mirror of
https://github.com/aria2/aria2.git
synced 2025-02-26 08:22:11 +00:00
6ae5882f3c
Added --bt-lpd-interface option to specify the interface to use for Local Peer Discovery. LpdMessageDispatcher object now has its own socket. LpdMessageReceiver's socket is binded to multicast address to only receive multicast packets. * src/BtSetup.cc * src/LpdMessageDispatcher.cc * src/LpdMessageDispatcher.h * src/LpdMessageReceiver.cc * src/LpdMessageReceiver.h * src/LpdReceiveMessageCommand.cc * src/LpdReceiveMessageCommand.h * src/OptionHandlerFactory.cc * src/SocketCore.cc * src/SocketCore.h * src/prefs.cc * src/prefs.h * src/usage_text.h * test/LpdMessageDispatcherTest.cc * test/LpdMessageReceiverTest.cc
70 lines
2.1 KiB
C++
70 lines
2.1 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));
|
|
recvsock->bind(LPD_MULTICAST_ADDR, LPD_MULTICAST_PORT);
|
|
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
|