connection_or.c 47 KB

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