xref: /xnu-10063.101.15/tests/ipsec.m (revision 94d3b452840153a99b38a3a9659680b2a006908e)
1*94d3b452SApple OSS Distributions#import <darwintest.h>
2*94d3b452SApple OSS Distributions#import <darwintest_utils.h>
3*94d3b452SApple OSS Distributions#import <dispatch/dispatch.h>
4*94d3b452SApple OSS Distributions#import <net/pfkeyv2.h>
5*94d3b452SApple OSS Distributions#import <netinet6/ipsec.h>
6*94d3b452SApple OSS Distributions#import <arpa/inet.h>
7*94d3b452SApple OSS Distributions#import <NetworkExtension/NetworkExtensionPrivate.h>
8*94d3b452SApple OSS Distributions#import <System/net/bpf.h>
9*94d3b452SApple OSS Distributions#import <System/netinet/ip.h>
10*94d3b452SApple OSS Distributions#import <System/netinet/ip6.h>
11*94d3b452SApple OSS Distributions
12*94d3b452SApple OSS Distributions#import <sys/mbuf.h>
13*94d3b452SApple OSS Distributions
14*94d3b452SApple OSS Distributions
15*94d3b452SApple OSS DistributionsT_GLOBAL_META(
16*94d3b452SApple OSS Distributions	T_META_NAMESPACE("xnu.ipsec"),
17*94d3b452SApple OSS Distributions	T_META_ASROOT(true),
18*94d3b452SApple OSS Distributions	T_META_CHECK_LEAKS(false));
19*94d3b452SApple OSS Distributions
20*94d3b452SApple OSS Distributionstypedef enum {
21*94d3b452SApple OSS Distributions	TEST_INVALID = 0,
22*94d3b452SApple OSS Distributions	TEST_IPSEC_IPv4_ENCAPSULATE_PANIC = 1,
23*94d3b452SApple OSS Distributions	TEST_IPSEC_IPv6_ENCAPSULATE_PANIC = 2,
24*94d3b452SApple OSS Distributions} test_identifier;
25*94d3b452SApple OSS Distributions
26*94d3b452SApple OSS Distributions#define TEST_SRC_ADDRESS_IPv4                           "10.0.0.2"
27*94d3b452SApple OSS Distributions#define TEST_DST_ADDRESS_IPv4                           "10.0.0.3"
28*94d3b452SApple OSS Distributions#define TEST_IPSEC_IPv4_INTERFACE_ADDRESS               "192.168.10.10"
29*94d3b452SApple OSS Distributions#define TEST_IPSEC_IPv6_INTERFACE_ADDRESS               "fdd3:0f89:9afd:9b9c::1234"
30*94d3b452SApple OSS Distributions#define TEST_DELEGATE_IPSEC_INTERFACE_ADDRESS           "192.168.20.10"
31*94d3b452SApple OSS Distributions#define TEST_IPSEC_IPv4_INTERFACE_MASK                  "255.255.255.255"
32*94d3b452SApple OSS Distributions#define TEST_IPSEC_IPv6_INTERFACE_MASK                  "ffff:ffff:ffff:ffff::"
33*94d3b452SApple OSS Distributions
34*94d3b452SApple OSS Distributionsstatic test_identifier test_id = TEST_INVALID;
35*94d3b452SApple OSS Distributionsstatic dispatch_source_t pfkey_source = NULL;
36*94d3b452SApple OSS Distributionsstatic NEVirtualInterfaceRef ipsecInterface = NULL;
37*94d3b452SApple OSS Distributionsstatic NEVirtualInterfaceRef delegateIPsecInterface = NULL;
38*94d3b452SApple OSS Distributionsstatic int bpf_fd = -1;
39*94d3b452SApple OSS Distributions
40*94d3b452SApple OSS Distributionsstatic void bpf_write(int fd);
41*94d3b452SApple OSS Distributionsstatic void pfkey_cleanup(void);
42*94d3b452SApple OSS Distributionsstatic void pfkey_process_message_test_encapsulate_panic(uint8_t **mhp, int pfkey_socket);
43*94d3b452SApple OSS Distributions
44*94d3b452SApple OSS Distributionsstatic void(*const process_pfkey_message_tests[])(uint8_t * *mhp, int pfkey_socket) =
45*94d3b452SApple OSS Distributions{
46*94d3b452SApple OSS Distributions	NULL,
47*94d3b452SApple OSS Distributions	pfkey_process_message_test_encapsulate_panic,    // TEST_IPSEC_IPv4_ENCAPSULATE_PANIC
48*94d3b452SApple OSS Distributions	pfkey_process_message_test_encapsulate_panic,    // TEST_IPSEC_IPv6_ENCAPSULATE_PANIC
49*94d3b452SApple OSS Distributions};
50*94d3b452SApple OSS Distributions
51*94d3b452SApple OSS Distributionsstatic void
52*94d3b452SApple OSS Distributionssend_pkey_add_sa(int pfkey_socket, uint32_t spi, const char *src, const char *dst, int family)
53*94d3b452SApple OSS Distributions{
54*94d3b452SApple OSS Distributions	uint8_t payload[MCLBYTES] __attribute__ ((aligned(32)));
55*94d3b452SApple OSS Distributions	bzero(payload, sizeof(payload));
56*94d3b452SApple OSS Distributions	uint16_t tlen = 0;
57*94d3b452SApple OSS Distributions
58*94d3b452SApple OSS Distributions	struct sadb_msg *msg_payload = (struct sadb_msg *)payload;
59*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_version = PF_KEY_V2;
60*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_type = SADB_ADD;
61*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_errno = 0;
62*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_satype = SADB_SATYPE_ESP;
63*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_len = PFKEY_UNIT64(tlen);
64*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_reserved = 0;
65*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_seq = 0;
66*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_pid = (u_int32_t)getpid();
67*94d3b452SApple OSS Distributions	tlen += sizeof(*msg_payload);
68*94d3b452SApple OSS Distributions
69*94d3b452SApple OSS Distributions	struct sadb_sa_2 *sa2_payload = (struct sadb_sa_2 *)(void *)(payload + tlen);
70*94d3b452SApple OSS Distributions	sa2_payload->sa.sadb_sa_len = PFKEY_UNIT64(sizeof(*sa2_payload));
71*94d3b452SApple OSS Distributions	sa2_payload->sa.sadb_sa_exttype = SADB_EXT_SA;
72*94d3b452SApple OSS Distributions	sa2_payload->sa.sadb_sa_spi = htonl(spi);
73*94d3b452SApple OSS Distributions	sa2_payload->sa.sadb_sa_replay = 4;
74*94d3b452SApple OSS Distributions	sa2_payload->sa.sadb_sa_state = SADB_SASTATE_LARVAL;
75*94d3b452SApple OSS Distributions	sa2_payload->sa.sadb_sa_auth = SADB_X_AALG_SHA2_256;
76*94d3b452SApple OSS Distributions	sa2_payload->sa.sadb_sa_encrypt = SADB_X_EALG_AESCBC;
77*94d3b452SApple OSS Distributions	sa2_payload->sa.sadb_sa_flags |= (SADB_X_EXT_NATT | SADB_X_EXT_NATT_KEEPALIVE);
78*94d3b452SApple OSS Distributions	sa2_payload->sadb_sa_natt_src_port = htons(4500);
79*94d3b452SApple OSS Distributions	sa2_payload->sadb_sa_natt_port = 4500;
80*94d3b452SApple OSS Distributions	sa2_payload->sadb_sa_natt_interval = 20;
81*94d3b452SApple OSS Distributions	sa2_payload->sadb_sa_natt_offload_interval = 0;
82*94d3b452SApple OSS Distributions	tlen += sizeof(*sa2_payload);
83*94d3b452SApple OSS Distributions
84*94d3b452SApple OSS Distributions	struct sadb_x_sa2 *sa2_x_payload = (struct sadb_x_sa2 *)(void *)(payload + tlen);
85*94d3b452SApple OSS Distributions	sa2_x_payload->sadb_x_sa2_len = PFKEY_UNIT64(sizeof(*sa2_x_payload));
86*94d3b452SApple OSS Distributions	sa2_x_payload->sadb_x_sa2_exttype = SADB_X_EXT_SA2;
87*94d3b452SApple OSS Distributions	sa2_x_payload->sadb_x_sa2_mode = IPSEC_MODE_TUNNEL;
88*94d3b452SApple OSS Distributions	sa2_x_payload->sadb_x_sa2_reqid = 0;
89*94d3b452SApple OSS Distributions	tlen += sizeof(*sa2_x_payload);
90*94d3b452SApple OSS Distributions
91*94d3b452SApple OSS Distributions	uint8_t prefixlen = (family == AF_INET) ? (sizeof(struct in_addr) << 3) : (sizeof(struct in6_addr) << 3);
92*94d3b452SApple OSS Distributions
93*94d3b452SApple OSS Distributions	struct sadb_address *src_address_payload = (struct sadb_address *)(void *)(payload + tlen);
94*94d3b452SApple OSS Distributions	src_address_payload->sadb_address_exttype = SADB_EXT_ADDRESS_SRC & 0xffff;
95*94d3b452SApple OSS Distributions	src_address_payload->sadb_address_proto = IPSEC_ULPROTO_ANY & 0xff;
96*94d3b452SApple OSS Distributions	src_address_payload->sadb_address_prefixlen = prefixlen;
97*94d3b452SApple OSS Distributions	src_address_payload->sadb_address_reserved = 0;
98*94d3b452SApple OSS Distributions	tlen += sizeof(*src_address_payload);
99*94d3b452SApple OSS Distributions
100*94d3b452SApple OSS Distributions	if (family == AF_INET) {
101*94d3b452SApple OSS Distributions		struct sockaddr_in *src4 = (struct sockaddr_in *)(void *)(payload + tlen);
102*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_EQ_INT(inet_pton(AF_INET, src, &src4->sin_addr), 1, "src address fail");
103*94d3b452SApple OSS Distributions		src4->sin_family = AF_INET;
104*94d3b452SApple OSS Distributions		src4->sin_len = sizeof(*src4);
105*94d3b452SApple OSS Distributions		uint16_t len = sizeof(*src_address_payload) + PFKEY_ALIGN8(src4->sin_len);
106*94d3b452SApple OSS Distributions		src_address_payload->sadb_address_len = PFKEY_UNIT64(len);
107*94d3b452SApple OSS Distributions		tlen += PFKEY_ALIGN8(src4->sin_len);
108*94d3b452SApple OSS Distributions	} else {
109*94d3b452SApple OSS Distributions		struct sockaddr_in6 *src6 = (struct sockaddr_in6 *)(void *)(payload + tlen);
110*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_EQ_INT(inet_pton(AF_INET6, src, &src6->sin6_addr), 1, "src address fail");
111*94d3b452SApple OSS Distributions		src6->sin6_family = AF_INET6;
112*94d3b452SApple OSS Distributions		src6->sin6_len = sizeof(*src6);
113*94d3b452SApple OSS Distributions		uint16_t len = sizeof(*src_address_payload) + PFKEY_ALIGN8(src6->sin6_len);
114*94d3b452SApple OSS Distributions		src_address_payload->sadb_address_len = PFKEY_UNIT64(len);
115*94d3b452SApple OSS Distributions		tlen += PFKEY_ALIGN8(src6->sin6_len);
116*94d3b452SApple OSS Distributions	}
117*94d3b452SApple OSS Distributions
118*94d3b452SApple OSS Distributions	struct sadb_address *dst_address_payload = (struct sadb_address *)(void *)(payload + tlen);
119*94d3b452SApple OSS Distributions	dst_address_payload->sadb_address_exttype = SADB_EXT_ADDRESS_DST & 0xffff;
120*94d3b452SApple OSS Distributions	dst_address_payload->sadb_address_proto = IPSEC_ULPROTO_ANY & 0xff;
121*94d3b452SApple OSS Distributions	dst_address_payload->sadb_address_prefixlen = prefixlen;
122*94d3b452SApple OSS Distributions	dst_address_payload->sadb_address_reserved = 0;
123*94d3b452SApple OSS Distributions	tlen += sizeof(*dst_address_payload);
124*94d3b452SApple OSS Distributions
125*94d3b452SApple OSS Distributions	if (family == AF_INET) {
126*94d3b452SApple OSS Distributions		struct sockaddr_in *dst4 = (struct sockaddr_in *)(void *)(payload + tlen);
127*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_EQ_INT(inet_pton(AF_INET, dst, &dst4->sin_addr), 1, "dst address fail");
128*94d3b452SApple OSS Distributions		dst4->sin_family = AF_INET;
129*94d3b452SApple OSS Distributions		dst4->sin_len = sizeof(*dst4);
130*94d3b452SApple OSS Distributions		uint16_t len = sizeof(*dst_address_payload) + PFKEY_ALIGN8(dst4->sin_len);
131*94d3b452SApple OSS Distributions		dst_address_payload->sadb_address_len = PFKEY_UNIT64(len);
132*94d3b452SApple OSS Distributions		tlen += PFKEY_ALIGN8(dst4->sin_len);
133*94d3b452SApple OSS Distributions	} else {
134*94d3b452SApple OSS Distributions		struct sockaddr_in6 *dst6 = (struct sockaddr_in6 *)(void *)(payload + tlen);
135*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_EQ_INT(inet_pton(AF_INET6, dst, &dst6->sin6_addr), 1, "dst address fail");
136*94d3b452SApple OSS Distributions		dst6->sin6_family = AF_INET6;
137*94d3b452SApple OSS Distributions		dst6->sin6_len = sizeof(*dst6);
138*94d3b452SApple OSS Distributions		uint16_t len = sizeof(*dst_address_payload) + PFKEY_ALIGN8(dst6->sin6_len);
139*94d3b452SApple OSS Distributions		dst_address_payload->sadb_address_len = PFKEY_UNIT64(len);
140*94d3b452SApple OSS Distributions		tlen += PFKEY_ALIGN8(dst6->sin6_len);
141*94d3b452SApple OSS Distributions	}
142*94d3b452SApple OSS Distributions
143*94d3b452SApple OSS Distributions	CFStringRef ipsecIfName = NEVirtualInterfaceCopyName(ipsecInterface);
144*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(ipsecIfName, "failed to get ipsec interface name");
145*94d3b452SApple OSS Distributions	char ifname[IFNAMSIZ];
146*94d3b452SApple OSS Distributions	CFStringGetCString(ipsecIfName, ifname, IFNAMSIZ, kCFStringEncodingUTF8);
147*94d3b452SApple OSS Distributions
148*94d3b452SApple OSS Distributions	CFStringRef delegateIPsecIfName = NEVirtualInterfaceCopyName(delegateIPsecInterface);
149*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(delegateIPsecIfName, "failed to get delegate ipsec interface name");
150*94d3b452SApple OSS Distributions	char delegateIfname[IFNAMSIZ];
151*94d3b452SApple OSS Distributions	CFStringGetCString(delegateIPsecIfName, delegateIfname, IFNAMSIZ, kCFStringEncodingUTF8);
152*94d3b452SApple OSS Distributions
153*94d3b452SApple OSS Distributions	struct sadb_x_ipsecif *ipsec_if_payload = (struct sadb_x_ipsecif *)(void *)(payload + tlen);
154*94d3b452SApple OSS Distributions	ipsec_if_payload->sadb_x_ipsecif_len = PFKEY_UNIT64(sizeof(*ipsec_if_payload));
155*94d3b452SApple OSS Distributions	ipsec_if_payload->sadb_x_ipsecif_exttype = SADB_X_EXT_IPSECIF;
156*94d3b452SApple OSS Distributions	strncpy(ipsec_if_payload->sadb_x_ipsecif_ipsec_if, ifname, strlen(ifname));
157*94d3b452SApple OSS Distributions	strncpy(ipsec_if_payload->sadb_x_ipsecif_outgoing_if, delegateIfname, strlen(delegateIfname));
158*94d3b452SApple OSS Distributions	tlen += sizeof(*ipsec_if_payload);
159*94d3b452SApple OSS Distributions
160*94d3b452SApple OSS Distributions	struct sadb_key *encrypt_key_payload = (struct sadb_key *)(void *)(payload + tlen);
161*94d3b452SApple OSS Distributions	uint16_t len = sizeof(*encrypt_key_payload) + PFKEY_ALIGN8(32);
162*94d3b452SApple OSS Distributions	encrypt_key_payload->sadb_key_len = PFKEY_UNIT64(len);
163*94d3b452SApple OSS Distributions	encrypt_key_payload->sadb_key_exttype = SADB_EXT_KEY_ENCRYPT;
164*94d3b452SApple OSS Distributions	encrypt_key_payload->sadb_key_bits = (uint16_t)(32 << 3);
165*94d3b452SApple OSS Distributions	encrypt_key_payload->sadb_key_reserved = 0;
166*94d3b452SApple OSS Distributions	tlen += sizeof(*encrypt_key_payload);
167*94d3b452SApple OSS Distributions	arc4random_buf(payload + tlen, 32);
168*94d3b452SApple OSS Distributions	tlen += PFKEY_ALIGN8(32);
169*94d3b452SApple OSS Distributions
170*94d3b452SApple OSS Distributions	struct sadb_key *auth_key_payload = (struct sadb_key *)(void *)(payload + tlen);
171*94d3b452SApple OSS Distributions	len = sizeof(*auth_key_payload) + PFKEY_ALIGN8(32);
172*94d3b452SApple OSS Distributions	auth_key_payload->sadb_key_len = PFKEY_UNIT64(len);
173*94d3b452SApple OSS Distributions	auth_key_payload->sadb_key_exttype = SADB_EXT_KEY_AUTH;
174*94d3b452SApple OSS Distributions	auth_key_payload->sadb_key_bits = (uint16_t)(32 << 3);
175*94d3b452SApple OSS Distributions	auth_key_payload->sadb_key_reserved = 0;
176*94d3b452SApple OSS Distributions	tlen += sizeof(*auth_key_payload);
177*94d3b452SApple OSS Distributions	arc4random_buf(payload + tlen, 32);
178*94d3b452SApple OSS Distributions	tlen += PFKEY_ALIGN8(32);
179*94d3b452SApple OSS Distributions
180*94d3b452SApple OSS Distributions	struct sadb_lifetime *hard_lifetime_payload = (struct sadb_lifetime *)(void *)(payload + tlen);
181*94d3b452SApple OSS Distributions	hard_lifetime_payload->sadb_lifetime_len = PFKEY_UNIT64(sizeof(*hard_lifetime_payload));
182*94d3b452SApple OSS Distributions	hard_lifetime_payload->sadb_lifetime_exttype = SADB_EXT_LIFETIME_HARD;
183*94d3b452SApple OSS Distributions	tlen += sizeof(*hard_lifetime_payload);
184*94d3b452SApple OSS Distributions
185*94d3b452SApple OSS Distributions	struct sadb_lifetime *soft_lifetime_payload = (struct sadb_lifetime *)(void *)(payload + tlen);
186*94d3b452SApple OSS Distributions	soft_lifetime_payload->sadb_lifetime_len = PFKEY_UNIT64(sizeof(*soft_lifetime_payload));
187*94d3b452SApple OSS Distributions	soft_lifetime_payload->sadb_lifetime_exttype = SADB_EXT_LIFETIME_SOFT;
188*94d3b452SApple OSS Distributions	tlen += sizeof(*soft_lifetime_payload);
189*94d3b452SApple OSS Distributions
190*94d3b452SApple OSS Distributions	// Update the total length
191*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_len = PFKEY_UNIT64(tlen);
192*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_POSIX_SUCCESS(send(pfkey_socket, payload, (size_t)PFKEY_UNUNIT64(msg_payload->sadb_msg_len), 0), "pfkey send update sa");
193*94d3b452SApple OSS Distributions}
194*94d3b452SApple OSS Distributions
195*94d3b452SApple OSS Distributionsstatic void
196*94d3b452SApple OSS Distributionssend_pfkey_flush_sa(int pfkey_socket)
197*94d3b452SApple OSS Distributions{
198*94d3b452SApple OSS Distributions	uint8_t payload[MCLBYTES] __attribute__ ((aligned(32)));
199*94d3b452SApple OSS Distributions	bzero(payload, sizeof(payload));
200*94d3b452SApple OSS Distributions	uint16_t tlen = 0;
201*94d3b452SApple OSS Distributions
202*94d3b452SApple OSS Distributions	struct sadb_msg *msg_payload = (struct sadb_msg *)payload;
203*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_version = PF_KEY_V2;
204*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_type = SADB_FLUSH;
205*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_errno = 0;
206*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_satype = SADB_SATYPE_UNSPEC;
207*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_len = PFKEY_UNIT64(tlen);
208*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_reserved = 0;
209*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_seq = 0;
210*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_pid = (u_int32_t)getpid();
211*94d3b452SApple OSS Distributions	tlen += sizeof(*msg_payload);
212*94d3b452SApple OSS Distributions
213*94d3b452SApple OSS Distributions	// Update the total length
214*94d3b452SApple OSS Distributions	msg_payload->sadb_msg_len = PFKEY_UNIT64(tlen);
215*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_POSIX_SUCCESS(send(pfkey_socket, payload, (size_t)PFKEY_UNUNIT64(msg_payload->sadb_msg_len), 0), "pfkey flush sa");
216*94d3b452SApple OSS Distributions}
217*94d3b452SApple OSS Distributions
218*94d3b452SApple OSS Distributionsstatic void
219*94d3b452SApple OSS Distributionspfkey_cleanup(void)
220*94d3b452SApple OSS Distributions{
221*94d3b452SApple OSS Distributions	if (pfkey_source != NULL) {
222*94d3b452SApple OSS Distributions		int pfkey_socket = (int)dispatch_source_get_handle(pfkey_source);
223*94d3b452SApple OSS Distributions		if (pfkey_socket > 0) {
224*94d3b452SApple OSS Distributions			send_pfkey_flush_sa(pfkey_socket);
225*94d3b452SApple OSS Distributions		}
226*94d3b452SApple OSS Distributions		dispatch_source_cancel(pfkey_source);
227*94d3b452SApple OSS Distributions		pfkey_source = NULL;
228*94d3b452SApple OSS Distributions	}
229*94d3b452SApple OSS Distributions}
230*94d3b452SApple OSS Distributions
231*94d3b452SApple OSS Distributionsstatic void
232*94d3b452SApple OSS Distributionspfkey_align(struct sadb_msg *msg, uint8_t **mhp)
233*94d3b452SApple OSS Distributions{
234*94d3b452SApple OSS Distributions	struct sadb_ext *ext;
235*94d3b452SApple OSS Distributions	int i;
236*94d3b452SApple OSS Distributions	uint8_t *p;
237*94d3b452SApple OSS Distributions	uint8_t *ep;     /* XXX should be passed from upper layer */
238*94d3b452SApple OSS Distributions
239*94d3b452SApple OSS Distributions	/* validity check */
240*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(msg, "pfkey align msg");
241*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(mhp, "pfkey align mhp");
242*94d3b452SApple OSS Distributions
243*94d3b452SApple OSS Distributions	/* initialize */
244*94d3b452SApple OSS Distributions	for (i = 0; i < SADB_EXT_MAX + 1; i++) {
245*94d3b452SApple OSS Distributions		mhp[i] = NULL;
246*94d3b452SApple OSS Distributions	}
247*94d3b452SApple OSS Distributions
248*94d3b452SApple OSS Distributions	mhp[0] = (void *)msg;
249*94d3b452SApple OSS Distributions
250*94d3b452SApple OSS Distributions	/* initialize */
251*94d3b452SApple OSS Distributions	p = (void *) msg;
252*94d3b452SApple OSS Distributions	ep = p + PFKEY_UNUNIT64(msg->sadb_msg_len);
253*94d3b452SApple OSS Distributions
254*94d3b452SApple OSS Distributions	/* skip base header */
255*94d3b452SApple OSS Distributions	p += sizeof(struct sadb_msg);
256*94d3b452SApple OSS Distributions
257*94d3b452SApple OSS Distributions	while (p < ep) {
258*94d3b452SApple OSS Distributions		ext = (void *)p;
259*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_GE_PTR((void *)ep, (void *)(p + sizeof(*ext)), "pfkey extension header beyond end of buffer");
260*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_GE_ULONG((unsigned long)PFKEY_EXTLEN(ext), sizeof(*ext), "pfkey extension shorter than extension header");
261*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_GE_PTR((void *)ep, (void *)(p + PFKEY_EXTLEN(ext)), "pfkey extension length beyond end of buffer");
262*94d3b452SApple OSS Distributions
263*94d3b452SApple OSS Distributions		T_QUIET; T_EXPECT_NULL(mhp[ext->sadb_ext_type], "duplicate extension type %u payload", ext->sadb_ext_type);
264*94d3b452SApple OSS Distributions
265*94d3b452SApple OSS Distributions		/* set pointer */
266*94d3b452SApple OSS Distributions		switch (ext->sadb_ext_type) {
267*94d3b452SApple OSS Distributions		case SADB_EXT_SA:
268*94d3b452SApple OSS Distributions		case SADB_EXT_LIFETIME_CURRENT:
269*94d3b452SApple OSS Distributions		case SADB_EXT_LIFETIME_HARD:
270*94d3b452SApple OSS Distributions		case SADB_EXT_LIFETIME_SOFT:
271*94d3b452SApple OSS Distributions		case SADB_EXT_ADDRESS_SRC:
272*94d3b452SApple OSS Distributions		case SADB_EXT_ADDRESS_DST:
273*94d3b452SApple OSS Distributions		case SADB_EXT_ADDRESS_PROXY:
274*94d3b452SApple OSS Distributions		case SADB_EXT_KEY_AUTH:
275*94d3b452SApple OSS Distributions		/* XXX should to be check weak keys. */
276*94d3b452SApple OSS Distributions		case SADB_EXT_KEY_ENCRYPT:
277*94d3b452SApple OSS Distributions		/* XXX should to be check weak keys. */
278*94d3b452SApple OSS Distributions		case SADB_EXT_IDENTITY_SRC:
279*94d3b452SApple OSS Distributions		case SADB_EXT_IDENTITY_DST:
280*94d3b452SApple OSS Distributions		case SADB_EXT_SENSITIVITY:
281*94d3b452SApple OSS Distributions		case SADB_EXT_PROPOSAL:
282*94d3b452SApple OSS Distributions		case SADB_EXT_SUPPORTED_AUTH:
283*94d3b452SApple OSS Distributions		case SADB_EXT_SUPPORTED_ENCRYPT:
284*94d3b452SApple OSS Distributions		case SADB_EXT_SPIRANGE:
285*94d3b452SApple OSS Distributions		case SADB_X_EXT_POLICY:
286*94d3b452SApple OSS Distributions		case SADB_X_EXT_SA2:
287*94d3b452SApple OSS Distributions		case SADB_EXT_SESSION_ID:
288*94d3b452SApple OSS Distributions		case SADB_EXT_SASTAT:
289*94d3b452SApple OSS Distributions#ifdef SADB_X_EXT_NAT_T_TYPE
290*94d3b452SApple OSS Distributions		case SADB_X_EXT_NAT_T_TYPE:
291*94d3b452SApple OSS Distributions		case SADB_X_EXT_NAT_T_SPORT:
292*94d3b452SApple OSS Distributions		case SADB_X_EXT_NAT_T_DPORT:
293*94d3b452SApple OSS Distributions		case SADB_X_EXT_NAT_T_OA:
294*94d3b452SApple OSS Distributions#endif
295*94d3b452SApple OSS Distributions#ifdef SADB_X_EXT_TAG
296*94d3b452SApple OSS Distributions		case SADB_X_EXT_TAG:
297*94d3b452SApple OSS Distributions#endif
298*94d3b452SApple OSS Distributions#ifdef SADB_X_EXT_PACKET
299*94d3b452SApple OSS Distributions		case SADB_X_EXT_PACKET:
300*94d3b452SApple OSS Distributions#endif
301*94d3b452SApple OSS Distributions		case SADB_X_EXT_IPSECIF:
302*94d3b452SApple OSS Distributions		case SADB_X_EXT_ADDR_RANGE_SRC_START:
303*94d3b452SApple OSS Distributions		case SADB_X_EXT_ADDR_RANGE_SRC_END:
304*94d3b452SApple OSS Distributions		case SADB_X_EXT_ADDR_RANGE_DST_START:
305*94d3b452SApple OSS Distributions		case SADB_X_EXT_ADDR_RANGE_DST_END:
306*94d3b452SApple OSS Distributions#ifdef SADB_MIGRATE
307*94d3b452SApple OSS Distributions		case SADB_EXT_MIGRATE_ADDRESS_SRC:
308*94d3b452SApple OSS Distributions		case SADB_EXT_MIGRATE_ADDRESS_DST:
309*94d3b452SApple OSS Distributions		case SADB_X_EXT_MIGRATE_IPSECIF:
310*94d3b452SApple OSS Distributions#endif
311*94d3b452SApple OSS Distributions			mhp[ext->sadb_ext_type] = (void *)ext;
312*94d3b452SApple OSS Distributions			break;
313*94d3b452SApple OSS Distributions		default:
314*94d3b452SApple OSS Distributions			T_FAIL("bad extension type %u", ext->sadb_ext_type);
315*94d3b452SApple OSS Distributions			T_END;
316*94d3b452SApple OSS Distributions		}
317*94d3b452SApple OSS Distributions
318*94d3b452SApple OSS Distributions		p += PFKEY_EXTLEN(ext);
319*94d3b452SApple OSS Distributions	}
320*94d3b452SApple OSS Distributions
321*94d3b452SApple OSS Distributions	T_QUIET; T_EXPECT_EQ_PTR((void *)ep, (void *)p, "invalid pfkey message length");
322*94d3b452SApple OSS Distributions	return;
323*94d3b452SApple OSS Distributions}
324*94d3b452SApple OSS Distributions
325*94d3b452SApple OSS Distributionsstatic void
326*94d3b452SApple OSS Distributionspfkey_process_message_test_encapsulate_panic(uint8_t **mhp, __unused int pfkey_socket)
327*94d3b452SApple OSS Distributions{
328*94d3b452SApple OSS Distributions	struct sadb_msg *message = (struct sadb_msg *)(void *)mhp[0];
329*94d3b452SApple OSS Distributions	static uint32_t spi = 0;
330*94d3b452SApple OSS Distributions	static uint8_t added_sa_counter = 0;
331*94d3b452SApple OSS Distributions
332*94d3b452SApple OSS Distributions	if (message->sadb_msg_pid != (uint32_t)getpid()) {
333*94d3b452SApple OSS Distributions		return;
334*94d3b452SApple OSS Distributions	}
335*94d3b452SApple OSS Distributions
336*94d3b452SApple OSS Distributions	if (message->sadb_msg_errno != 0) {
337*94d3b452SApple OSS Distributions		T_FAIL("SADB add SA received error %d", message->sadb_msg_errno);
338*94d3b452SApple OSS Distributions		T_END;
339*94d3b452SApple OSS Distributions	}
340*94d3b452SApple OSS Distributions
341*94d3b452SApple OSS Distributions	switch (message->sadb_msg_type) {
342*94d3b452SApple OSS Distributions	case SADB_ADD:
343*94d3b452SApple OSS Distributions	{
344*94d3b452SApple OSS Distributions		struct sadb_sa *sa_message = (struct sadb_sa *)(void *)mhp[SADB_EXT_SA];
345*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_NOTNULL(sa_message, "add sa message is NULL");
346*94d3b452SApple OSS Distributions		spi = ntohl(sa_message->sadb_sa_spi);
347*94d3b452SApple OSS Distributions		T_LOG("added sa 0x%x", spi);
348*94d3b452SApple OSS Distributions		added_sa_counter++;
349*94d3b452SApple OSS Distributions		if (added_sa_counter == 2) {
350*94d3b452SApple OSS Distributions			bpf_write(bpf_fd);
351*94d3b452SApple OSS Distributions		}
352*94d3b452SApple OSS Distributions		break;
353*94d3b452SApple OSS Distributions	}
354*94d3b452SApple OSS Distributions	case SADB_FLUSH:
355*94d3b452SApple OSS Distributions	case SADB_X_SPDFLUSH:
356*94d3b452SApple OSS Distributions		break;
357*94d3b452SApple OSS Distributions	default:
358*94d3b452SApple OSS Distributions		T_FAIL("bad SADB message type %u", message->sadb_msg_type);
359*94d3b452SApple OSS Distributions		T_END;
360*94d3b452SApple OSS Distributions	}
361*94d3b452SApple OSS Distributions	return;
362*94d3b452SApple OSS Distributions}
363*94d3b452SApple OSS Distributions
364*94d3b452SApple OSS Distributionsstatic void
365*94d3b452SApple OSS Distributionsrecv_pfkey_message(int pfkey_socket)
366*94d3b452SApple OSS Distributions{
367*94d3b452SApple OSS Distributions	uint8_t buffer[8192] __attribute__((aligned(4)));
368*94d3b452SApple OSS Distributions	struct iovec iovecs[1] = {
369*94d3b452SApple OSS Distributions		{ buffer, sizeof(buffer) },
370*94d3b452SApple OSS Distributions	};
371*94d3b452SApple OSS Distributions	struct msghdr msg = {
372*94d3b452SApple OSS Distributions		NULL,
373*94d3b452SApple OSS Distributions		0,
374*94d3b452SApple OSS Distributions		iovecs,
375*94d3b452SApple OSS Distributions		sizeof(iovecs) / sizeof(iovecs[0]),
376*94d3b452SApple OSS Distributions		NULL,
377*94d3b452SApple OSS Distributions		0,
378*94d3b452SApple OSS Distributions		0,
379*94d3b452SApple OSS Distributions	};
380*94d3b452SApple OSS Distributions
381*94d3b452SApple OSS Distributions	do {
382*94d3b452SApple OSS Distributions		ssize_t result = -1;
383*94d3b452SApple OSS Distributions		memset(buffer, 0, sizeof(buffer));
384*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_POSIX_SUCCESS(result = recvmsg(pfkey_socket, &msg, 0), NULL);
385*94d3b452SApple OSS Distributions
386*94d3b452SApple OSS Distributions		if (result > 0) {
387*94d3b452SApple OSS Distributions			T_QUIET; T_ASSERT_GE_ULONG((size_t)result, sizeof(struct sadb_msg), "Invalid PFKey message size: %zu", result);
388*94d3b452SApple OSS Distributions			struct sadb_msg *hdr = (struct sadb_msg *)buffer;
389*94d3b452SApple OSS Distributions			uint8_t *mhp[SADB_EXT_MAX + 1];
390*94d3b452SApple OSS Distributions			pfkey_align(hdr, mhp);
391*94d3b452SApple OSS Distributions			(*process_pfkey_message_tests[test_id])(mhp, pfkey_socket);
392*94d3b452SApple OSS Distributions		} else if (result == 0) {
393*94d3b452SApple OSS Distributions			T_LOG("PFKey socket received EOF");
394*94d3b452SApple OSS Distributions			break;
395*94d3b452SApple OSS Distributions		}
396*94d3b452SApple OSS Distributions	} while (1);
397*94d3b452SApple OSS Distributions}
398*94d3b452SApple OSS Distributions
399*94d3b452SApple OSS Distributionsstatic int
400*94d3b452SApple OSS Distributionspfkey_setup_socket(void)
401*94d3b452SApple OSS Distributions{
402*94d3b452SApple OSS Distributions	int pfkey_socket = -1;
403*94d3b452SApple OSS Distributions
404*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_POSIX_SUCCESS(pfkey_socket = socket(PF_KEY, SOCK_RAW, PF_KEY_V2), NULL);
405*94d3b452SApple OSS Distributions
406*94d3b452SApple OSS Distributions	struct mbstat mbstat;
407*94d3b452SApple OSS Distributions	size_t len_mbstat = sizeof(struct mbstat);
408*94d3b452SApple OSS Distributions
409*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_POSIX_SUCCESS(sysctlbyname("kern.ipc.mbstat", &mbstat, &len_mbstat, NULL, 0), "kern.ipc.mbstat");
410*94d3b452SApple OSS Distributions
411*94d3b452SApple OSS Distributions	int bufsiz = 0;
412*94d3b452SApple OSS Distributions	const unsigned long newbufk = 1536;
413*94d3b452SApple OSS Distributions	unsigned long oldmax;
414*94d3b452SApple OSS Distributions	size_t  oldmaxsize = sizeof(oldmax);
415*94d3b452SApple OSS Distributions	unsigned long newmax = newbufk * (1024 + mbstat.m_msize / 2);
416*94d3b452SApple OSS Distributions
417*94d3b452SApple OSS Distributions	if (sysctlbyname("kern.ipc.maxsockbuf", &oldmax, &oldmaxsize, &newmax, sizeof(newmax)) != 0) {
418*94d3b452SApple OSS Distributions		bufsiz = 233016;        /* Max allowed by default */
419*94d3b452SApple OSS Distributions	} else {
420*94d3b452SApple OSS Distributions		bufsiz = newbufk * 1024;
421*94d3b452SApple OSS Distributions	}
422*94d3b452SApple OSS Distributions
423*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_POSIX_SUCCESS(setsockopt(pfkey_socket, SOL_SOCKET, SO_SNDBUF, &bufsiz, sizeof(bufsiz)), "pfkey set snd socket buf failed %d", bufsiz);
424*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_POSIX_SUCCESS(setsockopt(pfkey_socket, SOL_SOCKET, SO_RCVBUF, &bufsiz, sizeof(bufsiz)), "pfkey set recv socket buf failed %d", bufsiz);
425*94d3b452SApple OSS Distributions
426*94d3b452SApple OSS Distributions	pfkey_source = dispatch_source_create(DISPATCH_SOURCE_TYPE_READ, (uintptr_t)pfkey_socket, 0, dispatch_get_main_queue());
427*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(pfkey_source, "dispatch_source_create(DISPATCH_SOURCE_TYPE_READ, ...)");
428*94d3b452SApple OSS Distributions	dispatch_source_set_event_handler(pfkey_source, ^{
429*94d3b452SApple OSS Distributions		recv_pfkey_message(pfkey_socket);
430*94d3b452SApple OSS Distributions	});
431*94d3b452SApple OSS Distributions	dispatch_source_set_cancel_handler(pfkey_source, ^{
432*94d3b452SApple OSS Distributions		close(pfkey_socket);
433*94d3b452SApple OSS Distributions	});
434*94d3b452SApple OSS Distributions	dispatch_resume(pfkey_source);
435*94d3b452SApple OSS Distributions	return pfkey_socket;
436*94d3b452SApple OSS Distributions}
437*94d3b452SApple OSS Distributions
438*94d3b452SApple OSS Distributionsstatic int
439*94d3b452SApple OSS Distributionsbpf_new(void)
440*94d3b452SApple OSS Distributions{
441*94d3b452SApple OSS Distributions	char bpfdev[256];
442*94d3b452SApple OSS Distributions	int i;
443*94d3b452SApple OSS Distributions	int fd = -1;
444*94d3b452SApple OSS Distributions
445*94d3b452SApple OSS Distributions	for (i = 0; true; i++) {
446*94d3b452SApple OSS Distributions		snprintf(bpfdev, sizeof(bpfdev), "/dev/bpf%d", i);
447*94d3b452SApple OSS Distributions		fd = open(bpfdev, O_RDWR, 0);
448*94d3b452SApple OSS Distributions		if (fd >= 0) {
449*94d3b452SApple OSS Distributions			break;
450*94d3b452SApple OSS Distributions		}
451*94d3b452SApple OSS Distributions		if (errno != EBUSY) {
452*94d3b452SApple OSS Distributions			break;
453*94d3b452SApple OSS Distributions		}
454*94d3b452SApple OSS Distributions	}
455*94d3b452SApple OSS Distributions	return fd;
456*94d3b452SApple OSS Distributions}
457*94d3b452SApple OSS Distributions
458*94d3b452SApple OSS Distributionsstatic int
459*94d3b452SApple OSS Distributionsbpf_setif(int fd, const char *en_name)
460*94d3b452SApple OSS Distributions{
461*94d3b452SApple OSS Distributions	struct ifreq ifr;
462*94d3b452SApple OSS Distributions
463*94d3b452SApple OSS Distributions	strlcpy(ifr.ifr_name, en_name, sizeof(ifr.ifr_name));
464*94d3b452SApple OSS Distributions	return ioctl(fd, BIOCSETIF, &ifr);
465*94d3b452SApple OSS Distributions}
466*94d3b452SApple OSS Distributions
467*94d3b452SApple OSS Distributionsstatic int
468*94d3b452SApple OSS Distributionsbpf_sethdr_complete(int fd)
469*94d3b452SApple OSS Distributions{
470*94d3b452SApple OSS Distributions	u_int8_t hdr_complete = 1;
471*94d3b452SApple OSS Distributions	return ioctl(fd, BIOCSHDRCMPLT, &hdr_complete);
472*94d3b452SApple OSS Distributions}
473*94d3b452SApple OSS Distributions
474*94d3b452SApple OSS Distributionsstatic void
475*94d3b452SApple OSS Distributionsbpf_write(int fd)
476*94d3b452SApple OSS Distributions{
477*94d3b452SApple OSS Distributions	if (test_id == TEST_IPSEC_IPv4_ENCAPSULATE_PANIC) {
478*94d3b452SApple OSS Distributions		char buffer[500];
479*94d3b452SApple OSS Distributions		struct ip *ipheader = (void *)buffer;
480*94d3b452SApple OSS Distributions		ipheader->ip_v = IPVERSION;
481*94d3b452SApple OSS Distributions		ipheader->ip_hl = (sizeof(struct ip) - 4) >> 2;
482*94d3b452SApple OSS Distributions		ipheader->ip_ttl = MAXTTL;
483*94d3b452SApple OSS Distributions		ipheader->ip_p = IPPROTO_UDP;
484*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_POSIX_SUCCESS(write(fd, buffer, 500), "bpf write call failed");
485*94d3b452SApple OSS Distributions		T_PASS("wrote bad ip header successfully");
486*94d3b452SApple OSS Distributions		T_END;
487*94d3b452SApple OSS Distributions	} else if (test_id == TEST_IPSEC_IPv6_ENCAPSULATE_PANIC) {
488*94d3b452SApple OSS Distributions		struct ip6_hdr ip6 = {0};
489*94d3b452SApple OSS Distributions		ip6.ip6_vfc |= IPV6_VERSION;
490*94d3b452SApple OSS Distributions		T_QUIET; T_ASSERT_POSIX_SUCCESS(write(fd, &ip6, sizeof(ip6) - 4), "bpf write call failed");
491*94d3b452SApple OSS Distributions		T_PASS("wrote bad ipv6 header successfully");
492*94d3b452SApple OSS Distributions		T_END;
493*94d3b452SApple OSS Distributions	}
494*94d3b452SApple OSS Distributions}
495*94d3b452SApple OSS Distributions
496*94d3b452SApple OSS Distributionsstatic void
497*94d3b452SApple OSS Distributionsbpf_socket_setup(void)
498*94d3b452SApple OSS Distributions{
499*94d3b452SApple OSS Distributions	int status = -1;
500*94d3b452SApple OSS Distributions
501*94d3b452SApple OSS Distributions	bpf_fd = bpf_new();
502*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NE(bpf_fd, -1, "failed to create bpf file descriptor");
503*94d3b452SApple OSS Distributions
504*94d3b452SApple OSS Distributions	CFStringRef ipsecIfName = NEVirtualInterfaceCopyName(ipsecInterface);
505*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(ipsecIfName, "failed to get ipsec interface name");
506*94d3b452SApple OSS Distributions
507*94d3b452SApple OSS Distributions	char ifname[IFNAMSIZ];
508*94d3b452SApple OSS Distributions	CFStringGetCString(ipsecIfName, ifname, IFNAMSIZ, kCFStringEncodingUTF8);
509*94d3b452SApple OSS Distributions
510*94d3b452SApple OSS Distributions	status = bpf_setif(bpf_fd, ifname);
511*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NE(status, -1, "failed to set bpf interface");
512*94d3b452SApple OSS Distributions
513*94d3b452SApple OSS Distributions	status = bpf_sethdr_complete(bpf_fd);
514*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NE(status, -1, "failed to set bpf header complete");
515*94d3b452SApple OSS Distributions}
516*94d3b452SApple OSS Distributions
517*94d3b452SApple OSS Distributionsstatic NEVirtualInterfaceRef
518*94d3b452SApple OSS Distributionsipsec_interface_setup(CFStringRef interfaceAddress, CFStringRef interfaceMask)
519*94d3b452SApple OSS Distributions{
520*94d3b452SApple OSS Distributions	Boolean status = FALSE;
521*94d3b452SApple OSS Distributions
522*94d3b452SApple OSS Distributions	NEVirtualInterfaceRef interface = NEVirtualInterfaceCreate(NULL, kNEVirtualInterfaceValTypeIPSec, dispatch_get_main_queue(), NULL);
523*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(interface, "ipsec interface creation failed");
524*94d3b452SApple OSS Distributions	status = NEVirtualInterfaceSetMTU(interface, 1400);
525*94d3b452SApple OSS Distributions	if (status == FALSE) {
526*94d3b452SApple OSS Distributions		T_FAIL("Failed to set MTU on ipsec interface");
527*94d3b452SApple OSS Distributions		T_END;
528*94d3b452SApple OSS Distributions	}
529*94d3b452SApple OSS Distributions
530*94d3b452SApple OSS Distributions	status = NEVirtualInterfaceAddAddress(interface, interfaceAddress, interfaceMask);
531*94d3b452SApple OSS Distributions	if (status == FALSE) {
532*94d3b452SApple OSS Distributions		T_FAIL("Failed to set address on ipsec interface");
533*94d3b452SApple OSS Distributions		T_END;
534*94d3b452SApple OSS Distributions	}
535*94d3b452SApple OSS Distributions
536*94d3b452SApple OSS Distributions	CFStringRef ipsecIfName = NEVirtualInterfaceCopyName(interface);
537*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(ipsecIfName, "failed to get ipsec interface name");
538*94d3b452SApple OSS Distributions
539*94d3b452SApple OSS Distributions	char ifname[IFNAMSIZ];
540*94d3b452SApple OSS Distributions	CFStringGetCString(ipsecIfName, ifname, IFNAMSIZ, kCFStringEncodingUTF8);
541*94d3b452SApple OSS Distributions
542*94d3b452SApple OSS Distributions	T_LOG("%s interface setup", ifname);
543*94d3b452SApple OSS Distributions	return interface;
544*94d3b452SApple OSS Distributions}
545*94d3b452SApple OSS Distributions
546*94d3b452SApple OSS Distributionsstatic void
547*94d3b452SApple OSS Distributionsipsec_interface_set_delegate(NEVirtualInterfaceRef interface, CFStringRef delegateInterfaceName)
548*94d3b452SApple OSS Distributions{
549*94d3b452SApple OSS Distributions	Boolean status = NEVirtualInterfaceSetDelegateInterface(interface, delegateInterfaceName);
550*94d3b452SApple OSS Distributions	if (status == FALSE) {
551*94d3b452SApple OSS Distributions		T_FAIL("Failed to set delegate on ipsec interface");
552*94d3b452SApple OSS Distributions		T_END;
553*94d3b452SApple OSS Distributions	}
554*94d3b452SApple OSS Distributions
555*94d3b452SApple OSS Distributions	return;
556*94d3b452SApple OSS Distributions}
557*94d3b452SApple OSS Distributions
558*94d3b452SApple OSS Distributionsstatic void
559*94d3b452SApple OSS Distributionsipsec_cleanup(void)
560*94d3b452SApple OSS Distributions{
561*94d3b452SApple OSS Distributions	pfkey_cleanup();
562*94d3b452SApple OSS Distributions
563*94d3b452SApple OSS Distributions	if (ipsecInterface != NULL) {
564*94d3b452SApple OSS Distributions		NEVirtualInterfaceInvalidate(ipsecInterface);
565*94d3b452SApple OSS Distributions		ipsecInterface = NULL;
566*94d3b452SApple OSS Distributions	}
567*94d3b452SApple OSS Distributions
568*94d3b452SApple OSS Distributions	if (delegateIPsecInterface != NULL) {
569*94d3b452SApple OSS Distributions		NEVirtualInterfaceInvalidate(delegateIPsecInterface);
570*94d3b452SApple OSS Distributions		delegateIPsecInterface = NULL;
571*94d3b452SApple OSS Distributions	}
572*94d3b452SApple OSS Distributions
573*94d3b452SApple OSS Distributions	if (bpf_fd != -1) {
574*94d3b452SApple OSS Distributions		close(bpf_fd);
575*94d3b452SApple OSS Distributions		bpf_fd = -1;
576*94d3b452SApple OSS Distributions	}
577*94d3b452SApple OSS Distributions}
578*94d3b452SApple OSS Distributions
579*94d3b452SApple OSS DistributionsT_DECL(ipsec_ipv4_encapsulate_panic_63139357, "ipsec: outer ip header length less than 20")
580*94d3b452SApple OSS Distributions{
581*94d3b452SApple OSS Distributions	test_id = TEST_IPSEC_IPv4_ENCAPSULATE_PANIC;
582*94d3b452SApple OSS Distributions
583*94d3b452SApple OSS Distributions	T_ATEND(ipsec_cleanup);
584*94d3b452SApple OSS Distributions
585*94d3b452SApple OSS Distributions	ipsecInterface = ipsec_interface_setup(CFSTR(TEST_IPSEC_IPv4_INTERFACE_ADDRESS), CFSTR(TEST_IPSEC_IPv4_INTERFACE_MASK));
586*94d3b452SApple OSS Distributions	delegateIPsecInterface = ipsec_interface_setup(CFSTR(TEST_DELEGATE_IPSEC_INTERFACE_ADDRESS), CFSTR(TEST_IPSEC_IPv4_INTERFACE_MASK));
587*94d3b452SApple OSS Distributions
588*94d3b452SApple OSS Distributions	CFStringRef delegateIPsecIfName = NEVirtualInterfaceCopyName(delegateIPsecInterface);
589*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(delegateIPsecIfName, "failed to get ipsec interface name");
590*94d3b452SApple OSS Distributions	ipsec_interface_set_delegate(ipsecInterface, delegateIPsecIfName);
591*94d3b452SApple OSS Distributions
592*94d3b452SApple OSS Distributions	bpf_socket_setup();
593*94d3b452SApple OSS Distributions
594*94d3b452SApple OSS Distributions	int pfkey_socket = pfkey_setup_socket();
595*94d3b452SApple OSS Distributions	send_pfkey_flush_sa(pfkey_socket);
596*94d3b452SApple OSS Distributions
597*94d3b452SApple OSS Distributions	send_pkey_add_sa(pfkey_socket, 0x12345678, TEST_SRC_ADDRESS_IPv4, TEST_DST_ADDRESS_IPv4, AF_INET);
598*94d3b452SApple OSS Distributions	send_pkey_add_sa(pfkey_socket, 0x23456789, TEST_SRC_ADDRESS_IPv4, TEST_DST_ADDRESS_IPv4, AF_INET);
599*94d3b452SApple OSS Distributions
600*94d3b452SApple OSS Distributions	dispatch_main();
601*94d3b452SApple OSS Distributions}
602*94d3b452SApple OSS Distributions
603*94d3b452SApple OSS DistributionsT_DECL(ipsec_ipv6_encapsulate_panic_63139357, "ipsec: payload less than IPv6 header")
604*94d3b452SApple OSS Distributions{
605*94d3b452SApple OSS Distributions	test_id = TEST_IPSEC_IPv6_ENCAPSULATE_PANIC;
606*94d3b452SApple OSS Distributions
607*94d3b452SApple OSS Distributions	T_ATEND(ipsec_cleanup);
608*94d3b452SApple OSS Distributions
609*94d3b452SApple OSS Distributions	ipsecInterface = ipsec_interface_setup(CFSTR(TEST_IPSEC_IPv6_INTERFACE_ADDRESS), CFSTR(TEST_IPSEC_IPv6_INTERFACE_MASK));
610*94d3b452SApple OSS Distributions	delegateIPsecInterface = ipsec_interface_setup(CFSTR(TEST_DELEGATE_IPSEC_INTERFACE_ADDRESS), CFSTR(TEST_IPSEC_IPv4_INTERFACE_MASK));
611*94d3b452SApple OSS Distributions
612*94d3b452SApple OSS Distributions	CFStringRef delegateIPsecIfName = NEVirtualInterfaceCopyName(delegateIPsecInterface);
613*94d3b452SApple OSS Distributions	T_QUIET; T_ASSERT_NOTNULL(delegateIPsecIfName, "failed to get ipsec interface name");
614*94d3b452SApple OSS Distributions	ipsec_interface_set_delegate(ipsecInterface, delegateIPsecIfName);
615*94d3b452SApple OSS Distributions
616*94d3b452SApple OSS Distributions	bpf_socket_setup();
617*94d3b452SApple OSS Distributions
618*94d3b452SApple OSS Distributions	int pfkey_socket = pfkey_setup_socket();
619*94d3b452SApple OSS Distributions	send_pfkey_flush_sa(pfkey_socket);
620*94d3b452SApple OSS Distributions
621*94d3b452SApple OSS Distributions	send_pkey_add_sa(pfkey_socket, 0x12345678, TEST_SRC_ADDRESS_IPv4, TEST_DST_ADDRESS_IPv4, AF_INET);
622*94d3b452SApple OSS Distributions	send_pkey_add_sa(pfkey_socket, 0x23456789, TEST_SRC_ADDRESS_IPv4, TEST_DST_ADDRESS_IPv4, AF_INET);
623*94d3b452SApple OSS Distributions
624*94d3b452SApple OSS Distributions	dispatch_main();
625*94d3b452SApple OSS Distributions}
626