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