1 files changed, 103 insertions, 0 deletions
diff --git a/Impostor-dev/src/Impostor.Server/Net/Manager/ClientManager.cs b/Impostor-dev/src/Impostor.Server/Net/Manager/ClientManager.cs
new file mode 100644
index 0000000..51e22d8
--- /dev/null
+++ b/Impostor-dev/src/Impostor.Server/Net/Manager/ClientManager.cs
@@ -0,0 +1,103 @@
+using System.Collections.Concurrent;
+using System.Collections.Generic;
+using System.Linq;
+using System.Threading;
+using System.Threading.Tasks;
+using Impostor.Api.Innersloth;
+using Impostor.Api.Net;
+using Impostor.Api.Net.Messages;
+using Impostor.Api.Net.Messages.S2C;
+using Impostor.Hazel;
+using Impostor.Server.Config;
+using Impostor.Server.Net.Factories;
+using Microsoft.Extensions.Logging;
+
+namespace Impostor.Server.Net.Manager
+{
+    internal partial class ClientManager
+    {
+        private static HashSet<int> SupportedVersions { get; } = new HashSet<int>
+        {
+            GameVersion.GetVersion(2020, 09, 07), // 2020.09.07 - 2020.09.22
+            GameVersion.GetVersion(2020, 10, 08), // 2020.10.08
+            GameVersion.GetVersion(2020, 11, 17), // 2020.11.17
+        };
+
+        private readonly ILogger<ClientManager> _logger;
+        private readonly ConcurrentDictionary<int, ClientBase> _clients;
+        private readonly IClientFactory _clientFactory;
+        private int _idLast;
+
+        public ClientManager(ILogger<ClientManager> logger, IClientFactory clientFactory)
+        {
+            _logger = logger;
+            _clientFactory = clientFactory;
+            _clients = new ConcurrentDictionary<int, ClientBase>();
+        }
+
+        public IEnumerable<ClientBase> Clients => _clients.Values;
+
+        public int NextId()
+        {
+            var clientId = Interlocked.Increment(ref _idLast);
+
+            if (clientId < 1)
+            {
+                // Super rare but reset the _idLast because of overflow.
+                _idLast = 0;
+
+                // And get a new id.
+                clientId = Interlocked.Increment(ref _idLast);
+            }
+
+            return clientId;
+        }
+
+        public async ValueTask RegisterConnectionAsync(IHazelConnection connection, string name, int clientVersion)
+        {
+            if (!SupportedVersions.Contains(clientVersion))
+            {
+                using var packet = MessageWriter.Get(MessageType.Reliable);
+                Message01JoinGameS2C.SerializeError(packet, false, DisconnectReason.IncorrectVersion);
+                await connection.SendAsync(packet);
+                return;
+            }
+
+            if (name.Length > 10)
+            {
+                using var packet = MessageWriter.Get(MessageType.Reliable);
+                Message01JoinGameS2C.SerializeError(packet, false, DisconnectReason.Custom, DisconnectMessages.UsernameLength);
+                await connection.SendAsync(packet);
+                return;
+            }
+
+            if (string.IsNullOrWhiteSpace(name) || !name.All(TextBox.IsCharAllowed))
+            {
+                using var packet = MessageWriter.Get(MessageType.Reliable);
+                Message01JoinGameS2C.SerializeError(packet, false, DisconnectReason.Custom, DisconnectMessages.UsernameIllegalCharacters);
+                await connection.SendAsync(packet);
+                return;
+            }
+
+            var client = _clientFactory.Create(connection, name, clientVersion);
+            var id = NextId();
+
+            client.Id = id;
+            _logger.LogTrace("Client connected.");
+            _clients.TryAdd(id, client);
+        }
+
+        public void Remove(IClient client)
+        {
+            _logger.LogTrace("Client disconnected.");
+            _clients.TryRemove(client.Id, out _);
+        }
+
+        public bool Validate(IClient client)
+        {
+            return client.Id != 0
+                   && _clients.TryGetValue(client.Id, out var registeredClient)
+                   && ReferenceEquals(client, registeredClient);
+        }
+    }
+}
+\ No newline at end of file