connection_or.c 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804
  1. /* Copyright (c) 2001 Matej Pfajfar.
  2. * Copyright (c) 2001-2004, Roger Dingledine.
  3. * Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson. */
  4. /* See LICENSE for licensing information */
  5. /* $Id$ */
  6. const char connection_or_c_id[] =
  7. "$Id$";
  8. /**
  9. * \file connection_or.c
  10. * \brief Functions to handle OR connections, TLS handshaking, and
  11. * cells on the network.
  12. **/
  13. #include "or.h"
  14. /** How much clock skew do we tolerate when checking certificates for
  15. * known routers? (sec) */
  16. #define TIGHT_CERT_ALLOW_SKEW (90*60)
  17. static int connection_tls_finish_handshake(connection_t *conn);
  18. static int connection_or_process_cells_from_inbuf(connection_t *conn);
  19. /**************************************************************/
  20. /** Map from identity digest of connected OR or desired OR to a connection_t
  21. * with that identity digest. If there is more than one such connection_t,
  22. * they form a linked list, with next_with_same_id as the next pointer.*/
  23. static digestmap_t *orconn_identity_map = NULL;
  24. /** If conn is listed in orconn_identity_map, remove it, and clear
  25. * conn->identity_digest. */
  26. void
  27. connection_or_remove_from_identity_map(connection_t *conn)
  28. {
  29. connection_t *tmp;
  30. tor_assert(conn);
  31. tor_assert(conn->type == CONN_TYPE_OR);
  32. if (!orconn_identity_map)
  33. return;
  34. tmp = digestmap_get(orconn_identity_map, conn->identity_digest);
  35. if (!tmp)
  36. return;
  37. if (conn == tmp) {
  38. if (conn->next_with_same_id)
  39. digestmap_set(orconn_identity_map, conn->identity_digest,
  40. conn->next_with_same_id);
  41. else
  42. digestmap_remove(orconn_identity_map, conn->identity_digest);
  43. } else {
  44. while (tmp->next_with_same_id) {
  45. if (tmp->next_with_same_id == conn) {
  46. tmp->next_with_same_id = conn->next_with_same_id;
  47. break;
  48. }
  49. tmp = tmp->next_with_same_id;
  50. }
  51. }
  52. memset(conn->identity_digest, 0, DIGEST_LEN);
  53. conn->next_with_same_id = NULL;
  54. }
  55. /** Remove all entries from the identity-to-orconn map, and clear
  56. * all identities in OR conns.*/
  57. void
  58. connection_or_clear_identity_map(void)
  59. {
  60. int i, n;
  61. connection_t **carray;
  62. get_connection_array(&carray,&n);
  63. for (i = 0; i < n; ++i) {
  64. connection_t* conn = carray[i];
  65. if (conn->type == CONN_TYPE_OR) {
  66. memset(conn->identity_digest, 0, DIGEST_LEN);
  67. conn->next_with_same_id = NULL;
  68. }
  69. }
  70. if (orconn_identity_map) {
  71. digestmap_free(orconn_identity_map, NULL);
  72. orconn_identity_map = NULL;
  73. }
  74. }
  75. /** Change conn->identity_digest to digest, and add conn into
  76. * orconn_digest_map. */
  77. static void
  78. connection_or_set_identity_digest(connection_t *conn, const char *digest)
  79. {
  80. connection_t *tmp;
  81. tor_assert(conn);
  82. tor_assert(conn->type == CONN_TYPE_OR);
  83. tor_assert(digest);
  84. if (!orconn_identity_map)
  85. orconn_identity_map = digestmap_new();
  86. if (!memcmp(conn->identity_digest, digest, DIGEST_LEN))
  87. return;
  88. if (tor_digest_is_zero(conn->identity_digest))
  89. connection_or_remove_from_identity_map(conn);
  90. memcpy(conn->identity_digest, digest, DIGEST_LEN);
  91. tmp = digestmap_set(orconn_identity_map, digest, conn);
  92. conn->next_with_same_id = tmp;
  93. /* Checking code; remove once I'm sure this works. XXXX*/
  94. for (; tmp; tmp = tmp->next_with_same_id) {
  95. tor_assert(!memcmp(tmp->identity_digest, digest, DIGEST_LEN));
  96. tor_assert(tmp != conn);
  97. }
  98. }
  99. /** Pack the cell_t host-order structure <b>src</b> into network-order
  100. * in the buffer <b>dest</b>. See tor-spec.txt for details about the
  101. * wire format.
  102. */
  103. static void
  104. cell_pack(char *dest, const cell_t *src)
  105. {
  106. *(uint16_t*)dest = htons(src->circ_id);
  107. *(uint8_t*)(dest+2) = src->command;
  108. memcpy(dest+3, src->payload, CELL_PAYLOAD_SIZE);
  109. }
  110. /** Unpack the network-order buffer <b>src</b> into a host-order
  111. * cell_t structure <b>dest</b>.
  112. */
  113. static void
  114. cell_unpack(cell_t *dest, const char *src)
  115. {
  116. dest->circ_id = ntohs(*(uint16_t*)(src));
  117. dest->command = *(uint8_t*)(src+2);
  118. memcpy(dest->payload, src+3, CELL_PAYLOAD_SIZE);
  119. }
  120. int
  121. connection_or_reached_eof(connection_t *conn)
  122. {
  123. info(LD_OR,"OR connection reached EOF. Closing.");
  124. connection_mark_for_close(conn);
  125. return 0;
  126. }
  127. /** Read conn's inbuf. If the http response from the proxy is all
  128. * here, make sure it's good news, and begin the tls handshake. If
  129. * it's bad news, close the connection and return -1. Else return 0
  130. * and hope for better luck next time.
  131. */
  132. static int
  133. connection_or_read_proxy_response(connection_t *conn)
  134. {
  135. char *headers;
  136. char *reason=NULL;
  137. int status_code;
  138. time_t date_header;
  139. int compression;
  140. switch (fetch_from_buf_http(conn->inbuf,
  141. &headers, MAX_HEADERS_SIZE,
  142. NULL, NULL, 10000, 0)) {
  143. case -1: /* overflow */
  144. warn(LD_PROTOCOL,"Your https proxy sent back an oversized response. "
  145. "Closing.");
  146. return -1;
  147. case 0:
  148. info(LD_OR,"https proxy response not all here yet. Waiting.");
  149. return 0;
  150. /* case 1, fall through */
  151. }
  152. if (parse_http_response(headers, &status_code, &date_header,
  153. &compression, &reason) < 0) {
  154. warn(LD_OR,"Unparseable headers from proxy (connecting to '%s'). Closing.",
  155. conn->address);
  156. tor_free(headers);
  157. return -1;
  158. }
  159. if (!reason) reason = tor_strdup("[no reason given]");
  160. if (status_code == 200) {
  161. info(LD_OR,
  162. "HTTPS connect to '%s' successful! (200 \"%s\") Starting TLS.",
  163. conn->address, reason);
  164. tor_free(reason);
  165. if (connection_tls_start_handshake(conn, 0) < 0) {
  166. /* TLS handshaking error of some kind. */
  167. connection_mark_for_close(conn);
  168. return -1;
  169. }
  170. return 0;
  171. }
  172. /* else, bad news on the status code */
  173. warn(LD_OR,
  174. "The https proxy sent back an unexpected status code %d (\"%s\"). "
  175. "Closing.",
  176. status_code, reason);
  177. tor_free(reason);
  178. connection_mark_for_close(conn);
  179. return -1;
  180. }
  181. /** Handle any new bytes that have come in on connection <b>conn</b>.
  182. * If conn is in 'open' state, hand it to
  183. * connection_or_process_cells_from_inbuf()
  184. * (else do nothing).
  185. */
  186. int
  187. connection_or_process_inbuf(connection_t *conn)
  188. {
  189. tor_assert(conn);
  190. tor_assert(conn->type == CONN_TYPE_OR);
  191. switch (conn->state) {
  192. case OR_CONN_STATE_PROXY_READING:
  193. return connection_or_read_proxy_response(conn);
  194. case OR_CONN_STATE_OPEN:
  195. return connection_or_process_cells_from_inbuf(conn);
  196. default:
  197. return 0; /* don't do anything */
  198. }
  199. }
  200. /** Connection <b>conn</b> has finished writing and has no bytes left on
  201. * its outbuf.
  202. *
  203. * Otherwise it's in state "open": stop writing and return.
  204. *
  205. * If <b>conn</b> is broken, mark it for close and return -1, else
  206. * return 0.
  207. */
  208. int
  209. connection_or_finished_flushing(connection_t *conn)
  210. {
  211. tor_assert(conn);
  212. tor_assert(conn->type == CONN_TYPE_OR);
  213. assert_connection_ok(conn,0);
  214. switch (conn->state) {
  215. case OR_CONN_STATE_PROXY_FLUSHING:
  216. debug(LD_OR,"finished sending CONNECT to proxy.");
  217. conn->state = OR_CONN_STATE_PROXY_READING;
  218. connection_stop_writing(conn);
  219. break;
  220. case OR_CONN_STATE_OPEN:
  221. connection_stop_writing(conn);
  222. break;
  223. default:
  224. err(LD_BUG,"BUG: called in unexpected state %d.", conn->state);
  225. tor_fragile_assert();
  226. return -1;
  227. }
  228. return 0;
  229. }
  230. /** Connected handler for OR connections: begin the TLS handshake.
  231. */
  232. int
  233. connection_or_finished_connecting(connection_t *conn)
  234. {
  235. tor_assert(conn);
  236. tor_assert(conn->type == CONN_TYPE_OR);
  237. tor_assert(conn->state == OR_CONN_STATE_CONNECTING);
  238. debug(LD_OR,"OR connect() to router at %s:%u finished.",
  239. conn->address,conn->port);
  240. if (get_options()->HttpsProxy) {
  241. char buf[1024];
  242. char addrbuf[INET_NTOA_BUF_LEN];
  243. struct in_addr in;
  244. char *base64_authenticator=NULL;
  245. const char *authenticator = get_options()->HttpsProxyAuthenticator;
  246. in.s_addr = htonl(conn->addr);
  247. tor_inet_ntoa(&in, addrbuf, sizeof(addrbuf));
  248. if (authenticator) {
  249. base64_authenticator = alloc_http_authenticator(authenticator);
  250. if (!base64_authenticator)
  251. warn(LD_OR, "Encoding https authenticator failed");
  252. }
  253. if (base64_authenticator) {
  254. tor_snprintf(buf, sizeof(buf), "CONNECT %s:%d HTTP/1.1\r\n"
  255. "Proxy-Authorization: Basic %s\r\n\r\n", addrbuf,
  256. conn->port, base64_authenticator);
  257. tor_free(base64_authenticator);
  258. } else {
  259. tor_snprintf(buf, sizeof(buf), "CONNECT %s:%d HTTP/1.0\r\n\r\n",
  260. addrbuf, conn->port);
  261. }
  262. connection_write_to_buf(buf, strlen(buf), conn);
  263. conn->state = OR_CONN_STATE_PROXY_FLUSHING;
  264. return 0;
  265. }
  266. if (connection_tls_start_handshake(conn, 0) < 0) {
  267. /* TLS handshaking error of some kind. */
  268. connection_mark_for_close(conn);
  269. return -1;
  270. }
  271. return 0;
  272. }
  273. /** Initialize <b>conn</b> to include all the relevant data from <b>router</b>.
  274. * This function is called either from connection_or_connect(), if
  275. * we initiated the connect, or from connection_tls_finish_handshake()
  276. * if the other side initiated it.
  277. */
  278. static void
  279. connection_or_init_conn_from_router(connection_t *conn, routerinfo_t *router)
  280. {
  281. or_options_t *options = get_options();
  282. conn->addr = router->addr;
  283. conn->port = router->or_port;
  284. conn->receiver_bucket = conn->bandwidth = (int)options->BandwidthBurst;
  285. conn->identity_pkey = crypto_pk_dup_key(router->identity_pkey);
  286. connection_or_set_identity_digest(conn, router->cache_info.identity_digest);
  287. conn->nickname = tor_strdup(router->nickname);
  288. tor_free(conn->address);
  289. conn->address = tor_strdup(router->address);
  290. }
  291. /** If we don't necessarily know the router we're connecting to, but we
  292. * have an addr/port/id_digest, then fill in as much as we can. Start
  293. * by checking to see if this describes a router we know. */
  294. static void
  295. connection_or_init_conn_from_address(connection_t *conn,
  296. uint32_t addr, uint16_t port,
  297. const char *id_digest)
  298. {
  299. const char *n;
  300. or_options_t *options = get_options();
  301. routerinfo_t *r = router_get_by_digest(id_digest);
  302. if (r) {
  303. connection_or_init_conn_from_router(conn,r);
  304. return;
  305. }
  306. conn->addr = addr;
  307. conn->port = port;
  308. /* This next part isn't really right, but it's good enough for now. */
  309. conn->receiver_bucket = conn->bandwidth = (int)options->BandwidthBurst;
  310. connection_or_set_identity_digest(conn, id_digest);
  311. /* If we're an authoritative directory server, we may know a
  312. * nickname for this router. */
  313. n = dirserv_get_nickname_by_digest(id_digest);
  314. if (n) {
  315. conn->nickname = tor_strdup(n);
  316. } else {
  317. conn->nickname = tor_malloc(HEX_DIGEST_LEN+2);
  318. conn->nickname[0] = '$';
  319. base16_encode(conn->nickname+1, HEX_DIGEST_LEN+1,
  320. conn->identity_digest, DIGEST_LEN);
  321. }
  322. tor_free(conn->address);
  323. conn->address = tor_dup_addr(addr);
  324. }
  325. /** Return the best connection of type OR with the
  326. * digest <b>digest</b> that we have, or NULL if we have none.
  327. *
  328. * 1) Don't return it if it's marked for close.
  329. * 2) If there are any open conns, ignore non-open conns.
  330. * 3) If there are any non-obsolete conns, ignore obsolete conns.
  331. * 4) Then if there are any non-empty conns, ignore empty conns.
  332. * 5) Of the remaining conns, prefer newer conns.
  333. */
  334. connection_t *
  335. connection_or_get_by_identity_digest(const char *digest)
  336. {
  337. int newer;
  338. connection_t *conn, *best=NULL;
  339. if (!orconn_identity_map)
  340. return NULL;
  341. conn = digestmap_get(orconn_identity_map, digest);
  342. for (; conn; conn = conn->next_with_same_id) {
  343. tor_assert(conn->magic == CONNECTION_MAGIC);
  344. tor_assert(conn->type == CONN_TYPE_OR);
  345. tor_assert(!memcmp(conn->identity_digest, digest, DIGEST_LEN));
  346. if (conn->marked_for_close)
  347. continue;
  348. if (!best) {
  349. best = conn; /* whatever it is, it's better than nothing. */
  350. continue;
  351. }
  352. if (best->state == OR_CONN_STATE_OPEN &&
  353. conn->state != OR_CONN_STATE_OPEN)
  354. continue; /* avoid non-open conns if we can */
  355. newer = best->timestamp_created < conn->timestamp_created;
  356. if (conn->is_obsolete && (!best->is_obsolete || !newer))
  357. continue; /* we have something, and it's better than this. */
  358. if (best->n_circuits && !conn->n_circuits)
  359. continue; /* prefer conns with circuits on them */
  360. if (newer)
  361. best = conn; /* lastly, prefer newer conns */
  362. }
  363. return best;
  364. }
  365. /** Launch a new OR connection to <b>addr</b>:<b>port</b> and expect to
  366. * handshake with an OR with identity digest <b>id_digest</b>.
  367. *
  368. * If <b>id_digest</b> is me, do nothing. If we're already connected to it,
  369. * return that connection. If the connect() is in progress, set the
  370. * new conn's state to 'connecting' and return it. If connect() succeeds,
  371. * call connection_tls_start_handshake() on it.
  372. *
  373. * This function is called from router_retry_connections(), for
  374. * ORs connecting to ORs, and circuit_establish_circuit(), for
  375. * OPs connecting to ORs.
  376. *
  377. * Return the launched conn, or NULL if it failed.
  378. */
  379. connection_t *
  380. connection_or_connect(uint32_t addr, uint16_t port, const char *id_digest)
  381. {
  382. connection_t *conn;
  383. routerinfo_t *me;
  384. or_options_t *options = get_options();
  385. tor_assert(id_digest);
  386. if (server_mode(options) && (me=router_get_my_routerinfo()) &&
  387. router_digest_is_me(id_digest)) {
  388. info(LD_PROTOCOL,"Client asked me to connect to myself. Refusing.");
  389. return NULL;
  390. }
  391. conn = connection_new(CONN_TYPE_OR);
  392. /* set up conn so it's got all the data we need to remember */
  393. connection_or_init_conn_from_address(conn, addr, port, id_digest);
  394. conn->state = OR_CONN_STATE_CONNECTING;
  395. control_event_or_conn_status(conn, OR_CONN_EVENT_LAUNCHED);
  396. if (options->HttpsProxy) {
  397. /* we shouldn't connect directly. use the https proxy instead. */
  398. addr = options->HttpsProxyAddr;
  399. port = options->HttpsProxyPort;
  400. }
  401. switch (connection_connect(conn, conn->address, addr, port)) {
  402. case -1:
  403. /* If the connection failed immediately, and we're using
  404. * an https proxy, our https proxy is down. Don't blame the
  405. * Tor server. */
  406. if (!options->HttpsProxy) {
  407. router_mark_as_down(conn->identity_digest);
  408. entry_guard_set_status(conn->identity_digest, 0);
  409. }
  410. control_event_or_conn_status(conn, OR_CONN_EVENT_FAILED);
  411. connection_free(conn);
  412. return NULL;
  413. case 0:
  414. connection_watch_events(conn, EV_READ | EV_WRITE);
  415. /* writable indicates finish, readable indicates broken link,
  416. error indicates broken link on windows */
  417. return conn;
  418. /* case 1: fall through */
  419. }
  420. if (connection_or_finished_connecting(conn) < 0) {
  421. /* already marked for close */
  422. return NULL;
  423. }
  424. return conn;
  425. }
  426. /** Begin the tls handshake with <b>conn</b>. <b>receiving</b> is 0 if
  427. * we initiated the connection, else it's 1.
  428. *
  429. * Assign a new tls object to conn->tls, begin reading on <b>conn</b>, and
  430. * pass <b>conn</b> to connection_tls_continue_handshake().
  431. *
  432. * Return -1 if <b>conn</b> is broken, else return 0.
  433. */
  434. int
  435. connection_tls_start_handshake(connection_t *conn, int receiving)
  436. {
  437. conn->state = OR_CONN_STATE_HANDSHAKING;
  438. conn->tls = tor_tls_new(conn->s, receiving, 0);
  439. if (!conn->tls) {
  440. warn(LD_BUG,"tor_tls_new failed. Closing.");
  441. return -1;
  442. }
  443. connection_start_reading(conn);
  444. debug(LD_OR,"starting TLS handshake on fd %d", conn->s);
  445. if (connection_tls_continue_handshake(conn) < 0) {
  446. return -1;
  447. }
  448. return 0;
  449. }
  450. /** Move forward with the tls handshake. If it finishes, hand
  451. * <b>conn</b> to connection_tls_finish_handshake().
  452. *
  453. * Return -1 if <b>conn</b> is broken, else return 0.
  454. */
  455. int
  456. connection_tls_continue_handshake(connection_t *conn)
  457. {
  458. check_no_tls_errors();
  459. switch (tor_tls_handshake(conn->tls)) {
  460. case TOR_TLS_ERROR:
  461. case TOR_TLS_CLOSE:
  462. info(LD_OR,"tls error. breaking connection.");
  463. return -1;
  464. case TOR_TLS_DONE:
  465. return connection_tls_finish_handshake(conn);
  466. case TOR_TLS_WANTWRITE:
  467. connection_start_writing(conn);
  468. debug(LD_OR,"wanted write");
  469. return 0;
  470. case TOR_TLS_WANTREAD: /* handshaking conns are *always* reading */
  471. debug(LD_OR,"wanted read");
  472. return 0;
  473. }
  474. return 0;
  475. }
  476. /** Return 1 if we initiated this connection, or 0 if it started
  477. * out as an incoming connection.
  478. *
  479. * This is implemented for now by checking to see if
  480. * conn-\>identity_digest is set or not. Perhaps we should add a flag
  481. * one day so we're clearer.
  482. */
  483. int
  484. connection_or_nonopen_was_started_here(connection_t *conn)
  485. {
  486. tor_assert(conn->type == CONN_TYPE_OR);
  487. if (tor_digest_is_zero(conn->identity_digest))
  488. return 0;
  489. else
  490. return 1;
  491. }
  492. /** Conn just completed its handshake. Return 0 if all is well, and
  493. * return -1 if he is lying, broken, or otherwise something is wrong.
  494. *
  495. * Make sure he sent a correctly formed certificate. If it has a
  496. * recognized (approved) nickname, make sure his identity key matches
  497. * it. If I initiated the connection, make sure it's the right guy.
  498. *
  499. * If we return 0, write a hash of the identity key into digest_rcvd,
  500. * which must have DIGEST_LEN space in it. (If we return -1 this
  501. * buffer is undefined.)
  502. *
  503. * As side effects,
  504. * 1) Set conn->circ_id_type according to tor-spec.txt
  505. * 2) If we're an authdirserver and we initiated the connection: drop all
  506. * descriptors that claim to be on that IP/port but that aren't
  507. * this guy; and note that this guy is reachable.
  508. */
  509. static int
  510. connection_or_check_valid_handshake(connection_t *conn, char *digest_rcvd)
  511. {
  512. routerinfo_t *router;
  513. crypto_pk_env_t *identity_rcvd=NULL;
  514. char nickname[MAX_NICKNAME_LEN+1];
  515. or_options_t *options = get_options();
  516. int severity = server_mode(options) ? LOG_PROTOCOL_WARN : LOG_WARN;
  517. check_no_tls_errors();
  518. if (! tor_tls_peer_has_cert(conn->tls)) {
  519. info(LD_PROTOCOL,"Peer didn't send a cert! Closing.");
  520. return -1;
  521. }
  522. check_no_tls_errors();
  523. if (tor_tls_get_peer_cert_nickname(conn->tls, nickname, sizeof(nickname))) {
  524. log_fn(severity,LD_PROTOCOL,"Other side (%s:%d) has a cert without a "
  525. "valid nickname. Closing.",
  526. conn->address, conn->port);
  527. return -1;
  528. }
  529. check_no_tls_errors();
  530. debug(LD_OR, "Other side (%s:%d) claims to be router '%s'",
  531. conn->address, conn->port, nickname);
  532. if (tor_tls_verify(severity, conn->tls, &identity_rcvd) < 0) {
  533. log_fn(severity,LD_OR,"Other side, which claims to be router '%s' (%s:%d),"
  534. " has a cert but it's invalid. Closing.",
  535. nickname, conn->address, conn->port);
  536. return -1;
  537. }
  538. check_no_tls_errors();
  539. debug(LD_OR,"The router's cert is valid.");
  540. crypto_pk_get_digest(identity_rcvd, digest_rcvd);
  541. if (crypto_pk_cmp_keys(get_identity_key(), identity_rcvd)<0) {
  542. conn->circ_id_type = CIRC_ID_TYPE_LOWER;
  543. } else {
  544. conn->circ_id_type = CIRC_ID_TYPE_HIGHER;
  545. }
  546. crypto_free_pk_env(identity_rcvd);
  547. router = router_get_by_nickname(nickname, 0);
  548. if (router && /* we know this nickname */
  549. router->is_named && /* make sure it's the right guy */
  550. memcmp(digest_rcvd, router->cache_info.identity_digest,DIGEST_LEN) !=0) {
  551. log_fn(severity, LD_OR,
  552. "Identity key not as expected for router claiming to be "
  553. "'%s' (%s:%d)",
  554. nickname, conn->address, conn->port);
  555. return -1;
  556. }
  557. if (connection_or_nonopen_was_started_here(conn)) {
  558. int as_advertised = 1;
  559. if (memcmp(digest_rcvd, conn->identity_digest, DIGEST_LEN)) {
  560. /* I was aiming for a particular digest. I didn't get it! */
  561. char seen[HEX_DIGEST_LEN+1];
  562. char expected[HEX_DIGEST_LEN+1];
  563. base16_encode(seen, sizeof(seen), digest_rcvd, DIGEST_LEN);
  564. base16_encode(expected, sizeof(expected), conn->identity_digest,
  565. DIGEST_LEN);
  566. log_fn(severity, LD_OR,
  567. "Identity key not as expected for router at %s:%d: wanted %s "
  568. "but got %s",
  569. conn->address, conn->port, expected, seen);
  570. entry_guard_set_status(conn->identity_digest, 0);
  571. control_event_or_conn_status(conn, OR_CONN_EVENT_FAILED);
  572. as_advertised = 0;
  573. }
  574. if (authdir_mode(options)) {
  575. /* We initiated this connection to address:port. Drop all routers
  576. * with the same address:port and a different key or nickname.
  577. */
  578. dirserv_orconn_tls_done(conn->address, conn->port,
  579. digest_rcvd, nickname, as_advertised);
  580. }
  581. if (!as_advertised)
  582. return -1;
  583. }
  584. return 0;
  585. }
  586. /** The tls handshake is finished.
  587. *
  588. * Make sure we are happy with the person we just handshaked with.
  589. *
  590. * If he initiated the connection, make sure he's not already connected,
  591. * then initialize conn from the information in router.
  592. *
  593. * If I'm not a server, set bandwidth to the default OP bandwidth.
  594. *
  595. * If all is successful, call circuit_n_conn_done() to handle events
  596. * that have been pending on the tls handshake completion. Also set the
  597. * directory to be dirty (only matters if I'm an authdirserver).
  598. */
  599. static int
  600. connection_tls_finish_handshake(connection_t *conn)
  601. {
  602. char digest_rcvd[DIGEST_LEN];
  603. int started_here = connection_or_nonopen_was_started_here(conn);
  604. debug(LD_OR,"tls handshake done. verifying.");
  605. if (connection_or_check_valid_handshake(conn, digest_rcvd) < 0)
  606. return -1;
  607. if (!started_here) {
  608. #if 0
  609. connection_t *c;
  610. if ((c=connection_or_get_by_identity_digest(digest_rcvd))) {
  611. debug(LD_OR,"Router '%s' is already connected on fd %d. Dropping fd %d.",
  612. c->nickname, c->s, conn->s);
  613. return -1;
  614. }
  615. #endif
  616. connection_or_init_conn_from_address(conn,conn->addr,conn->port,
  617. digest_rcvd);
  618. }
  619. if (!server_mode(get_options())) { /* If I'm an OP... */
  620. conn->receiver_bucket = conn->bandwidth = DEFAULT_BANDWIDTH_OP;
  621. }
  622. directory_set_dirty();
  623. conn->state = OR_CONN_STATE_OPEN;
  624. control_event_or_conn_status(conn, OR_CONN_EVENT_CONNECTED);
  625. if (started_here) {
  626. rep_hist_note_connect_succeeded(conn->identity_digest, time(NULL));
  627. if (entry_guard_set_status(conn->identity_digest, 1) < 0) {
  628. /* pending circs get closed in circuit_about_to_close_connection() */
  629. return -1;
  630. }
  631. }
  632. connection_watch_events(conn, EV_READ);
  633. circuit_n_conn_done(conn, 1); /* send the pending creates, if any. */
  634. return 0;
  635. }
  636. /** Pack <b>cell</b> into wire-format, and write it onto <b>conn</b>'s
  637. * outbuf.
  638. *
  639. * If it's an OR conn, and an entire TLS record is
  640. * ready, then try to flush the record now.
  641. */
  642. void
  643. connection_or_write_cell_to_buf(const cell_t *cell, connection_t *conn)
  644. {
  645. char networkcell[CELL_NETWORK_SIZE];
  646. char *n = networkcell;
  647. tor_assert(cell);
  648. tor_assert(conn);
  649. tor_assert(connection_speaks_cells(conn));
  650. cell_pack(n, cell);
  651. connection_write_to_buf(n, CELL_NETWORK_SIZE, conn);
  652. #define MIN_TLS_FLUSHLEN 15872
  653. /* openssl tls record size is 16383, this is close. The goal here is to
  654. * push data out as soon as we know there's enough for a tls record, so
  655. * during periods of high load we won't read the entire megabyte from
  656. * input before pushing any data out. It also has the feature of not
  657. * growing huge outbufs unless something is slow. */
  658. if (conn->outbuf_flushlen-CELL_NETWORK_SIZE < MIN_TLS_FLUSHLEN &&
  659. conn->outbuf_flushlen >= MIN_TLS_FLUSHLEN) {
  660. int extra = conn->outbuf_flushlen - MIN_TLS_FLUSHLEN;
  661. conn->outbuf_flushlen = MIN_TLS_FLUSHLEN;
  662. if (connection_handle_write(conn) < 0) {
  663. if (!conn->marked_for_close) {
  664. /* this connection is broken. remove it. */
  665. warn(LD_BUG,
  666. "Bug: unhandled error on write for OR conn (fd %d); removing",
  667. conn->s);
  668. tor_fragile_assert();
  669. conn->has_sent_end = 1; /* don't cry wolf about duplicate close */
  670. /* XXX do we need a close-immediate here, so we don't try to flush? */
  671. connection_mark_for_close(conn);
  672. }
  673. return;
  674. }
  675. if (extra) {
  676. conn->outbuf_flushlen += extra;
  677. connection_start_writing(conn);
  678. }
  679. }
  680. }
  681. /** Process cells from <b>conn</b>'s inbuf.
  682. *
  683. * Loop: while inbuf contains a cell, pull it off the inbuf, unpack it,
  684. * and hand it to command_process_cell().
  685. *
  686. * Always return 0.
  687. */
  688. static int
  689. connection_or_process_cells_from_inbuf(connection_t *conn)
  690. {
  691. char buf[CELL_NETWORK_SIZE];
  692. cell_t cell;
  693. loop:
  694. debug(LD_OR,"%d: starting, inbuf_datalen %d (%d pending in tls object).",
  695. conn->s,(int)buf_datalen(conn->inbuf),
  696. tor_tls_get_pending_bytes(conn->tls));
  697. if (buf_datalen(conn->inbuf) < CELL_NETWORK_SIZE) /* whole response
  698. available? */
  699. return 0; /* not yet */
  700. connection_fetch_from_buf(buf, CELL_NETWORK_SIZE, conn);
  701. /* retrieve cell info from buf (create the host-order struct from the
  702. * network-order string) */
  703. cell_unpack(&cell, buf);
  704. command_process_cell(&cell, conn);
  705. goto loop; /* process the remainder of the buffer */
  706. }
  707. /** Write a destroy cell with circ ID <b>circ_id</b> and reason <b>reason</b>
  708. * onto OR connection <b>conn</b>. Don't perform range-checking on reason:
  709. * we may want to propagate reasons from other cells.
  710. *
  711. * Return 0.
  712. */
  713. int
  714. connection_or_send_destroy(uint16_t circ_id, connection_t *conn, int reason)
  715. {
  716. cell_t cell;
  717. tor_assert(conn);
  718. tor_assert(connection_speaks_cells(conn));
  719. memset(&cell, 0, sizeof(cell_t));
  720. cell.circ_id = circ_id;
  721. cell.command = CELL_DESTROY;
  722. cell.payload[0] = (uint8_t) reason;
  723. debug(LD_OR,"Sending destroy (circID %d).", circ_id);
  724. connection_or_write_cell_to_buf(&cell, conn);
  725. return 0;
  726. }