xref: /netbsd-src/external/bsd/wpa/dist/src/eap_common/ikev2_common.c (revision 36ebd06e5ab61115eab7acca17a2350fc8222071)
18dbcf02cSchristos /*
28dbcf02cSchristos  * IKEv2 common routines for initiator and responder
38dbcf02cSchristos  * Copyright (c) 2007, Jouni Malinen <j@w1.fi>
48dbcf02cSchristos  *
5e604d861Schristos  * This software may be distributed under the terms of the BSD license.
6e604d861Schristos  * See README for more details.
78dbcf02cSchristos  */
88dbcf02cSchristos 
98dbcf02cSchristos #include "includes.h"
108dbcf02cSchristos 
118dbcf02cSchristos #include "common.h"
128dbcf02cSchristos #include "crypto/crypto.h"
138dbcf02cSchristos #include "crypto/md5.h"
148dbcf02cSchristos #include "crypto/sha1.h"
15111b9fd8Schristos #include "crypto/random.h"
168dbcf02cSchristos #include "ikev2_common.h"
178dbcf02cSchristos 
188dbcf02cSchristos 
19*36ebd06eSchristos static const struct ikev2_integ_alg ikev2_integ_algs[] = {
208dbcf02cSchristos 	{ AUTH_HMAC_SHA1_96, 20, 12 },
218dbcf02cSchristos 	{ AUTH_HMAC_MD5_96, 16, 12 }
228dbcf02cSchristos };
238dbcf02cSchristos 
243c260e60Schristos #define NUM_INTEG_ALGS ARRAY_SIZE(ikev2_integ_algs)
258dbcf02cSchristos 
268dbcf02cSchristos 
27*36ebd06eSchristos static const struct ikev2_prf_alg ikev2_prf_algs[] = {
288dbcf02cSchristos 	{ PRF_HMAC_SHA1, 20, 20 },
298dbcf02cSchristos 	{ PRF_HMAC_MD5, 16, 16 }
308dbcf02cSchristos };
318dbcf02cSchristos 
323c260e60Schristos #define NUM_PRF_ALGS ARRAY_SIZE(ikev2_prf_algs)
338dbcf02cSchristos 
348dbcf02cSchristos 
35*36ebd06eSchristos static const struct ikev2_encr_alg ikev2_encr_algs[] = {
368dbcf02cSchristos 	{ ENCR_AES_CBC, 16, 16 }, /* only 128-bit keys supported for now */
378dbcf02cSchristos 	{ ENCR_3DES, 24, 8 }
388dbcf02cSchristos };
398dbcf02cSchristos 
403c260e60Schristos #define NUM_ENCR_ALGS ARRAY_SIZE(ikev2_encr_algs)
418dbcf02cSchristos 
428dbcf02cSchristos 
ikev2_get_integ(int id)438dbcf02cSchristos const struct ikev2_integ_alg * ikev2_get_integ(int id)
448dbcf02cSchristos {
458dbcf02cSchristos 	size_t i;
468dbcf02cSchristos 
478dbcf02cSchristos 	for (i = 0; i < NUM_INTEG_ALGS; i++) {
488dbcf02cSchristos 		if (ikev2_integ_algs[i].id == id)
498dbcf02cSchristos 			return &ikev2_integ_algs[i];
508dbcf02cSchristos 	}
518dbcf02cSchristos 
528dbcf02cSchristos 	return NULL;
538dbcf02cSchristos }
548dbcf02cSchristos 
558dbcf02cSchristos 
ikev2_integ_hash(int alg,const u8 * key,size_t key_len,const u8 * data,size_t data_len,u8 * hash)568dbcf02cSchristos int ikev2_integ_hash(int alg, const u8 *key, size_t key_len, const u8 *data,
578dbcf02cSchristos 		     size_t data_len, u8 *hash)
588dbcf02cSchristos {
598dbcf02cSchristos 	u8 tmphash[IKEV2_MAX_HASH_LEN];
608dbcf02cSchristos 
618dbcf02cSchristos 	switch (alg) {
628dbcf02cSchristos 	case AUTH_HMAC_SHA1_96:
638dbcf02cSchristos 		if (key_len != 20)
648dbcf02cSchristos 			return -1;
65*36ebd06eSchristos 		if (hmac_sha1(key, key_len, data, data_len, tmphash) < 0)
66*36ebd06eSchristos 			return -1;
678dbcf02cSchristos 		os_memcpy(hash, tmphash, 12);
688dbcf02cSchristos 		break;
698dbcf02cSchristos 	case AUTH_HMAC_MD5_96:
708dbcf02cSchristos 		if (key_len != 16)
718dbcf02cSchristos 			return -1;
72*36ebd06eSchristos 		if (hmac_md5(key, key_len, data, data_len, tmphash) < 0)
73*36ebd06eSchristos 			return -1;
748dbcf02cSchristos 		os_memcpy(hash, tmphash, 12);
758dbcf02cSchristos 		break;
768dbcf02cSchristos 	default:
778dbcf02cSchristos 		return -1;
788dbcf02cSchristos 	}
798dbcf02cSchristos 
808dbcf02cSchristos 	return 0;
818dbcf02cSchristos }
828dbcf02cSchristos 
838dbcf02cSchristos 
ikev2_get_prf(int id)848dbcf02cSchristos const struct ikev2_prf_alg * ikev2_get_prf(int id)
858dbcf02cSchristos {
868dbcf02cSchristos 	size_t i;
878dbcf02cSchristos 
888dbcf02cSchristos 	for (i = 0; i < NUM_PRF_ALGS; i++) {
898dbcf02cSchristos 		if (ikev2_prf_algs[i].id == id)
908dbcf02cSchristos 			return &ikev2_prf_algs[i];
918dbcf02cSchristos 	}
928dbcf02cSchristos 
938dbcf02cSchristos 	return NULL;
948dbcf02cSchristos }
958dbcf02cSchristos 
968dbcf02cSchristos 
ikev2_prf_hash(int alg,const u8 * key,size_t key_len,size_t num_elem,const u8 * addr[],const size_t * len,u8 * hash)978dbcf02cSchristos int ikev2_prf_hash(int alg, const u8 *key, size_t key_len,
988dbcf02cSchristos 		   size_t num_elem, const u8 *addr[], const size_t *len,
998dbcf02cSchristos 		   u8 *hash)
1008dbcf02cSchristos {
1018dbcf02cSchristos 	switch (alg) {
1028dbcf02cSchristos 	case PRF_HMAC_SHA1:
103*36ebd06eSchristos 		return hmac_sha1_vector(key, key_len, num_elem, addr, len,
104*36ebd06eSchristos 					hash);
1058dbcf02cSchristos 	case PRF_HMAC_MD5:
106*36ebd06eSchristos 		return hmac_md5_vector(key, key_len, num_elem, addr, len, hash);
1078dbcf02cSchristos 	default:
1088dbcf02cSchristos 		return -1;
1098dbcf02cSchristos 	}
1108dbcf02cSchristos }
1118dbcf02cSchristos 
1128dbcf02cSchristos 
ikev2_prf_plus(int alg,const u8 * key,size_t key_len,const u8 * data,size_t data_len,u8 * out,size_t out_len)1138dbcf02cSchristos int ikev2_prf_plus(int alg, const u8 *key, size_t key_len,
1148dbcf02cSchristos 		   const u8 *data, size_t data_len,
1158dbcf02cSchristos 		   u8 *out, size_t out_len)
1168dbcf02cSchristos {
1178dbcf02cSchristos 	u8 hash[IKEV2_MAX_HASH_LEN];
1188dbcf02cSchristos 	size_t hash_len;
1198dbcf02cSchristos 	u8 iter, *pos, *end;
1208dbcf02cSchristos 	const u8 *addr[3];
1218dbcf02cSchristos 	size_t len[3];
1228dbcf02cSchristos 	const struct ikev2_prf_alg *prf;
1238dbcf02cSchristos 	int res;
1248dbcf02cSchristos 
1258dbcf02cSchristos 	prf = ikev2_get_prf(alg);
1268dbcf02cSchristos 	if (prf == NULL)
1278dbcf02cSchristos 		return -1;
1288dbcf02cSchristos 	hash_len = prf->hash_len;
1298dbcf02cSchristos 
1308dbcf02cSchristos 	addr[0] = hash;
1318dbcf02cSchristos 	len[0] = hash_len;
1328dbcf02cSchristos 	addr[1] = data;
1338dbcf02cSchristos 	len[1] = data_len;
1348dbcf02cSchristos 	addr[2] = &iter;
1358dbcf02cSchristos 	len[2] = 1;
1368dbcf02cSchristos 
1378dbcf02cSchristos 	pos = out;
1388dbcf02cSchristos 	end = out + out_len;
1398dbcf02cSchristos 	iter = 1;
1408dbcf02cSchristos 	while (pos < end) {
1418dbcf02cSchristos 		size_t clen;
1428dbcf02cSchristos 		if (iter == 1)
1438dbcf02cSchristos 			res = ikev2_prf_hash(alg, key, key_len, 2, &addr[1],
1448dbcf02cSchristos 					     &len[1], hash);
1458dbcf02cSchristos 		else
1468dbcf02cSchristos 			res = ikev2_prf_hash(alg, key, key_len, 3, addr, len,
1478dbcf02cSchristos 					     hash);
1488dbcf02cSchristos 		if (res < 0)
1498dbcf02cSchristos 			return -1;
1508dbcf02cSchristos 		clen = hash_len;
1518dbcf02cSchristos 		if ((int) clen > end - pos)
1528dbcf02cSchristos 			clen = end - pos;
1538dbcf02cSchristos 		os_memcpy(pos, hash, clen);
1548dbcf02cSchristos 		pos += clen;
1558dbcf02cSchristos 		iter++;
1568dbcf02cSchristos 	}
1578dbcf02cSchristos 
1588dbcf02cSchristos 	return 0;
1598dbcf02cSchristos }
1608dbcf02cSchristos 
1618dbcf02cSchristos 
ikev2_get_encr(int id)1628dbcf02cSchristos const struct ikev2_encr_alg * ikev2_get_encr(int id)
1638dbcf02cSchristos {
1648dbcf02cSchristos 	size_t i;
1658dbcf02cSchristos 
1668dbcf02cSchristos 	for (i = 0; i < NUM_ENCR_ALGS; i++) {
1678dbcf02cSchristos 		if (ikev2_encr_algs[i].id == id)
1688dbcf02cSchristos 			return &ikev2_encr_algs[i];
1698dbcf02cSchristos 	}
1708dbcf02cSchristos 
1718dbcf02cSchristos 	return NULL;
1728dbcf02cSchristos }
1738dbcf02cSchristos 
1748dbcf02cSchristos 
ikev2_encr_encrypt(int alg,const u8 * key,size_t key_len,const u8 * iv,const u8 * plain,u8 * crypt,size_t len)1758dbcf02cSchristos int ikev2_encr_encrypt(int alg, const u8 *key, size_t key_len, const u8 *iv,
1768dbcf02cSchristos 		       const u8 *plain, u8 *crypt, size_t len)
1778dbcf02cSchristos {
1788dbcf02cSchristos 	struct crypto_cipher *cipher;
1798dbcf02cSchristos 	int encr_alg;
1808dbcf02cSchristos 
1818dbcf02cSchristos 	switch (alg) {
1828dbcf02cSchristos 	case ENCR_3DES:
1838dbcf02cSchristos 		encr_alg = CRYPTO_CIPHER_ALG_3DES;
1848dbcf02cSchristos 		break;
1858dbcf02cSchristos 	case ENCR_AES_CBC:
1868dbcf02cSchristos 		encr_alg = CRYPTO_CIPHER_ALG_AES;
1878dbcf02cSchristos 		break;
1888dbcf02cSchristos 	default:
1898dbcf02cSchristos 		wpa_printf(MSG_DEBUG, "IKEV2: Unsupported encr alg %d", alg);
1908dbcf02cSchristos 		return -1;
1918dbcf02cSchristos 	}
1928dbcf02cSchristos 
1938dbcf02cSchristos 	cipher = crypto_cipher_init(encr_alg, iv, key, key_len);
1948dbcf02cSchristos 	if (cipher == NULL) {
1958dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Failed to initialize cipher");
1968dbcf02cSchristos 		return -1;
1978dbcf02cSchristos 	}
1988dbcf02cSchristos 
1998dbcf02cSchristos 	if (crypto_cipher_encrypt(cipher, plain, crypt, len) < 0) {
2008dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Encryption failed");
2018dbcf02cSchristos 		crypto_cipher_deinit(cipher);
2028dbcf02cSchristos 		return -1;
2038dbcf02cSchristos 	}
2048dbcf02cSchristos 	crypto_cipher_deinit(cipher);
2058dbcf02cSchristos 
2068dbcf02cSchristos 	return 0;
2078dbcf02cSchristos }
2088dbcf02cSchristos 
2098dbcf02cSchristos 
ikev2_encr_decrypt(int alg,const u8 * key,size_t key_len,const u8 * iv,const u8 * crypt,u8 * plain,size_t len)2108dbcf02cSchristos int ikev2_encr_decrypt(int alg, const u8 *key, size_t key_len, const u8 *iv,
2118dbcf02cSchristos 		       const u8 *crypt, u8 *plain, size_t len)
2128dbcf02cSchristos {
2138dbcf02cSchristos 	struct crypto_cipher *cipher;
2148dbcf02cSchristos 	int encr_alg;
2158dbcf02cSchristos 
2168dbcf02cSchristos 	switch (alg) {
2178dbcf02cSchristos 	case ENCR_3DES:
2188dbcf02cSchristos 		encr_alg = CRYPTO_CIPHER_ALG_3DES;
2198dbcf02cSchristos 		break;
2208dbcf02cSchristos 	case ENCR_AES_CBC:
2218dbcf02cSchristos 		encr_alg = CRYPTO_CIPHER_ALG_AES;
2228dbcf02cSchristos 		break;
2238dbcf02cSchristos 	default:
2248dbcf02cSchristos 		wpa_printf(MSG_DEBUG, "IKEV2: Unsupported encr alg %d", alg);
2258dbcf02cSchristos 		return -1;
2268dbcf02cSchristos 	}
2278dbcf02cSchristos 
2288dbcf02cSchristos 	cipher = crypto_cipher_init(encr_alg, iv, key, key_len);
2298dbcf02cSchristos 	if (cipher == NULL) {
2308dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Failed to initialize cipher");
2318dbcf02cSchristos 		return -1;
2328dbcf02cSchristos 	}
2338dbcf02cSchristos 
2348dbcf02cSchristos 	if (crypto_cipher_decrypt(cipher, crypt, plain, len) < 0) {
2358dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Decryption failed");
2368dbcf02cSchristos 		crypto_cipher_deinit(cipher);
2378dbcf02cSchristos 		return -1;
2388dbcf02cSchristos 	}
2398dbcf02cSchristos 	crypto_cipher_deinit(cipher);
2408dbcf02cSchristos 
2418dbcf02cSchristos 	return 0;
2428dbcf02cSchristos }
2438dbcf02cSchristos 
2448dbcf02cSchristos 
ikev2_parse_payloads(struct ikev2_payloads * payloads,u8 next_payload,const u8 * pos,const u8 * end)2458dbcf02cSchristos int ikev2_parse_payloads(struct ikev2_payloads *payloads,
2468dbcf02cSchristos 			 u8 next_payload, const u8 *pos, const u8 *end)
2478dbcf02cSchristos {
2488dbcf02cSchristos 	const struct ikev2_payload_hdr *phdr;
2498dbcf02cSchristos 
2508dbcf02cSchristos 	os_memset(payloads, 0, sizeof(*payloads));
2518dbcf02cSchristos 
2528dbcf02cSchristos 	while (next_payload != IKEV2_PAYLOAD_NO_NEXT_PAYLOAD) {
253bb610346Schristos 		unsigned int plen, pdatalen, left;
2548dbcf02cSchristos 		const u8 *pdata;
2558dbcf02cSchristos 		wpa_printf(MSG_DEBUG, "IKEV2: Processing payload %u",
2568dbcf02cSchristos 			   next_payload);
257bb610346Schristos 		if (end < pos)
258bb610346Schristos 			return -1;
259bb610346Schristos 		left = end - pos;
260bb610346Schristos 		if (left < sizeof(*phdr)) {
2618dbcf02cSchristos 			wpa_printf(MSG_INFO, "IKEV2:   Too short message for "
2628dbcf02cSchristos 				   "payload header (left=%ld)",
2638dbcf02cSchristos 				   (long) (end - pos));
264bb610346Schristos 			return -1;
2658dbcf02cSchristos 		}
2668dbcf02cSchristos 		phdr = (const struct ikev2_payload_hdr *) pos;
2678dbcf02cSchristos 		plen = WPA_GET_BE16(phdr->payload_length);
268bb610346Schristos 		if (plen < sizeof(*phdr) || plen > left) {
2698dbcf02cSchristos 			wpa_printf(MSG_INFO, "IKEV2:   Invalid payload header "
2708dbcf02cSchristos 				   "length %d", plen);
2718dbcf02cSchristos 			return -1;
2728dbcf02cSchristos 		}
2738dbcf02cSchristos 
2748dbcf02cSchristos 		wpa_printf(MSG_DEBUG, "IKEV2:   Next Payload: %u  Flags: 0x%x"
275bb610346Schristos 			   "  Payload Length: %u",
2768dbcf02cSchristos 			   phdr->next_payload, phdr->flags, plen);
2778dbcf02cSchristos 
2788dbcf02cSchristos 		pdata = (const u8 *) (phdr + 1);
2798dbcf02cSchristos 		pdatalen = plen - sizeof(*phdr);
2808dbcf02cSchristos 
2818dbcf02cSchristos 		switch (next_payload) {
2828dbcf02cSchristos 		case IKEV2_PAYLOAD_SA:
2838dbcf02cSchristos 			wpa_printf(MSG_DEBUG, "IKEV2:   Payload: Security "
2848dbcf02cSchristos 				   "Association");
2858dbcf02cSchristos 			payloads->sa = pdata;
2868dbcf02cSchristos 			payloads->sa_len = pdatalen;
2878dbcf02cSchristos 			break;
2888dbcf02cSchristos 		case IKEV2_PAYLOAD_KEY_EXCHANGE:
2898dbcf02cSchristos 			wpa_printf(MSG_DEBUG, "IKEV2:   Payload: Key "
2908dbcf02cSchristos 				   "Exchange");
2918dbcf02cSchristos 			payloads->ke = pdata;
2928dbcf02cSchristos 			payloads->ke_len = pdatalen;
2938dbcf02cSchristos 			break;
2948dbcf02cSchristos 		case IKEV2_PAYLOAD_IDi:
2958dbcf02cSchristos 			wpa_printf(MSG_DEBUG, "IKEV2:   Payload: IDi");
2968dbcf02cSchristos 			payloads->idi = pdata;
2978dbcf02cSchristos 			payloads->idi_len = pdatalen;
2988dbcf02cSchristos 			break;
2998dbcf02cSchristos 		case IKEV2_PAYLOAD_IDr:
3008dbcf02cSchristos 			wpa_printf(MSG_DEBUG, "IKEV2:   Payload: IDr");
3018dbcf02cSchristos 			payloads->idr = pdata;
3028dbcf02cSchristos 			payloads->idr_len = pdatalen;
3038dbcf02cSchristos 			break;
3048dbcf02cSchristos 		case IKEV2_PAYLOAD_CERTIFICATE:
3058dbcf02cSchristos 			wpa_printf(MSG_DEBUG, "IKEV2:   Payload: Certificate");
3068dbcf02cSchristos 			payloads->cert = pdata;
3078dbcf02cSchristos 			payloads->cert_len = pdatalen;
3088dbcf02cSchristos 			break;
3098dbcf02cSchristos 		case IKEV2_PAYLOAD_AUTHENTICATION:
3108dbcf02cSchristos 			wpa_printf(MSG_DEBUG, "IKEV2:   Payload: "
3118dbcf02cSchristos 				   "Authentication");
3128dbcf02cSchristos 			payloads->auth = pdata;
3138dbcf02cSchristos 			payloads->auth_len = pdatalen;
3148dbcf02cSchristos 			break;
3158dbcf02cSchristos 		case IKEV2_PAYLOAD_NONCE:
3168dbcf02cSchristos 			wpa_printf(MSG_DEBUG, "IKEV2:   Payload: Nonce");
3178dbcf02cSchristos 			payloads->nonce = pdata;
3188dbcf02cSchristos 			payloads->nonce_len = pdatalen;
3198dbcf02cSchristos 			break;
3208dbcf02cSchristos 		case IKEV2_PAYLOAD_ENCRYPTED:
3218dbcf02cSchristos 			wpa_printf(MSG_DEBUG, "IKEV2:   Payload: Encrypted");
3228dbcf02cSchristos 			payloads->encrypted = pdata;
3238dbcf02cSchristos 			payloads->encrypted_len = pdatalen;
3248dbcf02cSchristos 			break;
3258dbcf02cSchristos 		case IKEV2_PAYLOAD_NOTIFICATION:
3268dbcf02cSchristos 			wpa_printf(MSG_DEBUG, "IKEV2:   Payload: "
3278dbcf02cSchristos 				   "Notification");
3288dbcf02cSchristos 			payloads->notification = pdata;
3298dbcf02cSchristos 			payloads->notification_len = pdatalen;
3308dbcf02cSchristos 			break;
3318dbcf02cSchristos 		default:
3328dbcf02cSchristos 			if (phdr->flags & IKEV2_PAYLOAD_FLAGS_CRITICAL) {
3338dbcf02cSchristos 				wpa_printf(MSG_INFO, "IKEV2:   Unsupported "
3348dbcf02cSchristos 					   "critical payload %u - reject the "
3358dbcf02cSchristos 					   "entire message", next_payload);
3368dbcf02cSchristos 				return -1;
3378dbcf02cSchristos 			} else {
3388dbcf02cSchristos 				wpa_printf(MSG_DEBUG, "IKEV2:   Skipped "
3398dbcf02cSchristos 					   "unsupported payload %u",
3408dbcf02cSchristos 					   next_payload);
3418dbcf02cSchristos 			}
3428dbcf02cSchristos 		}
3438dbcf02cSchristos 
3448dbcf02cSchristos 		if (next_payload == IKEV2_PAYLOAD_ENCRYPTED &&
3458dbcf02cSchristos 		    pos + plen == end) {
3468dbcf02cSchristos 			/*
3478dbcf02cSchristos 			 * Next Payload in the case of Encrypted Payload is
3488dbcf02cSchristos 			 * actually the payload type for the first embedded
3498dbcf02cSchristos 			 * payload.
3508dbcf02cSchristos 			 */
3518dbcf02cSchristos 			payloads->encr_next_payload = phdr->next_payload;
3528dbcf02cSchristos 			next_payload = IKEV2_PAYLOAD_NO_NEXT_PAYLOAD;
3538dbcf02cSchristos 		} else
3548dbcf02cSchristos 			next_payload = phdr->next_payload;
3558dbcf02cSchristos 
3568dbcf02cSchristos 		pos += plen;
3578dbcf02cSchristos 	}
3588dbcf02cSchristos 
3598dbcf02cSchristos 	if (pos != end) {
3608dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Unexpected extra data after "
3618dbcf02cSchristos 			   "payloads");
3628dbcf02cSchristos 		return -1;
3638dbcf02cSchristos 	}
3648dbcf02cSchristos 
3658dbcf02cSchristos 	return 0;
3668dbcf02cSchristos }
3678dbcf02cSchristos 
3688dbcf02cSchristos 
ikev2_derive_auth_data(int prf_alg,const struct wpabuf * sign_msg,const u8 * ID,size_t ID_len,u8 ID_type,struct ikev2_keys * keys,int initiator,const u8 * shared_secret,size_t shared_secret_len,const u8 * nonce,size_t nonce_len,const u8 * key_pad,size_t key_pad_len,u8 * auth_data)3698dbcf02cSchristos int ikev2_derive_auth_data(int prf_alg, const struct wpabuf *sign_msg,
3708dbcf02cSchristos 			   const u8 *ID, size_t ID_len, u8 ID_type,
3718dbcf02cSchristos 			   struct ikev2_keys *keys, int initiator,
3728dbcf02cSchristos 			   const u8 *shared_secret, size_t shared_secret_len,
3738dbcf02cSchristos 			   const u8 *nonce, size_t nonce_len,
3748dbcf02cSchristos 			   const u8 *key_pad, size_t key_pad_len,
3758dbcf02cSchristos 			   u8 *auth_data)
3768dbcf02cSchristos {
3778dbcf02cSchristos 	size_t sign_len, buf_len;
3788dbcf02cSchristos 	u8 *sign_data, *pos, *buf, hash[IKEV2_MAX_HASH_LEN];
3798dbcf02cSchristos 	const struct ikev2_prf_alg *prf;
3808dbcf02cSchristos 	const u8 *SK_p = initiator ? keys->SK_pi : keys->SK_pr;
3818dbcf02cSchristos 
3828dbcf02cSchristos 	prf = ikev2_get_prf(prf_alg);
3838dbcf02cSchristos 	if (sign_msg == NULL || ID == NULL || SK_p == NULL ||
3848dbcf02cSchristos 	    shared_secret == NULL || nonce == NULL || prf == NULL)
3858dbcf02cSchristos 		return -1;
3868dbcf02cSchristos 
3878dbcf02cSchristos 	/* prf(SK_pi/r,IDi/r') */
3888dbcf02cSchristos 	buf_len = 4 + ID_len;
3898dbcf02cSchristos 	buf = os_zalloc(buf_len);
3908dbcf02cSchristos 	if (buf == NULL)
3918dbcf02cSchristos 		return -1;
3928dbcf02cSchristos 	buf[0] = ID_type;
3938dbcf02cSchristos 	os_memcpy(buf + 4, ID, ID_len);
3948dbcf02cSchristos 	if (ikev2_prf_hash(prf->id, SK_p, keys->SK_prf_len,
3958dbcf02cSchristos 			   1, (const u8 **) &buf, &buf_len, hash) < 0) {
3968dbcf02cSchristos 		os_free(buf);
3978dbcf02cSchristos 		return -1;
3988dbcf02cSchristos 	}
3998dbcf02cSchristos 	os_free(buf);
4008dbcf02cSchristos 
4018dbcf02cSchristos 	/* sign_data = msg | Nr/i | prf(SK_pi/r,IDi/r') */
4028dbcf02cSchristos 	sign_len = wpabuf_len(sign_msg) + nonce_len + prf->hash_len;
4038dbcf02cSchristos 	sign_data = os_malloc(sign_len);
4048dbcf02cSchristos 	if (sign_data == NULL)
4058dbcf02cSchristos 		return -1;
4068dbcf02cSchristos 	pos = sign_data;
4078dbcf02cSchristos 	os_memcpy(pos, wpabuf_head(sign_msg), wpabuf_len(sign_msg));
4088dbcf02cSchristos 	pos += wpabuf_len(sign_msg);
4098dbcf02cSchristos 	os_memcpy(pos, nonce, nonce_len);
4108dbcf02cSchristos 	pos += nonce_len;
4118dbcf02cSchristos 	os_memcpy(pos, hash, prf->hash_len);
4128dbcf02cSchristos 
4138dbcf02cSchristos 	/* AUTH = prf(prf(Shared Secret, key pad, sign_data) */
4148dbcf02cSchristos 	if (ikev2_prf_hash(prf->id, shared_secret, shared_secret_len, 1,
4158dbcf02cSchristos 			   &key_pad, &key_pad_len, hash) < 0 ||
4168dbcf02cSchristos 	    ikev2_prf_hash(prf->id, hash, prf->hash_len, 1,
4178dbcf02cSchristos 			   (const u8 **) &sign_data, &sign_len, auth_data) < 0)
4188dbcf02cSchristos 	{
4198dbcf02cSchristos 		os_free(sign_data);
4208dbcf02cSchristos 		return -1;
4218dbcf02cSchristos 	}
4228dbcf02cSchristos 	os_free(sign_data);
4238dbcf02cSchristos 
4248dbcf02cSchristos 	return 0;
4258dbcf02cSchristos }
4268dbcf02cSchristos 
4278dbcf02cSchristos 
ikev2_decrypt_payload(int encr_id,int integ_id,struct ikev2_keys * keys,int initiator,const struct ikev2_hdr * hdr,const u8 * encrypted,size_t encrypted_len,size_t * res_len)4288dbcf02cSchristos u8 * ikev2_decrypt_payload(int encr_id, int integ_id,
4298dbcf02cSchristos 			   struct ikev2_keys *keys, int initiator,
4308dbcf02cSchristos 			   const struct ikev2_hdr *hdr,
4318dbcf02cSchristos 			   const u8 *encrypted, size_t encrypted_len,
4328dbcf02cSchristos 			   size_t *res_len)
4338dbcf02cSchristos {
4348dbcf02cSchristos 	size_t iv_len;
4358dbcf02cSchristos 	const u8 *pos, *end, *iv, *integ;
4368dbcf02cSchristos 	u8 hash[IKEV2_MAX_HASH_LEN], *decrypted;
4378dbcf02cSchristos 	size_t decrypted_len, pad_len;
4388dbcf02cSchristos 	const struct ikev2_integ_alg *integ_alg;
4398dbcf02cSchristos 	const struct ikev2_encr_alg *encr_alg;
4408dbcf02cSchristos 	const u8 *SK_e = initiator ? keys->SK_ei : keys->SK_er;
4418dbcf02cSchristos 	const u8 *SK_a = initiator ? keys->SK_ai : keys->SK_ar;
4428dbcf02cSchristos 
4438dbcf02cSchristos 	if (encrypted == NULL) {
4448dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: No Encrypted payload in SA_AUTH");
4458dbcf02cSchristos 		return NULL;
4468dbcf02cSchristos 	}
4478dbcf02cSchristos 
4488dbcf02cSchristos 	encr_alg = ikev2_get_encr(encr_id);
4498dbcf02cSchristos 	if (encr_alg == NULL) {
4508dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Unsupported encryption type");
4518dbcf02cSchristos 		return NULL;
4528dbcf02cSchristos 	}
4538dbcf02cSchristos 	iv_len = encr_alg->block_size;
4548dbcf02cSchristos 
4558dbcf02cSchristos 	integ_alg = ikev2_get_integ(integ_id);
4568dbcf02cSchristos 	if (integ_alg == NULL) {
4578dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Unsupported intergrity type");
4588dbcf02cSchristos 		return NULL;
4598dbcf02cSchristos 	}
4608dbcf02cSchristos 
4618dbcf02cSchristos 	if (encrypted_len < iv_len + 1 + integ_alg->hash_len) {
4628dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: No room for IV or Integrity "
4638dbcf02cSchristos 			  "Checksum");
4648dbcf02cSchristos 		return NULL;
4658dbcf02cSchristos 	}
4668dbcf02cSchristos 
4678dbcf02cSchristos 	iv = encrypted;
4688dbcf02cSchristos 	pos = iv + iv_len;
4698dbcf02cSchristos 	end = encrypted + encrypted_len;
4708dbcf02cSchristos 	integ = end - integ_alg->hash_len;
4718dbcf02cSchristos 
4728dbcf02cSchristos 	if (SK_a == NULL) {
4738dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: No SK_a available");
4748dbcf02cSchristos 		return NULL;
4758dbcf02cSchristos 	}
4768dbcf02cSchristos 	if (ikev2_integ_hash(integ_id, SK_a, keys->SK_integ_len,
4778dbcf02cSchristos 			     (const u8 *) hdr,
4788dbcf02cSchristos 			     integ - (const u8 *) hdr, hash) < 0) {
4798dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Failed to calculate integrity "
4808dbcf02cSchristos 			   "hash");
4818dbcf02cSchristos 		return NULL;
4828dbcf02cSchristos 	}
4833c260e60Schristos 	if (os_memcmp_const(integ, hash, integ_alg->hash_len) != 0) {
4848dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Incorrect Integrity Checksum "
4858dbcf02cSchristos 			   "Data");
4868dbcf02cSchristos 		return NULL;
4878dbcf02cSchristos 	}
4888dbcf02cSchristos 
4898dbcf02cSchristos 	if (SK_e == NULL) {
4908dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: No SK_e available");
4918dbcf02cSchristos 		return NULL;
4928dbcf02cSchristos 	}
4938dbcf02cSchristos 
4948dbcf02cSchristos 	decrypted_len = integ - pos;
4958dbcf02cSchristos 	decrypted = os_malloc(decrypted_len);
4968dbcf02cSchristos 	if (decrypted == NULL)
4978dbcf02cSchristos 		return NULL;
4988dbcf02cSchristos 
4998dbcf02cSchristos 	if (ikev2_encr_decrypt(encr_alg->id, SK_e, keys->SK_encr_len, iv, pos,
5008dbcf02cSchristos 			       decrypted, decrypted_len) < 0) {
5018dbcf02cSchristos 		os_free(decrypted);
5028dbcf02cSchristos 		return NULL;
5038dbcf02cSchristos 	}
5048dbcf02cSchristos 
5058dbcf02cSchristos 	pad_len = decrypted[decrypted_len - 1];
5068dbcf02cSchristos 	if (decrypted_len < pad_len + 1) {
5078dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Invalid padding in encrypted "
5088dbcf02cSchristos 			   "payload");
5098dbcf02cSchristos 		os_free(decrypted);
5108dbcf02cSchristos 		return NULL;
5118dbcf02cSchristos 	}
5128dbcf02cSchristos 
5138dbcf02cSchristos 	decrypted_len -= pad_len + 1;
5148dbcf02cSchristos 
5158dbcf02cSchristos 	*res_len = decrypted_len;
5168dbcf02cSchristos 	return decrypted;
5178dbcf02cSchristos }
5188dbcf02cSchristos 
5198dbcf02cSchristos 
ikev2_update_hdr(struct wpabuf * msg)5208dbcf02cSchristos void ikev2_update_hdr(struct wpabuf *msg)
5218dbcf02cSchristos {
5228dbcf02cSchristos 	struct ikev2_hdr *hdr;
5238dbcf02cSchristos 
5248dbcf02cSchristos 	/* Update lenth field in HDR */
5258dbcf02cSchristos 	hdr = wpabuf_mhead(msg);
5268dbcf02cSchristos 	WPA_PUT_BE32(hdr->length, wpabuf_len(msg));
5278dbcf02cSchristos }
5288dbcf02cSchristos 
5298dbcf02cSchristos 
ikev2_build_encrypted(int encr_id,int integ_id,struct ikev2_keys * keys,int initiator,struct wpabuf * msg,struct wpabuf * plain,u8 next_payload)5308dbcf02cSchristos int ikev2_build_encrypted(int encr_id, int integ_id, struct ikev2_keys *keys,
5318dbcf02cSchristos 			  int initiator, struct wpabuf *msg,
5328dbcf02cSchristos 			  struct wpabuf *plain, u8 next_payload)
5338dbcf02cSchristos {
5348dbcf02cSchristos 	struct ikev2_payload_hdr *phdr;
5358dbcf02cSchristos 	size_t plen;
5368dbcf02cSchristos 	size_t iv_len, pad_len;
5378dbcf02cSchristos 	u8 *icv, *iv;
5388dbcf02cSchristos 	const struct ikev2_integ_alg *integ_alg;
5398dbcf02cSchristos 	const struct ikev2_encr_alg *encr_alg;
5408dbcf02cSchristos 	const u8 *SK_e = initiator ? keys->SK_ei : keys->SK_er;
5418dbcf02cSchristos 	const u8 *SK_a = initiator ? keys->SK_ai : keys->SK_ar;
5428dbcf02cSchristos 
5438dbcf02cSchristos 	wpa_printf(MSG_DEBUG, "IKEV2: Adding Encrypted payload");
5448dbcf02cSchristos 
5458dbcf02cSchristos 	/* Encr - RFC 4306, Sect. 3.14 */
5468dbcf02cSchristos 
5478dbcf02cSchristos 	encr_alg = ikev2_get_encr(encr_id);
5488dbcf02cSchristos 	if (encr_alg == NULL) {
5498dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Unsupported encryption type");
5508dbcf02cSchristos 		return -1;
5518dbcf02cSchristos 	}
5528dbcf02cSchristos 	iv_len = encr_alg->block_size;
5538dbcf02cSchristos 
5548dbcf02cSchristos 	integ_alg = ikev2_get_integ(integ_id);
5558dbcf02cSchristos 	if (integ_alg == NULL) {
5568dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Unsupported intergrity type");
5578dbcf02cSchristos 		return -1;
5588dbcf02cSchristos 	}
5598dbcf02cSchristos 
5608dbcf02cSchristos 	if (SK_e == NULL) {
5618dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: No SK_e available");
5628dbcf02cSchristos 		return -1;
5638dbcf02cSchristos 	}
5648dbcf02cSchristos 
5658dbcf02cSchristos 	if (SK_a == NULL) {
5668dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: No SK_a available");
5678dbcf02cSchristos 		return -1;
5688dbcf02cSchristos 	}
5698dbcf02cSchristos 
5708dbcf02cSchristos 	phdr = wpabuf_put(msg, sizeof(*phdr));
5718dbcf02cSchristos 	phdr->next_payload = next_payload;
5728dbcf02cSchristos 	phdr->flags = 0;
5738dbcf02cSchristos 
5748dbcf02cSchristos 	iv = wpabuf_put(msg, iv_len);
575111b9fd8Schristos 	if (random_get_bytes(iv, iv_len)) {
5768dbcf02cSchristos 		wpa_printf(MSG_INFO, "IKEV2: Could not generate IV");
5778dbcf02cSchristos 		return -1;
5788dbcf02cSchristos 	}
5798dbcf02cSchristos 
5808dbcf02cSchristos 	pad_len = iv_len - (wpabuf_len(plain) + 1) % iv_len;
5818dbcf02cSchristos 	if (pad_len == iv_len)
5828dbcf02cSchristos 		pad_len = 0;
5838dbcf02cSchristos 	wpabuf_put(plain, pad_len);
5848dbcf02cSchristos 	wpabuf_put_u8(plain, pad_len);
5858dbcf02cSchristos 
5868dbcf02cSchristos 	if (ikev2_encr_encrypt(encr_alg->id, SK_e, keys->SK_encr_len, iv,
5878dbcf02cSchristos 			       wpabuf_head(plain), wpabuf_mhead(plain),
5888dbcf02cSchristos 			       wpabuf_len(plain)) < 0)
5898dbcf02cSchristos 		return -1;
5908dbcf02cSchristos 
5918dbcf02cSchristos 	wpabuf_put_buf(msg, plain);
5928dbcf02cSchristos 
5938dbcf02cSchristos 	/* Need to update all headers (Length fields) prior to hash func */
5948dbcf02cSchristos 	icv = wpabuf_put(msg, integ_alg->hash_len);
5958dbcf02cSchristos 	plen = (u8 *) wpabuf_put(msg, 0) - (u8 *) phdr;
5968dbcf02cSchristos 	WPA_PUT_BE16(phdr->payload_length, plen);
5978dbcf02cSchristos 
5988dbcf02cSchristos 	ikev2_update_hdr(msg);
5998dbcf02cSchristos 
6008dbcf02cSchristos 	return ikev2_integ_hash(integ_id, SK_a, keys->SK_integ_len,
6018dbcf02cSchristos 				wpabuf_head(msg),
6028dbcf02cSchristos 				wpabuf_len(msg) - integ_alg->hash_len, icv);
6038dbcf02cSchristos 
6048dbcf02cSchristos 	return 0;
6058dbcf02cSchristos }
6068dbcf02cSchristos 
6078dbcf02cSchristos 
ikev2_keys_set(struct ikev2_keys * keys)6088dbcf02cSchristos int ikev2_keys_set(struct ikev2_keys *keys)
6098dbcf02cSchristos {
6108dbcf02cSchristos 	return keys->SK_d && keys->SK_ai && keys->SK_ar && keys->SK_ei &&
6118dbcf02cSchristos 		keys->SK_er && keys->SK_pi && keys->SK_pr;
6128dbcf02cSchristos }
6138dbcf02cSchristos 
6148dbcf02cSchristos 
ikev2_free_keys(struct ikev2_keys * keys)6158dbcf02cSchristos void ikev2_free_keys(struct ikev2_keys *keys)
6168dbcf02cSchristos {
6178dbcf02cSchristos 	os_free(keys->SK_d);
6188dbcf02cSchristos 	os_free(keys->SK_ai);
6198dbcf02cSchristos 	os_free(keys->SK_ar);
6208dbcf02cSchristos 	os_free(keys->SK_ei);
6218dbcf02cSchristos 	os_free(keys->SK_er);
6228dbcf02cSchristos 	os_free(keys->SK_pi);
6238dbcf02cSchristos 	os_free(keys->SK_pr);
6248dbcf02cSchristos 	keys->SK_d = keys->SK_ai = keys->SK_ar = keys->SK_ei = keys->SK_er =
6258dbcf02cSchristos 		keys->SK_pi = keys->SK_pr = NULL;
6268dbcf02cSchristos }
6278dbcf02cSchristos 
6288dbcf02cSchristos 
ikev2_derive_sk_keys(const struct ikev2_prf_alg * prf,const struct ikev2_integ_alg * integ,const struct ikev2_encr_alg * encr,const u8 * skeyseed,const u8 * data,size_t data_len,struct ikev2_keys * keys)6298dbcf02cSchristos int ikev2_derive_sk_keys(const struct ikev2_prf_alg *prf,
6308dbcf02cSchristos 			 const struct ikev2_integ_alg *integ,
6318dbcf02cSchristos 			 const struct ikev2_encr_alg *encr,
6328dbcf02cSchristos 			 const u8 *skeyseed, const u8 *data, size_t data_len,
6338dbcf02cSchristos 			 struct ikev2_keys *keys)
6348dbcf02cSchristos {
6358dbcf02cSchristos 	u8 *keybuf, *pos;
6368dbcf02cSchristos 	size_t keybuf_len;
6378dbcf02cSchristos 
6388dbcf02cSchristos 	/*
6398dbcf02cSchristos 	 * {SK_d | SK_ai | SK_ar | SK_ei | SK_er | SK_pi | SK_pr } =
6408dbcf02cSchristos 	 *	prf+(SKEYSEED, Ni | Nr | SPIi | SPIr )
6418dbcf02cSchristos 	 */
6428dbcf02cSchristos 	ikev2_free_keys(keys);
6438dbcf02cSchristos 	keys->SK_d_len = prf->key_len;
6448dbcf02cSchristos 	keys->SK_integ_len = integ->key_len;
6458dbcf02cSchristos 	keys->SK_encr_len = encr->key_len;
6468dbcf02cSchristos 	keys->SK_prf_len = prf->key_len;
6478dbcf02cSchristos 
6488dbcf02cSchristos 	keybuf_len = keys->SK_d_len + 2 * keys->SK_integ_len +
6498dbcf02cSchristos 		2 * keys->SK_encr_len + 2 * keys->SK_prf_len;
6508dbcf02cSchristos 	keybuf = os_malloc(keybuf_len);
6518dbcf02cSchristos 	if (keybuf == NULL)
6528dbcf02cSchristos 		return -1;
6538dbcf02cSchristos 
6548dbcf02cSchristos 	if (ikev2_prf_plus(prf->id, skeyseed, prf->hash_len,
6558dbcf02cSchristos 			   data, data_len, keybuf, keybuf_len)) {
6568dbcf02cSchristos 		os_free(keybuf);
6578dbcf02cSchristos 		return -1;
6588dbcf02cSchristos 	}
6598dbcf02cSchristos 
6608dbcf02cSchristos 	pos = keybuf;
6618dbcf02cSchristos 
6628dbcf02cSchristos 	keys->SK_d = os_malloc(keys->SK_d_len);
6638dbcf02cSchristos 	if (keys->SK_d) {
6648dbcf02cSchristos 		os_memcpy(keys->SK_d, pos, keys->SK_d_len);
6658dbcf02cSchristos 		wpa_hexdump_key(MSG_DEBUG, "IKEV2: SK_d",
6668dbcf02cSchristos 				keys->SK_d, keys->SK_d_len);
6678dbcf02cSchristos 	}
6688dbcf02cSchristos 	pos += keys->SK_d_len;
6698dbcf02cSchristos 
6708dbcf02cSchristos 	keys->SK_ai = os_malloc(keys->SK_integ_len);
6718dbcf02cSchristos 	if (keys->SK_ai) {
6728dbcf02cSchristos 		os_memcpy(keys->SK_ai, pos, keys->SK_integ_len);
6738dbcf02cSchristos 		wpa_hexdump_key(MSG_DEBUG, "IKEV2: SK_ai",
6748dbcf02cSchristos 				keys->SK_ai, keys->SK_integ_len);
6758dbcf02cSchristos 	}
6768dbcf02cSchristos 	pos += keys->SK_integ_len;
6778dbcf02cSchristos 
6788dbcf02cSchristos 	keys->SK_ar = os_malloc(keys->SK_integ_len);
6798dbcf02cSchristos 	if (keys->SK_ar) {
6808dbcf02cSchristos 		os_memcpy(keys->SK_ar, pos, keys->SK_integ_len);
6818dbcf02cSchristos 		wpa_hexdump_key(MSG_DEBUG, "IKEV2: SK_ar",
6828dbcf02cSchristos 				keys->SK_ar, keys->SK_integ_len);
6838dbcf02cSchristos 	}
6848dbcf02cSchristos 	pos += keys->SK_integ_len;
6858dbcf02cSchristos 
6868dbcf02cSchristos 	keys->SK_ei = os_malloc(keys->SK_encr_len);
6878dbcf02cSchristos 	if (keys->SK_ei) {
6888dbcf02cSchristos 		os_memcpy(keys->SK_ei, pos, keys->SK_encr_len);
6898dbcf02cSchristos 		wpa_hexdump_key(MSG_DEBUG, "IKEV2: SK_ei",
6908dbcf02cSchristos 				keys->SK_ei, keys->SK_encr_len);
6918dbcf02cSchristos 	}
6928dbcf02cSchristos 	pos += keys->SK_encr_len;
6938dbcf02cSchristos 
6948dbcf02cSchristos 	keys->SK_er = os_malloc(keys->SK_encr_len);
6958dbcf02cSchristos 	if (keys->SK_er) {
6968dbcf02cSchristos 		os_memcpy(keys->SK_er, pos, keys->SK_encr_len);
6978dbcf02cSchristos 		wpa_hexdump_key(MSG_DEBUG, "IKEV2: SK_er",
6988dbcf02cSchristos 				keys->SK_er, keys->SK_encr_len);
6998dbcf02cSchristos 	}
7008dbcf02cSchristos 	pos += keys->SK_encr_len;
7018dbcf02cSchristos 
7028dbcf02cSchristos 	keys->SK_pi = os_malloc(keys->SK_prf_len);
7038dbcf02cSchristos 	if (keys->SK_pi) {
7048dbcf02cSchristos 		os_memcpy(keys->SK_pi, pos, keys->SK_prf_len);
7058dbcf02cSchristos 		wpa_hexdump_key(MSG_DEBUG, "IKEV2: SK_pi",
7068dbcf02cSchristos 				keys->SK_pi, keys->SK_prf_len);
7078dbcf02cSchristos 	}
7088dbcf02cSchristos 	pos += keys->SK_prf_len;
7098dbcf02cSchristos 
7108dbcf02cSchristos 	keys->SK_pr = os_malloc(keys->SK_prf_len);
7118dbcf02cSchristos 	if (keys->SK_pr) {
7128dbcf02cSchristos 		os_memcpy(keys->SK_pr, pos, keys->SK_prf_len);
7138dbcf02cSchristos 		wpa_hexdump_key(MSG_DEBUG, "IKEV2: SK_pr",
7148dbcf02cSchristos 				keys->SK_pr, keys->SK_prf_len);
7158dbcf02cSchristos 	}
7168dbcf02cSchristos 
7178dbcf02cSchristos 	os_free(keybuf);
7188dbcf02cSchristos 
7198dbcf02cSchristos 	if (!ikev2_keys_set(keys)) {
7208dbcf02cSchristos 		ikev2_free_keys(keys);
7218dbcf02cSchristos 		return -1;
7228dbcf02cSchristos 	}
7238dbcf02cSchristos 
7248dbcf02cSchristos 	return 0;
7258dbcf02cSchristos }
726