mainloop.c 82 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550
  1. /* Copyright (c) 2001 Matej Pfajfar.
  2. * Copyright (c) 2001-2004, Roger Dingledine.
  3. * Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
  4. * Copyright (c) 2007-2019, The Tor Project, Inc. */
  5. /* See LICENSE for licensing information */
  6. /**
  7. * \file mainloop.c
  8. * \brief Toplevel module. Handles signals, multiplexes between
  9. * connections, implements main loop, and drives scheduled events.
  10. *
  11. * For the main loop itself; see run_main_loop_once(). It invokes the rest of
  12. * Tor mostly through Libevent callbacks. Libevent callbacks can happen when
  13. * a timer elapses, a signal is received, a socket is ready to read or write,
  14. * or an event is manually activated.
  15. *
  16. * Most events in Tor are driven from these callbacks:
  17. * <ul>
  18. * <li>conn_read_callback() and conn_write_callback() here, which are
  19. * invoked when a socket is ready to read or write respectively.
  20. * <li>signal_callback(), which handles incoming signals.
  21. * </ul>
  22. * Other events are used for specific purposes, or for building more complex
  23. * control structures. If you search for usage of tor_libevent_new(), you
  24. * will find all the events that we construct in Tor.
  25. *
  26. * Tor has numerous housekeeping operations that need to happen
  27. * regularly. They are handled in different ways:
  28. * <ul>
  29. * <li>The most frequent operations are handled after every read or write
  30. * event, at the end of connection_handle_read() and
  31. * connection_handle_write().
  32. *
  33. * <li>The next most frequent operations happen after each invocation of the
  34. * main loop, in run_main_loop_once().
  35. *
  36. * <li>Once per second, we run all of the operations listed in
  37. * second_elapsed_callback(), and in its child, run_scheduled_events().
  38. *
  39. * <li>Once-a-second operations are handled in second_elapsed_callback().
  40. *
  41. * <li>More infrequent operations take place based on the periodic event
  42. * driver in periodic.c . These are stored in the periodic_events[]
  43. * table.
  44. * </ul>
  45. *
  46. **/
  47. #define MAINLOOP_PRIVATE
  48. #include "core/or/or.h"
  49. #include "app/config/config.h"
  50. #include "app/config/statefile.h"
  51. #include "app/main/ntmain.h"
  52. #include "core/mainloop/connection.h"
  53. #include "core/mainloop/cpuworker.h"
  54. #include "core/mainloop/mainloop.h"
  55. #include "core/mainloop/netstatus.h"
  56. #include "core/mainloop/periodic.h"
  57. #include "core/or/channel.h"
  58. #include "core/or/channelpadding.h"
  59. #include "core/or/channeltls.h"
  60. #include "core/or/circuitbuild.h"
  61. #include "core/or/circuitlist.h"
  62. #include "core/or/circuituse.h"
  63. #include "core/or/connection_edge.h"
  64. #include "core/or/connection_or.h"
  65. #include "core/or/dos.h"
  66. #include "core/or/status.h"
  67. #include "feature/client/addressmap.h"
  68. #include "feature/client/bridges.h"
  69. #include "feature/client/dnsserv.h"
  70. #include "feature/client/entrynodes.h"
  71. #include "feature/client/transports.h"
  72. #include "feature/control/control.h"
  73. #include "feature/control/control_events.h"
  74. #include "feature/dirauth/authmode.h"
  75. #include "feature/dircache/consdiffmgr.h"
  76. #include "feature/dircache/dirserv.h"
  77. #include "feature/dircommon/directory.h"
  78. #include "feature/hibernate/hibernate.h"
  79. #include "feature/hs/hs_cache.h"
  80. #include "feature/hs/hs_client.h"
  81. #include "feature/hs/hs_service.h"
  82. #include "feature/nodelist/microdesc.h"
  83. #include "feature/nodelist/networkstatus.h"
  84. #include "feature/nodelist/nodelist.h"
  85. #include "feature/nodelist/routerlist.h"
  86. #include "feature/relay/dns.h"
  87. #include "feature/relay/routerkeys.h"
  88. #include "feature/relay/routermode.h"
  89. #include "feature/relay/selftest.h"
  90. #include "feature/rend/rendcache.h"
  91. #include "feature/rend/rendservice.h"
  92. #include "feature/stats/geoip_stats.h"
  93. #include "feature/stats/predict_ports.h"
  94. #include "feature/stats/rephist.h"
  95. #include "lib/buf/buffers.h"
  96. #include "lib/crypt_ops/crypto_rand.h"
  97. #include "lib/err/backtrace.h"
  98. #include "lib/tls/buffers_tls.h"
  99. #include "lib/net/buffers_net.h"
  100. #include "lib/evloop/compat_libevent.h"
  101. #include <event2/event.h>
  102. #include "core/or/cell_st.h"
  103. #include "core/or/entry_connection_st.h"
  104. #include "feature/nodelist/networkstatus_st.h"
  105. #include "core/or/or_connection_st.h"
  106. #include "app/config/or_state_st.h"
  107. #include "feature/nodelist/routerinfo_st.h"
  108. #include "core/or/socks_request_st.h"
  109. #ifdef HAVE_UNISTD_H
  110. #include <unistd.h>
  111. #endif
  112. #ifdef HAVE_SYSTEMD
  113. # if defined(__COVERITY__) && !defined(__INCLUDE_LEVEL__)
  114. /* Systemd's use of gcc's __INCLUDE_LEVEL__ extension macro appears to confuse
  115. * Coverity. Here's a kludge to unconfuse it.
  116. */
  117. # define __INCLUDE_LEVEL__ 2
  118. #endif /* defined(__COVERITY__) && !defined(__INCLUDE_LEVEL__) */
  119. #include <systemd/sd-daemon.h>
  120. #endif /* defined(HAVE_SYSTEMD) */
  121. /* Token bucket for all traffic. */
  122. token_bucket_rw_t global_bucket;
  123. /* Token bucket for relayed traffic. */
  124. token_bucket_rw_t global_relayed_bucket;
  125. /* XXX we might want to keep stats about global_relayed_*_bucket too. Or not.*/
  126. /** How many bytes have we read since we started the process? */
  127. static uint64_t stats_n_bytes_read = 0;
  128. /** How many bytes have we written since we started the process? */
  129. static uint64_t stats_n_bytes_written = 0;
  130. /** What time did this process start up? */
  131. time_t time_of_process_start = 0;
  132. /** How many seconds have we been running? */
  133. static long stats_n_seconds_working = 0;
  134. /** How many times have we returned from the main loop successfully? */
  135. static uint64_t stats_n_main_loop_successes = 0;
  136. /** How many times have we received an error from the main loop? */
  137. static uint64_t stats_n_main_loop_errors = 0;
  138. /** How many times have we returned from the main loop with no events. */
  139. static uint64_t stats_n_main_loop_idle = 0;
  140. /** How often will we honor SIGNEWNYM requests? */
  141. #define MAX_SIGNEWNYM_RATE 10
  142. /** When did we last process a SIGNEWNYM request? */
  143. static time_t time_of_last_signewnym = 0;
  144. /** Is there a signewnym request we're currently waiting to handle? */
  145. static int signewnym_is_pending = 0;
  146. /** Mainloop event for the deferred signewnym call. */
  147. static mainloop_event_t *handle_deferred_signewnym_ev = NULL;
  148. /** How many times have we called newnym? */
  149. static unsigned newnym_epoch = 0;
  150. /** Smartlist of all open connections. */
  151. STATIC smartlist_t *connection_array = NULL;
  152. /** List of connections that have been marked for close and need to be freed
  153. * and removed from connection_array. */
  154. static smartlist_t *closeable_connection_lst = NULL;
  155. /** List of linked connections that are currently reading data into their
  156. * inbuf from their partner's outbuf. */
  157. static smartlist_t *active_linked_connection_lst = NULL;
  158. /** Flag: Set to true iff we entered the current libevent main loop via
  159. * <b>loop_once</b>. If so, there's no need to trigger a loopexit in order
  160. * to handle linked connections. */
  161. static int called_loop_once = 0;
  162. /** Flag: if true, it's time to shut down, so the main loop should exit as
  163. * soon as possible.
  164. */
  165. static int main_loop_should_exit = 0;
  166. /** The return value that the main loop should yield when it exits, if
  167. * main_loop_should_exit is true.
  168. */
  169. static int main_loop_exit_value = 0;
  170. /** We set this to 1 when we've opened a circuit, so we can print a log
  171. * entry to inform the user that Tor is working. We set it to 0 when
  172. * we think the fact that we once opened a circuit doesn't mean we can do so
  173. * any longer (a big time jump happened, when we notice our directory is
  174. * heinously out-of-date, etc.
  175. */
  176. static int can_complete_circuits = 0;
  177. /** How often do we check for router descriptors that we should download
  178. * when we have too little directory info? */
  179. #define GREEDY_DESCRIPTOR_RETRY_INTERVAL (10)
  180. /** How often do we check for router descriptors that we should download
  181. * when we have enough directory info? */
  182. #define LAZY_DESCRIPTOR_RETRY_INTERVAL (60)
  183. static int conn_close_if_marked(int i);
  184. static void connection_start_reading_from_linked_conn(connection_t *conn);
  185. static int connection_should_read_from_linked_conn(connection_t *conn);
  186. static void conn_read_callback(evutil_socket_t fd, short event, void *_conn);
  187. static void conn_write_callback(evutil_socket_t fd, short event, void *_conn);
  188. static void shutdown_did_not_work_callback(evutil_socket_t fd, short event,
  189. void *arg) ATTR_NORETURN;
  190. /****************************************************************************
  191. *
  192. * This section contains accessors and other methods on the connection_array
  193. * variables (which are global within this file and unavailable outside it).
  194. *
  195. ****************************************************************************/
  196. /** Return 1 if we have successfully built a circuit, and nothing has changed
  197. * to make us think that maybe we can't.
  198. */
  199. int
  200. have_completed_a_circuit(void)
  201. {
  202. return can_complete_circuits;
  203. }
  204. /** Note that we have successfully built a circuit, so that reachability
  205. * testing and introduction points and so on may be attempted. */
  206. void
  207. note_that_we_completed_a_circuit(void)
  208. {
  209. can_complete_circuits = 1;
  210. }
  211. /** Note that something has happened (like a clock jump, or DisableNetwork) to
  212. * make us think that maybe we can't complete circuits. */
  213. void
  214. note_that_we_maybe_cant_complete_circuits(void)
  215. {
  216. can_complete_circuits = 0;
  217. }
  218. /** Add <b>conn</b> to the array of connections that we can poll on. The
  219. * connection's socket must be set; the connection starts out
  220. * non-reading and non-writing.
  221. */
  222. int
  223. connection_add_impl(connection_t *conn, int is_connecting)
  224. {
  225. tor_assert(conn);
  226. tor_assert(SOCKET_OK(conn->s) ||
  227. conn->linked ||
  228. (conn->type == CONN_TYPE_AP &&
  229. TO_EDGE_CONN(conn)->is_dns_request));
  230. tor_assert(conn->conn_array_index == -1); /* can only connection_add once */
  231. conn->conn_array_index = smartlist_len(connection_array);
  232. smartlist_add(connection_array, conn);
  233. (void) is_connecting;
  234. if (SOCKET_OK(conn->s) || conn->linked) {
  235. conn->read_event = tor_event_new(tor_libevent_get_base(),
  236. conn->s, EV_READ|EV_PERSIST, conn_read_callback, conn);
  237. conn->write_event = tor_event_new(tor_libevent_get_base(),
  238. conn->s, EV_WRITE|EV_PERSIST, conn_write_callback, conn);
  239. /* XXXX CHECK FOR NULL RETURN! */
  240. }
  241. log_debug(LD_NET,"new conn type %s, socket %d, address %s, n_conns %d.",
  242. conn_type_to_string(conn->type), (int)conn->s, conn->address,
  243. smartlist_len(connection_array));
  244. return 0;
  245. }
  246. /** Tell libevent that we don't care about <b>conn</b> any more. */
  247. void
  248. connection_unregister_events(connection_t *conn)
  249. {
  250. if (conn->read_event) {
  251. if (event_del(conn->read_event))
  252. log_warn(LD_BUG, "Error removing read event for %d", (int)conn->s);
  253. tor_free(conn->read_event);
  254. }
  255. if (conn->write_event) {
  256. if (event_del(conn->write_event))
  257. log_warn(LD_BUG, "Error removing write event for %d", (int)conn->s);
  258. tor_free(conn->write_event);
  259. }
  260. if (conn->type == CONN_TYPE_AP_DNS_LISTENER) {
  261. dnsserv_close_listener(conn);
  262. }
  263. }
  264. /** Remove the connection from the global list, and remove the
  265. * corresponding poll entry. Calling this function will shift the last
  266. * connection (if any) into the position occupied by conn.
  267. */
  268. int
  269. connection_remove(connection_t *conn)
  270. {
  271. int current_index;
  272. connection_t *tmp;
  273. tor_assert(conn);
  274. log_debug(LD_NET,"removing socket %d (type %s), n_conns now %d",
  275. (int)conn->s, conn_type_to_string(conn->type),
  276. smartlist_len(connection_array));
  277. if (conn->type == CONN_TYPE_AP && conn->socket_family == AF_UNIX) {
  278. log_info(LD_NET, "Closing SOCKS Unix socket connection");
  279. }
  280. control_event_conn_bandwidth(conn);
  281. tor_assert(conn->conn_array_index >= 0);
  282. current_index = conn->conn_array_index;
  283. connection_unregister_events(conn); /* This is redundant, but cheap. */
  284. if (current_index == smartlist_len(connection_array)-1) { /* at the end */
  285. smartlist_del(connection_array, current_index);
  286. return 0;
  287. }
  288. /* replace this one with the one at the end */
  289. smartlist_del(connection_array, current_index);
  290. tmp = smartlist_get(connection_array, current_index);
  291. tmp->conn_array_index = current_index;
  292. return 0;
  293. }
  294. /** If <b>conn</b> is an edge conn, remove it from the list
  295. * of conn's on this circuit. If it's not on an edge,
  296. * flush and send destroys for all circuits on this conn.
  297. *
  298. * Remove it from connection_array (if applicable) and
  299. * from closeable_connection_list.
  300. *
  301. * Then free it.
  302. */
  303. static void
  304. connection_unlink(connection_t *conn)
  305. {
  306. connection_about_to_close_connection(conn);
  307. if (conn->conn_array_index >= 0) {
  308. connection_remove(conn);
  309. }
  310. if (conn->linked_conn) {
  311. conn->linked_conn->linked_conn = NULL;
  312. if (! conn->linked_conn->marked_for_close &&
  313. conn->linked_conn->reading_from_linked_conn)
  314. connection_start_reading(conn->linked_conn);
  315. conn->linked_conn = NULL;
  316. }
  317. smartlist_remove(closeable_connection_lst, conn);
  318. smartlist_remove(active_linked_connection_lst, conn);
  319. if (conn->type == CONN_TYPE_EXIT) {
  320. assert_connection_edge_not_dns_pending(TO_EDGE_CONN(conn));
  321. }
  322. if (conn->type == CONN_TYPE_OR) {
  323. if (!tor_digest_is_zero(TO_OR_CONN(conn)->identity_digest))
  324. connection_or_clear_identity(TO_OR_CONN(conn));
  325. /* connection_unlink() can only get called if the connection
  326. * was already on the closeable list, and it got there by
  327. * connection_mark_for_close(), which was called from
  328. * connection_or_close_normally() or
  329. * connection_or_close_for_error(), so the channel should
  330. * already be in CHANNEL_STATE_CLOSING, and then the
  331. * connection_about_to_close_connection() goes to
  332. * connection_or_about_to_close(), which calls channel_closed()
  333. * to notify the channel_t layer, and closed the channel, so
  334. * nothing more to do here to deal with the channel associated
  335. * with an orconn.
  336. */
  337. }
  338. connection_free(conn);
  339. }
  340. /** Event that invokes schedule_active_linked_connections_cb. */
  341. static mainloop_event_t *schedule_active_linked_connections_event = NULL;
  342. /**
  343. * Callback: used to activate read events for all linked connections, so
  344. * libevent knows to call their read callbacks. This callback run as a
  345. * postloop event, so that the events _it_ activates don't happen until
  346. * Libevent has a chance to check for other events.
  347. */
  348. static void
  349. schedule_active_linked_connections_cb(mainloop_event_t *event, void *arg)
  350. {
  351. (void)event;
  352. (void)arg;
  353. /* All active linked conns should get their read events activated,
  354. * so that libevent knows to run their callbacks. */
  355. SMARTLIST_FOREACH(active_linked_connection_lst, connection_t *, conn,
  356. event_active(conn->read_event, EV_READ, 1));
  357. /* Reactivate the event if we still have connections in the active list.
  358. *
  359. * A linked connection doesn't get woken up by I/O but rather artificially
  360. * by this event callback. It has directory data spooled in it and it is
  361. * sent incrementally by small chunks unless spool_eagerly is true. For that
  362. * to happen, we need to induce the activation of the read event so it can
  363. * be flushed. */
  364. if (smartlist_len(active_linked_connection_lst)) {
  365. mainloop_event_activate(schedule_active_linked_connections_event);
  366. }
  367. }
  368. /** Initialize the global connection list, closeable connection list,
  369. * and active connection list. */
  370. void
  371. tor_init_connection_lists(void)
  372. {
  373. if (!connection_array)
  374. connection_array = smartlist_new();
  375. if (!closeable_connection_lst)
  376. closeable_connection_lst = smartlist_new();
  377. if (!active_linked_connection_lst)
  378. active_linked_connection_lst = smartlist_new();
  379. }
  380. /** Schedule <b>conn</b> to be closed. **/
  381. void
  382. add_connection_to_closeable_list(connection_t *conn)
  383. {
  384. tor_assert(!smartlist_contains(closeable_connection_lst, conn));
  385. tor_assert(conn->marked_for_close);
  386. assert_connection_ok(conn, time(NULL));
  387. smartlist_add(closeable_connection_lst, conn);
  388. mainloop_schedule_postloop_cleanup();
  389. }
  390. /** Return 1 if conn is on the closeable list, else return 0. */
  391. int
  392. connection_is_on_closeable_list(connection_t *conn)
  393. {
  394. return smartlist_contains(closeable_connection_lst, conn);
  395. }
  396. /** Return true iff conn is in the current poll array. */
  397. int
  398. connection_in_array(connection_t *conn)
  399. {
  400. return smartlist_contains(connection_array, conn);
  401. }
  402. /** Set <b>*array</b> to an array of all connections. <b>*array</b> must not
  403. * be modified.
  404. */
  405. MOCK_IMPL(smartlist_t *,
  406. get_connection_array, (void))
  407. {
  408. if (!connection_array)
  409. connection_array = smartlist_new();
  410. return connection_array;
  411. }
  412. /**
  413. * Return the amount of network traffic read, in bytes, over the life of this
  414. * process.
  415. */
  416. MOCK_IMPL(uint64_t,
  417. get_bytes_read,(void))
  418. {
  419. return stats_n_bytes_read;
  420. }
  421. /**
  422. * Return the amount of network traffic read, in bytes, over the life of this
  423. * process.
  424. */
  425. MOCK_IMPL(uint64_t,
  426. get_bytes_written,(void))
  427. {
  428. return stats_n_bytes_written;
  429. }
  430. /**
  431. * Increment the amount of network traffic read and written, over the life of
  432. * this process.
  433. */
  434. void
  435. stats_increment_bytes_read_and_written(uint64_t r, uint64_t w)
  436. {
  437. stats_n_bytes_read += r;
  438. stats_n_bytes_written += w;
  439. }
  440. /** Set the event mask on <b>conn</b> to <b>events</b>. (The event
  441. * mask is a bitmask whose bits are READ_EVENT and WRITE_EVENT)
  442. */
  443. void
  444. connection_watch_events(connection_t *conn, watchable_events_t events)
  445. {
  446. if (events & READ_EVENT)
  447. connection_start_reading(conn);
  448. else
  449. connection_stop_reading(conn);
  450. if (events & WRITE_EVENT)
  451. connection_start_writing(conn);
  452. else
  453. connection_stop_writing(conn);
  454. }
  455. /** Return true iff <b>conn</b> is listening for read events. */
  456. int
  457. connection_is_reading(connection_t *conn)
  458. {
  459. tor_assert(conn);
  460. return conn->reading_from_linked_conn ||
  461. (conn->read_event && event_pending(conn->read_event, EV_READ, NULL));
  462. }
  463. /** Reset our main loop counters. */
  464. void
  465. reset_main_loop_counters(void)
  466. {
  467. stats_n_main_loop_successes = 0;
  468. stats_n_main_loop_errors = 0;
  469. stats_n_main_loop_idle = 0;
  470. }
  471. /** Increment the main loop success counter. */
  472. static void
  473. increment_main_loop_success_count(void)
  474. {
  475. ++stats_n_main_loop_successes;
  476. }
  477. /** Get the main loop success counter. */
  478. uint64_t
  479. get_main_loop_success_count(void)
  480. {
  481. return stats_n_main_loop_successes;
  482. }
  483. /** Increment the main loop error counter. */
  484. static void
  485. increment_main_loop_error_count(void)
  486. {
  487. ++stats_n_main_loop_errors;
  488. }
  489. /** Get the main loop error counter. */
  490. uint64_t
  491. get_main_loop_error_count(void)
  492. {
  493. return stats_n_main_loop_errors;
  494. }
  495. /** Increment the main loop idle counter. */
  496. static void
  497. increment_main_loop_idle_count(void)
  498. {
  499. ++stats_n_main_loop_idle;
  500. }
  501. /** Get the main loop idle counter. */
  502. uint64_t
  503. get_main_loop_idle_count(void)
  504. {
  505. return stats_n_main_loop_idle;
  506. }
  507. /** Check whether <b>conn</b> is correct in having (or not having) a
  508. * read/write event (passed in <b>ev</b>). On success, return 0. On failure,
  509. * log a warning and return -1. */
  510. static int
  511. connection_check_event(connection_t *conn, struct event *ev)
  512. {
  513. int bad;
  514. if (conn->type == CONN_TYPE_AP && TO_EDGE_CONN(conn)->is_dns_request) {
  515. /* DNS requests which we launch through the dnsserv.c module do not have
  516. * any underlying socket or any underlying linked connection, so they
  517. * shouldn't have any attached events either.
  518. */
  519. bad = ev != NULL;
  520. } else {
  521. /* Everything else should have an underlying socket, or a linked
  522. * connection (which is also tracked with a read_event/write_event pair).
  523. */
  524. bad = ev == NULL;
  525. }
  526. if (bad) {
  527. log_warn(LD_BUG, "Event missing on connection %p [%s;%s]. "
  528. "socket=%d. linked=%d. "
  529. "is_dns_request=%d. Marked_for_close=%s:%d",
  530. conn,
  531. conn_type_to_string(conn->type),
  532. conn_state_to_string(conn->type, conn->state),
  533. (int)conn->s, (int)conn->linked,
  534. (conn->type == CONN_TYPE_AP &&
  535. TO_EDGE_CONN(conn)->is_dns_request),
  536. conn->marked_for_close_file ? conn->marked_for_close_file : "-",
  537. conn->marked_for_close
  538. );
  539. log_backtrace(LOG_WARN, LD_BUG, "Backtrace attached.");
  540. return -1;
  541. }
  542. return 0;
  543. }
  544. /** Tell the main loop to stop notifying <b>conn</b> of any read events. */
  545. MOCK_IMPL(void,
  546. connection_stop_reading,(connection_t *conn))
  547. {
  548. tor_assert(conn);
  549. if (connection_check_event(conn, conn->read_event) < 0) {
  550. return;
  551. }
  552. if (conn->linked) {
  553. conn->reading_from_linked_conn = 0;
  554. connection_stop_reading_from_linked_conn(conn);
  555. } else {
  556. if (event_del(conn->read_event))
  557. log_warn(LD_NET, "Error from libevent setting read event state for %d "
  558. "to unwatched: %s",
  559. (int)conn->s,
  560. tor_socket_strerror(tor_socket_errno(conn->s)));
  561. }
  562. }
  563. /** Tell the main loop to start notifying <b>conn</b> of any read events. */
  564. MOCK_IMPL(void,
  565. connection_start_reading,(connection_t *conn))
  566. {
  567. tor_assert(conn);
  568. if (connection_check_event(conn, conn->read_event) < 0) {
  569. return;
  570. }
  571. if (conn->linked) {
  572. conn->reading_from_linked_conn = 1;
  573. if (connection_should_read_from_linked_conn(conn))
  574. connection_start_reading_from_linked_conn(conn);
  575. } else {
  576. if (event_add(conn->read_event, NULL))
  577. log_warn(LD_NET, "Error from libevent setting read event state for %d "
  578. "to watched: %s",
  579. (int)conn->s,
  580. tor_socket_strerror(tor_socket_errno(conn->s)));
  581. }
  582. }
  583. /** Return true iff <b>conn</b> is listening for write events. */
  584. int
  585. connection_is_writing(connection_t *conn)
  586. {
  587. tor_assert(conn);
  588. return conn->writing_to_linked_conn ||
  589. (conn->write_event && event_pending(conn->write_event, EV_WRITE, NULL));
  590. }
  591. /** Tell the main loop to stop notifying <b>conn</b> of any write events. */
  592. MOCK_IMPL(void,
  593. connection_stop_writing,(connection_t *conn))
  594. {
  595. tor_assert(conn);
  596. if (connection_check_event(conn, conn->write_event) < 0) {
  597. return;
  598. }
  599. if (conn->linked) {
  600. conn->writing_to_linked_conn = 0;
  601. if (conn->linked_conn)
  602. connection_stop_reading_from_linked_conn(conn->linked_conn);
  603. } else {
  604. if (event_del(conn->write_event))
  605. log_warn(LD_NET, "Error from libevent setting write event state for %d "
  606. "to unwatched: %s",
  607. (int)conn->s,
  608. tor_socket_strerror(tor_socket_errno(conn->s)));
  609. }
  610. }
  611. /** Tell the main loop to start notifying <b>conn</b> of any write events. */
  612. MOCK_IMPL(void,
  613. connection_start_writing,(connection_t *conn))
  614. {
  615. tor_assert(conn);
  616. if (connection_check_event(conn, conn->write_event) < 0) {
  617. return;
  618. }
  619. if (conn->linked) {
  620. conn->writing_to_linked_conn = 1;
  621. if (conn->linked_conn &&
  622. connection_should_read_from_linked_conn(conn->linked_conn))
  623. connection_start_reading_from_linked_conn(conn->linked_conn);
  624. } else {
  625. if (event_add(conn->write_event, NULL))
  626. log_warn(LD_NET, "Error from libevent setting write event state for %d "
  627. "to watched: %s",
  628. (int)conn->s,
  629. tor_socket_strerror(tor_socket_errno(conn->s)));
  630. }
  631. }
  632. /** Return true iff <b>conn</b> is linked conn, and reading from the conn
  633. * linked to it would be good and feasible. (Reading is "feasible" if the
  634. * other conn exists and has data in its outbuf, and is "good" if we have our
  635. * reading_from_linked_conn flag set and the other conn has its
  636. * writing_to_linked_conn flag set.)*/
  637. static int
  638. connection_should_read_from_linked_conn(connection_t *conn)
  639. {
  640. if (conn->linked && conn->reading_from_linked_conn) {
  641. if (! conn->linked_conn ||
  642. (conn->linked_conn->writing_to_linked_conn &&
  643. buf_datalen(conn->linked_conn->outbuf)))
  644. return 1;
  645. }
  646. return 0;
  647. }
  648. /** Event to run 'shutdown did not work callback'. */
  649. static struct event *shutdown_did_not_work_event = NULL;
  650. /** Failsafe measure that should never actually be necessary: If
  651. * tor_shutdown_event_loop_and_exit() somehow doesn't successfully exit the
  652. * event loop, then this callback will kill Tor with an assertion failure
  653. * seconds later
  654. */
  655. static void
  656. shutdown_did_not_work_callback(evutil_socket_t fd, short event, void *arg)
  657. {
  658. // LCOV_EXCL_START
  659. (void) fd;
  660. (void) event;
  661. (void) arg;
  662. tor_assert_unreached();
  663. // LCOV_EXCL_STOP
  664. }
  665. #ifdef ENABLE_RESTART_DEBUGGING
  666. static struct event *tor_shutdown_event_loop_for_restart_event = NULL;
  667. static void
  668. tor_shutdown_event_loop_for_restart_cb(
  669. evutil_socket_t fd, short event, void *arg)
  670. {
  671. (void)fd;
  672. (void)event;
  673. (void)arg;
  674. tor_event_free(tor_shutdown_event_loop_for_restart_event);
  675. tor_shutdown_event_loop_and_exit(0);
  676. }
  677. #endif /* defined(ENABLE_RESTART_DEBUGGING) */
  678. /**
  679. * After finishing the current callback (if any), shut down the main loop,
  680. * clean up the process, and exit with <b>exitcode</b>.
  681. */
  682. void
  683. tor_shutdown_event_loop_and_exit(int exitcode)
  684. {
  685. if (main_loop_should_exit)
  686. return; /* Ignore multiple calls to this function. */
  687. main_loop_should_exit = 1;
  688. main_loop_exit_value = exitcode;
  689. /* Die with an assertion failure in ten seconds, if for some reason we don't
  690. * exit normally. */
  691. /* XXXX We should consider this code if it's never used. */
  692. struct timeval ten_seconds = { 10, 0 };
  693. shutdown_did_not_work_event = tor_evtimer_new(
  694. tor_libevent_get_base(),
  695. shutdown_did_not_work_callback, NULL);
  696. event_add(shutdown_did_not_work_event, &ten_seconds);
  697. /* Unlike exit_loop_after_delay(), exit_loop_after_callback
  698. * prevents other callbacks from running. */
  699. tor_libevent_exit_loop_after_callback(tor_libevent_get_base());
  700. }
  701. /** Return true iff tor_shutdown_event_loop_and_exit() has been called. */
  702. int
  703. tor_event_loop_shutdown_is_pending(void)
  704. {
  705. return main_loop_should_exit;
  706. }
  707. /** Helper: Tell the main loop to begin reading bytes into <b>conn</b> from
  708. * its linked connection, if it is not doing so already. Called by
  709. * connection_start_reading and connection_start_writing as appropriate. */
  710. static void
  711. connection_start_reading_from_linked_conn(connection_t *conn)
  712. {
  713. tor_assert(conn);
  714. tor_assert(conn->linked == 1);
  715. if (!conn->active_on_link) {
  716. conn->active_on_link = 1;
  717. smartlist_add(active_linked_connection_lst, conn);
  718. mainloop_event_activate(schedule_active_linked_connections_event);
  719. } else {
  720. tor_assert(smartlist_contains(active_linked_connection_lst, conn));
  721. }
  722. }
  723. /** Tell the main loop to stop reading bytes into <b>conn</b> from its linked
  724. * connection, if is currently doing so. Called by connection_stop_reading,
  725. * connection_stop_writing, and connection_read. */
  726. void
  727. connection_stop_reading_from_linked_conn(connection_t *conn)
  728. {
  729. tor_assert(conn);
  730. tor_assert(conn->linked == 1);
  731. if (conn->active_on_link) {
  732. conn->active_on_link = 0;
  733. /* FFFF We could keep an index here so we can smartlist_del
  734. * cleanly. On the other hand, this doesn't show up on profiles,
  735. * so let's leave it alone for now. */
  736. smartlist_remove(active_linked_connection_lst, conn);
  737. } else {
  738. tor_assert(!smartlist_contains(active_linked_connection_lst, conn));
  739. }
  740. }
  741. /** Close all connections that have been scheduled to get closed. */
  742. STATIC void
  743. close_closeable_connections(void)
  744. {
  745. int i;
  746. for (i = 0; i < smartlist_len(closeable_connection_lst); ) {
  747. connection_t *conn = smartlist_get(closeable_connection_lst, i);
  748. if (conn->conn_array_index < 0) {
  749. connection_unlink(conn); /* blow it away right now */
  750. } else {
  751. if (!conn_close_if_marked(conn->conn_array_index))
  752. ++i;
  753. }
  754. }
  755. }
  756. /** Count moribund connections for the OOS handler */
  757. MOCK_IMPL(int,
  758. connection_count_moribund, (void))
  759. {
  760. int moribund = 0;
  761. /*
  762. * Count things we'll try to kill when close_closeable_connections()
  763. * runs next.
  764. */
  765. SMARTLIST_FOREACH_BEGIN(closeable_connection_lst, connection_t *, conn) {
  766. if (SOCKET_OK(conn->s) && connection_is_moribund(conn)) ++moribund;
  767. } SMARTLIST_FOREACH_END(conn);
  768. return moribund;
  769. }
  770. /** Libevent callback: this gets invoked when (connection_t*)<b>conn</b> has
  771. * some data to read. */
  772. static void
  773. conn_read_callback(evutil_socket_t fd, short event, void *_conn)
  774. {
  775. connection_t *conn = _conn;
  776. (void)fd;
  777. (void)event;
  778. log_debug(LD_NET,"socket %d wants to read.",(int)conn->s);
  779. /* assert_connection_ok(conn, time(NULL)); */
  780. if (connection_handle_read(conn) < 0) {
  781. if (!conn->marked_for_close) {
  782. #ifndef _WIN32
  783. log_warn(LD_BUG,"Unhandled error on read for %s connection "
  784. "(fd %d); removing",
  785. conn_type_to_string(conn->type), (int)conn->s);
  786. tor_fragile_assert();
  787. #endif /* !defined(_WIN32) */
  788. if (CONN_IS_EDGE(conn))
  789. connection_edge_end_errno(TO_EDGE_CONN(conn));
  790. connection_mark_for_close(conn);
  791. }
  792. }
  793. assert_connection_ok(conn, time(NULL));
  794. if (smartlist_len(closeable_connection_lst))
  795. close_closeable_connections();
  796. }
  797. /** Libevent callback: this gets invoked when (connection_t*)<b>conn</b> has
  798. * some data to write. */
  799. static void
  800. conn_write_callback(evutil_socket_t fd, short events, void *_conn)
  801. {
  802. connection_t *conn = _conn;
  803. (void)fd;
  804. (void)events;
  805. LOG_FN_CONN(conn, (LOG_DEBUG, LD_NET, "socket %d wants to write.",
  806. (int)conn->s));
  807. /* assert_connection_ok(conn, time(NULL)); */
  808. if (connection_handle_write(conn, 0) < 0) {
  809. if (!conn->marked_for_close) {
  810. /* this connection is broken. remove it. */
  811. log_fn(LOG_WARN,LD_BUG,
  812. "unhandled error on write for %s connection (fd %d); removing",
  813. conn_type_to_string(conn->type), (int)conn->s);
  814. tor_fragile_assert();
  815. if (CONN_IS_EDGE(conn)) {
  816. /* otherwise we cry wolf about duplicate close */
  817. edge_connection_t *edge_conn = TO_EDGE_CONN(conn);
  818. if (!edge_conn->end_reason)
  819. edge_conn->end_reason = END_STREAM_REASON_INTERNAL;
  820. edge_conn->edge_has_sent_end = 1;
  821. }
  822. connection_close_immediate(conn); /* So we don't try to flush. */
  823. connection_mark_for_close(conn);
  824. }
  825. }
  826. assert_connection_ok(conn, time(NULL));
  827. if (smartlist_len(closeable_connection_lst))
  828. close_closeable_connections();
  829. }
  830. /** If the connection at connection_array[i] is marked for close, then:
  831. * - If it has data that it wants to flush, try to flush it.
  832. * - If it _still_ has data to flush, and conn->hold_open_until_flushed is
  833. * true, then leave the connection open and return.
  834. * - Otherwise, remove the connection from connection_array and from
  835. * all other lists, close it, and free it.
  836. * Returns 1 if the connection was closed, 0 otherwise.
  837. */
  838. static int
  839. conn_close_if_marked(int i)
  840. {
  841. connection_t *conn;
  842. int retval;
  843. time_t now;
  844. conn = smartlist_get(connection_array, i);
  845. if (!conn->marked_for_close)
  846. return 0; /* nothing to see here, move along */
  847. now = time(NULL);
  848. assert_connection_ok(conn, now);
  849. /* assert_all_pending_dns_resolves_ok(); */
  850. log_debug(LD_NET,"Cleaning up connection (fd "TOR_SOCKET_T_FORMAT").",
  851. conn->s);
  852. /* If the connection we are about to close was trying to connect to
  853. a proxy server and failed, the client won't be able to use that
  854. proxy. We should warn the user about this. */
  855. if (conn->proxy_state == PROXY_INFANT)
  856. log_failed_proxy_connection(conn);
  857. if ((SOCKET_OK(conn->s) || conn->linked_conn) &&
  858. connection_wants_to_flush(conn)) {
  859. /* s == -1 means it's an incomplete edge connection, or that the socket
  860. * has already been closed as unflushable. */
  861. ssize_t sz = connection_bucket_write_limit(conn, now);
  862. if (!conn->hold_open_until_flushed)
  863. log_info(LD_NET,
  864. "Conn (addr %s, fd %d, type %s, state %d) marked, but wants "
  865. "to flush %d bytes. (Marked at %s:%d)",
  866. escaped_safe_str_client(conn->address),
  867. (int)conn->s, conn_type_to_string(conn->type), conn->state,
  868. (int)conn->outbuf_flushlen,
  869. conn->marked_for_close_file, conn->marked_for_close);
  870. if (conn->linked_conn) {
  871. retval = buf_move_to_buf(conn->linked_conn->inbuf, conn->outbuf,
  872. &conn->outbuf_flushlen);
  873. if (retval >= 0) {
  874. /* The linked conn will notice that it has data when it notices that
  875. * we're gone. */
  876. connection_start_reading_from_linked_conn(conn->linked_conn);
  877. }
  878. log_debug(LD_GENERAL, "Flushed last %d bytes from a linked conn; "
  879. "%d left; flushlen %d; wants-to-flush==%d", retval,
  880. (int)connection_get_outbuf_len(conn),
  881. (int)conn->outbuf_flushlen,
  882. connection_wants_to_flush(conn));
  883. } else if (connection_speaks_cells(conn)) {
  884. if (conn->state == OR_CONN_STATE_OPEN) {
  885. retval = buf_flush_to_tls(conn->outbuf, TO_OR_CONN(conn)->tls, sz,
  886. &conn->outbuf_flushlen);
  887. } else
  888. retval = -1; /* never flush non-open broken tls connections */
  889. } else {
  890. retval = buf_flush_to_socket(conn->outbuf, conn->s, sz,
  891. &conn->outbuf_flushlen);
  892. }
  893. if (retval >= 0 && /* Technically, we could survive things like
  894. TLS_WANT_WRITE here. But don't bother for now. */
  895. conn->hold_open_until_flushed && connection_wants_to_flush(conn)) {
  896. if (retval > 0) {
  897. LOG_FN_CONN(conn, (LOG_INFO,LD_NET,
  898. "Holding conn (fd %d) open for more flushing.",
  899. (int)conn->s));
  900. conn->timestamp_last_write_allowed = now; /* reset so we can flush
  901. * more */
  902. } else if (sz == 0) {
  903. /* Also, retval==0. If we get here, we didn't want to write anything
  904. * (because of rate-limiting) and we didn't. */
  905. /* Connection must flush before closing, but it's being rate-limited.
  906. * Let's remove from Libevent, and mark it as blocked on bandwidth
  907. * so it will be re-added on next token bucket refill. Prevents
  908. * busy Libevent loops where we keep ending up here and returning
  909. * 0 until we are no longer blocked on bandwidth.
  910. */
  911. connection_consider_empty_write_buckets(conn);
  912. /* Make sure that consider_empty_buckets really disabled the
  913. * connection: */
  914. if (BUG(connection_is_writing(conn))) {
  915. connection_write_bw_exhausted(conn, true);
  916. }
  917. /* The connection is being held due to write rate limit and thus will
  918. * flush its data later. We need to stop reading because this
  919. * connection is about to be closed once flushed. It should not
  920. * process anything more coming in at this stage. */
  921. connection_stop_reading(conn);
  922. }
  923. return 0;
  924. }
  925. if (connection_wants_to_flush(conn)) {
  926. log_fn(LOG_INFO, LD_NET, "We stalled too much while trying to write %d "
  927. "bytes to address %s. If this happens a lot, either "
  928. "something is wrong with your network connection, or "
  929. "something is wrong with theirs. "
  930. "(fd %d, type %s, state %d, marked at %s:%d).",
  931. (int)connection_get_outbuf_len(conn),
  932. escaped_safe_str_client(conn->address),
  933. (int)conn->s, conn_type_to_string(conn->type), conn->state,
  934. conn->marked_for_close_file,
  935. conn->marked_for_close);
  936. }
  937. }
  938. connection_unlink(conn); /* unlink, remove, free */
  939. return 1;
  940. }
  941. /** Implementation for directory_all_unreachable. This is done in a callback,
  942. * since otherwise it would complicate Tor's control-flow graph beyond all
  943. * reason.
  944. */
  945. static void
  946. directory_all_unreachable_cb(mainloop_event_t *event, void *arg)
  947. {
  948. (void)event;
  949. (void)arg;
  950. connection_t *conn;
  951. while ((conn = connection_get_by_type_state(CONN_TYPE_AP,
  952. AP_CONN_STATE_CIRCUIT_WAIT))) {
  953. entry_connection_t *entry_conn = TO_ENTRY_CONN(conn);
  954. log_notice(LD_NET,
  955. "Is your network connection down? "
  956. "Failing connection to '%s:%d'.",
  957. safe_str_client(entry_conn->socks_request->address),
  958. entry_conn->socks_request->port);
  959. connection_mark_unattached_ap(entry_conn,
  960. END_STREAM_REASON_NET_UNREACHABLE);
  961. }
  962. control_event_general_error("DIR_ALL_UNREACHABLE");
  963. }
  964. static mainloop_event_t *directory_all_unreachable_cb_event = NULL;
  965. /** We've just tried every dirserver we know about, and none of
  966. * them were reachable. Assume the network is down. Change state
  967. * so next time an application connection arrives we'll delay it
  968. * and try another directory fetch. Kill off all the circuit_wait
  969. * streams that are waiting now, since they will all timeout anyway.
  970. */
  971. void
  972. directory_all_unreachable(time_t now)
  973. {
  974. (void)now;
  975. reset_uptime(); /* reset it */
  976. if (!directory_all_unreachable_cb_event) {
  977. directory_all_unreachable_cb_event =
  978. mainloop_event_new(directory_all_unreachable_cb, NULL);
  979. tor_assert(directory_all_unreachable_cb_event);
  980. }
  981. mainloop_event_activate(directory_all_unreachable_cb_event);
  982. }
  983. /** This function is called whenever we successfully pull down some new
  984. * network statuses or server descriptors. */
  985. void
  986. directory_info_has_arrived(time_t now, int from_cache, int suppress_logs)
  987. {
  988. const or_options_t *options = get_options();
  989. /* if we have enough dir info, then update our guard status with
  990. * whatever we just learned. */
  991. int invalidate_circs = guards_update_all();
  992. if (invalidate_circs) {
  993. circuit_mark_all_unused_circs();
  994. circuit_mark_all_dirty_circs_as_unusable();
  995. }
  996. if (!router_have_minimum_dir_info()) {
  997. int quiet = suppress_logs || from_cache ||
  998. directory_too_idle_to_fetch_descriptors(options, now);
  999. tor_log(quiet ? LOG_INFO : LOG_NOTICE, LD_DIR,
  1000. "I learned some more directory information, but not enough to "
  1001. "build a circuit: %s", get_dir_info_status_string());
  1002. update_all_descriptor_downloads(now);
  1003. return;
  1004. } else {
  1005. if (directory_fetches_from_authorities(options)) {
  1006. update_all_descriptor_downloads(now);
  1007. }
  1008. /* Don't even bother trying to get extrainfo until the rest of our
  1009. * directory info is up-to-date */
  1010. if (options->DownloadExtraInfo)
  1011. update_extrainfo_downloads(now);
  1012. }
  1013. if (server_mode(options) && !net_is_disabled() && !from_cache &&
  1014. (have_completed_a_circuit() || !any_predicted_circuits(now)))
  1015. router_do_reachability_checks(1, 1);
  1016. }
  1017. /** Perform regular maintenance tasks for a single connection. This
  1018. * function gets run once per second per connection by run_scheduled_events.
  1019. */
  1020. static void
  1021. run_connection_housekeeping(int i, time_t now)
  1022. {
  1023. cell_t cell;
  1024. connection_t *conn = smartlist_get(connection_array, i);
  1025. const or_options_t *options = get_options();
  1026. or_connection_t *or_conn;
  1027. channel_t *chan = NULL;
  1028. int have_any_circuits;
  1029. int past_keepalive =
  1030. now >= conn->timestamp_last_write_allowed + options->KeepalivePeriod;
  1031. if (conn->outbuf && !connection_get_outbuf_len(conn) &&
  1032. conn->type == CONN_TYPE_OR)
  1033. TO_OR_CONN(conn)->timestamp_lastempty = now;
  1034. if (conn->marked_for_close) {
  1035. /* nothing to do here */
  1036. return;
  1037. }
  1038. /* Expire any directory connections that haven't been active (sent
  1039. * if a server or received if a client) for 5 min */
  1040. if (conn->type == CONN_TYPE_DIR &&
  1041. ((DIR_CONN_IS_SERVER(conn) &&
  1042. conn->timestamp_last_write_allowed
  1043. + options->TestingDirConnectionMaxStall < now) ||
  1044. (!DIR_CONN_IS_SERVER(conn) &&
  1045. conn->timestamp_last_read_allowed
  1046. + options->TestingDirConnectionMaxStall < now))) {
  1047. log_info(LD_DIR,"Expiring wedged directory conn (fd %d, purpose %d)",
  1048. (int)conn->s, conn->purpose);
  1049. /* This check is temporary; it's to let us know whether we should consider
  1050. * parsing partial serverdesc responses. */
  1051. if (conn->purpose == DIR_PURPOSE_FETCH_SERVERDESC &&
  1052. connection_get_inbuf_len(conn) >= 1024) {
  1053. log_info(LD_DIR,"Trying to extract information from wedged server desc "
  1054. "download.");
  1055. connection_dir_reached_eof(TO_DIR_CONN(conn));
  1056. } else {
  1057. connection_mark_for_close(conn);
  1058. }
  1059. return;
  1060. }
  1061. if (!connection_speaks_cells(conn))
  1062. return; /* we're all done here, the rest is just for OR conns */
  1063. /* If we haven't flushed to an OR connection for a while, then either nuke
  1064. the connection or send a keepalive, depending. */
  1065. or_conn = TO_OR_CONN(conn);
  1066. tor_assert(conn->outbuf);
  1067. chan = TLS_CHAN_TO_BASE(or_conn->chan);
  1068. tor_assert(chan);
  1069. if (channel_num_circuits(chan) != 0) {
  1070. have_any_circuits = 1;
  1071. chan->timestamp_last_had_circuits = now;
  1072. } else {
  1073. have_any_circuits = 0;
  1074. }
  1075. if (channel_is_bad_for_new_circs(TLS_CHAN_TO_BASE(or_conn->chan)) &&
  1076. ! have_any_circuits) {
  1077. /* It's bad for new circuits, and has no unmarked circuits on it:
  1078. * mark it now. */
  1079. log_info(LD_OR,
  1080. "Expiring non-used OR connection to fd %d (%s:%d) [Too old].",
  1081. (int)conn->s, conn->address, conn->port);
  1082. if (conn->state == OR_CONN_STATE_CONNECTING)
  1083. connection_or_connect_failed(TO_OR_CONN(conn),
  1084. END_OR_CONN_REASON_TIMEOUT,
  1085. "Tor gave up on the connection");
  1086. connection_or_close_normally(TO_OR_CONN(conn), 1);
  1087. } else if (!connection_state_is_open(conn)) {
  1088. if (past_keepalive) {
  1089. /* We never managed to actually get this connection open and happy. */
  1090. log_info(LD_OR,"Expiring non-open OR connection to fd %d (%s:%d).",
  1091. (int)conn->s,conn->address, conn->port);
  1092. connection_or_close_normally(TO_OR_CONN(conn), 0);
  1093. }
  1094. } else if (we_are_hibernating() &&
  1095. ! have_any_circuits &&
  1096. !connection_get_outbuf_len(conn)) {
  1097. /* We're hibernating or shutting down, there's no circuits, and nothing to
  1098. * flush.*/
  1099. log_info(LD_OR,"Expiring non-used OR connection to fd %d (%s:%d) "
  1100. "[Hibernating or exiting].",
  1101. (int)conn->s,conn->address, conn->port);
  1102. connection_or_close_normally(TO_OR_CONN(conn), 1);
  1103. } else if (!have_any_circuits &&
  1104. now - or_conn->idle_timeout >=
  1105. chan->timestamp_last_had_circuits) {
  1106. log_info(LD_OR,"Expiring non-used OR connection %"PRIu64" to fd %d "
  1107. "(%s:%d) [no circuits for %d; timeout %d; %scanonical].",
  1108. (chan->global_identifier),
  1109. (int)conn->s, conn->address, conn->port,
  1110. (int)(now - chan->timestamp_last_had_circuits),
  1111. or_conn->idle_timeout,
  1112. or_conn->is_canonical ? "" : "non");
  1113. connection_or_close_normally(TO_OR_CONN(conn), 0);
  1114. } else if (
  1115. now >= or_conn->timestamp_lastempty + options->KeepalivePeriod*10 &&
  1116. now >=
  1117. conn->timestamp_last_write_allowed + options->KeepalivePeriod*10) {
  1118. log_fn(LOG_PROTOCOL_WARN,LD_PROTOCOL,
  1119. "Expiring stuck OR connection to fd %d (%s:%d). (%d bytes to "
  1120. "flush; %d seconds since last write)",
  1121. (int)conn->s, conn->address, conn->port,
  1122. (int)connection_get_outbuf_len(conn),
  1123. (int)(now-conn->timestamp_last_write_allowed));
  1124. connection_or_close_normally(TO_OR_CONN(conn), 0);
  1125. } else if (past_keepalive && !connection_get_outbuf_len(conn)) {
  1126. /* send a padding cell */
  1127. log_fn(LOG_DEBUG,LD_OR,"Sending keepalive to (%s:%d)",
  1128. conn->address, conn->port);
  1129. memset(&cell,0,sizeof(cell_t));
  1130. cell.command = CELL_PADDING;
  1131. connection_or_write_cell_to_buf(&cell, or_conn);
  1132. } else {
  1133. channelpadding_decide_to_pad_channel(chan);
  1134. }
  1135. }
  1136. /** Honor a NEWNYM request: make future requests unlinkable to past
  1137. * requests. */
  1138. static void
  1139. signewnym_impl(time_t now)
  1140. {
  1141. const or_options_t *options = get_options();
  1142. if (!proxy_mode(options)) {
  1143. log_info(LD_CONTROL, "Ignoring SIGNAL NEWNYM because client functionality "
  1144. "is disabled.");
  1145. return;
  1146. }
  1147. circuit_mark_all_dirty_circs_as_unusable();
  1148. addressmap_clear_transient();
  1149. hs_client_purge_state();
  1150. time_of_last_signewnym = now;
  1151. signewnym_is_pending = 0;
  1152. ++newnym_epoch;
  1153. control_event_signal(SIGNEWNYM);
  1154. }
  1155. /** Callback: run a deferred signewnym. */
  1156. static void
  1157. handle_deferred_signewnym_cb(mainloop_event_t *event, void *arg)
  1158. {
  1159. (void)event;
  1160. (void)arg;
  1161. log_info(LD_CONTROL, "Honoring delayed NEWNYM request");
  1162. do_signewnym(time(NULL));
  1163. }
  1164. /** Either perform a signewnym or schedule one, depending on rate limiting. */
  1165. void
  1166. do_signewnym(time_t now)
  1167. {
  1168. if (time_of_last_signewnym + MAX_SIGNEWNYM_RATE > now) {
  1169. const time_t delay_sec =
  1170. time_of_last_signewnym + MAX_SIGNEWNYM_RATE - now;
  1171. if (! signewnym_is_pending) {
  1172. signewnym_is_pending = 1;
  1173. if (!handle_deferred_signewnym_ev) {
  1174. handle_deferred_signewnym_ev =
  1175. mainloop_event_postloop_new(handle_deferred_signewnym_cb, NULL);
  1176. }
  1177. const struct timeval delay_tv = { delay_sec, 0 };
  1178. mainloop_event_schedule(handle_deferred_signewnym_ev, &delay_tv);
  1179. }
  1180. log_notice(LD_CONTROL,
  1181. "Rate limiting NEWNYM request: delaying by %d second(s)",
  1182. (int)(delay_sec));
  1183. } else {
  1184. signewnym_impl(now);
  1185. }
  1186. }
  1187. /** Return the number of times that signewnym has been called. */
  1188. unsigned
  1189. get_signewnym_epoch(void)
  1190. {
  1191. return newnym_epoch;
  1192. }
  1193. /** True iff we have initialized all the members of <b>periodic_events</b>.
  1194. * Used to prevent double-initialization. */
  1195. static int periodic_events_initialized = 0;
  1196. /* Declare all the timer callback functions... */
  1197. #undef CALLBACK
  1198. #define CALLBACK(name) \
  1199. static int name ## _callback(time_t, const or_options_t *)
  1200. CALLBACK(add_entropy);
  1201. CALLBACK(check_expired_networkstatus);
  1202. CALLBACK(clean_caches);
  1203. CALLBACK(clean_consdiffmgr);
  1204. CALLBACK(fetch_networkstatus);
  1205. CALLBACK(heartbeat);
  1206. CALLBACK(hs_service);
  1207. CALLBACK(launch_descriptor_fetches);
  1208. CALLBACK(prune_old_routers);
  1209. CALLBACK(record_bridge_stats);
  1210. CALLBACK(rend_cache_failure_clean);
  1211. CALLBACK(reset_padding_counts);
  1212. CALLBACK(retry_listeners);
  1213. CALLBACK(rotate_x509_certificate);
  1214. CALLBACK(save_state);
  1215. CALLBACK(write_stats_file);
  1216. CALLBACK(control_per_second_events);
  1217. CALLBACK(second_elapsed);
  1218. #undef CALLBACK
  1219. /* Now we declare an array of periodic_event_item_t for each periodic event */
  1220. #define CALLBACK(name, r, f) \
  1221. PERIODIC_EVENT(name, PERIODIC_EVENT_ROLE_ ## r, f)
  1222. #define FL(name) (PERIODIC_EVENT_FLAG_ ## name)
  1223. STATIC periodic_event_item_t mainloop_periodic_events[] = {
  1224. /* Everyone needs to run these. They need to have very long timeouts for
  1225. * that to be safe. */
  1226. CALLBACK(add_entropy, ALL, 0),
  1227. CALLBACK(heartbeat, ALL, 0),
  1228. CALLBACK(reset_padding_counts, ALL, 0),
  1229. /* This is a legacy catch-all callback that runs once per second if
  1230. * we are online and active. */
  1231. CALLBACK(second_elapsed, NET_PARTICIPANT,
  1232. FL(NEED_NET)|FL(RUN_ON_DISABLE)),
  1233. /* XXXX Do we have a reason to do this on a callback? Does it do any good at
  1234. * all? For now, if we're dormant, we can let our listeners decay. */
  1235. CALLBACK(retry_listeners, NET_PARTICIPANT, FL(NEED_NET)),
  1236. /* We need to do these if we're participating in the Tor network. */
  1237. CALLBACK(check_expired_networkstatus, NET_PARTICIPANT, 0),
  1238. CALLBACK(fetch_networkstatus, NET_PARTICIPANT, 0),
  1239. CALLBACK(launch_descriptor_fetches, NET_PARTICIPANT, FL(NEED_NET)),
  1240. CALLBACK(rotate_x509_certificate, NET_PARTICIPANT, 0),
  1241. CALLBACK(check_network_participation, NET_PARTICIPANT, 0),
  1242. /* We need to do these if we're participating in the Tor network, and
  1243. * immediately before we stop. */
  1244. CALLBACK(clean_caches, NET_PARTICIPANT, FL(RUN_ON_DISABLE)),
  1245. CALLBACK(save_state, NET_PARTICIPANT, FL(RUN_ON_DISABLE)),
  1246. CALLBACK(write_stats_file, NET_PARTICIPANT, FL(RUN_ON_DISABLE)),
  1247. CALLBACK(prune_old_routers, NET_PARTICIPANT, FL(RUN_ON_DISABLE)),
  1248. /* Hidden Service service only. */
  1249. CALLBACK(hs_service, HS_SERVICE, FL(NEED_NET)), // XXXX break this down more
  1250. /* Bridge only. */
  1251. CALLBACK(record_bridge_stats, BRIDGE, 0),
  1252. /* Client only. */
  1253. /* XXXX this could be restricted to CLIENT+NET_PARTICIPANT */
  1254. CALLBACK(rend_cache_failure_clean, NET_PARTICIPANT, FL(RUN_ON_DISABLE)),
  1255. /* Directory server only. */
  1256. CALLBACK(clean_consdiffmgr, DIRSERVER, 0),
  1257. /* Controller with per-second events only. */
  1258. CALLBACK(control_per_second_events, CONTROLEV, 0),
  1259. END_OF_PERIODIC_EVENTS
  1260. };
  1261. #undef CALLBACK
  1262. #undef FL
  1263. /* These are pointers to members of periodic_events[] that are used to
  1264. * implement particular callbacks. We keep them separate here so that we
  1265. * can access them by name. We also keep them inside periodic_events[]
  1266. * so that we can implement "reset all timers" in a reasonable way. */
  1267. static periodic_event_item_t *fetch_networkstatus_event=NULL;
  1268. static periodic_event_item_t *launch_descriptor_fetches_event=NULL;
  1269. static periodic_event_item_t *check_dns_honesty_event=NULL;
  1270. static periodic_event_item_t *save_state_event=NULL;
  1271. static periodic_event_item_t *prune_old_routers_event=NULL;
  1272. /** Reset all the periodic events so we'll do all our actions again as if we
  1273. * just started up.
  1274. * Useful if our clock just moved back a long time from the future,
  1275. * so we don't wait until that future arrives again before acting.
  1276. */
  1277. void
  1278. reset_all_main_loop_timers(void)
  1279. {
  1280. periodic_events_reset_all();
  1281. }
  1282. /** Return a bitmask of the roles this tor instance is configured for using
  1283. * the given options. */
  1284. STATIC int
  1285. get_my_roles(const or_options_t *options)
  1286. {
  1287. tor_assert(options);
  1288. int roles = PERIODIC_EVENT_ROLE_ALL;
  1289. int is_bridge = options->BridgeRelay;
  1290. int is_relay = server_mode(options);
  1291. int is_dirauth = authdir_mode_v3(options);
  1292. int is_bridgeauth = authdir_mode_bridge(options);
  1293. int is_hidden_service = !!hs_service_get_num_services() ||
  1294. !!rend_num_services();
  1295. int is_dirserver = dir_server_mode(options);
  1296. int sending_control_events = control_any_per_second_event_enabled();
  1297. /* We also consider tor to have the role of a client if the ControlPort is
  1298. * set because a lot of things can be done over the control port which
  1299. * requires tor to have basic functionnalities. */
  1300. int is_client = options_any_client_port_set(options) ||
  1301. options->ControlPort_set ||
  1302. options->OwningControllerFD != UINT64_MAX;
  1303. int is_net_participant = is_participating_on_network() ||
  1304. is_relay || is_hidden_service;
  1305. if (is_bridge) roles |= PERIODIC_EVENT_ROLE_BRIDGE;
  1306. if (is_client) roles |= PERIODIC_EVENT_ROLE_CLIENT;
  1307. if (is_relay) roles |= PERIODIC_EVENT_ROLE_RELAY;
  1308. if (is_dirauth) roles |= PERIODIC_EVENT_ROLE_DIRAUTH;
  1309. if (is_bridgeauth) roles |= PERIODIC_EVENT_ROLE_BRIDGEAUTH;
  1310. if (is_hidden_service) roles |= PERIODIC_EVENT_ROLE_HS_SERVICE;
  1311. if (is_dirserver) roles |= PERIODIC_EVENT_ROLE_DIRSERVER;
  1312. if (is_net_participant) roles |= PERIODIC_EVENT_ROLE_NET_PARTICIPANT;
  1313. if (sending_control_events) roles |= PERIODIC_EVENT_ROLE_CONTROLEV;
  1314. return roles;
  1315. }
  1316. /** Event to run initialize_periodic_events_cb */
  1317. static struct event *initialize_periodic_events_event = NULL;
  1318. /** Helper, run one second after setup:
  1319. * Initializes all members of periodic_events and starts them running.
  1320. *
  1321. * (We do this one second after setup for backward-compatibility reasons;
  1322. * it might not actually be necessary.) */
  1323. static void
  1324. initialize_periodic_events_cb(evutil_socket_t fd, short events, void *data)
  1325. {
  1326. (void) fd;
  1327. (void) events;
  1328. (void) data;
  1329. tor_event_free(initialize_periodic_events_event);
  1330. rescan_periodic_events(get_options());
  1331. }
  1332. /** Set up all the members of mainloop_periodic_events[], and configure them
  1333. * all to be launched from a callback. */
  1334. void
  1335. initialize_periodic_events(void)
  1336. {
  1337. if (periodic_events_initialized)
  1338. return;
  1339. periodic_events_initialized = 1;
  1340. for (int i = 0; mainloop_periodic_events[i].name; ++i) {
  1341. periodic_events_register(&mainloop_periodic_events[i]);
  1342. }
  1343. /* Set up all periodic events. We'll launch them by roles. */
  1344. #define NAMED_CALLBACK(name) \
  1345. STMT_BEGIN name ## _event = periodic_events_find( #name ); STMT_END
  1346. NAMED_CALLBACK(prune_old_routers);
  1347. NAMED_CALLBACK(fetch_networkstatus);
  1348. NAMED_CALLBACK(launch_descriptor_fetches);
  1349. NAMED_CALLBACK(check_dns_honesty);
  1350. NAMED_CALLBACK(save_state);
  1351. }
  1352. STATIC void
  1353. teardown_periodic_events(void)
  1354. {
  1355. periodic_events_disconnect_all();
  1356. fetch_networkstatus_event = NULL;
  1357. launch_descriptor_fetches_event = NULL;
  1358. check_dns_honesty_event = NULL;
  1359. save_state_event = NULL;
  1360. prune_old_routers_event = NULL;
  1361. periodic_events_initialized = 0;
  1362. }
  1363. static mainloop_event_t *rescan_periodic_events_ev = NULL;
  1364. /** Callback: rescan the periodic event list. */
  1365. static void
  1366. rescan_periodic_events_cb(mainloop_event_t *event, void *arg)
  1367. {
  1368. (void)event;
  1369. (void)arg;
  1370. rescan_periodic_events(get_options());
  1371. }
  1372. /**
  1373. * Schedule an event that will rescan which periodic events should run.
  1374. **/
  1375. MOCK_IMPL(void,
  1376. schedule_rescan_periodic_events,(void))
  1377. {
  1378. if (!rescan_periodic_events_ev) {
  1379. rescan_periodic_events_ev =
  1380. mainloop_event_new(rescan_periodic_events_cb, NULL);
  1381. }
  1382. mainloop_event_activate(rescan_periodic_events_ev);
  1383. }
  1384. /** Do a pass at all our periodic events, disable those we don't need anymore
  1385. * and enable those we need now using the given options. */
  1386. void
  1387. rescan_periodic_events(const or_options_t *options)
  1388. {
  1389. tor_assert(options);
  1390. periodic_events_rescan_by_roles(get_my_roles(options), net_is_disabled());
  1391. }
  1392. /* We just got new options globally set, see if we need to enabled or disable
  1393. * periodic events. */
  1394. void
  1395. periodic_events_on_new_options(const or_options_t *options)
  1396. {
  1397. rescan_periodic_events(options);
  1398. }
  1399. /**
  1400. * Update our schedule so that we'll check whether we need to fetch directory
  1401. * info immediately.
  1402. */
  1403. void
  1404. reschedule_directory_downloads(void)
  1405. {
  1406. tor_assert(fetch_networkstatus_event);
  1407. tor_assert(launch_descriptor_fetches_event);
  1408. periodic_event_reschedule(fetch_networkstatus_event);
  1409. periodic_event_reschedule(launch_descriptor_fetches_event);
  1410. }
  1411. /** Mainloop callback: clean up circuits, channels, and connections
  1412. * that are pending close. */
  1413. static void
  1414. postloop_cleanup_cb(mainloop_event_t *ev, void *arg)
  1415. {
  1416. (void)ev;
  1417. (void)arg;
  1418. circuit_close_all_marked();
  1419. close_closeable_connections();
  1420. channel_run_cleanup();
  1421. channel_listener_run_cleanup();
  1422. }
  1423. /** Event to run postloop_cleanup_cb */
  1424. static mainloop_event_t *postloop_cleanup_ev=NULL;
  1425. /** Schedule a post-loop event to clean up marked channels, connections, and
  1426. * circuits. */
  1427. void
  1428. mainloop_schedule_postloop_cleanup(void)
  1429. {
  1430. if (PREDICT_UNLIKELY(postloop_cleanup_ev == NULL)) {
  1431. // (It's possible that we can get here if we decide to close a connection
  1432. // in the earliest stages of our configuration, before we create events.)
  1433. return;
  1434. }
  1435. mainloop_event_activate(postloop_cleanup_ev);
  1436. }
  1437. /** Event to run 'scheduled_shutdown_cb' */
  1438. static mainloop_event_t *scheduled_shutdown_ev=NULL;
  1439. /** Callback: run a scheduled shutdown */
  1440. static void
  1441. scheduled_shutdown_cb(mainloop_event_t *ev, void *arg)
  1442. {
  1443. (void)ev;
  1444. (void)arg;
  1445. log_notice(LD_GENERAL, "Clean shutdown finished. Exiting.");
  1446. tor_shutdown_event_loop_and_exit(0);
  1447. }
  1448. /** Schedule the mainloop to exit after <b>delay_sec</b> seconds. */
  1449. void
  1450. mainloop_schedule_shutdown(int delay_sec)
  1451. {
  1452. const struct timeval delay_tv = { delay_sec, 0 };
  1453. if (! scheduled_shutdown_ev) {
  1454. scheduled_shutdown_ev = mainloop_event_new(scheduled_shutdown_cb, NULL);
  1455. }
  1456. mainloop_event_schedule(scheduled_shutdown_ev, &delay_tv);
  1457. }
  1458. /** Perform regular maintenance tasks. This function gets run once per
  1459. * second.
  1460. */
  1461. static int
  1462. second_elapsed_callback(time_t now, const or_options_t *options)
  1463. {
  1464. /* 0. See if our bandwidth limits are exhausted and we should hibernate
  1465. *
  1466. * Note: we have redundant mechanisms to handle the case where it's
  1467. * time to wake up from hibernation; or where we have a scheduled
  1468. * shutdown and it's time to run it, but this will also handle those.
  1469. */
  1470. consider_hibernation(now);
  1471. /* Maybe enough time elapsed for us to reconsider a circuit. */
  1472. circuit_upgrade_circuits_from_guard_wait();
  1473. if (options->UseBridges && !net_is_disabled()) {
  1474. /* Note: this check uses net_is_disabled(), not should_delay_dir_fetches()
  1475. * -- the latter is only for fetching consensus-derived directory info. */
  1476. // TODO: client
  1477. // Also, schedule this rather than probing 1x / sec
  1478. fetch_bridge_descriptors(options, now);
  1479. }
  1480. if (accounting_is_enabled(options)) {
  1481. // TODO: refactor or rewrite?
  1482. accounting_run_housekeeping(now);
  1483. }
  1484. /* 3a. Every second, we examine pending circuits and prune the
  1485. * ones which have been pending for more than a few seconds.
  1486. * We do this before step 4, so it can try building more if
  1487. * it's not comfortable with the number of available circuits.
  1488. */
  1489. /* (If our circuit build timeout can ever become lower than a second (which
  1490. * it can't, currently), we should do this more often.) */
  1491. // TODO: All expire stuff can become NET_PARTICIPANT, RUN_ON_DISABLE
  1492. circuit_expire_building();
  1493. circuit_expire_waiting_for_better_guard();
  1494. /* 3b. Also look at pending streams and prune the ones that 'began'
  1495. * a long time ago but haven't gotten a 'connected' yet.
  1496. * Do this before step 4, so we can put them back into pending
  1497. * state to be picked up by the new circuit.
  1498. */
  1499. connection_ap_expire_beginning();
  1500. /* 3c. And expire connections that we've held open for too long.
  1501. */
  1502. connection_expire_held_open();
  1503. /* 4. Every second, we try a new circuit if there are no valid
  1504. * circuits. Every NewCircuitPeriod seconds, we expire circuits
  1505. * that became dirty more than MaxCircuitDirtiness seconds ago,
  1506. * and we make a new circ if there are no clean circuits.
  1507. */
  1508. const int have_dir_info = router_have_minimum_dir_info();
  1509. if (have_dir_info && !net_is_disabled()) {
  1510. circuit_build_needed_circs(now);
  1511. } else {
  1512. circuit_expire_old_circs_as_needed(now);
  1513. }
  1514. /* 5. We do housekeeping for each connection... */
  1515. channel_update_bad_for_new_circs(NULL, 0);
  1516. int i;
  1517. for (i=0;i<smartlist_len(connection_array);i++) {
  1518. run_connection_housekeeping(i, now);
  1519. }
  1520. /* Run again in a second. */
  1521. return 1;
  1522. }
  1523. /**
  1524. * Periodic callback: Every {LAZY,GREEDY}_DESCRIPTOR_RETRY_INTERVAL,
  1525. * see about fetching descriptors, microdescriptors, and extrainfo
  1526. * documents.
  1527. */
  1528. static int
  1529. launch_descriptor_fetches_callback(time_t now, const or_options_t *options)
  1530. {
  1531. if (should_delay_dir_fetches(options, NULL))
  1532. return PERIODIC_EVENT_NO_UPDATE;
  1533. update_all_descriptor_downloads(now);
  1534. update_extrainfo_downloads(now);
  1535. if (router_have_minimum_dir_info())
  1536. return LAZY_DESCRIPTOR_RETRY_INTERVAL;
  1537. else
  1538. return GREEDY_DESCRIPTOR_RETRY_INTERVAL;
  1539. }
  1540. /**
  1541. * Periodic event: Rotate our X.509 certificates and TLS keys once every
  1542. * MAX_SSL_KEY_LIFETIME_INTERNAL.
  1543. */
  1544. static int
  1545. rotate_x509_certificate_callback(time_t now, const or_options_t *options)
  1546. {
  1547. static int first = 1;
  1548. (void)now;
  1549. (void)options;
  1550. if (first) {
  1551. first = 0;
  1552. return MAX_SSL_KEY_LIFETIME_INTERNAL;
  1553. }
  1554. /* 1b. Every MAX_SSL_KEY_LIFETIME_INTERNAL seconds, we change our
  1555. * TLS context. */
  1556. log_info(LD_GENERAL,"Rotating tls context.");
  1557. if (router_initialize_tls_context() < 0) {
  1558. log_err(LD_BUG, "Error reinitializing TLS context");
  1559. tor_assert_unreached();
  1560. }
  1561. if (generate_ed_link_cert(options, now, 1)) {
  1562. log_err(LD_OR, "Unable to update Ed25519->TLS link certificate for "
  1563. "new TLS context.");
  1564. tor_assert_unreached();
  1565. }
  1566. /* We also make sure to rotate the TLS connections themselves if they've
  1567. * been up for too long -- but that's done via is_bad_for_new_circs in
  1568. * run_connection_housekeeping() above. */
  1569. return MAX_SSL_KEY_LIFETIME_INTERNAL;
  1570. }
  1571. /**
  1572. * Periodic callback: once an hour, grab some more entropy from the
  1573. * kernel and feed it to our CSPRNG.
  1574. **/
  1575. static int
  1576. add_entropy_callback(time_t now, const or_options_t *options)
  1577. {
  1578. (void)now;
  1579. (void)options;
  1580. /* We already seeded once, so don't die on failure. */
  1581. if (crypto_seed_rng() < 0) {
  1582. log_warn(LD_GENERAL, "Tried to re-seed RNG, but failed. We already "
  1583. "seeded once, though, so we won't exit here.");
  1584. }
  1585. /** How often do we add more entropy to OpenSSL's RNG pool? */
  1586. #define ENTROPY_INTERVAL (60*60)
  1587. return ENTROPY_INTERVAL;
  1588. }
  1589. /** Periodic callback: if there has been no network usage in a while,
  1590. * enter a dormant state. */
  1591. STATIC int
  1592. check_network_participation_callback(time_t now, const or_options_t *options)
  1593. {
  1594. /* If we're a server, we can't become dormant. */
  1595. if (server_mode(options)) {
  1596. goto found_activity;
  1597. }
  1598. /* If we're running an onion service, we can't become dormant. */
  1599. /* XXXX this would be nice to change, so that we can be dormant with a
  1600. * service. */
  1601. if (hs_service_get_num_services() || rend_num_services()) {
  1602. goto found_activity;
  1603. }
  1604. /* If we have any currently open entry streams other than "linked"
  1605. * connections used for directory requests, those count as user activity.
  1606. */
  1607. if (options->DormantTimeoutDisabledByIdleStreams) {
  1608. if (connection_get_by_type_nonlinked(CONN_TYPE_AP) != NULL) {
  1609. goto found_activity;
  1610. }
  1611. }
  1612. /* XXXX Make this configurable? */
  1613. /** How often do we check whether we have had network activity? */
  1614. #define CHECK_PARTICIPATION_INTERVAL (5*60)
  1615. /* Become dormant if there has been no user activity in a long time.
  1616. * (The funny checks below are in order to prevent overflow.) */
  1617. time_t time_since_last_activity = 0;
  1618. if (get_last_user_activity_time() < now)
  1619. time_since_last_activity = now - get_last_user_activity_time();
  1620. if (time_since_last_activity >= options->DormantClientTimeout) {
  1621. log_notice(LD_GENERAL, "No user activity in a long time: becoming"
  1622. " dormant.");
  1623. set_network_participation(false);
  1624. rescan_periodic_events(options);
  1625. }
  1626. return CHECK_PARTICIPATION_INTERVAL;
  1627. found_activity:
  1628. note_user_activity(now);
  1629. return CHECK_PARTICIPATION_INTERVAL;
  1630. }
  1631. /**
  1632. * Periodic callback: If our consensus is too old, recalculate whether
  1633. * we can actually use it.
  1634. */
  1635. static int
  1636. check_expired_networkstatus_callback(time_t now, const or_options_t *options)
  1637. {
  1638. (void)options;
  1639. /* Check whether our networkstatus has expired. */
  1640. networkstatus_t *ns = networkstatus_get_latest_consensus();
  1641. /* Use reasonably live consensuses until they are no longer reasonably live.
  1642. */
  1643. if (ns && !networkstatus_consensus_reasonably_live(ns, now) &&
  1644. router_have_minimum_dir_info()) {
  1645. router_dir_info_changed();
  1646. }
  1647. #define CHECK_EXPIRED_NS_INTERVAL (2*60)
  1648. return CHECK_EXPIRED_NS_INTERVAL;
  1649. }
  1650. /**
  1651. * Scheduled callback: Save the state file to disk if appropriate.
  1652. */
  1653. static int
  1654. save_state_callback(time_t now, const or_options_t *options)
  1655. {
  1656. (void) options;
  1657. (void) or_state_save(now); // only saves if appropriate
  1658. const time_t next_write = get_or_state()->next_write;
  1659. if (next_write == TIME_MAX) {
  1660. return 86400;
  1661. }
  1662. return safe_timer_diff(now, next_write);
  1663. }
  1664. /** Reschedule the event for saving the state file.
  1665. *
  1666. * Run this when the state becomes dirty. */
  1667. void
  1668. reschedule_or_state_save(void)
  1669. {
  1670. if (save_state_event == NULL) {
  1671. /* This can happen early on during startup. */
  1672. return;
  1673. }
  1674. periodic_event_reschedule(save_state_event);
  1675. }
  1676. /**
  1677. * Periodic callback: Write statistics to disk if appropriate.
  1678. */
  1679. static int
  1680. write_stats_file_callback(time_t now, const or_options_t *options)
  1681. {
  1682. /* 1g. Check whether we should write statistics to disk.
  1683. */
  1684. #define CHECK_WRITE_STATS_INTERVAL (60*60)
  1685. time_t next_time_to_write_stats_files = now + CHECK_WRITE_STATS_INTERVAL;
  1686. if (options->CellStatistics) {
  1687. time_t next_write =
  1688. rep_hist_buffer_stats_write(now);
  1689. if (next_write && next_write < next_time_to_write_stats_files)
  1690. next_time_to_write_stats_files = next_write;
  1691. }
  1692. if (options->DirReqStatistics) {
  1693. time_t next_write = geoip_dirreq_stats_write(now);
  1694. if (next_write && next_write < next_time_to_write_stats_files)
  1695. next_time_to_write_stats_files = next_write;
  1696. }
  1697. if (options->EntryStatistics) {
  1698. time_t next_write = geoip_entry_stats_write(now);
  1699. if (next_write && next_write < next_time_to_write_stats_files)
  1700. next_time_to_write_stats_files = next_write;
  1701. }
  1702. if (options->HiddenServiceStatistics) {
  1703. time_t next_write = rep_hist_hs_stats_write(now);
  1704. if (next_write && next_write < next_time_to_write_stats_files)
  1705. next_time_to_write_stats_files = next_write;
  1706. }
  1707. if (options->ExitPortStatistics) {
  1708. time_t next_write = rep_hist_exit_stats_write(now);
  1709. if (next_write && next_write < next_time_to_write_stats_files)
  1710. next_time_to_write_stats_files = next_write;
  1711. }
  1712. if (options->ConnDirectionStatistics) {
  1713. time_t next_write = rep_hist_conn_stats_write(now);
  1714. if (next_write && next_write < next_time_to_write_stats_files)
  1715. next_time_to_write_stats_files = next_write;
  1716. }
  1717. if (options->BridgeAuthoritativeDir) {
  1718. time_t next_write = rep_hist_desc_stats_write(now);
  1719. if (next_write && next_write < next_time_to_write_stats_files)
  1720. next_time_to_write_stats_files = next_write;
  1721. }
  1722. return safe_timer_diff(now, next_time_to_write_stats_files);
  1723. }
  1724. static int
  1725. reset_padding_counts_callback(time_t now, const or_options_t *options)
  1726. {
  1727. if (options->PaddingStatistics) {
  1728. rep_hist_prep_published_padding_counts(now);
  1729. }
  1730. rep_hist_reset_padding_counts();
  1731. return REPHIST_CELL_PADDING_COUNTS_INTERVAL;
  1732. }
  1733. static int should_init_bridge_stats = 1;
  1734. /**
  1735. * Periodic callback: Write bridge statistics to disk if appropriate.
  1736. */
  1737. static int
  1738. record_bridge_stats_callback(time_t now, const or_options_t *options)
  1739. {
  1740. /* 1h. Check whether we should write bridge statistics to disk.
  1741. */
  1742. if (should_record_bridge_info(options)) {
  1743. if (should_init_bridge_stats) {
  1744. /* (Re-)initialize bridge statistics. */
  1745. geoip_bridge_stats_init(now);
  1746. should_init_bridge_stats = 0;
  1747. return WRITE_STATS_INTERVAL;
  1748. } else {
  1749. /* Possibly write bridge statistics to disk and ask when to write
  1750. * them next time. */
  1751. time_t next = geoip_bridge_stats_write(now);
  1752. return safe_timer_diff(now, next);
  1753. }
  1754. } else if (!should_init_bridge_stats) {
  1755. /* Bridge mode was turned off. Ensure that stats are re-initialized
  1756. * next time bridge mode is turned on. */
  1757. should_init_bridge_stats = 1;
  1758. }
  1759. return PERIODIC_EVENT_NO_UPDATE;
  1760. }
  1761. /**
  1762. * Periodic callback: Clean in-memory caches every once in a while
  1763. */
  1764. static int
  1765. clean_caches_callback(time_t now, const or_options_t *options)
  1766. {
  1767. /* Remove old information from rephist and the rend cache. */
  1768. rep_history_clean(now - options->RephistTrackTime);
  1769. rend_cache_clean(now, REND_CACHE_TYPE_SERVICE);
  1770. hs_cache_clean_as_client(now);
  1771. hs_cache_clean_as_dir(now);
  1772. microdesc_cache_rebuild(NULL, 0);
  1773. #define CLEAN_CACHES_INTERVAL (30*60)
  1774. return CLEAN_CACHES_INTERVAL;
  1775. }
  1776. /**
  1777. * Periodic callback: Clean the cache of failed hidden service lookups
  1778. * frequently.
  1779. */
  1780. static int
  1781. rend_cache_failure_clean_callback(time_t now, const or_options_t *options)
  1782. {
  1783. (void)options;
  1784. /* We don't keep entries that are more than five minutes old so we try to
  1785. * clean it as soon as we can since we want to make sure the client waits
  1786. * as little as possible for reachability reasons. */
  1787. rend_cache_failure_clean(now);
  1788. hs_cache_client_intro_state_clean(now);
  1789. return 30;
  1790. }
  1791. /**
  1792. * Periodic callback: prune routerlist of old information about Tor network.
  1793. */
  1794. static int
  1795. prune_old_routers_callback(time_t now, const or_options_t *options)
  1796. {
  1797. #define ROUTERLIST_PRUNING_INTERVAL (60*60) // 1 hour.
  1798. (void)now;
  1799. (void)options;
  1800. if (!net_is_disabled()) {
  1801. /* If any networkstatus documents are no longer recent, we need to
  1802. * update all the descriptors' running status. */
  1803. /* Remove dead routers. */
  1804. log_debug(LD_GENERAL, "Pruning routerlist...");
  1805. routerlist_remove_old_routers();
  1806. }
  1807. return ROUTERLIST_PRUNING_INTERVAL;
  1808. }
  1809. /**
  1810. * Periodic event: once a minute, (or every second if TestingTorNetwork, or
  1811. * during client bootstrap), check whether we want to download any
  1812. * networkstatus documents. */
  1813. static int
  1814. fetch_networkstatus_callback(time_t now, const or_options_t *options)
  1815. {
  1816. /* How often do we check whether we should download network status
  1817. * documents? */
  1818. const int we_are_bootstrapping = networkstatus_consensus_is_bootstrapping(
  1819. now);
  1820. const int prefer_mirrors = !directory_fetches_from_authorities(
  1821. get_options());
  1822. int networkstatus_dl_check_interval = 60;
  1823. /* check more often when testing, or when bootstrapping from mirrors
  1824. * (connection limits prevent too many connections being made) */
  1825. if (options->TestingTorNetwork
  1826. || (we_are_bootstrapping && prefer_mirrors)) {
  1827. networkstatus_dl_check_interval = 1;
  1828. }
  1829. if (should_delay_dir_fetches(options, NULL))
  1830. return PERIODIC_EVENT_NO_UPDATE;
  1831. update_networkstatus_downloads(now);
  1832. return networkstatus_dl_check_interval;
  1833. }
  1834. /**
  1835. * Periodic callback: Every 60 seconds, we relaunch listeners if any died. */
  1836. static int
  1837. retry_listeners_callback(time_t now, const or_options_t *options)
  1838. {
  1839. (void)now;
  1840. (void)options;
  1841. if (!net_is_disabled()) {
  1842. retry_all_listeners(NULL, 0);
  1843. return 60;
  1844. }
  1845. return PERIODIC_EVENT_NO_UPDATE;
  1846. }
  1847. static int heartbeat_callback_first_time = 1;
  1848. /**
  1849. * Periodic callback: write the heartbeat message in the logs.
  1850. *
  1851. * If writing the heartbeat message to the logs fails for some reason, retry
  1852. * again after <b>MIN_HEARTBEAT_PERIOD</b> seconds.
  1853. */
  1854. static int
  1855. heartbeat_callback(time_t now, const or_options_t *options)
  1856. {
  1857. /* Check if heartbeat is disabled */
  1858. if (!options->HeartbeatPeriod) {
  1859. return PERIODIC_EVENT_NO_UPDATE;
  1860. }
  1861. /* Skip the first one. */
  1862. if (heartbeat_callback_first_time) {
  1863. heartbeat_callback_first_time = 0;
  1864. return options->HeartbeatPeriod;
  1865. }
  1866. /* Write the heartbeat message */
  1867. if (log_heartbeat(now) == 0) {
  1868. return options->HeartbeatPeriod;
  1869. } else {
  1870. /* If we couldn't write the heartbeat log message, try again in the minimum
  1871. * interval of time. */
  1872. return MIN_HEARTBEAT_PERIOD;
  1873. }
  1874. }
  1875. #define CDM_CLEAN_CALLBACK_INTERVAL 600
  1876. static int
  1877. clean_consdiffmgr_callback(time_t now, const or_options_t *options)
  1878. {
  1879. (void)now;
  1880. if (dir_server_mode(options)) {
  1881. consdiffmgr_cleanup();
  1882. }
  1883. return CDM_CLEAN_CALLBACK_INTERVAL;
  1884. }
  1885. /*
  1886. * Periodic callback: Run scheduled events for HS service. This is called
  1887. * every second.
  1888. */
  1889. static int
  1890. hs_service_callback(time_t now, const or_options_t *options)
  1891. {
  1892. (void) options;
  1893. /* We need to at least be able to build circuits and that we actually have
  1894. * a working network. */
  1895. if (!have_completed_a_circuit() || net_is_disabled() ||
  1896. networkstatus_get_live_consensus(now) == NULL) {
  1897. goto end;
  1898. }
  1899. hs_service_run_scheduled_events(now);
  1900. end:
  1901. /* Every 1 second. */
  1902. return 1;
  1903. }
  1904. /*
  1905. * Periodic callback: Send once-per-second events to the controller(s).
  1906. * This is called every second.
  1907. */
  1908. static int
  1909. control_per_second_events_callback(time_t now, const or_options_t *options)
  1910. {
  1911. (void) options;
  1912. (void) now;
  1913. control_per_second_events();
  1914. return 1;
  1915. }
  1916. /** Last time that update_current_time was called. */
  1917. static time_t current_second = 0;
  1918. /** Last time that update_current_time updated current_second. */
  1919. static monotime_coarse_t current_second_last_changed;
  1920. /**
  1921. * Set the current time to "now", which should be the value returned by
  1922. * time(). Check for clock jumps and track the total number of seconds we
  1923. * have been running.
  1924. */
  1925. void
  1926. update_current_time(time_t now)
  1927. {
  1928. if (PREDICT_LIKELY(now == current_second)) {
  1929. /* We call this function a lot. Most frequently, the current second
  1930. * will not have changed, so we just return. */
  1931. return;
  1932. }
  1933. const time_t seconds_elapsed = current_second ? (now - current_second) : 0;
  1934. /* Check the wall clock against the monotonic clock, so we can
  1935. * better tell idleness from clock jumps and/or other shenanigans. */
  1936. monotime_coarse_t last_updated;
  1937. memcpy(&last_updated, &current_second_last_changed, sizeof(last_updated));
  1938. monotime_coarse_get(&current_second_last_changed);
  1939. /** How much clock jumping means that we should adjust our idea of when
  1940. * to go dormant? */
  1941. #define NUM_JUMPED_SECONDS_BEFORE_NETSTATUS_UPDATE 20
  1942. /* Don't go dormant early or late just because we jumped in time. */
  1943. if (ABS(seconds_elapsed) >= NUM_JUMPED_SECONDS_BEFORE_NETSTATUS_UPDATE) {
  1944. if (is_participating_on_network()) {
  1945. netstatus_note_clock_jumped(seconds_elapsed);
  1946. }
  1947. }
  1948. /** How much clock jumping do we tolerate? */
  1949. #define NUM_JUMPED_SECONDS_BEFORE_WARN 100
  1950. /** How much idleness do we tolerate? */
  1951. #define NUM_IDLE_SECONDS_BEFORE_WARN 3600
  1952. if (seconds_elapsed < -NUM_JUMPED_SECONDS_BEFORE_WARN) {
  1953. // moving back in time is always a bad sign.
  1954. circuit_note_clock_jumped(seconds_elapsed, false);
  1955. } else if (seconds_elapsed >= NUM_JUMPED_SECONDS_BEFORE_WARN) {
  1956. /* Compare the monotonic clock to the result of time(). */
  1957. const int32_t monotime_msec_passed =
  1958. monotime_coarse_diff_msec32(&last_updated,
  1959. &current_second_last_changed);
  1960. const int monotime_sec_passed = monotime_msec_passed / 1000;
  1961. const int discrepancy = monotime_sec_passed - (int)seconds_elapsed;
  1962. /* If the monotonic clock deviates from time(NULL), we have a couple of
  1963. * possibilities. On some systems, this means we have been suspended or
  1964. * sleeping. Everywhere, it can mean that the wall-clock time has
  1965. * been changed -- for example, with settimeofday().
  1966. *
  1967. * On the other hand, if the monotonic time matches with the wall-clock
  1968. * time, we've probably just been idle for a while, with no events firing.
  1969. * we tolerate much more of that.
  1970. */
  1971. const bool clock_jumped = abs(discrepancy) > 2;
  1972. if (clock_jumped || seconds_elapsed >= NUM_IDLE_SECONDS_BEFORE_WARN) {
  1973. circuit_note_clock_jumped(seconds_elapsed, ! clock_jumped);
  1974. }
  1975. } else if (seconds_elapsed > 0) {
  1976. stats_n_seconds_working += seconds_elapsed;
  1977. }
  1978. update_approx_time(now);
  1979. current_second = now;
  1980. }
  1981. #ifdef HAVE_SYSTEMD_209
  1982. static periodic_timer_t *systemd_watchdog_timer = NULL;
  1983. /** Libevent callback: invoked to reset systemd watchdog. */
  1984. static void
  1985. systemd_watchdog_callback(periodic_timer_t *timer, void *arg)
  1986. {
  1987. (void)timer;
  1988. (void)arg;
  1989. sd_notify(0, "WATCHDOG=1");
  1990. }
  1991. #endif /* defined(HAVE_SYSTEMD_209) */
  1992. #define UPTIME_CUTOFF_FOR_NEW_BANDWIDTH_TEST (6*60*60)
  1993. /** Called when our IP address seems to have changed. <b>at_interface</b>
  1994. * should be true if we detected a change in our interface, and false if we
  1995. * detected a change in our published address. */
  1996. void
  1997. ip_address_changed(int at_interface)
  1998. {
  1999. const or_options_t *options = get_options();
  2000. int server = server_mode(options);
  2001. int exit_reject_interfaces = (server && options->ExitRelay
  2002. && options->ExitPolicyRejectLocalInterfaces);
  2003. if (at_interface) {
  2004. if (! server) {
  2005. /* Okay, change our keys. */
  2006. if (init_keys_client() < 0)
  2007. log_warn(LD_GENERAL, "Unable to rotate keys after IP change!");
  2008. }
  2009. } else {
  2010. if (server) {
  2011. if (get_uptime() > UPTIME_CUTOFF_FOR_NEW_BANDWIDTH_TEST)
  2012. reset_bandwidth_test();
  2013. reset_uptime();
  2014. router_reset_reachability();
  2015. }
  2016. }
  2017. /* Exit relays incorporate interface addresses in their exit policies when
  2018. * ExitPolicyRejectLocalInterfaces is set */
  2019. if (exit_reject_interfaces || (server && !at_interface)) {
  2020. mark_my_descriptor_dirty("IP address changed");
  2021. }
  2022. dns_servers_relaunch_checks();
  2023. }
  2024. /** Forget what we've learned about the correctness of our DNS servers, and
  2025. * start learning again. */
  2026. void
  2027. dns_servers_relaunch_checks(void)
  2028. {
  2029. if (server_mode(get_options())) {
  2030. dns_reset_correctness_checks();
  2031. if (check_dns_honesty_event) {
  2032. periodic_event_reschedule(check_dns_honesty_event);
  2033. }
  2034. }
  2035. }
  2036. /** Initialize some mainloop_event_t objects that we require. */
  2037. void
  2038. initialize_mainloop_events(void)
  2039. {
  2040. if (!schedule_active_linked_connections_event) {
  2041. schedule_active_linked_connections_event =
  2042. mainloop_event_postloop_new(schedule_active_linked_connections_cb, NULL);
  2043. }
  2044. if (!postloop_cleanup_ev) {
  2045. postloop_cleanup_ev =
  2046. mainloop_event_postloop_new(postloop_cleanup_cb, NULL);
  2047. }
  2048. }
  2049. /** Tor main loop. */
  2050. int
  2051. do_main_loop(void)
  2052. {
  2053. /* initialize the periodic events first, so that code that depends on the
  2054. * events being present does not assert.
  2055. */
  2056. tor_assert(periodic_events_initialized);
  2057. initialize_mainloop_events();
  2058. periodic_events_connect_all();
  2059. struct timeval one_second = { 1, 0 };
  2060. initialize_periodic_events_event = tor_evtimer_new(
  2061. tor_libevent_get_base(),
  2062. initialize_periodic_events_cb, NULL);
  2063. event_add(initialize_periodic_events_event, &one_second);
  2064. #ifdef HAVE_SYSTEMD_209
  2065. uint64_t watchdog_delay;
  2066. /* set up systemd watchdog notification. */
  2067. if (sd_watchdog_enabled(1, &watchdog_delay) > 0) {
  2068. if (! systemd_watchdog_timer) {
  2069. struct timeval watchdog;
  2070. /* The manager will "act on" us if we don't send them a notification
  2071. * every 'watchdog_delay' microseconds. So, send notifications twice
  2072. * that often. */
  2073. watchdog_delay /= 2;
  2074. watchdog.tv_sec = watchdog_delay / 1000000;
  2075. watchdog.tv_usec = watchdog_delay % 1000000;
  2076. systemd_watchdog_timer = periodic_timer_new(tor_libevent_get_base(),
  2077. &watchdog,
  2078. systemd_watchdog_callback,
  2079. NULL);
  2080. tor_assert(systemd_watchdog_timer);
  2081. }
  2082. }
  2083. #endif /* defined(HAVE_SYSTEMD_209) */
  2084. #ifdef ENABLE_RESTART_DEBUGGING
  2085. {
  2086. static int first_time = 1;
  2087. if (first_time && getenv("TOR_DEBUG_RESTART")) {
  2088. first_time = 0;
  2089. const char *sec_str = getenv("TOR_DEBUG_RESTART_AFTER_SECONDS");
  2090. long sec;
  2091. int sec_ok=0;
  2092. if (sec_str &&
  2093. (sec = tor_parse_long(sec_str, 10, 0, INT_MAX, &sec_ok, NULL)) &&
  2094. sec_ok) {
  2095. /* Okay, we parsed the seconds. */
  2096. } else {
  2097. sec = 5;
  2098. }
  2099. struct timeval restart_after = { (time_t) sec, 0 };
  2100. tor_shutdown_event_loop_for_restart_event =
  2101. tor_evtimer_new(tor_libevent_get_base(),
  2102. tor_shutdown_event_loop_for_restart_cb, NULL);
  2103. event_add(tor_shutdown_event_loop_for_restart_event, &restart_after);
  2104. }
  2105. }
  2106. #endif /* defined(ENABLE_RESTART_DEBUGGING) */
  2107. return run_main_loop_until_done();
  2108. }
  2109. #ifndef _WIN32
  2110. /** Rate-limiter for EINVAL-type libevent warnings. */
  2111. static ratelim_t libevent_error_ratelim = RATELIM_INIT(10);
  2112. #endif
  2113. /**
  2114. * Run the main loop a single time. Return 0 for "exit"; -1 for "exit with
  2115. * error", and 1 for "run this again."
  2116. */
  2117. static int
  2118. run_main_loop_once(void)
  2119. {
  2120. int loop_result;
  2121. if (nt_service_is_stopping())
  2122. return 0;
  2123. if (main_loop_should_exit)
  2124. return 0;
  2125. #ifndef _WIN32
  2126. /* Make it easier to tell whether libevent failure is our fault or not. */
  2127. errno = 0;
  2128. #endif
  2129. if (get_options()->MainloopStats) {
  2130. /* We always enforce that EVLOOP_ONCE is passed to event_base_loop() if we
  2131. * are collecting main loop statistics. */
  2132. called_loop_once = 1;
  2133. } else {
  2134. called_loop_once = 0;
  2135. }
  2136. /* Make sure we know (about) what time it is. */
  2137. update_approx_time(time(NULL));
  2138. /* Here it is: the main loop. Here we tell Libevent to poll until we have
  2139. * an event, or the second ends, or until we have some active linked
  2140. * connections to trigger events for. Libevent will wait till one
  2141. * of these happens, then run all the appropriate callbacks. */
  2142. loop_result = tor_libevent_run_event_loop(tor_libevent_get_base(),
  2143. called_loop_once);
  2144. if (get_options()->MainloopStats) {
  2145. /* Update our main loop counters. */
  2146. if (loop_result == 0) {
  2147. // The call was successful.
  2148. increment_main_loop_success_count();
  2149. } else if (loop_result == -1) {
  2150. // The call was erroneous.
  2151. increment_main_loop_error_count();
  2152. } else if (loop_result == 1) {
  2153. // The call didn't have any active or pending events
  2154. // to handle.
  2155. increment_main_loop_idle_count();
  2156. }
  2157. }
  2158. /* Oh, the loop failed. That might be an error that we need to
  2159. * catch, but more likely, it's just an interrupted poll() call or something,
  2160. * and we should try again. */
  2161. if (loop_result < 0) {
  2162. int e = tor_socket_errno(-1);
  2163. /* let the program survive things like ^z */
  2164. if (e != EINTR && !ERRNO_IS_EINPROGRESS(e)) {
  2165. log_err(LD_NET,"libevent call with %s failed: %s [%d]",
  2166. tor_libevent_get_method(), tor_socket_strerror(e), e);
  2167. return -1;
  2168. #ifndef _WIN32
  2169. } else if (e == EINVAL) {
  2170. log_fn_ratelim(&libevent_error_ratelim, LOG_WARN, LD_NET,
  2171. "EINVAL from libevent: should you upgrade libevent?");
  2172. if (libevent_error_ratelim.n_calls_since_last_time > 8) {
  2173. log_err(LD_NET, "Too many libevent errors, too fast: dying");
  2174. return -1;
  2175. }
  2176. #endif /* !defined(_WIN32) */
  2177. } else {
  2178. tor_assert_nonfatal_once(! ERRNO_IS_EINPROGRESS(e));
  2179. log_debug(LD_NET,"libevent call interrupted.");
  2180. /* You can't trust the results of this poll(). Go back to the
  2181. * top of the big for loop. */
  2182. return 1;
  2183. }
  2184. }
  2185. if (main_loop_should_exit)
  2186. return 0;
  2187. return 1;
  2188. }
  2189. /** Run the run_main_loop_once() function until it declares itself done,
  2190. * and return its final return value.
  2191. *
  2192. * Shadow won't invoke this function, so don't fill it up with things.
  2193. */
  2194. STATIC int
  2195. run_main_loop_until_done(void)
  2196. {
  2197. int loop_result = 1;
  2198. main_loop_should_exit = 0;
  2199. main_loop_exit_value = 0;
  2200. do {
  2201. loop_result = run_main_loop_once();
  2202. } while (loop_result == 1);
  2203. if (main_loop_should_exit)
  2204. return main_loop_exit_value;
  2205. else
  2206. return loop_result;
  2207. }
  2208. /** Returns Tor's uptime. */
  2209. MOCK_IMPL(long,
  2210. get_uptime,(void))
  2211. {
  2212. return stats_n_seconds_working;
  2213. }
  2214. /** Reset Tor's uptime. */
  2215. MOCK_IMPL(void,
  2216. reset_uptime,(void))
  2217. {
  2218. stats_n_seconds_working = 0;
  2219. }
  2220. void
  2221. tor_mainloop_free_all(void)
  2222. {
  2223. smartlist_free(connection_array);
  2224. smartlist_free(closeable_connection_lst);
  2225. smartlist_free(active_linked_connection_lst);
  2226. teardown_periodic_events();
  2227. tor_event_free(shutdown_did_not_work_event);
  2228. tor_event_free(initialize_periodic_events_event);
  2229. mainloop_event_free(directory_all_unreachable_cb_event);
  2230. mainloop_event_free(schedule_active_linked_connections_event);
  2231. mainloop_event_free(postloop_cleanup_ev);
  2232. mainloop_event_free(handle_deferred_signewnym_ev);
  2233. mainloop_event_free(scheduled_shutdown_ev);
  2234. mainloop_event_free(rescan_periodic_events_ev);
  2235. #ifdef HAVE_SYSTEMD_209
  2236. periodic_timer_free(systemd_watchdog_timer);
  2237. #endif
  2238. stats_n_bytes_read = stats_n_bytes_written = 0;
  2239. memset(&global_bucket, 0, sizeof(global_bucket));
  2240. memset(&global_relayed_bucket, 0, sizeof(global_relayed_bucket));
  2241. time_of_process_start = 0;
  2242. time_of_last_signewnym = 0;
  2243. signewnym_is_pending = 0;
  2244. newnym_epoch = 0;
  2245. called_loop_once = 0;
  2246. main_loop_should_exit = 0;
  2247. main_loop_exit_value = 0;
  2248. can_complete_circuits = 0;
  2249. quiet_level = 0;
  2250. should_init_bridge_stats = 1;
  2251. heartbeat_callback_first_time = 1;
  2252. current_second = 0;
  2253. memset(&current_second_last_changed, 0,
  2254. sizeof(current_second_last_changed));
  2255. }