diff options
Diffstat (limited to 'test/BtHandshakeMessageTest.cc')
| -rw-r--r-- | test/BtHandshakeMessageTest.cc | 85 |
1 files changed, 44 insertions, 41 deletions
diff --git a/test/BtHandshakeMessageTest.cc b/test/BtHandshakeMessageTest.cc index f8c17acc..a86ae54a 100644 --- a/test/BtHandshakeMessageTest.cc +++ b/test/BtHandshakeMessageTest.cc @@ -9,7 +9,7 @@ namespace aria2 { -class BtHandshakeMessageTest:public CppUnit::TestFixture { +class BtHandshakeMessageTest : public CppUnit::TestFixture { CPPUNIT_TEST_SUITE(BtHandshakeMessageTest); CPPUNIT_TEST(testCreate); @@ -17,11 +17,10 @@ class BtHandshakeMessageTest:public CppUnit::TestFixture { CPPUNIT_TEST(testToString); CPPUNIT_TEST(testSetDHTEnabled); CPPUNIT_TEST_SUITE_END(); -private: +private: public: - void setUp() { - } + void setUp() {} void testCreate(); void testCreateMessage(); @@ -31,54 +30,55 @@ public: static std::string BTPSTR; }; - std::string BtHandshakeMessageTest::BTPSTR = "BitTorrent protocol"; CPPUNIT_TEST_SUITE_REGISTRATION(BtHandshakeMessageTest); -void createHandshakeMessageData(unsigned char* msg) { +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 }; + 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 }; + 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 }; + 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() { +void BtHandshakeMessageTest::testCreate() +{ unsigned char msg[68]; createHandshakeMessageData(msg); - std::shared_ptr<BtHandshakeMessage> message = BtHandshakeMessage::create(&msg[0], sizeof(msg)); + std::shared_ptr<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( + 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::testCreateMessage() { - 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 }; +void BtHandshakeMessageTest::testCreateMessage() +{ + 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}; std::shared_ptr<BtHandshakeMessage> msg(new BtHandshakeMessage()); msg->setInfoHash(infoHash); @@ -89,24 +89,27 @@ void BtHandshakeMessageTest::testCreateMessage() { unsigned char* rawmsg = msg->createMessage(); CPPUNIT_ASSERT_EQUAL(util::toHex((const unsigned char*)data, 68), util::toHex(rawmsg, 68)); - delete [] rawmsg; + delete[] rawmsg; } -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 }; +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()); + 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() |