Changeset 5045

Timestamp:

Feb 15, 2008, 6:25:42 PM (14 years ago)

Author:

charles

Message:

make the code a little easier to read. make tr_ioFileTest()'s return value consistent with other inout funcs.

Location:

trunk/libtransmission

Files:

• inout.c (modified) (3 diffs)
• inout.h (modified) (1 diff)
• peer-msgs.c (modified) (1 diff)
• verify.c (modified) (1 diff)

trunk/libtransmission/inout.c

@@ -222 +222 @@
 ****/
 
-static int
+static tr_errno
 tr_ioRecalculateHash( const tr_torrent  * tor,
                       int                 pieceIndex,
@@ -245 +245 @@
     {
         const int bytesThisPass = MIN( bytesLeft, (int)sizeof(buf) );
-        int err = tr_ioRead( tor, pieceIndex, offset, bytesThisPass, buf );
+        tr_errno err = tr_ioRead( tor, pieceIndex, offset, bytesThisPass, buf );
         if( err )
             return err;
@@ -257 +257 @@
 }
 
-int
+tr_errno
 tr_ioTestPiece( const tr_torrent * tor, int pieceIndex )
 {
+    int err;
     uint8_t hash[SHA_DIGEST_LENGTH];
-    const int ret = tr_ioRecalculateHash( tor, pieceIndex, hash )
-        || memcmp( hash, tor->info.pieces[pieceIndex].hash, SHA_DIGEST_LENGTH );
+
+    err  = tr_ioRecalculateHash( tor, pieceIndex, hash );
+
+    if( !err && memcmp( hash, tor->info.pieces[pieceIndex].hash, SHA_DIGEST_LENGTH ) )
+        err = TR_ERROR;
+
     tr_dbg ("torrent [%s] piece %d hash check: %s",
-            tor->info.name, pieceIndex, ( ret ? "FAILED" : "OK" ));
-    return ret;
-}
-
-int
-tr_ioHash( tr_torrent * tor, int pieceIndex )
-{
-    int ret;
-    const int success = !tr_ioTestPiece( tor, pieceIndex );
-
-    if( success )
-    {
-        tr_dbg( "Piece %d hash OK", pieceIndex );
-        tr_cpPieceAdd( tor->completion, pieceIndex );
-        ret = TR_OK;
-    }
-    else
-    {
-        tr_err( "Piece %d hash FAILED", pieceIndex );
-        tr_cpPieceRem( tor->completion, pieceIndex );
-        ret = TR_ERROR;
-    }
-
-    tr_peerMgrSetBlame( tor->handle->peerMgr, tor->info.hash,
-                        pieceIndex, success );
-
-    return ret;
-}
+            tor->info.name, pieceIndex, ( err ? "FAILED" : "OK" ));
+
+    return err;
+}

trunk/libtransmission/inout.h

@@ -51 +51 @@
 
 /**
- * returns true if the piece matches its metainfo's SHA1 checksum,
- * false otherwise.
+ * returns 0 if the piece matches its metainfo's SHA1 checksum,
+ * or TR_ERROR_IO_* if there was a problem reading the piece,
+ * or TR_ERROR if the checksum didn't match.
  */
-int tr_ioTestPiece( const tr_torrent*, int piece );
-
-
-/**
- * tests the specified piece and uses the results to
- * update the torrent's "completion" and "blame" fields.
- */
-int tr_ioHash ( tr_torrent*, int piece );
+tr_errno tr_ioTestPiece( const tr_torrent*, int piece );
 
 

trunk/libtransmission/peer-msgs.c

@@ -1456 +1456 @@
     if( tr_cpPieceIsComplete( tor->completion, req->index ) )
     {
-        if( tr_ioHash( tor, req->index ) )
-        {
+        const tr_errno err = tr_ioTestPiece( tor, req->index );
+
+        tr_torrentSetHasPiece( tor, req->index, !err );
+        tr_torrentSetPieceChecked( tor, req->index, TRUE );
+        tr_peerMgrSetBlame( tor->handle->peerMgr, tor->info.hash, req->index, !err );
+
+        if( !err )
+            fireClientHave( msgs, req->index );
+        else
             gotBadPiece( msgs, req->index );
-            return 0;
-        }
-
-        tr_torrentSetPieceChecked( tor, req->index, TRUE );
-        fireClientHave( msgs, req->index );
     }
 

trunk/libtransmission/verify.c

@@ -91 +91 @@
         else if( !tr_torrentIsPieceChecked( tor, i ) )
         {
-            const int check = tr_ioTestPiece( tor, i );
-            tr_torrentSetHasPiece( tor, i, !check );
+            const tr_errno err = tr_ioTestPiece( tor, i );
+
+            if( !err ) /* yay */
+            {
+                tr_torrentSetHasPiece( tor, i, TRUE );
+            }
+            else
+            {
+                /* if we were wrong about it being complete,
+                 * reset and start again.  if we were right about
+                 * it being incomplete, do nothing -- we don't
+                 * want to lose blocks in those incomplete pieces */
+
+                if( tr_cpPieceIsComplete( tor->completion, i ) )
+                    tr_torrentSetHasPiece( tor, i, FALSE );
+            }
+               
             tr_torrentSetPieceChecked( tor, i, TRUE );
         }