test_address.c 31 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135
  1. /* Copyright (c) 2014-2015, The Tor Project, Inc. */
  2. /* See LICENSE for licensing information */
  3. #define ADDRESS_PRIVATE
  4. #include "orconfig.h"
  5. #ifdef _WIN32
  6. #include <winsock2.h>
  7. /* For access to structs needed by GetAdaptersAddresses */
  8. #include <iphlpapi.h>
  9. #endif
  10. #ifdef HAVE_IFADDRS_TO_SMARTLIST
  11. #include <net/if.h>
  12. #include <ifaddrs.h>
  13. #endif
  14. #ifdef HAVE_IFCONF_TO_SMARTLIST
  15. #ifdef HAVE_SYS_IOCTL_H
  16. #include <sys/ioctl.h>
  17. #endif
  18. #include <net/if.h>
  19. #endif
  20. #include "or.h"
  21. #include "address.h"
  22. #include "test.h"
  23. /** Return 1 iff <b>sockaddr1</b> and <b>sockaddr2</b> represent
  24. * the same IP address and port combination. Otherwise, return 0.
  25. */
  26. static uint8_t
  27. sockaddr_in_are_equal(struct sockaddr_in *sockaddr1,
  28. struct sockaddr_in *sockaddr2)
  29. {
  30. return ((sockaddr1->sin_family == sockaddr2->sin_family) &&
  31. (sockaddr1->sin_port == sockaddr2->sin_port) &&
  32. (sockaddr1->sin_addr.s_addr == sockaddr2->sin_addr.s_addr));
  33. }
  34. /** Return 1 iff <b>sockaddr1</b> and <b>sockaddr2</b> represent
  35. * the same IP address and port combination. Otherwise, return 0.
  36. */
  37. static uint8_t
  38. sockaddr_in6_are_equal(struct sockaddr_in6 *sockaddr1,
  39. struct sockaddr_in6 *sockaddr2)
  40. {
  41. return ((sockaddr1->sin6_family == sockaddr2->sin6_family) &&
  42. (sockaddr1->sin6_port == sockaddr2->sin6_port) &&
  43. (tor_memeq(sockaddr1->sin6_addr.s6_addr,
  44. sockaddr2->sin6_addr.s6_addr,16)));
  45. }
  46. /** Create a sockaddr_in structure from IP address string <b>ip_str</b>.
  47. *
  48. * If <b>out</b> is not NULL, write the result
  49. * to the memory address in <b>out</b>. Otherwise, allocate the memory
  50. * for result. On success, return pointer to result. Otherwise, return
  51. * NULL.
  52. */
  53. static struct sockaddr_in *
  54. sockaddr_in_from_string(const char *ip_str, struct sockaddr_in *out)
  55. {
  56. // [FIXME: add some error checking?]
  57. if (!out)
  58. out = tor_malloc_zero(sizeof(struct sockaddr_in));
  59. out->sin_family = AF_INET;
  60. out->sin_port = 0;
  61. tor_inet_pton(AF_INET,ip_str,&(out->sin_addr));
  62. return out;
  63. }
  64. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  65. * that is an IPv4 or IPv6 localhost address. Otherwise, return 0.
  66. */
  67. static int
  68. smartlist_contains_localhost_tor_addr(smartlist_t *smartlist)
  69. {
  70. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  71. if (tor_addr_is_loopback(tor_addr)) {
  72. return 1;
  73. }
  74. } SMARTLIST_FOREACH_END(tor_addr);
  75. return 0;
  76. }
  77. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  78. * that is an IPv4 or IPv6 multicast address. Otherwise, return 0.
  79. */
  80. static int
  81. smartlist_contains_multicast_tor_addr(smartlist_t *smartlist)
  82. {
  83. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  84. if (tor_addr_is_multicast(tor_addr)) {
  85. return 1;
  86. }
  87. } SMARTLIST_FOREACH_END(tor_addr);
  88. return 0;
  89. }
  90. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  91. * that is an IPv4 or IPv6 internal address. Otherwise, return 0.
  92. */
  93. static int
  94. smartlist_contains_internal_tor_addr(smartlist_t *smartlist)
  95. {
  96. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  97. if (tor_addr_is_internal(tor_addr, 0)) {
  98. return 1;
  99. }
  100. } SMARTLIST_FOREACH_END(tor_addr);
  101. return 0;
  102. }
  103. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  104. * that is NULL or the null tor_addr_t. Otherwise, return 0.
  105. */
  106. static int
  107. smartlist_contains_null_tor_addr(smartlist_t *smartlist)
  108. {
  109. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  110. if (tor_addr == NULL || tor_addr_is_null(tor_addr)) {
  111. return 1;
  112. }
  113. } SMARTLIST_FOREACH_END(tor_addr);
  114. return 0;
  115. }
  116. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  117. * that is an IPv4 address. Otherwise, return 0.
  118. */
  119. static int
  120. smartlist_contains_ipv4_tor_addr(smartlist_t *smartlist)
  121. {
  122. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  123. if (tor_addr_is_v4(tor_addr)) {
  124. return 1;
  125. }
  126. } SMARTLIST_FOREACH_END(tor_addr);
  127. return 0;
  128. }
  129. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  130. * that is an IPv6 address. Otherwise, return 0.
  131. */
  132. static int
  133. smartlist_contains_ipv6_tor_addr(smartlist_t *smartlist)
  134. {
  135. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  136. /* Since there's no tor_addr_is_v6, assume all non-v4s are v6 */
  137. if (!tor_addr_is_v4(tor_addr)) {
  138. return 1;
  139. }
  140. } SMARTLIST_FOREACH_END(tor_addr);
  141. return 0;
  142. }
  143. #ifdef HAVE_IFADDRS_TO_SMARTLIST
  144. static void
  145. test_address_ifaddrs_to_smartlist(void *arg)
  146. {
  147. struct ifaddrs *ifa = NULL;
  148. struct ifaddrs *ifa_ipv4 = NULL;
  149. struct ifaddrs *ifa_ipv6 = NULL;
  150. struct sockaddr_in *ipv4_sockaddr_local = NULL;
  151. struct sockaddr_in *netmask_slash8 = NULL;
  152. struct sockaddr_in *ipv4_sockaddr_remote = NULL;
  153. struct sockaddr_in6 *ipv6_sockaddr = NULL;
  154. smartlist_t *smartlist = NULL;
  155. tor_addr_t *tor_addr = NULL;
  156. struct sockaddr *sockaddr_to_check = NULL;
  157. socklen_t addr_len;
  158. (void)arg;
  159. netmask_slash8 = sockaddr_in_from_string("255.0.0.0",NULL);
  160. ipv4_sockaddr_local = sockaddr_in_from_string("127.0.0.1",NULL);
  161. ipv4_sockaddr_remote = sockaddr_in_from_string("128.52.160.20",NULL);
  162. ipv6_sockaddr = tor_malloc(sizeof(struct sockaddr_in6));
  163. ipv6_sockaddr->sin6_family = AF_INET6;
  164. ipv6_sockaddr->sin6_port = 0;
  165. tor_inet_pton(AF_INET6, "2001:db8:8714:3a90::12",
  166. &(ipv6_sockaddr->sin6_addr));
  167. ifa = tor_malloc(sizeof(struct ifaddrs));
  168. ifa_ipv4 = tor_malloc(sizeof(struct ifaddrs));
  169. ifa_ipv6 = tor_malloc(sizeof(struct ifaddrs));
  170. ifa->ifa_next = ifa_ipv4;
  171. ifa->ifa_name = tor_strdup("eth0");
  172. ifa->ifa_flags = IFF_UP | IFF_RUNNING;
  173. ifa->ifa_addr = (struct sockaddr *)ipv4_sockaddr_local;
  174. ifa->ifa_netmask = (struct sockaddr *)netmask_slash8;
  175. ifa->ifa_dstaddr = NULL;
  176. ifa->ifa_data = NULL;
  177. ifa_ipv4->ifa_next = ifa_ipv6;
  178. ifa_ipv4->ifa_name = tor_strdup("eth1");
  179. ifa_ipv4->ifa_flags = IFF_UP | IFF_RUNNING;
  180. ifa_ipv4->ifa_addr = (struct sockaddr *)ipv4_sockaddr_remote;
  181. ifa_ipv4->ifa_netmask = (struct sockaddr *)netmask_slash8;
  182. ifa_ipv4->ifa_dstaddr = NULL;
  183. ifa_ipv4->ifa_data = NULL;
  184. ifa_ipv6->ifa_next = NULL;
  185. ifa_ipv6->ifa_name = tor_strdup("eth2");
  186. ifa_ipv6->ifa_flags = IFF_UP | IFF_RUNNING;
  187. ifa_ipv6->ifa_addr = (struct sockaddr *)ipv6_sockaddr;
  188. ifa_ipv6->ifa_netmask = NULL;
  189. ifa_ipv6->ifa_dstaddr = NULL;
  190. ifa_ipv6->ifa_data = NULL;
  191. smartlist = ifaddrs_to_smartlist(ifa, AF_UNSPEC);
  192. tt_assert(smartlist);
  193. tt_assert(smartlist_len(smartlist) == 3);
  194. sockaddr_to_check = tor_malloc(sizeof(struct sockaddr_in6));
  195. tor_addr = smartlist_get(smartlist,0);
  196. addr_len =
  197. tor_addr_to_sockaddr(tor_addr,0,sockaddr_to_check,
  198. sizeof(struct sockaddr_in));
  199. tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
  200. tt_assert(sockaddr_in_are_equal((struct sockaddr_in *)sockaddr_to_check,
  201. ipv4_sockaddr_local));
  202. tor_addr = smartlist_get(smartlist,1);
  203. addr_len =
  204. tor_addr_to_sockaddr(tor_addr,0,sockaddr_to_check,
  205. sizeof(struct sockaddr_in));
  206. tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
  207. tt_assert(sockaddr_in_are_equal((struct sockaddr_in *)sockaddr_to_check,
  208. ipv4_sockaddr_remote));
  209. tor_addr = smartlist_get(smartlist,2);
  210. addr_len =
  211. tor_addr_to_sockaddr(tor_addr,0,sockaddr_to_check,
  212. sizeof(struct sockaddr_in6));
  213. tt_int_op(addr_len,==,sizeof(struct sockaddr_in6));
  214. tt_assert(sockaddr_in6_are_equal((struct sockaddr_in6*)sockaddr_to_check,
  215. ipv6_sockaddr));
  216. done:
  217. tor_free(netmask_slash8);
  218. tor_free(ipv4_sockaddr_local);
  219. tor_free(ipv4_sockaddr_remote);
  220. tor_free(ipv6_sockaddr);
  221. tor_free(ifa->ifa_name);
  222. tor_free(ifa_ipv4->ifa_name);
  223. tor_free(ifa_ipv6->ifa_name);
  224. tor_free(ifa);
  225. tor_free(ifa_ipv4);
  226. tor_free(ifa_ipv6);
  227. tor_free(sockaddr_to_check);
  228. if (smartlist) {
  229. SMARTLIST_FOREACH(smartlist, tor_addr_t *, t, tor_free(t));
  230. smartlist_free(smartlist);
  231. }
  232. return;
  233. }
  234. static void
  235. test_address_get_if_addrs_ifaddrs(void *arg)
  236. {
  237. smartlist_t *results = NULL;
  238. (void)arg;
  239. results = get_interface_addresses_ifaddrs(LOG_ERR, AF_UNSPEC);
  240. tt_assert(results);
  241. /* Some FreeBSD jails don't have localhost IP address. Instead, they only
  242. * have the address assigned to the jail (whatever that may be).
  243. * And a jail without a network connection might not have any addresses at
  244. * all. */
  245. tt_assert(!smartlist_contains_null_tor_addr(results));
  246. /* If there are addresses, they must be IPv4 or IPv6 */
  247. if (smartlist_len(results) > 0) {
  248. tt_assert(smartlist_contains_ipv4_tor_addr(results)
  249. || smartlist_contains_ipv6_tor_addr(results));
  250. }
  251. done:
  252. SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
  253. smartlist_free(results);
  254. return;
  255. }
  256. #endif
  257. #ifdef HAVE_IP_ADAPTER_TO_SMARTLIST
  258. static void
  259. test_address_get_if_addrs_win32(void *arg)
  260. {
  261. smartlist_t *results = NULL;
  262. (void)arg;
  263. results = get_interface_addresses_win32(LOG_ERR, AF_UNSPEC);
  264. tt_int_op(smartlist_len(results),>=,1);
  265. tt_assert(smartlist_contains_localhost_tor_addr(results));
  266. tt_assert(!smartlist_contains_null_tor_addr(results));
  267. /* If there are addresses, they must be IPv4 or IPv6 */
  268. if (smartlist_len(results) > 0) {
  269. tt_assert(smartlist_contains_ipv4_tor_addr(results)
  270. || smartlist_contains_ipv6_tor_addr(results));
  271. }
  272. done:
  273. SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
  274. tor_free(results);
  275. return;
  276. }
  277. static void
  278. test_address_ip_adapter_addresses_to_smartlist(void *arg)
  279. {
  280. IP_ADAPTER_ADDRESSES *addrs1;
  281. IP_ADAPTER_ADDRESSES *addrs2;
  282. IP_ADAPTER_UNICAST_ADDRESS *unicast11;
  283. IP_ADAPTER_UNICAST_ADDRESS *unicast12;
  284. IP_ADAPTER_UNICAST_ADDRESS *unicast21;
  285. smartlist_t *result = NULL;
  286. struct sockaddr_in *sockaddr_test1;
  287. struct sockaddr_in *sockaddr_test2;
  288. struct sockaddr_in *sockaddr_localhost;
  289. struct sockaddr_in *sockaddr_to_check;
  290. tor_addr_t *tor_addr;
  291. (void)arg;
  292. (void)sockaddr_in6_are_equal;
  293. sockaddr_to_check = tor_malloc_zero(sizeof(struct sockaddr_in));
  294. addrs1 =
  295. tor_malloc_zero(sizeof(IP_ADAPTER_ADDRESSES));
  296. addrs1->FirstUnicastAddress =
  297. unicast11 = tor_malloc_zero(sizeof(IP_ADAPTER_UNICAST_ADDRESS));
  298. sockaddr_test1 = sockaddr_in_from_string("86.59.30.40",NULL);
  299. unicast11->Address.lpSockaddr = (LPSOCKADDR)sockaddr_test1;
  300. unicast11->Next = unicast12 =
  301. tor_malloc_zero(sizeof(IP_ADAPTER_UNICAST_ADDRESS));
  302. sockaddr_test2 = sockaddr_in_from_string("93.95.227.222", NULL);
  303. unicast12->Address.lpSockaddr = (LPSOCKADDR)sockaddr_test2;
  304. addrs1->Next = addrs2 =
  305. tor_malloc_zero(sizeof(IP_ADAPTER_ADDRESSES));
  306. addrs2->FirstUnicastAddress =
  307. unicast21 = tor_malloc_zero(sizeof(IP_ADAPTER_UNICAST_ADDRESS));
  308. sockaddr_localhost = sockaddr_in_from_string("127.0.0.1", NULL);
  309. unicast21->Address.lpSockaddr = (LPSOCKADDR)sockaddr_localhost;
  310. result = ip_adapter_addresses_to_smartlist(addrs1);
  311. tt_assert(result);
  312. tt_assert(smartlist_len(result) == 3);
  313. tor_addr = smartlist_get(result,0);
  314. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  315. sizeof(struct sockaddr_in));
  316. tt_assert(sockaddr_in_are_equal(sockaddr_test1,sockaddr_to_check));
  317. tor_addr = smartlist_get(result,1);
  318. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  319. sizeof(struct sockaddr_in));
  320. tt_assert(sockaddr_in_are_equal(sockaddr_test2,sockaddr_to_check));
  321. tor_addr = smartlist_get(result,2);
  322. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  323. sizeof(struct sockaddr_in));
  324. tt_assert(sockaddr_in_are_equal(sockaddr_localhost,sockaddr_to_check));
  325. done:
  326. SMARTLIST_FOREACH(result, tor_addr_t *, t, tor_free(t));
  327. smartlist_free(result);
  328. tor_free(addrs1);
  329. tor_free(addrs2);
  330. tor_free(unicast11->Address.lpSockaddr);
  331. tor_free(unicast11);
  332. tor_free(unicast12->Address.lpSockaddr);
  333. tor_free(unicast12);
  334. tor_free(unicast21->Address.lpSockaddr);
  335. tor_free(unicast21);
  336. tor_free(sockaddr_to_check);
  337. return;
  338. }
  339. #endif
  340. #ifdef HAVE_IFCONF_TO_SMARTLIST
  341. static void
  342. test_address_ifreq_to_smartlist(void *arg)
  343. {
  344. smartlist_t *results = NULL;
  345. const tor_addr_t *tor_addr = NULL;
  346. struct sockaddr_in *sockaddr = NULL;
  347. struct sockaddr_in *sockaddr_eth1 = NULL;
  348. struct sockaddr_in *sockaddr_to_check = NULL;
  349. struct ifconf *ifc;
  350. struct ifreq *ifr;
  351. struct ifreq *ifr_next;
  352. socklen_t addr_len;
  353. (void)arg;
  354. sockaddr_to_check = tor_malloc(sizeof(struct sockaddr_in));
  355. ifr = tor_malloc(sizeof(struct ifreq));
  356. memset(ifr,0,sizeof(struct ifreq));
  357. strlcpy(ifr->ifr_name,"lo",3);
  358. sockaddr = (struct sockaddr_in *) &(ifr->ifr_ifru.ifru_addr);
  359. sockaddr_in_from_string("127.0.0.1",sockaddr);
  360. ifc = tor_malloc(sizeof(struct ifconf));
  361. memset(ifc,0,sizeof(struct ifconf));
  362. ifc->ifc_len = sizeof(struct ifreq);
  363. ifc->ifc_ifcu.ifcu_req = ifr;
  364. results = ifreq_to_smartlist(ifc->ifc_buf,ifc->ifc_len);
  365. tt_int_op(smartlist_len(results),==,1);
  366. tor_addr = smartlist_get(results, 0);
  367. addr_len =
  368. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  369. sizeof(struct sockaddr_in));
  370. tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
  371. tt_assert(sockaddr_in_are_equal(sockaddr,sockaddr_to_check));
  372. ifr = tor_realloc(ifr,2*sizeof(struct ifreq));
  373. ifr_next = ifr+1;
  374. strlcpy(ifr_next->ifr_name,"eth1",5);
  375. ifc->ifc_len = 2*sizeof(struct ifreq);
  376. ifc->ifc_ifcu.ifcu_req = ifr;
  377. sockaddr = (struct sockaddr_in *) &(ifr->ifr_ifru.ifru_addr);
  378. sockaddr_eth1 = (struct sockaddr_in *) &(ifr_next->ifr_ifru.ifru_addr);
  379. sockaddr_in_from_string("192.168.10.55",sockaddr_eth1);
  380. SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
  381. smartlist_free(results);
  382. results = ifreq_to_smartlist(ifc->ifc_buf,ifc->ifc_len);
  383. tt_int_op(smartlist_len(results),==,2);
  384. tor_addr = smartlist_get(results, 0);
  385. addr_len =
  386. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  387. sizeof(struct sockaddr_in));
  388. tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
  389. tt_assert(sockaddr_in_are_equal(sockaddr,sockaddr_to_check));
  390. tor_addr = smartlist_get(results, 1);
  391. addr_len =
  392. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  393. sizeof(struct sockaddr_in));
  394. tt_int_op(addr_len,==,sizeof(struct sockaddr_in));
  395. tt_assert(sockaddr_in_are_equal(sockaddr_eth1,sockaddr_to_check));
  396. done:
  397. tor_free(sockaddr_to_check);
  398. SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
  399. smartlist_free(results);
  400. tor_free(ifc);
  401. tor_free(ifr);
  402. return;
  403. }
  404. static void
  405. test_address_get_if_addrs_ioctl(void *arg)
  406. {
  407. smartlist_t *result = NULL;
  408. (void)arg;
  409. result = get_interface_addresses_ioctl(LOG_ERR, AF_INET);
  410. /* On an IPv6-only system, this will fail and return NULL
  411. tt_assert(result);
  412. */
  413. /* Some FreeBSD jails don't have localhost IP address. Instead, they only
  414. * have the address assigned to the jail (whatever that may be).
  415. * And a jail without a network connection might not have any addresses at
  416. * all. */
  417. if (result) {
  418. tt_assert(!smartlist_contains_null_tor_addr(result));
  419. /* If there are addresses, they must be IPv4 or IPv6.
  420. * (AIX supports IPv6 from SIOCGIFCONF.) */
  421. if (smartlist_len(result) > 0) {
  422. tt_assert(smartlist_contains_ipv4_tor_addr(result)
  423. || smartlist_contains_ipv6_tor_addr(result));
  424. }
  425. }
  426. done:
  427. if (result) {
  428. SMARTLIST_FOREACH(result, tor_addr_t *, t, tor_free(t));
  429. smartlist_free(result);
  430. }
  431. return;
  432. }
  433. #endif
  434. #define FAKE_SOCKET_FD (42)
  435. static tor_socket_t
  436. fake_open_socket(int domain, int type, int protocol)
  437. {
  438. (void)domain;
  439. (void)type;
  440. (void)protocol;
  441. return FAKE_SOCKET_FD;
  442. }
  443. static int last_connected_socket_fd = 0;
  444. static int connect_retval = 0;
  445. static tor_socket_t
  446. pretend_to_connect(tor_socket_t socket, const struct sockaddr *address,
  447. socklen_t address_len)
  448. {
  449. (void)address;
  450. (void)address_len;
  451. last_connected_socket_fd = socket;
  452. return connect_retval;
  453. }
  454. static struct sockaddr *mock_addr = NULL;
  455. static int
  456. fake_getsockname(tor_socket_t socket, struct sockaddr *address,
  457. socklen_t *address_len)
  458. {
  459. socklen_t bytes_to_copy = 0;
  460. (void) socket;
  461. if (!mock_addr)
  462. return -1;
  463. if (mock_addr->sa_family == AF_INET) {
  464. bytes_to_copy = sizeof(struct sockaddr_in);
  465. } else if (mock_addr->sa_family == AF_INET6) {
  466. bytes_to_copy = sizeof(struct sockaddr_in6);
  467. } else {
  468. return -1;
  469. }
  470. if (*address_len < bytes_to_copy) {
  471. return -1;
  472. }
  473. memcpy(address,mock_addr,bytes_to_copy);
  474. *address_len = bytes_to_copy;
  475. return 0;
  476. }
  477. static void
  478. test_address_udp_socket_trick_whitebox(void *arg)
  479. {
  480. int hack_retval;
  481. tor_addr_t *addr_from_hack = tor_malloc_zero(sizeof(tor_addr_t));
  482. struct sockaddr_in6 *mock_addr6;
  483. struct sockaddr_in6 *ipv6_to_check =
  484. tor_malloc_zero(sizeof(struct sockaddr_in6));
  485. (void)arg;
  486. MOCK(tor_open_socket,fake_open_socket);
  487. MOCK(tor_connect_socket,pretend_to_connect);
  488. MOCK(tor_getsockname,fake_getsockname);
  489. mock_addr = tor_malloc_zero(sizeof(struct sockaddr_storage));
  490. sockaddr_in_from_string("23.32.246.118",(struct sockaddr_in *)mock_addr);
  491. hack_retval =
  492. get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  493. AF_INET, addr_from_hack);
  494. tt_int_op(hack_retval,==,0);
  495. tt_assert(tor_addr_eq_ipv4h(addr_from_hack, 0x1720f676));
  496. /* Now, lets do an IPv6 case. */
  497. memset(mock_addr,0,sizeof(struct sockaddr_storage));
  498. mock_addr6 = (struct sockaddr_in6 *)mock_addr;
  499. mock_addr6->sin6_family = AF_INET6;
  500. mock_addr6->sin6_port = 0;
  501. tor_inet_pton(AF_INET6,"2001:cdba::3257:9652",&(mock_addr6->sin6_addr));
  502. hack_retval =
  503. get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  504. AF_INET6, addr_from_hack);
  505. tt_int_op(hack_retval,==,0);
  506. tor_addr_to_sockaddr(addr_from_hack,0,(struct sockaddr *)ipv6_to_check,
  507. sizeof(struct sockaddr_in6));
  508. tt_assert(sockaddr_in6_are_equal(mock_addr6,ipv6_to_check));
  509. UNMOCK(tor_open_socket);
  510. UNMOCK(tor_connect_socket);
  511. UNMOCK(tor_getsockname);
  512. done:
  513. tor_free(ipv6_to_check);
  514. tor_free(mock_addr);
  515. tor_free(addr_from_hack);
  516. return;
  517. }
  518. static void
  519. test_address_udp_socket_trick_blackbox(void *arg)
  520. {
  521. /* We want get_interface_address6_via_udp_socket_hack() to yield
  522. * the same valid address that get_interface_address6() returns.
  523. * If the latter is unable to find a valid address, we want
  524. * _hack() to fail and return-1.
  525. *
  526. * Furthermore, we want _hack() never to crash, even if
  527. * get_interface_addresses_raw() is returning NULL.
  528. */
  529. tor_addr_t addr4;
  530. tor_addr_t addr4_to_check;
  531. tor_addr_t addr6;
  532. tor_addr_t addr6_to_check;
  533. int retval, retval_reference;
  534. (void)arg;
  535. #if 0
  536. retval_reference = get_interface_address6(LOG_DEBUG,AF_INET,&addr4);
  537. retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  538. AF_INET,
  539. &addr4_to_check);
  540. tt_int_op(retval,==,retval_reference);
  541. tt_assert( (retval == -1 && retval_reference == -1) ||
  542. (tor_addr_compare(&addr4,&addr4_to_check,CMP_EXACT) == 0) );
  543. retval_reference = get_interface_address6(LOG_DEBUG,AF_INET6,&addr6);
  544. retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  545. AF_INET6,
  546. &addr6_to_check);
  547. tt_int_op(retval,==,retval_reference);
  548. tt_assert( (retval == -1 && retval_reference == -1) ||
  549. (tor_addr_compare(&addr6,&addr6_to_check,CMP_EXACT) == 0) );
  550. #else
  551. /* Both of the blackbox test cases fail horribly if:
  552. * * The host has no external addreses.
  553. * * There are multiple interfaces with either AF_INET or AF_INET6.
  554. * * The last address isn't the one associated with the default route.
  555. *
  556. * The tests SHOULD be re-enabled when #12377 is fixed correctly, but till
  557. * then this fails a lot, in situations we expect failures due to knowing
  558. * about the code being broken.
  559. */
  560. (void)addr4_to_check;
  561. (void)addr6_to_check;
  562. (void)addr6;
  563. (void) retval_reference;
  564. #endif
  565. /* When family is neither AF_INET nor AF_INET6, we want _hack to
  566. * fail and return -1.
  567. */
  568. retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  569. AF_INET+AF_INET6,&addr4);
  570. tt_assert(retval == -1);
  571. done:
  572. return;
  573. }
  574. static void
  575. test_address_get_if_addrs_list_internal(void *arg)
  576. {
  577. smartlist_t *results = NULL;
  578. (void)arg;
  579. results = get_interface_address_list(LOG_ERR, 1);
  580. tt_assert(results != NULL);
  581. /* When the network is down, a system might not have any non-local
  582. * non-multicast addresseses, not even internal ones.
  583. * Unit tests shouldn't fail because of this. */
  584. tt_int_op(smartlist_len(results),>=,0);
  585. tt_assert(!smartlist_contains_localhost_tor_addr(results));
  586. tt_assert(!smartlist_contains_multicast_tor_addr(results));
  587. /* The list may or may not contain internal addresses */
  588. tt_assert(!smartlist_contains_null_tor_addr(results));
  589. /* if there are any addresses, they must be IPv4 */
  590. if (smartlist_len(results) > 0) {
  591. tt_assert(smartlist_contains_ipv4_tor_addr(results));
  592. }
  593. tt_assert(!smartlist_contains_ipv6_tor_addr(results));
  594. done:
  595. free_interface_address_list(results);
  596. return;
  597. }
  598. static void
  599. test_address_get_if_addrs_list_no_internal(void *arg)
  600. {
  601. smartlist_t *results = NULL;
  602. (void)arg;
  603. results = get_interface_address_list(LOG_ERR, 0);
  604. tt_assert(results != NULL);
  605. /* Work even on systems with only internal IPv4 addresses */
  606. tt_int_op(smartlist_len(results),>=,0);
  607. tt_assert(!smartlist_contains_localhost_tor_addr(results));
  608. tt_assert(!smartlist_contains_multicast_tor_addr(results));
  609. tt_assert(!smartlist_contains_internal_tor_addr(results));
  610. tt_assert(!smartlist_contains_null_tor_addr(results));
  611. /* if there are any addresses, they must be IPv4 */
  612. if (smartlist_len(results) > 0) {
  613. tt_assert(smartlist_contains_ipv4_tor_addr(results));
  614. }
  615. tt_assert(!smartlist_contains_ipv6_tor_addr(results));
  616. done:
  617. free_interface_address_list(results);
  618. return;
  619. }
  620. static void
  621. test_address_get_if_addrs6_list_internal(void *arg)
  622. {
  623. smartlist_t *results = NULL;
  624. (void)arg;
  625. results = get_interface_address6_list(LOG_ERR, AF_INET6, 1);
  626. tt_assert(results != NULL);
  627. /* Work even on systems without IPv6 interfaces */
  628. tt_int_op(smartlist_len(results),>=,0);
  629. tt_assert(!smartlist_contains_localhost_tor_addr(results));
  630. tt_assert(!smartlist_contains_multicast_tor_addr(results));
  631. /* The list may or may not contain internal addresses */
  632. tt_assert(!smartlist_contains_null_tor_addr(results));
  633. /* if there are any addresses, they must be IPv6 */
  634. tt_assert(!smartlist_contains_ipv4_tor_addr(results));
  635. if (smartlist_len(results) > 0) {
  636. tt_assert(smartlist_contains_ipv6_tor_addr(results));
  637. }
  638. done:
  639. free_interface_address6_list(results);
  640. return;
  641. }
  642. static void
  643. test_address_get_if_addrs6_list_no_internal(void *arg)
  644. {
  645. smartlist_t *results = NULL;
  646. (void)arg;
  647. results = get_interface_address6_list(LOG_ERR, AF_INET6, 0);
  648. tt_assert(results != NULL);
  649. /* Work even on systems without IPv6 interfaces */
  650. tt_int_op(smartlist_len(results),>=,0);
  651. tt_assert(!smartlist_contains_localhost_tor_addr(results));
  652. tt_assert(!smartlist_contains_multicast_tor_addr(results));
  653. tt_assert(!smartlist_contains_internal_tor_addr(results));
  654. tt_assert(!smartlist_contains_null_tor_addr(results));
  655. /* if there are any addresses, they must be IPv6 */
  656. tt_assert(!smartlist_contains_ipv4_tor_addr(results));
  657. if (smartlist_len(results) > 0) {
  658. tt_assert(smartlist_contains_ipv6_tor_addr(results));
  659. }
  660. done:
  661. free_interface_address6_list(results);
  662. return;
  663. }
  664. static int called_get_interface_addresses_raw = 0;
  665. static smartlist_t *
  666. mock_get_interface_addresses_raw_fail(int severity, sa_family_t family)
  667. {
  668. (void)severity;
  669. (void)family;
  670. called_get_interface_addresses_raw++;
  671. return smartlist_new();
  672. }
  673. static int called_get_interface_address6_via_udp_socket_hack = 0;
  674. static int
  675. mock_get_interface_address6_via_udp_socket_hack_fail(int severity,
  676. sa_family_t family,
  677. tor_addr_t *addr)
  678. {
  679. (void)severity;
  680. (void)family;
  681. (void)addr;
  682. called_get_interface_address6_via_udp_socket_hack++;
  683. return -1;
  684. }
  685. static void
  686. test_address_get_if_addrs_internal_fail(void *arg)
  687. {
  688. smartlist_t *results1 = NULL, *results2 = NULL;
  689. int rv = 0;
  690. uint32_t ipv4h_addr = 0;
  691. tor_addr_t ipv6_addr;
  692. memset(&ipv6_addr, 0, sizeof(tor_addr_t));
  693. (void)arg;
  694. MOCK(get_interface_addresses_raw,
  695. mock_get_interface_addresses_raw_fail);
  696. MOCK(get_interface_address6_via_udp_socket_hack,
  697. mock_get_interface_address6_via_udp_socket_hack_fail);
  698. results1 = get_interface_address6_list(LOG_ERR, AF_INET6, 1);
  699. tt_assert(results1 != NULL);
  700. tt_int_op(smartlist_len(results1),==,0);
  701. results2 = get_interface_address_list(LOG_ERR, 1);
  702. tt_assert(results2 != NULL);
  703. tt_int_op(smartlist_len(results2),==,0);
  704. rv = get_interface_address6(LOG_ERR, AF_INET6, &ipv6_addr);
  705. tt_assert(rv == -1);
  706. rv = get_interface_address(LOG_ERR, &ipv4h_addr);
  707. tt_assert(rv == -1);
  708. done:
  709. UNMOCK(get_interface_addresses_raw);
  710. UNMOCK(get_interface_address6_via_udp_socket_hack);
  711. free_interface_address6_list(results1);
  712. free_interface_address6_list(results2);
  713. return;
  714. }
  715. static void
  716. test_address_get_if_addrs_no_internal_fail(void *arg)
  717. {
  718. smartlist_t *results1 = NULL, *results2 = NULL;
  719. (void)arg;
  720. MOCK(get_interface_addresses_raw,
  721. mock_get_interface_addresses_raw_fail);
  722. MOCK(get_interface_address6_via_udp_socket_hack,
  723. mock_get_interface_address6_via_udp_socket_hack_fail);
  724. results1 = get_interface_address6_list(LOG_ERR, AF_INET6, 0);
  725. tt_assert(results1 != NULL);
  726. tt_int_op(smartlist_len(results1),==,0);
  727. results2 = get_interface_address_list(LOG_ERR, 0);
  728. tt_assert(results2 != NULL);
  729. tt_int_op(smartlist_len(results2),==,0);
  730. done:
  731. UNMOCK(get_interface_addresses_raw);
  732. UNMOCK(get_interface_address6_via_udp_socket_hack);
  733. free_interface_address6_list(results1);
  734. free_interface_address6_list(results2);
  735. return;
  736. }
  737. static void
  738. test_address_get_if_addrs(void *arg)
  739. {
  740. int rv;
  741. uint32_t addr_h = 0;
  742. tor_addr_t tor_addr;
  743. (void)arg;
  744. rv = get_interface_address(LOG_ERR, &addr_h);
  745. /* When the network is down, a system might not have any non-local
  746. * non-multicast IPv4 addresses, not even internal ones.
  747. * Unit tests shouldn't fail because of this. */
  748. if (rv == 0) {
  749. tor_addr_from_ipv4h(&tor_addr, addr_h);
  750. tt_assert(!tor_addr_is_loopback(&tor_addr));
  751. tt_assert(!tor_addr_is_multicast(&tor_addr));
  752. /* The address may or may not be an internal address */
  753. tt_assert(tor_addr_is_v4(&tor_addr));
  754. }
  755. done:
  756. return;
  757. }
  758. static void
  759. test_address_get_if_addrs6(void *arg)
  760. {
  761. int rv;
  762. tor_addr_t tor_addr;
  763. (void)arg;
  764. rv = get_interface_address6(LOG_ERR, AF_INET6, &tor_addr);
  765. /* Work even on systems without IPv6 interfaces */
  766. if (rv == 0) {
  767. tt_assert(!tor_addr_is_loopback(&tor_addr));
  768. tt_assert(!tor_addr_is_multicast(&tor_addr));
  769. /* The address may or may not be an internal address */
  770. tt_assert(!tor_addr_is_v4(&tor_addr));
  771. }
  772. done:
  773. return;
  774. }
  775. static void
  776. test_address_tor_addr_to_in6(void *ignored)
  777. {
  778. (void)ignored;
  779. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  780. const struct in6_addr *res;
  781. uint8_t expected[16] = {42, 1, 2, 3, 4, 5, 6, 7, 8, 9,
  782. 10, 11, 12, 13, 14, 15};
  783. a->family = AF_INET;
  784. res = tor_addr_to_in6(a);
  785. tt_assert(!res);
  786. a->family = AF_INET6;
  787. memcpy(a->addr.in6_addr.s6_addr, expected, 16);
  788. res = tor_addr_to_in6(a);
  789. tt_assert(res);
  790. tt_mem_op(res->s6_addr, OP_EQ, expected, 16);
  791. done:
  792. tor_free(a);
  793. }
  794. static void
  795. test_address_tor_addr_to_in(void *ignored)
  796. {
  797. (void)ignored;
  798. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  799. const struct in_addr *res;
  800. a->family = AF_INET6;
  801. res = tor_addr_to_in(a);
  802. tt_assert(!res);
  803. a->family = AF_INET;
  804. a->addr.in_addr.s_addr = 44;
  805. res = tor_addr_to_in(a);
  806. tt_assert(res);
  807. tt_int_op(res->s_addr, OP_EQ, 44);
  808. done:
  809. tor_free(a);
  810. }
  811. static void
  812. test_address_tor_addr_to_ipv4n(void *ignored)
  813. {
  814. (void)ignored;
  815. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  816. uint32_t res;
  817. a->family = AF_INET6;
  818. res = tor_addr_to_ipv4n(a);
  819. tt_assert(!res);
  820. a->family = AF_INET;
  821. a->addr.in_addr.s_addr = 43;
  822. res = tor_addr_to_ipv4n(a);
  823. tt_assert(res);
  824. tt_int_op(res, OP_EQ, 43);
  825. done:
  826. tor_free(a);
  827. }
  828. static void
  829. test_address_tor_addr_to_mapped_ipv4h(void *ignored)
  830. {
  831. (void)ignored;
  832. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  833. uint32_t res;
  834. uint8_t toset[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 255, 255, 0, 0, 0, 42};
  835. a->family = AF_INET;
  836. res = tor_addr_to_mapped_ipv4h(a);
  837. tt_assert(!res);
  838. a->family = AF_INET6;
  839. memcpy(a->addr.in6_addr.s6_addr, toset, 16);
  840. res = tor_addr_to_mapped_ipv4h(a);
  841. tt_assert(res);
  842. tt_int_op(res, OP_EQ, 42);
  843. done:
  844. tor_free(a);
  845. }
  846. static void
  847. test_address_tor_addr_eq_ipv4h(void *ignored)
  848. {
  849. (void)ignored;
  850. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  851. int res;
  852. a->family = AF_INET6;
  853. res = tor_addr_eq_ipv4h(a, 42);
  854. tt_assert(!res);
  855. a->family = AF_INET;
  856. a->addr.in_addr.s_addr = 52;
  857. res = tor_addr_eq_ipv4h(a, 42);
  858. tt_assert(!res);
  859. a->addr.in_addr.s_addr = 52;
  860. res = tor_addr_eq_ipv4h(a, ntohl(52));
  861. tt_assert(res);
  862. done:
  863. tor_free(a);
  864. }
  865. #define ADDRESS_TEST(name, flags) \
  866. { #name, test_address_ ## name, flags, NULL, NULL }
  867. struct testcase_t address_tests[] = {
  868. ADDRESS_TEST(udp_socket_trick_whitebox, TT_FORK),
  869. ADDRESS_TEST(udp_socket_trick_blackbox, TT_FORK),
  870. ADDRESS_TEST(get_if_addrs_list_internal, 0),
  871. ADDRESS_TEST(get_if_addrs_list_no_internal, 0),
  872. ADDRESS_TEST(get_if_addrs6_list_internal, 0),
  873. ADDRESS_TEST(get_if_addrs6_list_no_internal, 0),
  874. ADDRESS_TEST(get_if_addrs_internal_fail, 0),
  875. ADDRESS_TEST(get_if_addrs_no_internal_fail, 0),
  876. ADDRESS_TEST(get_if_addrs, 0),
  877. ADDRESS_TEST(get_if_addrs6, 0),
  878. #ifdef HAVE_IFADDRS_TO_SMARTLIST
  879. ADDRESS_TEST(get_if_addrs_ifaddrs, TT_FORK),
  880. ADDRESS_TEST(ifaddrs_to_smartlist, 0),
  881. #endif
  882. #ifdef HAVE_IP_ADAPTER_TO_SMARTLIST
  883. ADDRESS_TEST(get_if_addrs_win32, TT_FORK),
  884. ADDRESS_TEST(ip_adapter_addresses_to_smartlist, 0),
  885. #endif
  886. #ifdef HAVE_IFCONF_TO_SMARTLIST
  887. ADDRESS_TEST(get_if_addrs_ioctl, TT_FORK),
  888. ADDRESS_TEST(ifreq_to_smartlist, 0),
  889. #endif
  890. ADDRESS_TEST(tor_addr_to_in6, 0),
  891. ADDRESS_TEST(tor_addr_to_in, 0),
  892. ADDRESS_TEST(tor_addr_to_ipv4n, 0),
  893. ADDRESS_TEST(tor_addr_to_mapped_ipv4h, 0),
  894. ADDRESS_TEST(tor_addr_eq_ipv4h, 0),
  895. END_OF_TESTCASES
  896. };