Home | History | Annotate | Line # | Download | only in wpa_supplicant
mesh_rsn.c revision 1.1.1.1.2.3 
      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.3  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.3  snj 			hapd = wpa_s->ifmsh->bss[0];
     47  1.1.1.1.2.3  snj 
     48  1.1.1.1.2.2  snj 			if (sta->sae_auth_retry > MESH_AUTH_RETRY) {
     49  1.1.1.1.2.3  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.3  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.3  snj static int __mesh_rsn_auth_init(struct mesh_rsn *rsn, const u8 *addr,
    140  1.1.1.1.2.3  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.3  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.3  snj 	conf.wpa_pairwise = rsn->pairwise_cipher;
    152  1.1.1.1.2.3  snj 	conf.rsn_pairwise = rsn->pairwise_cipher;
    153  1.1.1.1.2.3  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.3  snj #ifdef CONFIG_IEEE80211W
    157  1.1.1.1.2.3  snj 	conf.ieee80211w = ieee80211w;
    158  1.1.1.1.2.3  snj 	if (ieee80211w != NO_MGMT_FRAME_PROTECTION)
    159  1.1.1.1.2.3  snj 		conf.group_mgmt_cipher = rsn->mgmt_group_cipher;
    160  1.1.1.1.2.3  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.3  snj 	rsn->mgtk_len = wpa_cipher_key_len(conf.wpa_group);
    177  1.1.1.1.2.3  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.3  snj 	rsn->mgtk_key_id = 1;
    180  1.1.1.1.2.3  snj 
    181  1.1.1.1.2.3  snj #ifdef CONFIG_IEEE80211W
    182  1.1.1.1.2.3  snj 	if (ieee80211w != NO_MGMT_FRAME_PROTECTION) {
    183  1.1.1.1.2.3  snj 		rsn->igtk_len = wpa_cipher_key_len(conf.group_mgmt_cipher);
    184  1.1.1.1.2.3  snj 		if (random_get_bytes(rsn->igtk, rsn->igtk_len) < 0)
    185  1.1.1.1.2.3  snj 			return -1;
    186  1.1.1.1.2.3  snj 		rsn->igtk_key_id = 4;
    187  1.1.1.1.2.2  snj 
    188  1.1.1.1.2.3  snj 		/* group mgmt */
    189  1.1.1.1.2.3  snj 		wpa_hexdump_key(MSG_DEBUG, "mesh: Own TX IGTK",
    190  1.1.1.1.2.3  snj 				rsn->igtk, rsn->igtk_len);
    191  1.1.1.1.2.3  snj 		wpa_drv_set_key(rsn->wpa_s,
    192  1.1.1.1.2.3  snj 				wpa_cipher_to_alg(rsn->mgmt_group_cipher), NULL,
    193  1.1.1.1.2.3  snj 				rsn->igtk_key_id, 1,
    194  1.1.1.1.2.3  snj 				seq, sizeof(seq), rsn->igtk, rsn->igtk_len);
    195  1.1.1.1.2.3  snj 	}
    196  1.1.1.1.2.3  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.3  snj 	wpa_hexdump_key(MSG_DEBUG, "mesh: Own TX MGTK",
    200  1.1.1.1.2.3  snj 			rsn->mgtk, rsn->mgtk_len);
    201  1.1.1.1.2.3  snj 	wpa_drv_set_key(rsn->wpa_s, wpa_cipher_to_alg(rsn->group_cipher), NULL,
    202  1.1.1.1.2.3  snj 			rsn->mgtk_key_id, 1, seq, sizeof(seq),
    203  1.1.1.1.2.3  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.3  snj 	rsn->mgtk_len = 0;
    213  1.1.1.1.2.3  snj 	os_memset(rsn->igtk, 0, sizeof(rsn->igtk));
    214  1.1.1.1.2.3  snj 	rsn->igtk_len = 0;
    215  1.1.1.1.2.3  snj 	if (rsn->auth)
    216  1.1.1.1.2.3  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.3  snj 	mesh_rsn->pairwise_cipher = conf->pairwise_cipher;
    233  1.1.1.1.2.3  snj 	mesh_rsn->group_cipher = conf->group_cipher;
    234  1.1.1.1.2.3  snj 	mesh_rsn->mgmt_group_cipher = conf->mgmt_group_cipher;
    235  1.1.1.1.2.2  snj 
    236  1.1.1.1.2.3  snj 	if (__mesh_rsn_auth_init(mesh_rsn, wpa_s->own_addr,
    237  1.1.1.1.2.3  snj 				 conf->ieee80211w) < 0) {
    238  1.1.1.1.2.2  snj 		mesh_rsn_deinit(mesh_rsn);
    239  1.1.1.1.2.3  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.3  snj 	conf->rsn_ie = (u8 *) ie;
    247  1.1.1.1.2.3  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.3  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.3  snj 	pmksa = wpa_auth_pmksa_get(hapd->wpa_auth, sta->addr);
    337  1.1.1.1.2.3  snj 	if (pmksa) {
    338  1.1.1.1.2.3  snj 		if (!sta->wpa_sm)
    339  1.1.1.1.2.3  snj 			sta->wpa_sm = wpa_auth_sta_init(hapd->wpa_auth,
    340  1.1.1.1.2.3  snj 							sta->addr, NULL);
    341  1.1.1.1.2.3  snj 		if (!sta->wpa_sm) {
    342  1.1.1.1.2.3  snj 			wpa_printf(MSG_ERROR,
    343  1.1.1.1.2.3  snj 				   "mesh: Failed to initialize RSN state machine");
    344  1.1.1.1.2.3  snj 			return -1;
    345  1.1.1.1.2.3  snj 		}
    346  1.1.1.1.2.3  snj 
    347  1.1.1.1.2.3  snj 		wpa_printf(MSG_DEBUG,
    348  1.1.1.1.2.3  snj 			   "AUTH: Mesh PMKSA cache entry found for " MACSTR
    349  1.1.1.1.2.3  snj 			   " - try to use PMKSA caching instead of new SAE authentication",
    350  1.1.1.1.2.3  snj 			   MAC2STR(sta->addr));
    351  1.1.1.1.2.3  snj 		wpa_auth_pmksa_set_to_sm(pmksa, sta->wpa_sm, hapd->wpa_auth,
    352  1.1.1.1.2.3  snj 					 sta->sae->pmkid, sta->sae->pmk);
    353  1.1.1.1.2.3  snj 		sae_accept_sta(hapd, sta);
    354  1.1.1.1.2.3  snj 		sta->mesh_sae_pmksa_caching = 1;
    355  1.1.1.1.2.3  snj 		return 0;
    356  1.1.1.1.2.3  snj 	}
    357  1.1.1.1.2.3  snj 	sta->mesh_sae_pmksa_caching = 0;
    358  1.1.1.1.2.3  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.3  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.3  snj 	u8 *addr1, *addr2;
    390  1.1.1.1.2.3  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.3  snj 	/*
    393  1.1.1.1.2.3  snj 	 * AEK = KDF-Hash-256(PMK, "AEK Derivation", Selected AKM Suite ||
    394  1.1.1.1.2.3  snj 	 *       min(localMAC, peerMAC) || max(localMAC, peerMAC))
    395  1.1.1.1.2.3  snj 	 */
    396  1.1.1.1.2.3  snj 	/* Selected AKM Suite: SAE */
    397  1.1.1.1.2.3  snj 	RSN_SELECTOR_PUT(ptr, RSN_AUTH_KEY_MGMT_SAE);
    398  1.1.1.1.2.3  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.3  snj 	} else {
    404  1.1.1.1.2.3  snj 		addr1 = peer;
    405  1.1.1.1.2.3  snj 		addr2 = myaddr;
    406  1.1.1.1.2.2  snj 	}
    407  1.1.1.1.2.3  snj 	os_memcpy(ptr, addr1, ETH_ALEN);
    408  1.1.1.1.2.3  snj 	ptr += ETH_ALEN;
    409  1.1.1.1.2.3  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.3  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.3  snj 	/*
    426  1.1.1.1.2.3  snj 	 * MTK = KDF-Hash-Length(PMK, "Temporal Key Derivation", min(localNonce,
    427  1.1.1.1.2.3  snj 	 *  peerNonce) || max(localNonce, peerNonce) || min(localLinkID,
    428  1.1.1.1.2.3  snj 	 *  peerLinkID) || max(localLinkID, peerLinkID) || Selected AKM Suite ||
    429  1.1.1.1.2.3  snj 	 *  min(localMAC, peerMAC) || max(localMAC, peerMAC))
    430  1.1.1.1.2.3  snj 	 */
    431  1.1.1.1.2.2  snj 	ptr = context;
    432  1.1.1.1.2.3  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.3  snj 	os_memcpy(ptr, min, WPA_NONCE_LEN);
    440  1.1.1.1.2.3  snj 	ptr += WPA_NONCE_LEN;
    441  1.1.1.1.2.3  snj 	os_memcpy(ptr, max, WPA_NONCE_LEN);
    442  1.1.1.1.2.3  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.3  snj 		WPA_PUT_LE16(ptr, sta->my_lid);
    446  1.1.1.1.2.3  snj 		ptr += 2;
    447  1.1.1.1.2.3  snj 		WPA_PUT_LE16(ptr, sta->peer_lid);
    448  1.1.1.1.2.3  snj 		ptr += 2;
    449  1.1.1.1.2.2  snj 	} else {
    450  1.1.1.1.2.3  snj 		WPA_PUT_LE16(ptr, sta->peer_lid);
    451  1.1.1.1.2.3  snj 		ptr += 2;
    452  1.1.1.1.2.3  snj 		WPA_PUT_LE16(ptr, sta->my_lid);
    453  1.1.1.1.2.3  snj 		ptr += 2;
    454  1.1.1.1.2.2  snj 	}
    455  1.1.1.1.2.3  snj 
    456  1.1.1.1.2.3  snj 	/* Selected AKM Suite: SAE */
    457  1.1.1.1.2.3  snj 	RSN_SELECTOR_PUT(ptr, RSN_AUTH_KEY_MGMT_SAE);
    458  1.1.1.1.2.3  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.3  snj 	ptr += ETH_ALEN;
    469  1.1.1.1.2.3  snj 	os_memcpy(ptr, max, ETH_ALEN);
    470  1.1.1.1.2.2  snj 
    471  1.1.1.1.2.3  snj 	sta->mtk_len = wpa_cipher_key_len(wpa_s->mesh_rsn->pairwise_cipher);
    472  1.1.1.1.2.3  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.3  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.3  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.3  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.3  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.3  snj 	size_t len;
    506  1.1.1.1.2.2  snj 
    507  1.1.1.1.2.3  snj 	len = sizeof(*ampe);
    508  1.1.1.1.2.3  snj 	if (cat[1] == PLINK_OPEN)
    509  1.1.1.1.2.3  snj 		len += rsn->mgtk_len + WPA_KEY_RSC_LEN + 4;
    510  1.1.1.1.2.3  snj #ifdef CONFIG_IEEE80211W
    511  1.1.1.1.2.3  snj 	if (cat[1] == PLINK_OPEN && rsn->igtk_len)
    512  1.1.1.1.2.3  snj 		len += 2 + 6 + rsn->igtk_len;
    513  1.1.1.1.2.3  snj #endif /* CONFIG_IEEE80211W */
    514  1.1.1.1.2.3  snj 
    515  1.1.1.1.2.3  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.3  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.3  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.3  snj 			 RSN_CIPHER_SUITE_CCMP);
    533  1.1.1.1.2.3  snj 	os_memcpy(ampe->local_nonce, sta->my_nonce, WPA_NONCE_LEN);
    534  1.1.1.1.2.3  snj 	os_memcpy(ampe->peer_nonce, sta->peer_nonce, WPA_NONCE_LEN);
    535  1.1.1.1.2.3  snj 
    536  1.1.1.1.2.3  snj 	pos = (u8 *) (ampe + 1);
    537  1.1.1.1.2.3  snj 	if (cat[1] != PLINK_OPEN)
    538  1.1.1.1.2.3  snj 		goto skip_keys;
    539  1.1.1.1.2.3  snj 
    540  1.1.1.1.2.3  snj 	/* TODO: Key Replay Counter[8] optionally for
    541  1.1.1.1.2.3  snj 	 * Mesh Group Key Inform/Acknowledge frames */
    542  1.1.1.1.2.3  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.3  snj 	/*
    545  1.1.1.1.2.3  snj 	 * GTKdata[variable]:
    546  1.1.1.1.2.3  snj 	 * MGTK[variable] || Key RSC[8] || GTKExpirationTime[4]
    547  1.1.1.1.2.3  snj 	 */
    548  1.1.1.1.2.3  snj 	os_memcpy(pos, rsn->mgtk, rsn->mgtk_len);
    549  1.1.1.1.2.3  snj 	pos += rsn->mgtk_len;
    550  1.1.1.1.2.3  snj 	wpa_drv_get_seqnum(rsn->wpa_s, NULL, rsn->mgtk_key_id, pos);
    551  1.1.1.1.2.3  snj 	pos += WPA_KEY_RSC_LEN;
    552  1.1.1.1.2.3  snj 	/* Use fixed GTKExpirationTime for now */
    553  1.1.1.1.2.3  snj 	WPA_PUT_LE32(pos, 0xffffffff);
    554  1.1.1.1.2.3  snj 	pos += 4;
    555  1.1.1.1.2.3  snj 
    556  1.1.1.1.2.3  snj #ifdef CONFIG_IEEE80211W
    557  1.1.1.1.2.3  snj 	/*
    558  1.1.1.1.2.3  snj 	 * IGTKdata[variable]:
    559  1.1.1.1.2.3  snj 	 * Key ID[2], IPN[6], IGTK[variable]
    560  1.1.1.1.2.3  snj 	 */
    561  1.1.1.1.2.3  snj 	if (rsn->igtk_len) {
    562  1.1.1.1.2.3  snj 		WPA_PUT_LE16(pos, rsn->igtk_key_id);
    563  1.1.1.1.2.3  snj 		pos += 2;
    564  1.1.1.1.2.3  snj 		wpa_drv_get_seqnum(rsn->wpa_s, NULL, rsn->igtk_key_id, pos);
    565  1.1.1.1.2.3  snj 		pos += 6;
    566  1.1.1.1.2.3  snj 		os_memcpy(pos, rsn->igtk, rsn->igtk_len);
    567  1.1.1.1.2.3  snj 	}
    568  1.1.1.1.2.3  snj #endif /* CONFIG_IEEE80211W */
    569  1.1.1.1.2.3  snj 
    570  1.1.1.1.2.3  snj skip_keys:
    571  1.1.1.1.2.3  snj 	wpa_hexdump_key(MSG_DEBUG, "mesh: Plaintext AMPE element",
    572  1.1.1.1.2.3  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.3  snj 	wpabuf_put_u8(buf, WLAN_EID_MIC);
    576  1.1.1.1.2.3  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.3  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.3  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.3  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.3  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.3  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.3  snj 	size_t key_len;
    610  1.1.1.1.2.3  snj 
    611  1.1.1.1.2.3  snj 	if (!sta->sae) {
    612  1.1.1.1.2.3  snj 		struct hostapd_data *hapd = wpa_s->ifmsh->bss[0];
    613  1.1.1.1.2.3  snj 
    614  1.1.1.1.2.3  snj 		if (!wpa_auth_pmksa_get(hapd->wpa_auth, sta->addr)) {
    615  1.1.1.1.2.3  snj 			wpa_printf(MSG_INFO,
    616  1.1.1.1.2.3  snj 				   "Mesh RSN: SAE is not prepared yet");
    617  1.1.1.1.2.3  snj 			return -1;
    618  1.1.1.1.2.3  snj 		}
    619  1.1.1.1.2.3  snj 		mesh_rsn_auth_sae_sta(wpa_s, sta);
    620  1.1.1.1.2.3  snj 	}
    621  1.1.1.1.2.3  snj 
    622  1.1.1.1.2.3  snj 	if (chosen_pmk && os_memcmp(chosen_pmk, sta->sae->pmkid, PMKID_LEN)) {
    623  1.1.1.1.2.3  snj 		wpa_msg(wpa_s, MSG_DEBUG,
    624  1.1.1.1.2.3  snj 			"Mesh RSN: Invalid PMKID (Chosen PMK did not match calculated PMKID)");
    625  1.1.1.1.2.3  snj 		return -1;
    626  1.1.1.1.2.3  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.3  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.3  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.3  snj 	crypt_len -= AES_BLOCK_SIZE;
    661  1.1.1.1.2.3  snj 	wpa_hexdump_key(MSG_DEBUG, "mesh: Decrypted AMPE element",
    662  1.1.1.1.2.3  snj 			ampe_buf, crypt_len);
    663  1.1.1.1.2.3  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.3  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.3  snj 	pos = (u8 *) (ampe + 1);
    677  1.1.1.1.2.3  snj 	end = ampe_buf + ampe_ie_len;
    678  1.1.1.1.2.3  snj 	if (os_memcmp(ampe->peer_nonce, null_nonce, WPA_NONCE_LEN) != 0 &&
    679  1.1.1.1.2.3  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.3  snj 	/* TODO: Key Replay Counter[8] in Mesh Group Key Inform/Acknowledge
    688  1.1.1.1.2.3  snj 	 * frames */
    689  1.1.1.1.2.3  snj 
    690  1.1.1.1.2.3  snj 	/*
    691  1.1.1.1.2.3  snj 	 * GTKdata shall not be included in Mesh Peering Confirm. While the
    692  1.1.1.1.2.3  snj 	 * standard does not state the same about IGTKdata, that same constraint
    693  1.1.1.1.2.3  snj 	 * needs to apply for it. It makes no sense to include the keys in Mesh
    694  1.1.1.1.2.3  snj 	 * Peering Close frames either, so while the standard does not seem to
    695  1.1.1.1.2.3  snj 	 * have a shall statement for these, they are described without
    696  1.1.1.1.2.3  snj 	 * mentioning GTKdata.
    697  1.1.1.1.2.3  snj 	 *
    698  1.1.1.1.2.3  snj 	 * An earlier implementation used to add GTKdata to both Mesh Peering
    699  1.1.1.1.2.3  snj 	 * Open and Mesh Peering Confirm frames, so ignore the possibly present
    700  1.1.1.1.2.3  snj 	 * GTKdata frame without rejecting the frame as a backwards
    701  1.1.1.1.2.3  snj 	 * compatibility mechanism.
    702  1.1.1.1.2.3  snj 	 */
    703  1.1.1.1.2.3  snj 	if (cat[1] != PLINK_OPEN) {
    704  1.1.1.1.2.3  snj 		if (end > pos) {
    705  1.1.1.1.2.3  snj 			wpa_hexdump_key(MSG_DEBUG,
    706  1.1.1.1.2.3  snj 					"mesh: Ignore unexpected GTKdata(etc.) fields in the end of AMPE element in Mesh Peering Confirm/Close",
    707  1.1.1.1.2.3  snj 					pos, end - pos);
    708  1.1.1.1.2.3  snj 		}
    709  1.1.1.1.2.3  snj 		goto free;
    710  1.1.1.1.2.3  snj 	}
    711  1.1.1.1.2.3  snj 
    712  1.1.1.1.2.3  snj 	/*
    713  1.1.1.1.2.3  snj 	 * GTKdata[variable]:
    714  1.1.1.1.2.3  snj 	 * MGTK[variable] || Key RSC[8] || GTKExpirationTime[4]
    715  1.1.1.1.2.3  snj 	 */
    716  1.1.1.1.2.3  snj 	sta->mgtk_key_id = 1; /* FIX: Where to get Key ID? */
    717  1.1.1.1.2.3  snj 	key_len = wpa_cipher_key_len(wpa_s->mesh_rsn->group_cipher);
    718  1.1.1.1.2.3  snj 	if ((int) key_len + WPA_KEY_RSC_LEN + 4 > end - pos) {
    719  1.1.1.1.2.3  snj 		wpa_dbg(wpa_s, MSG_DEBUG, "mesh: Truncated AMPE element");
    720  1.1.1.1.2.3  snj 		ret = -1;
    721  1.1.1.1.2.3  snj 		goto free;
    722  1.1.1.1.2.3  snj 	}
    723  1.1.1.1.2.3  snj 	sta->mgtk_len = key_len;
    724  1.1.1.1.2.3  snj 	os_memcpy(sta->mgtk, pos, sta->mgtk_len);
    725  1.1.1.1.2.3  snj 	wpa_hexdump_key(MSG_DEBUG, "mesh: GTKdata - MGTK",
    726  1.1.1.1.2.3  snj 			sta->mgtk, sta->mgtk_len);
    727  1.1.1.1.2.3  snj 	pos += sta->mgtk_len;
    728  1.1.1.1.2.3  snj 	wpa_hexdump(MSG_DEBUG, "mesh: GTKdata - MGTK - Key RSC",
    729  1.1.1.1.2.3  snj 		    pos, WPA_KEY_RSC_LEN);
    730  1.1.1.1.2.3  snj 	os_memcpy(sta->mgtk_rsc, pos, sizeof(sta->mgtk_rsc));
    731  1.1.1.1.2.3  snj 	pos += WPA_KEY_RSC_LEN;
    732  1.1.1.1.2.3  snj 	wpa_printf(MSG_DEBUG,
    733  1.1.1.1.2.3  snj 		   "mesh: GTKdata - MGTK - GTKExpirationTime: %u seconds",
    734  1.1.1.1.2.3  snj 		   WPA_GET_LE32(pos));
    735  1.1.1.1.2.3  snj 	pos += 4;
    736  1.1.1.1.2.3  snj 
    737  1.1.1.1.2.3  snj #ifdef CONFIG_IEEE80211W
    738  1.1.1.1.2.3  snj 	/*
    739  1.1.1.1.2.3  snj 	 * IGTKdata[variable]:
    740  1.1.1.1.2.3  snj 	 * Key ID[2], IPN[6], IGTK[variable]
    741  1.1.1.1.2.3  snj 	 */
    742  1.1.1.1.2.3  snj 	key_len = wpa_cipher_key_len(wpa_s->mesh_rsn->mgmt_group_cipher);
    743  1.1.1.1.2.3  snj 	if (end - pos >= (int) (2 + 6 + key_len)) {
    744  1.1.1.1.2.3  snj 		sta->igtk_key_id = WPA_GET_LE16(pos);
    745  1.1.1.1.2.3  snj 		wpa_printf(MSG_DEBUG, "mesh: IGTKdata - Key ID %u",
    746  1.1.1.1.2.3  snj 			   sta->igtk_key_id);
    747  1.1.1.1.2.3  snj 		pos += 2;
    748  1.1.1.1.2.3  snj 		os_memcpy(sta->igtk_rsc, pos, sizeof(sta->igtk_rsc));
    749  1.1.1.1.2.3  snj 		wpa_hexdump(MSG_DEBUG, "mesh: IGTKdata - IPN",
    750  1.1.1.1.2.3  snj 			    sta->igtk_rsc, sizeof(sta->igtk_rsc));
    751  1.1.1.1.2.3  snj 		pos += 6;
    752  1.1.1.1.2.3  snj 		os_memcpy(sta->igtk, pos, key_len);
    753  1.1.1.1.2.3  snj 		sta->igtk_len = key_len;
    754  1.1.1.1.2.3  snj 		wpa_hexdump_key(MSG_DEBUG, "mesh: IGTKdata - IGTK",
    755  1.1.1.1.2.3  snj 				sta->igtk, sta->igtk_len);
    756  1.1.1.1.2.3  snj 	}
    757  1.1.1.1.2.3  snj #endif /* CONFIG_IEEE80211W */
    758  1.1.1.1.2.3  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