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
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
#include "unity.h"
#include "proto.h"
#include "log.h"
#include <string.h>
void test_begin_pings_bootstrap_node() {
struct message outbuff[10] = {0};
struct dht dht;
dht.self = (struct nodeid){.inner={0x42424242, 0x42424242, 0x42424242, 0x42424242, 0x42424242}};
struct message* message_cursor = outbuff;
proto_begin(&dht, &message_cursor, outbuff+10);
TEST_ASSERT_EQUAL_PTR(message_cursor, outbuff+1);
TEST_ASSERT_EQUAL(91, outbuff[0].payload_len);
TEST_ASSERT_EQUAL_CHAR_ARRAY("d1:ad2:id20:BBBBBBBBBBBBBBBBBBBB6:target20:", outbuff[0].payload, 43);
TEST_ASSERT_EQUAL_CHAR_ARRAY("e1:q9:find_node1:t1:01:y1:qe", outbuff[0].payload+63, 28);
}
void test_response_from_initial_probe() {
struct message outbuff[2] = {0};
struct sockaddr_storage remote;
socklen_t remote_len;
struct dht dht;
dht.self = (struct nodeid){.inner={0x42424242, 0x42424242, 0x42424242, 0x42424242, 0x42424242}};
{
struct message* message_cursor = outbuff;
proto_begin(&dht, &message_cursor, outbuff+2);
remote_len = outbuff[0].dest_len;
memcpy(&remote, &outbuff[0].dest, remote_len);
}
char buff[] = "d1:y1:r1:t1:01:rd2:id20:aaaaaaaaaaaaaaaaaaaa5:nodes26:bbbbbbbbbbbbbbbbbbbb\xFF\xFF\xFF\xFF\x00\x00""ee";
struct message* message_cursor = outbuff;
proto_run(&dht, buff, sizeof(buff), (struct sockaddr_in*)&remote, remote_len, &message_cursor, outbuff+2);
TEST_ASSERT_EQUAL_PTR_MESSAGE(message_cursor, outbuff+1, "Sends one packet");
TEST_ASSERT_EQUAL(91, outbuff[0].payload_len);
TEST_ASSERT_EQUAL_CHAR_ARRAY("d1:ad2:id20:BBBBBBBBBBBBBBBBBBBB6:target20:", outbuff[0].payload, 43);
TEST_ASSERT_EQUAL_CHAR_ARRAY("e1:q9:find_node1:t1:11:y1:qe", outbuff[0].payload+63, 28);
// The queried node gets added to the routing table
struct nodeid other = (struct nodeid){.inner={0x61616161, 0x61616161, 0x61616161, 0x61616161, 0x61616161}};
struct entry* entry = routing_get(&other);
TEST_ASSERT_NOT_NULL(entry);
TEST_ASSERT_EQUAL(((struct sockaddr_in*)&remote)->sin_addr.s_addr, entry->addr.ip);
TEST_ASSERT_EQUAL(((struct sockaddr_in*)&remote)->sin_port, entry->addr.port);
}
void test_reponse_from_wrong_ip() {
struct message outbuff[2] = {0};
struct dht dht;
dht.self = (struct nodeid){.inner={0x42424242, 0x42424242, 0x42424242, 0x42424242, 0x42424242}};
{
struct message* message_cursor = outbuff;
proto_begin(&dht, &message_cursor, outbuff+2);
}
// This is fragile, since the ip of the bootstrap node could change, and we
// look it up from DNS. Although it pretty unlikely that it would change to
// this ip
struct sockaddr_in remote;
remote.sin_family = AF_INET;
inet_pton(AF_INET, "255.255.255.255", &remote.sin_addr.s_addr);
remote.sin_port = htons(6881);
char buff[] = "d1:y1:r1:t1:01:rd2:id20:aaaaaaaaaaaaaaaaaaaa5:nodes26:bbbbbbbbbbbbbbbbbbbb\xFF\xFF\xFF\xFF\x00\x00""ee";
struct message* message_cursor = outbuff;
proto_run(&dht, buff, sizeof(buff), &remote, sizeof(remote), &message_cursor, outbuff+2);
// We shouldn't send any packets, since the response is rejected
TEST_ASSERT_EQUAL_PTR(message_cursor, outbuff);
// Since the ip was wrong we should not have accepted the node into the
// rounting table
struct nodeid other = (struct nodeid){.inner={0x61616161, 0x61616161, 0x61616161, 0x61616161, 0x61616161}};
struct entry* entry = routing_get(&other);
TEST_ASSERT_NULL(entry);
}
void test_ping() {
struct message outbuff[2] = {0};
struct dht dht;
dht.self = (struct nodeid){.inner={0x42424242, 0x42424242, 0x42424242, 0x42424242, 0x42424242}};
{
struct message* message_cursor = outbuff;
proto_begin(&dht, &message_cursor, outbuff+2);
}
struct sockaddr_in remote;
remote.sin_family = AF_INET;
inet_pton(AF_INET, "255.255.255.255", &remote.sin_addr.s_addr);
remote.sin_port = htons(6881);
char buff[] = "d1:ad2:id20:abcdefghij0123456789e1:q4:ping1:t2:aa1:y1:qe";
struct message* message_cursor = outbuff;
proto_run(&dht, buff, sizeof(buff), &remote, sizeof(remote), &message_cursor, outbuff+2);
// We should have sent a response
TEST_ASSERT_EQUAL_PTR(message_cursor, outbuff+1);
TEST_ASSERT_EQUAL(48, outbuff[0].payload_len);
TEST_ASSERT_EQUAL_CHAR_ARRAY("d1:t2:aa1:y1:r1:rd2:id20:BBBBBBBBBBBBBBBBBBBBee", outbuff[0].payload, 47);
}
|
