Changeset 2989

Timestamp:

Sep 9, 2007, 12:29:59 AM (16 years ago)

Author:

charles

Message:

update the request block array after the recheck finishes, so that we don't re-ask for blocks we've got

Location:

branches/encryption/libtransmission

Files:

• peer-mgr.c (modified) (2 diffs)
• peer-mgr.h (modified) (1 diff)
• torrent.c (modified) (1 diff)

branches/encryption/libtransmission/peer-mgr.c

@@ -307 +307 @@
         if( t->blocks[i].have || t->blocks[i].dnd )
             continue;
+
         if( !size ) { /* all peers full */
             fprintf( stderr, "all peers full...\n" );
@@ -626 +627 @@
 
 void
+tr_peerMgrUpdateCompletion( tr_peerMgr     * manager,
+                            const uint8_t  * torrentHash )
+{
+    uint32_t i;
+    Torrent * t = getExistingTorrent( manager, torrentHash );
+
+    for( i=0; i<t->blockCount; ++i ) {
+        assert( t->blocks[i].block == i );
+        t->blocks[i].have = tr_cpBlockIsComplete( t->tor->completion, i ) ? 1 : 0;
+    }
+}
+
+void
 tr_peerMgrAddTorrent( tr_peerMgr * manager,
                       tr_torrent * tor )

branches/encryption/libtransmission/peer-mgr.h

@@ -73 +73 @@
                            struct tr_torrent  * tor );
 
+void tr_peerMgrUpdateCompletion( tr_peerMgr     * manager,
+                                 const uint8_t  * torrentHash );
+
+
 void tr_peerMgrRemoveTorrent( tr_peerMgr     * manager,
                                const uint8_t  * torrentHash );

branches/encryption/libtransmission/torrent.c

@@ -268 +268 @@
 {
 fprintf( stderr, "recheck done -- setting torrent %s to %d\n", tor->info.name, tor->runStatusAfterCheck );
+    tr_peerMgrUpdateCompletion( tor->handle->peerMgr, tor->info.hash );
     tor->runStatus = tor->runStatusAfterCheck;
     fastResumeSave( tor );