Suite B: PMKID derivation for AKM 00-0F-AC:11
[mech_eap.git] / src / rsn_supp / wpa.c
index 7c0ac87..9c840c6 100644 (file)
@@ -1,15 +1,9 @@
 /*
  * WPA Supplicant - WPA state machine and EAPOL-Key processing
- * Copyright (c) 2003-2010, Jouni Malinen <j@w1.fi>
+ * Copyright (c) 2003-2012, Jouni Malinen <j@w1.fi>
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License version 2 as
- * published by the Free Software Foundation.
- *
- * Alternatively, this software may be distributed under the terms of BSD
- * license.
- *
- * See README and COPYING for more details.
+ * This software may be distributed under the terms of the BSD license.
+ * See README for more details.
  */
 
 #include "includes.h"
@@ -95,9 +89,12 @@ void wpa_sm_key_request(struct wpa_sm *sm, int error, int pairwise)
        int key_info, ver;
        u8 bssid[ETH_ALEN], *rbuf;
 
-       if (wpa_key_mgmt_ft(sm->key_mgmt) || wpa_key_mgmt_sha256(sm->key_mgmt))
+       if (sm->key_mgmt == WPA_KEY_MGMT_OSEN)
+               ver = WPA_KEY_INFO_TYPE_AKM_DEFINED;
+       else if (wpa_key_mgmt_ft(sm->key_mgmt) ||
+                wpa_key_mgmt_sha256(sm->key_mgmt))
                ver = WPA_KEY_INFO_TYPE_AES_128_CMAC;
-       else if (sm->pairwise_cipher == WPA_CIPHER_CCMP)
+       else if (sm->pairwise_cipher != WPA_CIPHER_TKIP)
                ver = WPA_KEY_INFO_TYPE_HMAC_SHA1_AES;
        else
                ver = WPA_KEY_INFO_TYPE_HMAC_MD5_RC4;
@@ -113,7 +110,8 @@ void wpa_sm_key_request(struct wpa_sm *sm, int error, int pairwise)
        if (rbuf == NULL)
                return;
 
-       reply->type = sm->proto == WPA_PROTO_RSN ?
+       reply->type = (sm->proto == WPA_PROTO_RSN ||
+                      sm->proto == WPA_PROTO_OSEN) ?
                EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
        key_info = WPA_KEY_INFO_REQUEST | ver;
        if (sm->ptk_set)
@@ -140,6 +138,24 @@ void wpa_sm_key_request(struct wpa_sm *sm, int error, int pairwise)
 }
 
 
+static void wpa_supplicant_key_mgmt_set_pmk(struct wpa_sm *sm)
+{
+#ifdef CONFIG_IEEE80211R
+       if (sm->key_mgmt == WPA_KEY_MGMT_FT_IEEE8021X) {
+               if (wpa_sm_key_mgmt_set_pmk(sm, sm->xxkey, sm->xxkey_len))
+                       wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
+                               "RSN: Cannot set low order 256 bits of MSK for key management offload");
+       } else {
+#endif /* CONFIG_IEEE80211R */
+               if (wpa_sm_key_mgmt_set_pmk(sm, sm->pmk, sm->pmk_len))
+                       wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
+                               "RSN: Cannot set PMK for key management offload");
+#ifdef CONFIG_IEEE80211R
+       }
+#endif /* CONFIG_IEEE80211R */
+}
+
+
 static int wpa_supplicant_get_pmk(struct wpa_sm *sm,
                                  const unsigned char *src_addr,
                                  const u8 *pmkid)
@@ -151,7 +167,8 @@ static int wpa_supplicant_get_pmk(struct wpa_sm *sm,
                 * not have enough time to get the association information
                 * event before receiving this 1/4 message, so try to find a
                 * matching PMKSA cache entry here. */
-               sm->cur_pmksa = pmksa_cache_get(sm->pmksa, src_addr, pmkid);
+               sm->cur_pmksa = pmksa_cache_get(sm->pmksa, src_addr, pmkid,
+                                               NULL);
                if (sm->cur_pmksa) {
                        wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
                                "RSN: found matching PMKID from PMKSA cache");
@@ -163,7 +180,7 @@ static int wpa_supplicant_get_pmk(struct wpa_sm *sm,
        }
 
        if (pmkid && sm->cur_pmksa &&
-           os_memcmp(pmkid, sm->cur_pmksa->pmkid, PMKID_LEN) == 0) {
+           os_memcmp_const(pmkid, sm->cur_pmksa->pmkid, PMKID_LEN) == 0) {
                wpa_hexdump(MSG_DEBUG, "RSN: matched PMKID", pmkid, PMKID_LEN);
                wpa_sm_set_pmk_from_pmksa(sm);
                wpa_hexdump_key(MSG_DEBUG, "RSN: PMK from PMKSA cache",
@@ -195,22 +212,32 @@ static int wpa_supplicant_get_pmk(struct wpa_sm *sm,
 #endif /* CONFIG_IEEE80211R */
                }
                if (res == 0) {
+                       struct rsn_pmksa_cache_entry *sa = NULL;
                        wpa_hexdump_key(MSG_DEBUG, "WPA: PMK from EAPOL state "
                                        "machines", sm->pmk, pmk_len);
                        sm->pmk_len = pmk_len;
+                       wpa_supplicant_key_mgmt_set_pmk(sm);
                        if (sm->proto == WPA_PROTO_RSN &&
+                           !wpa_key_mgmt_suite_b(sm->key_mgmt) &&
                            !wpa_key_mgmt_ft(sm->key_mgmt)) {
-                               pmksa_cache_add(sm->pmksa, sm->pmk, pmk_len,
-                                               src_addr, sm->own_addr,
-                                               sm->network_ctx, sm->key_mgmt);
+                               sa = pmksa_cache_add(sm->pmksa,
+                                                    sm->pmk, pmk_len,
+                                                    NULL, 0,
+                                                    src_addr, sm->own_addr,
+                                                    sm->network_ctx,
+                                                    sm->key_mgmt);
                        }
                        if (!sm->cur_pmksa && pmkid &&
-                           pmksa_cache_get(sm->pmksa, src_addr, pmkid)) {
+                           pmksa_cache_get(sm->pmksa, src_addr, pmkid, NULL))
+                       {
                                wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
                                        "RSN: the new PMK matches with the "
                                        "PMKID");
                                abort_cached = 0;
                        }
+
+                       if (!sm->cur_pmksa)
+                               sm->cur_pmksa = sa;
                } else {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: Failed to get master session key from "
@@ -229,7 +256,9 @@ static int wpa_supplicant_get_pmk(struct wpa_sm *sm,
        }
 
        if (abort_cached && wpa_key_mgmt_wpa_ieee8021x(sm->key_mgmt) &&
-           !wpa_key_mgmt_ft(sm->key_mgmt)) {
+           !wpa_key_mgmt_suite_b(sm->key_mgmt) &&
+           !wpa_key_mgmt_ft(sm->key_mgmt) && sm->key_mgmt != WPA_KEY_MGMT_OSEN)
+       {
                /* Send EAPOL-Start to trigger full EAP authentication. */
                u8 *buf;
                size_t buflen;
@@ -323,11 +352,12 @@ int wpa_supplicant_send_2_of_4(struct wpa_sm *sm, const unsigned char *dst,
                return -1;
        }
 
-       reply->type = sm->proto == WPA_PROTO_RSN ?
+       reply->type = (sm->proto == WPA_PROTO_RSN ||
+                      sm->proto == WPA_PROTO_OSEN) ?
                EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
        WPA_PUT_BE16(reply->key_info,
                     ver | WPA_KEY_INFO_KEY_TYPE | WPA_KEY_INFO_MIC);
-       if (sm->proto == WPA_PROTO_RSN)
+       if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN)
                WPA_PUT_BE16(reply->key_length, 0);
        else
                os_memcpy(reply->key_length, key->key_length, 2);
@@ -354,7 +384,7 @@ static int wpa_derive_ptk(struct wpa_sm *sm, const unsigned char *src_addr,
                          const struct wpa_eapol_key *key,
                          struct wpa_ptk *ptk)
 {
-       size_t ptk_len = sm->pairwise_cipher == WPA_CIPHER_CCMP ? 48 : 64;
+       size_t ptk_len = wpa_cipher_key_len(sm->pairwise_cipher) + 32;
 #ifdef CONFIG_IEEE80211R
        if (wpa_key_mgmt_ft(sm->key_mgmt))
                return wpa_derive_ptk_ft(sm, src_addr, key, ptk, ptk_len);
@@ -371,12 +401,14 @@ static int wpa_derive_ptk(struct wpa_sm *sm, const unsigned char *src_addr,
 static void wpa_supplicant_process_1_of_4(struct wpa_sm *sm,
                                          const unsigned char *src_addr,
                                          const struct wpa_eapol_key *key,
-                                         u16 ver)
+                                         u16 ver, const u8 *key_data,
+                                         size_t key_data_len)
 {
        struct wpa_eapol_ie_parse ie;
        struct wpa_ptk *ptk;
-       u8 buf[8];
        int res;
+       u8 *kde, *kde_buf = NULL;
+       size_t kde_len;
 
        if (wpa_sm_get_network_ctx(sm) == NULL) {
                wpa_msg(sm->ctx->msg_ctx, MSG_WARNING, "WPA: No SSID info "
@@ -390,19 +422,17 @@ static void wpa_supplicant_process_1_of_4(struct wpa_sm *sm,
 
        os_memset(&ie, 0, sizeof(ie));
 
-#ifndef CONFIG_NO_WPA2
-       if (sm->proto == WPA_PROTO_RSN) {
+       if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN) {
                /* RSN: msg 1/4 should contain PMKID for the selected PMK */
-               const u8 *_buf = (const u8 *) (key + 1);
-               size_t len = WPA_GET_BE16(key->key_data_length);
-               wpa_hexdump(MSG_DEBUG, "RSN: msg 1/4 key data", _buf, len);
-               wpa_supplicant_parse_ies(_buf, len, &ie);
+               wpa_hexdump(MSG_DEBUG, "RSN: msg 1/4 key data",
+                           key_data, key_data_len);
+               if (wpa_supplicant_parse_ies(key_data, key_data_len, &ie) < 0)
+                       goto failed;
                if (ie.pmkid) {
                        wpa_hexdump(MSG_DEBUG, "RSN: PMKID from "
                                    "Authenticator", ie.pmkid, PMKID_LEN);
                }
        }
-#endif /* CONFIG_NO_WPA2 */
 
        res = wpa_supplicant_get_pmk(sm, src_addr, ie.pmkid);
        if (res == -2) {
@@ -428,21 +458,49 @@ static void wpa_supplicant_process_1_of_4(struct wpa_sm *sm,
         * been verified when processing message 3/4. */
        ptk = &sm->tptk;
        wpa_derive_ptk(sm, src_addr, key, ptk);
-       /* Supplicant: swap tx/rx Mic keys */
-       os_memcpy(buf, ptk->u.auth.tx_mic_key, 8);
-       os_memcpy(ptk->u.auth.tx_mic_key, ptk->u.auth.rx_mic_key, 8);
-       os_memcpy(ptk->u.auth.rx_mic_key, buf, 8);
+       if (sm->pairwise_cipher == WPA_CIPHER_TKIP) {
+               u8 buf[8];
+               /* Supplicant: swap tx/rx Mic keys */
+               os_memcpy(buf, ptk->u.auth.tx_mic_key, 8);
+               os_memcpy(ptk->u.auth.tx_mic_key, ptk->u.auth.rx_mic_key, 8);
+               os_memcpy(ptk->u.auth.rx_mic_key, buf, 8);
+               os_memset(buf, 0, sizeof(buf));
+       }
        sm->tptk_set = 1;
 
+       kde = sm->assoc_wpa_ie;
+       kde_len = sm->assoc_wpa_ie_len;
+
+#ifdef CONFIG_P2P
+       if (sm->p2p) {
+               kde_buf = os_malloc(kde_len + 2 + RSN_SELECTOR_LEN + 1);
+               if (kde_buf) {
+                       u8 *pos;
+                       wpa_printf(MSG_DEBUG, "P2P: Add IP Address Request KDE "
+                                  "into EAPOL-Key 2/4");
+                       os_memcpy(kde_buf, kde, kde_len);
+                       kde = kde_buf;
+                       pos = kde + kde_len;
+                       *pos++ = WLAN_EID_VENDOR_SPECIFIC;
+                       *pos++ = RSN_SELECTOR_LEN + 1;
+                       RSN_SELECTOR_PUT(pos, WFA_KEY_DATA_IP_ADDR_REQ);
+                       pos += RSN_SELECTOR_LEN;
+                       *pos++ = 0x01;
+                       kde_len = pos - kde;
+               }
+       }
+#endif /* CONFIG_P2P */
+
        if (wpa_supplicant_send_2_of_4(sm, sm->bssid, key, ver, sm->snonce,
-                                      sm->assoc_wpa_ie, sm->assoc_wpa_ie_len,
-                                      ptk))
+                                      kde, kde_len, ptk))
                goto failed;
 
+       os_free(kde_buf);
        os_memcpy(sm->anonce, key->key_nonce, WPA_NONCE_LEN);
        return;
 
 failed:
+       os_free(kde_buf);
        wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
 }
 
@@ -476,7 +534,7 @@ static void wpa_supplicant_key_neg_complete(struct wpa_sm *sm,
                 * Start preauthentication after a short wait to avoid a
                 * possible race condition between the data receive and key
                 * configuration after the 4-Way Handshake. This increases the
-                * likelyhood of the first preauth EAPOL-Start frame getting to
+                * likelihood of the first preauth EAPOL-Start frame getting to
                 * the target AP.
                 */
                eloop_register_timeout(1, 0, wpa_sm_start_preauth, sm, NULL);
@@ -517,29 +575,24 @@ static int wpa_supplicant_install_ptk(struct wpa_sm *sm,
        wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
                "WPA: Installing PTK to the driver");
 
-       switch (sm->pairwise_cipher) {
-       case WPA_CIPHER_CCMP:
-               alg = WPA_ALG_CCMP;
-               keylen = 16;
-               rsclen = 6;
-               break;
-       case WPA_CIPHER_TKIP:
-               alg = WPA_ALG_TKIP;
-               keylen = 32;
-               rsclen = 6;
-               break;
-       case WPA_CIPHER_NONE:
+       if (sm->pairwise_cipher == WPA_CIPHER_NONE) {
                wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Pairwise Cipher "
                        "Suite: NONE - do not use pairwise keys");
                return 0;
-       default:
+       }
+
+       if (!wpa_cipher_valid_pairwise(sm->pairwise_cipher)) {
                wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                        "WPA: Unsupported pairwise cipher %d",
                        sm->pairwise_cipher);
                return -1;
        }
 
-       if (sm->proto == WPA_PROTO_RSN) {
+       alg = wpa_cipher_to_alg(sm->pairwise_cipher);
+       keylen = wpa_cipher_key_len(sm->pairwise_cipher);
+       rsclen = wpa_cipher_rsc_len(sm->pairwise_cipher);
+
+       if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN) {
                key_rsc = null_rsc;
        } else {
                key_rsc = key->key_rsc;
@@ -571,55 +624,25 @@ static int wpa_supplicant_check_group_cipher(struct wpa_sm *sm,
                                             int *key_rsc_len,
                                             enum wpa_alg *alg)
 {
-       int ret = 0;
+       int klen;
 
-       switch (group_cipher) {
-       case WPA_CIPHER_CCMP:
-               if (keylen != 16 || maxkeylen < 16) {
-                       ret = -1;
-                       break;
-               }
-               *key_rsc_len = 6;
-               *alg = WPA_ALG_CCMP;
-               break;
-       case WPA_CIPHER_TKIP:
-               if (keylen != 32 || maxkeylen < 32) {
-                       ret = -1;
-                       break;
-               }
-               *key_rsc_len = 6;
-               *alg = WPA_ALG_TKIP;
-               break;
-       case WPA_CIPHER_WEP104:
-               if (keylen != 13 || maxkeylen < 13) {
-                       ret = -1;
-                       break;
-               }
-               *key_rsc_len = 0;
-               *alg = WPA_ALG_WEP;
-               break;
-       case WPA_CIPHER_WEP40:
-               if (keylen != 5 || maxkeylen < 5) {
-                       ret = -1;
-                       break;
-               }
-               *key_rsc_len = 0;
-               *alg = WPA_ALG_WEP;
-               break;
-       default:
+       *alg = wpa_cipher_to_alg(group_cipher);
+       if (*alg == WPA_ALG_NONE) {
                wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                        "WPA: Unsupported Group Cipher %d",
                        group_cipher);
                return -1;
        }
+       *key_rsc_len = wpa_cipher_rsc_len(group_cipher);
 
-       if (ret < 0 ) {
+       klen = wpa_cipher_key_len(group_cipher);
+       if (keylen != klen || maxkeylen < klen) {
                wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                        "WPA: Unsupported %s Group Cipher key length %d (%d)",
                        wpa_cipher_txt(group_cipher), keylen, maxkeylen);
+               return -1;
        }
-
-       return ret;
+       return 0;
 }
 
 
@@ -657,6 +680,7 @@ static int wpa_supplicant_install_gtk(struct wpa_sm *sm,
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: Failed to set GTK to the driver "
                                "(Group only)");
+                       os_memset(gtk_buf, 0, sizeof(gtk_buf));
                        return -1;
                }
        } else if (wpa_sm_set_key(sm, gd->alg, broadcast_ether_addr,
@@ -666,8 +690,10 @@ static int wpa_supplicant_install_gtk(struct wpa_sm *sm,
                        "WPA: Failed to set GTK to "
                        "the driver (alg=%d keylen=%d keyidx=%d)",
                        gd->alg, gd->gtk_len, gd->keyidx);
+               os_memset(gtk_buf, 0, sizeof(gtk_buf));
                return -1;
        }
+       os_memset(gtk_buf, 0, sizeof(gtk_buf));
 
        return 0;
 }
@@ -696,7 +722,6 @@ static int wpa_supplicant_pairwise_gtk(struct wpa_sm *sm,
                                       const u8 *gtk, size_t gtk_len,
                                       int key_info)
 {
-#ifndef CONFIG_NO_WPA2
        struct wpa_gtk_data gd;
 
        /*
@@ -723,21 +748,21 @@ static int wpa_supplicant_pairwise_gtk(struct wpa_sm *sm,
        os_memcpy(gd.gtk, gtk, gtk_len);
        gd.gtk_len = gtk_len;
 
-       if (wpa_supplicant_check_group_cipher(sm, sm->group_cipher,
-                                             gtk_len, gtk_len,
-                                             &gd.key_rsc_len, &gd.alg) ||
-           wpa_supplicant_install_gtk(sm, &gd, key->key_rsc)) {
+       if (sm->group_cipher != WPA_CIPHER_GTK_NOT_USED &&
+           (wpa_supplicant_check_group_cipher(sm, sm->group_cipher,
+                                              gtk_len, gtk_len,
+                                              &gd.key_rsc_len, &gd.alg) ||
+            wpa_supplicant_install_gtk(sm, &gd, key->key_rsc))) {
                wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
                        "RSN: Failed to install GTK");
+               os_memset(&gd, 0, sizeof(gd));
                return -1;
        }
+       os_memset(&gd, 0, sizeof(gd));
 
        wpa_supplicant_key_neg_complete(sm, sm->bssid,
                                        key_info & WPA_KEY_INFO_SECURE);
        return 0;
-#else /* CONFIG_NO_WPA2 */
-       return -1;
-#endif /* CONFIG_NO_WPA2 */
 }
 
 
@@ -745,13 +770,15 @@ static int ieee80211w_set_keys(struct wpa_sm *sm,
                               struct wpa_eapol_ie_parse *ie)
 {
 #ifdef CONFIG_IEEE80211W
-       if (sm->mgmt_group_cipher != WPA_CIPHER_AES_128_CMAC)
+       if (!wpa_cipher_valid_mgmt_group(sm->mgmt_group_cipher))
                return 0;
 
        if (ie->igtk) {
+               size_t len;
                const struct wpa_igtk_kde *igtk;
                u16 keyidx;
-               if (ie->igtk_len != sizeof(*igtk))
+               len = wpa_cipher_key_len(sm->mgmt_group_cipher);
+               if (ie->igtk_len != WPA_IGTK_KDE_PREFIX_LEN + len)
                        return -1;
                igtk = (const struct wpa_igtk_kde *) ie->igtk;
                keyidx = WPA_GET_LE16(igtk->keyid);
@@ -759,15 +786,16 @@ static int ieee80211w_set_keys(struct wpa_sm *sm,
                        "pn %02x%02x%02x%02x%02x%02x",
                        keyidx, MAC2STR(igtk->pn));
                wpa_hexdump_key(MSG_DEBUG, "WPA: IGTK",
-                               igtk->igtk, WPA_IGTK_LEN);
+                               igtk->igtk, len);
                if (keyidx > 4095) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: Invalid IGTK KeyID %d", keyidx);
                        return -1;
                }
-               if (wpa_sm_set_key(sm, WPA_ALG_IGTK, broadcast_ether_addr,
+               if (wpa_sm_set_key(sm, wpa_cipher_to_alg(sm->mgmt_group_cipher),
+                                  broadcast_ether_addr,
                                   keyidx, 0, igtk->pn, sizeof(igtk->pn),
-                                  igtk->igtk, WPA_IGTK_LEN) < 0) {
+                                  igtk->igtk, len) < 0) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: Failed to configure IGTK to the driver");
                        return -1;
@@ -815,7 +843,7 @@ static void wpa_report_ie_mismatch(struct wpa_sm *sm,
                            rsn_ie, rsn_ie_len);
        }
 
-       wpa_sm_disassociate(sm, WLAN_REASON_IE_IN_4WAY_DIFFERS);
+       wpa_sm_deauthenticate(sm, WLAN_REASON_IE_IN_4WAY_DIFFERS);
 }
 
 
@@ -900,7 +928,8 @@ static int ft_validate_rsnie(struct wpa_sm *sm,
                return -1;
        }
 
-       if (os_memcmp(rsn.pmkid, sm->pmk_r1_name, WPA_PMK_NAME_LEN) != 0) {
+       if (os_memcmp_const(rsn.pmkid, sm->pmk_r1_name, WPA_PMK_NAME_LEN) != 0)
+       {
                wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
                        "FT: PMKR1Name mismatch in "
                        "FT 4-way handshake message 3/4");
@@ -1020,45 +1049,37 @@ static int wpa_supplicant_validate_ie(struct wpa_sm *sm,
  * @key: Pointer to the EAPOL-Key frame header
  * @ver: Version bits from EAPOL-Key Key Info
  * @key_info: Key Info
- * @kde: KDEs to include the EAPOL-Key frame
- * @kde_len: Length of KDEs
  * @ptk: PTK to use for keyed hash and encryption
  * Returns: 0 on success, -1 on failure
  */
 int wpa_supplicant_send_4_of_4(struct wpa_sm *sm, const unsigned char *dst,
                               const struct wpa_eapol_key *key,
                               u16 ver, u16 key_info,
-                              const u8 *kde, size_t kde_len,
                               struct wpa_ptk *ptk)
 {
        size_t rlen;
        struct wpa_eapol_key *reply;
        u8 *rbuf;
 
-       if (kde)
-               wpa_hexdump(MSG_DEBUG, "WPA: KDE for msg 4/4", kde, kde_len);
-
        rbuf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_KEY, NULL,
-                                 sizeof(*reply) + kde_len,
-                                 &rlen, (void *) &reply);
+                                 sizeof(*reply), &rlen, (void *) &reply);
        if (rbuf == NULL)
                return -1;
 
-       reply->type = sm->proto == WPA_PROTO_RSN ?
+       reply->type = (sm->proto == WPA_PROTO_RSN ||
+                      sm->proto == WPA_PROTO_OSEN) ?
                EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
        key_info &= WPA_KEY_INFO_SECURE;
        key_info |= ver | WPA_KEY_INFO_KEY_TYPE | WPA_KEY_INFO_MIC;
        WPA_PUT_BE16(reply->key_info, key_info);
-       if (sm->proto == WPA_PROTO_RSN)
+       if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN)
                WPA_PUT_BE16(reply->key_length, 0);
        else
                os_memcpy(reply->key_length, key->key_length, 2);
        os_memcpy(reply->replay_counter, key->replay_counter,
                  WPA_REPLAY_COUNTER_LEN);
 
-       WPA_PUT_BE16(reply->key_data_length, kde_len);
-       if (kde)
-               os_memcpy(reply + 1, kde, kde_len);
+       WPA_PUT_BE16(reply->key_data_length, 0);
 
        wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Sending EAPOL-Key 4/4");
        wpa_eapol_key_send(sm, ptk->kck, ver, dst, ETH_P_EAPOL,
@@ -1070,10 +1091,10 @@ int wpa_supplicant_send_4_of_4(struct wpa_sm *sm, const unsigned char *dst,
 
 static void wpa_supplicant_process_3_of_4(struct wpa_sm *sm,
                                          const struct wpa_eapol_key *key,
-                                         u16 ver)
+                                         u16 ver, const u8 *key_data,
+                                         size_t key_data_len)
 {
-       u16 key_info, keylen, len;
-       const u8 *pos;
+       u16 key_info, keylen;
        struct wpa_eapol_ie_parse ie;
 
        wpa_sm_set_state(sm, WPA_4WAY_HANDSHAKE);
@@ -1082,10 +1103,9 @@ static void wpa_supplicant_process_3_of_4(struct wpa_sm *sm,
 
        key_info = WPA_GET_BE16(key->key_info);
 
-       pos = (const u8 *) (key + 1);
-       len = WPA_GET_BE16(key->key_data_length);
-       wpa_hexdump(MSG_DEBUG, "WPA: IE KeyData", pos, len);
-       wpa_supplicant_parse_ies(pos, len, &ie);
+       wpa_hexdump(MSG_DEBUG, "WPA: IE KeyData", key_data, key_data_len);
+       if (wpa_supplicant_parse_ies(key_data, key_data_len, &ie) < 0)
+               goto failed;
        if (ie.gtk && !(key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
                wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                        "WPA: GTK IE in unencrypted key data");
@@ -1098,7 +1118,10 @@ static void wpa_supplicant_process_3_of_4(struct wpa_sm *sm,
                goto failed;
        }
 
-       if (ie.igtk && ie.igtk_len != sizeof(struct wpa_igtk_kde)) {
+       if (ie.igtk &&
+           wpa_cipher_valid_mgmt_group(sm->mgmt_group_cipher) &&
+           ie.igtk_len != WPA_IGTK_KDE_PREFIX_LEN +
+           (unsigned int) wpa_cipher_key_len(sm->mgmt_group_cipher)) {
                wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                        "WPA: Invalid IGTK KDE length %lu",
                        (unsigned long) ie.igtk_len);
@@ -1118,27 +1141,24 @@ static void wpa_supplicant_process_3_of_4(struct wpa_sm *sm,
        }
 
        keylen = WPA_GET_BE16(key->key_length);
-       switch (sm->pairwise_cipher) {
-       case WPA_CIPHER_CCMP:
-               if (keylen != 16) {
-                       wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
-                               "WPA: Invalid CCMP key length %d (src=" MACSTR
-                               ")", keylen, MAC2STR(sm->bssid));
-                       goto failed;
-               }
-               break;
-       case WPA_CIPHER_TKIP:
-               if (keylen != 32) {
-                       wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
-                               "WPA: Invalid TKIP key length %d (src=" MACSTR
-                               ")", keylen, MAC2STR(sm->bssid));
-                       goto failed;
-               }
-               break;
+       if (keylen != wpa_cipher_key_len(sm->pairwise_cipher)) {
+               wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
+                       "WPA: Invalid %s key length %d (src=" MACSTR
+                       ")", wpa_cipher_txt(sm->pairwise_cipher), keylen,
+                       MAC2STR(sm->bssid));
+               goto failed;
+       }
+
+#ifdef CONFIG_P2P
+       if (ie.ip_addr_alloc) {
+               os_memcpy(sm->p2p_ip_addr, ie.ip_addr_alloc, 3 * 4);
+               wpa_hexdump(MSG_DEBUG, "P2P: IP address info",
+                           sm->p2p_ip_addr, sizeof(sm->p2p_ip_addr));
        }
+#endif /* CONFIG_P2P */
 
        if (wpa_supplicant_send_4_of_4(sm, sm->bssid, key, ver, key_info,
-                                      NULL, 0, &sm->ptk)) {
+                                      &sm->ptk)) {
                goto failed;
        }
 
@@ -1160,7 +1180,10 @@ static void wpa_supplicant_process_3_of_4(struct wpa_sm *sm,
        }
        wpa_sm_set_state(sm, WPA_GROUP_HANDSHAKE);
 
-       if (ie.gtk &&
+       if (sm->group_cipher == WPA_CIPHER_GTK_NOT_USED) {
+               wpa_supplicant_key_neg_complete(sm, sm->bssid,
+                                               key_info & WPA_KEY_INFO_SECURE);
+       } else if (ie.gtk &&
            wpa_supplicant_pairwise_gtk(sm, key,
                                        ie.gtk, ie.gtk_len, key_info) < 0) {
                wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
@@ -1174,6 +1197,20 @@ static void wpa_supplicant_process_3_of_4(struct wpa_sm *sm,
                goto failed;
        }
 
+       if (ie.gtk)
+               wpa_sm_set_rekey_offload(sm);
+
+       if (sm->proto == WPA_PROTO_RSN && wpa_key_mgmt_suite_b(sm->key_mgmt)) {
+               struct rsn_pmksa_cache_entry *sa;
+
+               sa = pmksa_cache_add(sm->pmksa, sm->pmk, sm->pmk_len,
+                                    sm->ptk.kck, sizeof(sm->ptk.kck),
+                                    sm->bssid, sm->own_addr,
+                                    sm->network_ctx, sm->key_mgmt);
+               if (!sm->cur_pmksa)
+                       sm->cur_pmksa = sa;
+       }
+
        return;
 
 failed:
@@ -1191,7 +1228,8 @@ static int wpa_supplicant_process_1_of_2_rsn(struct wpa_sm *sm,
        struct wpa_eapol_ie_parse ie;
 
        wpa_hexdump(MSG_DEBUG, "RSN: msg 1/2 key data", keydata, keydatalen);
-       wpa_supplicant_parse_ies(keydata, keydatalen, &ie);
+       if (wpa_supplicant_parse_ies(keydata, keydatalen, &ie) < 0)
+               return -1;
        if (ie.gtk && !(key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
                wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                        "WPA: GTK IE in unencrypted key data");
@@ -1232,22 +1270,14 @@ static int wpa_supplicant_process_1_of_2_rsn(struct wpa_sm *sm,
 
 static int wpa_supplicant_process_1_of_2_wpa(struct wpa_sm *sm,
                                             const struct wpa_eapol_key *key,
-                                            size_t keydatalen, int key_info,
-                                            size_t extra_len, u16 ver,
-                                            struct wpa_gtk_data *gd)
+                                            const u8 *key_data,
+                                            size_t key_data_len, u16 key_info,
+                                            u16 ver, struct wpa_gtk_data *gd)
 {
        size_t maxkeylen;
-       u8 ek[32];
 
        gd->gtk_len = WPA_GET_BE16(key->key_length);
-       maxkeylen = keydatalen;
-       if (keydatalen > extra_len) {
-               wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
-                       "WPA: Truncated EAPOL-Key packet: "
-                       "key_data_length=%lu > extra_len=%lu",
-                       (unsigned long) keydatalen, (unsigned long) extra_len);
-               return -1;
-       }
+       maxkeylen = key_data_len;
        if (ver == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
                if (maxkeylen < 8) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
@@ -1266,37 +1296,40 @@ static int wpa_supplicant_process_1_of_2_wpa(struct wpa_sm *sm,
        gd->keyidx = (key_info & WPA_KEY_INFO_KEY_INDEX_MASK) >>
                WPA_KEY_INFO_KEY_INDEX_SHIFT;
        if (ver == WPA_KEY_INFO_TYPE_HMAC_MD5_RC4) {
-               os_memcpy(ek, key->key_iv, 16);
-               os_memcpy(ek + 16, sm->ptk.kek, 16);
-               if (keydatalen > sizeof(gd->gtk)) {
+               u8 ek[32];
+               if (key_data_len > sizeof(gd->gtk)) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: RC4 key data too long (%lu)",
-                               (unsigned long) keydatalen);
+                               (unsigned long) key_data_len);
                        return -1;
                }
-               os_memcpy(gd->gtk, key + 1, keydatalen);
-               if (rc4_skip(ek, 32, 256, gd->gtk, keydatalen)) {
+               os_memcpy(ek, key->key_iv, 16);
+               os_memcpy(ek + 16, sm->ptk.kek, 16);
+               os_memcpy(gd->gtk, key_data, key_data_len);
+               if (rc4_skip(ek, 32, 256, gd->gtk, key_data_len)) {
+                       os_memset(ek, 0, sizeof(ek));
                        wpa_msg(sm->ctx->msg_ctx, MSG_ERROR,
                                "WPA: RC4 failed");
                        return -1;
                }
+               os_memset(ek, 0, sizeof(ek));
        } else if (ver == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
-               if (keydatalen % 8) {
+               if (maxkeylen % 8) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: Unsupported AES-WRAP len %lu",
-                               (unsigned long) keydatalen);
+                               (unsigned long) maxkeylen);
                        return -1;
                }
                if (maxkeylen > sizeof(gd->gtk)) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: AES-WRAP key data "
                                "too long (keydatalen=%lu maxkeylen=%lu)",
-                               (unsigned long) keydatalen,
+                               (unsigned long) key_data_len,
                                (unsigned long) maxkeylen);
                        return -1;
                }
-               if (aes_unwrap(sm->ptk.kek, maxkeylen / 8,
-                              (const u8 *) (key + 1), gd->gtk)) {
+               if (aes_unwrap(sm->ptk.kek, 16, maxkeylen / 8, key_data,
+                              gd->gtk)) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: AES unwrap failed - could not decrypt "
                                "GTK");
@@ -1326,12 +1359,13 @@ static int wpa_supplicant_send_2_of_2(struct wpa_sm *sm,
        if (rbuf == NULL)
                return -1;
 
-       reply->type = sm->proto == WPA_PROTO_RSN ?
+       reply->type = (sm->proto == WPA_PROTO_RSN ||
+                      sm->proto == WPA_PROTO_OSEN) ?
                EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
        key_info &= WPA_KEY_INFO_KEY_INDEX_MASK;
        key_info |= ver | WPA_KEY_INFO_MIC | WPA_KEY_INFO_SECURE;
        WPA_PUT_BE16(reply->key_info, key_info);
-       if (sm->proto == WPA_PROTO_RSN)
+       if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN)
                WPA_PUT_BE16(reply->key_length, 0);
        else
                os_memcpy(reply->key_length, key->key_length, 2);
@@ -1351,9 +1385,10 @@ static int wpa_supplicant_send_2_of_2(struct wpa_sm *sm,
 static void wpa_supplicant_process_1_of_2(struct wpa_sm *sm,
                                          const unsigned char *src_addr,
                                          const struct wpa_eapol_key *key,
-                                         int extra_len, u16 ver)
+                                         const u8 *key_data,
+                                         size_t key_data_len, u16 ver)
 {
-       u16 key_info, keydatalen;
+       u16 key_info;
        int rekey, ret;
        struct wpa_gtk_data gd;
 
@@ -1364,17 +1399,15 @@ static void wpa_supplicant_process_1_of_2(struct wpa_sm *sm,
                "Handshake from " MACSTR " (ver=%d)", MAC2STR(src_addr), ver);
 
        key_info = WPA_GET_BE16(key->key_info);
-       keydatalen = WPA_GET_BE16(key->key_data_length);
 
-       if (sm->proto == WPA_PROTO_RSN) {
-               ret = wpa_supplicant_process_1_of_2_rsn(sm,
-                                                       (const u8 *) (key + 1),
-                                                       keydatalen, key_info,
+       if (sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN) {
+               ret = wpa_supplicant_process_1_of_2_rsn(sm, key_data,
+                                                       key_data_len, key_info,
                                                        &gd);
        } else {
-               ret = wpa_supplicant_process_1_of_2_wpa(sm, key, keydatalen,
-                                                       key_info, extra_len,
-                                                       ver, &gd);
+               ret = wpa_supplicant_process_1_of_2_wpa(sm, key, key_data,
+                                                       key_data_len,
+                                                       key_info, ver, &gd);
        }
 
        wpa_sm_set_state(sm, WPA_GROUP_HANDSHAKE);
@@ -1397,6 +1430,9 @@ static void wpa_supplicant_process_1_of_2(struct wpa_sm *sm,
                                                key_info &
                                                WPA_KEY_INFO_SECURE);
        }
+
+       wpa_sm_set_rekey_offload(sm);
+
        return;
 
 failed:
@@ -1417,7 +1453,7 @@ static int wpa_supplicant_verify_eapol_key_mic(struct wpa_sm *sm,
                os_memset(key->key_mic, 0, 16);
                wpa_eapol_key_mic(sm->tptk.kck, ver, buf, len,
                                  key->key_mic);
-               if (os_memcmp(mic, key->key_mic, 16) != 0) {
+               if (os_memcmp_const(mic, key->key_mic, 16) != 0) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: Invalid EAPOL-Key MIC "
                                "when using TPTK - ignoring TPTK");
@@ -1426,6 +1462,7 @@ static int wpa_supplicant_verify_eapol_key_mic(struct wpa_sm *sm,
                        sm->tptk_set = 0;
                        sm->ptk_set = 1;
                        os_memcpy(&sm->ptk, &sm->tptk, sizeof(sm->ptk));
+                       os_memset(&sm->tptk, 0, sizeof(sm->tptk));
                }
        }
 
@@ -1433,7 +1470,7 @@ static int wpa_supplicant_verify_eapol_key_mic(struct wpa_sm *sm,
                os_memset(key->key_mic, 0, 16);
                wpa_eapol_key_mic(sm->ptk.kck, ver, buf, len,
                                  key->key_mic);
-               if (os_memcmp(mic, key->key_mic, 16) != 0) {
+               if (os_memcmp_const(mic, key->key_mic, 16) != 0) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: Invalid EAPOL-Key MIC - "
                                "dropping packet");
@@ -1458,12 +1495,11 @@ static int wpa_supplicant_verify_eapol_key_mic(struct wpa_sm *sm,
 
 /* Decrypt RSN EAPOL-Key key data (RC4 or AES-WRAP) */
 static int wpa_supplicant_decrypt_key_data(struct wpa_sm *sm,
-                                          struct wpa_eapol_key *key, u16 ver)
+                                          struct wpa_eapol_key *key, u16 ver,
+                                          u8 *key_data, size_t *key_data_len)
 {
-       u16 keydatalen = WPA_GET_BE16(key->key_data_length);
-
        wpa_hexdump(MSG_DEBUG, "RSN: encrypted key data",
-                   (u8 *) (key + 1), keydatalen);
+                   key_data, *key_data_len);
        if (!sm->ptk_set) {
                wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                        "WPA: PTK not available, cannot decrypt EAPOL-Key Key "
@@ -1477,45 +1513,48 @@ static int wpa_supplicant_decrypt_key_data(struct wpa_sm *sm,
                u8 ek[32];
                os_memcpy(ek, key->key_iv, 16);
                os_memcpy(ek + 16, sm->ptk.kek, 16);
-               if (rc4_skip(ek, 32, 256, (u8 *) (key + 1), keydatalen)) {
+               if (rc4_skip(ek, 32, 256, key_data, *key_data_len)) {
+                       os_memset(ek, 0, sizeof(ek));
                        wpa_msg(sm->ctx->msg_ctx, MSG_ERROR,
                                "WPA: RC4 failed");
                        return -1;
                }
+               os_memset(ek, 0, sizeof(ek));
        } else if (ver == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES ||
-                  ver == WPA_KEY_INFO_TYPE_AES_128_CMAC) {
+                  ver == WPA_KEY_INFO_TYPE_AES_128_CMAC ||
+                  sm->key_mgmt == WPA_KEY_MGMT_OSEN) {
                u8 *buf;
-               if (keydatalen % 8) {
+               if (*key_data_len < 8 || *key_data_len % 8) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
-                               "WPA: Unsupported AES-WRAP len %d",
-                               keydatalen);
+                               "WPA: Unsupported AES-WRAP len %u",
+                               (unsigned int) *key_data_len);
                        return -1;
                }
-               keydatalen -= 8; /* AES-WRAP adds 8 bytes */
-               buf = os_malloc(keydatalen);
+               *key_data_len -= 8; /* AES-WRAP adds 8 bytes */
+               buf = os_malloc(*key_data_len);
                if (buf == NULL) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: No memory for AES-UNWRAP buffer");
                        return -1;
                }
-               if (aes_unwrap(sm->ptk.kek, keydatalen / 8,
-                              (u8 *) (key + 1), buf)) {
+               if (aes_unwrap(sm->ptk.kek, 16, *key_data_len / 8,
+                              key_data, buf)) {
                        os_free(buf);
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: AES unwrap failed - "
                                "could not decrypt EAPOL-Key key data");
                        return -1;
                }
-               os_memcpy(key + 1, buf, keydatalen);
+               os_memcpy(key_data, buf, *key_data_len);
                os_free(buf);
-               WPA_PUT_BE16(key->key_data_length, keydatalen);
+               WPA_PUT_BE16(key->key_data_length, *key_data_len);
        } else {
                wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                        "WPA: Unsupported key_info type %d", ver);
                return -1;
        }
        wpa_hexdump_key(MSG_DEBUG, "WPA: decrypted EAPOL-Key key data",
-                       (u8 *) (key + 1), keydatalen);
+                       key_data, *key_data_len);
        return 0;
 }
 
@@ -1589,13 +1628,14 @@ static void wpa_eapol_key_dump(struct wpa_sm *sm,
 int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
                    const u8 *buf, size_t len)
 {
-       size_t plen, data_len, extra_len;
-       struct ieee802_1x_hdr *hdr;
+       size_t plen, data_len, key_data_len;
+       const struct ieee802_1x_hdr *hdr;
        struct wpa_eapol_key *key;
        u16 key_info, ver;
-       u8 *tmp;
+       u8 *tmp = NULL;
        int ret = -1;
        struct wpa_peerkey *peerkey = NULL;
+       u8 *key_data;
 
 #ifdef CONFIG_IEEE80211R
        sm->ft_completed = 0;
@@ -1610,13 +1650,7 @@ int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
                return 0;
        }
 
-       tmp = os_malloc(len);
-       if (tmp == NULL)
-               return -1;
-       os_memcpy(tmp, buf, len);
-
-       hdr = (struct ieee802_1x_hdr *) tmp;
-       key = (struct wpa_eapol_key *) (hdr + 1);
+       hdr = (const struct ieee802_1x_hdr *) buf;
        plen = be_to_host16(hdr->length);
        data_len = plen + sizeof(*hdr);
        wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
@@ -1633,6 +1667,7 @@ int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
                ret = 0;
                goto out;
        }
+       wpa_hexdump(MSG_MSGDUMP, "WPA: RX EAPOL-Key", buf, len);
        if (plen > len - sizeof(*hdr) || plen < sizeof(*key)) {
                wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
                        "WPA: EAPOL frame payload size %lu "
@@ -1641,6 +1676,22 @@ int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
                ret = 0;
                goto out;
        }
+       if (data_len < len) {
+               wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
+                       "WPA: ignoring %lu bytes after the IEEE 802.1X data",
+                       (unsigned long) len - data_len);
+       }
+
+       /*
+        * Make a copy of the frame since we need to modify the buffer during
+        * MAC validation and Key Data decryption.
+        */
+       tmp = os_malloc(data_len);
+       if (tmp == NULL)
+               goto out;
+       os_memcpy(tmp, buf, data_len);
+       key = (struct wpa_eapol_key *) (tmp + sizeof(struct ieee802_1x_hdr));
+       key_data = (u8 *) (key + 1);
 
        if (key->type != EAPOL_KEY_TYPE_WPA && key->type != EAPOL_KEY_TYPE_RSN)
        {
@@ -1652,26 +1703,38 @@ int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
        }
        wpa_eapol_key_dump(sm, key);
 
-       eapol_sm_notify_lower_layer_success(sm->eapol, 0);
-       wpa_hexdump(MSG_MSGDUMP, "WPA: RX EAPOL-Key", tmp, len);
-       if (data_len < len) {
-               wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
-                       "WPA: ignoring %lu bytes after the IEEE 802.1X data",
-                       (unsigned long) len - data_len);
+       key_data_len = WPA_GET_BE16(key->key_data_length);
+       if (key_data_len > plen - sizeof(struct wpa_eapol_key)) {
+               wpa_msg(sm->ctx->msg_ctx, MSG_INFO, "WPA: Invalid EAPOL-Key "
+                       "frame - key_data overflow (%u > %u)",
+                       (unsigned int) key_data_len,
+                       (unsigned int) (plen - sizeof(struct wpa_eapol_key)));
+               goto out;
        }
+
+       eapol_sm_notify_lower_layer_success(sm->eapol, 0);
        key_info = WPA_GET_BE16(key->key_info);
        ver = key_info & WPA_KEY_INFO_TYPE_MASK;
        if (ver != WPA_KEY_INFO_TYPE_HMAC_MD5_RC4 &&
 #if defined(CONFIG_IEEE80211R) || defined(CONFIG_IEEE80211W)
            ver != WPA_KEY_INFO_TYPE_AES_128_CMAC &&
 #endif /* CONFIG_IEEE80211R || CONFIG_IEEE80211W */
-           ver != WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
+           ver != WPA_KEY_INFO_TYPE_HMAC_SHA1_AES &&
+           sm->key_mgmt != WPA_KEY_MGMT_OSEN) {
                wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
                        "WPA: Unsupported EAPOL-Key descriptor version %d",
                        ver);
                goto out;
        }
 
+       if (sm->key_mgmt == WPA_KEY_MGMT_OSEN &&
+           ver != WPA_KEY_INFO_TYPE_AKM_DEFINED) {
+               wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
+                       "OSEN: Unsupported EAPOL-Key descriptor version %d",
+                       ver);
+               goto out;
+       }
+
 #ifdef CONFIG_IEEE80211R
        if (wpa_key_mgmt_ft(sm->key_mgmt)) {
                /* IEEE 802.11r uses a new key_info type (AES-128-CMAC). */
@@ -1684,7 +1747,8 @@ int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
 #endif /* CONFIG_IEEE80211R */
 #ifdef CONFIG_IEEE80211W
        if (wpa_key_mgmt_sha256(sm->key_mgmt)) {
-               if (ver != WPA_KEY_INFO_TYPE_AES_128_CMAC) {
+               if (ver != WPA_KEY_INFO_TYPE_AES_128_CMAC &&
+                   sm->key_mgmt != WPA_KEY_MGMT_OSEN) {
                        wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
                                "WPA: AP did not use the "
                                "negotiated AES-128-CMAC");
@@ -1706,8 +1770,17 @@ int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
                        wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
                                "WPA: Backwards compatibility: allow invalid "
                                "version for non-CCMP group keys");
+               } else if (ver == WPA_KEY_INFO_TYPE_AES_128_CMAC) {
+                       wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
+                               "WPA: Interoperability workaround: allow incorrect (should have been HMAC-SHA1), but stronger (is AES-128-CMAC), descriptor version to be used");
                } else
                        goto out;
+       } else if (sm->pairwise_cipher == WPA_CIPHER_GCMP &&
+                  ver != WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
+               wpa_msg(sm->ctx->msg_ctx, MSG_INFO,
+                       "WPA: GCMP is used, but EAPOL-Key "
+                       "descriptor version (%d) is not 2", ver);
+               goto out;
        }
 
 #ifdef CONFIG_PEERKEY
@@ -1782,22 +1855,11 @@ int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
                goto out;
 #endif /* CONFIG_PEERKEY */
 
-       extra_len = data_len - sizeof(*hdr) - sizeof(*key);
-
-       if (WPA_GET_BE16(key->key_data_length) > extra_len) {
-               wpa_msg(sm->ctx->msg_ctx, MSG_INFO, "WPA: Invalid EAPOL-Key "
-                       "frame - key_data overflow (%d > %lu)",
-                       WPA_GET_BE16(key->key_data_length),
-                       (unsigned long) extra_len);
-               goto out;
-       }
-       extra_len = WPA_GET_BE16(key->key_data_length);
-
-       if (sm->proto == WPA_PROTO_RSN &&
+       if ((sm->proto == WPA_PROTO_RSN || sm->proto == WPA_PROTO_OSEN) &&
            (key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
-               if (wpa_supplicant_decrypt_key_data(sm, key, ver))
+               if (wpa_supplicant_decrypt_key_data(sm, key, ver, key_data,
+                                                   &key_data_len))
                        goto out;
-               extra_len = WPA_GET_BE16(key->key_data_length);
        }
 
        if (key_info & WPA_KEY_INFO_KEY_TYPE) {
@@ -1809,24 +1871,28 @@ int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
                }
                if (peerkey) {
                        /* PeerKey 4-Way Handshake */
-                       peerkey_rx_eapol_4way(sm, peerkey, key, key_info, ver);
+                       peerkey_rx_eapol_4way(sm, peerkey, key, key_info, ver,
+                                             key_data, key_data_len);
                } else if (key_info & WPA_KEY_INFO_MIC) {
                        /* 3/4 4-Way Handshake */
-                       wpa_supplicant_process_3_of_4(sm, key, ver);
+                       wpa_supplicant_process_3_of_4(sm, key, ver, key_data,
+                                                     key_data_len);
                } else {
                        /* 1/4 4-Way Handshake */
                        wpa_supplicant_process_1_of_4(sm, src_addr, key,
-                                                     ver);
+                                                     ver, key_data,
+                                                     key_data_len);
                }
        } else if (key_info & WPA_KEY_INFO_SMK_MESSAGE) {
                /* PeerKey SMK Handshake */
-               peerkey_rx_eapol_smk(sm, src_addr, key, extra_len, key_info,
+               peerkey_rx_eapol_smk(sm, src_addr, key, key_data_len, key_info,
                                     ver);
        } else {
                if (key_info & WPA_KEY_INFO_MIC) {
                        /* 1/2 Group Key Handshake */
                        wpa_supplicant_process_1_of_2(sm, src_addr, key,
-                                                     extra_len, ver);
+                                                     key_data, key_data_len,
+                                                     ver);
                } else {
                        wpa_msg(sm->ctx->msg_ctx, MSG_WARNING,
                                "WPA: EAPOL-Key (Group) without Mic bit - "
@@ -1843,28 +1909,12 @@ out:
 
 
 #ifdef CONFIG_CTRL_IFACE
-static int wpa_cipher_bits(int cipher)
-{
-       switch (cipher) {
-       case WPA_CIPHER_CCMP:
-               return 128;
-       case WPA_CIPHER_TKIP:
-               return 256;
-       case WPA_CIPHER_WEP104:
-               return 104;
-       case WPA_CIPHER_WEP40:
-               return 40;
-       default:
-               return 0;
-       }
-}
-
-
 static u32 wpa_key_mgmt_suite(struct wpa_sm *sm)
 {
        switch (sm->key_mgmt) {
        case WPA_KEY_MGMT_IEEE8021X:
-               return (sm->proto == WPA_PROTO_RSN ?
+               return ((sm->proto == WPA_PROTO_RSN ||
+                        sm->proto == WPA_PROTO_OSEN) ?
                        RSN_AUTH_KEY_MGMT_UNSPEC_802_1X :
                        WPA_AUTH_KEY_MGMT_UNSPEC_802_1X);
        case WPA_KEY_MGMT_PSK:
@@ -1883,32 +1933,14 @@ static u32 wpa_key_mgmt_suite(struct wpa_sm *sm)
        case WPA_KEY_MGMT_PSK_SHA256:
                return RSN_AUTH_KEY_MGMT_PSK_SHA256;
 #endif /* CONFIG_IEEE80211W */
+       case WPA_KEY_MGMT_CCKM:
+               return (sm->proto == WPA_PROTO_RSN ?
+                       RSN_AUTH_KEY_MGMT_CCKM:
+                       WPA_AUTH_KEY_MGMT_CCKM);
        case WPA_KEY_MGMT_WPA_NONE:
                return WPA_AUTH_KEY_MGMT_NONE;
-       default:
-               return 0;
-       }
-}
-
-
-static u32 wpa_cipher_suite(struct wpa_sm *sm, int cipher)
-{
-       switch (cipher) {
-       case WPA_CIPHER_CCMP:
-               return (sm->proto == WPA_PROTO_RSN ?
-                       RSN_CIPHER_SUITE_CCMP : WPA_CIPHER_SUITE_CCMP);
-       case WPA_CIPHER_TKIP:
-               return (sm->proto == WPA_PROTO_RSN ?
-                       RSN_CIPHER_SUITE_TKIP : WPA_CIPHER_SUITE_TKIP);
-       case WPA_CIPHER_WEP104:
-               return (sm->proto == WPA_PROTO_RSN ?
-                       RSN_CIPHER_SUITE_WEP104 : WPA_CIPHER_SUITE_WEP104);
-       case WPA_CIPHER_WEP40:
-               return (sm->proto == WPA_PROTO_RSN ?
-                       RSN_CIPHER_SUITE_WEP40 : WPA_CIPHER_SUITE_WEP40);
-       case WPA_CIPHER_NONE:
-               return (sm->proto == WPA_PROTO_RSN ?
-                       RSN_CIPHER_SUITE_NONE : WPA_CIPHER_SUITE_NONE);
+       case WPA_KEY_MGMT_IEEE8021X_SUITE_B:
+               return RSN_AUTH_KEY_MGMT_802_1X_SUITE_B;
        default:
                return 0;
        }
@@ -1962,7 +1994,7 @@ int wpa_sm_get_mib(struct wpa_sm *sm, char *buf, size_t buflen)
                          rsna ? "TRUE" : "FALSE",
                          rsna ? "TRUE" : "FALSE",
                          RSN_VERSION,
-                         wpa_cipher_bits(sm->group_cipher),
+                         wpa_cipher_key_len(sm->group_cipher) * 8,
                          sm->dot11RSNAConfigPMKLifetime,
                          sm->dot11RSNAConfigPMKReauthThreshold,
                          sm->dot11RSNAConfigSATimeout);
@@ -1982,12 +2014,16 @@ int wpa_sm_get_mib(struct wpa_sm *sm, char *buf, size_t buflen)
                "dot11RSNAConfigNumberOfGTKSAReplayCounters=0\n"
                "dot11RSNA4WayHandshakeFailures=%u\n",
                RSN_SUITE_ARG(wpa_key_mgmt_suite(sm)),
-               RSN_SUITE_ARG(wpa_cipher_suite(sm, sm->pairwise_cipher)),
-               RSN_SUITE_ARG(wpa_cipher_suite(sm, sm->group_cipher)),
+               RSN_SUITE_ARG(wpa_cipher_to_suite(sm->proto,
+                                                 sm->pairwise_cipher)),
+               RSN_SUITE_ARG(wpa_cipher_to_suite(sm->proto,
+                                                 sm->group_cipher)),
                pmkid_txt,
                RSN_SUITE_ARG(wpa_key_mgmt_suite(sm)),
-               RSN_SUITE_ARG(wpa_cipher_suite(sm, sm->pairwise_cipher)),
-               RSN_SUITE_ARG(wpa_cipher_suite(sm, sm->group_cipher)),
+               RSN_SUITE_ARG(wpa_cipher_to_suite(sm->proto,
+                                                 sm->pairwise_cipher)),
+               RSN_SUITE_ARG(wpa_cipher_to_suite(sm->proto,
+                                                 sm->group_cipher)),
                sm->dot11RSNA4WayHandshakeFailures);
        if (ret >= 0 && (size_t) ret < buflen)
                len += ret;
@@ -1998,25 +2034,40 @@ int wpa_sm_get_mib(struct wpa_sm *sm, char *buf, size_t buflen)
 
 
 static void wpa_sm_pmksa_free_cb(struct rsn_pmksa_cache_entry *entry,
-                                void *ctx, int replace)
+                                void *ctx, enum pmksa_free_reason reason)
 {
        struct wpa_sm *sm = ctx;
+       int deauth = 0;
+
+       wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "RSN: PMKSA cache entry free_cb: "
+               MACSTR " reason=%d", MAC2STR(entry->aa), reason);
+
+       if (sm->cur_pmksa == entry) {
+               wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
+                       "RSN: %s current PMKSA entry",
+                       reason == PMKSA_REPLACE ? "replaced" : "removed");
+               pmksa_cache_clear_current(sm);
 
-       if (sm->cur_pmksa == entry ||
+               /*
+                * If an entry is simply being replaced, there's no need to
+                * deauthenticate because it will be immediately re-added.
+                * This happens when EAP authentication is completed again
+                * (reauth or failed PMKSA caching attempt).
+                */
+               if (reason != PMKSA_REPLACE)
+                       deauth = 1;
+       }
+
+       if (reason == PMKSA_EXPIRE &&
            (sm->pmk_len == entry->pmk_len &&
             os_memcmp(sm->pmk, entry->pmk, sm->pmk_len) == 0)) {
                wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG,
-                       "RSN: removed current PMKSA entry");
-               sm->cur_pmksa = NULL;
-
-               if (replace) {
-                       /* A new entry is being added, so no need to
-                        * deauthenticate in this case. This happens when EAP
-                        * authentication is completed again (reauth or failed
-                        * PMKSA caching attempt). */
-                       return;
-               }
+                       "RSN: deauthenticating due to expired PMK");
+               pmksa_cache_clear_current(sm);
+               deauth = 1;
+       }
 
+       if (deauth) {
                os_memset(sm->pmk, 0, sizeof(sm->pmk));
                wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
        }
@@ -2128,12 +2179,18 @@ void wpa_sm_notify_assoc(struct wpa_sm *sm, const u8 *bssid)
                 */
                wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Clear old PTK");
                sm->ptk_set = 0;
+               os_memset(&sm->ptk, 0, sizeof(sm->ptk));
                sm->tptk_set = 0;
+               os_memset(&sm->tptk, 0, sizeof(sm->tptk));
        }
 
 #ifdef CONFIG_TDLS
        wpa_tdls_assoc(sm);
 #endif /* CONFIG_TDLS */
+
+#ifdef CONFIG_P2P
+       os_memset(sm->p2p_ip_addr, 0, sizeof(sm->p2p_ip_addr));
+#endif /* CONFIG_P2P */
 }
 
 
@@ -2146,7 +2203,9 @@ void wpa_sm_notify_assoc(struct wpa_sm *sm, const u8 *bssid)
  */
 void wpa_sm_notify_disassoc(struct wpa_sm *sm)
 {
+       peerkey_deinit(sm);
        rsn_preauth_deinit(sm);
+       pmksa_cache_clear_current(sm);
        if (wpa_sm_get_state(sm) == WPA_4WAY_HANDSHAKE)
                sm->dot11RSNA4WayHandshakeFailures++;
 #ifdef CONFIG_TDLS
@@ -2160,10 +2219,12 @@ void wpa_sm_notify_disassoc(struct wpa_sm *sm)
  * @sm: Pointer to WPA state machine data from wpa_sm_init()
  * @pmk: The new PMK
  * @pmk_len: The length of the new PMK in bytes
+ * @bssid: AA to add into PMKSA cache or %NULL to not cache the PMK
  *
  * Configure the PMK for WPA state machine.
  */
-void wpa_sm_set_pmk(struct wpa_sm *sm, const u8 *pmk, size_t pmk_len)
+void wpa_sm_set_pmk(struct wpa_sm *sm, const u8 *pmk, size_t pmk_len,
+                   const u8 *bssid)
 {
        if (sm == NULL)
                return;
@@ -2176,6 +2237,12 @@ void wpa_sm_set_pmk(struct wpa_sm *sm, const u8 *pmk, size_t pmk_len)
        sm->xxkey_len = pmk_len;
        os_memcpy(sm->xxkey, pmk, pmk_len);
 #endif /* CONFIG_IEEE80211R */
+
+       if (bssid) {
+               pmksa_cache_add(sm->pmksa, pmk, pmk_len, NULL, 0,
+                               bssid, sm->own_addr,
+                               sm->network_ctx, sm->key_mgmt);
+       }
 }
 
 
@@ -2255,6 +2322,7 @@ void wpa_sm_set_config(struct wpa_sm *sm, struct rsn_supp_config *config)
                } else
                        sm->ssid_len = 0;
                sm->wpa_ptk_rekey = config->wpa_ptk_rekey;
+               sm->p2p = config->p2p;
        } else {
                sm->network_ctx = NULL;
                sm->peerkey_enabled = 0;
@@ -2264,9 +2332,8 @@ void wpa_sm_set_config(struct wpa_sm *sm, struct rsn_supp_config *config)
                sm->eap_conf_ctx = NULL;
                sm->ssid_len = 0;
                sm->wpa_ptk_rekey = 0;
+               sm->p2p = 0;
        }
-       if (config == NULL || config->network_ctx != sm->network_ctx)
-               pmksa_cache_notify_reconfig(sm->pmksa);
 }
 
 
@@ -2376,44 +2443,6 @@ int wpa_sm_set_param(struct wpa_sm *sm, enum wpa_sm_conf_params param,
 
 
 /**
- * wpa_sm_get_param - Get WPA state machine parameters
- * @sm: Pointer to WPA state machine data from wpa_sm_init()
- * @param: Parameter field
- * Returns: Parameter value
- */
-unsigned int wpa_sm_get_param(struct wpa_sm *sm, enum wpa_sm_conf_params param)
-{
-       if (sm == NULL)
-               return 0;
-
-       switch (param) {
-       case RSNA_PMK_LIFETIME:
-               return sm->dot11RSNAConfigPMKLifetime;
-       case RSNA_PMK_REAUTH_THRESHOLD:
-               return sm->dot11RSNAConfigPMKReauthThreshold;
-       case RSNA_SA_TIMEOUT:
-               return sm->dot11RSNAConfigSATimeout;
-       case WPA_PARAM_PROTO:
-               return sm->proto;
-       case WPA_PARAM_PAIRWISE:
-               return sm->pairwise_cipher;
-       case WPA_PARAM_GROUP:
-               return sm->group_cipher;
-       case WPA_PARAM_KEY_MGMT:
-               return sm->key_mgmt;
-#ifdef CONFIG_IEEE80211W
-       case WPA_PARAM_MGMT_GROUP:
-               return sm->mgmt_group_cipher;
-#endif /* CONFIG_IEEE80211W */
-       case WPA_PARAM_RSN_ENABLED:
-               return sm->rsn_enabled;
-       default:
-               return 0;
-       }
-}
-
-
-/**
  * wpa_sm_get_status - Get WPA state machine
  * @sm: Pointer to WPA state machine data from wpa_sm_init()
  * @buf: Buffer for status information
@@ -2441,10 +2470,41 @@ int wpa_sm_get_status(struct wpa_sm *sm, char *buf, size_t buflen,
        if (ret < 0 || ret >= end - pos)
                return pos - buf;
        pos += ret;
+
+       if (sm->mfp != NO_MGMT_FRAME_PROTECTION && sm->ap_rsn_ie) {
+               struct wpa_ie_data rsn;
+               if (wpa_parse_wpa_ie_rsn(sm->ap_rsn_ie, sm->ap_rsn_ie_len, &rsn)
+                   >= 0 &&
+                   rsn.capabilities & (WPA_CAPABILITY_MFPR |
+                                       WPA_CAPABILITY_MFPC)) {
+                       ret = os_snprintf(pos, end - pos, "pmf=%d\n",
+                                         (rsn.capabilities &
+                                          WPA_CAPABILITY_MFPR) ? 2 : 1);
+                       if (ret < 0 || ret >= end - pos)
+                               return pos - buf;
+                       pos += ret;
+               }
+       }
+
        return pos - buf;
 }
 
 
+int wpa_sm_pmf_enabled(struct wpa_sm *sm)
+{
+       struct wpa_ie_data rsn;
+
+       if (sm->mfp == NO_MGMT_FRAME_PROTECTION || !sm->ap_rsn_ie)
+               return 0;
+
+       if (wpa_parse_wpa_ie_rsn(sm->ap_rsn_ie, sm->ap_rsn_ie_len, &rsn) >= 0 &&
+           rsn.capabilities & (WPA_CAPABILITY_MFPR | WPA_CAPABILITY_MFPC))
+               return 1;
+
+       return 0;
+}
+
+
 /**
  * wpa_sm_set_assoc_wpa_ie_default - Generate own WPA/RSN IE from configuration
  * @sm: Pointer to WPA state machine data from wpa_sm_init()
@@ -2619,14 +2679,11 @@ int wpa_sm_parse_own_wpa_ie(struct wpa_sm *sm, struct wpa_ie_data *data)
 
 int wpa_sm_pmksa_cache_list(struct wpa_sm *sm, char *buf, size_t len)
 {
-#ifndef CONFIG_NO_WPA2
        return pmksa_cache_list(sm->pmksa, buf, len);
-#else /* CONFIG_NO_WPA2 */
-       return -1;
-#endif /* CONFIG_NO_WPA2 */
 }
 
 
+#ifdef CONFIG_TESTING_OPTIONS
 void wpa_sm_drop_sa(struct wpa_sm *sm)
 {
        wpa_dbg(sm->ctx->msg_ctx, MSG_DEBUG, "WPA: Clear old PMK and PTK");
@@ -2636,6 +2693,7 @@ void wpa_sm_drop_sa(struct wpa_sm *sm)
        os_memset(&sm->ptk, 0, sizeof(sm->ptk));
        os_memset(&sm->tptk, 0, sizeof(sm->tptk));
 }
+#endif /* CONFIG_TESTING_OPTIONS */
 
 
 int wpa_sm_has_ptk(struct wpa_sm *sm)
@@ -2644,3 +2702,154 @@ int wpa_sm_has_ptk(struct wpa_sm *sm)
                return 0;
        return sm->ptk_set;
 }
+
+
+void wpa_sm_update_replay_ctr(struct wpa_sm *sm, const u8 *replay_ctr)
+{
+       os_memcpy(sm->rx_replay_counter, replay_ctr, WPA_REPLAY_COUNTER_LEN);
+}
+
+
+void wpa_sm_pmksa_cache_flush(struct wpa_sm *sm, void *network_ctx)
+{
+       pmksa_cache_flush(sm->pmksa, network_ctx, NULL, 0);
+}
+
+
+#ifdef CONFIG_WNM
+int wpa_wnmsleep_install_key(struct wpa_sm *sm, u8 subelem_id, u8 *buf)
+{
+       u16 keyinfo;
+       u8 keylen;  /* plaintext key len */
+       u8 *key_rsc;
+
+       if (subelem_id == WNM_SLEEP_SUBELEM_GTK) {
+               struct wpa_gtk_data gd;
+
+               os_memset(&gd, 0, sizeof(gd));
+               keylen = wpa_cipher_key_len(sm->group_cipher);
+               gd.key_rsc_len = wpa_cipher_rsc_len(sm->group_cipher);
+               gd.alg = wpa_cipher_to_alg(sm->group_cipher);
+               if (gd.alg == WPA_ALG_NONE) {
+                       wpa_printf(MSG_DEBUG, "Unsupported group cipher suite");
+                       return -1;
+               }
+
+               key_rsc = buf + 5;
+               keyinfo = WPA_GET_LE16(buf + 2);
+               gd.gtk_len = keylen;
+               if (gd.gtk_len != buf[4]) {
+                       wpa_printf(MSG_DEBUG, "GTK len mismatch len %d vs %d",
+                                  gd.gtk_len, buf[4]);
+                       return -1;
+               }
+               gd.keyidx = keyinfo & 0x03; /* B0 - B1 */
+               gd.tx = wpa_supplicant_gtk_tx_bit_workaround(
+                        sm, !!(keyinfo & WPA_KEY_INFO_TXRX));
+
+               os_memcpy(gd.gtk, buf + 13, gd.gtk_len);
+
+               wpa_hexdump_key(MSG_DEBUG, "Install GTK (WNM SLEEP)",
+                               gd.gtk, gd.gtk_len);
+               if (wpa_supplicant_install_gtk(sm, &gd, key_rsc)) {
+                       os_memset(&gd, 0, sizeof(gd));
+                       wpa_printf(MSG_DEBUG, "Failed to install the GTK in "
+                                  "WNM mode");
+                       return -1;
+               }
+               os_memset(&gd, 0, sizeof(gd));
+#ifdef CONFIG_IEEE80211W
+       } else if (subelem_id == WNM_SLEEP_SUBELEM_IGTK) {
+               struct wpa_igtk_kde igd;
+               u16 keyidx;
+
+               os_memset(&igd, 0, sizeof(igd));
+               keylen = wpa_cipher_key_len(sm->mgmt_group_cipher);
+               os_memcpy(igd.keyid, buf + 2, 2);
+               os_memcpy(igd.pn, buf + 4, 6);
+
+               keyidx = WPA_GET_LE16(igd.keyid);
+               os_memcpy(igd.igtk, buf + 10, keylen);
+
+               wpa_hexdump_key(MSG_DEBUG, "Install IGTK (WNM SLEEP)",
+                               igd.igtk, keylen);
+               if (wpa_sm_set_key(sm, wpa_cipher_to_alg(sm->mgmt_group_cipher),
+                                  broadcast_ether_addr,
+                                  keyidx, 0, igd.pn, sizeof(igd.pn),
+                                  igd.igtk, keylen) < 0) {
+                       wpa_printf(MSG_DEBUG, "Failed to install the IGTK in "
+                                  "WNM mode");
+                       os_memset(&igd, 0, sizeof(igd));
+                       return -1;
+               }
+               os_memset(&igd, 0, sizeof(igd));
+#endif /* CONFIG_IEEE80211W */
+       } else {
+               wpa_printf(MSG_DEBUG, "Unknown element id");
+               return -1;
+       }
+
+       return 0;
+}
+#endif /* CONFIG_WNM */
+
+
+#ifdef CONFIG_PEERKEY
+int wpa_sm_rx_eapol_peerkey(struct wpa_sm *sm, const u8 *src_addr,
+                           const u8 *buf, size_t len)
+{
+       struct wpa_peerkey *peerkey;
+
+       for (peerkey = sm->peerkey; peerkey; peerkey = peerkey->next) {
+               if (os_memcmp(peerkey->addr, src_addr, ETH_ALEN) == 0)
+                       break;
+       }
+
+       if (!peerkey)
+               return 0;
+
+       wpa_sm_rx_eapol(sm, src_addr, buf, len);
+
+       return 1;
+}
+#endif /* CONFIG_PEERKEY */
+
+
+#ifdef CONFIG_P2P
+
+int wpa_sm_get_p2p_ip_addr(struct wpa_sm *sm, u8 *buf)
+{
+       if (sm == NULL || WPA_GET_BE32(sm->p2p_ip_addr) == 0)
+               return -1;
+       os_memcpy(buf, sm->p2p_ip_addr, 3 * 4);
+       return 0;
+}
+
+#endif /* CONFIG_P2P */
+
+
+void wpa_sm_set_rx_replay_ctr(struct wpa_sm *sm, const u8 *rx_replay_counter)
+{
+       if (rx_replay_counter == NULL)
+               return;
+
+       os_memcpy(sm->rx_replay_counter, rx_replay_counter,
+                 WPA_REPLAY_COUNTER_LEN);
+       sm->rx_replay_counter_set = 1;
+       wpa_printf(MSG_DEBUG, "Updated key replay counter");
+}
+
+
+void wpa_sm_set_ptk_kck_kek(struct wpa_sm *sm, const u8 *ptk_kck,
+                           const u8 *ptk_kek)
+{
+       if (ptk_kck) {
+               os_memcpy(sm->ptk.kck, ptk_kck, 16);
+               wpa_printf(MSG_DEBUG, "Updated PTK KCK");
+       }
+       if (ptk_kek) {
+               os_memcpy(sm->ptk.kek, ptk_kek, 16);
+               wpa_printf(MSG_DEBUG, "Updated PTK KEK");
+       }
+       sm->ptk_set = 1;
+}