mirror of
https://github.com/aria2/aria2.git
synced 2026-04-13 08:18:46 +00:00
26d6692376
Added IPv6 DHT. Added --dht-entry-porint6, --dht-file-path6, --dht-listen-addr6 and --enable-dht6 option. IPv6 DHT is disabled by default. To use IPv6 DHT, you need to use --enable-dht6 and specify a global unicast address to --dht-listen-addr6. IPv6 DHT is highly experimental. * src/BtSetup.cc * src/DHTAutoSaveCommand.cc * src/DHTAutoSaveCommand.h * src/DHTConnectionImpl.cc * src/DHTConnectionImpl.h * src/DHTEntryPointNameResolveCommand.cc * src/DHTFindNodeReplyMessage.cc * src/DHTFindNodeReplyMessage.h * src/DHTGetPeersMessage.cc * src/DHTGetPeersReplyMessage.cc * src/DHTGetPeersReplyMessage.h * src/DHTMessageFactory.h * src/DHTMessageFactoryImpl.cc * src/DHTMessageFactoryImpl.h * src/DHTMessageTracker.cc * src/DHTRegistry.cc * src/DHTRegistry.h * src/DHTRoutingTableDeserializer.cc * src/DHTRoutingTableDeserializer.h * src/DHTRoutingTableSerializer.cc * src/DHTRoutingTableSerializer.h * src/DHTSetup.cc * src/DHTSetup.h * src/FtpConnection.cc * src/LpdMessageReceiver.cc * src/OptionHandlerFactory.cc * src/OptionHandlerImpl.h * src/PeerInteractionCommand.cc * src/RequestGroup.cc * src/SocketCore.cc * src/SocketCore.h * src/bittorrent_helper.cc * src/bittorrent_helper.h * src/prefs.cc * src/prefs.h * src/usage_text.h * test/DHTConnectionImplTest.cc * test/DHTFindNodeReplyMessageTest.cc * test/DHTGetPeersMessageTest.cc * test/DHTGetPeersReplyMessageTest.cc * test/DHTMessageFactoryImplTest.cc * test/DHTRoutingTableDeserializerTest.cc * test/DHTRoutingTableSerializerTest.cc * test/LpdMessageDispatcherTest.cc * test/MockDHTMessageFactory.h
117 lines
3.6 KiB
C++
117 lines
3.6 KiB
C++
#include "DHTRoutingTableDeserializer.h"
|
|
|
|
#include <cstring>
|
|
#include <sstream>
|
|
#include <iostream>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "DHTRoutingTableSerializer.h"
|
|
#include "Exception.h"
|
|
#include "util.h"
|
|
#include "DHTNode.h"
|
|
#include "array_fun.h"
|
|
#include "DHTConstants.h"
|
|
#include "a2netcompat.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class DHTRoutingTableDeserializerTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(DHTRoutingTableDeserializerTest);
|
|
CPPUNIT_TEST(testDeserialize);
|
|
CPPUNIT_TEST(testDeserialize6);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
public:
|
|
void setUp() {}
|
|
|
|
void tearDown() {}
|
|
|
|
void testDeserialize();
|
|
|
|
void testDeserialize6();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(DHTRoutingTableDeserializerTest);
|
|
|
|
void DHTRoutingTableDeserializerTest::testDeserialize()
|
|
{
|
|
SharedHandle<DHTNode> localNode(new DHTNode());
|
|
|
|
SharedHandle<DHTNode> nodesSrc[3];
|
|
for(size_t i = 0; i < A2_ARRAY_LEN(nodesSrc); ++i) {
|
|
nodesSrc[i].reset(new DHTNode());
|
|
nodesSrc[i]->setIPAddress("192.168.0."+util::uitos(i+1));
|
|
nodesSrc[i]->setPort(6881+i);
|
|
}
|
|
nodesSrc[1]->setIPAddress("non-numerical-name");
|
|
std::vector<SharedHandle<DHTNode> > nodes(vbegin(nodesSrc), vend(nodesSrc));
|
|
|
|
DHTRoutingTableSerializer s(AF_INET);
|
|
s.setLocalNode(localNode);
|
|
s.setNodes(nodes);
|
|
|
|
std::stringstream ss;
|
|
s.serialize(ss);
|
|
|
|
DHTRoutingTableDeserializer d(AF_INET);
|
|
d.deserialize(ss);
|
|
|
|
CPPUNIT_ASSERT(memcmp(localNode->getID(), d.getLocalNode()->getID(),
|
|
DHT_ID_LENGTH) == 0);
|
|
|
|
std::cout << d.getSerializedTime().getTime() << std::endl;
|
|
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, d.getNodes().size());
|
|
const std::vector<SharedHandle<DHTNode> >& dsnodes = d.getNodes();
|
|
CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.1"), dsnodes[0]->getIPAddress());
|
|
CPPUNIT_ASSERT_EQUAL((uint16_t)6881, dsnodes[0]->getPort());
|
|
CPPUNIT_ASSERT(memcmp(nodes[0]->getID(), dsnodes[0]->getID(), DHT_ID_LENGTH) == 0);
|
|
CPPUNIT_ASSERT_EQUAL(std::string("192.168.0.3"), dsnodes[1]->getIPAddress());
|
|
CPPUNIT_ASSERT_EQUAL((uint16_t)6883, dsnodes[1]->getPort());
|
|
CPPUNIT_ASSERT(memcmp(nodes[2]->getID(), dsnodes[1]->getID(), DHT_ID_LENGTH) == 0);
|
|
}
|
|
|
|
void DHTRoutingTableDeserializerTest::testDeserialize6()
|
|
{
|
|
SharedHandle<DHTNode> localNode(new DHTNode());
|
|
|
|
SharedHandle<DHTNode> nodesSrc[3];
|
|
for(size_t i = 0; i < A2_ARRAY_LEN(nodesSrc); ++i) {
|
|
nodesSrc[i].reset(new DHTNode());
|
|
nodesSrc[i]->setIPAddress("2001::100"+util::uitos(i+1));
|
|
nodesSrc[i]->setPort(6881+i);
|
|
}
|
|
nodesSrc[1]->setIPAddress("non-numerical-name");
|
|
std::vector<SharedHandle<DHTNode> > nodes(vbegin(nodesSrc), vend(nodesSrc));
|
|
|
|
DHTRoutingTableSerializer s(AF_INET6);
|
|
s.setLocalNode(localNode);
|
|
s.setNodes(nodes);
|
|
|
|
std::stringstream ss;
|
|
s.serialize(ss);
|
|
|
|
DHTRoutingTableDeserializer d(AF_INET6);
|
|
d.deserialize(ss);
|
|
|
|
CPPUNIT_ASSERT(memcmp(localNode->getID(), d.getLocalNode()->getID(),
|
|
DHT_ID_LENGTH) == 0);
|
|
|
|
std::cout << d.getSerializedTime().getTime() << std::endl;
|
|
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, d.getNodes().size());
|
|
const std::vector<SharedHandle<DHTNode> >& dsnodes = d.getNodes();
|
|
CPPUNIT_ASSERT_EQUAL(std::string("2001::1001"), dsnodes[0]->getIPAddress());
|
|
CPPUNIT_ASSERT_EQUAL((uint16_t)6881, dsnodes[0]->getPort());
|
|
CPPUNIT_ASSERT(memcmp(nodes[0]->getID(), dsnodes[0]->getID(),
|
|
DHT_ID_LENGTH) == 0);
|
|
CPPUNIT_ASSERT_EQUAL(std::string("2001::1003"), dsnodes[1]->getIPAddress());
|
|
CPPUNIT_ASSERT_EQUAL((uint16_t)6883, dsnodes[1]->getPort());
|
|
CPPUNIT_ASSERT(memcmp(nodes[2]->getID(), dsnodes[1]->getID(),
|
|
DHT_ID_LENGTH) == 0);
|
|
}
|
|
|
|
} // namespace aria2
|