mainloop.c 94 KB

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