protected virtual void ReceiveCallback(IAsyncResult AR) { try { Socket Sock = (Socket)AR.AsyncState; int NumBytesRead = Sock.EndReceive(AR); /** Cant do anything with this! **/ if (NumBytesRead == 0) { return; } byte[] TmpBuf = new byte[NumBytesRead]; Buffer.BlockCopy(m_RecvBuf, 0, TmpBuf, 0, NumBytesRead); m_RecvBuf = new byte[11024]; //Clear, to make sure this buffer is always fresh. if (m_TempPacket == null) //No temporary data was stored - beginning of new packet! { //The packet is given an ID of 0x00 because its ID is currently unknown. PacketStream UnknownPacket = new PacketStream(0x00, (ushort)NumBytesRead, TmpBuf); byte ID = UnknownPacket.PeekByte(0); Logger.Log("Received packet: " + ID, LogLevel.info); ushort PacketLength = 0; m_Handler = FindPacketHandler(ID); PacketLength = m_Handler.Length; if (PacketLength == 0) //Variable length! { if (NumBytesRead >= (int)PacketHeaders.ENCRYPTED) PacketLength = UnknownPacket.PeekUShort(1); } else { if (m_TempPacket == null) m_TempPacket = new PacketStream(ID, PacketLength); byte[] TmpBuffer = new byte[NumBytesRead]; //Store the number of bytes that were read in the temporary buffer. Logger.Log("Got data, but not a full header - stored " + NumBytesRead.ToString() + "bytes!\r\n\r\n", LogLevel.info); Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, NumBytesRead); m_TempPacket.WriteBytes(TmpBuffer); TmpBuffer = null; } if (PacketLength != 0) { if (NumBytesRead == PacketLength) { Logger.Log("Got packet - exact length!\r\n\r\n", LogLevel.info); OnPacket(new ProcessedPacket(ID, m_Handler.Encrypted, m_Handler.VariableLength, PacketLength, m_ClientEncryptor, UnknownPacket.ToArray()), m_Handler); } else if (NumBytesRead < PacketLength) { if (m_TempPacket == null) m_TempPacket = new PacketStream(ID, PacketLength); byte[] TmpBuffer = new byte[NumBytesRead]; //Store the number of bytes that were read in the temporary buffer. Logger.Log("Got data, but not a full packet - stored " + NumBytesRead.ToString() + "bytes!\r\n\r\n", LogLevel.info); Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, NumBytesRead); m_TempPacket.WriteBytes(TmpBuffer); TmpBuffer = null; } else if (NumBytesRead > PacketLength) { Logger.Log("Received more bytes than needed for packet. Excess: " + (NumBytesRead - PacketLength) + "\r\n", LogLevel.info); PacketHandler OldHandler = m_Handler; byte[] PacketBuffer = new byte[PacketLength]; Buffer.BlockCopy(UnknownPacket.ToArray(), 0, PacketBuffer, 0, PacketBuffer.Length); OnPacket(new ProcessedPacket(ID, OldHandler.Encrypted, OldHandler.VariableLength, OldHandler.Length, m_ClientEncryptor, PacketBuffer), OldHandler); byte[] TmpBuffer = new byte[NumBytesRead - PacketLength]; Buffer.BlockCopy(UnknownPacket.ToArray(), (NumBytesRead - PacketLength) - 1, TmpBuffer, 0, TmpBuffer.Length); m_TempPacket = new PacketStream(TmpBuffer[0], (ushort)(NumBytesRead - PacketLength), TmpBuffer); m_TempPacket.WriteBytes(TmpBuffer); m_Handler = FindPacketHandler(ID); PacketLength = m_Handler.Length; if (PacketLength == 0) //Variable length! { if (m_TempPacket.BufferLength >= (int)PacketHeaders.ENCRYPTED) { PacketLength = UnknownPacket.PeekUShort(1); m_TempPacket.SetLength(PacketLength); } } TmpBuffer = null; } } } else { m_TempPacket.WriteBytes(TmpBuf); if (m_TempPacket.Length == 0) //Variable length! { if (NumBytesRead >= (int)PacketHeaders.ENCRYPTED) m_TempPacket.SetLength(m_TempPacket.PeekUShort(1)); else { byte[] TmpBuffer = new byte[NumBytesRead]; //Store the number of bytes that were read in the temporary buffer. Logger.Log("Got data, but not a full header - stored " + NumBytesRead.ToString() + "bytes!\r\n\r\n", LogLevel.info); Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, NumBytesRead); m_TempPacket.WriteBytes(TmpBuffer); BeginReceive(); TmpBuffer = null; } } if (m_TempPacket.BufferLength == m_TempPacket.Length) { Logger.Log("Got packet - exact length!\r\n\r\n", LogLevel.info); OnPacket(new ProcessedPacket(m_TempPacket.PacketID, m_Handler.Encrypted, m_Handler.VariableLength, (ushort)m_TempPacket.Length, m_ClientEncryptor, m_TempPacket.ToArray()), m_Handler); } else if (m_TempPacket.BufferLength < m_TempPacket.Length) { byte[] TmpBuffer = new byte[NumBytesRead]; //Store the number of bytes that were read in the temporary buffer. Logger.Log("Got data, but not a full packet - stored " + NumBytesRead.ToString() + "bytes!\r\n\r\n", LogLevel.info); Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, NumBytesRead); m_TempPacket.WriteBytes(TmpBuffer); TmpBuffer = null; } else if (m_TempPacket.BufferLength > m_TempPacket.Length) { Logger.Log("Received more bytes than needed for packet. Excess: " + (m_TempPacket.BufferLength - m_TempPacket.Length) + "\r\n", LogLevel.info); PacketHandler OldHandler = m_Handler; byte[] PacketBuffer = new byte[m_TempPacket.Length]; Buffer.BlockCopy(m_TempPacket.ToArray(), 0, PacketBuffer, 0, PacketBuffer.Length); OnPacket(new ProcessedPacket(OldHandler.ID, OldHandler.Encrypted, OldHandler.VariableLength, OldHandler.Length, m_ClientEncryptor, PacketBuffer), OldHandler); byte[] TmpBuffer = new byte[m_TempPacket.BufferLength - m_TempPacket.Length]; Buffer.BlockCopy(m_TempPacket.ToArray(), ((int)m_TempPacket.BufferLength - (int)m_TempPacket.Length), TmpBuffer, 0, TmpBuffer.Length); m_TempPacket = new PacketStream(TmpBuffer[0], (ushort)(m_TempPacket.BufferLength - m_TempPacket.Length), TmpBuffer); m_Handler = FindPacketHandler(m_TempPacket.PacketID); ushort PacketLength = m_Handler.Length; if (PacketLength == 0) //Variable length! { if (m_TempPacket.BufferLength >= (int)PacketHeaders.ENCRYPTED) { PacketLength = m_TempPacket.PeekUShort(1); m_TempPacket.SetLength(PacketLength); } } } } m_Sock.BeginReceive(m_RecvBuf, 0, m_RecvBuf.Length, SocketFlags.None, new AsyncCallback(ReceiveCallback), m_Sock); } catch (SocketException) { Disconnect(); } }
private void OnPacket(ProcessedPacket packet, PacketHandler handler) { if (OnReceivedData != null) { OnReceivedData(packet); } packetQueue.Enqueue(new KeyValuePair<ProcessedPacket, PacketHandler>(packet, handler)); //handler.Handler(this, packet); }
private void OnPacket(ProcessedPacket packet, PacketHandler handler) { if (OnReceivedData != null) { OnReceivedData(packet); } handler.Handler(this, packet); }