From d111b204918a1795cc42499f4d88ad17af99c3a8 Mon Sep 17 00:00:00 2001
From: lidgren <lidgren@users.noreply.github.com>
Date: Fri, 11 Mar 2011 08:58:58 +0000
Subject: [PATCH] Fixed issue with disconnected by server

---
 Lidgren.Network/NetConnection.Handshake.cs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/Lidgren.Network/NetConnection.Handshake.cs b/Lidgren.Network/NetConnection.Handshake.cs
index 724613e..9f7f632 100644
--- a/Lidgren.Network/NetConnection.Handshake.cs
+++ b/Lidgren.Network/NetConnection.Handshake.cs
@@ -108,7 +108,8 @@ namespace Lidgren.Network
 			SetStatus(NetConnectionStatus.Disconnected, reason);
 
 			// in case we're still in handshake
-			m_peer.m_handshakes.Remove(m_remoteEndpoint);
+			lock (m_peer.m_handshakes)
+				m_peer.m_handshakes.Remove(m_remoteEndpoint);
 
 			m_disconnectRequested = false;
 			m_connectRequested = false;
@@ -371,7 +372,7 @@ namespace Lidgren.Network
 					catch
 					{
 					}
-					SetStatus(NetConnectionStatus.Disconnected, reason);
+					ExecuteDisconnect(reason, false);
 					break;
 				default:
 					m_peer.LogDebug("Unhandled type during handshake: " + tp + " length: " + payloadLength);