net/mac80211/key.c

   1 /*
   2  * Copyright 2002-2005, Instant802 Networks, Inc.
   3  * Copyright 2005-2006, Devicescape Software, Inc.
   4  * Copyright 2006-2007  Jiri Benc <jbenc@suse.cz>
   5  * Copyright 2007-2008  Johannes Berg <johannes@sipsolutions.net>
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License version 2 as
   9  * published by the Free Software Foundation.
  10  */
  11
  12 #include <linux/if_ether.h>
  13 #include <linux/etherdevice.h>
  14 #include <linux/list.h>
  15 #include <linux/rcupdate.h>
  16 #include <linux/rtnetlink.h>
  17 #include <linux/slab.h>
  18 #include <linux/export.h>
  19 #include <net/mac80211.h>
  20 #include "ieee80211_i.h"
  21 #include "driver-ops.h"
  22 #include "debugfs_key.h"
  23 #include "aes_ccm.h"
  24 #include "aes_cmac.h"
  25
  26
  27 /**
  28  * DOC: Key handling basics
  29  *
  30  * Key handling in mac80211 is done based on per-interface (sub_if_data)
  31  * keys and per-station keys. Since each station belongs to an interface,
  32  * each station key also belongs to that interface.
  33  *
  34  * Hardware acceleration is done on a best-effort basis for algorithms
  35  * that are implemented in software,  for each key the hardware is asked
  36  * to enable that key for offloading but if it cannot do that the key is
  37  * simply kept for software encryption (unless it is for an algorithm
  38  * that isn't implemented in software).
  39  * There is currently no way of knowing whether a key is handled in SW
  40  * or HW except by looking into debugfs.
  41  *
  42  * All key management is internally protected by a mutex. Within all
  43  * other parts of mac80211, key references are, just as STA structure
  44  * references, protected by RCU. Note, however, that some things are
  45  * unprotected, namely the key->sta dereferences within the hardware
  46  * acceleration functions. This means that sta_info_destroy() must
  47  * remove the key which waits for an RCU grace period.
  48  */
  49
  50 static const u8 bcast_addr[ETH_ALEN] = { 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF };
  51
  52 static void assert_key_lock(struct ieee80211_local *local)
  53 {
  54         lockdep_assert_held(&local->key_mtx);
  55 }
  56
  57 static struct ieee80211_sta *get_sta_for_key(struct ieee80211_key *key)
  58 {
  59         if (key->sta)
  60                 return &key->sta->sta;
  61
  62         return NULL;
  63 }
  64
  65 static void increment_tailroom_need_count(struct ieee80211_sub_if_data *sdata)
  66 {
  67         /*
  68          * When this count is zero, SKB resizing for allocating tailroom
  69          * for IV or MMIC is skipped. But, this check has created two race
  70          * cases in xmit path while transiting from zero count to one:
  71          *
  72          * 1. SKB resize was skipped because no key was added but just before
  73          * the xmit key is added and SW encryption kicks off.
  74          *
  75          * 2. SKB resize was skipped because all the keys were hw planted but
  76          * just before xmit one of the key is deleted and SW encryption kicks
  77          * off.
  78          *
  79          * In both the above case SW encryption will find not enough space for
  80          * tailroom and exits with WARN_ON. (See WARN_ONs at wpa.c)
  81          *
  82          * Solution has been explained at
  83          * http://mid.gmane.org/1308590980.4322.19.camel@jlt3.sipsolutions.net
  84          */
  85
  86         if (!sdata->crypto_tx_tailroom_needed_cnt++) {
  87                 /*
  88                  * Flush all XMIT packets currently using HW encryption or no
  89                  * encryption at all if the count transition is from 0 -> 1.
  90                  */
  91                 synchronize_net();
  92         }
  93 }
  94
  95 static int ieee80211_key_enable_hw_accel(struct ieee80211_key *key)
  96 {
  97         struct ieee80211_sub_if_data *sdata;
  98         struct ieee80211_sta *sta;
  99         int ret;
 100
 101         might_sleep();
 102
 103         if (!key->local->ops->set_key)
 104                 goto out_unsupported;
 105
 106         assert_key_lock(key->local);
 107
 108         sta = get_sta_for_key(key);
 109
 110         /*
 111          * If this is a per-STA GTK, check if it
 112          * is supported; if not, return.
 113          */
 114         if (sta && !(key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE) &&
 115             !(key->local->hw.flags & IEEE80211_HW_SUPPORTS_PER_STA_GTK))
 116                 goto out_unsupported;
 117
 118         sdata = key->sdata;
 119         if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
 120                 /*
 121                  * The driver doesn't know anything about VLAN interfaces.
 122                  * Hence, don't send GTKs for VLAN interfaces to the driver.
 123                  */
 124                 if (!(key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE))
 125                         goto out_unsupported;
 126                 sdata = container_of(sdata->bss,
 127                                      struct ieee80211_sub_if_data,
 128                                      u.ap);
 129         }
 130
 131         ret = drv_set_key(key->local, SET_KEY, sdata, sta, &key->conf);
 132
 133         if (!ret) {
 134                 key->flags |= KEY_FLAG_UPLOADED_TO_HARDWARE;
 135
 136                 if (!((key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_MMIC) ||
 137                       (key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV) ||
 138                       (key->conf.flags & IEEE80211_KEY_FLAG_PUT_IV_SPACE)))
 139                         sdata->crypto_tx_tailroom_needed_cnt--;
 140
 141                 WARN_ON((key->conf.flags & IEEE80211_KEY_FLAG_PUT_IV_SPACE) &&
 142                         (key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV));
 143
 144                 return 0;
 145         }
 146
 147         if (ret != -ENOSPC && ret != -EOPNOTSUPP)
 148                 wiphy_err(key->local->hw.wiphy,
 149                           "failed to set key (%d, %pM) to hardware (%d)\n",
 150                           key->conf.keyidx, sta ? sta->addr : bcast_addr, ret);
 151
 152  out_unsupported:
 153         switch (key->conf.cipher) {
 154         case WLAN_CIPHER_SUITE_WEP40:
 155         case WLAN_CIPHER_SUITE_WEP104:
 156         case WLAN_CIPHER_SUITE_TKIP:
 157         case WLAN_CIPHER_SUITE_CCMP:
 158         case WLAN_CIPHER_SUITE_AES_CMAC:
 159                 /* all of these we can do in software */
 160                 return 0;
 161         default:
 162                 return -EINVAL;
 163         }
 164 }
 165
 166 static void ieee80211_key_disable_hw_accel(struct ieee80211_key *key)
 167 {
 168         struct ieee80211_sub_if_data *sdata;
 169         struct ieee80211_sta *sta;
 170         int ret;
 171
 172         might_sleep();
 173
 174         if (!key || !key->local->ops->set_key)
 175                 return;
 176
 177         assert_key_lock(key->local);
 178
 179         if (!(key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE))
 180                 return;
 181
 182         sta = get_sta_for_key(key);
 183         sdata = key->sdata;
 184
 185         if (!((key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_MMIC) ||
 186               (key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV) ||
 187               (key->conf.flags & IEEE80211_KEY_FLAG_PUT_IV_SPACE)))
 188                 increment_tailroom_need_count(sdata);
 189
 190         if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN)
 191                 sdata = container_of(sdata->bss,
 192                                      struct ieee80211_sub_if_data,
 193                                      u.ap);
 194
 195         ret = drv_set_key(key->local, DISABLE_KEY, sdata,
 196                           sta, &key->conf);
 197
 198         if (ret)
 199                 wiphy_err(key->local->hw.wiphy,
 200                           "failed to remove key (%d, %pM) from hardware (%d)\n",
 201                           key->conf.keyidx, sta ? sta->addr : bcast_addr, ret);
 202
 203         key->flags &= ~KEY_FLAG_UPLOADED_TO_HARDWARE;
 204 }
 205
 206 void ieee80211_key_removed(struct ieee80211_key_conf *key_conf)
 207 {
 208         struct ieee80211_key *key;
 209
 210         key = container_of(key_conf, struct ieee80211_key, conf);
 211
 212         might_sleep();
 213         assert_key_lock(key->local);
 214
 215         key->flags &= ~KEY_FLAG_UPLOADED_TO_HARDWARE;
 216
 217         /*
 218          * Flush TX path to avoid attempts to use this key
 219          * after this function returns. Until then, drivers
 220          * must be prepared to handle the key.
 221          */
 222         synchronize_rcu();
 223 }
 224 EXPORT_SYMBOL_GPL(ieee80211_key_removed);
 225
 226 static void __ieee80211_set_default_key(struct ieee80211_sub_if_data *sdata,
 227                                         int idx, bool uni, bool multi)
 228 {
 229         struct ieee80211_key *key = NULL;
 230
 231         assert_key_lock(sdata->local);
 232
 233         if (idx >= 0 && idx < NUM_DEFAULT_KEYS)
 234                 key = key_mtx_dereference(sdata->local, sdata->keys[idx]);
 235
 236         if (uni)
 237                 rcu_assign_pointer(sdata->default_unicast_key, key);
 238         if (multi)
 239                 rcu_assign_pointer(sdata->default_multicast_key, key);
 240
 241         ieee80211_debugfs_key_update_default(sdata);
 242 }
 243
 244 void ieee80211_set_default_key(struct ieee80211_sub_if_data *sdata, int idx,
 245                                bool uni, bool multi)
 246 {
 247         mutex_lock(&sdata->local->key_mtx);
 248         __ieee80211_set_default_key(sdata, idx, uni, multi);
 249         mutex_unlock(&sdata->local->key_mtx);
 250 }
 251
 252 static void
 253 __ieee80211_set_default_mgmt_key(struct ieee80211_sub_if_data *sdata, int idx)
 254 {
 255         struct ieee80211_key *key = NULL;
 256
 257         assert_key_lock(sdata->local);
 258
 259         if (idx >= NUM_DEFAULT_KEYS &&
 260             idx < NUM_DEFAULT_KEYS + NUM_DEFAULT_MGMT_KEYS)
 261                 key = key_mtx_dereference(sdata->local, sdata->keys[idx]);
 262
 263         rcu_assign_pointer(sdata->default_mgmt_key, key);
 264
 265         ieee80211_debugfs_key_update_default(sdata);
 266 }
 267
 268 void ieee80211_set_default_mgmt_key(struct ieee80211_sub_if_data *sdata,
 269                                     int idx)
 270 {
 271         mutex_lock(&sdata->local->key_mtx);
 272         __ieee80211_set_default_mgmt_key(sdata, idx);
 273         mutex_unlock(&sdata->local->key_mtx);
 274 }
 275
 276
 277 static void __ieee80211_key_replace(struct ieee80211_sub_if_data *sdata,
 278                                     struct sta_info *sta,
 279                                     bool pairwise,
 280                                     struct ieee80211_key *old,
 281                                     struct ieee80211_key *new)
 282 {
 283         int idx;
 284         bool defunikey, defmultikey, defmgmtkey;
 285
 286         if (new)
 287                 list_add_tail(&new->list, &sdata->key_list);
 288
 289         if (sta && pairwise) {
 290                 rcu_assign_pointer(sta->ptk, new);
 291         } else if (sta) {
 292                 if (old)
 293                         idx = old->conf.keyidx;
 294                 else
 295                         idx = new->conf.keyidx;
 296                 rcu_assign_pointer(sta->gtk[idx], new);
 297         } else {
 298                 WARN_ON(new && old && new->conf.keyidx != old->conf.keyidx);
 299
 300                 if (old)
 301                         idx = old->conf.keyidx;
 302                 else
 303                         idx = new->conf.keyidx;
 304
 305                 defunikey = old &&
 306                         old == key_mtx_dereference(sdata->local,
 307                                                 sdata->default_unicast_key);
 308                 defmultikey = old &&
 309                         old == key_mtx_dereference(sdata->local,
 310                                                 sdata->default_multicast_key);
 311                 defmgmtkey = old &&
 312                         old == key_mtx_dereference(sdata->local,
 313                                                 sdata->default_mgmt_key);
 314
 315                 if (defunikey && !new)
 316                         __ieee80211_set_default_key(sdata, -1, true, false);
 317                 if (defmultikey && !new)
 318                         __ieee80211_set_default_key(sdata, -1, false, true);
 319                 if (defmgmtkey && !new)
 320                         __ieee80211_set_default_mgmt_key(sdata, -1);
 321
 322                 rcu_assign_pointer(sdata->keys[idx], new);
 323                 if (defunikey && new)
 324                         __ieee80211_set_default_key(sdata, new->conf.keyidx,
 325                                                     true, false);
 326                 if (defmultikey && new)
 327                         __ieee80211_set_default_key(sdata, new->conf.keyidx,
 328                                                     false, true);
 329                 if (defmgmtkey && new)
 330                         __ieee80211_set_default_mgmt_key(sdata,
 331                                                          new->conf.keyidx);
 332         }
 333
 334         if (old)
 335                 list_del(&old->list);
 336 }
 337
 338 struct ieee80211_key *ieee80211_key_alloc(u32 cipher, int idx, size_t key_len,
 339                                           const u8 *key_data,
 340                                           size_t seq_len, const u8 *seq)
 341 {
 342         struct ieee80211_key *key;
 343         int i, j, err;
 344
 345         BUG_ON(idx < 0 || idx >= NUM_DEFAULT_KEYS + NUM_DEFAULT_MGMT_KEYS);
 346
 347         key = kzalloc(sizeof(struct ieee80211_key) + key_len, GFP_KERNEL);
 348         if (!key)
 349                 return ERR_PTR(-ENOMEM);
 350
 351         /*
 352          * Default to software encryption; we'll later upload the
 353          * key to the hardware if possible.
 354          */
 355         key->conf.flags = 0;
 356         key->flags = 0;
 357
 358         key->conf.cipher = cipher;
 359         key->conf.keyidx = idx;
 360         key->conf.keylen = key_len;
 361         switch (cipher) {
 362         case WLAN_CIPHER_SUITE_WEP40:
 363         case WLAN_CIPHER_SUITE_WEP104:
 364                 key->conf.iv_len = WEP_IV_LEN;
 365                 key->conf.icv_len = WEP_ICV_LEN;
 366                 break;
 367         case WLAN_CIPHER_SUITE_TKIP:
 368                 key->conf.iv_len = TKIP_IV_LEN;
 369                 key->conf.icv_len = TKIP_ICV_LEN;
 370                 if (seq) {
 371                         for (i = 0; i < NUM_RX_DATA_QUEUES; i++) {
 372                                 key->u.tkip.rx[i].iv32 =
 373                                         get_unaligned_le32(&seq[2]);
 374                                 key->u.tkip.rx[i].iv16 =
 375                                         get_unaligned_le16(seq);
 376                         }
 377                 }
 378                 spin_lock_init(&key->u.tkip.txlock);
 379                 break;
 380         case WLAN_CIPHER_SUITE_CCMP:
 381                 key->conf.iv_len = CCMP_HDR_LEN;
 382                 key->conf.icv_len = CCMP_MIC_LEN;
 383                 if (seq) {
 384                         for (i = 0; i < NUM_RX_DATA_QUEUES + 1; i++)
 385                                 for (j = 0; j < CCMP_PN_LEN; j++)
 386                                         key->u.ccmp.rx_pn[i][j] =
 387                                                 seq[CCMP_PN_LEN - j - 1];
 388                 }
 389                 /*
 390                  * Initialize AES key state here as an optimization so that
 391                  * it does not need to be initialized for every packet.
 392                  */
 393                 key->u.ccmp.tfm = ieee80211_aes_key_setup_encrypt(key_data);
 394                 if (IS_ERR(key->u.ccmp.tfm)) {
 395                         err = PTR_ERR(key->u.ccmp.tfm);
 396                         kfree(key);
 397                         return ERR_PTR(err);
 398                 }
 399                 break;
 400         case WLAN_CIPHER_SUITE_AES_CMAC:
 401                 key->conf.iv_len = 0;
 402                 key->conf.icv_len = sizeof(struct ieee80211_mmie);
 403                 if (seq)
 404                         for (j = 0; j < 6; j++)
 405                                 key->u.aes_cmac.rx_pn[j] = seq[6 - j - 1];
 406                 /*
 407                  * Initialize AES key state here as an optimization so that
 408                  * it does not need to be initialized for every packet.
 409                  */
 410                 key->u.aes_cmac.tfm =
 411                         ieee80211_aes_cmac_key_setup(key_data);
 412                 if (IS_ERR(key->u.aes_cmac.tfm)) {
 413                         err = PTR_ERR(key->u.aes_cmac.tfm);
 414                         kfree(key);
 415                         return ERR_PTR(err);
 416                 }
 417                 break;
 418         }
 419         memcpy(key->conf.key, key_data, key_len);
 420         INIT_LIST_HEAD(&key->list);
 421
 422         return key;
 423 }
 424
 425 static void __ieee80211_key_destroy(struct ieee80211_key *key)
 426 {
 427         if (!key)
 428                 return;
 429
 430         /*
 431          * Synchronize so the TX path can no longer be using
 432          * this key before we free/remove it.
 433          */
 434         synchronize_rcu();
 435
 436         if (key->local)
 437                 ieee80211_key_disable_hw_accel(key);
 438
 439         if (key->conf.cipher == WLAN_CIPHER_SUITE_CCMP)
 440                 ieee80211_aes_key_free(key->u.ccmp.tfm);
 441         if (key->conf.cipher == WLAN_CIPHER_SUITE_AES_CMAC)
 442                 ieee80211_aes_cmac_key_free(key->u.aes_cmac.tfm);
 443         if (key->local) {
 444                 ieee80211_debugfs_key_remove(key);
 445                 key->sdata->crypto_tx_tailroom_needed_cnt--;
 446         }
 447
 448         kfree(key);
 449 }
 450
 451 int ieee80211_key_link(struct ieee80211_key *key,
 452                        struct ieee80211_sub_if_data *sdata,
 453                        struct sta_info *sta)
 454 {
 455         struct ieee80211_key *old_key;
 456         int idx, ret;
 457         bool pairwise;
 458
 459         BUG_ON(!sdata);
 460         BUG_ON(!key);
 461
 462         pairwise = key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE;
 463         idx = key->conf.keyidx;
 464         key->local = sdata->local;
 465         key->sdata = sdata;
 466         key->sta = sta;
 467
 468         if (sta) {
 469                 /*
 470                  * some hardware cannot handle TKIP with QoS, so
 471                  * we indicate whether QoS could be in use.
 472                  */
 473                 if (test_sta_flag(sta, WLAN_STA_WME))
 474                         key->conf.flags |= IEEE80211_KEY_FLAG_WMM_STA;
 475         } else {
 476                 if (sdata->vif.type == NL80211_IFTYPE_STATION) {
 477                         struct sta_info *ap;
 478
 479                         /*
 480                          * We're getting a sta pointer in, so must be under
 481                          * appropriate locking for sta_info_get().
 482                          */
 483
 484                         /* same here, the AP could be using QoS */
 485                         ap = sta_info_get(key->sdata, key->sdata->u.mgd.bssid);
 486                         if (ap) {
 487                                 if (test_sta_flag(ap, WLAN_STA_WME))
 488                                         key->conf.flags |=
 489                                                 IEEE80211_KEY_FLAG_WMM_STA;
 490                         }
 491                 }
 492         }
 493
 494         mutex_lock(&sdata->local->key_mtx);
 495
 496         if (sta && pairwise)
 497                 old_key = key_mtx_dereference(sdata->local, sta->ptk);
 498         else if (sta)
 499                 old_key = key_mtx_dereference(sdata->local, sta->gtk[idx]);
 500         else
 501                 old_key = key_mtx_dereference(sdata->local, sdata->keys[idx]);
 502
 503         increment_tailroom_need_count(sdata);
 504
 505         __ieee80211_key_replace(sdata, sta, pairwise, old_key, key);
 506         __ieee80211_key_destroy(old_key);
 507
 508         ieee80211_debugfs_key_add(key);
 509
 510         ret = ieee80211_key_enable_hw_accel(key);
 511
 512         mutex_unlock(&sdata->local->key_mtx);
 513
 514         return ret;
 515 }
 516
 517 void __ieee80211_key_free(struct ieee80211_key *key)
 518 {
 519         if (!key)
 520                 return;
 521
 522         /*
 523          * Replace key with nothingness if it was ever used.
 524          */
 525         if (key->sdata)
 526                 __ieee80211_key_replace(key->sdata, key->sta,
 527                                 key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE,
 528                                 key, NULL);
 529         __ieee80211_key_destroy(key);
 530 }
 531
 532 void ieee80211_key_free(struct ieee80211_local *local,
 533                         struct ieee80211_key *key)
 534 {
 535         mutex_lock(&local->key_mtx);
 536         __ieee80211_key_free(key);
 537         mutex_unlock(&local->key_mtx);
 538 }
 539
 540 void ieee80211_enable_keys(struct ieee80211_sub_if_data *sdata)
 541 {
 542         struct ieee80211_key *key;
 543
 544         ASSERT_RTNL();
 545
 546         if (WARN_ON(!ieee80211_sdata_running(sdata)))
 547                 return;
 548
 549         mutex_lock(&sdata->local->key_mtx);
 550
 551         sdata->crypto_tx_tailroom_needed_cnt = 0;
 552
 553         list_for_each_entry(key, &sdata->key_list, list) {
 554                 increment_tailroom_need_count(sdata);
 555                 ieee80211_key_enable_hw_accel(key);
 556         }
 557
 558         mutex_unlock(&sdata->local->key_mtx);
 559 }
 560
 561 void ieee80211_iter_keys(struct ieee80211_hw *hw,
 562                          struct ieee80211_vif *vif,
 563                          void (*iter)(struct ieee80211_hw *hw,
 564                                       struct ieee80211_vif *vif,
 565                                       struct ieee80211_sta *sta,
 566                                       struct ieee80211_key_conf *key,
 567                                       void *data),
 568                          void *iter_data)
 569 {
 570         struct ieee80211_local *local = hw_to_local(hw);
 571         struct ieee80211_key *key;
 572         struct ieee80211_sub_if_data *sdata;
 573
 574         ASSERT_RTNL();
 575
 576         mutex_lock(&local->key_mtx);
 577         if (vif) {
 578                 sdata = vif_to_sdata(vif);
 579                 list_for_each_entry(key, &sdata->key_list, list)
 580                         iter(hw, &sdata->vif,
 581                              key->sta ? &key->sta->sta : NULL,
 582                              &key->conf, iter_data);
 583         } else {
 584                 list_for_each_entry(sdata, &local->interfaces, list)
 585                         list_for_each_entry(key, &sdata->key_list, list)
 586                                 iter(hw, &sdata->vif,
 587                                      key->sta ? &key->sta->sta : NULL,
 588                                      &key->conf, iter_data);
 589         }
 590         mutex_unlock(&local->key_mtx);
 591 }
 592 EXPORT_SYMBOL(ieee80211_iter_keys);
 593
 594 void ieee80211_disable_keys(struct ieee80211_sub_if_data *sdata)
 595 {
 596         struct ieee80211_key *key;
 597
 598         ASSERT_RTNL();
 599
 600         mutex_lock(&sdata->local->key_mtx);
 601
 602         list_for_each_entry(key, &sdata->key_list, list)
 603                 ieee80211_key_disable_hw_accel(key);
 604
 605         mutex_unlock(&sdata->local->key_mtx);
 606 }
 607
 608 void ieee80211_free_keys(struct ieee80211_sub_if_data *sdata)
 609 {
 610         struct ieee80211_key *key, *tmp;
 611
 612         mutex_lock(&sdata->local->key_mtx);
 613
 614         ieee80211_debugfs_key_remove_mgmt_default(sdata);
 615
 616         list_for_each_entry_safe(key, tmp, &sdata->key_list, list)
 617                 __ieee80211_key_free(key);
 618
 619         ieee80211_debugfs_key_update_default(sdata);
 620
 621         mutex_unlock(&sdata->local->key_mtx);
 622 }
 623
 624
 625 void ieee80211_gtk_rekey_notify(struct ieee80211_vif *vif, const u8 *bssid,
 626                                 const u8 *replay_ctr, gfp_t gfp)
 627 {
 628         struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
 629
 630         trace_api_gtk_rekey_notify(sdata, bssid, replay_ctr);
 631
 632         cfg80211_gtk_rekey_notify(sdata->dev, bssid, replay_ctr, gfp);
 633 }
 634 EXPORT_SYMBOL_GPL(ieee80211_gtk_rekey_notify);
 635
 636 void ieee80211_get_key_tx_seq(struct ieee80211_key_conf *keyconf,
 637                               struct ieee80211_key_seq *seq)
 638 {
 639         struct ieee80211_key *key;
 640         u64 pn64;
 641
 642         if (WARN_ON(!(keyconf->flags & IEEE80211_KEY_FLAG_GENERATE_IV)))
 643                 return;
 644
 645         key = container_of(keyconf, struct ieee80211_key, conf);
 646
 647         switch (key->conf.cipher) {
 648         case WLAN_CIPHER_SUITE_TKIP:
 649                 seq->tkip.iv32 = key->u.tkip.tx.iv32;
 650                 seq->tkip.iv16 = key->u.tkip.tx.iv16;
 651                 break;
 652         case WLAN_CIPHER_SUITE_CCMP:
 653                 pn64 = atomic64_read(&key->u.ccmp.tx_pn);
 654                 seq->ccmp.pn[5] = pn64;
 655                 seq->ccmp.pn[4] = pn64 >> 8;
 656                 seq->ccmp.pn[3] = pn64 >> 16;
 657                 seq->ccmp.pn[2] = pn64 >> 24;
 658                 seq->ccmp.pn[1] = pn64 >> 32;
 659                 seq->ccmp.pn[0] = pn64 >> 40;
 660                 break;
 661         case WLAN_CIPHER_SUITE_AES_CMAC:
 662                 pn64 = atomic64_read(&key->u.aes_cmac.tx_pn);
 663                 seq->ccmp.pn[5] = pn64;
 664                 seq->ccmp.pn[4] = pn64 >> 8;
 665                 seq->ccmp.pn[3] = pn64 >> 16;
 666                 seq->ccmp.pn[2] = pn64 >> 24;
 667                 seq->ccmp.pn[1] = pn64 >> 32;
 668                 seq->ccmp.pn[0] = pn64 >> 40;
 669                 break;
 670         default:
 671                 WARN_ON(1);
 672         }
 673 }
 674 EXPORT_SYMBOL(ieee80211_get_key_tx_seq);
 675
 676 void ieee80211_get_key_rx_seq(struct ieee80211_key_conf *keyconf,
 677                               int tid, struct ieee80211_key_seq *seq)
 678 {
 679         struct ieee80211_key *key;
 680         const u8 *pn;
 681
 682         key = container_of(keyconf, struct ieee80211_key, conf);
 683
 684         switch (key->conf.cipher) {
 685         case WLAN_CIPHER_SUITE_TKIP:
 686                 if (WARN_ON(tid < 0 || tid >= NUM_RX_DATA_QUEUES))
 687                         return;
 688                 seq->tkip.iv32 = key->u.tkip.rx[tid].iv32;
 689                 seq->tkip.iv16 = key->u.tkip.rx[tid].iv16;
 690                 break;
 691         case WLAN_CIPHER_SUITE_CCMP:
 692                 if (WARN_ON(tid < -1 || tid >= NUM_RX_DATA_QUEUES))
 693                         return;
 694                 if (tid < 0)
 695                         pn = key->u.ccmp.rx_pn[NUM_RX_DATA_QUEUES];
 696                 else
 697                         pn = key->u.ccmp.rx_pn[tid];
 698                 memcpy(seq->ccmp.pn, pn, CCMP_PN_LEN);
 699                 break;
 700         case WLAN_CIPHER_SUITE_AES_CMAC:
 701                 if (WARN_ON(tid != 0))
 702                         return;
 703                 pn = key->u.aes_cmac.rx_pn;
 704                 memcpy(seq->aes_cmac.pn, pn, CMAC_PN_LEN);
 705                 break;
 706         }
 707 }
 708 EXPORT_SYMBOL(ieee80211_get_key_rx_seq);