test_rendcache.c 42 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327
  1. /* Copyright (c) 2010-2015, The Tor Project, Inc. */
  2. /* See LICENSE for licensing information */
  3. #include "orconfig.h"
  4. #include "or.h"
  5. #include "test.h"
  6. #define RENDCACHE_PRIVATE
  7. #include "rendcache.h"
  8. #include "router.h"
  9. #include "routerlist.h"
  10. #include "config.h"
  11. #include <openssl/rsa.h>
  12. #include "rend_test_helpers.h"
  13. #define NS_MODULE rend_cache
  14. static const int RECENT_TIME = -10;
  15. static const int TIME_IN_THE_PAST = -(REND_CACHE_MAX_AGE + \
  16. REND_CACHE_MAX_SKEW + 10);
  17. static const int TIME_IN_THE_FUTURE = REND_CACHE_MAX_SKEW + 10;
  18. extern strmap_t *rend_cache;
  19. extern digestmap_t *rend_cache_v2_dir;
  20. extern strmap_t *rend_cache_failure;
  21. extern size_t rend_cache_total_allocation;
  22. static rend_data_t *
  23. mock_rend_data(const char *onion_address)
  24. {
  25. rend_data_t *rend_query = tor_malloc_zero(sizeof(rend_data_t));
  26. strncpy(rend_query->onion_address, onion_address,
  27. REND_SERVICE_ID_LEN_BASE32+1);
  28. rend_query->auth_type = REND_NO_AUTH;
  29. rend_query->hsdirs_fp = smartlist_new();
  30. smartlist_add(rend_query->hsdirs_fp, tor_memdup("aaaaaaaaaaaaaaaaaaaaaaaa",
  31. DIGEST_LEN));
  32. return rend_query;
  33. }
  34. static void
  35. test_rend_cache_lookup_entry(void *data)
  36. {
  37. int ret;
  38. rend_data_t *mock_rend_query = NULL;
  39. char desc_id_base32[REND_DESC_ID_V2_LEN_BASE32 + 1];
  40. rend_cache_entry_t *entry = NULL;
  41. rend_encoded_v2_service_descriptor_t *desc_holder = NULL;
  42. char *service_id = NULL;
  43. (void)data;
  44. rend_cache_init();
  45. generate_desc(RECENT_TIME, &desc_holder, &service_id, 3);
  46. ret = rend_cache_lookup_entry("abababababababab", 0, NULL);
  47. tt_int_op(ret, OP_EQ, -ENOENT);
  48. ret = rend_cache_lookup_entry("invalid query", 2, NULL);
  49. tt_int_op(ret, OP_EQ, -EINVAL);
  50. ret = rend_cache_lookup_entry("abababababababab", 2, NULL);
  51. tt_int_op(ret, OP_EQ, -ENOENT);
  52. ret = rend_cache_lookup_entry("abababababababab", 4224, NULL);
  53. tt_int_op(ret, OP_EQ, -ENOENT);
  54. mock_rend_query = mock_rend_data(service_id);
  55. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  56. DIGEST_LEN);
  57. rend_cache_store_v2_desc_as_client(desc_holder->desc_str, desc_id_base32,
  58. mock_rend_query, NULL);
  59. ret = rend_cache_lookup_entry(service_id, 2, NULL);
  60. tt_int_op(ret, OP_EQ, 0);
  61. ret = rend_cache_lookup_entry(service_id, 2, &entry);
  62. tt_assert(entry);
  63. tt_int_op(entry->len, OP_EQ, strlen(desc_holder->desc_str));
  64. tt_str_op(entry->desc, OP_EQ, desc_holder->desc_str);
  65. done:
  66. rend_encoded_v2_service_descriptor_free(desc_holder);
  67. tor_free(service_id);
  68. rend_cache_free_all();
  69. rend_data_free(mock_rend_query);
  70. }
  71. static void
  72. test_rend_cache_store_v2_desc_as_client(void *data)
  73. {
  74. rend_cache_store_status_t ret;
  75. rend_data_t *mock_rend_query;
  76. char desc_id_base32[REND_DESC_ID_V2_LEN_BASE32 + 1];
  77. rend_cache_entry_t *entry = NULL;
  78. rend_encoded_v2_service_descriptor_t *desc_holder = NULL;
  79. char *service_id = NULL;
  80. char client_cookie[REND_DESC_COOKIE_LEN];
  81. (void)data;
  82. rend_cache_init();
  83. generate_desc(RECENT_TIME, &desc_holder, &service_id, 3);
  84. // Test success
  85. mock_rend_query = mock_rend_data(service_id);
  86. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  87. DIGEST_LEN);
  88. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  89. desc_id_base32, mock_rend_query,
  90. &entry);
  91. tt_int_op(ret, OP_EQ, RCS_OKAY);
  92. tt_assert(entry);
  93. tt_int_op(entry->len, OP_EQ, strlen(desc_holder->desc_str));
  94. tt_str_op(entry->desc, OP_EQ, desc_holder->desc_str);
  95. // Test various failure modes
  96. // TODO: a too long desc_id_base32 argument crashes the function
  97. /* ret = rend_cache_store_v2_desc_as_client( */
  98. /* desc_holder->desc_str, */
  99. /* "3TOOLONG3TOOLONG3TOOLONG3TOOLONG3TOOLONG3TOOLONG", */
  100. /* &mock_rend_query, NULL); */
  101. /* tt_int_op(ret, OP_EQ, RCS_BADDESC); */
  102. // Test bad base32 failure
  103. // This causes an assertion failure if we're running with assertions.
  104. // But when doing coverage, we can test it.
  105. #ifdef TOR_COVERAGE
  106. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  107. "!xqunszqnaolrrfmtzgaki7mxelgvkj", mock_rend_query, NULL);
  108. tt_int_op(ret, OP_EQ, RCS_BADDESC);
  109. #endif
  110. // Test invalid descriptor
  111. ret = rend_cache_store_v2_desc_as_client("invalid descriptor",
  112. "3xqunszqnaolrrfmtzgaki7mxelgvkje", mock_rend_query, NULL);
  113. tt_int_op(ret, OP_EQ, RCS_BADDESC);
  114. // TODO: it doesn't seem to be possible to test invalid service ID condition.
  115. // that means it is likely not possible to have that condition without
  116. // earlier conditions failing first (such as signature checking of the desc)
  117. rend_cache_free_all();
  118. // Test mismatch between service ID and onion address
  119. rend_cache_init();
  120. strncpy(mock_rend_query->onion_address, "abc", REND_SERVICE_ID_LEN_BASE32+1);
  121. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  122. desc_id_base32,
  123. mock_rend_query, NULL);
  124. tt_int_op(ret, OP_EQ, RCS_BADDESC);
  125. rend_cache_free_all();
  126. rend_data_free(mock_rend_query);
  127. // Test incorrect descriptor ID
  128. rend_cache_init();
  129. mock_rend_query = mock_rend_data(service_id);
  130. desc_id_base32[0]++;
  131. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  132. desc_id_base32, mock_rend_query,
  133. NULL);
  134. tt_int_op(ret, OP_EQ, RCS_BADDESC);
  135. desc_id_base32[0]--;
  136. rend_cache_free_all();
  137. // Test too old descriptor
  138. rend_cache_init();
  139. rend_encoded_v2_service_descriptor_free(desc_holder);
  140. tor_free(service_id);
  141. rend_data_free(mock_rend_query);
  142. generate_desc(TIME_IN_THE_PAST, &desc_holder, &service_id, 3);
  143. mock_rend_query = mock_rend_data(service_id);
  144. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  145. DIGEST_LEN);
  146. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  147. desc_id_base32,
  148. mock_rend_query, NULL);
  149. tt_int_op(ret, OP_EQ, RCS_BADDESC);
  150. rend_cache_free_all();
  151. // Test too new descriptor (in the future)
  152. rend_cache_init();
  153. rend_encoded_v2_service_descriptor_free(desc_holder);
  154. tor_free(service_id);
  155. rend_data_free(mock_rend_query);
  156. generate_desc(TIME_IN_THE_FUTURE, &desc_holder, &service_id, 3);
  157. mock_rend_query = mock_rend_data(service_id);
  158. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  159. DIGEST_LEN);
  160. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  161. desc_id_base32, mock_rend_query,
  162. NULL);
  163. tt_int_op(ret, OP_EQ, RCS_BADDESC);
  164. rend_cache_free_all();
  165. // Test when a descriptor is already in the cache
  166. rend_cache_init();
  167. rend_encoded_v2_service_descriptor_free(desc_holder);
  168. tor_free(service_id);
  169. rend_data_free(mock_rend_query);
  170. generate_desc(RECENT_TIME, &desc_holder, &service_id, 3);
  171. mock_rend_query = mock_rend_data(service_id);
  172. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  173. DIGEST_LEN);
  174. rend_cache_store_v2_desc_as_client(desc_holder->desc_str, desc_id_base32,
  175. mock_rend_query, NULL);
  176. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  177. desc_id_base32, mock_rend_query,
  178. NULL);
  179. tt_int_op(ret, OP_EQ, RCS_OKAY);
  180. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  181. desc_id_base32, mock_rend_query,
  182. &entry);
  183. tt_int_op(ret, OP_EQ, RCS_OKAY);
  184. tt_assert(entry);
  185. rend_cache_free_all();
  186. // Test unsuccessful decrypting of introduction points
  187. rend_cache_init();
  188. rend_encoded_v2_service_descriptor_free(desc_holder);
  189. tor_free(service_id);
  190. rend_data_free(mock_rend_query);
  191. generate_desc(RECENT_TIME, &desc_holder, &service_id, 3);
  192. mock_rend_query = mock_rend_data(service_id);
  193. mock_rend_query->auth_type = REND_BASIC_AUTH;
  194. client_cookie[0] = 'A';
  195. memcpy(mock_rend_query->descriptor_cookie, client_cookie,
  196. REND_DESC_COOKIE_LEN);
  197. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  198. DIGEST_LEN);
  199. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  200. desc_id_base32, mock_rend_query,
  201. NULL);
  202. tt_int_op(ret, OP_EQ, RCS_OKAY);
  203. rend_cache_free_all();
  204. // Test successful run when we have REND_BASIC_AUTH but not cookie
  205. rend_cache_init();
  206. rend_encoded_v2_service_descriptor_free(desc_holder);
  207. tor_free(service_id);
  208. rend_data_free(mock_rend_query);
  209. generate_desc(RECENT_TIME, &desc_holder, &service_id, 3);
  210. mock_rend_query = mock_rend_data(service_id);
  211. mock_rend_query->auth_type = REND_BASIC_AUTH;
  212. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  213. DIGEST_LEN);
  214. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  215. desc_id_base32, mock_rend_query,
  216. NULL);
  217. tt_int_op(ret, OP_EQ, RCS_OKAY);
  218. rend_cache_free_all();
  219. // Test when we have no introduction points
  220. rend_cache_init();
  221. rend_encoded_v2_service_descriptor_free(desc_holder);
  222. tor_free(service_id);
  223. rend_data_free(mock_rend_query);
  224. generate_desc(RECENT_TIME, &desc_holder, &service_id, 0);
  225. mock_rend_query = mock_rend_data(service_id);
  226. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  227. DIGEST_LEN);
  228. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  229. desc_id_base32, mock_rend_query,
  230. NULL);
  231. tt_int_op(ret, OP_EQ, RCS_BADDESC);
  232. rend_cache_free_all();
  233. // Test when we have too many intro points
  234. rend_cache_init();
  235. rend_encoded_v2_service_descriptor_free(desc_holder);
  236. tor_free(service_id);
  237. rend_data_free(mock_rend_query);
  238. generate_desc(RECENT_TIME, &desc_holder, &service_id, MAX_INTRO_POINTS+1);
  239. mock_rend_query = mock_rend_data(service_id);
  240. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  241. DIGEST_LEN);
  242. ret = rend_cache_store_v2_desc_as_client(desc_holder->desc_str,
  243. desc_id_base32, mock_rend_query,
  244. NULL);
  245. tt_int_op(ret, OP_EQ, RCS_BADDESC);
  246. done:
  247. rend_encoded_v2_service_descriptor_free(desc_holder);
  248. tor_free(service_id);
  249. rend_cache_free_all();
  250. rend_data_free(mock_rend_query);
  251. }
  252. static void
  253. test_rend_cache_store_v2_desc_as_client_with_different_time(void *data)
  254. {
  255. rend_cache_store_status_t ret;
  256. rend_data_t *mock_rend_query;
  257. char desc_id_base32[REND_DESC_ID_V2_LEN_BASE32 + 1];
  258. rend_service_descriptor_t *generated = NULL;
  259. smartlist_t *descs = smartlist_new();
  260. time_t t;
  261. char *service_id = NULL;
  262. rend_encoded_v2_service_descriptor_t *desc_holder_newer;
  263. rend_encoded_v2_service_descriptor_t *desc_holder_older;
  264. t = time(NULL);
  265. rend_cache_init();
  266. create_descriptor(&generated, &service_id, 3);
  267. generated->timestamp = t + RECENT_TIME;
  268. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  269. REND_NO_AUTH, NULL, NULL);
  270. desc_holder_newer = ((rend_encoded_v2_service_descriptor_t *)
  271. smartlist_get(descs, 0));
  272. smartlist_set(descs, 0, NULL);
  273. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  274. rend_encoded_v2_service_descriptor_free(d));
  275. smartlist_free(descs);
  276. descs = smartlist_new();
  277. generated->timestamp = (t + RECENT_TIME) - 20;
  278. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  279. REND_NO_AUTH, NULL, NULL);
  280. desc_holder_older = ((rend_encoded_v2_service_descriptor_t *)
  281. smartlist_get(descs, 0));
  282. smartlist_set(descs, 0, NULL);
  283. (void)data;
  284. // Test when a descriptor is already in the cache and it is newer than the
  285. // one we submit
  286. mock_rend_query = mock_rend_data(service_id);
  287. base32_encode(desc_id_base32, sizeof(desc_id_base32),
  288. desc_holder_newer->desc_id, DIGEST_LEN);
  289. rend_cache_store_v2_desc_as_client(desc_holder_newer->desc_str,
  290. desc_id_base32, mock_rend_query, NULL);
  291. ret = rend_cache_store_v2_desc_as_client(desc_holder_older->desc_str,
  292. desc_id_base32, mock_rend_query,
  293. NULL);
  294. tt_int_op(ret, OP_EQ, RCS_OKAY);
  295. rend_cache_free_all();
  296. // Test when an old descriptor is in the cache and we submit a newer one
  297. rend_cache_init();
  298. rend_cache_store_v2_desc_as_client(desc_holder_older->desc_str,
  299. desc_id_base32, mock_rend_query, NULL);
  300. ret = rend_cache_store_v2_desc_as_client(desc_holder_newer->desc_str,
  301. desc_id_base32, mock_rend_query,
  302. NULL);
  303. tt_int_op(ret, OP_EQ, RCS_OKAY);
  304. done:
  305. rend_encoded_v2_service_descriptor_free(desc_holder_newer);
  306. rend_encoded_v2_service_descriptor_free(desc_holder_older);
  307. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  308. rend_encoded_v2_service_descriptor_free(d));
  309. smartlist_free(descs);
  310. rend_service_descriptor_free(generated);
  311. tor_free(service_id);
  312. rend_cache_free_all();
  313. rend_data_free(mock_rend_query);
  314. }
  315. #define NS_SUBMODULE lookup_v2_desc_as_dir
  316. NS_DECL(const routerinfo_t *, router_get_my_routerinfo, (void));
  317. NS_DECL(int, hid_serv_responsible_for_desc_id, (const char *id));
  318. static routerinfo_t *mock_routerinfo;
  319. static int hid_serv_responsible_for_desc_id_response;
  320. static const routerinfo_t *
  321. NS(router_get_my_routerinfo)(void)
  322. {
  323. if (!mock_routerinfo) {
  324. mock_routerinfo = tor_malloc(sizeof(routerinfo_t));
  325. }
  326. return mock_routerinfo;
  327. }
  328. static int
  329. NS(hid_serv_responsible_for_desc_id)(const char *id)
  330. {
  331. (void)id;
  332. return hid_serv_responsible_for_desc_id_response;
  333. }
  334. static void
  335. test_rend_cache_lookup_v2_desc_as_dir(void *data)
  336. {
  337. int ret;
  338. char desc_id_base32[REND_DESC_ID_V2_LEN_BASE32 + 1];
  339. rend_encoded_v2_service_descriptor_t *desc_holder = NULL;
  340. char *service_id = NULL;
  341. const char *ret_desc = NULL;
  342. (void)data;
  343. NS_MOCK(router_get_my_routerinfo);
  344. NS_MOCK(hid_serv_responsible_for_desc_id);
  345. rend_cache_init();
  346. // Test invalid base32
  347. ret = rend_cache_lookup_v2_desc_as_dir("!bababababababab", NULL);
  348. tt_int_op(ret, OP_EQ, -1);
  349. // Test non-existent descriptor but well formed
  350. ret = rend_cache_lookup_v2_desc_as_dir("3xqunszqnaolrrfmtzgaki7mxelgvkje",
  351. NULL);
  352. tt_int_op(ret, OP_EQ, 0);
  353. // Test existing descriptor
  354. hid_serv_responsible_for_desc_id_response = 1;
  355. generate_desc(RECENT_TIME, &desc_holder, &service_id, 3);
  356. rend_cache_store_v2_desc_as_dir(desc_holder->desc_str);
  357. base32_encode(desc_id_base32, sizeof(desc_id_base32), desc_holder->desc_id,
  358. DIGEST_LEN);
  359. ret = rend_cache_lookup_v2_desc_as_dir(desc_id_base32, &ret_desc);
  360. tt_int_op(ret, OP_EQ, 1);
  361. tt_assert(ret_desc);
  362. done:
  363. NS_UNMOCK(router_get_my_routerinfo);
  364. NS_UNMOCK(hid_serv_responsible_for_desc_id);
  365. tor_free(mock_routerinfo);
  366. rend_cache_free_all();
  367. rend_encoded_v2_service_descriptor_free(desc_holder);
  368. tor_free(service_id);
  369. }
  370. #undef NS_SUBMODULE
  371. #define NS_SUBMODULE store_v2_desc_as_dir
  372. NS_DECL(const routerinfo_t *, router_get_my_routerinfo, (void));
  373. NS_DECL(int, hid_serv_responsible_for_desc_id, (const char *id));
  374. static const routerinfo_t *
  375. NS(router_get_my_routerinfo)(void)
  376. {
  377. return mock_routerinfo;
  378. }
  379. static int
  380. NS(hid_serv_responsible_for_desc_id)(const char *id)
  381. {
  382. (void)id;
  383. return hid_serv_responsible_for_desc_id_response;
  384. }
  385. static void
  386. test_rend_cache_store_v2_desc_as_dir(void *data)
  387. {
  388. (void)data;
  389. rend_cache_store_status_t ret;
  390. rend_encoded_v2_service_descriptor_t *desc_holder = NULL;
  391. char *service_id = NULL;
  392. NS_MOCK(router_get_my_routerinfo);
  393. NS_MOCK(hid_serv_responsible_for_desc_id);
  394. rend_cache_init();
  395. // Test when we are not an HS dir
  396. mock_routerinfo = NULL;
  397. ret = rend_cache_store_v2_desc_as_dir("");
  398. tt_int_op(ret, OP_EQ, RCS_NOTDIR);
  399. // Test when we can't parse the descriptor
  400. mock_routerinfo = tor_malloc(sizeof(routerinfo_t));
  401. hid_serv_responsible_for_desc_id_response = 1;
  402. ret = rend_cache_store_v2_desc_as_dir("unparseable");
  403. tt_int_op(ret, OP_EQ, RCS_BADDESC);
  404. // Test when we are not responsible for an HS
  405. hid_serv_responsible_for_desc_id_response = 0;
  406. generate_desc(RECENT_TIME, &desc_holder, &service_id, 3);
  407. ret = rend_cache_store_v2_desc_as_dir(desc_holder->desc_str);
  408. tt_int_op(ret, OP_EQ, RCS_OKAY);
  409. rend_encoded_v2_service_descriptor_free(desc_holder);
  410. tor_free(service_id);
  411. // Test when we have an old descriptor
  412. hid_serv_responsible_for_desc_id_response = 1;
  413. generate_desc(TIME_IN_THE_PAST, &desc_holder, &service_id, 3);
  414. ret = rend_cache_store_v2_desc_as_dir(desc_holder->desc_str);
  415. tt_int_op(ret, OP_EQ, RCS_OKAY);
  416. rend_encoded_v2_service_descriptor_free(desc_holder);
  417. tor_free(service_id);
  418. // Test when we have a descriptor in the future
  419. generate_desc(TIME_IN_THE_FUTURE, &desc_holder, &service_id, 3);
  420. ret = rend_cache_store_v2_desc_as_dir(desc_holder->desc_str);
  421. tt_int_op(ret, OP_EQ, RCS_OKAY);
  422. rend_encoded_v2_service_descriptor_free(desc_holder);
  423. tor_free(service_id);
  424. // Test when two descriptors
  425. generate_desc(TIME_IN_THE_FUTURE, &desc_holder, &service_id, 3);
  426. ret = rend_cache_store_v2_desc_as_dir(desc_holder->desc_str);
  427. tt_int_op(ret, OP_EQ, RCS_OKAY);
  428. rend_encoded_v2_service_descriptor_free(desc_holder);
  429. tor_free(service_id);
  430. // Test when asking for hidden service statistics HiddenServiceStatistics
  431. rend_cache_purge();
  432. generate_desc(RECENT_TIME, &desc_holder, &service_id, 3);
  433. get_options_mutable()->HiddenServiceStatistics = 1;
  434. ret = rend_cache_store_v2_desc_as_dir(desc_holder->desc_str);
  435. tt_int_op(ret, OP_EQ, RCS_OKAY);
  436. done:
  437. NS_UNMOCK(router_get_my_routerinfo);
  438. NS_UNMOCK(hid_serv_responsible_for_desc_id);
  439. rend_encoded_v2_service_descriptor_free(desc_holder);
  440. tor_free(service_id);
  441. rend_cache_free_all();
  442. }
  443. static void
  444. test_rend_cache_store_v2_desc_as_dir_with_different_time(void *data)
  445. {
  446. (void)data;
  447. rend_cache_store_status_t ret;
  448. rend_service_descriptor_t *generated = NULL;
  449. smartlist_t *descs = smartlist_new();
  450. time_t t;
  451. char *service_id = NULL;
  452. rend_encoded_v2_service_descriptor_t *desc_holder_newer;
  453. rend_encoded_v2_service_descriptor_t *desc_holder_older;
  454. NS_MOCK(router_get_my_routerinfo);
  455. NS_MOCK(hid_serv_responsible_for_desc_id);
  456. rend_cache_init();
  457. t = time(NULL);
  458. create_descriptor(&generated, &service_id, 3);
  459. generated->timestamp = t + RECENT_TIME;
  460. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  461. REND_NO_AUTH, NULL, NULL);
  462. desc_holder_newer = ((rend_encoded_v2_service_descriptor_t *)
  463. smartlist_get(descs, 0));
  464. smartlist_set(descs, 0, NULL);
  465. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  466. rend_encoded_v2_service_descriptor_free(d));
  467. smartlist_free(descs);
  468. descs = smartlist_new();
  469. generated->timestamp = (t + RECENT_TIME) - 20;
  470. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  471. REND_NO_AUTH, NULL, NULL);
  472. desc_holder_older = ((rend_encoded_v2_service_descriptor_t *)
  473. smartlist_get(descs, 0));
  474. smartlist_set(descs, 0, NULL);
  475. // Test when we have a newer descriptor stored
  476. mock_routerinfo = tor_malloc(sizeof(routerinfo_t));
  477. hid_serv_responsible_for_desc_id_response = 1;
  478. rend_cache_store_v2_desc_as_dir(desc_holder_newer->desc_str);
  479. ret = rend_cache_store_v2_desc_as_dir(desc_holder_older->desc_str);
  480. tt_int_op(ret, OP_EQ, RCS_OKAY);
  481. // Test when we have an old descriptor stored
  482. rend_cache_purge();
  483. rend_cache_store_v2_desc_as_dir(desc_holder_older->desc_str);
  484. ret = rend_cache_store_v2_desc_as_dir(desc_holder_newer->desc_str);
  485. tt_int_op(ret, OP_EQ, RCS_OKAY);
  486. done:
  487. NS_UNMOCK(router_get_my_routerinfo);
  488. NS_UNMOCK(hid_serv_responsible_for_desc_id);
  489. rend_cache_free_all();
  490. rend_service_descriptor_free(generated);
  491. tor_free(service_id);
  492. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  493. rend_encoded_v2_service_descriptor_free(d));
  494. smartlist_free(descs);
  495. rend_encoded_v2_service_descriptor_free(desc_holder_newer);
  496. rend_encoded_v2_service_descriptor_free(desc_holder_older);
  497. }
  498. static void
  499. test_rend_cache_store_v2_desc_as_dir_with_different_content(void *data)
  500. {
  501. (void)data;
  502. rend_cache_store_status_t ret;
  503. rend_service_descriptor_t *generated = NULL;
  504. smartlist_t *descs = smartlist_new();
  505. time_t t;
  506. char *service_id = NULL;
  507. rend_encoded_v2_service_descriptor_t *desc_holder_one = NULL;
  508. rend_encoded_v2_service_descriptor_t *desc_holder_two = NULL;
  509. NS_MOCK(router_get_my_routerinfo);
  510. NS_MOCK(hid_serv_responsible_for_desc_id);
  511. rend_cache_init();
  512. t = time(NULL);
  513. create_descriptor(&generated, &service_id, 3);
  514. generated->timestamp = t + RECENT_TIME;
  515. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  516. REND_NO_AUTH, NULL, NULL);
  517. desc_holder_one = ((rend_encoded_v2_service_descriptor_t *)
  518. smartlist_get(descs, 0));
  519. smartlist_set(descs, 0, NULL);
  520. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  521. rend_encoded_v2_service_descriptor_free(d));
  522. smartlist_free(descs);
  523. descs = smartlist_new();
  524. generated->timestamp = t + RECENT_TIME;
  525. generated->protocols = 41;
  526. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  527. REND_NO_AUTH, NULL, NULL);
  528. desc_holder_two = ((rend_encoded_v2_service_descriptor_t *)
  529. smartlist_get(descs, 0));
  530. smartlist_set(descs, 0, NULL);
  531. // Test when we have another descriptor stored, with a different descriptor
  532. mock_routerinfo = tor_malloc(sizeof(routerinfo_t));
  533. hid_serv_responsible_for_desc_id_response = 1;
  534. rend_cache_store_v2_desc_as_dir(desc_holder_one->desc_str);
  535. ret = rend_cache_store_v2_desc_as_dir(desc_holder_two->desc_str);
  536. tt_int_op(ret, OP_EQ, RCS_OKAY);
  537. done:
  538. NS_UNMOCK(router_get_my_routerinfo);
  539. NS_UNMOCK(hid_serv_responsible_for_desc_id);
  540. rend_cache_free_all();
  541. rend_service_descriptor_free(generated);
  542. tor_free(service_id);
  543. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  544. rend_encoded_v2_service_descriptor_free(d));
  545. smartlist_free(descs);
  546. rend_encoded_v2_service_descriptor_free(desc_holder_one);
  547. rend_encoded_v2_service_descriptor_free(desc_holder_two);
  548. }
  549. #undef NS_SUBMODULE
  550. static void
  551. test_rend_cache_init(void *data)
  552. {
  553. (void)data;
  554. tt_assert_msg(!rend_cache, "rend_cache should be NULL when starting");
  555. tt_assert_msg(!rend_cache_v2_dir, "rend_cache_v2_dir should be NULL "
  556. "when starting");
  557. tt_assert_msg(!rend_cache_failure, "rend_cache_failure should be NULL when "
  558. "starting");
  559. rend_cache_init();
  560. tt_assert_msg(rend_cache, "rend_cache should not be NULL after initing");
  561. tt_assert_msg(rend_cache_v2_dir, "rend_cache_v2_dir should not be NULL "
  562. "after initing");
  563. tt_assert_msg(rend_cache_failure, "rend_cache_failure should not be NULL "
  564. "after initing");
  565. tt_int_op(strmap_size(rend_cache), OP_EQ, 0);
  566. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 0);
  567. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 0);
  568. done:
  569. rend_cache_free_all();
  570. }
  571. static void
  572. test_rend_cache_decrement_allocation(void *data)
  573. {
  574. (void)data;
  575. // Test when the cache has enough allocations
  576. rend_cache_total_allocation = 10;
  577. rend_cache_decrement_allocation(3);
  578. tt_int_op(rend_cache_total_allocation, OP_EQ, 7);
  579. // Test when there are not enough allocations
  580. rend_cache_total_allocation = 1;
  581. rend_cache_decrement_allocation(2);
  582. tt_int_op(rend_cache_total_allocation, OP_EQ, 0);
  583. // And again
  584. rend_cache_decrement_allocation(2);
  585. tt_int_op(rend_cache_total_allocation, OP_EQ, 0);
  586. done:
  587. (void)0;
  588. }
  589. static void
  590. test_rend_cache_increment_allocation(void *data)
  591. {
  592. (void)data;
  593. // Test when the cache is not overflowing
  594. rend_cache_total_allocation = 5;
  595. rend_cache_increment_allocation(3);
  596. tt_int_op(rend_cache_total_allocation, OP_EQ, 8);
  597. // Test when there are too many allocations
  598. rend_cache_total_allocation = SIZE_MAX-1;
  599. rend_cache_increment_allocation(2);
  600. tt_u64_op(rend_cache_total_allocation, OP_EQ, SIZE_MAX);
  601. // And again
  602. rend_cache_increment_allocation(2);
  603. tt_u64_op(rend_cache_total_allocation, OP_EQ, SIZE_MAX);
  604. done:
  605. (void)0;
  606. }
  607. static void
  608. test_rend_cache_failure_intro_entry_new(void *data)
  609. {
  610. time_t now;
  611. rend_cache_failure_intro_t *entry;
  612. rend_intro_point_failure_t failure;
  613. (void)data;
  614. failure = INTRO_POINT_FAILURE_TIMEOUT;
  615. now = time(NULL);
  616. entry = rend_cache_failure_intro_entry_new(failure);
  617. tt_int_op(entry->failure_type, OP_EQ, INTRO_POINT_FAILURE_TIMEOUT);
  618. tt_int_op(entry->created_ts, OP_GE, now-5);
  619. tt_int_op(entry->created_ts, OP_LE, now+5);
  620. done:
  621. tor_free(entry);
  622. }
  623. static void
  624. test_rend_cache_failure_intro_lookup(void *data)
  625. {
  626. (void)data;
  627. int ret;
  628. rend_cache_failure_t *failure;
  629. rend_cache_failure_intro_t *ip;
  630. rend_cache_failure_intro_t *entry;
  631. rend_cache_init();
  632. failure = rend_cache_failure_entry_new();
  633. ip = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  634. digestmap_set(failure->intro_failures, "ip1", ip);
  635. strmap_set_lc(rend_cache_failure, "foo1", failure);
  636. // Test not found
  637. ret = cache_failure_intro_lookup((const uint8_t *)"foo1", "foo2", NULL);
  638. tt_int_op(ret, OP_EQ, 0);
  639. // Test found with no intro failures in it
  640. ret = cache_failure_intro_lookup((const uint8_t *)"ip2", "foo1", NULL);
  641. tt_int_op(ret, OP_EQ, 0);
  642. // Test found
  643. ret = cache_failure_intro_lookup((const uint8_t *)"ip1", "foo1", NULL);
  644. tt_int_op(ret, OP_EQ, 1);
  645. // Test found and asking for entry
  646. cache_failure_intro_lookup((const uint8_t *)"ip1", "foo1", &entry);
  647. tt_assert(entry);
  648. tt_assert(entry == ip);
  649. done:
  650. rend_cache_free_all();
  651. }
  652. static void
  653. test_rend_cache_clean(void *data)
  654. {
  655. rend_cache_entry_t *one, *two;
  656. rend_service_descriptor_t *desc_one, *desc_two;
  657. strmap_iter_t *iter = NULL;
  658. const char *key;
  659. void *val;
  660. (void)data;
  661. rend_cache_init();
  662. // Test with empty rendcache
  663. rend_cache_clean(time(NULL), REND_CACHE_TYPE_CLIENT);
  664. tt_int_op(strmap_size(rend_cache), OP_EQ, 0);
  665. // Test with two old entries
  666. one = tor_malloc_zero(sizeof(rend_cache_entry_t));
  667. two = tor_malloc_zero(sizeof(rend_cache_entry_t));
  668. desc_one = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  669. desc_two = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  670. one->parsed = desc_one;
  671. two->parsed = desc_two;
  672. desc_one->timestamp = time(NULL) + TIME_IN_THE_PAST;
  673. desc_two->timestamp = (time(NULL) + TIME_IN_THE_PAST) - 10;
  674. desc_one->pk = pk_generate(0);
  675. desc_two->pk = pk_generate(1);
  676. strmap_set_lc(rend_cache, "foo1", one);
  677. strmap_set_lc(rend_cache, "foo2", two);
  678. rend_cache_clean(time(NULL), REND_CACHE_TYPE_CLIENT);
  679. tt_int_op(strmap_size(rend_cache), OP_EQ, 0);
  680. // Test with one old entry and one newer entry
  681. one = tor_malloc_zero(sizeof(rend_cache_entry_t));
  682. two = tor_malloc_zero(sizeof(rend_cache_entry_t));
  683. desc_one = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  684. desc_two = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  685. one->parsed = desc_one;
  686. two->parsed = desc_two;
  687. desc_one->timestamp = (time(NULL) + TIME_IN_THE_PAST) - 10;
  688. desc_two->timestamp = time(NULL) - 100;
  689. desc_one->pk = pk_generate(0);
  690. desc_two->pk = pk_generate(1);
  691. strmap_set_lc(rend_cache, "foo1", one);
  692. strmap_set_lc(rend_cache, "foo2", two);
  693. rend_cache_clean(time(NULL), REND_CACHE_TYPE_CLIENT);
  694. tt_int_op(strmap_size(rend_cache), OP_EQ, 1);
  695. iter = strmap_iter_init(rend_cache);
  696. strmap_iter_get(iter, &key, &val);
  697. tt_str_op(key, OP_EQ, "foo2");
  698. done:
  699. rend_cache_free_all();
  700. }
  701. static void
  702. test_rend_cache_failure_entry_new(void *data)
  703. {
  704. rend_cache_failure_t *failure;
  705. (void)data;
  706. failure = rend_cache_failure_entry_new();
  707. tt_assert(failure);
  708. tt_int_op(digestmap_size(failure->intro_failures), OP_EQ, 0);
  709. done:
  710. tor_free(failure);
  711. }
  712. static void
  713. test_rend_cache_failure_entry_free(void *data)
  714. {
  715. (void)data;
  716. // Test that it can deal with a NULL argument
  717. rend_cache_failure_entry_free(NULL);
  718. /* done: */
  719. /* (void)0; */
  720. }
  721. static void
  722. test_rend_cache_failure_clean(void *data)
  723. {
  724. rend_cache_failure_t *failure;
  725. rend_cache_failure_intro_t *ip_one, *ip_two;
  726. (void)data;
  727. rend_cache_init();
  728. // Test with empty failure cache
  729. rend_cache_failure_clean(time(NULL));
  730. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 0);
  731. // Test with one empty failure entry
  732. failure = rend_cache_failure_entry_new();
  733. strmap_set_lc(rend_cache_failure, "foo1", failure);
  734. rend_cache_failure_clean(time(NULL));
  735. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 0);
  736. // Test with one new intro point
  737. failure = rend_cache_failure_entry_new();
  738. ip_one = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  739. digestmap_set(failure->intro_failures, "ip1", ip_one);
  740. strmap_set_lc(rend_cache_failure, "foo1", failure);
  741. rend_cache_failure_clean(time(NULL));
  742. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 1);
  743. // Test with one old intro point
  744. rend_cache_failure_purge();
  745. failure = rend_cache_failure_entry_new();
  746. ip_one = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  747. ip_one->created_ts = time(NULL) - 7*60;
  748. digestmap_set(failure->intro_failures, "ip1", ip_one);
  749. strmap_set_lc(rend_cache_failure, "foo1", failure);
  750. rend_cache_failure_clean(time(NULL));
  751. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 0);
  752. // Test with one old intro point and one new one
  753. rend_cache_failure_purge();
  754. failure = rend_cache_failure_entry_new();
  755. ip_one = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  756. ip_one->created_ts = time(NULL) - 7*60;
  757. digestmap_set(failure->intro_failures, "ip1", ip_one);
  758. ip_two = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  759. ip_two->created_ts = time(NULL) - 2*60;
  760. digestmap_set(failure->intro_failures, "ip2", ip_two);
  761. strmap_set_lc(rend_cache_failure, "foo1", failure);
  762. rend_cache_failure_clean(time(NULL));
  763. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 1);
  764. tt_int_op(digestmap_size(failure->intro_failures), OP_EQ, 1);
  765. done:
  766. rend_cache_free_all();
  767. }
  768. static void
  769. test_rend_cache_failure_remove(void *data)
  770. {
  771. rend_service_descriptor_t *desc;
  772. (void)data;
  773. rend_cache_init();
  774. // Test that it deals well with a NULL desc
  775. rend_cache_failure_remove(NULL);
  776. // Test a descriptor that isn't in the cache
  777. desc = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  778. desc->pk = pk_generate(0);
  779. rend_cache_failure_remove(desc);
  780. // There seems to not exist any way of getting rend_cache_failure_remove()
  781. // to fail because of a problem with rend_get_service_id from here
  782. rend_cache_free_all();
  783. /* done: */
  784. /* (void)0; */
  785. }
  786. static void
  787. test_rend_cache_free_all(void *data)
  788. {
  789. rend_cache_failure_t *failure;
  790. rend_cache_entry_t *one;
  791. rend_service_descriptor_t *desc_one;
  792. (void)data;
  793. rend_cache_init();
  794. failure = rend_cache_failure_entry_new();
  795. strmap_set_lc(rend_cache_failure, "foo1", failure);
  796. one = tor_malloc_zero(sizeof(rend_cache_entry_t));
  797. desc_one = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  798. one->parsed = desc_one;
  799. desc_one->timestamp = time(NULL) + TIME_IN_THE_PAST;
  800. desc_one->pk = pk_generate(0);
  801. strmap_set_lc(rend_cache, "foo1", one);
  802. rend_cache_free_all();
  803. tt_assert(!rend_cache);
  804. tt_assert(!rend_cache_v2_dir);
  805. tt_assert(!rend_cache_failure);
  806. tt_assert(!rend_cache_total_allocation);
  807. done:
  808. rend_cache_free_all();
  809. }
  810. static void
  811. test_rend_cache_entry_free(void *data)
  812. {
  813. (void)data;
  814. rend_cache_entry_t *e;
  815. // Handles NULL correctly
  816. rend_cache_entry_free(NULL);
  817. // Handles NULL descriptor correctly
  818. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  819. rend_cache_entry_free(e);
  820. // Handles non-NULL descriptor correctly
  821. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  822. e->desc = (char *)malloc(10);
  823. rend_cache_entry_free(e);
  824. /* done: */
  825. /* (void)0; */
  826. }
  827. static void
  828. test_rend_cache_purge(void *data)
  829. {
  830. strmap_t *our_rend_cache;
  831. (void)data;
  832. // Deals with a NULL rend_cache
  833. rend_cache_purge();
  834. tt_assert(rend_cache);
  835. tt_int_op(strmap_size(rend_cache), OP_EQ, 0);
  836. // Deals with existing rend_cache
  837. rend_cache_free_all();
  838. rend_cache_init();
  839. our_rend_cache = rend_cache;
  840. rend_cache_purge();
  841. tt_assert(rend_cache);
  842. tt_assert(rend_cache == our_rend_cache);
  843. done:
  844. rend_cache_free_all();
  845. }
  846. static void
  847. test_rend_cache_failure_intro_add(void *data)
  848. {
  849. (void)data;
  850. rend_cache_failure_t *fail_entry;
  851. rend_cache_failure_intro_t *entry;
  852. rend_cache_init();
  853. // Adds non-existing entry
  854. cache_failure_intro_add((const uint8_t *)"foo1", "foo2",
  855. INTRO_POINT_FAILURE_TIMEOUT);
  856. fail_entry = strmap_get_lc(rend_cache_failure, "foo2");
  857. tt_assert(fail_entry);
  858. tt_int_op(digestmap_size(fail_entry->intro_failures), OP_EQ, 1);
  859. entry = digestmap_get(fail_entry->intro_failures, "foo1");
  860. tt_assert(entry);
  861. // Adds existing entry
  862. cache_failure_intro_add((const uint8_t *)"foo1", "foo2",
  863. INTRO_POINT_FAILURE_TIMEOUT);
  864. fail_entry = strmap_get_lc(rend_cache_failure, "foo2");
  865. tt_assert(fail_entry);
  866. tt_int_op(digestmap_size(fail_entry->intro_failures), OP_EQ, 1);
  867. entry = digestmap_get(fail_entry->intro_failures, "foo1");
  868. tt_assert(entry);
  869. done:
  870. rend_cache_free_all();
  871. }
  872. static void
  873. test_rend_cache_intro_failure_note(void *data)
  874. {
  875. (void)data;
  876. rend_cache_failure_t *fail_entry;
  877. rend_cache_failure_intro_t *entry;
  878. rend_cache_init();
  879. // Test not found
  880. rend_cache_intro_failure_note(INTRO_POINT_FAILURE_TIMEOUT,
  881. (const uint8_t *)"foo1", "foo2");
  882. fail_entry = strmap_get_lc(rend_cache_failure, "foo2");
  883. tt_assert(fail_entry);
  884. tt_int_op(digestmap_size(fail_entry->intro_failures), OP_EQ, 1);
  885. entry = digestmap_get(fail_entry->intro_failures, "foo1");
  886. tt_assert(entry);
  887. tt_int_op(entry->failure_type, OP_EQ, INTRO_POINT_FAILURE_TIMEOUT);
  888. // Test found
  889. rend_cache_intro_failure_note(INTRO_POINT_FAILURE_UNREACHABLE,
  890. (const uint8_t *)"foo1", "foo2");
  891. tt_int_op(entry->failure_type, OP_EQ, INTRO_POINT_FAILURE_UNREACHABLE);
  892. done:
  893. rend_cache_free_all();
  894. }
  895. #define NS_SUBMODULE clean_v2_descs_as_dir
  896. NS_DECL(int, hid_serv_responsible_for_desc_id, (const char *id));
  897. static int
  898. NS(hid_serv_responsible_for_desc_id)(const char *id)
  899. {
  900. (void)id;
  901. return hid_serv_responsible_for_desc_id_response;
  902. }
  903. static void
  904. test_rend_cache_clean_v2_descs_as_dir(void *data)
  905. {
  906. rend_cache_entry_t *e;
  907. time_t now;
  908. rend_service_descriptor_t *desc;
  909. now = time(NULL);
  910. (void)data;
  911. NS_MOCK(hid_serv_responsible_for_desc_id);
  912. rend_cache_init();
  913. // Test running with an empty cache
  914. rend_cache_clean_v2_descs_as_dir(now, 0);
  915. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 0);
  916. // Test with only one new entry
  917. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  918. e->last_served = now;
  919. desc = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  920. desc->timestamp = now;
  921. desc->pk = pk_generate(0);
  922. e->parsed = desc;
  923. digestmap_set(rend_cache_v2_dir, "abcde", e);
  924. hid_serv_responsible_for_desc_id_response = 1;
  925. rend_cache_clean_v2_descs_as_dir(now, 0);
  926. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 1);
  927. // Test with one old entry
  928. desc->timestamp = now - (REND_CACHE_MAX_AGE + REND_CACHE_MAX_SKEW + 1000);
  929. rend_cache_clean_v2_descs_as_dir(now, 0);
  930. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 0);
  931. // Test with one entry that is not under the responsibility of this
  932. // hidden service
  933. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  934. e->last_served = now;
  935. desc = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  936. desc->timestamp = now;
  937. desc->pk = pk_generate(0);
  938. e->parsed = desc;
  939. digestmap_set(rend_cache_v2_dir, "abcde", e);
  940. hid_serv_responsible_for_desc_id_response = 0;
  941. rend_cache_clean_v2_descs_as_dir(now, 0);
  942. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 0);
  943. // Test with one entry that has an old last served
  944. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  945. e->last_served = now - (REND_CACHE_MAX_AGE + REND_CACHE_MAX_SKEW + 1000);
  946. desc = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  947. desc->timestamp = now;
  948. desc->pk = pk_generate(0);
  949. e->parsed = desc;
  950. digestmap_set(rend_cache_v2_dir, "abcde", e);
  951. hid_serv_responsible_for_desc_id_response = 1;
  952. rend_cache_clean_v2_descs_as_dir(now, 0);
  953. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 0);
  954. // Test a run through asking for a large force_remove
  955. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  956. e->last_served = now;
  957. desc = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  958. desc->timestamp = now;
  959. desc->pk = pk_generate(0);
  960. e->parsed = desc;
  961. digestmap_set(rend_cache_v2_dir, "abcde", e);
  962. hid_serv_responsible_for_desc_id_response = 1;
  963. rend_cache_clean_v2_descs_as_dir(now, 20000);
  964. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 1);
  965. done:
  966. NS_UNMOCK(hid_serv_responsible_for_desc_id);
  967. rend_cache_free_all();
  968. }
  969. #undef NS_SUBMODULE
  970. static void
  971. test_rend_cache_entry_allocation(void *data)
  972. {
  973. (void)data;
  974. size_t ret;
  975. rend_cache_entry_t *e;
  976. // Handles a null argument
  977. ret = rend_cache_entry_allocation(NULL);
  978. tt_int_op(ret, OP_EQ, 0);
  979. // Handles a non-null argument
  980. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  981. ret = rend_cache_entry_allocation(e);
  982. tt_int_op(ret, OP_GT, sizeof(rend_cache_entry_t));
  983. done:
  984. (void)0;
  985. }
  986. static void
  987. test_rend_cache_failure_intro_entry_free(void *data)
  988. {
  989. (void)data;
  990. rend_cache_failure_intro_t *entry;
  991. // Handles a null argument
  992. rend_cache_failure_intro_entry_free(NULL);
  993. // Handles a non-null argument
  994. entry = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  995. rend_cache_failure_intro_entry_free(entry);
  996. }
  997. static void
  998. test_rend_cache_failure_purge(void *data)
  999. {
  1000. (void)data;
  1001. // Handles a null failure cache
  1002. strmap_free(rend_cache_failure, rend_cache_failure_entry_free_);
  1003. rend_cache_failure = NULL;
  1004. rend_cache_failure_purge();
  1005. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 0);
  1006. done:
  1007. rend_cache_free_all();
  1008. }
  1009. static void
  1010. test_rend_cache_validate_intro_point_failure(void *data)
  1011. {
  1012. (void)data;
  1013. rend_service_descriptor_t *desc = NULL;
  1014. char *service_id = NULL;
  1015. rend_intro_point_t *intro = NULL;
  1016. const uint8_t *identity = NULL;
  1017. rend_cache_failure_t *failure;
  1018. rend_cache_failure_intro_t *ip;
  1019. rend_cache_init();
  1020. create_descriptor(&desc, &service_id, 3);
  1021. desc->timestamp = time(NULL) + RECENT_TIME;
  1022. intro = (rend_intro_point_t *)smartlist_get(desc->intro_nodes, 0);
  1023. identity = (uint8_t *) intro->extend_info->identity_digest;
  1024. failure = rend_cache_failure_entry_new();
  1025. ip = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  1026. digestmap_set(failure->intro_failures, (char *)identity, ip);
  1027. strmap_set_lc(rend_cache_failure, service_id, failure);
  1028. // Test when we have an intro point in our cache
  1029. validate_intro_point_failure(desc, service_id);
  1030. tt_int_op(smartlist_len(desc->intro_nodes), OP_EQ, 2);
  1031. done:
  1032. rend_cache_free_all();
  1033. rend_service_descriptor_free(desc);
  1034. tor_free(service_id);
  1035. }
  1036. struct testcase_t rend_cache_tests[] = {
  1037. { "init", test_rend_cache_init, 0, NULL, NULL },
  1038. { "decrement_allocation", test_rend_cache_decrement_allocation, 0,
  1039. NULL, NULL },
  1040. { "increment_allocation", test_rend_cache_increment_allocation, 0,
  1041. NULL, NULL },
  1042. { "clean", test_rend_cache_clean, TT_FORK, NULL, NULL },
  1043. { "clean_v2_descs_as_dir", test_rend_cache_clean_v2_descs_as_dir, 0,
  1044. NULL, NULL },
  1045. { "entry_allocation", test_rend_cache_entry_allocation, 0, NULL, NULL },
  1046. { "entry_free", test_rend_cache_entry_free, 0, NULL, NULL },
  1047. { "failure_intro_entry_free", test_rend_cache_failure_intro_entry_free, 0,
  1048. NULL, NULL },
  1049. { "free_all", test_rend_cache_free_all, 0, NULL, NULL },
  1050. { "purge", test_rend_cache_purge, 0, NULL, NULL },
  1051. { "failure_clean", test_rend_cache_failure_clean, 0, NULL, NULL },
  1052. { "failure_entry_new", test_rend_cache_failure_entry_new, 0, NULL, NULL },
  1053. { "failure_entry_free", test_rend_cache_failure_entry_free, 0, NULL, NULL },
  1054. { "failure_intro_add", test_rend_cache_failure_intro_add, 0, NULL, NULL },
  1055. { "failure_intro_entry_new", test_rend_cache_failure_intro_entry_new, 0,
  1056. NULL, NULL },
  1057. { "failure_intro_lookup", test_rend_cache_failure_intro_lookup, 0,
  1058. NULL, NULL },
  1059. { "failure_purge", test_rend_cache_failure_purge, 0, NULL, NULL },
  1060. { "failure_remove", test_rend_cache_failure_remove, 0, NULL, NULL },
  1061. { "intro_failure_note", test_rend_cache_intro_failure_note, 0, NULL, NULL },
  1062. { "lookup", test_rend_cache_lookup_entry, 0, NULL, NULL },
  1063. { "lookup_v2_desc_as_dir", test_rend_cache_lookup_v2_desc_as_dir, 0,
  1064. NULL, NULL },
  1065. { "store_v2_desc_as_client", test_rend_cache_store_v2_desc_as_client, 0,
  1066. NULL, NULL },
  1067. { "store_v2_desc_as_client_with_different_time",
  1068. test_rend_cache_store_v2_desc_as_client_with_different_time, 0,
  1069. NULL, NULL },
  1070. { "store_v2_desc_as_dir", test_rend_cache_store_v2_desc_as_dir, 0,
  1071. NULL, NULL },
  1072. { "store_v2_desc_as_dir_with_different_time",
  1073. test_rend_cache_store_v2_desc_as_dir_with_different_time, 0, NULL, NULL },
  1074. { "store_v2_desc_as_dir_with_different_content",
  1075. test_rend_cache_store_v2_desc_as_dir_with_different_content, 0,
  1076. NULL, NULL },
  1077. { "validate_intro_point_failure",
  1078. test_rend_cache_validate_intro_point_failure, 0, NULL, NULL },
  1079. END_OF_TESTCASES
  1080. };