rephist.c 82 KB

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