1 /*
2  * WPA/RSN - Shared functions for supplicant and authenticator
3  * Copyright (c) 2002-2013, Jouni Malinen <j@w1.fi>
4  *
5  * This software may be distributed under the terms of the BSD license.
6  * See README for more details.
7  */
8 
9 #include "includes.h"
10 
11 #include "common.h"
12 #include "crypto/md5.h"
13 #include "crypto/sha1.h"
14 #include "crypto/sha256.h"
15 #include "crypto/aes_wrap.h"
16 #include "crypto/crypto.h"
17 #include "ieee802_11_defs.h"
18 #include "defs.h"
19 #include "wpa_common.h"
20 
21 
22 /**
23  * wpa_eapol_key_mic - Calculate EAPOL-Key MIC
24  * @key: EAPOL-Key Key Confirmation Key (KCK)
25  * @ver: Key descriptor version (WPA_KEY_INFO_TYPE_*)
26  * @buf: Pointer to the beginning of the EAPOL header (version field)
27  * @len: Length of the EAPOL frame (from EAPOL header to the end of the frame)
28  * @mic: Pointer to the buffer to which the EAPOL-Key MIC is written
29  * Returns: 0 on success, -1 on failure
30  *
31  * Calculate EAPOL-Key MIC for an EAPOL-Key packet. The EAPOL-Key MIC field has
32  * to be cleared (all zeroes) when calling this function.
33  *
34  * Note: 'IEEE Std 802.11i-2004 - 8.5.2 EAPOL-Key frames' has an error in the
35  * description of the Key MIC calculation. It includes packet data from the
36  * beginning of the EAPOL-Key header, not EAPOL header. This incorrect change
37  * happened during final editing of the standard and the correct behavior is
38  * defined in the last draft (IEEE 802.11i/D10).
39  */
40 int wpa_eapol_key_mic(const u8 *key, int ver, const u8 *buf, size_t len,
41 		      u8 *mic)
42 {
43 	u8 hash[SHA1_MAC_LEN];
44 
45 	switch (ver) {
46 #ifndef CONFIG_FIPS
47 	case WPA_KEY_INFO_TYPE_HMAC_MD5_RC4:
48 		return hmac_md5(key, 16, buf, len, mic);
49 #endif /* CONFIG_FIPS */
50 	case WPA_KEY_INFO_TYPE_HMAC_SHA1_AES:
51 		if (hmac_sha1(key, 16, buf, len, hash))
52 			return -1;
53 		os_memcpy(mic, hash, MD5_MAC_LEN);
54 		break;
55 #if defined(CONFIG_IEEE80211R) || defined(CONFIG_IEEE80211W)
56 	case WPA_KEY_INFO_TYPE_AES_128_CMAC:
57 		return omac1_aes_128(key, buf, len, mic);
58 #endif /* CONFIG_IEEE80211R || CONFIG_IEEE80211W */
59 	default:
60 		return -1;
61 	}
62 
63 	return 0;
64 }
65 
66 
67 /**
68  * wpa_pmk_to_ptk - Calculate PTK from PMK, addresses, and nonces
69  * @pmk: Pairwise master key
70  * @pmk_len: Length of PMK
71  * @label: Label to use in derivation
72  * @addr1: AA or SA
73  * @addr2: SA or AA
74  * @nonce1: ANonce or SNonce
75  * @nonce2: SNonce or ANonce
76  * @ptk: Buffer for pairwise transient key
77  * @ptk_len: Length of PTK
78  * @use_sha256: Whether to use SHA256-based KDF
79  *
80  * IEEE Std 802.11i-2004 - 8.5.1.2 Pairwise key hierarchy
81  * PTK = PRF-X(PMK, "Pairwise key expansion",
82  *             Min(AA, SA) || Max(AA, SA) ||
83  *             Min(ANonce, SNonce) || Max(ANonce, SNonce))
84  *
85  * STK = PRF-X(SMK, "Peer key expansion",
86  *             Min(MAC_I, MAC_P) || Max(MAC_I, MAC_P) ||
87  *             Min(INonce, PNonce) || Max(INonce, PNonce))
88  */
89 void wpa_pmk_to_ptk(const u8 *pmk, size_t pmk_len, const char *label,
90 		    const u8 *addr1, const u8 *addr2,
91 		    const u8 *nonce1, const u8 *nonce2,
92 		    u8 *ptk, size_t ptk_len, int use_sha256)
93 {
94 	u8 data[2 * ETH_ALEN + 2 * WPA_NONCE_LEN];
95 
96 	if (os_memcmp(addr1, addr2, ETH_ALEN) < 0) {
97 		os_memcpy(data, addr1, ETH_ALEN);
98 		os_memcpy(data + ETH_ALEN, addr2, ETH_ALEN);
99 	} else {
100 		os_memcpy(data, addr2, ETH_ALEN);
101 		os_memcpy(data + ETH_ALEN, addr1, ETH_ALEN);
102 	}
103 
104 	if (os_memcmp(nonce1, nonce2, WPA_NONCE_LEN) < 0) {
105 		os_memcpy(data + 2 * ETH_ALEN, nonce1, WPA_NONCE_LEN);
106 		os_memcpy(data + 2 * ETH_ALEN + WPA_NONCE_LEN, nonce2,
107 			  WPA_NONCE_LEN);
108 	} else {
109 		os_memcpy(data + 2 * ETH_ALEN, nonce2, WPA_NONCE_LEN);
110 		os_memcpy(data + 2 * ETH_ALEN + WPA_NONCE_LEN, nonce1,
111 			  WPA_NONCE_LEN);
112 	}
113 
114 #ifdef CONFIG_IEEE80211W
115 	if (use_sha256)
116 		sha256_prf(pmk, pmk_len, label, data, sizeof(data),
117 			   ptk, ptk_len);
118 	else
119 #endif /* CONFIG_IEEE80211W */
120 		sha1_prf(pmk, pmk_len, label, data, sizeof(data), ptk,
121 			 ptk_len);
122 
123 	wpa_printf(MSG_DEBUG, "WPA: PTK derivation - A1=" MACSTR " A2=" MACSTR,
124 		   MAC2STR(addr1), MAC2STR(addr2));
125 	wpa_hexdump(MSG_DEBUG, "WPA: Nonce1", nonce1, WPA_NONCE_LEN);
126 	wpa_hexdump(MSG_DEBUG, "WPA: Nonce2", nonce2, WPA_NONCE_LEN);
127 	wpa_hexdump_key(MSG_DEBUG, "WPA: PMK", pmk, pmk_len);
128 	wpa_hexdump_key(MSG_DEBUG, "WPA: PTK", ptk, ptk_len);
129 }
130 
131 
132 #ifdef CONFIG_IEEE80211R
133 int wpa_ft_mic(const u8 *kck, const u8 *sta_addr, const u8 *ap_addr,
134 	       u8 transaction_seqnum, const u8 *mdie, size_t mdie_len,
135 	       const u8 *ftie, size_t ftie_len,
136 	       const u8 *rsnie, size_t rsnie_len,
137 	       const u8 *ric, size_t ric_len, u8 *mic)
138 {
139 	u8 *buf, *pos;
140 	size_t buf_len;
141 
142 	buf_len = 2 * ETH_ALEN + 1 + mdie_len + ftie_len + rsnie_len + ric_len;
143 	buf = os_malloc(buf_len);
144 	if (buf == NULL)
145 		return -1;
146 
147 	pos = buf;
148 	os_memcpy(pos, sta_addr, ETH_ALEN);
149 	pos += ETH_ALEN;
150 	os_memcpy(pos, ap_addr, ETH_ALEN);
151 	pos += ETH_ALEN;
152 	*pos++ = transaction_seqnum;
153 	if (rsnie) {
154 		os_memcpy(pos, rsnie, rsnie_len);
155 		pos += rsnie_len;
156 	}
157 	if (mdie) {
158 		os_memcpy(pos, mdie, mdie_len);
159 		pos += mdie_len;
160 	}
161 	if (ftie) {
162 		struct rsn_ftie *_ftie;
163 		os_memcpy(pos, ftie, ftie_len);
164 		if (ftie_len < 2 + sizeof(*_ftie)) {
165 			os_free(buf);
166 			return -1;
167 		}
168 		_ftie = (struct rsn_ftie *) (pos + 2);
169 		os_memset(_ftie->mic, 0, sizeof(_ftie->mic));
170 		pos += ftie_len;
171 	}
172 	if (ric) {
173 		os_memcpy(pos, ric, ric_len);
174 		pos += ric_len;
175 	}
176 
177 	wpa_hexdump(MSG_MSGDUMP, "FT: MIC data", buf, pos - buf);
178 	if (omac1_aes_128(kck, buf, pos - buf, mic)) {
179 		os_free(buf);
180 		return -1;
181 	}
182 
183 	os_free(buf);
184 
185 	return 0;
186 }
187 
188 
189 static int wpa_ft_parse_ftie(const u8 *ie, size_t ie_len,
190 			     struct wpa_ft_ies *parse)
191 {
192 	const u8 *end, *pos;
193 
194 	parse->ftie = ie;
195 	parse->ftie_len = ie_len;
196 
197 	pos = ie + sizeof(struct rsn_ftie);
198 	end = ie + ie_len;
199 
200 	while (pos + 2 <= end && pos + 2 + pos[1] <= end) {
201 		switch (pos[0]) {
202 		case FTIE_SUBELEM_R1KH_ID:
203 			if (pos[1] != FT_R1KH_ID_LEN) {
204 				wpa_printf(MSG_DEBUG, "FT: Invalid R1KH-ID "
205 					   "length in FTIE: %d", pos[1]);
206 				return -1;
207 			}
208 			parse->r1kh_id = pos + 2;
209 			break;
210 		case FTIE_SUBELEM_GTK:
211 			parse->gtk = pos + 2;
212 			parse->gtk_len = pos[1];
213 			break;
214 		case FTIE_SUBELEM_R0KH_ID:
215 			if (pos[1] < 1 || pos[1] > FT_R0KH_ID_MAX_LEN) {
216 				wpa_printf(MSG_DEBUG, "FT: Invalid R0KH-ID "
217 					   "length in FTIE: %d", pos[1]);
218 				return -1;
219 			}
220 			parse->r0kh_id = pos + 2;
221 			parse->r0kh_id_len = pos[1];
222 			break;
223 #ifdef CONFIG_IEEE80211W
224 		case FTIE_SUBELEM_IGTK:
225 			parse->igtk = pos + 2;
226 			parse->igtk_len = pos[1];
227 			break;
228 #endif /* CONFIG_IEEE80211W */
229 		}
230 
231 		pos += 2 + pos[1];
232 	}
233 
234 	return 0;
235 }
236 
237 
238 int wpa_ft_parse_ies(const u8 *ies, size_t ies_len,
239 		     struct wpa_ft_ies *parse)
240 {
241 	const u8 *end, *pos;
242 	struct wpa_ie_data data;
243 	int ret;
244 	const struct rsn_ftie *ftie;
245 	int prot_ie_count = 0;
246 
247 	os_memset(parse, 0, sizeof(*parse));
248 	if (ies == NULL)
249 		return 0;
250 
251 	pos = ies;
252 	end = ies + ies_len;
253 	while (pos + 2 <= end && pos + 2 + pos[1] <= end) {
254 		switch (pos[0]) {
255 		case WLAN_EID_RSN:
256 			parse->rsn = pos + 2;
257 			parse->rsn_len = pos[1];
258 			ret = wpa_parse_wpa_ie_rsn(parse->rsn - 2,
259 						   parse->rsn_len + 2,
260 						   &data);
261 			if (ret < 0) {
262 				wpa_printf(MSG_DEBUG, "FT: Failed to parse "
263 					   "RSN IE: %d", ret);
264 				return -1;
265 			}
266 			if (data.num_pmkid == 1 && data.pmkid)
267 				parse->rsn_pmkid = data.pmkid;
268 			break;
269 		case WLAN_EID_MOBILITY_DOMAIN:
270 			parse->mdie = pos + 2;
271 			parse->mdie_len = pos[1];
272 			break;
273 		case WLAN_EID_FAST_BSS_TRANSITION:
274 			if (pos[1] < sizeof(*ftie))
275 				return -1;
276 			ftie = (const struct rsn_ftie *) (pos + 2);
277 			prot_ie_count = ftie->mic_control[1];
278 			if (wpa_ft_parse_ftie(pos + 2, pos[1], parse) < 0)
279 				return -1;
280 			break;
281 		case WLAN_EID_TIMEOUT_INTERVAL:
282 			parse->tie = pos + 2;
283 			parse->tie_len = pos[1];
284 			break;
285 		case WLAN_EID_RIC_DATA:
286 			if (parse->ric == NULL)
287 				parse->ric = pos;
288 			break;
289 		}
290 
291 		pos += 2 + pos[1];
292 	}
293 
294 	if (prot_ie_count == 0)
295 		return 0; /* no MIC */
296 
297 	/*
298 	 * Check that the protected IE count matches with IEs included in the
299 	 * frame.
300 	 */
301 	if (parse->rsn)
302 		prot_ie_count--;
303 	if (parse->mdie)
304 		prot_ie_count--;
305 	if (parse->ftie)
306 		prot_ie_count--;
307 	if (prot_ie_count < 0) {
308 		wpa_printf(MSG_DEBUG, "FT: Some required IEs not included in "
309 			   "the protected IE count");
310 		return -1;
311 	}
312 
313 	if (prot_ie_count == 0 && parse->ric) {
314 		wpa_printf(MSG_DEBUG, "FT: RIC IE(s) in the frame, but not "
315 			   "included in protected IE count");
316 		return -1;
317 	}
318 
319 	/* Determine the end of the RIC IE(s) */
320 	pos = parse->ric;
321 	while (pos && pos + 2 <= end && pos + 2 + pos[1] <= end &&
322 	       prot_ie_count) {
323 		prot_ie_count--;
324 		pos += 2 + pos[1];
325 	}
326 	parse->ric_len = pos - parse->ric;
327 	if (prot_ie_count) {
328 		wpa_printf(MSG_DEBUG, "FT: %d protected IEs missing from "
329 			   "frame", (int) prot_ie_count);
330 		return -1;
331 	}
332 
333 	return 0;
334 }
335 #endif /* CONFIG_IEEE80211R */
336 
337 
338 static int rsn_selector_to_bitfield(const u8 *s)
339 {
340 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_NONE)
341 		return WPA_CIPHER_NONE;
342 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_WEP40)
343 		return WPA_CIPHER_WEP40;
344 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_TKIP)
345 		return WPA_CIPHER_TKIP;
346 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_CCMP)
347 		return WPA_CIPHER_CCMP;
348 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_WEP104)
349 		return WPA_CIPHER_WEP104;
350 #ifdef CONFIG_IEEE80211W
351 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_AES_128_CMAC)
352 		return WPA_CIPHER_AES_128_CMAC;
353 #endif /* CONFIG_IEEE80211W */
354 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_GCMP)
355 		return WPA_CIPHER_GCMP;
356 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_CCMP_256)
357 		return WPA_CIPHER_CCMP_256;
358 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_GCMP_256)
359 		return WPA_CIPHER_GCMP_256;
360 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_BIP_GMAC_128)
361 		return WPA_CIPHER_BIP_GMAC_128;
362 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_BIP_GMAC_256)
363 		return WPA_CIPHER_BIP_GMAC_256;
364 	if (RSN_SELECTOR_GET(s) == RSN_CIPHER_SUITE_BIP_CMAC_256)
365 		return WPA_CIPHER_BIP_CMAC_256;
366 	return 0;
367 }
368 
369 
370 static int rsn_key_mgmt_to_bitfield(const u8 *s)
371 {
372 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_UNSPEC_802_1X)
373 		return WPA_KEY_MGMT_IEEE8021X;
374 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X)
375 		return WPA_KEY_MGMT_PSK;
376 #ifdef CONFIG_IEEE80211R
377 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_802_1X)
378 		return WPA_KEY_MGMT_FT_IEEE8021X;
379 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_PSK)
380 		return WPA_KEY_MGMT_FT_PSK;
381 #endif /* CONFIG_IEEE80211R */
382 #ifdef CONFIG_IEEE80211W
383 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_802_1X_SHA256)
384 		return WPA_KEY_MGMT_IEEE8021X_SHA256;
385 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_PSK_SHA256)
386 		return WPA_KEY_MGMT_PSK_SHA256;
387 #endif /* CONFIG_IEEE80211W */
388 #ifdef CONFIG_SAE
389 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_SAE)
390 		return WPA_KEY_MGMT_SAE;
391 	if (RSN_SELECTOR_GET(s) == RSN_AUTH_KEY_MGMT_FT_SAE)
392 		return WPA_KEY_MGMT_FT_SAE;
393 #endif /* CONFIG_SAE */
394 	return 0;
395 }
396 
397 
398 /**
399  * wpa_parse_wpa_ie_rsn - Parse RSN IE
400  * @rsn_ie: Buffer containing RSN IE
401  * @rsn_ie_len: RSN IE buffer length (including IE number and length octets)
402  * @data: Pointer to structure that will be filled in with parsed data
403  * Returns: 0 on success, <0 on failure
404  */
405 int wpa_parse_wpa_ie_rsn(const u8 *rsn_ie, size_t rsn_ie_len,
406 			 struct wpa_ie_data *data)
407 {
408 	const struct rsn_ie_hdr *hdr;
409 	const u8 *pos;
410 	int left;
411 	int i, count;
412 
413 	os_memset(data, 0, sizeof(*data));
414 	data->proto = WPA_PROTO_RSN;
415 	data->pairwise_cipher = WPA_CIPHER_CCMP;
416 	data->group_cipher = WPA_CIPHER_CCMP;
417 	data->key_mgmt = WPA_KEY_MGMT_IEEE8021X;
418 	data->capabilities = 0;
419 	data->pmkid = NULL;
420 	data->num_pmkid = 0;
421 #ifdef CONFIG_IEEE80211W
422 	data->mgmt_group_cipher = WPA_CIPHER_AES_128_CMAC;
423 #else /* CONFIG_IEEE80211W */
424 	data->mgmt_group_cipher = 0;
425 #endif /* CONFIG_IEEE80211W */
426 
427 	if (rsn_ie_len == 0) {
428 		/* No RSN IE - fail silently */
429 		return -1;
430 	}
431 
432 	if (rsn_ie_len < sizeof(struct rsn_ie_hdr)) {
433 		wpa_printf(MSG_DEBUG, "%s: ie len too short %lu",
434 			   __func__, (unsigned long) rsn_ie_len);
435 		return -1;
436 	}
437 
438 	hdr = (const struct rsn_ie_hdr *) rsn_ie;
439 
440 	if (hdr->elem_id != WLAN_EID_RSN ||
441 	    hdr->len != rsn_ie_len - 2 ||
442 	    WPA_GET_LE16(hdr->version) != RSN_VERSION) {
443 		wpa_printf(MSG_DEBUG, "%s: malformed ie or unknown version",
444 			   __func__);
445 		return -2;
446 	}
447 
448 	pos = (const u8 *) (hdr + 1);
449 	left = rsn_ie_len - sizeof(*hdr);
450 
451 	if (left >= RSN_SELECTOR_LEN) {
452 		data->group_cipher = rsn_selector_to_bitfield(pos);
453 #ifdef CONFIG_IEEE80211W
454 		if (data->group_cipher == WPA_CIPHER_AES_128_CMAC) {
455 			wpa_printf(MSG_DEBUG, "%s: AES-128-CMAC used as group "
456 				   "cipher", __func__);
457 			return -1;
458 		}
459 #endif /* CONFIG_IEEE80211W */
460 		pos += RSN_SELECTOR_LEN;
461 		left -= RSN_SELECTOR_LEN;
462 	} else if (left > 0) {
463 		wpa_printf(MSG_DEBUG, "%s: ie length mismatch, %u too much",
464 			   __func__, left);
465 		return -3;
466 	}
467 
468 	if (left >= 2) {
469 		data->pairwise_cipher = 0;
470 		count = WPA_GET_LE16(pos);
471 		pos += 2;
472 		left -= 2;
473 		if (count == 0 || left < count * RSN_SELECTOR_LEN) {
474 			wpa_printf(MSG_DEBUG, "%s: ie count botch (pairwise), "
475 				   "count %u left %u", __func__, count, left);
476 			return -4;
477 		}
478 		for (i = 0; i < count; i++) {
479 			data->pairwise_cipher |= rsn_selector_to_bitfield(pos);
480 			pos += RSN_SELECTOR_LEN;
481 			left -= RSN_SELECTOR_LEN;
482 		}
483 #ifdef CONFIG_IEEE80211W
484 		if (data->pairwise_cipher & WPA_CIPHER_AES_128_CMAC) {
485 			wpa_printf(MSG_DEBUG, "%s: AES-128-CMAC used as "
486 				   "pairwise cipher", __func__);
487 			return -1;
488 		}
489 #endif /* CONFIG_IEEE80211W */
490 	} else if (left == 1) {
491 		wpa_printf(MSG_DEBUG, "%s: ie too short (for key mgmt)",
492 			   __func__);
493 		return -5;
494 	}
495 
496 	if (left >= 2) {
497 		data->key_mgmt = 0;
498 		count = WPA_GET_LE16(pos);
499 		pos += 2;
500 		left -= 2;
501 		if (count == 0 || left < count * RSN_SELECTOR_LEN) {
502 			wpa_printf(MSG_DEBUG, "%s: ie count botch (key mgmt), "
503 				   "count %u left %u", __func__, count, left);
504 			return -6;
505 		}
506 		for (i = 0; i < count; i++) {
507 			data->key_mgmt |= rsn_key_mgmt_to_bitfield(pos);
508 			pos += RSN_SELECTOR_LEN;
509 			left -= RSN_SELECTOR_LEN;
510 		}
511 	} else if (left == 1) {
512 		wpa_printf(MSG_DEBUG, "%s: ie too short (for capabilities)",
513 			   __func__);
514 		return -7;
515 	}
516 
517 	if (left >= 2) {
518 		data->capabilities = WPA_GET_LE16(pos);
519 		pos += 2;
520 		left -= 2;
521 	}
522 
523 	if (left >= 2) {
524 		data->num_pmkid = WPA_GET_LE16(pos);
525 		pos += 2;
526 		left -= 2;
527 		if (left < (int) data->num_pmkid * PMKID_LEN) {
528 			wpa_printf(MSG_DEBUG, "%s: PMKID underflow "
529 				   "(num_pmkid=%lu left=%d)",
530 				   __func__, (unsigned long) data->num_pmkid,
531 				   left);
532 			data->num_pmkid = 0;
533 			return -9;
534 		} else {
535 			data->pmkid = pos;
536 			pos += data->num_pmkid * PMKID_LEN;
537 			left -= data->num_pmkid * PMKID_LEN;
538 		}
539 	}
540 
541 #ifdef CONFIG_IEEE80211W
542 	if (left >= 4) {
543 		data->mgmt_group_cipher = rsn_selector_to_bitfield(pos);
544 		if (data->mgmt_group_cipher != WPA_CIPHER_AES_128_CMAC) {
545 			wpa_printf(MSG_DEBUG, "%s: Unsupported management "
546 				   "group cipher 0x%x", __func__,
547 				   data->mgmt_group_cipher);
548 			return -10;
549 		}
550 		pos += RSN_SELECTOR_LEN;
551 		left -= RSN_SELECTOR_LEN;
552 	}
553 #endif /* CONFIG_IEEE80211W */
554 
555 	if (left > 0) {
556 		wpa_printf(MSG_DEBUG, "%s: ie has %u trailing bytes - ignored",
557 			   __func__, left);
558 	}
559 
560 	return 0;
561 }
562 
563 
564 static int wpa_selector_to_bitfield(const u8 *s)
565 {
566 	if (RSN_SELECTOR_GET(s) == WPA_CIPHER_SUITE_NONE)
567 		return WPA_CIPHER_NONE;
568 	if (RSN_SELECTOR_GET(s) == WPA_CIPHER_SUITE_WEP40)
569 		return WPA_CIPHER_WEP40;
570 	if (RSN_SELECTOR_GET(s) == WPA_CIPHER_SUITE_TKIP)
571 		return WPA_CIPHER_TKIP;
572 	if (RSN_SELECTOR_GET(s) == WPA_CIPHER_SUITE_CCMP)
573 		return WPA_CIPHER_CCMP;
574 	if (RSN_SELECTOR_GET(s) == WPA_CIPHER_SUITE_WEP104)
575 		return WPA_CIPHER_WEP104;
576 	return 0;
577 }
578 
579 
580 static int wpa_key_mgmt_to_bitfield(const u8 *s)
581 {
582 	if (RSN_SELECTOR_GET(s) == WPA_AUTH_KEY_MGMT_UNSPEC_802_1X)
583 		return WPA_KEY_MGMT_IEEE8021X;
584 	if (RSN_SELECTOR_GET(s) == WPA_AUTH_KEY_MGMT_PSK_OVER_802_1X)
585 		return WPA_KEY_MGMT_PSK;
586 	if (RSN_SELECTOR_GET(s) == WPA_AUTH_KEY_MGMT_NONE)
587 		return WPA_KEY_MGMT_WPA_NONE;
588 	return 0;
589 }
590 
591 
592 int wpa_parse_wpa_ie_wpa(const u8 *wpa_ie, size_t wpa_ie_len,
593 			 struct wpa_ie_data *data)
594 {
595 	const struct wpa_ie_hdr *hdr;
596 	const u8 *pos;
597 	int left;
598 	int i, count;
599 
600 	os_memset(data, 0, sizeof(*data));
601 	data->proto = WPA_PROTO_WPA;
602 	data->pairwise_cipher = WPA_CIPHER_TKIP;
603 	data->group_cipher = WPA_CIPHER_TKIP;
604 	data->key_mgmt = WPA_KEY_MGMT_IEEE8021X;
605 	data->capabilities = 0;
606 	data->pmkid = NULL;
607 	data->num_pmkid = 0;
608 	data->mgmt_group_cipher = 0;
609 
610 	if (wpa_ie_len == 0) {
611 		/* No WPA IE - fail silently */
612 		return -1;
613 	}
614 
615 	if (wpa_ie_len < sizeof(struct wpa_ie_hdr)) {
616 		wpa_printf(MSG_DEBUG, "%s: ie len too short %lu",
617 			   __func__, (unsigned long) wpa_ie_len);
618 		return -1;
619 	}
620 
621 	hdr = (const struct wpa_ie_hdr *) wpa_ie;
622 
623 	if (hdr->elem_id != WLAN_EID_VENDOR_SPECIFIC ||
624 	    hdr->len != wpa_ie_len - 2 ||
625 	    RSN_SELECTOR_GET(hdr->oui) != WPA_OUI_TYPE ||
626 	    WPA_GET_LE16(hdr->version) != WPA_VERSION) {
627 		wpa_printf(MSG_DEBUG, "%s: malformed ie or unknown version",
628 			   __func__);
629 		return -2;
630 	}
631 
632 	pos = (const u8 *) (hdr + 1);
633 	left = wpa_ie_len - sizeof(*hdr);
634 
635 	if (left >= WPA_SELECTOR_LEN) {
636 		data->group_cipher = wpa_selector_to_bitfield(pos);
637 		pos += WPA_SELECTOR_LEN;
638 		left -= WPA_SELECTOR_LEN;
639 	} else if (left > 0) {
640 		wpa_printf(MSG_DEBUG, "%s: ie length mismatch, %u too much",
641 			   __func__, left);
642 		return -3;
643 	}
644 
645 	if (left >= 2) {
646 		data->pairwise_cipher = 0;
647 		count = WPA_GET_LE16(pos);
648 		pos += 2;
649 		left -= 2;
650 		if (count == 0 || left < count * WPA_SELECTOR_LEN) {
651 			wpa_printf(MSG_DEBUG, "%s: ie count botch (pairwise), "
652 				   "count %u left %u", __func__, count, left);
653 			return -4;
654 		}
655 		for (i = 0; i < count; i++) {
656 			data->pairwise_cipher |= wpa_selector_to_bitfield(pos);
657 			pos += WPA_SELECTOR_LEN;
658 			left -= WPA_SELECTOR_LEN;
659 		}
660 	} else if (left == 1) {
661 		wpa_printf(MSG_DEBUG, "%s: ie too short (for key mgmt)",
662 			   __func__);
663 		return -5;
664 	}
665 
666 	if (left >= 2) {
667 		data->key_mgmt = 0;
668 		count = WPA_GET_LE16(pos);
669 		pos += 2;
670 		left -= 2;
671 		if (count == 0 || left < count * WPA_SELECTOR_LEN) {
672 			wpa_printf(MSG_DEBUG, "%s: ie count botch (key mgmt), "
673 				   "count %u left %u", __func__, count, left);
674 			return -6;
675 		}
676 		for (i = 0; i < count; i++) {
677 			data->key_mgmt |= wpa_key_mgmt_to_bitfield(pos);
678 			pos += WPA_SELECTOR_LEN;
679 			left -= WPA_SELECTOR_LEN;
680 		}
681 	} else if (left == 1) {
682 		wpa_printf(MSG_DEBUG, "%s: ie too short (for capabilities)",
683 			   __func__);
684 		return -7;
685 	}
686 
687 	if (left >= 2) {
688 		data->capabilities = WPA_GET_LE16(pos);
689 		pos += 2;
690 		left -= 2;
691 	}
692 
693 	if (left > 0) {
694 		wpa_printf(MSG_DEBUG, "%s: ie has %u trailing bytes - ignored",
695 			   __func__, left);
696 	}
697 
698 	return 0;
699 }
700 
701 
702 #ifdef CONFIG_IEEE80211R
703 
704 /**
705  * wpa_derive_pmk_r0 - Derive PMK-R0 and PMKR0Name
706  *
707  * IEEE Std 802.11r-2008 - 8.5.1.5.3
708  */
709 void wpa_derive_pmk_r0(const u8 *xxkey, size_t xxkey_len,
710 		       const u8 *ssid, size_t ssid_len,
711 		       const u8 *mdid, const u8 *r0kh_id, size_t r0kh_id_len,
712 		       const u8 *s0kh_id, u8 *pmk_r0, u8 *pmk_r0_name)
713 {
714 	u8 buf[1 + WPA_MAX_SSID_LEN + MOBILITY_DOMAIN_ID_LEN + 1 +
715 	       FT_R0KH_ID_MAX_LEN + ETH_ALEN];
716 	u8 *pos, r0_key_data[48], hash[32];
717 	const u8 *addr[2];
718 	size_t len[2];
719 
720 	/*
721 	 * R0-Key-Data = KDF-384(XXKey, "FT-R0",
722 	 *                       SSIDlength || SSID || MDID || R0KHlength ||
723 	 *                       R0KH-ID || S0KH-ID)
724 	 * XXKey is either the second 256 bits of MSK or PSK.
725 	 * PMK-R0 = L(R0-Key-Data, 0, 256)
726 	 * PMK-R0Name-Salt = L(R0-Key-Data, 256, 128)
727 	 */
728 	if (ssid_len > WPA_MAX_SSID_LEN || r0kh_id_len > FT_R0KH_ID_MAX_LEN)
729 		return;
730 	pos = buf;
731 	*pos++ = ssid_len;
732 	os_memcpy(pos, ssid, ssid_len);
733 	pos += ssid_len;
734 	os_memcpy(pos, mdid, MOBILITY_DOMAIN_ID_LEN);
735 	pos += MOBILITY_DOMAIN_ID_LEN;
736 	*pos++ = r0kh_id_len;
737 	os_memcpy(pos, r0kh_id, r0kh_id_len);
738 	pos += r0kh_id_len;
739 	os_memcpy(pos, s0kh_id, ETH_ALEN);
740 	pos += ETH_ALEN;
741 
742 	sha256_prf(xxkey, xxkey_len, "FT-R0", buf, pos - buf,
743 		   r0_key_data, sizeof(r0_key_data));
744 	os_memcpy(pmk_r0, r0_key_data, PMK_LEN);
745 
746 	/*
747 	 * PMKR0Name = Truncate-128(SHA-256("FT-R0N" || PMK-R0Name-Salt)
748 	 */
749 	addr[0] = (const u8 *) "FT-R0N";
750 	len[0] = 6;
751 	addr[1] = r0_key_data + PMK_LEN;
752 	len[1] = 16;
753 
754 	sha256_vector(2, addr, len, hash);
755 	os_memcpy(pmk_r0_name, hash, WPA_PMK_NAME_LEN);
756 }
757 
758 
759 /**
760  * wpa_derive_pmk_r1_name - Derive PMKR1Name
761  *
762  * IEEE Std 802.11r-2008 - 8.5.1.5.4
763  */
764 void wpa_derive_pmk_r1_name(const u8 *pmk_r0_name, const u8 *r1kh_id,
765 			    const u8 *s1kh_id, u8 *pmk_r1_name)
766 {
767 	u8 hash[32];
768 	const u8 *addr[4];
769 	size_t len[4];
770 
771 	/*
772 	 * PMKR1Name = Truncate-128(SHA-256("FT-R1N" || PMKR0Name ||
773 	 *                                  R1KH-ID || S1KH-ID))
774 	 */
775 	addr[0] = (const u8 *) "FT-R1N";
776 	len[0] = 6;
777 	addr[1] = pmk_r0_name;
778 	len[1] = WPA_PMK_NAME_LEN;
779 	addr[2] = r1kh_id;
780 	len[2] = FT_R1KH_ID_LEN;
781 	addr[3] = s1kh_id;
782 	len[3] = ETH_ALEN;
783 
784 	sha256_vector(4, addr, len, hash);
785 	os_memcpy(pmk_r1_name, hash, WPA_PMK_NAME_LEN);
786 }
787 
788 
789 /**
790  * wpa_derive_pmk_r1 - Derive PMK-R1 and PMKR1Name from PMK-R0
791  *
792  * IEEE Std 802.11r-2008 - 8.5.1.5.4
793  */
794 void wpa_derive_pmk_r1(const u8 *pmk_r0, const u8 *pmk_r0_name,
795 		       const u8 *r1kh_id, const u8 *s1kh_id,
796 		       u8 *pmk_r1, u8 *pmk_r1_name)
797 {
798 	u8 buf[FT_R1KH_ID_LEN + ETH_ALEN];
799 	u8 *pos;
800 
801 	/* PMK-R1 = KDF-256(PMK-R0, "FT-R1", R1KH-ID || S1KH-ID) */
802 	pos = buf;
803 	os_memcpy(pos, r1kh_id, FT_R1KH_ID_LEN);
804 	pos += FT_R1KH_ID_LEN;
805 	os_memcpy(pos, s1kh_id, ETH_ALEN);
806 	pos += ETH_ALEN;
807 
808 	sha256_prf(pmk_r0, PMK_LEN, "FT-R1", buf, pos - buf, pmk_r1, PMK_LEN);
809 
810 	wpa_derive_pmk_r1_name(pmk_r0_name, r1kh_id, s1kh_id, pmk_r1_name);
811 }
812 
813 
814 /**
815  * wpa_pmk_r1_to_ptk - Derive PTK and PTKName from PMK-R1
816  *
817  * IEEE Std 802.11r-2008 - 8.5.1.5.5
818  */
819 void wpa_pmk_r1_to_ptk(const u8 *pmk_r1, const u8 *snonce, const u8 *anonce,
820 		       const u8 *sta_addr, const u8 *bssid,
821 		       const u8 *pmk_r1_name,
822 		       u8 *ptk, size_t ptk_len, u8 *ptk_name)
823 {
824 	u8 buf[2 * WPA_NONCE_LEN + 2 * ETH_ALEN];
825 	u8 *pos, hash[32];
826 	const u8 *addr[6];
827 	size_t len[6];
828 
829 	/*
830 	 * PTK = KDF-PTKLen(PMK-R1, "FT-PTK", SNonce || ANonce ||
831 	 *                  BSSID || STA-ADDR)
832 	 */
833 	pos = buf;
834 	os_memcpy(pos, snonce, WPA_NONCE_LEN);
835 	pos += WPA_NONCE_LEN;
836 	os_memcpy(pos, anonce, WPA_NONCE_LEN);
837 	pos += WPA_NONCE_LEN;
838 	os_memcpy(pos, bssid, ETH_ALEN);
839 	pos += ETH_ALEN;
840 	os_memcpy(pos, sta_addr, ETH_ALEN);
841 	pos += ETH_ALEN;
842 
843 	sha256_prf(pmk_r1, PMK_LEN, "FT-PTK", buf, pos - buf, ptk, ptk_len);
844 
845 	/*
846 	 * PTKName = Truncate-128(SHA-256(PMKR1Name || "FT-PTKN" || SNonce ||
847 	 *                                ANonce || BSSID || STA-ADDR))
848 	 */
849 	addr[0] = pmk_r1_name;
850 	len[0] = WPA_PMK_NAME_LEN;
851 	addr[1] = (const u8 *) "FT-PTKN";
852 	len[1] = 7;
853 	addr[2] = snonce;
854 	len[2] = WPA_NONCE_LEN;
855 	addr[3] = anonce;
856 	len[3] = WPA_NONCE_LEN;
857 	addr[4] = bssid;
858 	len[4] = ETH_ALEN;
859 	addr[5] = sta_addr;
860 	len[5] = ETH_ALEN;
861 
862 	sha256_vector(6, addr, len, hash);
863 	os_memcpy(ptk_name, hash, WPA_PMK_NAME_LEN);
864 }
865 
866 #endif /* CONFIG_IEEE80211R */
867 
868 
869 /**
870  * rsn_pmkid - Calculate PMK identifier
871  * @pmk: Pairwise master key
872  * @pmk_len: Length of pmk in bytes
873  * @aa: Authenticator address
874  * @spa: Supplicant address
875  * @pmkid: Buffer for PMKID
876  * @use_sha256: Whether to use SHA256-based KDF
877  *
878  * IEEE Std 802.11i-2004 - 8.5.1.2 Pairwise key hierarchy
879  * PMKID = HMAC-SHA1-128(PMK, "PMK Name" || AA || SPA)
880  */
881 void rsn_pmkid(const u8 *pmk, size_t pmk_len, const u8 *aa, const u8 *spa,
882 	       u8 *pmkid, int use_sha256)
883 {
884 	char *title = "PMK Name";
885 	const u8 *addr[3];
886 	const size_t len[3] = { 8, ETH_ALEN, ETH_ALEN };
887 	unsigned char hash[SHA256_MAC_LEN];
888 
889 	addr[0] = (u8 *) title;
890 	addr[1] = aa;
891 	addr[2] = spa;
892 
893 #ifdef CONFIG_IEEE80211W
894 	if (use_sha256)
895 		hmac_sha256_vector(pmk, pmk_len, 3, addr, len, hash);
896 	else
897 #endif /* CONFIG_IEEE80211W */
898 		hmac_sha1_vector(pmk, pmk_len, 3, addr, len, hash);
899 	os_memcpy(pmkid, hash, PMKID_LEN);
900 }
901 
902 
903 /**
904  * wpa_cipher_txt - Convert cipher suite to a text string
905  * @cipher: Cipher suite (WPA_CIPHER_* enum)
906  * Returns: Pointer to a text string of the cipher suite name
907  */
908 const char * wpa_cipher_txt(int cipher)
909 {
910 	switch (cipher) {
911 	case WPA_CIPHER_NONE:
912 		return "NONE";
913 	case WPA_CIPHER_WEP40:
914 		return "WEP-40";
915 	case WPA_CIPHER_WEP104:
916 		return "WEP-104";
917 	case WPA_CIPHER_TKIP:
918 		return "TKIP";
919 	case WPA_CIPHER_CCMP:
920 		return "CCMP";
921 	case WPA_CIPHER_CCMP | WPA_CIPHER_TKIP:
922 		return "CCMP+TKIP";
923 	case WPA_CIPHER_GCMP:
924 		return "GCMP";
925 	case WPA_CIPHER_GCMP_256:
926 		return "GCMP-256";
927 	case WPA_CIPHER_CCMP_256:
928 		return "CCMP-256";
929 	case WPA_CIPHER_GTK_NOT_USED:
930 		return "GTK_NOT_USED";
931 	default:
932 		return "UNKNOWN";
933 	}
934 }
935 
936 
937 /**
938  * wpa_key_mgmt_txt - Convert key management suite to a text string
939  * @key_mgmt: Key management suite (WPA_KEY_MGMT_* enum)
940  * @proto: WPA/WPA2 version (WPA_PROTO_*)
941  * Returns: Pointer to a text string of the key management suite name
942  */
943 const char * wpa_key_mgmt_txt(int key_mgmt, int proto)
944 {
945 	switch (key_mgmt) {
946 	case WPA_KEY_MGMT_IEEE8021X:
947 		if (proto == (WPA_PROTO_RSN | WPA_PROTO_WPA))
948 			return "WPA2+WPA/IEEE 802.1X/EAP";
949 		return proto == WPA_PROTO_RSN ?
950 			"WPA2/IEEE 802.1X/EAP" : "WPA/IEEE 802.1X/EAP";
951 	case WPA_KEY_MGMT_PSK:
952 		if (proto == (WPA_PROTO_RSN | WPA_PROTO_WPA))
953 			return "WPA2-PSK+WPA-PSK";
954 		return proto == WPA_PROTO_RSN ?
955 			"WPA2-PSK" : "WPA-PSK";
956 	case WPA_KEY_MGMT_NONE:
957 		return "NONE";
958 	case WPA_KEY_MGMT_IEEE8021X_NO_WPA:
959 		return "IEEE 802.1X (no WPA)";
960 #ifdef CONFIG_IEEE80211R
961 	case WPA_KEY_MGMT_FT_IEEE8021X:
962 		return "FT-EAP";
963 	case WPA_KEY_MGMT_FT_PSK:
964 		return "FT-PSK";
965 #endif /* CONFIG_IEEE80211R */
966 #ifdef CONFIG_IEEE80211W
967 	case WPA_KEY_MGMT_IEEE8021X_SHA256:
968 		return "WPA2-EAP-SHA256";
969 	case WPA_KEY_MGMT_PSK_SHA256:
970 		return "WPA2-PSK-SHA256";
971 #endif /* CONFIG_IEEE80211W */
972 	default:
973 		return "UNKNOWN";
974 	}
975 }
976 
977 
978 int wpa_compare_rsn_ie(int ft_initial_assoc,
979 		       const u8 *ie1, size_t ie1len,
980 		       const u8 *ie2, size_t ie2len)
981 {
982 	if (ie1 == NULL || ie2 == NULL)
983 		return -1;
984 
985 	if (ie1len == ie2len && os_memcmp(ie1, ie2, ie1len) == 0)
986 		return 0; /* identical IEs */
987 
988 #ifdef CONFIG_IEEE80211R
989 	if (ft_initial_assoc) {
990 		struct wpa_ie_data ie1d, ie2d;
991 		/*
992 		 * The PMKID-List in RSN IE is different between Beacon/Probe
993 		 * Response/(Re)Association Request frames and EAPOL-Key
994 		 * messages in FT initial mobility domain association. Allow
995 		 * for this, but verify that other parts of the RSN IEs are
996 		 * identical.
997 		 */
998 		if (wpa_parse_wpa_ie_rsn(ie1, ie1len, &ie1d) < 0 ||
999 		    wpa_parse_wpa_ie_rsn(ie2, ie2len, &ie2d) < 0)
1000 			return -1;
1001 		if (ie1d.proto == ie2d.proto &&
1002 		    ie1d.pairwise_cipher == ie2d.pairwise_cipher &&
1003 		    ie1d.group_cipher == ie2d.group_cipher &&
1004 		    ie1d.key_mgmt == ie2d.key_mgmt &&
1005 		    ie1d.capabilities == ie2d.capabilities &&
1006 		    ie1d.mgmt_group_cipher == ie2d.mgmt_group_cipher)
1007 			return 0;
1008 	}
1009 #endif /* CONFIG_IEEE80211R */
1010 
1011 	return -1;
1012 }
1013 
1014 
1015 #ifdef CONFIG_IEEE80211R
1016 int wpa_insert_pmkid(u8 *ies, size_t ies_len, const u8 *pmkid)
1017 {
1018 	u8 *start, *end, *rpos, *rend;
1019 	int added = 0;
1020 
1021 	start = ies;
1022 	end = ies + ies_len;
1023 
1024 	while (start < end) {
1025 		if (*start == WLAN_EID_RSN)
1026 			break;
1027 		start += 2 + start[1];
1028 	}
1029 	if (start >= end) {
1030 		wpa_printf(MSG_ERROR, "FT: Could not find RSN IE in "
1031 			   "IEs data");
1032 		return -1;
1033 	}
1034 	wpa_hexdump(MSG_DEBUG, "FT: RSN IE before modification",
1035 		    start, 2 + start[1]);
1036 
1037 	/* Find start of PMKID-Count */
1038 	rpos = start + 2;
1039 	rend = rpos + start[1];
1040 
1041 	/* Skip Version and Group Data Cipher Suite */
1042 	rpos += 2 + 4;
1043 	/* Skip Pairwise Cipher Suite Count and List */
1044 	rpos += 2 + WPA_GET_LE16(rpos) * RSN_SELECTOR_LEN;
1045 	/* Skip AKM Suite Count and List */
1046 	rpos += 2 + WPA_GET_LE16(rpos) * RSN_SELECTOR_LEN;
1047 
1048 	if (rpos == rend) {
1049 		/* Add RSN Capabilities */
1050 		os_memmove(rpos + 2, rpos, end - rpos);
1051 		*rpos++ = 0;
1052 		*rpos++ = 0;
1053 	} else {
1054 		/* Skip RSN Capabilities */
1055 		rpos += 2;
1056 		if (rpos > rend) {
1057 			wpa_printf(MSG_ERROR, "FT: Could not parse RSN IE in "
1058 				   "IEs data");
1059 			return -1;
1060 		}
1061 	}
1062 
1063 	if (rpos == rend) {
1064 		/* No PMKID-Count field included; add it */
1065 		os_memmove(rpos + 2 + PMKID_LEN, rpos, end - rpos);
1066 		WPA_PUT_LE16(rpos, 1);
1067 		rpos += 2;
1068 		os_memcpy(rpos, pmkid, PMKID_LEN);
1069 		added += 2 + PMKID_LEN;
1070 		start[1] += 2 + PMKID_LEN;
1071 	} else {
1072 		/* PMKID-Count was included; use it */
1073 		if (WPA_GET_LE16(rpos) != 0) {
1074 			wpa_printf(MSG_ERROR, "FT: Unexpected PMKID "
1075 				   "in RSN IE in EAPOL-Key data");
1076 			return -1;
1077 		}
1078 		WPA_PUT_LE16(rpos, 1);
1079 		rpos += 2;
1080 		os_memmove(rpos + PMKID_LEN, rpos, end - rpos);
1081 		os_memcpy(rpos, pmkid, PMKID_LEN);
1082 		added += PMKID_LEN;
1083 		start[1] += PMKID_LEN;
1084 	}
1085 
1086 	wpa_hexdump(MSG_DEBUG, "FT: RSN IE after modification "
1087 		    "(PMKID inserted)", start, 2 + start[1]);
1088 
1089 	return added;
1090 }
1091 #endif /* CONFIG_IEEE80211R */
1092 
1093 
1094 int wpa_cipher_key_len(int cipher)
1095 {
1096 	switch (cipher) {
1097 	case WPA_CIPHER_CCMP_256:
1098 	case WPA_CIPHER_GCMP_256:
1099 		return 32;
1100 	case WPA_CIPHER_CCMP:
1101 	case WPA_CIPHER_GCMP:
1102 		return 16;
1103 	case WPA_CIPHER_TKIP:
1104 		return 32;
1105 	case WPA_CIPHER_WEP104:
1106 		return 13;
1107 	case WPA_CIPHER_WEP40:
1108 		return 5;
1109 	}
1110 
1111 	return 0;
1112 }
1113 
1114 
1115 int wpa_cipher_rsc_len(int cipher)
1116 {
1117 	switch (cipher) {
1118 	case WPA_CIPHER_CCMP_256:
1119 	case WPA_CIPHER_GCMP_256:
1120 	case WPA_CIPHER_CCMP:
1121 	case WPA_CIPHER_GCMP:
1122 	case WPA_CIPHER_TKIP:
1123 		return 6;
1124 	case WPA_CIPHER_WEP104:
1125 	case WPA_CIPHER_WEP40:
1126 		return 0;
1127 	}
1128 
1129 	return 0;
1130 }
1131 
1132 
1133 int wpa_cipher_to_alg(int cipher)
1134 {
1135 	switch (cipher) {
1136 	case WPA_CIPHER_CCMP_256:
1137 		return WPA_ALG_CCMP_256;
1138 	case WPA_CIPHER_GCMP_256:
1139 		return WPA_ALG_GCMP_256;
1140 	case WPA_CIPHER_CCMP:
1141 		return WPA_ALG_CCMP;
1142 	case WPA_CIPHER_GCMP:
1143 		return WPA_ALG_GCMP;
1144 	case WPA_CIPHER_TKIP:
1145 		return WPA_ALG_TKIP;
1146 	case WPA_CIPHER_WEP104:
1147 	case WPA_CIPHER_WEP40:
1148 		return WPA_ALG_WEP;
1149 	}
1150 	return WPA_ALG_NONE;
1151 }
1152 
1153 
1154 int wpa_cipher_valid_pairwise(int cipher)
1155 {
1156 	return cipher == WPA_CIPHER_CCMP_256 ||
1157 		cipher == WPA_CIPHER_GCMP_256 ||
1158 		cipher == WPA_CIPHER_CCMP ||
1159 		cipher == WPA_CIPHER_GCMP ||
1160 		cipher == WPA_CIPHER_TKIP;
1161 }
1162 
1163 
1164 u32 wpa_cipher_to_suite(int proto, int cipher)
1165 {
1166 	if (cipher & WPA_CIPHER_CCMP_256)
1167 		return RSN_CIPHER_SUITE_CCMP_256;
1168 	if (cipher & WPA_CIPHER_GCMP_256)
1169 		return RSN_CIPHER_SUITE_GCMP_256;
1170 	if (cipher & WPA_CIPHER_CCMP)
1171 		return (proto == WPA_PROTO_RSN ?
1172 			RSN_CIPHER_SUITE_CCMP : WPA_CIPHER_SUITE_CCMP);
1173 	if (cipher & WPA_CIPHER_GCMP)
1174 		return RSN_CIPHER_SUITE_GCMP;
1175 	if (cipher & WPA_CIPHER_TKIP)
1176 		return (proto == WPA_PROTO_RSN ?
1177 			RSN_CIPHER_SUITE_TKIP : WPA_CIPHER_SUITE_TKIP);
1178 	if (cipher & WPA_CIPHER_WEP104)
1179 		return (proto == WPA_PROTO_RSN ?
1180 			RSN_CIPHER_SUITE_WEP104 : WPA_CIPHER_SUITE_WEP104);
1181 	if (cipher & WPA_CIPHER_WEP40)
1182 		return (proto == WPA_PROTO_RSN ?
1183 			RSN_CIPHER_SUITE_WEP40 : WPA_CIPHER_SUITE_WEP40);
1184 	if (cipher & WPA_CIPHER_NONE)
1185 		return (proto == WPA_PROTO_RSN ?
1186 			RSN_CIPHER_SUITE_NONE : WPA_CIPHER_SUITE_NONE);
1187 	if (cipher & WPA_CIPHER_GTK_NOT_USED)
1188 		return RSN_CIPHER_SUITE_NO_GROUP_ADDRESSED;
1189 	return 0;
1190 }
1191 
1192 
1193 int rsn_cipher_put_suites(u8 *pos, int ciphers)
1194 {
1195 	int num_suites = 0;
1196 
1197 	if (ciphers & WPA_CIPHER_CCMP_256) {
1198 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_CCMP_256);
1199 		pos += RSN_SELECTOR_LEN;
1200 		num_suites++;
1201 	}
1202 	if (ciphers & WPA_CIPHER_GCMP_256) {
1203 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_GCMP_256);
1204 		pos += RSN_SELECTOR_LEN;
1205 		num_suites++;
1206 	}
1207 	if (ciphers & WPA_CIPHER_CCMP) {
1208 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_CCMP);
1209 		pos += RSN_SELECTOR_LEN;
1210 		num_suites++;
1211 	}
1212 	if (ciphers & WPA_CIPHER_GCMP) {
1213 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_GCMP);
1214 		pos += RSN_SELECTOR_LEN;
1215 		num_suites++;
1216 	}
1217 	if (ciphers & WPA_CIPHER_TKIP) {
1218 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_TKIP);
1219 		pos += RSN_SELECTOR_LEN;
1220 		num_suites++;
1221 	}
1222 	if (ciphers & WPA_CIPHER_NONE) {
1223 		RSN_SELECTOR_PUT(pos, RSN_CIPHER_SUITE_NONE);
1224 		pos += RSN_SELECTOR_LEN;
1225 		num_suites++;
1226 	}
1227 
1228 	return num_suites;
1229 }
1230 
1231 
1232 int wpa_cipher_put_suites(u8 *pos, int ciphers)
1233 {
1234 	int num_suites = 0;
1235 
1236 	if (ciphers & WPA_CIPHER_CCMP) {
1237 		RSN_SELECTOR_PUT(pos, WPA_CIPHER_SUITE_CCMP);
1238 		pos += WPA_SELECTOR_LEN;
1239 		num_suites++;
1240 	}
1241 	if (ciphers & WPA_CIPHER_TKIP) {
1242 		RSN_SELECTOR_PUT(pos, WPA_CIPHER_SUITE_TKIP);
1243 		pos += WPA_SELECTOR_LEN;
1244 		num_suites++;
1245 	}
1246 	if (ciphers & WPA_CIPHER_NONE) {
1247 		RSN_SELECTOR_PUT(pos, WPA_CIPHER_SUITE_NONE);
1248 		pos += WPA_SELECTOR_LEN;
1249 		num_suites++;
1250 	}
1251 
1252 	return num_suites;
1253 }
1254 
1255 
1256 int wpa_pick_pairwise_cipher(int ciphers, int none_allowed)
1257 {
1258 	if (ciphers & WPA_CIPHER_CCMP_256)
1259 		return WPA_CIPHER_CCMP_256;
1260 	if (ciphers & WPA_CIPHER_GCMP_256)
1261 		return WPA_CIPHER_GCMP_256;
1262 	if (ciphers & WPA_CIPHER_CCMP)
1263 		return WPA_CIPHER_CCMP;
1264 	if (ciphers & WPA_CIPHER_GCMP)
1265 		return WPA_CIPHER_GCMP;
1266 	if (ciphers & WPA_CIPHER_TKIP)
1267 		return WPA_CIPHER_TKIP;
1268 	if (none_allowed && (ciphers & WPA_CIPHER_NONE))
1269 		return WPA_CIPHER_NONE;
1270 	return -1;
1271 }
1272 
1273 
1274 int wpa_pick_group_cipher(int ciphers)
1275 {
1276 	if (ciphers & WPA_CIPHER_CCMP_256)
1277 		return WPA_CIPHER_CCMP_256;
1278 	if (ciphers & WPA_CIPHER_GCMP_256)
1279 		return WPA_CIPHER_GCMP_256;
1280 	if (ciphers & WPA_CIPHER_CCMP)
1281 		return WPA_CIPHER_CCMP;
1282 	if (ciphers & WPA_CIPHER_GCMP)
1283 		return WPA_CIPHER_GCMP;
1284 	if (ciphers & WPA_CIPHER_GTK_NOT_USED)
1285 		return WPA_CIPHER_GTK_NOT_USED;
1286 	if (ciphers & WPA_CIPHER_TKIP)
1287 		return WPA_CIPHER_TKIP;
1288 	if (ciphers & WPA_CIPHER_WEP104)
1289 		return WPA_CIPHER_WEP104;
1290 	if (ciphers & WPA_CIPHER_WEP40)
1291 		return WPA_CIPHER_WEP40;
1292 	return -1;
1293 }
1294 
1295 
1296 int wpa_parse_cipher(const char *value)
1297 {
1298 	int val = 0, last;
1299 	char *start, *end, *buf;
1300 
1301 	buf = os_strdup(value);
1302 	if (buf == NULL)
1303 		return -1;
1304 	start = buf;
1305 
1306 	while (*start != '\0') {
1307 		while (*start == ' ' || *start == '\t')
1308 			start++;
1309 		if (*start == '\0')
1310 			break;
1311 		end = start;
1312 		while (*end != ' ' && *end != '\t' && *end != '\0')
1313 			end++;
1314 		last = *end == '\0';
1315 		*end = '\0';
1316 		if (os_strcmp(start, "CCMP-256") == 0)
1317 			val |= WPA_CIPHER_CCMP_256;
1318 		else if (os_strcmp(start, "GCMP-256") == 0)
1319 			val |= WPA_CIPHER_GCMP_256;
1320 		else if (os_strcmp(start, "CCMP") == 0)
1321 			val |= WPA_CIPHER_CCMP;
1322 		else if (os_strcmp(start, "GCMP") == 0)
1323 			val |= WPA_CIPHER_GCMP;
1324 		else if (os_strcmp(start, "TKIP") == 0)
1325 			val |= WPA_CIPHER_TKIP;
1326 		else if (os_strcmp(start, "WEP104") == 0)
1327 			val |= WPA_CIPHER_WEP104;
1328 		else if (os_strcmp(start, "WEP40") == 0)
1329 			val |= WPA_CIPHER_WEP40;
1330 		else if (os_strcmp(start, "NONE") == 0)
1331 			val |= WPA_CIPHER_NONE;
1332 		else if (os_strcmp(start, "GTK_NOT_USED") == 0)
1333 			val |= WPA_CIPHER_GTK_NOT_USED;
1334 		else {
1335 			os_free(buf);
1336 			return -1;
1337 		}
1338 
1339 		if (last)
1340 			break;
1341 		start = end + 1;
1342 	}
1343 	os_free(buf);
1344 
1345 	return val;
1346 }
1347 
1348 
1349 int wpa_write_ciphers(char *start, char *end, int ciphers, const char *delim)
1350 {
1351 	char *pos = start;
1352 	int ret;
1353 
1354 	if (ciphers & WPA_CIPHER_CCMP_256) {
1355 		ret = os_snprintf(pos, end - pos, "%sCCMP-256",
1356 				  pos == start ? "" : delim);
1357 		if (ret < 0 || ret >= end - pos)
1358 			return -1;
1359 		pos += ret;
1360 	}
1361 	if (ciphers & WPA_CIPHER_GCMP_256) {
1362 		ret = os_snprintf(pos, end - pos, "%sGCMP-256",
1363 				  pos == start ? "" : delim);
1364 		if (ret < 0 || ret >= end - pos)
1365 			return -1;
1366 		pos += ret;
1367 	}
1368 	if (ciphers & WPA_CIPHER_CCMP) {
1369 		ret = os_snprintf(pos, end - pos, "%sCCMP",
1370 				  pos == start ? "" : delim);
1371 		if (ret < 0 || ret >= end - pos)
1372 			return -1;
1373 		pos += ret;
1374 	}
1375 	if (ciphers & WPA_CIPHER_GCMP) {
1376 		ret = os_snprintf(pos, end - pos, "%sGCMP",
1377 				  pos == start ? "" : delim);
1378 		if (ret < 0 || ret >= end - pos)
1379 			return -1;
1380 		pos += ret;
1381 	}
1382 	if (ciphers & WPA_CIPHER_TKIP) {
1383 		ret = os_snprintf(pos, end - pos, "%sTKIP",
1384 				  pos == start ? "" : delim);
1385 		if (ret < 0 || ret >= end - pos)
1386 			return -1;
1387 		pos += ret;
1388 	}
1389 	if (ciphers & WPA_CIPHER_WEP104) {
1390 		ret = os_snprintf(pos, end - pos, "%sWEP104",
1391 				  pos == start ? "" : delim);
1392 		if (ret < 0 || ret >= end - pos)
1393 			return -1;
1394 		pos += ret;
1395 	}
1396 	if (ciphers & WPA_CIPHER_WEP40) {
1397 		ret = os_snprintf(pos, end - pos, "%sWEP40",
1398 				  pos == start ? "" : delim);
1399 		if (ret < 0 || ret >= end - pos)
1400 			return -1;
1401 		pos += ret;
1402 	}
1403 	if (ciphers & WPA_CIPHER_NONE) {
1404 		ret = os_snprintf(pos, end - pos, "%sNONE",
1405 				  pos == start ? "" : delim);
1406 		if (ret < 0 || ret >= end - pos)
1407 			return -1;
1408 		pos += ret;
1409 	}
1410 
1411 	return pos - start;
1412 }
1413 
1414 
1415 int wpa_select_ap_group_cipher(int wpa, int wpa_pairwise, int rsn_pairwise)
1416 {
1417 	int pairwise = 0;
1418 
1419 	/* Select group cipher based on the enabled pairwise cipher suites */
1420 	if (wpa & 1)
1421 		pairwise |= wpa_pairwise;
1422 	if (wpa & 2)
1423 		pairwise |= rsn_pairwise;
1424 
1425 	if (pairwise & WPA_CIPHER_TKIP)
1426 		return WPA_CIPHER_TKIP;
1427 	if ((pairwise & (WPA_CIPHER_CCMP | WPA_CIPHER_GCMP)) == WPA_CIPHER_GCMP)
1428 		return WPA_CIPHER_GCMP;
1429 	if ((pairwise & (WPA_CIPHER_GCMP_256 | WPA_CIPHER_CCMP |
1430 			 WPA_CIPHER_GCMP)) == WPA_CIPHER_GCMP_256)
1431 		return WPA_CIPHER_GCMP_256;
1432 	if ((pairwise & (WPA_CIPHER_CCMP_256 | WPA_CIPHER_CCMP |
1433 			 WPA_CIPHER_GCMP)) == WPA_CIPHER_CCMP_256)
1434 		return WPA_CIPHER_CCMP_256;
1435 	return WPA_CIPHER_CCMP;
1436 }
1437