1 /*
2 * Copyright (c) 2023 Apple Inc. All rights reserved.
3 *
4 * @APPLE_OSREFERENCE_LICENSE_HEADER_START@
5 *
6 * This file contains Original Code and/or Modifications of Original Code
7 * as defined in and that are subject to the Apple Public Source License
8 * Version 2.0 (the 'License'). You may not use this file except in
9 * compliance with the License. The rights granted to you under the License
10 * may not be used to create, or enable the creation or redistribution of,
11 * unlawful or unlicensed copies of an Apple operating system, or to
12 * circumvent, violate, or enable the circumvention or violation of, any
13 * terms of an Apple operating system software license agreement.
14 *
15 * Please obtain a copy of the License at
16 * http://www.opensource.apple.com/apsl/ and read it before using this file.
17 *
18 * The Original Code and all software distributed under the License are
19 * distributed on an 'AS IS' basis, WITHOUT WARRANTY OF ANY KIND, EITHER
20 * EXPRESS OR IMPLIED, AND APPLE HEREBY DISCLAIMS ALL SUCH WARRANTIES,
21 * INCLUDING WITHOUT LIMITATION, ANY WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE, QUIET ENJOYMENT OR NON-INFRINGEMENT.
23 * Please see the License for the specific language governing rights and
24 * limitations under the License.
25 *
26 * @APPLE_OSREFERENCE_LICENSE_HEADER_END@
27 */
28
29
30 /*
31 * Disconnect should happen when passed a NULL address
32 * Verify we can reconnect after a disconnect
33 */
34
35 #include <sys/fcntl.h>
36 #include <sys/socket.h>
37
38 #include <net/if.h>
39 #include <net/route.h>
40
41 #include <netinet/in.h>
42
43 #include <stdbool.h>
44 #include <stdlib.h>
45 #include <string.h>
46 #include <unistd.h>
47
48 #include <arpa/inet.h>
49
50 #include <darwintest.h>
51
52 T_GLOBAL_META(
53 T_META_NAMESPACE("xnu.net"),
54 T_META_RADAR_COMPONENT_NAME("xnu"),
55 T_META_RADAR_COMPONENT_VERSION("networking")
56 );
57
58 #define MAX_IPv6_STR_LEN 64
59
60 static char l_addr_str[MAX_IPv6_STR_LEN];
61 static char f_addr_str[MAX_IPv6_STR_LEN];
62
63 const struct in6_addr in6addr_any = IN6ADDR_ANY_INIT;
64 #define s6_addr32 __u6_addr.__u6_addr32
65
66 #define RTM_BUFLEN (sizeof(struct rt_msghdr) + 6 * SOCK_MAXADDRLEN)
67
68 #define ROUNDUP(a) \
69 ((a) > 0 ? (1 + (((a) - 1) | (sizeof(long) - 1))) : sizeof(long))
70 #define ADVANCE(x, n) (x += ROUNDUP((n)->sa_len))
71
72 static bool
has_v4_default_route(void)73 has_v4_default_route(void)
74 {
75 bool result = false;
76 struct rt_msghdr *rtm = NULL;
77 struct sockaddr_in sin = {};
78
79 sin.sin_len = sizeof(struct sockaddr_in);
80 sin.sin_family = AF_INET;
81 sin.sin_addr.s_addr = INADDR_ANY;
82
83 T_QUIET; T_ASSERT_NOTNULL(rtm = (struct rt_msghdr *)calloc(1, RTM_BUFLEN), NULL);
84
85 rtm->rtm_msglen = sizeof(struct rt_msghdr) + sizeof(struct sockaddr_in);
86 rtm->rtm_version = RTM_VERSION;
87 rtm->rtm_type = RTM_GET;
88 rtm->rtm_flags = RTF_UP | RTF_STATIC | RTF_GATEWAY | RTF_HOST;
89 rtm->rtm_addrs = RTA_DST;
90 rtm->rtm_pid = getpid();
91 rtm->rtm_seq = 1;
92
93 uint8_t *cp = (unsigned char *)(rtm + 1);
94
95 bcopy(&sin, cp, sin.sin_len);
96 cp += ROUNDUP(sin.sin_len);
97
98 u_short len = (u_short)(cp - (uint8_t *)rtm);
99
100 rtm->rtm_msglen = len;
101
102 int fd;
103 T_QUIET; T_ASSERT_POSIX_SUCCESS(fd = socket(PF_ROUTE, SOCK_RAW, 0), NULL);
104
105 ssize_t sent = send(fd, rtm, len, 0);
106 if (sent == len) {
107 result = true;
108 } else {
109 result = false;
110 }
111
112 (void) close(fd);
113 free(rtm);
114
115 return result;
116 }
117
118 static bool
has_v6_default_route(void)119 has_v6_default_route(void)
120 {
121 bool result = false;
122 struct rt_msghdr *rtm = NULL;
123 struct sockaddr_in6 sin6 = {};
124
125 sin6.sin6_len = sizeof(struct sockaddr_in6);
126 sin6.sin6_family = AF_INET6;
127
128 T_QUIET; T_ASSERT_NOTNULL(rtm = (struct rt_msghdr *)calloc(1, RTM_BUFLEN), NULL);
129
130 rtm->rtm_msglen = sizeof(struct rt_msghdr) + sizeof(struct sockaddr_in);
131 rtm->rtm_version = RTM_VERSION;
132 rtm->rtm_type = RTM_GET;
133 rtm->rtm_flags = RTF_UP | RTF_STATIC | RTF_GATEWAY | RTF_HOST;
134 rtm->rtm_addrs = RTA_DST;
135 rtm->rtm_pid = getpid();
136 rtm->rtm_seq = 1;
137
138 uint8_t *cp = (unsigned char *)(rtm + 1);
139
140 bcopy(&sin6, cp, sin6.sin6_len);
141 cp += ROUNDUP(sin6.sin6_len);
142
143 u_short len = (u_short)(cp - (uint8_t *)rtm);
144
145 rtm->rtm_msglen = len;
146
147 int fd;
148 T_QUIET; T_ASSERT_POSIX_SUCCESS(fd = socket(PF_ROUTE, SOCK_RAW, 0), NULL);
149
150 ssize_t sent = send(fd, rtm, len, 0);
151 if (sent == len) {
152 result = true;
153 } else {
154 result = false;
155 }
156
157 (void) close(fd);
158 free(rtm);
159
160 return result;
161 }
162
163
164 static void
init_sin_address(struct sockaddr_in * sin)165 init_sin_address(struct sockaddr_in *sin)
166 {
167 memset(sin, 0, sizeof(struct sockaddr_in));
168 sin->sin_len = sizeof(struct sockaddr_in);
169 sin->sin_family = AF_INET;
170 }
171
172 static void
init_sin6_address(struct sockaddr_in6 * sin6)173 init_sin6_address(struct sockaddr_in6 *sin6)
174 {
175 memset(sin6, 0, sizeof(struct sockaddr_in6));
176 sin6->sin6_len = sizeof(struct sockaddr_in6);
177 sin6->sin6_family = AF_INET6;
178 }
179
180 static void
udp_disconnect_v4(int client_fd,struct sockaddr_in * sin_null,int expected_error)181 udp_disconnect_v4(int client_fd, struct sockaddr_in *sin_null, int expected_error)
182 {
183 if (expected_error == 0) {
184 socklen_t socklen;
185 struct sockaddr_in sin_local = {};
186 struct sockaddr_in sin_peer = {};
187
188 // Disconnect
189 T_EXPECT_POSIX_FAILURE(connect(client_fd, (struct sockaddr *)sin_null, sizeof(struct sockaddr_in)), EADDRNOTAVAIL, NULL);
190
191 socklen = sizeof(sin_local);
192 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin_local, &socklen), NULL);
193 (void)inet_ntop(AF_INET, &sin_local.sin_addr, l_addr_str, sizeof(l_addr_str));
194
195 socklen = sizeof(sin_peer);
196 T_EXPECT_POSIX_FAILURE(getpeername(client_fd, (struct sockaddr *)&sin_peer, &socklen), ENOTCONN, NULL);
197 (void)inet_ntop(AF_INET, &sin_peer.sin_addr, f_addr_str, sizeof(f_addr_str));
198
199 T_LOG("disconnected with %s:%u to %s:%u",
200 l_addr_str, ntohs(sin_local.sin_port),
201 f_addr_str, ntohs(sin_peer.sin_port));
202 } else {
203 T_EXPECT_POSIX_FAILURE(connect(client_fd, (struct sockaddr *)sin_null, sizeof(struct sockaddr_in)), expected_error, NULL);
204 }
205 }
206
207 static void
udp_connect_v4(int client_fd,struct sockaddr_in * sin_to)208 udp_connect_v4(int client_fd, struct sockaddr_in *sin_to)
209 {
210 int listen_fd = -1;
211 socklen_t socklen;
212 struct sockaddr_in sin_local = {};
213 struct sockaddr_in sin_peer = {};
214 struct sockaddr_in sin;
215
216 init_sin_address(&sin);
217 init_sin_address(&sin_local);
218 init_sin_address(&sin_peer);
219
220 T_ASSERT_POSIX_SUCCESS(listen_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP), NULL);
221
222 T_ASSERT_POSIX_SUCCESS(bind(listen_fd, (struct sockaddr *)&sin, sizeof(sin)), NULL);
223
224 socklen = sizeof(sin);
225 T_ASSERT_POSIX_SUCCESS(getsockname(listen_fd, (struct sockaddr *)&sin, &socklen), NULL);
226
227 T_LOG("listening on port: %u", ntohs(sin.sin_port));
228 sin_to->sin_port = sin.sin_port;
229
230 T_LOG("connect with sin_len: %u sin_family: %u sin_port: %u sin_addr: 0x%08x",
231 sin_to->sin_len, sin_to->sin_family, ntohs(sin_to->sin_port), ntohl(sin_to->sin_addr.s_addr));
232
233 T_EXPECT_POSIX_SUCCESS(connect(client_fd, (struct sockaddr *)sin_to, sizeof(struct sockaddr_in)), NULL);
234
235 socklen = sizeof(sin_local);
236 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin_local, &socklen), NULL);
237 (void)inet_ntop(AF_INET, &sin_local.sin_addr, l_addr_str, sizeof(l_addr_str));
238
239 socklen = sizeof(sin_peer);
240 T_ASSERT_POSIX_SUCCESS(getpeername(client_fd, (struct sockaddr *)&sin_peer, &socklen), NULL);
241 (void)inet_ntop(AF_INET, &sin_peer.sin_addr, f_addr_str, sizeof(f_addr_str));
242
243 T_LOG("connected with %s:%u to %s:%u",
244 l_addr_str, ntohs(sin_local.sin_port),
245 f_addr_str, ntohs(sin_peer.sin_port));
246
247 T_ASSERT_POSIX_SUCCESS(close(listen_fd), NULL);
248 }
249
250 T_DECL(udp_disconnect_null_ipv4, "UDP connect with a IPv4 loopback address")
251 {
252 struct sockaddr_in sin = {};
253 init_sin_address(&sin);
254 T_ASSERT_EQ(inet_pton(AF_INET, "127.0.0.1", &sin.sin_addr), 1, NULL);
255
256 struct sockaddr_in sin_null = {};
257 init_sin_address(&sin_null);
258
259 int s = -1;
260 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP), NULL);
261
262 udp_connect_v4(s, &sin);
263 udp_disconnect_v4(s, &sin_null, 0);
264 udp_connect_v4(s, &sin);
265
266 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
267 }
268
269 static void
udp_disconnect_v6(int client_fd,struct sockaddr_in6 * sin6_null,int expected_error)270 udp_disconnect_v6(int client_fd,
271 struct sockaddr_in6 *sin6_null, int expected_error)
272 {
273 if (expected_error == 0) {
274 socklen_t socklen;
275 struct sockaddr_in6 sin6_local = {};
276 struct sockaddr_in6 sin6_peer = {};
277
278 // Disconnect
279 T_EXPECT_POSIX_FAILURE(connect(client_fd, (struct sockaddr *)sin6_null, sizeof(struct sockaddr_in6)), EADDRNOTAVAIL, NULL);
280
281 socklen = sizeof(sin6_local);
282 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin6_local, &socklen), NULL);
283 (void)inet_ntop(AF_INET6, &sin6_local.sin6_addr, l_addr_str, sizeof(l_addr_str));
284
285 socklen = sizeof(sin6_peer);
286 T_EXPECT_POSIX_FAILURE(getpeername(client_fd, (struct sockaddr *)&sin6_peer, &socklen), ENOTCONN, NULL);
287 (void)inet_ntop(AF_INET6, &sin6_peer.sin6_addr, f_addr_str, sizeof(f_addr_str));
288
289 T_LOG("re=connected from %s:%u to %s:%u",
290 l_addr_str, ntohs(sin6_local.sin6_port),
291 f_addr_str, ntohs(sin6_peer.sin6_port));
292 } else {
293 T_EXPECT_POSIX_FAILURE(connect(client_fd, (struct sockaddr *)sin6_null, sizeof(struct sockaddr_in6)), expected_error, NULL);
294 }
295 }
296
297 static void
udp_connect_v6(int client_fd,struct sockaddr_in6 * sin6_to)298 udp_connect_v6(int client_fd, struct sockaddr_in6 *sin6_to)
299 {
300 int listen_fd = -1;
301 socklen_t socklen;
302 int off = 0;
303 struct sockaddr_in6 sin6_local = {};
304 struct sockaddr_in6 sin6_peer = {};
305 struct sockaddr_in6 sin6;
306
307 init_sin6_address(&sin6);
308 init_sin6_address(&sin6_local);
309 init_sin6_address(&sin6_peer);
310
311 T_ASSERT_POSIX_SUCCESS(listen_fd = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP), NULL);
312
313 T_ASSERT_POSIX_SUCCESS(setsockopt(listen_fd, IPPROTO_IPV6, IPV6_V6ONLY, &off, sizeof(off)), NULL);
314
315 T_ASSERT_POSIX_SUCCESS(bind(listen_fd, (struct sockaddr *)&sin6, sizeof(sin6)), NULL);
316
317 socklen = sizeof(sin6);
318 T_ASSERT_POSIX_SUCCESS(getsockname(listen_fd, (struct sockaddr *)&sin6, &socklen), NULL);
319
320 T_LOG("listening on port: %u", ntohs(sin6.sin6_port));
321 sin6_to->sin6_port = sin6.sin6_port;
322
323 (void)inet_ntop(AF_INET6, &sin6_to->sin6_addr, l_addr_str, sizeof(l_addr_str));
324 T_LOG("connecting with sin6_len: %u sin6_family: %u sin6_port: %u sin6_addr: %s",
325 sin6_to->sin6_len, sin6_to->sin6_family, ntohs(sin6_to->sin6_port), l_addr_str);
326
327 T_EXPECT_POSIX_SUCCESS(connect(client_fd, (struct sockaddr *)sin6_to, sizeof(struct sockaddr_in6)), NULL);
328
329 socklen = sizeof(sin6_local);
330 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin6_local, &socklen), NULL);
331 (void)inet_ntop(AF_INET6, &sin6_local.sin6_addr, l_addr_str, sizeof(l_addr_str));
332
333 socklen = sizeof(sin6_peer);
334 T_ASSERT_POSIX_SUCCESS(getpeername(client_fd, (struct sockaddr *)&sin6_peer, &socklen), NULL);
335 (void)inet_ntop(AF_INET6, &sin6_peer.sin6_addr, f_addr_str, sizeof(f_addr_str));
336
337 T_LOG("connected with %s:%u to %s:%u",
338 l_addr_str, ntohs(sin6_local.sin6_port),
339 f_addr_str, ntohs(sin6_peer.sin6_port));
340
341 T_ASSERT_POSIX_SUCCESS(close(listen_fd), NULL);
342 }
343
344 T_DECL(udp_disconnect_null_ipv6, "UDP connect with IPv4 multicast mapped IPv6 address")
345 {
346 if (!has_v6_default_route()) {
347 T_SKIP("test require IPv4 default route");
348 }
349
350 int s = -1;
351 struct sockaddr_in6 sin6 = {};
352
353 init_sin6_address(&sin6);
354 T_ASSERT_EQ(inet_pton(AF_INET6, "::1", &sin6.sin6_addr), 1, NULL);
355
356 struct sockaddr_in6 sin6_null = {};
357 init_sin6_address(&sin6_null);
358
359 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP), NULL);
360
361 udp_connect_v6(s, &sin6);
362 udp_disconnect_v6(s, &sin6_null, 0);
363 udp_connect_v6(s, &sin6);
364
365 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
366 }
367
368 T_DECL(udp_disconnect_null_ipv4_mapped_ipv6, "UDP connect with IPv4 multicast mapped IPv6 address")
369 {
370 if (!has_v4_default_route()) {
371 T_SKIP("test require IPv4 default route");
372 }
373
374 struct sockaddr_in6 sin6 = {};
375 init_sin6_address(&sin6);
376 T_ASSERT_EQ(inet_pton(AF_INET6, "::ffff:127.0.0.1", &sin6.sin6_addr), 1, NULL);
377
378 struct sockaddr_in6 sin6_null = {};
379 init_sin6_address(&sin6_null);
380
381 int s = -1;
382 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP), NULL);
383
384 udp_connect_v6(s, &sin6);
385 udp_disconnect_v6(s, &sin6_null, 0);
386 udp_connect_v6(s, &sin6);
387
388 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
389 }
390
391 T_DECL(udp_disconnect_mapped_ipv4_mapped_ipv6, "UDP connect with IPv4 multicast mapped IPv6 address")
392 {
393 if (!has_v4_default_route()) {
394 T_SKIP("test require IPv4 default route");
395 }
396
397 int s = -1;
398 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP), NULL);
399
400 struct sockaddr_in6 sin6 = {};
401 init_sin6_address(&sin6);
402 T_ASSERT_EQ(inet_pton(AF_INET6, "::ffff:127.0.0.1", &sin6.sin6_addr), 1, NULL);
403
404 struct sockaddr_in6 sin6_null = {};
405 init_sin6_address(&sin6_null);
406 sin6_null.sin6_addr.s6_addr[10] = 0xff;
407 sin6_null.sin6_addr.s6_addr[11] = 0xff;
408
409 udp_connect_v6(s, &sin6);
410 udp_disconnect_v6(s, &sin6_null, 0);
411 udp_connect_v6(s, &sin6);
412
413 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
414 }
415