test_controller.c 40 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295
  1. /* Copyright (c) 2015-2016, The Tor Project, Inc. */
  2. /* See LICENSE for licensing information */
  3. #define CONTROL_PRIVATE
  4. #include "or.h"
  5. #include "control.h"
  6. #include "entrynodes.h"
  7. #include "networkstatus.h"
  8. #include "rendservice.h"
  9. #include "routerlist.h"
  10. #include "test.h"
  11. static void
  12. test_add_onion_helper_keyarg(void *arg)
  13. {
  14. crypto_pk_t *pk = NULL;
  15. crypto_pk_t *pk2 = NULL;
  16. const char *key_new_alg = NULL;
  17. char *key_new_blob = NULL;
  18. char *err_msg = NULL;
  19. char *encoded = NULL;
  20. char *arg_str = NULL;
  21. (void) arg;
  22. /* Test explicit RSA1024 key generation. */
  23. pk = add_onion_helper_keyarg("NEW:RSA1024", 0, &key_new_alg, &key_new_blob,
  24. &err_msg);
  25. tt_assert(pk);
  26. tt_str_op(key_new_alg, OP_EQ, "RSA1024");
  27. tt_assert(key_new_blob);
  28. tt_assert(!err_msg);
  29. /* Test "BEST" key generation (Assumes BEST = RSA1024). */
  30. crypto_pk_free(pk);
  31. tor_free(key_new_blob);
  32. pk = add_onion_helper_keyarg("NEW:BEST", 0, &key_new_alg, &key_new_blob,
  33. &err_msg);
  34. tt_assert(pk);
  35. tt_str_op(key_new_alg, OP_EQ, "RSA1024");
  36. tt_assert(key_new_blob);
  37. tt_assert(!err_msg);
  38. /* Test discarding the private key. */
  39. crypto_pk_free(pk);
  40. tor_free(key_new_blob);
  41. pk = add_onion_helper_keyarg("NEW:BEST", 1, &key_new_alg, &key_new_blob,
  42. &err_msg);
  43. tt_assert(pk);
  44. tt_assert(!key_new_alg);
  45. tt_assert(!key_new_blob);
  46. tt_assert(!err_msg);
  47. /* Test generating a invalid key type. */
  48. crypto_pk_free(pk);
  49. pk = add_onion_helper_keyarg("NEW:RSA512", 0, &key_new_alg, &key_new_blob,
  50. &err_msg);
  51. tt_assert(!pk);
  52. tt_assert(!key_new_alg);
  53. tt_assert(!key_new_blob);
  54. tt_assert(err_msg);
  55. /* Test loading a RSA1024 key. */
  56. tor_free(err_msg);
  57. pk = pk_generate(0);
  58. tt_int_op(0, OP_EQ, crypto_pk_base64_encode(pk, &encoded));
  59. tor_asprintf(&arg_str, "RSA1024:%s", encoded);
  60. pk2 = add_onion_helper_keyarg(arg_str, 0, &key_new_alg, &key_new_blob,
  61. &err_msg);
  62. tt_assert(pk2);
  63. tt_assert(!key_new_alg);
  64. tt_assert(!key_new_blob);
  65. tt_assert(!err_msg);
  66. tt_assert(crypto_pk_cmp_keys(pk, pk2) == 0);
  67. /* Test loading a invalid key type. */
  68. tor_free(arg_str);
  69. crypto_pk_free(pk); pk = NULL;
  70. tor_asprintf(&arg_str, "RSA512:%s", encoded);
  71. pk = add_onion_helper_keyarg(arg_str, 0, &key_new_alg, &key_new_blob,
  72. &err_msg);
  73. tt_assert(!pk);
  74. tt_assert(!key_new_alg);
  75. tt_assert(!key_new_blob);
  76. tt_assert(err_msg);
  77. /* Test loading a invalid key. */
  78. tor_free(arg_str);
  79. crypto_pk_free(pk); pk = NULL;
  80. tor_free(err_msg);
  81. encoded[strlen(encoded)/2] = '\0';
  82. tor_asprintf(&arg_str, "RSA1024:%s", encoded);
  83. pk = add_onion_helper_keyarg(arg_str, 0, &key_new_alg, &key_new_blob,
  84. &err_msg);
  85. tt_assert(!pk);
  86. tt_assert(!key_new_alg);
  87. tt_assert(!key_new_blob);
  88. tt_assert(err_msg);
  89. done:
  90. crypto_pk_free(pk);
  91. crypto_pk_free(pk2);
  92. tor_free(key_new_blob);
  93. tor_free(err_msg);
  94. tor_free(encoded);
  95. tor_free(arg_str);
  96. }
  97. static void
  98. test_rend_service_parse_port_config(void *arg)
  99. {
  100. const char *sep = ",";
  101. rend_service_port_config_t *cfg = NULL;
  102. char *err_msg = NULL;
  103. (void)arg;
  104. /* Test "VIRTPORT" only. */
  105. cfg = rend_service_parse_port_config("80", sep, &err_msg);
  106. tt_assert(cfg);
  107. tt_assert(!err_msg);
  108. /* Test "VIRTPORT,TARGET" (Target is port). */
  109. rend_service_port_config_free(cfg);
  110. cfg = rend_service_parse_port_config("80,8080", sep, &err_msg);
  111. tt_assert(cfg);
  112. tt_assert(!err_msg);
  113. /* Test "VIRTPORT,TARGET" (Target is IPv4:port). */
  114. rend_service_port_config_free(cfg);
  115. cfg = rend_service_parse_port_config("80,192.0.2.1:8080", sep, &err_msg);
  116. tt_assert(cfg);
  117. tt_assert(!err_msg);
  118. /* Test "VIRTPORT,TARGET" (Target is IPv6:port). */
  119. rend_service_port_config_free(cfg);
  120. cfg = rend_service_parse_port_config("80,[2001:db8::1]:8080", sep, &err_msg);
  121. tt_assert(cfg);
  122. tt_assert(!err_msg);
  123. /* XXX: Someone should add tests for AF_UNIX targets if supported. */
  124. /* Test empty config. */
  125. rend_service_port_config_free(cfg);
  126. cfg = rend_service_parse_port_config("", sep, &err_msg);
  127. tt_assert(!cfg);
  128. tt_assert(err_msg);
  129. /* Test invalid port. */
  130. tor_free(err_msg);
  131. cfg = rend_service_parse_port_config("90001", sep, &err_msg);
  132. tt_assert(!cfg);
  133. tt_assert(err_msg);
  134. done:
  135. rend_service_port_config_free(cfg);
  136. tor_free(err_msg);
  137. }
  138. static void
  139. test_add_onion_helper_clientauth(void *arg)
  140. {
  141. rend_authorized_client_t *client = NULL;
  142. char *err_msg = NULL;
  143. int created = 0;
  144. (void)arg;
  145. /* Test "ClientName" only. */
  146. client = add_onion_helper_clientauth("alice", &created, &err_msg);
  147. tt_assert(client);
  148. tt_assert(created);
  149. tt_assert(!err_msg);
  150. rend_authorized_client_free(client);
  151. /* Test "ClientName:Blob" */
  152. client = add_onion_helper_clientauth("alice:475hGBHPlq7Mc0cRZitK/B",
  153. &created, &err_msg);
  154. tt_assert(client);
  155. tt_assert(!created);
  156. tt_assert(!err_msg);
  157. rend_authorized_client_free(client);
  158. /* Test invalid client names */
  159. client = add_onion_helper_clientauth("no*asterisks*allowed", &created,
  160. &err_msg);
  161. tt_assert(!client);
  162. tt_assert(err_msg);
  163. tor_free(err_msg);
  164. /* Test invalid auth cookie */
  165. client = add_onion_helper_clientauth("alice:12345", &created, &err_msg);
  166. tt_assert(!client);
  167. tt_assert(err_msg);
  168. tor_free(err_msg);
  169. /* Test invalid syntax */
  170. client = add_onion_helper_clientauth(":475hGBHPlq7Mc0cRZitK/B", &created,
  171. &err_msg);
  172. tt_assert(!client);
  173. tt_assert(err_msg);
  174. tor_free(err_msg);
  175. done:
  176. rend_authorized_client_free(client);
  177. tor_free(err_msg);
  178. }
  179. /* Mocks and data/variables used for GETINFO download status tests */
  180. static const download_status_t dl_status_default =
  181. { 0, 0, 0, DL_SCHED_CONSENSUS, DL_WANT_ANY_DIRSERVER,
  182. DL_SCHED_INCREMENT_FAILURE, DL_SCHED_RANDOM_EXPONENTIAL, 0, 0 };
  183. static download_status_t ns_dl_status[N_CONSENSUS_FLAVORS];
  184. static download_status_t ns_dl_status_bootstrap[N_CONSENSUS_FLAVORS];
  185. static download_status_t ns_dl_status_running[N_CONSENSUS_FLAVORS];
  186. /*
  187. * These should explore all the possible cases of download_status_to_string()
  188. * in control.c
  189. */
  190. static const download_status_t dls_sample_1 =
  191. { 1467163900, 0, 0, DL_SCHED_GENERIC, DL_WANT_ANY_DIRSERVER,
  192. DL_SCHED_INCREMENT_FAILURE, DL_SCHED_DETERMINISTIC, 0, 0 };
  193. static const char * dls_sample_1_str =
  194. "next-attempt-at 2016-06-29 01:31:40\n"
  195. "n-download-failures 0\n"
  196. "n-download-attempts 0\n"
  197. "schedule DL_SCHED_GENERIC\n"
  198. "want-authority DL_WANT_ANY_DIRSERVER\n"
  199. "increment-on DL_SCHED_INCREMENT_FAILURE\n"
  200. "backoff DL_SCHED_DETERMINISTIC\n";
  201. static const download_status_t dls_sample_2 =
  202. { 1467164400, 1, 2, DL_SCHED_CONSENSUS, DL_WANT_AUTHORITY,
  203. DL_SCHED_INCREMENT_FAILURE, DL_SCHED_DETERMINISTIC, 0, 0 };
  204. static const char * dls_sample_2_str =
  205. "next-attempt-at 2016-06-29 01:40:00\n"
  206. "n-download-failures 1\n"
  207. "n-download-attempts 2\n"
  208. "schedule DL_SCHED_CONSENSUS\n"
  209. "want-authority DL_WANT_AUTHORITY\n"
  210. "increment-on DL_SCHED_INCREMENT_FAILURE\n"
  211. "backoff DL_SCHED_DETERMINISTIC\n";
  212. static const download_status_t dls_sample_3 =
  213. { 1467154400, 12, 25, DL_SCHED_BRIDGE, DL_WANT_ANY_DIRSERVER,
  214. DL_SCHED_INCREMENT_ATTEMPT, DL_SCHED_DETERMINISTIC, 0, 0 };
  215. static const char * dls_sample_3_str =
  216. "next-attempt-at 2016-06-28 22:53:20\n"
  217. "n-download-failures 12\n"
  218. "n-download-attempts 25\n"
  219. "schedule DL_SCHED_BRIDGE\n"
  220. "want-authority DL_WANT_ANY_DIRSERVER\n"
  221. "increment-on DL_SCHED_INCREMENT_ATTEMPT\n"
  222. "backoff DL_SCHED_DETERMINISTIC\n";
  223. static const download_status_t dls_sample_4 =
  224. { 1467166600, 3, 0, DL_SCHED_GENERIC, DL_WANT_ANY_DIRSERVER,
  225. DL_SCHED_INCREMENT_FAILURE, DL_SCHED_RANDOM_EXPONENTIAL, 0, 0 };
  226. static const char * dls_sample_4_str =
  227. "next-attempt-at 2016-06-29 02:16:40\n"
  228. "n-download-failures 3\n"
  229. "n-download-attempts 0\n"
  230. "schedule DL_SCHED_GENERIC\n"
  231. "want-authority DL_WANT_ANY_DIRSERVER\n"
  232. "increment-on DL_SCHED_INCREMENT_FAILURE\n"
  233. "backoff DL_SCHED_RANDOM_EXPONENTIAL\n"
  234. "last-backoff-position 0\n"
  235. "last-delay-used 0\n";
  236. static const download_status_t dls_sample_5 =
  237. { 1467164600, 3, 7, DL_SCHED_CONSENSUS, DL_WANT_ANY_DIRSERVER,
  238. DL_SCHED_INCREMENT_FAILURE, DL_SCHED_RANDOM_EXPONENTIAL, 1, 2112, };
  239. static const char * dls_sample_5_str =
  240. "next-attempt-at 2016-06-29 01:43:20\n"
  241. "n-download-failures 3\n"
  242. "n-download-attempts 7\n"
  243. "schedule DL_SCHED_CONSENSUS\n"
  244. "want-authority DL_WANT_ANY_DIRSERVER\n"
  245. "increment-on DL_SCHED_INCREMENT_FAILURE\n"
  246. "backoff DL_SCHED_RANDOM_EXPONENTIAL\n"
  247. "last-backoff-position 1\n"
  248. "last-delay-used 2112\n";
  249. static const download_status_t dls_sample_6 =
  250. { 1467164200, 4, 9, DL_SCHED_CONSENSUS, DL_WANT_AUTHORITY,
  251. DL_SCHED_INCREMENT_ATTEMPT, DL_SCHED_RANDOM_EXPONENTIAL, 3, 432 };
  252. static const char * dls_sample_6_str =
  253. "next-attempt-at 2016-06-29 01:36:40\n"
  254. "n-download-failures 4\n"
  255. "n-download-attempts 9\n"
  256. "schedule DL_SCHED_CONSENSUS\n"
  257. "want-authority DL_WANT_AUTHORITY\n"
  258. "increment-on DL_SCHED_INCREMENT_ATTEMPT\n"
  259. "backoff DL_SCHED_RANDOM_EXPONENTIAL\n"
  260. "last-backoff-position 3\n"
  261. "last-delay-used 432\n";
  262. /* Simulated auth certs */
  263. static const char *auth_id_digest_1_str =
  264. "63CDD326DFEF0CA020BDD3FEB45A3286FE13A061";
  265. static download_status_t auth_def_cert_download_status_1;
  266. static const char *auth_id_digest_2_str =
  267. "2C209FCDD8D48DC049777B8DC2C0F94A0408BE99";
  268. static download_status_t auth_def_cert_download_status_2;
  269. /* Expected form of digest list returned for GETINFO downloads/cert/fps */
  270. static const char *auth_id_digest_expected_list =
  271. "63CDD326DFEF0CA020BDD3FEB45A3286FE13A061\n"
  272. "2C209FCDD8D48DC049777B8DC2C0F94A0408BE99\n";
  273. /* Signing keys for simulated auth 1 */
  274. static const char *auth_1_sk_1_str =
  275. "AA69566029B1F023BA09451B8F1B10952384EB58";
  276. static download_status_t auth_1_sk_1_dls;
  277. static const char *auth_1_sk_2_str =
  278. "710865C7F06B73C5292695A8C34F1C94F769FF72";
  279. static download_status_t auth_1_sk_2_dls;
  280. /*
  281. * Expected form of sk digest list for
  282. * GETINFO downloads/cert/<auth_id_digest_1_str>/sks
  283. */
  284. static const char *auth_1_sk_digest_expected_list =
  285. "AA69566029B1F023BA09451B8F1B10952384EB58\n"
  286. "710865C7F06B73C5292695A8C34F1C94F769FF72\n";
  287. /* Signing keys for simulated auth 2 */
  288. static const char *auth_2_sk_1_str =
  289. "4299047E00D070AD6703FE00BE7AA756DB061E62";
  290. static download_status_t auth_2_sk_1_dls;
  291. static const char *auth_2_sk_2_str =
  292. "9451B8F1B10952384EB58B5F230C0BB701626C9B";
  293. static download_status_t auth_2_sk_2_dls;
  294. /*
  295. * Expected form of sk digest list for
  296. * GETINFO downloads/cert/<auth_id_digest_2_str>/sks
  297. */
  298. static const char *auth_2_sk_digest_expected_list =
  299. "4299047E00D070AD6703FE00BE7AA756DB061E62\n"
  300. "9451B8F1B10952384EB58B5F230C0BB701626C9B\n";
  301. /* Simulated router descriptor digests or bridge identity digests */
  302. static const char *descbr_digest_1_str =
  303. "616408544C7345822696074A1A3DFA16AB381CBD";
  304. static download_status_t descbr_digest_1_dl;
  305. static const char *descbr_digest_2_str =
  306. "06E8067246967265DBCB6641631B530EFEC12DC3";
  307. static download_status_t descbr_digest_2_dl;
  308. /* Expected form of digest list returned for GETINFO downloads/desc/descs */
  309. static const char *descbr_expected_list =
  310. "616408544C7345822696074A1A3DFA16AB381CBD\n"
  311. "06E8067246967265DBCB6641631B530EFEC12DC3\n";
  312. /*
  313. * Flag to make all descbr queries fail, to simulate not being
  314. * configured such that such queries make sense.
  315. */
  316. static int disable_descbr = 0;
  317. static void
  318. reset_mocked_dl_statuses(void)
  319. {
  320. int i;
  321. for (i = 0; i < N_CONSENSUS_FLAVORS; ++i) {
  322. memcpy(&(ns_dl_status[i]), &dl_status_default,
  323. sizeof(download_status_t));
  324. memcpy(&(ns_dl_status_bootstrap[i]), &dl_status_default,
  325. sizeof(download_status_t));
  326. memcpy(&(ns_dl_status_running[i]), &dl_status_default,
  327. sizeof(download_status_t));
  328. }
  329. memcpy(&auth_def_cert_download_status_1, &dl_status_default,
  330. sizeof(download_status_t));
  331. memcpy(&auth_def_cert_download_status_2, &dl_status_default,
  332. sizeof(download_status_t));
  333. memcpy(&auth_1_sk_1_dls, &dl_status_default,
  334. sizeof(download_status_t));
  335. memcpy(&auth_1_sk_2_dls, &dl_status_default,
  336. sizeof(download_status_t));
  337. memcpy(&auth_2_sk_1_dls, &dl_status_default,
  338. sizeof(download_status_t));
  339. memcpy(&auth_2_sk_2_dls, &dl_status_default,
  340. sizeof(download_status_t));
  341. memcpy(&descbr_digest_1_dl, &dl_status_default,
  342. sizeof(download_status_t));
  343. memcpy(&descbr_digest_2_dl, &dl_status_default,
  344. sizeof(download_status_t));
  345. }
  346. static download_status_t *
  347. ns_dl_status_mock(consensus_flavor_t flavor)
  348. {
  349. return &(ns_dl_status[flavor]);
  350. }
  351. static download_status_t *
  352. ns_dl_status_bootstrap_mock(consensus_flavor_t flavor)
  353. {
  354. return &(ns_dl_status_bootstrap[flavor]);
  355. }
  356. static download_status_t *
  357. ns_dl_status_running_mock(consensus_flavor_t flavor)
  358. {
  359. return &(ns_dl_status_running[flavor]);
  360. }
  361. static void
  362. setup_ns_mocks(void)
  363. {
  364. MOCK(networkstatus_get_dl_status_by_flavor, ns_dl_status_mock);
  365. MOCK(networkstatus_get_dl_status_by_flavor_bootstrap,
  366. ns_dl_status_bootstrap_mock);
  367. MOCK(networkstatus_get_dl_status_by_flavor_running,
  368. ns_dl_status_running_mock);
  369. reset_mocked_dl_statuses();
  370. }
  371. static void
  372. clear_ns_mocks(void)
  373. {
  374. UNMOCK(networkstatus_get_dl_status_by_flavor);
  375. UNMOCK(networkstatus_get_dl_status_by_flavor_bootstrap);
  376. UNMOCK(networkstatus_get_dl_status_by_flavor_running);
  377. }
  378. static smartlist_t *
  379. cert_dl_status_auth_ids_mock(void)
  380. {
  381. char digest[DIGEST_LEN], *tmp;
  382. int len;
  383. smartlist_t *list = NULL;
  384. /* Just pretend we have only the two hard-coded digests listed above */
  385. list = smartlist_new();
  386. len = base16_decode(digest, DIGEST_LEN,
  387. auth_id_digest_1_str, strlen(auth_id_digest_1_str));
  388. tt_int_op(len, OP_EQ, DIGEST_LEN);
  389. tmp = tor_malloc(DIGEST_LEN);
  390. memcpy(tmp, digest, DIGEST_LEN);
  391. smartlist_add(list, tmp);
  392. len = base16_decode(digest, DIGEST_LEN,
  393. auth_id_digest_2_str, strlen(auth_id_digest_2_str));
  394. tt_int_op(len, OP_EQ, DIGEST_LEN);
  395. tmp = tor_malloc(DIGEST_LEN);
  396. memcpy(tmp, digest, DIGEST_LEN);
  397. smartlist_add(list, tmp);
  398. done:
  399. return list;
  400. }
  401. static download_status_t *
  402. cert_dl_status_def_for_auth_mock(const char *digest)
  403. {
  404. download_status_t *dl = NULL;
  405. char digest_str[HEX_DIGEST_LEN+1];
  406. tt_assert(digest != NULL);
  407. base16_encode(digest_str, HEX_DIGEST_LEN + 1,
  408. digest, DIGEST_LEN);
  409. digest_str[HEX_DIGEST_LEN] = '\0';
  410. if (strcmp(digest_str, auth_id_digest_1_str) == 0) {
  411. dl = &auth_def_cert_download_status_1;
  412. } else if (strcmp(digest_str, auth_id_digest_2_str) == 0) {
  413. dl = &auth_def_cert_download_status_2;
  414. }
  415. done:
  416. return dl;
  417. }
  418. static smartlist_t *
  419. cert_dl_status_sks_for_auth_id_mock(const char *digest)
  420. {
  421. smartlist_t *list = NULL;
  422. char sk[DIGEST_LEN];
  423. char digest_str[HEX_DIGEST_LEN+1];
  424. char *tmp;
  425. int len;
  426. tt_assert(digest != NULL);
  427. base16_encode(digest_str, HEX_DIGEST_LEN + 1,
  428. digest, DIGEST_LEN);
  429. digest_str[HEX_DIGEST_LEN] = '\0';
  430. /*
  431. * Build a list of two hard-coded digests, depending on what we
  432. * were just passed.
  433. */
  434. if (strcmp(digest_str, auth_id_digest_1_str) == 0) {
  435. list = smartlist_new();
  436. len = base16_decode(sk, DIGEST_LEN,
  437. auth_1_sk_1_str, strlen(auth_1_sk_1_str));
  438. tt_int_op(len, OP_EQ, DIGEST_LEN);
  439. tmp = tor_malloc(DIGEST_LEN);
  440. memcpy(tmp, sk, DIGEST_LEN);
  441. smartlist_add(list, tmp);
  442. len = base16_decode(sk, DIGEST_LEN,
  443. auth_1_sk_2_str, strlen(auth_1_sk_2_str));
  444. tt_int_op(len, OP_EQ, DIGEST_LEN);
  445. tmp = tor_malloc(DIGEST_LEN);
  446. memcpy(tmp, sk, DIGEST_LEN);
  447. smartlist_add(list, tmp);
  448. } else if (strcmp(digest_str, auth_id_digest_2_str) == 0) {
  449. list = smartlist_new();
  450. len = base16_decode(sk, DIGEST_LEN,
  451. auth_2_sk_1_str, strlen(auth_2_sk_1_str));
  452. tt_int_op(len, OP_EQ, DIGEST_LEN);
  453. tmp = tor_malloc(DIGEST_LEN);
  454. memcpy(tmp, sk, DIGEST_LEN);
  455. smartlist_add(list, tmp);
  456. len = base16_decode(sk, DIGEST_LEN,
  457. auth_2_sk_2_str, strlen(auth_2_sk_2_str));
  458. tt_int_op(len, OP_EQ, DIGEST_LEN);
  459. tmp = tor_malloc(DIGEST_LEN);
  460. memcpy(tmp, sk, DIGEST_LEN);
  461. smartlist_add(list, tmp);
  462. }
  463. done:
  464. return list;
  465. }
  466. static download_status_t *
  467. cert_dl_status_fp_sk_mock(const char *fp_digest, const char *sk_digest)
  468. {
  469. download_status_t *dl = NULL;
  470. char fp_digest_str[HEX_DIGEST_LEN+1], sk_digest_str[HEX_DIGEST_LEN+1];
  471. /*
  472. * Unpack the digests so we can compare them and figure out which
  473. * dl status we want.
  474. */
  475. tt_assert(fp_digest != NULL);
  476. base16_encode(fp_digest_str, HEX_DIGEST_LEN + 1,
  477. fp_digest, DIGEST_LEN);
  478. fp_digest_str[HEX_DIGEST_LEN] = '\0';
  479. tt_assert(sk_digest != NULL);
  480. base16_encode(sk_digest_str, HEX_DIGEST_LEN + 1,
  481. sk_digest, DIGEST_LEN);
  482. sk_digest_str[HEX_DIGEST_LEN] = '\0';
  483. if (strcmp(fp_digest_str, auth_id_digest_1_str) == 0) {
  484. if (strcmp(sk_digest_str, auth_1_sk_1_str) == 0) {
  485. dl = &auth_1_sk_1_dls;
  486. } else if (strcmp(sk_digest_str, auth_1_sk_2_str) == 0) {
  487. dl = &auth_1_sk_2_dls;
  488. }
  489. } else if (strcmp(fp_digest_str, auth_id_digest_2_str) == 0) {
  490. if (strcmp(sk_digest_str, auth_2_sk_1_str) == 0) {
  491. dl = &auth_2_sk_1_dls;
  492. } else if (strcmp(sk_digest_str, auth_2_sk_2_str) == 0) {
  493. dl = &auth_2_sk_2_dls;
  494. }
  495. }
  496. done:
  497. return dl;
  498. }
  499. static void
  500. setup_cert_mocks(void)
  501. {
  502. MOCK(list_authority_ids_with_downloads, cert_dl_status_auth_ids_mock);
  503. MOCK(id_only_download_status_for_authority_id,
  504. cert_dl_status_def_for_auth_mock);
  505. MOCK(list_sk_digests_for_authority_id,
  506. cert_dl_status_sks_for_auth_id_mock);
  507. MOCK(download_status_for_authority_id_and_sk,
  508. cert_dl_status_fp_sk_mock);
  509. reset_mocked_dl_statuses();
  510. }
  511. static void
  512. clear_cert_mocks(void)
  513. {
  514. UNMOCK(list_authority_ids_with_downloads);
  515. UNMOCK(id_only_download_status_for_authority_id);
  516. UNMOCK(list_sk_digests_for_authority_id);
  517. UNMOCK(download_status_for_authority_id_and_sk);
  518. }
  519. static smartlist_t *
  520. descbr_get_digests_mock(void)
  521. {
  522. char digest[DIGEST_LEN], *tmp;
  523. int len;
  524. smartlist_t *list = NULL;
  525. if (!disable_descbr) {
  526. /* Just pretend we have only the two hard-coded digests listed above */
  527. list = smartlist_new();
  528. len = base16_decode(digest, DIGEST_LEN,
  529. descbr_digest_1_str, strlen(descbr_digest_1_str));
  530. tt_int_op(len, OP_EQ, DIGEST_LEN);
  531. tmp = tor_malloc(DIGEST_LEN);
  532. memcpy(tmp, digest, DIGEST_LEN);
  533. smartlist_add(list, tmp);
  534. len = base16_decode(digest, DIGEST_LEN,
  535. descbr_digest_2_str, strlen(descbr_digest_2_str));
  536. tt_int_op(len, OP_EQ, DIGEST_LEN);
  537. tmp = tor_malloc(DIGEST_LEN);
  538. memcpy(tmp, digest, DIGEST_LEN);
  539. smartlist_add(list, tmp);
  540. }
  541. done:
  542. return list;
  543. }
  544. static download_status_t *
  545. descbr_get_dl_by_digest_mock(const char *digest)
  546. {
  547. download_status_t *dl = NULL;
  548. char digest_str[HEX_DIGEST_LEN+1];
  549. if (!disable_descbr) {
  550. tt_assert(digest != NULL);
  551. base16_encode(digest_str, HEX_DIGEST_LEN + 1,
  552. digest, DIGEST_LEN);
  553. digest_str[HEX_DIGEST_LEN] = '\0';
  554. if (strcmp(digest_str, descbr_digest_1_str) == 0) {
  555. dl = &descbr_digest_1_dl;
  556. } else if (strcmp(digest_str, descbr_digest_2_str) == 0) {
  557. dl = &descbr_digest_2_dl;
  558. }
  559. }
  560. done:
  561. return dl;
  562. }
  563. static void
  564. setup_desc_mocks(void)
  565. {
  566. MOCK(router_get_descriptor_digests,
  567. descbr_get_digests_mock);
  568. MOCK(router_get_dl_status_by_descriptor_digest,
  569. descbr_get_dl_by_digest_mock);
  570. reset_mocked_dl_statuses();
  571. }
  572. static void
  573. clear_desc_mocks(void)
  574. {
  575. UNMOCK(router_get_descriptor_digests);
  576. UNMOCK(router_get_dl_status_by_descriptor_digest);
  577. }
  578. static void
  579. setup_bridge_mocks(void)
  580. {
  581. disable_descbr = 0;
  582. MOCK(list_bridge_identities,
  583. descbr_get_digests_mock);
  584. MOCK(get_bridge_dl_status_by_id,
  585. descbr_get_dl_by_digest_mock);
  586. reset_mocked_dl_statuses();
  587. }
  588. static void
  589. clear_bridge_mocks(void)
  590. {
  591. UNMOCK(list_bridge_identities);
  592. UNMOCK(get_bridge_dl_status_by_id);
  593. disable_descbr = 0;
  594. }
  595. static void
  596. test_download_status_consensus(void *arg)
  597. {
  598. /* We just need one of these to pass, it doesn't matter what's in it */
  599. control_connection_t dummy;
  600. /* Get results out */
  601. char *answer = NULL;
  602. const char *errmsg = NULL;
  603. (void)arg;
  604. /* Check that the unknown prefix case works; no mocks needed yet */
  605. getinfo_helper_downloads(&dummy, "downloads/foo", &answer, &errmsg);
  606. tt_assert(answer == NULL);
  607. tt_str_op(errmsg, OP_EQ, "Unknown download status query");
  608. setup_ns_mocks();
  609. /*
  610. * Check returning serialized dlstatuses, and implicitly also test
  611. * download_status_to_string().
  612. */
  613. /* Case 1 default/FLAV_NS*/
  614. memcpy(&(ns_dl_status[FLAV_NS]), &dls_sample_1,
  615. sizeof(download_status_t));
  616. getinfo_helper_downloads(&dummy, "downloads/networkstatus/ns",
  617. &answer, &errmsg);
  618. tt_assert(answer != NULL);
  619. tt_assert(errmsg == NULL);
  620. tt_str_op(answer, OP_EQ, dls_sample_1_str);
  621. tor_free(answer);
  622. errmsg = NULL;
  623. /* Case 2 default/FLAV_MICRODESC */
  624. memcpy(&(ns_dl_status[FLAV_MICRODESC]), &dls_sample_2,
  625. sizeof(download_status_t));
  626. getinfo_helper_downloads(&dummy, "downloads/networkstatus/microdesc",
  627. &answer, &errmsg);
  628. tt_assert(answer != NULL);
  629. tt_assert(errmsg == NULL);
  630. tt_str_op(answer, OP_EQ, dls_sample_2_str);
  631. tor_free(answer);
  632. errmsg = NULL;
  633. /* Case 3 bootstrap/FLAV_NS */
  634. memcpy(&(ns_dl_status_bootstrap[FLAV_NS]), &dls_sample_3,
  635. sizeof(download_status_t));
  636. getinfo_helper_downloads(&dummy, "downloads/networkstatus/ns/bootstrap",
  637. &answer, &errmsg);
  638. tt_assert(answer != NULL);
  639. tt_assert(errmsg == NULL);
  640. tt_str_op(answer, OP_EQ, dls_sample_3_str);
  641. tor_free(answer);
  642. errmsg = NULL;
  643. /* Case 4 bootstrap/FLAV_MICRODESC */
  644. memcpy(&(ns_dl_status_bootstrap[FLAV_MICRODESC]), &dls_sample_4,
  645. sizeof(download_status_t));
  646. getinfo_helper_downloads(&dummy,
  647. "downloads/networkstatus/microdesc/bootstrap",
  648. &answer, &errmsg);
  649. tt_assert(answer != NULL);
  650. tt_assert(errmsg == NULL);
  651. tt_str_op(answer, OP_EQ, dls_sample_4_str);
  652. tor_free(answer);
  653. errmsg = NULL;
  654. /* Case 5 running/FLAV_NS */
  655. memcpy(&(ns_dl_status_running[FLAV_NS]), &dls_sample_5,
  656. sizeof(download_status_t));
  657. getinfo_helper_downloads(&dummy,
  658. "downloads/networkstatus/ns/running",
  659. &answer, &errmsg);
  660. tt_assert(answer != NULL);
  661. tt_assert(errmsg == NULL);
  662. tt_str_op(answer, OP_EQ, dls_sample_5_str);
  663. tor_free(answer);
  664. errmsg = NULL;
  665. /* Case 6 running/FLAV_MICRODESC */
  666. memcpy(&(ns_dl_status_running[FLAV_MICRODESC]), &dls_sample_6,
  667. sizeof(download_status_t));
  668. getinfo_helper_downloads(&dummy,
  669. "downloads/networkstatus/microdesc/running",
  670. &answer, &errmsg);
  671. tt_assert(answer != NULL);
  672. tt_assert(errmsg == NULL);
  673. tt_str_op(answer, OP_EQ, dls_sample_6_str);
  674. tor_free(answer);
  675. errmsg = NULL;
  676. /* Now check the error case */
  677. getinfo_helper_downloads(&dummy, "downloads/networkstatus/foo",
  678. &answer, &errmsg);
  679. tt_assert(answer == NULL);
  680. tt_assert(errmsg != NULL);
  681. tt_str_op(errmsg, OP_EQ, "Unknown flavor");
  682. errmsg = NULL;
  683. done:
  684. clear_ns_mocks();
  685. tor_free(answer);
  686. return;
  687. }
  688. static void
  689. test_download_status_cert(void *arg)
  690. {
  691. /* We just need one of these to pass, it doesn't matter what's in it */
  692. control_connection_t dummy;
  693. /* Get results out */
  694. char *question = NULL;
  695. char *answer = NULL;
  696. const char *errmsg = NULL;
  697. (void)arg;
  698. setup_cert_mocks();
  699. /*
  700. * Check returning serialized dlstatuses and digest lists, and implicitly
  701. * also test download_status_to_string() and digest_list_to_string().
  702. */
  703. /* Case 1 - list of authority identity fingerprints */
  704. getinfo_helper_downloads(&dummy,
  705. "downloads/cert/fps",
  706. &answer, &errmsg);
  707. tt_assert(answer != NULL);
  708. tt_assert(errmsg == NULL);
  709. tt_str_op(answer, OP_EQ, auth_id_digest_expected_list);
  710. tor_free(answer);
  711. errmsg = NULL;
  712. /* Case 2 - download status for default cert for 1st auth id */
  713. memcpy(&auth_def_cert_download_status_1, &dls_sample_1,
  714. sizeof(download_status_t));
  715. tor_asprintf(&question, "downloads/cert/fp/%s", auth_id_digest_1_str);
  716. tt_assert(question != NULL);
  717. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  718. tt_assert(answer != NULL);
  719. tt_assert(errmsg == NULL);
  720. tt_str_op(answer, OP_EQ, dls_sample_1_str);
  721. tor_free(question);
  722. tor_free(answer);
  723. errmsg = NULL;
  724. /* Case 3 - download status for default cert for 2nd auth id */
  725. memcpy(&auth_def_cert_download_status_2, &dls_sample_2,
  726. sizeof(download_status_t));
  727. tor_asprintf(&question, "downloads/cert/fp/%s", auth_id_digest_2_str);
  728. tt_assert(question != NULL);
  729. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  730. tt_assert(answer != NULL);
  731. tt_assert(errmsg == NULL);
  732. tt_str_op(answer, OP_EQ, dls_sample_2_str);
  733. tor_free(question);
  734. tor_free(answer);
  735. errmsg = NULL;
  736. /* Case 4 - list of signing key digests for 1st auth id */
  737. tor_asprintf(&question, "downloads/cert/fp/%s/sks", auth_id_digest_1_str);
  738. tt_assert(question != NULL);
  739. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  740. tt_assert(answer != NULL);
  741. tt_assert(errmsg == NULL);
  742. tt_str_op(answer, OP_EQ, auth_1_sk_digest_expected_list);
  743. tor_free(question);
  744. tor_free(answer);
  745. errmsg = NULL;
  746. /* Case 5 - list of signing key digests for 2nd auth id */
  747. tor_asprintf(&question, "downloads/cert/fp/%s/sks", auth_id_digest_2_str);
  748. tt_assert(question != NULL);
  749. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  750. tt_assert(answer != NULL);
  751. tt_assert(errmsg == NULL);
  752. tt_str_op(answer, OP_EQ, auth_2_sk_digest_expected_list);
  753. tor_free(question);
  754. tor_free(answer);
  755. errmsg = NULL;
  756. /* Case 6 - download status for 1st auth id, 1st sk */
  757. memcpy(&auth_1_sk_1_dls, &dls_sample_3,
  758. sizeof(download_status_t));
  759. tor_asprintf(&question, "downloads/cert/fp/%s/%s",
  760. auth_id_digest_1_str, auth_1_sk_1_str);
  761. tt_assert(question != NULL);
  762. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  763. tt_assert(answer != NULL);
  764. tt_assert(errmsg == NULL);
  765. tt_str_op(answer, OP_EQ, dls_sample_3_str);
  766. tor_free(question);
  767. tor_free(answer);
  768. errmsg = NULL;
  769. /* Case 7 - download status for 1st auth id, 2nd sk */
  770. memcpy(&auth_1_sk_2_dls, &dls_sample_4,
  771. sizeof(download_status_t));
  772. tor_asprintf(&question, "downloads/cert/fp/%s/%s",
  773. auth_id_digest_1_str, auth_1_sk_2_str);
  774. tt_assert(question != NULL);
  775. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  776. tt_assert(answer != NULL);
  777. tt_assert(errmsg == NULL);
  778. tt_str_op(answer, OP_EQ, dls_sample_4_str);
  779. tor_free(question);
  780. tor_free(answer);
  781. errmsg = NULL;
  782. /* Case 8 - download status for 2nd auth id, 1st sk */
  783. memcpy(&auth_2_sk_1_dls, &dls_sample_5,
  784. sizeof(download_status_t));
  785. tor_asprintf(&question, "downloads/cert/fp/%s/%s",
  786. auth_id_digest_2_str, auth_2_sk_1_str);
  787. tt_assert(question != NULL);
  788. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  789. tt_assert(answer != NULL);
  790. tt_assert(errmsg == NULL);
  791. tt_str_op(answer, OP_EQ, dls_sample_5_str);
  792. tor_free(question);
  793. tor_free(answer);
  794. errmsg = NULL;
  795. /* Case 9 - download status for 2nd auth id, 2nd sk */
  796. memcpy(&auth_2_sk_2_dls, &dls_sample_6,
  797. sizeof(download_status_t));
  798. tor_asprintf(&question, "downloads/cert/fp/%s/%s",
  799. auth_id_digest_2_str, auth_2_sk_2_str);
  800. tt_assert(question != NULL);
  801. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  802. tt_assert(answer != NULL);
  803. tt_assert(errmsg == NULL);
  804. tt_str_op(answer, OP_EQ, dls_sample_6_str);
  805. tor_free(question);
  806. tor_free(answer);
  807. errmsg = NULL;
  808. /* Now check the error cases */
  809. /* Case 1 - query is garbage after downloads/cert/ part */
  810. getinfo_helper_downloads(&dummy, "downloads/cert/blahdeblah",
  811. &answer, &errmsg);
  812. tt_assert(answer == NULL);
  813. tt_assert(errmsg != NULL);
  814. tt_str_op(errmsg, OP_EQ, "Unknown certificate download status query");
  815. errmsg = NULL;
  816. /*
  817. * Case 2 - looks like downloads/cert/fp/<fp>, but <fp> isn't even
  818. * the right length for a digest.
  819. */
  820. getinfo_helper_downloads(&dummy, "downloads/cert/fp/2B1D36D32B2942406",
  821. &answer, &errmsg);
  822. tt_assert(answer == NULL);
  823. tt_assert(errmsg != NULL);
  824. tt_str_op(errmsg, OP_EQ, "That didn't look like a digest");
  825. errmsg = NULL;
  826. /*
  827. * Case 3 - looks like downloads/cert/fp/<fp>, and <fp> is digest-sized,
  828. * but not parseable as one.
  829. */
  830. getinfo_helper_downloads(&dummy,
  831. "downloads/cert/fp/82F52AF55D250115FE44D3GC81D49643241D56A1",
  832. &answer, &errmsg);
  833. tt_assert(answer == NULL);
  834. tt_assert(errmsg != NULL);
  835. tt_str_op(errmsg, OP_EQ, "That didn't look like a digest");
  836. errmsg = NULL;
  837. /*
  838. * Case 4 - downloads/cert/fp/<fp>, and <fp> is not a known authority
  839. * identity digest
  840. */
  841. getinfo_helper_downloads(&dummy,
  842. "downloads/cert/fp/AC4F23B5745BDD2A77997B85B1FD85D05C2E0F61",
  843. &answer, &errmsg);
  844. tt_assert(answer == NULL);
  845. tt_assert(errmsg != NULL);
  846. tt_str_op(errmsg, OP_EQ,
  847. "Failed to get download status for this authority identity digest");
  848. errmsg = NULL;
  849. /*
  850. * Case 5 - looks like downloads/cert/fp/<fp>/<anything>, but <fp> doesn't
  851. * parse as a sensible digest.
  852. */
  853. getinfo_helper_downloads(&dummy,
  854. "downloads/cert/fp/82F52AF55D250115FE44D3GC81D49643241D56A1/blah",
  855. &answer, &errmsg);
  856. tt_assert(answer == NULL);
  857. tt_assert(errmsg != NULL);
  858. tt_str_op(errmsg, OP_EQ, "That didn't look like an identity digest");
  859. errmsg = NULL;
  860. /*
  861. * Case 6 - looks like downloads/cert/fp/<fp>/<anything>, but <fp> doesn't
  862. * parse as a sensible digest.
  863. */
  864. getinfo_helper_downloads(&dummy,
  865. "downloads/cert/fp/82F52AF55D25/blah",
  866. &answer, &errmsg);
  867. tt_assert(answer == NULL);
  868. tt_assert(errmsg != NULL);
  869. tt_str_op(errmsg, OP_EQ, "That didn't look like an identity digest");
  870. errmsg = NULL;
  871. /*
  872. * Case 7 - downloads/cert/fp/<fp>/sks, and <fp> is not a known authority
  873. * digest.
  874. */
  875. getinfo_helper_downloads(&dummy,
  876. "downloads/cert/fp/AC4F23B5745BDD2A77997B85B1FD85D05C2E0F61/sks",
  877. &answer, &errmsg);
  878. tt_assert(answer == NULL);
  879. tt_assert(errmsg != NULL);
  880. tt_str_op(errmsg, OP_EQ,
  881. "Failed to get list of signing key digests for this authority "
  882. "identity digest");
  883. errmsg = NULL;
  884. /*
  885. * Case 8 - looks like downloads/cert/fp/<fp>/<sk>, but <sk> doesn't
  886. * parse as a signing key digest.
  887. */
  888. getinfo_helper_downloads(&dummy,
  889. "downloads/cert/fp/AC4F23B5745BDD2A77997B85B1FD85D05C2E0F61/"
  890. "82F52AF55D250115FE44D3GC81D49643241D56A1",
  891. &answer, &errmsg);
  892. tt_assert(answer == NULL);
  893. tt_assert(errmsg != NULL);
  894. tt_str_op(errmsg, OP_EQ, "That didn't look like a signing key digest");
  895. errmsg = NULL;
  896. /*
  897. * Case 9 - looks like downloads/cert/fp/<fp>/<sk>, but <sk> doesn't
  898. * parse as a signing key digest.
  899. */
  900. getinfo_helper_downloads(&dummy,
  901. "downloads/cert/fp/AC4F23B5745BDD2A77997B85B1FD85D05C2E0F61/"
  902. "82F52AF55D250115FE44D",
  903. &answer, &errmsg);
  904. tt_assert(answer == NULL);
  905. tt_assert(errmsg != NULL);
  906. tt_str_op(errmsg, OP_EQ, "That didn't look like a signing key digest");
  907. errmsg = NULL;
  908. /*
  909. * Case 10 - downloads/cert/fp/<fp>/<sk>, but <fp> isn't a known
  910. * authority identity digest.
  911. */
  912. getinfo_helper_downloads(&dummy,
  913. "downloads/cert/fp/C6B05DF332F74DB9A13498EE3BBC7AA2F69FCB45/"
  914. "3A214FC21AE25B012C2ECCB5F4EC8A3602D0545D",
  915. &answer, &errmsg);
  916. tt_assert(answer == NULL);
  917. tt_assert(errmsg != NULL);
  918. tt_str_op(errmsg, OP_EQ,
  919. "Failed to get download status for this identity/"
  920. "signing key digest pair");
  921. errmsg = NULL;
  922. /*
  923. * Case 11 - downloads/cert/fp/<fp>/<sk>, but <sk> isn't a known
  924. * signing key digest.
  925. */
  926. getinfo_helper_downloads(&dummy,
  927. "downloads/cert/fp/63CDD326DFEF0CA020BDD3FEB45A3286FE13A061/"
  928. "3A214FC21AE25B012C2ECCB5F4EC8A3602D0545D",
  929. &answer, &errmsg);
  930. tt_assert(answer == NULL);
  931. tt_assert(errmsg != NULL);
  932. tt_str_op(errmsg, OP_EQ,
  933. "Failed to get download status for this identity/"
  934. "signing key digest pair");
  935. errmsg = NULL;
  936. /*
  937. * Case 12 - downloads/cert/fp/<fp>/<sk>, but <sk> is on the list for
  938. * a different authority identity digest.
  939. */
  940. getinfo_helper_downloads(&dummy,
  941. "downloads/cert/fp/63CDD326DFEF0CA020BDD3FEB45A3286FE13A061/"
  942. "9451B8F1B10952384EB58B5F230C0BB701626C9B",
  943. &answer, &errmsg);
  944. tt_assert(answer == NULL);
  945. tt_assert(errmsg != NULL);
  946. tt_str_op(errmsg, OP_EQ,
  947. "Failed to get download status for this identity/"
  948. "signing key digest pair");
  949. errmsg = NULL;
  950. done:
  951. clear_cert_mocks();
  952. tor_free(answer);
  953. return;
  954. }
  955. static void
  956. test_download_status_desc(void *arg)
  957. {
  958. /* We just need one of these to pass, it doesn't matter what's in it */
  959. control_connection_t dummy;
  960. /* Get results out */
  961. char *question = NULL;
  962. char *answer = NULL;
  963. const char *errmsg = NULL;
  964. (void)arg;
  965. setup_desc_mocks();
  966. /*
  967. * Check returning serialized dlstatuses and digest lists, and implicitly
  968. * also test download_status_to_string() and digest_list_to_string().
  969. */
  970. /* Case 1 - list of router descriptor digests */
  971. getinfo_helper_downloads(&dummy,
  972. "downloads/desc/descs",
  973. &answer, &errmsg);
  974. tt_assert(answer != NULL);
  975. tt_assert(errmsg == NULL);
  976. tt_str_op(answer, OP_EQ, descbr_expected_list);
  977. tor_free(answer);
  978. errmsg = NULL;
  979. /* Case 2 - get download status for router descriptor 1 */
  980. memcpy(&descbr_digest_1_dl, &dls_sample_1,
  981. sizeof(download_status_t));
  982. tor_asprintf(&question, "downloads/desc/%s", descbr_digest_1_str);
  983. tt_assert(question != NULL);
  984. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  985. tt_assert(answer != NULL);
  986. tt_assert(errmsg == NULL);
  987. tt_str_op(answer, OP_EQ, dls_sample_1_str);
  988. tor_free(question);
  989. tor_free(answer);
  990. errmsg = NULL;
  991. /* Case 3 - get download status for router descriptor 1 */
  992. memcpy(&descbr_digest_2_dl, &dls_sample_2,
  993. sizeof(download_status_t));
  994. tor_asprintf(&question, "downloads/desc/%s", descbr_digest_2_str);
  995. tt_assert(question != NULL);
  996. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  997. tt_assert(answer != NULL);
  998. tt_assert(errmsg == NULL);
  999. tt_str_op(answer, OP_EQ, dls_sample_2_str);
  1000. tor_free(question);
  1001. tor_free(answer);
  1002. errmsg = NULL;
  1003. /* Now check the error cases */
  1004. /* Case 1 - non-digest-length garbage after downloads/desc */
  1005. getinfo_helper_downloads(&dummy, "downloads/desc/blahdeblah",
  1006. &answer, &errmsg);
  1007. tt_assert(answer == NULL);
  1008. tt_assert(errmsg != NULL);
  1009. tt_str_op(errmsg, OP_EQ, "Unknown router descriptor download status query");
  1010. errmsg = NULL;
  1011. /* Case 2 - nonparseable digest-shaped thing */
  1012. getinfo_helper_downloads(
  1013. &dummy,
  1014. "downloads/desc/774EC52FD9A5B80A6FACZE536616E8022E3470AG",
  1015. &answer, &errmsg);
  1016. tt_assert(answer == NULL);
  1017. tt_assert(errmsg != NULL);
  1018. tt_str_op(errmsg, OP_EQ, "That didn't look like a digest");
  1019. errmsg = NULL;
  1020. /* Case 3 - digest we have no descriptor for */
  1021. getinfo_helper_downloads(
  1022. &dummy,
  1023. "downloads/desc/B05B46135B0B2C04EBE1DD6A6AE4B12D7CD2226A",
  1024. &answer, &errmsg);
  1025. tt_assert(answer == NULL);
  1026. tt_assert(errmsg != NULL);
  1027. tt_str_op(errmsg, OP_EQ, "No such descriptor digest found");
  1028. errmsg = NULL;
  1029. /* Case 4 - microdescs only */
  1030. disable_descbr = 1;
  1031. getinfo_helper_downloads(&dummy,
  1032. "downloads/desc/descs",
  1033. &answer, &errmsg);
  1034. tt_assert(answer == NULL);
  1035. tt_assert(errmsg != NULL);
  1036. tt_str_op(errmsg, OP_EQ,
  1037. "We don't seem to have a networkstatus-flavored consensus");
  1038. errmsg = NULL;
  1039. disable_descbr = 0;
  1040. done:
  1041. clear_desc_mocks();
  1042. tor_free(answer);
  1043. return;
  1044. }
  1045. static void
  1046. test_download_status_bridge(void *arg)
  1047. {
  1048. /* We just need one of these to pass, it doesn't matter what's in it */
  1049. control_connection_t dummy;
  1050. /* Get results out */
  1051. char *question = NULL;
  1052. char *answer = NULL;
  1053. const char *errmsg = NULL;
  1054. (void)arg;
  1055. setup_bridge_mocks();
  1056. /*
  1057. * Check returning serialized dlstatuses and digest lists, and implicitly
  1058. * also test download_status_to_string() and digest_list_to_string().
  1059. */
  1060. /* Case 1 - list of bridge identity digests */
  1061. getinfo_helper_downloads(&dummy,
  1062. "downloads/bridge/bridges",
  1063. &answer, &errmsg);
  1064. tt_assert(answer != NULL);
  1065. tt_assert(errmsg == NULL);
  1066. tt_str_op(answer, OP_EQ, descbr_expected_list);
  1067. tor_free(answer);
  1068. errmsg = NULL;
  1069. /* Case 2 - get download status for bridge descriptor 1 */
  1070. memcpy(&descbr_digest_1_dl, &dls_sample_3,
  1071. sizeof(download_status_t));
  1072. tor_asprintf(&question, "downloads/bridge/%s", descbr_digest_1_str);
  1073. tt_assert(question != NULL);
  1074. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  1075. tt_assert(answer != NULL);
  1076. tt_assert(errmsg == NULL);
  1077. tt_str_op(answer, OP_EQ, dls_sample_3_str);
  1078. tor_free(question);
  1079. tor_free(answer);
  1080. errmsg = NULL;
  1081. /* Case 3 - get download status for router descriptor 1 */
  1082. memcpy(&descbr_digest_2_dl, &dls_sample_4,
  1083. sizeof(download_status_t));
  1084. tor_asprintf(&question, "downloads/bridge/%s", descbr_digest_2_str);
  1085. tt_assert(question != NULL);
  1086. getinfo_helper_downloads(&dummy, question, &answer, &errmsg);
  1087. tt_assert(answer != NULL);
  1088. tt_assert(errmsg == NULL);
  1089. tt_str_op(answer, OP_EQ, dls_sample_4_str);
  1090. tor_free(question);
  1091. tor_free(answer);
  1092. errmsg = NULL;
  1093. /* Now check the error cases */
  1094. /* Case 1 - non-digest-length garbage after downloads/bridge */
  1095. getinfo_helper_downloads(&dummy, "downloads/bridge/blahdeblah",
  1096. &answer, &errmsg);
  1097. tt_assert(answer == NULL);
  1098. tt_assert(errmsg != NULL);
  1099. tt_str_op(errmsg, OP_EQ, "Unknown bridge descriptor download status query");
  1100. errmsg = NULL;
  1101. /* Case 2 - nonparseable digest-shaped thing */
  1102. getinfo_helper_downloads(
  1103. &dummy,
  1104. "downloads/bridge/774EC52FD9A5B80A6FACZE536616E8022E3470AG",
  1105. &answer, &errmsg);
  1106. tt_assert(answer == NULL);
  1107. tt_assert(errmsg != NULL);
  1108. tt_str_op(errmsg, OP_EQ, "That didn't look like a digest");
  1109. errmsg = NULL;
  1110. /* Case 3 - digest we have no descriptor for */
  1111. getinfo_helper_downloads(
  1112. &dummy,
  1113. "downloads/bridge/B05B46135B0B2C04EBE1DD6A6AE4B12D7CD2226A",
  1114. &answer, &errmsg);
  1115. tt_assert(answer == NULL);
  1116. tt_assert(errmsg != NULL);
  1117. tt_str_op(errmsg, OP_EQ, "No such bridge identity digest found");
  1118. errmsg = NULL;
  1119. /* Case 4 - bridges disabled */
  1120. disable_descbr = 1;
  1121. getinfo_helper_downloads(&dummy,
  1122. "downloads/bridge/bridges",
  1123. &answer, &errmsg);
  1124. tt_assert(answer == NULL);
  1125. tt_assert(errmsg != NULL);
  1126. tt_str_op(errmsg, OP_EQ, "We don't seem to be using bridges");
  1127. errmsg = NULL;
  1128. disable_descbr = 0;
  1129. done:
  1130. clear_bridge_mocks();
  1131. tor_free(answer);
  1132. return;
  1133. }
  1134. struct testcase_t controller_tests[] = {
  1135. { "add_onion_helper_keyarg", test_add_onion_helper_keyarg, 0, NULL, NULL },
  1136. { "rend_service_parse_port_config", test_rend_service_parse_port_config, 0,
  1137. NULL, NULL },
  1138. { "add_onion_helper_clientauth", test_add_onion_helper_clientauth, 0, NULL,
  1139. NULL },
  1140. { "download_status_consensus", test_download_status_consensus, 0, NULL,
  1141. NULL },
  1142. { "download_status_cert", test_download_status_cert, 0, NULL,
  1143. NULL },
  1144. { "download_status_desc", test_download_status_desc, 0, NULL, NULL },
  1145. { "download_status_bridge", test_download_status_bridge, 0, NULL, NULL },
  1146. END_OF_TESTCASES
  1147. };