mesh_rsn.c revision 1.1.1.1.2.2
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 #define MESH_AUTH_BLOCK_DURATION 3600 31 1.1.1.1.2.2 snj 32 1.1.1.1.2.2 snj void mesh_auth_timer(void *eloop_ctx, void *user_data) 33 1.1.1.1.2.2 snj { 34 1.1.1.1.2.2 snj struct wpa_supplicant *wpa_s = eloop_ctx; 35 1.1.1.1.2.2 snj struct sta_info *sta = user_data; 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 snj if (sta->sae_auth_retry > MESH_AUTH_RETRY) { 47 1.1.1.1.2.2 snj ap_free_sta(wpa_s->ifmsh->bss[0], sta); 48 1.1.1.1.2.2 snj return; 49 1.1.1.1.2.2 snj } 50 1.1.1.1.2.2 snj 51 1.1.1.1.2.2 snj /* block the STA if exceeded the number of attempts */ 52 1.1.1.1.2.2 snj wpa_mesh_set_plink_state(wpa_s, sta, PLINK_BLOCKED); 53 1.1.1.1.2.2 snj sta->sae->state = SAE_NOTHING; 54 1.1.1.1.2.2 snj if (wpa_s->mesh_auth_block_duration < 55 1.1.1.1.2.2 snj MESH_AUTH_BLOCK_DURATION) 56 1.1.1.1.2.2 snj wpa_s->mesh_auth_block_duration += 60; 57 1.1.1.1.2.2 snj eloop_register_timeout(wpa_s->mesh_auth_block_duration, 58 1.1.1.1.2.2 snj 0, mesh_auth_timer, wpa_s, sta); 59 1.1.1.1.2.2 snj wpa_msg(wpa_s, MSG_INFO, MESH_SAE_AUTH_BLOCKED "addr=" 60 1.1.1.1.2.2 snj MACSTR " duration=%d", 61 1.1.1.1.2.2 snj MAC2STR(sta->addr), 62 1.1.1.1.2.2 snj wpa_s->mesh_auth_block_duration); 63 1.1.1.1.2.2 snj } 64 1.1.1.1.2.2 snj sta->sae_auth_retry++; 65 1.1.1.1.2.2 snj } 66 1.1.1.1.2.2 snj } 67 1.1.1.1.2.2 snj 68 1.1.1.1.2.2 snj 69 1.1.1.1.2.2 snj static void auth_logger(void *ctx, const u8 *addr, logger_level level, 70 1.1.1.1.2.2 snj const char *txt) 71 1.1.1.1.2.2 snj { 72 1.1.1.1.2.2 snj if (addr) 73 1.1.1.1.2.2 snj wpa_printf(MSG_DEBUG, "AUTH: " MACSTR " - %s", 74 1.1.1.1.2.2 snj MAC2STR(addr), txt); 75 1.1.1.1.2.2 snj else 76 1.1.1.1.2.2 snj wpa_printf(MSG_DEBUG, "AUTH: %s", txt); 77 1.1.1.1.2.2 snj } 78 1.1.1.1.2.2 snj 79 1.1.1.1.2.2 snj 80 1.1.1.1.2.2 snj static const u8 *auth_get_psk(void *ctx, const u8 *addr, 81 1.1.1.1.2.2 snj const u8 *p2p_dev_addr, const u8 *prev_psk) 82 1.1.1.1.2.2 snj { 83 1.1.1.1.2.2 snj struct mesh_rsn *mesh_rsn = ctx; 84 1.1.1.1.2.2 snj struct hostapd_data *hapd = mesh_rsn->wpa_s->ifmsh->bss[0]; 85 1.1.1.1.2.2 snj struct sta_info *sta = ap_get_sta(hapd, addr); 86 1.1.1.1.2.2 snj 87 1.1.1.1.2.2 snj wpa_printf(MSG_DEBUG, "AUTH: %s (addr=" MACSTR " prev_psk=%p)", 88 1.1.1.1.2.2 snj __func__, MAC2STR(addr), prev_psk); 89 1.1.1.1.2.2 snj 90 1.1.1.1.2.2 snj if (sta && sta->auth_alg == WLAN_AUTH_SAE) { 91 1.1.1.1.2.2 snj if (!sta->sae || prev_psk) 92 1.1.1.1.2.2 snj return NULL; 93 1.1.1.1.2.2 snj return sta->sae->pmk; 94 1.1.1.1.2.2 snj } 95 1.1.1.1.2.2 snj 96 1.1.1.1.2.2 snj return NULL; 97 1.1.1.1.2.2 snj } 98 1.1.1.1.2.2 snj 99 1.1.1.1.2.2 snj 100 1.1.1.1.2.2 snj static int auth_set_key(void *ctx, int vlan_id, enum wpa_alg alg, 101 1.1.1.1.2.2 snj const u8 *addr, int idx, u8 *key, size_t key_len) 102 1.1.1.1.2.2 snj { 103 1.1.1.1.2.2 snj struct mesh_rsn *mesh_rsn = ctx; 104 1.1.1.1.2.2 snj u8 seq[6]; 105 1.1.1.1.2.2 snj 106 1.1.1.1.2.2 snj os_memset(seq, 0, sizeof(seq)); 107 1.1.1.1.2.2 snj 108 1.1.1.1.2.2 snj if (addr) { 109 1.1.1.1.2.2 snj wpa_printf(MSG_DEBUG, "AUTH: %s(alg=%d addr=" MACSTR 110 1.1.1.1.2.2 snj " key_idx=%d)", 111 1.1.1.1.2.2 snj __func__, alg, MAC2STR(addr), idx); 112 1.1.1.1.2.2 snj } else { 113 1.1.1.1.2.2 snj wpa_printf(MSG_DEBUG, "AUTH: %s(alg=%d key_idx=%d)", 114 1.1.1.1.2.2 snj __func__, alg, idx); 115 1.1.1.1.2.2 snj } 116 1.1.1.1.2.2 snj wpa_hexdump_key(MSG_DEBUG, "AUTH: set_key - key", key, key_len); 117 1.1.1.1.2.2 snj 118 1.1.1.1.2.2 snj return wpa_drv_set_key(mesh_rsn->wpa_s, alg, addr, idx, 119 1.1.1.1.2.2 snj 1, seq, 6, key, key_len); 120 1.1.1.1.2.2 snj } 121 1.1.1.1.2.2 snj 122 1.1.1.1.2.2 snj 123 1.1.1.1.2.2 snj static int auth_start_ampe(void *ctx, const u8 *addr) 124 1.1.1.1.2.2 snj { 125 1.1.1.1.2.2 snj struct mesh_rsn *mesh_rsn = ctx; 126 1.1.1.1.2.2 snj struct hostapd_data *hapd; 127 1.1.1.1.2.2 snj struct sta_info *sta; 128 1.1.1.1.2.2 snj 129 1.1.1.1.2.2 snj if (mesh_rsn->wpa_s->current_ssid->mode != WPAS_MODE_MESH) 130 1.1.1.1.2.2 snj return -1; 131 1.1.1.1.2.2 snj 132 1.1.1.1.2.2 snj hapd = mesh_rsn->wpa_s->ifmsh->bss[0]; 133 1.1.1.1.2.2 snj sta = ap_get_sta(hapd, addr); 134 1.1.1.1.2.2 snj if (sta) 135 1.1.1.1.2.2 snj eloop_cancel_timeout(mesh_auth_timer, mesh_rsn->wpa_s, sta); 136 1.1.1.1.2.2 snj 137 1.1.1.1.2.2 snj mesh_mpm_auth_peer(mesh_rsn->wpa_s, addr); 138 1.1.1.1.2.2 snj return 0; 139 1.1.1.1.2.2 snj } 140 1.1.1.1.2.2 snj 141 1.1.1.1.2.2 snj 142 1.1.1.1.2.2 snj static int __mesh_rsn_auth_init(struct mesh_rsn *rsn, const u8 *addr) 143 1.1.1.1.2.2 snj { 144 1.1.1.1.2.2 snj struct wpa_auth_config conf; 145 1.1.1.1.2.2 snj struct wpa_auth_callbacks cb; 146 1.1.1.1.2.2 snj u8 seq[6] = {}; 147 1.1.1.1.2.2 snj 148 1.1.1.1.2.2 snj wpa_printf(MSG_DEBUG, "AUTH: Initializing group state machine"); 149 1.1.1.1.2.2 snj 150 1.1.1.1.2.2 snj os_memset(&conf, 0, sizeof(conf)); 151 1.1.1.1.2.2 snj conf.wpa = 2; 152 1.1.1.1.2.2 snj conf.wpa_key_mgmt = WPA_KEY_MGMT_SAE; 153 1.1.1.1.2.2 snj conf.wpa_pairwise = WPA_CIPHER_CCMP; 154 1.1.1.1.2.2 snj conf.rsn_pairwise = WPA_CIPHER_CCMP; 155 1.1.1.1.2.2 snj conf.wpa_group = WPA_CIPHER_CCMP; 156 1.1.1.1.2.2 snj conf.eapol_version = 0; 157 1.1.1.1.2.2 snj conf.wpa_group_rekey = -1; 158 1.1.1.1.2.2 snj 159 1.1.1.1.2.2 snj os_memset(&cb, 0, sizeof(cb)); 160 1.1.1.1.2.2 snj cb.ctx = rsn; 161 1.1.1.1.2.2 snj cb.logger = auth_logger; 162 1.1.1.1.2.2 snj cb.get_psk = auth_get_psk; 163 1.1.1.1.2.2 snj cb.set_key = auth_set_key; 164 1.1.1.1.2.2 snj cb.start_ampe = auth_start_ampe; 165 1.1.1.1.2.2 snj 166 1.1.1.1.2.2 snj rsn->auth = wpa_init(addr, &conf, &cb); 167 1.1.1.1.2.2 snj if (rsn->auth == NULL) { 168 1.1.1.1.2.2 snj wpa_printf(MSG_DEBUG, "AUTH: wpa_init() failed"); 169 1.1.1.1.2.2 snj return -1; 170 1.1.1.1.2.2 snj } 171 1.1.1.1.2.2 snj 172 1.1.1.1.2.2 snj /* TODO: support rekeying */ 173 1.1.1.1.2.2 snj if (random_get_bytes(rsn->mgtk, 16) < 0) { 174 1.1.1.1.2.2 snj wpa_deinit(rsn->auth); 175 1.1.1.1.2.2 snj return -1; 176 1.1.1.1.2.2 snj } 177 1.1.1.1.2.2 snj 178 1.1.1.1.2.2 snj /* group mgmt */ 179 1.1.1.1.2.2 snj wpa_drv_set_key(rsn->wpa_s, WPA_ALG_IGTK, NULL, 4, 1, 180 1.1.1.1.2.2 snj seq, sizeof(seq), rsn->mgtk, sizeof(rsn->mgtk)); 181 1.1.1.1.2.2 snj 182 1.1.1.1.2.2 snj /* group privacy / data frames */ 183 1.1.1.1.2.2 snj wpa_drv_set_key(rsn->wpa_s, WPA_ALG_CCMP, NULL, 1, 1, 184 1.1.1.1.2.2 snj seq, sizeof(seq), rsn->mgtk, sizeof(rsn->mgtk)); 185 1.1.1.1.2.2 snj 186 1.1.1.1.2.2 snj return 0; 187 1.1.1.1.2.2 snj } 188 1.1.1.1.2.2 snj 189 1.1.1.1.2.2 snj 190 1.1.1.1.2.2 snj static void mesh_rsn_deinit(struct mesh_rsn *rsn) 191 1.1.1.1.2.2 snj { 192 1.1.1.1.2.2 snj os_memset(rsn->mgtk, 0, sizeof(rsn->mgtk)); 193 1.1.1.1.2.2 snj wpa_deinit(rsn->auth); 194 1.1.1.1.2.2 snj } 195 1.1.1.1.2.2 snj 196 1.1.1.1.2.2 snj 197 1.1.1.1.2.2 snj struct mesh_rsn *mesh_rsn_auth_init(struct wpa_supplicant *wpa_s, 198 1.1.1.1.2.2 snj struct mesh_conf *conf) 199 1.1.1.1.2.2 snj { 200 1.1.1.1.2.2 snj struct mesh_rsn *mesh_rsn; 201 1.1.1.1.2.2 snj struct hostapd_data *bss = wpa_s->ifmsh->bss[0]; 202 1.1.1.1.2.2 snj const u8 *ie; 203 1.1.1.1.2.2 snj size_t ie_len; 204 1.1.1.1.2.2 snj 205 1.1.1.1.2.2 snj mesh_rsn = os_zalloc(sizeof(*mesh_rsn)); 206 1.1.1.1.2.2 snj if (mesh_rsn == NULL) 207 1.1.1.1.2.2 snj return NULL; 208 1.1.1.1.2.2 snj mesh_rsn->wpa_s = wpa_s; 209 1.1.1.1.2.2 snj 210 1.1.1.1.2.2 snj if (__mesh_rsn_auth_init(mesh_rsn, wpa_s->own_addr) < 0) { 211 1.1.1.1.2.2 snj mesh_rsn_deinit(mesh_rsn); 212 1.1.1.1.2.2 snj return NULL; 213 1.1.1.1.2.2 snj } 214 1.1.1.1.2.2 snj 215 1.1.1.1.2.2 snj bss->wpa_auth = mesh_rsn->auth; 216 1.1.1.1.2.2 snj 217 1.1.1.1.2.2 snj ie = wpa_auth_get_wpa_ie(mesh_rsn->auth, &ie_len); 218 1.1.1.1.2.2 snj conf->ies = (u8 *) ie; 219 1.1.1.1.2.2 snj conf->ie_len = ie_len; 220 1.1.1.1.2.2 snj 221 1.1.1.1.2.2 snj wpa_supplicant_rsn_supp_set_config(wpa_s, wpa_s->current_ssid); 222 1.1.1.1.2.2 snj 223 1.1.1.1.2.2 snj return mesh_rsn; 224 1.1.1.1.2.2 snj } 225 1.1.1.1.2.2 snj 226 1.1.1.1.2.2 snj 227 1.1.1.1.2.2 snj static int index_within_array(const int *array, int idx) 228 1.1.1.1.2.2 snj { 229 1.1.1.1.2.2 snj int i; 230 1.1.1.1.2.2 snj 231 1.1.1.1.2.2 snj for (i = 0; i < idx; i++) { 232 1.1.1.1.2.2 snj if (array[i] == -1) 233 1.1.1.1.2.2 snj return 0; 234 1.1.1.1.2.2 snj } 235 1.1.1.1.2.2 snj 236 1.1.1.1.2.2 snj return 1; 237 1.1.1.1.2.2 snj } 238 1.1.1.1.2.2 snj 239 1.1.1.1.2.2 snj 240 1.1.1.1.2.2 snj static int mesh_rsn_sae_group(struct wpa_supplicant *wpa_s, 241 1.1.1.1.2.2 snj struct sae_data *sae) 242 1.1.1.1.2.2 snj { 243 1.1.1.1.2.2 snj int *groups = wpa_s->ifmsh->bss[0]->conf->sae_groups; 244 1.1.1.1.2.2 snj 245 1.1.1.1.2.2 snj /* Configuration may have changed, so validate current index */ 246 1.1.1.1.2.2 snj if (!index_within_array(groups, wpa_s->mesh_rsn->sae_group_index)) 247 1.1.1.1.2.2 snj return -1; 248 1.1.1.1.2.2 snj 249 1.1.1.1.2.2 snj for (;;) { 250 1.1.1.1.2.2 snj int group = groups[wpa_s->mesh_rsn->sae_group_index]; 251 1.1.1.1.2.2 snj 252 1.1.1.1.2.2 snj if (group <= 0) 253 1.1.1.1.2.2 snj break; 254 1.1.1.1.2.2 snj if (sae_set_group(sae, group) == 0) { 255 1.1.1.1.2.2 snj wpa_dbg(wpa_s, MSG_DEBUG, "SME: Selected SAE group %d", 256 1.1.1.1.2.2 snj sae->group); 257 1.1.1.1.2.2 snj return 0; 258 1.1.1.1.2.2 snj } 259 1.1.1.1.2.2 snj wpa_s->mesh_rsn->sae_group_index++; 260 1.1.1.1.2.2 snj } 261 1.1.1.1.2.2 snj 262 1.1.1.1.2.2 snj return -1; 263 1.1.1.1.2.2 snj } 264 1.1.1.1.2.2 snj 265 1.1.1.1.2.2 snj 266 1.1.1.1.2.2 snj static int mesh_rsn_build_sae_commit(struct wpa_supplicant *wpa_s, 267 1.1.1.1.2.2 snj struct wpa_ssid *ssid, 268 1.1.1.1.2.2 snj struct sta_info *sta) 269 1.1.1.1.2.2 snj { 270 1.1.1.1.2.2 snj if (ssid->passphrase == NULL) { 271 1.1.1.1.2.2 snj wpa_msg(wpa_s, MSG_DEBUG, "SAE: No password available"); 272 1.1.1.1.2.2 snj return -1; 273 1.1.1.1.2.2 snj } 274 1.1.1.1.2.2 snj 275 1.1.1.1.2.2 snj if (mesh_rsn_sae_group(wpa_s, sta->sae) < 0) { 276 1.1.1.1.2.2 snj wpa_msg(wpa_s, MSG_DEBUG, "SAE: Failed to select group"); 277 1.1.1.1.2.2 snj return -1; 278 1.1.1.1.2.2 snj } 279 1.1.1.1.2.2 snj 280 1.1.1.1.2.2 snj return sae_prepare_commit(wpa_s->own_addr, sta->addr, 281 1.1.1.1.2.2 snj (u8 *) ssid->passphrase, 282 1.1.1.1.2.2 snj os_strlen(ssid->passphrase), sta->sae); 283 1.1.1.1.2.2 snj } 284 1.1.1.1.2.2 snj 285 1.1.1.1.2.2 snj 286 1.1.1.1.2.2 snj /* initiate new SAE authentication with sta */ 287 1.1.1.1.2.2 snj int mesh_rsn_auth_sae_sta(struct wpa_supplicant *wpa_s, 288 1.1.1.1.2.2 snj struct sta_info *sta) 289 1.1.1.1.2.2 snj { 290 1.1.1.1.2.2 snj struct hostapd_data *hapd = wpa_s->ifmsh->bss[0]; 291 1.1.1.1.2.2 snj struct wpa_ssid *ssid = wpa_s->current_ssid; 292 1.1.1.1.2.2 snj unsigned int rnd; 293 1.1.1.1.2.2 snj int ret; 294 1.1.1.1.2.2 snj 295 1.1.1.1.2.2 snj if (!ssid) { 296 1.1.1.1.2.2 snj wpa_msg(wpa_s, MSG_DEBUG, 297 1.1.1.1.2.2 snj "AUTH: No current_ssid known to initiate new SAE"); 298 1.1.1.1.2.2 snj return -1; 299 1.1.1.1.2.2 snj } 300 1.1.1.1.2.2 snj 301 1.1.1.1.2.2 snj if (!sta->sae) { 302 1.1.1.1.2.2 snj sta->sae = os_zalloc(sizeof(*sta->sae)); 303 1.1.1.1.2.2 snj if (sta->sae == NULL) 304 1.1.1.1.2.2 snj return -1; 305 1.1.1.1.2.2 snj } 306 1.1.1.1.2.2 snj 307 1.1.1.1.2.2 snj if (mesh_rsn_build_sae_commit(wpa_s, ssid, sta)) 308 1.1.1.1.2.2 snj return -1; 309 1.1.1.1.2.2 snj 310 1.1.1.1.2.2 snj wpa_msg(wpa_s, MSG_DEBUG, 311 1.1.1.1.2.2 snj "AUTH: started authentication with SAE peer: " MACSTR, 312 1.1.1.1.2.2 snj MAC2STR(sta->addr)); 313 1.1.1.1.2.2 snj 314 1.1.1.1.2.2 snj wpa_supplicant_set_state(wpa_s, WPA_AUTHENTICATING); 315 1.1.1.1.2.2 snj ret = auth_sae_init_committed(hapd, sta); 316 1.1.1.1.2.2 snj if (ret) 317 1.1.1.1.2.2 snj return ret; 318 1.1.1.1.2.2 snj 319 1.1.1.1.2.2 snj eloop_cancel_timeout(mesh_auth_timer, wpa_s, sta); 320 1.1.1.1.2.2 snj rnd = rand() % MESH_AUTH_TIMEOUT; 321 1.1.1.1.2.2 snj eloop_register_timeout(MESH_AUTH_TIMEOUT + rnd, 0, mesh_auth_timer, 322 1.1.1.1.2.2 snj wpa_s, sta); 323 1.1.1.1.2.2 snj return 0; 324 1.1.1.1.2.2 snj } 325 1.1.1.1.2.2 snj 326 1.1.1.1.2.2 snj 327 1.1.1.1.2.2 snj void mesh_rsn_get_pmkid(struct mesh_rsn *rsn, struct sta_info *sta, u8 *pmkid) 328 1.1.1.1.2.2 snj { 329 1.1.1.1.2.2 snj /* don't expect wpa auth to cache the pmkid for now */ 330 1.1.1.1.2.2 snj rsn_pmkid(sta->sae->pmk, PMK_LEN, rsn->wpa_s->own_addr, 331 1.1.1.1.2.2 snj sta->addr, pmkid, 332 1.1.1.1.2.2 snj wpa_key_mgmt_sha256(wpa_auth_sta_key_mgmt(sta->wpa_sm))); 333 1.1.1.1.2.2 snj } 334 1.1.1.1.2.2 snj 335 1.1.1.1.2.2 snj 336 1.1.1.1.2.2 snj static void 337 1.1.1.1.2.2 snj mesh_rsn_derive_aek(struct mesh_rsn *rsn, struct sta_info *sta) 338 1.1.1.1.2.2 snj { 339 1.1.1.1.2.2 snj u8 *myaddr = rsn->wpa_s->own_addr; 340 1.1.1.1.2.2 snj u8 *peer = sta->addr; 341 1.1.1.1.2.2 snj u8 *addr1 = peer, *addr2 = myaddr; 342 1.1.1.1.2.2 snj u8 context[AES_BLOCK_SIZE]; 343 1.1.1.1.2.2 snj 344 1.1.1.1.2.2 snj /* SAE */ 345 1.1.1.1.2.2 snj RSN_SELECTOR_PUT(context, wpa_cipher_to_suite(0, WPA_CIPHER_GCMP)); 346 1.1.1.1.2.2 snj 347 1.1.1.1.2.2 snj if (os_memcmp(myaddr, peer, ETH_ALEN) < 0) { 348 1.1.1.1.2.2 snj addr1 = myaddr; 349 1.1.1.1.2.2 snj addr2 = peer; 350 1.1.1.1.2.2 snj } 351 1.1.1.1.2.2 snj os_memcpy(context + 4, addr1, ETH_ALEN); 352 1.1.1.1.2.2 snj os_memcpy(context + 10, addr2, ETH_ALEN); 353 1.1.1.1.2.2 snj 354 1.1.1.1.2.2 snj sha256_prf(sta->sae->pmk, sizeof(sta->sae->pmk), "AEK Derivation", 355 1.1.1.1.2.2 snj context, sizeof(context), sta->aek, sizeof(sta->aek)); 356 1.1.1.1.2.2 snj } 357 1.1.1.1.2.2 snj 358 1.1.1.1.2.2 snj 359 1.1.1.1.2.2 snj /* derive mesh temporal key from pmk */ 360 1.1.1.1.2.2 snj int mesh_rsn_derive_mtk(struct wpa_supplicant *wpa_s, struct sta_info *sta) 361 1.1.1.1.2.2 snj { 362 1.1.1.1.2.2 snj u8 *ptr; 363 1.1.1.1.2.2 snj u8 *min, *max; 364 1.1.1.1.2.2 snj u16 min_lid, max_lid; 365 1.1.1.1.2.2 snj size_t nonce_len = sizeof(sta->my_nonce); 366 1.1.1.1.2.2 snj size_t lid_len = sizeof(sta->my_lid); 367 1.1.1.1.2.2 snj u8 *myaddr = wpa_s->own_addr; 368 1.1.1.1.2.2 snj u8 *peer = sta->addr; 369 1.1.1.1.2.2 snj /* 2 nonces, 2 linkids, akm suite, 2 mac addrs */ 370 1.1.1.1.2.2 snj u8 context[64 + 4 + 4 + 12]; 371 1.1.1.1.2.2 snj 372 1.1.1.1.2.2 snj ptr = context; 373 1.1.1.1.2.2 snj if (os_memcmp(sta->my_nonce, sta->peer_nonce, nonce_len) < 0) { 374 1.1.1.1.2.2 snj min = sta->my_nonce; 375 1.1.1.1.2.2 snj max = sta->peer_nonce; 376 1.1.1.1.2.2 snj } else { 377 1.1.1.1.2.2 snj min = sta->peer_nonce; 378 1.1.1.1.2.2 snj max = sta->my_nonce; 379 1.1.1.1.2.2 snj } 380 1.1.1.1.2.2 snj os_memcpy(ptr, min, nonce_len); 381 1.1.1.1.2.2 snj os_memcpy(ptr + nonce_len, max, nonce_len); 382 1.1.1.1.2.2 snj ptr += 2 * nonce_len; 383 1.1.1.1.2.2 snj 384 1.1.1.1.2.2 snj if (sta->my_lid < sta->peer_lid) { 385 1.1.1.1.2.2 snj min_lid = host_to_le16(sta->my_lid); 386 1.1.1.1.2.2 snj max_lid = host_to_le16(sta->peer_lid); 387 1.1.1.1.2.2 snj } else { 388 1.1.1.1.2.2 snj min_lid = host_to_le16(sta->peer_lid); 389 1.1.1.1.2.2 snj max_lid = host_to_le16(sta->my_lid); 390 1.1.1.1.2.2 snj } 391 1.1.1.1.2.2 snj os_memcpy(ptr, &min_lid, lid_len); 392 1.1.1.1.2.2 snj os_memcpy(ptr + lid_len, &max_lid, lid_len); 393 1.1.1.1.2.2 snj ptr += 2 * lid_len; 394 1.1.1.1.2.2 snj 395 1.1.1.1.2.2 snj /* SAE */ 396 1.1.1.1.2.2 snj RSN_SELECTOR_PUT(ptr, wpa_cipher_to_suite(0, WPA_CIPHER_GCMP)); 397 1.1.1.1.2.2 snj ptr += 4; 398 1.1.1.1.2.2 snj 399 1.1.1.1.2.2 snj if (os_memcmp(myaddr, peer, ETH_ALEN) < 0) { 400 1.1.1.1.2.2 snj min = myaddr; 401 1.1.1.1.2.2 snj max = peer; 402 1.1.1.1.2.2 snj } else { 403 1.1.1.1.2.2 snj min = peer; 404 1.1.1.1.2.2 snj max = myaddr; 405 1.1.1.1.2.2 snj } 406 1.1.1.1.2.2 snj os_memcpy(ptr, min, ETH_ALEN); 407 1.1.1.1.2.2 snj os_memcpy(ptr + ETH_ALEN, max, ETH_ALEN); 408 1.1.1.1.2.2 snj 409 1.1.1.1.2.2 snj sha256_prf(sta->sae->pmk, sizeof(sta->sae->pmk), 410 1.1.1.1.2.2 snj "Temporal Key Derivation", context, sizeof(context), 411 1.1.1.1.2.2 snj sta->mtk, sizeof(sta->mtk)); 412 1.1.1.1.2.2 snj return 0; 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 void mesh_rsn_init_ampe_sta(struct wpa_supplicant *wpa_s, struct sta_info *sta) 417 1.1.1.1.2.2 snj { 418 1.1.1.1.2.2 snj if (random_get_bytes(sta->my_nonce, 32) < 0) { 419 1.1.1.1.2.2 snj wpa_printf(MSG_INFO, "mesh: Failed to derive random nonce"); 420 1.1.1.1.2.2 snj /* TODO: How to handle this more cleanly? */ 421 1.1.1.1.2.2 snj } 422 1.1.1.1.2.2 snj os_memset(sta->peer_nonce, 0, 32); 423 1.1.1.1.2.2 snj mesh_rsn_derive_aek(wpa_s->mesh_rsn, sta); 424 1.1.1.1.2.2 snj } 425 1.1.1.1.2.2 snj 426 1.1.1.1.2.2 snj 427 1.1.1.1.2.2 snj /* insert AMPE and encrypted MIC at @ie. 428 1.1.1.1.2.2 snj * @mesh_rsn: mesh RSN context 429 1.1.1.1.2.2 snj * @sta: STA we're sending to 430 1.1.1.1.2.2 snj * @cat: pointer to category code in frame header. 431 1.1.1.1.2.2 snj * @buf: wpabuf to add encrypted AMPE and MIC to. 432 1.1.1.1.2.2 snj * */ 433 1.1.1.1.2.2 snj int mesh_rsn_protect_frame(struct mesh_rsn *rsn, struct sta_info *sta, 434 1.1.1.1.2.2 snj const u8 *cat, struct wpabuf *buf) 435 1.1.1.1.2.2 snj { 436 1.1.1.1.2.2 snj struct ieee80211_ampe_ie *ampe; 437 1.1.1.1.2.2 snj u8 const *ie = wpabuf_head_u8(buf) + wpabuf_len(buf); 438 1.1.1.1.2.2 snj u8 *ampe_ie = NULL, *mic_ie = NULL, *mic_payload; 439 1.1.1.1.2.2 snj const u8 *aad[] = { rsn->wpa_s->own_addr, sta->addr, cat }; 440 1.1.1.1.2.2 snj const size_t aad_len[] = { ETH_ALEN, ETH_ALEN, ie - cat }; 441 1.1.1.1.2.2 snj int ret = 0; 442 1.1.1.1.2.2 snj 443 1.1.1.1.2.2 snj if (AES_BLOCK_SIZE + 2 + sizeof(*ampe) + 2 > wpabuf_tailroom(buf)) { 444 1.1.1.1.2.2 snj wpa_printf(MSG_ERROR, "protect frame: buffer too small"); 445 1.1.1.1.2.2 snj return -EINVAL; 446 1.1.1.1.2.2 snj } 447 1.1.1.1.2.2 snj 448 1.1.1.1.2.2 snj ampe_ie = os_zalloc(2 + sizeof(*ampe)); 449 1.1.1.1.2.2 snj if (!ampe_ie) { 450 1.1.1.1.2.2 snj wpa_printf(MSG_ERROR, "protect frame: out of memory"); 451 1.1.1.1.2.2 snj return -ENOMEM; 452 1.1.1.1.2.2 snj } 453 1.1.1.1.2.2 snj 454 1.1.1.1.2.2 snj mic_ie = os_zalloc(2 + AES_BLOCK_SIZE); 455 1.1.1.1.2.2 snj if (!mic_ie) { 456 1.1.1.1.2.2 snj wpa_printf(MSG_ERROR, "protect frame: out of memory"); 457 1.1.1.1.2.2 snj ret = -ENOMEM; 458 1.1.1.1.2.2 snj goto free; 459 1.1.1.1.2.2 snj } 460 1.1.1.1.2.2 snj 461 1.1.1.1.2.2 snj /* IE: AMPE */ 462 1.1.1.1.2.2 snj ampe_ie[0] = WLAN_EID_AMPE; 463 1.1.1.1.2.2 snj ampe_ie[1] = sizeof(*ampe); 464 1.1.1.1.2.2 snj ampe = (struct ieee80211_ampe_ie *) (ampe_ie + 2); 465 1.1.1.1.2.2 snj 466 1.1.1.1.2.2 snj RSN_SELECTOR_PUT(ampe->selected_pairwise_suite, 467 1.1.1.1.2.2 snj wpa_cipher_to_suite(WPA_PROTO_RSN, WPA_CIPHER_CCMP)); 468 1.1.1.1.2.2 snj os_memcpy(ampe->local_nonce, sta->my_nonce, 32); 469 1.1.1.1.2.2 snj os_memcpy(ampe->peer_nonce, sta->peer_nonce, 32); 470 1.1.1.1.2.2 snj /* incomplete: see 13.5.4 */ 471 1.1.1.1.2.2 snj /* TODO: static mgtk for now since we don't support rekeying! */ 472 1.1.1.1.2.2 snj os_memcpy(ampe->mgtk, rsn->mgtk, 16); 473 1.1.1.1.2.2 snj /* TODO: Populate Key RSC */ 474 1.1.1.1.2.2 snj /* expire in 13 decades or so */ 475 1.1.1.1.2.2 snj os_memset(ampe->key_expiration, 0xff, 4); 476 1.1.1.1.2.2 snj 477 1.1.1.1.2.2 snj /* IE: MIC */ 478 1.1.1.1.2.2 snj mic_ie[0] = WLAN_EID_MIC; 479 1.1.1.1.2.2 snj mic_ie[1] = AES_BLOCK_SIZE; 480 1.1.1.1.2.2 snj wpabuf_put_data(buf, mic_ie, 2); 481 1.1.1.1.2.2 snj /* MIC field is output ciphertext */ 482 1.1.1.1.2.2 snj 483 1.1.1.1.2.2 snj /* encrypt after MIC */ 484 1.1.1.1.2.2 snj mic_payload = (u8 *) wpabuf_put(buf, 2 + sizeof(*ampe) + 485 1.1.1.1.2.2 snj AES_BLOCK_SIZE); 486 1.1.1.1.2.2 snj 487 1.1.1.1.2.2 snj if (aes_siv_encrypt(sta->aek, ampe_ie, 2 + sizeof(*ampe), 3, 488 1.1.1.1.2.2 snj aad, aad_len, mic_payload)) { 489 1.1.1.1.2.2 snj wpa_printf(MSG_ERROR, "protect frame: failed to encrypt"); 490 1.1.1.1.2.2 snj ret = -ENOMEM; 491 1.1.1.1.2.2 snj goto free; 492 1.1.1.1.2.2 snj } 493 1.1.1.1.2.2 snj 494 1.1.1.1.2.2 snj free: 495 1.1.1.1.2.2 snj os_free(ampe_ie); 496 1.1.1.1.2.2 snj os_free(mic_ie); 497 1.1.1.1.2.2 snj 498 1.1.1.1.2.2 snj return ret; 499 1.1.1.1.2.2 snj } 500 1.1.1.1.2.2 snj 501 1.1.1.1.2.2 snj 502 1.1.1.1.2.2 snj int mesh_rsn_process_ampe(struct wpa_supplicant *wpa_s, struct sta_info *sta, 503 1.1.1.1.2.2 snj struct ieee802_11_elems *elems, const u8 *cat, 504 1.1.1.1.2.2 snj const u8 *start, size_t elems_len) 505 1.1.1.1.2.2 snj { 506 1.1.1.1.2.2 snj int ret = 0; 507 1.1.1.1.2.2 snj struct ieee80211_ampe_ie *ampe; 508 1.1.1.1.2.2 snj u8 null_nonce[32] = {}; 509 1.1.1.1.2.2 snj u8 ampe_eid; 510 1.1.1.1.2.2 snj u8 ampe_ie_len; 511 1.1.1.1.2.2 snj u8 *ampe_buf, *crypt = NULL; 512 1.1.1.1.2.2 snj size_t crypt_len; 513 1.1.1.1.2.2 snj const u8 *aad[] = { sta->addr, wpa_s->own_addr, cat }; 514 1.1.1.1.2.2 snj const size_t aad_len[] = { ETH_ALEN, ETH_ALEN, 515 1.1.1.1.2.2 snj (elems->mic - 2) - cat }; 516 1.1.1.1.2.2 snj 517 1.1.1.1.2.2 snj if (!elems->mic || elems->mic_len < AES_BLOCK_SIZE) { 518 1.1.1.1.2.2 snj wpa_msg(wpa_s, MSG_DEBUG, "Mesh RSN: missing mic ie"); 519 1.1.1.1.2.2 snj return -1; 520 1.1.1.1.2.2 snj } 521 1.1.1.1.2.2 snj 522 1.1.1.1.2.2 snj ampe_buf = (u8 *) elems->mic + elems->mic_len; 523 1.1.1.1.2.2 snj if ((int) elems_len < ampe_buf - start) 524 1.1.1.1.2.2 snj return -1; 525 1.1.1.1.2.2 snj 526 1.1.1.1.2.2 snj crypt_len = elems_len - (elems->mic - start); 527 1.1.1.1.2.2 snj if (crypt_len < 2) { 528 1.1.1.1.2.2 snj wpa_msg(wpa_s, MSG_DEBUG, "Mesh RSN: missing ampe ie"); 529 1.1.1.1.2.2 snj return -1; 530 1.1.1.1.2.2 snj } 531 1.1.1.1.2.2 snj 532 1.1.1.1.2.2 snj /* crypt is modified by siv_decrypt */ 533 1.1.1.1.2.2 snj crypt = os_zalloc(crypt_len); 534 1.1.1.1.2.2 snj if (!crypt) { 535 1.1.1.1.2.2 snj wpa_printf(MSG_ERROR, "Mesh RSN: out of memory"); 536 1.1.1.1.2.2 snj ret = -ENOMEM; 537 1.1.1.1.2.2 snj goto free; 538 1.1.1.1.2.2 snj } 539 1.1.1.1.2.2 snj 540 1.1.1.1.2.2 snj os_memcpy(crypt, elems->mic, crypt_len); 541 1.1.1.1.2.2 snj 542 1.1.1.1.2.2 snj if (aes_siv_decrypt(sta->aek, crypt, crypt_len, 3, 543 1.1.1.1.2.2 snj aad, aad_len, ampe_buf)) { 544 1.1.1.1.2.2 snj wpa_printf(MSG_ERROR, "Mesh RSN: frame verification failed!"); 545 1.1.1.1.2.2 snj ret = -1; 546 1.1.1.1.2.2 snj goto free; 547 1.1.1.1.2.2 snj } 548 1.1.1.1.2.2 snj 549 1.1.1.1.2.2 snj ampe_eid = *ampe_buf++; 550 1.1.1.1.2.2 snj ampe_ie_len = *ampe_buf++; 551 1.1.1.1.2.2 snj 552 1.1.1.1.2.2 snj if (ampe_eid != WLAN_EID_AMPE || 553 1.1.1.1.2.2 snj ampe_ie_len < sizeof(struct ieee80211_ampe_ie)) { 554 1.1.1.1.2.2 snj wpa_msg(wpa_s, MSG_DEBUG, "Mesh RSN: invalid ampe ie"); 555 1.1.1.1.2.2 snj ret = -1; 556 1.1.1.1.2.2 snj goto free; 557 1.1.1.1.2.2 snj } 558 1.1.1.1.2.2 snj 559 1.1.1.1.2.2 snj ampe = (struct ieee80211_ampe_ie *) ampe_buf; 560 1.1.1.1.2.2 snj if (os_memcmp(ampe->peer_nonce, null_nonce, 32) != 0 && 561 1.1.1.1.2.2 snj os_memcmp(ampe->peer_nonce, sta->my_nonce, 32) != 0) { 562 1.1.1.1.2.2 snj wpa_msg(wpa_s, MSG_DEBUG, "Mesh RSN: invalid peer nonce"); 563 1.1.1.1.2.2 snj ret = -1; 564 1.1.1.1.2.2 snj goto free; 565 1.1.1.1.2.2 snj } 566 1.1.1.1.2.2 snj os_memcpy(sta->peer_nonce, ampe->local_nonce, 567 1.1.1.1.2.2 snj sizeof(ampe->local_nonce)); 568 1.1.1.1.2.2 snj os_memcpy(sta->mgtk, ampe->mgtk, sizeof(ampe->mgtk)); 569 1.1.1.1.2.2 snj 570 1.1.1.1.2.2 snj /* todo parse mgtk expiration */ 571 1.1.1.1.2.2 snj free: 572 1.1.1.1.2.2 snj os_free(crypt); 573 1.1.1.1.2.2 snj return ret; 574 1.1.1.1.2.2 snj } 575
Indexes created Mon Mar 30 00:23:01 UTC 2026