Home | History | Annotate | Line # | Download | only in wpa_supplicant
mesh_rsn.c revision 1.1.1.1.2.2.2.1 
      1      1.1.1.1.2.2  snj /*
      2      1.1.1.1.2.2  snj  * WPA Supplicant - Mesh RSN routines
      3      1.1.1.1.2.2  snj  * Copyright (c) 2013-2014, cozybit, Inc.  All rights reserved.
      4      1.1.1.1.2.2  snj  *
      5      1.1.1.1.2.2  snj  * This software may be distributed under the terms of the BSD license.
      6      1.1.1.1.2.2  snj  * See README for more details.
      7      1.1.1.1.2.2  snj  */
      8      1.1.1.1.2.2  snj 
      9      1.1.1.1.2.2  snj #include "utils/includes.h"
     10      1.1.1.1.2.2  snj 
     11      1.1.1.1.2.2  snj #include "utils/common.h"
     12      1.1.1.1.2.2  snj #include "utils/eloop.h"
     13      1.1.1.1.2.2  snj #include "crypto/sha256.h"
     14      1.1.1.1.2.2  snj #include "crypto/random.h"
     15      1.1.1.1.2.2  snj #include "crypto/aes.h"
     16      1.1.1.1.2.2  snj #include "crypto/aes_siv.h"
     17      1.1.1.1.2.2  snj #include "rsn_supp/wpa.h"
     18      1.1.1.1.2.2  snj #include "ap/hostapd.h"
     19      1.1.1.1.2.2  snj #include "ap/wpa_auth.h"
     20      1.1.1.1.2.2  snj #include "ap/sta_info.h"
     21      1.1.1.1.2.2  snj #include "ap/ieee802_11.h"
     22      1.1.1.1.2.2  snj #include "wpa_supplicant_i.h"
     23      1.1.1.1.2.2  snj #include "driver_i.h"
     24      1.1.1.1.2.2  snj #include "wpas_glue.h"
     25      1.1.1.1.2.2  snj #include "mesh_mpm.h"
     26      1.1.1.1.2.2  snj #include "mesh_rsn.h"
     27      1.1.1.1.2.2  snj 
     28      1.1.1.1.2.2  snj #define MESH_AUTH_TIMEOUT 10
     29      1.1.1.1.2.2  snj #define MESH_AUTH_RETRY 3
     30      1.1.1.1.2.2  snj 
     31      1.1.1.1.2.2  snj void mesh_auth_timer(void *eloop_ctx, void *user_data)
     32      1.1.1.1.2.2  snj {
     33      1.1.1.1.2.2  snj 	struct wpa_supplicant *wpa_s = eloop_ctx;
     34      1.1.1.1.2.2  snj 	struct sta_info *sta = user_data;
     35  1.1.1.1.2.2.2.1  snj 	struct hostapd_data *hapd;
     36      1.1.1.1.2.2  snj 
     37      1.1.1.1.2.2  snj 	if (sta->sae->state != SAE_ACCEPTED) {
     38      1.1.1.1.2.2  snj 		wpa_printf(MSG_DEBUG, "AUTH: Re-authenticate with " MACSTR
     39      1.1.1.1.2.2  snj 			   " (attempt %d) ",
     40      1.1.1.1.2.2  snj 			   MAC2STR(sta->addr), sta->sae_auth_retry);
     41      1.1.1.1.2.2  snj 		wpa_msg(wpa_s, MSG_INFO, MESH_SAE_AUTH_FAILURE "addr=" MACSTR,
     42      1.1.1.1.2.2  snj 			MAC2STR(sta->addr));
     43      1.1.1.1.2.2  snj 		if (sta->sae_auth_retry < MESH_AUTH_RETRY) {
     44      1.1.1.1.2.2  snj 			mesh_rsn_auth_sae_sta(wpa_s, sta);
     45      1.1.1.1.2.2  snj 		} else {
     46  1.1.1.1.2.2.2.1  snj 			hapd = wpa_s->ifmsh->bss[0];
     47  1.1.1.1.2.2.2.1  snj 
     48      1.1.1.1.2.2  snj 			if (sta->sae_auth_retry > MESH_AUTH_RETRY) {
     49  1.1.1.1.2.2.2.1  snj 				ap_free_sta(hapd, sta);
     50      1.1.1.1.2.2  snj 				return;
     51      1.1.1.1.2.2  snj 			}
     52      1.1.1.1.2.2  snj 
     53      1.1.1.1.2.2  snj 			/* block the STA if exceeded the number of attempts */
     54      1.1.1.1.2.2  snj 			wpa_mesh_set_plink_state(wpa_s, sta, PLINK_BLOCKED);
     55      1.1.1.1.2.2  snj 			sta->sae->state = SAE_NOTHING;
     56      1.1.1.1.2.2  snj 			wpa_msg(wpa_s, MSG_INFO, MESH_SAE_AUTH_BLOCKED "addr="
     57      1.1.1.1.2.2  snj 				MACSTR " duration=%d",
     58      1.1.1.1.2.2  snj 				MAC2STR(sta->addr),
     59  1.1.1.1.2.2.2.1  snj 				hapd->conf->ap_max_inactivity);
     60      1.1.1.1.2.2  snj 		}
     61      1.1.1.1.2.2  snj 		sta->sae_auth_retry++;
     62      1.1.1.1.2.2  snj 	}
     63      1.1.1.1.2.2  snj }
     64      1.1.1.1.2.2  snj 
     65      1.1.1.1.2.2  snj 
     66      1.1.1.1.2.2  snj static void auth_logger(void *ctx, const u8 *addr, logger_level level,
     67      1.1.1.1.2.2  snj 			const char *txt)
     68      1.1.1.1.2.2  snj {
     69      1.1.1.1.2.2  snj 	if (addr)
     70      1.1.1.1.2.2  snj 		wpa_printf(MSG_DEBUG, "AUTH: " MACSTR " - %s",
     71      1.1.1.1.2.2  snj 			   MAC2STR(addr), txt);
     72      1.1.1.1.2.2  snj 	else
     73      1.1.1.1.2.2  snj 		wpa_printf(MSG_DEBUG, "AUTH: %s", txt);
     74      1.1.1.1.2.2  snj }
     75      1.1.1.1.2.2  snj 
     76      1.1.1.1.2.2  snj 
     77      1.1.1.1.2.2  snj static const u8 *auth_get_psk(void *ctx, const u8 *addr,
     78      1.1.1.1.2.2  snj 			      const u8 *p2p_dev_addr, const u8 *prev_psk)
     79      1.1.1.1.2.2  snj {
     80      1.1.1.1.2.2  snj 	struct mesh_rsn *mesh_rsn = ctx;
     81      1.1.1.1.2.2  snj 	struct hostapd_data *hapd = mesh_rsn->wpa_s->ifmsh->bss[0];
     82      1.1.1.1.2.2  snj 	struct sta_info *sta = ap_get_sta(hapd, addr);
     83      1.1.1.1.2.2  snj 
     84      1.1.1.1.2.2  snj 	wpa_printf(MSG_DEBUG, "AUTH: %s (addr=" MACSTR " prev_psk=%p)",
     85      1.1.1.1.2.2  snj 		   __func__, MAC2STR(addr), prev_psk);
     86      1.1.1.1.2.2  snj 
     87      1.1.1.1.2.2  snj 	if (sta && sta->auth_alg == WLAN_AUTH_SAE) {
     88      1.1.1.1.2.2  snj 		if (!sta->sae || prev_psk)
     89      1.1.1.1.2.2  snj 			return NULL;
     90      1.1.1.1.2.2  snj 		return sta->sae->pmk;
     91      1.1.1.1.2.2  snj 	}
     92      1.1.1.1.2.2  snj 
     93      1.1.1.1.2.2  snj 	return NULL;
     94      1.1.1.1.2.2  snj }
     95      1.1.1.1.2.2  snj 
     96      1.1.1.1.2.2  snj 
     97      1.1.1.1.2.2  snj static int auth_set_key(void *ctx, int vlan_id, enum wpa_alg alg,
     98      1.1.1.1.2.2  snj 			const u8 *addr, int idx, u8 *key, size_t key_len)
     99      1.1.1.1.2.2  snj {
    100      1.1.1.1.2.2  snj 	struct mesh_rsn *mesh_rsn = ctx;
    101      1.1.1.1.2.2  snj 	u8 seq[6];
    102      1.1.1.1.2.2  snj 
    103      1.1.1.1.2.2  snj 	os_memset(seq, 0, sizeof(seq));
    104      1.1.1.1.2.2  snj 
    105      1.1.1.1.2.2  snj 	if (addr) {
    106      1.1.1.1.2.2  snj 		wpa_printf(MSG_DEBUG, "AUTH: %s(alg=%d addr=" MACSTR
    107      1.1.1.1.2.2  snj 			   " key_idx=%d)",
    108      1.1.1.1.2.2  snj 			   __func__, alg, MAC2STR(addr), idx);
    109      1.1.1.1.2.2  snj 	} else {
    110      1.1.1.1.2.2  snj 		wpa_printf(MSG_DEBUG, "AUTH: %s(alg=%d key_idx=%d)",
    111      1.1.1.1.2.2  snj 			   __func__, alg, idx);
    112      1.1.1.1.2.2  snj 	}
    113      1.1.1.1.2.2  snj 	wpa_hexdump_key(MSG_DEBUG, "AUTH: set_key - key", key, key_len);
    114      1.1.1.1.2.2  snj 
    115      1.1.1.1.2.2  snj 	return wpa_drv_set_key(mesh_rsn->wpa_s, alg, addr, idx,
    116      1.1.1.1.2.2  snj 			       1, seq, 6, key, key_len);
    117      1.1.1.1.2.2  snj }
    118      1.1.1.1.2.2  snj 
    119      1.1.1.1.2.2  snj 
    120      1.1.1.1.2.2  snj static int auth_start_ampe(void *ctx, const u8 *addr)
    121      1.1.1.1.2.2  snj {
    122      1.1.1.1.2.2  snj 	struct mesh_rsn *mesh_rsn = ctx;
    123      1.1.1.1.2.2  snj 	struct hostapd_data *hapd;
    124      1.1.1.1.2.2  snj 	struct sta_info *sta;
    125      1.1.1.1.2.2  snj 
    126      1.1.1.1.2.2  snj 	if (mesh_rsn->wpa_s->current_ssid->mode != WPAS_MODE_MESH)
    127      1.1.1.1.2.2  snj 		return -1;
    128      1.1.1.1.2.2  snj 
    129      1.1.1.1.2.2  snj 	hapd = mesh_rsn->wpa_s->ifmsh->bss[0];
    130      1.1.1.1.2.2  snj 	sta = ap_get_sta(hapd, addr);
    131      1.1.1.1.2.2  snj 	if (sta)
    132      1.1.1.1.2.2  snj 		eloop_cancel_timeout(mesh_auth_timer, mesh_rsn->wpa_s, sta);
    133      1.1.1.1.2.2  snj 
    134      1.1.1.1.2.2  snj 	mesh_mpm_auth_peer(mesh_rsn->wpa_s, addr);
    135      1.1.1.1.2.2  snj 	return 0;
    136      1.1.1.1.2.2  snj }
    137      1.1.1.1.2.2  snj 
    138      1.1.1.1.2.2  snj 
    139  1.1.1.1.2.2.2.1  snj static int __mesh_rsn_auth_init(struct mesh_rsn *rsn, const u8 *addr,
    140  1.1.1.1.2.2.2.1  snj 				enum mfp_options ieee80211w)
    141      1.1.1.1.2.2  snj {
    142      1.1.1.1.2.2  snj 	struct wpa_auth_config conf;
    143      1.1.1.1.2.2  snj 	struct wpa_auth_callbacks cb;
    144      1.1.1.1.2.2  snj 	u8 seq[6] = {};
    145      1.1.1.1.2.2  snj 
    146      1.1.1.1.2.2  snj 	wpa_printf(MSG_DEBUG, "AUTH: Initializing group state machine");
    147      1.1.1.1.2.2  snj 
    148      1.1.1.1.2.2  snj 	os_memset(&conf, 0, sizeof(conf));
    149  1.1.1.1.2.2.2.1  snj 	conf.wpa = WPA_PROTO_RSN;
    150      1.1.1.1.2.2  snj 	conf.wpa_key_mgmt = WPA_KEY_MGMT_SAE;
    151  1.1.1.1.2.2.2.1  snj 	conf.wpa_pairwise = rsn->pairwise_cipher;
    152  1.1.1.1.2.2.2.1  snj 	conf.rsn_pairwise = rsn->pairwise_cipher;
    153  1.1.1.1.2.2.2.1  snj 	conf.wpa_group = rsn->group_cipher;
    154      1.1.1.1.2.2  snj 	conf.eapol_version = 0;
    155      1.1.1.1.2.2  snj 	conf.wpa_group_rekey = -1;
    156  1.1.1.1.2.2.2.1  snj #ifdef CONFIG_IEEE80211W
    157  1.1.1.1.2.2.2.1  snj 	conf.ieee80211w = ieee80211w;
    158  1.1.1.1.2.2.2.1  snj 	if (ieee80211w != NO_MGMT_FRAME_PROTECTION)
    159  1.1.1.1.2.2.2.1  snj 		conf.group_mgmt_cipher = rsn->mgmt_group_cipher;
    160  1.1.1.1.2.2.2.1  snj #endif /* CONFIG_IEEE80211W */
    161      1.1.1.1.2.2  snj 
    162      1.1.1.1.2.2  snj 	os_memset(&cb, 0, sizeof(cb));
    163      1.1.1.1.2.2  snj 	cb.ctx = rsn;
    164      1.1.1.1.2.2  snj 	cb.logger = auth_logger;
    165      1.1.1.1.2.2  snj 	cb.get_psk = auth_get_psk;
    166      1.1.1.1.2.2  snj 	cb.set_key = auth_set_key;
    167      1.1.1.1.2.2  snj 	cb.start_ampe = auth_start_ampe;
    168      1.1.1.1.2.2  snj 
    169      1.1.1.1.2.2  snj 	rsn->auth = wpa_init(addr, &conf, &cb);
    170      1.1.1.1.2.2  snj 	if (rsn->auth == NULL) {
    171      1.1.1.1.2.2  snj 		wpa_printf(MSG_DEBUG, "AUTH: wpa_init() failed");
    172      1.1.1.1.2.2  snj 		return -1;
    173      1.1.1.1.2.2  snj 	}
    174      1.1.1.1.2.2  snj 
    175      1.1.1.1.2.2  snj 	/* TODO: support rekeying */
    176  1.1.1.1.2.2.2.1  snj 	rsn->mgtk_len = wpa_cipher_key_len(conf.wpa_group);
    177  1.1.1.1.2.2.2.1  snj 	if (random_get_bytes(rsn->mgtk, rsn->mgtk_len) < 0)
    178      1.1.1.1.2.2  snj 		return -1;
    179  1.1.1.1.2.2.2.1  snj 	rsn->mgtk_key_id = 1;
    180  1.1.1.1.2.2.2.1  snj 
    181  1.1.1.1.2.2.2.1  snj #ifdef CONFIG_IEEE80211W
    182  1.1.1.1.2.2.2.1  snj 	if (ieee80211w != NO_MGMT_FRAME_PROTECTION) {
    183  1.1.1.1.2.2.2.1  snj 		rsn->igtk_len = wpa_cipher_key_len(conf.group_mgmt_cipher);
    184  1.1.1.1.2.2.2.1  snj 		if (random_get_bytes(rsn->igtk, rsn->igtk_len) < 0)
    185  1.1.1.1.2.2.2.1  snj 			return -1;
    186  1.1.1.1.2.2.2.1  snj 		rsn->igtk_key_id = 4;
    187      1.1.1.1.2.2  snj 
    188  1.1.1.1.2.2.2.1  snj 		/* group mgmt */
    189  1.1.1.1.2.2.2.1  snj 		wpa_hexdump_key(MSG_DEBUG, "mesh: Own TX IGTK",
    190  1.1.1.1.2.2.2.1  snj 				rsn->igtk, rsn->igtk_len);
    191  1.1.1.1.2.2.2.1  snj 		wpa_drv_set_key(rsn->wpa_s,
    192  1.1.1.1.2.2.2.1  snj 				wpa_cipher_to_alg(rsn->mgmt_group_cipher), NULL,
    193  1.1.1.1.2.2.2.1  snj 				rsn->igtk_key_id, 1,
    194  1.1.1.1.2.2.2.1  snj 				seq, sizeof(seq), rsn->igtk, rsn->igtk_len);
    195  1.1.1.1.2.2.2.1  snj 	}
    196  1.1.1.1.2.2.2.1  snj #endif /* CONFIG_IEEE80211W */
    197      1.1.1.1.2.2  snj 
    198      1.1.1.1.2.2  snj 	/* group privacy / data frames */
    199  1.1.1.1.2.2.2.1  snj 	wpa_hexdump_key(MSG_DEBUG, "mesh: Own TX MGTK",
    200  1.1.1.1.2.2.2.1  snj 			rsn->mgtk, rsn->mgtk_len);
    201  1.1.1.1.2.2.2.1  snj 	wpa_drv_set_key(rsn->wpa_s, wpa_cipher_to_alg(rsn->group_cipher), NULL,
    202  1.1.1.1.2.2.2.1  snj 			rsn->mgtk_key_id, 1, seq, sizeof(seq),
    203  1.1.1.1.2.2.2.1  snj 			rsn->mgtk, rsn->mgtk_len);
    204      1.1.1.1.2.2  snj 
    205      1.1.1.1.2.2  snj 	return 0;
    206      1.1.1.1.2.2  snj }
    207      1.1.1.1.2.2  snj 
    208      1.1.1.1.2.2  snj 
    209      1.1.1.1.2.2  snj static void mesh_rsn_deinit(struct mesh_rsn *rsn)
    210      1.1.1.1.2.2  snj {
    211      1.1.1.1.2.2  snj 	os_memset(rsn->mgtk, 0, sizeof(rsn->mgtk));
    212  1.1.1.1.2.2.2.1  snj 	rsn->mgtk_len = 0;
    213  1.1.1.1.2.2.2.1  snj 	os_memset(rsn->igtk, 0, sizeof(rsn->igtk));
    214  1.1.1.1.2.2.2.1  snj 	rsn->igtk_len = 0;
    215  1.1.1.1.2.2.2.1  snj 	if (rsn->auth)
    216  1.1.1.1.2.2.2.1  snj 		wpa_deinit(rsn->auth);
    217      1.1.1.1.2.2  snj }
    218      1.1.1.1.2.2  snj 
    219      1.1.1.1.2.2  snj 
    220      1.1.1.1.2.2  snj struct mesh_rsn *mesh_rsn_auth_init(struct wpa_supplicant *wpa_s,
    221      1.1.1.1.2.2  snj 				    struct mesh_conf *conf)
    222      1.1.1.1.2.2  snj {
    223      1.1.1.1.2.2  snj 	struct mesh_rsn *mesh_rsn;
    224      1.1.1.1.2.2  snj 	struct hostapd_data *bss = wpa_s->ifmsh->bss[0];
    225      1.1.1.1.2.2  snj 	const u8 *ie;
    226      1.1.1.1.2.2  snj 	size_t ie_len;
    227      1.1.1.1.2.2  snj 
    228      1.1.1.1.2.2  snj 	mesh_rsn = os_zalloc(sizeof(*mesh_rsn));
    229      1.1.1.1.2.2  snj 	if (mesh_rsn == NULL)
    230      1.1.1.1.2.2  snj 		return NULL;
    231      1.1.1.1.2.2  snj 	mesh_rsn->wpa_s = wpa_s;
    232  1.1.1.1.2.2.2.1  snj 	mesh_rsn->pairwise_cipher = conf->pairwise_cipher;
    233  1.1.1.1.2.2.2.1  snj 	mesh_rsn->group_cipher = conf->group_cipher;
    234  1.1.1.1.2.2.2.1  snj 	mesh_rsn->mgmt_group_cipher = conf->mgmt_group_cipher;
    235      1.1.1.1.2.2  snj 
    236  1.1.1.1.2.2.2.1  snj 	if (__mesh_rsn_auth_init(mesh_rsn, wpa_s->own_addr,
    237  1.1.1.1.2.2.2.1  snj 				 conf->ieee80211w) < 0) {
    238      1.1.1.1.2.2  snj 		mesh_rsn_deinit(mesh_rsn);
    239  1.1.1.1.2.2.2.1  snj 		os_free(mesh_rsn);
    240      1.1.1.1.2.2  snj 		return NULL;
    241      1.1.1.1.2.2  snj 	}
    242      1.1.1.1.2.2  snj 
    243      1.1.1.1.2.2  snj 	bss->wpa_auth = mesh_rsn->auth;
    244      1.1.1.1.2.2  snj 
    245      1.1.1.1.2.2  snj 	ie = wpa_auth_get_wpa_ie(mesh_rsn->auth, &ie_len);
    246  1.1.1.1.2.2.2.1  snj 	conf->rsn_ie = (u8 *) ie;
    247  1.1.1.1.2.2.2.1  snj 	conf->rsn_ie_len = ie_len;
    248      1.1.1.1.2.2  snj 
    249      1.1.1.1.2.2  snj 	wpa_supplicant_rsn_supp_set_config(wpa_s, wpa_s->current_ssid);
    250      1.1.1.1.2.2  snj 
    251      1.1.1.1.2.2  snj 	return mesh_rsn;
    252      1.1.1.1.2.2  snj }
    253      1.1.1.1.2.2  snj 
    254      1.1.1.1.2.2  snj 
    255      1.1.1.1.2.2  snj static int index_within_array(const int *array, int idx)
    256      1.1.1.1.2.2  snj {
    257      1.1.1.1.2.2  snj 	int i;
    258      1.1.1.1.2.2  snj 
    259      1.1.1.1.2.2  snj 	for (i = 0; i < idx; i++) {
    260      1.1.1.1.2.2  snj 		if (array[i] == -1)
    261      1.1.1.1.2.2  snj 			return 0;
    262      1.1.1.1.2.2  snj 	}
    263      1.1.1.1.2.2  snj 
    264      1.1.1.1.2.2  snj 	return 1;
    265      1.1.1.1.2.2  snj }
    266      1.1.1.1.2.2  snj 
    267      1.1.1.1.2.2  snj 
    268      1.1.1.1.2.2  snj static int mesh_rsn_sae_group(struct wpa_supplicant *wpa_s,
    269      1.1.1.1.2.2  snj 			      struct sae_data *sae)
    270      1.1.1.1.2.2  snj {
    271      1.1.1.1.2.2  snj 	int *groups = wpa_s->ifmsh->bss[0]->conf->sae_groups;
    272      1.1.1.1.2.2  snj 
    273      1.1.1.1.2.2  snj 	/* Configuration may have changed, so validate current index */
    274      1.1.1.1.2.2  snj 	if (!index_within_array(groups, wpa_s->mesh_rsn->sae_group_index))
    275      1.1.1.1.2.2  snj 		return -1;
    276      1.1.1.1.2.2  snj 
    277      1.1.1.1.2.2  snj 	for (;;) {
    278      1.1.1.1.2.2  snj 		int group = groups[wpa_s->mesh_rsn->sae_group_index];
    279      1.1.1.1.2.2  snj 
    280      1.1.1.1.2.2  snj 		if (group <= 0)
    281      1.1.1.1.2.2  snj 			break;
    282      1.1.1.1.2.2  snj 		if (sae_set_group(sae, group) == 0) {
    283      1.1.1.1.2.2  snj 			wpa_dbg(wpa_s, MSG_DEBUG, "SME: Selected SAE group %d",
    284      1.1.1.1.2.2  snj 				sae->group);
    285      1.1.1.1.2.2  snj 			return 0;
    286      1.1.1.1.2.2  snj 		}
    287      1.1.1.1.2.2  snj 		wpa_s->mesh_rsn->sae_group_index++;
    288      1.1.1.1.2.2  snj 	}
    289      1.1.1.1.2.2  snj 
    290      1.1.1.1.2.2  snj 	return -1;
    291      1.1.1.1.2.2  snj }
    292      1.1.1.1.2.2  snj 
    293      1.1.1.1.2.2  snj 
    294      1.1.1.1.2.2  snj static int mesh_rsn_build_sae_commit(struct wpa_supplicant *wpa_s,
    295      1.1.1.1.2.2  snj 				     struct wpa_ssid *ssid,
    296      1.1.1.1.2.2  snj 				     struct sta_info *sta)
    297      1.1.1.1.2.2  snj {
    298      1.1.1.1.2.2  snj 	if (ssid->passphrase == NULL) {
    299      1.1.1.1.2.2  snj 		wpa_msg(wpa_s, MSG_DEBUG, "SAE: No password available");
    300      1.1.1.1.2.2  snj 		return -1;
    301      1.1.1.1.2.2  snj 	}
    302      1.1.1.1.2.2  snj 
    303      1.1.1.1.2.2  snj 	if (mesh_rsn_sae_group(wpa_s, sta->sae) < 0) {
    304      1.1.1.1.2.2  snj 		wpa_msg(wpa_s, MSG_DEBUG, "SAE: Failed to select group");
    305      1.1.1.1.2.2  snj 		return -1;
    306      1.1.1.1.2.2  snj 	}
    307      1.1.1.1.2.2  snj 
    308      1.1.1.1.2.2  snj 	return sae_prepare_commit(wpa_s->own_addr, sta->addr,
    309      1.1.1.1.2.2  snj 				  (u8 *) ssid->passphrase,
    310      1.1.1.1.2.2  snj 				  os_strlen(ssid->passphrase), sta->sae);
    311      1.1.1.1.2.2  snj }
    312      1.1.1.1.2.2  snj 
    313      1.1.1.1.2.2  snj 
    314      1.1.1.1.2.2  snj /* initiate new SAE authentication with sta */
    315      1.1.1.1.2.2  snj int mesh_rsn_auth_sae_sta(struct wpa_supplicant *wpa_s,
    316      1.1.1.1.2.2  snj 			  struct sta_info *sta)
    317      1.1.1.1.2.2  snj {
    318      1.1.1.1.2.2  snj 	struct hostapd_data *hapd = wpa_s->ifmsh->bss[0];
    319      1.1.1.1.2.2  snj 	struct wpa_ssid *ssid = wpa_s->current_ssid;
    320  1.1.1.1.2.2.2.1  snj 	struct rsn_pmksa_cache_entry *pmksa;
    321      1.1.1.1.2.2  snj 	unsigned int rnd;
    322      1.1.1.1.2.2  snj 	int ret;
    323      1.1.1.1.2.2  snj 
    324      1.1.1.1.2.2  snj 	if (!ssid) {
    325      1.1.1.1.2.2  snj 		wpa_msg(wpa_s, MSG_DEBUG,
    326      1.1.1.1.2.2  snj 			"AUTH: No current_ssid known to initiate new SAE");
    327      1.1.1.1.2.2  snj 		return -1;
    328      1.1.1.1.2.2  snj 	}
    329      1.1.1.1.2.2  snj 
    330      1.1.1.1.2.2  snj 	if (!sta->sae) {
    331      1.1.1.1.2.2  snj 		sta->sae = os_zalloc(sizeof(*sta->sae));
    332      1.1.1.1.2.2  snj 		if (sta->sae == NULL)
    333      1.1.1.1.2.2  snj 			return -1;
    334      1.1.1.1.2.2  snj 	}
    335      1.1.1.1.2.2  snj 
    336  1.1.1.1.2.2.2.1  snj 	pmksa = wpa_auth_pmksa_get(hapd->wpa_auth, sta->addr);
    337  1.1.1.1.2.2.2.1  snj 	if (pmksa) {
    338  1.1.1.1.2.2.2.1  snj 		if (!sta->wpa_sm)
    339  1.1.1.1.2.2.2.1  snj 			sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
    340  1.1.1.1.2.2.2.1  snj 							sta->addr, NULL);
    341  1.1.1.1.2.2.2.1  snj 		if (!sta->wpa_sm) {
    342  1.1.1.1.2.2.2.1  snj 			wpa_printf(MSG_ERROR,
    343  1.1.1.1.2.2.2.1  snj 				   "mesh: Failed to initialize RSN state machine");
    344  1.1.1.1.2.2.2.1  snj 			return -1;
    345  1.1.1.1.2.2.2.1  snj 		}
    346  1.1.1.1.2.2.2.1  snj 
    347  1.1.1.1.2.2.2.1  snj 		wpa_printf(MSG_DEBUG,
    348  1.1.1.1.2.2.2.1  snj 			   "AUTH: Mesh PMKSA cache entry found for " MACSTR
    349  1.1.1.1.2.2.2.1  snj 			   " - try to use PMKSA caching instead of new SAE authentication",
    350  1.1.1.1.2.2.2.1  snj 			   MAC2STR(sta->addr));
    351  1.1.1.1.2.2.2.1  snj 		wpa_auth_pmksa_set_to_sm(pmksa, sta->wpa_sm, hapd->wpa_auth,
    352  1.1.1.1.2.2.2.1  snj 					 sta->sae->pmkid, sta->sae->pmk);
    353  1.1.1.1.2.2.2.1  snj 		sae_accept_sta(hapd, sta);
    354  1.1.1.1.2.2.2.1  snj 		sta->mesh_sae_pmksa_caching = 1;
    355  1.1.1.1.2.2.2.1  snj 		return 0;
    356  1.1.1.1.2.2.2.1  snj 	}
    357  1.1.1.1.2.2.2.1  snj 	sta->mesh_sae_pmksa_caching = 0;
    358  1.1.1.1.2.2.2.1  snj 
    359      1.1.1.1.2.2  snj 	if (mesh_rsn_build_sae_commit(wpa_s, ssid, sta))
    360      1.1.1.1.2.2  snj 		return -1;
    361      1.1.1.1.2.2  snj 
    362      1.1.1.1.2.2  snj 	wpa_msg(wpa_s, MSG_DEBUG,
    363      1.1.1.1.2.2  snj 		"AUTH: started authentication with SAE peer: " MACSTR,
    364      1.1.1.1.2.2  snj 		MAC2STR(sta->addr));
    365      1.1.1.1.2.2  snj 
    366      1.1.1.1.2.2  snj 	ret = auth_sae_init_committed(hapd, sta);
    367      1.1.1.1.2.2  snj 	if (ret)
    368      1.1.1.1.2.2  snj 		return ret;
    369      1.1.1.1.2.2  snj 
    370      1.1.1.1.2.2  snj 	eloop_cancel_timeout(mesh_auth_timer, wpa_s, sta);
    371      1.1.1.1.2.2  snj 	rnd = rand() % MESH_AUTH_TIMEOUT;
    372      1.1.1.1.2.2  snj 	eloop_register_timeout(MESH_AUTH_TIMEOUT + rnd, 0, mesh_auth_timer,
    373      1.1.1.1.2.2  snj 			       wpa_s, sta);
    374      1.1.1.1.2.2  snj 	return 0;
    375      1.1.1.1.2.2  snj }
    376      1.1.1.1.2.2  snj 
    377      1.1.1.1.2.2  snj 
    378      1.1.1.1.2.2  snj void mesh_rsn_get_pmkid(struct mesh_rsn *rsn, struct sta_info *sta, u8 *pmkid)
    379      1.1.1.1.2.2  snj {
    380  1.1.1.1.2.2.2.1  snj 	os_memcpy(pmkid, sta->sae->pmkid, SAE_PMKID_LEN);
    381      1.1.1.1.2.2  snj }
    382      1.1.1.1.2.2  snj 
    383      1.1.1.1.2.2  snj 
    384      1.1.1.1.2.2  snj static void
    385      1.1.1.1.2.2  snj mesh_rsn_derive_aek(struct mesh_rsn *rsn, struct sta_info *sta)
    386      1.1.1.1.2.2  snj {
    387      1.1.1.1.2.2  snj 	u8 *myaddr = rsn->wpa_s->own_addr;
    388      1.1.1.1.2.2  snj 	u8 *peer = sta->addr;
    389  1.1.1.1.2.2.2.1  snj 	u8 *addr1, *addr2;
    390  1.1.1.1.2.2.2.1  snj 	u8 context[RSN_SELECTOR_LEN + 2 * ETH_ALEN], *ptr = context;
    391      1.1.1.1.2.2  snj 
    392  1.1.1.1.2.2.2.1  snj 	/*
    393  1.1.1.1.2.2.2.1  snj 	 * AEK = KDF-Hash-256(PMK, "AEK Derivation", Selected AKM Suite ||
    394  1.1.1.1.2.2.2.1  snj 	 *       min(localMAC, peerMAC) || max(localMAC, peerMAC))
    395  1.1.1.1.2.2.2.1  snj 	 */
    396  1.1.1.1.2.2.2.1  snj 	/* Selected AKM Suite: SAE */
    397  1.1.1.1.2.2.2.1  snj 	RSN_SELECTOR_PUT(ptr, RSN_AUTH_KEY_MGMT_SAE);
    398  1.1.1.1.2.2.2.1  snj 	ptr += RSN_SELECTOR_LEN;
    399      1.1.1.1.2.2  snj 
    400      1.1.1.1.2.2  snj 	if (os_memcmp(myaddr, peer, ETH_ALEN) < 0) {
    401      1.1.1.1.2.2  snj 		addr1 = myaddr;
    402      1.1.1.1.2.2  snj 		addr2 = peer;
    403  1.1.1.1.2.2.2.1  snj 	} else {
    404  1.1.1.1.2.2.2.1  snj 		addr1 = peer;
    405  1.1.1.1.2.2.2.1  snj 		addr2 = myaddr;
    406      1.1.1.1.2.2  snj 	}
    407  1.1.1.1.2.2.2.1  snj 	os_memcpy(ptr, addr1, ETH_ALEN);
    408  1.1.1.1.2.2.2.1  snj 	ptr += ETH_ALEN;
    409  1.1.1.1.2.2.2.1  snj 	os_memcpy(ptr, addr2, ETH_ALEN);
    410      1.1.1.1.2.2  snj 
    411      1.1.1.1.2.2  snj 	sha256_prf(sta->sae->pmk, sizeof(sta->sae->pmk), "AEK Derivation",
    412      1.1.1.1.2.2  snj 		   context, sizeof(context), sta->aek, sizeof(sta->aek));
    413      1.1.1.1.2.2  snj }
    414      1.1.1.1.2.2  snj 
    415      1.1.1.1.2.2  snj 
    416      1.1.1.1.2.2  snj /* derive mesh temporal key from pmk */
    417      1.1.1.1.2.2  snj int mesh_rsn_derive_mtk(struct wpa_supplicant *wpa_s, struct sta_info *sta)
    418      1.1.1.1.2.2  snj {
    419      1.1.1.1.2.2  snj 	u8 *ptr;
    420      1.1.1.1.2.2  snj 	u8 *min, *max;
    421      1.1.1.1.2.2  snj 	u8 *myaddr = wpa_s->own_addr;
    422      1.1.1.1.2.2  snj 	u8 *peer = sta->addr;
    423  1.1.1.1.2.2.2.1  snj 	u8 context[2 * WPA_NONCE_LEN + 2 * 2 + RSN_SELECTOR_LEN + 2 * ETH_ALEN];
    424      1.1.1.1.2.2  snj 
    425  1.1.1.1.2.2.2.1  snj 	/*
    426  1.1.1.1.2.2.2.1  snj 	 * MTK = KDF-Hash-Length(PMK, "Temporal Key Derivation", min(localNonce,
    427  1.1.1.1.2.2.2.1  snj 	 *  peerNonce) || max(localNonce, peerNonce) || min(localLinkID,
    428  1.1.1.1.2.2.2.1  snj 	 *  peerLinkID) || max(localLinkID, peerLinkID) || Selected AKM Suite ||
    429  1.1.1.1.2.2.2.1  snj 	 *  min(localMAC, peerMAC) || max(localMAC, peerMAC))
    430  1.1.1.1.2.2.2.1  snj 	 */
    431      1.1.1.1.2.2  snj 	ptr = context;
    432  1.1.1.1.2.2.2.1  snj 	if (os_memcmp(sta->my_nonce, sta->peer_nonce, WPA_NONCE_LEN) < 0) {
    433      1.1.1.1.2.2  snj 		min = sta->my_nonce;
    434      1.1.1.1.2.2  snj 		max = sta->peer_nonce;
    435      1.1.1.1.2.2  snj 	} else {
    436      1.1.1.1.2.2  snj 		min = sta->peer_nonce;
    437      1.1.1.1.2.2  snj 		max = sta->my_nonce;
    438      1.1.1.1.2.2  snj 	}
    439  1.1.1.1.2.2.2.1  snj 	os_memcpy(ptr, min, WPA_NONCE_LEN);
    440  1.1.1.1.2.2.2.1  snj 	ptr += WPA_NONCE_LEN;
    441  1.1.1.1.2.2.2.1  snj 	os_memcpy(ptr, max, WPA_NONCE_LEN);
    442  1.1.1.1.2.2.2.1  snj 	ptr += WPA_NONCE_LEN;
    443      1.1.1.1.2.2  snj 
    444      1.1.1.1.2.2  snj 	if (sta->my_lid < sta->peer_lid) {
    445  1.1.1.1.2.2.2.1  snj 		WPA_PUT_LE16(ptr, sta->my_lid);
    446  1.1.1.1.2.2.2.1  snj 		ptr += 2;
    447  1.1.1.1.2.2.2.1  snj 		WPA_PUT_LE16(ptr, sta->peer_lid);
    448  1.1.1.1.2.2.2.1  snj 		ptr += 2;
    449      1.1.1.1.2.2  snj 	} else {
    450  1.1.1.1.2.2.2.1  snj 		WPA_PUT_LE16(ptr, sta->peer_lid);
    451  1.1.1.1.2.2.2.1  snj 		ptr += 2;
    452  1.1.1.1.2.2.2.1  snj 		WPA_PUT_LE16(ptr, sta->my_lid);
    453  1.1.1.1.2.2.2.1  snj 		ptr += 2;
    454      1.1.1.1.2.2  snj 	}
    455  1.1.1.1.2.2.2.1  snj 
    456  1.1.1.1.2.2.2.1  snj 	/* Selected AKM Suite: SAE */
    457  1.1.1.1.2.2.2.1  snj 	RSN_SELECTOR_PUT(ptr, RSN_AUTH_KEY_MGMT_SAE);
    458  1.1.1.1.2.2.2.1  snj 	ptr += RSN_SELECTOR_LEN;
    459      1.1.1.1.2.2  snj 
    460      1.1.1.1.2.2  snj 	if (os_memcmp(myaddr, peer, ETH_ALEN) < 0) {
    461      1.1.1.1.2.2  snj 		min = myaddr;
    462      1.1.1.1.2.2  snj 		max = peer;
    463      1.1.1.1.2.2  snj 	} else {
    464      1.1.1.1.2.2  snj 		min = peer;
    465      1.1.1.1.2.2  snj 		max = myaddr;
    466      1.1.1.1.2.2  snj 	}
    467      1.1.1.1.2.2  snj 	os_memcpy(ptr, min, ETH_ALEN);
    468  1.1.1.1.2.2.2.1  snj 	ptr += ETH_ALEN;
    469  1.1.1.1.2.2.2.1  snj 	os_memcpy(ptr, max, ETH_ALEN);
    470      1.1.1.1.2.2  snj 
    471  1.1.1.1.2.2.2.1  snj 	sta->mtk_len = wpa_cipher_key_len(wpa_s->mesh_rsn->pairwise_cipher);
    472  1.1.1.1.2.2.2.1  snj 	sha256_prf(sta->sae->pmk, SAE_PMK_LEN,
    473      1.1.1.1.2.2  snj 		   "Temporal Key Derivation", context, sizeof(context),
    474  1.1.1.1.2.2.2.1  snj 		   sta->mtk, sta->mtk_len);
    475      1.1.1.1.2.2  snj 	return 0;
    476      1.1.1.1.2.2  snj }
    477      1.1.1.1.2.2  snj 
    478      1.1.1.1.2.2  snj 
    479      1.1.1.1.2.2  snj void mesh_rsn_init_ampe_sta(struct wpa_supplicant *wpa_s, struct sta_info *sta)
    480      1.1.1.1.2.2  snj {
    481  1.1.1.1.2.2.2.1  snj 	if (random_get_bytes(sta->my_nonce, WPA_NONCE_LEN) < 0) {
    482      1.1.1.1.2.2  snj 		wpa_printf(MSG_INFO, "mesh: Failed to derive random nonce");
    483      1.1.1.1.2.2  snj 		/* TODO: How to handle this more cleanly? */
    484      1.1.1.1.2.2  snj 	}
    485  1.1.1.1.2.2.2.1  snj 	os_memset(sta->peer_nonce, 0, WPA_NONCE_LEN);
    486      1.1.1.1.2.2  snj 	mesh_rsn_derive_aek(wpa_s->mesh_rsn, sta);
    487      1.1.1.1.2.2  snj }
    488      1.1.1.1.2.2  snj 
    489      1.1.1.1.2.2  snj 
    490      1.1.1.1.2.2  snj /* insert AMPE and encrypted MIC at @ie.
    491      1.1.1.1.2.2  snj  * @mesh_rsn: mesh RSN context
    492      1.1.1.1.2.2  snj  * @sta: STA we're sending to
    493      1.1.1.1.2.2  snj  * @cat: pointer to category code in frame header.
    494      1.1.1.1.2.2  snj  * @buf: wpabuf to add encrypted AMPE and MIC to.
    495      1.1.1.1.2.2  snj  * */
    496      1.1.1.1.2.2  snj int mesh_rsn_protect_frame(struct mesh_rsn *rsn, struct sta_info *sta,
    497      1.1.1.1.2.2  snj 			   const u8 *cat, struct wpabuf *buf)
    498      1.1.1.1.2.2  snj {
    499      1.1.1.1.2.2  snj 	struct ieee80211_ampe_ie *ampe;
    500      1.1.1.1.2.2  snj 	u8 const *ie = wpabuf_head_u8(buf) + wpabuf_len(buf);
    501  1.1.1.1.2.2.2.1  snj 	u8 *ampe_ie, *pos, *mic_payload;
    502      1.1.1.1.2.2  snj 	const u8 *aad[] = { rsn->wpa_s->own_addr, sta->addr, cat };
    503      1.1.1.1.2.2  snj 	const size_t aad_len[] = { ETH_ALEN, ETH_ALEN, ie - cat };
    504      1.1.1.1.2.2  snj 	int ret = 0;
    505  1.1.1.1.2.2.2.1  snj 	size_t len;
    506      1.1.1.1.2.2  snj 
    507  1.1.1.1.2.2.2.1  snj 	len = sizeof(*ampe);
    508  1.1.1.1.2.2.2.1  snj 	if (cat[1] == PLINK_OPEN)
    509  1.1.1.1.2.2.2.1  snj 		len += rsn->mgtk_len + WPA_KEY_RSC_LEN + 4;
    510  1.1.1.1.2.2.2.1  snj #ifdef CONFIG_IEEE80211W
    511  1.1.1.1.2.2.2.1  snj 	if (cat[1] == PLINK_OPEN && rsn->igtk_len)
    512  1.1.1.1.2.2.2.1  snj 		len += 2 + 6 + rsn->igtk_len;
    513  1.1.1.1.2.2.2.1  snj #endif /* CONFIG_IEEE80211W */
    514  1.1.1.1.2.2.2.1  snj 
    515  1.1.1.1.2.2.2.1  snj 	if (2 + AES_BLOCK_SIZE + 2 + len > wpabuf_tailroom(buf)) {
    516      1.1.1.1.2.2  snj 		wpa_printf(MSG_ERROR, "protect frame: buffer too small");
    517      1.1.1.1.2.2  snj 		return -EINVAL;
    518      1.1.1.1.2.2  snj 	}
    519      1.1.1.1.2.2  snj 
    520  1.1.1.1.2.2.2.1  snj 	ampe_ie = os_zalloc(2 + len);
    521      1.1.1.1.2.2  snj 	if (!ampe_ie) {
    522      1.1.1.1.2.2  snj 		wpa_printf(MSG_ERROR, "protect frame: out of memory");
    523      1.1.1.1.2.2  snj 		return -ENOMEM;
    524      1.1.1.1.2.2  snj 	}
    525      1.1.1.1.2.2  snj 
    526      1.1.1.1.2.2  snj 	/*  IE: AMPE */
    527      1.1.1.1.2.2  snj 	ampe_ie[0] = WLAN_EID_AMPE;
    528  1.1.1.1.2.2.2.1  snj 	ampe_ie[1] = len;
    529      1.1.1.1.2.2  snj 	ampe = (struct ieee80211_ampe_ie *) (ampe_ie + 2);
    530      1.1.1.1.2.2  snj 
    531      1.1.1.1.2.2  snj 	RSN_SELECTOR_PUT(ampe->selected_pairwise_suite,
    532  1.1.1.1.2.2.2.1  snj 			 RSN_CIPHER_SUITE_CCMP);
    533  1.1.1.1.2.2.2.1  snj 	os_memcpy(ampe->local_nonce, sta->my_nonce, WPA_NONCE_LEN);
    534  1.1.1.1.2.2.2.1  snj 	os_memcpy(ampe->peer_nonce, sta->peer_nonce, WPA_NONCE_LEN);
    535  1.1.1.1.2.2.2.1  snj 
    536  1.1.1.1.2.2.2.1  snj 	pos = (u8 *) (ampe + 1);
    537  1.1.1.1.2.2.2.1  snj 	if (cat[1] != PLINK_OPEN)
    538  1.1.1.1.2.2.2.1  snj 		goto skip_keys;
    539  1.1.1.1.2.2.2.1  snj 
    540  1.1.1.1.2.2.2.1  snj 	/* TODO: Key Replay Counter[8] optionally for
    541  1.1.1.1.2.2.2.1  snj 	 * Mesh Group Key Inform/Acknowledge frames */
    542  1.1.1.1.2.2.2.1  snj 
    543      1.1.1.1.2.2  snj 	/* TODO: static mgtk for now since we don't support rekeying! */
    544  1.1.1.1.2.2.2.1  snj 	/*
    545  1.1.1.1.2.2.2.1  snj 	 * GTKdata[variable]:
    546  1.1.1.1.2.2.2.1  snj 	 * MGTK[variable] || Key RSC[8] || GTKExpirationTime[4]
    547  1.1.1.1.2.2.2.1  snj 	 */
    548  1.1.1.1.2.2.2.1  snj 	os_memcpy(pos, rsn->mgtk, rsn->mgtk_len);
    549  1.1.1.1.2.2.2.1  snj 	pos += rsn->mgtk_len;
    550  1.1.1.1.2.2.2.1  snj 	wpa_drv_get_seqnum(rsn->wpa_s, NULL, rsn->mgtk_key_id, pos);
    551  1.1.1.1.2.2.2.1  snj 	pos += WPA_KEY_RSC_LEN;
    552  1.1.1.1.2.2.2.1  snj 	/* Use fixed GTKExpirationTime for now */
    553  1.1.1.1.2.2.2.1  snj 	WPA_PUT_LE32(pos, 0xffffffff);
    554  1.1.1.1.2.2.2.1  snj 	pos += 4;
    555  1.1.1.1.2.2.2.1  snj 
    556  1.1.1.1.2.2.2.1  snj #ifdef CONFIG_IEEE80211W
    557  1.1.1.1.2.2.2.1  snj 	/*
    558  1.1.1.1.2.2.2.1  snj 	 * IGTKdata[variable]:
    559  1.1.1.1.2.2.2.1  snj 	 * Key ID[2], IPN[6], IGTK[variable]
    560  1.1.1.1.2.2.2.1  snj 	 */
    561  1.1.1.1.2.2.2.1  snj 	if (rsn->igtk_len) {
    562  1.1.1.1.2.2.2.1  snj 		WPA_PUT_LE16(pos, rsn->igtk_key_id);
    563  1.1.1.1.2.2.2.1  snj 		pos += 2;
    564  1.1.1.1.2.2.2.1  snj 		wpa_drv_get_seqnum(rsn->wpa_s, NULL, rsn->igtk_key_id, pos);
    565  1.1.1.1.2.2.2.1  snj 		pos += 6;
    566  1.1.1.1.2.2.2.1  snj 		os_memcpy(pos, rsn->igtk, rsn->igtk_len);
    567  1.1.1.1.2.2.2.1  snj 	}
    568  1.1.1.1.2.2.2.1  snj #endif /* CONFIG_IEEE80211W */
    569  1.1.1.1.2.2.2.1  snj 
    570  1.1.1.1.2.2.2.1  snj skip_keys:
    571  1.1.1.1.2.2.2.1  snj 	wpa_hexdump_key(MSG_DEBUG, "mesh: Plaintext AMPE element",
    572  1.1.1.1.2.2.2.1  snj 			ampe_ie, 2 + len);
    573      1.1.1.1.2.2  snj 
    574      1.1.1.1.2.2  snj 	/* IE: MIC */
    575  1.1.1.1.2.2.2.1  snj 	wpabuf_put_u8(buf, WLAN_EID_MIC);
    576  1.1.1.1.2.2.2.1  snj 	wpabuf_put_u8(buf, AES_BLOCK_SIZE);
    577      1.1.1.1.2.2  snj 	/* MIC field is output ciphertext */
    578      1.1.1.1.2.2  snj 
    579      1.1.1.1.2.2  snj 	/* encrypt after MIC */
    580  1.1.1.1.2.2.2.1  snj 	mic_payload = wpabuf_put(buf, 2 + len + AES_BLOCK_SIZE);
    581      1.1.1.1.2.2  snj 
    582  1.1.1.1.2.2.2.1  snj 	if (aes_siv_encrypt(sta->aek, ampe_ie, 2 + len, 3,
    583      1.1.1.1.2.2  snj 			    aad, aad_len, mic_payload)) {
    584      1.1.1.1.2.2  snj 		wpa_printf(MSG_ERROR, "protect frame: failed to encrypt");
    585      1.1.1.1.2.2  snj 		ret = -ENOMEM;
    586      1.1.1.1.2.2  snj 	}
    587      1.1.1.1.2.2  snj 
    588      1.1.1.1.2.2  snj 	os_free(ampe_ie);
    589      1.1.1.1.2.2  snj 
    590      1.1.1.1.2.2  snj 	return ret;
    591      1.1.1.1.2.2  snj }
    592      1.1.1.1.2.2  snj 
    593      1.1.1.1.2.2  snj 
    594      1.1.1.1.2.2  snj int mesh_rsn_process_ampe(struct wpa_supplicant *wpa_s, struct sta_info *sta,
    595      1.1.1.1.2.2  snj 			  struct ieee802_11_elems *elems, const u8 *cat,
    596  1.1.1.1.2.2.2.1  snj 			  const u8 *chosen_pmk,
    597      1.1.1.1.2.2  snj 			  const u8 *start, size_t elems_len)
    598      1.1.1.1.2.2  snj {
    599      1.1.1.1.2.2  snj 	int ret = 0;
    600      1.1.1.1.2.2  snj 	struct ieee80211_ampe_ie *ampe;
    601  1.1.1.1.2.2.2.1  snj 	u8 null_nonce[WPA_NONCE_LEN] = {};
    602      1.1.1.1.2.2  snj 	u8 ampe_eid;
    603      1.1.1.1.2.2  snj 	u8 ampe_ie_len;
    604  1.1.1.1.2.2.2.1  snj 	u8 *ampe_buf, *crypt = NULL, *pos, *end;
    605      1.1.1.1.2.2  snj 	size_t crypt_len;
    606      1.1.1.1.2.2  snj 	const u8 *aad[] = { sta->addr, wpa_s->own_addr, cat };
    607      1.1.1.1.2.2  snj 	const size_t aad_len[] = { ETH_ALEN, ETH_ALEN,
    608      1.1.1.1.2.2  snj 				   (elems->mic - 2) - cat };
    609  1.1.1.1.2.2.2.1  snj 	size_t key_len;
    610  1.1.1.1.2.2.2.1  snj 
    611  1.1.1.1.2.2.2.1  snj 	if (!sta->sae) {
    612  1.1.1.1.2.2.2.1  snj 		struct hostapd_data *hapd = wpa_s->ifmsh->bss[0];
    613  1.1.1.1.2.2.2.1  snj 
    614  1.1.1.1.2.2.2.1  snj 		if (!wpa_auth_pmksa_get(hapd->wpa_auth, sta->addr)) {
    615  1.1.1.1.2.2.2.1  snj 			wpa_printf(MSG_INFO,
    616  1.1.1.1.2.2.2.1  snj 				   "Mesh RSN: SAE is not prepared yet");
    617  1.1.1.1.2.2.2.1  snj 			return -1;
    618  1.1.1.1.2.2.2.1  snj 		}
    619  1.1.1.1.2.2.2.1  snj 		mesh_rsn_auth_sae_sta(wpa_s, sta);
    620  1.1.1.1.2.2.2.1  snj 	}
    621  1.1.1.1.2.2.2.1  snj 
    622  1.1.1.1.2.2.2.1  snj 	if (chosen_pmk && os_memcmp(chosen_pmk, sta->sae->pmkid, PMKID_LEN)) {
    623  1.1.1.1.2.2.2.1  snj 		wpa_msg(wpa_s, MSG_DEBUG,
    624  1.1.1.1.2.2.2.1  snj 			"Mesh RSN: Invalid PMKID (Chosen PMK did not match calculated PMKID)");
    625  1.1.1.1.2.2.2.1  snj 		return -1;
    626  1.1.1.1.2.2.2.1  snj 	}
    627      1.1.1.1.2.2  snj 
    628      1.1.1.1.2.2  snj 	if (!elems->mic || elems->mic_len < AES_BLOCK_SIZE) {
    629      1.1.1.1.2.2  snj 		wpa_msg(wpa_s, MSG_DEBUG, "Mesh RSN: missing mic ie");
    630      1.1.1.1.2.2  snj 		return -1;
    631      1.1.1.1.2.2  snj 	}
    632      1.1.1.1.2.2  snj 
    633      1.1.1.1.2.2  snj 	ampe_buf = (u8 *) elems->mic + elems->mic_len;
    634      1.1.1.1.2.2  snj 	if ((int) elems_len < ampe_buf - start)
    635      1.1.1.1.2.2  snj 		return -1;
    636      1.1.1.1.2.2  snj 
    637      1.1.1.1.2.2  snj 	crypt_len = elems_len - (elems->mic - start);
    638  1.1.1.1.2.2.2.1  snj 	if (crypt_len < 2 + AES_BLOCK_SIZE) {
    639      1.1.1.1.2.2  snj 		wpa_msg(wpa_s, MSG_DEBUG, "Mesh RSN: missing ampe ie");
    640      1.1.1.1.2.2  snj 		return -1;
    641      1.1.1.1.2.2  snj 	}
    642      1.1.1.1.2.2  snj 
    643      1.1.1.1.2.2  snj 	/* crypt is modified by siv_decrypt */
    644      1.1.1.1.2.2  snj 	crypt = os_zalloc(crypt_len);
    645      1.1.1.1.2.2  snj 	if (!crypt) {
    646      1.1.1.1.2.2  snj 		wpa_printf(MSG_ERROR, "Mesh RSN: out of memory");
    647      1.1.1.1.2.2  snj 		ret = -ENOMEM;
    648      1.1.1.1.2.2  snj 		goto free;
    649      1.1.1.1.2.2  snj 	}
    650      1.1.1.1.2.2  snj 
    651      1.1.1.1.2.2  snj 	os_memcpy(crypt, elems->mic, crypt_len);
    652      1.1.1.1.2.2  snj 
    653      1.1.1.1.2.2  snj 	if (aes_siv_decrypt(sta->aek, crypt, crypt_len, 3,
    654      1.1.1.1.2.2  snj 			    aad, aad_len, ampe_buf)) {
    655      1.1.1.1.2.2  snj 		wpa_printf(MSG_ERROR, "Mesh RSN: frame verification failed!");
    656  1.1.1.1.2.2.2.1  snj 		ret = -2;
    657      1.1.1.1.2.2  snj 		goto free;
    658      1.1.1.1.2.2  snj 	}
    659      1.1.1.1.2.2  snj 
    660  1.1.1.1.2.2.2.1  snj 	crypt_len -= AES_BLOCK_SIZE;
    661  1.1.1.1.2.2.2.1  snj 	wpa_hexdump_key(MSG_DEBUG, "mesh: Decrypted AMPE element",
    662  1.1.1.1.2.2.2.1  snj 			ampe_buf, crypt_len);
    663  1.1.1.1.2.2.2.1  snj 
    664      1.1.1.1.2.2  snj 	ampe_eid = *ampe_buf++;
    665      1.1.1.1.2.2  snj 	ampe_ie_len = *ampe_buf++;
    666      1.1.1.1.2.2  snj 
    667      1.1.1.1.2.2  snj 	if (ampe_eid != WLAN_EID_AMPE ||
    668  1.1.1.1.2.2.2.1  snj 	    (size_t) 2 + ampe_ie_len > crypt_len ||
    669      1.1.1.1.2.2  snj 	    ampe_ie_len < sizeof(struct ieee80211_ampe_ie)) {
    670      1.1.1.1.2.2  snj 		wpa_msg(wpa_s, MSG_DEBUG, "Mesh RSN: invalid ampe ie");
    671      1.1.1.1.2.2  snj 		ret = -1;
    672      1.1.1.1.2.2  snj 		goto free;
    673      1.1.1.1.2.2  snj 	}
    674      1.1.1.1.2.2  snj 
    675      1.1.1.1.2.2  snj 	ampe = (struct ieee80211_ampe_ie *) ampe_buf;
    676  1.1.1.1.2.2.2.1  snj 	pos = (u8 *) (ampe + 1);
    677  1.1.1.1.2.2.2.1  snj 	end = ampe_buf + ampe_ie_len;
    678  1.1.1.1.2.2.2.1  snj 	if (os_memcmp(ampe->peer_nonce, null_nonce, WPA_NONCE_LEN) != 0 &&
    679  1.1.1.1.2.2.2.1  snj 	    os_memcmp(ampe->peer_nonce, sta->my_nonce, WPA_NONCE_LEN) != 0) {
    680      1.1.1.1.2.2  snj 		wpa_msg(wpa_s, MSG_DEBUG, "Mesh RSN: invalid peer nonce");
    681      1.1.1.1.2.2  snj 		ret = -1;
    682      1.1.1.1.2.2  snj 		goto free;
    683      1.1.1.1.2.2  snj 	}
    684      1.1.1.1.2.2  snj 	os_memcpy(sta->peer_nonce, ampe->local_nonce,
    685      1.1.1.1.2.2  snj 		  sizeof(ampe->local_nonce));
    686      1.1.1.1.2.2  snj 
    687  1.1.1.1.2.2.2.1  snj 	/* TODO: Key Replay Counter[8] in Mesh Group Key Inform/Acknowledge
    688  1.1.1.1.2.2.2.1  snj 	 * frames */
    689  1.1.1.1.2.2.2.1  snj 
    690  1.1.1.1.2.2.2.1  snj 	/*
    691  1.1.1.1.2.2.2.1  snj 	 * GTKdata shall not be included in Mesh Peering Confirm. While the
    692  1.1.1.1.2.2.2.1  snj 	 * standard does not state the same about IGTKdata, that same constraint
    693  1.1.1.1.2.2.2.1  snj 	 * needs to apply for it. It makes no sense to include the keys in Mesh
    694  1.1.1.1.2.2.2.1  snj 	 * Peering Close frames either, so while the standard does not seem to
    695  1.1.1.1.2.2.2.1  snj 	 * have a shall statement for these, they are described without
    696  1.1.1.1.2.2.2.1  snj 	 * mentioning GTKdata.
    697  1.1.1.1.2.2.2.1  snj 	 *
    698  1.1.1.1.2.2.2.1  snj 	 * An earlier implementation used to add GTKdata to both Mesh Peering
    699  1.1.1.1.2.2.2.1  snj 	 * Open and Mesh Peering Confirm frames, so ignore the possibly present
    700  1.1.1.1.2.2.2.1  snj 	 * GTKdata frame without rejecting the frame as a backwards
    701  1.1.1.1.2.2.2.1  snj 	 * compatibility mechanism.
    702  1.1.1.1.2.2.2.1  snj 	 */
    703  1.1.1.1.2.2.2.1  snj 	if (cat[1] != PLINK_OPEN) {
    704  1.1.1.1.2.2.2.1  snj 		if (end > pos) {
    705  1.1.1.1.2.2.2.1  snj 			wpa_hexdump_key(MSG_DEBUG,
    706  1.1.1.1.2.2.2.1  snj 					"mesh: Ignore unexpected GTKdata(etc.) fields in the end of AMPE element in Mesh Peering Confirm/Close",
    707  1.1.1.1.2.2.2.1  snj 					pos, end - pos);
    708  1.1.1.1.2.2.2.1  snj 		}
    709  1.1.1.1.2.2.2.1  snj 		goto free;
    710  1.1.1.1.2.2.2.1  snj 	}
    711  1.1.1.1.2.2.2.1  snj 
    712  1.1.1.1.2.2.2.1  snj 	/*
    713  1.1.1.1.2.2.2.1  snj 	 * GTKdata[variable]:
    714  1.1.1.1.2.2.2.1  snj 	 * MGTK[variable] || Key RSC[8] || GTKExpirationTime[4]
    715  1.1.1.1.2.2.2.1  snj 	 */
    716  1.1.1.1.2.2.2.1  snj 	sta->mgtk_key_id = 1; /* FIX: Where to get Key ID? */
    717  1.1.1.1.2.2.2.1  snj 	key_len = wpa_cipher_key_len(wpa_s->mesh_rsn->group_cipher);
    718  1.1.1.1.2.2.2.1  snj 	if ((int) key_len + WPA_KEY_RSC_LEN + 4 > end - pos) {
    719  1.1.1.1.2.2.2.1  snj 		wpa_dbg(wpa_s, MSG_DEBUG, "mesh: Truncated AMPE element");
    720  1.1.1.1.2.2.2.1  snj 		ret = -1;
    721  1.1.1.1.2.2.2.1  snj 		goto free;
    722  1.1.1.1.2.2.2.1  snj 	}
    723  1.1.1.1.2.2.2.1  snj 	sta->mgtk_len = key_len;
    724  1.1.1.1.2.2.2.1  snj 	os_memcpy(sta->mgtk, pos, sta->mgtk_len);
    725  1.1.1.1.2.2.2.1  snj 	wpa_hexdump_key(MSG_DEBUG, "mesh: GTKdata - MGTK",
    726  1.1.1.1.2.2.2.1  snj 			sta->mgtk, sta->mgtk_len);
    727  1.1.1.1.2.2.2.1  snj 	pos += sta->mgtk_len;
    728  1.1.1.1.2.2.2.1  snj 	wpa_hexdump(MSG_DEBUG, "mesh: GTKdata - MGTK - Key RSC",
    729  1.1.1.1.2.2.2.1  snj 		    pos, WPA_KEY_RSC_LEN);
    730  1.1.1.1.2.2.2.1  snj 	os_memcpy(sta->mgtk_rsc, pos, sizeof(sta->mgtk_rsc));
    731  1.1.1.1.2.2.2.1  snj 	pos += WPA_KEY_RSC_LEN;
    732  1.1.1.1.2.2.2.1  snj 	wpa_printf(MSG_DEBUG,
    733  1.1.1.1.2.2.2.1  snj 		   "mesh: GTKdata - MGTK - GTKExpirationTime: %u seconds",
    734  1.1.1.1.2.2.2.1  snj 		   WPA_GET_LE32(pos));
    735  1.1.1.1.2.2.2.1  snj 	pos += 4;
    736  1.1.1.1.2.2.2.1  snj 
    737  1.1.1.1.2.2.2.1  snj #ifdef CONFIG_IEEE80211W
    738  1.1.1.1.2.2.2.1  snj 	/*
    739  1.1.1.1.2.2.2.1  snj 	 * IGTKdata[variable]:
    740  1.1.1.1.2.2.2.1  snj 	 * Key ID[2], IPN[6], IGTK[variable]
    741  1.1.1.1.2.2.2.1  snj 	 */
    742  1.1.1.1.2.2.2.1  snj 	key_len = wpa_cipher_key_len(wpa_s->mesh_rsn->mgmt_group_cipher);
    743  1.1.1.1.2.2.2.1  snj 	if (end - pos >= (int) (2 + 6 + key_len)) {
    744  1.1.1.1.2.2.2.1  snj 		sta->igtk_key_id = WPA_GET_LE16(pos);
    745  1.1.1.1.2.2.2.1  snj 		wpa_printf(MSG_DEBUG, "mesh: IGTKdata - Key ID %u",
    746  1.1.1.1.2.2.2.1  snj 			   sta->igtk_key_id);
    747  1.1.1.1.2.2.2.1  snj 		pos += 2;
    748  1.1.1.1.2.2.2.1  snj 		os_memcpy(sta->igtk_rsc, pos, sizeof(sta->igtk_rsc));
    749  1.1.1.1.2.2.2.1  snj 		wpa_hexdump(MSG_DEBUG, "mesh: IGTKdata - IPN",
    750  1.1.1.1.2.2.2.1  snj 			    sta->igtk_rsc, sizeof(sta->igtk_rsc));
    751  1.1.1.1.2.2.2.1  snj 		pos += 6;
    752  1.1.1.1.2.2.2.1  snj 		os_memcpy(sta->igtk, pos, key_len);
    753  1.1.1.1.2.2.2.1  snj 		sta->igtk_len = key_len;
    754  1.1.1.1.2.2.2.1  snj 		wpa_hexdump_key(MSG_DEBUG, "mesh: IGTKdata - IGTK",
    755  1.1.1.1.2.2.2.1  snj 				sta->igtk, sta->igtk_len);
    756  1.1.1.1.2.2.2.1  snj 	}
    757  1.1.1.1.2.2.2.1  snj #endif /* CONFIG_IEEE80211W */
    758  1.1.1.1.2.2.2.1  snj 
    759      1.1.1.1.2.2  snj free:
    760      1.1.1.1.2.2  snj 	os_free(crypt);
    761      1.1.1.1.2.2  snj 	return ret;
    762      1.1.1.1.2.2  snj }
    763