Преглед на файлове

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

Tatsuhiro Tsujikawa преди 14 години
родител
ревизия
2eed1c3914
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  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;
 }