mirror of
https://github.com/aria2/aria2.git
synced 2025-02-26 08:22:11 +00:00
3505201f33
Rewritten SharedHandle. Now copy constructor taking raw pointer has keyword explicit and SharedHandle's default constructor initializes its internal obj to null, old implementation initializes it using obj's default constructor. To assign null, write SharedHandle<T> x(...); x.reset(); TODO: test/SharedHandleTest.cc needs more tests. * src/SharedHandle.h
127 lines
3.7 KiB
C++
127 lines
3.7 KiB
C++
#include "DefaultExtensionMessageFactory.h"
|
|
#include "Peer.h"
|
|
#include "MockBtContext.h"
|
|
#include "PeerMessageUtil.h"
|
|
#include "HandshakeExtensionMessage.h"
|
|
#include "UTPexExtensionMessage.h"
|
|
#include "Exception.h"
|
|
#include "BtRegistry.h"
|
|
#include "BtRuntime.h"
|
|
#include "FileEntry.h"
|
|
#include <iostream>
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
namespace aria2 {
|
|
|
|
class DefaultExtensionMessageFactoryTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(DefaultExtensionMessageFactoryTest);
|
|
CPPUNIT_TEST(testCreateMessage_unknown);
|
|
CPPUNIT_TEST(testCreateMessage_Handshake);
|
|
CPPUNIT_TEST(testCreateMessage_UTPex);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
private:
|
|
SharedHandle<MockBtContext> _btContext;
|
|
SharedHandle<Peer> _peer;
|
|
public:
|
|
void setUp()
|
|
{
|
|
BtRegistry::unregisterAll();
|
|
SharedHandle<MockBtContext> btContext(new MockBtContext());
|
|
unsigned char infohash[20];
|
|
memset(infohash, 0, sizeof(infohash));
|
|
btContext->setInfoHash(infohash);
|
|
_btContext = btContext;
|
|
|
|
SharedHandle<BtRuntime> btRuntime(new BtRuntime());
|
|
BtRegistry::registerBtRuntime(_btContext->getInfoHashAsString(),
|
|
btRuntime);
|
|
|
|
_peer.reset(new Peer("192.168.0.1", 6969));
|
|
_peer->allocateSessionResource(1024, 1024*1024);
|
|
_peer->setExtension("ut_pex", 1);
|
|
}
|
|
|
|
void tearDown()
|
|
{
|
|
BtRegistry::unregisterAll();
|
|
}
|
|
|
|
void testCreateMessage_unknown();
|
|
void testCreateMessage_Handshake();
|
|
void testCreateMessage_UTPex();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(DefaultExtensionMessageFactoryTest);
|
|
|
|
void DefaultExtensionMessageFactoryTest::testCreateMessage_unknown()
|
|
{
|
|
DefaultExtensionMessageFactory factory;
|
|
factory.setBtContext(_btContext);
|
|
factory.setPeer(_peer);
|
|
_peer->setExtension("foo", 255);
|
|
|
|
char id[1] = { 255 };
|
|
|
|
std::string data = std::string(&id[0], &id[1]);
|
|
try {
|
|
// this test fails because localhost doesn't have extension id = 255.
|
|
factory.createMessage(reinterpret_cast<const unsigned char*>(data.c_str()),
|
|
data.size());
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(Exception* e) {
|
|
std::cerr << *e << std::endl;
|
|
delete e;
|
|
}
|
|
}
|
|
|
|
void DefaultExtensionMessageFactoryTest::testCreateMessage_Handshake()
|
|
{
|
|
DefaultExtensionMessageFactory factory;
|
|
factory.setBtContext(_btContext);
|
|
factory.setPeer(_peer);
|
|
|
|
char id[1] = { 0 };
|
|
|
|
std::string data = std::string(&id[0], &id[1])+"d1:v5:aria2e";
|
|
SharedHandle<HandshakeExtensionMessage> m
|
|
(dynamic_pointer_cast<HandshakeExtensionMessage>
|
|
(factory.createMessage(reinterpret_cast<const unsigned char*>(data.c_str()),
|
|
data.size())));
|
|
CPPUNIT_ASSERT_EQUAL(std::string("aria2"), m->getClientVersion());
|
|
}
|
|
|
|
void DefaultExtensionMessageFactoryTest::testCreateMessage_UTPex()
|
|
{
|
|
DefaultExtensionMessageFactory factory;
|
|
factory.setBtContext(_btContext);
|
|
factory.setPeer(_peer);
|
|
|
|
unsigned char c1[6];
|
|
unsigned char c2[6];
|
|
unsigned char c3[6];
|
|
unsigned char c4[6];
|
|
PeerMessageUtil::createcompact(c1, "192.168.0.1", 6881);
|
|
PeerMessageUtil::createcompact(c2, "10.1.1.2", 9999);
|
|
PeerMessageUtil::createcompact(c3, "192.168.0.2", 6882);
|
|
PeerMessageUtil::createcompact(c4, "10.1.1.3",10000);
|
|
|
|
char id[1] = { factory.getExtensionMessageID("ut_pex") };
|
|
|
|
std::string data = std::string(&id[0], &id[1])+"d5:added12:"+
|
|
std::string(&c1[0], &c1[6])+std::string(&c2[0], &c2[6])+
|
|
"7:added.f2:207:dropped12:"+
|
|
std::string(&c3[0], &c3[6])+std::string(&c4[0], &c4[6])+
|
|
"e";
|
|
|
|
SharedHandle<UTPexExtensionMessage> m
|
|
(dynamic_pointer_cast<UTPexExtensionMessage>
|
|
(factory.createMessage(reinterpret_cast<const unsigned char*>(data.c_str()),
|
|
data.size())));
|
|
CPPUNIT_ASSERT_EQUAL(factory.getExtensionMessageID("ut_pex"),
|
|
m->getExtensionMessageID());
|
|
}
|
|
|
|
} // namespace aria2
|