remove session when socket is closed

longshine · May 14, 2014 · d97272f · d97272f
1 parent 20db663
commit d97272f
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/Mina.NET/Transport/Socket/AsyncSocketSession.cs b/Mina.NET/Transport/Socket/AsyncSocketSession.cs
@@ -147,10 +147,17 @@ public void ProcessSend(SocketAsyncEventArgs e)
             {
                 EndSend(e.BytesTransferred);
             }
-            else
+            else if (e.SocketError != SocketError.OperationAborted
+                && e.SocketError != SocketError.Interrupted
+                && e.SocketError != SocketError.ConnectionReset)
             {
                 EndSend(new SocketException((Int32)e.SocketError));
             }
+            else
+            {
+                // closed
+                Processor.Remove(this);
+            }
         }
 
         /// <inheritdoc/>
@@ -213,10 +220,16 @@ public void ProcessReceive(SocketAsyncEventArgs e)
                 }
             }
             else if (e.SocketError != SocketError.OperationAborted
-                && e.SocketError != SocketError.Interrupted)
+                && e.SocketError != SocketError.Interrupted
+                && e.SocketError != SocketError.ConnectionReset)
             {
                 EndReceive(new SocketException((Int32)e.SocketError));
             }
+            else
+            {
+                // closed
+                Processor.Remove(this);
+            }
         }
     }
 }