internal void UpdateRequest(NetConnectRequestPacket connRequest)
{
if (connRequest.ConnectionTime >= ConnectionTime)
{
ConnectionTime = connRequest.ConnectionTime;
ConnectionNumber = connRequest.ConnectionNumber;
}
}
internal ConnectRequestResult ProcessConnectRequest(NetConnectRequestPacket connRequest)
{
//current or new request
switch (_connectionState)
{
//P2P case
case ConnectionState.Outgoing:
//fast check
if (connRequest.ConnectionTime < _connectTime)
{
return(ConnectRequestResult.P2PLose);
}
//slow rare case check
else if (connRequest.ConnectionTime == _connectTime)
{
var remoteBytes = EndPoint.Serialize();
var localBytes = connRequest.TargetAddress;
for (int i = remoteBytes.Size - 1; i >= 0; i--)
{
byte rb = remoteBytes[i];
if (rb == localBytes[i])
{
continue;
}
if (rb < localBytes[i])
{
return(ConnectRequestResult.P2PLose);
}
}
}
break;
case ConnectionState.Connected:
//Old connect request
if (connRequest.ConnectionTime == _connectTime)
{
//just reply accept
NetManager.SendRaw(_connectAcceptPacket, EndPoint);
}
//New connect request
else if (connRequest.ConnectionTime > _connectTime)
{
return(ConnectRequestResult.Reconnection);
}
break;
case ConnectionState.Disconnected:
case ConnectionState.ShutdownRequested:
if (connRequest.ConnectionTime >= _connectTime)
{
return(ConnectRequestResult.NewConnection);
}
break;
}
return(ConnectRequestResult.None);
}
//"Connect to" constructor
internal NetPeer(NetManager netManager, IPEndPoint remoteEndPoint, int id, byte connectNum, NetDataWriter connectData)
: this(netManager, remoteEndPoint, id)
{
_connectTime = DateTime.UtcNow.Ticks;
_connectionState = ConnectionState.Outgoing;
ConnectionNum = connectNum;
//Make initial packet
_connectRequestPacket = NetConnectRequestPacket.Make(connectData, _connectTime);
_connectRequestPacket.ConnectionNumber = connectNum;
//Send request
NetManager.SendRaw(_connectRequestPacket, EndPoint);
NetDebug.Write(NetLogLevel.Trace, "[CC] ConnectId: {0}, ConnectNum: {1}", _connectTime, connectNum);
}
//"Connect to" constructor
internal NetPeer(NetManager netManager, IPEndPoint remoteEndPoint, byte connectNum, NetDataWriter connectData) : this(netManager, remoteEndPoint)
{
Initialize();
_connectId = DateTime.UtcNow.Ticks;
_connectionState = ConnectionState.InProgress;
ConnectionNum = connectNum;
//Make initial packet
_connectRequestPacket = NetConnectRequestPacket.Make(connectData, _connectId);
_connectRequestPacket.ConnectionNumber = connectNum;
//Send request
_netManager.SendRaw(_connectRequestPacket, _remoteEndPoint);
NetUtils.DebugWrite(ConsoleColor.Cyan, "[CC] ConnectId: {0}, ConnectNum: {1}", _connectId, connectNum);
}
internal void UpdateRequest(NetConnectRequestPacket connectRequest)
{
//old request
if (connectRequest.ConnectionTime < InternalPacket.ConnectionTime)
{
return;
}
if (connectRequest.ConnectionTime == InternalPacket.ConnectionTime &&
connectRequest.ConnectionNumber == InternalPacket.ConnectionNumber)
{
return;
}
InternalPacket = connectRequest;
}
internal ConnectRequestResult ProcessConnectRequest(NetConnectRequestPacket connRequest)
{
//current or new request
switch (_connectionState)
{
//P2P case or just ID update
case ConnectionState.InProgress:
case ConnectionState.Incoming:
_connectionState = ConnectionState.Incoming;
//change connect id if newer
if (connRequest.ConnectionId >= _connectId)
{
//Change connect id
_connectId = connRequest.ConnectionId;
ConnectionNum = connRequest.ConnectionNumber;
}
return(_connectionState == ConnectionState.InProgress
? ConnectRequestResult.P2PConnection
: ConnectRequestResult.None);
case ConnectionState.Connected:
//Old connect request
if (connRequest.ConnectionId == _connectId)
{
//just reply accept
_netManager.SendRaw(_connectAcceptPacket, _remoteEndPoint);
}
//New connect request
else if (connRequest.ConnectionId > _connectId)
{
return(ConnectRequestResult.Reconnection);
}
break;
case ConnectionState.Disconnected:
case ConnectionState.ShutdownRequested:
if (connRequest.ConnectionId >= _connectId)
{
return(ConnectRequestResult.NewConnection);
}
break;
}
return(ConnectRequestResult.None);
}
internal ConnectRequestResult ProcessConnectRequest(NetConnectRequestPacket connRequest)
{
//current or new request
switch (_connectionState)
{
//P2P case or just ID update
case ConnectionState.Outgoing:
//change connect id if newer
if (connRequest.ConnectionTime >= _connectTime)
{
//Change connect id
_connectTime = connRequest.ConnectionTime;
ConnectionNum = connRequest.ConnectionNumber;
}
return(ConnectRequestResult.P2PConnection);
case ConnectionState.Connected:
//Old connect request
if (connRequest.ConnectionTime == _connectTime)
{
//just reply accept
NetManager.SendRaw(_connectAcceptPacket, EndPoint);
}
//New connect request
else if (connRequest.ConnectionTime > _connectTime)
{
return(ConnectRequestResult.Reconnection);
}
break;
case ConnectionState.Disconnected:
case ConnectionState.ShutdownRequested:
if (connRequest.ConnectionTime >= _connectTime)
{
return(ConnectRequestResult.NewConnection);
}
break;
}
return(ConnectRequestResult.None);
}
private void DataReceived(byte[] reusableBuffer, int count, IPEndPoint remoteEndPoint)
{
#if STATS_ENABLED
Statistics.PacketsReceived++;
Statistics.BytesReceived += (uint)count;
#endif
//Try read packet
NetPacket packet = NetPacketPool.GetPacket(count, false);
if (!packet.FromBytes(reusableBuffer, 0, count))
{
NetPacketPool.Recycle(packet);
NetUtils.DebugWriteError("[NM] DataReceived: bad!");
return;
}
//get peer
//Check normal packets
NetPeer netPeer;
//old packets protection
bool peerFound = _peers.TryGetValue(remoteEndPoint, out netPeer);
//Check unconnected
switch (packet.Property)
{
case PacketProperty.DiscoveryRequest:
if (!DiscoveryEnabled)
{
break;
}
CreateEvent(NetEvent.EType.DiscoveryRequest, remoteEndPoint: remoteEndPoint, readerSource: packet);
break;
case PacketProperty.DiscoveryResponse:
CreateEvent(NetEvent.EType.DiscoveryResponse, remoteEndPoint: remoteEndPoint, readerSource: packet);
break;
case PacketProperty.UnconnectedMessage:
if (!UnconnectedMessagesEnabled)
{
break;
}
CreateEvent(NetEvent.EType.ReceiveUnconnected, remoteEndPoint: remoteEndPoint, readerSource: packet);
break;
case PacketProperty.NatIntroduction:
case PacketProperty.NatIntroductionRequest:
case PacketProperty.NatPunchMessage:
if (NatPunchEnabled)
{
NatPunchModule.ProcessMessage(remoteEndPoint, packet);
}
break;
case PacketProperty.Disconnect:
if (peerFound)
{
var disconnectResult = netPeer.ProcessDisconnect(packet);
if (disconnectResult == DisconnectResult.None)
{
NetPacketPool.Recycle(packet);
return;
}
if (disconnectResult == DisconnectResult.Disconnect)
{
_connectedPeersCount--;
}
CreateEvent(
NetEvent.EType.Disconnect,
netPeer,
disconnectReason: disconnectResult == DisconnectResult.Disconnect
? DisconnectReason.RemoteConnectionClose
: DisconnectReason.ConnectionRejected,
readerSource: packet
);
}
else
{
NetPacketPool.Recycle(packet);
}
//Send shutdown
SendRaw(new[] { (byte)PacketProperty.ShutdownOk }, 0, 1, remoteEndPoint);
break;
case PacketProperty.ConnectAccept:
var connAccept = NetConnectAcceptPacket.FromData(packet);
if (connAccept != null && peerFound && netPeer.ProcessConnectAccept(connAccept))
{
CreateEvent(NetEvent.EType.Connect, netPeer);
}
break;
case PacketProperty.ConnectRequest:
var connRequest = NetConnectRequestPacket.FromData(packet);
if (connRequest != null)
{
ProcessConnectRequest(remoteEndPoint, netPeer, connRequest);
}
break;
default:
if (peerFound)
{
netPeer.ProcessPacket(packet);
}
break;
}
}
private void ProcessConnectRequest(
IPEndPoint remoteEndPoint,
NetPeer netPeer,
NetConnectRequestPacket connRequest)
{
byte connectionNumber = connRequest.ConnectionNumber;
NetUtils.DebugWrite("ConnectRequest LastId: {0}, NewId: {1}, EP: {2}", netPeer.ConnectId, connRequest.ConnectionId, remoteEndPoint);
//if we have peer
if (netPeer != null)
{
var processResult = netPeer.ProcessConnectRequest(connRequest);
switch (processResult)
{
case ConnectRequestResult.Reconnection:
_connectedPeersCount--;
CreateEvent(NetEvent.EType.Disconnect, netPeer, disconnectReason: DisconnectReason.RemoteConnectionClose);
_peers.RemovePeer(netPeer);
//go to new connection
break;
case ConnectRequestResult.NewConnection:
_peers.RemovePeer(netPeer);
//go to new connection
break;
case ConnectRequestResult.P2PConnection:
CreateEvent(
NetEvent.EType.ConnectionRequest,
connectionRequest: new ConnectionRequest(
netPeer.ConnectId,
connectionNumber,
ConnectionRequestType.PeerToPeer,
connRequest.Data,
netPeer,
this)
);
return;
default:
//no operations needed
return;
}
//ConnectRequestResult.NewConnection
//Set next connection number
connectionNumber = (byte)((netPeer.ConnectionNum + 1) % NetConstants.MaxConnectionNumber);
//To reconnect peer
}
//Add new peer and craete ConnectRequest event
NetUtils.DebugWrite("[NM] Creating request event: " + connRequest.ConnectionId);
netPeer = new NetPeer(this, remoteEndPoint);
if (_peers.TryAdd(netPeer))
{
CreateEvent(NetEvent.EType.ConnectionRequest, connectionRequest: new ConnectionRequest(
connRequest.ConnectionId,
connectionNumber,
ConnectionRequestType.Incoming,
connRequest.Data,
netPeer,
this));
}
}
internal ConnectionRequest(IPEndPoint remoteEndPoint, NetConnectRequestPacket requestPacket, NetManager listener)
{
InternalPacket = requestPacket;
RemoteEndPoint = remoteEndPoint;
_listener = listener;
}