contrib/hostapd/src/rsn_supp/wpa.c

   1 /*
   2  * WPA Supplicant - WPA state machine and EAPOL-Key processing
   3  * Copyright (c) 2003-2008, Jouni Malinen <j@w1.fi>
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License version 2 as
   7  * published by the Free Software Foundation.
   8  *
   9  * Alternatively, this software may be distributed under the terms of BSD
  10  * license.
  11  *
  12  * See README and COPYING for more details.
  13  */
  14
  15 #include "includes.h"
  16
  17 #include "common.h"
  18 #include "rc4.h"
  19 #include "aes_wrap.h"
  20 #include "wpa.h"
  21 #include "eloop.h"
  22 #include "eapol_supp/eapol_supp_sm.h"
  23 #include "preauth.h"
  24 #include "pmksa_cache.h"
  25 #include "wpa_i.h"
  26 #include "wpa_ie.h"
  27 #include "peerkey.h"
  28 #include "ieee802_11_defs.h"
  29
  30
  31 /**
  32  * wpa_cipher_txt - Convert cipher suite to a text string
  33  * @cipher: Cipher suite (WPA_CIPHER_* enum)
  34  * Returns: Pointer to a text string of the cipher suite name
  35  */
  36 static const char * wpa_cipher_txt(int cipher)
  37 {
  38         switch (cipher) {
  39         case WPA_CIPHER_NONE:
  40                 return "NONE";
  41         case WPA_CIPHER_WEP40:
  42                 return "WEP-40";
  43         case WPA_CIPHER_WEP104:
  44                 return "WEP-104";
  45         case WPA_CIPHER_TKIP:
  46                 return "TKIP";
  47         case WPA_CIPHER_CCMP:
  48                 return "CCMP";
  49         default:
  50                 return "UNKNOWN";
  51         }
  52 }
  53
  54
  55 /**
  56  * wpa_key_mgmt_txt - Convert key management suite to a text string
  57  * @key_mgmt: Key management suite (WPA_KEY_MGMT_* enum)
  58  * @proto: WPA/WPA2 version (WPA_PROTO_*)
  59  * Returns: Pointer to a text string of the key management suite name
  60  */
  61 static const char * wpa_key_mgmt_txt(int key_mgmt, int proto)
  62 {
  63         switch (key_mgmt) {
  64         case WPA_KEY_MGMT_IEEE8021X:
  65                 return proto == WPA_PROTO_RSN ?
  66                         "WPA2/IEEE 802.1X/EAP" : "WPA/IEEE 802.1X/EAP";
  67         case WPA_KEY_MGMT_PSK:
  68                 return proto == WPA_PROTO_RSN ?
  69                         "WPA2-PSK" : "WPA-PSK";
  70         case WPA_KEY_MGMT_NONE:
  71                 return "NONE";
  72         case WPA_KEY_MGMT_IEEE8021X_NO_WPA:
  73                 return "IEEE 802.1X (no WPA)";
  74 #ifdef CONFIG_IEEE80211R
  75         case WPA_KEY_MGMT_FT_IEEE8021X:
  76                 return "FT-EAP";
  77         case WPA_KEY_MGMT_FT_PSK:
  78                 return "FT-PSK";
  79 #endif /* CONFIG_IEEE80211R */
  80 #ifdef CONFIG_IEEE80211W
  81         case WPA_KEY_MGMT_IEEE8021X_SHA256:
  82                 return "WPA2-EAP-SHA256";
  83         case WPA_KEY_MGMT_PSK_SHA256:
  84                 return "WPA2-PSK-SHA256";
  85 #endif /* CONFIG_IEEE80211W */
  86         default:
  87                 return "UNKNOWN";
  88         }
  89 }
  90
  91
  92 /**
  93  * wpa_eapol_key_send - Send WPA/RSN EAPOL-Key message
  94  * @sm: Pointer to WPA state machine data from wpa_sm_init()
  95  * @kck: Key Confirmation Key (KCK, part of PTK)
  96  * @ver: Version field from Key Info
  97  * @dest: Destination address for the frame
  98  * @proto: Ethertype (usually ETH_P_EAPOL)
  99  * @msg: EAPOL-Key message
 100  * @msg_len: Length of message
 101  * @key_mic: Pointer to the buffer to which the EAPOL-Key MIC is written
 102  */
 103 void wpa_eapol_key_send(struct wpa_sm *sm, const u8 *kck,
 104                         int ver, const u8 *dest, u16 proto,
 105                         u8 *msg, size_t msg_len, u8 *key_mic)
 106 {
 107         if (is_zero_ether_addr(dest) && is_zero_ether_addr(sm->bssid)) {
 108                 /*
 109                  * Association event was not yet received; try to fetch
 110                  * BSSID from the driver.
 111                  */
 112                 if (wpa_sm_get_bssid(sm, sm->bssid) < 0) {
 113                         wpa_printf(MSG_DEBUG, "WPA: Failed to read BSSID for "
 114                                    "EAPOL-Key destination address");
 115                 } else {
 116                         dest = sm->bssid;
 117                         wpa_printf(MSG_DEBUG, "WPA: Use BSSID (" MACSTR
 118                                    ") as the destination for EAPOL-Key",
 119                                    MAC2STR(dest));
 120                 }
 121         }
 122         if (key_mic)
 123                 wpa_eapol_key_mic(kck, ver, msg, msg_len, key_mic);
 124         wpa_hexdump(MSG_MSGDUMP, "WPA: TX EAPOL-Key", msg, msg_len);
 125         wpa_sm_ether_send(sm, dest, proto, msg, msg_len);
 126         eapol_sm_notify_tx_eapol_key(sm->eapol);
 127         os_free(msg);
 128 }
 129
 130
 131 /**
 132  * wpa_sm_key_request - Send EAPOL-Key Request
 133  * @sm: Pointer to WPA state machine data from wpa_sm_init()
 134  * @error: Indicate whether this is an Michael MIC error report
 135  * @pairwise: 1 = error report for pairwise packet, 0 = for group packet
 136  *
 137  * Send an EAPOL-Key Request to the current authenticator. This function is
 138  * used to request rekeying and it is usually called when a local Michael MIC
 139  * failure is detected.
 140  */
 141 void wpa_sm_key_request(struct wpa_sm *sm, int error, int pairwise)
 142 {
 143         size_t rlen;
 144         struct wpa_eapol_key *reply;
 145         int key_info, ver;
 146         u8 bssid[ETH_ALEN], *rbuf;
 147
 148         if (wpa_key_mgmt_ft(sm->key_mgmt) || wpa_key_mgmt_sha256(sm->key_mgmt))
 149                 ver = WPA_KEY_INFO_TYPE_AES_128_CMAC;
 150         else if (sm->pairwise_cipher == WPA_CIPHER_CCMP)
 151                 ver = WPA_KEY_INFO_TYPE_HMAC_SHA1_AES;
 152         else
 153                 ver = WPA_KEY_INFO_TYPE_HMAC_MD5_RC4;
 154
 155         if (wpa_sm_get_bssid(sm, bssid) < 0) {
 156                 wpa_printf(MSG_WARNING, "Failed to read BSSID for EAPOL-Key "
 157                            "request");
 158                 return;
 159         }
 160
 161         rbuf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_KEY, NULL,
 162                                   sizeof(*reply), &rlen, (void *) &reply);
 163         if (rbuf == NULL)
 164                 return;
 165
 166         reply->type = sm->proto == WPA_PROTO_RSN ?
 167                 EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
 168         key_info = WPA_KEY_INFO_REQUEST | ver;
 169         if (sm->ptk_set)
 170                 key_info |= WPA_KEY_INFO_MIC;
 171         if (error)
 172                 key_info |= WPA_KEY_INFO_ERROR;
 173         if (pairwise)
 174                 key_info |= WPA_KEY_INFO_KEY_TYPE;
 175         WPA_PUT_BE16(reply->key_info, key_info);
 176         WPA_PUT_BE16(reply->key_length, 0);
 177         os_memcpy(reply->replay_counter, sm->request_counter,
 178                   WPA_REPLAY_COUNTER_LEN);
 179         inc_byte_array(sm->request_counter, WPA_REPLAY_COUNTER_LEN);
 180
 181         WPA_PUT_BE16(reply->key_data_length, 0);
 182
 183         wpa_printf(MSG_INFO, "WPA: Sending EAPOL-Key Request (error=%d "
 184                    "pairwise=%d ptk_set=%d len=%lu)",
 185                    error, pairwise, sm->ptk_set, (unsigned long) rlen);
 186         wpa_eapol_key_send(sm, sm->ptk.kck, ver, bssid, ETH_P_EAPOL,
 187                            rbuf, rlen, key_info & WPA_KEY_INFO_MIC ?
 188                            reply->key_mic : NULL);
 189 }
 190
 191
 192 static int wpa_supplicant_get_pmk(struct wpa_sm *sm,
 193                                   const unsigned char *src_addr,
 194                                   const u8 *pmkid)
 195 {
 196         int abort_cached = 0;
 197
 198         if (pmkid && !sm->cur_pmksa) {
 199                 /* When using drivers that generate RSN IE, wpa_supplicant may
 200                  * not have enough time to get the association information
 201                  * event before receiving this 1/4 message, so try to find a
 202                  * matching PMKSA cache entry here. */
 203                 sm->cur_pmksa = pmksa_cache_get(sm->pmksa, src_addr, pmkid);
 204                 if (sm->cur_pmksa) {
 205                         wpa_printf(MSG_DEBUG, "RSN: found matching PMKID from "
 206                                    "PMKSA cache");
 207                 } else {
 208                         wpa_printf(MSG_DEBUG, "RSN: no matching PMKID found");
 209                         abort_cached = 1;
 210                 }
 211         }
 212
 213         if (pmkid && sm->cur_pmksa &&
 214             os_memcmp(pmkid, sm->cur_pmksa->pmkid, PMKID_LEN) == 0) {
 215                 wpa_hexdump(MSG_DEBUG, "RSN: matched PMKID", pmkid, PMKID_LEN);
 216                 wpa_sm_set_pmk_from_pmksa(sm);
 217                 wpa_hexdump_key(MSG_DEBUG, "RSN: PMK from PMKSA cache",
 218                                 sm->pmk, sm->pmk_len);
 219                 eapol_sm_notify_cached(sm->eapol);
 220 #ifdef CONFIG_IEEE80211R
 221                 sm->xxkey_len = 0;
 222 #endif /* CONFIG_IEEE80211R */
 223         } else if (wpa_key_mgmt_wpa_ieee8021x(sm->key_mgmt) && sm->eapol) {
 224                 int res, pmk_len;
 225                 pmk_len = PMK_LEN;
 226                 res = eapol_sm_get_key(sm->eapol, sm->pmk, PMK_LEN);
 227                 if (res) {
 228                         /*
 229                          * EAP-LEAP is an exception from other EAP methods: it
 230                          * uses only 16-byte PMK.
 231                          */
 232                         res = eapol_sm_get_key(sm->eapol, sm->pmk, 16);
 233                         pmk_len = 16;
 234                 } else {
 235 #ifdef CONFIG_IEEE80211R
 236                         u8 buf[2 * PMK_LEN];
 237                         if (eapol_sm_get_key(sm->eapol, buf, 2 * PMK_LEN) == 0)
 238                         {
 239                                 os_memcpy(sm->xxkey, buf + PMK_LEN, PMK_LEN);
 240                                 sm->xxkey_len = PMK_LEN;
 241                                 os_memset(buf, 0, sizeof(buf));
 242                         }
 243 #endif /* CONFIG_IEEE80211R */
 244                 }
 245                 if (res == 0) {
 246                         wpa_hexdump_key(MSG_DEBUG, "WPA: PMK from EAPOL state "
 247                                         "machines", sm->pmk, pmk_len);
 248                         sm->pmk_len = pmk_len;
 249                         pmksa_cache_add(sm->pmksa, sm->pmk, pmk_len, src_addr,
 250                                         sm->own_addr, sm->network_ctx,
 251                                         sm->key_mgmt);
 252                         if (!sm->cur_pmksa && pmkid &&
 253                             pmksa_cache_get(sm->pmksa, src_addr, pmkid)) {
 254                                 wpa_printf(MSG_DEBUG, "RSN: the new PMK "
 255                                            "matches with the PMKID");
 256                                 abort_cached = 0;
 257                         }
 258                 } else {
 259                         wpa_msg(sm->ctx->ctx, MSG_WARNING,
 260                                 "WPA: Failed to get master session key from "
 261                                 "EAPOL state machines");
 262                         wpa_msg(sm->ctx->ctx, MSG_WARNING,
 263                                 "WPA: Key handshake aborted");
 264                         if (sm->cur_pmksa) {
 265                                 wpa_printf(MSG_DEBUG, "RSN: Cancelled PMKSA "
 266                                            "caching attempt");
 267                                 sm->cur_pmksa = NULL;
 268                                 abort_cached = 1;
 269                         } else if (!abort_cached) {
 270                                 return -1;
 271                         }
 272                 }
 273         }
 274
 275         if (abort_cached && wpa_key_mgmt_wpa_ieee8021x(sm->key_mgmt)) {
 276                 /* Send EAPOL-Start to trigger full EAP authentication. */
 277                 u8 *buf;
 278                 size_t buflen;
 279
 280                 wpa_printf(MSG_DEBUG, "RSN: no PMKSA entry found - trigger "
 281                            "full EAP authentication");
 282                 buf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_START,
 283                                          NULL, 0, &buflen, NULL);
 284                 if (buf) {
 285                         wpa_sm_ether_send(sm, sm->bssid, ETH_P_EAPOL,
 286                                           buf, buflen);
 287                         os_free(buf);
 288                 }
 289
 290                 return -1;
 291         }
 292
 293         return 0;
 294 }
 295
 296
 297 /**
 298  * wpa_supplicant_send_2_of_4 - Send message 2 of WPA/RSN 4-Way Handshake
 299  * @sm: Pointer to WPA state machine data from wpa_sm_init()
 300  * @dst: Destination address for the frame
 301  * @key: Pointer to the EAPOL-Key frame header
 302  * @ver: Version bits from EAPOL-Key Key Info
 303  * @nonce: Nonce value for the EAPOL-Key frame
 304  * @wpa_ie: WPA/RSN IE
 305  * @wpa_ie_len: Length of the WPA/RSN IE
 306  * @ptk: PTK to use for keyed hash and encryption
 307  * Returns: 0 on success, -1 on failure
 308  */
 309 int wpa_supplicant_send_2_of_4(struct wpa_sm *sm, const unsigned char *dst,
 310                                const struct wpa_eapol_key *key,
 311                                int ver, const u8 *nonce,
 312                                const u8 *wpa_ie, size_t wpa_ie_len,
 313                                struct wpa_ptk *ptk)
 314 {
 315         size_t rlen;
 316         struct wpa_eapol_key *reply;
 317         u8 *rbuf;
 318
 319         if (wpa_ie == NULL) {
 320                 wpa_printf(MSG_WARNING, "WPA: No wpa_ie set - cannot "
 321                            "generate msg 2/4");
 322                 return -1;
 323         }
 324
 325         wpa_hexdump(MSG_DEBUG, "WPA: WPA IE for msg 2/4", wpa_ie, wpa_ie_len);
 326
 327         rbuf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_KEY,
 328                                   NULL, sizeof(*reply) + wpa_ie_len,
 329                                   &rlen, (void *) &reply);
 330         if (rbuf == NULL)
 331                 return -1;
 332
 333         reply->type = sm->proto == WPA_PROTO_RSN ?
 334                 EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
 335         WPA_PUT_BE16(reply->key_info,
 336                      ver | WPA_KEY_INFO_KEY_TYPE | WPA_KEY_INFO_MIC);
 337         if (sm->proto == WPA_PROTO_RSN)
 338                 WPA_PUT_BE16(reply->key_length, 0);
 339         else
 340                 os_memcpy(reply->key_length, key->key_length, 2);
 341         os_memcpy(reply->replay_counter, key->replay_counter,
 342                   WPA_REPLAY_COUNTER_LEN);
 343
 344         WPA_PUT_BE16(reply->key_data_length, wpa_ie_len);
 345         os_memcpy(reply + 1, wpa_ie, wpa_ie_len);
 346
 347         os_memcpy(reply->key_nonce, nonce, WPA_NONCE_LEN);
 348
 349         wpa_printf(MSG_DEBUG, "WPA: Sending EAPOL-Key 2/4");
 350         wpa_eapol_key_send(sm, ptk->kck, ver, dst, ETH_P_EAPOL,
 351                            rbuf, rlen, reply->key_mic);
 352
 353         return 0;
 354 }
 355
 356
 357 static int wpa_derive_ptk(struct wpa_sm *sm, const unsigned char *src_addr,
 358                           const struct wpa_eapol_key *key,
 359                           struct wpa_ptk *ptk)
 360 {
 361         size_t ptk_len = sm->pairwise_cipher == WPA_CIPHER_CCMP ? 48 : 64;
 362 #ifdef CONFIG_IEEE80211R
 363         if (wpa_key_mgmt_ft(sm->key_mgmt))
 364                 return wpa_derive_ptk_ft(sm, src_addr, key, ptk, ptk_len);
 365 #endif /* CONFIG_IEEE80211R */
 366
 367         wpa_pmk_to_ptk(sm->pmk, sm->pmk_len, "Pairwise key expansion",
 368                        sm->own_addr, sm->bssid, sm->snonce, key->key_nonce,
 369                        (u8 *) ptk, ptk_len,
 370                        wpa_key_mgmt_sha256(sm->key_mgmt));
 371         return 0;
 372 }
 373
 374
 375 static void wpa_supplicant_process_1_of_4(struct wpa_sm *sm,
 376                                           const unsigned char *src_addr,
 377                                           const struct wpa_eapol_key *key,
 378                                           u16 ver)
 379 {
 380         struct wpa_eapol_ie_parse ie;
 381         struct wpa_ptk *ptk;
 382         u8 buf[8];
 383
 384         if (wpa_sm_get_network_ctx(sm) == NULL) {
 385                 wpa_printf(MSG_WARNING, "WPA: No SSID info found (msg 1 of "
 386                            "4).");
 387                 return;
 388         }
 389
 390         wpa_sm_set_state(sm, WPA_4WAY_HANDSHAKE);
 391         wpa_printf(MSG_DEBUG, "WPA: RX message 1 of 4-Way Handshake from "
 392                    MACSTR " (ver=%d)", MAC2STR(src_addr), ver);
 393
 394         os_memset(&ie, 0, sizeof(ie));
 395
 396 #ifndef CONFIG_NO_WPA2
 397         if (sm->proto == WPA_PROTO_RSN) {
 398                 /* RSN: msg 1/4 should contain PMKID for the selected PMK */
 399                 const u8 *_buf = (const u8 *) (key + 1);
 400                 size_t len = WPA_GET_BE16(key->key_data_length);
 401                 wpa_hexdump(MSG_DEBUG, "RSN: msg 1/4 key data", _buf, len);
 402                 wpa_supplicant_parse_ies(_buf, len, &ie);
 403                 if (ie.pmkid) {
 404                         wpa_hexdump(MSG_DEBUG, "RSN: PMKID from "
 405                                     "Authenticator", ie.pmkid, PMKID_LEN);
 406                 }
 407         }
 408 #endif /* CONFIG_NO_WPA2 */
 409
 410         if (wpa_supplicant_get_pmk(sm, src_addr, ie.pmkid))
 411                 goto failed;
 412
 413         if (sm->renew_snonce) {
 414                 if (os_get_random(sm->snonce, WPA_NONCE_LEN)) {
 415                         wpa_msg(sm->ctx->ctx, MSG_WARNING,
 416                                 "WPA: Failed to get random data for SNonce");
 417                         goto failed;
 418                 }
 419                 sm->renew_snonce = 0;
 420                 wpa_hexdump(MSG_DEBUG, "WPA: Renewed SNonce",
 421                             sm->snonce, WPA_NONCE_LEN);
 422         }
 423
 424         /* Calculate PTK which will be stored as a temporary PTK until it has
 425          * been verified when processing message 3/4. */
 426         ptk = &sm->tptk;
 427         wpa_derive_ptk(sm, src_addr, key, ptk);
 428         /* Supplicant: swap tx/rx Mic keys */
 429         os_memcpy(buf, ptk->u.auth.tx_mic_key, 8);
 430         os_memcpy(ptk->u.auth.tx_mic_key, ptk->u.auth.rx_mic_key, 8);
 431         os_memcpy(ptk->u.auth.rx_mic_key, buf, 8);
 432         sm->tptk_set = 1;
 433
 434         if (wpa_supplicant_send_2_of_4(sm, sm->bssid, key, ver, sm->snonce,
 435                                        sm->assoc_wpa_ie, sm->assoc_wpa_ie_len,
 436                                        ptk))
 437                 goto failed;
 438
 439         os_memcpy(sm->anonce, key->key_nonce, WPA_NONCE_LEN);
 440         return;
 441
 442 failed:
 443         wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
 444 }
 445
 446
 447 static void wpa_sm_start_preauth(void *eloop_ctx, void *timeout_ctx)
 448 {
 449         struct wpa_sm *sm = eloop_ctx;
 450         rsn_preauth_candidate_process(sm);
 451 }
 452
 453
 454 static void wpa_supplicant_key_neg_complete(struct wpa_sm *sm,
 455                                             const u8 *addr, int secure)
 456 {
 457         wpa_msg(sm->ctx->ctx, MSG_INFO, "WPA: Key negotiation completed with "
 458                 MACSTR " [PTK=%s GTK=%s]", MAC2STR(addr),
 459                 wpa_cipher_txt(sm->pairwise_cipher),
 460                 wpa_cipher_txt(sm->group_cipher));
 461         wpa_sm_cancel_auth_timeout(sm);
 462         wpa_sm_set_state(sm, WPA_COMPLETED);
 463
 464         if (secure) {
 465                 wpa_sm_mlme_setprotection(
 466                         sm, addr, MLME_SETPROTECTION_PROTECT_TYPE_RX_TX,
 467                         MLME_SETPROTECTION_KEY_TYPE_PAIRWISE);
 468                 eapol_sm_notify_portValid(sm->eapol, TRUE);
 469                 if (wpa_key_mgmt_wpa_psk(sm->key_mgmt))
 470                         eapol_sm_notify_eap_success(sm->eapol, TRUE);
 471                 /*
 472                  * Start preauthentication after a short wait to avoid a
 473                  * possible race condition between the data receive and key
 474                  * configuration after the 4-Way Handshake. This increases the
 475                  * likelyhood of the first preauth EAPOL-Start frame getting to
 476                  * the target AP.
 477                  */
 478                 eloop_register_timeout(1, 0, wpa_sm_start_preauth, sm, NULL);
 479         }
 480
 481         if (sm->cur_pmksa && sm->cur_pmksa->opportunistic) {
 482                 wpa_printf(MSG_DEBUG, "RSN: Authenticator accepted "
 483                            "opportunistic PMKSA entry - marking it valid");
 484                 sm->cur_pmksa->opportunistic = 0;
 485         }
 486
 487 #ifdef CONFIG_IEEE80211R
 488         if (wpa_key_mgmt_ft(sm->key_mgmt)) {
 489                 /* Prepare for the next transition */
 490                 wpa_ft_prepare_auth_request(sm);
 491         }
 492 #endif /* CONFIG_IEEE80211R */
 493 }
 494
 495
 496 static void wpa_sm_rekey_ptk(void *eloop_ctx, void *timeout_ctx)
 497 {
 498         struct wpa_sm *sm = eloop_ctx;
 499         wpa_printf(MSG_DEBUG, "WPA: Request PTK rekeying");
 500         wpa_sm_key_request(sm, 0, 1);
 501 }
 502
 503
 504 static int wpa_supplicant_install_ptk(struct wpa_sm *sm,
 505                                       const struct wpa_eapol_key *key)
 506 {
 507         int keylen, rsclen;
 508         wpa_alg alg;
 509         const u8 *key_rsc;
 510         u8 null_rsc[8] = { 0, 0, 0, 0, 0, 0, 0, 0 };
 511
 512         wpa_printf(MSG_DEBUG, "WPA: Installing PTK to the driver.");
 513
 514         switch (sm->pairwise_cipher) {
 515         case WPA_CIPHER_CCMP:
 516                 alg = WPA_ALG_CCMP;
 517                 keylen = 16;
 518                 rsclen = 6;
 519                 break;
 520         case WPA_CIPHER_TKIP:
 521                 alg = WPA_ALG_TKIP;
 522                 keylen = 32;
 523                 rsclen = 6;
 524                 break;
 525         case WPA_CIPHER_NONE:
 526                 wpa_printf(MSG_DEBUG, "WPA: Pairwise Cipher Suite: "
 527                            "NONE - do not use pairwise keys");
 528                 return 0;
 529         default:
 530                 wpa_printf(MSG_WARNING, "WPA: Unsupported pairwise cipher %d",
 531                            sm->pairwise_cipher);
 532                 return -1;
 533         }
 534
 535         if (sm->proto == WPA_PROTO_RSN) {
 536                 key_rsc = null_rsc;
 537         } else {
 538                 key_rsc = key->key_rsc;
 539                 wpa_hexdump(MSG_DEBUG, "WPA: RSC", key_rsc, rsclen);
 540         }
 541
 542         if (wpa_sm_set_key(sm, alg, sm->bssid, 0, 1, key_rsc, rsclen,
 543                            (u8 *) sm->ptk.tk1, keylen) < 0) {
 544                 wpa_printf(MSG_WARNING, "WPA: Failed to set PTK to the "
 545                            "driver (alg=%d keylen=%d bssid=" MACSTR ")",
 546                            alg, keylen, MAC2STR(sm->bssid));
 547                 return -1;
 548         }
 549
 550         if (sm->wpa_ptk_rekey) {
 551                 eloop_cancel_timeout(wpa_sm_rekey_ptk, sm, NULL);
 552                 eloop_register_timeout(sm->wpa_ptk_rekey, 0, wpa_sm_rekey_ptk,
 553                                        sm, NULL);
 554         }
 555
 556         return 0;
 557 }
 558
 559
 560 static int wpa_supplicant_check_group_cipher(int group_cipher,
 561                                              int keylen, int maxkeylen,
 562                                              int *key_rsc_len, wpa_alg *alg)
 563 {
 564         int ret = 0;
 565
 566         switch (group_cipher) {
 567         case WPA_CIPHER_CCMP:
 568                 if (keylen != 16 || maxkeylen < 16) {
 569                         ret = -1;
 570                         break;
 571                 }
 572                 *key_rsc_len = 6;
 573                 *alg = WPA_ALG_CCMP;
 574                 break;
 575         case WPA_CIPHER_TKIP:
 576                 if (keylen != 32 || maxkeylen < 32) {
 577                         ret = -1;
 578                         break;
 579                 }
 580                 *key_rsc_len = 6;
 581                 *alg = WPA_ALG_TKIP;
 582                 break;
 583         case WPA_CIPHER_WEP104:
 584                 if (keylen != 13 || maxkeylen < 13) {
 585                         ret = -1;
 586                         break;
 587                 }
 588                 *key_rsc_len = 0;
 589                 *alg = WPA_ALG_WEP;
 590                 break;
 591         case WPA_CIPHER_WEP40:
 592                 if (keylen != 5 || maxkeylen < 5) {
 593                         ret = -1;
 594                         break;
 595                 }
 596                 *key_rsc_len = 0;
 597                 *alg = WPA_ALG_WEP;
 598                 break;
 599         default:
 600                 wpa_printf(MSG_WARNING, "WPA: Unsupported Group Cipher %d",
 601                            group_cipher);
 602                 return -1;
 603         }
 604
 605         if (ret < 0 ) {
 606                 wpa_printf(MSG_WARNING, "WPA: Unsupported %s Group Cipher key "
 607                            "length %d (%d).",
 608                            wpa_cipher_txt(group_cipher), keylen, maxkeylen);
 609         }
 610
 611         return ret;
 612 }
 613
 614
 615 struct wpa_gtk_data {
 616         wpa_alg alg;
 617         int tx, key_rsc_len, keyidx;
 618         u8 gtk[32];
 619         int gtk_len;
 620 };
 621
 622
 623 static int wpa_supplicant_install_gtk(struct wpa_sm *sm,
 624                                       const struct wpa_gtk_data *gd,
 625                                       const u8 *key_rsc)
 626 {
 627         const u8 *_gtk = gd->gtk;
 628         u8 gtk_buf[32];
 629
 630         wpa_hexdump_key(MSG_DEBUG, "WPA: Group Key", gd->gtk, gd->gtk_len);
 631         wpa_printf(MSG_DEBUG, "WPA: Installing GTK to the driver "
 632                    "(keyidx=%d tx=%d len=%d).", gd->keyidx, gd->tx,
 633                    gd->gtk_len);
 634         wpa_hexdump(MSG_DEBUG, "WPA: RSC", key_rsc, gd->key_rsc_len);
 635         if (sm->group_cipher == WPA_CIPHER_TKIP) {
 636                 /* Swap Tx/Rx keys for Michael MIC */
 637                 os_memcpy(gtk_buf, gd->gtk, 16);
 638                 os_memcpy(gtk_buf + 16, gd->gtk + 24, 8);
 639                 os_memcpy(gtk_buf + 24, gd->gtk + 16, 8);
 640                 _gtk = gtk_buf;
 641         }
 642         if (sm->pairwise_cipher == WPA_CIPHER_NONE) {
 643                 if (wpa_sm_set_key(sm, gd->alg,
 644                                    (u8 *) "\xff\xff\xff\xff\xff\xff",
 645                                    gd->keyidx, 1, key_rsc, gd->key_rsc_len,
 646                                    _gtk, gd->gtk_len) < 0) {
 647                         wpa_printf(MSG_WARNING, "WPA: Failed to set "
 648                                    "GTK to the driver (Group only).");
 649                         return -1;
 650                 }
 651         } else if (wpa_sm_set_key(sm, gd->alg,
 652                                   (u8 *) "\xff\xff\xff\xff\xff\xff",
 653                                   gd->keyidx, gd->tx, key_rsc, gd->key_rsc_len,
 654                                   _gtk, gd->gtk_len) < 0) {
 655                 wpa_printf(MSG_WARNING, "WPA: Failed to set GTK to "
 656                            "the driver (alg=%d keylen=%d keyidx=%d)",
 657                            gd->alg, gd->gtk_len, gd->keyidx);
 658                 return -1;
 659         }
 660
 661         return 0;
 662 }
 663
 664
 665 static int wpa_supplicant_gtk_tx_bit_workaround(const struct wpa_sm *sm,
 666                                                 int tx)
 667 {
 668         if (tx && sm->pairwise_cipher != WPA_CIPHER_NONE) {
 669                 /* Ignore Tx bit for GTK if a pairwise key is used. One AP
 670                  * seemed to set this bit (incorrectly, since Tx is only when
 671                  * doing Group Key only APs) and without this workaround, the
 672                  * data connection does not work because wpa_supplicant
 673                  * configured non-zero keyidx to be used for unicast. */
 674                 wpa_printf(MSG_INFO, "WPA: Tx bit set for GTK, but pairwise "
 675                            "keys are used - ignore Tx bit");
 676                 return 0;
 677         }
 678         return tx;
 679 }
 680
 681
 682 static int wpa_supplicant_pairwise_gtk(struct wpa_sm *sm,
 683                                        const struct wpa_eapol_key *key,
 684                                        const u8 *gtk, size_t gtk_len,
 685                                        int key_info)
 686 {
 687 #ifndef CONFIG_NO_WPA2
 688         struct wpa_gtk_data gd;
 689
 690         /*
 691          * IEEE Std 802.11i-2004 - 8.5.2 EAPOL-Key frames - Figure 43x
 692          * GTK KDE format:
 693          * KeyID[bits 0-1], Tx [bit 2], Reserved [bits 3-7]
 694          * Reserved [bits 0-7]
 695          * GTK
 696          */
 697
 698         os_memset(&gd, 0, sizeof(gd));
 699         wpa_hexdump_key(MSG_DEBUG, "RSN: received GTK in pairwise handshake",
 700                         gtk, gtk_len);
 701
 702         if (gtk_len < 2 || gtk_len - 2 > sizeof(gd.gtk))
 703                 return -1;
 704
 705         gd.keyidx = gtk[0] & 0x3;
 706         gd.tx = wpa_supplicant_gtk_tx_bit_workaround(sm,
 707                                                      !!(gtk[0] & BIT(2)));
 708         gtk += 2;
 709         gtk_len -= 2;
 710
 711         os_memcpy(gd.gtk, gtk, gtk_len);
 712         gd.gtk_len = gtk_len;
 713
 714         if (wpa_supplicant_check_group_cipher(sm->group_cipher,
 715                                               gtk_len, gtk_len,
 716                                               &gd.key_rsc_len, &gd.alg) ||
 717             wpa_supplicant_install_gtk(sm, &gd, key->key_rsc)) {
 718                 wpa_printf(MSG_DEBUG, "RSN: Failed to install GTK");
 719                 return -1;
 720         }
 721
 722         wpa_supplicant_key_neg_complete(sm, sm->bssid,
 723                                         key_info & WPA_KEY_INFO_SECURE);
 724         return 0;
 725 #else /* CONFIG_NO_WPA2 */
 726         return -1;
 727 #endif /* CONFIG_NO_WPA2 */
 728 }
 729
 730
 731 static int ieee80211w_set_keys(struct wpa_sm *sm,
 732                                struct wpa_eapol_ie_parse *ie)
 733 {
 734 #ifdef CONFIG_IEEE80211W
 735         if (sm->mgmt_group_cipher != WPA_CIPHER_AES_128_CMAC)
 736                 return 0;
 737
 738         if (ie->igtk) {
 739                 const struct wpa_igtk_kde *igtk;
 740                 u16 keyidx;
 741                 if (ie->igtk_len != sizeof(*igtk))
 742                         return -1;
 743                 igtk = (const struct wpa_igtk_kde *) ie->igtk;
 744                 keyidx = WPA_GET_LE16(igtk->keyid);
 745                 wpa_printf(MSG_DEBUG, "WPA: IGTK keyid %d "
 746                            "pn %02x%02x%02x%02x%02x%02x",
 747                            keyidx, MAC2STR(igtk->pn));
 748                 wpa_hexdump_key(MSG_DEBUG, "WPA: IGTK",
 749                                 igtk->igtk, WPA_IGTK_LEN);
 750                 if (keyidx > 4095) {
 751                         wpa_printf(MSG_WARNING, "WPA: Invalid IGTK KeyID %d",
 752                                    keyidx);
 753                         return -1;
 754                 }
 755                 if (wpa_sm_set_key(sm, WPA_ALG_IGTK,
 756                                    (u8 *) "\xff\xff\xff\xff\xff\xff",
 757                                    keyidx, 0, igtk->pn, sizeof(igtk->pn),
 758                                    igtk->igtk, WPA_IGTK_LEN) < 0) {
 759                         wpa_printf(MSG_WARNING, "WPA: Failed to configure IGTK"
 760                                    " to the driver");
 761                         return -1;
 762                 }
 763         }
 764
 765         return 0;
 766 #else /* CONFIG_IEEE80211W */
 767         return 0;
 768 #endif /* CONFIG_IEEE80211W */
 769 }
 770
 771
 772 static void wpa_report_ie_mismatch(struct wpa_sm *sm,
 773                                    const char *reason, const u8 *src_addr,
 774                                    const u8 *wpa_ie, size_t wpa_ie_len,
 775                                    const u8 *rsn_ie, size_t rsn_ie_len)
 776 {
 777         wpa_msg(sm->ctx->ctx, MSG_WARNING, "WPA: %s (src=" MACSTR ")",
 778                 reason, MAC2STR(src_addr));
 779
 780         if (sm->ap_wpa_ie) {
 781                 wpa_hexdump(MSG_INFO, "WPA: WPA IE in Beacon/ProbeResp",
 782                             sm->ap_wpa_ie, sm->ap_wpa_ie_len);
 783         }
 784         if (wpa_ie) {
 785                 if (!sm->ap_wpa_ie) {
 786                         wpa_printf(MSG_INFO, "WPA: No WPA IE in "
 787                                    "Beacon/ProbeResp");
 788                 }
 789                 wpa_hexdump(MSG_INFO, "WPA: WPA IE in 3/4 msg",
 790                             wpa_ie, wpa_ie_len);
 791         }
 792
 793         if (sm->ap_rsn_ie) {
 794                 wpa_hexdump(MSG_INFO, "WPA: RSN IE in Beacon/ProbeResp",
 795                             sm->ap_rsn_ie, sm->ap_rsn_ie_len);
 796         }
 797         if (rsn_ie) {
 798                 if (!sm->ap_rsn_ie) {
 799                         wpa_printf(MSG_INFO, "WPA: No RSN IE in "
 800                                    "Beacon/ProbeResp");
 801                 }
 802                 wpa_hexdump(MSG_INFO, "WPA: RSN IE in 3/4 msg",
 803                             rsn_ie, rsn_ie_len);
 804         }
 805
 806         wpa_sm_disassociate(sm, WLAN_REASON_IE_IN_4WAY_DIFFERS);
 807 }
 808
 809
 810 static int wpa_supplicant_validate_ie(struct wpa_sm *sm,
 811                                       const unsigned char *src_addr,
 812                                       struct wpa_eapol_ie_parse *ie)
 813 {
 814         if (sm->ap_wpa_ie == NULL && sm->ap_rsn_ie == NULL) {
 815                 wpa_printf(MSG_DEBUG, "WPA: No WPA/RSN IE for this AP known. "
 816                            "Trying to get from scan results");
 817                 if (wpa_sm_get_beacon_ie(sm) < 0) {
 818                         wpa_printf(MSG_WARNING, "WPA: Could not find AP from "
 819                                    "the scan results");
 820                 } else {
 821                         wpa_printf(MSG_DEBUG, "WPA: Found the current AP from "
 822                                    "updated scan results");
 823                 }
 824         }
 825
 826         if (ie->wpa_ie == NULL && ie->rsn_ie == NULL &&
 827             (sm->ap_wpa_ie || sm->ap_rsn_ie)) {
 828                 wpa_report_ie_mismatch(sm, "IE in 3/4 msg does not match "
 829                                        "with IE in Beacon/ProbeResp (no IE?)",
 830                                        src_addr, ie->wpa_ie, ie->wpa_ie_len,
 831                                        ie->rsn_ie, ie->rsn_ie_len);
 832                 return -1;
 833         }
 834
 835         if ((ie->wpa_ie && sm->ap_wpa_ie &&
 836              (ie->wpa_ie_len != sm->ap_wpa_ie_len ||
 837               os_memcmp(ie->wpa_ie, sm->ap_wpa_ie, ie->wpa_ie_len) != 0)) ||
 838             (ie->rsn_ie && sm->ap_rsn_ie &&
 839              (ie->rsn_ie_len != sm->ap_rsn_ie_len ||
 840               os_memcmp(ie->rsn_ie, sm->ap_rsn_ie, ie->rsn_ie_len) != 0))) {
 841                 wpa_report_ie_mismatch(sm, "IE in 3/4 msg does not match "
 842                                        "with IE in Beacon/ProbeResp",
 843                                        src_addr, ie->wpa_ie, ie->wpa_ie_len,
 844                                        ie->rsn_ie, ie->rsn_ie_len);
 845                 return -1;
 846         }
 847
 848         if (sm->proto == WPA_PROTO_WPA &&
 849             ie->rsn_ie && sm->ap_rsn_ie == NULL && sm->rsn_enabled) {
 850                 wpa_report_ie_mismatch(sm, "Possible downgrade attack "
 851                                        "detected - RSN was enabled and RSN IE "
 852                                        "was in msg 3/4, but not in "
 853                                        "Beacon/ProbeResp",
 854                                        src_addr, ie->wpa_ie, ie->wpa_ie_len,
 855                                        ie->rsn_ie, ie->rsn_ie_len);
 856                 return -1;
 857         }
 858
 859 #ifdef CONFIG_IEEE80211R
 860         if (wpa_key_mgmt_ft(sm->key_mgmt)) {
 861                 struct rsn_mdie *mdie;
 862                 /* TODO: verify that full MDIE matches with the one from scan
 863                  * results, not only mobility domain */
 864                 mdie = (struct rsn_mdie *) (ie->mdie + 2);
 865                 if (ie->mdie == NULL || ie->mdie_len < 2 + sizeof(*mdie) ||
 866                     os_memcmp(mdie->mobility_domain, sm->mobility_domain,
 867                               MOBILITY_DOMAIN_ID_LEN) != 0) {
 868                         wpa_printf(MSG_DEBUG, "FT: MDIE in msg 3/4 did not "
 869                                    "match with the current mobility domain");
 870                         return -1;
 871                 }
 872         }
 873 #endif /* CONFIG_IEEE80211R */
 874
 875         return 0;
 876 }
 877
 878
 879 /**
 880  * wpa_supplicant_send_4_of_4 - Send message 4 of WPA/RSN 4-Way Handshake
 881  * @sm: Pointer to WPA state machine data from wpa_sm_init()
 882  * @dst: Destination address for the frame
 883  * @key: Pointer to the EAPOL-Key frame header
 884  * @ver: Version bits from EAPOL-Key Key Info
 885  * @key_info: Key Info
 886  * @kde: KDEs to include the EAPOL-Key frame
 887  * @kde_len: Length of KDEs
 888  * @ptk: PTK to use for keyed hash and encryption
 889  * Returns: 0 on success, -1 on failure
 890  */
 891 int wpa_supplicant_send_4_of_4(struct wpa_sm *sm, const unsigned char *dst,
 892                                const struct wpa_eapol_key *key,
 893                                u16 ver, u16 key_info,
 894                                const u8 *kde, size_t kde_len,
 895                                struct wpa_ptk *ptk)
 896 {
 897         size_t rlen;
 898         struct wpa_eapol_key *reply;
 899         u8 *rbuf;
 900
 901         if (kde)
 902                 wpa_hexdump(MSG_DEBUG, "WPA: KDE for msg 4/4", kde, kde_len);
 903
 904         rbuf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_KEY, NULL,
 905                                   sizeof(*reply) + kde_len,
 906                                   &rlen, (void *) &reply);
 907         if (rbuf == NULL)
 908                 return -1;
 909
 910         reply->type = sm->proto == WPA_PROTO_RSN ?
 911                 EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
 912         key_info &= WPA_KEY_INFO_SECURE;
 913         key_info |= ver | WPA_KEY_INFO_KEY_TYPE | WPA_KEY_INFO_MIC;
 914         WPA_PUT_BE16(reply->key_info, key_info);
 915         if (sm->proto == WPA_PROTO_RSN)
 916                 WPA_PUT_BE16(reply->key_length, 0);
 917         else
 918                 os_memcpy(reply->key_length, key->key_length, 2);
 919         os_memcpy(reply->replay_counter, key->replay_counter,
 920                   WPA_REPLAY_COUNTER_LEN);
 921
 922         WPA_PUT_BE16(reply->key_data_length, kde_len);
 923         if (kde)
 924                 os_memcpy(reply + 1, kde, kde_len);
 925
 926         wpa_printf(MSG_DEBUG, "WPA: Sending EAPOL-Key 4/4");
 927         wpa_eapol_key_send(sm, ptk->kck, ver, dst, ETH_P_EAPOL,
 928                            rbuf, rlen, reply->key_mic);
 929
 930         return 0;
 931 }
 932
 933
 934 static void wpa_supplicant_process_3_of_4(struct wpa_sm *sm,
 935                                           const struct wpa_eapol_key *key,
 936                                           u16 ver)
 937 {
 938         u16 key_info, keylen, len;
 939         const u8 *pos;
 940         struct wpa_eapol_ie_parse ie;
 941
 942         wpa_sm_set_state(sm, WPA_4WAY_HANDSHAKE);
 943         wpa_printf(MSG_DEBUG, "WPA: RX message 3 of 4-Way Handshake from "
 944                    MACSTR " (ver=%d)", MAC2STR(sm->bssid), ver);
 945
 946         key_info = WPA_GET_BE16(key->key_info);
 947
 948         pos = (const u8 *) (key + 1);
 949         len = WPA_GET_BE16(key->key_data_length);
 950         wpa_hexdump(MSG_DEBUG, "WPA: IE KeyData", pos, len);
 951         wpa_supplicant_parse_ies(pos, len, &ie);
 952         if (ie.gtk && !(key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
 953                 wpa_printf(MSG_WARNING, "WPA: GTK IE in unencrypted key data");
 954                 goto failed;
 955         }
 956 #ifdef CONFIG_IEEE80211W
 957         if (ie.igtk && !(key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
 958                 wpa_printf(MSG_WARNING, "WPA: IGTK KDE in unencrypted key "
 959                            "data");
 960                 goto failed;
 961         }
 962
 963         if (ie.igtk && ie.igtk_len != sizeof(struct wpa_igtk_kde)) {
 964                 wpa_printf(MSG_WARNING, "WPA: Invalid IGTK KDE length %lu",
 965                            (unsigned long) ie.igtk_len);
 966                 goto failed;
 967         }
 968 #endif /* CONFIG_IEEE80211W */
 969
 970         if (wpa_supplicant_validate_ie(sm, sm->bssid, &ie) < 0)
 971                 goto failed;
 972
 973         if (os_memcmp(sm->anonce, key->key_nonce, WPA_NONCE_LEN) != 0) {
 974                 wpa_printf(MSG_WARNING, "WPA: ANonce from message 1 of 4-Way "
 975                            "Handshake differs from 3 of 4-Way Handshake - drop"
 976                            " packet (src=" MACSTR ")", MAC2STR(sm->bssid));
 977                 goto failed;
 978         }
 979
 980         keylen = WPA_GET_BE16(key->key_length);
 981         switch (sm->pairwise_cipher) {
 982         case WPA_CIPHER_CCMP:
 983                 if (keylen != 16) {
 984                         wpa_printf(MSG_WARNING, "WPA: Invalid CCMP key length "
 985                                    "%d (src=" MACSTR ")",
 986                                    keylen, MAC2STR(sm->bssid));
 987                         goto failed;
 988                 }
 989                 break;
 990         case WPA_CIPHER_TKIP:
 991                 if (keylen != 32) {
 992                         wpa_printf(MSG_WARNING, "WPA: Invalid TKIP key length "
 993                                    "%d (src=" MACSTR ")",
 994                                    keylen, MAC2STR(sm->bssid));
 995                         goto failed;
 996                 }
 997                 break;
 998         }
 999
1000         if (wpa_supplicant_send_4_of_4(sm, sm->bssid, key, ver, key_info,
1001                                        NULL, 0, &sm->ptk)) {
1002                 goto failed;
1003         }
1004
1005         /* SNonce was successfully used in msg 3/4, so mark it to be renewed
1006          * for the next 4-Way Handshake. If msg 3 is received again, the old
1007          * SNonce will still be used to avoid changing PTK. */
1008         sm->renew_snonce = 1;
1009
1010         if (key_info & WPA_KEY_INFO_INSTALL) {
1011                 if (wpa_supplicant_install_ptk(sm, key))
1012                         goto failed;
1013         }
1014
1015         if (key_info & WPA_KEY_INFO_SECURE) {
1016                 wpa_sm_mlme_setprotection(
1017                         sm, sm->bssid, MLME_SETPROTECTION_PROTECT_TYPE_RX,
1018                         MLME_SETPROTECTION_KEY_TYPE_PAIRWISE);
1019                 eapol_sm_notify_portValid(sm->eapol, TRUE);
1020         }
1021         wpa_sm_set_state(sm, WPA_GROUP_HANDSHAKE);
1022
1023         if (ie.gtk &&
1024             wpa_supplicant_pairwise_gtk(sm, key,
1025                                         ie.gtk, ie.gtk_len, key_info) < 0) {
1026                 wpa_printf(MSG_INFO, "RSN: Failed to configure GTK");
1027                 goto failed;
1028         }
1029
1030         if (ieee80211w_set_keys(sm, &ie) < 0) {
1031                 wpa_printf(MSG_INFO, "RSN: Failed to configure IGTK");
1032                 goto failed;
1033         }
1034
1035         return;
1036
1037 failed:
1038         wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
1039 }
1040
1041
1042 static int wpa_supplicant_process_1_of_2_rsn(struct wpa_sm *sm,
1043                                              const u8 *keydata,
1044                                              size_t keydatalen,
1045                                              u16 key_info,
1046                                              struct wpa_gtk_data *gd)
1047 {
1048         int maxkeylen;
1049         struct wpa_eapol_ie_parse ie;
1050
1051         wpa_hexdump(MSG_DEBUG, "RSN: msg 1/2 key data", keydata, keydatalen);
1052         wpa_supplicant_parse_ies(keydata, keydatalen, &ie);
1053         if (ie.gtk && !(key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
1054                 wpa_printf(MSG_WARNING, "WPA: GTK IE in unencrypted key data");
1055                 return -1;
1056         }
1057         if (ie.gtk == NULL) {
1058                 wpa_printf(MSG_INFO, "WPA: No GTK IE in Group Key msg 1/2");
1059                 return -1;
1060         }
1061         maxkeylen = gd->gtk_len = ie.gtk_len - 2;
1062
1063         if (wpa_supplicant_check_group_cipher(sm->group_cipher,
1064                                               gd->gtk_len, maxkeylen,
1065                                               &gd->key_rsc_len, &gd->alg))
1066                 return -1;
1067
1068         wpa_hexdump(MSG_DEBUG, "RSN: received GTK in group key handshake",
1069                     ie.gtk, ie.gtk_len);
1070         gd->keyidx = ie.gtk[0] & 0x3;
1071         gd->tx = wpa_supplicant_gtk_tx_bit_workaround(sm,
1072                                                       !!(ie.gtk[0] & BIT(2)));
1073         if (ie.gtk_len - 2 > sizeof(gd->gtk)) {
1074                 wpa_printf(MSG_INFO, "RSN: Too long GTK in GTK IE "
1075                            "(len=%lu)", (unsigned long) ie.gtk_len - 2);
1076                 return -1;
1077         }
1078         os_memcpy(gd->gtk, ie.gtk + 2, ie.gtk_len - 2);
1079
1080         if (ieee80211w_set_keys(sm, &ie) < 0)
1081                 wpa_printf(MSG_INFO, "RSN: Failed to configure IGTK");
1082
1083         return 0;
1084 }
1085
1086
1087 static int wpa_supplicant_process_1_of_2_wpa(struct wpa_sm *sm,
1088                                              const struct wpa_eapol_key *key,
1089                                              size_t keydatalen, int key_info,
1090                                              size_t extra_len, u16 ver,
1091                                              struct wpa_gtk_data *gd)
1092 {
1093         size_t maxkeylen;
1094         u8 ek[32];
1095
1096         gd->gtk_len = WPA_GET_BE16(key->key_length);
1097         maxkeylen = keydatalen;
1098         if (keydatalen > extra_len) {
1099                 wpa_printf(MSG_INFO, "WPA: Truncated EAPOL-Key packet:"
1100                            " key_data_length=%lu > extra_len=%lu",
1101                            (unsigned long) keydatalen,
1102                            (unsigned long) extra_len);
1103                 return -1;
1104         }
1105         if (ver == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
1106                 if (maxkeylen < 8) {
1107                         wpa_printf(MSG_INFO, "WPA: Too short maxkeylen (%lu)",
1108                                    (unsigned long) maxkeylen);
1109                         return -1;
1110                 }
1111                 maxkeylen -= 8;
1112         }
1113
1114         if (wpa_supplicant_check_group_cipher(sm->group_cipher,
1115                                               gd->gtk_len, maxkeylen,
1116                                               &gd->key_rsc_len, &gd->alg))
1117                 return -1;
1118
1119         gd->keyidx = (key_info & WPA_KEY_INFO_KEY_INDEX_MASK) >>
1120                 WPA_KEY_INFO_KEY_INDEX_SHIFT;
1121         if (ver == WPA_KEY_INFO_TYPE_HMAC_MD5_RC4) {
1122                 os_memcpy(ek, key->key_iv, 16);
1123                 os_memcpy(ek + 16, sm->ptk.kek, 16);
1124                 if (keydatalen > sizeof(gd->gtk)) {
1125                         wpa_printf(MSG_WARNING, "WPA: RC4 key data "
1126                                    "too long (%lu)",
1127                                    (unsigned long) keydatalen);
1128                         return -1;
1129                 }
1130                 os_memcpy(gd->gtk, key + 1, keydatalen);
1131                 rc4_skip(ek, 32, 256, gd->gtk, keydatalen);
1132         } else if (ver == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
1133                 if (keydatalen % 8) {
1134                         wpa_printf(MSG_WARNING, "WPA: Unsupported AES-WRAP "
1135                                    "len %lu", (unsigned long) keydatalen);
1136                         return -1;
1137                 }
1138                 if (maxkeylen > sizeof(gd->gtk)) {
1139                         wpa_printf(MSG_WARNING, "WPA: AES-WRAP key data "
1140                                    "too long (keydatalen=%lu maxkeylen=%lu)",
1141                                    (unsigned long) keydatalen,
1142                                    (unsigned long) maxkeylen);
1143                         return -1;
1144                 }
1145                 if (aes_unwrap(sm->ptk.kek, maxkeylen / 8,
1146                                (const u8 *) (key + 1), gd->gtk)) {
1147                         wpa_printf(MSG_WARNING, "WPA: AES unwrap "
1148                                    "failed - could not decrypt GTK");
1149                         return -1;
1150                 }
1151         } else {
1152                 wpa_printf(MSG_WARNING, "WPA: Unsupported key_info type %d",
1153                            ver);
1154                 return -1;
1155         }
1156         gd->tx = wpa_supplicant_gtk_tx_bit_workaround(
1157                 sm, !!(key_info & WPA_KEY_INFO_TXRX));
1158         return 0;
1159 }
1160
1161
1162 static int wpa_supplicant_send_2_of_2(struct wpa_sm *sm,
1163                                       const struct wpa_eapol_key *key,
1164                                       int ver, u16 key_info)
1165 {
1166         size_t rlen;
1167         struct wpa_eapol_key *reply;
1168         u8 *rbuf;
1169
1170         rbuf = wpa_sm_alloc_eapol(sm, IEEE802_1X_TYPE_EAPOL_KEY, NULL,
1171                                   sizeof(*reply), &rlen, (void *) &reply);
1172         if (rbuf == NULL)
1173                 return -1;
1174
1175         reply->type = sm->proto == WPA_PROTO_RSN ?
1176                 EAPOL_KEY_TYPE_RSN : EAPOL_KEY_TYPE_WPA;
1177         key_info &= WPA_KEY_INFO_KEY_INDEX_MASK;
1178         key_info |= ver | WPA_KEY_INFO_MIC | WPA_KEY_INFO_SECURE;
1179         WPA_PUT_BE16(reply->key_info, key_info);
1180         if (sm->proto == WPA_PROTO_RSN)
1181                 WPA_PUT_BE16(reply->key_length, 0);
1182         else
1183                 os_memcpy(reply->key_length, key->key_length, 2);
1184         os_memcpy(reply->replay_counter, key->replay_counter,
1185                   WPA_REPLAY_COUNTER_LEN);
1186
1187         WPA_PUT_BE16(reply->key_data_length, 0);
1188
1189         wpa_printf(MSG_DEBUG, "WPA: Sending EAPOL-Key 2/2");
1190         wpa_eapol_key_send(sm, sm->ptk.kck, ver, sm->bssid, ETH_P_EAPOL,
1191                            rbuf, rlen, reply->key_mic);
1192
1193         return 0;
1194 }
1195
1196
1197 static void wpa_supplicant_process_1_of_2(struct wpa_sm *sm,
1198                                           const unsigned char *src_addr,
1199                                           const struct wpa_eapol_key *key,
1200                                           int extra_len, u16 ver)
1201 {
1202         u16 key_info, keydatalen;
1203         int rekey, ret;
1204         struct wpa_gtk_data gd;
1205
1206         os_memset(&gd, 0, sizeof(gd));
1207
1208         rekey = wpa_sm_get_state(sm) == WPA_COMPLETED;
1209         wpa_printf(MSG_DEBUG, "WPA: RX message 1 of Group Key Handshake from "
1210                    MACSTR " (ver=%d)", MAC2STR(src_addr), ver);
1211
1212         key_info = WPA_GET_BE16(key->key_info);
1213         keydatalen = WPA_GET_BE16(key->key_data_length);
1214
1215         if (sm->proto == WPA_PROTO_RSN) {
1216                 ret = wpa_supplicant_process_1_of_2_rsn(sm,
1217                                                         (const u8 *) (key + 1),
1218                                                         keydatalen, key_info,
1219                                                         &gd);
1220         } else {
1221                 ret = wpa_supplicant_process_1_of_2_wpa(sm, key, keydatalen,
1222                                                         key_info, extra_len,
1223                                                         ver, &gd);
1224         }
1225
1226         wpa_sm_set_state(sm, WPA_GROUP_HANDSHAKE);
1227
1228         if (ret)
1229                 goto failed;
1230
1231         if (wpa_supplicant_install_gtk(sm, &gd, key->key_rsc) ||
1232             wpa_supplicant_send_2_of_2(sm, key, ver, key_info))
1233                 goto failed;
1234
1235         if (rekey) {
1236                 wpa_msg(sm->ctx->ctx, MSG_INFO, "WPA: Group rekeying "
1237                         "completed with " MACSTR " [GTK=%s]",
1238                         MAC2STR(sm->bssid), wpa_cipher_txt(sm->group_cipher));
1239                 wpa_sm_cancel_auth_timeout(sm);
1240                 wpa_sm_set_state(sm, WPA_COMPLETED);
1241         } else {
1242                 wpa_supplicant_key_neg_complete(sm, sm->bssid,
1243                                                 key_info &
1244                                                 WPA_KEY_INFO_SECURE);
1245         }
1246         return;
1247
1248 failed:
1249         wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
1250 }
1251
1252
1253 static int wpa_supplicant_verify_eapol_key_mic(struct wpa_sm *sm,
1254                                                struct wpa_eapol_key *key,
1255                                                u16 ver,
1256                                                const u8 *buf, size_t len)
1257 {
1258         u8 mic[16];
1259         int ok = 0;
1260
1261         os_memcpy(mic, key->key_mic, 16);
1262         if (sm->tptk_set) {
1263                 os_memset(key->key_mic, 0, 16);
1264                 wpa_eapol_key_mic(sm->tptk.kck, ver, buf, len,
1265                                   key->key_mic);
1266                 if (os_memcmp(mic, key->key_mic, 16) != 0) {
1267                         wpa_printf(MSG_WARNING, "WPA: Invalid EAPOL-Key MIC "
1268                                    "when using TPTK - ignoring TPTK");
1269                 } else {
1270                         ok = 1;
1271                         sm->tptk_set = 0;
1272                         sm->ptk_set = 1;
1273                         os_memcpy(&sm->ptk, &sm->tptk, sizeof(sm->ptk));
1274                 }
1275         }
1276
1277         if (!ok && sm->ptk_set) {
1278                 os_memset(key->key_mic, 0, 16);
1279                 wpa_eapol_key_mic(sm->ptk.kck, ver, buf, len,
1280                                   key->key_mic);
1281                 if (os_memcmp(mic, key->key_mic, 16) != 0) {
1282                         wpa_printf(MSG_WARNING, "WPA: Invalid EAPOL-Key MIC "
1283                                    "- dropping packet");
1284                         return -1;
1285                 }
1286                 ok = 1;
1287         }
1288
1289         if (!ok) {
1290                 wpa_printf(MSG_WARNING, "WPA: Could not verify EAPOL-Key MIC "
1291                            "- dropping packet");
1292                 return -1;
1293         }
1294
1295         os_memcpy(sm->rx_replay_counter, key->replay_counter,
1296                   WPA_REPLAY_COUNTER_LEN);
1297         sm->rx_replay_counter_set = 1;
1298         return 0;
1299 }
1300
1301
1302 /* Decrypt RSN EAPOL-Key key data (RC4 or AES-WRAP) */
1303 static int wpa_supplicant_decrypt_key_data(struct wpa_sm *sm,
1304                                            struct wpa_eapol_key *key, u16 ver)
1305 {
1306         u16 keydatalen = WPA_GET_BE16(key->key_data_length);
1307
1308         wpa_hexdump(MSG_DEBUG, "RSN: encrypted key data",
1309                     (u8 *) (key + 1), keydatalen);
1310         if (!sm->ptk_set) {
1311                 wpa_printf(MSG_WARNING, "WPA: PTK not available, "
1312                            "cannot decrypt EAPOL-Key key data.");
1313                 return -1;
1314         }
1315
1316         /* Decrypt key data here so that this operation does not need
1317          * to be implemented separately for each message type. */
1318         if (ver == WPA_KEY_INFO_TYPE_HMAC_MD5_RC4) {
1319                 u8 ek[32];
1320                 os_memcpy(ek, key->key_iv, 16);
1321                 os_memcpy(ek + 16, sm->ptk.kek, 16);
1322                 rc4_skip(ek, 32, 256, (u8 *) (key + 1), keydatalen);
1323         } else if (ver == WPA_KEY_INFO_TYPE_HMAC_SHA1_AES ||
1324                    ver == WPA_KEY_INFO_TYPE_AES_128_CMAC) {
1325                 u8 *buf;
1326                 if (keydatalen % 8) {
1327                         wpa_printf(MSG_WARNING, "WPA: Unsupported "
1328                                    "AES-WRAP len %d", keydatalen);
1329                         return -1;
1330                 }
1331                 keydatalen -= 8; /* AES-WRAP adds 8 bytes */
1332                 buf = os_malloc(keydatalen);
1333                 if (buf == NULL) {
1334                         wpa_printf(MSG_WARNING, "WPA: No memory for "
1335                                    "AES-UNWRAP buffer");
1336                         return -1;
1337                 }
1338                 if (aes_unwrap(sm->ptk.kek, keydatalen / 8,
1339                                (u8 *) (key + 1), buf)) {
1340                         os_free(buf);
1341                         wpa_printf(MSG_WARNING, "WPA: AES unwrap failed - "
1342                                    "could not decrypt EAPOL-Key key data");
1343                         return -1;
1344                 }
1345                 os_memcpy(key + 1, buf, keydatalen);
1346                 os_free(buf);
1347                 WPA_PUT_BE16(key->key_data_length, keydatalen);
1348         } else {
1349                 wpa_printf(MSG_WARNING, "WPA: Unsupported key_info type %d",
1350                            ver);
1351                 return -1;
1352         }
1353         wpa_hexdump_key(MSG_DEBUG, "WPA: decrypted EAPOL-Key key data",
1354                         (u8 *) (key + 1), keydatalen);
1355         return 0;
1356 }
1357
1358
1359 /**
1360  * wpa_sm_aborted_cached - Notify WPA that PMKSA caching was aborted
1361  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1362  */
1363 void wpa_sm_aborted_cached(struct wpa_sm *sm)
1364 {
1365         if (sm && sm->cur_pmksa) {
1366                 wpa_printf(MSG_DEBUG, "RSN: Cancelling PMKSA caching attempt");
1367                 sm->cur_pmksa = NULL;
1368         }
1369 }
1370
1371
1372 static void wpa_eapol_key_dump(const struct wpa_eapol_key *key)
1373 {
1374 #ifndef CONFIG_NO_STDOUT_DEBUG
1375         u16 key_info = WPA_GET_BE16(key->key_info);
1376
1377         wpa_printf(MSG_DEBUG, "  EAPOL-Key type=%d", key->type);
1378         wpa_printf(MSG_DEBUG, "  key_info 0x%x (ver=%d keyidx=%d rsvd=%d %s"
1379                    "%s%s%s%s%s%s%s)",
1380                    key_info, key_info & WPA_KEY_INFO_TYPE_MASK,
1381                    (key_info & WPA_KEY_INFO_KEY_INDEX_MASK) >>
1382                    WPA_KEY_INFO_KEY_INDEX_SHIFT,
1383                    (key_info & (BIT(13) | BIT(14) | BIT(15))) >> 13,
1384                    key_info & WPA_KEY_INFO_KEY_TYPE ? "Pairwise" : "Group",
1385                    key_info & WPA_KEY_INFO_INSTALL ? " Install" : "",
1386                    key_info & WPA_KEY_INFO_ACK ? " Ack" : "",
1387                    key_info & WPA_KEY_INFO_MIC ? " MIC" : "",
1388                    key_info & WPA_KEY_INFO_SECURE ? " Secure" : "",
1389                    key_info & WPA_KEY_INFO_ERROR ? " Error" : "",
1390                    key_info & WPA_KEY_INFO_REQUEST ? " Request" : "",
1391                    key_info & WPA_KEY_INFO_ENCR_KEY_DATA ? " Encr" : "");
1392         wpa_printf(MSG_DEBUG, "  key_length=%u key_data_length=%u",
1393                    WPA_GET_BE16(key->key_length),
1394                    WPA_GET_BE16(key->key_data_length));
1395         wpa_hexdump(MSG_DEBUG, "  replay_counter",
1396                     key->replay_counter, WPA_REPLAY_COUNTER_LEN);
1397         wpa_hexdump(MSG_DEBUG, "  key_nonce", key->key_nonce, WPA_NONCE_LEN);
1398         wpa_hexdump(MSG_DEBUG, "  key_iv", key->key_iv, 16);
1399         wpa_hexdump(MSG_DEBUG, "  key_rsc", key->key_rsc, 8);
1400         wpa_hexdump(MSG_DEBUG, "  key_id (reserved)", key->key_id, 8);
1401         wpa_hexdump(MSG_DEBUG, "  key_mic", key->key_mic, 16);
1402 #endif /* CONFIG_NO_STDOUT_DEBUG */
1403 }
1404
1405
1406 /**
1407  * wpa_sm_rx_eapol - Process received WPA EAPOL frames
1408  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1409  * @src_addr: Source MAC address of the EAPOL packet
1410  * @buf: Pointer to the beginning of the EAPOL data (EAPOL header)
1411  * @len: Length of the EAPOL frame
1412  * Returns: 1 = WPA EAPOL-Key processed, 0 = not a WPA EAPOL-Key, -1 failure
1413  *
1414  * This function is called for each received EAPOL frame. Other than EAPOL-Key
1415  * frames can be skipped if filtering is done elsewhere. wpa_sm_rx_eapol() is
1416  * only processing WPA and WPA2 EAPOL-Key frames.
1417  *
1418  * The received EAPOL-Key packets are validated and valid packets are replied
1419  * to. In addition, key material (PTK, GTK) is configured at the end of a
1420  * successful key handshake.
1421  */
1422 int wpa_sm_rx_eapol(struct wpa_sm *sm, const u8 *src_addr,
1423                     const u8 *buf, size_t len)
1424 {
1425         size_t plen, data_len, extra_len;
1426         struct ieee802_1x_hdr *hdr;
1427         struct wpa_eapol_key *key;
1428         u16 key_info, ver;
1429         u8 *tmp;
1430         int ret = -1;
1431         struct wpa_peerkey *peerkey = NULL;
1432
1433 #ifdef CONFIG_IEEE80211R
1434         sm->ft_completed = 0;
1435 #endif /* CONFIG_IEEE80211R */
1436
1437         if (len < sizeof(*hdr) + sizeof(*key)) {
1438                 wpa_printf(MSG_DEBUG, "WPA: EAPOL frame too short to be a WPA "
1439                            "EAPOL-Key (len %lu, expecting at least %lu)",
1440                            (unsigned long) len,
1441                            (unsigned long) sizeof(*hdr) + sizeof(*key));
1442                 return 0;
1443         }
1444
1445         tmp = os_malloc(len);
1446         if (tmp == NULL)
1447                 return -1;
1448         os_memcpy(tmp, buf, len);
1449
1450         hdr = (struct ieee802_1x_hdr *) tmp;
1451         key = (struct wpa_eapol_key *) (hdr + 1);
1452         plen = be_to_host16(hdr->length);
1453         data_len = plen + sizeof(*hdr);
1454         wpa_printf(MSG_DEBUG, "IEEE 802.1X RX: version=%d type=%d length=%lu",
1455                    hdr->version, hdr->type, (unsigned long) plen);
1456
1457         if (hdr->version < EAPOL_VERSION) {
1458                 /* TODO: backwards compatibility */
1459         }
1460         if (hdr->type != IEEE802_1X_TYPE_EAPOL_KEY) {
1461                 wpa_printf(MSG_DEBUG, "WPA: EAPOL frame (type %u) discarded, "
1462                         "not a Key frame", hdr->type);
1463                 ret = 0;
1464                 goto out;
1465         }
1466         if (plen > len - sizeof(*hdr) || plen < sizeof(*key)) {
1467                 wpa_printf(MSG_DEBUG, "WPA: EAPOL frame payload size %lu "
1468                            "invalid (frame size %lu)",
1469                            (unsigned long) plen, (unsigned long) len);
1470                 ret = 0;
1471                 goto out;
1472         }
1473
1474         if (key->type != EAPOL_KEY_TYPE_WPA && key->type != EAPOL_KEY_TYPE_RSN)
1475         {
1476                 wpa_printf(MSG_DEBUG, "WPA: EAPOL-Key type (%d) unknown, "
1477                            "discarded", key->type);
1478                 ret = 0;
1479                 goto out;
1480         }
1481         wpa_eapol_key_dump(key);
1482
1483         eapol_sm_notify_lower_layer_success(sm->eapol, 0);
1484         wpa_hexdump(MSG_MSGDUMP, "WPA: RX EAPOL-Key", tmp, len);
1485         if (data_len < len) {
1486                 wpa_printf(MSG_DEBUG, "WPA: ignoring %lu bytes after the IEEE "
1487                            "802.1X data", (unsigned long) len - data_len);
1488         }
1489         key_info = WPA_GET_BE16(key->key_info);
1490         ver = key_info & WPA_KEY_INFO_TYPE_MASK;
1491         if (ver != WPA_KEY_INFO_TYPE_HMAC_MD5_RC4 &&
1492 #if defined(CONFIG_IEEE80211R) || defined(CONFIG_IEEE80211W)
1493             ver != WPA_KEY_INFO_TYPE_AES_128_CMAC &&
1494 #endif /* CONFIG_IEEE80211R || CONFIG_IEEE80211W */
1495             ver != WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
1496                 wpa_printf(MSG_INFO, "WPA: Unsupported EAPOL-Key descriptor "
1497                            "version %d.", ver);
1498                 goto out;
1499         }
1500
1501 #ifdef CONFIG_IEEE80211R
1502         if (wpa_key_mgmt_ft(sm->key_mgmt)) {
1503                 /* IEEE 802.11r uses a new key_info type (AES-128-CMAC). */
1504                 if (ver != WPA_KEY_INFO_TYPE_AES_128_CMAC) {
1505                         wpa_printf(MSG_INFO, "FT: AP did not use "
1506                                    "AES-128-CMAC.");
1507                         goto out;
1508                 }
1509         } else
1510 #endif /* CONFIG_IEEE80211R */
1511 #ifdef CONFIG_IEEE80211W
1512         if (wpa_key_mgmt_sha256(sm->key_mgmt)) {
1513                 if (ver != WPA_KEY_INFO_TYPE_AES_128_CMAC) {
1514                         wpa_printf(MSG_INFO, "WPA: AP did not use the "
1515                                    "negotiated AES-128-CMAC.");
1516                         goto out;
1517                 }
1518         } else
1519 #endif /* CONFIG_IEEE80211W */
1520         if (sm->pairwise_cipher == WPA_CIPHER_CCMP &&
1521             ver != WPA_KEY_INFO_TYPE_HMAC_SHA1_AES) {
1522                 wpa_printf(MSG_INFO, "WPA: CCMP is used, but EAPOL-Key "
1523                            "descriptor version (%d) is not 2.", ver);
1524                 if (sm->group_cipher != WPA_CIPHER_CCMP &&
1525                     !(key_info & WPA_KEY_INFO_KEY_TYPE)) {
1526                         /* Earlier versions of IEEE 802.11i did not explicitly
1527                          * require version 2 descriptor for all EAPOL-Key
1528                          * packets, so allow group keys to use version 1 if
1529                          * CCMP is not used for them. */
1530                         wpa_printf(MSG_INFO, "WPA: Backwards compatibility: "
1531                                    "allow invalid version for non-CCMP group "
1532                                    "keys");
1533                 } else
1534                         goto out;
1535         }
1536
1537 #ifdef CONFIG_PEERKEY
1538         for (peerkey = sm->peerkey; peerkey; peerkey = peerkey->next) {
1539                 if (os_memcmp(peerkey->addr, src_addr, ETH_ALEN) == 0)
1540                         break;
1541         }
1542
1543         if (!(key_info & WPA_KEY_INFO_SMK_MESSAGE) && peerkey) {
1544                 if (!peerkey->initiator && peerkey->replay_counter_set &&
1545                     os_memcmp(key->replay_counter, peerkey->replay_counter,
1546                               WPA_REPLAY_COUNTER_LEN) <= 0) {
1547                         wpa_printf(MSG_WARNING, "RSN: EAPOL-Key Replay "
1548                                    "Counter did not increase (STK) - dropping "
1549                                    "packet");
1550                         goto out;
1551                 } else if (peerkey->initiator) {
1552                         u8 _tmp[WPA_REPLAY_COUNTER_LEN];
1553                         os_memcpy(_tmp, key->replay_counter,
1554                                   WPA_REPLAY_COUNTER_LEN);
1555                         inc_byte_array(_tmp, WPA_REPLAY_COUNTER_LEN);
1556                         if (os_memcmp(_tmp, peerkey->replay_counter,
1557                                       WPA_REPLAY_COUNTER_LEN) != 0) {
1558                                 wpa_printf(MSG_DEBUG, "RSN: EAPOL-Key Replay "
1559                                            "Counter did not match (STK) - "
1560                                            "dropping packet");
1561                                 goto out;
1562                         }
1563                 }
1564         }
1565
1566         if (peerkey && peerkey->initiator && (key_info & WPA_KEY_INFO_ACK)) {
1567                 wpa_printf(MSG_INFO, "RSN: Ack bit in key_info from STK peer");
1568                 goto out;
1569         }
1570 #endif /* CONFIG_PEERKEY */
1571
1572         if (!peerkey && sm->rx_replay_counter_set &&
1573             os_memcmp(key->replay_counter, sm->rx_replay_counter,
1574                       WPA_REPLAY_COUNTER_LEN) <= 0) {
1575                 wpa_printf(MSG_WARNING, "WPA: EAPOL-Key Replay Counter did not"
1576                            " increase - dropping packet");
1577                 goto out;
1578         }
1579
1580         if (!(key_info & (WPA_KEY_INFO_ACK | WPA_KEY_INFO_SMK_MESSAGE))
1581 #ifdef CONFIG_PEERKEY
1582             && (peerkey == NULL || !peerkey->initiator)
1583 #endif /* CONFIG_PEERKEY */
1584                 ) {
1585                 wpa_printf(MSG_INFO, "WPA: No Ack bit in key_info");
1586                 goto out;
1587         }
1588
1589         if (key_info & WPA_KEY_INFO_REQUEST) {
1590                 wpa_printf(MSG_INFO, "WPA: EAPOL-Key with Request bit - "
1591                            "dropped");
1592                 goto out;
1593         }
1594
1595         if ((key_info & WPA_KEY_INFO_MIC) && !peerkey &&
1596             wpa_supplicant_verify_eapol_key_mic(sm, key, ver, tmp, data_len))
1597                 goto out;
1598
1599 #ifdef CONFIG_PEERKEY
1600         if ((key_info & WPA_KEY_INFO_MIC) && peerkey &&
1601             peerkey_verify_eapol_key_mic(sm, peerkey, key, ver, tmp, data_len))
1602                 goto out;
1603 #endif /* CONFIG_PEERKEY */
1604
1605         extra_len = data_len - sizeof(*hdr) - sizeof(*key);
1606
1607         if (WPA_GET_BE16(key->key_data_length) > extra_len) {
1608                 wpa_msg(sm->ctx->ctx, MSG_INFO, "WPA: Invalid EAPOL-Key "
1609                         "frame - key_data overflow (%d > %lu)",
1610                         WPA_GET_BE16(key->key_data_length),
1611                         (unsigned long) extra_len);
1612                 goto out;
1613         }
1614         extra_len = WPA_GET_BE16(key->key_data_length);
1615
1616         if (sm->proto == WPA_PROTO_RSN &&
1617             (key_info & WPA_KEY_INFO_ENCR_KEY_DATA)) {
1618                 if (wpa_supplicant_decrypt_key_data(sm, key, ver))
1619                         goto out;
1620                 extra_len = WPA_GET_BE16(key->key_data_length);
1621         }
1622
1623         if (key_info & WPA_KEY_INFO_KEY_TYPE) {
1624                 if (key_info & WPA_KEY_INFO_KEY_INDEX_MASK) {
1625                         wpa_printf(MSG_WARNING, "WPA: Ignored EAPOL-Key "
1626                                    "(Pairwise) with non-zero key index");
1627                         goto out;
1628                 }
1629                 if (peerkey) {
1630                         /* PeerKey 4-Way Handshake */
1631                         peerkey_rx_eapol_4way(sm, peerkey, key, key_info, ver);
1632                 } else if (key_info & WPA_KEY_INFO_MIC) {
1633                         /* 3/4 4-Way Handshake */
1634                         wpa_supplicant_process_3_of_4(sm, key, ver);
1635                 } else {
1636                         /* 1/4 4-Way Handshake */
1637                         wpa_supplicant_process_1_of_4(sm, src_addr, key,
1638                                                       ver);
1639                 }
1640         } else if (key_info & WPA_KEY_INFO_SMK_MESSAGE) {
1641                 /* PeerKey SMK Handshake */
1642                 peerkey_rx_eapol_smk(sm, src_addr, key, extra_len, key_info,
1643                                      ver);
1644         } else {
1645                 if (key_info & WPA_KEY_INFO_MIC) {
1646                         /* 1/2 Group Key Handshake */
1647                         wpa_supplicant_process_1_of_2(sm, src_addr, key,
1648                                                       extra_len, ver);
1649                 } else {
1650                         wpa_printf(MSG_WARNING, "WPA: EAPOL-Key (Group) "
1651                                    "without Mic bit - dropped");
1652                 }
1653         }
1654
1655         ret = 1;
1656
1657 out:
1658         os_free(tmp);
1659         return ret;
1660 }
1661
1662
1663 #ifdef CONFIG_CTRL_IFACE
1664 static int wpa_cipher_bits(int cipher)
1665 {
1666         switch (cipher) {
1667         case WPA_CIPHER_CCMP:
1668                 return 128;
1669         case WPA_CIPHER_TKIP:
1670                 return 256;
1671         case WPA_CIPHER_WEP104:
1672                 return 104;
1673         case WPA_CIPHER_WEP40:
1674                 return 40;
1675         default:
1676                 return 0;
1677         }
1678 }
1679
1680
1681 static u32 wpa_key_mgmt_suite(struct wpa_sm *sm)
1682 {
1683         switch (sm->key_mgmt) {
1684         case WPA_KEY_MGMT_IEEE8021X:
1685                 return (sm->proto == WPA_PROTO_RSN ?
1686                         RSN_AUTH_KEY_MGMT_UNSPEC_802_1X :
1687                         WPA_AUTH_KEY_MGMT_UNSPEC_802_1X);
1688         case WPA_KEY_MGMT_PSK:
1689                 return (sm->proto == WPA_PROTO_RSN ?
1690                         RSN_AUTH_KEY_MGMT_PSK_OVER_802_1X :
1691                         WPA_AUTH_KEY_MGMT_PSK_OVER_802_1X);
1692 #ifdef CONFIG_IEEE80211R
1693         case WPA_KEY_MGMT_FT_IEEE8021X:
1694                 return RSN_AUTH_KEY_MGMT_FT_802_1X;
1695         case WPA_KEY_MGMT_FT_PSK:
1696                 return RSN_AUTH_KEY_MGMT_FT_PSK;
1697 #endif /* CONFIG_IEEE80211R */
1698 #ifdef CONFIG_IEEE80211W
1699         case WPA_KEY_MGMT_IEEE8021X_SHA256:
1700                 return RSN_AUTH_KEY_MGMT_802_1X_SHA256;
1701         case WPA_KEY_MGMT_PSK_SHA256:
1702                 return RSN_AUTH_KEY_MGMT_PSK_SHA256;
1703 #endif /* CONFIG_IEEE80211W */
1704         case WPA_KEY_MGMT_WPA_NONE:
1705                 return WPA_AUTH_KEY_MGMT_NONE;
1706         default:
1707                 return 0;
1708         }
1709 }
1710
1711
1712 static u32 wpa_cipher_suite(struct wpa_sm *sm, int cipher)
1713 {
1714         switch (cipher) {
1715         case WPA_CIPHER_CCMP:
1716                 return (sm->proto == WPA_PROTO_RSN ?
1717                         RSN_CIPHER_SUITE_CCMP : WPA_CIPHER_SUITE_CCMP);
1718         case WPA_CIPHER_TKIP:
1719                 return (sm->proto == WPA_PROTO_RSN ?
1720                         RSN_CIPHER_SUITE_TKIP : WPA_CIPHER_SUITE_TKIP);
1721         case WPA_CIPHER_WEP104:
1722                 return (sm->proto == WPA_PROTO_RSN ?
1723                         RSN_CIPHER_SUITE_WEP104 : WPA_CIPHER_SUITE_WEP104);
1724         case WPA_CIPHER_WEP40:
1725                 return (sm->proto == WPA_PROTO_RSN ?
1726                         RSN_CIPHER_SUITE_WEP40 : WPA_CIPHER_SUITE_WEP40);
1727         case WPA_CIPHER_NONE:
1728                 return (sm->proto == WPA_PROTO_RSN ?
1729                         RSN_CIPHER_SUITE_NONE : WPA_CIPHER_SUITE_NONE);
1730         default:
1731                 return 0;
1732         }
1733 }
1734
1735
1736 #define RSN_SUITE "%02x-%02x-%02x-%d"
1737 #define RSN_SUITE_ARG(s) \
1738 ((s) >> 24) & 0xff, ((s) >> 16) & 0xff, ((s) >> 8) & 0xff, (s) & 0xff
1739
1740 /**
1741  * wpa_sm_get_mib - Dump text list of MIB entries
1742  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1743  * @buf: Buffer for the list
1744  * @buflen: Length of the buffer
1745  * Returns: Number of bytes written to buffer
1746  *
1747  * This function is used fetch dot11 MIB variables.
1748  */
1749 int wpa_sm_get_mib(struct wpa_sm *sm, char *buf, size_t buflen)
1750 {
1751         char pmkid_txt[PMKID_LEN * 2 + 1];
1752         int rsna, ret;
1753         size_t len;
1754
1755         if (sm->cur_pmksa) {
1756                 wpa_snprintf_hex(pmkid_txt, sizeof(pmkid_txt),
1757                                  sm->cur_pmksa->pmkid, PMKID_LEN);
1758         } else
1759                 pmkid_txt[0] = '\0';
1760
1761         if ((wpa_key_mgmt_wpa_psk(sm->key_mgmt) ||
1762              wpa_key_mgmt_wpa_ieee8021x(sm->key_mgmt)) &&
1763             sm->proto == WPA_PROTO_RSN)
1764                 rsna = 1;
1765         else
1766                 rsna = 0;
1767
1768         ret = os_snprintf(buf, buflen,
1769                           "dot11RSNAOptionImplemented=TRUE\n"
1770                           "dot11RSNAPreauthenticationImplemented=TRUE\n"
1771                           "dot11RSNAEnabled=%s\n"
1772                           "dot11RSNAPreauthenticationEnabled=%s\n"
1773                           "dot11RSNAConfigVersion=%d\n"
1774                           "dot11RSNAConfigPairwiseKeysSupported=5\n"
1775                           "dot11RSNAConfigGroupCipherSize=%d\n"
1776                           "dot11RSNAConfigPMKLifetime=%d\n"
1777                           "dot11RSNAConfigPMKReauthThreshold=%d\n"
1778                           "dot11RSNAConfigNumberOfPTKSAReplayCounters=1\n"
1779                           "dot11RSNAConfigSATimeout=%d\n",
1780                           rsna ? "TRUE" : "FALSE",
1781                           rsna ? "TRUE" : "FALSE",
1782                           RSN_VERSION,
1783                           wpa_cipher_bits(sm->group_cipher),
1784                           sm->dot11RSNAConfigPMKLifetime,
1785                           sm->dot11RSNAConfigPMKReauthThreshold,
1786                           sm->dot11RSNAConfigSATimeout);
1787         if (ret < 0 || (size_t) ret >= buflen)
1788                 return 0;
1789         len = ret;
1790
1791         ret = os_snprintf(
1792                 buf + len, buflen - len,
1793                 "dot11RSNAAuthenticationSuiteSelected=" RSN_SUITE "\n"
1794                 "dot11RSNAPairwiseCipherSelected=" RSN_SUITE "\n"
1795                 "dot11RSNAGroupCipherSelected=" RSN_SUITE "\n"
1796                 "dot11RSNAPMKIDUsed=%s\n"
1797                 "dot11RSNAAuthenticationSuiteRequested=" RSN_SUITE "\n"
1798                 "dot11RSNAPairwiseCipherRequested=" RSN_SUITE "\n"
1799                 "dot11RSNAGroupCipherRequested=" RSN_SUITE "\n"
1800                 "dot11RSNAConfigNumberOfGTKSAReplayCounters=0\n"
1801                 "dot11RSNA4WayHandshakeFailures=%u\n",
1802                 RSN_SUITE_ARG(wpa_key_mgmt_suite(sm)),
1803                 RSN_SUITE_ARG(wpa_cipher_suite(sm, sm->pairwise_cipher)),
1804                 RSN_SUITE_ARG(wpa_cipher_suite(sm, sm->group_cipher)),
1805                 pmkid_txt,
1806                 RSN_SUITE_ARG(wpa_key_mgmt_suite(sm)),
1807                 RSN_SUITE_ARG(wpa_cipher_suite(sm, sm->pairwise_cipher)),
1808                 RSN_SUITE_ARG(wpa_cipher_suite(sm, sm->group_cipher)),
1809                 sm->dot11RSNA4WayHandshakeFailures);
1810         if (ret >= 0 && (size_t) ret < buflen)
1811                 len += ret;
1812
1813         return (int) len;
1814 }
1815 #endif /* CONFIG_CTRL_IFACE */
1816
1817
1818 static void wpa_sm_pmksa_free_cb(struct rsn_pmksa_cache_entry *entry,
1819                                  void *ctx, int replace)
1820 {
1821         struct wpa_sm *sm = ctx;
1822
1823         if (sm->cur_pmksa == entry ||
1824             (sm->pmk_len == entry->pmk_len &&
1825              os_memcmp(sm->pmk, entry->pmk, sm->pmk_len) == 0)) {
1826                 wpa_printf(MSG_DEBUG, "RSN: removed current PMKSA entry");
1827                 sm->cur_pmksa = NULL;
1828
1829                 if (replace) {
1830                         /* A new entry is being added, so no need to
1831                          * deauthenticate in this case. This happens when EAP
1832                          * authentication is completed again (reauth or failed
1833                          * PMKSA caching attempt). */
1834                         return;
1835                 }
1836
1837                 os_memset(sm->pmk, 0, sizeof(sm->pmk));
1838                 wpa_sm_deauthenticate(sm, WLAN_REASON_UNSPECIFIED);
1839         }
1840 }
1841
1842
1843 /**
1844  * wpa_sm_init - Initialize WPA state machine
1845  * @ctx: Context pointer for callbacks; this needs to be an allocated buffer
1846  * Returns: Pointer to the allocated WPA state machine data
1847  *
1848  * This function is used to allocate a new WPA state machine and the returned
1849  * value is passed to all WPA state machine calls.
1850  */
1851 struct wpa_sm * wpa_sm_init(struct wpa_sm_ctx *ctx)
1852 {
1853         struct wpa_sm *sm;
1854
1855         sm = os_zalloc(sizeof(*sm));
1856         if (sm == NULL)
1857                 return NULL;
1858         sm->renew_snonce = 1;
1859         sm->ctx = ctx;
1860
1861         sm->dot11RSNAConfigPMKLifetime = 43200;
1862         sm->dot11RSNAConfigPMKReauthThreshold = 70;
1863         sm->dot11RSNAConfigSATimeout = 60;
1864
1865         sm->pmksa = pmksa_cache_init(wpa_sm_pmksa_free_cb, sm, sm);
1866         if (sm->pmksa == NULL) {
1867                 wpa_printf(MSG_ERROR, "RSN: PMKSA cache initialization "
1868                            "failed");
1869                 os_free(sm);
1870                 return NULL;
1871         }
1872
1873         return sm;
1874 }
1875
1876
1877 /**
1878  * wpa_sm_deinit - Deinitialize WPA state machine
1879  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1880  */
1881 void wpa_sm_deinit(struct wpa_sm *sm)
1882 {
1883         if (sm == NULL)
1884                 return;
1885         pmksa_cache_deinit(sm->pmksa);
1886         eloop_cancel_timeout(wpa_sm_start_preauth, sm, NULL);
1887         eloop_cancel_timeout(wpa_sm_rekey_ptk, sm, NULL);
1888         os_free(sm->assoc_wpa_ie);
1889         os_free(sm->ap_wpa_ie);
1890         os_free(sm->ap_rsn_ie);
1891         os_free(sm->ctx);
1892         peerkey_deinit(sm);
1893         os_free(sm);
1894 }
1895
1896
1897 /**
1898  * wpa_sm_notify_assoc - Notify WPA state machine about association
1899  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1900  * @bssid: The BSSID of the new association
1901  *
1902  * This function is called to let WPA state machine know that the connection
1903  * was established.
1904  */
1905 void wpa_sm_notify_assoc(struct wpa_sm *sm, const u8 *bssid)
1906 {
1907         int clear_ptk = 1;
1908
1909         if (sm == NULL)
1910                 return;
1911
1912         wpa_printf(MSG_DEBUG, "WPA: Association event - clear replay counter");
1913         os_memcpy(sm->bssid, bssid, ETH_ALEN);
1914         os_memset(sm->rx_replay_counter, 0, WPA_REPLAY_COUNTER_LEN);
1915         sm->rx_replay_counter_set = 0;
1916         sm->renew_snonce = 1;
1917         if (os_memcmp(sm->preauth_bssid, bssid, ETH_ALEN) == 0)
1918                 rsn_preauth_deinit(sm);
1919
1920 #ifdef CONFIG_IEEE80211R
1921         if (wpa_ft_is_completed(sm)) {
1922                 wpa_supplicant_key_neg_complete(sm, sm->bssid, 1);
1923
1924                 /* Prepare for the next transition */
1925                 wpa_ft_prepare_auth_request(sm);
1926
1927                 clear_ptk = 0;
1928         }
1929 #endif /* CONFIG_IEEE80211R */
1930
1931         if (clear_ptk) {
1932                 /*
1933                  * IEEE 802.11, 8.4.10: Delete PTK SA on (re)association if
1934                  * this is not part of a Fast BSS Transition.
1935                  */
1936                 wpa_printf(MSG_DEBUG, "WPA: Clear old PTK");
1937                 sm->ptk_set = 0;
1938                 sm->tptk_set = 0;
1939         }
1940 }
1941
1942
1943 /**
1944  * wpa_sm_notify_disassoc - Notify WPA state machine about disassociation
1945  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1946  *
1947  * This function is called to let WPA state machine know that the connection
1948  * was lost. This will abort any existing pre-authentication session.
1949  */
1950 void wpa_sm_notify_disassoc(struct wpa_sm *sm)
1951 {
1952         rsn_preauth_deinit(sm);
1953         if (wpa_sm_get_state(sm) == WPA_4WAY_HANDSHAKE)
1954                 sm->dot11RSNA4WayHandshakeFailures++;
1955 }
1956
1957
1958 /**
1959  * wpa_sm_set_pmk - Set PMK
1960  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1961  * @pmk: The new PMK
1962  * @pmk_len: The length of the new PMK in bytes
1963  *
1964  * Configure the PMK for WPA state machine.
1965  */
1966 void wpa_sm_set_pmk(struct wpa_sm *sm, const u8 *pmk, size_t pmk_len)
1967 {
1968         if (sm == NULL)
1969                 return;
1970
1971         sm->pmk_len = pmk_len;
1972         os_memcpy(sm->pmk, pmk, pmk_len);
1973
1974 #ifdef CONFIG_IEEE80211R
1975         /* Set XXKey to be PSK for FT key derivation */
1976         sm->xxkey_len = pmk_len;
1977         os_memcpy(sm->xxkey, pmk, pmk_len);
1978 #endif /* CONFIG_IEEE80211R */
1979 }
1980
1981
1982 /**
1983  * wpa_sm_set_pmk_from_pmksa - Set PMK based on the current PMKSA
1984  * @sm: Pointer to WPA state machine data from wpa_sm_init()
1985  *
1986  * Take the PMK from the current PMKSA into use. If no PMKSA is active, the PMK
1987  * will be cleared.
1988  */
1989 void wpa_sm_set_pmk_from_pmksa(struct wpa_sm *sm)
1990 {
1991         if (sm == NULL)
1992                 return;
1993
1994         if (sm->cur_pmksa) {
1995                 sm->pmk_len = sm->cur_pmksa->pmk_len;
1996                 os_memcpy(sm->pmk, sm->cur_pmksa->pmk, sm->pmk_len);
1997         } else {
1998                 sm->pmk_len = PMK_LEN;
1999                 os_memset(sm->pmk, 0, PMK_LEN);
2000         }
2001 }
2002
2003
2004 /**
2005  * wpa_sm_set_fast_reauth - Set fast reauthentication (EAP) enabled/disabled
2006  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2007  * @fast_reauth: Whether fast reauthentication (EAP) is allowed
2008  */
2009 void wpa_sm_set_fast_reauth(struct wpa_sm *sm, int fast_reauth)
2010 {
2011         if (sm)
2012                 sm->fast_reauth = fast_reauth;
2013 }
2014
2015
2016 /**
2017  * wpa_sm_set_scard_ctx - Set context pointer for smartcard callbacks
2018  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2019  * @scard_ctx: Context pointer for smartcard related callback functions
2020  */
2021 void wpa_sm_set_scard_ctx(struct wpa_sm *sm, void *scard_ctx)
2022 {
2023         if (sm == NULL)
2024                 return;
2025         sm->scard_ctx = scard_ctx;
2026         if (sm->preauth_eapol)
2027                 eapol_sm_register_scard_ctx(sm->preauth_eapol, scard_ctx);
2028 }
2029
2030
2031 /**
2032  * wpa_sm_set_config - Notification of current configration change
2033  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2034  * @config: Pointer to current network configuration
2035  *
2036  * Notify WPA state machine that configuration has changed. config will be
2037  * stored as a backpointer to network configuration. This can be %NULL to clear
2038  * the stored pointed.
2039  */
2040 void wpa_sm_set_config(struct wpa_sm *sm, struct rsn_supp_config *config)
2041 {
2042         if (!sm)
2043                 return;
2044
2045         if (config) {
2046                 sm->network_ctx = config->network_ctx;
2047                 sm->peerkey_enabled = config->peerkey_enabled;
2048                 sm->allowed_pairwise_cipher = config->allowed_pairwise_cipher;
2049                 sm->proactive_key_caching = config->proactive_key_caching;
2050                 sm->eap_workaround = config->eap_workaround;
2051                 sm->eap_conf_ctx = config->eap_conf_ctx;
2052                 if (config->ssid) {
2053                         os_memcpy(sm->ssid, config->ssid, config->ssid_len);
2054                         sm->ssid_len = config->ssid_len;
2055                 } else
2056                         sm->ssid_len = 0;
2057                 sm->wpa_ptk_rekey = config->wpa_ptk_rekey;
2058         } else {
2059                 sm->network_ctx = NULL;
2060                 sm->peerkey_enabled = 0;
2061                 sm->allowed_pairwise_cipher = 0;
2062                 sm->proactive_key_caching = 0;
2063                 sm->eap_workaround = 0;
2064                 sm->eap_conf_ctx = NULL;
2065                 sm->ssid_len = 0;
2066                 sm->wpa_ptk_rekey = 0;
2067         }
2068         if (config == NULL || config->network_ctx != sm->network_ctx)
2069                 pmksa_cache_notify_reconfig(sm->pmksa);
2070 }
2071
2072
2073 /**
2074  * wpa_sm_set_own_addr - Set own MAC address
2075  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2076  * @addr: Own MAC address
2077  */
2078 void wpa_sm_set_own_addr(struct wpa_sm *sm, const u8 *addr)
2079 {
2080         if (sm)
2081                 os_memcpy(sm->own_addr, addr, ETH_ALEN);
2082 }
2083
2084
2085 /**
2086  * wpa_sm_set_ifname - Set network interface name
2087  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2088  * @ifname: Interface name
2089  * @bridge_ifname: Optional bridge interface name (for pre-auth)
2090  */
2091 void wpa_sm_set_ifname(struct wpa_sm *sm, const char *ifname,
2092                        const char *bridge_ifname)
2093 {
2094         if (sm) {
2095                 sm->ifname = ifname;
2096                 sm->bridge_ifname = bridge_ifname;
2097         }
2098 }
2099
2100
2101 /**
2102  * wpa_sm_set_eapol - Set EAPOL state machine pointer
2103  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2104  * @eapol: Pointer to EAPOL state machine allocated with eapol_sm_init()
2105  */
2106 void wpa_sm_set_eapol(struct wpa_sm *sm, struct eapol_sm *eapol)
2107 {
2108         if (sm)
2109                 sm->eapol = eapol;
2110 }
2111
2112
2113 /**
2114  * wpa_sm_set_param - Set WPA state machine parameters
2115  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2116  * @param: Parameter field
2117  * @value: Parameter value
2118  * Returns: 0 on success, -1 on failure
2119  */
2120 int wpa_sm_set_param(struct wpa_sm *sm, enum wpa_sm_conf_params param,
2121                      unsigned int value)
2122 {
2123         int ret = 0;
2124
2125         if (sm == NULL)
2126                 return -1;
2127
2128         switch (param) {
2129         case RSNA_PMK_LIFETIME:
2130                 if (value > 0)
2131                         sm->dot11RSNAConfigPMKLifetime = value;
2132                 else
2133                         ret = -1;
2134                 break;
2135         case RSNA_PMK_REAUTH_THRESHOLD:
2136                 if (value > 0 && value <= 100)
2137                         sm->dot11RSNAConfigPMKReauthThreshold = value;
2138                 else
2139                         ret = -1;
2140                 break;
2141         case RSNA_SA_TIMEOUT:
2142                 if (value > 0)
2143                         sm->dot11RSNAConfigSATimeout = value;
2144                 else
2145                         ret = -1;
2146                 break;
2147         case WPA_PARAM_PROTO:
2148                 sm->proto = value;
2149                 break;
2150         case WPA_PARAM_PAIRWISE:
2151                 sm->pairwise_cipher = value;
2152                 break;
2153         case WPA_PARAM_GROUP:
2154                 sm->group_cipher = value;
2155                 break;
2156         case WPA_PARAM_KEY_MGMT:
2157                 sm->key_mgmt = value;
2158                 break;
2159 #ifdef CONFIG_IEEE80211W
2160         case WPA_PARAM_MGMT_GROUP:
2161                 sm->mgmt_group_cipher = value;
2162                 break;
2163 #endif /* CONFIG_IEEE80211W */
2164         case WPA_PARAM_RSN_ENABLED:
2165                 sm->rsn_enabled = value;
2166                 break;
2167         default:
2168                 break;
2169         }
2170
2171         return ret;
2172 }
2173
2174
2175 /**
2176  * wpa_sm_get_param - Get WPA state machine parameters
2177  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2178  * @param: Parameter field
2179  * Returns: Parameter value
2180  */
2181 unsigned int wpa_sm_get_param(struct wpa_sm *sm, enum wpa_sm_conf_params param)
2182 {
2183         if (sm == NULL)
2184                 return 0;
2185
2186         switch (param) {
2187         case RSNA_PMK_LIFETIME:
2188                 return sm->dot11RSNAConfigPMKLifetime;
2189         case RSNA_PMK_REAUTH_THRESHOLD:
2190                 return sm->dot11RSNAConfigPMKReauthThreshold;
2191         case RSNA_SA_TIMEOUT:
2192                 return sm->dot11RSNAConfigSATimeout;
2193         case WPA_PARAM_PROTO:
2194                 return sm->proto;
2195         case WPA_PARAM_PAIRWISE:
2196                 return sm->pairwise_cipher;
2197         case WPA_PARAM_GROUP:
2198                 return sm->group_cipher;
2199         case WPA_PARAM_KEY_MGMT:
2200                 return sm->key_mgmt;
2201 #ifdef CONFIG_IEEE80211W
2202         case WPA_PARAM_MGMT_GROUP:
2203                 return sm->mgmt_group_cipher;
2204 #endif /* CONFIG_IEEE80211W */
2205         case WPA_PARAM_RSN_ENABLED:
2206                 return sm->rsn_enabled;
2207         default:
2208                 return 0;
2209         }
2210 }
2211
2212
2213 /**
2214  * wpa_sm_get_status - Get WPA state machine
2215  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2216  * @buf: Buffer for status information
2217  * @buflen: Maximum buffer length
2218  * @verbose: Whether to include verbose status information
2219  * Returns: Number of bytes written to buf.
2220  *
2221  * Query WPA state machine for status information. This function fills in
2222  * a text area with current status information. If the buffer (buf) is not
2223  * large enough, status information will be truncated to fit the buffer.
2224  */
2225 int wpa_sm_get_status(struct wpa_sm *sm, char *buf, size_t buflen,
2226                       int verbose)
2227 {
2228         char *pos = buf, *end = buf + buflen;
2229         int ret;
2230
2231         ret = os_snprintf(pos, end - pos,
2232                           "pairwise_cipher=%s\n"
2233                           "group_cipher=%s\n"
2234                           "key_mgmt=%s\n",
2235                           wpa_cipher_txt(sm->pairwise_cipher),
2236                           wpa_cipher_txt(sm->group_cipher),
2237                           wpa_key_mgmt_txt(sm->key_mgmt, sm->proto));
2238         if (ret < 0 || ret >= end - pos)
2239                 return pos - buf;
2240         pos += ret;
2241         return pos - buf;
2242 }
2243
2244
2245 /**
2246  * wpa_sm_set_assoc_wpa_ie_default - Generate own WPA/RSN IE from configuration
2247  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2248  * @wpa_ie: Pointer to buffer for WPA/RSN IE
2249  * @wpa_ie_len: Pointer to the length of the wpa_ie buffer
2250  * Returns: 0 on success, -1 on failure
2251  */
2252 int wpa_sm_set_assoc_wpa_ie_default(struct wpa_sm *sm, u8 *wpa_ie,
2253                                     size_t *wpa_ie_len)
2254 {
2255         int res;
2256
2257         if (sm == NULL)
2258                 return -1;
2259
2260         res = wpa_gen_wpa_ie(sm, wpa_ie, *wpa_ie_len);
2261         if (res < 0)
2262                 return -1;
2263         *wpa_ie_len = res;
2264
2265         wpa_hexdump(MSG_DEBUG, "WPA: Set own WPA IE default",
2266                     wpa_ie, *wpa_ie_len);
2267
2268         if (sm->assoc_wpa_ie == NULL) {
2269                 /*
2270                  * Make a copy of the WPA/RSN IE so that 4-Way Handshake gets
2271                  * the correct version of the IE even if PMKSA caching is
2272                  * aborted (which would remove PMKID from IE generation).
2273                  */
2274                 sm->assoc_wpa_ie = os_malloc(*wpa_ie_len);
2275                 if (sm->assoc_wpa_ie == NULL)
2276                         return -1;
2277
2278                 os_memcpy(sm->assoc_wpa_ie, wpa_ie, *wpa_ie_len);
2279                 sm->assoc_wpa_ie_len = *wpa_ie_len;
2280         }
2281
2282         return 0;
2283 }
2284
2285
2286 /**
2287  * wpa_sm_set_assoc_wpa_ie - Set own WPA/RSN IE from (Re)AssocReq
2288  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2289  * @ie: Pointer to IE data (starting from id)
2290  * @len: IE length
2291  * Returns: 0 on success, -1 on failure
2292  *
2293  * Inform WPA state machine about the WPA/RSN IE used in (Re)Association
2294  * Request frame. The IE will be used to override the default value generated
2295  * with wpa_sm_set_assoc_wpa_ie_default().
2296  */
2297 int wpa_sm_set_assoc_wpa_ie(struct wpa_sm *sm, const u8 *ie, size_t len)
2298 {
2299         if (sm == NULL)
2300                 return -1;
2301
2302         os_free(sm->assoc_wpa_ie);
2303         if (ie == NULL || len == 0) {
2304                 wpa_printf(MSG_DEBUG, "WPA: clearing own WPA/RSN IE");
2305                 sm->assoc_wpa_ie = NULL;
2306                 sm->assoc_wpa_ie_len = 0;
2307         } else {
2308                 wpa_hexdump(MSG_DEBUG, "WPA: set own WPA/RSN IE", ie, len);
2309                 sm->assoc_wpa_ie = os_malloc(len);
2310                 if (sm->assoc_wpa_ie == NULL)
2311                         return -1;
2312
2313                 os_memcpy(sm->assoc_wpa_ie, ie, len);
2314                 sm->assoc_wpa_ie_len = len;
2315         }
2316
2317         return 0;
2318 }
2319
2320
2321 /**
2322  * wpa_sm_set_ap_wpa_ie - Set AP WPA IE from Beacon/ProbeResp
2323  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2324  * @ie: Pointer to IE data (starting from id)
2325  * @len: IE length
2326  * Returns: 0 on success, -1 on failure
2327  *
2328  * Inform WPA state machine about the WPA IE used in Beacon / Probe Response
2329  * frame.
2330  */
2331 int wpa_sm_set_ap_wpa_ie(struct wpa_sm *sm, const u8 *ie, size_t len)
2332 {
2333         if (sm == NULL)
2334                 return -1;
2335
2336         os_free(sm->ap_wpa_ie);
2337         if (ie == NULL || len == 0) {
2338                 wpa_printf(MSG_DEBUG, "WPA: clearing AP WPA IE");
2339                 sm->ap_wpa_ie = NULL;
2340                 sm->ap_wpa_ie_len = 0;
2341         } else {
2342                 wpa_hexdump(MSG_DEBUG, "WPA: set AP WPA IE", ie, len);
2343                 sm->ap_wpa_ie = os_malloc(len);
2344                 if (sm->ap_wpa_ie == NULL)
2345                         return -1;
2346
2347                 os_memcpy(sm->ap_wpa_ie, ie, len);
2348                 sm->ap_wpa_ie_len = len;
2349         }
2350
2351         return 0;
2352 }
2353
2354
2355 /**
2356  * wpa_sm_set_ap_rsn_ie - Set AP RSN IE from Beacon/ProbeResp
2357  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2358  * @ie: Pointer to IE data (starting from id)
2359  * @len: IE length
2360  * Returns: 0 on success, -1 on failure
2361  *
2362  * Inform WPA state machine about the RSN IE used in Beacon / Probe Response
2363  * frame.
2364  */
2365 int wpa_sm_set_ap_rsn_ie(struct wpa_sm *sm, const u8 *ie, size_t len)
2366 {
2367         if (sm == NULL)
2368                 return -1;
2369
2370         os_free(sm->ap_rsn_ie);
2371         if (ie == NULL || len == 0) {
2372                 wpa_printf(MSG_DEBUG, "WPA: clearing AP RSN IE");
2373                 sm->ap_rsn_ie = NULL;
2374                 sm->ap_rsn_ie_len = 0;
2375         } else {
2376                 wpa_hexdump(MSG_DEBUG, "WPA: set AP RSN IE", ie, len);
2377                 sm->ap_rsn_ie = os_malloc(len);
2378                 if (sm->ap_rsn_ie == NULL)
2379                         return -1;
2380
2381                 os_memcpy(sm->ap_rsn_ie, ie, len);
2382                 sm->ap_rsn_ie_len = len;
2383         }
2384
2385         return 0;
2386 }
2387
2388
2389 /**
2390  * wpa_sm_parse_own_wpa_ie - Parse own WPA/RSN IE
2391  * @sm: Pointer to WPA state machine data from wpa_sm_init()
2392  * @data: Pointer to data area for parsing results
2393  * Returns: 0 on success, -1 if IE is not known, or -2 on parsing failure
2394  *
2395  * Parse the contents of the own WPA or RSN IE from (Re)AssocReq and write the
2396  * parsed data into data.
2397  */
2398 int wpa_sm_parse_own_wpa_ie(struct wpa_sm *sm, struct wpa_ie_data *data)
2399 {
2400         if (sm == NULL || sm->assoc_wpa_ie == NULL) {
2401                 wpa_printf(MSG_DEBUG, "WPA: No WPA/RSN IE available from "
2402                            "association info");
2403                 return -1;
2404         }
2405         if (wpa_parse_wpa_ie(sm->assoc_wpa_ie, sm->assoc_wpa_ie_len, data))
2406                 return -2;
2407         return 0;
2408 }