test_routerlist.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616
  1. /* Copyright (c) 2014-2017, The Tor Project, Inc. */
  2. /* See LICENSE for licensing information */
  3. #include "orconfig.h"
  4. #include <math.h>
  5. #include <time.h>
  6. #define CONNECTION_PRIVATE
  7. #define DIRECTORY_PRIVATE
  8. #define DIRVOTE_PRIVATE
  9. #define ENTRYNODES_PRIVATE
  10. #define HIBERNATE_PRIVATE
  11. #define NETWORKSTATUS_PRIVATE
  12. #define ROUTERLIST_PRIVATE
  13. #define TOR_UNIT_TESTING
  14. #include "or.h"
  15. #include "config.h"
  16. #include "connection.h"
  17. #include "container.h"
  18. #include "control.h"
  19. #include "crypto_rand.h"
  20. #include "directory.h"
  21. #include "dirvote.h"
  22. #include "entrynodes.h"
  23. #include "hibernate.h"
  24. #include "microdesc.h"
  25. #include "networkstatus.h"
  26. #include "nodelist.h"
  27. #include "policies.h"
  28. #include "router.h"
  29. #include "routerlist.h"
  30. #include "routerset.h"
  31. #include "routerparse.h"
  32. #include "shared_random.h"
  33. #include "statefile.h"
  34. #include "test.h"
  35. #include "test_dir_common.h"
  36. #include "log_test_helpers.h"
  37. void construct_consensus(char **consensus_text_md);
  38. static authority_cert_t *mock_cert;
  39. static authority_cert_t *
  40. get_my_v3_authority_cert_m(void)
  41. {
  42. tor_assert(mock_cert);
  43. return mock_cert;
  44. }
  45. /* 4 digests + 3 sep + pre + post + NULL */
  46. static char output[4*BASE64_DIGEST256_LEN+3+2+2+1];
  47. static void
  48. mock_get_from_dirserver(uint8_t dir_purpose, uint8_t router_purpose,
  49. const char *resource, int pds_flags,
  50. download_want_authority_t want_authority)
  51. {
  52. (void)dir_purpose;
  53. (void)router_purpose;
  54. (void)pds_flags;
  55. (void)want_authority;
  56. tt_assert(resource);
  57. strlcpy(output, resource, sizeof(output));
  58. done:
  59. ;
  60. }
  61. static void
  62. test_routerlist_initiate_descriptor_downloads(void *arg)
  63. {
  64. const char *prose = "unhurried and wise, we perceive.";
  65. smartlist_t *digests = smartlist_new();
  66. (void)arg;
  67. for (int i = 0; i < 20; i++) {
  68. smartlist_add(digests, (char*)prose);
  69. }
  70. MOCK(directory_get_from_dirserver, mock_get_from_dirserver);
  71. initiate_descriptor_downloads(NULL, DIR_PURPOSE_FETCH_MICRODESC,
  72. digests, 3, 7, 0);
  73. UNMOCK(directory_get_from_dirserver);
  74. tt_str_op(output, OP_EQ, "d/"
  75. "dW5odXJyaWVkIGFuZCB3aXNlLCB3ZSBwZXJjZWl2ZS4-"
  76. "dW5odXJyaWVkIGFuZCB3aXNlLCB3ZSBwZXJjZWl2ZS4-"
  77. "dW5odXJyaWVkIGFuZCB3aXNlLCB3ZSBwZXJjZWl2ZS4-"
  78. "dW5odXJyaWVkIGFuZCB3aXNlLCB3ZSBwZXJjZWl2ZS4"
  79. ".z");
  80. done:
  81. smartlist_free(digests);
  82. }
  83. static int count = 0;
  84. static void
  85. mock_initiate_descriptor_downloads(const routerstatus_t *source,
  86. int purpose, smartlist_t *digests,
  87. int lo, int hi, int pds_flags)
  88. {
  89. (void)source;
  90. (void)purpose;
  91. (void)digests;
  92. (void)pds_flags;
  93. (void)hi;
  94. (void)lo;
  95. count += 1;
  96. }
  97. static void
  98. test_routerlist_launch_descriptor_downloads(void *arg)
  99. {
  100. smartlist_t *downloadable = smartlist_new();
  101. time_t now = time(NULL);
  102. char *cp;
  103. (void)arg;
  104. for (int i = 0; i < 100; i++) {
  105. cp = tor_malloc(DIGEST256_LEN);
  106. tt_assert(cp);
  107. crypto_rand(cp, DIGEST256_LEN);
  108. smartlist_add(downloadable, cp);
  109. }
  110. MOCK(initiate_descriptor_downloads, mock_initiate_descriptor_downloads);
  111. launch_descriptor_downloads(DIR_PURPOSE_FETCH_MICRODESC, downloadable,
  112. NULL, now);
  113. tt_int_op(3, OP_EQ, count);
  114. UNMOCK(initiate_descriptor_downloads);
  115. done:
  116. SMARTLIST_FOREACH(downloadable, char *, cp1, tor_free(cp1));
  117. smartlist_free(downloadable);
  118. }
  119. void
  120. construct_consensus(char **consensus_text_md)
  121. {
  122. networkstatus_t *vote = NULL;
  123. networkstatus_t *v1 = NULL, *v2 = NULL, *v3 = NULL;
  124. networkstatus_voter_info_t *voter = NULL;
  125. authority_cert_t *cert1=NULL, *cert2=NULL, *cert3=NULL;
  126. crypto_pk_t *sign_skey_1=NULL, *sign_skey_2=NULL, *sign_skey_3=NULL;
  127. crypto_pk_t *sign_skey_leg=NULL;
  128. time_t now = time(NULL);
  129. smartlist_t *votes = NULL;
  130. int n_vrs;
  131. tt_assert(!dir_common_authority_pk_init(&cert1, &cert2, &cert3,
  132. &sign_skey_1, &sign_skey_2,
  133. &sign_skey_3));
  134. sign_skey_leg = pk_generate(4);
  135. dir_common_construct_vote_1(&vote, cert1, sign_skey_1,
  136. &dir_common_gen_routerstatus_for_v3ns,
  137. &v1, &n_vrs, now, 1);
  138. networkstatus_vote_free(vote);
  139. tt_assert(v1);
  140. tt_int_op(n_vrs, OP_EQ, 4);
  141. tt_int_op(smartlist_len(v1->routerstatus_list), OP_EQ, 4);
  142. dir_common_construct_vote_2(&vote, cert2, sign_skey_2,
  143. &dir_common_gen_routerstatus_for_v3ns,
  144. &v2, &n_vrs, now, 1);
  145. networkstatus_vote_free(vote);
  146. tt_assert(v2);
  147. tt_int_op(n_vrs, OP_EQ, 4);
  148. tt_int_op(smartlist_len(v2->routerstatus_list), OP_EQ, 4);
  149. dir_common_construct_vote_3(&vote, cert3, sign_skey_3,
  150. &dir_common_gen_routerstatus_for_v3ns,
  151. &v3, &n_vrs, now, 1);
  152. tt_assert(v3);
  153. tt_int_op(n_vrs, OP_EQ, 4);
  154. tt_int_op(smartlist_len(v3->routerstatus_list), OP_EQ, 4);
  155. networkstatus_vote_free(vote);
  156. votes = smartlist_new();
  157. smartlist_add(votes, v1);
  158. smartlist_add(votes, v2);
  159. smartlist_add(votes, v3);
  160. *consensus_text_md = networkstatus_compute_consensus(votes, 3,
  161. cert1->identity_key,
  162. sign_skey_1,
  163. "AAAAAAAAAAAAAAAAAAAA",
  164. sign_skey_leg,
  165. FLAV_MICRODESC);
  166. tt_assert(*consensus_text_md);
  167. done:
  168. tor_free(voter);
  169. networkstatus_vote_free(v1);
  170. networkstatus_vote_free(v2);
  171. networkstatus_vote_free(v3);
  172. smartlist_free(votes);
  173. authority_cert_free(cert1);
  174. authority_cert_free(cert2);
  175. authority_cert_free(cert3);
  176. crypto_pk_free(sign_skey_1);
  177. crypto_pk_free(sign_skey_2);
  178. crypto_pk_free(sign_skey_3);
  179. crypto_pk_free(sign_skey_leg);
  180. }
  181. static int mock_usable_consensus_flavor_value = FLAV_NS;
  182. static int
  183. mock_usable_consensus_flavor(void)
  184. {
  185. return mock_usable_consensus_flavor_value;
  186. }
  187. static void
  188. test_router_pick_directory_server_impl(void *arg)
  189. {
  190. (void)arg;
  191. networkstatus_t *con_md = NULL;
  192. char *consensus_text_md = NULL;
  193. int flags = PDS_IGNORE_FASCISTFIREWALL|PDS_RETRY_IF_NO_SERVERS;
  194. or_options_t *options = get_options_mutable();
  195. const routerstatus_t *rs = NULL;
  196. options->UseMicrodescriptors = 1;
  197. char *router1_id = NULL, *router2_id = NULL, *router3_id = NULL;
  198. node_t *node_router1 = NULL, *node_router2 = NULL, *node_router3 = NULL;
  199. config_line_t *policy_line = NULL;
  200. time_t now = time(NULL);
  201. int tmp_dirport1, tmp_dirport3;
  202. (void)arg;
  203. MOCK(usable_consensus_flavor, mock_usable_consensus_flavor);
  204. /* With no consensus, we must be bootstrapping, regardless of time or flavor
  205. */
  206. mock_usable_consensus_flavor_value = FLAV_NS;
  207. tt_assert(networkstatus_consensus_is_bootstrapping(now));
  208. tt_assert(networkstatus_consensus_is_bootstrapping(now + 2000));
  209. tt_assert(networkstatus_consensus_is_bootstrapping(now + 2*24*60*60));
  210. tt_assert(networkstatus_consensus_is_bootstrapping(now - 2*24*60*60));
  211. mock_usable_consensus_flavor_value = FLAV_MICRODESC;
  212. tt_assert(networkstatus_consensus_is_bootstrapping(now));
  213. tt_assert(networkstatus_consensus_is_bootstrapping(now + 2000));
  214. tt_assert(networkstatus_consensus_is_bootstrapping(now + 2*24*60*60));
  215. tt_assert(networkstatus_consensus_is_bootstrapping(now - 2*24*60*60));
  216. /* Init SR subsystem. */
  217. MOCK(get_my_v3_authority_cert, get_my_v3_authority_cert_m);
  218. mock_cert = authority_cert_parse_from_string(AUTHORITY_CERT_1, NULL);
  219. sr_init(0);
  220. UNMOCK(get_my_v3_authority_cert);
  221. /* No consensus available, fail early */
  222. rs = router_pick_directory_server_impl(V3_DIRINFO, (const int) 0, NULL);
  223. tt_ptr_op(rs, OP_EQ, NULL);
  224. construct_consensus(&consensus_text_md);
  225. tt_assert(consensus_text_md);
  226. con_md = networkstatus_parse_vote_from_string(consensus_text_md, NULL,
  227. NS_TYPE_CONSENSUS);
  228. tt_assert(con_md);
  229. tt_int_op(con_md->flavor,OP_EQ, FLAV_MICRODESC);
  230. tt_assert(con_md->routerstatus_list);
  231. tt_int_op(smartlist_len(con_md->routerstatus_list), OP_EQ, 3);
  232. tt_assert(!networkstatus_set_current_consensus_from_ns(con_md,
  233. "microdesc"));
  234. /* If the consensus time or flavor doesn't match, we are still
  235. * bootstrapping */
  236. mock_usable_consensus_flavor_value = FLAV_NS;
  237. tt_assert(networkstatus_consensus_is_bootstrapping(now));
  238. tt_assert(networkstatus_consensus_is_bootstrapping(now + 2000));
  239. tt_assert(networkstatus_consensus_is_bootstrapping(now + 2*24*60*60));
  240. tt_assert(networkstatus_consensus_is_bootstrapping(now - 2*24*60*60));
  241. /* With a valid consensus for the current time and flavor, we stop
  242. * bootstrapping, even if we have no certificates */
  243. mock_usable_consensus_flavor_value = FLAV_MICRODESC;
  244. tt_assert(!networkstatus_consensus_is_bootstrapping(now + 2000));
  245. tt_assert(!networkstatus_consensus_is_bootstrapping(con_md->valid_after));
  246. tt_assert(!networkstatus_consensus_is_bootstrapping(con_md->valid_until));
  247. tt_assert(!networkstatus_consensus_is_bootstrapping(con_md->valid_until
  248. + 24*60*60));
  249. /* These times are outside the test validity period */
  250. tt_assert(networkstatus_consensus_is_bootstrapping(now));
  251. tt_assert(networkstatus_consensus_is_bootstrapping(now + 2*24*60*60));
  252. tt_assert(networkstatus_consensus_is_bootstrapping(now - 2*24*60*60));
  253. nodelist_set_consensus(con_md);
  254. nodelist_assert_ok();
  255. rs = router_pick_directory_server_impl(V3_DIRINFO, flags, NULL);
  256. /* We should not fail now we have a consensus and routerstatus_list
  257. * and nodelist are populated. */
  258. tt_ptr_op(rs, OP_NE, NULL);
  259. /* Manipulate the nodes so we get the dir server we expect */
  260. router1_id = tor_malloc(DIGEST_LEN);
  261. memset(router1_id, TEST_DIR_ROUTER_ID_1, DIGEST_LEN);
  262. router2_id = tor_malloc(DIGEST_LEN);
  263. memset(router2_id, TEST_DIR_ROUTER_ID_2, DIGEST_LEN);
  264. router3_id = tor_malloc(DIGEST_LEN);
  265. memset(router3_id, TEST_DIR_ROUTER_ID_3, DIGEST_LEN);
  266. node_router1 = node_get_mutable_by_id(router1_id);
  267. node_router2 = node_get_mutable_by_id(router2_id);
  268. node_router3 = node_get_mutable_by_id(router3_id);
  269. node_router1->is_possible_guard = 1;
  270. node_router1->is_running = 0;
  271. node_router3->is_running = 0;
  272. rs = router_pick_directory_server_impl(V3_DIRINFO, flags, NULL);
  273. tt_ptr_op(rs, OP_NE, NULL);
  274. tt_assert(tor_memeq(rs->identity_digest, router2_id, DIGEST_LEN));
  275. rs = NULL;
  276. node_router1->is_running = 1;
  277. node_router3->is_running = 1;
  278. node_router1->rs->is_v2_dir = 0;
  279. node_router3->rs->is_v2_dir = 0;
  280. tmp_dirport1 = node_router1->rs->dir_port;
  281. tmp_dirport3 = node_router3->rs->dir_port;
  282. node_router1->rs->dir_port = 0;
  283. node_router3->rs->dir_port = 0;
  284. rs = router_pick_directory_server_impl(V3_DIRINFO, flags, NULL);
  285. tt_ptr_op(rs, OP_NE, NULL);
  286. tt_assert(tor_memeq(rs->identity_digest, router2_id, DIGEST_LEN));
  287. rs = NULL;
  288. node_router1->rs->is_v2_dir = 1;
  289. node_router3->rs->is_v2_dir = 1;
  290. node_router1->rs->dir_port = tmp_dirport1;
  291. node_router3->rs->dir_port = tmp_dirport3;
  292. node_router1->is_valid = 0;
  293. node_router3->is_valid = 0;
  294. rs = router_pick_directory_server_impl(V3_DIRINFO, flags, NULL);
  295. tt_ptr_op(rs, OP_NE, NULL);
  296. tt_assert(tor_memeq(rs->identity_digest, router2_id, DIGEST_LEN));
  297. rs = NULL;
  298. node_router1->is_valid = 1;
  299. node_router3->is_valid = 1;
  300. /* Manipulate overloaded */
  301. node_router2->rs->last_dir_503_at = now;
  302. node_router3->rs->last_dir_503_at = now;
  303. rs = router_pick_directory_server_impl(V3_DIRINFO, flags, NULL);
  304. tt_ptr_op(rs, OP_NE, NULL);
  305. tt_assert(tor_memeq(rs->identity_digest, router1_id, DIGEST_LEN));
  306. node_router2->rs->last_dir_503_at = 0;
  307. node_router3->rs->last_dir_503_at = 0;
  308. /* Set a Fascist firewall */
  309. flags &= ~ PDS_IGNORE_FASCISTFIREWALL;
  310. policy_line = tor_malloc_zero(sizeof(config_line_t));
  311. policy_line->key = tor_strdup("ReachableORAddresses");
  312. policy_line->value = tor_strdup("accept *:442, reject *:*");
  313. options->ReachableORAddresses = policy_line;
  314. policies_parse_from_options(options);
  315. node_router1->rs->or_port = 444;
  316. node_router2->rs->or_port = 443;
  317. node_router3->rs->or_port = 442;
  318. rs = router_pick_directory_server_impl(V3_DIRINFO, flags, NULL);
  319. tt_ptr_op(rs, OP_NE, NULL);
  320. tt_assert(tor_memeq(rs->identity_digest, router3_id, DIGEST_LEN));
  321. node_router1->rs->or_port = 442;
  322. node_router2->rs->or_port = 443;
  323. node_router3->rs->or_port = 444;
  324. rs = router_pick_directory_server_impl(V3_DIRINFO, flags, NULL);
  325. tt_ptr_op(rs, OP_NE, NULL);
  326. tt_assert(tor_memeq(rs->identity_digest, router1_id, DIGEST_LEN));
  327. /* Fascist firewall and overloaded */
  328. node_router1->rs->or_port = 442;
  329. node_router2->rs->or_port = 443;
  330. node_router3->rs->or_port = 442;
  331. node_router3->rs->last_dir_503_at = now;
  332. rs = router_pick_directory_server_impl(V3_DIRINFO, flags, NULL);
  333. tt_ptr_op(rs, OP_NE, NULL);
  334. tt_assert(tor_memeq(rs->identity_digest, router1_id, DIGEST_LEN));
  335. node_router3->rs->last_dir_503_at = 0;
  336. /* Fascists against OR and Dir */
  337. policy_line = tor_malloc_zero(sizeof(config_line_t));
  338. policy_line->key = tor_strdup("ReachableAddresses");
  339. policy_line->value = tor_strdup("accept *:80, reject *:*");
  340. options->ReachableDirAddresses = policy_line;
  341. policies_parse_from_options(options);
  342. node_router1->rs->or_port = 442;
  343. node_router2->rs->or_port = 441;
  344. node_router3->rs->or_port = 443;
  345. node_router1->rs->dir_port = 80;
  346. node_router2->rs->dir_port = 80;
  347. node_router3->rs->dir_port = 81;
  348. node_router1->rs->last_dir_503_at = now;
  349. rs = router_pick_directory_server_impl(V3_DIRINFO, flags, NULL);
  350. tt_ptr_op(rs, OP_NE, NULL);
  351. tt_assert(tor_memeq(rs->identity_digest, router1_id, DIGEST_LEN));
  352. node_router1->rs->last_dir_503_at = 0;
  353. done:
  354. UNMOCK(usable_consensus_flavor);
  355. if (router1_id)
  356. tor_free(router1_id);
  357. if (router2_id)
  358. tor_free(router2_id);
  359. if (router3_id)
  360. tor_free(router3_id);
  361. if (options->ReachableORAddresses ||
  362. options->ReachableDirAddresses)
  363. policies_free_all();
  364. tor_free(consensus_text_md);
  365. networkstatus_vote_free(con_md);
  366. }
  367. static or_state_t *dummy_state = NULL;
  368. static or_state_t *
  369. get_or_state_replacement(void)
  370. {
  371. return dummy_state;
  372. }
  373. static void
  374. mock_directory_initiate_request(directory_request_t *req)
  375. {
  376. (void)req;
  377. return;
  378. }
  379. static circuit_guard_state_t *
  380. mock_circuit_guard_state_new(entry_guard_t *guard, unsigned state,
  381. entry_guard_restriction_t *rst)
  382. {
  383. (void) guard;
  384. (void) state;
  385. (void) rst;
  386. return NULL;
  387. }
  388. /** Test that we will use our directory guards to fetch mds even if we don't
  389. * have any dirinfo (tests bug #23862). */
  390. static void
  391. test_directory_guard_fetch_with_no_dirinfo(void *arg)
  392. {
  393. int retval;
  394. char *consensus_text_md = NULL;
  395. or_options_t *options = get_options_mutable();
  396. (void) arg;
  397. hibernate_set_state_for_testing_(HIBERNATE_STATE_LIVE);
  398. /* Initialize the SRV subsystem */
  399. MOCK(get_my_v3_authority_cert, get_my_v3_authority_cert_m);
  400. mock_cert = authority_cert_parse_from_string(AUTHORITY_CERT_1, NULL);
  401. sr_init(0);
  402. UNMOCK(get_my_v3_authority_cert);
  403. /* Initialize the entry node configuration from the ticket */
  404. options->UseEntryGuards = 1;
  405. options->StrictNodes = 1;
  406. get_options_mutable()->EntryNodes = routerset_new();
  407. routerset_parse(get_options_mutable()->EntryNodes,
  408. "2121212121212121212121212121212121212121", "foo");
  409. /* Mock some functions */
  410. dummy_state = tor_malloc_zero(sizeof(or_state_t));
  411. MOCK(get_or_state, get_or_state_replacement);
  412. MOCK(directory_initiate_request, mock_directory_initiate_request);
  413. /* we need to mock this one to avoid memleaks */
  414. MOCK(circuit_guard_state_new, mock_circuit_guard_state_new);
  415. /* Call guards_update_all() to simulate loading our state file (see
  416. * entry_guards_load_guards_from_state() and ticket #23989). */
  417. guards_update_all();
  418. /* Test logic: Simulate the arrival of a new consensus when we have no
  419. * dirinfo at all. Tor will need to fetch the mds from the consensus. Make
  420. * sure that Tor will use the specified entry guard instead of relying on the
  421. * fallback directories. */
  422. /* Fixup the dirconn that will deliver the consensus */
  423. dir_connection_t *conn = dir_connection_new(AF_INET);
  424. tor_addr_from_ipv4h(&conn->base_.addr, 0x7f000001);
  425. conn->base_.port = 8800;
  426. TO_CONN(conn)->address = tor_strdup("127.0.0.1");
  427. conn->base_.purpose = DIR_PURPOSE_FETCH_CONSENSUS;
  428. conn->requested_resource = tor_strdup("ns");
  429. /* Construct a consensus */
  430. construct_consensus(&consensus_text_md);
  431. tt_assert(consensus_text_md);
  432. /* Place the consensus in the dirconn */
  433. response_handler_args_t args;
  434. memset(&args, 0, sizeof(response_handler_args_t));
  435. args.status_code = 200;
  436. args.body = consensus_text_md;
  437. args.body_len = strlen(consensus_text_md);
  438. /* Update approx time so that the consensus is considered live */
  439. update_approx_time(time(NULL)+1010);
  440. setup_capture_of_logs(LOG_DEBUG);
  441. /* Now handle the consensus */
  442. retval = handle_response_fetch_consensus(conn, &args);
  443. tt_int_op(retval, OP_EQ, 0);
  444. /* Make sure that our primary guard was chosen */
  445. expect_log_msg_containing("Selected primary guard router3");
  446. done:
  447. tor_free(consensus_text_md);
  448. tor_free(dummy_state);
  449. connection_free_minimal(TO_CONN(conn));
  450. entry_guards_free_all();
  451. teardown_capture_of_logs();
  452. }
  453. static connection_t *mocked_connection = NULL;
  454. /* Mock connection_get_by_type_addr_port_purpose by returning
  455. * mocked_connection. */
  456. static connection_t *
  457. mock_connection_get_by_type_addr_port_purpose(int type,
  458. const tor_addr_t *addr,
  459. uint16_t port, int purpose)
  460. {
  461. (void)type;
  462. (void)addr;
  463. (void)port;
  464. (void)purpose;
  465. return mocked_connection;
  466. }
  467. #define TEST_ADDR_STR "127.0.0.1"
  468. #define TEST_DIR_PORT 12345
  469. static void
  470. test_routerlist_router_is_already_dir_fetching(void *arg)
  471. {
  472. (void)arg;
  473. tor_addr_port_t test_ap, null_addr_ap, zero_port_ap;
  474. /* Setup */
  475. tor_addr_parse(&test_ap.addr, TEST_ADDR_STR);
  476. test_ap.port = TEST_DIR_PORT;
  477. tor_addr_make_null(&null_addr_ap.addr, AF_INET6);
  478. null_addr_ap.port = TEST_DIR_PORT;
  479. tor_addr_parse(&zero_port_ap.addr, TEST_ADDR_STR);
  480. zero_port_ap.port = 0;
  481. MOCK(connection_get_by_type_addr_port_purpose,
  482. mock_connection_get_by_type_addr_port_purpose);
  483. /* Test that we never get 1 from a NULL connection */
  484. mocked_connection = NULL;
  485. tt_int_op(router_is_already_dir_fetching(&test_ap, 1, 1), OP_EQ, 0);
  486. tt_int_op(router_is_already_dir_fetching(&test_ap, 1, 0), OP_EQ, 0);
  487. tt_int_op(router_is_already_dir_fetching(&test_ap, 0, 1), OP_EQ, 0);
  488. /* We always expect 0 in these cases */
  489. tt_int_op(router_is_already_dir_fetching(&test_ap, 0, 0), OP_EQ, 0);
  490. tt_int_op(router_is_already_dir_fetching(NULL, 1, 1), OP_EQ, 0);
  491. tt_int_op(router_is_already_dir_fetching(&null_addr_ap, 1, 1), OP_EQ, 0);
  492. tt_int_op(router_is_already_dir_fetching(&zero_port_ap, 1, 1), OP_EQ, 0);
  493. /* Test that we get 1 with a connection in the appropriate circumstances */
  494. mocked_connection = connection_new(CONN_TYPE_DIR, AF_INET);
  495. tt_int_op(router_is_already_dir_fetching(&test_ap, 1, 1), OP_EQ, 1);
  496. tt_int_op(router_is_already_dir_fetching(&test_ap, 1, 0), OP_EQ, 1);
  497. tt_int_op(router_is_already_dir_fetching(&test_ap, 0, 1), OP_EQ, 1);
  498. /* Test that we get 0 even with a connection in the appropriate
  499. * circumstances */
  500. tt_int_op(router_is_already_dir_fetching(&test_ap, 0, 0), OP_EQ, 0);
  501. tt_int_op(router_is_already_dir_fetching(NULL, 1, 1), OP_EQ, 0);
  502. tt_int_op(router_is_already_dir_fetching(&null_addr_ap, 1, 1), OP_EQ, 0);
  503. tt_int_op(router_is_already_dir_fetching(&zero_port_ap, 1, 1), OP_EQ, 0);
  504. done:
  505. /* If a connection is never set up, connection_free chokes on it. */
  506. if (mocked_connection) {
  507. buf_free(mocked_connection->inbuf);
  508. buf_free(mocked_connection->outbuf);
  509. }
  510. tor_free(mocked_connection);
  511. UNMOCK(connection_get_by_type_addr_port_purpose);
  512. }
  513. #undef TEST_ADDR_STR
  514. #undef TEST_DIR_PORT
  515. #define NODE(name, flags) \
  516. { #name, test_routerlist_##name, (flags), NULL, NULL }
  517. #define ROUTER(name,flags) \
  518. { #name, test_router_##name, (flags), NULL, NULL }
  519. struct testcase_t routerlist_tests[] = {
  520. NODE(initiate_descriptor_downloads, 0),
  521. NODE(launch_descriptor_downloads, 0),
  522. NODE(router_is_already_dir_fetching, TT_FORK),
  523. ROUTER(pick_directory_server_impl, TT_FORK),
  524. { "directory_guard_fetch_with_no_dirinfo",
  525. test_directory_guard_fetch_with_no_dirinfo, TT_FORK, NULL, NULL },
  526. END_OF_TESTCASES
  527. };