Revize 1434de93

Re #8619. Connection handling improvements + simple parser of sent messages.

ClientConnectionTest/ClientTest/ClientTest/Program.cs

@@ -226 +226 @@
             //---send the text---
             Console.WriteLine("Sending : " + textToSend);
             nwStream.Write(bytesToSend, 0, bytesToSend.Length);
-			if (textToSend.Equals("bye!"))
-			{
-                break;
-			}
+			
 
             //---read back the text---
             byte[] bytesToRead = new byte[client.ReceiveBufferSize];
             int bytesRead = nwStream.Read(bytesToRead, 0, client.ReceiveBufferSize);
             Console.WriteLine("Received : " + Encoding.ASCII.GetString(bytesToRead, 0, bytesRead));
+
+            if (textToSend.Equals("bye!"))
+            {
+                break;
+            }
         }
         
         client.Close();

Server/ServerApp/Connection/SocketListener.cs

@@ -31 +31 @@
         {
         }
 
-        public static void StartListening()
+        public void StartListening()
         {
             // Data buffer for incoming data.
             byte[] bytes = new Byte[1024];
@@ -78 +78 @@
 
         }
 
-        public static void AcceptCallback(IAsyncResult ar)
+        public void AcceptCallback(IAsyncResult ar)
         {
             // Signal the main thread to continue.
             allDone.Set();
+            Console.WriteLine("Accepted a connection.");
 
             // Get the socket that handles the client request.
             Socket listener = (Socket)ar.AsyncState;
@@ -94 +95 @@
             
         }
 
-        public static void ReadCallback(IAsyncResult ar)
+        public void ReadCallback(IAsyncResult ar)
         {
+
+            
+
+
             String content = String.Empty;
 
             // Retrieve the state object and the handler socket
@@ -103 +108 @@
             StateObject state = (StateObject)ar.AsyncState;
             Socket handler = state.workSocket;
 
-            // Read data from the client socket. 
+            if (!SocketConnected(handler))
+			{
+                Console.WriteLine("Closing connection - unexpected...");
+                CloseConnection(handler);
+                return;
+            }
+
+                // Read data from the client socket. 
             int bytesRead = handler.EndReceive(ar);
 
             if (bytesRead > 0)
@@ -119 +131 @@
                     // All the data has been read from the 
                     // client. Display it on the console.
                     Console.WriteLine("Read {0} bytes from socket. \n Data : {1}", content.Length, content);
+
+                    content = ParseMessageAndBuildAnswer(content);
+                    if (content.Equals("bye!"))
+                    {
+                        Console.WriteLine("Closing connection - expected...");
+                        //handler.Shutdown(SocketShutdown.Both);
+                        //handler.Close();
+                        CloseConnection(handler);
+                        return;
+                    }
+
                     // Echo the data back to the client.
                     Send(handler, content);
+					
 
+                    state.sb.Clear();
                     handler.BeginReceive(state.buffer, 0, StateObject.BufferSize, 0, new AsyncCallback(ReadCallback), state);
                 }
                 else
@@ -132 +157 @@
             }
         }
 
-        private static void Send(Socket handler, String data)
+        private void Send(Socket handler, String data)
         {
             // Convert the string data to byte data using ASCII encoding.
             byte[] byteData = Encoding.ASCII.GetBytes(data);
@@ -141 +166 @@
             handler.BeginSend(byteData, 0, byteData.Length, 0, new AsyncCallback(SendCallback), handler);
         }
 
-        private static void SendCallback(IAsyncResult ar)
+        private void SendCallback(IAsyncResult ar)
         {
             try
             {
@@ -152 +177 @@
                 int bytesSent = handler.EndSend(ar);
                 Console.WriteLine("Sent {0} bytes to client.", bytesSent);
 
+
                 //handler.Shutdown(SocketShutdown.Both);
                 //handler.Close();
 
@@ -161 +187 @@
                 Console.WriteLine(e.ToString());
             }
         }
+
+
+
+
+
+
+
+        private string ParseMessageAndBuildAnswer(string message)
+		{
+            string ans = "";
+
+			if (message.Equals("hello!"))
+			{
+                ans = "hello to you too.";
+			}
+            else if (message.Equals("weather!"))
+			{
+                ans = "weather will be nice";
+            }
+            else if (message.Equals("bye!"))
+			{
+                ans = "bye!";
+			}
+            else
+			{
+                // here bye too, if we want to disconnect clients who don't follow protocol
+                ans = "bye!";//"i didnt understand that";
+
+            }
+
+            return ans;
+		}
+
+
+        private void CloseConnection(Socket s)
+		{
+            s.Shutdown(SocketShutdown.Both);
+            s.Close();
+        }
+
+        private bool SocketConnected(Socket s)
+        {
+            bool part1 = s.Poll(1000, SelectMode.SelectRead);
+            bool part2 = (s.Available == 0);
+            if (part1 && part2)
+                return false;
+            else
+                return true;
+        }
     }
+   
 }

Server/ServerApp/Program.cs

@@ -36 +36 @@
 
 
             // test - connection:
-            AsynchronousSocketListener.StartListening();
+            AsynchronousSocketListener asl = new AsynchronousSocketListener();
+            asl.StartListening();
            
 
             Console.ReadLine();