//"Accept" incoming constructor
internal void Accept(long connectId, byte connectNum)
{
Initialize();
_connectId = connectId;
_connectionState = ConnectionState.Connected;
ConnectionNum = connectNum;
//Make initial packet
_connectAcceptPacket = NetConnectAcceptPacket.Make(_connectId, connectNum, false);
//Send
_netManager.SendRaw(_connectAcceptPacket, _remoteEndPoint);
NetUtils.DebugWrite(ConsoleColor.Cyan, "[CC] ConnectId: {0}", _connectId);
}
internal bool ProcessConnectAccept(NetConnectAcceptPacket packet)
{
if (_connectionState != ConnectionState.InProgress)
{
return(false);
}
//check connection id
if (packet.ConnectionId != _connectId)
{
_connectId = packet.ConnectionId;
NetUtils.DebugWrite(ConsoleColor.Cyan, "[NC] Server set new connectId from {0} to {1}", packet.ConnectionId, _connectId);
}
//check connect num
ConnectionNum = packet.ConnectionNumber;
NetUtils.DebugWrite(ConsoleColor.Cyan, "[NC] Received connection accept");
_timeSinceLastPacket = 0;
_connectionState = ConnectionState.Connected;
return(true);
}
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
NetEvent netEvent;
bool peerFound = _peers.TryGetValue(remoteEndPoint, out netPeer);
//Check unconnected
switch (packet.Property)
{
case PacketProperty.DiscoveryRequest:
if (!DiscoveryEnabled)
{
break;
}
netEvent = CreateEvent(NetEventType.DiscoveryRequest);
netEvent.RemoteEndPoint = remoteEndPoint;
netEvent.DataReader.SetSource(packet.RawData, NetConstants.HeaderSize, count);
EnqueueEvent(netEvent);
break;
case PacketProperty.DiscoveryResponse:
netEvent = CreateEvent(NetEventType.DiscoveryResponse);
netEvent.RemoteEndPoint = remoteEndPoint;
netEvent.DataReader.SetSource(packet.RawData, NetConstants.HeaderSize, count);
EnqueueEvent(netEvent);
break;
case PacketProperty.UnconnectedMessage:
if (!UnconnectedMessagesEnabled)
{
break;
}
netEvent = CreateEvent(NetEventType.ReceiveUnconnected);
netEvent.RemoteEndPoint = remoteEndPoint;
netEvent.DataReader.SetSource(packet.RawData, NetConstants.HeaderSize, count);
EnqueueEvent(netEvent);
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--;
}
netEvent = CreateEvent(NetEventType.Disconnect);
netEvent.Peer = netPeer;
netEvent.DataReader.SetSource(packet.RawData, 9, packet.Size);
netEvent.DisconnectReason = disconnectResult == DisconnectResult.Disconnect
? DisconnectReason.RemoteConnectionClose
: DisconnectReason.ConnectionRejected;
EnqueueEvent(netEvent);
}
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))
{
var connectEvent = CreateEvent(NetEventType.Connect);
connectEvent.Peer = netPeer;
EnqueueEvent(connectEvent);
}
break;
case PacketProperty.ConnectRequest:
var connRequest = NetConnectRequestPacket.FromData(packet);
if (connRequest != null)
{
ProcessConnectRequest(remoteEndPoint, netPeer, connRequest);
}
break;
default:
if (peerFound)
{
netPeer.ProcessPacket(packet);
}
break;
}
}
