xref: /freebsd/contrib/wpa/src/common/wpa_common.c (revision 641a6cfb86023499caafe26a4d821a0b885cf00b)
1 /*
2  * WPA/RSN - Shared functions for supplicant and authenticator
3  * Copyright (c) 2002-2008, Jouni Malinen <j@w1.fi>
4  *
5  * This program is free software; you can redistribute it and/or modify
6  * it under the terms of the GNU General Public License version 2 as
7  * published by the Free Software Foundation.
8  *
9  * Alternatively, this software may be distributed under the terms of BSD
10  * license.
11  *
12  * See README and COPYING for more details.
13  */
14 
15 #include "includes.h"
16 
17 #include "common.h"
18 #include "crypto/md5.h"
19 #include "crypto/sha1.h"
20 #include "crypto/sha256.h"
21 #include "crypto/aes_wrap.h"
22 #include "crypto/crypto.h"
23 #include "ieee802_11_defs.h"
24 #include "defs.h"
25 #include "wpa_common.h"
26 
27 
28 /**
29  * wpa_eapol_key_mic - Calculate EAPOL-Key MIC
30  * @key: EAPOL-Key Key Confirmation Key (KCK)
31  * @ver: Key descriptor version (WPA_KEY_INFO_TYPE_*)
32  * @buf: Pointer to the beginning of the EAPOL header (version field)
33  * @len: Length of the EAPOL frame (from EAPOL header to the end of the frame)
34  * @mic: Pointer to the buffer to which the EAPOL-Key MIC is written
35  * Returns: 0 on success, -1 on failure
36  *
37  * Calculate EAPOL-Key MIC for an EAPOL-Key packet. The EAPOL-Key MIC field has
38  * to be cleared (all zeroes) when calling this function.
39  *
40  * Note: 'IEEE Std 802.11i-2004 - 8.5.2 EAPOL-Key frames' has an error in the
41  * description of the Key MIC calculation. It includes packet data from the
42  * beginning of the EAPOL-Key header, not EAPOL header. This incorrect change
43  * happened during final editing of the standard and the correct behavior is
44  * defined in the last draft (IEEE 802.11i/D10).
45  */
46 int wpa_eapol_key_mic(const u8 *key, int ver, const u8 *buf, size_t len,
47 		      u8 *mic)
48 {
49 	u8 hash[SHA1_MAC_LEN];
50 
51 	switch (ver) {
52 	case WPA_KEY_INFO_TYPE_HMAC_MD5_RC4:
53 		return hmac_md5(key, 16, buf, len, mic);
54 	case WPA_KEY_INFO_TYPE_HMAC_SHA1_AES:
55 		if (hmac_sha1(key, 16, buf, len, hash))
56 			return -1;
57 		os_memcpy(mic, hash, MD5_MAC_LEN);
58 		break;
59 #if defined(CONFIG_IEEE80211R) || defined(CONFIG_IEEE80211W)
60 	case WPA_KEY_INFO_TYPE_AES_128_CMAC:
61 		return omac1_aes_128(key, buf, len, mic);
62 #endif /* CONFIG_IEEE80211R || CONFIG_IEEE80211W */
63 	default:
64 		return -1;
65 	}
66 
67 	return 0;
68 }
69 
70 
71 /**
72  * wpa_pmk_to_ptk - Calculate PTK from PMK, addresses, and nonces
73  * @pmk: Pairwise master key
74  * @pmk_len: Length of PMK
75  * @label: Label to use in derivation
76  * @addr1: AA or SA
77  * @addr2: SA or AA
78  * @nonce1: ANonce or SNonce
79  * @nonce2: SNonce or ANonce
80  * @ptk: Buffer for pairwise transient key
81  * @ptk_len: Length of PTK
82  * @use_sha256: Whether to use SHA256-based KDF
83  *
84  * IEEE Std 802.11i-2004 - 8.5.1.2 Pairwise key hierarchy
85  * PTK = PRF-X(PMK, "Pairwise key expansion",
86  *             Min(AA, SA) || Max(AA, SA) ||
87  *             Min(ANonce, SNonce) || Max(ANonce, SNonce))
88  *
89  * STK = PRF-X(SMK, "Peer key expansion",
90  *             Min(MAC_I, MAC_P) || Max(MAC_I, MAC_P) ||
91  *             Min(INonce, PNonce) || Max(INonce, PNonce))
92  */
93 void wpa_pmk_to_ptk(const u8 *pmk, size_t pmk_len, const char *label,
94 		    const u8 *addr1, const u8 *addr2,
95 		    const u8 *nonce1, const u8 *nonce2,
96 		    u8 *ptk, size_t ptk_len, int use_sha256)
97 {
98 	u8 data[2 * ETH_ALEN + 2 * WPA_NONCE_LEN];
99 
100 	if (os_memcmp(addr1, addr2, ETH_ALEN) < 0) {
101 		os_memcpy(data, addr1, ETH_ALEN);
102 		os_memcpy(data + ETH_ALEN, addr2, ETH_ALEN);
103 	} else {
104 		os_memcpy(data, addr2, ETH_ALEN);
105 		os_memcpy(data + ETH_ALEN, addr1, ETH_ALEN);
106 	}
107 
108 	if (os_memcmp(nonce1, nonce2, WPA_NONCE_LEN) < 0) {
109 		os_memcpy(data + 2 * ETH_ALEN, nonce1, WPA_NONCE_LEN);
110 		os_memcpy(data + 2 * ETH_ALEN + WPA_NONCE_LEN, nonce2,
111 			  WPA_NONCE_LEN);
112 	} else {
113 		os_memcpy(data + 2 * ETH_ALEN, nonce2, WPA_NONCE_LEN);
114 		os_memcpy(data + 2 * ETH_ALEN + WPA_NONCE_LEN, nonce1,
115 			  WPA_NONCE_LEN);
116 	}
117 
118 #ifdef CONFIG_IEEE80211W
119 	if (use_sha256)
120 		sha256_prf(pmk, pmk_len, label, data, sizeof(data),
121 			   ptk, ptk_len);
122 	else
123 #endif /* CONFIG_IEEE80211W */
124 		sha1_prf(pmk, pmk_len, label, data, sizeof(data), ptk,
125 			 ptk_len);
126 
127 	wpa_printf(MSG_DEBUG, "WPA: PTK derivation - A1=" MACSTR " A2=" MACSTR,
128 		   MAC2STR(addr1), MAC2STR(addr2));
129 	wpa_hexdump_key(MSG_DEBUG, "WPA: PMK", pmk, pmk_len);
130 	wpa_hexdump_key(MSG_DEBUG, "WPA: PTK", ptk, ptk_len);
131 }
132 
133 
134 #ifdef CONFIG_IEEE80211R
135 int wpa_ft_mic(const u8 *kck, const u8 *sta_addr, const u8 *ap_addr,
136 	       u8 transaction_seqnum, const u8 *mdie, size_t mdie_len,
137 	       const u8 *ftie, size_t ftie_len,
138 	       const u8 *rsnie, size_t rsnie_len,
139 	       const u8 *ric, size_t ric_len, u8 *mic)
140 {
141 	u8 *buf, *pos;
142 	size_t buf_len;
143 
144 	buf_len = 2 * ETH_ALEN + 1 + mdie_len + ftie_len + rsnie_len + ric_len;
145 	buf = os_malloc(buf_len);
146 	if (buf == NULL)
147 		return -1;
148 
149 	pos = buf;
150 	os_memcpy(pos, sta_addr, ETH_ALEN);
151 	pos += ETH_ALEN;
152 	os_memcpy(pos, ap_addr, ETH_ALEN);
153 	pos += ETH_ALEN;
154 	*pos++ = transaction_seqnum;
155 	if (rsnie) {
156 		os_memcpy(pos, rsnie, rsnie_len);
157 		pos += rsnie_len;
158 	}
159 	if (mdie) {
160 		os_memcpy(pos, mdie, mdie_len);
161 		pos += mdie_len;
162 	}
163 	if (ftie) {
164 		struct rsn_ftie *_ftie;
165 		os_memcpy(pos, ftie, ftie_len);
166 		if (ftie_len < 2 + sizeof(*_ftie)) {
167 			os_free(buf);
168 			return -1;
169 		}
170 		_ftie = (struct rsn_ftie *) (pos + 2);
171 		os_memset(_ftie->mic, 0, sizeof(_ftie->mic));
172 		pos += ftie_len;
173 	}
174 	if (ric) {
175 		os_memcpy(pos, ric, ric_len);
176 		pos += ric_len;
177 	}
178 
179 	wpa_hexdump(MSG_MSGDUMP, "FT: MIC data", buf, pos - buf);
180 	if (omac1_aes_128(kck, buf, pos - buf, mic)) {
181 		os_free(buf);
182 		return -1;
183 	}
184 
185 	os_free(buf);
186 
187 	return 0;
188 }
189 #endif /* CONFIG_IEEE80211R */
190 
191 
192 #ifndef CONFIG_NO_WPA2
193 static int rsn_selector_to_bitfield(const u8 *s)
194 {
195 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_NONE)
196 		return WPA_CIPHER_NONE;
197 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_WEP40)
198 		return WPA_CIPHER_WEP40;
199 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_TKIP)
200 		return WPA_CIPHER_TKIP;
201 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_CCMP)
202 		return WPA_CIPHER_CCMP;
203 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_WEP104)
204 		return WPA_CIPHER_WEP104;
205 #ifdef CONFIG_IEEE80211W
206 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_AES_128_CMAC)
207 		return WPA_CIPHER_AES_128_CMAC;
208 #endif /* CONFIG_IEEE80211W */
209 	return 0;
210 }
211 
212 
213 static int rsn_key_mgmt_to_bitfield(const u8 *s)
214 {
215 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_UNSPEC_802_1X)
216 		return WPA_KEY_MGMT_IEEE8021X;
217 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X)
218 		return WPA_KEY_MGMT_PSK;
219 #ifdef CONFIG_IEEE80211R
220 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_802_1X)
221 		return WPA_KEY_MGMT_FT_IEEE8021X;
222 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_PSK)
223 		return WPA_KEY_MGMT_FT_PSK;
224 #endif /* CONFIG_IEEE80211R */
225 #ifdef CONFIG_IEEE80211W
226 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_802_1X_SHA256)
227 		return WPA_KEY_MGMT_IEEE8021X_SHA256;
228 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_PSK_SHA256)
229 		return WPA_KEY_MGMT_PSK_SHA256;
230 #endif /* CONFIG_IEEE80211W */
231 	return 0;
232 }
233 #endif /* CONFIG_NO_WPA2 */
234 
235 
236 /**
237  * wpa_parse_wpa_ie_rsn - Parse RSN IE
238  * @rsn_ie: Buffer containing RSN IE
239  * @rsn_ie_len: RSN IE buffer length (including IE number and length octets)
240  * @data: Pointer to structure that will be filled in with parsed data
241  * Returns: 0 on success, <0 on failure
242  */
243 int wpa_parse_wpa_ie_rsn(const u8 *rsn_ie, size_t rsn_ie_len,
244 			 struct wpa_ie_data *data)
245 {
246 #ifndef CONFIG_NO_WPA2
247 	const struct rsn_ie_hdr *hdr;
248 	const u8 *pos;
249 	int left;
250 	int i, count;
251 
252 	os_memset(data, 0, sizeof(*data));
253 	data->proto = WPA_PROTO_RSN;
254 	data->pairwise_cipher = WPA_CIPHER_CCMP;
255 	data->group_cipher = WPA_CIPHER_CCMP;
256 	data->key_mgmt = WPA_KEY_MGMT_IEEE8021X;
257 	data->capabilities = 0;
258 	data->pmkid = NULL;
259 	data->num_pmkid = 0;
260 #ifdef CONFIG_IEEE80211W
261 	data->mgmt_group_cipher = WPA_CIPHER_AES_128_CMAC;
262 #else /* CONFIG_IEEE80211W */
263 	data->mgmt_group_cipher = 0;
264 #endif /* CONFIG_IEEE80211W */
265 
266 	if (rsn_ie_len == 0) {
267 		/* No RSN IE - fail silently */
268 		return -1;
269 	}
270 
271 	if (rsn_ie_len < sizeof(struct rsn_ie_hdr)) {
272 		wpa_printf(MSG_DEBUG, "%s: ie len too short %lu",
273 			   __func__, (unsigned long) rsn_ie_len);
274 		return -1;
275 	}
276 
277 	hdr = (const struct rsn_ie_hdr *) rsn_ie;
278 
279 	if (hdr->elem_id != WLAN_EID_RSN ||
280 	    hdr->len != rsn_ie_len - 2 ||
281 	    WPA_GET_LE16(hdr->version) != RSN_VERSION) {
282 		wpa_printf(MSG_DEBUG, "%s: malformed ie or unknown version",
283 			   __func__);
284 		return -2;
285 	}
286 
287 	pos = (const u8 *) (hdr + 1);
288 	left = rsn_ie_len - sizeof(*hdr);
289 
290 	if (left >= RSN_SELECTOR_LEN) {
291 		data->group_cipher = rsn_selector_to_bitfield(pos);
292 #ifdef CONFIG_IEEE80211W
293 		if (data->group_cipher == WPA_CIPHER_AES_128_CMAC) {
294 			wpa_printf(MSG_DEBUG, "%s: AES-128-CMAC used as group "
295 				   "cipher", __func__);
296 			return -1;
297 		}
298 #endif /* CONFIG_IEEE80211W */
299 		pos += RSN_SELECTOR_LEN;
300 		left -= RSN_SELECTOR_LEN;
301 	} else if (left > 0) {
302 		wpa_printf(MSG_DEBUG, "%s: ie length mismatch, %u too much",
303 			   __func__, left);
304 		return -3;
305 	}
306 
307 	if (left >= 2) {
308 		data->pairwise_cipher = 0;
309 		count = WPA_GET_LE16(pos);
310 		pos += 2;
311 		left -= 2;
312 		if (count == 0 || left < count * RSN_SELECTOR_LEN) {
313 			wpa_printf(MSG_DEBUG, "%s: ie count botch (pairwise), "
314 				   "count %u left %u", __func__, count, left);
315 			return -4;
316 		}
317 		for (i = 0; i < count; i++) {
318 			data->pairwise_cipher |= rsn_selector_to_bitfield(pos);
319 			pos += RSN_SELECTOR_LEN;
320 			left -= RSN_SELECTOR_LEN;
321 		}
322 #ifdef CONFIG_IEEE80211W
323 		if (data->pairwise_cipher & WPA_CIPHER_AES_128_CMAC) {
324 			wpa_printf(MSG_DEBUG, "%s: AES-128-CMAC used as "
325 				   "pairwise cipher", __func__);
326 			return -1;
327 		}
328 #endif /* CONFIG_IEEE80211W */
329 	} else if (left == 1) {
330 		wpa_printf(MSG_DEBUG, "%s: ie too short (for key mgmt)",
331 			   __func__);
332 		return -5;
333 	}
334 
335 	if (left >= 2) {
336 		data->key_mgmt = 0;
337 		count = WPA_GET_LE16(pos);
338 		pos += 2;
339 		left -= 2;
340 		if (count == 0 || left < count * RSN_SELECTOR_LEN) {
341 			wpa_printf(MSG_DEBUG, "%s: ie count botch (key mgmt), "
342 				   "count %u left %u", __func__, count, left);
343 			return -6;
344 		}
345 		for (i = 0; i < count; i++) {
346 			data->key_mgmt |= rsn_key_mgmt_to_bitfield(pos);
347 			pos += RSN_SELECTOR_LEN;
348 			left -= RSN_SELECTOR_LEN;
349 		}
350 	} else if (left == 1) {
351 		wpa_printf(MSG_DEBUG, "%s: ie too short (for capabilities)",
352 			   __func__);
353 		return -7;
354 	}
355 
356 	if (left >= 2) {
357 		data->capabilities = WPA_GET_LE16(pos);
358 		pos += 2;
359 		left -= 2;
360 	}
361 
362 	if (left >= 2) {
363 		data->num_pmkid = WPA_GET_LE16(pos);
364 		pos += 2;
365 		left -= 2;
366 		if (left < (int) data->num_pmkid * PMKID_LEN) {
367 			wpa_printf(MSG_DEBUG, "%s: PMKID underflow "
368 				   "(num_pmkid=%lu left=%d)",
369 				   __func__, (unsigned long) data->num_pmkid,
370 				   left);
371 			data->num_pmkid = 0;
372 			return -9;
373 		} else {
374 			data->pmkid = pos;
375 			pos += data->num_pmkid * PMKID_LEN;
376 			left -= data->num_pmkid * PMKID_LEN;
377 		}
378 	}
379 
380 #ifdef CONFIG_IEEE80211W
381 	if (left >= 4) {
382 		data->mgmt_group_cipher = rsn_selector_to_bitfield(pos);
383 		if (data->mgmt_group_cipher != WPA_CIPHER_AES_128_CMAC) {
384 			wpa_printf(MSG_DEBUG, "%s: Unsupported management "
385 				   "group cipher 0x%x", __func__,
386 				   data->mgmt_group_cipher);
387 			return -10;
388 		}
389 		pos += RSN_SELECTOR_LEN;
390 		left -= RSN_SELECTOR_LEN;
391 	}
392 #endif /* CONFIG_IEEE80211W */
393 
394 	if (left > 0) {
395 		wpa_printf(MSG_DEBUG, "%s: ie has %u trailing bytes - ignored",
396 			   __func__, left);
397 	}
398 
399 	return 0;
400 #else /* CONFIG_NO_WPA2 */
401 	return -1;
402 #endif /* CONFIG_NO_WPA2 */
403 }
404 
405 
406 #ifdef CONFIG_IEEE80211R
407 
408 /**
409  * wpa_derive_pmk_r0 - Derive PMK-R0 and PMKR0Name
410  *
411  * IEEE Std 802.11r-2008 - 8.5.1.5.3
412  */
413 void wpa_derive_pmk_r0(const u8 *xxkey, size_t xxkey_len,
414 		       const u8 *ssid, size_t ssid_len,
415 		       const u8 *mdid, const u8 *r0kh_id, size_t r0kh_id_len,
416 		       const u8 *s0kh_id, u8 *pmk_r0, u8 *pmk_r0_name)
417 {
418 	u8 buf[1 + WPA_MAX_SSID_LEN + MOBILITY_DOMAIN_ID_LEN + 1 +
419 	       FT_R0KH_ID_MAX_LEN + ETH_ALEN];
420 	u8 *pos, r0_key_data[48], hash[32];
421 	const u8 *addr[2];
422 	size_t len[2];
423 
424 	/*
425 	 * R0-Key-Data = KDF-384(XXKey, "FT-R0",
426 	 *                       SSIDlength || SSID || MDID || R0KHlength ||
427 	 *                       R0KH-ID || S0KH-ID)
428 	 * XXKey is either the second 256 bits of MSK or PSK.
429 	 * PMK-R0 = L(R0-Key-Data, 0, 256)
430 	 * PMK-R0Name-Salt = L(R0-Key-Data, 256, 128)
431 	 */
432 	if (ssid_len > WPA_MAX_SSID_LEN || r0kh_id_len > FT_R0KH_ID_MAX_LEN)
433 		return;
434 	pos = buf;
435 	*pos++ = ssid_len;
436 	os_memcpy(pos, ssid, ssid_len);
437 	pos += ssid_len;
438 	os_memcpy(pos, mdid, MOBILITY_DOMAIN_ID_LEN);
439 	pos += MOBILITY_DOMAIN_ID_LEN;
440 	*pos++ = r0kh_id_len;
441 	os_memcpy(pos, r0kh_id, r0kh_id_len);
442 	pos += r0kh_id_len;
443 	os_memcpy(pos, s0kh_id, ETH_ALEN);
444 	pos += ETH_ALEN;
445 
446 	sha256_prf(xxkey, xxkey_len, "FT-R0", buf, pos - buf,
447 		   r0_key_data, sizeof(r0_key_data));
448 	os_memcpy(pmk_r0, r0_key_data, PMK_LEN);
449 
450 	/*
451 	 * PMKR0Name = Truncate-128(SHA-256("FT-R0N" || PMK-R0Name-Salt)
452 	 */
453 	addr[0] = (const u8 *) "FT-R0N";
454 	len[0] = 6;
455 	addr[1] = r0_key_data + PMK_LEN;
456 	len[1] = 16;
457 
458 	sha256_vector(2, addr, len, hash);
459 	os_memcpy(pmk_r0_name, hash, WPA_PMK_NAME_LEN);
460 }
461 
462 
463 /**
464  * wpa_derive_pmk_r1_name - Derive PMKR1Name
465  *
466  * IEEE Std 802.11r-2008 - 8.5.1.5.4
467  */
468 void wpa_derive_pmk_r1_name(const u8 *pmk_r0_name, const u8 *r1kh_id,
469 			    const u8 *s1kh_id, u8 *pmk_r1_name)
470 {
471 	u8 hash[32];
472 	const u8 *addr[4];
473 	size_t len[4];
474 
475 	/*
476 	 * PMKR1Name = Truncate-128(SHA-256("FT-R1N" || PMKR0Name ||
477 	 *                                  R1KH-ID || S1KH-ID))
478 	 */
479 	addr[0] = (const u8 *) "FT-R1N";
480 	len[0] = 6;
481 	addr[1] = pmk_r0_name;
482 	len[1] = WPA_PMK_NAME_LEN;
483 	addr[2] = r1kh_id;
484 	len[2] = FT_R1KH_ID_LEN;
485 	addr[3] = s1kh_id;
486 	len[3] = ETH_ALEN;
487 
488 	sha256_vector(4, addr, len, hash);
489 	os_memcpy(pmk_r1_name, hash, WPA_PMK_NAME_LEN);
490 }
491 
492 
493 /**
494  * wpa_derive_pmk_r1 - Derive PMK-R1 and PMKR1Name from PMK-R0
495  *
496  * IEEE Std 802.11r-2008 - 8.5.1.5.4
497  */
498 void wpa_derive_pmk_r1(const u8 *pmk_r0, const u8 *pmk_r0_name,
499 		       const u8 *r1kh_id, const u8 *s1kh_id,
500 		       u8 *pmk_r1, u8 *pmk_r1_name)
501 {
502 	u8 buf[FT_R1KH_ID_LEN + ETH_ALEN];
503 	u8 *pos;
504 
505 	/* PMK-R1 = KDF-256(PMK-R0, "FT-R1", R1KH-ID || S1KH-ID) */
506 	pos = buf;
507 	os_memcpy(pos, r1kh_id, FT_R1KH_ID_LEN);
508 	pos += FT_R1KH_ID_LEN;
509 	os_memcpy(pos, s1kh_id, ETH_ALEN);
510 	pos += ETH_ALEN;
511 
512 	sha256_prf(pmk_r0, PMK_LEN, "FT-R1", buf, pos - buf, pmk_r1, PMK_LEN);
513 
514 	wpa_derive_pmk_r1_name(pmk_r0_name, r1kh_id, s1kh_id, pmk_r1_name);
515 }
516 
517 
518 /**
519  * wpa_pmk_r1_to_ptk - Derive PTK and PTKName from PMK-R1
520  *
521  * IEEE Std 802.11r-2008 - 8.5.1.5.5
522  */
523 void wpa_pmk_r1_to_ptk(const u8 *pmk_r1, const u8 *snonce, const u8 *anonce,
524 		       const u8 *sta_addr, const u8 *bssid,
525 		       const u8 *pmk_r1_name,
526 		       u8 *ptk, size_t ptk_len, u8 *ptk_name)
527 {
528 	u8 buf[2 * WPA_NONCE_LEN + 2 * ETH_ALEN];
529 	u8 *pos, hash[32];
530 	const u8 *addr[6];
531 	size_t len[6];
532 
533 	/*
534 	 * PTK = KDF-PTKLen(PMK-R1, "FT-PTK", SNonce || ANonce ||
535 	 *                  BSSID || STA-ADDR)
536 	 */
537 	pos = buf;
538 	os_memcpy(pos, snonce, WPA_NONCE_LEN);
539 	pos += WPA_NONCE_LEN;
540 	os_memcpy(pos, anonce, WPA_NONCE_LEN);
541 	pos += WPA_NONCE_LEN;
542 	os_memcpy(pos, bssid, ETH_ALEN);
543 	pos += ETH_ALEN;
544 	os_memcpy(pos, sta_addr, ETH_ALEN);
545 	pos += ETH_ALEN;
546 
547 	sha256_prf(pmk_r1, PMK_LEN, "FT-PTK", buf, pos - buf, ptk, ptk_len);
548 
549 	/*
550 	 * PTKName = Truncate-128(SHA-256(PMKR1Name || "FT-PTKN" || SNonce ||
551 	 *                                ANonce || BSSID || STA-ADDR))
552 	 */
553 	addr[0] = pmk_r1_name;
554 	len[0] = WPA_PMK_NAME_LEN;
555 	addr[1] = (const u8 *) "FT-PTKN";
556 	len[1] = 7;
557 	addr[2] = snonce;
558 	len[2] = WPA_NONCE_LEN;
559 	addr[3] = anonce;
560 	len[3] = WPA_NONCE_LEN;
561 	addr[4] = bssid;
562 	len[4] = ETH_ALEN;
563 	addr[5] = sta_addr;
564 	len[5] = ETH_ALEN;
565 
566 	sha256_vector(6, addr, len, hash);
567 	os_memcpy(ptk_name, hash, WPA_PMK_NAME_LEN);
568 }
569 
570 #endif /* CONFIG_IEEE80211R */
571 
572 
573 /**
574  * rsn_pmkid - Calculate PMK identifier
575  * @pmk: Pairwise master key
576  * @pmk_len: Length of pmk in bytes
577  * @aa: Authenticator address
578  * @spa: Supplicant address
579  * @pmkid: Buffer for PMKID
580  * @use_sha256: Whether to use SHA256-based KDF
581  *
582  * IEEE Std 802.11i-2004 - 8.5.1.2 Pairwise key hierarchy
583  * PMKID = HMAC-SHA1-128(PMK, "PMK Name" || AA || SPA)
584  */
585 void rsn_pmkid(const u8 *pmk, size_t pmk_len, const u8 *aa, const u8 *spa,
586 	       u8 *pmkid, int use_sha256)
587 {
588 	char *title = "PMK Name";
589 	const u8 *addr[3];
590 	const size_t len[3] = { 8, ETH_ALEN, ETH_ALEN };
591 	unsigned char hash[SHA256_MAC_LEN];
592 
593 	addr[0] = (u8 *) title;
594 	addr[1] = aa;
595 	addr[2] = spa;
596 
597 #ifdef CONFIG_IEEE80211W
598 	if (use_sha256)
599 		hmac_sha256_vector(pmk, pmk_len, 3, addr, len, hash);
600 	else
601 #endif /* CONFIG_IEEE80211W */
602 		hmac_sha1_vector(pmk, pmk_len, 3, addr, len, hash);
603 	os_memcpy(pmkid, hash, PMKID_LEN);
604 }
605 
606 
607 /**
608  * wpa_cipher_txt - Convert cipher suite to a text string
609  * @cipher: Cipher suite (WPA_CIPHER_* enum)
610  * Returns: Pointer to a text string of the cipher suite name
611  */
612 const char * wpa_cipher_txt(int cipher)
613 {
614 	switch (cipher) {
615 	case WPA_CIPHER_NONE:
616 		return "NONE";
617 	case WPA_CIPHER_WEP40:
618 		return "WEP-40";
619 	case WPA_CIPHER_WEP104:
620 		return "WEP-104";
621 	case WPA_CIPHER_TKIP:
622 		return "TKIP";
623 	case WPA_CIPHER_CCMP:
624 		return "CCMP";
625 	case WPA_CIPHER_CCMP | WPA_CIPHER_TKIP:
626 		return "CCMP+TKIP";
627 	default:
628 		return "UNKNOWN";
629 	}
630 }
631 
632 
633 /**
634  * wpa_key_mgmt_txt - Convert key management suite to a text string
635  * @key_mgmt: Key management suite (WPA_KEY_MGMT_* enum)
636  * @proto: WPA/WPA2 version (WPA_PROTO_*)
637  * Returns: Pointer to a text string of the key management suite name
638  */
639 const char * wpa_key_mgmt_txt(int key_mgmt, int proto)
640 {
641 	switch (key_mgmt) {
642 	case WPA_KEY_MGMT_IEEE8021X:
643 		if (proto == (WPA_PROTO_RSN | WPA_PROTO_WPA))
644 			return "WPA2+WPA/IEEE 802.1X/EAP";
645 		return proto == WPA_PROTO_RSN ?
646 			"WPA2/IEEE 802.1X/EAP" : "WPA/IEEE 802.1X/EAP";
647 	case WPA_KEY_MGMT_PSK:
648 		if (proto == (WPA_PROTO_RSN | WPA_PROTO_WPA))
649 			return "WPA2-PSK+WPA-PSK";
650 		return proto == WPA_PROTO_RSN ?
651 			"WPA2-PSK" : "WPA-PSK";
652 	case WPA_KEY_MGMT_NONE:
653 		return "NONE";
654 	case WPA_KEY_MGMT_IEEE8021X_NO_WPA:
655 		return "IEEE 802.1X (no WPA)";
656 #ifdef CONFIG_IEEE80211R
657 	case WPA_KEY_MGMT_FT_IEEE8021X:
658 		return "FT-EAP";
659 	case WPA_KEY_MGMT_FT_PSK:
660 		return "FT-PSK";
661 #endif /* CONFIG_IEEE80211R */
662 #ifdef CONFIG_IEEE80211W
663 	case WPA_KEY_MGMT_IEEE8021X_SHA256:
664 		return "WPA2-EAP-SHA256";
665 	case WPA_KEY_MGMT_PSK_SHA256:
666 		return "WPA2-PSK-SHA256";
667 #endif /* CONFIG_IEEE80211W */
668 	default:
669 		return "UNKNOWN";
670 	}
671 }
672 
673 
674 int wpa_compare_rsn_ie(int ft_initial_assoc,
675 		       const u8 *ie1, size_t ie1len,
676 		       const u8 *ie2, size_t ie2len)
677 {
678 	if (ie1 == NULL || ie2 == NULL)
679 		return -1;
680 
681 	if (ie1len == ie2len && os_memcmp(ie1, ie2, ie1len) == 0)
682 		return 0; /* identical IEs */
683 
684 #ifdef CONFIG_IEEE80211R
685 	if (ft_initial_assoc) {
686 		struct wpa_ie_data ie1d, ie2d;
687 		/*
688 		 * The PMKID-List in RSN IE is different between Beacon/Probe
689 		 * Response/(Re)Association Request frames and EAPOL-Key
690 		 * messages in FT initial mobility domain association. Allow
691 		 * for this, but verify that other parts of the RSN IEs are
692 		 * identical.
693 		 */
694 		if (wpa_parse_wpa_ie_rsn(ie1, ie1len, &ie1d) < 0 ||
695 		    wpa_parse_wpa_ie_rsn(ie2, ie2len, &ie2d) < 0)
696 			return -1;
697 		if (ie1d.proto == ie2d.proto &&
698 		    ie1d.pairwise_cipher == ie2d.pairwise_cipher &&
699 		    ie1d.group_cipher == ie2d.group_cipher &&
700 		    ie1d.key_mgmt == ie2d.key_mgmt &&
701 		    ie1d.capabilities == ie2d.capabilities &&
702 		    ie1d.mgmt_group_cipher == ie2d.mgmt_group_cipher)
703 			return 0;
704 	}
705 #endif /* CONFIG_IEEE80211R */
706 
707 	return -1;
708 }
709 
710 
711 #ifdef CONFIG_IEEE80211R
712 int wpa_insert_pmkid(u8 *ies, size_t ies_len, const u8 *pmkid)
713 {
714 	u8 *start, *end, *rpos, *rend;
715 	int added = 0;
716 
717 	start = ies;
718 	end = ies + ies_len;
719 
720 	while (start < end) {
721 		if (*start == WLAN_EID_RSN)
722 			break;
723 		start += 2 + start[1];
724 	}
725 	if (start >= end) {
726 		wpa_printf(MSG_ERROR, "FT: Could not find RSN IE in "
727 			   "IEs data");
728 		return -1;
729 	}
730 	wpa_hexdump(MSG_DEBUG, "FT: RSN IE before modification",
731 		    start, 2 + start[1]);
732 
733 	/* Find start of PMKID-Count */
734 	rpos = start + 2;
735 	rend = rpos + start[1];
736 
737 	/* Skip Version and Group Data Cipher Suite */
738 	rpos += 2 + 4;
739 	/* Skip Pairwise Cipher Suite Count and List */
740 	rpos += 2 + WPA_GET_LE16(rpos) * RSN_SELECTOR_LEN;
741 	/* Skip AKM Suite Count and List */
742 	rpos += 2 + WPA_GET_LE16(rpos) * RSN_SELECTOR_LEN;
743 
744 	if (rpos == rend) {
745 		/* Add RSN Capabilities */
746 		os_memmove(rpos + 2, rpos, end - rpos);
747 		*rpos++ = 0;
748 		*rpos++ = 0;
749 	} else {
750 		/* Skip RSN Capabilities */
751 		rpos += 2;
752 		if (rpos > rend) {
753 			wpa_printf(MSG_ERROR, "FT: Could not parse RSN IE in "
754 				   "IEs data");
755 			return -1;
756 		}
757 	}
758 
759 	if (rpos == rend) {
760 		/* No PMKID-Count field included; add it */
761 		os_memmove(rpos + 2 + PMKID_LEN, rpos, end - rpos);
762 		WPA_PUT_LE16(rpos, 1);
763 		rpos += 2;
764 		os_memcpy(rpos, pmkid, PMKID_LEN);
765 		added += 2 + PMKID_LEN;
766 		start[1] += 2 + PMKID_LEN;
767 	} else {
768 		/* PMKID-Count was included; use it */
769 		if (WPA_GET_LE16(rpos) != 0) {
770 			wpa_printf(MSG_ERROR, "FT: Unexpected PMKID "
771 				   "in RSN IE in EAPOL-Key data");
772 			return -1;
773 		}
774 		WPA_PUT_LE16(rpos, 1);
775 		rpos += 2;
776 		os_memmove(rpos + PMKID_LEN, rpos, end - rpos);
777 		os_memcpy(rpos, pmkid, PMKID_LEN);
778 		added += PMKID_LEN;
779 		start[1] += PMKID_LEN;
780 	}
781 
782 	wpa_hexdump(MSG_DEBUG, "FT: RSN IE after modification "
783 		    "(PMKID inserted)", start, 2 + start[1]);
784 
785 	return added;
786 }
787 #endif /* CONFIG_IEEE80211R */
788