connection_or.c 47 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380
  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-2010, The Tor Project, Inc. */
  5. /* See LICENSE for licensing information */
  6. /**
  7. * \file connection_or.c
  8. * \brief Functions to handle OR connections, TLS handshaking, and
  9. * cells on the network.
  10. **/
  11. #include "or.h"
  12. #include "buffers.h"
  13. #include "circuitbuild.h"
  14. #include "command.h"
  15. #include "config.h"
  16. #include "connection.h"
  17. #include "connection_or.h"
  18. #include "control.h"
  19. #include "geoip.h"
  20. #include "router.h"
  21. #include "routerlist.h"
  22. static int connection_tls_finish_handshake(or_connection_t *conn);
  23. static int connection_or_process_cells_from_inbuf(or_connection_t *conn);
  24. static int connection_or_send_versions(or_connection_t *conn);
  25. static int connection_init_or_handshake_state(or_connection_t *conn,
  26. int started_here);
  27. static int connection_or_check_valid_tls_handshake(or_connection_t *conn,
  28. int started_here,
  29. char *digest_rcvd_out);
  30. /**************************************************************/
  31. /** Map from identity digest of connected OR or desired OR to a connection_t
  32. * with that identity digest. If there is more than one such connection_t,
  33. * they form a linked list, with next_with_same_id as the next pointer. */
  34. static digestmap_t *orconn_identity_map = NULL;
  35. /** If conn is listed in orconn_identity_map, remove it, and clear
  36. * conn->identity_digest. Otherwise do nothing. */
  37. void
  38. connection_or_remove_from_identity_map(or_connection_t *conn)
  39. {
  40. or_connection_t *tmp;
  41. tor_assert(conn);
  42. if (!orconn_identity_map)
  43. return;
  44. tmp = digestmap_get(orconn_identity_map, conn->identity_digest);
  45. if (!tmp) {
  46. if (!tor_digest_is_zero(conn->identity_digest)) {
  47. log_warn(LD_BUG, "Didn't find connection '%s' on identity map when "
  48. "trying to remove it.",
  49. conn->nickname ? conn->nickname : "NULL");
  50. }
  51. return;
  52. }
  53. if (conn == tmp) {
  54. if (conn->next_with_same_id)
  55. digestmap_set(orconn_identity_map, conn->identity_digest,
  56. conn->next_with_same_id);
  57. else
  58. digestmap_remove(orconn_identity_map, conn->identity_digest);
  59. } else {
  60. while (tmp->next_with_same_id) {
  61. if (tmp->next_with_same_id == conn) {
  62. tmp->next_with_same_id = conn->next_with_same_id;
  63. break;
  64. }
  65. tmp = tmp->next_with_same_id;
  66. }
  67. }
  68. memset(conn->identity_digest, 0, DIGEST_LEN);
  69. conn->next_with_same_id = NULL;
  70. }
  71. /** Remove all entries from the identity-to-orconn map, and clear
  72. * all identities in OR conns.*/
  73. void
  74. connection_or_clear_identity_map(void)
  75. {
  76. smartlist_t *conns = get_connection_array();
  77. SMARTLIST_FOREACH(conns, connection_t *, conn,
  78. {
  79. if (conn->type == CONN_TYPE_OR) {
  80. or_connection_t *or_conn = TO_OR_CONN(conn);
  81. memset(or_conn->identity_digest, 0, DIGEST_LEN);
  82. or_conn->next_with_same_id = NULL;
  83. }
  84. });
  85. digestmap_free(orconn_identity_map, NULL);
  86. orconn_identity_map = NULL;
  87. }
  88. /** Change conn->identity_digest to digest, and add conn into
  89. * orconn_digest_map. */
  90. static void
  91. connection_or_set_identity_digest(or_connection_t *conn, const char *digest)
  92. {
  93. or_connection_t *tmp;
  94. tor_assert(conn);
  95. tor_assert(digest);
  96. if (!orconn_identity_map)
  97. orconn_identity_map = digestmap_new();
  98. if (!memcmp(conn->identity_digest, digest, DIGEST_LEN))
  99. return;
  100. /* If the identity was set previously, remove the old mapping. */
  101. if (! tor_digest_is_zero(conn->identity_digest))
  102. connection_or_remove_from_identity_map(conn);
  103. memcpy(conn->identity_digest, digest, DIGEST_LEN);
  104. /* If we're setting the ID to zero, don't add a mapping. */
  105. if (tor_digest_is_zero(digest))
  106. return;
  107. tmp = digestmap_set(orconn_identity_map, digest, conn);
  108. conn->next_with_same_id = tmp;
  109. #if 1
  110. /* Testing code to check for bugs in representation. */
  111. for (; tmp; tmp = tmp->next_with_same_id) {
  112. tor_assert(!memcmp(tmp->identity_digest, digest, DIGEST_LEN));
  113. tor_assert(tmp != conn);
  114. }
  115. #endif
  116. }
  117. /** Pack the cell_t host-order structure <b>src</b> into network-order
  118. * in the buffer <b>dest</b>. See tor-spec.txt for details about the
  119. * wire format.
  120. *
  121. * Note that this function doesn't touch <b>dst</b>-\>next: the caller
  122. * should set it or clear it as appropriate.
  123. */
  124. void
  125. cell_pack(packed_cell_t *dst, const cell_t *src)
  126. {
  127. char *dest = dst->body;
  128. *(uint16_t*)dest = htons(src->circ_id);
  129. *(uint8_t*)(dest+2) = src->command;
  130. memcpy(dest+3, src->payload, CELL_PAYLOAD_SIZE);
  131. }
  132. /** Unpack the network-order buffer <b>src</b> into a host-order
  133. * cell_t structure <b>dest</b>.
  134. */
  135. static void
  136. cell_unpack(cell_t *dest, const char *src)
  137. {
  138. dest->circ_id = ntohs(*(uint16_t*)(src));
  139. dest->command = *(uint8_t*)(src+2);
  140. memcpy(dest->payload, src+3, CELL_PAYLOAD_SIZE);
  141. }
  142. /** Write the header of <b>cell</b> into the first VAR_CELL_HEADER_SIZE
  143. * bytes of <b>hdr_out</b>. */
  144. void
  145. var_cell_pack_header(const var_cell_t *cell, char *hdr_out)
  146. {
  147. set_uint16(hdr_out, htons(cell->circ_id));
  148. set_uint8(hdr_out+2, cell->command);
  149. set_uint16(hdr_out+3, htons(cell->payload_len));
  150. }
  151. /** Allocate and return a new var_cell_t with <b>payload_len</b> bytes of
  152. * payload space. */
  153. var_cell_t *
  154. var_cell_new(uint16_t payload_len)
  155. {
  156. var_cell_t *cell = tor_malloc(sizeof(var_cell_t)+payload_len-1);
  157. cell->payload_len = payload_len;
  158. cell->command = 0;
  159. cell->circ_id = 0;
  160. return cell;
  161. }
  162. /** Release all space held by <b>cell</b>. */
  163. void
  164. var_cell_free(var_cell_t *cell)
  165. {
  166. tor_free(cell);
  167. }
  168. /** We've received an EOF from <b>conn</b>. Mark it for close and return. */
  169. int
  170. connection_or_reached_eof(or_connection_t *conn)
  171. {
  172. log_info(LD_OR,"OR connection reached EOF. Closing.");
  173. connection_mark_for_close(TO_CONN(conn));
  174. return 0;
  175. }
  176. /** Handle any new bytes that have come in on connection <b>conn</b>.
  177. * If conn is in 'open' state, hand it to
  178. * connection_or_process_cells_from_inbuf()
  179. * (else do nothing).
  180. */
  181. int
  182. connection_or_process_inbuf(or_connection_t *conn)
  183. {
  184. int ret;
  185. tor_assert(conn);
  186. switch (conn->_base.state) {
  187. case OR_CONN_STATE_PROXY_HANDSHAKING:
  188. ret = connection_read_proxy_handshake(TO_CONN(conn));
  189. /* start TLS after handshake completion, or deal with error */
  190. if (ret == 1) {
  191. tor_assert(TO_CONN(conn)->proxy_state == PROXY_CONNECTED);
  192. if (connection_tls_start_handshake(conn, 0) < 0)
  193. ret = -1;
  194. }
  195. if (ret < 0) {
  196. connection_mark_for_close(TO_CONN(conn));
  197. }
  198. return ret;
  199. case OR_CONN_STATE_OPEN:
  200. case OR_CONN_STATE_OR_HANDSHAKING:
  201. return connection_or_process_cells_from_inbuf(conn);
  202. default:
  203. return 0; /* don't do anything */
  204. }
  205. }
  206. /** When adding cells to an OR connection's outbuf, keep adding until the
  207. * outbuf is at least this long, or we run out of cells. */
  208. #define OR_CONN_HIGHWATER (32*1024)
  209. /** Add cells to an OR connection's outbuf whenever the outbuf's data length
  210. * drops below this size. */
  211. #define OR_CONN_LOWWATER (16*1024)
  212. /** Called whenever we have flushed some data on an or_conn: add more data
  213. * from active circuits. */
  214. int
  215. connection_or_flushed_some(or_connection_t *conn)
  216. {
  217. size_t datalen = buf_datalen(conn->_base.outbuf);
  218. /* If we're under the low water mark, add cells until we're just over the
  219. * high water mark. */
  220. if (datalen < OR_CONN_LOWWATER) {
  221. ssize_t n = (OR_CONN_HIGHWATER - datalen + CELL_NETWORK_SIZE-1)
  222. / CELL_NETWORK_SIZE;
  223. time_t now = approx_time();
  224. while (conn->active_circuits && n > 0) {
  225. int flushed;
  226. flushed = connection_or_flush_from_first_active_circuit(conn, 1, now);
  227. n -= flushed;
  228. }
  229. }
  230. return 0;
  231. }
  232. /** Connection <b>conn</b> has finished writing and has no bytes left on
  233. * its outbuf.
  234. *
  235. * Otherwise it's in state "open": stop writing and return.
  236. *
  237. * If <b>conn</b> is broken, mark it for close and return -1, else
  238. * return 0.
  239. */
  240. int
  241. connection_or_finished_flushing(or_connection_t *conn)
  242. {
  243. tor_assert(conn);
  244. assert_connection_ok(TO_CONN(conn),0);
  245. switch (conn->_base.state) {
  246. case OR_CONN_STATE_PROXY_HANDSHAKING:
  247. case OR_CONN_STATE_OPEN:
  248. case OR_CONN_STATE_OR_HANDSHAKING:
  249. connection_stop_writing(TO_CONN(conn));
  250. break;
  251. default:
  252. log_err(LD_BUG,"Called in unexpected state %d.", conn->_base.state);
  253. tor_fragile_assert();
  254. return -1;
  255. }
  256. return 0;
  257. }
  258. /** Connected handler for OR connections: begin the TLS handshake.
  259. */
  260. int
  261. connection_or_finished_connecting(or_connection_t *or_conn)
  262. {
  263. int proxy_type;
  264. connection_t *conn;
  265. tor_assert(or_conn);
  266. conn = TO_CONN(or_conn);
  267. tor_assert(conn->state == OR_CONN_STATE_CONNECTING);
  268. log_debug(LD_HANDSHAKE,"OR connect() to router at %s:%u finished.",
  269. conn->address,conn->port);
  270. control_event_bootstrap(BOOTSTRAP_STATUS_HANDSHAKE, 0);
  271. proxy_type = PROXY_NONE;
  272. if (get_options()->HttpsProxy)
  273. proxy_type = PROXY_CONNECT;
  274. else if (get_options()->Socks4Proxy)
  275. proxy_type = PROXY_SOCKS4;
  276. else if (get_options()->Socks5Proxy)
  277. proxy_type = PROXY_SOCKS5;
  278. if (proxy_type != PROXY_NONE) {
  279. /* start proxy handshake */
  280. if (connection_proxy_connect(conn, proxy_type) < 0) {
  281. connection_mark_for_close(conn);
  282. return -1;
  283. }
  284. connection_start_reading(conn);
  285. conn->state = OR_CONN_STATE_PROXY_HANDSHAKING;
  286. return 0;
  287. }
  288. if (connection_tls_start_handshake(or_conn, 0) < 0) {
  289. /* TLS handshaking error of some kind. */
  290. connection_mark_for_close(conn);
  291. return -1;
  292. }
  293. return 0;
  294. }
  295. /** Return 1 if identity digest <b>id_digest</b> is known to be a
  296. * currently or recently running relay. Otherwise return 0. */
  297. int
  298. connection_or_digest_is_known_relay(const char *id_digest)
  299. {
  300. if (router_get_consensus_status_by_id(id_digest))
  301. return 1; /* It's in the consensus: "yes" */
  302. if (router_get_by_digest(id_digest))
  303. return 1; /* Not in the consensus, but we have a descriptor for
  304. * it. Probably it was in a recent consensus. "Yes". */
  305. return 0;
  306. }
  307. /** If we don't necessarily know the router we're connecting to, but we
  308. * have an addr/port/id_digest, then fill in as much as we can. Start
  309. * by checking to see if this describes a router we know. */
  310. static void
  311. connection_or_init_conn_from_address(or_connection_t *conn,
  312. const tor_addr_t *addr, uint16_t port,
  313. const char *id_digest,
  314. int started_here)
  315. {
  316. or_options_t *options = get_options();
  317. int rate, burst; /* per-connection rate limiting params */
  318. routerinfo_t *r = router_get_by_digest(id_digest);
  319. connection_or_set_identity_digest(conn, id_digest);
  320. if (connection_or_digest_is_known_relay(id_digest)) {
  321. /* It's in the consensus, or we have a descriptor for it meaning it
  322. * was probably in a recent consensus. It's a recognized relay:
  323. * give it full bandwidth. */
  324. rate = (int)options->BandwidthRate;
  325. burst = (int)options->BandwidthBurst;
  326. } else {
  327. /* Not a recognized relay. Squeeze it down based on the suggested
  328. * bandwidth parameters in the consensus, but allow local config
  329. * options to override. */
  330. rate = options->PerConnBWRate ? (int)options->PerConnBWRate :
  331. (int)networkstatus_get_param(NULL, "bwconnrate",
  332. (int)options->BandwidthRate);
  333. burst = options->PerConnBWBurst ? (int)options->PerConnBWBurst :
  334. (int)networkstatus_get_param(NULL, "bwconnburst",
  335. (int)options->BandwidthBurst);
  336. }
  337. conn->bandwidthrate = rate;
  338. conn->read_bucket = conn->write_bucket = conn->bandwidthburst = burst;
  339. conn->_base.port = port;
  340. tor_addr_copy(&conn->_base.addr, addr);
  341. tor_addr_copy(&conn->real_addr, addr);
  342. if (r) {
  343. /* XXXX proposal 118 will make this more complex. */
  344. if (tor_addr_eq_ipv4h(&conn->_base.addr, r->addr))
  345. conn->is_canonical = 1;
  346. if (!started_here) {
  347. /* Override the addr/port, so our log messages will make sense.
  348. * This is dangerous, since if we ever try looking up a conn by
  349. * its actual addr/port, we won't remember. Careful! */
  350. /* XXXX arma: this is stupid, and it's the reason we need real_addr
  351. * to track is_canonical properly. What requires it? */
  352. /* XXXX <arma> i believe the reason we did this, originally, is because
  353. * we wanted to log what OR a connection was to, and if we logged the
  354. * right IP address and port 56244, that wouldn't be as helpful. now we
  355. * log the "right" port too, so we know if it's moria1 or moria2.
  356. */
  357. tor_addr_from_ipv4h(&conn->_base.addr, r->addr);
  358. conn->_base.port = r->or_port;
  359. }
  360. conn->nickname = tor_strdup(r->nickname);
  361. tor_free(conn->_base.address);
  362. conn->_base.address = tor_strdup(r->address);
  363. } else {
  364. const char *n;
  365. /* If we're an authoritative directory server, we may know a
  366. * nickname for this router. */
  367. n = dirserv_get_nickname_by_digest(id_digest);
  368. if (n) {
  369. conn->nickname = tor_strdup(n);
  370. } else {
  371. conn->nickname = tor_malloc(HEX_DIGEST_LEN+2);
  372. conn->nickname[0] = '$';
  373. base16_encode(conn->nickname+1, HEX_DIGEST_LEN+1,
  374. conn->identity_digest, DIGEST_LEN);
  375. }
  376. tor_free(conn->_base.address);
  377. conn->_base.address = tor_dup_addr(addr);
  378. }
  379. }
  380. /** Return true iff <b>a</b> is "better" than <b>b</b> for new circuits.
  381. *
  382. * A more canonical connection is always better than a less canonical
  383. * connection. That aside, a connection is better if it has circuits and the
  384. * other does not, or if it was created more recently.
  385. *
  386. * Requires that both input connections are open; not is_bad_for_new_circs,
  387. * and not impossibly non-canonical.
  388. *
  389. * If </b>forgive_new_connections</b> is true, then we do not call
  390. * <b>a</b>better than <b>b</b> simply because b has no circuits,
  391. * unless b is also relatively old.
  392. */
  393. static int
  394. connection_or_is_better(time_t now,
  395. const or_connection_t *a,
  396. const or_connection_t *b,
  397. int forgive_new_connections)
  398. {
  399. int newer;
  400. /** Do not definitively deprecate a new connection with no circuits on it
  401. * until this much time has passed. */
  402. #define NEW_CONN_GRACE_PERIOD (15*60)
  403. if (b->is_canonical && !a->is_canonical)
  404. return 0; /* A canonical connection is better than a non-canonical
  405. * one, no matter how new it is or which has circuits. */
  406. newer = b->_base.timestamp_created < a->_base.timestamp_created;
  407. if (
  408. /* We prefer canonical connections regardless of newness. */
  409. (!b->is_canonical && a->is_canonical) ||
  410. /* If both have circuits we prefer the newer: */
  411. (b->n_circuits && a->n_circuits && newer) ||
  412. /* If neither has circuits we prefer the newer: */
  413. (!b->n_circuits && !a->n_circuits && newer))
  414. return 1;
  415. /* If one has no circuits and the other does... */
  416. if (!b->n_circuits && a->n_circuits) {
  417. /* Then it's bad, unless it's in its grace period and we're forgiving. */
  418. if (forgive_new_connections &&
  419. now < b->_base.timestamp_created + NEW_CONN_GRACE_PERIOD)
  420. return 0;
  421. else
  422. return 1;
  423. }
  424. return 0;
  425. }
  426. /** Return the OR connection we should use to extend a circuit to the router
  427. * whose identity is <b>digest</b>, and whose address we believe (or have been
  428. * told in an extend cell) is <b>target_addr</b>. If there is no good
  429. * connection, set *<b>msg_out</b> to a message describing the connection's
  430. * state and our next action, and set <b>launch_out</b> to a boolean for
  431. * whether we should launch a new connection or not.
  432. */
  433. or_connection_t *
  434. connection_or_get_for_extend(const char *digest,
  435. const tor_addr_t *target_addr,
  436. const char **msg_out,
  437. int *launch_out)
  438. {
  439. or_connection_t *conn, *best=NULL;
  440. int n_inprogress_goodaddr = 0, n_old = 0, n_noncanonical = 0, n_possible = 0;
  441. time_t now = approx_time();
  442. tor_assert(msg_out);
  443. tor_assert(launch_out);
  444. if (!orconn_identity_map) {
  445. *msg_out = "Router not connected (nothing is). Connecting.";
  446. *launch_out = 1;
  447. return NULL;
  448. }
  449. conn = digestmap_get(orconn_identity_map, digest);
  450. for (; conn; conn = conn->next_with_same_id) {
  451. tor_assert(conn->_base.magic == OR_CONNECTION_MAGIC);
  452. tor_assert(conn->_base.type == CONN_TYPE_OR);
  453. tor_assert(!memcmp(conn->identity_digest, digest, DIGEST_LEN));
  454. if (conn->_base.marked_for_close)
  455. continue;
  456. /* Never return a non-open connection. */
  457. if (conn->_base.state != OR_CONN_STATE_OPEN) {
  458. /* If the address matches, don't launch a new connection for this
  459. * circuit. */
  460. if (!tor_addr_compare(&conn->real_addr, target_addr, CMP_EXACT))
  461. ++n_inprogress_goodaddr;
  462. continue;
  463. }
  464. /* Never return a connection that shouldn't be used for circs. */
  465. if (conn->is_bad_for_new_circs) {
  466. ++n_old;
  467. continue;
  468. }
  469. /* Never return a non-canonical connection using a recent link protocol
  470. * if the address is not what we wanted.
  471. *
  472. * (For old link protocols, we can't rely on is_canonical getting
  473. * set properly if we're talking to the right address, since we might
  474. * have an out-of-date descriptor, and we will get no NETINFO cell to
  475. * tell us about the right address.) */
  476. if (!conn->is_canonical && conn->link_proto >= 2 &&
  477. tor_addr_compare(&conn->real_addr, target_addr, CMP_EXACT)) {
  478. ++n_noncanonical;
  479. continue;
  480. }
  481. ++n_possible;
  482. if (!best) {
  483. best = conn; /* If we have no 'best' so far, this one is good enough. */
  484. continue;
  485. }
  486. if (connection_or_is_better(now, conn, best, 0))
  487. best = conn;
  488. }
  489. if (best) {
  490. *msg_out = "Connection is fine; using it.";
  491. *launch_out = 0;
  492. return best;
  493. } else if (n_inprogress_goodaddr) {
  494. *msg_out = "Connection in progress; waiting.";
  495. *launch_out = 0;
  496. return NULL;
  497. } else if (n_old || n_noncanonical) {
  498. *msg_out = "Connections all too old, or too non-canonical. "
  499. " Launching a new one.";
  500. *launch_out = 1;
  501. return NULL;
  502. } else {
  503. *msg_out = "Not connected. Connecting.";
  504. *launch_out = 1;
  505. return NULL;
  506. }
  507. }
  508. /** How old do we let a connection to an OR get before deciding it's
  509. * too old for new circuits? */
  510. #define TIME_BEFORE_OR_CONN_IS_TOO_OLD (60*60*24*7)
  511. /** Given the head of the linked list for all the or_connections with a given
  512. * identity, set elements of that list as is_bad_for_new_circs() as
  513. * appropriate. Helper for connection_or_set_bad_connections().
  514. */
  515. static void
  516. connection_or_group_set_badness(or_connection_t *head)
  517. {
  518. or_connection_t *or_conn = NULL, *best = NULL;
  519. int n_old = 0, n_inprogress = 0, n_canonical = 0, n_other = 0;
  520. time_t now = time(NULL);
  521. /* Pass 1: expire everything that's old, and see what the status of
  522. * everything else is. */
  523. for (or_conn = head; or_conn; or_conn = or_conn->next_with_same_id) {
  524. if (or_conn->_base.marked_for_close ||
  525. or_conn->is_bad_for_new_circs)
  526. continue;
  527. if (or_conn->_base.timestamp_created + TIME_BEFORE_OR_CONN_IS_TOO_OLD
  528. < now) {
  529. log_info(LD_OR,
  530. "Marking OR conn to %s:%d as too old for new circuits "
  531. "(fd %d, %d secs old).",
  532. or_conn->_base.address, or_conn->_base.port, or_conn->_base.s,
  533. (int)(now - or_conn->_base.timestamp_created));
  534. or_conn->is_bad_for_new_circs = 1;
  535. }
  536. if (or_conn->is_bad_for_new_circs) {
  537. ++n_old;
  538. } else if (or_conn->_base.state != OR_CONN_STATE_OPEN) {
  539. ++n_inprogress;
  540. } else if (or_conn->is_canonical) {
  541. ++n_canonical;
  542. } else {
  543. ++n_other;
  544. }
  545. }
  546. /* Pass 2: We know how about how good the best connection is.
  547. * expire everything that's worse, and find the very best if we can. */
  548. for (or_conn = head; or_conn; or_conn = or_conn->next_with_same_id) {
  549. if (or_conn->_base.marked_for_close ||
  550. or_conn->is_bad_for_new_circs)
  551. continue; /* This one doesn't need to be marked bad. */
  552. if (or_conn->_base.state != OR_CONN_STATE_OPEN)
  553. continue; /* Don't mark anything bad until we have seen what happens
  554. * when the connection finishes. */
  555. if (n_canonical && !or_conn->is_canonical) {
  556. /* We have at least one open canonical connection to this router,
  557. * and this one is open but not canonical. Mark it bad. */
  558. log_info(LD_OR,
  559. "Marking OR conn to %s:%d as unsuitable for new circuits: "
  560. "(fd %d, %d secs old). It is not canonical, and we have "
  561. "another connection to that OR that is.",
  562. or_conn->_base.address, or_conn->_base.port, or_conn->_base.s,
  563. (int)(now - or_conn->_base.timestamp_created));
  564. or_conn->is_bad_for_new_circs = 1;
  565. continue;
  566. }
  567. if (!best || connection_or_is_better(now, or_conn, best, 0))
  568. best = or_conn;
  569. }
  570. if (!best)
  571. return;
  572. /* Pass 3: One connection to OR is best. If it's canonical, mark as bad
  573. * every other open connection. If it's non-canonical, mark as bad
  574. * every other open connection to the same address.
  575. *
  576. * XXXX This isn't optimal; if we have connections to an OR at multiple
  577. * addresses, we'd like to pick the best _for each address_, and mark as
  578. * bad every open connection that isn't best for its address. But this
  579. * can only occur in cases where the other OR is old (so we have no
  580. * canonical connection to it), or where all the connections to the OR are
  581. * at noncanonical addresses and we have no good direct connection (which
  582. * means we aren't at risk of attaching circuits to it anyway). As
  583. * 0.1.2.x dies out, the first case will go away, and the second one is
  584. * "mostly harmless", so a fix can wait until somebody is bored.
  585. */
  586. for (or_conn = head; or_conn; or_conn = or_conn->next_with_same_id) {
  587. if (or_conn->_base.marked_for_close ||
  588. or_conn->is_bad_for_new_circs ||
  589. or_conn->_base.state != OR_CONN_STATE_OPEN)
  590. continue;
  591. if (or_conn != best && connection_or_is_better(now, best, or_conn, 1)) {
  592. /* This isn't the best conn, _and_ the best conn is better than it,
  593. even when we're being forgiving. */
  594. if (best->is_canonical) {
  595. log_info(LD_OR,
  596. "Marking OR conn to %s:%d as unsuitable for new circuits: "
  597. "(fd %d, %d secs old). We have a better canonical one "
  598. "(fd %d; %d secs old).",
  599. or_conn->_base.address, or_conn->_base.port, or_conn->_base.s,
  600. (int)(now - or_conn->_base.timestamp_created),
  601. best->_base.s, (int)(now - best->_base.timestamp_created));
  602. or_conn->is_bad_for_new_circs = 1;
  603. } else if (!tor_addr_compare(&or_conn->real_addr,
  604. &best->real_addr, CMP_EXACT)) {
  605. log_info(LD_OR,
  606. "Marking OR conn to %s:%d as unsuitable for new circuits: "
  607. "(fd %d, %d secs old). We have a better one with the "
  608. "same address (fd %d; %d secs old).",
  609. or_conn->_base.address, or_conn->_base.port, or_conn->_base.s,
  610. (int)(now - or_conn->_base.timestamp_created),
  611. best->_base.s, (int)(now - best->_base.timestamp_created));
  612. or_conn->is_bad_for_new_circs = 1;
  613. }
  614. }
  615. }
  616. }
  617. /** Go through all the OR connections, and set the is_bad_for_new_circs
  618. * flag on:
  619. * - all connections that are too old.
  620. * - all open non-canonical connections for which a canonical connection
  621. * exists to the same router.
  622. * - all open canonical connections for which a 'better' canonical
  623. * connection exists to the same router.
  624. * - all open non-canonical connections for which a 'better' non-canonical
  625. * connection exists to the same router at the same address.
  626. *
  627. * See connection_or_is_better() for our idea of what makes one OR connection
  628. * better than another.
  629. */
  630. void
  631. connection_or_set_bad_connections(void)
  632. {
  633. if (!orconn_identity_map)
  634. return;
  635. DIGESTMAP_FOREACH(orconn_identity_map, identity, or_connection_t *, conn) {
  636. connection_or_group_set_badness(conn);
  637. } DIGESTMAP_FOREACH_END;
  638. }
  639. /** <b>conn</b> is in the 'connecting' state, and it failed to complete
  640. * a TCP connection. Send notifications appropriately.
  641. *
  642. * <b>reason</b> specifies the or_conn_end_reason for the failure;
  643. * <b>msg</b> specifies the strerror-style error message.
  644. */
  645. void
  646. connection_or_connect_failed(or_connection_t *conn,
  647. int reason, const char *msg)
  648. {
  649. control_event_or_conn_status(conn, OR_CONN_EVENT_FAILED, reason);
  650. if (!authdir_mode_tests_reachability(get_options()))
  651. control_event_bootstrap_problem(msg, reason);
  652. }
  653. /** Launch a new OR connection to <b>addr</b>:<b>port</b> and expect to
  654. * handshake with an OR with identity digest <b>id_digest</b>.
  655. *
  656. * If <b>id_digest</b> is me, do nothing. If we're already connected to it,
  657. * return that connection. If the connect() is in progress, set the
  658. * new conn's state to 'connecting' and return it. If connect() succeeds,
  659. * call connection_tls_start_handshake() on it.
  660. *
  661. * This function is called from router_retry_connections(), for
  662. * ORs connecting to ORs, and circuit_establish_circuit(), for
  663. * OPs connecting to ORs.
  664. *
  665. * Return the launched conn, or NULL if it failed.
  666. */
  667. or_connection_t *
  668. connection_or_connect(const tor_addr_t *_addr, uint16_t port,
  669. const char *id_digest)
  670. {
  671. or_connection_t *conn;
  672. or_options_t *options = get_options();
  673. int socket_error = 0;
  674. int using_proxy = 0;
  675. tor_addr_t addr;
  676. tor_assert(_addr);
  677. tor_assert(id_digest);
  678. tor_addr_copy(&addr, _addr);
  679. if (server_mode(options) && router_digest_is_me(id_digest)) {
  680. log_info(LD_PROTOCOL,"Client asked me to connect to myself. Refusing.");
  681. return NULL;
  682. }
  683. conn = or_connection_new(AF_INET);
  684. /* set up conn so it's got all the data we need to remember */
  685. connection_or_init_conn_from_address(conn, &addr, port, id_digest, 1);
  686. conn->_base.state = OR_CONN_STATE_CONNECTING;
  687. control_event_or_conn_status(conn, OR_CONN_EVENT_LAUNCHED, 0);
  688. /* use a proxy server if available */
  689. if (options->HttpsProxy) {
  690. using_proxy = 1;
  691. tor_addr_copy(&addr, &options->HttpsProxyAddr);
  692. port = options->HttpsProxyPort;
  693. } else if (options->Socks4Proxy) {
  694. using_proxy = 1;
  695. tor_addr_copy(&addr, &options->Socks4ProxyAddr);
  696. port = options->Socks4ProxyPort;
  697. } else if (options->Socks5Proxy) {
  698. using_proxy = 1;
  699. tor_addr_copy(&addr, &options->Socks5ProxyAddr);
  700. port = options->Socks5ProxyPort;
  701. }
  702. switch (connection_connect(TO_CONN(conn), conn->_base.address,
  703. &addr, port, &socket_error)) {
  704. case -1:
  705. /* If the connection failed immediately, and we're using
  706. * a proxy, our proxy is down. Don't blame the Tor server. */
  707. if (!using_proxy)
  708. entry_guard_register_connect_status(conn->identity_digest,
  709. 0, 1, time(NULL));
  710. connection_or_connect_failed(conn,
  711. errno_to_orconn_end_reason(socket_error),
  712. tor_socket_strerror(socket_error));
  713. connection_free(TO_CONN(conn));
  714. return NULL;
  715. case 0:
  716. connection_watch_events(TO_CONN(conn), READ_EVENT | WRITE_EVENT);
  717. /* writable indicates finish, readable indicates broken link,
  718. error indicates broken link on windows */
  719. return conn;
  720. /* case 1: fall through */
  721. }
  722. if (connection_or_finished_connecting(conn) < 0) {
  723. /* already marked for close */
  724. return NULL;
  725. }
  726. return conn;
  727. }
  728. /** Begin the tls handshake with <b>conn</b>. <b>receiving</b> is 0 if
  729. * we initiated the connection, else it's 1.
  730. *
  731. * Assign a new tls object to conn->tls, begin reading on <b>conn</b>, and
  732. * pass <b>conn</b> to connection_tls_continue_handshake().
  733. *
  734. * Return -1 if <b>conn</b> is broken, else return 0.
  735. */
  736. int
  737. connection_tls_start_handshake(or_connection_t *conn, int receiving)
  738. {
  739. conn->_base.state = OR_CONN_STATE_TLS_HANDSHAKING;
  740. conn->tls = tor_tls_new(conn->_base.s, receiving);
  741. tor_tls_set_logged_address(conn->tls, // XXX client and relay?
  742. escaped_safe_str(conn->_base.address));
  743. if (!conn->tls) {
  744. log_warn(LD_BUG,"tor_tls_new failed. Closing.");
  745. return -1;
  746. }
  747. connection_start_reading(TO_CONN(conn));
  748. log_debug(LD_HANDSHAKE,"starting TLS handshake on fd %d", conn->_base.s);
  749. note_crypto_pk_op(receiving ? TLS_HANDSHAKE_S : TLS_HANDSHAKE_C);
  750. if (connection_tls_continue_handshake(conn) < 0) {
  751. return -1;
  752. }
  753. return 0;
  754. }
  755. /** Invoked on the server side from inside tor_tls_read() when the server
  756. * gets a successful TLS renegotiation from the client. */
  757. static void
  758. connection_or_tls_renegotiated_cb(tor_tls_t *tls, void *_conn)
  759. {
  760. or_connection_t *conn = _conn;
  761. (void)tls;
  762. /* Don't invoke this again. */
  763. tor_tls_set_renegotiate_callback(tls, NULL, NULL);
  764. tor_tls_block_renegotiation(tls);
  765. if (connection_tls_finish_handshake(conn) < 0) {
  766. /* XXXX_TLS double-check that it's ok to do this from inside read. */
  767. /* XXXX_TLS double-check that this verifies certificates. */
  768. connection_mark_for_close(TO_CONN(conn));
  769. }
  770. }
  771. /** Move forward with the tls handshake. If it finishes, hand
  772. * <b>conn</b> to connection_tls_finish_handshake().
  773. *
  774. * Return -1 if <b>conn</b> is broken, else return 0.
  775. */
  776. int
  777. connection_tls_continue_handshake(or_connection_t *conn)
  778. {
  779. int result;
  780. check_no_tls_errors();
  781. again:
  782. if (conn->_base.state == OR_CONN_STATE_TLS_CLIENT_RENEGOTIATING) {
  783. // log_notice(LD_OR, "Renegotiate with %p", conn->tls);
  784. result = tor_tls_renegotiate(conn->tls);
  785. // log_notice(LD_OR, "Result: %d", result);
  786. } else {
  787. tor_assert(conn->_base.state == OR_CONN_STATE_TLS_HANDSHAKING);
  788. // log_notice(LD_OR, "Continue handshake with %p", conn->tls);
  789. result = tor_tls_handshake(conn->tls);
  790. // log_notice(LD_OR, "Result: %d", result);
  791. }
  792. switch (result) {
  793. CASE_TOR_TLS_ERROR_ANY:
  794. log_info(LD_OR,"tls error [%s]. breaking connection.",
  795. tor_tls_err_to_string(result));
  796. return -1;
  797. case TOR_TLS_DONE:
  798. if (! tor_tls_used_v1_handshake(conn->tls)) {
  799. if (!tor_tls_is_server(conn->tls)) {
  800. if (conn->_base.state == OR_CONN_STATE_TLS_HANDSHAKING) {
  801. // log_notice(LD_OR,"Done. state was TLS_HANDSHAKING.");
  802. conn->_base.state = OR_CONN_STATE_TLS_CLIENT_RENEGOTIATING;
  803. goto again;
  804. }
  805. // log_notice(LD_OR,"Done. state was %d.", conn->_base.state);
  806. } else {
  807. /* improved handshake, but not a client. */
  808. tor_tls_set_renegotiate_callback(conn->tls,
  809. connection_or_tls_renegotiated_cb,
  810. conn);
  811. conn->_base.state = OR_CONN_STATE_TLS_SERVER_RENEGOTIATING;
  812. connection_stop_writing(TO_CONN(conn));
  813. connection_start_reading(TO_CONN(conn));
  814. return 0;
  815. }
  816. }
  817. return connection_tls_finish_handshake(conn);
  818. case TOR_TLS_WANTWRITE:
  819. connection_start_writing(TO_CONN(conn));
  820. log_debug(LD_OR,"wanted write");
  821. return 0;
  822. case TOR_TLS_WANTREAD: /* handshaking conns are *always* reading */
  823. log_debug(LD_OR,"wanted read");
  824. return 0;
  825. case TOR_TLS_CLOSE:
  826. log_info(LD_OR,"tls closed. breaking connection.");
  827. return -1;
  828. }
  829. return 0;
  830. }
  831. /** Return 1 if we initiated this connection, or 0 if it started
  832. * out as an incoming connection.
  833. */
  834. int
  835. connection_or_nonopen_was_started_here(or_connection_t *conn)
  836. {
  837. tor_assert(conn->_base.type == CONN_TYPE_OR);
  838. if (!conn->tls)
  839. return 1; /* it's still in proxy states or something */
  840. if (conn->handshake_state)
  841. return conn->handshake_state->started_here;
  842. return !tor_tls_is_server(conn->tls);
  843. }
  844. /** <b>Conn</b> just completed its handshake. Return 0 if all is well, and
  845. * return -1 if he is lying, broken, or otherwise something is wrong.
  846. *
  847. * If we initiated this connection (<b>started_here</b> is true), make sure
  848. * the other side sent a correctly formed certificate. If I initiated the
  849. * connection, make sure it's the right guy.
  850. *
  851. * Otherwise (if we _didn't_ initiate this connection), it's okay for
  852. * the certificate to be weird or absent.
  853. *
  854. * If we return 0, and the certificate is as expected, write a hash of the
  855. * identity key into <b>digest_rcvd_out</b>, which must have DIGEST_LEN
  856. * space in it.
  857. * If the certificate is invalid or missing on an incoming connection,
  858. * we return 0 and set <b>digest_rcvd_out</b> to DIGEST_LEN NUL bytes.
  859. * (If we return -1, the contents of this buffer are undefined.)
  860. *
  861. * As side effects,
  862. * 1) Set conn->circ_id_type according to tor-spec.txt.
  863. * 2) If we're an authdirserver and we initiated the connection: drop all
  864. * descriptors that claim to be on that IP/port but that aren't
  865. * this guy; and note that this guy is reachable.
  866. * 3) If this is a bridge and we didn't configure its identity
  867. * fingerprint, remember the keyid we just learned.
  868. */
  869. static int
  870. connection_or_check_valid_tls_handshake(or_connection_t *conn,
  871. int started_here,
  872. char *digest_rcvd_out)
  873. {
  874. crypto_pk_env_t *identity_rcvd=NULL;
  875. or_options_t *options = get_options();
  876. int severity = server_mode(options) ? LOG_PROTOCOL_WARN : LOG_WARN;
  877. const char *safe_address =
  878. started_here ? conn->_base.address :
  879. safe_str_client(conn->_base.address);
  880. const char *conn_type = started_here ? "outgoing" : "incoming";
  881. int has_cert = 0, has_identity=0;
  882. check_no_tls_errors();
  883. has_cert = tor_tls_peer_has_cert(conn->tls);
  884. if (started_here && !has_cert) {
  885. log_info(LD_HANDSHAKE,"Tried connecting to router at %s:%d, but it didn't "
  886. "send a cert! Closing.",
  887. safe_address, conn->_base.port);
  888. return -1;
  889. } else if (!has_cert) {
  890. log_debug(LD_HANDSHAKE,"Got incoming connection with no certificate. "
  891. "That's ok.");
  892. }
  893. check_no_tls_errors();
  894. if (has_cert) {
  895. int v = tor_tls_verify(started_here?severity:LOG_INFO,
  896. conn->tls, &identity_rcvd);
  897. if (started_here && v<0) {
  898. log_fn(severity,LD_HANDSHAKE,"Tried connecting to router at %s:%d: It"
  899. " has a cert but it's invalid. Closing.",
  900. safe_address, conn->_base.port);
  901. return -1;
  902. } else if (v<0) {
  903. log_info(LD_HANDSHAKE,"Incoming connection gave us an invalid cert "
  904. "chain; ignoring.");
  905. } else {
  906. log_debug(LD_HANDSHAKE,
  907. "The certificate seems to be valid on %s connection "
  908. "with %s:%d", conn_type, safe_address, conn->_base.port);
  909. }
  910. check_no_tls_errors();
  911. }
  912. if (identity_rcvd) {
  913. has_identity = 1;
  914. crypto_pk_get_digest(identity_rcvd, digest_rcvd_out);
  915. if (crypto_pk_cmp_keys(get_identity_key(), identity_rcvd)<0) {
  916. conn->circ_id_type = CIRC_ID_TYPE_LOWER;
  917. } else {
  918. conn->circ_id_type = CIRC_ID_TYPE_HIGHER;
  919. }
  920. crypto_free_pk_env(identity_rcvd);
  921. } else {
  922. memset(digest_rcvd_out, 0, DIGEST_LEN);
  923. conn->circ_id_type = CIRC_ID_TYPE_NEITHER;
  924. }
  925. if (started_here && tor_digest_is_zero(conn->identity_digest)) {
  926. connection_or_set_identity_digest(conn, digest_rcvd_out);
  927. tor_free(conn->nickname);
  928. conn->nickname = tor_malloc(HEX_DIGEST_LEN+2);
  929. conn->nickname[0] = '$';
  930. base16_encode(conn->nickname+1, HEX_DIGEST_LEN+1,
  931. conn->identity_digest, DIGEST_LEN);
  932. log_info(LD_HANDSHAKE, "Connected to router %s at %s:%d without knowing "
  933. "its key. Hoping for the best.",
  934. conn->nickname, conn->_base.address, conn->_base.port);
  935. /* if it's a bridge and we didn't know its identity fingerprint, now
  936. * we do -- remember it for future attempts. */
  937. learned_router_identity(&conn->_base.addr, conn->_base.port,
  938. digest_rcvd_out);
  939. }
  940. if (started_here) {
  941. int as_advertised = 1;
  942. tor_assert(has_cert);
  943. tor_assert(has_identity);
  944. if (memcmp(digest_rcvd_out, conn->identity_digest, DIGEST_LEN)) {
  945. /* I was aiming for a particular digest. I didn't get it! */
  946. char seen[HEX_DIGEST_LEN+1];
  947. char expected[HEX_DIGEST_LEN+1];
  948. base16_encode(seen, sizeof(seen), digest_rcvd_out, DIGEST_LEN);
  949. base16_encode(expected, sizeof(expected), conn->identity_digest,
  950. DIGEST_LEN);
  951. log_fn(severity, LD_HANDSHAKE,
  952. "Tried connecting to router at %s:%d, but identity key was not "
  953. "as expected: wanted %s but got %s.",
  954. conn->_base.address, conn->_base.port, expected, seen);
  955. entry_guard_register_connect_status(conn->identity_digest, 0, 1,
  956. time(NULL));
  957. control_event_or_conn_status(conn, OR_CONN_EVENT_FAILED,
  958. END_OR_CONN_REASON_OR_IDENTITY);
  959. if (!authdir_mode_tests_reachability(options))
  960. control_event_bootstrap_problem("foo", END_OR_CONN_REASON_OR_IDENTITY);
  961. as_advertised = 0;
  962. }
  963. if (authdir_mode_tests_reachability(options)) {
  964. /* We initiated this connection to address:port. Drop all routers
  965. * with the same address:port and a different key.
  966. */
  967. dirserv_orconn_tls_done(conn->_base.address, conn->_base.port,
  968. digest_rcvd_out, as_advertised);
  969. }
  970. if (!as_advertised)
  971. return -1;
  972. }
  973. return 0;
  974. }
  975. /** The tls handshake is finished.
  976. *
  977. * Make sure we are happy with the person we just handshaked with.
  978. *
  979. * If he initiated the connection, make sure he's not already connected,
  980. * then initialize conn from the information in router.
  981. *
  982. * If all is successful, call circuit_n_conn_done() to handle events
  983. * that have been pending on the <tls handshake completion. Also set the
  984. * directory to be dirty (only matters if I'm an authdirserver).
  985. */
  986. static int
  987. connection_tls_finish_handshake(or_connection_t *conn)
  988. {
  989. char digest_rcvd[DIGEST_LEN];
  990. int started_here = connection_or_nonopen_was_started_here(conn);
  991. log_debug(LD_HANDSHAKE,"tls handshake with %s done. verifying.",
  992. safe_str_client(conn->_base.address));
  993. directory_set_dirty();
  994. if (connection_or_check_valid_tls_handshake(conn, started_here,
  995. digest_rcvd) < 0)
  996. return -1;
  997. circuit_build_times_network_is_live(&circ_times);
  998. if (tor_tls_used_v1_handshake(conn->tls)) {
  999. conn->link_proto = 1;
  1000. if (!started_here) {
  1001. connection_or_init_conn_from_address(conn, &conn->_base.addr,
  1002. conn->_base.port, digest_rcvd, 0);
  1003. }
  1004. tor_tls_block_renegotiation(conn->tls);
  1005. return connection_or_set_state_open(conn);
  1006. } else {
  1007. conn->_base.state = OR_CONN_STATE_OR_HANDSHAKING;
  1008. if (connection_init_or_handshake_state(conn, started_here) < 0)
  1009. return -1;
  1010. if (!started_here) {
  1011. connection_or_init_conn_from_address(conn, &conn->_base.addr,
  1012. conn->_base.port, digest_rcvd, 0);
  1013. }
  1014. return connection_or_send_versions(conn);
  1015. }
  1016. }
  1017. /** Allocate a new connection handshake state for the connection
  1018. * <b>conn</b>. Return 0 on success, -1 on failure. */
  1019. static int
  1020. connection_init_or_handshake_state(or_connection_t *conn, int started_here)
  1021. {
  1022. or_handshake_state_t *s;
  1023. s = conn->handshake_state = tor_malloc_zero(sizeof(or_handshake_state_t));
  1024. s->started_here = started_here ? 1 : 0;
  1025. return 0;
  1026. }
  1027. /** Free all storage held by <b>state</b>. */
  1028. void
  1029. or_handshake_state_free(or_handshake_state_t *state)
  1030. {
  1031. if (!state)
  1032. return;
  1033. memset(state, 0xBE, sizeof(or_handshake_state_t));
  1034. tor_free(state);
  1035. }
  1036. /** Set <b>conn</b>'s state to OR_CONN_STATE_OPEN, and tell other subsystems
  1037. * as appropriate. Called when we are done with all TLS and OR handshaking.
  1038. */
  1039. int
  1040. connection_or_set_state_open(or_connection_t *conn)
  1041. {
  1042. int started_here = connection_or_nonopen_was_started_here(conn);
  1043. time_t now = time(NULL);
  1044. conn->_base.state = OR_CONN_STATE_OPEN;
  1045. control_event_or_conn_status(conn, OR_CONN_EVENT_CONNECTED, 0);
  1046. if (started_here) {
  1047. circuit_build_times_network_is_live(&circ_times);
  1048. rep_hist_note_connect_succeeded(conn->identity_digest, now);
  1049. if (entry_guard_register_connect_status(conn->identity_digest,
  1050. 1, 0, now) < 0) {
  1051. /* Close any circuits pending on this conn. We leave it in state
  1052. * 'open' though, because it didn't actually *fail* -- we just
  1053. * chose not to use it. (Otherwise
  1054. * connection_about_to_close_connection() will call a big pile of
  1055. * functions to indicate we shouldn't try it again.) */
  1056. log_debug(LD_OR, "New entry guard was reachable, but closing this "
  1057. "connection so we can retry the earlier entry guards.");
  1058. circuit_n_conn_done(conn, 0);
  1059. return -1;
  1060. }
  1061. router_set_status(conn->identity_digest, 1);
  1062. } else {
  1063. /* only report it to the geoip module if it's not a known router */
  1064. if (!router_get_by_digest(conn->identity_digest)) {
  1065. if (tor_addr_family(&TO_CONN(conn)->addr) == AF_INET) {
  1066. /*XXXX IP6 support ipv6 geoip.*/
  1067. uint32_t a = tor_addr_to_ipv4h(&TO_CONN(conn)->addr);
  1068. geoip_note_client_seen(GEOIP_CLIENT_CONNECT, a, now);
  1069. }
  1070. }
  1071. }
  1072. or_handshake_state_free(conn->handshake_state);
  1073. conn->handshake_state = NULL;
  1074. connection_start_reading(TO_CONN(conn));
  1075. circuit_n_conn_done(conn, 1); /* send the pending creates, if any. */
  1076. return 0;
  1077. }
  1078. /** Pack <b>cell</b> into wire-format, and write it onto <b>conn</b>'s outbuf.
  1079. * For cells that use or affect a circuit, this should only be called by
  1080. * connection_or_flush_from_first_active_circuit().
  1081. */
  1082. void
  1083. connection_or_write_cell_to_buf(const cell_t *cell, or_connection_t *conn)
  1084. {
  1085. packed_cell_t networkcell;
  1086. tor_assert(cell);
  1087. tor_assert(conn);
  1088. cell_pack(&networkcell, cell);
  1089. connection_write_to_buf(networkcell.body, CELL_NETWORK_SIZE, TO_CONN(conn));
  1090. if (cell->command != CELL_PADDING)
  1091. conn->timestamp_last_added_nonpadding = approx_time();
  1092. }
  1093. /** Pack a variable-length <b>cell</b> into wire-format, and write it onto
  1094. * <b>conn</b>'s outbuf. Right now, this <em>DOES NOT</em> support cells that
  1095. * affect a circuit.
  1096. */
  1097. void
  1098. connection_or_write_var_cell_to_buf(const var_cell_t *cell,
  1099. or_connection_t *conn)
  1100. {
  1101. char hdr[VAR_CELL_HEADER_SIZE];
  1102. tor_assert(cell);
  1103. tor_assert(conn);
  1104. var_cell_pack_header(cell, hdr);
  1105. connection_write_to_buf(hdr, sizeof(hdr), TO_CONN(conn));
  1106. connection_write_to_buf(cell->payload, cell->payload_len, TO_CONN(conn));
  1107. if (cell->command != CELL_PADDING)
  1108. conn->timestamp_last_added_nonpadding = approx_time();
  1109. }
  1110. /** See whether there's a variable-length cell waiting on <b>conn</b>'s
  1111. * inbuf. Return values as for fetch_var_cell_from_buf(). */
  1112. static int
  1113. connection_fetch_var_cell_from_buf(or_connection_t *conn, var_cell_t **out)
  1114. {
  1115. return fetch_var_cell_from_buf(conn->_base.inbuf, out, conn->link_proto);
  1116. }
  1117. /** Process cells from <b>conn</b>'s inbuf.
  1118. *
  1119. * Loop: while inbuf contains a cell, pull it off the inbuf, unpack it,
  1120. * and hand it to command_process_cell().
  1121. *
  1122. * Always return 0.
  1123. */
  1124. static int
  1125. connection_or_process_cells_from_inbuf(or_connection_t *conn)
  1126. {
  1127. var_cell_t *var_cell;
  1128. while (1) {
  1129. log_debug(LD_OR,
  1130. "%d: starting, inbuf_datalen %d (%d pending in tls object).",
  1131. conn->_base.s,(int)buf_datalen(conn->_base.inbuf),
  1132. tor_tls_get_pending_bytes(conn->tls));
  1133. if (connection_fetch_var_cell_from_buf(conn, &var_cell)) {
  1134. if (!var_cell)
  1135. return 0; /* not yet. */
  1136. circuit_build_times_network_is_live(&circ_times);
  1137. command_process_var_cell(var_cell, conn);
  1138. var_cell_free(var_cell);
  1139. } else {
  1140. char buf[CELL_NETWORK_SIZE];
  1141. cell_t cell;
  1142. if (buf_datalen(conn->_base.inbuf) < CELL_NETWORK_SIZE) /* whole response
  1143. available? */
  1144. return 0; /* not yet */
  1145. circuit_build_times_network_is_live(&circ_times);
  1146. connection_fetch_from_buf(buf, CELL_NETWORK_SIZE, TO_CONN(conn));
  1147. /* retrieve cell info from buf (create the host-order struct from the
  1148. * network-order string) */
  1149. cell_unpack(&cell, buf);
  1150. command_process_cell(&cell, conn);
  1151. }
  1152. }
  1153. }
  1154. /** Write a destroy cell with circ ID <b>circ_id</b> and reason <b>reason</b>
  1155. * onto OR connection <b>conn</b>. Don't perform range-checking on reason:
  1156. * we may want to propagate reasons from other cells.
  1157. *
  1158. * Return 0.
  1159. */
  1160. int
  1161. connection_or_send_destroy(circid_t circ_id, or_connection_t *conn, int reason)
  1162. {
  1163. cell_t cell;
  1164. tor_assert(conn);
  1165. memset(&cell, 0, sizeof(cell_t));
  1166. cell.circ_id = circ_id;
  1167. cell.command = CELL_DESTROY;
  1168. cell.payload[0] = (uint8_t) reason;
  1169. log_debug(LD_OR,"Sending destroy (circID %d).", circ_id);
  1170. /* XXXX It's possible that under some circumstances, we want the destroy
  1171. * to take precedence over other data waiting on the circuit's cell queue.
  1172. */
  1173. connection_or_write_cell_to_buf(&cell, conn);
  1174. return 0;
  1175. }
  1176. /** Array of recognized link protocol versions. */
  1177. static const uint16_t or_protocol_versions[] = { 1, 2 };
  1178. /** Number of versions in <b>or_protocol_versions</b>. */
  1179. static const int n_or_protocol_versions =
  1180. (int)( sizeof(or_protocol_versions)/sizeof(uint16_t) );
  1181. /** Return true iff <b>v</b> is a link protocol version that this Tor
  1182. * implementation believes it can support. */
  1183. int
  1184. is_or_protocol_version_known(uint16_t v)
  1185. {
  1186. int i;
  1187. for (i = 0; i < n_or_protocol_versions; ++i) {
  1188. if (or_protocol_versions[i] == v)
  1189. return 1;
  1190. }
  1191. return 0;
  1192. }
  1193. /** Send a VERSIONS cell on <b>conn</b>, telling the other host about the
  1194. * link protocol versions that this Tor can support. */
  1195. static int
  1196. connection_or_send_versions(or_connection_t *conn)
  1197. {
  1198. var_cell_t *cell;
  1199. int i;
  1200. tor_assert(conn->handshake_state &&
  1201. !conn->handshake_state->sent_versions_at);
  1202. cell = var_cell_new(n_or_protocol_versions * 2);
  1203. cell->command = CELL_VERSIONS;
  1204. for (i = 0; i < n_or_protocol_versions; ++i) {
  1205. uint16_t v = or_protocol_versions[i];
  1206. set_uint16(cell->payload+(2*i), htons(v));
  1207. }
  1208. connection_or_write_var_cell_to_buf(cell, conn);
  1209. conn->handshake_state->sent_versions_at = time(NULL);
  1210. var_cell_free(cell);
  1211. return 0;
  1212. }
  1213. /** Send a NETINFO cell on <b>conn</b>, telling the other server what we know
  1214. * about their address, our address, and the current time. */
  1215. int
  1216. connection_or_send_netinfo(or_connection_t *conn)
  1217. {
  1218. cell_t cell;
  1219. time_t now = time(NULL);
  1220. routerinfo_t *me;
  1221. int len;
  1222. char *out;
  1223. memset(&cell, 0, sizeof(cell_t));
  1224. cell.command = CELL_NETINFO;
  1225. /* Timestamp. */
  1226. set_uint32(cell.payload, htonl((uint32_t)now));
  1227. /* Their address. */
  1228. out = cell.payload + 4;
  1229. len = append_address_to_payload(out, &conn->_base.addr);
  1230. if (len<0)
  1231. return -1;
  1232. out += len;
  1233. /* My address. */
  1234. if ((me = router_get_my_routerinfo())) {
  1235. tor_addr_t my_addr;
  1236. *out++ = 1; /* only one address is supported. */
  1237. tor_addr_from_ipv4h(&my_addr, me->addr);
  1238. len = append_address_to_payload(out, &my_addr);
  1239. if (len < 0)
  1240. return -1;
  1241. out += len;
  1242. } else {
  1243. *out++ = 0;
  1244. }
  1245. connection_or_write_cell_to_buf(&cell, conn);
  1246. return 0;
  1247. }