1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
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);
}
}
}
|
