connection_or.c 47 KB

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