test_addr.c 39 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064
  1. /* Copyright (c) 2001-2004, Roger Dingledine.
  2. * Copyright (c) 2004-2006, Roger Dingledine, Nick Mathewson.
  3. * Copyright (c) 2007-2013, The Tor Project, Inc. */
  4. /* See LICENSE for licensing information */
  5. #define ADDRESSMAP_PRIVATE
  6. #include "orconfig.h"
  7. #include "or.h"
  8. #include "test.h"
  9. #include "addressmap.h"
  10. static void
  11. test_addr_basic(void)
  12. {
  13. uint32_t u32;
  14. uint16_t u16;
  15. char *cp;
  16. /* Test addr_port_lookup */
  17. cp = NULL; u32 = 3; u16 = 3;
  18. test_assert(!addr_port_lookup(LOG_WARN, "1.2.3.4", &cp, &u32, &u16));
  19. test_streq(cp, "1.2.3.4");
  20. test_eq(u32, 0x01020304u);
  21. test_eq(u16, 0);
  22. tor_free(cp);
  23. test_assert(!addr_port_lookup(LOG_WARN, "4.3.2.1:99", &cp, &u32, &u16));
  24. test_streq(cp, "4.3.2.1");
  25. test_eq(u32, 0x04030201u);
  26. test_eq(u16, 99);
  27. tor_free(cp);
  28. test_assert(!addr_port_lookup(LOG_WARN, "nonexistent.address:4040",
  29. &cp, NULL, &u16));
  30. test_streq(cp, "nonexistent.address");
  31. test_eq(u16, 4040);
  32. tor_free(cp);
  33. test_assert(!addr_port_lookup(LOG_WARN, "localhost:9999", &cp, &u32, &u16));
  34. test_streq(cp, "localhost");
  35. test_eq(u32, 0x7f000001u);
  36. test_eq(u16, 9999);
  37. tor_free(cp);
  38. u32 = 3;
  39. test_assert(!addr_port_lookup(LOG_WARN, "localhost", NULL, &u32, &u16));
  40. test_eq_ptr(cp, NULL);
  41. test_eq(u32, 0x7f000001u);
  42. test_eq(u16, 0);
  43. tor_free(cp);
  44. test_assert(addr_port_lookup(LOG_WARN, "localhost:3", &cp, &u32, NULL));
  45. tor_free(cp);
  46. test_eq(0, addr_mask_get_bits(0x0u));
  47. test_eq(32, addr_mask_get_bits(0xFFFFFFFFu));
  48. test_eq(16, addr_mask_get_bits(0xFFFF0000u));
  49. test_eq(31, addr_mask_get_bits(0xFFFFFFFEu));
  50. test_eq(1, addr_mask_get_bits(0x80000000u));
  51. /* Test inet_ntop */
  52. {
  53. char tmpbuf[TOR_ADDR_BUF_LEN];
  54. const char *ip = "176.192.208.224";
  55. struct in_addr in;
  56. /* good round trip */
  57. test_eq(tor_inet_pton(AF_INET, ip, &in), 1);
  58. test_eq_ptr(tor_inet_ntop(AF_INET, &in, tmpbuf, sizeof(tmpbuf)), &tmpbuf);
  59. test_streq(tmpbuf, ip);
  60. /* just enough buffer length */
  61. test_streq(tor_inet_ntop(AF_INET, &in, tmpbuf, strlen(ip) + 1), ip);
  62. /* too short buffer */
  63. test_eq_ptr(tor_inet_ntop(AF_INET, &in, tmpbuf, strlen(ip)), NULL);
  64. }
  65. done:
  66. tor_free(cp);
  67. }
  68. #define test_op_ip6_(a,op,b,e1,e2) \
  69. STMT_BEGIN \
  70. tt_assert_test_fmt_type(a,b,e1" "#op" "e2,struct in6_addr*, \
  71. (memcmp(val1_->s6_addr, val2_->s6_addr, 16) op 0), \
  72. char *, "%s", \
  73. { int i; char *cp; \
  74. cp = print_ = tor_malloc(64); \
  75. for (i=0;i<16;++i) { \
  76. tor_snprintf(cp, 3,"%02x", (unsigned)value_->s6_addr[i]);\
  77. cp += 2; \
  78. if (i != 15) *cp++ = ':'; \
  79. } \
  80. }, \
  81. { tor_free(print_); }, \
  82. TT_EXIT_TEST_FUNCTION \
  83. ); \
  84. STMT_END
  85. /** Helper: Assert that two strings both decode as IPv6 addresses with
  86. * tor_inet_pton(), and both decode to the same address. */
  87. #define test_pton6_same(a,b) STMT_BEGIN \
  88. test_eq(tor_inet_pton(AF_INET6, a, &a1), 1); \
  89. test_eq(tor_inet_pton(AF_INET6, b, &a2), 1); \
  90. test_op_ip6_(&a1,==,&a2,#a,#b); \
  91. STMT_END
  92. /** Helper: Assert that <b>a</b> is recognized as a bad IPv6 address by
  93. * tor_inet_pton(). */
  94. #define test_pton6_bad(a) \
  95. test_eq(0, tor_inet_pton(AF_INET6, a, &a1))
  96. /** Helper: assert that <b>a</b>, when parsed by tor_inet_pton() and displayed
  97. * with tor_inet_ntop(), yields <b>b</b>. Also assert that <b>b</b> parses to
  98. * the same value as <b>a</b>. */
  99. #define test_ntop6_reduces(a,b) STMT_BEGIN \
  100. test_eq(tor_inet_pton(AF_INET6, a, &a1), 1); \
  101. test_streq(tor_inet_ntop(AF_INET6, &a1, buf, sizeof(buf)), b); \
  102. test_eq(tor_inet_pton(AF_INET6, b, &a2), 1); \
  103. test_op_ip6_(&a1, ==, &a2, a, b); \
  104. STMT_END
  105. /** Helper: assert that <b>a</b> parses by tor_inet_pton() into a address that
  106. * passes tor_addr_is_internal() with <b>for_listening</b>. */
  107. #define test_internal_ip(a,for_listening) STMT_BEGIN \
  108. test_eq(tor_inet_pton(AF_INET6, a, &t1.addr.in6_addr), 1); \
  109. t1.family = AF_INET6; \
  110. if (!tor_addr_is_internal(&t1, for_listening)) \
  111. test_fail_msg( a "was not internal."); \
  112. STMT_END
  113. /** Helper: assert that <b>a</b> parses by tor_inet_pton() into a address that
  114. * does not pass tor_addr_is_internal() with <b>for_listening</b>. */
  115. #define test_external_ip(a,for_listening) STMT_BEGIN \
  116. test_eq(tor_inet_pton(AF_INET6, a, &t1.addr.in6_addr), 1); \
  117. t1.family = AF_INET6; \
  118. if (tor_addr_is_internal(&t1, for_listening)) \
  119. test_fail_msg(a "was not external."); \
  120. STMT_END
  121. /** Helper: Assert that <b>a</b> and <b>b</b>, when parsed by
  122. * tor_inet_pton(), give addresses that compare in the order defined by
  123. * <b>op</b> with tor_addr_compare(). */
  124. #define test_addr_compare(a, op, b) STMT_BEGIN \
  125. test_eq(tor_inet_pton(AF_INET6, a, &t1.addr.in6_addr), 1); \
  126. test_eq(tor_inet_pton(AF_INET6, b, &t2.addr.in6_addr), 1); \
  127. t1.family = t2.family = AF_INET6; \
  128. r = tor_addr_compare(&t1,&t2,CMP_SEMANTIC); \
  129. if (!(r op 0)) \
  130. test_fail_msg("failed: tor_addr_compare("a","b") "#op" 0"); \
  131. STMT_END
  132. /** Helper: Assert that <b>a</b> and <b>b</b>, when parsed by
  133. * tor_inet_pton(), give addresses that compare in the order defined by
  134. * <b>op</b> with tor_addr_compare_masked() with <b>m</b> masked. */
  135. #define test_addr_compare_masked(a, op, b, m) STMT_BEGIN \
  136. test_eq(tor_inet_pton(AF_INET6, a, &t1.addr.in6_addr), 1); \
  137. test_eq(tor_inet_pton(AF_INET6, b, &t2.addr.in6_addr), 1); \
  138. t1.family = t2.family = AF_INET6; \
  139. r = tor_addr_compare_masked(&t1,&t2,m,CMP_SEMANTIC); \
  140. if (!(r op 0)) \
  141. test_fail_msg("failed: tor_addr_compare_masked("a","b","#m") "#op" 0"); \
  142. STMT_END
  143. /** Helper: assert that <b>xx</b> is parseable as a masked IPv6 address with
  144. * ports by tor_parse_mask_addr_ports(), with family <b>f</b>, IP address
  145. * as 4 32-bit words <b>ip1...ip4</b>, mask bits as <b>mm</b>, and port range
  146. * as <b>pt1..pt2</b>. */
  147. #define test_addr_mask_ports_parse(xx, f, ip1, ip2, ip3, ip4, mm, pt1, pt2) \
  148. STMT_BEGIN \
  149. test_eq(tor_addr_parse_mask_ports(xx, 0, &t1, &mask, &port1, &port2), \
  150. f); \
  151. p1=tor_inet_ntop(AF_INET6, &t1.addr.in6_addr, bug, sizeof(bug)); \
  152. test_eq(htonl(ip1), tor_addr_to_in6_addr32(&t1)[0]); \
  153. test_eq(htonl(ip2), tor_addr_to_in6_addr32(&t1)[1]); \
  154. test_eq(htonl(ip3), tor_addr_to_in6_addr32(&t1)[2]); \
  155. test_eq(htonl(ip4), tor_addr_to_in6_addr32(&t1)[3]); \
  156. test_eq(mask, mm); \
  157. test_eq(port1, pt1); \
  158. test_eq(port2, pt2); \
  159. STMT_END
  160. /** Run unit tests for IPv6 encoding/decoding/manipulation functions. */
  161. static void
  162. test_addr_ip6_helpers(void)
  163. {
  164. char buf[TOR_ADDR_BUF_LEN], bug[TOR_ADDR_BUF_LEN];
  165. char rbuf[REVERSE_LOOKUP_NAME_BUF_LEN];
  166. struct in6_addr a1, a2;
  167. tor_addr_t t1, t2;
  168. int r, i;
  169. uint16_t port1, port2;
  170. maskbits_t mask;
  171. const char *p1;
  172. struct sockaddr_storage sa_storage;
  173. struct sockaddr_in *sin;
  174. struct sockaddr_in6 *sin6;
  175. /* Test tor_inet_ntop and tor_inet_pton: IPv6 */
  176. {
  177. const char *ip = "2001::1234";
  178. const char *ip_ffff = "::ffff:192.168.1.2";
  179. /* good round trip */
  180. test_eq(tor_inet_pton(AF_INET6, ip, &a1), 1);
  181. test_eq_ptr(tor_inet_ntop(AF_INET6, &a1, buf, sizeof(buf)), &buf);
  182. test_streq(buf, ip);
  183. /* good round trip - ::ffff:0:0 style */
  184. test_eq(tor_inet_pton(AF_INET6, ip_ffff, &a2), 1);
  185. test_eq_ptr(tor_inet_ntop(AF_INET6, &a2, buf, sizeof(buf)), &buf);
  186. test_streq(buf, ip_ffff);
  187. /* just long enough buffer (remember \0) */
  188. test_streq(tor_inet_ntop(AF_INET6, &a1, buf, strlen(ip)+1), ip);
  189. test_streq(tor_inet_ntop(AF_INET6, &a2, buf, strlen(ip_ffff)+1),
  190. ip_ffff);
  191. /* too short buffer (remember \0) */
  192. test_eq_ptr(tor_inet_ntop(AF_INET6, &a1, buf, strlen(ip)), NULL);
  193. test_eq_ptr(tor_inet_ntop(AF_INET6, &a2, buf, strlen(ip_ffff)), NULL);
  194. }
  195. /* ==== Converting to and from sockaddr_t. */
  196. sin = (struct sockaddr_in *)&sa_storage;
  197. sin->sin_family = AF_INET;
  198. sin->sin_port = htons(9090);
  199. sin->sin_addr.s_addr = htonl(0x7f7f0102); /*127.127.1.2*/
  200. tor_addr_from_sockaddr(&t1, (struct sockaddr *)sin, &port1);
  201. test_eq(tor_addr_family(&t1), AF_INET);
  202. test_eq(tor_addr_to_ipv4h(&t1), 0x7f7f0102);
  203. tt_int_op(port1, ==, 9090);
  204. memset(&sa_storage, 0, sizeof(sa_storage));
  205. test_eq(sizeof(struct sockaddr_in),
  206. tor_addr_to_sockaddr(&t1, 1234, (struct sockaddr *)&sa_storage,
  207. sizeof(sa_storage)));
  208. test_eq(1234, ntohs(sin->sin_port));
  209. test_eq(0x7f7f0102, ntohl(sin->sin_addr.s_addr));
  210. memset(&sa_storage, 0, sizeof(sa_storage));
  211. sin6 = (struct sockaddr_in6 *)&sa_storage;
  212. sin6->sin6_family = AF_INET6;
  213. sin6->sin6_port = htons(7070);
  214. sin6->sin6_addr.s6_addr[0] = 128;
  215. tor_addr_from_sockaddr(&t1, (struct sockaddr *)sin6, &port1);
  216. test_eq(tor_addr_family(&t1), AF_INET6);
  217. tt_int_op(port1, ==, 7070);
  218. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 0);
  219. test_streq(p1, "8000::");
  220. memset(&sa_storage, 0, sizeof(sa_storage));
  221. test_eq(sizeof(struct sockaddr_in6),
  222. tor_addr_to_sockaddr(&t1, 9999, (struct sockaddr *)&sa_storage,
  223. sizeof(sa_storage)));
  224. test_eq(AF_INET6, sin6->sin6_family);
  225. test_eq(9999, ntohs(sin6->sin6_port));
  226. test_eq(0x80000000, ntohl(S6_ADDR32(sin6->sin6_addr)[0]));
  227. /* ==== tor_addr_lookup: static cases. (Can't test dns without knowing we
  228. * have a good resolver. */
  229. test_eq(0, tor_addr_lookup("127.128.129.130", AF_UNSPEC, &t1));
  230. test_eq(AF_INET, tor_addr_family(&t1));
  231. test_eq(tor_addr_to_ipv4h(&t1), 0x7f808182);
  232. test_eq(0, tor_addr_lookup("9000::5", AF_UNSPEC, &t1));
  233. test_eq(AF_INET6, tor_addr_family(&t1));
  234. test_eq(0x90, tor_addr_to_in6_addr8(&t1)[0]);
  235. test_assert(tor_mem_is_zero((char*)tor_addr_to_in6_addr8(&t1)+1, 14));
  236. test_eq(0x05, tor_addr_to_in6_addr8(&t1)[15]);
  237. /* === Test pton: valid af_inet6 */
  238. /* Simple, valid parsing. */
  239. r = tor_inet_pton(AF_INET6,
  240. "0102:0304:0506:0708:090A:0B0C:0D0E:0F10", &a1);
  241. test_assert(r==1);
  242. for (i=0;i<16;++i) { test_eq(i+1, (int)a1.s6_addr[i]); }
  243. /* ipv4 ending. */
  244. test_pton6_same("0102:0304:0506:0708:090A:0B0C:0D0E:0F10",
  245. "0102:0304:0506:0708:090A:0B0C:13.14.15.16");
  246. /* shortened words. */
  247. test_pton6_same("0001:0099:BEEF:0000:0123:FFFF:0001:0001",
  248. "1:99:BEEF:0:0123:FFFF:1:1");
  249. /* zeros at the beginning */
  250. test_pton6_same("0000:0000:0000:0000:0009:C0A8:0001:0001",
  251. "::9:c0a8:1:1");
  252. test_pton6_same("0000:0000:0000:0000:0009:C0A8:0001:0001",
  253. "::9:c0a8:0.1.0.1");
  254. /* zeros in the middle. */
  255. test_pton6_same("fe80:0000:0000:0000:0202:1111:0001:0001",
  256. "fe80::202:1111:1:1");
  257. /* zeros at the end. */
  258. test_pton6_same("1000:0001:0000:0007:0000:0000:0000:0000",
  259. "1000:1:0:7::");
  260. /* === Test ntop: af_inet6 */
  261. test_ntop6_reduces("0:0:0:0:0:0:0:0", "::");
  262. test_ntop6_reduces("0001:0099:BEEF:0006:0123:FFFF:0001:0001",
  263. "1:99:beef:6:123:ffff:1:1");
  264. //test_ntop6_reduces("0:0:0:0:0:0:c0a8:0101", "::192.168.1.1");
  265. test_ntop6_reduces("0:0:0:0:0:ffff:c0a8:0101", "::ffff:192.168.1.1");
  266. test_ntop6_reduces("002:0:0000:0:3::4", "2::3:0:0:4");
  267. test_ntop6_reduces("0:0::1:0:3", "::1:0:3");
  268. test_ntop6_reduces("008:0::0", "8::");
  269. test_ntop6_reduces("0:0:0:0:0:ffff::1", "::ffff:0.0.0.1");
  270. test_ntop6_reduces("abcd:0:0:0:0:0:7f00::", "abcd::7f00:0");
  271. test_ntop6_reduces("0000:0000:0000:0000:0009:C0A8:0001:0001",
  272. "::9:c0a8:1:1");
  273. test_ntop6_reduces("fe80:0000:0000:0000:0202:1111:0001:0001",
  274. "fe80::202:1111:1:1");
  275. test_ntop6_reduces("1000:0001:0000:0007:0000:0000:0000:0000",
  276. "1000:1:0:7::");
  277. /* Bad af param */
  278. test_eq(tor_inet_pton(AF_UNSPEC, 0, 0), -1);
  279. /* === Test pton: invalid in6. */
  280. test_pton6_bad("foobar.");
  281. test_pton6_bad("-1::");
  282. test_pton6_bad("00001::");
  283. test_pton6_bad("10000::");
  284. test_pton6_bad("::10000");
  285. test_pton6_bad("55555::");
  286. test_pton6_bad("9:-60::");
  287. test_pton6_bad("9:+60::");
  288. test_pton6_bad("9|60::");
  289. test_pton6_bad("0x60::");
  290. test_pton6_bad("::0x60");
  291. test_pton6_bad("9:0x60::");
  292. test_pton6_bad("1:2:33333:4:0002:3::");
  293. test_pton6_bad("1:2:3333:4:fish:3::");
  294. test_pton6_bad("1:2:3:4:5:6:7:8:9");
  295. test_pton6_bad("1:2:3:4:5:6:7");
  296. test_pton6_bad("1:2:3:4:5:6:1.2.3.4.5");
  297. test_pton6_bad("1:2:3:4:5:6:1.2.3");
  298. test_pton6_bad("::1.2.3");
  299. test_pton6_bad("::1.2.3.4.5");
  300. test_pton6_bad("::ffff:0xff.0.0.0");
  301. test_pton6_bad("::ffff:ff.0.0.0");
  302. test_pton6_bad("::ffff:256.0.0.0");
  303. test_pton6_bad("::ffff:-1.0.0.0");
  304. test_pton6_bad("99");
  305. test_pton6_bad("");
  306. test_pton6_bad(".");
  307. test_pton6_bad(":");
  308. test_pton6_bad("1::2::3:4");
  309. test_pton6_bad("a:::b:c");
  310. test_pton6_bad(":::a:b:c");
  311. test_pton6_bad("a:b:c:::");
  312. test_pton6_bad("1.2.3.4");
  313. test_pton6_bad(":1.2.3.4");
  314. test_pton6_bad(".2.3.4");
  315. /* test internal checking */
  316. test_external_ip("fbff:ffff::2:7", 0);
  317. test_internal_ip("fc01::2:7", 0);
  318. test_internal_ip("fc01::02:7", 0);
  319. test_internal_ip("fc01::002:7", 0);
  320. test_internal_ip("fc01::0002:7", 0);
  321. test_internal_ip("fdff:ffff::f:f", 0);
  322. test_external_ip("fe00::3:f", 0);
  323. test_external_ip("fe7f:ffff::2:7", 0);
  324. test_internal_ip("fe80::2:7", 0);
  325. test_internal_ip("febf:ffff::f:f", 0);
  326. test_internal_ip("fec0::2:7:7", 0);
  327. test_internal_ip("feff:ffff::e:7:7", 0);
  328. test_external_ip("ff00::e:7:7", 0);
  329. test_internal_ip("::", 0);
  330. test_internal_ip("::1", 0);
  331. test_internal_ip("::1", 1);
  332. test_internal_ip("::", 0);
  333. test_external_ip("::", 1);
  334. test_external_ip("::2", 0);
  335. test_external_ip("2001::", 0);
  336. test_external_ip("ffff::", 0);
  337. test_external_ip("::ffff:0.0.0.0", 1);
  338. test_internal_ip("::ffff:0.0.0.0", 0);
  339. test_internal_ip("::ffff:0.255.255.255", 0);
  340. test_external_ip("::ffff:1.0.0.0", 0);
  341. test_external_ip("::ffff:9.255.255.255", 0);
  342. test_internal_ip("::ffff:10.0.0.0", 0);
  343. test_internal_ip("::ffff:10.255.255.255", 0);
  344. test_external_ip("::ffff:11.0.0.0", 0);
  345. test_external_ip("::ffff:126.255.255.255", 0);
  346. test_internal_ip("::ffff:127.0.0.0", 0);
  347. test_internal_ip("::ffff:127.255.255.255", 0);
  348. test_external_ip("::ffff:128.0.0.0", 0);
  349. test_external_ip("::ffff:172.15.255.255", 0);
  350. test_internal_ip("::ffff:172.16.0.0", 0);
  351. test_internal_ip("::ffff:172.31.255.255", 0);
  352. test_external_ip("::ffff:172.32.0.0", 0);
  353. test_external_ip("::ffff:192.167.255.255", 0);
  354. test_internal_ip("::ffff:192.168.0.0", 0);
  355. test_internal_ip("::ffff:192.168.255.255", 0);
  356. test_external_ip("::ffff:192.169.0.0", 0);
  357. test_external_ip("::ffff:169.253.255.255", 0);
  358. test_internal_ip("::ffff:169.254.0.0", 0);
  359. test_internal_ip("::ffff:169.254.255.255", 0);
  360. test_external_ip("::ffff:169.255.0.0", 0);
  361. /* tor_addr_compare(tor_addr_t x2) */
  362. test_addr_compare("ffff::", ==, "ffff::0");
  363. test_addr_compare("0::3:2:1", <, "0::ffff:0.3.2.1");
  364. test_addr_compare("0::2:2:1", <, "0::ffff:0.3.2.1");
  365. test_addr_compare("0::ffff:0.3.2.1", >, "0::0:0:0");
  366. test_addr_compare("0::ffff:5.2.2.1", <, "::ffff:6.0.0.0"); /* XXXX wrong. */
  367. tor_addr_parse_mask_ports("[::ffff:2.3.4.5]", 0, &t1, NULL, NULL, NULL);
  368. tor_addr_parse_mask_ports("2.3.4.5", 0, &t2, NULL, NULL, NULL);
  369. test_assert(tor_addr_compare(&t1, &t2, CMP_SEMANTIC) == 0);
  370. tor_addr_parse_mask_ports("[::ffff:2.3.4.4]", 0, &t1, NULL, NULL, NULL);
  371. tor_addr_parse_mask_ports("2.3.4.5", 0, &t2, NULL, NULL, NULL);
  372. test_assert(tor_addr_compare(&t1, &t2, CMP_SEMANTIC) < 0);
  373. /* test compare_masked */
  374. test_addr_compare_masked("ffff::", ==, "ffff::0", 128);
  375. test_addr_compare_masked("ffff::", ==, "ffff::0", 64);
  376. test_addr_compare_masked("0::2:2:1", <, "0::8000:2:1", 81);
  377. test_addr_compare_masked("0::2:2:1", ==, "0::8000:2:1", 80);
  378. /* Test undecorated tor_addr_to_str */
  379. test_eq(AF_INET6, tor_addr_parse(&t1, "[123:45:6789::5005:11]"));
  380. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 0);
  381. test_streq(p1, "123:45:6789::5005:11");
  382. test_eq(AF_INET, tor_addr_parse(&t1, "18.0.0.1"));
  383. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 0);
  384. test_streq(p1, "18.0.0.1");
  385. /* Test decorated tor_addr_to_str */
  386. test_eq(AF_INET6, tor_addr_parse(&t1, "[123:45:6789::5005:11]"));
  387. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  388. test_streq(p1, "[123:45:6789::5005:11]");
  389. test_eq(AF_INET, tor_addr_parse(&t1, "18.0.0.1"));
  390. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  391. test_streq(p1, "18.0.0.1");
  392. /* Test buffer bounds checking of tor_addr_to_str */
  393. test_eq(AF_INET6, tor_addr_parse(&t1, "::")); /* 2 + \0 */
  394. test_eq_ptr(tor_addr_to_str(buf, &t1, 2, 0), NULL); /* too short buf */
  395. test_streq(tor_addr_to_str(buf, &t1, 3, 0), "::");
  396. test_eq_ptr(tor_addr_to_str(buf, &t1, 4, 1), NULL); /* too short buf */
  397. test_streq(tor_addr_to_str(buf, &t1, 5, 1), "[::]");
  398. test_eq(AF_INET6, tor_addr_parse(&t1, "2000::1337")); /* 10 + \0 */
  399. test_eq_ptr(tor_addr_to_str(buf, &t1, 10, 0), NULL); /* too short buf */
  400. test_streq(tor_addr_to_str(buf, &t1, 11, 0), "2000::1337");
  401. test_eq_ptr(tor_addr_to_str(buf, &t1, 12, 1), NULL); /* too short buf */
  402. test_streq(tor_addr_to_str(buf, &t1, 13, 1), "[2000::1337]");
  403. test_eq(AF_INET, tor_addr_parse(&t1, "1.2.3.4")); /* 7 + \0 */
  404. test_eq_ptr(tor_addr_to_str(buf, &t1, 7, 0), NULL); /* too short buf */
  405. test_streq(tor_addr_to_str(buf, &t1, 8, 0), "1.2.3.4");
  406. test_eq(AF_INET, tor_addr_parse(&t1, "255.255.255.255")); /* 15 + \0 */
  407. test_eq_ptr(tor_addr_to_str(buf, &t1, 15, 0), NULL); /* too short buf */
  408. test_streq(tor_addr_to_str(buf, &t1, 16, 0), "255.255.255.255");
  409. test_eq_ptr(tor_addr_to_str(buf, &t1, 15, 1), NULL); /* too short buf */
  410. test_streq(tor_addr_to_str(buf, &t1, 16, 1), "255.255.255.255");
  411. t1.family = AF_UNSPEC;
  412. test_eq_ptr(tor_addr_to_str(buf, &t1, sizeof(buf), 0), NULL);
  413. /* Test tor_addr_parse_PTR_name */
  414. i = tor_addr_parse_PTR_name(&t1, "Foobar.baz", AF_UNSPEC, 0);
  415. test_eq(0, i);
  416. i = tor_addr_parse_PTR_name(&t1, "Foobar.baz", AF_UNSPEC, 1);
  417. test_eq(0, i);
  418. i = tor_addr_parse_PTR_name(&t1, "9999999999999999999999999999.in-addr.arpa",
  419. AF_UNSPEC, 1);
  420. test_eq(-1, i);
  421. i = tor_addr_parse_PTR_name(&t1, "1.0.168.192.in-addr.arpa",
  422. AF_UNSPEC, 1);
  423. test_eq(1, i);
  424. test_eq(tor_addr_family(&t1), AF_INET);
  425. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  426. test_streq(p1, "192.168.0.1");
  427. i = tor_addr_parse_PTR_name(&t1, "192.168.0.99", AF_UNSPEC, 0);
  428. test_eq(0, i);
  429. i = tor_addr_parse_PTR_name(&t1, "192.168.0.99", AF_UNSPEC, 1);
  430. test_eq(1, i);
  431. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  432. test_streq(p1, "192.168.0.99");
  433. memset(&t1, 0, sizeof(t1));
  434. i = tor_addr_parse_PTR_name(&t1,
  435. "0.1.2.3.4.5.6.7.8.9.a.b.c.d.e.f."
  436. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  437. "ip6.ARPA",
  438. AF_UNSPEC, 0);
  439. test_eq(1, i);
  440. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  441. test_streq(p1, "[9dee:effe:ebe1:beef:fedc:ba98:7654:3210]");
  442. /* Failing cases. */
  443. i = tor_addr_parse_PTR_name(&t1,
  444. "6.7.8.9.a.b.c.d.e.f."
  445. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  446. "ip6.ARPA",
  447. AF_UNSPEC, 0);
  448. test_eq(i, -1);
  449. i = tor_addr_parse_PTR_name(&t1,
  450. "6.7.8.9.a.b.c.d.e.f.a.b.c.d.e.f.0."
  451. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  452. "ip6.ARPA",
  453. AF_UNSPEC, 0);
  454. test_eq(i, -1);
  455. i = tor_addr_parse_PTR_name(&t1,
  456. "6.7.8.9.a.b.c.d.e.f.X.0.0.0.0.9."
  457. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  458. "ip6.ARPA",
  459. AF_UNSPEC, 0);
  460. test_eq(i, -1);
  461. i = tor_addr_parse_PTR_name(&t1, "32.1.1.in-addr.arpa",
  462. AF_UNSPEC, 0);
  463. test_eq(i, -1);
  464. i = tor_addr_parse_PTR_name(&t1, ".in-addr.arpa",
  465. AF_UNSPEC, 0);
  466. test_eq(i, -1);
  467. i = tor_addr_parse_PTR_name(&t1, "1.2.3.4.5.in-addr.arpa",
  468. AF_UNSPEC, 0);
  469. test_eq(i, -1);
  470. i = tor_addr_parse_PTR_name(&t1, "1.2.3.4.5.in-addr.arpa",
  471. AF_INET6, 0);
  472. test_eq(i, -1);
  473. i = tor_addr_parse_PTR_name(&t1,
  474. "6.7.8.9.a.b.c.d.e.f.a.b.c.d.e.0."
  475. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  476. "ip6.ARPA",
  477. AF_INET, 0);
  478. test_eq(i, -1);
  479. /* === Test tor_addr_to_PTR_name */
  480. /* Stage IPv4 addr */
  481. memset(&sa_storage, 0, sizeof(sa_storage));
  482. sin = (struct sockaddr_in *)&sa_storage;
  483. sin->sin_family = AF_INET;
  484. sin->sin_addr.s_addr = htonl(0x7f010203); /* 127.1.2.3 */
  485. tor_addr_from_sockaddr(&t1, (struct sockaddr *)sin, NULL);
  486. /* Check IPv4 PTR - too short buffer */
  487. test_eq(tor_addr_to_PTR_name(rbuf, 1, &t1), -1);
  488. test_eq(tor_addr_to_PTR_name(rbuf,
  489. strlen("3.2.1.127.in-addr.arpa") - 1,
  490. &t1), -1);
  491. /* Check IPv4 PTR - valid addr */
  492. test_eq(tor_addr_to_PTR_name(rbuf, sizeof(rbuf), &t1),
  493. strlen("3.2.1.127.in-addr.arpa"));
  494. test_streq(rbuf, "3.2.1.127.in-addr.arpa");
  495. /* Invalid addr family */
  496. t1.family = AF_UNSPEC;
  497. test_eq(tor_addr_to_PTR_name(rbuf, sizeof(rbuf), &t1), -1);
  498. /* Stage IPv6 addr */
  499. memset(&sa_storage, 0, sizeof(sa_storage));
  500. sin6 = (struct sockaddr_in6 *)&sa_storage;
  501. sin6->sin6_family = AF_INET6;
  502. sin6->sin6_addr.s6_addr[0] = 0x80; /* 8000::abcd */
  503. sin6->sin6_addr.s6_addr[14] = 0xab;
  504. sin6->sin6_addr.s6_addr[15] = 0xcd;
  505. tor_addr_from_sockaddr(&t1, (struct sockaddr *)sin6, NULL);
  506. {
  507. const char* addr_PTR = "d.c.b.a.0.0.0.0.0.0.0.0.0.0.0.0."
  508. "0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.8.ip6.arpa";
  509. /* Check IPv6 PTR - too short buffer */
  510. test_eq(tor_addr_to_PTR_name(rbuf, 0, &t1), -1);
  511. test_eq(tor_addr_to_PTR_name(rbuf, strlen(addr_PTR) - 1, &t1), -1);
  512. /* Check IPv6 PTR - valid addr */
  513. test_eq(tor_addr_to_PTR_name(rbuf, sizeof(rbuf), &t1),
  514. strlen(addr_PTR));
  515. test_streq(rbuf, addr_PTR);
  516. }
  517. /* XXXX turn this into a separate function; it's not all IPv6. */
  518. /* test tor_addr_parse_mask_ports */
  519. test_addr_mask_ports_parse("[::f]/17:47-95", AF_INET6,
  520. 0, 0, 0, 0x0000000f, 17, 47, 95);
  521. test_streq(p1, "::f");
  522. //test_addr_parse("[::fefe:4.1.1.7/120]:999-1000");
  523. //test_addr_parse_check("::fefe:401:107", 120, 999, 1000);
  524. test_addr_mask_ports_parse("[::ffff:4.1.1.7]/120:443", AF_INET6,
  525. 0, 0, 0x0000ffff, 0x04010107, 120, 443, 443);
  526. test_streq(p1, "::ffff:4.1.1.7");
  527. test_addr_mask_ports_parse("[abcd:2::44a:0]:2-65000", AF_INET6,
  528. 0xabcd0002, 0, 0, 0x044a0000, 128, 2, 65000);
  529. test_streq(p1, "abcd:2::44a:0");
  530. /* Try some long addresses. */
  531. r=tor_addr_parse_mask_ports("[ffff:1111:1111:1111:1111:1111:1111:1111]",
  532. 0, &t1, NULL, NULL, NULL);
  533. test_assert(r == AF_INET6);
  534. r=tor_addr_parse_mask_ports("[ffff:1111:1111:1111:1111:1111:1111:11111]",
  535. 0, &t1, NULL, NULL, NULL);
  536. test_assert(r == -1);
  537. r=tor_addr_parse_mask_ports("[ffff:1111:1111:1111:1111:1111:1111:1111:1]",
  538. 0, &t1, NULL, NULL, NULL);
  539. test_assert(r == -1);
  540. r=tor_addr_parse_mask_ports(
  541. "[ffff:1111:1111:1111:1111:1111:1111:ffff:"
  542. "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:"
  543. "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:"
  544. "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff]",
  545. 0, &t1, NULL, NULL, NULL);
  546. test_assert(r == -1);
  547. /* Try some failing cases. */
  548. r=tor_addr_parse_mask_ports("[fefef::]/112", 0, &t1, NULL, NULL, NULL);
  549. test_assert(r == -1);
  550. r=tor_addr_parse_mask_ports("[fefe::/112", 0, &t1, NULL, NULL, NULL);
  551. test_assert(r == -1);
  552. r=tor_addr_parse_mask_ports("[fefe::", 0, &t1, NULL, NULL, NULL);
  553. test_assert(r == -1);
  554. r=tor_addr_parse_mask_ports("[fefe::X]", 0, &t1, NULL, NULL, NULL);
  555. test_assert(r == -1);
  556. r=tor_addr_parse_mask_ports("efef::/112", 0, &t1, NULL, NULL, NULL);
  557. test_assert(r == -1);
  558. r=tor_addr_parse_mask_ports("[f:f:f:f:f:f:f:f::]",0,&t1, NULL, NULL, NULL);
  559. test_assert(r == -1);
  560. r=tor_addr_parse_mask_ports("[::f:f:f:f:f:f:f:f]",0,&t1, NULL, NULL, NULL);
  561. test_assert(r == -1);
  562. r=tor_addr_parse_mask_ports("[f:f:f:f:f:f:f:f:f]",0,&t1, NULL, NULL, NULL);
  563. test_assert(r == -1);
  564. r=tor_addr_parse_mask_ports("[f:f:f:f:f::]/fred",0,&t1,&mask, NULL, NULL);
  565. test_assert(r == -1);
  566. r=tor_addr_parse_mask_ports("[f:f:f:f:f::]/255.255.0.0",
  567. 0,&t1, NULL, NULL, NULL);
  568. test_assert(r == -1);
  569. /* This one will get rejected because it isn't a pure prefix. */
  570. r=tor_addr_parse_mask_ports("1.1.2.3/255.255.64.0",0,&t1, &mask,NULL,NULL);
  571. test_assert(r == -1);
  572. /* Test for V4-mapped address with mask < 96. (arguably not valid) */
  573. r=tor_addr_parse_mask_ports("[::ffff:1.1.2.2/33]",0,&t1, &mask, NULL, NULL);
  574. test_assert(r == -1);
  575. r=tor_addr_parse_mask_ports("1.1.2.2/33",0,&t1, &mask, NULL, NULL);
  576. test_assert(r == -1);
  577. /* Try extended wildcard addresses with out TAPMP_EXTENDED_STAR*/
  578. r=tor_addr_parse_mask_ports("*4",0,&t1, &mask, NULL, NULL);
  579. test_assert(r == -1);
  580. r=tor_addr_parse_mask_ports("*6",0,&t1, &mask, NULL, NULL);
  581. test_assert(r == -1);
  582. #if 0
  583. /* Try a mask with a wildcard. */
  584. r=tor_addr_parse_mask_ports("*/16",0,&t1, &mask, NULL, NULL);
  585. test_assert(r == -1);
  586. r=tor_addr_parse_mask_ports("*4/16",TAPMP_EXTENDED_STAR,
  587. &t1, &mask, NULL, NULL);
  588. test_assert(r == -1);
  589. r=tor_addr_parse_mask_ports("*6/30",TAPMP_EXTENDED_STAR,
  590. &t1, &mask, NULL, NULL);
  591. test_assert(r == -1);
  592. #endif
  593. /* Basic mask tests*/
  594. r=tor_addr_parse_mask_ports("1.1.2.2/31",0,&t1, &mask, NULL, NULL);
  595. test_assert(r == AF_INET);
  596. tt_int_op(mask,==,31);
  597. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  598. tt_int_op(tor_addr_to_ipv4h(&t1),==,0x01010202);
  599. r=tor_addr_parse_mask_ports("3.4.16.032:1-2",0,&t1, &mask, &port1, &port2);
  600. test_assert(r == AF_INET);
  601. tt_int_op(mask,==,32);
  602. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  603. tt_int_op(tor_addr_to_ipv4h(&t1),==,0x03041020);
  604. test_assert(port1 == 1);
  605. test_assert(port2 == 2);
  606. r=tor_addr_parse_mask_ports("1.1.2.3/255.255.128.0",0,&t1, &mask,NULL,NULL);
  607. test_assert(r == AF_INET);
  608. tt_int_op(mask,==,17);
  609. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  610. tt_int_op(tor_addr_to_ipv4h(&t1),==,0x01010203);
  611. r=tor_addr_parse_mask_ports("[efef::]/112",0,&t1, &mask, &port1, &port2);
  612. test_assert(r == AF_INET6);
  613. test_assert(port1 == 1);
  614. test_assert(port2 == 65535);
  615. /* Try regular wildcard behavior without TAPMP_EXTENDED_STAR */
  616. r=tor_addr_parse_mask_ports("*:80-443",0,&t1,&mask,&port1,&port2);
  617. tt_int_op(r,==,AF_INET); /* Old users of this always get inet */
  618. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  619. tt_int_op(tor_addr_to_ipv4h(&t1),==,0);
  620. tt_int_op(mask,==,0);
  621. tt_int_op(port1,==,80);
  622. tt_int_op(port2,==,443);
  623. /* Now try wildcards *with* TAPMP_EXTENDED_STAR */
  624. r=tor_addr_parse_mask_ports("*:8000-9000",TAPMP_EXTENDED_STAR,
  625. &t1,&mask,&port1,&port2);
  626. tt_int_op(r,==,AF_UNSPEC);
  627. tt_int_op(tor_addr_family(&t1),==,AF_UNSPEC);
  628. tt_int_op(mask,==,0);
  629. tt_int_op(port1,==,8000);
  630. tt_int_op(port2,==,9000);
  631. r=tor_addr_parse_mask_ports("*4:6667",TAPMP_EXTENDED_STAR,
  632. &t1,&mask,&port1,&port2);
  633. tt_int_op(r,==,AF_INET);
  634. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  635. tt_int_op(tor_addr_to_ipv4h(&t1),==,0);
  636. tt_int_op(mask,==,0);
  637. tt_int_op(port1,==,6667);
  638. tt_int_op(port2,==,6667);
  639. r=tor_addr_parse_mask_ports("*6",TAPMP_EXTENDED_STAR,
  640. &t1,&mask,&port1,&port2);
  641. tt_int_op(r,==,AF_INET6);
  642. tt_int_op(tor_addr_family(&t1),==,AF_INET6);
  643. tt_assert(tor_mem_is_zero((const char*)tor_addr_to_in6_addr32(&t1), 16));
  644. tt_int_op(mask,==,0);
  645. tt_int_op(port1,==,1);
  646. tt_int_op(port2,==,65535);
  647. /* make sure inet address lengths >= max */
  648. test_assert(INET_NTOA_BUF_LEN >= sizeof("255.255.255.255"));
  649. test_assert(TOR_ADDR_BUF_LEN >=
  650. sizeof("ffff:ffff:ffff:ffff:ffff:ffff:255.255.255.255"));
  651. test_assert(sizeof(tor_addr_t) >= sizeof(struct in6_addr));
  652. /* get interface addresses */
  653. r = get_interface_address6(LOG_DEBUG, AF_INET, &t1);
  654. i = get_interface_address6(LOG_DEBUG, AF_INET6, &t2);
  655. TT_BLATHER(("v4 address: %s (family=%d)", fmt_addr(&t1),
  656. tor_addr_family(&t1)));
  657. TT_BLATHER(("v6 address: %s (family=%d)", fmt_addr(&t2),
  658. tor_addr_family(&t2)));
  659. done:
  660. ;
  661. }
  662. /** Test tor_addr_port_parse(). */
  663. static void
  664. test_addr_parse(void)
  665. {
  666. int r;
  667. tor_addr_t addr;
  668. char buf[TOR_ADDR_BUF_LEN];
  669. uint16_t port = 0;
  670. /* Correct call. */
  671. r= tor_addr_port_parse(LOG_DEBUG,
  672. "192.0.2.1:1234",
  673. &addr, &port, -1);
  674. test_assert(r == 0);
  675. tor_addr_to_str(buf, &addr, sizeof(buf), 0);
  676. test_streq(buf, "192.0.2.1");
  677. test_eq(port, 1234);
  678. r= tor_addr_port_parse(LOG_DEBUG,
  679. "[::1]:1234",
  680. &addr, &port, -1);
  681. test_assert(r == 0);
  682. tor_addr_to_str(buf, &addr, sizeof(buf), 0);
  683. test_streq(buf, "::1");
  684. test_eq(port, 1234);
  685. /* Domain name. */
  686. r= tor_addr_port_parse(LOG_DEBUG,
  687. "torproject.org:1234",
  688. &addr, &port, -1);
  689. test_assert(r == -1);
  690. /* Only IP. */
  691. r= tor_addr_port_parse(LOG_DEBUG,
  692. "192.0.2.2",
  693. &addr, &port, -1);
  694. test_assert(r == -1);
  695. r= tor_addr_port_parse(LOG_DEBUG,
  696. "192.0.2.2",
  697. &addr, &port, 200);
  698. test_assert(r == 0);
  699. tt_int_op(port,==,200);
  700. r= tor_addr_port_parse(LOG_DEBUG,
  701. "[::1]",
  702. &addr, &port, -1);
  703. test_assert(r == -1);
  704. r= tor_addr_port_parse(LOG_DEBUG,
  705. "[::1]",
  706. &addr, &port, 400);
  707. test_assert(r == 0);
  708. tt_int_op(port,==,400);
  709. /* Bad port. */
  710. r= tor_addr_port_parse(LOG_DEBUG,
  711. "192.0.2.2:66666",
  712. &addr, &port, -1);
  713. test_assert(r == -1);
  714. r= tor_addr_port_parse(LOG_DEBUG,
  715. "192.0.2.2:66666",
  716. &addr, &port, 200);
  717. test_assert(r == -1);
  718. /* Only domain name */
  719. r= tor_addr_port_parse(LOG_DEBUG,
  720. "torproject.org",
  721. &addr, &port, -1);
  722. test_assert(r == -1);
  723. r= tor_addr_port_parse(LOG_DEBUG,
  724. "torproject.org",
  725. &addr, &port, 200);
  726. test_assert(r == -1);
  727. /* Bad IP address */
  728. r= tor_addr_port_parse(LOG_DEBUG,
  729. "192.0.2:1234",
  730. &addr, &port, -1);
  731. test_assert(r == -1);
  732. /* Make sure that the default port has lower priority than the real
  733. one */
  734. r= tor_addr_port_parse(LOG_DEBUG,
  735. "192.0.2.2:1337",
  736. &addr, &port, 200);
  737. test_assert(r == 0);
  738. tt_int_op(port,==,1337);
  739. r= tor_addr_port_parse(LOG_DEBUG,
  740. "[::1]:1369",
  741. &addr, &port, 200);
  742. test_assert(r == 0);
  743. tt_int_op(port,==,1369);
  744. done:
  745. ;
  746. }
  747. static void
  748. update_difference(int ipv6, uint8_t *d,
  749. const tor_addr_t *a, const tor_addr_t *b)
  750. {
  751. const int n_bytes = ipv6 ? 16 : 4;
  752. uint8_t a_tmp[4], b_tmp[4];
  753. const uint8_t *ba, *bb;
  754. int i;
  755. if (ipv6) {
  756. ba = tor_addr_to_in6_addr8(a);
  757. bb = tor_addr_to_in6_addr8(b);
  758. } else {
  759. set_uint32(a_tmp, tor_addr_to_ipv4n(a));
  760. set_uint32(b_tmp, tor_addr_to_ipv4n(b));
  761. ba = a_tmp; bb = b_tmp;
  762. }
  763. for (i = 0; i < n_bytes; ++i) {
  764. d[i] |= ba[i] ^ bb[i];
  765. }
  766. }
  767. static void
  768. test_virtaddrmap(void *data)
  769. {
  770. /* Let's start with a bunch of random addresses. */
  771. int ipv6, bits, iter, b;
  772. virtual_addr_conf_t cfg[2];
  773. uint8_t bytes[16];
  774. (void)data;
  775. tor_addr_parse(&cfg[0].addr, "64.65.0.0");
  776. tor_addr_parse(&cfg[1].addr, "3491:c0c0::");
  777. for (ipv6 = 0; ipv6 <= 1; ++ipv6) {
  778. for (bits = 0; bits < 18; ++bits) {
  779. tor_addr_t last_a;
  780. cfg[ipv6].bits = bits;
  781. memset(bytes, 0, sizeof(bytes));
  782. tor_addr_copy(&last_a, &cfg[ipv6].addr);
  783. /* Generate 128 addresses with each addr/bits combination. */
  784. for (iter = 0; iter < 128; ++iter) {
  785. tor_addr_t a;
  786. get_random_virtual_addr(&cfg[ipv6], &a);
  787. //printf("%s\n", fmt_addr(&a));
  788. /* Make sure that the first b bits match the configured network */
  789. tt_int_op(0, ==, tor_addr_compare_masked(&a, &cfg[ipv6].addr,
  790. bits, CMP_EXACT));
  791. /* And track which bits have been different between pairs of
  792. * addresses */
  793. update_difference(ipv6, bytes, &last_a, &a);
  794. }
  795. /* Now make sure all but the first 'bits' bits of bytes are true */
  796. for (b = bits+1; b < (ipv6?128:32); ++b) {
  797. tt_assert(1 & (bytes[b/8] >> (7-(b&7))));
  798. }
  799. }
  800. }
  801. done:
  802. ;
  803. }
  804. static void
  805. test_addr_localname(void *arg)
  806. {
  807. (void)arg;
  808. tt_assert(tor_addr_hostname_is_local("localhost"));
  809. tt_assert(tor_addr_hostname_is_local("LOCALHOST"));
  810. tt_assert(tor_addr_hostname_is_local("LocalHost"));
  811. tt_assert(tor_addr_hostname_is_local("local"));
  812. tt_assert(tor_addr_hostname_is_local("LOCAL"));
  813. tt_assert(tor_addr_hostname_is_local("here.now.local"));
  814. tt_assert(tor_addr_hostname_is_local("here.now.LOCAL"));
  815. tt_assert(!tor_addr_hostname_is_local(" localhost"));
  816. tt_assert(!tor_addr_hostname_is_local("www.torproject.org"));
  817. done:
  818. ;
  819. }
  820. static void
  821. test_addr_dup_ip(void *arg)
  822. {
  823. char *v = NULL;
  824. (void)arg;
  825. #define CHECK(ip, s) do { \
  826. v = tor_dup_ip(ip); \
  827. tt_str_op(v,==,(s)); \
  828. tor_free(v); \
  829. } while (0)
  830. CHECK(0xffffffff, "255.255.255.255");
  831. CHECK(0x00000000, "0.0.0.0");
  832. CHECK(0x7f000001, "127.0.0.1");
  833. CHECK(0x01020304, "1.2.3.4");
  834. #undef CHECK
  835. done:
  836. tor_free(v);
  837. }
  838. static void
  839. test_addr_sockaddr_to_str(void *arg)
  840. {
  841. char *v = NULL;
  842. struct sockaddr_in sin;
  843. struct sockaddr_in6 sin6;
  844. struct sockaddr_storage ss;
  845. #ifdef HAVE_SYS_UN_H
  846. struct sockaddr_un s_un;
  847. #endif
  848. #define CHECK(sa, s) do { \
  849. v = tor_sockaddr_to_str((const struct sockaddr*) &(sa)); \
  850. tt_str_op(v,==,(s)); \
  851. tor_free(v); \
  852. } while (0)
  853. (void)arg;
  854. memset(&ss,0,sizeof(ss));
  855. ss.ss_family = AF_UNSPEC;
  856. CHECK(ss, "unspec");
  857. memset(&sin,0,sizeof(sin));
  858. sin.sin_family = AF_INET;
  859. sin.sin_addr.s_addr = htonl(0x7f808001);
  860. sin.sin_port = htons(1234);
  861. CHECK(sin, "127.128.128.1:1234");
  862. #ifdef HAVE_SYS_UN_H
  863. memset(&s_un,0,sizeof(s_un));
  864. s_un.sun_family = AF_UNIX;
  865. strlcpy(s_un.sun_path, "/here/is/a/path", sizeof(s_un.sun_path));
  866. CHECK(s_un, "unix:/here/is/a/path");
  867. #endif
  868. memset(&sin6,0,sizeof(sin6));
  869. sin6.sin6_family = AF_INET6;
  870. memcpy(sin6.sin6_addr.s6_addr, "\x20\x00\x00\x00\x00\x00\x00\x00"
  871. "\x00\x1a\x2b\x3c\x4d\x5e\x00\x01", 16);
  872. sin6.sin6_port = htons(1234);
  873. CHECK(sin6, "[2000::1a:2b3c:4d5e:1]:1234");
  874. done:
  875. tor_free(v);
  876. }
  877. static void
  878. test_addr_is_loopback(void *data)
  879. {
  880. static const struct loopback_item {
  881. const char *name;
  882. int is_loopback;
  883. } loopback_items[] = {
  884. { "::1", 1 },
  885. { "127.0.0.1", 1 },
  886. { "127.99.100.101", 1 },
  887. { "128.99.100.101", 0 },
  888. { "8.8.8.8", 0 },
  889. { "0.0.0.0", 0 },
  890. { "::2", 0 },
  891. { "::", 0 },
  892. { "::1.0.0.0", 0 },
  893. { NULL, 0 }
  894. };
  895. int i;
  896. tor_addr_t addr;
  897. (void)data;
  898. for (i=0; loopback_items[i].name; ++i) {
  899. tt_int_op(tor_addr_parse(&addr, loopback_items[i].name), >=, 0);
  900. tt_int_op(tor_addr_is_loopback(&addr), ==, loopback_items[i].is_loopback);
  901. }
  902. tor_addr_make_unspec(&addr);
  903. tt_int_op(tor_addr_is_loopback(&addr), ==, 0);
  904. done:
  905. ;
  906. }
  907. static void
  908. test_addr_make_null(void *data)
  909. {
  910. tor_addr_t *addr = tor_malloc(sizeof(*addr));
  911. tor_addr_t *zeros = tor_malloc_zero(sizeof(*addr));
  912. char buf[TOR_ADDR_BUF_LEN];
  913. (void) data;
  914. /* Ensure that before tor_addr_make_null, addr != 0's */
  915. memset(addr, 1, sizeof(*addr));
  916. tt_int_op(memcmp(addr, zeros, sizeof(*addr)), !=, 0);
  917. /* Test with AF == AF_INET */
  918. zeros->family = AF_INET;
  919. tor_addr_make_null(addr, AF_INET);
  920. tt_int_op(memcmp(addr, zeros, sizeof(*addr)), ==, 0);
  921. tt_str_op(tor_addr_to_str(buf, addr, sizeof(buf), 0), ==, "0.0.0.0");
  922. /* Test with AF == AF_INET6 */
  923. memset(addr, 1, sizeof(*addr));
  924. zeros->family = AF_INET6;
  925. tor_addr_make_null(addr, AF_INET6);
  926. tt_int_op(memcmp(addr, zeros, sizeof(*addr)), ==, 0);
  927. tt_str_op(tor_addr_to_str(buf, addr, sizeof(buf), 0), ==, "::");
  928. done:
  929. tor_free(addr);
  930. tor_free(zeros);
  931. }
  932. #define ADDR_LEGACY(name) \
  933. { #name, legacy_test_helper, 0, &legacy_setup, test_addr_ ## name }
  934. struct testcase_t addr_tests[] = {
  935. ADDR_LEGACY(basic),
  936. ADDR_LEGACY(ip6_helpers),
  937. ADDR_LEGACY(parse),
  938. { "virtaddr", test_virtaddrmap, 0, NULL, NULL },
  939. { "localname", test_addr_localname, 0, NULL, NULL },
  940. { "dup_ip", test_addr_dup_ip, 0, NULL, NULL },
  941. { "sockaddr_to_str", test_addr_sockaddr_to_str, 0, NULL, NULL },
  942. { "is_loopback", test_addr_is_loopback, 0, NULL, NULL },
  943. { "make_null", test_addr_make_null, 0, NULL, NULL },
  944. END_OF_TESTCASES
  945. };