contrib/wpa/src/ap/wpa_auth.c

   1 /*
   2  * IEEE 802.11 RSN / WPA Authenticator
   3  * Copyright (c) 2004-2019, Jouni Malinen <j@w1.fi>
   4  *
   5  * This software may be distributed under the terms of the BSD license.
   6  * See README for more details.
   7  */
   8
   9 #include "utils/includes.h"
  10
  11 #include "utils/common.h"
  12 #include "utils/eloop.h"
  13 #include "utils/state_machine.h"
  14 #include "utils/bitfield.h"
  15 #include "common/ieee802_11_defs.h"
  16 #include "common/ocv.h"
  17 #include "crypto/aes.h"
  18 #include "crypto/aes_wrap.h"
  19 #include "crypto/aes_siv.h"
  20 #include "crypto/crypto.h"
  21 #include "crypto/sha1.h"
  22 #include "crypto/sha256.h"
  23 #include "crypto/sha384.h"
  24 #include "crypto/random.h"
  25 #include "eapol_auth/eapol_auth_sm.h"
  26 #include "drivers/driver.h"
  27 #include "ap_config.h"
  28 #include "ieee802_11.h"
  29 #include "wpa_auth.h"
  30 #include "pmksa_cache_auth.h"
  31 #include "wpa_auth_i.h"
  32 #include "wpa_auth_ie.h"
  33
  34 #define STATE_MACHINE_DATA struct wpa_state_machine
  35 #define STATE_MACHINE_DEBUG_PREFIX "WPA"
  36 #define STATE_MACHINE_ADDR sm->addr
  37
  38
  39 static void wpa_send_eapol_timeout(void *eloop_ctx, void *timeout_ctx);
  40 static int wpa_sm_step(struct wpa_state_machine *sm);
  41 static int wpa_verify_key_mic(int akmp, size_t pmk_len, struct wpa_ptk *PTK,
  42                               u8 *data, size_t data_len);
  43 #ifdef CONFIG_FILS
  44 static int wpa_aead_decrypt(struct wpa_state_machine *sm, struct wpa_ptk *ptk,
  45                             u8 *buf, size_t buf_len, u16 *_key_data_len);
  46 static struct wpabuf * fils_prepare_plainbuf(struct wpa_state_machine *sm,
  47                                              const struct wpabuf *hlp);
  48 #endif /* CONFIG_FILS */
  49 static void wpa_sm_call_step(void *eloop_ctx, void *timeout_ctx);
  50 static void wpa_group_sm_step(struct wpa_authenticator *wpa_auth,
  51                               struct wpa_group *group);
  52 static void wpa_request_new_ptk(struct wpa_state_machine *sm);
  53 static int wpa_gtk_update(struct wpa_authenticator *wpa_auth,
  54                           struct wpa_group *group);
  55 static int wpa_group_config_group_keys(struct wpa_authenticator *wpa_auth,
  56                                        struct wpa_group *group);
  57 static int wpa_derive_ptk(struct wpa_state_machine *sm, const u8 *snonce,
  58                           const u8 *pmk, unsigned int pmk_len,
  59                           struct wpa_ptk *ptk);
  60 static void wpa_group_free(struct wpa_authenticator *wpa_auth,
  61                            struct wpa_group *group);
  62 static void wpa_group_get(struct wpa_authenticator *wpa_auth,
  63                           struct wpa_group *group);
  64 static void wpa_group_put(struct wpa_authenticator *wpa_auth,
  65                           struct wpa_group *group);
  66 static u8 * ieee80211w_kde_add(struct wpa_state_machine *sm, u8 *pos);
  67
  68 static const u32 eapol_key_timeout_first = 100; /* ms */
  69 static const u32 eapol_key_timeout_subseq = 1000; /* ms */
  70 static const u32 eapol_key_timeout_first_group = 500; /* ms */
  71 static const u32 eapol_key_timeout_no_retrans = 4000; /* ms */
  72
  73 /* TODO: make these configurable */
  74 static const int dot11RSNAConfigPMKLifetime = 43200;
  75 static const int dot11RSNAConfigPMKReauthThreshold = 70;
  76 static const int dot11RSNAConfigSATimeout = 60;
  77
  78
  79 static inline int wpa_auth_mic_failure_report(
  80         struct wpa_authenticator *wpa_auth, const u8 *addr)
  81 {
  82         if (wpa_auth->cb->mic_failure_report)
  83                 return wpa_auth->cb->mic_failure_report(wpa_auth->cb_ctx, addr);
  84         return 0;
  85 }
  86
  87
  88 static inline void wpa_auth_psk_failure_report(
  89         struct wpa_authenticator *wpa_auth, const u8 *addr)
  90 {
  91         if (wpa_auth->cb->psk_failure_report)
  92                 wpa_auth->cb->psk_failure_report(wpa_auth->cb_ctx, addr);
  93 }
  94
  95
  96 static inline void wpa_auth_set_eapol(struct wpa_authenticator *wpa_auth,
  97                                       const u8 *addr, wpa_eapol_variable var,
  98                                       int value)
  99 {
 100         if (wpa_auth->cb->set_eapol)
 101                 wpa_auth->cb->set_eapol(wpa_auth->cb_ctx, addr, var, value);
 102 }
 103
 104
 105 static inline int wpa_auth_get_eapol(struct wpa_authenticator *wpa_auth,
 106                                      const u8 *addr, wpa_eapol_variable var)
 107 {
 108         if (wpa_auth->cb->get_eapol == NULL)
 109                 return -1;
 110         return wpa_auth->cb->get_eapol(wpa_auth->cb_ctx, addr, var);
 111 }
 112
 113
 114 static inline const u8 * wpa_auth_get_psk(struct wpa_authenticator *wpa_auth,
 115                                           const u8 *addr,
 116                                           const u8 *p2p_dev_addr,
 117                                           const u8 *prev_psk, size_t *psk_len,
 118                                           int *vlan_id)
 119 {
 120         if (wpa_auth->cb->get_psk == NULL)
 121                 return NULL;
 122         return wpa_auth->cb->get_psk(wpa_auth->cb_ctx, addr, p2p_dev_addr,
 123                                      prev_psk, psk_len, vlan_id);
 124 }
 125
 126
 127 static inline int wpa_auth_get_msk(struct wpa_authenticator *wpa_auth,
 128                                    const u8 *addr, u8 *msk, size_t *len)
 129 {
 130         if (wpa_auth->cb->get_msk == NULL)
 131                 return -1;
 132         return wpa_auth->cb->get_msk(wpa_auth->cb_ctx, addr, msk, len);
 133 }
 134
 135
 136 static inline int wpa_auth_set_key(struct wpa_authenticator *wpa_auth,
 137                                    int vlan_id,
 138                                    enum wpa_alg alg, const u8 *addr, int idx,
 139                                    u8 *key, size_t key_len)
 140 {
 141         if (wpa_auth->cb->set_key == NULL)
 142                 return -1;
 143         return wpa_auth->cb->set_key(wpa_auth->cb_ctx, vlan_id, alg, addr, idx,
 144                                      key, key_len);
 145 }
 146
 147
 148 static inline int wpa_auth_get_seqnum(struct wpa_authenticator *wpa_auth,
 149                                       const u8 *addr, int idx, u8 *seq)
 150 {
 151         if (wpa_auth->cb->get_seqnum == NULL)
 152                 return -1;
 153         return wpa_auth->cb->get_seqnum(wpa_auth->cb_ctx, addr, idx, seq);
 154 }
 155
 156
 157 static inline int
 158 wpa_auth_send_eapol(struct wpa_authenticator *wpa_auth, const u8 *addr,
 159                     const u8 *data, size_t data_len, int encrypt)
 160 {
 161         if (wpa_auth->cb->send_eapol == NULL)
 162                 return -1;
 163         return wpa_auth->cb->send_eapol(wpa_auth->cb_ctx, addr, data, data_len,
 164                                         encrypt);
 165 }
 166
 167
 168 #ifdef CONFIG_MESH
 169 static inline int wpa_auth_start_ampe(struct wpa_authenticator *wpa_auth,
 170                                       const u8 *addr)
 171 {
 172         if (wpa_auth->cb->start_ampe == NULL)
 173                 return -1;
 174         return wpa_auth->cb->start_ampe(wpa_auth->cb_ctx, addr);
 175 }
 176 #endif /* CONFIG_MESH */
 177
 178
 179 int wpa_auth_for_each_sta(struct wpa_authenticator *wpa_auth,
 180                           int (*cb)(struct wpa_state_machine *sm, void *ctx),
 181                           void *cb_ctx)
 182 {
 183         if (wpa_auth->cb->for_each_sta == NULL)
 184                 return 0;
 185         return wpa_auth->cb->for_each_sta(wpa_auth->cb_ctx, cb, cb_ctx);
 186 }
 187
 188
 189 int wpa_auth_for_each_auth(struct wpa_authenticator *wpa_auth,
 190                            int (*cb)(struct wpa_authenticator *a, void *ctx),
 191                            void *cb_ctx)
 192 {
 193         if (wpa_auth->cb->for_each_auth == NULL)
 194                 return 0;
 195         return wpa_auth->cb->for_each_auth(wpa_auth->cb_ctx, cb, cb_ctx);
 196 }
 197
 198
 199 void wpa_auth_logger(struct wpa_authenticator *wpa_auth, const u8 *addr,
 200                      logger_level level, const char *txt)
 201 {
 202         if (wpa_auth->cb->logger == NULL)
 203                 return;
 204         wpa_auth->cb->logger(wpa_auth->cb_ctx, addr, level, txt);
 205 }
 206
 207
 208 void wpa_auth_vlogger(struct wpa_authenticator *wpa_auth, const u8 *addr,
 209                       logger_level level, const char *fmt, ...)
 210 {
 211         char *format;
 212         int maxlen;
 213         va_list ap;
 214
 215         if (wpa_auth->cb->logger == NULL)
 216                 return;
 217
 218         maxlen = os_strlen(fmt) + 100;
 219         format = os_malloc(maxlen);
 220         if (!format)
 221                 return;
 222
 223         va_start(ap, fmt);
 224         vsnprintf(format, maxlen, fmt, ap);
 225         va_end(ap);
 226
 227         wpa_auth_logger(wpa_auth, addr, level, format);
 228
 229         os_free(format);
 230 }
 231
 232
 233 static void wpa_sta_disconnect(struct wpa_authenticator *wpa_auth,
 234                                const u8 *addr, u16 reason)
 235 {
 236         if (wpa_auth->cb->disconnect == NULL)
 237                 return;
 238         wpa_printf(MSG_DEBUG, "wpa_sta_disconnect STA " MACSTR " (reason %u)",
 239                    MAC2STR(addr), reason);
 240         wpa_auth->cb->disconnect(wpa_auth->cb_ctx, addr, reason);
 241 }
 242
 243
 244 #ifdef CONFIG_OCV
 245 static int wpa_channel_info(struct wpa_authenticator *wpa_auth,
 246                             struct wpa_channel_info *ci)
 247 {
 248         if (!wpa_auth->cb->channel_info)
 249                 return -1;
 250         return wpa_auth->cb->channel_info(wpa_auth->cb_ctx, ci);
 251 }
 252 #endif /* CONFIG_OCV */
 253
 254
 255 static int wpa_auth_update_vlan(struct wpa_authenticator *wpa_auth,
 256                                 const u8 *addr, int vlan_id)
 257 {
 258         if (!wpa_auth->cb->update_vlan)
 259                 return -1;
 260         return wpa_auth->cb->update_vlan(wpa_auth->cb_ctx, addr, vlan_id);
 261 }
 262
 263
 264 static void wpa_rekey_gmk(void *eloop_ctx, void *timeout_ctx)
 265 {
 266         struct wpa_authenticator *wpa_auth = eloop_ctx;
 267
 268         if (random_get_bytes(wpa_auth->group->GMK, WPA_GMK_LEN)) {
 269                 wpa_printf(MSG_ERROR, "Failed to get random data for WPA "
 270                            "initialization.");
 271         } else {
 272                 wpa_auth_logger(wpa_auth, NULL, LOGGER_DEBUG, "GMK rekeyd");
 273                 wpa_hexdump_key(MSG_DEBUG, "GMK",
 274                                 wpa_auth->group->GMK, WPA_GMK_LEN);
 275         }
 276
 277         if (wpa_auth->conf.wpa_gmk_rekey) {
 278                 eloop_register_timeout(wpa_auth->conf.wpa_gmk_rekey, 0,
 279                                        wpa_rekey_gmk, wpa_auth, NULL);
 280         }
 281 }
 282
 283
 284 static void wpa_rekey_gtk(void *eloop_ctx, void *timeout_ctx)
 285 {
 286         struct wpa_authenticator *wpa_auth = eloop_ctx;
 287         struct wpa_group *group, *next;
 288
 289         wpa_auth_logger(wpa_auth, NULL, LOGGER_DEBUG, "rekeying GTK");
 290         group = wpa_auth->group;
 291         while (group) {
 292                 wpa_group_get(wpa_auth, group);
 293
 294                 group->GTKReKey = TRUE;
 295                 do {
 296                         group->changed = FALSE;
 297                         wpa_group_sm_step(wpa_auth, group);
 298                 } while (group->changed);
 299
 300                 next = group->next;
 301                 wpa_group_put(wpa_auth, group);
 302                 group = next;
 303         }
 304
 305         if (wpa_auth->conf.wpa_group_rekey) {
 306                 eloop_register_timeout(wpa_auth->conf.wpa_group_rekey,
 307                                        0, wpa_rekey_gtk, wpa_auth, NULL);
 308         }
 309 }
 310
 311
 312 static void wpa_rekey_ptk(void *eloop_ctx, void *timeout_ctx)
 313 {
 314         struct wpa_authenticator *wpa_auth = eloop_ctx;
 315         struct wpa_state_machine *sm = timeout_ctx;
 316
 317         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_DEBUG, "rekeying PTK");
 318         wpa_request_new_ptk(sm);
 319         wpa_sm_step(sm);
 320 }
 321
 322
 323 void wpa_auth_set_ptk_rekey_timer(struct wpa_state_machine *sm)
 324 {
 325         if (sm && sm->wpa_auth->conf.wpa_ptk_rekey) {
 326                 wpa_printf(MSG_DEBUG, "WPA: Start PTK rekeying timer for "
 327                            MACSTR " (%d seconds)", MAC2STR(sm->addr),
 328                            sm->wpa_auth->conf.wpa_ptk_rekey);
 329                 eloop_cancel_timeout(wpa_rekey_ptk, sm->wpa_auth, sm);
 330                 eloop_register_timeout(sm->wpa_auth->conf.wpa_ptk_rekey, 0,
 331                                        wpa_rekey_ptk, sm->wpa_auth, sm);
 332         }
 333 }
 334
 335
 336 static int wpa_auth_pmksa_clear_cb(struct wpa_state_machine *sm, void *ctx)
 337 {
 338         if (sm->pmksa == ctx)
 339                 sm->pmksa = NULL;
 340         return 0;
 341 }
 342
 343
 344 static void wpa_auth_pmksa_free_cb(struct rsn_pmksa_cache_entry *entry,
 345                                    void *ctx)
 346 {
 347         struct wpa_authenticator *wpa_auth = ctx;
 348         wpa_auth_for_each_sta(wpa_auth, wpa_auth_pmksa_clear_cb, entry);
 349 }
 350
 351
 352 static int wpa_group_init_gmk_and_counter(struct wpa_authenticator *wpa_auth,
 353                                           struct wpa_group *group)
 354 {
 355         u8 buf[ETH_ALEN + 8 + sizeof(unsigned long)];
 356         u8 rkey[32];
 357         unsigned long ptr;
 358
 359         if (random_get_bytes(group->GMK, WPA_GMK_LEN) < 0)
 360                 return -1;
 361         wpa_hexdump_key(MSG_DEBUG, "GMK", group->GMK, WPA_GMK_LEN);
 362
 363         /*
 364          * Counter = PRF-256(Random number, "Init Counter",
 365          *                   Local MAC Address || Time)
 366          */
 367         os_memcpy(buf, wpa_auth->addr, ETH_ALEN);
 368         wpa_get_ntp_timestamp(buf + ETH_ALEN);
 369         ptr = (unsigned long) group;
 370         os_memcpy(buf + ETH_ALEN + 8, &ptr, sizeof(ptr));
 371 #ifdef TEST_FUZZ
 372         os_memset(buf + ETH_ALEN, 0xab, 8);
 373         os_memset(buf + ETH_ALEN + 8, 0xcd, sizeof(ptr));
 374 #endif /* TEST_FUZZ */
 375         if (random_get_bytes(rkey, sizeof(rkey)) < 0)
 376                 return -1;
 377
 378         if (sha1_prf(rkey, sizeof(rkey), "Init Counter", buf, sizeof(buf),
 379                      group->Counter, WPA_NONCE_LEN) < 0)
 380                 return -1;
 381         wpa_hexdump_key(MSG_DEBUG, "Key Counter",
 382                         group->Counter, WPA_NONCE_LEN);
 383
 384         return 0;
 385 }
 386
 387
 388 static struct wpa_group * wpa_group_init(struct wpa_authenticator *wpa_auth,
 389                                          int vlan_id, int delay_init)
 390 {
 391         struct wpa_group *group;
 392
 393         group = os_zalloc(sizeof(struct wpa_group));
 394         if (group == NULL)
 395                 return NULL;
 396
 397         group->GTKAuthenticator = TRUE;
 398         group->vlan_id = vlan_id;
 399         group->GTK_len = wpa_cipher_key_len(wpa_auth->conf.wpa_group);
 400
 401         if (random_pool_ready() != 1) {
 402                 wpa_printf(MSG_INFO, "WPA: Not enough entropy in random pool "
 403                            "for secure operations - update keys later when "
 404                            "the first station connects");
 405         }
 406
 407         /*
 408          * Set initial GMK/Counter value here. The actual values that will be
 409          * used in negotiations will be set once the first station tries to
 410          * connect. This allows more time for collecting additional randomness
 411          * on embedded devices.
 412          */
 413         if (wpa_group_init_gmk_and_counter(wpa_auth, group) < 0) {
 414                 wpa_printf(MSG_ERROR, "Failed to get random data for WPA "
 415                            "initialization.");
 416                 os_free(group);
 417                 return NULL;
 418         }
 419
 420         group->GInit = TRUE;
 421         if (delay_init) {
 422                 wpa_printf(MSG_DEBUG, "WPA: Delay group state machine start "
 423                            "until Beacon frames have been configured");
 424                 /* Initialization is completed in wpa_init_keys(). */
 425         } else {
 426                 wpa_group_sm_step(wpa_auth, group);
 427                 group->GInit = FALSE;
 428                 wpa_group_sm_step(wpa_auth, group);
 429         }
 430
 431         return group;
 432 }
 433
 434
 435 /**
 436  * wpa_init - Initialize WPA authenticator
 437  * @addr: Authenticator address
 438  * @conf: Configuration for WPA authenticator
 439  * @cb: Callback functions for WPA authenticator
 440  * Returns: Pointer to WPA authenticator data or %NULL on failure
 441  */
 442 struct wpa_authenticator * wpa_init(const u8 *addr,
 443                                     struct wpa_auth_config *conf,
 444                                     const struct wpa_auth_callbacks *cb,
 445                                     void *cb_ctx)
 446 {
 447         struct wpa_authenticator *wpa_auth;
 448
 449         wpa_auth = os_zalloc(sizeof(struct wpa_authenticator));
 450         if (wpa_auth == NULL)
 451                 return NULL;
 452         os_memcpy(wpa_auth->addr, addr, ETH_ALEN);
 453         os_memcpy(&wpa_auth->conf, conf, sizeof(*conf));
 454         wpa_auth->cb = cb;
 455         wpa_auth->cb_ctx = cb_ctx;
 456
 457         if (wpa_auth_gen_wpa_ie(wpa_auth)) {
 458                 wpa_printf(MSG_ERROR, "Could not generate WPA IE.");
 459                 os_free(wpa_auth);
 460                 return NULL;
 461         }
 462
 463         wpa_auth->group = wpa_group_init(wpa_auth, 0, 1);
 464         if (wpa_auth->group == NULL) {
 465                 os_free(wpa_auth->wpa_ie);
 466                 os_free(wpa_auth);
 467                 return NULL;
 468         }
 469
 470         wpa_auth->pmksa = pmksa_cache_auth_init(wpa_auth_pmksa_free_cb,
 471                                                 wpa_auth);
 472         if (wpa_auth->pmksa == NULL) {
 473                 wpa_printf(MSG_ERROR, "PMKSA cache initialization failed.");
 474                 os_free(wpa_auth->group);
 475                 os_free(wpa_auth->wpa_ie);
 476                 os_free(wpa_auth);
 477                 return NULL;
 478         }
 479
 480 #ifdef CONFIG_IEEE80211R_AP
 481         wpa_auth->ft_pmk_cache = wpa_ft_pmk_cache_init();
 482         if (wpa_auth->ft_pmk_cache == NULL) {
 483                 wpa_printf(MSG_ERROR, "FT PMK cache initialization failed.");
 484                 os_free(wpa_auth->group);
 485                 os_free(wpa_auth->wpa_ie);
 486                 pmksa_cache_auth_deinit(wpa_auth->pmksa);
 487                 os_free(wpa_auth);
 488                 return NULL;
 489         }
 490 #endif /* CONFIG_IEEE80211R_AP */
 491
 492         if (wpa_auth->conf.wpa_gmk_rekey) {
 493                 eloop_register_timeout(wpa_auth->conf.wpa_gmk_rekey, 0,
 494                                        wpa_rekey_gmk, wpa_auth, NULL);
 495         }
 496
 497         if (wpa_auth->conf.wpa_group_rekey) {
 498                 eloop_register_timeout(wpa_auth->conf.wpa_group_rekey, 0,
 499                                        wpa_rekey_gtk, wpa_auth, NULL);
 500         }
 501
 502 #ifdef CONFIG_P2P
 503         if (WPA_GET_BE32(conf->ip_addr_start)) {
 504                 int count = WPA_GET_BE32(conf->ip_addr_end) -
 505                         WPA_GET_BE32(conf->ip_addr_start) + 1;
 506                 if (count > 1000)
 507                         count = 1000;
 508                 if (count > 0)
 509                         wpa_auth->ip_pool = bitfield_alloc(count);
 510         }
 511 #endif /* CONFIG_P2P */
 512
 513         return wpa_auth;
 514 }
 515
 516
 517 int wpa_init_keys(struct wpa_authenticator *wpa_auth)
 518 {
 519         struct wpa_group *group = wpa_auth->group;
 520
 521         wpa_printf(MSG_DEBUG, "WPA: Start group state machine to set initial "
 522                    "keys");
 523         wpa_group_sm_step(wpa_auth, group);
 524         group->GInit = FALSE;
 525         wpa_group_sm_step(wpa_auth, group);
 526         if (group->wpa_group_state == WPA_GROUP_FATAL_FAILURE)
 527                 return -1;
 528         return 0;
 529 }
 530
 531
 532 /**
 533  * wpa_deinit - Deinitialize WPA authenticator
 534  * @wpa_auth: Pointer to WPA authenticator data from wpa_init()
 535  */
 536 void wpa_deinit(struct wpa_authenticator *wpa_auth)
 537 {
 538         struct wpa_group *group, *prev;
 539
 540         eloop_cancel_timeout(wpa_rekey_gmk, wpa_auth, NULL);
 541         eloop_cancel_timeout(wpa_rekey_gtk, wpa_auth, NULL);
 542
 543         pmksa_cache_auth_deinit(wpa_auth->pmksa);
 544
 545 #ifdef CONFIG_IEEE80211R_AP
 546         wpa_ft_pmk_cache_deinit(wpa_auth->ft_pmk_cache);
 547         wpa_auth->ft_pmk_cache = NULL;
 548         wpa_ft_deinit(wpa_auth);
 549 #endif /* CONFIG_IEEE80211R_AP */
 550
 551 #ifdef CONFIG_P2P
 552         bitfield_free(wpa_auth->ip_pool);
 553 #endif /* CONFIG_P2P */
 554
 555
 556         os_free(wpa_auth->wpa_ie);
 557
 558         group = wpa_auth->group;
 559         while (group) {
 560                 prev = group;
 561                 group = group->next;
 562                 os_free(prev);
 563         }
 564
 565         os_free(wpa_auth);
 566 }
 567
 568
 569 /**
 570  * wpa_reconfig - Update WPA authenticator configuration
 571  * @wpa_auth: Pointer to WPA authenticator data from wpa_init()
 572  * @conf: Configuration for WPA authenticator
 573  */
 574 int wpa_reconfig(struct wpa_authenticator *wpa_auth,
 575                  struct wpa_auth_config *conf)
 576 {
 577         struct wpa_group *group;
 578         if (wpa_auth == NULL)
 579                 return 0;
 580
 581         os_memcpy(&wpa_auth->conf, conf, sizeof(*conf));
 582         if (wpa_auth_gen_wpa_ie(wpa_auth)) {
 583                 wpa_printf(MSG_ERROR, "Could not generate WPA IE.");
 584                 return -1;
 585         }
 586
 587         /*
 588          * Reinitialize GTK to make sure it is suitable for the new
 589          * configuration.
 590          */
 591         group = wpa_auth->group;
 592         group->GTK_len = wpa_cipher_key_len(wpa_auth->conf.wpa_group);
 593         group->GInit = TRUE;
 594         wpa_group_sm_step(wpa_auth, group);
 595         group->GInit = FALSE;
 596         wpa_group_sm_step(wpa_auth, group);
 597
 598         return 0;
 599 }
 600
 601
 602 struct wpa_state_machine *
 603 wpa_auth_sta_init(struct wpa_authenticator *wpa_auth, const u8 *addr,
 604                   const u8 *p2p_dev_addr)
 605 {
 606         struct wpa_state_machine *sm;
 607
 608         if (wpa_auth->group->wpa_group_state == WPA_GROUP_FATAL_FAILURE)
 609                 return NULL;
 610
 611         sm = os_zalloc(sizeof(struct wpa_state_machine));
 612         if (sm == NULL)
 613                 return NULL;
 614         os_memcpy(sm->addr, addr, ETH_ALEN);
 615         if (p2p_dev_addr)
 616                 os_memcpy(sm->p2p_dev_addr, p2p_dev_addr, ETH_ALEN);
 617
 618         sm->wpa_auth = wpa_auth;
 619         sm->group = wpa_auth->group;
 620         wpa_group_get(sm->wpa_auth, sm->group);
 621
 622         return sm;
 623 }
 624
 625
 626 int wpa_auth_sta_associated(struct wpa_authenticator *wpa_auth,
 627                             struct wpa_state_machine *sm)
 628 {
 629         if (wpa_auth == NULL || !wpa_auth->conf.wpa || sm == NULL)
 630                 return -1;
 631
 632 #ifdef CONFIG_IEEE80211R_AP
 633         if (sm->ft_completed) {
 634                 wpa_auth_logger(wpa_auth, sm->addr, LOGGER_DEBUG,
 635                                 "FT authentication already completed - do not "
 636                                 "start 4-way handshake");
 637                 /* Go to PTKINITDONE state to allow GTK rekeying */
 638                 sm->wpa_ptk_state = WPA_PTK_PTKINITDONE;
 639                 sm->Pair = TRUE;
 640                 return 0;
 641         }
 642 #endif /* CONFIG_IEEE80211R_AP */
 643
 644 #ifdef CONFIG_FILS
 645         if (sm->fils_completed) {
 646                 wpa_auth_logger(wpa_auth, sm->addr, LOGGER_DEBUG,
 647                                 "FILS authentication already completed - do not start 4-way handshake");
 648                 /* Go to PTKINITDONE state to allow GTK rekeying */
 649                 sm->wpa_ptk_state = WPA_PTK_PTKINITDONE;
 650                 sm->Pair = TRUE;
 651                 return 0;
 652         }
 653 #endif /* CONFIG_FILS */
 654
 655         if (sm->started) {
 656                 os_memset(&sm->key_replay, 0, sizeof(sm->key_replay));
 657                 sm->ReAuthenticationRequest = TRUE;
 658                 return wpa_sm_step(sm);
 659         }
 660
 661         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_DEBUG,
 662                         "start authentication");
 663         sm->started = 1;
 664
 665         sm->Init = TRUE;
 666         if (wpa_sm_step(sm) == 1)
 667                 return 1; /* should not really happen */
 668         sm->Init = FALSE;
 669         sm->AuthenticationRequest = TRUE;
 670         return wpa_sm_step(sm);
 671 }
 672
 673
 674 void wpa_auth_sta_no_wpa(struct wpa_state_machine *sm)
 675 {
 676         /* WPA/RSN was not used - clear WPA state. This is needed if the STA
 677          * reassociates back to the same AP while the previous entry for the
 678          * STA has not yet been removed. */
 679         if (sm == NULL)
 680                 return;
 681
 682         sm->wpa_key_mgmt = 0;
 683 }
 684
 685
 686 static void wpa_free_sta_sm(struct wpa_state_machine *sm)
 687 {
 688 #ifdef CONFIG_P2P
 689         if (WPA_GET_BE32(sm->ip_addr)) {
 690                 u32 start;
 691                 wpa_printf(MSG_DEBUG, "P2P: Free assigned IP "
 692                            "address %u.%u.%u.%u from " MACSTR,
 693                            sm->ip_addr[0], sm->ip_addr[1],
 694                            sm->ip_addr[2], sm->ip_addr[3],
 695                            MAC2STR(sm->addr));
 696                 start = WPA_GET_BE32(sm->wpa_auth->conf.ip_addr_start);
 697                 bitfield_clear(sm->wpa_auth->ip_pool,
 698                                WPA_GET_BE32(sm->ip_addr) - start);
 699         }
 700 #endif /* CONFIG_P2P */
 701         if (sm->GUpdateStationKeys) {
 702                 sm->group->GKeyDoneStations--;
 703                 sm->GUpdateStationKeys = FALSE;
 704         }
 705 #ifdef CONFIG_IEEE80211R_AP
 706         os_free(sm->assoc_resp_ftie);
 707         wpabuf_free(sm->ft_pending_req_ies);
 708 #endif /* CONFIG_IEEE80211R_AP */
 709         os_free(sm->last_rx_eapol_key);
 710         os_free(sm->wpa_ie);
 711         wpa_group_put(sm->wpa_auth, sm->group);
 712 #ifdef CONFIG_DPP2
 713         wpabuf_clear_free(sm->dpp_z);
 714 #endif /* CONFIG_DPP2 */
 715         bin_clear_free(sm, sizeof(*sm));
 716 }
 717
 718
 719 void wpa_auth_sta_deinit(struct wpa_state_machine *sm)
 720 {
 721         if (sm == NULL)
 722                 return;
 723
 724         if (sm->wpa_auth->conf.wpa_strict_rekey && sm->has_GTK) {
 725                 wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
 726                                 "strict rekeying - force GTK rekey since STA "
 727                                 "is leaving");
 728                 if (eloop_deplete_timeout(0, 500000, wpa_rekey_gtk,
 729                                           sm->wpa_auth, NULL) == -1)
 730                         eloop_register_timeout(0, 500000, wpa_rekey_gtk, sm->wpa_auth,
 731                                                NULL);
 732         }
 733
 734         eloop_cancel_timeout(wpa_send_eapol_timeout, sm->wpa_auth, sm);
 735         sm->pending_1_of_4_timeout = 0;
 736         eloop_cancel_timeout(wpa_sm_call_step, sm, NULL);
 737         eloop_cancel_timeout(wpa_rekey_ptk, sm->wpa_auth, sm);
 738 #ifdef CONFIG_IEEE80211R_AP
 739         wpa_ft_sta_deinit(sm);
 740 #endif /* CONFIG_IEEE80211R_AP */
 741         if (sm->in_step_loop) {
 742                 /* Must not free state machine while wpa_sm_step() is running.
 743                  * Freeing will be completed in the end of wpa_sm_step(). */
 744                 wpa_printf(MSG_DEBUG, "WPA: Registering pending STA state "
 745                            "machine deinit for " MACSTR, MAC2STR(sm->addr));
 746                 sm->pending_deinit = 1;
 747         } else
 748                 wpa_free_sta_sm(sm);
 749 }
 750
 751
 752 static void wpa_request_new_ptk(struct wpa_state_machine *sm)
 753 {
 754         if (sm == NULL)
 755                 return;
 756
 757         sm->PTKRequest = TRUE;
 758         sm->PTK_valid = 0;
 759 }
 760
 761
 762 static int wpa_replay_counter_valid(struct wpa_key_replay_counter *ctr,
 763                                     const u8 *replay_counter)
 764 {
 765         int i;
 766         for (i = 0; i < RSNA_MAX_EAPOL_RETRIES; i++) {
 767                 if (!ctr[i].valid)
 768                         break;
 769                 if (os_memcmp(replay_counter, ctr[i].counter,
 770                               WPA_REPLAY_COUNTER_LEN) == 0)
 771                         return 1;
 772         }
 773         return 0;
 774 }
 775
 776
 777 static void wpa_replay_counter_mark_invalid(struct wpa_key_replay_counter *ctr,
 778                                             const u8 *replay_counter)
 779 {
 780         int i;
 781         for (i = 0; i < RSNA_MAX_EAPOL_RETRIES; i++) {
 782                 if (ctr[i].valid &&
 783                     (replay_counter == NULL ||
 784                      os_memcmp(replay_counter, ctr[i].counter,
 785                                WPA_REPLAY_COUNTER_LEN) == 0))
 786                         ctr[i].valid = FALSE;
 787         }
 788 }
 789
 790
 791 #ifdef CONFIG_IEEE80211R_AP
 792 static int ft_check_msg_2_of_4(struct wpa_authenticator *wpa_auth,
 793                                struct wpa_state_machine *sm,
 794                                struct wpa_eapol_ie_parse *kde)
 795 {
 796         struct wpa_ie_data ie;
 797         struct rsn_mdie *mdie;
 798
 799         if (wpa_parse_wpa_ie_rsn(kde->rsn_ie, kde->rsn_ie_len, &ie) < 0 ||
 800             ie.num_pmkid != 1 || ie.pmkid == NULL) {
 801                 wpa_printf(MSG_DEBUG, "FT: No PMKR1Name in "
 802                            "FT 4-way handshake message 2/4");
 803                 return -1;
 804         }
 805
 806         os_memcpy(sm->sup_pmk_r1_name, ie.pmkid, PMKID_LEN);
 807         wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name from Supplicant",
 808                     sm->sup_pmk_r1_name, PMKID_LEN);
 809
 810         if (!kde->mdie || !kde->ftie) {
 811                 wpa_printf(MSG_DEBUG, "FT: No %s in FT 4-way handshake "
 812                            "message 2/4", kde->mdie ? "FTIE" : "MDIE");
 813                 return -1;
 814         }
 815
 816         mdie = (struct rsn_mdie *) (kde->mdie + 2);
 817         if (kde->mdie[1] < sizeof(struct rsn_mdie) ||
 818             os_memcmp(wpa_auth->conf.mobility_domain, mdie->mobility_domain,
 819                       MOBILITY_DOMAIN_ID_LEN) != 0) {
 820                 wpa_printf(MSG_DEBUG, "FT: MDIE mismatch");
 821                 return -1;
 822         }
 823
 824         if (sm->assoc_resp_ftie &&
 825             (kde->ftie[1] != sm->assoc_resp_ftie[1] ||
 826              os_memcmp(kde->ftie, sm->assoc_resp_ftie,
 827                        2 + sm->assoc_resp_ftie[1]) != 0)) {
 828                 wpa_printf(MSG_DEBUG, "FT: FTIE mismatch");
 829                 wpa_hexdump(MSG_DEBUG, "FT: FTIE in EAPOL-Key msg 2/4",
 830                             kde->ftie, kde->ftie_len);
 831                 wpa_hexdump(MSG_DEBUG, "FT: FTIE in (Re)AssocResp",
 832                             sm->assoc_resp_ftie, 2 + sm->assoc_resp_ftie[1]);
 833                 return -1;
 834         }
 835
 836         return 0;
 837 }
 838 #endif /* CONFIG_IEEE80211R_AP */
 839
 840
 841 static int wpa_receive_error_report(struct wpa_authenticator *wpa_auth,
 842                                     struct wpa_state_machine *sm, int group)
 843 {
 844         /* Supplicant reported a Michael MIC error */
 845         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_INFO,
 846                          "received EAPOL-Key Error Request "
 847                          "(STA detected Michael MIC failure (group=%d))",
 848                          group);
 849
 850         if (group && wpa_auth->conf.wpa_group != WPA_CIPHER_TKIP) {
 851                 wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
 852                                 "ignore Michael MIC failure report since "
 853                                 "group cipher is not TKIP");
 854         } else if (!group && sm->pairwise != WPA_CIPHER_TKIP) {
 855                 wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
 856                                 "ignore Michael MIC failure report since "
 857                                 "pairwise cipher is not TKIP");
 858         } else {
 859                 if (wpa_auth_mic_failure_report(wpa_auth, sm->addr) > 0)
 860                         return 1; /* STA entry was removed */
 861                 sm->dot11RSNAStatsTKIPRemoteMICFailures++;
 862                 wpa_auth->dot11RSNAStatsTKIPRemoteMICFailures++;
 863         }
 864
 865         /*
 866          * Error report is not a request for a new key handshake, but since
 867          * Authenticator may do it, let's change the keys now anyway.
 868          */
 869         wpa_request_new_ptk(sm);
 870         return 0;
 871 }
 872
 873
 874 static int wpa_try_alt_snonce(struct wpa_state_machine *sm, u8 *data,
 875                               size_t data_len)
 876 {
 877         struct wpa_ptk PTK;
 878         int ok = 0;
 879         const u8 *pmk = NULL;
 880         size_t pmk_len;
 881         int vlan_id = 0;
 882
 883         os_memset(&PTK, 0, sizeof(PTK));
 884         for (;;) {
 885                 if (wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt) &&
 886                     !wpa_key_mgmt_sae(sm->wpa_key_mgmt)) {
 887                         pmk = wpa_auth_get_psk(sm->wpa_auth, sm->addr,
 888                                                sm->p2p_dev_addr, pmk, &pmk_len,
 889                                                &vlan_id);
 890                         if (pmk == NULL)
 891                                 break;
 892 #ifdef CONFIG_IEEE80211R_AP
 893                         if (wpa_key_mgmt_ft_psk(sm->wpa_key_mgmt)) {
 894                                 os_memcpy(sm->xxkey, pmk, pmk_len);
 895                                 sm->xxkey_len = pmk_len;
 896                         }
 897 #endif /* CONFIG_IEEE80211R_AP */
 898                 } else {
 899                         pmk = sm->PMK;
 900                         pmk_len = sm->pmk_len;
 901                 }
 902
 903                 if (wpa_derive_ptk(sm, sm->alt_SNonce, pmk, pmk_len, &PTK) < 0)
 904                         break;
 905
 906                 if (wpa_verify_key_mic(sm->wpa_key_mgmt, pmk_len, &PTK,
 907                                        data, data_len) == 0) {
 908                         if (sm->PMK != pmk) {
 909                                 os_memcpy(sm->PMK, pmk, pmk_len);
 910                                 sm->pmk_len = pmk_len;
 911                         }
 912                         ok = 1;
 913                         break;
 914                 }
 915
 916                 if (!wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt) ||
 917                     wpa_key_mgmt_sae(sm->wpa_key_mgmt))
 918                         break;
 919         }
 920
 921         if (!ok) {
 922                 wpa_printf(MSG_DEBUG,
 923                            "WPA: Earlier SNonce did not result in matching MIC");
 924                 return -1;
 925         }
 926
 927         wpa_printf(MSG_DEBUG,
 928                    "WPA: Earlier SNonce resulted in matching MIC");
 929         sm->alt_snonce_valid = 0;
 930
 931         if (vlan_id && wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt) &&
 932             wpa_auth_update_vlan(sm->wpa_auth, sm->addr, vlan_id) < 0)
 933                 return -1;
 934
 935         os_memcpy(sm->SNonce, sm->alt_SNonce, WPA_NONCE_LEN);
 936         os_memcpy(&sm->PTK, &PTK, sizeof(PTK));
 937         forced_memzero(&PTK, sizeof(PTK));
 938         sm->PTK_valid = TRUE;
 939
 940         return 0;
 941 }
 942
 943
 944 void wpa_receive(struct wpa_authenticator *wpa_auth,
 945                  struct wpa_state_machine *sm,
 946                  u8 *data, size_t data_len)
 947 {
 948         struct ieee802_1x_hdr *hdr;
 949         struct wpa_eapol_key *key;
 950         u16 key_info, key_data_length;
 951         enum { PAIRWISE_2, PAIRWISE_4, GROUP_2, REQUEST } msg;
 952         char *msgtxt;
 953         struct wpa_eapol_ie_parse kde;
 954         const u8 *key_data;
 955         size_t keyhdrlen, mic_len;
 956         u8 *mic;
 957
 958         if (wpa_auth == NULL || !wpa_auth->conf.wpa || sm == NULL)
 959                 return;
 960         wpa_hexdump(MSG_MSGDUMP, "WPA: RX EAPOL data", data, data_len);
 961
 962         mic_len = wpa_mic_len(sm->wpa_key_mgmt, sm->pmk_len);
 963         keyhdrlen = sizeof(*key) + mic_len + 2;
 964
 965         if (data_len < sizeof(*hdr) + keyhdrlen) {
 966                 wpa_printf(MSG_DEBUG, "WPA: Ignore too short EAPOL-Key frame");
 967                 return;
 968         }
 969
 970         hdr = (struct ieee802_1x_hdr *) data;
 971         key = (struct wpa_eapol_key *) (hdr + 1);
 972         mic = (u8 *) (key + 1);
 973         key_info = WPA_GET_BE16(key->key_info);
 974         key_data = mic + mic_len + 2;
 975         key_data_length = WPA_GET_BE16(mic + mic_len);
 976         wpa_printf(MSG_DEBUG, "WPA: Received EAPOL-Key from " MACSTR
 977                    " key_info=0x%x type=%u mic_len=%u key_data_length=%u",
 978                    MAC2STR(sm->addr), key_info, key->type,
 979                    (unsigned int) mic_len, key_data_length);
 980         wpa_hexdump(MSG_MSGDUMP,
 981                     "WPA: EAPOL-Key header (ending before Key MIC)",
 982                     key, sizeof(*key));
 983         wpa_hexdump(MSG_MSGDUMP, "WPA: EAPOL-Key Key MIC",
 984                     mic, mic_len);
 985         if (key_data_length > data_len - sizeof(*hdr) - keyhdrlen) {
 986                 wpa_printf(MSG_INFO, "WPA: Invalid EAPOL-Key frame - "
 987                            "key_data overflow (%d > %lu)",
 988                            key_data_length,
 989                            (unsigned long) (data_len - sizeof(*hdr) -
 990                                             keyhdrlen));
 991                 return;
 992         }
 993
 994         if (sm->wpa == WPA_VERSION_WPA2) {
 995                 if (key->type == EAPOL_KEY_TYPE_WPA) {
 996                         /*
 997                          * Some deployed station implementations seem to send
 998                          * msg 4/4 with incorrect type value in WPA2 mode.
 999                          */
1000                         wpa_printf(MSG_DEBUG, "Workaround: Allow EAPOL-Key "
1001                                    "with unexpected WPA type in RSN mode");
1002                 } else if (key->type != EAPOL_KEY_TYPE_RSN) {
1003                         wpa_printf(MSG_DEBUG, "Ignore EAPOL-Key with "
1004                                    "unexpected type %d in RSN mode",
1005                                    key->type);
1006                         return;
1007                 }
1008         } else {
1009                 if (key->type != EAPOL_KEY_TYPE_WPA) {
1010                         wpa_printf(MSG_DEBUG, "Ignore EAPOL-Key with "
1011                                    "unexpected type %d in WPA mode",
1012                                    key->type);
1013                         return;
1014                 }
1015         }
1016
1017         wpa_hexdump(MSG_DEBUG, "WPA: Received Key Nonce", key->key_nonce,
1018                     WPA_NONCE_LEN);
1019         wpa_hexdump(MSG_DEBUG, "WPA: Received Replay Counter",
1020                     key->replay_counter, WPA_REPLAY_COUNTER_LEN);
1021
1022         /* FIX: verify that the EAPOL-Key frame was encrypted if pairwise keys
1023          * are set */
1024
1025         if (key_info & WPA_KEY_INFO_SMK_MESSAGE) {
1026                 wpa_printf(MSG_DEBUG, "WPA: Ignore SMK message");
1027                 return;
1028         }
1029
1030         if (key_info & WPA_KEY_INFO_REQUEST) {
1031                 msg = REQUEST;
1032                 msgtxt = "Request";
1033         } else if (!(key_info & WPA_KEY_INFO_KEY_TYPE)) {
1034                 msg = GROUP_2;
1035                 msgtxt = "2/2 Group";
1036         } else if (key_data_length == 0 ||
1037                    (mic_len == 0 && (key_info & WPA_KEY_INFO_ENCR_KEY_DATA) &&
1038                     key_data_length == AES_BLOCK_SIZE)) {
1039                 msg = PAIRWISE_4;
1040                 msgtxt = "4/4 Pairwise";
1041         } else {
1042                 msg = PAIRWISE_2;
1043                 msgtxt = "2/4 Pairwise";
1044         }
1045
1046         if (msg == REQUEST || msg == PAIRWISE_2 || msg == PAIRWISE_4 ||
1047             msg == GROUP_2) {
1048                 u16 ver = key_info & WPA_KEY_INFO_TYPE_MASK;
1049                 if (sm->pairwise == WPA_CIPHER_CCMP ||
1050                     sm->pairwise == WPA_CIPHER_GCMP) {
1051                         if (wpa_use_cmac(sm->wpa_key_mgmt) &&
1052                             !wpa_use_akm_defined(sm->wpa_key_mgmt) &&
1053                             ver != WPA_KEY_INFO_TYPE_AES_128_CMAC) {
1054                                 wpa_auth_logger(wpa_auth, sm->addr,
1055                                                 LOGGER_WARNING,
1056                                                 "advertised support for "
1057                                                 "AES-128-CMAC, but did not "
1058                                                 "use it");
1059                                 return;
1060                         }
1061
1062                         if (!wpa_use_cmac(sm->wpa_key_mgmt) &&
1063                             !wpa_use_akm_defined(sm->wpa_key_mgmt) &&
1064                             ver != WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
1065                                 wpa_auth_logger(wpa_auth, sm->addr,
1066                                                 LOGGER_WARNING,
1067                                                 "did not use HMAC-SHA1-AES "
1068                                                 "with CCMP/GCMP");
1069                                 return;
1070                         }
1071                 }
1072
1073                 if (wpa_use_akm_defined(sm->wpa_key_mgmt) &&
1074                     ver != WPA_KEY_INFO_TYPE_AKM_DEFINED) {
1075                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_WARNING,
1076                                         "did not use EAPOL-Key descriptor version 0 as required for AKM-defined cases");
1077                         return;
1078                 }
1079         }
1080
1081         if (key_info & WPA_KEY_INFO_REQUEST) {
1082                 if (sm->req_replay_counter_used &&
1083                     os_memcmp(key->replay_counter, sm->req_replay_counter,
1084                               WPA_REPLAY_COUNTER_LEN) <= 0) {
1085                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_WARNING,
1086                                         "received EAPOL-Key request with "
1087                                         "replayed counter");
1088                         return;
1089                 }
1090         }
1091
1092         if (!(key_info & WPA_KEY_INFO_REQUEST) &&
1093             !wpa_replay_counter_valid(sm->key_replay, key->replay_counter)) {
1094                 int i;
1095
1096                 if (msg == PAIRWISE_2 &&
1097                     wpa_replay_counter_valid(sm->prev_key_replay,
1098                                              key->replay_counter) &&
1099                     sm->wpa_ptk_state == WPA_PTK_PTKINITNEGOTIATING &&
1100                     os_memcmp(sm->SNonce, key->key_nonce, WPA_NONCE_LEN) != 0)
1101                 {
1102                         /*
1103                          * Some supplicant implementations (e.g., Windows XP
1104                          * WZC) update SNonce for each EAPOL-Key 2/4. This
1105                          * breaks the workaround on accepting any of the
1106                          * pending requests, so allow the SNonce to be updated
1107                          * even if we have already sent out EAPOL-Key 3/4.
1108                          */
1109                         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_DEBUG,
1110                                          "Process SNonce update from STA "
1111                                          "based on retransmitted EAPOL-Key "
1112                                          "1/4");
1113                         sm->update_snonce = 1;
1114                         os_memcpy(sm->alt_SNonce, sm->SNonce, WPA_NONCE_LEN);
1115                         sm->alt_snonce_valid = TRUE;
1116                         os_memcpy(sm->alt_replay_counter,
1117                                   sm->key_replay[0].counter,
1118                                   WPA_REPLAY_COUNTER_LEN);
1119                         goto continue_processing;
1120                 }
1121
1122                 if (msg == PAIRWISE_4 && sm->alt_snonce_valid &&
1123                     sm->wpa_ptk_state == WPA_PTK_PTKINITNEGOTIATING &&
1124                     os_memcmp(key->replay_counter, sm->alt_replay_counter,
1125                               WPA_REPLAY_COUNTER_LEN) == 0) {
1126                         /*
1127                          * Supplicant may still be using the old SNonce since
1128                          * there was two EAPOL-Key 2/4 messages and they had
1129                          * different SNonce values.
1130                          */
1131                         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_DEBUG,
1132                                          "Try to process received EAPOL-Key 4/4 based on old Replay Counter and SNonce from an earlier EAPOL-Key 1/4");
1133                         goto continue_processing;
1134                 }
1135
1136                 if (msg == PAIRWISE_2 &&
1137                     wpa_replay_counter_valid(sm->prev_key_replay,
1138                                              key->replay_counter) &&
1139                     sm->wpa_ptk_state == WPA_PTK_PTKINITNEGOTIATING) {
1140                         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_DEBUG,
1141                                          "ignore retransmitted EAPOL-Key %s - "
1142                                          "SNonce did not change", msgtxt);
1143                 } else {
1144                         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_DEBUG,
1145                                          "received EAPOL-Key %s with "
1146                                          "unexpected replay counter", msgtxt);
1147                 }
1148                 for (i = 0; i < RSNA_MAX_EAPOL_RETRIES; i++) {
1149                         if (!sm->key_replay[i].valid)
1150                                 break;
1151                         wpa_hexdump(MSG_DEBUG, "pending replay counter",
1152                                     sm->key_replay[i].counter,
1153                                     WPA_REPLAY_COUNTER_LEN);
1154                 }
1155                 wpa_hexdump(MSG_DEBUG, "received replay counter",
1156                             key->replay_counter, WPA_REPLAY_COUNTER_LEN);
1157                 return;
1158         }
1159
1160 continue_processing:
1161 #ifdef CONFIG_FILS
1162         if (sm->wpa == WPA_VERSION_WPA2 && mic_len == 0 &&
1163             !(key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
1164                 wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_DEBUG,
1165                                  "WPA: Encr Key Data bit not set even though AEAD cipher is supposed to be used - drop frame");
1166                 return;
1167         }
1168 #endif /* CONFIG_FILS */
1169
1170         switch (msg) {
1171         case PAIRWISE_2:
1172                 if (sm->wpa_ptk_state != WPA_PTK_PTKSTART &&
1173                     sm->wpa_ptk_state != WPA_PTK_PTKCALCNEGOTIATING &&
1174                     (!sm->update_snonce ||
1175                      sm->wpa_ptk_state != WPA_PTK_PTKINITNEGOTIATING)) {
1176                         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_INFO,
1177                                          "received EAPOL-Key msg 2/4 in "
1178                                          "invalid state (%d) - dropped",
1179                                          sm->wpa_ptk_state);
1180                         return;
1181                 }
1182                 random_add_randomness(key->key_nonce, WPA_NONCE_LEN);
1183                 if (sm->group->reject_4way_hs_for_entropy) {
1184                         /*
1185                          * The system did not have enough entropy to generate
1186                          * strong random numbers. Reject the first 4-way
1187                          * handshake(s) and collect some entropy based on the
1188                          * information from it. Once enough entropy is
1189                          * available, the next atempt will trigger GMK/Key
1190                          * Counter update and the station will be allowed to
1191                          * continue.
1192                          */
1193                         wpa_printf(MSG_DEBUG, "WPA: Reject 4-way handshake to "
1194                                    "collect more entropy for random number "
1195                                    "generation");
1196                         random_mark_pool_ready();
1197                         wpa_sta_disconnect(wpa_auth, sm->addr,
1198                                            WLAN_REASON_PREV_AUTH_NOT_VALID);
1199                         return;
1200                 }
1201                 break;
1202         case PAIRWISE_4:
1203                 if (sm->wpa_ptk_state != WPA_PTK_PTKINITNEGOTIATING ||
1204                     !sm->PTK_valid) {
1205                         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_INFO,
1206                                          "received EAPOL-Key msg 4/4 in "
1207                                          "invalid state (%d) - dropped",
1208                                          sm->wpa_ptk_state);
1209                         return;
1210                 }
1211                 break;
1212         case GROUP_2:
1213                 if (sm->wpa_ptk_group_state != WPA_PTK_GROUP_REKEYNEGOTIATING
1214                     || !sm->PTK_valid) {
1215                         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_INFO,
1216                                          "received EAPOL-Key msg 2/2 in "
1217                                          "invalid state (%d) - dropped",
1218                                          sm->wpa_ptk_group_state);
1219                         return;
1220                 }
1221                 break;
1222         case REQUEST:
1223                 break;
1224         }
1225
1226         wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_DEBUG,
1227                          "received EAPOL-Key frame (%s)", msgtxt);
1228
1229         if (key_info & WPA_KEY_INFO_ACK) {
1230                 wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
1231                                 "received invalid EAPOL-Key: Key Ack set");
1232                 return;
1233         }
1234
1235         if (!wpa_key_mgmt_fils(sm->wpa_key_mgmt) &&
1236             !(key_info & WPA_KEY_INFO_MIC)) {
1237                 wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
1238                                 "received invalid EAPOL-Key: Key MIC not set");
1239                 return;
1240         }
1241
1242 #ifdef CONFIG_FILS
1243         if (wpa_key_mgmt_fils(sm->wpa_key_mgmt) &&
1244             (key_info & WPA_KEY_INFO_MIC)) {
1245                 wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
1246                                 "received invalid EAPOL-Key: Key MIC set");
1247                 return;
1248         }
1249 #endif /* CONFIG_FILS */
1250
1251         sm->MICVerified = FALSE;
1252         if (sm->PTK_valid && !sm->update_snonce) {
1253                 if (mic_len &&
1254                     wpa_verify_key_mic(sm->wpa_key_mgmt, sm->pmk_len, &sm->PTK,
1255                                        data, data_len) &&
1256                     (msg != PAIRWISE_4 || !sm->alt_snonce_valid ||
1257                      wpa_try_alt_snonce(sm, data, data_len))) {
1258                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
1259                                         "received EAPOL-Key with invalid MIC");
1260 #ifdef TEST_FUZZ
1261                         wpa_printf(MSG_INFO,
1262                                    "TEST: Ignore Key MIC failure for fuzz testing");
1263                         goto continue_fuzz;
1264 #endif /* TEST_FUZZ */
1265                         return;
1266                 }
1267 #ifdef CONFIG_FILS
1268                 if (!mic_len &&
1269                     wpa_aead_decrypt(sm, &sm->PTK, data, data_len,
1270                                      &key_data_length) < 0) {
1271                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
1272                                         "received EAPOL-Key with invalid MIC");
1273 #ifdef TEST_FUZZ
1274                         wpa_printf(MSG_INFO,
1275                                    "TEST: Ignore Key MIC failure for fuzz testing");
1276                         goto continue_fuzz;
1277 #endif /* TEST_FUZZ */
1278                         return;
1279                 }
1280 #endif /* CONFIG_FILS */
1281 #ifdef TEST_FUZZ
1282         continue_fuzz:
1283 #endif /* TEST_FUZZ */
1284                 sm->MICVerified = TRUE;
1285                 eloop_cancel_timeout(wpa_send_eapol_timeout, wpa_auth, sm);
1286                 sm->pending_1_of_4_timeout = 0;
1287         }
1288
1289         if (key_info & WPA_KEY_INFO_REQUEST) {
1290                 if (sm->MICVerified) {
1291                         sm->req_replay_counter_used = 1;
1292                         os_memcpy(sm->req_replay_counter, key->replay_counter,
1293                                   WPA_REPLAY_COUNTER_LEN);
1294                 } else {
1295                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
1296                                         "received EAPOL-Key request with "
1297                                         "invalid MIC");
1298                         return;
1299                 }
1300
1301                 /*
1302                  * TODO: should decrypt key data field if encryption was used;
1303                  * even though MAC address KDE is not normally encrypted,
1304                  * supplicant is allowed to encrypt it.
1305                  */
1306                 if (key_info & WPA_KEY_INFO_ERROR) {
1307                         if (wpa_receive_error_report(
1308                                     wpa_auth, sm,
1309                                     !(key_info & WPA_KEY_INFO_KEY_TYPE)) > 0)
1310                                 return; /* STA entry was removed */
1311                 } else if (key_info & WPA_KEY_INFO_KEY_TYPE) {
1312                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
1313                                         "received EAPOL-Key Request for new "
1314                                         "4-Way Handshake");
1315                         wpa_request_new_ptk(sm);
1316                 } else if (key_data_length > 0 &&
1317                            wpa_parse_kde_ies(key_data, key_data_length,
1318                                              &kde) == 0 &&
1319                            kde.mac_addr) {
1320                 } else {
1321                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
1322                                         "received EAPOL-Key Request for GTK "
1323                                         "rekeying");
1324                         eloop_cancel_timeout(wpa_rekey_gtk, wpa_auth, NULL);
1325                         wpa_rekey_gtk(wpa_auth, NULL);
1326                 }
1327         } else {
1328                 /* Do not allow the same key replay counter to be reused. */
1329                 wpa_replay_counter_mark_invalid(sm->key_replay,
1330                                                 key->replay_counter);
1331
1332                 if (msg == PAIRWISE_2) {
1333                         /*
1334                          * Maintain a copy of the pending EAPOL-Key frames in
1335                          * case the EAPOL-Key frame was retransmitted. This is
1336                          * needed to allow EAPOL-Key msg 2/4 reply to another
1337                          * pending msg 1/4 to update the SNonce to work around
1338                          * unexpected supplicant behavior.
1339                          */
1340                         os_memcpy(sm->prev_key_replay, sm->key_replay,
1341                                   sizeof(sm->key_replay));
1342                 } else {
1343                         os_memset(sm->prev_key_replay, 0,
1344                                   sizeof(sm->prev_key_replay));
1345                 }
1346
1347                 /*
1348                  * Make sure old valid counters are not accepted anymore and
1349                  * do not get copied again.
1350                  */
1351                 wpa_replay_counter_mark_invalid(sm->key_replay, NULL);
1352         }
1353
1354         os_free(sm->last_rx_eapol_key);
1355         sm->last_rx_eapol_key = os_memdup(data, data_len);
1356         if (sm->last_rx_eapol_key == NULL)
1357                 return;
1358         sm->last_rx_eapol_key_len = data_len;
1359
1360         sm->rx_eapol_key_secure = !!(key_info & WPA_KEY_INFO_SECURE);
1361         sm->EAPOLKeyReceived = TRUE;
1362         sm->EAPOLKeyPairwise = !!(key_info & WPA_KEY_INFO_KEY_TYPE);
1363         sm->EAPOLKeyRequest = !!(key_info & WPA_KEY_INFO_REQUEST);
1364         os_memcpy(sm->SNonce, key->key_nonce, WPA_NONCE_LEN);
1365         wpa_sm_step(sm);
1366 }
1367
1368
1369 static int wpa_gmk_to_gtk(const u8 *gmk, const char *label, const u8 *addr,
1370                           const u8 *gnonce, u8 *gtk, size_t gtk_len)
1371 {
1372         u8 data[ETH_ALEN + WPA_NONCE_LEN + 8 + WPA_GTK_MAX_LEN];
1373         u8 *pos;
1374         int ret = 0;
1375
1376         /* GTK = PRF-X(GMK, "Group key expansion",
1377          *      AA || GNonce || Time || random data)
1378          * The example described in the IEEE 802.11 standard uses only AA and
1379          * GNonce as inputs here. Add some more entropy since this derivation
1380          * is done only at the Authenticator and as such, does not need to be
1381          * exactly same.
1382          */
1383         os_memset(data, 0, sizeof(data));
1384         os_memcpy(data, addr, ETH_ALEN);
1385         os_memcpy(data + ETH_ALEN, gnonce, WPA_NONCE_LEN);
1386         pos = data + ETH_ALEN + WPA_NONCE_LEN;
1387         wpa_get_ntp_timestamp(pos);
1388 #ifdef TEST_FUZZ
1389         os_memset(pos, 0xef, 8);
1390 #endif /* TEST_FUZZ */
1391         pos += 8;
1392         if (random_get_bytes(pos, gtk_len) < 0)
1393                 ret = -1;
1394
1395 #ifdef CONFIG_SHA384
1396         if (sha384_prf(gmk, WPA_GMK_LEN, label, data, sizeof(data),
1397                        gtk, gtk_len) < 0)
1398                 ret = -1;
1399 #else /* CONFIG_SHA384 */
1400 #ifdef CONFIG_SHA256
1401         if (sha256_prf(gmk, WPA_GMK_LEN, label, data, sizeof(data),
1402                        gtk, gtk_len) < 0)
1403                 ret = -1;
1404 #else /* CONFIG_SHA256 */
1405         if (sha1_prf(gmk, WPA_GMK_LEN, label, data, sizeof(data),
1406                      gtk, gtk_len) < 0)
1407                 ret = -1;
1408 #endif /* CONFIG_SHA256 */
1409 #endif /* CONFIG_SHA384 */
1410
1411         forced_memzero(data, sizeof(data));
1412
1413         return ret;
1414 }
1415
1416
1417 static void wpa_send_eapol_timeout(void *eloop_ctx, void *timeout_ctx)
1418 {
1419         struct wpa_authenticator *wpa_auth = eloop_ctx;
1420         struct wpa_state_machine *sm = timeout_ctx;
1421
1422         sm->pending_1_of_4_timeout = 0;
1423         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_DEBUG, "EAPOL-Key timeout");
1424         sm->TimeoutEvt = TRUE;
1425         wpa_sm_step(sm);
1426 }
1427
1428
1429 void __wpa_send_eapol(struct wpa_authenticator *wpa_auth,
1430                       struct wpa_state_machine *sm, int key_info,
1431                       const u8 *key_rsc, const u8 *nonce,
1432                       const u8 *kde, size_t kde_len,
1433                       int keyidx, int encr, int force_version)
1434 {
1435         struct ieee802_1x_hdr *hdr;
1436         struct wpa_eapol_key *key;
1437         size_t len, mic_len, keyhdrlen;
1438         int alg;
1439         int key_data_len, pad_len = 0;
1440         u8 *buf, *pos;
1441         int version, pairwise;
1442         int i;
1443         u8 *key_mic, *key_data;
1444
1445         mic_len = wpa_mic_len(sm->wpa_key_mgmt, sm->pmk_len);
1446         keyhdrlen = sizeof(*key) + mic_len + 2;
1447
1448         len = sizeof(struct ieee802_1x_hdr) + keyhdrlen;
1449
1450         if (force_version)
1451                 version = force_version;
1452         else if (wpa_use_akm_defined(sm->wpa_key_mgmt))
1453                 version = WPA_KEY_INFO_TYPE_AKM_DEFINED;
1454         else if (wpa_use_cmac(sm->wpa_key_mgmt))
1455                 version = WPA_KEY_INFO_TYPE_AES_128_CMAC;
1456         else if (sm->pairwise != WPA_CIPHER_TKIP)
1457                 version = WPA_KEY_INFO_TYPE_HMAC_SHA1_AES;
1458         else
1459                 version = WPA_KEY_INFO_TYPE_HMAC_MD5_RC4;
1460
1461         pairwise = !!(key_info & WPA_KEY_INFO_KEY_TYPE);
1462
1463         wpa_printf(MSG_DEBUG, "WPA: Send EAPOL(version=%d secure=%d mic=%d "
1464                    "ack=%d install=%d pairwise=%d kde_len=%lu keyidx=%d "
1465                    "encr=%d)",
1466                    version,
1467                    (key_info & WPA_KEY_INFO_SECURE) ? 1 : 0,
1468                    (key_info & WPA_KEY_INFO_MIC) ? 1 : 0,
1469                    (key_info & WPA_KEY_INFO_ACK) ? 1 : 0,
1470                    (key_info & WPA_KEY_INFO_INSTALL) ? 1 : 0,
1471                    pairwise, (unsigned long) kde_len, keyidx, encr);
1472
1473         key_data_len = kde_len;
1474
1475         if ((version == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES ||
1476              wpa_use_aes_key_wrap(sm->wpa_key_mgmt) ||
1477              version == WPA_KEY_INFO_TYPE_AES_128_CMAC) && encr) {
1478                 pad_len = key_data_len % 8;
1479                 if (pad_len)
1480                         pad_len = 8 - pad_len;
1481                 key_data_len += pad_len + 8;
1482         }
1483
1484         len += key_data_len;
1485         if (!mic_len && encr)
1486                 len += AES_BLOCK_SIZE;
1487
1488         hdr = os_zalloc(len);
1489         if (hdr == NULL)
1490                 return;
1491         hdr->version = wpa_auth->conf.eapol_version;
1492         hdr->type = IEEE802_1X_TYPE_EAPOL_KEY;
1493         hdr->length = host_to_be16(len  - sizeof(*hdr));
1494         key = (struct wpa_eapol_key *) (hdr + 1);
1495         key_mic = (u8 *) (key + 1);
1496         key_data = ((u8 *) (hdr + 1)) + keyhdrlen;
1497
1498         key->type = sm->wpa == WPA_VERSION_WPA2 ?
1499                 EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
1500         key_info |= version;
1501         if (encr && sm->wpa == WPA_VERSION_WPA2)
1502                 key_info |= WPA_KEY_INFO_ENCR_KEY_DATA;
1503         if (sm->wpa != WPA_VERSION_WPA2)
1504                 key_info |= keyidx << WPA_KEY_INFO_KEY_INDEX_SHIFT;
1505         WPA_PUT_BE16(key->key_info, key_info);
1506
1507         alg = pairwise ? sm->pairwise : wpa_auth->conf.wpa_group;
1508         if (sm->wpa == WPA_VERSION_WPA2 && !pairwise)
1509                 WPA_PUT_BE16(key->key_length, 0);
1510         else
1511                 WPA_PUT_BE16(key->key_length, wpa_cipher_key_len(alg));
1512
1513         for (i = RSNA_MAX_EAPOL_RETRIES - 1; i > 0; i--) {
1514                 sm->key_replay[i].valid = sm->key_replay[i - 1].valid;
1515                 os_memcpy(sm->key_replay[i].counter,
1516                           sm->key_replay[i - 1].counter,
1517                           WPA_REPLAY_COUNTER_LEN);
1518         }
1519         inc_byte_array(sm->key_replay[0].counter, WPA_REPLAY_COUNTER_LEN);
1520         os_memcpy(key->replay_counter, sm->key_replay[0].counter,
1521                   WPA_REPLAY_COUNTER_LEN);
1522         wpa_hexdump(MSG_DEBUG, "WPA: Replay Counter",
1523                     key->replay_counter, WPA_REPLAY_COUNTER_LEN);
1524         sm->key_replay[0].valid = TRUE;
1525
1526         if (nonce)
1527                 os_memcpy(key->key_nonce, nonce, WPA_NONCE_LEN);
1528
1529         if (key_rsc)
1530                 os_memcpy(key->key_rsc, key_rsc, WPA_KEY_RSC_LEN);
1531
1532         if (kde && !encr) {
1533                 os_memcpy(key_data, kde, kde_len);
1534                 WPA_PUT_BE16(key_mic + mic_len, kde_len);
1535 #ifdef CONFIG_FILS
1536         } else if (!mic_len && kde) {
1537                 const u8 *aad[1];
1538                 size_t aad_len[1];
1539
1540                 WPA_PUT_BE16(key_mic, AES_BLOCK_SIZE + kde_len);
1541                 wpa_hexdump_key(MSG_DEBUG, "Plaintext EAPOL-Key Key Data",
1542                                 kde, kde_len);
1543
1544                 wpa_hexdump_key(MSG_DEBUG, "WPA: KEK",
1545                                 sm->PTK.kek, sm->PTK.kek_len);
1546                 /* AES-SIV AAD from EAPOL protocol version field (inclusive) to
1547                  * to Key Data (exclusive). */
1548                 aad[0] = (u8 *) hdr;
1549                 aad_len[0] = key_mic + 2 - (u8 *) hdr;
1550                 if (aes_siv_encrypt(sm->PTK.kek, sm->PTK.kek_len, kde, kde_len,
1551                                     1, aad, aad_len, key_mic + 2) < 0) {
1552                         wpa_printf(MSG_DEBUG, "WPA: AES-SIV encryption failed");
1553                         return;
1554                 }
1555
1556                 wpa_hexdump(MSG_DEBUG, "WPA: Encrypted Key Data from SIV",
1557                             key_mic + 2, AES_BLOCK_SIZE + kde_len);
1558 #endif /* CONFIG_FILS */
1559         } else if (encr && kde) {
1560                 buf = os_zalloc(key_data_len);
1561                 if (buf == NULL) {
1562                         os_free(hdr);
1563                         return;
1564                 }
1565                 pos = buf;
1566                 os_memcpy(pos, kde, kde_len);
1567                 pos += kde_len;
1568
1569                 if (pad_len)
1570                         *pos++ = 0xdd;
1571
1572                 wpa_hexdump_key(MSG_DEBUG, "Plaintext EAPOL-Key Key Data",
1573                                 buf, key_data_len);
1574                 if (version == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES ||
1575                     wpa_use_aes_key_wrap(sm->wpa_key_mgmt) ||
1576                     version == WPA_KEY_INFO_TYPE_AES_128_CMAC) {
1577                         wpa_printf(MSG_DEBUG,
1578                                    "WPA: Encrypt Key Data using AES-WRAP (KEK length %u)",
1579                                    (unsigned int) sm->PTK.kek_len);
1580                         if (aes_wrap(sm->PTK.kek, sm->PTK.kek_len,
1581                                      (key_data_len - 8) / 8, buf, key_data)) {
1582                                 os_free(hdr);
1583                                 os_free(buf);
1584                                 return;
1585                         }
1586                         WPA_PUT_BE16(key_mic + mic_len, key_data_len);
1587 #ifndef CONFIG_NO_RC4
1588                 } else if (sm->PTK.kek_len == 16) {
1589                         u8 ek[32];
1590
1591                         wpa_printf(MSG_DEBUG,
1592                                    "WPA: Encrypt Key Data using RC4");
1593                         os_memcpy(key->key_iv,
1594                                   sm->group->Counter + WPA_NONCE_LEN - 16, 16);
1595                         inc_byte_array(sm->group->Counter, WPA_NONCE_LEN);
1596                         os_memcpy(ek, key->key_iv, 16);
1597                         os_memcpy(ek + 16, sm->PTK.kek, sm->PTK.kek_len);
1598                         os_memcpy(key_data, buf, key_data_len);
1599                         rc4_skip(ek, 32, 256, key_data, key_data_len);
1600                         WPA_PUT_BE16(key_mic + mic_len, key_data_len);
1601 #endif /* CONFIG_NO_RC4 */
1602                 } else {
1603                         os_free(hdr);
1604                         os_free(buf);
1605                         return;
1606                 }
1607                 os_free(buf);
1608         }
1609
1610         if (key_info & WPA_KEY_INFO_MIC) {
1611                 if (!sm->PTK_valid || !mic_len) {
1612                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_DEBUG,
1613                                         "PTK not valid when sending EAPOL-Key "
1614                                         "frame");
1615                         os_free(hdr);
1616                         return;
1617                 }
1618
1619                 if (wpa_eapol_key_mic(sm->PTK.kck, sm->PTK.kck_len,
1620                                       sm->wpa_key_mgmt, version,
1621                                       (u8 *) hdr, len, key_mic) < 0) {
1622                         os_free(hdr);
1623                         return;
1624                 }
1625 #ifdef CONFIG_TESTING_OPTIONS
1626                 if (!pairwise &&
1627                     wpa_auth->conf.corrupt_gtk_rekey_mic_probability > 0.0 &&
1628                     drand48() <
1629                     wpa_auth->conf.corrupt_gtk_rekey_mic_probability) {
1630                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
1631                                         "Corrupting group EAPOL-Key Key MIC");
1632                         key_mic[0]++;
1633                 }
1634 #endif /* CONFIG_TESTING_OPTIONS */
1635         }
1636
1637         wpa_auth_set_eapol(sm->wpa_auth, sm->addr, WPA_EAPOL_inc_EapolFramesTx,
1638                            1);
1639         wpa_auth_send_eapol(wpa_auth, sm->addr, (u8 *) hdr, len,
1640                             sm->pairwise_set);
1641         os_free(hdr);
1642 }
1643
1644
1645 static void wpa_send_eapol(struct wpa_authenticator *wpa_auth,
1646                            struct wpa_state_machine *sm, int key_info,
1647                            const u8 *key_rsc, const u8 *nonce,
1648                            const u8 *kde, size_t kde_len,
1649                            int keyidx, int encr)
1650 {
1651         int timeout_ms;
1652         int pairwise = key_info & WPA_KEY_INFO_KEY_TYPE;
1653         u32 ctr;
1654
1655         if (sm == NULL)
1656                 return;
1657
1658         __wpa_send_eapol(wpa_auth, sm, key_info, key_rsc, nonce, kde, kde_len,
1659                          keyidx, encr, 0);
1660
1661         ctr = pairwise ? sm->TimeoutCtr : sm->GTimeoutCtr;
1662         if (ctr == 1 && wpa_auth->conf.tx_status)
1663                 timeout_ms = pairwise ? eapol_key_timeout_first :
1664                         eapol_key_timeout_first_group;
1665         else
1666                 timeout_ms = eapol_key_timeout_subseq;
1667         if (wpa_auth->conf.wpa_disable_eapol_key_retries &&
1668             (!pairwise || (key_info & WPA_KEY_INFO_MIC)))
1669                 timeout_ms = eapol_key_timeout_no_retrans;
1670         if (pairwise && ctr == 1 && !(key_info & WPA_KEY_INFO_MIC))
1671                 sm->pending_1_of_4_timeout = 1;
1672 #ifdef TEST_FUZZ
1673         timeout_ms = 1;
1674 #endif /* TEST_FUZZ */
1675         wpa_printf(MSG_DEBUG, "WPA: Use EAPOL-Key timeout of %u ms (retry "
1676                    "counter %u)", timeout_ms, ctr);
1677         eloop_register_timeout(timeout_ms / 1000, (timeout_ms % 1000) * 1000,
1678                                wpa_send_eapol_timeout, wpa_auth, sm);
1679 }
1680
1681
1682 static int wpa_verify_key_mic(int akmp, size_t pmk_len, struct wpa_ptk *PTK,
1683                               u8 *data, size_t data_len)
1684 {
1685         struct ieee802_1x_hdr *hdr;
1686         struct wpa_eapol_key *key;
1687         u16 key_info;
1688         int ret = 0;
1689         u8 mic[WPA_EAPOL_KEY_MIC_MAX_LEN], *mic_pos;
1690         size_t mic_len = wpa_mic_len(akmp, pmk_len);
1691
1692         if (data_len < sizeof(*hdr) + sizeof(*key))
1693                 return -1;
1694
1695         hdr = (struct ieee802_1x_hdr *) data;
1696         key = (struct wpa_eapol_key *) (hdr + 1);
1697         mic_pos = (u8 *) (key + 1);
1698         key_info = WPA_GET_BE16(key->key_info);
1699         os_memcpy(mic, mic_pos, mic_len);
1700         os_memset(mic_pos, 0, mic_len);
1701         if (wpa_eapol_key_mic(PTK->kck, PTK->kck_len, akmp,
1702                               key_info & WPA_KEY_INFO_TYPE_MASK,
1703                               data, data_len, mic_pos) ||
1704             os_memcmp_const(mic, mic_pos, mic_len) != 0)
1705                 ret = -1;
1706         os_memcpy(mic_pos, mic, mic_len);
1707         return ret;
1708 }
1709
1710
1711 void wpa_remove_ptk(struct wpa_state_machine *sm)
1712 {
1713         sm->PTK_valid = FALSE;
1714         os_memset(&sm->PTK, 0, sizeof(sm->PTK));
1715         if (wpa_auth_set_key(sm->wpa_auth, 0, WPA_ALG_NONE, sm->addr, 0, NULL,
1716                              0))
1717                 wpa_printf(MSG_DEBUG,
1718                            "RSN: PTK removal from the driver failed");
1719         sm->pairwise_set = FALSE;
1720         eloop_cancel_timeout(wpa_rekey_ptk, sm->wpa_auth, sm);
1721 }
1722
1723
1724 int wpa_auth_sm_event(struct wpa_state_machine *sm, enum wpa_event event)
1725 {
1726         int remove_ptk = 1;
1727
1728         if (sm == NULL)
1729                 return -1;
1730
1731         wpa_auth_vlogger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
1732                          "event %d notification", event);
1733
1734         switch (event) {
1735         case WPA_AUTH:
1736 #ifdef CONFIG_MESH
1737                 /* PTKs are derived through AMPE */
1738                 if (wpa_auth_start_ampe(sm->wpa_auth, sm->addr)) {
1739                         /* not mesh */
1740                         break;
1741                 }
1742                 return 0;
1743 #endif /* CONFIG_MESH */
1744         case WPA_ASSOC:
1745                 break;
1746         case WPA_DEAUTH:
1747         case WPA_DISASSOC:
1748                 sm->DeauthenticationRequest = TRUE;
1749 #ifdef CONFIG_IEEE80211R_AP
1750                 os_memset(sm->PMK, 0, sizeof(sm->PMK));
1751                 sm->pmk_len = 0;
1752                 os_memset(sm->xxkey, 0, sizeof(sm->xxkey));
1753                 sm->xxkey_len = 0;
1754                 os_memset(sm->pmk_r1, 0, sizeof(sm->pmk_r1));
1755                 sm->pmk_r1_len = 0;
1756 #endif /* CONFIG_IEEE80211R_AP */
1757                 break;
1758         case WPA_REAUTH:
1759         case WPA_REAUTH_EAPOL:
1760                 if (!sm->started) {
1761                         /*
1762                          * When using WPS, we may end up here if the STA
1763                          * manages to re-associate without the previous STA
1764                          * entry getting removed. Consequently, we need to make
1765                          * sure that the WPA state machines gets initialized
1766                          * properly at this point.
1767                          */
1768                         wpa_printf(MSG_DEBUG, "WPA state machine had not been "
1769                                    "started - initialize now");
1770                         sm->started = 1;
1771                         sm->Init = TRUE;
1772                         if (wpa_sm_step(sm) == 1)
1773                                 return 1; /* should not really happen */
1774                         sm->Init = FALSE;
1775                         sm->AuthenticationRequest = TRUE;
1776                         break;
1777                 }
1778                 if (sm->GUpdateStationKeys) {
1779                         /*
1780                          * Reauthentication cancels the pending group key
1781                          * update for this STA.
1782                          */
1783                         sm->group->GKeyDoneStations--;
1784                         sm->GUpdateStationKeys = FALSE;
1785                         sm->PtkGroupInit = TRUE;
1786                 }
1787                 sm->ReAuthenticationRequest = TRUE;
1788                 break;
1789         case WPA_ASSOC_FT:
1790 #ifdef CONFIG_IEEE80211R_AP
1791                 wpa_printf(MSG_DEBUG, "FT: Retry PTK configuration "
1792                            "after association");
1793                 wpa_ft_install_ptk(sm);
1794
1795                 /* Using FT protocol, not WPA auth state machine */
1796                 sm->ft_completed = 1;
1797                 wpa_auth_set_ptk_rekey_timer(sm);
1798                 return 0;
1799 #else /* CONFIG_IEEE80211R_AP */
1800                 break;
1801 #endif /* CONFIG_IEEE80211R_AP */
1802         case WPA_ASSOC_FILS:
1803 #ifdef CONFIG_FILS
1804                 wpa_printf(MSG_DEBUG,
1805                            "FILS: TK configuration after association");
1806                 fils_set_tk(sm);
1807                 sm->fils_completed = 1;
1808                 return 0;
1809 #else /* CONFIG_FILS */
1810                 break;
1811 #endif /* CONFIG_FILS */
1812         case WPA_DRV_STA_REMOVED:
1813                 sm->tk_already_set = FALSE;
1814                 return 0;
1815         }
1816
1817 #ifdef CONFIG_IEEE80211R_AP
1818         sm->ft_completed = 0;
1819 #endif /* CONFIG_IEEE80211R_AP */
1820
1821 #ifdef CONFIG_IEEE80211W
1822         if (sm->mgmt_frame_prot && event == WPA_AUTH)
1823                 remove_ptk = 0;
1824 #endif /* CONFIG_IEEE80211W */
1825 #ifdef CONFIG_FILS
1826         if (wpa_key_mgmt_fils(sm->wpa_key_mgmt) &&
1827             (event == WPA_AUTH || event == WPA_ASSOC))
1828                 remove_ptk = 0;
1829 #endif /* CONFIG_FILS */
1830
1831         if (remove_ptk) {
1832                 sm->PTK_valid = FALSE;
1833                 os_memset(&sm->PTK, 0, sizeof(sm->PTK));
1834
1835                 if (event != WPA_REAUTH_EAPOL)
1836                         wpa_remove_ptk(sm);
1837         }
1838
1839         if (sm->in_step_loop) {
1840                 /*
1841                  * wpa_sm_step() is already running - avoid recursive call to
1842                  * it by making the existing loop process the new update.
1843                  */
1844                 sm->changed = TRUE;
1845                 return 0;
1846         }
1847         return wpa_sm_step(sm);
1848 }
1849
1850
1851 SM_STATE(WPA_PTK, INITIALIZE)
1852 {
1853         SM_ENTRY_MA(WPA_PTK, INITIALIZE, wpa_ptk);
1854         if (sm->Init) {
1855                 /* Init flag is not cleared here, so avoid busy
1856                  * loop by claiming nothing changed. */
1857                 sm->changed = FALSE;
1858         }
1859
1860         sm->keycount = 0;
1861         if (sm->GUpdateStationKeys)
1862                 sm->group->GKeyDoneStations--;
1863         sm->GUpdateStationKeys = FALSE;
1864         if (sm->wpa == WPA_VERSION_WPA)
1865                 sm->PInitAKeys = FALSE;
1866         if (1 /* Unicast cipher supported AND (ESS OR ((IBSS or WDS) and
1867                * Local AA > Remote AA)) */) {
1868                 sm->Pair = TRUE;
1869         }
1870         wpa_auth_set_eapol(sm->wpa_auth, sm->addr, WPA_EAPOL_portEnabled, 0);
1871         wpa_remove_ptk(sm);
1872         wpa_auth_set_eapol(sm->wpa_auth, sm->addr, WPA_EAPOL_portValid, 0);
1873         sm->TimeoutCtr = 0;
1874         if (wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt) ||
1875             sm->wpa_key_mgmt == WPA_KEY_MGMT_DPP ||
1876             sm->wpa_key_mgmt == WPA_KEY_MGMT_OWE) {
1877                 wpa_auth_set_eapol(sm->wpa_auth, sm->addr,
1878                                    WPA_EAPOL_authorized, 0);
1879         }
1880 }
1881
1882
1883 SM_STATE(WPA_PTK, DISCONNECT)
1884 {
1885         u16 reason = sm->disconnect_reason;
1886
1887         SM_ENTRY_MA(WPA_PTK, DISCONNECT, wpa_ptk);
1888         sm->Disconnect = FALSE;
1889         sm->disconnect_reason = 0;
1890         if (!reason)
1891                 reason = WLAN_REASON_PREV_AUTH_NOT_VALID;
1892         wpa_sta_disconnect(sm->wpa_auth, sm->addr, reason);
1893 }
1894
1895
1896 SM_STATE(WPA_PTK, DISCONNECTED)
1897 {
1898         SM_ENTRY_MA(WPA_PTK, DISCONNECTED, wpa_ptk);
1899         sm->DeauthenticationRequest = FALSE;
1900 }
1901
1902
1903 SM_STATE(WPA_PTK, AUTHENTICATION)
1904 {
1905         SM_ENTRY_MA(WPA_PTK, AUTHENTICATION, wpa_ptk);
1906         os_memset(&sm->PTK, 0, sizeof(sm->PTK));
1907         sm->PTK_valid = FALSE;
1908         wpa_auth_set_eapol(sm->wpa_auth, sm->addr, WPA_EAPOL_portControl_Auto,
1909                            1);
1910         wpa_auth_set_eapol(sm->wpa_auth, sm->addr, WPA_EAPOL_portEnabled, 1);
1911         sm->AuthenticationRequest = FALSE;
1912 }
1913
1914
1915 static void wpa_group_ensure_init(struct wpa_authenticator *wpa_auth,
1916                                   struct wpa_group *group)
1917 {
1918         if (group->first_sta_seen)
1919                 return;
1920         /*
1921          * System has run bit further than at the time hostapd was started
1922          * potentially very early during boot up. This provides better chances
1923          * of collecting more randomness on embedded systems. Re-initialize the
1924          * GMK and Counter here to improve their strength if there was not
1925          * enough entropy available immediately after system startup.
1926          */
1927         wpa_printf(MSG_DEBUG, "WPA: Re-initialize GMK/Counter on first "
1928                    "station");
1929         if (random_pool_ready() != 1) {
1930                 wpa_printf(MSG_INFO, "WPA: Not enough entropy in random pool "
1931                            "to proceed - reject first 4-way handshake");
1932                 group->reject_4way_hs_for_entropy = TRUE;
1933         } else {
1934                 group->first_sta_seen = TRUE;
1935                 group->reject_4way_hs_for_entropy = FALSE;
1936         }
1937
1938         if (wpa_group_init_gmk_and_counter(wpa_auth, group) < 0 ||
1939             wpa_gtk_update(wpa_auth, group) < 0 ||
1940             wpa_group_config_group_keys(wpa_auth, group) < 0) {
1941                 wpa_printf(MSG_INFO, "WPA: GMK/GTK setup failed");
1942                 group->first_sta_seen = FALSE;
1943                 group->reject_4way_hs_for_entropy = TRUE;
1944         }
1945 }
1946
1947
1948 SM_STATE(WPA_PTK, AUTHENTICATION2)
1949 {
1950         SM_ENTRY_MA(WPA_PTK, AUTHENTICATION2, wpa_ptk);
1951
1952         wpa_group_ensure_init(sm->wpa_auth, sm->group);
1953         sm->ReAuthenticationRequest = FALSE;
1954
1955         /*
1956          * Definition of ANonce selection in IEEE Std 802.11i-2004 is somewhat
1957          * ambiguous. The Authenticator state machine uses a counter that is
1958          * incremented by one for each 4-way handshake. However, the security
1959          * analysis of 4-way handshake points out that unpredictable nonces
1960          * help in preventing precomputation attacks. Instead of the state
1961          * machine definition, use an unpredictable nonce value here to provide
1962          * stronger protection against potential precomputation attacks.
1963          */
1964         if (random_get_bytes(sm->ANonce, WPA_NONCE_LEN)) {
1965                 wpa_printf(MSG_ERROR, "WPA: Failed to get random data for "
1966                            "ANonce.");
1967                 sm->Disconnect = TRUE;
1968                 return;
1969         }
1970         wpa_hexdump(MSG_DEBUG, "WPA: Assign ANonce", sm->ANonce,
1971                     WPA_NONCE_LEN);
1972         /* IEEE 802.11i does not clear TimeoutCtr here, but this is more
1973          * logical place than INITIALIZE since AUTHENTICATION2 can be
1974          * re-entered on ReAuthenticationRequest without going through
1975          * INITIALIZE. */
1976         sm->TimeoutCtr = 0;
1977 }
1978
1979
1980 static int wpa_auth_sm_ptk_update(struct wpa_state_machine *sm)
1981 {
1982         if (random_get_bytes(sm->ANonce, WPA_NONCE_LEN)) {
1983                 wpa_printf(MSG_ERROR,
1984                            "WPA: Failed to get random data for ANonce");
1985                 sm->Disconnect = TRUE;
1986                 return -1;
1987         }
1988         wpa_hexdump(MSG_DEBUG, "WPA: Assign new ANonce", sm->ANonce,
1989                     WPA_NONCE_LEN);
1990         sm->TimeoutCtr = 0;
1991         return 0;
1992 }
1993
1994
1995 SM_STATE(WPA_PTK, INITPMK)
1996 {
1997         u8 msk[2 * PMK_LEN];
1998         size_t len = 2 * PMK_LEN;
1999
2000         SM_ENTRY_MA(WPA_PTK, INITPMK, wpa_ptk);
2001 #ifdef CONFIG_IEEE80211R_AP
2002         sm->xxkey_len = 0;
2003 #endif /* CONFIG_IEEE80211R_AP */
2004         if (sm->pmksa) {
2005                 wpa_printf(MSG_DEBUG, "WPA: PMK from PMKSA cache");
2006                 os_memcpy(sm->PMK, sm->pmksa->pmk, sm->pmksa->pmk_len);
2007                 sm->pmk_len = sm->pmksa->pmk_len;
2008 #ifdef CONFIG_DPP
2009         } else if (sm->wpa_key_mgmt == WPA_KEY_MGMT_DPP) {
2010                 wpa_printf(MSG_DEBUG,
2011                            "DPP: No PMKSA cache entry for STA - reject connection");
2012                 sm->Disconnect = TRUE;
2013                 sm->disconnect_reason = WLAN_REASON_INVALID_PMKID;
2014                 return;
2015 #endif /* CONFIG_DPP */
2016         } else if (wpa_auth_get_msk(sm->wpa_auth, sm->addr, msk, &len) == 0) {
2017                 unsigned int pmk_len;
2018
2019                 if (wpa_key_mgmt_sha384(sm->wpa_key_mgmt))
2020                         pmk_len = PMK_LEN_SUITE_B_192;
2021                 else
2022                         pmk_len = PMK_LEN;
2023                 wpa_printf(MSG_DEBUG, "WPA: PMK from EAPOL state machine "
2024                            "(MSK len=%lu PMK len=%u)", (unsigned long) len,
2025                            pmk_len);
2026                 if (len < pmk_len) {
2027                         wpa_printf(MSG_DEBUG,
2028                                    "WPA: MSK not long enough (%u) to create PMK (%u)",
2029                                    (unsigned int) len, (unsigned int) pmk_len);
2030                         sm->Disconnect = TRUE;
2031                         return;
2032                 }
2033                 os_memcpy(sm->PMK, msk, pmk_len);
2034                 sm->pmk_len = pmk_len;
2035 #ifdef CONFIG_IEEE80211R_AP
2036                 if (len >= 2 * PMK_LEN) {
2037                         if (wpa_key_mgmt_sha384(sm->wpa_key_mgmt)) {
2038                                 os_memcpy(sm->xxkey, msk, SHA384_MAC_LEN);
2039                                 sm->xxkey_len = SHA384_MAC_LEN;
2040                         } else {
2041                                 os_memcpy(sm->xxkey, msk + PMK_LEN, PMK_LEN);
2042                                 sm->xxkey_len = PMK_LEN;
2043                         }
2044                 }
2045 #endif /* CONFIG_IEEE80211R_AP */
2046         } else {
2047                 wpa_printf(MSG_DEBUG, "WPA: Could not get PMK, get_msk: %p",
2048                            sm->wpa_auth->cb->get_msk);
2049                 sm->Disconnect = TRUE;
2050                 return;
2051         }
2052         forced_memzero(msk, sizeof(msk));
2053
2054         sm->req_replay_counter_used = 0;
2055         /* IEEE 802.11i does not set keyRun to FALSE, but not doing this
2056          * will break reauthentication since EAPOL state machines may not be
2057          * get into AUTHENTICATING state that clears keyRun before WPA state
2058          * machine enters AUTHENTICATION2 state and goes immediately to INITPMK
2059          * state and takes PMK from the previously used AAA Key. This will
2060          * eventually fail in 4-Way Handshake because Supplicant uses PMK
2061          * derived from the new AAA Key. Setting keyRun = FALSE here seems to
2062          * be good workaround for this issue. */
2063         wpa_auth_set_eapol(sm->wpa_auth, sm->addr, WPA_EAPOL_keyRun, 0);
2064 }
2065
2066
2067 SM_STATE(WPA_PTK, INITPSK)
2068 {
2069         const u8 *psk;
2070         size_t psk_len;
2071
2072         SM_ENTRY_MA(WPA_PTK, INITPSK, wpa_ptk);
2073         psk = wpa_auth_get_psk(sm->wpa_auth, sm->addr, sm->p2p_dev_addr, NULL,
2074                                &psk_len, NULL);
2075         if (psk) {
2076                 os_memcpy(sm->PMK, psk, psk_len);
2077                 sm->pmk_len = psk_len;
2078 #ifdef CONFIG_IEEE80211R_AP
2079                 os_memcpy(sm->xxkey, psk, PMK_LEN);
2080                 sm->xxkey_len = PMK_LEN;
2081 #endif /* CONFIG_IEEE80211R_AP */
2082         }
2083 #ifdef CONFIG_SAE
2084         if (wpa_auth_uses_sae(sm) && sm->pmksa) {
2085                 wpa_printf(MSG_DEBUG, "SAE: PMK from PMKSA cache");
2086                 os_memcpy(sm->PMK, sm->pmksa->pmk, sm->pmksa->pmk_len);
2087                 sm->pmk_len = sm->pmksa->pmk_len;
2088 #ifdef CONFIG_IEEE80211R_AP
2089                 os_memcpy(sm->xxkey, sm->pmksa->pmk, sm->pmksa->pmk_len);
2090                 sm->xxkey_len = sm->pmksa->pmk_len;
2091 #endif /* CONFIG_IEEE80211R_AP */
2092         }
2093 #endif /* CONFIG_SAE */
2094         sm->req_replay_counter_used = 0;
2095 }
2096
2097
2098 SM_STATE(WPA_PTK, PTKSTART)
2099 {
2100         u8 buf[2 + RSN_SELECTOR_LEN + PMKID_LEN], *pmkid = NULL;
2101         size_t pmkid_len = 0;
2102
2103         SM_ENTRY_MA(WPA_PTK, PTKSTART, wpa_ptk);
2104         sm->PTKRequest = FALSE;
2105         sm->TimeoutEvt = FALSE;
2106         sm->alt_snonce_valid = FALSE;
2107
2108         sm->TimeoutCtr++;
2109         if (sm->TimeoutCtr > sm->wpa_auth->conf.wpa_pairwise_update_count) {
2110                 /* No point in sending the EAPOL-Key - we will disconnect
2111                  * immediately following this. */
2112                 return;
2113         }
2114
2115         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
2116                         "sending 1/4 msg of 4-Way Handshake");
2117         /*
2118          * For infrastructure BSS cases, it is better for the AP not to include
2119          * the PMKID KDE in EAPOL-Key msg 1/4 since it could be used to initiate
2120          * offline search for the passphrase/PSK without having to be able to
2121          * capture a 4-way handshake from a STA that has access to the network.
2122          *
2123          * For IBSS cases, addition of PMKID KDE could be considered even with
2124          * WPA2-PSK cases that use multiple PSKs, but only if there is a single
2125          * possible PSK for this STA. However, this should not be done unless
2126          * there is support for using that information on the supplicant side.
2127          * The concern about exposing PMKID unnecessarily in infrastructure BSS
2128          * cases would also apply here, but at least in the IBSS case, this
2129          * would cover a potential real use case.
2130          */
2131         if (sm->wpa == WPA_VERSION_WPA2 &&
2132             (wpa_key_mgmt_wpa_ieee8021x(sm->wpa_key_mgmt) ||
2133              (sm->wpa_key_mgmt == WPA_KEY_MGMT_OWE && sm->pmksa) ||
2134              wpa_key_mgmt_sae(sm->wpa_key_mgmt)) &&
2135             sm->wpa_key_mgmt != WPA_KEY_MGMT_OSEN) {
2136                 pmkid = buf;
2137                 pmkid_len = 2 + RSN_SELECTOR_LEN + PMKID_LEN;
2138                 pmkid[0] = WLAN_EID_VENDOR_SPECIFIC;
2139                 pmkid[1] = RSN_SELECTOR_LEN + PMKID_LEN;
2140                 RSN_SELECTOR_PUT(&pmkid[2], RSN_KEY_DATA_PMKID);
2141                 if (sm->pmksa) {
2142                         wpa_hexdump(MSG_DEBUG,
2143                                     "RSN: Message 1/4 PMKID from PMKSA entry",
2144                                     sm->pmksa->pmkid, PMKID_LEN);
2145                         os_memcpy(&pmkid[2 + RSN_SELECTOR_LEN],
2146                                   sm->pmksa->pmkid, PMKID_LEN);
2147                 } else if (wpa_key_mgmt_suite_b(sm->wpa_key_mgmt)) {
2148                         /* No KCK available to derive PMKID */
2149                         wpa_printf(MSG_DEBUG,
2150                                    "RSN: No KCK available to derive PMKID for message 1/4");
2151                         pmkid = NULL;
2152 #ifdef CONFIG_FILS
2153                 } else if (wpa_key_mgmt_fils(sm->wpa_key_mgmt)) {
2154                         if (sm->pmkid_set) {
2155                                 wpa_hexdump(MSG_DEBUG,
2156                                             "RSN: Message 1/4 PMKID from FILS/ERP",
2157                                             sm->pmkid, PMKID_LEN);
2158                                 os_memcpy(&pmkid[2 + RSN_SELECTOR_LEN],
2159                                           sm->pmkid, PMKID_LEN);
2160                         } else {
2161                                 /* No PMKID available */
2162                                 wpa_printf(MSG_DEBUG,
2163                                            "RSN: No FILS/ERP PMKID available for message 1/4");
2164                                 pmkid = NULL;
2165                         }
2166 #endif /* CONFIG_FILS */
2167 #ifdef CONFIG_IEEE80211R_AP
2168                 } else if (wpa_key_mgmt_ft(sm->wpa_key_mgmt) &&
2169                            sm->ft_completed) {
2170                         wpa_printf(MSG_DEBUG,
2171                                    "FT: No PMKID in message 1/4 when using FT protocol");
2172                         pmkid = NULL;
2173                         pmkid_len = 0;
2174 #endif /* CONFIG_IEEE80211R_AP */
2175 #ifdef CONFIG_SAE
2176                 } else if (wpa_key_mgmt_sae(sm->wpa_key_mgmt)) {
2177                         if (sm->pmkid_set) {
2178                                 wpa_hexdump(MSG_DEBUG,
2179                                             "RSN: Message 1/4 PMKID from SAE",
2180                                             sm->pmkid, PMKID_LEN);
2181                                 os_memcpy(&pmkid[2 + RSN_SELECTOR_LEN],
2182                                           sm->pmkid, PMKID_LEN);
2183                         } else {
2184                                 /* No PMKID available */
2185                                 wpa_printf(MSG_DEBUG,
2186                                            "RSN: No SAE PMKID available for message 1/4");
2187                                 pmkid = NULL;
2188                         }
2189 #endif /* CONFIG_SAE */
2190                 } else {
2191                         /*
2192                          * Calculate PMKID since no PMKSA cache entry was
2193                          * available with pre-calculated PMKID.
2194                          */
2195                         rsn_pmkid(sm->PMK, sm->pmk_len, sm->wpa_auth->addr,
2196                                   sm->addr, &pmkid[2 + RSN_SELECTOR_LEN],
2197                                   sm->wpa_key_mgmt);
2198                         wpa_hexdump(MSG_DEBUG,
2199                                     "RSN: Message 1/4 PMKID derived from PMK",
2200                                     &pmkid[2 + RSN_SELECTOR_LEN], PMKID_LEN);
2201                 }
2202         }
2203         wpa_send_eapol(sm->wpa_auth, sm,
2204                        WPA_KEY_INFO_ACK | WPA_KEY_INFO_KEY_TYPE, NULL,
2205                        sm->ANonce, pmkid, pmkid_len, 0, 0);
2206 }
2207
2208
2209 static int wpa_derive_ptk(struct wpa_state_machine *sm, const u8 *snonce,
2210                           const u8 *pmk, unsigned int pmk_len,
2211                           struct wpa_ptk *ptk)
2212 {
2213         const u8 *z = NULL;
2214         size_t z_len = 0;
2215
2216 #ifdef CONFIG_IEEE80211R_AP
2217         if (wpa_key_mgmt_ft(sm->wpa_key_mgmt)) {
2218                 if (sm->ft_completed) {
2219                         u8 ptk_name[WPA_PMK_NAME_LEN];
2220
2221                         return wpa_pmk_r1_to_ptk(sm->pmk_r1, sm->pmk_r1_len,
2222                                                  sm->SNonce, sm->ANonce,
2223                                                  sm->addr, sm->wpa_auth->addr,
2224                                                  sm->pmk_r1_name,
2225                                                  ptk, ptk_name,
2226                                                  sm->wpa_key_mgmt,
2227                                                  sm->pairwise);
2228                 }
2229                 return wpa_auth_derive_ptk_ft(sm, ptk);
2230         }
2231 #endif /* CONFIG_IEEE80211R_AP */
2232
2233 #ifdef CONFIG_DPP2
2234         if (sm->wpa_key_mgmt == WPA_KEY_MGMT_DPP && sm->dpp_z) {
2235                 z = wpabuf_head(sm->dpp_z);
2236                 z_len = wpabuf_len(sm->dpp_z);
2237         }
2238 #endif /* CONFIG_DPP2 */
2239
2240         return wpa_pmk_to_ptk(pmk, pmk_len, "Pairwise key expansion",
2241                               sm->wpa_auth->addr, sm->addr, sm->ANonce, snonce,
2242                               ptk, sm->wpa_key_mgmt, sm->pairwise, z, z_len);
2243 }
2244
2245
2246 #ifdef CONFIG_FILS
2247
2248 int fils_auth_pmk_to_ptk(struct wpa_state_machine *sm, const u8 *pmk,
2249                          size_t pmk_len, const u8 *snonce, const u8 *anonce,
2250                          const u8 *dhss, size_t dhss_len,
2251                          struct wpabuf *g_sta, struct wpabuf *g_ap)
2252 {
2253         u8 ick[FILS_ICK_MAX_LEN];
2254         size_t ick_len;
2255         int res;
2256         u8 fils_ft[FILS_FT_MAX_LEN];
2257         size_t fils_ft_len = 0;
2258
2259         res = fils_pmk_to_ptk(pmk, pmk_len, sm->addr, sm->wpa_auth->addr,
2260                               snonce, anonce, dhss, dhss_len,
2261                               &sm->PTK, ick, &ick_len,
2262                               sm->wpa_key_mgmt, sm->pairwise,
2263                               fils_ft, &fils_ft_len);
2264         if (res < 0)
2265                 return res;
2266         sm->PTK_valid = TRUE;
2267         sm->tk_already_set = FALSE;
2268
2269 #ifdef CONFIG_IEEE80211R_AP
2270         if (fils_ft_len) {
2271                 struct wpa_authenticator *wpa_auth = sm->wpa_auth;
2272                 struct wpa_auth_config *conf = &wpa_auth->conf;
2273                 u8 pmk_r0[PMK_LEN_MAX], pmk_r0_name[WPA_PMK_NAME_LEN];
2274                 int use_sha384 = wpa_key_mgmt_sha384(sm->wpa_key_mgmt);
2275                 size_t pmk_r0_len = use_sha384 ? SHA384_MAC_LEN : PMK_LEN;
2276
2277                 if (wpa_derive_pmk_r0(fils_ft, fils_ft_len,
2278                                       conf->ssid, conf->ssid_len,
2279                                       conf->mobility_domain,
2280                                       conf->r0_key_holder,
2281                                       conf->r0_key_holder_len,
2282                                       sm->addr, pmk_r0, pmk_r0_name,
2283                                       use_sha384) < 0)
2284                         return -1;
2285
2286                 wpa_hexdump_key(MSG_DEBUG, "FILS+FT: PMK-R0",
2287                                 pmk_r0, pmk_r0_len);
2288                 wpa_hexdump(MSG_DEBUG, "FILS+FT: PMKR0Name",
2289                             pmk_r0_name, WPA_PMK_NAME_LEN);
2290                 wpa_ft_store_pmk_fils(sm, pmk_r0, pmk_r0_name);
2291                 forced_memzero(fils_ft, sizeof(fils_ft));
2292
2293                 res = wpa_derive_pmk_r1_name(pmk_r0_name, conf->r1_key_holder,
2294                                              sm->addr, sm->pmk_r1_name,
2295                                              use_sha384);
2296                 forced_memzero(pmk_r0, PMK_LEN_MAX);
2297                 if (res < 0)
2298                         return -1;
2299                 wpa_hexdump(MSG_DEBUG, "FILS+FT: PMKR1Name", sm->pmk_r1_name,
2300                             WPA_PMK_NAME_LEN);
2301                 sm->pmk_r1_name_valid = 1;
2302         }
2303 #endif /* CONFIG_IEEE80211R_AP */
2304
2305         res = fils_key_auth_sk(ick, ick_len, snonce, anonce,
2306                                sm->addr, sm->wpa_auth->addr,
2307                                g_sta ? wpabuf_head(g_sta) : NULL,
2308                                g_sta ? wpabuf_len(g_sta) : 0,
2309                                g_ap ? wpabuf_head(g_ap) : NULL,
2310                                g_ap ? wpabuf_len(g_ap) : 0,
2311                                sm->wpa_key_mgmt, sm->fils_key_auth_sta,
2312                                sm->fils_key_auth_ap,
2313                                &sm->fils_key_auth_len);
2314         forced_memzero(ick, sizeof(ick));
2315
2316         /* Store nonces for (Re)Association Request/Response frame processing */
2317         os_memcpy(sm->SNonce, snonce, FILS_NONCE_LEN);
2318         os_memcpy(sm->ANonce, anonce, FILS_NONCE_LEN);
2319
2320         return res;
2321 }
2322
2323
2324 static int wpa_aead_decrypt(struct wpa_state_machine *sm, struct wpa_ptk *ptk,
2325                             u8 *buf, size_t buf_len, u16 *_key_data_len)
2326 {
2327         struct ieee802_1x_hdr *hdr;
2328         struct wpa_eapol_key *key;
2329         u8 *pos;
2330         u16 key_data_len;
2331         u8 *tmp;
2332         const u8 *aad[1];
2333         size_t aad_len[1];
2334
2335         hdr = (struct ieee802_1x_hdr *) buf;
2336         key = (struct wpa_eapol_key *) (hdr + 1);
2337         pos = (u8 *) (key + 1);
2338         key_data_len = WPA_GET_BE16(pos);
2339         if (key_data_len < AES_BLOCK_SIZE ||
2340             key_data_len > buf_len - sizeof(*hdr) - sizeof(*key) - 2) {
2341                 wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_INFO,
2342                                 "No room for AES-SIV data in the frame");
2343                 return -1;
2344         }
2345         pos += 2; /* Pointing at the Encrypted Key Data field */
2346
2347         tmp = os_malloc(key_data_len);
2348         if (!tmp)
2349                 return -1;
2350
2351         /* AES-SIV AAD from EAPOL protocol version field (inclusive) to
2352          * to Key Data (exclusive). */
2353         aad[0] = buf;
2354         aad_len[0] = pos - buf;
2355         if (aes_siv_decrypt(ptk->kek, ptk->kek_len, pos, key_data_len,
2356                             1, aad, aad_len, tmp) < 0) {
2357                 wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_INFO,
2358                                 "Invalid AES-SIV data in the frame");
2359                 bin_clear_free(tmp, key_data_len);
2360                 return -1;
2361         }
2362
2363         /* AEAD decryption and validation completed successfully */
2364         key_data_len -= AES_BLOCK_SIZE;
2365         wpa_hexdump_key(MSG_DEBUG, "WPA: Decrypted Key Data",
2366                         tmp, key_data_len);
2367
2368         /* Replace Key Data field with the decrypted version */
2369         os_memcpy(pos, tmp, key_data_len);
2370         pos -= 2; /* Key Data Length field */
2371         WPA_PUT_BE16(pos, key_data_len);
2372         bin_clear_free(tmp, key_data_len);
2373         if (_key_data_len)
2374                 *_key_data_len = key_data_len;
2375         return 0;
2376 }
2377
2378
2379 const u8 * wpa_fils_validate_fils_session(struct wpa_state_machine *sm,
2380                                           const u8 *ies, size_t ies_len,
2381                                           const u8 *fils_session)
2382 {
2383         const u8 *ie, *end;
2384         const u8 *session = NULL;
2385
2386         if (!wpa_key_mgmt_fils(sm->wpa_key_mgmt)) {
2387                 wpa_printf(MSG_DEBUG,
2388                            "FILS: Not a FILS AKM - reject association");
2389                 return NULL;
2390         }
2391
2392         /* Verify Session element */
2393         ie = ies;
2394         end = ((const u8 *) ie) + ies_len;
2395         while (ie + 1 < end) {
2396                 if (ie + 2 + ie[1] > end)
2397                         break;
2398                 if (ie[0] == WLAN_EID_EXTENSION &&
2399                     ie[1] >= 1 + FILS_SESSION_LEN &&
2400                     ie[2] == WLAN_EID_EXT_FILS_SESSION) {
2401                         session = ie;
2402                         break;
2403                 }
2404                 ie += 2 + ie[1];
2405         }
2406
2407         if (!session) {
2408                 wpa_printf(MSG_DEBUG,
2409                            "FILS: %s: Could not find FILS Session element in Assoc Req - reject",
2410                            __func__);
2411                 return NULL;
2412         }
2413
2414         if (!fils_session) {
2415                 wpa_printf(MSG_DEBUG,
2416                            "FILS: %s: Could not find FILS Session element in STA entry - reject",
2417                            __func__);
2418                 return NULL;
2419         }
2420
2421         if (os_memcmp(fils_session, session + 3, FILS_SESSION_LEN) != 0) {
2422                 wpa_printf(MSG_DEBUG, "FILS: Session mismatch");
2423                 wpa_hexdump(MSG_DEBUG, "FILS: Expected FILS Session",
2424                             fils_session, FILS_SESSION_LEN);
2425                 wpa_hexdump(MSG_DEBUG, "FILS: Received FILS Session",
2426                             session + 3, FILS_SESSION_LEN);
2427                 return NULL;
2428         }
2429         return session;
2430 }
2431
2432
2433 int wpa_fils_validate_key_confirm(struct wpa_state_machine *sm, const u8 *ies,
2434                                   size_t ies_len)
2435 {
2436         struct ieee802_11_elems elems;
2437
2438         if (ieee802_11_parse_elems(ies, ies_len, &elems, 1) == ParseFailed) {
2439                 wpa_printf(MSG_DEBUG,
2440                            "FILS: Failed to parse decrypted elements");
2441                 return -1;
2442         }
2443
2444         if (!elems.fils_session) {
2445                 wpa_printf(MSG_DEBUG, "FILS: No FILS Session element");
2446                 return -1;
2447         }
2448
2449         if (!elems.fils_key_confirm) {
2450                 wpa_printf(MSG_DEBUG, "FILS: No FILS Key Confirm element");
2451                 return -1;
2452         }
2453
2454         if (elems.fils_key_confirm_len != sm->fils_key_auth_len) {
2455                 wpa_printf(MSG_DEBUG,
2456                            "FILS: Unexpected Key-Auth length %d (expected %d)",
2457                            elems.fils_key_confirm_len,
2458                            (int) sm->fils_key_auth_len);
2459                 return -1;
2460         }
2461
2462         if (os_memcmp(elems.fils_key_confirm, sm->fils_key_auth_sta,
2463                       sm->fils_key_auth_len) != 0) {
2464                 wpa_printf(MSG_DEBUG, "FILS: Key-Auth mismatch");
2465                 wpa_hexdump(MSG_DEBUG, "FILS: Received Key-Auth",
2466                             elems.fils_key_confirm, elems.fils_key_confirm_len);
2467                 wpa_hexdump(MSG_DEBUG, "FILS: Expected Key-Auth",
2468                             sm->fils_key_auth_sta, sm->fils_key_auth_len);
2469                 return -1;
2470         }
2471
2472         return 0;
2473 }
2474
2475
2476 int fils_decrypt_assoc(struct wpa_state_machine *sm, const u8 *fils_session,
2477                        const struct ieee80211_mgmt *mgmt, size_t frame_len,
2478                        u8 *pos, size_t left)
2479 {
2480         u16 fc, stype;
2481         const u8 *end, *ie_start, *ie, *session, *crypt;
2482         const u8 *aad[5];
2483         size_t aad_len[5];
2484
2485         if (!sm || !sm->PTK_valid) {
2486                 wpa_printf(MSG_DEBUG,
2487                            "FILS: No KEK to decrypt Assocication Request frame");
2488                 return -1;
2489         }
2490
2491         if (!wpa_key_mgmt_fils(sm->wpa_key_mgmt)) {
2492                 wpa_printf(MSG_DEBUG,
2493                            "FILS: Not a FILS AKM - reject association");
2494                 return -1;
2495         }
2496
2497         end = ((const u8 *) mgmt) + frame_len;
2498         fc = le_to_host16(mgmt->frame_control);
2499         stype = WLAN_FC_GET_STYPE(fc);
2500         if (stype == WLAN_FC_STYPE_REASSOC_REQ)
2501                 ie_start = mgmt->u.reassoc_req.variable;
2502         else
2503                 ie_start = mgmt->u.assoc_req.variable;
2504         ie = ie_start;
2505
2506         /*
2507          * Find FILS Session element which is the last unencrypted element in
2508          * the frame.
2509          */
2510         session = wpa_fils_validate_fils_session(sm, ie, end - ie,
2511                                                  fils_session);
2512         if (!session) {
2513                 wpa_printf(MSG_DEBUG, "FILS: Session validation failed");
2514                 return -1;
2515         }
2516
2517         crypt = session + 2 + session[1];
2518
2519         if (end - crypt < AES_BLOCK_SIZE) {
2520                 wpa_printf(MSG_DEBUG,
2521                            "FILS: Too short frame to include AES-SIV data");
2522                 return -1;
2523         }
2524
2525         /* AES-SIV AAD vectors */
2526
2527         /* The STA's MAC address */
2528         aad[0] = mgmt->sa;
2529         aad_len[0] = ETH_ALEN;
2530         /* The AP's BSSID */
2531         aad[1] = mgmt->da;
2532         aad_len[1] = ETH_ALEN;
2533         /* The STA's nonce */
2534         aad[2] = sm->SNonce;
2535         aad_len[2] = FILS_NONCE_LEN;
2536         /* The AP's nonce */
2537         aad[3] = sm->ANonce;
2538         aad_len[3] = FILS_NONCE_LEN;
2539         /*
2540          * The (Re)Association Request frame from the Capability Information
2541          * field to the FILS Session element (both inclusive).
2542          */
2543         aad[4] = (const u8 *) &mgmt->u.assoc_req.capab_info;
2544         aad_len[4] = crypt - aad[4];
2545
2546         if (aes_siv_decrypt(sm->PTK.kek, sm->PTK.kek_len, crypt, end - crypt,
2547                             5, aad, aad_len, pos + (crypt - ie_start)) < 0) {
2548                 wpa_printf(MSG_DEBUG,
2549                            "FILS: Invalid AES-SIV data in the frame");
2550                 return -1;
2551         }
2552         wpa_hexdump(MSG_DEBUG, "FILS: Decrypted Association Request elements",
2553                     pos, left - AES_BLOCK_SIZE);
2554
2555         if (wpa_fils_validate_key_confirm(sm, pos, left - AES_BLOCK_SIZE) < 0) {
2556                 wpa_printf(MSG_DEBUG, "FILS: Key Confirm validation failed");
2557                 return -1;
2558         }
2559
2560         return left - AES_BLOCK_SIZE;
2561 }
2562
2563
2564 int fils_encrypt_assoc(struct wpa_state_machine *sm, u8 *buf,
2565                        size_t current_len, size_t max_len,
2566                        const struct wpabuf *hlp)
2567 {
2568         u8 *end = buf + max_len;
2569         u8 *pos = buf + current_len;
2570         struct ieee80211_mgmt *mgmt;
2571         struct wpabuf *plain;
2572         const u8 *aad[5];
2573         size_t aad_len[5];
2574
2575         if (!sm || !sm->PTK_valid)
2576                 return -1;
2577
2578         wpa_hexdump(MSG_DEBUG,
2579                     "FILS: Association Response frame before FILS processing",
2580                     buf, current_len);
2581
2582         mgmt = (struct ieee80211_mgmt *) buf;
2583
2584         /* AES-SIV AAD vectors */
2585
2586         /* The AP's BSSID */
2587         aad[0] = mgmt->sa;
2588         aad_len[0] = ETH_ALEN;
2589         /* The STA's MAC address */
2590         aad[1] = mgmt->da;
2591         aad_len[1] = ETH_ALEN;
2592         /* The AP's nonce */
2593         aad[2] = sm->ANonce;
2594         aad_len[2] = FILS_NONCE_LEN;
2595         /* The STA's nonce */
2596         aad[3] = sm->SNonce;
2597         aad_len[3] = FILS_NONCE_LEN;
2598         /*
2599          * The (Re)Association Response frame from the Capability Information
2600          * field (the same offset in both Association and Reassociation
2601          * Response frames) to the FILS Session element (both inclusive).
2602          */
2603         aad[4] = (const u8 *) &mgmt->u.assoc_resp.capab_info;
2604         aad_len[4] = pos - aad[4];
2605
2606         /* The following elements will be encrypted with AES-SIV */
2607         plain = fils_prepare_plainbuf(sm, hlp);
2608         if (!plain) {
2609                 wpa_printf(MSG_DEBUG, "FILS: Plain buffer prep failed");
2610                 return -1;
2611         }
2612
2613         if (pos + wpabuf_len(plain) + AES_BLOCK_SIZE > end) {
2614                 wpa_printf(MSG_DEBUG,
2615                            "FILS: Not enough room for FILS elements");
2616                 wpabuf_clear_free(plain);
2617                 return -1;
2618         }
2619
2620         wpa_hexdump_buf_key(MSG_DEBUG, "FILS: Association Response plaintext",
2621                             plain);
2622
2623         if (aes_siv_encrypt(sm->PTK.kek, sm->PTK.kek_len,
2624                             wpabuf_head(plain), wpabuf_len(plain),
2625                             5, aad, aad_len, pos) < 0) {
2626                 wpabuf_clear_free(plain);
2627                 return -1;
2628         }
2629
2630         wpa_hexdump(MSG_DEBUG,
2631                     "FILS: Encrypted Association Response elements",
2632                     pos, AES_BLOCK_SIZE + wpabuf_len(plain));
2633         current_len += wpabuf_len(plain) + AES_BLOCK_SIZE;
2634         wpabuf_clear_free(plain);
2635
2636         sm->fils_completed = 1;
2637
2638         return current_len;
2639 }
2640
2641
2642 static struct wpabuf * fils_prepare_plainbuf(struct wpa_state_machine *sm,
2643                                              const struct wpabuf *hlp)
2644 {
2645         struct wpabuf *plain;
2646         u8 *len, *tmp, *tmp2;
2647         u8 hdr[2];
2648         u8 *gtk, dummy_gtk[32];
2649         size_t gtk_len;
2650         struct wpa_group *gsm;
2651
2652         plain = wpabuf_alloc(1000);
2653         if (!plain)
2654                 return NULL;
2655
2656         /* TODO: FILS Public Key */
2657
2658         /* FILS Key Confirmation */
2659         wpabuf_put_u8(plain, WLAN_EID_EXTENSION); /* Element ID */
2660         wpabuf_put_u8(plain, 1 + sm->fils_key_auth_len); /* Length */
2661         /* Element ID Extension */
2662         wpabuf_put_u8(plain, WLAN_EID_EXT_FILS_KEY_CONFIRM);
2663         wpabuf_put_data(plain, sm->fils_key_auth_ap, sm->fils_key_auth_len);
2664
2665         /* FILS HLP Container */
2666         if (hlp)
2667                 wpabuf_put_buf(plain, hlp);
2668
2669         /* TODO: FILS IP Address Assignment */
2670
2671         /* Key Delivery */
2672         gsm = sm->group;
2673         wpabuf_put_u8(plain, WLAN_EID_EXTENSION); /* Element ID */
2674         len = wpabuf_put(plain, 1);
2675         wpabuf_put_u8(plain, WLAN_EID_EXT_KEY_DELIVERY);
2676         wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN,
2677                             wpabuf_put(plain, WPA_KEY_RSC_LEN));
2678         /* GTK KDE */
2679         gtk = gsm->GTK[gsm->GN - 1];
2680         gtk_len = gsm->GTK_len;
2681         if (sm->wpa_auth->conf.disable_gtk ||
2682             sm->wpa_key_mgmt == WPA_KEY_MGMT_OSEN) {
2683                 /*
2684                  * Provide unique random GTK to each STA to prevent use
2685                  * of GTK in the BSS.
2686                  */
2687                 if (random_get_bytes(dummy_gtk, gtk_len) < 0) {
2688                         wpabuf_clear_free(plain);
2689                         return NULL;
2690                 }
2691                 gtk = dummy_gtk;
2692         }
2693         hdr[0] = gsm->GN & 0x03;
2694         hdr[1] = 0;
2695         tmp = wpabuf_put(plain, 0);
2696         tmp2 = wpa_add_kde(tmp, RSN_KEY_DATA_GROUPKEY, hdr, 2,
2697                            gtk, gtk_len);
2698         wpabuf_put(plain, tmp2 - tmp);
2699
2700         /* IGTK KDE */
2701         tmp = wpabuf_put(plain, 0);
2702         tmp2 = ieee80211w_kde_add(sm, tmp);
2703         wpabuf_put(plain, tmp2 - tmp);
2704
2705         *len = (u8 *) wpabuf_put(plain, 0) - len - 1;
2706
2707 #ifdef CONFIG_OCV
2708         if (wpa_auth_uses_ocv(sm)) {
2709                 struct wpa_channel_info ci;
2710                 u8 *pos;
2711
2712                 if (wpa_channel_info(sm->wpa_auth, &ci) != 0) {
2713                         wpa_printf(MSG_WARNING,
2714                                    "FILS: Failed to get channel info for OCI element");
2715                         wpabuf_clear_free(plain);
2716                         return NULL;
2717                 }
2718
2719                 pos = wpabuf_put(plain, OCV_OCI_EXTENDED_LEN);
2720                 if (ocv_insert_extended_oci(&ci, pos) < 0) {
2721                         wpabuf_clear_free(plain);
2722                         return NULL;
2723                 }
2724         }
2725 #endif /* CONFIG_OCV */
2726
2727         return plain;
2728 }
2729
2730
2731 int fils_set_tk(struct wpa_state_machine *sm)
2732 {
2733         enum wpa_alg alg;
2734         int klen;
2735
2736         if (!sm || !sm->PTK_valid) {
2737                 wpa_printf(MSG_DEBUG, "FILS: No valid PTK available to set TK");
2738                 return -1;
2739         }
2740         if (sm->tk_already_set) {
2741                 wpa_printf(MSG_DEBUG, "FILS: TK already set to the driver");
2742                 return -1;
2743         }
2744
2745         alg = wpa_cipher_to_alg(sm->pairwise);
2746         klen = wpa_cipher_key_len(sm->pairwise);
2747
2748         wpa_printf(MSG_DEBUG, "FILS: Configure TK to the driver");
2749         if (wpa_auth_set_key(sm->wpa_auth, 0, alg, sm->addr, 0,
2750                              sm->PTK.tk, klen)) {
2751                 wpa_printf(MSG_DEBUG, "FILS: Failed to set TK to the driver");
2752                 return -1;
2753         }
2754         sm->tk_already_set = TRUE;
2755
2756         return 0;
2757 }
2758
2759
2760 u8 * hostapd_eid_assoc_fils_session(struct wpa_state_machine *sm, u8 *buf,
2761                                     const u8 *fils_session, struct wpabuf *hlp)
2762 {
2763         struct wpabuf *plain;
2764         u8 *pos = buf;
2765
2766         /* FILS Session */
2767         *pos++ = WLAN_EID_EXTENSION; /* Element ID */
2768         *pos++ = 1 + FILS_SESSION_LEN; /* Length */
2769         *pos++ = WLAN_EID_EXT_FILS_SESSION; /* Element ID Extension */
2770         os_memcpy(pos, fils_session, FILS_SESSION_LEN);
2771         pos += FILS_SESSION_LEN;
2772
2773         plain = fils_prepare_plainbuf(sm, hlp);
2774         if (!plain) {
2775                 wpa_printf(MSG_DEBUG, "FILS: Plain buffer prep failed");
2776                 return NULL;
2777         }
2778
2779         os_memcpy(pos, wpabuf_head(plain), wpabuf_len(plain));
2780         pos += wpabuf_len(plain);
2781
2782         wpa_printf(MSG_DEBUG, "%s: plain buf_len: %u", __func__,
2783                    (unsigned int) wpabuf_len(plain));
2784         wpabuf_clear_free(plain);
2785         sm->fils_completed = 1;
2786         return pos;
2787 }
2788
2789 #endif /* CONFIG_FILS */
2790
2791
2792 #ifdef CONFIG_OCV
2793 int get_sta_tx_parameters(struct wpa_state_machine *sm, int ap_max_chanwidth,
2794                           int ap_seg1_idx, int *bandwidth, int *seg1_idx)
2795 {
2796         struct wpa_authenticator *wpa_auth = sm->wpa_auth;
2797
2798         if (!wpa_auth->cb->get_sta_tx_params)
2799                 return -1;
2800         return wpa_auth->cb->get_sta_tx_params(wpa_auth->cb_ctx, sm->addr,
2801                                                ap_max_chanwidth, ap_seg1_idx,
2802                                                bandwidth, seg1_idx);
2803 }
2804 #endif /* CONFIG_OCV */
2805
2806
2807 SM_STATE(WPA_PTK, PTKCALCNEGOTIATING)
2808 {
2809         struct wpa_authenticator *wpa_auth = sm->wpa_auth;
2810         struct wpa_ptk PTK;
2811         int ok = 0, psk_found = 0;
2812         const u8 *pmk = NULL;
2813         size_t pmk_len;
2814         int ft;
2815         const u8 *eapol_key_ie, *key_data, *mic;
2816         u16 key_data_length;
2817         size_t mic_len, eapol_key_ie_len;
2818         struct ieee802_1x_hdr *hdr;
2819         struct wpa_eapol_key *key;
2820         struct wpa_eapol_ie_parse kde;
2821         int vlan_id = 0;
2822
2823         SM_ENTRY_MA(WPA_PTK, PTKCALCNEGOTIATING, wpa_ptk);
2824         sm->EAPOLKeyReceived = FALSE;
2825         sm->update_snonce = FALSE;
2826         os_memset(&PTK, 0, sizeof(PTK));
2827
2828         mic_len = wpa_mic_len(sm->wpa_key_mgmt, sm->pmk_len);
2829
2830         /* WPA with IEEE 802.1X: use the derived PMK from EAP
2831          * WPA-PSK: iterate through possible PSKs and select the one matching
2832          * the packet */
2833         for (;;) {
2834                 if (wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt) &&
2835                     !wpa_key_mgmt_sae(sm->wpa_key_mgmt)) {
2836                         pmk = wpa_auth_get_psk(sm->wpa_auth, sm->addr,
2837                                                sm->p2p_dev_addr, pmk, &pmk_len,
2838                                                &vlan_id);
2839                         if (pmk == NULL)
2840                                 break;
2841                         psk_found = 1;
2842 #ifdef CONFIG_IEEE80211R_AP
2843                         if (wpa_key_mgmt_ft_psk(sm->wpa_key_mgmt)) {
2844                                 os_memcpy(sm->xxkey, pmk, pmk_len);
2845                                 sm->xxkey_len = pmk_len;
2846                         }
2847 #endif /* CONFIG_IEEE80211R_AP */
2848                 } else {
2849                         pmk = sm->PMK;
2850                         pmk_len = sm->pmk_len;
2851                 }
2852
2853                 if ((!pmk || !pmk_len) && sm->pmksa) {
2854                         wpa_printf(MSG_DEBUG, "WPA: Use PMK from PMKSA cache");
2855                         pmk = sm->pmksa->pmk;
2856                         pmk_len = sm->pmksa->pmk_len;
2857                 }
2858
2859                 if (wpa_derive_ptk(sm, sm->SNonce, pmk, pmk_len, &PTK) < 0)
2860                         break;
2861
2862                 if (mic_len &&
2863                     wpa_verify_key_mic(sm->wpa_key_mgmt, pmk_len, &PTK,
2864                                        sm->last_rx_eapol_key,
2865                                        sm->last_rx_eapol_key_len) == 0) {
2866                         if (sm->PMK != pmk) {
2867                                 os_memcpy(sm->PMK, pmk, pmk_len);
2868                                 sm->pmk_len = pmk_len;
2869                         }
2870                         ok = 1;
2871                         break;
2872                 }
2873
2874 #ifdef CONFIG_FILS
2875                 if (!mic_len &&
2876                     wpa_aead_decrypt(sm, &PTK, sm->last_rx_eapol_key,
2877                                      sm->last_rx_eapol_key_len, NULL) == 0) {
2878                         ok = 1;
2879                         break;
2880                 }
2881 #endif /* CONFIG_FILS */
2882
2883                 if (!wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt) ||
2884                     wpa_key_mgmt_sae(sm->wpa_key_mgmt))
2885                         break;
2886         }
2887
2888         if (!ok) {
2889                 wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
2890                                 "invalid MIC in msg 2/4 of 4-Way Handshake");
2891                 if (psk_found)
2892                         wpa_auth_psk_failure_report(sm->wpa_auth, sm->addr);
2893                 return;
2894         }
2895
2896         /*
2897          * Note: last_rx_eapol_key length fields have already been validated in
2898          * wpa_receive().
2899          */
2900         hdr = (struct ieee802_1x_hdr *) sm->last_rx_eapol_key;
2901         key = (struct wpa_eapol_key *) (hdr + 1);
2902         mic = (u8 *) (key + 1);
2903         key_data = mic + mic_len + 2;
2904         key_data_length = WPA_GET_BE16(mic + mic_len);
2905         if (key_data_length > sm->last_rx_eapol_key_len - sizeof(*hdr) -
2906             sizeof(*key) - mic_len - 2)
2907                 return;
2908
2909         if (wpa_parse_kde_ies(key_data, key_data_length, &kde) < 0) {
2910                 wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_INFO,
2911                                  "received EAPOL-Key msg 2/4 with invalid Key Data contents");
2912                 return;
2913         }
2914         if (kde.rsn_ie) {
2915                 eapol_key_ie = kde.rsn_ie;
2916                 eapol_key_ie_len = kde.rsn_ie_len;
2917         } else if (kde.osen) {
2918                 eapol_key_ie = kde.osen;
2919                 eapol_key_ie_len = kde.osen_len;
2920         } else {
2921                 eapol_key_ie = kde.wpa_ie;
2922                 eapol_key_ie_len = kde.wpa_ie_len;
2923         }
2924         ft = sm->wpa == WPA_VERSION_WPA2 && wpa_key_mgmt_ft(sm->wpa_key_mgmt);
2925         if (sm->wpa_ie == NULL ||
2926             wpa_compare_rsn_ie(ft, sm->wpa_ie, sm->wpa_ie_len,
2927                                eapol_key_ie, eapol_key_ie_len)) {
2928                 wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
2929                                 "WPA IE from (Re)AssocReq did not match with msg 2/4");
2930                 if (sm->wpa_ie) {
2931                         wpa_hexdump(MSG_DEBUG, "WPA IE in AssocReq",
2932                                     sm->wpa_ie, sm->wpa_ie_len);
2933                 }
2934                 wpa_hexdump(MSG_DEBUG, "WPA IE in msg 2/4",
2935                             eapol_key_ie, eapol_key_ie_len);
2936                 /* MLME-DEAUTHENTICATE.request */
2937                 wpa_sta_disconnect(wpa_auth, sm->addr,
2938                                    WLAN_REASON_PREV_AUTH_NOT_VALID);
2939                 return;
2940         }
2941 #ifdef CONFIG_OCV
2942         if (wpa_auth_uses_ocv(sm)) {
2943                 struct wpa_channel_info ci;
2944                 int tx_chanwidth;
2945                 int tx_seg1_idx;
2946
2947                 if (wpa_channel_info(wpa_auth, &ci) != 0) {
2948                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
2949                                         "Failed to get channel info to validate received OCI in EAPOL-Key 2/4");
2950                         return;
2951                 }
2952
2953                 if (get_sta_tx_parameters(sm,
2954                                           channel_width_to_int(ci.chanwidth),
2955                                           ci.seg1_idx, &tx_chanwidth,
2956                                           &tx_seg1_idx) < 0)
2957                         return;
2958
2959                 if (ocv_verify_tx_params(kde.oci, kde.oci_len, &ci,
2960                                          tx_chanwidth, tx_seg1_idx) != 0) {
2961                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
2962                                         ocv_errorstr);
2963                         return;
2964                 }
2965         }
2966 #endif /* CONFIG_OCV */
2967 #ifdef CONFIG_IEEE80211R_AP
2968         if (ft && ft_check_msg_2_of_4(wpa_auth, sm, &kde) < 0) {
2969                 wpa_sta_disconnect(wpa_auth, sm->addr,
2970                                    WLAN_REASON_PREV_AUTH_NOT_VALID);
2971                 return;
2972         }
2973 #endif /* CONFIG_IEEE80211R_AP */
2974 #ifdef CONFIG_P2P
2975         if (kde.ip_addr_req && kde.ip_addr_req[0] &&
2976             wpa_auth->ip_pool && WPA_GET_BE32(sm->ip_addr) == 0) {
2977                 int idx;
2978                 wpa_printf(MSG_DEBUG,
2979                            "P2P: IP address requested in EAPOL-Key exchange");
2980                 idx = bitfield_get_first_zero(wpa_auth->ip_pool);
2981                 if (idx >= 0) {
2982                         u32 start = WPA_GET_BE32(wpa_auth->conf.ip_addr_start);
2983                         bitfield_set(wpa_auth->ip_pool, idx);
2984                         WPA_PUT_BE32(sm->ip_addr, start + idx);
2985                         wpa_printf(MSG_DEBUG,
2986                                    "P2P: Assigned IP address %u.%u.%u.%u to "
2987                                    MACSTR, sm->ip_addr[0], sm->ip_addr[1],
2988                                    sm->ip_addr[2], sm->ip_addr[3],
2989                                    MAC2STR(sm->addr));
2990                 }
2991         }
2992 #endif /* CONFIG_P2P */
2993
2994 #ifdef CONFIG_IEEE80211R_AP
2995         if (sm->wpa == WPA_VERSION_WPA2 && wpa_key_mgmt_ft(sm->wpa_key_mgmt)) {
2996                 /*
2997                  * Verify that PMKR1Name from EAPOL-Key message 2/4 matches
2998                  * with the value we derived.
2999                  */
3000                 if (os_memcmp_const(sm->sup_pmk_r1_name, sm->pmk_r1_name,
3001                                     WPA_PMK_NAME_LEN) != 0) {
3002                         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
3003                                         "PMKR1Name mismatch in FT 4-way "
3004                                         "handshake");
3005                         wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name from "
3006                                     "Supplicant",
3007                                     sm->sup_pmk_r1_name, WPA_PMK_NAME_LEN);
3008                         wpa_hexdump(MSG_DEBUG, "FT: Derived PMKR1Name",
3009                                     sm->pmk_r1_name, WPA_PMK_NAME_LEN);
3010                         return;
3011                 }
3012         }
3013 #endif /* CONFIG_IEEE80211R_AP */
3014
3015         if (vlan_id && wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt) &&
3016             wpa_auth_update_vlan(wpa_auth, sm->addr, vlan_id) < 0) {
3017                 wpa_sta_disconnect(wpa_auth, sm->addr,
3018                                    WLAN_REASON_PREV_AUTH_NOT_VALID);
3019                 return;
3020         }
3021
3022         sm->pending_1_of_4_timeout = 0;
3023         eloop_cancel_timeout(wpa_send_eapol_timeout, sm->wpa_auth, sm);
3024
3025         if (wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt)) {
3026                 /* PSK may have changed from the previous choice, so update
3027                  * state machine data based on whatever PSK was selected here.
3028                  */
3029                 os_memcpy(sm->PMK, pmk, PMK_LEN);
3030                 sm->pmk_len = PMK_LEN;
3031         }
3032
3033         sm->MICVerified = TRUE;
3034
3035         os_memcpy(&sm->PTK, &PTK, sizeof(PTK));
3036         forced_memzero(&PTK, sizeof(PTK));
3037         sm->PTK_valid = TRUE;
3038 }
3039
3040
3041 SM_STATE(WPA_PTK, PTKCALCNEGOTIATING2)
3042 {
3043         SM_ENTRY_MA(WPA_PTK, PTKCALCNEGOTIATING2, wpa_ptk);
3044         sm->TimeoutCtr = 0;
3045 }
3046
3047
3048 #ifdef CONFIG_IEEE80211W
3049
3050 static int ieee80211w_kde_len(struct wpa_state_machine *sm)
3051 {
3052         if (sm->mgmt_frame_prot) {
3053                 size_t len;
3054                 len = wpa_cipher_key_len(sm->wpa_auth->conf.group_mgmt_cipher);
3055                 return 2 + RSN_SELECTOR_LEN + WPA_IGTK_KDE_PREFIX_LEN + len;
3056         }
3057
3058         return 0;
3059 }
3060
3061
3062 static u8 * ieee80211w_kde_add(struct wpa_state_machine *sm, u8 *pos)
3063 {
3064         struct wpa_igtk_kde igtk;
3065         struct wpa_group *gsm = sm->group;
3066         u8 rsc[WPA_KEY_RSC_LEN];
3067         size_t len = wpa_cipher_key_len(sm->wpa_auth->conf.group_mgmt_cipher);
3068
3069         if (!sm->mgmt_frame_prot)
3070                 return pos;
3071
3072         igtk.keyid[0] = gsm->GN_igtk;
3073         igtk.keyid[1] = 0;
3074         if (gsm->wpa_group_state != WPA_GROUP_SETKEYSDONE ||
3075             wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN_igtk, rsc) < 0)
3076                 os_memset(igtk.pn, 0, sizeof(igtk.pn));
3077         else
3078                 os_memcpy(igtk.pn, rsc, sizeof(igtk.pn));
3079         os_memcpy(igtk.igtk, gsm->IGTK[gsm->GN_igtk - 4], len);
3080         if (sm->wpa_auth->conf.disable_gtk ||
3081             sm->wpa_key_mgmt == WPA_KEY_MGMT_OSEN) {
3082                 /*
3083                  * Provide unique random IGTK to each STA to prevent use of
3084                  * IGTK in the BSS.
3085                  */
3086                 if (random_get_bytes(igtk.igtk, len) < 0)
3087                         return pos;
3088         }
3089         pos = wpa_add_kde(pos, RSN_KEY_DATA_IGTK,
3090                           (const u8 *) &igtk, WPA_IGTK_KDE_PREFIX_LEN + len,
3091                           NULL, 0);
3092
3093         return pos;
3094 }
3095
3096 #else /* CONFIG_IEEE80211W */
3097
3098 static int ieee80211w_kde_len(struct wpa_state_machine *sm)
3099 {
3100         return 0;
3101 }
3102
3103
3104 static u8 * ieee80211w_kde_add(struct wpa_state_machine *sm, u8 *pos)
3105 {
3106         return pos;
3107 }
3108
3109 #endif /* CONFIG_IEEE80211W */
3110
3111
3112 static int ocv_oci_len(struct wpa_state_machine *sm)
3113 {
3114 #ifdef CONFIG_OCV
3115         if (wpa_auth_uses_ocv(sm))
3116                 return OCV_OCI_KDE_LEN;
3117 #endif /* CONFIG_OCV */
3118         return 0;
3119 }
3120
3121 static int ocv_oci_add(struct wpa_state_machine *sm, u8 **argpos)
3122 {
3123 #ifdef CONFIG_OCV
3124         struct wpa_channel_info ci;
3125
3126         if (!wpa_auth_uses_ocv(sm))
3127                 return 0;
3128
3129         if (wpa_channel_info(sm->wpa_auth, &ci) != 0) {
3130                 wpa_printf(MSG_WARNING,
3131                            "Failed to get channel info for OCI element");
3132                 return -1;
3133         }
3134
3135         return ocv_insert_oci_kde(&ci, argpos);
3136 #else /* CONFIG_OCV */
3137         return 0;
3138 #endif /* CONFIG_OCV */
3139 }
3140
3141
3142 SM_STATE(WPA_PTK, PTKINITNEGOTIATING)
3143 {
3144         u8 rsc[WPA_KEY_RSC_LEN], *_rsc, *gtk, *kde, *pos, dummy_gtk[32];
3145         size_t gtk_len, kde_len;
3146         struct wpa_group *gsm = sm->group;
3147         u8 *wpa_ie;
3148         int wpa_ie_len, secure, keyidx, encr = 0;
3149
3150         SM_ENTRY_MA(WPA_PTK, PTKINITNEGOTIATING, wpa_ptk);
3151         sm->TimeoutEvt = FALSE;
3152
3153         sm->TimeoutCtr++;
3154         if (sm->wpa_auth->conf.wpa_disable_eapol_key_retries &&
3155             sm->TimeoutCtr > 1) {
3156                 /* Do not allow retransmission of EAPOL-Key msg 3/4 */
3157                 return;
3158         }
3159         if (sm->TimeoutCtr > sm->wpa_auth->conf.wpa_pairwise_update_count) {
3160                 /* No point in sending the EAPOL-Key - we will disconnect
3161                  * immediately following this. */
3162                 return;
3163         }
3164
3165         /* Send EAPOL(1, 1, 1, Pair, P, RSC, ANonce, MIC(PTK), RSNIE, [MDIE],
3166            GTK[GN], IGTK, [FTIE], [TIE * 2])
3167          */
3168         os_memset(rsc, 0, WPA_KEY_RSC_LEN);
3169         wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN, rsc);
3170         /* If FT is used, wpa_auth->wpa_ie includes both RSNIE and MDIE */
3171         wpa_ie = sm->wpa_auth->wpa_ie;
3172         wpa_ie_len = sm->wpa_auth->wpa_ie_len;
3173         if (sm->wpa == WPA_VERSION_WPA &&
3174             (sm->wpa_auth->conf.wpa & WPA_PROTO_RSN) &&
3175             wpa_ie_len > wpa_ie[1] + 2 && wpa_ie[0] == WLAN_EID_RSN) {
3176                 /* WPA-only STA, remove RSN IE and possible MDIE */
3177                 wpa_ie = wpa_ie + wpa_ie[1] + 2;
3178                 if (wpa_ie[0] == WLAN_EID_MOBILITY_DOMAIN)
3179                         wpa_ie = wpa_ie + wpa_ie[1] + 2;
3180                 wpa_ie_len = wpa_ie[1] + 2;
3181         }
3182         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
3183                         "sending 3/4 msg of 4-Way Handshake");
3184         if (sm->wpa == WPA_VERSION_WPA2) {
3185                 /* WPA2 send GTK in the 4-way handshake */
3186                 secure = 1;
3187                 gtk = gsm->GTK[gsm->GN - 1];
3188                 gtk_len = gsm->GTK_len;
3189                 if (sm->wpa_auth->conf.disable_gtk ||
3190                     sm->wpa_key_mgmt == WPA_KEY_MGMT_OSEN) {
3191                         /*
3192                          * Provide unique random GTK to each STA to prevent use
3193                          * of GTK in the BSS.
3194                          */
3195                         if (random_get_bytes(dummy_gtk, gtk_len) < 0)
3196                                 return;
3197                         gtk = dummy_gtk;
3198                 }
3199                 keyidx = gsm->GN;
3200                 _rsc = rsc;
3201                 encr = 1;
3202         } else {
3203                 /* WPA does not include GTK in msg 3/4 */
3204                 secure = 0;
3205                 gtk = NULL;
3206                 gtk_len = 0;
3207                 keyidx = 0;
3208                 _rsc = NULL;
3209                 if (sm->rx_eapol_key_secure) {
3210                         /*
3211                          * It looks like Windows 7 supplicant tries to use
3212                          * Secure bit in msg 2/4 after having reported Michael
3213                          * MIC failure and it then rejects the 4-way handshake
3214                          * if msg 3/4 does not set Secure bit. Work around this
3215                          * by setting the Secure bit here even in the case of
3216                          * WPA if the supplicant used it first.
3217                          */
3218                         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
3219                                         "STA used Secure bit in WPA msg 2/4 - "
3220                                         "set Secure for 3/4 as workaround");
3221                         secure = 1;
3222                 }
3223         }
3224
3225         kde_len = wpa_ie_len + ieee80211w_kde_len(sm) + ocv_oci_len(sm);
3226         if (gtk)
3227                 kde_len += 2 + RSN_SELECTOR_LEN + 2 + gtk_len;
3228 #ifdef CONFIG_IEEE80211R_AP
3229         if (wpa_key_mgmt_ft(sm->wpa_key_mgmt)) {
3230                 kde_len += 2 + PMKID_LEN; /* PMKR1Name into RSN IE */
3231                 kde_len += 300; /* FTIE + 2 * TIE */
3232         }
3233 #endif /* CONFIG_IEEE80211R_AP */
3234 #ifdef CONFIG_P2P
3235         if (WPA_GET_BE32(sm->ip_addr) > 0)
3236                 kde_len += 2 + RSN_SELECTOR_LEN + 3 * 4;
3237 #endif /* CONFIG_P2P */
3238         kde = os_malloc(kde_len);
3239         if (kde == NULL)
3240                 return;
3241
3242         pos = kde;
3243         os_memcpy(pos, wpa_ie, wpa_ie_len);
3244         pos += wpa_ie_len;
3245 #ifdef CONFIG_IEEE80211R_AP
3246         if (wpa_key_mgmt_ft(sm->wpa_key_mgmt)) {
3247                 int res;
3248                 size_t elen;
3249
3250                 elen = pos - kde;
3251                 res = wpa_insert_pmkid(kde, &elen, sm->pmk_r1_name);
3252                 if (res < 0) {
3253                         wpa_printf(MSG_ERROR, "FT: Failed to insert "
3254                                    "PMKR1Name into RSN IE in EAPOL-Key data");
3255                         os_free(kde);
3256                         return;
3257                 }
3258                 pos -= wpa_ie_len;
3259                 pos += elen;
3260         }
3261 #endif /* CONFIG_IEEE80211R_AP */
3262         if (gtk) {
3263                 u8 hdr[2];
3264                 hdr[0] = keyidx & 0x03;
3265                 hdr[1] = 0;
3266                 pos = wpa_add_kde(pos, RSN_KEY_DATA_GROUPKEY, hdr, 2,
3267                                   gtk, gtk_len);
3268         }
3269         pos = ieee80211w_kde_add(sm, pos);
3270         if (ocv_oci_add(sm, &pos) < 0) {
3271                 os_free(kde);
3272                 return;
3273         }
3274
3275 #ifdef CONFIG_IEEE80211R_AP
3276         if (wpa_key_mgmt_ft(sm->wpa_key_mgmt)) {
3277                 int res;
3278                 struct wpa_auth_config *conf;
3279
3280                 conf = &sm->wpa_auth->conf;
3281                 if (sm->assoc_resp_ftie &&
3282                     kde + kde_len - pos >= 2 + sm->assoc_resp_ftie[1]) {
3283                         os_memcpy(pos, sm->assoc_resp_ftie,
3284                                   2 + sm->assoc_resp_ftie[1]);
3285                         res = 2 + sm->assoc_resp_ftie[1];
3286                 } else {
3287                         int use_sha384 = wpa_key_mgmt_sha384(sm->wpa_key_mgmt);
3288
3289                         res = wpa_write_ftie(conf, use_sha384,
3290                                              conf->r0_key_holder,
3291                                              conf->r0_key_holder_len,
3292                                              NULL, NULL, pos,
3293                                              kde + kde_len - pos,
3294                                              NULL, 0);
3295                 }
3296                 if (res < 0) {
3297                         wpa_printf(MSG_ERROR, "FT: Failed to insert FTIE "
3298                                    "into EAPOL-Key Key Data");
3299                         os_free(kde);
3300                         return;
3301                 }
3302                 pos += res;
3303
3304                 /* TIE[ReassociationDeadline] (TU) */
3305                 *pos++ = WLAN_EID_TIMEOUT_INTERVAL;
3306                 *pos++ = 5;
3307                 *pos++ = WLAN_TIMEOUT_REASSOC_DEADLINE;
3308                 WPA_PUT_LE32(pos, conf->reassociation_deadline);
3309                 pos += 4;
3310
3311                 /* TIE[KeyLifetime] (seconds) */
3312                 *pos++ = WLAN_EID_TIMEOUT_INTERVAL;
3313                 *pos++ = 5;
3314                 *pos++ = WLAN_TIMEOUT_KEY_LIFETIME;
3315                 WPA_PUT_LE32(pos, conf->r0_key_lifetime);
3316                 pos += 4;
3317         }
3318 #endif /* CONFIG_IEEE80211R_AP */
3319 #ifdef CONFIG_P2P
3320         if (WPA_GET_BE32(sm->ip_addr) > 0) {
3321                 u8 addr[3 * 4];
3322                 os_memcpy(addr, sm->ip_addr, 4);
3323                 os_memcpy(addr + 4, sm->wpa_auth->conf.ip_addr_mask, 4);
3324                 os_memcpy(addr + 8, sm->wpa_auth->conf.ip_addr_go, 4);
3325                 pos = wpa_add_kde(pos, WFA_KEY_DATA_IP_ADDR_ALLOC,
3326                                   addr, sizeof(addr), NULL, 0);
3327         }
3328 #endif /* CONFIG_P2P */
3329
3330         wpa_send_eapol(sm->wpa_auth, sm,
3331                        (secure ? WPA_KEY_INFO_SECURE : 0) |
3332                        (wpa_mic_len(sm->wpa_key_mgmt, sm->pmk_len) ?
3333                         WPA_KEY_INFO_MIC : 0) |
3334                        WPA_KEY_INFO_ACK | WPA_KEY_INFO_INSTALL |
3335                        WPA_KEY_INFO_KEY_TYPE,
3336                        _rsc, sm->ANonce, kde, pos - kde, keyidx, encr);
3337         os_free(kde);
3338 }
3339
3340
3341 SM_STATE(WPA_PTK, PTKINITDONE)
3342 {
3343         SM_ENTRY_MA(WPA_PTK, PTKINITDONE, wpa_ptk);
3344         sm->EAPOLKeyReceived = FALSE;
3345         if (sm->Pair) {
3346                 enum wpa_alg alg = wpa_cipher_to_alg(sm->pairwise);
3347                 int klen = wpa_cipher_key_len(sm->pairwise);
3348                 if (wpa_auth_set_key(sm->wpa_auth, 0, alg, sm->addr, 0,
3349                                      sm->PTK.tk, klen)) {
3350                         wpa_sta_disconnect(sm->wpa_auth, sm->addr,
3351                                            WLAN_REASON_PREV_AUTH_NOT_VALID);
3352                         return;
3353                 }
3354                 /* FIX: MLME-SetProtection.Request(TA, Tx_Rx) */
3355                 sm->pairwise_set = TRUE;
3356
3357                 wpa_auth_set_ptk_rekey_timer(sm);
3358
3359                 if (wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt) ||
3360                     sm->wpa_key_mgmt == WPA_KEY_MGMT_DPP ||
3361                     sm->wpa_key_mgmt == WPA_KEY_MGMT_OWE) {
3362                         wpa_auth_set_eapol(sm->wpa_auth, sm->addr,
3363                                            WPA_EAPOL_authorized, 1);
3364                 }
3365         }
3366
3367         if (0 /* IBSS == TRUE */) {
3368                 sm->keycount++;
3369                 if (sm->keycount == 2) {
3370                         wpa_auth_set_eapol(sm->wpa_auth, sm->addr,
3371                                            WPA_EAPOL_portValid, 1);
3372                 }
3373         } else {
3374                 wpa_auth_set_eapol(sm->wpa_auth, sm->addr, WPA_EAPOL_portValid,
3375                                    1);
3376         }
3377         wpa_auth_set_eapol(sm->wpa_auth, sm->addr, WPA_EAPOL_keyAvailable, 0);
3378         wpa_auth_set_eapol(sm->wpa_auth, sm->addr, WPA_EAPOL_keyDone, 1);
3379         if (sm->wpa == WPA_VERSION_WPA)
3380                 sm->PInitAKeys = TRUE;
3381         else
3382                 sm->has_GTK = TRUE;
3383         wpa_auth_vlogger(sm->wpa_auth, sm->addr, LOGGER_INFO,
3384                          "pairwise key handshake completed (%s)",
3385                          sm->wpa == WPA_VERSION_WPA ? "WPA" : "RSN");
3386
3387 #ifdef CONFIG_IEEE80211R_AP
3388         wpa_ft_push_pmk_r1(sm->wpa_auth, sm->addr);
3389 #endif /* CONFIG_IEEE80211R_AP */
3390 }
3391
3392
3393 SM_STEP(WPA_PTK)
3394 {
3395         struct wpa_authenticator *wpa_auth = sm->wpa_auth;
3396
3397         if (sm->Init)
3398                 SM_ENTER(WPA_PTK, INITIALIZE);
3399         else if (sm->Disconnect
3400                  /* || FIX: dot11RSNAConfigSALifetime timeout */) {
3401                 wpa_auth_logger(wpa_auth, sm->addr, LOGGER_DEBUG,
3402                                 "WPA_PTK: sm->Disconnect");
3403                 SM_ENTER(WPA_PTK, DISCONNECT);
3404         }
3405         else if (sm->DeauthenticationRequest)
3406                 SM_ENTER(WPA_PTK, DISCONNECTED);
3407         else if (sm->AuthenticationRequest)
3408                 SM_ENTER(WPA_PTK, AUTHENTICATION);
3409         else if (sm->ReAuthenticationRequest)
3410                 SM_ENTER(WPA_PTK, AUTHENTICATION2);
3411         else if (sm->PTKRequest) {
3412                 if (wpa_auth_sm_ptk_update(sm) < 0)
3413                         SM_ENTER(WPA_PTK, DISCONNECTED);
3414                 else
3415                         SM_ENTER(WPA_PTK, PTKSTART);
3416         } else switch (sm->wpa_ptk_state) {
3417         case WPA_PTK_INITIALIZE:
3418                 break;
3419         case WPA_PTK_DISCONNECT:
3420                 SM_ENTER(WPA_PTK, DISCONNECTED);
3421                 break;
3422         case WPA_PTK_DISCONNECTED:
3423                 SM_ENTER(WPA_PTK, INITIALIZE);
3424                 break;
3425         case WPA_PTK_AUTHENTICATION:
3426                 SM_ENTER(WPA_PTK, AUTHENTICATION2);
3427                 break;
3428         case WPA_PTK_AUTHENTICATION2:
3429                 if (wpa_key_mgmt_wpa_ieee8021x(sm->wpa_key_mgmt) &&
3430                     wpa_auth_get_eapol(sm->wpa_auth, sm->addr,
3431                                        WPA_EAPOL_keyRun) > 0)
3432                         SM_ENTER(WPA_PTK, INITPMK);
3433                 else if (wpa_key_mgmt_wpa_psk(sm->wpa_key_mgmt) ||
3434                          sm->wpa_key_mgmt == WPA_KEY_MGMT_OWE
3435                          /* FIX: && 802.1X::keyRun */)
3436                         SM_ENTER(WPA_PTK, INITPSK);
3437                 else if (sm->wpa_key_mgmt == WPA_KEY_MGMT_DPP)
3438                         SM_ENTER(WPA_PTK, INITPMK);
3439                 break;
3440         case WPA_PTK_INITPMK:
3441                 if (wpa_auth_get_eapol(sm->wpa_auth, sm->addr,
3442                                        WPA_EAPOL_keyAvailable) > 0) {
3443                         SM_ENTER(WPA_PTK, PTKSTART);
3444 #ifdef CONFIG_DPP
3445                 } else if (sm->wpa_key_mgmt == WPA_KEY_MGMT_DPP && sm->pmksa) {
3446                         SM_ENTER(WPA_PTK, PTKSTART);
3447 #endif /* CONFIG_DPP */
3448                 } else {
3449                         wpa_auth->dot11RSNA4WayHandshakeFailures++;
3450                         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_INFO,
3451                                         "INITPMK - keyAvailable = false");
3452                         SM_ENTER(WPA_PTK, DISCONNECT);
3453                 }
3454                 break;
3455         case WPA_PTK_INITPSK:
3456                 if (wpa_auth_get_psk(sm->wpa_auth, sm->addr, sm->p2p_dev_addr,
3457                                      NULL, NULL, NULL)) {
3458                         SM_ENTER(WPA_PTK, PTKSTART);
3459 #ifdef CONFIG_SAE
3460                 } else if (wpa_auth_uses_sae(sm) && sm->pmksa) {
3461                         SM_ENTER(WPA_PTK, PTKSTART);
3462 #endif /* CONFIG_SAE */
3463                 } else {
3464                         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_INFO,
3465                                         "no PSK configured for the STA");
3466                         wpa_auth->dot11RSNA4WayHandshakeFailures++;
3467                         SM_ENTER(WPA_PTK, DISCONNECT);
3468                 }
3469                 break;
3470         case WPA_PTK_PTKSTART:
3471                 if (sm->EAPOLKeyReceived && !sm->EAPOLKeyRequest &&
3472                     sm->EAPOLKeyPairwise)
3473                         SM_ENTER(WPA_PTK, PTKCALCNEGOTIATING);
3474                 else if (sm->TimeoutCtr >
3475                          sm->wpa_auth->conf.wpa_pairwise_update_count) {
3476                         wpa_auth->dot11RSNA4WayHandshakeFailures++;
3477                         wpa_auth_vlogger(
3478                                 sm->wpa_auth, sm->addr, LOGGER_DEBUG,
3479                                 "PTKSTART: Retry limit %u reached",
3480                                 sm->wpa_auth->conf.wpa_pairwise_update_count);
3481                         SM_ENTER(WPA_PTK, DISCONNECT);
3482                 } else if (sm->TimeoutEvt)
3483                         SM_ENTER(WPA_PTK, PTKSTART);
3484                 break;
3485         case WPA_PTK_PTKCALCNEGOTIATING:
3486                 if (sm->MICVerified)
3487                         SM_ENTER(WPA_PTK, PTKCALCNEGOTIATING2);
3488                 else if (sm->EAPOLKeyReceived && !sm->EAPOLKeyRequest &&
3489                          sm->EAPOLKeyPairwise)
3490                         SM_ENTER(WPA_PTK, PTKCALCNEGOTIATING);
3491                 else if (sm->TimeoutEvt)
3492                         SM_ENTER(WPA_PTK, PTKSTART);
3493                 break;
3494         case WPA_PTK_PTKCALCNEGOTIATING2:
3495                 SM_ENTER(WPA_PTK, PTKINITNEGOTIATING);
3496                 break;
3497         case WPA_PTK_PTKINITNEGOTIATING:
3498                 if (sm->update_snonce)
3499                         SM_ENTER(WPA_PTK, PTKCALCNEGOTIATING);
3500                 else if (sm->EAPOLKeyReceived && !sm->EAPOLKeyRequest &&
3501                          sm->EAPOLKeyPairwise && sm->MICVerified)
3502                         SM_ENTER(WPA_PTK, PTKINITDONE);
3503                 else if (sm->TimeoutCtr >
3504                          sm->wpa_auth->conf.wpa_pairwise_update_count ||
3505                          (sm->wpa_auth->conf.wpa_disable_eapol_key_retries &&
3506                           sm->TimeoutCtr > 1)) {
3507                         wpa_auth->dot11RSNA4WayHandshakeFailures++;
3508                         wpa_auth_vlogger(
3509                                 sm->wpa_auth, sm->addr, LOGGER_DEBUG,
3510                                 "PTKINITNEGOTIATING: Retry limit %u reached",
3511                                 sm->wpa_auth->conf.wpa_pairwise_update_count);
3512                         SM_ENTER(WPA_PTK, DISCONNECT);
3513                 } else if (sm->TimeoutEvt)
3514                         SM_ENTER(WPA_PTK, PTKINITNEGOTIATING);
3515                 break;
3516         case WPA_PTK_PTKINITDONE:
3517                 break;
3518         }
3519 }
3520
3521
3522 SM_STATE(WPA_PTK_GROUP, IDLE)
3523 {
3524         SM_ENTRY_MA(WPA_PTK_GROUP, IDLE, wpa_ptk_group);
3525         if (sm->Init) {
3526                 /* Init flag is not cleared here, so avoid busy
3527                  * loop by claiming nothing changed. */
3528                 sm->changed = FALSE;
3529         }
3530         sm->GTimeoutCtr = 0;
3531 }
3532
3533
3534 SM_STATE(WPA_PTK_GROUP, REKEYNEGOTIATING)
3535 {
3536         u8 rsc[WPA_KEY_RSC_LEN];
3537         struct wpa_group *gsm = sm->group;
3538         const u8 *kde;
3539         u8 *kde_buf = NULL, *pos, hdr[2];
3540         size_t kde_len;
3541         u8 *gtk, dummy_gtk[32];
3542
3543         SM_ENTRY_MA(WPA_PTK_GROUP, REKEYNEGOTIATING, wpa_ptk_group);
3544
3545         sm->GTimeoutCtr++;
3546         if (sm->wpa_auth->conf.wpa_disable_eapol_key_retries &&
3547             sm->GTimeoutCtr > 1) {
3548                 /* Do not allow retransmission of EAPOL-Key group msg 1/2 */
3549                 return;
3550         }
3551         if (sm->GTimeoutCtr > sm->wpa_auth->conf.wpa_group_update_count) {
3552                 /* No point in sending the EAPOL-Key - we will disconnect
3553                  * immediately following this. */
3554                 return;
3555         }
3556
3557         if (sm->wpa == WPA_VERSION_WPA)
3558                 sm->PInitAKeys = FALSE;
3559         sm->TimeoutEvt = FALSE;
3560         /* Send EAPOL(1, 1, 1, !Pair, G, RSC, GNonce, MIC(PTK), GTK[GN]) */
3561         os_memset(rsc, 0, WPA_KEY_RSC_LEN);
3562         if (gsm->wpa_group_state == WPA_GROUP_SETKEYSDONE)
3563                 wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN, rsc);
3564         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
3565                         "sending 1/2 msg of Group Key Handshake");
3566
3567         gtk = gsm->GTK[gsm->GN - 1];
3568         if (sm->wpa_auth->conf.disable_gtk ||
3569             sm->wpa_key_mgmt == WPA_KEY_MGMT_OSEN) {
3570                 /*
3571                  * Provide unique random GTK to each STA to prevent use
3572                  * of GTK in the BSS.
3573                  */
3574                 if (random_get_bytes(dummy_gtk, gsm->GTK_len) < 0)
3575                         return;
3576                 gtk = dummy_gtk;
3577         }
3578         if (sm->wpa == WPA_VERSION_WPA2) {
3579                 kde_len = 2 + RSN_SELECTOR_LEN + 2 + gsm->GTK_len +
3580                         ieee80211w_kde_len(sm) + ocv_oci_len(sm);
3581                 kde_buf = os_malloc(kde_len);
3582                 if (kde_buf == NULL)
3583                         return;
3584
3585                 kde = pos = kde_buf;
3586                 hdr[0] = gsm->GN & 0x03;
3587                 hdr[1] = 0;
3588                 pos = wpa_add_kde(pos, RSN_KEY_DATA_GROUPKEY, hdr, 2,
3589                                   gtk, gsm->GTK_len);
3590                 pos = ieee80211w_kde_add(sm, pos);
3591                 if (ocv_oci_add(sm, &pos) < 0) {
3592                         os_free(kde_buf);
3593                         return;
3594                 }
3595                 kde_len = pos - kde;
3596         } else {
3597                 kde = gtk;
3598                 kde_len = gsm->GTK_len;
3599         }
3600
3601         wpa_send_eapol(sm->wpa_auth, sm,
3602                        WPA_KEY_INFO_SECURE |
3603                        (wpa_mic_len(sm->wpa_key_mgmt, sm->pmk_len) ?
3604                         WPA_KEY_INFO_MIC : 0) |
3605                        WPA_KEY_INFO_ACK |
3606                        (!sm->Pair ? WPA_KEY_INFO_INSTALL : 0),
3607                        rsc, NULL, kde, kde_len, gsm->GN, 1);
3608
3609         os_free(kde_buf);
3610 }
3611
3612
3613 SM_STATE(WPA_PTK_GROUP, REKEYESTABLISHED)
3614 {
3615 #ifdef CONFIG_OCV
3616         struct wpa_authenticator *wpa_auth = sm->wpa_auth;
3617         const u8 *key_data, *mic;
3618         struct ieee802_1x_hdr *hdr;
3619         struct wpa_eapol_key *key;
3620         struct wpa_eapol_ie_parse kde;
3621         size_t mic_len;
3622         u16 key_data_length;
3623 #endif /* CONFIG_OCV */
3624
3625         SM_ENTRY_MA(WPA_PTK_GROUP, REKEYESTABLISHED, wpa_ptk_group);
3626         sm->EAPOLKeyReceived = FALSE;
3627
3628 #ifdef CONFIG_OCV
3629         mic_len = wpa_mic_len(sm->wpa_key_mgmt, sm->pmk_len);
3630
3631         /*
3632          * Note: last_rx_eapol_key length fields have already been validated in
3633          * wpa_receive().
3634          */
3635         hdr = (struct ieee802_1x_hdr *) sm->last_rx_eapol_key;
3636         key = (struct wpa_eapol_key *) (hdr + 1);
3637         mic = (u8 *) (key + 1);
3638         key_data = mic + mic_len + 2;
3639         key_data_length = WPA_GET_BE16(mic + mic_len);
3640         if (key_data_length > sm->last_rx_eapol_key_len - sizeof(*hdr) -
3641             sizeof(*key) - mic_len - 2)
3642                 return;
3643
3644         if (wpa_parse_kde_ies(key_data, key_data_length, &kde) < 0) {
3645                 wpa_auth_vlogger(wpa_auth, sm->addr, LOGGER_INFO,
3646                                  "received EAPOL-Key group msg 2/2 with invalid Key Data contents");
3647                 return;
3648         }
3649
3650         if (wpa_auth_uses_ocv(sm)) {
3651                 struct wpa_channel_info ci;
3652                 int tx_chanwidth;
3653                 int tx_seg1_idx;
3654
3655                 if (wpa_channel_info(wpa_auth, &ci) != 0) {
3656                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
3657                                         "Failed to get channel info to validate received OCI in EAPOL-Key group 1/2");
3658                         return;
3659                 }
3660
3661                 if (get_sta_tx_parameters(sm,
3662                                           channel_width_to_int(ci.chanwidth),
3663                                           ci.seg1_idx, &tx_chanwidth,
3664                                           &tx_seg1_idx) < 0)
3665                         return;
3666
3667                 if (ocv_verify_tx_params(kde.oci, kde.oci_len, &ci,
3668                                          tx_chanwidth, tx_seg1_idx) != 0) {
3669                         wpa_auth_logger(wpa_auth, sm->addr, LOGGER_INFO,
3670                                         ocv_errorstr);
3671                         return;
3672                 }
3673         }
3674 #endif /* CONFIG_OCV */
3675
3676         if (sm->GUpdateStationKeys)
3677                 sm->group->GKeyDoneStations--;
3678         sm->GUpdateStationKeys = FALSE;
3679         sm->GTimeoutCtr = 0;
3680         /* FIX: MLME.SetProtection.Request(TA, Tx_Rx) */
3681         wpa_auth_vlogger(sm->wpa_auth, sm->addr, LOGGER_INFO,
3682                          "group key handshake completed (%s)",
3683                          sm->wpa == WPA_VERSION_WPA ? "WPA" : "RSN");
3684         sm->has_GTK = TRUE;
3685 }
3686
3687
3688 SM_STATE(WPA_PTK_GROUP, KEYERROR)
3689 {
3690         SM_ENTRY_MA(WPA_PTK_GROUP, KEYERROR, wpa_ptk_group);
3691         if (sm->GUpdateStationKeys)
3692                 sm->group->GKeyDoneStations--;
3693         sm->GUpdateStationKeys = FALSE;
3694         sm->Disconnect = TRUE;
3695         wpa_auth_vlogger(sm->wpa_auth, sm->addr, LOGGER_INFO,
3696                          "group key handshake failed (%s) after %u tries",
3697                          sm->wpa == WPA_VERSION_WPA ? "WPA" : "RSN",
3698                          sm->wpa_auth->conf.wpa_group_update_count);
3699 }
3700
3701
3702 SM_STEP(WPA_PTK_GROUP)
3703 {
3704         if (sm->Init || sm->PtkGroupInit) {
3705                 SM_ENTER(WPA_PTK_GROUP, IDLE);
3706                 sm->PtkGroupInit = FALSE;
3707         } else switch (sm->wpa_ptk_group_state) {
3708         case WPA_PTK_GROUP_IDLE:
3709                 if (sm->GUpdateStationKeys ||
3710                     (sm->wpa == WPA_VERSION_WPA && sm->PInitAKeys))
3711                         SM_ENTER(WPA_PTK_GROUP, REKEYNEGOTIATING);
3712                 break;
3713         case WPA_PTK_GROUP_REKEYNEGOTIATING:
3714                 if (sm->EAPOLKeyReceived && !sm->EAPOLKeyRequest &&
3715                     !sm->EAPOLKeyPairwise && sm->MICVerified)
3716                         SM_ENTER(WPA_PTK_GROUP, REKEYESTABLISHED);
3717                 else if (sm->GTimeoutCtr >
3718                          sm->wpa_auth->conf.wpa_group_update_count ||
3719                          (sm->wpa_auth->conf.wpa_disable_eapol_key_retries &&
3720                           sm->GTimeoutCtr > 1))
3721                         SM_ENTER(WPA_PTK_GROUP, KEYERROR);
3722                 else if (sm->TimeoutEvt)
3723                         SM_ENTER(WPA_PTK_GROUP, REKEYNEGOTIATING);
3724                 break;
3725         case WPA_PTK_GROUP_KEYERROR:
3726                 SM_ENTER(WPA_PTK_GROUP, IDLE);
3727                 break;
3728         case WPA_PTK_GROUP_REKEYESTABLISHED:
3729                 SM_ENTER(WPA_PTK_GROUP, IDLE);
3730                 break;
3731         }
3732 }
3733
3734
3735 static int wpa_gtk_update(struct wpa_authenticator *wpa_auth,
3736                           struct wpa_group *group)
3737 {
3738         int ret = 0;
3739
3740         os_memcpy(group->GNonce, group->Counter, WPA_NONCE_LEN);
3741         inc_byte_array(group->Counter, WPA_NONCE_LEN);
3742         if (wpa_gmk_to_gtk(group->GMK, "Group key expansion",
3743                            wpa_auth->addr, group->GNonce,
3744                            group->GTK[group->GN - 1], group->GTK_len) < 0)
3745                 ret = -1;
3746         wpa_hexdump_key(MSG_DEBUG, "GTK",
3747                         group->GTK[group->GN - 1], group->GTK_len);
3748
3749 #ifdef CONFIG_IEEE80211W
3750         if (wpa_auth->conf.ieee80211w != NO_MGMT_FRAME_PROTECTION) {
3751                 size_t len;
3752                 len = wpa_cipher_key_len(wpa_auth->conf.group_mgmt_cipher);
3753                 os_memcpy(group->GNonce, group->Counter, WPA_NONCE_LEN);
3754                 inc_byte_array(group->Counter, WPA_NONCE_LEN);
3755                 if (wpa_gmk_to_gtk(group->GMK, "IGTK key expansion",
3756                                    wpa_auth->addr, group->GNonce,
3757                                    group->IGTK[group->GN_igtk - 4], len) < 0)
3758                         ret = -1;
3759                 wpa_hexdump_key(MSG_DEBUG, "IGTK",
3760                                 group->IGTK[group->GN_igtk - 4], len);
3761         }
3762 #endif /* CONFIG_IEEE80211W */
3763
3764         return ret;
3765 }
3766
3767
3768 static void wpa_group_gtk_init(struct wpa_authenticator *wpa_auth,
3769                                struct wpa_group *group)
3770 {
3771         wpa_printf(MSG_DEBUG, "WPA: group state machine entering state "
3772                    "GTK_INIT (VLAN-ID %d)", group->vlan_id);
3773         group->changed = FALSE; /* GInit is not cleared here; avoid loop */
3774         group->wpa_group_state = WPA_GROUP_GTK_INIT;
3775
3776         /* GTK[0..N] = 0 */
3777         os_memset(group->GTK, 0, sizeof(group->GTK));
3778         group->GN = 1;
3779         group->GM = 2;
3780 #ifdef CONFIG_IEEE80211W
3781         group->GN_igtk = 4;
3782         group->GM_igtk = 5;
3783 #endif /* CONFIG_IEEE80211W */
3784         /* GTK[GN] = CalcGTK() */
3785         wpa_gtk_update(wpa_auth, group);
3786 }
3787
3788
3789 static int wpa_group_update_sta(struct wpa_state_machine *sm, void *ctx)
3790 {
3791         if (ctx != NULL && ctx != sm->group)
3792                 return 0;
3793
3794         if (sm->wpa_ptk_state != WPA_PTK_PTKINITDONE) {
3795                 wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
3796                                 "Not in PTKINITDONE; skip Group Key update");
3797                 sm->GUpdateStationKeys = FALSE;
3798                 return 0;
3799         }
3800         if (sm->GUpdateStationKeys) {
3801                 /*
3802                  * This should not really happen, so add a debug log entry.
3803                  * Since we clear the GKeyDoneStations before the loop, the
3804                  * station needs to be counted here anyway.
3805                  */
3806                 wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
3807                                 "GUpdateStationKeys was already set when "
3808                                 "marking station for GTK rekeying");
3809         }
3810
3811         /* Do not rekey GTK/IGTK when STA is in WNM-Sleep Mode */
3812         if (sm->is_wnmsleep)
3813                 return 0;
3814
3815         sm->group->GKeyDoneStations++;
3816         sm->GUpdateStationKeys = TRUE;
3817
3818         wpa_sm_step(sm);
3819         return 0;
3820 }
3821
3822
3823 #ifdef CONFIG_WNM_AP
3824 /* update GTK when exiting WNM-Sleep Mode */
3825 void wpa_wnmsleep_rekey_gtk(struct wpa_state_machine *sm)
3826 {
3827         if (sm == NULL || sm->is_wnmsleep)
3828                 return;
3829
3830         wpa_group_update_sta(sm, NULL);
3831 }
3832
3833
3834 void wpa_set_wnmsleep(struct wpa_state_machine *sm, int flag)
3835 {
3836         if (sm)
3837                 sm->is_wnmsleep = !!flag;
3838 }
3839
3840
3841 int wpa_wnmsleep_gtk_subelem(struct wpa_state_machine *sm, u8 *pos)
3842 {
3843         struct wpa_group *gsm = sm->group;
3844         u8 *start = pos;
3845
3846         /*
3847          * GTK subelement:
3848          * Sub-elem ID[1] | Length[1] | Key Info[2] | Key Length[1] | RSC[8] |
3849          * Key[5..32]
3850          */
3851         *pos++ = WNM_SLEEP_SUBELEM_GTK;
3852         *pos++ = 11 + gsm->GTK_len;
3853         /* Key ID in B0-B1 of Key Info */
3854         WPA_PUT_LE16(pos, gsm->GN & 0x03);
3855         pos += 2;
3856         *pos++ = gsm->GTK_len;
3857         if (wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN, pos) != 0)
3858                 return 0;
3859         pos += 8;
3860         os_memcpy(pos, gsm->GTK[gsm->GN - 1], gsm->GTK_len);
3861         pos += gsm->GTK_len;
3862
3863         wpa_printf(MSG_DEBUG, "WNM: GTK Key ID %u in WNM-Sleep Mode exit",
3864                    gsm->GN);
3865         wpa_hexdump_key(MSG_DEBUG, "WNM: GTK in WNM-Sleep Mode exit",
3866                         gsm->GTK[gsm->GN - 1], gsm->GTK_len);
3867
3868         return pos - start;
3869 }
3870
3871
3872 #ifdef CONFIG_IEEE80211W
3873 int wpa_wnmsleep_igtk_subelem(struct wpa_state_machine *sm, u8 *pos)
3874 {
3875         struct wpa_group *gsm = sm->group;
3876         u8 *start = pos;
3877         size_t len = wpa_cipher_key_len(sm->wpa_auth->conf.group_mgmt_cipher);
3878
3879         /*
3880          * IGTK subelement:
3881          * Sub-elem ID[1] | Length[1] | KeyID[2] | PN[6] | Key[16]
3882          */
3883         *pos++ = WNM_SLEEP_SUBELEM_IGTK;
3884         *pos++ = 2 + 6 + len;
3885         WPA_PUT_LE16(pos, gsm->GN_igtk);
3886         pos += 2;
3887         if (wpa_auth_get_seqnum(sm->wpa_auth, NULL, gsm->GN_igtk, pos) != 0)
3888                 return 0;
3889         pos += 6;
3890
3891         os_memcpy(pos, gsm->IGTK[gsm->GN_igtk - 4], len);
3892         pos += len;
3893
3894         wpa_printf(MSG_DEBUG, "WNM: IGTK Key ID %u in WNM-Sleep Mode exit",
3895                    gsm->GN_igtk);
3896         wpa_hexdump_key(MSG_DEBUG, "WNM: IGTK in WNM-Sleep Mode exit",
3897                         gsm->IGTK[gsm->GN_igtk - 4], len);
3898
3899         return pos - start;
3900 }
3901 #endif /* CONFIG_IEEE80211W */
3902 #endif /* CONFIG_WNM_AP */
3903
3904
3905 static void wpa_group_setkeys(struct wpa_authenticator *wpa_auth,
3906                               struct wpa_group *group)
3907 {
3908         int tmp;
3909
3910         wpa_printf(MSG_DEBUG, "WPA: group state machine entering state "
3911                    "SETKEYS (VLAN-ID %d)", group->vlan_id);
3912         group->changed = TRUE;
3913         group->wpa_group_state = WPA_GROUP_SETKEYS;
3914         group->GTKReKey = FALSE;
3915         tmp = group->GM;
3916         group->GM = group->GN;
3917         group->GN = tmp;
3918 #ifdef CONFIG_IEEE80211W
3919         tmp = group->GM_igtk;
3920         group->GM_igtk = group->GN_igtk;
3921         group->GN_igtk = tmp;
3922 #endif /* CONFIG_IEEE80211W */
3923         /* "GKeyDoneStations = GNoStations" is done in more robust way by
3924          * counting the STAs that are marked with GUpdateStationKeys instead of
3925          * including all STAs that could be in not-yet-completed state. */
3926         wpa_gtk_update(wpa_auth, group);
3927
3928         if (group->GKeyDoneStations) {
3929                 wpa_printf(MSG_DEBUG, "wpa_group_setkeys: Unexpected "
3930                            "GKeyDoneStations=%d when starting new GTK rekey",
3931                            group->GKeyDoneStations);
3932                 group->GKeyDoneStations = 0;
3933         }
3934         wpa_auth_for_each_sta(wpa_auth, wpa_group_update_sta, group);
3935         wpa_printf(MSG_DEBUG, "wpa_group_setkeys: GKeyDoneStations=%d",
3936                    group->GKeyDoneStations);
3937 }
3938
3939
3940 static int wpa_group_config_group_keys(struct wpa_authenticator *wpa_auth,
3941                                        struct wpa_group *group)
3942 {
3943         int ret = 0;
3944
3945         if (wpa_auth_set_key(wpa_auth, group->vlan_id,
3946                              wpa_cipher_to_alg(wpa_auth->conf.wpa_group),
3947                              broadcast_ether_addr, group->GN,
3948                              group->GTK[group->GN - 1], group->GTK_len) < 0)
3949                 ret = -1;
3950
3951 #ifdef CONFIG_IEEE80211W
3952         if (wpa_auth->conf.ieee80211w != NO_MGMT_FRAME_PROTECTION) {
3953                 enum wpa_alg alg;
3954                 size_t len;
3955
3956                 alg = wpa_cipher_to_alg(wpa_auth->conf.group_mgmt_cipher);
3957                 len = wpa_cipher_key_len(wpa_auth->conf.group_mgmt_cipher);
3958
3959                 if (ret == 0 &&
3960                     wpa_auth_set_key(wpa_auth, group->vlan_id, alg,
3961                                      broadcast_ether_addr, group->GN_igtk,
3962                                      group->IGTK[group->GN_igtk - 4], len) < 0)
3963                         ret = -1;
3964         }
3965 #endif /* CONFIG_IEEE80211W */
3966
3967         return ret;
3968 }
3969
3970
3971 static int wpa_group_disconnect_cb(struct wpa_state_machine *sm, void *ctx)
3972 {
3973         if (sm->group == ctx) {
3974                 wpa_printf(MSG_DEBUG, "WPA: Mark STA " MACSTR
3975                            " for discconnection due to fatal failure",
3976                            MAC2STR(sm->addr));
3977                 sm->Disconnect = TRUE;
3978         }
3979
3980         return 0;
3981 }
3982
3983
3984 static void wpa_group_fatal_failure(struct wpa_authenticator *wpa_auth,
3985                                     struct wpa_group *group)
3986 {
3987         wpa_printf(MSG_DEBUG, "WPA: group state machine entering state FATAL_FAILURE");
3988         group->changed = TRUE;
3989         group->wpa_group_state = WPA_GROUP_FATAL_FAILURE;
3990         wpa_auth_for_each_sta(wpa_auth, wpa_group_disconnect_cb, group);
3991 }
3992
3993
3994 static int wpa_group_setkeysdone(struct wpa_authenticator *wpa_auth,
3995                                  struct wpa_group *group)
3996 {
3997         wpa_printf(MSG_DEBUG, "WPA: group state machine entering state "
3998                    "SETKEYSDONE (VLAN-ID %d)", group->vlan_id);
3999         group->changed = TRUE;
4000         group->wpa_group_state = WPA_GROUP_SETKEYSDONE;
4001
4002         if (wpa_group_config_group_keys(wpa_auth, group) < 0) {
4003                 wpa_group_fatal_failure(wpa_auth, group);
4004                 return -1;
4005         }
4006
4007         return 0;
4008 }
4009
4010
4011 static void wpa_group_sm_step(struct wpa_authenticator *wpa_auth,
4012                               struct wpa_group *group)
4013 {
4014         if (group->GInit) {
4015                 wpa_group_gtk_init(wpa_auth, group);
4016         } else if (group->wpa_group_state == WPA_GROUP_FATAL_FAILURE) {
4017                 /* Do not allow group operations */
4018         } else if (group->wpa_group_state == WPA_GROUP_GTK_INIT &&
4019                    group->GTKAuthenticator) {
4020                 wpa_group_setkeysdone(wpa_auth, group);
4021         } else if (group->wpa_group_state == WPA_GROUP_SETKEYSDONE &&
4022                    group->GTKReKey) {
4023                 wpa_group_setkeys(wpa_auth, group);
4024         } else if (group->wpa_group_state == WPA_GROUP_SETKEYS) {
4025                 if (group->GKeyDoneStations == 0)
4026                         wpa_group_setkeysdone(wpa_auth, group);
4027                 else if (group->GTKReKey)
4028                         wpa_group_setkeys(wpa_auth, group);
4029         }
4030 }
4031
4032
4033 static int wpa_sm_step(struct wpa_state_machine *sm)
4034 {
4035         if (sm == NULL)
4036                 return 0;
4037
4038         if (sm->in_step_loop) {
4039                 /* This should not happen, but if it does, make sure we do not
4040                  * end up freeing the state machine too early by exiting the
4041                  * recursive call. */
4042                 wpa_printf(MSG_ERROR, "WPA: wpa_sm_step() called recursively");
4043                 return 0;
4044         }
4045
4046         sm->in_step_loop = 1;
4047         do {
4048                 if (sm->pending_deinit)
4049                         break;
4050
4051                 sm->changed = FALSE;
4052                 sm->wpa_auth->group->changed = FALSE;
4053
4054                 SM_STEP_RUN(WPA_PTK);
4055                 if (sm->pending_deinit)
4056                         break;
4057                 SM_STEP_RUN(WPA_PTK_GROUP);
4058                 if (sm->pending_deinit)
4059                         break;
4060                 wpa_group_sm_step(sm->wpa_auth, sm->group);
4061         } while (sm->changed || sm->wpa_auth->group->changed);
4062         sm->in_step_loop = 0;
4063
4064         if (sm->pending_deinit) {
4065                 wpa_printf(MSG_DEBUG, "WPA: Completing pending STA state "
4066                            "machine deinit for " MACSTR, MAC2STR(sm->addr));
4067                 wpa_free_sta_sm(sm);
4068                 return 1;
4069         }
4070         return 0;
4071 }
4072
4073
4074 static void wpa_sm_call_step(void *eloop_ctx, void *timeout_ctx)
4075 {
4076         struct wpa_state_machine *sm = eloop_ctx;
4077         wpa_sm_step(sm);
4078 }
4079
4080
4081 void wpa_auth_sm_notify(struct wpa_state_machine *sm)
4082 {
4083         if (sm == NULL)
4084                 return;
4085         eloop_register_timeout(0, 0, wpa_sm_call_step, sm, NULL);
4086 }
4087
4088
4089 void wpa_gtk_rekey(struct wpa_authenticator *wpa_auth)
4090 {
4091         int tmp, i;
4092         struct wpa_group *group;
4093
4094         if (wpa_auth == NULL)
4095                 return;
4096
4097         group = wpa_auth->group;
4098
4099         for (i = 0; i < 2; i++) {
4100                 tmp = group->GM;
4101                 group->GM = group->GN;
4102                 group->GN = tmp;
4103 #ifdef CONFIG_IEEE80211W
4104                 tmp = group->GM_igtk;
4105                 group->GM_igtk = group->GN_igtk;
4106                 group->GN_igtk = tmp;
4107 #endif /* CONFIG_IEEE80211W */
4108                 wpa_gtk_update(wpa_auth, group);
4109                 wpa_group_config_group_keys(wpa_auth, group);
4110         }
4111 }
4112
4113
4114 static const char * wpa_bool_txt(int val)
4115 {
4116         return val ? "TRUE" : "FALSE";
4117 }
4118
4119
4120 #define RSN_SUITE "%02x-%02x-%02x-%d"
4121 #define RSN_SUITE_ARG(s) \
4122 ((s) >> 24) & 0xff, ((s) >> 16) & 0xff, ((s) >> 8) & 0xff, (s) & 0xff
4123
4124 int wpa_get_mib(struct wpa_authenticator *wpa_auth, char *buf, size_t buflen)
4125 {
4126         int len = 0, ret;
4127         char pmkid_txt[PMKID_LEN * 2 + 1];
4128 #ifdef CONFIG_RSN_PREAUTH
4129         const int preauth = 1;
4130 #else /* CONFIG_RSN_PREAUTH */
4131         const int preauth = 0;
4132 #endif /* CONFIG_RSN_PREAUTH */
4133
4134         if (wpa_auth == NULL)
4135                 return len;
4136
4137         ret = os_snprintf(buf + len, buflen - len,
4138                           "dot11RSNAOptionImplemented=TRUE\n"
4139                           "dot11RSNAPreauthenticationImplemented=%s\n"
4140                           "dot11RSNAEnabled=%s\n"
4141                           "dot11RSNAPreauthenticationEnabled=%s\n",
4142                           wpa_bool_txt(preauth),
4143                           wpa_bool_txt(wpa_auth->conf.wpa & WPA_PROTO_RSN),
4144                           wpa_bool_txt(wpa_auth->conf.rsn_preauth));
4145         if (os_snprintf_error(buflen - len, ret))
4146                 return len;
4147         len += ret;
4148
4149         wpa_snprintf_hex(pmkid_txt, sizeof(pmkid_txt),
4150                          wpa_auth->dot11RSNAPMKIDUsed, PMKID_LEN);
4151
4152         ret = os_snprintf(
4153                 buf + len, buflen - len,
4154                 "dot11RSNAConfigVersion=%u\n"
4155                 "dot11RSNAConfigPairwiseKeysSupported=9999\n"
4156                 /* FIX: dot11RSNAConfigGroupCipher */
4157                 /* FIX: dot11RSNAConfigGroupRekeyMethod */
4158                 /* FIX: dot11RSNAConfigGroupRekeyTime */
4159                 /* FIX: dot11RSNAConfigGroupRekeyPackets */
4160                 "dot11RSNAConfigGroupRekeyStrict=%u\n"
4161                 "dot11RSNAConfigGroupUpdateCount=%u\n"
4162                 "dot11RSNAConfigPairwiseUpdateCount=%u\n"
4163                 "dot11RSNAConfigGroupCipherSize=%u\n"
4164                 "dot11RSNAConfigPMKLifetime=%u\n"
4165                 "dot11RSNAConfigPMKReauthThreshold=%u\n"
4166                 "dot11RSNAConfigNumberOfPTKSAReplayCounters=0\n"
4167                 "dot11RSNAConfigSATimeout=%u\n"
4168                 "dot11RSNAAuthenticationSuiteSelected=" RSN_SUITE "\n"
4169                 "dot11RSNAPairwiseCipherSelected=" RSN_SUITE "\n"
4170                 "dot11RSNAGroupCipherSelected=" RSN_SUITE "\n"
4171                 "dot11RSNAPMKIDUsed=%s\n"
4172                 "dot11RSNAAuthenticationSuiteRequested=" RSN_SUITE "\n"
4173                 "dot11RSNAPairwiseCipherRequested=" RSN_SUITE "\n"
4174                 "dot11RSNAGroupCipherRequested=" RSN_SUITE "\n"
4175                 "dot11RSNATKIPCounterMeasuresInvoked=%u\n"
4176                 "dot11RSNA4WayHandshakeFailures=%u\n"
4177                 "dot11RSNAConfigNumberOfGTKSAReplayCounters=0\n",
4178                 RSN_VERSION,
4179                 !!wpa_auth->conf.wpa_strict_rekey,
4180                 wpa_auth->conf.wpa_group_update_count,
4181                 wpa_auth->conf.wpa_pairwise_update_count,
4182                 wpa_cipher_key_len(wpa_auth->conf.wpa_group) * 8,
4183                 dot11RSNAConfigPMKLifetime,
4184                 dot11RSNAConfigPMKReauthThreshold,
4185                 dot11RSNAConfigSATimeout,
4186                 RSN_SUITE_ARG(wpa_auth->dot11RSNAAuthenticationSuiteSelected),
4187                 RSN_SUITE_ARG(wpa_auth->dot11RSNAPairwiseCipherSelected),
4188                 RSN_SUITE_ARG(wpa_auth->dot11RSNAGroupCipherSelected),
4189                 pmkid_txt,
4190                 RSN_SUITE_ARG(wpa_auth->dot11RSNAAuthenticationSuiteRequested),
4191                 RSN_SUITE_ARG(wpa_auth->dot11RSNAPairwiseCipherRequested),
4192                 RSN_SUITE_ARG(wpa_auth->dot11RSNAGroupCipherRequested),
4193                 wpa_auth->dot11RSNATKIPCounterMeasuresInvoked,
4194                 wpa_auth->dot11RSNA4WayHandshakeFailures);
4195         if (os_snprintf_error(buflen - len, ret))
4196                 return len;
4197         len += ret;
4198
4199         /* TODO: dot11RSNAConfigPairwiseCiphersTable */
4200         /* TODO: dot11RSNAConfigAuthenticationSuitesTable */
4201
4202         /* Private MIB */
4203         ret = os_snprintf(buf + len, buflen - len, "hostapdWPAGroupState=%d\n",
4204                           wpa_auth->group->wpa_group_state);
4205         if (os_snprintf_error(buflen - len, ret))
4206                 return len;
4207         len += ret;
4208
4209         return len;
4210 }
4211
4212
4213 int wpa_get_mib_sta(struct wpa_state_machine *sm, char *buf, size_t buflen)
4214 {
4215         int len = 0, ret;
4216         u32 pairwise = 0;
4217
4218         if (sm == NULL)
4219                 return 0;
4220
4221         /* TODO: FF-FF-FF-FF-FF-FF entry for broadcast/multicast stats */
4222
4223         /* dot11RSNAStatsEntry */
4224
4225         pairwise = wpa_cipher_to_suite(sm->wpa == WPA_VERSION_WPA2 ?
4226                                        WPA_PROTO_RSN : WPA_PROTO_WPA,
4227                                        sm->pairwise);
4228         if (pairwise == 0)
4229                 return 0;
4230
4231         ret = os_snprintf(
4232                 buf + len, buflen - len,
4233                 /* TODO: dot11RSNAStatsIndex */
4234                 "dot11RSNAStatsSTAAddress=" MACSTR "\n"
4235                 "dot11RSNAStatsVersion=1\n"
4236                 "dot11RSNAStatsSelectedPairwiseCipher=" RSN_SUITE "\n"
4237                 /* TODO: dot11RSNAStatsTKIPICVErrors */
4238                 "dot11RSNAStatsTKIPLocalMICFailures=%u\n"
4239                 "dot11RSNAStatsTKIPRemoteMICFailures=%u\n"
4240                 /* TODO: dot11RSNAStatsCCMPReplays */
4241                 /* TODO: dot11RSNAStatsCCMPDecryptErrors */
4242                 /* TODO: dot11RSNAStatsTKIPReplays */,
4243                 MAC2STR(sm->addr),
4244                 RSN_SUITE_ARG(pairwise),
4245                 sm->dot11RSNAStatsTKIPLocalMICFailures,
4246                 sm->dot11RSNAStatsTKIPRemoteMICFailures);
4247         if (os_snprintf_error(buflen - len, ret))
4248                 return len;
4249         len += ret;
4250
4251         /* Private MIB */
4252         ret = os_snprintf(buf + len, buflen - len,
4253                           "wpa=%d\n"
4254                           "AKMSuiteSelector=" RSN_SUITE "\n"
4255                           "hostapdWPAPTKState=%d\n"
4256                           "hostapdWPAPTKGroupState=%d\n",
4257                           sm->wpa,
4258                           RSN_SUITE_ARG(wpa_akm_to_suite(sm->wpa_key_mgmt)),
4259                           sm->wpa_ptk_state,
4260                           sm->wpa_ptk_group_state);
4261         if (os_snprintf_error(buflen - len, ret))
4262                 return len;
4263         len += ret;
4264
4265         return len;
4266 }
4267
4268
4269 void wpa_auth_countermeasures_start(struct wpa_authenticator *wpa_auth)
4270 {
4271         if (wpa_auth)
4272                 wpa_auth->dot11RSNATKIPCounterMeasuresInvoked++;
4273 }
4274
4275
4276 int wpa_auth_pairwise_set(struct wpa_state_machine *sm)
4277 {
4278         return sm && sm->pairwise_set;
4279 }
4280
4281
4282 int wpa_auth_get_pairwise(struct wpa_state_machine *sm)
4283 {
4284         return sm->pairwise;
4285 }
4286
4287
4288 const u8 * wpa_auth_get_pmk(struct wpa_state_machine *sm, int *len)
4289 {
4290         if (!sm)
4291                 return NULL;
4292         *len = sm->pmk_len;
4293         return sm->PMK;
4294 }
4295
4296
4297 int wpa_auth_sta_key_mgmt(struct wpa_state_machine *sm)
4298 {
4299         if (sm == NULL)
4300                 return -1;
4301         return sm->wpa_key_mgmt;
4302 }
4303
4304
4305 int wpa_auth_sta_wpa_version(struct wpa_state_machine *sm)
4306 {
4307         if (sm == NULL)
4308                 return 0;
4309         return sm->wpa;
4310 }
4311
4312
4313 int wpa_auth_sta_ft_tk_already_set(struct wpa_state_machine *sm)
4314 {
4315         if (!sm || !wpa_key_mgmt_ft(sm->wpa_key_mgmt))
4316                 return 0;
4317         return sm->tk_already_set;
4318 }
4319
4320
4321 int wpa_auth_sta_fils_tk_already_set(struct wpa_state_machine *sm)
4322 {
4323         if (!sm || !wpa_key_mgmt_fils(sm->wpa_key_mgmt))
4324                 return 0;
4325         return sm->tk_already_set;
4326 }
4327
4328
4329 int wpa_auth_sta_clear_pmksa(struct wpa_state_machine *sm,
4330                              struct rsn_pmksa_cache_entry *entry)
4331 {
4332         if (sm == NULL || sm->pmksa != entry)
4333                 return -1;
4334         sm->pmksa = NULL;
4335         return 0;
4336 }
4337
4338
4339 struct rsn_pmksa_cache_entry *
4340 wpa_auth_sta_get_pmksa(struct wpa_state_machine *sm)
4341 {
4342         return sm ? sm->pmksa : NULL;
4343 }
4344
4345
4346 void wpa_auth_sta_local_mic_failure_report(struct wpa_state_machine *sm)
4347 {
4348         if (sm)
4349                 sm->dot11RSNAStatsTKIPLocalMICFailures++;
4350 }
4351
4352
4353 const u8 * wpa_auth_get_wpa_ie(struct wpa_authenticator *wpa_auth, size_t *len)
4354 {
4355         if (wpa_auth == NULL)
4356                 return NULL;
4357         *len = wpa_auth->wpa_ie_len;
4358         return wpa_auth->wpa_ie;
4359 }
4360
4361
4362 int wpa_auth_pmksa_add(struct wpa_state_machine *sm, const u8 *pmk,
4363                        unsigned int pmk_len,
4364                        int session_timeout, struct eapol_state_machine *eapol)
4365 {
4366         if (sm == NULL || sm->wpa != WPA_VERSION_WPA2 ||
4367             sm->wpa_auth->conf.disable_pmksa_caching)
4368                 return -1;
4369
4370 #ifdef CONFIG_IEEE80211R_AP
4371         if (pmk_len >= 2 * PMK_LEN && wpa_key_mgmt_ft(sm->wpa_key_mgmt) &&
4372             wpa_key_mgmt_wpa_ieee8021x(sm->wpa_key_mgmt) &&
4373             !wpa_key_mgmt_sha384(sm->wpa_key_mgmt)) {
4374                 /* Cache MPMK/XXKey instead of initial part from MSK */
4375                 pmk = pmk + PMK_LEN;
4376                 pmk_len = PMK_LEN;
4377         } else
4378 #endif /* CONFIG_IEEE80211R_AP */
4379         if (wpa_key_mgmt_sha384(sm->wpa_key_mgmt)) {
4380                 if (pmk_len > PMK_LEN_SUITE_B_192)
4381                         pmk_len = PMK_LEN_SUITE_B_192;
4382         } else if (pmk_len > PMK_LEN) {
4383                 pmk_len = PMK_LEN;
4384         }
4385
4386         wpa_hexdump_key(MSG_DEBUG, "RSN: Cache PMK", pmk, pmk_len);
4387         if (pmksa_cache_auth_add(sm->wpa_auth->pmksa, pmk, pmk_len, NULL,
4388                                  sm->PTK.kck, sm->PTK.kck_len,
4389                                  sm->wpa_auth->addr, sm->addr, session_timeout,
4390                                  eapol, sm->wpa_key_mgmt))
4391                 return 0;
4392
4393         return -1;
4394 }
4395
4396
4397 int wpa_auth_pmksa_add_preauth(struct wpa_authenticator *wpa_auth,
4398                                const u8 *pmk, size_t len, const u8 *sta_addr,
4399                                int session_timeout,
4400                                struct eapol_state_machine *eapol)
4401 {
4402         if (wpa_auth == NULL)
4403                 return -1;
4404
4405         wpa_hexdump_key(MSG_DEBUG, "RSN: Cache PMK from preauth", pmk, len);
4406         if (pmksa_cache_auth_add(wpa_auth->pmksa, pmk, len, NULL,
4407                                  NULL, 0,
4408                                  wpa_auth->addr,
4409                                  sta_addr, session_timeout, eapol,
4410                                  WPA_KEY_MGMT_IEEE8021X))
4411                 return 0;
4412
4413         return -1;
4414 }
4415
4416
4417 int wpa_auth_pmksa_add_sae(struct wpa_authenticator *wpa_auth, const u8 *addr,
4418                            const u8 *pmk, const u8 *pmkid)
4419 {
4420         if (wpa_auth->conf.disable_pmksa_caching)
4421                 return -1;
4422
4423         wpa_hexdump_key(MSG_DEBUG, "RSN: Cache PMK from SAE", pmk, PMK_LEN);
4424         if (pmksa_cache_auth_add(wpa_auth->pmksa, pmk, PMK_LEN, pmkid,
4425                                  NULL, 0,
4426                                  wpa_auth->addr, addr, 0, NULL,
4427                                  WPA_KEY_MGMT_SAE))
4428                 return 0;
4429
4430         return -1;
4431 }
4432
4433
4434 void wpa_auth_add_sae_pmkid(struct wpa_state_machine *sm, const u8 *pmkid)
4435 {
4436         os_memcpy(sm->pmkid, pmkid, PMKID_LEN);
4437         sm->pmkid_set = 1;
4438 }
4439
4440
4441 int wpa_auth_pmksa_add2(struct wpa_authenticator *wpa_auth, const u8 *addr,
4442                         const u8 *pmk, size_t pmk_len, const u8 *pmkid,
4443                         int session_timeout, int akmp)
4444 {
4445         if (wpa_auth->conf.disable_pmksa_caching)
4446                 return -1;
4447
4448         wpa_hexdump_key(MSG_DEBUG, "RSN: Cache PMK (2)", pmk, PMK_LEN);
4449         if (pmksa_cache_auth_add(wpa_auth->pmksa, pmk, pmk_len, pmkid,
4450                                  NULL, 0, wpa_auth->addr, addr, session_timeout,
4451                                  NULL, akmp))
4452                 return 0;
4453
4454         return -1;
4455 }
4456
4457
4458 void wpa_auth_pmksa_remove(struct wpa_authenticator *wpa_auth,
4459                            const u8 *sta_addr)
4460 {
4461         struct rsn_pmksa_cache_entry *pmksa;
4462
4463         if (wpa_auth == NULL || wpa_auth->pmksa == NULL)
4464                 return;
4465         pmksa = pmksa_cache_auth_get(wpa_auth->pmksa, sta_addr, NULL);
4466         if (pmksa) {
4467                 wpa_printf(MSG_DEBUG, "WPA: Remove PMKSA cache entry for "
4468                            MACSTR " based on request", MAC2STR(sta_addr));
4469                 pmksa_cache_free_entry(wpa_auth->pmksa, pmksa);
4470         }
4471 }
4472
4473
4474 int wpa_auth_pmksa_list(struct wpa_authenticator *wpa_auth, char *buf,
4475                         size_t len)
4476 {
4477         if (!wpa_auth || !wpa_auth->pmksa)
4478                 return 0;
4479         return pmksa_cache_auth_list(wpa_auth->pmksa, buf, len);
4480 }
4481
4482
4483 void wpa_auth_pmksa_flush(struct wpa_authenticator *wpa_auth)
4484 {
4485         if (wpa_auth && wpa_auth->pmksa)
4486                 pmksa_cache_auth_flush(wpa_auth->pmksa);
4487 }
4488
4489
4490 #ifdef CONFIG_PMKSA_CACHE_EXTERNAL
4491 #ifdef CONFIG_MESH
4492
4493 int wpa_auth_pmksa_list_mesh(struct wpa_authenticator *wpa_auth, const u8 *addr,
4494                              char *buf, size_t len)
4495 {
4496         if (!wpa_auth || !wpa_auth->pmksa)
4497                 return 0;
4498
4499         return pmksa_cache_auth_list_mesh(wpa_auth->pmksa, addr, buf, len);
4500 }
4501
4502
4503 struct rsn_pmksa_cache_entry *
4504 wpa_auth_pmksa_create_entry(const u8 *aa, const u8 *spa, const u8 *pmk,
4505                             const u8 *pmkid, int expiration)
4506 {
4507         struct rsn_pmksa_cache_entry *entry;
4508         struct os_reltime now;
4509
4510         entry = pmksa_cache_auth_create_entry(pmk, PMK_LEN, pmkid, NULL, 0, aa,
4511                                               spa, 0, NULL, WPA_KEY_MGMT_SAE);
4512         if (!entry)
4513                 return NULL;
4514
4515         os_get_reltime(&now);
4516         entry->expiration = now.sec + expiration;
4517         return entry;
4518 }
4519
4520
4521 int wpa_auth_pmksa_add_entry(struct wpa_authenticator *wpa_auth,
4522                              struct rsn_pmksa_cache_entry *entry)
4523 {
4524         int ret;
4525
4526         if (!wpa_auth || !wpa_auth->pmksa)
4527                 return -1;
4528
4529         ret = pmksa_cache_auth_add_entry(wpa_auth->pmksa, entry);
4530         if (ret < 0)
4531                 wpa_printf(MSG_DEBUG,
4532                            "RSN: Failed to store external PMKSA cache for "
4533                            MACSTR, MAC2STR(entry->spa));
4534
4535         return ret;
4536 }
4537
4538 #endif /* CONFIG_MESH */
4539 #endif /* CONFIG_PMKSA_CACHE_EXTERNAL */
4540
4541
4542 struct rsn_pmksa_cache_entry *
4543 wpa_auth_pmksa_get(struct wpa_authenticator *wpa_auth, const u8 *sta_addr,
4544                    const u8 *pmkid)
4545 {
4546         if (!wpa_auth || !wpa_auth->pmksa)
4547                 return NULL;
4548         return pmksa_cache_auth_get(wpa_auth->pmksa, sta_addr, pmkid);
4549 }
4550
4551
4552 void wpa_auth_pmksa_set_to_sm(struct rsn_pmksa_cache_entry *pmksa,
4553                               struct wpa_state_machine *sm,
4554                               struct wpa_authenticator *wpa_auth,
4555                               u8 *pmkid, u8 *pmk)
4556 {
4557         if (!sm)
4558                 return;
4559
4560         sm->pmksa = pmksa;
4561         os_memcpy(pmk, pmksa->pmk, PMK_LEN);
4562         os_memcpy(pmkid, pmksa->pmkid, PMKID_LEN);
4563         os_memcpy(wpa_auth->dot11RSNAPMKIDUsed, pmksa->pmkid, PMKID_LEN);
4564 }
4565
4566
4567 /*
4568  * Remove and free the group from wpa_authenticator. This is triggered by a
4569  * callback to make sure nobody is currently iterating the group list while it
4570  * gets modified.
4571  */
4572 static void wpa_group_free(struct wpa_authenticator *wpa_auth,
4573                            struct wpa_group *group)
4574 {
4575         struct wpa_group *prev = wpa_auth->group;
4576
4577         wpa_printf(MSG_DEBUG, "WPA: Remove group state machine for VLAN-ID %d",
4578                    group->vlan_id);
4579
4580         while (prev) {
4581                 if (prev->next == group) {
4582                         /* This never frees the special first group as needed */
4583                         prev->next = group->next;
4584                         os_free(group);
4585                         break;
4586                 }
4587                 prev = prev->next;
4588         }
4589
4590 }
4591
4592
4593 /* Increase the reference counter for group */
4594 static void wpa_group_get(struct wpa_authenticator *wpa_auth,
4595                           struct wpa_group *group)
4596 {
4597         /* Skip the special first group */
4598         if (wpa_auth->group == group)
4599                 return;
4600
4601         group->references++;
4602 }
4603
4604
4605 /* Decrease the reference counter and maybe free the group */
4606 static void wpa_group_put(struct wpa_authenticator *wpa_auth,
4607                           struct wpa_group *group)
4608 {
4609         /* Skip the special first group */
4610         if (wpa_auth->group == group)
4611                 return;
4612
4613         group->references--;
4614         if (group->references)
4615                 return;
4616         wpa_group_free(wpa_auth, group);
4617 }
4618
4619
4620 /*
4621  * Add a group that has its references counter set to zero. Caller needs to
4622  * call wpa_group_get() on the return value to mark the entry in use.
4623  */
4624 static struct wpa_group *
4625 wpa_auth_add_group(struct wpa_authenticator *wpa_auth, int vlan_id)
4626 {
4627         struct wpa_group *group;
4628
4629         if (wpa_auth == NULL || wpa_auth->group == NULL)
4630                 return NULL;
4631
4632         wpa_printf(MSG_DEBUG, "WPA: Add group state machine for VLAN-ID %d",
4633                    vlan_id);
4634         group = wpa_group_init(wpa_auth, vlan_id, 0);
4635         if (group == NULL)
4636                 return NULL;
4637
4638         group->next = wpa_auth->group->next;
4639         wpa_auth->group->next = group;
4640
4641         return group;
4642 }
4643
4644
4645 /*
4646  * Enforce that the group state machine for the VLAN is running, increase
4647  * reference counter as interface is up. References might have been increased
4648  * even if a negative value is returned.
4649  * Returns: -1 on error (group missing, group already failed); otherwise, 0
4650  */
4651 int wpa_auth_ensure_group(struct wpa_authenticator *wpa_auth, int vlan_id)
4652 {
4653         struct wpa_group *group;
4654
4655         if (wpa_auth == NULL)
4656                 return 0;
4657
4658         group = wpa_auth->group;
4659         while (group) {
4660                 if (group->vlan_id == vlan_id)
4661                         break;
4662                 group = group->next;
4663         }
4664
4665         if (group == NULL) {
4666                 group = wpa_auth_add_group(wpa_auth, vlan_id);
4667                 if (group == NULL)
4668                         return -1;
4669         }
4670
4671         wpa_printf(MSG_DEBUG,
4672                    "WPA: Ensure group state machine running for VLAN ID %d",
4673                    vlan_id);
4674
4675         wpa_group_get(wpa_auth, group);
4676         group->num_setup_iface++;
4677
4678         if (group->wpa_group_state == WPA_GROUP_FATAL_FAILURE)
4679                 return -1;
4680
4681         return 0;
4682 }
4683
4684
4685 /*
4686  * Decrease reference counter, expected to be zero afterwards.
4687  * returns: -1 on error (group not found, group in fail state)
4688  *          -2 if wpa_group is still referenced
4689  *           0 else
4690  */
4691 int wpa_auth_release_group(struct wpa_authenticator *wpa_auth, int vlan_id)
4692 {
4693         struct wpa_group *group;
4694         int ret = 0;
4695
4696         if (wpa_auth == NULL)
4697                 return 0;
4698
4699         group = wpa_auth->group;
4700         while (group) {
4701                 if (group->vlan_id == vlan_id)
4702                         break;
4703                 group = group->next;
4704         }
4705
4706         if (group == NULL)
4707                 return -1;
4708
4709         wpa_printf(MSG_DEBUG,
4710                    "WPA: Try stopping group state machine for VLAN ID %d",
4711                    vlan_id);
4712
4713         if (group->num_setup_iface <= 0) {
4714                 wpa_printf(MSG_ERROR,
4715                            "WPA: wpa_auth_release_group called more often than wpa_auth_ensure_group for VLAN ID %d, skipping.",
4716                            vlan_id);
4717                 return -1;
4718         }
4719         group->num_setup_iface--;
4720
4721         if (group->wpa_group_state == WPA_GROUP_FATAL_FAILURE)
4722                 ret = -1;
4723
4724         if (group->references > 1) {
4725                 wpa_printf(MSG_DEBUG,
4726                            "WPA: Cannot stop group state machine for VLAN ID %d as references are still hold",
4727                            vlan_id);
4728                 ret = -2;
4729         }
4730
4731         wpa_group_put(wpa_auth, group);
4732
4733         return ret;
4734 }
4735
4736
4737 int wpa_auth_sta_set_vlan(struct wpa_state_machine *sm, int vlan_id)
4738 {
4739         struct wpa_group *group;
4740
4741         if (sm == NULL || sm->wpa_auth == NULL)
4742                 return 0;
4743
4744         group = sm->wpa_auth->group;
4745         while (group) {
4746                 if (group->vlan_id == vlan_id)
4747                         break;
4748                 group = group->next;
4749         }
4750
4751         if (group == NULL) {
4752                 group = wpa_auth_add_group(sm->wpa_auth, vlan_id);
4753                 if (group == NULL)
4754                         return -1;
4755         }
4756
4757         if (sm->group == group)
4758                 return 0;
4759
4760         if (group->wpa_group_state == WPA_GROUP_FATAL_FAILURE)
4761                 return -1;
4762
4763         wpa_printf(MSG_DEBUG, "WPA: Moving STA " MACSTR " to use group state "
4764                    "machine for VLAN ID %d", MAC2STR(sm->addr), vlan_id);
4765
4766         wpa_group_get(sm->wpa_auth, group);
4767         wpa_group_put(sm->wpa_auth, sm->group);
4768         sm->group = group;
4769
4770         return 0;
4771 }
4772
4773
4774 void wpa_auth_eapol_key_tx_status(struct wpa_authenticator *wpa_auth,
4775                                   struct wpa_state_machine *sm, int ack)
4776 {
4777         if (wpa_auth == NULL || sm == NULL)
4778                 return;
4779         wpa_printf(MSG_DEBUG, "WPA: EAPOL-Key TX status for STA " MACSTR
4780                    " ack=%d", MAC2STR(sm->addr), ack);
4781         if (sm->pending_1_of_4_timeout && ack) {
4782                 /*
4783                  * Some deployed supplicant implementations update their SNonce
4784                  * for each EAPOL-Key 2/4 message even within the same 4-way
4785                  * handshake and then fail to use the first SNonce when
4786                  * deriving the PTK. This results in unsuccessful 4-way
4787                  * handshake whenever the relatively short initial timeout is
4788                  * reached and EAPOL-Key 1/4 is retransmitted. Try to work
4789                  * around this by increasing the timeout now that we know that
4790                  * the station has received the frame.
4791                  */
4792                 int timeout_ms = eapol_key_timeout_subseq;
4793                 wpa_printf(MSG_DEBUG, "WPA: Increase initial EAPOL-Key 1/4 "
4794                            "timeout by %u ms because of acknowledged frame",
4795                            timeout_ms);
4796                 eloop_cancel_timeout(wpa_send_eapol_timeout, wpa_auth, sm);
4797                 eloop_register_timeout(timeout_ms / 1000,
4798                                        (timeout_ms % 1000) * 1000,
4799                                        wpa_send_eapol_timeout, wpa_auth, sm);
4800         }
4801
4802 #ifdef CONFIG_TESTING_OPTIONS
4803         if (sm->eapol_status_cb) {
4804                 sm->eapol_status_cb(sm->eapol_status_cb_ctx1,
4805                                     sm->eapol_status_cb_ctx2);
4806                 sm->eapol_status_cb = NULL;
4807         }
4808 #endif /* CONFIG_TESTING_OPTIONS */
4809 }
4810
4811
4812 int wpa_auth_uses_sae(struct wpa_state_machine *sm)
4813 {
4814         if (sm == NULL)
4815                 return 0;
4816         return wpa_key_mgmt_sae(sm->wpa_key_mgmt);
4817 }
4818
4819
4820 int wpa_auth_uses_ft_sae(struct wpa_state_machine *sm)
4821 {
4822         if (sm == NULL)
4823                 return 0;
4824         return sm->wpa_key_mgmt == WPA_KEY_MGMT_FT_SAE;
4825 }
4826
4827
4828 #ifdef CONFIG_P2P
4829 int wpa_auth_get_ip_addr(struct wpa_state_machine *sm, u8 *addr)
4830 {
4831         if (sm == NULL || WPA_GET_BE32(sm->ip_addr) == 0)
4832                 return -1;
4833         os_memcpy(addr, sm->ip_addr, 4);
4834         return 0;
4835 }
4836 #endif /* CONFIG_P2P */
4837
4838
4839 int wpa_auth_radius_das_disconnect_pmksa(struct wpa_authenticator *wpa_auth,
4840                                          struct radius_das_attrs *attr)
4841 {
4842         return pmksa_cache_auth_radius_das_disconnect(wpa_auth->pmksa, attr);
4843 }
4844
4845
4846 void wpa_auth_reconfig_group_keys(struct wpa_authenticator *wpa_auth)
4847 {
4848         struct wpa_group *group;
4849
4850         if (!wpa_auth)
4851                 return;
4852         for (group = wpa_auth->group; group; group = group->next)
4853                 wpa_group_config_group_keys(wpa_auth, group);
4854 }
4855
4856
4857 #ifdef CONFIG_FILS
4858
4859 struct wpa_auth_fils_iter_data {
4860         struct wpa_authenticator *auth;
4861         const u8 *cache_id;
4862         struct rsn_pmksa_cache_entry *pmksa;
4863         const u8 *spa;
4864         const u8 *pmkid;
4865 };
4866
4867
4868 static int wpa_auth_fils_iter(struct wpa_authenticator *a, void *ctx)
4869 {
4870         struct wpa_auth_fils_iter_data *data = ctx;
4871
4872         if (a == data->auth || !a->conf.fils_cache_id_set ||
4873             os_memcmp(a->conf.fils_cache_id, data->cache_id,
4874                       FILS_CACHE_ID_LEN) != 0)
4875                 return 0;
4876         data->pmksa = pmksa_cache_auth_get(a->pmksa, data->spa, data->pmkid);
4877         return data->pmksa != NULL;
4878 }
4879
4880
4881 struct rsn_pmksa_cache_entry *
4882 wpa_auth_pmksa_get_fils_cache_id(struct wpa_authenticator *wpa_auth,
4883                                  const u8 *sta_addr, const u8 *pmkid)
4884 {
4885         struct wpa_auth_fils_iter_data idata;
4886
4887         if (!wpa_auth->conf.fils_cache_id_set)
4888                 return NULL;
4889         idata.auth = wpa_auth;
4890         idata.cache_id = wpa_auth->conf.fils_cache_id;
4891         idata.pmksa = NULL;
4892         idata.spa = sta_addr;
4893         idata.pmkid = pmkid;
4894         wpa_auth_for_each_auth(wpa_auth, wpa_auth_fils_iter, &idata);
4895         return idata.pmksa;
4896 }
4897
4898
4899 #ifdef CONFIG_IEEE80211R_AP
4900 int wpa_auth_write_fte(struct wpa_authenticator *wpa_auth, int use_sha384,
4901                        u8 *buf, size_t len)
4902 {
4903         struct wpa_auth_config *conf = &wpa_auth->conf;
4904
4905         return wpa_write_ftie(conf, use_sha384, conf->r0_key_holder,
4906                               conf->r0_key_holder_len,
4907                               NULL, NULL, buf, len, NULL, 0);
4908 }
4909 #endif /* CONFIG_IEEE80211R_AP */
4910
4911
4912 void wpa_auth_get_fils_aead_params(struct wpa_state_machine *sm,
4913                                    u8 *fils_anonce, u8 *fils_snonce,
4914                                    u8 *fils_kek, size_t *fils_kek_len)
4915 {
4916         os_memcpy(fils_anonce, sm->ANonce, WPA_NONCE_LEN);
4917         os_memcpy(fils_snonce, sm->SNonce, WPA_NONCE_LEN);
4918         os_memcpy(fils_kek, sm->PTK.kek, WPA_KEK_MAX_LEN);
4919         *fils_kek_len = sm->PTK.kek_len;
4920 }
4921
4922
4923 void wpa_auth_add_fils_pmk_pmkid(struct wpa_state_machine *sm, const u8 *pmk,
4924                                  size_t pmk_len, const u8 *pmkid)
4925 {
4926         os_memcpy(sm->PMK, pmk, pmk_len);
4927         sm->pmk_len = pmk_len;
4928         os_memcpy(sm->pmkid, pmkid, PMKID_LEN);
4929         sm->pmkid_set = 1;
4930 }
4931
4932 #endif /* CONFIG_FILS */
4933
4934
4935 void wpa_auth_set_auth_alg(struct wpa_state_machine *sm, u16 auth_alg)
4936 {
4937         if (sm)
4938                 sm->auth_alg = auth_alg;
4939 }
4940
4941
4942 #ifdef CONFIG_DPP2
4943 void wpa_auth_set_dpp_z(struct wpa_state_machine *sm, const struct wpabuf *z)
4944 {
4945         if (sm) {
4946                 wpabuf_clear_free(sm->dpp_z);
4947                 sm->dpp_z = z ? wpabuf_dup(z) : NULL;
4948         }
4949 }
4950 #endif /* CONFIG_DPP2 */
4951
4952
4953 #ifdef CONFIG_TESTING_OPTIONS
4954
4955 int wpa_auth_resend_m1(struct wpa_state_machine *sm, int change_anonce,
4956                        void (*cb)(void *ctx1, void *ctx2),
4957                        void *ctx1, void *ctx2)
4958 {
4959         const u8 *anonce = sm->ANonce;
4960         u8 anonce_buf[WPA_NONCE_LEN];
4961
4962         if (change_anonce) {
4963                 if (random_get_bytes(anonce_buf, WPA_NONCE_LEN))
4964                         return -1;
4965                 anonce = anonce_buf;
4966         }
4967
4968         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
4969                         "sending 1/4 msg of 4-Way Handshake (TESTING)");
4970         wpa_send_eapol(sm->wpa_auth, sm,
4971                        WPA_KEY_INFO_ACK | WPA_KEY_INFO_KEY_TYPE, NULL,
4972                        anonce, NULL, 0, 0, 0);
4973         return 0;
4974 }
4975
4976
4977 int wpa_auth_resend_m3(struct wpa_state_machine *sm,
4978                        void (*cb)(void *ctx1, void *ctx2),
4979                        void *ctx1, void *ctx2)
4980 {
4981         u8 rsc[WPA_KEY_RSC_LEN], *_rsc, *gtk, *kde, *pos;
4982 #ifdef CONFIG_IEEE80211W
4983         u8 *opos;
4984 #endif /* CONFIG_IEEE80211W */
4985         size_t gtk_len, kde_len;
4986         struct wpa_group *gsm = sm->group;
4987         u8 *wpa_ie;
4988         int wpa_ie_len, secure, keyidx, encr = 0;
4989
4990         /* Send EAPOL(1, 1, 1, Pair, P, RSC, ANonce, MIC(PTK), RSNIE, [MDIE],
4991            GTK[GN], IGTK, [FTIE], [TIE * 2])
4992          */
4993
4994         /* Use 0 RSC */
4995         os_memset(rsc, 0, WPA_KEY_RSC_LEN);
4996         /* If FT is used, wpa_auth->wpa_ie includes both RSNIE and MDIE */
4997         wpa_ie = sm->wpa_auth->wpa_ie;
4998         wpa_ie_len = sm->wpa_auth->wpa_ie_len;
4999         if (sm->wpa == WPA_VERSION_WPA &&
5000             (sm->wpa_auth->conf.wpa & WPA_PROTO_RSN) &&
5001             wpa_ie_len > wpa_ie[1] + 2 && wpa_ie[0] == WLAN_EID_RSN) {
5002                 /* WPA-only STA, remove RSN IE and possible MDIE */
5003                 wpa_ie = wpa_ie + wpa_ie[1] + 2;
5004                 if (wpa_ie[0] == WLAN_EID_MOBILITY_DOMAIN)
5005                         wpa_ie = wpa_ie + wpa_ie[1] + 2;
5006                 wpa_ie_len = wpa_ie[1] + 2;
5007         }
5008         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
5009                         "sending 3/4 msg of 4-Way Handshake (TESTING)");
5010         if (sm->wpa == WPA_VERSION_WPA2) {
5011                 /* WPA2 send GTK in the 4-way handshake */
5012                 secure = 1;
5013                 gtk = gsm->GTK[gsm->GN - 1];
5014                 gtk_len = gsm->GTK_len;
5015                 keyidx = gsm->GN;
5016                 _rsc = rsc;
5017                 encr = 1;
5018         } else {
5019                 /* WPA does not include GTK in msg 3/4 */
5020                 secure = 0;
5021                 gtk = NULL;
5022                 gtk_len = 0;
5023                 keyidx = 0;
5024                 _rsc = NULL;
5025                 if (sm->rx_eapol_key_secure) {
5026                         /*
5027                          * It looks like Windows 7 supplicant tries to use
5028                          * Secure bit in msg 2/4 after having reported Michael
5029                          * MIC failure and it then rejects the 4-way handshake
5030                          * if msg 3/4 does not set Secure bit. Work around this
5031                          * by setting the Secure bit here even in the case of
5032                          * WPA if the supplicant used it first.
5033                          */
5034                         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
5035                                         "STA used Secure bit in WPA msg 2/4 - "
5036                                         "set Secure for 3/4 as workaround");
5037                         secure = 1;
5038                 }
5039         }
5040
5041         kde_len = wpa_ie_len + ieee80211w_kde_len(sm) + ocv_oci_len(sm);
5042         if (gtk)
5043                 kde_len += 2 + RSN_SELECTOR_LEN + 2 + gtk_len;
5044 #ifdef CONFIG_IEEE80211R_AP
5045         if (wpa_key_mgmt_ft(sm->wpa_key_mgmt)) {
5046                 kde_len += 2 + PMKID_LEN; /* PMKR1Name into RSN IE */
5047                 kde_len += 300; /* FTIE + 2 * TIE */
5048         }
5049 #endif /* CONFIG_IEEE80211R_AP */
5050         kde = os_malloc(kde_len);
5051         if (kde == NULL)
5052                 return -1;
5053
5054         pos = kde;
5055         os_memcpy(pos, wpa_ie, wpa_ie_len);
5056         pos += wpa_ie_len;
5057 #ifdef CONFIG_IEEE80211R_AP
5058         if (wpa_key_mgmt_ft(sm->wpa_key_mgmt)) {
5059                 int res;
5060                 size_t elen;
5061
5062                 elen = pos - kde;
5063                 res = wpa_insert_pmkid(kde, &elen, sm->pmk_r1_name);
5064                 if (res < 0) {
5065                         wpa_printf(MSG_ERROR, "FT: Failed to insert "
5066                                    "PMKR1Name into RSN IE in EAPOL-Key data");
5067                         os_free(kde);
5068                         return -1;
5069                 }
5070                 pos -= wpa_ie_len;
5071                 pos += elen;
5072         }
5073 #endif /* CONFIG_IEEE80211R_AP */
5074         if (gtk) {
5075                 u8 hdr[2];
5076                 hdr[0] = keyidx & 0x03;
5077                 hdr[1] = 0;
5078                 pos = wpa_add_kde(pos, RSN_KEY_DATA_GROUPKEY, hdr, 2,
5079                                   gtk, gtk_len);
5080         }
5081 #ifdef CONFIG_IEEE80211W
5082         opos = pos;
5083         pos = ieee80211w_kde_add(sm, pos);
5084         if (pos - opos >= 2 + RSN_SELECTOR_LEN + WPA_IGTK_KDE_PREFIX_LEN) {
5085                 /* skip KDE header and keyid */
5086                 opos += 2 + RSN_SELECTOR_LEN + 2;
5087                 os_memset(opos, 0, 6); /* clear PN */
5088         }
5089 #endif /* CONFIG_IEEE80211W */
5090         if (ocv_oci_add(sm, &pos) < 0) {
5091                 os_free(kde);
5092                 return -1;
5093         }
5094
5095 #ifdef CONFIG_IEEE80211R_AP
5096         if (wpa_key_mgmt_ft(sm->wpa_key_mgmt)) {
5097                 int res;
5098                 struct wpa_auth_config *conf;
5099
5100                 conf = &sm->wpa_auth->conf;
5101                 if (sm->assoc_resp_ftie &&
5102                     kde + kde_len - pos >= 2 + sm->assoc_resp_ftie[1]) {
5103                         os_memcpy(pos, sm->assoc_resp_ftie,
5104                                   2 + sm->assoc_resp_ftie[1]);
5105                         res = 2 + sm->assoc_resp_ftie[1];
5106                 } else {
5107                         int use_sha384 = wpa_key_mgmt_sha384(sm->wpa_key_mgmt);
5108
5109                         res = wpa_write_ftie(conf, use_sha384,
5110                                              conf->r0_key_holder,
5111                                              conf->r0_key_holder_len,
5112                                              NULL, NULL, pos,
5113                                              kde + kde_len - pos,
5114                                              NULL, 0);
5115                 }
5116                 if (res < 0) {
5117                         wpa_printf(MSG_ERROR, "FT: Failed to insert FTIE "
5118                                    "into EAPOL-Key Key Data");
5119                         os_free(kde);
5120                         return -1;
5121                 }
5122                 pos += res;
5123
5124                 /* TIE[ReassociationDeadline] (TU) */
5125                 *pos++ = WLAN_EID_TIMEOUT_INTERVAL;
5126                 *pos++ = 5;
5127                 *pos++ = WLAN_TIMEOUT_REASSOC_DEADLINE;
5128                 WPA_PUT_LE32(pos, conf->reassociation_deadline);
5129                 pos += 4;
5130
5131                 /* TIE[KeyLifetime] (seconds) */
5132                 *pos++ = WLAN_EID_TIMEOUT_INTERVAL;
5133                 *pos++ = 5;
5134                 *pos++ = WLAN_TIMEOUT_KEY_LIFETIME;
5135                 WPA_PUT_LE32(pos, conf->r0_key_lifetime);
5136                 pos += 4;
5137         }
5138 #endif /* CONFIG_IEEE80211R_AP */
5139
5140         wpa_send_eapol(sm->wpa_auth, sm,
5141                        (secure ? WPA_KEY_INFO_SECURE : 0) |
5142                        (wpa_mic_len(sm->wpa_key_mgmt, sm->pmk_len) ?
5143                         WPA_KEY_INFO_MIC : 0) |
5144                        WPA_KEY_INFO_ACK | WPA_KEY_INFO_INSTALL |
5145                        WPA_KEY_INFO_KEY_TYPE,
5146                        _rsc, sm->ANonce, kde, pos - kde, keyidx, encr);
5147         os_free(kde);
5148         return 0;
5149 }
5150
5151
5152 int wpa_auth_resend_group_m1(struct wpa_state_machine *sm,
5153                              void (*cb)(void *ctx1, void *ctx2),
5154                              void *ctx1, void *ctx2)
5155 {
5156         u8 rsc[WPA_KEY_RSC_LEN];
5157         struct wpa_group *gsm = sm->group;
5158         const u8 *kde;
5159         u8 *kde_buf = NULL, *pos, hdr[2];
5160 #ifdef CONFIG_IEEE80211W
5161         u8 *opos;
5162 #endif /* CONFIG_IEEE80211W */
5163         size_t kde_len;
5164         u8 *gtk;
5165
5166         /* Send EAPOL(1, 1, 1, !Pair, G, RSC, GNonce, MIC(PTK), GTK[GN]) */
5167         os_memset(rsc, 0, WPA_KEY_RSC_LEN);
5168         /* Use 0 RSC */
5169         wpa_auth_logger(sm->wpa_auth, sm->addr, LOGGER_DEBUG,
5170                         "sending 1/2 msg of Group Key Handshake (TESTING)");
5171
5172         gtk = gsm->GTK[gsm->GN - 1];
5173         if (sm->wpa == WPA_VERSION_WPA2) {
5174                 kde_len = 2 + RSN_SELECTOR_LEN + 2 + gsm->GTK_len +
5175                         ieee80211w_kde_len(sm) + ocv_oci_len(sm);
5176                 kde_buf = os_malloc(kde_len);
5177                 if (kde_buf == NULL)
5178                         return -1;
5179
5180                 kde = pos = kde_buf;
5181                 hdr[0] = gsm->GN & 0x03;
5182                 hdr[1] = 0;
5183                 pos = wpa_add_kde(pos, RSN_KEY_DATA_GROUPKEY, hdr, 2,
5184                                   gtk, gsm->GTK_len);
5185 #ifdef CONFIG_IEEE80211W
5186                 opos = pos;
5187                 pos = ieee80211w_kde_add(sm, pos);
5188                 if (pos - opos >=
5189                     2 + RSN_SELECTOR_LEN + WPA_IGTK_KDE_PREFIX_LEN) {
5190                         /* skip KDE header and keyid */
5191                         opos += 2 + RSN_SELECTOR_LEN + 2;
5192                         os_memset(opos, 0, 6); /* clear PN */
5193                 }
5194 #endif /* CONFIG_IEEE80211W */
5195                 if (ocv_oci_add(sm, &pos) < 0) {
5196                         os_free(kde_buf);
5197                         return -1;
5198                 }
5199                 kde_len = pos - kde;
5200         } else {
5201                 kde = gtk;
5202                 kde_len = gsm->GTK_len;
5203         }
5204
5205         sm->eapol_status_cb = cb;
5206         sm->eapol_status_cb_ctx1 = ctx1;
5207         sm->eapol_status_cb_ctx2 = ctx2;
5208
5209         wpa_send_eapol(sm->wpa_auth, sm,
5210                        WPA_KEY_INFO_SECURE |
5211                        (wpa_mic_len(sm->wpa_key_mgmt, sm->pmk_len) ?
5212                         WPA_KEY_INFO_MIC : 0) |
5213                        WPA_KEY_INFO_ACK |
5214                        (!sm->Pair ? WPA_KEY_INFO_INSTALL : 0),
5215                        rsc, NULL, kde, kde_len, gsm->GN, 1);
5216
5217         os_free(kde_buf);
5218         return 0;
5219 }
5220
5221
5222 int wpa_auth_rekey_gtk(struct wpa_authenticator *wpa_auth)
5223 {
5224         if (!wpa_auth)
5225                 return -1;
5226         eloop_cancel_timeout(wpa_rekey_gtk, wpa_auth, NULL);
5227         return eloop_register_timeout(0, 0, wpa_rekey_gtk, wpa_auth, NULL);
5228 }
5229
5230 #endif /* CONFIG_TESTING_OPTIONS */