diff --git a/tests/common.c b/tests/common.c index ff1531c..92f2d4c 100644 --- a/tests/common.c +++ b/tests/common.c @@ -1,5 +1,6 @@ #include #include +#include #include #include @@ -43,13 +44,73 @@ START_TEST(test_parse_format_ipv4_listen_all) } END_TEST +START_TEST(test_parse_format_ipv6) +{ + char *host = "2001:0db8:0505:0::123:0abc"; + char *compact = "2001:db8:505::123:abc"; + unsigned char v6_bits[] = { + 0x20, 0x01, 0x0d, 0xb8, 0x05, 0x05, 0x00, 0x00, + 0x00, 0x00, 0x00, 0x00, 0x01, 0x23, 0x0a, 0xbc, + }; + char *formatted; + struct sockaddr_storage addr; + struct sockaddr_in6 *v6addr; + int addr_len; + + addr_len = get_addr(host, 53, AF_UNSPEC, 0, &addr); + fail_unless(addr_len == sizeof(struct sockaddr_in6)); + + v6addr = (struct sockaddr_in6 *) &addr; + fail_if(memcmp(&v6addr->sin6_addr, v6_bits, sizeof(v6_bits))); + fail_unless(v6addr->sin6_port == htons(53)); + + formatted = format_addr(&addr, addr_len); + fail_if(strcmp(compact, formatted)); +} +END_TEST + +START_TEST(test_parse_format_ipv4_mapped_ipv6) +{ + char *v4mapped = "::FFFF:192.168.2.10"; + char *host = "192.168.2.10"; + unsigned char v6_bits[] = { + 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, + 0x00, 0x00, 0xff, 0xff, 0xc0, 0xa8, 0x02, 0x0a, + }; + char *formatted; + struct sockaddr_storage addr; + struct sockaddr_in6 *v6addr; + int addr_len; + + addr_len = get_addr(v4mapped, 53, AF_INET6, 0, &addr); + fail_unless(addr_len == sizeof(struct sockaddr_in6)); + + v6addr = (struct sockaddr_in6 *) &addr; + fail_if(memcmp(&v6addr->sin6_addr, v6_bits, sizeof(v6_bits))); + fail_unless(v6addr->sin6_port == htons(53)); + + /* Format as IPv4 address */ + formatted = format_addr(&addr, addr_len); + fail_if(strcmp(host, formatted)); +} +END_TEST + TCase * test_common_create_tests() { TCase *tc; + int sock; tc = tcase_create("Common"); tcase_add_test(tc, test_parse_format_ipv4); tcase_add_test(tc, test_parse_format_ipv4_listen_all); + + /* Tests require IPv6 support */ + sock = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP); + if (sock >= 0) { + close(sock); + tcase_add_test(tc, test_parse_format_ipv6); + tcase_add_test(tc, test_parse_format_ipv4_mapped_ipv6); + } return tc; }