From eee7b16b2cf5808966b11d1ead12f572b24a8b22 Mon Sep 17 00:00:00 2001
From: antiduh <antiduh@e114e3f1-24f7-e311-a78f-001bfcf5e160>
Date: Tue, 8 Jul 2014 17:51:24 +0000
Subject: [PATCH] Better client handling when the server connection goes away.

---
 TestProtocol/CustomConnection.cs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/TestProtocol/CustomConnection.cs b/TestProtocol/CustomConnection.cs
index 8a68881..274b67f 100644
--- a/TestProtocol/CustomConnection.cs
+++ b/TestProtocol/CustomConnection.cs
@@ -143,7 +143,8 @@ namespace TestProtocol
                     if( e.SocketErrorCode == SocketError.ConnectionAborted ||
                         e.SocketErrorCode == SocketError.Interrupted ||
                         e.SocketErrorCode == SocketError.OperationAborted ||
-                        e.SocketErrorCode == SocketError.Shutdown )
+                        e.SocketErrorCode == SocketError.Shutdown || 
+                        e.SocketErrorCode == SocketError.ConnectionReset )
                     {
                         // Shutting down.
                         break;