1 /* -*- Mode: c; tab-width: 8; indent-tabs-mode: 1; c-basic-offset: 8; -*- */
2
3 #include <darwintest.h>
4 #include <poll.h>
5 #include <sys/socket.h>
6 #include <unistd.h>
7 #include <netinet/in.h>
8 #include <arpa/inet.h>
9 #include <errno.h>
10
11 static int
sockv6_open(void)12 sockv6_open(void)
13 {
14 int s;
15
16 s = socket(AF_INET6, SOCK_DGRAM, 0);
17 T_QUIET;
18 T_ASSERT_POSIX_SUCCESS(s, "socket(AF_INET6, SOCK_DGRAM, 0)");
19 return s;
20 }
21
22 static int
sockv6_bind(int s,in_port_t port)23 sockv6_bind(int s, in_port_t port)
24 {
25 struct sockaddr_in6 sin6;
26
27 bzero(&sin6, sizeof(sin6));
28 sin6.sin6_len = sizeof(sin6);
29 sin6.sin6_family = AF_INET6;
30 sin6.sin6_port = port;
31 return bind(s, (const struct sockaddr *)&sin6, sizeof(sin6));
32 }
33
34 static void
sockv6_set_v6only(int s)35 sockv6_set_v6only(int s)
36 {
37 int on = 1;
38 int ret;
39
40 ret = setsockopt(s, IPPROTO_IPV6, IPV6_V6ONLY, &on, sizeof(on));
41 T_QUIET;
42 T_ASSERT_POSIX_SUCCESS(ret, "setsockopt(%d, IPV6_ONLY)", s);
43 }
44
45 static bool
alloc_and_bind_ports(in_port_t port_start,in_port_t port_end,int bind_attempts)46 alloc_and_bind_ports(in_port_t port_start, in_port_t port_end,
47 int bind_attempts)
48 {
49 int bound_count = 0;
50 bool success = true;
51
52 for (in_port_t i = port_start; success && i <= port_end; i++) {
53 int s6 = -1;
54 int s6_other = -1;
55 int ret;
56
57 s6 = sockv6_open();
58 sockv6_set_v6only(s6);
59 if (sockv6_bind(s6, i) != 0) {
60 /* find the next available port */
61 goto loop_done;
62 }
63 s6_other = sockv6_open();
64 ret = sockv6_bind(s6_other, i);
65 T_WITH_ERRNO;
66 T_QUIET;
67 T_ASSERT_TRUE(ret != 0, "socket %d bind %d", s6_other, i);
68 /*
69 * After bind fails, try binding to a different port.
70 * For non-root user, this will panic without the fix for
71 * <rdar://problem/35243417>.
72 */
73 if (sockv6_bind(s6_other, i + 1) == 0) {
74 bound_count++;
75 if (bound_count >= bind_attempts) {
76 break;
77 }
78 }
79 loop_done:
80 if (s6 >= 0) {
81 close(s6);
82 }
83 if (s6_other >= 0) {
84 close(s6_other);
85 }
86 }
87 T_ASSERT_TRUE(bound_count == bind_attempts,
88 "number of successful binds %d (out of %d)",
89 bound_count, bind_attempts);
90 return success;
91 }
92
93
94 T_DECL(socket_bind_35243417,
95 "bind IPv6 only UDP socket, then bind IPv6 socket.",
96 T_META_ASROOT(false),
97 T_META_CHECK_LEAKS(false),
98 T_META_TAG_VM_PREFERRED)
99 {
100 alloc_and_bind_ports(1, 65534, 10);
101 }
102
103 T_DECL(socket_bind_35243417_root,
104 "bind IPv6 only UDP socket, then bind IPv6 socket.",
105 T_META_ASROOT(true),
106 T_META_TAG_VM_PREFERRED)
107 {
108 alloc_and_bind_ports(1, 65534, 10);
109 }
110