mirror of
https://github.com/aria2/aria2.git
synced 2025-04-04 21:17:41 +03:00
f5b68379d5
Removed incomingPeer. Set 0 to peer's port if it is not a listening
port.
* src/DefaultPeerStorage.{h, cc}
* test/DefaultPeerStorageTest.cc
* src/HandshakeExtensionMessage.cc
* test/HandshakeExtensionMessageTest.cc
* src/Peer.{h, cc}: Added ipaddr and port to identity comparison.
* src/PeerStorage.h
* test/MockPeerStorage.h
* src/PeerListenCommand.cc
135 lines
3.9 KiB
C++
135 lines
3.9 KiB
C++
#include "HandshakeExtensionMessage.h"
|
|
#include "Peer.h"
|
|
#include "MockBtContext.h"
|
|
#include "Exception.h"
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
class HandshakeExtensionMessageTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(HandshakeExtensionMessageTest);
|
|
CPPUNIT_TEST(testGetExtensionMessageID);
|
|
CPPUNIT_TEST(testGetExtensionName);
|
|
CPPUNIT_TEST(testGetBencodedData);
|
|
CPPUNIT_TEST(testToString);
|
|
CPPUNIT_TEST(testDoReceivedAction);
|
|
CPPUNIT_TEST(testCreate);
|
|
CPPUNIT_TEST(testCreate_stringnum);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
BtContextHandle _btContext;
|
|
public:
|
|
HandshakeExtensionMessageTest():_btContext(0) {}
|
|
|
|
void setUp() {}
|
|
|
|
void tearDown() {}
|
|
|
|
void testGetExtensionMessageID();
|
|
void testGetExtensionName();
|
|
void testGetBencodedData();
|
|
void testToString();
|
|
void testDoReceivedAction();
|
|
void testCreate();
|
|
void testCreate_stringnum();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(HandshakeExtensionMessageTest);
|
|
|
|
void HandshakeExtensionMessageTest::testGetExtensionMessageID()
|
|
{
|
|
HandshakeExtensionMessage msg;
|
|
CPPUNIT_ASSERT_EQUAL((uint8_t)0, msg.getExtensionMessageID());
|
|
}
|
|
|
|
void HandshakeExtensionMessageTest::testGetExtensionName()
|
|
{
|
|
HandshakeExtensionMessage msg;
|
|
CPPUNIT_ASSERT_EQUAL(string("handshake"), msg.getExtensionName());
|
|
}
|
|
|
|
void HandshakeExtensionMessageTest::testGetBencodedData()
|
|
{
|
|
HandshakeExtensionMessage msg;
|
|
msg.setClientVersion("aria2");
|
|
msg.setTCPPort(6889);
|
|
msg.setExtension("ut_pex", 1);
|
|
msg.setExtension("a2_dht", 2);
|
|
CPPUNIT_ASSERT_EQUAL(string("d1:v5:aria21:pi6889e1:md6:a2_dhti2e6:ut_pexi1eee"), msg.getBencodedData());
|
|
}
|
|
|
|
void HandshakeExtensionMessageTest::testToString()
|
|
{
|
|
HandshakeExtensionMessage msg;
|
|
msg.setClientVersion("aria2");
|
|
msg.setTCPPort(6889);
|
|
msg.setExtension("ut_pex", 1);
|
|
msg.setExtension("a2_dht", 2);
|
|
CPPUNIT_ASSERT_EQUAL(string("handshake client=aria2, tcpPort=6889, a2_dht=2, ut_pex=1"), msg.toString());
|
|
}
|
|
|
|
void HandshakeExtensionMessageTest::testDoReceivedAction()
|
|
{
|
|
PeerHandle peer = new Peer("192.168.0.1", 0, 1, 1);
|
|
HandshakeExtensionMessage msg;
|
|
msg.setClientVersion("aria2");
|
|
msg.setTCPPort(6889);
|
|
msg.setExtension("ut_pex", 1);
|
|
msg.setExtension("a2_dht", 2);
|
|
msg.setPeer(peer);
|
|
msg.setBtContext(_btContext);
|
|
|
|
msg.doReceivedAction();
|
|
|
|
CPPUNIT_ASSERT_EQUAL((uint16_t)6889, peer->port);
|
|
CPPUNIT_ASSERT_EQUAL((uint8_t)1, peer->getExtensionMessageID("ut_pex"));
|
|
CPPUNIT_ASSERT_EQUAL((uint8_t)2, peer->getExtensionMessageID("a2_dht"));
|
|
}
|
|
|
|
void HandshakeExtensionMessageTest::testCreate()
|
|
{
|
|
string in = "0d1:pi6881e1:v5:aria21:md6:ut_pexi1eee";
|
|
HandshakeExtensionMessageHandle m =
|
|
HandshakeExtensionMessage::create(in.c_str(), in.size());
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2"), m->getClientVersion());
|
|
CPPUNIT_ASSERT_EQUAL((uint16_t)6881, m->getTCPPort());
|
|
CPPUNIT_ASSERT_EQUAL((uint8_t)1, m->getExtensionMessageID("ut_pex"));
|
|
|
|
try {
|
|
// bad payload format
|
|
string in = "011:hello world";
|
|
HandshakeExtensionMessage::create(in.c_str(), in.size());
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(Exception* e) {
|
|
cerr << *e << endl;
|
|
delete e;
|
|
}
|
|
try {
|
|
// malformed dencoded message
|
|
string in = "011:hello";
|
|
HandshakeExtensionMessage::create(in.c_str(), in.size());
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(Exception* e) {
|
|
cerr << *e << endl;
|
|
delete e;
|
|
}
|
|
try {
|
|
// 0 length data
|
|
string in = "";
|
|
HandshakeExtensionMessage::create(in.c_str(), in.size());
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(Exception* e) {
|
|
cerr << *e << endl;
|
|
delete e;
|
|
}
|
|
}
|
|
|
|
void HandshakeExtensionMessageTest::testCreate_stringnum()
|
|
{
|
|
string in = "0d1:p4:68811:v5:aria21:md6:ut_pex1:1ee";
|
|
HandshakeExtensionMessageHandle m =
|
|
HandshakeExtensionMessage::create(in.c_str(), in.size());
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2"), m->getClientVersion());
|
|
CPPUNIT_ASSERT_EQUAL((uint16_t)6881, m->getTCPPort());
|
|
CPPUNIT_ASSERT_EQUAL((uint8_t)1, m->getExtensionMessageID("ut_pex"));
|
|
}
|