Changeset 3759

Timestamp:

Nov 8, 2007, 9:20:08 PM (15 years ago)

Author:

charles

Message:

added `plaintext preferred' encryption option.

Location:

trunk/libtransmission

Files:

• handshake.c (modified) (9 diffs)
• transmission.h (modified) (1 diff)

trunk/libtransmission/handshake.c

@@ -90 +90 @@
     unsigned int havePeerID                   : 1;
     unsigned int haveSentBitTorrentHandshake  : 1;
-    unsigned int allowUnencryptedPeers        : 1;
     tr_peerIo * io;
     tr_crypto * crypto;
@@ -97 +96 @@
     uint8_t mySecret[KEY_LEN];
     uint8_t state;
+    tr_encryption_mode encryptionMode;
     uint16_t pad_c_len;
     uint16_t pad_d_len;
@@ -325 +325 @@
 
 static uint32_t
-getCryptoProvide( const tr_handshake * handshake UNUSED )
-{
-    uint32_t i = 0;
-
-    i |= CRYPTO_PROVIDE_CRYPTO; /* always allow crypto */
-
-#if 0
-    /* by the time we send a crypto_provide, we _know_
-     * the peer supports encryption. */
-    if( handshake->allowUnencryptedPeers )
-        i |= CRYPTO_PROVIDE_PLAINTEXT;
-#endif
-
-   return i;
+getCryptoProvide( const tr_handshake * handshake )
+{
+    uint32_t provide = 0;
+
+    switch( handshake->encryptionMode )
+    {
+        case TR_ENCRYPTION_REQUIRED:
+        case TR_ENCRYPTION_PREFERRED:
+            provide |= CRYPTO_PROVIDE_CRYPTO;
+            break;
+
+        case TR_PLAINTEXT_PREFERRED:
+            provide |= CRYPTO_PROVIDE_CRYPTO | CRYPTO_PROVIDE_PLAINTEXT;
+            break;
+    }
+
+    return provide;
+}
+
+static uint32_t
+getCryptoSelect( const tr_handshake * handshake, uint32_t crypto_provide )
+{
+    uint32_t choices[4];
+    int i, nChoices=0;
+
+    switch( handshake->encryptionMode )
+    {
+        case TR_ENCRYPTION_REQUIRED:
+            choices[nChoices++] = CRYPTO_PROVIDE_CRYPTO;
+            break;
+
+        case TR_ENCRYPTION_PREFERRED:
+            choices[nChoices++] = CRYPTO_PROVIDE_CRYPTO;
+            choices[nChoices++] = CRYPTO_PROVIDE_PLAINTEXT;
+            break;
+
+        case TR_PLAINTEXT_PREFERRED:
+            choices[nChoices++] = CRYPTO_PROVIDE_PLAINTEXT;
+            choices[nChoices++] = CRYPTO_PROVIDE_CRYPTO;
+            break;
+    }
+
+    for( i=0; i<nChoices; ++i )
+        if( crypto_provide & choices[i] )
+            return choices[i];
+
+    return 0;
 }
 
@@ -545 +578 @@
         tr_peerIoSetEncryption( handshake->io, PEER_ENCRYPTION_NONE );
 
-        if( !handshake->allowUnencryptedPeers )
+        if( handshake->encryptionMode == TR_ENCRYPTION_REQUIRED )
         {
             dbgmsg( handshake, "peer is unencrypted, and we're disallowing that" );
@@ -841 +874 @@
     }
 
-dbgmsg( handshake, "sending crypto_select" );
     /* send crypto_select */
-    {
-dbgmsg( handshake, "handshake->crypto_provide is %d", (int)handshake->crypto_provide );
-        if( handshake->crypto_provide & CRYPTO_PROVIDE_CRYPTO )
-            crypto_select = CRYPTO_PROVIDE_CRYPTO;
-        else if( handshake->allowUnencryptedPeers )
-            crypto_select = CRYPTO_PROVIDE_PLAINTEXT;
-        else {
-dbgmsg( handshake, "gronk..." );
-            evbuffer_free( outbuf );
-            tr_handshakeDone( handshake, FALSE );
-            return READ_DONE;
-        }
-
-dbgmsg( handshake, "we select crypto_select as %d...", (int)crypto_select );
+    crypto_select = getCryptoSelect( handshake, handshake->crypto_provide );
+    if( crypto_select ) {
+        dbgmsg( handshake, "selecting crypto mode '%d'", (int)crypto_select );
         tr_peerIoWriteUint32( handshake->io, outbuf, crypto_select );
+    } else {
+        dbgmsg( handshake, "peer didn't offer an encryption mode we like." );
+        evbuffer_free( outbuf );
+        tr_handshakeDone( handshake, FALSE );
+        return READ_DONE;
     }
 
@@ -965 +991 @@
      * try a plaintext handshake */
     if(    ( ( handshake->state == AWAITING_YB ) || ( handshake->state == AWAITING_VC ) )
-        && ( handshake->allowUnencryptedPeers )
+        && ( handshake->encryptionMode != TR_ENCRYPTION_REQUIRED )
         && ( !tr_peerIoReconnect( handshake->io ) ) )
     {
@@ -991 +1017 @@
 tr_handshake*
 tr_handshakeNew( tr_peerIo           * io,
-                 tr_encryption_mode    encryption_mode,
+                 tr_encryption_mode    encryptionMode,
                  handshakeDoneCB       doneCB,
                  void                * doneUserData )
@@ -1000 +1026 @@
     handshake->io = io;
     handshake->crypto = tr_peerIoGetCrypto( io );
-    handshake->allowUnencryptedPeers = encryption_mode!=TR_ENCRYPTION_REQUIRED;
+    handshake->encryptionMode = encryptionMode;
     handshake->doneCB = doneCB;
     handshake->doneUserData = doneUserData;
@@ -1010 +1036 @@
     if( tr_peerIoIsIncoming( handshake->io ) )
         setReadState( handshake, AWAITING_HANDSHAKE );
-    else
+    else if( encryptionMode != TR_PLAINTEXT_PREFERRED )
         sendYa( handshake );
+    else {
+        int msgSize; 
+        uint8_t * msg = buildHandshakeMessage( handshake, &msgSize );
+        handshake->haveSentBitTorrentHandshake = 1;
+        setReadState( handshake, AWAITING_HANDSHAKE );
+        tr_peerIoWrite( handshake->io, msg, msgSize );
+        tr_free( msg );
+    }
 
     return handshake;

trunk/libtransmission/transmission.h

@@ -117 +117 @@
 typedef enum
 {
+    TR_PLAINTEXT_PREFERRED,
     TR_ENCRYPTION_PREFERRED,
     TR_ENCRYPTION_REQUIRED