123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134 |
- /* Copyright (c) 2014-2015, The Tor Project, Inc. */
- /* See LICENSE for licensing information */
- #define ADDRESS_PRIVATE
- #include "orconfig.h"
- #ifdef _WIN32
- #include <winsock2.h>
- /* For access to structs needed by GetAdaptersAddresses */
- #include <iphlpapi.h>
- #endif
- #ifdef HAVE_IFADDRS_TO_SMARTLIST
- #include <net/if.h>
- #include <ifaddrs.h>
- #endif
- #ifdef HAVE_IFCONF_TO_SMARTLIST
- #ifdef HAVE_SYS_IOCTL_H
- #include <sys/ioctl.h>
- #endif
- #include <net/if.h>
- #endif
- #include "or.h"
- #include "address.h"
- #include "test.h"
- /** Return 1 iff <b>sockaddr1</b> and <b>sockaddr2</b> represent
- * the same IP address and port combination. Otherwise, return 0.
- */
- static uint8_t
- sockaddr_in_are_equal(struct sockaddr_in *sockaddr1,
- struct sockaddr_in *sockaddr2)
- {
- return ((sockaddr1->sin_family == sockaddr2->sin_family) &&
- (sockaddr1->sin_port == sockaddr2->sin_port) &&
- (sockaddr1->sin_addr.s_addr == sockaddr2->sin_addr.s_addr));
- }
- /** Return 1 iff <b>sockaddr1</b> and <b>sockaddr2</b> represent
- * the same IP address and port combination. Otherwise, return 0.
- */
- static uint8_t
- sockaddr_in6_are_equal(struct sockaddr_in6 *sockaddr1,
- struct sockaddr_in6 *sockaddr2)
- {
- return ((sockaddr1->sin6_family == sockaddr2->sin6_family) &&
- (sockaddr1->sin6_port == sockaddr2->sin6_port) &&
- (tor_memeq(sockaddr1->sin6_addr.s6_addr,
- sockaddr2->sin6_addr.s6_addr,16)));
- }
- /** Create a sockaddr_in structure from IP address string <b>ip_str</b>.
- *
- * If <b>out</b> is not NULL, write the result
- * to the memory address in <b>out</b>. Otherwise, allocate the memory
- * for result. On success, return pointer to result. Otherwise, return
- * NULL.
- */
- static struct sockaddr_in *
- sockaddr_in_from_string(const char *ip_str, struct sockaddr_in *out)
- {
- // [FIXME: add some error checking?]
- if (!out)
- out = tor_malloc_zero(sizeof(struct sockaddr_in));
- out->sin_family = AF_INET;
- out->sin_port = 0;
- tor_inet_pton(AF_INET,ip_str,&(out->sin_addr));
- return out;
- }
- /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
- * that is an IPv4 or IPv6 localhost address. Otherwise, return 0.
- */
- static int
- smartlist_contains_localhost_tor_addr(smartlist_t *smartlist)
- {
- SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
- if (tor_addr_is_loopback(tor_addr)) {
- return 1;
- }
- } SMARTLIST_FOREACH_END(tor_addr);
- return 0;
- }
- /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
- * that is an IPv4 or IPv6 multicast address. Otherwise, return 0.
- */
- static int
- smartlist_contains_multicast_tor_addr(smartlist_t *smartlist)
- {
- SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
- if (tor_addr_is_multicast(tor_addr)) {
- return 1;
- }
- } SMARTLIST_FOREACH_END(tor_addr);
- return 0;
- }
- /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
- * that is an IPv4 or IPv6 internal address. Otherwise, return 0.
- */
- static int
- smartlist_contains_internal_tor_addr(smartlist_t *smartlist)
- {
- SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
- if (tor_addr_is_internal(tor_addr, 0)) {
- return 1;
- }
- } SMARTLIST_FOREACH_END(tor_addr);
- return 0;
- }
- /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
- * that is NULL or the null tor_addr_t. Otherwise, return 0.
- */
- static int
- smartlist_contains_null_tor_addr(smartlist_t *smartlist)
- {
- SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
- if (tor_addr == NULL || tor_addr_is_null(tor_addr)) {
- return 1;
- }
- } SMARTLIST_FOREACH_END(tor_addr);
- return 0;
- }
- /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
- * that is an IPv4 address. Otherwise, return 0.
- */
- static int
- smartlist_contains_ipv4_tor_addr(smartlist_t *smartlist)
- {
- SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
- if (tor_addr_is_v4(tor_addr)) {
- return 1;
- }
- } SMARTLIST_FOREACH_END(tor_addr);
- return 0;
- }
- /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
- * that is an IPv6 address. Otherwise, return 0.
- */
- static int
- smartlist_contains_ipv6_tor_addr(smartlist_t *smartlist)
- {
- SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
- /* Since there's no tor_addr_is_v6, assume all non-v4s are v6 */
- if (!tor_addr_is_v4(tor_addr)) {
- return 1;
- }
- } SMARTLIST_FOREACH_END(tor_addr);
- return 0;
- }
- #ifdef HAVE_IFADDRS_TO_SMARTLIST
- static void
- test_address_ifaddrs_to_smartlist(void *arg)
- {
- struct ifaddrs *ifa = NULL;
- struct ifaddrs *ifa_ipv4 = NULL;
- struct ifaddrs *ifa_ipv6 = NULL;
- struct sockaddr_in *ipv4_sockaddr_local = NULL;
- struct sockaddr_in *netmask_slash8 = NULL;
- struct sockaddr_in *ipv4_sockaddr_remote = NULL;
- struct sockaddr_in6 *ipv6_sockaddr = NULL;
- smartlist_t *smartlist = NULL;
- tor_addr_t *tor_addr = NULL;
- struct sockaddr *sockaddr_to_check = NULL;
- socklen_t addr_len;
- (void)arg;
- netmask_slash8 = sockaddr_in_from_string("255.0.0.0",NULL);
- ipv4_sockaddr_local = sockaddr_in_from_string("127.0.0.1",NULL);
- ipv4_sockaddr_remote = sockaddr_in_from_string("128.52.160.20",NULL);
- ipv6_sockaddr = tor_malloc(sizeof(struct sockaddr_in6));
- ipv6_sockaddr->sin6_family = AF_INET6;
- ipv6_sockaddr->sin6_port = 0;
- tor_inet_pton(AF_INET6, "2001:db8:8714:3a90::12",
- &(ipv6_sockaddr->sin6_addr));
- ifa = tor_malloc(sizeof(struct ifaddrs));
- ifa_ipv4 = tor_malloc(sizeof(struct ifaddrs));
- ifa_ipv6 = tor_malloc(sizeof(struct ifaddrs));
- ifa->ifa_next = ifa_ipv4;
- ifa->ifa_name = tor_strdup("eth0");
- ifa->ifa_flags = IFF_UP | IFF_RUNNING;
- ifa->ifa_addr = (struct sockaddr *)ipv4_sockaddr_local;
- ifa->ifa_netmask = (struct sockaddr *)netmask_slash8;
- ifa->ifa_dstaddr = NULL;
- ifa->ifa_data = NULL;
- ifa_ipv4->ifa_next = ifa_ipv6;
- ifa_ipv4->ifa_name = tor_strdup("eth1");
- ifa_ipv4->ifa_flags = IFF_UP | IFF_RUNNING;
- ifa_ipv4->ifa_addr = (struct sockaddr *)ipv4_sockaddr_remote;
- ifa_ipv4->ifa_netmask = (struct sockaddr *)netmask_slash8;
- ifa_ipv4->ifa_dstaddr = NULL;
- ifa_ipv4->ifa_data = NULL;
- ifa_ipv6->ifa_next = NULL;
- ifa_ipv6->ifa_name = tor_strdup("eth2");
- ifa_ipv6->ifa_flags = IFF_UP | IFF_RUNNING;
- ifa_ipv6->ifa_addr = (struct sockaddr *)ipv6_sockaddr;
- ifa_ipv6->ifa_netmask = NULL;
- ifa_ipv6->ifa_dstaddr = NULL;
- ifa_ipv6->ifa_data = NULL;
- smartlist = ifaddrs_to_smartlist(ifa);
- tt_assert(smartlist);
- tt_assert(smartlist_len(smartlist) == 3);
- sockaddr_to_check = tor_malloc(sizeof(struct sockaddr_in6));
- tor_addr = smartlist_get(smartlist,0);
- addr_len =
- tor_addr_to_sockaddr(tor_addr,0,sockaddr_to_check,
- sizeof(struct sockaddr_in));
- tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
- tt_assert(sockaddr_in_are_equal((struct sockaddr_in *)sockaddr_to_check,
- ipv4_sockaddr_local));
- tor_addr = smartlist_get(smartlist,1);
- addr_len =
- tor_addr_to_sockaddr(tor_addr,0,sockaddr_to_check,
- sizeof(struct sockaddr_in));
- tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
- tt_assert(sockaddr_in_are_equal((struct sockaddr_in *)sockaddr_to_check,
- ipv4_sockaddr_remote));
- tor_addr = smartlist_get(smartlist,2);
- addr_len =
- tor_addr_to_sockaddr(tor_addr,0,sockaddr_to_check,
- sizeof(struct sockaddr_in6));
- tt_int_op(addr_len,==,sizeof(struct sockaddr_in6));
- tt_assert(sockaddr_in6_are_equal((struct sockaddr_in6*)sockaddr_to_check,
- ipv6_sockaddr));
- done:
- tor_free(netmask_slash8);
- tor_free(ipv4_sockaddr_local);
- tor_free(ipv4_sockaddr_remote);
- tor_free(ipv6_sockaddr);
- tor_free(ifa->ifa_name);
- tor_free(ifa_ipv4->ifa_name);
- tor_free(ifa_ipv6->ifa_name);
- tor_free(ifa);
- tor_free(ifa_ipv4);
- tor_free(ifa_ipv6);
- tor_free(sockaddr_to_check);
- if (smartlist) {
- SMARTLIST_FOREACH(smartlist, tor_addr_t *, t, tor_free(t));
- smartlist_free(smartlist);
- }
- return;
- }
- static void
- test_address_get_if_addrs_ifaddrs(void *arg)
- {
- smartlist_t *results = NULL;
- (void)arg;
- results = get_interface_addresses_ifaddrs(LOG_ERR);
- tt_assert(results);
- /* Some FreeBSD jails don't have localhost IP address. Instead, they only
- * have the address assigned to the jail (whatever that may be).
- * And a jail without a network connection might not have any addresses at
- * all. */
- tt_assert(!smartlist_contains_null_tor_addr(results));
- /* If there are addresses, they must be IPv4 or IPv6 */
- if (smartlist_len(results) > 0) {
- tt_assert(smartlist_contains_ipv4_tor_addr(results)
- || smartlist_contains_ipv6_tor_addr(results));
- }
- done:
- SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
- smartlist_free(results);
- return;
- }
- #endif
- #ifdef HAVE_IP_ADAPTER_TO_SMARTLIST
- static void
- test_address_get_if_addrs_win32(void *arg)
- {
- smartlist_t *results = NULL;
- (void)arg;
- results = get_interface_addresses_win32(LOG_ERR);
- tt_int_op(smartlist_len(results),>=,1);
- tt_assert(smartlist_contains_localhost_tor_addr(results));
- tt_assert(!smartlist_contains_null_tor_addr(results));
- /* If there are addresses, they must be IPv4 or IPv6 */
- if (smartlist_len(results) > 0) {
- tt_assert(smartlist_contains_ipv4_tor_addr(results)
- || smartlist_contains_ipv6_tor_addr(results));
- }
- done:
- SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
- tor_free(results);
- return;
- }
- static void
- test_address_ip_adapter_addresses_to_smartlist(void *arg)
- {
- IP_ADAPTER_ADDRESSES *addrs1;
- IP_ADAPTER_ADDRESSES *addrs2;
- IP_ADAPTER_UNICAST_ADDRESS *unicast11;
- IP_ADAPTER_UNICAST_ADDRESS *unicast12;
- IP_ADAPTER_UNICAST_ADDRESS *unicast21;
- smartlist_t *result = NULL;
- struct sockaddr_in *sockaddr_test1;
- struct sockaddr_in *sockaddr_test2;
- struct sockaddr_in *sockaddr_localhost;
- struct sockaddr_in *sockaddr_to_check;
- tor_addr_t *tor_addr;
- (void)arg;
- (void)sockaddr_in6_are_equal;
- sockaddr_to_check = tor_malloc_zero(sizeof(struct sockaddr_in));
- addrs1 =
- tor_malloc_zero(sizeof(IP_ADAPTER_ADDRESSES));
- addrs1->FirstUnicastAddress =
- unicast11 = tor_malloc_zero(sizeof(IP_ADAPTER_UNICAST_ADDRESS));
- sockaddr_test1 = sockaddr_in_from_string("86.59.30.40",NULL);
- unicast11->Address.lpSockaddr = (LPSOCKADDR)sockaddr_test1;
- unicast11->Next = unicast12 =
- tor_malloc_zero(sizeof(IP_ADAPTER_UNICAST_ADDRESS));
- sockaddr_test2 = sockaddr_in_from_string("93.95.227.222", NULL);
- unicast12->Address.lpSockaddr = (LPSOCKADDR)sockaddr_test2;
- addrs1->Next = addrs2 =
- tor_malloc_zero(sizeof(IP_ADAPTER_ADDRESSES));
- addrs2->FirstUnicastAddress =
- unicast21 = tor_malloc_zero(sizeof(IP_ADAPTER_UNICAST_ADDRESS));
- sockaddr_localhost = sockaddr_in_from_string("127.0.0.1", NULL);
- unicast21->Address.lpSockaddr = (LPSOCKADDR)sockaddr_localhost;
- result = ip_adapter_addresses_to_smartlist(addrs1);
- tt_assert(result);
- tt_assert(smartlist_len(result) == 3);
- tor_addr = smartlist_get(result,0);
- tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
- sizeof(struct sockaddr_in));
- tt_assert(sockaddr_in_are_equal(sockaddr_test1,sockaddr_to_check));
- tor_addr = smartlist_get(result,1);
- tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
- sizeof(struct sockaddr_in));
- tt_assert(sockaddr_in_are_equal(sockaddr_test2,sockaddr_to_check));
- tor_addr = smartlist_get(result,2);
- tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
- sizeof(struct sockaddr_in));
- tt_assert(sockaddr_in_are_equal(sockaddr_localhost,sockaddr_to_check));
- done:
- SMARTLIST_FOREACH(result, tor_addr_t *, t, tor_free(t));
- smartlist_free(result);
- tor_free(addrs1);
- tor_free(addrs2);
- tor_free(unicast11->Address.lpSockaddr);
- tor_free(unicast11);
- tor_free(unicast12->Address.lpSockaddr);
- tor_free(unicast12);
- tor_free(unicast21->Address.lpSockaddr);
- tor_free(unicast21);
- tor_free(sockaddr_to_check);
- return;
- }
- #endif
- #ifdef HAVE_IFCONF_TO_SMARTLIST
- static void
- test_address_ifreq_to_smartlist(void *arg)
- {
- smartlist_t *results = NULL;
- const tor_addr_t *tor_addr = NULL;
- struct sockaddr_in *sockaddr = NULL;
- struct sockaddr_in *sockaddr_eth1 = NULL;
- struct sockaddr_in *sockaddr_to_check = NULL;
- struct ifconf *ifc;
- struct ifreq *ifr;
- struct ifreq *ifr_next;
- socklen_t addr_len;
- (void)arg;
- sockaddr_to_check = tor_malloc(sizeof(struct sockaddr_in));
- ifr = tor_malloc(sizeof(struct ifreq));
- memset(ifr,0,sizeof(struct ifreq));
- strlcpy(ifr->ifr_name,"lo",3);
- sockaddr = (struct sockaddr_in *) &(ifr->ifr_ifru.ifru_addr);
- sockaddr_in_from_string("127.0.0.1",sockaddr);
- ifc = tor_malloc(sizeof(struct ifconf));
- memset(ifc,0,sizeof(struct ifconf));
- ifc->ifc_len = sizeof(struct ifreq);
- ifc->ifc_ifcu.ifcu_req = ifr;
- results = ifreq_to_smartlist(ifc->ifc_buf,ifc->ifc_len);
- tt_int_op(smartlist_len(results),==,1);
- tor_addr = smartlist_get(results, 0);
- addr_len =
- tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
- sizeof(struct sockaddr_in));
- tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
- tt_assert(sockaddr_in_are_equal(sockaddr,sockaddr_to_check));
- ifr = tor_realloc(ifr,2*sizeof(struct ifreq));
- ifr_next = ifr+1;
- strlcpy(ifr_next->ifr_name,"eth1",5);
- ifc->ifc_len = 2*sizeof(struct ifreq);
- ifc->ifc_ifcu.ifcu_req = ifr;
- sockaddr = (struct sockaddr_in *) &(ifr->ifr_ifru.ifru_addr);
- sockaddr_eth1 = (struct sockaddr_in *) &(ifr_next->ifr_ifru.ifru_addr);
- sockaddr_in_from_string("192.168.10.55",sockaddr_eth1);
- SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
- smartlist_free(results);
- results = ifreq_to_smartlist(ifc->ifc_buf,ifc->ifc_len);
- tt_int_op(smartlist_len(results),==,2);
- tor_addr = smartlist_get(results, 0);
- addr_len =
- tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
- sizeof(struct sockaddr_in));
- tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
- tt_assert(sockaddr_in_are_equal(sockaddr,sockaddr_to_check));
- tor_addr = smartlist_get(results, 1);
- addr_len =
- tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
- sizeof(struct sockaddr_in));
- tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
- tt_assert(sockaddr_in_are_equal(sockaddr_eth1,sockaddr_to_check));
- done:
- tor_free(sockaddr_to_check);
- SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
- smartlist_free(results);
- tor_free(ifc);
- tor_free(ifr);
- return;
- }
- static void
- test_address_get_if_addrs_ioctl(void *arg)
- {
- smartlist_t *result = NULL;
- (void)arg;
- result = get_interface_addresses_ioctl(LOG_ERR);
- /* On an IPv6-only system, this will fail and return NULL
- tt_assert(result);
- */
- /* Some FreeBSD jails don't have localhost IP address. Instead, they only
- * have the address assigned to the jail (whatever that may be).
- * And a jail without a network connection might not have any addresses at
- * all. */
- if (result) {
- tt_assert(!smartlist_contains_null_tor_addr(result));
- /* If there are addresses, they must be IPv4 or IPv6.
- * (AIX supports IPv6 from SIOCGIFCONF.) */
- if (smartlist_len(result) > 0) {
- tt_assert(smartlist_contains_ipv4_tor_addr(result)
- || smartlist_contains_ipv6_tor_addr(result));
- }
- }
- done:
- if (result) {
- SMARTLIST_FOREACH(result, tor_addr_t *, t, tor_free(t));
- smartlist_free(result);
- }
- return;
- }
- #endif
- #define FAKE_SOCKET_FD (42)
- static tor_socket_t
- fake_open_socket(int domain, int type, int protocol)
- {
- (void)domain;
- (void)type;
- (void)protocol;
- return FAKE_SOCKET_FD;
- }
- static int last_connected_socket_fd = 0;
- static int connect_retval = 0;
- static tor_socket_t
- pretend_to_connect(tor_socket_t socket, const struct sockaddr *address,
- socklen_t address_len)
- {
- (void)address;
- (void)address_len;
- last_connected_socket_fd = socket;
- return connect_retval;
- }
- static struct sockaddr *mock_addr = NULL;
- static int
- fake_getsockname(tor_socket_t socket, struct sockaddr *address,
- socklen_t *address_len)
- {
- socklen_t bytes_to_copy = 0;
- (void) socket;
- if (!mock_addr)
- return -1;
- if (mock_addr->sa_family == AF_INET) {
- bytes_to_copy = sizeof(struct sockaddr_in);
- } else if (mock_addr->sa_family == AF_INET6) {
- bytes_to_copy = sizeof(struct sockaddr_in6);
- } else {
- return -1;
- }
- if (*address_len < bytes_to_copy) {
- return -1;
- }
- memcpy(address,mock_addr,bytes_to_copy);
- *address_len = bytes_to_copy;
- return 0;
- }
- static void
- test_address_udp_socket_trick_whitebox(void *arg)
- {
- int hack_retval;
- tor_addr_t *addr_from_hack = tor_malloc_zero(sizeof(tor_addr_t));
- struct sockaddr_in6 *mock_addr6;
- struct sockaddr_in6 *ipv6_to_check =
- tor_malloc_zero(sizeof(struct sockaddr_in6));
- (void)arg;
- MOCK(tor_open_socket,fake_open_socket);
- MOCK(tor_connect_socket,pretend_to_connect);
- MOCK(tor_getsockname,fake_getsockname);
- mock_addr = tor_malloc_zero(sizeof(struct sockaddr_storage));
- sockaddr_in_from_string("23.32.246.118",(struct sockaddr_in *)mock_addr);
- hack_retval =
- get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
- AF_INET, addr_from_hack);
- tt_int_op(hack_retval,==,0);
- tt_assert(tor_addr_eq_ipv4h(addr_from_hack, 0x1720f676));
- /* Now, lets do an IPv6 case. */
- memset(mock_addr,0,sizeof(struct sockaddr_storage));
- mock_addr6 = (struct sockaddr_in6 *)mock_addr;
- mock_addr6->sin6_family = AF_INET6;
- mock_addr6->sin6_port = 0;
- tor_inet_pton(AF_INET6,"2001:cdba::3257:9652",&(mock_addr6->sin6_addr));
- hack_retval =
- get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
- AF_INET6, addr_from_hack);
- tt_int_op(hack_retval,==,0);
- tor_addr_to_sockaddr(addr_from_hack,0,(struct sockaddr *)ipv6_to_check,
- sizeof(struct sockaddr_in6));
- tt_assert(sockaddr_in6_are_equal(mock_addr6,ipv6_to_check));
- UNMOCK(tor_open_socket);
- UNMOCK(tor_connect_socket);
- UNMOCK(tor_getsockname);
- done:
- tor_free(ipv6_to_check);
- tor_free(mock_addr);
- tor_free(addr_from_hack);
- return;
- }
- static void
- test_address_udp_socket_trick_blackbox(void *arg)
- {
- /* We want get_interface_address6_via_udp_socket_hack() to yield
- * the same valid address that get_interface_address6() returns.
- * If the latter is unable to find a valid address, we want
- * _hack() to fail and return-1.
- *
- * Furthermore, we want _hack() never to crash, even if
- * get_interface_addresses_raw() is returning NULL.
- */
- tor_addr_t addr4;
- tor_addr_t addr4_to_check;
- tor_addr_t addr6;
- tor_addr_t addr6_to_check;
- int retval, retval_reference;
- (void)arg;
- #if 0
- retval_reference = get_interface_address6(LOG_DEBUG,AF_INET,&addr4);
- retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
- AF_INET,
- &addr4_to_check);
- tt_int_op(retval,==,retval_reference);
- tt_assert( (retval == -1 && retval_reference == -1) ||
- (tor_addr_compare(&addr4,&addr4_to_check,CMP_EXACT) == 0) );
- retval_reference = get_interface_address6(LOG_DEBUG,AF_INET6,&addr6);
- retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
- AF_INET6,
- &addr6_to_check);
- tt_int_op(retval,==,retval_reference);
- tt_assert( (retval == -1 && retval_reference == -1) ||
- (tor_addr_compare(&addr6,&addr6_to_check,CMP_EXACT) == 0) );
- #else
- /* Both of the blackbox test cases fail horribly if:
- * * The host has no external addreses.
- * * There are multiple interfaces with either AF_INET or AF_INET6.
- * * The last address isn't the one associated with the default route.
- *
- * The tests SHOULD be re-enabled when #12377 is fixed correctly, but till
- * then this fails a lot, in situations we expect failures due to knowing
- * about the code being broken.
- */
- (void)addr4_to_check;
- (void)addr6_to_check;
- (void)addr6;
- (void) retval_reference;
- #endif
- /* When family is neither AF_INET nor AF_INET6, we want _hack to
- * fail and return -1.
- */
- retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
- AF_INET+AF_INET6,&addr4);
- tt_assert(retval == -1);
- done:
- return;
- }
- static void
- test_address_get_if_addrs_list_internal(void *arg)
- {
- smartlist_t *results = NULL;
- (void)arg;
- results = get_interface_address_list(LOG_ERR, 1);
- tt_assert(results != NULL);
- /* When the network is down, a system might not have any non-local
- * non-multicast addresseses, not even internal ones.
- * Unit tests shouldn't fail because of this. */
- tt_int_op(smartlist_len(results),>=,0);
- tt_assert(!smartlist_contains_localhost_tor_addr(results));
- tt_assert(!smartlist_contains_multicast_tor_addr(results));
- /* The list may or may not contain internal addresses */
- tt_assert(!smartlist_contains_null_tor_addr(results));
- /* if there are any addresses, they must be IPv4 */
- if (smartlist_len(results) > 0) {
- tt_assert(smartlist_contains_ipv4_tor_addr(results));
- }
- tt_assert(!smartlist_contains_ipv6_tor_addr(results));
- done:
- free_interface_address_list(results);
- return;
- }
- static void
- test_address_get_if_addrs_list_no_internal(void *arg)
- {
- smartlist_t *results = NULL;
- (void)arg;
- results = get_interface_address_list(LOG_ERR, 0);
- tt_assert(results != NULL);
- /* Work even on systems with only internal IPv4 addresses */
- tt_int_op(smartlist_len(results),>=,0);
- tt_assert(!smartlist_contains_localhost_tor_addr(results));
- tt_assert(!smartlist_contains_multicast_tor_addr(results));
- tt_assert(!smartlist_contains_internal_tor_addr(results));
- tt_assert(!smartlist_contains_null_tor_addr(results));
- /* if there are any addresses, they must be IPv4 */
- if (smartlist_len(results) > 0) {
- tt_assert(smartlist_contains_ipv4_tor_addr(results));
- }
- tt_assert(!smartlist_contains_ipv6_tor_addr(results));
- done:
- free_interface_address_list(results);
- return;
- }
- static void
- test_address_get_if_addrs6_list_internal(void *arg)
- {
- smartlist_t *results = NULL;
- (void)arg;
- results = get_interface_address6_list(LOG_ERR, AF_INET6, 1);
- tt_assert(results != NULL);
- /* Work even on systems without IPv6 interfaces */
- tt_int_op(smartlist_len(results),>=,0);
- tt_assert(!smartlist_contains_localhost_tor_addr(results));
- tt_assert(!smartlist_contains_multicast_tor_addr(results));
- /* The list may or may not contain internal addresses */
- tt_assert(!smartlist_contains_null_tor_addr(results));
- /* if there are any addresses, they must be IPv6 */
- tt_assert(!smartlist_contains_ipv4_tor_addr(results));
- if (smartlist_len(results) > 0) {
- tt_assert(smartlist_contains_ipv6_tor_addr(results));
- }
- done:
- free_interface_address6_list(results);
- return;
- }
- static void
- test_address_get_if_addrs6_list_no_internal(void *arg)
- {
- smartlist_t *results = NULL;
- (void)arg;
- results = get_interface_address6_list(LOG_ERR, AF_INET6, 0);
- tt_assert(results != NULL);
- /* Work even on systems without IPv6 interfaces */
- tt_int_op(smartlist_len(results),>=,0);
- tt_assert(!smartlist_contains_localhost_tor_addr(results));
- tt_assert(!smartlist_contains_multicast_tor_addr(results));
- tt_assert(!smartlist_contains_internal_tor_addr(results));
- tt_assert(!smartlist_contains_null_tor_addr(results));
- /* if there are any addresses, they must be IPv6 */
- tt_assert(!smartlist_contains_ipv4_tor_addr(results));
- if (smartlist_len(results) > 0) {
- tt_assert(smartlist_contains_ipv6_tor_addr(results));
- }
- done:
- free_interface_address6_list(results);
- return;
- }
- static int called_get_interface_addresses_raw = 0;
- static smartlist_t *
- mock_get_interface_addresses_raw_fail(int severity)
- {
- (void)severity;
- called_get_interface_addresses_raw++;
- return smartlist_new();
- }
- static int called_get_interface_address6_via_udp_socket_hack = 0;
- static int
- mock_get_interface_address6_via_udp_socket_hack_fail(int severity,
- sa_family_t family,
- tor_addr_t *addr)
- {
- (void)severity;
- (void)family;
- (void)addr;
- called_get_interface_address6_via_udp_socket_hack++;
- return -1;
- }
- static void
- test_address_get_if_addrs_internal_fail(void *arg)
- {
- smartlist_t *results1 = NULL, *results2 = NULL;
- int rv = 0;
- uint32_t ipv4h_addr = 0;
- tor_addr_t ipv6_addr;
- memset(&ipv6_addr, 0, sizeof(tor_addr_t));
- (void)arg;
- MOCK(get_interface_addresses_raw,
- mock_get_interface_addresses_raw_fail);
- MOCK(get_interface_address6_via_udp_socket_hack,
- mock_get_interface_address6_via_udp_socket_hack_fail);
- results1 = get_interface_address6_list(LOG_ERR, AF_INET6, 1);
- tt_assert(results1 != NULL);
- tt_int_op(smartlist_len(results1),==,0);
- results2 = get_interface_address_list(LOG_ERR, 1);
- tt_assert(results2 != NULL);
- tt_int_op(smartlist_len(results2),==,0);
- rv = get_interface_address6(LOG_ERR, AF_INET6, &ipv6_addr);
- tt_assert(rv == -1);
- rv = get_interface_address(LOG_ERR, &ipv4h_addr);
- tt_assert(rv == -1);
- done:
- UNMOCK(get_interface_addresses_raw);
- UNMOCK(get_interface_address6_via_udp_socket_hack);
- free_interface_address6_list(results1);
- free_interface_address6_list(results2);
- return;
- }
- static void
- test_address_get_if_addrs_no_internal_fail(void *arg)
- {
- smartlist_t *results1 = NULL, *results2 = NULL;
- (void)arg;
- MOCK(get_interface_addresses_raw,
- mock_get_interface_addresses_raw_fail);
- MOCK(get_interface_address6_via_udp_socket_hack,
- mock_get_interface_address6_via_udp_socket_hack_fail);
- results1 = get_interface_address6_list(LOG_ERR, AF_INET6, 0);
- tt_assert(results1 != NULL);
- tt_int_op(smartlist_len(results1),==,0);
- results2 = get_interface_address_list(LOG_ERR, 0);
- tt_assert(results2 != NULL);
- tt_int_op(smartlist_len(results2),==,0);
- done:
- UNMOCK(get_interface_addresses_raw);
- UNMOCK(get_interface_address6_via_udp_socket_hack);
- free_interface_address6_list(results1);
- free_interface_address6_list(results2);
- return;
- }
- static void
- test_address_get_if_addrs(void *arg)
- {
- int rv;
- uint32_t addr_h = 0;
- tor_addr_t tor_addr;
- (void)arg;
- rv = get_interface_address(LOG_ERR, &addr_h);
- /* When the network is down, a system might not have any non-local
- * non-multicast IPv4 addresses, not even internal ones.
- * Unit tests shouldn't fail because of this. */
- if (rv == 0) {
- tor_addr_from_ipv4h(&tor_addr, addr_h);
- tt_assert(!tor_addr_is_loopback(&tor_addr));
- tt_assert(!tor_addr_is_multicast(&tor_addr));
- /* The address may or may not be an internal address */
- tt_assert(tor_addr_is_v4(&tor_addr));
- }
- done:
- return;
- }
- static void
- test_address_get_if_addrs6(void *arg)
- {
- int rv;
- tor_addr_t tor_addr;
- (void)arg;
- rv = get_interface_address6(LOG_ERR, AF_INET6, &tor_addr);
- /* Work even on systems without IPv6 interfaces */
- if (rv == 0) {
- tt_assert(!tor_addr_is_loopback(&tor_addr));
- tt_assert(!tor_addr_is_multicast(&tor_addr));
- /* The address may or may not be an internal address */
- tt_assert(!tor_addr_is_v4(&tor_addr));
- }
- done:
- return;
- }
- static void
- test_address_tor_addr_to_in6(void *ignored)
- {
- (void)ignored;
- tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
- const struct in6_addr *res;
- uint8_t expected[16] = {42, 1, 2, 3, 4, 5, 6, 7, 8, 9,
- 10, 11, 12, 13, 14, 15};
- a->family = AF_INET;
- res = tor_addr_to_in6(a);
- tt_assert(!res);
- a->family = AF_INET6;
- memcpy(a->addr.in6_addr.s6_addr, expected, 16);
- res = tor_addr_to_in6(a);
- tt_assert(res);
- tt_mem_op(res->s6_addr, OP_EQ, expected, 16);
- done:
- tor_free(a);
- }
- static void
- test_address_tor_addr_to_in(void *ignored)
- {
- (void)ignored;
- tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
- const struct in_addr *res;
- a->family = AF_INET6;
- res = tor_addr_to_in(a);
- tt_assert(!res);
- a->family = AF_INET;
- a->addr.in_addr.s_addr = 44;
- res = tor_addr_to_in(a);
- tt_assert(res);
- tt_int_op(res->s_addr, OP_EQ, 44);
- done:
- tor_free(a);
- }
- static void
- test_address_tor_addr_to_ipv4n(void *ignored)
- {
- (void)ignored;
- tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
- uint32_t res;
- a->family = AF_INET6;
- res = tor_addr_to_ipv4n(a);
- tt_assert(!res);
- a->family = AF_INET;
- a->addr.in_addr.s_addr = 43;
- res = tor_addr_to_ipv4n(a);
- tt_assert(res);
- tt_int_op(res, OP_EQ, 43);
- done:
- tor_free(a);
- }
- static void
- test_address_tor_addr_to_mapped_ipv4h(void *ignored)
- {
- (void)ignored;
- tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
- uint32_t res;
- uint8_t toset[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 255, 255, 0, 0, 0, 42};
- a->family = AF_INET;
- res = tor_addr_to_mapped_ipv4h(a);
- tt_assert(!res);
- a->family = AF_INET6;
- memcpy(a->addr.in6_addr.s6_addr, toset, 16);
- res = tor_addr_to_mapped_ipv4h(a);
- tt_assert(res);
- tt_int_op(res, OP_EQ, 42);
- done:
- tor_free(a);
- }
- static void
- test_address_tor_addr_eq_ipv4h(void *ignored)
- {
- (void)ignored;
- tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
- int res;
- a->family = AF_INET6;
- res = tor_addr_eq_ipv4h(a, 42);
- tt_assert(!res);
- a->family = AF_INET;
- a->addr.in_addr.s_addr = 52;
- res = tor_addr_eq_ipv4h(a, 42);
- tt_assert(!res);
- a->addr.in_addr.s_addr = 52;
- res = tor_addr_eq_ipv4h(a, ntohl(52));
- tt_assert(res);
- done:
- tor_free(a);
- }
- #define ADDRESS_TEST(name, flags) \
- { #name, test_address_ ## name, flags, NULL, NULL }
- struct testcase_t address_tests[] = {
- ADDRESS_TEST(udp_socket_trick_whitebox, TT_FORK),
- ADDRESS_TEST(udp_socket_trick_blackbox, TT_FORK),
- ADDRESS_TEST(get_if_addrs_list_internal, 0),
- ADDRESS_TEST(get_if_addrs_list_no_internal, 0),
- ADDRESS_TEST(get_if_addrs6_list_internal, 0),
- ADDRESS_TEST(get_if_addrs6_list_no_internal, 0),
- ADDRESS_TEST(get_if_addrs_internal_fail, 0),
- ADDRESS_TEST(get_if_addrs_no_internal_fail, 0),
- ADDRESS_TEST(get_if_addrs, 0),
- ADDRESS_TEST(get_if_addrs6, 0),
- #ifdef HAVE_IFADDRS_TO_SMARTLIST
- ADDRESS_TEST(get_if_addrs_ifaddrs, TT_FORK),
- ADDRESS_TEST(ifaddrs_to_smartlist, 0),
- #endif
- #ifdef HAVE_IP_ADAPTER_TO_SMARTLIST
- ADDRESS_TEST(get_if_addrs_win32, TT_FORK),
- ADDRESS_TEST(ip_adapter_addresses_to_smartlist, 0),
- #endif
- #ifdef HAVE_IFCONF_TO_SMARTLIST
- ADDRESS_TEST(get_if_addrs_ioctl, TT_FORK),
- ADDRESS_TEST(ifreq_to_smartlist, 0),
- #endif
- ADDRESS_TEST(tor_addr_to_in6, 0),
- ADDRESS_TEST(tor_addr_to_in, 0),
- ADDRESS_TEST(tor_addr_to_ipv4n, 0),
- ADDRESS_TEST(tor_addr_to_mapped_ipv4h, 0),
- ADDRESS_TEST(tor_addr_eq_ipv4h, 0),
- END_OF_TESTCASES
- };
|