mirror of
https://github.com/aria2/aria2.git
synced 2024-12-14 09:28:55 +00:00
3505201f33
Rewritten SharedHandle. Now copy constructor taking raw pointer has keyword explicit and SharedHandle's default constructor initializes its internal obj to null, old implementation initializes it using obj's default constructor. To assign null, write SharedHandle<T> x(...); x.reset(); TODO: test/SharedHandleTest.cc needs more tests. * src/SharedHandle.h
119 lines
4.0 KiB
C++
119 lines
4.0 KiB
C++
#include "BtHandshakeMessage.h"
|
|
#include "PeerMessageUtil.h"
|
|
#include "Util.h"
|
|
#include "BtConstants.h"
|
|
#include <cstring>
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class BtHandshakeMessageTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(BtHandshakeMessageTest);
|
|
CPPUNIT_TEST(testCreate);
|
|
CPPUNIT_TEST(testGetMessage);
|
|
CPPUNIT_TEST(testToString);
|
|
CPPUNIT_TEST(testSetDHTEnabled);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
|
|
public:
|
|
void setUp() {
|
|
}
|
|
|
|
void testCreate();
|
|
void testGetMessage();
|
|
void testToString();
|
|
void testSetDHTEnabled();
|
|
|
|
static std::string BTPSTR;
|
|
};
|
|
|
|
|
|
std::string BtHandshakeMessageTest::BTPSTR = "BitTorrent protocol";
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(BtHandshakeMessageTest);
|
|
|
|
void createHandshakeMessageData(unsigned char* msg) {
|
|
msg[0] = 19;
|
|
memcpy(&msg[1], BtHandshakeMessageTest::BTPSTR.c_str(),
|
|
BtHandshakeMessageTest::BTPSTR.size());
|
|
unsigned char reserved[] = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x10, 0x00, 0x04 };
|
|
memcpy(&msg[20], reserved, sizeof(reserved));
|
|
unsigned char infoHash[] = { 0xff, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xff, 0xff, 0xff };
|
|
memcpy(&msg[28], infoHash, sizeof(infoHash));
|
|
unsigned char peerId[] = { 0xf0, 0xf0, 0xf0, 0xf0, 0xf0,
|
|
0xf0, 0xf0, 0xf0, 0xf0, 0xf0,
|
|
0xf0, 0xf0, 0xf0, 0xf0, 0xf0,
|
|
0xf0, 0xf0, 0xf0, 0xf0, 0xf0 };
|
|
memcpy(&msg[48], peerId, sizeof(peerId));
|
|
}
|
|
|
|
void BtHandshakeMessageTest::testCreate() {
|
|
unsigned char msg[68];
|
|
createHandshakeMessageData(msg);
|
|
SharedHandle<BtHandshakeMessage> message = BtHandshakeMessage::create(&msg[0], sizeof(msg));
|
|
CPPUNIT_ASSERT_EQUAL((uint8_t)INT8_MAX, message->getId());
|
|
CPPUNIT_ASSERT_EQUAL((uint8_t)19, message->getPstrlen());
|
|
CPPUNIT_ASSERT_EQUAL(Util::toHex((const unsigned char*)BTPSTR.c_str(), BTPSTR.size()),
|
|
Util::toHex(message->getPstr(), BtHandshakeMessage::PSTR_LENGTH));
|
|
CPPUNIT_ASSERT_EQUAL(std::string("0000000000100004"),
|
|
Util::toHex(message->getReserved(), BtHandshakeMessage::RESERVED_LENGTH));
|
|
CPPUNIT_ASSERT_EQUAL(std::string("ffffffffffffffffffffffffffffffffffffffff"),
|
|
Util::toHex(message->getInfoHash(), INFO_HASH_LENGTH));
|
|
CPPUNIT_ASSERT_EQUAL(std::string("f0f0f0f0f0f0f0f0f0f0f0f0f0f0f0f0f0f0f0f0"),
|
|
Util::toHex(message->getPeerId(), PEER_ID_LENGTH));
|
|
}
|
|
|
|
void BtHandshakeMessageTest::testGetMessage() {
|
|
unsigned char infoHash[] = { 0xff, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xff, 0xff, 0xff };
|
|
unsigned char peerId[] = { 0xf0, 0xf0, 0xf0, 0xf0, 0xf0,
|
|
0xf0, 0xf0, 0xf0, 0xf0, 0xf0,
|
|
0xf0, 0xf0, 0xf0, 0xf0, 0xf0,
|
|
0xf0, 0xf0, 0xf0, 0xf0, 0xf0 };
|
|
|
|
SharedHandle<BtHandshakeMessage> msg(new BtHandshakeMessage());
|
|
msg->setInfoHash(infoHash);
|
|
msg->setPeerId(peerId);
|
|
|
|
unsigned char data[68];
|
|
createHandshakeMessageData(data);
|
|
CPPUNIT_ASSERT_EQUAL(Util::toHex((const unsigned char*)data, 68),
|
|
Util::toHex((const unsigned char*)msg->getMessage(), 68));
|
|
}
|
|
|
|
void BtHandshakeMessageTest::testToString() {
|
|
unsigned char infoHash[] = { 0xff, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xff, 0xff, 0xff,
|
|
0xff, 0xff, 0xff, 0xff, 0xff };
|
|
unsigned char peerId[] = { 0xf0, 0xf0, 0xf0, 0xf0, 0xf0,
|
|
0xf0, 0xf0, 0xf0, 0xf0, 0xf0,
|
|
0xf0, 0xf0, 0xf0, 0xf0, 0xf0,
|
|
0xf0, 0xf0, 0xf0, 0xf0, 0xf0 };
|
|
|
|
BtHandshakeMessage msg;
|
|
msg.setInfoHash(infoHash);
|
|
msg.setPeerId(peerId);
|
|
|
|
CPPUNIT_ASSERT_EQUAL(std::string("handshake peerId=%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0%f0, reserved=0000000000100004"), msg.toString());
|
|
}
|
|
|
|
void BtHandshakeMessageTest::testSetDHTEnabled()
|
|
{
|
|
BtHandshakeMessage msg;
|
|
CPPUNIT_ASSERT(!msg.isDHTEnabled());
|
|
msg.setDHTEnabled(false);
|
|
CPPUNIT_ASSERT(!msg.isDHTEnabled());
|
|
msg.setDHTEnabled(true);
|
|
CPPUNIT_ASSERT(msg.isDHTEnabled());
|
|
}
|
|
|
|
} // namespace aria2
|