test_rendcache.c 42 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340
  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. strlcpy(rend_query->onion_address, onion_address,
  27. sizeof(rend_query->onion_address));
  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. tor_free(mock_routerinfo);
  443. }
  444. static void
  445. test_rend_cache_store_v2_desc_as_dir_with_different_time(void *data)
  446. {
  447. (void)data;
  448. rend_cache_store_status_t ret;
  449. rend_service_descriptor_t *generated = NULL;
  450. smartlist_t *descs = smartlist_new();
  451. time_t t;
  452. char *service_id = NULL;
  453. rend_encoded_v2_service_descriptor_t *desc_holder_newer;
  454. rend_encoded_v2_service_descriptor_t *desc_holder_older;
  455. NS_MOCK(router_get_my_routerinfo);
  456. NS_MOCK(hid_serv_responsible_for_desc_id);
  457. rend_cache_init();
  458. t = time(NULL);
  459. create_descriptor(&generated, &service_id, 3);
  460. generated->timestamp = t + RECENT_TIME;
  461. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  462. REND_NO_AUTH, NULL, NULL);
  463. desc_holder_newer = ((rend_encoded_v2_service_descriptor_t *)
  464. smartlist_get(descs, 0));
  465. smartlist_set(descs, 0, NULL);
  466. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  467. rend_encoded_v2_service_descriptor_free(d));
  468. smartlist_free(descs);
  469. descs = smartlist_new();
  470. generated->timestamp = (t + RECENT_TIME) - 20;
  471. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  472. REND_NO_AUTH, NULL, NULL);
  473. desc_holder_older = ((rend_encoded_v2_service_descriptor_t *)
  474. smartlist_get(descs, 0));
  475. smartlist_set(descs, 0, NULL);
  476. // Test when we have a newer descriptor stored
  477. mock_routerinfo = tor_malloc(sizeof(routerinfo_t));
  478. hid_serv_responsible_for_desc_id_response = 1;
  479. rend_cache_store_v2_desc_as_dir(desc_holder_newer->desc_str);
  480. ret = rend_cache_store_v2_desc_as_dir(desc_holder_older->desc_str);
  481. tt_int_op(ret, OP_EQ, RCS_OKAY);
  482. // Test when we have an old descriptor stored
  483. rend_cache_purge();
  484. rend_cache_store_v2_desc_as_dir(desc_holder_older->desc_str);
  485. ret = rend_cache_store_v2_desc_as_dir(desc_holder_newer->desc_str);
  486. tt_int_op(ret, OP_EQ, RCS_OKAY);
  487. done:
  488. NS_UNMOCK(router_get_my_routerinfo);
  489. NS_UNMOCK(hid_serv_responsible_for_desc_id);
  490. rend_cache_free_all();
  491. rend_service_descriptor_free(generated);
  492. tor_free(service_id);
  493. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  494. rend_encoded_v2_service_descriptor_free(d));
  495. smartlist_free(descs);
  496. rend_encoded_v2_service_descriptor_free(desc_holder_newer);
  497. rend_encoded_v2_service_descriptor_free(desc_holder_older);
  498. tor_free(mock_routerinfo);
  499. }
  500. static void
  501. test_rend_cache_store_v2_desc_as_dir_with_different_content(void *data)
  502. {
  503. (void)data;
  504. rend_cache_store_status_t ret;
  505. rend_service_descriptor_t *generated = NULL;
  506. smartlist_t *descs = smartlist_new();
  507. time_t t;
  508. char *service_id = NULL;
  509. rend_encoded_v2_service_descriptor_t *desc_holder_one = NULL;
  510. rend_encoded_v2_service_descriptor_t *desc_holder_two = NULL;
  511. NS_MOCK(router_get_my_routerinfo);
  512. NS_MOCK(hid_serv_responsible_for_desc_id);
  513. rend_cache_init();
  514. t = time(NULL);
  515. create_descriptor(&generated, &service_id, 3);
  516. generated->timestamp = t + RECENT_TIME;
  517. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  518. REND_NO_AUTH, NULL, NULL);
  519. desc_holder_one = ((rend_encoded_v2_service_descriptor_t *)
  520. smartlist_get(descs, 0));
  521. smartlist_set(descs, 0, NULL);
  522. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  523. rend_encoded_v2_service_descriptor_free(d));
  524. smartlist_free(descs);
  525. descs = smartlist_new();
  526. generated->timestamp = t + RECENT_TIME;
  527. generated->protocols = 41;
  528. rend_encode_v2_descriptors(descs, generated, t + RECENT_TIME, 0,
  529. REND_NO_AUTH, NULL, NULL);
  530. desc_holder_two = ((rend_encoded_v2_service_descriptor_t *)
  531. smartlist_get(descs, 0));
  532. smartlist_set(descs, 0, NULL);
  533. // Test when we have another descriptor stored, with a different descriptor
  534. mock_routerinfo = tor_malloc(sizeof(routerinfo_t));
  535. hid_serv_responsible_for_desc_id_response = 1;
  536. rend_cache_store_v2_desc_as_dir(desc_holder_one->desc_str);
  537. ret = rend_cache_store_v2_desc_as_dir(desc_holder_two->desc_str);
  538. tt_int_op(ret, OP_EQ, RCS_OKAY);
  539. done:
  540. NS_UNMOCK(router_get_my_routerinfo);
  541. NS_UNMOCK(hid_serv_responsible_for_desc_id);
  542. rend_cache_free_all();
  543. rend_service_descriptor_free(generated);
  544. tor_free(service_id);
  545. SMARTLIST_FOREACH(descs, rend_encoded_v2_service_descriptor_t *, d,
  546. rend_encoded_v2_service_descriptor_free(d));
  547. smartlist_free(descs);
  548. rend_encoded_v2_service_descriptor_free(desc_holder_one);
  549. rend_encoded_v2_service_descriptor_free(desc_holder_two);
  550. }
  551. #undef NS_SUBMODULE
  552. static void
  553. test_rend_cache_init(void *data)
  554. {
  555. (void)data;
  556. tt_assert_msg(!rend_cache, "rend_cache should be NULL when starting");
  557. tt_assert_msg(!rend_cache_v2_dir, "rend_cache_v2_dir should be NULL "
  558. "when starting");
  559. tt_assert_msg(!rend_cache_failure, "rend_cache_failure should be NULL when "
  560. "starting");
  561. rend_cache_init();
  562. tt_assert_msg(rend_cache, "rend_cache should not be NULL after initing");
  563. tt_assert_msg(rend_cache_v2_dir, "rend_cache_v2_dir should not be NULL "
  564. "after initing");
  565. tt_assert_msg(rend_cache_failure, "rend_cache_failure should not be NULL "
  566. "after initing");
  567. tt_int_op(strmap_size(rend_cache), OP_EQ, 0);
  568. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 0);
  569. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 0);
  570. done:
  571. rend_cache_free_all();
  572. }
  573. static void
  574. test_rend_cache_decrement_allocation(void *data)
  575. {
  576. (void)data;
  577. // Test when the cache has enough allocations
  578. rend_cache_total_allocation = 10;
  579. rend_cache_decrement_allocation(3);
  580. tt_int_op(rend_cache_total_allocation, OP_EQ, 7);
  581. // Test when there are not enough allocations
  582. rend_cache_total_allocation = 1;
  583. rend_cache_decrement_allocation(2);
  584. tt_int_op(rend_cache_total_allocation, OP_EQ, 0);
  585. // And again
  586. rend_cache_decrement_allocation(2);
  587. tt_int_op(rend_cache_total_allocation, OP_EQ, 0);
  588. done:
  589. (void)0;
  590. }
  591. static void
  592. test_rend_cache_increment_allocation(void *data)
  593. {
  594. (void)data;
  595. // Test when the cache is not overflowing
  596. rend_cache_total_allocation = 5;
  597. rend_cache_increment_allocation(3);
  598. tt_int_op(rend_cache_total_allocation, OP_EQ, 8);
  599. // Test when there are too many allocations
  600. rend_cache_total_allocation = SIZE_MAX-1;
  601. rend_cache_increment_allocation(2);
  602. tt_u64_op(rend_cache_total_allocation, OP_EQ, SIZE_MAX);
  603. // And again
  604. rend_cache_increment_allocation(2);
  605. tt_u64_op(rend_cache_total_allocation, OP_EQ, SIZE_MAX);
  606. done:
  607. (void)0;
  608. }
  609. static void
  610. test_rend_cache_failure_intro_entry_new(void *data)
  611. {
  612. time_t now;
  613. rend_cache_failure_intro_t *entry;
  614. rend_intro_point_failure_t failure;
  615. (void)data;
  616. failure = INTRO_POINT_FAILURE_TIMEOUT;
  617. now = time(NULL);
  618. entry = rend_cache_failure_intro_entry_new(failure);
  619. tt_int_op(entry->failure_type, OP_EQ, INTRO_POINT_FAILURE_TIMEOUT);
  620. tt_int_op(entry->created_ts, OP_GE, now-5);
  621. tt_int_op(entry->created_ts, OP_LE, now+5);
  622. done:
  623. tor_free(entry);
  624. }
  625. static void
  626. test_rend_cache_failure_intro_lookup(void *data)
  627. {
  628. (void)data;
  629. int ret;
  630. rend_cache_failure_t *failure;
  631. rend_cache_failure_intro_t *ip;
  632. rend_cache_failure_intro_t *entry;
  633. const char key_ip_one[DIGEST_LEN] = "ip1";
  634. const char key_ip_two[DIGEST_LEN] = "ip2";
  635. const char key_foo[DIGEST_LEN] = "foo1";
  636. rend_cache_init();
  637. failure = rend_cache_failure_entry_new();
  638. ip = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  639. digestmap_set(failure->intro_failures, key_ip_one, ip);
  640. strmap_set_lc(rend_cache_failure, "foo1", failure);
  641. // Test not found
  642. ret = cache_failure_intro_lookup((const uint8_t *) key_foo, "foo2", NULL);
  643. tt_int_op(ret, OP_EQ, 0);
  644. // Test found with no intro failures in it
  645. ret = cache_failure_intro_lookup((const uint8_t *) key_ip_two, "foo1", NULL);
  646. tt_int_op(ret, OP_EQ, 0);
  647. // Test found
  648. ret = cache_failure_intro_lookup((const uint8_t *) key_ip_one, "foo1", NULL);
  649. tt_int_op(ret, OP_EQ, 1);
  650. // Test found and asking for entry
  651. cache_failure_intro_lookup((const uint8_t *) key_ip_one, "foo1", &entry);
  652. tt_assert(entry);
  653. tt_assert(entry == ip);
  654. done:
  655. rend_cache_free_all();
  656. }
  657. static void
  658. test_rend_cache_clean(void *data)
  659. {
  660. rend_cache_entry_t *one, *two;
  661. rend_service_descriptor_t *desc_one, *desc_two;
  662. strmap_iter_t *iter = NULL;
  663. const char *key;
  664. void *val;
  665. (void)data;
  666. rend_cache_init();
  667. // Test with empty rendcache
  668. rend_cache_clean(time(NULL), REND_CACHE_TYPE_CLIENT);
  669. tt_int_op(strmap_size(rend_cache), OP_EQ, 0);
  670. // Test with two old entries
  671. one = tor_malloc_zero(sizeof(rend_cache_entry_t));
  672. two = tor_malloc_zero(sizeof(rend_cache_entry_t));
  673. desc_one = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  674. desc_two = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  675. one->parsed = desc_one;
  676. two->parsed = desc_two;
  677. desc_one->timestamp = time(NULL) + TIME_IN_THE_PAST;
  678. desc_two->timestamp = (time(NULL) + TIME_IN_THE_PAST) - 10;
  679. desc_one->pk = pk_generate(0);
  680. desc_two->pk = pk_generate(1);
  681. strmap_set_lc(rend_cache, "foo1", one);
  682. strmap_set_lc(rend_cache, "foo2", two);
  683. rend_cache_clean(time(NULL), REND_CACHE_TYPE_CLIENT);
  684. tt_int_op(strmap_size(rend_cache), OP_EQ, 0);
  685. // Test with one old entry and one newer entry
  686. one = tor_malloc_zero(sizeof(rend_cache_entry_t));
  687. two = tor_malloc_zero(sizeof(rend_cache_entry_t));
  688. desc_one = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  689. desc_two = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  690. one->parsed = desc_one;
  691. two->parsed = desc_two;
  692. desc_one->timestamp = (time(NULL) + TIME_IN_THE_PAST) - 10;
  693. desc_two->timestamp = time(NULL) - 100;
  694. desc_one->pk = pk_generate(0);
  695. desc_two->pk = pk_generate(1);
  696. strmap_set_lc(rend_cache, "foo1", one);
  697. strmap_set_lc(rend_cache, "foo2", two);
  698. rend_cache_clean(time(NULL), REND_CACHE_TYPE_CLIENT);
  699. tt_int_op(strmap_size(rend_cache), OP_EQ, 1);
  700. iter = strmap_iter_init(rend_cache);
  701. strmap_iter_get(iter, &key, &val);
  702. tt_str_op(key, OP_EQ, "foo2");
  703. done:
  704. rend_cache_free_all();
  705. }
  706. static void
  707. test_rend_cache_failure_entry_new(void *data)
  708. {
  709. rend_cache_failure_t *failure;
  710. (void)data;
  711. failure = rend_cache_failure_entry_new();
  712. tt_assert(failure);
  713. tt_int_op(digestmap_size(failure->intro_failures), OP_EQ, 0);
  714. done:
  715. rend_cache_failure_entry_free(failure);
  716. }
  717. static void
  718. test_rend_cache_failure_entry_free(void *data)
  719. {
  720. (void)data;
  721. // Test that it can deal with a NULL argument
  722. rend_cache_failure_entry_free(NULL);
  723. /* done: */
  724. /* (void)0; */
  725. }
  726. static void
  727. test_rend_cache_failure_clean(void *data)
  728. {
  729. rend_cache_failure_t *failure;
  730. rend_cache_failure_intro_t *ip_one, *ip_two;
  731. const char key_one[DIGEST_LEN] = "ip1";
  732. const char key_two[DIGEST_LEN] = "ip2";
  733. (void)data;
  734. rend_cache_init();
  735. // Test with empty failure cache
  736. rend_cache_failure_clean(time(NULL));
  737. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 0);
  738. // Test with one empty failure entry
  739. failure = rend_cache_failure_entry_new();
  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, 0);
  743. // Test with one new intro point
  744. failure = rend_cache_failure_entry_new();
  745. ip_one = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  746. digestmap_set(failure->intro_failures, key_one, ip_one);
  747. strmap_set_lc(rend_cache_failure, "foo1", failure);
  748. rend_cache_failure_clean(time(NULL));
  749. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 1);
  750. // Test with one old intro point
  751. rend_cache_failure_purge();
  752. failure = rend_cache_failure_entry_new();
  753. ip_one = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  754. ip_one->created_ts = time(NULL) - 7*60;
  755. digestmap_set(failure->intro_failures, key_one, ip_one);
  756. strmap_set_lc(rend_cache_failure, "foo1", failure);
  757. rend_cache_failure_clean(time(NULL));
  758. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 0);
  759. // Test with one old intro point and one new one
  760. rend_cache_failure_purge();
  761. failure = rend_cache_failure_entry_new();
  762. ip_one = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  763. ip_one->created_ts = time(NULL) - 7*60;
  764. digestmap_set(failure->intro_failures, key_one, ip_one);
  765. ip_two = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  766. ip_two->created_ts = time(NULL) - 2*60;
  767. digestmap_set(failure->intro_failures, key_two, ip_two);
  768. strmap_set_lc(rend_cache_failure, "foo1", failure);
  769. rend_cache_failure_clean(time(NULL));
  770. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 1);
  771. tt_int_op(digestmap_size(failure->intro_failures), OP_EQ, 1);
  772. done:
  773. rend_cache_free_all();
  774. }
  775. static void
  776. test_rend_cache_failure_remove(void *data)
  777. {
  778. rend_service_descriptor_t *desc;
  779. (void)data;
  780. rend_cache_init();
  781. // Test that it deals well with a NULL desc
  782. rend_cache_failure_remove(NULL);
  783. // Test a descriptor that isn't in the cache
  784. desc = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  785. desc->pk = pk_generate(0);
  786. rend_cache_failure_remove(desc);
  787. // There seems to not exist any way of getting rend_cache_failure_remove()
  788. // to fail because of a problem with rend_get_service_id from here
  789. rend_cache_free_all();
  790. rend_service_descriptor_free(desc);
  791. /* done: */
  792. /* (void)0; */
  793. }
  794. static void
  795. test_rend_cache_free_all(void *data)
  796. {
  797. rend_cache_failure_t *failure;
  798. rend_cache_entry_t *one;
  799. rend_service_descriptor_t *desc_one;
  800. (void)data;
  801. rend_cache_init();
  802. failure = rend_cache_failure_entry_new();
  803. strmap_set_lc(rend_cache_failure, "foo1", failure);
  804. one = tor_malloc_zero(sizeof(rend_cache_entry_t));
  805. desc_one = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  806. one->parsed = desc_one;
  807. desc_one->timestamp = time(NULL) + TIME_IN_THE_PAST;
  808. desc_one->pk = pk_generate(0);
  809. strmap_set_lc(rend_cache, "foo1", one);
  810. rend_cache_free_all();
  811. tt_assert(!rend_cache);
  812. tt_assert(!rend_cache_v2_dir);
  813. tt_assert(!rend_cache_failure);
  814. tt_assert(!rend_cache_total_allocation);
  815. done:
  816. rend_cache_free_all();
  817. }
  818. static void
  819. test_rend_cache_entry_free(void *data)
  820. {
  821. (void)data;
  822. rend_cache_entry_t *e;
  823. // Handles NULL correctly
  824. rend_cache_entry_free(NULL);
  825. // Handles NULL descriptor correctly
  826. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  827. rend_cache_entry_free(e);
  828. // Handles non-NULL descriptor correctly
  829. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  830. e->desc = (char *)malloc(10);
  831. rend_cache_entry_free(e);
  832. /* done: */
  833. /* (void)0; */
  834. }
  835. static void
  836. test_rend_cache_purge(void *data)
  837. {
  838. (void)data;
  839. // Deals with a NULL rend_cache
  840. rend_cache_purge();
  841. tt_assert(rend_cache);
  842. tt_assert(strmap_size(rend_cache) == 0);
  843. // Deals with existing rend_cache
  844. rend_cache_free_all();
  845. rend_cache_init();
  846. tt_assert(rend_cache);
  847. tt_assert(strmap_size(rend_cache) == 0);
  848. rend_cache_purge();
  849. tt_assert(rend_cache);
  850. tt_assert(strmap_size(rend_cache) == 0);
  851. done:
  852. rend_cache_free_all();
  853. }
  854. static void
  855. test_rend_cache_failure_intro_add(void *data)
  856. {
  857. (void)data;
  858. rend_cache_failure_t *fail_entry;
  859. rend_cache_failure_intro_t *entry;
  860. const char identity[DIGEST_LEN] = "foo1";
  861. rend_cache_init();
  862. // Adds non-existing entry
  863. cache_failure_intro_add((const uint8_t *) identity, "foo2",
  864. INTRO_POINT_FAILURE_TIMEOUT);
  865. fail_entry = strmap_get_lc(rend_cache_failure, "foo2");
  866. tt_assert(fail_entry);
  867. tt_int_op(digestmap_size(fail_entry->intro_failures), OP_EQ, 1);
  868. entry = digestmap_get(fail_entry->intro_failures, identity);
  869. tt_assert(entry);
  870. // Adds existing entry
  871. cache_failure_intro_add((const uint8_t *) identity, "foo2",
  872. INTRO_POINT_FAILURE_TIMEOUT);
  873. fail_entry = strmap_get_lc(rend_cache_failure, "foo2");
  874. tt_assert(fail_entry);
  875. tt_int_op(digestmap_size(fail_entry->intro_failures), OP_EQ, 1);
  876. entry = digestmap_get(fail_entry->intro_failures, identity);
  877. tt_assert(entry);
  878. done:
  879. rend_cache_free_all();
  880. }
  881. static void
  882. test_rend_cache_intro_failure_note(void *data)
  883. {
  884. (void)data;
  885. rend_cache_failure_t *fail_entry;
  886. rend_cache_failure_intro_t *entry;
  887. const char key[DIGEST_LEN] = "foo1";
  888. rend_cache_init();
  889. // Test not found
  890. rend_cache_intro_failure_note(INTRO_POINT_FAILURE_TIMEOUT,
  891. (const uint8_t *) key, "foo2");
  892. fail_entry = strmap_get_lc(rend_cache_failure, "foo2");
  893. tt_assert(fail_entry);
  894. tt_int_op(digestmap_size(fail_entry->intro_failures), OP_EQ, 1);
  895. entry = digestmap_get(fail_entry->intro_failures, key);
  896. tt_assert(entry);
  897. tt_int_op(entry->failure_type, OP_EQ, INTRO_POINT_FAILURE_TIMEOUT);
  898. // Test found
  899. rend_cache_intro_failure_note(INTRO_POINT_FAILURE_UNREACHABLE,
  900. (const uint8_t *) key, "foo2");
  901. tt_int_op(entry->failure_type, OP_EQ, INTRO_POINT_FAILURE_UNREACHABLE);
  902. done:
  903. rend_cache_free_all();
  904. }
  905. #define NS_SUBMODULE clean_v2_descs_as_dir
  906. NS_DECL(int, hid_serv_responsible_for_desc_id, (const char *id));
  907. static int
  908. NS(hid_serv_responsible_for_desc_id)(const char *id)
  909. {
  910. (void)id;
  911. return hid_serv_responsible_for_desc_id_response;
  912. }
  913. static void
  914. test_rend_cache_clean_v2_descs_as_dir(void *data)
  915. {
  916. rend_cache_entry_t *e;
  917. time_t now;
  918. rend_service_descriptor_t *desc;
  919. now = time(NULL);
  920. const char key[DIGEST_LEN] = "abcde";
  921. (void)data;
  922. NS_MOCK(hid_serv_responsible_for_desc_id);
  923. rend_cache_init();
  924. // Test running with an empty cache
  925. rend_cache_clean_v2_descs_as_dir(now, 0);
  926. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 0);
  927. // Test with only one new entry
  928. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  929. e->last_served = now;
  930. desc = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  931. desc->timestamp = now;
  932. desc->pk = pk_generate(0);
  933. e->parsed = desc;
  934. digestmap_set(rend_cache_v2_dir, key, e);
  935. hid_serv_responsible_for_desc_id_response = 1;
  936. rend_cache_clean_v2_descs_as_dir(now, 0);
  937. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 1);
  938. // Test with one old entry
  939. desc->timestamp = now - (REND_CACHE_MAX_AGE + REND_CACHE_MAX_SKEW + 1000);
  940. rend_cache_clean_v2_descs_as_dir(now, 0);
  941. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 0);
  942. // Test with one entry that is not under the responsibility of this
  943. // hidden service
  944. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  945. e->last_served = now;
  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, key, e);
  951. hid_serv_responsible_for_desc_id_response = 0;
  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 with one entry that has an old last served
  955. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  956. e->last_served = now - (REND_CACHE_MAX_AGE + REND_CACHE_MAX_SKEW + 1000);
  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, key, e);
  962. hid_serv_responsible_for_desc_id_response = 1;
  963. rend_cache_clean_v2_descs_as_dir(now, 0);
  964. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 0);
  965. // Test a run through asking for a large force_remove
  966. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  967. e->last_served = now;
  968. desc = tor_malloc_zero(sizeof(rend_service_descriptor_t));
  969. desc->timestamp = now;
  970. desc->pk = pk_generate(0);
  971. e->parsed = desc;
  972. digestmap_set(rend_cache_v2_dir, key, e);
  973. hid_serv_responsible_for_desc_id_response = 1;
  974. rend_cache_clean_v2_descs_as_dir(now, 20000);
  975. tt_int_op(digestmap_size(rend_cache_v2_dir), OP_EQ, 1);
  976. done:
  977. NS_UNMOCK(hid_serv_responsible_for_desc_id);
  978. rend_cache_free_all();
  979. }
  980. #undef NS_SUBMODULE
  981. static void
  982. test_rend_cache_entry_allocation(void *data)
  983. {
  984. (void)data;
  985. size_t ret;
  986. rend_cache_entry_t *e = NULL;
  987. // Handles a null argument
  988. ret = rend_cache_entry_allocation(NULL);
  989. tt_int_op(ret, OP_EQ, 0);
  990. // Handles a non-null argument
  991. e = tor_malloc_zero(sizeof(rend_cache_entry_t));
  992. ret = rend_cache_entry_allocation(e);
  993. tt_int_op(ret, OP_GT, sizeof(rend_cache_entry_t));
  994. done:
  995. tor_free(e);
  996. }
  997. static void
  998. test_rend_cache_failure_intro_entry_free(void *data)
  999. {
  1000. (void)data;
  1001. rend_cache_failure_intro_t *entry;
  1002. // Handles a null argument
  1003. rend_cache_failure_intro_entry_free(NULL);
  1004. // Handles a non-null argument
  1005. entry = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  1006. rend_cache_failure_intro_entry_free(entry);
  1007. }
  1008. static void
  1009. test_rend_cache_failure_purge(void *data)
  1010. {
  1011. (void)data;
  1012. // Handles a null failure cache
  1013. strmap_free(rend_cache_failure, rend_cache_failure_entry_free_);
  1014. rend_cache_failure = NULL;
  1015. rend_cache_failure_purge();
  1016. tt_ptr_op(rend_cache_failure, OP_NE, NULL);
  1017. tt_int_op(strmap_size(rend_cache_failure), OP_EQ, 0);
  1018. done:
  1019. rend_cache_free_all();
  1020. }
  1021. static void
  1022. test_rend_cache_validate_intro_point_failure(void *data)
  1023. {
  1024. (void)data;
  1025. rend_service_descriptor_t *desc = NULL;
  1026. char *service_id = NULL;
  1027. rend_intro_point_t *intro = NULL;
  1028. const uint8_t *identity = NULL;
  1029. rend_cache_failure_t *failure;
  1030. rend_cache_failure_intro_t *ip;
  1031. rend_cache_init();
  1032. create_descriptor(&desc, &service_id, 3);
  1033. desc->timestamp = time(NULL) + RECENT_TIME;
  1034. intro = (rend_intro_point_t *)smartlist_get(desc->intro_nodes, 0);
  1035. identity = (uint8_t *) intro->extend_info->identity_digest;
  1036. failure = rend_cache_failure_entry_new();
  1037. ip = rend_cache_failure_intro_entry_new(INTRO_POINT_FAILURE_TIMEOUT);
  1038. digestmap_set(failure->intro_failures, (char *)identity, ip);
  1039. strmap_set_lc(rend_cache_failure, service_id, failure);
  1040. // Test when we have an intro point in our cache
  1041. validate_intro_point_failure(desc, service_id);
  1042. tt_int_op(smartlist_len(desc->intro_nodes), OP_EQ, 2);
  1043. done:
  1044. rend_cache_free_all();
  1045. rend_service_descriptor_free(desc);
  1046. tor_free(service_id);
  1047. }
  1048. struct testcase_t rend_cache_tests[] = {
  1049. { "init", test_rend_cache_init, 0, NULL, NULL },
  1050. { "decrement_allocation", test_rend_cache_decrement_allocation, 0,
  1051. NULL, NULL },
  1052. { "increment_allocation", test_rend_cache_increment_allocation, 0,
  1053. NULL, NULL },
  1054. { "clean", test_rend_cache_clean, TT_FORK, NULL, NULL },
  1055. { "clean_v2_descs_as_dir", test_rend_cache_clean_v2_descs_as_dir, 0,
  1056. NULL, NULL },
  1057. { "entry_allocation", test_rend_cache_entry_allocation, 0, NULL, NULL },
  1058. { "entry_free", test_rend_cache_entry_free, 0, NULL, NULL },
  1059. { "failure_intro_entry_free", test_rend_cache_failure_intro_entry_free, 0,
  1060. NULL, NULL },
  1061. { "free_all", test_rend_cache_free_all, 0, NULL, NULL },
  1062. { "purge", test_rend_cache_purge, 0, NULL, NULL },
  1063. { "failure_clean", test_rend_cache_failure_clean, 0, NULL, NULL },
  1064. { "failure_entry_new", test_rend_cache_failure_entry_new, 0, NULL, NULL },
  1065. { "failure_entry_free", test_rend_cache_failure_entry_free, 0, NULL, NULL },
  1066. { "failure_intro_add", test_rend_cache_failure_intro_add, 0, NULL, NULL },
  1067. { "failure_intro_entry_new", test_rend_cache_failure_intro_entry_new, 0,
  1068. NULL, NULL },
  1069. { "failure_intro_lookup", test_rend_cache_failure_intro_lookup, 0,
  1070. NULL, NULL },
  1071. { "failure_purge", test_rend_cache_failure_purge, 0, NULL, NULL },
  1072. { "failure_remove", test_rend_cache_failure_remove, 0, NULL, NULL },
  1073. { "intro_failure_note", test_rend_cache_intro_failure_note, 0, NULL, NULL },
  1074. { "lookup", test_rend_cache_lookup_entry, 0, NULL, NULL },
  1075. { "lookup_v2_desc_as_dir", test_rend_cache_lookup_v2_desc_as_dir, 0,
  1076. NULL, NULL },
  1077. { "store_v2_desc_as_client", test_rend_cache_store_v2_desc_as_client, 0,
  1078. NULL, NULL },
  1079. { "store_v2_desc_as_client_with_different_time",
  1080. test_rend_cache_store_v2_desc_as_client_with_different_time, 0,
  1081. NULL, NULL },
  1082. { "store_v2_desc_as_dir", test_rend_cache_store_v2_desc_as_dir, 0,
  1083. NULL, NULL },
  1084. { "store_v2_desc_as_dir_with_different_time",
  1085. test_rend_cache_store_v2_desc_as_dir_with_different_time, 0, NULL, NULL },
  1086. { "store_v2_desc_as_dir_with_different_content",
  1087. test_rend_cache_store_v2_desc_as_dir_with_different_content, 0,
  1088. NULL, NULL },
  1089. { "validate_intro_point_failure",
  1090. test_rend_cache_validate_intro_point_failure, 0, NULL, NULL },
  1091. END_OF_TESTCASES
  1092. };