Changeset 1594

Timestamp:

Mar 26, 2007, 7:19:33 PM (16 years ago)

Author:

joshe

Message:

Don't save or load cached peers for private torrents.

Location:

trunk/libtransmission

Files:

• fastresume.h (modified) (2 diffs)
• internal.h (modified) (1 diff)
• peer.c (modified) (1 diff)
• peeraz.h (modified) (1 diff)
• peerext.h (modified) (2 diffs)
• torrent.c (modified) (3 diffs)
• transmission.h (modified) (1 diff)

trunk/libtransmission/fastresume.h

@@ -172 +172 @@
     fastResumeWriteData( FR_ID_UPLOADED, &total, 8, 1, file );
 
-    /* Write IPs and ports of connectable peers, if any */
-    if( ( size = tr_peerGetConnectable( tor, &buf ) ) > 0 )
-    {
-        fastResumeWriteData( FR_ID_PEERS, buf, size, 1, file );
-        free( buf );
+    if( !( TR_FLAG_PRIVATE & tor->info.flags ) )
+    {
+        /* Write IPs and ports of connectable peers, if any */
+        if( ( size = tr_peerGetConnectable( tor, &buf ) ) > 0 )
+        {
+            fastResumeWriteData( FR_ID_PEERS, buf, size, 1, file );
+            free( buf );
+        }
     }
 
@@ -375 +378 @@
 
             case FR_ID_PEERS:
-            {
-                uint8_t * buf = malloc( len );
-                if( 1 != fread( buf, len, 1, file ) )
+                if( !( TR_FLAG_PRIVATE & tor->info.flags ) )
                 {
+                    uint8_t * buf = malloc( len );
+                    if( 1 != fread( buf, len, 1, file ) )
+                    {
+                        free( buf );
+                        fclose( file );
+                        return 1;
+                    }
+                    tr_torrentAddCompact( tor, TR_PEER_FROM_CACHE,
+                                          buf, len / 6 );
                     free( buf );
-                    fclose( file );
-                    return 1;
                 }
-                tr_torrentAddCompact( tor, TR_PEER_FROM_CACHE, buf, len / 6 );
-                free( buf );
                 continue;
-            }
 
             default:

trunk/libtransmission/internal.h

@@ -202 +202 @@
     uint64_t          uploadedPrev;
 
+    uint8_t           pexDisabled;
+
     tr_stat_t         stats[2];
     int               statCur;

trunk/libtransmission/peer.c

@@ -224 +224 @@
 void tr_peerSetPrivate( tr_peer_t * peer, int private )
 {
+    if( peer->private == private )
+    {
+        return;
+    }
+
     peer->private = private;
+
+    if( !private )
+    {
+        peer->lastPex = 0;
+    }
+
+    if( EXTENDED_HANDSHAKE == peer->extStatus )
+    {
+        sendExtended( peer->tor, peer, EXTENDED_HANDSHAKE_ID );
+    }
 }
 

trunk/libtransmission/peeraz.h

@@ -152 +152 @@
             continue;
         }
+        /* XXX azureus sucks, we can't enable or disable pex after handshake */
         if( AZ_MSG_AZ_PEER_EXCHANGE == azmsgId( idx ) && peer->private )
         {

trunk/libtransmission/peerext.h

@@ -128 +128 @@
     /* create dict of supported extended messages */
     tr_bencInit( msgsval, TYPE_DICT );
-    if( !peer->private )
-    {
-        /* for public torrents advertise utorrent pex message */
-        if( tr_bencDictAppendNofree( msgsval, "ut_pex", &pexval, NULL ) )
-        {
-            tr_bencFree( &val );
-            return NULL;
-        }
-        tr_bencInitInt( pexval, EXTENDED_PEX_ID );
-    }
+    if( tr_bencDictAppendNofree( msgsval, "ut_pex", &pexval, NULL ) )
+    {
+        tr_bencFree( &val );
+        return NULL;
+    }
+    /* for public torrents advertise utorrent pex message */
+    tr_bencInitInt( pexval, ( peer->private ? 0 : EXTENDED_PEX_ID ) );
 
     /* our listening port */
@@ -233 +230 @@
         {
             peer->pexStatus = 0;
-            if( !peer->private && 0x0 < sub->val.i && 0xff >= sub->val.i )
+            if( 0x0 < sub->val.i && 0xff >= sub->val.i )
             {
                 peer->pexStatus = sub->val.i;

trunk/libtransmission/torrent.c

@@ -138 +138 @@
     }
 
+    tor->pexDisabled = 0;
+
     /* Block size: usually 16 ko, or less if we have to */
     tor->blockSize  = MIN( inf->pieceSize, 1 << 14 );
@@ -268 +270 @@
     }
     tor->peerCount = 0;
+    tr_lockUnlock( &tor->lock );
+}
+
+void tr_torrentDisablePex( tr_torrent_t * tor, int disable )
+{
+    int ii;
+
+    if( TR_FLAG_PRIVATE & tor->info.flags )
+    {
+        return;
+    }
+
+    tr_lockLock( &tor->lock );
+
+    if( tor->pexDisabled == disable )
+    {
+        tr_lockUnlock( &tor->lock );
+        return;
+    }
+
+    tor->pexDisabled = disable;
+    for( ii = 0; ii < tor->peerCount; ii++ )
+    {
+        tr_peerSetPrivate( tor->peers[ii], disable );
+    }
+
     tr_lockUnlock( &tor->lock );
 }
@@ -636 +664 @@
     }
 
-    tr_peerSetPrivate( peer, tor->info.flags & TR_FLAG_PRIVATE );
+    tr_peerSetPrivate( peer, tor->info.flags & TR_FLAG_PRIVATE ||
+                       tor->pexDisabled );
     tr_peerSetTorrent( peer, tor );
     tor->peers[tor->peerCount++] = peer;

trunk/libtransmission/transmission.h

@@ -224 +224 @@
                                     int flags, int * error );
 
+/***********************************************************************
+ * tr_torrentDisablePex
+ ***********************************************************************
+ * Disable or enable peer exchange for this torrent. Peer exchange is
+ * enabled by default, except for private torrents where pex is
+ * disabled and cannot be enabled.
+ **********************************************************************/
+void tr_torrentDisablePex( tr_torrent_t *, int disable );
+
+/***********************************************************************
+ * tr_torrentScrape
+ ***********************************************************************
+ * Return torrent metainfo.
+ **********************************************************************/
 typedef struct tr_info_s tr_info_t;
 tr_info_t * tr_torrentInfo( tr_torrent_t * );