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
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
|
// SPDX-License-Identifier: GPL-2.0-or-later
/*
* Copyright (c) 2017-2019 Petr Vorel <pvorel@suse.cz>
* Copyright (c) 2019 Martin Doucha <mdoucha@suse.cz>
*/
#include <errno.h>
#include <netdb.h>
#include <string.h>
#include <stdlib.h>
#define TST_NO_DEFAULT_MAIN
#include "tst_test.h"
#include "tst_net.h"
#include "tst_private.h"
void tst_print_svar(const char *name, const char *val)
{
if (name && val)
printf("export %s=\"%s\"\n", name, val);
}
void tst_print_svar_change(const char *name, const char *val)
{
if (name && val)
printf("export %s=\"${%s:-%s}\"\n", name, name, val);
}
/*
* Function bit_count is from ipcalc project, ipcalc.c.
*/
static int tst_bit_count(uint32_t i)
{
int c = 0;
unsigned int seen_one = 0;
while (i > 0) {
if (i & 1) {
seen_one = 1;
c++;
} else {
if (seen_one)
return -1;
}
i >>= 1;
}
return c;
}
/*
* Function mask2prefix is from ipcalc project, ipcalc.c.
*/
static int tst_mask2prefix(struct in_addr mask)
{
return tst_bit_count(ntohl(mask.s_addr));
}
/*
* Function ipv4_mask_to_int is from ipcalc project, ipcalc.c.
*/
static int tst_ipv4_mask_to_int(const char *prefix)
{
int ret;
struct in_addr in;
ret = inet_pton(AF_INET, prefix, &in);
if (ret == 0)
return -1;
return tst_mask2prefix(in);
}
/*
* Function safe_atoi is from ipcalc project, ipcalc.c.
*/
static int tst_safe_atoi(const char *s, int *ret_i)
{
char *x = NULL;
long l;
errno = 0;
l = strtol(s, &x, 0);
if (!x || x == s || *x || errno)
return errno > 0 ? -errno : -EINVAL;
if ((long)(int)l != l)
return -ERANGE;
*ret_i = (int)l;
return 0;
}
/*
* Function get_prefix use code from ipcalc project, str_to_prefix/ipcalc.c.
*/
int tst_get_prefix(const char *ip_str, int is_ipv6)
{
char *prefix_str = NULL;
int prefix = -1, r;
prefix_str = strchr(ip_str, '/');
if (!prefix_str)
return -1;
*(prefix_str++) = '\0';
if (!is_ipv6 && strchr(prefix_str, '.'))
prefix = tst_ipv4_mask_to_int(prefix_str);
else {
r = tst_safe_atoi(prefix_str, &prefix);
if (r != 0)
tst_brk_comment("conversion error: '%s' is not integer",
prefix_str);
}
if (prefix < 0 || ((is_ipv6 && prefix > MAX_IPV6_PREFIX) ||
(!is_ipv6 && prefix > MAX_IPV4_PREFIX)))
tst_brk_comment("bad %s prefix: %s", is_ipv6 ? "IPv6" : "IPv4",
prefix_str);
return prefix;
}
void tst_get_in_addr(const char *ip_str, struct in_addr *ip)
{
if (inet_pton(AF_INET, ip_str, ip) <= 0)
tst_brk_comment("bad IPv4 address: '%s'", ip_str);
}
void tst_get_in6_addr(const char *ip_str, struct in6_addr *ip6)
{
if (inet_pton(AF_INET6, ip_str, ip6) <= 0)
tst_brk_comment("bad IPv6 address: '%s'", ip_str);
}
socklen_t tst_get_connect_address(int sock, struct sockaddr_storage *addr)
{
struct sockaddr_in *inet_ptr;
struct sockaddr_in6 *inet6_ptr;
size_t tmp_size;
socklen_t ret = sizeof(*addr);
SAFE_GETSOCKNAME(sock, (struct sockaddr*)addr, &ret);
/* Sanitize wildcard addresses */
switch (addr->ss_family) {
case AF_INET:
inet_ptr = (struct sockaddr_in*)addr;
switch (ntohl(inet_ptr->sin_addr.s_addr)) {
case INADDR_ANY:
case INADDR_BROADCAST:
inet_ptr->sin_addr.s_addr = htonl(INADDR_LOOPBACK);
break;
}
break;
case AF_INET6:
inet6_ptr = (struct sockaddr_in6*)addr;
tmp_size = sizeof(struct in6_addr);
if (!memcmp(&inet6_ptr->sin6_addr, &in6addr_any, tmp_size)) {
memcpy(&inet6_ptr->sin6_addr, &in6addr_loopback,
tmp_size);
}
break;
}
return ret;
}
void tst_init_sockaddr_inet(struct sockaddr_in *sa, const char *ip_str, uint16_t port)
{
memset(sa, 0, sizeof(struct sockaddr_in));
sa->sin_family = AF_INET;
sa->sin_port = htons(port);
tst_get_in_addr(ip_str, &sa->sin_addr);
}
void tst_init_sockaddr_inet_bin(struct sockaddr_in *sa, uint32_t ip_val, uint16_t port)
{
memset(sa, 0, sizeof(struct sockaddr_in));
sa->sin_family = AF_INET;
sa->sin_port = htons(port);
sa->sin_addr.s_addr = htonl(ip_val);
}
void tst_init_sockaddr_inet6(struct sockaddr_in6 *sa, const char *ip_str, uint16_t port)
{
memset(sa, 0, sizeof(struct sockaddr_in6));
sa->sin6_family = AF_INET6;
sa->sin6_port = htons(port);
tst_get_in6_addr(ip_str, &sa->sin6_addr);
}
void tst_init_sockaddr_inet6_bin(struct sockaddr_in6 *sa, const struct in6_addr *ip_val, uint16_t port)
{
memset(sa, 0, sizeof(struct sockaddr_in6));
sa->sin6_family = AF_INET6;
sa->sin6_port = htons(port);
memcpy(&sa->sin6_addr, ip_val, sizeof(struct in6_addr));
}
void safe_getaddrinfo(const char *file, const int lineno, const char *src_addr,
const char *port, const struct addrinfo *hints,
struct addrinfo **addr_info)
{
int err = getaddrinfo(src_addr, port, hints, addr_info);
if (err)
tst_brk(TBROK, "%s:%d: getaddrinfo failed, %s", file, lineno,
gai_strerror(err));
if (!*addr_info)
tst_brk(TBROK, "%s:%d: failed to get the address", file, lineno);
}
|
