1*94d3b452SApple OSS Distributions /* -*- compile-command: "xcrun --sdk iphoneos.internal make net_tuntests" -*- */
2*94d3b452SApple OSS Distributions
3*94d3b452SApple OSS Distributions #include <inttypes.h>
4*94d3b452SApple OSS Distributions #include <stdbool.h>
5*94d3b452SApple OSS Distributions #include <stdio.h>
6*94d3b452SApple OSS Distributions #include <stdlib.h>
7*94d3b452SApple OSS Distributions #include <time.h>
8*94d3b452SApple OSS Distributions #include <string.h>
9*94d3b452SApple OSS Distributions #include <unistd.h>
10*94d3b452SApple OSS Distributions #include <poll.h>
11*94d3b452SApple OSS Distributions #include <sys/types.h>
12*94d3b452SApple OSS Distributions #include <sys/event.h>
13*94d3b452SApple OSS Distributions #include <sys/time.h>
14*94d3b452SApple OSS Distributions #include <uuid/uuid.h>
15*94d3b452SApple OSS Distributions #include <arpa/inet.h>
16*94d3b452SApple OSS Distributions #include <sys/sysctl.h>
17*94d3b452SApple OSS Distributions #include <sys/kern_control.h>
18*94d3b452SApple OSS Distributions #include <sys/ioctl.h>
19*94d3b452SApple OSS Distributions #include <sys/socket.h>
20*94d3b452SApple OSS Distributions #include <sys/kern_control.h>
21*94d3b452SApple OSS Distributions #include <sys/sys_domain.h>
22*94d3b452SApple OSS Distributions
23*94d3b452SApple OSS Distributions #include <net/if.h>
24*94d3b452SApple OSS Distributions #include <net/if_ipsec.h>
25*94d3b452SApple OSS Distributions #include <net/if_utun.h>
26*94d3b452SApple OSS Distributions #include <netinet/in.h>
27*94d3b452SApple OSS Distributions #include <netinet/in_var.h>
28*94d3b452SApple OSS Distributions #include <net/pfkeyv2.h>
29*94d3b452SApple OSS Distributions #include <netinet6/ipsec.h>
30*94d3b452SApple OSS Distributions
31*94d3b452SApple OSS Distributions #include <darwintest.h>
32*94d3b452SApple OSS Distributions #include <darwintest_utils.h>
33*94d3b452SApple OSS Distributions
34*94d3b452SApple OSS Distributions #include <skywalk/os_skywalk_private.h> // for SK_FEATURE_*
35*94d3b452SApple OSS Distributions
36*94d3b452SApple OSS Distributions T_GLOBAL_META(T_META_NAMESPACE("xnu.net.tun"));
37*94d3b452SApple OSS Distributions
38*94d3b452SApple OSS Distributions /* Disable all these test until <rdar://problem/49124468> is fixed */
39*94d3b452SApple OSS Distributions T_GLOBAL_META(T_META_ENABLED(false));
40*94d3b452SApple OSS Distributions
41*94d3b452SApple OSS Distributions #if 0
42*94d3b452SApple OSS Distributions #undef T_QUIET
43*94d3b452SApple OSS Distributions #define T_QUIET
44*94d3b452SApple OSS Distributions #endif
45*94d3b452SApple OSS Distributions
46*94d3b452SApple OSS Distributions #if 0
47*94d3b452SApple OSS Distributions static void
48*94d3b452SApple OSS Distributions log_hexdump(const void *inp, size_t len)
49*94d3b452SApple OSS Distributions {
50*94d3b452SApple OSS Distributions unsigned i, off = 0;
51*94d3b452SApple OSS Distributions char buf[9 + 16 * 3 + 1];
52*94d3b452SApple OSS Distributions for (i = 0; i < len; i++) {
53*94d3b452SApple OSS Distributions if (i % 16 == 0) {
54*94d3b452SApple OSS Distributions off = (unsigned)snprintf(buf, sizeof(buf), "%08x:", i);
55*94d3b452SApple OSS Distributions }
56*94d3b452SApple OSS Distributions off += (unsigned)snprintf(buf + off, sizeof(buf) - off, " %02x", (((const uint8_t *)inp)[i]) & 0xff);
57*94d3b452SApple OSS Distributions if (i % 16 == 15) {
58*94d3b452SApple OSS Distributions T_LOG("%s", buf);
59*94d3b452SApple OSS Distributions }
60*94d3b452SApple OSS Distributions }
61*94d3b452SApple OSS Distributions if (len % 16) {
62*94d3b452SApple OSS Distributions T_LOG("%s", buf);
63*94d3b452SApple OSS Distributions }
64*94d3b452SApple OSS Distributions }
65*94d3b452SApple OSS Distributions #else
66*94d3b452SApple OSS Distributions static void
log_hexdump(const void * inp,size_t len)67*94d3b452SApple OSS Distributions log_hexdump(const void *inp, size_t len)
68*94d3b452SApple OSS Distributions {
69*94d3b452SApple OSS Distributions #pragma unused(inp, len)
70*94d3b452SApple OSS Distributions }
71*94d3b452SApple OSS Distributions #endif
72*94d3b452SApple OSS Distributions
73*94d3b452SApple OSS Distributions static bool
is_netagent_enabled(void)74*94d3b452SApple OSS Distributions is_netagent_enabled(void)
75*94d3b452SApple OSS Distributions {
76*94d3b452SApple OSS Distributions int enabled = 0;
77*94d3b452SApple OSS Distributions size_t len = sizeof(enabled);
78*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(sysctlbyname("net.link.generic.system.enable_netagent", &enabled, &len, NULL, 0), NULL);
79*94d3b452SApple OSS Distributions T_QUIET; T_ASSERT_EQ(len, sizeof(enabled), NULL);
80*94d3b452SApple OSS Distributions return enabled == 1;
81*94d3b452SApple OSS Distributions }
82*94d3b452SApple OSS Distributions
83*94d3b452SApple OSS Distributions static bool g_is_ipsec_test;
84*94d3b452SApple OSS Distributions static bool g_is_utun_test;
85*94d3b452SApple OSS Distributions static int g_OPT_ENABLE_NETIF = -1;
86*94d3b452SApple OSS Distributions static int g_OPT_ENABLE_FLOWSWITCH = -1;
87*94d3b452SApple OSS Distributions static int g_OPT_ENABLE_CHANNEL = -1;
88*94d3b452SApple OSS Distributions static int g_OPT_GET_CHANNEL_UUID = -1;
89*94d3b452SApple OSS Distributions static int g_OPT_IFNAME = -1;
90*94d3b452SApple OSS Distributions static char *g_CONTROL_NAME = NULL;
91*94d3b452SApple OSS Distributions
92*94d3b452SApple OSS Distributions static int create_tunsock_old(int enable_netif, int enable_flowswitch, int channel_count, uuid_t uuid[]);
93*94d3b452SApple OSS Distributions static int create_tunsock_new(int enable_netif, int enable_flowswitch, int channel_count, uuid_t uuid[]);
94*94d3b452SApple OSS Distributions static int (*create_tunsock)(int enable_netif, int enable_flowswitch, int channel_count, uuid_t uuid[]);
95*94d3b452SApple OSS Distributions
96*94d3b452SApple OSS Distributions static void
setup_ipsec_test(void)97*94d3b452SApple OSS Distributions setup_ipsec_test(void)
98*94d3b452SApple OSS Distributions {
99*94d3b452SApple OSS Distributions T_LOG("Configuring for ipsec tests");
100*94d3b452SApple OSS Distributions g_OPT_ENABLE_NETIF = IPSEC_OPT_ENABLE_NETIF;
101*94d3b452SApple OSS Distributions g_OPT_ENABLE_FLOWSWITCH = IPSEC_OPT_ENABLE_FLOWSWITCH;
102*94d3b452SApple OSS Distributions g_OPT_ENABLE_CHANNEL = IPSEC_OPT_ENABLE_CHANNEL;
103*94d3b452SApple OSS Distributions g_OPT_GET_CHANNEL_UUID = IPSEC_OPT_GET_CHANNEL_UUID;
104*94d3b452SApple OSS Distributions g_OPT_IFNAME = IPSEC_OPT_IFNAME;
105*94d3b452SApple OSS Distributions g_CONTROL_NAME = IPSEC_CONTROL_NAME;
106*94d3b452SApple OSS Distributions create_tunsock = create_tunsock_new;
107*94d3b452SApple OSS Distributions g_is_ipsec_test = true;
108*94d3b452SApple OSS Distributions }
109*94d3b452SApple OSS Distributions
110*94d3b452SApple OSS Distributions static void
setup_utun_test(void)111*94d3b452SApple OSS Distributions setup_utun_test(void)
112*94d3b452SApple OSS Distributions {
113*94d3b452SApple OSS Distributions T_LOG("Configuring for utun tests");
114*94d3b452SApple OSS Distributions g_OPT_ENABLE_NETIF = UTUN_OPT_ENABLE_NETIF;
115*94d3b452SApple OSS Distributions g_OPT_ENABLE_FLOWSWITCH = UTUN_OPT_ENABLE_FLOWSWITCH;
116*94d3b452SApple OSS Distributions g_OPT_ENABLE_CHANNEL = UTUN_OPT_ENABLE_CHANNEL;
117*94d3b452SApple OSS Distributions g_OPT_GET_CHANNEL_UUID = UTUN_OPT_GET_CHANNEL_UUID;
118*94d3b452SApple OSS Distributions g_OPT_IFNAME = UTUN_OPT_IFNAME;
119*94d3b452SApple OSS Distributions g_CONTROL_NAME = UTUN_CONTROL_NAME;
120*94d3b452SApple OSS Distributions create_tunsock = create_tunsock_old;
121*94d3b452SApple OSS Distributions g_is_utun_test = true;
122*94d3b452SApple OSS Distributions }
123*94d3b452SApple OSS Distributions
124*94d3b452SApple OSS Distributions static bool
setblocking(int s,bool blocking)125*94d3b452SApple OSS Distributions setblocking(int s, bool blocking)
126*94d3b452SApple OSS Distributions {
127*94d3b452SApple OSS Distributions int flags;
128*94d3b452SApple OSS Distributions bool ret;
129*94d3b452SApple OSS Distributions
130*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(flags = fcntl(s, F_GETFL, 0), NULL);
131*94d3b452SApple OSS Distributions
132*94d3b452SApple OSS Distributions ret = !(flags & O_NONBLOCK);
133*94d3b452SApple OSS Distributions
134*94d3b452SApple OSS Distributions if (blocking) {
135*94d3b452SApple OSS Distributions flags &= ~O_NONBLOCK;
136*94d3b452SApple OSS Distributions } else {
137*94d3b452SApple OSS Distributions flags |= O_NONBLOCK;
138*94d3b452SApple OSS Distributions }
139*94d3b452SApple OSS Distributions
140*94d3b452SApple OSS Distributions #if 0
141*94d3b452SApple OSS Distributions T_LOG("Setting fd %d from %s to %s\n",
142*94d3b452SApple OSS Distributions s, ret ? "blocking" : "nonblocking",
143*94d3b452SApple OSS Distributions blocking ? "blocking" : "nonblocking");
144*94d3b452SApple OSS Distributions #endif
145*94d3b452SApple OSS Distributions
146*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(flags = fcntl(s, F_SETFL, flags), NULL);
147*94d3b452SApple OSS Distributions
148*94d3b452SApple OSS Distributions return ret;
149*94d3b452SApple OSS Distributions }
150*94d3b452SApple OSS Distributions
151*94d3b452SApple OSS Distributions
152*94d3b452SApple OSS Distributions static void
check_enables(int tunsock,int enable_netif,int enable_flowswitch,int channel_count,uuid_t uuid[])153*94d3b452SApple OSS Distributions check_enables(int tunsock, int enable_netif, int enable_flowswitch, int channel_count, uuid_t uuid[])
154*94d3b452SApple OSS Distributions {
155*94d3b452SApple OSS Distributions int scratch;
156*94d3b452SApple OSS Distributions socklen_t scratchlen, uuidlen;
157*94d3b452SApple OSS Distributions uuid_t scratchuuid[channel_count];
158*94d3b452SApple OSS Distributions if (!uuid) {
159*94d3b452SApple OSS Distributions uuid = scratchuuid;
160*94d3b452SApple OSS Distributions }
161*94d3b452SApple OSS Distributions
162*94d3b452SApple OSS Distributions //T_LOG("checking tunsock %d", tunsock);
163*94d3b452SApple OSS Distributions
164*94d3b452SApple OSS Distributions if (g_is_ipsec_test && channel_count && !enable_netif) {
165*94d3b452SApple OSS Distributions /* Unfortunately, the connect incorrectly unwinds the bind if it get an error.
166*94d3b452SApple OSS Distributions * until that is fixed, expect EINVAL here
167*94d3b452SApple OSS Distributions */
168*94d3b452SApple OSS Distributions scratchlen = sizeof(scratch);
169*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_NETIF,
170*94d3b452SApple OSS Distributions &scratch, &scratchlen), EINVAL, NULL);
171*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
172*94d3b452SApple OSS Distributions &scratch, &scratchlen), EINVAL, NULL);
173*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
174*94d3b452SApple OSS Distributions &scratch, &scratchlen), EINVAL, NULL);
175*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
176*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
177*94d3b452SApple OSS Distributions }
178*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
179*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
180*94d3b452SApple OSS Distributions uuid, &uuidlen), EINVAL, NULL);
181*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
182*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_TRUE(uuid_is_null(uuid[i]), NULL);
183*94d3b452SApple OSS Distributions }
184*94d3b452SApple OSS Distributions return;
185*94d3b452SApple OSS Distributions }
186*94d3b452SApple OSS Distributions
187*94d3b452SApple OSS Distributions
188*94d3b452SApple OSS Distributions scratchlen = sizeof(scratch);
189*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_NETIF,
190*94d3b452SApple OSS Distributions &scratch, &scratchlen), NULL);
191*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)scratchlen, sizeof(scratch), NULL);
192*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ(scratch, enable_netif, NULL);
193*94d3b452SApple OSS Distributions
194*94d3b452SApple OSS Distributions scratchlen = sizeof(scratch);
195*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
196*94d3b452SApple OSS Distributions &scratch, &scratchlen), NULL);
197*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)scratchlen, sizeof(scratch), NULL);
198*94d3b452SApple OSS Distributions if (is_netagent_enabled()) {
199*94d3b452SApple OSS Distributions if (enable_netif) {
200*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ(scratch, enable_flowswitch, NULL);
201*94d3b452SApple OSS Distributions } else {
202*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ(scratch, 0, NULL);
203*94d3b452SApple OSS Distributions }
204*94d3b452SApple OSS Distributions } else {
205*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ(scratch, 0, NULL);
206*94d3b452SApple OSS Distributions }
207*94d3b452SApple OSS Distributions
208*94d3b452SApple OSS Distributions scratchlen = sizeof(scratch);
209*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
210*94d3b452SApple OSS Distributions &scratch, &scratchlen), NULL);
211*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)scratchlen, sizeof(scratch), NULL);
212*94d3b452SApple OSS Distributions if (g_is_ipsec_test && !enable_netif) {
213*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ(scratch, 0, NULL);
214*94d3b452SApple OSS Distributions } else {
215*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ(scratch, (int)channel_count, NULL);
216*94d3b452SApple OSS Distributions }
217*94d3b452SApple OSS Distributions
218*94d3b452SApple OSS Distributions if (scratch) {
219*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
220*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
221*94d3b452SApple OSS Distributions }
222*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
223*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
224*94d3b452SApple OSS Distributions uuid, &uuidlen), NULL);
225*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
226*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
227*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_FALSE(uuid_is_null(uuid[i]), NULL);
228*94d3b452SApple OSS Distributions }
229*94d3b452SApple OSS Distributions } else {
230*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
231*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
232*94d3b452SApple OSS Distributions }
233*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
234*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
235*94d3b452SApple OSS Distributions uuid, &uuidlen), ENXIO, NULL);
236*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
237*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
238*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_TRUE(uuid_is_null(uuid[i]), NULL);
239*94d3b452SApple OSS Distributions }
240*94d3b452SApple OSS Distributions }
241*94d3b452SApple OSS Distributions }
242*94d3b452SApple OSS Distributions
243*94d3b452SApple OSS Distributions static void
tunsock_get_ifname(int s,char ifname[IFXNAMSIZ])244*94d3b452SApple OSS Distributions tunsock_get_ifname(int s, char ifname[IFXNAMSIZ])
245*94d3b452SApple OSS Distributions {
246*94d3b452SApple OSS Distributions socklen_t optlen = IFXNAMSIZ;
247*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_ASSERT_POSIX_ZERO(getsockopt(s, SYSPROTO_CONTROL, g_OPT_IFNAME, ifname, &optlen), NULL);
248*94d3b452SApple OSS Distributions T_QUIET; T_ASSERT_TRUE(optlen > 0, NULL);
249*94d3b452SApple OSS Distributions T_QUIET; T_ASSERT_TRUE(ifname[optlen - 1] == '\0', NULL);
250*94d3b452SApple OSS Distributions T_QUIET; T_ASSERT_TRUE(strlen(ifname) + 1 == optlen, "got ifname \"%s\" len %zd expected %u", ifname, strlen(ifname), optlen);
251*94d3b452SApple OSS Distributions }
252*94d3b452SApple OSS Distributions
253*94d3b452SApple OSS Distributions static short
ifnet_get_flags(int s,const char ifname[IFNAMSIZ])254*94d3b452SApple OSS Distributions ifnet_get_flags(int s, const char ifname[IFNAMSIZ])
255*94d3b452SApple OSS Distributions {
256*94d3b452SApple OSS Distributions struct ifreq ifr;
257*94d3b452SApple OSS Distributions memset(&ifr, 0, sizeof(ifr));
258*94d3b452SApple OSS Distributions strlcpy(ifr.ifr_name, ifname, sizeof(ifr.ifr_name));
259*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(ioctl(s, SIOCGIFFLAGS, (caddr_t)&ifr), NULL);
260*94d3b452SApple OSS Distributions return ifr.ifr_flags;
261*94d3b452SApple OSS Distributions }
262*94d3b452SApple OSS Distributions
263*94d3b452SApple OSS Distributions static void
ifnet_add_addr4(const char ifname[IFNAMSIZ],struct in_addr * addr,struct in_addr * mask,struct in_addr * broadaddr)264*94d3b452SApple OSS Distributions ifnet_add_addr4(const char ifname[IFNAMSIZ], struct in_addr *addr, struct in_addr *mask, struct in_addr *broadaddr)
265*94d3b452SApple OSS Distributions {
266*94d3b452SApple OSS Distributions struct sockaddr_in *sin;
267*94d3b452SApple OSS Distributions struct in_aliasreq ifra;
268*94d3b452SApple OSS Distributions int s;
269*94d3b452SApple OSS Distributions
270*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(s = socket(AF_INET, SOCK_DGRAM, 0), NULL);
271*94d3b452SApple OSS Distributions
272*94d3b452SApple OSS Distributions memset(&ifra, 0, sizeof(ifra));
273*94d3b452SApple OSS Distributions strlcpy(ifra.ifra_name, ifname, sizeof(ifra.ifra_name));
274*94d3b452SApple OSS Distributions
275*94d3b452SApple OSS Distributions if (addr != NULL) {
276*94d3b452SApple OSS Distributions sin = &ifra.ifra_addr;
277*94d3b452SApple OSS Distributions sin->sin_len = sizeof(*sin);
278*94d3b452SApple OSS Distributions sin->sin_family = AF_INET;
279*94d3b452SApple OSS Distributions sin->sin_addr = *addr;
280*94d3b452SApple OSS Distributions }
281*94d3b452SApple OSS Distributions
282*94d3b452SApple OSS Distributions if (mask != NULL) {
283*94d3b452SApple OSS Distributions sin = &ifra.ifra_mask;
284*94d3b452SApple OSS Distributions sin->sin_len = sizeof(*sin);
285*94d3b452SApple OSS Distributions sin->sin_family = AF_INET;
286*94d3b452SApple OSS Distributions sin->sin_addr = *mask;
287*94d3b452SApple OSS Distributions }
288*94d3b452SApple OSS Distributions
289*94d3b452SApple OSS Distributions if (broadaddr != NULL || (addr != NULL &&
290*94d3b452SApple OSS Distributions (ifnet_get_flags(s, ifname) & IFF_POINTOPOINT) != 0)) {
291*94d3b452SApple OSS Distributions sin = &ifra.ifra_broadaddr;
292*94d3b452SApple OSS Distributions sin->sin_len = sizeof(*sin);
293*94d3b452SApple OSS Distributions sin->sin_family = AF_INET;
294*94d3b452SApple OSS Distributions sin->sin_addr = (broadaddr != NULL) ? *broadaddr : *addr;
295*94d3b452SApple OSS Distributions }
296*94d3b452SApple OSS Distributions
297*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(ioctl(s, SIOCAIFADDR, &ifra), NULL);
298*94d3b452SApple OSS Distributions
299*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(s), NULL);
300*94d3b452SApple OSS Distributions }
301*94d3b452SApple OSS Distributions
302*94d3b452SApple OSS Distributions static int g_pfkeyso = -1;
303*94d3b452SApple OSS Distributions static struct in_addr g_addr1, g_addr2;
304*94d3b452SApple OSS Distributions
305*94d3b452SApple OSS Distributions static void
create_sa(const char ifname[IFXNAMSIZ],uint8_t type,uint32_t spi,struct in_addr * src,struct in_addr * dst)306*94d3b452SApple OSS Distributions create_sa(const char ifname[IFXNAMSIZ], uint8_t type, uint32_t spi, struct in_addr *src, struct in_addr *dst)
307*94d3b452SApple OSS Distributions {
308*94d3b452SApple OSS Distributions if (g_pfkeyso == -1) {
309*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(g_pfkeyso = socket(PF_KEY, SOCK_RAW, PF_KEY_V2), NULL);
310*94d3b452SApple OSS Distributions }
311*94d3b452SApple OSS Distributions
312*94d3b452SApple OSS Distributions /*
313*94d3b452SApple OSS Distributions * <base, SA, (lifetime(HS),) address(SD), (address(P),)
314*94d3b452SApple OSS Distributions * key(AE), (identity(SD),) (sensitivity)>
315*94d3b452SApple OSS Distributions */
316*94d3b452SApple OSS Distributions
317*94d3b452SApple OSS Distributions struct {
318*94d3b452SApple OSS Distributions struct sadb_msg msg __attribute((aligned(sizeof(uint64_t))));
319*94d3b452SApple OSS Distributions struct sadb_key key __attribute((aligned(sizeof(uint64_t))));
320*94d3b452SApple OSS Distributions struct sadb_sa sa __attribute((aligned(sizeof(uint64_t))));
321*94d3b452SApple OSS Distributions struct sadb_x_sa2 sa2 __attribute((aligned(sizeof(uint64_t))));
322*94d3b452SApple OSS Distributions struct sadb_x_ipsecif ipsecif __attribute((aligned(sizeof(uint64_t))));
323*94d3b452SApple OSS Distributions struct {
324*94d3b452SApple OSS Distributions struct sadb_address addr __attribute((aligned(sizeof(uint64_t))));
325*94d3b452SApple OSS Distributions struct sockaddr_in saddr __attribute((aligned(sizeof(uint64_t))));
326*94d3b452SApple OSS Distributions } src;
327*94d3b452SApple OSS Distributions struct {
328*94d3b452SApple OSS Distributions struct sadb_address addr __attribute((aligned(sizeof(uint64_t))));
329*94d3b452SApple OSS Distributions struct sockaddr_in saddr __attribute((aligned(sizeof(uint64_t))));
330*94d3b452SApple OSS Distributions } dst;
331*94d3b452SApple OSS Distributions } addcmd;
332*94d3b452SApple OSS Distributions
333*94d3b452SApple OSS Distributions memset(&addcmd, 0, sizeof(addcmd));
334*94d3b452SApple OSS Distributions
335*94d3b452SApple OSS Distributions addcmd.msg.sadb_msg_version = PF_KEY_V2;
336*94d3b452SApple OSS Distributions addcmd.msg.sadb_msg_type = type;
337*94d3b452SApple OSS Distributions addcmd.msg.sadb_msg_errno = 0;
338*94d3b452SApple OSS Distributions addcmd.msg.sadb_msg_satype = SADB_SATYPE_ESP;
339*94d3b452SApple OSS Distributions addcmd.msg.sadb_msg_len = PFKEY_UNIT64(sizeof(addcmd));
340*94d3b452SApple OSS Distributions addcmd.msg.sadb_msg_reserved = 0;
341*94d3b452SApple OSS Distributions addcmd.msg.sadb_msg_seq = 0;
342*94d3b452SApple OSS Distributions addcmd.msg.sadb_msg_pid = (unsigned)getpid();
343*94d3b452SApple OSS Distributions
344*94d3b452SApple OSS Distributions addcmd.key.sadb_key_len = PFKEY_UNIT64(sizeof(addcmd.key));
345*94d3b452SApple OSS Distributions addcmd.key.sadb_key_exttype = SADB_EXT_KEY_ENCRYPT;
346*94d3b452SApple OSS Distributions addcmd.key.sadb_key_bits = 0;
347*94d3b452SApple OSS Distributions addcmd.key.sadb_key_reserved = 0;
348*94d3b452SApple OSS Distributions
349*94d3b452SApple OSS Distributions addcmd.sa.sadb_sa_len = PFKEY_UNIT64(sizeof(addcmd.sa));
350*94d3b452SApple OSS Distributions addcmd.sa.sadb_sa_exttype = SADB_EXT_SA;
351*94d3b452SApple OSS Distributions addcmd.sa.sadb_sa_spi = htonl(spi);
352*94d3b452SApple OSS Distributions addcmd.sa.sadb_sa_replay = 0;
353*94d3b452SApple OSS Distributions addcmd.sa.sadb_sa_state = 0;
354*94d3b452SApple OSS Distributions addcmd.sa.sadb_sa_auth = SADB_AALG_NONE;
355*94d3b452SApple OSS Distributions addcmd.sa.sadb_sa_encrypt = SADB_EALG_NULL;
356*94d3b452SApple OSS Distributions addcmd.sa.sadb_sa_flags = SADB_X_EXT_CYCSEQ;
357*94d3b452SApple OSS Distributions
358*94d3b452SApple OSS Distributions addcmd.sa2.sadb_x_sa2_len = PFKEY_UNIT64(sizeof(addcmd.sa2));
359*94d3b452SApple OSS Distributions addcmd.sa2.sadb_x_sa2_exttype = SADB_X_EXT_SA2;
360*94d3b452SApple OSS Distributions addcmd.sa2.sadb_x_sa2_mode = IPSEC_MODE_ANY;
361*94d3b452SApple OSS Distributions addcmd.sa2.sadb_x_sa2_alwaysexpire = 1;
362*94d3b452SApple OSS Distributions addcmd.sa2.sadb_x_sa2_flags = SADB_X_EXT_SA2_DELETE_ON_DETACH;
363*94d3b452SApple OSS Distributions addcmd.sa2.sadb_x_sa2_sequence = 0;
364*94d3b452SApple OSS Distributions addcmd.sa2.sadb_x_sa2_reqid = 0;
365*94d3b452SApple OSS Distributions
366*94d3b452SApple OSS Distributions addcmd.ipsecif.sadb_x_ipsecif_len = PFKEY_UNIT64(sizeof(addcmd.ipsecif));
367*94d3b452SApple OSS Distributions addcmd.ipsecif.sadb_x_ipsecif_exttype = SADB_X_EXT_IPSECIF;
368*94d3b452SApple OSS Distributions memset(addcmd.ipsecif.sadb_x_ipsecif_internal_if, 0, sizeof(addcmd.ipsecif.sadb_x_ipsecif_internal_if));
369*94d3b452SApple OSS Distributions memset(addcmd.ipsecif.sadb_x_ipsecif_outgoing_if, 0, sizeof(addcmd.ipsecif.sadb_x_ipsecif_outgoing_if));
370*94d3b452SApple OSS Distributions strlcpy(addcmd.ipsecif.sadb_x_ipsecif_ipsec_if, ifname, sizeof(addcmd.ipsecif.sadb_x_ipsecif_ipsec_if));
371*94d3b452SApple OSS Distributions addcmd.ipsecif.sadb_x_ipsecif_init_disabled = 0;
372*94d3b452SApple OSS Distributions addcmd.ipsecif.reserved = 0;
373*94d3b452SApple OSS Distributions
374*94d3b452SApple OSS Distributions addcmd.src.addr.sadb_address_len = PFKEY_UNIT64(sizeof(addcmd.src));
375*94d3b452SApple OSS Distributions addcmd.src.addr.sadb_address_exttype = SADB_EXT_ADDRESS_SRC;
376*94d3b452SApple OSS Distributions addcmd.src.addr.sadb_address_proto = IPSEC_ULPROTO_ANY;
377*94d3b452SApple OSS Distributions addcmd.src.addr.sadb_address_prefixlen = sizeof(struct in_addr) << 3; //XXX Why?
378*94d3b452SApple OSS Distributions addcmd.src.addr.sadb_address_reserved = 0;
379*94d3b452SApple OSS Distributions addcmd.src.saddr.sin_len = sizeof(addcmd.src.saddr);
380*94d3b452SApple OSS Distributions addcmd.src.saddr.sin_family = AF_INET;
381*94d3b452SApple OSS Distributions addcmd.src.saddr.sin_port = htons(0);
382*94d3b452SApple OSS Distributions addcmd.src.saddr.sin_addr = *src;
383*94d3b452SApple OSS Distributions
384*94d3b452SApple OSS Distributions addcmd.dst.addr.sadb_address_len = PFKEY_UNIT64(sizeof(addcmd.dst));
385*94d3b452SApple OSS Distributions addcmd.dst.addr.sadb_address_exttype = SADB_EXT_ADDRESS_DST;
386*94d3b452SApple OSS Distributions addcmd.dst.addr.sadb_address_proto = IPSEC_ULPROTO_ANY;
387*94d3b452SApple OSS Distributions addcmd.dst.addr.sadb_address_prefixlen = sizeof(struct in_addr) << 3; //XXX Why?
388*94d3b452SApple OSS Distributions addcmd.dst.addr.sadb_address_reserved = 0;
389*94d3b452SApple OSS Distributions addcmd.dst.saddr.sin_len = sizeof(addcmd.dst.saddr);
390*94d3b452SApple OSS Distributions addcmd.dst.saddr.sin_family = AF_INET;
391*94d3b452SApple OSS Distributions addcmd.dst.saddr.sin_port = htons(0);
392*94d3b452SApple OSS Distributions addcmd.dst.saddr.sin_addr = *dst;
393*94d3b452SApple OSS Distributions
394*94d3b452SApple OSS Distributions log_hexdump(&addcmd, sizeof(addcmd));
395*94d3b452SApple OSS Distributions
396*94d3b452SApple OSS Distributions ssize_t slen;
397*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(slen = send(g_pfkeyso, &addcmd, sizeof(addcmd), 0), NULL);
398*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ(slen, (ssize_t)sizeof(addcmd), NULL);
399*94d3b452SApple OSS Distributions }
400*94d3b452SApple OSS Distributions
401*94d3b452SApple OSS Distributions /* This version of the test expects channels to be enabled after connect.
402*94d3b452SApple OSS Distributions * Once the utun driver is converted, switch to create_tunsock_new
403*94d3b452SApple OSS Distributions */
404*94d3b452SApple OSS Distributions static int
create_tunsock_old(int enable_netif,int enable_flowswitch,int channel_count,uuid_t uuid[])405*94d3b452SApple OSS Distributions create_tunsock_old(int enable_netif, int enable_flowswitch, int channel_count, uuid_t uuid[])
406*94d3b452SApple OSS Distributions {
407*94d3b452SApple OSS Distributions int tunsock;
408*94d3b452SApple OSS Distributions struct ctl_info kernctl_info;
409*94d3b452SApple OSS Distributions struct sockaddr_ctl kernctl_addr;
410*94d3b452SApple OSS Distributions uuid_t scratchuuid[channel_count];
411*94d3b452SApple OSS Distributions if (!uuid) {
412*94d3b452SApple OSS Distributions uuid = scratchuuid;
413*94d3b452SApple OSS Distributions }
414*94d3b452SApple OSS Distributions socklen_t uuidlen;
415*94d3b452SApple OSS Distributions
416*94d3b452SApple OSS Distributions startover:
417*94d3b452SApple OSS Distributions
418*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(tunsock = socket(PF_SYSTEM, SOCK_DGRAM, SYSPROTO_CONTROL), NULL);
419*94d3b452SApple OSS Distributions
420*94d3b452SApple OSS Distributions memset(&kernctl_info, 0, sizeof(kernctl_info));
421*94d3b452SApple OSS Distributions strlcpy(kernctl_info.ctl_name, g_CONTROL_NAME, sizeof(kernctl_info.ctl_name));
422*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(ioctl(tunsock, CTLIOCGINFO, &kernctl_info), NULL);
423*94d3b452SApple OSS Distributions
424*94d3b452SApple OSS Distributions memset(&kernctl_addr, 0, sizeof(kernctl_addr));
425*94d3b452SApple OSS Distributions kernctl_addr.sc_len = sizeof(kernctl_addr);
426*94d3b452SApple OSS Distributions kernctl_addr.sc_family = AF_SYSTEM;
427*94d3b452SApple OSS Distributions kernctl_addr.ss_sysaddr = AF_SYS_CONTROL;
428*94d3b452SApple OSS Distributions kernctl_addr.sc_id = kernctl_info.ctl_id;
429*94d3b452SApple OSS Distributions kernctl_addr.sc_unit = 0;
430*94d3b452SApple OSS Distributions
431*94d3b452SApple OSS Distributions T_LOG("%s: enable_netif = %d, enable_flowswitch = %d, channel_count = %d",
432*94d3b452SApple OSS Distributions __func__, enable_netif, enable_flowswitch, channel_count);
433*94d3b452SApple OSS Distributions
434*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_NETIF,
435*94d3b452SApple OSS Distributions &enable_netif, sizeof(enable_netif)), EINVAL, NULL);
436*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
437*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), EINVAL, NULL);
438*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
439*94d3b452SApple OSS Distributions &channel_count, sizeof(channel_count)), EINVAL, NULL);
440*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
441*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
442*94d3b452SApple OSS Distributions }
443*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
444*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
445*94d3b452SApple OSS Distributions uuid, &uuidlen), EINVAL, NULL);
446*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
447*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
448*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_TRUE(uuid_is_null(uuid[i]), NULL);
449*94d3b452SApple OSS Distributions }
450*94d3b452SApple OSS Distributions
451*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(bind(tunsock, (struct sockaddr *)&kernctl_addr, sizeof(kernctl_addr)), NULL);
452*94d3b452SApple OSS Distributions
453*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_NETIF,
454*94d3b452SApple OSS Distributions &enable_netif, sizeof(enable_netif)), NULL);
455*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
456*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), EINVAL, NULL);
457*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
458*94d3b452SApple OSS Distributions &channel_count, sizeof(channel_count)), EINVAL, NULL);
459*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
460*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
461*94d3b452SApple OSS Distributions }
462*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
463*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
464*94d3b452SApple OSS Distributions uuid, &uuidlen), ENXIO, NULL);
465*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
466*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
467*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_TRUE(uuid_is_null(uuid[i]), NULL);
468*94d3b452SApple OSS Distributions }
469*94d3b452SApple OSS Distributions
470*94d3b452SApple OSS Distributions int error = connect(tunsock, (struct sockaddr *)&kernctl_addr, sizeof(kernctl_addr));
471*94d3b452SApple OSS Distributions if (error == -1 && errno == EBUSY) {
472*94d3b452SApple OSS Distributions /* XXX remove this retry nonsense when this is fixed:
473*94d3b452SApple OSS Distributions * <rdar://problem/37340313> creating an interface without specifying specific interface name should not return EBUSY
474*94d3b452SApple OSS Distributions */
475*94d3b452SApple OSS Distributions close(tunsock);
476*94d3b452SApple OSS Distributions T_LOG("connect got EBUSY, sleeping 1 second before retry");
477*94d3b452SApple OSS Distributions sleep(1);
478*94d3b452SApple OSS Distributions goto startover;
479*94d3b452SApple OSS Distributions }
480*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(error, "connect()");
481*94d3b452SApple OSS Distributions
482*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_NETIF,
483*94d3b452SApple OSS Distributions &enable_netif, sizeof(enable_netif)), EINVAL, NULL);
484*94d3b452SApple OSS Distributions
485*94d3b452SApple OSS Distributions if (is_netagent_enabled()) {
486*94d3b452SApple OSS Distributions if (enable_netif) {
487*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
488*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), NULL);
489*94d3b452SApple OSS Distributions } else {
490*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
491*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), ENOENT, NULL);
492*94d3b452SApple OSS Distributions }
493*94d3b452SApple OSS Distributions } else {
494*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
495*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), ENOTSUP, NULL);
496*94d3b452SApple OSS Distributions }
497*94d3b452SApple OSS Distributions
498*94d3b452SApple OSS Distributions if (channel_count) {
499*94d3b452SApple OSS Distributions if (g_is_ipsec_test && !enable_netif) {
500*94d3b452SApple OSS Distributions /* ipsec doesn't support channels without a netif */
501*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
502*94d3b452SApple OSS Distributions &channel_count, sizeof(channel_count)), EOPNOTSUPP, NULL);
503*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
504*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
505*94d3b452SApple OSS Distributions }
506*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
507*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
508*94d3b452SApple OSS Distributions uuid, &uuidlen), ENXIO, NULL);
509*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
510*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
511*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_TRUE(uuid_is_null(uuid[i]), NULL);
512*94d3b452SApple OSS Distributions }
513*94d3b452SApple OSS Distributions } else {
514*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
515*94d3b452SApple OSS Distributions &channel_count, sizeof(channel_count)), NULL);
516*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
517*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
518*94d3b452SApple OSS Distributions }
519*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
520*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
521*94d3b452SApple OSS Distributions uuid, &uuidlen), NULL);
522*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
523*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
524*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_FALSE(uuid_is_null(uuid[i]), NULL);
525*94d3b452SApple OSS Distributions }
526*94d3b452SApple OSS Distributions }
527*94d3b452SApple OSS Distributions } else {
528*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
529*94d3b452SApple OSS Distributions &channel_count, sizeof(channel_count)), ENXIO, NULL);
530*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
531*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
532*94d3b452SApple OSS Distributions }
533*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
534*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
535*94d3b452SApple OSS Distributions uuid, &uuidlen), ENXIO, NULL);
536*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
537*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
538*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_TRUE(uuid_is_null(uuid[i]), NULL);
539*94d3b452SApple OSS Distributions }
540*94d3b452SApple OSS Distributions }
541*94d3b452SApple OSS Distributions
542*94d3b452SApple OSS Distributions check_enables(tunsock, enable_netif, enable_flowswitch, channel_count, uuid);
543*94d3b452SApple OSS Distributions
544*94d3b452SApple OSS Distributions //T_LOG("Returning tunsock %d", tunsock);
545*94d3b452SApple OSS Distributions
546*94d3b452SApple OSS Distributions return tunsock;
547*94d3b452SApple OSS Distributions }
548*94d3b452SApple OSS Distributions
549*94d3b452SApple OSS Distributions /* This version of the test expects channels to be enabled before connect
550*94d3b452SApple OSS Distributions * Once the utun driver is converted, rename this to just create_tunsock
551*94d3b452SApple OSS Distributions */
552*94d3b452SApple OSS Distributions static int
create_tunsock_new(int enable_netif,int enable_flowswitch,int channel_count,uuid_t uuid[])553*94d3b452SApple OSS Distributions create_tunsock_new(int enable_netif, int enable_flowswitch, int channel_count, uuid_t uuid[])
554*94d3b452SApple OSS Distributions {
555*94d3b452SApple OSS Distributions int tunsock;
556*94d3b452SApple OSS Distributions struct ctl_info kernctl_info;
557*94d3b452SApple OSS Distributions struct sockaddr_ctl kernctl_addr;
558*94d3b452SApple OSS Distributions uuid_t scratchuuid[channel_count];
559*94d3b452SApple OSS Distributions if (!uuid) {
560*94d3b452SApple OSS Distributions uuid = scratchuuid;
561*94d3b452SApple OSS Distributions }
562*94d3b452SApple OSS Distributions socklen_t uuidlen;
563*94d3b452SApple OSS Distributions
564*94d3b452SApple OSS Distributions startover:
565*94d3b452SApple OSS Distributions
566*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(tunsock = socket(PF_SYSTEM, SOCK_DGRAM, SYSPROTO_CONTROL), NULL);
567*94d3b452SApple OSS Distributions
568*94d3b452SApple OSS Distributions memset(&kernctl_info, 0, sizeof(kernctl_info));
569*94d3b452SApple OSS Distributions strlcpy(kernctl_info.ctl_name, g_CONTROL_NAME, sizeof(kernctl_info.ctl_name));
570*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(ioctl(tunsock, CTLIOCGINFO, &kernctl_info), NULL);
571*94d3b452SApple OSS Distributions
572*94d3b452SApple OSS Distributions memset(&kernctl_addr, 0, sizeof(kernctl_addr));
573*94d3b452SApple OSS Distributions kernctl_addr.sc_len = sizeof(kernctl_addr);
574*94d3b452SApple OSS Distributions kernctl_addr.sc_family = AF_SYSTEM;
575*94d3b452SApple OSS Distributions kernctl_addr.ss_sysaddr = AF_SYS_CONTROL;
576*94d3b452SApple OSS Distributions kernctl_addr.sc_id = kernctl_info.ctl_id;
577*94d3b452SApple OSS Distributions kernctl_addr.sc_unit = 0;
578*94d3b452SApple OSS Distributions
579*94d3b452SApple OSS Distributions T_LOG("%s: enable_netif = %d, enable_flowswitch = %d, channel_count = %d",
580*94d3b452SApple OSS Distributions __func__, enable_netif, enable_flowswitch, channel_count);
581*94d3b452SApple OSS Distributions
582*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_NETIF,
583*94d3b452SApple OSS Distributions &enable_netif, sizeof(enable_netif)), EINVAL, NULL);
584*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
585*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), EINVAL, NULL);
586*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
587*94d3b452SApple OSS Distributions &channel_count, sizeof(channel_count)), EINVAL, NULL);
588*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
589*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
590*94d3b452SApple OSS Distributions }
591*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
592*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
593*94d3b452SApple OSS Distributions uuid, &uuidlen), EINVAL, NULL);
594*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
595*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
596*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_TRUE(uuid_is_null(uuid[i]), NULL);
597*94d3b452SApple OSS Distributions }
598*94d3b452SApple OSS Distributions
599*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(bind(tunsock, (struct sockaddr *)&kernctl_addr, sizeof(kernctl_addr)), NULL);
600*94d3b452SApple OSS Distributions
601*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_NETIF,
602*94d3b452SApple OSS Distributions &enable_netif, sizeof(enable_netif)), NULL);
603*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
604*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), EINVAL, NULL);
605*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
606*94d3b452SApple OSS Distributions &channel_count, sizeof(channel_count)), NULL);
607*94d3b452SApple OSS Distributions
608*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
609*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
610*94d3b452SApple OSS Distributions }
611*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
612*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
613*94d3b452SApple OSS Distributions uuid, &uuidlen), ENXIO, NULL);
614*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
615*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
616*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_TRUE(uuid_is_null(uuid[i]), NULL);
617*94d3b452SApple OSS Distributions }
618*94d3b452SApple OSS Distributions
619*94d3b452SApple OSS Distributions int error = connect(tunsock, (struct sockaddr *)&kernctl_addr, sizeof(kernctl_addr));
620*94d3b452SApple OSS Distributions if (error == -1 && errno == EBUSY) {
621*94d3b452SApple OSS Distributions /* XXX remove this retry nonsense when this is fixed:
622*94d3b452SApple OSS Distributions * <rdar://problem/37340313> creating an interface without specifying specific interface name should not return EBUSY
623*94d3b452SApple OSS Distributions */
624*94d3b452SApple OSS Distributions close(tunsock);
625*94d3b452SApple OSS Distributions T_LOG("connect got EBUSY, sleeping 1 second before retry");
626*94d3b452SApple OSS Distributions sleep(1);
627*94d3b452SApple OSS Distributions goto startover;
628*94d3b452SApple OSS Distributions }
629*94d3b452SApple OSS Distributions if (g_is_ipsec_test && channel_count && !enable_netif) {
630*94d3b452SApple OSS Distributions /* ipsec doesn't support channels without a netif */
631*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(error, ENOTSUP, "connect() == -1 && errno == ENOTSUP");
632*94d3b452SApple OSS Distributions } else {
633*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(error, "connect() == 0");
634*94d3b452SApple OSS Distributions }
635*94d3b452SApple OSS Distributions
636*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_NETIF,
637*94d3b452SApple OSS Distributions &enable_netif, sizeof(enable_netif)), EINVAL, NULL);
638*94d3b452SApple OSS Distributions
639*94d3b452SApple OSS Distributions if (g_is_ipsec_test && channel_count && !enable_netif) {
640*94d3b452SApple OSS Distributions /* Connect failed above, so we get EINVAL */
641*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
642*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), EINVAL, NULL);
643*94d3b452SApple OSS Distributions } else {
644*94d3b452SApple OSS Distributions if (is_netagent_enabled()) {
645*94d3b452SApple OSS Distributions if (enable_netif) {
646*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
647*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), NULL);
648*94d3b452SApple OSS Distributions } else {
649*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
650*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), ENOENT, NULL);
651*94d3b452SApple OSS Distributions }
652*94d3b452SApple OSS Distributions } else {
653*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_FLOWSWITCH,
654*94d3b452SApple OSS Distributions &enable_flowswitch, sizeof(enable_flowswitch)), ENOTSUP, NULL);
655*94d3b452SApple OSS Distributions }
656*94d3b452SApple OSS Distributions }
657*94d3b452SApple OSS Distributions
658*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(setsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_ENABLE_CHANNEL,
659*94d3b452SApple OSS Distributions &channel_count, sizeof(channel_count)), EINVAL, NULL);
660*94d3b452SApple OSS Distributions
661*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
662*94d3b452SApple OSS Distributions uuid_clear(uuid[i]);
663*94d3b452SApple OSS Distributions }
664*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
665*94d3b452SApple OSS Distributions if (!channel_count || (g_is_ipsec_test && channel_count && !enable_netif)) {
666*94d3b452SApple OSS Distributions /* ipsec doesn't support channels without a netif */
667*94d3b452SApple OSS Distributions if (g_is_ipsec_test && channel_count && !enable_netif) {
668*94d3b452SApple OSS Distributions /* Unfortunately, the connect incorrectly unwinds the bind if it get an error.
669*94d3b452SApple OSS Distributions * until that is fixed, expect EINVAL here
670*94d3b452SApple OSS Distributions */
671*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
672*94d3b452SApple OSS Distributions uuid, &uuidlen), EINVAL, NULL);
673*94d3b452SApple OSS Distributions } else {
674*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
675*94d3b452SApple OSS Distributions uuid, &uuidlen), ENXIO, NULL);
676*94d3b452SApple OSS Distributions }
677*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
678*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
679*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_TRUE(uuid_is_null(uuid[i]), NULL);
680*94d3b452SApple OSS Distributions }
681*94d3b452SApple OSS Distributions } else {
682*94d3b452SApple OSS Distributions uuidlen = sizeof(uuid_t) * (unsigned int)channel_count;
683*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(getsockopt(tunsock, SYSPROTO_CONTROL, g_OPT_GET_CHANNEL_UUID,
684*94d3b452SApple OSS Distributions uuid, &uuidlen), NULL);
685*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG((unsigned long)uuidlen, sizeof(uuid_t) * (unsigned long)channel_count, NULL);
686*94d3b452SApple OSS Distributions for (int i = 0; i < channel_count; i++) {
687*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_FALSE(uuid_is_null(uuid[i]), NULL);
688*94d3b452SApple OSS Distributions }
689*94d3b452SApple OSS Distributions }
690*94d3b452SApple OSS Distributions
691*94d3b452SApple OSS Distributions check_enables(tunsock, enable_netif, enable_flowswitch, channel_count, uuid);
692*94d3b452SApple OSS Distributions
693*94d3b452SApple OSS Distributions //T_LOG("Returning tunsock %d", tunsock);
694*94d3b452SApple OSS Distributions
695*94d3b452SApple OSS Distributions return tunsock;
696*94d3b452SApple OSS Distributions }
697*94d3b452SApple OSS Distributions
698*94d3b452SApple OSS Distributions static int (*create_tunsock)(int enable_netif, int enable_flowswitch, int channel_count, uuid_t uuid[]) = create_tunsock_new;
699*94d3b452SApple OSS Distributions
700*94d3b452SApple OSS Distributions #if 0
701*94d3b452SApple OSS Distributions static void
702*94d3b452SApple OSS Distributions ipsec_stats(void)
703*94d3b452SApple OSS Distributions {
704*94d3b452SApple OSS Distributions struct ifmibdata ifmd;
705*94d3b452SApple OSS Distributions
706*94d3b452SApple OSS Distributions len = sizeof(struct ifmibdata);
707*94d3b452SApple OSS Distributions name[3] = IFMIB_IFDATA;
708*94d3b452SApple OSS Distributions name[4] = interesting_row;
709*94d3b452SApple OSS Distributions name[5] = IpFDATA_GENERAL;
710*94d3b452SApple OSS Distributions if (sysctl(name, 6, &ifmd, &len, (void *)0, 0) == -1) {
711*94d3b452SApple OSS Distributions err(1, "sysctl IFDATA_GENERAL %d", interesting_row);
712*94d3b452SApple OSS Distributions }
713*94d3b452SApple OSS Distributions }
714*94d3b452SApple OSS Distributions #endif
715*94d3b452SApple OSS Distributions
716*94d3b452SApple OSS Distributions static void
permute_enables(void)717*94d3b452SApple OSS Distributions permute_enables(void)
718*94d3b452SApple OSS Distributions {
719*94d3b452SApple OSS Distributions int tunsock;
720*94d3b452SApple OSS Distributions T_EXPECT_GE(tunsock = create_tunsock(false, false, false, NULL), 0, NULL);
721*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(tunsock), NULL);
722*94d3b452SApple OSS Distributions T_EXPECT_GE(tunsock = create_tunsock(false, false, true, NULL), 0, NULL);
723*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(tunsock), NULL);
724*94d3b452SApple OSS Distributions T_EXPECT_GE(tunsock = create_tunsock(false, true, false, NULL), 0, NULL);
725*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(tunsock), NULL);
726*94d3b452SApple OSS Distributions T_EXPECT_GE(tunsock = create_tunsock(false, true, true, NULL), 0, NULL);
727*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(tunsock), NULL);
728*94d3b452SApple OSS Distributions T_EXPECT_GE(tunsock = create_tunsock(true, false, false, NULL), 0, NULL);
729*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(tunsock), NULL);
730*94d3b452SApple OSS Distributions T_EXPECT_GE(tunsock = create_tunsock(true, false, true, NULL), 0, NULL);
731*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(tunsock), NULL);
732*94d3b452SApple OSS Distributions T_EXPECT_GE(tunsock = create_tunsock(true, true, false, NULL), 0, NULL);
733*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(tunsock), NULL);
734*94d3b452SApple OSS Distributions T_EXPECT_GE(tunsock = create_tunsock(true, true, true, NULL), 0, NULL);
735*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(tunsock), NULL);
736*94d3b452SApple OSS Distributions }
737*94d3b452SApple OSS Distributions
738*94d3b452SApple OSS Distributions T_DECL(ipsec_enables, "This test checks combinations of netif/channel/flowswitch on ipsec")
739*94d3b452SApple OSS Distributions {
740*94d3b452SApple OSS Distributions setup_ipsec_test();
741*94d3b452SApple OSS Distributions permute_enables();
742*94d3b452SApple OSS Distributions }
743*94d3b452SApple OSS Distributions
744*94d3b452SApple OSS Distributions T_DECL(utun_enables, "This test checks combinations of netif/channel/flowswitch on utun")
745*94d3b452SApple OSS Distributions {
746*94d3b452SApple OSS Distributions setup_utun_test();
747*94d3b452SApple OSS Distributions permute_enables();
748*94d3b452SApple OSS Distributions }
749*94d3b452SApple OSS Distributions
750*94d3b452SApple OSS Distributions static int g_tunsock = -1;
751*94d3b452SApple OSS Distributions
752*94d3b452SApple OSS Distributions static void
cleanup_tunsock(void)753*94d3b452SApple OSS Distributions cleanup_tunsock(void)
754*94d3b452SApple OSS Distributions {
755*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(g_tunsock), NULL);
756*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(close(g_tunsock), EBADF, NULL);
757*94d3b452SApple OSS Distributions if (g_is_ipsec_test) {
758*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(g_pfkeyso), NULL);
759*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(close(g_pfkeyso), EBADF, NULL);
760*94d3b452SApple OSS Distributions }
761*94d3b452SApple OSS Distributions }
762*94d3b452SApple OSS Distributions
763*94d3b452SApple OSS Distributions static void
setup_tunsock(int channel_count,uuid_t uuids[])764*94d3b452SApple OSS Distributions setup_tunsock(int channel_count, uuid_t uuids[])
765*94d3b452SApple OSS Distributions {
766*94d3b452SApple OSS Distributions T_ASSERT_GE(g_tunsock = create_tunsock(true, false, channel_count, uuids), 0, NULL);
767*94d3b452SApple OSS Distributions T_ATEND(cleanup_tunsock);
768*94d3b452SApple OSS Distributions
769*94d3b452SApple OSS Distributions char ifname[IFXNAMSIZ];
770*94d3b452SApple OSS Distributions tunsock_get_ifname(g_tunsock, ifname);
771*94d3b452SApple OSS Distributions
772*94d3b452SApple OSS Distributions T_LOG("Created interface %s", ifname);
773*94d3b452SApple OSS Distributions
774*94d3b452SApple OSS Distributions uint32_t ifaddr = (10 << 24) | ((unsigned)getpid() & 0xffff) << 8 | 160;
775*94d3b452SApple OSS Distributions struct in_addr mask;
776*94d3b452SApple OSS Distributions g_addr1.s_addr = htonl(ifaddr);
777*94d3b452SApple OSS Distributions g_addr2.s_addr = htonl(ifaddr + 1);
778*94d3b452SApple OSS Distributions mask.s_addr = htonl(0xffffffff);
779*94d3b452SApple OSS Distributions
780*94d3b452SApple OSS Distributions ifnet_add_addr4(ifname, &g_addr1, &mask, &g_addr2);
781*94d3b452SApple OSS Distributions
782*94d3b452SApple OSS Distributions if (g_is_ipsec_test) {
783*94d3b452SApple OSS Distributions create_sa(ifname, SADB_ADD, 12345, &g_addr1, &g_addr2);
784*94d3b452SApple OSS Distributions create_sa(ifname, SADB_ADD, 12346, &g_addr2, &g_addr1);
785*94d3b452SApple OSS Distributions }
786*94d3b452SApple OSS Distributions }
787*94d3b452SApple OSS Distributions
788*94d3b452SApple OSS Distributions T_DECL(setup_ipsec, "This test sets up an ipsec interface")
789*94d3b452SApple OSS Distributions {
790*94d3b452SApple OSS Distributions setup_ipsec_test();
791*94d3b452SApple OSS Distributions setup_tunsock(1, NULL);
792*94d3b452SApple OSS Distributions }
793*94d3b452SApple OSS Distributions
794*94d3b452SApple OSS Distributions T_DECL(setup_utun, "This test sets up a utun interface")
795*94d3b452SApple OSS Distributions {
796*94d3b452SApple OSS Distributions setup_utun_test();
797*94d3b452SApple OSS Distributions setup_tunsock(1, NULL);
798*94d3b452SApple OSS Distributions }
799*94d3b452SApple OSS Distributions
800*94d3b452SApple OSS Distributions static const int SOCKET_TRAFFIC_CLASSES[] = {
801*94d3b452SApple OSS Distributions SO_TC_BK_SYS, // BK
802*94d3b452SApple OSS Distributions SO_TC_BK, // BK
803*94d3b452SApple OSS Distributions SO_TC_BE, // BE
804*94d3b452SApple OSS Distributions SO_TC_RD, // BE
805*94d3b452SApple OSS Distributions SO_TC_OAM, // BE
806*94d3b452SApple OSS Distributions SO_TC_AV, // VI
807*94d3b452SApple OSS Distributions SO_TC_RV, // VI
808*94d3b452SApple OSS Distributions SO_TC_VI, // VI
809*94d3b452SApple OSS Distributions SO_TC_VO, // VO
810*94d3b452SApple OSS Distributions SO_TC_CTL, // VO
811*94d3b452SApple OSS Distributions };
812*94d3b452SApple OSS Distributions
813*94d3b452SApple OSS Distributions // this should match ipsec_find_tx_ring_by_svc in ipsec driver
814*94d3b452SApple OSS Distributions static const int SOCKET_TC_TO_RING[] = {
815*94d3b452SApple OSS Distributions 3,
816*94d3b452SApple OSS Distributions 3,
817*94d3b452SApple OSS Distributions 2,
818*94d3b452SApple OSS Distributions 2,
819*94d3b452SApple OSS Distributions 2,
820*94d3b452SApple OSS Distributions 1,
821*94d3b452SApple OSS Distributions 1,
822*94d3b452SApple OSS Distributions 1,
823*94d3b452SApple OSS Distributions 0,
824*94d3b452SApple OSS Distributions 0,
825*94d3b452SApple OSS Distributions };
826*94d3b452SApple OSS Distributions
827*94d3b452SApple OSS Distributions /* How many sockets map to this ring */
828*94d3b452SApple OSS Distributions static const int RING_TO_TC_COUNT[] = {
829*94d3b452SApple OSS Distributions 2, 3, 3, 2,
830*94d3b452SApple OSS Distributions };
831*94d3b452SApple OSS Distributions
832*94d3b452SApple OSS Distributions static void
setup_channels_and_rings(int kq,int channel_count,channel_t channels[],channel_ring_t rxrings[],channel_ring_t txrings[],uuid_t uuids[],int cfds[])833*94d3b452SApple OSS Distributions setup_channels_and_rings(int kq, int channel_count, channel_t channels[], channel_ring_t rxrings[], channel_ring_t txrings[], uuid_t uuids[], int cfds[])
834*94d3b452SApple OSS Distributions {
835*94d3b452SApple OSS Distributions setup_tunsock(channel_count, uuids);
836*94d3b452SApple OSS Distributions
837*94d3b452SApple OSS Distributions #if 0
838*94d3b452SApple OSS Distributions // give time to enable a tcpdump if desired
839*94d3b452SApple OSS Distributions T_LOG("Sleeping 10");
840*94d3b452SApple OSS Distributions sleep(10);
841*94d3b452SApple OSS Distributions T_LOG("Done");
842*94d3b452SApple OSS Distributions #endif
843*94d3b452SApple OSS Distributions
844*94d3b452SApple OSS Distributions for (int ri = 0; ri < channel_count; ri++) {
845*94d3b452SApple OSS Distributions if (rxrings) {
846*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_ASSERT_NOTNULL(channels[ri] = os_channel_create(uuids[ri], 0), NULL);
847*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_ASSERT_NOTNULL(rxrings[ri] = os_channel_rx_ring(channels[ri],
848*94d3b452SApple OSS Distributions os_channel_ring_id(channels[ri], CHANNEL_FIRST_RX_RING)), NULL);
849*94d3b452SApple OSS Distributions }
850*94d3b452SApple OSS Distributions if (txrings) {
851*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_ASSERT_NOTNULL(channels[ri] = os_channel_create(uuids[ri], 0), NULL);
852*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_ASSERT_NOTNULL(rxrings[ri] = os_channel_rx_ring(channels[ri],
853*94d3b452SApple OSS Distributions os_channel_ring_id(channels[ri], CHANNEL_FIRST_TX_RING)), NULL);
854*94d3b452SApple OSS Distributions }
855*94d3b452SApple OSS Distributions
856*94d3b452SApple OSS Distributions struct kevent kev;
857*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(cfds[ri] = os_channel_get_fd(channels[ri]), NULL);
858*94d3b452SApple OSS Distributions EV_SET(&kev, cfds[ri], EVFILT_READ, EV_ADD | EV_ENABLE, 0, 0, (void *)(uintptr_t)ri);
859*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(kevent(kq, &kev, 1, NULL, 0, NULL), NULL);
860*94d3b452SApple OSS Distributions }
861*94d3b452SApple OSS Distributions }
862*94d3b452SApple OSS Distributions
863*94d3b452SApple OSS Distributions static void
cleanup_channels_and_rings(int channel_count,channel_t channels[],channel_ring_t rxrings[],channel_ring_t txrings[],uuid_t uuids[])864*94d3b452SApple OSS Distributions cleanup_channels_and_rings(int channel_count, channel_t channels[], channel_ring_t rxrings[], channel_ring_t txrings[], uuid_t uuids[])
865*94d3b452SApple OSS Distributions {
866*94d3b452SApple OSS Distributions for (int ri = 0; ri < channel_count; ri++) {
867*94d3b452SApple OSS Distributions if (rxrings) {
868*94d3b452SApple OSS Distributions rxrings[ri] = NULL;
869*94d3b452SApple OSS Distributions }
870*94d3b452SApple OSS Distributions if (txrings) {
871*94d3b452SApple OSS Distributions rxrings[ri] = NULL;
872*94d3b452SApple OSS Distributions }
873*94d3b452SApple OSS Distributions os_channel_destroy(channels[ri]);
874*94d3b452SApple OSS Distributions channels[ri] = NULL;
875*94d3b452SApple OSS Distributions uuid_clear(uuids[ri]);
876*94d3b452SApple OSS Distributions }
877*94d3b452SApple OSS Distributions }
878*94d3b452SApple OSS Distributions
879*94d3b452SApple OSS Distributions static void
setup_sockets(int sockets[SO_TC_MAX],int type)880*94d3b452SApple OSS Distributions setup_sockets(int sockets[SO_TC_MAX], int type)
881*94d3b452SApple OSS Distributions {
882*94d3b452SApple OSS Distributions for (int si = 0; si < SO_TC_MAX; si++) {
883*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(sockets[si] = socket(PF_INET, type, 0), NULL);
884*94d3b452SApple OSS Distributions
885*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(setsockopt(sockets[si], SOL_SOCKET,
886*94d3b452SApple OSS Distributions SO_TRAFFIC_CLASS, &SOCKET_TRAFFIC_CLASSES[si], sizeof(SOCKET_TRAFFIC_CLASSES[si])), NULL);
887*94d3b452SApple OSS Distributions
888*94d3b452SApple OSS Distributions // XXX setsockopt(IP_BOUND_IF) here?
889*94d3b452SApple OSS Distributions
890*94d3b452SApple OSS Distributions struct sockaddr_in sin;
891*94d3b452SApple OSS Distributions memset(&sin, 0, sizeof(sin));
892*94d3b452SApple OSS Distributions sin.sin_len = sizeof(sin);
893*94d3b452SApple OSS Distributions sin.sin_family = AF_INET;
894*94d3b452SApple OSS Distributions sin.sin_addr = g_addr1;
895*94d3b452SApple OSS Distributions
896*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(bind(sockets[si], (struct sockaddr *)&sin, sizeof(sin)), NULL);
897*94d3b452SApple OSS Distributions
898*94d3b452SApple OSS Distributions char sbuf[INET6_ADDRSTRLEN];
899*94d3b452SApple OSS Distributions inet_ntop(sin.sin_family, &sin.sin_addr.s_addr, sbuf, sizeof(sbuf));
900*94d3b452SApple OSS Distributions #if 0
901*94d3b452SApple OSS Distributions T_LOG("%s socket %d bound to %s port %d",
902*94d3b452SApple OSS Distributions type == SOCK_DGRAM ? "udp" : type == SOCK_STREAM ? "tcp" : "???",
903*94d3b452SApple OSS Distributions sockets[si], sbuf, ntohs(sin.sin_port));
904*94d3b452SApple OSS Distributions #endif
905*94d3b452SApple OSS Distributions setblocking(sockets[si], false);
906*94d3b452SApple OSS Distributions }
907*94d3b452SApple OSS Distributions }
908*94d3b452SApple OSS Distributions
909*94d3b452SApple OSS Distributions static void
cleanup_sockets(int sockets[SO_TC_MAX])910*94d3b452SApple OSS Distributions cleanup_sockets(int sockets[SO_TC_MAX])
911*94d3b452SApple OSS Distributions {
912*94d3b452SApple OSS Distributions for (int si = 0; si < SO_TC_MAX; si++) {
913*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(sockets[si]), NULL);
914*94d3b452SApple OSS Distributions sockets[si] = -1;
915*94d3b452SApple OSS Distributions }
916*94d3b452SApple OSS Distributions }
917*94d3b452SApple OSS Distributions
918*94d3b452SApple OSS Distributions static void
drain_ring(channel_ring_t rxring)919*94d3b452SApple OSS Distributions drain_ring(channel_ring_t rxring)
920*94d3b452SApple OSS Distributions {
921*94d3b452SApple OSS Distributions uint32_t i, sc = os_channel_available_slot_count(rxring);
922*94d3b452SApple OSS Distributions channel_slot_t rxprev = NULL;
923*94d3b452SApple OSS Distributions for (i = 0; i < sc; i++) {
924*94d3b452SApple OSS Distributions slot_prop_t rxprop;
925*94d3b452SApple OSS Distributions channel_slot_t rxslot;
926*94d3b452SApple OSS Distributions
927*94d3b452SApple OSS Distributions memset(&rxprop, 0, sizeof(rxprop));
928*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_NOTNULL(rxslot = os_channel_get_next_slot(rxring, rxprev, &rxprop), NULL);
929*94d3b452SApple OSS Distributions T_QUIET; T_ASSERT_NE_UINT(0, rxprop.sp_len, NULL);
930*94d3b452SApple OSS Distributions T_QUIET; T_ASSERT_NOTNULL((void *)rxprop.sp_buf_ptr, NULL);
931*94d3b452SApple OSS Distributions
932*94d3b452SApple OSS Distributions log_hexdump((void *)rxprop.sp_buf_ptr, rxprop.sp_len);
933*94d3b452SApple OSS Distributions
934*94d3b452SApple OSS Distributions rxprev = rxslot;
935*94d3b452SApple OSS Distributions }
936*94d3b452SApple OSS Distributions if (sc) {
937*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_ZERO(os_channel_advance_slot(rxring, rxprev), NULL);
938*94d3b452SApple OSS Distributions }
939*94d3b452SApple OSS Distributions }
940*94d3b452SApple OSS Distributions
941*94d3b452SApple OSS Distributions static void
send_one_packet(int s,int type)942*94d3b452SApple OSS Distributions send_one_packet(int s, int type)
943*94d3b452SApple OSS Distributions {
944*94d3b452SApple OSS Distributions struct sockaddr_in sin;
945*94d3b452SApple OSS Distributions memset(&sin, 0, sizeof(sin));
946*94d3b452SApple OSS Distributions sin.sin_len = sizeof(sin);
947*94d3b452SApple OSS Distributions sin.sin_family = AF_INET;
948*94d3b452SApple OSS Distributions sin.sin_addr = g_addr2;
949*94d3b452SApple OSS Distributions sin.sin_port = ntohs(12345);
950*94d3b452SApple OSS Distributions
951*94d3b452SApple OSS Distributions if (type == SOCK_STREAM) {
952*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_FAILURE(connect(s, (struct sockaddr *)&sin, sizeof(sin)), EINPROGRESS, NULL);
953*94d3b452SApple OSS Distributions }
954*94d3b452SApple OSS Distributions if (type == SOCK_DGRAM) {
955*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_EQ_LONG((long)sizeof(s), sendto(s, &s, sizeof(s), 0,
956*94d3b452SApple OSS Distributions (struct sockaddr *)&sin, sizeof(sin)), NULL);
957*94d3b452SApple OSS Distributions }
958*94d3b452SApple OSS Distributions }
959*94d3b452SApple OSS Distributions
960*94d3b452SApple OSS Distributions static void
expect_empty_rings(int channel_count,channel_ring_t rings[])961*94d3b452SApple OSS Distributions expect_empty_rings(int channel_count, channel_ring_t rings[])
962*94d3b452SApple OSS Distributions {
963*94d3b452SApple OSS Distributions /* Check all the rings and make sure there are no packets */
964*94d3b452SApple OSS Distributions for (int ri = 0; ri < channel_count; ri++) {
965*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_UINT(0U, os_channel_available_slot_count(rings[ri]), NULL);
966*94d3b452SApple OSS Distributions }
967*94d3b452SApple OSS Distributions }
968*94d3b452SApple OSS Distributions
969*94d3b452SApple OSS Distributions static void
xfer_1_packet_singly(int channel_count,int type)970*94d3b452SApple OSS Distributions xfer_1_packet_singly(int channel_count, int type)
971*94d3b452SApple OSS Distributions {
972*94d3b452SApple OSS Distributions uuid_t uuids[channel_count];
973*94d3b452SApple OSS Distributions channel_t channels[channel_count];
974*94d3b452SApple OSS Distributions int sockets[SO_TC_MAX];
975*94d3b452SApple OSS Distributions channel_ring_t rxrings[channel_count];
976*94d3b452SApple OSS Distributions int cfds[channel_count];
977*94d3b452SApple OSS Distributions int kq;
978*94d3b452SApple OSS Distributions
979*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(kq = kqueue(), NULL);
980*94d3b452SApple OSS Distributions
981*94d3b452SApple OSS Distributions setup_channels_and_rings(kq, channel_count, channels, rxrings, NULL, uuids, cfds);
982*94d3b452SApple OSS Distributions
983*94d3b452SApple OSS Distributions setup_sockets(sockets, type);
984*94d3b452SApple OSS Distributions
985*94d3b452SApple OSS Distributions for (int si = 0; si < SO_TC_MAX; si++) {
986*94d3b452SApple OSS Distributions expect_empty_rings(channel_count, rxrings);
987*94d3b452SApple OSS Distributions
988*94d3b452SApple OSS Distributions send_one_packet(sockets[si], type);
989*94d3b452SApple OSS Distributions
990*94d3b452SApple OSS Distributions int expected_ring = channel_count == 1 ? 0 : SOCKET_TC_TO_RING[si];
991*94d3b452SApple OSS Distributions
992*94d3b452SApple OSS Distributions /* Wait for the packet delivery and check that it's only one packet and on the correct ring */
993*94d3b452SApple OSS Distributions struct kevent kev[channel_count + 1];
994*94d3b452SApple OSS Distributions int nev;
995*94d3b452SApple OSS Distributions memset(kev, 0, sizeof(kev));
996*94d3b452SApple OSS Distributions struct timespec to = { 0, 100 * NSEC_PER_MSEC }; // 100 ms
997*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(nev = kevent(kq, NULL, 0, kev, channel_count + 1, &to), NULL);
998*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_INT(nev, 1, NULL);
999*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_PTR((void *)kev[0].ident, (void *)(uintptr_t)cfds[expected_ring], NULL);
1000*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_PTR(kev[0].udata, (void *)(uintptr_t)expected_ring, NULL);
1001*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_SHORT(kev[0].filter, (short)EVFILT_READ, NULL);
1002*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_FALSE(kev[0].flags & EV_ERROR, NULL);
1003*94d3b452SApple OSS Distributions
1004*94d3b452SApple OSS Distributions /* Make sure it comes out the expected interface */
1005*94d3b452SApple OSS Distributions for (int ri = 0; ri < channel_count; ri++) {
1006*94d3b452SApple OSS Distributions errno = 0;
1007*94d3b452SApple OSS Distributions
1008*94d3b452SApple OSS Distributions uint32_t sc = os_channel_available_slot_count(rxrings[ri]);
1009*94d3b452SApple OSS Distributions
1010*94d3b452SApple OSS Distributions /* Check that the packet appears only on the expected ring and
1011*94d3b452SApple OSS Distributions * is the only packet on the expected ring.
1012*94d3b452SApple OSS Distributions */
1013*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_UINT(ri == expected_ring, sc, NULL);
1014*94d3b452SApple OSS Distributions
1015*94d3b452SApple OSS Distributions if ((ri == expected_ring) == sc) {
1016*94d3b452SApple OSS Distributions T_PASS("tc index %d ring %d expected ring %d slot count %u", si, ri, expected_ring, sc);
1017*94d3b452SApple OSS Distributions } else {
1018*94d3b452SApple OSS Distributions T_FAIL("tc index %d ring %d expected ring %d slot count %u", si, ri, expected_ring, sc);
1019*94d3b452SApple OSS Distributions }
1020*94d3b452SApple OSS Distributions
1021*94d3b452SApple OSS Distributions drain_ring(rxrings[ri]);
1022*94d3b452SApple OSS Distributions }
1023*94d3b452SApple OSS Distributions }
1024*94d3b452SApple OSS Distributions
1025*94d3b452SApple OSS Distributions cleanup_sockets(sockets);
1026*94d3b452SApple OSS Distributions
1027*94d3b452SApple OSS Distributions cleanup_channels_and_rings(channel_count, channels, rxrings, NULL, uuids);
1028*94d3b452SApple OSS Distributions
1029*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(kq), NULL);
1030*94d3b452SApple OSS Distributions }
1031*94d3b452SApple OSS Distributions
1032*94d3b452SApple OSS Distributions T_DECL(ipsec35889979u1s, "transfers 1 packet at a time of each sevice class over udp to a single ring")
1033*94d3b452SApple OSS Distributions {
1034*94d3b452SApple OSS Distributions setup_ipsec_test();
1035*94d3b452SApple OSS Distributions xfer_1_packet_singly(1, SOCK_DGRAM);
1036*94d3b452SApple OSS Distributions }
1037*94d3b452SApple OSS Distributions
1038*94d3b452SApple OSS Distributions T_DECL(ipsec35889979u4s, "transfers 1 packet at a time of each sevice class over udp to 4 rings")
1039*94d3b452SApple OSS Distributions {
1040*94d3b452SApple OSS Distributions setup_ipsec_test();
1041*94d3b452SApple OSS Distributions xfer_1_packet_singly(4, SOCK_DGRAM);
1042*94d3b452SApple OSS Distributions }
1043*94d3b452SApple OSS Distributions
1044*94d3b452SApple OSS Distributions T_DECL(ipsec35889979t1s, "transfers 1 packet at a time of each sevice class over tcp to a single ring")
1045*94d3b452SApple OSS Distributions {
1046*94d3b452SApple OSS Distributions setup_ipsec_test();
1047*94d3b452SApple OSS Distributions xfer_1_packet_singly(1, SOCK_STREAM);
1048*94d3b452SApple OSS Distributions }
1049*94d3b452SApple OSS Distributions
1050*94d3b452SApple OSS Distributions
1051*94d3b452SApple OSS Distributions T_DECL(ipsec35889979t4s, "transfers 1 packet at a time of each sevice class over tcp to 4 rings",
1052*94d3b452SApple OSS Distributions /* This test will fail because tcp syn packets get elevated
1053*94d3b452SApple OSS Distributions * due to ack prioritization
1054*94d3b452SApple OSS Distributions */
1055*94d3b452SApple OSS Distributions T_META_ENABLED(false))
1056*94d3b452SApple OSS Distributions {
1057*94d3b452SApple OSS Distributions setup_ipsec_test();
1058*94d3b452SApple OSS Distributions xfer_1_packet_singly(4, SOCK_STREAM);
1059*94d3b452SApple OSS Distributions }
1060*94d3b452SApple OSS Distributions
1061*94d3b452SApple OSS Distributions static void
xfer_1_packet_together(int channel_count,int type)1062*94d3b452SApple OSS Distributions xfer_1_packet_together(int channel_count, int type)
1063*94d3b452SApple OSS Distributions {
1064*94d3b452SApple OSS Distributions uuid_t uuids[channel_count];
1065*94d3b452SApple OSS Distributions channel_t channels[channel_count];
1066*94d3b452SApple OSS Distributions int sockets[SO_TC_MAX];
1067*94d3b452SApple OSS Distributions channel_ring_t rxrings[channel_count];
1068*94d3b452SApple OSS Distributions int cfds[channel_count];
1069*94d3b452SApple OSS Distributions int kq;
1070*94d3b452SApple OSS Distributions
1071*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(kq = kqueue(), NULL);
1072*94d3b452SApple OSS Distributions
1073*94d3b452SApple OSS Distributions setup_channels_and_rings(kq, channel_count, channels, rxrings, NULL, uuids, cfds);
1074*94d3b452SApple OSS Distributions
1075*94d3b452SApple OSS Distributions setup_sockets(sockets, type);
1076*94d3b452SApple OSS Distributions
1077*94d3b452SApple OSS Distributions for (int si = 0; si < SO_TC_MAX; si++) {
1078*94d3b452SApple OSS Distributions expect_empty_rings(channel_count, rxrings);
1079*94d3b452SApple OSS Distributions
1080*94d3b452SApple OSS Distributions send_one_packet(sockets[si], type);
1081*94d3b452SApple OSS Distributions }
1082*94d3b452SApple OSS Distributions
1083*94d3b452SApple OSS Distributions /* Sleep to make sure all packets get delivered */
1084*94d3b452SApple OSS Distributions struct timespec to = { 0, 100 * NSEC_PER_MSEC }; // 100 ms
1085*94d3b452SApple OSS Distributions nanosleep(&to, NULL);
1086*94d3b452SApple OSS Distributions
1087*94d3b452SApple OSS Distributions /* Wait for the packet delivery and check that all rings event */
1088*94d3b452SApple OSS Distributions struct kevent kev[channel_count + 1];
1089*94d3b452SApple OSS Distributions int nev;
1090*94d3b452SApple OSS Distributions memset(kev, 0, sizeof(kev));
1091*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_POSIX_SUCCESS(nev = kevent(kq, NULL, 0, kev, channel_count + 1, &to), NULL);
1092*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_INT(nev, channel_count, NULL);
1093*94d3b452SApple OSS Distributions
1094*94d3b452SApple OSS Distributions uint32_t found[channel_count];
1095*94d3b452SApple OSS Distributions memset(found, 0, sizeof(found));
1096*94d3b452SApple OSS Distributions for (int e = 0; e < nev; e++) {
1097*94d3b452SApple OSS Distributions T_LOG("kevent %lu filter 0x%4x flags 0x%04x fflags 0x%08x data %"PRIdPTR" udata %p",
1098*94d3b452SApple OSS Distributions kev[e].ident, kev[e].filter, kev[e].flags, kev[e].fflags, kev[e].data, kev[e].udata);
1099*94d3b452SApple OSS Distributions
1100*94d3b452SApple OSS Distributions T_QUIET; T_ASSERT_GE_PTR(kev[e].udata, (void *)0, NULL);
1101*94d3b452SApple OSS Distributions T_QUIET; T_ASSERT_LT_PTR(kev[e].udata, (void *)(intptr_t)channel_count, NULL);
1102*94d3b452SApple OSS Distributions int ri = (int)kev[e].udata;
1103*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_UINT(found[ri], 0U, NULL);
1104*94d3b452SApple OSS Distributions
1105*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_ULONG(kev[e].ident, (uintptr_t)cfds[ri], NULL);
1106*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_SHORT(kev[e].filter, (short)EVFILT_READ, NULL);
1107*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_FALSE(kev[e].flags & EV_ERROR, NULL);
1108*94d3b452SApple OSS Distributions
1109*94d3b452SApple OSS Distributions if (channel_count == 1) {
1110*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_LONG(kev[e].data, (long)SO_TC_MAX, NULL);
1111*94d3b452SApple OSS Distributions } else {
1112*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_EQ_LONG(kev[e].data, (long)RING_TO_TC_COUNT[ri], NULL);
1113*94d3b452SApple OSS Distributions }
1114*94d3b452SApple OSS Distributions
1115*94d3b452SApple OSS Distributions found[ri] += (uint32_t)kev[e].data;
1116*94d3b452SApple OSS Distributions }
1117*94d3b452SApple OSS Distributions /* Check that something came out of all rings */
1118*94d3b452SApple OSS Distributions for (int ri = 0; ri < channel_count; ri++) {
1119*94d3b452SApple OSS Distributions T_QUIET; T_EXPECT_NE_UINT(found[ri], 0U, NULL);
1120*94d3b452SApple OSS Distributions }
1121*94d3b452SApple OSS Distributions
1122*94d3b452SApple OSS Distributions /* Make sure it comes out the expected interface */
1123*94d3b452SApple OSS Distributions for (int ri = 0; ri < channel_count; ri++) {
1124*94d3b452SApple OSS Distributions uint32_t sc = os_channel_available_slot_count(rxrings[ri]);
1125*94d3b452SApple OSS Distributions if (channel_count == 1) {
1126*94d3b452SApple OSS Distributions if (sc == SO_TC_MAX) {
1127*94d3b452SApple OSS Distributions T_PASS("ring %d got %"PRIu32" slots expecting %"PRIu32"", ri, sc, SO_TC_MAX);
1128*94d3b452SApple OSS Distributions } else {
1129*94d3b452SApple OSS Distributions T_FAIL("ring %d got %"PRIu32" slots expecting %"PRIu32"", ri, sc, SO_TC_MAX);
1130*94d3b452SApple OSS Distributions }
1131*94d3b452SApple OSS Distributions } else {
1132*94d3b452SApple OSS Distributions if (sc == (uint32_t)RING_TO_TC_COUNT[ri]) {
1133*94d3b452SApple OSS Distributions T_PASS("ring %d got %"PRIu32" slots expecting %"PRIu32"", ri, sc, (uint32_t)RING_TO_TC_COUNT[ri]);
1134*94d3b452SApple OSS Distributions } else {
1135*94d3b452SApple OSS Distributions T_FAIL("ring %d got %"PRIu32" slots expecting %"PRIu32"", ri, sc, (uint32_t)RING_TO_TC_COUNT[ri]);
1136*94d3b452SApple OSS Distributions }
1137*94d3b452SApple OSS Distributions }
1138*94d3b452SApple OSS Distributions
1139*94d3b452SApple OSS Distributions drain_ring(rxrings[ri]);
1140*94d3b452SApple OSS Distributions }
1141*94d3b452SApple OSS Distributions
1142*94d3b452SApple OSS Distributions cleanup_sockets(sockets);
1143*94d3b452SApple OSS Distributions
1144*94d3b452SApple OSS Distributions cleanup_channels_and_rings(channel_count, channels, rxrings, NULL, uuids);
1145*94d3b452SApple OSS Distributions
1146*94d3b452SApple OSS Distributions T_QUIET; T_WITH_ERRNO; T_EXPECT_POSIX_ZERO(close(kq), NULL);
1147*94d3b452SApple OSS Distributions }
1148*94d3b452SApple OSS Distributions
1149*94d3b452SApple OSS Distributions T_DECL(ipsec35889979u1m, "transfers 1 packet together of each sevice class over udp to a single ring")
1150*94d3b452SApple OSS Distributions {
1151*94d3b452SApple OSS Distributions setup_ipsec_test();
1152*94d3b452SApple OSS Distributions xfer_1_packet_together(1, SOCK_DGRAM);
1153*94d3b452SApple OSS Distributions }
1154*94d3b452SApple OSS Distributions
1155*94d3b452SApple OSS Distributions T_DECL(ipsec35889979u4m, "transfers 1 packet together of each sevice class over udp to 4 rings")
1156*94d3b452SApple OSS Distributions {
1157*94d3b452SApple OSS Distributions setup_ipsec_test();
1158*94d3b452SApple OSS Distributions xfer_1_packet_together(4, SOCK_DGRAM);
1159*94d3b452SApple OSS Distributions }
1160*94d3b452SApple OSS Distributions
1161*94d3b452SApple OSS Distributions T_DECL(ipsec35889979t1m, "transfers 1 packet together of each sevice class over tcp to a single ring")
1162*94d3b452SApple OSS Distributions {
1163*94d3b452SApple OSS Distributions setup_ipsec_test();
1164*94d3b452SApple OSS Distributions xfer_1_packet_together(1, SOCK_STREAM);
1165*94d3b452SApple OSS Distributions }
1166*94d3b452SApple OSS Distributions
1167*94d3b452SApple OSS Distributions T_DECL(ipsec35889979t4m, "transfers 1 packet together of each sevice class over tcp to 4 rings",
1168*94d3b452SApple OSS Distributions /* This test will fail because tcp syn packets get elevated
1169*94d3b452SApple OSS Distributions * due to ack prioritization
1170*94d3b452SApple OSS Distributions */
1171*94d3b452SApple OSS Distributions T_META_ENABLED(false))
1172*94d3b452SApple OSS Distributions {
1173*94d3b452SApple OSS Distributions setup_ipsec_test();
1174*94d3b452SApple OSS Distributions xfer_1_packet_together(4, SOCK_STREAM);
1175*94d3b452SApple OSS Distributions }
1176