public static bool ScheduleCacheUpdater(string cacheDir, string steamCmdDir, int autoUpdatePeriod)
{
var schedulerKey = $"ArkServerManager\\AutoUpdateSeverCache";
var rootSrcPath = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
var scriptPath = Path.Combine(rootSrcPath, "Lib", "ServerUpdater", "AutoUpdateServerCache.ps1");
var logPath = Path.Combine(cacheDir, "UpdateCache.log");
//
// Write the command to execute and copy mcrcon for updaters to use.
//
var cacheUpdateCmdPath = Path.Combine(cacheDir, "UpdateCache.cmd");
ScriptUtils.WriteCommandScript(cacheUpdateCmdPath, $"powershell -ExecutionPolicy Bypass -File \"{scriptPath}\" \"{cacheDir}\" \"{steamCmdDir}\" > \"{logPath}\"");
PlaceMCRcon(cacheDir);
//
// Schedule the task
//
var builder = new StringBuilder();
builder.AppendLine($"schtasks /Delete /TN {schedulerKey} /F");
if (autoUpdatePeriod != 0)
{
builder.AppendLine($"schTasks /Create /TN {schedulerKey} /TR \"'{cacheUpdateCmdPath}'\" /SC MINUTE /MO {autoUpdatePeriod} /NP /RL LIMITED ");
builder.AppendLine("IF ERRORLEVEL 1 EXIT 1");
}
builder.AppendLine("EXIT 0");
var script = builder.ToString();
return(ScriptUtils.RunElevatedShellScript(nameof(ScheduleCacheUpdater), script));
}
public static bool ScheduleAutoStart(string updateKey, bool enableAutoStart, string command, string args)
{
var schedulerKey = $"ArkServerManager\\AutoStart_{updateKey}";
var builder = new StringBuilder();
builder.AppendLine($"schtasks /Delete /TN {schedulerKey} /F");
if (enableAutoStart)
{
if (String.IsNullOrWhiteSpace(args))
{
builder.AppendLine($"schtasks /Create /TN {schedulerKey} /TR \"'{command}'\" /SC ONSTART /DELAY 0001:00 /NP /RL HIGHEST");
}
else
{
builder.AppendLine($"schtasks /Create /TN {schedulerKey} /TR \"'{command}' '{args}'\" /SC ONSTART /DELAY 0001:00 /NP /RL HIGHEST");
}
builder.AppendLine("IF ERRORLEVEL 1 EXIT 1");
}
builder.AppendLine("EXIT 0");
var script = builder.ToString();
bool result = ScriptUtils.RunElevatedShellScript(nameof(ScheduleAutoStart), script);
return(result);
}
public static bool EnsurePortsOpen(string exeName, int[] ports, string ruleName)
{
string rulesCommand = String.Format(@"
netsh advfirewall firewall delete rule all program = ""{0}""
netsh advfirewall firewall add rule name = ""{1} TCP"" action = allow program = ""{0}"" enable = yes localport={2} remoteport = any protocol = tcp dir = in
IF ERRORLEVEL 1 EXIT 1
netsh advfirewall firewall add rule name = ""{1} UDP"" action = allow program = ""{0}"" enable = yes localport={2} remoteport = any protocol = udp dir = in
IF ERRORLEVEL 1 EXIT 1
", exeName, ruleName, String.Join(",", ports));
return(ScriptUtils.RunElevatedShellScript(nameof(EnsurePortsOpen), rulesCommand));
}
public static bool ScheduleUpdates(string updateKey, int autoUpdatePeriod, string cacheDir, string installDir, string rconIP, int rconPort, string rconPass, int graceMinutes, TimeSpan?forceRestartTime)
{
var schedulerKey = $"ArkServerManager\\AutoUpgrade_{updateKey}";
var forceSchedulerKey = $"ArkServerManager\\AutoUpgrade_Force_{updateKey}";
var rootSrcPath = Path.GetDirectoryName(Assembly.GetExecutingAssembly().Location);
var scriptPath = Path.Combine(rootSrcPath, "Lib", "ServerUpdater", "AutoUpdateFromCache.ps1");
var serverUpdateCmdPath = Path.Combine(installDir, "ShooterGame", "Saved", "Config", "WindowsServer", "UpdateServerFromCache.cmd");
var logPath = Path.Combine(installDir, "ShooterGame", "Saved", "Config", "WindowsServer", "UpdateServerFromCache.log");
ScriptUtils.WriteCommandScript(serverUpdateCmdPath, $"powershell -ExecutionPolicy Bypass -File \"{scriptPath}\" \"{cacheDir}\" \"{installDir}\" \"{rconIP}\" \"{rconPort}\" \"{rconPass}\" \"{graceMinutes}\" > \"{logPath}\"");
string forceServerUpdateCmdPath = null;
if (forceRestartTime.HasValue)
{
forceServerUpdateCmdPath = Path.Combine(installDir, "ShooterGame", "Saved", "Config", "WindowsServer", "ForceUpdateServerFromCache.cmd");
var cmdBuilder = new StringBuilder();
cmdBuilder.AppendLine($"echo force_update > {(installDir + @"\ForceUpdate.txt").AsQuoted()} ");
cmdBuilder.AppendLine($"schTasks /Run /TN {schedulerKey}");
ScriptUtils.WriteCommandScript(forceServerUpdateCmdPath, cmdBuilder.ToString());
}
PlaceMCRcon(cacheDir);
var builder = new StringBuilder();
builder.AppendLine($"schtasks /Delete /TN {schedulerKey} /F");
builder.AppendLine($"schtasks /Delete /TN {forceSchedulerKey} /F");
if (autoUpdatePeriod != 0)
{
builder.AppendLine($"schTasks /Create /TN {schedulerKey} /TR \"'{serverUpdateCmdPath}'\" /SC MINUTE /MO {autoUpdatePeriod} /NP /RL LIMITED ");
builder.AppendLine("IF ERRORLEVEL 1 EXIT 1");
if (forceRestartTime.HasValue)
{
builder.AppendLine($"schTasks /Create /TN {forceSchedulerKey} /TR \"'{forceServerUpdateCmdPath}'\" /SC DAILY /ST {forceRestartTime.Value.Hours:D2}:{forceRestartTime.Value.Minutes:D2} /NP");
builder.AppendLine("IF ERRORLEVEL 1 EXIT 1");
}
builder.AppendLine($"schtasks /Run /TN {schedulerKey}");
builder.AppendLine("IF ERRORLEVEL 1 EXIT 1");
}
builder.AppendLine("EXIT 0");
var script = builder.ToString();
return(ScriptUtils.RunElevatedShellScript(nameof(ScheduleUpdates), script));
}
public static bool EnsurePortsOpen(string exeName, int[] ports, string ruleName)
{
StringBuilder rulesCommand = new StringBuilder();
rulesCommand.AppendFormat("netsh advfirewall firewall delete rule all program = \"{0}\"", exeName);
rulesCommand.AppendLine();
rulesCommand.AppendFormat("netsh advfirewall firewall add rule name = \"{0} TCP\" action = allow program = \"{1}\" enable = yes localport={2} remoteport = any protocol = tcp dir = in", ruleName, exeName, String.Join(",", ports));
rulesCommand.AppendLine();
rulesCommand.AppendFormat("IF ERRORLEVEL 1 EXIT 1");
rulesCommand.AppendLine();
rulesCommand.AppendFormat("netsh advfirewall firewall add rule name = \"{0} UDP\" action = allow program = \"{1}\" enable = yes localport={2} remoteport = any protocol = udp dir = in", ruleName, exeName, String.Join(",", ports));
rulesCommand.AppendLine();
rulesCommand.AppendFormat("IF ERRORLEVEL 1 EXIT 1");
rulesCommand.AppendLine();
return(ScriptUtils.RunElevatedShellScript(nameof(EnsurePortsOpen), rulesCommand.ToString()));
}
