1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
#include "DHTTokenTracker.h"
#include <cppunit/extensions/HelperMacros.h>
#include "Exception.h"
#include "util.h"
#include "DHTConstants.h"
namespace aria2 {
class DHTTokenTrackerTest:public CppUnit::TestFixture {
CPPUNIT_TEST_SUITE(DHTTokenTrackerTest);
CPPUNIT_TEST(testGenerateToken);
CPPUNIT_TEST_SUITE_END();
public:
void setUp() {}
void tearDown() {}
void testGenerateToken();
};
CPPUNIT_TEST_SUITE_REGISTRATION(DHTTokenTrackerTest);
void DHTTokenTrackerTest::testGenerateToken()
{
unsigned char infohash[DHT_ID_LENGTH];
util::generateRandomData(infohash, DHT_ID_LENGTH);
std::string ipaddr = "192.168.0.1";
uint16_t port = 6881;
DHTTokenTracker tracker;
std::string token = tracker.generateToken(infohash, ipaddr, port);
CPPUNIT_ASSERT(tracker.validateToken(token, infohash, ipaddr, port));
tracker.updateTokenSecret();
CPPUNIT_ASSERT(tracker.validateToken(token, infohash, ipaddr, port));
std::string newtoken = tracker.generateToken(infohash, ipaddr, port);
tracker.updateTokenSecret();
CPPUNIT_ASSERT(!tracker.validateToken(token, infohash, ipaddr, port));
CPPUNIT_ASSERT(tracker.validateToken(newtoken, infohash, ipaddr, port));
}
} // namespace aria2
|
