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

Rewritten LpdMessageReceiver::receiveMessage()

pull/28/head
Tatsuhiro Tsujikawa 2012-09-26 23:45:58 +09:00
parent 3a917ff695
commit b9f972665b
3 changed files with 29 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
src/LpdMessageReceiver.cc
View File

@ -80,21 +80,37 @@ bool LpdMessageReceiver::init(const std::string& localAddr)
SharedHandle<LpdMessage> LpdMessageReceiver::receiveMessage()
{
SharedHandle<LpdMessage> msg;
try {
while(1) {
unsigned char buf[200];
std::pair<std::string, uint16_t> peerAddr;
ssize_t length = socket_->readDataFrom(buf, sizeof(buf), peerAddr);
if(length == 0) {
ssize_t length;
try {
length = socket_->readDataFrom(buf, sizeof(buf), peerAddr);
if(length == 0) {
return msg;
}
} catch(RecoverableException& e) {
A2_LOG_INFO_EX("Failed to receive LPD message.", e);
return msg;
}
HttpHeaderProcessor proc(HttpHeaderProcessor::SERVER_PARSER);
if(!proc.parse(buf, length)) {
msg.reset(new LpdMessage());
return msg;
try {
if(!proc.parse(buf, length)) {
// UDP packet must contain whole HTTP header block.
continue;
}
} catch(RecoverableException& e) {
A2_LOG_INFO_EX("Failed to parse LPD message.", e);
continue;
}
const SharedHandle<HttpHeader>& header = proc.getResult();
const std::string& infoHashString = header->find(HttpHeader::INFOHASH);
uint16_t port = header->findAsInt(HttpHeader::PORT);
uint32_t port = 0;
if(!util::parseUIntNoThrow(port, header->find(HttpHeader::PORT)) ||
port > UINT16_MAX || port == 0) {
A2_LOG_INFO(fmt("Bad LPD port=%u", port));
continue;
}
A2_LOG_INFO(fmt("LPD message received infohash=%s, port=%u from %s",
infoHashString.c_str(),
port,
@ -102,11 +118,10 @@ SharedHandle<LpdMessage> LpdMessageReceiver::receiveMessage()
std::string infoHash;
if(infoHashString.size() != 40 ||
(infoHash = util::fromHex(infoHashString.begin(),
infoHashString.end())).empty() ||
port == 0) {
A2_LOG_INFO(fmt("LPD bad request. infohash=%s", infoHashString.c_str()));
msg.reset(new LpdMessage());
return msg;
infoHashString.end())).empty()) {
A2_LOG_INFO(fmt("LPD bad request. infohash=%s",
infoHashString.c_str()));
continue;
}
SharedHandle<Peer> peer(new Peer(peerAddr.first, port, false));
if(util::inPrivateAddress(peerAddr.first)) {
@ -114,10 +129,6 @@ SharedHandle<LpdMessage> LpdMessageReceiver::receiveMessage()
}
msg.reset(new LpdMessage(peer, infoHash));
return msg;
} catch(RecoverableException& e) {
A2_LOG_INFO_EX("Failed to receive LPD message.", e);
msg.reset(new LpdMessage());
return msg;
}
}

4
src/LpdReceiveMessageCommand.cc
View File

@ -77,10 +77,6 @@ bool LpdReceiveMessageCommand::execute()
if(!m) {
break;
}
if(!m->peer) {
// bad message
continue;
}
SharedHandle<BtRegistry> reg = e_->getBtRegistry();
SharedHandle<DownloadContext> dctx = reg->getDownloadContext(m->infoHash);
if(!dctx) {

8
test/LpdMessageReceiverTest.cc
View File

@ -67,9 +67,7 @@ void LpdMessageReceiverTest::testReceiveMessage()
rcv.getSocket()->isReadable(5);
msg = rcv.receiveMessage();
CPPUNIT_ASSERT(msg);
CPPUNIT_ASSERT(!msg->peer);
CPPUNIT_ASSERT(msg->infoHash.empty());
CPPUNIT_ASSERT(!msg);
// Bad port
request =
@ -81,9 +79,7 @@ void LpdMessageReceiverTest::testReceiveMessage()
rcv.getSocket()->isReadable(5);
msg = rcv.receiveMessage();
CPPUNIT_ASSERT(msg);
CPPUNIT_ASSERT(!msg->peer);
CPPUNIT_ASSERT(msg->infoHash.empty());
CPPUNIT_ASSERT(!msg);
// No data available
msg = rcv.receiveMessage();