mirror of
https://github.com/aria2/aria2.git
synced 2026-04-02 10:55:00 +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
74 lines
2.2 KiB
C++
74 lines
2.2 KiB
C++
#include "LpdMessageDispatcher.h"
|
|
|
|
#include <cstring>
|
|
#include <sstream>
|
|
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
#include "Exception.h"
|
|
#include "util.h"
|
|
#include "LpdMessageDispatcher.h"
|
|
#include "SocketCore.h"
|
|
#include "BtConstants.h"
|
|
|
|
namespace aria2 {
|
|
|
|
class LpdMessageDispatcherTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(LpdMessageDispatcherTest);
|
|
CPPUNIT_TEST(testCreateLpdRequest);
|
|
CPPUNIT_TEST(testSendMessage);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
public:
|
|
void testCreateLpdRequest();
|
|
void testSendMessage();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(LpdMessageDispatcherTest);
|
|
|
|
void LpdMessageDispatcherTest::testCreateLpdRequest()
|
|
{
|
|
std::string infoHashString = "cd41c7fdddfd034a15a04d7ff881216e01c4ceaf";
|
|
CPPUNIT_ASSERT_EQUAL
|
|
(std::string("BT-SEARCH * HTTP/1.1\r\n"
|
|
"Host: 239.192.152.143:6771\r\n"
|
|
"Port: 6000\r\n"
|
|
"Infohash: cd41c7fdddfd034a15a04d7ff881216e01c4ceaf\r\n"
|
|
"\r\n\r\n"),
|
|
bittorrent::createLpdRequest("239.192.152.143", 6771,
|
|
util::fromHex(infoHashString), 6000));
|
|
}
|
|
|
|
void LpdMessageDispatcherTest::testSendMessage()
|
|
{
|
|
SharedHandle<SocketCore> recvsock(new SocketCore(SOCK_DGRAM));
|
|
#ifdef __MINGW32__
|
|
recvsock->bindWithFamily(LPD_MULTICAST_PORT, AF_INET);
|
|
#else // !__MINGW32__
|
|
recvsock->bind(LPD_MULTICAST_ADDR, LPD_MULTICAST_PORT, AF_INET);
|
|
#endif // !__MINGW32__
|
|
recvsock->joinMulticastGroup(LPD_MULTICAST_ADDR, LPD_MULTICAST_PORT, "");
|
|
|
|
LpdMessageDispatcher d("cd41c7fdddfd034a15a04d7ff881216e01c4ceaf", 6000,
|
|
LPD_MULTICAST_ADDR, LPD_MULTICAST_PORT);
|
|
d.init("", 0, 1);
|
|
|
|
CPPUNIT_ASSERT(d.sendMessage());
|
|
|
|
unsigned char buf[200];
|
|
|
|
std::pair<std::string, uint16_t> peer;
|
|
ssize_t nbytes = recvsock->readDataFrom(buf, sizeof(buf), peer);
|
|
buf[nbytes] = '\0';
|
|
std::stringstream temp;
|
|
temp << "BT-SEARCH * HTTP/1.1\r\n"
|
|
<< "Host: " << LPD_MULTICAST_ADDR << ":" << LPD_MULTICAST_PORT << "\r\n"
|
|
<< "Port: " << d.getPort() << "\r\n"
|
|
<< "Infohash: " << util::toHex(d.getInfoHash()) << "\r\n"
|
|
<< "\r\n\r\n";
|
|
CPPUNIT_ASSERT_EQUAL(temp.str(), std::string(&buf[0], &buf[nbytes]));
|
|
}
|
|
|
|
} // namespace aria2
|