mirror of https://github.com/aria2/aria2
285 lines
9.0 KiB
C++
285 lines
9.0 KiB
C++
#include "BtRequestMessage.h"
|
|
|
|
#include <cstring>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "bittorrent_helper.h"
|
|
#include "MockBtMessage.h"
|
|
#include "MockPieceStorage.h"
|
|
#include "MockBtMessageFactory.h"
|
|
#include "MockBtMessageDispatcher.h"
|
|
#include "BtAbortOutstandingRequestEvent.h"
|
|
#include "Peer.h"
|
|
#include "FileEntry.h"
|
|
#include "BtHandshakeMessage.h"
|
|
#include "RangeBtMessageValidator.h"
|
|
#include "DlAbortEx.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class BtRequestMessageTest : public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(BtRequestMessageTest);
|
|
CPPUNIT_TEST(testCreate);
|
|
CPPUNIT_TEST(testCreateMessage);
|
|
CPPUNIT_TEST(testDoReceivedAction_hasPieceAndAmNotChoking);
|
|
CPPUNIT_TEST(
|
|
testDoReceivedAction_hasPieceAndAmChokingAndFastExtensionEnabled);
|
|
CPPUNIT_TEST(
|
|
testDoReceivedAction_hasPieceAndAmChokingAndFastExtensionDisabled);
|
|
CPPUNIT_TEST(testDoReceivedAction_doesntHavePieceAndFastExtensionEnabled);
|
|
CPPUNIT_TEST(testDoReceivedAction_doesntHavePieceAndFastExtensionDisabled);
|
|
CPPUNIT_TEST(testHandleAbortRequestEvent);
|
|
CPPUNIT_TEST(testHandleAbortRequestEvent_indexNoMatch);
|
|
CPPUNIT_TEST(testHandleAbortRequestEvent_alreadyInvalidated);
|
|
CPPUNIT_TEST(testToString);
|
|
CPPUNIT_TEST(testValidate);
|
|
CPPUNIT_TEST(testValidate_lengthTooLong);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
|
|
private:
|
|
public:
|
|
void testCreate();
|
|
void testCreateMessage();
|
|
void testDoReceivedAction_hasPieceAndAmNotChoking();
|
|
void testDoReceivedAction_hasPieceAndAmChokingAndFastExtensionEnabled();
|
|
void testDoReceivedAction_hasPieceAndAmChokingAndFastExtensionDisabled();
|
|
void testDoReceivedAction_doesntHavePieceAndFastExtensionEnabled();
|
|
void testDoReceivedAction_doesntHavePieceAndFastExtensionDisabled();
|
|
void testHandleAbortRequestEvent();
|
|
void testHandleAbortRequestEvent_indexNoMatch();
|
|
void testHandleAbortRequestEvent_alreadyInvalidated();
|
|
void testToString();
|
|
void testValidate();
|
|
void testValidate_lengthTooLong();
|
|
|
|
class MockPieceStorage2 : public MockPieceStorage {
|
|
public:
|
|
virtual bool hasPiece(size_t index) CXX11_OVERRIDE { return index == 1; }
|
|
};
|
|
|
|
class MockBtMessageFactory2 : public MockBtMessageFactory {
|
|
public:
|
|
virtual std::unique_ptr<BtPieceMessage>
|
|
createPieceMessage(size_t index, int32_t begin,
|
|
int32_t length) CXX11_OVERRIDE
|
|
{
|
|
return make_unique<BtPieceMessage>(index, begin, length);
|
|
}
|
|
|
|
virtual std::unique_ptr<BtRejectMessage>
|
|
createRejectMessage(size_t index, int32_t begin,
|
|
int32_t length) CXX11_OVERRIDE
|
|
{
|
|
return make_unique<BtRejectMessage>(index, begin, length);
|
|
}
|
|
};
|
|
|
|
std::unique_ptr<MockPieceStorage> pieceStorage_;
|
|
std::shared_ptr<Peer> peer_;
|
|
std::unique_ptr<MockBtMessageDispatcher> dispatcher_;
|
|
std::unique_ptr<MockBtMessageFactory> messageFactory_;
|
|
std::unique_ptr<BtRequestMessage> msg;
|
|
|
|
void setUp()
|
|
{
|
|
pieceStorage_ = make_unique<MockPieceStorage2>();
|
|
|
|
peer_ = std::make_shared<Peer>("host", 6969);
|
|
peer_->allocateSessionResource(16_k, 256_k);
|
|
|
|
dispatcher_ = make_unique<MockBtMessageDispatcher>();
|
|
|
|
messageFactory_ = make_unique<MockBtMessageFactory2>();
|
|
|
|
msg = make_unique<BtRequestMessage>();
|
|
msg->setPeer(peer_);
|
|
msg->setIndex(1);
|
|
msg->setBegin(16);
|
|
msg->setLength(32);
|
|
msg->setBlockIndex(2);
|
|
msg->setBtMessageDispatcher(dispatcher_.get());
|
|
msg->setBtMessageFactory(messageFactory_.get());
|
|
msg->setPieceStorage(pieceStorage_.get());
|
|
}
|
|
};
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(BtRequestMessageTest);
|
|
|
|
void BtRequestMessageTest::testCreate()
|
|
{
|
|
unsigned char msg[17];
|
|
bittorrent::createPeerMessageString(msg, sizeof(msg), 13, 6);
|
|
bittorrent::setIntParam(&msg[5], 12345);
|
|
bittorrent::setIntParam(&msg[9], 256);
|
|
bittorrent::setIntParam(&msg[13], 1_k);
|
|
auto pm = BtRequestMessage::create(&msg[4], 13);
|
|
CPPUNIT_ASSERT_EQUAL((uint8_t)6, pm->getId());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)12345, pm->getIndex());
|
|
CPPUNIT_ASSERT_EQUAL(256, pm->getBegin());
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)1_k, pm->getLength());
|
|
|
|
// case: payload size is wrong
|
|
try {
|
|
unsigned char msg[18];
|
|
bittorrent::createPeerMessageString(msg, sizeof(msg), 14, 6);
|
|
BtRequestMessage::create(&msg[4], 14);
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
}
|
|
catch (...) {
|
|
}
|
|
// case: id is wrong
|
|
try {
|
|
unsigned char msg[17];
|
|
bittorrent::createPeerMessageString(msg, sizeof(msg), 13, 7);
|
|
BtRequestMessage::create(&msg[4], 13);
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
}
|
|
catch (...) {
|
|
}
|
|
}
|
|
|
|
void BtRequestMessageTest::testCreateMessage()
|
|
{
|
|
BtRequestMessage msg;
|
|
msg.setIndex(12345);
|
|
msg.setBegin(256);
|
|
msg.setLength(1_k);
|
|
unsigned char data[17];
|
|
bittorrent::createPeerMessageString(data, sizeof(data), 13, 6);
|
|
bittorrent::setIntParam(&data[5], 12345);
|
|
bittorrent::setIntParam(&data[9], 256);
|
|
bittorrent::setIntParam(&data[13], 1_k);
|
|
auto rawmsg = msg.createMessage();
|
|
CPPUNIT_ASSERT_EQUAL((size_t)17, rawmsg.size());
|
|
CPPUNIT_ASSERT(std::equal(std::begin(rawmsg), std::end(rawmsg), data));
|
|
}
|
|
|
|
void BtRequestMessageTest::testDoReceivedAction_hasPieceAndAmNotChoking()
|
|
{
|
|
peer_->amChoking(false);
|
|
msg->doReceivedAction();
|
|
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, dispatcher_->messageQueue.size());
|
|
CPPUNIT_ASSERT(BtPieceMessage::ID ==
|
|
dispatcher_->messageQueue.front()->getId());
|
|
auto pieceMsg = static_cast<const BtPieceMessage*>(
|
|
dispatcher_->messageQueue.front().get());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, pieceMsg->getIndex());
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)16, pieceMsg->getBegin());
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)32, pieceMsg->getBlockLength());
|
|
}
|
|
|
|
void BtRequestMessageTest::
|
|
testDoReceivedAction_hasPieceAndAmChokingAndFastExtensionEnabled()
|
|
{
|
|
peer_->amChoking(true);
|
|
peer_->setFastExtensionEnabled(true);
|
|
msg->doReceivedAction();
|
|
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, dispatcher_->messageQueue.size());
|
|
CPPUNIT_ASSERT(BtRejectMessage::ID ==
|
|
dispatcher_->messageQueue.front()->getId());
|
|
auto rejectMsg = static_cast<const BtRejectMessage*>(
|
|
dispatcher_->messageQueue.front().get());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, rejectMsg->getIndex());
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)16, rejectMsg->getBegin());
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)32, rejectMsg->getLength());
|
|
}
|
|
|
|
void BtRequestMessageTest::
|
|
testDoReceivedAction_hasPieceAndAmChokingAndFastExtensionDisabled()
|
|
{
|
|
peer_->amChoking(true);
|
|
msg->doReceivedAction();
|
|
|
|
CPPUNIT_ASSERT_EQUAL((size_t)0, dispatcher_->messageQueue.size());
|
|
}
|
|
|
|
void BtRequestMessageTest::
|
|
testDoReceivedAction_doesntHavePieceAndFastExtensionEnabled()
|
|
{
|
|
msg->setIndex(2);
|
|
peer_->amChoking(false);
|
|
peer_->setFastExtensionEnabled(true);
|
|
msg->doReceivedAction();
|
|
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, dispatcher_->messageQueue.size());
|
|
CPPUNIT_ASSERT(BtRejectMessage::ID ==
|
|
dispatcher_->messageQueue.front()->getId());
|
|
auto rejectMsg = static_cast<const BtRejectMessage*>(
|
|
dispatcher_->messageQueue.front().get());
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, rejectMsg->getIndex());
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)16, rejectMsg->getBegin());
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)32, rejectMsg->getLength());
|
|
}
|
|
|
|
void BtRequestMessageTest::
|
|
testDoReceivedAction_doesntHavePieceAndFastExtensionDisabled()
|
|
{
|
|
msg->setIndex(2);
|
|
peer_->amChoking(false);
|
|
msg->doReceivedAction();
|
|
|
|
CPPUNIT_ASSERT_EQUAL((size_t)0, dispatcher_->messageQueue.size());
|
|
}
|
|
|
|
void BtRequestMessageTest::testHandleAbortRequestEvent()
|
|
{
|
|
auto piece = std::make_shared<Piece>(1, 16_k);
|
|
CPPUNIT_ASSERT(!msg->isInvalidate());
|
|
msg->onAbortOutstandingRequestEvent(BtAbortOutstandingRequestEvent(piece));
|
|
CPPUNIT_ASSERT(msg->isInvalidate());
|
|
}
|
|
|
|
void BtRequestMessageTest::testHandleAbortRequestEvent_indexNoMatch()
|
|
{
|
|
auto piece = std::make_shared<Piece>(2, 16_k);
|
|
CPPUNIT_ASSERT(!msg->isInvalidate());
|
|
msg->onAbortOutstandingRequestEvent(BtAbortOutstandingRequestEvent(piece));
|
|
CPPUNIT_ASSERT(!msg->isInvalidate());
|
|
}
|
|
|
|
void BtRequestMessageTest::testHandleAbortRequestEvent_alreadyInvalidated()
|
|
{
|
|
auto piece = std::make_shared<Piece>(1, 16_k);
|
|
msg->setInvalidate(true);
|
|
CPPUNIT_ASSERT(msg->isInvalidate());
|
|
msg->onAbortOutstandingRequestEvent(BtAbortOutstandingRequestEvent(piece));
|
|
CPPUNIT_ASSERT(msg->isInvalidate());
|
|
}
|
|
|
|
void BtRequestMessageTest::testToString()
|
|
{
|
|
CPPUNIT_ASSERT_EQUAL(std::string("request index=1, begin=16, length=32"),
|
|
msg->toString());
|
|
}
|
|
|
|
void BtRequestMessageTest::testValidate()
|
|
{
|
|
BtRequestMessage msg(0, 0, 16_k);
|
|
msg.setBtMessageValidator(
|
|
make_unique<RangeBtMessageValidator>(&msg, 1_k, 256_k));
|
|
msg.validate();
|
|
}
|
|
|
|
void BtRequestMessageTest::testValidate_lengthTooLong()
|
|
{
|
|
BtRequestMessage msg(0, 0, MAX_BLOCK_LENGTH + 1);
|
|
msg.setBtMessageValidator(
|
|
make_unique<RangeBtMessageValidator>(&msg, 1_k, 256_k));
|
|
try {
|
|
msg.validate();
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
}
|
|
catch (DlAbortEx& e) {
|
|
CPPUNIT_ASSERT_EQUAL("Length too long: " +
|
|
util::uitos(MAX_BLOCK_LENGTH + 1) + " > " +
|
|
util::uitos(MAX_BLOCK_LENGTH / 1024) + "KB",
|
|
std::string(e.what()));
|
|
}
|
|
}
|
|
|
|
} // namespace aria2
|