test_address.c 38 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372
  1. /* Copyright (c) 2014-2019, 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 /* defined(HAVE_IFCONF_TO_SMARTLIST) */
  20. #include "core/or/or.h"
  21. #include "app/config/config.h"
  22. #include "feature/dirauth/process_descs.h"
  23. #include "feature/nodelist/routerinfo_st.h"
  24. #include "feature/nodelist/node_st.h"
  25. #include "feature/nodelist/nodelist.h"
  26. #include "lib/net/address.h"
  27. #include "test/test.h"
  28. #include "test/log_test_helpers.h"
  29. /** Return 1 iff <b>sockaddr1</b> and <b>sockaddr2</b> represent
  30. * the same IP address and port combination. Otherwise, return 0.
  31. */
  32. static uint8_t
  33. sockaddr_in_are_equal(struct sockaddr_in *sockaddr1,
  34. struct sockaddr_in *sockaddr2)
  35. {
  36. return ((sockaddr1->sin_family == sockaddr2->sin_family) &&
  37. (sockaddr1->sin_port == sockaddr2->sin_port) &&
  38. (sockaddr1->sin_addr.s_addr == sockaddr2->sin_addr.s_addr));
  39. }
  40. /** Return 1 iff <b>sockaddr1</b> and <b>sockaddr2</b> represent
  41. * the same IP address and port combination. Otherwise, return 0.
  42. */
  43. static uint8_t
  44. sockaddr_in6_are_equal(struct sockaddr_in6 *sockaddr1,
  45. struct sockaddr_in6 *sockaddr2)
  46. {
  47. return ((sockaddr1->sin6_family == sockaddr2->sin6_family) &&
  48. (sockaddr1->sin6_port == sockaddr2->sin6_port) &&
  49. (tor_memeq(sockaddr1->sin6_addr.s6_addr,
  50. sockaddr2->sin6_addr.s6_addr,16)));
  51. }
  52. /** Create a sockaddr_in structure from IP address string <b>ip_str</b>.
  53. *
  54. * If <b>out</b> is not NULL, write the result
  55. * to the memory address in <b>out</b>. Otherwise, allocate the memory
  56. * for result. On success, return pointer to result. Otherwise, return
  57. * NULL.
  58. */
  59. static struct sockaddr_in *
  60. sockaddr_in_from_string(const char *ip_str, struct sockaddr_in *out)
  61. {
  62. // [FIXME: add some error checking?]
  63. if (!out)
  64. out = tor_malloc_zero(sizeof(struct sockaddr_in));
  65. out->sin_family = AF_INET;
  66. out->sin_port = 0;
  67. tor_inet_pton(AF_INET,ip_str,&(out->sin_addr));
  68. return out;
  69. }
  70. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  71. * that is an IPv4 or IPv6 localhost address. Otherwise, return 0.
  72. */
  73. static int
  74. smartlist_contains_localhost_tor_addr(smartlist_t *smartlist)
  75. {
  76. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  77. if (tor_addr_is_loopback(tor_addr)) {
  78. return 1;
  79. }
  80. } SMARTLIST_FOREACH_END(tor_addr);
  81. return 0;
  82. }
  83. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  84. * that is an IPv4 or IPv6 multicast address. Otherwise, return 0.
  85. */
  86. static int
  87. smartlist_contains_multicast_tor_addr(smartlist_t *smartlist)
  88. {
  89. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  90. if (tor_addr_is_multicast(tor_addr)) {
  91. return 1;
  92. }
  93. } SMARTLIST_FOREACH_END(tor_addr);
  94. return 0;
  95. }
  96. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  97. * that is an IPv4 or IPv6 internal address. Otherwise, return 0.
  98. */
  99. static int
  100. smartlist_contains_internal_tor_addr(smartlist_t *smartlist)
  101. {
  102. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  103. if (tor_addr_is_internal(tor_addr, 0)) {
  104. return 1;
  105. }
  106. } SMARTLIST_FOREACH_END(tor_addr);
  107. return 0;
  108. }
  109. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  110. * that is NULL or the null tor_addr_t. Otherwise, return 0.
  111. */
  112. static int
  113. smartlist_contains_null_tor_addr(smartlist_t *smartlist)
  114. {
  115. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  116. if (tor_addr == NULL || tor_addr_is_null(tor_addr)) {
  117. return 1;
  118. }
  119. } SMARTLIST_FOREACH_END(tor_addr);
  120. return 0;
  121. }
  122. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  123. * that is an IPv4 address. Otherwise, return 0.
  124. */
  125. static int
  126. smartlist_contains_ipv4_tor_addr(smartlist_t *smartlist)
  127. {
  128. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  129. if (tor_addr_is_v4(tor_addr)) {
  130. return 1;
  131. }
  132. } SMARTLIST_FOREACH_END(tor_addr);
  133. return 0;
  134. }
  135. /** Return 1 iff <b>smartlist</b> contains a tor_addr_t structure
  136. * that is an IPv6 address. Otherwise, return 0.
  137. */
  138. static int
  139. smartlist_contains_ipv6_tor_addr(smartlist_t *smartlist)
  140. {
  141. SMARTLIST_FOREACH_BEGIN(smartlist, tor_addr_t *, tor_addr) {
  142. /* Since there's no tor_addr_is_v6, assume all non-v4s are v6 */
  143. if (!tor_addr_is_v4(tor_addr)) {
  144. return 1;
  145. }
  146. } SMARTLIST_FOREACH_END(tor_addr);
  147. return 0;
  148. }
  149. #ifdef HAVE_IFADDRS_TO_SMARTLIST
  150. static void
  151. test_address_ifaddrs_to_smartlist(void *arg)
  152. {
  153. struct ifaddrs *ifa = NULL;
  154. struct ifaddrs *ifa_ipv4 = NULL;
  155. struct ifaddrs *ifa_ipv6 = NULL;
  156. struct sockaddr_in *ipv4_sockaddr_local = NULL;
  157. struct sockaddr_in *netmask_slash8 = NULL;
  158. struct sockaddr_in *ipv4_sockaddr_remote = NULL;
  159. struct sockaddr_in6 *ipv6_sockaddr = NULL;
  160. smartlist_t *smartlist = NULL;
  161. tor_addr_t *tor_addr = NULL;
  162. struct sockaddr *sockaddr_to_check = NULL;
  163. socklen_t addr_len;
  164. (void)arg;
  165. netmask_slash8 = sockaddr_in_from_string("255.0.0.0",NULL);
  166. ipv4_sockaddr_local = sockaddr_in_from_string("127.0.0.1",NULL);
  167. ipv4_sockaddr_remote = sockaddr_in_from_string("128.52.160.20",NULL);
  168. ipv6_sockaddr = tor_malloc(sizeof(struct sockaddr_in6));
  169. ipv6_sockaddr->sin6_family = AF_INET6;
  170. ipv6_sockaddr->sin6_port = 0;
  171. tor_inet_pton(AF_INET6, "2001:db8:8714:3a90::12",
  172. &(ipv6_sockaddr->sin6_addr));
  173. ifa = tor_malloc(sizeof(struct ifaddrs));
  174. ifa_ipv4 = tor_malloc(sizeof(struct ifaddrs));
  175. ifa_ipv6 = tor_malloc(sizeof(struct ifaddrs));
  176. ifa->ifa_next = ifa_ipv4;
  177. ifa->ifa_name = tor_strdup("eth0");
  178. ifa->ifa_flags = IFF_UP | IFF_RUNNING;
  179. ifa->ifa_addr = (struct sockaddr *)ipv4_sockaddr_local;
  180. ifa->ifa_netmask = (struct sockaddr *)netmask_slash8;
  181. ifa->ifa_dstaddr = NULL;
  182. ifa->ifa_data = NULL;
  183. ifa_ipv4->ifa_next = ifa_ipv6;
  184. ifa_ipv4->ifa_name = tor_strdup("eth1");
  185. ifa_ipv4->ifa_flags = IFF_UP | IFF_RUNNING;
  186. ifa_ipv4->ifa_addr = (struct sockaddr *)ipv4_sockaddr_remote;
  187. ifa_ipv4->ifa_netmask = (struct sockaddr *)netmask_slash8;
  188. ifa_ipv4->ifa_dstaddr = NULL;
  189. ifa_ipv4->ifa_data = NULL;
  190. ifa_ipv6->ifa_next = NULL;
  191. ifa_ipv6->ifa_name = tor_strdup("eth2");
  192. ifa_ipv6->ifa_flags = IFF_UP | IFF_RUNNING;
  193. ifa_ipv6->ifa_addr = (struct sockaddr *)ipv6_sockaddr;
  194. ifa_ipv6->ifa_netmask = NULL;
  195. ifa_ipv6->ifa_dstaddr = NULL;
  196. ifa_ipv6->ifa_data = NULL;
  197. smartlist = ifaddrs_to_smartlist(ifa, AF_UNSPEC);
  198. tt_assert(smartlist);
  199. tt_int_op(smartlist_len(smartlist), OP_EQ, 3);
  200. sockaddr_to_check = tor_malloc(sizeof(struct sockaddr_in6));
  201. tor_addr = smartlist_get(smartlist,0);
  202. addr_len =
  203. tor_addr_to_sockaddr(tor_addr,0,sockaddr_to_check,
  204. sizeof(struct sockaddr_in));
  205. tt_int_op(addr_len,OP_EQ,sizeof(struct sockaddr_in));
  206. tt_assert(sockaddr_in_are_equal((struct sockaddr_in *)sockaddr_to_check,
  207. ipv4_sockaddr_local));
  208. tor_addr = smartlist_get(smartlist,1);
  209. addr_len =
  210. tor_addr_to_sockaddr(tor_addr,0,sockaddr_to_check,
  211. sizeof(struct sockaddr_in));
  212. tt_int_op(addr_len,OP_EQ,sizeof(struct sockaddr_in));
  213. tt_assert(sockaddr_in_are_equal((struct sockaddr_in *)sockaddr_to_check,
  214. ipv4_sockaddr_remote));
  215. tor_addr = smartlist_get(smartlist,2);
  216. addr_len =
  217. tor_addr_to_sockaddr(tor_addr,0,sockaddr_to_check,
  218. sizeof(struct sockaddr_in6));
  219. tt_int_op(addr_len,OP_EQ,sizeof(struct sockaddr_in6));
  220. tt_assert(sockaddr_in6_are_equal((struct sockaddr_in6*)sockaddr_to_check,
  221. ipv6_sockaddr));
  222. done:
  223. tor_free(netmask_slash8);
  224. tor_free(ipv4_sockaddr_local);
  225. tor_free(ipv4_sockaddr_remote);
  226. tor_free(ipv6_sockaddr);
  227. tor_free(ifa->ifa_name);
  228. tor_free(ifa_ipv4->ifa_name);
  229. tor_free(ifa_ipv6->ifa_name);
  230. tor_free(ifa);
  231. tor_free(ifa_ipv4);
  232. tor_free(ifa_ipv6);
  233. tor_free(sockaddr_to_check);
  234. if (smartlist) {
  235. SMARTLIST_FOREACH(smartlist, tor_addr_t *, t, tor_free(t));
  236. smartlist_free(smartlist);
  237. }
  238. return;
  239. }
  240. static void
  241. test_address_get_if_addrs_ifaddrs(void *arg)
  242. {
  243. smartlist_t *results = NULL;
  244. (void)arg;
  245. results = get_interface_addresses_ifaddrs(LOG_ERR, AF_UNSPEC);
  246. tt_assert(results);
  247. /* Some FreeBSD jails don't have localhost IP address. Instead, they only
  248. * have the address assigned to the jail (whatever that may be).
  249. * And a jail without a network connection might not have any addresses at
  250. * all. */
  251. tt_assert(!smartlist_contains_null_tor_addr(results));
  252. /* If there are addresses, they must be IPv4 or IPv6 */
  253. if (smartlist_len(results) > 0) {
  254. tt_assert(smartlist_contains_ipv4_tor_addr(results)
  255. || smartlist_contains_ipv6_tor_addr(results));
  256. }
  257. done:
  258. if (results) {
  259. SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
  260. }
  261. smartlist_free(results);
  262. return;
  263. }
  264. #endif /* defined(HAVE_IFADDRS_TO_SMARTLIST) */
  265. #ifdef HAVE_IP_ADAPTER_TO_SMARTLIST
  266. static void
  267. test_address_get_if_addrs_win32(void *arg)
  268. {
  269. smartlist_t *results = NULL;
  270. (void)arg;
  271. results = get_interface_addresses_win32(LOG_ERR, AF_UNSPEC);
  272. tt_int_op(smartlist_len(results),OP_GE,1);
  273. tt_assert(smartlist_contains_localhost_tor_addr(results));
  274. tt_assert(!smartlist_contains_null_tor_addr(results));
  275. /* If there are addresses, they must be IPv4 or IPv6 */
  276. if (smartlist_len(results) > 0) {
  277. tt_assert(smartlist_contains_ipv4_tor_addr(results)
  278. || smartlist_contains_ipv6_tor_addr(results));
  279. }
  280. done:
  281. SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
  282. tor_free(results);
  283. return;
  284. }
  285. static void
  286. test_address_ip_adapter_addresses_to_smartlist(void *arg)
  287. {
  288. IP_ADAPTER_ADDRESSES *addrs1;
  289. IP_ADAPTER_ADDRESSES *addrs2;
  290. IP_ADAPTER_UNICAST_ADDRESS *unicast11;
  291. IP_ADAPTER_UNICAST_ADDRESS *unicast12;
  292. IP_ADAPTER_UNICAST_ADDRESS *unicast21;
  293. smartlist_t *result = NULL;
  294. struct sockaddr_in *sockaddr_test1;
  295. struct sockaddr_in *sockaddr_test2;
  296. struct sockaddr_in *sockaddr_localhost;
  297. struct sockaddr_in *sockaddr_to_check;
  298. tor_addr_t *tor_addr;
  299. (void)arg;
  300. (void)sockaddr_in6_are_equal;
  301. sockaddr_to_check = tor_malloc_zero(sizeof(struct sockaddr_in));
  302. addrs1 =
  303. tor_malloc_zero(sizeof(IP_ADAPTER_ADDRESSES));
  304. addrs1->FirstUnicastAddress =
  305. unicast11 = tor_malloc_zero(sizeof(IP_ADAPTER_UNICAST_ADDRESS));
  306. sockaddr_test1 = sockaddr_in_from_string("86.59.30.40",NULL);
  307. unicast11->Address.lpSockaddr = (LPSOCKADDR)sockaddr_test1;
  308. unicast11->Next = unicast12 =
  309. tor_malloc_zero(sizeof(IP_ADAPTER_UNICAST_ADDRESS));
  310. sockaddr_test2 = sockaddr_in_from_string("93.95.227.222", NULL);
  311. unicast12->Address.lpSockaddr = (LPSOCKADDR)sockaddr_test2;
  312. addrs1->Next = addrs2 =
  313. tor_malloc_zero(sizeof(IP_ADAPTER_ADDRESSES));
  314. addrs2->FirstUnicastAddress =
  315. unicast21 = tor_malloc_zero(sizeof(IP_ADAPTER_UNICAST_ADDRESS));
  316. sockaddr_localhost = sockaddr_in_from_string("127.0.0.1", NULL);
  317. unicast21->Address.lpSockaddr = (LPSOCKADDR)sockaddr_localhost;
  318. result = ip_adapter_addresses_to_smartlist(addrs1);
  319. tt_assert(result);
  320. tt_int_op(smartlist_len(result), OP_EQ, 3);
  321. tor_addr = smartlist_get(result,0);
  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_test1,sockaddr_to_check));
  325. tor_addr = smartlist_get(result,1);
  326. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  327. sizeof(struct sockaddr_in));
  328. tt_assert(sockaddr_in_are_equal(sockaddr_test2,sockaddr_to_check));
  329. tor_addr = smartlist_get(result,2);
  330. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  331. sizeof(struct sockaddr_in));
  332. tt_assert(sockaddr_in_are_equal(sockaddr_localhost,sockaddr_to_check));
  333. done:
  334. SMARTLIST_FOREACH(result, tor_addr_t *, t, tor_free(t));
  335. smartlist_free(result);
  336. tor_free(addrs1);
  337. tor_free(addrs2);
  338. tor_free(unicast11->Address.lpSockaddr);
  339. tor_free(unicast11);
  340. tor_free(unicast12->Address.lpSockaddr);
  341. tor_free(unicast12);
  342. tor_free(unicast21->Address.lpSockaddr);
  343. tor_free(unicast21);
  344. tor_free(sockaddr_to_check);
  345. return;
  346. }
  347. #endif /* defined(HAVE_IP_ADAPTER_TO_SMARTLIST) */
  348. #ifdef HAVE_IFCONF_TO_SMARTLIST
  349. static void
  350. test_address_ifreq_to_smartlist(void *arg)
  351. {
  352. smartlist_t *results = NULL;
  353. const tor_addr_t *tor_addr = NULL;
  354. struct sockaddr_in *sockaddr = NULL;
  355. struct sockaddr_in *sockaddr_eth1 = NULL;
  356. struct sockaddr_in *sockaddr_to_check = NULL;
  357. struct ifconf *ifc;
  358. struct ifreq *ifr;
  359. struct ifreq *ifr_next;
  360. socklen_t addr_len;
  361. (void)arg;
  362. sockaddr_to_check = tor_malloc(sizeof(struct sockaddr_in));
  363. ifr = tor_malloc(sizeof(struct ifreq));
  364. memset(ifr,0,sizeof(struct ifreq));
  365. strlcpy(ifr->ifr_name,"lo",3);
  366. sockaddr = (struct sockaddr_in *) &(ifr->ifr_ifru.ifru_addr);
  367. sockaddr_in_from_string("127.0.0.1",sockaddr);
  368. ifc = tor_malloc(sizeof(struct ifconf));
  369. memset(ifc,0,sizeof(struct ifconf));
  370. ifc->ifc_len = sizeof(struct ifreq);
  371. ifc->ifc_ifcu.ifcu_req = ifr;
  372. results = ifreq_to_smartlist(ifc->ifc_buf,ifc->ifc_len);
  373. tt_int_op(smartlist_len(results),OP_EQ,1);
  374. tor_addr = smartlist_get(results, 0);
  375. addr_len =
  376. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  377. sizeof(struct sockaddr_in));
  378. tt_int_op(addr_len,OP_EQ,sizeof(struct sockaddr_in));
  379. tt_assert(sockaddr_in_are_equal(sockaddr,sockaddr_to_check));
  380. ifr = tor_realloc(ifr,2*sizeof(struct ifreq));
  381. ifr_next = ifr+1;
  382. strlcpy(ifr_next->ifr_name,"eth1",5);
  383. ifc->ifc_len = 2*sizeof(struct ifreq);
  384. ifc->ifc_ifcu.ifcu_req = ifr;
  385. sockaddr = (struct sockaddr_in *) &(ifr->ifr_ifru.ifru_addr);
  386. sockaddr_eth1 = (struct sockaddr_in *) &(ifr_next->ifr_ifru.ifru_addr);
  387. sockaddr_in_from_string("192.168.10.55",sockaddr_eth1);
  388. SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
  389. smartlist_free(results);
  390. results = ifreq_to_smartlist(ifc->ifc_buf,ifc->ifc_len);
  391. tt_int_op(smartlist_len(results),OP_EQ,2);
  392. tor_addr = smartlist_get(results, 0);
  393. addr_len =
  394. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  395. sizeof(struct sockaddr_in));
  396. tt_int_op(addr_len,OP_EQ,sizeof(struct sockaddr_in));
  397. tt_assert(sockaddr_in_are_equal(sockaddr,sockaddr_to_check));
  398. tor_addr = smartlist_get(results, 1);
  399. addr_len =
  400. tor_addr_to_sockaddr(tor_addr,0,(struct sockaddr *)sockaddr_to_check,
  401. sizeof(struct sockaddr_in));
  402. tt_int_op(addr_len,OP_EQ,sizeof(struct sockaddr_in));
  403. tt_assert(sockaddr_in_are_equal(sockaddr_eth1,sockaddr_to_check));
  404. done:
  405. tor_free(sockaddr_to_check);
  406. SMARTLIST_FOREACH(results, tor_addr_t *, t, tor_free(t));
  407. smartlist_free(results);
  408. tor_free(ifc);
  409. tor_free(ifr);
  410. return;
  411. }
  412. static void
  413. test_address_get_if_addrs_ioctl(void *arg)
  414. {
  415. smartlist_t *result = NULL;
  416. (void)arg;
  417. result = get_interface_addresses_ioctl(LOG_ERR, AF_INET);
  418. /* On an IPv6-only system, this will fail and return NULL
  419. tt_assert(result);
  420. */
  421. /* Some FreeBSD jails don't have localhost IP address. Instead, they only
  422. * have the address assigned to the jail (whatever that may be).
  423. * And a jail without a network connection might not have any addresses at
  424. * all. */
  425. if (result) {
  426. tt_assert(!smartlist_contains_null_tor_addr(result));
  427. /* If there are addresses, they must be IPv4 or IPv6.
  428. * (AIX supports IPv6 from SIOCGIFCONF.) */
  429. if (smartlist_len(result) > 0) {
  430. tt_assert(smartlist_contains_ipv4_tor_addr(result)
  431. || smartlist_contains_ipv6_tor_addr(result));
  432. }
  433. }
  434. done:
  435. if (result) {
  436. SMARTLIST_FOREACH(result, tor_addr_t *, t, tor_free(t));
  437. smartlist_free(result);
  438. }
  439. return;
  440. }
  441. #endif /* defined(HAVE_IFCONF_TO_SMARTLIST) */
  442. #define FAKE_SOCKET_FD (42)
  443. static tor_socket_t
  444. fake_open_socket(int domain, int type, int protocol)
  445. {
  446. (void)domain;
  447. (void)type;
  448. (void)protocol;
  449. return FAKE_SOCKET_FD;
  450. }
  451. static int
  452. fake_close_socket(tor_socket_t s)
  453. {
  454. (void)s;
  455. return 0;
  456. }
  457. static int last_connected_socket_fd = 0;
  458. static int connect_retval = 0;
  459. static tor_socket_t
  460. pretend_to_connect(tor_socket_t sock, const struct sockaddr *address,
  461. socklen_t address_len)
  462. {
  463. (void)address;
  464. (void)address_len;
  465. last_connected_socket_fd = sock;
  466. return connect_retval;
  467. }
  468. static struct sockaddr *mock_addr = NULL;
  469. static int
  470. fake_getsockname(tor_socket_t sock, struct sockaddr *address,
  471. socklen_t *address_len)
  472. {
  473. socklen_t bytes_to_copy = 0;
  474. (void) sock;
  475. if (!mock_addr)
  476. return -1;
  477. if (mock_addr->sa_family == AF_INET) {
  478. bytes_to_copy = sizeof(struct sockaddr_in);
  479. } else if (mock_addr->sa_family == AF_INET6) {
  480. bytes_to_copy = sizeof(struct sockaddr_in6);
  481. } else {
  482. return -1;
  483. }
  484. if (*address_len < bytes_to_copy) {
  485. return -1;
  486. }
  487. memcpy(address,mock_addr,bytes_to_copy);
  488. *address_len = bytes_to_copy;
  489. return 0;
  490. }
  491. static void
  492. test_address_udp_socket_trick_whitebox(void *arg)
  493. {
  494. int hack_retval;
  495. tor_addr_t *addr_from_hack = tor_malloc_zero(sizeof(tor_addr_t));
  496. struct sockaddr_in6 *mock_addr6;
  497. struct sockaddr_in6 *ipv6_to_check =
  498. tor_malloc_zero(sizeof(struct sockaddr_in6));
  499. (void)arg;
  500. MOCK(tor_open_socket,fake_open_socket);
  501. MOCK(tor_connect_socket,pretend_to_connect);
  502. MOCK(tor_getsockname,fake_getsockname);
  503. MOCK(tor_close_socket,fake_close_socket);
  504. mock_addr = tor_malloc_zero(sizeof(struct sockaddr_storage));
  505. sockaddr_in_from_string("23.32.246.118",(struct sockaddr_in *)mock_addr);
  506. hack_retval =
  507. get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  508. AF_INET, addr_from_hack);
  509. tt_int_op(hack_retval,OP_EQ,0);
  510. tt_assert(tor_addr_eq_ipv4h(addr_from_hack, 0x1720f676));
  511. /* Now, lets do an IPv6 case. */
  512. memset(mock_addr,0,sizeof(struct sockaddr_storage));
  513. mock_addr6 = (struct sockaddr_in6 *)mock_addr;
  514. mock_addr6->sin6_family = AF_INET6;
  515. mock_addr6->sin6_port = 0;
  516. tor_inet_pton(AF_INET6,"2001:cdba::3257:9652",&(mock_addr6->sin6_addr));
  517. hack_retval =
  518. get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  519. AF_INET6, addr_from_hack);
  520. tt_int_op(hack_retval,OP_EQ,0);
  521. tor_addr_to_sockaddr(addr_from_hack,0,(struct sockaddr *)ipv6_to_check,
  522. sizeof(struct sockaddr_in6));
  523. tt_assert(sockaddr_in6_are_equal(mock_addr6,ipv6_to_check));
  524. done:
  525. UNMOCK(tor_open_socket);
  526. UNMOCK(tor_connect_socket);
  527. UNMOCK(tor_getsockname);
  528. UNMOCK(tor_close_socket);
  529. tor_free(ipv6_to_check);
  530. tor_free(mock_addr);
  531. tor_free(addr_from_hack);
  532. return;
  533. }
  534. static void
  535. test_address_udp_socket_trick_blackbox(void *arg)
  536. {
  537. /* We want get_interface_address6_via_udp_socket_hack() to yield
  538. * the same valid address that get_interface_address6() returns.
  539. * If the latter is unable to find a valid address, we want
  540. * _hack() to fail and return-1.
  541. *
  542. * Furthermore, we want _hack() never to crash, even if
  543. * get_interface_addresses_raw() is returning NULL.
  544. */
  545. tor_addr_t addr4;
  546. tor_addr_t addr4_to_check;
  547. tor_addr_t addr6;
  548. tor_addr_t addr6_to_check;
  549. int retval, retval_reference;
  550. (void)arg;
  551. #if 0
  552. retval_reference = get_interface_address6(LOG_DEBUG,AF_INET,&addr4);
  553. retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  554. AF_INET,
  555. &addr4_to_check);
  556. tt_int_op(retval,OP_EQ,retval_reference);
  557. tt_assert( (retval == -1 && retval_reference == -1) ||
  558. (tor_addr_compare(&addr4,&addr4_to_check,CMP_EXACT) == 0) );
  559. retval_reference = get_interface_address6(LOG_DEBUG,AF_INET6,&addr6);
  560. retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  561. AF_INET6,
  562. &addr6_to_check);
  563. tt_int_op(retval,OP_EQ,retval_reference);
  564. tt_assert( (retval == -1 && retval_reference == -1) ||
  565. (tor_addr_compare(&addr6,&addr6_to_check,CMP_EXACT) == 0) );
  566. #else /* !(0) */
  567. /* Both of the blackbox test cases fail horribly if:
  568. * * The host has no external addreses.
  569. * * There are multiple interfaces with either AF_INET or AF_INET6.
  570. * * The last address isn't the one associated with the default route.
  571. *
  572. * The tests SHOULD be re-enabled when #12377 is fixed correctly, but till
  573. * then this fails a lot, in situations we expect failures due to knowing
  574. * about the code being broken.
  575. */
  576. (void)addr4_to_check;
  577. (void)addr6_to_check;
  578. (void)addr6;
  579. (void) retval_reference;
  580. #endif /* 0 */
  581. /* When family is neither AF_INET nor AF_INET6, we want _hack to
  582. * fail and return -1.
  583. */
  584. retval = get_interface_address6_via_udp_socket_hack(LOG_DEBUG,
  585. AF_INET+AF_INET6,&addr4);
  586. tt_int_op(retval, OP_EQ, -1);
  587. done:
  588. return;
  589. }
  590. static void
  591. test_address_get_if_addrs_list_internal(void *arg)
  592. {
  593. smartlist_t *results = NULL;
  594. (void)arg;
  595. results = get_interface_address_list(LOG_WARN, 1);
  596. tt_ptr_op(results, OP_NE, NULL);
  597. /* When the network is down, a system might not have any non-local
  598. * non-multicast addresseses, not even internal ones.
  599. * Unit tests shouldn't fail because of this. */
  600. tt_int_op(smartlist_len(results),OP_GE,0);
  601. tt_assert(!smartlist_contains_localhost_tor_addr(results));
  602. tt_assert(!smartlist_contains_multicast_tor_addr(results));
  603. /* The list may or may not contain internal addresses */
  604. tt_assert(!smartlist_contains_null_tor_addr(results));
  605. /* if there are any addresses, they must be IPv4 */
  606. if (smartlist_len(results) > 0) {
  607. tt_assert(smartlist_contains_ipv4_tor_addr(results));
  608. }
  609. tt_assert(!smartlist_contains_ipv6_tor_addr(results));
  610. done:
  611. interface_address_list_free(results);
  612. return;
  613. }
  614. static void
  615. test_address_get_if_addrs_list_no_internal(void *arg)
  616. {
  617. smartlist_t *results = NULL;
  618. (void)arg;
  619. results = get_interface_address_list(LOG_WARN, 0);
  620. tt_ptr_op(results, OP_NE, NULL);
  621. /* Work even on systems with only internal IPv4 addresses */
  622. tt_int_op(smartlist_len(results),OP_GE,0);
  623. tt_assert(!smartlist_contains_localhost_tor_addr(results));
  624. tt_assert(!smartlist_contains_multicast_tor_addr(results));
  625. tt_assert(!smartlist_contains_internal_tor_addr(results));
  626. tt_assert(!smartlist_contains_null_tor_addr(results));
  627. /* if there are any addresses, they must be IPv4 */
  628. if (smartlist_len(results) > 0) {
  629. tt_assert(smartlist_contains_ipv4_tor_addr(results));
  630. }
  631. tt_assert(!smartlist_contains_ipv6_tor_addr(results));
  632. done:
  633. interface_address_list_free(results);
  634. return;
  635. }
  636. static void
  637. test_address_get_if_addrs6_list_internal(void *arg)
  638. {
  639. smartlist_t *results = NULL;
  640. (void)arg;
  641. /* We might drop a log_err */
  642. setup_full_capture_of_logs(LOG_ERR);
  643. results = get_interface_address6_list(LOG_ERR, AF_INET6, 1);
  644. tt_int_op(smartlist_len(mock_saved_logs()), OP_LE, 1);
  645. if (smartlist_len(mock_saved_logs()) == 1) {
  646. expect_log_msg_containing_either4("connect() failed",
  647. "unable to create socket",
  648. "Address that we determined via UDP "
  649. "socket magic is unsuitable for public "
  650. "comms.",
  651. "getsockname() to determine interface "
  652. "failed");
  653. }
  654. teardown_capture_of_logs();
  655. tt_ptr_op(results, OP_NE, NULL);
  656. /* Work even on systems without IPv6 interfaces */
  657. tt_int_op(smartlist_len(results),OP_GE,0);
  658. tt_assert(!smartlist_contains_localhost_tor_addr(results));
  659. tt_assert(!smartlist_contains_multicast_tor_addr(results));
  660. /* The list may or may not contain internal addresses */
  661. tt_assert(!smartlist_contains_null_tor_addr(results));
  662. /* if there are any addresses, they must be IPv6 */
  663. tt_assert(!smartlist_contains_ipv4_tor_addr(results));
  664. if (smartlist_len(results) > 0) {
  665. tt_assert(smartlist_contains_ipv6_tor_addr(results));
  666. }
  667. done:
  668. interface_address6_list_free(results);
  669. teardown_capture_of_logs();
  670. return;
  671. }
  672. static void
  673. test_address_get_if_addrs6_list_no_internal(void *arg)
  674. {
  675. smartlist_t *results = NULL;
  676. (void)arg;
  677. /* We might drop a log_err */
  678. setup_full_capture_of_logs(LOG_ERR);
  679. results = get_interface_address6_list(LOG_ERR, AF_INET6, 0);
  680. tt_int_op(smartlist_len(mock_saved_logs()), OP_LE, 1);
  681. if (smartlist_len(mock_saved_logs()) == 1) {
  682. expect_log_msg_containing_either4("connect() failed",
  683. "unable to create socket",
  684. "Address that we determined via UDP "
  685. "socket magic is unsuitable for public "
  686. "comms.",
  687. "getsockname() to determine interface "
  688. "failed");
  689. }
  690. teardown_capture_of_logs();
  691. tt_ptr_op(results, OP_NE, NULL);
  692. /* Work even on systems without IPv6 interfaces */
  693. tt_int_op(smartlist_len(results),OP_GE,0);
  694. tt_assert(!smartlist_contains_localhost_tor_addr(results));
  695. tt_assert(!smartlist_contains_multicast_tor_addr(results));
  696. tt_assert(!smartlist_contains_internal_tor_addr(results));
  697. tt_assert(!smartlist_contains_null_tor_addr(results));
  698. /* if there are any addresses, they must be IPv6 */
  699. tt_assert(!smartlist_contains_ipv4_tor_addr(results));
  700. if (smartlist_len(results) > 0) {
  701. tt_assert(smartlist_contains_ipv6_tor_addr(results));
  702. }
  703. done:
  704. teardown_capture_of_logs();
  705. interface_address6_list_free(results);
  706. return;
  707. }
  708. static int called_get_interface_addresses_raw = 0;
  709. static smartlist_t *
  710. mock_get_interface_addresses_raw_fail(int severity, sa_family_t family)
  711. {
  712. (void)severity;
  713. (void)family;
  714. called_get_interface_addresses_raw++;
  715. return smartlist_new();
  716. }
  717. static int called_get_interface_address6_via_udp_socket_hack = 0;
  718. static int
  719. mock_get_interface_address6_via_udp_socket_hack_fail(int severity,
  720. sa_family_t family,
  721. tor_addr_t *addr)
  722. {
  723. (void)severity;
  724. (void)family;
  725. (void)addr;
  726. called_get_interface_address6_via_udp_socket_hack++;
  727. return -1;
  728. }
  729. static void
  730. test_address_get_if_addrs_internal_fail(void *arg)
  731. {
  732. smartlist_t *results1 = NULL, *results2 = NULL;
  733. int rv = 0;
  734. uint32_t ipv4h_addr = 0;
  735. tor_addr_t ipv6_addr;
  736. memset(&ipv6_addr, 0, sizeof(tor_addr_t));
  737. (void)arg;
  738. MOCK(get_interface_addresses_raw,
  739. mock_get_interface_addresses_raw_fail);
  740. MOCK(get_interface_address6_via_udp_socket_hack,
  741. mock_get_interface_address6_via_udp_socket_hack_fail);
  742. results1 = get_interface_address6_list(LOG_ERR, AF_INET6, 1);
  743. tt_ptr_op(results1, OP_NE, NULL);
  744. tt_int_op(smartlist_len(results1),OP_EQ,0);
  745. results2 = get_interface_address_list(LOG_ERR, 1);
  746. tt_ptr_op(results2, OP_NE, NULL);
  747. tt_int_op(smartlist_len(results2),OP_EQ,0);
  748. rv = get_interface_address6(LOG_ERR, AF_INET6, &ipv6_addr);
  749. tt_int_op(rv, OP_EQ, -1);
  750. rv = get_interface_address(LOG_ERR, &ipv4h_addr);
  751. tt_int_op(rv, OP_EQ, -1);
  752. done:
  753. UNMOCK(get_interface_addresses_raw);
  754. UNMOCK(get_interface_address6_via_udp_socket_hack);
  755. interface_address6_list_free(results1);
  756. interface_address6_list_free(results2);
  757. return;
  758. }
  759. static void
  760. test_address_get_if_addrs_no_internal_fail(void *arg)
  761. {
  762. smartlist_t *results1 = NULL, *results2 = NULL;
  763. (void)arg;
  764. MOCK(get_interface_addresses_raw,
  765. mock_get_interface_addresses_raw_fail);
  766. MOCK(get_interface_address6_via_udp_socket_hack,
  767. mock_get_interface_address6_via_udp_socket_hack_fail);
  768. results1 = get_interface_address6_list(LOG_ERR, AF_INET6, 0);
  769. tt_ptr_op(results1, OP_NE, NULL);
  770. tt_int_op(smartlist_len(results1),OP_EQ,0);
  771. results2 = get_interface_address_list(LOG_ERR, 0);
  772. tt_ptr_op(results2, OP_NE, NULL);
  773. tt_int_op(smartlist_len(results2),OP_EQ,0);
  774. done:
  775. UNMOCK(get_interface_addresses_raw);
  776. UNMOCK(get_interface_address6_via_udp_socket_hack);
  777. interface_address6_list_free(results1);
  778. interface_address6_list_free(results2);
  779. return;
  780. }
  781. static void
  782. test_address_get_if_addrs(void *arg)
  783. {
  784. int rv;
  785. uint32_t addr_h = 0;
  786. tor_addr_t tor_addr;
  787. (void)arg;
  788. rv = get_interface_address(LOG_WARN, &addr_h);
  789. /* When the network is down, a system might not have any non-local
  790. * non-multicast IPv4 addresses, not even internal ones.
  791. * Unit tests shouldn't fail because of this. */
  792. if (rv == 0) {
  793. tor_addr_from_ipv4h(&tor_addr, addr_h);
  794. tt_assert(!tor_addr_is_loopback(&tor_addr));
  795. tt_assert(!tor_addr_is_multicast(&tor_addr));
  796. /* The address may or may not be an internal address */
  797. tt_assert(tor_addr_is_v4(&tor_addr));
  798. }
  799. done:
  800. return;
  801. }
  802. static void
  803. test_address_get_if_addrs6(void *arg)
  804. {
  805. int rv;
  806. tor_addr_t tor_addr;
  807. (void)arg;
  808. rv = get_interface_address6(LOG_WARN, AF_INET6, &tor_addr);
  809. /* Work even on systems without IPv6 interfaces */
  810. if (rv == 0) {
  811. tt_assert(!tor_addr_is_loopback(&tor_addr));
  812. tt_assert(!tor_addr_is_multicast(&tor_addr));
  813. /* The address may or may not be an internal address */
  814. tt_assert(!tor_addr_is_v4(&tor_addr));
  815. }
  816. done:
  817. return;
  818. }
  819. static void
  820. test_address_tor_addr_to_in6(void *ignored)
  821. {
  822. (void)ignored;
  823. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  824. const struct in6_addr *res;
  825. uint8_t expected[16] = {42, 1, 2, 3, 4, 5, 6, 7, 8, 9,
  826. 10, 11, 12, 13, 14, 15};
  827. a->family = AF_INET;
  828. res = tor_addr_to_in6(a);
  829. tt_assert(!res);
  830. a->family = AF_INET6;
  831. memcpy(a->addr.in6_addr.s6_addr, expected, 16);
  832. res = tor_addr_to_in6(a);
  833. tt_assert(res);
  834. tt_mem_op(res->s6_addr, OP_EQ, expected, 16);
  835. done:
  836. tor_free(a);
  837. }
  838. static void
  839. test_address_tor_addr_to_in(void *ignored)
  840. {
  841. (void)ignored;
  842. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  843. const struct in_addr *res;
  844. a->family = AF_INET6;
  845. res = tor_addr_to_in(a);
  846. tt_assert(!res);
  847. a->family = AF_INET;
  848. a->addr.in_addr.s_addr = 44;
  849. res = tor_addr_to_in(a);
  850. tt_assert(res);
  851. tt_int_op(res->s_addr, OP_EQ, 44);
  852. done:
  853. tor_free(a);
  854. }
  855. static void
  856. test_address_tor_addr_to_ipv4n(void *ignored)
  857. {
  858. (void)ignored;
  859. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  860. uint32_t res;
  861. a->family = AF_INET6;
  862. res = tor_addr_to_ipv4n(a);
  863. tt_assert(!res);
  864. a->family = AF_INET;
  865. a->addr.in_addr.s_addr = 43;
  866. res = tor_addr_to_ipv4n(a);
  867. tt_assert(res);
  868. tt_int_op(res, OP_EQ, 43);
  869. done:
  870. tor_free(a);
  871. }
  872. static void
  873. test_address_tor_addr_to_mapped_ipv4h(void *ignored)
  874. {
  875. (void)ignored;
  876. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  877. uint32_t res;
  878. uint8_t toset[16] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 255, 255, 0, 0, 0, 42};
  879. a->family = AF_INET;
  880. res = tor_addr_to_mapped_ipv4h(a);
  881. tt_assert(!res);
  882. a->family = AF_INET6;
  883. memcpy(a->addr.in6_addr.s6_addr, toset, 16);
  884. res = tor_addr_to_mapped_ipv4h(a);
  885. tt_assert(res);
  886. tt_int_op(res, OP_EQ, 42);
  887. done:
  888. tor_free(a);
  889. }
  890. static void
  891. test_address_tor_addr_eq_ipv4h(void *ignored)
  892. {
  893. (void)ignored;
  894. tor_addr_t *a = tor_malloc_zero(sizeof(tor_addr_t));
  895. int res;
  896. a->family = AF_INET6;
  897. res = tor_addr_eq_ipv4h(a, 42);
  898. tt_assert(!res);
  899. a->family = AF_INET;
  900. a->addr.in_addr.s_addr = 52;
  901. res = tor_addr_eq_ipv4h(a, 42);
  902. tt_assert(!res);
  903. a->addr.in_addr.s_addr = 52;
  904. res = tor_addr_eq_ipv4h(a, ntohl(52));
  905. tt_assert(res);
  906. done:
  907. tor_free(a);
  908. }
  909. static void
  910. test_address_tor_addr_in_same_network_family(void *ignored)
  911. {
  912. (void)ignored;
  913. tor_addr_t a, b;
  914. tor_addr_parse(&a, "8.8.8.8");
  915. tor_addr_parse(&b, "8.8.4.4");
  916. tt_int_op(addrs_in_same_network_family(&a, &b), OP_EQ, 1);
  917. tor_addr_parse(&a, "8.8.8.8");
  918. tor_addr_parse(&b, "1.1.1.1");
  919. tt_int_op(addrs_in_same_network_family(&a, &b), OP_EQ, 0);
  920. tor_addr_parse(&a, "8.8.8.8");
  921. tor_addr_parse(&b, "2001:4860:4860::8844");
  922. tt_int_op(addrs_in_same_network_family(&a, &b), OP_EQ, 0);
  923. tor_addr_parse(&a, "2001:4860:4860::8888");
  924. tor_addr_parse(&b, "2001:4860:4860::8844");
  925. tt_int_op(addrs_in_same_network_family(&a, &b), OP_EQ, 1);
  926. tor_addr_parse(&a, "2001:4860:4860::8888");
  927. tor_addr_parse(&b, "2001:470:20::2");
  928. tt_int_op(addrs_in_same_network_family(&a, &b), OP_EQ, 0);
  929. done:
  930. return;
  931. }
  932. static node_t *
  933. helper_create_mock_node(char id_char)
  934. {
  935. node_t *node = tor_malloc_zero(sizeof(node_t));
  936. routerinfo_t *ri = tor_malloc_zero(sizeof(routerinfo_t));
  937. tor_addr_make_null(&ri->ipv6_addr, AF_INET6);
  938. node->ri = ri;
  939. memset(node->identity, id_char, sizeof(node->identity));
  940. return node;
  941. }
  942. static void
  943. helper_free_mock_node(node_t *node)
  944. {
  945. if (!node)
  946. return;
  947. tor_free(node->ri);
  948. tor_free(node);
  949. }
  950. #define NODE_SET_IPV4(node, ipv4_addr, ipv4_port) { \
  951. tor_addr_t addr; \
  952. tor_addr_parse(&addr, ipv4_addr); \
  953. node->ri->addr = tor_addr_to_ipv4h(&addr); \
  954. node->ri->or_port = ipv4_port; \
  955. }
  956. #define NODE_CLEAR_IPV4(node) { \
  957. node->ri->addr = 0; \
  958. node->ri->or_port = 0; \
  959. }
  960. #define NODE_SET_IPV6(node, ipv6_addr_str, ipv6_port) { \
  961. tor_addr_parse(&node->ri->ipv6_addr, ipv6_addr_str); \
  962. node->ri->ipv6_orport = ipv6_port; \
  963. }
  964. static void
  965. test_address_tor_node_in_same_network_family(void *ignored)
  966. {
  967. (void)ignored;
  968. node_t *node_a = helper_create_mock_node('a');
  969. node_t *node_b = helper_create_mock_node('b');
  970. NODE_SET_IPV4(node_a, "8.8.8.8", 1);
  971. NODE_SET_IPV4(node_b, "8.8.4.4", 1);
  972. tt_int_op(nodes_in_same_family(node_a, node_b), OP_EQ, 1);
  973. NODE_SET_IPV4(node_a, "8.8.8.8", 1);
  974. NODE_SET_IPV4(node_b, "1.1.1.1", 1);
  975. tt_int_op(nodes_in_same_family(node_a, node_b), OP_EQ, 0);
  976. NODE_CLEAR_IPV4(node_a);
  977. NODE_SET_IPV6(node_a, "2001:470:20::2", 1);
  978. tt_int_op(nodes_in_same_family(node_a, node_b), OP_EQ, 0);
  979. NODE_CLEAR_IPV4(node_b);
  980. NODE_SET_IPV6(node_b, "2606:4700:4700::1111", 1);
  981. tt_int_op(nodes_in_same_family(node_a, node_b), OP_EQ, 0);
  982. NODE_SET_IPV6(node_a, "2606:4700:4700::1001", 1);
  983. tt_int_op(nodes_in_same_family(node_a, node_b), OP_EQ, 1);
  984. done:
  985. helper_free_mock_node(node_a);
  986. helper_free_mock_node(node_b);
  987. }
  988. static or_options_t mock_options;
  989. static const or_options_t *
  990. mock_get_options(void)
  991. {
  992. return &mock_options;
  993. }
  994. /* Test dirserv_router_has_valid_address() on a stub routerinfo, with only its
  995. * address fields set. Use IPv4 ipv4_addr_str and IPv6 ipv6_addr_str.
  996. * Fail if it does not return rv. */
  997. #define TEST_ROUTER_VALID_ADDRESS_HELPER(ipv4_addr_str, ipv6_addr_str, rv) \
  998. STMT_BEGIN \
  999. ri = tor_malloc_zero(sizeof(routerinfo_t)); \
  1000. tor_addr_t addr; \
  1001. tor_addr_parse(&addr, (ipv4_addr_str)); \
  1002. ri->addr = tor_addr_to_ipv4h(&addr); \
  1003. tor_addr_parse(&ri->ipv6_addr, (ipv6_addr_str)); \
  1004. tt_int_op(dirserv_router_has_valid_address(ri), OP_EQ, (rv)); \
  1005. tor_free(ri); \
  1006. STMT_END
  1007. /* Like TEST_ROUTER_VALID_ADDRESS_HELPER(), but always passes a null
  1008. * IPv6 address. */
  1009. #define CHECK_RI_ADDR(ipv4_addr_str, rv) \
  1010. TEST_ROUTER_VALID_ADDRESS_HELPER(ipv4_addr_str, "::", rv)
  1011. /* Like TEST_ROUTER_VALID_ADDRESS_HELPER(), but always passes a non-internal
  1012. * IPv4 address, so that the IPv6 check is reached. */
  1013. #define CHECK_RI_ADDR6(ipv6_addr_str, rv) \
  1014. TEST_ROUTER_VALID_ADDRESS_HELPER("1.0.0.1", ipv6_addr_str, rv)
  1015. static void
  1016. test_address_dirserv_router_addr_private(void *opt_dir_allow_private)
  1017. {
  1018. /* A stub routerinfo structure, with only its address fields set. */
  1019. routerinfo_t *ri = NULL;
  1020. /* The expected return value for private addresses.
  1021. * Modified if DirAllowPrivateAddresses is 1. */
  1022. int private_rv = -1;
  1023. memset(&mock_options, 0, sizeof(or_options_t));
  1024. MOCK(get_options, mock_get_options);
  1025. if (opt_dir_allow_private) {
  1026. mock_options.DirAllowPrivateAddresses = 1;
  1027. private_rv = 0;
  1028. }
  1029. CHECK_RI_ADDR("1.0.0.1", 0);
  1030. CHECK_RI_ADDR("10.0.0.1", private_rv);
  1031. CHECK_RI_ADDR6("2600::1", 0);
  1032. CHECK_RI_ADDR6("fe80::1", private_rv);
  1033. /* Null addresses */
  1034. /* IPv4 null fails, regardless of IPv6 */
  1035. CHECK_RI_ADDR("0.0.0.0", private_rv);
  1036. TEST_ROUTER_VALID_ADDRESS_HELPER("0.0.0.0", "::", private_rv);
  1037. /* IPv6 null succeeds, because IPv4 is not null */
  1038. CHECK_RI_ADDR6("::", 0);
  1039. /* Byte-zeroed null addresses */
  1040. /* IPv4 null fails, regardless of IPv6 */
  1041. {
  1042. ri = tor_malloc_zero(sizeof(routerinfo_t));
  1043. tt_int_op(dirserv_router_has_valid_address(ri), OP_EQ, private_rv);
  1044. tor_free(ri);
  1045. }
  1046. /* IPv6 null succeeds, because IPv4 is not internal */
  1047. {
  1048. ri = tor_malloc_zero(sizeof(routerinfo_t));
  1049. ri->addr = 16777217; /* 1.0.0.1 */
  1050. tt_int_op(dirserv_router_has_valid_address(ri), OP_EQ, 0);
  1051. tor_free(ri);
  1052. }
  1053. done:
  1054. tor_free(ri);
  1055. UNMOCK(get_options);
  1056. }
  1057. #define ADDRESS_TEST(name, flags) \
  1058. { #name, test_address_ ## name, flags, NULL, NULL }
  1059. #define ADDRESS_TEST_STR_ARG(name, flags, str_arg) \
  1060. { #name "/" str_arg, test_address_ ## name, flags, &passthrough_setup, \
  1061. (void *)(str_arg) }
  1062. struct testcase_t address_tests[] = {
  1063. ADDRESS_TEST(udp_socket_trick_whitebox, TT_FORK),
  1064. ADDRESS_TEST(udp_socket_trick_blackbox, TT_FORK),
  1065. ADDRESS_TEST(get_if_addrs_list_internal, 0),
  1066. ADDRESS_TEST(get_if_addrs_list_no_internal, 0),
  1067. ADDRESS_TEST(get_if_addrs6_list_internal, 0),
  1068. ADDRESS_TEST(get_if_addrs6_list_no_internal, TT_FORK),
  1069. ADDRESS_TEST(get_if_addrs_internal_fail, 0),
  1070. ADDRESS_TEST(get_if_addrs_no_internal_fail, 0),
  1071. ADDRESS_TEST(get_if_addrs, 0),
  1072. ADDRESS_TEST(get_if_addrs6, 0),
  1073. #ifdef HAVE_IFADDRS_TO_SMARTLIST
  1074. ADDRESS_TEST(get_if_addrs_ifaddrs, TT_FORK),
  1075. ADDRESS_TEST(ifaddrs_to_smartlist, 0),
  1076. #endif
  1077. #ifdef HAVE_IP_ADAPTER_TO_SMARTLIST
  1078. ADDRESS_TEST(get_if_addrs_win32, TT_FORK),
  1079. ADDRESS_TEST(ip_adapter_addresses_to_smartlist, 0),
  1080. #endif
  1081. #ifdef HAVE_IFCONF_TO_SMARTLIST
  1082. ADDRESS_TEST(get_if_addrs_ioctl, TT_FORK),
  1083. ADDRESS_TEST(ifreq_to_smartlist, 0),
  1084. #endif
  1085. ADDRESS_TEST(tor_addr_to_in6, 0),
  1086. ADDRESS_TEST(tor_addr_to_in, 0),
  1087. ADDRESS_TEST(tor_addr_to_ipv4n, 0),
  1088. ADDRESS_TEST(tor_addr_to_mapped_ipv4h, 0),
  1089. ADDRESS_TEST(tor_addr_eq_ipv4h, 0),
  1090. ADDRESS_TEST(tor_addr_in_same_network_family, 0),
  1091. ADDRESS_TEST(tor_node_in_same_network_family, 0),
  1092. ADDRESS_TEST(dirserv_router_addr_private, 0),
  1093. ADDRESS_TEST_STR_ARG(dirserv_router_addr_private, 0, "allow_private"),
  1094. END_OF_TESTCASES
  1095. };