1 /*
2 * Copyright (c) 2023-2024 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 #include "net_test_lib.h"
53
54 T_GLOBAL_META(
55 T_META_NAMESPACE("xnu.net"),
56 T_META_RADAR_COMPONENT_NAME("xnu"),
57 T_META_RADAR_COMPONENT_VERSION("networking")
58 );
59
60 #define MAX_IPv6_STR_LEN 64
61
62 static char l_addr_str[MAX_IPv6_STR_LEN];
63 static char f_addr_str[MAX_IPv6_STR_LEN];
64
65 const struct in6_addr in6addr_any = IN6ADDR_ANY_INIT;
66 #define s6_addr32 __u6_addr.__u6_addr32
67
68
69 static void
init_sin_address(struct sockaddr_in * sin)70 init_sin_address(struct sockaddr_in *sin)
71 {
72 memset(sin, 0, sizeof(struct sockaddr_in));
73 sin->sin_len = sizeof(struct sockaddr_in);
74 sin->sin_family = AF_INET;
75 }
76
77 static void
init_sin6_address(struct sockaddr_in6 * sin6)78 init_sin6_address(struct sockaddr_in6 *sin6)
79 {
80 memset(sin6, 0, sizeof(struct sockaddr_in6));
81 sin6->sin6_len = sizeof(struct sockaddr_in6);
82 sin6->sin6_family = AF_INET6;
83 }
84
85 static void
udp_disconnect_v4(int client_fd,struct sockaddr_in * sin_null,int expected_error)86 udp_disconnect_v4(int client_fd, struct sockaddr_in *sin_null, int expected_error)
87 {
88 if (expected_error == 0) {
89 socklen_t socklen;
90 struct sockaddr_in sin_local = { 0 };
91 struct sockaddr_in sin_peer = { 0 };
92
93 // Disconnect
94 T_EXPECT_POSIX_FAILURE(connect(client_fd, (struct sockaddr *)sin_null, sizeof(struct sockaddr_in)), EADDRNOTAVAIL, NULL);
95
96 socklen = sizeof(sin_local);
97 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin_local, &socklen), NULL);
98 (void)inet_ntop(AF_INET, &sin_local.sin_addr, l_addr_str, sizeof(l_addr_str));
99
100 socklen = sizeof(sin_peer);
101 T_EXPECT_POSIX_FAILURE(getpeername(client_fd, (struct sockaddr *)&sin_peer, &socklen), ENOTCONN, NULL);
102 (void)inet_ntop(AF_INET, &sin_peer.sin_addr, f_addr_str, sizeof(f_addr_str));
103
104 T_LOG("disconnected with %s:%u to %s:%u",
105 l_addr_str, ntohs(sin_local.sin_port),
106 f_addr_str, ntohs(sin_peer.sin_port));
107 } else {
108 T_EXPECT_POSIX_FAILURE(connect(client_fd, (struct sockaddr *)sin_null, sizeof(struct sockaddr_in)), expected_error, NULL);
109 }
110 }
111
112 static void
udp_connect_v4(int client_fd,struct sockaddr_in * sin_to)113 udp_connect_v4(int client_fd, struct sockaddr_in *sin_to)
114 {
115 int listen_fd = -1;
116 socklen_t socklen;
117 struct sockaddr_in sin_local = { 0 };
118 struct sockaddr_in sin_peer = { 0 };
119 struct sockaddr_in sin;
120
121 init_sin_address(&sin);
122 init_sin_address(&sin_local);
123 init_sin_address(&sin_peer);
124
125 T_ASSERT_POSIX_SUCCESS(listen_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP), NULL);
126
127 T_ASSERT_POSIX_SUCCESS(bind(listen_fd, (struct sockaddr *)&sin, sizeof(sin)), NULL);
128
129 socklen = sizeof(sin);
130 T_ASSERT_POSIX_SUCCESS(getsockname(listen_fd, (struct sockaddr *)&sin, &socklen), NULL);
131
132 T_LOG("listening on port: %u", ntohs(sin.sin_port));
133 sin_to->sin_port = sin.sin_port;
134
135 T_LOG("connect with sin_len: %u sin_family: %u sin_port: %u sin_addr: 0x%08x",
136 sin_to->sin_len, sin_to->sin_family, ntohs(sin_to->sin_port), ntohl(sin_to->sin_addr.s_addr));
137
138 T_EXPECT_POSIX_SUCCESS(connect(client_fd, (struct sockaddr *)sin_to, sizeof(struct sockaddr_in)), NULL);
139
140 socklen = sizeof(sin_local);
141 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin_local, &socklen), NULL);
142 (void)inet_ntop(AF_INET, &sin_local.sin_addr, l_addr_str, sizeof(l_addr_str));
143
144 socklen = sizeof(sin_peer);
145 T_ASSERT_POSIX_SUCCESS(getpeername(client_fd, (struct sockaddr *)&sin_peer, &socklen), NULL);
146 (void)inet_ntop(AF_INET, &sin_peer.sin_addr, f_addr_str, sizeof(f_addr_str));
147
148 T_LOG("connected with %s:%u to %s:%u",
149 l_addr_str, ntohs(sin_local.sin_port),
150 f_addr_str, ntohs(sin_peer.sin_port));
151
152 T_ASSERT_POSIX_SUCCESS(close(listen_fd), NULL);
153 }
154
155 T_DECL(udp_disconnect_null_ipv4, "UDP connect with a IPv4 loopback address", T_META_TAG_VM_PREFERRED)
156 {
157 struct sockaddr_in sin = { 0 };
158 init_sin_address(&sin);
159 T_ASSERT_EQ(inet_pton(AF_INET, "127.0.0.1", &sin.sin_addr), 1, NULL);
160
161 struct sockaddr_in sin_null = { 0 };
162 init_sin_address(&sin_null);
163
164 int s = -1;
165 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP), NULL);
166
167 udp_connect_v4(s, &sin);
168 udp_disconnect_v4(s, &sin_null, 0);
169 udp_connect_v4(s, &sin);
170
171 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
172 }
173
174 static void
udp_disconnect_v6(int client_fd,struct sockaddr_in6 * sin6_null,int expected_error)175 udp_disconnect_v6(int client_fd,
176 struct sockaddr_in6 *sin6_null, int expected_error)
177 {
178 if (expected_error == 0) {
179 socklen_t socklen;
180 struct sockaddr_in6 sin6_local = { 0 };
181 struct sockaddr_in6 sin6_peer = { 0 };
182
183 // Disconnect
184 T_EXPECT_POSIX_FAILURE(connect(client_fd, (struct sockaddr *)sin6_null, sizeof(struct sockaddr_in6)), EADDRNOTAVAIL, NULL);
185
186 socklen = sizeof(sin6_local);
187 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin6_local, &socklen), NULL);
188 (void)inet_ntop(AF_INET6, &sin6_local.sin6_addr, l_addr_str, sizeof(l_addr_str));
189
190 socklen = sizeof(sin6_peer);
191 T_EXPECT_POSIX_FAILURE(getpeername(client_fd, (struct sockaddr *)&sin6_peer, &socklen), ENOTCONN, NULL);
192 (void)inet_ntop(AF_INET6, &sin6_peer.sin6_addr, f_addr_str, sizeof(f_addr_str));
193
194 T_LOG("re=connected from %s:%u to %s:%u",
195 l_addr_str, ntohs(sin6_local.sin6_port),
196 f_addr_str, ntohs(sin6_peer.sin6_port));
197 } else {
198 T_EXPECT_POSIX_FAILURE(connect(client_fd, (struct sockaddr *)sin6_null, sizeof(struct sockaddr_in6)), expected_error, NULL);
199 }
200 }
201
202 static void
udp_connect_v6(int client_fd,struct sockaddr_in6 * sin6_to)203 udp_connect_v6(int client_fd, struct sockaddr_in6 *sin6_to)
204 {
205 int listen_fd = -1;
206 socklen_t socklen;
207 int off = 0;
208 struct sockaddr_in6 sin6_local = { 0 };
209 struct sockaddr_in6 sin6_peer = { 0 };
210 struct sockaddr_in6 sin6;
211
212 init_sin6_address(&sin6);
213 init_sin6_address(&sin6_local);
214 init_sin6_address(&sin6_peer);
215
216 T_ASSERT_POSIX_SUCCESS(listen_fd = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP), NULL);
217
218 T_ASSERT_POSIX_SUCCESS(setsockopt(listen_fd, IPPROTO_IPV6, IPV6_V6ONLY, &off, sizeof(off)), NULL);
219
220 T_ASSERT_POSIX_SUCCESS(bind(listen_fd, (struct sockaddr *)&sin6, sizeof(sin6)), NULL);
221
222 socklen = sizeof(sin6);
223 T_ASSERT_POSIX_SUCCESS(getsockname(listen_fd, (struct sockaddr *)&sin6, &socklen), NULL);
224
225 T_LOG("listening on port: %u", ntohs(sin6.sin6_port));
226 sin6_to->sin6_port = sin6.sin6_port;
227
228 (void)inet_ntop(AF_INET6, &sin6_to->sin6_addr, l_addr_str, sizeof(l_addr_str));
229 T_LOG("connecting with sin6_len: %u sin6_family: %u sin6_port: %u sin6_addr: %s",
230 sin6_to->sin6_len, sin6_to->sin6_family, ntohs(sin6_to->sin6_port), l_addr_str);
231
232 T_EXPECT_POSIX_SUCCESS(connect(client_fd, (struct sockaddr *)sin6_to, sizeof(struct sockaddr_in6)), NULL);
233
234 socklen = sizeof(sin6_local);
235 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin6_local, &socklen), NULL);
236 (void)inet_ntop(AF_INET6, &sin6_local.sin6_addr, l_addr_str, sizeof(l_addr_str));
237
238 socklen = sizeof(sin6_peer);
239 T_ASSERT_POSIX_SUCCESS(getpeername(client_fd, (struct sockaddr *)&sin6_peer, &socklen), NULL);
240 (void)inet_ntop(AF_INET6, &sin6_peer.sin6_addr, f_addr_str, sizeof(f_addr_str));
241
242 T_LOG("connected with %s:%u to %s:%u",
243 l_addr_str, ntohs(sin6_local.sin6_port),
244 f_addr_str, ntohs(sin6_peer.sin6_port));
245
246 T_ASSERT_POSIX_SUCCESS(close(listen_fd), NULL);
247 }
248
249 T_DECL(udp_disconnect_null_ipv6, "UDP connect with IPv4 multicast mapped IPv6 address", T_META_TAG_VM_PREFERRED)
250 {
251 if (!has_ipv6_default_route()) {
252 T_SKIP("test require IPv4 default route");
253 }
254
255 int s = -1;
256 struct sockaddr_in6 sin6 = { 0 };
257
258 init_sin6_address(&sin6);
259 T_ASSERT_EQ(inet_pton(AF_INET6, "::1", &sin6.sin6_addr), 1, NULL);
260
261 struct sockaddr_in6 sin6_null = { 0 };
262 init_sin6_address(&sin6_null);
263
264 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP), NULL);
265
266 udp_connect_v6(s, &sin6);
267 udp_disconnect_v6(s, &sin6_null, 0);
268 udp_connect_v6(s, &sin6);
269
270 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
271 }
272
273 T_DECL(udp_disconnect_null_ipv4_mapped_ipv6, "UDP connect with IPv4 multicast mapped IPv6 address", T_META_TAG_VM_PREFERRED)
274 {
275 if (!has_ipv4_default_route()) {
276 T_SKIP("test require IPv4 default route");
277 }
278
279 struct sockaddr_in6 sin6 = { 0 };
280 init_sin6_address(&sin6);
281 T_ASSERT_EQ(inet_pton(AF_INET6, "::ffff:127.0.0.1", &sin6.sin6_addr), 1, NULL);
282
283 struct sockaddr_in6 sin6_null = { 0 };
284 init_sin6_address(&sin6_null);
285
286 int s = -1;
287 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP), NULL);
288
289 udp_connect_v6(s, &sin6);
290 udp_disconnect_v6(s, &sin6_null, 0);
291 udp_connect_v6(s, &sin6);
292
293 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
294 }
295
296 T_DECL(udp_disconnect_mapped_ipv4_mapped_ipv6, "UDP connect with IPv4 multicast mapped IPv6 address", T_META_TAG_VM_PREFERRED)
297 {
298 if (!has_ipv4_default_route()) {
299 T_SKIP("test require IPv4 default route");
300 }
301
302 int s = -1;
303 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_DGRAM, IPPROTO_UDP), NULL);
304
305 struct sockaddr_in6 sin6 = { 0 };
306 init_sin6_address(&sin6);
307 T_ASSERT_EQ(inet_pton(AF_INET6, "::ffff:127.0.0.1", &sin6.sin6_addr), 1, NULL);
308
309 struct sockaddr_in6 sin6_null = { 0 };
310 init_sin6_address(&sin6_null);
311 sin6_null.sin6_addr.s6_addr[10] = 0xff;
312 sin6_null.sin6_addr.s6_addr[11] = 0xff;
313
314 udp_connect_v6(s, &sin6);
315 udp_disconnect_v6(s, &sin6_null, 0);
316 udp_connect_v6(s, &sin6);
317
318 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
319 }
320