Browse Source

Fixed bug which causes segmentation fault when bad piece is got.

Tatsuhiro Tsujikawa 14 years ago
parent
commit
2eed1c3914
1 changed files with 1 additions and 1 deletions
  1. 1 1
      src/DefaultBtMessageFactory.cc

+ 1 - 1
src/DefaultBtMessageFactory.cc

@@ -178,6 +178,7 @@ DefaultBtMessageFactory::createBtMessage
         temp->setBtMessageValidator(validator);
       }
       temp->setDownloadContext(downloadContext_);
+      temp->setPeerStorage(peerStorage_);
       msg = temp;
       break;
     }
@@ -340,7 +341,6 @@ DefaultBtMessageFactory::createPieceMessage
                                  pieceStorage_->getPieceLength(index)));
   msg->setBtMessageValidator(validator);
   msg->setDownloadContext(downloadContext_);
-  msg->setPeerStorage(peerStorage_);
   setCommonProperty(msg);
   return msg;
 }