private void ReceiveCallback(IAsyncResult AR)
{
try
{
Socket Sock = (Socket)AR.AsyncState;
int NumBytesRead = Sock.EndReceive(AR);
Log.LogThis("Received: " + NumBytesRead + " bytes!", eloglevel.info);
byte[] TmpBuf = new byte[NumBytesRead];
Buffer.BlockCopy(m_RecvBuf, 0, TmpBuf, 0, NumBytesRead);
//The packet is given an ID of 0x00 because its ID is currently unknown.
PacketStream TempPacket = new PacketStream(0x00, NumBytesRead, TmpBuf);
byte ID = TempPacket.PeekByte(0);
int PacketLength = 0;
bool FoundMatchingID = false;
FoundMatchingID = FindMatchingPacketID(ID);
if (FoundMatchingID)
{
PacketLength = m_PacketIDs[ID];
Log.LogThis("PacketLength: " + PacketLength, eloglevel.info);
Log.LogThis("Found matching PacketID (" + ID + ")!\r\n\r\n", eloglevel.info);
if (NumBytesRead == PacketLength)
{
Log.LogThis("Got packet - exact length!\r\n\r\n", eloglevel.info);
m_RecvBuf = new byte[11024];
OnReceivedData(new PacketStream(ID, PacketLength, TempPacket.ToArray()));
}
else if (NumBytesRead < PacketLength)
{
m_TempPacket = new PacketStream(ID, PacketLength);
byte[] TmpBuffer = new byte[NumBytesRead];
//Store the number of bytes that were read in the temporary buffer.
Log.LogThis("Got data, but not a full packet - stored " +
NumBytesRead.ToString() + "bytes!\r\n\r\n", eloglevel.info);
Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, NumBytesRead);
m_TempPacket.WriteBytes(TmpBuffer);
//And reset the buffers!
m_RecvBuf = new byte[11024];
TmpBuffer = null;
}
else if (PacketLength == 0)
{
Log.LogThis("Received variable length packet!\r\n", eloglevel.info);
if (NumBytesRead > 2)
{
PacketLength = TempPacket.PeekByte(1);
if (NumBytesRead == PacketLength)
{
Log.LogThis("Received exact number of bytes for packet!\r\n", eloglevel.info);
m_RecvBuf = new byte[11024];
m_TempPacket = null;
OnReceivedData(new PacketStream(ID, PacketLength, TempPacket.ToArray()));
}
else if (NumBytesRead < PacketLength)
{
Log.LogThis("Didn't receive entire packet - stored: " + PacketLength + " bytes!\r\n",
eloglevel.info);
TempPacket.SetLength(PacketLength);
m_TempPacket = TempPacket;
m_RecvBuf = new byte[11024];
}
else if (NumBytesRead > PacketLength)
{
Log.LogThis("Received more bytes than needed for packet. Excess: " +
(NumBytesRead - PacketLength) + "\r\n", eloglevel.info);
byte[] TmpBuffer = new byte[NumBytesRead - PacketLength];
Buffer.BlockCopy(TempPacket.ToArray(), 0, TmpBuffer, 0, TmpBuffer.Length);
m_TempPacket = new PacketStream(TmpBuffer[0], NumBytesRead - PacketLength,
TmpBuffer);
byte[] PacketBuffer = new byte[PacketLength];
Buffer.BlockCopy(TempPacket.ToArray(), 0, PacketBuffer, 0, PacketBuffer.Length);
m_RecvBuf = new byte[11024];
OnReceivedData(new PacketStream(ID, PacketLength, PacketBuffer));
}
}
}
}
else
{
if (m_TempPacket != null)
{
if (m_TempPacket.Length < m_TempPacket.BufferLength)
{
//Received the exact number of bytes needed to complete the stored packet.
if ((m_TempPacket.BufferLength + NumBytesRead) == m_TempPacket.Length)
{
byte[] TmpBuffer = new byte[NumBytesRead];
Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, NumBytesRead);
m_RecvBuf = new byte[11024];
TmpBuffer = null;
}
//Received more than the number of bytes needed to complete the packet!
else if ((m_TempPacket.BufferLength + NumBytesRead) > m_TempPacket.Length)
{
int Target = (int)((m_TempPacket.BufferLength + NumBytesRead) - m_TempPacket.Length);
byte[] TmpBuffer = new byte[Target];
Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, Target);
m_TempPacket.WriteBytes(TmpBuffer);
//Now we have a full packet, so call the received event!
OnReceivedData(new PacketStream(m_TempPacket.PacketID,
(int)m_TempPacket.Length, m_TempPacket.ToArray()));
//Copy the remaining bytes in the receiving buffer.
TmpBuffer = new byte[NumBytesRead - Target];
Buffer.BlockCopy(m_RecvBuf, Target, TmpBuffer, 0, (NumBytesRead - Target));
//Give the temporary packet an ID of 0x00 since we don't know its ID yet.
TempPacket = new PacketStream(0x00, NumBytesRead - Target, TmpBuffer);
ID = TempPacket.PeekByte(0);
//This SHOULD be an existing ID, but let's sanity-check it...
if (FindMatchingPacketID(ID))
{
m_TempPacket = new PacketStream(ID, m_PacketIDs[ID], TempPacket.ToArray());
//Congratulations, you just received another packet!
if (m_TempPacket.Length == m_TempPacket.BufferLength)
{
OnReceivedData(new PacketStream(m_TempPacket.PacketID,
(int)m_TempPacket.Length, m_TempPacket.ToArray()));
//No more data to store on this read, so reset everything...
m_TempPacket = null;
TmpBuffer = null;
m_RecvBuf = new byte[11024];
}
}
else
{
//Houston, we have a problem (this should never occur)!
}
}
}
}
}
m_Sock.BeginReceive(m_RecvBuf, 0, m_RecvBuf.Length, SocketFlags.None,
new AsyncCallback(ReceiveCallback), m_Sock);
}
catch (SocketException E)
{
Log.LogThis("SocketException: " + E.ToString(), eloglevel.info);
Disconnect();
}
}
private void ReceiveCallback(IAsyncResult AR)
{
try
{
Socket Sock = (Socket)AR.AsyncState;
int NumBytesRead = Sock.EndReceive(AR);
/** Cant do anything with this! **/
if (NumBytesRead == 0)
{
return;
}
Log.LogThis("Received: " + NumBytesRead + " bytes!", eloglevel.info);
byte[] TmpBuf = new byte[NumBytesRead];
Buffer.BlockCopy(m_RecvBuf, 0, TmpBuf, 0, NumBytesRead);
//The packet is given an ID of 0x00 because its ID is currently unknown.
PacketStream TempPacket = new PacketStream(0x00, NumBytesRead, TmpBuf);
byte ID = TempPacket.PeekByte(0);
int PacketLength = 0;
var handler = FindPacketHandler(ID);
if (handler != null)
{
PacketLength = handler.Length;
Log.LogThis("PacketLength: " + PacketLength, eloglevel.info);
Log.LogThis("Found matching PacketID (" + ID + ")!\r\n\r\n", eloglevel.info);
if (NumBytesRead == PacketLength)
{
Log.LogThis("Got packet - exact length!\r\n\r\n", eloglevel.info);
m_RecvBuf = new byte[11024];
OnPacket(new PacketStream(ID, PacketLength, TempPacket.ToArray()), handler);
}
else if (NumBytesRead < PacketLength)
{
m_TempPacket = new PacketStream(ID, PacketLength);
byte[] TmpBuffer = new byte[NumBytesRead];
//Store the number of bytes that were read in the temporary buffer.
Log.LogThis("Got data, but not a full packet - stored " +
NumBytesRead.ToString() + "bytes!\r\n\r\n", eloglevel.info);
Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, NumBytesRead);
m_TempPacket.WriteBytes(TmpBuffer);
//And reset the buffers!
m_RecvBuf = new byte[11024];
TmpBuffer = null;
}
else if (PacketLength == 0)
{
Log.LogThis("Received variable length packet!\r\n", eloglevel.info);
if (NumBytesRead > (int)PacketHeaders.UNENCRYPTED) //Header is 3 bytes.
{
PacketLength = TempPacket.PeekUShort(1);
if (NumBytesRead == PacketLength)
{
Log.LogThis("Received exact number of bytes for packet!\r\n", eloglevel.info);
m_RecvBuf = new byte[11024];
m_TempPacket = null;
OnPacket(new PacketStream(ID, PacketLength, TempPacket.ToArray()), handler);
}
else if (NumBytesRead < PacketLength)
{
Log.LogThis("Didn't receive entire packet - stored: " + PacketLength + " bytes!\r\n",
eloglevel.info);
TempPacket.SetLength(PacketLength);
m_TempPacket = TempPacket;
m_RecvBuf = new byte[11024];
}
else if (NumBytesRead > PacketLength)
{
Log.LogThis("Received more bytes than needed for packet. Excess: " +
(NumBytesRead - PacketLength) + "\r\n", eloglevel.info);
byte[] TmpBuffer = new byte[NumBytesRead - PacketLength];
Buffer.BlockCopy(TempPacket.ToArray(), 0, TmpBuffer, 0, TmpBuffer.Length);
m_TempPacket = new PacketStream(TmpBuffer[0], NumBytesRead - PacketLength,
TmpBuffer);
byte[] PacketBuffer = new byte[PacketLength];
Buffer.BlockCopy(TempPacket.ToArray(), 0, PacketBuffer, 0, PacketBuffer.Length);
m_RecvBuf = new byte[11024];
OnPacket(new PacketStream(ID, PacketLength, PacketBuffer), handler);
}
}
}
}
else
{
if (m_TempPacket != null)
{
if (m_TempPacket.Length < m_TempPacket.BufferLength)
{
//Received the exact number of bytes needed to complete the stored packet.
if ((m_TempPacket.BufferLength + NumBytesRead) == m_TempPacket.Length)
{
byte[] TmpBuffer = new byte[NumBytesRead];
Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, NumBytesRead);
m_RecvBuf = new byte[11024];
TmpBuffer = null;
}
//Received more than the number of bytes needed to complete the packet!
else if ((m_TempPacket.BufferLength + NumBytesRead) > m_TempPacket.Length)
{
int Target = (int)((m_TempPacket.BufferLength + NumBytesRead) - m_TempPacket.Length);
byte[] TmpBuffer = new byte[Target];
Buffer.BlockCopy(m_RecvBuf, 0, TmpBuffer, 0, Target);
m_TempPacket.WriteBytes(TmpBuffer);
//Now we have a full packet, so call the received event!
OnPacket(new PacketStream(m_TempPacket.PacketID,
(int)m_TempPacket.Length, m_TempPacket.ToArray()), handler);
//Copy the remaining bytes in the receiving buffer.
TmpBuffer = new byte[NumBytesRead - Target];
Buffer.BlockCopy(m_RecvBuf, Target, TmpBuffer, 0, (NumBytesRead - Target));
//Give the temporary packet an ID of 0x00 since we don't know its ID yet.
TempPacket = new PacketStream(0x00, NumBytesRead - Target, TmpBuffer);
ID = TempPacket.PeekByte(0);
handler = FindPacketHandler(ID);
//This SHOULD be an existing ID, but let's sanity-check it...
if (handler != null)
{
m_TempPacket = new PacketStream(ID, handler.Length, TempPacket.ToArray());
//Congratulations, you just received another packet!
if (m_TempPacket.Length == m_TempPacket.BufferLength)
{
OnPacket(new PacketStream(m_TempPacket.PacketID,
(int)m_TempPacket.Length, m_TempPacket.ToArray()), handler);
//No more data to store on this read, so reset everything...
m_TempPacket = null;
TmpBuffer = null;
m_RecvBuf = new byte[11024];
}
}
else
{
//Houston, we have a problem (this should never occur)!
}
}
}
}
}
m_Sock.BeginReceive(m_RecvBuf, 0, m_RecvBuf.Length, SocketFlags.None,
new AsyncCallback(ReceiveCallback), m_Sock);
}
catch (SocketException E)
{
Log.LogThis("SocketException: " + E.ToString(), eloglevel.info);
Disconnect();
}
}