Changeset 6985

Timestamp:

Oct 29, 2008, 7:30:17 PM (12 years ago)

Author:

charles

Message:

(libT) possible fix for #1389: bitcomet handshake with TransmissionBT failed

File:

• trunk/libtransmission/handshake.c (modified) (1 diff)

trunk/libtransmission/handshake.c

@@ -1019 +1019 @@
     struct evbuffer * inbuf = EVBUFFER_INPUT ( evin );
     ReadState         ret;
+    int               readyForMore = TRUE;
 
     dbgmsg( handshake, "handling canRead; state is [%s]",
            getStateName( handshake->state ) );
 
-    switch( handshake->state )
-    {
-        case AWAITING_HANDSHAKE:
-            ret = readHandshake    ( handshake, inbuf ); break;
-
-        case AWAITING_PEER_ID:
-            ret = readPeerId       ( handshake, inbuf ); break;
-
-        case AWAITING_YA:
-            ret = readYa           ( handshake, inbuf ); break;
-
-        case AWAITING_PAD_A:
-            ret = readPadA         ( handshake, inbuf ); break;
-
-        case AWAITING_CRYPTO_PROVIDE:
-            ret = readCryptoProvide( handshake, inbuf ); break;
-
-        case AWAITING_PAD_C:
-            ret = readPadC         ( handshake, inbuf ); break;
-
-        case AWAITING_IA:
-            ret = readIA           ( handshake, inbuf ); break;
-
-        case AWAITING_YB:
-            ret = readYb           ( handshake, inbuf ); break;
-
-        case AWAITING_VC:
-            ret = readVC           ( handshake, inbuf ); break;
-
-        case AWAITING_CRYPTO_SELECT:
-            ret = readCryptoSelect ( handshake, inbuf ); break;
-
-        case AWAITING_PAD_D:
-            ret = readPadD         ( handshake, inbuf ); break;
-
-        default:
-            assert( 0 );
+    while( readyForMore )
+    {
+        switch( handshake->state )
+        {
+            case AWAITING_HANDSHAKE:
+                ret = readHandshake    ( handshake, inbuf ); break;
+
+            case AWAITING_PEER_ID:
+                ret = readPeerId       ( handshake, inbuf ); break;
+
+            case AWAITING_YA:
+                ret = readYa           ( handshake, inbuf ); break;
+
+            case AWAITING_PAD_A:
+                ret = readPadA         ( handshake, inbuf ); break;
+
+            case AWAITING_CRYPTO_PROVIDE:
+                ret = readCryptoProvide( handshake, inbuf ); break;
+
+            case AWAITING_PAD_C:
+                ret = readPadC         ( handshake, inbuf ); break;
+
+            case AWAITING_IA:
+                ret = readIA           ( handshake, inbuf ); break;
+
+            case AWAITING_YB:
+                ret = readYb           ( handshake, inbuf ); break;
+
+            case AWAITING_VC:
+                ret = readVC           ( handshake, inbuf ); break;
+
+            case AWAITING_CRYPTO_SELECT:
+                ret = readCryptoSelect ( handshake, inbuf ); break;
+
+            case AWAITING_PAD_D:
+                ret = readPadD         ( handshake, inbuf ); break;
+
+            default:
+                assert( 0 );
+        }
+
+        switch( handshake->state )
+        {
+            case AWAITING_PAD_C:
+                readyForMore = EVBUFFER_LENGTH( inbuf ) >= handshake->pad_c_len;
+                break;
+
+            case AWAITING_PAD_D:
+                readyForMore = EVBUFFER_LENGTH( inbuf ) >= handshake->pad_d_len;
+                break;
+
+            case AWAITING_IA:
+                readyForMore = EVBUFFER_LENGTH( inbuf ) >= handshake->ia_len;
+                break;
+
+            default:
+                readyForMore = ret == READ_NOW;
+                break;
+        }
     }