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

Fix bug that causes bogus EOF connection failure in BitTorrent downloads

pull/704/head
Tatsuhiro Tsujikawa 2016-07-09 21:21:32 +09:00
parent 41df1607f6
commit ae11b7a85d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/PeerConnection.cc
View File

@ -124,7 +124,7 @@ bool PeerConnection::receiveMessage(unsigned char* data, size_t& dataLength)
} }
break; break;
case (BT_MSG_READ_PAYLOAD): case (BT_MSG_READ_PAYLOAD):
// We chosen the bufferCapacity_ so that whole message, // We chose the bufferCapacity_ so that whole message,
// including 4 bytes length and payload, in it. So here we // including 4 bytes length and payload, in it. So here we
// just make sure that it happens. // just make sure that it happens.
if (resbufLength_ - msgOffset_ >= 4 + currentPayloadLength_) { if (resbufLength_ - msgOffset_ >= 4 + currentPayloadLength_) {
@ -150,7 +150,7 @@ bool PeerConnection::receiveMessage(unsigned char* data, size_t& dataLength)
else { else {
assert(resbufOffset_ == resbufLength_); assert(resbufOffset_ == resbufLength_);
if (resbufLength_ != 0) { if (resbufLength_ != 0) {
if (msgOffset_ == 0 && resbufLength_ == currentPayloadLength_ + 4) { if (resbufLength_ - msgOffset_ == currentPayloadLength_ + 4) {
// All bytes in buffer have been processed, so clear it // All bytes in buffer have been processed, so clear it
// away. // away.
resbufLength_ = 0; resbufLength_ = 0;