1
2
3
4
5
6
7
8
9
10
11
12
13
14#include <linux/if_ether.h>
15#include <linux/etherdevice.h>
16#include <linux/list.h>
17#include <linux/rcupdate.h>
18#include <linux/rtnetlink.h>
19#include <linux/slab.h>
20#include <linux/export.h>
21#include <net/mac80211.h>
22#include <asm/unaligned.h>
23#include "ieee80211_i.h"
24#include "driver-ops.h"
25#include "debugfs_key.h"
26#include "aes_ccm.h"
27#include "aes_cmac.h"
28#include "aes_gmac.h"
29#include "aes_gcm.h"
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55static const u8 bcast_addr[ETH_ALEN] = { 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF };
56
57static void assert_key_lock(struct ieee80211_local *local)
58{
59 lockdep_assert_held(&local->key_mtx);
60}
61
62static void
63update_vlan_tailroom_need_count(struct ieee80211_sub_if_data *sdata, int delta)
64{
65 struct ieee80211_sub_if_data *vlan;
66
67 if (sdata->vif.type != NL80211_IFTYPE_AP)
68 return;
69
70
71 assert_key_lock(sdata->local);
72
73 rcu_read_lock();
74
75 list_for_each_entry_rcu(vlan, &sdata->u.ap.vlans, u.vlan.list)
76 vlan->crypto_tx_tailroom_needed_cnt += delta;
77
78 rcu_read_unlock();
79}
80
81static void increment_tailroom_need_count(struct ieee80211_sub_if_data *sdata)
82{
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102 assert_key_lock(sdata->local);
103
104 update_vlan_tailroom_need_count(sdata, 1);
105
106 if (!sdata->crypto_tx_tailroom_needed_cnt++) {
107
108
109
110
111 synchronize_net();
112 }
113}
114
115static void decrease_tailroom_need_count(struct ieee80211_sub_if_data *sdata,
116 int delta)
117{
118 assert_key_lock(sdata->local);
119
120 WARN_ON_ONCE(sdata->crypto_tx_tailroom_needed_cnt < delta);
121
122 update_vlan_tailroom_need_count(sdata, -delta);
123 sdata->crypto_tx_tailroom_needed_cnt -= delta;
124}
125
126static int ieee80211_key_enable_hw_accel(struct ieee80211_key *key)
127{
128 struct ieee80211_sub_if_data *sdata;
129 struct sta_info *sta;
130 int ret = -EOPNOTSUPP;
131
132 might_sleep();
133
134 if (key->flags & KEY_FLAG_TAINTED) {
135
136
137
138
139
140
141
142 key->flags &= ~KEY_FLAG_UPLOADED_TO_HARDWARE;
143 return -EINVAL;
144 }
145
146 if (!key->local->ops->set_key)
147 goto out_unsupported;
148
149 assert_key_lock(key->local);
150
151 sta = key->sta;
152
153
154
155
156
157 if (sta && !(key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE) &&
158 !ieee80211_hw_check(&key->local->hw, SUPPORTS_PER_STA_GTK))
159 goto out_unsupported;
160
161 if (sta && !sta->uploaded)
162 goto out_unsupported;
163
164 sdata = key->sdata;
165 if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
166
167
168
169
170 if (!(key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE))
171 goto out_unsupported;
172 }
173
174 ret = drv_set_key(key->local, SET_KEY, sdata,
175 sta ? &sta->sta : NULL, &key->conf);
176
177 if (!ret) {
178 key->flags |= KEY_FLAG_UPLOADED_TO_HARDWARE;
179
180 if (!((key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_MMIC) ||
181 (key->conf.flags & IEEE80211_KEY_FLAG_RESERVE_TAILROOM)))
182 decrease_tailroom_need_count(sdata, 1);
183
184 WARN_ON((key->conf.flags & IEEE80211_KEY_FLAG_PUT_IV_SPACE) &&
185 (key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_IV));
186
187 return 0;
188 }
189
190 if (ret != -ENOSPC && ret != -EOPNOTSUPP && ret != 1)
191 sdata_err(sdata,
192 "failed to set key (%d, %pM) to hardware (%d)\n",
193 key->conf.keyidx,
194 sta ? sta->sta.addr : bcast_addr, ret);
195
196 out_unsupported:
197 switch (key->conf.cipher) {
198 case WLAN_CIPHER_SUITE_WEP40:
199 case WLAN_CIPHER_SUITE_WEP104:
200 case WLAN_CIPHER_SUITE_TKIP:
201 case WLAN_CIPHER_SUITE_CCMP:
202 case WLAN_CIPHER_SUITE_CCMP_256:
203 case WLAN_CIPHER_SUITE_AES_CMAC:
204 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
205 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
206 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
207 case WLAN_CIPHER_SUITE_GCMP:
208 case WLAN_CIPHER_SUITE_GCMP_256:
209
210 if (ret == 1)
211 return 0;
212 if (ieee80211_hw_check(&key->local->hw, SW_CRYPTO_CONTROL))
213 return -EINVAL;
214 return 0;
215 default:
216 return -EINVAL;
217 }
218}
219
220static void ieee80211_key_disable_hw_accel(struct ieee80211_key *key)
221{
222 struct ieee80211_sub_if_data *sdata;
223 struct sta_info *sta;
224 int ret;
225
226 might_sleep();
227
228 if (!key || !key->local->ops->set_key)
229 return;
230
231 assert_key_lock(key->local);
232
233 if (!(key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE))
234 return;
235
236 sta = key->sta;
237 sdata = key->sdata;
238
239 if (!((key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_MMIC) ||
240 (key->conf.flags & IEEE80211_KEY_FLAG_RESERVE_TAILROOM)))
241 increment_tailroom_need_count(sdata);
242
243 ret = drv_set_key(key->local, DISABLE_KEY, sdata,
244 sta ? &sta->sta : NULL, &key->conf);
245
246 if (ret)
247 sdata_err(sdata,
248 "failed to remove key (%d, %pM) from hardware (%d)\n",
249 key->conf.keyidx,
250 sta ? sta->sta.addr : bcast_addr, ret);
251
252 key->flags &= ~KEY_FLAG_UPLOADED_TO_HARDWARE;
253}
254
255static void __ieee80211_set_default_key(struct ieee80211_sub_if_data *sdata,
256 int idx, bool uni, bool multi)
257{
258 struct ieee80211_key *key = NULL;
259
260 assert_key_lock(sdata->local);
261
262 if (idx >= 0 && idx < NUM_DEFAULT_KEYS)
263 key = key_mtx_dereference(sdata->local, sdata->keys[idx]);
264
265 if (uni) {
266 rcu_assign_pointer(sdata->default_unicast_key, key);
267 ieee80211_check_fast_xmit_iface(sdata);
268 drv_set_default_unicast_key(sdata->local, sdata, idx);
269 }
270
271 if (multi)
272 rcu_assign_pointer(sdata->default_multicast_key, key);
273
274 ieee80211_debugfs_key_update_default(sdata);
275}
276
277void ieee80211_set_default_key(struct ieee80211_sub_if_data *sdata, int idx,
278 bool uni, bool multi)
279{
280 mutex_lock(&sdata->local->key_mtx);
281 __ieee80211_set_default_key(sdata, idx, uni, multi);
282 mutex_unlock(&sdata->local->key_mtx);
283}
284
285static void
286__ieee80211_set_default_mgmt_key(struct ieee80211_sub_if_data *sdata, int idx)
287{
288 struct ieee80211_key *key = NULL;
289
290 assert_key_lock(sdata->local);
291
292 if (idx >= NUM_DEFAULT_KEYS &&
293 idx < NUM_DEFAULT_KEYS + NUM_DEFAULT_MGMT_KEYS)
294 key = key_mtx_dereference(sdata->local, sdata->keys[idx]);
295
296 rcu_assign_pointer(sdata->default_mgmt_key, key);
297
298 ieee80211_debugfs_key_update_default(sdata);
299}
300
301void ieee80211_set_default_mgmt_key(struct ieee80211_sub_if_data *sdata,
302 int idx)
303{
304 mutex_lock(&sdata->local->key_mtx);
305 __ieee80211_set_default_mgmt_key(sdata, idx);
306 mutex_unlock(&sdata->local->key_mtx);
307}
308
309
310static void ieee80211_key_replace(struct ieee80211_sub_if_data *sdata,
311 struct sta_info *sta,
312 bool pairwise,
313 struct ieee80211_key *old,
314 struct ieee80211_key *new)
315{
316 int idx;
317 bool defunikey, defmultikey, defmgmtkey;
318
319
320 if (WARN_ON(!new && !old))
321 return;
322
323 if (new)
324 list_add_tail_rcu(&new->list, &sdata->key_list);
325
326 WARN_ON(new && old && new->conf.keyidx != old->conf.keyidx);
327
328 if (old)
329 idx = old->conf.keyidx;
330 else
331 idx = new->conf.keyidx;
332
333 if (sta) {
334 if (pairwise) {
335 rcu_assign_pointer(sta->ptk[idx], new);
336 sta->ptk_idx = idx;
337 ieee80211_check_fast_xmit(sta);
338 } else {
339 rcu_assign_pointer(sta->gtk[idx], new);
340 }
341 } else {
342 defunikey = old &&
343 old == key_mtx_dereference(sdata->local,
344 sdata->default_unicast_key);
345 defmultikey = old &&
346 old == key_mtx_dereference(sdata->local,
347 sdata->default_multicast_key);
348 defmgmtkey = old &&
349 old == key_mtx_dereference(sdata->local,
350 sdata->default_mgmt_key);
351
352 if (defunikey && !new)
353 __ieee80211_set_default_key(sdata, -1, true, false);
354 if (defmultikey && !new)
355 __ieee80211_set_default_key(sdata, -1, false, true);
356 if (defmgmtkey && !new)
357 __ieee80211_set_default_mgmt_key(sdata, -1);
358
359 rcu_assign_pointer(sdata->keys[idx], new);
360 if (defunikey && new)
361 __ieee80211_set_default_key(sdata, new->conf.keyidx,
362 true, false);
363 if (defmultikey && new)
364 __ieee80211_set_default_key(sdata, new->conf.keyidx,
365 false, true);
366 if (defmgmtkey && new)
367 __ieee80211_set_default_mgmt_key(sdata,
368 new->conf.keyidx);
369 }
370
371 if (old)
372 list_del_rcu(&old->list);
373}
374
375struct ieee80211_key *
376ieee80211_key_alloc(u32 cipher, int idx, size_t key_len,
377 const u8 *key_data,
378 size_t seq_len, const u8 *seq,
379 const struct ieee80211_cipher_scheme *cs)
380{
381 struct ieee80211_key *key;
382 int i, j, err;
383
384 if (WARN_ON(idx < 0 || idx >= NUM_DEFAULT_KEYS + NUM_DEFAULT_MGMT_KEYS))
385 return ERR_PTR(-EINVAL);
386
387 key = kzalloc(sizeof(struct ieee80211_key) + key_len, GFP_KERNEL);
388 if (!key)
389 return ERR_PTR(-ENOMEM);
390
391
392
393
394
395 key->conf.flags = 0;
396 key->flags = 0;
397
398 key->conf.cipher = cipher;
399 key->conf.keyidx = idx;
400 key->conf.keylen = key_len;
401 switch (cipher) {
402 case WLAN_CIPHER_SUITE_WEP40:
403 case WLAN_CIPHER_SUITE_WEP104:
404 key->conf.iv_len = IEEE80211_WEP_IV_LEN;
405 key->conf.icv_len = IEEE80211_WEP_ICV_LEN;
406 break;
407 case WLAN_CIPHER_SUITE_TKIP:
408 key->conf.iv_len = IEEE80211_TKIP_IV_LEN;
409 key->conf.icv_len = IEEE80211_TKIP_ICV_LEN;
410 if (seq) {
411 for (i = 0; i < IEEE80211_NUM_TIDS; i++) {
412 key->u.tkip.rx[i].iv32 =
413 get_unaligned_le32(&seq[2]);
414 key->u.tkip.rx[i].iv16 =
415 get_unaligned_le16(seq);
416 }
417 }
418 spin_lock_init(&key->u.tkip.txlock);
419 break;
420 case WLAN_CIPHER_SUITE_CCMP:
421 key->conf.iv_len = IEEE80211_CCMP_HDR_LEN;
422 key->conf.icv_len = IEEE80211_CCMP_MIC_LEN;
423 if (seq) {
424 for (i = 0; i < IEEE80211_NUM_TIDS + 1; i++)
425 for (j = 0; j < IEEE80211_CCMP_PN_LEN; j++)
426 key->u.ccmp.rx_pn[i][j] =
427 seq[IEEE80211_CCMP_PN_LEN - j - 1];
428 }
429
430
431
432
433 key->u.ccmp.tfm = ieee80211_aes_key_setup_encrypt(
434 key_data, key_len, IEEE80211_CCMP_MIC_LEN);
435 if (IS_ERR(key->u.ccmp.tfm)) {
436 err = PTR_ERR(key->u.ccmp.tfm);
437 kfree(key);
438 return ERR_PTR(err);
439 }
440 break;
441 case WLAN_CIPHER_SUITE_CCMP_256:
442 key->conf.iv_len = IEEE80211_CCMP_256_HDR_LEN;
443 key->conf.icv_len = IEEE80211_CCMP_256_MIC_LEN;
444 for (i = 0; seq && i < IEEE80211_NUM_TIDS + 1; i++)
445 for (j = 0; j < IEEE80211_CCMP_256_PN_LEN; j++)
446 key->u.ccmp.rx_pn[i][j] =
447 seq[IEEE80211_CCMP_256_PN_LEN - j - 1];
448
449
450
451 key->u.ccmp.tfm = ieee80211_aes_key_setup_encrypt(
452 key_data, key_len, IEEE80211_CCMP_256_MIC_LEN);
453 if (IS_ERR(key->u.ccmp.tfm)) {
454 err = PTR_ERR(key->u.ccmp.tfm);
455 kfree(key);
456 return ERR_PTR(err);
457 }
458 break;
459 case WLAN_CIPHER_SUITE_AES_CMAC:
460 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
461 key->conf.iv_len = 0;
462 if (cipher == WLAN_CIPHER_SUITE_AES_CMAC)
463 key->conf.icv_len = sizeof(struct ieee80211_mmie);
464 else
465 key->conf.icv_len = sizeof(struct ieee80211_mmie_16);
466 if (seq)
467 for (j = 0; j < IEEE80211_CMAC_PN_LEN; j++)
468 key->u.aes_cmac.rx_pn[j] =
469 seq[IEEE80211_CMAC_PN_LEN - j - 1];
470
471
472
473
474 key->u.aes_cmac.tfm =
475 ieee80211_aes_cmac_key_setup(key_data, key_len);
476 if (IS_ERR(key->u.aes_cmac.tfm)) {
477 err = PTR_ERR(key->u.aes_cmac.tfm);
478 kfree(key);
479 return ERR_PTR(err);
480 }
481 break;
482 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
483 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
484 key->conf.iv_len = 0;
485 key->conf.icv_len = sizeof(struct ieee80211_mmie_16);
486 if (seq)
487 for (j = 0; j < IEEE80211_GMAC_PN_LEN; j++)
488 key->u.aes_gmac.rx_pn[j] =
489 seq[IEEE80211_GMAC_PN_LEN - j - 1];
490
491
492
493 key->u.aes_gmac.tfm =
494 ieee80211_aes_gmac_key_setup(key_data, key_len);
495 if (IS_ERR(key->u.aes_gmac.tfm)) {
496 err = PTR_ERR(key->u.aes_gmac.tfm);
497 kfree(key);
498 return ERR_PTR(err);
499 }
500 break;
501 case WLAN_CIPHER_SUITE_GCMP:
502 case WLAN_CIPHER_SUITE_GCMP_256:
503 key->conf.iv_len = IEEE80211_GCMP_HDR_LEN;
504 key->conf.icv_len = IEEE80211_GCMP_MIC_LEN;
505 for (i = 0; seq && i < IEEE80211_NUM_TIDS + 1; i++)
506 for (j = 0; j < IEEE80211_GCMP_PN_LEN; j++)
507 key->u.gcmp.rx_pn[i][j] =
508 seq[IEEE80211_GCMP_PN_LEN - j - 1];
509
510
511
512 key->u.gcmp.tfm = ieee80211_aes_gcm_key_setup_encrypt(key_data,
513 key_len);
514 if (IS_ERR(key->u.gcmp.tfm)) {
515 err = PTR_ERR(key->u.gcmp.tfm);
516 kfree(key);
517 return ERR_PTR(err);
518 }
519 break;
520 default:
521 if (cs) {
522 if (seq_len && seq_len != cs->pn_len) {
523 kfree(key);
524 return ERR_PTR(-EINVAL);
525 }
526
527 key->conf.iv_len = cs->hdr_len;
528 key->conf.icv_len = cs->mic_len;
529 for (i = 0; i < IEEE80211_NUM_TIDS + 1; i++)
530 for (j = 0; j < seq_len; j++)
531 key->u.gen.rx_pn[i][j] =
532 seq[seq_len - j - 1];
533 key->flags |= KEY_FLAG_CIPHER_SCHEME;
534 }
535 }
536 memcpy(key->conf.key, key_data, key_len);
537 INIT_LIST_HEAD(&key->list);
538
539 return key;
540}
541
542static void ieee80211_key_free_common(struct ieee80211_key *key)
543{
544 switch (key->conf.cipher) {
545 case WLAN_CIPHER_SUITE_CCMP:
546 case WLAN_CIPHER_SUITE_CCMP_256:
547 ieee80211_aes_key_free(key->u.ccmp.tfm);
548 break;
549 case WLAN_CIPHER_SUITE_AES_CMAC:
550 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
551 ieee80211_aes_cmac_key_free(key->u.aes_cmac.tfm);
552 break;
553 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
554 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
555 ieee80211_aes_gmac_key_free(key->u.aes_gmac.tfm);
556 break;
557 case WLAN_CIPHER_SUITE_GCMP:
558 case WLAN_CIPHER_SUITE_GCMP_256:
559 ieee80211_aes_gcm_key_free(key->u.gcmp.tfm);
560 break;
561 }
562 kzfree(key);
563}
564
565static void __ieee80211_key_destroy(struct ieee80211_key *key,
566 bool delay_tailroom)
567{
568 if (key->local)
569 ieee80211_key_disable_hw_accel(key);
570
571 if (key->local) {
572 struct ieee80211_sub_if_data *sdata = key->sdata;
573
574 ieee80211_debugfs_key_remove(key);
575
576 if (delay_tailroom) {
577
578 sdata->crypto_tx_tailroom_pending_dec++;
579 schedule_delayed_work(&sdata->dec_tailroom_needed_wk,
580 HZ/2);
581 } else {
582 decrease_tailroom_need_count(sdata, 1);
583 }
584 }
585
586 ieee80211_key_free_common(key);
587}
588
589static void ieee80211_key_destroy(struct ieee80211_key *key,
590 bool delay_tailroom)
591{
592 if (!key)
593 return;
594
595
596
597
598
599 synchronize_net();
600
601 __ieee80211_key_destroy(key, delay_tailroom);
602}
603
604void ieee80211_key_free_unused(struct ieee80211_key *key)
605{
606 WARN_ON(key->sdata || key->local);
607 ieee80211_key_free_common(key);
608}
609
610int ieee80211_key_link(struct ieee80211_key *key,
611 struct ieee80211_sub_if_data *sdata,
612 struct sta_info *sta)
613{
614 struct ieee80211_local *local = sdata->local;
615 struct ieee80211_key *old_key;
616 int idx, ret;
617 bool pairwise;
618
619 pairwise = key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE;
620 idx = key->conf.keyidx;
621 key->local = sdata->local;
622 key->sdata = sdata;
623 key->sta = sta;
624
625 mutex_lock(&sdata->local->key_mtx);
626
627 if (sta && pairwise)
628 old_key = key_mtx_dereference(sdata->local, sta->ptk[idx]);
629 else if (sta)
630 old_key = key_mtx_dereference(sdata->local, sta->gtk[idx]);
631 else
632 old_key = key_mtx_dereference(sdata->local, sdata->keys[idx]);
633
634 increment_tailroom_need_count(sdata);
635
636 ieee80211_key_replace(sdata, sta, pairwise, old_key, key);
637 ieee80211_key_destroy(old_key, true);
638
639 ieee80211_debugfs_key_add(key);
640
641 if (!local->wowlan) {
642 ret = ieee80211_key_enable_hw_accel(key);
643 if (ret)
644 ieee80211_key_free(key, true);
645 } else {
646 ret = 0;
647 }
648
649 mutex_unlock(&sdata->local->key_mtx);
650
651 return ret;
652}
653
654void ieee80211_key_free(struct ieee80211_key *key, bool delay_tailroom)
655{
656 if (!key)
657 return;
658
659
660
661
662 if (key->sdata)
663 ieee80211_key_replace(key->sdata, key->sta,
664 key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE,
665 key, NULL);
666 ieee80211_key_destroy(key, delay_tailroom);
667}
668
669void ieee80211_enable_keys(struct ieee80211_sub_if_data *sdata)
670{
671 struct ieee80211_key *key;
672 struct ieee80211_sub_if_data *vlan;
673
674 ASSERT_RTNL();
675
676 if (WARN_ON(!ieee80211_sdata_running(sdata)))
677 return;
678
679 mutex_lock(&sdata->local->key_mtx);
680
681 WARN_ON_ONCE(sdata->crypto_tx_tailroom_needed_cnt ||
682 sdata->crypto_tx_tailroom_pending_dec);
683
684 if (sdata->vif.type == NL80211_IFTYPE_AP) {
685 list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
686 WARN_ON_ONCE(vlan->crypto_tx_tailroom_needed_cnt ||
687 vlan->crypto_tx_tailroom_pending_dec);
688 }
689
690 list_for_each_entry(key, &sdata->key_list, list) {
691 increment_tailroom_need_count(sdata);
692 ieee80211_key_enable_hw_accel(key);
693 }
694
695 mutex_unlock(&sdata->local->key_mtx);
696}
697
698void ieee80211_reset_crypto_tx_tailroom(struct ieee80211_sub_if_data *sdata)
699{
700 struct ieee80211_sub_if_data *vlan;
701
702 mutex_lock(&sdata->local->key_mtx);
703
704 sdata->crypto_tx_tailroom_needed_cnt = 0;
705
706 if (sdata->vif.type == NL80211_IFTYPE_AP) {
707 list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
708 vlan->crypto_tx_tailroom_needed_cnt = 0;
709 }
710
711 mutex_unlock(&sdata->local->key_mtx);
712}
713
714void ieee80211_iter_keys(struct ieee80211_hw *hw,
715 struct ieee80211_vif *vif,
716 void (*iter)(struct ieee80211_hw *hw,
717 struct ieee80211_vif *vif,
718 struct ieee80211_sta *sta,
719 struct ieee80211_key_conf *key,
720 void *data),
721 void *iter_data)
722{
723 struct ieee80211_local *local = hw_to_local(hw);
724 struct ieee80211_key *key, *tmp;
725 struct ieee80211_sub_if_data *sdata;
726
727 ASSERT_RTNL();
728
729 mutex_lock(&local->key_mtx);
730 if (vif) {
731 sdata = vif_to_sdata(vif);
732 list_for_each_entry_safe(key, tmp, &sdata->key_list, list)
733 iter(hw, &sdata->vif,
734 key->sta ? &key->sta->sta : NULL,
735 &key->conf, iter_data);
736 } else {
737 list_for_each_entry(sdata, &local->interfaces, list)
738 list_for_each_entry_safe(key, tmp,
739 &sdata->key_list, list)
740 iter(hw, &sdata->vif,
741 key->sta ? &key->sta->sta : NULL,
742 &key->conf, iter_data);
743 }
744 mutex_unlock(&local->key_mtx);
745}
746EXPORT_SYMBOL(ieee80211_iter_keys);
747
748static void
749_ieee80211_iter_keys_rcu(struct ieee80211_hw *hw,
750 struct ieee80211_sub_if_data *sdata,
751 void (*iter)(struct ieee80211_hw *hw,
752 struct ieee80211_vif *vif,
753 struct ieee80211_sta *sta,
754 struct ieee80211_key_conf *key,
755 void *data),
756 void *iter_data)
757{
758 struct ieee80211_key *key;
759
760 list_for_each_entry_rcu(key, &sdata->key_list, list) {
761
762 if (key->sta && key->sta->removed)
763 continue;
764 if (!(key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE))
765 continue;
766
767 iter(hw, &sdata->vif,
768 key->sta ? &key->sta->sta : NULL,
769 &key->conf, iter_data);
770 }
771}
772
773void ieee80211_iter_keys_rcu(struct ieee80211_hw *hw,
774 struct ieee80211_vif *vif,
775 void (*iter)(struct ieee80211_hw *hw,
776 struct ieee80211_vif *vif,
777 struct ieee80211_sta *sta,
778 struct ieee80211_key_conf *key,
779 void *data),
780 void *iter_data)
781{
782 struct ieee80211_local *local = hw_to_local(hw);
783 struct ieee80211_sub_if_data *sdata;
784
785 if (vif) {
786 sdata = vif_to_sdata(vif);
787 _ieee80211_iter_keys_rcu(hw, sdata, iter, iter_data);
788 } else {
789 list_for_each_entry_rcu(sdata, &local->interfaces, list)
790 _ieee80211_iter_keys_rcu(hw, sdata, iter, iter_data);
791 }
792}
793EXPORT_SYMBOL(ieee80211_iter_keys_rcu);
794
795static void ieee80211_free_keys_iface(struct ieee80211_sub_if_data *sdata,
796 struct list_head *keys)
797{
798 struct ieee80211_key *key, *tmp;
799
800 decrease_tailroom_need_count(sdata,
801 sdata->crypto_tx_tailroom_pending_dec);
802 sdata->crypto_tx_tailroom_pending_dec = 0;
803
804 ieee80211_debugfs_key_remove_mgmt_default(sdata);
805
806 list_for_each_entry_safe(key, tmp, &sdata->key_list, list) {
807 ieee80211_key_replace(key->sdata, key->sta,
808 key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE,
809 key, NULL);
810 list_add_tail(&key->list, keys);
811 }
812
813 ieee80211_debugfs_key_update_default(sdata);
814}
815
816void ieee80211_free_keys(struct ieee80211_sub_if_data *sdata,
817 bool force_synchronize)
818{
819 struct ieee80211_local *local = sdata->local;
820 struct ieee80211_sub_if_data *vlan;
821 struct ieee80211_sub_if_data *master;
822 struct ieee80211_key *key, *tmp;
823 LIST_HEAD(keys);
824
825 cancel_delayed_work_sync(&sdata->dec_tailroom_needed_wk);
826
827 mutex_lock(&local->key_mtx);
828
829 ieee80211_free_keys_iface(sdata, &keys);
830
831 if (sdata->vif.type == NL80211_IFTYPE_AP) {
832 list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
833 ieee80211_free_keys_iface(vlan, &keys);
834 }
835
836 if (!list_empty(&keys) || force_synchronize)
837 synchronize_net();
838 list_for_each_entry_safe(key, tmp, &keys, list)
839 __ieee80211_key_destroy(key, false);
840
841 if (sdata->vif.type == NL80211_IFTYPE_AP_VLAN) {
842 if (sdata->bss) {
843 master = container_of(sdata->bss,
844 struct ieee80211_sub_if_data,
845 u.ap);
846
847 WARN_ON_ONCE(sdata->crypto_tx_tailroom_needed_cnt !=
848 master->crypto_tx_tailroom_needed_cnt);
849 }
850 } else {
851 WARN_ON_ONCE(sdata->crypto_tx_tailroom_needed_cnt ||
852 sdata->crypto_tx_tailroom_pending_dec);
853 }
854
855 if (sdata->vif.type == NL80211_IFTYPE_AP) {
856 list_for_each_entry(vlan, &sdata->u.ap.vlans, u.vlan.list)
857 WARN_ON_ONCE(vlan->crypto_tx_tailroom_needed_cnt ||
858 vlan->crypto_tx_tailroom_pending_dec);
859 }
860
861 mutex_unlock(&local->key_mtx);
862}
863
864void ieee80211_free_sta_keys(struct ieee80211_local *local,
865 struct sta_info *sta)
866{
867 struct ieee80211_key *key;
868 int i;
869
870 mutex_lock(&local->key_mtx);
871 for (i = 0; i < ARRAY_SIZE(sta->gtk); i++) {
872 key = key_mtx_dereference(local, sta->gtk[i]);
873 if (!key)
874 continue;
875 ieee80211_key_replace(key->sdata, key->sta,
876 key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE,
877 key, NULL);
878 __ieee80211_key_destroy(key, true);
879 }
880
881 for (i = 0; i < NUM_DEFAULT_KEYS; i++) {
882 key = key_mtx_dereference(local, sta->ptk[i]);
883 if (!key)
884 continue;
885 ieee80211_key_replace(key->sdata, key->sta,
886 key->conf.flags & IEEE80211_KEY_FLAG_PAIRWISE,
887 key, NULL);
888 __ieee80211_key_destroy(key, true);
889 }
890
891 mutex_unlock(&local->key_mtx);
892}
893
894void ieee80211_delayed_tailroom_dec(struct work_struct *wk)
895{
896 struct ieee80211_sub_if_data *sdata;
897
898 sdata = container_of(wk, struct ieee80211_sub_if_data,
899 dec_tailroom_needed_wk.work);
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917 mutex_lock(&sdata->local->key_mtx);
918 decrease_tailroom_need_count(sdata,
919 sdata->crypto_tx_tailroom_pending_dec);
920 sdata->crypto_tx_tailroom_pending_dec = 0;
921 mutex_unlock(&sdata->local->key_mtx);
922}
923
924void ieee80211_gtk_rekey_notify(struct ieee80211_vif *vif, const u8 *bssid,
925 const u8 *replay_ctr, gfp_t gfp)
926{
927 struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
928
929 trace_api_gtk_rekey_notify(sdata, bssid, replay_ctr);
930
931 cfg80211_gtk_rekey_notify(sdata->dev, bssid, replay_ctr, gfp);
932}
933EXPORT_SYMBOL_GPL(ieee80211_gtk_rekey_notify);
934
935void ieee80211_get_key_tx_seq(struct ieee80211_key_conf *keyconf,
936 struct ieee80211_key_seq *seq)
937{
938 struct ieee80211_key *key;
939 u64 pn64;
940
941 if (WARN_ON(!(keyconf->flags & IEEE80211_KEY_FLAG_GENERATE_IV)))
942 return;
943
944 key = container_of(keyconf, struct ieee80211_key, conf);
945
946 switch (key->conf.cipher) {
947 case WLAN_CIPHER_SUITE_TKIP:
948 seq->tkip.iv32 = key->u.tkip.tx.iv32;
949 seq->tkip.iv16 = key->u.tkip.tx.iv16;
950 break;
951 case WLAN_CIPHER_SUITE_CCMP:
952 case WLAN_CIPHER_SUITE_CCMP_256:
953 case WLAN_CIPHER_SUITE_AES_CMAC:
954 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
955 BUILD_BUG_ON(offsetof(typeof(*seq), ccmp) !=
956 offsetof(typeof(*seq), aes_cmac));
957 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
958 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
959 BUILD_BUG_ON(offsetof(typeof(*seq), ccmp) !=
960 offsetof(typeof(*seq), aes_gmac));
961 case WLAN_CIPHER_SUITE_GCMP:
962 case WLAN_CIPHER_SUITE_GCMP_256:
963 BUILD_BUG_ON(offsetof(typeof(*seq), ccmp) !=
964 offsetof(typeof(*seq), gcmp));
965 pn64 = atomic64_read(&key->conf.tx_pn);
966 seq->ccmp.pn[5] = pn64;
967 seq->ccmp.pn[4] = pn64 >> 8;
968 seq->ccmp.pn[3] = pn64 >> 16;
969 seq->ccmp.pn[2] = pn64 >> 24;
970 seq->ccmp.pn[1] = pn64 >> 32;
971 seq->ccmp.pn[0] = pn64 >> 40;
972 break;
973 default:
974 WARN_ON(1);
975 }
976}
977EXPORT_SYMBOL(ieee80211_get_key_tx_seq);
978
979void ieee80211_get_key_rx_seq(struct ieee80211_key_conf *keyconf,
980 int tid, struct ieee80211_key_seq *seq)
981{
982 struct ieee80211_key *key;
983 const u8 *pn;
984
985 key = container_of(keyconf, struct ieee80211_key, conf);
986
987 switch (key->conf.cipher) {
988 case WLAN_CIPHER_SUITE_TKIP:
989 if (WARN_ON(tid < 0 || tid >= IEEE80211_NUM_TIDS))
990 return;
991 seq->tkip.iv32 = key->u.tkip.rx[tid].iv32;
992 seq->tkip.iv16 = key->u.tkip.rx[tid].iv16;
993 break;
994 case WLAN_CIPHER_SUITE_CCMP:
995 case WLAN_CIPHER_SUITE_CCMP_256:
996 if (WARN_ON(tid < -1 || tid >= IEEE80211_NUM_TIDS))
997 return;
998 if (tid < 0)
999 pn = key->u.ccmp.rx_pn[IEEE80211_NUM_TIDS];
1000 else
1001 pn = key->u.ccmp.rx_pn[tid];
1002 memcpy(seq->ccmp.pn, pn, IEEE80211_CCMP_PN_LEN);
1003 break;
1004 case WLAN_CIPHER_SUITE_AES_CMAC:
1005 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
1006 if (WARN_ON(tid != 0))
1007 return;
1008 pn = key->u.aes_cmac.rx_pn;
1009 memcpy(seq->aes_cmac.pn, pn, IEEE80211_CMAC_PN_LEN);
1010 break;
1011 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
1012 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
1013 if (WARN_ON(tid != 0))
1014 return;
1015 pn = key->u.aes_gmac.rx_pn;
1016 memcpy(seq->aes_gmac.pn, pn, IEEE80211_GMAC_PN_LEN);
1017 break;
1018 case WLAN_CIPHER_SUITE_GCMP:
1019 case WLAN_CIPHER_SUITE_GCMP_256:
1020 if (WARN_ON(tid < -1 || tid >= IEEE80211_NUM_TIDS))
1021 return;
1022 if (tid < 0)
1023 pn = key->u.gcmp.rx_pn[IEEE80211_NUM_TIDS];
1024 else
1025 pn = key->u.gcmp.rx_pn[tid];
1026 memcpy(seq->gcmp.pn, pn, IEEE80211_GCMP_PN_LEN);
1027 break;
1028 }
1029}
1030EXPORT_SYMBOL(ieee80211_get_key_rx_seq);
1031
1032void ieee80211_set_key_tx_seq(struct ieee80211_key_conf *keyconf,
1033 struct ieee80211_key_seq *seq)
1034{
1035 struct ieee80211_key *key;
1036 u64 pn64;
1037
1038 key = container_of(keyconf, struct ieee80211_key, conf);
1039
1040 switch (key->conf.cipher) {
1041 case WLAN_CIPHER_SUITE_TKIP:
1042 key->u.tkip.tx.iv32 = seq->tkip.iv32;
1043 key->u.tkip.tx.iv16 = seq->tkip.iv16;
1044 break;
1045 case WLAN_CIPHER_SUITE_CCMP:
1046 case WLAN_CIPHER_SUITE_CCMP_256:
1047 case WLAN_CIPHER_SUITE_AES_CMAC:
1048 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
1049 BUILD_BUG_ON(offsetof(typeof(*seq), ccmp) !=
1050 offsetof(typeof(*seq), aes_cmac));
1051 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
1052 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
1053 BUILD_BUG_ON(offsetof(typeof(*seq), ccmp) !=
1054 offsetof(typeof(*seq), aes_gmac));
1055 case WLAN_CIPHER_SUITE_GCMP:
1056 case WLAN_CIPHER_SUITE_GCMP_256:
1057 BUILD_BUG_ON(offsetof(typeof(*seq), ccmp) !=
1058 offsetof(typeof(*seq), gcmp));
1059 pn64 = (u64)seq->ccmp.pn[5] |
1060 ((u64)seq->ccmp.pn[4] << 8) |
1061 ((u64)seq->ccmp.pn[3] << 16) |
1062 ((u64)seq->ccmp.pn[2] << 24) |
1063 ((u64)seq->ccmp.pn[1] << 32) |
1064 ((u64)seq->ccmp.pn[0] << 40);
1065 atomic64_set(&key->conf.tx_pn, pn64);
1066 break;
1067 default:
1068 WARN_ON(1);
1069 break;
1070 }
1071}
1072EXPORT_SYMBOL_GPL(ieee80211_set_key_tx_seq);
1073
1074void ieee80211_set_key_rx_seq(struct ieee80211_key_conf *keyconf,
1075 int tid, struct ieee80211_key_seq *seq)
1076{
1077 struct ieee80211_key *key;
1078 u8 *pn;
1079
1080 key = container_of(keyconf, struct ieee80211_key, conf);
1081
1082 switch (key->conf.cipher) {
1083 case WLAN_CIPHER_SUITE_TKIP:
1084 if (WARN_ON(tid < 0 || tid >= IEEE80211_NUM_TIDS))
1085 return;
1086 key->u.tkip.rx[tid].iv32 = seq->tkip.iv32;
1087 key->u.tkip.rx[tid].iv16 = seq->tkip.iv16;
1088 break;
1089 case WLAN_CIPHER_SUITE_CCMP:
1090 case WLAN_CIPHER_SUITE_CCMP_256:
1091 if (WARN_ON(tid < -1 || tid >= IEEE80211_NUM_TIDS))
1092 return;
1093 if (tid < 0)
1094 pn = key->u.ccmp.rx_pn[IEEE80211_NUM_TIDS];
1095 else
1096 pn = key->u.ccmp.rx_pn[tid];
1097 memcpy(pn, seq->ccmp.pn, IEEE80211_CCMP_PN_LEN);
1098 break;
1099 case WLAN_CIPHER_SUITE_AES_CMAC:
1100 case WLAN_CIPHER_SUITE_BIP_CMAC_256:
1101 if (WARN_ON(tid != 0))
1102 return;
1103 pn = key->u.aes_cmac.rx_pn;
1104 memcpy(pn, seq->aes_cmac.pn, IEEE80211_CMAC_PN_LEN);
1105 break;
1106 case WLAN_CIPHER_SUITE_BIP_GMAC_128:
1107 case WLAN_CIPHER_SUITE_BIP_GMAC_256:
1108 if (WARN_ON(tid != 0))
1109 return;
1110 pn = key->u.aes_gmac.rx_pn;
1111 memcpy(pn, seq->aes_gmac.pn, IEEE80211_GMAC_PN_LEN);
1112 break;
1113 case WLAN_CIPHER_SUITE_GCMP:
1114 case WLAN_CIPHER_SUITE_GCMP_256:
1115 if (WARN_ON(tid < -1 || tid >= IEEE80211_NUM_TIDS))
1116 return;
1117 if (tid < 0)
1118 pn = key->u.gcmp.rx_pn[IEEE80211_NUM_TIDS];
1119 else
1120 pn = key->u.gcmp.rx_pn[tid];
1121 memcpy(pn, seq->gcmp.pn, IEEE80211_GCMP_PN_LEN);
1122 break;
1123 default:
1124 WARN_ON(1);
1125 break;
1126 }
1127}
1128EXPORT_SYMBOL_GPL(ieee80211_set_key_rx_seq);
1129
1130void ieee80211_remove_key(struct ieee80211_key_conf *keyconf)
1131{
1132 struct ieee80211_key *key;
1133
1134 key = container_of(keyconf, struct ieee80211_key, conf);
1135
1136 assert_key_lock(key->local);
1137
1138
1139
1140
1141
1142 if (key->flags & KEY_FLAG_UPLOADED_TO_HARDWARE) {
1143 key->flags &= ~KEY_FLAG_UPLOADED_TO_HARDWARE;
1144
1145 if (!((key->conf.flags & IEEE80211_KEY_FLAG_GENERATE_MMIC) ||
1146 (key->conf.flags & IEEE80211_KEY_FLAG_RESERVE_TAILROOM)))
1147 increment_tailroom_need_count(key->sdata);
1148 }
1149
1150 ieee80211_key_free(key, false);
1151}
1152EXPORT_SYMBOL_GPL(ieee80211_remove_key);
1153
1154struct ieee80211_key_conf *
1155ieee80211_gtk_rekey_add(struct ieee80211_vif *vif,
1156 struct ieee80211_key_conf *keyconf)
1157{
1158 struct ieee80211_sub_if_data *sdata = vif_to_sdata(vif);
1159 struct ieee80211_local *local = sdata->local;
1160 struct ieee80211_key *key;
1161 int err;
1162
1163 if (WARN_ON(!local->wowlan))
1164 return ERR_PTR(-EINVAL);
1165
1166 if (WARN_ON(vif->type != NL80211_IFTYPE_STATION))
1167 return ERR_PTR(-EINVAL);
1168
1169 key = ieee80211_key_alloc(keyconf->cipher, keyconf->keyidx,
1170 keyconf->keylen, keyconf->key,
1171 0, NULL, NULL);
1172 if (IS_ERR(key))
1173 return ERR_CAST(key);
1174
1175 if (sdata->u.mgd.mfp != IEEE80211_MFP_DISABLED)
1176 key->conf.flags |= IEEE80211_KEY_FLAG_RX_MGMT;
1177
1178 err = ieee80211_key_link(key, sdata, NULL);
1179 if (err)
1180 return ERR_PTR(err);
1181
1182 return &key->conf;
1183}
1184EXPORT_SYMBOL_GPL(ieee80211_gtk_rekey_add);
1185
