Agent/Phantom.Agent.Services/AgentServices.cs

@@ -6,6 +6,7 @@ using Phantom.Agent.Services.Instances;
 using Phantom.Common.Data.Agent;
 using Phantom.Utils.Actor;
 using Phantom.Utils.Logging;
+using Phantom.Utils.Tasks;
 using Serilog;
 
 namespace Phantom.Agent.Services;
@@ -17,6 +18,7 @@ public sealed class AgentServices {
 
 	private AgentFolders AgentFolders { get; }
 	private AgentState AgentState { get; }
+	private TaskManager TaskManager { get; }
 	private BackupManager BackupManager { get; }
 
 	internal JavaRuntimeRepository JavaRuntimeRepository { get; }
@@ -28,12 +30,13 @@ public sealed class AgentServices {
 		
 		this.AgentFolders = agentFolders;
 		this.AgentState = new AgentState();
+		this.TaskManager = new TaskManager(PhantomLogger.Create<TaskManager, AgentServices>());
 		this.BackupManager = new BackupManager(agentFolders, serviceConfiguration.MaxConcurrentCompressionTasks);
 		
 		this.JavaRuntimeRepository = new JavaRuntimeRepository();
 		this.InstanceTicketManager = new InstanceTicketManager(agentInfo, controllerConnection);
 		
-		var instanceManagerInit = new InstanceManagerActor.Init(controllerConnection, agentFolders, AgentState, JavaRuntimeRepository, InstanceTicketManager, BackupManager);
+		var instanceManagerInit = new InstanceManagerActor.Init(controllerConnection, agentFolders, AgentState, JavaRuntimeRepository, InstanceTicketManager, TaskManager, BackupManager);
 		this.InstanceManager = ActorSystem.ActorOf(InstanceManagerActor.Factory(instanceManagerInit), "InstanceManager");
 	}
 
@@ -47,6 +50,7 @@ public sealed class AgentServices {
 		Logger.Information("Stopping services...");
 		
 		await InstanceManager.Stop(new InstanceManagerActor.ShutdownCommand());
+		await TaskManager.Stop();
 		
 		BackupManager.Dispose();
 		

Agent/Phantom.Agent.Services/Backups/BackupScheduler.cs

@@ -24,7 +24,7 @@ sealed class BackupScheduler : CancellableBackgroundTask {
 	
 	public event EventHandler<BackupCreationResult>? BackupCompleted;
 
-	public BackupScheduler(InstanceContext context, InstanceProcess process, int serverPort) : base(PhantomLogger.Create<BackupScheduler>(context.ShortName)) {
+	public BackupScheduler(InstanceContext context, InstanceProcess process, int serverPort) : base(PhantomLogger.Create<BackupScheduler>(context.ShortName), context.Services.TaskManager, "Backup scheduler for " + context.ShortName) {
 		this.backupManager = context.Services.BackupManager;
 		this.context = context;
 		this.process = process;

Agent/Phantom.Agent.Services/Instances/InstanceManagerActor.cs

@@ -20,7 +20,7 @@ namespace Phantom.Agent.Services.Instances;
 sealed class InstanceManagerActor : ReceiveActor<InstanceManagerActor.ICommand> {
 	private static readonly ILogger Logger = PhantomLogger.Create<InstanceManagerActor>();
 
-	public readonly record struct Init(ControllerConnection ControllerConnection, AgentFolders AgentFolders, AgentState AgentState, JavaRuntimeRepository JavaRuntimeRepository, InstanceTicketManager InstanceTicketManager, BackupManager BackupManager);
+	public readonly record struct Init(ControllerConnection ControllerConnection, AgentFolders AgentFolders, AgentState AgentState, JavaRuntimeRepository JavaRuntimeRepository, InstanceTicketManager InstanceTicketManager, TaskManager TaskManager, BackupManager BackupManager);
 
 	public static Props<ICommand> Factory(Init init) {
 		return Props<ICommand>.Create(() => new InstanceManagerActor(init), new ActorConfiguration { SupervisorStrategy = SupervisorStrategies.Resume });
@@ -47,7 +47,7 @@ sealed class InstanceManagerActor : ReceiveActor<InstanceManagerActor.ICommand>
 		var minecraftServerExecutables = new MinecraftServerExecutables(init.AgentFolders.ServerExecutableFolderPath);
 		var launchServices = new LaunchServices(minecraftServerExecutables, init.JavaRuntimeRepository);
 
-		this.instanceServices = new InstanceServices(init.ControllerConnection, init.BackupManager, launchServices);
+		this.instanceServices = new InstanceServices(init.ControllerConnection, init.TaskManager, init.BackupManager, launchServices);
 		
 		ReceiveAndReply<ConfigureInstanceCommand, InstanceActionResult<ConfigureInstanceResult>>(ConfigureInstance);
 		ReceiveAndReply<LaunchInstanceCommand, InstanceActionResult<LaunchInstanceResult>>(LaunchInstance);

Agent/Phantom.Agent.Services/Instances/InstanceServices.cs

@@ -1,7 +1,8 @@
 using Phantom.Agent.Minecraft.Launcher;
 using Phantom.Agent.Rpc;
 using Phantom.Agent.Services.Backups;
+using Phantom.Utils.Tasks;
 
 namespace Phantom.Agent.Services.Instances;
 
-sealed record InstanceServices(ControllerConnection ControllerConnection, BackupManager BackupManager, LaunchServices LaunchServices);
+sealed record InstanceServices(ControllerConnection ControllerConnection, TaskManager TaskManager, BackupManager BackupManager, LaunchServices LaunchServices);

Agent/Phantom.Agent.Services/Instances/State/InstanceLogSender.cs

@@ -22,7 +22,7 @@ sealed class InstanceLogSender : CancellableBackgroundTask {
 	
 	private int droppedLinesSinceLastSend;
 
-	public InstanceLogSender(ControllerConnection controllerConnection, Guid instanceGuid, string loggerName) : base(PhantomLogger.Create<InstanceLogSender>(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);

Agent/Phantom.Agent.Services/Instances/State/InstanceRunningState.cs

@@ -31,7 +31,7 @@ sealed class InstanceRunningState : IDisposable {
 		this.Process = process;
 		this.cancellationToken = cancellationToken;
 
-		this.logSender = new InstanceLogSender(context.Services.ControllerConnection, context.InstanceGuid, context.ShortName);
+		this.logSender = new InstanceLogSender(context.Services.ControllerConnection, context.Services.TaskManager, context.InstanceGuid, context.ShortName);
 
 		this.backupScheduler = new BackupScheduler(context, process, configuration.ServerPort);
 		this.backupScheduler.BackupCompleted += OnScheduledBackupCompleted;

Common/Phantom.Common.Data.Web/EventLog/EventLogEventType.cs

@@ -7,7 +7,7 @@ public enum EventLogEventType {
 	InstanceStopped,
 	InstanceBackupSucceeded,
 	InstanceBackupSucceededWithWarnings,
-	InstanceBackupFailed
+	InstanceBackupFailed,
 }
 
 public static class EventLogEventTypeExtensions {
@@ -18,7 +18,7 @@ public static class EventLogEventTypeExtensions {
 		{ EventLogEventType.InstanceStopped, EventLogSubjectType.Instance },
 		{ EventLogEventType.InstanceBackupSucceeded, EventLogSubjectType.Instance },
 		{ EventLogEventType.InstanceBackupSucceededWithWarnings, EventLogSubjectType.Instance },
-		{ EventLogEventType.InstanceBackupFailed, EventLogSubjectType.Instance }
+		{ EventLogEventType.InstanceBackupFailed, EventLogSubjectType.Instance },
 	};
 
 	static EventLogEventTypeExtensions() {

Common/Phantom.Common.Data/AuthToken.cs

@@ -14,7 +14,9 @@ public sealed partial class AuthToken {
 	private readonly byte[] bytes;
 
 	internal AuthToken(byte[]? bytes) {
-		ArgumentNullException.ThrowIfNull(bytes);
+		if (bytes == null) {
+			throw new ArgumentNullException(nameof(bytes));
+		}
 
 		if (bytes.Length != Length) {
 			throw new ArgumentOutOfRangeException(nameof(bytes), "Invalid token length: " + bytes.Length + ". Token length must be exactly " + Length + " bytes.");

Common/Phantom.Common.Messages.Web/ToController/LogOutMessage.cs

@@ -1,10 +0,0 @@
-using System.Collections.Immutable;
-using MemoryPack;
-
-namespace Phantom.Common.Messages.Web.ToController;
-
-[MemoryPackable(GenerateType.VersionTolerant)]
-public sealed partial record LogOutMessage(
-	[property: MemoryPackOrder(0)] Guid UserGuid,
-	[property: MemoryPackOrder(1)] ImmutableArray<byte> SessionToken
-) : IMessageToController;

Common/Phantom.Common.Messages.Web/WebMessageRegistries.cs

@@ -24,22 +24,21 @@ public static class WebMessageRegistries {
 		ToController.Add<RegisterWebMessage>(0);
 		ToController.Add<UnregisterWebMessage>(1);
 		ToController.Add<LogInMessage, LogInSuccess?>(2);
-		ToController.Add<LogOutMessage>(3);
-		ToController.Add<CreateOrUpdateAdministratorUserMessage, CreateOrUpdateAdministratorUserResult>(4);
-		ToController.Add<CreateUserMessage, CreateUserResult>(5);
-		ToController.Add<DeleteUserMessage, DeleteUserResult>(6);
-		ToController.Add<GetUsersMessage, ImmutableArray<UserInfo>>(7);
-		ToController.Add<GetRolesMessage, ImmutableArray<RoleInfo>>(8);
-		ToController.Add<GetUserRolesMessage, ImmutableDictionary<Guid, ImmutableArray<Guid>>>(9);
-		ToController.Add<ChangeUserRolesMessage, ChangeUserRolesResult>(10);
-		ToController.Add<CreateOrUpdateInstanceMessage, InstanceActionResult<CreateOrUpdateInstanceResult>>(11);
-		ToController.Add<LaunchInstanceMessage, InstanceActionResult<LaunchInstanceResult>>(12);
-		ToController.Add<StopInstanceMessage, InstanceActionResult<StopInstanceResult>>(13);
-		ToController.Add<SendCommandToInstanceMessage, InstanceActionResult<SendCommandToInstanceResult>>(14);
-		ToController.Add<GetMinecraftVersionsMessage, ImmutableArray<MinecraftVersion>>(15);
-		ToController.Add<GetAgentJavaRuntimesMessage, ImmutableDictionary<Guid, ImmutableArray<TaggedJavaRuntime>>>(16);
-		ToController.Add<GetAuditLogMessage, ImmutableArray<AuditLogItem>>(17);
-		ToController.Add<GetEventLogMessage, ImmutableArray<EventLogItem>>(18);
+		ToController.Add<CreateOrUpdateAdministratorUserMessage, CreateOrUpdateAdministratorUserResult>(3);
+		ToController.Add<CreateUserMessage, CreateUserResult>(4);
+		ToController.Add<DeleteUserMessage, DeleteUserResult>(5);
+		ToController.Add<GetUsersMessage, ImmutableArray<UserInfo>>(6);
+		ToController.Add<GetRolesMessage, ImmutableArray<RoleInfo>>(7);
+		ToController.Add<GetUserRolesMessage, ImmutableDictionary<Guid, ImmutableArray<Guid>>>(8);
+		ToController.Add<ChangeUserRolesMessage, ChangeUserRolesResult>(9);
+		ToController.Add<CreateOrUpdateInstanceMessage, InstanceActionResult<CreateOrUpdateInstanceResult>>(10);
+		ToController.Add<LaunchInstanceMessage, InstanceActionResult<LaunchInstanceResult>>(11);
+		ToController.Add<StopInstanceMessage, InstanceActionResult<StopInstanceResult>>(12);
+		ToController.Add<SendCommandToInstanceMessage, InstanceActionResult<SendCommandToInstanceResult>>(13);
+		ToController.Add<GetMinecraftVersionsMessage, ImmutableArray<MinecraftVersion>>(14);
+		ToController.Add<GetAgentJavaRuntimesMessage, ImmutableDictionary<Guid, ImmutableArray<TaggedJavaRuntime>>>(15);
+		ToController.Add<GetAuditLogMessage, ImmutableArray<AuditLogItem>>(16);
+		ToController.Add<GetEventLogMessage, ImmutableArray<EventLogItem>>(17);
 		ToController.Add<ReplyMessage>(127);
 		
 		ToWeb.Add<RegisterWebResultMessage>(0);

Controller/Phantom.Controller.Database/ApplicationDbContext.cs

@@ -13,18 +13,18 @@ namespace Phantom.Controller.Database;
 
 [SuppressMessage("ReSharper", "AutoPropertyCanBeMadeGetOnly.Global")]
 public class ApplicationDbContext : DbContext {
-	public DbSet<UserEntity> Users { get; init; } = null!;
-	public DbSet<RoleEntity> Roles { get; init; } = null!;
-	public DbSet<PermissionEntity> Permissions { get; init; } = null!;
+	public DbSet<UserEntity> Users { get; set; } = null!;
+	public DbSet<RoleEntity> Roles { get; set; } = null!;
+	public DbSet<PermissionEntity> Permissions { get; set; } = null!;
 	
-	public DbSet<UserRoleEntity> UserRoles { get; init; } = null!;
-	public DbSet<UserPermissionEntity> UserPermissions { get; init; } = null!;
-	public DbSet<RolePermissionEntity> RolePermissions { get; init; } = null!;
+	public DbSet<UserRoleEntity> UserRoles { get; set; } = null!;
+	public DbSet<UserPermissionEntity> UserPermissions { get; set; } = null!;
+	public DbSet<RolePermissionEntity> RolePermissions { get; set; } = null!;
 	
-	public DbSet<AgentEntity> Agents { get; init; } = null!;
-	public DbSet<InstanceEntity> Instances { get; init; } = null!;
-	public DbSet<AuditLogEntity> AuditLog { get; init; } = null!;
-	public DbSet<EventLogEntity> EventLog { get; init; } = null!;
+	public DbSet<AgentEntity> Agents { get; set; } = null!;
+	public DbSet<InstanceEntity> Instances { get; set; } = null!;
+	public DbSet<AuditLogEntity> AuditLog { get; set; } = null!;
+	public DbSet<EventLogEntity> EventLog { get; set; } = null!;
 
 	public AgentEntityUpsert AgentUpsert { get; }
 	public InstanceEntityUpsert InstanceUpsert { get; }

Controller/Phantom.Controller.Database/Entities/AgentEntity.cs

@@ -9,7 +9,7 @@ namespace Phantom.Controller.Database.Entities;
 [SuppressMessage("ReSharper", "AutoPropertyCanBeMadeGetOnly.Global")]
 public sealed class AgentEntity {
 	[Key]
-	public Guid AgentGuid { get; init; }
+	public Guid AgentGuid { get; set; }
 	
 	public string Name { get; set; }
 	public ushort ProtocolVersion { get; set; }

Controller/Phantom.Controller.Database/Entities/AuditLogEntity.cs

@@ -13,16 +13,16 @@ public class AuditLogEntity : IDisposable {
 	[Key]
 	[DatabaseGenerated(DatabaseGeneratedOption.Identity)]
 	[SuppressMessage("ReSharper", "UnusedMember.Global")]
-	public long Id { get; init; }
+	public long Id { get; set; }
 
-	public Guid? UserGuid { get; init; }
-	public DateTime UtcTime { get; init; } // Note: Converting to UTC is not best practice, but for historical records it's good enough.
-	public AuditLogEventType EventType { get; init; }
-	public AuditLogSubjectType SubjectType { get; init; }
-	public string SubjectId { get; init; }
-	public JsonDocument? Data { get; init; }
+	public Guid? UserGuid { get; set; }
+	public DateTime UtcTime { get; set; } // Note: Converting to UTC is not best practice, but for historical records it's good enough.
+	public AuditLogEventType EventType { get; set; }
+	public AuditLogSubjectType SubjectType { get; set; }
+	public string SubjectId { get; set; }
+	public JsonDocument? Data { get; set; }
 
-	public virtual UserEntity? User { get; init; }
+	public virtual UserEntity? User { get; set; }
 	
 	[SuppressMessage("ReSharper", "UnusedMember.Global")]
 	internal AuditLogEntity() {

Controller/Phantom.Controller.Database/Entities/EventLogEntity.cs

@@ -11,14 +11,14 @@ namespace Phantom.Controller.Database.Entities;
 [SuppressMessage("ReSharper", "ClassWithVirtualMembersNeverInherited.Global")]
 public sealed class EventLogEntity : IDisposable {
 	[Key]
-	public Guid EventGuid { get; init; }
+	public Guid EventGuid { get; set; }
 
-	public DateTime UtcTime { get; init; } // Note: Converting to UTC is not best practice, but for historical records it's good enough.
-	public Guid? AgentGuid { get; init; }
-	public EventLogEventType EventType { get; init; }
-	public EventLogSubjectType SubjectType { get; init; }
-	public string SubjectId { get; init; }
-	public JsonDocument? Data { get; init; }
+	public DateTime UtcTime { get; set; } // Note: Converting to UTC is not best practice, but for historical records it's good enough.
+	public Guid? AgentGuid { get; set; }
+	public EventLogEventType EventType { get; set; }
+	public EventLogSubjectType SubjectType { get; set; }
+	public string SubjectId { get; set; }
+	public JsonDocument? Data { get; set; }
 	
 	[SuppressMessage("ReSharper", "UnusedMember.Global")]
 	internal EventLogEntity() {

Controller/Phantom.Controller.Database/Entities/InstanceEntity.cs

@@ -11,7 +11,7 @@ namespace Phantom.Controller.Database.Entities;
 [SuppressMessage("ReSharper", "MemberCanBePrivate.Global")]
 public sealed class InstanceEntity {
 	[Key]
-	public Guid InstanceGuid { get; init; }
+	public Guid InstanceGuid { get; set; }
 
 	public Guid AgentGuid { get; set; }
 

Controller/Phantom.Controller.Database/Entities/PermissionEntity.cs

@@ -6,7 +6,7 @@ namespace Phantom.Controller.Database.Entities;
 [Table("Permissions", Schema = "identity")]
 public sealed class PermissionEntity {
 	[Key]
-	public string Id { get; init; }
+	public string Id { get; set; }
 
 	public PermissionEntity(string id) {
 		Id = id;

Controller/Phantom.Controller.Database/Entities/RoleEntity.cs

@@ -7,9 +7,9 @@ namespace Phantom.Controller.Database.Entities;
 [Table("Roles", Schema = "identity")]
 public sealed class RoleEntity {
 	[Key]
-	public Guid RoleGuid { get; init; }
+	public Guid RoleGuid { get; set; }
 
-	public string Name { get; init; }
+	public string Name { get; set; }
 
 	public RoleEntity(Guid roleGuid, string name) {
 		RoleGuid = roleGuid;

Controller/Phantom.Controller.Database/Entities/RolePermissionEntity.cs

@@ -4,8 +4,8 @@ namespace Phantom.Controller.Database.Entities;
 
 [Table("RolePermissions", Schema = "identity")]
 public sealed class RolePermissionEntity {
-	public Guid RoleGuid { get; init; }
-	public string PermissionId { get; init; }
+	public Guid RoleGuid { get; set; }
+	public string PermissionId { get; set; }
 	
 	public RolePermissionEntity(Guid roleGuid, string permissionId) {
 		RoleGuid = roleGuid;

Controller/Phantom.Controller.Database/Entities/UserEntity.cs

@@ -7,9 +7,9 @@ namespace Phantom.Controller.Database.Entities;
 [Table("Users", Schema = "identity")]
 public sealed class UserEntity {
 	[Key]
-	public Guid UserGuid { get; init; }
+	public Guid UserGuid { get; set; }
 
-	public string Name { get; init; }
+	public string Name { get; set; }
 	public string PasswordHash { get; set; }
 
 	public UserEntity(Guid userGuid, string name, string passwordHash) {

Controller/Phantom.Controller.Database/Entities/UserPermissionEntity.cs

@@ -4,8 +4,8 @@ namespace Phantom.Controller.Database.Entities;
 
 [Table("UserPermissions", Schema = "identity")]
 public sealed class UserPermissionEntity {
-	public Guid UserGuid { get; init; }
-	public string PermissionId { get; init; }
+	public Guid UserGuid { get; set; }
+	public string PermissionId { get; set; }
 
 	public UserPermissionEntity(Guid userGuid, string permissionId) {
 		UserGuid = userGuid;

Controller/Phantom.Controller.Database/Entities/UserRoleEntity.cs

@@ -4,11 +4,11 @@ namespace Phantom.Controller.Database.Entities;
 
 [Table("UserRoles", Schema = "identity")]
 public sealed class UserRoleEntity {
-	public Guid UserGuid { get; init; }
-	public Guid RoleGuid { get; init; }
+	public Guid UserGuid { get; set; }
+	public Guid RoleGuid { get; set; }
 
-	public UserEntity User { get; init; }
-	public RoleEntity Role { get; init; }
+	public UserEntity User { get; set; }
+	public RoleEntity Role { get; set; }
 
 	public UserRoleEntity(Guid userGuid, Guid roleGuid) {
 		UserGuid = userGuid;

Controller/Phantom.Controller.Services/ControllerServices.cs

@@ -59,7 +59,7 @@ public sealed class ControllerServices : IDisposable {
 		this.PermissionManager = new PermissionManager(dbProvider);
 
 		this.UserRoleManager = new UserRoleManager(dbProvider);
-		this.UserLoginManager = new UserLoginManager(UserManager, PermissionManager, dbProvider);
+		this.UserLoginManager = new UserLoginManager(UserManager, PermissionManager);
 		this.AuditLogManager = new AuditLogManager(dbProvider);
 		this.EventLogManager = new EventLogManager(ActorSystem, dbProvider, shutdownCancellationToken);
 		

Controller/Phantom.Controller.Services/Rpc/AgentMessageHandlerActor.cs

@@ -1,4 +1,5 @@
-using Phantom.Common.Data.Replies;
+using Akka.Actor;
+using Phantom.Common.Data.Replies;
 using Phantom.Common.Messages.Agent;
 using Phantom.Common.Messages.Agent.BiDirectional;
 using Phantom.Common.Messages.Agent.ToAgent;
@@ -18,6 +19,8 @@ sealed class AgentMessageHandlerActor : ReceiveActor<IMessageToController> {
 		return Props<IMessageToController>.Create(() => new AgentMessageHandlerActor(init), new ActorConfiguration { SupervisorStrategy = SupervisorStrategies.Resume });
 	}
 
+	public IStash Stash { get; set; } = null!;
+	
 	private readonly Guid agentGuid;
 	private readonly RpcConnectionToClient<IMessageToAgent> connection;
 	private readonly AgentRegistrationHandler agentRegistrationHandler;

Controller/Phantom.Controller.Services/Rpc/WebMessageHandlerActor.cs

@@ -69,8 +69,6 @@ sealed class WebMessageHandlerActor : ReceiveActor<IMessageToController> {
 		
 		ReceiveAsync<RegisterWebMessage>(HandleRegisterWeb);
 		Receive<UnregisterWebMessage>(HandleUnregisterWeb);
-		ReceiveAndReplyLater<LogInMessage, LogInSuccess?>(HandleLogIn);
-		Receive<LogOutMessage>(HandleLogOut);
 		ReceiveAndReplyLater<CreateOrUpdateAdministratorUserMessage, CreateOrUpdateAdministratorUserResult>(HandleCreateOrUpdateAdministratorUser);
 		ReceiveAndReplyLater<CreateUserMessage, CreateUserResult>(HandleCreateUser);
 		ReceiveAndReplyLater<GetUsersMessage, ImmutableArray<UserInfo>>(HandleGetUsers);
@@ -86,6 +84,7 @@ sealed class WebMessageHandlerActor : ReceiveActor<IMessageToController> {
 		ReceiveAndReply<GetAgentJavaRuntimesMessage, ImmutableDictionary<Guid, ImmutableArray<TaggedJavaRuntime>>>(HandleGetAgentJavaRuntimes);
 		ReceiveAndReplyLater<GetAuditLogMessage, ImmutableArray<AuditLogItem>>(HandleGetAuditLog);
 		ReceiveAndReplyLater<GetEventLogMessage, ImmutableArray<EventLogItem>>(HandleGetEventLog);
+		ReceiveAndReplyLater<LogInMessage, LogInSuccess?>(HandleLogIn);
 		Receive<ReplyMessage>(HandleReply);
 	}
 
@@ -97,14 +96,6 @@ sealed class WebMessageHandlerActor : ReceiveActor<IMessageToController> {
 		connection.Close();
 	}
 
-	private Task<LogInSuccess?> HandleLogIn(LogInMessage message) {
-		return userLoginManager.LogIn(message.Username, message.Password);
-	}
-
-	private void HandleLogOut(LogOutMessage message) {
-		_ = userLoginManager.LogOut(message.UserGuid, message.SessionToken);
-	}
-	
 	private Task<CreateOrUpdateAdministratorUserResult> HandleCreateOrUpdateAdministratorUser(CreateOrUpdateAdministratorUserMessage message) {
 		return userManager.CreateOrUpdateAdministrator(message.Username, message.Password);
 	}
@@ -165,6 +156,10 @@ sealed class WebMessageHandlerActor : ReceiveActor<IMessageToController> {
 		return eventLogManager.GetMostRecentItems(message.Count);
 	}
 
+	private Task<LogInSuccess?> HandleLogIn(LogInMessage message) {
+		return userLoginManager.LogIn(message.Username, message.Password);
+	}
+
 	private void HandleReply(ReplyMessage message) {
 		connection.Receive(message);
 	}

Controller/Phantom.Controller.Services/Users/UserLoginManager.cs

@@ -2,23 +2,19 @@
 using System.Collections.Immutable;
 using System.Security.Cryptography;
 using Phantom.Common.Data.Web.Users;
-using Phantom.Controller.Database;
-using Phantom.Controller.Database.Repositories;
 
 namespace Phantom.Controller.Services.Users; 
 
 sealed class UserLoginManager {
 	private const int SessionIdBytes = 20;
-	private readonly ConcurrentDictionary<Guid, List<ImmutableArray<byte>>> sessionTokensByUserGuid = new ();
+	private readonly ConcurrentDictionary<string, List<ImmutableArray<byte>>> sessionTokensByUsername = new ();
 	
 	private readonly UserManager userManager;
 	private readonly PermissionManager permissionManager;
-	private readonly IDbContextProvider dbProvider;
 	
-	public UserLoginManager(UserManager userManager, PermissionManager permissionManager, IDbContextProvider dbProvider) {
+	public UserLoginManager(UserManager userManager, PermissionManager permissionManager) {
 		this.userManager = userManager;
 		this.permissionManager = permissionManager;
-		this.dbProvider = dbProvider;
 	}
 
 	public async Task<LogInSuccess?> LogIn(string username, string password) {
@@ -28,37 +24,11 @@ sealed class UserLoginManager {
 		}
 
 		var token = ImmutableArray.Create(RandomNumberGenerator.GetBytes(SessionIdBytes));
-		var sessionTokens = sessionTokensByUserGuid.GetOrAdd(user.UserGuid, static _ => new List<ImmutableArray<byte>>());
+		var sessionTokens = sessionTokensByUsername.GetOrAdd(username, static _ => new List<ImmutableArray<byte>>());
 		lock (sessionTokens) {
 			sessionTokens.Add(token);
 		}
 		
-		await using (var db = dbProvider.Lazy()) {
-			var auditLogWriter = new AuditLogRepository(db).Writer(user.UserGuid);
-			auditLogWriter.UserLoggedIn(user);
-			
-			await db.Ctx.SaveChangesAsync();
-		}
-
 		return new LogInSuccess(user.UserGuid, await permissionManager.FetchPermissionsForUserId(user.UserGuid), token);
 	}
-
-	public async Task LogOut(Guid userGuid, ImmutableArray<byte> sessionToken) {
-		if (!sessionTokensByUserGuid.TryGetValue(userGuid, out var sessionTokens)) {
-			return;
-		}
-
-		lock (sessionTokens) {
-			if (sessionTokens.RemoveAll(token => token.SequenceEqual(sessionToken)) == 0) {
-				return;
-			}
-		}
-
-		await using var db = dbProvider.Lazy();
-		
-		var auditLogWriter = new AuditLogRepository(db).Writer(userGuid);
-		auditLogWriter.UserLoggedOut(userGuid);
-			
-		await db.Ctx.SaveChangesAsync();
-	}
 }

Controller/Phantom.Controller/Program.cs

@@ -10,6 +10,7 @@ using Phantom.Utils.Logging;
 using Phantom.Utils.Rpc;
 using Phantom.Utils.Rpc.Runtime;
 using Phantom.Utils.Runtime;
+using Phantom.Utils.Tasks;
 
 var shutdownCancellationTokenSource = new CancellationTokenSource();
 var shutdownCancellationToken = shutdownCancellationTokenSource.Token;
@@ -63,12 +64,14 @@ try {
 		return new RpcConfiguration(serviceName, host, port, connectionKey.Certificate);
 	}
 
+	var rpcTaskManager = new TaskManager(PhantomLogger.Create<TaskManager>("Rpc"));
 	try {
 		await Task.WhenAll(
 			RpcServerRuntime.Launch(ConfigureRpc("Agent", agentRpcServerHost, agentRpcServerPort, agentKeyData), AgentMessageRegistries.Definitions, controllerServices.AgentRegistrationHandler, controllerServices.ActorSystem, shutdownCancellationToken),
 			RpcServerRuntime.Launch(ConfigureRpc("Web", webRpcServerHost, webRpcServerPort, webKeyData), WebMessageRegistries.Definitions, controllerServices.WebRegistrationHandler, controllerServices.ActorSystem, shutdownCancellationToken)
 		);
 	} finally {
+		await rpcTaskManager.Stop();
 		NetMQConfig.Cleanup();
 	}
 

Utils/Phantom.Utils.Actor/Mailbox/UnboundedJumpAheadMailbox.cs

@@ -1,12 +1,10 @@
-using System.Diagnostics.CodeAnalysis;
-using Akka.Actor;
+using Akka.Actor;
 using Akka.Configuration;
 using Akka.Dispatch;
 using Akka.Dispatch.MessageQueues;
 
 namespace Phantom.Utils.Actor.Mailbox;
 
-[SuppressMessage("ReSharper", "ClassNeverInstantiated.Global")]
 public sealed class UnboundedJumpAheadMailbox : MailboxType, IProducesMessageQueue<UnboundedJumpAheadMessageQueue> {
 	public const string Name = "unbounded-jump-ahead-mailbox";
 	

Utils/Phantom.Utils.Rpc/Runtime/RpcConnectionToClient.cs

@@ -9,7 +9,7 @@ public sealed class RpcConnectionToClient<TMessageBase> : RpcConnection<TMessage
 	private readonly uint routingId;
 
 	internal event EventHandler<RpcClientConnectionClosedEventArgs>? Closed;
-	private bool isClosed;
+	public bool IsClosed { get; private set; }
 
 	internal RpcConnectionToClient(ServerSocket socket, uint routingId, MessageRegistry<TMessageBase> messageRegistry, MessageReplyTracker replyTracker) : base(messageRegistry, replyTracker) {
 		this.socket = socket;
@@ -24,8 +24,8 @@ public sealed class RpcConnectionToClient<TMessageBase> : RpcConnection<TMessage
 		bool hasClosed = false;
 		
 		lock (this) {
-			if (!isClosed) {
-				isClosed = true;
+			if (!IsClosed) {
+				IsClosed = true;
 				hasClosed = true;
 			}
 		}

Utils/Phantom.Utils.Tests/Runtime/EnvironmentVariablesTests.cs

@@ -12,7 +12,7 @@ public class EnvironmentVariablesTests {
 	private readonly HashSet<string> createdVariables = new ();
 
 	private static void Discard<T>(T value) {
-		_ = value;
+		var _ = value;
 	}
 
 	private string CreateVariable(string value) {

Utils/Phantom.Utils/Collections/EnumerableExtensions.cs

@@ -1,8 +1,18 @@
 using System.Collections.Immutable;
+using System.Diagnostics.CodeAnalysis;
 
 namespace Phantom.Utils.Collections;
 
 public static class EnumerableExtensions {
+	[SuppressMessage("ReSharper", "LoopCanBeConvertedToQuery")]
+	public static IEnumerable<TSource> WhereNotNull<TSource>(this IEnumerable<TSource?> items) {
+		foreach (var item in items) {
+			if (item is not null) {
+				yield return item;
+			}
+		}
+	}
+	
 	public static async Task<ImmutableArray<TSource>> ToImmutableArrayAsync<TSource>(this IAsyncEnumerable<TSource> source, CancellationToken cancellationToken = default) {
 		var builder = ImmutableArray.CreateBuilder<TSource>();
 		

Utils/Phantom.Utils/Tasks/CancellableBackgroundTask.cs

@@ -8,13 +8,19 @@ public abstract class CancellableBackgroundTask {
 	protected ILogger Logger { get; }
 	protected CancellationToken CancellationToken { get; }
 
-	protected CancellableBackgroundTask(ILogger logger) {
+	private readonly TaskManager taskManager;
+	private readonly string taskName;
+	
+	protected CancellableBackgroundTask(ILogger logger, TaskManager taskManager, string taskName) {
 		this.Logger = logger;
 		this.CancellationToken = cancellationTokenSource.Token;
+		
+		this.taskManager = taskManager;
+		this.taskName = taskName;
 	}
 
 	protected void Start() {
-		Task.Run(Run, CancellationToken.None);
+		taskManager.Run(taskName, Run);
 	}
 
 	private async Task Run() {

Utils/Phantom.Utils/Tasks/Result.cs

@@ -1,6 +1,4 @@
-using System.Diagnostics.CodeAnalysis;
-
-namespace Phantom.Utils.Tasks;
+namespace Phantom.Utils.Tasks;
 
 public abstract record Result<TValue, TError> {
 	private Result() {}
@@ -44,7 +42,7 @@ public abstract record Result<TError> {
 		return new Fail(error);
 	}
 
-	public static implicit operator Result<TError>([SuppressMessage("ReSharper", "UnusedParameter.Global")] Result.OkType _) {
+	public static implicit operator Result<TError>(Result.OkType _) {
 		return new Ok();
 	}
 	

Utils/Phantom.Utils/Tasks/TaskManager.cs

@@ -0,0 +1,76 @@
+using System.Collections.Concurrent;
+using Phantom.Utils.Collections;
+using Serilog;
+
+namespace Phantom.Utils.Tasks;
+
+public sealed class TaskManager {
+	private readonly ILogger logger;
+	private readonly CancellationTokenSource cancellationTokenSource = new ();
+	private readonly CancellationToken cancellationToken;
+	
+	private readonly ConcurrentDictionary<Task, string> runningTasks = new (ReferenceEqualityComparer<Task>.Instance);
+	
+	public TaskManager(ILogger logger) {
+		this.logger = logger;
+		this.cancellationToken = cancellationTokenSource.Token;
+	}
+
+	private T Add<T>(string name, T task) where T : Task {
+		cancellationToken.ThrowIfCancellationRequested();
+		runningTasks.TryAdd(task, name);
+		task.ContinueWith(OnFinished, CancellationToken.None, TaskContinuationOptions.RunContinuationsAsynchronously, TaskScheduler.Default);
+		return task;
+	}
+
+	private void OnFinished(Task task) {
+		runningTasks.TryRemove(task, out _);
+	}
+
+	public Task Run(string name, Action action) {
+		return Add(name, Task.Run(action, cancellationToken));
+	}
+
+	public Task Run(string name, Func<Task> taskFunc) {
+		return Add(name, Task.Run(taskFunc, cancellationToken));
+	}
+
+	public Task<T> Run<T>(string name, Func<Task<T>> taskFunc) {
+		return Add(name, Task.Run(taskFunc, cancellationToken));
+	}
+
+	public async Task Stop() {
+		logger.Information("Stopping task manager...");
+		
+		cancellationTokenSource.Cancel();
+
+		var remainingTasksAwaiterTask = WaitForRemainingTasks();
+		while (true) {
+			var logStateTimeoutTask = Task.Delay(TimeSpan.FromSeconds(10), CancellationToken.None);
+			var completedTask = await Task.WhenAny(remainingTasksAwaiterTask, logStateTimeoutTask);
+			if (completedTask == logStateTimeoutTask) {
+				var remainingTaskNames = runningTasks.Values.Order().ToList();
+				var remainingTaskNameList = string.Join('\n', remainingTaskNames.Select(static name => "- " + name));
+				logger.Warning("Waiting for {TaskCount} task(s) to finish:\n{TaskNames}", remainingTaskNames.Count, remainingTaskNameList);
+			}
+			else {
+				break;
+			}
+		}
+
+		runningTasks.Clear();
+		cancellationTokenSource.Dispose();
+		
+		logger.Information("Task manager stopped.");
+	}
+
+	private async Task WaitForRemainingTasks() {
+		foreach (var task in runningTasks.Keys) {
+			try {
+				await task;
+			} catch (Exception) {
+				// ignored
+			}
+		}
+	}
+}

Utils/Phantom.Utils/Threading/ThreadSafeLinkedList.cs

@@ -0,0 +1,39 @@
+namespace Phantom.Utils.Threading;
+
+public sealed class ThreadSafeLinkedList<T> : IDisposable {
+	private readonly LinkedList<T> list = new ();
+	private readonly SemaphoreSlim semaphore = new (1, 1);
+
+	public async Task Add(T item, bool toFront, CancellationToken cancellationToken) {
+		await semaphore.WaitAsync(cancellationToken);
+		try {
+			if (toFront) {
+				list.AddFirst(item);
+			}
+			else {
+				list.AddLast(item);
+			}
+		} finally {
+			semaphore.Release();
+		}
+	}
+
+	public async Task<T?> TryTakeFromFront(CancellationToken cancellationToken) {
+		await semaphore.WaitAsync(cancellationToken);
+		try {
+			var firstNode = list.First;
+			if (firstNode == null) {
+				return default;
+			}
+
+			list.RemoveFirst();
+			return firstNode.Value;
+		} finally {
+			semaphore.Release();
+		}
+	}
+
+	public void Dispose() {
+		semaphore.Dispose();
+	}
+}

Utils/Phantom.Utils/Threading/ThreadSafeStructRef.cs

@@ -0,0 +1,28 @@
+namespace Phantom.Utils.Threading;
+
+public sealed class ThreadSafeStructRef<T> : IDisposable where T : struct {
+	private T? value;
+	private readonly SemaphoreSlim semaphore = new (1, 1);
+	
+	public async Task<T?> Get(CancellationToken cancellationToken) {
+		await semaphore.WaitAsync(cancellationToken);
+		try {
+			return value;
+		} finally {
+			semaphore.Release();
+		}
+	}
+
+	public async Task Set(T? value, CancellationToken cancellationToken) {
+		await semaphore.WaitAsync(cancellationToken);
+		try {
+			this.value = value;
+		} finally {
+			semaphore.Release();
+		}
+	}
+
+	public void Dispose() {
+		semaphore.Dispose();
+	}
+}

Web/Phantom.Web.Services/Authentication/UserLoginManager.cs

@@ -53,8 +53,8 @@ public sealed class UserLoginManager {
 
 	public async Task LogOut() {
 		var stored = await sessionBrowserStorage.Delete();
-		if (stored != null && sessionManager.Remove(stored.UserGuid, stored.Token)) {
-			await controllerConnection.Send(new LogOutMessage(stored.UserGuid, stored.Token));
+		if (stored != null) {
+			sessionManager.Remove(stored.UserGuid, stored.Token);
 		}
 
 		await navigation.NavigateTo(string.Empty);

Web/Phantom.Web.Services/Authentication/UserSessionManager.cs

@@ -23,13 +23,9 @@ public sealed class UserSessionManager {
 		return userSessions.TryGetValue(userGuid, out var sessions) && sessions.HasToken(token) ? sessions.UserInfo : null;
 	}
 
-	internal bool Remove(Guid userGuid, ImmutableArray<byte> token) {
+	internal void Remove(Guid userGuid, ImmutableArray<byte> token) {
 		if (userSessions.TryGetValue(userGuid, out var sessions)) {
 			sessions.RemoveToken(token);
-			return true;
-		}
-		else {
-			return false;
 		}
 	}
 }

Web/Phantom.Web/Program.cs

@@ -9,6 +9,7 @@ using Phantom.Utils.Logging;
 using Phantom.Utils.Rpc;
 using Phantom.Utils.Rpc.Sockets;
 using Phantom.Utils.Runtime;
+using Phantom.Utils.Tasks;
 using Phantom.Web;
 using Phantom.Web.Services;
 using Phantom.Web.Services.Rpc;
@@ -58,7 +59,8 @@ try {
 	var rpcSocket = RpcClientSocket.Connect(rpcConfiguration, WebMessageRegistries.Definitions, new RegisterWebMessage(webToken));
 
 	var webConfiguration = new WebLauncher.Configuration(PhantomLogger.Create("Web"), webServerHost, webServerPort, webBasePath, dataProtectionKeysPath, shutdownCancellationToken);
-	var webApplication = WebLauncher.CreateApplication(webConfiguration, applicationProperties, rpcSocket.Connection);
+	var taskManager = new TaskManager(PhantomLogger.Create<TaskManager>("Web"));
+	var webApplication = WebLauncher.CreateApplication(webConfiguration, taskManager, applicationProperties, rpcSocket.Connection);
 
 	using var actorSystem = ActorSystemFactory.Create("Web");
 	
@@ -86,6 +88,7 @@ try {
 		await WebLauncher.Launch(webConfiguration, webApplication);
 	} finally {
 		shutdownCancellationTokenSource.Cancel();
+		await taskManager.Stop();
 		
 		rpcDisconnectSemaphore.Release();
 		await rpcTask;

Web/Phantom.Web/WebLauncher.cs

@@ -1,6 +1,7 @@
 using Microsoft.AspNetCore.DataProtection;
 using Phantom.Common.Messages.Web;
 using Phantom.Utils.Rpc.Runtime;
+using Phantom.Utils.Tasks;
 using Phantom.Web.Services;
 using Serilog;
 using ILogger = Serilog.ILogger;
@@ -12,7 +13,7 @@ static class WebLauncher {
 		public string HttpUrl => "http://" + Host + ":" + Port;
 	}
 	
-	internal static WebApplication CreateApplication(Configuration config, ApplicationProperties applicationProperties, RpcConnectionToServer<IMessageToController> controllerConnection) {
+	internal static WebApplication CreateApplication(Configuration config, TaskManager taskManager, ApplicationProperties applicationProperties, RpcConnectionToServer<IMessageToController> controllerConnection) {
 		var assembly = typeof(WebLauncher).Assembly;
 		var builder = WebApplication.CreateBuilder(new WebApplicationOptions {
 			ApplicationName = assembly.GetName().Name,
@@ -28,6 +29,7 @@ static class WebLauncher {
 			builder.WebHost.UseStaticWebAssets();
 		}
 
+		builder.Services.AddSingleton(taskManager);
 		builder.Services.AddSingleton(applicationProperties);
 		builder.Services.AddSingleton(controllerConnection);
 		builder.Services.AddPhantomServices();