/* SPDX-License-Identifier: LGPL-2.1-or-later */ #include "in-addr-prefix-util.h" #include "tests.h" static void test_in_addr_prefix_to_string_one(int f, const char *addr, unsigned prefixlen) { union in_addr_union ua; assert_se(in_addr_from_string(f, addr, &ua) >= 0); const char *r = IN_ADDR_PREFIX_TO_STRING(f, &ua, prefixlen); assert_se(r); printf("%s: %s/%u == %s\n", __func__, addr, prefixlen, r); assert_se(startswith(r, addr)); ASSERT_STREQ(r, IN_ADDR_PREFIX_TO_STRING(f, &ua, prefixlen)); ASSERT_STREQ(IN_ADDR_PREFIX_TO_STRING(f, &ua, prefixlen), r); } TEST(in_addr_to_string_prefix) { test_in_addr_prefix_to_string_one(AF_INET, "192.168.0.1", 0); test_in_addr_prefix_to_string_one(AF_INET, "192.168.0.1", 1); test_in_addr_prefix_to_string_one(AF_INET, "192.168.0.1", 31); test_in_addr_prefix_to_string_one(AF_INET, "192.168.0.1", 32); test_in_addr_prefix_to_string_one(AF_INET, "192.168.0.1", 256); test_in_addr_prefix_to_string_one(AF_INET, "10.11.12.13", UINT_MAX); test_in_addr_prefix_to_string_one(AF_INET6, "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff", 0); test_in_addr_prefix_to_string_one(AF_INET6, "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff", UINT_MAX); test_in_addr_prefix_to_string_one(AF_INET6, "::1", 11); test_in_addr_prefix_to_string_one(AF_INET6, "fe80::", 33); } static void test_config_parse_in_addr_prefixes_one(int family, const union in_addr_union *addr, uint8_t prefixlen, Set **prefixes) { const char *str = IN_ADDR_PREFIX_TO_STRING(family, addr, prefixlen); assert_se(str); assert_se(config_parse_in_addr_prefixes("unit", "filename", 1, "Service", 1, "IPAddressAllow", 0, str, prefixes, NULL) >= 0); ASSERT_STREQ(str, IN_ADDR_PREFIX_TO_STRING(family, addr, prefixlen)); ASSERT_STREQ(IN_ADDR_PREFIX_TO_STRING(family, addr, prefixlen), str); } static void test_config_parse_in_addr_prefixes(Set **ret) { _cleanup_set_free_ Set *prefixes = NULL; log_info("/* %s() */", __func__); for (uint32_t i = 0; i < 256; i++) { /* ipv4 link-local address */ test_config_parse_in_addr_prefixes_one(AF_INET, &(union in_addr_union) { .in.s_addr = htobe32((UINT32_C(169) << 24) | (UINT32_C(254) << 16) | (i << 8)), }, 24, &prefixes); /* ipv6 multicast address */ test_config_parse_in_addr_prefixes_one(AF_INET6, &(union in_addr_union) { .in6.s6_addr[0] = 0xff, .in6.s6_addr[1] = i, }, 16, &prefixes); for (uint32_t j = 0; j < 256; j++) { test_config_parse_in_addr_prefixes_one(AF_INET, &(union in_addr_union) { .in.s_addr = htobe32((UINT32_C(169) << 24) | (UINT32_C(254) << 16) | (i << 8) | j), }, 32, &prefixes); test_config_parse_in_addr_prefixes_one(AF_INET6, &(union in_addr_union) { .in6.s6_addr[0] = 0xff, .in6.s6_addr[1] = i, .in6.s6_addr[2] = j, }, 24, &prefixes); } } *ret = TAKE_PTR(prefixes); } static void test_in_addr_prefixes_reduce(Set *prefixes) { log_info("/* %s() */", __func__); assert_se(set_size(prefixes) == 2 * 256 * 257); assert_se(!in_addr_prefixes_is_any(prefixes)); assert_se(in_addr_prefixes_reduce(prefixes) >= 0); assert_se(set_size(prefixes) == 2 * 256); assert_se(!in_addr_prefixes_is_any(prefixes)); assert_se(config_parse_in_addr_prefixes("unit", "filename", 1, "Service", 1, "IPAddressAllow", 0, "link-local", &prefixes, NULL) == 0); assert_se(set_size(prefixes) == 2 * 256 + 2); assert_se(!in_addr_prefixes_is_any(prefixes)); assert_se(in_addr_prefixes_reduce(prefixes) >= 0); assert_se(set_size(prefixes) == 256 + 2); assert_se(!in_addr_prefixes_is_any(prefixes)); assert_se(config_parse_in_addr_prefixes("unit", "filename", 1, "Service", 1, "IPAddressAllow", 0, "multicast", &prefixes, NULL) == 0); assert_se(set_size(prefixes) == 256 + 4); assert_se(!in_addr_prefixes_is_any(prefixes)); assert_se(in_addr_prefixes_reduce(prefixes) >= 0); assert_se(set_size(prefixes) == 4); assert_se(!in_addr_prefixes_is_any(prefixes)); assert_se(config_parse_in_addr_prefixes("unit", "filename", 1, "Service", 1, "IPAddressAllow", 0, "any", &prefixes, NULL) == 0); assert_se(set_size(prefixes) == 6); assert_se(in_addr_prefixes_is_any(prefixes)); assert_se(in_addr_prefixes_reduce(prefixes) >= 0); assert_se(set_size(prefixes) == 2); assert_se(in_addr_prefixes_is_any(prefixes)); } TEST(in_addr_prefixes) { _cleanup_set_free_ Set *prefixes = NULL; test_config_parse_in_addr_prefixes(&prefixes); test_in_addr_prefixes_reduce(prefixes); } DEFINE_TEST_MAIN(LOG_DEBUG);