github
/
aria2
mirror of https://github.com/aria2/aria2
1
0
Fork 0
Code Issues Releases Wiki Activity

Fixed the bug that peer is marked as seeder whenever it sends 

extension handshake with metadata size.

Peer must be marked as seeder only when client has not got metadata
yet. This bug causes aria2 shutdowns connection early when it gets
have message from a peer because it wrongly recognizes the peer as
seeder.
pull/1/head
Tatsuhiro Tsujikawa 2011-01-06 22:43:34 +09:00
parent e818f9c8a5
commit d429943d05
3 changed files with 36 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/HandshakeExtensionMessage.cc
View File

@ -111,9 +111,9 @@ void HandshakeExtensionMessage::doReceivedAction()
const std::map<std::string, uint8_t>::value_type& vt = *itr;
peer_->setExtension(vt.first, vt.second);
}
SharedHandle<TorrentAttribute> attrs =
bittorrent::getTorrentAttrs(dctx_);
if(attrs->metadata.empty() && !peer_->getExtensionMessageID("ut_metadata")) {
SharedHandle<TorrentAttribute> attrs = bittorrent::getTorrentAttrs(dctx_);
if(attrs->metadata.empty()) {
if(!peer_->getExtensionMessageID("ut_metadata")) {
// TODO In metadataGetMode, if peer don't support metadata
// transfer, should we drop connection? There is a possibility
// that peer can still tell us peers using PEX.
@ -124,11 +124,6 @@ void HandshakeExtensionMessage::doReceivedAction()
if(metadataSize_ != attrs->metadataSize) {
throw DL_ABORT_EX("Wrong metadata_size. Which one is correct!?");
}
if(!peer_->isSeeder()) {
peer_->reconfigureSessionResource(dctx_->getPieceLength(),
dctx_->getTotalLength());
peer_->setAllBitfield();
}
} else {
attrs->metadataSize = metadataSize_;
dctx_->getFirstFileEntry()->setLength(metadataSize_);
@ -140,15 +135,16 @@ void HandshakeExtensionMessage::doReceivedAction()
// We enter 'end game' mode from the start to get metadata
// quickly.
pieceStorage->enterEndGame();
}
peer_->reconfigureSessionResource(dctx_->getPieceLength(),
dctx_->getTotalLength());
peer_->setAllBitfield();
}
} else if(attrs->metadata.empty()) {
} else {
throw DL_ABORT_EX("Peer didn't provide metadata_size."
" It seems that it doesn't have whole metadata.");
}
}
}
void HandshakeExtensionMessage::setPeer(const SharedHandle<Peer>& peer)
{

1
src/Peer.cc
View File

@ -79,6 +79,7 @@ void Peer::allocateSessionResource(size_t pieceLength, uint64_t totalLength)
delete res_;
res_ = new PeerSessionResource(pieceLength, totalLength);
res_->getPeerStat().downloadStart();
updateSeeder();
}
void Peer::reconfigureSessionResource(size_t pieceLength, uint64_t totalLength)

8
test/HandshakeExtensionMessageTest.cc
View File

@ -117,9 +117,17 @@ void HandshakeExtensionMessageTest::testDoReceivedAction()
CPPUNIT_ASSERT_EQUAL((uint16_t)6889, peer->getPort());
CPPUNIT_ASSERT_EQUAL((uint8_t)1, peer->getExtensionMessageID("ut_pex"));
CPPUNIT_ASSERT_EQUAL((uint8_t)2, peer->getExtensionMessageID("a2_dht"));
CPPUNIT_ASSERT(peer->isSeeder());
CPPUNIT_ASSERT_EQUAL((size_t)1024, attrs->metadataSize);
CPPUNIT_ASSERT_EQUAL((uint64_t)1024, dctx->getTotalLength());
CPPUNIT_ASSERT(dctx->knowsTotalLength());
// See Peer is not marked as seeder if !attrs->metadata.empty()
peer->allocateSessionResource(1024, 1024*1024);
attrs->metadataSize = 1024;
attrs->metadata = std::string('0', attrs->metadataSize);
msg.doReceivedAction();
CPPUNIT_ASSERT(!peer->isSeeder());
}
void HandshakeExtensionMessageTest::testCreate()