rephist.c 74 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383
  1. /* Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
  2. * Copyright (c) 2007-2008, The Tor Project, Inc. */
  3. /* See LICENSE for licensing information */
  4. /**
  5. * \file rephist.c
  6. * \brief Basic history and "reputation" functionality to remember
  7. * which servers have worked in the past, how much bandwidth we've
  8. * been using, which ports we tend to want, and so on.
  9. **/
  10. #include "or.h"
  11. #include "ht.h"
  12. static void bw_arrays_init(void);
  13. static void predicted_ports_init(void);
  14. static void hs_usage_init(void);
  15. /** Total number of bytes currently allocated in fields used by rephist.c. */
  16. uint64_t rephist_total_alloc=0;
  17. /** Number of or_history_t objects currently allocated. */
  18. uint32_t rephist_total_num=0;
  19. /** If the total weighted run count of all runs for a router ever falls
  20. * below this amount, the router can be treated as having 0 MTBF. */
  21. #define STABILITY_EPSILON 0.0001
  22. /** Value by which to discount all old intervals for MTBF purposses. This
  23. * is compounded every STABILITY_INTERVAL. */
  24. #define STABILITY_ALPHA 0.95
  25. /** Interval at which to discount all old intervals for MTBF purposes. */
  26. #define STABILITY_INTERVAL (12*60*60)
  27. /* (This combination of ALPHA, INTERVAL, and EPSILON makes it so that an
  28. * interval that just ended counts twice as much as one that ended a week ago,
  29. * 20X as much as one that ended a month ago, and routers that have had no
  30. * uptime data for about half a year will get forgotten.) */
  31. /** History of an OR-\>OR link. */
  32. typedef struct link_history_t {
  33. /** When did we start tracking this list? */
  34. time_t since;
  35. /** When did we most recently note a change to this link */
  36. time_t changed;
  37. /** How many times did extending from OR1 to OR2 succeed? */
  38. unsigned long n_extend_ok;
  39. /** How many times did extending from OR1 to OR2 fail? */
  40. unsigned long n_extend_fail;
  41. } link_history_t;
  42. /** History of an OR. */
  43. typedef struct or_history_t {
  44. /** When did we start tracking this OR? */
  45. time_t since;
  46. /** When did we most recently note a change to this OR? */
  47. time_t changed;
  48. /** How many times did we successfully connect? */
  49. unsigned long n_conn_ok;
  50. /** How many times did we try to connect and fail?*/
  51. unsigned long n_conn_fail;
  52. /** How many seconds have we been connected to this OR before
  53. * 'up_since'? */
  54. unsigned long uptime;
  55. /** How many seconds have we been unable to connect to this OR before
  56. * 'down_since'? */
  57. unsigned long downtime;
  58. /** If nonzero, we have been connected since this time. */
  59. time_t up_since;
  60. /** If nonzero, we have been unable to connect since this time. */
  61. time_t down_since;
  62. /* === For MTBF tracking: */
  63. /** Weighted sum total of all times that this router has been online.
  64. */
  65. unsigned long weighted_run_length;
  66. /** If the router is now online (according to stability-checking rules),
  67. * when did it come online? */
  68. time_t start_of_run;
  69. /** Sum of weights for runs in weighted_run_length. */
  70. double total_run_weights;
  71. /* === For fractional uptime tracking: */
  72. time_t start_of_downtime;
  73. unsigned long weighted_uptime;
  74. unsigned long total_weighted_time;
  75. /** Map from hex OR2 identity digest to a link_history_t for the link
  76. * from this OR to OR2. */
  77. digestmap_t *link_history_map;
  78. } or_history_t;
  79. /** When did we last multiply all routers' weighted_run_length and
  80. * total_run_weights by STABILITY_ALPHA? */
  81. static time_t stability_last_downrated = 0;
  82. /** */
  83. static time_t started_tracking_stability = 0;
  84. /** Map from hex OR identity digest to or_history_t. */
  85. static digestmap_t *history_map = NULL;
  86. /** Return the or_history_t for the OR with identity digest <b>id</b>,
  87. * creating it if necessary. */
  88. static or_history_t *
  89. get_or_history(const char* id)
  90. {
  91. or_history_t *hist;
  92. if (tor_mem_is_zero(id, DIGEST_LEN))
  93. return NULL;
  94. hist = digestmap_get(history_map, id);
  95. if (!hist) {
  96. hist = tor_malloc_zero(sizeof(or_history_t));
  97. rephist_total_alloc += sizeof(or_history_t);
  98. rephist_total_num++;
  99. hist->link_history_map = digestmap_new();
  100. hist->since = hist->changed = time(NULL);
  101. digestmap_set(history_map, id, hist);
  102. }
  103. return hist;
  104. }
  105. /** Return the link_history_t for the link from the first named OR to
  106. * the second, creating it if necessary. (ORs are identified by
  107. * identity digest.)
  108. */
  109. static link_history_t *
  110. get_link_history(const char *from_id, const char *to_id)
  111. {
  112. or_history_t *orhist;
  113. link_history_t *lhist;
  114. orhist = get_or_history(from_id);
  115. if (!orhist)
  116. return NULL;
  117. if (tor_mem_is_zero(to_id, DIGEST_LEN))
  118. return NULL;
  119. lhist = (link_history_t*) digestmap_get(orhist->link_history_map, to_id);
  120. if (!lhist) {
  121. lhist = tor_malloc_zero(sizeof(link_history_t));
  122. rephist_total_alloc += sizeof(link_history_t);
  123. lhist->since = lhist->changed = time(NULL);
  124. digestmap_set(orhist->link_history_map, to_id, lhist);
  125. }
  126. return lhist;
  127. }
  128. /** Helper: free storage held by a single link history entry. */
  129. static void
  130. _free_link_history(void *val)
  131. {
  132. rephist_total_alloc -= sizeof(link_history_t);
  133. tor_free(val);
  134. }
  135. /** Helper: free storage held by a single OR history entry. */
  136. static void
  137. free_or_history(void *_hist)
  138. {
  139. or_history_t *hist = _hist;
  140. digestmap_free(hist->link_history_map, _free_link_history);
  141. rephist_total_alloc -= sizeof(or_history_t);
  142. rephist_total_num--;
  143. tor_free(hist);
  144. }
  145. /** Update an or_history_t object <b>hist</b> so that its uptime/downtime
  146. * count is up-to-date as of <b>when</b>.
  147. */
  148. static void
  149. update_or_history(or_history_t *hist, time_t when)
  150. {
  151. tor_assert(hist);
  152. if (hist->up_since) {
  153. tor_assert(!hist->down_since);
  154. hist->uptime += (when - hist->up_since);
  155. hist->up_since = when;
  156. } else if (hist->down_since) {
  157. hist->downtime += (when - hist->down_since);
  158. hist->down_since = when;
  159. }
  160. }
  161. /** Initialize the static data structures for tracking history. */
  162. void
  163. rep_hist_init(void)
  164. {
  165. history_map = digestmap_new();
  166. bw_arrays_init();
  167. predicted_ports_init();
  168. hs_usage_init();
  169. }
  170. /** Helper: note that we are no longer connected to the router with history
  171. * <b>hist</b>. If <b>failed</b>, the connection failed; otherwise, it was
  172. * closed correctly. */
  173. static void
  174. mark_or_down(or_history_t *hist, time_t when, int failed)
  175. {
  176. if (hist->up_since) {
  177. hist->uptime += (when - hist->up_since);
  178. hist->up_since = 0;
  179. }
  180. if (failed && !hist->down_since) {
  181. hist->down_since = when;
  182. }
  183. }
  184. /** Helper: note that we are connected to the router with history
  185. * <b>hist</b>. */
  186. static void
  187. mark_or_up(or_history_t *hist, time_t when)
  188. {
  189. if (hist->down_since) {
  190. hist->downtime += (when - hist->down_since);
  191. hist->down_since = 0;
  192. }
  193. if (!hist->up_since) {
  194. hist->up_since = when;
  195. }
  196. }
  197. /** Remember that an attempt to connect to the OR with identity digest
  198. * <b>id</b> failed at <b>when</b>.
  199. */
  200. void
  201. rep_hist_note_connect_failed(const char* id, time_t when)
  202. {
  203. or_history_t *hist;
  204. hist = get_or_history(id);
  205. if (!hist)
  206. return;
  207. ++hist->n_conn_fail;
  208. mark_or_down(hist, when, 1);
  209. hist->changed = when;
  210. }
  211. /** Remember that an attempt to connect to the OR with identity digest
  212. * <b>id</b> succeeded at <b>when</b>.
  213. */
  214. void
  215. rep_hist_note_connect_succeeded(const char* id, time_t when)
  216. {
  217. or_history_t *hist;
  218. hist = get_or_history(id);
  219. if (!hist)
  220. return;
  221. ++hist->n_conn_ok;
  222. mark_or_up(hist, when);
  223. hist->changed = when;
  224. }
  225. /** Remember that we intentionally closed our connection to the OR
  226. * with identity digest <b>id</b> at <b>when</b>.
  227. */
  228. void
  229. rep_hist_note_disconnect(const char* id, time_t when)
  230. {
  231. or_history_t *hist;
  232. hist = get_or_history(id);
  233. if (!hist)
  234. return;
  235. mark_or_down(hist, when, 0);
  236. hist->changed = when;
  237. }
  238. /** Remember that our connection to the OR with identity digest
  239. * <b>id</b> had an error and stopped working at <b>when</b>.
  240. */
  241. void
  242. rep_hist_note_connection_died(const char* id, time_t when)
  243. {
  244. or_history_t *hist;
  245. if (!id) {
  246. /* If conn has no identity, it didn't complete its handshake, or something
  247. * went wrong. Ignore it.
  248. */
  249. return;
  250. }
  251. hist = get_or_history(id);
  252. if (!hist)
  253. return;
  254. mark_or_down(hist, when, 1);
  255. hist->changed = when;
  256. }
  257. /** We have just decided that this router with identity digest <b>id</b> is
  258. * reachable, meaning we will give it a "Running" flag for the next while. */
  259. void
  260. rep_hist_note_router_reachable(const char *id, time_t when)
  261. {
  262. or_history_t *hist = get_or_history(id);
  263. int was_in_run = 1;
  264. char tbuf[ISO_TIME_LEN+1];
  265. tor_assert(hist);
  266. if (!started_tracking_stability)
  267. started_tracking_stability = time(NULL);
  268. if (!hist->start_of_run) {
  269. hist->start_of_run = when;
  270. was_in_run = 0;
  271. }
  272. if (hist->start_of_downtime) {
  273. long down_length;
  274. format_local_iso_time(tbuf, hist->start_of_downtime);
  275. log_info(LD_HIST, "Router %s is now Running; it had been down since %s.",
  276. hex_str(id, DIGEST_LEN), tbuf);
  277. if (was_in_run)
  278. log_info(LD_HIST, " (Paradoxically, it was already Running too.)");
  279. down_length = when - hist->start_of_downtime;
  280. hist->total_weighted_time += down_length;
  281. hist->start_of_downtime = 0;
  282. } else {
  283. format_local_iso_time(tbuf, hist->start_of_run);
  284. if (was_in_run)
  285. log_debug(LD_HIST, "Router %s is still Running; it has been Running "
  286. "since %s", hex_str(id, DIGEST_LEN), tbuf);
  287. else
  288. log_info(LD_HIST,"Router %s is now Running; it was previously untracked",
  289. hex_str(id, DIGEST_LEN));
  290. }
  291. }
  292. /** We have just decided that this router is unreachable, meaning
  293. * we are taking away its "Running" flag. */
  294. void
  295. rep_hist_note_router_unreachable(const char *id, time_t when)
  296. {
  297. or_history_t *hist = get_or_history(id);
  298. char tbuf[ISO_TIME_LEN+1];
  299. int was_running = 0;
  300. if (!started_tracking_stability)
  301. started_tracking_stability = time(NULL);
  302. tor_assert(hist);
  303. if (hist->start_of_run) {
  304. /*XXXX We could treat failed connections differently from failed
  305. * connect attempts. */
  306. long run_length = when - hist->start_of_run;
  307. format_local_iso_time(tbuf, hist->start_of_run);
  308. hist->weighted_run_length += run_length;
  309. hist->total_run_weights += 1.0;
  310. hist->start_of_run = 0;
  311. hist->weighted_uptime += run_length;
  312. hist->total_weighted_time += run_length;
  313. was_running = 1;
  314. log_info(LD_HIST, "Router %s is now non-Running: it had previously been "
  315. "Running since %s. Its total weighted uptime is %lu/%lu.",
  316. hex_str(id, DIGEST_LEN), tbuf, hist->weighted_uptime,
  317. hist->total_weighted_time);
  318. }
  319. if (!hist->start_of_downtime) {
  320. hist->start_of_downtime = when;
  321. if (!was_running)
  322. log_info(LD_HIST, "Router %s is now non-Running; it was previously "
  323. "untracked.", hex_str(id, DIGEST_LEN));
  324. } else {
  325. if (!was_running) {
  326. format_local_iso_time(tbuf, hist->start_of_downtime);
  327. log_info(LD_HIST, "Router %s is still non-Running; it has been "
  328. "non-Running since %s.", hex_str(id, DIGEST_LEN), tbuf);
  329. }
  330. }
  331. }
  332. /** Helper: Discount all old MTBF data, if it is time to do so. Return
  333. * the time at which we should next discount MTBF data. */
  334. time_t
  335. rep_hist_downrate_old_runs(time_t now)
  336. {
  337. digestmap_iter_t *orhist_it;
  338. const char *digest1;
  339. or_history_t *hist;
  340. void *hist_p;
  341. double alpha = 1.0;
  342. if (!history_map)
  343. history_map = digestmap_new();
  344. if (!stability_last_downrated)
  345. stability_last_downrated = now;
  346. if (stability_last_downrated + STABILITY_INTERVAL > now)
  347. return stability_last_downrated + STABILITY_INTERVAL;
  348. /* Okay, we should downrate the data. By how much? */
  349. while (stability_last_downrated + STABILITY_INTERVAL < now) {
  350. stability_last_downrated += STABILITY_INTERVAL;
  351. alpha *= STABILITY_ALPHA;
  352. }
  353. log_info(LD_HIST, "Discounting all old stability info by a factor of %lf",
  354. alpha);
  355. /* Multiply every w_r_l, t_r_w pair by alpha. */
  356. for (orhist_it = digestmap_iter_init(history_map);
  357. !digestmap_iter_done(orhist_it);
  358. orhist_it = digestmap_iter_next(history_map,orhist_it)) {
  359. digestmap_iter_get(orhist_it, &digest1, &hist_p);
  360. hist = hist_p;
  361. hist->weighted_run_length =
  362. (unsigned long)(hist->weighted_run_length * alpha);
  363. hist->total_run_weights *= alpha;
  364. hist->weighted_uptime = (unsigned long)(hist->weighted_uptime * alpha);
  365. hist->total_weighted_time = (unsigned long)
  366. (hist->total_weighted_time * alpha);
  367. }
  368. return stability_last_downrated + STABILITY_INTERVAL;
  369. }
  370. /** Helper: Return the weighted MTBF of the router with history <b>hist</b>. */
  371. static double
  372. get_stability(or_history_t *hist, time_t when)
  373. {
  374. unsigned long total = hist->weighted_run_length;
  375. double total_weights = hist->total_run_weights;
  376. if (hist->start_of_run) {
  377. /* We're currently in a run. Let total and total_weights hold the values
  378. * they would hold if the current run were to end now. */
  379. total += (when-hist->start_of_run);
  380. total_weights += 1.0;
  381. }
  382. if (total_weights < STABILITY_EPSILON) {
  383. /* Round down to zero, and avoid divide-by-zero. */
  384. return 0.0;
  385. }
  386. return total / total_weights;
  387. }
  388. /** Return the total amount of time we've been observing, with each run of
  389. * time downrated by the appropriate factor. */
  390. static long
  391. get_total_weighted_time(or_history_t *hist, time_t when)
  392. {
  393. long total = hist->total_weighted_time;
  394. if (hist->start_of_run) {
  395. total += (when - hist->start_of_run);
  396. } else if (hist->start_of_downtime) {
  397. total += (when - hist->start_of_downtime);
  398. }
  399. return total;
  400. }
  401. /** Helper: Return the weighted percent-of-time-online of the router with
  402. * history <b>hist</b>. */
  403. static double
  404. get_weighted_fractional_uptime(or_history_t *hist, time_t when)
  405. {
  406. unsigned long total = hist->total_weighted_time;
  407. unsigned long up = hist->weighted_uptime;
  408. if (hist->start_of_run) {
  409. long run_length = (when - hist->start_of_run);
  410. up += run_length;
  411. total += run_length;
  412. } else if (hist->start_of_downtime) {
  413. total += (when - hist->start_of_downtime);
  414. }
  415. if (!total) {
  416. /* Avoid calling anybody's uptime infinity (which should be impossible if
  417. * the code is working), or NaN (which can happen for any router we haven't
  418. * observed up or down yet). */
  419. return 0.0;
  420. }
  421. return ((double) up) / total;
  422. }
  423. /** Return an estimated MTBF for the router whose identity digest is
  424. * <b>id</b>. Return 0 if the router is unknown. */
  425. double
  426. rep_hist_get_stability(const char *id, time_t when)
  427. {
  428. or_history_t *hist = get_or_history(id);
  429. if (!hist)
  430. return 0.0;
  431. return get_stability(hist, when);
  432. }
  433. /** Return an estimated percent-of-time-online for the router whose identity
  434. * digest is <b>id</b>. Return 0 if the router is unknown. */
  435. double
  436. rep_hist_get_weighted_fractional_uptime(const char *id, time_t when)
  437. {
  438. or_history_t *hist = get_or_history(id);
  439. if (!hist)
  440. return 0.0;
  441. return get_weighted_fractional_uptime(hist, when);
  442. }
  443. /** Return a number representing how long we've known about the router whose
  444. * digest is <b>id</b>. Return 0 if the router is unknown.
  445. *
  446. * Be careful: this measure incresases monotonically as we know the router for
  447. * longer and longer, but it doesn't increase linearly.
  448. */
  449. long
  450. rep_hist_get_weighted_time_known(const char *id, time_t when)
  451. {
  452. or_history_t *hist = get_or_history(id);
  453. if (!hist)
  454. return 0;
  455. return get_total_weighted_time(hist, when);
  456. }
  457. /** Return true if we've been measuring MTBFs for long enough to
  458. * pronounce on Stability. */
  459. int
  460. rep_hist_have_measured_enough_stability(void)
  461. {
  462. /* XXXX021 This doesn't do so well when we change our opinion
  463. * as to whether we're tracking router stability. */
  464. return started_tracking_stability < time(NULL) - 4*60*60;
  465. }
  466. /** Remember that we successfully extended from the OR with identity
  467. * digest <b>from_id</b> to the OR with identity digest
  468. * <b>to_name</b>.
  469. */
  470. void
  471. rep_hist_note_extend_succeeded(const char *from_id, const char *to_id)
  472. {
  473. link_history_t *hist;
  474. /* log_fn(LOG_WARN, "EXTEND SUCCEEDED: %s->%s",from_name,to_name); */
  475. hist = get_link_history(from_id, to_id);
  476. if (!hist)
  477. return;
  478. ++hist->n_extend_ok;
  479. hist->changed = time(NULL);
  480. }
  481. /** Remember that we tried to extend from the OR with identity digest
  482. * <b>from_id</b> to the OR with identity digest <b>to_name</b>, but
  483. * failed.
  484. */
  485. void
  486. rep_hist_note_extend_failed(const char *from_id, const char *to_id)
  487. {
  488. link_history_t *hist;
  489. /* log_fn(LOG_WARN, "EXTEND FAILED: %s->%s",from_name,to_name); */
  490. hist = get_link_history(from_id, to_id);
  491. if (!hist)
  492. return;
  493. ++hist->n_extend_fail;
  494. hist->changed = time(NULL);
  495. }
  496. /** Log all the reliability data we have remembered, with the chosen
  497. * severity.
  498. */
  499. void
  500. rep_hist_dump_stats(time_t now, int severity)
  501. {
  502. digestmap_iter_t *lhist_it;
  503. digestmap_iter_t *orhist_it;
  504. const char *name1, *name2, *digest1, *digest2;
  505. char hexdigest1[HEX_DIGEST_LEN+1];
  506. or_history_t *or_history;
  507. link_history_t *link_history;
  508. void *or_history_p, *link_history_p;
  509. double uptime;
  510. char buffer[2048];
  511. size_t len;
  512. int ret;
  513. unsigned long upt, downt;
  514. routerinfo_t *r;
  515. rep_history_clean(now - get_options()->RephistTrackTime);
  516. log(severity, LD_HIST, "--------------- Dumping history information:");
  517. for (orhist_it = digestmap_iter_init(history_map);
  518. !digestmap_iter_done(orhist_it);
  519. orhist_it = digestmap_iter_next(history_map,orhist_it)) {
  520. double s;
  521. long stability;
  522. digestmap_iter_get(orhist_it, &digest1, &or_history_p);
  523. or_history = (or_history_t*) or_history_p;
  524. if ((r = router_get_by_digest(digest1)))
  525. name1 = r->nickname;
  526. else
  527. name1 = "(unknown)";
  528. base16_encode(hexdigest1, sizeof(hexdigest1), digest1, DIGEST_LEN);
  529. update_or_history(or_history, now);
  530. upt = or_history->uptime;
  531. downt = or_history->downtime;
  532. s = get_stability(or_history, now);
  533. stability = (long)s;
  534. if (upt+downt) {
  535. uptime = ((double)upt) / (upt+downt);
  536. } else {
  537. uptime=1.0;
  538. }
  539. log(severity, LD_HIST,
  540. "OR %s [%s]: %ld/%ld good connections; uptime %ld/%ld sec (%.2f%%); "
  541. "wmtbf %lu:%02lu:%02lu",
  542. name1, hexdigest1,
  543. or_history->n_conn_ok, or_history->n_conn_fail+or_history->n_conn_ok,
  544. upt, upt+downt, uptime*100.0,
  545. stability/3600, (stability/60)%60, stability%60);
  546. if (!digestmap_isempty(or_history->link_history_map)) {
  547. strlcpy(buffer, " Extend attempts: ", sizeof(buffer));
  548. len = strlen(buffer);
  549. for (lhist_it = digestmap_iter_init(or_history->link_history_map);
  550. !digestmap_iter_done(lhist_it);
  551. lhist_it = digestmap_iter_next(or_history->link_history_map,
  552. lhist_it)) {
  553. digestmap_iter_get(lhist_it, &digest2, &link_history_p);
  554. if ((r = router_get_by_digest(digest2)))
  555. name2 = r->nickname;
  556. else
  557. name2 = "(unknown)";
  558. link_history = (link_history_t*) link_history_p;
  559. ret = tor_snprintf(buffer+len, 2048-len, "%s(%ld/%ld); ", name2,
  560. link_history->n_extend_ok,
  561. link_history->n_extend_ok+link_history->n_extend_fail);
  562. if (ret<0)
  563. break;
  564. else
  565. len += ret;
  566. }
  567. log(severity, LD_HIST, "%s", buffer);
  568. }
  569. }
  570. }
  571. /** Remove history info for routers/links that haven't changed since
  572. * <b>before</b>.
  573. */
  574. void
  575. rep_history_clean(time_t before)
  576. {
  577. int authority = authdir_mode(get_options());
  578. or_history_t *or_history;
  579. link_history_t *link_history;
  580. void *or_history_p, *link_history_p;
  581. digestmap_iter_t *orhist_it, *lhist_it;
  582. const char *d1, *d2;
  583. orhist_it = digestmap_iter_init(history_map);
  584. while (!digestmap_iter_done(orhist_it)) {
  585. int remove;
  586. digestmap_iter_get(orhist_it, &d1, &or_history_p);
  587. or_history = or_history_p;
  588. remove = authority ? (or_history->total_run_weights < STABILITY_EPSILON &&
  589. !or_history->start_of_run)
  590. : (or_history->changed < before);
  591. if (remove) {
  592. orhist_it = digestmap_iter_next_rmv(history_map, orhist_it);
  593. free_or_history(or_history);
  594. continue;
  595. }
  596. for (lhist_it = digestmap_iter_init(or_history->link_history_map);
  597. !digestmap_iter_done(lhist_it); ) {
  598. digestmap_iter_get(lhist_it, &d2, &link_history_p);
  599. link_history = link_history_p;
  600. if (link_history->changed < before) {
  601. lhist_it = digestmap_iter_next_rmv(or_history->link_history_map,
  602. lhist_it);
  603. rephist_total_alloc -= sizeof(link_history_t);
  604. tor_free(link_history);
  605. continue;
  606. }
  607. lhist_it = digestmap_iter_next(or_history->link_history_map,lhist_it);
  608. }
  609. orhist_it = digestmap_iter_next(history_map, orhist_it);
  610. }
  611. }
  612. /** Write MTBF data to disk. Returns 0 on success, negative on failure. */
  613. int
  614. rep_hist_record_mtbf_data(void)
  615. {
  616. char time_buf[ISO_TIME_LEN+1];
  617. digestmap_iter_t *orhist_it;
  618. const char *digest;
  619. void *or_history_p;
  620. or_history_t *hist;
  621. open_file_t *open_file = NULL;
  622. FILE *f;
  623. {
  624. char *filename = get_datadir_fname("router-stability");
  625. f = start_writing_to_stdio_file(filename, OPEN_FLAGS_REPLACE|O_TEXT, 0600,
  626. &open_file);
  627. tor_free(filename);
  628. if (!f)
  629. return -1;
  630. }
  631. /* File format is:
  632. * FormatLine *KeywordLine Data
  633. *
  634. * FormatLine = "format 1" NL
  635. * KeywordLine = Keyword SP Arguments NL
  636. * Data = "data" NL *RouterMTBFLine "." NL
  637. * RouterMTBFLine = Fingerprint SP WeightedRunLen SP
  638. * TotalRunWeights [SP S=StartRunTime] NL
  639. */
  640. #define PUT(s) STMT_BEGIN if (fputs((s),f)<0) goto err; STMT_END
  641. #define PRINTF(args) STMT_BEGIN if (fprintf args <0) goto err; STMT_END
  642. PUT("format 2\n");
  643. format_iso_time(time_buf, time(NULL));
  644. PRINTF((f, "stored-at %s\n", time_buf));
  645. if (started_tracking_stability) {
  646. format_iso_time(time_buf, started_tracking_stability);
  647. PRINTF((f, "tracked-since %s\n", time_buf));
  648. }
  649. if (stability_last_downrated) {
  650. format_iso_time(time_buf, stability_last_downrated);
  651. PRINTF((f, "last-downrated %s\n", time_buf));
  652. }
  653. PUT("data\n");
  654. /* XXX Nick: now bridge auths record this for all routers too.
  655. * Should we make them record it only for bridge routers? -RD
  656. * Not for 0.2.0. -NM */
  657. for (orhist_it = digestmap_iter_init(history_map);
  658. !digestmap_iter_done(orhist_it);
  659. orhist_it = digestmap_iter_next(history_map,orhist_it)) {
  660. char dbuf[HEX_DIGEST_LEN+1];
  661. const char *t = NULL;
  662. digestmap_iter_get(orhist_it, &digest, &or_history_p);
  663. hist = (or_history_t*) or_history_p;
  664. base16_encode(dbuf, sizeof(dbuf), digest, DIGEST_LEN);
  665. PRINTF((f, "R %s\n", dbuf));
  666. if (hist->start_of_run > 0) {
  667. format_iso_time(time_buf, hist->start_of_run);
  668. t = time_buf;
  669. }
  670. PRINTF((f, "+MTBF %lu %.5lf%s%s\n",
  671. hist->weighted_run_length, hist->total_run_weights,
  672. t ? " S=" : "", t ? t : ""));
  673. t = NULL;
  674. if (hist->start_of_downtime > 0) {
  675. format_iso_time(time_buf, hist->start_of_downtime);
  676. t = time_buf;
  677. }
  678. PRINTF((f, "+WFU %lu %lu%s%s\n",
  679. hist->weighted_uptime, hist->total_weighted_time,
  680. t ? " S=" : "", t ? t : ""));
  681. }
  682. PUT(".\n");
  683. #undef PUT
  684. #undef PRINTF
  685. return finish_writing_to_file(open_file);
  686. err:
  687. abort_writing_to_file(open_file);
  688. return -1;
  689. }
  690. /** Format the current tracked status of the router in <b>hist</b> at time
  691. * <b>now</b> for analysis; return it in a newly allocated string. */
  692. static char *
  693. rep_hist_format_router_status(or_history_t *hist, time_t now)
  694. {
  695. char buf[1024];
  696. char sor_buf[ISO_TIME_LEN+1];
  697. char sod_buf[ISO_TIME_LEN+1];
  698. double wfu;
  699. double mtbf;
  700. int up = 0, down = 0;
  701. if (hist->start_of_run) {
  702. format_iso_time(sor_buf, hist->start_of_run);
  703. up = 1;
  704. }
  705. if (hist->start_of_downtime) {
  706. format_iso_time(sod_buf, hist->start_of_downtime);
  707. down = 1;
  708. }
  709. wfu = get_weighted_fractional_uptime(hist, now);
  710. mtbf = get_stability(hist, now);
  711. tor_snprintf(buf, sizeof(buf),
  712. "%s%s%s"
  713. "%s%s%s"
  714. "wfu %0.3lf\n"
  715. " weighted-time %lu\n"
  716. " weighted-uptime %lu\n"
  717. "mtbf %0.1lf\n"
  718. " weighted-run-length %lu\n"
  719. " total-run-weights %lf\n",
  720. up?"uptime-started ":"", up?sor_buf:"", up?" UTC\n":"",
  721. down?"downtime-started ":"", down?sod_buf:"", down?" UTC\n":"",
  722. wfu,
  723. hist->total_weighted_time,
  724. hist->weighted_uptime,
  725. mtbf,
  726. hist->weighted_run_length,
  727. hist->total_run_weights
  728. );
  729. return tor_strdup(buf);
  730. }
  731. /** The last stability analysis document that we created, or NULL if we never
  732. * have created one. */
  733. static char *last_stability_doc = NULL;
  734. /** The last time we created a stability analysis document, or 0 if we never
  735. * have created one. */
  736. static time_t built_last_stability_doc_at = 0;
  737. /** Shortest allowable time between building two stability documents. */
  738. #define MAX_STABILITY_DOC_BUILD_RATE (3*60)
  739. /** Return a pointer to a NUL-terminated document describing our view of the
  740. * stability of the routers we've been tracking. Return NULL on failure. */
  741. const char *
  742. rep_hist_get_router_stability_doc(time_t now)
  743. {
  744. char *result;
  745. smartlist_t *chunks;
  746. if (built_last_stability_doc_at + MAX_STABILITY_DOC_BUILD_RATE > now)
  747. return last_stability_doc;
  748. if (!history_map)
  749. return NULL;
  750. tor_free(last_stability_doc);
  751. chunks = smartlist_create();
  752. if (rep_hist_have_measured_enough_stability()) {
  753. smartlist_add(chunks, tor_strdup("we-have-enough-measurements\n"));
  754. } else {
  755. smartlist_add(chunks, tor_strdup("we-do-not-have-enough-measurements\n"));
  756. }
  757. DIGESTMAP_FOREACH(history_map, id, or_history_t *, hist) {
  758. routerinfo_t *ri;
  759. char dbuf[BASE64_DIGEST_LEN+1];
  760. char header_buf[512];
  761. char *info;
  762. digest_to_base64(dbuf, id);
  763. ri = router_get_by_digest(id);
  764. if (ri) {
  765. char *ip = tor_dup_ip(ri->addr);
  766. char tbuf[ISO_TIME_LEN+1];
  767. format_iso_time(tbuf, ri->cache_info.published_on);
  768. tor_snprintf(header_buf, sizeof(header_buf),
  769. "router %s %s %s\n"
  770. "published %s\n"
  771. "relevant-flags %s%s%s\n"
  772. "declared-uptime %ld\n",
  773. dbuf, ri->nickname, ip,
  774. tbuf,
  775. ri->is_running ? "Running " : "",
  776. ri->is_valid ? "Valid " : "",
  777. ri->is_hibernating ? "Hibernating " : "",
  778. ri->uptime);
  779. tor_free(ip);
  780. } else {
  781. tor_snprintf(header_buf, sizeof(header_buf),
  782. "router %s {no descriptor}\n", dbuf);
  783. }
  784. smartlist_add(chunks, tor_strdup(header_buf));
  785. info = rep_hist_format_router_status(hist, now);
  786. if (info)
  787. smartlist_add(chunks, info);
  788. } DIGESTMAP_FOREACH_END;
  789. result = smartlist_join_strings(chunks, "", 0, NULL);
  790. SMARTLIST_FOREACH(chunks, char *, cp, tor_free(cp));
  791. smartlist_free(chunks);
  792. last_stability_doc = result;
  793. built_last_stability_doc_at = time(NULL);
  794. return result;
  795. }
  796. /** Helper: return the first j >= i such that !strcmpstart(sl[j], prefix) and
  797. * such that no line sl[k] with i <= k < j starts with "R ". Return -1 if no
  798. * such line exists. */
  799. static int
  800. find_next_with(smartlist_t *sl, int i, const char *prefix)
  801. {
  802. for ( ; i < smartlist_len(sl); ++i) {
  803. const char *line = smartlist_get(sl, i);
  804. if (!strcmpstart(line, prefix))
  805. return i;
  806. if (!strcmpstart(line, "R "))
  807. return -1;
  808. }
  809. return -1;
  810. }
  811. /** How many bad times has parse_possibly_bad_iso_time parsed? */
  812. static int n_bogus_times = 0;
  813. /** Parse the ISO-formatted time in <b>s</b> into *<b>time_out</b>, but
  814. * rounds any pre-1970 date to Jan 1, 1970. */
  815. static int
  816. parse_possibly_bad_iso_time(const char *s, time_t *time_out)
  817. {
  818. int year;
  819. char b[5];
  820. strlcpy(b, s, sizeof(b));
  821. b[4] = '\0';
  822. year = (int)tor_parse_long(b, 10, 0, INT_MAX, NULL, NULL);
  823. if (year < 1970) {
  824. *time_out = 0;
  825. ++n_bogus_times;
  826. return 0;
  827. } else
  828. return parse_iso_time(s, time_out);
  829. }
  830. /** We've read a time <b>t</b> from a file stored at <b>stored_at</b>, which
  831. * says we started measuring at <b>started_measuring</b>. Return a new number
  832. * that's about as much before <b>now</b> as <b>t</b> was before
  833. * <b>stored_at</b>.
  834. */
  835. static INLINE time_t
  836. correct_time(time_t t, time_t now, time_t stored_at, time_t started_measuring)
  837. {
  838. if (t < started_measuring - 24*60*60*365)
  839. return 0;
  840. else if (t < started_measuring)
  841. return started_measuring;
  842. else if (t > stored_at)
  843. return 0;
  844. else {
  845. long run_length = stored_at - t;
  846. t = now - run_length;
  847. if (t < started_measuring)
  848. t = started_measuring;
  849. return t;
  850. }
  851. }
  852. /** Load MTBF data from disk. Returns 0 on success or recoverable error, -1
  853. * on failure. */
  854. int
  855. rep_hist_load_mtbf_data(time_t now)
  856. {
  857. /* XXXX won't handle being called while history is already populated. */
  858. smartlist_t *lines;
  859. const char *line = NULL;
  860. int r=0, i;
  861. time_t last_downrated = 0, stored_at = 0, tracked_since = 0;
  862. time_t latest_possible_start = now;
  863. long format = -1;
  864. {
  865. char *filename = get_datadir_fname("router-stability");
  866. char *d = read_file_to_str(filename, RFTS_IGNORE_MISSING, NULL);
  867. tor_free(filename);
  868. if (!d)
  869. return -1;
  870. lines = smartlist_create();
  871. smartlist_split_string(lines, d, "\n", SPLIT_SKIP_SPACE, 0);
  872. tor_free(d);
  873. }
  874. {
  875. const char *firstline;
  876. if (smartlist_len(lines)>4) {
  877. firstline = smartlist_get(lines, 0);
  878. if (!strcmpstart(firstline, "format "))
  879. format = tor_parse_long(firstline+strlen("format "),
  880. 10, -1, LONG_MAX, NULL, NULL);
  881. }
  882. }
  883. if (format != 1 && format != 2) {
  884. log_warn(LD_HIST,
  885. "Unrecognized format in mtbf history file. Skipping.");
  886. goto err;
  887. }
  888. for (i = 1; i < smartlist_len(lines); ++i) {
  889. line = smartlist_get(lines, i);
  890. if (!strcmp(line, "data"))
  891. break;
  892. if (!strcmpstart(line, "last-downrated ")) {
  893. if (parse_iso_time(line+strlen("last-downrated "), &last_downrated)<0)
  894. log_warn(LD_HIST,"Couldn't parse downrate time in mtbf "
  895. "history file.");
  896. }
  897. if (!strcmpstart(line, "stored-at ")) {
  898. if (parse_iso_time(line+strlen("stored-at "), &stored_at)<0)
  899. log_warn(LD_HIST,"Couldn't parse stored time in mtbf "
  900. "history file.");
  901. }
  902. if (!strcmpstart(line, "tracked-since ")) {
  903. if (parse_iso_time(line+strlen("tracked-since "), &tracked_since)<0)
  904. log_warn(LD_HIST,"Couldn't parse started-tracking time in mtbf "
  905. "history file.");
  906. }
  907. }
  908. if (last_downrated > now)
  909. last_downrated = now;
  910. if (tracked_since > now)
  911. tracked_since = now;
  912. if (!stored_at) {
  913. log_warn(LD_HIST, "No stored time recorded.");
  914. goto err;
  915. }
  916. if (line && !strcmp(line, "data"))
  917. ++i;
  918. n_bogus_times = 0;
  919. for (; i < smartlist_len(lines); ++i) {
  920. char digest[DIGEST_LEN];
  921. char hexbuf[HEX_DIGEST_LEN+1];
  922. char mtbf_timebuf[ISO_TIME_LEN+1];
  923. char wfu_timebuf[ISO_TIME_LEN+1];
  924. time_t start_of_run = 0;
  925. time_t start_of_downtime = 0;
  926. int have_mtbf = 0, have_wfu = 0;
  927. long wrl = 0;
  928. double trw = 0;
  929. long wt_uptime = 0, total_wt_time = 0;
  930. int n;
  931. or_history_t *hist;
  932. line = smartlist_get(lines, i);
  933. if (!strcmp(line, "."))
  934. break;
  935. mtbf_timebuf[0] = '\0';
  936. wfu_timebuf[0] = '\0';
  937. if (format == 1) {
  938. n = sscanf(line, "%40s %ld %lf S=%10s %8s",
  939. hexbuf, &wrl, &trw, mtbf_timebuf, mtbf_timebuf+11);
  940. if (n != 3 && n != 5) {
  941. log_warn(LD_HIST, "Couldn't scan line %s", escaped(line));
  942. continue;
  943. }
  944. have_mtbf = 1;
  945. } else {
  946. // format == 2.
  947. int mtbf_idx, wfu_idx;
  948. if (strcmpstart(line, "R ") || strlen(line) < 2+HEX_DIGEST_LEN)
  949. continue;
  950. strlcpy(hexbuf, line+2, sizeof(hexbuf));
  951. mtbf_idx = find_next_with(lines, i+1, "+MTBF ");
  952. wfu_idx = find_next_with(lines, i+1, "+WFU ");
  953. if (mtbf_idx >= 0) {
  954. const char *mtbfline = smartlist_get(lines, mtbf_idx);
  955. n = sscanf(mtbfline, "+MTBF %lu %lf S=%10s %8s",
  956. &wrl, &trw, mtbf_timebuf, mtbf_timebuf+11);
  957. if (n == 2 || n == 4) {
  958. have_mtbf = 1;
  959. } else {
  960. log_warn(LD_HIST, "Couldn't scan +MTBF line %s",
  961. escaped(mtbfline));
  962. }
  963. }
  964. if (wfu_idx >= 0) {
  965. const char *wfuline = smartlist_get(lines, wfu_idx);
  966. n = sscanf(wfuline, "+WFU %lu %lu S=%10s %8s",
  967. &wt_uptime, &total_wt_time,
  968. wfu_timebuf, wfu_timebuf+11);
  969. if (n == 2 || n == 4) {
  970. have_wfu = 1;
  971. } else {
  972. log_warn(LD_HIST, "Couldn't scan +WFU line %s", escaped(wfuline));
  973. }
  974. }
  975. if (wfu_idx > i)
  976. i = wfu_idx;
  977. if (mtbf_idx > i)
  978. i = mtbf_idx;
  979. }
  980. if (base16_decode(digest, DIGEST_LEN, hexbuf, HEX_DIGEST_LEN) < 0) {
  981. log_warn(LD_HIST, "Couldn't hex string %s", escaped(hexbuf));
  982. continue;
  983. }
  984. hist = get_or_history(digest);
  985. if (!hist)
  986. continue;
  987. if (have_mtbf) {
  988. if (mtbf_timebuf[0]) {
  989. mtbf_timebuf[10] = ' ';
  990. if (parse_possibly_bad_iso_time(mtbf_timebuf, &start_of_run)<0)
  991. log_warn(LD_HIST, "Couldn't parse time %s",
  992. escaped(mtbf_timebuf));
  993. }
  994. hist->start_of_run = correct_time(start_of_run, now, stored_at,
  995. tracked_since);
  996. if (hist->start_of_run < latest_possible_start + wrl)
  997. latest_possible_start = hist->start_of_run - wrl;
  998. hist->weighted_run_length = wrl;
  999. hist->total_run_weights = trw;
  1000. }
  1001. if (have_wfu) {
  1002. if (wfu_timebuf[0]) {
  1003. wfu_timebuf[10] = ' ';
  1004. if (parse_possibly_bad_iso_time(wfu_timebuf, &start_of_downtime)<0)
  1005. log_warn(LD_HIST, "Couldn't parse time %s", escaped(wfu_timebuf));
  1006. }
  1007. }
  1008. hist->start_of_downtime = correct_time(start_of_downtime, now, stored_at,
  1009. tracked_since);
  1010. hist->weighted_uptime = wt_uptime;
  1011. hist->total_weighted_time = total_wt_time;
  1012. }
  1013. if (strcmp(line, "."))
  1014. log_warn(LD_HIST, "Truncated MTBF file.");
  1015. if (tracked_since < 86400*365) /* Recover from insanely early value. */
  1016. tracked_since = latest_possible_start;
  1017. stability_last_downrated = last_downrated;
  1018. started_tracking_stability = tracked_since;
  1019. goto done;
  1020. err:
  1021. r = -1;
  1022. done:
  1023. SMARTLIST_FOREACH(lines, char *, cp, tor_free(cp));
  1024. smartlist_free(lines);
  1025. return r;
  1026. }
  1027. /** For how many seconds do we keep track of individual per-second bandwidth
  1028. * totals? */
  1029. #define NUM_SECS_ROLLING_MEASURE 10
  1030. /** How large are the intervals for which we track and report bandwidth use? */
  1031. #define NUM_SECS_BW_SUM_INTERVAL (15*60)
  1032. /** How far in the past do we remember and publish bandwidth use? */
  1033. #define NUM_SECS_BW_SUM_IS_VALID (24*60*60)
  1034. /** How many bandwidth usage intervals do we remember? (derived) */
  1035. #define NUM_TOTALS (NUM_SECS_BW_SUM_IS_VALID/NUM_SECS_BW_SUM_INTERVAL)
  1036. /** Structure to track bandwidth use, and remember the maxima for a given
  1037. * time period.
  1038. */
  1039. typedef struct bw_array_t {
  1040. /** Observation array: Total number of bytes transferred in each of the last
  1041. * NUM_SECS_ROLLING_MEASURE seconds. This is used as a circular array. */
  1042. uint64_t obs[NUM_SECS_ROLLING_MEASURE];
  1043. int cur_obs_idx; /**< Current position in obs. */
  1044. time_t cur_obs_time; /**< Time represented in obs[cur_obs_idx] */
  1045. uint64_t total_obs; /**< Total for all members of obs except
  1046. * obs[cur_obs_idx] */
  1047. uint64_t max_total; /**< Largest value that total_obs has taken on in the
  1048. * current period. */
  1049. uint64_t total_in_period; /**< Total bytes transferred in the current
  1050. * period. */
  1051. /** When does the next period begin? */
  1052. time_t next_period;
  1053. /** Where in 'maxima' should the maximum bandwidth usage for the current
  1054. * period be stored? */
  1055. int next_max_idx;
  1056. /** How many values in maxima/totals have been set ever? */
  1057. int num_maxes_set;
  1058. /** Circular array of the maximum
  1059. * bandwidth-per-NUM_SECS_ROLLING_MEASURE usage for the last
  1060. * NUM_TOTALS periods */
  1061. uint64_t maxima[NUM_TOTALS];
  1062. /** Circular array of the total bandwidth usage for the last NUM_TOTALS
  1063. * periods */
  1064. uint64_t totals[NUM_TOTALS];
  1065. } bw_array_t;
  1066. /** Shift the current period of b forward by one. */
  1067. static void
  1068. commit_max(bw_array_t *b)
  1069. {
  1070. /* Store total from current period. */
  1071. b->totals[b->next_max_idx] = b->total_in_period;
  1072. /* Store maximum from current period. */
  1073. b->maxima[b->next_max_idx++] = b->max_total;
  1074. /* Advance next_period and next_max_idx */
  1075. b->next_period += NUM_SECS_BW_SUM_INTERVAL;
  1076. if (b->next_max_idx == NUM_TOTALS)
  1077. b->next_max_idx = 0;
  1078. if (b->num_maxes_set < NUM_TOTALS)
  1079. ++b->num_maxes_set;
  1080. /* Reset max_total. */
  1081. b->max_total = 0;
  1082. /* Reset total_in_period. */
  1083. b->total_in_period = 0;
  1084. }
  1085. /** Shift the current observation time of 'b' forward by one second. */
  1086. static INLINE void
  1087. advance_obs(bw_array_t *b)
  1088. {
  1089. int nextidx;
  1090. uint64_t total;
  1091. /* Calculate the total bandwidth for the last NUM_SECS_ROLLING_MEASURE
  1092. * seconds; adjust max_total as needed.*/
  1093. total = b->total_obs + b->obs[b->cur_obs_idx];
  1094. if (total > b->max_total)
  1095. b->max_total = total;
  1096. nextidx = b->cur_obs_idx+1;
  1097. if (nextidx == NUM_SECS_ROLLING_MEASURE)
  1098. nextidx = 0;
  1099. b->total_obs = total - b->obs[nextidx];
  1100. b->obs[nextidx]=0;
  1101. b->cur_obs_idx = nextidx;
  1102. if (++b->cur_obs_time >= b->next_period)
  1103. commit_max(b);
  1104. }
  1105. /** Add <b>n</b> bytes to the number of bytes in <b>b</b> for second
  1106. * <b>when</b>. */
  1107. static INLINE void
  1108. add_obs(bw_array_t *b, time_t when, uint64_t n)
  1109. {
  1110. /* Don't record data in the past. */
  1111. if (when<b->cur_obs_time)
  1112. return;
  1113. /* If we're currently adding observations for an earlier second than
  1114. * 'when', advance b->cur_obs_time and b->cur_obs_idx by an
  1115. * appropriate number of seconds, and do all the other housekeeping */
  1116. while (when>b->cur_obs_time)
  1117. advance_obs(b);
  1118. b->obs[b->cur_obs_idx] += n;
  1119. b->total_in_period += n;
  1120. }
  1121. /** Allocate, initialize, and return a new bw_array. */
  1122. static bw_array_t *
  1123. bw_array_new(void)
  1124. {
  1125. bw_array_t *b;
  1126. time_t start;
  1127. b = tor_malloc_zero(sizeof(bw_array_t));
  1128. rephist_total_alloc += sizeof(bw_array_t);
  1129. start = time(NULL);
  1130. b->cur_obs_time = start;
  1131. b->next_period = start + NUM_SECS_BW_SUM_INTERVAL;
  1132. return b;
  1133. }
  1134. /** Recent history of bandwidth observations for read operations. */
  1135. static bw_array_t *read_array = NULL;
  1136. /** Recent history of bandwidth observations for write operations. */
  1137. static bw_array_t *write_array = NULL;
  1138. /** Set up read_array and write_array. */
  1139. static void
  1140. bw_arrays_init(void)
  1141. {
  1142. read_array = bw_array_new();
  1143. write_array = bw_array_new();
  1144. }
  1145. /** We read <b>num_bytes</b> more bytes in second <b>when</b>.
  1146. *
  1147. * Add num_bytes to the current running total for <b>when</b>.
  1148. *
  1149. * <b>when</b> can go back to time, but it's safe to ignore calls
  1150. * earlier than the latest <b>when</b> you've heard of.
  1151. */
  1152. void
  1153. rep_hist_note_bytes_written(size_t num_bytes, time_t when)
  1154. {
  1155. /* Maybe a circular array for recent seconds, and step to a new point
  1156. * every time a new second shows up. Or simpler is to just to have
  1157. * a normal array and push down each item every second; it's short.
  1158. */
  1159. /* When a new second has rolled over, compute the sum of the bytes we've
  1160. * seen over when-1 to when-1-NUM_SECS_ROLLING_MEASURE, and stick it
  1161. * somewhere. See rep_hist_bandwidth_assess() below.
  1162. */
  1163. add_obs(write_array, when, num_bytes);
  1164. }
  1165. /** We wrote <b>num_bytes</b> more bytes in second <b>when</b>.
  1166. * (like rep_hist_note_bytes_written() above)
  1167. */
  1168. void
  1169. rep_hist_note_bytes_read(size_t num_bytes, time_t when)
  1170. {
  1171. /* if we're smart, we can make this func and the one above share code */
  1172. add_obs(read_array, when, num_bytes);
  1173. }
  1174. /** Helper: Return the largest value in b->maxima. (This is equal to the
  1175. * most bandwidth used in any NUM_SECS_ROLLING_MEASURE period for the last
  1176. * NUM_SECS_BW_SUM_IS_VALID seconds.)
  1177. */
  1178. static uint64_t
  1179. find_largest_max(bw_array_t *b)
  1180. {
  1181. int i;
  1182. uint64_t max;
  1183. max=0;
  1184. for (i=0; i<NUM_TOTALS; ++i) {
  1185. if (b->maxima[i]>max)
  1186. max = b->maxima[i];
  1187. }
  1188. return max;
  1189. }
  1190. /** Find the largest sums in the past NUM_SECS_BW_SUM_IS_VALID (roughly)
  1191. * seconds. Find one sum for reading and one for writing. They don't have
  1192. * to be at the same time.
  1193. *
  1194. * Return the smaller of these sums, divided by NUM_SECS_ROLLING_MEASURE.
  1195. */
  1196. int
  1197. rep_hist_bandwidth_assess(void)
  1198. {
  1199. uint64_t w,r;
  1200. r = find_largest_max(read_array);
  1201. w = find_largest_max(write_array);
  1202. if (r>w)
  1203. return (int)(U64_TO_DBL(w)/NUM_SECS_ROLLING_MEASURE);
  1204. else
  1205. return (int)(U64_TO_DBL(r)/NUM_SECS_ROLLING_MEASURE);
  1206. }
  1207. /** Print the bandwidth history of b (either read_array or write_array)
  1208. * into the buffer pointed to by buf. The format is simply comma
  1209. * separated numbers, from oldest to newest.
  1210. *
  1211. * It returns the number of bytes written.
  1212. */
  1213. static size_t
  1214. rep_hist_fill_bandwidth_history(char *buf, size_t len, bw_array_t *b)
  1215. {
  1216. char *cp = buf;
  1217. int i, n;
  1218. or_options_t *options = get_options();
  1219. uint64_t cutoff;
  1220. if (b->num_maxes_set <= b->next_max_idx) {
  1221. /* We haven't been through the circular array yet; time starts at i=0.*/
  1222. i = 0;
  1223. } else {
  1224. /* We've been around the array at least once. The next i to be
  1225. overwritten is the oldest. */
  1226. i = b->next_max_idx;
  1227. }
  1228. if (options->RelayBandwidthRate) {
  1229. /* We don't want to report that we used more bandwidth than the max we're
  1230. * willing to relay; otherwise everybody will know how much traffic
  1231. * we used ourself. */
  1232. cutoff = options->RelayBandwidthRate * NUM_SECS_BW_SUM_INTERVAL;
  1233. } else {
  1234. cutoff = UINT64_MAX;
  1235. }
  1236. for (n=0; n<b->num_maxes_set; ++n,++i) {
  1237. uint64_t total;
  1238. if (i >= NUM_TOTALS)
  1239. i -= NUM_TOTALS;
  1240. tor_assert(i < NUM_TOTALS);
  1241. /* Round the bandwidth used down to the nearest 1k. */
  1242. total = b->totals[i] & ~0x3ff;
  1243. if (total > cutoff)
  1244. total = cutoff;
  1245. if (n==(b->num_maxes_set-1))
  1246. tor_snprintf(cp, len-(cp-buf), U64_FORMAT, U64_PRINTF_ARG(total));
  1247. else
  1248. tor_snprintf(cp, len-(cp-buf), U64_FORMAT",", U64_PRINTF_ARG(total));
  1249. cp += strlen(cp);
  1250. }
  1251. return cp-buf;
  1252. }
  1253. /** Allocate and return lines for representing this server's bandwidth
  1254. * history in its descriptor.
  1255. */
  1256. char *
  1257. rep_hist_get_bandwidth_lines(int for_extrainfo)
  1258. {
  1259. char *buf, *cp;
  1260. char t[ISO_TIME_LEN+1];
  1261. int r;
  1262. bw_array_t *b;
  1263. size_t len;
  1264. /* opt (read|write)-history yyyy-mm-dd HH:MM:SS (n s) n,n,n,n,n... */
  1265. len = (60+20*NUM_TOTALS)*2;
  1266. buf = tor_malloc_zero(len);
  1267. cp = buf;
  1268. for (r=0;r<2;++r) {
  1269. b = r?read_array:write_array;
  1270. tor_assert(b);
  1271. format_iso_time(t, b->next_period-NUM_SECS_BW_SUM_INTERVAL);
  1272. tor_snprintf(cp, len-(cp-buf), "%s%s %s (%d s) ",
  1273. for_extrainfo ? "" : "opt ",
  1274. r ? "read-history" : "write-history", t,
  1275. NUM_SECS_BW_SUM_INTERVAL);
  1276. cp += strlen(cp);
  1277. cp += rep_hist_fill_bandwidth_history(cp, len-(cp-buf), b);
  1278. strlcat(cp, "\n", len-(cp-buf));
  1279. ++cp;
  1280. }
  1281. return buf;
  1282. }
  1283. /** Update <b>state</b> with the newest bandwidth history. */
  1284. void
  1285. rep_hist_update_state(or_state_t *state)
  1286. {
  1287. int len, r;
  1288. char *buf, *cp;
  1289. smartlist_t **s_values;
  1290. time_t *s_begins;
  1291. int *s_interval;
  1292. bw_array_t *b;
  1293. len = 20*NUM_TOTALS+1;
  1294. buf = tor_malloc_zero(len);
  1295. for (r=0;r<2;++r) {
  1296. b = r?read_array:write_array;
  1297. s_begins = r?&state->BWHistoryReadEnds :&state->BWHistoryWriteEnds;
  1298. s_interval= r?&state->BWHistoryReadInterval:&state->BWHistoryWriteInterval;
  1299. s_values = r?&state->BWHistoryReadValues :&state->BWHistoryWriteValues;
  1300. if (*s_values) {
  1301. SMARTLIST_FOREACH(*s_values, char *, val, tor_free(val));
  1302. smartlist_free(*s_values);
  1303. }
  1304. if (! server_mode(get_options())) {
  1305. /* Clients don't need to store bandwidth history persistently;
  1306. * force these values to the defaults. */
  1307. /* FFFF we should pull the default out of config.c's state table,
  1308. * so we don't have two defaults. */
  1309. if (*s_begins != 0 || *s_interval != 900) {
  1310. time_t now = time(NULL);
  1311. time_t save_at = get_options()->AvoidDiskWrites ? now+3600 : now+600;
  1312. or_state_mark_dirty(state, save_at);
  1313. }
  1314. *s_begins = 0;
  1315. *s_interval = 900;
  1316. *s_values = smartlist_create();
  1317. continue;
  1318. }
  1319. *s_begins = b->next_period;
  1320. *s_interval = NUM_SECS_BW_SUM_INTERVAL;
  1321. cp = buf;
  1322. cp += rep_hist_fill_bandwidth_history(cp, len, b);
  1323. tor_snprintf(cp, len-(cp-buf), cp == buf ? U64_FORMAT : ","U64_FORMAT,
  1324. U64_PRINTF_ARG(b->total_in_period));
  1325. *s_values = smartlist_create();
  1326. if (server_mode(get_options()))
  1327. smartlist_split_string(*s_values, buf, ",", SPLIT_SKIP_SPACE, 0);
  1328. }
  1329. tor_free(buf);
  1330. if (server_mode(get_options())) {
  1331. or_state_mark_dirty(get_or_state(), time(NULL)+(2*3600));
  1332. }
  1333. }
  1334. /** Set bandwidth history from our saved state. */
  1335. int
  1336. rep_hist_load_state(or_state_t *state, char **err)
  1337. {
  1338. time_t s_begins, start;
  1339. time_t now = time(NULL);
  1340. uint64_t v;
  1341. int r,i,ok;
  1342. int all_ok = 1;
  1343. int s_interval;
  1344. smartlist_t *s_values;
  1345. bw_array_t *b;
  1346. /* Assert they already have been malloced */
  1347. tor_assert(read_array && write_array);
  1348. for (r=0;r<2;++r) {
  1349. b = r?read_array:write_array;
  1350. s_begins = r?state->BWHistoryReadEnds:state->BWHistoryWriteEnds;
  1351. s_interval = r?state->BWHistoryReadInterval:state->BWHistoryWriteInterval;
  1352. s_values = r?state->BWHistoryReadValues:state->BWHistoryWriteValues;
  1353. if (s_values && s_begins >= now - NUM_SECS_BW_SUM_INTERVAL*NUM_TOTALS) {
  1354. start = s_begins - s_interval*(smartlist_len(s_values));
  1355. if (start > now)
  1356. continue;
  1357. b->cur_obs_time = start;
  1358. b->next_period = start + NUM_SECS_BW_SUM_INTERVAL;
  1359. SMARTLIST_FOREACH(s_values, char *, cp, {
  1360. v = tor_parse_uint64(cp, 10, 0, UINT64_MAX, &ok, NULL);
  1361. if (!ok) {
  1362. all_ok=0;
  1363. log_notice(LD_HIST, "Could not parse '%s' into a number.'", cp);
  1364. }
  1365. if (start < now) {
  1366. add_obs(b, start, v);
  1367. start += NUM_SECS_BW_SUM_INTERVAL;
  1368. }
  1369. });
  1370. }
  1371. /* Clean up maxima and observed */
  1372. /* Do we really want to zero this for the purpose of max capacity? */
  1373. for (i=0; i<NUM_SECS_ROLLING_MEASURE; ++i) {
  1374. b->obs[i] = 0;
  1375. }
  1376. b->total_obs = 0;
  1377. for (i=0; i<NUM_TOTALS; ++i) {
  1378. b->maxima[i] = 0;
  1379. }
  1380. b->max_total = 0;
  1381. }
  1382. if (!all_ok) {
  1383. *err = tor_strdup("Parsing of bandwidth history values failed");
  1384. /* and create fresh arrays */
  1385. tor_free(read_array);
  1386. tor_free(write_array);
  1387. read_array = bw_array_new();
  1388. write_array = bw_array_new();
  1389. return -1;
  1390. }
  1391. return 0;
  1392. }
  1393. /*********************************************************************/
  1394. /** A list of port numbers that have been used recently. */
  1395. static smartlist_t *predicted_ports_list=NULL;
  1396. /** The corresponding most recently used time for each port. */
  1397. static smartlist_t *predicted_ports_times=NULL;
  1398. /** We just got an application request for a connection with
  1399. * port <b>port</b>. Remember it for the future, so we can keep
  1400. * some circuits open that will exit to this port.
  1401. */
  1402. static void
  1403. add_predicted_port(time_t now, uint16_t port)
  1404. {
  1405. /* XXXX we could just use uintptr_t here, I think. */
  1406. uint16_t *tmp_port = tor_malloc(sizeof(uint16_t));
  1407. time_t *tmp_time = tor_malloc(sizeof(time_t));
  1408. *tmp_port = port;
  1409. *tmp_time = now;
  1410. rephist_total_alloc += sizeof(uint16_t) + sizeof(time_t);
  1411. smartlist_add(predicted_ports_list, tmp_port);
  1412. smartlist_add(predicted_ports_times, tmp_time);
  1413. }
  1414. /** Initialize whatever memory and structs are needed for predicting
  1415. * which ports will be used. Also seed it with port 80, so we'll build
  1416. * circuits on start-up.
  1417. */
  1418. static void
  1419. predicted_ports_init(void)
  1420. {
  1421. predicted_ports_list = smartlist_create();
  1422. predicted_ports_times = smartlist_create();
  1423. add_predicted_port(time(NULL), 80); /* add one to kickstart us */
  1424. }
  1425. /** Free whatever memory is needed for predicting which ports will
  1426. * be used.
  1427. */
  1428. static void
  1429. predicted_ports_free(void)
  1430. {
  1431. rephist_total_alloc -= smartlist_len(predicted_ports_list)*sizeof(uint16_t);
  1432. SMARTLIST_FOREACH(predicted_ports_list, char *, cp, tor_free(cp));
  1433. smartlist_free(predicted_ports_list);
  1434. rephist_total_alloc -= smartlist_len(predicted_ports_times)*sizeof(time_t);
  1435. SMARTLIST_FOREACH(predicted_ports_times, char *, cp, tor_free(cp));
  1436. smartlist_free(predicted_ports_times);
  1437. }
  1438. /** Remember that <b>port</b> has been asked for as of time <b>now</b>.
  1439. * This is used for predicting what sorts of streams we'll make in the
  1440. * future and making exit circuits to anticipate that.
  1441. */
  1442. void
  1443. rep_hist_note_used_port(time_t now, uint16_t port)
  1444. {
  1445. int i;
  1446. uint16_t *tmp_port;
  1447. time_t *tmp_time;
  1448. tor_assert(predicted_ports_list);
  1449. tor_assert(predicted_ports_times);
  1450. if (!port) /* record nothing */
  1451. return;
  1452. for (i = 0; i < smartlist_len(predicted_ports_list); ++i) {
  1453. tmp_port = smartlist_get(predicted_ports_list, i);
  1454. tmp_time = smartlist_get(predicted_ports_times, i);
  1455. if (*tmp_port == port) {
  1456. *tmp_time = now;
  1457. return;
  1458. }
  1459. }
  1460. /* it's not there yet; we need to add it */
  1461. add_predicted_port(now, port);
  1462. }
  1463. /** For this long after we've seen a request for a given port, assume that
  1464. * we'll want to make connections to the same port in the future. */
  1465. #define PREDICTED_CIRCS_RELEVANCE_TIME (60*60)
  1466. /** Return a pointer to the list of port numbers that
  1467. * are likely to be asked for in the near future.
  1468. *
  1469. * The caller promises not to mess with it.
  1470. */
  1471. smartlist_t *
  1472. rep_hist_get_predicted_ports(time_t now)
  1473. {
  1474. int i;
  1475. uint16_t *tmp_port;
  1476. time_t *tmp_time;
  1477. tor_assert(predicted_ports_list);
  1478. tor_assert(predicted_ports_times);
  1479. /* clean out obsolete entries */
  1480. for (i = 0; i < smartlist_len(predicted_ports_list); ++i) {
  1481. tmp_time = smartlist_get(predicted_ports_times, i);
  1482. if (*tmp_time + PREDICTED_CIRCS_RELEVANCE_TIME < now) {
  1483. tmp_port = smartlist_get(predicted_ports_list, i);
  1484. log_debug(LD_CIRC, "Expiring predicted port %d", *tmp_port);
  1485. smartlist_del(predicted_ports_list, i);
  1486. smartlist_del(predicted_ports_times, i);
  1487. rephist_total_alloc -= sizeof(uint16_t)+sizeof(time_t);
  1488. tor_free(tmp_port);
  1489. tor_free(tmp_time);
  1490. i--;
  1491. }
  1492. }
  1493. return predicted_ports_list;
  1494. }
  1495. /** The user asked us to do a resolve. Rather than keeping track of
  1496. * timings and such of resolves, we fake it for now by treating
  1497. * it the same way as a connection to port 80. This way we will continue
  1498. * to have circuits lying around if the user only uses Tor for resolves.
  1499. */
  1500. void
  1501. rep_hist_note_used_resolve(time_t now)
  1502. {
  1503. rep_hist_note_used_port(now, 80);
  1504. }
  1505. /** The last time at which we needed an internal circ. */
  1506. static time_t predicted_internal_time = 0;
  1507. /** The last time we needed an internal circ with good uptime. */
  1508. static time_t predicted_internal_uptime_time = 0;
  1509. /** The last time we needed an internal circ with good capacity. */
  1510. static time_t predicted_internal_capacity_time = 0;
  1511. /** Remember that we used an internal circ at time <b>now</b>. */
  1512. void
  1513. rep_hist_note_used_internal(time_t now, int need_uptime, int need_capacity)
  1514. {
  1515. predicted_internal_time = now;
  1516. if (need_uptime)
  1517. predicted_internal_uptime_time = now;
  1518. if (need_capacity)
  1519. predicted_internal_capacity_time = now;
  1520. }
  1521. /** Return 1 if we've used an internal circ recently; else return 0. */
  1522. int
  1523. rep_hist_get_predicted_internal(time_t now, int *need_uptime,
  1524. int *need_capacity)
  1525. {
  1526. if (!predicted_internal_time) { /* initialize it */
  1527. predicted_internal_time = now;
  1528. predicted_internal_uptime_time = now;
  1529. predicted_internal_capacity_time = now;
  1530. }
  1531. if (predicted_internal_time + PREDICTED_CIRCS_RELEVANCE_TIME < now)
  1532. return 0; /* too long ago */
  1533. if (predicted_internal_uptime_time + PREDICTED_CIRCS_RELEVANCE_TIME >= now)
  1534. *need_uptime = 1;
  1535. if (predicted_internal_capacity_time + PREDICTED_CIRCS_RELEVANCE_TIME >= now)
  1536. *need_capacity = 1;
  1537. return 1;
  1538. }
  1539. /** Any ports used lately? These are pre-seeded if we just started
  1540. * up or if we're running a hidden service. */
  1541. int
  1542. any_predicted_circuits(time_t now)
  1543. {
  1544. return smartlist_len(predicted_ports_list) ||
  1545. predicted_internal_time + PREDICTED_CIRCS_RELEVANCE_TIME >= now;
  1546. }
  1547. /** Return 1 if we have no need for circuits currently, else return 0. */
  1548. int
  1549. rep_hist_circbuilding_dormant(time_t now)
  1550. {
  1551. if (any_predicted_circuits(now))
  1552. return 0;
  1553. /* see if we'll still need to build testing circuits */
  1554. if (server_mode(get_options()) &&
  1555. (!check_whether_orport_reachable() || !circuit_enough_testing_circs()))
  1556. return 0;
  1557. if (!check_whether_dirport_reachable())
  1558. return 0;
  1559. return 1;
  1560. }
  1561. /** Structure to track how many times we've done each public key operation. */
  1562. static struct {
  1563. /** How many directory objects have we signed? */
  1564. unsigned long n_signed_dir_objs;
  1565. /** How many routerdescs have we signed? */
  1566. unsigned long n_signed_routerdescs;
  1567. /** How many directory objects have we verified? */
  1568. unsigned long n_verified_dir_objs;
  1569. /** How many routerdescs have we verified */
  1570. unsigned long n_verified_routerdescs;
  1571. /** How many onionskins have we encrypted to build circuits? */
  1572. unsigned long n_onionskins_encrypted;
  1573. /** How many onionskins have we decrypted to do circuit build requests? */
  1574. unsigned long n_onionskins_decrypted;
  1575. /** How many times have we done the TLS handshake as a client? */
  1576. unsigned long n_tls_client_handshakes;
  1577. /** How many times have we done the TLS handshake as a server? */
  1578. unsigned long n_tls_server_handshakes;
  1579. /** How many PK operations have we done as a hidden service client? */
  1580. unsigned long n_rend_client_ops;
  1581. /** How many PK operations have we done as a hidden service midpoint? */
  1582. unsigned long n_rend_mid_ops;
  1583. /** How many PK operations have we done as a hidden service provider? */
  1584. unsigned long n_rend_server_ops;
  1585. } pk_op_counts = {0,0,0,0,0,0,0,0,0,0,0};
  1586. /** Increment the count of the number of times we've done <b>operation</b>. */
  1587. void
  1588. note_crypto_pk_op(pk_op_t operation)
  1589. {
  1590. switch (operation)
  1591. {
  1592. case SIGN_DIR:
  1593. pk_op_counts.n_signed_dir_objs++;
  1594. break;
  1595. case SIGN_RTR:
  1596. pk_op_counts.n_signed_routerdescs++;
  1597. break;
  1598. case VERIFY_DIR:
  1599. pk_op_counts.n_verified_dir_objs++;
  1600. break;
  1601. case VERIFY_RTR:
  1602. pk_op_counts.n_verified_routerdescs++;
  1603. break;
  1604. case ENC_ONIONSKIN:
  1605. pk_op_counts.n_onionskins_encrypted++;
  1606. break;
  1607. case DEC_ONIONSKIN:
  1608. pk_op_counts.n_onionskins_decrypted++;
  1609. break;
  1610. case TLS_HANDSHAKE_C:
  1611. pk_op_counts.n_tls_client_handshakes++;
  1612. break;
  1613. case TLS_HANDSHAKE_S:
  1614. pk_op_counts.n_tls_server_handshakes++;
  1615. break;
  1616. case REND_CLIENT:
  1617. pk_op_counts.n_rend_client_ops++;
  1618. break;
  1619. case REND_MID:
  1620. pk_op_counts.n_rend_mid_ops++;
  1621. break;
  1622. case REND_SERVER:
  1623. pk_op_counts.n_rend_server_ops++;
  1624. break;
  1625. default:
  1626. log_warn(LD_BUG, "Unknown pk operation %d", operation);
  1627. }
  1628. }
  1629. /** Log the number of times we've done each public/private-key operation. */
  1630. void
  1631. dump_pk_ops(int severity)
  1632. {
  1633. log(severity, LD_HIST,
  1634. "PK operations: %lu directory objects signed, "
  1635. "%lu directory objects verified, "
  1636. "%lu routerdescs signed, "
  1637. "%lu routerdescs verified, "
  1638. "%lu onionskins encrypted, "
  1639. "%lu onionskins decrypted, "
  1640. "%lu client-side TLS handshakes, "
  1641. "%lu server-side TLS handshakes, "
  1642. "%lu rendezvous client operations, "
  1643. "%lu rendezvous middle operations, "
  1644. "%lu rendezvous server operations.",
  1645. pk_op_counts.n_signed_dir_objs,
  1646. pk_op_counts.n_verified_dir_objs,
  1647. pk_op_counts.n_signed_routerdescs,
  1648. pk_op_counts.n_verified_routerdescs,
  1649. pk_op_counts.n_onionskins_encrypted,
  1650. pk_op_counts.n_onionskins_decrypted,
  1651. pk_op_counts.n_tls_client_handshakes,
  1652. pk_op_counts.n_tls_server_handshakes,
  1653. pk_op_counts.n_rend_client_ops,
  1654. pk_op_counts.n_rend_mid_ops,
  1655. pk_op_counts.n_rend_server_ops);
  1656. }
  1657. /** Free all storage held by the OR/link history caches, by the
  1658. * bandwidth history arrays, or by the port history. */
  1659. void
  1660. rep_hist_free_all(void)
  1661. {
  1662. digestmap_free(history_map, free_or_history);
  1663. tor_free(read_array);
  1664. tor_free(write_array);
  1665. tor_free(last_stability_doc);
  1666. built_last_stability_doc_at = 0;
  1667. predicted_ports_free();
  1668. }
  1669. /****************** hidden service usage statistics ******************/
  1670. /** How large are the intervals for which we track and report hidden service
  1671. * use? */
  1672. #define NUM_SECS_HS_USAGE_SUM_INTERVAL (15*60)
  1673. /** How far in the past do we remember and publish hidden service use? */
  1674. #define NUM_SECS_HS_USAGE_SUM_IS_VALID (24*60*60)
  1675. /** How many hidden service usage intervals do we remember? (derived) */
  1676. #define NUM_TOTALS_HS_USAGE (NUM_SECS_HS_USAGE_SUM_IS_VALID/ \
  1677. NUM_SECS_HS_USAGE_SUM_INTERVAL)
  1678. /** List element containing a service id and the count. */
  1679. typedef struct hs_usage_list_elem_t {
  1680. /** Service id of this elem. */
  1681. char service_id[REND_SERVICE_ID_LEN_BASE32+1];
  1682. /** Number of occurrences for the given service id. */
  1683. uint32_t count;
  1684. /* Pointer to next list elem */
  1685. struct hs_usage_list_elem_t *next;
  1686. } hs_usage_list_elem_t;
  1687. /** Ordered list that stores service ids and the number of observations. It is
  1688. * ordered by the number of occurrences in descending order. Its purpose is to
  1689. * calculate the frequency distribution when the period is over. */
  1690. typedef struct hs_usage_list_t {
  1691. /* Pointer to the first element in the list. */
  1692. hs_usage_list_elem_t *start;
  1693. /* Number of total occurrences for all list elements. */
  1694. uint32_t total_count;
  1695. /* Number of service ids, i.e. number of list elements. */
  1696. uint32_t total_service_ids;
  1697. } hs_usage_list_t;
  1698. /** Tracks service-related observations in the current period and their
  1699. * history. */
  1700. typedef struct hs_usage_service_related_observation_t {
  1701. /** Ordered list that stores service ids and the number of observations in
  1702. * the current period. It is ordered by the number of occurrences in
  1703. * descending order. Its purpose is to calculate the frequency distribution
  1704. * when the period is over. */
  1705. hs_usage_list_t *list;
  1706. /** Circular arrays that store the history of observations. totals stores all
  1707. * observations, twenty (ten, five) the number of observations related to a
  1708. * service id being accounted for the top 20 (10, 5) percent of all
  1709. * observations. */
  1710. uint32_t totals[NUM_TOTALS_HS_USAGE];
  1711. uint32_t five[NUM_TOTALS_HS_USAGE];
  1712. uint32_t ten[NUM_TOTALS_HS_USAGE];
  1713. uint32_t twenty[NUM_TOTALS_HS_USAGE];
  1714. } hs_usage_service_related_observation_t;
  1715. /** Tracks the history of general period-related observations, i.e. those that
  1716. * cannot be related to a specific service id. */
  1717. typedef struct hs_usage_general_period_related_observations_t {
  1718. /** Circular array that stores the history of observations. */
  1719. uint32_t totals[NUM_TOTALS_HS_USAGE];
  1720. } hs_usage_general_period_related_observations_t;
  1721. /** Keeps information about the current observation period and its relation to
  1722. * the histories of observations. */
  1723. typedef struct hs_usage_current_observation_period_t {
  1724. /** Where do we write the next history entry? */
  1725. int next_idx;
  1726. /** How many values in history have been set ever? (upper bound!) */
  1727. int num_set;
  1728. /** When did this period begin? */
  1729. time_t start_of_current_period;
  1730. /** When does the next period begin? */
  1731. time_t start_of_next_period;
  1732. } hs_usage_current_observation_period_t;
  1733. /** Usage statistics for the current observation period. */
  1734. static hs_usage_current_observation_period_t *current_period = NULL;
  1735. /* DOCDOC publish_total */
  1736. static hs_usage_service_related_observation_t *publish_total = NULL;
  1737. /* DOCDOC publish_novel */
  1738. static hs_usage_service_related_observation_t *publish_novel = NULL;
  1739. /* DOCDOC fetch_total */
  1740. static hs_usage_service_related_observation_t *fetch_total = NULL;
  1741. /* DOCDOC fetch_successful */
  1742. static hs_usage_service_related_observation_t *fetch_successful = NULL;
  1743. /* DOCDOC descs */
  1744. static hs_usage_general_period_related_observations_t *descs = NULL;
  1745. /** Creates an empty ordered list element. */
  1746. static hs_usage_list_elem_t *
  1747. hs_usage_list_elem_new(void)
  1748. {
  1749. hs_usage_list_elem_t *e;
  1750. e = tor_malloc_zero(sizeof(hs_usage_list_elem_t));
  1751. rephist_total_alloc += sizeof(hs_usage_list_elem_t);
  1752. e->count = 1;
  1753. e->next = NULL;
  1754. return e;
  1755. }
  1756. /** Creates an empty ordered list. */
  1757. static hs_usage_list_t *
  1758. hs_usage_list_new(void)
  1759. {
  1760. hs_usage_list_t *l;
  1761. l = tor_malloc_zero(sizeof(hs_usage_list_t));
  1762. rephist_total_alloc += sizeof(hs_usage_list_t);
  1763. l->start = NULL;
  1764. l->total_count = 0;
  1765. l->total_service_ids = 0;
  1766. return l;
  1767. }
  1768. /** Creates an empty structure for storing service-related observations. */
  1769. static hs_usage_service_related_observation_t *
  1770. hs_usage_service_related_observation_new(void)
  1771. {
  1772. hs_usage_service_related_observation_t *h;
  1773. h = tor_malloc_zero(sizeof(hs_usage_service_related_observation_t));
  1774. rephist_total_alloc += sizeof(hs_usage_service_related_observation_t);
  1775. h->list = hs_usage_list_new();
  1776. return h;
  1777. }
  1778. /** Creates an empty structure for storing general period-related
  1779. * observations. */
  1780. static hs_usage_general_period_related_observations_t *
  1781. hs_usage_general_period_related_observations_new(void)
  1782. {
  1783. hs_usage_general_period_related_observations_t *p;
  1784. p = tor_malloc_zero(sizeof(hs_usage_general_period_related_observations_t));
  1785. rephist_total_alloc+= sizeof(hs_usage_general_period_related_observations_t);
  1786. return p;
  1787. }
  1788. /** Creates an empty structure for storing period-specific information. */
  1789. static hs_usage_current_observation_period_t *
  1790. hs_usage_current_observation_period_new(void)
  1791. {
  1792. hs_usage_current_observation_period_t *c;
  1793. time_t now;
  1794. c = tor_malloc_zero(sizeof(hs_usage_current_observation_period_t));
  1795. rephist_total_alloc += sizeof(hs_usage_current_observation_period_t);
  1796. now = time(NULL);
  1797. c->start_of_current_period = now;
  1798. c->start_of_next_period = now + NUM_SECS_HS_USAGE_SUM_INTERVAL;
  1799. return c;
  1800. }
  1801. /** Initializes the structures for collecting hidden service usage data. */
  1802. static void
  1803. hs_usage_init(void)
  1804. {
  1805. current_period = hs_usage_current_observation_period_new();
  1806. publish_total = hs_usage_service_related_observation_new();
  1807. publish_novel = hs_usage_service_related_observation_new();
  1808. fetch_total = hs_usage_service_related_observation_new();
  1809. fetch_successful = hs_usage_service_related_observation_new();
  1810. descs = hs_usage_general_period_related_observations_new();
  1811. }
  1812. /** Clears the given ordered list by resetting its attributes and releasing
  1813. * the memory allocated by its elements. */
  1814. static void
  1815. hs_usage_list_clear(hs_usage_list_t *lst)
  1816. {
  1817. /* walk through elements and free memory */
  1818. hs_usage_list_elem_t *current = lst->start;
  1819. hs_usage_list_elem_t *tmp;
  1820. while (current != NULL) {
  1821. tmp = current->next;
  1822. rephist_total_alloc -= sizeof(hs_usage_list_elem_t);
  1823. tor_free(current);
  1824. current = tmp;
  1825. }
  1826. /* reset attributes */
  1827. lst->start = NULL;
  1828. lst->total_count = 0;
  1829. lst->total_service_ids = 0;
  1830. return;
  1831. }
  1832. /** Frees the memory used by the given list. */
  1833. static void
  1834. hs_usage_list_free(hs_usage_list_t *lst)
  1835. {
  1836. if (!lst)
  1837. return;
  1838. hs_usage_list_clear(lst);
  1839. rephist_total_alloc -= sizeof(hs_usage_list_t);
  1840. tor_free(lst);
  1841. }
  1842. /** Frees the memory used by the given service-related observations. */
  1843. static void
  1844. hs_usage_service_related_observation_free(
  1845. hs_usage_service_related_observation_t *s)
  1846. {
  1847. if (!s)
  1848. return;
  1849. hs_usage_list_free(s->list);
  1850. rephist_total_alloc -= sizeof(hs_usage_service_related_observation_t);
  1851. tor_free(s);
  1852. }
  1853. /** Frees the memory used by the given period-specific observations. */
  1854. static void
  1855. hs_usage_general_period_related_observations_free(
  1856. hs_usage_general_period_related_observations_t *s)
  1857. {
  1858. rephist_total_alloc-=sizeof(hs_usage_general_period_related_observations_t);
  1859. tor_free(s);
  1860. }
  1861. /** Frees the memory used by period-specific information. */
  1862. static void
  1863. hs_usage_current_observation_period_free(
  1864. hs_usage_current_observation_period_t *s)
  1865. {
  1866. rephist_total_alloc -= sizeof(hs_usage_current_observation_period_t);
  1867. tor_free(s);
  1868. }
  1869. /** Frees all memory that was used for collecting hidden service usage data. */
  1870. void
  1871. hs_usage_free_all(void)
  1872. {
  1873. hs_usage_general_period_related_observations_free(descs);
  1874. descs = NULL;
  1875. hs_usage_service_related_observation_free(fetch_successful);
  1876. hs_usage_service_related_observation_free(fetch_total);
  1877. hs_usage_service_related_observation_free(publish_novel);
  1878. hs_usage_service_related_observation_free(publish_total);
  1879. fetch_successful = fetch_total = publish_novel = publish_total = NULL;
  1880. hs_usage_current_observation_period_free(current_period);
  1881. current_period = NULL;
  1882. }
  1883. /** Inserts a new occurrence for the given service id to the given ordered
  1884. * list. */
  1885. static void
  1886. hs_usage_insert_value(hs_usage_list_t *lst, const char *service_id)
  1887. {
  1888. /* search if there is already an elem with same service_id in list */
  1889. hs_usage_list_elem_t *current = lst->start;
  1890. hs_usage_list_elem_t *previous = NULL;
  1891. while (current != NULL && strcasecmp(current->service_id,service_id)) {
  1892. previous = current;
  1893. current = current->next;
  1894. }
  1895. /* found an element with same service_id? */
  1896. if (current == NULL) {
  1897. /* not found! append to end (which could also be the end of a zero-length
  1898. * list), don't need to sort (1 is smallest value). */
  1899. /* create elem */
  1900. hs_usage_list_elem_t *e = hs_usage_list_elem_new();
  1901. /* update list attributes (one new elem, one new occurrence) */
  1902. lst->total_count++;
  1903. lst->total_service_ids++;
  1904. /* copy service id to elem */
  1905. strlcpy(e->service_id,service_id,sizeof(e->service_id));
  1906. /* let either l->start or previously last elem point to new elem */
  1907. if (lst->start == NULL) {
  1908. /* this is the first elem */
  1909. lst->start = e;
  1910. } else {
  1911. /* there were elems in the list before */
  1912. previous->next = e;
  1913. }
  1914. } else {
  1915. /* found! add occurrence to elem and consider resorting */
  1916. /* update list attributes (no new elem, but one new occurrence) */
  1917. lst->total_count++;
  1918. /* add occurrence to elem */
  1919. current->count++;
  1920. /* is it another than the first list elem? and has previous elem fewer
  1921. * count than current? then we need to resort */
  1922. if (previous != NULL && previous->count < current->count) {
  1923. /* yes! we need to resort */
  1924. /* remove current elem first */
  1925. previous->next = current->next;
  1926. /* can we prepend elem to all other elements? */
  1927. if (lst->start->count <= current->count) {
  1928. /* yes! prepend elem */
  1929. current->next = lst->start;
  1930. lst->start = current;
  1931. } else {
  1932. /* no! walk through list a second time and insert at correct place */
  1933. hs_usage_list_elem_t *insert_current = lst->start->next;
  1934. hs_usage_list_elem_t *insert_previous = lst->start;
  1935. while (insert_current != NULL &&
  1936. insert_current->count > current->count) {
  1937. insert_previous = insert_current;
  1938. insert_current = insert_current->next;
  1939. }
  1940. /* insert here */
  1941. current->next = insert_current;
  1942. insert_previous->next = current;
  1943. }
  1944. }
  1945. }
  1946. }
  1947. /** Writes the current service-related observations to the history array and
  1948. * clears the observations of the current period. */
  1949. static void
  1950. hs_usage_write_service_related_observations_to_history(
  1951. hs_usage_current_observation_period_t *p,
  1952. hs_usage_service_related_observation_t *h)
  1953. {
  1954. /* walk through the first 20 % of list elements and calculate frequency
  1955. * distributions */
  1956. /* maximum indices for the three frequencies */
  1957. int five_percent_idx = h->list->total_service_ids/20;
  1958. int ten_percent_idx = h->list->total_service_ids/10;
  1959. int twenty_percent_idx = h->list->total_service_ids/5;
  1960. /* temp values */
  1961. uint32_t five_percent = 0;
  1962. uint32_t ten_percent = 0;
  1963. uint32_t twenty_percent = 0;
  1964. /* walk through list */
  1965. hs_usage_list_elem_t *current = h->list->start;
  1966. int i=0;
  1967. while (current != NULL && i <= twenty_percent_idx) {
  1968. twenty_percent += current->count;
  1969. if (i <= ten_percent_idx)
  1970. ten_percent += current->count;
  1971. if (i <= five_percent_idx)
  1972. five_percent += current->count;
  1973. current = current->next;
  1974. i++;
  1975. }
  1976. /* copy frequencies */
  1977. h->twenty[p->next_idx] = twenty_percent;
  1978. h->ten[p->next_idx] = ten_percent;
  1979. h->five[p->next_idx] = five_percent;
  1980. /* copy total number of observations */
  1981. h->totals[p->next_idx] = h->list->total_count;
  1982. /* free memory of old list */
  1983. hs_usage_list_clear(h->list);
  1984. }
  1985. /** Advances to next observation period. */
  1986. static void
  1987. hs_usage_advance_current_observation_period(void)
  1988. {
  1989. /* aggregate observations to history, including frequency distribution
  1990. * arrays */
  1991. hs_usage_write_service_related_observations_to_history(
  1992. current_period, publish_total);
  1993. hs_usage_write_service_related_observations_to_history(
  1994. current_period, publish_novel);
  1995. hs_usage_write_service_related_observations_to_history(
  1996. current_period, fetch_total);
  1997. hs_usage_write_service_related_observations_to_history(
  1998. current_period, fetch_successful);
  1999. /* write current number of descriptors to descs history */
  2000. descs->totals[current_period->next_idx] = rend_cache_size();
  2001. /* advance to next period */
  2002. current_period->next_idx++;
  2003. if (current_period->next_idx == NUM_TOTALS_HS_USAGE)
  2004. current_period->next_idx = 0;
  2005. if (current_period->num_set < NUM_TOTALS_HS_USAGE)
  2006. ++current_period->num_set;
  2007. current_period->start_of_current_period=current_period->start_of_next_period;
  2008. current_period->start_of_next_period += NUM_SECS_HS_USAGE_SUM_INTERVAL;
  2009. }
  2010. /** Checks if the current period is up to date, and if not, advances it. */
  2011. static void
  2012. hs_usage_check_if_current_period_is_up_to_date(time_t now)
  2013. {
  2014. while (now > current_period->start_of_next_period) {
  2015. hs_usage_advance_current_observation_period();
  2016. }
  2017. }
  2018. /** Adds a service-related observation, maybe after advancing to next
  2019. * observation period. */
  2020. static void
  2021. hs_usage_add_service_related_observation(
  2022. hs_usage_service_related_observation_t *h,
  2023. time_t now,
  2024. const char *service_id)
  2025. {
  2026. if (now < current_period->start_of_current_period) {
  2027. /* don't record old data */
  2028. return;
  2029. }
  2030. /* check if we are up-to-date */
  2031. hs_usage_check_if_current_period_is_up_to_date(now);
  2032. /* add observation */
  2033. hs_usage_insert_value(h->list, service_id);
  2034. }
  2035. /** Adds the observation of storing a rendezvous service descriptor to our
  2036. * cache in our role as HS authoritative directory. */
  2037. void
  2038. hs_usage_note_publish_total(const char *service_id, time_t now)
  2039. {
  2040. hs_usage_add_service_related_observation(publish_total, now, service_id);
  2041. }
  2042. /** Adds the observation of storing a novel rendezvous service descriptor to
  2043. * our cache in our role as HS authoritative directory. */
  2044. void
  2045. hs_usage_note_publish_novel(const char *service_id, time_t now)
  2046. {
  2047. hs_usage_add_service_related_observation(publish_novel, now, service_id);
  2048. }
  2049. /** Adds the observation of being requested for a rendezvous service descriptor
  2050. * in our role as HS authoritative directory. */
  2051. void
  2052. hs_usage_note_fetch_total(const char *service_id, time_t now)
  2053. {
  2054. hs_usage_add_service_related_observation(fetch_total, now, service_id);
  2055. }
  2056. /** Adds the observation of being requested for a rendezvous service descriptor
  2057. * in our role as HS authoritative directory and being able to answer that
  2058. * request successfully. */
  2059. void
  2060. hs_usage_note_fetch_successful(const char *service_id, time_t now)
  2061. {
  2062. hs_usage_add_service_related_observation(fetch_successful, now, service_id);
  2063. }
  2064. /** Writes the given circular array to a string. */
  2065. static size_t
  2066. hs_usage_format_history(char *buf, size_t len, uint32_t *data)
  2067. {
  2068. char *cp = buf; /* pointer where we are in the buffer */
  2069. int i, n;
  2070. if (current_period->num_set <= current_period->next_idx) {
  2071. i = 0; /* not been through circular array */
  2072. } else {
  2073. i = current_period->next_idx;
  2074. }
  2075. for (n = 0; n < current_period->num_set; ++n,++i) {
  2076. if (i >= NUM_TOTALS_HS_USAGE)
  2077. i -= NUM_TOTALS_HS_USAGE;
  2078. tor_assert(i < NUM_TOTALS_HS_USAGE);
  2079. if (n == (current_period->num_set-1))
  2080. tor_snprintf(cp, len-(cp-buf), "%d", data[i]);
  2081. else
  2082. tor_snprintf(cp, len-(cp-buf), "%d,", data[i]);
  2083. cp += strlen(cp);
  2084. }
  2085. return cp-buf;
  2086. }
  2087. /** Writes the complete usage history as hidden service authoritative directory
  2088. * to a string. */
  2089. static char *
  2090. hs_usage_format_statistics(void)
  2091. {
  2092. char *buf, *cp, *s = NULL;
  2093. char t[ISO_TIME_LEN+1];
  2094. int r;
  2095. uint32_t *data = NULL;
  2096. size_t len;
  2097. len = (70+20*NUM_TOTALS_HS_USAGE)*11;
  2098. buf = tor_malloc_zero(len);
  2099. cp = buf;
  2100. for (r = 0; r < 11; ++r) {
  2101. switch (r) {
  2102. case 0:
  2103. s = (char*) "publish-total-history";
  2104. data = publish_total->totals;
  2105. break;
  2106. case 1:
  2107. s = (char*) "publish-novel-history";
  2108. data = publish_novel->totals;
  2109. break;
  2110. case 2:
  2111. s = (char*) "publish-top-5-percent-history";
  2112. data = publish_total->five;
  2113. break;
  2114. case 3:
  2115. s = (char*) "publish-top-10-percent-history";
  2116. data = publish_total->ten;
  2117. break;
  2118. case 4:
  2119. s = (char*) "publish-top-20-percent-history";
  2120. data = publish_total->twenty;
  2121. break;
  2122. case 5:
  2123. s = (char*) "fetch-total-history";
  2124. data = fetch_total->totals;
  2125. break;
  2126. case 6:
  2127. s = (char*) "fetch-successful-history";
  2128. data = fetch_successful->totals;
  2129. break;
  2130. case 7:
  2131. s = (char*) "fetch-top-5-percent-history";
  2132. data = fetch_total->five;
  2133. break;
  2134. case 8:
  2135. s = (char*) "fetch-top-10-percent-history";
  2136. data = fetch_total->ten;
  2137. break;
  2138. case 9:
  2139. s = (char*) "fetch-top-20-percent-history";
  2140. data = fetch_total->twenty;
  2141. break;
  2142. case 10:
  2143. s = (char*) "desc-total-history";
  2144. data = descs->totals;
  2145. break;
  2146. }
  2147. format_iso_time(t, current_period->start_of_current_period);
  2148. tor_snprintf(cp, len-(cp-buf), "%s %s (%d s) ", s, t,
  2149. NUM_SECS_HS_USAGE_SUM_INTERVAL);
  2150. cp += strlen(cp);
  2151. cp += hs_usage_format_history(cp, len-(cp-buf), data);
  2152. strlcat(cp, "\n", len-(cp-buf));
  2153. ++cp;
  2154. }
  2155. return buf;
  2156. }
  2157. /** Write current statistics about hidden service usage to file. */
  2158. void
  2159. hs_usage_write_statistics_to_file(time_t now)
  2160. {
  2161. char *buf;
  2162. size_t len;
  2163. char *fname;
  2164. or_options_t *options = get_options();
  2165. /* check if we are up-to-date */
  2166. hs_usage_check_if_current_period_is_up_to_date(now);
  2167. buf = hs_usage_format_statistics();
  2168. len = strlen(options->DataDirectory) + 16;
  2169. fname = tor_malloc(len);
  2170. tor_snprintf(fname, len, "%s"PATH_SEPARATOR"hsusage",
  2171. options->DataDirectory);
  2172. write_str_to_file(fname,buf,0);
  2173. tor_free(buf);
  2174. tor_free(fname);
  2175. }