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