1/*
2 * hostapd - IEEE 802.11r - Fast BSS Transition
3 * Copyright (c) 2004-2018, 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 "utils/list.h"
14#include "common/ieee802_11_defs.h"
15#include "common/ieee802_11_common.h"
16#include "common/ocv.h"
17#include "common/wpa_ctrl.h"
18#include "drivers/driver.h"
19#include "crypto/aes.h"
20#include "crypto/aes_siv.h"
21#include "crypto/aes_wrap.h"
22#include "crypto/sha384.h"
23#include "crypto/random.h"
24#include "ap_config.h"
25#include "ieee802_11.h"
26#include "wmm.h"
27#include "wpa_auth.h"
28#include "wpa_auth_i.h"
29#include "pmksa_cache_auth.h"
30
31
32#ifdef CONFIG_IEEE80211R_AP
33
34const unsigned int ftRRBseqTimeout = 10;
35const unsigned int ftRRBmaxQueueLen = 100;
36
37
38static int wpa_ft_send_rrb_auth_resp(struct wpa_state_machine *sm,
39				     const u8 *current_ap, const u8 *sta_addr,
40				     u16 status, const u8 *resp_ies,
41				     size_t resp_ies_len);
42static void ft_finish_pull(struct wpa_state_machine *sm);
43static void wpa_ft_expire_pull(void *eloop_ctx, void *timeout_ctx);
44static void wpa_ft_rrb_seq_timeout(void *eloop_ctx, void *timeout_ctx);
45
46struct tlv_list {
47	u16 type;
48	size_t len;
49	const u8 *data;
50};
51
52
53/**
54 * wpa_ft_rrb_decrypt - Decrypt FT RRB message
55 * @key: AES-SIV key for AEAD
56 * @key_len: Length of key in octets
57 * @enc: Pointer to encrypted TLVs
58 * @enc_len: Length of encrypted TLVs in octets
59 * @auth: Pointer to authenticated TLVs
60 * @auth_len: Length of authenticated TLVs in octets
61 * @src_addr: MAC address of the frame sender
62 * @type: Vendor-specific subtype of the RRB frame (FT_PACKET_*)
63 * @plain: Pointer to return the pointer to the allocated plaintext buffer;
64 *	needs to be freed by the caller if not NULL;
65 *	will only be returned on success
66 * @plain_len: Pointer to return the length of the allocated plaintext buffer
67 *	in octets
68 * Returns: 0 on success, -1 on error
69 */
70static int wpa_ft_rrb_decrypt(const u8 *key, const size_t key_len,
71			      const u8 *enc, size_t enc_len,
72			      const u8 *auth, const size_t auth_len,
73			      const u8 *src_addr, u8 type,
74			      u8 **plain, size_t *plain_size)
75{
76	const u8 *ad[3] = { src_addr, auth, &type };
77	size_t ad_len[3] = { ETH_ALEN, auth_len, sizeof(type) };
78
79	wpa_printf(MSG_DEBUG, "FT(RRB): src_addr=" MACSTR " type=%u",
80		   MAC2STR(src_addr), type);
81	wpa_hexdump_key(MSG_DEBUG, "FT(RRB): decrypt using key", key, key_len);
82	wpa_hexdump(MSG_DEBUG, "FT(RRB): encrypted TLVs", enc, enc_len);
83	wpa_hexdump(MSG_DEBUG, "FT(RRB): authenticated TLVs", auth, auth_len);
84
85	if (!key) { /* skip decryption */
86		*plain = os_memdup(enc, enc_len);
87		if (enc_len > 0 && !*plain)
88			goto err;
89
90		*plain_size = enc_len;
91
92		return 0;
93	}
94
95	*plain = NULL;
96
97	/* SIV overhead */
98	if (enc_len < AES_BLOCK_SIZE)
99		goto err;
100
101	*plain = os_zalloc(enc_len - AES_BLOCK_SIZE);
102	if (!*plain)
103		goto err;
104
105	if (aes_siv_decrypt(key, key_len, enc, enc_len, 3, ad, ad_len,
106			    *plain) < 0) {
107		if (enc_len < AES_BLOCK_SIZE + 2)
108			goto err;
109
110		/* Try to work around Ethernet devices that add extra
111		 * two octet padding even if the frame is longer than
112		 * the minimum Ethernet frame. */
113		enc_len -= 2;
114		if (aes_siv_decrypt(key, key_len, enc, enc_len, 3, ad, ad_len,
115				    *plain) < 0)
116			goto err;
117	}
118
119	*plain_size = enc_len - AES_BLOCK_SIZE;
120	wpa_hexdump_key(MSG_DEBUG, "FT(RRB): decrypted TLVs",
121			*plain, *plain_size);
122	return 0;
123err:
124	os_free(*plain);
125	*plain = NULL;
126	*plain_size = 0;
127
128	wpa_printf(MSG_ERROR, "FT(RRB): Failed to decrypt");
129
130	return -1;
131}
132
133
134/* get first tlv record in packet matching type
135 * @data (decrypted) packet
136 * @return 0 on success else -1
137 */
138static int wpa_ft_rrb_get_tlv(const u8 *plain, size_t plain_len,
139			      u16 type, size_t *tlv_len, const u8 **tlv_data)
140{
141	const struct ft_rrb_tlv *f;
142	size_t left;
143	le16 type16;
144	size_t len;
145
146	left = plain_len;
147	type16 = host_to_le16(type);
148
149	while (left >= sizeof(*f)) {
150		f = (const struct ft_rrb_tlv *) plain;
151
152		left -= sizeof(*f);
153		plain += sizeof(*f);
154		len = le_to_host16(f->len);
155
156		if (left < len) {
157			wpa_printf(MSG_DEBUG, "FT: RRB message truncated");
158			break;
159		}
160
161		if (f->type == type16) {
162			*tlv_len = len;
163			*tlv_data = plain;
164			return 0;
165		}
166
167		left -= len;
168		plain += len;
169	}
170
171	return -1;
172}
173
174
175static void wpa_ft_rrb_dump(const u8 *plain, const size_t plain_len)
176{
177	const struct ft_rrb_tlv *f;
178	size_t left;
179	size_t len;
180
181	left = plain_len;
182
183	wpa_printf(MSG_DEBUG, "FT: RRB dump message");
184	while (left >= sizeof(*f)) {
185		f = (const struct ft_rrb_tlv *) plain;
186
187		left -= sizeof(*f);
188		plain += sizeof(*f);
189		len = le_to_host16(f->len);
190
191		wpa_printf(MSG_DEBUG, "FT: RRB TLV type = %d, len = %zu",
192			   le_to_host16(f->type), len);
193
194		if (left < len) {
195			wpa_printf(MSG_DEBUG,
196				   "FT: RRB message truncated: left %zu bytes, need %zu",
197				   left, len);
198			break;
199		}
200
201		wpa_hexdump(MSG_DEBUG, "FT: RRB TLV data", plain, len);
202
203		left -= len;
204		plain += len;
205	}
206
207	if (left > 0)
208		wpa_hexdump(MSG_DEBUG, "FT: RRB TLV padding", plain, left);
209
210	wpa_printf(MSG_DEBUG, "FT: RRB dump message end");
211}
212
213
214static int cmp_int(const void *a, const void *b)
215{
216	int x, y;
217
218	x = *((int *) a);
219	y = *((int *) b);
220	return x - y;
221}
222
223
224static int wpa_ft_rrb_get_tlv_vlan(const u8 *plain, const size_t plain_len,
225				   struct vlan_description *vlan)
226{
227	struct ft_rrb_tlv *f;
228	size_t left;
229	size_t len;
230	int taggedidx;
231	int vlan_id;
232	int type;
233
234	left = plain_len;
235	taggedidx = 0;
236	os_memset(vlan, 0, sizeof(*vlan));
237
238	while (left >= sizeof(*f)) {
239		f = (struct ft_rrb_tlv *) plain;
240
241		left -= sizeof(*f);
242		plain += sizeof(*f);
243
244		len = le_to_host16(f->len);
245		type = le_to_host16(f->type);
246
247		if (left < len) {
248			wpa_printf(MSG_DEBUG, "FT: RRB message truncated");
249			return -1;
250		}
251
252		if (type != FT_RRB_VLAN_UNTAGGED && type != FT_RRB_VLAN_TAGGED)
253			goto skip;
254
255		if (type == FT_RRB_VLAN_UNTAGGED && len != sizeof(le16)) {
256			wpa_printf(MSG_DEBUG,
257				   "FT: RRB VLAN_UNTAGGED invalid length");
258			return -1;
259		}
260
261		if (type == FT_RRB_VLAN_TAGGED && len % sizeof(le16) != 0) {
262			wpa_printf(MSG_DEBUG,
263				   "FT: RRB VLAN_TAGGED invalid length");
264			return -1;
265		}
266
267		while (len >= sizeof(le16)) {
268			vlan_id = WPA_GET_LE16(plain);
269			plain += sizeof(le16);
270			left -= sizeof(le16);
271			len -= sizeof(le16);
272
273			if (vlan_id <= 0 || vlan_id > MAX_VLAN_ID) {
274				wpa_printf(MSG_DEBUG,
275					   "FT: RRB VLAN ID invalid %d",
276					   vlan_id);
277				continue;
278			}
279
280			if (type == FT_RRB_VLAN_UNTAGGED)
281				vlan->untagged = vlan_id;
282
283			if (type == FT_RRB_VLAN_TAGGED &&
284			    taggedidx < MAX_NUM_TAGGED_VLAN) {
285				vlan->tagged[taggedidx] = vlan_id;
286				taggedidx++;
287			} else if (type == FT_RRB_VLAN_TAGGED) {
288				wpa_printf(MSG_DEBUG, "FT: RRB too many VLANs");
289			}
290		}
291
292	skip:
293		left -= len;
294		plain += len;
295	}
296
297	if (taggedidx)
298		qsort(vlan->tagged, taggedidx, sizeof(int), cmp_int);
299
300	vlan->notempty = vlan->untagged || vlan->tagged[0];
301
302	return 0;
303}
304
305
306static size_t wpa_ft_tlv_len(const struct tlv_list *tlvs)
307{
308	size_t tlv_len = 0;
309	int i;
310
311	if (!tlvs)
312		return 0;
313
314	for (i = 0; tlvs[i].type != FT_RRB_LAST_EMPTY; i++) {
315		tlv_len += sizeof(struct ft_rrb_tlv);
316		tlv_len += tlvs[i].len;
317	}
318
319	return tlv_len;
320}
321
322
323static size_t wpa_ft_tlv_lin(const struct tlv_list *tlvs, u8 *start,
324			     u8 *endpos)
325{
326	int i;
327	size_t tlv_len;
328	struct ft_rrb_tlv *hdr;
329	u8 *pos;
330
331	if (!tlvs)
332		return 0;
333
334	tlv_len = 0;
335	pos = start;
336	for (i = 0; tlvs[i].type != FT_RRB_LAST_EMPTY; i++) {
337		if (tlv_len + sizeof(*hdr) > (size_t) (endpos - start))
338			return tlv_len;
339		tlv_len += sizeof(*hdr);
340		hdr = (struct ft_rrb_tlv *) pos;
341		hdr->type = host_to_le16(tlvs[i].type);
342		hdr->len = host_to_le16(tlvs[i].len);
343		pos = start + tlv_len;
344
345		if (tlv_len + tlvs[i].len > (size_t) (endpos - start))
346			return tlv_len;
347		if (tlvs[i].len == 0)
348			continue;
349		tlv_len += tlvs[i].len;
350		os_memcpy(pos, tlvs[i].data, tlvs[i].len);
351		pos = start + tlv_len;
352	}
353
354	return tlv_len;
355}
356
357
358static size_t wpa_ft_vlan_len(const struct vlan_description *vlan)
359{
360	size_t tlv_len = 0;
361	int i;
362
363	if (!vlan || !vlan->notempty)
364		return 0;
365
366	if (vlan->untagged) {
367		tlv_len += sizeof(struct ft_rrb_tlv);
368		tlv_len += sizeof(le16);
369	}
370	if (vlan->tagged[0])
371		tlv_len += sizeof(struct ft_rrb_tlv);
372	for (i = 0; i < MAX_NUM_TAGGED_VLAN && vlan->tagged[i]; i++)
373		tlv_len += sizeof(le16);
374
375	return tlv_len;
376}
377
378
379static size_t wpa_ft_vlan_lin(const struct vlan_description *vlan,
380			      u8 *start, u8 *endpos)
381{
382	size_t tlv_len;
383	int i, len;
384	struct ft_rrb_tlv *hdr;
385	u8 *pos = start;
386
387	if (!vlan || !vlan->notempty)
388		return 0;
389
390	tlv_len = 0;
391	if (vlan->untagged) {
392		tlv_len += sizeof(*hdr);
393		if (start + tlv_len > endpos)
394			return tlv_len;
395		hdr = (struct ft_rrb_tlv *) pos;
396		hdr->type = host_to_le16(FT_RRB_VLAN_UNTAGGED);
397		hdr->len = host_to_le16(sizeof(le16));
398		pos = start + tlv_len;
399
400		tlv_len += sizeof(le16);
401		if (start + tlv_len > endpos)
402			return tlv_len;
403		WPA_PUT_LE16(pos, vlan->untagged);
404		pos = start + tlv_len;
405	}
406
407	if (!vlan->tagged[0])
408		return tlv_len;
409
410	tlv_len += sizeof(*hdr);
411	if (start + tlv_len > endpos)
412		return tlv_len;
413	hdr = (struct ft_rrb_tlv *) pos;
414	hdr->type = host_to_le16(FT_RRB_VLAN_TAGGED);
415	len = 0; /* len is computed below */
416	pos = start + tlv_len;
417
418	for (i = 0; i < MAX_NUM_TAGGED_VLAN && vlan->tagged[i]; i++) {
419		tlv_len += sizeof(le16);
420		if (start + tlv_len > endpos)
421			break;
422		len += sizeof(le16);
423		WPA_PUT_LE16(pos, vlan->tagged[i]);
424		pos = start + tlv_len;
425	}
426
427	hdr->len = host_to_le16(len);
428
429	return tlv_len;
430}
431
432
433static int wpa_ft_rrb_lin(const struct tlv_list *tlvs1,
434			  const struct tlv_list *tlvs2,
435			  const struct vlan_description *vlan,
436			  u8 **plain, size_t *plain_len)
437{
438	u8 *pos, *endpos;
439	size_t tlv_len;
440
441	tlv_len = wpa_ft_tlv_len(tlvs1);
442	tlv_len += wpa_ft_tlv_len(tlvs2);
443	tlv_len += wpa_ft_vlan_len(vlan);
444
445	*plain_len = tlv_len;
446	*plain = os_zalloc(tlv_len);
447	if (!*plain) {
448		wpa_printf(MSG_ERROR, "FT: Failed to allocate plaintext");
449		goto err;
450	}
451
452	pos = *plain;
453	endpos = *plain + tlv_len;
454	pos += wpa_ft_tlv_lin(tlvs1, pos, endpos);
455	pos += wpa_ft_tlv_lin(tlvs2, pos, endpos);
456	pos += wpa_ft_vlan_lin(vlan, pos, endpos);
457
458	/* validity check */
459	if (pos != endpos) {
460		wpa_printf(MSG_ERROR, "FT: Length error building RRB");
461		goto err;
462	}
463
464	return 0;
465
466err:
467	os_free(*plain);
468	*plain = NULL;
469	*plain_len = 0;
470	return -1;
471}
472
473
474static int wpa_ft_rrb_encrypt(const u8 *key, const size_t key_len,
475			      const u8 *plain, const size_t plain_len,
476			      const u8 *auth, const size_t auth_len,
477			      const u8 *src_addr, u8 type, u8 *enc)
478{
479	const u8 *ad[3] = { src_addr, auth, &type };
480	size_t ad_len[3] = { ETH_ALEN, auth_len, sizeof(type) };
481
482	wpa_printf(MSG_DEBUG, "FT(RRB): src_addr=" MACSTR " type=%u",
483		   MAC2STR(src_addr), type);
484	wpa_hexdump_key(MSG_DEBUG, "FT(RRB): plaintext message",
485			plain, plain_len);
486	wpa_hexdump_key(MSG_DEBUG, "FT(RRB): encrypt using key", key, key_len);
487	wpa_hexdump(MSG_DEBUG, "FT(RRB): authenticated TLVs", auth, auth_len);
488
489	if (!key) {
490		/* encryption not needed, return plaintext as packet */
491		os_memcpy(enc, plain, plain_len);
492	} else if (aes_siv_encrypt(key, key_len, plain, plain_len,
493				   3, ad, ad_len, enc) < 0) {
494		wpa_printf(MSG_ERROR, "FT: Failed to encrypt RRB-OUI message");
495		return -1;
496	}
497	wpa_hexdump(MSG_DEBUG, "FT(RRB): encrypted TLVs",
498		    enc, plain_len + AES_BLOCK_SIZE);
499
500	return 0;
501}
502
503
504/**
505 * wpa_ft_rrb_build - Build and encrypt an FT RRB message
506 * @key: AES-SIV key for AEAD
507 * @key_len: Length of key in octets
508 * @tlvs_enc0: First set of to-be-encrypted TLVs
509 * @tlvs_enc1: Second set of to-be-encrypted TLVs
510 * @tlvs_auth: Set of to-be-authenticated TLVs
511 * @src_addr: MAC address of the frame sender
512 * @type: Vendor-specific subtype of the RRB frame (FT_PACKET_*)
513 * @packet Pointer to return the pointer to the allocated packet buffer;
514 *         needs to be freed by the caller if not null;
515 *         will only be returned on success
516 * @packet_len: Pointer to return the length of the allocated buffer in octets
517 * Returns: 0 on success, -1 on error
518 */
519static int wpa_ft_rrb_build(const u8 *key, const size_t key_len,
520			    const struct tlv_list *tlvs_enc0,
521			    const struct tlv_list *tlvs_enc1,
522			    const struct tlv_list *tlvs_auth,
523			    const struct vlan_description *vlan,
524			    const u8 *src_addr, u8 type,
525			    u8 **packet, size_t *packet_len)
526{
527	u8 *plain = NULL, *auth = NULL, *pos, *tmp;
528	size_t plain_len = 0, auth_len = 0;
529	int ret = -1;
530	size_t pad_len = 0;
531
532	*packet = NULL;
533	if (wpa_ft_rrb_lin(tlvs_enc0, tlvs_enc1, vlan, &plain, &plain_len) < 0)
534		goto out;
535
536	if (wpa_ft_rrb_lin(tlvs_auth, NULL, NULL, &auth, &auth_len) < 0)
537		goto out;
538
539	*packet_len = sizeof(u16) + auth_len + plain_len;
540	if (key)
541		*packet_len += AES_BLOCK_SIZE;
542#define RRB_MIN_MSG_LEN 64
543	if (*packet_len < RRB_MIN_MSG_LEN) {
544		pad_len = RRB_MIN_MSG_LEN - *packet_len;
545		if (pad_len < sizeof(struct ft_rrb_tlv))
546			pad_len = sizeof(struct ft_rrb_tlv);
547		wpa_printf(MSG_DEBUG,
548			   "FT: Pad message to minimum Ethernet frame length (%d --> %d)",
549			   (int) *packet_len, (int) (*packet_len + pad_len));
550		*packet_len += pad_len;
551		tmp = os_realloc(auth, auth_len + pad_len);
552		if (!tmp)
553			goto out;
554		auth = tmp;
555		pos = auth + auth_len;
556		WPA_PUT_LE16(pos, FT_RRB_LAST_EMPTY);
557		pos += 2;
558		WPA_PUT_LE16(pos, pad_len - sizeof(struct ft_rrb_tlv));
559		pos += 2;
560		os_memset(pos, 0, pad_len - sizeof(struct ft_rrb_tlv));
561		auth_len += pad_len;
562
563	}
564	*packet = os_zalloc(*packet_len);
565	if (!*packet)
566		goto out;
567
568	pos = *packet;
569	WPA_PUT_LE16(pos, auth_len);
570	pos += 2;
571	os_memcpy(pos, auth, auth_len);
572	pos += auth_len;
573	if (wpa_ft_rrb_encrypt(key, key_len, plain, plain_len, auth,
574			       auth_len, src_addr, type, pos) < 0)
575		goto out;
576	wpa_hexdump(MSG_MSGDUMP, "FT: RRB frame payload", *packet, *packet_len);
577
578	ret = 0;
579
580out:
581	bin_clear_free(plain, plain_len);
582	os_free(auth);
583
584	if (ret) {
585		wpa_printf(MSG_ERROR, "FT: Failed to build RRB-OUI message");
586		os_free(*packet);
587		*packet = NULL;
588		*packet_len = 0;
589	}
590
591	return ret;
592}
593
594
595#define RRB_GET_SRC(srcfield, type, field, txt, checklength) do { \
596	if (wpa_ft_rrb_get_tlv(srcfield, srcfield##_len, type, \
597				&f_##field##_len, &f_##field) < 0 || \
598	    (checklength > 0 && ((size_t) checklength) != f_##field##_len)) { \
599		wpa_printf(MSG_INFO, "FT: Missing required " #field \
600			   " in %s from " MACSTR, txt, MAC2STR(src_addr)); \
601		wpa_ft_rrb_dump(srcfield, srcfield##_len); \
602		goto out; \
603	} \
604} while (0)
605
606#define RRB_GET(type, field, txt, checklength) \
607	RRB_GET_SRC(plain, type, field, txt, checklength)
608#define RRB_GET_AUTH(type, field, txt, checklength) \
609	RRB_GET_SRC(auth, type, field, txt, checklength)
610
611#define RRB_GET_OPTIONAL_SRC(srcfield, type, field, txt, checklength) do { \
612	if (wpa_ft_rrb_get_tlv(srcfield, srcfield##_len, type, \
613				&f_##field##_len, &f_##field) < 0 || \
614	    (checklength > 0 && ((size_t) checklength) != f_##field##_len)) { \
615		wpa_printf(MSG_DEBUG, "FT: Missing optional " #field \
616			   " in %s from " MACSTR, txt, MAC2STR(src_addr)); \
617		f_##field##_len = 0; \
618		f_##field = NULL; \
619	} \
620} while (0)
621
622#define RRB_GET_OPTIONAL(type, field, txt, checklength) \
623	RRB_GET_OPTIONAL_SRC(plain, type, field, txt, checklength)
624#define RRB_GET_OPTIONAL_AUTH(type, field, txt, checklength) \
625	RRB_GET_OPTIONAL_SRC(auth, type, field, txt, checklength)
626
627static int wpa_ft_rrb_send(struct wpa_authenticator *wpa_auth, const u8 *dst,
628			   const u8 *data, size_t data_len)
629{
630	if (wpa_auth->cb->send_ether == NULL)
631		return -1;
632	wpa_printf(MSG_DEBUG, "FT: RRB send to " MACSTR, MAC2STR(dst));
633	return wpa_auth->cb->send_ether(wpa_auth->cb_ctx, dst, ETH_P_RRB,
634					data, data_len);
635}
636
637
638static int wpa_ft_rrb_oui_send(struct wpa_authenticator *wpa_auth,
639			       const u8 *dst, u8 oui_suffix,
640			       const u8 *data, size_t data_len)
641{
642	if (!wpa_auth->cb->send_oui)
643		return -1;
644	wpa_printf(MSG_DEBUG, "FT: RRB-OUI type %u send to " MACSTR " (len=%u)",
645		   oui_suffix, MAC2STR(dst), (unsigned int) data_len);
646	return wpa_auth->cb->send_oui(wpa_auth->cb_ctx, dst, oui_suffix, data,
647				      data_len);
648}
649
650
651static int wpa_ft_action_send(struct wpa_authenticator *wpa_auth,
652			      const u8 *dst, const u8 *data, size_t data_len)
653{
654	if (wpa_auth->cb->send_ft_action == NULL)
655		return -1;
656	return wpa_auth->cb->send_ft_action(wpa_auth->cb_ctx, dst,
657					    data, data_len);
658}
659
660
661static const u8 * wpa_ft_get_psk(struct wpa_authenticator *wpa_auth,
662				 const u8 *addr, const u8 *p2p_dev_addr,
663				 const u8 *prev_psk)
664{
665	if (wpa_auth->cb->get_psk == NULL)
666		return NULL;
667	return wpa_auth->cb->get_psk(wpa_auth->cb_ctx, addr, p2p_dev_addr,
668				     prev_psk, NULL, NULL);
669}
670
671
672static struct wpa_state_machine *
673wpa_ft_add_sta(struct wpa_authenticator *wpa_auth, const u8 *sta_addr)
674{
675	if (wpa_auth->cb->add_sta == NULL)
676		return NULL;
677	return wpa_auth->cb->add_sta(wpa_auth->cb_ctx, sta_addr);
678}
679
680
681static int wpa_ft_set_vlan(struct wpa_authenticator *wpa_auth,
682			   const u8 *sta_addr, struct vlan_description *vlan)
683{
684	if (!wpa_auth->cb->set_vlan)
685		return -1;
686	return wpa_auth->cb->set_vlan(wpa_auth->cb_ctx, sta_addr, vlan);
687}
688
689
690static int wpa_ft_get_vlan(struct wpa_authenticator *wpa_auth,
691			   const u8 *sta_addr, struct vlan_description *vlan)
692{
693	if (!wpa_auth->cb->get_vlan)
694		return -1;
695	return wpa_auth->cb->get_vlan(wpa_auth->cb_ctx, sta_addr, vlan);
696}
697
698
699static int
700wpa_ft_set_identity(struct wpa_authenticator *wpa_auth, const u8 *sta_addr,
701		    const u8 *identity, size_t identity_len)
702{
703	if (!wpa_auth->cb->set_identity)
704		return -1;
705	return wpa_auth->cb->set_identity(wpa_auth->cb_ctx, sta_addr, identity,
706					  identity_len);
707}
708
709
710static size_t
711wpa_ft_get_identity(struct wpa_authenticator *wpa_auth, const u8 *sta_addr,
712		    const u8 **buf)
713{
714	*buf = NULL;
715	if (!wpa_auth->cb->get_identity)
716		return 0;
717	return wpa_auth->cb->get_identity(wpa_auth->cb_ctx, sta_addr, buf);
718}
719
720
721static int
722wpa_ft_set_radius_cui(struct wpa_authenticator *wpa_auth, const u8 *sta_addr,
723		      const u8 *radius_cui, size_t radius_cui_len)
724{
725	if (!wpa_auth->cb->set_radius_cui)
726		return -1;
727	return wpa_auth->cb->set_radius_cui(wpa_auth->cb_ctx, sta_addr,
728					    radius_cui, radius_cui_len);
729}
730
731
732static size_t
733wpa_ft_get_radius_cui(struct wpa_authenticator *wpa_auth, const u8 *sta_addr,
734		      const u8 **buf)
735{
736	*buf = NULL;
737	if (!wpa_auth->cb->get_radius_cui)
738		return 0;
739	return wpa_auth->cb->get_radius_cui(wpa_auth->cb_ctx, sta_addr, buf);
740}
741
742
743static void
744wpa_ft_set_session_timeout(struct wpa_authenticator *wpa_auth,
745			    const u8 *sta_addr, int session_timeout)
746{
747	if (!wpa_auth->cb->set_session_timeout)
748		return;
749	wpa_auth->cb->set_session_timeout(wpa_auth->cb_ctx, sta_addr,
750					  session_timeout);
751}
752
753
754static int
755wpa_ft_get_session_timeout(struct wpa_authenticator *wpa_auth,
756			    const u8 *sta_addr)
757{
758	if (!wpa_auth->cb->get_session_timeout)
759		return 0;
760	return wpa_auth->cb->get_session_timeout(wpa_auth->cb_ctx, sta_addr);
761}
762
763
764static int wpa_ft_add_tspec(struct wpa_authenticator *wpa_auth,
765			    const u8 *sta_addr,
766			    u8 *tspec_ie, size_t tspec_ielen)
767{
768	if (wpa_auth->cb->add_tspec == NULL) {
769		wpa_printf(MSG_DEBUG, "FT: add_tspec is not initialized");
770		return -1;
771	}
772	return wpa_auth->cb->add_tspec(wpa_auth->cb_ctx, sta_addr, tspec_ie,
773				       tspec_ielen);
774}
775
776
777#ifdef CONFIG_OCV
778static int wpa_channel_info(struct wpa_authenticator *wpa_auth,
779			       struct wpa_channel_info *ci)
780{
781	if (!wpa_auth->cb->channel_info)
782		return -1;
783	return wpa_auth->cb->channel_info(wpa_auth->cb_ctx, ci);
784}
785#endif /* CONFIG_OCV */
786
787
788int wpa_write_mdie(struct wpa_auth_config *conf, u8 *buf, size_t len)
789{
790	u8 *pos = buf;
791	u8 capab;
792	if (len < 2 + sizeof(struct rsn_mdie))
793		return -1;
794
795	*pos++ = WLAN_EID_MOBILITY_DOMAIN;
796	*pos++ = MOBILITY_DOMAIN_ID_LEN + 1;
797	os_memcpy(pos, conf->mobility_domain, MOBILITY_DOMAIN_ID_LEN);
798	pos += MOBILITY_DOMAIN_ID_LEN;
799	capab = 0;
800	if (conf->ft_over_ds)
801		capab |= RSN_FT_CAPAB_FT_OVER_DS;
802	*pos++ = capab;
803
804	return pos - buf;
805}
806
807
808int wpa_write_ftie(struct wpa_auth_config *conf, int use_sha384,
809		   const u8 *r0kh_id, size_t r0kh_id_len,
810		   const u8 *anonce, const u8 *snonce,
811		   u8 *buf, size_t len, const u8 *subelem,
812		   size_t subelem_len, int rsnxe_used)
813{
814	u8 *pos = buf, *ielen;
815	size_t hdrlen = use_sha384 ? sizeof(struct rsn_ftie_sha384) :
816		sizeof(struct rsn_ftie);
817
818	if (len < 2 + hdrlen + 2 + FT_R1KH_ID_LEN + 2 + r0kh_id_len +
819	    subelem_len)
820		return -1;
821
822	*pos++ = WLAN_EID_FAST_BSS_TRANSITION;
823	ielen = pos++;
824
825	if (use_sha384) {
826		struct rsn_ftie_sha384 *hdr = (struct rsn_ftie_sha384 *) pos;
827
828		os_memset(hdr, 0, sizeof(*hdr));
829		pos += sizeof(*hdr);
830		WPA_PUT_LE16(hdr->mic_control, !!rsnxe_used);
831		if (anonce)
832			os_memcpy(hdr->anonce, anonce, WPA_NONCE_LEN);
833		if (snonce)
834			os_memcpy(hdr->snonce, snonce, WPA_NONCE_LEN);
835	} else {
836		struct rsn_ftie *hdr = (struct rsn_ftie *) pos;
837
838		os_memset(hdr, 0, sizeof(*hdr));
839		pos += sizeof(*hdr);
840		WPA_PUT_LE16(hdr->mic_control, !!rsnxe_used);
841		if (anonce)
842			os_memcpy(hdr->anonce, anonce, WPA_NONCE_LEN);
843		if (snonce)
844			os_memcpy(hdr->snonce, snonce, WPA_NONCE_LEN);
845	}
846
847	/* Optional Parameters */
848	*pos++ = FTIE_SUBELEM_R1KH_ID;
849	*pos++ = FT_R1KH_ID_LEN;
850	os_memcpy(pos, conf->r1_key_holder, FT_R1KH_ID_LEN);
851	pos += FT_R1KH_ID_LEN;
852
853	if (r0kh_id) {
854		*pos++ = FTIE_SUBELEM_R0KH_ID;
855		*pos++ = r0kh_id_len;
856		os_memcpy(pos, r0kh_id, r0kh_id_len);
857		pos += r0kh_id_len;
858	}
859
860	if (subelem) {
861		os_memcpy(pos, subelem, subelem_len);
862		pos += subelem_len;
863	}
864
865	*ielen = pos - buf - 2;
866
867	return pos - buf;
868}
869
870
871/* A packet to be handled after seq response */
872struct ft_remote_item {
873	struct dl_list list;
874
875	u8 nonce[FT_RRB_NONCE_LEN];
876	struct os_reltime nonce_ts;
877
878	u8 src_addr[ETH_ALEN];
879	u8 *enc;
880	size_t enc_len;
881	u8 *auth;
882	size_t auth_len;
883	int (*cb)(struct wpa_authenticator *wpa_auth,
884		  const u8 *src_addr,
885		  const u8 *enc, size_t enc_len,
886		  const u8 *auth, size_t auth_len,
887		  int no_defer);
888};
889
890
891static void wpa_ft_rrb_seq_free(struct ft_remote_item *item)
892{
893	eloop_cancel_timeout(wpa_ft_rrb_seq_timeout, ELOOP_ALL_CTX, item);
894	dl_list_del(&item->list);
895	bin_clear_free(item->enc, item->enc_len);
896	os_free(item->auth);
897	os_free(item);
898}
899
900
901static void wpa_ft_rrb_seq_flush(struct wpa_authenticator *wpa_auth,
902				 struct ft_remote_seq *rkh_seq, int cb)
903{
904	struct ft_remote_item *item, *n;
905
906	dl_list_for_each_safe(item, n, &rkh_seq->rx.queue,
907			      struct ft_remote_item, list) {
908		if (cb && item->cb)
909			item->cb(wpa_auth, item->src_addr, item->enc,
910				 item->enc_len, item->auth, item->auth_len, 1);
911		wpa_ft_rrb_seq_free(item);
912	}
913}
914
915
916static void wpa_ft_rrb_seq_timeout(void *eloop_ctx, void *timeout_ctx)
917{
918	struct ft_remote_item *item = timeout_ctx;
919
920	wpa_ft_rrb_seq_free(item);
921}
922
923
924static int
925wpa_ft_rrb_seq_req(struct wpa_authenticator *wpa_auth,
926		   struct ft_remote_seq *rkh_seq, const u8 *src_addr,
927		   const u8 *f_r0kh_id, size_t f_r0kh_id_len,
928		   const u8 *f_r1kh_id, const u8 *key, size_t key_len,
929		   const u8 *enc, size_t enc_len,
930		   const u8 *auth, size_t auth_len,
931		   int (*cb)(struct wpa_authenticator *wpa_auth,
932			     const u8 *src_addr,
933			     const u8 *enc, size_t enc_len,
934			     const u8 *auth, size_t auth_len,
935			     int no_defer))
936{
937	struct ft_remote_item *item = NULL;
938	u8 *packet = NULL;
939	size_t packet_len;
940	struct tlv_list seq_req_auth[] = {
941		{ .type = FT_RRB_NONCE, .len = FT_RRB_NONCE_LEN,
942		  .data = NULL /* to be filled: item->nonce */ },
943		{ .type = FT_RRB_R0KH_ID, .len = f_r0kh_id_len,
944		  .data = f_r0kh_id },
945		{ .type = FT_RRB_R1KH_ID, .len = FT_R1KH_ID_LEN,
946		  .data = f_r1kh_id },
947		{ .type = FT_RRB_LAST_EMPTY, .len = 0, .data = NULL },
948	};
949
950	if (dl_list_len(&rkh_seq->rx.queue) >= ftRRBmaxQueueLen) {
951		wpa_printf(MSG_DEBUG, "FT: Sequence number queue too long");
952		goto err;
953	}
954
955	wpa_printf(MSG_DEBUG, "FT: Send sequence number request from " MACSTR
956		   " to " MACSTR,
957		   MAC2STR(wpa_auth->addr), MAC2STR(src_addr));
958	item = os_zalloc(sizeof(*item));
959	if (!item)
960		goto err;
961
962	os_memcpy(item->src_addr, src_addr, ETH_ALEN);
963	item->cb = cb;
964
965	if (random_get_bytes(item->nonce, FT_RRB_NONCE_LEN) < 0) {
966		wpa_printf(MSG_DEBUG, "FT: Seq num nonce: out of random bytes");
967		goto err;
968	}
969
970	if (os_get_reltime(&item->nonce_ts) < 0)
971		goto err;
972
973	if (enc && enc_len > 0) {
974		item->enc = os_memdup(enc, enc_len);
975		item->enc_len = enc_len;
976		if (!item->enc)
977			goto err;
978	}
979
980	if (auth && auth_len > 0) {
981		item->auth = os_memdup(auth, auth_len);
982		item->auth_len = auth_len;
983		if (!item->auth)
984			goto err;
985	}
986
987	eloop_register_timeout(ftRRBseqTimeout, 0, wpa_ft_rrb_seq_timeout,
988			       wpa_auth, item);
989
990	seq_req_auth[0].data = item->nonce;
991
992	if (wpa_ft_rrb_build(key, key_len, NULL, NULL, seq_req_auth, NULL,
993			     wpa_auth->addr, FT_PACKET_R0KH_R1KH_SEQ_REQ,
994			     &packet, &packet_len) < 0) {
995		item = NULL; /* some other seq resp might still accept this */
996		goto err;
997	}
998
999	dl_list_add(&rkh_seq->rx.queue, &item->list);
1000
1001	wpa_ft_rrb_oui_send(wpa_auth, src_addr, FT_PACKET_R0KH_R1KH_SEQ_REQ,
1002			    packet, packet_len);
1003
1004	os_free(packet);
1005
1006	return 0;
1007err:
1008	wpa_printf(MSG_DEBUG, "FT: Failed to send sequence number request");
1009	if (item) {
1010		os_free(item->auth);
1011		bin_clear_free(item->enc, item->enc_len);
1012		os_free(item);
1013	}
1014
1015	return -1;
1016}
1017
1018
1019#define FT_RRB_SEQ_OK    0
1020#define FT_RRB_SEQ_DROP  1
1021#define FT_RRB_SEQ_DEFER 2
1022
1023static int
1024wpa_ft_rrb_seq_chk(struct ft_remote_seq *rkh_seq, const u8 *src_addr,
1025		   const u8 *enc, size_t enc_len,
1026		   const u8 *auth, size_t auth_len,
1027		   const char *msgtype, int no_defer)
1028{
1029	const u8 *f_seq;
1030	size_t f_seq_len;
1031	const struct ft_rrb_seq *msg_both;
1032	u32 msg_seq, msg_off, rkh_off;
1033	struct os_reltime now;
1034	unsigned int i;
1035
1036	RRB_GET_AUTH(FT_RRB_SEQ, seq, msgtype, sizeof(*msg_both));
1037	wpa_hexdump(MSG_DEBUG, "FT: sequence number", f_seq, f_seq_len);
1038	msg_both = (const struct ft_rrb_seq *) f_seq;
1039
1040	if (rkh_seq->rx.num_last == 0) {
1041		/* first packet from remote */
1042		goto defer;
1043	}
1044
1045	if (le_to_host32(msg_both->dom) != rkh_seq->rx.dom) {
1046		/* remote might have rebooted */
1047		goto defer;
1048	}
1049
1050	if (os_get_reltime(&now) == 0) {
1051		u32 msg_ts_now_remote, msg_ts_off;
1052		struct os_reltime now_remote;
1053
1054		os_reltime_sub(&now, &rkh_seq->rx.time_offset, &now_remote);
1055		msg_ts_now_remote = now_remote.sec;
1056		msg_ts_off = le_to_host32(msg_both->ts) -
1057			(msg_ts_now_remote - ftRRBseqTimeout);
1058		if (msg_ts_off > 2 * ftRRBseqTimeout)
1059			goto defer;
1060	}
1061
1062	msg_seq = le_to_host32(msg_both->seq);
1063	rkh_off = rkh_seq->rx.last[rkh_seq->rx.offsetidx];
1064	msg_off = msg_seq - rkh_off;
1065	if (msg_off > 0xC0000000)
1066		goto out; /* too old message, drop it */
1067
1068	if (msg_off <= 0x40000000) {
1069		for (i = 0; i < rkh_seq->rx.num_last; i++) {
1070			if (rkh_seq->rx.last[i] == msg_seq)
1071				goto out; /* duplicate message, drop it */
1072		}
1073
1074		return FT_RRB_SEQ_OK;
1075	}
1076
1077defer:
1078	if (no_defer)
1079		goto out;
1080
1081	wpa_printf(MSG_DEBUG, "FT: Possibly invalid sequence number in %s from "
1082		   MACSTR, msgtype, MAC2STR(src_addr));
1083
1084	return FT_RRB_SEQ_DEFER;
1085out:
1086	wpa_printf(MSG_DEBUG, "FT: Invalid sequence number in %s from " MACSTR,
1087		   msgtype, MAC2STR(src_addr));
1088
1089	return FT_RRB_SEQ_DROP;
1090}
1091
1092
1093static void
1094wpa_ft_rrb_seq_accept(struct wpa_authenticator *wpa_auth,
1095		      struct ft_remote_seq *rkh_seq, const u8 *src_addr,
1096		      const u8 *auth, size_t auth_len,
1097		      const char *msgtype)
1098{
1099	const u8 *f_seq;
1100	size_t f_seq_len;
1101	const struct ft_rrb_seq *msg_both;
1102	u32 msg_seq, msg_off, min_off, rkh_off;
1103	int minidx = 0;
1104	unsigned int i;
1105
1106	RRB_GET_AUTH(FT_RRB_SEQ, seq, msgtype, sizeof(*msg_both));
1107	msg_both = (const struct ft_rrb_seq *) f_seq;
1108
1109	msg_seq = le_to_host32(msg_both->seq);
1110
1111	if (rkh_seq->rx.num_last < FT_REMOTE_SEQ_BACKLOG) {
1112		rkh_seq->rx.last[rkh_seq->rx.num_last] = msg_seq;
1113		rkh_seq->rx.num_last++;
1114		return;
1115	}
1116
1117	rkh_off = rkh_seq->rx.last[rkh_seq->rx.offsetidx];
1118	for (i = 0; i < rkh_seq->rx.num_last; i++) {
1119		msg_off = rkh_seq->rx.last[i] - rkh_off;
1120		min_off = rkh_seq->rx.last[minidx] - rkh_off;
1121		if (msg_off < min_off && i != rkh_seq->rx.offsetidx)
1122			minidx = i;
1123	}
1124	rkh_seq->rx.last[rkh_seq->rx.offsetidx] = msg_seq;
1125	rkh_seq->rx.offsetidx = minidx;
1126
1127	return;
1128out:
1129	/* RRB_GET_AUTH should never fail here as
1130	 * wpa_ft_rrb_seq_chk() verified FT_RRB_SEQ presence. */
1131	wpa_printf(MSG_ERROR, "FT: %s() failed", __func__);
1132}
1133
1134
1135static int wpa_ft_new_seq(struct ft_remote_seq *rkh_seq,
1136			  struct ft_rrb_seq *f_seq)
1137{
1138	struct os_reltime now;
1139
1140	if (os_get_reltime(&now) < 0)
1141		return -1;
1142
1143	if (!rkh_seq->tx.dom) {
1144		if (random_get_bytes((u8 *) &rkh_seq->tx.seq,
1145				     sizeof(rkh_seq->tx.seq))) {
1146			wpa_printf(MSG_ERROR,
1147				   "FT: Failed to get random data for sequence number initialization");
1148			rkh_seq->tx.seq = now.usec;
1149		}
1150		if (random_get_bytes((u8 *) &rkh_seq->tx.dom,
1151				     sizeof(rkh_seq->tx.dom))) {
1152			wpa_printf(MSG_ERROR,
1153				   "FT: Failed to get random data for sequence number initialization");
1154			rkh_seq->tx.dom = now.usec;
1155		}
1156		rkh_seq->tx.dom |= 1;
1157	}
1158
1159	f_seq->dom = host_to_le32(rkh_seq->tx.dom);
1160	f_seq->seq = host_to_le32(rkh_seq->tx.seq);
1161	f_seq->ts = host_to_le32(now.sec);
1162
1163	rkh_seq->tx.seq++;
1164
1165	return 0;
1166}
1167
1168
1169struct wpa_ft_pmk_r0_sa {
1170	struct dl_list list;
1171	u8 pmk_r0[PMK_LEN_MAX];
1172	size_t pmk_r0_len;
1173	u8 pmk_r0_name[WPA_PMK_NAME_LEN];
1174	u8 spa[ETH_ALEN];
1175	int pairwise; /* Pairwise cipher suite, WPA_CIPHER_* */
1176	struct vlan_description *vlan;
1177	os_time_t expiration; /* 0 for no expiration */
1178	u8 *identity;
1179	size_t identity_len;
1180	u8 *radius_cui;
1181	size_t radius_cui_len;
1182	os_time_t session_timeout; /* 0 for no expiration */
1183	/* TODO: radius_class, EAP type */
1184	int pmk_r1_pushed;
1185};
1186
1187struct wpa_ft_pmk_r1_sa {
1188	struct dl_list list;
1189	u8 pmk_r1[PMK_LEN_MAX];
1190	size_t pmk_r1_len;
1191	u8 pmk_r1_name[WPA_PMK_NAME_LEN];
1192	u8 spa[ETH_ALEN];
1193	int pairwise; /* Pairwise cipher suite, WPA_CIPHER_* */
1194	struct vlan_description *vlan;
1195	u8 *identity;
1196	size_t identity_len;
1197	u8 *radius_cui;
1198	size_t radius_cui_len;
1199	os_time_t session_timeout; /* 0 for no expiration */
1200	/* TODO: radius_class, EAP type */
1201};
1202
1203struct wpa_ft_pmk_cache {
1204	struct dl_list pmk_r0; /* struct wpa_ft_pmk_r0_sa */
1205	struct dl_list pmk_r1; /* struct wpa_ft_pmk_r1_sa */
1206};
1207
1208
1209static void wpa_ft_expire_pmk_r0(void *eloop_ctx, void *timeout_ctx);
1210static void wpa_ft_expire_pmk_r1(void *eloop_ctx, void *timeout_ctx);
1211
1212
1213static void wpa_ft_free_pmk_r0(struct wpa_ft_pmk_r0_sa *r0)
1214{
1215	if (!r0)
1216		return;
1217
1218	dl_list_del(&r0->list);
1219	eloop_cancel_timeout(wpa_ft_expire_pmk_r0, r0, NULL);
1220
1221	os_memset(r0->pmk_r0, 0, PMK_LEN_MAX);
1222	os_free(r0->vlan);
1223	os_free(r0->identity);
1224	os_free(r0->radius_cui);
1225	os_free(r0);
1226}
1227
1228
1229static void wpa_ft_expire_pmk_r0(void *eloop_ctx, void *timeout_ctx)
1230{
1231	struct wpa_ft_pmk_r0_sa *r0 = eloop_ctx;
1232	struct os_reltime now;
1233	int expires_in;
1234	int session_timeout;
1235
1236	os_get_reltime(&now);
1237
1238	if (!r0)
1239		return;
1240
1241	expires_in = r0->expiration - now.sec;
1242	session_timeout = r0->session_timeout - now.sec;
1243	/* conditions to remove from cache:
1244	 * a) r0->expiration is set and hit
1245	 * -or-
1246	 * b) r0->session_timeout is set and hit
1247	 */
1248	if ((!r0->expiration || expires_in > 0) &&
1249	    (!r0->session_timeout || session_timeout > 0)) {
1250		wpa_printf(MSG_ERROR,
1251			   "FT: %s() called for non-expired entry %p",
1252			   __func__, r0);
1253		eloop_cancel_timeout(wpa_ft_expire_pmk_r0, r0, NULL);
1254		if (r0->expiration && expires_in > 0)
1255			eloop_register_timeout(expires_in + 1, 0,
1256					       wpa_ft_expire_pmk_r0, r0, NULL);
1257		if (r0->session_timeout && session_timeout > 0)
1258			eloop_register_timeout(session_timeout + 1, 0,
1259					       wpa_ft_expire_pmk_r0, r0, NULL);
1260		return;
1261	}
1262
1263	wpa_ft_free_pmk_r0(r0);
1264}
1265
1266
1267static void wpa_ft_free_pmk_r1(struct wpa_ft_pmk_r1_sa *r1)
1268{
1269	if (!r1)
1270		return;
1271
1272	dl_list_del(&r1->list);
1273	eloop_cancel_timeout(wpa_ft_expire_pmk_r1, r1, NULL);
1274
1275	os_memset(r1->pmk_r1, 0, PMK_LEN_MAX);
1276	os_free(r1->vlan);
1277	os_free(r1->identity);
1278	os_free(r1->radius_cui);
1279	os_free(r1);
1280}
1281
1282
1283static void wpa_ft_expire_pmk_r1(void *eloop_ctx, void *timeout_ctx)
1284{
1285	struct wpa_ft_pmk_r1_sa *r1 = eloop_ctx;
1286
1287	wpa_ft_free_pmk_r1(r1);
1288}
1289
1290
1291struct wpa_ft_pmk_cache * wpa_ft_pmk_cache_init(void)
1292{
1293	struct wpa_ft_pmk_cache *cache;
1294
1295	cache = os_zalloc(sizeof(*cache));
1296	if (cache) {
1297		dl_list_init(&cache->pmk_r0);
1298		dl_list_init(&cache->pmk_r1);
1299	}
1300
1301	return cache;
1302}
1303
1304
1305void wpa_ft_pmk_cache_deinit(struct wpa_ft_pmk_cache *cache)
1306{
1307	struct wpa_ft_pmk_r0_sa *r0, *r0prev;
1308	struct wpa_ft_pmk_r1_sa *r1, *r1prev;
1309
1310	dl_list_for_each_safe(r0, r0prev, &cache->pmk_r0,
1311			      struct wpa_ft_pmk_r0_sa, list)
1312		wpa_ft_free_pmk_r0(r0);
1313
1314	dl_list_for_each_safe(r1, r1prev, &cache->pmk_r1,
1315			      struct wpa_ft_pmk_r1_sa, list)
1316		wpa_ft_free_pmk_r1(r1);
1317
1318	os_free(cache);
1319}
1320
1321
1322static int wpa_ft_store_pmk_r0(struct wpa_authenticator *wpa_auth,
1323			       const u8 *spa, const u8 *pmk_r0,
1324			       size_t pmk_r0_len,
1325			       const u8 *pmk_r0_name, int pairwise,
1326			       const struct vlan_description *vlan,
1327			       int expires_in, int session_timeout,
1328			       const u8 *identity, size_t identity_len,
1329			       const u8 *radius_cui, size_t radius_cui_len)
1330{
1331	struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
1332	struct wpa_ft_pmk_r0_sa *r0;
1333	struct os_reltime now;
1334
1335	/* TODO: add limit on number of entries in cache */
1336	os_get_reltime(&now);
1337
1338	r0 = os_zalloc(sizeof(*r0));
1339	if (r0 == NULL)
1340		return -1;
1341
1342	os_memcpy(r0->pmk_r0, pmk_r0, pmk_r0_len);
1343	r0->pmk_r0_len = pmk_r0_len;
1344	os_memcpy(r0->pmk_r0_name, pmk_r0_name, WPA_PMK_NAME_LEN);
1345	os_memcpy(r0->spa, spa, ETH_ALEN);
1346	r0->pairwise = pairwise;
1347	if (expires_in > 0)
1348		r0->expiration = now.sec + expires_in;
1349	if (vlan && vlan->notempty) {
1350		r0->vlan = os_zalloc(sizeof(*vlan));
1351		if (!r0->vlan) {
1352			bin_clear_free(r0, sizeof(*r0));
1353			return -1;
1354		}
1355		*r0->vlan = *vlan;
1356	}
1357	if (identity) {
1358		r0->identity = os_malloc(identity_len);
1359		if (r0->identity) {
1360			os_memcpy(r0->identity, identity, identity_len);
1361			r0->identity_len = identity_len;
1362		}
1363	}
1364	if (radius_cui) {
1365		r0->radius_cui = os_malloc(radius_cui_len);
1366		if (r0->radius_cui) {
1367			os_memcpy(r0->radius_cui, radius_cui, radius_cui_len);
1368			r0->radius_cui_len = radius_cui_len;
1369		}
1370	}
1371	if (session_timeout > 0)
1372		r0->session_timeout = now.sec + session_timeout;
1373
1374	dl_list_add(&cache->pmk_r0, &r0->list);
1375	if (expires_in > 0)
1376		eloop_register_timeout(expires_in + 1, 0, wpa_ft_expire_pmk_r0,
1377				       r0, NULL);
1378	if (session_timeout > 0)
1379		eloop_register_timeout(session_timeout + 1, 0,
1380				       wpa_ft_expire_pmk_r0, r0, NULL);
1381
1382	return 0;
1383}
1384
1385
1386static int wpa_ft_fetch_pmk_r0(struct wpa_authenticator *wpa_auth,
1387			       const u8 *spa, const u8 *pmk_r0_name,
1388			       const struct wpa_ft_pmk_r0_sa **r0_out)
1389{
1390	struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
1391	struct wpa_ft_pmk_r0_sa *r0;
1392	struct os_reltime now;
1393
1394	os_get_reltime(&now);
1395	dl_list_for_each(r0, &cache->pmk_r0, struct wpa_ft_pmk_r0_sa, list) {
1396		if (os_memcmp(r0->spa, spa, ETH_ALEN) == 0 &&
1397		    os_memcmp_const(r0->pmk_r0_name, pmk_r0_name,
1398				    WPA_PMK_NAME_LEN) == 0) {
1399			*r0_out = r0;
1400			return 0;
1401		}
1402	}
1403
1404	*r0_out = NULL;
1405	return -1;
1406}
1407
1408
1409static int wpa_ft_store_pmk_r1(struct wpa_authenticator *wpa_auth,
1410			       const u8 *spa, const u8 *pmk_r1,
1411			       size_t pmk_r1_len,
1412			       const u8 *pmk_r1_name, int pairwise,
1413			       const struct vlan_description *vlan,
1414			       int expires_in, int session_timeout,
1415			       const u8 *identity, size_t identity_len,
1416			       const u8 *radius_cui, size_t radius_cui_len)
1417{
1418	struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
1419	int max_expires_in = wpa_auth->conf.r1_max_key_lifetime;
1420	struct wpa_ft_pmk_r1_sa *r1;
1421	struct os_reltime now;
1422
1423	/* TODO: limit on number of entries in cache */
1424	os_get_reltime(&now);
1425
1426	if (max_expires_in && (max_expires_in < expires_in || expires_in == 0))
1427		expires_in = max_expires_in;
1428
1429	r1 = os_zalloc(sizeof(*r1));
1430	if (r1 == NULL)
1431		return -1;
1432
1433	os_memcpy(r1->pmk_r1, pmk_r1, pmk_r1_len);
1434	r1->pmk_r1_len = pmk_r1_len;
1435	os_memcpy(r1->pmk_r1_name, pmk_r1_name, WPA_PMK_NAME_LEN);
1436	os_memcpy(r1->spa, spa, ETH_ALEN);
1437	r1->pairwise = pairwise;
1438	if (vlan && vlan->notempty) {
1439		r1->vlan = os_zalloc(sizeof(*vlan));
1440		if (!r1->vlan) {
1441			bin_clear_free(r1, sizeof(*r1));
1442			return -1;
1443		}
1444		*r1->vlan = *vlan;
1445	}
1446	if (identity) {
1447		r1->identity = os_malloc(identity_len);
1448		if (r1->identity) {
1449			os_memcpy(r1->identity, identity, identity_len);
1450			r1->identity_len = identity_len;
1451		}
1452	}
1453	if (radius_cui) {
1454		r1->radius_cui = os_malloc(radius_cui_len);
1455		if (r1->radius_cui) {
1456			os_memcpy(r1->radius_cui, radius_cui, radius_cui_len);
1457			r1->radius_cui_len = radius_cui_len;
1458		}
1459	}
1460	if (session_timeout > 0)
1461		r1->session_timeout = now.sec + session_timeout;
1462
1463	dl_list_add(&cache->pmk_r1, &r1->list);
1464
1465	if (expires_in > 0)
1466		eloop_register_timeout(expires_in + 1, 0, wpa_ft_expire_pmk_r1,
1467				       r1, NULL);
1468	if (session_timeout > 0)
1469		eloop_register_timeout(session_timeout + 1, 0,
1470				       wpa_ft_expire_pmk_r1, r1, NULL);
1471
1472	return 0;
1473}
1474
1475
1476int wpa_ft_fetch_pmk_r1(struct wpa_authenticator *wpa_auth,
1477			const u8 *spa, const u8 *pmk_r1_name,
1478			u8 *pmk_r1, size_t *pmk_r1_len, int *pairwise,
1479			struct vlan_description *vlan,
1480			const u8 **identity, size_t *identity_len,
1481			const u8 **radius_cui, size_t *radius_cui_len,
1482			int *session_timeout)
1483{
1484	struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
1485	struct wpa_ft_pmk_r1_sa *r1;
1486	struct os_reltime now;
1487
1488	os_get_reltime(&now);
1489
1490	dl_list_for_each(r1, &cache->pmk_r1, struct wpa_ft_pmk_r1_sa, list) {
1491		if (os_memcmp(r1->spa, spa, ETH_ALEN) == 0 &&
1492		    os_memcmp_const(r1->pmk_r1_name, pmk_r1_name,
1493				    WPA_PMK_NAME_LEN) == 0) {
1494			os_memcpy(pmk_r1, r1->pmk_r1, r1->pmk_r1_len);
1495			*pmk_r1_len = r1->pmk_r1_len;
1496			if (pairwise)
1497				*pairwise = r1->pairwise;
1498			if (vlan && r1->vlan)
1499				*vlan = *r1->vlan;
1500			if (vlan && !r1->vlan)
1501				os_memset(vlan, 0, sizeof(*vlan));
1502			if (identity && identity_len) {
1503				*identity = r1->identity;
1504				*identity_len = r1->identity_len;
1505			}
1506			if (radius_cui && radius_cui_len) {
1507				*radius_cui = r1->radius_cui;
1508				*radius_cui_len = r1->radius_cui_len;
1509			}
1510			if (session_timeout && r1->session_timeout > now.sec)
1511				*session_timeout = r1->session_timeout -
1512					now.sec;
1513			else if (session_timeout && r1->session_timeout)
1514				*session_timeout = 1;
1515			else if (session_timeout)
1516				*session_timeout = 0;
1517			return 0;
1518		}
1519	}
1520
1521	return -1;
1522}
1523
1524
1525static int wpa_ft_rrb_init_r0kh_seq(struct ft_remote_r0kh *r0kh)
1526{
1527	if (r0kh->seq)
1528		return 0;
1529
1530	r0kh->seq = os_zalloc(sizeof(*r0kh->seq));
1531	if (!r0kh->seq) {
1532		wpa_printf(MSG_DEBUG, "FT: Failed to allocate r0kh->seq");
1533		return -1;
1534	}
1535
1536	dl_list_init(&r0kh->seq->rx.queue);
1537
1538	return 0;
1539}
1540
1541
1542static void wpa_ft_rrb_lookup_r0kh(struct wpa_authenticator *wpa_auth,
1543				   const u8 *f_r0kh_id, size_t f_r0kh_id_len,
1544				   struct ft_remote_r0kh **r0kh_out,
1545				   struct ft_remote_r0kh **r0kh_wildcard)
1546{
1547	struct ft_remote_r0kh *r0kh;
1548
1549	*r0kh_wildcard = NULL;
1550	*r0kh_out = NULL;
1551
1552	if (wpa_auth->conf.r0kh_list)
1553		r0kh = *wpa_auth->conf.r0kh_list;
1554	else
1555		r0kh = NULL;
1556	for (; r0kh; r0kh = r0kh->next) {
1557		if (r0kh->id_len == 1 && r0kh->id[0] == '*')
1558			*r0kh_wildcard = r0kh;
1559		if (f_r0kh_id && r0kh->id_len == f_r0kh_id_len &&
1560		    os_memcmp_const(f_r0kh_id, r0kh->id, f_r0kh_id_len) == 0)
1561			*r0kh_out = r0kh;
1562	}
1563
1564	if (!*r0kh_out && !*r0kh_wildcard)
1565		wpa_printf(MSG_DEBUG, "FT: No matching R0KH found");
1566
1567	if (*r0kh_out && wpa_ft_rrb_init_r0kh_seq(*r0kh_out) < 0)
1568		*r0kh_out = NULL;
1569}
1570
1571
1572static int wpa_ft_rrb_init_r1kh_seq(struct ft_remote_r1kh *r1kh)
1573{
1574	if (r1kh->seq)
1575		return 0;
1576
1577	r1kh->seq = os_zalloc(sizeof(*r1kh->seq));
1578	if (!r1kh->seq) {
1579		wpa_printf(MSG_DEBUG, "FT: Failed to allocate r1kh->seq");
1580		return -1;
1581	}
1582
1583	dl_list_init(&r1kh->seq->rx.queue);
1584
1585	return 0;
1586}
1587
1588
1589static void wpa_ft_rrb_lookup_r1kh(struct wpa_authenticator *wpa_auth,
1590				   const u8 *f_r1kh_id,
1591				   struct ft_remote_r1kh **r1kh_out,
1592				   struct ft_remote_r1kh **r1kh_wildcard)
1593{
1594	struct ft_remote_r1kh *r1kh;
1595
1596	*r1kh_wildcard = NULL;
1597	*r1kh_out = NULL;
1598
1599	if (wpa_auth->conf.r1kh_list)
1600		r1kh = *wpa_auth->conf.r1kh_list;
1601	else
1602		r1kh = NULL;
1603	for (; r1kh; r1kh = r1kh->next) {
1604		if (is_zero_ether_addr(r1kh->addr) &&
1605		    is_zero_ether_addr(r1kh->id))
1606			*r1kh_wildcard = r1kh;
1607		if (f_r1kh_id &&
1608		    os_memcmp_const(r1kh->id, f_r1kh_id, FT_R1KH_ID_LEN) == 0)
1609			*r1kh_out = r1kh;
1610	}
1611
1612	if (!*r1kh_out && !*r1kh_wildcard)
1613		wpa_printf(MSG_DEBUG, "FT: No matching R1KH found");
1614
1615	if (*r1kh_out && wpa_ft_rrb_init_r1kh_seq(*r1kh_out) < 0)
1616		*r1kh_out = NULL;
1617}
1618
1619
1620static int wpa_ft_rrb_check_r0kh(struct wpa_authenticator *wpa_auth,
1621				 const u8 *f_r0kh_id, size_t f_r0kh_id_len)
1622{
1623	if (f_r0kh_id_len != wpa_auth->conf.r0_key_holder_len ||
1624	    os_memcmp_const(f_r0kh_id, wpa_auth->conf.r0_key_holder,
1625			    f_r0kh_id_len) != 0)
1626		return -1;
1627
1628	return 0;
1629}
1630
1631
1632static int wpa_ft_rrb_check_r1kh(struct wpa_authenticator *wpa_auth,
1633				 const u8 *f_r1kh_id)
1634{
1635	if (os_memcmp_const(f_r1kh_id, wpa_auth->conf.r1_key_holder,
1636			    FT_R1KH_ID_LEN) != 0)
1637		return -1;
1638
1639	return 0;
1640}
1641
1642
1643static void wpa_ft_rrb_del_r0kh(void *eloop_ctx, void *timeout_ctx)
1644{
1645	struct wpa_authenticator *wpa_auth = eloop_ctx;
1646	struct ft_remote_r0kh *r0kh, *prev = NULL;
1647
1648	if (!wpa_auth->conf.r0kh_list)
1649		return;
1650
1651	for (r0kh = *wpa_auth->conf.r0kh_list; r0kh; r0kh = r0kh->next) {
1652		if (r0kh == timeout_ctx)
1653			break;
1654		prev = r0kh;
1655	}
1656	if (!r0kh)
1657		return;
1658	if (prev)
1659		prev->next = r0kh->next;
1660	else
1661		*wpa_auth->conf.r0kh_list = r0kh->next;
1662	if (r0kh->seq)
1663		wpa_ft_rrb_seq_flush(wpa_auth, r0kh->seq, 0);
1664	os_free(r0kh->seq);
1665	os_free(r0kh);
1666}
1667
1668
1669static void wpa_ft_rrb_r0kh_replenish(struct wpa_authenticator *wpa_auth,
1670				      struct ft_remote_r0kh *r0kh, int timeout)
1671{
1672	if (timeout > 0)
1673		eloop_replenish_timeout(timeout, 0, wpa_ft_rrb_del_r0kh,
1674					wpa_auth, r0kh);
1675}
1676
1677
1678static void wpa_ft_rrb_r0kh_timeout(struct wpa_authenticator *wpa_auth,
1679				    struct ft_remote_r0kh *r0kh, int timeout)
1680{
1681	eloop_cancel_timeout(wpa_ft_rrb_del_r0kh, wpa_auth, r0kh);
1682
1683	if (timeout > 0)
1684		eloop_register_timeout(timeout, 0, wpa_ft_rrb_del_r0kh,
1685				       wpa_auth, r0kh);
1686}
1687
1688
1689static struct ft_remote_r0kh *
1690wpa_ft_rrb_add_r0kh(struct wpa_authenticator *wpa_auth,
1691		    struct ft_remote_r0kh *r0kh_wildcard,
1692		    const u8 *src_addr, const u8 *r0kh_id, size_t id_len,
1693		    int timeout)
1694{
1695	struct ft_remote_r0kh *r0kh;
1696
1697	if (!wpa_auth->conf.r0kh_list)
1698		return NULL;
1699
1700	r0kh = os_zalloc(sizeof(*r0kh));
1701	if (!r0kh)
1702		return NULL;
1703
1704	if (src_addr)
1705		os_memcpy(r0kh->addr, src_addr, sizeof(r0kh->addr));
1706
1707	if (id_len > FT_R0KH_ID_MAX_LEN)
1708		id_len = FT_R0KH_ID_MAX_LEN;
1709	os_memcpy(r0kh->id, r0kh_id, id_len);
1710	r0kh->id_len = id_len;
1711
1712	os_memcpy(r0kh->key, r0kh_wildcard->key, sizeof(r0kh->key));
1713
1714	r0kh->next = *wpa_auth->conf.r0kh_list;
1715	*wpa_auth->conf.r0kh_list = r0kh;
1716
1717	if (timeout > 0)
1718		eloop_register_timeout(timeout, 0, wpa_ft_rrb_del_r0kh,
1719				       wpa_auth, r0kh);
1720
1721	if (wpa_ft_rrb_init_r0kh_seq(r0kh) < 0)
1722		return NULL;
1723
1724	return r0kh;
1725}
1726
1727
1728static void wpa_ft_rrb_del_r1kh(void *eloop_ctx, void *timeout_ctx)
1729{
1730	struct wpa_authenticator *wpa_auth = eloop_ctx;
1731	struct ft_remote_r1kh *r1kh, *prev = NULL;
1732
1733	if (!wpa_auth->conf.r1kh_list)
1734		return;
1735
1736	for (r1kh = *wpa_auth->conf.r1kh_list; r1kh; r1kh = r1kh->next) {
1737		if (r1kh == timeout_ctx)
1738			break;
1739		prev = r1kh;
1740	}
1741	if (!r1kh)
1742		return;
1743	if (prev)
1744		prev->next = r1kh->next;
1745	else
1746		*wpa_auth->conf.r1kh_list = r1kh->next;
1747	if (r1kh->seq)
1748		wpa_ft_rrb_seq_flush(wpa_auth, r1kh->seq, 0);
1749	os_free(r1kh->seq);
1750	os_free(r1kh);
1751}
1752
1753
1754static void wpa_ft_rrb_r1kh_replenish(struct wpa_authenticator *wpa_auth,
1755				      struct ft_remote_r1kh *r1kh, int timeout)
1756{
1757	if (timeout > 0)
1758		eloop_replenish_timeout(timeout, 0, wpa_ft_rrb_del_r1kh,
1759					wpa_auth, r1kh);
1760}
1761
1762
1763static struct ft_remote_r1kh *
1764wpa_ft_rrb_add_r1kh(struct wpa_authenticator *wpa_auth,
1765		    struct ft_remote_r1kh *r1kh_wildcard,
1766		    const u8 *src_addr, const u8 *r1kh_id, int timeout)
1767{
1768	struct ft_remote_r1kh *r1kh;
1769
1770	if (!wpa_auth->conf.r1kh_list)
1771		return NULL;
1772
1773	r1kh = os_zalloc(sizeof(*r1kh));
1774	if (!r1kh)
1775		return NULL;
1776
1777	os_memcpy(r1kh->addr, src_addr, sizeof(r1kh->addr));
1778	os_memcpy(r1kh->id, r1kh_id, sizeof(r1kh->id));
1779	os_memcpy(r1kh->key, r1kh_wildcard->key, sizeof(r1kh->key));
1780	r1kh->next = *wpa_auth->conf.r1kh_list;
1781	*wpa_auth->conf.r1kh_list = r1kh;
1782
1783	if (timeout > 0)
1784		eloop_register_timeout(timeout, 0, wpa_ft_rrb_del_r1kh,
1785				       wpa_auth, r1kh);
1786
1787	if (wpa_ft_rrb_init_r1kh_seq(r1kh) < 0)
1788		return NULL;
1789
1790	return r1kh;
1791}
1792
1793
1794void wpa_ft_sta_deinit(struct wpa_state_machine *sm)
1795{
1796	eloop_cancel_timeout(wpa_ft_expire_pull, sm, NULL);
1797}
1798
1799
1800static void wpa_ft_deinit_seq(struct wpa_authenticator *wpa_auth)
1801{
1802	struct ft_remote_r0kh *r0kh;
1803	struct ft_remote_r1kh *r1kh;
1804
1805	eloop_cancel_timeout(wpa_ft_rrb_seq_timeout, wpa_auth, ELOOP_ALL_CTX);
1806
1807	if (wpa_auth->conf.r0kh_list)
1808		r0kh = *wpa_auth->conf.r0kh_list;
1809	else
1810		r0kh = NULL;
1811	for (; r0kh; r0kh = r0kh->next) {
1812		if (!r0kh->seq)
1813			continue;
1814		wpa_ft_rrb_seq_flush(wpa_auth, r0kh->seq, 0);
1815		os_free(r0kh->seq);
1816		r0kh->seq = NULL;
1817	}
1818
1819	if (wpa_auth->conf.r1kh_list)
1820		r1kh = *wpa_auth->conf.r1kh_list;
1821	else
1822		r1kh = NULL;
1823	for (; r1kh; r1kh = r1kh->next) {
1824		if (!r1kh->seq)
1825			continue;
1826		wpa_ft_rrb_seq_flush(wpa_auth, r1kh->seq, 0);
1827		os_free(r1kh->seq);
1828		r1kh->seq = NULL;
1829	}
1830}
1831
1832
1833static void wpa_ft_deinit_rkh_tmp(struct wpa_authenticator *wpa_auth)
1834{
1835	struct ft_remote_r0kh *r0kh, *r0kh_next, *r0kh_prev = NULL;
1836	struct ft_remote_r1kh *r1kh, *r1kh_next, *r1kh_prev = NULL;
1837
1838	if (wpa_auth->conf.r0kh_list)
1839		r0kh = *wpa_auth->conf.r0kh_list;
1840	else
1841		r0kh = NULL;
1842	while (r0kh) {
1843		r0kh_next = r0kh->next;
1844		if (eloop_cancel_timeout(wpa_ft_rrb_del_r0kh, wpa_auth,
1845					 r0kh) > 0) {
1846			if (r0kh_prev)
1847				r0kh_prev->next = r0kh_next;
1848			else
1849				*wpa_auth->conf.r0kh_list = r0kh_next;
1850			os_free(r0kh);
1851		} else {
1852			r0kh_prev = r0kh;
1853		}
1854		r0kh = r0kh_next;
1855	}
1856
1857	if (wpa_auth->conf.r1kh_list)
1858		r1kh = *wpa_auth->conf.r1kh_list;
1859	else
1860		r1kh = NULL;
1861	while (r1kh) {
1862		r1kh_next = r1kh->next;
1863		if (eloop_cancel_timeout(wpa_ft_rrb_del_r1kh, wpa_auth,
1864					 r1kh) > 0) {
1865			if (r1kh_prev)
1866				r1kh_prev->next = r1kh_next;
1867			else
1868				*wpa_auth->conf.r1kh_list = r1kh_next;
1869			os_free(r1kh);
1870		} else {
1871			r1kh_prev = r1kh;
1872		}
1873		r1kh = r1kh_next;
1874	}
1875}
1876
1877
1878void wpa_ft_deinit(struct wpa_authenticator *wpa_auth)
1879{
1880	wpa_ft_deinit_seq(wpa_auth);
1881	wpa_ft_deinit_rkh_tmp(wpa_auth);
1882}
1883
1884
1885static void wpa_ft_block_r0kh(struct wpa_authenticator *wpa_auth,
1886			      const u8 *f_r0kh_id, size_t f_r0kh_id_len)
1887{
1888	struct ft_remote_r0kh *r0kh, *r0kh_wildcard;
1889
1890	if (!wpa_auth->conf.rkh_neg_timeout)
1891		return;
1892
1893	wpa_ft_rrb_lookup_r0kh(wpa_auth, f_r0kh_id, f_r0kh_id_len,
1894			       &r0kh, &r0kh_wildcard);
1895
1896	if (!r0kh_wildcard) {
1897		/* r0kh removed after neg_timeout and might need re-adding */
1898		return;
1899	}
1900
1901	wpa_hexdump(MSG_DEBUG, "FT: Temporarily block R0KH-ID",
1902		    f_r0kh_id, f_r0kh_id_len);
1903
1904	if (r0kh) {
1905		wpa_ft_rrb_r0kh_timeout(wpa_auth, r0kh,
1906					wpa_auth->conf.rkh_neg_timeout);
1907		os_memset(r0kh->addr, 0, ETH_ALEN);
1908	} else
1909		wpa_ft_rrb_add_r0kh(wpa_auth, r0kh_wildcard, NULL, f_r0kh_id,
1910				    f_r0kh_id_len,
1911				    wpa_auth->conf.rkh_neg_timeout);
1912}
1913
1914
1915static void wpa_ft_expire_pull(void *eloop_ctx, void *timeout_ctx)
1916{
1917	struct wpa_state_machine *sm = eloop_ctx;
1918
1919	wpa_printf(MSG_DEBUG, "FT: Timeout pending pull request for " MACSTR,
1920		   MAC2STR(sm->addr));
1921	if (sm->ft_pending_pull_left_retries <= 0)
1922		wpa_ft_block_r0kh(sm->wpa_auth, sm->r0kh_id, sm->r0kh_id_len);
1923
1924	/* cancel multiple timeouts */
1925	eloop_cancel_timeout(wpa_ft_expire_pull, sm, NULL);
1926	ft_finish_pull(sm);
1927}
1928
1929
1930static int wpa_ft_pull_pmk_r1(struct wpa_state_machine *sm,
1931			      const u8 *ies, size_t ies_len,
1932			      const u8 *pmk_r0_name)
1933{
1934	struct ft_remote_r0kh *r0kh, *r0kh_wildcard;
1935	u8 *packet = NULL;
1936	const u8 *key, *f_r1kh_id = sm->wpa_auth->conf.r1_key_holder;
1937	size_t packet_len, key_len;
1938	struct ft_rrb_seq f_seq;
1939	int tsecs, tusecs, first;
1940	struct wpabuf *ft_pending_req_ies;
1941	int r0kh_timeout;
1942	struct tlv_list req_enc[] = {
1943		{ .type = FT_RRB_PMK_R0_NAME, .len = WPA_PMK_NAME_LEN,
1944		  .data = pmk_r0_name },
1945		{ .type = FT_RRB_S1KH_ID, .len = ETH_ALEN,
1946		  .data = sm->addr },
1947		{ .type = FT_RRB_LAST_EMPTY, .len = 0, .data = NULL },
1948	};
1949	struct tlv_list req_auth[] = {
1950		{ .type = FT_RRB_NONCE, .len = FT_RRB_NONCE_LEN,
1951		  .data = sm->ft_pending_pull_nonce },
1952		{ .type = FT_RRB_SEQ, .len = sizeof(f_seq),
1953		  .data = (u8 *) &f_seq },
1954		{ .type = FT_RRB_R0KH_ID, .len = sm->r0kh_id_len,
1955		  .data = sm->r0kh_id },
1956		{ .type = FT_RRB_R1KH_ID, .len = FT_R1KH_ID_LEN,
1957		  .data = f_r1kh_id },
1958		{ .type = FT_RRB_LAST_EMPTY, .len = 0, .data = NULL },
1959	};
1960
1961	if (sm->ft_pending_pull_left_retries <= 0)
1962		return -1;
1963	first = sm->ft_pending_pull_left_retries ==
1964		sm->wpa_auth->conf.rkh_pull_retries;
1965	sm->ft_pending_pull_left_retries--;
1966
1967	wpa_ft_rrb_lookup_r0kh(sm->wpa_auth, sm->r0kh_id, sm->r0kh_id_len,
1968			       &r0kh, &r0kh_wildcard);
1969
1970	/* Keep r0kh sufficiently long in the list for seq num check */
1971	r0kh_timeout = sm->wpa_auth->conf.rkh_pull_timeout / 1000 +
1972		1 + ftRRBseqTimeout;
1973	if (r0kh) {
1974		wpa_ft_rrb_r0kh_replenish(sm->wpa_auth, r0kh, r0kh_timeout);
1975	} else if (r0kh_wildcard) {
1976		wpa_printf(MSG_DEBUG, "FT: Using wildcard R0KH-ID");
1977		/* r0kh->addr: updated by SEQ_RESP and wpa_ft_expire_pull */
1978		r0kh = wpa_ft_rrb_add_r0kh(sm->wpa_auth, r0kh_wildcard,
1979					   r0kh_wildcard->addr,
1980					   sm->r0kh_id, sm->r0kh_id_len,
1981					   r0kh_timeout);
1982	}
1983	if (r0kh == NULL) {
1984		wpa_hexdump(MSG_DEBUG, "FT: Did not find R0KH-ID",
1985			    sm->r0kh_id, sm->r0kh_id_len);
1986		return -1;
1987	}
1988	if (is_zero_ether_addr(r0kh->addr)) {
1989		wpa_hexdump(MSG_DEBUG, "FT: R0KH-ID is temporarily blocked",
1990			    sm->r0kh_id, sm->r0kh_id_len);
1991		return -1;
1992	}
1993	if (os_memcmp(r0kh->addr, sm->wpa_auth->addr, ETH_ALEN) == 0) {
1994		wpa_printf(MSG_DEBUG,
1995			   "FT: R0KH-ID points to self - no matching key available");
1996		return -1;
1997	}
1998
1999	key = r0kh->key;
2000	key_len = sizeof(r0kh->key);
2001
2002	if (r0kh->seq->rx.num_last == 0) {
2003		/* A sequence request will be sent out anyway when pull
2004		 * response is received. Send it out now to avoid one RTT. */
2005		wpa_ft_rrb_seq_req(sm->wpa_auth, r0kh->seq, r0kh->addr,
2006				   r0kh->id, r0kh->id_len, f_r1kh_id, key,
2007				   key_len, NULL, 0, NULL, 0, NULL);
2008	}
2009
2010	wpa_printf(MSG_DEBUG, "FT: Send PMK-R1 pull request from " MACSTR
2011		   " to remote R0KH address " MACSTR,
2012		   MAC2STR(sm->wpa_auth->addr), MAC2STR(r0kh->addr));
2013
2014	if (first &&
2015	    random_get_bytes(sm->ft_pending_pull_nonce, FT_RRB_NONCE_LEN) < 0) {
2016		wpa_printf(MSG_DEBUG, "FT: Failed to get random data for "
2017			   "nonce");
2018		return -1;
2019	}
2020
2021	if (wpa_ft_new_seq(r0kh->seq, &f_seq) < 0) {
2022		wpa_printf(MSG_DEBUG, "FT: Failed to get seq num");
2023		return -1;
2024	}
2025
2026	if (wpa_ft_rrb_build(key, key_len, req_enc, NULL, req_auth, NULL,
2027			     sm->wpa_auth->addr, FT_PACKET_R0KH_R1KH_PULL,
2028			     &packet, &packet_len) < 0)
2029		return -1;
2030
2031	ft_pending_req_ies = wpabuf_alloc_copy(ies, ies_len);
2032	wpabuf_free(sm->ft_pending_req_ies);
2033	sm->ft_pending_req_ies = ft_pending_req_ies;
2034	if (!sm->ft_pending_req_ies) {
2035		os_free(packet);
2036		return -1;
2037	}
2038
2039	tsecs = sm->wpa_auth->conf.rkh_pull_timeout / 1000;
2040	tusecs = (sm->wpa_auth->conf.rkh_pull_timeout % 1000) * 1000;
2041	eloop_register_timeout(tsecs, tusecs, wpa_ft_expire_pull, sm, NULL);
2042
2043	wpa_ft_rrb_oui_send(sm->wpa_auth, r0kh->addr, FT_PACKET_R0KH_R1KH_PULL,
2044			    packet, packet_len);
2045
2046	os_free(packet);
2047
2048	return 0;
2049}
2050
2051
2052int wpa_ft_store_pmk_fils(struct wpa_state_machine *sm,
2053			  const u8 *pmk_r0, const u8 *pmk_r0_name)
2054{
2055	int expires_in = sm->wpa_auth->conf.r0_key_lifetime;
2056	struct vlan_description vlan;
2057	const u8 *identity, *radius_cui;
2058	size_t identity_len, radius_cui_len;
2059	int session_timeout;
2060	size_t pmk_r0_len = wpa_key_mgmt_sha384(sm->wpa_key_mgmt) ?
2061		SHA384_MAC_LEN : PMK_LEN;
2062
2063	if (wpa_ft_get_vlan(sm->wpa_auth, sm->addr, &vlan) < 0) {
2064		wpa_printf(MSG_DEBUG, "FT: vlan not available for STA " MACSTR,
2065			   MAC2STR(sm->addr));
2066		return -1;
2067	}
2068
2069	identity_len = wpa_ft_get_identity(sm->wpa_auth, sm->addr, &identity);
2070	radius_cui_len = wpa_ft_get_radius_cui(sm->wpa_auth, sm->addr,
2071					       &radius_cui);
2072	session_timeout = wpa_ft_get_session_timeout(sm->wpa_auth, sm->addr);
2073
2074	return wpa_ft_store_pmk_r0(sm->wpa_auth, sm->addr, pmk_r0, pmk_r0_len,
2075				   pmk_r0_name, sm->pairwise, &vlan, expires_in,
2076				   session_timeout, identity, identity_len,
2077				   radius_cui, radius_cui_len);
2078}
2079
2080
2081int wpa_auth_derive_ptk_ft(struct wpa_state_machine *sm, struct wpa_ptk *ptk)
2082{
2083	u8 pmk_r0[PMK_LEN_MAX], pmk_r0_name[WPA_PMK_NAME_LEN];
2084	size_t pmk_r0_len = wpa_key_mgmt_sha384(sm->wpa_key_mgmt) ?
2085		SHA384_MAC_LEN : PMK_LEN;
2086	size_t pmk_r1_len = pmk_r0_len;
2087	u8 pmk_r1[PMK_LEN_MAX];
2088	u8 ptk_name[WPA_PMK_NAME_LEN];
2089	const u8 *mdid = sm->wpa_auth->conf.mobility_domain;
2090	const u8 *r0kh = sm->wpa_auth->conf.r0_key_holder;
2091	size_t r0kh_len = sm->wpa_auth->conf.r0_key_holder_len;
2092	const u8 *r1kh = sm->wpa_auth->conf.r1_key_holder;
2093	const u8 *ssid = sm->wpa_auth->conf.ssid;
2094	size_t ssid_len = sm->wpa_auth->conf.ssid_len;
2095	int psk_local = sm->wpa_auth->conf.ft_psk_generate_local;
2096	int expires_in = sm->wpa_auth->conf.r0_key_lifetime;
2097	struct vlan_description vlan;
2098	const u8 *identity, *radius_cui;
2099	size_t identity_len, radius_cui_len;
2100	int session_timeout;
2101	const u8 *mpmk;
2102	size_t mpmk_len;
2103
2104	if (sm->xxkey_len > 0) {
2105		mpmk = sm->xxkey;
2106		mpmk_len = sm->xxkey_len;
2107	} else if (sm->pmksa) {
2108		mpmk = sm->pmksa->pmk;
2109		mpmk_len = sm->pmksa->pmk_len;
2110	} else {
2111		wpa_printf(MSG_DEBUG, "FT: XXKey not available for key "
2112			   "derivation");
2113		return -1;
2114	}
2115
2116	if (wpa_ft_get_vlan(sm->wpa_auth, sm->addr, &vlan) < 0) {
2117		wpa_printf(MSG_DEBUG, "FT: vlan not available for STA " MACSTR,
2118			   MAC2STR(sm->addr));
2119		return -1;
2120	}
2121
2122	identity_len = wpa_ft_get_identity(sm->wpa_auth, sm->addr, &identity);
2123	radius_cui_len = wpa_ft_get_radius_cui(sm->wpa_auth, sm->addr,
2124					       &radius_cui);
2125	session_timeout = wpa_ft_get_session_timeout(sm->wpa_auth, sm->addr);
2126
2127	if (wpa_derive_pmk_r0(mpmk, mpmk_len, ssid, ssid_len, mdid,
2128			      r0kh, r0kh_len, sm->addr,
2129			      pmk_r0, pmk_r0_name,
2130			      wpa_key_mgmt_sha384(sm->wpa_key_mgmt)) < 0)
2131		return -1;
2132	if (!psk_local || !wpa_key_mgmt_ft_psk(sm->wpa_key_mgmt))
2133		wpa_ft_store_pmk_r0(sm->wpa_auth, sm->addr, pmk_r0, pmk_r0_len,
2134				    pmk_r0_name,
2135				    sm->pairwise, &vlan, expires_in,
2136				    session_timeout, identity, identity_len,
2137				    radius_cui, radius_cui_len);
2138
2139	if (wpa_derive_pmk_r1(pmk_r0, pmk_r0_len, pmk_r0_name, r1kh, sm->addr,
2140			      pmk_r1, sm->pmk_r1_name) < 0)
2141		return -1;
2142	if (!psk_local || !wpa_key_mgmt_ft_psk(sm->wpa_key_mgmt))
2143		wpa_ft_store_pmk_r1(sm->wpa_auth, sm->addr, pmk_r1, pmk_r1_len,
2144				    sm->pmk_r1_name, sm->pairwise, &vlan,
2145				    expires_in, session_timeout, identity,
2146				    identity_len, radius_cui, radius_cui_len);
2147
2148	return wpa_pmk_r1_to_ptk(pmk_r1, pmk_r1_len, sm->SNonce, sm->ANonce,
2149				 sm->addr, sm->wpa_auth->addr, sm->pmk_r1_name,
2150				 ptk, ptk_name, sm->wpa_key_mgmt, sm->pairwise,
2151				 0);
2152}
2153
2154
2155static inline int wpa_auth_get_seqnum(struct wpa_authenticator *wpa_auth,
2156				      const u8 *addr, int idx, u8 *seq)
2157{
2158	if (wpa_auth->cb->get_seqnum == NULL)
2159		return -1;
2160	return wpa_auth->cb->get_seqnum(wpa_auth->cb_ctx, addr, idx, seq);
2161}
2162
2163
2164static u8 * wpa_ft_gtk_subelem(struct wpa_state_machine *sm, size_t *len)
2165{
2166	u8 *subelem;
2167	struct wpa_auth_config *conf = &sm->wpa_auth->conf;
2168	struct wpa_group *gsm = sm->group;
2169	size_t subelem_len, pad_len;
2170	const u8 *key;
2171	size_t key_len;
2172	u8 keybuf[WPA_GTK_MAX_LEN];
2173	const u8 *kek;
2174	size_t kek_len;
2175
2176	if (wpa_key_mgmt_fils(sm->wpa_key_mgmt)) {
2177		kek = sm->PTK.kek2;
2178		kek_len = sm->PTK.kek2_len;
2179	} else {
2180		kek = sm->PTK.kek;
2181		kek_len = sm->PTK.kek_len;
2182	}
2183
2184	key_len = gsm->GTK_len;
2185	if (key_len > sizeof(keybuf))
2186		return NULL;
2187
2188	/*
2189	 * Pad key for AES Key Wrap if it is not multiple of 8 bytes or is less
2190	 * than 16 bytes.
2191	 */
2192	pad_len = key_len % 8;
2193	if (pad_len)
2194		pad_len = 8 - pad_len;
2195	if (key_len + pad_len < 16)
2196		pad_len += 8;
2197	if (pad_len && key_len < sizeof(keybuf)) {
2198		os_memcpy(keybuf, gsm->GTK[gsm->GN - 1], key_len);
2199		if (conf->disable_gtk ||
2200		    sm->wpa_key_mgmt == WPA_KEY_MGMT_OSEN) {
2201			/*
2202			 * Provide unique random GTK to each STA to prevent use
2203			 * of GTK in the BSS.
2204			 */
2205			if (random_get_bytes(keybuf, key_len) < 0)
2206				return NULL;
2207		}
2208		os_memset(keybuf + key_len, 0, pad_len);
2209		keybuf[key_len] = 0xdd;
2210		key_len += pad_len;
2211		key = keybuf;
2212	} else if (conf->disable_gtk || sm->wpa_key_mgmt == WPA_KEY_MGMT_OSEN) {
2213		/*
2214		 * Provide unique random GTK to each STA to prevent use of GTK
2215		 * in the BSS.
2216		 */
2217		if (random_get_bytes(keybuf, key_len) < 0)
2218			return NULL;
2219		key = keybuf;
2220	} else {
2221		key = gsm->GTK[gsm->GN - 1];
2222	}
2223
2224	/*
2225	 * Sub-elem ID[1] | Length[1] | Key Info[2] | Key Length[1] | RSC[8] |
2226	 * Key[5..32].
2227	 */
2228	subelem_len = 13 + key_len + 8;
2229	subelem = os_zalloc(subelem_len);
2230	if (subelem == NULL)
2231		return NULL;
2232
2233	subelem[0] = FTIE_SUBELEM_GTK;
2234	subelem[1] = 11 + key_len + 8;
2235	/* Key ID in B0-B1 of Key Info */
2236	WPA_PUT_LE16(&subelem[2], gsm->GN & 0x03);
2237	subelem[4] = gsm->GTK_len;
2238	wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN, subelem + 5);
2239	if (aes_wrap(kek, kek_len, key_len / 8, key, subelem + 13)) {
2240		wpa_printf(MSG_DEBUG,
2241			   "FT: GTK subelem encryption failed: kek_len=%d",
2242			   (int) kek_len);
2243		os_free(subelem);
2244		return NULL;
2245	}
2246
2247	forced_memzero(keybuf, sizeof(keybuf));
2248	*len = subelem_len;
2249	return subelem;
2250}
2251
2252
2253static u8 * wpa_ft_igtk_subelem(struct wpa_state_machine *sm, size_t *len)
2254{
2255	u8 *subelem, *pos;
2256	struct wpa_auth_config *conf = &sm->wpa_auth->conf;
2257	struct wpa_group *gsm = sm->group;
2258	size_t subelem_len;
2259	const u8 *kek, *igtk;
2260	size_t kek_len;
2261	size_t igtk_len;
2262	u8 stub_igtk[WPA_IGTK_MAX_LEN];
2263
2264	if (wpa_key_mgmt_fils(sm->wpa_key_mgmt)) {
2265		kek = sm->PTK.kek2;
2266		kek_len = sm->PTK.kek2_len;
2267	} else {
2268		kek = sm->PTK.kek;
2269		kek_len = sm->PTK.kek_len;
2270	}
2271
2272	igtk_len = wpa_cipher_key_len(sm->wpa_auth->conf.group_mgmt_cipher);
2273
2274	/* Sub-elem ID[1] | Length[1] | KeyID[2] | IPN[6] | Key Length[1] |
2275	 * Key[16+8] */
2276	subelem_len = 1 + 1 + 2 + 6 + 1 + igtk_len + 8;
2277	subelem = os_zalloc(subelem_len);
2278	if (subelem == NULL)
2279		return NULL;
2280
2281	pos = subelem;
2282	*pos++ = FTIE_SUBELEM_IGTK;
2283	*pos++ = subelem_len - 2;
2284	WPA_PUT_LE16(pos, gsm->GN_igtk);
2285	pos += 2;
2286	wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN_igtk, pos);
2287	pos += 6;
2288	*pos++ = igtk_len;
2289	igtk = gsm->IGTK[gsm->GN_igtk - 4];
2290	if (conf->disable_gtk || sm->wpa_key_mgmt == WPA_KEY_MGMT_OSEN) {
2291		/*
2292		 * Provide unique random IGTK to each STA to prevent use of
2293		 * IGTK in the BSS.
2294		 */
2295		if (random_get_bytes(stub_igtk, igtk_len / 8) < 0) {
2296			os_free(subelem);
2297			return NULL;
2298		}
2299		igtk = stub_igtk;
2300	}
2301	if (aes_wrap(kek, kek_len, igtk_len / 8, igtk, pos)) {
2302		wpa_printf(MSG_DEBUG,
2303			   "FT: IGTK subelem encryption failed: kek_len=%d",
2304			   (int) kek_len);
2305		os_free(subelem);
2306		return NULL;
2307	}
2308
2309	*len = subelem_len;
2310	return subelem;
2311}
2312
2313
2314static u8 * wpa_ft_bigtk_subelem(struct wpa_state_machine *sm, size_t *len)
2315{
2316	u8 *subelem, *pos;
2317	struct wpa_group *gsm = sm->group;
2318	size_t subelem_len;
2319	const u8 *kek, *bigtk;
2320	size_t kek_len;
2321	size_t bigtk_len;
2322	u8 stub_bigtk[WPA_IGTK_MAX_LEN];
2323
2324	if (wpa_key_mgmt_fils(sm->wpa_key_mgmt)) {
2325		kek = sm->PTK.kek2;
2326		kek_len = sm->PTK.kek2_len;
2327	} else {
2328		kek = sm->PTK.kek;
2329		kek_len = sm->PTK.kek_len;
2330	}
2331
2332	bigtk_len = wpa_cipher_key_len(sm->wpa_auth->conf.group_mgmt_cipher);
2333
2334	/* Sub-elem ID[1] | Length[1] | KeyID[2] | BIPN[6] | Key Length[1] |
2335	 * Key[16+8] */
2336	subelem_len = 1 + 1 + 2 + 6 + 1 + bigtk_len + 8;
2337	subelem = os_zalloc(subelem_len);
2338	if (subelem == NULL)
2339		return NULL;
2340
2341	pos = subelem;
2342	*pos++ = FTIE_SUBELEM_BIGTK;
2343	*pos++ = subelem_len - 2;
2344	WPA_PUT_LE16(pos, gsm->GN_bigtk);
2345	pos += 2;
2346	wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN_bigtk, pos);
2347	pos += 6;
2348	*pos++ = bigtk_len;
2349	bigtk = gsm->IGTK[gsm->GN_bigtk - 6];
2350	if (sm->wpa_key_mgmt == WPA_KEY_MGMT_OSEN) {
2351		/*
2352		 * Provide unique random BIGTK to each OSEN STA to prevent use
2353		 * of BIGTK in the BSS.
2354		 */
2355		if (random_get_bytes(stub_bigtk, bigtk_len / 8) < 0) {
2356			os_free(subelem);
2357			return NULL;
2358		}
2359		bigtk = stub_bigtk;
2360	}
2361	if (aes_wrap(kek, kek_len, bigtk_len / 8, bigtk, pos)) {
2362		wpa_printf(MSG_DEBUG,
2363			   "FT: BIGTK subelem encryption failed: kek_len=%d",
2364			   (int) kek_len);
2365		os_free(subelem);
2366		return NULL;
2367	}
2368
2369	*len = subelem_len;
2370	return subelem;
2371}
2372
2373
2374static u8 * wpa_ft_process_rdie(struct wpa_state_machine *sm,
2375				u8 *pos, u8 *end, u8 id, u8 descr_count,
2376				const u8 *ies, size_t ies_len)
2377{
2378	struct ieee802_11_elems parse;
2379	struct rsn_rdie *rdie;
2380
2381	wpa_printf(MSG_DEBUG, "FT: Resource Request: id=%d descr_count=%d",
2382		   id, descr_count);
2383	wpa_hexdump(MSG_MSGDUMP, "FT: Resource descriptor IE(s)",
2384		    ies, ies_len);
2385
2386	if (end - pos < (int) sizeof(*rdie)) {
2387		wpa_printf(MSG_ERROR, "FT: Not enough room for response RDIE");
2388		return pos;
2389	}
2390
2391	*pos++ = WLAN_EID_RIC_DATA;
2392	*pos++ = sizeof(*rdie);
2393	rdie = (struct rsn_rdie *) pos;
2394	rdie->id = id;
2395	rdie->descr_count = 0;
2396	rdie->status_code = host_to_le16(WLAN_STATUS_SUCCESS);
2397	pos += sizeof(*rdie);
2398
2399	if (ieee802_11_parse_elems((u8 *) ies, ies_len, &parse, 1) ==
2400	    ParseFailed) {
2401		wpa_printf(MSG_DEBUG, "FT: Failed to parse request IEs");
2402		rdie->status_code =
2403			host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
2404		return pos;
2405	}
2406
2407	if (parse.wmm_tspec) {
2408		struct wmm_tspec_element *tspec;
2409
2410		if (parse.wmm_tspec_len + 2 < (int) sizeof(*tspec)) {
2411			wpa_printf(MSG_DEBUG, "FT: Too short WMM TSPEC IE "
2412				   "(%d)", (int) parse.wmm_tspec_len);
2413			rdie->status_code =
2414				host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
2415			return pos;
2416		}
2417		if (end - pos < (int) sizeof(*tspec)) {
2418			wpa_printf(MSG_ERROR, "FT: Not enough room for "
2419				   "response TSPEC");
2420			rdie->status_code =
2421				host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
2422			return pos;
2423		}
2424		tspec = (struct wmm_tspec_element *) pos;
2425		os_memcpy(tspec, parse.wmm_tspec - 2, sizeof(*tspec));
2426	}
2427
2428#ifdef NEED_AP_MLME
2429	if (parse.wmm_tspec && sm->wpa_auth->conf.ap_mlme) {
2430		int res;
2431
2432		res = wmm_process_tspec((struct wmm_tspec_element *) pos);
2433		wpa_printf(MSG_DEBUG, "FT: ADDTS processing result: %d", res);
2434		if (res == WMM_ADDTS_STATUS_INVALID_PARAMETERS)
2435			rdie->status_code =
2436				host_to_le16(WLAN_STATUS_INVALID_PARAMETERS);
2437		else if (res == WMM_ADDTS_STATUS_REFUSED)
2438			rdie->status_code =
2439				host_to_le16(WLAN_STATUS_REQUEST_DECLINED);
2440		else {
2441			/* TSPEC accepted; include updated TSPEC in response */
2442			rdie->descr_count = 1;
2443			pos += sizeof(struct wmm_tspec_element);
2444		}
2445		return pos;
2446	}
2447#endif /* NEED_AP_MLME */
2448
2449	if (parse.wmm_tspec && !sm->wpa_auth->conf.ap_mlme) {
2450		int res;
2451
2452		res = wpa_ft_add_tspec(sm->wpa_auth, sm->addr, pos,
2453				       sizeof(struct wmm_tspec_element));
2454		if (res >= 0) {
2455			if (res)
2456				rdie->status_code = host_to_le16(res);
2457			else {
2458				/* TSPEC accepted; include updated TSPEC in
2459				 * response */
2460				rdie->descr_count = 1;
2461				pos += sizeof(struct wmm_tspec_element);
2462			}
2463			return pos;
2464		}
2465	}
2466
2467	wpa_printf(MSG_DEBUG, "FT: No supported resource requested");
2468	rdie->status_code = host_to_le16(WLAN_STATUS_UNSPECIFIED_FAILURE);
2469	return pos;
2470}
2471
2472
2473static u8 * wpa_ft_process_ric(struct wpa_state_machine *sm, u8 *pos, u8 *end,
2474			       const u8 *ric, size_t ric_len)
2475{
2476	const u8 *rpos, *start;
2477	const struct rsn_rdie *rdie;
2478
2479	wpa_hexdump(MSG_MSGDUMP, "FT: RIC Request", ric, ric_len);
2480
2481	rpos = ric;
2482	while (rpos + sizeof(*rdie) < ric + ric_len) {
2483		if (rpos[0] != WLAN_EID_RIC_DATA || rpos[1] < sizeof(*rdie) ||
2484		    rpos + 2 + rpos[1] > ric + ric_len)
2485			break;
2486		rdie = (const struct rsn_rdie *) (rpos + 2);
2487		rpos += 2 + rpos[1];
2488		start = rpos;
2489
2490		while (rpos + 2 <= ric + ric_len &&
2491		       rpos + 2 + rpos[1] <= ric + ric_len) {
2492			if (rpos[0] == WLAN_EID_RIC_DATA)
2493				break;
2494			rpos += 2 + rpos[1];
2495		}
2496		pos = wpa_ft_process_rdie(sm, pos, end, rdie->id,
2497					  rdie->descr_count,
2498					  start, rpos - start);
2499	}
2500
2501	return pos;
2502}
2503
2504
2505u8 * wpa_sm_write_assoc_resp_ies(struct wpa_state_machine *sm, u8 *pos,
2506				 size_t max_len, int auth_alg,
2507				 const u8 *req_ies, size_t req_ies_len,
2508				 int omit_rsnxe)
2509{
2510	u8 *end, *mdie, *ftie, *rsnie = NULL, *r0kh_id, *subelem = NULL;
2511	u8 *fte_mic, *elem_count;
2512	size_t mdie_len, ftie_len, rsnie_len = 0, r0kh_id_len, subelem_len = 0;
2513	u8 rsnxe_buf[10], *rsnxe = rsnxe_buf;
2514	size_t rsnxe_len;
2515	int rsnxe_used;
2516	int res;
2517	struct wpa_auth_config *conf;
2518	struct wpa_ft_ies parse;
2519	u8 *ric_start;
2520	u8 *anonce, *snonce;
2521	const u8 *kck;
2522	size_t kck_len;
2523	int use_sha384;
2524
2525	if (sm == NULL)
2526		return pos;
2527
2528	use_sha384 = wpa_key_mgmt_sha384(sm->wpa_key_mgmt);
2529	conf = &sm->wpa_auth->conf;
2530
2531	if (!wpa_key_mgmt_ft(sm->wpa_key_mgmt))
2532		return pos;
2533
2534	end = pos + max_len;
2535
2536#ifdef CONFIG_TESTING_OPTIONS
2537	if (auth_alg == WLAN_AUTH_FT &&
2538	    sm->wpa_auth->conf.rsne_override_ft_set) {
2539		wpa_printf(MSG_DEBUG,
2540			   "TESTING: RSNE FT override for MIC calculation");
2541		rsnie = sm->wpa_auth->conf.rsne_override_ft;
2542		rsnie_len = sm->wpa_auth->conf.rsne_override_ft_len;
2543		if (end - pos < (long int) rsnie_len)
2544			return pos;
2545		os_memcpy(pos, rsnie, rsnie_len);
2546		rsnie = pos;
2547		pos += rsnie_len;
2548		if (rsnie_len > PMKID_LEN && sm->pmk_r1_name_valid) {
2549			int idx;
2550
2551			/* Replace all 0xff PMKID with the valid PMKR1Name */
2552			for (idx = 0; idx < PMKID_LEN; idx++) {
2553				if (rsnie[rsnie_len - 1 - idx] != 0xff)
2554					break;
2555			}
2556			if (idx == PMKID_LEN)
2557				os_memcpy(&rsnie[rsnie_len - PMKID_LEN],
2558					  sm->pmk_r1_name, WPA_PMK_NAME_LEN);
2559		}
2560	} else
2561#endif /* CONFIG_TESTING_OPTIONS */
2562	if (auth_alg == WLAN_AUTH_FT ||
2563	    ((auth_alg == WLAN_AUTH_FILS_SK ||
2564	      auth_alg == WLAN_AUTH_FILS_SK_PFS ||
2565	      auth_alg == WLAN_AUTH_FILS_PK) &&
2566	     (sm->wpa_key_mgmt & (WPA_KEY_MGMT_FT_FILS_SHA256 |
2567				  WPA_KEY_MGMT_FT_FILS_SHA384)))) {
2568		if (!sm->pmk_r1_name_valid) {
2569			wpa_printf(MSG_ERROR,
2570				   "FT: PMKR1Name is not valid for Assoc Resp RSNE");
2571			return NULL;
2572		}
2573		wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name for Assoc Resp RSNE",
2574			    sm->pmk_r1_name, WPA_PMK_NAME_LEN);
2575		/*
2576		 * RSN (only present if this is a Reassociation Response and
2577		 * part of a fast BSS transition; or if this is a
2578		 * (Re)Association Response frame during an FT initial mobility
2579		 * domain association using FILS)
2580		 */
2581		res = wpa_write_rsn_ie(conf, pos, end - pos, sm->pmk_r1_name);
2582		if (res < 0)
2583			return NULL;
2584		rsnie = pos;
2585		rsnie_len = res;
2586		pos += res;
2587	}
2588
2589	/* Mobility Domain Information */
2590	res = wpa_write_mdie(conf, pos, end - pos);
2591	if (res < 0)
2592		return NULL;
2593	mdie = pos;
2594	mdie_len = res;
2595	pos += res;
2596
2597	/* Fast BSS Transition Information */
2598	if (auth_alg == WLAN_AUTH_FT) {
2599		subelem = wpa_ft_gtk_subelem(sm, &subelem_len);
2600		if (!subelem) {
2601			wpa_printf(MSG_DEBUG,
2602				   "FT: Failed to add GTK subelement");
2603			return NULL;
2604		}
2605		r0kh_id = sm->r0kh_id;
2606		r0kh_id_len = sm->r0kh_id_len;
2607		anonce = sm->ANonce;
2608		snonce = sm->SNonce;
2609		if (sm->mgmt_frame_prot) {
2610			u8 *igtk;
2611			size_t igtk_len;
2612			u8 *nbuf;
2613			igtk = wpa_ft_igtk_subelem(sm, &igtk_len);
2614			if (igtk == NULL) {
2615				wpa_printf(MSG_DEBUG,
2616					   "FT: Failed to add IGTK subelement");
2617				os_free(subelem);
2618				return NULL;
2619			}
2620			nbuf = os_realloc(subelem, subelem_len + igtk_len);
2621			if (nbuf == NULL) {
2622				os_free(subelem);
2623				os_free(igtk);
2624				return NULL;
2625			}
2626			subelem = nbuf;
2627			os_memcpy(subelem + subelem_len, igtk, igtk_len);
2628			subelem_len += igtk_len;
2629			os_free(igtk);
2630		}
2631		if (sm->mgmt_frame_prot && conf->beacon_prot) {
2632			u8 *bigtk;
2633			size_t bigtk_len;
2634			u8 *nbuf;
2635
2636			bigtk = wpa_ft_bigtk_subelem(sm, &bigtk_len);
2637			if (!bigtk) {
2638				wpa_printf(MSG_DEBUG,
2639					   "FT: Failed to add BIGTK subelement");
2640				os_free(subelem);
2641				return NULL;
2642			}
2643			nbuf = os_realloc(subelem, subelem_len + bigtk_len);
2644			if (!nbuf) {
2645				os_free(subelem);
2646				os_free(bigtk);
2647				return NULL;
2648			}
2649			subelem = nbuf;
2650			os_memcpy(subelem + subelem_len, bigtk, bigtk_len);
2651			subelem_len += bigtk_len;
2652			os_free(bigtk);
2653		}
2654#ifdef CONFIG_OCV
2655		if (wpa_auth_uses_ocv(sm)) {
2656			struct wpa_channel_info ci;
2657			u8 *nbuf, *ocipos;
2658
2659			if (wpa_channel_info(sm->wpa_auth, &ci) != 0) {
2660				wpa_printf(MSG_WARNING,
2661					   "Failed to get channel info for OCI element");
2662				os_free(subelem);
2663				return NULL;
2664			}
2665#ifdef CONFIG_TESTING_OPTIONS
2666			if (conf->oci_freq_override_ft_assoc) {
2667				wpa_printf(MSG_INFO,
2668					   "TEST: Override OCI frequency %d -> %u MHz",
2669					   ci.frequency,
2670					   conf->oci_freq_override_ft_assoc);
2671				ci.frequency = conf->oci_freq_override_ft_assoc;
2672			}
2673#endif /* CONFIG_TESTING_OPTIONS */
2674
2675			subelem_len += 2 + OCV_OCI_LEN;
2676			nbuf = os_realloc(subelem, subelem_len);
2677			if (!nbuf) {
2678				os_free(subelem);
2679				return NULL;
2680			}
2681			subelem = nbuf;
2682
2683			ocipos = subelem + subelem_len - 2 - OCV_OCI_LEN;
2684			*ocipos++ = FTIE_SUBELEM_OCI;
2685			*ocipos++ = OCV_OCI_LEN;
2686			if (ocv_insert_oci(&ci, &ocipos) < 0) {
2687				os_free(subelem);
2688				return NULL;
2689			}
2690		}
2691#endif /* CONFIG_OCV */
2692	} else {
2693		r0kh_id = conf->r0_key_holder;
2694		r0kh_id_len = conf->r0_key_holder_len;
2695		anonce = NULL;
2696		snonce = NULL;
2697	}
2698	rsnxe_used = (auth_alg == WLAN_AUTH_FT) &&
2699		(conf->sae_pwe == 1 || conf->sae_pwe == 2);
2700#ifdef CONFIG_TESTING_OPTIONS
2701	if (sm->wpa_auth->conf.ft_rsnxe_used) {
2702		rsnxe_used = sm->wpa_auth->conf.ft_rsnxe_used == 1;
2703		wpa_printf(MSG_DEBUG, "TESTING: FT: Force RSNXE Used %d",
2704			   rsnxe_used);
2705	}
2706#endif /* CONFIG_TESTING_OPTIONS */
2707	res = wpa_write_ftie(conf, use_sha384, r0kh_id, r0kh_id_len,
2708			     anonce, snonce, pos, end - pos,
2709			     subelem, subelem_len, rsnxe_used);
2710	os_free(subelem);
2711	if (res < 0)
2712		return NULL;
2713	ftie = pos;
2714	ftie_len = res;
2715	pos += res;
2716
2717	if (use_sha384) {
2718		struct rsn_ftie_sha384 *_ftie =
2719			(struct rsn_ftie_sha384 *) (ftie + 2);
2720
2721		fte_mic = _ftie->mic;
2722		elem_count = &_ftie->mic_control[1];
2723	} else {
2724		struct rsn_ftie *_ftie = (struct rsn_ftie *) (ftie + 2);
2725
2726		fte_mic = _ftie->mic;
2727		elem_count = &_ftie->mic_control[1];
2728	}
2729	if (auth_alg == WLAN_AUTH_FT)
2730		*elem_count = 3; /* Information element count */
2731
2732	ric_start = pos;
2733	if (wpa_ft_parse_ies(req_ies, req_ies_len, &parse, use_sha384) == 0
2734	    && parse.ric) {
2735		pos = wpa_ft_process_ric(sm, pos, end, parse.ric,
2736					 parse.ric_len);
2737		if (auth_alg == WLAN_AUTH_FT)
2738			*elem_count +=
2739				ieee802_11_ie_count(ric_start,
2740						    pos - ric_start);
2741	}
2742	if (ric_start == pos)
2743		ric_start = NULL;
2744
2745	if (omit_rsnxe) {
2746		rsnxe_len = 0;
2747	} else {
2748		res = wpa_write_rsnxe(&sm->wpa_auth->conf, rsnxe,
2749				      sizeof(rsnxe_buf));
2750		if (res < 0)
2751			return NULL;
2752		rsnxe_len = res;
2753	}
2754#ifdef CONFIG_TESTING_OPTIONS
2755	if (auth_alg == WLAN_AUTH_FT &&
2756	    sm->wpa_auth->conf.rsnxe_override_ft_set) {
2757		wpa_printf(MSG_DEBUG,
2758			   "TESTING: RSNXE FT override for MIC calculation");
2759		rsnxe = sm->wpa_auth->conf.rsnxe_override_ft;
2760		rsnxe_len = sm->wpa_auth->conf.rsnxe_override_ft_len;
2761	}
2762#endif /* CONFIG_TESTING_OPTIONS */
2763	if (auth_alg == WLAN_AUTH_FT && rsnxe_len)
2764		*elem_count += 1;
2765
2766	if (wpa_key_mgmt_fils(sm->wpa_key_mgmt)) {
2767		kck = sm->PTK.kck2;
2768		kck_len = sm->PTK.kck2_len;
2769	} else {
2770		kck = sm->PTK.kck;
2771		kck_len = sm->PTK.kck_len;
2772	}
2773	if (auth_alg == WLAN_AUTH_FT &&
2774	    wpa_ft_mic(kck, kck_len, sm->addr, sm->wpa_auth->addr, 6,
2775		       mdie, mdie_len, ftie, ftie_len,
2776		       rsnie, rsnie_len,
2777		       ric_start, ric_start ? pos - ric_start : 0,
2778		       rsnxe_len ? rsnxe : NULL, rsnxe_len,
2779		       fte_mic) < 0) {
2780		wpa_printf(MSG_DEBUG, "FT: Failed to calculate MIC");
2781		return NULL;
2782	}
2783
2784	os_free(sm->assoc_resp_ftie);
2785	sm->assoc_resp_ftie = os_malloc(ftie_len);
2786	if (!sm->assoc_resp_ftie)
2787		return NULL;
2788	os_memcpy(sm->assoc_resp_ftie, ftie, ftie_len);
2789
2790	return pos;
2791}
2792
2793
2794static inline int wpa_auth_set_key(struct wpa_authenticator *wpa_auth,
2795				   int vlan_id,
2796				   enum wpa_alg alg, const u8 *addr, int idx,
2797				   u8 *key, size_t key_len,
2798				   enum key_flag key_flag)
2799{
2800	if (wpa_auth->cb->set_key == NULL)
2801		return -1;
2802	return wpa_auth->cb->set_key(wpa_auth->cb_ctx, vlan_id, alg, addr, idx,
2803				     key, key_len, key_flag);
2804}
2805
2806
2807static inline int wpa_auth_add_sta_ft(struct wpa_authenticator *wpa_auth,
2808				      const u8 *addr)
2809{
2810	if (!wpa_auth->cb->add_sta_ft)
2811		return -1;
2812	return wpa_auth->cb->add_sta_ft(wpa_auth->cb_ctx, addr);
2813}
2814
2815
2816void wpa_ft_install_ptk(struct wpa_state_machine *sm, int retry)
2817{
2818	enum wpa_alg alg;
2819	int klen;
2820
2821	/* MLME-SETKEYS.request(PTK) */
2822	alg = wpa_cipher_to_alg(sm->pairwise);
2823	klen = wpa_cipher_key_len(sm->pairwise);
2824	if (!wpa_cipher_valid_pairwise(sm->pairwise)) {
2825		wpa_printf(MSG_DEBUG, "FT: Unknown pairwise alg 0x%x - skip "
2826			   "PTK configuration", sm->pairwise);
2827		return;
2828	}
2829
2830	if (sm->tk_already_set) {
2831		/* Must avoid TK reconfiguration to prevent clearing of TX/RX
2832		 * PN in the driver */
2833		wpa_printf(MSG_DEBUG,
2834			   "FT: Do not re-install same PTK to the driver");
2835		return;
2836	}
2837
2838	if (!retry)
2839		wpa_auth_add_sta_ft(sm->wpa_auth, sm->addr);
2840
2841	/* FIX: add STA entry to kernel/driver here? The set_key will fail
2842	 * most likely without this.. At the moment, STA entry is added only
2843	 * after association has been completed. This function will be called
2844	 * again after association to get the PTK configured, but that could be
2845	 * optimized by adding the STA entry earlier.
2846	 */
2847	if (wpa_auth_set_key(sm->wpa_auth, 0, alg, sm->addr, sm->keyidx_active,
2848			     sm->PTK.tk, klen, KEY_FLAG_PAIRWISE_RX_TX))
2849		return;
2850
2851	/* FIX: MLME-SetProtection.Request(TA, Tx_Rx) */
2852	sm->pairwise_set = true;
2853	sm->tk_already_set = true;
2854}
2855
2856
2857/* Derive PMK-R1 from PSK, check all available PSK */
2858static int wpa_ft_psk_pmk_r1(struct wpa_state_machine *sm,
2859			     const u8 *req_pmk_r1_name,
2860			     u8 *out_pmk_r1, int *out_pairwise,
2861			     struct vlan_description *out_vlan,
2862			     const u8 **out_identity, size_t *out_identity_len,
2863			     const u8 **out_radius_cui,
2864			     size_t *out_radius_cui_len,
2865			     int *out_session_timeout)
2866{
2867	const u8 *pmk = NULL;
2868	u8 pmk_r0[PMK_LEN], pmk_r0_name[WPA_PMK_NAME_LEN];
2869	u8 pmk_r1[PMK_LEN], pmk_r1_name[WPA_PMK_NAME_LEN];
2870	struct wpa_authenticator *wpa_auth = sm->wpa_auth;
2871	const u8 *mdid = wpa_auth->conf.mobility_domain;
2872	const u8 *r0kh = sm->r0kh_id;
2873	size_t r0kh_len = sm->r0kh_id_len;
2874	const u8 *r1kh = wpa_auth->conf.r1_key_holder;
2875	const u8 *ssid = wpa_auth->conf.ssid;
2876	size_t ssid_len = wpa_auth->conf.ssid_len;
2877	int pairwise;
2878
2879	pairwise = sm->pairwise;
2880
2881	for (;;) {
2882		pmk = wpa_ft_get_psk(wpa_auth, sm->addr, sm->p2p_dev_addr,
2883				     pmk);
2884		if (pmk == NULL)
2885			break;
2886
2887		if (wpa_derive_pmk_r0(pmk, PMK_LEN, ssid, ssid_len, mdid, r0kh,
2888				      r0kh_len, sm->addr,
2889				      pmk_r0, pmk_r0_name, 0) < 0 ||
2890		    wpa_derive_pmk_r1(pmk_r0, PMK_LEN, pmk_r0_name, r1kh,
2891				      sm->addr, pmk_r1, pmk_r1_name) < 0 ||
2892		    os_memcmp_const(pmk_r1_name, req_pmk_r1_name,
2893				    WPA_PMK_NAME_LEN) != 0)
2894			continue;
2895
2896		/* We found a PSK that matches the requested pmk_r1_name */
2897		wpa_printf(MSG_DEBUG,
2898			   "FT: Found PSK to generate PMK-R1 locally");
2899		os_memcpy(out_pmk_r1, pmk_r1, PMK_LEN);
2900		if (out_pairwise)
2901			*out_pairwise = pairwise;
2902		os_memcpy(sm->PMK, pmk, PMK_LEN);
2903		sm->pmk_len = PMK_LEN;
2904		if (out_vlan &&
2905		    wpa_ft_get_vlan(sm->wpa_auth, sm->addr, out_vlan) < 0) {
2906			wpa_printf(MSG_DEBUG, "FT: vlan not available for STA "
2907				   MACSTR, MAC2STR(sm->addr));
2908			return -1;
2909		}
2910
2911		if (out_identity && out_identity_len) {
2912			*out_identity_len = wpa_ft_get_identity(
2913				sm->wpa_auth, sm->addr, out_identity);
2914		}
2915
2916		if (out_radius_cui && out_radius_cui_len) {
2917			*out_radius_cui_len = wpa_ft_get_radius_cui(
2918				sm->wpa_auth, sm->addr, out_radius_cui);
2919		}
2920
2921		if (out_session_timeout) {
2922			*out_session_timeout = wpa_ft_get_session_timeout(
2923				sm->wpa_auth, sm->addr);
2924		}
2925
2926		return 0;
2927	}
2928
2929	wpa_printf(MSG_DEBUG,
2930		   "FT: Did not find PSK to generate PMK-R1 locally");
2931	return -1;
2932}
2933
2934
2935/* Detect the configuration the station asked for.
2936 * Required to detect FT-PSK and pairwise cipher.
2937 */
2938static int wpa_ft_set_key_mgmt(struct wpa_state_machine *sm,
2939			       struct wpa_ft_ies *parse)
2940{
2941	int key_mgmt, ciphers;
2942
2943	if (sm->wpa_key_mgmt)
2944		return 0;
2945
2946	key_mgmt = parse->key_mgmt & sm->wpa_auth->conf.wpa_key_mgmt;
2947	if (!key_mgmt) {
2948		wpa_printf(MSG_DEBUG, "FT: Invalid key mgmt (0x%x) from "
2949			   MACSTR, parse->key_mgmt, MAC2STR(sm->addr));
2950		return -1;
2951	}
2952	if (key_mgmt & WPA_KEY_MGMT_FT_IEEE8021X)
2953		sm->wpa_key_mgmt = WPA_KEY_MGMT_FT_IEEE8021X;
2954#ifdef CONFIG_SHA384
2955	else if (key_mgmt & WPA_KEY_MGMT_FT_IEEE8021X_SHA384)
2956		sm->wpa_key_mgmt = WPA_KEY_MGMT_FT_IEEE8021X_SHA384;
2957#endif /* CONFIG_SHA384 */
2958	else if (key_mgmt & WPA_KEY_MGMT_FT_PSK)
2959		sm->wpa_key_mgmt = WPA_KEY_MGMT_FT_PSK;
2960#ifdef CONFIG_FILS
2961	else if (key_mgmt & WPA_KEY_MGMT_FT_FILS_SHA256)
2962		sm->wpa_key_mgmt = WPA_KEY_MGMT_FT_FILS_SHA256;
2963	else if (key_mgmt & WPA_KEY_MGMT_FT_FILS_SHA384)
2964		sm->wpa_key_mgmt = WPA_KEY_MGMT_FT_FILS_SHA384;
2965#endif /* CONFIG_FILS */
2966	ciphers = parse->pairwise_cipher & sm->wpa_auth->conf.rsn_pairwise;
2967	if (!ciphers) {
2968		wpa_printf(MSG_DEBUG, "FT: Invalid pairwise cipher (0x%x) from "
2969			   MACSTR,
2970			   parse->pairwise_cipher, MAC2STR(sm->addr));
2971		return -1;
2972	}
2973	sm->pairwise = wpa_pick_pairwise_cipher(ciphers, 0);
2974
2975	return 0;
2976}
2977
2978
2979static int wpa_ft_local_derive_pmk_r1(struct wpa_authenticator *wpa_auth,
2980				      struct wpa_state_machine *sm,
2981				      const u8 *r0kh_id, size_t r0kh_id_len,
2982				      const u8 *req_pmk_r0_name,
2983				      const u8 *req_pmk_r1_name,
2984				      u8 *out_pmk_r1, int *out_pairwise,
2985				      struct vlan_description *vlan,
2986				      const u8 **identity, size_t *identity_len,
2987				      const u8 **radius_cui,
2988				      size_t *radius_cui_len,
2989				      int *out_session_timeout)
2990{
2991	struct wpa_auth_config *conf = &wpa_auth->conf;
2992	const struct wpa_ft_pmk_r0_sa *r0;
2993	u8 pmk_r1_name[WPA_PMK_NAME_LEN];
2994	int expires_in = 0;
2995	int session_timeout = 0;
2996	struct os_reltime now;
2997
2998	if (conf->r0_key_holder_len != r0kh_id_len ||
2999	    os_memcmp(conf->r0_key_holder, r0kh_id, conf->r0_key_holder_len) !=
3000	    0)
3001		return -1; /* not our R0KH-ID */
3002
3003	wpa_printf(MSG_DEBUG, "FT: STA R0KH-ID matching local configuration");
3004	if (wpa_ft_fetch_pmk_r0(sm->wpa_auth, sm->addr, req_pmk_r0_name, &r0) <
3005	    0)
3006		return -1; /* no matching PMKR0Name in local cache */
3007
3008	wpa_printf(MSG_DEBUG, "FT: Requested PMKR0Name found in local cache");
3009
3010	if (wpa_derive_pmk_r1(r0->pmk_r0, r0->pmk_r0_len, r0->pmk_r0_name,
3011			      conf->r1_key_holder,
3012			      sm->addr, out_pmk_r1, pmk_r1_name) < 0)
3013		return -1;
3014
3015	os_get_reltime(&now);
3016	if (r0->expiration)
3017		expires_in = r0->expiration - now.sec;
3018
3019	if (r0->session_timeout)
3020		session_timeout = r0->session_timeout - now.sec;
3021
3022	wpa_ft_store_pmk_r1(wpa_auth, sm->addr, out_pmk_r1, r0->pmk_r0_len,
3023			    pmk_r1_name,
3024			    sm->pairwise, r0->vlan, expires_in, session_timeout,
3025			    r0->identity, r0->identity_len,
3026			    r0->radius_cui, r0->radius_cui_len);
3027
3028	*out_pairwise = sm->pairwise;
3029	if (vlan) {
3030		if (r0->vlan)
3031			*vlan = *r0->vlan;
3032		else
3033			os_memset(vlan, 0, sizeof(*vlan));
3034	}
3035
3036	if (identity && identity_len) {
3037		*identity = r0->identity;
3038		*identity_len = r0->identity_len;
3039	}
3040
3041	if (radius_cui && radius_cui_len) {
3042		*radius_cui = r0->radius_cui;
3043		*radius_cui_len = r0->radius_cui_len;
3044	}
3045
3046	*out_session_timeout = session_timeout;
3047
3048	return 0;
3049}
3050
3051
3052static int wpa_ft_process_auth_req(struct wpa_state_machine *sm,
3053				   const u8 *ies, size_t ies_len,
3054				   u8 **resp_ies, size_t *resp_ies_len)
3055{
3056	struct rsn_mdie *mdie;
3057	u8 pmk_r1[PMK_LEN_MAX], pmk_r1_name[WPA_PMK_NAME_LEN];
3058	u8 ptk_name[WPA_PMK_NAME_LEN];
3059	struct wpa_auth_config *conf;
3060	struct wpa_ft_ies parse;
3061	size_t buflen;
3062	int ret;
3063	u8 *pos, *end;
3064	int pairwise, session_timeout = 0;
3065	struct vlan_description vlan;
3066	const u8 *identity, *radius_cui;
3067	size_t identity_len = 0, radius_cui_len = 0;
3068	int use_sha384;
3069	size_t pmk_r1_len, kdk_len;
3070
3071	*resp_ies = NULL;
3072	*resp_ies_len = 0;
3073
3074	sm->pmk_r1_name_valid = 0;
3075	conf = &sm->wpa_auth->conf;
3076
3077	wpa_hexdump(MSG_DEBUG, "FT: Received authentication frame IEs",
3078		    ies, ies_len);
3079
3080	if (wpa_ft_parse_ies(ies, ies_len, &parse, -1)) {
3081		wpa_printf(MSG_DEBUG, "FT: Failed to parse FT IEs");
3082		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3083	}
3084	use_sha384 = wpa_key_mgmt_sha384(parse.key_mgmt);
3085	pmk_r1_len = use_sha384 ? SHA384_MAC_LEN : PMK_LEN;
3086
3087	mdie = (struct rsn_mdie *) parse.mdie;
3088	if (mdie == NULL || parse.mdie_len < sizeof(*mdie) ||
3089	    os_memcmp(mdie->mobility_domain,
3090		      sm->wpa_auth->conf.mobility_domain,
3091		      MOBILITY_DOMAIN_ID_LEN) != 0) {
3092		wpa_printf(MSG_DEBUG, "FT: Invalid MDIE");
3093		return WLAN_STATUS_INVALID_MDIE;
3094	}
3095
3096	if (use_sha384) {
3097		struct rsn_ftie_sha384 *ftie;
3098
3099		ftie = (struct rsn_ftie_sha384 *) parse.ftie;
3100		if (!ftie || parse.ftie_len < sizeof(*ftie)) {
3101			wpa_printf(MSG_DEBUG, "FT: Invalid FTIE");
3102			return WLAN_STATUS_INVALID_FTIE;
3103		}
3104
3105		os_memcpy(sm->SNonce, ftie->snonce, WPA_NONCE_LEN);
3106	} else {
3107		struct rsn_ftie *ftie;
3108
3109		ftie = (struct rsn_ftie *) parse.ftie;
3110		if (!ftie || parse.ftie_len < sizeof(*ftie)) {
3111			wpa_printf(MSG_DEBUG, "FT: Invalid FTIE");
3112			return WLAN_STATUS_INVALID_FTIE;
3113		}
3114
3115		os_memcpy(sm->SNonce, ftie->snonce, WPA_NONCE_LEN);
3116	}
3117
3118	if (parse.r0kh_id == NULL) {
3119		wpa_printf(MSG_DEBUG, "FT: Invalid FTIE - no R0KH-ID");
3120		return WLAN_STATUS_INVALID_FTIE;
3121	}
3122
3123	wpa_hexdump(MSG_DEBUG, "FT: STA R0KH-ID",
3124		    parse.r0kh_id, parse.r0kh_id_len);
3125	os_memcpy(sm->r0kh_id, parse.r0kh_id, parse.r0kh_id_len);
3126	sm->r0kh_id_len = parse.r0kh_id_len;
3127
3128	if (parse.rsn_pmkid == NULL) {
3129		wpa_printf(MSG_DEBUG, "FT: No PMKID in RSNIE");
3130		return WLAN_STATUS_INVALID_PMKID;
3131	}
3132
3133	if (wpa_ft_set_key_mgmt(sm, &parse) < 0)
3134		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3135
3136	wpa_hexdump(MSG_DEBUG, "FT: Requested PMKR0Name",
3137		    parse.rsn_pmkid, WPA_PMK_NAME_LEN);
3138	if (wpa_derive_pmk_r1_name(parse.rsn_pmkid,
3139				   sm->wpa_auth->conf.r1_key_holder, sm->addr,
3140				   pmk_r1_name, use_sha384) < 0)
3141		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3142
3143	if (conf->ft_psk_generate_local &&
3144	    wpa_key_mgmt_ft_psk(sm->wpa_key_mgmt)) {
3145		if (wpa_ft_psk_pmk_r1(sm, pmk_r1_name, pmk_r1, &pairwise,
3146				      &vlan, &identity, &identity_len,
3147				      &radius_cui, &radius_cui_len,
3148				      &session_timeout) < 0)
3149			return WLAN_STATUS_INVALID_PMKID;
3150		wpa_printf(MSG_DEBUG,
3151			   "FT: Generated PMK-R1 for FT-PSK locally");
3152	} else if (wpa_ft_fetch_pmk_r1(sm->wpa_auth, sm->addr, pmk_r1_name,
3153				       pmk_r1, &pmk_r1_len, &pairwise, &vlan,
3154				       &identity, &identity_len, &radius_cui,
3155				       &radius_cui_len, &session_timeout) < 0) {
3156		wpa_printf(MSG_DEBUG,
3157			   "FT: No PMK-R1 available in local cache for the requested PMKR1Name");
3158		if (wpa_ft_local_derive_pmk_r1(sm->wpa_auth, sm,
3159					       parse.r0kh_id, parse.r0kh_id_len,
3160					       parse.rsn_pmkid,
3161					       pmk_r1_name, pmk_r1, &pairwise,
3162					       &vlan, &identity, &identity_len,
3163					       &radius_cui, &radius_cui_len,
3164					       &session_timeout) == 0) {
3165			wpa_printf(MSG_DEBUG,
3166				   "FT: Generated PMK-R1 based on local PMK-R0");
3167			goto pmk_r1_derived;
3168		}
3169
3170		if (wpa_ft_pull_pmk_r1(sm, ies, ies_len, parse.rsn_pmkid) < 0) {
3171			wpa_printf(MSG_DEBUG,
3172				   "FT: Did not have matching PMK-R1 and either unknown or blocked R0KH-ID or NAK from R0KH");
3173			return WLAN_STATUS_INVALID_PMKID;
3174		}
3175
3176		return -1; /* Status pending */
3177	} else {
3178		wpa_printf(MSG_DEBUG, "FT: Found PMKR1Name from local cache");
3179	}
3180
3181pmk_r1_derived:
3182	wpa_hexdump_key(MSG_DEBUG, "FT: Selected PMK-R1", pmk_r1, pmk_r1_len);
3183	sm->pmk_r1_name_valid = 1;
3184	os_memcpy(sm->pmk_r1_name, pmk_r1_name, WPA_PMK_NAME_LEN);
3185	os_memcpy(sm->pmk_r1, pmk_r1, pmk_r1_len);
3186	sm->pmk_r1_len = pmk_r1_len;
3187
3188	if (random_get_bytes(sm->ANonce, WPA_NONCE_LEN)) {
3189		wpa_printf(MSG_DEBUG, "FT: Failed to get random data for "
3190			   "ANonce");
3191		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3192	}
3193
3194	wpa_hexdump(MSG_DEBUG, "FT: Received SNonce",
3195		    sm->SNonce, WPA_NONCE_LEN);
3196	wpa_hexdump(MSG_DEBUG, "FT: Generated ANonce",
3197		    sm->ANonce, WPA_NONCE_LEN);
3198
3199	if (sm->wpa_auth->conf.force_kdk_derivation ||
3200	    (sm->wpa_auth->conf.secure_ltf &&
3201	     ieee802_11_rsnx_capab(sm->rsnxe, WLAN_RSNX_CAPAB_SECURE_LTF)))
3202		kdk_len = WPA_KDK_MAX_LEN;
3203	else
3204		kdk_len = 0;
3205
3206	if (wpa_pmk_r1_to_ptk(pmk_r1, pmk_r1_len, sm->SNonce, sm->ANonce,
3207			      sm->addr, sm->wpa_auth->addr, pmk_r1_name,
3208			      &sm->PTK, ptk_name, sm->wpa_key_mgmt,
3209			      pairwise, kdk_len) < 0)
3210		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3211
3212	sm->pairwise = pairwise;
3213	sm->PTK_valid = true;
3214	sm->tk_already_set = false;
3215	wpa_ft_install_ptk(sm, 0);
3216
3217	if (wpa_ft_set_vlan(sm->wpa_auth, sm->addr, &vlan) < 0) {
3218		wpa_printf(MSG_DEBUG, "FT: Failed to configure VLAN");
3219		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3220	}
3221	if (wpa_ft_set_identity(sm->wpa_auth, sm->addr,
3222				identity, identity_len) < 0 ||
3223	    wpa_ft_set_radius_cui(sm->wpa_auth, sm->addr,
3224				  radius_cui, radius_cui_len) < 0) {
3225		wpa_printf(MSG_DEBUG, "FT: Failed to configure identity/CUI");
3226		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3227	}
3228	wpa_ft_set_session_timeout(sm->wpa_auth, sm->addr, session_timeout);
3229
3230	buflen = 2 + sizeof(struct rsn_mdie) + 2 + sizeof(struct rsn_ftie) +
3231		2 + FT_R1KH_ID_LEN + 200;
3232	*resp_ies = os_zalloc(buflen);
3233	if (*resp_ies == NULL)
3234		goto fail;
3235
3236	pos = *resp_ies;
3237	end = *resp_ies + buflen;
3238
3239	ret = wpa_write_rsn_ie(conf, pos, end - pos, parse.rsn_pmkid);
3240	if (ret < 0)
3241		goto fail;
3242	pos += ret;
3243
3244	ret = wpa_write_mdie(conf, pos, end - pos);
3245	if (ret < 0)
3246		goto fail;
3247	pos += ret;
3248
3249	ret = wpa_write_ftie(conf, use_sha384, parse.r0kh_id, parse.r0kh_id_len,
3250			     sm->ANonce, sm->SNonce, pos, end - pos, NULL, 0,
3251			     0);
3252	if (ret < 0)
3253		goto fail;
3254	pos += ret;
3255
3256	*resp_ies_len = pos - *resp_ies;
3257
3258	return WLAN_STATUS_SUCCESS;
3259fail:
3260	os_free(*resp_ies);
3261	*resp_ies = NULL;
3262	return WLAN_STATUS_UNSPECIFIED_FAILURE;
3263}
3264
3265
3266void wpa_ft_process_auth(struct wpa_state_machine *sm, const u8 *bssid,
3267			 u16 auth_transaction, const u8 *ies, size_t ies_len,
3268			 void (*cb)(void *ctx, const u8 *dst, const u8 *bssid,
3269				    u16 auth_transaction, u16 status,
3270				    const u8 *ies, size_t ies_len),
3271			 void *ctx)
3272{
3273	u16 status;
3274	u8 *resp_ies;
3275	size_t resp_ies_len;
3276	int res;
3277
3278	if (sm == NULL) {
3279		wpa_printf(MSG_DEBUG, "FT: Received authentication frame, but "
3280			   "WPA SM not available");
3281		return;
3282	}
3283
3284	wpa_printf(MSG_DEBUG, "FT: Received authentication frame: STA=" MACSTR
3285		   " BSSID=" MACSTR " transaction=%d",
3286		   MAC2STR(sm->addr), MAC2STR(bssid), auth_transaction);
3287	sm->ft_pending_cb = cb;
3288	sm->ft_pending_cb_ctx = ctx;
3289	sm->ft_pending_auth_transaction = auth_transaction;
3290	sm->ft_pending_pull_left_retries = sm->wpa_auth->conf.rkh_pull_retries;
3291	res = wpa_ft_process_auth_req(sm, ies, ies_len, &resp_ies,
3292				      &resp_ies_len);
3293	if (res < 0) {
3294		wpa_printf(MSG_DEBUG, "FT: Callback postponed until response is available");
3295		return;
3296	}
3297	status = res;
3298
3299	wpa_printf(MSG_DEBUG, "FT: FT authentication response: dst=" MACSTR
3300		   " auth_transaction=%d status=%u (%s)",
3301		   MAC2STR(sm->addr), auth_transaction + 1, status,
3302		   status2str(status));
3303	wpa_hexdump(MSG_DEBUG, "FT: Response IEs", resp_ies, resp_ies_len);
3304	cb(ctx, sm->addr, bssid, auth_transaction + 1, status,
3305	   resp_ies, resp_ies_len);
3306	os_free(resp_ies);
3307}
3308
3309
3310int wpa_ft_validate_reassoc(struct wpa_state_machine *sm, const u8 *ies,
3311			    size_t ies_len)
3312{
3313	struct wpa_ft_ies parse;
3314	struct rsn_mdie *mdie;
3315	u8 mic[WPA_EAPOL_KEY_MIC_MAX_LEN];
3316	size_t mic_len = 16;
3317	unsigned int count;
3318	const u8 *kck;
3319	size_t kck_len;
3320	int use_sha384;
3321	const u8 *anonce, *snonce, *fte_mic;
3322	u8 fte_elem_count;
3323	int rsnxe_used;
3324	struct wpa_auth_config *conf;
3325
3326	if (sm == NULL)
3327		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3328
3329	conf = &sm->wpa_auth->conf;
3330	use_sha384 = wpa_key_mgmt_sha384(sm->wpa_key_mgmt);
3331
3332	wpa_hexdump(MSG_DEBUG, "FT: Reassoc Req IEs", ies, ies_len);
3333
3334	if (wpa_ft_parse_ies(ies, ies_len, &parse, use_sha384) < 0) {
3335		wpa_printf(MSG_DEBUG, "FT: Failed to parse FT IEs");
3336		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3337	}
3338
3339	if (parse.rsn == NULL) {
3340		wpa_printf(MSG_DEBUG, "FT: No RSNIE in Reassoc Req");
3341		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3342	}
3343
3344	if (parse.rsn_pmkid == NULL) {
3345		wpa_printf(MSG_DEBUG, "FT: No PMKID in RSNIE");
3346		return WLAN_STATUS_INVALID_PMKID;
3347	}
3348
3349	if (os_memcmp_const(parse.rsn_pmkid, sm->pmk_r1_name, WPA_PMK_NAME_LEN)
3350	    != 0) {
3351		wpa_printf(MSG_DEBUG, "FT: PMKID in Reassoc Req did not match "
3352			   "with the PMKR1Name derived from auth request");
3353		return WLAN_STATUS_INVALID_PMKID;
3354	}
3355
3356	mdie = (struct rsn_mdie *) parse.mdie;
3357	if (mdie == NULL || parse.mdie_len < sizeof(*mdie) ||
3358	    os_memcmp(mdie->mobility_domain, conf->mobility_domain,
3359		      MOBILITY_DOMAIN_ID_LEN) != 0) {
3360		wpa_printf(MSG_DEBUG, "FT: Invalid MDIE");
3361		return WLAN_STATUS_INVALID_MDIE;
3362	}
3363
3364	if (use_sha384) {
3365		struct rsn_ftie_sha384 *ftie;
3366
3367		ftie = (struct rsn_ftie_sha384 *) parse.ftie;
3368		if (ftie == NULL || parse.ftie_len < sizeof(*ftie)) {
3369			wpa_printf(MSG_DEBUG, "FT: Invalid FTIE");
3370			return WLAN_STATUS_INVALID_FTIE;
3371		}
3372
3373		anonce = ftie->anonce;
3374		snonce = ftie->snonce;
3375		rsnxe_used = ftie->mic_control[0] & 0x01;
3376		fte_elem_count = ftie->mic_control[1];
3377		fte_mic = ftie->mic;
3378	} else {
3379		struct rsn_ftie *ftie;
3380
3381		ftie = (struct rsn_ftie *) parse.ftie;
3382		if (ftie == NULL || parse.ftie_len < sizeof(*ftie)) {
3383			wpa_printf(MSG_DEBUG, "FT: Invalid FTIE");
3384			return WLAN_STATUS_INVALID_FTIE;
3385		}
3386
3387		anonce = ftie->anonce;
3388		snonce = ftie->snonce;
3389		rsnxe_used = ftie->mic_control[0] & 0x01;
3390		fte_elem_count = ftie->mic_control[1];
3391		fte_mic = ftie->mic;
3392	}
3393
3394	if (os_memcmp(snonce, sm->SNonce, WPA_NONCE_LEN) != 0) {
3395		wpa_printf(MSG_DEBUG, "FT: SNonce mismatch in FTIE");
3396		wpa_hexdump(MSG_DEBUG, "FT: Received SNonce",
3397			    snonce, WPA_NONCE_LEN);
3398		wpa_hexdump(MSG_DEBUG, "FT: Expected SNonce",
3399			    sm->SNonce, WPA_NONCE_LEN);
3400		return WLAN_STATUS_INVALID_FTIE;
3401	}
3402
3403	if (os_memcmp(anonce, sm->ANonce, WPA_NONCE_LEN) != 0) {
3404		wpa_printf(MSG_DEBUG, "FT: ANonce mismatch in FTIE");
3405		wpa_hexdump(MSG_DEBUG, "FT: Received ANonce",
3406			    anonce, WPA_NONCE_LEN);
3407		wpa_hexdump(MSG_DEBUG, "FT: Expected ANonce",
3408			    sm->ANonce, WPA_NONCE_LEN);
3409		return WLAN_STATUS_INVALID_FTIE;
3410	}
3411
3412
3413	if (parse.r0kh_id == NULL) {
3414		wpa_printf(MSG_DEBUG, "FT: No R0KH-ID subelem in FTIE");
3415		return WLAN_STATUS_INVALID_FTIE;
3416	}
3417
3418	if (parse.r0kh_id_len != sm->r0kh_id_len ||
3419	    os_memcmp_const(parse.r0kh_id, sm->r0kh_id, parse.r0kh_id_len) != 0)
3420	{
3421		wpa_printf(MSG_DEBUG, "FT: R0KH-ID in FTIE did not match with "
3422			   "the current R0KH-ID");
3423		wpa_hexdump(MSG_DEBUG, "FT: R0KH-ID in FTIE",
3424			    parse.r0kh_id, parse.r0kh_id_len);
3425		wpa_hexdump(MSG_DEBUG, "FT: The current R0KH-ID",
3426			    sm->r0kh_id, sm->r0kh_id_len);
3427		return WLAN_STATUS_INVALID_FTIE;
3428	}
3429
3430	if (parse.r1kh_id == NULL) {
3431		wpa_printf(MSG_DEBUG, "FT: No R1KH-ID subelem in FTIE");
3432		return WLAN_STATUS_INVALID_FTIE;
3433	}
3434
3435	if (os_memcmp_const(parse.r1kh_id, conf->r1_key_holder,
3436			    FT_R1KH_ID_LEN) != 0) {
3437		wpa_printf(MSG_DEBUG, "FT: Unknown R1KH-ID used in "
3438			   "ReassocReq");
3439		wpa_hexdump(MSG_DEBUG, "FT: R1KH-ID in FTIE",
3440			    parse.r1kh_id, FT_R1KH_ID_LEN);
3441		wpa_hexdump(MSG_DEBUG, "FT: Expected R1KH-ID",
3442			    conf->r1_key_holder, FT_R1KH_ID_LEN);
3443		return WLAN_STATUS_INVALID_FTIE;
3444	}
3445
3446	if (parse.rsn_pmkid == NULL ||
3447	    os_memcmp_const(parse.rsn_pmkid, sm->pmk_r1_name, WPA_PMK_NAME_LEN))
3448	{
3449		wpa_printf(MSG_DEBUG, "FT: No matching PMKR1Name (PMKID) in "
3450			   "RSNIE (pmkid=%d)", !!parse.rsn_pmkid);
3451		return WLAN_STATUS_INVALID_PMKID;
3452	}
3453
3454	count = 3;
3455	if (parse.ric)
3456		count += ieee802_11_ie_count(parse.ric, parse.ric_len);
3457	if (parse.rsnxe)
3458		count++;
3459	if (fte_elem_count != count) {
3460		wpa_printf(MSG_DEBUG, "FT: Unexpected IE count in MIC "
3461			   "Control: received %u expected %u",
3462			   fte_elem_count, count);
3463		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3464	}
3465
3466	if (wpa_key_mgmt_fils(sm->wpa_key_mgmt)) {
3467		kck = sm->PTK.kck2;
3468		kck_len = sm->PTK.kck2_len;
3469	} else {
3470		kck = sm->PTK.kck;
3471		kck_len = sm->PTK.kck_len;
3472	}
3473	if (wpa_ft_mic(kck, kck_len, sm->addr, sm->wpa_auth->addr, 5,
3474		       parse.mdie - 2, parse.mdie_len + 2,
3475		       parse.ftie - 2, parse.ftie_len + 2,
3476		       parse.rsn - 2, parse.rsn_len + 2,
3477		       parse.ric, parse.ric_len,
3478		       parse.rsnxe ? parse.rsnxe - 2 : NULL,
3479		       parse.rsnxe ? parse.rsnxe_len + 2 : 0,
3480		       mic) < 0) {
3481		wpa_printf(MSG_DEBUG, "FT: Failed to calculate MIC");
3482		return WLAN_STATUS_UNSPECIFIED_FAILURE;
3483	}
3484
3485	if (os_memcmp_const(mic, fte_mic, mic_len) != 0) {
3486		wpa_printf(MSG_DEBUG, "FT: Invalid MIC in FTIE");
3487		wpa_printf(MSG_DEBUG, "FT: addr=" MACSTR " auth_addr=" MACSTR,
3488			   MAC2STR(sm->addr), MAC2STR(sm->wpa_auth->addr));
3489		wpa_hexdump(MSG_MSGDUMP, "FT: Received MIC",
3490			    fte_mic, mic_len);
3491		wpa_hexdump(MSG_MSGDUMP, "FT: Calculated MIC", mic, mic_len);
3492		wpa_hexdump(MSG_MSGDUMP, "FT: MDIE",
3493			    parse.mdie - 2, parse.mdie_len + 2);
3494		wpa_hexdump(MSG_MSGDUMP, "FT: FTIE",
3495			    parse.ftie - 2, parse.ftie_len + 2);
3496		wpa_hexdump(MSG_MSGDUMP, "FT: RSN",
3497			    parse.rsn - 2, parse.rsn_len + 2);
3498		wpa_hexdump(MSG_MSGDUMP, "FT: RSNXE",
3499			    parse.rsnxe ? parse.rsnxe - 2 : NULL,
3500			    parse.rsnxe ? parse.rsnxe_len + 2 : 0);
3501		return WLAN_STATUS_INVALID_FTIE;
3502	}
3503
3504	if (rsnxe_used && (conf->sae_pwe == 1 || conf->sae_pwe == 2) &&
3505	    !parse.rsnxe) {
3506		wpa_printf(MSG_INFO,
3507			   "FT: FTE indicated that STA uses RSNXE, but RSNXE was not included");
3508		return -1; /* discard request */
3509	}
3510
3511#ifdef CONFIG_OCV
3512	if (wpa_auth_uses_ocv(sm)) {
3513		struct wpa_channel_info ci;
3514		int tx_chanwidth;
3515		int tx_seg1_idx;
3516		enum oci_verify_result res;
3517
3518		if (wpa_channel_info(sm->wpa_auth, &ci) != 0) {
3519			wpa_printf(MSG_WARNING,
3520				   "Failed to get channel info to validate received OCI in (Re)Assoc Request");
3521			return WLAN_STATUS_UNSPECIFIED_FAILURE;
3522		}
3523
3524		if (get_sta_tx_parameters(sm,
3525					  channel_width_to_int(ci.chanwidth),
3526					  ci.seg1_idx, &tx_chanwidth,
3527					  &tx_seg1_idx) < 0)
3528			return WLAN_STATUS_UNSPECIFIED_FAILURE;
3529
3530		res = ocv_verify_tx_params(parse.oci, parse.oci_len, &ci,
3531					   tx_chanwidth, tx_seg1_idx);
3532		if (wpa_auth_uses_ocv(sm) == 2 && res == OCI_NOT_FOUND) {
3533			/* Work around misbehaving STAs */
3534			wpa_printf(MSG_INFO,
3535				   "Disable OCV with a STA that does not send OCI");
3536			wpa_auth_set_ocv(sm, 0);
3537		} else if (res != OCI_SUCCESS) {
3538			wpa_printf(MSG_WARNING, "OCV failed: %s", ocv_errorstr);
3539			if (sm->wpa_auth->conf.msg_ctx)
3540				wpa_msg(sm->wpa_auth->conf.msg_ctx, MSG_INFO,
3541					OCV_FAILURE "addr=" MACSTR
3542					" frame=ft-reassoc-req error=%s",
3543					MAC2STR(sm->addr), ocv_errorstr);
3544			return WLAN_STATUS_INVALID_FTIE;
3545		}
3546	}
3547#endif /* CONFIG_OCV */
3548
3549	return WLAN_STATUS_SUCCESS;
3550}
3551
3552
3553int wpa_ft_action_rx(struct wpa_state_machine *sm, const u8 *data, size_t len)
3554{
3555	const u8 *sta_addr, *target_ap;
3556	const u8 *ies;
3557	size_t ies_len;
3558	u8 action;
3559	struct ft_rrb_frame *frame;
3560
3561	if (sm == NULL)
3562		return -1;
3563
3564	/*
3565	 * data: Category[1] Action[1] STA_Address[6] Target_AP_Address[6]
3566	 * FT Request action frame body[variable]
3567	 */
3568
3569	if (len < 14) {
3570		wpa_printf(MSG_DEBUG, "FT: Too short FT Action frame "
3571			   "(len=%lu)", (unsigned long) len);
3572		return -1;
3573	}
3574
3575	action = data[1];
3576	sta_addr = data + 2;
3577	target_ap = data + 8;
3578	ies = data + 14;
3579	ies_len = len - 14;
3580
3581	wpa_printf(MSG_DEBUG, "FT: Received FT Action frame (STA=" MACSTR
3582		   " Target AP=" MACSTR " Action=%d)",
3583		   MAC2STR(sta_addr), MAC2STR(target_ap), action);
3584
3585	if (os_memcmp(sta_addr, sm->addr, ETH_ALEN) != 0) {
3586		wpa_printf(MSG_DEBUG, "FT: Mismatch in FT Action STA address: "
3587			   "STA=" MACSTR " STA-Address=" MACSTR,
3588			   MAC2STR(sm->addr), MAC2STR(sta_addr));
3589		return -1;
3590	}
3591
3592	/*
3593	 * Do some validity checking on the target AP address (not own and not
3594	 * broadcast. This could be extended to filter based on a list of known
3595	 * APs in the MD (if such a list were configured).
3596	 */
3597	if ((target_ap[0] & 0x01) ||
3598	    os_memcmp(target_ap, sm->wpa_auth->addr, ETH_ALEN) == 0) {
3599		wpa_printf(MSG_DEBUG, "FT: Invalid Target AP in FT Action "
3600			   "frame");
3601		return -1;
3602	}
3603
3604	wpa_hexdump(MSG_MSGDUMP, "FT: Action frame body", ies, ies_len);
3605
3606	if (!sm->wpa_auth->conf.ft_over_ds) {
3607		wpa_printf(MSG_DEBUG, "FT: Over-DS option disabled - reject");
3608		return -1;
3609	}
3610
3611	/* RRB - Forward action frame to the target AP */
3612	frame = os_malloc(sizeof(*frame) + len);
3613	if (frame == NULL)
3614		return -1;
3615	frame->frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
3616	frame->packet_type = FT_PACKET_REQUEST;
3617	frame->action_length = host_to_le16(len);
3618	os_memcpy(frame->ap_address, sm->wpa_auth->addr, ETH_ALEN);
3619	os_memcpy(frame + 1, data, len);
3620
3621	wpa_ft_rrb_send(sm->wpa_auth, target_ap, (u8 *) frame,
3622			sizeof(*frame) + len);
3623	os_free(frame);
3624
3625	return 0;
3626}
3627
3628
3629static void wpa_ft_rrb_rx_request_cb(void *ctx, const u8 *dst, const u8 *bssid,
3630				     u16 auth_transaction, u16 resp,
3631				     const u8 *ies, size_t ies_len)
3632{
3633	struct wpa_state_machine *sm = ctx;
3634	wpa_printf(MSG_DEBUG, "FT: Over-the-DS RX request cb for " MACSTR,
3635		   MAC2STR(sm->addr));
3636	wpa_ft_send_rrb_auth_resp(sm, sm->ft_pending_current_ap, sm->addr,
3637				  WLAN_STATUS_SUCCESS, ies, ies_len);
3638}
3639
3640
3641static int wpa_ft_rrb_rx_request(struct wpa_authenticator *wpa_auth,
3642				 const u8 *current_ap, const u8 *sta_addr,
3643				 const u8 *body, size_t len)
3644{
3645	struct wpa_state_machine *sm;
3646	u16 status;
3647	u8 *resp_ies;
3648	size_t resp_ies_len;
3649	int res;
3650
3651	sm = wpa_ft_add_sta(wpa_auth, sta_addr);
3652	if (sm == NULL) {
3653		wpa_printf(MSG_DEBUG, "FT: Failed to add new STA based on "
3654			   "RRB Request");
3655		return -1;
3656	}
3657
3658	wpa_hexdump(MSG_MSGDUMP, "FT: RRB Request Frame body", body, len);
3659
3660	sm->ft_pending_cb = wpa_ft_rrb_rx_request_cb;
3661	sm->ft_pending_cb_ctx = sm;
3662	os_memcpy(sm->ft_pending_current_ap, current_ap, ETH_ALEN);
3663	sm->ft_pending_pull_left_retries = sm->wpa_auth->conf.rkh_pull_retries;
3664	res = wpa_ft_process_auth_req(sm, body, len, &resp_ies,
3665				      &resp_ies_len);
3666	if (res < 0) {
3667		wpa_printf(MSG_DEBUG, "FT: No immediate response available - wait for pull response");
3668		return 0;
3669	}
3670	status = res;
3671
3672	res = wpa_ft_send_rrb_auth_resp(sm, current_ap, sta_addr, status,
3673					resp_ies, resp_ies_len);
3674	os_free(resp_ies);
3675	return res;
3676}
3677
3678
3679static int wpa_ft_send_rrb_auth_resp(struct wpa_state_machine *sm,
3680				     const u8 *current_ap, const u8 *sta_addr,
3681				     u16 status, const u8 *resp_ies,
3682				     size_t resp_ies_len)
3683{
3684	struct wpa_authenticator *wpa_auth = sm->wpa_auth;
3685	size_t rlen;
3686	struct ft_rrb_frame *frame;
3687	u8 *pos;
3688
3689	wpa_printf(MSG_DEBUG, "FT: RRB authentication response: STA=" MACSTR
3690		   " CurrentAP=" MACSTR " status=%u (%s)",
3691		   MAC2STR(sm->addr), MAC2STR(current_ap), status,
3692		   status2str(status));
3693	wpa_hexdump(MSG_DEBUG, "FT: Response IEs", resp_ies, resp_ies_len);
3694
3695	/* RRB - Forward action frame response to the Current AP */
3696
3697	/*
3698	 * data: Category[1] Action[1] STA_Address[6] Target_AP_Address[6]
3699	 * Status_Code[2] FT Request action frame body[variable]
3700	 */
3701	rlen = 2 + 2 * ETH_ALEN + 2 + resp_ies_len;
3702
3703	frame = os_malloc(sizeof(*frame) + rlen);
3704	if (frame == NULL)
3705		return -1;
3706	frame->frame_type = RSN_REMOTE_FRAME_TYPE_FT_RRB;
3707	frame->packet_type = FT_PACKET_RESPONSE;
3708	frame->action_length = host_to_le16(rlen);
3709	os_memcpy(frame->ap_address, wpa_auth->addr, ETH_ALEN);
3710	pos = (u8 *) (frame + 1);
3711	*pos++ = WLAN_ACTION_FT;
3712	*pos++ = 2; /* Action: Response */
3713	os_memcpy(pos, sta_addr, ETH_ALEN);
3714	pos += ETH_ALEN;
3715	os_memcpy(pos, wpa_auth->addr, ETH_ALEN);
3716	pos += ETH_ALEN;
3717	WPA_PUT_LE16(pos, status);
3718	pos += 2;
3719	if (resp_ies)
3720		os_memcpy(pos, resp_ies, resp_ies_len);
3721
3722	wpa_ft_rrb_send(wpa_auth, current_ap, (u8 *) frame,
3723			sizeof(*frame) + rlen);
3724	os_free(frame);
3725
3726	return 0;
3727}
3728
3729
3730static int wpa_ft_rrb_build_r0(const u8 *key, const size_t key_len,
3731			       const struct tlv_list *tlvs,
3732			       const struct wpa_ft_pmk_r0_sa *pmk_r0,
3733			       const u8 *r1kh_id, const u8 *s1kh_id,
3734			       const struct tlv_list *tlv_auth,
3735			       const u8 *src_addr, u8 type,
3736			       u8 **packet, size_t *packet_len)
3737{
3738	u8 pmk_r1[PMK_LEN_MAX];
3739	size_t pmk_r1_len = pmk_r0->pmk_r0_len;
3740	u8 pmk_r1_name[WPA_PMK_NAME_LEN];
3741	u8 f_pairwise[sizeof(le16)];
3742	u8 f_expires_in[sizeof(le16)];
3743	u8 f_session_timeout[sizeof(le32)];
3744	int expires_in;
3745	int session_timeout;
3746	struct os_reltime now;
3747	int ret;
3748	struct tlv_list sess_tlv[] = {
3749		{ .type = FT_RRB_PMK_R1, .len = pmk_r1_len,
3750		  .data = pmk_r1 },
3751		{ .type = FT_RRB_PMK_R1_NAME, .len = sizeof(pmk_r1_name),
3752		  .data = pmk_r1_name },
3753		{ .type = FT_RRB_PAIRWISE, .len = sizeof(f_pairwise),
3754		  .data = f_pairwise },
3755		{ .type = FT_RRB_EXPIRES_IN, .len = sizeof(f_expires_in),
3756		  .data = f_expires_in },
3757		{ .type = FT_RRB_IDENTITY, .len = pmk_r0->identity_len,
3758		  .data = pmk_r0->identity },
3759		{ .type = FT_RRB_RADIUS_CUI, .len = pmk_r0->radius_cui_len,
3760		  .data = pmk_r0->radius_cui },
3761		{ .type = FT_RRB_SESSION_TIMEOUT,
3762		  .len = sizeof(f_session_timeout),
3763		  .data = f_session_timeout },
3764		{ .type = FT_RRB_LAST_EMPTY, .len = 0, .data = NULL },
3765	};
3766
3767	wpa_printf(MSG_DEBUG, "FT: Derive PMK-R1 for peer AP");
3768	if (wpa_derive_pmk_r1(pmk_r0->pmk_r0, pmk_r0->pmk_r0_len,
3769			      pmk_r0->pmk_r0_name, r1kh_id,
3770			      s1kh_id, pmk_r1, pmk_r1_name) < 0)
3771		return -1;
3772	WPA_PUT_LE16(f_pairwise, pmk_r0->pairwise);
3773
3774	os_get_reltime(&now);
3775	if (pmk_r0->expiration > now.sec)
3776		expires_in = pmk_r0->expiration - now.sec;
3777	else if (pmk_r0->expiration)
3778		expires_in = 1;
3779	else
3780		expires_in = 0;
3781	WPA_PUT_LE16(f_expires_in, expires_in);
3782
3783	if (pmk_r0->session_timeout > now.sec)
3784		session_timeout = pmk_r0->session_timeout - now.sec;
3785	else if (pmk_r0->session_timeout)
3786		session_timeout = 1;
3787	else
3788		session_timeout = 0;
3789	WPA_PUT_LE32(f_session_timeout, session_timeout);
3790
3791	ret = wpa_ft_rrb_build(key, key_len, tlvs, sess_tlv, tlv_auth,
3792			       pmk_r0->vlan, src_addr, type,
3793			       packet, packet_len);
3794
3795	forced_memzero(pmk_r1, sizeof(pmk_r1));
3796
3797	return ret;
3798}
3799
3800
3801static int wpa_ft_rrb_rx_pull(struct wpa_authenticator *wpa_auth,
3802			      const u8 *src_addr,
3803			      const u8 *enc, size_t enc_len,
3804			      const u8 *auth, size_t auth_len,
3805			      int no_defer)
3806{
3807	const char *msgtype = "pull request";
3808	u8 *plain = NULL, *packet = NULL;
3809	size_t plain_len = 0, packet_len = 0;
3810	struct ft_remote_r1kh *r1kh, *r1kh_wildcard;
3811	const u8 *key;
3812	size_t key_len;
3813	int seq_ret;
3814	const u8 *f_nonce, *f_r0kh_id, *f_r1kh_id, *f_s1kh_id, *f_pmk_r0_name;
3815	size_t f_nonce_len, f_r0kh_id_len, f_r1kh_id_len, f_s1kh_id_len;
3816	size_t f_pmk_r0_name_len;
3817	const struct wpa_ft_pmk_r0_sa *r0;
3818	int ret;
3819	struct tlv_list resp[2];
3820	struct tlv_list resp_auth[5];
3821	struct ft_rrb_seq f_seq;
3822
3823	wpa_printf(MSG_DEBUG, "FT: Received PMK-R1 pull");
3824
3825	RRB_GET_AUTH(FT_RRB_R0KH_ID, r0kh_id, msgtype, -1);
3826	wpa_hexdump(MSG_DEBUG, "FT: R0KH-ID", f_r0kh_id, f_r0kh_id_len);
3827
3828	if (wpa_ft_rrb_check_r0kh(wpa_auth, f_r0kh_id, f_r0kh_id_len)) {
3829		wpa_printf(MSG_DEBUG, "FT: R0KH-ID mismatch");
3830		goto out;
3831	}
3832
3833	RRB_GET_AUTH(FT_RRB_R1KH_ID, r1kh_id, msgtype, FT_R1KH_ID_LEN);
3834	wpa_printf(MSG_DEBUG, "FT: R1KH-ID=" MACSTR, MAC2STR(f_r1kh_id));
3835
3836	wpa_ft_rrb_lookup_r1kh(wpa_auth, f_r1kh_id, &r1kh, &r1kh_wildcard);
3837	if (r1kh) {
3838		key = r1kh->key;
3839		key_len = sizeof(r1kh->key);
3840	} else if (r1kh_wildcard) {
3841		wpa_printf(MSG_DEBUG, "FT: Using wildcard R1KH-ID");
3842		key = r1kh_wildcard->key;
3843		key_len = sizeof(r1kh_wildcard->key);
3844	} else {
3845		goto out;
3846	}
3847
3848	RRB_GET_AUTH(FT_RRB_NONCE, nonce, "pull request", FT_RRB_NONCE_LEN);
3849	wpa_hexdump(MSG_DEBUG, "FT: nonce", f_nonce, f_nonce_len);
3850
3851	seq_ret = FT_RRB_SEQ_DROP;
3852	if (r1kh)
3853		seq_ret = wpa_ft_rrb_seq_chk(r1kh->seq, src_addr, enc, enc_len,
3854					     auth, auth_len, msgtype, no_defer);
3855	if (!no_defer && r1kh_wildcard &&
3856	    (!r1kh || os_memcmp(r1kh->addr, src_addr, ETH_ALEN) != 0)) {
3857		/* wildcard: r1kh-id unknown or changed addr -> do a seq req */
3858		seq_ret = FT_RRB_SEQ_DEFER;
3859	}
3860
3861	if (seq_ret == FT_RRB_SEQ_DROP)
3862		goto out;
3863
3864	if (wpa_ft_rrb_decrypt(key, key_len, enc, enc_len, auth, auth_len,
3865			       src_addr, FT_PACKET_R0KH_R1KH_PULL,
3866			       &plain, &plain_len) < 0)
3867		goto out;
3868
3869	if (!r1kh)
3870		r1kh = wpa_ft_rrb_add_r1kh(wpa_auth, r1kh_wildcard, src_addr,
3871					   f_r1kh_id,
3872					   wpa_auth->conf.rkh_pos_timeout);
3873	if (!r1kh)
3874		goto out;
3875
3876	if (seq_ret == FT_RRB_SEQ_DEFER) {
3877		wpa_ft_rrb_seq_req(wpa_auth, r1kh->seq, src_addr, f_r0kh_id,
3878				   f_r0kh_id_len, f_r1kh_id, key, key_len,
3879				   enc, enc_len, auth, auth_len,
3880				   &wpa_ft_rrb_rx_pull);
3881		goto out;
3882	}
3883
3884	wpa_ft_rrb_seq_accept(wpa_auth, r1kh->seq, src_addr, auth, auth_len,
3885			      msgtype);
3886	wpa_ft_rrb_r1kh_replenish(wpa_auth, r1kh,
3887				  wpa_auth->conf.rkh_pos_timeout);
3888
3889	RRB_GET(FT_RRB_PMK_R0_NAME, pmk_r0_name, msgtype, WPA_PMK_NAME_LEN);
3890	wpa_hexdump(MSG_DEBUG, "FT: PMKR0Name", f_pmk_r0_name,
3891		    f_pmk_r0_name_len);
3892
3893	RRB_GET(FT_RRB_S1KH_ID, s1kh_id, msgtype, ETH_ALEN);
3894	wpa_printf(MSG_DEBUG, "FT: S1KH-ID=" MACSTR, MAC2STR(f_s1kh_id));
3895
3896	if (wpa_ft_new_seq(r1kh->seq, &f_seq) < 0) {
3897		wpa_printf(MSG_DEBUG, "FT: Failed to get seq num");
3898		goto out;
3899	}
3900
3901	wpa_printf(MSG_DEBUG, "FT: Send PMK-R1 pull response from " MACSTR
3902		   " to " MACSTR,
3903		   MAC2STR(wpa_auth->addr), MAC2STR(src_addr));
3904
3905	resp[0].type = FT_RRB_S1KH_ID;
3906	resp[0].len = f_s1kh_id_len;
3907	resp[0].data = f_s1kh_id;
3908	resp[1].type = FT_RRB_LAST_EMPTY;
3909	resp[1].len = 0;
3910	resp[1].data = NULL;
3911
3912	resp_auth[0].type = FT_RRB_NONCE;
3913	resp_auth[0].len = f_nonce_len;
3914	resp_auth[0].data = f_nonce;
3915	resp_auth[1].type = FT_RRB_SEQ;
3916	resp_auth[1].len = sizeof(f_seq);
3917	resp_auth[1].data = (u8 *) &f_seq;
3918	resp_auth[2].type = FT_RRB_R0KH_ID;
3919	resp_auth[2].len = f_r0kh_id_len;
3920	resp_auth[2].data = f_r0kh_id;
3921	resp_auth[3].type = FT_RRB_R1KH_ID;
3922	resp_auth[3].len = f_r1kh_id_len;
3923	resp_auth[3].data = f_r1kh_id;
3924	resp_auth[4].type = FT_RRB_LAST_EMPTY;
3925	resp_auth[4].len = 0;
3926	resp_auth[4].data = NULL;
3927
3928	if (wpa_ft_fetch_pmk_r0(wpa_auth, f_s1kh_id, f_pmk_r0_name, &r0) < 0) {
3929		wpa_printf(MSG_DEBUG, "FT: No matching PMK-R0-Name found");
3930		ret = wpa_ft_rrb_build(key, key_len, resp, NULL, resp_auth,
3931				       NULL, wpa_auth->addr,
3932				       FT_PACKET_R0KH_R1KH_RESP,
3933				       &packet, &packet_len);
3934	} else {
3935		ret = wpa_ft_rrb_build_r0(key, key_len, resp, r0, f_r1kh_id,
3936					  f_s1kh_id, resp_auth, wpa_auth->addr,
3937					  FT_PACKET_R0KH_R1KH_RESP,
3938					  &packet, &packet_len);
3939	}
3940
3941	if (!ret)
3942		wpa_ft_rrb_oui_send(wpa_auth, src_addr,
3943				    FT_PACKET_R0KH_R1KH_RESP, packet,
3944				    packet_len);
3945
3946out:
3947	os_free(plain);
3948	os_free(packet);
3949
3950	return 0;
3951}
3952
3953
3954/* @returns  0 on success
3955 *          -1 on error
3956 *          -2 if FR_RRB_PAIRWISE is missing
3957 */
3958static int wpa_ft_rrb_rx_r1(struct wpa_authenticator *wpa_auth,
3959			    const u8 *src_addr, u8 type,
3960			    const u8 *enc, size_t enc_len,
3961			    const u8 *auth, size_t auth_len,
3962			    const char *msgtype, u8 *s1kh_id_out,
3963			    int (*cb)(struct wpa_authenticator *wpa_auth,
3964				      const u8 *src_addr,
3965				      const u8 *enc, size_t enc_len,
3966				      const u8 *auth, size_t auth_len,
3967				      int no_defer))
3968{
3969	u8 *plain = NULL;
3970	size_t plain_len = 0;
3971	struct ft_remote_r0kh *r0kh, *r0kh_wildcard;
3972	const u8 *key;
3973	size_t key_len;
3974	int seq_ret;
3975	const u8 *f_r1kh_id, *f_s1kh_id, *f_r0kh_id;
3976	const u8 *f_pmk_r1_name, *f_pairwise, *f_pmk_r1;
3977	const u8 *f_expires_in;
3978	size_t f_r1kh_id_len, f_s1kh_id_len, f_r0kh_id_len;
3979	const u8 *f_identity, *f_radius_cui;
3980	const u8 *f_session_timeout;
3981	size_t f_pmk_r1_name_len, f_pairwise_len, f_pmk_r1_len;
3982	size_t f_expires_in_len;
3983	size_t f_identity_len, f_radius_cui_len;
3984	size_t f_session_timeout_len;
3985	int pairwise;
3986	int ret = -1;
3987	int expires_in;
3988	int session_timeout;
3989	struct vlan_description vlan;
3990	size_t pmk_r1_len;
3991
3992	RRB_GET_AUTH(FT_RRB_R0KH_ID, r0kh_id, msgtype, -1);
3993	wpa_hexdump(MSG_DEBUG, "FT: R0KH-ID", f_r0kh_id, f_r0kh_id_len);
3994
3995	RRB_GET_AUTH(FT_RRB_R1KH_ID, r1kh_id, msgtype, FT_R1KH_ID_LEN);
3996	wpa_printf(MSG_DEBUG, "FT: R1KH-ID=" MACSTR, MAC2STR(f_r1kh_id));
3997
3998	if (wpa_ft_rrb_check_r1kh(wpa_auth, f_r1kh_id)) {
3999		wpa_printf(MSG_DEBUG, "FT: R1KH-ID mismatch");
4000		goto out;
4001	}
4002
4003	wpa_ft_rrb_lookup_r0kh(wpa_auth, f_r0kh_id, f_r0kh_id_len, &r0kh,
4004			       &r0kh_wildcard);
4005	if (r0kh) {
4006		key = r0kh->key;
4007		key_len = sizeof(r0kh->key);
4008	} else if (r0kh_wildcard) {
4009		wpa_printf(MSG_DEBUG, "FT: Using wildcard R0KH-ID");
4010		key = r0kh_wildcard->key;
4011		key_len = sizeof(r0kh_wildcard->key);
4012	} else {
4013		goto out;
4014	}
4015
4016	seq_ret = FT_RRB_SEQ_DROP;
4017	if (r0kh) {
4018		seq_ret = wpa_ft_rrb_seq_chk(r0kh->seq, src_addr, enc, enc_len,
4019					     auth, auth_len, msgtype,
4020					     cb ? 0 : 1);
4021	}
4022	if (cb && r0kh_wildcard &&
4023	    (!r0kh || os_memcmp(r0kh->addr, src_addr, ETH_ALEN) != 0)) {
4024		/* wildcard: r0kh-id unknown or changed addr -> do a seq req */
4025		seq_ret = FT_RRB_SEQ_DEFER;
4026	}
4027
4028	if (seq_ret == FT_RRB_SEQ_DROP)
4029		goto out;
4030
4031	if (wpa_ft_rrb_decrypt(key, key_len, enc, enc_len, auth, auth_len,
4032			       src_addr, type, &plain, &plain_len) < 0)
4033		goto out;
4034
4035	if (!r0kh)
4036		r0kh = wpa_ft_rrb_add_r0kh(wpa_auth, r0kh_wildcard, src_addr,
4037					   f_r0kh_id, f_r0kh_id_len,
4038					   wpa_auth->conf.rkh_pos_timeout);
4039	if (!r0kh)
4040		goto out;
4041
4042	if (seq_ret == FT_RRB_SEQ_DEFER) {
4043		wpa_ft_rrb_seq_req(wpa_auth, r0kh->seq, src_addr, f_r0kh_id,
4044				   f_r0kh_id_len, f_r1kh_id, key, key_len,
4045				   enc, enc_len, auth, auth_len, cb);
4046		goto out;
4047	}
4048
4049	wpa_ft_rrb_seq_accept(wpa_auth, r0kh->seq, src_addr, auth, auth_len,
4050			      msgtype);
4051	wpa_ft_rrb_r0kh_replenish(wpa_auth, r0kh,
4052				  wpa_auth->conf.rkh_pos_timeout);
4053
4054	RRB_GET(FT_RRB_S1KH_ID, s1kh_id, msgtype, ETH_ALEN);
4055	wpa_printf(MSG_DEBUG, "FT: S1KH-ID=" MACSTR, MAC2STR(f_s1kh_id));
4056
4057	if (s1kh_id_out)
4058		os_memcpy(s1kh_id_out, f_s1kh_id, ETH_ALEN);
4059
4060	ret = -2;
4061	RRB_GET(FT_RRB_PAIRWISE, pairwise, msgtype, sizeof(le16));
4062	wpa_hexdump(MSG_DEBUG, "FT: pairwise", f_pairwise, f_pairwise_len);
4063
4064	ret = -1;
4065	RRB_GET(FT_RRB_PMK_R1_NAME, pmk_r1_name, msgtype, WPA_PMK_NAME_LEN);
4066	wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name",
4067		    f_pmk_r1_name, WPA_PMK_NAME_LEN);
4068
4069	pmk_r1_len = PMK_LEN;
4070	if (wpa_ft_rrb_get_tlv(plain, plain_len, FT_RRB_PMK_R1, &f_pmk_r1_len,
4071			       &f_pmk_r1) == 0 &&
4072	    (f_pmk_r1_len == PMK_LEN || f_pmk_r1_len == SHA384_MAC_LEN))
4073		pmk_r1_len = f_pmk_r1_len;
4074	RRB_GET(FT_RRB_PMK_R1, pmk_r1, msgtype, pmk_r1_len);
4075	wpa_hexdump_key(MSG_DEBUG, "FT: PMK-R1", f_pmk_r1, pmk_r1_len);
4076
4077	pairwise = WPA_GET_LE16(f_pairwise);
4078
4079	RRB_GET_OPTIONAL(FT_RRB_EXPIRES_IN, expires_in, msgtype,
4080			 sizeof(le16));
4081	if (f_expires_in)
4082		expires_in = WPA_GET_LE16(f_expires_in);
4083	else
4084		expires_in = 0;
4085
4086	wpa_printf(MSG_DEBUG, "FT: PMK-R1 %s - expires_in=%d", msgtype,
4087		   expires_in);
4088
4089	if (wpa_ft_rrb_get_tlv_vlan(plain, plain_len, &vlan) < 0) {
4090		wpa_printf(MSG_DEBUG, "FT: Cannot parse vlan");
4091		wpa_ft_rrb_dump(plain, plain_len);
4092		goto out;
4093	}
4094
4095	wpa_printf(MSG_DEBUG, "FT: vlan %d%s",
4096		   le_to_host16(vlan.untagged), vlan.tagged[0] ? "+" : "");
4097
4098	RRB_GET_OPTIONAL(FT_RRB_IDENTITY, identity, msgtype, -1);
4099	if (f_identity)
4100		wpa_hexdump_ascii(MSG_DEBUG, "FT: Identity", f_identity,
4101				  f_identity_len);
4102
4103	RRB_GET_OPTIONAL(FT_RRB_RADIUS_CUI, radius_cui, msgtype, -1);
4104	if (f_radius_cui)
4105		wpa_hexdump_ascii(MSG_DEBUG, "FT: CUI", f_radius_cui,
4106				  f_radius_cui_len);
4107
4108	RRB_GET_OPTIONAL(FT_RRB_SESSION_TIMEOUT, session_timeout, msgtype,
4109			 sizeof(le32));
4110	if (f_session_timeout)
4111		session_timeout = WPA_GET_LE32(f_session_timeout);
4112	else
4113		session_timeout = 0;
4114	wpa_printf(MSG_DEBUG, "FT: session_timeout %d", session_timeout);
4115
4116	if (wpa_ft_store_pmk_r1(wpa_auth, f_s1kh_id, f_pmk_r1, pmk_r1_len,
4117				f_pmk_r1_name,
4118				pairwise, &vlan, expires_in, session_timeout,
4119				f_identity, f_identity_len, f_radius_cui,
4120				f_radius_cui_len) < 0)
4121		goto out;
4122
4123	ret = 0;
4124out:
4125	bin_clear_free(plain, plain_len);
4126
4127	return ret;
4128
4129}
4130
4131
4132static void ft_finish_pull(struct wpa_state_machine *sm)
4133{
4134	int res;
4135	u8 *resp_ies;
4136	size_t resp_ies_len;
4137	u16 status;
4138
4139	if (!sm->ft_pending_cb || !sm->ft_pending_req_ies)
4140		return;
4141
4142	res = wpa_ft_process_auth_req(sm, wpabuf_head(sm->ft_pending_req_ies),
4143				      wpabuf_len(sm->ft_pending_req_ies),
4144				      &resp_ies, &resp_ies_len);
4145	if (res < 0) {
4146		/* this loop is broken by ft_pending_pull_left_retries */
4147		wpa_printf(MSG_DEBUG,
4148			   "FT: Callback postponed until response is available");
4149		return;
4150	}
4151	wpabuf_free(sm->ft_pending_req_ies);
4152	sm->ft_pending_req_ies = NULL;
4153	status = res;
4154	wpa_printf(MSG_DEBUG, "FT: Postponed auth callback result for " MACSTR
4155		   " - status %u", MAC2STR(sm->addr), status);
4156
4157	sm->ft_pending_cb(sm->ft_pending_cb_ctx, sm->addr, sm->wpa_auth->addr,
4158			  sm->ft_pending_auth_transaction + 1, status,
4159			  resp_ies, resp_ies_len);
4160	os_free(resp_ies);
4161}
4162
4163
4164struct ft_get_sta_ctx {
4165	const u8 *nonce;
4166	const u8 *s1kh_id;
4167	struct wpa_state_machine *sm;
4168};
4169
4170
4171static int ft_get_sta_cb(struct wpa_state_machine *sm, void *ctx)
4172{
4173	struct ft_get_sta_ctx *info = ctx;
4174
4175	if ((info->s1kh_id &&
4176	     os_memcmp(info->s1kh_id, sm->addr, ETH_ALEN) != 0) ||
4177	    os_memcmp(info->nonce, sm->ft_pending_pull_nonce,
4178		      FT_RRB_NONCE_LEN) != 0 ||
4179	    sm->ft_pending_cb == NULL || sm->ft_pending_req_ies == NULL)
4180		return 0;
4181
4182	info->sm = sm;
4183
4184	return 1;
4185}
4186
4187
4188static int wpa_ft_rrb_rx_resp(struct wpa_authenticator *wpa_auth,
4189			      const u8 *src_addr,
4190			      const u8 *enc, size_t enc_len,
4191			      const u8 *auth, size_t auth_len,
4192			      int no_defer)
4193{
4194	const char *msgtype = "pull response";
4195	int nak, ret = -1;
4196	struct ft_get_sta_ctx ctx;
4197	u8 s1kh_id[ETH_ALEN];
4198	const u8 *f_nonce;
4199	size_t f_nonce_len;
4200
4201	wpa_printf(MSG_DEBUG, "FT: Received PMK-R1 pull response");
4202
4203	RRB_GET_AUTH(FT_RRB_NONCE, nonce, msgtype, FT_RRB_NONCE_LEN);
4204	wpa_hexdump(MSG_DEBUG, "FT: nonce", f_nonce, f_nonce_len);
4205
4206	os_memset(&ctx, 0, sizeof(ctx));
4207	ctx.nonce = f_nonce;
4208	if (!wpa_auth_for_each_sta(wpa_auth, ft_get_sta_cb, &ctx)) {
4209		/* nonce not found */
4210		wpa_printf(MSG_DEBUG, "FT: Invalid nonce");
4211		return -1;
4212	}
4213
4214	ret = wpa_ft_rrb_rx_r1(wpa_auth, src_addr, FT_PACKET_R0KH_R1KH_RESP,
4215			       enc, enc_len, auth, auth_len, msgtype, s1kh_id,
4216			       no_defer ? NULL : &wpa_ft_rrb_rx_resp);
4217	if (ret == -2) {
4218		ret = 0;
4219		nak = 1;
4220	} else {
4221		nak = 0;
4222	}
4223	if (ret < 0)
4224		return -1;
4225
4226	ctx.s1kh_id = s1kh_id;
4227	if (wpa_auth_for_each_sta(wpa_auth, ft_get_sta_cb, &ctx)) {
4228		wpa_printf(MSG_DEBUG,
4229			   "FT: Response to a pending pull request for " MACSTR,
4230			   MAC2STR(ctx.sm->addr));
4231		eloop_cancel_timeout(wpa_ft_expire_pull, ctx.sm, NULL);
4232		if (nak)
4233			ctx.sm->ft_pending_pull_left_retries = 0;
4234		ft_finish_pull(ctx.sm);
4235	}
4236
4237out:
4238	return ret;
4239}
4240
4241
4242static int wpa_ft_rrb_rx_push(struct wpa_authenticator *wpa_auth,
4243			      const u8 *src_addr,
4244			      const u8 *enc, size_t enc_len,
4245			      const u8 *auth, size_t auth_len, int no_defer)
4246{
4247	const char *msgtype = "push";
4248
4249	wpa_printf(MSG_DEBUG, "FT: Received PMK-R1 push");
4250
4251	if (wpa_ft_rrb_rx_r1(wpa_auth, src_addr, FT_PACKET_R0KH_R1KH_PUSH,
4252			     enc, enc_len, auth, auth_len, msgtype, NULL,
4253			     no_defer ? NULL : wpa_ft_rrb_rx_push) < 0)
4254		return -1;
4255
4256	return 0;
4257}
4258
4259
4260static int wpa_ft_rrb_rx_seq(struct wpa_authenticator *wpa_auth,
4261			     const u8 *src_addr, int type,
4262			     const u8 *enc, size_t enc_len,
4263			     const u8 *auth, size_t auth_len,
4264			     struct ft_remote_seq **rkh_seq,
4265			     u8 **key, size_t *key_len,
4266			     struct ft_remote_r0kh **r0kh_out,
4267			     struct ft_remote_r1kh **r1kh_out,
4268			     struct ft_remote_r0kh **r0kh_wildcard_out,
4269			     struct ft_remote_r1kh **r1kh_wildcard_out)
4270{
4271	struct ft_remote_r0kh *r0kh = NULL;
4272	struct ft_remote_r1kh *r1kh = NULL;
4273	const u8 *f_r0kh_id, *f_r1kh_id;
4274	size_t f_r0kh_id_len, f_r1kh_id_len;
4275	int to_r0kh, to_r1kh;
4276	u8 *plain = NULL;
4277	size_t plain_len = 0;
4278	struct ft_remote_r0kh *r0kh_wildcard;
4279	struct ft_remote_r1kh *r1kh_wildcard;
4280
4281	RRB_GET_AUTH(FT_RRB_R0KH_ID, r0kh_id, "seq", -1);
4282	RRB_GET_AUTH(FT_RRB_R1KH_ID, r1kh_id, "seq", FT_R1KH_ID_LEN);
4283
4284	to_r0kh = !wpa_ft_rrb_check_r0kh(wpa_auth, f_r0kh_id, f_r0kh_id_len);
4285	to_r1kh = !wpa_ft_rrb_check_r1kh(wpa_auth, f_r1kh_id);
4286
4287	if (to_r0kh && to_r1kh) {
4288		wpa_printf(MSG_DEBUG, "FT: seq - local R0KH-ID and R1KH-ID");
4289		goto out;
4290	}
4291
4292	if (!to_r0kh && !to_r1kh) {
4293		wpa_printf(MSG_DEBUG, "FT: seq - remote R0KH-ID and R1KH-ID");
4294		goto out;
4295	}
4296
4297	if (!to_r0kh) {
4298		wpa_ft_rrb_lookup_r0kh(wpa_auth, f_r0kh_id, f_r0kh_id_len,
4299				       &r0kh, &r0kh_wildcard);
4300		if (!r0kh_wildcard &&
4301		    (!r0kh || os_memcmp(r0kh->addr, src_addr, ETH_ALEN) != 0)) {
4302			wpa_hexdump(MSG_DEBUG, "FT: Did not find R0KH-ID",
4303				    f_r0kh_id, f_r0kh_id_len);
4304			goto out;
4305		}
4306		if (r0kh) {
4307			*key = r0kh->key;
4308			*key_len = sizeof(r0kh->key);
4309		} else {
4310			*key = r0kh_wildcard->key;
4311			*key_len = sizeof(r0kh_wildcard->key);
4312		}
4313	}
4314
4315	if (!to_r1kh) {
4316		wpa_ft_rrb_lookup_r1kh(wpa_auth, f_r1kh_id, &r1kh,
4317				       &r1kh_wildcard);
4318		if (!r1kh_wildcard &&
4319		    (!r1kh || os_memcmp(r1kh->addr, src_addr, ETH_ALEN) != 0)) {
4320			wpa_hexdump(MSG_DEBUG, "FT: Did not find R1KH-ID",
4321				    f_r1kh_id, FT_R1KH_ID_LEN);
4322			goto out;
4323		}
4324		if (r1kh) {
4325			*key = r1kh->key;
4326			*key_len = sizeof(r1kh->key);
4327		} else {
4328			*key = r1kh_wildcard->key;
4329			*key_len = sizeof(r1kh_wildcard->key);
4330		}
4331	}
4332
4333	if (wpa_ft_rrb_decrypt(*key, *key_len, enc, enc_len, auth, auth_len,
4334			       src_addr, type, &plain, &plain_len) < 0)
4335		goto out;
4336
4337	os_free(plain);
4338
4339	if (!to_r0kh) {
4340		if (!r0kh)
4341			r0kh = wpa_ft_rrb_add_r0kh(wpa_auth, r0kh_wildcard,
4342						   src_addr, f_r0kh_id,
4343						   f_r0kh_id_len,
4344						   ftRRBseqTimeout);
4345		if (!r0kh)
4346			goto out;
4347
4348		wpa_ft_rrb_r0kh_replenish(wpa_auth, r0kh, ftRRBseqTimeout);
4349		*rkh_seq = r0kh->seq;
4350		if (r0kh_out)
4351			*r0kh_out = r0kh;
4352		if (r0kh_wildcard_out)
4353			*r0kh_wildcard_out = r0kh_wildcard;
4354	}
4355
4356	if (!to_r1kh) {
4357		if (!r1kh)
4358			r1kh = wpa_ft_rrb_add_r1kh(wpa_auth, r1kh_wildcard,
4359						   src_addr, f_r1kh_id,
4360						   ftRRBseqTimeout);
4361		if (!r1kh)
4362			goto out;
4363
4364		wpa_ft_rrb_r1kh_replenish(wpa_auth, r1kh, ftRRBseqTimeout);
4365		*rkh_seq = r1kh->seq;
4366		if (r1kh_out)
4367			*r1kh_out = r1kh;
4368		if (r1kh_wildcard_out)
4369			*r1kh_wildcard_out = r1kh_wildcard;
4370	}
4371
4372	return 0;
4373out:
4374	return -1;
4375}
4376
4377
4378static int wpa_ft_rrb_rx_seq_req(struct wpa_authenticator *wpa_auth,
4379				 const u8 *src_addr,
4380				 const u8 *enc, size_t enc_len,
4381				 const u8 *auth, size_t auth_len,
4382				 int no_defer)
4383{
4384	int ret = -1;
4385	struct ft_rrb_seq f_seq;
4386	const u8 *f_nonce, *f_r0kh_id, *f_r1kh_id;
4387	size_t f_nonce_len, f_r0kh_id_len, f_r1kh_id_len;
4388	struct ft_remote_seq *rkh_seq = NULL;
4389	u8 *packet = NULL, *key = NULL;
4390	size_t packet_len = 0, key_len = 0;
4391	struct tlv_list seq_resp_auth[5];
4392
4393	wpa_printf(MSG_DEBUG, "FT: Received sequence number request");
4394
4395	if (wpa_ft_rrb_rx_seq(wpa_auth, src_addr, FT_PACKET_R0KH_R1KH_SEQ_REQ,
4396			      enc, enc_len, auth, auth_len, &rkh_seq, &key,
4397			      &key_len, NULL, NULL, NULL, NULL) < 0)
4398		goto out;
4399
4400	RRB_GET_AUTH(FT_RRB_NONCE, nonce, "seq request", FT_RRB_NONCE_LEN);
4401	wpa_hexdump(MSG_DEBUG, "FT: seq request - nonce", f_nonce, f_nonce_len);
4402
4403	RRB_GET_AUTH(FT_RRB_R0KH_ID, r0kh_id, "seq", -1);
4404	RRB_GET_AUTH(FT_RRB_R1KH_ID, r1kh_id, "seq", FT_R1KH_ID_LEN);
4405
4406	if (wpa_ft_new_seq(rkh_seq, &f_seq) < 0) {
4407		wpa_printf(MSG_DEBUG, "FT: Failed to get seq num");
4408		goto out;
4409	}
4410
4411	wpa_printf(MSG_DEBUG, "FT: Send sequence number response from " MACSTR
4412		   " to " MACSTR,
4413		   MAC2STR(wpa_auth->addr), MAC2STR(src_addr));
4414
4415	seq_resp_auth[0].type = FT_RRB_NONCE;
4416	seq_resp_auth[0].len = f_nonce_len;
4417	seq_resp_auth[0].data = f_nonce;
4418	seq_resp_auth[1].type = FT_RRB_SEQ;
4419	seq_resp_auth[1].len = sizeof(f_seq);
4420	seq_resp_auth[1].data = (u8 *) &f_seq;
4421	seq_resp_auth[2].type = FT_RRB_R0KH_ID;
4422	seq_resp_auth[2].len = f_r0kh_id_len;
4423	seq_resp_auth[2].data = f_r0kh_id;
4424	seq_resp_auth[3].type = FT_RRB_R1KH_ID;
4425	seq_resp_auth[3].len = FT_R1KH_ID_LEN;
4426	seq_resp_auth[3].data = f_r1kh_id;
4427	seq_resp_auth[4].type = FT_RRB_LAST_EMPTY;
4428	seq_resp_auth[4].len = 0;
4429	seq_resp_auth[4].data = NULL;
4430
4431	if (wpa_ft_rrb_build(key, key_len, NULL, NULL, seq_resp_auth, NULL,
4432			     wpa_auth->addr, FT_PACKET_R0KH_R1KH_SEQ_RESP,
4433			     &packet, &packet_len) < 0)
4434		goto out;
4435
4436	wpa_ft_rrb_oui_send(wpa_auth, src_addr,
4437			    FT_PACKET_R0KH_R1KH_SEQ_RESP, packet,
4438			    packet_len);
4439
4440out:
4441	os_free(packet);
4442
4443	return ret;
4444}
4445
4446
4447static int wpa_ft_rrb_rx_seq_resp(struct wpa_authenticator *wpa_auth,
4448				  const u8 *src_addr,
4449				  const u8 *enc, size_t enc_len,
4450				  const u8 *auth, size_t auth_len,
4451				  int no_defer)
4452{
4453	u8 *key = NULL;
4454	size_t key_len = 0;
4455	struct ft_remote_r0kh *r0kh = NULL, *r0kh_wildcard = NULL;
4456	struct ft_remote_r1kh *r1kh = NULL, *r1kh_wildcard = NULL;
4457	const u8 *f_nonce, *f_seq;
4458	size_t f_nonce_len, f_seq_len;
4459	struct ft_remote_seq *rkh_seq = NULL;
4460	struct ft_remote_item *item;
4461	struct os_reltime now, now_remote;
4462	int seq_ret, found;
4463	const struct ft_rrb_seq *msg_both;
4464	u32 msg_dom, msg_seq;
4465
4466	wpa_printf(MSG_DEBUG, "FT: Received sequence number response");
4467
4468	if (wpa_ft_rrb_rx_seq(wpa_auth, src_addr, FT_PACKET_R0KH_R1KH_SEQ_RESP,
4469			      enc, enc_len, auth, auth_len, &rkh_seq, &key,
4470			      &key_len, &r0kh, &r1kh, &r0kh_wildcard,
4471			      &r1kh_wildcard) < 0)
4472		goto out;
4473
4474	RRB_GET_AUTH(FT_RRB_NONCE, nonce, "seq response", FT_RRB_NONCE_LEN);
4475	wpa_hexdump(MSG_DEBUG, "FT: seq response - nonce", f_nonce,
4476		    f_nonce_len);
4477
4478	found = 0;
4479	dl_list_for_each(item, &rkh_seq->rx.queue, struct ft_remote_item,
4480			 list) {
4481		if (os_memcmp_const(f_nonce, item->nonce,
4482				    FT_RRB_NONCE_LEN) != 0 ||
4483		    os_get_reltime(&now) < 0 ||
4484		    os_reltime_expired(&now, &item->nonce_ts, ftRRBseqTimeout))
4485			continue;
4486
4487		found = 1;
4488		break;
4489	}
4490	if (!found) {
4491		wpa_printf(MSG_DEBUG, "FT: seq response - bad nonce");
4492		goto out;
4493	}
4494
4495	if (r0kh) {
4496		wpa_ft_rrb_r0kh_replenish(wpa_auth, r0kh,
4497					  wpa_auth->conf.rkh_pos_timeout);
4498		if (r0kh_wildcard)
4499			os_memcpy(r0kh->addr, src_addr, ETH_ALEN);
4500	}
4501
4502	if (r1kh) {
4503		wpa_ft_rrb_r1kh_replenish(wpa_auth, r1kh,
4504					  wpa_auth->conf.rkh_pos_timeout);
4505		if (r1kh_wildcard)
4506			os_memcpy(r1kh->addr, src_addr, ETH_ALEN);
4507	}
4508
4509	seq_ret = wpa_ft_rrb_seq_chk(rkh_seq, src_addr, enc, enc_len, auth,
4510				     auth_len, "seq response", 1);
4511	if (seq_ret == FT_RRB_SEQ_OK) {
4512		wpa_printf(MSG_DEBUG, "FT: seq response - valid seq number");
4513		wpa_ft_rrb_seq_accept(wpa_auth, rkh_seq, src_addr, auth,
4514				      auth_len, "seq response");
4515	} else {
4516		wpa_printf(MSG_DEBUG, "FT: seq response - reset seq number");
4517
4518		RRB_GET_AUTH(FT_RRB_SEQ, seq, "seq response",
4519			     sizeof(*msg_both));
4520		msg_both = (const struct ft_rrb_seq *) f_seq;
4521
4522		msg_dom = le_to_host32(msg_both->dom);
4523		msg_seq = le_to_host32(msg_both->seq);
4524		now_remote.sec = le_to_host32(msg_both->ts);
4525		now_remote.usec = 0;
4526
4527		rkh_seq->rx.num_last = 2;
4528		rkh_seq->rx.dom = msg_dom;
4529		rkh_seq->rx.offsetidx = 0;
4530		/* Accept some older, possibly cached packets as well */
4531		rkh_seq->rx.last[0] = msg_seq - FT_REMOTE_SEQ_BACKLOG -
4532			dl_list_len(&rkh_seq->rx.queue);
4533		rkh_seq->rx.last[1] = msg_seq;
4534
4535		/* local time - offset = remote time
4536		 * <=> local time - remote time = offset */
4537		os_reltime_sub(&now, &now_remote, &rkh_seq->rx.time_offset);
4538	}
4539
4540	wpa_ft_rrb_seq_flush(wpa_auth, rkh_seq, 1);
4541
4542	return 0;
4543out:
4544	return -1;
4545}
4546
4547
4548int wpa_ft_rrb_rx(struct wpa_authenticator *wpa_auth, const u8 *src_addr,
4549		  const u8 *data, size_t data_len)
4550{
4551	struct ft_rrb_frame *frame;
4552	u16 alen;
4553	const u8 *pos, *end, *start;
4554	u8 action;
4555	const u8 *sta_addr, *target_ap_addr;
4556
4557	wpa_printf(MSG_DEBUG, "FT: RRB received frame from remote AP " MACSTR,
4558		   MAC2STR(src_addr));
4559
4560	if (data_len < sizeof(*frame)) {
4561		wpa_printf(MSG_DEBUG, "FT: Too short RRB frame (data_len=%lu)",
4562			   (unsigned long) data_len);
4563		return -1;
4564	}
4565
4566	pos = data;
4567	frame = (struct ft_rrb_frame *) pos;
4568	pos += sizeof(*frame);
4569
4570	alen = le_to_host16(frame->action_length);
4571	wpa_printf(MSG_DEBUG, "FT: RRB frame - frame_type=%d packet_type=%d "
4572		   "action_length=%d ap_address=" MACSTR,
4573		   frame->frame_type, frame->packet_type, alen,
4574		   MAC2STR(frame->ap_address));
4575
4576	if (frame->frame_type != RSN_REMOTE_FRAME_TYPE_FT_RRB) {
4577		/* Discard frame per IEEE Std 802.11r-2008, 11A.10.3 */
4578		wpa_printf(MSG_DEBUG, "FT: RRB discarded frame with "
4579			   "unrecognized type %d", frame->frame_type);
4580		return -1;
4581	}
4582
4583	if (alen > data_len - sizeof(*frame)) {
4584		wpa_printf(MSG_DEBUG, "FT: RRB frame too short for action "
4585			   "frame");
4586		return -1;
4587	}
4588
4589	wpa_hexdump(MSG_MSGDUMP, "FT: RRB - FT Action frame", pos, alen);
4590
4591	if (alen < 1 + 1 + 2 * ETH_ALEN) {
4592		wpa_printf(MSG_DEBUG, "FT: Too short RRB frame (not enough "
4593			   "room for Action Frame body); alen=%lu",
4594			   (unsigned long) alen);
4595		return -1;
4596	}
4597	start = pos;
4598	end = pos + alen;
4599
4600	if (*pos != WLAN_ACTION_FT) {
4601		wpa_printf(MSG_DEBUG, "FT: Unexpected Action frame category "
4602			   "%d", *pos);
4603		return -1;
4604	}
4605
4606	pos++;
4607	action = *pos++;
4608	sta_addr = pos;
4609	pos += ETH_ALEN;
4610	target_ap_addr = pos;
4611	pos += ETH_ALEN;
4612	wpa_printf(MSG_DEBUG, "FT: RRB Action Frame: action=%d sta_addr="
4613		   MACSTR " target_ap_addr=" MACSTR,
4614		   action, MAC2STR(sta_addr), MAC2STR(target_ap_addr));
4615
4616	if (frame->packet_type == FT_PACKET_REQUEST) {
4617		wpa_printf(MSG_DEBUG, "FT: FT Packet Type - Request");
4618
4619		if (action != 1) {
4620			wpa_printf(MSG_DEBUG, "FT: Unexpected Action %d in "
4621				   "RRB Request", action);
4622			return -1;
4623		}
4624
4625		if (os_memcmp(target_ap_addr, wpa_auth->addr, ETH_ALEN) != 0) {
4626			wpa_printf(MSG_DEBUG, "FT: Target AP address in the "
4627				   "RRB Request does not match with own "
4628				   "address");
4629			return -1;
4630		}
4631
4632		if (wpa_ft_rrb_rx_request(wpa_auth, frame->ap_address,
4633					  sta_addr, pos, end - pos) < 0)
4634			return -1;
4635	} else if (frame->packet_type == FT_PACKET_RESPONSE) {
4636		u16 status_code;
4637
4638		if (end - pos < 2) {
4639			wpa_printf(MSG_DEBUG, "FT: Not enough room for status "
4640				   "code in RRB Response");
4641			return -1;
4642		}
4643		status_code = WPA_GET_LE16(pos);
4644
4645		wpa_printf(MSG_DEBUG, "FT: FT Packet Type - Response "
4646			   "(status_code=%d)", status_code);
4647
4648		if (wpa_ft_action_send(wpa_auth, sta_addr, start, alen) < 0)
4649			return -1;
4650	} else {
4651		wpa_printf(MSG_DEBUG, "FT: RRB discarded frame with unknown "
4652			   "packet_type %d", frame->packet_type);
4653		return -1;
4654	}
4655
4656	return 0;
4657}
4658
4659
4660void wpa_ft_rrb_oui_rx(struct wpa_authenticator *wpa_auth, const u8 *src_addr,
4661		       const u8 *dst_addr, u8 oui_suffix, const u8 *data,
4662		       size_t data_len)
4663{
4664	const u8 *auth, *enc;
4665	size_t alen, elen;
4666	int no_defer = 0;
4667
4668	wpa_printf(MSG_DEBUG, "FT: RRB-OUI(" MACSTR
4669		   ") received frame from remote AP "
4670		   MACSTR " oui_suffix=%u dst=" MACSTR,
4671		   MAC2STR(wpa_auth->addr), MAC2STR(src_addr), oui_suffix,
4672		   MAC2STR(dst_addr));
4673	wpa_hexdump(MSG_MSGDUMP, "FT: RRB frame payload", data, data_len);
4674
4675	if (is_multicast_ether_addr(src_addr)) {
4676		wpa_printf(MSG_DEBUG,
4677			   "FT: RRB-OUI received frame from multicast address "
4678			   MACSTR, MAC2STR(src_addr));
4679		return;
4680	}
4681
4682	if (is_multicast_ether_addr(dst_addr))
4683		no_defer = 1;
4684
4685	if (data_len < sizeof(u16)) {
4686		wpa_printf(MSG_DEBUG, "FT: RRB-OUI frame too short");
4687		return;
4688	}
4689
4690	alen = WPA_GET_LE16(data);
4691	if (data_len < sizeof(u16) + alen) {
4692		wpa_printf(MSG_DEBUG, "FT: RRB-OUI frame too short");
4693		return;
4694	}
4695
4696	auth = data + sizeof(u16);
4697	wpa_hexdump(MSG_MSGDUMP, "FT: Authenticated payload", auth, alen);
4698	enc = data + sizeof(u16) + alen;
4699	elen = data_len - sizeof(u16) - alen;
4700	wpa_hexdump(MSG_MSGDUMP, "FT: Encrypted payload", enc, elen);
4701
4702	switch (oui_suffix) {
4703	case FT_PACKET_R0KH_R1KH_PULL:
4704		wpa_ft_rrb_rx_pull(wpa_auth, src_addr, enc, elen, auth, alen,
4705				   no_defer);
4706		break;
4707	case FT_PACKET_R0KH_R1KH_RESP:
4708		wpa_ft_rrb_rx_resp(wpa_auth, src_addr, enc, elen, auth, alen,
4709				   no_defer);
4710		break;
4711	case FT_PACKET_R0KH_R1KH_PUSH:
4712		wpa_ft_rrb_rx_push(wpa_auth, src_addr, enc, elen, auth, alen,
4713				   no_defer);
4714		break;
4715	case FT_PACKET_R0KH_R1KH_SEQ_REQ:
4716		wpa_ft_rrb_rx_seq_req(wpa_auth, src_addr, enc, elen, auth, alen,
4717				      no_defer);
4718		break;
4719	case FT_PACKET_R0KH_R1KH_SEQ_RESP:
4720		wpa_ft_rrb_rx_seq_resp(wpa_auth, src_addr, enc, elen, auth,
4721				       alen, no_defer);
4722		break;
4723	}
4724}
4725
4726
4727static int wpa_ft_generate_pmk_r1(struct wpa_authenticator *wpa_auth,
4728				  struct wpa_ft_pmk_r0_sa *pmk_r0,
4729				  struct ft_remote_r1kh *r1kh,
4730				  const u8 *s1kh_id)
4731{
4732	u8 *packet;
4733	size_t packet_len;
4734	struct ft_rrb_seq f_seq;
4735	struct tlv_list push[] = {
4736		{ .type = FT_RRB_S1KH_ID, .len = ETH_ALEN,
4737		  .data = s1kh_id },
4738		{ .type = FT_RRB_PMK_R0_NAME, .len = WPA_PMK_NAME_LEN,
4739		  .data = pmk_r0->pmk_r0_name },
4740		{ .type = FT_RRB_LAST_EMPTY, .len = 0, .data = NULL },
4741	};
4742	struct tlv_list push_auth[] = {
4743		{ .type = FT_RRB_SEQ, .len = sizeof(f_seq),
4744		  .data = (u8 *) &f_seq },
4745		{ .type = FT_RRB_R0KH_ID,
4746		  .len = wpa_auth->conf.r0_key_holder_len,
4747		  .data = wpa_auth->conf.r0_key_holder },
4748		{ .type = FT_RRB_R1KH_ID, .len = FT_R1KH_ID_LEN,
4749		  .data = r1kh->id },
4750		{ .type = FT_RRB_LAST_EMPTY, .len = 0, .data = NULL },
4751	};
4752
4753	if (wpa_ft_new_seq(r1kh->seq, &f_seq) < 0) {
4754		wpa_printf(MSG_DEBUG, "FT: Failed to get seq num");
4755		return -1;
4756	}
4757
4758	wpa_printf(MSG_DEBUG, "FT: Send PMK-R1 push from " MACSTR
4759		   " to remote R0KH address " MACSTR,
4760		   MAC2STR(wpa_auth->addr), MAC2STR(r1kh->addr));
4761
4762	if (wpa_ft_rrb_build_r0(r1kh->key, sizeof(r1kh->key), push, pmk_r0,
4763				r1kh->id, s1kh_id, push_auth, wpa_auth->addr,
4764				FT_PACKET_R0KH_R1KH_PUSH,
4765				&packet, &packet_len) < 0)
4766		return -1;
4767
4768	wpa_ft_rrb_oui_send(wpa_auth, r1kh->addr, FT_PACKET_R0KH_R1KH_PUSH,
4769			    packet, packet_len);
4770
4771	os_free(packet);
4772	return 0;
4773}
4774
4775
4776void wpa_ft_push_pmk_r1(struct wpa_authenticator *wpa_auth, const u8 *addr)
4777{
4778	struct wpa_ft_pmk_cache *cache = wpa_auth->ft_pmk_cache;
4779	struct wpa_ft_pmk_r0_sa *r0, *r0found = NULL;
4780	struct ft_remote_r1kh *r1kh;
4781
4782	if (!wpa_auth->conf.pmk_r1_push)
4783		return;
4784	if (!wpa_auth->conf.r1kh_list)
4785		return;
4786
4787	dl_list_for_each(r0, &cache->pmk_r0, struct wpa_ft_pmk_r0_sa, list) {
4788		if (os_memcmp(r0->spa, addr, ETH_ALEN) == 0) {
4789			r0found = r0;
4790			break;
4791		}
4792	}
4793
4794	r0 = r0found;
4795	if (r0 == NULL || r0->pmk_r1_pushed)
4796		return;
4797	r0->pmk_r1_pushed = 1;
4798
4799	wpa_printf(MSG_DEBUG, "FT: Deriving and pushing PMK-R1 keys to R1KHs "
4800		   "for STA " MACSTR, MAC2STR(addr));
4801
4802	for (r1kh = *wpa_auth->conf.r1kh_list; r1kh; r1kh = r1kh->next) {
4803		if (is_zero_ether_addr(r1kh->addr) ||
4804		    is_zero_ether_addr(r1kh->id))
4805			continue;
4806		if (wpa_ft_rrb_init_r1kh_seq(r1kh) < 0)
4807			continue;
4808		wpa_ft_generate_pmk_r1(wpa_auth, r0, r1kh, addr);
4809	}
4810}
4811
4812#endif /* CONFIG_IEEE80211R_AP */
4813