If the remote handshake couldn't be received, ensure the connection i…

…s correctly disposed. svn path=/trunk/bitsharp/; revision=123693
alanmcgovern · Jan 17, 2009 · 918b42c · 918b42c
1 parent 2b466e5
commit 918b42c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/MonoTorrent/MonoTorrent.Client/Managers/ConnectionManager.cs b/src/MonoTorrent/MonoTorrent.Client/Managers/ConnectionManager.cs
@@ -349,8 +349,10 @@ private void PeerHandshakeReceived(bool succeeded, int count, object state)
             {
                 // If the connection is closed, just return
                 if (!succeeded)
+                {
+                    CleanupSocket(id, "Handshaking failed");
                     return;
-
+                }
                 // Decode the handshake and handle it
                 id.Decryptor.Decrypt(id.recieveBuffer.Array, id.recieveBuffer.Offset, count);
                 msg = new HandshakeMessage();
@@ -380,7 +382,7 @@ private void PeerHandshakeReceived(bool succeeded, int count, object state)
                     msg = new BitfieldMessage(id.TorrentManager.Bitfield);
                 }
 
-                if (id.SupportsLTMessages)
+                if (id.SupportsLTMessages && ClientEngine.SupportsExtended)
                 {
                     MessageBundle bundle = new MessageBundle();
                     bundle.Messages.Add(new ExtendedHandshakeMessage());