summaryrefslogtreecommitdiffstats
path: root/src/test/test-in-addr-prefix-util.c
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--src/test/test-in-addr-prefix-util.c121
1 files changed, 121 insertions, 0 deletions
diff --git a/src/test/test-in-addr-prefix-util.c b/src/test/test-in-addr-prefix-util.c
new file mode 100644
index 0000000..661ca8f
--- /dev/null
+++ b/src/test/test-in-addr-prefix-util.c
@@ -0,0 +1,121 @@
+/* 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_se(streq(r, IN_ADDR_PREFIX_TO_STRING(f, &ua, prefixlen)));
+ assert_se(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_se(streq(str, IN_ADDR_PREFIX_TO_STRING(family, addr, prefixlen)));
+ assert_se(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);