416 lines
13 KiB
C#
416 lines
13 KiB
C#
#region
|
|
|
|
using System;
|
|
using System.Collections.Generic;
|
|
using System.Diagnostics;
|
|
using System.IO;
|
|
using System.Linq;
|
|
using System.Text;
|
|
using System.Threading;
|
|
using Microsoft.Diagnostics.Runtime;
|
|
using Microsoft.Extensions.Configuration;
|
|
using NLog;
|
|
using PropertyChanged;
|
|
using Sandbox;
|
|
using Sandbox.Engine.Multiplayer;
|
|
using Sandbox.Game.Multiplayer;
|
|
using Sandbox.Game.World;
|
|
using Torch.API;
|
|
using Torch.API.Managers;
|
|
using Torch.API.Session;
|
|
using Torch.Commands;
|
|
using Torch.Managers.PatchManager;
|
|
using Torch.Mod;
|
|
using Torch.Server.Commands;
|
|
using Torch.Server.Managers;
|
|
using Torch.Utils;
|
|
using VRage;
|
|
using Timer = System.Threading.Timer;
|
|
|
|
#endregion
|
|
|
|
#pragma warning disable 618
|
|
|
|
namespace Torch.Server
|
|
{
|
|
public class TorchServer : TorchBase, ITorchServer
|
|
{
|
|
private float _simRatio;
|
|
private Stopwatch _uptime;
|
|
private Timer _watchdog;
|
|
private MultiplayerManagerDedicated _multiplayerManagerDedicated;
|
|
|
|
internal bool FatalException { get; set; }
|
|
|
|
private System.Timers.Timer _simUpdateTimer = new System.Timers.Timer(200);
|
|
private bool _simDirty;
|
|
|
|
//Here to trigger rebuild
|
|
/// <inheritdoc />
|
|
public TorchServer(ITorchConfig config, string instancePath, string instanceName, IConfiguration configuration) : base(config)
|
|
{
|
|
Configuration = configuration;
|
|
InstancePath = instancePath;
|
|
InstanceName = instanceName;
|
|
DedicatedInstance = new InstanceManager(this);
|
|
AddManager(DedicatedInstance);
|
|
if (config.EntityManagerEnabled)
|
|
AddManager(new EntityControlManager(this));
|
|
AddManager(new RemoteAPIManager(this));
|
|
|
|
var sessionManager = Managers.GetManager<ITorchSessionManager>();
|
|
sessionManager.AddFactory(_ => new MultiplayerManagerDedicated(this));
|
|
sessionManager.SessionStateChanged += OnSessionStateChanged;
|
|
|
|
// Needs to be done at some point after MyVRageWindows.Init
|
|
// where the debug listeners are registered
|
|
if (!((TorchConfig)Config).EnableAsserts)
|
|
MyDebug.Listeners.Clear();
|
|
|
|
_simUpdateTimer.Elapsed += SimUpdateElapsed;
|
|
_simUpdateTimer.Start();
|
|
}
|
|
|
|
private void SimUpdateElapsed(object sender, System.Timers.ElapsedEventArgs e)
|
|
{
|
|
if (_simDirty)
|
|
{
|
|
OnPropertyChanged(nameof(SimulationRatio));
|
|
_simDirty = false;
|
|
}
|
|
}
|
|
|
|
public bool HasRun { get; set; }
|
|
|
|
|
|
/// <inheritdoc />
|
|
public float SimulationRatio
|
|
{
|
|
get => _simRatio;
|
|
set
|
|
{
|
|
if (_simRatio.IsEqual(value, 0.01f))
|
|
return;
|
|
|
|
_simRatio = value;
|
|
_simDirty = true;
|
|
//SetValue(ref _simRatio, value);
|
|
}
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public TimeSpan ElapsedPlayTime { get; set; }
|
|
|
|
/// <inheritdoc />
|
|
public Thread GameThread => MySandboxGame.Static?.UpdateThread;
|
|
|
|
/// <inheritdoc />
|
|
public bool IsRunning { get; set; }
|
|
|
|
public bool CanRun { get; set; }
|
|
|
|
/// <inheritdoc />
|
|
public InstanceManager DedicatedInstance { get; }
|
|
|
|
/// <inheritdoc />
|
|
protected override uint SteamAppId => 244850;
|
|
|
|
/// <inheritdoc />
|
|
protected override string SteamAppName => "SpaceEngineersDedicated";
|
|
|
|
/// <inheritdoc />
|
|
public ServerState State { get; private set; }
|
|
|
|
public event Action<ITorchServer> Initialized;
|
|
|
|
public int OnlinePlayers { get; private set; }
|
|
|
|
/// <inheritdoc />
|
|
public override void Init()
|
|
{
|
|
Log.Info("Initializing server");
|
|
base.Init();
|
|
GetManager<InstanceManager>().LoadInstance(InstancePath);
|
|
CanRun = true;
|
|
Initialized?.Invoke(this);
|
|
Log.Info($"Initialized server '{InstanceName}' at '{InstancePath}'");
|
|
}
|
|
|
|
public override IConfiguration Configuration { get; }
|
|
|
|
/// <inheritdoc />
|
|
public override void Start()
|
|
{
|
|
if (State != ServerState.Stopped)
|
|
return;
|
|
|
|
if (IsRunning || HasRun)
|
|
{
|
|
Restart();
|
|
return;
|
|
}
|
|
|
|
State = ServerState.Starting;
|
|
IsRunning = true;
|
|
HasRun = true;
|
|
CanRun = false;
|
|
PatchManager.CommitInternal();
|
|
Log.Info("Starting server.");
|
|
MySandboxGame.ConfigDedicated = DedicatedInstance.DedicatedConfig.Model;
|
|
|
|
_uptime = Stopwatch.StartNew();
|
|
base.Start();
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public override void Stop()
|
|
{
|
|
if (State == ServerState.Stopped)
|
|
Log.Error("Server is already stopped");
|
|
if (Thread.CurrentThread == GameThread)
|
|
new Thread(StopInternal)
|
|
{
|
|
Name = "Stopping Thread"
|
|
}.Start();
|
|
else
|
|
StopInternal();
|
|
}
|
|
|
|
private void StopInternal()
|
|
{
|
|
Log.Info("Stopping server.");
|
|
base.Stop();
|
|
Log.Info("Server stopped.");
|
|
|
|
State = ServerState.Stopped;
|
|
IsRunning = false;
|
|
CanRun = true;
|
|
SimulationRatio = 0;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Restart the program.
|
|
/// </summary>
|
|
public override void Restart(bool save = true)
|
|
{
|
|
if (Config.DisconnectOnRestart)
|
|
{
|
|
foreach (var member in MyMultiplayer.Static.Members)
|
|
{
|
|
MyMultiplayer.Static.DisconnectClient(member);
|
|
}
|
|
|
|
Log.Info("Ejected all players from server for restart.");
|
|
}
|
|
|
|
new Thread(() =>
|
|
{
|
|
StopInternal();
|
|
LogManager.Flush();
|
|
|
|
#if DEBUG
|
|
Environment.Exit(0);
|
|
#endif
|
|
|
|
var exe = Path.Combine(AppContext.BaseDirectory, "Torch.Server.exe");
|
|
|
|
var args = Environment.GetCommandLineArgs();
|
|
|
|
for (var i = 0; i < args.Length; i++)
|
|
{
|
|
if (args[i].Contains(' '))
|
|
args[i] = $"\"{args[i]}\"";
|
|
|
|
if (!args[i].Contains("--tempAutostart", StringComparison.InvariantCultureIgnoreCase) &&
|
|
!args[i].Contains("--waitForPid", StringComparison.InvariantCultureIgnoreCase))
|
|
continue;
|
|
|
|
args[i] = string.Empty;
|
|
args[++i] = string.Empty;
|
|
}
|
|
|
|
Process.Start(exe, $"--waitForPid {Environment.ProcessId} --tempAutostart true {string.Join(" ", args)}");
|
|
|
|
Environment.Exit(0);
|
|
})
|
|
{
|
|
Name = "Restart thread"
|
|
}.Start();
|
|
}
|
|
|
|
[SuppressPropertyChangedWarnings]
|
|
private void OnSessionStateChanged(ITorchSession session, TorchSessionState newState)
|
|
{
|
|
switch (newState)
|
|
{
|
|
case TorchSessionState.Unloading:
|
|
_watchdog?.Dispose();
|
|
_watchdog = null;
|
|
break;
|
|
case TorchSessionState.Loaded:
|
|
_multiplayerManagerDedicated = CurrentSession.Managers.GetManager<MultiplayerManagerDedicated>();
|
|
_multiplayerManagerDedicated.PlayerJoined += MultiplayerManagerDedicatedOnPlayerJoined;
|
|
_multiplayerManagerDedicated.PlayerLeft += MultiplayerManagerDedicatedOnPlayerLeft;
|
|
CurrentSession.Managers.GetManager<CommandManager>().RegisterCommandModule(typeof(WhitelistCommands));
|
|
break;
|
|
case TorchSessionState.Loading:
|
|
case TorchSessionState.Unloaded:
|
|
break;
|
|
default:
|
|
throw new ArgumentOutOfRangeException(nameof(newState), newState, null);
|
|
}
|
|
|
|
}
|
|
|
|
private void MultiplayerManagerDedicatedOnPlayerLeft(IPlayer player)
|
|
{
|
|
OnlinePlayers--;
|
|
}
|
|
|
|
private void MultiplayerManagerDedicatedOnPlayerJoined(IPlayer player)
|
|
{
|
|
OnlinePlayers++;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public override void Init(object gameInstance)
|
|
{
|
|
base.Init(gameInstance);
|
|
if (gameInstance is MySandboxGame && MySession.Static != null)
|
|
State = ServerState.Running;
|
|
else
|
|
State = ServerState.Stopped;
|
|
}
|
|
|
|
/// <inheritdoc />
|
|
public override void Update()
|
|
{
|
|
base.Update();
|
|
// Stops 1.00-1.02 flicker.
|
|
SimulationRatio = Math.Min(Sync.ServerSimulationRatio, 1);
|
|
var elapsed = TimeSpan.FromSeconds(Math.Floor(_uptime.Elapsed.TotalSeconds));
|
|
ElapsedPlayTime = elapsed;
|
|
|
|
if (_watchdog == null && Config.TickTimeout > 0)
|
|
{
|
|
Log.Info("Starting server watchdog.");
|
|
_watchdog = new Timer(CheckServerResponding, this, TimeSpan.Zero,
|
|
TimeSpan.FromSeconds(Config.TickTimeout));
|
|
}
|
|
}
|
|
|
|
#region Freeze Detection
|
|
|
|
private static void CheckServerResponding(object state)
|
|
{
|
|
var server = (TorchServer)state;
|
|
var mre = new ManualResetEvent(false);
|
|
server.Invoke(() => mre.Set());
|
|
if (!mre.WaitOne(TimeSpan.FromSeconds(Instance.Config.TickTimeout)))
|
|
{
|
|
if (server.FatalException)
|
|
{
|
|
server._watchdog.Dispose();
|
|
return;
|
|
}
|
|
#if DEBUG
|
|
Log.Error(
|
|
$"Server watchdog detected that the server was frozen for at least {((TorchServer) state).Config.TickTimeout} seconds.");
|
|
Log.Error(DumpFrozenThread(MySandboxGame.Static.UpdateThread));
|
|
#else
|
|
Log.Error(DumpFrozenThread(MySandboxGame.Static.UpdateThread));
|
|
throw new TimeoutException($"Server watchdog detected that the server was frozen for at least {((TorchServer)state).Config.TickTimeout} seconds.");
|
|
#endif
|
|
}
|
|
|
|
Log.Debug("Server watchdog responded");
|
|
}
|
|
|
|
private static string DumpFrozenThread(Thread thread, int traces = 3, int pause = 5000)
|
|
{
|
|
var stacks = new List<string>(traces);
|
|
var totalSize = 0;
|
|
for (var i = 0; i < traces; i++)
|
|
{
|
|
string dump = DumpStack(thread);
|
|
totalSize += dump.Length;
|
|
stacks.Add(dump);
|
|
Thread.Sleep(pause);
|
|
}
|
|
|
|
string commonPrefix = StringUtils.CommonSuffix(stacks);
|
|
// Advance prefix to include the line terminator.
|
|
commonPrefix = commonPrefix.Substring(commonPrefix.IndexOf('\n') + 1);
|
|
|
|
var result = new StringBuilder(totalSize - (stacks.Count - 1) * commonPrefix.Length);
|
|
result.AppendLine($"Frozen thread dump {thread.Name}");
|
|
result.AppendLine("Common prefix:").AppendLine(commonPrefix);
|
|
for (var i = 0; i < stacks.Count; i++)
|
|
if (stacks[i].Length > commonPrefix.Length)
|
|
{
|
|
result.AppendLine($"Suffix {i}");
|
|
result.AppendLine(stacks[i].Substring(0, stacks[i].Length - commonPrefix.Length));
|
|
}
|
|
|
|
return result.ToString();
|
|
}
|
|
|
|
private static string DumpStack(Thread thread)
|
|
{
|
|
// Deprecated in .NET Core and later
|
|
// try
|
|
// {
|
|
// thread.Suspend();
|
|
// }
|
|
// catch
|
|
// {
|
|
// // ignored
|
|
// }
|
|
//
|
|
// var stack = new StackTrace(thread, true);
|
|
// try
|
|
// {
|
|
// thread.Resume();
|
|
// }
|
|
// catch
|
|
// {
|
|
// // ignored
|
|
// }
|
|
//
|
|
// return stack.ToString();
|
|
|
|
// Modified from https://www.examplefiles.net/cs/579311
|
|
using var target = DataTarget.CreateSnapshotAndAttach(Environment.ProcessId);
|
|
|
|
var runtime = target.ClrVersions[0].CreateRuntime();
|
|
|
|
var clrThread = runtime.Threads.First(b => b.ManagedThreadId == thread.ManagedThreadId);
|
|
|
|
var sb = new StringBuilder();
|
|
|
|
foreach (var frame in clrThread.EnumerateStackTrace())
|
|
{
|
|
sb.Append('\t');
|
|
switch (frame.Kind)
|
|
{
|
|
case ClrStackFrameKind.Unknown:
|
|
sb.AppendLine("[Unknown]");
|
|
break;
|
|
case ClrStackFrameKind.ManagedMethod:
|
|
sb.AppendLine(frame.Method?.Signature ?? "[Unable to get method signature]");
|
|
break;
|
|
case ClrStackFrameKind.Runtime:
|
|
sb.AppendLine("[CLR Runtime]");
|
|
break;
|
|
default:
|
|
throw new ArgumentOutOfRangeException(nameof(frame.Kind), frame.Kind, "Incorrect value in EnumerateStackTrace");
|
|
}
|
|
}
|
|
|
|
return sb.ToString();
|
|
}
|
|
|
|
#endregion
|
|
}
|
|
}
|