1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
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_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);
|
