Changeset 3405

Timestamp:

Oct 13, 2007, 11:15:43 PM (15 years ago)

Author:

charles

Message:

Try sharing a DH so that we don't spend so many cpu cycles making new ones when we're working through the peers trying to find good ones.

Location:

trunk/libtransmission

Files:

• crypto.c (modified) (6 diffs)
• peer-msgs.c (modified) (1 diff)

trunk/libtransmission/crypto.c

@@ -77 +77 @@
 struct tr_crypto
 {
-    DH * dh;
     RC4_KEY dec_key;
     RC4_KEY enc_key;
@@ -86 +85 @@
     uint8_t myPublicKey[KEY_LEN];
     uint8_t mySecret[KEY_LEN];
-
 };
 
-static void
-ensureKeyExists( tr_crypto * crypto )
-{
-    if( crypto->dh == NULL )
+/**
+***
+**/
+
+static DH*
+getSharedDH( void )
+{
+    static DH * dh = NULL;
+
+    if( dh == NULL )
     {
-        int len, offset;
-
-        crypto->dh = DH_new( );
-        crypto->dh->p = BN_bin2bn( dh_P, sizeof(dh_P), NULL );
-        crypto->dh->g = BN_bin2bn( dh_G, sizeof(dh_G), NULL );
-        DH_generate_key( crypto->dh );
-
-        /* DH can generate key sizes that are smaller than the size of
-           P with exponentially decreasing probability, in which case
-           the msb's of myPublicKey need to be zeroed appropriately. */
-        len = DH_size( crypto->dh );
-        offset = KEY_LEN - len;
-        assert( len <= KEY_LEN );
-        memset( crypto->myPublicKey, 0, offset );
-        BN_bn2bin( crypto->dh->pub_key, crypto->myPublicKey + offset );
+        dh = DH_new( );
+        dh->p = BN_bin2bn( dh_P, sizeof(dh_P), NULL );
+        dh->g = BN_bin2bn( dh_G, sizeof(dh_G), NULL );
+        DH_generate_key( dh );
     }
-}
-
-/**
-***
-**/
+
+    return dh;
+}
 
 tr_crypto * 
@@ -120 +111 @@
               int             isIncoming )
 {
+    int len, offset;
     tr_crypto * crypto;
+    DH * dh = getSharedDH( );
 
     crypto = tr_new0( tr_crypto, 1 );
     crypto->isIncoming = isIncoming ? 1 : 0;
-    crypto->dh = NULL;
     tr_cryptoSetTorrentHash( crypto, torrentHash );
+
+    /* DH can generate key sizes that are smaller than the size of
+       P with exponentially decreasing probability, in which case
+       the msb's of myPublicKey need to be zeroed appropriately. */
+    len = DH_size( dh );
+    offset = KEY_LEN - len;
+    assert( len <= KEY_LEN );
+    memset( crypto->myPublicKey, 0, offset );
+    BN_bn2bin( dh->pub_key, crypto->myPublicKey + offset );
+
     return crypto;
 }
@@ -132 +134 @@
 tr_cryptoFree( tr_crypto * crypto )
 {
-    assert( crypto != NULL );
-    if( crypto->dh != NULL )
-        DH_free( crypto->dh );
     tr_free( crypto );
 }
@@ -149 +148 @@
     uint8_t secret[KEY_LEN];
     BIGNUM * bn = BN_bin2bn( peerPublicKey, KEY_LEN, NULL );
-
-    ensureKeyExists( crypto );
-    assert( DH_size( crypto->dh ) == KEY_LEN );
-
-    len = DH_compute_key( secret, bn, crypto->dh );
+    DH * dh = getSharedDH( );
+    assert( DH_size(dh) == KEY_LEN );
+
+    len = DH_compute_key( secret, bn, dh );
     assert( len <= KEY_LEN );
     offset = KEY_LEN - len;
@@ -168 +166 @@
 tr_cryptoGetMyPublicKey( const tr_crypto * crypto, int * setme_len )
 {
-    ensureKeyExists( (tr_crypto*) crypto );
     *setme_len = KEY_LEN;
     return crypto->myPublicKey;

trunk/libtransmission/peer-msgs.c

@@ -70 +70 @@
     KEEPALIVE_INTERVAL_SECS = 90,          /* idle seconds before we send a keepalive */
     PEX_INTERVAL            = (60 * 1000), /* msec between calls to sendPex() */
-    PEER_PULSE_INTERVAL     = (66),        /* msec between calls to pulse() */
+    PEER_PULSE_INTERVAL     = (20),        /* msec between calls to pulse() */
     RATE_PULSE_INTERVAL     = (333),       /* msec between calls to ratePulse() */
 };