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