12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175 |
- /* Copyright (c) 2014-2016, 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"
- #include "log_test_helpers.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, AF_UNSPEC);
- 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, AF_UNSPEC);
- 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:
- if (results) {
- 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, AF_UNSPEC);
- 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, AF_INET);
- /* 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
- fake_close_socket(tor_socket_t s)
- {
- (void)s;
- return 0;
- }
- static int last_connected_socket_fd = 0;
- static int connect_retval = 0;
- static tor_socket_t
- pretend_to_connect(tor_socket_t sock, const struct sockaddr *address,
- socklen_t address_len)
- {
- (void)address;
- (void)address_len;
- last_connected_socket_fd = sock;
- return connect_retval;
- }
- static struct sockaddr *mock_addr = NULL;
- static int
- fake_getsockname(tor_socket_t sock, struct sockaddr *address,
- socklen_t *address_len)
- {
- socklen_t bytes_to_copy = 0;
- (void) sock;
- 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(tor_close_socket,fake_close_socket);
- 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));
- done:
- UNMOCK(tor_open_socket);
- UNMOCK(tor_connect_socket);
- UNMOCK(tor_getsockname);
- UNMOCK(tor_close_socket);
- 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;
- /* We might drop a log_err */
- setup_full_capture_of_logs(LOG_ERR);
- results = get_interface_address6_list(LOG_ERR, AF_INET6, 1);
- tt_int_op(smartlist_len(mock_saved_logs()), OP_LE, 1);
- if (smartlist_len(mock_saved_logs()) == 1) {
- expect_log_msg_containing_either4("connect() failed",
- "unable to create socket",
- "Address that we determined via UDP "
- "socket magic is unsuitable for public "
- "comms.",
- "getsockname() to determine interface "
- "failed");
- }
- teardown_capture_of_logs();
- 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);
- teardown_capture_of_logs();
- return;
- }
- static void
- test_address_get_if_addrs6_list_no_internal(void *arg)
- {
- smartlist_t *results = NULL;
- (void)arg;
- /* We might drop a log_err */
- setup_full_capture_of_logs(LOG_ERR);
- results = get_interface_address6_list(LOG_ERR, AF_INET6, 0);
- tt_int_op(smartlist_len(mock_saved_logs()), OP_LE, 1);
- if (smartlist_len(mock_saved_logs()) == 1) {
- expect_log_msg_containing_either4("connect() failed",
- "unable to create socket",
- "Address that we determined via UDP "
- "socket magic is unsuitable for public "
- "comms.",
- "getsockname() to determine interface "
- "failed");
- }
- teardown_capture_of_logs();
- 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:
- teardown_capture_of_logs();
- 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, sa_family_t family)
- {
- (void)severity;
- (void)family;
- 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, TT_FORK),
- 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
- };
|