contrib/wpa/src/rsn_supp/wpa.c

   1 /*
   2  * WPA Supplicant - WPA state machine and EAPOL-Key processing
   3  * Copyright (c) 2003-2015, Jouni Malinen <j@w1.fi>
   4  * Copyright(c) 2015 Intel Deutschland GmbH
   5  *
   6  * This software may be distributed under the terms of the BSD license.
   7  * See README for more details.
   8  */
   9
  10 #include "includes.h"
  11
  12 #include "common.h"
  13 #include "crypto/aes_wrap.h"
  14 #include "crypto/crypto.h"
  15 #include "crypto/random.h"
  16 #include "common/ieee802_11_defs.h"
  17 #include "eapol_supp/eapol_supp_sm.h"
  18 #include "wpa.h"
  19 #include "eloop.h"
  20 #include "preauth.h"
  21 #include "pmksa_cache.h"
  22 #include "wpa_i.h"
  23 #include "wpa_ie.h"
  24 #include "peerkey.h"
  25
  26
  27 static const u8 null_rsc[8] = { 0, 0, 0, 0, 0, 0, 0, 0 };
  28
  29
  30 /**
  31  * wpa_eapol_key_send - Send WPA/RSN EAPOL-Key message
  32  * @sm: Pointer to WPA state machine data from wpa_sm_init()
  33  * @kck: Key Confirmation Key (KCK, part of PTK)
  34  * @kck_len: KCK length in octets
  35  * @ver: Version field from Key Info
  36  * @dest: Destination address for the frame
  37  * @proto: Ethertype (usually ETH_P_EAPOL)
  38  * @msg: EAPOL-Key message
  39  * @msg_len: Length of message
  40  * @key_mic: Pointer to the buffer to which the EAPOL-Key MIC is written
  41  * Returns: >= 0 on success, < 0 on failure
  42  */
  43 int wpa_eapol_key_send(struct wpa_sm *sm, const u8 *kck, size_t kck_len,
  44                        int ver, const u8 *dest, u16 proto,
  45                        u8 *msg, size_t msg_len, u8 *key_mic)
  46 {
  47         int ret = -1;
  48         size_t mic_len = wpa_mic_len(sm->key_mgmt);
  49
  50         if (is_zero_ether_addr(dest) && is_zero_ether_addr(sm->bssid)) {
  51                 /*
  52                  * Association event was not yet received; try to fetch
  53                  * BSSID from the driver.
  54                  */
  55                 if (wpa_sm_get_bssid(sm, sm->bssid) < 0) {
  56                         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
  57                                 "WPA: Failed to read BSSID for "
  58                                 "EAPOL-Key destination address");
  59                 } else {
  60                         dest = sm->bssid;
  61                         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
  62                                 "WPA: Use BSSID (" MACSTR
  63                                 ") as the destination for EAPOL-Key",
  64                                 MAC2STR(dest));
  65                 }
  66         }
  67         if (key_mic &&
  68             wpa_eapol_key_mic(kck, kck_len, sm->key_mgmt, ver, msg, msg_len,
  69                               key_mic)) {
  70                 wpa_msg(sm->ctx->msg_ctx, MSG_ERROR,
  71                         "WPA: Failed to generate EAPOL-Key version %d key_mgmt 0x%x MIC",
  72                         ver, sm->key_mgmt);
  73                 goto out;
  74         }
  75         wpa_hexdump_key(MSG_DEBUG, "WPA: KCK", kck, kck_len);
  76         wpa_hexdump(MSG_DEBUG, "WPA: Derived Key MIC", key_mic, mic_len);
  77         wpa_hexdump(MSG_MSGDUMP, "WPA: TX EAPOL-Key", msg, msg_len);
  78         ret = wpa_sm_ether_send(sm, dest, proto, msg, msg_len);
  79         eapol_sm_notify_tx_eapol_key(sm->eapol);
  80 out:
  81         os_free(msg);
  82         return ret;
  83 }
  84
  85
  86 /**
  87  * wpa_sm_key_request - Send EAPOL-Key Request
  88  * @sm: Pointer to WPA state machine data from wpa_sm_init()
  89  * @error: Indicate whether this is an Michael MIC error report
  90  * @pairwise: 1 = error report for pairwise packet, 0 = for group packet
  91  *
  92  * Send an EAPOL-Key Request to the current authenticator. This function is
  93  * used to request rekeying and it is usually called when a local Michael MIC
  94  * failure is detected.
  95  */
  96 void wpa_sm_key_request(struct wpa_sm *sm, int error, int pairwise)
  97 {
  98         size_t mic_len, hdrlen, rlen;
  99         struct wpa_eapol_key *reply;
 100         struct wpa_eapol_key_192 *reply192;
 101         int key_info, ver;
 102         u8 bssid[ETH_ALEN], *rbuf, *key_mic;
 103
 104         if (sm->key_mgmt == WPA_KEY_MGMT_OSEN ||
 105             wpa_key_mgmt_suite_b(sm->key_mgmt))
 106                 ver = WPA_KEY_INFO_TYPE_AKM_DEFINED;
 107         else if (wpa_key_mgmt_ft(sm->key_mgmt) ||
 108                  wpa_key_mgmt_sha256(sm->key_mgmt))
 109                 ver = WPA_KEY_INFO_TYPE_AES_128_CMAC;
 110         else if (sm->pairwise_cipher != WPA_CIPHER_TKIP)
 111                 ver = WPA_KEY_INFO_TYPE_HMAC_SHA1_AES;
 112         else
 113                 ver = WPA_KEY_INFO_TYPE_HMAC_MD5_RC4;
 114
 115         if (wpa_sm_get_bssid(sm, bssid) < 0) {
 116                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 117                         "Failed to read BSSID for EAPOL-Key request");
 118                 return;
 119         }
 120
 121         mic_len = wpa_mic_len(sm->key_mgmt);
 122         hdrlen = mic_len == 24 ? sizeof(*reply192) : sizeof(*reply);
 123         rbuf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_KEY, NULL,
 124                                   hdrlen, &rlen, (void *) &reply);
 125         if (rbuf == NULL)
 126                 return;
 127         reply192 = (struct wpa_eapol_key_192 *) reply;
 128
 129         reply->type = (sm->proto == WPA_PROTO_RSN ||
 130                        sm->proto == WPA_PROTO_OSEN) ?
 131                 EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
 132         key_info = WPA_KEY_INFO_REQUEST | ver;
 133         if (sm->ptk_set)
 134                 key_info |= WPA_KEY_INFO_MIC | WPA_KEY_INFO_SECURE;
 135         if (error)
 136                 key_info |= WPA_KEY_INFO_ERROR;
 137         if (pairwise)
 138                 key_info |= WPA_KEY_INFO_KEY_TYPE;
 139         WPA_PUT_BE16(reply->key_info, key_info);
 140         WPA_PUT_BE16(reply->key_length, 0);
 141         os_memcpy(reply->replay_counter, sm->request_counter,
 142                   WPA_REPLAY_COUNTER_LEN);
 143         inc_byte_array(sm->request_counter, WPA_REPLAY_COUNTER_LEN);
 144
 145         if (mic_len == 24)
 146                 WPA_PUT_BE16(reply192->key_data_length, 0);
 147         else
 148                 WPA_PUT_BE16(reply->key_data_length, 0);
 149         if (!(key_info & WPA_KEY_INFO_MIC))
 150                 key_mic = NULL;
 151         else
 152                 key_mic = reply192->key_mic; /* same offset in reply */
 153
 154         wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
 155                 "WPA: Sending EAPOL-Key Request (error=%d "
 156                 "pairwise=%d ptk_set=%d len=%lu)",
 157                 error, pairwise, sm->ptk_set, (unsigned long) rlen);
 158         wpa_eapol_key_send(sm, sm->ptk.kck, sm->ptk.kck_len, ver, bssid,
 159                            ETH_P_EAPOL, rbuf, rlen, key_mic);
 160 }
 161
 162
 163 static void wpa_supplicant_key_mgmt_set_pmk(struct wpa_sm *sm)
 164 {
 165 #ifdef CONFIG_IEEE80211R
 166         if (sm->key_mgmt == WPA_KEY_MGMT_FT_IEEE8021X) {
 167                 if (wpa_sm_key_mgmt_set_pmk(sm, sm->xxkey, sm->xxkey_len))
 168                         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 169                                 "RSN: Cannot set low order 256 bits of MSK for key management offload");
 170         } else {
 171 #endif /* CONFIG_IEEE80211R */
 172                 if (wpa_sm_key_mgmt_set_pmk(sm, sm->pmk, sm->pmk_len))
 173                         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 174                                 "RSN: Cannot set PMK for key management offload");
 175 #ifdef CONFIG_IEEE80211R
 176         }
 177 #endif /* CONFIG_IEEE80211R */
 178 }
 179
 180
 181 static int wpa_supplicant_get_pmk(struct wpa_sm *sm,
 182                                   const unsigned char *src_addr,
 183                                   const u8 *pmkid)
 184 {
 185         int abort_cached = 0;
 186
 187         if (pmkid && !sm->cur_pmksa) {
 188                 /* When using drivers that generate RSN IE, wpa_supplicant may
 189                  * not have enough time to get the association information
 190                  * event before receiving this 1/4 message, so try to find a
 191                  * matching PMKSA cache entry here. */
 192                 sm->cur_pmksa = pmksa_cache_get(sm->pmksa, src_addr, pmkid,
 193                                                 NULL);
 194                 if (sm->cur_pmksa) {
 195                         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 196                                 "RSN: found matching PMKID from PMKSA cache");
 197                 } else {
 198                         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 199                                 "RSN: no matching PMKID found");
 200                         abort_cached = 1;
 201                 }
 202         }
 203
 204         if (pmkid && sm->cur_pmksa &&
 205             os_memcmp_const(pmkid, sm->cur_pmksa->pmkid, PMKID_LEN) == 0) {
 206                 wpa_hexdump(MSG_DEBUG, "RSN: matched PMKID", pmkid, PMKID_LEN);
 207                 wpa_sm_set_pmk_from_pmksa(sm);
 208                 wpa_hexdump_key(MSG_DEBUG, "RSN: PMK from PMKSA cache",
 209                                 sm->pmk, sm->pmk_len);
 210                 eapol_sm_notify_cached(sm->eapol);
 211 #ifdef CONFIG_IEEE80211R
 212                 sm->xxkey_len = 0;
 213 #endif /* CONFIG_IEEE80211R */
 214         } else if (wpa_key_mgmt_wpa_ieee8021x(sm->key_mgmt) && sm->eapol) {
 215                 int res, pmk_len;
 216
 217                 if (sm->key_mgmt & WPA_KEY_MGMT_IEEE8021X_SUITE_B_192)
 218                         pmk_len = PMK_LEN_SUITE_B_192;
 219                 else
 220                         pmk_len = PMK_LEN;
 221                 res = eapol_sm_get_key(sm->eapol, sm->pmk, pmk_len);
 222                 if (res) {
 223                         if (pmk_len == PMK_LEN) {
 224                                 /*
 225                                  * EAP-LEAP is an exception from other EAP
 226                                  * methods: it uses only 16-byte PMK.
 227                                  */
 228                                 res = eapol_sm_get_key(sm->eapol, sm->pmk, 16);
 229                                 pmk_len = 16;
 230                         }
 231                 } else {
 232 #ifdef CONFIG_IEEE80211R
 233                         u8 buf[2 * PMK_LEN];
 234                         if (eapol_sm_get_key(sm->eapol, buf, 2 * PMK_LEN) == 0)
 235                         {
 236                                 os_memcpy(sm->xxkey, buf + PMK_LEN, PMK_LEN);
 237                                 sm->xxkey_len = PMK_LEN;
 238                                 os_memset(buf, 0, sizeof(buf));
 239                         }
 240 #endif /* CONFIG_IEEE80211R */
 241                 }
 242                 if (res == 0) {
 243                         struct rsn_pmksa_cache_entry *sa = NULL;
 244                         wpa_hexdump_key(MSG_DEBUG, "WPA: PMK from EAPOL state "
 245                                         "machines", sm->pmk, pmk_len);
 246                         sm->pmk_len = pmk_len;
 247                         wpa_supplicant_key_mgmt_set_pmk(sm);
 248                         if (sm->proto == WPA_PROTO_RSN &&
 249                             !wpa_key_mgmt_suite_b(sm->key_mgmt) &&
 250                             !wpa_key_mgmt_ft(sm->key_mgmt)) {
 251                                 sa = pmksa_cache_add(sm->pmksa,
 252                                                      sm->pmk, pmk_len, NULL,
 253                                                      NULL, 0,
 254                                                      src_addr, sm->own_addr,
 255                                                      sm->network_ctx,
 256                                                      sm->key_mgmt);
 257                         }
 258                         if (!sm->cur_pmksa && pmkid &&
 259                             pmksa_cache_get(sm->pmksa, src_addr, pmkid, NULL))
 260                         {
 261                                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 262                                         "RSN: the new PMK matches with the "
 263                                         "PMKID");
 264                                 abort_cached = 0;
 265                         } else if (sa && !sm->cur_pmksa && pmkid) {
 266                                 /*
 267                                  * It looks like the authentication server
 268                                  * derived mismatching MSK. This should not
 269                                  * really happen, but bugs happen.. There is not
 270                                  * much we can do here without knowing what
 271                                  * exactly caused the server to misbehave.
 272                                  */
 273                                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
 274                                         "RSN: PMKID mismatch - authentication server may have derived different MSK?!");
 275                                 return -1;
 276                         }
 277
 278                         if (!sm->cur_pmksa)
 279                                 sm->cur_pmksa = sa;
 280                 } else {
 281                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 282                                 "WPA: Failed to get master session key from "
 283                                 "EAPOL state machines - key handshake "
 284                                 "aborted");
 285                         if (sm->cur_pmksa) {
 286                                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 287                                         "RSN: Cancelled PMKSA caching "
 288                                         "attempt");
 289                                 sm->cur_pmksa = NULL;
 290                                 abort_cached = 1;
 291                         } else if (!abort_cached) {
 292                                 return -1;
 293                         }
 294                 }
 295         }
 296
 297         if (abort_cached && wpa_key_mgmt_wpa_ieee8021x(sm->key_mgmt) &&
 298             !wpa_key_mgmt_suite_b(sm->key_mgmt) &&
 299             !wpa_key_mgmt_ft(sm->key_mgmt) && sm->key_mgmt != WPA_KEY_MGMT_OSEN)
 300         {
 301                 /* Send EAPOL-Start to trigger full EAP authentication. */
 302                 u8 *buf;
 303                 size_t buflen;
 304
 305                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 306                         "RSN: no PMKSA entry found - trigger "
 307                         "full EAP authentication");
 308                 buf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_START,
 309                                          NULL, 0, &buflen, NULL);
 310                 if (buf) {
 311                         wpa_sm_ether_send(sm, sm->bssid, ETH_P_EAPOL,
 312                                           buf, buflen);
 313                         os_free(buf);
 314                         return -2;
 315                 }
 316
 317                 return -1;
 318         }
 319
 320         return 0;
 321 }
 322
 323
 324 /**
 325  * wpa_supplicant_send_2_of_4 - Send message 2 of WPA/RSN 4-Way Handshake
 326  * @sm: Pointer to WPA state machine data from wpa_sm_init()
 327  * @dst: Destination address for the frame
 328  * @key: Pointer to the EAPOL-Key frame header
 329  * @ver: Version bits from EAPOL-Key Key Info
 330  * @nonce: Nonce value for the EAPOL-Key frame
 331  * @wpa_ie: WPA/RSN IE
 332  * @wpa_ie_len: Length of the WPA/RSN IE
 333  * @ptk: PTK to use for keyed hash and encryption
 334  * Returns: >= 0 on success, < 0 on failure
 335  */
 336 int wpa_supplicant_send_2_of_4(struct wpa_sm *sm, const unsigned char *dst,
 337                                const struct wpa_eapol_key *key,
 338                                int ver, const u8 *nonce,
 339                                const u8 *wpa_ie, size_t wpa_ie_len,
 340                                struct wpa_ptk *ptk)
 341 {
 342         size_t mic_len, hdrlen, rlen;
 343         struct wpa_eapol_key *reply;
 344         struct wpa_eapol_key_192 *reply192;
 345         u8 *rbuf, *key_mic;
 346         u8 *rsn_ie_buf = NULL;
 347
 348         if (wpa_ie == NULL) {
 349                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING, "WPA: No wpa_ie set - "
 350                         "cannot generate msg 2/4");
 351                 return -1;
 352         }
 353
 354 #ifdef CONFIG_IEEE80211R
 355         if (wpa_key_mgmt_ft(sm->key_mgmt)) {
 356                 int res;
 357
 358                 /*
 359                  * Add PMKR1Name into RSN IE (PMKID-List) and add MDIE and
 360                  * FTIE from (Re)Association Response.
 361                  */
 362                 rsn_ie_buf = os_malloc(wpa_ie_len + 2 + 2 + PMKID_LEN +
 363                                        sm->assoc_resp_ies_len);
 364                 if (rsn_ie_buf == NULL)
 365                         return -1;
 366                 os_memcpy(rsn_ie_buf, wpa_ie, wpa_ie_len);
 367                 res = wpa_insert_pmkid(rsn_ie_buf, &wpa_ie_len,
 368                                        sm->pmk_r1_name);
 369                 if (res < 0) {
 370                         os_free(rsn_ie_buf);
 371                         return -1;
 372                 }
 373
 374                 if (sm->assoc_resp_ies) {
 375                         os_memcpy(rsn_ie_buf + wpa_ie_len, sm->assoc_resp_ies,
 376                                   sm->assoc_resp_ies_len);
 377                         wpa_ie_len += sm->assoc_resp_ies_len;
 378                 }
 379
 380                 wpa_ie = rsn_ie_buf;
 381         }
 382 #endif /* CONFIG_IEEE80211R */
 383
 384         wpa_hexdump(MSG_DEBUG, "WPA: WPA IE for msg 2/4", wpa_ie, wpa_ie_len);
 385
 386         mic_len = wpa_mic_len(sm->key_mgmt);
 387         hdrlen = mic_len == 24 ? sizeof(*reply192) : sizeof(*reply);
 388         rbuf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_KEY,
 389                                   NULL, hdrlen + wpa_ie_len,
 390                                   &rlen, (void *) &reply);
 391         if (rbuf == NULL) {
 392                 os_free(rsn_ie_buf);
 393                 return -1;
 394         }
 395         reply192 = (struct wpa_eapol_key_192 *) reply;
 396
 397         reply->type = (sm->proto == WPA_PROTO_RSN ||
 398                        sm->proto == WPA_PROTO_OSEN) ?
 399                 EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
 400         WPA_PUT_BE16(reply->key_info,
 401                      ver | WPA_KEY_INFO_KEY_TYPE | WPA_KEY_INFO_MIC);
 402         if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN)
 403                 WPA_PUT_BE16(reply->key_length, 0);
 404         else
 405                 os_memcpy(reply->key_length, key->key_length, 2);
 406         os_memcpy(reply->replay_counter, key->replay_counter,
 407                   WPA_REPLAY_COUNTER_LEN);
 408         wpa_hexdump(MSG_DEBUG, "WPA: Replay Counter", reply->replay_counter,
 409                     WPA_REPLAY_COUNTER_LEN);
 410
 411         key_mic = reply192->key_mic; /* same offset for reply and reply192 */
 412         if (mic_len == 24) {
 413                 WPA_PUT_BE16(reply192->key_data_length, wpa_ie_len);
 414                 os_memcpy(reply192 + 1, wpa_ie, wpa_ie_len);
 415         } else {
 416                 WPA_PUT_BE16(reply->key_data_length, wpa_ie_len);
 417                 os_memcpy(reply + 1, wpa_ie, wpa_ie_len);
 418         }
 419         os_free(rsn_ie_buf);
 420
 421         os_memcpy(reply->key_nonce, nonce, WPA_NONCE_LEN);
 422
 423         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Sending EAPOL-Key 2/4");
 424         return wpa_eapol_key_send(sm, ptk->kck, ptk->kck_len, ver, dst,
 425                                   ETH_P_EAPOL, rbuf, rlen, key_mic);
 426 }
 427
 428
 429 static int wpa_derive_ptk(struct wpa_sm *sm, const unsigned char *src_addr,
 430                           const struct wpa_eapol_key *key, struct wpa_ptk *ptk)
 431 {
 432 #ifdef CONFIG_IEEE80211R
 433         if (wpa_key_mgmt_ft(sm->key_mgmt))
 434                 return wpa_derive_ptk_ft(sm, src_addr, key, ptk);
 435 #endif /* CONFIG_IEEE80211R */
 436
 437         return wpa_pmk_to_ptk(sm->pmk, sm->pmk_len, "Pairwise key expansion",
 438                               sm->own_addr, sm->bssid, sm->snonce,
 439                               key->key_nonce, ptk, sm->key_mgmt,
 440                               sm->pairwise_cipher);
 441 }
 442
 443
 444 static void wpa_supplicant_process_1_of_4(struct wpa_sm *sm,
 445                                           const unsigned char *src_addr,
 446                                           const struct wpa_eapol_key *key,
 447                                           u16 ver, const u8 *key_data,
 448                                           size_t key_data_len)
 449 {
 450         struct wpa_eapol_ie_parse ie;
 451         struct wpa_ptk *ptk;
 452         int res;
 453         u8 *kde, *kde_buf = NULL;
 454         size_t kde_len;
 455
 456         if (wpa_sm_get_network_ctx(sm) == NULL) {
 457                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING, "WPA: No SSID info "
 458                         "found (msg 1 of 4)");
 459                 return;
 460         }
 461
 462         wpa_sm_set_state(sm, WPA_4WAY_HANDSHAKE);
 463         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: RX message 1 of 4-Way "
 464                 "Handshake from " MACSTR " (ver=%d)", MAC2STR(src_addr), ver);
 465
 466         os_memset(&ie, 0, sizeof(ie));
 467
 468         if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN) {
 469                 /* RSN: msg 1/4 should contain PMKID for the selected PMK */
 470                 wpa_hexdump(MSG_DEBUG, "RSN: msg 1/4 key data",
 471                             key_data, key_data_len);
 472                 if (wpa_supplicant_parse_ies(key_data, key_data_len, &ie) < 0)
 473                         goto failed;
 474                 if (ie.pmkid) {
 475                         wpa_hexdump(MSG_DEBUG, "RSN: PMKID from "
 476                                     "Authenticator", ie.pmkid, PMKID_LEN);
 477                 }
 478         }
 479
 480         res = wpa_supplicant_get_pmk(sm, src_addr, ie.pmkid);
 481         if (res == -2) {
 482                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "RSN: Do not reply to "
 483                         "msg 1/4 - requesting full EAP authentication");
 484                 return;
 485         }
 486         if (res)
 487                 goto failed;
 488
 489         if (sm->renew_snonce) {
 490                 if (random_get_bytes(sm->snonce, WPA_NONCE_LEN)) {
 491                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 492                                 "WPA: Failed to get random data for SNonce");
 493                         goto failed;
 494                 }
 495                 sm->renew_snonce = 0;
 496                 wpa_hexdump(MSG_DEBUG, "WPA: Renewed SNonce",
 497                             sm->snonce, WPA_NONCE_LEN);
 498         }
 499
 500         /* Calculate PTK which will be stored as a temporary PTK until it has
 501          * been verified when processing message 3/4. */
 502         ptk = &sm->tptk;
 503         wpa_derive_ptk(sm, src_addr, key, ptk);
 504         if (sm->pairwise_cipher == WPA_CIPHER_TKIP) {
 505                 u8 buf[8];
 506                 /* Supplicant: swap tx/rx Mic keys */
 507                 os_memcpy(buf, &ptk->tk[16], 8);
 508                 os_memcpy(&ptk->tk[16], &ptk->tk[24], 8);
 509                 os_memcpy(&ptk->tk[24], buf, 8);
 510                 os_memset(buf, 0, sizeof(buf));
 511         }
 512         sm->tptk_set = 1;
 513
 514         kde = sm->assoc_wpa_ie;
 515         kde_len = sm->assoc_wpa_ie_len;
 516
 517 #ifdef CONFIG_P2P
 518         if (sm->p2p) {
 519                 kde_buf = os_malloc(kde_len + 2 + RSN_SELECTOR_LEN + 1);
 520                 if (kde_buf) {
 521                         u8 *pos;
 522                         wpa_printf(MSG_DEBUG, "P2P: Add IP Address Request KDE "
 523                                    "into EAPOL-Key 2/4");
 524                         os_memcpy(kde_buf, kde, kde_len);
 525                         kde = kde_buf;
 526                         pos = kde + kde_len;
 527                         *pos++ = WLAN_EID_VENDOR_SPECIFIC;
 528                         *pos++ = RSN_SELECTOR_LEN + 1;
 529                         RSN_SELECTOR_PUT(pos, WFA_KEY_DATA_IP_ADDR_REQ);
 530                         pos += RSN_SELECTOR_LEN;
 531                         *pos++ = 0x01;
 532                         kde_len = pos - kde;
 533                 }
 534         }
 535 #endif /* CONFIG_P2P */
 536
 537         if (wpa_supplicant_send_2_of_4(sm, sm->bssid, key, ver, sm->snonce,
 538                                        kde, kde_len, ptk) < 0)
 539                 goto failed;
 540
 541         os_free(kde_buf);
 542         os_memcpy(sm->anonce, key->key_nonce, WPA_NONCE_LEN);
 543         return;
 544
 545 failed:
 546         os_free(kde_buf);
 547         wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
 548 }
 549
 550
 551 static void wpa_sm_start_preauth(void *eloop_ctx, void *timeout_ctx)
 552 {
 553         struct wpa_sm *sm = eloop_ctx;
 554         rsn_preauth_candidate_process(sm);
 555 }
 556
 557
 558 static void wpa_supplicant_key_neg_complete(struct wpa_sm *sm,
 559                                             const u8 *addr, int secure)
 560 {
 561         wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
 562                 "WPA: Key negotiation completed with "
 563                 MACSTR " [PTK=%s GTK=%s]", MAC2STR(addr),
 564                 wpa_cipher_txt(sm->pairwise_cipher),
 565                 wpa_cipher_txt(sm->group_cipher));
 566         wpa_sm_cancel_auth_timeout(sm);
 567         wpa_sm_set_state(sm, WPA_COMPLETED);
 568
 569         if (secure) {
 570                 wpa_sm_mlme_setprotection(
 571                         sm, addr, MLME_SETPROTECTION_PROTECT_TYPE_RX_TX,
 572                         MLME_SETPROTECTION_KEY_TYPE_PAIRWISE);
 573                 eapol_sm_notify_portValid(sm->eapol, TRUE);
 574                 if (wpa_key_mgmt_wpa_psk(sm->key_mgmt))
 575                         eapol_sm_notify_eap_success(sm->eapol, TRUE);
 576                 /*
 577                  * Start preauthentication after a short wait to avoid a
 578                  * possible race condition between the data receive and key
 579                  * configuration after the 4-Way Handshake. This increases the
 580                  * likelihood of the first preauth EAPOL-Start frame getting to
 581                  * the target AP.
 582                  */
 583                 eloop_register_timeout(1, 0, wpa_sm_start_preauth, sm, NULL);
 584         }
 585
 586         if (sm->cur_pmksa && sm->cur_pmksa->opportunistic) {
 587                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 588                         "RSN: Authenticator accepted "
 589                         "opportunistic PMKSA entry - marking it valid");
 590                 sm->cur_pmksa->opportunistic = 0;
 591         }
 592
 593 #ifdef CONFIG_IEEE80211R
 594         if (wpa_key_mgmt_ft(sm->key_mgmt)) {
 595                 /* Prepare for the next transition */
 596                 wpa_ft_prepare_auth_request(sm, NULL);
 597         }
 598 #endif /* CONFIG_IEEE80211R */
 599 }
 600
 601
 602 static void wpa_sm_rekey_ptk(void *eloop_ctx, void *timeout_ctx)
 603 {
 604         struct wpa_sm *sm = eloop_ctx;
 605         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Request PTK rekeying");
 606         wpa_sm_key_request(sm, 0, 1);
 607 }
 608
 609
 610 static int wpa_supplicant_install_ptk(struct wpa_sm *sm,
 611                                       const struct wpa_eapol_key *key)
 612 {
 613         int keylen, rsclen;
 614         enum wpa_alg alg;
 615         const u8 *key_rsc;
 616
 617         if (sm->ptk.installed) {
 618                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 619                         "WPA: Do not re-install same PTK to the driver");
 620                 return 0;
 621         }
 622
 623         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 624                 "WPA: Installing PTK to the driver");
 625
 626         if (sm->pairwise_cipher == WPA_CIPHER_NONE) {
 627                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Pairwise Cipher "
 628                         "Suite: NONE - do not use pairwise keys");
 629                 return 0;
 630         }
 631
 632         if (!wpa_cipher_valid_pairwise(sm->pairwise_cipher)) {
 633                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 634                         "WPA: Unsupported pairwise cipher %d",
 635                         sm->pairwise_cipher);
 636                 return -1;
 637         }
 638
 639         alg = wpa_cipher_to_alg(sm->pairwise_cipher);
 640         keylen = wpa_cipher_key_len(sm->pairwise_cipher);
 641         rsclen = wpa_cipher_rsc_len(sm->pairwise_cipher);
 642
 643         if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN) {
 644                 key_rsc = null_rsc;
 645         } else {
 646                 key_rsc = key->key_rsc;
 647                 wpa_hexdump(MSG_DEBUG, "WPA: RSC", key_rsc, rsclen);
 648         }
 649
 650         if (wpa_sm_set_key(sm, alg, sm->bssid, 0, 1, key_rsc, rsclen,
 651                            sm->ptk.tk, keylen) < 0) {
 652                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 653                         "WPA: Failed to set PTK to the "
 654                         "driver (alg=%d keylen=%d bssid=" MACSTR ")",
 655                         alg, keylen, MAC2STR(sm->bssid));
 656                 return -1;
 657         }
 658
 659         /* TK is not needed anymore in supplicant */
 660         os_memset(sm->ptk.tk, 0, WPA_TK_MAX_LEN);
 661         sm->ptk.installed = 1;
 662
 663         if (sm->wpa_ptk_rekey) {
 664                 eloop_cancel_timeout(wpa_sm_rekey_ptk, sm, NULL);
 665                 eloop_register_timeout(sm->wpa_ptk_rekey, 0, wpa_sm_rekey_ptk,
 666                                        sm, NULL);
 667         }
 668
 669         return 0;
 670 }
 671
 672
 673 static int wpa_supplicant_check_group_cipher(struct wpa_sm *sm,
 674                                              int group_cipher,
 675                                              int keylen, int maxkeylen,
 676                                              int *key_rsc_len,
 677                                              enum wpa_alg *alg)
 678 {
 679         int klen;
 680
 681         *alg = wpa_cipher_to_alg(group_cipher);
 682         if (*alg == WPA_ALG_NONE) {
 683                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 684                         "WPA: Unsupported Group Cipher %d",
 685                         group_cipher);
 686                 return -1;
 687         }
 688         *key_rsc_len = wpa_cipher_rsc_len(group_cipher);
 689
 690         klen = wpa_cipher_key_len(group_cipher);
 691         if (keylen != klen || maxkeylen < klen) {
 692                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 693                         "WPA: Unsupported %s Group Cipher key length %d (%d)",
 694                         wpa_cipher_txt(group_cipher), keylen, maxkeylen);
 695                 return -1;
 696         }
 697         return 0;
 698 }
 699
 700
 701 struct wpa_gtk_data {
 702         enum wpa_alg alg;
 703         int tx, key_rsc_len, keyidx;
 704         u8 gtk[32];
 705         int gtk_len;
 706 };
 707
 708
 709 static int wpa_supplicant_install_gtk(struct wpa_sm *sm,
 710                                       const struct wpa_gtk_data *gd,
 711                                       const u8 *key_rsc, int wnm_sleep)
 712 {
 713         const u8 *_gtk = gd->gtk;
 714         u8 gtk_buf[32];
 715
 716         /* Detect possible key reinstallation */
 717         if ((sm->gtk.gtk_len == (size_t) gd->gtk_len &&
 718              os_memcmp(sm->gtk.gtk, gd->gtk, sm->gtk.gtk_len) == 0) ||
 719             (sm->gtk_wnm_sleep.gtk_len == (size_t) gd->gtk_len &&
 720              os_memcmp(sm->gtk_wnm_sleep.gtk, gd->gtk,
 721                        sm->gtk_wnm_sleep.gtk_len) == 0)) {
 722                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 723                         "WPA: Not reinstalling already in-use GTK to the driver (keyidx=%d tx=%d len=%d)",
 724                         gd->keyidx, gd->tx, gd->gtk_len);
 725                 return 0;
 726         }
 727
 728         wpa_hexdump_key(MSG_DEBUG, "WPA: Group Key", gd->gtk, gd->gtk_len);
 729         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 730                 "WPA: Installing GTK to the driver (keyidx=%d tx=%d len=%d)",
 731                 gd->keyidx, gd->tx, gd->gtk_len);
 732         wpa_hexdump(MSG_DEBUG, "WPA: RSC", key_rsc, gd->key_rsc_len);
 733         if (sm->group_cipher == WPA_CIPHER_TKIP) {
 734                 /* Swap Tx/Rx keys for Michael MIC */
 735                 os_memcpy(gtk_buf, gd->gtk, 16);
 736                 os_memcpy(gtk_buf + 16, gd->gtk + 24, 8);
 737                 os_memcpy(gtk_buf + 24, gd->gtk + 16, 8);
 738                 _gtk = gtk_buf;
 739         }
 740         if (sm->pairwise_cipher == WPA_CIPHER_NONE) {
 741                 if (wpa_sm_set_key(sm, gd->alg, NULL,
 742                                    gd->keyidx, 1, key_rsc, gd->key_rsc_len,
 743                                    _gtk, gd->gtk_len) < 0) {
 744                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 745                                 "WPA: Failed to set GTK to the driver "
 746                                 "(Group only)");
 747                         os_memset(gtk_buf, 0, sizeof(gtk_buf));
 748                         return -1;
 749                 }
 750         } else if (wpa_sm_set_key(sm, gd->alg, broadcast_ether_addr,
 751                                   gd->keyidx, gd->tx, key_rsc, gd->key_rsc_len,
 752                                   _gtk, gd->gtk_len) < 0) {
 753                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 754                         "WPA: Failed to set GTK to "
 755                         "the driver (alg=%d keylen=%d keyidx=%d)",
 756                         gd->alg, gd->gtk_len, gd->keyidx);
 757                 os_memset(gtk_buf, 0, sizeof(gtk_buf));
 758                 return -1;
 759         }
 760         os_memset(gtk_buf, 0, sizeof(gtk_buf));
 761
 762         if (wnm_sleep) {
 763                 sm->gtk_wnm_sleep.gtk_len = gd->gtk_len;
 764                 os_memcpy(sm->gtk_wnm_sleep.gtk, gd->gtk,
 765                           sm->gtk_wnm_sleep.gtk_len);
 766         } else {
 767                 sm->gtk.gtk_len = gd->gtk_len;
 768                 os_memcpy(sm->gtk.gtk, gd->gtk, sm->gtk.gtk_len);
 769         }
 770
 771         return 0;
 772 }
 773
 774
 775 static int wpa_supplicant_gtk_tx_bit_workaround(const struct wpa_sm *sm,
 776                                                 int tx)
 777 {
 778         if (tx && sm->pairwise_cipher != WPA_CIPHER_NONE) {
 779                 /* Ignore Tx bit for GTK if a pairwise key is used. One AP
 780                  * seemed to set this bit (incorrectly, since Tx is only when
 781                  * doing Group Key only APs) and without this workaround, the
 782                  * data connection does not work because wpa_supplicant
 783                  * configured non-zero keyidx to be used for unicast. */
 784                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
 785                         "WPA: Tx bit set for GTK, but pairwise "
 786                         "keys are used - ignore Tx bit");
 787                 return 0;
 788         }
 789         return tx;
 790 }
 791
 792
 793 static int wpa_supplicant_rsc_relaxation(const struct wpa_sm *sm,
 794                                          const u8 *rsc)
 795 {
 796         int rsclen;
 797
 798         if (!sm->wpa_rsc_relaxation)
 799                 return 0;
 800
 801         rsclen = wpa_cipher_rsc_len(sm->group_cipher);
 802
 803         /*
 804          * Try to detect RSC (endian) corruption issue where the AP sends
 805          * the RSC bytes in EAPOL-Key message in the wrong order, both if
 806          * it's actually a 6-byte field (as it should be) and if it treats
 807          * it as an 8-byte field.
 808          * An AP model known to have this bug is the Sapido RB-1632.
 809          */
 810         if (rsclen == 6 && ((rsc[5] && !rsc[0]) || rsc[6] || rsc[7])) {
 811                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 812                         "RSC %02x%02x%02x%02x%02x%02x%02x%02x is likely bogus, using 0",
 813                         rsc[0], rsc[1], rsc[2], rsc[3],
 814                         rsc[4], rsc[5], rsc[6], rsc[7]);
 815
 816                 return 1;
 817         }
 818
 819         return 0;
 820 }
 821
 822
 823 static int wpa_supplicant_pairwise_gtk(struct wpa_sm *sm,
 824                                        const struct wpa_eapol_key *key,
 825                                        const u8 *gtk, size_t gtk_len,
 826                                        int key_info)
 827 {
 828         struct wpa_gtk_data gd;
 829         const u8 *key_rsc;
 830
 831         /*
 832          * IEEE Std 802.11i-2004 - 8.5.2 EAPOL-Key frames - Figure 43x
 833          * GTK KDE format:
 834          * KeyID[bits 0-1], Tx [bit 2], Reserved [bits 3-7]
 835          * Reserved [bits 0-7]
 836          * GTK
 837          */
 838
 839         os_memset(&gd, 0, sizeof(gd));
 840         wpa_hexdump_key(MSG_DEBUG, "RSN: received GTK in pairwise handshake",
 841                         gtk, gtk_len);
 842
 843         if (gtk_len < 2 || gtk_len - 2 > sizeof(gd.gtk))
 844                 return -1;
 845
 846         gd.keyidx = gtk[0] & 0x3;
 847         gd.tx = wpa_supplicant_gtk_tx_bit_workaround(sm,
 848                                                      !!(gtk[0] & BIT(2)));
 849         gtk += 2;
 850         gtk_len -= 2;
 851
 852         os_memcpy(gd.gtk, gtk, gtk_len);
 853         gd.gtk_len = gtk_len;
 854
 855         key_rsc = key->key_rsc;
 856         if (wpa_supplicant_rsc_relaxation(sm, key->key_rsc))
 857                 key_rsc = null_rsc;
 858
 859         if (sm->group_cipher != WPA_CIPHER_GTK_NOT_USED &&
 860             (wpa_supplicant_check_group_cipher(sm, sm->group_cipher,
 861                                                gtk_len, gtk_len,
 862                                                &gd.key_rsc_len, &gd.alg) ||
 863              wpa_supplicant_install_gtk(sm, &gd, key_rsc, 0))) {
 864                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 865                         "RSN: Failed to install GTK");
 866                 os_memset(&gd, 0, sizeof(gd));
 867                 return -1;
 868         }
 869         os_memset(&gd, 0, sizeof(gd));
 870
 871         wpa_supplicant_key_neg_complete(sm, sm->bssid,
 872                                         key_info & WPA_KEY_INFO_SECURE);
 873         return 0;
 874 }
 875
 876
 877 #ifdef CONFIG_IEEE80211W
 878 static int wpa_supplicant_install_igtk(struct wpa_sm *sm,
 879                                        const struct wpa_igtk_kde *igtk,
 880                                        int wnm_sleep)
 881 {
 882         size_t len = wpa_cipher_key_len(sm->mgmt_group_cipher);
 883         u16 keyidx = WPA_GET_LE16(igtk->keyid);
 884
 885         /* Detect possible key reinstallation */
 886         if ((sm->igtk.igtk_len == len &&
 887              os_memcmp(sm->igtk.igtk, igtk->igtk, sm->igtk.igtk_len) == 0) ||
 888             (sm->igtk_wnm_sleep.igtk_len == len &&
 889              os_memcmp(sm->igtk_wnm_sleep.igtk, igtk->igtk,
 890                        sm->igtk_wnm_sleep.igtk_len) == 0)) {
 891                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 892                         "WPA: Not reinstalling already in-use IGTK to the driver (keyidx=%d)",
 893                         keyidx);
 894                 return  0;
 895         }
 896
 897         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
 898                 "WPA: IGTK keyid %d pn %02x%02x%02x%02x%02x%02x",
 899                 keyidx, MAC2STR(igtk->pn));
 900         wpa_hexdump_key(MSG_DEBUG, "WPA: IGTK", igtk->igtk, len);
 901         if (keyidx > 4095) {
 902                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 903                         "WPA: Invalid IGTK KeyID %d", keyidx);
 904                 return -1;
 905         }
 906         if (wpa_sm_set_key(sm, wpa_cipher_to_alg(sm->mgmt_group_cipher),
 907                            broadcast_ether_addr,
 908                            keyidx, 0, igtk->pn, sizeof(igtk->pn),
 909                            igtk->igtk, len) < 0) {
 910                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
 911                         "WPA: Failed to configure IGTK to the driver");
 912                 return -1;
 913         }
 914
 915         if (wnm_sleep) {
 916                 sm->igtk_wnm_sleep.igtk_len = len;
 917                 os_memcpy(sm->igtk_wnm_sleep.igtk, igtk->igtk,
 918                           sm->igtk_wnm_sleep.igtk_len);
 919         } else {
 920                 sm->igtk.igtk_len = len;
 921                 os_memcpy(sm->igtk.igtk, igtk->igtk, sm->igtk.igtk_len);
 922         }
 923
 924         return 0;
 925 }
 926 #endif /* CONFIG_IEEE80211W */
 927
 928
 929 static int ieee80211w_set_keys(struct wpa_sm *sm,
 930                                struct wpa_eapol_ie_parse *ie)
 931 {
 932 #ifdef CONFIG_IEEE80211W
 933         if (!wpa_cipher_valid_mgmt_group(sm->mgmt_group_cipher))
 934                 return 0;
 935
 936         if (ie->igtk) {
 937                 size_t len;
 938                 const struct wpa_igtk_kde *igtk;
 939
 940                 len = wpa_cipher_key_len(sm->mgmt_group_cipher);
 941                 if (ie->igtk_len != WPA_IGTK_KDE_PREFIX_LEN + len)
 942                         return -1;
 943
 944                 igtk = (const struct wpa_igtk_kde *) ie->igtk;
 945                 if (wpa_supplicant_install_igtk(sm, igtk, 0) < 0)
 946                         return -1;
 947         }
 948
 949         return 0;
 950 #else /* CONFIG_IEEE80211W */
 951         return 0;
 952 #endif /* CONFIG_IEEE80211W */
 953 }
 954
 955
 956 static void wpa_report_ie_mismatch(struct wpa_sm *sm,
 957                                    const char *reason, const u8 *src_addr,
 958                                    const u8 *wpa_ie, size_t wpa_ie_len,
 959                                    const u8 *rsn_ie, size_t rsn_ie_len)
 960 {
 961         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING, "WPA: %s (src=" MACSTR ")",
 962                 reason, MAC2STR(src_addr));
 963
 964         if (sm->ap_wpa_ie) {
 965                 wpa_hexdump(MSG_INFO, "WPA: WPA IE in Beacon/ProbeResp",
 966                             sm->ap_wpa_ie, sm->ap_wpa_ie_len);
 967         }
 968         if (wpa_ie) {
 969                 if (!sm->ap_wpa_ie) {
 970                         wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
 971                                 "WPA: No WPA IE in Beacon/ProbeResp");
 972                 }
 973                 wpa_hexdump(MSG_INFO, "WPA: WPA IE in 3/4 msg",
 974                             wpa_ie, wpa_ie_len);
 975         }
 976
 977         if (sm->ap_rsn_ie) {
 978                 wpa_hexdump(MSG_INFO, "WPA: RSN IE in Beacon/ProbeResp",
 979                             sm->ap_rsn_ie, sm->ap_rsn_ie_len);
 980         }
 981         if (rsn_ie) {
 982                 if (!sm->ap_rsn_ie) {
 983                         wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
 984                                 "WPA: No RSN IE in Beacon/ProbeResp");
 985                 }
 986                 wpa_hexdump(MSG_INFO, "WPA: RSN IE in 3/4 msg",
 987                             rsn_ie, rsn_ie_len);
 988         }
 989
 990         wpa_sm_deauthenticate(sm, WLAN_REASON_IE_IN_4WAY_DIFFERS);
 991 }
 992
 993
 994 #ifdef CONFIG_IEEE80211R
 995
 996 static int ft_validate_mdie(struct wpa_sm *sm,
 997                             const unsigned char *src_addr,
 998                             struct wpa_eapol_ie_parse *ie,
 999                             const u8 *assoc_resp_mdie)
1000 {
1001         struct rsn_mdie *mdie;
1002
1003         mdie = (struct rsn_mdie *) (ie->mdie + 2);
1004         if (ie->mdie == NULL || ie->mdie_len < 2 + sizeof(*mdie) ||
1005             os_memcmp(mdie->mobility_domain, sm->mobility_domain,
1006                       MOBILITY_DOMAIN_ID_LEN) != 0) {
1007                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "FT: MDIE in msg 3/4 did "
1008                         "not match with the current mobility domain");
1009                 return -1;
1010         }
1011
1012         if (assoc_resp_mdie &&
1013             (assoc_resp_mdie[1] != ie->mdie[1] ||
1014              os_memcmp(assoc_resp_mdie, ie->mdie, 2 + ie->mdie[1]) != 0)) {
1015                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "FT: MDIE mismatch");
1016                 wpa_hexdump(MSG_DEBUG, "FT: MDIE in EAPOL-Key msg 3/4",
1017                             ie->mdie, 2 + ie->mdie[1]);
1018                 wpa_hexdump(MSG_DEBUG, "FT: MDIE in (Re)Association Response",
1019                             assoc_resp_mdie, 2 + assoc_resp_mdie[1]);
1020                 return -1;
1021         }
1022
1023         return 0;
1024 }
1025
1026
1027 static int ft_validate_ftie(struct wpa_sm *sm,
1028                             const unsigned char *src_addr,
1029                             struct wpa_eapol_ie_parse *ie,
1030                             const u8 *assoc_resp_ftie)
1031 {
1032         if (ie->ftie == NULL) {
1033                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1034                         "FT: No FTIE in EAPOL-Key msg 3/4");
1035                 return -1;
1036         }
1037
1038         if (assoc_resp_ftie == NULL)
1039                 return 0;
1040
1041         if (assoc_resp_ftie[1] != ie->ftie[1] ||
1042             os_memcmp(assoc_resp_ftie, ie->ftie, 2 + ie->ftie[1]) != 0) {
1043                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "FT: FTIE mismatch");
1044                 wpa_hexdump(MSG_DEBUG, "FT: FTIE in EAPOL-Key msg 3/4",
1045                             ie->ftie, 2 + ie->ftie[1]);
1046                 wpa_hexdump(MSG_DEBUG, "FT: FTIE in (Re)Association Response",
1047                             assoc_resp_ftie, 2 + assoc_resp_ftie[1]);
1048                 return -1;
1049         }
1050
1051         return 0;
1052 }
1053
1054
1055 static int ft_validate_rsnie(struct wpa_sm *sm,
1056                              const unsigned char *src_addr,
1057                              struct wpa_eapol_ie_parse *ie)
1058 {
1059         struct wpa_ie_data rsn;
1060
1061         if (!ie->rsn_ie)
1062                 return 0;
1063
1064         /*
1065          * Verify that PMKR1Name from EAPOL-Key message 3/4
1066          * matches with the value we derived.
1067          */
1068         if (wpa_parse_wpa_ie_rsn(ie->rsn_ie, ie->rsn_ie_len, &rsn) < 0 ||
1069             rsn.num_pmkid != 1 || rsn.pmkid == NULL) {
1070                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "FT: No PMKR1Name in "
1071                         "FT 4-way handshake message 3/4");
1072                 return -1;
1073         }
1074
1075         if (os_memcmp_const(rsn.pmkid, sm->pmk_r1_name, WPA_PMK_NAME_LEN) != 0)
1076         {
1077                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1078                         "FT: PMKR1Name mismatch in "
1079                         "FT 4-way handshake message 3/4");
1080                 wpa_hexdump(MSG_DEBUG, "FT: PMKR1Name from Authenticator",
1081                             rsn.pmkid, WPA_PMK_NAME_LEN);
1082                 wpa_hexdump(MSG_DEBUG, "FT: Derived PMKR1Name",
1083                             sm->pmk_r1_name, WPA_PMK_NAME_LEN);
1084                 return -1;
1085         }
1086
1087         return 0;
1088 }
1089
1090
1091 static int wpa_supplicant_validate_ie_ft(struct wpa_sm *sm,
1092                                          const unsigned char *src_addr,
1093                                          struct wpa_eapol_ie_parse *ie)
1094 {
1095         const u8 *pos, *end, *mdie = NULL, *ftie = NULL;
1096
1097         if (sm->assoc_resp_ies) {
1098                 pos = sm->assoc_resp_ies;
1099                 end = pos + sm->assoc_resp_ies_len;
1100                 while (end - pos > 2) {
1101                         if (2 + pos[1] > end - pos)
1102                                 break;
1103                         switch (*pos) {
1104                         case WLAN_EID_MOBILITY_DOMAIN:
1105                                 mdie = pos;
1106                                 break;
1107                         case WLAN_EID_FAST_BSS_TRANSITION:
1108                                 ftie = pos;
1109                                 break;
1110                         }
1111                         pos += 2 + pos[1];
1112                 }
1113         }
1114
1115         if (ft_validate_mdie(sm, src_addr, ie, mdie) < 0 ||
1116             ft_validate_ftie(sm, src_addr, ie, ftie) < 0 ||
1117             ft_validate_rsnie(sm, src_addr, ie) < 0)
1118                 return -1;
1119
1120         return 0;
1121 }
1122
1123 #endif /* CONFIG_IEEE80211R */
1124
1125
1126 static int wpa_supplicant_validate_ie(struct wpa_sm *sm,
1127                                       const unsigned char *src_addr,
1128                                       struct wpa_eapol_ie_parse *ie)
1129 {
1130         if (sm->ap_wpa_ie == NULL && sm->ap_rsn_ie == NULL) {
1131                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1132                         "WPA: No WPA/RSN IE for this AP known. "
1133                         "Trying to get from scan results");
1134                 if (wpa_sm_get_beacon_ie(sm) < 0) {
1135                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1136                                 "WPA: Could not find AP from "
1137                                 "the scan results");
1138                 } else {
1139                         wpa_msg(sm->ctx->msg_ctx, MSG_DEBUG,
1140                                 "WPA: Found the current AP from "
1141                                 "updated scan results");
1142                 }
1143         }
1144
1145         if (ie->wpa_ie == NULL && ie->rsn_ie == NULL &&
1146             (sm->ap_wpa_ie || sm->ap_rsn_ie)) {
1147                 wpa_report_ie_mismatch(sm, "IE in 3/4 msg does not match "
1148                                        "with IE in Beacon/ProbeResp (no IE?)",
1149                                        src_addr, ie->wpa_ie, ie->wpa_ie_len,
1150                                        ie->rsn_ie, ie->rsn_ie_len);
1151                 return -1;
1152         }
1153
1154         if ((ie->wpa_ie && sm->ap_wpa_ie &&
1155              (ie->wpa_ie_len != sm->ap_wpa_ie_len ||
1156               os_memcmp(ie->wpa_ie, sm->ap_wpa_ie, ie->wpa_ie_len) != 0)) ||
1157             (ie->rsn_ie && sm->ap_rsn_ie &&
1158              wpa_compare_rsn_ie(wpa_key_mgmt_ft(sm->key_mgmt),
1159                                 sm->ap_rsn_ie, sm->ap_rsn_ie_len,
1160                                 ie->rsn_ie, ie->rsn_ie_len))) {
1161                 wpa_report_ie_mismatch(sm, "IE in 3/4 msg does not match "
1162                                        "with IE in Beacon/ProbeResp",
1163                                        src_addr, ie->wpa_ie, ie->wpa_ie_len,
1164                                        ie->rsn_ie, ie->rsn_ie_len);
1165                 return -1;
1166         }
1167
1168         if (sm->proto == WPA_PROTO_WPA &&
1169             ie->rsn_ie && sm->ap_rsn_ie == NULL && sm->rsn_enabled) {
1170                 wpa_report_ie_mismatch(sm, "Possible downgrade attack "
1171                                        "detected - RSN was enabled and RSN IE "
1172                                        "was in msg 3/4, but not in "
1173                                        "Beacon/ProbeResp",
1174                                        src_addr, ie->wpa_ie, ie->wpa_ie_len,
1175                                        ie->rsn_ie, ie->rsn_ie_len);
1176                 return -1;
1177         }
1178
1179 #ifdef CONFIG_IEEE80211R
1180         if (wpa_key_mgmt_ft(sm->key_mgmt) &&
1181             wpa_supplicant_validate_ie_ft(sm, src_addr, ie) < 0)
1182                 return -1;
1183 #endif /* CONFIG_IEEE80211R */
1184
1185         return 0;
1186 }
1187
1188
1189 /**
1190  * wpa_supplicant_send_4_of_4 - Send message 4 of WPA/RSN 4-Way Handshake
1191  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1192  * @dst: Destination address for the frame
1193  * @key: Pointer to the EAPOL-Key frame header
1194  * @ver: Version bits from EAPOL-Key Key Info
1195  * @key_info: Key Info
1196  * @ptk: PTK to use for keyed hash and encryption
1197  * Returns: >= 0 on success, < 0 on failure
1198  */
1199 int wpa_supplicant_send_4_of_4(struct wpa_sm *sm, const unsigned char *dst,
1200                                const struct wpa_eapol_key *key,
1201                                u16 ver, u16 key_info,
1202                                struct wpa_ptk *ptk)
1203 {
1204         size_t mic_len, hdrlen, rlen;
1205         struct wpa_eapol_key *reply;
1206         struct wpa_eapol_key_192 *reply192;
1207         u8 *rbuf, *key_mic;
1208
1209         mic_len = wpa_mic_len(sm->key_mgmt);
1210         hdrlen = mic_len == 24 ? sizeof(*reply192) : sizeof(*reply);
1211         rbuf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_KEY, NULL,
1212                                   hdrlen, &rlen, (void *) &reply);
1213         if (rbuf == NULL)
1214                 return -1;
1215         reply192 = (struct wpa_eapol_key_192 *) reply;
1216
1217         reply->type = (sm->proto == WPA_PROTO_RSN ||
1218                        sm->proto == WPA_PROTO_OSEN) ?
1219                 EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
1220         key_info &= WPA_KEY_INFO_SECURE;
1221         key_info |= ver | WPA_KEY_INFO_KEY_TYPE | WPA_KEY_INFO_MIC;
1222         WPA_PUT_BE16(reply->key_info, key_info);
1223         if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN)
1224                 WPA_PUT_BE16(reply->key_length, 0);
1225         else
1226                 os_memcpy(reply->key_length, key->key_length, 2);
1227         os_memcpy(reply->replay_counter, key->replay_counter,
1228                   WPA_REPLAY_COUNTER_LEN);
1229
1230         key_mic = reply192->key_mic; /* same offset for reply and reply192 */
1231         if (mic_len == 24)
1232                 WPA_PUT_BE16(reply192->key_data_length, 0);
1233         else
1234                 WPA_PUT_BE16(reply->key_data_length, 0);
1235
1236         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Sending EAPOL-Key 4/4");
1237         return wpa_eapol_key_send(sm, ptk->kck, ptk->kck_len, ver, dst,
1238                                   ETH_P_EAPOL, rbuf, rlen, key_mic);
1239 }
1240
1241
1242 static void wpa_supplicant_process_3_of_4(struct wpa_sm *sm,
1243                                           const struct wpa_eapol_key *key,
1244                                           u16 ver, const u8 *key_data,
1245                                           size_t key_data_len)
1246 {
1247         u16 key_info, keylen;
1248         struct wpa_eapol_ie_parse ie;
1249
1250         wpa_sm_set_state(sm, WPA_4WAY_HANDSHAKE);
1251         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: RX message 3 of 4-Way "
1252                 "Handshake from " MACSTR " (ver=%d)", MAC2STR(sm->bssid), ver);
1253
1254         key_info = WPA_GET_BE16(key->key_info);
1255
1256         wpa_hexdump(MSG_DEBUG, "WPA: IE KeyData", key_data, key_data_len);
1257         if (wpa_supplicant_parse_ies(key_data, key_data_len, &ie) < 0)
1258                 goto failed;
1259         if (ie.gtk && !(key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
1260                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1261                         "WPA: GTK IE in unencrypted key data");
1262                 goto failed;
1263         }
1264 #ifdef CONFIG_IEEE80211W
1265         if (ie.igtk && !(key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
1266                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1267                         "WPA: IGTK KDE in unencrypted key data");
1268                 goto failed;
1269         }
1270
1271         if (ie.igtk &&
1272             wpa_cipher_valid_mgmt_group(sm->mgmt_group_cipher) &&
1273             ie.igtk_len != WPA_IGTK_KDE_PREFIX_LEN +
1274             (unsigned int) wpa_cipher_key_len(sm->mgmt_group_cipher)) {
1275                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1276                         "WPA: Invalid IGTK KDE length %lu",
1277                         (unsigned long) ie.igtk_len);
1278                 goto failed;
1279         }
1280 #endif /* CONFIG_IEEE80211W */
1281
1282         if (wpa_supplicant_validate_ie(sm, sm->bssid, &ie) < 0)
1283                 goto failed;
1284
1285         if (os_memcmp(sm->anonce, key->key_nonce, WPA_NONCE_LEN) != 0) {
1286                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1287                         "WPA: ANonce from message 1 of 4-Way Handshake "
1288                         "differs from 3 of 4-Way Handshake - drop packet (src="
1289                         MACSTR ")", MAC2STR(sm->bssid));
1290                 goto failed;
1291         }
1292
1293         keylen = WPA_GET_BE16(key->key_length);
1294         if (keylen != wpa_cipher_key_len(sm->pairwise_cipher)) {
1295                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1296                         "WPA: Invalid %s key length %d (src=" MACSTR
1297                         ")", wpa_cipher_txt(sm->pairwise_cipher), keylen,
1298                         MAC2STR(sm->bssid));
1299                 goto failed;
1300         }
1301
1302 #ifdef CONFIG_P2P
1303         if (ie.ip_addr_alloc) {
1304                 os_memcpy(sm->p2p_ip_addr, ie.ip_addr_alloc, 3 * 4);
1305                 wpa_hexdump(MSG_DEBUG, "P2P: IP address info",
1306                             sm->p2p_ip_addr, sizeof(sm->p2p_ip_addr));
1307         }
1308 #endif /* CONFIG_P2P */
1309
1310         if (wpa_supplicant_send_4_of_4(sm, sm->bssid, key, ver, key_info,
1311                                        &sm->ptk) < 0) {
1312                 goto failed;
1313         }
1314
1315         /* SNonce was successfully used in msg 3/4, so mark it to be renewed
1316          * for the next 4-Way Handshake. If msg 3 is received again, the old
1317          * SNonce will still be used to avoid changing PTK. */
1318         sm->renew_snonce = 1;
1319
1320         if (key_info & WPA_KEY_INFO_INSTALL) {
1321                 if (wpa_supplicant_install_ptk(sm, key))
1322                         goto failed;
1323         }
1324
1325         if (key_info & WPA_KEY_INFO_SECURE) {
1326                 wpa_sm_mlme_setprotection(
1327                         sm, sm->bssid, MLME_SETPROTECTION_PROTECT_TYPE_RX,
1328                         MLME_SETPROTECTION_KEY_TYPE_PAIRWISE);
1329                 eapol_sm_notify_portValid(sm->eapol, TRUE);
1330         }
1331         wpa_sm_set_state(sm, WPA_GROUP_HANDSHAKE);
1332
1333         if (sm->group_cipher == WPA_CIPHER_GTK_NOT_USED) {
1334                 wpa_supplicant_key_neg_complete(sm, sm->bssid,
1335                                                 key_info & WPA_KEY_INFO_SECURE);
1336         } else if (ie.gtk &&
1337             wpa_supplicant_pairwise_gtk(sm, key,
1338                                         ie.gtk, ie.gtk_len, key_info) < 0) {
1339                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1340                         "RSN: Failed to configure GTK");
1341                 goto failed;
1342         }
1343
1344         if (ieee80211w_set_keys(sm, &ie) < 0) {
1345                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1346                         "RSN: Failed to configure IGTK");
1347                 goto failed;
1348         }
1349
1350         if (ie.gtk)
1351                 wpa_sm_set_rekey_offload(sm);
1352
1353         if (sm->proto == WPA_PROTO_RSN && wpa_key_mgmt_suite_b(sm->key_mgmt)) {
1354                 struct rsn_pmksa_cache_entry *sa;
1355
1356                 sa = pmksa_cache_add(sm->pmksa, sm->pmk, sm->pmk_len, NULL,
1357                                      sm->ptk.kck, sm->ptk.kck_len,
1358                                      sm->bssid, sm->own_addr,
1359                                      sm->network_ctx, sm->key_mgmt);
1360                 if (!sm->cur_pmksa)
1361                         sm->cur_pmksa = sa;
1362         }
1363
1364         sm->msg_3_of_4_ok = 1;
1365         return;
1366
1367 failed:
1368         wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
1369 }
1370
1371
1372 static int wpa_supplicant_process_1_of_2_rsn(struct wpa_sm *sm,
1373                                              const u8 *keydata,
1374                                              size_t keydatalen,
1375                                              u16 key_info,
1376                                              struct wpa_gtk_data *gd)
1377 {
1378         int maxkeylen;
1379         struct wpa_eapol_ie_parse ie;
1380
1381         wpa_hexdump(MSG_DEBUG, "RSN: msg 1/2 key data", keydata, keydatalen);
1382         if (wpa_supplicant_parse_ies(keydata, keydatalen, &ie) < 0)
1383                 return -1;
1384         if (ie.gtk && !(key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
1385                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1386                         "WPA: GTK IE in unencrypted key data");
1387                 return -1;
1388         }
1389         if (ie.gtk == NULL) {
1390                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1391                         "WPA: No GTK IE in Group Key msg 1/2");
1392                 return -1;
1393         }
1394         maxkeylen = gd->gtk_len = ie.gtk_len - 2;
1395
1396         if (wpa_supplicant_check_group_cipher(sm, sm->group_cipher,
1397                                               gd->gtk_len, maxkeylen,
1398                                               &gd->key_rsc_len, &gd->alg))
1399                 return -1;
1400
1401         wpa_hexdump_key(MSG_DEBUG, "RSN: received GTK in group key handshake",
1402                         ie.gtk, ie.gtk_len);
1403         gd->keyidx = ie.gtk[0] & 0x3;
1404         gd->tx = wpa_supplicant_gtk_tx_bit_workaround(sm,
1405                                                       !!(ie.gtk[0] & BIT(2)));
1406         if (ie.gtk_len - 2 > sizeof(gd->gtk)) {
1407                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1408                         "RSN: Too long GTK in GTK IE (len=%lu)",
1409                         (unsigned long) ie.gtk_len - 2);
1410                 return -1;
1411         }
1412         os_memcpy(gd->gtk, ie.gtk + 2, ie.gtk_len - 2);
1413
1414         if (ieee80211w_set_keys(sm, &ie) < 0)
1415                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1416                         "RSN: Failed to configure IGTK");
1417
1418         return 0;
1419 }
1420
1421
1422 static int wpa_supplicant_process_1_of_2_wpa(struct wpa_sm *sm,
1423                                              const struct wpa_eapol_key *key,
1424                                              const u8 *key_data,
1425                                              size_t key_data_len, u16 key_info,
1426                                              u16 ver, struct wpa_gtk_data *gd)
1427 {
1428         size_t maxkeylen;
1429         u16 gtk_len;
1430
1431         gtk_len = WPA_GET_BE16(key->key_length);
1432         maxkeylen = key_data_len;
1433         if (ver == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
1434                 if (maxkeylen < 8) {
1435                         wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1436                                 "WPA: Too short maxkeylen (%lu)",
1437                                 (unsigned long) maxkeylen);
1438                         return -1;
1439                 }
1440                 maxkeylen -= 8;
1441         }
1442
1443         if (gtk_len > maxkeylen ||
1444             wpa_supplicant_check_group_cipher(sm, sm->group_cipher,
1445                                               gtk_len, maxkeylen,
1446                                               &gd->key_rsc_len, &gd->alg))
1447                 return -1;
1448
1449         gd->gtk_len = gtk_len;
1450         gd->keyidx = (key_info & WPA_KEY_INFO_KEY_INDEX_MASK) >>
1451                 WPA_KEY_INFO_KEY_INDEX_SHIFT;
1452         if (ver == WPA_KEY_INFO_TYPE_HMAC_MD5_RC4 && sm->ptk.kek_len == 16) {
1453 #ifdef CONFIG_NO_RC4
1454                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1455                         "WPA: RC4 not supported in the build");
1456                 return -1;
1457 #else /* CONFIG_NO_RC4 */
1458                 u8 ek[32];
1459                 if (key_data_len > sizeof(gd->gtk)) {
1460                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1461                                 "WPA: RC4 key data too long (%lu)",
1462                                 (unsigned long) key_data_len);
1463                         return -1;
1464                 }
1465                 os_memcpy(ek, key->key_iv, 16);
1466                 os_memcpy(ek + 16, sm->ptk.kek, sm->ptk.kek_len);
1467                 os_memcpy(gd->gtk, key_data, key_data_len);
1468                 if (rc4_skip(ek, 32, 256, gd->gtk, key_data_len)) {
1469                         os_memset(ek, 0, sizeof(ek));
1470                         wpa_msg(sm->ctx->msg_ctx, MSG_ERROR,
1471                                 "WPA: RC4 failed");
1472                         return -1;
1473                 }
1474                 os_memset(ek, 0, sizeof(ek));
1475 #endif /* CONFIG_NO_RC4 */
1476         } else if (ver == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
1477                 if (maxkeylen % 8) {
1478                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1479                                 "WPA: Unsupported AES-WRAP len %lu",
1480                                 (unsigned long) maxkeylen);
1481                         return -1;
1482                 }
1483                 if (maxkeylen > sizeof(gd->gtk)) {
1484                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1485                                 "WPA: AES-WRAP key data "
1486                                 "too long (keydatalen=%lu maxkeylen=%lu)",
1487                                 (unsigned long) key_data_len,
1488                                 (unsigned long) maxkeylen);
1489                         return -1;
1490                 }
1491                 if (aes_unwrap(sm->ptk.kek, sm->ptk.kek_len, maxkeylen / 8,
1492                                key_data, gd->gtk)) {
1493                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1494                                 "WPA: AES unwrap failed - could not decrypt "
1495                                 "GTK");
1496                         return -1;
1497                 }
1498         } else {
1499                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1500                         "WPA: Unsupported key_info type %d", ver);
1501                 return -1;
1502         }
1503         gd->tx = wpa_supplicant_gtk_tx_bit_workaround(
1504                 sm, !!(key_info & WPA_KEY_INFO_TXRX));
1505         return 0;
1506 }
1507
1508
1509 static int wpa_supplicant_send_2_of_2(struct wpa_sm *sm,
1510                                       const struct wpa_eapol_key *key,
1511                                       int ver, u16 key_info)
1512 {
1513         size_t mic_len, hdrlen, rlen;
1514         struct wpa_eapol_key *reply;
1515         struct wpa_eapol_key_192 *reply192;
1516         u8 *rbuf, *key_mic;
1517
1518         mic_len = wpa_mic_len(sm->key_mgmt);
1519         hdrlen = mic_len == 24 ? sizeof(*reply192) : sizeof(*reply);
1520         rbuf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_KEY, NULL,
1521                                   hdrlen, &rlen, (void *) &reply);
1522         if (rbuf == NULL)
1523                 return -1;
1524         reply192 = (struct wpa_eapol_key_192 *) reply;
1525
1526         reply->type = (sm->proto == WPA_PROTO_RSN ||
1527                        sm->proto == WPA_PROTO_OSEN) ?
1528                 EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
1529         key_info &= WPA_KEY_INFO_KEY_INDEX_MASK;
1530         key_info |= ver | WPA_KEY_INFO_MIC | WPA_KEY_INFO_SECURE;
1531         WPA_PUT_BE16(reply->key_info, key_info);
1532         if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN)
1533                 WPA_PUT_BE16(reply->key_length, 0);
1534         else
1535                 os_memcpy(reply->key_length, key->key_length, 2);
1536         os_memcpy(reply->replay_counter, key->replay_counter,
1537                   WPA_REPLAY_COUNTER_LEN);
1538
1539         key_mic = reply192->key_mic; /* same offset for reply and reply192 */
1540         if (mic_len == 24)
1541                 WPA_PUT_BE16(reply192->key_data_length, 0);
1542         else
1543                 WPA_PUT_BE16(reply->key_data_length, 0);
1544
1545         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Sending EAPOL-Key 2/2");
1546         return wpa_eapol_key_send(sm, sm->ptk.kck, sm->ptk.kck_len, ver,
1547                                   sm->bssid, ETH_P_EAPOL, rbuf, rlen, key_mic);
1548 }
1549
1550
1551 static void wpa_supplicant_process_1_of_2(struct wpa_sm *sm,
1552                                           const unsigned char *src_addr,
1553                                           const struct wpa_eapol_key *key,
1554                                           const u8 *key_data,
1555                                           size_t key_data_len, u16 ver)
1556 {
1557         u16 key_info;
1558         int rekey, ret;
1559         struct wpa_gtk_data gd;
1560         const u8 *key_rsc;
1561
1562         if (!sm->msg_3_of_4_ok) {
1563                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1564                         "WPA: Group Key Handshake started prior to completion of 4-way handshake");
1565                 goto failed;
1566         }
1567
1568         os_memset(&gd, 0, sizeof(gd));
1569
1570         rekey = wpa_sm_get_state(sm) == WPA_COMPLETED;
1571         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: RX message 1 of Group Key "
1572                 "Handshake from " MACSTR " (ver=%d)", MAC2STR(src_addr), ver);
1573
1574         key_info = WPA_GET_BE16(key->key_info);
1575
1576         if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN) {
1577                 ret = wpa_supplicant_process_1_of_2_rsn(sm, key_data,
1578                                                         key_data_len, key_info,
1579                                                         &gd);
1580         } else {
1581                 ret = wpa_supplicant_process_1_of_2_wpa(sm, key, key_data,
1582                                                         key_data_len,
1583                                                         key_info, ver, &gd);
1584         }
1585
1586         wpa_sm_set_state(sm, WPA_GROUP_HANDSHAKE);
1587
1588         if (ret)
1589                 goto failed;
1590
1591         key_rsc = key->key_rsc;
1592         if (wpa_supplicant_rsc_relaxation(sm, key->key_rsc))
1593                 key_rsc = null_rsc;
1594
1595         if (wpa_supplicant_install_gtk(sm, &gd, key_rsc, 0) ||
1596             wpa_supplicant_send_2_of_2(sm, key, ver, key_info) < 0)
1597                 goto failed;
1598         os_memset(&gd, 0, sizeof(gd));
1599
1600         if (rekey) {
1601                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO, "WPA: Group rekeying "
1602                         "completed with " MACSTR " [GTK=%s]",
1603                         MAC2STR(sm->bssid), wpa_cipher_txt(sm->group_cipher));
1604                 wpa_sm_cancel_auth_timeout(sm);
1605                 wpa_sm_set_state(sm, WPA_COMPLETED);
1606         } else {
1607                 wpa_supplicant_key_neg_complete(sm, sm->bssid,
1608                                                 key_info &
1609                                                 WPA_KEY_INFO_SECURE);
1610         }
1611
1612         wpa_sm_set_rekey_offload(sm);
1613
1614         return;
1615
1616 failed:
1617         os_memset(&gd, 0, sizeof(gd));
1618         wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
1619 }
1620
1621
1622 static int wpa_supplicant_verify_eapol_key_mic(struct wpa_sm *sm,
1623                                                struct wpa_eapol_key_192 *key,
1624                                                u16 ver,
1625                                                const u8 *buf, size_t len)
1626 {
1627         u8 mic[WPA_EAPOL_KEY_MIC_MAX_LEN];
1628         int ok = 0;
1629         size_t mic_len = wpa_mic_len(sm->key_mgmt);
1630
1631         os_memcpy(mic, key->key_mic, mic_len);
1632         if (sm->tptk_set) {
1633                 os_memset(key->key_mic, 0, mic_len);
1634                 wpa_eapol_key_mic(sm->tptk.kck, sm->tptk.kck_len, sm->key_mgmt,
1635                                   ver, buf, len, key->key_mic);
1636                 if (os_memcmp_const(mic, key->key_mic, mic_len) != 0) {
1637                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1638                                 "WPA: Invalid EAPOL-Key MIC "
1639                                 "when using TPTK - ignoring TPTK");
1640                 } else {
1641                         ok = 1;
1642                         sm->tptk_set = 0;
1643                         sm->ptk_set = 1;
1644                         os_memcpy(&sm->ptk, &sm->tptk, sizeof(sm->ptk));
1645                         os_memset(&sm->tptk, 0, sizeof(sm->tptk));
1646                 }
1647         }
1648
1649         if (!ok && sm->ptk_set) {
1650                 os_memset(key->key_mic, 0, mic_len);
1651                 wpa_eapol_key_mic(sm->ptk.kck, sm->ptk.kck_len, sm->key_mgmt,
1652                                   ver, buf, len, key->key_mic);
1653                 if (os_memcmp_const(mic, key->key_mic, mic_len) != 0) {
1654                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1655                                 "WPA: Invalid EAPOL-Key MIC - "
1656                                 "dropping packet");
1657                         return -1;
1658                 }
1659                 ok = 1;
1660         }
1661
1662         if (!ok) {
1663                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1664                         "WPA: Could not verify EAPOL-Key MIC - "
1665                         "dropping packet");
1666                 return -1;
1667         }
1668
1669         os_memcpy(sm->rx_replay_counter, key->replay_counter,
1670                   WPA_REPLAY_COUNTER_LEN);
1671         sm->rx_replay_counter_set = 1;
1672         return 0;
1673 }
1674
1675
1676 /* Decrypt RSN EAPOL-Key key data (RC4 or AES-WRAP) */
1677 static int wpa_supplicant_decrypt_key_data(struct wpa_sm *sm,
1678                                            struct wpa_eapol_key *key, u16 ver,
1679                                            u8 *key_data, size_t *key_data_len)
1680 {
1681         wpa_hexdump(MSG_DEBUG, "RSN: encrypted key data",
1682                     key_data, *key_data_len);
1683         if (!sm->ptk_set) {
1684                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1685                         "WPA: PTK not available, cannot decrypt EAPOL-Key Key "
1686                         "Data");
1687                 return -1;
1688         }
1689
1690         /* Decrypt key data here so that this operation does not need
1691          * to be implemented separately for each message type. */
1692         if (ver == WPA_KEY_INFO_TYPE_HMAC_MD5_RC4 && sm->ptk.kek_len == 16) {
1693 #ifdef CONFIG_NO_RC4
1694                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1695                         "WPA: RC4 not supported in the build");
1696                 return -1;
1697 #else /* CONFIG_NO_RC4 */
1698                 u8 ek[32];
1699                 os_memcpy(ek, key->key_iv, 16);
1700                 os_memcpy(ek + 16, sm->ptk.kek, sm->ptk.kek_len);
1701                 if (rc4_skip(ek, 32, 256, key_data, *key_data_len)) {
1702                         os_memset(ek, 0, sizeof(ek));
1703                         wpa_msg(sm->ctx->msg_ctx, MSG_ERROR,
1704                                 "WPA: RC4 failed");
1705                         return -1;
1706                 }
1707                 os_memset(ek, 0, sizeof(ek));
1708 #endif /* CONFIG_NO_RC4 */
1709         } else if (ver == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES ||
1710                    ver == WPA_KEY_INFO_TYPE_AES_128_CMAC ||
1711                    sm->key_mgmt == WPA_KEY_MGMT_OSEN ||
1712                    wpa_key_mgmt_suite_b(sm->key_mgmt)) {
1713                 u8 *buf;
1714                 if (*key_data_len < 8 || *key_data_len % 8) {
1715                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1716                                 "WPA: Unsupported AES-WRAP len %u",
1717                                 (unsigned int) *key_data_len);
1718                         return -1;
1719                 }
1720                 *key_data_len -= 8; /* AES-WRAP adds 8 bytes */
1721                 buf = os_malloc(*key_data_len);
1722                 if (buf == NULL) {
1723                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1724                                 "WPA: No memory for AES-UNWRAP buffer");
1725                         return -1;
1726                 }
1727                 if (aes_unwrap(sm->ptk.kek, sm->ptk.kek_len, *key_data_len / 8,
1728                                key_data, buf)) {
1729                         bin_clear_free(buf, *key_data_len);
1730                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1731                                 "WPA: AES unwrap failed - "
1732                                 "could not decrypt EAPOL-Key key data");
1733                         return -1;
1734                 }
1735                 os_memcpy(key_data, buf, *key_data_len);
1736                 bin_clear_free(buf, *key_data_len);
1737                 WPA_PUT_BE16(key->key_data_length, *key_data_len);
1738         } else {
1739                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
1740                         "WPA: Unsupported key_info type %d", ver);
1741                 return -1;
1742         }
1743         wpa_hexdump_key(MSG_DEBUG, "WPA: decrypted EAPOL-Key key data",
1744                         key_data, *key_data_len);
1745         return 0;
1746 }
1747
1748
1749 /**
1750  * wpa_sm_aborted_cached - Notify WPA that PMKSA caching was aborted
1751  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1752  */
1753 void wpa_sm_aborted_cached(struct wpa_sm *sm)
1754 {
1755         if (sm && sm->cur_pmksa) {
1756                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1757                         "RSN: Cancelling PMKSA caching attempt");
1758                 sm->cur_pmksa = NULL;
1759         }
1760 }
1761
1762
1763 static void wpa_eapol_key_dump(struct wpa_sm *sm,
1764                                const struct wpa_eapol_key *key,
1765                                unsigned int key_data_len,
1766                                const u8 *mic, unsigned int mic_len)
1767 {
1768 #ifndef CONFIG_NO_STDOUT_DEBUG
1769         u16 key_info = WPA_GET_BE16(key->key_info);
1770
1771         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "  EAPOL-Key type=%d", key->type);
1772         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1773                 "  key_info 0x%x (ver=%d keyidx=%d rsvd=%d %s%s%s%s%s%s%s%s)",
1774                 key_info, key_info & WPA_KEY_INFO_TYPE_MASK,
1775                 (key_info & WPA_KEY_INFO_KEY_INDEX_MASK) >>
1776                 WPA_KEY_INFO_KEY_INDEX_SHIFT,
1777                 (key_info & (BIT(13) | BIT(14) | BIT(15))) >> 13,
1778                 key_info & WPA_KEY_INFO_KEY_TYPE ? "Pairwise" : "Group",
1779                 key_info & WPA_KEY_INFO_INSTALL ? " Install" : "",
1780                 key_info & WPA_KEY_INFO_ACK ? " Ack" : "",
1781                 key_info & WPA_KEY_INFO_MIC ? " MIC" : "",
1782                 key_info & WPA_KEY_INFO_SECURE ? " Secure" : "",
1783                 key_info & WPA_KEY_INFO_ERROR ? " Error" : "",
1784                 key_info & WPA_KEY_INFO_REQUEST ? " Request" : "",
1785                 key_info & WPA_KEY_INFO_ENCR_KEY_DATA ? " Encr" : "");
1786         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1787                 "  key_length=%u key_data_length=%u",
1788                 WPA_GET_BE16(key->key_length), key_data_len);
1789         wpa_hexdump(MSG_DEBUG, "  replay_counter",
1790                     key->replay_counter, WPA_REPLAY_COUNTER_LEN);
1791         wpa_hexdump(MSG_DEBUG, "  key_nonce", key->key_nonce, WPA_NONCE_LEN);
1792         wpa_hexdump(MSG_DEBUG, "  key_iv", key->key_iv, 16);
1793         wpa_hexdump(MSG_DEBUG, "  key_rsc", key->key_rsc, 8);
1794         wpa_hexdump(MSG_DEBUG, "  key_id (reserved)", key->key_id, 8);
1795         wpa_hexdump(MSG_DEBUG, "  key_mic", mic, mic_len);
1796 #endif /* CONFIG_NO_STDOUT_DEBUG */
1797 }
1798
1799
1800 /**
1801  * wpa_sm_rx_eapol - Process received WPA EAPOL frames
1802  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1803  * @src_addr: Source MAC address of the EAPOL packet
1804  * @buf: Pointer to the beginning of the EAPOL data (EAPOL header)
1805  * @len: Length of the EAPOL frame
1806  * Returns: 1 = WPA EAPOL-Key processed, 0 = not a WPA EAPOL-Key, -1 failure
1807  *
1808  * This function is called for each received EAPOL frame. Other than EAPOL-Key
1809  * frames can be skipped if filtering is done elsewhere. wpa_sm_rx_eapol() is
1810  * only processing WPA and WPA2 EAPOL-Key frames.
1811  *
1812  * The received EAPOL-Key packets are validated and valid packets are replied
1813  * to. In addition, key material (PTK, GTK) is configured at the end of a
1814  * successful key handshake.
1815  */
1816 int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
1817                     const u8 *buf, size_t len)
1818 {
1819         size_t plen, data_len, key_data_len;
1820         const struct ieee802_1x_hdr *hdr;
1821         struct wpa_eapol_key *key;
1822         struct wpa_eapol_key_192 *key192;
1823         u16 key_info, ver;
1824         u8 *tmp = NULL;
1825         int ret = -1;
1826         struct wpa_peerkey *peerkey = NULL;
1827         u8 *key_data;
1828         size_t mic_len, keyhdrlen;
1829
1830 #ifdef CONFIG_IEEE80211R
1831         sm->ft_completed = 0;
1832 #endif /* CONFIG_IEEE80211R */
1833
1834         mic_len = wpa_mic_len(sm->key_mgmt);
1835         keyhdrlen = mic_len == 24 ? sizeof(*key192) : sizeof(*key);
1836
1837         if (len < sizeof(*hdr) + keyhdrlen) {
1838                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1839                         "WPA: EAPOL frame too short to be a WPA "
1840                         "EAPOL-Key (len %lu, expecting at least %lu)",
1841                         (unsigned long) len,
1842                         (unsigned long) sizeof(*hdr) + keyhdrlen);
1843                 return 0;
1844         }
1845
1846         hdr = (const struct ieee802_1x_hdr *) buf;
1847         plen = be_to_host16(hdr->length);
1848         data_len = plen + sizeof(*hdr);
1849         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1850                 "IEEE 802.1X RX: version=%d type=%d length=%lu",
1851                 hdr->version, hdr->type, (unsigned long) plen);
1852
1853         if (hdr->version < EAPOL_VERSION) {
1854                 /* TODO: backwards compatibility */
1855         }
1856         if (hdr->type != IEEE802_1X_TYPE_EAPOL_KEY) {
1857                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1858                         "WPA: EAPOL frame (type %u) discarded, "
1859                         "not a Key frame", hdr->type);
1860                 ret = 0;
1861                 goto out;
1862         }
1863         wpa_hexdump(MSG_MSGDUMP, "WPA: RX EAPOL-Key", buf, len);
1864         if (plen > len - sizeof(*hdr) || plen < keyhdrlen) {
1865                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1866                         "WPA: EAPOL frame payload size %lu "
1867                         "invalid (frame size %lu)",
1868                         (unsigned long) plen, (unsigned long) len);
1869                 ret = 0;
1870                 goto out;
1871         }
1872         if (data_len < len) {
1873                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1874                         "WPA: ignoring %lu bytes after the IEEE 802.1X data",
1875                         (unsigned long) len - data_len);
1876         }
1877
1878         /*
1879          * Make a copy of the frame since we need to modify the buffer during
1880          * MAC validation and Key Data decryption.
1881          */
1882         tmp = os_malloc(data_len);
1883         if (tmp == NULL)
1884                 goto out;
1885         os_memcpy(tmp, buf, data_len);
1886         key = (struct wpa_eapol_key *) (tmp + sizeof(struct ieee802_1x_hdr));
1887         key192 = (struct wpa_eapol_key_192 *)
1888                 (tmp + sizeof(struct ieee802_1x_hdr));
1889         if (mic_len == 24)
1890                 key_data = (u8 *) (key192 + 1);
1891         else
1892                 key_data = (u8 *) (key + 1);
1893
1894         if (key->type != EAPOL_KEY_TYPE_WPA && key->type != EAPOL_KEY_TYPE_RSN)
1895         {
1896                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
1897                         "WPA: EAPOL-Key type (%d) unknown, discarded",
1898                         key->type);
1899                 ret = 0;
1900                 goto out;
1901         }
1902
1903         if (mic_len == 24)
1904                 key_data_len = WPA_GET_BE16(key192->key_data_length);
1905         else
1906                 key_data_len = WPA_GET_BE16(key->key_data_length);
1907         wpa_eapol_key_dump(sm, key, key_data_len, key192->key_mic, mic_len);
1908
1909         if (key_data_len > plen - keyhdrlen) {
1910                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO, "WPA: Invalid EAPOL-Key "
1911                         "frame - key_data overflow (%u > %u)",
1912                         (unsigned int) key_data_len,
1913                         (unsigned int) (plen - keyhdrlen));
1914                 goto out;
1915         }
1916
1917         eapol_sm_notify_lower_layer_success(sm->eapol, 0);
1918         key_info = WPA_GET_BE16(key->key_info);
1919         ver = key_info & WPA_KEY_INFO_TYPE_MASK;
1920         if (ver != WPA_KEY_INFO_TYPE_HMAC_MD5_RC4 &&
1921 #if defined(CONFIG_IEEE80211R) || defined(CONFIG_IEEE80211W)
1922             ver != WPA_KEY_INFO_TYPE_AES_128_CMAC &&
1923 #endif /* CONFIG_IEEE80211R || CONFIG_IEEE80211W */
1924             ver != WPA_KEY_INFO_TYPE_HMAC_SHA1_AES &&
1925             !wpa_key_mgmt_suite_b(sm->key_mgmt) &&
1926             sm->key_mgmt != WPA_KEY_MGMT_OSEN) {
1927                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1928                         "WPA: Unsupported EAPOL-Key descriptor version %d",
1929                         ver);
1930                 goto out;
1931         }
1932
1933         if (sm->key_mgmt == WPA_KEY_MGMT_OSEN &&
1934             ver != WPA_KEY_INFO_TYPE_AKM_DEFINED) {
1935                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1936                         "OSEN: Unsupported EAPOL-Key descriptor version %d",
1937                         ver);
1938                 goto out;
1939         }
1940
1941         if (wpa_key_mgmt_suite_b(sm->key_mgmt) &&
1942             ver != WPA_KEY_INFO_TYPE_AKM_DEFINED) {
1943                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1944                         "RSN: Unsupported EAPOL-Key descriptor version %d (expected AKM defined = 0)",
1945                         ver);
1946                 goto out;
1947         }
1948
1949 #ifdef CONFIG_IEEE80211R
1950         if (wpa_key_mgmt_ft(sm->key_mgmt)) {
1951                 /* IEEE 802.11r uses a new key_info type (AES-128-CMAC). */
1952                 if (ver != WPA_KEY_INFO_TYPE_AES_128_CMAC) {
1953                         wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1954                                 "FT: AP did not use AES-128-CMAC");
1955                         goto out;
1956                 }
1957         } else
1958 #endif /* CONFIG_IEEE80211R */
1959 #ifdef CONFIG_IEEE80211W
1960         if (wpa_key_mgmt_sha256(sm->key_mgmt)) {
1961                 if (ver != WPA_KEY_INFO_TYPE_AES_128_CMAC &&
1962                     sm->key_mgmt != WPA_KEY_MGMT_OSEN &&
1963                     !wpa_key_mgmt_suite_b(sm->key_mgmt)) {
1964                         wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1965                                 "WPA: AP did not use the "
1966                                 "negotiated AES-128-CMAC");
1967                         goto out;
1968                 }
1969         } else
1970 #endif /* CONFIG_IEEE80211W */
1971         if (sm->pairwise_cipher == WPA_CIPHER_CCMP &&
1972             !wpa_key_mgmt_suite_b(sm->key_mgmt) &&
1973             ver != WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
1974                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1975                         "WPA: CCMP is used, but EAPOL-Key "
1976                         "descriptor version (%d) is not 2", ver);
1977                 if (sm->group_cipher != WPA_CIPHER_CCMP &&
1978                     !(key_info & WPA_KEY_INFO_KEY_TYPE)) {
1979                         /* Earlier versions of IEEE 802.11i did not explicitly
1980                          * require version 2 descriptor for all EAPOL-Key
1981                          * packets, so allow group keys to use version 1 if
1982                          * CCMP is not used for them. */
1983                         wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1984                                 "WPA: Backwards compatibility: allow invalid "
1985                                 "version for non-CCMP group keys");
1986                 } else if (ver == WPA_KEY_INFO_TYPE_AES_128_CMAC) {
1987                         wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1988                                 "WPA: Interoperability workaround: allow incorrect (should have been HMAC-SHA1), but stronger (is AES-128-CMAC), descriptor version to be used");
1989                 } else
1990                         goto out;
1991         } else if (sm->pairwise_cipher == WPA_CIPHER_GCMP &&
1992                    !wpa_key_mgmt_suite_b(sm->key_mgmt) &&
1993                    ver != WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
1994                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
1995                         "WPA: GCMP is used, but EAPOL-Key "
1996                         "descriptor version (%d) is not 2", ver);
1997                 goto out;
1998         }
1999
2000 #ifdef CONFIG_PEERKEY
2001         for (peerkey = sm->peerkey; peerkey; peerkey = peerkey->next) {
2002                 if (os_memcmp(peerkey->addr, src_addr, ETH_ALEN) == 0)
2003                         break;
2004         }
2005
2006         if (!(key_info & WPA_KEY_INFO_SMK_MESSAGE) && peerkey) {
2007                 if (!peerkey->initiator && peerkey->replay_counter_set &&
2008                     os_memcmp(key->replay_counter, peerkey->replay_counter,
2009                               WPA_REPLAY_COUNTER_LEN) <= 0) {
2010                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
2011                                 "RSN: EAPOL-Key Replay Counter did not "
2012                                 "increase (STK) - dropping packet");
2013                         goto out;
2014                 } else if (peerkey->initiator) {
2015                         u8 _tmp[WPA_REPLAY_COUNTER_LEN];
2016                         os_memcpy(_tmp, key->replay_counter,
2017                                   WPA_REPLAY_COUNTER_LEN);
2018                         inc_byte_array(_tmp, WPA_REPLAY_COUNTER_LEN);
2019                         if (os_memcmp(_tmp, peerkey->replay_counter,
2020                                       WPA_REPLAY_COUNTER_LEN) != 0) {
2021                                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
2022                                         "RSN: EAPOL-Key Replay "
2023                                         "Counter did not match (STK) - "
2024                                         "dropping packet");
2025                                 goto out;
2026                         }
2027                 }
2028         }
2029
2030         if (peerkey && peerkey->initiator && (key_info & WPA_KEY_INFO_ACK)) {
2031                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
2032                         "RSN: Ack bit in key_info from STK peer");
2033                 goto out;
2034         }
2035 #endif /* CONFIG_PEERKEY */
2036
2037         if (!peerkey && sm->rx_replay_counter_set &&
2038             os_memcmp(key->replay_counter, sm->rx_replay_counter,
2039                       WPA_REPLAY_COUNTER_LEN) <= 0) {
2040                 wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
2041                         "WPA: EAPOL-Key Replay Counter did not increase - "
2042                         "dropping packet");
2043                 goto out;
2044         }
2045
2046         if (!(key_info & (WPA_KEY_INFO_ACK | WPA_KEY_INFO_SMK_MESSAGE))
2047 #ifdef CONFIG_PEERKEY
2048             && (peerkey == NULL || !peerkey->initiator)
2049 #endif /* CONFIG_PEERKEY */
2050                 ) {
2051                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
2052                         "WPA: No Ack bit in key_info");
2053                 goto out;
2054         }
2055
2056         if (key_info & WPA_KEY_INFO_REQUEST) {
2057                 wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
2058                         "WPA: EAPOL-Key with Request bit - dropped");
2059                 goto out;
2060         }
2061
2062         if ((key_info & WPA_KEY_INFO_MIC) && !peerkey &&
2063             wpa_supplicant_verify_eapol_key_mic(sm, key192, ver, tmp, data_len))
2064                 goto out;
2065
2066 #ifdef CONFIG_PEERKEY
2067         if ((key_info & WPA_KEY_INFO_MIC) && peerkey &&
2068             peerkey_verify_eapol_key_mic(sm, peerkey, key192, ver, tmp,
2069                                          data_len))
2070                 goto out;
2071 #endif /* CONFIG_PEERKEY */
2072
2073         if ((sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN) &&
2074             (key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
2075                 /*
2076                  * Only decrypt the Key Data field if the frame's authenticity
2077                  * was verified. When using AES-SIV (FILS), the MIC flag is not
2078                  * set, so this check should only be performed if mic_len != 0
2079                  * which is the case in this code branch.
2080                  */
2081                 if (!(key_info & WPA_KEY_INFO_MIC)) {
2082                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
2083                                 "WPA: Ignore EAPOL-Key with encrypted but unauthenticated data");
2084                         goto out;
2085                 }
2086                 if (wpa_supplicant_decrypt_key_data(sm, key, ver, key_data,
2087                                                     &key_data_len))
2088                         goto out;
2089         }
2090
2091         if (key_info & WPA_KEY_INFO_KEY_TYPE) {
2092                 if (key_info & WPA_KEY_INFO_KEY_INDEX_MASK) {
2093                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
2094                                 "WPA: Ignored EAPOL-Key (Pairwise) with "
2095                                 "non-zero key index");
2096                         goto out;
2097                 }
2098                 if (peerkey) {
2099                         /* PeerKey 4-Way Handshake */
2100                         peerkey_rx_eapol_4way(sm, peerkey, key, key_info, ver,
2101                                               key_data, key_data_len);
2102                 } else if (key_info & WPA_KEY_INFO_MIC) {
2103                         /* 3/4 4-Way Handshake */
2104                         wpa_supplicant_process_3_of_4(sm, key, ver, key_data,
2105                                                       key_data_len);
2106                 } else {
2107                         /* 1/4 4-Way Handshake */
2108                         wpa_supplicant_process_1_of_4(sm, src_addr, key,
2109                                                       ver, key_data,
2110                                                       key_data_len);
2111                 }
2112         } else if (key_info & WPA_KEY_INFO_SMK_MESSAGE) {
2113                 /* PeerKey SMK Handshake */
2114                 peerkey_rx_eapol_smk(sm, src_addr, key, key_data_len, key_info,
2115                                      ver);
2116         } else {
2117                 if (key_info & WPA_KEY_INFO_MIC) {
2118                         /* 1/2 Group Key Handshake */
2119                         wpa_supplicant_process_1_of_2(sm, src_addr, key,
2120                                                       key_data, key_data_len,
2121                                                       ver);
2122                 } else {
2123                         wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
2124                                 "WPA: EAPOL-Key (Group) without Mic bit - "
2125                                 "dropped");
2126                 }
2127         }
2128
2129         ret = 1;
2130
2131 out:
2132         bin_clear_free(tmp, data_len);
2133         return ret;
2134 }
2135
2136
2137 #ifdef CONFIG_CTRL_IFACE
2138 static u32 wpa_key_mgmt_suite(struct wpa_sm *sm)
2139 {
2140         switch (sm->key_mgmt) {
2141         case WPA_KEY_MGMT_IEEE8021X:
2142                 return ((sm->proto == WPA_PROTO_RSN ||
2143                          sm->proto == WPA_PROTO_OSEN) ?
2144                         RSN_AUTH_KEY_MGMT_UNSPEC_802_1X :
2145                         WPA_AUTH_KEY_MGMT_UNSPEC_802_1X);
2146         case WPA_KEY_MGMT_PSK:
2147                 return (sm->proto == WPA_PROTO_RSN ?
2148                         RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X :
2149                         WPA_AUTH_KEY_MGMT_PSK_OVER_802_1X);
2150 #ifdef CONFIG_IEEE80211R
2151         case WPA_KEY_MGMT_FT_IEEE8021X:
2152                 return RSN_AUTH_KEY_MGMT_FT_802_1X;
2153         case WPA_KEY_MGMT_FT_PSK:
2154                 return RSN_AUTH_KEY_MGMT_FT_PSK;
2155 #endif /* CONFIG_IEEE80211R */
2156 #ifdef CONFIG_IEEE80211W
2157         case WPA_KEY_MGMT_IEEE8021X_SHA256:
2158                 return RSN_AUTH_KEY_MGMT_802_1X_SHA256;
2159         case WPA_KEY_MGMT_PSK_SHA256:
2160                 return RSN_AUTH_KEY_MGMT_PSK_SHA256;
2161 #endif /* CONFIG_IEEE80211W */
2162         case WPA_KEY_MGMT_CCKM:
2163                 return (sm->proto == WPA_PROTO_RSN ?
2164                         RSN_AUTH_KEY_MGMT_CCKM:
2165                         WPA_AUTH_KEY_MGMT_CCKM);
2166         case WPA_KEY_MGMT_WPA_NONE:
2167                 return WPA_AUTH_KEY_MGMT_NONE;
2168         case WPA_KEY_MGMT_IEEE8021X_SUITE_B:
2169                 return RSN_AUTH_KEY_MGMT_802_1X_SUITE_B;
2170         case WPA_KEY_MGMT_IEEE8021X_SUITE_B_192:
2171                 return RSN_AUTH_KEY_MGMT_802_1X_SUITE_B_192;
2172         default:
2173                 return 0;
2174         }
2175 }
2176
2177
2178 #define RSN_SUITE "%02x-%02x-%02x-%d"
2179 #define RSN_SUITE_ARG(s) \
2180 ((s) >> 24) & 0xff, ((s) >> 16) & 0xff, ((s) >> 8) & 0xff, (s) & 0xff
2181
2182 /**
2183  * wpa_sm_get_mib - Dump text list of MIB entries
2184  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2185  * @buf: Buffer for the list
2186  * @buflen: Length of the buffer
2187  * Returns: Number of bytes written to buffer
2188  *
2189  * This function is used fetch dot11 MIB variables.
2190  */
2191 int wpa_sm_get_mib(struct wpa_sm *sm, char *buf, size_t buflen)
2192 {
2193         char pmkid_txt[PMKID_LEN * 2 + 1];
2194         int rsna, ret;
2195         size_t len;
2196
2197         if (sm->cur_pmksa) {
2198                 wpa_snprintf_hex(pmkid_txt, sizeof(pmkid_txt),
2199                                  sm->cur_pmksa->pmkid, PMKID_LEN);
2200         } else
2201                 pmkid_txt[0] = '\0';
2202
2203         if ((wpa_key_mgmt_wpa_psk(sm->key_mgmt) ||
2204              wpa_key_mgmt_wpa_ieee8021x(sm->key_mgmt)) &&
2205             sm->proto == WPA_PROTO_RSN)
2206                 rsna = 1;
2207         else
2208                 rsna = 0;
2209
2210         ret = os_snprintf(buf, buflen,
2211                           "dot11RSNAOptionImplemented=TRUE\n"
2212                           "dot11RSNAPreauthenticationImplemented=TRUE\n"
2213                           "dot11RSNAEnabled=%s\n"
2214                           "dot11RSNAPreauthenticationEnabled=%s\n"
2215                           "dot11RSNAConfigVersion=%d\n"
2216                           "dot11RSNAConfigPairwiseKeysSupported=5\n"
2217                           "dot11RSNAConfigGroupCipherSize=%d\n"
2218                           "dot11RSNAConfigPMKLifetime=%d\n"
2219                           "dot11RSNAConfigPMKReauthThreshold=%d\n"
2220                           "dot11RSNAConfigNumberOfPTKSAReplayCounters=1\n"
2221                           "dot11RSNAConfigSATimeout=%d\n",
2222                           rsna ? "TRUE" : "FALSE",
2223                           rsna ? "TRUE" : "FALSE",
2224                           RSN_VERSION,
2225                           wpa_cipher_key_len(sm->group_cipher) * 8,
2226                           sm->dot11RSNAConfigPMKLifetime,
2227                           sm->dot11RSNAConfigPMKReauthThreshold,
2228                           sm->dot11RSNAConfigSATimeout);
2229         if (os_snprintf_error(buflen, ret))
2230                 return 0;
2231         len = ret;
2232
2233         ret = os_snprintf(
2234                 buf + len, buflen - len,
2235                 "dot11RSNAAuthenticationSuiteSelected=" RSN_SUITE "\n"
2236                 "dot11RSNAPairwiseCipherSelected=" RSN_SUITE "\n"
2237                 "dot11RSNAGroupCipherSelected=" RSN_SUITE "\n"
2238                 "dot11RSNAPMKIDUsed=%s\n"
2239                 "dot11RSNAAuthenticationSuiteRequested=" RSN_SUITE "\n"
2240                 "dot11RSNAPairwiseCipherRequested=" RSN_SUITE "\n"
2241                 "dot11RSNAGroupCipherRequested=" RSN_SUITE "\n"
2242                 "dot11RSNAConfigNumberOfGTKSAReplayCounters=0\n"
2243                 "dot11RSNA4WayHandshakeFailures=%u\n",
2244                 RSN_SUITE_ARG(wpa_key_mgmt_suite(sm)),
2245                 RSN_SUITE_ARG(wpa_cipher_to_suite(sm->proto,
2246                                                   sm->pairwise_cipher)),
2247                 RSN_SUITE_ARG(wpa_cipher_to_suite(sm->proto,
2248                                                   sm->group_cipher)),
2249                 pmkid_txt,
2250                 RSN_SUITE_ARG(wpa_key_mgmt_suite(sm)),
2251                 RSN_SUITE_ARG(wpa_cipher_to_suite(sm->proto,
2252                                                   sm->pairwise_cipher)),
2253                 RSN_SUITE_ARG(wpa_cipher_to_suite(sm->proto,
2254                                                   sm->group_cipher)),
2255                 sm->dot11RSNA4WayHandshakeFailures);
2256         if (!os_snprintf_error(buflen - len, ret))
2257                 len += ret;
2258
2259         return (int) len;
2260 }
2261 #endif /* CONFIG_CTRL_IFACE */
2262
2263
2264 static void wpa_sm_pmksa_free_cb(struct rsn_pmksa_cache_entry *entry,
2265                                  void *ctx, enum pmksa_free_reason reason)
2266 {
2267         struct wpa_sm *sm = ctx;
2268         int deauth = 0;
2269
2270         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "RSN: PMKSA cache entry free_cb: "
2271                 MACSTR " reason=%d", MAC2STR(entry->aa), reason);
2272
2273         if (sm->cur_pmksa == entry) {
2274                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
2275                         "RSN: %s current PMKSA entry",
2276                         reason == PMKSA_REPLACE ? "replaced" : "removed");
2277                 pmksa_cache_clear_current(sm);
2278
2279                 /*
2280                  * If an entry is simply being replaced, there's no need to
2281                  * deauthenticate because it will be immediately re-added.
2282                  * This happens when EAP authentication is completed again
2283                  * (reauth or failed PMKSA caching attempt).
2284                  */
2285                 if (reason != PMKSA_REPLACE)
2286                         deauth = 1;
2287         }
2288
2289         if (reason == PMKSA_EXPIRE &&
2290             (sm->pmk_len == entry->pmk_len &&
2291              os_memcmp(sm->pmk, entry->pmk, sm->pmk_len) == 0)) {
2292                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
2293                         "RSN: deauthenticating due to expired PMK");
2294                 pmksa_cache_clear_current(sm);
2295                 deauth = 1;
2296         }
2297
2298         if (deauth) {
2299                 os_memset(sm->pmk, 0, sizeof(sm->pmk));
2300                 wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
2301         }
2302 }
2303
2304
2305 /**
2306  * wpa_sm_init - Initialize WPA state machine
2307  * @ctx: Context pointer for callbacks; this needs to be an allocated buffer
2308  * Returns: Pointer to the allocated WPA state machine data
2309  *
2310  * This function is used to allocate a new WPA state machine and the returned
2311  * value is passed to all WPA state machine calls.
2312  */
2313 struct wpa_sm * wpa_sm_init(struct wpa_sm_ctx *ctx)
2314 {
2315         struct wpa_sm *sm;
2316
2317         sm = os_zalloc(sizeof(*sm));
2318         if (sm == NULL)
2319                 return NULL;
2320         dl_list_init(&sm->pmksa_candidates);
2321         sm->renew_snonce = 1;
2322         sm->ctx = ctx;
2323
2324         sm->dot11RSNAConfigPMKLifetime = 43200;
2325         sm->dot11RSNAConfigPMKReauthThreshold = 70;
2326         sm->dot11RSNAConfigSATimeout = 60;
2327
2328         sm->pmksa = pmksa_cache_init(wpa_sm_pmksa_free_cb, sm, sm);
2329         if (sm->pmksa == NULL) {
2330                 wpa_msg(sm->ctx->msg_ctx, MSG_ERROR,
2331                         "RSN: PMKSA cache initialization failed");
2332                 os_free(sm);
2333                 return NULL;
2334         }
2335
2336         return sm;
2337 }
2338
2339
2340 /**
2341  * wpa_sm_deinit - Deinitialize WPA state machine
2342  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2343  */
2344 void wpa_sm_deinit(struct wpa_sm *sm)
2345 {
2346         if (sm == NULL)
2347                 return;
2348         pmksa_cache_deinit(sm->pmksa);
2349         eloop_cancel_timeout(wpa_sm_start_preauth, sm, NULL);
2350         eloop_cancel_timeout(wpa_sm_rekey_ptk, sm, NULL);
2351         os_free(sm->assoc_wpa_ie);
2352         os_free(sm->ap_wpa_ie);
2353         os_free(sm->ap_rsn_ie);
2354         wpa_sm_drop_sa(sm);
2355         os_free(sm->ctx);
2356         peerkey_deinit(sm);
2357 #ifdef CONFIG_IEEE80211R
2358         os_free(sm->assoc_resp_ies);
2359 #endif /* CONFIG_IEEE80211R */
2360 #ifdef CONFIG_TESTING_OPTIONS
2361         wpabuf_free(sm->test_assoc_ie);
2362 #endif /* CONFIG_TESTING_OPTIONS */
2363         os_free(sm);
2364 }
2365
2366
2367 /**
2368  * wpa_sm_notify_assoc - Notify WPA state machine about association
2369  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2370  * @bssid: The BSSID of the new association
2371  *
2372  * This function is called to let WPA state machine know that the connection
2373  * was established.
2374  */
2375 void wpa_sm_notify_assoc(struct wpa_sm *sm, const u8 *bssid)
2376 {
2377         int clear_keys = 1;
2378
2379         if (sm == NULL)
2380                 return;
2381
2382         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
2383                 "WPA: Association event - clear replay counter");
2384         os_memcpy(sm->bssid, bssid, ETH_ALEN);
2385         os_memset(sm->rx_replay_counter, 0, WPA_REPLAY_COUNTER_LEN);
2386         sm->rx_replay_counter_set = 0;
2387         sm->renew_snonce = 1;
2388         if (os_memcmp(sm->preauth_bssid, bssid, ETH_ALEN) == 0)
2389                 rsn_preauth_deinit(sm);
2390
2391 #ifdef CONFIG_IEEE80211R
2392         if (wpa_ft_is_completed(sm)) {
2393                 /*
2394                  * Clear portValid to kick EAPOL state machine to re-enter
2395                  * AUTHENTICATED state to get the EAPOL port Authorized.
2396                  */
2397                 eapol_sm_notify_portValid(sm->eapol, FALSE);
2398                 wpa_supplicant_key_neg_complete(sm, sm->bssid, 1);
2399
2400                 /* Prepare for the next transition */
2401                 wpa_ft_prepare_auth_request(sm, NULL);
2402
2403                 clear_keys = 0;
2404         }
2405 #endif /* CONFIG_IEEE80211R */
2406
2407         if (clear_keys) {
2408                 /*
2409                  * IEEE 802.11, 8.4.10: Delete PTK SA on (re)association if
2410                  * this is not part of a Fast BSS Transition.
2411                  */
2412                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Clear old PTK");
2413                 sm->ptk_set = 0;
2414                 os_memset(&sm->ptk, 0, sizeof(sm->ptk));
2415                 sm->tptk_set = 0;
2416                 os_memset(&sm->tptk, 0, sizeof(sm->tptk));
2417                 os_memset(&sm->gtk, 0, sizeof(sm->gtk));
2418                 os_memset(&sm->gtk_wnm_sleep, 0, sizeof(sm->gtk_wnm_sleep));
2419 #ifdef CONFIG_IEEE80211W
2420                 os_memset(&sm->igtk, 0, sizeof(sm->igtk));
2421                 os_memset(&sm->igtk_wnm_sleep, 0, sizeof(sm->igtk_wnm_sleep));
2422 #endif /* CONFIG_IEEE80211W */
2423         }
2424
2425 #ifdef CONFIG_TDLS
2426         wpa_tdls_assoc(sm);
2427 #endif /* CONFIG_TDLS */
2428
2429 #ifdef CONFIG_P2P
2430         os_memset(sm->p2p_ip_addr, 0, sizeof(sm->p2p_ip_addr));
2431 #endif /* CONFIG_P2P */
2432 }
2433
2434
2435 /**
2436  * wpa_sm_notify_disassoc - Notify WPA state machine about disassociation
2437  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2438  *
2439  * This function is called to let WPA state machine know that the connection
2440  * was lost. This will abort any existing pre-authentication session.
2441  */
2442 void wpa_sm_notify_disassoc(struct wpa_sm *sm)
2443 {
2444         eloop_cancel_timeout(wpa_sm_start_preauth, sm, NULL);
2445         eloop_cancel_timeout(wpa_sm_rekey_ptk, sm, NULL);
2446         peerkey_deinit(sm);
2447         rsn_preauth_deinit(sm);
2448         pmksa_cache_clear_current(sm);
2449         if (wpa_sm_get_state(sm) == WPA_4WAY_HANDSHAKE)
2450                 sm->dot11RSNA4WayHandshakeFailures++;
2451 #ifdef CONFIG_TDLS
2452         wpa_tdls_disassoc(sm);
2453 #endif /* CONFIG_TDLS */
2454 #ifdef CONFIG_IEEE80211R
2455         sm->ft_reassoc_completed = 0;
2456 #endif /* CONFIG_IEEE80211R */
2457
2458         /* Keys are not needed in the WPA state machine anymore */
2459         wpa_sm_drop_sa(sm);
2460
2461         sm->msg_3_of_4_ok = 0;
2462 }
2463
2464
2465 /**
2466  * wpa_sm_set_pmk - Set PMK
2467  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2468  * @pmk: The new PMK
2469  * @pmk_len: The length of the new PMK in bytes
2470  * @pmkid: Calculated PMKID
2471  * @bssid: AA to add into PMKSA cache or %NULL to not cache the PMK
2472  *
2473  * Configure the PMK for WPA state machine.
2474  */
2475 void wpa_sm_set_pmk(struct wpa_sm *sm, const u8 *pmk, size_t pmk_len,
2476                     const u8 *pmkid, const u8 *bssid)
2477 {
2478         if (sm == NULL)
2479                 return;
2480
2481         sm->pmk_len = pmk_len;
2482         os_memcpy(sm->pmk, pmk, pmk_len);
2483
2484 #ifdef CONFIG_IEEE80211R
2485         /* Set XXKey to be PSK for FT key derivation */
2486         sm->xxkey_len = pmk_len;
2487         os_memcpy(sm->xxkey, pmk, pmk_len);
2488 #endif /* CONFIG_IEEE80211R */
2489
2490         if (bssid) {
2491                 pmksa_cache_add(sm->pmksa, pmk, pmk_len, pmkid, NULL, 0,
2492                                 bssid, sm->own_addr,
2493                                 sm->network_ctx, sm->key_mgmt);
2494         }
2495 }
2496
2497
2498 /**
2499  * wpa_sm_set_pmk_from_pmksa - Set PMK based on the current PMKSA
2500  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2501  *
2502  * Take the PMK from the current PMKSA into use. If no PMKSA is active, the PMK
2503  * will be cleared.
2504  */
2505 void wpa_sm_set_pmk_from_pmksa(struct wpa_sm *sm)
2506 {
2507         if (sm == NULL)
2508                 return;
2509
2510         if (sm->cur_pmksa) {
2511                 sm->pmk_len = sm->cur_pmksa->pmk_len;
2512                 os_memcpy(sm->pmk, sm->cur_pmksa->pmk, sm->pmk_len);
2513         } else {
2514                 sm->pmk_len = PMK_LEN;
2515                 os_memset(sm->pmk, 0, PMK_LEN);
2516         }
2517 }
2518
2519
2520 /**
2521  * wpa_sm_set_fast_reauth - Set fast reauthentication (EAP) enabled/disabled
2522  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2523  * @fast_reauth: Whether fast reauthentication (EAP) is allowed
2524  */
2525 void wpa_sm_set_fast_reauth(struct wpa_sm *sm, int fast_reauth)
2526 {
2527         if (sm)
2528                 sm->fast_reauth = fast_reauth;
2529 }
2530
2531
2532 /**
2533  * wpa_sm_set_scard_ctx - Set context pointer for smartcard callbacks
2534  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2535  * @scard_ctx: Context pointer for smartcard related callback functions
2536  */
2537 void wpa_sm_set_scard_ctx(struct wpa_sm *sm, void *scard_ctx)
2538 {
2539         if (sm == NULL)
2540                 return;
2541         sm->scard_ctx = scard_ctx;
2542         if (sm->preauth_eapol)
2543                 eapol_sm_register_scard_ctx(sm->preauth_eapol, scard_ctx);
2544 }
2545
2546
2547 /**
2548  * wpa_sm_set_config - Notification of current configration change
2549  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2550  * @config: Pointer to current network configuration
2551  *
2552  * Notify WPA state machine that configuration has changed. config will be
2553  * stored as a backpointer to network configuration. This can be %NULL to clear
2554  * the stored pointed.
2555  */
2556 void wpa_sm_set_config(struct wpa_sm *sm, struct rsn_supp_config *config)
2557 {
2558         if (!sm)
2559                 return;
2560
2561         if (config) {
2562                 sm->network_ctx = config->network_ctx;
2563                 sm->peerkey_enabled = config->peerkey_enabled;
2564                 sm->allowed_pairwise_cipher = config->allowed_pairwise_cipher;
2565                 sm->proactive_key_caching = config->proactive_key_caching;
2566                 sm->eap_workaround = config->eap_workaround;
2567                 sm->eap_conf_ctx = config->eap_conf_ctx;
2568                 if (config->ssid) {
2569                         os_memcpy(sm->ssid, config->ssid, config->ssid_len);
2570                         sm->ssid_len = config->ssid_len;
2571                 } else
2572                         sm->ssid_len = 0;
2573                 sm->wpa_ptk_rekey = config->wpa_ptk_rekey;
2574                 sm->p2p = config->p2p;
2575                 sm->wpa_rsc_relaxation = config->wpa_rsc_relaxation;
2576         } else {
2577                 sm->network_ctx = NULL;
2578                 sm->peerkey_enabled = 0;
2579                 sm->allowed_pairwise_cipher = 0;
2580                 sm->proactive_key_caching = 0;
2581                 sm->eap_workaround = 0;
2582                 sm->eap_conf_ctx = NULL;
2583                 sm->ssid_len = 0;
2584                 sm->wpa_ptk_rekey = 0;
2585                 sm->p2p = 0;
2586                 sm->wpa_rsc_relaxation = 0;
2587         }
2588 }
2589
2590
2591 /**
2592  * wpa_sm_set_own_addr - Set own MAC address
2593  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2594  * @addr: Own MAC address
2595  */
2596 void wpa_sm_set_own_addr(struct wpa_sm *sm, const u8 *addr)
2597 {
2598         if (sm)
2599                 os_memcpy(sm->own_addr, addr, ETH_ALEN);
2600 }
2601
2602
2603 /**
2604  * wpa_sm_set_ifname - Set network interface name
2605  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2606  * @ifname: Interface name
2607  * @bridge_ifname: Optional bridge interface name (for pre-auth)
2608  */
2609 void wpa_sm_set_ifname(struct wpa_sm *sm, const char *ifname,
2610                        const char *bridge_ifname)
2611 {
2612         if (sm) {
2613                 sm->ifname = ifname;
2614                 sm->bridge_ifname = bridge_ifname;
2615         }
2616 }
2617
2618
2619 /**
2620  * wpa_sm_set_eapol - Set EAPOL state machine pointer
2621  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2622  * @eapol: Pointer to EAPOL state machine allocated with eapol_sm_init()
2623  */
2624 void wpa_sm_set_eapol(struct wpa_sm *sm, struct eapol_sm *eapol)
2625 {
2626         if (sm)
2627                 sm->eapol = eapol;
2628 }
2629
2630
2631 /**
2632  * wpa_sm_set_param - Set WPA state machine parameters
2633  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2634  * @param: Parameter field
2635  * @value: Parameter value
2636  * Returns: 0 on success, -1 on failure
2637  */
2638 int wpa_sm_set_param(struct wpa_sm *sm, enum wpa_sm_conf_params param,
2639                      unsigned int value)
2640 {
2641         int ret = 0;
2642
2643         if (sm == NULL)
2644                 return -1;
2645
2646         switch (param) {
2647         case RSNA_PMK_LIFETIME:
2648                 if (value > 0)
2649                         sm->dot11RSNAConfigPMKLifetime = value;
2650                 else
2651                         ret = -1;
2652                 break;
2653         case RSNA_PMK_REAUTH_THRESHOLD:
2654                 if (value > 0 && value <= 100)
2655                         sm->dot11RSNAConfigPMKReauthThreshold = value;
2656                 else
2657                         ret = -1;
2658                 break;
2659         case RSNA_SA_TIMEOUT:
2660                 if (value > 0)
2661                         sm->dot11RSNAConfigSATimeout = value;
2662                 else
2663                         ret = -1;
2664                 break;
2665         case WPA_PARAM_PROTO:
2666                 sm->proto = value;
2667                 break;
2668         case WPA_PARAM_PAIRWISE:
2669                 sm->pairwise_cipher = value;
2670                 break;
2671         case WPA_PARAM_GROUP:
2672                 sm->group_cipher = value;
2673                 break;
2674         case WPA_PARAM_KEY_MGMT:
2675                 sm->key_mgmt = value;
2676                 break;
2677 #ifdef CONFIG_IEEE80211W
2678         case WPA_PARAM_MGMT_GROUP:
2679                 sm->mgmt_group_cipher = value;
2680                 break;
2681 #endif /* CONFIG_IEEE80211W */
2682         case WPA_PARAM_RSN_ENABLED:
2683                 sm->rsn_enabled = value;
2684                 break;
2685         case WPA_PARAM_MFP:
2686                 sm->mfp = value;
2687                 break;
2688         default:
2689                 break;
2690         }
2691
2692         return ret;
2693 }
2694
2695
2696 /**
2697  * wpa_sm_get_status - Get WPA state machine
2698  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2699  * @buf: Buffer for status information
2700  * @buflen: Maximum buffer length
2701  * @verbose: Whether to include verbose status information
2702  * Returns: Number of bytes written to buf.
2703  *
2704  * Query WPA state machine for status information. This function fills in
2705  * a text area with current status information. If the buffer (buf) is not
2706  * large enough, status information will be truncated to fit the buffer.
2707  */
2708 int wpa_sm_get_status(struct wpa_sm *sm, char *buf, size_t buflen,
2709                       int verbose)
2710 {
2711         char *pos = buf, *end = buf + buflen;
2712         int ret;
2713
2714         ret = os_snprintf(pos, end - pos,
2715                           "pairwise_cipher=%s\n"
2716                           "group_cipher=%s\n"
2717                           "key_mgmt=%s\n",
2718                           wpa_cipher_txt(sm->pairwise_cipher),
2719                           wpa_cipher_txt(sm->group_cipher),
2720                           wpa_key_mgmt_txt(sm->key_mgmt, sm->proto));
2721         if (os_snprintf_error(end - pos, ret))
2722                 return pos - buf;
2723         pos += ret;
2724
2725         if (sm->mfp != NO_MGMT_FRAME_PROTECTION && sm->ap_rsn_ie) {
2726                 struct wpa_ie_data rsn;
2727                 if (wpa_parse_wpa_ie_rsn(sm->ap_rsn_ie, sm->ap_rsn_ie_len, &rsn)
2728                     >= 0 &&
2729                     rsn.capabilities & (WPA_CAPABILITY_MFPR |
2730                                         WPA_CAPABILITY_MFPC)) {
2731                         ret = os_snprintf(pos, end - pos, "pmf=%d\n",
2732                                           (rsn.capabilities &
2733                                            WPA_CAPABILITY_MFPR) ? 2 : 1);
2734                         if (os_snprintf_error(end - pos, ret))
2735                                 return pos - buf;
2736                         pos += ret;
2737                 }
2738         }
2739
2740         return pos - buf;
2741 }
2742
2743
2744 int wpa_sm_pmf_enabled(struct wpa_sm *sm)
2745 {
2746         struct wpa_ie_data rsn;
2747
2748         if (sm->mfp == NO_MGMT_FRAME_PROTECTION || !sm->ap_rsn_ie)
2749                 return 0;
2750
2751         if (wpa_parse_wpa_ie_rsn(sm->ap_rsn_ie, sm->ap_rsn_ie_len, &rsn) >= 0 &&
2752             rsn.capabilities & (WPA_CAPABILITY_MFPR | WPA_CAPABILITY_MFPC))
2753                 return 1;
2754
2755         return 0;
2756 }
2757
2758
2759 /**
2760  * wpa_sm_set_assoc_wpa_ie_default - Generate own WPA/RSN IE from configuration
2761  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2762  * @wpa_ie: Pointer to buffer for WPA/RSN IE
2763  * @wpa_ie_len: Pointer to the length of the wpa_ie buffer
2764  * Returns: 0 on success, -1 on failure
2765  */
2766 int wpa_sm_set_assoc_wpa_ie_default(struct wpa_sm *sm, u8 *wpa_ie,
2767                                     size_t *wpa_ie_len)
2768 {
2769         int res;
2770
2771         if (sm == NULL)
2772                 return -1;
2773
2774 #ifdef CONFIG_TESTING_OPTIONS
2775         if (sm->test_assoc_ie) {
2776                 wpa_printf(MSG_DEBUG,
2777                            "TESTING: Replace association WPA/RSN IE");
2778                 if (*wpa_ie_len < wpabuf_len(sm->test_assoc_ie))
2779                         return -1;
2780                 os_memcpy(wpa_ie, wpabuf_head(sm->test_assoc_ie),
2781                           wpabuf_len(sm->test_assoc_ie));
2782                 res = wpabuf_len(sm->test_assoc_ie);
2783         } else
2784 #endif /* CONFIG_TESTING_OPTIONS */
2785         res = wpa_gen_wpa_ie(sm, wpa_ie, *wpa_ie_len);
2786         if (res < 0)
2787                 return -1;
2788         *wpa_ie_len = res;
2789
2790         wpa_hexdump(MSG_DEBUG, "WPA: Set own WPA IE default",
2791                     wpa_ie, *wpa_ie_len);
2792
2793         if (sm->assoc_wpa_ie == NULL) {
2794                 /*
2795                  * Make a copy of the WPA/RSN IE so that 4-Way Handshake gets
2796                  * the correct version of the IE even if PMKSA caching is
2797                  * aborted (which would remove PMKID from IE generation).
2798                  */
2799                 sm->assoc_wpa_ie = os_malloc(*wpa_ie_len);
2800                 if (sm->assoc_wpa_ie == NULL)
2801                         return -1;
2802
2803                 os_memcpy(sm->assoc_wpa_ie, wpa_ie, *wpa_ie_len);
2804                 sm->assoc_wpa_ie_len = *wpa_ie_len;
2805         }
2806
2807         return 0;
2808 }
2809
2810
2811 /**
2812  * wpa_sm_set_assoc_wpa_ie - Set own WPA/RSN IE from (Re)AssocReq
2813  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2814  * @ie: Pointer to IE data (starting from id)
2815  * @len: IE length
2816  * Returns: 0 on success, -1 on failure
2817  *
2818  * Inform WPA state machine about the WPA/RSN IE used in (Re)Association
2819  * Request frame. The IE will be used to override the default value generated
2820  * with wpa_sm_set_assoc_wpa_ie_default().
2821  */
2822 int wpa_sm_set_assoc_wpa_ie(struct wpa_sm *sm, const u8 *ie, size_t len)
2823 {
2824         if (sm == NULL)
2825                 return -1;
2826
2827         os_free(sm->assoc_wpa_ie);
2828         if (ie == NULL || len == 0) {
2829                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
2830                         "WPA: clearing own WPA/RSN IE");
2831                 sm->assoc_wpa_ie = NULL;
2832                 sm->assoc_wpa_ie_len = 0;
2833         } else {
2834                 wpa_hexdump(MSG_DEBUG, "WPA: set own WPA/RSN IE", ie, len);
2835                 sm->assoc_wpa_ie = os_malloc(len);
2836                 if (sm->assoc_wpa_ie == NULL)
2837                         return -1;
2838
2839                 os_memcpy(sm->assoc_wpa_ie, ie, len);
2840                 sm->assoc_wpa_ie_len = len;
2841         }
2842
2843         return 0;
2844 }
2845
2846
2847 /**
2848  * wpa_sm_set_ap_wpa_ie - Set AP WPA IE from Beacon/ProbeResp
2849  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2850  * @ie: Pointer to IE data (starting from id)
2851  * @len: IE length
2852  * Returns: 0 on success, -1 on failure
2853  *
2854  * Inform WPA state machine about the WPA IE used in Beacon / Probe Response
2855  * frame.
2856  */
2857 int wpa_sm_set_ap_wpa_ie(struct wpa_sm *sm, const u8 *ie, size_t len)
2858 {
2859         if (sm == NULL)
2860                 return -1;
2861
2862         os_free(sm->ap_wpa_ie);
2863         if (ie == NULL || len == 0) {
2864                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
2865                         "WPA: clearing AP WPA IE");
2866                 sm->ap_wpa_ie = NULL;
2867                 sm->ap_wpa_ie_len = 0;
2868         } else {
2869                 wpa_hexdump(MSG_DEBUG, "WPA: set AP WPA IE", ie, len);
2870                 sm->ap_wpa_ie = os_malloc(len);
2871                 if (sm->ap_wpa_ie == NULL)
2872                         return -1;
2873
2874                 os_memcpy(sm->ap_wpa_ie, ie, len);
2875                 sm->ap_wpa_ie_len = len;
2876         }
2877
2878         return 0;
2879 }
2880
2881
2882 /**
2883  * wpa_sm_set_ap_rsn_ie - Set AP RSN IE from Beacon/ProbeResp
2884  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2885  * @ie: Pointer to IE data (starting from id)
2886  * @len: IE length
2887  * Returns: 0 on success, -1 on failure
2888  *
2889  * Inform WPA state machine about the RSN IE used in Beacon / Probe Response
2890  * frame.
2891  */
2892 int wpa_sm_set_ap_rsn_ie(struct wpa_sm *sm, const u8 *ie, size_t len)
2893 {
2894         if (sm == NULL)
2895                 return -1;
2896
2897         os_free(sm->ap_rsn_ie);
2898         if (ie == NULL || len == 0) {
2899                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
2900                         "WPA: clearing AP RSN IE");
2901                 sm->ap_rsn_ie = NULL;
2902                 sm->ap_rsn_ie_len = 0;
2903         } else {
2904                 wpa_hexdump(MSG_DEBUG, "WPA: set AP RSN IE", ie, len);
2905                 sm->ap_rsn_ie = os_malloc(len);
2906                 if (sm->ap_rsn_ie == NULL)
2907                         return -1;
2908
2909                 os_memcpy(sm->ap_rsn_ie, ie, len);
2910                 sm->ap_rsn_ie_len = len;
2911         }
2912
2913         return 0;
2914 }
2915
2916
2917 /**
2918  * wpa_sm_parse_own_wpa_ie - Parse own WPA/RSN IE
2919  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2920  * @data: Pointer to data area for parsing results
2921  * Returns: 0 on success, -1 if IE is not known, or -2 on parsing failure
2922  *
2923  * Parse the contents of the own WPA or RSN IE from (Re)AssocReq and write the
2924  * parsed data into data.
2925  */
2926 int wpa_sm_parse_own_wpa_ie(struct wpa_sm *sm, struct wpa_ie_data *data)
2927 {
2928         if (sm == NULL)
2929                 return -1;
2930
2931         if (sm->assoc_wpa_ie == NULL) {
2932                 wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
2933                         "WPA: No WPA/RSN IE available from association info");
2934                 return -1;
2935         }
2936         if (wpa_parse_wpa_ie(sm->assoc_wpa_ie, sm->assoc_wpa_ie_len, data))
2937                 return -2;
2938         return 0;
2939 }
2940
2941
2942 int wpa_sm_pmksa_cache_list(struct wpa_sm *sm, char *buf, size_t len)
2943 {
2944         return pmksa_cache_list(sm->pmksa, buf, len);
2945 }
2946
2947
2948 void wpa_sm_drop_sa(struct wpa_sm *sm)
2949 {
2950         wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Clear old PMK and PTK");
2951         sm->ptk_set = 0;
2952         sm->tptk_set = 0;
2953         os_memset(sm->pmk, 0, sizeof(sm->pmk));
2954         os_memset(&sm->ptk, 0, sizeof(sm->ptk));
2955         os_memset(&sm->tptk, 0, sizeof(sm->tptk));
2956         os_memset(&sm->gtk, 0, sizeof(sm->gtk));
2957         os_memset(&sm->gtk_wnm_sleep, 0, sizeof(sm->gtk_wnm_sleep));
2958 #ifdef CONFIG_IEEE80211W
2959         os_memset(&sm->igtk, 0, sizeof(sm->igtk));
2960         os_memset(&sm->igtk_wnm_sleep, 0, sizeof(sm->igtk_wnm_sleep));
2961 #endif /* CONFIG_IEEE80211W */
2962 #ifdef CONFIG_IEEE80211R
2963         os_memset(sm->xxkey, 0, sizeof(sm->xxkey));
2964         os_memset(sm->pmk_r0, 0, sizeof(sm->pmk_r0));
2965         os_memset(sm->pmk_r1, 0, sizeof(sm->pmk_r1));
2966 #endif /* CONFIG_IEEE80211R */
2967 }
2968
2969
2970 int wpa_sm_has_ptk(struct wpa_sm *sm)
2971 {
2972         if (sm == NULL)
2973                 return 0;
2974         return sm->ptk_set;
2975 }
2976
2977
2978 void wpa_sm_update_replay_ctr(struct wpa_sm *sm, const u8 *replay_ctr)
2979 {
2980         os_memcpy(sm->rx_replay_counter, replay_ctr, WPA_REPLAY_COUNTER_LEN);
2981 }
2982
2983
2984 void wpa_sm_pmksa_cache_flush(struct wpa_sm *sm, void *network_ctx)
2985 {
2986         pmksa_cache_flush(sm->pmksa, network_ctx, NULL, 0);
2987 }
2988
2989
2990 #ifdef CONFIG_WNM
2991 int wpa_wnmsleep_install_key(struct wpa_sm *sm, u8 subelem_id, u8 *buf)
2992 {
2993         u16 keyinfo;
2994         u8 keylen;  /* plaintext key len */
2995         u8 *key_rsc;
2996
2997         if (subelem_id == WNM_SLEEP_SUBELEM_GTK) {
2998                 struct wpa_gtk_data gd;
2999
3000                 os_memset(&gd, 0, sizeof(gd));
3001                 keylen = wpa_cipher_key_len(sm->group_cipher);
3002                 gd.key_rsc_len = wpa_cipher_rsc_len(sm->group_cipher);
3003                 gd.alg = wpa_cipher_to_alg(sm->group_cipher);
3004                 if (gd.alg == WPA_ALG_NONE) {
3005                         wpa_printf(MSG_DEBUG, "Unsupported group cipher suite");
3006                         return -1;
3007                 }
3008
3009                 key_rsc = buf + 5;
3010                 keyinfo = WPA_GET_LE16(buf + 2);
3011                 gd.gtk_len = keylen;
3012                 if (gd.gtk_len != buf[4]) {
3013                         wpa_printf(MSG_DEBUG, "GTK len mismatch len %d vs %d",
3014                                    gd.gtk_len, buf[4]);
3015                         return -1;
3016                 }
3017                 gd.keyidx = keyinfo & 0x03; /* B0 - B1 */
3018                 gd.tx = wpa_supplicant_gtk_tx_bit_workaround(
3019                          sm, !!(keyinfo & WPA_KEY_INFO_TXRX));
3020
3021                 os_memcpy(gd.gtk, buf + 13, gd.gtk_len);
3022
3023                 wpa_hexdump_key(MSG_DEBUG, "Install GTK (WNM SLEEP)",
3024                                 gd.gtk, gd.gtk_len);
3025                 if (wpa_supplicant_install_gtk(sm, &gd, key_rsc, 1)) {
3026                         os_memset(&gd, 0, sizeof(gd));
3027                         wpa_printf(MSG_DEBUG, "Failed to install the GTK in "
3028                                    "WNM mode");
3029                         return -1;
3030                 }
3031                 os_memset(&gd, 0, sizeof(gd));
3032 #ifdef CONFIG_IEEE80211W
3033         } else if (subelem_id == WNM_SLEEP_SUBELEM_IGTK) {
3034                 const struct wpa_igtk_kde *igtk;
3035
3036                 igtk = (const struct wpa_igtk_kde *) (buf + 2);
3037                 if (wpa_supplicant_install_igtk(sm, igtk, 1) < 0)
3038                         return -1;
3039 #endif /* CONFIG_IEEE80211W */
3040         } else {
3041                 wpa_printf(MSG_DEBUG, "Unknown element id");
3042                 return -1;
3043         }
3044
3045         return 0;
3046 }
3047 #endif /* CONFIG_WNM */
3048
3049
3050 #ifdef CONFIG_PEERKEY
3051 int wpa_sm_rx_eapol_peerkey(struct wpa_sm *sm, const u8 *src_addr,
3052                             const u8 *buf, size_t len)
3053 {
3054         struct wpa_peerkey *peerkey;
3055
3056         for (peerkey = sm->peerkey; peerkey; peerkey = peerkey->next) {
3057                 if (os_memcmp(peerkey->addr, src_addr, ETH_ALEN) == 0)
3058                         break;
3059         }
3060
3061         if (!peerkey)
3062                 return 0;
3063
3064         wpa_sm_rx_eapol(sm, src_addr, buf, len);
3065
3066         return 1;
3067 }
3068 #endif /* CONFIG_PEERKEY */
3069
3070
3071 #ifdef CONFIG_P2P
3072
3073 int wpa_sm_get_p2p_ip_addr(struct wpa_sm *sm, u8 *buf)
3074 {
3075         if (sm == NULL || WPA_GET_BE32(sm->p2p_ip_addr) == 0)
3076                 return -1;
3077         os_memcpy(buf, sm->p2p_ip_addr, 3 * 4);
3078         return 0;
3079 }
3080
3081 #endif /* CONFIG_P2P */
3082
3083
3084 void wpa_sm_set_rx_replay_ctr(struct wpa_sm *sm, const u8 *rx_replay_counter)
3085 {
3086         if (rx_replay_counter == NULL)
3087                 return;
3088
3089         os_memcpy(sm->rx_replay_counter, rx_replay_counter,
3090                   WPA_REPLAY_COUNTER_LEN);
3091         sm->rx_replay_counter_set = 1;
3092         wpa_printf(MSG_DEBUG, "Updated key replay counter");
3093 }
3094
3095
3096 void wpa_sm_set_ptk_kck_kek(struct wpa_sm *sm,
3097                             const u8 *ptk_kck, size_t ptk_kck_len,
3098                             const u8 *ptk_kek, size_t ptk_kek_len)
3099 {
3100         if (ptk_kck && ptk_kck_len <= WPA_KCK_MAX_LEN) {
3101                 os_memcpy(sm->ptk.kck, ptk_kck, ptk_kck_len);
3102                 sm->ptk.kck_len = ptk_kck_len;
3103                 wpa_printf(MSG_DEBUG, "Updated PTK KCK");
3104         }
3105         if (ptk_kek && ptk_kek_len <= WPA_KEK_MAX_LEN) {
3106                 os_memcpy(sm->ptk.kek, ptk_kek, ptk_kek_len);
3107                 sm->ptk.kek_len = ptk_kek_len;
3108                 wpa_printf(MSG_DEBUG, "Updated PTK KEK");
3109         }
3110         sm->ptk_set = 1;
3111 }
3112
3113
3114 #ifdef CONFIG_TESTING_OPTIONS
3115 void wpa_sm_set_test_assoc_ie(struct wpa_sm *sm, struct wpabuf *buf)
3116 {
3117         wpabuf_free(sm->test_assoc_ie);
3118         sm->test_assoc_ie = buf;
3119 }
3120 #endif /* CONFIG_TESTING_OPTIONS */