Exemplo n.º 1
0
        private static void ContinueReceiveCallback(IAsyncResult ar)
        {
            // Store the response state from the receive callback to continue receiving.
            ResponeState state = (ResponeState)ar.AsyncState;

            // Amount of bytes received
            int received;

            try
            {
                // Receive the data and store the amount of bytes
                received = state.Socket.EndReceive(ar);

                // Store the response as a string after conversion from bytes to string
                string response = Encoding.UTF8.GetString(Buffer);

                // Add the string to the response state response string builder
                state.Response.Append(Encoding.UTF8.GetString(Buffer).Substring(0, received));

                // Reduce the amount of bytes remaining by the amount received
                state.RemainingBytes -= received;

                // If there is still data left then continue receiving with the ContinueReceiveCallback
                if (state.RemainingBytes > 0)
                {
                    state.Socket.BeginReceive(Buffer, 0, BufferSize, SocketFlags.None, ContinueReceiveCallback, state);
                }
                else
                {
                    // Else recieve for a new request
                    state.Socket.BeginReceive(Buffer, 0, BufferSize, SocketFlags.None, ReceiveCallback, state.Socket);

                    // Handle the request.
                    RequestHandler.Handle(state.Response.ToString(), state.Socket);
                }

                // Information
                Console.WriteLine($"$ Request from {state.Socket.RemoteEndPoint}: {state.Response}");
            }
            catch (SocketException)
            {
                // If a socket exception is encountered then close the socket, shut it down and remove it.
                CloseErrorSocket(state.Socket);
                return;
            }
            catch (Exception)
            {
                // If any exception is encountered then close the socket, shut it down and remove it.
                CloseErrorSocket(state.Socket);
                return;
            }
        }
Exemplo n.º 2
0
        private static void ReceiveCallback(IAsyncResult ar)
        {
            // End receiving if server is closing
            if (Closing)
            {
                return;
            }

            // Create a ResponseState to pass into the async calls to retrieve all data if it is longer than the buffer
            ResponeState state = new ResponeState();

            state.Socket = (Socket)ar.AsyncState;

            // The amount of data received
            int received;

            try
            {
                // Receive the data and store the amount
                received = state.Socket.EndReceive(ar);

                // Get the current response
                string response = Encoding.UTF8.GetString(Buffer);

                // The length of data is the first thing received, it is parsed and stored in the ResponseState
                string length = response.Split(' ')[0];
                state.RemainingBytes = int.Parse(length);

                // Append the data received so far
                state.Response.Append(Encoding.UTF8.GetString(Buffer).Substring(length.Length + 1, received - length.Length - 1));

                // Reduce the received data from the remaining
                state.RemainingBytes -= received;

                // Information
                Console.WriteLine($"$ Request from {state.Socket.RemoteEndPoint}: {state.Response}");

                // If there is still data left receive more and call the callback to handle more data
                if (state.RemainingBytes > 0)
                {
                    state.Socket.BeginReceive(Buffer, 0, BufferSize, SocketFlags.None, ContinueReceiveCallback, state);
                }
                else
                {
                    // If the data received is EXT, then close the socket and shut it down and close the receiving loop
                    if (state.Response.ToString() == "EXT")
                    {
                        state.Socket.Shutdown(SocketShutdown.Both);
                        state.Socket.Close();
                        ConnectedSockets.Remove(state.Socket);
                        return;
                    }

                    // Continue receiving
                    state.Socket.BeginReceive(Buffer, 0, BufferSize, SocketFlags.None, ReceiveCallback, state.Socket);

                    // Handle the request sent from the client
                    RequestHandler.Handle(state.Response.ToString(), state.Socket);
                }
            }
            catch (SocketException)
            {
                // If a socket exception is encountered then close the socket, shut it down and remove it.
                CloseErrorSocket(state.Socket);
                return;
            }
            catch (Exception)
            {
                // If any exception is encountered then close the socket, shut it down and remove it.
                CloseErrorSocket(state.Socket);
                return;
            }
        }