public void CreatePacket()
        {
            const string str              = "not a very big deal";
            var          bytes            = new [] { byte.MinValue, byte.MaxValue, byte.MaxValue };
            var          protocol         = new BinaryProtocol(bytes, str);
            var          protocolBytes    = protocol.GetDataToSend();
            var          protocolRestored = new BinaryProtocol(protocolBytes);
            var          size             = BinaryProtocol.GetMessageLength(protocolBytes);

            Assert.True(protocolRestored.IsCheckSumValid());
            Assert.True(protocolBytes.Length == size);
        }
Esempio n. 2
0
        private void ReceiveCallback(IAsyncResult ar)
        {
            var buff = (byte[])ar.AsyncState;

            try
            {
                var bytesCount = Socket.EndReceive(ar);
                Stats.AddBytesReceived(bytesCount);

                if (bytesCount > 0)
                {
                    var bytesReceived = new byte[bytesCount];
                    Buffer.BlockCopy(buff, 0, bytesReceived, 0, bytesCount);

                    _messagesBuffer.Put(bytesReceived);

                    while (_messagesBuffer.Size > 4)
                    {
                        var messageLength = BinaryProtocol.GetMessageLength(_messagesBuffer.Peek(4));
                        if (_messagesBuffer.Size >= messageLength)
                        {
                            var messageBytes = _messagesBuffer.Get(messageLength);
                            var proto        = new BinaryProtocol(messageBytes);
                            var cmdName      = proto.GetCommandBytes();
                            CommandContainer <IClientCommand> command;

                            //if (Commands.TryGetValue(cmdName, out command))
                            //{
                            //    command.CommandRef.Execute(null, );
                            //    command.CommandRef
                            //}

                            RaiseReceive(new ClientGenericEventArgs <byte[]> {
                                DataReceived = bytesReceived, Protocol = proto
                            });
                        }
                        else
                        {
                            break;
                        }
                    }
                }

                Socket.BeginReceive(buff, 0, 1024, SocketFlags.None, ReceiveCallback, buff);
            }
            catch (Exception ex)
            {
                RaiseError(new ClientGenericEventArgs <Exception> {
                    DataReceived = ex
                });
            }
        }
        private void EndReceive(IAsyncResult ar)
        {
            var evArgs = (SocketAsyncEventArgs)ar.AsyncState;

            try
            {
                var args = (SocketAsyncEventArgsExtended)evArgs;
                switch (args.SocketError)
                {
                case SocketError.Success:
                    var bytesReceivedCount = args.AcceptSocket.EndReceive(ar);

                    if (args.AcceptSocket.Connected)
                    {
                        var receivedBytes = new byte[bytesReceivedCount];
                        Buffer.BlockCopy(_receiveBuffer, 0, receivedBytes, 0, bytesReceivedCount);
                        args.CurrentSocketStats.AddBytesReceived(args.BytesTransferred);
                        args.LastResponse = DateTime.UtcNow;

                        args.BytesBuffer.Put(receivedBytes);

                        while (args.BytesBuffer.Size > 4)
                        {
                            var messageLength = BinaryProtocol.GetMessageLength(args.BytesBuffer.Peek(args.BytesBuffer.Size));
                            if (args.BytesBuffer.Size >= messageLength)
                            {
                                var messageBytes     = args.BytesBuffer.Get(messageLength);
                                var messageBytesCopy = new byte[BinaryProtocol.HeaderLength + 1];

                                Buffer.BlockCopy(messageBytes, 0, messageBytesCopy, 0, BinaryProtocol.HeaderLength + 1);

                                var protocol    = new BinaryProtocol(messageBytesCopy);
                                var commandHash = protocol.GetCommandBytes();

                                Tuple <Transport.CommandsBase.IServerCommand, Type> command;
                                var userConnection = args.UserToken as ServerConnection <byte[]>;

                                if (Commands.TryGetValue(commandHash, out command))
                                {
                                    userConnection?.OnDataReceived(protocol, command.Item1, command.Item2);
                                }

                                ReceiveInvoke(new ServerGenericEventArgs <byte[]>
                                {
                                    ServerConnectionBase = args.UserToken as ServerConnection <byte[]>,
                                    DataReceived         = messageBytes,
                                    Protocol             = protocol
                                });
                            }
                            else
                            {
                                break;
                            }
                        }
                    }
                    break;

                case SocketError.ConnectionReset:
                    break;

                case SocketError.TimedOut:
#if DEBUG
                    Trace.WriteLine("SocketError.TimedOut");
#endif
                    break;

                default:
                    var connection = args.UserToken as ServerConnection <byte[]>;
                    DisconnectClient(connection?.ConnectionId);
                    break;
                }

                evArgs.AcceptSocket?.BeginReceive(_receiveBuffer, 0, _receiveBuffer.Length, SocketFlags.None, EndReceive, ListenArgs);
            }
            catch (Exception ex)
            {
                OnErrorInvoke(new ServerGenericEventArgs <Exception>
                {
                    DataReceived = ex
                });
                evArgs.AcceptSocket = null;
            }
        }