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