123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 |
- #include "DHTFindNodeReplyMessage.h"
- #include <cppunit/extensions/HelperMacros.h>
- #include "DHTNode.h"
- #include "Exception.h"
- #include "util.h"
- #include "DHTBucket.h"
- #include "bittorrent_helper.h"
- #include "bencode.h"
- namespace aria2 {
- class DHTFindNodeReplyMessageTest:public CppUnit::TestFixture {
- CPPUNIT_TEST_SUITE(DHTFindNodeReplyMessageTest);
- CPPUNIT_TEST(testGetBencodedMessage);
- CPPUNIT_TEST_SUITE_END();
- public:
- void setUp() {}
- void tearDown() {}
- void testGetBencodedMessage();
- };
- CPPUNIT_TEST_SUITE_REGISTRATION(DHTFindNodeReplyMessageTest);
- void DHTFindNodeReplyMessageTest::testGetBencodedMessage()
- {
- SharedHandle<DHTNode> localNode(new DHTNode());
- SharedHandle<DHTNode> remoteNode(new DHTNode());
- unsigned char tid[DHT_TRANSACTION_ID_LENGTH];
- util::generateRandomData(tid, DHT_TRANSACTION_ID_LENGTH);
- std::string transactionID(&tid[0], &tid[DHT_TRANSACTION_ID_LENGTH]);
- DHTFindNodeReplyMessage msg(localNode, remoteNode, transactionID);
- msg.setVersion("A200");
- std::string compactNodeInfo;
- SharedHandle<DHTNode> nodes[8];
- for(size_t i = 0; i < DHTBucket::K; ++i) {
- nodes[i].reset(new DHTNode());
- nodes[i]->setIPAddress("192.168.0."+util::uitos(i+1));
- nodes[i]->setPort(6881+i);
- unsigned char buf[6];
- CPPUNIT_ASSERT(bittorrent::createcompact
- (buf, nodes[i]->getIPAddress(), nodes[i]->getPort()));
- compactNodeInfo +=
- std::string(&nodes[i]->getID()[0], &nodes[i]->getID()[DHT_ID_LENGTH])+
- std::string(&buf[0], &buf[sizeof(buf)]);
- }
- msg.setClosestKNodes
- (std::vector<SharedHandle<DHTNode> >(&nodes[0], &nodes[DHTBucket::K]));
- std::string msgbody = msg.getBencodedMessage();
- BDE dict = BDE::dict();
- dict["t"] = transactionID;
- dict["v"] = BDE("A200");
- dict["y"] = BDE("r");
- BDE rDict = BDE::dict();
- rDict["id"] = BDE(localNode->getID(), DHT_ID_LENGTH);
- rDict["nodes"] = compactNodeInfo;
- dict["r"] = rDict;
- CPPUNIT_ASSERT_EQUAL(bencode::encode(dict), msgbody);
- }
- } // namespace aria2
|