protected override void OnHandshakeRequest(byte[] buf, int size, IPEndPoint remoteEP)
{
if (HandshakeRequest.TryUnpack(m_CookieProvider, m_RSA, buf, size, out var packet))
{
int offest = 0;
int clientId = BinaryUtil.ReadInt(packet.Payload, ref offest);
PeerEntry peer = null;
if (m_ClientIdPeerMap.TryGetValue(clientId, out int connectionId))
{
m_PeerManager.TryGetValue(connectionId, out peer);
}
else if (HandshakeRequestPayload.TryUnpack(packet.Payload, out var payload))
{
if (m_ClientIdBuffer.Count == ClientIdCapacity)
{
m_ClientIdPeerMap.Remove(m_ClientIdBuffer.Dequeue());
}
m_ClientIdBuffer.Enqueue(clientId);
var id = m_IdGenerator.Gen();
m_ClientIdPeerMap[clientId] = id;
var key = new EncryptorKey(packet, payload, id);
var encryptor = m_EncryptorGenerator.Generate(in key);
peer = new PeerEntry(id, clientId, encryptor, remoteEP);
m_PeerManager.Add(peer);
}
if (peer != null)
{
size = new HandshakeAccept(peer.ConnectionId).Pack(m_SendBuffer, peer.Encryptor);
m_Socket.Send(m_SendBuffer, 0, size, peer.EndPoint);
}
}
}
public EncryptorKey(HandshakeRequest request, HandshakeRequestPayload payload, int connectionId)
{
Cookie = request.Cookie;
MajorVersion = request.MajorVersion;
MinorVersion = request.MinorVersion;
Nonce = payload.ClientId;
ConnectionId = connectionId;
Random = payload.Randam;
}
public async Task <ClientConnectionImpl> Run()
{
try
{
m_Socket = new UdpSocket();
m_Socket.Bind(m_Settings.EndPoint.AddressFamily);
if (!m_Settings.UseP2P)
{
m_Socket.Connect(m_Settings.EndPoint);
}
byte[] cookie;
if (m_Settings.Cookie == null)
{
var buf = new ClientHello(Protocol.MajorVersion, Protocol.MinorVersion).Pack();
var req = new TimeoutRetryableRequester <ServerHello>(WaitServerHello(), () => Send(buf), m_Token);
var res = await req.Run();
cookie = res.Cookie;
}
else
{
cookie = m_Settings.Cookie;
}
{
m_Token.ThrowIfCancellationRequested();
var payload = new HandshakeRequestPayload(Random.GenInt(), m_Randam);
byte[] payloadBuf = payload.Pack();
if (m_Settings.RSA != null)
{
payloadBuf = m_Settings.RSA.Encrypt(payloadBuf, RSAEncryptionPadding.Pkcs1);
}
var request = new HandshakeRequest(cookie, payloadBuf);
m_EncryptorKey = new EncryptorKey(request, payload, 0);
var res = await(new TimeoutRetryableRequester <HandshakeResult>(WaitHandshakeAccept(), () => Send(request.Pack()), m_Token)).Run();
m_Token.ThrowIfCancellationRequested();
return(new ClientConnectionImpl(res.ConnectionId, m_Socket, m_Settings, res.Encryptor, m_EncryptorGenerator));
}
}
catch (Exception ex)
{
Log.Exception(ex);
m_Socket.Dispose();
m_EncryptorGenerator.Dispose();
throw;
}
}
