connection_edge.c
上传用户:awang829
上传日期:2019-07-14
资源大小:2356k
文件大小:105k
源码类别:

网络

开发平台:

Unix_Linux

  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-2009, The Tor Project, Inc. */
  5. /* See LICENSE for licensing information */
  6. /**
  7.  * file connection_edge.c
  8.  * brief Handle edge streams.
  9.  **/
  10. #include "or.h"
  11. #ifdef HAVE_LINUX_TYPES_H
  12. #include <linux/types.h>
  13. #endif
  14. #ifdef HAVE_LINUX_NETFILTER_IPV4_H
  15. #include <linux/netfilter_ipv4.h>
  16. #define TRANS_NETFILTER
  17. #endif
  18. #if defined(HAVE_NET_IF_H) && defined(HAVE_NET_PFVAR_H)
  19. #include <net/if.h>
  20. #include <net/pfvar.h>
  21. #define TRANS_PF
  22. #endif
  23. #define SOCKS4_GRANTED          90
  24. #define SOCKS4_REJECT           91
  25. static int connection_ap_handshake_process_socks(edge_connection_t *conn);
  26. static int connection_ap_process_natd(edge_connection_t *conn);
  27. static int connection_exit_connect_dir(edge_connection_t *exitconn);
  28. static int address_is_in_virtual_range(const char *addr);
  29. static int consider_plaintext_ports(edge_connection_t *conn, uint16_t port);
  30. static void clear_trackexithost_mappings(const char *exitname);
  31. /** An AP stream has failed/finished. If it hasn't already sent back
  32.  * a socks reply, send one now (based on endreason). Also set
  33.  * has_sent_end to 1, and mark the conn.
  34.  */
  35. void
  36. _connection_mark_unattached_ap(edge_connection_t *conn, int endreason,
  37.                                int line, const char *file)
  38. {
  39.   tor_assert(conn->_base.type == CONN_TYPE_AP);
  40.   conn->edge_has_sent_end = 1; /* no circ yet */
  41.   if (conn->_base.marked_for_close) {
  42.     /* This call will warn as appropriate. */
  43.     _connection_mark_for_close(TO_CONN(conn), line, file);
  44.     return;
  45.   }
  46.   if (!conn->socks_request->has_finished) {
  47.     if (endreason & END_STREAM_REASON_FLAG_ALREADY_SOCKS_REPLIED)
  48.       log_warn(LD_BUG,
  49.                "stream (marked at %s:%d) sending two socks replies?",
  50.                file, line);
  51.     if (SOCKS_COMMAND_IS_CONNECT(conn->socks_request->command))
  52.       connection_ap_handshake_socks_reply(conn, NULL, 0, endreason);
  53.     else if (SOCKS_COMMAND_IS_RESOLVE(conn->socks_request->command))
  54.       connection_ap_handshake_socks_resolved(conn,
  55.                                              RESOLVED_TYPE_ERROR_TRANSIENT,
  56.                                              0, NULL, -1, -1);
  57.     else /* unknown or no handshake at all. send no response. */
  58.       conn->socks_request->has_finished = 1;
  59.   }
  60.   _connection_mark_for_close(TO_CONN(conn), line, file);
  61.   conn->_base.hold_open_until_flushed = 1;
  62.   conn->end_reason = endreason;
  63. }
  64. /** There was an EOF. Send an end and mark the connection for close.
  65.  */
  66. int
  67. connection_edge_reached_eof(edge_connection_t *conn)
  68. {
  69.   if (buf_datalen(conn->_base.inbuf) &&
  70.       connection_state_is_open(TO_CONN(conn))) {
  71.     /* it still has stuff to process. don't let it die yet. */
  72.     return 0;
  73.   }
  74.   log_info(LD_EDGE,"conn (fd %d) reached eof. Closing.", conn->_base.s);
  75.   if (!conn->_base.marked_for_close) {
  76.     /* only mark it if not already marked. it's possible to
  77.      * get the 'end' right around when the client hangs up on us. */
  78.     connection_edge_end(conn, END_STREAM_REASON_DONE);
  79.     if (conn->socks_request) /* eof, so don't send a socks reply back */
  80.       conn->socks_request->has_finished = 1;
  81.     connection_mark_for_close(TO_CONN(conn));
  82.   }
  83.   return 0;
  84. }
  85. /** Handle new bytes on conn->inbuf based on state:
  86.  *   - If it's waiting for socks info, try to read another step of the
  87.  *     socks handshake out of conn->inbuf.
  88.  *   - If it's waiting for the original destination, fetch it.
  89.  *   - If it's open, then package more relay cells from the stream.
  90.  *   - Else, leave the bytes on inbuf alone for now.
  91.  *
  92.  * Mark and return -1 if there was an unexpected error with the conn,
  93.  * else return 0.
  94.  */
  95. int
  96. connection_edge_process_inbuf(edge_connection_t *conn, int package_partial)
  97. {
  98.   tor_assert(conn);
  99.   switch (conn->_base.state) {
  100.     case AP_CONN_STATE_SOCKS_WAIT:
  101.       if (connection_ap_handshake_process_socks(conn) < 0) {
  102.         /* already marked */
  103.         return -1;
  104.       }
  105.       return 0;
  106.     case AP_CONN_STATE_NATD_WAIT:
  107.       if (connection_ap_process_natd(conn) < 0) {
  108.         /* already marked */
  109.         return -1;
  110.       }
  111.       return 0;
  112.     case AP_CONN_STATE_OPEN:
  113.     case EXIT_CONN_STATE_OPEN:
  114.       if (connection_edge_package_raw_inbuf(conn, package_partial) < 0) {
  115.         /* (We already sent an end cell if possible) */
  116.         connection_mark_for_close(TO_CONN(conn));
  117.         return -1;
  118.       }
  119.       return 0;
  120.     case EXIT_CONN_STATE_CONNECTING:
  121.     case AP_CONN_STATE_RENDDESC_WAIT:
  122.     case AP_CONN_STATE_CIRCUIT_WAIT:
  123.     case AP_CONN_STATE_CONNECT_WAIT:
  124.     case AP_CONN_STATE_RESOLVE_WAIT:
  125.     case AP_CONN_STATE_CONTROLLER_WAIT:
  126.       log_info(LD_EDGE,
  127.                "data from edge while in '%s' state. Leaving it on buffer.",
  128.                conn_state_to_string(conn->_base.type, conn->_base.state));
  129.       return 0;
  130.   }
  131.   log_warn(LD_BUG,"Got unexpected state %d. Closing.",conn->_base.state);
  132.   tor_fragile_assert();
  133.   connection_edge_end(conn, END_STREAM_REASON_INTERNAL);
  134.   connection_mark_for_close(TO_CONN(conn));
  135.   return -1;
  136. }
  137. /** This edge needs to be closed, because its circuit has closed.
  138.  * Mark it for close and return 0.
  139.  */
  140. int
  141. connection_edge_destroy(circid_t circ_id, edge_connection_t *conn)
  142. {
  143.   if (!conn->_base.marked_for_close) {
  144.     log_info(LD_EDGE,
  145.              "CircID %d: At an edge. Marking connection for close.", circ_id);
  146.     if (conn->_base.type == CONN_TYPE_AP) {
  147.       connection_mark_unattached_ap(conn, END_STREAM_REASON_DESTROY);
  148.       control_event_stream_bandwidth(conn);
  149.       control_event_stream_status(conn, STREAM_EVENT_CLOSED,
  150.                                   END_STREAM_REASON_DESTROY);
  151.       conn->end_reason |= END_STREAM_REASON_FLAG_ALREADY_SENT_CLOSED;
  152.     } else {
  153.       /* closing the circuit, nothing to send an END to */
  154.       conn->edge_has_sent_end = 1;
  155.       conn->end_reason = END_STREAM_REASON_DESTROY;
  156.       conn->end_reason |= END_STREAM_REASON_FLAG_ALREADY_SENT_CLOSED;
  157.       connection_mark_for_close(TO_CONN(conn));
  158.       conn->_base.hold_open_until_flushed = 1;
  159.     }
  160.   }
  161.   conn->cpath_layer = NULL;
  162.   conn->on_circuit = NULL;
  163.   return 0;
  164. }
  165. /** Send a raw end cell to the stream with ID <b>stream_id</b> out over the
  166.  * <b>circ</b> towards the hop identified with <b>cpath_layer</b>. If this
  167.  * is not a client connection, set the relay end cell's reason for closing
  168.  * as <b>reason</b> */
  169. static int
  170. relay_send_end_cell_from_edge(streamid_t stream_id, circuit_t *circ,
  171.                               uint8_t reason, crypt_path_t *cpath_layer)
  172. {
  173.   char payload[1];
  174.   if (CIRCUIT_PURPOSE_IS_CLIENT(circ->purpose)) {
  175.     /* Never send the server an informative reason code; it doesn't need to
  176.      * know why the client stream is failing. */
  177.     reason = END_STREAM_REASON_MISC;
  178.   }
  179.   payload[0] = (char) reason;
  180.   return relay_send_command_from_edge(stream_id, circ, RELAY_COMMAND_END,
  181.                                       payload, 1, cpath_layer);
  182. }
  183. /** Send a relay end cell from stream <b>conn</b> down conn's circuit, and
  184.  * remember that we've done so.  If this is not a client connection, set the
  185.  * relay end cell's reason for closing as <b>reason</b>.
  186.  *
  187.  * Return -1 if this function has already been called on this conn,
  188.  * else return 0.
  189.  */
  190. int
  191. connection_edge_end(edge_connection_t *conn, uint8_t reason)
  192. {
  193.   char payload[RELAY_PAYLOAD_SIZE];
  194.   size_t payload_len=1;
  195.   circuit_t *circ;
  196.   uint8_t control_reason = reason;
  197.   if (conn->edge_has_sent_end) {
  198.     log_warn(LD_BUG,"(Harmless.) Calling connection_edge_end (reason %d) "
  199.              "on an already ended stream?", reason);
  200.     tor_fragile_assert();
  201.     return -1;
  202.   }
  203.   if (conn->_base.marked_for_close) {
  204.     log_warn(LD_BUG,
  205.              "called on conn that's already marked for close at %s:%d.",
  206.              conn->_base.marked_for_close_file, conn->_base.marked_for_close);
  207.     return 0;
  208.   }
  209.   circ = circuit_get_by_edge_conn(conn);
  210.   if (circ && CIRCUIT_PURPOSE_IS_CLIENT(circ->purpose)) {
  211.     /* If this is a client circuit, don't send the server an informative
  212.      * reason code; it doesn't need to know why the client stream is
  213.      * failing. */
  214.     reason = END_STREAM_REASON_MISC;
  215.   }
  216.   payload[0] = (char)reason;
  217.   if (reason == END_STREAM_REASON_EXITPOLICY &&
  218.       !connection_edge_is_rendezvous_stream(conn)) {
  219.     int addrlen;
  220.     if (tor_addr_family(&conn->_base.addr) == AF_INET) {
  221.       set_uint32(payload+1, tor_addr_to_ipv4n(&conn->_base.addr));
  222.       addrlen = 4;
  223.     } else {
  224.       memcpy(payload+1, tor_addr_to_in6_addr8(&conn->_base.addr), 16);
  225.       addrlen = 16;
  226.     }
  227.     set_uint32(payload+1+addrlen, htonl(dns_clip_ttl(conn->address_ttl)));
  228.     payload_len += 4+addrlen;
  229.   }
  230.   if (circ && !circ->marked_for_close) {
  231.     log_debug(LD_EDGE,"Sending end on conn (fd %d).",conn->_base.s);
  232.     connection_edge_send_command(conn, RELAY_COMMAND_END,
  233.                                  payload, payload_len);
  234.   } else {
  235.     log_debug(LD_EDGE,"No circ to send end on conn (fd %d).",
  236.               conn->_base.s);
  237.   }
  238.   conn->edge_has_sent_end = 1;
  239.   conn->end_reason = control_reason;
  240.   return 0;
  241. }
  242. /** An error has just occurred on an operation on an edge connection
  243.  * <b>conn</b>.  Extract the errno; convert it to an end reason, and send an
  244.  * appropriate relay end cell to the other end of the connection's circuit.
  245.  **/
  246. int
  247. connection_edge_end_errno(edge_connection_t *conn)
  248. {
  249.   uint8_t reason;
  250.   tor_assert(conn);
  251.   reason = errno_to_stream_end_reason(tor_socket_errno(conn->_base.s));
  252.   return connection_edge_end(conn, reason);
  253. }
  254. /** Connection <b>conn</b> has finished writing and has no bytes left on
  255.  * its outbuf.
  256.  *
  257.  * If it's in state 'open', stop writing, consider responding with a
  258.  * sendme, and return.
  259.  * Otherwise, stop writing and return.
  260.  *
  261.  * If <b>conn</b> is broken, mark it for close and return -1, else
  262.  * return 0.
  263.  */
  264. int
  265. connection_edge_finished_flushing(edge_connection_t *conn)
  266. {
  267.   tor_assert(conn);
  268.   switch (conn->_base.state) {
  269.     case AP_CONN_STATE_OPEN:
  270.     case EXIT_CONN_STATE_OPEN:
  271.       connection_stop_writing(TO_CONN(conn));
  272.       connection_edge_consider_sending_sendme(conn);
  273.       return 0;
  274.     case AP_CONN_STATE_SOCKS_WAIT:
  275.     case AP_CONN_STATE_NATD_WAIT:
  276.     case AP_CONN_STATE_RENDDESC_WAIT:
  277.     case AP_CONN_STATE_CIRCUIT_WAIT:
  278.     case AP_CONN_STATE_CONNECT_WAIT:
  279.     case AP_CONN_STATE_CONTROLLER_WAIT:
  280.       connection_stop_writing(TO_CONN(conn));
  281.       return 0;
  282.     default:
  283.       log_warn(LD_BUG, "Called in unexpected state %d.",conn->_base.state);
  284.       tor_fragile_assert();
  285.       return -1;
  286.   }
  287.   return 0;
  288. }
  289. /** Connected handler for exit connections: start writing pending
  290.  * data, deliver 'CONNECTED' relay cells as appropriate, and check
  291.  * any pending data that may have been received. */
  292. int
  293. connection_edge_finished_connecting(edge_connection_t *edge_conn)
  294. {
  295.   connection_t *conn;
  296.   tor_assert(edge_conn);
  297.   tor_assert(edge_conn->_base.type == CONN_TYPE_EXIT);
  298.   conn = TO_CONN(edge_conn);
  299.   tor_assert(conn->state == EXIT_CONN_STATE_CONNECTING);
  300.   log_info(LD_EXIT,"Exit connection to %s:%u (%s) established.",
  301.            escaped_safe_str(conn->address),conn->port,
  302.            safe_str(fmt_addr(&conn->addr)));
  303.   conn->state = EXIT_CONN_STATE_OPEN;
  304.   connection_watch_events(conn, EV_READ); /* stop writing, continue reading */
  305.   if (connection_wants_to_flush(conn)) /* in case there are any queued relay
  306.                                         * cells */
  307.     connection_start_writing(conn);
  308.   /* deliver a 'connected' relay cell back through the circuit. */
  309.   if (connection_edge_is_rendezvous_stream(edge_conn)) {
  310.     if (connection_edge_send_command(edge_conn,
  311.                                      RELAY_COMMAND_CONNECTED, NULL, 0) < 0)
  312.       return 0; /* circuit is closed, don't continue */
  313.   } else {
  314.     char connected_payload[20];
  315.     int connected_payload_len;
  316.     if (tor_addr_family(&conn->addr) == AF_INET) {
  317.       set_uint32(connected_payload, tor_addr_to_ipv4n(&conn->addr));
  318.       set_uint32(connected_payload+4,
  319.                  htonl(dns_clip_ttl(edge_conn->address_ttl)));
  320.       connected_payload_len = 8;
  321.     } else {
  322.       memcpy(connected_payload, tor_addr_to_in6_addr8(&conn->addr), 16);
  323.       set_uint32(connected_payload+16,
  324.                  htonl(dns_clip_ttl(edge_conn->address_ttl)));
  325.       connected_payload_len = 20;
  326.     }
  327.     if (connection_edge_send_command(edge_conn,
  328.                                  RELAY_COMMAND_CONNECTED,
  329.                                  connected_payload, connected_payload_len) < 0)
  330.       return 0; /* circuit is closed, don't continue */
  331.   }
  332.   tor_assert(edge_conn->package_window > 0);
  333.   /* in case the server has written anything */
  334.   return connection_edge_process_inbuf(edge_conn, 1);
  335. }
  336. /** Define a schedule for how long to wait between retrying
  337.  * application connections. Rather than waiting a fixed amount of
  338.  * time between each retry, we wait 10 seconds each for the first
  339.  * two tries, and 15 seconds for each retry after
  340.  * that. Hopefully this will improve the expected user experience. */
  341. static int
  342. compute_retry_timeout(edge_connection_t *conn)
  343. {
  344.   if (conn->num_socks_retries < 2) /* try 0 and try 1 */
  345.     return 10;
  346.   return 15;
  347. }
  348. /** Find all general-purpose AP streams waiting for a response that sent their
  349.  * begin/resolve cell >=15 seconds ago. Detach from their current circuit, and
  350.  * mark their current circuit as unsuitable for new streams. Then call
  351.  * connection_ap_handshake_attach_circuit() to attach to a new circuit (if
  352.  * available) or launch a new one.
  353.  *
  354.  * For rendezvous streams, simply give up after SocksTimeout seconds (with no
  355.  * retry attempt).
  356.  */
  357. void
  358. connection_ap_expire_beginning(void)
  359. {
  360.   edge_connection_t *conn;
  361.   circuit_t *circ;
  362.   time_t now = time(NULL);
  363.   or_options_t *options = get_options();
  364.   int severity;
  365.   int cutoff;
  366.   int seconds_idle, seconds_since_born;
  367.   smartlist_t *conns = get_connection_array();
  368.   SMARTLIST_FOREACH_BEGIN(conns, connection_t *, c) {
  369.     if (c->type != CONN_TYPE_AP || c->marked_for_close)
  370.       continue;
  371.     conn = TO_EDGE_CONN(c);
  372.     /* if it's an internal linked connection, don't yell its status. */
  373.     severity = (tor_addr_is_null(&conn->_base.addr) && !conn->_base.port)
  374.       ? LOG_INFO : LOG_NOTICE;
  375.     seconds_idle = (int)( now - conn->_base.timestamp_lastread );
  376.     seconds_since_born = (int)( now - conn->_base.timestamp_created );
  377.     if (conn->_base.state == AP_CONN_STATE_OPEN)
  378.       continue;
  379.     /* We already consider SocksTimeout in
  380.      * connection_ap_handshake_attach_circuit(), but we need to consider
  381.      * it here too because controllers that put streams in controller_wait
  382.      * state never ask Tor to attach the circuit. */
  383.     if (AP_CONN_STATE_IS_UNATTACHED(conn->_base.state)) {
  384.       if (seconds_since_born >= options->SocksTimeout) {
  385.         log_fn(severity, LD_APP,
  386.             "Tried for %d seconds to get a connection to %s:%d. "
  387.             "Giving up. (%s)",
  388.             seconds_since_born, safe_str(conn->socks_request->address),
  389.             conn->socks_request->port,
  390.             conn_state_to_string(CONN_TYPE_AP, conn->_base.state));
  391.         connection_mark_unattached_ap(conn, END_STREAM_REASON_TIMEOUT);
  392.       }
  393.       continue;
  394.     }
  395.     /* We're in state connect_wait or resolve_wait now -- waiting for a
  396.      * reply to our relay cell. See if we want to retry/give up. */
  397.     cutoff = compute_retry_timeout(conn);
  398.     if (seconds_idle < cutoff)
  399.       continue;
  400.     circ = circuit_get_by_edge_conn(conn);
  401.     if (!circ) { /* it's vanished? */
  402.       log_info(LD_APP,"Conn is waiting (address %s), but lost its circ.",
  403.                safe_str(conn->socks_request->address));
  404.       connection_mark_unattached_ap(conn, END_STREAM_REASON_TIMEOUT);
  405.       continue;
  406.     }
  407.     if (circ->purpose == CIRCUIT_PURPOSE_C_REND_JOINED) {
  408.       if (seconds_idle >= options->SocksTimeout) {
  409.         log_fn(severity, LD_REND,
  410.                "Rend stream is %d seconds late. Giving up on address"
  411.                " '%s.onion'.",
  412.                seconds_idle,
  413.                safe_str(conn->socks_request->address));
  414.         connection_edge_end(conn, END_STREAM_REASON_TIMEOUT);
  415.         connection_mark_unattached_ap(conn, END_STREAM_REASON_TIMEOUT);
  416.       }
  417.       continue;
  418.     }
  419.     tor_assert(circ->purpose == CIRCUIT_PURPOSE_C_GENERAL);
  420.     log_fn(cutoff < 15 ? LOG_INFO : severity, LD_APP,
  421.            "We tried for %d seconds to connect to '%s' using exit '%s'."
  422.            " Retrying on a new circuit.",
  423.            seconds_idle, safe_str(conn->socks_request->address),
  424.            conn->cpath_layer ?
  425.              conn->cpath_layer->extend_info->nickname : "*unnamed*");
  426.     /* send an end down the circuit */
  427.     connection_edge_end(conn, END_STREAM_REASON_TIMEOUT);
  428.     /* un-mark it as ending, since we're going to reuse it */
  429.     conn->edge_has_sent_end = 0;
  430.     conn->end_reason = 0;
  431.     /* kludge to make us not try this circuit again, yet to allow
  432.      * current streams on it to survive if they can: make it
  433.      * unattractive to use for new streams */
  434.     tor_assert(circ->timestamp_dirty);
  435.     circ->timestamp_dirty -= options->MaxCircuitDirtiness;
  436.     /* give our stream another 'cutoff' seconds to try */
  437.     conn->_base.timestamp_lastread += cutoff;
  438.     if (conn->num_socks_retries < 250) /* avoid overflow */
  439.       conn->num_socks_retries++;
  440.     /* move it back into 'pending' state, and try to attach. */
  441.     if (connection_ap_detach_retriable(conn, TO_ORIGIN_CIRCUIT(circ),
  442.                                        END_STREAM_REASON_TIMEOUT)<0) {
  443.       if (!conn->_base.marked_for_close)
  444.         connection_mark_unattached_ap(conn, END_STREAM_REASON_CANT_ATTACH);
  445.     }
  446.   } SMARTLIST_FOREACH_END(conn);
  447. }
  448. /** Tell any AP streams that are waiting for a new circuit to try again,
  449.  * either attaching to an available circ or launching a new one.
  450.  */
  451. void
  452. connection_ap_attach_pending(void)
  453. {
  454.   edge_connection_t *edge_conn;
  455.   smartlist_t *conns = get_connection_array();
  456.   SMARTLIST_FOREACH(conns, connection_t *, conn,
  457.   {
  458.     if (conn->marked_for_close ||
  459.         conn->type != CONN_TYPE_AP ||
  460.         conn->state != AP_CONN_STATE_CIRCUIT_WAIT)
  461.       continue;
  462.     edge_conn = TO_EDGE_CONN(conn);
  463.     if (connection_ap_handshake_attach_circuit(edge_conn) < 0) {
  464.       if (!edge_conn->_base.marked_for_close)
  465.         connection_mark_unattached_ap(edge_conn,
  466.                                       END_STREAM_REASON_CANT_ATTACH);
  467.     }
  468.   });
  469. }
  470. /** Tell any AP streams that are waiting for a one-hop tunnel to
  471.  * <b>failed_digest</b> that they are going to fail. */
  472. /* XXX022 We should get rid of this function, and instead attach
  473.  * one-hop streams to circ->p_streams so they get marked in
  474.  * circuit_mark_for_close like normal p_streams. */
  475. void
  476. connection_ap_fail_onehop(const char *failed_digest,
  477.                           cpath_build_state_t *build_state)
  478. {
  479.   edge_connection_t *edge_conn;
  480.   char digest[DIGEST_LEN];
  481.   smartlist_t *conns = get_connection_array();
  482.   SMARTLIST_FOREACH_BEGIN(conns, connection_t *, conn) {
  483.     if (conn->marked_for_close ||
  484.         conn->type != CONN_TYPE_AP ||
  485.         conn->state != AP_CONN_STATE_CIRCUIT_WAIT)
  486.       continue;
  487.     edge_conn = TO_EDGE_CONN(conn);
  488.     if (!edge_conn->want_onehop)
  489.       continue;
  490.     if (hexdigest_to_digest(edge_conn->chosen_exit_name, digest) < 0 ||
  491.         memcmp(digest, failed_digest, DIGEST_LEN))
  492.       continue;
  493.     if (tor_digest_is_zero(digest)) {
  494.       /* we don't know the digest; have to compare addr:port */
  495.       tor_addr_t addr;
  496.       if (!build_state || !build_state->chosen_exit ||
  497.           !edge_conn->socks_request || !edge_conn->socks_request->address)
  498.         continue;
  499.       if (tor_addr_from_str(&addr, edge_conn->socks_request->address)<0 ||
  500.           !tor_addr_eq(&build_state->chosen_exit->addr, &addr) ||
  501.           build_state->chosen_exit->port != edge_conn->socks_request->port)
  502.         continue;
  503.     }
  504.     log_info(LD_APP, "Closing one-hop stream to '%s/%s' because the OR conn "
  505.                      "just failed.", edge_conn->chosen_exit_name,
  506.                      edge_conn->socks_request->address);
  507.     connection_mark_unattached_ap(edge_conn, END_STREAM_REASON_TIMEOUT);
  508.   } SMARTLIST_FOREACH_END(conn);
  509. }
  510. /** A circuit failed to finish on its last hop <b>info</b>. If there
  511.  * are any streams waiting with this exit node in mind, but they
  512.  * don't absolutely require it, make them give up on it.
  513.  */
  514. void
  515. circuit_discard_optional_exit_enclaves(extend_info_t *info)
  516. {
  517.   edge_connection_t *edge_conn;
  518.   routerinfo_t *r1, *r2;
  519.   smartlist_t *conns = get_connection_array();
  520.   SMARTLIST_FOREACH_BEGIN(conns, connection_t *, conn) {
  521.     if (conn->marked_for_close ||
  522.         conn->type != CONN_TYPE_AP ||
  523.         conn->state != AP_CONN_STATE_CIRCUIT_WAIT)
  524.       continue;
  525.     edge_conn = TO_EDGE_CONN(conn);
  526.     if (!edge_conn->chosen_exit_optional &&
  527.         !edge_conn->chosen_exit_retries)
  528.       continue;
  529.     r1 = router_get_by_nickname(edge_conn->chosen_exit_name, 0);
  530.     r2 = router_get_by_nickname(info->nickname, 0);
  531.     if (!r1 || !r2 || r1 != r2)
  532.       continue;
  533.     tor_assert(edge_conn->socks_request);
  534.     if (edge_conn->chosen_exit_optional) {
  535.       log_info(LD_APP, "Giving up on enclave exit '%s' for destination %s.",
  536.                safe_str(edge_conn->chosen_exit_name),
  537.                escaped_safe_str(edge_conn->socks_request->address));
  538.       edge_conn->chosen_exit_optional = 0;
  539.       tor_free(edge_conn->chosen_exit_name); /* clears it */
  540.       /* if this port is dangerous, warn or reject it now that we don't
  541.        * think it'll be using an enclave. */
  542.       consider_plaintext_ports(edge_conn, edge_conn->socks_request->port);
  543.     }
  544.     if (edge_conn->chosen_exit_retries) {
  545.       if (--edge_conn->chosen_exit_retries == 0) { /* give up! */
  546.         clear_trackexithost_mappings(edge_conn->chosen_exit_name);
  547.         tor_free(edge_conn->chosen_exit_name); /* clears it */
  548.         /* if this port is dangerous, warn or reject it now that we don't
  549.          * think it'll be using an enclave. */
  550.         consider_plaintext_ports(edge_conn, edge_conn->socks_request->port);
  551.       }
  552.     }
  553.   } SMARTLIST_FOREACH_END(conn);
  554. }
  555. /** The AP connection <b>conn</b> has just failed while attaching or
  556.  * sending a BEGIN or resolving on <b>circ</b>, but another circuit
  557.  * might work. Detach the circuit, and either reattach it, launch a
  558.  * new circuit, tell the controller, or give up as a appropriate.
  559.  *
  560.  * Returns -1 on err, 1 on success, 0 on not-yet-sure.
  561.  */
  562. int
  563. connection_ap_detach_retriable(edge_connection_t *conn, origin_circuit_t *circ,
  564.                                int reason)
  565. {
  566.   control_event_stream_status(conn, STREAM_EVENT_FAILED_RETRIABLE, reason);
  567.   conn->_base.timestamp_lastread = time(NULL);
  568.   if (!get_options()->LeaveStreamsUnattached || conn->use_begindir) {
  569.     /* If we're attaching streams ourself, or if this connection is
  570.      * a tunneled directory connection, then just attach it. */
  571.     conn->_base.state = AP_CONN_STATE_CIRCUIT_WAIT;
  572.     circuit_detach_stream(TO_CIRCUIT(circ),conn);
  573.     return connection_ap_handshake_attach_circuit(conn);
  574.   } else {
  575.     conn->_base.state = AP_CONN_STATE_CONTROLLER_WAIT;
  576.     circuit_detach_stream(TO_CIRCUIT(circ),conn);
  577.     return 0;
  578.   }
  579. }
  580. /** A client-side struct to remember requests to rewrite addresses
  581.  * to new addresses. These structs are stored in the hash table
  582.  * "addressmap" below.
  583.  *
  584.  * There are 5 ways to set an address mapping:
  585.  * - A MapAddress command from the controller [permanent]
  586.  * - An AddressMap directive in the torrc [permanent]
  587.  * - When a TrackHostExits torrc directive is triggered [temporary]
  588.  * - When a DNS resolve succeeds [temporary]
  589.  * - When a DNS resolve fails [temporary]
  590.  *
  591.  * When an addressmap request is made but one is already registered,
  592.  * the new one is replaced only if the currently registered one has
  593.  * no "new_address" (that is, it's in the process of DNS resolve),
  594.  * or if the new one is permanent (expires==0 or 1).
  595.  *
  596.  * (We overload the 'expires' field, using "0" for mappings set via
  597.  * the configuration file, "1" for mappings set from the control
  598.  * interface, and other values for DNS and TrackHostExit mappings that can
  599.  * expire.)
  600.  */
  601. typedef struct {
  602.   char *new_address;
  603.   time_t expires;
  604.   addressmap_entry_source_t source:3;
  605.   short num_resolve_failures;
  606. } addressmap_entry_t;
  607. /** Entry for mapping addresses to which virtual address we mapped them to. */
  608. typedef struct {
  609.   char *ipv4_address;
  610.   char *hostname_address;
  611. } virtaddress_entry_t;
  612. /** A hash table to store client-side address rewrite instructions. */
  613. static strmap_t *addressmap=NULL;
  614. /**
  615.  * Table mapping addresses to which virtual address, if any, we
  616.  * assigned them to.
  617.  *
  618.  * We maintain the following invariant: if [A,B] is in
  619.  * virtaddress_reversemap, then B must be a virtual address, and [A,B]
  620.  * must be in addressmap.  We do not require that the converse hold:
  621.  * if it fails, then we could end up mapping two virtual addresses to
  622.  * the same address, which is no disaster.
  623.  **/
  624. static strmap_t *virtaddress_reversemap=NULL;
  625. /** Initialize addressmap. */
  626. void
  627. addressmap_init(void)
  628. {
  629.   addressmap = strmap_new();
  630.   virtaddress_reversemap = strmap_new();
  631. }
  632. /** Free the memory associated with the addressmap entry <b>_ent</b>. */
  633. static void
  634. addressmap_ent_free(void *_ent)
  635. {
  636.   addressmap_entry_t *ent = _ent;
  637.   tor_free(ent->new_address);
  638.   tor_free(ent);
  639. }
  640. /** Free storage held by a virtaddress_entry_t* entry in <b>ent</b>. */
  641. static void
  642. addressmap_virtaddress_ent_free(void *_ent)
  643. {
  644.   virtaddress_entry_t *ent = _ent;
  645.   tor_free(ent->ipv4_address);
  646.   tor_free(ent->hostname_address);
  647.   tor_free(ent);
  648. }
  649. /** Free storage held by a virtaddress_entry_t* entry in <b>ent</b>. */
  650. static void
  651. addressmap_virtaddress_remove(const char *address, addressmap_entry_t *ent)
  652. {
  653.   if (ent && ent->new_address &&
  654.       address_is_in_virtual_range(ent->new_address)) {
  655.     virtaddress_entry_t *ve =
  656.       strmap_get(virtaddress_reversemap, ent->new_address);
  657.     /*log_fn(LOG_NOTICE,"remove reverse mapping for %s",ent->new_address);*/
  658.     if (ve) {
  659.       if (!strcmp(address, ve->ipv4_address))
  660.         tor_free(ve->ipv4_address);
  661.       if (!strcmp(address, ve->hostname_address))
  662.         tor_free(ve->hostname_address);
  663.       if (!ve->ipv4_address && !ve->hostname_address) {
  664.         tor_free(ve);
  665.         strmap_remove(virtaddress_reversemap, ent->new_address);
  666.       }
  667.     }
  668.   }
  669. }
  670. /** Remove <b>ent</b> (which must be mapped to by <b>address</b>) from the
  671.  * client address maps. */
  672. static void
  673. addressmap_ent_remove(const char *address, addressmap_entry_t *ent)
  674. {
  675.   addressmap_virtaddress_remove(address, ent);
  676.   addressmap_ent_free(ent);
  677. }
  678. /** Unregister all TrackHostExits mappings from any address to
  679.  * *.exitname.exit. */
  680. static void
  681. clear_trackexithost_mappings(const char *exitname)
  682. {
  683.   char *suffix;
  684.   size_t suffix_len;
  685.   if (!addressmap || !exitname)
  686.     return;
  687.   suffix_len = strlen(exitname) + 16;
  688.   suffix = tor_malloc(suffix_len);
  689.   tor_snprintf(suffix, suffix_len, ".%s.exit", exitname);
  690.   tor_strlower(suffix);
  691.   STRMAP_FOREACH_MODIFY(addressmap, address, addressmap_entry_t *, ent) {
  692.     if (ent->source == ADDRMAPSRC_TRACKEXIT && !strcmpend(address, suffix)) {
  693.       addressmap_ent_remove(address, ent);
  694.       MAP_DEL_CURRENT(address);
  695.     }
  696.   } STRMAP_FOREACH_END;
  697.   tor_free(suffix);
  698. }
  699. /** Remove all entries from the addressmap that were set via the
  700.  * configuration file or the command line. */
  701. void
  702. addressmap_clear_configured(void)
  703. {
  704.   addressmap_get_mappings(NULL, 0, 0, 0);
  705. }
  706. /** Remove all entries from the addressmap that are set to expire, ever. */
  707. void
  708. addressmap_clear_transient(void)
  709. {
  710.   addressmap_get_mappings(NULL, 2, TIME_MAX, 0);
  711. }
  712. /** Clean out entries from the addressmap cache that were
  713.  * added long enough ago that they are no longer valid.
  714.  */
  715. void
  716. addressmap_clean(time_t now)
  717. {
  718.   addressmap_get_mappings(NULL, 2, now, 0);
  719. }
  720. /** Free all the elements in the addressmap, and free the addressmap
  721.  * itself. */
  722. void
  723. addressmap_free_all(void)
  724. {
  725.   if (addressmap) {
  726.     strmap_free(addressmap, addressmap_ent_free);
  727.     addressmap = NULL;
  728.   }
  729.   if (virtaddress_reversemap) {
  730.     strmap_free(virtaddress_reversemap, addressmap_virtaddress_ent_free);
  731.     virtaddress_reversemap = NULL;
  732.   }
  733. }
  734. /** Look at address, and rewrite it until it doesn't want any
  735.  * more rewrites; but don't get into an infinite loop.
  736.  * Don't write more than maxlen chars into address.  Return true if the
  737.  * address changed; false otherwise.  Set *<b>expires_out</b> to the
  738.  * expiry time of the result, or to <b>time_max</b> if the result does
  739.  * not expire.
  740.  */
  741. int
  742. addressmap_rewrite(char *address, size_t maxlen, time_t *expires_out)
  743. {
  744.   addressmap_entry_t *ent;
  745.   int rewrites;
  746.   char *cp;
  747.   time_t expires = TIME_MAX;
  748.   for (rewrites = 0; rewrites < 16; rewrites++) {
  749.     ent = strmap_get(addressmap, address);
  750.     if (!ent || !ent->new_address) {
  751.       if (expires_out)
  752.         *expires_out = expires;
  753.       return (rewrites > 0); /* done, no rewrite needed */
  754.     }
  755.     cp = tor_strdup(escaped_safe_str(ent->new_address));
  756.     log_info(LD_APP, "Addressmap: rewriting %s to %s",
  757.              escaped_safe_str(address), cp);
  758.     if (ent->expires > 1 && ent->expires < expires)
  759.       expires = ent->expires;
  760.     tor_free(cp);
  761.     strlcpy(address, ent->new_address, maxlen);
  762.   }
  763.   log_warn(LD_CONFIG,
  764.            "Loop detected: we've rewritten %s 16 times! Using it as-is.",
  765.            escaped_safe_str(address));
  766.   /* it's fine to rewrite a rewrite, but don't loop forever */
  767.   if (expires_out)
  768.     *expires_out = TIME_MAX;
  769.   return 1;
  770. }
  771. /** If we have a cached reverse DNS entry for the address stored in the
  772.  * <b>maxlen</b>-byte buffer <b>address</b> (typically, a dotted quad) then
  773.  * rewrite to the cached value and return 1.  Otherwise return 0.  Set
  774.  * *<b>expires_out</b> to the expiry time of the result, or to <b>time_max</b>
  775.  * if the result does not expire. */
  776. static int
  777. addressmap_rewrite_reverse(char *address, size_t maxlen, time_t *expires_out)
  778. {
  779.   size_t len = maxlen + 16;
  780.   char *s = tor_malloc(len), *cp;
  781.   addressmap_entry_t *ent;
  782.   int r = 0;
  783.   tor_snprintf(s, len, "REVERSE[%s]", address);
  784.   ent = strmap_get(addressmap, s);
  785.   if (ent) {
  786.     cp = tor_strdup(escaped_safe_str(ent->new_address));
  787.     log_info(LD_APP, "Rewrote reverse lookup %s -> %s",
  788.              escaped_safe_str(s), cp);
  789.     tor_free(cp);
  790.     strlcpy(address, ent->new_address, maxlen);
  791.     r = 1;
  792.   }
  793.   if (expires_out)
  794.     *expires_out = (ent && ent->expires > 1) ? ent->expires : TIME_MAX;
  795.   tor_free(s);
  796.   return r;
  797. }
  798. /** Return 1 if <b>address</b> is already registered, else return 0. If address
  799.  * is already registered, and <b>update_expires</b> is non-zero, then update
  800.  * the expiry time on the mapping with update_expires if it is a
  801.  * mapping created by TrackHostExits. */
  802. int
  803. addressmap_have_mapping(const char *address, int update_expiry)
  804. {
  805.   addressmap_entry_t *ent;
  806.   if (!(ent=strmap_get_lc(addressmap, address)))
  807.     return 0;
  808.   if (update_expiry && ent->source==ADDRMAPSRC_TRACKEXIT)
  809.     ent->expires=time(NULL) + update_expiry;
  810.   return 1;
  811. }
  812. /** Register a request to map <b>address</b> to <b>new_address</b>,
  813.  * which will expire on <b>expires</b> (or 0 if never expires from
  814.  * config file, 1 if never expires from controller, 2 if never expires
  815.  * (virtual address mapping) from the controller.)
  816.  *
  817.  * <b>new_address</b> should be a newly dup'ed string, which we'll use or
  818.  * free as appropriate. We will leave address alone.
  819.  *
  820.  * If <b>new_address</b> is NULL, or equal to <b>address</b>, remove
  821.  * any mappings that exist from <b>address</b>.
  822.  */
  823. void
  824. addressmap_register(const char *address, char *new_address, time_t expires,
  825.                     addressmap_entry_source_t source)
  826. {
  827.   addressmap_entry_t *ent;
  828.   ent = strmap_get(addressmap, address);
  829.   if (!new_address || !strcasecmp(address,new_address)) {
  830.     /* Remove the mapping, if any. */
  831.     tor_free(new_address);
  832.     if (ent) {
  833.       addressmap_ent_remove(address,ent);
  834.       strmap_remove(addressmap, address);
  835.     }
  836.     return;
  837.   }
  838.   if (!ent) { /* make a new one and register it */
  839.     ent = tor_malloc_zero(sizeof(addressmap_entry_t));
  840.     strmap_set(addressmap, address, ent);
  841.   } else if (ent->new_address) { /* we need to clean up the old mapping. */
  842.     if (expires > 1) {
  843.       log_info(LD_APP,"Temporary addressmap ('%s' to '%s') not performed, "
  844.                "since it's already mapped to '%s'",
  845.       safe_str(address), safe_str(new_address), safe_str(ent->new_address));
  846.       tor_free(new_address);
  847.       return;
  848.     }
  849.     if (address_is_in_virtual_range(ent->new_address) &&
  850.         expires != 2) {
  851.       /* XXX This isn't the perfect test; we want to avoid removing
  852.        * mappings set from the control interface _as virtual mapping */
  853.       addressmap_virtaddress_remove(address, ent);
  854.     }
  855.     tor_free(ent->new_address);
  856.   } /* else { we have an in-progress resolve with no mapping. } */
  857.   ent->new_address = new_address;
  858.   ent->expires = expires==2 ? 1 : expires;
  859.   ent->num_resolve_failures = 0;
  860.   ent->source = source;
  861.   log_info(LD_CONFIG, "Addressmap: (re)mapped '%s' to '%s'",
  862.            safe_str(address), safe_str(ent->new_address));
  863.   control_event_address_mapped(address, ent->new_address, expires, NULL);
  864. }
  865. /** An attempt to resolve <b>address</b> failed at some OR.
  866.  * Increment the number of resolve failures we have on record
  867.  * for it, and then return that number.
  868.  */
  869. int
  870. client_dns_incr_failures(const char *address)
  871. {
  872.   addressmap_entry_t *ent = strmap_get(addressmap, address);
  873.   if (!ent) {
  874.     ent = tor_malloc_zero(sizeof(addressmap_entry_t));
  875.     ent->expires = time(NULL) + MAX_DNS_ENTRY_AGE;
  876.     strmap_set(addressmap,address,ent);
  877.   }
  878.   if (ent->num_resolve_failures < SHORT_MAX)
  879.     ++ent->num_resolve_failures; /* don't overflow */
  880.   log_info(LD_APP, "Address %s now has %d resolve failures.",
  881.            safe_str(address), ent->num_resolve_failures);
  882.   return ent->num_resolve_failures;
  883. }
  884. /** If <b>address</b> is in the client DNS addressmap, reset
  885.  * the number of resolve failures we have on record for it.
  886.  * This is used when we fail a stream because it won't resolve:
  887.  * otherwise future attempts on that address will only try once.
  888.  */
  889. void
  890. client_dns_clear_failures(const char *address)
  891. {
  892.   addressmap_entry_t *ent = strmap_get(addressmap, address);
  893.   if (ent)
  894.     ent->num_resolve_failures = 0;
  895. }
  896. /** Record the fact that <b>address</b> resolved to <b>name</b>.
  897.  * We can now use this in subsequent streams via addressmap_rewrite()
  898.  * so we can more correctly choose an exit that will allow <b>address</b>.
  899.  *
  900.  * If <b>exitname</b> is defined, then append the addresses with
  901.  * ".exitname.exit" before registering the mapping.
  902.  *
  903.  * If <b>ttl</b> is nonnegative, the mapping will be valid for
  904.  * <b>ttl</b>seconds; otherwise, we use the default.
  905.  */
  906. static void
  907. client_dns_set_addressmap_impl(const char *address, const char *name,
  908.                                const char *exitname,
  909.                                int ttl)
  910. {
  911.   /* <address>.<hex or nickname>.exit  or just  <address> */
  912.   char extendedaddress[MAX_SOCKS_ADDR_LEN+MAX_VERBOSE_NICKNAME_LEN+10];
  913.   /* 123.123.123.123.<hex or nickname>.exit  or just  123.123.123.123 */
  914.   char extendedval[INET_NTOA_BUF_LEN+MAX_VERBOSE_NICKNAME_LEN+10];
  915.   tor_assert(address);
  916.   tor_assert(name);
  917.   if (ttl<0)
  918.     ttl = DEFAULT_DNS_TTL;
  919.   else
  920.     ttl = dns_clip_ttl(ttl);
  921.   if (exitname) {
  922.     /* XXXX fails to ever get attempts to get an exit address of
  923.      * google.com.digest[=~]nickname.exit; we need a syntax for this that
  924.      * won't make strict RFC952-compliant applications (like us) barf. */
  925.     tor_snprintf(extendedaddress, sizeof(extendedaddress),
  926.                  "%s.%s.exit", address, exitname);
  927.     tor_snprintf(extendedval, sizeof(extendedval),
  928.                  "%s.%s.exit", name, exitname);
  929.   } else {
  930.     tor_snprintf(extendedaddress, sizeof(extendedaddress),
  931.                  "%s", address);
  932.     tor_snprintf(extendedval, sizeof(extendedval),
  933.                  "%s", name);
  934.   }
  935.   addressmap_register(extendedaddress, tor_strdup(extendedval),
  936.                       time(NULL) + ttl, ADDRMAPSRC_DNS);
  937. }
  938. /** Record the fact that <b>address</b> resolved to <b>val</b>.
  939.  * We can now use this in subsequent streams via addressmap_rewrite()
  940.  * so we can more correctly choose an exit that will allow <b>address</b>.
  941.  *
  942.  * If <b>exitname</b> is defined, then append the addresses with
  943.  * ".exitname.exit" before registering the mapping.
  944.  *
  945.  * If <b>ttl</b> is nonnegative, the mapping will be valid for
  946.  * <b>ttl</b>seconds; otherwise, we use the default.
  947.  */
  948. void
  949. client_dns_set_addressmap(const char *address, uint32_t val,
  950.                           const char *exitname,
  951.                           int ttl)
  952. {
  953.   struct in_addr in;
  954.   char valbuf[INET_NTOA_BUF_LEN];
  955.   tor_assert(address);
  956.   if (tor_inet_aton(address, &in))
  957.     return; /* If address was an IP address already, don't add a mapping. */
  958.   in.s_addr = htonl(val);
  959.   tor_inet_ntoa(&in,valbuf,sizeof(valbuf));
  960.   client_dns_set_addressmap_impl(address, valbuf, exitname, ttl);
  961. }
  962. /** Add a cache entry noting that <b>address</b> (ordinarily a dotted quad)
  963.  * resolved via a RESOLVE_PTR request to the hostname <b>v</b>.
  964.  *
  965.  * If <b>exitname</b> is defined, then append the addresses with
  966.  * ".exitname.exit" before registering the mapping.
  967.  *
  968.  * If <b>ttl</b> is nonnegative, the mapping will be valid for
  969.  * <b>ttl</b>seconds; otherwise, we use the default.
  970.  */
  971. static void
  972. client_dns_set_reverse_addressmap(const char *address, const char *v,
  973.                                   const char *exitname,
  974.                                   int ttl)
  975. {
  976.   size_t len = strlen(address) + 16;
  977.   char *s = tor_malloc(len);
  978.   tor_snprintf(s, len, "REVERSE[%s]", address);
  979.   client_dns_set_addressmap_impl(s, v, exitname, ttl);
  980.   tor_free(s);
  981. }
  982. /* By default, we hand out 127.192.0.1 through 127.254.254.254.
  983.  * These addresses should map to localhost, so even if the
  984.  * application accidentally tried to connect to them directly (not
  985.  * via Tor), it wouldn't get too far astray.
  986.  *
  987.  * These options are configured by parse_virtual_addr_network().
  988.  */
  989. /** Which network should we use for virtual IPv4 addresses?  Only the first
  990.  * bits of this value are fixed. */
  991. static uint32_t virtual_addr_network = 0x7fc00000u;
  992. /** How many bits of <b>virtual_addr_network</b> are fixed? */
  993. static maskbits_t virtual_addr_netmask_bits = 10;
  994. /** What's the next virtual address we will hand out? */
  995. static uint32_t next_virtual_addr    = 0x7fc00000u;
  996. /** Read a netmask of the form 127.192.0.0/10 from "val", and check whether
  997.  * it's a valid set of virtual addresses to hand out in response to MAPADDRESS
  998.  * requests.  Return 0 on success; set *msg (if provided) to a newly allocated
  999.  * string and return -1 on failure.  If validate_only is false, sets the
  1000.  * actual virtual address range to the parsed value. */
  1001. int
  1002. parse_virtual_addr_network(const char *val, int validate_only,
  1003.                            char **msg)
  1004. {
  1005.   uint32_t addr;
  1006.   uint16_t port_min, port_max;
  1007.   maskbits_t bits;
  1008.   if (parse_addr_and_port_range(val, &addr, &bits, &port_min, &port_max)) {
  1009.     if (msg) *msg = tor_strdup("Error parsing VirtualAddressNetwork");
  1010.     return -1;
  1011.   }
  1012.   if (port_min != 1 || port_max != 65535) {
  1013.     if (msg) *msg = tor_strdup("Can't specify ports on VirtualAddressNetwork");
  1014.     return -1;
  1015.   }
  1016.   if (bits > 16) {
  1017.     if (msg) *msg = tor_strdup("VirtualAddressNetwork expects a /16 "
  1018.                                "network or larger");
  1019.     return -1;
  1020.   }
  1021.   if (validate_only)
  1022.     return 0;
  1023.   virtual_addr_network = (uint32_t)( addr & (0xfffffffful << (32-bits)) );
  1024.   virtual_addr_netmask_bits = bits;
  1025.   if (addr_mask_cmp_bits(next_virtual_addr, addr, bits))
  1026.     next_virtual_addr = addr;
  1027.   return 0;
  1028. }
  1029. /**
  1030.  * Return true iff <b>addr</b> is likely to have been returned by
  1031.  * client_dns_get_unused_address.
  1032.  **/
  1033. static int
  1034. address_is_in_virtual_range(const char *address)
  1035. {
  1036.   struct in_addr in;
  1037.   tor_assert(address);
  1038.   if (!strcasecmpend(address, ".virtual")) {
  1039.     return 1;
  1040.   } else if (tor_inet_aton(address, &in)) {
  1041.     uint32_t addr = ntohl(in.s_addr);
  1042.     if (!addr_mask_cmp_bits(addr, virtual_addr_network,
  1043.                             virtual_addr_netmask_bits))
  1044.       return 1;
  1045.   }
  1046.   return 0;
  1047. }
  1048. /** Return a newly allocated string holding an address of <b>type</b>
  1049.  * (one of RESOLVED_TYPE_{IPV4|HOSTNAME}) that has not yet been mapped,
  1050.  * and that is very unlikely to be the address of any real host.
  1051.  */
  1052. static char *
  1053. addressmap_get_virtual_address(int type)
  1054. {
  1055.   char buf[64];
  1056.   struct in_addr in;
  1057.   tor_assert(addressmap);
  1058.   if (type == RESOLVED_TYPE_HOSTNAME) {
  1059.     char rand[10];
  1060.     do {
  1061.       crypto_rand(rand, sizeof(rand));
  1062.       base32_encode(buf,sizeof(buf),rand,sizeof(rand));
  1063.       strlcat(buf, ".virtual", sizeof(buf));
  1064.     } while (strmap_get(addressmap, buf));
  1065.     return tor_strdup(buf);
  1066.   } else if (type == RESOLVED_TYPE_IPV4) {
  1067.     // This is an imperfect estimate of how many addresses are available, but
  1068.     // that's ok.
  1069.     uint32_t available = 1u << (32-virtual_addr_netmask_bits);
  1070.     while (available) {
  1071.       /* Don't hand out any .0 or .255 address. */
  1072.       while ((next_virtual_addr & 0xff) == 0 ||
  1073.              (next_virtual_addr & 0xff) == 0xff) {
  1074.         ++next_virtual_addr;
  1075.       }
  1076.       in.s_addr = htonl(next_virtual_addr);
  1077.       tor_inet_ntoa(&in, buf, sizeof(buf));
  1078.       if (!strmap_get(addressmap, buf)) {
  1079.         ++next_virtual_addr;
  1080.         break;
  1081.       }
  1082.       ++next_virtual_addr;
  1083.       --available;
  1084.       log_info(LD_CONFIG, "%d addrs available", (int)available);
  1085.       if (! --available) {
  1086.         log_warn(LD_CONFIG, "Ran out of virtual addresses!");
  1087.         return NULL;
  1088.       }
  1089.       if (addr_mask_cmp_bits(next_virtual_addr, virtual_addr_network,
  1090.                              virtual_addr_netmask_bits))
  1091.         next_virtual_addr = virtual_addr_network;
  1092.     }
  1093.     return tor_strdup(buf);
  1094.   } else {
  1095.     log_warn(LD_BUG, "Called with unsupported address type (%d)", type);
  1096.     return NULL;
  1097.   }
  1098. }
  1099. /** A controller has requested that we map some address of type
  1100.  * <b>type</b> to the address <b>new_address</b>.  Choose an address
  1101.  * that is unlikely to be used, and map it, and return it in a newly
  1102.  * allocated string.  If another address of the same type is already
  1103.  * mapped to <b>new_address</b>, try to return a copy of that address.
  1104.  *
  1105.  * The string in <b>new_address</b> may be freed, or inserted into a map
  1106.  * as appropriate.
  1107.  **/
  1108. const char *
  1109. addressmap_register_virtual_address(int type, char *new_address)
  1110. {
  1111.   char **addrp;
  1112.   virtaddress_entry_t *vent;
  1113.   tor_assert(new_address);
  1114.   tor_assert(addressmap);
  1115.   tor_assert(virtaddress_reversemap);
  1116.   vent = strmap_get(virtaddress_reversemap, new_address);
  1117.   if (!vent) {
  1118.     vent = tor_malloc_zero(sizeof(virtaddress_entry_t));
  1119.     strmap_set(virtaddress_reversemap, new_address, vent);
  1120.   }
  1121.   addrp = (type == RESOLVED_TYPE_IPV4) ?
  1122.     &vent->ipv4_address : &vent->hostname_address;
  1123.   if (*addrp) {
  1124.     addressmap_entry_t *ent = strmap_get(addressmap, *addrp);
  1125.     if (ent && ent->new_address &&
  1126.         !strcasecmp(new_address, ent->new_address)) {
  1127.       tor_free(new_address);
  1128.       return tor_strdup(*addrp);
  1129.     } else
  1130.       log_warn(LD_BUG,
  1131.                "Internal confusion: I thought that '%s' was mapped to by "
  1132.                "'%s', but '%s' really maps to '%s'. This is a harmless bug.",
  1133.                safe_str(new_address), safe_str(*addrp), safe_str(*addrp),
  1134.                ent?safe_str(ent->new_address):"(nothing)");
  1135.   }
  1136.   tor_free(*addrp);
  1137.   *addrp = addressmap_get_virtual_address(type);
  1138.   log_info(LD_APP, "Registering map from %s to %s", *addrp, new_address);
  1139.   addressmap_register(*addrp, new_address, 2, ADDRMAPSRC_CONTROLLER);
  1140. #if 0
  1141.   {
  1142.     /* Try to catch possible bugs */
  1143.     addressmap_entry_t *ent;
  1144.     ent = strmap_get(addressmap, *addrp);
  1145.     tor_assert(ent);
  1146.     tor_assert(!strcasecmp(ent->new_address,new_address));
  1147.     vent = strmap_get(virtaddress_reversemap, new_address);
  1148.     tor_assert(vent);
  1149.     tor_assert(!strcasecmp(*addrp,
  1150.                            (type == RESOLVED_TYPE_IPV4) ?
  1151.                            vent->ipv4_address : vent->hostname_address));
  1152.     log_info(LD_APP, "Map from %s to %s okay.",
  1153.            safe_str(*addrp),safe_str(new_address));
  1154.   }
  1155. #endif
  1156.   return *addrp;
  1157. }
  1158. /** Return 1 if <b>address</b> has funny characters in it like colons. Return
  1159.  * 0 if it's fine, or if we're configured to allow it anyway.  <b>client</b>
  1160.  * should be true if we're using this address as a client; false if we're
  1161.  * using it as a server.
  1162.  */
  1163. int
  1164. address_is_invalid_destination(const char *address, int client)
  1165. {
  1166.   if (client) {
  1167.     if (get_options()->AllowNonRFC953Hostnames)
  1168.       return 0;
  1169.   } else {
  1170.     if (get_options()->ServerDNSAllowNonRFC953Hostnames)
  1171.       return 0;
  1172.   }
  1173.   while (*address) {
  1174.     if (TOR_ISALNUM(*address) ||
  1175.         *address == '-' ||
  1176.         *address == '.' ||
  1177.         *address == '_') /* Underscore is not allowed, but Windows does it
  1178.                           * sometimes, just to thumb its nose at the IETF. */
  1179.       ++address;
  1180.     else
  1181.       return 1;
  1182.   }
  1183.   return 0;
  1184. }
  1185. /** Iterate over all address mappings which have expiry times between
  1186.  * min_expires and max_expires, inclusive.  If sl is provided, add an
  1187.  * "old-addr new-addr expiry" string to sl for each mapping, omitting
  1188.  * the expiry time if want_expiry is false. If sl is NULL, remove the
  1189.  * mappings.
  1190.  */
  1191. void
  1192. addressmap_get_mappings(smartlist_t *sl, time_t min_expires,
  1193.                         time_t max_expires, int want_expiry)
  1194. {
  1195.    strmap_iter_t *iter;
  1196.    const char *key;
  1197.    void *_val;
  1198.    addressmap_entry_t *val;
  1199.    if (!addressmap)
  1200.      addressmap_init();
  1201.    for (iter = strmap_iter_init(addressmap); !strmap_iter_done(iter); ) {
  1202.      strmap_iter_get(iter, &key, &_val);
  1203.      val = _val;
  1204.      if (val->expires >= min_expires && val->expires <= max_expires) {
  1205.        if (!sl) {
  1206.          iter = strmap_iter_next_rmv(addressmap,iter);
  1207.          addressmap_ent_remove(key, val);
  1208.          continue;
  1209.        } else if (val->new_address) {
  1210.          size_t len = strlen(key)+strlen(val->new_address)+ISO_TIME_LEN+5;
  1211.          char *line = tor_malloc(len);
  1212.          if (want_expiry) {
  1213.            if (val->expires < 3 || val->expires == TIME_MAX)
  1214.              tor_snprintf(line, len, "%s %s NEVER", key, val->new_address);
  1215.            else {
  1216.              char time[ISO_TIME_LEN+1];
  1217.              format_iso_time(time, val->expires);
  1218.              tor_snprintf(line, len, "%s %s "%s"", key, val->new_address,
  1219.                           time);
  1220.            }
  1221.          } else {
  1222.            tor_snprintf(line, len, "%s %s", key, val->new_address);
  1223.          }
  1224.          smartlist_add(sl, line);
  1225.        }
  1226.      }
  1227.      iter = strmap_iter_next(addressmap,iter);
  1228.    }
  1229. }
  1230. /** Check if <b>conn</b> is using a dangerous port. Then warn and/or
  1231.  * reject depending on our config options. */
  1232. static int
  1233. consider_plaintext_ports(edge_connection_t *conn, uint16_t port)
  1234. {
  1235.   or_options_t *options = get_options();
  1236.   int reject = smartlist_string_num_isin(options->RejectPlaintextPorts, port);
  1237.   if (smartlist_string_num_isin(options->WarnPlaintextPorts, port)) {
  1238.     log_warn(LD_APP, "Application request to port %d: this port is "
  1239.              "commonly used for unencrypted protocols. Please make sure "
  1240.              "you don't send anything you would mind the rest of the "
  1241.              "Internet reading!%s", port, reject ? " Closing." : "");
  1242.     control_event_client_status(LOG_WARN, "DANGEROUS_PORT PORT=%d RESULT=%s",
  1243.                                 port, reject ? "REJECT" : "WARN");
  1244.   }
  1245.   if (reject) {
  1246.     log_info(LD_APP, "Port %d listed in RejectPlaintextPorts. Closing.", port);
  1247.     connection_mark_unattached_ap(conn, END_STREAM_REASON_ENTRYPOLICY);
  1248.     return -1;
  1249.   }
  1250.   return 0;
  1251. }
  1252. /** How many times do we try connecting with an exit configured via
  1253.  * TrackHostExits before concluding that it won't work any more and trying a
  1254.  * different one? */
  1255. #define TRACKHOSTEXITS_RETRIES 5
  1256. /** Connection <b>conn</b> just finished its socks handshake, or the
  1257.  * controller asked us to take care of it. If <b>circ</b> is defined,
  1258.  * then that's where we'll want to attach it. Otherwise we have to
  1259.  * figure it out ourselves.
  1260.  *
  1261.  * First, parse whether it's a .exit address, remap it, and so on. Then
  1262.  * if it's for a general circuit, try to attach it to a circuit (or launch
  1263.  * one as needed), else if it's for a rendezvous circuit, fetch a
  1264.  * rendezvous descriptor first (or attach/launch a circuit if the
  1265.  * rendezvous descriptor is already here and fresh enough).
  1266.  *
  1267.  * The stream will exit from the hop
  1268.  * indicated by <b>cpath</b>, or from the last hop in circ's cpath if
  1269.  * <b>cpath</b> is NULL.
  1270.  */
  1271. int
  1272. connection_ap_handshake_rewrite_and_attach(edge_connection_t *conn,
  1273.                                            origin_circuit_t *circ,
  1274.                                            crypt_path_t *cpath)
  1275. {
  1276.   socks_request_t *socks = conn->socks_request;
  1277.   hostname_type_t addresstype;
  1278.   or_options_t *options = get_options();
  1279.   struct in_addr addr_tmp;
  1280.   int automap = 0;
  1281.   char orig_address[MAX_SOCKS_ADDR_LEN];
  1282.   time_t map_expires = TIME_MAX;
  1283.   int remapped_to_exit = 0;
  1284.   time_t now = time(NULL);
  1285.   tor_strlower(socks->address); /* normalize it */
  1286.   strlcpy(orig_address, socks->address, sizeof(orig_address));
  1287.   log_debug(LD_APP,"Client asked for %s:%d",
  1288.             safe_str(socks->address),
  1289.             socks->port);
  1290.   if (socks->command == SOCKS_COMMAND_RESOLVE &&
  1291.       !tor_inet_aton(socks->address, &addr_tmp) &&
  1292.       options->AutomapHostsOnResolve && options->AutomapHostsSuffixes) {
  1293.     SMARTLIST_FOREACH(options->AutomapHostsSuffixes, const char *, cp,
  1294.                       if (!strcasecmpend(socks->address, cp)) {
  1295.                         automap = 1;
  1296.                         break;
  1297.                       });
  1298.     if (automap) {
  1299.       const char *new_addr;
  1300.       new_addr = addressmap_register_virtual_address(
  1301.                               RESOLVED_TYPE_IPV4, tor_strdup(socks->address));
  1302.       tor_assert(new_addr);
  1303.       log_info(LD_APP, "Automapping %s to %s",
  1304.                escaped_safe_str(socks->address), safe_str(new_addr));
  1305.       strlcpy(socks->address, new_addr, sizeof(socks->address));
  1306.     }
  1307.   }
  1308.   if (socks->command == SOCKS_COMMAND_RESOLVE_PTR) {
  1309.     if (addressmap_rewrite_reverse(socks->address, sizeof(socks->address),
  1310.                                    &map_expires)) {
  1311.       char *result = tor_strdup(socks->address);
  1312.       /* remember _what_ is supposed to have been resolved. */
  1313.       tor_snprintf(socks->address, sizeof(socks->address), "REVERSE[%s]",
  1314.                   orig_address);
  1315.       connection_ap_handshake_socks_resolved(conn, RESOLVED_TYPE_HOSTNAME,
  1316.                                              strlen(result), result, -1,
  1317.                                              map_expires);
  1318.       connection_mark_unattached_ap(conn,
  1319.                                 END_STREAM_REASON_DONE |
  1320.                                 END_STREAM_REASON_FLAG_ALREADY_SOCKS_REPLIED);
  1321.       return 0;
  1322.     }
  1323.     if (options->ClientDNSRejectInternalAddresses) {
  1324.       /* Don't let people try to do a reverse lookup on 10.0.0.1. */
  1325.       tor_addr_t addr;
  1326.       int ok;
  1327.       ok = tor_addr_parse_reverse_lookup_name(
  1328.                                &addr, socks->address, AF_UNSPEC, 1);
  1329.       if (ok == 1 && tor_addr_is_internal(&addr, 0)) {
  1330.         connection_ap_handshake_socks_resolved(conn, RESOLVED_TYPE_ERROR,
  1331.                                                0, NULL, -1, TIME_MAX);
  1332.         connection_mark_unattached_ap(conn,
  1333.                                  END_STREAM_REASON_SOCKSPROTOCOL |
  1334.                                  END_STREAM_REASON_FLAG_ALREADY_SOCKS_REPLIED);
  1335.         return -1;
  1336.       }
  1337.     }
  1338.   } else if (!automap) {
  1339.     int started_without_chosen_exit = strcasecmpend(socks->address, ".exit");
  1340.     /* For address map controls, remap the address. */
  1341.     if (addressmap_rewrite(socks->address, sizeof(socks->address),
  1342.                            &map_expires)) {
  1343.       control_event_stream_status(conn, STREAM_EVENT_REMAP,
  1344.                                   REMAP_STREAM_SOURCE_CACHE);
  1345.       if (started_without_chosen_exit &&
  1346.           !strcasecmpend(socks->address, ".exit") &&
  1347.           map_expires < TIME_MAX)
  1348.         remapped_to_exit = 1;
  1349.     }
  1350.   }
  1351.   if (!automap && address_is_in_virtual_range(socks->address)) {
  1352.     /* This address was probably handed out by client_dns_get_unmapped_address,
  1353.      * but the mapping was discarded for some reason.  We *don't* want to send
  1354.      * the address through Tor; that's likely to fail, and may leak
  1355.      * information.
  1356.      */
  1357.     log_warn(LD_APP,"Missing mapping for virtual address '%s'. Refusing.",
  1358.              socks->address); /* don't safe_str() this yet. */
  1359.     connection_mark_unattached_ap(conn, END_STREAM_REASON_INTERNAL);
  1360.     return -1;
  1361.   }
  1362.   /* Parse the address provided by SOCKS.  Modify it in-place if it
  1363.    * specifies a hidden-service (.onion) or particular exit node (.exit).
  1364.    */
  1365.   addresstype = parse_extended_hostname(socks->address);
  1366.   if (addresstype == BAD_HOSTNAME) {
  1367.     log_warn(LD_APP, "Invalid onion hostname %s; rejecting",
  1368.              safe_str(socks->address));
  1369.     control_event_client_status(LOG_WARN, "SOCKS_BAD_HOSTNAME HOSTNAME=%s",
  1370.                                 escaped(socks->address));
  1371.     connection_mark_unattached_ap(conn, END_STREAM_REASON_TORPROTOCOL);
  1372.     return -1;
  1373.   }
  1374.   if (addresstype == EXIT_HOSTNAME) {
  1375.     /* foo.exit -- modify conn->chosen_exit_node to specify the exit
  1376.      * node, and conn->address to hold only the address portion.*/
  1377.     char *s = strrchr(socks->address,'.');
  1378.     tor_assert(!automap);
  1379.     if (s) {
  1380.       if (s[1] != '') {
  1381.         conn->chosen_exit_name = tor_strdup(s+1);
  1382.         if (remapped_to_exit) /* 5 tries before it expires the addressmap */
  1383.           conn->chosen_exit_retries = TRACKHOSTEXITS_RETRIES;
  1384.         *s = 0;
  1385.       } else {
  1386.         log_warn(LD_APP,"Malformed exit address '%s.exit'. Refusing.",
  1387.                  safe_str(socks->address));
  1388.         control_event_client_status(LOG_WARN, "SOCKS_BAD_HOSTNAME HOSTNAME=%s",
  1389.                                     escaped(socks->address));
  1390.         connection_mark_unattached_ap(conn, END_STREAM_REASON_TORPROTOCOL);
  1391.         return -1;
  1392.       }
  1393.     } else {
  1394.       routerinfo_t *r;
  1395.       conn->chosen_exit_name = tor_strdup(socks->address);
  1396.       r = router_get_by_nickname(conn->chosen_exit_name, 1);
  1397.       *socks->address = 0;
  1398.       if (r) {
  1399.         strlcpy(socks->address, r->address, sizeof(socks->address));
  1400.       } else {
  1401.         log_warn(LD_APP,
  1402.                  "Unrecognized server in exit address '%s.exit'. Refusing.",
  1403.                  safe_str(socks->address));
  1404.         connection_mark_unattached_ap(conn, END_STREAM_REASON_TORPROTOCOL);
  1405.         return -1;
  1406.       }
  1407.     }
  1408.   }
  1409.   if (addresstype != ONION_HOSTNAME) {
  1410.     /* not a hidden-service request (i.e. normal or .exit) */
  1411.     if (address_is_invalid_destination(socks->address, 1)) {
  1412.       control_event_client_status(LOG_WARN, "SOCKS_BAD_HOSTNAME HOSTNAME=%s",
  1413.                                   escaped(socks->address));
  1414.       log_warn(LD_APP,
  1415.                "Destination '%s' seems to be an invalid hostname. Failing.",
  1416.                safe_str(socks->address));
  1417.       connection_mark_unattached_ap(conn, END_STREAM_REASON_TORPROTOCOL);
  1418.       return -1;
  1419.     }
  1420.     if (socks->command == SOCKS_COMMAND_RESOLVE) {
  1421.       uint32_t answer;
  1422.       struct in_addr in;
  1423.       /* Reply to resolves immediately if we can. */
  1424.       if (strlen(socks->address) > RELAY_PAYLOAD_SIZE) {
  1425.         log_warn(LD_APP,"Address to be resolved is too large. Failing.");
  1426.         control_event_client_status(LOG_WARN, "SOCKS_BAD_HOSTNAME HOSTNAME=%s",
  1427.                                     escaped(socks->address));
  1428.         connection_ap_handshake_socks_resolved(conn,
  1429.                                                RESOLVED_TYPE_ERROR_TRANSIENT,
  1430.                                                0,NULL,-1,TIME_MAX);
  1431.         connection_mark_unattached_ap(conn,
  1432.                                 END_STREAM_REASON_SOCKSPROTOCOL |
  1433.                                 END_STREAM_REASON_FLAG_ALREADY_SOCKS_REPLIED);
  1434.         return -1;
  1435.       }
  1436.       if (tor_inet_aton(socks->address, &in)) { /* see if it's an IP already */
  1437.         /* leave it in network order */
  1438.         answer = in.s_addr;
  1439.         /* remember _what_ is supposed to have been resolved. */
  1440.         strlcpy(socks->address, orig_address, sizeof(socks->address));
  1441.         connection_ap_handshake_socks_resolved(conn,RESOLVED_TYPE_IPV4,4,
  1442.                                                (char*)&answer,-1,map_expires);
  1443.         connection_mark_unattached_ap(conn,
  1444.                                 END_STREAM_REASON_DONE |
  1445.                                 END_STREAM_REASON_FLAG_ALREADY_SOCKS_REPLIED);
  1446.         return 0;
  1447.       }
  1448.       tor_assert(!automap);
  1449.       rep_hist_note_used_resolve(now); /* help predict this next time */
  1450.     } else if (socks->command == SOCKS_COMMAND_CONNECT) {
  1451.       tor_assert(!automap);
  1452.       if (socks->port == 0) {
  1453.         log_notice(LD_APP,"Application asked to connect to port 0. Refusing.");
  1454.         connection_mark_unattached_ap(conn, END_STREAM_REASON_TORPROTOCOL);
  1455.         return -1;
  1456.       }
  1457.       if (!conn->use_begindir && !conn->chosen_exit_name && !circ) {
  1458.         /* see if we can find a suitable enclave exit */
  1459.         routerinfo_t *r =
  1460.           router_find_exact_exit_enclave(socks->address, socks->port);
  1461.         if (r) {
  1462.           log_info(LD_APP,
  1463.                    "Redirecting address %s to exit at enclave router %s",
  1464.                    safe_str(socks->address), r->nickname);
  1465.           /* use the hex digest, not nickname, in case there are two
  1466.              routers with this nickname */
  1467.           conn->chosen_exit_name =
  1468.             tor_strdup(hex_str(r->cache_info.identity_digest, DIGEST_LEN));
  1469.           conn->chosen_exit_optional = 1;
  1470.         }
  1471.       }
  1472.       /* warn or reject if it's using a dangerous port */
  1473.       if (!conn->use_begindir && !conn->chosen_exit_name && !circ)
  1474.         if (consider_plaintext_ports(conn, socks->port) < 0)
  1475.           return -1;
  1476.       if (!conn->use_begindir) {
  1477.         /* help predict this next time */
  1478.         rep_hist_note_used_port(now, socks->port);
  1479.       }
  1480.     } else if (socks->command == SOCKS_COMMAND_RESOLVE_PTR) {
  1481.       rep_hist_note_used_resolve(now); /* help predict this next time */
  1482.       /* no extra processing needed */
  1483.     } else {
  1484.       tor_fragile_assert();
  1485.     }
  1486.     conn->_base.state = AP_CONN_STATE_CIRCUIT_WAIT;
  1487.     if ((circ && connection_ap_handshake_attach_chosen_circuit(
  1488.                    conn, circ, cpath) < 0) ||
  1489.         (!circ &&
  1490.          connection_ap_handshake_attach_circuit(conn) < 0)) {
  1491.       if (!conn->_base.marked_for_close)
  1492.         connection_mark_unattached_ap(conn, END_STREAM_REASON_CANT_ATTACH);
  1493.       return -1;
  1494.     }
  1495.     return 0;
  1496.   } else {
  1497.     /* it's a hidden-service request */
  1498.     rend_cache_entry_t *entry;
  1499.     int r;
  1500.     rend_service_authorization_t *client_auth;
  1501.     tor_assert(!automap);
  1502.     if (SOCKS_COMMAND_IS_RESOLVE(socks->command)) {
  1503.       /* if it's a resolve request, fail it right now, rather than
  1504.        * building all the circuits and then realizing it won't work. */
  1505.       log_warn(LD_APP,
  1506.                "Resolve requests to hidden services not allowed. Failing.");
  1507.       connection_ap_handshake_socks_resolved(conn,RESOLVED_TYPE_ERROR,
  1508.                                              0,NULL,-1,TIME_MAX);
  1509.       connection_mark_unattached_ap(conn,
  1510.                                 END_STREAM_REASON_SOCKSPROTOCOL |
  1511.                                 END_STREAM_REASON_FLAG_ALREADY_SOCKS_REPLIED);
  1512.       return -1;
  1513.     }
  1514.     if (circ) {
  1515.       log_warn(LD_CONTROL, "Attachstream to a circuit is not "
  1516.                "supported for .onion addresses currently. Failing.");
  1517.       connection_mark_unattached_ap(conn, END_STREAM_REASON_TORPROTOCOL);
  1518.       return -1;
  1519.     }
  1520.     conn->rend_data = tor_malloc_zero(sizeof(rend_data_t));
  1521.     strlcpy(conn->rend_data->onion_address, socks->address,
  1522.             sizeof(conn->rend_data->onion_address));
  1523.     log_info(LD_REND,"Got a hidden service request for ID '%s'",
  1524.              safe_str(conn->rend_data->onion_address));
  1525.     /* see if we already have it cached */
  1526.     r = rend_cache_lookup_entry(conn->rend_data->onion_address, -1, &entry);
  1527.     if (r<0) {
  1528.       log_warn(LD_BUG,"Invalid service name '%s'",
  1529.                safe_str(conn->rend_data->onion_address));
  1530.       connection_mark_unattached_ap(conn, END_STREAM_REASON_TORPROTOCOL);
  1531.       return -1;
  1532.     }
  1533.     /* Help predict this next time. We're not sure if it will need
  1534.      * a stable circuit yet, but we know we'll need *something*. */
  1535.     rep_hist_note_used_internal(now, 0, 1);
  1536.     /* Look up if we have client authorization for it. */
  1537.     client_auth = rend_client_lookup_service_authorization(
  1538.                                           conn->rend_data->onion_address);
  1539.     if (client_auth) {
  1540.       log_info(LD_REND, "Using previously configured client authorization "
  1541.                         "for hidden service request.");
  1542.       memcpy(conn->rend_data->descriptor_cookie,
  1543.              client_auth->descriptor_cookie, REND_DESC_COOKIE_LEN);
  1544.       conn->rend_data->auth_type = client_auth->auth_type;
  1545.     }
  1546.     if (r==0) {
  1547.       conn->_base.state = AP_CONN_STATE_RENDDESC_WAIT;
  1548.       log_info(LD_REND, "Unknown descriptor %s. Fetching.",
  1549.                safe_str(conn->rend_data->onion_address));
  1550.       /* Fetch both, v0 and v2 rend descriptors in parallel. Use whichever
  1551.        * arrives first. Exception: When using client authorization, only
  1552.        * fetch v2 descriptors.*/
  1553.       rend_client_refetch_v2_renddesc(conn->rend_data);
  1554.       if (conn->rend_data->auth_type == REND_NO_AUTH)
  1555.         rend_client_refetch_renddesc(conn->rend_data->onion_address);
  1556.     } else { /* r > 0 */
  1557.       if (now - entry->received < NUM_SECONDS_BEFORE_HS_REFETCH) {
  1558.         conn->_base.state = AP_CONN_STATE_CIRCUIT_WAIT;
  1559.         log_info(LD_REND, "Descriptor is here and fresh enough. Great.");
  1560.         if (connection_ap_handshake_attach_circuit(conn) < 0) {
  1561.           if (!conn->_base.marked_for_close)
  1562.             connection_mark_unattached_ap(conn, END_STREAM_REASON_CANT_ATTACH);
  1563.           return -1;
  1564.         }
  1565.       } else {
  1566.         conn->_base.state = AP_CONN_STATE_RENDDESC_WAIT;
  1567.         log_info(LD_REND, "Stale descriptor %s. Re-fetching.",
  1568.                  safe_str(conn->rend_data->onion_address));
  1569.         /* Fetch both, v0 and v2 rend descriptors in parallel. Use whichever
  1570.          * arrives first. Exception: When using client authorization, only
  1571.          * fetch v2 descriptors.*/
  1572.         rend_client_refetch_v2_renddesc(conn->rend_data);
  1573.         if (conn->rend_data->auth_type == REND_NO_AUTH)
  1574.           rend_client_refetch_renddesc(conn->rend_data->onion_address);
  1575.       }
  1576.     }
  1577.     return 0;
  1578.   }
  1579.   return 0; /* unreached but keeps the compiler happy */
  1580. }
  1581. #ifdef TRANS_PF
  1582. static int pf_socket = -1;
  1583. int
  1584. get_pf_socket(void)
  1585. {
  1586.   int pf;
  1587.   /*  This should be opened before dropping privileges. */
  1588.   if (pf_socket >= 0)
  1589.     return pf_socket;
  1590. #ifdef OPENBSD
  1591.   /* only works on OpenBSD */
  1592.   pf = open("/dev/pf", O_RDONLY);
  1593. #else
  1594.   /* works on NetBSD and FreeBSD */
  1595.   pf = open("/dev/pf", O_RDWR);
  1596. #endif
  1597.   if (pf < 0) {
  1598.     log_warn(LD_NET, "open("/dev/pf") failed: %s", strerror(errno));
  1599.     return -1;
  1600.   }
  1601.   pf_socket = pf;
  1602.   return pf_socket;
  1603. }
  1604. #endif
  1605. /** Fetch the original destination address and port from a
  1606.  * system-specific interface and put them into a
  1607.  * socks_request_t as if they came from a socks request.
  1608.  *
  1609.  * Return -1 if an error prevents fetching the destination,
  1610.  * else return 0.
  1611.  */
  1612. static int
  1613. connection_ap_get_original_destination(edge_connection_t *conn,
  1614.                                        socks_request_t *req)
  1615. {
  1616. #ifdef TRANS_NETFILTER
  1617.   /* Linux 2.4+ */
  1618.   struct sockaddr_storage orig_dst;
  1619.   socklen_t orig_dst_len = sizeof(orig_dst);
  1620.   tor_addr_t addr;
  1621.   if (getsockopt(conn->_base.s, SOL_IP, SO_ORIGINAL_DST,
  1622.                  (struct sockaddr*)&orig_dst, &orig_dst_len) < 0) {
  1623.     int e = tor_socket_errno(conn->_base.s);
  1624.     log_warn(LD_NET, "getsockopt() failed: %s", tor_socket_strerror(e));
  1625.     return -1;
  1626.   }
  1627.   tor_addr_from_sockaddr(&addr, (struct sockaddr*)&orig_dst, &req->port);
  1628.   tor_addr_to_str(req->address, &addr, sizeof(req->address), 0);
  1629.   return 0;
  1630. #elif defined(TRANS_PF)
  1631.   struct sockaddr_storage proxy_addr;
  1632.   socklen_t proxy_addr_len = sizeof(proxy_addr);
  1633.   struct sockaddr *proxy_sa = (struct sockaddr*) &proxy_addr;
  1634.   struct pfioc_natlook pnl;
  1635.   tor_addr_t addr;
  1636.   int pf = -1;
  1637.   if (getsockname(conn->_base.s, (struct sockaddr*)&proxy_addr,
  1638.                   &proxy_addr_len) < 0) {
  1639.     int e = tor_socket_errno(conn->_base.s);
  1640.     log_warn(LD_NET, "getsockname() to determine transocks destination "
  1641.              "failed: %s", tor_socket_strerror(e));
  1642.     return -1;
  1643.   }
  1644.   memset(&pnl, 0, sizeof(pnl));
  1645.   pnl.proto           = IPPROTO_TCP;
  1646.   pnl.direction       = PF_OUT;
  1647.   if (proxy_sa->sa_family == AF_INET) {
  1648.     struct sockaddr_in *sin = (struct sockaddr_in *)proxy_sa;
  1649.     pnl.af              = AF_INET;
  1650.     pnl.saddr.v4.s_addr = tor_addr_to_ipv4n(&conn->_base.addr);
  1651.     pnl.sport           = htons(conn->_base.port);
  1652.     pnl.daddr.v4.s_addr = sin->sin_addr.s_addr;
  1653.     pnl.dport           = sin->sin_port;
  1654.   } else if (proxy_sa->sa_family == AF_INET6) {
  1655.     struct sockaddr_in6 *sin6 = (struct sockaddr_in6 *)proxy_sa;
  1656.     pnl.af = AF_INET6;
  1657.     memcpy(&pnl.saddr.v6, tor_addr_to_in6(&conn->_base.addr),
  1658.            sizeof(struct in6_addr));
  1659.     pnl.sport = htons(conn->_base.port);
  1660.     memcpy(&pnl.daddr.v6, &sin6->sin6_addr, sizeof(struct in6_addr));
  1661.     pnl.dport = sin6->sin6_port;
  1662.   } else {
  1663.     log_warn(LD_NET, "getsockname() gave an unexpected address family (%d)",
  1664.              (int)proxy_sa->sa_family);
  1665.     return -1;
  1666.   }
  1667.   pf = get_pf_socket();
  1668.   if (pf<0)
  1669.     return -1;
  1670.   if (ioctl(pf, DIOCNATLOOK, &pnl) < 0) {
  1671.     log_warn(LD_NET, "ioctl(DIOCNATLOOK) failed: %s", strerror(errno));
  1672.     return -1;
  1673.   }
  1674.   if (pnl.af == AF_INET) {
  1675.     tor_addr_from_ipv4n(&addr, pnl.rdaddr.v4.s_addr);
  1676.   } else if (pnl.af == AF_INET6) {
  1677.     tor_addr_from_in6(&addr, &pnl.rdaddr.v6);
  1678.   } else {
  1679.     tor_fragile_assert();
  1680.     return -1;
  1681.   }
  1682.   tor_addr_to_str(req->address, &addr, sizeof(req->address), 0);
  1683.   req->port = ntohs(pnl.rdport);
  1684.   return 0;
  1685. #else
  1686.   (void)conn;
  1687.   (void)req;
  1688.   log_warn(LD_BUG, "Called connection_ap_get_original_destination, but no "
  1689.            "transparent proxy method was configured.");
  1690.   return -1;
  1691. #endif
  1692. }
  1693. /** connection_edge_process_inbuf() found a conn in state
  1694.  * socks_wait. See if conn->inbuf has the right bytes to proceed with
  1695.  * the socks handshake.
  1696.  *
  1697.  * If the handshake is complete, send it to
  1698.  * connection_ap_handshake_rewrite_and_attach().
  1699.  *
  1700.  * Return -1 if an unexpected error with conn occurs (and mark it for close),
  1701.  * else return 0.
  1702.  */
  1703. static int
  1704. connection_ap_handshake_process_socks(edge_connection_t *conn)
  1705. {
  1706.   socks_request_t *socks;
  1707.   int sockshere;
  1708.   or_options_t *options = get_options();
  1709.   tor_assert(conn);
  1710.   tor_assert(conn->_base.type == CONN_TYPE_AP);
  1711.   tor_assert(conn->_base.state == AP_CONN_STATE_SOCKS_WAIT);
  1712.   tor_assert(conn->socks_request);
  1713.   socks = conn->socks_request;
  1714.   log_debug(LD_APP,"entered.");
  1715.   sockshere = fetch_from_buf_socks(conn->_base.inbuf, socks,
  1716.                                    options->TestSocks, options->SafeSocks);
  1717.   if (sockshere == 0) {
  1718.     if (socks->replylen) {
  1719.       connection_write_to_buf(socks->reply, socks->replylen, TO_CONN(conn));
  1720.       /* zero it out so we can do another round of negotiation */
  1721.       socks->replylen = 0;
  1722.     } else {
  1723.       log_debug(LD_APP,"socks handshake not all here yet.");
  1724.     }
  1725.     return 0;
  1726.   } else if (sockshere == -1) {
  1727.     if (socks->replylen) { /* we should send reply back */
  1728.       log_debug(LD_APP,"reply is already set for us. Using it.");
  1729.       connection_ap_handshake_socks_reply(conn, socks->reply, socks->replylen,
  1730.                                           END_STREAM_REASON_SOCKSPROTOCOL);
  1731.     } else {
  1732.       log_warn(LD_APP,"Fetching socks handshake failed. Closing.");
  1733.       connection_ap_handshake_socks_reply(conn, NULL, 0,
  1734.                                           END_STREAM_REASON_SOCKSPROTOCOL);
  1735.     }
  1736.     connection_mark_unattached_ap(conn,
  1737.                               END_STREAM_REASON_SOCKSPROTOCOL |
  1738.                               END_STREAM_REASON_FLAG_ALREADY_SOCKS_REPLIED);
  1739.     return -1;
  1740.   } /* else socks handshake is done, continue processing */
  1741.   if (hostname_is_noconnect_address(socks->address))
  1742.   {
  1743.     control_event_stream_status(conn, STREAM_EVENT_NEW, 0);
  1744.     control_event_stream_status(conn, STREAM_EVENT_CLOSED, 0);
  1745.     connection_mark_unattached_ap(conn, END_STREAM_REASON_DONE);
  1746.     return -1;
  1747.   }
  1748.   if (SOCKS_COMMAND_IS_CONNECT(socks->command))
  1749.     control_event_stream_status(conn, STREAM_EVENT_NEW, 0);
  1750.   else
  1751.     control_event_stream_status(conn, STREAM_EVENT_NEW_RESOLVE, 0);
  1752.   if (options->LeaveStreamsUnattached) {
  1753.     conn->_base.state = AP_CONN_STATE_CONTROLLER_WAIT;
  1754.     return 0;
  1755.   }
  1756.   return connection_ap_handshake_rewrite_and_attach(conn, NULL, NULL);
  1757. }
  1758. /** connection_init_accepted_conn() found a new trans AP conn.
  1759.  * Get the original destination and send it to
  1760.  * connection_ap_handshake_rewrite_and_attach().
  1761.  *
  1762.  * Return -1 if an unexpected error with conn (and it should be marked
  1763.  * for close), else return 0.
  1764.  */
  1765. int
  1766. connection_ap_process_transparent(edge_connection_t *conn)
  1767. {
  1768.   socks_request_t *socks;
  1769.   or_options_t *options = get_options();
  1770.   tor_assert(conn);
  1771.   tor_assert(conn->_base.type == CONN_TYPE_AP);
  1772.   tor_assert(conn->socks_request);
  1773.   socks = conn->socks_request;
  1774.   /* pretend that a socks handshake completed so we don't try to
  1775.    * send a socks reply down a transparent conn */
  1776.   socks->command = SOCKS_COMMAND_CONNECT;
  1777.   socks->has_finished = 1;
  1778.   log_debug(LD_APP,"entered.");
  1779.   if (connection_ap_get_original_destination(conn, socks) < 0) {
  1780.     log_warn(LD_APP,"Fetching original destination failed. Closing.");
  1781.     connection_mark_unattached_ap(conn,
  1782.                                END_STREAM_REASON_CANT_FETCH_ORIG_DEST);
  1783.     return -1;
  1784.   }
  1785.   /* we have the original destination */
  1786.   control_event_stream_status(conn, STREAM_EVENT_NEW, 0);
  1787.   if (options->LeaveStreamsUnattached) {
  1788.     conn->_base.state = AP_CONN_STATE_CONTROLLER_WAIT;
  1789.     return 0;
  1790.   }
  1791.   return connection_ap_handshake_rewrite_and_attach(conn, NULL, NULL);
  1792. }
  1793. /** connection_edge_process_inbuf() found a conn in state natd_wait. See if
  1794.  * conn->inbuf has the right bytes to proceed.  See FreeBSD's libalias(3) and
  1795.  * ProxyEncodeTcpStream() in src/lib/libalias/alias_proxy.c for the encoding
  1796.  * form of the original destination.
  1797.  *
  1798.  * If the original destination is complete, send it to
  1799.  * connection_ap_handshake_rewrite_and_attach().
  1800.  *
  1801.  * Return -1 if an unexpected error with conn (and it should be marked
  1802.  * for close), else return 0.
  1803.  */
  1804. static int
  1805. connection_ap_process_natd(edge_connection_t *conn)
  1806. {
  1807.   char tmp_buf[36], *tbuf, *daddr;
  1808.   size_t tlen = 30;
  1809.   int err, port_ok;
  1810.   socks_request_t *socks;
  1811.   or_options_t *options = get_options();
  1812.   tor_assert(conn);
  1813.   tor_assert(conn->_base.type == CONN_TYPE_AP);
  1814.   tor_assert(conn->_base.state == AP_CONN_STATE_NATD_WAIT);
  1815.   tor_assert(conn->socks_request);
  1816.   socks = conn->socks_request;
  1817.   log_debug(LD_APP,"entered.");
  1818.   /* look for LF-terminated "[DEST ip_addr port]"
  1819.    * where ip_addr is a dotted-quad and port is in string form */
  1820.   err = fetch_from_buf_line(conn->_base.inbuf, tmp_buf, &tlen);
  1821.   if (err == 0)
  1822.     return 0;
  1823.   if (err < 0) {
  1824.     log_warn(LD_APP,"Natd handshake failed (DEST too long). Closing");
  1825.     connection_mark_unattached_ap(conn, END_STREAM_REASON_INVALID_NATD_DEST);
  1826.     return -1;
  1827.   }
  1828.   if (strcmpstart(tmp_buf, "[DEST ")) {
  1829.     log_warn(LD_APP,"Natd handshake was ill-formed; closing. The client "
  1830.              "said: %s",
  1831.              escaped(tmp_buf));
  1832.     connection_mark_unattached_ap(conn, END_STREAM_REASON_INVALID_NATD_DEST);
  1833.     return -1;
  1834.   }
  1835.   daddr = tbuf = &tmp_buf[0] + 6; /* after end of "[DEST " */
  1836.   if (!(tbuf = strchr(tbuf, ' '))) {
  1837.     log_warn(LD_APP,"Natd handshake was ill-formed; closing. The client "
  1838.              "said: %s",
  1839.              escaped(tmp_buf));
  1840.     connection_mark_unattached_ap(conn, END_STREAM_REASON_INVALID_NATD_DEST);
  1841.     return -1;
  1842.   }
  1843.   *tbuf++ = '';
  1844.   /* pretend that a socks handshake completed so we don't try to
  1845.    * send a socks reply down a natd conn */
  1846.   strlcpy(socks->address, daddr, sizeof(socks->address));
  1847.   socks->port = (uint16_t)
  1848.     tor_parse_long(tbuf, 10, 1, 65535, &port_ok, &daddr);
  1849.   if (!port_ok) {
  1850.     log_warn(LD_APP,"Natd handshake failed; port %s is ill-formed or out "
  1851.              "of range.", escaped(tbuf));
  1852.     connection_mark_unattached_ap(conn, END_STREAM_REASON_INVALID_NATD_DEST);
  1853.     return -1;
  1854.   }
  1855.   socks->command = SOCKS_COMMAND_CONNECT;
  1856.   socks->has_finished = 1;
  1857.   control_event_stream_status(conn, STREAM_EVENT_NEW, 0);
  1858.   if (options->LeaveStreamsUnattached) {
  1859.     conn->_base.state = AP_CONN_STATE_CONTROLLER_WAIT;
  1860.     return 0;
  1861.   }
  1862.   conn->_base.state = AP_CONN_STATE_CIRCUIT_WAIT;
  1863.   return connection_ap_handshake_rewrite_and_attach(conn, NULL, NULL);
  1864. }
  1865. /** Iterate over the two bytes of stream_id until we get one that is not
  1866.  * already in use; return it. Return 0 if can't get a unique stream_id.
  1867.  */
  1868. static streamid_t
  1869. get_unique_stream_id_by_circ(origin_circuit_t *circ)
  1870. {
  1871.   edge_connection_t *tmpconn;
  1872.   streamid_t test_stream_id;
  1873.   uint32_t attempts=0;
  1874. again:
  1875.   test_stream_id = circ->next_stream_id++;
  1876.   if (++attempts > 1<<16) {
  1877.     /* Make sure we don't loop forever if all stream_id's are used. */
  1878.     log_warn(LD_APP,"No unused stream IDs. Failing.");
  1879.     return 0;
  1880.   }
  1881.   if (test_stream_id == 0)
  1882.     goto again;
  1883.   for (tmpconn = circ->p_streams; tmpconn; tmpconn=tmpconn->next_stream)
  1884.     if (tmpconn->stream_id == test_stream_id)
  1885.       goto again;
  1886.   return test_stream_id;
  1887. }
  1888. /** Write a relay begin cell, using destaddr and destport from ap_conn's
  1889.  * socks_request field, and send it down circ.
  1890.  *
  1891.  * If ap_conn is broken, mark it for close and return -1. Else return 0.
  1892.  */
  1893. int
  1894. connection_ap_handshake_send_begin(edge_connection_t *ap_conn)
  1895. {
  1896.   char payload[CELL_PAYLOAD_SIZE];
  1897.   int payload_len;
  1898.   int begin_type;
  1899.   origin_circuit_t *circ;
  1900.   tor_assert(ap_conn->on_circuit);
  1901.   circ = TO_ORIGIN_CIRCUIT(ap_conn->on_circuit);
  1902.   tor_assert(ap_conn->_base.type == CONN_TYPE_AP);
  1903.   tor_assert(ap_conn->_base.state == AP_CONN_STATE_CIRCUIT_WAIT);
  1904.   tor_assert(ap_conn->socks_request);
  1905.   tor_assert(SOCKS_COMMAND_IS_CONNECT(ap_conn->socks_request->command));
  1906.   ap_conn->stream_id = get_unique_stream_id_by_circ(circ);
  1907.   if (ap_conn->stream_id==0) {
  1908.     connection_mark_unattached_ap(ap_conn, END_STREAM_REASON_INTERNAL);
  1909.     circuit_mark_for_close(TO_CIRCUIT(circ), END_CIRC_REASON_RESOURCELIMIT);
  1910.     return -1;
  1911.   }
  1912.   tor_snprintf(payload,RELAY_PAYLOAD_SIZE, "%s:%d",
  1913.                (circ->_base.purpose == CIRCUIT_PURPOSE_C_GENERAL) ?
  1914.                  ap_conn->socks_request->address : "",
  1915.                ap_conn->socks_request->port);
  1916.   payload_len = (int)strlen(payload)+1;
  1917.   log_debug(LD_APP,
  1918.             "Sending relay cell to begin stream %d.", ap_conn->stream_id);
  1919.   begin_type = ap_conn->use_begindir ?
  1920.                  RELAY_COMMAND_BEGIN_DIR : RELAY_COMMAND_BEGIN;
  1921.   if (begin_type == RELAY_COMMAND_BEGIN) {
  1922.     tor_assert(circ->build_state->onehop_tunnel == 0);
  1923.   }
  1924.   if (connection_edge_send_command(ap_conn, begin_type,
  1925.                   begin_type == RELAY_COMMAND_BEGIN ? payload : NULL,
  1926.                   begin_type == RELAY_COMMAND_BEGIN ? payload_len : 0) < 0)
  1927.     return -1; /* circuit is closed, don't continue */
  1928.   ap_conn->package_window = STREAMWINDOW_START;
  1929.   ap_conn->deliver_window = STREAMWINDOW_START;
  1930.   ap_conn->_base.state = AP_CONN_STATE_CONNECT_WAIT;
  1931.   log_info(LD_APP,"Address/port sent, ap socket %d, n_circ_id %d",
  1932.            ap_conn->_base.s, circ->_base.n_circ_id);
  1933.   control_event_stream_status(ap_conn, STREAM_EVENT_SENT_CONNECT, 0);
  1934.   return 0;
  1935. }
  1936. /** Write a relay resolve cell, using destaddr and destport from ap_conn's
  1937.  * socks_request field, and send it down circ.
  1938.  *
  1939.  * If ap_conn is broken, mark it for close and return -1. Else return 0.
  1940.  */
  1941. int
  1942. connection_ap_handshake_send_resolve(edge_connection_t *ap_conn)
  1943. {
  1944.   int payload_len, command;
  1945.   const char *string_addr;
  1946.   char inaddr_buf[REVERSE_LOOKUP_NAME_BUF_LEN];
  1947.   origin_circuit_t *circ;
  1948.   tor_assert(ap_conn->on_circuit);
  1949.   circ = TO_ORIGIN_CIRCUIT(ap_conn->on_circuit);
  1950.   tor_assert(ap_conn->_base.type == CONN_TYPE_AP);
  1951.   tor_assert(ap_conn->_base.state == AP_CONN_STATE_CIRCUIT_WAIT);
  1952.   tor_assert(ap_conn->socks_request);
  1953.   tor_assert(circ->_base.purpose == CIRCUIT_PURPOSE_C_GENERAL);
  1954.   command = ap_conn->socks_request->command;
  1955.   tor_assert(SOCKS_COMMAND_IS_RESOLVE(command));
  1956.   ap_conn->stream_id = get_unique_stream_id_by_circ(circ);
  1957.   if (ap_conn->stream_id==0) {
  1958.     connection_mark_unattached_ap(ap_conn, END_STREAM_REASON_INTERNAL);
  1959.     /*XXXX022 _close_ the circuit because it's full?  That sounds dumb. */
  1960.     circuit_mark_for_close(TO_CIRCUIT(circ), END_CIRC_REASON_RESOURCELIMIT);
  1961.     return -1;
  1962.   }
  1963.   if (command == SOCKS_COMMAND_RESOLVE) {
  1964.     string_addr = ap_conn->socks_request->address;
  1965.     payload_len = (int)strlen(string_addr)+1;
  1966.   } else {
  1967.     /* command == SOCKS_COMMAND_RESOLVE_PTR */
  1968.     const char *a = ap_conn->socks_request->address;
  1969.     tor_addr_t addr;
  1970.     int r;
  1971.     /* We're doing a reverse lookup.  The input could be an IP address, or
  1972.      * could be an .in-addr.arpa or .ip6.arpa address */
  1973.     r = tor_addr_parse_reverse_lookup_name(&addr, a, AF_INET, 1);
  1974.     if (r <= 0) {
  1975.       log_warn(LD_APP, "Rejecting ill-formed reverse lookup of %s",
  1976.                safe_str(a));
  1977.       connection_mark_unattached_ap(ap_conn, END_STREAM_REASON_INTERNAL);
  1978.       return -1;
  1979.     }
  1980.     r = tor_addr_to_reverse_lookup_name(inaddr_buf, sizeof(inaddr_buf), &addr);
  1981.     if (r < 0) {
  1982.       log_warn(LD_BUG, "Couldn't generate reverse lookup hostname of %s",
  1983.                safe_str(a));
  1984.       connection_mark_unattached_ap(ap_conn, END_STREAM_REASON_INTERNAL);
  1985.       return -1;
  1986.     }
  1987.     string_addr = inaddr_buf;
  1988.     payload_len = (int)strlen(inaddr_buf)+1;
  1989.     tor_assert(payload_len <= (int)sizeof(inaddr_buf));
  1990.   }
  1991.   if (payload_len > RELAY_PAYLOAD_SIZE) {
  1992.     /* This should be impossible: we don't accept addresses this big. */
  1993.     connection_mark_unattached_ap(ap_conn, END_STREAM_REASON_INTERNAL);
  1994.     return -1;
  1995.   }
  1996.   log_debug(LD_APP,
  1997.             "Sending relay cell to begin stream %d.", ap_conn->stream_id);
  1998.   if (connection_edge_send_command(ap_conn,
  1999.                            RELAY_COMMAND_RESOLVE,
  2000.                            string_addr, payload_len) < 0)
  2001.     return -1; /* circuit is closed, don't continue */
  2002.   tor_free(ap_conn->_base.address); /* Maybe already set by dnsserv. */
  2003.   ap_conn->_base.address = tor_strdup("(Tor_internal)");
  2004.   ap_conn->_base.state = AP_CONN_STATE_RESOLVE_WAIT;
  2005.   log_info(LD_APP,"Address sent for resolve, ap socket %d, n_circ_id %d",
  2006.            ap_conn->_base.s, circ->_base.n_circ_id);
  2007.   control_event_stream_status(ap_conn, STREAM_EVENT_NEW, 0);
  2008.   control_event_stream_status(ap_conn, STREAM_EVENT_SENT_RESOLVE, 0);
  2009.   return 0;
  2010. }
  2011. /** Make an AP connection_t, make a new linked connection pair, and attach
  2012.  * one side to the conn, connection_add it, initialize it to circuit_wait,
  2013.  * and call connection_ap_handshake_attach_circuit(conn) on it.
  2014.  *
  2015.  * Return the other end of the linked connection pair, or -1 if error.
  2016.  */
  2017. edge_connection_t *
  2018. connection_ap_make_link(char *address, uint16_t port,
  2019.                         const char *digest, int use_begindir, int want_onehop)
  2020. {
  2021.   edge_connection_t *conn;
  2022.   log_info(LD_APP,"Making internal %s tunnel to %s:%d ...",
  2023.            want_onehop ? "direct" : "anonymized" , safe_str(address),port);
  2024.   conn = edge_connection_new(CONN_TYPE_AP, AF_INET);
  2025.   conn->_base.linked = 1; /* so that we can add it safely below. */
  2026.   /* populate conn->socks_request */
  2027.   /* leave version at zero, so the socks_reply is empty */
  2028.   conn->socks_request->socks_version = 0;
  2029.   conn->socks_request->has_finished = 0; /* waiting for 'connected' */
  2030.   strlcpy(conn->socks_request->address, address,
  2031.           sizeof(conn->socks_request->address));
  2032.   conn->socks_request->port = port;
  2033.   conn->socks_request->command = SOCKS_COMMAND_CONNECT;
  2034.   conn->want_onehop = want_onehop;
  2035.   conn->use_begindir = use_begindir;
  2036.   if (use_begindir) {
  2037.     conn->chosen_exit_name = tor_malloc(HEX_DIGEST_LEN+2);
  2038.     conn->chosen_exit_name[0] = '$';
  2039.     tor_assert(digest);
  2040.     base16_encode(conn->chosen_exit_name+1,HEX_DIGEST_LEN+1,
  2041.                   digest, DIGEST_LEN);
  2042.   }
  2043.   conn->_base.address = tor_strdup("(Tor_internal)");
  2044.   tor_addr_make_unspec(&conn->_base.addr);
  2045.   conn->_base.port = 0;
  2046.   if (connection_add(TO_CONN(conn)) < 0) { /* no space, forget it */
  2047.     connection_free(TO_CONN(conn));
  2048.     return NULL;
  2049.   }
  2050.   conn->_base.state = AP_CONN_STATE_CIRCUIT_WAIT;
  2051.   control_event_stream_status(conn, STREAM_EVENT_NEW, 0);
  2052.   /* attaching to a dirty circuit is fine */
  2053.   if (connection_ap_handshake_attach_circuit(conn) < 0) {
  2054.     if (!conn->_base.marked_for_close)
  2055.       connection_mark_unattached_ap(conn, END_STREAM_REASON_CANT_ATTACH);
  2056.     return NULL;
  2057.   }
  2058.   log_info(LD_APP,"... application connection created and linked.");
  2059.   return conn;
  2060. }
  2061. /** Notify any interested controller connections about a new hostname resolve
  2062.  * or resolve error.  Takes the same arguments as does
  2063.  * connection_ap_handshake_socks_resolved(). */
  2064. static void
  2065. tell_controller_about_resolved_result(edge_connection_t *conn,
  2066.                                       int answer_type,
  2067.                                       size_t answer_len,
  2068.                                       const char *answer,
  2069.                                       int ttl,
  2070.                                       time_t expires)
  2071. {
  2072.   if (ttl >= 0 && (answer_type == RESOLVED_TYPE_IPV4 ||
  2073.                    answer_type == RESOLVED_TYPE_HOSTNAME)) {
  2074.     return; /* we already told the controller. */
  2075.   } else if (answer_type == RESOLVED_TYPE_IPV4 && answer_len >= 4) {
  2076.     struct in_addr in;
  2077.     char buf[INET_NTOA_BUF_LEN];
  2078.     in.s_addr = get_uint32(answer);
  2079.     tor_inet_ntoa(&in, buf, sizeof(buf));
  2080.     control_event_address_mapped(conn->socks_request->address,
  2081.                                  buf, expires, NULL);
  2082.   } else if (answer_type == RESOLVED_TYPE_HOSTNAME && answer_len <256) {
  2083.     char *cp = tor_strndup(answer, answer_len);
  2084.     control_event_address_mapped(conn->socks_request->address,
  2085.                                  cp, expires, NULL);
  2086.     tor_free(cp);
  2087.   } else {
  2088.     control_event_address_mapped(conn->socks_request->address,
  2089.                                  "<error>",
  2090.                                  time(NULL)+ttl,
  2091.                                  "error=yes");
  2092.   }
  2093. }
  2094. /** Send an answer to an AP connection that has requested a DNS lookup via
  2095.  * SOCKS.  The type should be one of RESOLVED_TYPE_(IPV4|IPV6|HOSTNAME) or -1
  2096.  * for unreachable; the answer should be in the format specified in the socks
  2097.  * extensions document.  <b>ttl</b> is the ttl for the answer, or -1 on
  2098.  * certain errors or for values that didn't come via DNS.  <b>expires</b> is
  2099.  * a time when the answer expires, or -1 or TIME_MAX if there's a good TTL.
  2100.  **/
  2101. /* XXXX022 the use of the ttl and expires fields is nutty.  Let's make this
  2102.  * interface and those that use it less ugly. */
  2103. void
  2104. connection_ap_handshake_socks_resolved(edge_connection_t *conn,
  2105.                                        int answer_type,
  2106.                                        size_t answer_len,
  2107.                                        const char *answer,
  2108.                                        int ttl,
  2109.                                        time_t expires)
  2110. {
  2111.   char buf[384];
  2112.   size_t replylen;
  2113.   if (ttl >= 0) {
  2114.     if (answer_type == RESOLVED_TYPE_IPV4 && answer_len == 4) {
  2115.       uint32_t a = ntohl(get_uint32(answer));
  2116.       if (a)
  2117.         client_dns_set_addressmap(conn->socks_request->address, a,
  2118.                                   conn->chosen_exit_name, ttl);
  2119.     } else if (answer_type == RESOLVED_TYPE_HOSTNAME && answer_len < 256) {
  2120.       char *cp = tor_strndup(answer, answer_len);
  2121.       client_dns_set_reverse_addressmap(conn->socks_request->address,
  2122.                                         cp,
  2123.                                         conn->chosen_exit_name, ttl);
  2124.       tor_free(cp);
  2125.     }
  2126.   }
  2127.   if (conn->is_dns_request) {
  2128.     if (conn->dns_server_request) {
  2129.       /* We had a request on our DNS port: answer it. */
  2130.       dnsserv_resolved(conn, answer_type, answer_len, answer, ttl);
  2131.       conn->socks_request->has_finished = 1;
  2132.       return;
  2133.     } else {
  2134.       /* This must be a request from the controller. We already sent
  2135.        * a mapaddress if there's a ttl. */
  2136.       tell_controller_about_resolved_result(conn, answer_type, answer_len,
  2137.                                             answer, ttl, expires);
  2138.       conn->socks_request->has_finished = 1;
  2139.       return;
  2140.     }
  2141.     /* We shouldn't need to free conn here; it gets marked by the caller. */
  2142.   }
  2143.   if (conn->socks_request->socks_version == 4) {
  2144.     buf[0] = 0x00; /* version */
  2145.     if (answer_type == RESOLVED_TYPE_IPV4 && answer_len == 4) {
  2146.       buf[1] = SOCKS4_GRANTED;
  2147.       set_uint16(buf+2, 0);
  2148.       memcpy(buf+4, answer, 4); /* address */
  2149.       replylen = SOCKS4_NETWORK_LEN;
  2150.     } else { /* "error" */
  2151.       buf[1] = SOCKS4_REJECT;
  2152.       memset(buf+2, 0, 6);
  2153.       replylen = SOCKS4_NETWORK_LEN;
  2154.     }
  2155.   } else if (conn->socks_request->socks_version == 5) {
  2156.     /* SOCKS5 */
  2157.     buf[0] = 0x05; /* version */
  2158.     if (answer_type == RESOLVED_TYPE_IPV4 && answer_len == 4) {
  2159.       buf[1] = SOCKS5_SUCCEEDED;
  2160.       buf[2] = 0; /* reserved */
  2161.       buf[3] = 0x01; /* IPv4 address type */
  2162.       memcpy(buf+4, answer, 4); /* address */
  2163.       set_uint16(buf+8, 0); /* port == 0. */
  2164.       replylen = 10;
  2165.     } else if (answer_type == RESOLVED_TYPE_IPV6 && answer_len == 16) {
  2166.       buf[1] = SOCKS5_SUCCEEDED;
  2167.       buf[2] = 0; /* reserved */
  2168.       buf[3] = 0x04; /* IPv6 address type */
  2169.       memcpy(buf+4, answer, 16); /* address */
  2170.       set_uint16(buf+20, 0); /* port == 0. */
  2171.       replylen = 22;
  2172.     } else if (answer_type == RESOLVED_TYPE_HOSTNAME && answer_len < 256) {
  2173.       buf[1] = SOCKS5_SUCCEEDED;
  2174.       buf[2] = 0; /* reserved */
  2175.       buf[3] = 0x03; /* Domainname address type */
  2176.       buf[4] = (char)answer_len;
  2177.       memcpy(buf+5, answer, answer_len); /* address */
  2178.       set_uint16(buf+5+answer_len, 0); /* port == 0. */
  2179.       replylen = 5+answer_len+2;
  2180.     } else {
  2181.       buf[1] = SOCKS5_HOST_UNREACHABLE;
  2182.       memset(buf+2, 0, 8);
  2183.       replylen = 10;
  2184.     }
  2185.   } else {
  2186.     /* no socks version info; don't send anything back */
  2187.     return;
  2188.   }
  2189.   connection_ap_handshake_socks_reply(conn, buf, replylen,
  2190.           (answer_type == RESOLVED_TYPE_IPV4 ||
  2191.            answer_type == RESOLVED_TYPE_IPV6) ?
  2192.                                       0 : END_STREAM_REASON_RESOLVEFAILED);
  2193. }
  2194. /** Send a socks reply to stream <b>conn</b>, using the appropriate
  2195.  * socks version, etc, and mark <b>conn</b> as completed with SOCKS
  2196.  * handshaking.
  2197.  *
  2198.  * If <b>reply</b> is defined, then write <b>replylen</b> bytes of it to conn
  2199.  * and return, else reply based on <b>endreason</b> (one of
  2200.  * END_STREAM_REASON_*). If <b>reply</b> is undefined, <b>endreason</b> can't
  2201.  * be 0 or REASON_DONE.  Send endreason to the controller, if appropriate.
  2202.  */
  2203. void
  2204. connection_ap_handshake_socks_reply(edge_connection_t *conn, char *reply,
  2205.                                     size_t replylen, int endreason)
  2206. {
  2207.   char buf[256];
  2208.   socks5_reply_status_t status =
  2209.     stream_end_reason_to_socks5_response(endreason);
  2210.   tor_assert(conn->socks_request); /* make sure it's an AP stream */
  2211.   control_event_stream_status(conn,
  2212.      status==SOCKS5_SUCCEEDED ? STREAM_EVENT_SUCCEEDED : STREAM_EVENT_FAILED,
  2213.                               endreason);
  2214.   if (conn->socks_request->has_finished) {
  2215.     log_warn(LD_BUG, "(Harmless.) duplicate calls to "
  2216.              "connection_ap_handshake_socks_reply.");
  2217.     return;
  2218.   }
  2219.   if (replylen) { /* we already have a reply in mind */
  2220.     connection_write_to_buf(reply, replylen, TO_CONN(conn));
  2221.     conn->socks_request->has_finished = 1;
  2222.     return;
  2223.   }
  2224.   if (conn->socks_request->socks_version == 4) {
  2225.     memset(buf,0,SOCKS4_NETWORK_LEN);
  2226.     buf[1] = (status==SOCKS5_SUCCEEDED ? SOCKS4_GRANTED : SOCKS4_REJECT);
  2227.     /* leave version, destport, destip zero */
  2228.     connection_write_to_buf(buf, SOCKS4_NETWORK_LEN, TO_CONN(conn));
  2229.   } else if (conn->socks_request->socks_version == 5) {
  2230.     buf[0] = 5; /* version 5 */
  2231.     buf[1] = (char)status;
  2232.     buf[2] = 0;
  2233.     buf[3] = 1; /* ipv4 addr */
  2234.     memset(buf+4,0,6); /* Set external addr/port to 0.
  2235.                           The spec doesn't seem to say what to do here. -RD */
  2236.     connection_write_to_buf(buf,10,TO_CONN(conn));
  2237.   }
  2238.   /* If socks_version isn't 4 or 5, don't send anything.
  2239.    * This can happen in the case of AP bridges. */
  2240.   conn->socks_request->has_finished = 1;
  2241.   return;
  2242. }
  2243. /** A relay 'begin' or 'begin_dir' cell has arrived, and either we are
  2244.  * an exit hop for the circuit, or we are the origin and it is a
  2245.  * rendezvous begin.
  2246.  *
  2247.  * Launch a new exit connection and initialize things appropriately.
  2248.  *
  2249.  * If it's a rendezvous stream, call connection_exit_connect() on
  2250.  * it.
  2251.  *
  2252.  * For general streams, call dns_resolve() on it first, and only call
  2253.  * connection_exit_connect() if the dns answer is already known.
  2254.  *
  2255.  * Note that we don't call connection_add() on the new stream! We wait
  2256.  * for connection_exit_connect() to do that.
  2257.  *
  2258.  * Return -(some circuit end reason) if we want to tear down <b>circ</b>.
  2259.  * Else return 0.
  2260.  */
  2261. int
  2262. connection_exit_begin_conn(cell_t *cell, circuit_t *circ)
  2263. {
  2264.   edge_connection_t *n_stream;
  2265.   relay_header_t rh;
  2266.   char *address=NULL;
  2267.   uint16_t port;
  2268.   or_circuit_t *or_circ = NULL;
  2269.   assert_circuit_ok(circ);
  2270.   if (!CIRCUIT_IS_ORIGIN(circ))
  2271.     or_circ = TO_OR_CIRCUIT(circ);
  2272.   relay_header_unpack(&rh, cell->payload);
  2273.   /* Note: we have to use relay_send_command_from_edge here, not
  2274.    * connection_edge_end or connection_edge_send_command, since those require
  2275.    * that we have a stream connected to a circuit, and we don't connect to a
  2276.    * circuit until we have a pending/successful resolve. */
  2277.   if (!server_mode(get_options()) &&
  2278.       circ->purpose != CIRCUIT_PURPOSE_S_REND_JOINED) {
  2279.     log_fn(LOG_PROTOCOL_WARN, LD_PROTOCOL,
  2280.            "Relay begin cell at non-server. Closing.");
  2281.     relay_send_end_cell_from_edge(rh.stream_id, circ,
  2282.                                   END_STREAM_REASON_EXITPOLICY, NULL);
  2283.     return 0;
  2284.   }
  2285.   if (rh.command == RELAY_COMMAND_BEGIN) {
  2286.     if (!memchr(cell->payload+RELAY_HEADER_SIZE, 0, rh.length)) {
  2287.       log_fn(LOG_PROTOCOL_WARN, LD_PROTOCOL,
  2288.              "Relay begin cell has no \0. Closing.");
  2289.       relay_send_end_cell_from_edge(rh.stream_id, circ,
  2290.                                     END_STREAM_REASON_TORPROTOCOL, NULL);
  2291.       return 0;
  2292.     }
  2293.     if (parse_addr_port(LOG_PROTOCOL_WARN, cell->payload+RELAY_HEADER_SIZE,
  2294.                         &address,NULL,&port)<0) {
  2295.       log_fn(LOG_PROTOCOL_WARN, LD_PROTOCOL,
  2296.              "Unable to parse addr:port in relay begin cell. Closing.");
  2297.       relay_send_end_cell_from_edge(rh.stream_id, circ,
  2298.                                     END_STREAM_REASON_TORPROTOCOL, NULL);
  2299.       return 0;
  2300.     }
  2301.     if (port==0) {
  2302.       log_fn(LOG_PROTOCOL_WARN, LD_PROTOCOL,
  2303.              "Missing port in relay begin cell. Closing.");
  2304.       relay_send_end_cell_from_edge(rh.stream_id, circ,
  2305.                                     END_STREAM_REASON_TORPROTOCOL, NULL);
  2306.       tor_free(address);
  2307.       return 0;
  2308.     }
  2309.     if (or_circ && or_circ->is_first_hop &&
  2310.         !get_options()->AllowSingleHopExits) {
  2311.       /* Don't let clients use us as a single-hop proxy, unless the user
  2312.        * has explicitly allowed that in the config.  It attracts attackers
  2313.        * and users who'd be better off with, well, single-hop proxies.
  2314.        */
  2315.       log_fn(LOG_PROTOCOL_WARN, LD_PROTOCOL,
  2316.              "Attempt to open a stream on first hop of circuit. Closing.");
  2317.       relay_send_end_cell_from_edge(rh.stream_id, circ,
  2318.                                     END_STREAM_REASON_TORPROTOCOL, NULL);
  2319.       tor_free(address);
  2320.       return 0;
  2321.     }
  2322.   } else if (rh.command == RELAY_COMMAND_BEGIN_DIR) {
  2323.     if (!directory_permits_begindir_requests(get_options()) ||
  2324.         circ->purpose != CIRCUIT_PURPOSE_OR) {
  2325.       relay_send_end_cell_from_edge(rh.stream_id, circ,
  2326.                                     END_STREAM_REASON_NOTDIRECTORY, NULL);
  2327.       return 0;
  2328.     }
  2329.     /* Make sure to get the 'real' address of the previous hop: the
  2330.      * caller might want to know whether his IP address has changed, and
  2331.      * we might already have corrected _base.addr[ess] for the relay's
  2332.      * canonical IP address. */
  2333.     if (or_circ && or_circ->p_conn)
  2334.       address = tor_dup_addr(&or_circ->p_conn->real_addr);
  2335.     else
  2336.       address = tor_strdup("127.0.0.1");
  2337.     port = 1; /* XXXX This value is never actually used anywhere, and there
  2338.                * isn't "really" a connection here.  But we
  2339.                * need to set it to something nonzero. */
  2340.   } else {
  2341.     log_warn(LD_BUG, "Got an unexpected command %d", (int)rh.command);
  2342.     relay_send_end_cell_from_edge(rh.stream_id, circ,
  2343.                                   END_STREAM_REASON_INTERNAL, NULL);
  2344.     return 0;
  2345.   }
  2346.   log_debug(LD_EXIT,"Creating new exit connection.");
  2347.   n_stream = edge_connection_new(CONN_TYPE_EXIT, AF_INET);
  2348.   n_stream->_base.purpose = EXIT_PURPOSE_CONNECT;
  2349.   n_stream->stream_id = rh.stream_id;
  2350.   n_stream->_base.port = port;
  2351.   /* leave n_stream->s at -1, because it's not yet valid */
  2352.   n_stream->package_window = STREAMWINDOW_START;
  2353.   n_stream->deliver_window = STREAMWINDOW_START;
  2354.   if (circ->purpose == CIRCUIT_PURPOSE_S_REND_JOINED) {
  2355.     origin_circuit_t *origin_circ = TO_ORIGIN_CIRCUIT(circ);
  2356.     log_info(LD_REND,"begin is for rendezvous. configuring stream.");
  2357.     n_stream->_base.address = tor_strdup("(rendezvous)");
  2358.     n_stream->_base.state = EXIT_CONN_STATE_CONNECTING;
  2359.     n_stream->rend_data = rend_data_dup(origin_circ->rend_data);
  2360.     tor_assert(connection_edge_is_rendezvous_stream(n_stream));
  2361.     assert_circuit_ok(circ);
  2362.     if (rend_service_set_connection_addr_port(n_stream, origin_circ) < 0) {
  2363.       log_info(LD_REND,"Didn't find rendezvous service (port %d)",
  2364.                n_stream->_base.port);
  2365.       relay_send_end_cell_from_edge(rh.stream_id, circ,
  2366.                                     END_STREAM_REASON_EXITPOLICY,
  2367.                                     origin_circ->cpath->prev);
  2368.       connection_free(TO_CONN(n_stream));
  2369.       tor_free(address);
  2370.       return 0;
  2371.     }
  2372.     assert_circuit_ok(circ);
  2373.     log_debug(LD_REND,"Finished assigning addr/port");
  2374.     n_stream->cpath_layer = origin_circ->cpath->prev; /* link it */
  2375.     /* add it into the linked list of n_streams on this circuit */
  2376.     n_stream->next_stream = origin_circ->p_streams;
  2377.     n_stream->on_circuit = circ;
  2378.     origin_circ->p_streams = n_stream;
  2379.     assert_circuit_ok(circ);
  2380.     connection_exit_connect(n_stream);
  2381.     tor_free(address);
  2382.     return 0;
  2383.   }
  2384.   tor_strlower(address);
  2385.   n_stream->_base.address = address;
  2386.   n_stream->_base.state = EXIT_CONN_STATE_RESOLVEFAILED;
  2387.   /* default to failed, change in dns_resolve if it turns out not to fail */
  2388.   if (we_are_hibernating()) {
  2389.     relay_send_end_cell_from_edge(rh.stream_id, circ,
  2390.                                   END_STREAM_REASON_HIBERNATING, NULL);
  2391.     connection_free(TO_CONN(n_stream));
  2392.     return 0;
  2393.   }
  2394.   n_stream->on_circuit = circ;
  2395.   if (rh.command == RELAY_COMMAND_BEGIN_DIR) {
  2396.     tor_assert(or_circ);
  2397.     if (or_circ->p_conn && !tor_addr_is_null(&or_circ->p_conn->real_addr))
  2398.       tor_addr_assign(&n_stream->_base.addr, &or_circ->p_conn->real_addr);
  2399.     return connection_exit_connect_dir(n_stream);
  2400.   }
  2401.   log_debug(LD_EXIT,"about to start the dns_resolve().");
  2402.   /* send it off to the gethostbyname farm */
  2403.   switch (dns_resolve(n_stream)) {
  2404.     case 1: /* resolve worked; now n_stream is attached to circ. */
  2405.       assert_circuit_ok(circ);
  2406.       log_debug(LD_EXIT,"about to call connection_exit_connect().");
  2407.       connection_exit_connect(n_stream);
  2408.       return 0;
  2409.     case -1: /* resolve failed */
  2410.       relay_send_end_cell_from_edge(rh.stream_id, circ,
  2411.                                     END_STREAM_REASON_RESOLVEFAILED, NULL);
  2412.       /* n_stream got freed. don't touch it. */
  2413.       break;
  2414.     case 0: /* resolve added to pending list */
  2415.       assert_circuit_ok(circ);
  2416.       break;
  2417.   }
  2418.   return 0;
  2419. }
  2420. /**
  2421.  * Called when we receive a RELAY_COMMAND_RESOLVE cell 'cell' along the
  2422.  * circuit <b>circ</b>;
  2423.  * begin resolving the hostname, and (eventually) reply with a RESOLVED cell.
  2424.  */
  2425. int
  2426. connection_exit_begin_resolve(cell_t *cell, or_circuit_t *circ)
  2427. {
  2428.   edge_connection_t *dummy_conn;
  2429.   relay_header_t rh;
  2430.   assert_circuit_ok(TO_CIRCUIT(circ));
  2431.   relay_header_unpack(&rh, cell->payload);
  2432.   /* This 'dummy_conn' only exists to remember the stream ID
  2433.    * associated with the resolve request; and to make the
  2434.    * implementation of dns.c more uniform.  (We really only need to
  2435.    * remember the circuit, the stream ID, and the hostname to be
  2436.    * resolved; but if we didn't store them in a connection like this,
  2437.    * the housekeeping in dns.c would get way more complicated.)
  2438.    */
  2439.   dummy_conn = edge_connection_new(CONN_TYPE_EXIT, AF_INET);
  2440.   dummy_conn->stream_id = rh.stream_id;
  2441.   dummy_conn->_base.address = tor_strndup(cell->payload+RELAY_HEADER_SIZE,
  2442.                                           rh.length);
  2443.   dummy_conn->_base.port = 0;
  2444.   dummy_conn->_base.state = EXIT_CONN_STATE_RESOLVEFAILED;
  2445.   dummy_conn->_base.purpose = EXIT_PURPOSE_RESOLVE;
  2446.   dummy_conn->on_circuit = TO_CIRCUIT(circ);
  2447.   /* send it off to the gethostbyname farm */
  2448.   switch (dns_resolve(dummy_conn)) {
  2449.     case -1: /* Impossible to resolve; a resolved cell was sent. */
  2450.       /* Connection freed; don't touch it. */
  2451.       return 0;
  2452.     case 1: /* The result was cached; a resolved cell was sent. */
  2453.       if (!dummy_conn->_base.marked_for_close)
  2454.         connection_free(TO_CONN(dummy_conn));
  2455.       return 0;
  2456.     case 0: /* resolve added to pending list */
  2457.       assert_circuit_ok(TO_CIRCUIT(circ));
  2458.       break;
  2459.   }
  2460.   return 0;
  2461. }
  2462. /** Connect to conn's specified addr and port. If it worked, conn
  2463.  * has now been added to the connection_array.
  2464.  *
  2465.  * Send back a connected cell. Include the resolved IP of the destination
  2466.  * address, but <em>only</em> if it's a general exit stream. (Rendezvous
  2467.  * streams must not reveal what IP they connected to.)
  2468.  */
  2469. void
  2470. connection_exit_connect(edge_connection_t *edge_conn)
  2471. {
  2472.   const tor_addr_t *addr;
  2473.   uint16_t port;
  2474.   connection_t *conn = TO_CONN(edge_conn);
  2475.   int socket_error = 0;
  2476.   if (!connection_edge_is_rendezvous_stream(edge_conn) &&
  2477.       router_compare_to_my_exit_policy(edge_conn)) {
  2478.     log_info(LD_EXIT,"%s:%d failed exit policy. Closing.",
  2479.              escaped_safe_str(conn->address), conn->port);
  2480.     connection_edge_end(edge_conn, END_STREAM_REASON_EXITPOLICY);
  2481.     circuit_detach_stream(circuit_get_by_edge_conn(edge_conn), edge_conn);
  2482.     connection_free(conn);
  2483.     return;
  2484.   }
  2485.   addr = &conn->addr;
  2486.   port = conn->port;
  2487.   log_debug(LD_EXIT,"about to try connecting");
  2488.   switch (connection_connect(conn, conn->address, addr, port, &socket_error)) {
  2489.     case -1:
  2490.       /* XXX021 use socket_error below rather than trying to piece things
  2491.        * together from the current errno, which may have been clobbered. */
  2492.       connection_edge_end_errno(edge_conn);
  2493.       circuit_detach_stream(circuit_get_by_edge_conn(edge_conn), edge_conn);
  2494.       connection_free(conn);
  2495.       return;
  2496.     case 0:
  2497.       conn->state = EXIT_CONN_STATE_CONNECTING;
  2498.       connection_watch_events(conn, EV_WRITE | EV_READ);
  2499.       /* writable indicates finish;
  2500.        * readable/error indicates broken link in windows-land. */
  2501.       return;
  2502.     /* case 1: fall through */
  2503.   }
  2504.   conn->state = EXIT_CONN_STATE_OPEN;
  2505.   if (connection_wants_to_flush(conn)) {
  2506.     /* in case there are any queued data cells */
  2507.     log_warn(LD_BUG,"newly connected conn had data waiting!");
  2508. //    connection_start_writing(conn);
  2509.   }
  2510.   connection_watch_events(conn, EV_READ);
  2511.   /* also, deliver a 'connected' cell back through the circuit. */
  2512.   if (connection_edge_is_rendezvous_stream(edge_conn)) {
  2513.     /* rendezvous stream */
  2514.     /* don't send an address back! */
  2515.     connection_edge_send_command(edge_conn,
  2516.                                  RELAY_COMMAND_CONNECTED,
  2517.                                  NULL, 0);
  2518.   } else { /* normal stream */
  2519.     char connected_payload[20];
  2520.     int connected_payload_len;
  2521.     if (tor_addr_family(&conn->addr) == AF_INET) {
  2522.       set_uint32(connected_payload, tor_addr_to_ipv4n(&conn->addr));
  2523.       connected_payload_len = 4;
  2524.     } else {
  2525.       memcpy(connected_payload, tor_addr_to_in6_addr8(&conn->addr), 16);
  2526.       connected_payload_len = 16;
  2527.     }
  2528.     set_uint32(connected_payload+connected_payload_len,
  2529.                htonl(dns_clip_ttl(edge_conn->address_ttl)));
  2530.     connected_payload_len += 4;
  2531.     connection_edge_send_command(edge_conn,
  2532.                                  RELAY_COMMAND_CONNECTED,
  2533.                                  connected_payload, connected_payload_len);
  2534.   }
  2535. }
  2536. /** Given an exit conn that should attach to us as a directory server, open a
  2537.  * bridge connection with a linked connection pair, create a new directory
  2538.  * conn, and join them together.  Return 0 on success (or if there was an
  2539.  * error we could send back an end cell for).  Return -(some circuit end
  2540.  * reason) if the circuit needs to be torn down.  Either connects
  2541.  * <b>exitconn</b>, frees it, or marks it, as appropriate.
  2542.  */
  2543. static int
  2544. connection_exit_connect_dir(edge_connection_t *exitconn)
  2545. {
  2546.   dir_connection_t *dirconn = NULL;
  2547.   or_circuit_t *circ = TO_OR_CIRCUIT(exitconn->on_circuit);
  2548.   log_info(LD_EXIT, "Opening local connection for anonymized directory exit");
  2549.   exitconn->_base.state = EXIT_CONN_STATE_OPEN;
  2550.   dirconn = dir_connection_new(AF_INET);
  2551.   tor_addr_assign(&dirconn->_base.addr, &exitconn->_base.addr);
  2552.   dirconn->_base.port = 0;
  2553.   dirconn->_base.address = tor_strdup(exitconn->_base.address);
  2554.   dirconn->_base.type = CONN_TYPE_DIR;
  2555.   dirconn->_base.purpose = DIR_PURPOSE_SERVER;
  2556.   dirconn->_base.state = DIR_CONN_STATE_SERVER_COMMAND_WAIT;
  2557.   connection_link_connections(TO_CONN(dirconn), TO_CONN(exitconn));
  2558.   if (connection_add(TO_CONN(exitconn))<0) {
  2559.     connection_edge_end(exitconn, END_STREAM_REASON_RESOURCELIMIT);
  2560.     connection_free(TO_CONN(exitconn));
  2561.     connection_free(TO_CONN(dirconn));
  2562.     return 0;
  2563.   }
  2564.   /* link exitconn to circ, now that we know we can use it. */
  2565.   exitconn->next_stream = circ->n_streams;
  2566.   circ->n_streams = exitconn;
  2567.   if (connection_add(TO_CONN(dirconn))<0) {
  2568.     connection_edge_end(exitconn, END_STREAM_REASON_RESOURCELIMIT);
  2569.     connection_close_immediate(TO_CONN(exitconn));
  2570.     connection_mark_for_close(TO_CONN(exitconn));
  2571.     connection_free(TO_CONN(dirconn));
  2572.     return 0;
  2573.   }
  2574.   connection_start_reading(TO_CONN(dirconn));
  2575.   connection_start_reading(TO_CONN(exitconn));
  2576.   if (connection_edge_send_command(exitconn,
  2577.                                    RELAY_COMMAND_CONNECTED, NULL, 0) < 0) {
  2578.     connection_mark_for_close(TO_CONN(exitconn));
  2579.     connection_mark_for_close(TO_CONN(dirconn));
  2580.     return 0;
  2581.   }
  2582.   return 0;
  2583. }
  2584. /** Return 1 if <b>conn</b> is a rendezvous stream, or 0 if
  2585.  * it is a general stream.
  2586.  */
  2587. int
  2588. connection_edge_is_rendezvous_stream(edge_connection_t *conn)
  2589. {
  2590.   tor_assert(conn);
  2591.   if (conn->rend_data)
  2592.     return 1;
  2593.   return 0;
  2594. }
  2595. /** Return 1 if router <b>exit</b> is likely to allow stream <b>conn</b>
  2596.  * to exit from it, or 0 if it probably will not allow it.
  2597.  * (We might be uncertain if conn's destination address has not yet been
  2598.  * resolved.)
  2599.  */
  2600. int
  2601. connection_ap_can_use_exit(edge_connection_t *conn, routerinfo_t *exit)
  2602. {
  2603.   tor_assert(conn);
  2604.   tor_assert(conn->_base.type == CONN_TYPE_AP);
  2605.   tor_assert(conn->socks_request);
  2606.   tor_assert(exit);
  2607.   /* If a particular exit node has been requested for the new connection,
  2608.    * make sure the exit node of the existing circuit matches exactly.
  2609.    */
  2610.   if (conn->chosen_exit_name) {
  2611.     routerinfo_t *chosen_exit =
  2612.       router_get_by_nickname(conn->chosen_exit_name, 1);
  2613.     if (!chosen_exit || memcmp(chosen_exit->cache_info.identity_digest,
  2614.                                exit->cache_info.identity_digest, DIGEST_LEN)) {
  2615.       /* doesn't match */
  2616. //      log_debug(LD_APP,"Requested node '%s', considering node '%s'. No.",
  2617. //                conn->chosen_exit_name, exit->nickname);
  2618.       return 0;
  2619.     }
  2620.   }
  2621.   if (conn->socks_request->command == SOCKS_COMMAND_CONNECT &&
  2622.       !conn->use_begindir) {
  2623.     struct in_addr in;
  2624.     uint32_t addr = 0;
  2625.     addr_policy_result_t r;
  2626.     if (tor_inet_aton(conn->socks_request->address, &in))
  2627.       addr = ntohl(in.s_addr);
  2628.     r = compare_addr_to_addr_policy(addr, conn->socks_request->port,
  2629.                                     exit->exit_policy);
  2630.     if (r == ADDR_POLICY_REJECTED)
  2631.       return 0; /* We know the address, and the exit policy rejects it. */
  2632.     if (r == ADDR_POLICY_PROBABLY_REJECTED && !conn->chosen_exit_name)
  2633.       return 0; /* We don't know the addr, but the exit policy rejects most
  2634.                  * addresses with this port. Since the user didn't ask for
  2635.                  * this node, err on the side of caution. */
  2636.   } else if (SOCKS_COMMAND_IS_RESOLVE(conn->socks_request->command)) {
  2637.     /* Can't support reverse lookups without eventdns. */
  2638.     if (conn->socks_request->command == SOCKS_COMMAND_RESOLVE_PTR &&
  2639.         exit->has_old_dnsworkers)
  2640.       return 0;
  2641.     /* Don't send DNS requests to non-exit servers by default. */
  2642.     if (!conn->chosen_exit_name && policy_is_reject_star(exit->exit_policy))
  2643.       return 0;
  2644.   }
  2645.   return 1;
  2646. }
  2647. /** If address is of the form "y.onion" with a well-formed handle y:
  2648.  *     Put a NUL after y, lower-case it, and return ONION_HOSTNAME.
  2649.  *
  2650.  * If address is of the form "y.exit":
  2651.  *     Put a NUL after y and return EXIT_HOSTNAME.
  2652.  *
  2653.  * Otherwise:
  2654.  *     Return NORMAL_HOSTNAME and change nothing.
  2655.  */
  2656. hostname_type_t
  2657. parse_extended_hostname(char *address)
  2658. {
  2659.     char *s;
  2660.     char query[REND_SERVICE_ID_LEN_BASE32+1];
  2661.     s = strrchr(address,'.');
  2662.     if (!s)
  2663.       return NORMAL_HOSTNAME; /* no dot, thus normal */
  2664.     if (!strcmp(s+1,"exit")) {
  2665.       *s = 0; /* NUL-terminate it */
  2666.       return EXIT_HOSTNAME; /* .exit */
  2667.     }
  2668.     if (strcmp(s+1,"onion"))
  2669.       return NORMAL_HOSTNAME; /* neither .exit nor .onion, thus normal */
  2670.     /* so it is .onion */
  2671.     *s = 0; /* NUL-terminate it */
  2672.     if (strlcpy(query, address, REND_SERVICE_ID_LEN_BASE32+1) >=
  2673.         REND_SERVICE_ID_LEN_BASE32+1)
  2674.       goto failed;
  2675.     if (rend_valid_service_id(query)) {
  2676.       return ONION_HOSTNAME; /* success */
  2677.     }
  2678. failed:
  2679.     /* otherwise, return to previous state and return 0 */
  2680.     *s = '.';
  2681.     return BAD_HOSTNAME;
  2682. }
  2683. /** Check if the address is of the form "y.noconnect"
  2684.  */
  2685. int
  2686. hostname_is_noconnect_address(const char *address)
  2687. {
  2688.   return ! strcasecmpend(address, ".noconnect");
  2689. }