public override void Unload()
{
ThreadRunning = false;
if (CanInterrupt)
{
UpdateThread.Interrupt();
}
base.Unload();
}
private void SocketDataRecieved(IAsyncResult e)
{
RecieveStruct state = (RecieveStruct)e.AsyncState;
int ecode = 0;
try
{
ecode = state.sock.EndReceive(e);
}
catch (SocketException) {
sockets.Remove(state.sock);
return;
}
catch (ObjectDisposedException) {
sockets.Remove(state.sock);
return;
}
if (ecode == 0)
{
//We should NEVER get zero data.
sockets.Remove(state.sock);
return;
}
WSPacket packet;
try
{
packet = DecodePacket(state.buffer, state.sock);
}
catch (Exception) {
CloseSocket(state.sock);
return;
}
if (packet.opcode == OPCODE_INTERNAL_FAIL)
{
//Decoder could not get any more data from the packet, so the connection must be terminated.
sockets.Remove(state.sock);
return;
}
if (packet.lastPacket)
{
if (fragments.ContainsKey(state.sock))
{
byte[] lastPacket = fragments[state.sock];
byte[] buffer = new byte[lastPacket.Length + packet.binaryData.Length];
Array.Copy(lastPacket, buffer, lastPacket.Length);
Array.Copy(packet.binaryData, 0, buffer, lastPacket.Length, packet.binaryData.Length);
fragments[state.sock] = buffer;
packet.binaryData = buffer;
if (packet.opcode == OPCODE_TEXT)
{
packet.textData = Encoding.UTF8.GetString(packet.binaryData);
}
fragments.Remove(state.sock);
}
packet.sender = state.sock;
//Do something with the packet.
if (packet.opcode == OPCODE_CLOSE)
{
state.sock.Close();
}
else if (packet.opcode == OPCODE_PING)
{
WSPacket newpack = new WSPacket();
packet.opcode = 0xA;
packet.length = 0;
packet.lastPacket = true;
byte[] newdata = EncodePacket(newpack);
state.sock.BeginSend(newdata, 0, newdata.Length, SocketFlags.None, SocketDataSent, state);
}
else
{
processingQueue.Enqueue(packet);
if (CanInterrupt)
{
UpdateThread.Interrupt();
}
}
}
else
{
if (!fragments.ContainsKey(state.sock))
{
fragments.Add(state.sock, packet.binaryData);
}
else
{
byte[] lastPacket = fragments[state.sock];
byte[] buffer = new byte[lastPacket.Length + packet.binaryData.Length];
lastPacket.CopyTo(buffer, 0);
packet.binaryData.CopyTo(buffer, lastPacket.Length);
fragments[state.sock] = buffer;
}
}
if (state.sock.Connected)
{
state.sock.BeginReceive(state.buffer, 0, WSREADSIZE, SocketFlags.None, SocketDataRecieved, state);
}
}
