diff options
Diffstat (limited to 'auto_tests/onion_test.c')
-rw-r--r-- | auto_tests/onion_test.c | 27 |
1 files changed, 19 insertions, 8 deletions
diff --git a/auto_tests/onion_test.c b/auto_tests/onion_test.c index a0d4f4bc..16acf676 100644 --- a/auto_tests/onion_test.c +++ b/auto_tests/onion_test.c | |||
@@ -19,6 +19,22 @@ | |||
19 | 19 | ||
20 | #include "helpers.h" | 20 | #include "helpers.h" |
21 | 21 | ||
22 | #ifndef USE_IPV6 | ||
23 | #define USE_IPV6 1 | ||
24 | #endif | ||
25 | |||
26 | static inline IP get_loopback() | ||
27 | { | ||
28 | IP ip; | ||
29 | #if USE_IPV6 | ||
30 | ip.family = TOX_AF_INET6; | ||
31 | ip.ip6 = get_ip6_loopback(); | ||
32 | #else | ||
33 | ip.family = TOX_AF_INET; | ||
34 | ip.ip4 = get_ip4_loopback(); | ||
35 | #endif | ||
36 | return ip; | ||
37 | } | ||
22 | static void do_onion(Onion *onion) | 38 | static void do_onion(Onion *onion) |
23 | { | 39 | { |
24 | networking_poll(onion->net, NULL); | 40 | networking_poll(onion->net, NULL); |
@@ -141,9 +157,7 @@ static int handle_test_4(void *object, IP_Port source, const uint8_t *packet, ui | |||
141 | 157 | ||
142 | START_TEST(test_basic) | 158 | START_TEST(test_basic) |
143 | { | 159 | { |
144 | IP ip; | 160 | IP ip = get_loopback(); |
145 | ip_init(&ip, 1); | ||
146 | ip.ip6.uint8[15] = 1; | ||
147 | Onion *onion1 = new_onion(new_DHT(NULL, new_networking(NULL, ip, 34567), true)); | 161 | Onion *onion1 = new_onion(new_DHT(NULL, new_networking(NULL, ip, 34567), true)); |
148 | Onion *onion2 = new_onion(new_DHT(NULL, new_networking(NULL, ip, 34568), true)); | 162 | Onion *onion2 = new_onion(new_DHT(NULL, new_networking(NULL, ip, 34568), true)); |
149 | ck_assert_msg((onion1 != NULL) && (onion2 != NULL), "Onion failed initializing."); | 163 | ck_assert_msg((onion1 != NULL) && (onion2 != NULL), "Onion failed initializing."); |
@@ -281,8 +295,7 @@ typedef struct { | |||
281 | 295 | ||
282 | static Onions *new_onions(uint16_t port) | 296 | static Onions *new_onions(uint16_t port) |
283 | { | 297 | { |
284 | IP ip; | 298 | IP ip = get_loopback(); |
285 | ip_init(&ip, 1); | ||
286 | ip.ip6.uint8[15] = 1; | 299 | ip.ip6.uint8[15] = 1; |
287 | Onions *on = (Onions *)malloc(sizeof(Onions)); | 300 | Onions *on = (Onions *)malloc(sizeof(Onions)); |
288 | DHT *dht = new_DHT(NULL, new_networking(NULL, ip, port), true); | 301 | DHT *dht = new_DHT(NULL, new_networking(NULL, ip, port), true); |
@@ -386,9 +399,7 @@ START_TEST(test_announce) | |||
386 | ck_assert_msg(onions[i] != 0, "Failed to create onions. %u"); | 399 | ck_assert_msg(onions[i] != 0, "Failed to create onions. %u"); |
387 | } | 400 | } |
388 | 401 | ||
389 | IP ip; | 402 | IP ip = get_loopback(); |
390 | ip_init(&ip, 1); | ||
391 | ip.ip6.uint8[15] = 1; | ||
392 | 403 | ||
393 | for (i = 3; i < NUM_ONIONS; ++i) { | 404 | for (i = 3; i < NUM_ONIONS; ++i) { |
394 | IP_Port ip_port = {ip, onions[i - 1]->onion->net->port}; | 405 | IP_Port ip_port = {ip, onions[i - 1]->onion->net->port}; |