test_address_set.c 5.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175
  1. /* Copyright (c) 2017, The Tor Project, Inc. */
  2. /* See LICENSE for licensing information */
  3. #include "or.h"
  4. #include "crypto_rand.h"
  5. #include "address_set.h"
  6. #include "microdesc.h"
  7. #include "networkstatus.h"
  8. #include "nodelist.h"
  9. #include "routerlist.h"
  10. #include "torcert.h"
  11. #include "test.h"
  12. static networkstatus_t *dummy_ns = NULL;
  13. static networkstatus_t *
  14. mock_networkstatus_get_latest_consensus(void)
  15. {
  16. return dummy_ns;
  17. }
  18. static networkstatus_t *
  19. mock_networkstatus_get_latest_consensus_by_flavor(consensus_flavor_t f)
  20. {
  21. tor_assert(f == FLAV_MICRODESC);
  22. return dummy_ns;
  23. }
  24. /* Number of address a single node_t can have. Default to the production
  25. * value. This is to control the size of the bloom filter. */
  26. static int addr_per_node = 2;
  27. static int
  28. mock_get_estimated_address_per_node(void)
  29. {
  30. return addr_per_node;
  31. }
  32. static void
  33. test_contains(void *arg)
  34. {
  35. int ret;
  36. address_set_t *set = NULL;
  37. (void) arg;
  38. /* Setup an IPv4 and IPv6 addresses. */
  39. tor_addr_t addr_v6;
  40. tor_addr_parse(&addr_v6, "1:2:3:4::");
  41. tor_addr_t addr_v4;
  42. tor_addr_parse(&addr_v4, "42.42.42.42");
  43. uint32_t ipv4h = tor_addr_to_ipv4h(&addr_v4);
  44. /* Make it very big so the chance of failing the contain test will be
  45. * extremely rare. */
  46. set = address_set_new(1024);
  47. tt_assert(set);
  48. /* Add and lookup IPv6. */
  49. address_set_add(set, &addr_v6);
  50. ret = address_set_probably_contains(set, &addr_v6);
  51. tt_int_op(ret, OP_EQ, 1);
  52. /* Add and lookup IPv4. */
  53. address_set_add_ipv4h(set, ipv4h);
  54. ret = address_set_probably_contains(set, &addr_v4);
  55. tt_int_op(ret, OP_EQ, 1);
  56. /* Try a lookup of rubbish. */
  57. tor_addr_t dummy_addr;
  58. memset(&dummy_addr, 'A', sizeof(dummy_addr));
  59. dummy_addr.family = AF_INET;
  60. ret = address_set_probably_contains(set, &dummy_addr);
  61. tt_int_op(ret, OP_EQ, 0);
  62. dummy_addr.family = AF_INET6;
  63. ret = address_set_probably_contains(set, &dummy_addr);
  64. tt_int_op(ret, OP_EQ, 0);
  65. done:
  66. address_set_free(set);
  67. }
  68. static void
  69. test_nodelist(void *arg)
  70. {
  71. int ret;
  72. routerstatus_t *rs = NULL; microdesc_t *md = NULL; routerinfo_t *ri = NULL;
  73. (void) arg;
  74. MOCK(networkstatus_get_latest_consensus,
  75. mock_networkstatus_get_latest_consensus);
  76. MOCK(networkstatus_get_latest_consensus_by_flavor,
  77. mock_networkstatus_get_latest_consensus_by_flavor);
  78. MOCK(get_estimated_address_per_node,
  79. mock_get_estimated_address_per_node);
  80. dummy_ns = tor_malloc_zero(sizeof(*dummy_ns));
  81. dummy_ns->flavor = FLAV_MICRODESC;
  82. dummy_ns->routerstatus_list = smartlist_new();
  83. tor_addr_t addr_v4, addr_v6, dummy_addr;
  84. tor_addr_parse(&addr_v4, "42.42.42.42");
  85. uint32_t ipv4h = tor_addr_to_ipv4h(&addr_v4);
  86. tor_addr_parse(&addr_v6, "1:2:3:4::");
  87. memset(&dummy_addr, 'A', sizeof(dummy_addr));
  88. /* This will make the nodelist bloom filter very large
  89. * (the_nodelist->node_addrs) so we will fail the contain test rarely. */
  90. addr_per_node = 1024;
  91. /* No node no nothing. The lookups should be empty. */
  92. nodelist_set_consensus(dummy_ns);
  93. /* The address set should be empty. */
  94. ret = nodelist_probably_contains_address(&addr_v4);
  95. tt_int_op(ret, OP_EQ, 0);
  96. ret = nodelist_probably_contains_address(&addr_v6);
  97. tt_int_op(ret, OP_EQ, 0);
  98. dummy_addr.family = AF_INET;
  99. ret = nodelist_probably_contains_address(&dummy_addr);
  100. tt_int_op(ret, OP_EQ, 0);
  101. dummy_addr.family = AF_INET6;
  102. ret = nodelist_probably_contains_address(&dummy_addr);
  103. tt_int_op(ret, OP_EQ, 0);
  104. md = tor_malloc_zero(sizeof(*md));
  105. ri = tor_malloc_zero(sizeof(*ri));
  106. rs = tor_malloc_zero(sizeof(*rs));
  107. crypto_rand(rs->identity_digest, sizeof(rs->identity_digest));
  108. crypto_rand(md->digest, sizeof(md->digest));
  109. memcpy(rs->descriptor_digest, md->digest, DIGEST256_LEN);
  110. /* Setup the rs, ri and md addresses. */
  111. rs->addr = ipv4h;
  112. tor_addr_parse(&rs->ipv6_addr, "1:2:3:4::");
  113. ri->addr = ipv4h;
  114. tor_addr_parse(&ri->ipv6_addr, "1:2:3:4::");
  115. tor_addr_parse(&md->ipv6_addr, "1:2:3:4::");
  116. /* Add the rs to the consensus becoming a node_t. */
  117. smartlist_add(dummy_ns->routerstatus_list, rs);
  118. nodelist_set_consensus(dummy_ns);
  119. /* At this point, the address set should be initialized in the nodelist and
  120. * we should be able to lookup. */
  121. ret = nodelist_probably_contains_address(&addr_v4);
  122. tt_int_op(ret, OP_EQ, 1);
  123. ret = nodelist_probably_contains_address(&addr_v6);
  124. tt_int_op(ret, OP_EQ, 1);
  125. /* Lookup unknown address. */
  126. dummy_addr.family = AF_INET;
  127. ret = nodelist_probably_contains_address(&dummy_addr);
  128. tt_int_op(ret, OP_EQ, 0);
  129. dummy_addr.family = AF_INET6;
  130. ret = nodelist_probably_contains_address(&dummy_addr);
  131. tt_int_op(ret, OP_EQ, 0);
  132. done:
  133. routerstatus_free(rs); routerinfo_free(ri); microdesc_free(md);
  134. smartlist_clear(dummy_ns->routerstatus_list);
  135. networkstatus_vote_free(dummy_ns);
  136. UNMOCK(networkstatus_get_latest_consensus);
  137. UNMOCK(networkstatus_get_latest_consensus_by_flavor);
  138. UNMOCK(get_estimated_address_per_node);
  139. }
  140. struct testcase_t address_set_tests[] = {
  141. { "contains", test_contains, TT_FORK,
  142. NULL, NULL },
  143. { "nodelist", test_nodelist, TT_FORK,
  144. NULL, NULL },
  145. END_OF_TESTCASES
  146. };