mainloop.c 94 KB

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