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