mirror of https://github.com/aria2/aria2
344 lines
11 KiB
C++
344 lines
11 KiB
C++
#include "DefaultBtContext.h"
|
|
#include "Util.h"
|
|
#include "Exception.h"
|
|
#include "AnnounceTier.h"
|
|
#include "FixedNumberRandomizer.h"
|
|
#include <cppunit/extensions/HelperMacros.h>
|
|
|
|
using namespace std;
|
|
|
|
class DefaultBtContextTest:public CppUnit::TestFixture {
|
|
|
|
CPPUNIT_TEST_SUITE(DefaultBtContextTest);
|
|
CPPUNIT_TEST(testGetInfoHash);
|
|
CPPUNIT_TEST(testGetPieceHash);
|
|
CPPUNIT_TEST(testGetFileEntries);
|
|
CPPUNIT_TEST(testGetTotalLength);
|
|
CPPUNIT_TEST(testGetFileEntriesSingle);
|
|
CPPUNIT_TEST(testGetTotalLengthSingle);
|
|
CPPUNIT_TEST(testGetFileModeMulti);
|
|
CPPUNIT_TEST(testGetFileModeSingle);
|
|
CPPUNIT_TEST(testGetNameMulti);
|
|
CPPUNIT_TEST(testGetNameSingle);
|
|
CPPUNIT_TEST(testGetAnnounceTier);
|
|
CPPUNIT_TEST(testGetAnnounceTierAnnounceList);
|
|
CPPUNIT_TEST(testGetPieceLength);
|
|
CPPUNIT_TEST(testGetInfoHashAsString);
|
|
CPPUNIT_TEST(testGetPeerId);
|
|
CPPUNIT_TEST(testComputeFastSet);
|
|
CPPUNIT_TEST(testGetFileEntries_multiFileUrlList);
|
|
CPPUNIT_TEST(testGetFileEntries_singleFileUrlList);
|
|
CPPUNIT_TEST(testLoadFromMemory);
|
|
CPPUNIT_TEST(testLoadFromMemory_somethingMissing);
|
|
CPPUNIT_TEST_SUITE_END();
|
|
public:
|
|
void setUp() {
|
|
}
|
|
|
|
void testGetInfoHash();
|
|
void testGetPieceHash();
|
|
void testGetFileEntries();
|
|
void testGetTotalLength();
|
|
void testGetFileEntriesSingle();
|
|
void testGetTotalLengthSingle();
|
|
void testGetFileModeMulti();
|
|
void testGetFileModeSingle();
|
|
void testGetNameMulti();
|
|
void testGetNameSingle();
|
|
void testGetAnnounceTier();
|
|
void testGetAnnounceTierAnnounceList();
|
|
void testGetPieceLength();
|
|
void testGetInfoHashAsString();
|
|
void testGetPeerId();
|
|
void testComputeFastSet();
|
|
void testGetFileEntries_multiFileUrlList();
|
|
void testGetFileEntries_singleFileUrlList();
|
|
void testLoadFromMemory();
|
|
void testLoadFromMemory_somethingMissing();
|
|
};
|
|
|
|
|
|
CPPUNIT_TEST_SUITE_REGISTRATION(DefaultBtContextTest);
|
|
|
|
void DefaultBtContextTest::testGetInfoHash() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("test.torrent");
|
|
|
|
string correctHash = "248d0a1cd08284299de78d5c1ed359bb46717d8c";
|
|
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)20, btContext.getInfoHashLength());
|
|
CPPUNIT_ASSERT_EQUAL(correctHash, Util::toHex(btContext.getInfoHash(),
|
|
btContext.getInfoHashLength()));
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetPieceHash() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("test.torrent");
|
|
|
|
CPPUNIT_ASSERT_EQUAL(Util::toHex((const unsigned char*)"AAAAAAAAAAAAAAAAAAAA", 20),
|
|
btContext.getPieceHash(0));
|
|
CPPUNIT_ASSERT_EQUAL(Util::toHex((const unsigned char*)"BBBBBBBBBBBBBBBBBBBB", 20),
|
|
btContext.getPieceHash(1));
|
|
CPPUNIT_ASSERT_EQUAL(Util::toHex((const unsigned char*)"CCCCCCCCCCCCCCCCCCCC", 20),
|
|
btContext.getPieceHash(2));
|
|
CPPUNIT_ASSERT_EQUAL(string(""),
|
|
btContext.getPieceHash(-1));
|
|
CPPUNIT_ASSERT_EQUAL(string(""),
|
|
btContext.getPieceHash(3));
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetFileEntries() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("test.torrent");
|
|
// This is multi-file torrent.
|
|
FileEntries fileEntries = btContext.getFileEntries();
|
|
// There are 2 file entries.
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, fileEntries.size());
|
|
FileEntries::iterator itr = fileEntries.begin();
|
|
|
|
FileEntryHandle fileEntry1 = *itr;
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2/src/aria2c"),
|
|
fileEntry1->getPath());
|
|
itr++;
|
|
FileEntryHandle fileEntry2 = *itr;
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2-0.2.2.tar.bz2"),
|
|
fileEntry2->getPath());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetFileEntriesSingle() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("single.torrent");
|
|
// This is multi-file torrent.
|
|
FileEntries fileEntries = btContext.getFileEntries();
|
|
// There is 1 file entry.
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, fileEntries.size());
|
|
FileEntries::iterator itr = fileEntries.begin();
|
|
|
|
FileEntryHandle fileEntry1 = *itr;
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2-0.8.2.tar.bz2"),
|
|
fileEntry1->getPath());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetTotalLength() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("test.torrent");
|
|
|
|
CPPUNIT_ASSERT_EQUAL((long long int)384,
|
|
btContext.getTotalLength());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetTotalLengthSingle() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("single.torrent");
|
|
|
|
CPPUNIT_ASSERT_EQUAL((long long int)384,
|
|
btContext.getTotalLength());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetFileModeMulti() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("test.torrent");
|
|
|
|
CPPUNIT_ASSERT_EQUAL(BtContext::MULTI,
|
|
btContext.getFileMode());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetFileModeSingle() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("single.torrent");
|
|
|
|
CPPUNIT_ASSERT_EQUAL(BtContext::SINGLE,
|
|
btContext.getFileMode());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetNameMulti() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("test.torrent");
|
|
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2-test"),
|
|
btContext.getName());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetNameSingle() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("single.torrent");
|
|
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2-0.8.2.tar.bz2"),
|
|
btContext.getName());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetAnnounceTier() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("single.torrent");
|
|
|
|
AnnounceTiers tiers = btContext.getAnnounceTiers();
|
|
|
|
// There is 1 tier.
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1,
|
|
tiers.size());
|
|
|
|
AnnounceTiers::iterator itr = tiers.begin();
|
|
AnnounceTierHandle tier1 = *itr;
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1,
|
|
tier1->urls.size());
|
|
CPPUNIT_ASSERT_EQUAL(string("http://aria.rednoah.com/announce.php"),
|
|
tier1->urls.at(0));
|
|
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetAnnounceTierAnnounceList() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("test.torrent");
|
|
|
|
AnnounceTiers tiers = btContext.getAnnounceTiers();
|
|
|
|
// There are 3 tiers.
|
|
CPPUNIT_ASSERT_EQUAL((size_t)3,
|
|
tiers.size());
|
|
|
|
AnnounceTierHandle tier1 = tiers.at(0);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1,
|
|
tier1->urls.size());
|
|
CPPUNIT_ASSERT_EQUAL(string("http://tracker1"),
|
|
tier1->urls.at(0));
|
|
|
|
AnnounceTierHandle tier2 = tiers.at(1);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1,
|
|
tier2->urls.size());
|
|
CPPUNIT_ASSERT_EQUAL(string("http://tracker2"),
|
|
tier2->urls.at(0));
|
|
|
|
AnnounceTierHandle tier3 = tiers.at(2);
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1,
|
|
tier3->urls.size());
|
|
CPPUNIT_ASSERT_EQUAL(string("http://tracker3"),
|
|
tier3->urls.at(0));
|
|
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetPieceLength() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("test.torrent");
|
|
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)128,
|
|
btContext.getPieceLength());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetInfoHashAsString() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("test.torrent");
|
|
|
|
CPPUNIT_ASSERT_EQUAL(string("248d0a1cd08284299de78d5c1ed359bb46717d8c"),
|
|
btContext.getInfoHashAsString());
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetPeerId() {
|
|
DefaultBtContext btContext;
|
|
btContext.setRandomizer(new FixedNumberRandomizer());
|
|
CPPUNIT_ASSERT_EQUAL(string("%2daria2%2dAAAAAAAAAAAAA"), Util::torrentUrlencode(btContext.getPeerId(), 20));
|
|
}
|
|
|
|
void DefaultBtContextTest::testComputeFastSet()
|
|
{
|
|
string ipaddr = "192.168.0.1";
|
|
unsigned char infoHash[20];
|
|
memset(infoHash, 0, sizeof(infoHash));
|
|
infoHash[0] = 0xff;
|
|
|
|
int pieces = 1000;
|
|
int fastSetSize = 10;
|
|
|
|
DefaultBtContext btContext;
|
|
btContext.setInfoHash(infoHash);
|
|
btContext.setNumPieces(pieces);
|
|
|
|
Integers fastSet = btContext.computeFastSet(ipaddr, fastSetSize);
|
|
//for_each(fastSet.begin(), fastSet.end(), Printer());
|
|
//cerr << endl;
|
|
int ans1[] = { 686, 459, 278, 200, 404, 834, 64, 203, 760, 950 };
|
|
Integers ansSet1(&ans1[0], &ans1[10]);
|
|
CPPUNIT_ASSERT(equal(fastSet.begin(), fastSet.end(), ansSet1.begin()));
|
|
|
|
ipaddr = "10.0.0.1";
|
|
fastSet = btContext.computeFastSet(ipaddr, fastSetSize);
|
|
int ans2[] = { 568, 188, 466, 452, 550, 662, 109, 226, 398, 11 };
|
|
Integers ansSet2(&ans2[0], &ans2[10]);
|
|
CPPUNIT_ASSERT(equal(fastSet.begin(), fastSet.end(), ansSet2.begin()));
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetFileEntries_multiFileUrlList() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("url-list-multiFile.torrent");
|
|
// This is multi-file torrent.
|
|
FileEntries fileEntries = btContext.getFileEntries();
|
|
// There are 2 file entries.
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, fileEntries.size());
|
|
FileEntries::iterator itr = fileEntries.begin();
|
|
|
|
FileEntryHandle fileEntry1 = *itr;
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2/src/aria2c"),
|
|
fileEntry1->getPath());
|
|
Strings uris1 = fileEntry1->getAssociatedUris();
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, uris1.size());
|
|
CPPUNIT_ASSERT_EQUAL(string("http://localhost/dist/aria2-test/aria2/src/aria2c"),
|
|
uris1[0]);
|
|
CPPUNIT_ASSERT_EQUAL(string("http://mirror/dist/aria2-test/aria2/src/aria2c"),
|
|
uris1[1]);
|
|
|
|
itr++;
|
|
FileEntryHandle fileEntry2 = *itr;
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2-0.2.2.tar.bz2"),
|
|
fileEntry2->getPath());
|
|
Strings uris2 = fileEntry2->getAssociatedUris();
|
|
CPPUNIT_ASSERT_EQUAL((size_t)2, uris2.size());
|
|
CPPUNIT_ASSERT_EQUAL(string("http://localhost/dist/aria2-test/aria2-0.2.2.tar.bz2"),
|
|
uris2[0]);
|
|
CPPUNIT_ASSERT_EQUAL(string("http://mirror/dist/aria2-test/aria2-0.2.2.tar.bz2"),
|
|
uris2[1]);
|
|
}
|
|
|
|
void DefaultBtContextTest::testGetFileEntries_singleFileUrlList() {
|
|
DefaultBtContext btContext;
|
|
btContext.load("url-list-singleFile.torrent");
|
|
// This is multi-file torrent.
|
|
FileEntries fileEntries = btContext.getFileEntries();
|
|
// There are 1 file entries.
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, fileEntries.size());
|
|
|
|
FileEntryHandle fileEntry1 = fileEntries.front();
|
|
CPPUNIT_ASSERT_EQUAL(string("aria2.tar.bz2"),
|
|
fileEntry1->getPath());
|
|
Strings uris1 = fileEntry1->getAssociatedUris();
|
|
CPPUNIT_ASSERT_EQUAL((size_t)1, uris1.size());
|
|
CPPUNIT_ASSERT_EQUAL(string("http://localhost/dist/aria2.tar.bz2"),
|
|
uris1[0]);
|
|
}
|
|
|
|
void DefaultBtContextTest::testLoadFromMemory()
|
|
{
|
|
string memory = "d8:announce36:http://aria.rednoah.com/announce.php13:announce-listll16:http://tracker1 el15:http://tracker2el15:http://tracker3ee7:comment17:REDNOAH.COM RULES13:creation datei1123456789e4:infod5:filesld6:lengthi284e4:pathl5:aria23:src6:aria2ceed6:lengthi100e4:pathl19:aria2-0.2.2.tar.bz2eee4:name10:aria2-test12:piece lengthi128e6:pieces60:AAAAAAAAAAAAAAAAAAAABBBBBBBBBBBBBBBBBBBBCCCCCCCCCCCCCCCCCCCCee";
|
|
|
|
DefaultBtContext btContext;
|
|
btContext.loadFromMemory(memory.c_str(), memory.size(), "default");
|
|
|
|
string correctHash = "248d0a1cd08284299de78d5c1ed359bb46717d8c";
|
|
|
|
CPPUNIT_ASSERT_EQUAL((int32_t)20, btContext.getInfoHashLength());
|
|
CPPUNIT_ASSERT_EQUAL(correctHash, Util::toHex(btContext.getInfoHash(),
|
|
btContext.getInfoHashLength()));
|
|
}
|
|
|
|
void DefaultBtContextTest::testLoadFromMemory_somethingMissing()
|
|
{
|
|
// pieces missing
|
|
try {
|
|
string memory = "d8:announce36:http://aria.rednoah.com/announce.php4:infod4:name13:aria2.tar.bz26:lengthi262144eee";
|
|
DefaultBtContext btContext;
|
|
btContext.loadFromMemory(memory.c_str(), memory.size(), "default");
|
|
CPPUNIT_FAIL("exception must be thrown.");
|
|
} catch(Exception* e) {
|
|
cerr << *e << endl;
|
|
delete e;
|
|
}
|
|
}
|