Exemplo n.º 1
0
        protected void SendData(Socket socket, string data)
        {
            WSPacket packet = new WSPacket();

            packet.opcode     = 0x1;
            packet.textData   = data;
            packet.lastPacket = true;
            byte[]        newdata = EncodePacket(packet);
            RecieveStruct rs      = new RecieveStruct();

            rs.sock = socket;
            socket.BeginSend(newdata, 0, newdata.Length, SocketFlags.None, SocketDataSent, rs);
        }
Exemplo n.º 2
0
        protected void CloseSocket(Socket socket){
            WSPacket packet = new WSPacket();
            packet.opcode = OPCODE_CLOSE;
            packet.lastPacket = true;
            byte[] newdata = EncodePacket(packet);
            try
            {
                socket.Send(newdata);
                socket.Close();
            }
            catch(ObjectDisposedException){

            }
        }
Exemplo n.º 3
0
        protected void CloseSocket(Socket socket)
        {
            WSPacket packet = new WSPacket();

            packet.opcode     = OPCODE_CLOSE;
            packet.lastPacket = true;
            byte[] newdata = EncodePacket(packet);
            try
            {
                socket.Send(newdata);
                socket.Close();
            }
            catch (ObjectDisposedException) {
            }
        }
Exemplo n.º 4
0
        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);
            }
        }
Exemplo n.º 5
0
 public virtual void HandlePacket(WSPacket packet){
     SendData(packet.sender, packet.textData);
 }
Exemplo n.º 6
0
        private byte[] EncodePacket(WSPacket packet){
            int packetSize = 2;
            if (packet.opcode == 1)
            {
                packet.binaryData = Encoding.UTF8.GetBytes(packet.textData);
                packet.length = (ulong)packet.binaryData.Length;
            }
            if ((int)packet.length <= 125)
            {
                //Nothing
            }
            else if (packet.length < (long)Int16.MaxValue)
            {
                if (packet.length < byte.MaxValue)
                {
                    packetSize += 2;
                }
                else
                {
                    packetSize += 2;
                }
            }
            else
            {
                packetSize += 8;
            }
            packetSize += (int)packet.length;
            byte[] bpacket = new byte[packetSize];


            if (packet.lastPacket)
            {
                bpacket[0] = (byte)128;
            }
            bpacket[0] = Convert.ToByte((int)bpacket[0] + packet.opcode);//Fin + OpCode

            if (packet.length <= 125)
            {
                bpacket[1] = (byte)packet.length; 
            }
            else if (packet.length < (int)Int16.MaxValue)
            {
                bpacket[1] = 126;
            }
            else
            {
                bpacket[1] = 127;
            }
            bpacket[1] = Convert.ToByte((int)bpacket[1]);//Masking

            if (packetSize > 125)
            {
                byte[] blen = BitConverter.GetBytes(packet.length);
                if (BitConverter.IsLittleEndian)
                {
                    Array.Reverse(blen);
                }
                if (packetSize <= Int16.MaxValue)
                {
                    blen = new byte[2]{ blen[6], blen[7] };
                }
                for (int i = 0; i < blen.Length; i++)
                {
                    bpacket[2 + i] = blen[i];
                }
            }

            //Masking - Don't do it
            //byte[] mask = new byte[4];
            //new Random().NextBytes(mask);
            int offset = (packetSize - (int)packet.length);
            //for (int i = 0; i < 4; i++)
            //{
            //    bpacket[offset + i] = mask[i];
            //}
            //offset += 4;
            for (int i = 0; i < (int)packet.length; i++)
            {
                bpacket[offset + i] = packet.binaryData[i];
            }

            return bpacket;
        }
Exemplo n.º 7
0
        private WSPacket DecodePacket(byte[] buffer,Socket sock){
            int byteCounter = 0;
            WSPacket packet = new WSPacket();
            BitArray byteZero = BytesToBits(buffer,0,1);
            bool moreToCome = !byteZero.Get(0);
            int opcode = Convert.ToInt16(byteZero.Get(4))*8 + 
                Convert.ToInt16(byteZero.Get(5))*4 +  
                Convert.ToInt16(byteZero.Get(6))*2 +  
                Convert.ToInt16(byteZero.Get(7))*1;
            BitArray byteOne = BytesToBits(buffer,1,1);
            bool isMasked = byteOne.Get(0);
            if (!isMasked)
            {
                sock.Disconnect(false);
            }
            UInt64 length = 0;
            uint factor = 0;
            uint bit = 0;
            for (var i = 1; i < 8; i++)
            {
                factor = (uint)Math.Pow(2, 7 - i);
                bit = Convert.ToUInt16(byteOne.Get(i));
                length += bit * factor;
            }

            if (length > 125)
            {
                int bytesToGet = 2;
                if (length == 127)
                {
                    bytesToGet = 8;
                }

                byte[] lenbuffer = new byte[bytesToGet];
                if (sock.Connected)
                {
                    sock.Receive(lenbuffer);
                }
                else
                {
                    packet.opcode = OPCODE_INTERNAL_FAIL;
                    return packet;
                }
                BitArray lengthArr = BytesToBits(lenbuffer,0,bytesToGet);
                byteCounter += bytesToGet;
                length = 0;
                for (var i = 0; i < bytesToGet * 8; i++)
                {
                    factor = (uint)Math.Pow(2, ((bytesToGet * 8)-1) - i);
                    bit = Convert.ToUInt16(lengthArr.Get(i));
                    length += bit * factor;
                }
            }

            byte[] mask = new byte[4];
            if (sock.Connected)
            {
                sock.Receive(mask);
            }
            else
            {
                packet.opcode = OPCODE_INTERNAL_FAIL;
                return packet;
            }

            byte[] edata = new byte[length];
            byte[] ddata = new byte[edata.Length];

            if (sock.Connected)
            {
                int bcount=0;
                while (bcount < edata.Length)
                {
                    bcount += sock.Receive(edata,bcount,edata.Length-bcount,0);
                }
            }
            else
            {
                packet.opcode = OPCODE_INTERNAL_FAIL;
                return packet;
            }

            for (int i = 0; i < edata.Length; i++) {
                ddata[i] = (byte)((int)edata[i] ^ mask[i % 4]);
            }
            packet.binaryData = ddata;
            packet.length = length;
            packet.opcode = (uint)opcode;
            packet.lastPacket = !moreToCome;
            if (opcode == 1 && !moreToCome)
            {
                packet.textData = Encoding.UTF8.GetString(ddata);
            }
            return packet;
        }
Exemplo n.º 8
0
 protected void SendData(Socket socket,string data){
     WSPacket packet = new WSPacket();
     packet.opcode = 0x1;
     packet.textData = data;
     packet.lastPacket = true;
     byte[] newdata = EncodePacket(packet);
     RecieveStruct rs = new RecieveStruct();
     rs.sock = socket;
     socket.BeginSend(newdata, 0, newdata.Length, SocketFlags.None, SocketDataSent, rs);
 }
Exemplo n.º 9
0
        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);
            }
        }
Exemplo n.º 10
0
 public virtual void HandlePacket(WSPacket packet)
 {
     SendData(packet.sender, packet.textData);
 }
Exemplo n.º 11
0
        private byte[] EncodePacket(WSPacket packet)
        {
            int packetSize = 2;

            if (packet.opcode == 1)
            {
                packet.binaryData = Encoding.UTF8.GetBytes(packet.textData);
                packet.length     = (ulong)packet.binaryData.Length;
            }
            if ((int)packet.length <= 125)
            {
                //Nothing
            }
            else if (packet.length < (long)Int16.MaxValue)
            {
                if (packet.length < byte.MaxValue)
                {
                    packetSize += 2;
                }
                else
                {
                    packetSize += 2;
                }
            }
            else
            {
                packetSize += 8;
            }
            packetSize += (int)packet.length;
            byte[] bpacket = new byte[packetSize];


            if (packet.lastPacket)
            {
                bpacket[0] = (byte)128;
            }
            bpacket[0] = Convert.ToByte((int)bpacket[0] + packet.opcode);//Fin + OpCode

            if (packet.length <= 125)
            {
                bpacket[1] = (byte)packet.length;
            }
            else if (packet.length < (int)Int16.MaxValue)
            {
                bpacket[1] = 126;
            }
            else
            {
                bpacket[1] = 127;
            }
            bpacket[1] = Convert.ToByte((int)bpacket[1]);//Masking

            if (packetSize > 125)
            {
                byte[] blen = BitConverter.GetBytes(packet.length);
                if (BitConverter.IsLittleEndian)
                {
                    Array.Reverse(blen);
                }
                if (packetSize <= Int16.MaxValue)
                {
                    blen = new byte[2] {
                        blen[6], blen[7]
                    };
                }
                for (int i = 0; i < blen.Length; i++)
                {
                    bpacket[2 + i] = blen[i];
                }
            }

            //Masking - Don't do it
            //byte[] mask = new byte[4];
            //new Random().NextBytes(mask);
            int offset = (packetSize - (int)packet.length);

            //for (int i = 0; i < 4; i++)
            //{
            //    bpacket[offset + i] = mask[i];
            //}
            //offset += 4;
            for (int i = 0; i < (int)packet.length; i++)
            {
                bpacket[offset + i] = packet.binaryData[i];
            }

            return(bpacket);
        }
Exemplo n.º 12
0
        private WSPacket DecodePacket(byte[] buffer, Socket sock)
        {
            int      byteCounter = 0;
            WSPacket packet      = new WSPacket();
            BitArray byteZero    = BytesToBits(buffer, 0, 1);
            bool     moreToCome  = !byteZero.Get(0);
            int      opcode      = Convert.ToInt16(byteZero.Get(4)) * 8 +
                                   Convert.ToInt16(byteZero.Get(5)) * 4 +
                                   Convert.ToInt16(byteZero.Get(6)) * 2 +
                                   Convert.ToInt16(byteZero.Get(7)) * 1;
            BitArray byteOne  = BytesToBits(buffer, 1, 1);
            bool     isMasked = byteOne.Get(0);

            if (!isMasked)
            {
                sock.Disconnect(false);
            }
            UInt64 length = 0;
            uint   factor = 0;
            uint   bit    = 0;

            for (var i = 1; i < 8; i++)
            {
                factor  = (uint)Math.Pow(2, 7 - i);
                bit     = Convert.ToUInt16(byteOne.Get(i));
                length += bit * factor;
            }

            if (length > 125)
            {
                int bytesToGet = 2;
                if (length == 127)
                {
                    bytesToGet = 8;
                }

                byte[] lenbuffer = new byte[bytesToGet];
                if (sock.Connected)
                {
                    sock.Receive(lenbuffer);
                }
                else
                {
                    packet.opcode = OPCODE_INTERNAL_FAIL;
                    return(packet);
                }
                BitArray lengthArr = BytesToBits(lenbuffer, 0, bytesToGet);
                byteCounter += bytesToGet;
                length       = 0;
                for (var i = 0; i < bytesToGet * 8; i++)
                {
                    factor  = (uint)Math.Pow(2, ((bytesToGet * 8) - 1) - i);
                    bit     = Convert.ToUInt16(lengthArr.Get(i));
                    length += bit * factor;
                }
            }

            byte[] mask = new byte[4];
            if (sock.Connected)
            {
                sock.Receive(mask);
            }
            else
            {
                packet.opcode = OPCODE_INTERNAL_FAIL;
                return(packet);
            }

            byte[] edata = new byte[length];
            byte[] ddata = new byte[edata.Length];

            if (sock.Connected)
            {
                int bcount = 0;
                while (bcount < edata.Length)
                {
                    bcount += sock.Receive(edata, bcount, edata.Length - bcount, 0);
                }
            }
            else
            {
                packet.opcode = OPCODE_INTERNAL_FAIL;
                return(packet);
            }

            for (int i = 0; i < edata.Length; i++)
            {
                ddata[i] = (byte)((int)edata[i] ^ mask[i % 4]);
            }
            packet.binaryData = ddata;
            packet.length     = length;
            packet.opcode     = (uint)opcode;
            packet.lastPacket = !moreToCome;
            if (opcode == 1 && !moreToCome)
            {
                packet.textData = Encoding.UTF8.GetString(ddata);
            }
            return(packet);
        }