xref: /freebsd/contrib/wpa/src/ap/wpa_auth_ft.c (revision 6472ac3d8a86336899b6cfb789a4cd9897e3fab5)
1 /*
2  * hostapd - IEEE 802.11r - Fast BSS Transition
3  * Copyright (c) 2004-2009, 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 "utils/includes.h"
16 
17 #include "utils/common.h"
18 #include "common/ieee802_11_defs.h"
19 #include "common/ieee802_11_common.h"
20 #include "crypto/aes_wrap.h"
21 #include "ap_config.h"
22 #include "ieee802_11.h"
23 #include "wmm.h"
24 #include "wpa_auth.h"
25 #include "wpa_auth_i.h"
26 #include "wpa_auth_ie.h"
27 
28 
29 #ifdef CONFIG_IEEE80211R
30 
31 struct wpa_ft_ies {
32 	const u8 *mdie;
33 	size_t mdie_len;
34 	const u8 *ftie;
35 	size_t ftie_len;
36 	const u8 *r1kh_id;
37 	const u8 *gtk;
38 	size_t gtk_len;
39 	const u8 *r0kh_id;
40 	size_t r0kh_id_len;
41 	const u8 *rsn;
42 	size_t rsn_len;
43 	const u8 *rsn_pmkid;
44 	const u8 *ric;
45 	size_t ric_len;
46 };
47 
48 
49 static int wpa_ft_parse_ies(const u8 *ies, size_t ies_len,
50 			    struct wpa_ft_ies *parse);
51 
52 
53 static int wpa_ft_rrb_send(struct wpa_authenticator *wpa_auth, const u8 *dst,
54 			   const u8 *data, size_t data_len)
55 {
56 	if (wpa_auth->cb.send_ether == NULL)
57 		return -1;
58 	wpa_printf(MSG_DEBUG, "FT: RRB send to " MACSTR, MAC2STR(dst));
59 	return wpa_auth->cb.send_ether(wpa_auth->cb.ctx, dst, ETH_P_RRB,
60 				       data, data_len);
61 }
62 
63 
64 static int wpa_ft_action_send(struct wpa_authenticator *wpa_auth,
65 			      const u8 *dst, const u8 *data, size_t data_len)
66 {
67 	if (wpa_auth->cb.send_ft_action == NULL)
68 		return -1;
69 	return wpa_auth->cb.send_ft_action(wpa_auth->cb.ctx, dst,
70 					   data, data_len);
71 }
72 
73 
74 static struct wpa_state_machine *
75 wpa_ft_add_sta(struct wpa_authenticator *wpa_auth, const u8 *sta_addr)
76 {
77 	if (wpa_auth->cb.add_sta == NULL)
78 		return NULL;
79 	return wpa_auth->cb.add_sta(wpa_auth->cb.ctx, sta_addr);
80 }
81 
82 
83 int wpa_write_mdie(struct wpa_auth_config *conf, u8 *buf, size_t len)
84 {
85 	u8 *pos = buf;
86 	u8 capab;
87 	if (len < 2 + sizeof(struct rsn_mdie))
88 		return -1;
89 
90 	*pos++ = WLAN_EID_MOBILITY_DOMAIN;
91 	*pos++ = MOBILITY_DOMAIN_ID_LEN + 1;
92 	os_memcpy(pos, conf->mobility_domain, MOBILITY_DOMAIN_ID_LEN);
93 	pos += MOBILITY_DOMAIN_ID_LEN;
94 	capab = RSN_FT_CAPAB_FT_OVER_DS;
95 	*pos++ = capab;
96 
97 	return pos - buf;
98 }
99 
100 
101 int wpa_write_ftie(struct wpa_auth_config *conf, const u8 *r0kh_id,
102 		   size_t r0kh_id_len,
103 		   const u8 *anonce, const u8 *snonce,
104 		   u8 *buf, size_t len, const u8 *subelem,
105 		   size_t subelem_len)
106 {
107 	u8 *pos = buf, *ielen;
108 	struct rsn_ftie *hdr;
109 
110 	if (len < 2 + sizeof(*hdr) + 2 + FT_R1KH_ID_LEN + 2 + r0kh_id_len +
111 	    subelem_len)
112 		return -1;
113 
114 	*pos++ = WLAN_EID_FAST_BSS_TRANSITION;
115 	ielen = pos++;
116 
117 	hdr = (struct rsn_ftie *) pos;
118 	os_memset(hdr, 0, sizeof(*hdr));
119 	pos += sizeof(*hdr);
120 	WPA_PUT_LE16(hdr->mic_control, 0);
121 	if (anonce)
122 		os_memcpy(hdr->anonce, anonce, WPA_NONCE_LEN);
123 	if (snonce)
124 		os_memcpy(hdr->snonce, snonce, WPA_NONCE_LEN);
125 
126 	/* Optional Parameters */
127 	*pos++ = FTIE_SUBELEM_R1KH_ID;
128 	*pos++ = FT_R1KH_ID_LEN;
129 	os_memcpy(pos, conf->r1_key_holder, FT_R1KH_ID_LEN);
130 	pos += FT_R1KH_ID_LEN;
131 
132 	if (r0kh_id) {
133 		*pos++ = FTIE_SUBELEM_R0KH_ID;
134 		*pos++ = r0kh_id_len;
135 		os_memcpy(pos, r0kh_id, r0kh_id_len);
136 		pos += r0kh_id_len;
137 	}
138 
139 	if (subelem) {
140 		os_memcpy(pos, subelem, subelem_len);
141 		pos += subelem_len;
142 	}
143 
144 	*ielen = pos - buf - 2;
145 
146 	return pos - buf;
147 }
148 
149 
150 struct wpa_ft_pmk_r0_sa {
151 	struct wpa_ft_pmk_r0_sa *next;
152 	u8 pmk_r0[PMK_LEN];
153 	u8 pmk_r0_name[WPA_PMK_NAME_LEN];
154 	u8 spa[ETH_ALEN];
155 	int pairwise; /* Pairwise cipher suite, WPA_CIPHER_* */
156 	/* TODO: expiration, identity, radius_class, EAP type, VLAN ID */
157 	int pmk_r1_pushed;
158 };
159 
160 struct wpa_ft_pmk_r1_sa {
161 	struct wpa_ft_pmk_r1_sa *next;
162 	u8 pmk_r1[PMK_LEN];
163 	u8 pmk_r1_name[WPA_PMK_NAME_LEN];
164 	u8 spa[ETH_ALEN];
165 	int pairwise; /* Pairwise cipher suite, WPA_CIPHER_* */
166 	/* TODO: expiration, identity, radius_class, EAP type, VLAN ID */
167 };
168 
169 struct wpa_ft_pmk_cache {
170 	struct wpa_ft_pmk_r0_sa *pmk_r0;
171 	struct wpa_ft_pmk_r1_sa *pmk_r1;
172 };
173 
174 struct wpa_ft_pmk_cache * wpa_ft_pmk_cache_init(void)
175 {
176 	struct wpa_ft_pmk_cache *cache;
177 
178 	cache = os_zalloc(sizeof(*cache));
179 
180 	return cache;
181 }
182 
183 
184 void wpa_ft_pmk_cache_deinit(struct wpa_ft_pmk_cache *cache)
185 {
186 	struct wpa_ft_pmk_r0_sa *r0, *r0prev;
187 	struct wpa_ft_pmk_r1_sa *r1, *r1prev;
188 
189 	r0 = cache->pmk_r0;
190 	while (r0) {
191 		r0prev = r0;
192 		r0 = r0->next;
193 		os_memset(r0prev->pmk_r0, 0, PMK_LEN);
194 		os_free(r0prev);
195 	}
196 
197 	r1 = cache->pmk_r1;
198 	while (r1) {
199 		r1prev = r1;
200 		r1 = r1->next;
201 		os_memset(r1prev->pmk_r1, 0, PMK_LEN);
202 		os_free(r1prev);
203 	}
204 
205 	os_free(cache);
206 }
207 
208 
209 static int wpa_ft_store_pmk_r0(struct wpa_authenticator *wpa_auth,
210 			       const u8 *spa, const u8 *pmk_r0,
211 			       const u8 *pmk_r0_name, int pairwise)
212 {
213 	struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
214 	struct wpa_ft_pmk_r0_sa *r0;
215 
216 	/* TODO: add expiration and limit on number of entries in cache */
217 
218 	r0 = os_zalloc(sizeof(*r0));
219 	if (r0 == NULL)
220 		return -1;
221 
222 	os_memcpy(r0->pmk_r0, pmk_r0, PMK_LEN);
223 	os_memcpy(r0->pmk_r0_name, pmk_r0_name, WPA_PMK_NAME_LEN);
224 	os_memcpy(r0->spa, spa, ETH_ALEN);
225 	r0->pairwise = pairwise;
226 
227 	r0->next = cache->pmk_r0;
228 	cache->pmk_r0 = r0;
229 
230 	return 0;
231 }
232 
233 
234 static int wpa_ft_fetch_pmk_r0(struct wpa_authenticator *wpa_auth,
235 			       const u8 *spa, const u8 *pmk_r0_name,
236 			       u8 *pmk_r0, int *pairwise)
237 {
238 	struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
239 	struct wpa_ft_pmk_r0_sa *r0;
240 
241 	r0 = cache->pmk_r0;
242 	while (r0) {
243 		if (os_memcmp(r0->spa, spa, ETH_ALEN) == 0 &&
244 		    os_memcmp(r0->pmk_r0_name, pmk_r0_name, WPA_PMK_NAME_LEN)
245 		    == 0) {
246 			os_memcpy(pmk_r0, r0->pmk_r0, PMK_LEN);
247 			if (pairwise)
248 				*pairwise = r0->pairwise;
249 			return 0;
250 		}
251 
252 		r0 = r0->next;
253 	}
254 
255 	return -1;
256 }
257 
258 
259 static int wpa_ft_store_pmk_r1(struct wpa_authenticator *wpa_auth,
260 			       const u8 *spa, const u8 *pmk_r1,
261 			       const u8 *pmk_r1_name, int pairwise)
262 {
263 	struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
264 	struct wpa_ft_pmk_r1_sa *r1;
265 
266 	/* TODO: add expiration and limit on number of entries in cache */
267 
268 	r1 = os_zalloc(sizeof(*r1));
269 	if (r1 == NULL)
270 		return -1;
271 
272 	os_memcpy(r1->pmk_r1, pmk_r1, PMK_LEN);
273 	os_memcpy(r1->pmk_r1_name, pmk_r1_name, WPA_PMK_NAME_LEN);
274 	os_memcpy(r1->spa, spa, ETH_ALEN);
275 	r1->pairwise = pairwise;
276 
277 	r1->next = cache->pmk_r1;
278 	cache->pmk_r1 = r1;
279 
280 	return 0;
281 }
282 
283 
284 static int wpa_ft_fetch_pmk_r1(struct wpa_authenticator *wpa_auth,
285 			       const u8 *spa, const u8 *pmk_r1_name,
286 			       u8 *pmk_r1, int *pairwise)
287 {
288 	struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
289 	struct wpa_ft_pmk_r1_sa *r1;
290 
291 	r1 = cache->pmk_r1;
292 	while (r1) {
293 		if (os_memcmp(r1->spa, spa, ETH_ALEN) == 0 &&
294 		    os_memcmp(r1->pmk_r1_name, pmk_r1_name, WPA_PMK_NAME_LEN)
295 		    == 0) {
296 			os_memcpy(pmk_r1, r1->pmk_r1, PMK_LEN);
297 			if (pairwise)
298 				*pairwise = r1->pairwise;
299 			return 0;
300 		}
301 
302 		r1 = r1->next;
303 	}
304 
305 	return -1;
306 }
307 
308 
309 static int wpa_ft_pull_pmk_r1(struct wpa_authenticator *wpa_auth,
310 			      const u8 *s1kh_id, const u8 *r0kh_id,
311 			      size_t r0kh_id_len, const u8 *pmk_r0_name)
312 {
313 	struct ft_remote_r0kh *r0kh;
314 	struct ft_r0kh_r1kh_pull_frame frame, f;
315 
316 	r0kh = wpa_auth->conf.r0kh_list;
317 	while (r0kh) {
318 		if (r0kh->id_len == r0kh_id_len &&
319 		    os_memcmp(r0kh->id, r0kh_id, r0kh_id_len) == 0)
320 			break;
321 		r0kh = r0kh->next;
322 	}
323 	if (r0kh == NULL)
324 		return -1;
325 
326 	wpa_printf(MSG_DEBUG, "FT: Send PMK-R1 pull request to remote R0KH "
327 		   "address " MACSTR, MAC2STR(r0kh->addr));
328 
329 	os_memset(&frame, 0, sizeof(frame));
330 	frame.frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
331 	frame.packet_type = FT_PACKET_R0KH_R1KH_PULL;
332 	frame.data_length = host_to_le16(FT_R0KH_R1KH_PULL_DATA_LEN);
333 	os_memcpy(frame.ap_address, wpa_auth->addr, ETH_ALEN);
334 
335 	/* aes_wrap() does not support inplace encryption, so use a temporary
336 	 * buffer for the data. */
337 	if (os_get_random(f.nonce, sizeof(f.nonce))) {
338 		wpa_printf(MSG_DEBUG, "FT: Failed to get random data for "
339 			   "nonce");
340 		return -1;
341 	}
342 	os_memcpy(f.pmk_r0_name, pmk_r0_name, WPA_PMK_NAME_LEN);
343 	os_memcpy(f.r1kh_id, wpa_auth->conf.r1_key_holder, FT_R1KH_ID_LEN);
344 	os_memcpy(f.s1kh_id, s1kh_id, ETH_ALEN);
345 
346 	if (aes_wrap(r0kh->key, (FT_R0KH_R1KH_PULL_DATA_LEN + 7) / 8,
347 		     f.nonce, frame.nonce) < 0)
348 		return -1;
349 
350 	wpa_ft_rrb_send(wpa_auth, r0kh->addr, (u8 *) &frame, sizeof(frame));
351 
352 	return 0;
353 }
354 
355 
356 int wpa_auth_derive_ptk_ft(struct wpa_state_machine *sm, const u8 *pmk,
357 			   struct wpa_ptk *ptk, size_t ptk_len)
358 {
359 	u8 pmk_r0[PMK_LEN], pmk_r0_name[WPA_PMK_NAME_LEN];
360 	u8 pmk_r1[PMK_LEN];
361 	u8 ptk_name[WPA_PMK_NAME_LEN];
362 	const u8 *mdid = sm->wpa_auth->conf.mobility_domain;
363 	const u8 *r0kh = sm->wpa_auth->conf.r0_key_holder;
364 	size_t r0kh_len = sm->wpa_auth->conf.r0_key_holder_len;
365 	const u8 *r1kh = sm->wpa_auth->conf.r1_key_holder;
366 	const u8 *ssid = sm->wpa_auth->conf.ssid;
367 	size_t ssid_len = sm->wpa_auth->conf.ssid_len;
368 
369 
370 	if (sm->xxkey_len == 0) {
371 		wpa_printf(MSG_DEBUG, "FT: XXKey not available for key "
372 			   "derivation");
373 		return -1;
374 	}
375 
376 	wpa_derive_pmk_r0(sm->xxkey, sm->xxkey_len, ssid, ssid_len, mdid,
377 			  r0kh, r0kh_len, sm->addr, pmk_r0, pmk_r0_name);
378 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R0", pmk_r0, PMK_LEN);
379 	wpa_hexdump(MSG_DEBUG, "FT: PMKR0Name", pmk_r0_name, WPA_PMK_NAME_LEN);
380 	wpa_ft_store_pmk_r0(sm->wpa_auth, sm->addr, pmk_r0, pmk_r0_name,
381 			    sm->pairwise);
382 
383 	wpa_derive_pmk_r1(pmk_r0, pmk_r0_name, r1kh, sm->addr,
384 			  pmk_r1, sm->pmk_r1_name);
385 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1", pmk_r1, PMK_LEN);
386 	wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name", sm->pmk_r1_name,
387 		    WPA_PMK_NAME_LEN);
388 	wpa_ft_store_pmk_r1(sm->wpa_auth, sm->addr, pmk_r1, sm->pmk_r1_name,
389 			    sm->pairwise);
390 
391 	wpa_pmk_r1_to_ptk(pmk_r1, sm->SNonce, sm->ANonce, sm->addr,
392 			  sm->wpa_auth->addr, sm->pmk_r1_name,
393 			  (u8 *) ptk, ptk_len, ptk_name);
394 	wpa_hexdump_key(MSG_DEBUG, "FT: PTK", (u8 *) ptk, ptk_len);
395 	wpa_hexdump(MSG_DEBUG, "FT: PTKName", ptk_name, WPA_PMK_NAME_LEN);
396 
397 	return 0;
398 }
399 
400 
401 static inline int wpa_auth_get_seqnum(struct wpa_authenticator *wpa_auth,
402 				      const u8 *addr, int idx, u8 *seq)
403 {
404 	if (wpa_auth->cb.get_seqnum == NULL)
405 		return -1;
406 	return wpa_auth->cb.get_seqnum(wpa_auth->cb.ctx, addr, idx, seq);
407 }
408 
409 
410 static u8 * wpa_ft_gtk_subelem(struct wpa_state_machine *sm, size_t *len)
411 {
412 	u8 *subelem;
413 	struct wpa_group *gsm = sm->group;
414 	size_t subelem_len, pad_len;
415 	const u8 *key;
416 	size_t key_len;
417 	u8 keybuf[32];
418 
419 	key_len = gsm->GTK_len;
420 	if (key_len > sizeof(keybuf))
421 		return NULL;
422 
423 	/*
424 	 * Pad key for AES Key Wrap if it is not multiple of 8 bytes or is less
425 	 * than 16 bytes.
426 	 */
427 	pad_len = key_len % 8;
428 	if (pad_len)
429 		pad_len = 8 - pad_len;
430 	if (key_len + pad_len < 16)
431 		pad_len += 8;
432 	if (pad_len) {
433 		os_memcpy(keybuf, gsm->GTK[gsm->GN - 1], key_len);
434 		os_memset(keybuf + key_len, 0, pad_len);
435 		keybuf[key_len] = 0xdd;
436 		key_len += pad_len;
437 		key = keybuf;
438 	} else
439 		key = gsm->GTK[gsm->GN - 1];
440 
441 	/*
442 	 * Sub-elem ID[1] | Length[1] | Key Info[2] | Key Length[1] | RSC[8] |
443 	 * Key[5..32].
444 	 */
445 	subelem_len = 13 + key_len + 8;
446 	subelem = os_zalloc(subelem_len);
447 	if (subelem == NULL)
448 		return NULL;
449 
450 	subelem[0] = FTIE_SUBELEM_GTK;
451 	subelem[1] = 11 + key_len + 8;
452 	/* Key ID in B0-B1 of Key Info */
453 	WPA_PUT_LE16(&subelem[2], gsm->GN & 0x03);
454 	subelem[4] = gsm->GTK_len;
455 	wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN, subelem + 5);
456 	if (aes_wrap(sm->PTK.kek, key_len / 8, key, subelem + 13)) {
457 		os_free(subelem);
458 		return NULL;
459 	}
460 
461 	*len = subelem_len;
462 	return subelem;
463 }
464 
465 
466 #ifdef CONFIG_IEEE80211W
467 static u8 * wpa_ft_igtk_subelem(struct wpa_state_machine *sm, size_t *len)
468 {
469 	u8 *subelem, *pos;
470 	struct wpa_group *gsm = sm->group;
471 	size_t subelem_len;
472 
473 	/* Sub-elem ID[1] | Length[1] | KeyID[2] | IPN[6] | Key Length[1] |
474 	 * Key[16+8] */
475 	subelem_len = 1 + 1 + 2 + 6 + 1 + WPA_IGTK_LEN + 8;
476 	subelem = os_zalloc(subelem_len);
477 	if (subelem == NULL)
478 		return NULL;
479 
480 	pos = subelem;
481 	*pos++ = FTIE_SUBELEM_IGTK;
482 	*pos++ = subelem_len - 2;
483 	WPA_PUT_LE16(pos, gsm->GN_igtk);
484 	pos += 2;
485 	wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN_igtk, pos);
486 	pos += 6;
487 	*pos++ = WPA_IGTK_LEN;
488 	if (aes_wrap(sm->PTK.kek, WPA_IGTK_LEN / 8,
489 		     gsm->IGTK[gsm->GN_igtk - 4], pos)) {
490 		os_free(subelem);
491 		return NULL;
492 	}
493 
494 	*len = subelem_len;
495 	return subelem;
496 }
497 #endif /* CONFIG_IEEE80211W */
498 
499 
500 static u8 * wpa_ft_process_rdie(u8 *pos, u8 *end, u8 id, u8 descr_count,
501 				const u8 *ies, size_t ies_len)
502 {
503 	struct ieee802_11_elems parse;
504 	struct rsn_rdie *rdie;
505 
506 	wpa_printf(MSG_DEBUG, "FT: Resource Request: id=%d descr_count=%d",
507 		   id, descr_count);
508 	wpa_hexdump(MSG_MSGDUMP, "FT: Resource descriptor IE(s)",
509 		    ies, ies_len);
510 
511 	if (end - pos < (int) sizeof(*rdie)) {
512 		wpa_printf(MSG_ERROR, "FT: Not enough room for response RDIE");
513 		return pos;
514 	}
515 
516 	*pos++ = WLAN_EID_RIC_DATA;
517 	*pos++ = sizeof(*rdie);
518 	rdie = (struct rsn_rdie *) pos;
519 	rdie->id = id;
520 	rdie->descr_count = 0;
521 	rdie->status_code = host_to_le16(WLAN_STATUS_SUCCESS);
522 	pos += sizeof(*rdie);
523 
524 	if (ieee802_11_parse_elems((u8 *) ies, ies_len, &parse, 1) ==
525 	    ParseFailed) {
526 		wpa_printf(MSG_DEBUG, "FT: Failed to parse request IEs");
527 		rdie->status_code =
528 			host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
529 		return pos;
530 	}
531 
532 #ifdef NEED_AP_MLME
533 	if (parse.wmm_tspec) {
534 		struct wmm_tspec_element *tspec;
535 		int res;
536 
537 		if (parse.wmm_tspec_len + 2 < (int) sizeof(*tspec)) {
538 			wpa_printf(MSG_DEBUG, "FT: Too short WMM TSPEC IE "
539 				   "(%d)", (int) parse.wmm_tspec_len);
540 			rdie->status_code =
541 				host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
542 			return pos;
543 		}
544 		if (end - pos < (int) sizeof(*tspec)) {
545 			wpa_printf(MSG_ERROR, "FT: Not enough room for "
546 				   "response TSPEC");
547 			rdie->status_code =
548 				host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
549 			return pos;
550 		}
551 		tspec = (struct wmm_tspec_element *) pos;
552 		os_memcpy(tspec, parse.wmm_tspec - 2, sizeof(*tspec));
553 		res = wmm_process_tspec(tspec);
554 		wpa_printf(MSG_DEBUG, "FT: ADDTS processing result: %d", res);
555 		if (res == WMM_ADDTS_STATUS_INVALID_PARAMETERS)
556 			rdie->status_code =
557 				host_to_le16(WLAN_STATUS_INVALID_PARAMETERS);
558 		else if (res == WMM_ADDTS_STATUS_REFUSED)
559 			rdie->status_code =
560 				host_to_le16(WLAN_STATUS_REQUEST_DECLINED);
561 		else {
562 			/* TSPEC accepted; include updated TSPEC in response */
563 			rdie->descr_count = 1;
564 			pos += sizeof(*tspec);
565 		}
566 		return pos;
567 	}
568 #endif /* NEED_AP_MLME */
569 
570 	wpa_printf(MSG_DEBUG, "FT: No supported resource requested");
571 	rdie->status_code = host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
572 	return pos;
573 }
574 
575 
576 static u8 * wpa_ft_process_ric(u8 *pos, u8 *end, const u8 *ric, size_t ric_len)
577 {
578 	const u8 *rpos, *start;
579 	const struct rsn_rdie *rdie;
580 
581 	wpa_hexdump(MSG_MSGDUMP, "FT: RIC Request", ric, ric_len);
582 
583 	rpos = ric;
584 	while (rpos + sizeof(*rdie) < ric + ric_len) {
585 		if (rpos[0] != WLAN_EID_RIC_DATA || rpos[1] < sizeof(*rdie) ||
586 		    rpos + 2 + rpos[1] > ric + ric_len)
587 			break;
588 		rdie = (const struct rsn_rdie *) (rpos + 2);
589 		rpos += 2 + rpos[1];
590 		start = rpos;
591 
592 		while (rpos + 2 <= ric + ric_len &&
593 		       rpos + 2 + rpos[1] <= ric + ric_len) {
594 			if (rpos[0] == WLAN_EID_RIC_DATA)
595 				break;
596 			rpos += 2 + rpos[1];
597 		}
598 		pos = wpa_ft_process_rdie(pos, end, rdie->id,
599 					  rdie->descr_count,
600 					  start, rpos - start);
601 	}
602 
603 	return pos;
604 }
605 
606 
607 u8 * wpa_sm_write_assoc_resp_ies(struct wpa_state_machine *sm, u8 *pos,
608 				 size_t max_len, int auth_alg,
609 				 const u8 *req_ies, size_t req_ies_len)
610 {
611 	u8 *end, *mdie, *ftie, *rsnie = NULL, *r0kh_id, *subelem = NULL;
612 	size_t mdie_len, ftie_len, rsnie_len = 0, r0kh_id_len, subelem_len = 0;
613 	int res;
614 	struct wpa_auth_config *conf;
615 	struct rsn_ftie *_ftie;
616 	struct wpa_ft_ies parse;
617 	u8 *ric_start;
618 	u8 *anonce, *snonce;
619 
620 	if (sm == NULL)
621 		return pos;
622 
623 	conf = &sm->wpa_auth->conf;
624 
625 	if (sm->wpa_key_mgmt != WPA_KEY_MGMT_FT_IEEE8021X &&
626 	    sm->wpa_key_mgmt != WPA_KEY_MGMT_FT_PSK)
627 		return pos;
628 
629 	end = pos + max_len;
630 
631 	if (auth_alg == WLAN_AUTH_FT) {
632 		/*
633 		 * RSN (only present if this is a Reassociation Response and
634 		 * part of a fast BSS transition)
635 		 */
636 		res = wpa_write_rsn_ie(conf, pos, end - pos, sm->pmk_r1_name);
637 		if (res < 0)
638 			return pos;
639 		rsnie = pos;
640 		rsnie_len = res;
641 		pos += res;
642 	}
643 
644 	/* Mobility Domain Information */
645 	res = wpa_write_mdie(conf, pos, end - pos);
646 	if (res < 0)
647 		return pos;
648 	mdie = pos;
649 	mdie_len = res;
650 	pos += res;
651 
652 	/* Fast BSS Transition Information */
653 	if (auth_alg == WLAN_AUTH_FT) {
654 		subelem = wpa_ft_gtk_subelem(sm, &subelem_len);
655 		r0kh_id = sm->r0kh_id;
656 		r0kh_id_len = sm->r0kh_id_len;
657 		anonce = sm->ANonce;
658 		snonce = sm->SNonce;
659 #ifdef CONFIG_IEEE80211W
660 		if (sm->mgmt_frame_prot) {
661 			u8 *igtk;
662 			size_t igtk_len;
663 			u8 *nbuf;
664 			igtk = wpa_ft_igtk_subelem(sm, &igtk_len);
665 			if (igtk == NULL) {
666 				os_free(subelem);
667 				return pos;
668 			}
669 			nbuf = os_realloc(subelem, subelem_len + igtk_len);
670 			if (nbuf == NULL) {
671 				os_free(subelem);
672 				os_free(igtk);
673 				return pos;
674 			}
675 			subelem = nbuf;
676 			os_memcpy(subelem + subelem_len, igtk, igtk_len);
677 			subelem_len += igtk_len;
678 			os_free(igtk);
679 		}
680 #endif /* CONFIG_IEEE80211W */
681 	} else {
682 		r0kh_id = conf->r0_key_holder;
683 		r0kh_id_len = conf->r0_key_holder_len;
684 		anonce = NULL;
685 		snonce = NULL;
686 	}
687 	res = wpa_write_ftie(conf, r0kh_id, r0kh_id_len, anonce, snonce, pos,
688 			     end - pos, subelem, subelem_len);
689 	os_free(subelem);
690 	if (res < 0)
691 		return pos;
692 	ftie = pos;
693 	ftie_len = res;
694 	pos += res;
695 
696 	os_free(sm->assoc_resp_ftie);
697 	sm->assoc_resp_ftie = os_malloc(ftie_len);
698 	if (sm->assoc_resp_ftie)
699 		os_memcpy(sm->assoc_resp_ftie, ftie, ftie_len);
700 
701 	_ftie = (struct rsn_ftie *) (ftie + 2);
702 	if (auth_alg == WLAN_AUTH_FT)
703 		_ftie->mic_control[1] = 3; /* Information element count */
704 
705 	ric_start = pos;
706 	if (wpa_ft_parse_ies(req_ies, req_ies_len, &parse) == 0 && parse.ric) {
707 		pos = wpa_ft_process_ric(pos, end, parse.ric, parse.ric_len);
708 		if (auth_alg == WLAN_AUTH_FT)
709 			_ftie->mic_control[1] +=
710 				ieee802_11_ie_count(ric_start,
711 						    pos - ric_start);
712 	}
713 	if (ric_start == pos)
714 		ric_start = NULL;
715 
716 	if (auth_alg == WLAN_AUTH_FT &&
717 	    wpa_ft_mic(sm->PTK.kck, sm->addr, sm->wpa_auth->addr, 6,
718 		       mdie, mdie_len, ftie, ftie_len,
719 		       rsnie, rsnie_len,
720 		       ric_start, ric_start ? pos - ric_start : 0,
721 		       _ftie->mic) < 0)
722 		wpa_printf(MSG_DEBUG, "FT: Failed to calculate MIC");
723 
724 	return pos;
725 }
726 
727 
728 static int wpa_ft_parse_ftie(const u8 *ie, size_t ie_len,
729 			     struct wpa_ft_ies *parse)
730 {
731 	const u8 *end, *pos;
732 
733 	parse->ftie = ie;
734 	parse->ftie_len = ie_len;
735 
736 	pos = ie + sizeof(struct rsn_ftie);
737 	end = ie + ie_len;
738 
739 	while (pos + 2 <= end && pos + 2 + pos[1] <= end) {
740 		switch (pos[0]) {
741 		case FTIE_SUBELEM_R1KH_ID:
742 			if (pos[1] != FT_R1KH_ID_LEN) {
743 				wpa_printf(MSG_DEBUG, "FT: Invalid R1KH-ID "
744 					   "length in FTIE: %d", pos[1]);
745 				return -1;
746 			}
747 			parse->r1kh_id = pos + 2;
748 			break;
749 		case FTIE_SUBELEM_GTK:
750 			parse->gtk = pos + 2;
751 			parse->gtk_len = pos[1];
752 			break;
753 		case FTIE_SUBELEM_R0KH_ID:
754 			if (pos[1] < 1 || pos[1] > FT_R0KH_ID_MAX_LEN) {
755 				wpa_printf(MSG_DEBUG, "FT: Invalid R0KH-ID "
756 					   "length in FTIE: %d", pos[1]);
757 				return -1;
758 			}
759 			parse->r0kh_id = pos + 2;
760 			parse->r0kh_id_len = pos[1];
761 			break;
762 		}
763 
764 		pos += 2 + pos[1];
765 	}
766 
767 	return 0;
768 }
769 
770 
771 static int wpa_ft_parse_ies(const u8 *ies, size_t ies_len,
772 			    struct wpa_ft_ies *parse)
773 {
774 	const u8 *end, *pos;
775 	struct wpa_ie_data data;
776 	int ret;
777 	const struct rsn_ftie *ftie;
778 	int prot_ie_count = 0;
779 
780 	os_memset(parse, 0, sizeof(*parse));
781 	if (ies == NULL)
782 		return 0;
783 
784 	pos = ies;
785 	end = ies + ies_len;
786 	while (pos + 2 <= end && pos + 2 + pos[1] <= end) {
787 		switch (pos[0]) {
788 		case WLAN_EID_RSN:
789 			parse->rsn = pos + 2;
790 			parse->rsn_len = pos[1];
791 			ret = wpa_parse_wpa_ie_rsn(parse->rsn - 2,
792 						   parse->rsn_len + 2,
793 						   &data);
794 			if (ret < 0) {
795 				wpa_printf(MSG_DEBUG, "FT: Failed to parse "
796 					   "RSN IE: %d", ret);
797 				return -1;
798 			}
799 			if (data.num_pmkid == 1 && data.pmkid)
800 				parse->rsn_pmkid = data.pmkid;
801 			break;
802 		case WLAN_EID_MOBILITY_DOMAIN:
803 			parse->mdie = pos + 2;
804 			parse->mdie_len = pos[1];
805 			break;
806 		case WLAN_EID_FAST_BSS_TRANSITION:
807 			if (pos[1] < sizeof(*ftie))
808 				return -1;
809 			ftie = (const struct rsn_ftie *) (pos + 2);
810 			prot_ie_count = ftie->mic_control[1];
811 			if (wpa_ft_parse_ftie(pos + 2, pos[1], parse) < 0)
812 				return -1;
813 			break;
814 		case WLAN_EID_RIC_DATA:
815 			if (parse->ric == NULL)
816 				parse->ric = pos;
817 		}
818 
819 		pos += 2 + pos[1];
820 	}
821 
822 	if (prot_ie_count == 0)
823 		return 0; /* no MIC */
824 
825 	/*
826 	 * Check that the protected IE count matches with IEs included in the
827 	 * frame.
828 	 */
829 	if (parse->rsn)
830 		prot_ie_count--;
831 	if (parse->mdie)
832 		prot_ie_count--;
833 	if (parse->ftie)
834 		prot_ie_count--;
835 	if (prot_ie_count < 0) {
836 		wpa_printf(MSG_DEBUG, "FT: Some required IEs not included in "
837 			   "the protected IE count");
838 		return -1;
839 	}
840 
841 	if (prot_ie_count == 0 && parse->ric) {
842 		wpa_printf(MSG_DEBUG, "FT: RIC IE(s) in the frame, but not "
843 			   "included in protected IE count");
844 		return -1;
845 	}
846 
847 	/* Determine the end of the RIC IE(s) */
848 	pos = parse->ric;
849 	while (pos && pos + 2 <= end && pos + 2 + pos[1] <= end &&
850 	       prot_ie_count) {
851 		prot_ie_count--;
852 		pos += 2 + pos[1];
853 	}
854 	parse->ric_len = pos - parse->ric;
855 	if (prot_ie_count) {
856 		wpa_printf(MSG_DEBUG, "FT: %d protected IEs missing from "
857 			   "frame", (int) prot_ie_count);
858 		return -1;
859 	}
860 
861 	return 0;
862 }
863 
864 
865 static inline int wpa_auth_set_key(struct wpa_authenticator *wpa_auth,
866 				   int vlan_id,
867 				   enum wpa_alg alg, const u8 *addr, int idx,
868 				   u8 *key, size_t key_len)
869 {
870 	if (wpa_auth->cb.set_key == NULL)
871 		return -1;
872 	return wpa_auth->cb.set_key(wpa_auth->cb.ctx, vlan_id, alg, addr, idx,
873 				    key, key_len);
874 }
875 
876 
877 void wpa_ft_install_ptk(struct wpa_state_machine *sm)
878 {
879 	enum wpa_alg alg;
880 	int klen;
881 
882 	/* MLME-SETKEYS.request(PTK) */
883 	if (sm->pairwise == WPA_CIPHER_TKIP) {
884 		alg = WPA_ALG_TKIP;
885 		klen = 32;
886 	} else if (sm->pairwise == WPA_CIPHER_CCMP) {
887 		alg = WPA_ALG_CCMP;
888 		klen = 16;
889 	} else {
890 		wpa_printf(MSG_DEBUG, "FT: Unknown pairwise alg 0x%x - skip "
891 			   "PTK configuration", sm->pairwise);
892 		return;
893 	}
894 
895 	/* FIX: add STA entry to kernel/driver here? The set_key will fail
896 	 * most likely without this.. At the moment, STA entry is added only
897 	 * after association has been completed. This function will be called
898 	 * again after association to get the PTK configured, but that could be
899 	 * optimized by adding the STA entry earlier.
900 	 */
901 	if (wpa_auth_set_key(sm->wpa_auth, 0, alg, sm->addr, 0,
902 			     sm->PTK.tk1, klen))
903 		return;
904 
905 	/* FIX: MLME-SetProtection.Request(TA, Tx_Rx) */
906 	sm->pairwise_set = TRUE;
907 }
908 
909 
910 static u16 wpa_ft_process_auth_req(struct wpa_state_machine *sm,
911 				   const u8 *ies, size_t ies_len,
912 				   u8 **resp_ies, size_t *resp_ies_len)
913 {
914 	struct rsn_mdie *mdie;
915 	struct rsn_ftie *ftie;
916 	u8 pmk_r1[PMK_LEN], pmk_r1_name[WPA_PMK_NAME_LEN];
917 	u8 ptk_name[WPA_PMK_NAME_LEN];
918 	struct wpa_auth_config *conf;
919 	struct wpa_ft_ies parse;
920 	size_t buflen, ptk_len;
921 	int ret;
922 	u8 *pos, *end;
923 	int pairwise;
924 
925 	*resp_ies = NULL;
926 	*resp_ies_len = 0;
927 
928 	sm->pmk_r1_name_valid = 0;
929 	conf = &sm->wpa_auth->conf;
930 
931 	wpa_hexdump(MSG_DEBUG, "FT: Received authentication frame IEs",
932 		    ies, ies_len);
933 
934 	if (wpa_ft_parse_ies(ies, ies_len, &parse) < 0) {
935 		wpa_printf(MSG_DEBUG, "FT: Failed to parse FT IEs");
936 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
937 	}
938 
939 	mdie = (struct rsn_mdie *) parse.mdie;
940 	if (mdie == NULL || parse.mdie_len < sizeof(*mdie) ||
941 	    os_memcmp(mdie->mobility_domain,
942 		      sm->wpa_auth->conf.mobility_domain,
943 		      MOBILITY_DOMAIN_ID_LEN) != 0) {
944 		wpa_printf(MSG_DEBUG, "FT: Invalid MDIE");
945 		return WLAN_STATUS_INVALID_MDIE;
946 	}
947 
948 	ftie = (struct rsn_ftie *) parse.ftie;
949 	if (ftie == NULL || parse.ftie_len < sizeof(*ftie)) {
950 		wpa_printf(MSG_DEBUG, "FT: Invalid FTIE");
951 		return WLAN_STATUS_INVALID_FTIE;
952 	}
953 
954 	os_memcpy(sm->SNonce, ftie->snonce, WPA_NONCE_LEN);
955 
956 	if (parse.r0kh_id == NULL) {
957 		wpa_printf(MSG_DEBUG, "FT: Invalid FTIE - no R0KH-ID");
958 		return WLAN_STATUS_INVALID_FTIE;
959 	}
960 
961 	wpa_hexdump(MSG_DEBUG, "FT: STA R0KH-ID",
962 		    parse.r0kh_id, parse.r0kh_id_len);
963 	os_memcpy(sm->r0kh_id, parse.r0kh_id, parse.r0kh_id_len);
964 	sm->r0kh_id_len = parse.r0kh_id_len;
965 
966 	if (parse.rsn_pmkid == NULL) {
967 		wpa_printf(MSG_DEBUG, "FT: No PMKID in RSNIE");
968 		return WLAN_STATUS_INVALID_PMKID;
969 	}
970 
971 	wpa_hexdump(MSG_DEBUG, "FT: Requested PMKR0Name",
972 		    parse.rsn_pmkid, WPA_PMK_NAME_LEN);
973 	wpa_derive_pmk_r1_name(parse.rsn_pmkid,
974 			       sm->wpa_auth->conf.r1_key_holder, sm->addr,
975 			       pmk_r1_name);
976 	wpa_hexdump(MSG_DEBUG, "FT: Derived requested PMKR1Name",
977 		    pmk_r1_name, WPA_PMK_NAME_LEN);
978 
979 	if (wpa_ft_fetch_pmk_r1(sm->wpa_auth, sm->addr, pmk_r1_name, pmk_r1,
980 		    &pairwise) < 0) {
981 		if (wpa_ft_pull_pmk_r1(sm->wpa_auth, sm->addr, sm->r0kh_id,
982 				       sm->r0kh_id_len, parse.rsn_pmkid) < 0) {
983 			wpa_printf(MSG_DEBUG, "FT: Did not have matching "
984 				   "PMK-R1 and unknown R0KH-ID");
985 			return WLAN_STATUS_INVALID_PMKID;
986 		}
987 
988 		/*
989 		 * TODO: Should return "status pending" (and the caller should
990 		 * not send out response now). The real response will be sent
991 		 * once the response from R0KH is received.
992 		 */
993 		return WLAN_STATUS_INVALID_PMKID;
994 	}
995 
996 	wpa_hexdump_key(MSG_DEBUG, "FT: Selected PMK-R1", pmk_r1, PMK_LEN);
997 	sm->pmk_r1_name_valid = 1;
998 	os_memcpy(sm->pmk_r1_name, pmk_r1_name, WPA_PMK_NAME_LEN);
999 
1000 	if (os_get_random(sm->ANonce, WPA_NONCE_LEN)) {
1001 		wpa_printf(MSG_DEBUG, "FT: Failed to get random data for "
1002 			   "ANonce");
1003 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1004 	}
1005 
1006 	wpa_hexdump(MSG_DEBUG, "FT: Received SNonce",
1007 		    sm->SNonce, WPA_NONCE_LEN);
1008 	wpa_hexdump(MSG_DEBUG, "FT: Generated ANonce",
1009 		    sm->ANonce, WPA_NONCE_LEN);
1010 
1011 	ptk_len = pairwise != WPA_CIPHER_CCMP ? 64 : 48;
1012 	wpa_pmk_r1_to_ptk(pmk_r1, sm->SNonce, sm->ANonce, sm->addr,
1013 			  sm->wpa_auth->addr, pmk_r1_name,
1014 			  (u8 *) &sm->PTK, ptk_len, ptk_name);
1015 	wpa_hexdump_key(MSG_DEBUG, "FT: PTK",
1016 			(u8 *) &sm->PTK, ptk_len);
1017 	wpa_hexdump(MSG_DEBUG, "FT: PTKName", ptk_name, WPA_PMK_NAME_LEN);
1018 
1019 	sm->pairwise = pairwise;
1020 	wpa_ft_install_ptk(sm);
1021 
1022 	buflen = 2 + sizeof(struct rsn_mdie) + 2 + sizeof(struct rsn_ftie) +
1023 		2 + FT_R1KH_ID_LEN + 200;
1024 	*resp_ies = os_zalloc(buflen);
1025 	if (*resp_ies == NULL) {
1026 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1027 	}
1028 
1029 	pos = *resp_ies;
1030 	end = *resp_ies + buflen;
1031 
1032 	ret = wpa_write_rsn_ie(conf, pos, end - pos, parse.rsn_pmkid);
1033 	if (ret < 0) {
1034 		os_free(*resp_ies);
1035 		*resp_ies = NULL;
1036 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1037 	}
1038 	pos += ret;
1039 
1040 	ret = wpa_write_mdie(conf, pos, end - pos);
1041 	if (ret < 0) {
1042 		os_free(*resp_ies);
1043 		*resp_ies = NULL;
1044 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1045 	}
1046 	pos += ret;
1047 
1048 	ret = wpa_write_ftie(conf, parse.r0kh_id, parse.r0kh_id_len,
1049 			     sm->ANonce, sm->SNonce, pos, end - pos, NULL, 0);
1050 	if (ret < 0) {
1051 		os_free(*resp_ies);
1052 		*resp_ies = NULL;
1053 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1054 	}
1055 	pos += ret;
1056 
1057 	*resp_ies_len = pos - *resp_ies;
1058 
1059 	return WLAN_STATUS_SUCCESS;
1060 }
1061 
1062 
1063 void wpa_ft_process_auth(struct wpa_state_machine *sm, const u8 *bssid,
1064 			 u16 auth_transaction, const u8 *ies, size_t ies_len,
1065 			 void (*cb)(void *ctx, const u8 *dst, const u8 *bssid,
1066 				    u16 auth_transaction, u16 status,
1067 				    const u8 *ies, size_t ies_len),
1068 			 void *ctx)
1069 {
1070 	u16 status;
1071 	u8 *resp_ies;
1072 	size_t resp_ies_len;
1073 
1074 	if (sm == NULL) {
1075 		wpa_printf(MSG_DEBUG, "FT: Received authentication frame, but "
1076 			   "WPA SM not available");
1077 		return;
1078 	}
1079 
1080 	wpa_printf(MSG_DEBUG, "FT: Received authentication frame: STA=" MACSTR
1081 		   " BSSID=" MACSTR " transaction=%d",
1082 		   MAC2STR(sm->addr), MAC2STR(bssid), auth_transaction);
1083 	status = wpa_ft_process_auth_req(sm, ies, ies_len, &resp_ies,
1084 					 &resp_ies_len);
1085 
1086 	wpa_printf(MSG_DEBUG, "FT: FT authentication response: dst=" MACSTR
1087 		   " auth_transaction=%d status=%d",
1088 		   MAC2STR(sm->addr), auth_transaction + 1, status);
1089 	wpa_hexdump(MSG_DEBUG, "FT: Response IEs", resp_ies, resp_ies_len);
1090 	cb(ctx, sm->addr, bssid, auth_transaction + 1, status,
1091 	   resp_ies, resp_ies_len);
1092 	os_free(resp_ies);
1093 }
1094 
1095 
1096 u16 wpa_ft_validate_reassoc(struct wpa_state_machine *sm, const u8 *ies,
1097 			    size_t ies_len)
1098 {
1099 	struct wpa_ft_ies parse;
1100 	struct rsn_mdie *mdie;
1101 	struct rsn_ftie *ftie;
1102 	u8 mic[16];
1103 	unsigned int count;
1104 
1105 	if (sm == NULL)
1106 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1107 
1108 	wpa_hexdump(MSG_DEBUG, "FT: Reassoc Req IEs", ies, ies_len);
1109 
1110 	if (wpa_ft_parse_ies(ies, ies_len, &parse) < 0) {
1111 		wpa_printf(MSG_DEBUG, "FT: Failed to parse FT IEs");
1112 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1113 	}
1114 
1115 	if (parse.rsn == NULL) {
1116 		wpa_printf(MSG_DEBUG, "FT: No RSNIE in Reassoc Req");
1117 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1118 	}
1119 
1120 	if (parse.rsn_pmkid == NULL) {
1121 		wpa_printf(MSG_DEBUG, "FT: No PMKID in RSNIE");
1122 		return WLAN_STATUS_INVALID_PMKID;
1123 	}
1124 
1125 	if (os_memcmp(parse.rsn_pmkid, sm->pmk_r1_name, WPA_PMK_NAME_LEN) != 0)
1126 	{
1127 		wpa_printf(MSG_DEBUG, "FT: PMKID in Reassoc Req did not match "
1128 			   "with the PMKR1Name derived from auth request");
1129 		return WLAN_STATUS_INVALID_PMKID;
1130 	}
1131 
1132 	mdie = (struct rsn_mdie *) parse.mdie;
1133 	if (mdie == NULL || parse.mdie_len < sizeof(*mdie) ||
1134 	    os_memcmp(mdie->mobility_domain,
1135 		      sm->wpa_auth->conf.mobility_domain,
1136 		      MOBILITY_DOMAIN_ID_LEN) != 0) {
1137 		wpa_printf(MSG_DEBUG, "FT: Invalid MDIE");
1138 		return WLAN_STATUS_INVALID_MDIE;
1139 	}
1140 
1141 	ftie = (struct rsn_ftie *) parse.ftie;
1142 	if (ftie == NULL || parse.ftie_len < sizeof(*ftie)) {
1143 		wpa_printf(MSG_DEBUG, "FT: Invalid FTIE");
1144 		return WLAN_STATUS_INVALID_FTIE;
1145 	}
1146 
1147 	if (os_memcmp(ftie->snonce, sm->SNonce, WPA_NONCE_LEN) != 0) {
1148 		wpa_printf(MSG_DEBUG, "FT: SNonce mismatch in FTIE");
1149 		wpa_hexdump(MSG_DEBUG, "FT: Received SNonce",
1150 			    ftie->snonce, WPA_NONCE_LEN);
1151 		wpa_hexdump(MSG_DEBUG, "FT: Expected SNonce",
1152 			    sm->SNonce, WPA_NONCE_LEN);
1153 		return -1;
1154 	}
1155 
1156 	if (os_memcmp(ftie->anonce, sm->ANonce, WPA_NONCE_LEN) != 0) {
1157 		wpa_printf(MSG_DEBUG, "FT: ANonce mismatch in FTIE");
1158 		wpa_hexdump(MSG_DEBUG, "FT: Received ANonce",
1159 			    ftie->anonce, WPA_NONCE_LEN);
1160 		wpa_hexdump(MSG_DEBUG, "FT: Expected ANonce",
1161 			    sm->ANonce, WPA_NONCE_LEN);
1162 		return -1;
1163 	}
1164 
1165 
1166 	if (parse.r0kh_id == NULL) {
1167 		wpa_printf(MSG_DEBUG, "FT: No R0KH-ID subelem in FTIE");
1168 		return -1;
1169 	}
1170 
1171 	if (parse.r0kh_id_len != sm->r0kh_id_len ||
1172 	    os_memcmp(parse.r0kh_id, sm->r0kh_id, parse.r0kh_id_len) != 0) {
1173 		wpa_printf(MSG_DEBUG, "FT: R0KH-ID in FTIE did not match with "
1174 			   "the current R0KH-ID");
1175 		wpa_hexdump(MSG_DEBUG, "FT: R0KH-ID in FTIE",
1176 			    parse.r0kh_id, parse.r0kh_id_len);
1177 		wpa_hexdump(MSG_DEBUG, "FT: The current R0KH-ID",
1178 			    sm->r0kh_id, sm->r0kh_id_len);
1179 		return -1;
1180 	}
1181 
1182 	if (parse.r1kh_id == NULL) {
1183 		wpa_printf(MSG_DEBUG, "FT: No R1KH-ID subelem in FTIE");
1184 		return -1;
1185 	}
1186 
1187 	if (os_memcmp(parse.r1kh_id, sm->wpa_auth->conf.r1_key_holder,
1188 		      FT_R1KH_ID_LEN) != 0) {
1189 		wpa_printf(MSG_DEBUG, "FT: Unknown R1KH-ID used in "
1190 			   "ReassocReq");
1191 		wpa_hexdump(MSG_DEBUG, "FT: R1KH-ID in FTIE",
1192 			    parse.r1kh_id, FT_R1KH_ID_LEN);
1193 		wpa_hexdump(MSG_DEBUG, "FT: Expected R1KH-ID",
1194 			    sm->wpa_auth->conf.r1_key_holder, FT_R1KH_ID_LEN);
1195 		return -1;
1196 	}
1197 
1198 	if (parse.rsn_pmkid == NULL ||
1199 	    os_memcmp(parse.rsn_pmkid, sm->pmk_r1_name, WPA_PMK_NAME_LEN)) {
1200 		wpa_printf(MSG_DEBUG, "FT: No matching PMKR1Name (PMKID) in "
1201 			   "RSNIE (pmkid=%d)", !!parse.rsn_pmkid);
1202 		return -1;
1203 	}
1204 
1205 	count = 3;
1206 	if (parse.ric)
1207 		count++;
1208 	if (ftie->mic_control[1] != count) {
1209 		wpa_printf(MSG_DEBUG, "FT: Unexpected IE count in MIC "
1210 			   "Control: received %u expected %u",
1211 			   ftie->mic_control[1], count);
1212 		return -1;
1213 	}
1214 
1215 	if (wpa_ft_mic(sm->PTK.kck, sm->addr, sm->wpa_auth->addr, 5,
1216 		       parse.mdie - 2, parse.mdie_len + 2,
1217 		       parse.ftie - 2, parse.ftie_len + 2,
1218 		       parse.rsn - 2, parse.rsn_len + 2,
1219 		       parse.ric, parse.ric_len,
1220 		       mic) < 0) {
1221 		wpa_printf(MSG_DEBUG, "FT: Failed to calculate MIC");
1222 		return WLAN_STATUS_UNSPECIFIED_FAILURE;
1223 	}
1224 
1225 	if (os_memcmp(mic, ftie->mic, 16) != 0) {
1226 		wpa_printf(MSG_DEBUG, "FT: Invalid MIC in FTIE");
1227 		wpa_hexdump(MSG_MSGDUMP, "FT: Received MIC", ftie->mic, 16);
1228 		wpa_hexdump(MSG_MSGDUMP, "FT: Calculated MIC", mic, 16);
1229 		return WLAN_STATUS_INVALID_FTIE;
1230 	}
1231 
1232 	return WLAN_STATUS_SUCCESS;
1233 }
1234 
1235 
1236 int wpa_ft_action_rx(struct wpa_state_machine *sm, const u8 *data, size_t len)
1237 {
1238 	const u8 *sta_addr, *target_ap;
1239 	const u8 *ies;
1240 	size_t ies_len;
1241 	u8 action;
1242 	struct ft_rrb_frame *frame;
1243 
1244 	if (sm == NULL)
1245 		return -1;
1246 
1247 	/*
1248 	 * data: Category[1] Action[1] STA_Address[6] Target_AP_Address[6]
1249 	 * FT Request action frame body[variable]
1250 	 */
1251 
1252 	if (len < 14) {
1253 		wpa_printf(MSG_DEBUG, "FT: Too short FT Action frame "
1254 			   "(len=%lu)", (unsigned long) len);
1255 		return -1;
1256 	}
1257 
1258 	action = data[1];
1259 	sta_addr = data + 2;
1260 	target_ap = data + 8;
1261 	ies = data + 14;
1262 	ies_len = len - 14;
1263 
1264 	wpa_printf(MSG_DEBUG, "FT: Received FT Action frame (STA=" MACSTR
1265 		   " Target AP=" MACSTR " Action=%d)",
1266 		   MAC2STR(sta_addr), MAC2STR(target_ap), action);
1267 
1268 	if (os_memcmp(sta_addr, sm->addr, ETH_ALEN) != 0) {
1269 		wpa_printf(MSG_DEBUG, "FT: Mismatch in FT Action STA address: "
1270 			   "STA=" MACSTR " STA-Address=" MACSTR,
1271 			   MAC2STR(sm->addr), MAC2STR(sta_addr));
1272 		return -1;
1273 	}
1274 
1275 	/*
1276 	 * Do some sanity checking on the target AP address (not own and not
1277 	 * broadcast. This could be extended to filter based on a list of known
1278 	 * APs in the MD (if such a list were configured).
1279 	 */
1280 	if ((target_ap[0] & 0x01) ||
1281 	    os_memcmp(target_ap, sm->wpa_auth->addr, ETH_ALEN) == 0) {
1282 		wpa_printf(MSG_DEBUG, "FT: Invalid Target AP in FT Action "
1283 			   "frame");
1284 		return -1;
1285 	}
1286 
1287 	wpa_hexdump(MSG_MSGDUMP, "FT: Action frame body", ies, ies_len);
1288 
1289 	/* RRB - Forward action frame to the target AP */
1290 	frame = os_malloc(sizeof(*frame) + len);
1291 	frame->frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
1292 	frame->packet_type = FT_PACKET_REQUEST;
1293 	frame->action_length = host_to_le16(len);
1294 	os_memcpy(frame->ap_address, sm->wpa_auth->addr, ETH_ALEN);
1295 	os_memcpy(frame + 1, data, len);
1296 
1297 	wpa_ft_rrb_send(sm->wpa_auth, target_ap, (u8 *) frame,
1298 			sizeof(*frame) + len);
1299 	os_free(frame);
1300 
1301 	return 0;
1302 }
1303 
1304 
1305 static int wpa_ft_rrb_rx_request(struct wpa_authenticator *wpa_auth,
1306 				 const u8 *current_ap, const u8 *sta_addr,
1307 				 const u8 *body, size_t len)
1308 {
1309 	struct wpa_state_machine *sm;
1310 	u16 status;
1311 	u8 *resp_ies, *pos;
1312 	size_t resp_ies_len, rlen;
1313 	struct ft_rrb_frame *frame;
1314 
1315 	sm = wpa_ft_add_sta(wpa_auth, sta_addr);
1316 	if (sm == NULL) {
1317 		wpa_printf(MSG_DEBUG, "FT: Failed to add new STA based on "
1318 			   "RRB Request");
1319 		return -1;
1320 	}
1321 
1322 	wpa_hexdump(MSG_MSGDUMP, "FT: RRB Request Frame body", body, len);
1323 
1324 	status = wpa_ft_process_auth_req(sm, body, len, &resp_ies,
1325 					 &resp_ies_len);
1326 
1327 	wpa_printf(MSG_DEBUG, "FT: RRB authentication response: STA=" MACSTR
1328 		   " CurrentAP=" MACSTR " status=%d",
1329 		   MAC2STR(sm->addr), MAC2STR(current_ap), status);
1330 	wpa_hexdump(MSG_DEBUG, "FT: Response IEs", resp_ies, resp_ies_len);
1331 
1332 	/* RRB - Forward action frame response to the Current AP */
1333 
1334 	/*
1335 	 * data: Category[1] Action[1] STA_Address[6] Target_AP_Address[6]
1336 	 * Status_Code[2] FT Request action frame body[variable]
1337 	 */
1338 	rlen = 2 + 2 * ETH_ALEN + 2 + resp_ies_len;
1339 
1340 	frame = os_malloc(sizeof(*frame) + rlen);
1341 	frame->frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
1342 	frame->packet_type = FT_PACKET_RESPONSE;
1343 	frame->action_length = host_to_le16(rlen);
1344 	os_memcpy(frame->ap_address, wpa_auth->addr, ETH_ALEN);
1345 	pos = (u8 *) (frame + 1);
1346 	*pos++ = WLAN_ACTION_FT;
1347 	*pos++ = 2; /* Action: Response */
1348 	os_memcpy(pos, sta_addr, ETH_ALEN);
1349 	pos += ETH_ALEN;
1350 	os_memcpy(pos, wpa_auth->addr, ETH_ALEN);
1351 	pos += ETH_ALEN;
1352 	WPA_PUT_LE16(pos, status);
1353 	pos += 2;
1354 	if (resp_ies) {
1355 		os_memcpy(pos, resp_ies, resp_ies_len);
1356 		os_free(resp_ies);
1357 	}
1358 
1359 	wpa_ft_rrb_send(wpa_auth, current_ap, (u8 *) frame,
1360 			sizeof(*frame) + rlen);
1361 	os_free(frame);
1362 
1363 	return 0;
1364 }
1365 
1366 
1367 static int wpa_ft_rrb_rx_pull(struct wpa_authenticator *wpa_auth,
1368 			      const u8 *src_addr,
1369 			      const u8 *data, size_t data_len)
1370 {
1371 	struct ft_r0kh_r1kh_pull_frame *frame, f;
1372 	struct ft_remote_r1kh *r1kh;
1373 	struct ft_r0kh_r1kh_resp_frame resp, r;
1374 	u8 pmk_r0[PMK_LEN];
1375 	int pairwise;
1376 
1377 	wpa_printf(MSG_DEBUG, "FT: Received PMK-R1 pull");
1378 
1379 	if (data_len < sizeof(*frame))
1380 		return -1;
1381 
1382 	r1kh = wpa_auth->conf.r1kh_list;
1383 	while (r1kh) {
1384 		if (os_memcmp(r1kh->addr, src_addr, ETH_ALEN) == 0)
1385 			break;
1386 		r1kh = r1kh->next;
1387 	}
1388 	if (r1kh == NULL) {
1389 		wpa_printf(MSG_DEBUG, "FT: No matching R1KH address found for "
1390 			   "PMK-R1 pull source address " MACSTR,
1391 			   MAC2STR(src_addr));
1392 		return -1;
1393 	}
1394 
1395 	frame = (struct ft_r0kh_r1kh_pull_frame *) data;
1396 	/* aes_unwrap() does not support inplace decryption, so use a temporary
1397 	 * buffer for the data. */
1398 	if (aes_unwrap(r1kh->key, (FT_R0KH_R1KH_PULL_DATA_LEN + 7) / 8,
1399 		       frame->nonce, f.nonce) < 0) {
1400 		wpa_printf(MSG_DEBUG, "FT: Failed to decrypt PMK-R1 pull "
1401 			   "request from " MACSTR, MAC2STR(src_addr));
1402 		return -1;
1403 	}
1404 
1405 	wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 pull - nonce",
1406 		    f.nonce, sizeof(f.nonce));
1407 	wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 pull - PMKR0Name",
1408 		    f.pmk_r0_name, WPA_PMK_NAME_LEN);
1409 	wpa_printf(MSG_DEBUG, "FT: PMK-R1 pull - R1KH-ID=" MACSTR "S1KH-ID="
1410 		   MACSTR, MAC2STR(f.r1kh_id), MAC2STR(f.s1kh_id));
1411 
1412 	os_memset(&resp, 0, sizeof(resp));
1413 	resp.frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
1414 	resp.packet_type = FT_PACKET_R0KH_R1KH_RESP;
1415 	resp.data_length = host_to_le16(FT_R0KH_R1KH_RESP_DATA_LEN);
1416 	os_memcpy(resp.ap_address, wpa_auth->addr, ETH_ALEN);
1417 
1418 	/* aes_wrap() does not support inplace encryption, so use a temporary
1419 	 * buffer for the data. */
1420 	os_memcpy(r.nonce, f.nonce, sizeof(f.nonce));
1421 	os_memcpy(r.r1kh_id, f.r1kh_id, FT_R1KH_ID_LEN);
1422 	os_memcpy(r.s1kh_id, f.s1kh_id, ETH_ALEN);
1423 	if (wpa_ft_fetch_pmk_r0(wpa_auth, f.s1kh_id, f.pmk_r0_name, pmk_r0,
1424 				&pairwise) < 0) {
1425 		wpa_printf(MSG_DEBUG, "FT: No matching PMKR0Name found for "
1426 			   "PMK-R1 pull");
1427 		return -1;
1428 	}
1429 
1430 	wpa_derive_pmk_r1(pmk_r0, f.pmk_r0_name, f.r1kh_id, f.s1kh_id,
1431 			  r.pmk_r1, r.pmk_r1_name);
1432 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1", r.pmk_r1, PMK_LEN);
1433 	wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name", r.pmk_r1_name,
1434 		    WPA_PMK_NAME_LEN);
1435 	r.pairwise = host_to_le16(pairwise);
1436 
1437 	if (aes_wrap(r1kh->key, (FT_R0KH_R1KH_RESP_DATA_LEN + 7) / 8,
1438 		     r.nonce, resp.nonce) < 0) {
1439 		os_memset(pmk_r0, 0, PMK_LEN);
1440 		return -1;
1441 	}
1442 
1443 	os_memset(pmk_r0, 0, PMK_LEN);
1444 
1445 	wpa_ft_rrb_send(wpa_auth, src_addr, (u8 *) &resp, sizeof(resp));
1446 
1447 	return 0;
1448 }
1449 
1450 
1451 static int wpa_ft_rrb_rx_resp(struct wpa_authenticator *wpa_auth,
1452 			      const u8 *src_addr,
1453 			      const u8 *data, size_t data_len)
1454 {
1455 	struct ft_r0kh_r1kh_resp_frame *frame, f;
1456 	struct ft_remote_r0kh *r0kh;
1457 	int pairwise;
1458 
1459 	wpa_printf(MSG_DEBUG, "FT: Received PMK-R1 pull response");
1460 
1461 	if (data_len < sizeof(*frame))
1462 		return -1;
1463 
1464 	r0kh = wpa_auth->conf.r0kh_list;
1465 	while (r0kh) {
1466 		if (os_memcmp(r0kh->addr, src_addr, ETH_ALEN) == 0)
1467 			break;
1468 		r0kh = r0kh->next;
1469 	}
1470 	if (r0kh == NULL) {
1471 		wpa_printf(MSG_DEBUG, "FT: No matching R0KH address found for "
1472 			   "PMK-R0 pull response source address " MACSTR,
1473 			   MAC2STR(src_addr));
1474 		return -1;
1475 	}
1476 
1477 	frame = (struct ft_r0kh_r1kh_resp_frame *) data;
1478 	/* aes_unwrap() does not support inplace decryption, so use a temporary
1479 	 * buffer for the data. */
1480 	if (aes_unwrap(r0kh->key, (FT_R0KH_R1KH_RESP_DATA_LEN + 7) / 8,
1481 		       frame->nonce, f.nonce) < 0) {
1482 		wpa_printf(MSG_DEBUG, "FT: Failed to decrypt PMK-R1 pull "
1483 			   "response from " MACSTR, MAC2STR(src_addr));
1484 		return -1;
1485 	}
1486 
1487 	if (os_memcmp(f.r1kh_id, wpa_auth->conf.r1_key_holder, FT_R1KH_ID_LEN)
1488 	    != 0) {
1489 		wpa_printf(MSG_DEBUG, "FT: PMK-R1 pull response did not use a "
1490 			   "matching R1KH-ID");
1491 		return -1;
1492 	}
1493 
1494 	/* TODO: verify that <nonce,s1kh_id> matches with a pending request
1495 	 * and call this requests callback function to finish request
1496 	 * processing */
1497 
1498 	pairwise = le_to_host16(f.pairwise);
1499 	wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 pull - nonce",
1500 		    f.nonce, sizeof(f.nonce));
1501 	wpa_printf(MSG_DEBUG, "FT: PMK-R1 pull - R1KH-ID=" MACSTR "S1KH-ID="
1502 		   MACSTR " pairwise=0x%x",
1503 		   MAC2STR(f.r1kh_id), MAC2STR(f.s1kh_id), pairwise);
1504 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1 pull - PMK-R1",
1505 			f.pmk_r1, PMK_LEN);
1506 	wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 pull - PMKR1Name",
1507 			f.pmk_r1_name, WPA_PMK_NAME_LEN);
1508 
1509 	wpa_ft_store_pmk_r1(wpa_auth, f.s1kh_id, f.pmk_r1, f.pmk_r1_name,
1510 			    pairwise);
1511 	os_memset(f.pmk_r1, 0, PMK_LEN);
1512 
1513 	return 0;
1514 }
1515 
1516 
1517 static int wpa_ft_rrb_rx_push(struct wpa_authenticator *wpa_auth,
1518 			      const u8 *src_addr,
1519 			      const u8 *data, size_t data_len)
1520 {
1521 	struct ft_r0kh_r1kh_push_frame *frame, f;
1522 	struct ft_remote_r0kh *r0kh;
1523 	struct os_time now;
1524 	os_time_t tsend;
1525 	int pairwise;
1526 
1527 	wpa_printf(MSG_DEBUG, "FT: Received PMK-R1 push");
1528 
1529 	if (data_len < sizeof(*frame))
1530 		return -1;
1531 
1532 	r0kh = wpa_auth->conf.r0kh_list;
1533 	while (r0kh) {
1534 		if (os_memcmp(r0kh->addr, src_addr, ETH_ALEN) == 0)
1535 			break;
1536 		r0kh = r0kh->next;
1537 	}
1538 	if (r0kh == NULL) {
1539 		wpa_printf(MSG_DEBUG, "FT: No matching R0KH address found for "
1540 			   "PMK-R0 push source address " MACSTR,
1541 			   MAC2STR(src_addr));
1542 		return -1;
1543 	}
1544 
1545 	frame = (struct ft_r0kh_r1kh_push_frame *) data;
1546 	/* aes_unwrap() does not support inplace decryption, so use a temporary
1547 	 * buffer for the data. */
1548 	if (aes_unwrap(r0kh->key, (FT_R0KH_R1KH_PUSH_DATA_LEN + 7) / 8,
1549 		       frame->timestamp, f.timestamp) < 0) {
1550 		wpa_printf(MSG_DEBUG, "FT: Failed to decrypt PMK-R1 push from "
1551 			   MACSTR, MAC2STR(src_addr));
1552 		return -1;
1553 	}
1554 
1555 	os_get_time(&now);
1556 	tsend = WPA_GET_LE32(f.timestamp);
1557 	if ((now.sec > tsend && now.sec - tsend > 60) ||
1558 	    (now.sec < tsend && tsend - now.sec > 60)) {
1559 		wpa_printf(MSG_DEBUG, "FT: PMK-R1 push did not have a valid "
1560 			   "timestamp: sender time %d own time %d\n",
1561 			   (int) tsend, (int) now.sec);
1562 		return -1;
1563 	}
1564 
1565 	if (os_memcmp(f.r1kh_id, wpa_auth->conf.r1_key_holder, FT_R1KH_ID_LEN)
1566 	    != 0) {
1567 		wpa_printf(MSG_DEBUG, "FT: PMK-R1 push did not use a matching "
1568 			   "R1KH-ID (received " MACSTR " own " MACSTR ")",
1569 			   MAC2STR(f.r1kh_id),
1570 			   MAC2STR(wpa_auth->conf.r1_key_holder));
1571 		return -1;
1572 	}
1573 
1574 	pairwise = le_to_host16(f.pairwise);
1575 	wpa_printf(MSG_DEBUG, "FT: PMK-R1 push - R1KH-ID=" MACSTR " S1KH-ID="
1576 		   MACSTR " pairwise=0x%x",
1577 		   MAC2STR(f.r1kh_id), MAC2STR(f.s1kh_id), pairwise);
1578 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1 push - PMK-R1",
1579 			f.pmk_r1, PMK_LEN);
1580 	wpa_hexdump(MSG_DEBUG, "FT: PMK-R1 push - PMKR1Name",
1581 			f.pmk_r1_name, WPA_PMK_NAME_LEN);
1582 
1583 	wpa_ft_store_pmk_r1(wpa_auth, f.s1kh_id, f.pmk_r1, f.pmk_r1_name,
1584 			    pairwise);
1585 	os_memset(f.pmk_r1, 0, PMK_LEN);
1586 
1587 	return 0;
1588 }
1589 
1590 
1591 int wpa_ft_rrb_rx(struct wpa_authenticator *wpa_auth, const u8 *src_addr,
1592 		  const u8 *data, size_t data_len)
1593 {
1594 	struct ft_rrb_frame *frame;
1595 	u16 alen;
1596 	const u8 *pos, *end, *start;
1597 	u8 action;
1598 	const u8 *sta_addr, *target_ap_addr;
1599 
1600 	wpa_printf(MSG_DEBUG, "FT: RRB received frame from remote AP " MACSTR,
1601 		   MAC2STR(src_addr));
1602 
1603 	if (data_len < sizeof(*frame)) {
1604 		wpa_printf(MSG_DEBUG, "FT: Too short RRB frame (data_len=%lu)",
1605 			   (unsigned long) data_len);
1606 		return -1;
1607 	}
1608 
1609 	pos = data;
1610 	frame = (struct ft_rrb_frame *) pos;
1611 	pos += sizeof(*frame);
1612 
1613 	alen = le_to_host16(frame->action_length);
1614 	wpa_printf(MSG_DEBUG, "FT: RRB frame - frame_type=%d packet_type=%d "
1615 		   "action_length=%d ap_address=" MACSTR,
1616 		   frame->frame_type, frame->packet_type, alen,
1617 		   MAC2STR(frame->ap_address));
1618 
1619 	if (frame->frame_type != RSN_REMOTE_FRAME_TYPE_FT_RRB) {
1620 		/* Discard frame per IEEE Std 802.11r-2008, 11A.10.3 */
1621 		wpa_printf(MSG_DEBUG, "FT: RRB discarded frame with "
1622 			   "unrecognized type %d", frame->frame_type);
1623 		return -1;
1624 	}
1625 
1626 	if (alen > data_len - sizeof(*frame)) {
1627 		wpa_printf(MSG_DEBUG, "FT: RRB frame too short for action "
1628 			   "frame");
1629 		return -1;
1630 	}
1631 
1632 	if (frame->packet_type == FT_PACKET_R0KH_R1KH_PULL)
1633 		return wpa_ft_rrb_rx_pull(wpa_auth, src_addr, data, data_len);
1634 	if (frame->packet_type == FT_PACKET_R0KH_R1KH_RESP)
1635 		return wpa_ft_rrb_rx_resp(wpa_auth, src_addr, data, data_len);
1636 	if (frame->packet_type == FT_PACKET_R0KH_R1KH_PUSH)
1637 		return wpa_ft_rrb_rx_push(wpa_auth, src_addr, data, data_len);
1638 
1639 	wpa_hexdump(MSG_MSGDUMP, "FT: RRB - FT Action frame", pos, alen);
1640 
1641 	if (alen < 1 + 1 + 2 * ETH_ALEN) {
1642 		wpa_printf(MSG_DEBUG, "FT: Too short RRB frame (not enough "
1643 			   "room for Action Frame body); alen=%lu",
1644 			   (unsigned long) alen);
1645 		return -1;
1646 	}
1647 	start = pos;
1648 	end = pos + alen;
1649 
1650 	if (*pos != WLAN_ACTION_FT) {
1651 		wpa_printf(MSG_DEBUG, "FT: Unexpected Action frame category "
1652 			   "%d", *pos);
1653 		return -1;
1654 	}
1655 
1656 	pos++;
1657 	action = *pos++;
1658 	sta_addr = pos;
1659 	pos += ETH_ALEN;
1660 	target_ap_addr = pos;
1661 	pos += ETH_ALEN;
1662 	wpa_printf(MSG_DEBUG, "FT: RRB Action Frame: action=%d sta_addr="
1663 		   MACSTR " target_ap_addr=" MACSTR,
1664 		   action, MAC2STR(sta_addr), MAC2STR(target_ap_addr));
1665 
1666 	if (frame->packet_type == FT_PACKET_REQUEST) {
1667 		wpa_printf(MSG_DEBUG, "FT: FT Packet Type - Request");
1668 
1669 		if (action != 1) {
1670 			wpa_printf(MSG_DEBUG, "FT: Unexpected Action %d in "
1671 				   "RRB Request", action);
1672 			return -1;
1673 		}
1674 
1675 		if (os_memcmp(target_ap_addr, wpa_auth->addr, ETH_ALEN) != 0) {
1676 			wpa_printf(MSG_DEBUG, "FT: Target AP address in the "
1677 				   "RRB Request does not match with own "
1678 				   "address");
1679 			return -1;
1680 		}
1681 
1682 		if (wpa_ft_rrb_rx_request(wpa_auth, frame->ap_address,
1683 					  sta_addr, pos, end - pos) < 0)
1684 			return -1;
1685 	} else if (frame->packet_type == FT_PACKET_RESPONSE) {
1686 		u16 status_code;
1687 
1688 		if (end - pos < 2) {
1689 			wpa_printf(MSG_DEBUG, "FT: Not enough room for status "
1690 				   "code in RRB Response");
1691 			return -1;
1692 		}
1693 		status_code = WPA_GET_LE16(pos);
1694 		pos += 2;
1695 
1696 		wpa_printf(MSG_DEBUG, "FT: FT Packet Type - Response "
1697 			   "(status_code=%d)", status_code);
1698 
1699 		if (wpa_ft_action_send(wpa_auth, sta_addr, start, alen) < 0)
1700 			return -1;
1701 	} else {
1702 		wpa_printf(MSG_DEBUG, "FT: RRB discarded frame with unknown "
1703 			   "packet_type %d", frame->packet_type);
1704 		return -1;
1705 	}
1706 
1707 	return 0;
1708 }
1709 
1710 
1711 static void wpa_ft_generate_pmk_r1(struct wpa_authenticator *wpa_auth,
1712 				   struct wpa_ft_pmk_r0_sa *pmk_r0,
1713 				   struct ft_remote_r1kh *r1kh,
1714 				   const u8 *s1kh_id, int pairwise)
1715 {
1716 	struct ft_r0kh_r1kh_push_frame frame, f;
1717 	struct os_time now;
1718 
1719 	os_memset(&frame, 0, sizeof(frame));
1720 	frame.frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
1721 	frame.packet_type = FT_PACKET_R0KH_R1KH_PUSH;
1722 	frame.data_length = host_to_le16(FT_R0KH_R1KH_PUSH_DATA_LEN);
1723 	os_memcpy(frame.ap_address, wpa_auth->addr, ETH_ALEN);
1724 
1725 	/* aes_wrap() does not support inplace encryption, so use a temporary
1726 	 * buffer for the data. */
1727 	os_memcpy(f.r1kh_id, r1kh->id, FT_R1KH_ID_LEN);
1728 	os_memcpy(f.s1kh_id, s1kh_id, ETH_ALEN);
1729 	os_memcpy(f.pmk_r0_name, pmk_r0->pmk_r0_name, WPA_PMK_NAME_LEN);
1730 	wpa_derive_pmk_r1(pmk_r0->pmk_r0, pmk_r0->pmk_r0_name, r1kh->id,
1731 			  s1kh_id, f.pmk_r1, f.pmk_r1_name);
1732 	wpa_printf(MSG_DEBUG, "FT: R1KH-ID " MACSTR, MAC2STR(r1kh->id));
1733 	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1", f.pmk_r1, PMK_LEN);
1734 	wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name", f.pmk_r1_name,
1735 		    WPA_PMK_NAME_LEN);
1736 	os_get_time(&now);
1737 	WPA_PUT_LE32(f.timestamp, now.sec);
1738 	f.pairwise = host_to_le16(pairwise);
1739 	if (aes_wrap(r1kh->key, (FT_R0KH_R1KH_PUSH_DATA_LEN + 7) / 8,
1740 		     f.timestamp, frame.timestamp) < 0)
1741 		return;
1742 
1743 	wpa_ft_rrb_send(wpa_auth, r1kh->addr, (u8 *) &frame, sizeof(frame));
1744 }
1745 
1746 
1747 void wpa_ft_push_pmk_r1(struct wpa_authenticator *wpa_auth, const u8 *addr)
1748 {
1749 	struct wpa_ft_pmk_r0_sa *r0;
1750 	struct ft_remote_r1kh *r1kh;
1751 
1752 	if (!wpa_auth->conf.pmk_r1_push)
1753 		return;
1754 
1755 	r0 = wpa_auth->ft_pmk_cache->pmk_r0;
1756 	while (r0) {
1757 		if (os_memcmp(r0->spa, addr, ETH_ALEN) == 0)
1758 			break;
1759 		r0 = r0->next;
1760 	}
1761 
1762 	if (r0 == NULL || r0->pmk_r1_pushed)
1763 		return;
1764 	r0->pmk_r1_pushed = 1;
1765 
1766 	wpa_printf(MSG_DEBUG, "FT: Deriving and pushing PMK-R1 keys to R1KHs "
1767 		   "for STA " MACSTR, MAC2STR(addr));
1768 
1769 	r1kh = wpa_auth->conf.r1kh_list;
1770 	while (r1kh) {
1771 		wpa_ft_generate_pmk_r1(wpa_auth, r0, r1kh, addr, r0->pairwise);
1772 		r1kh = r1kh->next;
1773 	}
1774 }
1775 
1776 #endif /* CONFIG_IEEE80211R */
1777