1 /*
2 * Copyright (c) 2022-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 #include <sys/fcntl.h>
30 #include <sys/socket.h>
31 #include <net/if.h>
32 #include <netinet/in.h>
33 #include <netinet/tcp.h>
34 #include <arpa/inet.h>
35
36 #include <darwintest.h>
37 #include <string.h>
38 #include <unistd.h>
39
40 T_GLOBAL_META(T_META_NAMESPACE("xnu.net"));
41
42 #define MAX_IPv6_STR_LEN 64
43
44 static char l_addr_str[MAX_IPv6_STR_LEN];
45 static char f_addr_str[MAX_IPv6_STR_LEN];
46
47 const struct in6_addr in6addr_any = IN6ADDR_ANY_INIT;
48 #define s6_addr32 __u6_addr.__u6_addr32
49
50 static void
init_sin_address(struct sockaddr_in * sin)51 init_sin_address(struct sockaddr_in *sin)
52 {
53 memset(sin, 0, sizeof(struct sockaddr_in));
54 sin->sin_len = sizeof(struct sockaddr_in);
55 sin->sin_family = AF_INET;
56 }
57
58 static void
init_sin6_address(struct sockaddr_in6 * sin6)59 init_sin6_address(struct sockaddr_in6 *sin6)
60 {
61 memset(sin6, 0, sizeof(struct sockaddr_in6));
62 sin6->sin6_len = sizeof(struct sockaddr_in6);
63 sin6->sin6_family = AF_INET6;
64 }
65
66 static int
tcp_send_implied_connect_v4(int client_fd,struct sockaddr_in * sin_to,int expected_error)67 tcp_send_implied_connect_v4(int client_fd, struct sockaddr_in *sin_to, int expected_error)
68 {
69 int listen_fd = -1;
70 socklen_t socklen;
71 int val = 10;
72 struct sockaddr_in sin_local = {};
73 struct sockaddr_in sin_peer = {};
74 struct sockaddr_in sin;
75
76 init_sin_address(&sin);
77 init_sin_address(&sin_local);
78 init_sin_address(&sin_peer);
79
80 T_ASSERT_POSIX_SUCCESS(listen_fd = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP), NULL);
81
82 T_ASSERT_POSIX_SUCCESS(bind(listen_fd, (struct sockaddr *)&sin, sizeof(sin)), NULL);
83
84 socklen = sizeof(sin);
85 T_ASSERT_POSIX_SUCCESS(getsockname(listen_fd, (struct sockaddr *)&sin, &socklen), NULL);
86
87 T_LOG("listening on port: %u", ntohs(sin.sin_port));
88 sin_to->sin_port = sin.sin_port;
89
90 T_ASSERT_POSIX_SUCCESS(listen(listen_fd, 10), NULL);
91
92 T_ASSERT_POSIX_SUCCESS(setsockopt(client_fd, IPPROTO_TCP, TCP_CONNECTIONTIMEOUT, &val, sizeof(val)), NULL);
93
94 T_LOG("sendmsg with sin_len: %u sin_family: %u sin_port: %u sin_addr: 0x%08x expected_error: %d",
95 sin_to->sin_len, sin_to->sin_family, ntohs(sin_to->sin_port), ntohl(sin_to->sin_addr.s_addr), expected_error);
96
97 struct iovec iovec = { .iov_base = "", .iov_len = 0 };
98 struct msghdr msg = {};
99 struct cmsghdr *cmsghdrp;
100 char buf[CMSG_SPACE(sizeof(int))];
101
102 msg.msg_name = sin_to;
103 msg.msg_namelen = sin_to->sin_len;
104 msg.msg_iov = &iovec;
105 msg.msg_iovlen = 1;
106 msg.msg_control = buf;
107 msg.msg_controllen = CMSG_SPACE(sizeof(int));
108
109 cmsghdrp = CMSG_FIRSTHDR(&msg);
110 cmsghdrp->cmsg_len = CMSG_LEN(sizeof(int));
111 cmsghdrp->cmsg_level = SOL_SOCKET;
112 cmsghdrp->cmsg_type = SCM_RIGHTS;
113
114 memcpy(CMSG_DATA(cmsghdrp), &client_fd, sizeof(client_fd));
115
116 ssize_t retval = sendmsg(client_fd, &msg, 0);
117
118 if (expected_error == 0) {
119 T_EXPECT_POSIX_SUCCESS(retval, "sendmsg(client_fd, &msg, 0)");
120 } else {
121 T_EXPECT_POSIX_FAILURE(retval, expected_error, "sendmsg(client_fd, &msg, 0)");
122 }
123
124 if (retval == 0) {
125 socklen = sizeof(sin_local);
126 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin_local, &socklen), NULL);
127 (void)inet_ntop(AF_INET, &sin_local.sin_addr, l_addr_str, sizeof(l_addr_str));
128
129 socklen = sizeof(sin_peer);
130 T_ASSERT_POSIX_SUCCESS(getpeername(client_fd, (struct sockaddr *)&sin_peer, &socklen), NULL);
131 (void)inet_ntop(AF_INET, &sin_peer.sin_addr, f_addr_str, sizeof(f_addr_str));
132
133 T_LOG("connected from %s:%u to %s:%u",
134 l_addr_str, ntohs(sin_local.sin_port),
135 f_addr_str, ntohs(sin_peer.sin_port));
136 }
137
138 T_ASSERT_POSIX_SUCCESS(close(listen_fd), NULL);
139
140 return 0;
141 }
142
143 static int
tcp_send_implied_connect_v6(int client_fd,struct sockaddr_in6 * sin6_to,int expected_error)144 tcp_send_implied_connect_v6(int client_fd, struct sockaddr_in6 *sin6_to, int expected_error)
145 {
146 int listen_fd = -1;
147 socklen_t socklen;
148 int off = 0;
149 int val = 10;
150 struct iovec iovec = { .iov_base = "", .iov_len = 0 };
151 struct msghdr msg = {};
152 struct cmsghdr *cmsghdrp;
153 char buf[CMSG_SPACE(sizeof(int))];
154 struct sockaddr_in6 sin6_local = {};
155 struct sockaddr_in6 sin6_peer = {};
156 struct sockaddr_in6 sin6;
157
158 init_sin6_address(&sin6);
159 init_sin6_address(&sin6_local);
160 init_sin6_address(&sin6_peer);
161
162 T_ASSERT_POSIX_SUCCESS(listen_fd = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
163
164 T_ASSERT_POSIX_SUCCESS(setsockopt(listen_fd, IPPROTO_IPV6, IPV6_V6ONLY, &off, sizeof(off)), NULL);
165
166 T_ASSERT_POSIX_SUCCESS(bind(listen_fd, (struct sockaddr *)&sin6, sizeof(sin6)), NULL);
167
168 socklen = sizeof(sin6);
169 T_ASSERT_POSIX_SUCCESS(getsockname(listen_fd, (struct sockaddr *)&sin6, &socklen), NULL);
170
171 T_LOG("listening on port: %u", ntohs(sin6.sin6_port));
172 sin6_to->sin6_port = sin6.sin6_port;
173
174 T_ASSERT_POSIX_SUCCESS(listen(listen_fd, 10), NULL);
175
176 T_ASSERT_POSIX_SUCCESS(setsockopt(client_fd, IPPROTO_TCP, TCP_CONNECTIONTIMEOUT, &val, sizeof(val)), NULL);
177
178 msg.msg_name = sin6_to;
179 msg.msg_namelen = sin6_to->sin6_len;
180 msg.msg_iov = &iovec;
181 msg.msg_iovlen = 1;
182 msg.msg_control = buf;
183 msg.msg_controllen = CMSG_SPACE(sizeof(int));
184
185 cmsghdrp = CMSG_FIRSTHDR(&msg);
186 cmsghdrp->cmsg_len = CMSG_LEN(sizeof(int));
187 cmsghdrp->cmsg_level = SOL_SOCKET;
188 cmsghdrp->cmsg_type = SCM_RIGHTS;
189
190 memcpy(CMSG_DATA(cmsghdrp), &client_fd, sizeof(client_fd));
191
192 (void)inet_ntop(AF_INET6, &sin6_to->sin6_addr, l_addr_str, sizeof(l_addr_str));
193
194 T_LOG("sendmsg with sin6_len: %u sin6_family: %u sin6_port: %u sin6_addr: %s expected_error: %d",
195 sin6_to->sin6_len, sin6_to->sin6_family, ntohs(sin6_to->sin6_port), l_addr_str, expected_error);
196
197 ssize_t retval = sendmsg(client_fd, &msg, 0);
198
199 if (expected_error == 0) {
200 T_EXPECT_POSIX_SUCCESS(retval, "sendmsg(client_fd, &msg, 0)");
201 } else {
202 T_EXPECT_POSIX_FAILURE(retval, expected_error, "sendmsg(client_fd, &msg, 0)");
203 }
204
205 if (retval == 0) {
206 socklen = sizeof(sin6_local);
207 T_ASSERT_POSIX_SUCCESS(getsockname(client_fd, (struct sockaddr *)&sin6_local, &socklen), NULL);
208 (void)inet_ntop(AF_INET6, &sin6_local.sin6_addr, l_addr_str, sizeof(l_addr_str));
209
210 socklen = sizeof(sin6_peer);
211 T_ASSERT_POSIX_SUCCESS(getpeername(client_fd, (struct sockaddr *)&sin6_peer, &socklen), NULL);
212 (void)inet_ntop(AF_INET6, &sin6_peer.sin6_addr, f_addr_str, sizeof(f_addr_str));
213
214 T_LOG("connected from %s:%u to %s:%u",
215 l_addr_str, ntohs(sin6_local.sin6_port),
216 f_addr_str, ntohs(sin6_peer.sin6_port));
217 }
218
219 T_ASSERT_POSIX_SUCCESS(close(listen_fd), NULL);
220
221 return 0;
222 }
223
224
225 T_DECL(tcp_send_implied_connect_ipv4_loopback, "TCP send implied connect with a IPv4 loopback address", T_META_TAG_VM_PREFERRED)
226 {
227 int s = -1;
228 struct sockaddr_in sin = {};
229
230 init_sin_address(&sin);
231 T_ASSERT_EQ(inet_pton(AF_INET, "127.0.0.1", &sin.sin_addr), 1, NULL);
232
233 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP), NULL);
234
235 T_EXPECT_NULL(tcp_send_implied_connect_v4(s, &sin, 0), NULL);
236
237 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
238 }
239
240
241 T_DECL(tcp_send_implied_connect_ipv4_multicast, "TCP send implied connect with an IPv4 multicast address", T_META_TAG_VM_PREFERRED)
242 {
243 int s = -1;
244 struct sockaddr_in sin = {};
245
246 init_sin_address(&sin);
247 T_ASSERT_EQ(inet_pton(AF_INET, "224.0.0.1", &sin.sin_addr), 1, NULL);
248
249 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP), NULL);
250
251 T_EXPECT_NULL(tcp_send_implied_connect_v4(s, &sin, EAFNOSUPPORT), NULL);
252
253 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
254 }
255
256 T_DECL(tcp_send_implied_connect_ipv4_broadcast, "TCP send implied connect with the IPv4 broadcast address", T_META_TAG_VM_PREFERRED)
257 {
258 int s = -1;
259 struct sockaddr_in sin = {};
260
261 init_sin_address(&sin);
262 T_ASSERT_EQ(inet_pton(AF_INET, "255.255.255.255", &sin.sin_addr), 1, NULL);
263
264 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP), NULL);
265
266 T_EXPECT_NULL(tcp_send_implied_connect_v4(s, &sin, EAFNOSUPPORT), NULL);
267
268 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
269 }
270
271 T_DECL(tcp_send_implied_connect_ipv4_null, "TCP send implied connect with the null IPv4 address", T_META_TAG_VM_PREFERRED)
272 {
273 int s = -1;
274 struct sockaddr_in sin = {};
275
276 init_sin_address(&sin);
277 T_ASSERT_EQ(inet_pton(AF_INET, "0.0.0.0", &sin.sin_addr), 1, NULL);
278
279 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP), NULL);
280
281 T_EXPECT_NULL(tcp_send_implied_connect_v4(s, &sin, 0), NULL);
282
283 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
284 }
285
286 T_DECL(tcp_send_implied_connect_ipv6_loopback, "TCP send implied connect with the IPv6 loopback address", T_META_TAG_VM_PREFERRED)
287 {
288 int s = -1;
289 struct sockaddr_in6 sin6 = {};
290
291 init_sin6_address(&sin6);
292 T_ASSERT_EQ(inet_pton(AF_INET6, "::1", &sin6.sin6_addr), 1, NULL);
293
294 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
295
296 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin6, 0), NULL);
297
298 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
299 }
300
301 T_DECL(tcp_send_implied_connect_ipv6_multicast, "TCP send implied connect with a IPv6 multicast address", T_META_TAG_VM_PREFERRED)
302 {
303 int s = -1;
304 struct sockaddr_in6 sin6 = {};
305
306 init_sin6_address(&sin6);
307 sin6.sin6_scope_id = if_nametoindex("lo0");
308 T_ASSERT_EQ(inet_pton(AF_INET6, "ff01::1", &sin6.sin6_addr), 1, NULL);
309
310 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
311
312 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin6, EAFNOSUPPORT), NULL);
313
314 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
315 }
316
317 T_DECL(tcp_send_implied_connect_null_ipv6, "TCP send implied connect with the IPv6 null address", T_META_TAG_VM_PREFERRED)
318 {
319 int s = -1;
320 struct sockaddr_in6 sin6 = {};
321
322 init_sin6_address(&sin6);
323 T_ASSERT_EQ(inet_pton(AF_INET6, "::", &sin6.sin6_addr), 1, NULL);
324
325 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
326
327 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin6, 0), NULL);
328
329 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
330 }
331
332 // panics without rdar://124092232
333 T_DECL(tcp_send_implied_connect_ipv4_loopback_via_ipv6_rdar_124092232, "TCP send implied v6 connect with IPv4 loopback", T_META_TAG_VM_PREFERRED)
334 {
335 int s = -1;
336 struct sockaddr_in sin = {0};
337
338 init_sin_address(&sin);
339 T_ASSERT_EQ(inet_pton(AF_INET, "127.0.0.1", &sin.sin_addr), 1, NULL);
340
341 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
342
343 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin, EAFNOSUPPORT), NULL);
344
345 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
346 }
347
348 T_DECL(tcp_send_implied_connect_ipv4_multicast_mapped_ipv6, "TCP send implied connect with IPv4 multicast mapped IPv6 address", T_META_TAG_VM_PREFERRED)
349 {
350 int s = -1;
351 struct sockaddr_in6 sin6 = {};
352
353 init_sin6_address(&sin6);
354 T_ASSERT_EQ(inet_pton(AF_INET6, "::ffff:224.0.0.1", &sin6.sin6_addr), 1, NULL);
355
356 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
357
358 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin6, EAFNOSUPPORT), NULL);
359
360 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
361 }
362
363 T_DECL(tcp_send_implied_connect_ipv4_broadcast_mapped_ipv6, "TCP send implied connect with IPv4 broadcast mapped IPv6 address", T_META_TAG_VM_PREFERRED)
364 {
365 int s = -1;
366 struct sockaddr_in6 sin6 = {};
367
368 init_sin6_address(&sin6);
369 T_ASSERT_EQ(inet_pton(AF_INET6, "::ffff:255.255.255.255", &sin6.sin6_addr), 1, NULL);
370
371 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
372
373 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin6, EAFNOSUPPORT), NULL);
374
375 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
376 }
377
378 T_DECL(tcp_send_implied_connect_ipv4_null_mapped_ipv6, "TCP send implied connect with IPv4 null mapped IPv6 address", T_META_TAG_VM_PREFERRED)
379 {
380 int s = -1;
381 struct sockaddr_in6 sin6 = {};
382
383 init_sin6_address(&sin6);
384 T_ASSERT_EQ(inet_pton(AF_INET6, "::ffff:0.0.0.0", &sin6.sin6_addr), 1, NULL);
385
386 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
387
388 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin6, 0), NULL);
389
390 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
391 }
392
393 T_DECL(tcp_send_implied_connect_ipv4_multicast_compatible_ipv6, "TCP send implied connect with IPv4 multicast compatible IPv6 address", T_META_TAG_VM_PREFERRED)
394 {
395 int s = -1;
396 struct sockaddr_in6 sin6 = {};
397
398 init_sin6_address(&sin6);
399 T_ASSERT_EQ(inet_pton(AF_INET6, "::224.0.0.1", &sin6.sin6_addr), 1, NULL);
400
401 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
402
403 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin6, EAFNOSUPPORT), NULL);
404
405 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
406 }
407
408 T_DECL(tcp_send_implied_connect_ipv4_broadcast_compatible_ipv6, "TCP send implied connect with IPv4 broadcast compatible IPv6 address", T_META_TAG_VM_PREFERRED)
409 {
410 int s = -1;
411 struct sockaddr_in6 sin6 = {};
412
413 init_sin6_address(&sin6);
414 T_ASSERT_EQ(inet_pton(AF_INET6, "::255.255.255.255", &sin6.sin6_addr), 1, NULL);
415
416 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
417
418 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin6, EAFNOSUPPORT), NULL);
419
420 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
421 }
422
423 T_DECL(tcp_send_implied_connect_ipv4_null_compatible_ipv6, "TCP send implied connect with IPv4 null compatible IPv6 address", T_META_TAG_VM_PREFERRED)
424 {
425 int s = -1;
426 struct sockaddr_in6 sin6 = {};
427
428 init_sin6_address(&sin6);
429 T_ASSERT_EQ(inet_pton(AF_INET6, "::0.0.0.0", &sin6.sin6_addr), 1, NULL);
430
431 T_ASSERT_POSIX_SUCCESS(s = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP), NULL);
432
433 T_EXPECT_NULL(tcp_send_implied_connect_v6(s, &sin6, 0), NULL);
434
435 T_ASSERT_POSIX_SUCCESS(close(s), NULL);
436 }
437