mirror of
https://github.com/aria2/aria2.git
synced 2025-04-05 13:37:40 +03:00
9eea959455
On HAVE message reception and PIECE message transmission, update peer bitfield and piece stat only if current peer bitfield indicates it does not have the piece.
134 lines
3.4 KiB
C++
134 lines
3.4 KiB
C++
#include "BtHaveMessage.h"
|
|
|
|
#include <cstring>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "bittorrent_helper.h"
|
|
#include "Peer.h"
|
|
#include "MockPieceStorage.h"
|
|
#include "DlAbortEx.h"
|
|
#include "FileEntry.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class BtHaveMessageTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(BtHaveMessageTest);
|
|
CPPUNIT_TEST(testCreate);
|
|
CPPUNIT_TEST(testCreateMessage);
|
|
CPPUNIT_TEST(testDoReceivedAction);
|
|
CPPUNIT_TEST(testDoReceivedAction_goodByeSeeder);
|
|
CPPUNIT_TEST(testToString);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
|
|
public:
|
|
void setUp() {
|
|
}
|
|
|
|
void testCreate();
|
|
void testCreateMessage();
|
|
void testDoReceivedAction();
|
|
void testDoReceivedAction_goodByeSeeder();
|
|
void testToString();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(BtHaveMessageTest);
|
|
|
|
void BtHaveMessageTest::testCreate() {
|
|
unsigned char msg[9];
|
|
bittorrent::createPeerMessageString(msg, sizeof(msg), 5, 4);
|
|
bittorrent::setIntParam(&msg[5], 12345);
|
|
SharedHandle<BtHaveMessage> pm(BtHaveMessage::create(&msg[4], 5));
|
|
CPPUNIT_ASSERT_EQUAL((uint8_t)4, pm->getId());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)12345, pm->getIndex());
|
|
|
|
// case: payload size is wrong
|
|
try {
|
|
unsigned char msg[10];
|
|
bittorrent::createPeerMessageString(msg, sizeof(msg), 6, 4);
|
|
BtHaveMessage::create(&msg[4], 2);
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(...) {
|
|
}
|
|
// case: id is wrong
|
|
try {
|
|
unsigned char msg[9];
|
|
bittorrent::createPeerMessageString(msg, sizeof(msg), 5, 5);
|
|
BtHaveMessage::create(&msg[4], 1);
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(...) {
|
|
}
|
|
}
|
|
|
|
void BtHaveMessageTest::testCreateMessage() {
|
|
BtHaveMessage msg;
|
|
msg.setIndex(12345);
|
|
unsigned char data[9];
|
|
bittorrent::createPeerMessageString(data, sizeof(data), 5, 4);
|
|
bittorrent::setIntParam(&data[5], 12345);
|
|
unsigned char* rawmsg = msg.createMessage();
|
|
CPPUNIT_ASSERT(memcmp(rawmsg, data, 9) == 0);
|
|
delete [] rawmsg;
|
|
}
|
|
|
|
void BtHaveMessageTest::testDoReceivedAction() {
|
|
SharedHandle<Peer> peer(new Peer("host", 6969));
|
|
peer->allocateSessionResource(16*1024, 256*1024);
|
|
BtHaveMessage msg;
|
|
msg.setIndex(1);
|
|
msg.setPeer(peer);
|
|
SharedHandle<MockPieceStorage> pieceStorage(new MockPieceStorage());
|
|
msg.setPieceStorage(pieceStorage);
|
|
|
|
CPPUNIT_ASSERT(!peer->hasPiece(msg.getIndex()));
|
|
|
|
msg.doReceivedAction();
|
|
|
|
CPPUNIT_ASSERT(peer->hasPiece(msg.getIndex()));
|
|
}
|
|
|
|
void BtHaveMessageTest::testDoReceivedAction_goodByeSeeder()
|
|
{
|
|
SharedHandle<Peer> peer(new Peer("ip", 6000));
|
|
peer->allocateSessionResource(1024, 2*1024);
|
|
BtHaveMessage msg;
|
|
msg.setIndex(0);
|
|
msg.setPeer(peer);
|
|
SharedHandle<MockPieceStorage> pieceStorage(new MockPieceStorage());
|
|
msg.setPieceStorage(pieceStorage);
|
|
|
|
// peer is not seeder and client have not completed download
|
|
msg.doReceivedAction();
|
|
|
|
pieceStorage->setDownloadFinished(true);
|
|
|
|
// client have completed download but, peer is not seeder
|
|
msg.doReceivedAction();
|
|
|
|
msg.setIndex(1);
|
|
pieceStorage->setDownloadFinished(false);
|
|
|
|
// peer is a seeder but client have not completed download
|
|
msg.doReceivedAction();
|
|
|
|
pieceStorage->setDownloadFinished(true);
|
|
peer->updateBitfield(1, 0);
|
|
try {
|
|
msg.doReceivedAction();
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(DlAbortEx& e) {
|
|
// success
|
|
}
|
|
}
|
|
|
|
void BtHaveMessageTest::testToString() {
|
|
BtHaveMessage msg;
|
|
msg.setIndex(1);
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("have index=1"), msg.toString());
|
|
}
|
|
|
|
} // namespace aria2
|