test_addr.c 39 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061
  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 internal checking */
  313. test_external_ip("fbff:ffff::2:7", 0);
  314. test_internal_ip("fc01::2:7", 0);
  315. test_internal_ip("fc01::02:7", 0);
  316. test_internal_ip("fc01::002:7", 0);
  317. test_internal_ip("fc01::0002:7", 0);
  318. test_internal_ip("fdff:ffff::f:f", 0);
  319. test_external_ip("fe00::3:f", 0);
  320. test_external_ip("fe7f:ffff::2:7", 0);
  321. test_internal_ip("fe80::2:7", 0);
  322. test_internal_ip("febf:ffff::f:f", 0);
  323. test_internal_ip("fec0::2:7:7", 0);
  324. test_internal_ip("feff:ffff::e:7:7", 0);
  325. test_external_ip("ff00::e:7:7", 0);
  326. test_internal_ip("::", 0);
  327. test_internal_ip("::1", 0);
  328. test_internal_ip("::1", 1);
  329. test_internal_ip("::", 0);
  330. test_external_ip("::", 1);
  331. test_external_ip("::2", 0);
  332. test_external_ip("2001::", 0);
  333. test_external_ip("ffff::", 0);
  334. test_external_ip("::ffff:0.0.0.0", 1);
  335. test_internal_ip("::ffff:0.0.0.0", 0);
  336. test_internal_ip("::ffff:0.255.255.255", 0);
  337. test_external_ip("::ffff:1.0.0.0", 0);
  338. test_external_ip("::ffff:9.255.255.255", 0);
  339. test_internal_ip("::ffff:10.0.0.0", 0);
  340. test_internal_ip("::ffff:10.255.255.255", 0);
  341. test_external_ip("::ffff:11.0.0.0", 0);
  342. test_external_ip("::ffff:126.255.255.255", 0);
  343. test_internal_ip("::ffff:127.0.0.0", 0);
  344. test_internal_ip("::ffff:127.255.255.255", 0);
  345. test_external_ip("::ffff:128.0.0.0", 0);
  346. test_external_ip("::ffff:172.15.255.255", 0);
  347. test_internal_ip("::ffff:172.16.0.0", 0);
  348. test_internal_ip("::ffff:172.31.255.255", 0);
  349. test_external_ip("::ffff:172.32.0.0", 0);
  350. test_external_ip("::ffff:192.167.255.255", 0);
  351. test_internal_ip("::ffff:192.168.0.0", 0);
  352. test_internal_ip("::ffff:192.168.255.255", 0);
  353. test_external_ip("::ffff:192.169.0.0", 0);
  354. test_external_ip("::ffff:169.253.255.255", 0);
  355. test_internal_ip("::ffff:169.254.0.0", 0);
  356. test_internal_ip("::ffff:169.254.255.255", 0);
  357. test_external_ip("::ffff:169.255.0.0", 0);
  358. /* tor_addr_compare(tor_addr_t x2) */
  359. test_addr_compare("ffff::", ==, "ffff::0");
  360. test_addr_compare("0::3:2:1", <, "0::ffff:0.3.2.1");
  361. test_addr_compare("0::2:2:1", <, "0::ffff:0.3.2.1");
  362. test_addr_compare("0::ffff:0.3.2.1", >, "0::0:0:0");
  363. test_addr_compare("0::ffff:5.2.2.1", <, "::ffff:6.0.0.0"); /* XXXX wrong. */
  364. tor_addr_parse_mask_ports("[::ffff:2.3.4.5]", 0, &t1, NULL, NULL, NULL);
  365. tor_addr_parse_mask_ports("2.3.4.5", 0, &t2, NULL, NULL, NULL);
  366. test_assert(tor_addr_compare(&t1, &t2, CMP_SEMANTIC) == 0);
  367. tor_addr_parse_mask_ports("[::ffff:2.3.4.4]", 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. /* test compare_masked */
  371. test_addr_compare_masked("ffff::", ==, "ffff::0", 128);
  372. test_addr_compare_masked("ffff::", ==, "ffff::0", 64);
  373. test_addr_compare_masked("0::2:2:1", <, "0::8000:2:1", 81);
  374. test_addr_compare_masked("0::2:2:1", ==, "0::8000:2:1", 80);
  375. /* Test undecorated tor_addr_to_str */
  376. test_eq(AF_INET6, tor_addr_parse(&t1, "[123:45:6789::5005:11]"));
  377. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 0);
  378. test_streq(p1, "123:45:6789::5005:11");
  379. test_eq(AF_INET, tor_addr_parse(&t1, "18.0.0.1"));
  380. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 0);
  381. test_streq(p1, "18.0.0.1");
  382. /* Test decorated tor_addr_to_str */
  383. test_eq(AF_INET6, tor_addr_parse(&t1, "[123:45:6789::5005:11]"));
  384. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  385. test_streq(p1, "[123:45:6789::5005:11]");
  386. test_eq(AF_INET, tor_addr_parse(&t1, "18.0.0.1"));
  387. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  388. test_streq(p1, "18.0.0.1");
  389. /* Test buffer bounds checking of tor_addr_to_str */
  390. test_eq(AF_INET6, tor_addr_parse(&t1, "::")); /* 2 + \0 */
  391. test_eq_ptr(tor_addr_to_str(buf, &t1, 2, 0), NULL); /* too short buf */
  392. test_streq(tor_addr_to_str(buf, &t1, 3, 0), "::");
  393. test_eq_ptr(tor_addr_to_str(buf, &t1, 4, 1), NULL); /* too short buf */
  394. test_streq(tor_addr_to_str(buf, &t1, 5, 1), "[::]");
  395. test_eq(AF_INET6, tor_addr_parse(&t1, "2000::1337")); /* 10 + \0 */
  396. test_eq_ptr(tor_addr_to_str(buf, &t1, 10, 0), NULL); /* too short buf */
  397. test_streq(tor_addr_to_str(buf, &t1, 11, 0), "2000::1337");
  398. test_eq_ptr(tor_addr_to_str(buf, &t1, 12, 1), NULL); /* too short buf */
  399. test_streq(tor_addr_to_str(buf, &t1, 13, 1), "[2000::1337]");
  400. test_eq(AF_INET, tor_addr_parse(&t1, "1.2.3.4")); /* 7 + \0 */
  401. test_eq_ptr(tor_addr_to_str(buf, &t1, 7, 0), NULL); /* too short buf */
  402. test_streq(tor_addr_to_str(buf, &t1, 8, 0), "1.2.3.4");
  403. test_eq(AF_INET, tor_addr_parse(&t1, "255.255.255.255")); /* 15 + \0 */
  404. test_eq_ptr(tor_addr_to_str(buf, &t1, 15, 0), NULL); /* too short buf */
  405. test_streq(tor_addr_to_str(buf, &t1, 16, 0), "255.255.255.255");
  406. test_eq_ptr(tor_addr_to_str(buf, &t1, 15, 1), NULL); /* too short buf */
  407. test_streq(tor_addr_to_str(buf, &t1, 16, 1), "255.255.255.255");
  408. t1.family = AF_UNSPEC;
  409. test_eq_ptr(tor_addr_to_str(buf, &t1, sizeof(buf), 0), NULL);
  410. /* Test tor_addr_parse_PTR_name */
  411. i = tor_addr_parse_PTR_name(&t1, "Foobar.baz", AF_UNSPEC, 0);
  412. test_eq(0, i);
  413. i = tor_addr_parse_PTR_name(&t1, "Foobar.baz", AF_UNSPEC, 1);
  414. test_eq(0, i);
  415. i = tor_addr_parse_PTR_name(&t1, "9999999999999999999999999999.in-addr.arpa",
  416. AF_UNSPEC, 1);
  417. test_eq(-1, i);
  418. i = tor_addr_parse_PTR_name(&t1, "1.0.168.192.in-addr.arpa",
  419. AF_UNSPEC, 1);
  420. test_eq(1, i);
  421. test_eq(tor_addr_family(&t1), AF_INET);
  422. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  423. test_streq(p1, "192.168.0.1");
  424. i = tor_addr_parse_PTR_name(&t1, "192.168.0.99", AF_UNSPEC, 0);
  425. test_eq(0, i);
  426. i = tor_addr_parse_PTR_name(&t1, "192.168.0.99", AF_UNSPEC, 1);
  427. test_eq(1, i);
  428. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  429. test_streq(p1, "192.168.0.99");
  430. memset(&t1, 0, sizeof(t1));
  431. i = tor_addr_parse_PTR_name(&t1,
  432. "0.1.2.3.4.5.6.7.8.9.a.b.c.d.e.f."
  433. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  434. "ip6.ARPA",
  435. AF_UNSPEC, 0);
  436. test_eq(1, i);
  437. p1 = tor_addr_to_str(buf, &t1, sizeof(buf), 1);
  438. test_streq(p1, "[9dee:effe:ebe1:beef:fedc:ba98:7654:3210]");
  439. /* Failing cases. */
  440. i = tor_addr_parse_PTR_name(&t1,
  441. "6.7.8.9.a.b.c.d.e.f."
  442. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  443. "ip6.ARPA",
  444. AF_UNSPEC, 0);
  445. test_eq(i, -1);
  446. i = tor_addr_parse_PTR_name(&t1,
  447. "6.7.8.9.a.b.c.d.e.f.a.b.c.d.e.f.0."
  448. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  449. "ip6.ARPA",
  450. AF_UNSPEC, 0);
  451. test_eq(i, -1);
  452. i = tor_addr_parse_PTR_name(&t1,
  453. "6.7.8.9.a.b.c.d.e.f.X.0.0.0.0.9."
  454. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  455. "ip6.ARPA",
  456. AF_UNSPEC, 0);
  457. test_eq(i, -1);
  458. i = tor_addr_parse_PTR_name(&t1, "32.1.1.in-addr.arpa",
  459. AF_UNSPEC, 0);
  460. test_eq(i, -1);
  461. i = tor_addr_parse_PTR_name(&t1, ".in-addr.arpa",
  462. AF_UNSPEC, 0);
  463. test_eq(i, -1);
  464. i = tor_addr_parse_PTR_name(&t1, "1.2.3.4.5.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_INET6, 0);
  469. test_eq(i, -1);
  470. i = tor_addr_parse_PTR_name(&t1,
  471. "6.7.8.9.a.b.c.d.e.f.a.b.c.d.e.0."
  472. "f.e.e.b.1.e.b.e.e.f.f.e.e.e.d.9."
  473. "ip6.ARPA",
  474. AF_INET, 0);
  475. test_eq(i, -1);
  476. /* === Test tor_addr_to_PTR_name */
  477. /* Stage IPv4 addr */
  478. memset(&sa_storage, 0, sizeof(sa_storage));
  479. sin = (struct sockaddr_in *)&sa_storage;
  480. sin->sin_family = AF_INET;
  481. sin->sin_addr.s_addr = htonl(0x7f010203); /* 127.1.2.3 */
  482. tor_addr_from_sockaddr(&t1, (struct sockaddr *)sin, NULL);
  483. /* Check IPv4 PTR - too short buffer */
  484. test_eq(tor_addr_to_PTR_name(rbuf, 1, &t1), -1);
  485. test_eq(tor_addr_to_PTR_name(rbuf,
  486. strlen("3.2.1.127.in-addr.arpa") - 1,
  487. &t1), -1);
  488. /* Check IPv4 PTR - valid addr */
  489. test_eq(tor_addr_to_PTR_name(rbuf, sizeof(rbuf), &t1),
  490. strlen("3.2.1.127.in-addr.arpa"));
  491. test_streq(rbuf, "3.2.1.127.in-addr.arpa");
  492. /* Invalid addr family */
  493. t1.family = AF_UNSPEC;
  494. test_eq(tor_addr_to_PTR_name(rbuf, sizeof(rbuf), &t1), -1);
  495. /* Stage IPv6 addr */
  496. memset(&sa_storage, 0, sizeof(sa_storage));
  497. sin6 = (struct sockaddr_in6 *)&sa_storage;
  498. sin6->sin6_family = AF_INET6;
  499. sin6->sin6_addr.s6_addr[0] = 0x80; /* 8000::abcd */
  500. sin6->sin6_addr.s6_addr[14] = 0xab;
  501. sin6->sin6_addr.s6_addr[15] = 0xcd;
  502. tor_addr_from_sockaddr(&t1, (struct sockaddr *)sin6, NULL);
  503. {
  504. const char* addr_PTR = "d.c.b.a.0.0.0.0.0.0.0.0.0.0.0.0."
  505. "0.0.0.0.0.0.0.0.0.0.0.0.0.0.0.8.ip6.arpa";
  506. /* Check IPv6 PTR - too short buffer */
  507. test_eq(tor_addr_to_PTR_name(rbuf, 0, &t1), -1);
  508. test_eq(tor_addr_to_PTR_name(rbuf, strlen(addr_PTR) - 1, &t1), -1);
  509. /* Check IPv6 PTR - valid addr */
  510. test_eq(tor_addr_to_PTR_name(rbuf, sizeof(rbuf), &t1),
  511. strlen(addr_PTR));
  512. test_streq(rbuf, addr_PTR);
  513. }
  514. /* XXXX turn this into a separate function; it's not all IPv6. */
  515. /* test tor_addr_parse_mask_ports */
  516. test_addr_mask_ports_parse("[::f]/17:47-95", AF_INET6,
  517. 0, 0, 0, 0x0000000f, 17, 47, 95);
  518. test_streq(p1, "::f");
  519. //test_addr_parse("[::fefe:4.1.1.7/120]:999-1000");
  520. //test_addr_parse_check("::fefe:401:107", 120, 999, 1000);
  521. test_addr_mask_ports_parse("[::ffff:4.1.1.7]/120:443", AF_INET6,
  522. 0, 0, 0x0000ffff, 0x04010107, 120, 443, 443);
  523. test_streq(p1, "::ffff:4.1.1.7");
  524. test_addr_mask_ports_parse("[abcd:2::44a:0]:2-65000", AF_INET6,
  525. 0xabcd0002, 0, 0, 0x044a0000, 128, 2, 65000);
  526. test_streq(p1, "abcd:2::44a:0");
  527. /* Try some long addresses. */
  528. r=tor_addr_parse_mask_ports("[ffff:1111:1111:1111:1111:1111:1111:1111]",
  529. 0, &t1, NULL, NULL, NULL);
  530. test_assert(r == AF_INET6);
  531. r=tor_addr_parse_mask_ports("[ffff:1111:1111:1111:1111:1111:1111:11111]",
  532. 0, &t1, NULL, NULL, NULL);
  533. test_assert(r == -1);
  534. r=tor_addr_parse_mask_ports("[ffff:1111:1111:1111:1111:1111:1111:1111:1]",
  535. 0, &t1, NULL, NULL, NULL);
  536. test_assert(r == -1);
  537. r=tor_addr_parse_mask_ports(
  538. "[ffff:1111:1111:1111:1111:1111:1111:ffff:"
  539. "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:"
  540. "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:"
  541. "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff]",
  542. 0, &t1, NULL, NULL, NULL);
  543. test_assert(r == -1);
  544. /* Try some failing cases. */
  545. r=tor_addr_parse_mask_ports("[fefef::]/112", 0, &t1, NULL, NULL, NULL);
  546. test_assert(r == -1);
  547. r=tor_addr_parse_mask_ports("[fefe::/112", 0, &t1, NULL, NULL, NULL);
  548. test_assert(r == -1);
  549. r=tor_addr_parse_mask_ports("[fefe::", 0, &t1, NULL, NULL, NULL);
  550. test_assert(r == -1);
  551. r=tor_addr_parse_mask_ports("[fefe::X]", 0, &t1, NULL, NULL, NULL);
  552. test_assert(r == -1);
  553. r=tor_addr_parse_mask_ports("efef::/112", 0, &t1, NULL, NULL, NULL);
  554. test_assert(r == -1);
  555. r=tor_addr_parse_mask_ports("[f:f:f:f:f:f:f:f::]",0,&t1, NULL, NULL, NULL);
  556. test_assert(r == -1);
  557. r=tor_addr_parse_mask_ports("[::f:f:f:f:f:f:f:f]",0,&t1, NULL, NULL, NULL);
  558. test_assert(r == -1);
  559. r=tor_addr_parse_mask_ports("[f:f:f:f:f:f:f:f:f]",0,&t1, NULL, NULL, NULL);
  560. test_assert(r == -1);
  561. r=tor_addr_parse_mask_ports("[f:f:f:f:f::]/fred",0,&t1,&mask, NULL, NULL);
  562. test_assert(r == -1);
  563. r=tor_addr_parse_mask_ports("[f:f:f:f:f::]/255.255.0.0",
  564. 0,&t1, NULL, NULL, NULL);
  565. test_assert(r == -1);
  566. /* This one will get rejected because it isn't a pure prefix. */
  567. r=tor_addr_parse_mask_ports("1.1.2.3/255.255.64.0",0,&t1, &mask,NULL,NULL);
  568. test_assert(r == -1);
  569. /* Test for V4-mapped address with mask < 96. (arguably not valid) */
  570. r=tor_addr_parse_mask_ports("[::ffff:1.1.2.2/33]",0,&t1, &mask, NULL, NULL);
  571. test_assert(r == -1);
  572. r=tor_addr_parse_mask_ports("1.1.2.2/33",0,&t1, &mask, NULL, NULL);
  573. test_assert(r == -1);
  574. /* Try extended wildcard addresses with out TAPMP_EXTENDED_STAR*/
  575. r=tor_addr_parse_mask_ports("*4",0,&t1, &mask, NULL, NULL);
  576. test_assert(r == -1);
  577. r=tor_addr_parse_mask_ports("*6",0,&t1, &mask, NULL, NULL);
  578. test_assert(r == -1);
  579. #if 0
  580. /* Try a mask with a wildcard. */
  581. r=tor_addr_parse_mask_ports("*/16",0,&t1, &mask, NULL, NULL);
  582. test_assert(r == -1);
  583. r=tor_addr_parse_mask_ports("*4/16",TAPMP_EXTENDED_STAR,
  584. &t1, &mask, NULL, NULL);
  585. test_assert(r == -1);
  586. r=tor_addr_parse_mask_ports("*6/30",TAPMP_EXTENDED_STAR,
  587. &t1, &mask, NULL, NULL);
  588. test_assert(r == -1);
  589. #endif
  590. /* Basic mask tests*/
  591. r=tor_addr_parse_mask_ports("1.1.2.2/31",0,&t1, &mask, NULL, NULL);
  592. test_assert(r == AF_INET);
  593. tt_int_op(mask,==,31);
  594. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  595. tt_int_op(tor_addr_to_ipv4h(&t1),==,0x01010202);
  596. r=tor_addr_parse_mask_ports("3.4.16.032:1-2",0,&t1, &mask, &port1, &port2);
  597. test_assert(r == AF_INET);
  598. tt_int_op(mask,==,32);
  599. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  600. tt_int_op(tor_addr_to_ipv4h(&t1),==,0x03041020);
  601. test_assert(port1 == 1);
  602. test_assert(port2 == 2);
  603. r=tor_addr_parse_mask_ports("1.1.2.3/255.255.128.0",0,&t1, &mask,NULL,NULL);
  604. test_assert(r == AF_INET);
  605. tt_int_op(mask,==,17);
  606. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  607. tt_int_op(tor_addr_to_ipv4h(&t1),==,0x01010203);
  608. r=tor_addr_parse_mask_ports("[efef::]/112",0,&t1, &mask, &port1, &port2);
  609. test_assert(r == AF_INET6);
  610. test_assert(port1 == 1);
  611. test_assert(port2 == 65535);
  612. /* Try regular wildcard behavior without TAPMP_EXTENDED_STAR */
  613. r=tor_addr_parse_mask_ports("*:80-443",0,&t1,&mask,&port1,&port2);
  614. tt_int_op(r,==,AF_INET); /* Old users of this always get inet */
  615. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  616. tt_int_op(tor_addr_to_ipv4h(&t1),==,0);
  617. tt_int_op(mask,==,0);
  618. tt_int_op(port1,==,80);
  619. tt_int_op(port2,==,443);
  620. /* Now try wildcards *with* TAPMP_EXTENDED_STAR */
  621. r=tor_addr_parse_mask_ports("*:8000-9000",TAPMP_EXTENDED_STAR,
  622. &t1,&mask,&port1,&port2);
  623. tt_int_op(r,==,AF_UNSPEC);
  624. tt_int_op(tor_addr_family(&t1),==,AF_UNSPEC);
  625. tt_int_op(mask,==,0);
  626. tt_int_op(port1,==,8000);
  627. tt_int_op(port2,==,9000);
  628. r=tor_addr_parse_mask_ports("*4:6667",TAPMP_EXTENDED_STAR,
  629. &t1,&mask,&port1,&port2);
  630. tt_int_op(r,==,AF_INET);
  631. tt_int_op(tor_addr_family(&t1),==,AF_INET);
  632. tt_int_op(tor_addr_to_ipv4h(&t1),==,0);
  633. tt_int_op(mask,==,0);
  634. tt_int_op(port1,==,6667);
  635. tt_int_op(port2,==,6667);
  636. r=tor_addr_parse_mask_ports("*6",TAPMP_EXTENDED_STAR,
  637. &t1,&mask,&port1,&port2);
  638. tt_int_op(r,==,AF_INET6);
  639. tt_int_op(tor_addr_family(&t1),==,AF_INET6);
  640. tt_assert(tor_mem_is_zero((const char*)tor_addr_to_in6_addr32(&t1), 16));
  641. tt_int_op(mask,==,0);
  642. tt_int_op(port1,==,1);
  643. tt_int_op(port2,==,65535);
  644. /* make sure inet address lengths >= max */
  645. test_assert(INET_NTOA_BUF_LEN >= sizeof("255.255.255.255"));
  646. test_assert(TOR_ADDR_BUF_LEN >=
  647. sizeof("ffff:ffff:ffff:ffff:ffff:ffff:255.255.255.255"));
  648. test_assert(sizeof(tor_addr_t) >= sizeof(struct in6_addr));
  649. /* get interface addresses */
  650. r = get_interface_address6(LOG_DEBUG, AF_INET, &t1);
  651. i = get_interface_address6(LOG_DEBUG, AF_INET6, &t2);
  652. TT_BLATHER(("v4 address: %s (family=%d)", fmt_addr(&t1),
  653. tor_addr_family(&t1)));
  654. TT_BLATHER(("v6 address: %s (family=%d)", fmt_addr(&t2),
  655. tor_addr_family(&t2)));
  656. done:
  657. ;
  658. }
  659. /** Test tor_addr_port_parse(). */
  660. static void
  661. test_addr_parse(void)
  662. {
  663. int r;
  664. tor_addr_t addr;
  665. char buf[TOR_ADDR_BUF_LEN];
  666. uint16_t port = 0;
  667. /* Correct call. */
  668. r= tor_addr_port_parse(LOG_DEBUG,
  669. "192.0.2.1:1234",
  670. &addr, &port, -1);
  671. test_assert(r == 0);
  672. tor_addr_to_str(buf, &addr, sizeof(buf), 0);
  673. test_streq(buf, "192.0.2.1");
  674. test_eq(port, 1234);
  675. r= tor_addr_port_parse(LOG_DEBUG,
  676. "[::1]:1234",
  677. &addr, &port, -1);
  678. test_assert(r == 0);
  679. tor_addr_to_str(buf, &addr, sizeof(buf), 0);
  680. test_streq(buf, "::1");
  681. test_eq(port, 1234);
  682. /* Domain name. */
  683. r= tor_addr_port_parse(LOG_DEBUG,
  684. "torproject.org:1234",
  685. &addr, &port, -1);
  686. test_assert(r == -1);
  687. /* Only IP. */
  688. r= tor_addr_port_parse(LOG_DEBUG,
  689. "192.0.2.2",
  690. &addr, &port, -1);
  691. test_assert(r == -1);
  692. r= tor_addr_port_parse(LOG_DEBUG,
  693. "192.0.2.2",
  694. &addr, &port, 200);
  695. test_assert(r == 0);
  696. tt_int_op(port,==,200);
  697. r= tor_addr_port_parse(LOG_DEBUG,
  698. "[::1]",
  699. &addr, &port, -1);
  700. test_assert(r == -1);
  701. r= tor_addr_port_parse(LOG_DEBUG,
  702. "[::1]",
  703. &addr, &port, 400);
  704. test_assert(r == 0);
  705. tt_int_op(port,==,400);
  706. /* Bad port. */
  707. r= tor_addr_port_parse(LOG_DEBUG,
  708. "192.0.2.2:66666",
  709. &addr, &port, -1);
  710. test_assert(r == -1);
  711. r= tor_addr_port_parse(LOG_DEBUG,
  712. "192.0.2.2:66666",
  713. &addr, &port, 200);
  714. test_assert(r == -1);
  715. /* Only domain name */
  716. r= tor_addr_port_parse(LOG_DEBUG,
  717. "torproject.org",
  718. &addr, &port, -1);
  719. test_assert(r == -1);
  720. r= tor_addr_port_parse(LOG_DEBUG,
  721. "torproject.org",
  722. &addr, &port, 200);
  723. test_assert(r == -1);
  724. /* Bad IP address */
  725. r= tor_addr_port_parse(LOG_DEBUG,
  726. "192.0.2:1234",
  727. &addr, &port, -1);
  728. test_assert(r == -1);
  729. /* Make sure that the default port has lower priority than the real
  730. one */
  731. r= tor_addr_port_parse(LOG_DEBUG,
  732. "192.0.2.2:1337",
  733. &addr, &port, 200);
  734. test_assert(r == 0);
  735. tt_int_op(port,==,1337);
  736. r= tor_addr_port_parse(LOG_DEBUG,
  737. "[::1]:1369",
  738. &addr, &port, 200);
  739. test_assert(r == 0);
  740. tt_int_op(port,==,1369);
  741. done:
  742. ;
  743. }
  744. static void
  745. update_difference(int ipv6, uint8_t *d,
  746. const tor_addr_t *a, const tor_addr_t *b)
  747. {
  748. const int n_bytes = ipv6 ? 16 : 4;
  749. uint8_t a_tmp[4], b_tmp[4];
  750. const uint8_t *ba, *bb;
  751. int i;
  752. if (ipv6) {
  753. ba = tor_addr_to_in6_addr8(a);
  754. bb = tor_addr_to_in6_addr8(b);
  755. } else {
  756. set_uint32(a_tmp, tor_addr_to_ipv4n(a));
  757. set_uint32(b_tmp, tor_addr_to_ipv4n(b));
  758. ba = a_tmp; bb = b_tmp;
  759. }
  760. for (i = 0; i < n_bytes; ++i) {
  761. d[i] |= ba[i] ^ bb[i];
  762. }
  763. }
  764. static void
  765. test_virtaddrmap(void *data)
  766. {
  767. /* Let's start with a bunch of random addresses. */
  768. int ipv6, bits, iter, b;
  769. virtual_addr_conf_t cfg[2];
  770. uint8_t bytes[16];
  771. (void)data;
  772. tor_addr_parse(&cfg[0].addr, "64.65.0.0");
  773. tor_addr_parse(&cfg[1].addr, "3491:c0c0::");
  774. for (ipv6 = 0; ipv6 <= 1; ++ipv6) {
  775. for (bits = 0; bits < 18; ++bits) {
  776. tor_addr_t last_a;
  777. cfg[ipv6].bits = bits;
  778. memset(bytes, 0, sizeof(bytes));
  779. tor_addr_copy(&last_a, &cfg[ipv6].addr);
  780. /* Generate 128 addresses with each addr/bits combination. */
  781. for (iter = 0; iter < 128; ++iter) {
  782. tor_addr_t a;
  783. get_random_virtual_addr(&cfg[ipv6], &a);
  784. //printf("%s\n", fmt_addr(&a));
  785. /* Make sure that the first b bits match the configured network */
  786. tt_int_op(0, ==, tor_addr_compare_masked(&a, &cfg[ipv6].addr,
  787. bits, CMP_EXACT));
  788. /* And track which bits have been different between pairs of
  789. * addresses */
  790. update_difference(ipv6, bytes, &last_a, &a);
  791. }
  792. /* Now make sure all but the first 'bits' bits of bytes are true */
  793. for (b = bits+1; b < (ipv6?128:32); ++b) {
  794. tt_assert(1 & (bytes[b/8] >> (7-(b&7))));
  795. }
  796. }
  797. }
  798. done:
  799. ;
  800. }
  801. static void
  802. test_addr_localname(void *arg)
  803. {
  804. (void)arg;
  805. tt_assert(tor_addr_hostname_is_local("localhost"));
  806. tt_assert(tor_addr_hostname_is_local("LOCALHOST"));
  807. tt_assert(tor_addr_hostname_is_local("LocalHost"));
  808. tt_assert(tor_addr_hostname_is_local("local"));
  809. tt_assert(tor_addr_hostname_is_local("LOCAL"));
  810. tt_assert(tor_addr_hostname_is_local("here.now.local"));
  811. tt_assert(tor_addr_hostname_is_local("here.now.LOCAL"));
  812. tt_assert(!tor_addr_hostname_is_local(" localhost"));
  813. tt_assert(!tor_addr_hostname_is_local("www.torproject.org"));
  814. done:
  815. ;
  816. }
  817. static void
  818. test_addr_dup_ip(void *arg)
  819. {
  820. char *v = NULL;
  821. (void)arg;
  822. #define CHECK(ip, s) do { \
  823. v = tor_dup_ip(ip); \
  824. tt_str_op(v,==,(s)); \
  825. tor_free(v); \
  826. } while (0)
  827. CHECK(0xffffffff, "255.255.255.255");
  828. CHECK(0x00000000, "0.0.0.0");
  829. CHECK(0x7f000001, "127.0.0.1");
  830. CHECK(0x01020304, "1.2.3.4");
  831. #undef CHECK
  832. done:
  833. tor_free(v);
  834. }
  835. static void
  836. test_addr_sockaddr_to_str(void *arg)
  837. {
  838. char *v = NULL;
  839. struct sockaddr_in sin;
  840. struct sockaddr_in6 sin6;
  841. struct sockaddr_storage ss;
  842. #ifdef HAVE_SYS_UN_H
  843. struct sockaddr_un s_un;
  844. #endif
  845. #define CHECK(sa, s) do { \
  846. v = tor_sockaddr_to_str((const struct sockaddr*) &(sa)); \
  847. tt_str_op(v,==,(s)); \
  848. tor_free(v); \
  849. } while (0)
  850. (void)arg;
  851. memset(&ss,0,sizeof(ss));
  852. ss.ss_family = AF_UNSPEC;
  853. CHECK(ss, "unspec");
  854. memset(&sin,0,sizeof(sin));
  855. sin.sin_family = AF_INET;
  856. sin.sin_addr.s_addr = htonl(0x7f808001);
  857. sin.sin_port = htons(1234);
  858. CHECK(sin, "127.128.128.1:1234");
  859. #ifdef HAVE_SYS_UN_H
  860. memset(&s_un,0,sizeof(s_un));
  861. s_un.sun_family = AF_UNIX;
  862. strlcpy(s_un.sun_path, "/here/is/a/path", sizeof(s_un.sun_path));
  863. CHECK(s_un, "unix:/here/is/a/path");
  864. #endif
  865. memset(&sin6,0,sizeof(sin6));
  866. sin6.sin6_family = AF_INET6;
  867. memcpy(sin6.sin6_addr.s6_addr, "\x20\x00\x00\x00\x00\x00\x00\x00"
  868. "\x00\x1a\x2b\x3c\x4d\x5e\x00\x01", 16);
  869. sin6.sin6_port = htons(1234);
  870. CHECK(sin6, "[2000::1a:2b3c:4d5e:1]:1234");
  871. done:
  872. tor_free(v);
  873. }
  874. static void
  875. test_addr_is_loopback(void *data)
  876. {
  877. static const struct loopback_item {
  878. const char *name;
  879. int is_loopback;
  880. } loopback_items[] = {
  881. { "::1", 1 },
  882. { "127.0.0.1", 1 },
  883. { "127.99.100.101", 1 },
  884. { "128.99.100.101", 0 },
  885. { "8.8.8.8", 0 },
  886. { "0.0.0.0", 0 },
  887. { "::2", 0 },
  888. { "::", 0 },
  889. { "::1.0.0.0", 0 },
  890. { NULL, 0 }
  891. };
  892. int i;
  893. tor_addr_t addr;
  894. (void)data;
  895. for (i=0; loopback_items[i].name; ++i) {
  896. tt_int_op(tor_addr_parse(&addr, loopback_items[i].name), >=, 0);
  897. tt_int_op(tor_addr_is_loopback(&addr), ==, loopback_items[i].is_loopback);
  898. }
  899. tor_addr_make_unspec(&addr);
  900. tt_int_op(tor_addr_is_loopback(&addr), ==, 0);
  901. done:
  902. ;
  903. }
  904. static void
  905. test_addr_make_null(void *data)
  906. {
  907. tor_addr_t *addr = tor_malloc(sizeof(*addr));
  908. tor_addr_t *zeros = tor_malloc_zero(sizeof(*addr));
  909. char buf[TOR_ADDR_BUF_LEN];
  910. (void) data;
  911. /* Ensure that before tor_addr_make_null, addr != 0's */
  912. memset(addr, 1, sizeof(*addr));
  913. tt_int_op(memcmp(addr, zeros, sizeof(*addr)), !=, 0);
  914. /* Test with AF == AF_INET */
  915. zeros->family = AF_INET;
  916. tor_addr_make_null(addr, AF_INET);
  917. tt_int_op(memcmp(addr, zeros, sizeof(*addr)), ==, 0);
  918. tt_str_op(tor_addr_to_str(buf, addr, sizeof(buf), 0), ==, "0.0.0.0");
  919. /* Test with AF == AF_INET6 */
  920. memset(addr, 1, sizeof(*addr));
  921. zeros->family = AF_INET6;
  922. tor_addr_make_null(addr, AF_INET6);
  923. tt_int_op(memcmp(addr, zeros, sizeof(*addr)), ==, 0);
  924. tt_str_op(tor_addr_to_str(buf, addr, sizeof(buf), 0), ==, "::");
  925. done:
  926. tor_free(addr);
  927. tor_free(zeros);
  928. }
  929. #define ADDR_LEGACY(name) \
  930. { #name, legacy_test_helper, 0, &legacy_setup, test_addr_ ## name }
  931. struct testcase_t addr_tests[] = {
  932. ADDR_LEGACY(basic),
  933. ADDR_LEGACY(ip6_helpers),
  934. ADDR_LEGACY(parse),
  935. { "virtaddr", test_virtaddrmap, 0, NULL, NULL },
  936. { "localname", test_addr_localname, 0, NULL, NULL },
  937. { "dup_ip", test_addr_dup_ip, 0, NULL, NULL },
  938. { "sockaddr_to_str", test_addr_sockaddr_to_str, 0, NULL, NULL },
  939. { "is_loopback", test_addr_is_loopback, 0, NULL, NULL },
  940. { "make_null", test_addr_make_null, 0, NULL, NULL },
  941. END_OF_TESTCASES
  942. };