mirror of
				https://github.com/chylex/Minecraft-Phantom-Panel.git
				synced 2025-10-31 11:17:15 +01:00 
			
		
		
		
	Compare commits
	
		
			37 Commits
		
	
	
		
			wip-launch
			...
			15d45fe1a3
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 15d45fe1a3 | |||
| 0b8a870d10 | |||
| 16888c9b10 | |||
| bfefb9063b | |||
| cd332a6571 | |||
| 2a9bb9e6ac | |||
| 55b853d227 | |||
| 627e7436fd | |||
| 339b958e45 | |||
| 3a17eee8d0 | |||
| 2c623cbd9a | |||
| 346eab0b1b | |||
| 956f1e779b | |||
| 2e8238a3bc | |||
| 346ca5de75 | |||
| 3e93ad9217 | |||
| 07c31ac70b | |||
| c89e30cd3f | |||
| 0091fa11fb | |||
| 20ff2871fa | |||
| f211708509 | |||
| ad7964677e | |||
| 626f141a2c | |||
| 46dba1a4fa | |||
| 476de00cfd | |||
| 2180e0c067 | |||
| a9646c5bbc | |||
| 01d6648b1f | |||
| def6c41a77 | |||
| 3f976295bd | |||
| 9d9734d1fd | |||
| 267c5ad921 | |||
| f5b40a92e2 | |||
| 2c413160f6 | |||
| 0af14c3262 | |||
| 0ab165fd21 | |||
| 6f11f65d91 | 
| @@ -1,6 +1,6 @@ | ||||
| <component name="ProjectRunConfigurationManager"> | ||||
|   <configuration default="false" name="Agent 1" type="DotNetProject" factoryName=".NET Project"> | ||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/Agent/Phantom.Agent/bin/Debug/net7.0/Phantom.Agent.exe" /> | ||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Agent/debug/Phantom.Agent.exe" /> | ||||
|     <option name="PROGRAM_PARAMETERS" value="" /> | ||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Agent1" /> | ||||
|     <option name="PASS_PARENT_ENVS" value="1" /> | ||||
| @@ -9,10 +9,10 @@ | ||||
|       <env name="AGENT_NAME" value="Agent 1" /> | ||||
|       <env name="ALLOWED_RCON_PORTS" value="25575,27000,27001" /> | ||||
|       <env name="ALLOWED_SERVER_PORTS" value="25565,26000,26001" /> | ||||
|       <env name="CONTROLLER_HOST" value="localhost" /> | ||||
|       <env name="JAVA_SEARCH_PATH" value="~/.jdks" /> | ||||
|       <env name="MAX_INSTANCES" value="3" /> | ||||
|       <env name="MAX_MEMORY" value="12G" /> | ||||
|       <env name="SERVER_HOST" value="localhost" /> | ||||
|     </envs> | ||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> | ||||
|     <option name="USE_MONO" value="0" /> | ||||
| @@ -22,7 +22,7 @@ | ||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> | ||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> | ||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> | ||||
|     <option name="PROJECT_TFM" value="net7.0" /> | ||||
|     <option name="PROJECT_TFM" value="net8.0" /> | ||||
|     <method v="2"> | ||||
|       <option name="Build" /> | ||||
|     </method> | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| <component name="ProjectRunConfigurationManager"> | ||||
|   <configuration default="false" name="Agent 2" type="DotNetProject" factoryName=".NET Project"> | ||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/Agent/Phantom.Agent/bin/Debug/net7.0/Phantom.Agent.exe" /> | ||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Agent/debug/Phantom.Agent.exe" /> | ||||
|     <option name="PROGRAM_PARAMETERS" value="" /> | ||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Agent2" /> | ||||
|     <option name="PASS_PARENT_ENVS" value="1" /> | ||||
| @@ -9,10 +9,10 @@ | ||||
|       <env name="AGENT_NAME" value="Agent 2" /> | ||||
|       <env name="ALLOWED_RCON_PORTS" value="27002-27006" /> | ||||
|       <env name="ALLOWED_SERVER_PORTS" value="26002-26006" /> | ||||
|       <env name="CONTROLLER_HOST" value="localhost" /> | ||||
|       <env name="JAVA_SEARCH_PATH" value="~/.jdks" /> | ||||
|       <env name="MAX_INSTANCES" value="5" /> | ||||
|       <env name="MAX_MEMORY" value="10G" /> | ||||
|       <env name="SERVER_HOST" value="localhost" /> | ||||
|     </envs> | ||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> | ||||
|     <option name="USE_MONO" value="0" /> | ||||
| @@ -22,7 +22,7 @@ | ||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> | ||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> | ||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> | ||||
|     <option name="PROJECT_TFM" value="net7.0" /> | ||||
|     <option name="PROJECT_TFM" value="net8.0" /> | ||||
|     <method v="2"> | ||||
|       <option name="Build" /> | ||||
|     </method> | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| <component name="ProjectRunConfigurationManager"> | ||||
|   <configuration default="false" name="Agent 3" type="DotNetProject" factoryName=".NET Project"> | ||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/Agent/Phantom.Agent/bin/Debug/net7.0/Phantom.Agent.exe" /> | ||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Agent/debug/Phantom.Agent.exe" /> | ||||
|     <option name="PROGRAM_PARAMETERS" value="" /> | ||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Agent3" /> | ||||
|     <option name="PASS_PARENT_ENVS" value="1" /> | ||||
| @@ -9,10 +9,10 @@ | ||||
|       <env name="AGENT_NAME" value="Agent 3" /> | ||||
|       <env name="ALLOWED_RCON_PORTS" value="27007" /> | ||||
|       <env name="ALLOWED_SERVER_PORTS" value="26007" /> | ||||
|       <env name="CONTROLLER_HOST" value="localhost" /> | ||||
|       <env name="JAVA_SEARCH_PATH" value="~/.jdks" /> | ||||
|       <env name="MAX_INSTANCES" value="1" /> | ||||
|       <env name="MAX_MEMORY" value="2560M" /> | ||||
|       <env name="SERVER_HOST" value="localhost" /> | ||||
|     </envs> | ||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> | ||||
|     <option name="USE_MONO" value="0" /> | ||||
| @@ -22,7 +22,7 @@ | ||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> | ||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> | ||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> | ||||
|     <option name="PROJECT_TFM" value="net7.0" /> | ||||
|     <option name="PROJECT_TFM" value="net8.0" /> | ||||
|     <method v="2"> | ||||
|       <option name="Build" /> | ||||
|     </method> | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| <component name="ProjectRunConfigurationManager"> | ||||
|   <configuration default="false" name="Compile Bootstrap" type="js.build_tools.npm"> | ||||
|     <package-json value="$PROJECT_DIR$/Server/Phantom.Server.Web.Bootstrap/package.json" /> | ||||
|     <package-json value="$PROJECT_DIR$/Web/Phantom.Web.Bootstrap/package.json" /> | ||||
|     <command value="run" /> | ||||
|     <scripts> | ||||
|       <script value="compile" /> | ||||
| @@ -9,4 +9,4 @@ | ||||
|     <envs /> | ||||
|     <method v="2" /> | ||||
|   </configuration> | ||||
| </component> | ||||
| </component> | ||||
| @@ -1,9 +1,10 @@ | ||||
| <component name="ProjectRunConfigurationManager"> | ||||
|   <configuration default="false" name="Server + Agent x3" type="CompoundRunConfigurationType"> | ||||
|   <configuration default="false" name="Controller + Web + Agent x3" type="CompoundRunConfigurationType"> | ||||
|     <toRun name="Agent 1" type="DotNetProject" /> | ||||
|     <toRun name="Agent 2" type="DotNetProject" /> | ||||
|     <toRun name="Agent 3" type="DotNetProject" /> | ||||
|     <toRun name="Server" type="DotNetProject" /> | ||||
|     <toRun name="Controller" type="DotNetProject" /> | ||||
|     <toRun name="Web" type="DotNetProject" /> | ||||
|     <method v="2" /> | ||||
|   </configuration> | ||||
| </component> | ||||
| </component> | ||||
							
								
								
									
										8
									
								
								.run/Controller + Web + Agent.run.xml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										8
									
								
								.run/Controller + Web + Agent.run.xml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,8 @@ | ||||
| <component name="ProjectRunConfigurationManager"> | ||||
|   <configuration default="false" name="Controller + Web + Agent" type="CompoundRunConfigurationType"> | ||||
|     <toRun name="Agent 1" type="DotNetProject" /> | ||||
|     <toRun name="Controller" type="DotNetProject" /> | ||||
|     <toRun name="Web" type="DotNetProject" /> | ||||
|     <method v="2" /> | ||||
|   </configuration> | ||||
| </component> | ||||
| @@ -1,30 +1,29 @@ | ||||
| <component name="ProjectRunConfigurationManager"> | ||||
|   <configuration default="false" name="Server" type="DotNetProject" factoryName=".NET Project"> | ||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/Server/Phantom.Server/bin/Debug/net7.0/Phantom.Server.exe" /> | ||||
|   <configuration default="false" name="Controller" type="DotNetProject" factoryName=".NET Project"> | ||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Controller/debug/Phantom.Controller.exe" /> | ||||
|     <option name="PROGRAM_PARAMETERS" value="" /> | ||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Server" /> | ||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Controller" /> | ||||
|     <option name="PASS_PARENT_ENVS" value="1" /> | ||||
|     <envs> | ||||
|       <env name="ASPNETCORE_ENVIRONMENT" value="Development" /> | ||||
|       <env name="AGENT_RPC_SERVER_HOST" value="localhost" /> | ||||
|       <env name="PG_DATABASE" value="postgres" /> | ||||
|       <env name="PG_HOST" value="localhost" /> | ||||
|       <env name="PG_PASS" value="development" /> | ||||
|       <env name="PG_PORT" value="9402" /> | ||||
|       <env name="PG_PORT" value="9403" /> | ||||
|       <env name="PG_USER" value="postgres" /> | ||||
|       <env name="RPC_SERVER_HOST" value="localhost" /> | ||||
|       <env name="WEB_SERVER_HOST" value="localhost" /> | ||||
|       <env name="WEB_RPC_SERVER_HOST" value="localhost" /> | ||||
|     </envs> | ||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> | ||||
|     <option name="USE_MONO" value="0" /> | ||||
|     <option name="RUNTIME_ARGUMENTS" value="" /> | ||||
|     <option name="PROJECT_PATH" value="$PROJECT_DIR$/Server/Phantom.Server/Phantom.Server.csproj" /> | ||||
|     <option name="PROJECT_PATH" value="$PROJECT_DIR$/Controller/Phantom.Controller/Phantom.Controller.csproj" /> | ||||
|     <option name="PROJECT_EXE_PATH_TRACKING" value="1" /> | ||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> | ||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> | ||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> | ||||
|     <option name="PROJECT_TFM" value="net7.0" /> | ||||
|     <option name="PROJECT_TFM" value="net8.0" /> | ||||
|     <method v="2"> | ||||
|       <option name="Build" /> | ||||
|     </method> | ||||
|   </configuration> | ||||
| </component> | ||||
| </component> | ||||
| @@ -1,7 +0,0 @@ | ||||
| <component name="ProjectRunConfigurationManager"> | ||||
|   <configuration default="false" name="Server + Agent" type="CompoundRunConfigurationType"> | ||||
|     <toRun name="Agent 1" type="DotNetProject" /> | ||||
|     <toRun name="Server" type="DotNetProject" /> | ||||
|     <method v="2" /> | ||||
|   </configuration> | ||||
| </component> | ||||
							
								
								
									
										26
									
								
								.run/Web.run.xml
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										26
									
								
								.run/Web.run.xml
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,26 @@ | ||||
| <component name="ProjectRunConfigurationManager"> | ||||
|   <configuration default="false" name="Web" type="DotNetProject" factoryName=".NET Project"> | ||||
|     <option name="EXE_PATH" value="$PROJECT_DIR$/.artifacts/bin/Phantom.Web/debug/Phantom.Web.exe" /> | ||||
|     <option name="PROGRAM_PARAMETERS" value="" /> | ||||
|     <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$/.workdir/Web" /> | ||||
|     <option name="PASS_PARENT_ENVS" value="1" /> | ||||
|     <envs> | ||||
|       <env name="ASPNETCORE_ENVIRONMENT" value="Development" /> | ||||
|       <env name="CONTROLLER_HOST" value="localhost" /> | ||||
|       <env name="WEB_KEY" value="BMNHM9RRPMCBBY29D9XHS6KBKZSRY7F5XFN27YZX96XXWJC2NM2D6YRHM9PZN9JGQGCSJ6FMB2GGZ" /> | ||||
|       <env name="WEB_SERVER_HOST" value="localhost" /> | ||||
|     </envs> | ||||
|     <option name="USE_EXTERNAL_CONSOLE" value="0" /> | ||||
|     <option name="USE_MONO" value="0" /> | ||||
|     <option name="RUNTIME_ARGUMENTS" value="" /> | ||||
|     <option name="PROJECT_PATH" value="$PROJECT_DIR$/Web/Phantom.Web/Phantom.Web.csproj" /> | ||||
|     <option name="PROJECT_EXE_PATH_TRACKING" value="1" /> | ||||
|     <option name="PROJECT_ARGUMENTS_TRACKING" value="1" /> | ||||
|     <option name="PROJECT_WORKING_DIRECTORY_TRACKING" value="0" /> | ||||
|     <option name="PROJECT_KIND" value="DotNetCore" /> | ||||
|     <option name="PROJECT_TFM" value="net8.0" /> | ||||
|     <method v="2"> | ||||
|       <option name="Build" /> | ||||
|     </method> | ||||
|   </configuration> | ||||
| </component> | ||||
							
								
								
									
										1
									
								
								.workdir/Controller/.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								.workdir/Controller/.gitignore
									
									
									
									
										vendored
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | ||||
|  | ||||
							
								
								
									
										2
									
								
								.workdir/Controller/secrets/web.key
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										2
									
								
								.workdir/Controller/secrets/web.key
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,2 @@ | ||||
| <EFBFBD><EFBFBD>h?Ο<05>Bx | ||||
| <02> | ||||
							
								
								
									
										1
									
								
								.workdir/Controller/secrets/web.secret
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										1
									
								
								.workdir/Controller/secrets/web.secret
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1 @@ | ||||
| T<EFBFBD>./g<11><>N<EFBFBD><4E>t<EFBFBD>$<24>!<21>(<28><>#<23>~<7E><>}<14><: | ||||
| @@ -5,4 +5,4 @@ if [%1]==[] ( | ||||
|     exit | ||||
| ) | ||||
|  | ||||
| dotnet ef migrations add %~1 --project Server/Phantom.Server.Database.Postgres | ||||
| dotnet ef migrations add %~1 --project Controller/Phantom.Controller.Database.Postgres --msbuildprojectextensionspath .artifacts/obj/Phantom.Controller.Database.Postgres | ||||
|   | ||||
| @@ -3,4 +3,4 @@ if [ -z "$1" ]; then | ||||
|     exit 1 | ||||
| fi | ||||
|  | ||||
| dotnet ef migrations add "$1" --project Server/Phantom.Server.Database.Postgres | ||||
| dotnet ef migrations add "$1" --project Controller/Phantom.Controller.Database.Postgres --msbuildprojectextensionspath .artifacts/obj/Phantom.Controller.Database.Postgres | ||||
|   | ||||
| @@ -1,4 +1,4 @@ | ||||
| namespace Phantom.Agent.Minecraft.Command;  | ||||
| namespace Phantom.Agent.Minecraft.Command; | ||||
|  | ||||
| public static class MinecraftCommand { | ||||
| 	public const string SaveOn = "save-on"; | ||||
|   | ||||
| @@ -1,13 +1,12 @@ | ||||
| using Phantom.Utils.Collections; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Utils.Processes; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Instance;  | ||||
| namespace Phantom.Agent.Minecraft.Instance; | ||||
|  | ||||
| public sealed class InstanceProcess : IDisposable { | ||||
| 	public InstanceProperties InstanceProperties { get; } | ||||
| 	public CancellableSemaphore BackupSemaphore { get; } = new (1, 1); | ||||
| 	 | ||||
| 	private readonly RingBuffer<string> outputBuffer = new (10000); | ||||
| 	private readonly RingBuffer<string> outputBuffer = new (100); | ||||
| 	private event EventHandler<string>? OutputEvent; | ||||
|  | ||||
| 	public event EventHandler? Ended; | ||||
| @@ -61,7 +60,6 @@ public sealed class InstanceProcess : IDisposable { | ||||
|  | ||||
| 	public void Dispose() { | ||||
| 		process.Dispose(); | ||||
| 		BackupSemaphore.Dispose(); | ||||
| 		OutputEvent = null; | ||||
| 		Ended = null; | ||||
| 	} | ||||
|   | ||||
| @@ -0,0 +1,92 @@ | ||||
| using System.Text; | ||||
| using Kajabity.Tools.Java; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Java; | ||||
|  | ||||
| sealed class JavaPropertiesFileEditor { | ||||
| 	private static readonly Encoding Encoding = Encoding.GetEncoding("ISO-8859-1"); | ||||
|  | ||||
| 	private readonly Dictionary<string, string> overriddenProperties = new (); | ||||
|  | ||||
| 	public void Set(string key, string value) { | ||||
| 		overriddenProperties[key] = value; | ||||
| 	} | ||||
|  | ||||
| 	public async Task EditOrCreate(string filePath) { | ||||
| 		if (File.Exists(filePath)) { | ||||
| 			string tmpFilePath = filePath + ".tmp"; | ||||
| 			File.Copy(filePath, tmpFilePath, overwrite: true); | ||||
| 			await EditFromCopyOrCreate(filePath, tmpFilePath); | ||||
| 			File.Move(tmpFilePath, filePath, overwrite: true); | ||||
| 		} | ||||
| 		else { | ||||
| 			await EditFromCopyOrCreate(null, filePath); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private async Task EditFromCopyOrCreate(string? sourceFilePath, string targetFilePath) { | ||||
| 		var properties = new JavaProperties(); | ||||
|  | ||||
| 		if (sourceFilePath != null) { | ||||
| 			// TODO replace with custom async parser | ||||
| 			await using var sourceStream = new FileStream(sourceFilePath, FileMode.Open, FileAccess.Read, FileShare.Read); | ||||
| 			properties.Load(sourceStream, Encoding); | ||||
| 		} | ||||
|  | ||||
| 		foreach (var (key, value) in overriddenProperties) { | ||||
| 			properties[key] = value; | ||||
| 		} | ||||
|  | ||||
| 		await using var targetStream = new FileStream(targetFilePath, FileMode.Create, FileAccess.Write, FileShare.Read); | ||||
| 		await using var targetWriter = new StreamWriter(targetStream, Encoding); | ||||
|  | ||||
| 		await targetWriter.WriteLineAsync("# Properties"); | ||||
|  | ||||
| 		foreach (var (key, value) in properties) { | ||||
| 			await WriteProperty(targetWriter, key, value); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private static async Task WriteProperty(StreamWriter writer, string key, string value) { | ||||
| 		await WritePropertyComponent(writer, key, escapeSpaces: true); | ||||
| 		await writer.WriteAsync('='); | ||||
| 		await WritePropertyComponent(writer, value, escapeSpaces: false); | ||||
| 		await writer.WriteLineAsync(); | ||||
| 	} | ||||
|  | ||||
| 	private static async Task WritePropertyComponent(TextWriter writer, string component, bool escapeSpaces) { | ||||
| 		for (int index = 0; index < component.Length; index++) { | ||||
| 			var c = component[index]; | ||||
| 			switch (c) { | ||||
| 				case '\\': | ||||
| 				case '#': | ||||
| 				case '!': | ||||
| 				case '=': | ||||
| 				case ':': | ||||
| 				case ' ' when escapeSpaces || index == 0: | ||||
| 					await writer.WriteAsync('\\'); | ||||
| 					await writer.WriteAsync(c); | ||||
| 					break; | ||||
| 				case var _ when c > 31 && c < 127: | ||||
| 					await writer.WriteAsync(c); | ||||
| 					break; | ||||
| 				case '\t': | ||||
| 					await writer.WriteAsync("\\t"); | ||||
| 					break; | ||||
| 				case '\n': | ||||
| 					await writer.WriteAsync("\\n"); | ||||
| 					break; | ||||
| 				case '\r': | ||||
| 					await writer.WriteAsync("\\r"); | ||||
| 					break; | ||||
| 				case '\f': | ||||
| 					await writer.WriteAsync("\\f"); | ||||
| 					break; | ||||
| 				default: | ||||
| 					await writer.WriteAsync("\\u"); | ||||
| 					await writer.WriteAsync(((int) c).ToString("X4")); | ||||
| 					break; | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -79,10 +79,9 @@ public sealed class JavaRuntimeDiscovery { | ||||
| 			WorkingDirectory = Path.GetDirectoryName(javaExecutablePath), | ||||
| 			Arguments = "-XshowSettings:properties -version", | ||||
| 			RedirectStandardInput = false, | ||||
| 			RedirectStandardOutput = true, | ||||
| 			RedirectStandardOutput = false, | ||||
| 			RedirectStandardError = true, | ||||
| 			UseShellExecute = false, | ||||
| 			CreateNoWindow = false | ||||
| 			UseShellExecute = false | ||||
| 		}; | ||||
|  | ||||
| 		var process = new Process { StartInfo = startInfo }; | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| using Phantom.Common.Data.Java; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Java;  | ||||
| namespace Phantom.Agent.Minecraft.Java; | ||||
|  | ||||
| public sealed class JavaRuntimeExecutable { | ||||
| 	internal string ExecutablePath { get; } | ||||
|   | ||||
| @@ -1,5 +1,4 @@ | ||||
| using System.Collections.Immutable; | ||||
| using System.Collections.ObjectModel; | ||||
| using System.Collections.ObjectModel; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Java; | ||||
|  | ||||
| @@ -7,12 +6,12 @@ sealed class JvmArgumentBuilder { | ||||
| 	private readonly JvmProperties basicProperties; | ||||
| 	private readonly List<string> customArguments = new (); | ||||
|  | ||||
| 	public JvmArgumentBuilder(JvmProperties basicProperties, ImmutableArray<string> customArguments) { | ||||
| 	public JvmArgumentBuilder(JvmProperties basicProperties) { | ||||
| 		this.basicProperties = basicProperties; | ||||
| 	} | ||||
|  | ||||
| 		foreach (var jvmArgument in customArguments) { | ||||
| 			this.customArguments.Add(jvmArgument); | ||||
| 		} | ||||
| 	public void Add(string argument) { | ||||
| 		customArguments.Add(argument); | ||||
| 	} | ||||
|  | ||||
| 	public void AddProperty(string key, string value) { | ||||
| @@ -24,8 +23,8 @@ sealed class JvmArgumentBuilder { | ||||
| 			target.Add(property); | ||||
| 		} | ||||
|  | ||||
| 		// In case of duplicate JVM arguments, typically the last one wins. | ||||
| 		target.Add("-Xms" + basicProperties.InitialHeapMegabytes + "M"); | ||||
| 		target.Add("-Xmx" + basicProperties.MaximumHeapMegabytes + "M"); | ||||
| 		target.Add("-Xrs"); | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,10 +1,8 @@ | ||||
| using System.Text; | ||||
| using Kajabity.Tools.Java; | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Agent.Minecraft.Java; | ||||
| using Phantom.Agent.Minecraft.Server; | ||||
| using Phantom.Common.Minecraft; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Utils.Processes; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Launcher; | ||||
| @@ -23,10 +21,6 @@ public abstract class BaseLauncher : IServerLauncher { | ||||
| 			return new LaunchResult.InvalidJavaRuntime(); | ||||
| 		} | ||||
|  | ||||
| 		if (JvmArgumentsHelper.Validate(instanceProperties.JvmArguments) != null) { | ||||
| 			return new LaunchResult.InvalidJvmArguments(); | ||||
| 		} | ||||
|  | ||||
| 		var vanillaServerJarPath = await services.ServerExecutables.DownloadAndGetPath(instanceProperties.LaunchProperties.ServerDownloadInfo, MinecraftVersion, downloadProgressEventHandler, cancellationToken); | ||||
| 		if (vanillaServerJarPath == null) { | ||||
| 			return new LaunchResult.CouldNotDownloadMinecraftServer(); | ||||
| @@ -62,16 +56,8 @@ public abstract class BaseLauncher : IServerLauncher { | ||||
| 			UseShellExecute = false | ||||
| 		}; | ||||
| 		 | ||||
| 		var jvmArguments = new JvmArgumentBuilder(instanceProperties.JvmProperties, instanceProperties.JvmArguments); | ||||
| 		CustomizeJvmArguments(jvmArguments); | ||||
|  | ||||
| 		var processArguments = processConfigurator.ArgumentList; | ||||
| 		jvmArguments.Build(processArguments); | ||||
| 		 | ||||
| 		foreach (var extraArgument in serverJar.ExtraArgs) { | ||||
| 			processArguments.Add(extraArgument); | ||||
| 		} | ||||
| 		 | ||||
| 		PrepareJvmArguments(serverJar).Build(processArguments); | ||||
| 		processArguments.Add("-jar"); | ||||
| 		processArguments.Add(serverJar.FilePath); | ||||
| 		processArguments.Add("nogui"); | ||||
| @@ -96,6 +82,21 @@ public abstract class BaseLauncher : IServerLauncher { | ||||
| 		return new LaunchResult.Success(instanceProcess); | ||||
| 	} | ||||
|  | ||||
| 	private JvmArgumentBuilder PrepareJvmArguments(ServerJarInfo serverJar) { | ||||
| 		var builder = new JvmArgumentBuilder(instanceProperties.JvmProperties); | ||||
|  | ||||
| 		foreach (string argument in instanceProperties.JvmArguments) { | ||||
| 			builder.Add(argument); | ||||
| 		} | ||||
|  | ||||
| 		foreach (var argument in serverJar.ExtraArgs) { | ||||
| 			builder.Add(argument); | ||||
| 		} | ||||
|  | ||||
| 		CustomizeJvmArguments(builder); | ||||
| 		return builder; | ||||
| 	} | ||||
|  | ||||
| 	private protected virtual void CustomizeJvmArguments(JvmArgumentBuilder arguments) {} | ||||
|  | ||||
| 	private protected virtual Task<ServerJarInfo> PrepareServerJar(ILogger logger, string serverJarPath, CancellationToken cancellationToken) { | ||||
| @@ -108,21 +109,8 @@ public abstract class BaseLauncher : IServerLauncher { | ||||
| 	} | ||||
|  | ||||
| 	private static async Task UpdateServerProperties(InstanceProperties instanceProperties) { | ||||
| 		var serverPropertiesFilePath = Path.Combine(instanceProperties.InstanceFolder, "server.properties"); | ||||
| 		var serverPropertiesData = new JavaProperties(); | ||||
|  | ||||
| 		await using var fileStream = new FileStream(serverPropertiesFilePath, FileMode.OpenOrCreate, FileAccess.ReadWrite, FileShare.None); | ||||
| 		try { | ||||
| 			serverPropertiesData.Load(fileStream); | ||||
| 		} catch (ParseException e) { | ||||
| 			throw new Exception("Could not parse server.properties file: " + serverPropertiesFilePath, e); | ||||
| 		} | ||||
| 		 | ||||
| 		instanceProperties.ServerProperties.SetTo(serverPropertiesData); | ||||
|  | ||||
| 		fileStream.Seek(0L, SeekOrigin.Begin); | ||||
| 		fileStream.SetLength(0L); | ||||
| 		 | ||||
| 		serverPropertiesData.Store(fileStream, true); | ||||
| 		var serverPropertiesEditor = new JavaPropertiesFileEditor(); | ||||
| 		instanceProperties.ServerProperties.SetTo(serverPropertiesEditor); | ||||
| 		await serverPropertiesEditor.EditOrCreate(Path.Combine(instanceProperties.InstanceFolder, "server.properties")); | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -8,8 +8,6 @@ public abstract record LaunchResult { | ||||
| 	public sealed record Success(InstanceProcess Process) : LaunchResult; | ||||
|  | ||||
| 	public sealed record InvalidJavaRuntime : LaunchResult; | ||||
| 	 | ||||
| 	public sealed record InvalidJvmArguments : LaunchResult; | ||||
|  | ||||
| 	public sealed record CouldNotDownloadMinecraftServer : LaunchResult; | ||||
| 	 | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| using Phantom.Agent.Minecraft.Java; | ||||
| using Phantom.Agent.Minecraft.Server; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Launcher;  | ||||
| namespace Phantom.Agent.Minecraft.Launcher; | ||||
|  | ||||
| public sealed record LaunchServices(MinecraftServerExecutables ServerExecutables, JavaRuntimeRepository JavaRuntimeRepository); | ||||
|   | ||||
| @@ -3,7 +3,7 @@ using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Utils.IO; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Launcher.Types;  | ||||
| namespace Phantom.Agent.Minecraft.Launcher.Types; | ||||
|  | ||||
| public sealed class FabricLauncher : BaseLauncher { | ||||
| 	public FabricLauncher(InstanceProperties instanceProperties) : base(instanceProperties) {} | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| using Phantom.Agent.Minecraft.Server; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Launcher.Types;  | ||||
| namespace Phantom.Agent.Minecraft.Launcher.Types; | ||||
|  | ||||
| public sealed class InvalidLauncher : IServerLauncher { | ||||
| 	public static InvalidLauncher Instance { get; } = new (); | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Launcher.Types;  | ||||
| namespace Phantom.Agent.Minecraft.Launcher.Types; | ||||
|  | ||||
| public sealed class VanillaLauncher : BaseLauncher { | ||||
| 	public VanillaLauncher(InstanceProperties instanceProperties) : base(instanceProperties) {} | ||||
|   | ||||
| @@ -1,7 +1,6 @@ | ||||
| <Project Sdk="Microsoft.NET.Sdk"> | ||||
|    | ||||
|   <PropertyGroup> | ||||
|     <TargetFramework>net7.0</TargetFramework> | ||||
|     <ImplicitUsings>enable</ImplicitUsings> | ||||
|     <Nullable>enable</Nullable> | ||||
|   </PropertyGroup> | ||||
| @@ -13,11 +12,7 @@ | ||||
|   <ItemGroup> | ||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Data\Phantom.Common.Data.csproj" /> | ||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Logging\Phantom.Common.Logging.csproj" /> | ||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Minecraft\Phantom.Common.Minecraft.csproj" /> | ||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils.Collections\Phantom.Utils.Collections.csproj" /> | ||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils.Cryptography\Phantom.Utils.Cryptography.csproj" /> | ||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils.IO\Phantom.Utils.IO.csproj" /> | ||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils.Runtime\Phantom.Utils.Runtime.csproj" /> | ||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils\Phantom.Utils.csproj" /> | ||||
|   </ItemGroup> | ||||
|  | ||||
| </Project> | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| using Kajabity.Tools.Java; | ||||
| using Phantom.Agent.Minecraft.Java; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Properties;  | ||||
| namespace Phantom.Agent.Minecraft.Properties; | ||||
|  | ||||
| abstract class MinecraftServerProperty<T> { | ||||
| 	private readonly string key; | ||||
| @@ -12,7 +12,7 @@ abstract class MinecraftServerProperty<T> { | ||||
| 	protected abstract T Read(string value); | ||||
| 	protected abstract string Write(T value); | ||||
|  | ||||
| 	public void Set(JavaProperties properties, T value) { | ||||
| 		properties.SetProperty(key, Write(value)); | ||||
| 	public void Set(JavaPropertiesFileEditor properties, T value) { | ||||
| 		properties.Set(key, Write(value)); | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,4 +1,4 @@ | ||||
| using Kajabity.Tools.Java; | ||||
| using Phantom.Agent.Minecraft.Java; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Properties; | ||||
|  | ||||
| @@ -7,7 +7,7 @@ public sealed record ServerProperties( | ||||
| 	ushort RconPort, | ||||
| 	bool EnableRcon = true | ||||
| ) { | ||||
| 	internal void SetTo(JavaProperties properties) { | ||||
| 	internal void SetTo(JavaPropertiesFileEditor properties) { | ||||
| 		MinecraftServerProperties.ServerPort.Set(properties, ServerPort); | ||||
| 		MinecraftServerProperties.RconPort.Set(properties, RconPort); | ||||
| 		MinecraftServerProperties.EnableRcon.Set(properties, EnableRcon); | ||||
|   | ||||
| @@ -1,4 +1,4 @@ | ||||
| namespace Phantom.Agent.Minecraft.Server;  | ||||
| namespace Phantom.Agent.Minecraft.Server; | ||||
|  | ||||
| public sealed class DownloadProgressEventArgs : EventArgs { | ||||
| 	public ulong DownloadedBytes { get; } | ||||
|   | ||||
| @@ -1,3 +1,3 @@ | ||||
| namespace Phantom.Agent.Minecraft.Server;  | ||||
| namespace Phantom.Agent.Minecraft.Server; | ||||
|  | ||||
| sealed record MinecraftServerExecutableDownloadListener(EventHandler<DownloadProgressEventArgs> DownloadProgressEventHandler, CancellationToken CancellationToken); | ||||
|   | ||||
| @@ -10,7 +10,7 @@ public sealed partial class MinecraftServerExecutables { | ||||
| 	private static readonly ILogger Logger = PhantomLogger.Create<MinecraftServerExecutables>(); | ||||
|  | ||||
| 	[GeneratedRegex(@"[^a-zA-Z0-9_\-\.]", RegexOptions.Compiled)] | ||||
| 	private static partial Regex VersionFolderSanitizeRegex(); | ||||
| 	private static partial Regex SanitizePathRegex(); | ||||
|  | ||||
| 	private readonly string basePath; | ||||
| 	private readonly Dictionary<string, MinecraftServerExecutableDownloader> runningDownloadersByVersion = new (); | ||||
| @@ -20,7 +20,7 @@ public sealed partial class MinecraftServerExecutables { | ||||
| 	} | ||||
|  | ||||
| 	internal async Task<string?> DownloadAndGetPath(FileDownloadInfo? fileDownloadInfo, string minecraftVersion, EventHandler<DownloadProgressEventArgs> progressEventHandler, CancellationToken cancellationToken) { | ||||
| 		string serverExecutableFolderPath = Path.Combine(basePath, VersionFolderSanitizeRegex().Replace(minecraftVersion, "_")); | ||||
| 		string serverExecutableFolderPath = Path.Combine(basePath, SanitizePathRegex().IsMatch(minecraftVersion) ? SanitizePathRegex().Replace(minecraftVersion, "_") : minecraftVersion); | ||||
| 		string serverExecutableFilePath = Path.Combine(serverExecutableFolderPath, "server.jar"); | ||||
|  | ||||
| 		if (File.Exists(serverExecutableFilePath)) { | ||||
|   | ||||
| @@ -6,7 +6,7 @@ using System.Text; | ||||
| using Phantom.Common.Logging; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Minecraft.Server;  | ||||
| namespace Phantom.Agent.Minecraft.Server; | ||||
|  | ||||
| public sealed class ServerStatusProtocol { | ||||
| 	private readonly ILogger logger; | ||||
| @@ -78,7 +78,7 @@ public sealed class ServerStatusProtocol { | ||||
| 				return null; | ||||
| 			} | ||||
|  | ||||
| 			string onlinePlayerCountStr = Encoding.BigEndianUnicode.GetString(messageBuffer[(separator1 + 1)..(separator2 - 1)]); | ||||
| 			string onlinePlayerCountStr = Encoding.BigEndianUnicode.GetString(messageBuffer.AsSpan((separator1 + 1)..(separator2 - 1))); | ||||
| 			if (!int.TryParse(onlinePlayerCountStr, out int onlinePlayerCount)) { | ||||
| 				logger.Error("Could not parse online player count in response from server: {OnlinePlayerCount}.", onlinePlayerCountStr); | ||||
| 				return null; | ||||
|   | ||||
							
								
								
									
										25
									
								
								Agent/Phantom.Agent.Rpc/ControllerConnection.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								Agent/Phantom.Agent.Rpc/ControllerConnection.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,25 @@ | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Common.Messages.Agent; | ||||
| using Phantom.Utils.Rpc; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Rpc; | ||||
|  | ||||
| public sealed class ControllerConnection { | ||||
| 	private static readonly ILogger Logger = PhantomLogger.Create(nameof(ControllerConnection)); | ||||
|  | ||||
| 	private readonly RpcConnectionToServer<IMessageToControllerListener> connection; | ||||
| 	 | ||||
| 	public ControllerConnection(RpcConnectionToServer<IMessageToControllerListener> connection) { | ||||
| 		this.connection = connection; | ||||
| 		Logger.Information("Connection ready."); | ||||
| 	} | ||||
|  | ||||
| 	public Task Send<TMessage>(TMessage message) where TMessage : IMessageToController { | ||||
| 		return connection.Send(message); | ||||
| 	} | ||||
|  | ||||
| 	public Task<TReply?> Send<TMessage, TReply>(TMessage message, TimeSpan waitForReplyTime, CancellationToken waitForReplyCancellationToken) where TMessage : IMessageToController<TReply> where TReply : class { | ||||
| 		return connection.Send<TMessage, TReply>(message, waitForReplyTime, waitForReplyCancellationToken); | ||||
| 	} | ||||
| } | ||||
| @@ -1,5 +1,7 @@ | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Common.Messages.ToServer; | ||||
| using Phantom.Common.Messages.Agent; | ||||
| using Phantom.Common.Messages.Agent.ToController; | ||||
| using Phantom.Utils.Rpc; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Rpc; | ||||
| @@ -9,10 +11,10 @@ sealed class KeepAliveLoop { | ||||
|  | ||||
| 	private static readonly TimeSpan KeepAliveInterval = TimeSpan.FromSeconds(10); | ||||
|  | ||||
| 	private readonly RpcServerConnection connection; | ||||
| 	private readonly RpcConnectionToServer<IMessageToControllerListener> connection; | ||||
| 	private readonly CancellationTokenSource cancellationTokenSource = new (); | ||||
|  | ||||
| 	public KeepAliveLoop(RpcServerConnection connection) { | ||||
| 	public KeepAliveLoop(RpcConnectionToServer<IMessageToControllerListener> connection) { | ||||
| 		this.connection = connection; | ||||
| 		Task.Run(Run); | ||||
| 	} | ||||
|   | ||||
| @@ -1,13 +1,12 @@ | ||||
| <Project Sdk="Microsoft.NET.Sdk"> | ||||
|    | ||||
|   <PropertyGroup> | ||||
|     <TargetFramework>net7.0</TargetFramework> | ||||
|     <ImplicitUsings>enable</ImplicitUsings> | ||||
|     <Nullable>enable</Nullable> | ||||
|   </PropertyGroup> | ||||
|    | ||||
|   <ItemGroup> | ||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Messages\Phantom.Common.Messages.csproj" /> | ||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Messages.Agent\Phantom.Common.Messages.Agent.csproj" /> | ||||
|   </ItemGroup> | ||||
|  | ||||
| </Project> | ||||
|   | ||||
							
								
								
									
										38
									
								
								Agent/Phantom.Agent.Rpc/RpcClientRuntime.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										38
									
								
								Agent/Phantom.Agent.Rpc/RpcClientRuntime.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,38 @@ | ||||
| using NetMQ; | ||||
| using NetMQ.Sockets; | ||||
| using Phantom.Common.Data.Agent; | ||||
| using Phantom.Common.Messages.Agent; | ||||
| using Phantom.Common.Messages.Agent.BiDirectional; | ||||
| using Phantom.Common.Messages.Agent.ToController; | ||||
| using Phantom.Utils.Rpc; | ||||
| using Phantom.Utils.Rpc.Sockets; | ||||
| using Phantom.Utils.Tasks; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Rpc; | ||||
|  | ||||
| public sealed class RpcClientRuntime : RpcClientRuntime<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> { | ||||
| 	public static Task Launch(RpcClientSocket<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> socket, AgentInfo agentInfo, IMessageToAgentListener messageListener, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) { | ||||
| 		return new RpcClientRuntime(socket, messageListener, disconnectSemaphore, receiveCancellationToken).Launch(); | ||||
| 	} | ||||
|  | ||||
| 	private RpcClientRuntime(RpcClientSocket<IMessageToAgentListener, IMessageToControllerListener, ReplyMessage> socket, IMessageToAgentListener messageListener, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) : base(socket, messageListener, disconnectSemaphore, receiveCancellationToken) {} | ||||
|  | ||||
| 	protected override void RunWithConnection(ClientSocket socket, RpcConnectionToServer<IMessageToControllerListener> connection, ILogger logger, TaskManager taskManager) { | ||||
| 		var keepAliveLoop = new KeepAliveLoop(connection); | ||||
| 		try { | ||||
| 			base.RunWithConnection(socket, connection, logger, taskManager); | ||||
| 		} finally { | ||||
| 			keepAliveLoop.Cancel(); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	protected override async Task Disconnect(ClientSocket socket, ILogger logger) { | ||||
| 		var unregisterMessageBytes = AgentMessageRegistries.ToController.Write(new UnregisterAgentMessage()).ToArray(); | ||||
| 		try { | ||||
| 			await socket.SendAsync(unregisterMessageBytes).AsTask().WaitAsync(TimeSpan.FromSeconds(5), CancellationToken.None); | ||||
| 		} catch (TimeoutException) { | ||||
| 			logger.Error("Timed out communicating agent shutdown with the controller."); | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -1,107 +0,0 @@ | ||||
| using NetMQ; | ||||
| using NetMQ.Sockets; | ||||
| using Phantom.Common.Data.Agent; | ||||
| using Phantom.Common.Messages; | ||||
| using Phantom.Common.Messages.BiDirectional; | ||||
| using Phantom.Common.Messages.ToServer; | ||||
| using Phantom.Utils.Rpc; | ||||
| using Phantom.Utils.Rpc.Message; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Serilog; | ||||
| using Serilog.Events; | ||||
|  | ||||
| namespace Phantom.Agent.Rpc; | ||||
|  | ||||
| public sealed class RpcLauncher : RpcRuntime<ClientSocket> { | ||||
| 	public static async Task Launch(RpcConfiguration config, AgentAuthToken authToken, AgentInfo agentInfo, Func<RpcServerConnection, IMessageToAgentListener> listenerFactory, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) { | ||||
| 		var socket = new ClientSocket(); | ||||
| 		var options = socket.Options; | ||||
|  | ||||
| 		options.CurveServerCertificate = config.ServerCertificate; | ||||
| 		options.CurveCertificate = new NetMQCertificate(); | ||||
| 		options.HelloMessage = MessageRegistries.ToServer.Write(new RegisterAgentMessage(authToken, agentInfo)).ToArray(); | ||||
|  | ||||
| 		await new RpcLauncher(config, socket, agentInfo.Guid, listenerFactory, disconnectSemaphore, receiveCancellationToken).Launch(); | ||||
| 	} | ||||
|  | ||||
| 	private readonly RpcConfiguration config; | ||||
| 	private readonly Guid agentGuid; | ||||
| 	private readonly Func<RpcServerConnection, IMessageToAgentListener> messageListenerFactory; | ||||
|  | ||||
| 	private readonly SemaphoreSlim disconnectSemaphore; | ||||
| 	private readonly CancellationToken receiveCancellationToken; | ||||
|  | ||||
| 	private RpcLauncher(RpcConfiguration config, ClientSocket socket, Guid agentGuid, Func<RpcServerConnection, IMessageToAgentListener> messageListenerFactory, SemaphoreSlim disconnectSemaphore, CancellationToken receiveCancellationToken) : base(config, socket) { | ||||
| 		this.config = config; | ||||
| 		this.agentGuid = agentGuid; | ||||
| 		this.messageListenerFactory = messageListenerFactory; | ||||
| 		this.disconnectSemaphore = disconnectSemaphore; | ||||
| 		this.receiveCancellationToken = receiveCancellationToken; | ||||
| 	} | ||||
|  | ||||
| 	protected override void Connect(ClientSocket socket) { | ||||
| 		var logger = config.RuntimeLogger; | ||||
| 		var url = config.TcpUrl; | ||||
|  | ||||
| 		logger.Information("Starting ZeroMQ client and connecting to {Url}...", url); | ||||
| 		socket.Connect(url); | ||||
| 		logger.Information("ZeroMQ client ready."); | ||||
| 	} | ||||
|  | ||||
| 	protected override void Run(ClientSocket socket, MessageReplyTracker replyTracker, TaskManager taskManager) { | ||||
| 		var connection = new RpcServerConnection(socket, replyTracker); | ||||
| 		ServerMessaging.SetCurrentConnection(connection); | ||||
| 		 | ||||
| 		var logger = config.RuntimeLogger; | ||||
| 		var handler = new MessageToAgentHandler(messageListenerFactory(connection), logger, taskManager, receiveCancellationToken); | ||||
| 		var keepAliveLoop = new KeepAliveLoop(connection); | ||||
|  | ||||
| 		try { | ||||
| 			while (!receiveCancellationToken.IsCancellationRequested) { | ||||
| 				var data = socket.Receive(receiveCancellationToken); | ||||
| 				 | ||||
| 				LogMessageType(logger, data); | ||||
| 				 | ||||
| 				if (data.Length > 0) { | ||||
| 					MessageRegistries.ToAgent.Handle(data, handler); | ||||
| 				} | ||||
| 			} | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			// Ignore. | ||||
| 		} finally { | ||||
| 			logger.Debug("ZeroMQ client stopped receiving messages."); | ||||
|  | ||||
| 			disconnectSemaphore.Wait(CancellationToken.None); | ||||
| 			keepAliveLoop.Cancel(); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private static void LogMessageType(ILogger logger, ReadOnlyMemory<byte> data) { | ||||
| 		if (!logger.IsEnabled(LogEventLevel.Verbose)) { | ||||
| 			return; | ||||
| 		} | ||||
|  | ||||
| 		if (data.Length > 0 && MessageRegistries.ToAgent.TryGetType(data, out var type)) { | ||||
| 			logger.Verbose("Received {MessageType} ({Bytes} B) from server.", type.Name, data.Length); | ||||
| 		} | ||||
| 		else { | ||||
| 			logger.Verbose("Received {Bytes} B message from server.", data.Length); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	protected override async Task Disconnect() { | ||||
| 		var unregisterTimeoutTask = Task.Delay(TimeSpan.FromSeconds(5), CancellationToken.None); | ||||
| 		var finishedTask = await Task.WhenAny(ServerMessaging.Send(new UnregisterAgentMessage(agentGuid)), unregisterTimeoutTask); | ||||
| 		if (finishedTask == unregisterTimeoutTask) { | ||||
| 			config.RuntimeLogger.Error("Timed out communicating agent shutdown with the server."); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private sealed class MessageToAgentHandler : MessageHandler<IMessageToAgentListener> { | ||||
| 		public MessageToAgentHandler(IMessageToAgentListener listener, ILogger logger, TaskManager taskManager, CancellationToken cancellationToken) : base(listener, logger, taskManager, cancellationToken) {} | ||||
| 		 | ||||
| 		protected override Task SendReply(uint sequenceId, byte[] serializedReply) { | ||||
| 			return ServerMessaging.Send(new ReplyMessage(sequenceId, serializedReply)); | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -1,41 +0,0 @@ | ||||
| using NetMQ; | ||||
| using NetMQ.Sockets; | ||||
| using Phantom.Common.Messages; | ||||
| using Phantom.Common.Messages.BiDirectional; | ||||
| using Phantom.Utils.Rpc.Message; | ||||
|  | ||||
| namespace Phantom.Agent.Rpc;  | ||||
|  | ||||
| public sealed class RpcServerConnection { | ||||
| 	private readonly ClientSocket socket; | ||||
| 	private readonly MessageReplyTracker replyTracker; | ||||
|  | ||||
| 	internal RpcServerConnection(ClientSocket socket, MessageReplyTracker replyTracker) { | ||||
| 		this.socket = socket; | ||||
| 		this.replyTracker = replyTracker; | ||||
| 	} | ||||
|  | ||||
| 	internal async Task Send<TMessage>(TMessage message) where TMessage : IMessageToServer { | ||||
| 		var bytes = MessageRegistries.ToServer.Write(message).ToArray(); | ||||
| 		if (bytes.Length > 0) { | ||||
| 			await socket.SendAsync(bytes); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	internal async Task<TReply?> Send<TMessage, TReply>(TMessage message, TimeSpan waitForReplyTime, CancellationToken waitForReplyCancellationToken) where TMessage : IMessageToServer<TReply> where TReply : class { | ||||
| 		var sequenceId = replyTracker.RegisterReply(); | ||||
| 		 | ||||
| 		var bytes = MessageRegistries.ToServer.Write<TMessage, TReply>(sequenceId, message).ToArray(); | ||||
| 		if (bytes.Length == 0) { | ||||
| 			replyTracker.ForgetReply(sequenceId); | ||||
| 			return null; | ||||
| 		} | ||||
|  | ||||
| 		await socket.SendAsync(bytes); | ||||
| 		return await replyTracker.WaitForReply<TReply>(sequenceId, waitForReplyTime, waitForReplyCancellationToken); | ||||
| 	} | ||||
|  | ||||
| 	public void Receive(ReplyMessage message) { | ||||
| 		replyTracker.ReceiveReply(message.SequenceId, message.SerializedReply); | ||||
| 	} | ||||
| } | ||||
| @@ -1,34 +0,0 @@ | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Common.Messages; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Rpc; | ||||
|  | ||||
| public static class ServerMessaging { | ||||
| 	private static readonly ILogger Logger = PhantomLogger.Create(nameof(ServerMessaging)); | ||||
| 	 | ||||
| 	private static RpcServerConnection? CurrentConnection { get; set; } | ||||
| 	private static RpcServerConnection CurrentConnectionOrThrow => CurrentConnection ?? throw new InvalidOperationException("Server connection not ready."); | ||||
| 	 | ||||
| 	private static readonly object SetCurrentConnectionLock = new (); | ||||
|  | ||||
| 	internal static void SetCurrentConnection(RpcServerConnection connection) { | ||||
| 		lock (SetCurrentConnectionLock) { | ||||
| 			if (CurrentConnection != null) { | ||||
| 				throw new InvalidOperationException("Server connection can only be set once."); | ||||
| 			} | ||||
| 			 | ||||
| 			CurrentConnection = connection; | ||||
| 		} | ||||
| 		 | ||||
| 		Logger.Information("Server connection ready."); | ||||
| 	} | ||||
|  | ||||
| 	public static Task Send<TMessage>(TMessage message) where TMessage : IMessageToServer { | ||||
| 		return CurrentConnectionOrThrow.Send(message); | ||||
| 	} | ||||
|  | ||||
| 	public static Task<TReply?> Send<TMessage, TReply>(TMessage message, TimeSpan waitForReplyTime, CancellationToken waitForReplyCancellationToken) where TMessage : IMessageToServer<TReply> where TReply : class { | ||||
| 		return CurrentConnectionOrThrow.Send<TMessage, TReply>(message, waitForReplyTime, waitForReplyCancellationToken); | ||||
| 	} | ||||
| } | ||||
| @@ -0,0 +1,3 @@ | ||||
| namespace Phantom.Agent.Services; | ||||
|  | ||||
| public readonly record struct AgentServiceConfiguration(int MaxConcurrentCompressionTasks); | ||||
| @@ -1,9 +1,10 @@ | ||||
| using Phantom.Agent.Minecraft.Java; | ||||
| using Phantom.Agent.Rpc; | ||||
| using Phantom.Agent.Services.Backups; | ||||
| using Phantom.Agent.Services.Instances; | ||||
| using Phantom.Common.Data.Agent; | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Utils.Tasks; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Services; | ||||
| @@ -18,12 +19,12 @@ public sealed class AgentServices { | ||||
| 	internal JavaRuntimeRepository JavaRuntimeRepository { get; } | ||||
| 	internal InstanceSessionManager InstanceSessionManager { get; } | ||||
|  | ||||
| 	public AgentServices(AgentInfo agentInfo, AgentFolders agentFolders) { | ||||
| 	public AgentServices(AgentInfo agentInfo, AgentFolders agentFolders, AgentServiceConfiguration serviceConfiguration, ControllerConnection controllerConnection) { | ||||
| 		this.AgentFolders = agentFolders; | ||||
| 		this.TaskManager = new TaskManager(PhantomLogger.Create<TaskManager, AgentServices>()); | ||||
| 		this.BackupManager = new BackupManager(agentFolders); | ||||
| 		this.BackupManager = new BackupManager(agentFolders, serviceConfiguration.MaxConcurrentCompressionTasks); | ||||
| 		this.JavaRuntimeRepository = new JavaRuntimeRepository(); | ||||
| 		this.InstanceSessionManager = new InstanceSessionManager(agentInfo, agentFolders, JavaRuntimeRepository, TaskManager, BackupManager); | ||||
| 		this.InstanceSessionManager = new InstanceSessionManager(controllerConnection, agentInfo, agentFolders, JavaRuntimeRepository, TaskManager, BackupManager); | ||||
| 	} | ||||
|  | ||||
| 	public async Task Initialize() { | ||||
| @@ -35,11 +36,11 @@ public sealed class AgentServices { | ||||
| 	public async Task Shutdown() { | ||||
| 		Logger.Information("Stopping services..."); | ||||
| 		 | ||||
| 		await InstanceSessionManager.StopAll(); | ||||
| 		InstanceSessionManager.Dispose(); | ||||
| 		 | ||||
| 		await InstanceSessionManager.DisposeAsync(); | ||||
| 		await TaskManager.Stop(); | ||||
| 		 | ||||
| 		BackupManager.Dispose(); | ||||
| 		 | ||||
| 		Logger.Information("Services stopped."); | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -7,7 +7,7 @@ using Phantom.Common.Logging; | ||||
| using Phantom.Utils.IO; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Backups;  | ||||
| namespace Phantom.Agent.Services.Backups; | ||||
|  | ||||
| sealed class BackupArchiver { | ||||
| 	private readonly string destinationBasePath; | ||||
|   | ||||
| @@ -1,8 +1,8 @@ | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Utils.Processes; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Backups;  | ||||
| namespace Phantom.Agent.Services.Backups; | ||||
|  | ||||
| static class BackupCompressor { | ||||
| 	private static ILogger Logger { get; } = PhantomLogger.Create(nameof(BackupCompressor)); | ||||
| @@ -47,10 +47,8 @@ static class BackupCompressor { | ||||
| 				Quality, | ||||
| 				Memory, | ||||
| 				Threads, | ||||
| 				"-c", | ||||
| 				"--rm", | ||||
| 				"--no-progress", | ||||
| 				"-c", | ||||
| 				"-o", destinationFilePath, | ||||
| 				"--", sourceFilePath | ||||
| 			} | ||||
|   | ||||
| @@ -5,44 +5,34 @@ using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Backups; | ||||
|  | ||||
| sealed class BackupManager { | ||||
| sealed class BackupManager : IDisposable { | ||||
| 	private readonly string destinationBasePath; | ||||
| 	private readonly string temporaryBasePath; | ||||
| 	private readonly SemaphoreSlim compressionSemaphore; | ||||
|  | ||||
| 	public BackupManager(AgentFolders agentFolders) { | ||||
| 	public BackupManager(AgentFolders agentFolders, int maxConcurrentCompressionTasks) { | ||||
| 		this.destinationBasePath = agentFolders.BackupsFolderPath; | ||||
| 		this.temporaryBasePath = Path.Combine(agentFolders.TemporaryFolderPath, "backups"); | ||||
| 		this.compressionSemaphore = new SemaphoreSlim(maxConcurrentCompressionTasks, maxConcurrentCompressionTasks); | ||||
| 	} | ||||
|  | ||||
| 	public async Task<BackupCreationResult> CreateBackup(string loggerName, InstanceProcess process, CancellationToken cancellationToken) { | ||||
| 		try { | ||||
| 			if (!await process.BackupSemaphore.Wait(TimeSpan.FromSeconds(1), cancellationToken)) { | ||||
| 				return new BackupCreationResult(BackupCreationResultKind.BackupAlreadyRunning); | ||||
| 			} | ||||
| 		} catch (ObjectDisposedException) { | ||||
| 			return new BackupCreationResult(BackupCreationResultKind.InstanceNotRunning); | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			return new BackupCreationResult(BackupCreationResultKind.InstanceNotRunning); | ||||
| 		} | ||||
| 	public Task<BackupCreationResult> CreateBackup(string loggerName, InstanceProcess process, CancellationToken cancellationToken) { | ||||
| 		return new BackupCreator(this, loggerName, process, cancellationToken).CreateBackup(); | ||||
| 	} | ||||
|  | ||||
| 		try { | ||||
| 			return await new BackupCreator(destinationBasePath, temporaryBasePath, loggerName, process, cancellationToken).CreateBackup(); | ||||
| 		} finally { | ||||
| 			process.BackupSemaphore.Release(); | ||||
| 		} | ||||
| 	public void Dispose() { | ||||
| 		compressionSemaphore.Dispose(); | ||||
| 	} | ||||
|  | ||||
| 	private sealed class BackupCreator { | ||||
| 		private readonly string destinationBasePath; | ||||
| 		private readonly string temporaryBasePath; | ||||
| 		private readonly BackupManager manager; | ||||
| 		private readonly string loggerName; | ||||
| 		private readonly ILogger logger; | ||||
| 		private readonly InstanceProcess process; | ||||
| 		private readonly CancellationToken cancellationToken; | ||||
|  | ||||
| 		public BackupCreator(string destinationBasePath, string temporaryBasePath, string loggerName, InstanceProcess process, CancellationToken cancellationToken) { | ||||
| 			this.destinationBasePath = destinationBasePath; | ||||
| 			this.temporaryBasePath = temporaryBasePath; | ||||
| 		public BackupCreator(BackupManager manager, string loggerName, InstanceProcess process, CancellationToken cancellationToken) { | ||||
| 			this.manager = manager; | ||||
| 			this.loggerName = loggerName; | ||||
| 			this.logger = PhantomLogger.Create<BackupManager>(loggerName); | ||||
| 			this.process = process; | ||||
| @@ -72,7 +62,7 @@ sealed class BackupManager { | ||||
| 			try { | ||||
| 				await dispatcher.DisableAutomaticSaving(); | ||||
| 				await dispatcher.SaveAllChunks(); | ||||
| 				return await new BackupArchiver(destinationBasePath, temporaryBasePath, loggerName, process.InstanceProperties, cancellationToken).ArchiveWorld(resultBuilder); | ||||
| 				return await new BackupArchiver(manager.destinationBasePath, manager.temporaryBasePath, loggerName, process.InstanceProperties, cancellationToken).ArchiveWorld(resultBuilder); | ||||
| 			} catch (OperationCanceledException) { | ||||
| 				resultBuilder.Kind = BackupCreationResultKind.BackupCancelled; | ||||
| 				logger.Warning("Backup creation was cancelled."); | ||||
| @@ -85,24 +75,34 @@ sealed class BackupManager { | ||||
| 				try { | ||||
| 					await dispatcher.EnableAutomaticSaving(); | ||||
| 				} catch (OperationCanceledException) { | ||||
| 					// ignore | ||||
| 					// Ignore. | ||||
| 				} catch (Exception e) { | ||||
| 					resultBuilder.Warnings |= BackupCreationWarnings.CouldNotRestoreAutomaticSaving; | ||||
| 					logger.Error(e, "Caught exception while enabling automatic saving after creating an instance backup."); | ||||
| 				} | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		 | ||||
| 		private async Task CompressWorldArchive(string filePath, BackupCreationResult.Builder resultBuilder) { | ||||
| 			var compressedFilePath = await BackupCompressor.Compress(filePath, cancellationToken); | ||||
| 			if (compressedFilePath == null) { | ||||
| 				resultBuilder.Warnings |= BackupCreationWarnings.CouldNotCompressWorldArchive; | ||||
| 			if (!await manager.compressionSemaphore.WaitAsync(TimeSpan.FromSeconds(1), cancellationToken)) { | ||||
| 				logger.Information("Too many compression tasks running, waiting for one of them to complete..."); | ||||
| 				await manager.compressionSemaphore.WaitAsync(cancellationToken); | ||||
| 			} | ||||
|  | ||||
| 			logger.Information("Compressing backup..."); | ||||
| 			try { | ||||
| 				var compressedFilePath = await BackupCompressor.Compress(filePath, cancellationToken); | ||||
| 				if (compressedFilePath == null) { | ||||
| 					resultBuilder.Warnings |= BackupCreationWarnings.CouldNotCompressWorldArchive; | ||||
| 				} | ||||
| 			} finally { | ||||
| 				manager.compressionSemaphore.Release(); | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		private void LogBackupResult(BackupCreationResult result) { | ||||
| 			if (result.Kind != BackupCreationResultKind.Success) { | ||||
| 				logger.Warning("Backup failed: {Reason}", result.Kind.ToSentence()); | ||||
| 				logger.Warning("Backup failed: {Reason}", DescribeResult(result.Kind)); | ||||
| 				return; | ||||
| 			} | ||||
| 			 | ||||
| @@ -114,5 +114,19 @@ sealed class BackupManager { | ||||
| 				logger.Information("Backup finished successfully."); | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		private static string DescribeResult(BackupCreationResultKind kind) { | ||||
| 			return kind switch { | ||||
| 				BackupCreationResultKind.Success                            => "Backup created successfully.", | ||||
| 				BackupCreationResultKind.InstanceNotRunning                 => "Instance is not running.", | ||||
| 				BackupCreationResultKind.BackupCancelled                    => "Backup cancelled.", | ||||
| 				BackupCreationResultKind.BackupAlreadyRunning               => "A backup is already being created.", | ||||
| 				BackupCreationResultKind.BackupFileAlreadyExists            => "Backup with the same name already exists.", | ||||
| 				BackupCreationResultKind.CouldNotCreateBackupFolder         => "Could not create backup folder.", | ||||
| 				BackupCreationResultKind.CouldNotCopyWorldToTemporaryFolder => "Could not copy world to temporary folder.", | ||||
| 				BackupCreationResultKind.CouldNotCreateWorldArchive         => "Could not create world archive.", | ||||
| 				_                                                           => "Unknown error." | ||||
| 			}; | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,8 +1,11 @@ | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Agent.Minecraft.Server; | ||||
| using Phantom.Agent.Services.Instances; | ||||
| using Phantom.Agent.Services.Instances.Procedures; | ||||
| using Phantom.Common.Data.Backups; | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Utils.Tasks; | ||||
| using Phantom.Utils.Threading; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Backups; | ||||
|  | ||||
| @@ -12,21 +15,23 @@ sealed class BackupScheduler : CancellableBackgroundTask { | ||||
| 	private static readonly TimeSpan BackupInterval = TimeSpan.FromMinutes(30); | ||||
| 	private static readonly TimeSpan BackupFailureRetryDelay = TimeSpan.FromMinutes(5); | ||||
|  | ||||
| 	private readonly string loggerName; | ||||
| 	private readonly BackupManager backupManager; | ||||
| 	private readonly InstanceProcess process; | ||||
| 	private readonly IInstanceContext context; | ||||
| 	private readonly SemaphoreSlim backupSemaphore = new (1, 1); | ||||
| 	private readonly int serverPort; | ||||
| 	private readonly ServerStatusProtocol serverStatusProtocol; | ||||
| 	private readonly ManualResetEventSlim serverOutputWhileWaitingForOnlinePlayers = new (); | ||||
| 	 | ||||
| 	public event EventHandler<BackupCreationResult>? BackupCompleted;  | ||||
| 	public event EventHandler<BackupCreationResult>? BackupCompleted; | ||||
|  | ||||
| 	public BackupScheduler(TaskManager taskManager, BackupManager backupManager, InstanceProcess process, int serverPort, string loggerName) : base(PhantomLogger.Create<BackupScheduler>(loggerName), taskManager, "Backup scheduler for " + loggerName) { | ||||
| 		this.loggerName = loggerName; | ||||
| 	public BackupScheduler(TaskManager taskManager, BackupManager backupManager, InstanceProcess process, IInstanceContext context, int serverPort) : base(PhantomLogger.Create<BackupScheduler>(context.ShortName), taskManager, "Backup scheduler for " + context.ShortName) { | ||||
| 		this.backupManager = backupManager; | ||||
| 		this.process = process; | ||||
| 		this.context = context; | ||||
| 		this.serverPort = serverPort; | ||||
| 		this.serverStatusProtocol = new ServerStatusProtocol(loggerName); | ||||
| 		this.serverStatusProtocol = new ServerStatusProtocol(context.ShortName); | ||||
| 		Start(); | ||||
| 	} | ||||
|  | ||||
| 	protected override async Task RunTask() { | ||||
| @@ -50,7 +55,17 @@ sealed class BackupScheduler : CancellableBackgroundTask { | ||||
| 	} | ||||
|  | ||||
| 	private async Task<BackupCreationResult> CreateBackup() { | ||||
| 		return await backupManager.CreateBackup(loggerName, process, CancellationToken.None); | ||||
| 		if (!await backupSemaphore.WaitAsync(TimeSpan.FromSeconds(1))) { | ||||
| 			return new BackupCreationResult(BackupCreationResultKind.BackupAlreadyRunning); | ||||
| 		} | ||||
| 		 | ||||
| 		try { | ||||
| 			var procedure = new BackupInstanceProcedure(backupManager); | ||||
| 			context.EnqueueProcedure(procedure); | ||||
| 			return await procedure.Result; | ||||
| 		} finally { | ||||
| 			backupSemaphore.Release(); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private async Task WaitForOnlinePlayers() { | ||||
| @@ -93,4 +108,9 @@ sealed class BackupScheduler : CancellableBackgroundTask { | ||||
| 			Logger.Debug("Detected server output, signalling to check for online players again."); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	protected override void Dispose() { | ||||
| 		backupSemaphore.Dispose(); | ||||
| 		serverOutputWhileWaitingForOnlinePlayers.Dispose(); | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,7 +1,7 @@ | ||||
| using System.Text.RegularExpressions; | ||||
| using Phantom.Agent.Minecraft.Command; | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Utils.Tasks; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Backups; | ||||
| @@ -14,9 +14,9 @@ sealed partial class BackupServerCommandDispatcher : IDisposable { | ||||
| 	private readonly InstanceProcess process; | ||||
| 	private readonly CancellationToken cancellationToken; | ||||
|  | ||||
| 	private readonly TaskCompletionSource automaticSavingDisabled = Tasks.CreateCompletionSource(); | ||||
| 	private readonly TaskCompletionSource savedTheGame = Tasks.CreateCompletionSource(); | ||||
| 	private readonly TaskCompletionSource automaticSavingEnabled = Tasks.CreateCompletionSource(); | ||||
| 	private readonly TaskCompletionSource automaticSavingDisabled = AsyncTasks.CreateCompletionSource(); | ||||
| 	private readonly TaskCompletionSource savedTheGame = AsyncTasks.CreateCompletionSource(); | ||||
| 	private readonly TaskCompletionSource automaticSavingEnabled = AsyncTasks.CreateCompletionSource(); | ||||
|  | ||||
| 	public BackupServerCommandDispatcher(ILogger logger, InstanceProcess process, CancellationToken cancellationToken) { | ||||
| 		this.logger = logger; | ||||
|   | ||||
							
								
								
									
										25
									
								
								Agent/Phantom.Agent.Services/Instances/IInstanceContext.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								Agent/Phantom.Agent.Services/Instances/IInstanceContext.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,25 @@ | ||||
| using Phantom.Agent.Services.Instances.Procedures; | ||||
| using Phantom.Agent.Services.Instances.States; | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances; | ||||
|  | ||||
| interface IInstanceContext { | ||||
| 	string ShortName { get; } | ||||
| 	ILogger Logger { get; } | ||||
| 	 | ||||
| 	InstanceServices Services { get; } | ||||
| 	IInstanceState CurrentState { get; } | ||||
|  | ||||
| 	void SetStatus(IInstanceStatus newStatus); | ||||
| 	void ReportEvent(IInstanceEvent instanceEvent); | ||||
| 	void EnqueueProcedure(IInstanceProcedure procedure, bool immediate = false); | ||||
| } | ||||
|  | ||||
| static class InstanceContextExtensions { | ||||
| 	public static void SetLaunchFailedStatusAndReportEvent(this IInstanceContext context, InstanceLaunchFailReason reason) { | ||||
| 		context.SetStatus(InstanceStatus.Failed(reason)); | ||||
| 		context.ReportEvent(new InstanceLaunchFailedEvent(reason)); | ||||
| 	} | ||||
| } | ||||
| @@ -1,27 +1,21 @@ | ||||
| using Phantom.Agent.Minecraft.Launcher; | ||||
| using Phantom.Agent.Rpc; | ||||
| using Phantom.Agent.Services.Instances.Procedures; | ||||
| using Phantom.Agent.Services.Instances.States; | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Phantom.Common.Data.Minecraft; | ||||
| using Phantom.Common.Data.Replies; | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Common.Messages.ToServer; | ||||
| using Phantom.Common.Messages.Agent.ToController; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances; | ||||
|  | ||||
| sealed class Instance : IDisposable { | ||||
| 	private static uint loggerSequenceId = 0; | ||||
|  | ||||
| 	private static string GetLoggerName(Guid guid) { | ||||
| 		var prefix = guid.ToString(); | ||||
| 		return prefix[..prefix.IndexOf('-')] + "/" + Interlocked.Increment(ref loggerSequenceId); | ||||
| 	} | ||||
|  | ||||
| sealed class Instance : IAsyncDisposable { | ||||
| 	private InstanceServices Services { get; } | ||||
| 	 | ||||
| 	public InstanceConfiguration Configuration { get; private set; } | ||||
| 	private IServerLauncher Launcher { get; set; } | ||||
| 	private readonly SemaphoreSlim configurationSemaphore = new (1, 1); | ||||
|  | ||||
| 	private readonly string shortName; | ||||
| 	private readonly ILogger logger; | ||||
| @@ -30,14 +24,14 @@ sealed class Instance : IDisposable { | ||||
| 	private int statusUpdateCounter; | ||||
| 	 | ||||
| 	private IInstanceState currentState; | ||||
| 	private readonly SemaphoreSlim stateTransitioningActionSemaphore = new (1, 1); | ||||
|  | ||||
| 	public bool IsRunning => currentState is not InstanceNotRunningState; | ||||
| 	 | ||||
| 	public event EventHandler? IsRunningChanged;  | ||||
| 	public event EventHandler? IsRunningChanged; | ||||
| 	 | ||||
| 	private readonly InstanceProcedureManager procedureManager; | ||||
|  | ||||
| 	public Instance(InstanceServices services, InstanceConfiguration configuration, IServerLauncher launcher) { | ||||
| 		this.shortName = GetLoggerName(configuration.InstanceGuid); | ||||
| 	public Instance(string shortName, InstanceServices services, InstanceConfiguration configuration, IServerLauncher launcher) { | ||||
| 		this.shortName = shortName; | ||||
| 		this.logger = PhantomLogger.Create<Instance>(shortName); | ||||
|  | ||||
| 		this.Services = services; | ||||
| @@ -46,6 +40,8 @@ sealed class Instance : IDisposable { | ||||
| 		 | ||||
| 		this.currentState = new InstanceNotRunningState(); | ||||
| 		this.currentStatus = InstanceStatus.NotRunning; | ||||
| 		 | ||||
| 		this.procedureManager = new InstanceProcedureManager(this, new Context(this), services.TaskManager); | ||||
| 	} | ||||
|  | ||||
| 	private void TryUpdateStatus(string taskName, Func<Task> getUpdateTask) { | ||||
| @@ -60,23 +56,25 @@ sealed class Instance : IDisposable { | ||||
|  | ||||
| 	public void ReportLastStatus() { | ||||
| 		TryUpdateStatus("Report last status of instance " + shortName, async () => { | ||||
| 			await ServerMessaging.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, currentStatus)); | ||||
| 			await Services.ControllerConnection.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, currentStatus)); | ||||
| 		}); | ||||
| 	} | ||||
|  | ||||
| 	private void ReportAndSetStatus(IInstanceStatus status) { | ||||
| 		TryUpdateStatus("Report status of instance " + shortName + " as " + status.GetType().Name, async () => { | ||||
| 			currentStatus = status; | ||||
| 			await ServerMessaging.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, status)); | ||||
| 			if (status != currentStatus) { | ||||
| 				currentStatus = status; | ||||
| 				await Services.ControllerConnection.Send(new ReportInstanceStatusMessage(Configuration.InstanceGuid, status)); | ||||
| 			} | ||||
| 		}); | ||||
| 	} | ||||
|  | ||||
| 	private void ReportEvent(IInstanceEvent instanceEvent) { | ||||
| 		var message = new ReportInstanceEventMessage(Guid.NewGuid(), DateTime.UtcNow, Configuration.InstanceGuid, instanceEvent); | ||||
| 		Services.TaskManager.Run("Report event for instance " + shortName, async () => await ServerMessaging.Send(message)); | ||||
| 		Services.TaskManager.Run("Report event for instance " + shortName, async () => await Services.ControllerConnection.Send(message)); | ||||
| 	} | ||||
| 	 | ||||
| 	private void TransitionState(IInstanceState newState) { | ||||
| 	internal void TransitionState(IInstanceState newState) { | ||||
| 		if (currentState == newState) { | ||||
| 			return; | ||||
| 		} | ||||
| @@ -96,114 +94,94 @@ sealed class Instance : IDisposable { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private T TransitionStateAndReturn<T>((IInstanceState State, T Result) newStateAndResult) { | ||||
| 		TransitionState(newStateAndResult.State); | ||||
| 		return newStateAndResult.Result; | ||||
| 	} | ||||
|  | ||||
| 	public async Task Reconfigure(InstanceConfiguration configuration, IServerLauncher launcher, CancellationToken cancellationToken) { | ||||
| 		await stateTransitioningActionSemaphore.WaitAsync(cancellationToken); | ||||
| 		await configurationSemaphore.WaitAsync(cancellationToken); | ||||
| 		try { | ||||
| 			Configuration = configuration; | ||||
| 			Launcher = launcher; | ||||
| 		} finally { | ||||
| 			stateTransitioningActionSemaphore.Release(); | ||||
| 			configurationSemaphore.Release(); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	public async Task<LaunchInstanceResult> Launch(CancellationToken shutdownCancellationToken) { | ||||
| 		await stateTransitioningActionSemaphore.WaitAsync(shutdownCancellationToken); | ||||
| 	public async Task<LaunchInstanceResult> Launch(CancellationToken cancellationToken) { | ||||
| 		if (IsRunning) { | ||||
| 			return LaunchInstanceResult.InstanceAlreadyRunning; | ||||
| 		} | ||||
|  | ||||
| 		if (await procedureManager.GetCurrentProcedure(cancellationToken) is LaunchInstanceProcedure) { | ||||
| 			return LaunchInstanceResult.InstanceAlreadyLaunching; | ||||
| 		} | ||||
| 		 | ||||
| 		LaunchInstanceProcedure procedure; | ||||
| 		 | ||||
| 		await configurationSemaphore.WaitAsync(cancellationToken); | ||||
| 		try { | ||||
| 			return TransitionStateAndReturn(currentState.Launch(new InstanceContextImpl(this, shutdownCancellationToken))); | ||||
| 		} catch (Exception e) { | ||||
| 			logger.Error(e, "Caught exception while launching instance."); | ||||
| 			return LaunchInstanceResult.UnknownError; | ||||
| 			procedure = new LaunchInstanceProcedure(Configuration, Launcher); | ||||
| 		} finally { | ||||
| 			stateTransitioningActionSemaphore.Release(); | ||||
| 			configurationSemaphore.Release(); | ||||
| 		} | ||||
| 		 | ||||
| 		ReportAndSetStatus(InstanceStatus.Launching); | ||||
| 		await procedureManager.Enqueue(procedure); | ||||
| 		return LaunchInstanceResult.LaunchInitiated; | ||||
| 	} | ||||
|  | ||||
| 	public async Task<StopInstanceResult> Stop(MinecraftStopStrategy stopStrategy) { | ||||
| 		await stateTransitioningActionSemaphore.WaitAsync(); | ||||
| 		try { | ||||
| 			return TransitionStateAndReturn(currentState.Stop(stopStrategy)); | ||||
| 		} catch (Exception e) { | ||||
| 			logger.Error(e, "Caught exception while stopping instance."); | ||||
| 			return StopInstanceResult.UnknownError; | ||||
| 		} finally { | ||||
| 			stateTransitioningActionSemaphore.Release(); | ||||
| 	public async Task<StopInstanceResult> Stop(MinecraftStopStrategy stopStrategy, CancellationToken cancellationToken) { | ||||
| 		if (!IsRunning) { | ||||
| 			return StopInstanceResult.InstanceAlreadyStopped; | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	public async Task StopAndWait(TimeSpan waitTime) { | ||||
| 		await Stop(MinecraftStopStrategy.Instant); | ||||
|  | ||||
| 		using var waitTokenSource = new CancellationTokenSource(waitTime); | ||||
| 		var waitToken = waitTokenSource.Token; | ||||
|  | ||||
| 		while (currentState is not InstanceNotRunningState) { | ||||
| 			await Task.Delay(TimeSpan.FromMilliseconds(250), waitToken); | ||||
| 		 | ||||
| 		if (await procedureManager.GetCurrentProcedure(cancellationToken) is StopInstanceProcedure) { | ||||
| 			return StopInstanceResult.InstanceAlreadyStopping; | ||||
| 		} | ||||
| 		 | ||||
| 		ReportAndSetStatus(InstanceStatus.Stopping); | ||||
| 		await procedureManager.Enqueue(new StopInstanceProcedure(stopStrategy)); | ||||
| 		return StopInstanceResult.StopInitiated; | ||||
| 	} | ||||
|  | ||||
| 	public async Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||
| 		return await currentState.SendCommand(command, cancellationToken); | ||||
| 	} | ||||
|  | ||||
| 	private sealed class InstanceContextImpl : InstanceContext { | ||||
| 		private readonly Instance instance; | ||||
| 		private readonly CancellationToken shutdownCancellationToken; | ||||
| 		 | ||||
| 		public InstanceContextImpl(Instance instance, CancellationToken shutdownCancellationToken) : base(instance.Services, instance.Configuration, instance.Launcher) { | ||||
| 			this.instance = instance; | ||||
| 			this.shutdownCancellationToken = shutdownCancellationToken; | ||||
| 	public async ValueTask DisposeAsync() { | ||||
| 		await procedureManager.DisposeAsync(); | ||||
|  | ||||
| 		while (currentState is not InstanceNotRunningState) { | ||||
| 			await Task.Delay(TimeSpan.FromMilliseconds(250), CancellationToken.None); | ||||
| 		} | ||||
|  | ||||
| 		public override ILogger Logger => instance.logger; | ||||
| 		public override string ShortName => instance.shortName; | ||||
|  | ||||
| 		public override void SetStatus(IInstanceStatus newStatus) { | ||||
| 			instance.ReportAndSetStatus(newStatus); | ||||
| 		} | ||||
|  | ||||
| 		public override void ReportEvent(IInstanceEvent instanceEvent) { | ||||
| 			instance.ReportEvent(instanceEvent); | ||||
| 		} | ||||
|  | ||||
| 		public override void TransitionState(Func<(IInstanceState, IInstanceStatus?)> newStateAndStatus) { | ||||
| 			instance.stateTransitioningActionSemaphore.Wait(CancellationToken.None); | ||||
| 			try { | ||||
| 				var (state, status) = newStateAndStatus(); | ||||
| 				 | ||||
| 				if (!instance.IsRunning) { | ||||
| 					// Only InstanceSessionManager is allowed to transition an instance out of a non-running state. | ||||
| 					instance.logger.Debug("Cancelled state transition to {State} because instance is not running.", state.GetType().Name); | ||||
| 					return; | ||||
| 				} | ||||
| 				 | ||||
| 				if (state is not InstanceNotRunningState && shutdownCancellationToken.IsCancellationRequested) { | ||||
| 					instance.logger.Debug("Cancelled state transition to {State} due to Agent shutdown.", state.GetType().Name); | ||||
| 					return; | ||||
| 				} | ||||
|  | ||||
| 				if (status != null) { | ||||
| 					SetStatus(status); | ||||
| 				} | ||||
|  | ||||
| 				instance.TransitionState(state); | ||||
| 			} catch (Exception e) { | ||||
| 				instance.logger.Error(e, "Caught exception during state transition."); | ||||
| 			} finally { | ||||
| 				instance.stateTransitioningActionSemaphore.Release(); | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	public void Dispose() { | ||||
| 		stateTransitioningActionSemaphore.Dispose(); | ||||
| 		 | ||||
| 		if (currentState is IDisposable disposable) { | ||||
| 			disposable.Dispose(); | ||||
| 		} | ||||
| 		 | ||||
| 		configurationSemaphore.Dispose(); | ||||
| 	} | ||||
|  | ||||
| 	private sealed class Context : IInstanceContext { | ||||
| 		public string ShortName => instance.shortName; | ||||
| 		public ILogger Logger => instance.logger; | ||||
| 		 | ||||
| 		public InstanceServices Services => instance.Services; | ||||
| 		public IInstanceState CurrentState => instance.currentState; | ||||
|  | ||||
| 		private readonly Instance instance; | ||||
| 		 | ||||
| 		public Context(Instance instance) { | ||||
| 			this.instance = instance; | ||||
| 		} | ||||
|  | ||||
| 		public void SetStatus(IInstanceStatus newStatus) { | ||||
| 			instance.ReportAndSetStatus(newStatus); | ||||
| 		} | ||||
|  | ||||
| 		public void ReportEvent(IInstanceEvent instanceEvent) { | ||||
| 			instance.ReportEvent(instanceEvent); | ||||
| 		} | ||||
|  | ||||
| 		public void EnqueueProcedure(IInstanceProcedure procedure, bool immediate) { | ||||
| 			Services.TaskManager.Run("Enqueue procedure for instance " + instance.shortName, () => instance.procedureManager.Enqueue(procedure, immediate)); | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,35 +0,0 @@ | ||||
| using Phantom.Agent.Minecraft.Launcher; | ||||
| using Phantom.Agent.Services.Instances.States; | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances; | ||||
|  | ||||
| abstract class InstanceContext { | ||||
| 	public InstanceServices Services { get; } | ||||
| 	public InstanceConfiguration Configuration { get; } | ||||
| 	public IServerLauncher Launcher { get; } | ||||
| 	 | ||||
| 	public abstract ILogger Logger { get; } | ||||
| 	public abstract string ShortName { get; } | ||||
|  | ||||
| 	protected InstanceContext(InstanceServices services, InstanceConfiguration configuration, IServerLauncher launcher) { | ||||
| 		Services = services; | ||||
| 		Configuration = configuration; | ||||
| 		Launcher = launcher; | ||||
| 	} | ||||
|  | ||||
| 	public abstract void SetStatus(IInstanceStatus newStatus); | ||||
|  | ||||
| 	public void SetLaunchFailedStatusAndReportEvent(InstanceLaunchFailReason reason) { | ||||
| 		SetStatus(InstanceStatus.Failed(reason)); | ||||
| 		ReportEvent(new InstanceLaunchFailedEvent(reason)); | ||||
| 	} | ||||
|  | ||||
| 	public abstract void ReportEvent(IInstanceEvent instanceEvent); | ||||
| 	public abstract void TransitionState(Func<(IInstanceState, IInstanceStatus?)> newStateAndStatus); | ||||
|  | ||||
| 	public void TransitionState(IInstanceState newState, IInstanceStatus? newStatus = null) { | ||||
| 		TransitionState(() => (newState, newStatus)); | ||||
| 	} | ||||
| } | ||||
| @@ -1,71 +1,84 @@ | ||||
| using System.Collections.Immutable; | ||||
| using System.Threading.Channels; | ||||
| using Phantom.Agent.Rpc; | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Common.Messages.ToServer; | ||||
| using Phantom.Utils.Collections; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Common.Messages.Agent.ToController; | ||||
| using Phantom.Utils.Tasks; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances; | ||||
|  | ||||
| sealed class InstanceLogSender : CancellableBackgroundTask { | ||||
| 	private static readonly BoundedChannelOptions BufferOptions = new (capacity: 64) { | ||||
| 		SingleReader = true, | ||||
| 		SingleWriter = true, | ||||
| 		FullMode = BoundedChannelFullMode.DropNewest | ||||
| 	}; | ||||
| 	 | ||||
| 	private static readonly TimeSpan SendDelay = TimeSpan.FromMilliseconds(200); | ||||
|  | ||||
| 	private readonly ControllerConnection controllerConnection; | ||||
| 	private readonly Guid instanceGuid; | ||||
| 	private readonly Channel<string> outputChannel; | ||||
| 	 | ||||
| 	private int droppedLinesSinceLastSend; | ||||
|  | ||||
| 	private readonly SemaphoreSlim semaphore = new (1, 1); | ||||
| 	private readonly RingBuffer<string> buffer = new (1000); | ||||
|  | ||||
| 	public InstanceLogSender(TaskManager taskManager, Guid instanceGuid, string loggerName) : base(PhantomLogger.Create<InstanceLogSender>(loggerName), taskManager, "Instance log sender for " + loggerName) { | ||||
| 	public InstanceLogSender(ControllerConnection controllerConnection, TaskManager taskManager, Guid instanceGuid, string loggerName) : base(PhantomLogger.Create<InstanceLogSender>(loggerName), taskManager, "Instance log sender for " + loggerName) { | ||||
| 		this.controllerConnection = controllerConnection; | ||||
| 		this.instanceGuid = instanceGuid; | ||||
| 		this.outputChannel = Channel.CreateBounded<string>(BufferOptions, OnLineDropped); | ||||
| 		Start(); | ||||
| 	} | ||||
| 	 | ||||
| 	protected override async Task RunTask() { | ||||
| 		var lineReader = outputChannel.Reader; | ||||
| 		var lineBuilder = ImmutableArray.CreateBuilder<string>(); | ||||
| 		 | ||||
| 		try { | ||||
| 			while (!CancellationToken.IsCancellationRequested) { | ||||
| 				await SendOutputToServer(await DequeueOrThrow()); | ||||
| 			while (await lineReader.WaitToReadAsync(CancellationToken)) { | ||||
| 				await Task.Delay(SendDelay, CancellationToken); | ||||
| 				await SendOutputToServer(ReadLinesFromChannel(lineReader, lineBuilder)); | ||||
| 			} | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			// Ignore. | ||||
| 		} | ||||
|  | ||||
| 		// Flush remaining lines. | ||||
| 		await SendOutputToServer(DequeueWithoutSemaphore()); | ||||
| 		await SendOutputToServer(ReadLinesFromChannel(lineReader, lineBuilder)); | ||||
| 	} | ||||
|  | ||||
| 	private ImmutableArray<string> ReadLinesFromChannel(ChannelReader<string> reader, ImmutableArray<string>.Builder builder) { | ||||
| 		builder.Clear(); | ||||
| 		 | ||||
| 		while (reader.TryRead(out string? line)) { | ||||
| 			builder.Add(line); | ||||
| 		} | ||||
| 		 | ||||
| 		int droppedLines = Interlocked.Exchange(ref droppedLinesSinceLastSend, 0); | ||||
| 		if (droppedLines > 0) { | ||||
| 			builder.Add($"Dropped {droppedLines} {(droppedLines == 1 ? "line" : "lines")} due to buffer overflow."); | ||||
| 		} | ||||
| 		 | ||||
| 		return builder.ToImmutable(); | ||||
| 	} | ||||
|  | ||||
| 	private async Task SendOutputToServer(ImmutableArray<string> lines) { | ||||
| 		if (!lines.IsEmpty) { | ||||
| 			await ServerMessaging.Send(new InstanceOutputMessage(instanceGuid, lines)); | ||||
| 			await controllerConnection.Send(new InstanceOutputMessage(instanceGuid, lines)); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private ImmutableArray<string> DequeueWithoutSemaphore() { | ||||
| 		ImmutableArray<string> lines = buffer.Count > 0 ? buffer.EnumerateLast(uint.MaxValue).ToImmutableArray() : ImmutableArray<string>.Empty; | ||||
| 		buffer.Clear(); | ||||
| 		return lines; | ||||
| 	} | ||||
|  | ||||
| 	private async Task<ImmutableArray<string>> DequeueOrThrow() { | ||||
| 		await semaphore.WaitAsync(CancellationToken); | ||||
|  | ||||
| 		try { | ||||
| 			return DequeueWithoutSemaphore(); | ||||
| 		} finally { | ||||
| 			semaphore.Release(); | ||||
| 		} | ||||
| 	private void OnLineDropped(string line) { | ||||
| 		Logger.Warning("Buffer is full, dropped line: {Line}", line); | ||||
| 		Interlocked.Increment(ref droppedLinesSinceLastSend); | ||||
| 	} | ||||
|  | ||||
| 	public void Enqueue(string line) { | ||||
| 		try { | ||||
| 			semaphore.Wait(CancellationToken); | ||||
| 		} catch (Exception) { | ||||
| 			return; | ||||
| 		} | ||||
| 		outputChannel.Writer.TryWrite(line); | ||||
| 	} | ||||
|  | ||||
| 		try { | ||||
| 			buffer.Add(line); | ||||
| 		} finally { | ||||
| 			semaphore.Release(); | ||||
| 	protected override void Dispose() { | ||||
| 		if (!outputChannel.Writer.TryComplete()) { | ||||
| 			Logger.Error("Could not mark channel as completed."); | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -0,0 +1,85 @@ | ||||
| using Phantom.Agent.Services.Instances.Procedures; | ||||
| using Phantom.Common.Data.Minecraft; | ||||
| using Phantom.Utils.Tasks; | ||||
| using Phantom.Utils.Threading; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances; | ||||
|  | ||||
| sealed class InstanceProcedureManager : IAsyncDisposable { | ||||
| 	private readonly record struct CurrentProcedure(IInstanceProcedure Procedure, CancellationTokenSource CancellationTokenSource); | ||||
|  | ||||
| 	private readonly ThreadSafeStructRef<CurrentProcedure> currentProcedure = new (); | ||||
| 	private readonly ThreadSafeLinkedList<IInstanceProcedure> procedureQueue = new (); | ||||
| 	private readonly AutoResetEvent procedureQueueReady = new (false); | ||||
| 	private readonly ManualResetEventSlim procedureQueueFinished = new (false); | ||||
|  | ||||
| 	private readonly Instance instance; | ||||
| 	private readonly IInstanceContext context; | ||||
| 	private readonly CancellationTokenSource shutdownCancellationTokenSource = new (); | ||||
|  | ||||
| 	public InstanceProcedureManager(Instance instance, IInstanceContext context, TaskManager taskManager) { | ||||
| 		this.instance = instance; | ||||
| 		this.context = context; | ||||
| 		taskManager.Run("Procedure manager for instance " + context.ShortName, Run); | ||||
| 	} | ||||
|  | ||||
| 	public async Task Enqueue(IInstanceProcedure procedure, bool immediate = false) { | ||||
| 		await procedureQueue.Add(procedure, toFront: immediate, shutdownCancellationTokenSource.Token); | ||||
| 		procedureQueueReady.Set(); | ||||
| 	} | ||||
|  | ||||
| 	public async Task<IInstanceProcedure?> GetCurrentProcedure(CancellationToken cancellationToken) { | ||||
| 		return (await currentProcedure.Get(cancellationToken))?.Procedure; | ||||
| 	} | ||||
|  | ||||
| 	private async Task Run() { | ||||
| 		try { | ||||
| 			var shutdownCancellationToken = shutdownCancellationTokenSource.Token; | ||||
| 			while (true) { | ||||
| 				await procedureQueueReady.WaitOneAsync(shutdownCancellationToken); | ||||
| 				while (await procedureQueue.TryTakeFromFront(shutdownCancellationToken) is {} nextProcedure) { | ||||
| 					using var procedureCancellationTokenSource = new CancellationTokenSource(); | ||||
| 					await currentProcedure.Set(new CurrentProcedure(nextProcedure, procedureCancellationTokenSource), shutdownCancellationToken); | ||||
| 					await RunProcedure(nextProcedure, procedureCancellationTokenSource.Token); | ||||
| 					await currentProcedure.Set(null, shutdownCancellationToken); | ||||
| 				} | ||||
| 			} | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			// Ignore. | ||||
| 		} | ||||
|  | ||||
| 		await RunProcedure(new StopInstanceProcedure(MinecraftStopStrategy.Instant), CancellationToken.None); | ||||
| 		procedureQueueFinished.Set(); | ||||
| 	} | ||||
|  | ||||
| 	private async Task RunProcedure(IInstanceProcedure procedure, CancellationToken cancellationToken) { | ||||
| 		var procedureName = procedure.GetType().Name; | ||||
|  | ||||
| 		context.Logger.Debug("Started procedure: {Procedure}", procedureName); | ||||
| 		try { | ||||
| 			var newState = await procedure.Run(context, cancellationToken); | ||||
| 			context.Logger.Debug("Finished procedure: {Procedure}", procedureName); | ||||
|  | ||||
| 			if (newState != null) { | ||||
| 				instance.TransitionState(newState); | ||||
| 			} | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			context.Logger.Debug("Cancelled procedure: {Procedure}", procedureName); | ||||
| 		} catch (Exception e) { | ||||
| 			context.Logger.Error(e, "Caught exception while running procedure: {Procedure}", procedureName); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	public async ValueTask DisposeAsync() { | ||||
| 		shutdownCancellationTokenSource.Cancel(); | ||||
|  | ||||
| 		(await currentProcedure.Get(CancellationToken.None))?.CancellationTokenSource.Cancel(); | ||||
| 		await procedureQueueFinished.WaitHandle.WaitOneAsync(); | ||||
|  | ||||
| 		currentProcedure.Dispose(); | ||||
| 		procedureQueue.Dispose(); | ||||
| 		procedureQueueReady.Dispose(); | ||||
| 		procedureQueueFinished.Dispose(); | ||||
| 		shutdownCancellationTokenSource.Dispose(); | ||||
| 	} | ||||
| } | ||||
| @@ -1,7 +1,8 @@ | ||||
| using Phantom.Agent.Minecraft.Launcher; | ||||
| using Phantom.Agent.Rpc; | ||||
| using Phantom.Agent.Services.Backups; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Utils.Tasks; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances;  | ||||
| namespace Phantom.Agent.Services.Instances; | ||||
|  | ||||
| sealed record InstanceServices(TaskManager TaskManager, PortManager PortManager, BackupManager BackupManager, LaunchServices LaunchServices); | ||||
| sealed record InstanceServices(ControllerConnection ControllerConnection, TaskManager TaskManager, PortManager PortManager, BackupManager BackupManager, LaunchServices LaunchServices); | ||||
|   | ||||
| @@ -14,16 +14,17 @@ using Phantom.Common.Data.Instance; | ||||
| using Phantom.Common.Data.Minecraft; | ||||
| using Phantom.Common.Data.Replies; | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Common.Messages.ToServer; | ||||
| using Phantom.Common.Messages.Agent.ToController; | ||||
| using Phantom.Utils.IO; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Utils.Tasks; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances; | ||||
|  | ||||
| sealed class InstanceSessionManager : IDisposable { | ||||
| sealed class InstanceSessionManager : IAsyncDisposable { | ||||
| 	private static readonly ILogger Logger = PhantomLogger.Create<InstanceSessionManager>(); | ||||
| 	 | ||||
|  | ||||
| 	private readonly ControllerConnection controllerConnection; | ||||
| 	private readonly AgentInfo agentInfo; | ||||
| 	private readonly string basePath; | ||||
|  | ||||
| @@ -34,16 +35,19 @@ sealed class InstanceSessionManager : IDisposable { | ||||
| 	private readonly CancellationToken shutdownCancellationToken; | ||||
| 	private readonly SemaphoreSlim semaphore = new (1, 1); | ||||
|  | ||||
| 	public InstanceSessionManager(AgentInfo agentInfo, AgentFolders agentFolders, JavaRuntimeRepository javaRuntimeRepository, TaskManager taskManager, BackupManager backupManager) { | ||||
| 	private uint instanceLoggerSequenceId = 0; | ||||
|  | ||||
| 	public InstanceSessionManager(ControllerConnection controllerConnection, AgentInfo agentInfo, AgentFolders agentFolders, JavaRuntimeRepository javaRuntimeRepository, TaskManager taskManager, BackupManager backupManager) { | ||||
| 		this.controllerConnection = controllerConnection; | ||||
| 		this.agentInfo = agentInfo; | ||||
| 		this.basePath = agentFolders.InstancesFolderPath; | ||||
| 		this.shutdownCancellationToken = shutdownCancellationTokenSource.Token; | ||||
| 		 | ||||
|  | ||||
| 		var minecraftServerExecutables = new MinecraftServerExecutables(agentFolders.ServerExecutableFolderPath); | ||||
| 		var launchServices = new LaunchServices(minecraftServerExecutables, javaRuntimeRepository); | ||||
| 		var portManager = new PortManager(agentInfo.AllowedServerPorts, agentInfo.AllowedRconPorts); | ||||
| 		 | ||||
| 		this.instanceServices = new InstanceServices(taskManager, portManager, backupManager, launchServices); | ||||
|  | ||||
| 		this.instanceServices = new InstanceServices(controllerConnection, taskManager, portManager, backupManager, launchServices); | ||||
| 	} | ||||
|  | ||||
| 	private async Task<InstanceActionResult<T>> AcquireSemaphoreAndRun<T>(Func<Task<InstanceActionResult<T>>> func) { | ||||
| @@ -76,7 +80,7 @@ sealed class InstanceSessionManager : IDisposable { | ||||
| 			var instanceGuid = configuration.InstanceGuid; | ||||
| 			var instanceFolder = Path.Combine(basePath, instanceGuid.ToString()); | ||||
| 			Directories.Create(instanceFolder, Chmod.URWX_GRX); | ||||
| 			 | ||||
|  | ||||
| 			var heapMegabytes = configuration.MemoryAllocation.InMegabytes; | ||||
| 			var jvmProperties = new JvmProperties( | ||||
| 				InitialHeapMegabytes: heapMegabytes / 2, | ||||
| @@ -103,15 +107,15 @@ sealed class InstanceSessionManager : IDisposable { | ||||
| 			if (instances.TryGetValue(instanceGuid, out var instance)) { | ||||
| 				await instance.Reconfigure(configuration, launcher, shutdownCancellationToken); | ||||
| 				Logger.Information("Reconfigured instance \"{Name}\" (GUID {Guid}).", configuration.InstanceName, configuration.InstanceGuid); | ||||
| 				 | ||||
|  | ||||
| 				if (alwaysReportStatus) { | ||||
| 					instance.ReportLastStatus(); | ||||
| 				} | ||||
| 			} | ||||
| 			else { | ||||
| 				instances[instanceGuid] = instance = new Instance(instanceServices, configuration, launcher); | ||||
| 				instances[instanceGuid] = instance = new Instance(GetInstanceLoggerName(instanceGuid), instanceServices, configuration, launcher); | ||||
| 				Logger.Information("Created instance \"{Name}\" (GUID {Guid}).", configuration.InstanceName, configuration.InstanceGuid); | ||||
| 				 | ||||
|  | ||||
| 				instance.ReportLastStatus(); | ||||
| 				instance.IsRunningChanged += OnInstanceIsRunningChanged; | ||||
| 			} | ||||
| @@ -124,6 +128,11 @@ sealed class InstanceSessionManager : IDisposable { | ||||
| 		}); | ||||
| 	} | ||||
|  | ||||
| 	private string GetInstanceLoggerName(Guid guid) { | ||||
| 		var prefix = guid.ToString(); | ||||
| 		return prefix[..prefix.IndexOf('-')] + "/" + Interlocked.Increment(ref instanceLoggerSequenceId); | ||||
| 	} | ||||
|  | ||||
| 	private ImmutableArray<Instance> GetRunningInstancesInternal() { | ||||
| 		return instances.Values.Where(static instance => instance.IsRunning).ToImmutableArray(); | ||||
| 	} | ||||
| @@ -139,7 +148,7 @@ sealed class InstanceSessionManager : IDisposable { | ||||
| 				var runningInstances = GetRunningInstancesInternal(); | ||||
| 				var runningInstanceCount = runningInstances.Length; | ||||
| 				var runningInstanceMemory = runningInstances.Aggregate(RamAllocationUnits.Zero, static (total, instance) => total + instance.Configuration.MemoryAllocation); | ||||
| 				await ServerMessaging.Send(new ReportAgentStatusMessage(runningInstanceCount, runningInstanceMemory)); | ||||
| 				await controllerConnection.Send(new ReportAgentStatusMessage(runningInstanceCount, runningInstanceMemory)); | ||||
| 			} finally { | ||||
| 				semaphore.Release(); | ||||
| 			} | ||||
| @@ -167,38 +176,23 @@ sealed class InstanceSessionManager : IDisposable { | ||||
| 	} | ||||
|  | ||||
| 	public Task<InstanceActionResult<StopInstanceResult>> Stop(Guid instanceGuid, MinecraftStopStrategy stopStrategy) { | ||||
| 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, instance => instance.Stop(stopStrategy)); | ||||
| 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, instance => instance.Stop(stopStrategy, shutdownCancellationToken)); | ||||
| 	} | ||||
|  | ||||
| 	public Task<InstanceActionResult<SendCommandToInstanceResult>> SendCommand(Guid instanceGuid, string command) { | ||||
| 		return AcquireSemaphoreAndRunWithInstance(instanceGuid, async instance => await instance.SendCommand(command, shutdownCancellationToken) ? SendCommandToInstanceResult.Success : SendCommandToInstanceResult.UnknownError); | ||||
| 	} | ||||
|  | ||||
| 	public async Task StopAll() { | ||||
| 		shutdownCancellationTokenSource.Cancel(); | ||||
| 		 | ||||
| 	public async ValueTask DisposeAsync() { | ||||
| 		Logger.Information("Stopping all instances..."); | ||||
| 		 | ||||
| 		shutdownCancellationTokenSource.Cancel(); | ||||
|  | ||||
| 		await semaphore.WaitAsync(CancellationToken.None); | ||||
| 		try { | ||||
| 			await Task.WhenAll(instances.Values.Select(static instance => instance.StopAndWait(TimeSpan.FromSeconds(30)))); | ||||
| 			DisposeAllInstances(); | ||||
| 		} finally { | ||||
| 			semaphore.Release(); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	public void Dispose() { | ||||
| 		DisposeAllInstances(); | ||||
| 		await Task.WhenAll(instances.Values.Select(static instance => instance.DisposeAsync().AsTask())); | ||||
| 		instances.Clear(); | ||||
| 		 | ||||
| 		shutdownCancellationTokenSource.Dispose(); | ||||
| 		semaphore.Dispose(); | ||||
| 	} | ||||
|  | ||||
| 	private void DisposeAllInstances() { | ||||
| 		foreach (var (_, instance) in instances) { | ||||
| 			instance.Dispose(); | ||||
| 		} | ||||
|  | ||||
| 		instances.Clear(); | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -0,0 +1,29 @@ | ||||
| using Phantom.Agent.Services.Backups; | ||||
| using Phantom.Agent.Services.Instances.States; | ||||
| using Phantom.Common.Data.Backups; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.Procedures; | ||||
|  | ||||
| sealed record BackupInstanceProcedure(BackupManager BackupManager) : IInstanceProcedure { | ||||
| 	private readonly TaskCompletionSource<BackupCreationResult> resultCompletionSource = new (); | ||||
|  | ||||
| 	public Task<BackupCreationResult> Result => resultCompletionSource.Task; | ||||
|  | ||||
| 	public async Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken) { | ||||
| 		if (context.CurrentState is not InstanceRunningState runningState || runningState.Process.HasEnded) { | ||||
| 			resultCompletionSource.SetResult(new BackupCreationResult(BackupCreationResultKind.InstanceNotRunning)); | ||||
| 			return null; | ||||
| 		} | ||||
|  | ||||
| 		try { | ||||
| 			var result = await BackupManager.CreateBackup(context.ShortName, runningState.Process, cancellationToken); | ||||
| 			resultCompletionSource.SetResult(result); | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			resultCompletionSource.SetCanceled(cancellationToken); | ||||
| 		} catch (Exception e) { | ||||
| 			resultCompletionSource.SetException(e); | ||||
| 		} | ||||
|  | ||||
| 		return null; | ||||
| 	} | ||||
| } | ||||
| @@ -0,0 +1,7 @@ | ||||
| using Phantom.Agent.Services.Instances.States; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.Procedures; | ||||
|  | ||||
| interface IInstanceProcedure { | ||||
| 	Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken); | ||||
| } | ||||
| @@ -0,0 +1,97 @@ | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Agent.Minecraft.Launcher; | ||||
| using Phantom.Agent.Minecraft.Server; | ||||
| using Phantom.Agent.Services.Instances.States; | ||||
| using Phantom.Common.Data.Instance; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.Procedures; | ||||
|  | ||||
| sealed record LaunchInstanceProcedure(InstanceConfiguration Configuration, IServerLauncher Launcher, bool IsRestarting = false) : IInstanceProcedure { | ||||
| 	public async Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken) { | ||||
| 		if (!IsRestarting && context.CurrentState is InstanceRunningState) { | ||||
| 			return null; | ||||
| 		} | ||||
| 		 | ||||
| 		context.SetStatus(IsRestarting ? InstanceStatus.Restarting : InstanceStatus.Launching); | ||||
|  | ||||
| 		InstanceLaunchFailReason? failReason = context.Services.PortManager.Reserve(Configuration) switch { | ||||
| 			PortManager.Result.ServerPortNotAllowed   => InstanceLaunchFailReason.ServerPortNotAllowed, | ||||
| 			PortManager.Result.ServerPortAlreadyInUse => InstanceLaunchFailReason.ServerPortAlreadyInUse, | ||||
| 			PortManager.Result.RconPortNotAllowed     => InstanceLaunchFailReason.RconPortNotAllowed, | ||||
| 			PortManager.Result.RconPortAlreadyInUse   => InstanceLaunchFailReason.RconPortAlreadyInUse, | ||||
| 			_                                         => null | ||||
| 		}; | ||||
|  | ||||
| 		if (failReason is {} reason) { | ||||
| 			context.SetLaunchFailedStatusAndReportEvent(reason); | ||||
| 			return new InstanceNotRunningState(); | ||||
| 		} | ||||
| 		 | ||||
| 		context.Logger.Information("Session starting..."); | ||||
| 		try { | ||||
| 			InstanceProcess process = await DoLaunch(context, cancellationToken); | ||||
| 			return new InstanceRunningState(Configuration, Launcher, process, context); | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			context.SetStatus(InstanceStatus.NotRunning); | ||||
| 		} catch (LaunchFailureException e) { | ||||
| 			context.Logger.Error(e.LogMessage); | ||||
| 			context.SetLaunchFailedStatusAndReportEvent(e.Reason); | ||||
| 		} catch (Exception e) { | ||||
| 			context.Logger.Error(e, "Caught exception while launching instance."); | ||||
| 			context.SetLaunchFailedStatusAndReportEvent(InstanceLaunchFailReason.UnknownError); | ||||
| 		} | ||||
| 		 | ||||
| 		context.Services.PortManager.Release(Configuration); | ||||
| 		return new InstanceNotRunningState(); | ||||
| 	} | ||||
|  | ||||
| 	private async Task<InstanceProcess> DoLaunch(IInstanceContext context, CancellationToken cancellationToken) { | ||||
| 		cancellationToken.ThrowIfCancellationRequested(); | ||||
|  | ||||
| 		byte lastDownloadProgress = byte.MaxValue; | ||||
|  | ||||
| 		void OnDownloadProgress(object? sender, DownloadProgressEventArgs args) { | ||||
| 			byte progress = (byte) Math.Min(args.DownloadedBytes * 100 / args.TotalBytes, 100); | ||||
|  | ||||
| 			if (lastDownloadProgress != progress) { | ||||
| 				lastDownloadProgress = progress; | ||||
| 				context.SetStatus(InstanceStatus.Downloading(progress)); | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		var launchResult = await Launcher.Launch(context.Logger, context.Services.LaunchServices, OnDownloadProgress, cancellationToken); | ||||
| 		if (launchResult is LaunchResult.InvalidJavaRuntime) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.JavaRuntimeNotFound, "Session failed to launch, invalid Java runtime."); | ||||
| 		} | ||||
| 		else if (launchResult is LaunchResult.CouldNotDownloadMinecraftServer) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotDownloadMinecraftServer, "Session failed to launch, could not download Minecraft server."); | ||||
| 		} | ||||
| 		else if (launchResult is LaunchResult.CouldNotPrepareMinecraftServerLauncher) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotPrepareMinecraftServerLauncher, "Session failed to launch, could not prepare Minecraft server launcher."); | ||||
| 		} | ||||
| 		else if (launchResult is LaunchResult.CouldNotConfigureMinecraftServer) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotConfigureMinecraftServer, "Session failed to launch, could not configure Minecraft server."); | ||||
| 		} | ||||
| 		else if (launchResult is LaunchResult.CouldNotStartMinecraftServer) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotStartMinecraftServer, "Session failed to launch, could not start Minecraft server."); | ||||
| 		} | ||||
|  | ||||
| 		if (launchResult is not LaunchResult.Success launchSuccess) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.UnknownError, "Session failed to launch."); | ||||
| 		} | ||||
|  | ||||
| 		context.SetStatus(InstanceStatus.Running); | ||||
| 		context.ReportEvent(InstanceEvent.LaunchSucceded); | ||||
| 		return launchSuccess.Process; | ||||
| 	} | ||||
|  | ||||
| 	private sealed class LaunchFailureException : Exception { | ||||
| 		public InstanceLaunchFailReason Reason { get; } | ||||
| 		public string LogMessage { get; } | ||||
|  | ||||
| 		public LaunchFailureException(InstanceLaunchFailReason reason, string logMessage) { | ||||
| 			this.Reason = reason; | ||||
| 			this.LogMessage = logMessage; | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -0,0 +1,17 @@ | ||||
| using Phantom.Agent.Services.Instances.States; | ||||
| using Phantom.Common.Data.Instance; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.Procedures; | ||||
|  | ||||
| sealed record SetInstanceToNotRunningStateProcedure(IInstanceStatus Status) : IInstanceProcedure { | ||||
| 	public Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken) { | ||||
| 		if (context.CurrentState is InstanceRunningState { Process.HasEnded: true }) { | ||||
| 			context.SetStatus(Status); | ||||
| 			context.ReportEvent(InstanceEvent.Stopped); | ||||
| 			return Task.FromResult<IInstanceState?>(new InstanceNotRunningState()); | ||||
| 		} | ||||
| 		else { | ||||
| 			return Task.FromResult<IInstanceState?>(null); | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -0,0 +1,103 @@ | ||||
| using System.Diagnostics; | ||||
| using Phantom.Agent.Minecraft.Command; | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Agent.Services.Instances.States; | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Phantom.Common.Data.Minecraft; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.Procedures; | ||||
|  | ||||
| sealed record StopInstanceProcedure(MinecraftStopStrategy StopStrategy) : IInstanceProcedure { | ||||
| 	private static readonly ushort[] Stops = { 60, 30, 10, 5, 4, 3, 2, 1, 0 }; | ||||
|  | ||||
| 	public async Task<IInstanceState?> Run(IInstanceContext context, CancellationToken cancellationToken) { | ||||
| 		if (context.CurrentState is not InstanceRunningState runningState) { | ||||
| 			return null; | ||||
| 		} | ||||
|  | ||||
| 		var process = runningState.Process; | ||||
|  | ||||
| 		runningState.IsStopping = true; | ||||
| 		context.SetStatus(InstanceStatus.Stopping); | ||||
|  | ||||
| 		var seconds = StopStrategy.Seconds; | ||||
| 		if (seconds > 0) { | ||||
| 			try { | ||||
| 				await CountDownWithAnnouncements(context, process, seconds, cancellationToken); | ||||
| 			} catch (OperationCanceledException) { | ||||
| 				runningState.IsStopping = false; | ||||
| 				return null; | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		try { | ||||
| 			// Too late to cancel the stop procedure now. | ||||
| 			if (!process.HasEnded) { | ||||
| 				context.Logger.Information("Session stopping now."); | ||||
| 				await DoStop(context, process); | ||||
| 			} | ||||
| 		} finally { | ||||
| 			context.Logger.Information("Session stopped."); | ||||
| 			context.SetStatus(InstanceStatus.NotRunning); | ||||
| 			context.ReportEvent(InstanceEvent.Stopped); | ||||
| 		} | ||||
|  | ||||
| 		return new InstanceNotRunningState(); | ||||
| 	} | ||||
|  | ||||
| 	private async Task CountDownWithAnnouncements(IInstanceContext context, InstanceProcess process, ushort seconds, CancellationToken cancellationToken) { | ||||
| 		context.Logger.Information("Session stopping in {Seconds} seconds.", seconds); | ||||
|  | ||||
| 		foreach (var stop in Stops) { | ||||
| 			// TODO change to event-based cancellation | ||||
| 			if (process.HasEnded) { | ||||
| 				return; | ||||
| 			} | ||||
|  | ||||
| 			if (seconds > stop) { | ||||
| 				await process.SendCommand(GetCountDownAnnouncementCommand(seconds), cancellationToken); | ||||
| 				await Task.Delay(TimeSpan.FromSeconds(seconds - stop), cancellationToken); | ||||
| 				seconds = stop; | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private static string GetCountDownAnnouncementCommand(ushort seconds) { | ||||
| 		return MinecraftCommand.Say("Server shutting down in " + seconds + (seconds == 1 ? " second." : " seconds.")); | ||||
| 	} | ||||
|  | ||||
| 	private async Task DoStop(IInstanceContext context, InstanceProcess process) { | ||||
| 		context.Logger.Information("Sending stop command..."); | ||||
| 		await TrySendStopCommand(context, process); | ||||
|  | ||||
| 		context.Logger.Information("Waiting for session to end..."); | ||||
| 		await WaitForSessionToEnd(context, process); | ||||
| 	} | ||||
|  | ||||
| 	private async Task TrySendStopCommand(IInstanceContext context, InstanceProcess process) { | ||||
| 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(5)); | ||||
| 		try { | ||||
| 			await process.SendCommand(MinecraftCommand.Stop, timeout.Token); | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			// Ignore. | ||||
| 		} catch (ObjectDisposedException e) when (e.ObjectName == typeof(Process).FullName && process.HasEnded) { | ||||
| 			// Ignore. | ||||
| 		} catch (Exception e) { | ||||
| 			context.Logger.Warning(e, "Caught exception while sending stop command."); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private async Task WaitForSessionToEnd(IInstanceContext context, InstanceProcess process) { | ||||
| 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(55)); | ||||
| 		try { | ||||
| 			await process.WaitForExit(timeout.Token); | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			try { | ||||
| 				context.Logger.Warning("Waiting timed out, killing session..."); | ||||
| 				process.Kill(); | ||||
| 			} catch (Exception e) { | ||||
| 				context.Logger.Error(e, "Caught exception while killing session."); | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
| @@ -1,28 +0,0 @@ | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.Sessions;  | ||||
|  | ||||
| sealed class InstanceSession : IDisposable { | ||||
| 	private readonly InstanceProcess process; | ||||
| 	private readonly InstanceContext context; | ||||
| 	private readonly InstanceLogSender logSender; | ||||
|  | ||||
| 	public InstanceSession(InstanceProcess process, InstanceContext context) { | ||||
| 		this.process = process; | ||||
| 		this.context = context; | ||||
| 		this.logSender = new InstanceLogSender(context.Services.TaskManager, context.Configuration.InstanceGuid, context.ShortName); | ||||
| 		 | ||||
| 		this.process.AddOutputListener(SessionOutput); | ||||
| 	} | ||||
| 	 | ||||
| 	private void SessionOutput(object? sender, string line) { | ||||
| 		context.Logger.Debug("[Server] {Line}", line); | ||||
| 		logSender.Enqueue(line); | ||||
| 	} | ||||
| 	 | ||||
| 	public void Dispose() { | ||||
| 		logSender.Stop(); | ||||
| 		process.Dispose(); | ||||
| 		context.Services.PortManager.Release(context.Configuration); | ||||
| 	} | ||||
| } | ||||
| @@ -1,11 +1,6 @@ | ||||
| using Phantom.Common.Data.Minecraft; | ||||
| using Phantom.Common.Data.Replies; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.States;  | ||||
| namespace Phantom.Agent.Services.Instances.States; | ||||
|  | ||||
| interface IInstanceState { | ||||
| 	void Initialize(); | ||||
| 	(IInstanceState, LaunchInstanceResult) Launch(InstanceContext context); | ||||
| 	(IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy); | ||||
| 	Task<bool> SendCommand(string command, CancellationToken cancellationToken); | ||||
| } | ||||
|   | ||||
| @@ -1,127 +0,0 @@ | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Agent.Minecraft.Launcher; | ||||
| using Phantom.Agent.Minecraft.Server; | ||||
| using Phantom.Agent.Services.Instances.Sessions; | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Phantom.Common.Data.Minecraft; | ||||
| using Phantom.Common.Data.Replies; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.States; | ||||
|  | ||||
| sealed class InstanceLaunchingState : IInstanceState, IDisposable { | ||||
| 	private readonly InstanceContext context; | ||||
| 	private readonly CancellationTokenSource cancellationTokenSource = new (); | ||||
| 	private byte lastDownloadProgress = byte.MaxValue; | ||||
|  | ||||
| 	public InstanceLaunchingState(InstanceContext context) { | ||||
| 		this.context = context; | ||||
| 	} | ||||
|  | ||||
| 	public void Initialize() { | ||||
| 		context.Logger.Information("Session starting..."); | ||||
|  | ||||
| 		var launchTask = context.Services.TaskManager.Run("Launch procedure for instance " + context.ShortName, DoLaunch); | ||||
| 		launchTask.ContinueWith(OnLaunchSuccess, CancellationToken.None, TaskContinuationOptions.OnlyOnRanToCompletion, TaskScheduler.Default); | ||||
| 		launchTask.ContinueWith(OnLaunchFailure, CancellationToken.None, TaskContinuationOptions.NotOnRanToCompletion, TaskScheduler.Default); | ||||
| 	} | ||||
|  | ||||
| 	private async Task<InstanceProcess> DoLaunch() { | ||||
| 		var cancellationToken = cancellationTokenSource.Token; | ||||
| 		cancellationToken.ThrowIfCancellationRequested(); | ||||
|  | ||||
| 		void OnDownloadProgress(object? sender, DownloadProgressEventArgs args) { | ||||
| 			byte progress = (byte) Math.Min(args.DownloadedBytes * 100 / args.TotalBytes, 100); | ||||
|  | ||||
| 			if (lastDownloadProgress != progress) { | ||||
| 				lastDownloadProgress = progress; | ||||
| 				context.SetStatus(InstanceStatus.Downloading(progress)); | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		var launchResult = await context.Launcher.Launch(context.Logger, context.Services.LaunchServices, OnDownloadProgress, cancellationToken); | ||||
| 		if (launchResult is LaunchResult.InvalidJavaRuntime) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.JavaRuntimeNotFound, "Session failed to launch, invalid Java runtime."); | ||||
| 		} | ||||
| 		else if (launchResult is LaunchResult.InvalidJvmArguments) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.InvalidJvmArguments, "Session failed to launch, invalid JVM arguments."); | ||||
| 		} | ||||
| 		else if (launchResult is LaunchResult.CouldNotDownloadMinecraftServer) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotDownloadMinecraftServer, "Session failed to launch, could not download Minecraft server."); | ||||
| 		} | ||||
| 		else if (launchResult is LaunchResult.CouldNotPrepareMinecraftServerLauncher) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotPrepareMinecraftServerLauncher, "Session failed to launch, could not prepare Minecraft server launcher."); | ||||
| 		} | ||||
| 		else if (launchResult is LaunchResult.CouldNotConfigureMinecraftServer) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotConfigureMinecraftServer, "Session failed to launch, could not configure Minecraft server."); | ||||
| 		} | ||||
| 		else if (launchResult is LaunchResult.CouldNotStartMinecraftServer) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.CouldNotStartMinecraftServer, "Session failed to launch, could not start Minecraft server."); | ||||
| 		} | ||||
|  | ||||
| 		if (launchResult is not LaunchResult.Success launchSuccess) { | ||||
| 			throw new LaunchFailureException(InstanceLaunchFailReason.UnknownError, "Session failed to launch."); | ||||
| 		} | ||||
|  | ||||
| 		context.SetStatus(InstanceStatus.Launching); | ||||
| 		return launchSuccess.Process; | ||||
| 	} | ||||
|  | ||||
| 	private void OnLaunchSuccess(Task<InstanceProcess> task) { | ||||
| 		context.TransitionState(() => { | ||||
| 			context.ReportEvent(InstanceEvent.LaunchSucceded); | ||||
| 			 | ||||
| 			var process = task.Result; | ||||
| 			var session = new InstanceSession(process, context); | ||||
| 			 | ||||
| 			if (cancellationTokenSource.IsCancellationRequested) { | ||||
| 				return (new InstanceStoppingState(context, process, session), InstanceStatus.Stopping); | ||||
| 			} | ||||
| 			else { | ||||
| 				return (new InstanceRunningState(context, process, session), null); | ||||
| 			} | ||||
| 		}); | ||||
| 	} | ||||
|  | ||||
| 	private void OnLaunchFailure(Task task) { | ||||
| 		if (task.IsFaulted) { | ||||
| 			if (task.Exception is { InnerException: LaunchFailureException e }) { | ||||
| 				context.Logger.Error(e.LogMessage); | ||||
| 				context.SetLaunchFailedStatusAndReportEvent(e.Reason); | ||||
| 			} | ||||
| 			else { | ||||
| 				context.Logger.Error(task.Exception, "Caught exception while launching instance."); | ||||
| 				context.SetLaunchFailedStatusAndReportEvent(InstanceLaunchFailReason.UnknownError); | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		context.Services.PortManager.Release(context.Configuration); | ||||
| 		context.TransitionState(new InstanceNotRunningState()); | ||||
| 	} | ||||
|  | ||||
| 	private sealed class LaunchFailureException : Exception { | ||||
| 		public InstanceLaunchFailReason Reason { get; } | ||||
| 		public string LogMessage { get; } | ||||
|  | ||||
| 		public LaunchFailureException(InstanceLaunchFailReason reason, string logMessage) { | ||||
| 			this.Reason = reason; | ||||
| 			this.LogMessage = logMessage; | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	public (IInstanceState, LaunchInstanceResult) Launch(InstanceContext context) { | ||||
| 		return (this, LaunchInstanceResult.InstanceAlreadyLaunching); | ||||
| 	} | ||||
|  | ||||
| 	public (IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy) { | ||||
| 		cancellationTokenSource.Cancel(); | ||||
| 		return (this, StopInstanceResult.StopInitiated); | ||||
| 	} | ||||
|  | ||||
| 	public Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||
| 		return Task.FromResult(false); | ||||
| 	} | ||||
|  | ||||
| 	public void Dispose() { | ||||
| 		cancellationTokenSource.Dispose(); | ||||
| 	} | ||||
| } | ||||
| @@ -1,34 +1,8 @@ | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Phantom.Common.Data.Minecraft; | ||||
| using Phantom.Common.Data.Replies; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.States;  | ||||
| namespace Phantom.Agent.Services.Instances.States; | ||||
|  | ||||
| sealed class InstanceNotRunningState : IInstanceState { | ||||
| 	public void Initialize() {} | ||||
|  | ||||
| 	public (IInstanceState, LaunchInstanceResult) Launch(InstanceContext context) { | ||||
| 		InstanceLaunchFailReason? failReason = context.Services.PortManager.Reserve(context.Configuration) switch { | ||||
| 			PortManager.Result.ServerPortNotAllowed   => InstanceLaunchFailReason.ServerPortNotAllowed, | ||||
| 			PortManager.Result.ServerPortAlreadyInUse => InstanceLaunchFailReason.ServerPortAlreadyInUse, | ||||
| 			PortManager.Result.RconPortNotAllowed     => InstanceLaunchFailReason.RconPortNotAllowed, | ||||
| 			PortManager.Result.RconPortAlreadyInUse   => InstanceLaunchFailReason.RconPortAlreadyInUse, | ||||
| 			_                                         => null | ||||
| 		}; | ||||
|  | ||||
| 		if (failReason is {} reason) { | ||||
| 			context.SetLaunchFailedStatusAndReportEvent(reason); | ||||
| 			return (this, LaunchInstanceResult.LaunchInitiated); | ||||
| 		} | ||||
| 		 | ||||
| 		context.SetStatus(InstanceStatus.Launching); | ||||
| 		return (new InstanceLaunchingState(context), LaunchInstanceResult.LaunchInitiated); | ||||
| 	} | ||||
|  | ||||
| 	public (IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy) { | ||||
| 		return (this, StopInstanceResult.InstanceAlreadyStopped); | ||||
| 	} | ||||
|  | ||||
| 	public Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||
| 		return Task.FromResult(false); | ||||
| 	} | ||||
|   | ||||
| @@ -1,133 +1,81 @@ | ||||
| using Phantom.Agent.Minecraft.Command; | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Agent.Minecraft.Launcher; | ||||
| using Phantom.Agent.Services.Backups; | ||||
| using Phantom.Agent.Services.Instances.Sessions; | ||||
| using Phantom.Agent.Services.Instances.Procedures; | ||||
| using Phantom.Common.Data.Backups; | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Phantom.Common.Data.Minecraft; | ||||
| using Phantom.Common.Data.Replies; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.States; | ||||
|  | ||||
| sealed class InstanceRunningState : IInstanceState { | ||||
| 	private readonly InstanceContext context; | ||||
| 	private readonly InstanceProcess process; | ||||
| 	private readonly BackupScheduler backupScheduler; | ||||
| 	private readonly RunningSessionDisposer runningSessionDisposer; | ||||
| 	 | ||||
| 	private readonly CancellationTokenSource delayedStopCancellationTokenSource = new (); | ||||
| 	private bool stateOwnsDelayedStopCancellationTokenSource = true; | ||||
| 	private bool isStopping; | ||||
| sealed class InstanceRunningState : IInstanceState, IDisposable { | ||||
| 	public InstanceProcess Process { get; } | ||||
|  | ||||
| 	public InstanceRunningState(InstanceContext context, InstanceProcess process, InstanceSession session) { | ||||
| 	internal bool IsStopping { get; set; } | ||||
|  | ||||
| 	private readonly InstanceConfiguration configuration; | ||||
| 	private readonly IServerLauncher launcher; | ||||
| 	private readonly IInstanceContext context; | ||||
|  | ||||
| 	private readonly InstanceLogSender logSender; | ||||
| 	private readonly BackupScheduler backupScheduler; | ||||
|  | ||||
| 	private bool isDisposed; | ||||
|  | ||||
| 	public InstanceRunningState(InstanceConfiguration configuration, IServerLauncher launcher, InstanceProcess process, IInstanceContext context) { | ||||
| 		this.configuration = configuration; | ||||
| 		this.launcher = launcher; | ||||
| 		this.context = context; | ||||
| 		this.process = process; | ||||
| 		this.backupScheduler = new BackupScheduler(context.Services.TaskManager, context.Services.BackupManager, process, context.Configuration.ServerPort, context.ShortName); | ||||
| 		this.Process = process; | ||||
|  | ||||
| 		this.logSender = new InstanceLogSender(context.Services.ControllerConnection, context.Services.TaskManager, configuration.InstanceGuid, context.ShortName); | ||||
|  | ||||
| 		this.backupScheduler = new BackupScheduler(context.Services.TaskManager, context.Services.BackupManager, process, context, configuration.ServerPort); | ||||
| 		this.backupScheduler.BackupCompleted += OnScheduledBackupCompleted; | ||||
| 		this.runningSessionDisposer = new RunningSessionDisposer(this, session); | ||||
| 	} | ||||
|  | ||||
| 	public void Initialize() { | ||||
| 		process.Ended += ProcessEnded; | ||||
| 		 | ||||
| 		if (process.HasEnded) { | ||||
| 			if (runningSessionDisposer.Dispose()) { | ||||
| 		Process.Ended += ProcessEnded; | ||||
|  | ||||
| 		if (Process.HasEnded) { | ||||
| 			if (TryDispose()) { | ||||
| 				context.Logger.Warning("Session ended immediately after it was started."); | ||||
| 				context.ReportEvent(InstanceEvent.Stopped); | ||||
| 				context.Services.TaskManager.Run("Transition state of instance " + context.ShortName + " to not running", () => context.TransitionState(new InstanceNotRunningState(), InstanceStatus.Failed(InstanceLaunchFailReason.UnknownError))); | ||||
| 				context.EnqueueProcedure(new SetInstanceToNotRunningStateProcedure(InstanceStatus.Failed(InstanceLaunchFailReason.UnknownError)), immediate: true); | ||||
| 			} | ||||
| 		} | ||||
| 		else { | ||||
| 			context.SetStatus(InstanceStatus.Running); | ||||
| 			context.Logger.Information("Session started."); | ||||
| 			Process.AddOutputListener(SessionOutput); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private void SessionOutput(object? sender, string line) { | ||||
| 		context.Logger.Debug("[Server] {Line}", line); | ||||
| 		logSender.Enqueue(line); | ||||
| 	} | ||||
|  | ||||
| 	private void ProcessEnded(object? sender, EventArgs e) { | ||||
| 		if (!runningSessionDisposer.Dispose()) { | ||||
| 		if (!TryDispose()) { | ||||
| 			return; | ||||
| 		} | ||||
|  | ||||
| 		if (isStopping) { | ||||
| 			context.Logger.Information("Session ended."); | ||||
| 			context.ReportEvent(InstanceEvent.Stopped); | ||||
| 			context.TransitionState(new InstanceNotRunningState(), InstanceStatus.NotRunning); | ||||
| 		if (IsStopping) { | ||||
| 			context.EnqueueProcedure(new SetInstanceToNotRunningStateProcedure(InstanceStatus.NotRunning), immediate: true); | ||||
| 		} | ||||
| 		else { | ||||
| 			context.Logger.Information("Session ended unexpectedly, restarting..."); | ||||
| 			context.ReportEvent(InstanceEvent.Crashed); | ||||
| 			context.TransitionState(new InstanceLaunchingState(context), InstanceStatus.Restarting); | ||||
| 			context.EnqueueProcedure(new LaunchInstanceProcedure(configuration, launcher, IsRestarting: true)); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	public (IInstanceState, LaunchInstanceResult) Launch(InstanceContext context) { | ||||
| 		return (this, LaunchInstanceResult.InstanceAlreadyRunning); | ||||
| 	} | ||||
|  | ||||
| 	public (IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy) { | ||||
| 		if (stopStrategy == MinecraftStopStrategy.Instant) { | ||||
| 			CancelDelayedStop(); | ||||
| 			return (PrepareStoppedState(), StopInstanceResult.StopInitiated); | ||||
| 		} | ||||
|  | ||||
| 		if (isStopping) { | ||||
| 			// TODO change delay or something | ||||
| 			return (this, StopInstanceResult.InstanceAlreadyStopping); | ||||
| 		} | ||||
| 		 | ||||
| 		isStopping = true; | ||||
| 		context.Services.TaskManager.Run("Delayed stop timer for instance " + context.ShortName, () => StopLater(stopStrategy.Seconds)); | ||||
| 		return (this, StopInstanceResult.StopInitiated); | ||||
| 	} | ||||
|  | ||||
| 	private IInstanceState PrepareStoppedState() { | ||||
| 		process.Ended -= ProcessEnded; | ||||
| 		backupScheduler.Stop(); | ||||
| 		return new InstanceStoppingState(context, process, runningSessionDisposer); | ||||
| 	} | ||||
|  | ||||
| 	private void CancelDelayedStop() { | ||||
| 		try { | ||||
| 			delayedStopCancellationTokenSource.Cancel(); | ||||
| 		} catch (ObjectDisposedException) { | ||||
| 			// ignore | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private async Task StopLater(int seconds) { | ||||
| 		var cancellationToken = delayedStopCancellationTokenSource.Token; | ||||
|  | ||||
| 		try { | ||||
| 			stateOwnsDelayedStopCancellationTokenSource = false; | ||||
|  | ||||
| 			int[] stops = { 60, 30, 10, 5, 4, 3, 2, 1, 0 }; | ||||
| 			 | ||||
| 			foreach (var stop in stops) { | ||||
| 				if (seconds > stop) { | ||||
| 					await SendCommand(MinecraftCommand.Say("Server shutting down in " + seconds + (seconds == 1 ? " second." : " seconds.")), cancellationToken); | ||||
| 					await Task.Delay(TimeSpan.FromSeconds(seconds - stop), cancellationToken); | ||||
| 					seconds = stop; | ||||
| 				} | ||||
| 			} | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			context.Logger.Debug("Cancelled delayed stop."); | ||||
| 			return; | ||||
| 		} catch (ObjectDisposedException) { | ||||
| 			return; | ||||
| 		} catch (Exception e) { | ||||
| 			context.Logger.Warning(e, "Caught exception during delayed stop."); | ||||
| 			return; | ||||
| 		} finally { | ||||
| 			delayedStopCancellationTokenSource.Dispose(); | ||||
| 		} | ||||
|  | ||||
| 		context.TransitionState(PrepareStoppedState()); | ||||
| 	private void OnScheduledBackupCompleted(object? sender, BackupCreationResult e) { | ||||
| 		context.ReportEvent(new InstanceBackupCompletedEvent(e.Kind, e.Warnings)); | ||||
| 	} | ||||
|  | ||||
| 	public async Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||
| 		try { | ||||
| 			context.Logger.Information("Sending command: {Command}", command); | ||||
| 			await process.SendCommand(command, cancellationToken); | ||||
| 			await Process.SendCommand(command, cancellationToken); | ||||
| 			return true; | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			return false; | ||||
| @@ -137,42 +85,25 @@ sealed class InstanceRunningState : IInstanceState { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private void OnScheduledBackupCompleted(object? sender, BackupCreationResult e) { | ||||
| 		context.ReportEvent(new InstanceBackupCompletedEvent(e.Kind, e.Warnings)); | ||||
| 	private bool TryDispose() { | ||||
| 		lock (this) { | ||||
| 			if (isDisposed) { | ||||
| 				return false; | ||||
| 			} | ||||
|  | ||||
| 			isDisposed = true; | ||||
| 		} | ||||
|  | ||||
| 		logSender.Stop(); | ||||
| 		backupScheduler.Stop(); | ||||
| 		 | ||||
| 		Process.Dispose(); | ||||
| 		context.Services.PortManager.Release(configuration); | ||||
| 		 | ||||
| 		return true; | ||||
| 	} | ||||
|  | ||||
| 	private sealed class RunningSessionDisposer : IDisposable { | ||||
| 		private readonly InstanceRunningState state; | ||||
| 		private readonly InstanceSession session; | ||||
| 		private bool isDisposed; | ||||
|  | ||||
| 		public RunningSessionDisposer(InstanceRunningState state, InstanceSession session) { | ||||
| 			this.state = state; | ||||
| 			this.session = session; | ||||
| 		} | ||||
|  | ||||
| 		public bool Dispose() { | ||||
| 			lock (this) { | ||||
| 				if (isDisposed) { | ||||
| 					return false; | ||||
| 				} | ||||
|  | ||||
| 				isDisposed = true; | ||||
| 			} | ||||
|  | ||||
| 			if (state.stateOwnsDelayedStopCancellationTokenSource) { | ||||
| 				state.delayedStopCancellationTokenSource.Dispose(); | ||||
| 			} | ||||
| 			else { | ||||
| 				state.CancelDelayedStop(); | ||||
| 			} | ||||
| 			 | ||||
| 			session.Dispose(); | ||||
| 			return true; | ||||
| 		} | ||||
|  | ||||
| 		void IDisposable.Dispose() { | ||||
| 			Dispose(); | ||||
| 		} | ||||
| 	public void Dispose() { | ||||
| 		TryDispose(); | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,89 +0,0 @@ | ||||
| using System.Diagnostics; | ||||
| using Phantom.Agent.Minecraft.Command; | ||||
| using Phantom.Agent.Minecraft.Instance; | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Phantom.Common.Data.Minecraft; | ||||
| using Phantom.Common.Data.Replies; | ||||
|  | ||||
| namespace Phantom.Agent.Services.Instances.States;  | ||||
|  | ||||
| sealed class InstanceStoppingState : IInstanceState, IDisposable { | ||||
| 	private readonly InstanceContext context; | ||||
| 	private readonly InstanceProcess process; | ||||
| 	private readonly IDisposable sessionDisposer; | ||||
|  | ||||
| 	public InstanceStoppingState(InstanceContext context, InstanceProcess process, IDisposable sessionDisposer) { | ||||
| 		this.context = context; | ||||
| 		this.process = process; | ||||
| 		this.sessionDisposer = sessionDisposer; | ||||
| 	} | ||||
|  | ||||
| 	public void Initialize() { | ||||
| 		context.Logger.Information("Session stopping."); | ||||
| 		context.SetStatus(InstanceStatus.Stopping); | ||||
| 		context.Services.TaskManager.Run("Stop procedure for instance " + context.ShortName, DoStop); | ||||
| 	} | ||||
|  | ||||
| 	private async Task DoStop() { | ||||
| 		try { | ||||
| 			// Do not release the semaphore after this point. | ||||
| 			if (!await process.BackupSemaphore.CancelAndWait(TimeSpan.FromSeconds(1))) { | ||||
| 				context.Logger.Information("Waiting for backup to finish..."); | ||||
| 				await process.BackupSemaphore.CancelAndWait(Timeout.InfiniteTimeSpan); | ||||
| 			} | ||||
| 			 | ||||
| 			context.Logger.Information("Sending stop command..."); | ||||
| 			await DoSendStopCommand(); | ||||
|  | ||||
| 			context.Logger.Information("Waiting for session to end..."); | ||||
| 			await DoWaitForSessionToEnd(); | ||||
| 		} finally { | ||||
| 			context.Logger.Information("Session stopped."); | ||||
| 			context.ReportEvent(InstanceEvent.Stopped); | ||||
| 			context.TransitionState(new InstanceNotRunningState(), InstanceStatus.NotRunning); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private async Task DoSendStopCommand() { | ||||
| 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(5)); | ||||
| 		try { | ||||
| 			await process.SendCommand(MinecraftCommand.Stop, timeout.Token); | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			// ignore | ||||
| 		} catch (ObjectDisposedException e) when (e.ObjectName == typeof(Process).FullName && process.HasEnded) { | ||||
| 			// ignore | ||||
| 		} catch (Exception e) { | ||||
| 			context.Logger.Warning(e, "Caught exception while sending stop command."); | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private async Task DoWaitForSessionToEnd() { | ||||
| 		using var timeout = new CancellationTokenSource(TimeSpan.FromSeconds(55)); | ||||
| 		try { | ||||
| 			await process.WaitForExit(timeout.Token); | ||||
| 		} catch (OperationCanceledException) { | ||||
| 			try { | ||||
| 				context.Logger.Warning("Waiting timed out, killing session..."); | ||||
| 				process.Kill(); | ||||
| 			} catch (Exception e) { | ||||
| 				context.Logger.Error(e, "Caught exception while killing session."); | ||||
| 			} | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	public (IInstanceState, LaunchInstanceResult) Launch(InstanceContext context) { | ||||
| 		return (this, LaunchInstanceResult.InstanceIsStopping); | ||||
| 	} | ||||
|  | ||||
| 	public (IInstanceState, StopInstanceResult) Stop(MinecraftStopStrategy stopStrategy) { | ||||
| 		return (this, StopInstanceResult.InstanceAlreadyStopping); // TODO maybe provide a way to kill? | ||||
| 	} | ||||
|  | ||||
| 	public Task<bool> SendCommand(string command, CancellationToken cancellationToken) { | ||||
| 		return Task.FromResult(false); | ||||
| 	} | ||||
|  | ||||
| 	public void Dispose() { | ||||
| 		sessionDisposer.Dispose(); | ||||
| 	} | ||||
| } | ||||
| @@ -1,13 +1,12 @@ | ||||
| <Project Sdk="Microsoft.NET.Sdk"> | ||||
|    | ||||
|   <PropertyGroup> | ||||
|     <TargetFramework>net7.0</TargetFramework> | ||||
|     <ImplicitUsings>enable</ImplicitUsings> | ||||
|     <Nullable>enable</Nullable> | ||||
|   </PropertyGroup> | ||||
|    | ||||
|   <ItemGroup> | ||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Messages\Phantom.Common.Messages.csproj" /> | ||||
|     <ProjectReference Include="..\..\Common\Phantom.Common.Messages.Agent\Phantom.Common.Messages.Agent.csproj" /> | ||||
|     <ProjectReference Include="..\Phantom.Agent.Minecraft\Phantom.Agent.Minecraft.csproj" /> | ||||
|     <ProjectReference Include="..\Phantom.Agent.Rpc\Phantom.Agent.Rpc.csproj" /> | ||||
|   </ItemGroup> | ||||
|   | ||||
| @@ -1,11 +1,11 @@ | ||||
| using Phantom.Agent.Rpc; | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Phantom.Common.Data.Instance; | ||||
| using Phantom.Common.Data.Replies; | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Common.Messages; | ||||
| using Phantom.Common.Messages.BiDirectional; | ||||
| using Phantom.Common.Messages.ToAgent; | ||||
| using Phantom.Common.Messages.ToServer; | ||||
| using Phantom.Common.Messages.Agent; | ||||
| using Phantom.Common.Messages.Agent.BiDirectional; | ||||
| using Phantom.Common.Messages.Agent.ToAgent; | ||||
| using Phantom.Common.Messages.Agent.ToController; | ||||
| using Phantom.Utils.Rpc; | ||||
| using Phantom.Utils.Rpc.Message; | ||||
| using Serilog; | ||||
|  | ||||
| @@ -14,11 +14,11 @@ namespace Phantom.Agent.Services.Rpc; | ||||
| public sealed class MessageListener : IMessageToAgentListener { | ||||
| 	private static ILogger Logger { get; } = PhantomLogger.Create<MessageListener>(); | ||||
|  | ||||
| 	private readonly RpcServerConnection connection; | ||||
| 	private readonly RpcConnectionToServer<IMessageToControllerListener> connection; | ||||
| 	private readonly AgentServices agent; | ||||
| 	private readonly CancellationTokenSource shutdownTokenSource; | ||||
|  | ||||
| 	public MessageListener(RpcServerConnection connection, AgentServices agent, CancellationTokenSource shutdownTokenSource) { | ||||
| 	public MessageListener(RpcConnectionToServer<IMessageToControllerListener> connection, AgentServices agent, CancellationTokenSource shutdownTokenSource) { | ||||
| 		this.connection = connection; | ||||
| 		this.agent = agent; | ||||
| 		this.shutdownTokenSource = shutdownTokenSource; | ||||
| @@ -40,7 +40,7 @@ public sealed class MessageListener : IMessageToAgentListener { | ||||
| 			} | ||||
| 		} | ||||
|  | ||||
| 		await ServerMessaging.Send(new AdvertiseJavaRuntimesMessage(agent.JavaRuntimeRepository.All)); | ||||
| 		await connection.Send(new AdvertiseJavaRuntimesMessage(agent.JavaRuntimeRepository.All)); | ||||
| 		await agent.InstanceSessionManager.RefreshAgentStatus(); | ||||
| 		 | ||||
| 		return NoReply.Instance; | ||||
| @@ -54,6 +54,8 @@ public sealed class MessageListener : IMessageToAgentListener { | ||||
| 		}; | ||||
|  | ||||
| 		Logger.Fatal("Agent authentication failed: {Error}", errorMessage); | ||||
| 		 | ||||
| 		PhantomLogger.Dispose(); | ||||
| 		Environment.Exit(1); | ||||
|  | ||||
| 		return Task.FromResult(NoReply.Instance); | ||||
|   | ||||
| @@ -1,5 +1,5 @@ | ||||
| using NetMQ; | ||||
| using Phantom.Common.Data.Agent; | ||||
| using Phantom.Common.Data; | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Utils.Cryptography; | ||||
| using Phantom.Utils.IO; | ||||
| @@ -10,7 +10,7 @@ namespace Phantom.Agent; | ||||
| static class AgentKey { | ||||
| 	private static ILogger Logger { get; } = PhantomLogger.Create(nameof(AgentKey)); | ||||
|  | ||||
| 	public static Task<(NetMQCertificate, AgentAuthToken)?> Load(string? agentKeyToken, string? agentKeyFilePath) { | ||||
| 	public static Task<(NetMQCertificate, AuthToken)?> Load(string? agentKeyToken, string? agentKeyFilePath) { | ||||
| 		if (agentKeyFilePath != null) { | ||||
| 			return LoadFromFile(agentKeyFilePath); | ||||
| 		} | ||||
| @@ -22,7 +22,7 @@ static class AgentKey { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private static async Task<(NetMQCertificate, AgentAuthToken)?> LoadFromFile(string agentKeyFilePath) { | ||||
| 	private static async Task<(NetMQCertificate, AuthToken)?> LoadFromFile(string agentKeyFilePath) { | ||||
| 		if (!File.Exists(agentKeyFilePath)) { | ||||
| 			Logger.Fatal("Missing agent key file: {AgentKeyFilePath}", agentKeyFilePath); | ||||
| 			return null; | ||||
| @@ -41,7 +41,7 @@ static class AgentKey { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private static (NetMQCertificate, AgentAuthToken)? LoadFromToken(string agentKey) { | ||||
| 	private static (NetMQCertificate, AuthToken)? LoadFromToken(string agentKey) { | ||||
| 		try { | ||||
| 			return LoadFromBytes(TokenGenerator.DecodeBytes(agentKey)); | ||||
| 		} catch (Exception) { | ||||
| @@ -50,11 +50,11 @@ static class AgentKey { | ||||
| 		} | ||||
| 	} | ||||
|  | ||||
| 	private static (NetMQCertificate, AgentAuthToken)? LoadFromBytes(byte[] agentKey) { | ||||
| 		var (publicKey, agentToken) = AgentKeyData.FromBytes(agentKey); | ||||
| 		var serverCertificate = NetMQCertificate.FromPublicKey(publicKey); | ||||
| 	private static (NetMQCertificate, AuthToken)? LoadFromBytes(byte[] agentKey) { | ||||
| 		var (publicKey, agentToken) = ConnectionCommonKey.FromBytes(agentKey); | ||||
| 		var controllerCertificate = NetMQCertificate.FromPublicKey(publicKey); | ||||
| 		 | ||||
| 		Logger.Information("Loaded agent key."); | ||||
| 		return (serverCertificate, agentToken); | ||||
| 		return (controllerCertificate, agentToken); | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -3,7 +3,7 @@ using Phantom.Common.Logging; | ||||
| using Phantom.Utils.IO; | ||||
| using Serilog; | ||||
|  | ||||
| namespace Phantom.Agent;  | ||||
| namespace Phantom.Agent; | ||||
|  | ||||
| static class GuidFile { | ||||
| 	private static ILogger Logger { get; } = PhantomLogger.Create(nameof(GuidFile)); | ||||
|   | ||||
| @@ -1,7 +1,6 @@ | ||||
| <Project Sdk="Microsoft.NET.Sdk"> | ||||
|    | ||||
|   <PropertyGroup> | ||||
|     <TargetFramework>net7.0</TargetFramework> | ||||
|     <ImplicitUsings>enable</ImplicitUsings> | ||||
|     <Nullable>enable</Nullable> | ||||
|   </PropertyGroup> | ||||
| @@ -12,8 +11,7 @@ | ||||
|   </PropertyGroup> | ||||
|    | ||||
|   <ItemGroup> | ||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils.IO\Phantom.Utils.IO.csproj" /> | ||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils.Runtime\Phantom.Utils.Runtime.csproj" /> | ||||
|     <ProjectReference Include="..\..\Utils\Phantom.Utils\Phantom.Utils.csproj" /> | ||||
|     <ProjectReference Include="..\Phantom.Agent.Minecraft\Phantom.Agent.Minecraft.csproj" /> | ||||
|     <ProjectReference Include="..\Phantom.Agent.Services\Phantom.Agent.Services.csproj" /> | ||||
|   </ItemGroup> | ||||
|   | ||||
| @@ -1,12 +1,17 @@ | ||||
| using System.Reflection; | ||||
| using NetMQ; | ||||
| using Phantom.Agent; | ||||
| using Phantom.Agent.Rpc; | ||||
| using Phantom.Agent.Services; | ||||
| using Phantom.Agent.Services.Rpc; | ||||
| using Phantom.Common.Data.Agent; | ||||
| using Phantom.Common.Logging; | ||||
| using Phantom.Common.Messages.Agent; | ||||
| using Phantom.Common.Messages.Agent.ToController; | ||||
| using Phantom.Utils.Rpc; | ||||
| using Phantom.Utils.Rpc.Sockets; | ||||
| using Phantom.Utils.Runtime; | ||||
| using Phantom.Utils.Tasks; | ||||
|  | ||||
| const int ProtocolVersion = 1; | ||||
|  | ||||
| @@ -17,44 +22,45 @@ PosixSignals.RegisterCancellation(shutdownCancellationTokenSource, static () => | ||||
| 	PhantomLogger.Root.InformationHeading("Stopping Phantom Panel agent..."); | ||||
| }); | ||||
|  | ||||
| ThreadPool.SetMinThreads(workerThreads: 2, completionPortThreads: 1); | ||||
|  | ||||
| try { | ||||
| 	var fullVersion = AssemblyAttributes.GetFullVersion(Assembly.GetExecutingAssembly()); | ||||
|  | ||||
| 	PhantomLogger.Root.InformationHeading("Initializing Phantom Panel agent..."); | ||||
| 	PhantomLogger.Root.Information("Agent version: {Version}", fullVersion); | ||||
|  | ||||
| 	var (serverHost, serverPort, javaSearchPath, agentKeyToken, agentKeyFilePath, agentName, maxInstances, maxMemory, allowedServerPorts, allowedRconPorts) = Variables.LoadOrExit(); | ||||
| 	var (controllerHost, controllerPort, javaSearchPath, agentKeyToken, agentKeyFilePath, agentName, maxInstances, maxMemory, allowedServerPorts, allowedRconPorts, maxConcurrentBackupCompressionTasks) = Variables.LoadOrStop(); | ||||
|  | ||||
| 	var agentKey = await AgentKey.Load(agentKeyToken, agentKeyFilePath); | ||||
| 	if (agentKey == null) { | ||||
| 		Environment.Exit(1); | ||||
| 		return 1; | ||||
| 	} | ||||
|  | ||||
| 	var folders = new AgentFolders("./data", "./temp", javaSearchPath); | ||||
| 	if (!folders.TryCreate()) { | ||||
| 		Environment.Exit(1); | ||||
| 		return 1; | ||||
| 	} | ||||
|  | ||||
| 	var agentGuid = await GuidFile.CreateOrLoad(folders.DataFolderPath); | ||||
| 	if (agentGuid == null) { | ||||
| 		Environment.Exit(1); | ||||
| 		return 1; | ||||
| 	} | ||||
|  | ||||
| 	var (serverCertificate, agentToken) = agentKey.Value; | ||||
| 	var (controllerCertificate, agentToken) = agentKey.Value; | ||||
| 	var agentInfo = new AgentInfo(agentGuid.Value, agentName, ProtocolVersion, fullVersion, maxInstances, maxMemory, allowedServerPorts, allowedRconPorts); | ||||
| 	var agentServices = new AgentServices(agentInfo, folders); | ||||
|  | ||||
| 	MessageListener MessageListenerFactory(RpcServerConnection connection) { | ||||
| 		return new MessageListener(connection, agentServices, shutdownCancellationTokenSource); | ||||
| 	} | ||||
|  | ||||
| 	 | ||||
| 	PhantomLogger.Root.InformationHeading("Launching Phantom Panel agent..."); | ||||
| 	 | ||||
| 	var rpcConfiguration = new RpcConfiguration(PhantomLogger.Create("Rpc"), PhantomLogger.Create<TaskManager>("Rpc"), controllerHost, controllerPort, controllerCertificate); | ||||
| 	var rpcSocket = RpcClientSocket.Connect(rpcConfiguration, AgentMessageRegistries.Definitions, new RegisterAgentMessage(agentToken, agentInfo)); | ||||
|  | ||||
| 	var agentServices = new AgentServices(agentInfo, folders, new AgentServiceConfiguration(maxConcurrentBackupCompressionTasks), new ControllerConnection(rpcSocket.Connection)); | ||||
| 	await agentServices.Initialize(); | ||||
|  | ||||
| 	var rpcDisconnectSemaphore = new SemaphoreSlim(0, 1); | ||||
| 	var rpcConfiguration = new RpcConfiguration(PhantomLogger.Create("Rpc"), PhantomLogger.Create<TaskManager>("Rpc"), serverHost, serverPort, serverCertificate); | ||||
| 	var rpcTask = RpcLauncher.Launch(rpcConfiguration, agentToken, agentInfo, MessageListenerFactory, rpcDisconnectSemaphore, shutdownCancellationToken); | ||||
| 	var rpcMessageListener = new MessageListener(rpcSocket.Connection, agentServices, shutdownCancellationTokenSource); | ||||
| 	var rpcTask = RpcClientRuntime.Launch(rpcSocket, agentInfo, rpcMessageListener, rpcDisconnectSemaphore, shutdownCancellationToken); | ||||
| 	try { | ||||
| 		await rpcTask.WaitAsync(shutdownCancellationToken); | ||||
| 	} finally { | ||||
| @@ -64,11 +70,18 @@ try { | ||||
| 		rpcDisconnectSemaphore.Release(); | ||||
| 		await rpcTask; | ||||
| 		rpcDisconnectSemaphore.Dispose(); | ||||
| 		 | ||||
| 		NetMQConfig.Cleanup(); | ||||
| 	} | ||||
|  | ||||
| 	return 0; | ||||
| } catch (OperationCanceledException) { | ||||
| 	// Ignore. | ||||
| 	return 0; | ||||
| } catch (StopProcedureException) { | ||||
| 	return 1; | ||||
| } catch (Exception e) { | ||||
| 	PhantomLogger.Root.Fatal(e, "Caught exception in entry point."); | ||||
| 	return 1; | ||||
| } finally { | ||||
| 	shutdownCancellationTokenSource.Dispose(); | ||||
|  | ||||
|   | ||||
| @@ -6,8 +6,8 @@ using Phantom.Utils.Runtime; | ||||
| namespace Phantom.Agent; | ||||
|  | ||||
| sealed record Variables( | ||||
| 	string ServerHost, | ||||
| 	ushort ServerPort, | ||||
| 	string ControllerHost, | ||||
| 	ushort ControllerPort, | ||||
| 	string JavaSearchPath, | ||||
| 	string? AgentKeyToken, | ||||
| 	string? AgentKeyFilePath, | ||||
| @@ -15,15 +15,16 @@ sealed record Variables( | ||||
| 	ushort MaxInstances, | ||||
| 	RamAllocationUnits MaxMemory, | ||||
| 	AllowedPorts AllowedServerPorts, | ||||
| 	AllowedPorts AllowedRconPorts | ||||
| 	AllowedPorts AllowedRconPorts, | ||||
| 	ushort MaxConcurrentBackupCompressionTasks | ||||
| ) { | ||||
| 	private static Variables LoadOrThrow() { | ||||
| 		var (agentKeyToken, agentKeyFilePath) = EnvironmentVariables.GetEitherString("AGENT_KEY", "AGENT_KEY_FILE").Require; | ||||
| 		var javaSearchPath = EnvironmentVariables.GetString("JAVA_SEARCH_PATH").WithDefaultGetter(GetDefaultJavaSearchPath); | ||||
|  | ||||
| 		return new Variables( | ||||
| 			EnvironmentVariables.GetString("SERVER_HOST").Require, | ||||
| 			EnvironmentVariables.GetPortNumber("SERVER_PORT").WithDefault(9401), | ||||
| 			EnvironmentVariables.GetString("CONTROLLER_HOST").Require, | ||||
| 			EnvironmentVariables.GetPortNumber("CONTROLLER_PORT").WithDefault(9401), | ||||
| 			javaSearchPath, | ||||
| 			agentKeyToken, | ||||
| 			agentKeyFilePath, | ||||
| @@ -31,7 +32,8 @@ sealed record Variables( | ||||
| 			(ushort) EnvironmentVariables.GetInteger("MAX_INSTANCES", min: 1, max: 10000).Require, | ||||
| 			EnvironmentVariables.GetString("MAX_MEMORY").MapParse(RamAllocationUnits.FromString).Require, | ||||
| 			EnvironmentVariables.GetString("ALLOWED_SERVER_PORTS").MapParse(AllowedPorts.FromString).Require, | ||||
| 			EnvironmentVariables.GetString("ALLOWED_RCON_PORTS").MapParse(AllowedPorts.FromString).Require | ||||
| 			EnvironmentVariables.GetString("ALLOWED_RCON_PORTS").MapParse(AllowedPorts.FromString).Require, | ||||
| 			(ushort) EnvironmentVariables.GetInteger("MAX_CONCURRENT_BACKUP_COMPRESSION_TASKS", min: 1, max: 10000).WithDefault(1) | ||||
| 		); | ||||
| 	} | ||||
|  | ||||
| @@ -39,13 +41,12 @@ sealed record Variables( | ||||
| 		return JavaRuntimeDiscovery.GetSystemSearchPath() ?? throw new Exception("Could not automatically determine the path to Java installations on this system. Please set the JAVA_SEARCH_PATH environment variable to the folder containing Java installations."); | ||||
| 	} | ||||
|  | ||||
| 	public static Variables LoadOrExit() { | ||||
| 	public static Variables LoadOrStop() { | ||||
| 		try { | ||||
| 			return LoadOrThrow(); | ||||
| 		} catch (Exception e) { | ||||
| 			PhantomLogger.Root.Fatal(e.Message); | ||||
| 			Environment.Exit(1); | ||||
| 			throw; | ||||
| 			throw StopProcedureException.Instance; | ||||
| 		} | ||||
| 	} | ||||
| } | ||||
|   | ||||
| @@ -1,6 +1,6 @@ | ||||
| using NUnit.Framework; | ||||
|  | ||||
| namespace Phantom.Common.Data.Tests;  | ||||
| namespace Phantom.Common.Data.Tests; | ||||
|  | ||||
| [TestFixture] | ||||
| public sealed class AllowedPortsTests { | ||||
|   | ||||
| @@ -1,7 +1,6 @@ | ||||
| <Project Sdk="Microsoft.NET.Sdk"> | ||||
|    | ||||
|   <PropertyGroup> | ||||
|     <TargetFramework>net7.0</TargetFramework> | ||||
|     <ImplicitUsings>enable</ImplicitUsings> | ||||
|     <Nullable>enable</Nullable> | ||||
|   </PropertyGroup> | ||||
| @@ -15,7 +14,6 @@ | ||||
|     <PackageReference Include="NUnit" /> | ||||
|     <PackageReference Include="NUnit3TestAdapter" /> | ||||
|     <PackageReference Include="NUnit.Analyzers" /> | ||||
|     <PackageReference Include="coverlet.collector" /> | ||||
|   </ItemGroup> | ||||
|    | ||||
|   <ItemGroup> | ||||
|   | ||||
| @@ -1,4 +1,4 @@ | ||||
| namespace Phantom.Server.Database.Enums; | ||||
| namespace Phantom.Common.Data.Web.AuditLog; | ||||
| 
 | ||||
| public enum AuditLogEventType { | ||||
| 	AdministratorUserCreated, | ||||
| @@ -6,6 +6,7 @@ public enum AuditLogEventType { | ||||
| 	UserLoggedIn, | ||||
| 	UserLoggedOut, | ||||
| 	UserCreated, | ||||
| 	UserPasswordChanged, | ||||
| 	UserRolesChanged, | ||||
| 	UserDeleted, | ||||
| 	InstanceCreated, | ||||
| @@ -22,6 +23,7 @@ public static class AuditLogEventTypeExtensions { | ||||
| 		{ AuditLogEventType.UserLoggedIn,              AuditLogSubjectType.User }, | ||||
| 		{ AuditLogEventType.UserLoggedOut,             AuditLogSubjectType.User }, | ||||
| 		{ AuditLogEventType.UserCreated,               AuditLogSubjectType.User }, | ||||
| 		{ AuditLogEventType.UserPasswordChanged,       AuditLogSubjectType.User }, | ||||
| 		{ AuditLogEventType.UserRolesChanged,          AuditLogSubjectType.User }, | ||||
| 		{ AuditLogEventType.UserDeleted,               AuditLogSubjectType.User }, | ||||
| 		{ AuditLogEventType.InstanceCreated,           AuditLogSubjectType.Instance }, | ||||
| @@ -39,7 +41,7 @@ public static class AuditLogEventTypeExtensions { | ||||
| 		} | ||||
| 	} | ||||
| 
 | ||||
| 	internal static AuditLogSubjectType GetSubjectType(this AuditLogEventType type) { | ||||
| 	public static AuditLogSubjectType GetSubjectType(this AuditLogEventType type) { | ||||
| 		return SubjectTypes[type]; | ||||
| 	} | ||||
| } | ||||
							
								
								
									
										5
									
								
								Common/Phantom.Common.Data.Web/AuditLog/AuditLogItem.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										5
									
								
								Common/Phantom.Common.Data.Web/AuditLog/AuditLogItem.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,5 @@ | ||||
| using System.Text.Json; | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.AuditLog; | ||||
|  | ||||
| public sealed record AuditLogItem(DateTime UtcTime, Guid? UserGuid, string? UserName, AuditLogEventType EventType, AuditLogSubjectType SubjectType, string? SubjectId, JsonDocument? Data); | ||||
| @@ -1,4 +1,4 @@ | ||||
| namespace Phantom.Server.Database.Enums;  | ||||
| namespace Phantom.Common.Data.Web.AuditLog; | ||||
| 
 | ||||
| public enum AuditLogSubjectType { | ||||
| 	User, | ||||
| @@ -1,4 +1,4 @@ | ||||
| namespace Phantom.Server.Database.Enums; | ||||
| namespace Phantom.Common.Data.Web.EventLog; | ||||
| 
 | ||||
| public enum EventLogEventType { | ||||
| 	InstanceLaunchSucceded, | ||||
| @@ -10,7 +10,7 @@ public enum EventLogEventType { | ||||
| 	InstanceBackupFailed, | ||||
| } | ||||
| 
 | ||||
| internal static class EventLogEventTypeExtensions { | ||||
| public static class EventLogEventTypeExtensions { | ||||
| 	private static readonly Dictionary<EventLogEventType, EventLogSubjectType> SubjectTypes = new () { | ||||
| 		{ EventLogEventType.InstanceLaunchSucceded, EventLogSubjectType.Instance }, | ||||
| 		{ EventLogEventType.InstanceLaunchFailed, EventLogSubjectType.Instance }, | ||||
| @@ -1,6 +1,5 @@ | ||||
| using System.Text.Json; | ||||
| using Phantom.Server.Database.Enums; | ||||
| 
 | ||||
| namespace Phantom.Server.Services.Events;  | ||||
| namespace Phantom.Common.Data.Web.EventLog; | ||||
| 
 | ||||
| public sealed record EventLogItem(DateTime UtcTime, Guid? AgentGuid, EventLogEventType EventType, EventLogSubjectType SubjectType, string SubjectId, JsonDocument? Data); | ||||
| @@ -1,4 +1,4 @@ | ||||
| namespace Phantom.Server.Database.Enums;  | ||||
| namespace Phantom.Common.Data.Web.EventLog; | ||||
| 
 | ||||
| public enum EventLogSubjectType { | ||||
| 	Instance | ||||
| @@ -1,17 +1,21 @@ | ||||
| using System.Collections.Immutable; | ||||
| 
 | ||||
| namespace Phantom.Common.Minecraft; | ||||
| namespace Phantom.Common.Data.Web.Minecraft; | ||||
| 
 | ||||
| public static class JvmArgumentsHelper { | ||||
| 	public static ImmutableArray<string> Split(string arguments) { | ||||
| 		return arguments.Split('\n', StringSplitOptions.TrimEntries | StringSplitOptions.RemoveEmptyEntries).ToImmutableArray(); | ||||
| 	} | ||||
| 
 | ||||
| 	public static string Join(ImmutableArray<string> arguments) { | ||||
| 		return string.Join('\n', arguments); | ||||
| 	} | ||||
| 
 | ||||
| 	public static ValidationError? Validate(string arguments) { | ||||
| 		return Validate(Split(arguments)); | ||||
| 	} | ||||
| 	 | ||||
| 	public static ValidationError? Validate(ImmutableArray<string> arguments) { | ||||
| 
 | ||||
| 	private static ValidationError? Validate(ImmutableArray<string> arguments) { | ||||
| 		if (!arguments.All(static argument => argument.StartsWith('-'))) { | ||||
| 			return ValidationError.InvalidFormat; | ||||
| 		} | ||||
| @@ -28,22 +32,9 @@ public static class JvmArgumentsHelper { | ||||
| 		return null; | ||||
| 	} | ||||
| 
 | ||||
| 	public static string Join(ImmutableArray<string> arguments) { | ||||
| 		return string.Join('\n', arguments); | ||||
| 	} | ||||
| 
 | ||||
| 	public enum ValidationError { | ||||
| 		InvalidFormat, | ||||
| 		XmxNotAllowed, | ||||
| 		XmsNotAllowed | ||||
| 	} | ||||
| 
 | ||||
| 	public static string ToSentence(this ValidationError? result) { | ||||
| 		return result switch { | ||||
| 			ValidationError.InvalidFormat => "Invalid format.", | ||||
| 			ValidationError.XmxNotAllowed => "The -Xmx argument must not be specified manually.", | ||||
| 			ValidationError.XmsNotAllowed => "The -Xms argument must not be specified manually.", | ||||
| 			_                             => throw new ArgumentOutOfRangeException(nameof(result), result, null) | ||||
| 		}; | ||||
| 	} | ||||
| } | ||||
| @@ -1,9 +1,13 @@ | ||||
| <Project Sdk="Microsoft.NET.Sdk"> | ||||
|    | ||||
|   <PropertyGroup> | ||||
|     <TargetFramework>net7.0</TargetFramework> | ||||
|     <ImplicitUsings>enable</ImplicitUsings> | ||||
|     <Nullable>enable</Nullable> | ||||
|   </PropertyGroup> | ||||
| 
 | ||||
|    | ||||
|   <ItemGroup> | ||||
|     <PackageReference Include="BCrypt.Net-Next.StrongName" /> | ||||
|     <PackageReference Include="MemoryPack" /> | ||||
|   </ItemGroup> | ||||
|    | ||||
| </Project> | ||||
							
								
								
									
										8
									
								
								Common/Phantom.Common.Data.Web/Users/AddRoleError.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										8
									
								
								Common/Phantom.Common.Data.Web/Users/AddRoleError.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,8 @@ | ||||
| namespace Phantom.Common.Data.Web.Users; | ||||
|  | ||||
| public enum AddRoleError : byte { | ||||
| 	NameIsEmpty, | ||||
| 	NameIsTooLong, | ||||
| 	NameAlreadyExists, | ||||
| 	UnknownError | ||||
| } | ||||
							
								
								
									
										28
									
								
								Common/Phantom.Common.Data.Web/Users/AddUserError.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										28
									
								
								Common/Phantom.Common.Data.Web/Users/AddUserError.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,28 @@ | ||||
| using System.Collections.Immutable; | ||||
| using MemoryPack; | ||||
| using Phantom.Common.Data.Web.Users.AddUserErrors; | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users { | ||||
| 	[MemoryPackable] | ||||
| 	[MemoryPackUnion(0, typeof(NameIsInvalid))] | ||||
| 	[MemoryPackUnion(1, typeof(PasswordIsInvalid))] | ||||
| 	[MemoryPackUnion(2, typeof(NameAlreadyExists))] | ||||
| 	[MemoryPackUnion(3, typeof(UnknownError))] | ||||
| 	public abstract partial record AddUserError { | ||||
| 		internal AddUserError() {} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users.AddUserErrors { | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record NameIsInvalid([property: MemoryPackOrder(0)] UsernameRequirementViolation Violation) : AddUserError; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record PasswordIsInvalid([property: MemoryPackOrder(0)] ImmutableArray<PasswordRequirementViolation> Violations) : AddUserError; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record NameAlreadyExists : AddUserError; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record UnknownError : AddUserError; | ||||
| } | ||||
| @@ -0,0 +1,31 @@ | ||||
| using MemoryPack; | ||||
| using Phantom.Common.Data.Web.Users.CreateOrUpdateAdministratorUserResults; | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users { | ||||
| 	[MemoryPackable] | ||||
| 	[MemoryPackUnion(0, typeof(Success))] | ||||
| 	[MemoryPackUnion(1, typeof(CreationFailed))] | ||||
| 	[MemoryPackUnion(2, typeof(UpdatingFailed))] | ||||
| 	[MemoryPackUnion(3, typeof(AddingToRoleFailed))] | ||||
| 	[MemoryPackUnion(4, typeof(UnknownError))] | ||||
| 	public abstract partial record CreateOrUpdateAdministratorUserResult { | ||||
| 		internal CreateOrUpdateAdministratorUserResult() {} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users.CreateOrUpdateAdministratorUserResults { | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record Success([property: MemoryPackOrder(0)] UserInfo User) : CreateOrUpdateAdministratorUserResult; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record CreationFailed([property: MemoryPackOrder(0)] AddUserError Error) : CreateOrUpdateAdministratorUserResult; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record UpdatingFailed([property: MemoryPackOrder(0)] SetUserPasswordError Error) : CreateOrUpdateAdministratorUserResult; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record AddingToRoleFailed : CreateOrUpdateAdministratorUserResult; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record UnknownError : CreateOrUpdateAdministratorUserResult; | ||||
| } | ||||
							
								
								
									
										7
									
								
								Common/Phantom.Common.Data.Web/Users/DeleteUserResult.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										7
									
								
								Common/Phantom.Common.Data.Web/Users/DeleteUserResult.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,7 @@ | ||||
| namespace Phantom.Common.Data.Web.Users; | ||||
|  | ||||
| public enum DeleteUserResult : byte { | ||||
| 	Deleted, | ||||
| 	NotFound, | ||||
| 	Failed | ||||
| } | ||||
							
								
								
									
										25
									
								
								Common/Phantom.Common.Data.Web/Users/IdentityPermissions.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										25
									
								
								Common/Phantom.Common.Data.Web/Users/IdentityPermissions.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,25 @@ | ||||
| using System.Collections.Immutable; | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users; | ||||
|  | ||||
| public sealed class IdentityPermissions { | ||||
| 	public static IdentityPermissions None { get; } = new (ImmutableHashSet<string>.Empty); | ||||
| 	 | ||||
| 	private readonly ImmutableHashSet<string> permissionIds; | ||||
|  | ||||
| 	public IdentityPermissions(ImmutableHashSet<string> permissionIdsQuery) { | ||||
| 		this.permissionIds = permissionIdsQuery; | ||||
| 	} | ||||
|  | ||||
| 	public bool Check(Permission? permission) { | ||||
| 		while (permission != null) { | ||||
| 			if (!permissionIds.Contains(permission.Id)) { | ||||
| 				return false; | ||||
| 			} | ||||
|  | ||||
| 			permission = permission.Parent; | ||||
| 		} | ||||
|  | ||||
| 		return true; | ||||
| 	} | ||||
| } | ||||
| @@ -0,0 +1,27 @@ | ||||
| using MemoryPack; | ||||
| using Phantom.Common.Data.Web.Users.PasswordRequirementViolations; | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users { | ||||
| 	[MemoryPackable] | ||||
| 	[MemoryPackUnion(0, typeof(TooShort))] | ||||
| 	[MemoryPackUnion(1, typeof(MustContainLowercaseLetter))] | ||||
| 	[MemoryPackUnion(2, typeof(MustContainUppercaseLetter))] | ||||
| 	[MemoryPackUnion(3, typeof(MustContainDigit))] | ||||
| 	public abstract partial record PasswordRequirementViolation { | ||||
| 		internal PasswordRequirementViolation() {} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users.PasswordRequirementViolations { | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record TooShort([property: MemoryPackOrder(0)] int MinimumLength) : PasswordRequirementViolation; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record MustContainLowercaseLetter : PasswordRequirementViolation; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record MustContainUppercaseLetter : PasswordRequirementViolation; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record MustContainDigit : PasswordRequirementViolation; | ||||
| } | ||||
| @@ -1,4 +1,4 @@ | ||||
| namespace Phantom.Server.Web.Identity.Data; | ||||
| namespace Phantom.Common.Data.Web.Users; | ||||
| 
 | ||||
| public sealed record Permission(string Id, Permission? Parent) { | ||||
| 	private static readonly List<Permission> AllPermissions = new (); | ||||
							
								
								
									
										9
									
								
								Common/Phantom.Common.Data.Web/Users/RoleInfo.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								Common/Phantom.Common.Data.Web/Users/RoleInfo.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,9 @@ | ||||
| using MemoryPack; | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users; | ||||
|  | ||||
| [MemoryPackable(GenerateType.VersionTolerant)] | ||||
| public sealed partial record RoleInfo( | ||||
| 	[property: MemoryPackOrder(0)] Guid Guid, | ||||
| 	[property: MemoryPackOrder(1)] string Name | ||||
| ); | ||||
							
								
								
									
										24
									
								
								Common/Phantom.Common.Data.Web/Users/SetUserPasswordError.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										24
									
								
								Common/Phantom.Common.Data.Web/Users/SetUserPasswordError.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,24 @@ | ||||
| using System.Collections.Immutable; | ||||
| using MemoryPack; | ||||
| using Phantom.Common.Data.Web.Users.SetUserPasswordErrors; | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users { | ||||
| 	[MemoryPackable] | ||||
| 	[MemoryPackUnion(0, typeof(UserNotFound))] | ||||
| 	[MemoryPackUnion(1, typeof(PasswordIsInvalid))] | ||||
| 	[MemoryPackUnion(2, typeof(UnknownError))] | ||||
| 	public abstract partial record SetUserPasswordError { | ||||
| 		internal SetUserPasswordError() {} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users.SetUserPasswordErrors { | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record UserNotFound : SetUserPasswordError; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record PasswordIsInvalid([property: MemoryPackOrder(0)] ImmutableArray<PasswordRequirementViolation> Violations) : SetUserPasswordError; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record UnknownError : SetUserPasswordError; | ||||
| } | ||||
							
								
								
									
										9
									
								
								Common/Phantom.Common.Data.Web/Users/UserInfo.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										9
									
								
								Common/Phantom.Common.Data.Web/Users/UserInfo.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,9 @@ | ||||
| using MemoryPack; | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users; | ||||
|  | ||||
| [MemoryPackable(GenerateType.VersionTolerant)] | ||||
| public sealed partial record UserInfo( | ||||
| 	[property: MemoryPackOrder(0)] Guid Guid, | ||||
| 	[property: MemoryPackOrder(1)] string Name | ||||
| ); | ||||
							
								
								
									
										12
									
								
								Common/Phantom.Common.Data.Web/Users/UserPasswords.cs
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										12
									
								
								Common/Phantom.Common.Data.Web/Users/UserPasswords.cs
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,12 @@ | ||||
| namespace Phantom.Common.Data.Web.Users; | ||||
|  | ||||
| public static class UserPasswords { | ||||
| 	public static string Hash(string password) { | ||||
| 		return BCrypt.Net.BCrypt.HashPassword(password, workFactor: 12); | ||||
| 	} | ||||
|  | ||||
| 	public static bool Verify(string password, string hash) { | ||||
| 		// TODO rehash | ||||
| 		return BCrypt.Net.BCrypt.Verify(password, hash); | ||||
| 	} | ||||
| } | ||||
| @@ -0,0 +1,19 @@ | ||||
| using MemoryPack; | ||||
| using Phantom.Common.Data.Web.Users.UsernameRequirementViolations; | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users { | ||||
| 	[MemoryPackable] | ||||
| 	[MemoryPackUnion(0, typeof(IsEmpty))] | ||||
| 	[MemoryPackUnion(1, typeof(TooLong))] | ||||
| 	public abstract partial record UsernameRequirementViolation { | ||||
| 		internal UsernameRequirementViolation() {} | ||||
| 	} | ||||
| } | ||||
|  | ||||
| namespace Phantom.Common.Data.Web.Users.UsernameRequirementViolations { | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record IsEmpty : UsernameRequirementViolation; | ||||
|  | ||||
| 	[MemoryPackable(GenerateType.VersionTolerant)] | ||||
| 	public sealed partial record TooLong([property: MemoryPackOrder(0)] int MaxLength) : UsernameRequirementViolation; | ||||
| } | ||||
| @@ -2,7 +2,7 @@ | ||||
|  | ||||
| namespace Phantom.Common.Data.Agent; | ||||
|  | ||||
| [MemoryPackable] | ||||
| [MemoryPackable(GenerateType.VersionTolerant)] | ||||
| public sealed partial record AgentInfo( | ||||
| 	[property: MemoryPackOrder(0)] Guid Guid, | ||||
| 	[property: MemoryPackOrder(1)] string Name, | ||||
|   | ||||
| @@ -1,18 +0,0 @@ | ||||
| namespace Phantom.Common.Data.Agent; | ||||
|  | ||||
| public static class AgentKeyData { | ||||
| 	private const byte TokenLength = AgentAuthToken.Length; | ||||
|  | ||||
| 	public static byte[] ToBytes(byte[] publicKey, AgentAuthToken agentToken) { | ||||
| 		Span<byte> agentKey = stackalloc byte[TokenLength + publicKey.Length]; | ||||
| 		agentToken.WriteTo(agentKey[..TokenLength]); | ||||
| 		publicKey.CopyTo(agentKey[TokenLength..]); | ||||
| 		return agentKey.ToArray(); | ||||
| 	} | ||||
|  | ||||
| 	public static (byte[] PublicKey, AgentAuthToken AgentToken) FromBytes(byte[] agentKey) { | ||||
| 		var token = new AgentAuthToken(agentKey[..TokenLength]); | ||||
| 		var publicKey = agentKey[TokenLength..]; | ||||
| 		return (publicKey, token); | ||||
| 	} | ||||
| } | ||||
| @@ -4,7 +4,7 @@ using MemoryPack; | ||||
|  | ||||
| namespace Phantom.Common.Data; | ||||
|  | ||||
| [MemoryPackable] | ||||
| [MemoryPackable(GenerateType.VersionTolerant)] | ||||
| public sealed partial class AllowedPorts { | ||||
| 	[MemoryPackOrder(0)] | ||||
| 	[MemoryPackInclude] | ||||
| @@ -53,6 +53,6 @@ public sealed partial class AllowedPorts { | ||||
| 	} | ||||
|  | ||||
| 	public static AllowedPorts FromString(string definitions) { | ||||
| 		return FromString((ReadOnlySpan<char>) definitions); | ||||
| 		return FromString(definitions.AsSpan()); | ||||
| 	} | ||||
| } | ||||
|   | ||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user