xref: /freebsd/contrib/wpa/src/common/wpa_common.c (revision 4b72b91a7132df1f77bbae194e1071ac621f1edb)
139beb93cSSam Leffler /*
239beb93cSSam Leffler  * WPA/RSN - Shared functions for supplicant and authenticator
385732ac8SCy Schubert  * Copyright (c) 2002-2018, Jouni Malinen <j@w1.fi>
439beb93cSSam Leffler  *
5f05cddf9SRui Paulo  * This software may be distributed under the terms of the BSD license.
6f05cddf9SRui Paulo  * See README for more details.
739beb93cSSam Leffler  */
839beb93cSSam Leffler 
939beb93cSSam Leffler #include "includes.h"
1039beb93cSSam Leffler 
1139beb93cSSam Leffler #include "common.h"
12e28a4053SRui Paulo #include "crypto/md5.h"
13e28a4053SRui Paulo #include "crypto/sha1.h"
14e28a4053SRui Paulo #include "crypto/sha256.h"
155b9c547cSRui Paulo #include "crypto/sha384.h"
1685732ac8SCy Schubert #include "crypto/sha512.h"
17e28a4053SRui Paulo #include "crypto/aes_wrap.h"
18e28a4053SRui Paulo #include "crypto/crypto.h"
1939beb93cSSam Leffler #include "ieee802_11_defs.h"
2039beb93cSSam Leffler #include "defs.h"
2139beb93cSSam Leffler #include "wpa_common.h"
2239beb93cSSam Leffler 
2339beb93cSSam Leffler 
2485732ac8SCy Schubert static unsigned int wpa_kck_len(int akmp, size_t pmk_len)
255b9c547cSRui Paulo {
2685732ac8SCy Schubert 	switch (akmp) {
2785732ac8SCy Schubert 	case WPA_KEY_MGMT_IEEE8021X_SUITE_B_192:
2885732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_IEEE8021X_SHA384:
295b9c547cSRui Paulo 		return 24;
3085732ac8SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA256:
3185732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA256:
3285732ac8SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA384:
3385732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA384:
3485732ac8SCy Schubert 		return 0;
3585732ac8SCy Schubert 	case WPA_KEY_MGMT_DPP:
3685732ac8SCy Schubert 		return pmk_len / 2;
3785732ac8SCy Schubert 	case WPA_KEY_MGMT_OWE:
3885732ac8SCy Schubert 		return pmk_len / 2;
3985732ac8SCy Schubert 	default:
405b9c547cSRui Paulo 		return 16;
415b9c547cSRui Paulo 	}
4285732ac8SCy Schubert }
435b9c547cSRui Paulo 
445b9c547cSRui Paulo 
4585732ac8SCy Schubert #ifdef CONFIG_IEEE80211R
4685732ac8SCy Schubert static unsigned int wpa_kck2_len(int akmp)
475b9c547cSRui Paulo {
4885732ac8SCy Schubert 	switch (akmp) {
4985732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA256:
5085732ac8SCy Schubert 		return 16;
5185732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA384:
5285732ac8SCy Schubert 		return 24;
5385732ac8SCy Schubert 	default:
5485732ac8SCy Schubert 		return 0;
5585732ac8SCy Schubert 	}
5685732ac8SCy Schubert }
5785732ac8SCy Schubert #endif /* CONFIG_IEEE80211R */
5885732ac8SCy Schubert 
5985732ac8SCy Schubert 
6085732ac8SCy Schubert static unsigned int wpa_kek_len(int akmp, size_t pmk_len)
6185732ac8SCy Schubert {
6285732ac8SCy Schubert 	switch (akmp) {
6385732ac8SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA384:
6485732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA384:
6585732ac8SCy Schubert 		return 64;
6685732ac8SCy Schubert 	case WPA_KEY_MGMT_IEEE8021X_SUITE_B_192:
6785732ac8SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA256:
6885732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA256:
6985732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_IEEE8021X_SHA384:
705b9c547cSRui Paulo 		return 32;
7185732ac8SCy Schubert 	case WPA_KEY_MGMT_DPP:
7285732ac8SCy Schubert 		return pmk_len <= 32 ? 16 : 32;
7385732ac8SCy Schubert 	case WPA_KEY_MGMT_OWE:
7485732ac8SCy Schubert 		return pmk_len <= 32 ? 16 : 32;
7585732ac8SCy Schubert 	default:
765b9c547cSRui Paulo 		return 16;
775b9c547cSRui Paulo 	}
7885732ac8SCy Schubert }
795b9c547cSRui Paulo 
805b9c547cSRui Paulo 
8185732ac8SCy Schubert #ifdef CONFIG_IEEE80211R
8285732ac8SCy Schubert static unsigned int wpa_kek2_len(int akmp)
835b9c547cSRui Paulo {
8485732ac8SCy Schubert 	switch (akmp) {
8585732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA256:
865b9c547cSRui Paulo 		return 16;
8785732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA384:
8885732ac8SCy Schubert 		return 32;
8985732ac8SCy Schubert 	default:
9085732ac8SCy Schubert 		return 0;
9185732ac8SCy Schubert 	}
9285732ac8SCy Schubert }
9385732ac8SCy Schubert #endif /* CONFIG_IEEE80211R */
9485732ac8SCy Schubert 
9585732ac8SCy Schubert 
9685732ac8SCy Schubert unsigned int wpa_mic_len(int akmp, size_t pmk_len)
9785732ac8SCy Schubert {
9885732ac8SCy Schubert 	switch (akmp) {
9985732ac8SCy Schubert 	case WPA_KEY_MGMT_IEEE8021X_SUITE_B_192:
10085732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_IEEE8021X_SHA384:
10185732ac8SCy Schubert 		return 24;
10285732ac8SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA256:
10385732ac8SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA384:
10485732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA256:
10585732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA384:
10685732ac8SCy Schubert 		return 0;
10785732ac8SCy Schubert 	case WPA_KEY_MGMT_DPP:
10885732ac8SCy Schubert 		return pmk_len / 2;
10985732ac8SCy Schubert 	case WPA_KEY_MGMT_OWE:
11085732ac8SCy Schubert 		return pmk_len / 2;
11185732ac8SCy Schubert 	default:
11285732ac8SCy Schubert 		return 16;
11385732ac8SCy Schubert 	}
11485732ac8SCy Schubert }
11585732ac8SCy Schubert 
11685732ac8SCy Schubert 
11785732ac8SCy Schubert /**
11885732ac8SCy Schubert  * wpa_use_akm_defined - Is AKM-defined Key Descriptor Version used
11985732ac8SCy Schubert  * @akmp: WPA_KEY_MGMT_* used in key derivation
12085732ac8SCy Schubert  * Returns: 1 if AKM-defined Key Descriptor Version is used; 0 otherwise
12185732ac8SCy Schubert  */
12285732ac8SCy Schubert int wpa_use_akm_defined(int akmp)
12385732ac8SCy Schubert {
12485732ac8SCy Schubert 	return akmp == WPA_KEY_MGMT_OSEN ||
12585732ac8SCy Schubert 		akmp == WPA_KEY_MGMT_OWE ||
12685732ac8SCy Schubert 		akmp == WPA_KEY_MGMT_DPP ||
12785732ac8SCy Schubert 		akmp == WPA_KEY_MGMT_FT_IEEE8021X_SHA384 ||
12885732ac8SCy Schubert 		wpa_key_mgmt_sae(akmp) ||
12985732ac8SCy Schubert 		wpa_key_mgmt_suite_b(akmp) ||
13085732ac8SCy Schubert 		wpa_key_mgmt_fils(akmp);
13185732ac8SCy Schubert }
13285732ac8SCy Schubert 
13385732ac8SCy Schubert 
13485732ac8SCy Schubert /**
13585732ac8SCy Schubert  * wpa_use_cmac - Is CMAC integrity algorithm used for EAPOL-Key MIC
13685732ac8SCy Schubert  * @akmp: WPA_KEY_MGMT_* used in key derivation
13785732ac8SCy Schubert  * Returns: 1 if CMAC is used; 0 otherwise
13885732ac8SCy Schubert  */
13985732ac8SCy Schubert int wpa_use_cmac(int akmp)
14085732ac8SCy Schubert {
14185732ac8SCy Schubert 	return akmp == WPA_KEY_MGMT_OSEN ||
14285732ac8SCy Schubert 		akmp == WPA_KEY_MGMT_OWE ||
14385732ac8SCy Schubert 		akmp == WPA_KEY_MGMT_DPP ||
14485732ac8SCy Schubert 		wpa_key_mgmt_ft(akmp) ||
14585732ac8SCy Schubert 		wpa_key_mgmt_sha256(akmp) ||
14685732ac8SCy Schubert 		wpa_key_mgmt_sae(akmp) ||
14785732ac8SCy Schubert 		wpa_key_mgmt_suite_b(akmp);
14885732ac8SCy Schubert }
14985732ac8SCy Schubert 
15085732ac8SCy Schubert 
15185732ac8SCy Schubert /**
15285732ac8SCy Schubert  * wpa_use_aes_key_wrap - Is AES Keywrap algorithm used for EAPOL-Key Key Data
15385732ac8SCy Schubert  * @akmp: WPA_KEY_MGMT_* used in key derivation
15485732ac8SCy Schubert  * Returns: 1 if AES Keywrap is used; 0 otherwise
15585732ac8SCy Schubert  *
15685732ac8SCy Schubert  * Note: AKM 00-0F-AC:1 and 00-0F-AC:2 have special rules for selecting whether
15785732ac8SCy Schubert  * to use AES Keywrap based on the negotiated pairwise cipher. This function
15885732ac8SCy Schubert  * does not cover those special cases.
15985732ac8SCy Schubert  */
16085732ac8SCy Schubert int wpa_use_aes_key_wrap(int akmp)
16185732ac8SCy Schubert {
16285732ac8SCy Schubert 	return akmp == WPA_KEY_MGMT_OSEN ||
16385732ac8SCy Schubert 		akmp == WPA_KEY_MGMT_OWE ||
16485732ac8SCy Schubert 		akmp == WPA_KEY_MGMT_DPP ||
16585732ac8SCy Schubert 		wpa_key_mgmt_ft(akmp) ||
16685732ac8SCy Schubert 		wpa_key_mgmt_sha256(akmp) ||
16785732ac8SCy Schubert 		wpa_key_mgmt_sae(akmp) ||
16885732ac8SCy Schubert 		wpa_key_mgmt_suite_b(akmp);
1695b9c547cSRui Paulo }
1705b9c547cSRui Paulo 
1715b9c547cSRui Paulo 
17239beb93cSSam Leffler /**
17339beb93cSSam Leffler  * wpa_eapol_key_mic - Calculate EAPOL-Key MIC
17439beb93cSSam Leffler  * @key: EAPOL-Key Key Confirmation Key (KCK)
1755b9c547cSRui Paulo  * @key_len: KCK length in octets
1765b9c547cSRui Paulo  * @akmp: WPA_KEY_MGMT_* used in key derivation
17739beb93cSSam Leffler  * @ver: Key descriptor version (WPA_KEY_INFO_TYPE_*)
17839beb93cSSam Leffler  * @buf: Pointer to the beginning of the EAPOL header (version field)
17939beb93cSSam Leffler  * @len: Length of the EAPOL frame (from EAPOL header to the end of the frame)
18039beb93cSSam Leffler  * @mic: Pointer to the buffer to which the EAPOL-Key MIC is written
18139beb93cSSam Leffler  * Returns: 0 on success, -1 on failure
18239beb93cSSam Leffler  *
18339beb93cSSam Leffler  * Calculate EAPOL-Key MIC for an EAPOL-Key packet. The EAPOL-Key MIC field has
18439beb93cSSam Leffler  * to be cleared (all zeroes) when calling this function.
18539beb93cSSam Leffler  *
18639beb93cSSam Leffler  * Note: 'IEEE Std 802.11i-2004 - 8.5.2 EAPOL-Key frames' has an error in the
18739beb93cSSam Leffler  * description of the Key MIC calculation. It includes packet data from the
18839beb93cSSam Leffler  * beginning of the EAPOL-Key header, not EAPOL header. This incorrect change
18939beb93cSSam Leffler  * happened during final editing of the standard and the correct behavior is
19039beb93cSSam Leffler  * defined in the last draft (IEEE 802.11i/D10).
19139beb93cSSam Leffler  */
1925b9c547cSRui Paulo int wpa_eapol_key_mic(const u8 *key, size_t key_len, int akmp, int ver,
1935b9c547cSRui Paulo 		      const u8 *buf, size_t len, u8 *mic)
19439beb93cSSam Leffler {
19585732ac8SCy Schubert 	u8 hash[SHA512_MAC_LEN];
19685732ac8SCy Schubert 
19785732ac8SCy Schubert 	if (key_len == 0) {
19885732ac8SCy Schubert 		wpa_printf(MSG_DEBUG,
19985732ac8SCy Schubert 			   "WPA: KCK not set - cannot calculate MIC");
20085732ac8SCy Schubert 		return -1;
20185732ac8SCy Schubert 	}
20239beb93cSSam Leffler 
20339beb93cSSam Leffler 	switch (ver) {
204f05cddf9SRui Paulo #ifndef CONFIG_FIPS
20539beb93cSSam Leffler 	case WPA_KEY_INFO_TYPE_HMAC_MD5_RC4:
20685732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: EAPOL-Key MIC using HMAC-MD5");
2075b9c547cSRui Paulo 		return hmac_md5(key, key_len, buf, len, mic);
208f05cddf9SRui Paulo #endif /* CONFIG_FIPS */
20939beb93cSSam Leffler 	case WPA_KEY_INFO_TYPE_HMAC_SHA1_AES:
21085732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: EAPOL-Key MIC using HMAC-SHA1");
2115b9c547cSRui Paulo 		if (hmac_sha1(key, key_len, buf, len, hash))
212e28a4053SRui Paulo 			return -1;
21339beb93cSSam Leffler 		os_memcpy(mic, hash, MD5_MAC_LEN);
21439beb93cSSam Leffler 		break;
21539beb93cSSam Leffler 	case WPA_KEY_INFO_TYPE_AES_128_CMAC:
21685732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: EAPOL-Key MIC using AES-CMAC");
21739beb93cSSam Leffler 		return omac1_aes_128(key, buf, len, mic);
2185b9c547cSRui Paulo 	case WPA_KEY_INFO_TYPE_AKM_DEFINED:
2195b9c547cSRui Paulo 		switch (akmp) {
22085732ac8SCy Schubert #ifdef CONFIG_SAE
22185732ac8SCy Schubert 		case WPA_KEY_MGMT_SAE:
22285732ac8SCy Schubert 		case WPA_KEY_MGMT_FT_SAE:
22385732ac8SCy Schubert 			wpa_printf(MSG_DEBUG,
22485732ac8SCy Schubert 				   "WPA: EAPOL-Key MIC using AES-CMAC (AKM-defined - SAE)");
22585732ac8SCy Schubert 			return omac1_aes_128(key, buf, len, mic);
22685732ac8SCy Schubert #endif /* CONFIG_SAE */
2275b9c547cSRui Paulo #ifdef CONFIG_HS20
2285b9c547cSRui Paulo 		case WPA_KEY_MGMT_OSEN:
22985732ac8SCy Schubert 			wpa_printf(MSG_DEBUG,
23085732ac8SCy Schubert 				   "WPA: EAPOL-Key MIC using AES-CMAC (AKM-defined - OSEN)");
2315b9c547cSRui Paulo 			return omac1_aes_128(key, buf, len, mic);
2325b9c547cSRui Paulo #endif /* CONFIG_HS20 */
2335b9c547cSRui Paulo #ifdef CONFIG_SUITEB
2345b9c547cSRui Paulo 		case WPA_KEY_MGMT_IEEE8021X_SUITE_B:
23585732ac8SCy Schubert 			wpa_printf(MSG_DEBUG,
23685732ac8SCy Schubert 				   "WPA: EAPOL-Key MIC using HMAC-SHA256 (AKM-defined - Suite B)");
2375b9c547cSRui Paulo 			if (hmac_sha256(key, key_len, buf, len, hash))
2385b9c547cSRui Paulo 				return -1;
2395b9c547cSRui Paulo 			os_memcpy(mic, hash, MD5_MAC_LEN);
2405b9c547cSRui Paulo 			break;
2415b9c547cSRui Paulo #endif /* CONFIG_SUITEB */
2425b9c547cSRui Paulo #ifdef CONFIG_SUITEB192
2435b9c547cSRui Paulo 		case WPA_KEY_MGMT_IEEE8021X_SUITE_B_192:
24485732ac8SCy Schubert 			wpa_printf(MSG_DEBUG,
24585732ac8SCy Schubert 				   "WPA: EAPOL-Key MIC using HMAC-SHA384 (AKM-defined - Suite B 192-bit)");
2465b9c547cSRui Paulo 			if (hmac_sha384(key, key_len, buf, len, hash))
2475b9c547cSRui Paulo 				return -1;
2485b9c547cSRui Paulo 			os_memcpy(mic, hash, 24);
2495b9c547cSRui Paulo 			break;
2505b9c547cSRui Paulo #endif /* CONFIG_SUITEB192 */
25185732ac8SCy Schubert #ifdef CONFIG_OWE
25285732ac8SCy Schubert 		case WPA_KEY_MGMT_OWE:
25385732ac8SCy Schubert 			wpa_printf(MSG_DEBUG,
25485732ac8SCy Schubert 				   "WPA: EAPOL-Key MIC using HMAC-SHA%u (AKM-defined - OWE)",
25585732ac8SCy Schubert 				   (unsigned int) key_len * 8 * 2);
25685732ac8SCy Schubert 			if (key_len == 128 / 8) {
25785732ac8SCy Schubert 				if (hmac_sha256(key, key_len, buf, len, hash))
25885732ac8SCy Schubert 					return -1;
25985732ac8SCy Schubert 			} else if (key_len == 192 / 8) {
26085732ac8SCy Schubert 				if (hmac_sha384(key, key_len, buf, len, hash))
26185732ac8SCy Schubert 					return -1;
26285732ac8SCy Schubert 			} else if (key_len == 256 / 8) {
26385732ac8SCy Schubert 				if (hmac_sha512(key, key_len, buf, len, hash))
26485732ac8SCy Schubert 					return -1;
26585732ac8SCy Schubert 			} else {
26685732ac8SCy Schubert 				wpa_printf(MSG_INFO,
26785732ac8SCy Schubert 					   "OWE: Unsupported KCK length: %u",
26885732ac8SCy Schubert 					   (unsigned int) key_len);
26985732ac8SCy Schubert 				return -1;
27085732ac8SCy Schubert 			}
27185732ac8SCy Schubert 			os_memcpy(mic, hash, key_len);
27285732ac8SCy Schubert 			break;
27385732ac8SCy Schubert #endif /* CONFIG_OWE */
27485732ac8SCy Schubert #ifdef CONFIG_DPP
27585732ac8SCy Schubert 		case WPA_KEY_MGMT_DPP:
27685732ac8SCy Schubert 			wpa_printf(MSG_DEBUG,
27785732ac8SCy Schubert 				   "WPA: EAPOL-Key MIC using HMAC-SHA%u (AKM-defined - DPP)",
27885732ac8SCy Schubert 				   (unsigned int) key_len * 8 * 2);
27985732ac8SCy Schubert 			if (key_len == 128 / 8) {
28085732ac8SCy Schubert 				if (hmac_sha256(key, key_len, buf, len, hash))
28185732ac8SCy Schubert 					return -1;
28285732ac8SCy Schubert 			} else if (key_len == 192 / 8) {
28385732ac8SCy Schubert 				if (hmac_sha384(key, key_len, buf, len, hash))
28485732ac8SCy Schubert 					return -1;
28585732ac8SCy Schubert 			} else if (key_len == 256 / 8) {
28685732ac8SCy Schubert 				if (hmac_sha512(key, key_len, buf, len, hash))
28785732ac8SCy Schubert 					return -1;
28885732ac8SCy Schubert 			} else {
28985732ac8SCy Schubert 				wpa_printf(MSG_INFO,
29085732ac8SCy Schubert 					   "DPP: Unsupported KCK length: %u",
29185732ac8SCy Schubert 					   (unsigned int) key_len);
29285732ac8SCy Schubert 				return -1;
29385732ac8SCy Schubert 			}
29485732ac8SCy Schubert 			os_memcpy(mic, hash, key_len);
29585732ac8SCy Schubert 			break;
29685732ac8SCy Schubert #endif /* CONFIG_DPP */
29785732ac8SCy Schubert #if defined(CONFIG_IEEE80211R) && defined(CONFIG_SHA384)
29885732ac8SCy Schubert 		case WPA_KEY_MGMT_FT_IEEE8021X_SHA384:
29985732ac8SCy Schubert 			wpa_printf(MSG_DEBUG,
30085732ac8SCy Schubert 				   "WPA: EAPOL-Key MIC using HMAC-SHA384 (AKM-defined - FT 802.1X SHA384)");
30185732ac8SCy Schubert 			if (hmac_sha384(key, key_len, buf, len, hash))
30285732ac8SCy Schubert 				return -1;
30385732ac8SCy Schubert 			os_memcpy(mic, hash, 24);
30485732ac8SCy Schubert 			break;
30585732ac8SCy Schubert #endif /* CONFIG_IEEE80211R && CONFIG_SHA384 */
3065b9c547cSRui Paulo 		default:
30785732ac8SCy Schubert 			wpa_printf(MSG_DEBUG,
30885732ac8SCy Schubert 				   "WPA: EAPOL-Key MIC algorithm not known (AKM-defined - akmp=0x%x)",
30985732ac8SCy Schubert 				   akmp);
3105b9c547cSRui Paulo 			return -1;
3115b9c547cSRui Paulo 		}
3125b9c547cSRui Paulo 		break;
31339beb93cSSam Leffler 	default:
31485732ac8SCy Schubert 		wpa_printf(MSG_DEBUG,
31585732ac8SCy Schubert 			   "WPA: EAPOL-Key MIC algorithm not known (ver=%d)",
31685732ac8SCy Schubert 			   ver);
31739beb93cSSam Leffler 		return -1;
31839beb93cSSam Leffler 	}
31939beb93cSSam Leffler 
32039beb93cSSam Leffler 	return 0;
32139beb93cSSam Leffler }
32239beb93cSSam Leffler 
32339beb93cSSam Leffler 
32439beb93cSSam Leffler /**
32539beb93cSSam Leffler  * wpa_pmk_to_ptk - Calculate PTK from PMK, addresses, and nonces
32639beb93cSSam Leffler  * @pmk: Pairwise master key
32739beb93cSSam Leffler  * @pmk_len: Length of PMK
32839beb93cSSam Leffler  * @label: Label to use in derivation
32939beb93cSSam Leffler  * @addr1: AA or SA
33039beb93cSSam Leffler  * @addr2: SA or AA
33139beb93cSSam Leffler  * @nonce1: ANonce or SNonce
33239beb93cSSam Leffler  * @nonce2: SNonce or ANonce
33339beb93cSSam Leffler  * @ptk: Buffer for pairwise transient key
3345b9c547cSRui Paulo  * @akmp: Negotiated AKM
3355b9c547cSRui Paulo  * @cipher: Negotiated pairwise cipher
336c1d255d3SCy Schubert  * @kdk_len: The length in octets that should be derived for KDK
3375b9c547cSRui Paulo  * Returns: 0 on success, -1 on failure
33839beb93cSSam Leffler  *
33939beb93cSSam Leffler  * IEEE Std 802.11i-2004 - 8.5.1.2 Pairwise key hierarchy
34039beb93cSSam Leffler  * PTK = PRF-X(PMK, "Pairwise key expansion",
34139beb93cSSam Leffler  *             Min(AA, SA) || Max(AA, SA) ||
3424bc52338SCy Schubert  *             Min(ANonce, SNonce) || Max(ANonce, SNonce)
3434bc52338SCy Schubert  *             [ || Z.x ])
3444bc52338SCy Schubert  *
3454bc52338SCy Schubert  * The optional Z.x component is used only with DPP and that part is not defined
3464bc52338SCy Schubert  * in IEEE 802.11.
34739beb93cSSam Leffler  */
3485b9c547cSRui Paulo int wpa_pmk_to_ptk(const u8 *pmk, size_t pmk_len, const char *label,
34939beb93cSSam Leffler 		   const u8 *addr1, const u8 *addr2,
35039beb93cSSam Leffler 		   const u8 *nonce1, const u8 *nonce2,
3514bc52338SCy Schubert 		   struct wpa_ptk *ptk, int akmp, int cipher,
352c1d255d3SCy Schubert 		   const u8 *z, size_t z_len, size_t kdk_len)
35339beb93cSSam Leffler {
3544bc52338SCy Schubert #define MAX_Z_LEN 66 /* with NIST P-521 */
3554bc52338SCy Schubert 	u8 data[2 * ETH_ALEN + 2 * WPA_NONCE_LEN + MAX_Z_LEN];
3564bc52338SCy Schubert 	size_t data_len = 2 * ETH_ALEN + 2 * WPA_NONCE_LEN;
357c1d255d3SCy Schubert 	u8 tmp[WPA_KCK_MAX_LEN + WPA_KEK_MAX_LEN + WPA_TK_MAX_LEN +
358c1d255d3SCy Schubert 		WPA_KDK_MAX_LEN];
3595b9c547cSRui Paulo 	size_t ptk_len;
360c1d255d3SCy Schubert #ifdef CONFIG_OWE
361c1d255d3SCy Schubert 	int owe_ptk_workaround = 0;
362c1d255d3SCy Schubert 
363c1d255d3SCy Schubert 	if (akmp == (WPA_KEY_MGMT_OWE | WPA_KEY_MGMT_PSK_SHA256)) {
364c1d255d3SCy Schubert 		owe_ptk_workaround = 1;
365c1d255d3SCy Schubert 		akmp = WPA_KEY_MGMT_OWE;
366c1d255d3SCy Schubert 	}
367c1d255d3SCy Schubert #endif /* CONFIG_OWE */
36839beb93cSSam Leffler 
36985732ac8SCy Schubert 	if (pmk_len == 0) {
37085732ac8SCy Schubert 		wpa_printf(MSG_ERROR, "WPA: No PMK set for PTK derivation");
37185732ac8SCy Schubert 		return -1;
37285732ac8SCy Schubert 	}
37385732ac8SCy Schubert 
3744bc52338SCy Schubert 	if (z_len > MAX_Z_LEN)
3754bc52338SCy Schubert 		return -1;
3764bc52338SCy Schubert 
37739beb93cSSam Leffler 	if (os_memcmp(addr1, addr2, ETH_ALEN) < 0) {
37839beb93cSSam Leffler 		os_memcpy(data, addr1, ETH_ALEN);
37939beb93cSSam Leffler 		os_memcpy(data + ETH_ALEN, addr2, ETH_ALEN);
38039beb93cSSam Leffler 	} else {
38139beb93cSSam Leffler 		os_memcpy(data, addr2, ETH_ALEN);
38239beb93cSSam Leffler 		os_memcpy(data + ETH_ALEN, addr1, ETH_ALEN);
38339beb93cSSam Leffler 	}
38439beb93cSSam Leffler 
38539beb93cSSam Leffler 	if (os_memcmp(nonce1, nonce2, WPA_NONCE_LEN) < 0) {
38639beb93cSSam Leffler 		os_memcpy(data + 2 * ETH_ALEN, nonce1, WPA_NONCE_LEN);
38739beb93cSSam Leffler 		os_memcpy(data + 2 * ETH_ALEN + WPA_NONCE_LEN, nonce2,
38839beb93cSSam Leffler 			  WPA_NONCE_LEN);
38939beb93cSSam Leffler 	} else {
39039beb93cSSam Leffler 		os_memcpy(data + 2 * ETH_ALEN, nonce2, WPA_NONCE_LEN);
39139beb93cSSam Leffler 		os_memcpy(data + 2 * ETH_ALEN + WPA_NONCE_LEN, nonce1,
39239beb93cSSam Leffler 			  WPA_NONCE_LEN);
39339beb93cSSam Leffler 	}
39439beb93cSSam Leffler 
3954bc52338SCy Schubert 	if (z && z_len) {
3964bc52338SCy Schubert 		os_memcpy(data + 2 * ETH_ALEN + 2 * WPA_NONCE_LEN, z, z_len);
3974bc52338SCy Schubert 		data_len += z_len;
3984bc52338SCy Schubert 	}
3994bc52338SCy Schubert 
400c1d255d3SCy Schubert 	if (kdk_len > WPA_KDK_MAX_LEN) {
401c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR,
402c1d255d3SCy Schubert 			   "WPA: KDK len=%zu exceeds max supported len",
403c1d255d3SCy Schubert 			   kdk_len);
404c1d255d3SCy Schubert 		return -1;
405c1d255d3SCy Schubert 	}
406c1d255d3SCy Schubert 
40785732ac8SCy Schubert 	ptk->kck_len = wpa_kck_len(akmp, pmk_len);
40885732ac8SCy Schubert 	ptk->kek_len = wpa_kek_len(akmp, pmk_len);
4095b9c547cSRui Paulo 	ptk->tk_len = wpa_cipher_key_len(cipher);
410c1d255d3SCy Schubert 	ptk->kdk_len = kdk_len;
41185732ac8SCy Schubert 	if (ptk->tk_len == 0) {
41285732ac8SCy Schubert 		wpa_printf(MSG_ERROR,
41385732ac8SCy Schubert 			   "WPA: Unsupported cipher (0x%x) used in PTK derivation",
41485732ac8SCy Schubert 			   cipher);
41585732ac8SCy Schubert 		return -1;
41685732ac8SCy Schubert 	}
417c1d255d3SCy Schubert 	ptk_len = ptk->kck_len + ptk->kek_len + ptk->tk_len + ptk->kdk_len;
4185b9c547cSRui Paulo 
41985732ac8SCy Schubert 	if (wpa_key_mgmt_sha384(akmp)) {
42085732ac8SCy Schubert #if defined(CONFIG_SUITEB192) || defined(CONFIG_FILS)
42185732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: PTK derivation using PRF(SHA384)");
4224bc52338SCy Schubert 		if (sha384_prf(pmk, pmk_len, label, data, data_len,
42385732ac8SCy Schubert 			       tmp, ptk_len) < 0)
42485732ac8SCy Schubert 			return -1;
42585732ac8SCy Schubert #else /* CONFIG_SUITEB192 || CONFIG_FILS */
42685732ac8SCy Schubert 		return -1;
42785732ac8SCy Schubert #endif /* CONFIG_SUITEB192 || CONFIG_FILS */
428c1d255d3SCy Schubert 	} else if (wpa_key_mgmt_sha256(akmp)) {
42985732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: PTK derivation using PRF(SHA256)");
4304bc52338SCy Schubert 		if (sha256_prf(pmk, pmk_len, label, data, data_len,
43185732ac8SCy Schubert 			       tmp, ptk_len) < 0)
43285732ac8SCy Schubert 			return -1;
433c1d255d3SCy Schubert #ifdef CONFIG_OWE
434c1d255d3SCy Schubert 	} else if (akmp == WPA_KEY_MGMT_OWE && (pmk_len == 32 ||
435c1d255d3SCy Schubert 						owe_ptk_workaround)) {
436c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: PTK derivation using PRF(SHA256)");
437c1d255d3SCy Schubert 		if (sha256_prf(pmk, pmk_len, label, data, data_len,
438c1d255d3SCy Schubert 			       tmp, ptk_len) < 0)
43985732ac8SCy Schubert 			return -1;
440c1d255d3SCy Schubert 	} else if (akmp == WPA_KEY_MGMT_OWE && pmk_len == 48) {
441c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: PTK derivation using PRF(SHA384)");
442c1d255d3SCy Schubert 		if (sha384_prf(pmk, pmk_len, label, data, data_len,
443c1d255d3SCy Schubert 			       tmp, ptk_len) < 0)
444c1d255d3SCy Schubert 			return -1;
445c1d255d3SCy Schubert 	} else if (akmp == WPA_KEY_MGMT_OWE && pmk_len == 64) {
446c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: PTK derivation using PRF(SHA512)");
447c1d255d3SCy Schubert 		if (sha512_prf(pmk, pmk_len, label, data, data_len,
448c1d255d3SCy Schubert 			       tmp, ptk_len) < 0)
449c1d255d3SCy Schubert 			return -1;
450c1d255d3SCy Schubert 	} else if (akmp == WPA_KEY_MGMT_OWE) {
451c1d255d3SCy Schubert 		wpa_printf(MSG_INFO, "OWE: Unknown PMK length %u",
452c1d255d3SCy Schubert 			   (unsigned int) pmk_len);
453c1d255d3SCy Schubert 		return -1;
454c1d255d3SCy Schubert #endif /* CONFIG_OWE */
45585732ac8SCy Schubert #ifdef CONFIG_DPP
45685732ac8SCy Schubert 	} else if (akmp == WPA_KEY_MGMT_DPP && pmk_len == 32) {
45785732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: PTK derivation using PRF(SHA256)");
4584bc52338SCy Schubert 		if (sha256_prf(pmk, pmk_len, label, data, data_len,
45985732ac8SCy Schubert 			       tmp, ptk_len) < 0)
46085732ac8SCy Schubert 			return -1;
46185732ac8SCy Schubert 	} else if (akmp == WPA_KEY_MGMT_DPP && pmk_len == 48) {
46285732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: PTK derivation using PRF(SHA384)");
4634bc52338SCy Schubert 		if (sha384_prf(pmk, pmk_len, label, data, data_len,
46485732ac8SCy Schubert 			       tmp, ptk_len) < 0)
46585732ac8SCy Schubert 			return -1;
46685732ac8SCy Schubert 	} else if (akmp == WPA_KEY_MGMT_DPP && pmk_len == 64) {
46785732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: PTK derivation using PRF(SHA512)");
4684bc52338SCy Schubert 		if (sha512_prf(pmk, pmk_len, label, data, data_len,
46985732ac8SCy Schubert 			       tmp, ptk_len) < 0)
47085732ac8SCy Schubert 			return -1;
47185732ac8SCy Schubert 	} else if (akmp == WPA_KEY_MGMT_DPP) {
47285732ac8SCy Schubert 		wpa_printf(MSG_INFO, "DPP: Unknown PMK length %u",
47385732ac8SCy Schubert 			   (unsigned int) pmk_len);
47485732ac8SCy Schubert 		return -1;
47585732ac8SCy Schubert #endif /* CONFIG_DPP */
47685732ac8SCy Schubert 	} else {
47785732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "WPA: PTK derivation using PRF(SHA1)");
4784bc52338SCy Schubert 		if (sha1_prf(pmk, pmk_len, label, data, data_len, tmp,
47985732ac8SCy Schubert 			     ptk_len) < 0)
48085732ac8SCy Schubert 			return -1;
48185732ac8SCy Schubert 	}
48239beb93cSSam Leffler 
48339beb93cSSam Leffler 	wpa_printf(MSG_DEBUG, "WPA: PTK derivation - A1=" MACSTR " A2=" MACSTR,
48439beb93cSSam Leffler 		   MAC2STR(addr1), MAC2STR(addr2));
485f05cddf9SRui Paulo 	wpa_hexdump(MSG_DEBUG, "WPA: Nonce1", nonce1, WPA_NONCE_LEN);
486f05cddf9SRui Paulo 	wpa_hexdump(MSG_DEBUG, "WPA: Nonce2", nonce2, WPA_NONCE_LEN);
4874bc52338SCy Schubert 	if (z && z_len)
4884bc52338SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "WPA: Z.x", z, z_len);
48939beb93cSSam Leffler 	wpa_hexdump_key(MSG_DEBUG, "WPA: PMK", pmk, pmk_len);
4905b9c547cSRui Paulo 	wpa_hexdump_key(MSG_DEBUG, "WPA: PTK", tmp, ptk_len);
4915b9c547cSRui Paulo 
4925b9c547cSRui Paulo 	os_memcpy(ptk->kck, tmp, ptk->kck_len);
4935b9c547cSRui Paulo 	wpa_hexdump_key(MSG_DEBUG, "WPA: KCK", ptk->kck, ptk->kck_len);
4945b9c547cSRui Paulo 
4955b9c547cSRui Paulo 	os_memcpy(ptk->kek, tmp + ptk->kck_len, ptk->kek_len);
4965b9c547cSRui Paulo 	wpa_hexdump_key(MSG_DEBUG, "WPA: KEK", ptk->kek, ptk->kek_len);
4975b9c547cSRui Paulo 
4985b9c547cSRui Paulo 	os_memcpy(ptk->tk, tmp + ptk->kck_len + ptk->kek_len, ptk->tk_len);
4995b9c547cSRui Paulo 	wpa_hexdump_key(MSG_DEBUG, "WPA: TK", ptk->tk, ptk->tk_len);
5005b9c547cSRui Paulo 
501c1d255d3SCy Schubert 	if (kdk_len) {
502c1d255d3SCy Schubert 		os_memcpy(ptk->kdk, tmp + ptk->kck_len + ptk->kek_len +
503c1d255d3SCy Schubert 			  ptk->tk_len, ptk->kdk_len);
504c1d255d3SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "WPA: KDK", ptk->kdk, ptk->kdk_len);
505c1d255d3SCy Schubert 	}
506c1d255d3SCy Schubert 
50785732ac8SCy Schubert 	ptk->kek2_len = 0;
50885732ac8SCy Schubert 	ptk->kck2_len = 0;
50985732ac8SCy Schubert 
5105b9c547cSRui Paulo 	os_memset(tmp, 0, sizeof(tmp));
5114bc52338SCy Schubert 	os_memset(data, 0, data_len);
5125b9c547cSRui Paulo 	return 0;
51339beb93cSSam Leffler }
51439beb93cSSam Leffler 
51585732ac8SCy Schubert #ifdef CONFIG_FILS
51685732ac8SCy Schubert 
51785732ac8SCy Schubert int fils_rmsk_to_pmk(int akmp, const u8 *rmsk, size_t rmsk_len,
51885732ac8SCy Schubert 		     const u8 *snonce, const u8 *anonce, const u8 *dh_ss,
51985732ac8SCy Schubert 		     size_t dh_ss_len, u8 *pmk, size_t *pmk_len)
52085732ac8SCy Schubert {
52185732ac8SCy Schubert 	u8 nonces[2 * FILS_NONCE_LEN];
52285732ac8SCy Schubert 	const u8 *addr[2];
52385732ac8SCy Schubert 	size_t len[2];
52485732ac8SCy Schubert 	size_t num_elem;
52585732ac8SCy Schubert 	int res;
52685732ac8SCy Schubert 
52785732ac8SCy Schubert 	/* PMK = HMAC-Hash(SNonce || ANonce, rMSK [ || DHss ]) */
52885732ac8SCy Schubert 	wpa_printf(MSG_DEBUG, "FILS: rMSK to PMK derivation");
52985732ac8SCy Schubert 
53085732ac8SCy Schubert 	if (wpa_key_mgmt_sha384(akmp))
53185732ac8SCy Schubert 		*pmk_len = SHA384_MAC_LEN;
53285732ac8SCy Schubert 	else if (wpa_key_mgmt_sha256(akmp))
53385732ac8SCy Schubert 		*pmk_len = SHA256_MAC_LEN;
53485732ac8SCy Schubert 	else
53585732ac8SCy Schubert 		return -1;
53685732ac8SCy Schubert 
53785732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FILS: rMSK", rmsk, rmsk_len);
53885732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: SNonce", snonce, FILS_NONCE_LEN);
53985732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: ANonce", anonce, FILS_NONCE_LEN);
54085732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: DHss", dh_ss, dh_ss_len);
54185732ac8SCy Schubert 
54285732ac8SCy Schubert 	os_memcpy(nonces, snonce, FILS_NONCE_LEN);
54385732ac8SCy Schubert 	os_memcpy(&nonces[FILS_NONCE_LEN], anonce, FILS_NONCE_LEN);
54485732ac8SCy Schubert 	addr[0] = rmsk;
54585732ac8SCy Schubert 	len[0] = rmsk_len;
54685732ac8SCy Schubert 	num_elem = 1;
54785732ac8SCy Schubert 	if (dh_ss) {
54885732ac8SCy Schubert 		addr[1] = dh_ss;
54985732ac8SCy Schubert 		len[1] = dh_ss_len;
55085732ac8SCy Schubert 		num_elem++;
55185732ac8SCy Schubert 	}
55285732ac8SCy Schubert 	if (wpa_key_mgmt_sha384(akmp))
55385732ac8SCy Schubert 		res = hmac_sha384_vector(nonces, 2 * FILS_NONCE_LEN, num_elem,
55485732ac8SCy Schubert 					 addr, len, pmk);
55585732ac8SCy Schubert 	else
55685732ac8SCy Schubert 		res = hmac_sha256_vector(nonces, 2 * FILS_NONCE_LEN, num_elem,
55785732ac8SCy Schubert 					 addr, len, pmk);
55885732ac8SCy Schubert 	if (res == 0)
55985732ac8SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "FILS: PMK", pmk, *pmk_len);
56085732ac8SCy Schubert 	else
56185732ac8SCy Schubert 		*pmk_len = 0;
56285732ac8SCy Schubert 	return res;
56385732ac8SCy Schubert }
56485732ac8SCy Schubert 
56585732ac8SCy Schubert 
56685732ac8SCy Schubert int fils_pmkid_erp(int akmp, const u8 *reauth, size_t reauth_len,
56785732ac8SCy Schubert 		   u8 *pmkid)
56885732ac8SCy Schubert {
56985732ac8SCy Schubert 	const u8 *addr[1];
57085732ac8SCy Schubert 	size_t len[1];
57185732ac8SCy Schubert 	u8 hash[SHA384_MAC_LEN];
57285732ac8SCy Schubert 	int res;
57385732ac8SCy Schubert 
57485732ac8SCy Schubert 	/* PMKID = Truncate-128(Hash(EAP-Initiate/Reauth)) */
57585732ac8SCy Schubert 	addr[0] = reauth;
57685732ac8SCy Schubert 	len[0] = reauth_len;
57785732ac8SCy Schubert 	if (wpa_key_mgmt_sha384(akmp))
57885732ac8SCy Schubert 		res = sha384_vector(1, addr, len, hash);
57985732ac8SCy Schubert 	else if (wpa_key_mgmt_sha256(akmp))
58085732ac8SCy Schubert 		res = sha256_vector(1, addr, len, hash);
58185732ac8SCy Schubert 	else
58285732ac8SCy Schubert 		return -1;
58385732ac8SCy Schubert 	if (res)
58485732ac8SCy Schubert 		return res;
58585732ac8SCy Schubert 	os_memcpy(pmkid, hash, PMKID_LEN);
58685732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: PMKID", pmkid, PMKID_LEN);
58785732ac8SCy Schubert 	return 0;
58885732ac8SCy Schubert }
58985732ac8SCy Schubert 
59085732ac8SCy Schubert 
59185732ac8SCy Schubert int fils_pmk_to_ptk(const u8 *pmk, size_t pmk_len, const u8 *spa, const u8 *aa,
59285732ac8SCy Schubert 		    const u8 *snonce, const u8 *anonce, const u8 *dhss,
59385732ac8SCy Schubert 		    size_t dhss_len, struct wpa_ptk *ptk,
59485732ac8SCy Schubert 		    u8 *ick, size_t *ick_len, int akmp, int cipher,
595c1d255d3SCy Schubert 		    u8 *fils_ft, size_t *fils_ft_len, size_t kdk_len)
59685732ac8SCy Schubert {
59785732ac8SCy Schubert 	u8 *data, *pos;
59885732ac8SCy Schubert 	size_t data_len;
59985732ac8SCy Schubert 	u8 tmp[FILS_ICK_MAX_LEN + WPA_KEK_MAX_LEN + WPA_TK_MAX_LEN +
600c1d255d3SCy Schubert 	       FILS_FT_MAX_LEN + WPA_KDK_MAX_LEN];
60185732ac8SCy Schubert 	size_t key_data_len;
60285732ac8SCy Schubert 	const char *label = "FILS PTK Derivation";
60385732ac8SCy Schubert 	int ret = -1;
604c1d255d3SCy Schubert 	size_t offset;
60585732ac8SCy Schubert 
60685732ac8SCy Schubert 	/*
60785732ac8SCy Schubert 	 * FILS-Key-Data = PRF-X(PMK, "FILS PTK Derivation",
60885732ac8SCy Schubert 	 *                       SPA || AA || SNonce || ANonce [ || DHss ])
60985732ac8SCy Schubert 	 * ICK = L(FILS-Key-Data, 0, ICK_bits)
61085732ac8SCy Schubert 	 * KEK = L(FILS-Key-Data, ICK_bits, KEK_bits)
61185732ac8SCy Schubert 	 * TK = L(FILS-Key-Data, ICK_bits + KEK_bits, TK_bits)
61285732ac8SCy Schubert 	 * If doing FT initial mobility domain association:
61385732ac8SCy Schubert 	 * FILS-FT = L(FILS-Key-Data, ICK_bits + KEK_bits + TK_bits,
61485732ac8SCy Schubert 	 *             FILS-FT_bits)
615c1d255d3SCy Schubert 	 * When a KDK is derived:
616c1d255d3SCy Schubert 	 * KDK = L(FILS-Key-Data, ICK_bits + KEK_bits + TK_bits + FILS-FT_bits,
617c1d255d3SCy Schubert 	 *	   KDK_bits)
61885732ac8SCy Schubert 	 */
61985732ac8SCy Schubert 	data_len = 2 * ETH_ALEN + 2 * FILS_NONCE_LEN + dhss_len;
62085732ac8SCy Schubert 	data = os_malloc(data_len);
62185732ac8SCy Schubert 	if (!data)
62285732ac8SCy Schubert 		goto err;
62385732ac8SCy Schubert 	pos = data;
62485732ac8SCy Schubert 	os_memcpy(pos, spa, ETH_ALEN);
62585732ac8SCy Schubert 	pos += ETH_ALEN;
62685732ac8SCy Schubert 	os_memcpy(pos, aa, ETH_ALEN);
62785732ac8SCy Schubert 	pos += ETH_ALEN;
62885732ac8SCy Schubert 	os_memcpy(pos, snonce, FILS_NONCE_LEN);
62985732ac8SCy Schubert 	pos += FILS_NONCE_LEN;
63085732ac8SCy Schubert 	os_memcpy(pos, anonce, FILS_NONCE_LEN);
63185732ac8SCy Schubert 	pos += FILS_NONCE_LEN;
63285732ac8SCy Schubert 	if (dhss)
63385732ac8SCy Schubert 		os_memcpy(pos, dhss, dhss_len);
63485732ac8SCy Schubert 
63585732ac8SCy Schubert 	ptk->kck_len = 0;
63685732ac8SCy Schubert 	ptk->kek_len = wpa_kek_len(akmp, pmk_len);
63785732ac8SCy Schubert 	ptk->tk_len = wpa_cipher_key_len(cipher);
63885732ac8SCy Schubert 	if (wpa_key_mgmt_sha384(akmp))
63985732ac8SCy Schubert 		*ick_len = 48;
64085732ac8SCy Schubert 	else if (wpa_key_mgmt_sha256(akmp))
64185732ac8SCy Schubert 		*ick_len = 32;
64285732ac8SCy Schubert 	else
64385732ac8SCy Schubert 		goto err;
64485732ac8SCy Schubert 	key_data_len = *ick_len + ptk->kek_len + ptk->tk_len;
64585732ac8SCy Schubert 
646c1d255d3SCy Schubert 	if (kdk_len) {
647c1d255d3SCy Schubert 		if (kdk_len > WPA_KDK_MAX_LEN) {
648c1d255d3SCy Schubert 			wpa_printf(MSG_ERROR, "FILS: KDK len=%zu too big",
649c1d255d3SCy Schubert 				   kdk_len);
650c1d255d3SCy Schubert 			goto err;
651c1d255d3SCy Schubert 		}
652c1d255d3SCy Schubert 
653c1d255d3SCy Schubert 		ptk->kdk_len = kdk_len;
654c1d255d3SCy Schubert 		key_data_len += kdk_len;
655c1d255d3SCy Schubert 	} else {
656c1d255d3SCy Schubert 		ptk->kdk_len = 0;
657c1d255d3SCy Schubert 	}
658c1d255d3SCy Schubert 
65985732ac8SCy Schubert 	if (fils_ft && fils_ft_len) {
66085732ac8SCy Schubert 		if (akmp == WPA_KEY_MGMT_FT_FILS_SHA256) {
66185732ac8SCy Schubert 			*fils_ft_len = 32;
66285732ac8SCy Schubert 		} else if (akmp == WPA_KEY_MGMT_FT_FILS_SHA384) {
66385732ac8SCy Schubert 			*fils_ft_len = 48;
66485732ac8SCy Schubert 		} else {
66585732ac8SCy Schubert 			*fils_ft_len = 0;
66685732ac8SCy Schubert 			fils_ft = NULL;
66785732ac8SCy Schubert 		}
66885732ac8SCy Schubert 		key_data_len += *fils_ft_len;
66985732ac8SCy Schubert 	}
67085732ac8SCy Schubert 
67185732ac8SCy Schubert 	if (wpa_key_mgmt_sha384(akmp)) {
67285732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "FILS: PTK derivation using PRF(SHA384)");
67385732ac8SCy Schubert 		if (sha384_prf(pmk, pmk_len, label, data, data_len,
67485732ac8SCy Schubert 			       tmp, key_data_len) < 0)
67585732ac8SCy Schubert 			goto err;
67685732ac8SCy Schubert 	} else {
67785732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "FILS: PTK derivation using PRF(SHA256)");
67885732ac8SCy Schubert 		if (sha256_prf(pmk, pmk_len, label, data, data_len,
67985732ac8SCy Schubert 			       tmp, key_data_len) < 0)
68085732ac8SCy Schubert 			goto err;
68185732ac8SCy Schubert 	}
68285732ac8SCy Schubert 
68385732ac8SCy Schubert 	wpa_printf(MSG_DEBUG, "FILS: PTK derivation - SPA=" MACSTR
68485732ac8SCy Schubert 		   " AA=" MACSTR, MAC2STR(spa), MAC2STR(aa));
68585732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: SNonce", snonce, FILS_NONCE_LEN);
68685732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: ANonce", anonce, FILS_NONCE_LEN);
68785732ac8SCy Schubert 	if (dhss)
68885732ac8SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "FILS: DHss", dhss, dhss_len);
68985732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FILS: PMK", pmk, pmk_len);
69085732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FILS: FILS-Key-Data", tmp, key_data_len);
69185732ac8SCy Schubert 
69285732ac8SCy Schubert 	os_memcpy(ick, tmp, *ick_len);
693c1d255d3SCy Schubert 	offset = *ick_len;
69485732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FILS: ICK", ick, *ick_len);
69585732ac8SCy Schubert 
696c1d255d3SCy Schubert 	os_memcpy(ptk->kek, tmp + offset, ptk->kek_len);
69785732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FILS: KEK", ptk->kek, ptk->kek_len);
698c1d255d3SCy Schubert 	offset += ptk->kek_len;
69985732ac8SCy Schubert 
700c1d255d3SCy Schubert 	os_memcpy(ptk->tk, tmp + offset, ptk->tk_len);
70185732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FILS: TK", ptk->tk, ptk->tk_len);
702c1d255d3SCy Schubert 	offset += ptk->tk_len;
70385732ac8SCy Schubert 
70485732ac8SCy Schubert 	if (fils_ft && fils_ft_len) {
705c1d255d3SCy Schubert 		os_memcpy(fils_ft, tmp + offset, *fils_ft_len);
70685732ac8SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "FILS: FILS-FT",
70785732ac8SCy Schubert 				fils_ft, *fils_ft_len);
708c1d255d3SCy Schubert 		offset += *fils_ft_len;
709c1d255d3SCy Schubert 	}
710c1d255d3SCy Schubert 
711c1d255d3SCy Schubert 	if (ptk->kdk_len) {
712c1d255d3SCy Schubert 		os_memcpy(ptk->kdk, tmp + offset, ptk->kdk_len);
713c1d255d3SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "FILS: KDK", ptk->kdk, ptk->kdk_len);
71485732ac8SCy Schubert 	}
71585732ac8SCy Schubert 
71685732ac8SCy Schubert 	ptk->kek2_len = 0;
71785732ac8SCy Schubert 	ptk->kck2_len = 0;
71885732ac8SCy Schubert 
71985732ac8SCy Schubert 	os_memset(tmp, 0, sizeof(tmp));
72085732ac8SCy Schubert 	ret = 0;
72185732ac8SCy Schubert err:
72285732ac8SCy Schubert 	bin_clear_free(data, data_len);
72385732ac8SCy Schubert 	return ret;
72485732ac8SCy Schubert }
72585732ac8SCy Schubert 
72685732ac8SCy Schubert 
72785732ac8SCy Schubert int fils_key_auth_sk(const u8 *ick, size_t ick_len, const u8 *snonce,
72885732ac8SCy Schubert 		     const u8 *anonce, const u8 *sta_addr, const u8 *bssid,
72985732ac8SCy Schubert 		     const u8 *g_sta, size_t g_sta_len,
73085732ac8SCy Schubert 		     const u8 *g_ap, size_t g_ap_len,
73185732ac8SCy Schubert 		     int akmp, u8 *key_auth_sta, u8 *key_auth_ap,
73285732ac8SCy Schubert 		     size_t *key_auth_len)
73385732ac8SCy Schubert {
73485732ac8SCy Schubert 	const u8 *addr[6];
73585732ac8SCy Schubert 	size_t len[6];
73685732ac8SCy Schubert 	size_t num_elem = 4;
73785732ac8SCy Schubert 	int res;
73885732ac8SCy Schubert 
73985732ac8SCy Schubert 	wpa_printf(MSG_DEBUG, "FILS: Key-Auth derivation: STA-MAC=" MACSTR
74085732ac8SCy Schubert 		   " AP-BSSID=" MACSTR, MAC2STR(sta_addr), MAC2STR(bssid));
74185732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FILS: ICK", ick, ick_len);
74285732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: SNonce", snonce, FILS_NONCE_LEN);
74385732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: ANonce", anonce, FILS_NONCE_LEN);
74485732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: gSTA", g_sta, g_sta_len);
74585732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: gAP", g_ap, g_ap_len);
74685732ac8SCy Schubert 
74785732ac8SCy Schubert 	/*
74885732ac8SCy Schubert 	 * For (Re)Association Request frame (STA->AP):
74985732ac8SCy Schubert 	 * Key-Auth = HMAC-Hash(ICK, SNonce || ANonce || STA-MAC || AP-BSSID
75085732ac8SCy Schubert 	 *                      [ || gSTA || gAP ])
75185732ac8SCy Schubert 	 */
75285732ac8SCy Schubert 	addr[0] = snonce;
75385732ac8SCy Schubert 	len[0] = FILS_NONCE_LEN;
75485732ac8SCy Schubert 	addr[1] = anonce;
75585732ac8SCy Schubert 	len[1] = FILS_NONCE_LEN;
75685732ac8SCy Schubert 	addr[2] = sta_addr;
75785732ac8SCy Schubert 	len[2] = ETH_ALEN;
75885732ac8SCy Schubert 	addr[3] = bssid;
75985732ac8SCy Schubert 	len[3] = ETH_ALEN;
760c1d255d3SCy Schubert 	if (g_sta && g_sta_len && g_ap && g_ap_len) {
76185732ac8SCy Schubert 		addr[4] = g_sta;
76285732ac8SCy Schubert 		len[4] = g_sta_len;
76385732ac8SCy Schubert 		addr[5] = g_ap;
76485732ac8SCy Schubert 		len[5] = g_ap_len;
76585732ac8SCy Schubert 		num_elem = 6;
76685732ac8SCy Schubert 	}
76785732ac8SCy Schubert 
76885732ac8SCy Schubert 	if (wpa_key_mgmt_sha384(akmp)) {
76985732ac8SCy Schubert 		*key_auth_len = 48;
77085732ac8SCy Schubert 		res = hmac_sha384_vector(ick, ick_len, num_elem, addr, len,
77185732ac8SCy Schubert 					 key_auth_sta);
77285732ac8SCy Schubert 	} else if (wpa_key_mgmt_sha256(akmp)) {
77385732ac8SCy Schubert 		*key_auth_len = 32;
77485732ac8SCy Schubert 		res = hmac_sha256_vector(ick, ick_len, num_elem, addr, len,
77585732ac8SCy Schubert 					 key_auth_sta);
77685732ac8SCy Schubert 	} else {
77785732ac8SCy Schubert 		return -1;
77885732ac8SCy Schubert 	}
77985732ac8SCy Schubert 	if (res < 0)
78085732ac8SCy Schubert 		return res;
78185732ac8SCy Schubert 
78285732ac8SCy Schubert 	/*
78385732ac8SCy Schubert 	 * For (Re)Association Response frame (AP->STA):
78485732ac8SCy Schubert 	 * Key-Auth = HMAC-Hash(ICK, ANonce || SNonce || AP-BSSID || STA-MAC
78585732ac8SCy Schubert 	 *                      [ || gAP || gSTA ])
78685732ac8SCy Schubert 	 */
78785732ac8SCy Schubert 	addr[0] = anonce;
78885732ac8SCy Schubert 	addr[1] = snonce;
78985732ac8SCy Schubert 	addr[2] = bssid;
79085732ac8SCy Schubert 	addr[3] = sta_addr;
791c1d255d3SCy Schubert 	if (g_sta && g_sta_len && g_ap && g_ap_len) {
79285732ac8SCy Schubert 		addr[4] = g_ap;
79385732ac8SCy Schubert 		len[4] = g_ap_len;
79485732ac8SCy Schubert 		addr[5] = g_sta;
79585732ac8SCy Schubert 		len[5] = g_sta_len;
79685732ac8SCy Schubert 	}
79785732ac8SCy Schubert 
79885732ac8SCy Schubert 	if (wpa_key_mgmt_sha384(akmp))
79985732ac8SCy Schubert 		res = hmac_sha384_vector(ick, ick_len, num_elem, addr, len,
80085732ac8SCy Schubert 					 key_auth_ap);
80185732ac8SCy Schubert 	else if (wpa_key_mgmt_sha256(akmp))
80285732ac8SCy Schubert 		res = hmac_sha256_vector(ick, ick_len, num_elem, addr, len,
80385732ac8SCy Schubert 					 key_auth_ap);
80485732ac8SCy Schubert 	if (res < 0)
80585732ac8SCy Schubert 		return res;
80685732ac8SCy Schubert 
80785732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: Key-Auth (STA)",
80885732ac8SCy Schubert 		    key_auth_sta, *key_auth_len);
80985732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FILS: Key-Auth (AP)",
81085732ac8SCy Schubert 		    key_auth_ap, *key_auth_len);
81185732ac8SCy Schubert 
81285732ac8SCy Schubert 	return 0;
81385732ac8SCy Schubert }
81485732ac8SCy Schubert 
81585732ac8SCy Schubert #endif /* CONFIG_FILS */
81685732ac8SCy Schubert 
81739beb93cSSam Leffler 
81839beb93cSSam Leffler #ifdef CONFIG_IEEE80211R
8195b9c547cSRui Paulo int wpa_ft_mic(const u8 *kck, size_t kck_len, const u8 *sta_addr,
8205b9c547cSRui Paulo 	       const u8 *ap_addr, u8 transaction_seqnum,
8215b9c547cSRui Paulo 	       const u8 *mdie, size_t mdie_len,
82239beb93cSSam Leffler 	       const u8 *ftie, size_t ftie_len,
82339beb93cSSam Leffler 	       const u8 *rsnie, size_t rsnie_len,
824c1d255d3SCy Schubert 	       const u8 *ric, size_t ric_len,
825c1d255d3SCy Schubert 	       const u8 *rsnxe, size_t rsnxe_len,
826c1d255d3SCy Schubert 	       u8 *mic)
82739beb93cSSam Leffler {
828c1d255d3SCy Schubert 	const u8 *addr[10];
829c1d255d3SCy Schubert 	size_t len[10];
830325151a3SRui Paulo 	size_t i, num_elem = 0;
83185732ac8SCy Schubert 	u8 zero_mic[24];
83285732ac8SCy Schubert 	size_t mic_len, fte_fixed_len;
83339beb93cSSam Leffler 
83485732ac8SCy Schubert 	if (kck_len == 16) {
83585732ac8SCy Schubert 		mic_len = 16;
83685732ac8SCy Schubert #ifdef CONFIG_SHA384
83785732ac8SCy Schubert 	} else if (kck_len == 24) {
83885732ac8SCy Schubert 		mic_len = 24;
83985732ac8SCy Schubert #endif /* CONFIG_SHA384 */
84085732ac8SCy Schubert 	} else {
8415b9c547cSRui Paulo 		wpa_printf(MSG_WARNING, "FT: Unsupported KCK length %u",
8425b9c547cSRui Paulo 			   (unsigned int) kck_len);
8435b9c547cSRui Paulo 		return -1;
8445b9c547cSRui Paulo 	}
8455b9c547cSRui Paulo 
84685732ac8SCy Schubert 	fte_fixed_len = sizeof(struct rsn_ftie) - 16 + mic_len;
84785732ac8SCy Schubert 
848325151a3SRui Paulo 	addr[num_elem] = sta_addr;
849325151a3SRui Paulo 	len[num_elem] = ETH_ALEN;
850325151a3SRui Paulo 	num_elem++;
85139beb93cSSam Leffler 
852325151a3SRui Paulo 	addr[num_elem] = ap_addr;
853325151a3SRui Paulo 	len[num_elem] = ETH_ALEN;
854325151a3SRui Paulo 	num_elem++;
855325151a3SRui Paulo 
856325151a3SRui Paulo 	addr[num_elem] = &transaction_seqnum;
857325151a3SRui Paulo 	len[num_elem] = 1;
858325151a3SRui Paulo 	num_elem++;
859325151a3SRui Paulo 
86039beb93cSSam Leffler 	if (rsnie) {
861325151a3SRui Paulo 		addr[num_elem] = rsnie;
862325151a3SRui Paulo 		len[num_elem] = rsnie_len;
863325151a3SRui Paulo 		num_elem++;
86439beb93cSSam Leffler 	}
86539beb93cSSam Leffler 	if (mdie) {
866325151a3SRui Paulo 		addr[num_elem] = mdie;
867325151a3SRui Paulo 		len[num_elem] = mdie_len;
868325151a3SRui Paulo 		num_elem++;
86939beb93cSSam Leffler 	}
87039beb93cSSam Leffler 	if (ftie) {
87185732ac8SCy Schubert 		if (ftie_len < 2 + fte_fixed_len)
87239beb93cSSam Leffler 			return -1;
873325151a3SRui Paulo 
874325151a3SRui Paulo 		/* IE hdr and mic_control */
875325151a3SRui Paulo 		addr[num_elem] = ftie;
876325151a3SRui Paulo 		len[num_elem] = 2 + 2;
877325151a3SRui Paulo 		num_elem++;
878325151a3SRui Paulo 
879325151a3SRui Paulo 		/* MIC field with all zeros */
88085732ac8SCy Schubert 		os_memset(zero_mic, 0, mic_len);
881325151a3SRui Paulo 		addr[num_elem] = zero_mic;
88285732ac8SCy Schubert 		len[num_elem] = mic_len;
883325151a3SRui Paulo 		num_elem++;
884325151a3SRui Paulo 
885325151a3SRui Paulo 		/* Rest of FTIE */
88685732ac8SCy Schubert 		addr[num_elem] = ftie + 2 + 2 + mic_len;
88785732ac8SCy Schubert 		len[num_elem] = ftie_len - (2 + 2 + mic_len);
888325151a3SRui Paulo 		num_elem++;
88939beb93cSSam Leffler 	}
89039beb93cSSam Leffler 	if (ric) {
891325151a3SRui Paulo 		addr[num_elem] = ric;
892325151a3SRui Paulo 		len[num_elem] = ric_len;
893325151a3SRui Paulo 		num_elem++;
89439beb93cSSam Leffler 	}
89539beb93cSSam Leffler 
896c1d255d3SCy Schubert 	if (rsnxe) {
897c1d255d3SCy Schubert 		addr[num_elem] = rsnxe;
898c1d255d3SCy Schubert 		len[num_elem] = rsnxe_len;
899c1d255d3SCy Schubert 		num_elem++;
900c1d255d3SCy Schubert 	}
901c1d255d3SCy Schubert 
902325151a3SRui Paulo 	for (i = 0; i < num_elem; i++)
903325151a3SRui Paulo 		wpa_hexdump(MSG_MSGDUMP, "FT: MIC data", addr[i], len[i]);
90485732ac8SCy Schubert #ifdef CONFIG_SHA384
90585732ac8SCy Schubert 	if (kck_len == 24) {
90685732ac8SCy Schubert 		u8 hash[SHA384_MAC_LEN];
90785732ac8SCy Schubert 
90885732ac8SCy Schubert 		if (hmac_sha384_vector(kck, kck_len, num_elem, addr, len, hash))
90985732ac8SCy Schubert 			return -1;
91085732ac8SCy Schubert 		os_memcpy(mic, hash, 24);
91185732ac8SCy Schubert 	}
91285732ac8SCy Schubert #endif /* CONFIG_SHA384 */
91385732ac8SCy Schubert 	if (kck_len == 16 &&
91485732ac8SCy Schubert 	    omac1_aes_128_vector(kck, num_elem, addr, len, mic))
91539beb93cSSam Leffler 		return -1;
91639beb93cSSam Leffler 
91739beb93cSSam Leffler 	return 0;
91839beb93cSSam Leffler }
919f05cddf9SRui Paulo 
920f05cddf9SRui Paulo 
921f05cddf9SRui Paulo static int wpa_ft_parse_ftie(const u8 *ie, size_t ie_len,
92285732ac8SCy Schubert 			     struct wpa_ft_ies *parse, int use_sha384)
923f05cddf9SRui Paulo {
924f05cddf9SRui Paulo 	const u8 *end, *pos;
925f05cddf9SRui Paulo 
926f05cddf9SRui Paulo 	parse->ftie = ie;
927f05cddf9SRui Paulo 	parse->ftie_len = ie_len;
928f05cddf9SRui Paulo 
92985732ac8SCy Schubert 	pos = ie + (use_sha384 ? sizeof(struct rsn_ftie_sha384) :
93085732ac8SCy Schubert 		    sizeof(struct rsn_ftie));
931f05cddf9SRui Paulo 	end = ie + ie_len;
93285732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FT: Parse FTE subelements", pos, end - pos);
933f05cddf9SRui Paulo 
934780fb4a2SCy Schubert 	while (end - pos >= 2) {
935780fb4a2SCy Schubert 		u8 id, len;
936780fb4a2SCy Schubert 
937780fb4a2SCy Schubert 		id = *pos++;
938780fb4a2SCy Schubert 		len = *pos++;
93985732ac8SCy Schubert 		if (len > end - pos) {
94085732ac8SCy Schubert 			wpa_printf(MSG_DEBUG, "FT: Truncated subelement");
941780fb4a2SCy Schubert 			break;
94285732ac8SCy Schubert 		}
943780fb4a2SCy Schubert 
944780fb4a2SCy Schubert 		switch (id) {
945f05cddf9SRui Paulo 		case FTIE_SUBELEM_R1KH_ID:
946780fb4a2SCy Schubert 			if (len != FT_R1KH_ID_LEN) {
947780fb4a2SCy Schubert 				wpa_printf(MSG_DEBUG,
948780fb4a2SCy Schubert 					   "FT: Invalid R1KH-ID length in FTIE: %d",
949780fb4a2SCy Schubert 					   len);
950f05cddf9SRui Paulo 				return -1;
951f05cddf9SRui Paulo 			}
952780fb4a2SCy Schubert 			parse->r1kh_id = pos;
953f05cddf9SRui Paulo 			break;
954f05cddf9SRui Paulo 		case FTIE_SUBELEM_GTK:
955780fb4a2SCy Schubert 			parse->gtk = pos;
956780fb4a2SCy Schubert 			parse->gtk_len = len;
957f05cddf9SRui Paulo 			break;
958f05cddf9SRui Paulo 		case FTIE_SUBELEM_R0KH_ID:
959780fb4a2SCy Schubert 			if (len < 1 || len > FT_R0KH_ID_MAX_LEN) {
960780fb4a2SCy Schubert 				wpa_printf(MSG_DEBUG,
961780fb4a2SCy Schubert 					   "FT: Invalid R0KH-ID length in FTIE: %d",
962780fb4a2SCy Schubert 					   len);
963f05cddf9SRui Paulo 				return -1;
964f05cddf9SRui Paulo 			}
965780fb4a2SCy Schubert 			parse->r0kh_id = pos;
966780fb4a2SCy Schubert 			parse->r0kh_id_len = len;
967f05cddf9SRui Paulo 			break;
968f05cddf9SRui Paulo 		case FTIE_SUBELEM_IGTK:
969780fb4a2SCy Schubert 			parse->igtk = pos;
970780fb4a2SCy Schubert 			parse->igtk_len = len;
971f05cddf9SRui Paulo 			break;
9724bc52338SCy Schubert #ifdef CONFIG_OCV
9734bc52338SCy Schubert 		case FTIE_SUBELEM_OCI:
9744bc52338SCy Schubert 			parse->oci = pos;
9754bc52338SCy Schubert 			parse->oci_len = len;
9764bc52338SCy Schubert 			break;
9774bc52338SCy Schubert #endif /* CONFIG_OCV */
978c1d255d3SCy Schubert 		case FTIE_SUBELEM_BIGTK:
979c1d255d3SCy Schubert 			parse->bigtk = pos;
980c1d255d3SCy Schubert 			parse->bigtk_len = len;
981c1d255d3SCy Schubert 			break;
98285732ac8SCy Schubert 		default:
98385732ac8SCy Schubert 			wpa_printf(MSG_DEBUG, "FT: Unknown subelem id %u", id);
98485732ac8SCy Schubert 			break;
985f05cddf9SRui Paulo 		}
986f05cddf9SRui Paulo 
987780fb4a2SCy Schubert 		pos += len;
988f05cddf9SRui Paulo 	}
989f05cddf9SRui Paulo 
990f05cddf9SRui Paulo 	return 0;
991f05cddf9SRui Paulo }
992f05cddf9SRui Paulo 
993f05cddf9SRui Paulo 
994f05cddf9SRui Paulo int wpa_ft_parse_ies(const u8 *ies, size_t ies_len,
99585732ac8SCy Schubert 		     struct wpa_ft_ies *parse, int use_sha384)
996f05cddf9SRui Paulo {
997f05cddf9SRui Paulo 	const u8 *end, *pos;
998f05cddf9SRui Paulo 	struct wpa_ie_data data;
999f05cddf9SRui Paulo 	int ret;
1000f05cddf9SRui Paulo 	const struct rsn_ftie *ftie;
1001f05cddf9SRui Paulo 	int prot_ie_count = 0;
100285732ac8SCy Schubert 	int update_use_sha384 = 0;
100385732ac8SCy Schubert 
100485732ac8SCy Schubert 	if (use_sha384 < 0) {
100585732ac8SCy Schubert 		use_sha384 = 0;
100685732ac8SCy Schubert 		update_use_sha384 = 1;
100785732ac8SCy Schubert 	}
1008f05cddf9SRui Paulo 
1009f05cddf9SRui Paulo 	os_memset(parse, 0, sizeof(*parse));
1010f05cddf9SRui Paulo 	if (ies == NULL)
1011f05cddf9SRui Paulo 		return 0;
1012f05cddf9SRui Paulo 
1013f05cddf9SRui Paulo 	pos = ies;
1014f05cddf9SRui Paulo 	end = ies + ies_len;
1015780fb4a2SCy Schubert 	while (end - pos >= 2) {
1016780fb4a2SCy Schubert 		u8 id, len;
1017780fb4a2SCy Schubert 
1018780fb4a2SCy Schubert 		id = *pos++;
1019780fb4a2SCy Schubert 		len = *pos++;
1020780fb4a2SCy Schubert 		if (len > end - pos)
1021780fb4a2SCy Schubert 			break;
1022780fb4a2SCy Schubert 
1023780fb4a2SCy Schubert 		switch (id) {
1024f05cddf9SRui Paulo 		case WLAN_EID_RSN:
102585732ac8SCy Schubert 			wpa_hexdump(MSG_DEBUG, "FT: RSNE", pos, len);
1026780fb4a2SCy Schubert 			parse->rsn = pos;
1027780fb4a2SCy Schubert 			parse->rsn_len = len;
1028f05cddf9SRui Paulo 			ret = wpa_parse_wpa_ie_rsn(parse->rsn - 2,
1029f05cddf9SRui Paulo 						   parse->rsn_len + 2,
1030f05cddf9SRui Paulo 						   &data);
1031f05cddf9SRui Paulo 			if (ret < 0) {
1032f05cddf9SRui Paulo 				wpa_printf(MSG_DEBUG, "FT: Failed to parse "
1033f05cddf9SRui Paulo 					   "RSN IE: %d", ret);
1034f05cddf9SRui Paulo 				return -1;
1035f05cddf9SRui Paulo 			}
1036c1d255d3SCy Schubert 			parse->rsn_capab = data.capabilities;
1037f05cddf9SRui Paulo 			if (data.num_pmkid == 1 && data.pmkid)
1038f05cddf9SRui Paulo 				parse->rsn_pmkid = data.pmkid;
103985732ac8SCy Schubert 			parse->key_mgmt = data.key_mgmt;
104085732ac8SCy Schubert 			parse->pairwise_cipher = data.pairwise_cipher;
104185732ac8SCy Schubert 			if (update_use_sha384) {
104285732ac8SCy Schubert 				use_sha384 =
104385732ac8SCy Schubert 					wpa_key_mgmt_sha384(parse->key_mgmt);
104485732ac8SCy Schubert 				update_use_sha384 = 0;
104585732ac8SCy Schubert 			}
1046f05cddf9SRui Paulo 			break;
1047c1d255d3SCy Schubert 		case WLAN_EID_RSNX:
1048c1d255d3SCy Schubert 			wpa_hexdump(MSG_DEBUG, "FT: RSNXE", pos, len);
1049c1d255d3SCy Schubert 			if (len < 1)
1050c1d255d3SCy Schubert 				break;
1051c1d255d3SCy Schubert 			parse->rsnxe = pos;
1052c1d255d3SCy Schubert 			parse->rsnxe_len = len;
1053c1d255d3SCy Schubert 			break;
1054f05cddf9SRui Paulo 		case WLAN_EID_MOBILITY_DOMAIN:
105585732ac8SCy Schubert 			wpa_hexdump(MSG_DEBUG, "FT: MDE", pos, len);
1056780fb4a2SCy Schubert 			if (len < sizeof(struct rsn_mdie))
1057325151a3SRui Paulo 				return -1;
1058780fb4a2SCy Schubert 			parse->mdie = pos;
1059780fb4a2SCy Schubert 			parse->mdie_len = len;
1060f05cddf9SRui Paulo 			break;
1061f05cddf9SRui Paulo 		case WLAN_EID_FAST_BSS_TRANSITION:
106285732ac8SCy Schubert 			wpa_hexdump(MSG_DEBUG, "FT: FTE", pos, len);
106385732ac8SCy Schubert 			if (use_sha384) {
106485732ac8SCy Schubert 				const struct rsn_ftie_sha384 *ftie_sha384;
106585732ac8SCy Schubert 
106685732ac8SCy Schubert 				if (len < sizeof(*ftie_sha384))
106785732ac8SCy Schubert 					return -1;
106885732ac8SCy Schubert 				ftie_sha384 =
106985732ac8SCy Schubert 					(const struct rsn_ftie_sha384 *) pos;
107085732ac8SCy Schubert 				wpa_hexdump(MSG_DEBUG, "FT: FTE-MIC Control",
107185732ac8SCy Schubert 					    ftie_sha384->mic_control, 2);
107285732ac8SCy Schubert 				wpa_hexdump(MSG_DEBUG, "FT: FTE-MIC",
107385732ac8SCy Schubert 					    ftie_sha384->mic,
107485732ac8SCy Schubert 					    sizeof(ftie_sha384->mic));
1075c1d255d3SCy Schubert 				parse->fte_anonce = ftie_sha384->anonce;
107685732ac8SCy Schubert 				wpa_hexdump(MSG_DEBUG, "FT: FTE-ANonce",
107785732ac8SCy Schubert 					    ftie_sha384->anonce,
107885732ac8SCy Schubert 					    WPA_NONCE_LEN);
1079c1d255d3SCy Schubert 				parse->fte_snonce = ftie_sha384->snonce;
108085732ac8SCy Schubert 				wpa_hexdump(MSG_DEBUG, "FT: FTE-SNonce",
108185732ac8SCy Schubert 					    ftie_sha384->snonce,
108285732ac8SCy Schubert 					    WPA_NONCE_LEN);
108385732ac8SCy Schubert 				prot_ie_count = ftie_sha384->mic_control[1];
108485732ac8SCy Schubert 				if (wpa_ft_parse_ftie(pos, len, parse, 1) < 0)
108585732ac8SCy Schubert 					return -1;
108685732ac8SCy Schubert 				break;
108785732ac8SCy Schubert 			}
108885732ac8SCy Schubert 
1089780fb4a2SCy Schubert 			if (len < sizeof(*ftie))
1090f05cddf9SRui Paulo 				return -1;
1091780fb4a2SCy Schubert 			ftie = (const struct rsn_ftie *) pos;
109285732ac8SCy Schubert 			wpa_hexdump(MSG_DEBUG, "FT: FTE-MIC Control",
109385732ac8SCy Schubert 				    ftie->mic_control, 2);
109485732ac8SCy Schubert 			wpa_hexdump(MSG_DEBUG, "FT: FTE-MIC",
109585732ac8SCy Schubert 				    ftie->mic, sizeof(ftie->mic));
1096c1d255d3SCy Schubert 			parse->fte_anonce = ftie->anonce;
109785732ac8SCy Schubert 			wpa_hexdump(MSG_DEBUG, "FT: FTE-ANonce",
109885732ac8SCy Schubert 				    ftie->anonce, WPA_NONCE_LEN);
1099c1d255d3SCy Schubert 			parse->fte_snonce = ftie->snonce;
110085732ac8SCy Schubert 			wpa_hexdump(MSG_DEBUG, "FT: FTE-SNonce",
110185732ac8SCy Schubert 				    ftie->snonce, WPA_NONCE_LEN);
1102f05cddf9SRui Paulo 			prot_ie_count = ftie->mic_control[1];
110385732ac8SCy Schubert 			if (wpa_ft_parse_ftie(pos, len, parse, 0) < 0)
1104f05cddf9SRui Paulo 				return -1;
1105f05cddf9SRui Paulo 			break;
1106f05cddf9SRui Paulo 		case WLAN_EID_TIMEOUT_INTERVAL:
110785732ac8SCy Schubert 			wpa_hexdump(MSG_DEBUG, "FT: Timeout Interval",
110885732ac8SCy Schubert 				    pos, len);
1109780fb4a2SCy Schubert 			if (len != 5)
1110325151a3SRui Paulo 				break;
1111780fb4a2SCy Schubert 			parse->tie = pos;
1112780fb4a2SCy Schubert 			parse->tie_len = len;
1113f05cddf9SRui Paulo 			break;
1114f05cddf9SRui Paulo 		case WLAN_EID_RIC_DATA:
1115f05cddf9SRui Paulo 			if (parse->ric == NULL)
1116780fb4a2SCy Schubert 				parse->ric = pos - 2;
1117f05cddf9SRui Paulo 			break;
1118f05cddf9SRui Paulo 		}
1119f05cddf9SRui Paulo 
1120780fb4a2SCy Schubert 		pos += len;
1121f05cddf9SRui Paulo 	}
1122f05cddf9SRui Paulo 
1123f05cddf9SRui Paulo 	if (prot_ie_count == 0)
1124f05cddf9SRui Paulo 		return 0; /* no MIC */
1125f05cddf9SRui Paulo 
1126f05cddf9SRui Paulo 	/*
1127f05cddf9SRui Paulo 	 * Check that the protected IE count matches with IEs included in the
1128f05cddf9SRui Paulo 	 * frame.
1129f05cddf9SRui Paulo 	 */
1130f05cddf9SRui Paulo 	if (parse->rsn)
1131f05cddf9SRui Paulo 		prot_ie_count--;
1132f05cddf9SRui Paulo 	if (parse->mdie)
1133f05cddf9SRui Paulo 		prot_ie_count--;
1134f05cddf9SRui Paulo 	if (parse->ftie)
1135f05cddf9SRui Paulo 		prot_ie_count--;
1136c1d255d3SCy Schubert 	if (parse->rsnxe)
1137c1d255d3SCy Schubert 		prot_ie_count--;
1138f05cddf9SRui Paulo 	if (prot_ie_count < 0) {
1139f05cddf9SRui Paulo 		wpa_printf(MSG_DEBUG, "FT: Some required IEs not included in "
1140f05cddf9SRui Paulo 			   "the protected IE count");
1141f05cddf9SRui Paulo 		return -1;
1142f05cddf9SRui Paulo 	}
1143f05cddf9SRui Paulo 
1144f05cddf9SRui Paulo 	if (prot_ie_count == 0 && parse->ric) {
1145f05cddf9SRui Paulo 		wpa_printf(MSG_DEBUG, "FT: RIC IE(s) in the frame, but not "
1146f05cddf9SRui Paulo 			   "included in protected IE count");
1147f05cddf9SRui Paulo 		return -1;
1148f05cddf9SRui Paulo 	}
1149f05cddf9SRui Paulo 
1150f05cddf9SRui Paulo 	/* Determine the end of the RIC IE(s) */
1151780fb4a2SCy Schubert 	if (parse->ric) {
1152f05cddf9SRui Paulo 		pos = parse->ric;
1153780fb4a2SCy Schubert 		while (end - pos >= 2 && 2 + pos[1] <= end - pos &&
1154f05cddf9SRui Paulo 		       prot_ie_count) {
1155f05cddf9SRui Paulo 			prot_ie_count--;
1156f05cddf9SRui Paulo 			pos += 2 + pos[1];
1157f05cddf9SRui Paulo 		}
1158f05cddf9SRui Paulo 		parse->ric_len = pos - parse->ric;
1159780fb4a2SCy Schubert 	}
1160f05cddf9SRui Paulo 	if (prot_ie_count) {
1161f05cddf9SRui Paulo 		wpa_printf(MSG_DEBUG, "FT: %d protected IEs missing from "
1162f05cddf9SRui Paulo 			   "frame", (int) prot_ie_count);
1163f05cddf9SRui Paulo 		return -1;
1164f05cddf9SRui Paulo 	}
1165f05cddf9SRui Paulo 
1166f05cddf9SRui Paulo 	return 0;
1167f05cddf9SRui Paulo }
116839beb93cSSam Leffler #endif /* CONFIG_IEEE80211R */
116939beb93cSSam Leffler 
117039beb93cSSam Leffler 
1171c1d255d3SCy Schubert #ifdef CONFIG_PASN
1172c1d255d3SCy Schubert 
1173c1d255d3SCy Schubert /*
1174c1d255d3SCy Schubert  * pasn_use_sha384 - Should SHA384 be used or SHA256
1175c1d255d3SCy Schubert  *
1176c1d255d3SCy Schubert  * @akmp: Authentication and key management protocol
1177c1d255d3SCy Schubert  * @cipher: The cipher suite
1178c1d255d3SCy Schubert  *
1179c1d255d3SCy Schubert  * According to IEEE P802.11az/D2.7, 12.12.7, the hash algorithm to use is the
1180c1d255d3SCy Schubert  * hash algorithm defined for the Base AKM (see Table 9-151 (AKM suite
1181c1d255d3SCy Schubert  * selectors)). When there is no Base AKM, the hash algorithm is selected based
1182c1d255d3SCy Schubert  * on the pairwise cipher suite provided in the RSNE by the AP in the second
1183c1d255d3SCy Schubert  * PASN frame. SHA-256 is used as the hash algorithm, except for the ciphers
1184c1d255d3SCy Schubert  * 00-0F-AC:9 and 00-0F-AC:10 for which SHA-384 is used.
1185c1d255d3SCy Schubert  */
1186c1d255d3SCy Schubert static bool pasn_use_sha384(int akmp, int cipher)
1187c1d255d3SCy Schubert {
1188c1d255d3SCy Schubert 	return (akmp == WPA_KEY_MGMT_PASN && (cipher == WPA_CIPHER_CCMP_256 ||
1189c1d255d3SCy Schubert 					      cipher == WPA_CIPHER_GCMP_256)) ||
1190c1d255d3SCy Schubert 		wpa_key_mgmt_sha384(akmp);
1191c1d255d3SCy Schubert }
1192c1d255d3SCy Schubert 
1193c1d255d3SCy Schubert 
1194c1d255d3SCy Schubert /**
1195c1d255d3SCy Schubert  * pasn_pmk_to_ptk - Calculate PASN PTK from PMK, addresses, etc.
1196c1d255d3SCy Schubert  * @pmk: Pairwise master key
1197c1d255d3SCy Schubert  * @pmk_len: Length of PMK
1198c1d255d3SCy Schubert  * @spa: Suppplicant address
1199c1d255d3SCy Schubert  * @bssid: AP BSSID
1200c1d255d3SCy Schubert  * @dhss: Is the shared secret (DHss) derived from the PASN ephemeral key
1201c1d255d3SCy Schubert  *	exchange encoded as an octet string
1202c1d255d3SCy Schubert  * @dhss_len: The length of dhss in octets
1203c1d255d3SCy Schubert  * @ptk: Buffer for pairwise transient key
1204c1d255d3SCy Schubert  * @akmp: Negotiated AKM
1205c1d255d3SCy Schubert  * @cipher: Negotiated pairwise cipher
1206c1d255d3SCy Schubert  * @kdk_len: the length in octets that should be derived for HTLK. Can be zero.
1207c1d255d3SCy Schubert  * Returns: 0 on success, -1 on failure
1208c1d255d3SCy Schubert  */
1209c1d255d3SCy Schubert int pasn_pmk_to_ptk(const u8 *pmk, size_t pmk_len,
1210c1d255d3SCy Schubert 		    const u8 *spa, const u8 *bssid,
1211c1d255d3SCy Schubert 		    const u8 *dhss, size_t dhss_len,
1212c1d255d3SCy Schubert 		    struct wpa_ptk *ptk, int akmp, int cipher,
1213c1d255d3SCy Schubert 		    size_t kdk_len)
1214c1d255d3SCy Schubert {
1215c1d255d3SCy Schubert 	u8 tmp[WPA_KCK_MAX_LEN + WPA_TK_MAX_LEN + WPA_KDK_MAX_LEN];
1216c1d255d3SCy Schubert 	u8 *data;
1217c1d255d3SCy Schubert 	size_t data_len, ptk_len;
1218c1d255d3SCy Schubert 	int ret = -1;
1219c1d255d3SCy Schubert 	const char *label = "PASN PTK Derivation";
1220c1d255d3SCy Schubert 
1221c1d255d3SCy Schubert 	if (!pmk || !pmk_len) {
1222c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR, "PASN: No PMK set for PTK derivation");
1223c1d255d3SCy Schubert 		return -1;
1224c1d255d3SCy Schubert 	}
1225c1d255d3SCy Schubert 
1226c1d255d3SCy Schubert 	if (!dhss || !dhss_len) {
1227c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR, "PASN: No DHss set for PTK derivation");
1228c1d255d3SCy Schubert 		return -1;
1229c1d255d3SCy Schubert 	}
1230c1d255d3SCy Schubert 
1231c1d255d3SCy Schubert 	/*
1232c1d255d3SCy Schubert 	 * PASN-PTK = KDF(PMK, “PASN PTK Derivation”, SPA || BSSID || DHss)
1233c1d255d3SCy Schubert 	 *
1234c1d255d3SCy Schubert 	 * KCK = L(PASN-PTK, 0, 256)
1235c1d255d3SCy Schubert 	 * TK = L(PASN-PTK, 256, TK_bits)
1236c1d255d3SCy Schubert 	 * KDK = L(PASN-PTK, 256 + TK_bits, kdk_len * 8)
1237c1d255d3SCy Schubert 	 */
1238c1d255d3SCy Schubert 	data_len = 2 * ETH_ALEN + dhss_len;
1239c1d255d3SCy Schubert 	data = os_zalloc(data_len);
1240c1d255d3SCy Schubert 	if (!data)
1241c1d255d3SCy Schubert 		return -1;
1242c1d255d3SCy Schubert 
1243c1d255d3SCy Schubert 	os_memcpy(data, spa, ETH_ALEN);
1244c1d255d3SCy Schubert 	os_memcpy(data + ETH_ALEN, bssid, ETH_ALEN);
1245c1d255d3SCy Schubert 	os_memcpy(data + 2 * ETH_ALEN, dhss, dhss_len);
1246c1d255d3SCy Schubert 
1247c1d255d3SCy Schubert 	ptk->kck_len = WPA_PASN_KCK_LEN;
1248c1d255d3SCy Schubert 	ptk->tk_len = wpa_cipher_key_len(cipher);
1249c1d255d3SCy Schubert 	ptk->kdk_len = kdk_len;
1250c1d255d3SCy Schubert 	ptk->kek_len = 0;
1251c1d255d3SCy Schubert 	ptk->kek2_len = 0;
1252c1d255d3SCy Schubert 	ptk->kck2_len = 0;
1253c1d255d3SCy Schubert 
1254c1d255d3SCy Schubert 	if (ptk->tk_len == 0) {
1255c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR,
1256c1d255d3SCy Schubert 			   "PASN: Unsupported cipher (0x%x) used in PTK derivation",
1257c1d255d3SCy Schubert 			   cipher);
1258c1d255d3SCy Schubert 		goto err;
1259c1d255d3SCy Schubert 	}
1260c1d255d3SCy Schubert 
1261c1d255d3SCy Schubert 	ptk_len = ptk->kck_len + ptk->tk_len + ptk->kdk_len;
1262c1d255d3SCy Schubert 	if (ptk_len > sizeof(tmp))
1263c1d255d3SCy Schubert 		goto err;
1264c1d255d3SCy Schubert 
1265c1d255d3SCy Schubert 	if (pasn_use_sha384(akmp, cipher)) {
1266c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: PTK derivation using SHA384");
1267c1d255d3SCy Schubert 
1268c1d255d3SCy Schubert 		if (sha384_prf(pmk, pmk_len, label, data, data_len, tmp,
1269c1d255d3SCy Schubert 			       ptk_len) < 0)
1270c1d255d3SCy Schubert 			goto err;
1271c1d255d3SCy Schubert 	} else {
1272c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: PTK derivation using SHA256");
1273c1d255d3SCy Schubert 
1274c1d255d3SCy Schubert 		if (sha256_prf(pmk, pmk_len, label, data, data_len, tmp,
1275c1d255d3SCy Schubert 			       ptk_len) < 0)
1276c1d255d3SCy Schubert 			goto err;
1277c1d255d3SCy Schubert 	}
1278c1d255d3SCy Schubert 
1279c1d255d3SCy Schubert 	wpa_printf(MSG_DEBUG,
1280c1d255d3SCy Schubert 		   "PASN: PTK derivation: SPA=" MACSTR " BSSID=" MACSTR,
1281c1d255d3SCy Schubert 		   MAC2STR(spa), MAC2STR(bssid));
1282c1d255d3SCy Schubert 
1283c1d255d3SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "PASN: DHss", dhss, dhss_len);
1284c1d255d3SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "PASN: PMK", pmk, pmk_len);
1285c1d255d3SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "PASN: PASN-PTK", tmp, ptk_len);
1286c1d255d3SCy Schubert 
1287c1d255d3SCy Schubert 	os_memcpy(ptk->kck, tmp, WPA_PASN_KCK_LEN);
1288c1d255d3SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "PASN: KCK:", ptk->kck, WPA_PASN_KCK_LEN);
1289c1d255d3SCy Schubert 
1290c1d255d3SCy Schubert 	os_memcpy(ptk->tk, tmp + WPA_PASN_KCK_LEN, ptk->tk_len);
1291c1d255d3SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "PASN: TK:", ptk->tk, ptk->tk_len);
1292c1d255d3SCy Schubert 
1293c1d255d3SCy Schubert 	if (kdk_len) {
1294c1d255d3SCy Schubert 		os_memcpy(ptk->kdk, tmp + WPA_PASN_KCK_LEN + ptk->tk_len,
1295c1d255d3SCy Schubert 			  ptk->kdk_len);
1296c1d255d3SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "PASN: KDK:",
1297c1d255d3SCy Schubert 				ptk->kdk, ptk->kdk_len);
1298c1d255d3SCy Schubert 	}
1299c1d255d3SCy Schubert 
1300c1d255d3SCy Schubert 	forced_memzero(tmp, sizeof(tmp));
1301c1d255d3SCy Schubert 	ret = 0;
1302c1d255d3SCy Schubert err:
1303c1d255d3SCy Schubert 	bin_clear_free(data, data_len);
1304c1d255d3SCy Schubert 	return ret;
1305c1d255d3SCy Schubert }
1306c1d255d3SCy Schubert 
1307c1d255d3SCy Schubert 
1308c1d255d3SCy Schubert /*
1309c1d255d3SCy Schubert  * pasn_mic_len - Returns the MIC length for PASN authentication
1310c1d255d3SCy Schubert  */
1311c1d255d3SCy Schubert u8 pasn_mic_len(int akmp, int cipher)
1312c1d255d3SCy Schubert {
1313c1d255d3SCy Schubert 	if (pasn_use_sha384(akmp, cipher))
1314c1d255d3SCy Schubert 		return 24;
1315c1d255d3SCy Schubert 
1316c1d255d3SCy Schubert 	return 16;
1317c1d255d3SCy Schubert }
1318c1d255d3SCy Schubert 
1319c1d255d3SCy Schubert 
1320c1d255d3SCy Schubert /**
1321c1d255d3SCy Schubert  * pasn_mic - Calculate PASN MIC
1322c1d255d3SCy Schubert  * @kck: The key confirmation key for the PASN PTKSA
1323c1d255d3SCy Schubert  * @akmp: Negotiated AKM
1324c1d255d3SCy Schubert  * @cipher: Negotiated pairwise cipher
1325c1d255d3SCy Schubert  * @addr1: For the 2nd PASN frame supplicant address; for the 3rd frame the
1326c1d255d3SCy Schubert  *	BSSID
1327c1d255d3SCy Schubert  * @addr2: For the 2nd PASN frame the BSSID; for the 3rd frame the supplicant
1328c1d255d3SCy Schubert  *	address
1329c1d255d3SCy Schubert  * @data: For calculating the MIC for the 2nd PASN frame, this should hold the
1330c1d255d3SCy Schubert  *	Beacon frame RSNE + RSNXE. For calculating the MIC for the 3rd PASN
1331c1d255d3SCy Schubert  *	frame, this should hold the hash of the body of the PASN 1st frame.
1332c1d255d3SCy Schubert  * @data_len: The length of data
1333c1d255d3SCy Schubert  * @frame: The body of the PASN frame including the MIC element with the octets
1334c1d255d3SCy Schubert  *	in the MIC field of the MIC element set to 0.
1335c1d255d3SCy Schubert  * @frame_len: The length of frame
1336c1d255d3SCy Schubert  * @mic: Buffer to hold the MIC on success. Should be big enough to handle the
1337c1d255d3SCy Schubert  *	maximal MIC length
1338c1d255d3SCy Schubert  * Returns: 0 on success, -1 on failure
1339c1d255d3SCy Schubert  */
1340c1d255d3SCy Schubert int pasn_mic(const u8 *kck, int akmp, int cipher,
1341c1d255d3SCy Schubert 	     const u8 *addr1, const u8 *addr2,
1342c1d255d3SCy Schubert 	     const u8 *data, size_t data_len,
1343c1d255d3SCy Schubert 	     const u8 *frame, size_t frame_len, u8 *mic)
1344c1d255d3SCy Schubert {
1345c1d255d3SCy Schubert 	u8 *buf;
1346c1d255d3SCy Schubert 	u8 hash[SHA384_MAC_LEN];
1347c1d255d3SCy Schubert 	size_t buf_len = 2 * ETH_ALEN + data_len + frame_len;
1348c1d255d3SCy Schubert 	int ret = -1;
1349c1d255d3SCy Schubert 
1350c1d255d3SCy Schubert 	if (!kck) {
1351c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR, "PASN: No KCK for MIC calculation");
1352c1d255d3SCy Schubert 		return -1;
1353c1d255d3SCy Schubert 	}
1354c1d255d3SCy Schubert 
1355c1d255d3SCy Schubert 	if (!data || !data_len) {
1356c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR, "PASN: invalid data for MIC calculation");
1357c1d255d3SCy Schubert 		return -1;
1358c1d255d3SCy Schubert 	}
1359c1d255d3SCy Schubert 
1360c1d255d3SCy Schubert 	if (!frame || !frame_len) {
1361c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR, "PASN: invalid data for MIC calculation");
1362c1d255d3SCy Schubert 		return -1;
1363c1d255d3SCy Schubert 	}
1364c1d255d3SCy Schubert 
1365c1d255d3SCy Schubert 	buf = os_zalloc(buf_len);
1366c1d255d3SCy Schubert 	if (!buf)
1367c1d255d3SCy Schubert 		return -1;
1368c1d255d3SCy Schubert 
1369c1d255d3SCy Schubert 	os_memcpy(buf, addr1, ETH_ALEN);
1370c1d255d3SCy Schubert 	os_memcpy(buf + ETH_ALEN, addr2, ETH_ALEN);
1371c1d255d3SCy Schubert 
1372c1d255d3SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "PASN: MIC: data", data, data_len);
1373c1d255d3SCy Schubert 	os_memcpy(buf + 2 * ETH_ALEN, data, data_len);
1374c1d255d3SCy Schubert 
1375c1d255d3SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "PASN: MIC: frame", frame, frame_len);
1376c1d255d3SCy Schubert 	os_memcpy(buf + 2 * ETH_ALEN + data_len, frame, frame_len);
1377c1d255d3SCy Schubert 
1378c1d255d3SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "PASN: MIC: KCK", kck, WPA_PASN_KCK_LEN);
1379c1d255d3SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "PASN: MIC: buf", buf, buf_len);
1380c1d255d3SCy Schubert 
1381c1d255d3SCy Schubert 	if (pasn_use_sha384(akmp, cipher)) {
1382c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: MIC using HMAC-SHA384");
1383c1d255d3SCy Schubert 
1384c1d255d3SCy Schubert 		if (hmac_sha384(kck, WPA_PASN_KCK_LEN, buf, buf_len, hash))
1385c1d255d3SCy Schubert 			goto err;
1386c1d255d3SCy Schubert 
1387c1d255d3SCy Schubert 		os_memcpy(mic, hash, 24);
1388c1d255d3SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "PASN: MIC: mic: ", mic, 24);
1389c1d255d3SCy Schubert 	} else {
1390c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: MIC using HMAC-SHA256");
1391c1d255d3SCy Schubert 
1392c1d255d3SCy Schubert 		if (hmac_sha256(kck, WPA_PASN_KCK_LEN, buf, buf_len, hash))
1393c1d255d3SCy Schubert 			goto err;
1394c1d255d3SCy Schubert 
1395c1d255d3SCy Schubert 		os_memcpy(mic, hash, 16);
1396c1d255d3SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "PASN: MIC: mic: ", mic, 16);
1397c1d255d3SCy Schubert 	}
1398c1d255d3SCy Schubert 
1399c1d255d3SCy Schubert 	ret = 0;
1400c1d255d3SCy Schubert err:
1401c1d255d3SCy Schubert 	bin_clear_free(buf, buf_len);
1402c1d255d3SCy Schubert 	return ret;
1403c1d255d3SCy Schubert }
1404c1d255d3SCy Schubert 
1405c1d255d3SCy Schubert 
1406c1d255d3SCy Schubert /**
1407c1d255d3SCy Schubert  * pasn_auth_frame_hash - Computes a hash of an Authentication frame body
1408c1d255d3SCy Schubert  * @akmp: Negotiated AKM
1409c1d255d3SCy Schubert  * @cipher: Negotiated pairwise cipher
1410c1d255d3SCy Schubert  * @data: Pointer to the Authentication frame body
1411c1d255d3SCy Schubert  * @len: Length of the Authentication frame body
1412c1d255d3SCy Schubert  * @hash: On return would hold the computed hash. Should be big enough to handle
1413c1d255d3SCy Schubert  *	SHA384.
1414c1d255d3SCy Schubert  * Returns: 0 on success, -1 on failure
1415c1d255d3SCy Schubert  */
1416c1d255d3SCy Schubert int pasn_auth_frame_hash(int akmp, int cipher, const u8 *data, size_t len,
1417c1d255d3SCy Schubert 			 u8 *hash)
1418c1d255d3SCy Schubert {
1419c1d255d3SCy Schubert 	if (pasn_use_sha384(akmp, cipher)) {
1420c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Frame hash using SHA-384");
1421c1d255d3SCy Schubert 		return sha384_vector(1, &data, &len, hash);
1422c1d255d3SCy Schubert 	} else {
1423c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Frame hash using SHA-256");
1424c1d255d3SCy Schubert 		return sha256_vector(1, &data, &len, hash);
1425c1d255d3SCy Schubert 	}
1426c1d255d3SCy Schubert }
1427c1d255d3SCy Schubert 
1428c1d255d3SCy Schubert #endif /* CONFIG_PASN */
1429c1d255d3SCy Schubert 
1430c1d255d3SCy Schubert 
143139beb93cSSam Leffler static int rsn_selector_to_bitfield(const u8 *s)
143239beb93cSSam Leffler {
143339beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_NONE)
143439beb93cSSam Leffler 		return WPA_CIPHER_NONE;
143539beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_TKIP)
143639beb93cSSam Leffler 		return WPA_CIPHER_TKIP;
143739beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_CCMP)
143839beb93cSSam Leffler 		return WPA_CIPHER_CCMP;
143939beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_AES_128_CMAC)
144039beb93cSSam Leffler 		return WPA_CIPHER_AES_128_CMAC;
1441f05cddf9SRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_GCMP)
1442f05cddf9SRui Paulo 		return WPA_CIPHER_GCMP;
14435b9c547cSRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_CCMP_256)
14445b9c547cSRui Paulo 		return WPA_CIPHER_CCMP_256;
14455b9c547cSRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_GCMP_256)
14465b9c547cSRui Paulo 		return WPA_CIPHER_GCMP_256;
14475b9c547cSRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_BIP_GMAC_128)
14485b9c547cSRui Paulo 		return WPA_CIPHER_BIP_GMAC_128;
14495b9c547cSRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_BIP_GMAC_256)
14505b9c547cSRui Paulo 		return WPA_CIPHER_BIP_GMAC_256;
14515b9c547cSRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_BIP_CMAC_256)
14525b9c547cSRui Paulo 		return WPA_CIPHER_BIP_CMAC_256;
14535b9c547cSRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED)
14545b9c547cSRui Paulo 		return WPA_CIPHER_GTK_NOT_USED;
145539beb93cSSam Leffler 	return 0;
145639beb93cSSam Leffler }
145739beb93cSSam Leffler 
145839beb93cSSam Leffler 
145939beb93cSSam Leffler static int rsn_key_mgmt_to_bitfield(const u8 *s)
146039beb93cSSam Leffler {
146139beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_UNSPEC_802_1X)
146239beb93cSSam Leffler 		return WPA_KEY_MGMT_IEEE8021X;
146339beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X)
146439beb93cSSam Leffler 		return WPA_KEY_MGMT_PSK;
146539beb93cSSam Leffler #ifdef CONFIG_IEEE80211R
146639beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_802_1X)
146739beb93cSSam Leffler 		return WPA_KEY_MGMT_FT_IEEE8021X;
146839beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_PSK)
146939beb93cSSam Leffler 		return WPA_KEY_MGMT_FT_PSK;
147085732ac8SCy Schubert #ifdef CONFIG_SHA384
147185732ac8SCy Schubert 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_802_1X_SHA384)
147285732ac8SCy Schubert 		return WPA_KEY_MGMT_FT_IEEE8021X_SHA384;
147385732ac8SCy Schubert #endif /* CONFIG_SHA384 */
147439beb93cSSam Leffler #endif /* CONFIG_IEEE80211R */
147539beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_802_1X_SHA256)
147639beb93cSSam Leffler 		return WPA_KEY_MGMT_IEEE8021X_SHA256;
147739beb93cSSam Leffler 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_PSK_SHA256)
147839beb93cSSam Leffler 		return WPA_KEY_MGMT_PSK_SHA256;
1479f05cddf9SRui Paulo #ifdef CONFIG_SAE
1480f05cddf9SRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_SAE)
1481f05cddf9SRui Paulo 		return WPA_KEY_MGMT_SAE;
1482f05cddf9SRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_SAE)
1483f05cddf9SRui Paulo 		return WPA_KEY_MGMT_FT_SAE;
1484f05cddf9SRui Paulo #endif /* CONFIG_SAE */
14855b9c547cSRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_802_1X_SUITE_B)
14865b9c547cSRui Paulo 		return WPA_KEY_MGMT_IEEE8021X_SUITE_B;
14875b9c547cSRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_802_1X_SUITE_B_192)
14885b9c547cSRui Paulo 		return WPA_KEY_MGMT_IEEE8021X_SUITE_B_192;
148985732ac8SCy Schubert 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FILS_SHA256)
149085732ac8SCy Schubert 		return WPA_KEY_MGMT_FILS_SHA256;
149185732ac8SCy Schubert 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FILS_SHA384)
149285732ac8SCy Schubert 		return WPA_KEY_MGMT_FILS_SHA384;
149385732ac8SCy Schubert 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_FILS_SHA256)
149485732ac8SCy Schubert 		return WPA_KEY_MGMT_FT_FILS_SHA256;
149585732ac8SCy Schubert 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_FILS_SHA384)
149685732ac8SCy Schubert 		return WPA_KEY_MGMT_FT_FILS_SHA384;
149785732ac8SCy Schubert #ifdef CONFIG_OWE
149885732ac8SCy Schubert 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_OWE)
149985732ac8SCy Schubert 		return WPA_KEY_MGMT_OWE;
150085732ac8SCy Schubert #endif /* CONFIG_OWE */
150185732ac8SCy Schubert #ifdef CONFIG_DPP
150285732ac8SCy Schubert 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_DPP)
150385732ac8SCy Schubert 		return WPA_KEY_MGMT_DPP;
150485732ac8SCy Schubert #endif /* CONFIG_DPP */
1505325151a3SRui Paulo 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_OSEN)
1506325151a3SRui Paulo 		return WPA_KEY_MGMT_OSEN;
1507c1d255d3SCy Schubert #ifdef CONFIG_PASN
1508c1d255d3SCy Schubert 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_PASN)
1509c1d255d3SCy Schubert 		return WPA_KEY_MGMT_PASN;
1510c1d255d3SCy Schubert #endif /* CONFIG_PASN */
151139beb93cSSam Leffler 	return 0;
151239beb93cSSam Leffler }
15135b9c547cSRui Paulo 
15145b9c547cSRui Paulo 
1515325151a3SRui Paulo int wpa_cipher_valid_group(int cipher)
15165b9c547cSRui Paulo {
15175b9c547cSRui Paulo 	return wpa_cipher_valid_pairwise(cipher) ||
15185b9c547cSRui Paulo 		cipher == WPA_CIPHER_GTK_NOT_USED;
15195b9c547cSRui Paulo }
15205b9c547cSRui Paulo 
15215b9c547cSRui Paulo 
15225b9c547cSRui Paulo int wpa_cipher_valid_mgmt_group(int cipher)
15235b9c547cSRui Paulo {
1524c1d255d3SCy Schubert 	return cipher == WPA_CIPHER_GTK_NOT_USED ||
1525c1d255d3SCy Schubert 		cipher == WPA_CIPHER_AES_128_CMAC ||
15265b9c547cSRui Paulo 		cipher == WPA_CIPHER_BIP_GMAC_128 ||
15275b9c547cSRui Paulo 		cipher == WPA_CIPHER_BIP_GMAC_256 ||
15285b9c547cSRui Paulo 		cipher == WPA_CIPHER_BIP_CMAC_256;
15295b9c547cSRui Paulo }
153039beb93cSSam Leffler 
153139beb93cSSam Leffler 
153239beb93cSSam Leffler /**
153339beb93cSSam Leffler  * wpa_parse_wpa_ie_rsn - Parse RSN IE
153439beb93cSSam Leffler  * @rsn_ie: Buffer containing RSN IE
153539beb93cSSam Leffler  * @rsn_ie_len: RSN IE buffer length (including IE number and length octets)
153639beb93cSSam Leffler  * @data: Pointer to structure that will be filled in with parsed data
153739beb93cSSam Leffler  * Returns: 0 on success, <0 on failure
153839beb93cSSam Leffler  */
153939beb93cSSam Leffler int wpa_parse_wpa_ie_rsn(const u8 *rsn_ie, size_t rsn_ie_len,
154039beb93cSSam Leffler 			 struct wpa_ie_data *data)
154139beb93cSSam Leffler {
154239beb93cSSam Leffler 	const u8 *pos;
154339beb93cSSam Leffler 	int left;
154439beb93cSSam Leffler 	int i, count;
154539beb93cSSam Leffler 
154639beb93cSSam Leffler 	os_memset(data, 0, sizeof(*data));
154739beb93cSSam Leffler 	data->proto = WPA_PROTO_RSN;
154839beb93cSSam Leffler 	data->pairwise_cipher = WPA_CIPHER_CCMP;
154939beb93cSSam Leffler 	data->group_cipher = WPA_CIPHER_CCMP;
155039beb93cSSam Leffler 	data->key_mgmt = WPA_KEY_MGMT_IEEE8021X;
155139beb93cSSam Leffler 	data->capabilities = 0;
155239beb93cSSam Leffler 	data->pmkid = NULL;
155339beb93cSSam Leffler 	data->num_pmkid = 0;
155439beb93cSSam Leffler 	data->mgmt_group_cipher = WPA_CIPHER_AES_128_CMAC;
155539beb93cSSam Leffler 
155639beb93cSSam Leffler 	if (rsn_ie_len == 0) {
155739beb93cSSam Leffler 		/* No RSN IE - fail silently */
155839beb93cSSam Leffler 		return -1;
155939beb93cSSam Leffler 	}
156039beb93cSSam Leffler 
156139beb93cSSam Leffler 	if (rsn_ie_len < sizeof(struct rsn_ie_hdr)) {
156239beb93cSSam Leffler 		wpa_printf(MSG_DEBUG, "%s: ie len too short %lu",
156339beb93cSSam Leffler 			   __func__, (unsigned long) rsn_ie_len);
156439beb93cSSam Leffler 		return -1;
156539beb93cSSam Leffler 	}
156639beb93cSSam Leffler 
1567325151a3SRui Paulo 	if (rsn_ie_len >= 6 && rsn_ie[1] >= 4 &&
1568325151a3SRui Paulo 	    rsn_ie[1] == rsn_ie_len - 2 &&
1569325151a3SRui Paulo 	    WPA_GET_BE32(&rsn_ie[2]) == OSEN_IE_VENDOR_TYPE) {
1570325151a3SRui Paulo 		pos = rsn_ie + 6;
1571325151a3SRui Paulo 		left = rsn_ie_len - 6;
1572325151a3SRui Paulo 
157385732ac8SCy Schubert 		data->group_cipher = WPA_CIPHER_GTK_NOT_USED;
15744bc52338SCy Schubert 		data->has_group = 1;
157585732ac8SCy Schubert 		data->key_mgmt = WPA_KEY_MGMT_OSEN;
1576325151a3SRui Paulo 		data->proto = WPA_PROTO_OSEN;
1577325151a3SRui Paulo 	} else {
1578325151a3SRui Paulo 		const struct rsn_ie_hdr *hdr;
1579325151a3SRui Paulo 
158039beb93cSSam Leffler 		hdr = (const struct rsn_ie_hdr *) rsn_ie;
158139beb93cSSam Leffler 
158239beb93cSSam Leffler 		if (hdr->elem_id != WLAN_EID_RSN ||
158339beb93cSSam Leffler 		    hdr->len != rsn_ie_len - 2 ||
158439beb93cSSam Leffler 		    WPA_GET_LE16(hdr->version) != RSN_VERSION) {
158539beb93cSSam Leffler 			wpa_printf(MSG_DEBUG, "%s: malformed ie or unknown version",
158639beb93cSSam Leffler 				   __func__);
158739beb93cSSam Leffler 			return -2;
158839beb93cSSam Leffler 		}
158939beb93cSSam Leffler 
159039beb93cSSam Leffler 		pos = (const u8 *) (hdr + 1);
159139beb93cSSam Leffler 		left = rsn_ie_len - sizeof(*hdr);
1592325151a3SRui Paulo 	}
159339beb93cSSam Leffler 
159439beb93cSSam Leffler 	if (left >= RSN_SELECTOR_LEN) {
159539beb93cSSam Leffler 		data->group_cipher = rsn_selector_to_bitfield(pos);
15964bc52338SCy Schubert 		data->has_group = 1;
15975b9c547cSRui Paulo 		if (!wpa_cipher_valid_group(data->group_cipher)) {
1598780fb4a2SCy Schubert 			wpa_printf(MSG_DEBUG,
1599780fb4a2SCy Schubert 				   "%s: invalid group cipher 0x%x (%08x)",
1600780fb4a2SCy Schubert 				   __func__, data->group_cipher,
1601780fb4a2SCy Schubert 				   WPA_GET_BE32(pos));
160239beb93cSSam Leffler 			return -1;
160339beb93cSSam Leffler 		}
160439beb93cSSam Leffler 		pos += RSN_SELECTOR_LEN;
160539beb93cSSam Leffler 		left -= RSN_SELECTOR_LEN;
160639beb93cSSam Leffler 	} else if (left > 0) {
160739beb93cSSam Leffler 		wpa_printf(MSG_DEBUG, "%s: ie length mismatch, %u too much",
160839beb93cSSam Leffler 			   __func__, left);
160939beb93cSSam Leffler 		return -3;
161039beb93cSSam Leffler 	}
161139beb93cSSam Leffler 
161239beb93cSSam Leffler 	if (left >= 2) {
161339beb93cSSam Leffler 		data->pairwise_cipher = 0;
161439beb93cSSam Leffler 		count = WPA_GET_LE16(pos);
161539beb93cSSam Leffler 		pos += 2;
161639beb93cSSam Leffler 		left -= 2;
16175b9c547cSRui Paulo 		if (count == 0 || count > left / RSN_SELECTOR_LEN) {
161839beb93cSSam Leffler 			wpa_printf(MSG_DEBUG, "%s: ie count botch (pairwise), "
161939beb93cSSam Leffler 				   "count %u left %u", __func__, count, left);
162039beb93cSSam Leffler 			return -4;
162139beb93cSSam Leffler 		}
16224bc52338SCy Schubert 		if (count)
16234bc52338SCy Schubert 			data->has_pairwise = 1;
162439beb93cSSam Leffler 		for (i = 0; i < count; i++) {
162539beb93cSSam Leffler 			data->pairwise_cipher |= rsn_selector_to_bitfield(pos);
162639beb93cSSam Leffler 			pos += RSN_SELECTOR_LEN;
162739beb93cSSam Leffler 			left -= RSN_SELECTOR_LEN;
162839beb93cSSam Leffler 		}
162939beb93cSSam Leffler 		if (data->pairwise_cipher & WPA_CIPHER_AES_128_CMAC) {
163039beb93cSSam Leffler 			wpa_printf(MSG_DEBUG, "%s: AES-128-CMAC used as "
163139beb93cSSam Leffler 				   "pairwise cipher", __func__);
163239beb93cSSam Leffler 			return -1;
163339beb93cSSam Leffler 		}
163439beb93cSSam Leffler 	} else if (left == 1) {
163539beb93cSSam Leffler 		wpa_printf(MSG_DEBUG, "%s: ie too short (for key mgmt)",
163639beb93cSSam Leffler 			   __func__);
163739beb93cSSam Leffler 		return -5;
163839beb93cSSam Leffler 	}
163939beb93cSSam Leffler 
164039beb93cSSam Leffler 	if (left >= 2) {
164139beb93cSSam Leffler 		data->key_mgmt = 0;
164239beb93cSSam Leffler 		count = WPA_GET_LE16(pos);
164339beb93cSSam Leffler 		pos += 2;
164439beb93cSSam Leffler 		left -= 2;
16455b9c547cSRui Paulo 		if (count == 0 || count > left / RSN_SELECTOR_LEN) {
164639beb93cSSam Leffler 			wpa_printf(MSG_DEBUG, "%s: ie count botch (key mgmt), "
164739beb93cSSam Leffler 				   "count %u left %u", __func__, count, left);
164839beb93cSSam Leffler 			return -6;
164939beb93cSSam Leffler 		}
165039beb93cSSam Leffler 		for (i = 0; i < count; i++) {
165139beb93cSSam Leffler 			data->key_mgmt |= rsn_key_mgmt_to_bitfield(pos);
165239beb93cSSam Leffler 			pos += RSN_SELECTOR_LEN;
165339beb93cSSam Leffler 			left -= RSN_SELECTOR_LEN;
165439beb93cSSam Leffler 		}
165539beb93cSSam Leffler 	} else if (left == 1) {
165639beb93cSSam Leffler 		wpa_printf(MSG_DEBUG, "%s: ie too short (for capabilities)",
165739beb93cSSam Leffler 			   __func__);
165839beb93cSSam Leffler 		return -7;
165939beb93cSSam Leffler 	}
166039beb93cSSam Leffler 
166139beb93cSSam Leffler 	if (left >= 2) {
166239beb93cSSam Leffler 		data->capabilities = WPA_GET_LE16(pos);
166339beb93cSSam Leffler 		pos += 2;
166439beb93cSSam Leffler 		left -= 2;
166539beb93cSSam Leffler 	}
166639beb93cSSam Leffler 
166739beb93cSSam Leffler 	if (left >= 2) {
16685b9c547cSRui Paulo 		u16 num_pmkid = WPA_GET_LE16(pos);
166939beb93cSSam Leffler 		pos += 2;
167039beb93cSSam Leffler 		left -= 2;
16715b9c547cSRui Paulo 		if (num_pmkid > (unsigned int) left / PMKID_LEN) {
167239beb93cSSam Leffler 			wpa_printf(MSG_DEBUG, "%s: PMKID underflow "
16735b9c547cSRui Paulo 				   "(num_pmkid=%u left=%d)",
16745b9c547cSRui Paulo 				   __func__, num_pmkid, left);
167539beb93cSSam Leffler 			data->num_pmkid = 0;
167639beb93cSSam Leffler 			return -9;
167739beb93cSSam Leffler 		} else {
16785b9c547cSRui Paulo 			data->num_pmkid = num_pmkid;
167939beb93cSSam Leffler 			data->pmkid = pos;
168039beb93cSSam Leffler 			pos += data->num_pmkid * PMKID_LEN;
168139beb93cSSam Leffler 			left -= data->num_pmkid * PMKID_LEN;
168239beb93cSSam Leffler 		}
168339beb93cSSam Leffler 	}
168439beb93cSSam Leffler 
168539beb93cSSam Leffler 	if (left >= 4) {
168639beb93cSSam Leffler 		data->mgmt_group_cipher = rsn_selector_to_bitfield(pos);
16875b9c547cSRui Paulo 		if (!wpa_cipher_valid_mgmt_group(data->mgmt_group_cipher)) {
1688780fb4a2SCy Schubert 			wpa_printf(MSG_DEBUG,
1689780fb4a2SCy Schubert 				   "%s: Unsupported management group cipher 0x%x (%08x)",
1690780fb4a2SCy Schubert 				   __func__, data->mgmt_group_cipher,
1691780fb4a2SCy Schubert 				   WPA_GET_BE32(pos));
169239beb93cSSam Leffler 			return -10;
169339beb93cSSam Leffler 		}
169439beb93cSSam Leffler 		pos += RSN_SELECTOR_LEN;
169539beb93cSSam Leffler 		left -= RSN_SELECTOR_LEN;
169639beb93cSSam Leffler 	}
169739beb93cSSam Leffler 
169839beb93cSSam Leffler 	if (left > 0) {
16995b9c547cSRui Paulo 		wpa_hexdump(MSG_DEBUG,
17005b9c547cSRui Paulo 			    "wpa_parse_wpa_ie_rsn: ignore trailing bytes",
17015b9c547cSRui Paulo 			    pos, left);
170239beb93cSSam Leffler 	}
170339beb93cSSam Leffler 
170439beb93cSSam Leffler 	return 0;
170539beb93cSSam Leffler }
170639beb93cSSam Leffler 
170739beb93cSSam Leffler 
1708f05cddf9SRui Paulo static int wpa_selector_to_bitfield(const u8 *s)
1709f05cddf9SRui Paulo {
1710f05cddf9SRui Paulo 	if (RSN_SELECTOR_GET(s) == WPA_CIPHER_SUITE_NONE)
1711f05cddf9SRui Paulo 		return WPA_CIPHER_NONE;
1712f05cddf9SRui Paulo 	if (RSN_SELECTOR_GET(s) == WPA_CIPHER_SUITE_TKIP)
1713f05cddf9SRui Paulo 		return WPA_CIPHER_TKIP;
1714f05cddf9SRui Paulo 	if (RSN_SELECTOR_GET(s) == WPA_CIPHER_SUITE_CCMP)
1715f05cddf9SRui Paulo 		return WPA_CIPHER_CCMP;
1716f05cddf9SRui Paulo 	return 0;
1717f05cddf9SRui Paulo }
1718f05cddf9SRui Paulo 
1719f05cddf9SRui Paulo 
1720f05cddf9SRui Paulo static int wpa_key_mgmt_to_bitfield(const u8 *s)
1721f05cddf9SRui Paulo {
1722f05cddf9SRui Paulo 	if (RSN_SELECTOR_GET(s) == WPA_AUTH_KEY_MGMT_UNSPEC_802_1X)
1723f05cddf9SRui Paulo 		return WPA_KEY_MGMT_IEEE8021X;
1724f05cddf9SRui Paulo 	if (RSN_SELECTOR_GET(s) == WPA_AUTH_KEY_MGMT_PSK_OVER_802_1X)
1725f05cddf9SRui Paulo 		return WPA_KEY_MGMT_PSK;
1726f05cddf9SRui Paulo 	if (RSN_SELECTOR_GET(s) == WPA_AUTH_KEY_MGMT_NONE)
1727f05cddf9SRui Paulo 		return WPA_KEY_MGMT_WPA_NONE;
1728f05cddf9SRui Paulo 	return 0;
1729f05cddf9SRui Paulo }
1730f05cddf9SRui Paulo 
1731f05cddf9SRui Paulo 
1732f05cddf9SRui Paulo int wpa_parse_wpa_ie_wpa(const u8 *wpa_ie, size_t wpa_ie_len,
1733f05cddf9SRui Paulo 			 struct wpa_ie_data *data)
1734f05cddf9SRui Paulo {
1735f05cddf9SRui Paulo 	const struct wpa_ie_hdr *hdr;
1736f05cddf9SRui Paulo 	const u8 *pos;
1737f05cddf9SRui Paulo 	int left;
1738f05cddf9SRui Paulo 	int i, count;
1739f05cddf9SRui Paulo 
1740f05cddf9SRui Paulo 	os_memset(data, 0, sizeof(*data));
1741f05cddf9SRui Paulo 	data->proto = WPA_PROTO_WPA;
1742f05cddf9SRui Paulo 	data->pairwise_cipher = WPA_CIPHER_TKIP;
1743f05cddf9SRui Paulo 	data->group_cipher = WPA_CIPHER_TKIP;
1744f05cddf9SRui Paulo 	data->key_mgmt = WPA_KEY_MGMT_IEEE8021X;
1745f05cddf9SRui Paulo 	data->capabilities = 0;
1746f05cddf9SRui Paulo 	data->pmkid = NULL;
1747f05cddf9SRui Paulo 	data->num_pmkid = 0;
1748f05cddf9SRui Paulo 	data->mgmt_group_cipher = 0;
1749f05cddf9SRui Paulo 
1750f05cddf9SRui Paulo 	if (wpa_ie_len < sizeof(struct wpa_ie_hdr)) {
1751f05cddf9SRui Paulo 		wpa_printf(MSG_DEBUG, "%s: ie len too short %lu",
1752f05cddf9SRui Paulo 			   __func__, (unsigned long) wpa_ie_len);
1753f05cddf9SRui Paulo 		return -1;
1754f05cddf9SRui Paulo 	}
1755f05cddf9SRui Paulo 
1756f05cddf9SRui Paulo 	hdr = (const struct wpa_ie_hdr *) wpa_ie;
1757f05cddf9SRui Paulo 
1758f05cddf9SRui Paulo 	if (hdr->elem_id != WLAN_EID_VENDOR_SPECIFIC ||
1759f05cddf9SRui Paulo 	    hdr->len != wpa_ie_len - 2 ||
1760f05cddf9SRui Paulo 	    RSN_SELECTOR_GET(hdr->oui) != WPA_OUI_TYPE ||
1761f05cddf9SRui Paulo 	    WPA_GET_LE16(hdr->version) != WPA_VERSION) {
1762f05cddf9SRui Paulo 		wpa_printf(MSG_DEBUG, "%s: malformed ie or unknown version",
1763f05cddf9SRui Paulo 			   __func__);
1764f05cddf9SRui Paulo 		return -2;
1765f05cddf9SRui Paulo 	}
1766f05cddf9SRui Paulo 
1767f05cddf9SRui Paulo 	pos = (const u8 *) (hdr + 1);
1768f05cddf9SRui Paulo 	left = wpa_ie_len - sizeof(*hdr);
1769f05cddf9SRui Paulo 
1770f05cddf9SRui Paulo 	if (left >= WPA_SELECTOR_LEN) {
1771f05cddf9SRui Paulo 		data->group_cipher = wpa_selector_to_bitfield(pos);
1772f05cddf9SRui Paulo 		pos += WPA_SELECTOR_LEN;
1773f05cddf9SRui Paulo 		left -= WPA_SELECTOR_LEN;
1774f05cddf9SRui Paulo 	} else if (left > 0) {
1775f05cddf9SRui Paulo 		wpa_printf(MSG_DEBUG, "%s: ie length mismatch, %u too much",
1776f05cddf9SRui Paulo 			   __func__, left);
1777f05cddf9SRui Paulo 		return -3;
1778f05cddf9SRui Paulo 	}
1779f05cddf9SRui Paulo 
1780f05cddf9SRui Paulo 	if (left >= 2) {
1781f05cddf9SRui Paulo 		data->pairwise_cipher = 0;
1782f05cddf9SRui Paulo 		count = WPA_GET_LE16(pos);
1783f05cddf9SRui Paulo 		pos += 2;
1784f05cddf9SRui Paulo 		left -= 2;
17855b9c547cSRui Paulo 		if (count == 0 || count > left / WPA_SELECTOR_LEN) {
1786f05cddf9SRui Paulo 			wpa_printf(MSG_DEBUG, "%s: ie count botch (pairwise), "
1787f05cddf9SRui Paulo 				   "count %u left %u", __func__, count, left);
1788f05cddf9SRui Paulo 			return -4;
1789f05cddf9SRui Paulo 		}
1790f05cddf9SRui Paulo 		for (i = 0; i < count; i++) {
1791f05cddf9SRui Paulo 			data->pairwise_cipher |= wpa_selector_to_bitfield(pos);
1792f05cddf9SRui Paulo 			pos += WPA_SELECTOR_LEN;
1793f05cddf9SRui Paulo 			left -= WPA_SELECTOR_LEN;
1794f05cddf9SRui Paulo 		}
1795f05cddf9SRui Paulo 	} else if (left == 1) {
1796f05cddf9SRui Paulo 		wpa_printf(MSG_DEBUG, "%s: ie too short (for key mgmt)",
1797f05cddf9SRui Paulo 			   __func__);
1798f05cddf9SRui Paulo 		return -5;
1799f05cddf9SRui Paulo 	}
1800f05cddf9SRui Paulo 
1801f05cddf9SRui Paulo 	if (left >= 2) {
1802f05cddf9SRui Paulo 		data->key_mgmt = 0;
1803f05cddf9SRui Paulo 		count = WPA_GET_LE16(pos);
1804f05cddf9SRui Paulo 		pos += 2;
1805f05cddf9SRui Paulo 		left -= 2;
18065b9c547cSRui Paulo 		if (count == 0 || count > left / WPA_SELECTOR_LEN) {
1807f05cddf9SRui Paulo 			wpa_printf(MSG_DEBUG, "%s: ie count botch (key mgmt), "
1808f05cddf9SRui Paulo 				   "count %u left %u", __func__, count, left);
1809f05cddf9SRui Paulo 			return -6;
1810f05cddf9SRui Paulo 		}
1811f05cddf9SRui Paulo 		for (i = 0; i < count; i++) {
1812f05cddf9SRui Paulo 			data->key_mgmt |= wpa_key_mgmt_to_bitfield(pos);
1813f05cddf9SRui Paulo 			pos += WPA_SELECTOR_LEN;
1814f05cddf9SRui Paulo 			left -= WPA_SELECTOR_LEN;
1815f05cddf9SRui Paulo 		}
1816f05cddf9SRui Paulo 	} else if (left == 1) {
1817f05cddf9SRui Paulo 		wpa_printf(MSG_DEBUG, "%s: ie too short (for capabilities)",
1818f05cddf9SRui Paulo 			   __func__);
1819f05cddf9SRui Paulo 		return -7;
1820f05cddf9SRui Paulo 	}
1821f05cddf9SRui Paulo 
1822f05cddf9SRui Paulo 	if (left >= 2) {
1823f05cddf9SRui Paulo 		data->capabilities = WPA_GET_LE16(pos);
1824f05cddf9SRui Paulo 		pos += 2;
1825f05cddf9SRui Paulo 		left -= 2;
1826f05cddf9SRui Paulo 	}
1827f05cddf9SRui Paulo 
1828f05cddf9SRui Paulo 	if (left > 0) {
18295b9c547cSRui Paulo 		wpa_hexdump(MSG_DEBUG,
18305b9c547cSRui Paulo 			    "wpa_parse_wpa_ie_wpa: ignore trailing bytes",
18315b9c547cSRui Paulo 			    pos, left);
1832f05cddf9SRui Paulo 	}
1833f05cddf9SRui Paulo 
1834f05cddf9SRui Paulo 	return 0;
1835f05cddf9SRui Paulo }
1836f05cddf9SRui Paulo 
1837f05cddf9SRui Paulo 
18384bc52338SCy Schubert int wpa_default_rsn_cipher(int freq)
18394bc52338SCy Schubert {
18404bc52338SCy Schubert 	if (freq > 56160)
18414bc52338SCy Schubert 		return WPA_CIPHER_GCMP; /* DMG */
18424bc52338SCy Schubert 
18434bc52338SCy Schubert 	return WPA_CIPHER_CCMP;
18444bc52338SCy Schubert }
18454bc52338SCy Schubert 
18464bc52338SCy Schubert 
184739beb93cSSam Leffler #ifdef CONFIG_IEEE80211R
184839beb93cSSam Leffler 
184939beb93cSSam Leffler /**
185039beb93cSSam Leffler  * wpa_derive_pmk_r0 - Derive PMK-R0 and PMKR0Name
185139beb93cSSam Leffler  *
185239beb93cSSam Leffler  * IEEE Std 802.11r-2008 - 8.5.1.5.3
185339beb93cSSam Leffler  */
185485732ac8SCy Schubert int wpa_derive_pmk_r0(const u8 *xxkey, size_t xxkey_len,
185539beb93cSSam Leffler 		      const u8 *ssid, size_t ssid_len,
185639beb93cSSam Leffler 		      const u8 *mdid, const u8 *r0kh_id, size_t r0kh_id_len,
185785732ac8SCy Schubert 		      const u8 *s0kh_id, u8 *pmk_r0, u8 *pmk_r0_name,
185885732ac8SCy Schubert 		      int use_sha384)
185939beb93cSSam Leffler {
1860325151a3SRui Paulo 	u8 buf[1 + SSID_MAX_LEN + MOBILITY_DOMAIN_ID_LEN + 1 +
186139beb93cSSam Leffler 	       FT_R0KH_ID_MAX_LEN + ETH_ALEN];
186285732ac8SCy Schubert 	u8 *pos, r0_key_data[64], hash[48];
186339beb93cSSam Leffler 	const u8 *addr[2];
186439beb93cSSam Leffler 	size_t len[2];
186585732ac8SCy Schubert 	size_t q = use_sha384 ? 48 : 32;
186685732ac8SCy Schubert 	size_t r0_key_data_len = q + 16;
186739beb93cSSam Leffler 
186839beb93cSSam Leffler 	/*
186939beb93cSSam Leffler 	 * R0-Key-Data = KDF-384(XXKey, "FT-R0",
187039beb93cSSam Leffler 	 *                       SSIDlength || SSID || MDID || R0KHlength ||
187139beb93cSSam Leffler 	 *                       R0KH-ID || S0KH-ID)
187285732ac8SCy Schubert 	 * XXKey is either the second 256 bits of MSK or PSK; or the first
187385732ac8SCy Schubert 	 * 384 bits of MSK for FT-EAP-SHA384.
187485732ac8SCy Schubert 	 * PMK-R0 = L(R0-Key-Data, 0, Q)
187585732ac8SCy Schubert 	 * PMK-R0Name-Salt = L(R0-Key-Data, Q, 128)
187685732ac8SCy Schubert 	 * Q = 384 for FT-EAP-SHA384; otherwise, 256
187739beb93cSSam Leffler 	 */
1878325151a3SRui Paulo 	if (ssid_len > SSID_MAX_LEN || r0kh_id_len > FT_R0KH_ID_MAX_LEN)
187985732ac8SCy Schubert 		return -1;
188085732ac8SCy Schubert 	wpa_printf(MSG_DEBUG, "FT: Derive PMK-R0 using KDF-%s",
188185732ac8SCy Schubert 		   use_sha384 ? "SHA384" : "SHA256");
188285732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FT: XXKey", xxkey, xxkey_len);
188385732ac8SCy Schubert 	wpa_hexdump_ascii(MSG_DEBUG, "FT: SSID", ssid, ssid_len);
188485732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FT: MDID", mdid, MOBILITY_DOMAIN_ID_LEN);
188585732ac8SCy Schubert 	wpa_hexdump_ascii(MSG_DEBUG, "FT: R0KH-ID", r0kh_id, r0kh_id_len);
188685732ac8SCy Schubert 	wpa_printf(MSG_DEBUG, "FT: S0KH-ID: " MACSTR, MAC2STR(s0kh_id));
188739beb93cSSam Leffler 	pos = buf;
188839beb93cSSam Leffler 	*pos++ = ssid_len;
188939beb93cSSam Leffler 	os_memcpy(pos, ssid, ssid_len);
189039beb93cSSam Leffler 	pos += ssid_len;
189139beb93cSSam Leffler 	os_memcpy(pos, mdid, MOBILITY_DOMAIN_ID_LEN);
189239beb93cSSam Leffler 	pos += MOBILITY_DOMAIN_ID_LEN;
189339beb93cSSam Leffler 	*pos++ = r0kh_id_len;
189439beb93cSSam Leffler 	os_memcpy(pos, r0kh_id, r0kh_id_len);
189539beb93cSSam Leffler 	pos += r0kh_id_len;
189639beb93cSSam Leffler 	os_memcpy(pos, s0kh_id, ETH_ALEN);
189739beb93cSSam Leffler 	pos += ETH_ALEN;
189839beb93cSSam Leffler 
189985732ac8SCy Schubert #ifdef CONFIG_SHA384
190085732ac8SCy Schubert 	if (use_sha384) {
190185732ac8SCy Schubert 		if (xxkey_len != SHA384_MAC_LEN) {
190285732ac8SCy Schubert 			wpa_printf(MSG_ERROR,
190385732ac8SCy Schubert 				   "FT: Unexpected XXKey length %d (expected %d)",
190485732ac8SCy Schubert 				   (int) xxkey_len, SHA384_MAC_LEN);
190585732ac8SCy Schubert 			return -1;
190685732ac8SCy Schubert 		}
190785732ac8SCy Schubert 		if (sha384_prf(xxkey, xxkey_len, "FT-R0", buf, pos - buf,
190885732ac8SCy Schubert 			       r0_key_data, r0_key_data_len) < 0)
190985732ac8SCy Schubert 			return -1;
191085732ac8SCy Schubert 	}
191185732ac8SCy Schubert #endif /* CONFIG_SHA384 */
191285732ac8SCy Schubert 	if (!use_sha384) {
191385732ac8SCy Schubert 		if (xxkey_len != PMK_LEN) {
191485732ac8SCy Schubert 			wpa_printf(MSG_ERROR,
191585732ac8SCy Schubert 				   "FT: Unexpected XXKey length %d (expected %d)",
191685732ac8SCy Schubert 				   (int) xxkey_len, PMK_LEN);
191785732ac8SCy Schubert 			return -1;
191885732ac8SCy Schubert 		}
191985732ac8SCy Schubert 		if (sha256_prf(xxkey, xxkey_len, "FT-R0", buf, pos - buf,
192085732ac8SCy Schubert 			       r0_key_data, r0_key_data_len) < 0)
192185732ac8SCy Schubert 			return -1;
192285732ac8SCy Schubert 	}
192385732ac8SCy Schubert 	os_memcpy(pmk_r0, r0_key_data, q);
192485732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R0", pmk_r0, q);
192585732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R0Name-Salt", &r0_key_data[q], 16);
192639beb93cSSam Leffler 
192739beb93cSSam Leffler 	/*
192885732ac8SCy Schubert 	 * PMKR0Name = Truncate-128(Hash("FT-R0N" || PMK-R0Name-Salt)
192939beb93cSSam Leffler 	 */
193039beb93cSSam Leffler 	addr[0] = (const u8 *) "FT-R0N";
193139beb93cSSam Leffler 	len[0] = 6;
193285732ac8SCy Schubert 	addr[1] = &r0_key_data[q];
193339beb93cSSam Leffler 	len[1] = 16;
193439beb93cSSam Leffler 
193585732ac8SCy Schubert #ifdef CONFIG_SHA384
193685732ac8SCy Schubert 	if (use_sha384 && sha384_vector(2, addr, len, hash) < 0)
193785732ac8SCy Schubert 		return -1;
193885732ac8SCy Schubert #endif /* CONFIG_SHA384 */
193985732ac8SCy Schubert 	if (!use_sha384 && sha256_vector(2, addr, len, hash) < 0)
194085732ac8SCy Schubert 		return -1;
194139beb93cSSam Leffler 	os_memcpy(pmk_r0_name, hash, WPA_PMK_NAME_LEN);
1942c1d255d3SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FT: PMKR0Name", pmk_r0_name, WPA_PMK_NAME_LEN);
1943c1d255d3SCy Schubert 	forced_memzero(r0_key_data, sizeof(r0_key_data));
194485732ac8SCy Schubert 	return 0;
194539beb93cSSam Leffler }
194639beb93cSSam Leffler 
194739beb93cSSam Leffler 
194839beb93cSSam Leffler /**
194939beb93cSSam Leffler  * wpa_derive_pmk_r1_name - Derive PMKR1Name
195039beb93cSSam Leffler  *
195139beb93cSSam Leffler  * IEEE Std 802.11r-2008 - 8.5.1.5.4
195239beb93cSSam Leffler  */
195385732ac8SCy Schubert int wpa_derive_pmk_r1_name(const u8 *pmk_r0_name, const u8 *r1kh_id,
195485732ac8SCy Schubert 			   const u8 *s1kh_id, u8 *pmk_r1_name, int use_sha384)
195539beb93cSSam Leffler {
195685732ac8SCy Schubert 	u8 hash[48];
195739beb93cSSam Leffler 	const u8 *addr[4];
195839beb93cSSam Leffler 	size_t len[4];
195939beb93cSSam Leffler 
196039beb93cSSam Leffler 	/*
196185732ac8SCy Schubert 	 * PMKR1Name = Truncate-128(Hash("FT-R1N" || PMKR0Name ||
196239beb93cSSam Leffler 	 *                               R1KH-ID || S1KH-ID))
196339beb93cSSam Leffler 	 */
196439beb93cSSam Leffler 	addr[0] = (const u8 *) "FT-R1N";
196539beb93cSSam Leffler 	len[0] = 6;
196639beb93cSSam Leffler 	addr[1] = pmk_r0_name;
196739beb93cSSam Leffler 	len[1] = WPA_PMK_NAME_LEN;
196839beb93cSSam Leffler 	addr[2] = r1kh_id;
196939beb93cSSam Leffler 	len[2] = FT_R1KH_ID_LEN;
197039beb93cSSam Leffler 	addr[3] = s1kh_id;
197139beb93cSSam Leffler 	len[3] = ETH_ALEN;
197239beb93cSSam Leffler 
197385732ac8SCy Schubert #ifdef CONFIG_SHA384
197485732ac8SCy Schubert 	if (use_sha384 && sha384_vector(4, addr, len, hash) < 0)
197585732ac8SCy Schubert 		return -1;
197685732ac8SCy Schubert #endif /* CONFIG_SHA384 */
197785732ac8SCy Schubert 	if (!use_sha384 && sha256_vector(4, addr, len, hash) < 0)
197885732ac8SCy Schubert 		return -1;
197939beb93cSSam Leffler 	os_memcpy(pmk_r1_name, hash, WPA_PMK_NAME_LEN);
1980c1d255d3SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name", pmk_r1_name, WPA_PMK_NAME_LEN);
198185732ac8SCy Schubert 	return 0;
198239beb93cSSam Leffler }
198339beb93cSSam Leffler 
198439beb93cSSam Leffler 
198539beb93cSSam Leffler /**
198639beb93cSSam Leffler  * wpa_derive_pmk_r1 - Derive PMK-R1 and PMKR1Name from PMK-R0
198739beb93cSSam Leffler  *
198839beb93cSSam Leffler  * IEEE Std 802.11r-2008 - 8.5.1.5.4
198939beb93cSSam Leffler  */
199085732ac8SCy Schubert int wpa_derive_pmk_r1(const u8 *pmk_r0, size_t pmk_r0_len,
199185732ac8SCy Schubert 		      const u8 *pmk_r0_name,
199239beb93cSSam Leffler 		      const u8 *r1kh_id, const u8 *s1kh_id,
199339beb93cSSam Leffler 		      u8 *pmk_r1, u8 *pmk_r1_name)
199439beb93cSSam Leffler {
199539beb93cSSam Leffler 	u8 buf[FT_R1KH_ID_LEN + ETH_ALEN];
199639beb93cSSam Leffler 	u8 *pos;
199739beb93cSSam Leffler 
199839beb93cSSam Leffler 	/* PMK-R1 = KDF-256(PMK-R0, "FT-R1", R1KH-ID || S1KH-ID) */
199985732ac8SCy Schubert 	wpa_printf(MSG_DEBUG, "FT: Derive PMK-R1 using KDF-%s",
200085732ac8SCy Schubert 		   pmk_r0_len == SHA384_MAC_LEN ? "SHA384" : "SHA256");
200185732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R0", pmk_r0, pmk_r0_len);
200285732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FT: R1KH-ID", r1kh_id, FT_R1KH_ID_LEN);
200385732ac8SCy Schubert 	wpa_printf(MSG_DEBUG, "FT: S1KH-ID: " MACSTR, MAC2STR(s1kh_id));
200439beb93cSSam Leffler 	pos = buf;
200539beb93cSSam Leffler 	os_memcpy(pos, r1kh_id, FT_R1KH_ID_LEN);
200639beb93cSSam Leffler 	pos += FT_R1KH_ID_LEN;
200739beb93cSSam Leffler 	os_memcpy(pos, s1kh_id, ETH_ALEN);
200839beb93cSSam Leffler 	pos += ETH_ALEN;
200939beb93cSSam Leffler 
201085732ac8SCy Schubert #ifdef CONFIG_SHA384
201185732ac8SCy Schubert 	if (pmk_r0_len == SHA384_MAC_LEN &&
201285732ac8SCy Schubert 	    sha384_prf(pmk_r0, pmk_r0_len, "FT-R1",
201385732ac8SCy Schubert 		       buf, pos - buf, pmk_r1, pmk_r0_len) < 0)
201485732ac8SCy Schubert 		return -1;
201585732ac8SCy Schubert #endif /* CONFIG_SHA384 */
201685732ac8SCy Schubert 	if (pmk_r0_len == PMK_LEN &&
201785732ac8SCy Schubert 	    sha256_prf(pmk_r0, pmk_r0_len, "FT-R1",
201885732ac8SCy Schubert 		       buf, pos - buf, pmk_r1, pmk_r0_len) < 0)
201985732ac8SCy Schubert 		return -1;
202085732ac8SCy Schubert 	if (pmk_r0_len != SHA384_MAC_LEN && pmk_r0_len != PMK_LEN) {
202185732ac8SCy Schubert 		wpa_printf(MSG_ERROR, "FT: Unexpected PMK-R0 length %d",
202285732ac8SCy Schubert 			   (int) pmk_r0_len);
202385732ac8SCy Schubert 		return -1;
202485732ac8SCy Schubert 	}
202585732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1", pmk_r1, pmk_r0_len);
202639beb93cSSam Leffler 
202785732ac8SCy Schubert 	return wpa_derive_pmk_r1_name(pmk_r0_name, r1kh_id, s1kh_id,
202885732ac8SCy Schubert 				      pmk_r1_name,
202985732ac8SCy Schubert 				      pmk_r0_len == SHA384_MAC_LEN);
203039beb93cSSam Leffler }
203139beb93cSSam Leffler 
203239beb93cSSam Leffler 
203339beb93cSSam Leffler /**
203439beb93cSSam Leffler  * wpa_pmk_r1_to_ptk - Derive PTK and PTKName from PMK-R1
203539beb93cSSam Leffler  *
203639beb93cSSam Leffler  * IEEE Std 802.11r-2008 - 8.5.1.5.5
203739beb93cSSam Leffler  */
203885732ac8SCy Schubert int wpa_pmk_r1_to_ptk(const u8 *pmk_r1, size_t pmk_r1_len,
203985732ac8SCy Schubert 		      const u8 *snonce, const u8 *anonce,
204039beb93cSSam Leffler 		      const u8 *sta_addr, const u8 *bssid,
204139beb93cSSam Leffler 		      const u8 *pmk_r1_name,
2042c1d255d3SCy Schubert 		      struct wpa_ptk *ptk, u8 *ptk_name, int akmp, int cipher,
2043c1d255d3SCy Schubert 		      size_t kdk_len)
204439beb93cSSam Leffler {
204539beb93cSSam Leffler 	u8 buf[2 * WPA_NONCE_LEN + 2 * ETH_ALEN];
204639beb93cSSam Leffler 	u8 *pos, hash[32];
204739beb93cSSam Leffler 	const u8 *addr[6];
204839beb93cSSam Leffler 	size_t len[6];
2049c1d255d3SCy Schubert 	u8 tmp[2 * WPA_KCK_MAX_LEN + 2 * WPA_KEK_MAX_LEN + WPA_TK_MAX_LEN +
2050c1d255d3SCy Schubert 	       WPA_KDK_MAX_LEN];
205185732ac8SCy Schubert 	size_t ptk_len, offset;
205285732ac8SCy Schubert 	int use_sha384 = wpa_key_mgmt_sha384(akmp);
205339beb93cSSam Leffler 
2054c1d255d3SCy Schubert 	if (kdk_len > WPA_KDK_MAX_LEN) {
2055c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR,
2056c1d255d3SCy Schubert 			   "FT: KDK len=%zu exceeds max supported len",
2057c1d255d3SCy Schubert 			   kdk_len);
2058c1d255d3SCy Schubert 		return -1;
2059c1d255d3SCy Schubert 	}
2060c1d255d3SCy Schubert 
206139beb93cSSam Leffler 	/*
206239beb93cSSam Leffler 	 * PTK = KDF-PTKLen(PMK-R1, "FT-PTK", SNonce || ANonce ||
206339beb93cSSam Leffler 	 *                  BSSID || STA-ADDR)
206439beb93cSSam Leffler 	 */
206585732ac8SCy Schubert 	wpa_printf(MSG_DEBUG, "FT: Derive PTK using KDF-%s",
206685732ac8SCy Schubert 		   use_sha384 ? "SHA384" : "SHA256");
206785732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1", pmk_r1, pmk_r1_len);
206885732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FT: SNonce", snonce, WPA_NONCE_LEN);
206985732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FT: ANonce", anonce, WPA_NONCE_LEN);
207085732ac8SCy Schubert 	wpa_printf(MSG_DEBUG, "FT: BSSID=" MACSTR " STA-ADDR=" MACSTR,
207185732ac8SCy Schubert 		   MAC2STR(bssid), MAC2STR(sta_addr));
207239beb93cSSam Leffler 	pos = buf;
207339beb93cSSam Leffler 	os_memcpy(pos, snonce, WPA_NONCE_LEN);
207439beb93cSSam Leffler 	pos += WPA_NONCE_LEN;
207539beb93cSSam Leffler 	os_memcpy(pos, anonce, WPA_NONCE_LEN);
207639beb93cSSam Leffler 	pos += WPA_NONCE_LEN;
207739beb93cSSam Leffler 	os_memcpy(pos, bssid, ETH_ALEN);
207839beb93cSSam Leffler 	pos += ETH_ALEN;
207939beb93cSSam Leffler 	os_memcpy(pos, sta_addr, ETH_ALEN);
208039beb93cSSam Leffler 	pos += ETH_ALEN;
208139beb93cSSam Leffler 
208285732ac8SCy Schubert 	ptk->kck_len = wpa_kck_len(akmp, PMK_LEN);
208385732ac8SCy Schubert 	ptk->kck2_len = wpa_kck2_len(akmp);
208485732ac8SCy Schubert 	ptk->kek_len = wpa_kek_len(akmp, PMK_LEN);
208585732ac8SCy Schubert 	ptk->kek2_len = wpa_kek2_len(akmp);
20865b9c547cSRui Paulo 	ptk->tk_len = wpa_cipher_key_len(cipher);
2087c1d255d3SCy Schubert 	ptk->kdk_len = kdk_len;
208885732ac8SCy Schubert 	ptk_len = ptk->kck_len + ptk->kek_len + ptk->tk_len +
2089c1d255d3SCy Schubert 		ptk->kck2_len + ptk->kek2_len + ptk->kdk_len;
20905b9c547cSRui Paulo 
209185732ac8SCy Schubert #ifdef CONFIG_SHA384
209285732ac8SCy Schubert 	if (use_sha384) {
209385732ac8SCy Schubert 		if (pmk_r1_len != SHA384_MAC_LEN) {
209485732ac8SCy Schubert 			wpa_printf(MSG_ERROR,
209585732ac8SCy Schubert 				   "FT: Unexpected PMK-R1 length %d (expected %d)",
209685732ac8SCy Schubert 				   (int) pmk_r1_len, SHA384_MAC_LEN);
209785732ac8SCy Schubert 			return -1;
209885732ac8SCy Schubert 		}
209985732ac8SCy Schubert 		if (sha384_prf(pmk_r1, pmk_r1_len, "FT-PTK",
210085732ac8SCy Schubert 			       buf, pos - buf, tmp, ptk_len) < 0)
210185732ac8SCy Schubert 			return -1;
210285732ac8SCy Schubert 	}
210385732ac8SCy Schubert #endif /* CONFIG_SHA384 */
210485732ac8SCy Schubert 	if (!use_sha384) {
210585732ac8SCy Schubert 		if (pmk_r1_len != PMK_LEN) {
210685732ac8SCy Schubert 			wpa_printf(MSG_ERROR,
210785732ac8SCy Schubert 				   "FT: Unexpected PMK-R1 length %d (expected %d)",
210885732ac8SCy Schubert 				   (int) pmk_r1_len, PMK_LEN);
210985732ac8SCy Schubert 			return -1;
211085732ac8SCy Schubert 		}
211185732ac8SCy Schubert 		if (sha256_prf(pmk_r1, pmk_r1_len, "FT-PTK",
211285732ac8SCy Schubert 			       buf, pos - buf, tmp, ptk_len) < 0)
211385732ac8SCy Schubert 			return -1;
211485732ac8SCy Schubert 	}
211585732ac8SCy Schubert 	wpa_hexdump_key(MSG_DEBUG, "FT: PTK", tmp, ptk_len);
211639beb93cSSam Leffler 
211739beb93cSSam Leffler 	/*
211839beb93cSSam Leffler 	 * PTKName = Truncate-128(SHA-256(PMKR1Name || "FT-PTKN" || SNonce ||
211939beb93cSSam Leffler 	 *                                ANonce || BSSID || STA-ADDR))
212039beb93cSSam Leffler 	 */
212185732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name", pmk_r1_name, WPA_PMK_NAME_LEN);
212239beb93cSSam Leffler 	addr[0] = pmk_r1_name;
212339beb93cSSam Leffler 	len[0] = WPA_PMK_NAME_LEN;
212439beb93cSSam Leffler 	addr[1] = (const u8 *) "FT-PTKN";
212539beb93cSSam Leffler 	len[1] = 7;
212639beb93cSSam Leffler 	addr[2] = snonce;
212739beb93cSSam Leffler 	len[2] = WPA_NONCE_LEN;
212839beb93cSSam Leffler 	addr[3] = anonce;
212939beb93cSSam Leffler 	len[3] = WPA_NONCE_LEN;
213039beb93cSSam Leffler 	addr[4] = bssid;
213139beb93cSSam Leffler 	len[4] = ETH_ALEN;
213239beb93cSSam Leffler 	addr[5] = sta_addr;
213339beb93cSSam Leffler 	len[5] = ETH_ALEN;
213439beb93cSSam Leffler 
213585732ac8SCy Schubert 	if (sha256_vector(6, addr, len, hash) < 0)
213685732ac8SCy Schubert 		return -1;
213739beb93cSSam Leffler 	os_memcpy(ptk_name, hash, WPA_PMK_NAME_LEN);
21385b9c547cSRui Paulo 
21395b9c547cSRui Paulo 	os_memcpy(ptk->kck, tmp, ptk->kck_len);
214085732ac8SCy Schubert 	offset = ptk->kck_len;
214185732ac8SCy Schubert 	os_memcpy(ptk->kek, tmp + offset, ptk->kek_len);
214285732ac8SCy Schubert 	offset += ptk->kek_len;
214385732ac8SCy Schubert 	os_memcpy(ptk->tk, tmp + offset, ptk->tk_len);
214485732ac8SCy Schubert 	offset += ptk->tk_len;
214585732ac8SCy Schubert 	os_memcpy(ptk->kck2, tmp + offset, ptk->kck2_len);
21464bc52338SCy Schubert 	offset += ptk->kck2_len;
214785732ac8SCy Schubert 	os_memcpy(ptk->kek2, tmp + offset, ptk->kek2_len);
2148c1d255d3SCy Schubert 	offset += ptk->kek2_len;
2149c1d255d3SCy Schubert 	os_memcpy(ptk->kdk, tmp + offset, ptk->kdk_len);
21505b9c547cSRui Paulo 
21515b9c547cSRui Paulo 	wpa_hexdump_key(MSG_DEBUG, "FT: KCK", ptk->kck, ptk->kck_len);
21525b9c547cSRui Paulo 	wpa_hexdump_key(MSG_DEBUG, "FT: KEK", ptk->kek, ptk->kek_len);
215385732ac8SCy Schubert 	if (ptk->kck2_len)
215485732ac8SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "FT: KCK2",
215585732ac8SCy Schubert 				ptk->kck2, ptk->kck2_len);
215685732ac8SCy Schubert 	if (ptk->kek2_len)
215785732ac8SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "FT: KEK2",
215885732ac8SCy Schubert 				ptk->kek2, ptk->kek2_len);
2159c1d255d3SCy Schubert 	if (ptk->kdk_len)
2160c1d255d3SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "FT: KDK", ptk->kdk, ptk->kdk_len);
2161c1d255d3SCy Schubert 
21625b9c547cSRui Paulo 	wpa_hexdump_key(MSG_DEBUG, "FT: TK", ptk->tk, ptk->tk_len);
21635b9c547cSRui Paulo 	wpa_hexdump(MSG_DEBUG, "FT: PTKName", ptk_name, WPA_PMK_NAME_LEN);
21645b9c547cSRui Paulo 
2165c1d255d3SCy Schubert 	forced_memzero(tmp, sizeof(tmp));
21665b9c547cSRui Paulo 
21675b9c547cSRui Paulo 	return 0;
216839beb93cSSam Leffler }
216939beb93cSSam Leffler 
217039beb93cSSam Leffler #endif /* CONFIG_IEEE80211R */
2171e28a4053SRui Paulo 
2172e28a4053SRui Paulo 
2173e28a4053SRui Paulo /**
2174e28a4053SRui Paulo  * rsn_pmkid - Calculate PMK identifier
2175e28a4053SRui Paulo  * @pmk: Pairwise master key
2176e28a4053SRui Paulo  * @pmk_len: Length of pmk in bytes
2177e28a4053SRui Paulo  * @aa: Authenticator address
2178e28a4053SRui Paulo  * @spa: Supplicant address
2179e28a4053SRui Paulo  * @pmkid: Buffer for PMKID
218085732ac8SCy Schubert  * @akmp: Negotiated key management protocol
2181e28a4053SRui Paulo  *
218285732ac8SCy Schubert  * IEEE Std 802.11-2016 - 12.7.1.3 Pairwise key hierarchy
218385732ac8SCy Schubert  * AKM: 00-0F-AC:5, 00-0F-AC:6, 00-0F-AC:14, 00-0F-AC:16
218485732ac8SCy Schubert  * PMKID = Truncate-128(HMAC-SHA-256(PMK, "PMK Name" || AA || SPA))
218585732ac8SCy Schubert  * AKM: 00-0F-AC:11
218685732ac8SCy Schubert  * See rsn_pmkid_suite_b()
218785732ac8SCy Schubert  * AKM: 00-0F-AC:12
218885732ac8SCy Schubert  * See rsn_pmkid_suite_b_192()
218985732ac8SCy Schubert  * AKM: 00-0F-AC:13, 00-0F-AC:15, 00-0F-AC:17
219085732ac8SCy Schubert  * PMKID = Truncate-128(HMAC-SHA-384(PMK, "PMK Name" || AA || SPA))
219185732ac8SCy Schubert  * Otherwise:
219285732ac8SCy Schubert  * PMKID = Truncate-128(HMAC-SHA-1(PMK, "PMK Name" || AA || SPA))
2193e28a4053SRui Paulo  */
2194e28a4053SRui Paulo void rsn_pmkid(const u8 *pmk, size_t pmk_len, const u8 *aa, const u8 *spa,
219585732ac8SCy Schubert 	       u8 *pmkid, int akmp)
2196e28a4053SRui Paulo {
2197e28a4053SRui Paulo 	char *title = "PMK Name";
2198e28a4053SRui Paulo 	const u8 *addr[3];
2199e28a4053SRui Paulo 	const size_t len[3] = { 8, ETH_ALEN, ETH_ALEN };
220085732ac8SCy Schubert 	unsigned char hash[SHA384_MAC_LEN];
2201e28a4053SRui Paulo 
2202e28a4053SRui Paulo 	addr[0] = (u8 *) title;
2203e28a4053SRui Paulo 	addr[1] = aa;
2204e28a4053SRui Paulo 	addr[2] = spa;
2205e28a4053SRui Paulo 
220685732ac8SCy Schubert 	if (0) {
220785732ac8SCy Schubert #if defined(CONFIG_FILS) || defined(CONFIG_SHA384)
220885732ac8SCy Schubert 	} else if (wpa_key_mgmt_sha384(akmp)) {
220985732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "RSN: Derive PMKID using HMAC-SHA-384");
221085732ac8SCy Schubert 		hmac_sha384_vector(pmk, pmk_len, 3, addr, len, hash);
221185732ac8SCy Schubert #endif /* CONFIG_FILS || CONFIG_SHA384 */
221285732ac8SCy Schubert 	} else if (wpa_key_mgmt_sha256(akmp)) {
221385732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "RSN: Derive PMKID using HMAC-SHA-256");
2214e28a4053SRui Paulo 		hmac_sha256_vector(pmk, pmk_len, 3, addr, len, hash);
221585732ac8SCy Schubert 	} else {
221685732ac8SCy Schubert 		wpa_printf(MSG_DEBUG, "RSN: Derive PMKID using HMAC-SHA-1");
2217e28a4053SRui Paulo 		hmac_sha1_vector(pmk, pmk_len, 3, addr, len, hash);
221885732ac8SCy Schubert 	}
221985732ac8SCy Schubert 	wpa_hexdump(MSG_DEBUG, "RSN: Derived PMKID", hash, PMKID_LEN);
2220e28a4053SRui Paulo 	os_memcpy(pmkid, hash, PMKID_LEN);
2221e28a4053SRui Paulo }
2222e28a4053SRui Paulo 
2223e28a4053SRui Paulo 
22245b9c547cSRui Paulo #ifdef CONFIG_SUITEB
22255b9c547cSRui Paulo /**
22265b9c547cSRui Paulo  * rsn_pmkid_suite_b - Calculate PMK identifier for Suite B AKM
22275b9c547cSRui Paulo  * @kck: Key confirmation key
22285b9c547cSRui Paulo  * @kck_len: Length of kck in bytes
22295b9c547cSRui Paulo  * @aa: Authenticator address
22305b9c547cSRui Paulo  * @spa: Supplicant address
22315b9c547cSRui Paulo  * @pmkid: Buffer for PMKID
22325b9c547cSRui Paulo  * Returns: 0 on success, -1 on failure
22335b9c547cSRui Paulo  *
22345b9c547cSRui Paulo  * IEEE Std 802.11ac-2013 - 11.6.1.3 Pairwise key hierarchy
22355b9c547cSRui Paulo  * PMKID = Truncate(HMAC-SHA-256(KCK, "PMK Name" || AA || SPA))
22365b9c547cSRui Paulo  */
22375b9c547cSRui Paulo int rsn_pmkid_suite_b(const u8 *kck, size_t kck_len, const u8 *aa,
22385b9c547cSRui Paulo 		      const u8 *spa, u8 *pmkid)
22395b9c547cSRui Paulo {
22405b9c547cSRui Paulo 	char *title = "PMK Name";
22415b9c547cSRui Paulo 	const u8 *addr[3];
22425b9c547cSRui Paulo 	const size_t len[3] = { 8, ETH_ALEN, ETH_ALEN };
22435b9c547cSRui Paulo 	unsigned char hash[SHA256_MAC_LEN];
22445b9c547cSRui Paulo 
22455b9c547cSRui Paulo 	addr[0] = (u8 *) title;
22465b9c547cSRui Paulo 	addr[1] = aa;
22475b9c547cSRui Paulo 	addr[2] = spa;
22485b9c547cSRui Paulo 
22495b9c547cSRui Paulo 	if (hmac_sha256_vector(kck, kck_len, 3, addr, len, hash) < 0)
22505b9c547cSRui Paulo 		return -1;
22515b9c547cSRui Paulo 	os_memcpy(pmkid, hash, PMKID_LEN);
22525b9c547cSRui Paulo 	return 0;
22535b9c547cSRui Paulo }
22545b9c547cSRui Paulo #endif /* CONFIG_SUITEB */
22555b9c547cSRui Paulo 
22565b9c547cSRui Paulo 
22575b9c547cSRui Paulo #ifdef CONFIG_SUITEB192
22585b9c547cSRui Paulo /**
22595b9c547cSRui Paulo  * rsn_pmkid_suite_b_192 - Calculate PMK identifier for Suite B AKM
22605b9c547cSRui Paulo  * @kck: Key confirmation key
22615b9c547cSRui Paulo  * @kck_len: Length of kck in bytes
22625b9c547cSRui Paulo  * @aa: Authenticator address
22635b9c547cSRui Paulo  * @spa: Supplicant address
22645b9c547cSRui Paulo  * @pmkid: Buffer for PMKID
22655b9c547cSRui Paulo  * Returns: 0 on success, -1 on failure
22665b9c547cSRui Paulo  *
22675b9c547cSRui Paulo  * IEEE Std 802.11ac-2013 - 11.6.1.3 Pairwise key hierarchy
22685b9c547cSRui Paulo  * PMKID = Truncate(HMAC-SHA-384(KCK, "PMK Name" || AA || SPA))
22695b9c547cSRui Paulo  */
22705b9c547cSRui Paulo int rsn_pmkid_suite_b_192(const u8 *kck, size_t kck_len, const u8 *aa,
22715b9c547cSRui Paulo 			  const u8 *spa, u8 *pmkid)
22725b9c547cSRui Paulo {
22735b9c547cSRui Paulo 	char *title = "PMK Name";
22745b9c547cSRui Paulo 	const u8 *addr[3];
22755b9c547cSRui Paulo 	const size_t len[3] = { 8, ETH_ALEN, ETH_ALEN };
22765b9c547cSRui Paulo 	unsigned char hash[SHA384_MAC_LEN];
22775b9c547cSRui Paulo 
22785b9c547cSRui Paulo 	addr[0] = (u8 *) title;
22795b9c547cSRui Paulo 	addr[1] = aa;
22805b9c547cSRui Paulo 	addr[2] = spa;
22815b9c547cSRui Paulo 
22825b9c547cSRui Paulo 	if (hmac_sha384_vector(kck, kck_len, 3, addr, len, hash) < 0)
22835b9c547cSRui Paulo 		return -1;
22845b9c547cSRui Paulo 	os_memcpy(pmkid, hash, PMKID_LEN);
22855b9c547cSRui Paulo 	return 0;
22865b9c547cSRui Paulo }
22875b9c547cSRui Paulo #endif /* CONFIG_SUITEB192 */
22885b9c547cSRui Paulo 
22895b9c547cSRui Paulo 
2290e28a4053SRui Paulo /**
2291e28a4053SRui Paulo  * wpa_cipher_txt - Convert cipher suite to a text string
2292e28a4053SRui Paulo  * @cipher: Cipher suite (WPA_CIPHER_* enum)
2293e28a4053SRui Paulo  * Returns: Pointer to a text string of the cipher suite name
2294e28a4053SRui Paulo  */
2295e28a4053SRui Paulo const char * wpa_cipher_txt(int cipher)
2296e28a4053SRui Paulo {
2297e28a4053SRui Paulo 	switch (cipher) {
2298e28a4053SRui Paulo 	case WPA_CIPHER_NONE:
2299e28a4053SRui Paulo 		return "NONE";
2300c1d255d3SCy Schubert #ifdef CONFIG_WEP
2301e28a4053SRui Paulo 	case WPA_CIPHER_WEP40:
2302e28a4053SRui Paulo 		return "WEP-40";
2303e28a4053SRui Paulo 	case WPA_CIPHER_WEP104:
2304e28a4053SRui Paulo 		return "WEP-104";
2305c1d255d3SCy Schubert #endif /* CONFIG_WEP */
2306e28a4053SRui Paulo 	case WPA_CIPHER_TKIP:
2307e28a4053SRui Paulo 		return "TKIP";
2308e28a4053SRui Paulo 	case WPA_CIPHER_CCMP:
2309e28a4053SRui Paulo 		return "CCMP";
2310e28a4053SRui Paulo 	case WPA_CIPHER_CCMP | WPA_CIPHER_TKIP:
2311e28a4053SRui Paulo 		return "CCMP+TKIP";
2312f05cddf9SRui Paulo 	case WPA_CIPHER_GCMP:
2313f05cddf9SRui Paulo 		return "GCMP";
23145b9c547cSRui Paulo 	case WPA_CIPHER_GCMP_256:
23155b9c547cSRui Paulo 		return "GCMP-256";
23165b9c547cSRui Paulo 	case WPA_CIPHER_CCMP_256:
23175b9c547cSRui Paulo 		return "CCMP-256";
231885732ac8SCy Schubert 	case WPA_CIPHER_AES_128_CMAC:
231985732ac8SCy Schubert 		return "BIP";
232085732ac8SCy Schubert 	case WPA_CIPHER_BIP_GMAC_128:
232185732ac8SCy Schubert 		return "BIP-GMAC-128";
232285732ac8SCy Schubert 	case WPA_CIPHER_BIP_GMAC_256:
232385732ac8SCy Schubert 		return "BIP-GMAC-256";
232485732ac8SCy Schubert 	case WPA_CIPHER_BIP_CMAC_256:
232585732ac8SCy Schubert 		return "BIP-CMAC-256";
23265b9c547cSRui Paulo 	case WPA_CIPHER_GTK_NOT_USED:
23275b9c547cSRui Paulo 		return "GTK_NOT_USED";
2328e28a4053SRui Paulo 	default:
2329e28a4053SRui Paulo 		return "UNKNOWN";
2330e28a4053SRui Paulo 	}
2331e28a4053SRui Paulo }
2332e28a4053SRui Paulo 
2333e28a4053SRui Paulo 
2334e28a4053SRui Paulo /**
2335e28a4053SRui Paulo  * wpa_key_mgmt_txt - Convert key management suite to a text string
2336e28a4053SRui Paulo  * @key_mgmt: Key management suite (WPA_KEY_MGMT_* enum)
2337e28a4053SRui Paulo  * @proto: WPA/WPA2 version (WPA_PROTO_*)
2338e28a4053SRui Paulo  * Returns: Pointer to a text string of the key management suite name
2339e28a4053SRui Paulo  */
2340e28a4053SRui Paulo const char * wpa_key_mgmt_txt(int key_mgmt, int proto)
2341e28a4053SRui Paulo {
2342e28a4053SRui Paulo 	switch (key_mgmt) {
2343e28a4053SRui Paulo 	case WPA_KEY_MGMT_IEEE8021X:
2344e28a4053SRui Paulo 		if (proto == (WPA_PROTO_RSN | WPA_PROTO_WPA))
2345e28a4053SRui Paulo 			return "WPA2+WPA/IEEE 802.1X/EAP";
2346e28a4053SRui Paulo 		return proto == WPA_PROTO_RSN ?
2347e28a4053SRui Paulo 			"WPA2/IEEE 802.1X/EAP" : "WPA/IEEE 802.1X/EAP";
2348e28a4053SRui Paulo 	case WPA_KEY_MGMT_PSK:
2349e28a4053SRui Paulo 		if (proto == (WPA_PROTO_RSN | WPA_PROTO_WPA))
2350e28a4053SRui Paulo 			return "WPA2-PSK+WPA-PSK";
2351e28a4053SRui Paulo 		return proto == WPA_PROTO_RSN ?
2352e28a4053SRui Paulo 			"WPA2-PSK" : "WPA-PSK";
2353e28a4053SRui Paulo 	case WPA_KEY_MGMT_NONE:
2354e28a4053SRui Paulo 		return "NONE";
2355780fb4a2SCy Schubert 	case WPA_KEY_MGMT_WPA_NONE:
2356780fb4a2SCy Schubert 		return "WPA-NONE";
2357e28a4053SRui Paulo 	case WPA_KEY_MGMT_IEEE8021X_NO_WPA:
2358e28a4053SRui Paulo 		return "IEEE 802.1X (no WPA)";
2359e28a4053SRui Paulo #ifdef CONFIG_IEEE80211R
2360e28a4053SRui Paulo 	case WPA_KEY_MGMT_FT_IEEE8021X:
2361e28a4053SRui Paulo 		return "FT-EAP";
236285732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_IEEE8021X_SHA384:
236385732ac8SCy Schubert 		return "FT-EAP-SHA384";
2364e28a4053SRui Paulo 	case WPA_KEY_MGMT_FT_PSK:
2365e28a4053SRui Paulo 		return "FT-PSK";
2366e28a4053SRui Paulo #endif /* CONFIG_IEEE80211R */
2367e28a4053SRui Paulo 	case WPA_KEY_MGMT_IEEE8021X_SHA256:
2368e28a4053SRui Paulo 		return "WPA2-EAP-SHA256";
2369e28a4053SRui Paulo 	case WPA_KEY_MGMT_PSK_SHA256:
2370e28a4053SRui Paulo 		return "WPA2-PSK-SHA256";
23715b9c547cSRui Paulo 	case WPA_KEY_MGMT_WPS:
23725b9c547cSRui Paulo 		return "WPS";
23735b9c547cSRui Paulo 	case WPA_KEY_MGMT_SAE:
23745b9c547cSRui Paulo 		return "SAE";
23755b9c547cSRui Paulo 	case WPA_KEY_MGMT_FT_SAE:
23765b9c547cSRui Paulo 		return "FT-SAE";
23775b9c547cSRui Paulo 	case WPA_KEY_MGMT_OSEN:
23785b9c547cSRui Paulo 		return "OSEN";
23795b9c547cSRui Paulo 	case WPA_KEY_MGMT_IEEE8021X_SUITE_B:
23805b9c547cSRui Paulo 		return "WPA2-EAP-SUITE-B";
23815b9c547cSRui Paulo 	case WPA_KEY_MGMT_IEEE8021X_SUITE_B_192:
23825b9c547cSRui Paulo 		return "WPA2-EAP-SUITE-B-192";
238385732ac8SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA256:
238485732ac8SCy Schubert 		return "FILS-SHA256";
238585732ac8SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA384:
238685732ac8SCy Schubert 		return "FILS-SHA384";
238785732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA256:
238885732ac8SCy Schubert 		return "FT-FILS-SHA256";
238985732ac8SCy Schubert 	case WPA_KEY_MGMT_FT_FILS_SHA384:
239085732ac8SCy Schubert 		return "FT-FILS-SHA384";
239185732ac8SCy Schubert 	case WPA_KEY_MGMT_OWE:
239285732ac8SCy Schubert 		return "OWE";
239385732ac8SCy Schubert 	case WPA_KEY_MGMT_DPP:
239485732ac8SCy Schubert 		return "DPP";
2395c1d255d3SCy Schubert 	case WPA_KEY_MGMT_PASN:
2396c1d255d3SCy Schubert 		return "PASN";
2397e28a4053SRui Paulo 	default:
2398e28a4053SRui Paulo 		return "UNKNOWN";
2399e28a4053SRui Paulo 	}
2400e28a4053SRui Paulo }
2401e28a4053SRui Paulo 
2402e28a4053SRui Paulo 
24035b9c547cSRui Paulo u32 wpa_akm_to_suite(int akm)
24045b9c547cSRui Paulo {
240585732ac8SCy Schubert 	if (akm & WPA_KEY_MGMT_FT_IEEE8021X_SHA384)
240685732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_FT_802_1X_SHA384;
24075b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_FT_IEEE8021X)
240885732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_FT_802_1X;
24095b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_FT_PSK)
241085732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_FT_PSK;
24115b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_IEEE8021X_SHA256)
241285732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_802_1X_SHA256;
24135b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_IEEE8021X)
241485732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_UNSPEC_802_1X;
24155b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_PSK_SHA256)
241685732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_PSK_SHA256;
24175b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_PSK)
241885732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X;
24195b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_CCKM)
242085732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_CCKM;
24215b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_OSEN)
242285732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_OSEN;
24235b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_IEEE8021X_SUITE_B)
242485732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_802_1X_SUITE_B;
24255b9c547cSRui Paulo 	if (akm & WPA_KEY_MGMT_IEEE8021X_SUITE_B_192)
242685732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_802_1X_SUITE_B_192;
242785732ac8SCy Schubert 	if (akm & WPA_KEY_MGMT_FILS_SHA256)
242885732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_FILS_SHA256;
242985732ac8SCy Schubert 	if (akm & WPA_KEY_MGMT_FILS_SHA384)
243085732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_FILS_SHA384;
243185732ac8SCy Schubert 	if (akm & WPA_KEY_MGMT_FT_FILS_SHA256)
243285732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_FT_FILS_SHA256;
243385732ac8SCy Schubert 	if (akm & WPA_KEY_MGMT_FT_FILS_SHA384)
243485732ac8SCy Schubert 		return RSN_AUTH_KEY_MGMT_FT_FILS_SHA384;
2435206b73d0SCy Schubert 	if (akm & WPA_KEY_MGMT_SAE)
2436206b73d0SCy Schubert 		return RSN_AUTH_KEY_MGMT_SAE;
2437206b73d0SCy Schubert 	if (akm & WPA_KEY_MGMT_FT_SAE)
2438206b73d0SCy Schubert 		return RSN_AUTH_KEY_MGMT_FT_SAE;
2439206b73d0SCy Schubert 	if (akm & WPA_KEY_MGMT_OWE)
2440206b73d0SCy Schubert 		return RSN_AUTH_KEY_MGMT_OWE;
2441206b73d0SCy Schubert 	if (akm & WPA_KEY_MGMT_DPP)
2442206b73d0SCy Schubert 		return RSN_AUTH_KEY_MGMT_DPP;
24435b9c547cSRui Paulo 	return 0;
24445b9c547cSRui Paulo }
24455b9c547cSRui Paulo 
24465b9c547cSRui Paulo 
2447e28a4053SRui Paulo int wpa_compare_rsn_ie(int ft_initial_assoc,
2448e28a4053SRui Paulo 		       const u8 *ie1, size_t ie1len,
2449e28a4053SRui Paulo 		       const u8 *ie2, size_t ie2len)
2450e28a4053SRui Paulo {
2451e28a4053SRui Paulo 	if (ie1 == NULL || ie2 == NULL)
2452e28a4053SRui Paulo 		return -1;
2453e28a4053SRui Paulo 
2454e28a4053SRui Paulo 	if (ie1len == ie2len && os_memcmp(ie1, ie2, ie1len) == 0)
2455e28a4053SRui Paulo 		return 0; /* identical IEs */
2456e28a4053SRui Paulo 
2457e28a4053SRui Paulo #ifdef CONFIG_IEEE80211R
2458e28a4053SRui Paulo 	if (ft_initial_assoc) {
2459e28a4053SRui Paulo 		struct wpa_ie_data ie1d, ie2d;
2460e28a4053SRui Paulo 		/*
2461e28a4053SRui Paulo 		 * The PMKID-List in RSN IE is different between Beacon/Probe
2462e28a4053SRui Paulo 		 * Response/(Re)Association Request frames and EAPOL-Key
2463e28a4053SRui Paulo 		 * messages in FT initial mobility domain association. Allow
2464e28a4053SRui Paulo 		 * for this, but verify that other parts of the RSN IEs are
2465e28a4053SRui Paulo 		 * identical.
2466e28a4053SRui Paulo 		 */
2467e28a4053SRui Paulo 		if (wpa_parse_wpa_ie_rsn(ie1, ie1len, &ie1d) < 0 ||
2468e28a4053SRui Paulo 		    wpa_parse_wpa_ie_rsn(ie2, ie2len, &ie2d) < 0)
2469e28a4053SRui Paulo 			return -1;
2470e28a4053SRui Paulo 		if (ie1d.proto == ie2d.proto &&
2471e28a4053SRui Paulo 		    ie1d.pairwise_cipher == ie2d.pairwise_cipher &&
2472e28a4053SRui Paulo 		    ie1d.group_cipher == ie2d.group_cipher &&
2473e28a4053SRui Paulo 		    ie1d.key_mgmt == ie2d.key_mgmt &&
2474e28a4053SRui Paulo 		    ie1d.capabilities == ie2d.capabilities &&
2475e28a4053SRui Paulo 		    ie1d.mgmt_group_cipher == ie2d.mgmt_group_cipher)
2476e28a4053SRui Paulo 			return 0;
2477e28a4053SRui Paulo 	}
2478e28a4053SRui Paulo #endif /* CONFIG_IEEE80211R */
2479e28a4053SRui Paulo 
2480e28a4053SRui Paulo 	return -1;
2481e28a4053SRui Paulo }
2482e28a4053SRui Paulo 
2483e28a4053SRui Paulo 
2484780fb4a2SCy Schubert int wpa_insert_pmkid(u8 *ies, size_t *ies_len, const u8 *pmkid)
2485e28a4053SRui Paulo {
2486e28a4053SRui Paulo 	u8 *start, *end, *rpos, *rend;
2487e28a4053SRui Paulo 	int added = 0;
2488e28a4053SRui Paulo 
2489e28a4053SRui Paulo 	start = ies;
2490780fb4a2SCy Schubert 	end = ies + *ies_len;
2491e28a4053SRui Paulo 
2492e28a4053SRui Paulo 	while (start < end) {
2493e28a4053SRui Paulo 		if (*start == WLAN_EID_RSN)
2494e28a4053SRui Paulo 			break;
2495e28a4053SRui Paulo 		start += 2 + start[1];
2496e28a4053SRui Paulo 	}
2497e28a4053SRui Paulo 	if (start >= end) {
2498c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR, "RSN: Could not find RSNE in IEs data");
2499e28a4053SRui Paulo 		return -1;
2500e28a4053SRui Paulo 	}
2501c1d255d3SCy Schubert 	wpa_hexdump(MSG_DEBUG, "RSN: RSNE before modification",
2502e28a4053SRui Paulo 		    start, 2 + start[1]);
2503e28a4053SRui Paulo 
2504e28a4053SRui Paulo 	/* Find start of PMKID-Count */
2505e28a4053SRui Paulo 	rpos = start + 2;
2506e28a4053SRui Paulo 	rend = rpos + start[1];
2507e28a4053SRui Paulo 
2508e28a4053SRui Paulo 	/* Skip Version and Group Data Cipher Suite */
2509e28a4053SRui Paulo 	rpos += 2 + 4;
2510e28a4053SRui Paulo 	/* Skip Pairwise Cipher Suite Count and List */
2511e28a4053SRui Paulo 	rpos += 2 + WPA_GET_LE16(rpos) * RSN_SELECTOR_LEN;
2512e28a4053SRui Paulo 	/* Skip AKM Suite Count and List */
2513e28a4053SRui Paulo 	rpos += 2 + WPA_GET_LE16(rpos) * RSN_SELECTOR_LEN;
2514e28a4053SRui Paulo 
2515e28a4053SRui Paulo 	if (rpos == rend) {
2516e28a4053SRui Paulo 		/* Add RSN Capabilities */
2517e28a4053SRui Paulo 		os_memmove(rpos + 2, rpos, end - rpos);
2518e28a4053SRui Paulo 		*rpos++ = 0;
2519e28a4053SRui Paulo 		*rpos++ = 0;
2520325151a3SRui Paulo 		added += 2;
2521325151a3SRui Paulo 		start[1] += 2;
2522325151a3SRui Paulo 		rend = rpos;
2523e28a4053SRui Paulo 	} else {
2524e28a4053SRui Paulo 		/* Skip RSN Capabilities */
2525e28a4053SRui Paulo 		rpos += 2;
2526e28a4053SRui Paulo 		if (rpos > rend) {
2527c1d255d3SCy Schubert 			wpa_printf(MSG_ERROR,
2528c1d255d3SCy Schubert 				   "RSN: Could not parse RSNE in IEs data");
2529e28a4053SRui Paulo 			return -1;
2530e28a4053SRui Paulo 		}
2531e28a4053SRui Paulo 	}
2532e28a4053SRui Paulo 
2533e28a4053SRui Paulo 	if (rpos == rend) {
2534e28a4053SRui Paulo 		/* No PMKID-Count field included; add it */
2535325151a3SRui Paulo 		os_memmove(rpos + 2 + PMKID_LEN, rpos, end + added - rpos);
2536e28a4053SRui Paulo 		WPA_PUT_LE16(rpos, 1);
2537e28a4053SRui Paulo 		rpos += 2;
2538e28a4053SRui Paulo 		os_memcpy(rpos, pmkid, PMKID_LEN);
2539e28a4053SRui Paulo 		added += 2 + PMKID_LEN;
2540e28a4053SRui Paulo 		start[1] += 2 + PMKID_LEN;
2541e28a4053SRui Paulo 	} else {
2542780fb4a2SCy Schubert 		u16 num_pmkid;
2543780fb4a2SCy Schubert 
2544780fb4a2SCy Schubert 		if (rend - rpos < 2)
2545e28a4053SRui Paulo 			return -1;
2546780fb4a2SCy Schubert 		num_pmkid = WPA_GET_LE16(rpos);
2547780fb4a2SCy Schubert 		/* PMKID-Count was included; use it */
2548780fb4a2SCy Schubert 		if (num_pmkid != 0) {
2549780fb4a2SCy Schubert 			u8 *after;
2550780fb4a2SCy Schubert 
2551780fb4a2SCy Schubert 			if (num_pmkid * PMKID_LEN > rend - rpos - 2)
2552780fb4a2SCy Schubert 				return -1;
2553780fb4a2SCy Schubert 			/*
2554780fb4a2SCy Schubert 			 * PMKID may have been included in RSN IE in
2555780fb4a2SCy Schubert 			 * (Re)Association Request frame, so remove the old
2556780fb4a2SCy Schubert 			 * PMKID(s) first before adding the new one.
2557780fb4a2SCy Schubert 			 */
2558780fb4a2SCy Schubert 			wpa_printf(MSG_DEBUG,
2559c1d255d3SCy Schubert 				   "RSN: Remove %u old PMKID(s) from RSNE",
2560780fb4a2SCy Schubert 				   num_pmkid);
2561780fb4a2SCy Schubert 			after = rpos + 2 + num_pmkid * PMKID_LEN;
2562c1d255d3SCy Schubert 			os_memmove(rpos + 2, after, end - after);
2563780fb4a2SCy Schubert 			start[1] -= num_pmkid * PMKID_LEN;
2564780fb4a2SCy Schubert 			added -= num_pmkid * PMKID_LEN;
2565e28a4053SRui Paulo 		}
2566e28a4053SRui Paulo 		WPA_PUT_LE16(rpos, 1);
2567e28a4053SRui Paulo 		rpos += 2;
2568325151a3SRui Paulo 		os_memmove(rpos + PMKID_LEN, rpos, end + added - rpos);
2569e28a4053SRui Paulo 		os_memcpy(rpos, pmkid, PMKID_LEN);
2570e28a4053SRui Paulo 		added += PMKID_LEN;
2571e28a4053SRui Paulo 		start[1] += PMKID_LEN;
2572e28a4053SRui Paulo 	}
2573e28a4053SRui Paulo 
2574c1d255d3SCy Schubert 	wpa_hexdump(MSG_DEBUG, "RSN: RSNE after modification (PMKID inserted)",
2575c1d255d3SCy Schubert 		    start, 2 + start[1]);
2576e28a4053SRui Paulo 
2577780fb4a2SCy Schubert 	*ies_len += added;
2578780fb4a2SCy Schubert 
2579780fb4a2SCy Schubert 	return 0;
2580e28a4053SRui Paulo }
2581f05cddf9SRui Paulo 
2582f05cddf9SRui Paulo 
2583f05cddf9SRui Paulo int wpa_cipher_key_len(int cipher)
2584f05cddf9SRui Paulo {
2585f05cddf9SRui Paulo 	switch (cipher) {
25865b9c547cSRui Paulo 	case WPA_CIPHER_CCMP_256:
25875b9c547cSRui Paulo 	case WPA_CIPHER_GCMP_256:
25885b9c547cSRui Paulo 	case WPA_CIPHER_BIP_GMAC_256:
25895b9c547cSRui Paulo 	case WPA_CIPHER_BIP_CMAC_256:
25905b9c547cSRui Paulo 		return 32;
2591f05cddf9SRui Paulo 	case WPA_CIPHER_CCMP:
2592f05cddf9SRui Paulo 	case WPA_CIPHER_GCMP:
25935b9c547cSRui Paulo 	case WPA_CIPHER_AES_128_CMAC:
25945b9c547cSRui Paulo 	case WPA_CIPHER_BIP_GMAC_128:
2595f05cddf9SRui Paulo 		return 16;
2596f05cddf9SRui Paulo 	case WPA_CIPHER_TKIP:
2597f05cddf9SRui Paulo 		return 32;
2598f05cddf9SRui Paulo 	}
2599f05cddf9SRui Paulo 
2600f05cddf9SRui Paulo 	return 0;
2601f05cddf9SRui Paulo }
2602f05cddf9SRui Paulo 
2603f05cddf9SRui Paulo 
2604f05cddf9SRui Paulo int wpa_cipher_rsc_len(int cipher)
2605f05cddf9SRui Paulo {
2606f05cddf9SRui Paulo 	switch (cipher) {
26075b9c547cSRui Paulo 	case WPA_CIPHER_CCMP_256:
26085b9c547cSRui Paulo 	case WPA_CIPHER_GCMP_256:
2609f05cddf9SRui Paulo 	case WPA_CIPHER_CCMP:
2610f05cddf9SRui Paulo 	case WPA_CIPHER_GCMP:
2611f05cddf9SRui Paulo 	case WPA_CIPHER_TKIP:
2612f05cddf9SRui Paulo 		return 6;
2613f05cddf9SRui Paulo 	}
2614f05cddf9SRui Paulo 
2615f05cddf9SRui Paulo 	return 0;
2616f05cddf9SRui Paulo }
2617f05cddf9SRui Paulo 
2618f05cddf9SRui Paulo 
261985732ac8SCy Schubert enum wpa_alg wpa_cipher_to_alg(int cipher)
2620f05cddf9SRui Paulo {
2621f05cddf9SRui Paulo 	switch (cipher) {
26225b9c547cSRui Paulo 	case WPA_CIPHER_CCMP_256:
26235b9c547cSRui Paulo 		return WPA_ALG_CCMP_256;
26245b9c547cSRui Paulo 	case WPA_CIPHER_GCMP_256:
26255b9c547cSRui Paulo 		return WPA_ALG_GCMP_256;
2626f05cddf9SRui Paulo 	case WPA_CIPHER_CCMP:
2627f05cddf9SRui Paulo 		return WPA_ALG_CCMP;
2628f05cddf9SRui Paulo 	case WPA_CIPHER_GCMP:
2629f05cddf9SRui Paulo 		return WPA_ALG_GCMP;
2630f05cddf9SRui Paulo 	case WPA_CIPHER_TKIP:
2631f05cddf9SRui Paulo 		return WPA_ALG_TKIP;
26325b9c547cSRui Paulo 	case WPA_CIPHER_AES_128_CMAC:
2633c1d255d3SCy Schubert 		return WPA_ALG_BIP_CMAC_128;
26345b9c547cSRui Paulo 	case WPA_CIPHER_BIP_GMAC_128:
26355b9c547cSRui Paulo 		return WPA_ALG_BIP_GMAC_128;
26365b9c547cSRui Paulo 	case WPA_CIPHER_BIP_GMAC_256:
26375b9c547cSRui Paulo 		return WPA_ALG_BIP_GMAC_256;
26385b9c547cSRui Paulo 	case WPA_CIPHER_BIP_CMAC_256:
26395b9c547cSRui Paulo 		return WPA_ALG_BIP_CMAC_256;
2640f05cddf9SRui Paulo 	}
2641f05cddf9SRui Paulo 	return WPA_ALG_NONE;
2642f05cddf9SRui Paulo }
2643f05cddf9SRui Paulo 
2644f05cddf9SRui Paulo 
2645f05cddf9SRui Paulo int wpa_cipher_valid_pairwise(int cipher)
2646f05cddf9SRui Paulo {
2647c1d255d3SCy Schubert #ifdef CONFIG_NO_TKIP
2648c1d255d3SCy Schubert 	return cipher == WPA_CIPHER_CCMP_256 ||
2649c1d255d3SCy Schubert 		cipher == WPA_CIPHER_GCMP_256 ||
2650c1d255d3SCy Schubert 		cipher == WPA_CIPHER_CCMP ||
2651c1d255d3SCy Schubert 		cipher == WPA_CIPHER_GCMP;
2652c1d255d3SCy Schubert #else /* CONFIG_NO_TKIP */
26535b9c547cSRui Paulo 	return cipher == WPA_CIPHER_CCMP_256 ||
26545b9c547cSRui Paulo 		cipher == WPA_CIPHER_GCMP_256 ||
26555b9c547cSRui Paulo 		cipher == WPA_CIPHER_CCMP ||
2656f05cddf9SRui Paulo 		cipher == WPA_CIPHER_GCMP ||
2657f05cddf9SRui Paulo 		cipher == WPA_CIPHER_TKIP;
2658c1d255d3SCy Schubert #endif /* CONFIG_NO_TKIP */
2659f05cddf9SRui Paulo }
2660f05cddf9SRui Paulo 
2661f05cddf9SRui Paulo 
2662f05cddf9SRui Paulo u32 wpa_cipher_to_suite(int proto, int cipher)
2663f05cddf9SRui Paulo {
26645b9c547cSRui Paulo 	if (cipher & WPA_CIPHER_CCMP_256)
26655b9c547cSRui Paulo 		return RSN_CIPHER_SUITE_CCMP_256;
26665b9c547cSRui Paulo 	if (cipher & WPA_CIPHER_GCMP_256)
26675b9c547cSRui Paulo 		return RSN_CIPHER_SUITE_GCMP_256;
2668f05cddf9SRui Paulo 	if (cipher & WPA_CIPHER_CCMP)
2669f05cddf9SRui Paulo 		return (proto == WPA_PROTO_RSN ?
2670f05cddf9SRui Paulo 			RSN_CIPHER_SUITE_CCMP : WPA_CIPHER_SUITE_CCMP);
2671f05cddf9SRui Paulo 	if (cipher & WPA_CIPHER_GCMP)
2672f05cddf9SRui Paulo 		return RSN_CIPHER_SUITE_GCMP;
2673f05cddf9SRui Paulo 	if (cipher & WPA_CIPHER_TKIP)
2674f05cddf9SRui Paulo 		return (proto == WPA_PROTO_RSN ?
2675f05cddf9SRui Paulo 			RSN_CIPHER_SUITE_TKIP : WPA_CIPHER_SUITE_TKIP);
2676f05cddf9SRui Paulo 	if (cipher & WPA_CIPHER_NONE)
2677f05cddf9SRui Paulo 		return (proto == WPA_PROTO_RSN ?
2678f05cddf9SRui Paulo 			RSN_CIPHER_SUITE_NONE : WPA_CIPHER_SUITE_NONE);
26795b9c547cSRui Paulo 	if (cipher & WPA_CIPHER_GTK_NOT_USED)
26805b9c547cSRui Paulo 		return RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED;
26815b9c547cSRui Paulo 	if (cipher & WPA_CIPHER_AES_128_CMAC)
26825b9c547cSRui Paulo 		return RSN_CIPHER_SUITE_AES_128_CMAC;
26835b9c547cSRui Paulo 	if (cipher & WPA_CIPHER_BIP_GMAC_128)
26845b9c547cSRui Paulo 		return RSN_CIPHER_SUITE_BIP_GMAC_128;
26855b9c547cSRui Paulo 	if (cipher & WPA_CIPHER_BIP_GMAC_256)
26865b9c547cSRui Paulo 		return RSN_CIPHER_SUITE_BIP_GMAC_256;
26875b9c547cSRui Paulo 	if (cipher & WPA_CIPHER_BIP_CMAC_256)
26885b9c547cSRui Paulo 		return RSN_CIPHER_SUITE_BIP_CMAC_256;
2689f05cddf9SRui Paulo 	return 0;
2690f05cddf9SRui Paulo }
2691f05cddf9SRui Paulo 
2692f05cddf9SRui Paulo 
26935b9c547cSRui Paulo int rsn_cipher_put_suites(u8 *start, int ciphers)
2694f05cddf9SRui Paulo {
26955b9c547cSRui Paulo 	u8 *pos = start;
2696f05cddf9SRui Paulo 
26975b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_CCMP_256) {
26985b9c547cSRui Paulo 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_CCMP_256);
26995b9c547cSRui Paulo 		pos += RSN_SELECTOR_LEN;
27005b9c547cSRui Paulo 	}
27015b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_GCMP_256) {
27025b9c547cSRui Paulo 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_GCMP_256);
27035b9c547cSRui Paulo 		pos += RSN_SELECTOR_LEN;
27045b9c547cSRui Paulo 	}
2705f05cddf9SRui Paulo 	if (ciphers & WPA_CIPHER_CCMP) {
2706f05cddf9SRui Paulo 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_CCMP);
2707f05cddf9SRui Paulo 		pos += RSN_SELECTOR_LEN;
2708f05cddf9SRui Paulo 	}
2709f05cddf9SRui Paulo 	if (ciphers & WPA_CIPHER_GCMP) {
2710f05cddf9SRui Paulo 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_GCMP);
2711f05cddf9SRui Paulo 		pos += RSN_SELECTOR_LEN;
2712f05cddf9SRui Paulo 	}
2713f05cddf9SRui Paulo 	if (ciphers & WPA_CIPHER_TKIP) {
2714f05cddf9SRui Paulo 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_TKIP);
2715f05cddf9SRui Paulo 		pos += RSN_SELECTOR_LEN;
2716f05cddf9SRui Paulo 	}
2717f05cddf9SRui Paulo 	if (ciphers & WPA_CIPHER_NONE) {
2718f05cddf9SRui Paulo 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_NONE);
2719f05cddf9SRui Paulo 		pos += RSN_SELECTOR_LEN;
2720f05cddf9SRui Paulo 	}
2721f05cddf9SRui Paulo 
27225b9c547cSRui Paulo 	return (pos - start) / RSN_SELECTOR_LEN;
2723f05cddf9SRui Paulo }
2724f05cddf9SRui Paulo 
2725f05cddf9SRui Paulo 
27265b9c547cSRui Paulo int wpa_cipher_put_suites(u8 *start, int ciphers)
2727f05cddf9SRui Paulo {
27285b9c547cSRui Paulo 	u8 *pos = start;
2729f05cddf9SRui Paulo 
2730f05cddf9SRui Paulo 	if (ciphers & WPA_CIPHER_CCMP) {
2731f05cddf9SRui Paulo 		RSN_SELECTOR_PUT(pos, WPA_CIPHER_SUITE_CCMP);
2732f05cddf9SRui Paulo 		pos += WPA_SELECTOR_LEN;
2733f05cddf9SRui Paulo 	}
2734f05cddf9SRui Paulo 	if (ciphers & WPA_CIPHER_TKIP) {
2735f05cddf9SRui Paulo 		RSN_SELECTOR_PUT(pos, WPA_CIPHER_SUITE_TKIP);
2736f05cddf9SRui Paulo 		pos += WPA_SELECTOR_LEN;
2737f05cddf9SRui Paulo 	}
2738f05cddf9SRui Paulo 	if (ciphers & WPA_CIPHER_NONE) {
2739f05cddf9SRui Paulo 		RSN_SELECTOR_PUT(pos, WPA_CIPHER_SUITE_NONE);
2740f05cddf9SRui Paulo 		pos += WPA_SELECTOR_LEN;
2741f05cddf9SRui Paulo 	}
2742f05cddf9SRui Paulo 
27435b9c547cSRui Paulo 	return (pos - start) / RSN_SELECTOR_LEN;
27445b9c547cSRui Paulo }
27455b9c547cSRui Paulo 
27465b9c547cSRui Paulo 
27475b9c547cSRui Paulo int wpa_pick_pairwise_cipher(int ciphers, int none_allowed)
27485b9c547cSRui Paulo {
27495b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_CCMP_256)
27505b9c547cSRui Paulo 		return WPA_CIPHER_CCMP_256;
27515b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_GCMP_256)
27525b9c547cSRui Paulo 		return WPA_CIPHER_GCMP_256;
27535b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_CCMP)
27545b9c547cSRui Paulo 		return WPA_CIPHER_CCMP;
27555b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_GCMP)
27565b9c547cSRui Paulo 		return WPA_CIPHER_GCMP;
27575b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_TKIP)
27585b9c547cSRui Paulo 		return WPA_CIPHER_TKIP;
27595b9c547cSRui Paulo 	if (none_allowed && (ciphers & WPA_CIPHER_NONE))
27605b9c547cSRui Paulo 		return WPA_CIPHER_NONE;
27615b9c547cSRui Paulo 	return -1;
27625b9c547cSRui Paulo }
27635b9c547cSRui Paulo 
27645b9c547cSRui Paulo 
27655b9c547cSRui Paulo int wpa_pick_group_cipher(int ciphers)
27665b9c547cSRui Paulo {
27675b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_CCMP_256)
27685b9c547cSRui Paulo 		return WPA_CIPHER_CCMP_256;
27695b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_GCMP_256)
27705b9c547cSRui Paulo 		return WPA_CIPHER_GCMP_256;
27715b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_CCMP)
27725b9c547cSRui Paulo 		return WPA_CIPHER_CCMP;
27735b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_GCMP)
27745b9c547cSRui Paulo 		return WPA_CIPHER_GCMP;
27755b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_GTK_NOT_USED)
27765b9c547cSRui Paulo 		return WPA_CIPHER_GTK_NOT_USED;
27775b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_TKIP)
27785b9c547cSRui Paulo 		return WPA_CIPHER_TKIP;
27795b9c547cSRui Paulo 	return -1;
27805b9c547cSRui Paulo }
27815b9c547cSRui Paulo 
27825b9c547cSRui Paulo 
27835b9c547cSRui Paulo int wpa_parse_cipher(const char *value)
27845b9c547cSRui Paulo {
27855b9c547cSRui Paulo 	int val = 0, last;
27865b9c547cSRui Paulo 	char *start, *end, *buf;
27875b9c547cSRui Paulo 
27885b9c547cSRui Paulo 	buf = os_strdup(value);
27895b9c547cSRui Paulo 	if (buf == NULL)
27905b9c547cSRui Paulo 		return -1;
27915b9c547cSRui Paulo 	start = buf;
27925b9c547cSRui Paulo 
27935b9c547cSRui Paulo 	while (*start != '\0') {
27945b9c547cSRui Paulo 		while (*start == ' ' || *start == '\t')
27955b9c547cSRui Paulo 			start++;
27965b9c547cSRui Paulo 		if (*start == '\0')
27975b9c547cSRui Paulo 			break;
27985b9c547cSRui Paulo 		end = start;
27995b9c547cSRui Paulo 		while (*end != ' ' && *end != '\t' && *end != '\0')
28005b9c547cSRui Paulo 			end++;
28015b9c547cSRui Paulo 		last = *end == '\0';
28025b9c547cSRui Paulo 		*end = '\0';
28035b9c547cSRui Paulo 		if (os_strcmp(start, "CCMP-256") == 0)
28045b9c547cSRui Paulo 			val |= WPA_CIPHER_CCMP_256;
28055b9c547cSRui Paulo 		else if (os_strcmp(start, "GCMP-256") == 0)
28065b9c547cSRui Paulo 			val |= WPA_CIPHER_GCMP_256;
28075b9c547cSRui Paulo 		else if (os_strcmp(start, "CCMP") == 0)
28085b9c547cSRui Paulo 			val |= WPA_CIPHER_CCMP;
28095b9c547cSRui Paulo 		else if (os_strcmp(start, "GCMP") == 0)
28105b9c547cSRui Paulo 			val |= WPA_CIPHER_GCMP;
2811c1d255d3SCy Schubert #ifndef CONFIG_NO_TKIP
28125b9c547cSRui Paulo 		else if (os_strcmp(start, "TKIP") == 0)
28135b9c547cSRui Paulo 			val |= WPA_CIPHER_TKIP;
2814c1d255d3SCy Schubert #endif /* CONFIG_NO_TKIP */
2815c1d255d3SCy Schubert #ifdef CONFIG_WEP
28165b9c547cSRui Paulo 		else if (os_strcmp(start, "WEP104") == 0)
28175b9c547cSRui Paulo 			val |= WPA_CIPHER_WEP104;
28185b9c547cSRui Paulo 		else if (os_strcmp(start, "WEP40") == 0)
28195b9c547cSRui Paulo 			val |= WPA_CIPHER_WEP40;
2820c1d255d3SCy Schubert #endif /* CONFIG_WEP */
28215b9c547cSRui Paulo 		else if (os_strcmp(start, "NONE") == 0)
28225b9c547cSRui Paulo 			val |= WPA_CIPHER_NONE;
28235b9c547cSRui Paulo 		else if (os_strcmp(start, "GTK_NOT_USED") == 0)
28245b9c547cSRui Paulo 			val |= WPA_CIPHER_GTK_NOT_USED;
282585732ac8SCy Schubert 		else if (os_strcmp(start, "AES-128-CMAC") == 0)
282685732ac8SCy Schubert 			val |= WPA_CIPHER_AES_128_CMAC;
282785732ac8SCy Schubert 		else if (os_strcmp(start, "BIP-GMAC-128") == 0)
282885732ac8SCy Schubert 			val |= WPA_CIPHER_BIP_GMAC_128;
282985732ac8SCy Schubert 		else if (os_strcmp(start, "BIP-GMAC-256") == 0)
283085732ac8SCy Schubert 			val |= WPA_CIPHER_BIP_GMAC_256;
283185732ac8SCy Schubert 		else if (os_strcmp(start, "BIP-CMAC-256") == 0)
283285732ac8SCy Schubert 			val |= WPA_CIPHER_BIP_CMAC_256;
28335b9c547cSRui Paulo 		else {
28345b9c547cSRui Paulo 			os_free(buf);
28355b9c547cSRui Paulo 			return -1;
28365b9c547cSRui Paulo 		}
28375b9c547cSRui Paulo 
28385b9c547cSRui Paulo 		if (last)
28395b9c547cSRui Paulo 			break;
28405b9c547cSRui Paulo 		start = end + 1;
28415b9c547cSRui Paulo 	}
28425b9c547cSRui Paulo 	os_free(buf);
28435b9c547cSRui Paulo 
28445b9c547cSRui Paulo 	return val;
28455b9c547cSRui Paulo }
28465b9c547cSRui Paulo 
28475b9c547cSRui Paulo 
28485b9c547cSRui Paulo int wpa_write_ciphers(char *start, char *end, int ciphers, const char *delim)
28495b9c547cSRui Paulo {
28505b9c547cSRui Paulo 	char *pos = start;
28515b9c547cSRui Paulo 	int ret;
28525b9c547cSRui Paulo 
28535b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_CCMP_256) {
28545b9c547cSRui Paulo 		ret = os_snprintf(pos, end - pos, "%sCCMP-256",
28555b9c547cSRui Paulo 				  pos == start ? "" : delim);
28565b9c547cSRui Paulo 		if (os_snprintf_error(end - pos, ret))
28575b9c547cSRui Paulo 			return -1;
28585b9c547cSRui Paulo 		pos += ret;
28595b9c547cSRui Paulo 	}
28605b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_GCMP_256) {
28615b9c547cSRui Paulo 		ret = os_snprintf(pos, end - pos, "%sGCMP-256",
28625b9c547cSRui Paulo 				  pos == start ? "" : delim);
28635b9c547cSRui Paulo 		if (os_snprintf_error(end - pos, ret))
28645b9c547cSRui Paulo 			return -1;
28655b9c547cSRui Paulo 		pos += ret;
28665b9c547cSRui Paulo 	}
28675b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_CCMP) {
28685b9c547cSRui Paulo 		ret = os_snprintf(pos, end - pos, "%sCCMP",
28695b9c547cSRui Paulo 				  pos == start ? "" : delim);
28705b9c547cSRui Paulo 		if (os_snprintf_error(end - pos, ret))
28715b9c547cSRui Paulo 			return -1;
28725b9c547cSRui Paulo 		pos += ret;
28735b9c547cSRui Paulo 	}
28745b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_GCMP) {
28755b9c547cSRui Paulo 		ret = os_snprintf(pos, end - pos, "%sGCMP",
28765b9c547cSRui Paulo 				  pos == start ? "" : delim);
28775b9c547cSRui Paulo 		if (os_snprintf_error(end - pos, ret))
28785b9c547cSRui Paulo 			return -1;
28795b9c547cSRui Paulo 		pos += ret;
28805b9c547cSRui Paulo 	}
28815b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_TKIP) {
28825b9c547cSRui Paulo 		ret = os_snprintf(pos, end - pos, "%sTKIP",
28835b9c547cSRui Paulo 				  pos == start ? "" : delim);
28845b9c547cSRui Paulo 		if (os_snprintf_error(end - pos, ret))
28855b9c547cSRui Paulo 			return -1;
28865b9c547cSRui Paulo 		pos += ret;
28875b9c547cSRui Paulo 	}
288885732ac8SCy Schubert 	if (ciphers & WPA_CIPHER_AES_128_CMAC) {
288985732ac8SCy Schubert 		ret = os_snprintf(pos, end - pos, "%sAES-128-CMAC",
289085732ac8SCy Schubert 				  pos == start ? "" : delim);
289185732ac8SCy Schubert 		if (os_snprintf_error(end - pos, ret))
289285732ac8SCy Schubert 			return -1;
289385732ac8SCy Schubert 		pos += ret;
289485732ac8SCy Schubert 	}
289585732ac8SCy Schubert 	if (ciphers & WPA_CIPHER_BIP_GMAC_128) {
289685732ac8SCy Schubert 		ret = os_snprintf(pos, end - pos, "%sBIP-GMAC-128",
289785732ac8SCy Schubert 				  pos == start ? "" : delim);
289885732ac8SCy Schubert 		if (os_snprintf_error(end - pos, ret))
289985732ac8SCy Schubert 			return -1;
290085732ac8SCy Schubert 		pos += ret;
290185732ac8SCy Schubert 	}
290285732ac8SCy Schubert 	if (ciphers & WPA_CIPHER_BIP_GMAC_256) {
290385732ac8SCy Schubert 		ret = os_snprintf(pos, end - pos, "%sBIP-GMAC-256",
290485732ac8SCy Schubert 				  pos == start ? "" : delim);
290585732ac8SCy Schubert 		if (os_snprintf_error(end - pos, ret))
290685732ac8SCy Schubert 			return -1;
290785732ac8SCy Schubert 		pos += ret;
290885732ac8SCy Schubert 	}
290985732ac8SCy Schubert 	if (ciphers & WPA_CIPHER_BIP_CMAC_256) {
291085732ac8SCy Schubert 		ret = os_snprintf(pos, end - pos, "%sBIP-CMAC-256",
291185732ac8SCy Schubert 				  pos == start ? "" : delim);
291285732ac8SCy Schubert 		if (os_snprintf_error(end - pos, ret))
291385732ac8SCy Schubert 			return -1;
291485732ac8SCy Schubert 		pos += ret;
291585732ac8SCy Schubert 	}
29165b9c547cSRui Paulo 	if (ciphers & WPA_CIPHER_NONE) {
29175b9c547cSRui Paulo 		ret = os_snprintf(pos, end - pos, "%sNONE",
29185b9c547cSRui Paulo 				  pos == start ? "" : delim);
29195b9c547cSRui Paulo 		if (os_snprintf_error(end - pos, ret))
29205b9c547cSRui Paulo 			return -1;
29215b9c547cSRui Paulo 		pos += ret;
29225b9c547cSRui Paulo 	}
29235b9c547cSRui Paulo 
29245b9c547cSRui Paulo 	return pos - start;
29255b9c547cSRui Paulo }
29265b9c547cSRui Paulo 
29275b9c547cSRui Paulo 
29285b9c547cSRui Paulo int wpa_select_ap_group_cipher(int wpa, int wpa_pairwise, int rsn_pairwise)
29295b9c547cSRui Paulo {
29305b9c547cSRui Paulo 	int pairwise = 0;
29315b9c547cSRui Paulo 
29325b9c547cSRui Paulo 	/* Select group cipher based on the enabled pairwise cipher suites */
29335b9c547cSRui Paulo 	if (wpa & 1)
29345b9c547cSRui Paulo 		pairwise |= wpa_pairwise;
29355b9c547cSRui Paulo 	if (wpa & 2)
29365b9c547cSRui Paulo 		pairwise |= rsn_pairwise;
29375b9c547cSRui Paulo 
29385b9c547cSRui Paulo 	if (pairwise & WPA_CIPHER_TKIP)
29395b9c547cSRui Paulo 		return WPA_CIPHER_TKIP;
29405b9c547cSRui Paulo 	if ((pairwise & (WPA_CIPHER_CCMP | WPA_CIPHER_GCMP)) == WPA_CIPHER_GCMP)
29415b9c547cSRui Paulo 		return WPA_CIPHER_GCMP;
29425b9c547cSRui Paulo 	if ((pairwise & (WPA_CIPHER_GCMP_256 | WPA_CIPHER_CCMP |
29435b9c547cSRui Paulo 			 WPA_CIPHER_GCMP)) == WPA_CIPHER_GCMP_256)
29445b9c547cSRui Paulo 		return WPA_CIPHER_GCMP_256;
29455b9c547cSRui Paulo 	if ((pairwise & (WPA_CIPHER_CCMP_256 | WPA_CIPHER_CCMP |
29465b9c547cSRui Paulo 			 WPA_CIPHER_GCMP)) == WPA_CIPHER_CCMP_256)
29475b9c547cSRui Paulo 		return WPA_CIPHER_CCMP_256;
29485b9c547cSRui Paulo 	return WPA_CIPHER_CCMP;
2949f05cddf9SRui Paulo }
295085732ac8SCy Schubert 
295185732ac8SCy Schubert 
295285732ac8SCy Schubert #ifdef CONFIG_FILS
295385732ac8SCy Schubert int fils_domain_name_hash(const char *domain, u8 *hash)
295485732ac8SCy Schubert {
295585732ac8SCy Schubert 	char buf[255], *wpos = buf;
295685732ac8SCy Schubert 	const char *pos = domain;
295785732ac8SCy Schubert 	size_t len;
295885732ac8SCy Schubert 	const u8 *addr[1];
295985732ac8SCy Schubert 	u8 mac[SHA256_MAC_LEN];
296085732ac8SCy Schubert 
296185732ac8SCy Schubert 	for (len = 0; len < sizeof(buf) && *pos; len++) {
296285732ac8SCy Schubert 		if (isalpha(*pos) && isupper(*pos))
296385732ac8SCy Schubert 			*wpos++ = tolower(*pos);
296485732ac8SCy Schubert 		else
296585732ac8SCy Schubert 			*wpos++ = *pos;
296685732ac8SCy Schubert 		pos++;
296785732ac8SCy Schubert 	}
296885732ac8SCy Schubert 
296985732ac8SCy Schubert 	addr[0] = (const u8 *) buf;
297085732ac8SCy Schubert 	if (sha256_vector(1, addr, &len, mac) < 0)
297185732ac8SCy Schubert 		return -1;
297285732ac8SCy Schubert 	os_memcpy(hash, mac, 2);
297385732ac8SCy Schubert 	return 0;
297485732ac8SCy Schubert }
297585732ac8SCy Schubert #endif /* CONFIG_FILS */
2976c1d255d3SCy Schubert 
2977c1d255d3SCy Schubert 
2978c1d255d3SCy Schubert /**
2979c1d255d3SCy Schubert  * wpa_parse_vendor_specific - Parse Vendor Specific IEs
2980c1d255d3SCy Schubert  * @pos: Pointer to the IE header
2981c1d255d3SCy Schubert  * @end: Pointer to the end of the Key Data buffer
2982c1d255d3SCy Schubert  * @ie: Pointer to parsed IE data
2983c1d255d3SCy Schubert  */
2984c1d255d3SCy Schubert static void wpa_parse_vendor_specific(const u8 *pos, const u8 *end,
2985c1d255d3SCy Schubert 				      struct wpa_eapol_ie_parse *ie)
2986c1d255d3SCy Schubert {
2987c1d255d3SCy Schubert 	unsigned int oui;
2988c1d255d3SCy Schubert 
2989c1d255d3SCy Schubert 	if (pos[1] < 4) {
2990c1d255d3SCy Schubert 		wpa_printf(MSG_MSGDUMP,
2991c1d255d3SCy Schubert 			   "Too short vendor specific IE ignored (len=%u)",
2992c1d255d3SCy Schubert 			   pos[1]);
2993c1d255d3SCy Schubert 		return;
2994c1d255d3SCy Schubert 	}
2995c1d255d3SCy Schubert 
2996c1d255d3SCy Schubert 	oui = WPA_GET_BE24(&pos[2]);
2997c1d255d3SCy Schubert 	if (oui == OUI_MICROSOFT && pos[5] == WMM_OUI_TYPE && pos[1] > 4) {
2998c1d255d3SCy Schubert 		if (pos[6] == WMM_OUI_SUBTYPE_INFORMATION_ELEMENT) {
2999c1d255d3SCy Schubert 			ie->wmm = &pos[2];
3000c1d255d3SCy Schubert 			ie->wmm_len = pos[1];
3001c1d255d3SCy Schubert 			wpa_hexdump(MSG_DEBUG, "WPA: WMM IE",
3002c1d255d3SCy Schubert 				    ie->wmm, ie->wmm_len);
3003c1d255d3SCy Schubert 		} else if (pos[6] == WMM_OUI_SUBTYPE_PARAMETER_ELEMENT) {
3004c1d255d3SCy Schubert 			ie->wmm = &pos[2];
3005c1d255d3SCy Schubert 			ie->wmm_len = pos[1];
3006c1d255d3SCy Schubert 			wpa_hexdump(MSG_DEBUG, "WPA: WMM Parameter Element",
3007c1d255d3SCy Schubert 				    ie->wmm, ie->wmm_len);
3008c1d255d3SCy Schubert 		}
3009c1d255d3SCy Schubert 	}
3010c1d255d3SCy Schubert }
3011c1d255d3SCy Schubert 
3012c1d255d3SCy Schubert 
3013c1d255d3SCy Schubert /**
3014c1d255d3SCy Schubert  * wpa_parse_generic - Parse EAPOL-Key Key Data Generic IEs
3015c1d255d3SCy Schubert  * @pos: Pointer to the IE header
3016c1d255d3SCy Schubert  * @ie: Pointer to parsed IE data
3017c1d255d3SCy Schubert  * Returns: 0 on success, 1 if end mark is found, 2 if KDE is not recognized
3018c1d255d3SCy Schubert  */
3019c1d255d3SCy Schubert static int wpa_parse_generic(const u8 *pos, struct wpa_eapol_ie_parse *ie)
3020c1d255d3SCy Schubert {
3021c1d255d3SCy Schubert 	if (pos[1] == 0)
3022c1d255d3SCy Schubert 		return 1;
3023c1d255d3SCy Schubert 
3024c1d255d3SCy Schubert 	if (pos[1] >= 6 &&
3025c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == WPA_OUI_TYPE &&
3026c1d255d3SCy Schubert 	    pos[2 + WPA_SELECTOR_LEN] == 1 &&
3027c1d255d3SCy Schubert 	    pos[2 + WPA_SELECTOR_LEN + 1] == 0) {
3028c1d255d3SCy Schubert 		ie->wpa_ie = pos;
3029c1d255d3SCy Schubert 		ie->wpa_ie_len = pos[1] + 2;
3030c1d255d3SCy Schubert 		wpa_hexdump(MSG_DEBUG, "WPA: WPA IE in EAPOL-Key",
3031c1d255d3SCy Schubert 			    ie->wpa_ie, ie->wpa_ie_len);
3032c1d255d3SCy Schubert 		return 0;
3033c1d255d3SCy Schubert 	}
3034c1d255d3SCy Schubert 
3035c1d255d3SCy Schubert 	if (pos[1] >= 4 && WPA_GET_BE32(pos + 2) == OSEN_IE_VENDOR_TYPE) {
3036c1d255d3SCy Schubert 		ie->osen = pos;
3037c1d255d3SCy Schubert 		ie->osen_len = pos[1] + 2;
3038c1d255d3SCy Schubert 		return 0;
3039c1d255d3SCy Schubert 	}
3040c1d255d3SCy Schubert 
3041c1d255d3SCy Schubert 	if (pos[1] >= RSN_SELECTOR_LEN + PMKID_LEN &&
3042c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_PMKID) {
3043c1d255d3SCy Schubert 		ie->pmkid = pos + 2 + RSN_SELECTOR_LEN;
3044c1d255d3SCy Schubert 		wpa_hexdump(MSG_DEBUG, "WPA: PMKID in EAPOL-Key",
3045c1d255d3SCy Schubert 			    pos, pos[1] + 2);
3046c1d255d3SCy Schubert 		return 0;
3047c1d255d3SCy Schubert 	}
3048c1d255d3SCy Schubert 
3049c1d255d3SCy Schubert 	if (pos[1] >= RSN_SELECTOR_LEN + 2 &&
3050c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_KEYID) {
3051c1d255d3SCy Schubert 		ie->key_id = pos + 2 + RSN_SELECTOR_LEN;
3052c1d255d3SCy Schubert 		wpa_hexdump(MSG_DEBUG, "WPA: KeyID in EAPOL-Key",
3053c1d255d3SCy Schubert 			    pos, pos[1] + 2);
3054c1d255d3SCy Schubert 		return 0;
3055c1d255d3SCy Schubert 	}
3056c1d255d3SCy Schubert 
3057c1d255d3SCy Schubert 	if (pos[1] > RSN_SELECTOR_LEN + 2 &&
3058c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_GROUPKEY) {
3059c1d255d3SCy Schubert 		ie->gtk = pos + 2 + RSN_SELECTOR_LEN;
3060c1d255d3SCy Schubert 		ie->gtk_len = pos[1] - RSN_SELECTOR_LEN;
3061c1d255d3SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "WPA: GTK in EAPOL-Key",
3062c1d255d3SCy Schubert 				pos, pos[1] + 2);
3063c1d255d3SCy Schubert 		return 0;
3064c1d255d3SCy Schubert 	}
3065c1d255d3SCy Schubert 
3066c1d255d3SCy Schubert 	if (pos[1] > RSN_SELECTOR_LEN + 2 &&
3067c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_MAC_ADDR) {
3068c1d255d3SCy Schubert 		ie->mac_addr = pos + 2 + RSN_SELECTOR_LEN;
3069c1d255d3SCy Schubert 		ie->mac_addr_len = pos[1] - RSN_SELECTOR_LEN;
3070c1d255d3SCy Schubert 		wpa_hexdump(MSG_DEBUG, "WPA: MAC Address in EAPOL-Key",
3071c1d255d3SCy Schubert 			    pos, pos[1] + 2);
3072c1d255d3SCy Schubert 		return 0;
3073c1d255d3SCy Schubert 	}
3074c1d255d3SCy Schubert 
3075c1d255d3SCy Schubert 	if (pos[1] > RSN_SELECTOR_LEN + 2 &&
3076c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_IGTK) {
3077c1d255d3SCy Schubert 		ie->igtk = pos + 2 + RSN_SELECTOR_LEN;
3078c1d255d3SCy Schubert 		ie->igtk_len = pos[1] - RSN_SELECTOR_LEN;
3079c1d255d3SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "WPA: IGTK in EAPOL-Key",
3080c1d255d3SCy Schubert 				pos, pos[1] + 2);
3081c1d255d3SCy Schubert 		return 0;
3082c1d255d3SCy Schubert 	}
3083c1d255d3SCy Schubert 
3084c1d255d3SCy Schubert 	if (pos[1] > RSN_SELECTOR_LEN + 2 &&
3085c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_BIGTK) {
3086c1d255d3SCy Schubert 		ie->bigtk = pos + 2 + RSN_SELECTOR_LEN;
3087c1d255d3SCy Schubert 		ie->bigtk_len = pos[1] - RSN_SELECTOR_LEN;
3088c1d255d3SCy Schubert 		wpa_hexdump_key(MSG_DEBUG, "WPA: BIGTK in EAPOL-Key",
3089c1d255d3SCy Schubert 				pos, pos[1] + 2);
3090c1d255d3SCy Schubert 		return 0;
3091c1d255d3SCy Schubert 	}
3092c1d255d3SCy Schubert 
3093c1d255d3SCy Schubert 	if (pos[1] >= RSN_SELECTOR_LEN + 1 &&
3094c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == WFA_KEY_DATA_IP_ADDR_REQ) {
3095c1d255d3SCy Schubert 		ie->ip_addr_req = pos + 2 + RSN_SELECTOR_LEN;
3096c1d255d3SCy Schubert 		wpa_hexdump(MSG_DEBUG, "WPA: IP Address Request in EAPOL-Key",
3097c1d255d3SCy Schubert 			    ie->ip_addr_req, pos[1] - RSN_SELECTOR_LEN);
3098c1d255d3SCy Schubert 		return 0;
3099c1d255d3SCy Schubert 	}
3100c1d255d3SCy Schubert 
3101c1d255d3SCy Schubert 	if (pos[1] >= RSN_SELECTOR_LEN + 3 * 4 &&
3102c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == WFA_KEY_DATA_IP_ADDR_ALLOC) {
3103c1d255d3SCy Schubert 		ie->ip_addr_alloc = pos + 2 + RSN_SELECTOR_LEN;
3104c1d255d3SCy Schubert 		wpa_hexdump(MSG_DEBUG,
3105c1d255d3SCy Schubert 			    "WPA: IP Address Allocation in EAPOL-Key",
3106c1d255d3SCy Schubert 			    ie->ip_addr_alloc, pos[1] - RSN_SELECTOR_LEN);
3107c1d255d3SCy Schubert 		return 0;
3108c1d255d3SCy Schubert 	}
3109c1d255d3SCy Schubert 
3110c1d255d3SCy Schubert 	if (pos[1] > RSN_SELECTOR_LEN + 2 &&
3111c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == RSN_KEY_DATA_OCI) {
3112c1d255d3SCy Schubert 		ie->oci = pos + 2 + RSN_SELECTOR_LEN;
3113c1d255d3SCy Schubert 		ie->oci_len = pos[1] - RSN_SELECTOR_LEN;
3114c1d255d3SCy Schubert 		wpa_hexdump(MSG_DEBUG, "WPA: OCI KDE in EAPOL-Key",
3115c1d255d3SCy Schubert 			    pos, pos[1] + 2);
3116c1d255d3SCy Schubert 		return 0;
3117c1d255d3SCy Schubert 	}
3118c1d255d3SCy Schubert 
3119c1d255d3SCy Schubert 	if (pos[1] >= RSN_SELECTOR_LEN + 1 &&
3120c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == WFA_KEY_DATA_TRANSITION_DISABLE) {
3121c1d255d3SCy Schubert 		ie->transition_disable = pos + 2 + RSN_SELECTOR_LEN;
3122c1d255d3SCy Schubert 		ie->transition_disable_len = pos[1] - RSN_SELECTOR_LEN;
3123c1d255d3SCy Schubert 		wpa_hexdump(MSG_DEBUG,
3124c1d255d3SCy Schubert 			    "WPA: Transition Disable KDE in EAPOL-Key",
3125c1d255d3SCy Schubert 			    pos, pos[1] + 2);
3126c1d255d3SCy Schubert 		return 0;
3127c1d255d3SCy Schubert 	}
3128c1d255d3SCy Schubert 
3129c1d255d3SCy Schubert 	if (pos[1] >= RSN_SELECTOR_LEN + 2 &&
3130c1d255d3SCy Schubert 	    RSN_SELECTOR_GET(pos + 2) == WFA_KEY_DATA_DPP) {
3131c1d255d3SCy Schubert 		ie->dpp_kde = pos + 2 + RSN_SELECTOR_LEN;
3132c1d255d3SCy Schubert 		ie->dpp_kde_len = pos[1] - RSN_SELECTOR_LEN;
3133c1d255d3SCy Schubert 		wpa_hexdump(MSG_DEBUG, "WPA: DPP KDE in EAPOL-Key",
3134c1d255d3SCy Schubert 			    pos, pos[1] + 2);
3135c1d255d3SCy Schubert 		return 0;
3136c1d255d3SCy Schubert 	}
3137c1d255d3SCy Schubert 
3138c1d255d3SCy Schubert 	return 2;
3139c1d255d3SCy Schubert }
3140c1d255d3SCy Schubert 
3141c1d255d3SCy Schubert 
3142c1d255d3SCy Schubert /**
3143c1d255d3SCy Schubert  * wpa_parse_kde_ies - Parse EAPOL-Key Key Data IEs
3144c1d255d3SCy Schubert  * @buf: Pointer to the Key Data buffer
3145c1d255d3SCy Schubert  * @len: Key Data Length
3146c1d255d3SCy Schubert  * @ie: Pointer to parsed IE data
3147c1d255d3SCy Schubert  * Returns: 0 on success, -1 on failure
3148c1d255d3SCy Schubert  */
3149c1d255d3SCy Schubert int wpa_parse_kde_ies(const u8 *buf, size_t len, struct wpa_eapol_ie_parse *ie)
3150c1d255d3SCy Schubert {
3151c1d255d3SCy Schubert 	const u8 *pos, *end;
3152c1d255d3SCy Schubert 	int ret = 0;
3153c1d255d3SCy Schubert 
3154c1d255d3SCy Schubert 	os_memset(ie, 0, sizeof(*ie));
3155c1d255d3SCy Schubert 	for (pos = buf, end = pos + len; end - pos > 1; pos += 2 + pos[1]) {
3156c1d255d3SCy Schubert 		if (pos[0] == 0xdd &&
3157c1d255d3SCy Schubert 		    ((pos == buf + len - 1) || pos[1] == 0)) {
3158c1d255d3SCy Schubert 			/* Ignore padding */
3159c1d255d3SCy Schubert 			break;
3160c1d255d3SCy Schubert 		}
3161c1d255d3SCy Schubert 		if (2 + pos[1] > end - pos) {
3162c1d255d3SCy Schubert 			wpa_printf(MSG_DEBUG,
3163c1d255d3SCy Schubert 				   "WPA: EAPOL-Key Key Data underflow (ie=%d len=%d pos=%d)",
3164c1d255d3SCy Schubert 				   pos[0], pos[1], (int) (pos - buf));
3165c1d255d3SCy Schubert 			wpa_hexdump_key(MSG_DEBUG, "WPA: Key Data", buf, len);
3166c1d255d3SCy Schubert 			ret = -1;
3167c1d255d3SCy Schubert 			break;
3168c1d255d3SCy Schubert 		}
3169c1d255d3SCy Schubert 		if (*pos == WLAN_EID_RSN) {
3170c1d255d3SCy Schubert 			ie->rsn_ie = pos;
3171c1d255d3SCy Schubert 			ie->rsn_ie_len = pos[1] + 2;
3172c1d255d3SCy Schubert 			wpa_hexdump(MSG_DEBUG, "WPA: RSN IE in EAPOL-Key",
3173c1d255d3SCy Schubert 				    ie->rsn_ie, ie->rsn_ie_len);
3174c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_RSNX) {
3175c1d255d3SCy Schubert 			ie->rsnxe = pos;
3176c1d255d3SCy Schubert 			ie->rsnxe_len = pos[1] + 2;
3177c1d255d3SCy Schubert 			wpa_hexdump(MSG_DEBUG, "WPA: RSNXE in EAPOL-Key",
3178c1d255d3SCy Schubert 				    ie->rsnxe, ie->rsnxe_len);
3179c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_MOBILITY_DOMAIN) {
3180c1d255d3SCy Schubert 			ie->mdie = pos;
3181c1d255d3SCy Schubert 			ie->mdie_len = pos[1] + 2;
3182c1d255d3SCy Schubert 			wpa_hexdump(MSG_DEBUG, "WPA: MDIE in EAPOL-Key",
3183c1d255d3SCy Schubert 				    ie->mdie, ie->mdie_len);
3184c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_FAST_BSS_TRANSITION) {
3185c1d255d3SCy Schubert 			ie->ftie = pos;
3186c1d255d3SCy Schubert 			ie->ftie_len = pos[1] + 2;
3187c1d255d3SCy Schubert 			wpa_hexdump(MSG_DEBUG, "WPA: FTIE in EAPOL-Key",
3188c1d255d3SCy Schubert 				    ie->ftie, ie->ftie_len);
3189c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_TIMEOUT_INTERVAL && pos[1] >= 5) {
3190c1d255d3SCy Schubert 			if (pos[2] == WLAN_TIMEOUT_REASSOC_DEADLINE) {
3191c1d255d3SCy Schubert 				ie->reassoc_deadline = pos;
3192c1d255d3SCy Schubert 				wpa_hexdump(MSG_DEBUG, "WPA: Reassoc Deadline "
3193c1d255d3SCy Schubert 					    "in EAPOL-Key",
3194c1d255d3SCy Schubert 					    ie->reassoc_deadline, pos[1] + 2);
3195c1d255d3SCy Schubert 			} else if (pos[2] == WLAN_TIMEOUT_KEY_LIFETIME) {
3196c1d255d3SCy Schubert 				ie->key_lifetime = pos;
3197c1d255d3SCy Schubert 				wpa_hexdump(MSG_DEBUG, "WPA: KeyLifetime "
3198c1d255d3SCy Schubert 					    "in EAPOL-Key",
3199c1d255d3SCy Schubert 					    ie->key_lifetime, pos[1] + 2);
3200c1d255d3SCy Schubert 			} else {
3201c1d255d3SCy Schubert 				wpa_hexdump(MSG_DEBUG, "WPA: Unrecognized "
3202c1d255d3SCy Schubert 					    "EAPOL-Key Key Data IE",
3203c1d255d3SCy Schubert 					    pos, 2 + pos[1]);
3204c1d255d3SCy Schubert 			}
3205c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_LINK_ID) {
3206c1d255d3SCy Schubert 			if (pos[1] >= 18) {
3207c1d255d3SCy Schubert 				ie->lnkid = pos;
3208c1d255d3SCy Schubert 				ie->lnkid_len = pos[1] + 2;
3209c1d255d3SCy Schubert 			}
3210c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_EXT_CAPAB) {
3211c1d255d3SCy Schubert 			ie->ext_capab = pos;
3212c1d255d3SCy Schubert 			ie->ext_capab_len = pos[1] + 2;
3213c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_SUPP_RATES) {
3214c1d255d3SCy Schubert 			ie->supp_rates = pos;
3215c1d255d3SCy Schubert 			ie->supp_rates_len = pos[1] + 2;
3216c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_EXT_SUPP_RATES) {
3217c1d255d3SCy Schubert 			ie->ext_supp_rates = pos;
3218c1d255d3SCy Schubert 			ie->ext_supp_rates_len = pos[1] + 2;
3219c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_HT_CAP &&
3220c1d255d3SCy Schubert 			   pos[1] >= sizeof(struct ieee80211_ht_capabilities)) {
3221c1d255d3SCy Schubert 			ie->ht_capabilities = pos + 2;
3222c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_VHT_AID) {
3223c1d255d3SCy Schubert 			if (pos[1] >= 2)
3224c1d255d3SCy Schubert 				ie->aid = WPA_GET_LE16(pos + 2) & 0x3fff;
3225c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_VHT_CAP &&
3226c1d255d3SCy Schubert 			   pos[1] >= sizeof(struct ieee80211_vht_capabilities))
3227c1d255d3SCy Schubert 		{
3228c1d255d3SCy Schubert 			ie->vht_capabilities = pos + 2;
3229c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_EXTENSION &&
3230c1d255d3SCy Schubert 			   pos[1] >= 1 + IEEE80211_HE_CAPAB_MIN_LEN &&
3231c1d255d3SCy Schubert 			   pos[2] == WLAN_EID_EXT_HE_CAPABILITIES) {
3232c1d255d3SCy Schubert 			ie->he_capabilities = pos + 3;
3233c1d255d3SCy Schubert 			ie->he_capab_len = pos[1] - 1;
3234*4b72b91aSCy Schubert 		} else if (*pos == WLAN_EID_EXTENSION &&
3235*4b72b91aSCy Schubert 			   pos[1] >= 1 +
3236*4b72b91aSCy Schubert 			   sizeof(struct ieee80211_he_6ghz_band_cap) &&
3237*4b72b91aSCy Schubert 			   pos[2] == WLAN_EID_EXT_HE_6GHZ_BAND_CAP) {
3238*4b72b91aSCy Schubert 			ie->he_6ghz_capabilities = pos + 3;
3239c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_QOS && pos[1] >= 1) {
3240c1d255d3SCy Schubert 			ie->qosinfo = pos[2];
3241c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_SUPPORTED_CHANNELS) {
3242c1d255d3SCy Schubert 			ie->supp_channels = pos + 2;
3243c1d255d3SCy Schubert 			ie->supp_channels_len = pos[1];
3244c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_SUPPORTED_OPERATING_CLASSES) {
3245c1d255d3SCy Schubert 			/*
3246c1d255d3SCy Schubert 			 * The value of the Length field of the Supported
3247c1d255d3SCy Schubert 			 * Operating Classes element is between 2 and 253.
3248c1d255d3SCy Schubert 			 * Silently skip invalid elements to avoid interop
3249c1d255d3SCy Schubert 			 * issues when trying to use the value.
3250c1d255d3SCy Schubert 			 */
3251c1d255d3SCy Schubert 			if (pos[1] >= 2 && pos[1] <= 253) {
3252c1d255d3SCy Schubert 				ie->supp_oper_classes = pos + 2;
3253c1d255d3SCy Schubert 				ie->supp_oper_classes_len = pos[1];
3254c1d255d3SCy Schubert 			}
3255c1d255d3SCy Schubert 		} else if (*pos == WLAN_EID_VENDOR_SPECIFIC) {
3256c1d255d3SCy Schubert 			ret = wpa_parse_generic(pos, ie);
3257c1d255d3SCy Schubert 			if (ret == 1) {
3258c1d255d3SCy Schubert 				/* end mark found */
3259c1d255d3SCy Schubert 				ret = 0;
3260c1d255d3SCy Schubert 				break;
3261c1d255d3SCy Schubert 			}
3262c1d255d3SCy Schubert 
3263c1d255d3SCy Schubert 			if (ret == 2) {
3264c1d255d3SCy Schubert 				/* not a known KDE */
3265c1d255d3SCy Schubert 				wpa_parse_vendor_specific(pos, end, ie);
3266c1d255d3SCy Schubert 			}
3267c1d255d3SCy Schubert 
3268c1d255d3SCy Schubert 			ret = 0;
3269c1d255d3SCy Schubert 		} else {
3270c1d255d3SCy Schubert 			wpa_hexdump(MSG_DEBUG,
3271c1d255d3SCy Schubert 				    "WPA: Unrecognized EAPOL-Key Key Data IE",
3272c1d255d3SCy Schubert 				    pos, 2 + pos[1]);
3273c1d255d3SCy Schubert 		}
3274c1d255d3SCy Schubert 	}
3275c1d255d3SCy Schubert 
3276c1d255d3SCy Schubert 	return ret;
3277c1d255d3SCy Schubert }
3278c1d255d3SCy Schubert 
3279c1d255d3SCy Schubert 
3280c1d255d3SCy Schubert #ifdef CONFIG_PASN
3281c1d255d3SCy Schubert 
3282c1d255d3SCy Schubert /*
3283c1d255d3SCy Schubert  * wpa_pasn_build_auth_header - Add the MAC header and initialize Authentication
3284c1d255d3SCy Schubert  * frame for PASN
3285c1d255d3SCy Schubert  *
3286c1d255d3SCy Schubert  * @buf: Buffer in which the header will be added
3287c1d255d3SCy Schubert  * @bssid: The BSSID of the AP
3288c1d255d3SCy Schubert  * @src: Source address
3289c1d255d3SCy Schubert  * @dst: Destination address
3290c1d255d3SCy Schubert  * @trans_seq: Authentication transaction sequence number
3291c1d255d3SCy Schubert  * @status: Authentication status
3292c1d255d3SCy Schubert  */
3293c1d255d3SCy Schubert void wpa_pasn_build_auth_header(struct wpabuf *buf, const u8 *bssid,
3294c1d255d3SCy Schubert 				const u8 *src, const u8 *dst,
3295c1d255d3SCy Schubert 				u8 trans_seq, u16 status)
3296c1d255d3SCy Schubert {
3297c1d255d3SCy Schubert 	struct ieee80211_mgmt *auth;
3298c1d255d3SCy Schubert 
3299c1d255d3SCy Schubert 	wpa_printf(MSG_DEBUG, "PASN: Add authentication header. trans_seq=%u",
3300c1d255d3SCy Schubert 		   trans_seq);
3301c1d255d3SCy Schubert 
3302c1d255d3SCy Schubert 	auth = wpabuf_put(buf, offsetof(struct ieee80211_mgmt,
3303c1d255d3SCy Schubert 					u.auth.variable));
3304c1d255d3SCy Schubert 
3305c1d255d3SCy Schubert 	auth->frame_control = host_to_le16((WLAN_FC_TYPE_MGMT << 2) |
3306c1d255d3SCy Schubert 					   (WLAN_FC_STYPE_AUTH << 4));
3307c1d255d3SCy Schubert 
3308c1d255d3SCy Schubert 	os_memcpy(auth->da, dst, ETH_ALEN);
3309c1d255d3SCy Schubert 	os_memcpy(auth->sa, src, ETH_ALEN);
3310c1d255d3SCy Schubert 	os_memcpy(auth->bssid, bssid, ETH_ALEN);
3311c1d255d3SCy Schubert 	auth->seq_ctrl = 0;
3312c1d255d3SCy Schubert 
3313c1d255d3SCy Schubert 	auth->u.auth.auth_alg = host_to_le16(WLAN_AUTH_PASN);
3314c1d255d3SCy Schubert 	auth->u.auth.auth_transaction = host_to_le16(trans_seq);
3315c1d255d3SCy Schubert 	auth->u.auth.status_code = host_to_le16(status);
3316c1d255d3SCy Schubert }
3317c1d255d3SCy Schubert 
3318c1d255d3SCy Schubert 
3319c1d255d3SCy Schubert /*
3320c1d255d3SCy Schubert  * wpa_pasn_add_rsne - Add an RSNE for PASN authentication
3321c1d255d3SCy Schubert  * @buf: Buffer in which the IE will be added
3322c1d255d3SCy Schubert  * @pmkid: Optional PMKID. Can be NULL.
3323c1d255d3SCy Schubert  * @akmp: Authentication and key management protocol
3324c1d255d3SCy Schubert  * @cipher: The cipher suite
3325c1d255d3SCy Schubert  */
3326c1d255d3SCy Schubert int wpa_pasn_add_rsne(struct wpabuf *buf, const u8 *pmkid, int akmp, int cipher)
3327c1d255d3SCy Schubert {
3328c1d255d3SCy Schubert 	struct rsn_ie_hdr *hdr;
3329c1d255d3SCy Schubert 	u32 suite;
3330c1d255d3SCy Schubert 	u16 capab;
3331c1d255d3SCy Schubert 	u8 *pos;
3332c1d255d3SCy Schubert 	u8 rsne_len;
3333c1d255d3SCy Schubert 
3334c1d255d3SCy Schubert 	wpa_printf(MSG_DEBUG, "PASN: Add RSNE");
3335c1d255d3SCy Schubert 
3336c1d255d3SCy Schubert 	rsne_len = sizeof(*hdr) + RSN_SELECTOR_LEN +
3337c1d255d3SCy Schubert 		2 + RSN_SELECTOR_LEN + 2 + RSN_SELECTOR_LEN +
3338c1d255d3SCy Schubert 		2 + RSN_SELECTOR_LEN + 2 + (pmkid ? PMKID_LEN : 0);
3339c1d255d3SCy Schubert 
3340c1d255d3SCy Schubert 	if (wpabuf_tailroom(buf) < rsne_len)
3341c1d255d3SCy Schubert 		return -1;
3342c1d255d3SCy Schubert 	hdr = wpabuf_put(buf, rsne_len);
3343c1d255d3SCy Schubert 	hdr->elem_id = WLAN_EID_RSN;
3344c1d255d3SCy Schubert 	hdr->len = rsne_len - 2;
3345c1d255d3SCy Schubert 	WPA_PUT_LE16(hdr->version, RSN_VERSION);
3346c1d255d3SCy Schubert 	pos = (u8 *) (hdr + 1);
3347c1d255d3SCy Schubert 
3348c1d255d3SCy Schubert 	/* Group addressed data is not allowed */
3349c1d255d3SCy Schubert 	RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED);
3350c1d255d3SCy Schubert 	pos += RSN_SELECTOR_LEN;
3351c1d255d3SCy Schubert 
3352c1d255d3SCy Schubert 	/* Add the pairwise cipher */
3353c1d255d3SCy Schubert 	WPA_PUT_LE16(pos, 1);
3354c1d255d3SCy Schubert 	pos += 2;
3355c1d255d3SCy Schubert 	suite = wpa_cipher_to_suite(WPA_PROTO_RSN, cipher);
3356c1d255d3SCy Schubert 	RSN_SELECTOR_PUT(pos, suite);
3357c1d255d3SCy Schubert 	pos += RSN_SELECTOR_LEN;
3358c1d255d3SCy Schubert 
3359c1d255d3SCy Schubert 	/* Add the AKM suite */
3360c1d255d3SCy Schubert 	WPA_PUT_LE16(pos, 1);
3361c1d255d3SCy Schubert 	pos += 2;
3362c1d255d3SCy Schubert 
3363c1d255d3SCy Schubert 	switch (akmp) {
3364c1d255d3SCy Schubert 	case WPA_KEY_MGMT_PASN:
3365c1d255d3SCy Schubert 		RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_PASN);
3366c1d255d3SCy Schubert 		break;
3367c1d255d3SCy Schubert #ifdef CONFIG_SAE
3368c1d255d3SCy Schubert 	case WPA_KEY_MGMT_SAE:
3369c1d255d3SCy Schubert 		RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_SAE);
3370c1d255d3SCy Schubert 		break;
3371c1d255d3SCy Schubert #endif /* CONFIG_SAE */
3372c1d255d3SCy Schubert #ifdef CONFIG_FILS
3373c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA256:
3374c1d255d3SCy Schubert 		RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_FILS_SHA256);
3375c1d255d3SCy Schubert 		break;
3376c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA384:
3377c1d255d3SCy Schubert 		RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_FILS_SHA384);
3378c1d255d3SCy Schubert 		break;
3379c1d255d3SCy Schubert #endif /* CONFIG_FILS */
3380c1d255d3SCy Schubert #ifdef CONFIG_IEEE80211R
3381c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FT_PSK:
3382c1d255d3SCy Schubert 		RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_FT_PSK);
3383c1d255d3SCy Schubert 		break;
3384c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FT_IEEE8021X:
3385c1d255d3SCy Schubert 		RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_FT_802_1X);
3386c1d255d3SCy Schubert 		break;
3387c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FT_IEEE8021X_SHA384:
3388c1d255d3SCy Schubert 		RSN_SELECTOR_PUT(pos, RSN_AUTH_KEY_MGMT_FT_802_1X_SHA384);
3389c1d255d3SCy Schubert 		break;
3390c1d255d3SCy Schubert #endif /* CONFIG_IEEE80211R */
3391c1d255d3SCy Schubert 	default:
3392c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR, "PASN: Invalid AKMP=0x%x", akmp);
3393c1d255d3SCy Schubert 		return -1;
3394c1d255d3SCy Schubert 	}
3395c1d255d3SCy Schubert 	pos += RSN_SELECTOR_LEN;
3396c1d255d3SCy Schubert 
3397c1d255d3SCy Schubert 	/* RSN Capabilities: PASN mandates both MFP capable and required */
3398c1d255d3SCy Schubert 	capab = WPA_CAPABILITY_MFPC | WPA_CAPABILITY_MFPR;
3399c1d255d3SCy Schubert 	WPA_PUT_LE16(pos, capab);
3400c1d255d3SCy Schubert 	pos += 2;
3401c1d255d3SCy Schubert 
3402c1d255d3SCy Schubert 	if (pmkid) {
3403c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Adding PMKID");
3404c1d255d3SCy Schubert 
3405c1d255d3SCy Schubert 		WPA_PUT_LE16(pos, 1);
3406c1d255d3SCy Schubert 		pos += 2;
3407c1d255d3SCy Schubert 		os_memcpy(pos, pmkid, PMKID_LEN);
3408c1d255d3SCy Schubert 		pos += PMKID_LEN;
3409c1d255d3SCy Schubert 	} else {
3410c1d255d3SCy Schubert 		WPA_PUT_LE16(pos, 0);
3411c1d255d3SCy Schubert 		pos += 2;
3412c1d255d3SCy Schubert 	}
3413c1d255d3SCy Schubert 
3414c1d255d3SCy Schubert 	/* Group addressed management is not allowed */
3415c1d255d3SCy Schubert 	RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED);
3416c1d255d3SCy Schubert 
3417c1d255d3SCy Schubert 	return 0;
3418c1d255d3SCy Schubert }
3419c1d255d3SCy Schubert 
3420c1d255d3SCy Schubert 
3421c1d255d3SCy Schubert /*
3422c1d255d3SCy Schubert  * wpa_pasn_add_parameter_ie - Add PASN Parameters IE for PASN authentication
3423c1d255d3SCy Schubert  * @buf: Buffer in which the IE will be added
3424c1d255d3SCy Schubert  * @pasn_group: Finite Cyclic Group ID for PASN authentication
3425c1d255d3SCy Schubert  * @wrapped_data_format: Format of the data in the Wrapped Data IE
3426c1d255d3SCy Schubert  * @pubkey: A buffer holding the local public key. Can be NULL
3427c1d255d3SCy Schubert  * @compressed: In case pubkey is included, indicates if the public key is
3428c1d255d3SCy Schubert  *     compressed (only x coordinate is included) or not (both x and y
3429c1d255d3SCy Schubert  *     coordinates are included)
3430c1d255d3SCy Schubert  * @comeback: A buffer holding the comeback token. Can be NULL
3431c1d255d3SCy Schubert  * @after: If comeback is set, defined the comeback time in seconds. -1 to not
3432c1d255d3SCy Schubert  *	include the Comeback After field (frames from non-AP STA).
3433c1d255d3SCy Schubert  */
3434c1d255d3SCy Schubert void wpa_pasn_add_parameter_ie(struct wpabuf *buf, u16 pasn_group,
3435c1d255d3SCy Schubert 			       u8 wrapped_data_format,
3436c1d255d3SCy Schubert 			       const struct wpabuf *pubkey, bool compressed,
3437c1d255d3SCy Schubert 			       const struct wpabuf *comeback, int after)
3438c1d255d3SCy Schubert {
3439c1d255d3SCy Schubert 	struct pasn_parameter_ie *params;
3440c1d255d3SCy Schubert 
3441c1d255d3SCy Schubert 	wpa_printf(MSG_DEBUG, "PASN: Add PASN Parameters element");
3442c1d255d3SCy Schubert 
3443c1d255d3SCy Schubert 	params = wpabuf_put(buf, sizeof(*params));
3444c1d255d3SCy Schubert 
3445c1d255d3SCy Schubert 	params->id = WLAN_EID_EXTENSION;
3446c1d255d3SCy Schubert 	params->len = sizeof(*params) - 2;
3447c1d255d3SCy Schubert 	params->id_ext = WLAN_EID_EXT_PASN_PARAMS;
3448c1d255d3SCy Schubert 	params->control = 0;
3449c1d255d3SCy Schubert 	params->wrapped_data_format = wrapped_data_format;
3450c1d255d3SCy Schubert 
3451c1d255d3SCy Schubert 	if (comeback) {
3452c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Adding comeback data");
3453c1d255d3SCy Schubert 
3454c1d255d3SCy Schubert 		/*
3455c1d255d3SCy Schubert 		 * 2 octets for the 'after' field + 1 octet for the length +
3456c1d255d3SCy Schubert 		 * actual cookie data
3457c1d255d3SCy Schubert 		 */
3458c1d255d3SCy Schubert 		if (after >= 0)
3459c1d255d3SCy Schubert 			params->len += 2;
3460c1d255d3SCy Schubert 		params->len += 1 + wpabuf_len(comeback);
3461c1d255d3SCy Schubert 		params->control |= WPA_PASN_CTRL_COMEBACK_INFO_PRESENT;
3462c1d255d3SCy Schubert 
3463c1d255d3SCy Schubert 		if (after >= 0)
3464c1d255d3SCy Schubert 			wpabuf_put_le16(buf, after);
3465c1d255d3SCy Schubert 		wpabuf_put_u8(buf, wpabuf_len(comeback));
3466c1d255d3SCy Schubert 		wpabuf_put_buf(buf, comeback);
3467c1d255d3SCy Schubert 	}
3468c1d255d3SCy Schubert 
3469c1d255d3SCy Schubert 	if (pubkey) {
3470c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG,
3471c1d255d3SCy Schubert 			   "PASN: Adding public key and group ID %u",
3472c1d255d3SCy Schubert 			   pasn_group);
3473c1d255d3SCy Schubert 
3474c1d255d3SCy Schubert 		/*
3475c1d255d3SCy Schubert 		 * 2 octets for the finite cyclic group + 2 octets public key
3476c1d255d3SCy Schubert 		 * length + 1 octet for the compressed/uncompressed indication +
3477c1d255d3SCy Schubert 		 * the actual key.
3478c1d255d3SCy Schubert 		 */
3479c1d255d3SCy Schubert 		params->len += 2 + 1 + 1 + wpabuf_len(pubkey);
3480c1d255d3SCy Schubert 		params->control |= WPA_PASN_CTRL_GROUP_AND_KEY_PRESENT;
3481c1d255d3SCy Schubert 
3482c1d255d3SCy Schubert 		wpabuf_put_le16(buf, pasn_group);
3483c1d255d3SCy Schubert 
3484c1d255d3SCy Schubert 		/*
3485c1d255d3SCy Schubert 		 * The first octet indicates whether the public key is
3486c1d255d3SCy Schubert 		 * compressed, as defined in RFC 5480 section 2.2.
3487c1d255d3SCy Schubert 		 */
3488c1d255d3SCy Schubert 		wpabuf_put_u8(buf, wpabuf_len(pubkey) + 1);
3489c1d255d3SCy Schubert 		wpabuf_put_u8(buf, compressed ? WPA_PASN_PUBKEY_COMPRESSED_0 :
3490c1d255d3SCy Schubert 			      WPA_PASN_PUBKEY_UNCOMPRESSED);
3491c1d255d3SCy Schubert 
3492c1d255d3SCy Schubert 		wpabuf_put_buf(buf, pubkey);
3493c1d255d3SCy Schubert 	}
3494c1d255d3SCy Schubert }
3495c1d255d3SCy Schubert 
3496c1d255d3SCy Schubert /*
3497c1d255d3SCy Schubert  * wpa_pasn_add_wrapped_data - Add a Wrapped Data IE to PASN Authentication
3498c1d255d3SCy Schubert  * frame. If needed, the Wrapped Data IE would be fragmented.
3499c1d255d3SCy Schubert  *
3500c1d255d3SCy Schubert  * @buf: Buffer in which the IE will be added
3501c1d255d3SCy Schubert  * @wrapped_data_buf: Buffer holding the wrapped data
3502c1d255d3SCy Schubert  */
3503c1d255d3SCy Schubert int wpa_pasn_add_wrapped_data(struct wpabuf *buf,
3504c1d255d3SCy Schubert 			      struct wpabuf *wrapped_data_buf)
3505c1d255d3SCy Schubert {
3506c1d255d3SCy Schubert 	const u8 *data;
3507c1d255d3SCy Schubert 	size_t data_len;
3508c1d255d3SCy Schubert 	u8 len;
3509c1d255d3SCy Schubert 
3510c1d255d3SCy Schubert 	if (!wrapped_data_buf)
3511c1d255d3SCy Schubert 		return 0;
3512c1d255d3SCy Schubert 
3513c1d255d3SCy Schubert 	wpa_printf(MSG_DEBUG, "PASN: Add wrapped data");
3514c1d255d3SCy Schubert 
3515c1d255d3SCy Schubert 	data = wpabuf_head_u8(wrapped_data_buf);
3516c1d255d3SCy Schubert 	data_len = wpabuf_len(wrapped_data_buf);
3517c1d255d3SCy Schubert 
3518c1d255d3SCy Schubert 	/* nothing to add */
3519c1d255d3SCy Schubert 	if (!data_len)
3520c1d255d3SCy Schubert 		return 0;
3521c1d255d3SCy Schubert 
3522c1d255d3SCy Schubert 	if (data_len <= 254)
3523c1d255d3SCy Schubert 		len = 1 + data_len;
3524c1d255d3SCy Schubert 	else
3525c1d255d3SCy Schubert 		len = 255;
3526c1d255d3SCy Schubert 
3527c1d255d3SCy Schubert 	if (wpabuf_tailroom(buf) < 3 + data_len)
3528c1d255d3SCy Schubert 		return -1;
3529c1d255d3SCy Schubert 
3530c1d255d3SCy Schubert 	wpabuf_put_u8(buf, WLAN_EID_EXTENSION);
3531c1d255d3SCy Schubert 	wpabuf_put_u8(buf, len);
3532c1d255d3SCy Schubert 	wpabuf_put_u8(buf, WLAN_EID_EXT_WRAPPED_DATA);
3533c1d255d3SCy Schubert 	wpabuf_put_data(buf, data, len - 1);
3534c1d255d3SCy Schubert 
3535c1d255d3SCy Schubert 	data += len - 1;
3536c1d255d3SCy Schubert 	data_len -= len - 1;
3537c1d255d3SCy Schubert 
3538c1d255d3SCy Schubert 	while (data_len) {
3539c1d255d3SCy Schubert 		if (wpabuf_tailroom(buf) < 1 + data_len)
3540c1d255d3SCy Schubert 			return -1;
3541c1d255d3SCy Schubert 		wpabuf_put_u8(buf, WLAN_EID_FRAGMENT);
3542c1d255d3SCy Schubert 		len = data_len > 255 ? 255 : data_len;
3543c1d255d3SCy Schubert 		wpabuf_put_u8(buf, len);
3544c1d255d3SCy Schubert 		wpabuf_put_data(buf, data, len);
3545c1d255d3SCy Schubert 		data += len;
3546c1d255d3SCy Schubert 		data_len -= len;
3547c1d255d3SCy Schubert 	}
3548c1d255d3SCy Schubert 
3549c1d255d3SCy Schubert 	return 0;
3550c1d255d3SCy Schubert }
3551c1d255d3SCy Schubert 
3552c1d255d3SCy Schubert 
3553c1d255d3SCy Schubert /*
3554c1d255d3SCy Schubert  * wpa_pasn_validate_rsne - Validate PSAN specific data of RSNE
3555c1d255d3SCy Schubert  * @data: Parsed representation of an RSNE
3556c1d255d3SCy Schubert  * Returns -1 for invalid data; otherwise 0
3557c1d255d3SCy Schubert  */
3558c1d255d3SCy Schubert int wpa_pasn_validate_rsne(const struct wpa_ie_data *data)
3559c1d255d3SCy Schubert {
3560c1d255d3SCy Schubert 	u16 capab = WPA_CAPABILITY_MFPC | WPA_CAPABILITY_MFPR;
3561c1d255d3SCy Schubert 
3562c1d255d3SCy Schubert 	if (data->proto != WPA_PROTO_RSN)
3563c1d255d3SCy Schubert 		return -1;
3564c1d255d3SCy Schubert 
3565c1d255d3SCy Schubert 	if ((data->capabilities & capab) != capab) {
3566c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Invalid RSNE capabilities");
3567c1d255d3SCy Schubert 		return -1;
3568c1d255d3SCy Schubert 	}
3569c1d255d3SCy Schubert 
3570c1d255d3SCy Schubert 	if (!data->has_group || data->group_cipher != WPA_CIPHER_GTK_NOT_USED) {
3571c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Invalid group data cipher");
3572c1d255d3SCy Schubert 		return -1;
3573c1d255d3SCy Schubert 	}
3574c1d255d3SCy Schubert 
3575c1d255d3SCy Schubert 	if (!data->has_pairwise || !data->pairwise_cipher ||
3576c1d255d3SCy Schubert 	    (data->pairwise_cipher & (data->pairwise_cipher - 1))) {
3577c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: No valid pairwise suite");
3578c1d255d3SCy Schubert 		return -1;
3579c1d255d3SCy Schubert 	}
3580c1d255d3SCy Schubert 
3581c1d255d3SCy Schubert 	switch (data->key_mgmt) {
3582c1d255d3SCy Schubert #ifdef CONFIG_SAE
3583c1d255d3SCy Schubert 	case WPA_KEY_MGMT_SAE:
3584c1d255d3SCy Schubert 	/* fall through */
3585c1d255d3SCy Schubert #endif /* CONFIG_SAE */
3586c1d255d3SCy Schubert #ifdef CONFIG_FILS
3587c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA256:
3588c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FILS_SHA384:
3589c1d255d3SCy Schubert 	/* fall through */
3590c1d255d3SCy Schubert #endif /* CONFIG_FILS */
3591c1d255d3SCy Schubert #ifdef CONFIG_IEEE80211R
3592c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FT_PSK:
3593c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FT_IEEE8021X:
3594c1d255d3SCy Schubert 	case WPA_KEY_MGMT_FT_IEEE8021X_SHA384:
3595c1d255d3SCy Schubert 	/* fall through */
3596c1d255d3SCy Schubert #endif /* CONFIG_IEEE80211R */
3597c1d255d3SCy Schubert 	case WPA_KEY_MGMT_PASN:
3598c1d255d3SCy Schubert 		break;
3599c1d255d3SCy Schubert 	default:
3600c1d255d3SCy Schubert 		wpa_printf(MSG_ERROR, "PASN: invalid key_mgmt: 0x%0x",
3601c1d255d3SCy Schubert 			   data->key_mgmt);
3602c1d255d3SCy Schubert 		return -1;
3603c1d255d3SCy Schubert 	}
3604c1d255d3SCy Schubert 
3605c1d255d3SCy Schubert 	if (data->mgmt_group_cipher != WPA_CIPHER_GTK_NOT_USED) {
3606c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Invalid group mgmt cipher");
3607c1d255d3SCy Schubert 		return -1;
3608c1d255d3SCy Schubert 	}
3609c1d255d3SCy Schubert 
3610c1d255d3SCy Schubert 	if (data->num_pmkid > 1) {
3611c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Invalid number of PMKIDs");
3612c1d255d3SCy Schubert 		return -1;
3613c1d255d3SCy Schubert 	}
3614c1d255d3SCy Schubert 
3615c1d255d3SCy Schubert 	return 0;
3616c1d255d3SCy Schubert }
3617c1d255d3SCy Schubert 
3618c1d255d3SCy Schubert 
3619c1d255d3SCy Schubert /*
3620c1d255d3SCy Schubert  * wpa_pasn_parse_parameter_ie - Validates PASN Parameters IE
3621c1d255d3SCy Schubert  * @data: Pointer to the PASN Parameters IE (starting with the EID).
3622c1d255d3SCy Schubert  * @len: Length of the data in the PASN Parameters IE
3623c1d255d3SCy Schubert  * @from_ap: Whether this was received from an AP
3624c1d255d3SCy Schubert  * @pasn_params: On successful return would hold the parsed PASN parameters.
3625c1d255d3SCy Schubert  * Returns: -1 for invalid data; otherwise 0
3626c1d255d3SCy Schubert  *
3627c1d255d3SCy Schubert  * Note: On successful return, the pointers in &pasn_params point to the data in
3628c1d255d3SCy Schubert  * the IE and are not locally allocated (so they should not be freed etc.).
3629c1d255d3SCy Schubert  */
3630c1d255d3SCy Schubert int wpa_pasn_parse_parameter_ie(const u8 *data, u8 len, bool from_ap,
3631c1d255d3SCy Schubert 				struct wpa_pasn_params_data *pasn_params)
3632c1d255d3SCy Schubert {
3633c1d255d3SCy Schubert 	struct pasn_parameter_ie *params = (struct pasn_parameter_ie *) data;
3634c1d255d3SCy Schubert 	const u8 *pos = (const u8 *) (params + 1);
3635c1d255d3SCy Schubert 
3636c1d255d3SCy Schubert 	if (!pasn_params) {
3637c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Invalid params");
3638c1d255d3SCy Schubert 		return -1;
3639c1d255d3SCy Schubert 	}
3640c1d255d3SCy Schubert 
3641c1d255d3SCy Schubert 	if (!params || ((size_t) (params->len + 2) < sizeof(*params)) ||
3642c1d255d3SCy Schubert 	    len < sizeof(*params) || params->len + 2 != len) {
3643c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG,
3644c1d255d3SCy Schubert 			   "PASN: Invalid parameters IE. len=(%u, %u)",
3645c1d255d3SCy Schubert 			   params ? params->len : 0, len);
3646c1d255d3SCy Schubert 		return -1;
3647c1d255d3SCy Schubert 	}
3648c1d255d3SCy Schubert 
3649c1d255d3SCy Schubert 	os_memset(pasn_params, 0, sizeof(*pasn_params));
3650c1d255d3SCy Schubert 
3651c1d255d3SCy Schubert 	switch (params->wrapped_data_format) {
3652c1d255d3SCy Schubert 	case WPA_PASN_WRAPPED_DATA_NO:
3653c1d255d3SCy Schubert 	case WPA_PASN_WRAPPED_DATA_SAE:
3654c1d255d3SCy Schubert 	case WPA_PASN_WRAPPED_DATA_FILS_SK:
3655c1d255d3SCy Schubert 	case WPA_PASN_WRAPPED_DATA_FT:
3656c1d255d3SCy Schubert 		break;
3657c1d255d3SCy Schubert 	default:
3658c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG, "PASN: Invalid wrapped data format");
3659c1d255d3SCy Schubert 		return -1;
3660c1d255d3SCy Schubert 	}
3661c1d255d3SCy Schubert 
3662c1d255d3SCy Schubert 	pasn_params->wrapped_data_format = params->wrapped_data_format;
3663c1d255d3SCy Schubert 
3664c1d255d3SCy Schubert 	len -= sizeof(*params);
3665c1d255d3SCy Schubert 
3666c1d255d3SCy Schubert 	if (params->control & WPA_PASN_CTRL_COMEBACK_INFO_PRESENT) {
3667c1d255d3SCy Schubert 		if (from_ap) {
3668c1d255d3SCy Schubert 			if (len < 2) {
3669c1d255d3SCy Schubert 				wpa_printf(MSG_DEBUG,
3670c1d255d3SCy Schubert 					   "PASN: Invalid Parameters IE: Truncated Comeback After");
3671c1d255d3SCy Schubert 				return -1;
3672c1d255d3SCy Schubert 			}
3673c1d255d3SCy Schubert 			pasn_params->after = WPA_GET_LE16(pos);
3674c1d255d3SCy Schubert 			pos += 2;
3675c1d255d3SCy Schubert 			len -= 2;
3676c1d255d3SCy Schubert 		}
3677c1d255d3SCy Schubert 
3678c1d255d3SCy Schubert 		if (len < 1 || len < 1 + *pos) {
3679c1d255d3SCy Schubert 			wpa_printf(MSG_DEBUG,
3680c1d255d3SCy Schubert 				   "PASN: Invalid Parameters IE: comeback len");
3681c1d255d3SCy Schubert 			return -1;
3682c1d255d3SCy Schubert 		}
3683c1d255d3SCy Schubert 
3684c1d255d3SCy Schubert 		pasn_params->comeback_len = *pos++;
3685c1d255d3SCy Schubert 		len--;
3686c1d255d3SCy Schubert 		pasn_params->comeback = pos;
3687c1d255d3SCy Schubert 		len -=  pasn_params->comeback_len;
3688c1d255d3SCy Schubert 		pos += pasn_params->comeback_len;
3689c1d255d3SCy Schubert 	}
3690c1d255d3SCy Schubert 
3691c1d255d3SCy Schubert 	if (params->control & WPA_PASN_CTRL_GROUP_AND_KEY_PRESENT) {
3692c1d255d3SCy Schubert 		if (len < 3 || len < 3 + pos[2]) {
3693c1d255d3SCy Schubert 			wpa_printf(MSG_DEBUG,
3694c1d255d3SCy Schubert 				   "PASN: Invalid Parameters IE: group and key");
3695c1d255d3SCy Schubert 			return -1;
3696c1d255d3SCy Schubert 		}
3697c1d255d3SCy Schubert 
3698c1d255d3SCy Schubert 		pasn_params->group = WPA_GET_LE16(pos);
3699c1d255d3SCy Schubert 		pos += 2;
3700c1d255d3SCy Schubert 		len -= 2;
3701c1d255d3SCy Schubert 		pasn_params->pubkey_len = *pos++;
3702c1d255d3SCy Schubert 		len--;
3703c1d255d3SCy Schubert 		pasn_params->pubkey = pos;
3704c1d255d3SCy Schubert 		len -= pasn_params->pubkey_len;
3705c1d255d3SCy Schubert 		pos += pasn_params->pubkey_len;
3706c1d255d3SCy Schubert 	}
3707c1d255d3SCy Schubert 
3708c1d255d3SCy Schubert 	if (len) {
3709c1d255d3SCy Schubert 		wpa_printf(MSG_DEBUG,
3710c1d255d3SCy Schubert 			   "PASN: Invalid Parameters IE. Bytes left=%u", len);
3711c1d255d3SCy Schubert 		return -1;
3712c1d255d3SCy Schubert 	}
3713c1d255d3SCy Schubert 
3714c1d255d3SCy Schubert 	return 0;
3715c1d255d3SCy Schubert }
3716c1d255d3SCy Schubert 
3717c1d255d3SCy Schubert 
3718c1d255d3SCy Schubert void wpa_pasn_add_rsnxe(struct wpabuf *buf, u16 capab)
3719c1d255d3SCy Schubert {
3720c1d255d3SCy Schubert 	size_t flen;
3721c1d255d3SCy Schubert 
3722c1d255d3SCy Schubert 	flen = (capab & 0xff00) ? 2 : 1;
3723c1d255d3SCy Schubert 	if (!capab)
3724c1d255d3SCy Schubert 		return; /* no supported extended RSN capabilities */
3725c1d255d3SCy Schubert 	if (wpabuf_tailroom(buf) < 2 + flen)
3726c1d255d3SCy Schubert 		return;
3727c1d255d3SCy Schubert 	capab |= flen - 1; /* bit 0-3 = Field length (n - 1) */
3728c1d255d3SCy Schubert 
3729c1d255d3SCy Schubert 	wpabuf_put_u8(buf, WLAN_EID_RSNX);
3730c1d255d3SCy Schubert 	wpabuf_put_u8(buf, flen);
3731c1d255d3SCy Schubert 	wpabuf_put_u8(buf, capab & 0x00ff);
3732c1d255d3SCy Schubert 	capab >>= 8;
3733c1d255d3SCy Schubert 	if (capab)
3734c1d255d3SCy Schubert 		wpabuf_put_u8(buf, capab);
3735c1d255d3SCy Schubert }
3736c1d255d3SCy Schubert 
3737c1d255d3SCy Schubert #endif /* CONFIG_PASN */
3738