connection_or.c 47 KB

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