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