block: ublk: switch to ioctl command encoding
[linux-block.git] / net / mac802154 / llsec.c
CommitLineData
1802d0be 1// SPDX-License-Identifier: GPL-2.0-only
5d637d5a
PB
2/*
3 * Copyright (C) 2014 Fraunhofer ITWM
4 *
5d637d5a
PB
5 * Written by:
6 * Phoebe Buckheister <phoebe.buckheister@itwm.fraunhofer.de>
7 */
8
9#include <linux/err.h>
10#include <linux/bug.h>
11#include <linux/completion.h>
4ca24aca 12#include <linux/ieee802154.h>
b2d09103
IM
13#include <linux/rculist.h>
14
8a1a2b71 15#include <crypto/aead.h>
96953718 16#include <crypto/skcipher.h>
5d637d5a 17
0f1556bc 18#include "ieee802154_i.h"
5d637d5a
PB
19#include "llsec.h"
20
21static void llsec_key_put(struct mac802154_llsec_key *key);
22static bool llsec_key_id_equal(const struct ieee802154_llsec_key_id *a,
23 const struct ieee802154_llsec_key_id *b);
24
25static void llsec_dev_free(struct mac802154_llsec_device *dev);
26
27void mac802154_llsec_init(struct mac802154_llsec *sec)
28{
29 memset(sec, 0, sizeof(*sec));
30
31 memset(&sec->params.default_key_source, 0xFF, IEEE802154_ADDR_LEN);
32
33 INIT_LIST_HEAD(&sec->table.security_levels);
34 INIT_LIST_HEAD(&sec->table.devices);
35 INIT_LIST_HEAD(&sec->table.keys);
36 hash_init(sec->devices_short);
37 hash_init(sec->devices_hw);
38 rwlock_init(&sec->lock);
39}
40
41void mac802154_llsec_destroy(struct mac802154_llsec *sec)
42{
43 struct ieee802154_llsec_seclevel *sl, *sn;
44 struct ieee802154_llsec_device *dev, *dn;
45 struct ieee802154_llsec_key_entry *key, *kn;
46
47 list_for_each_entry_safe(sl, sn, &sec->table.security_levels, list) {
48 struct mac802154_llsec_seclevel *msl;
49
50 msl = container_of(sl, struct mac802154_llsec_seclevel, level);
51 list_del(&sl->list);
453431a5 52 kfree_sensitive(msl);
5d637d5a
PB
53 }
54
55 list_for_each_entry_safe(dev, dn, &sec->table.devices, list) {
56 struct mac802154_llsec_device *mdev;
57
58 mdev = container_of(dev, struct mac802154_llsec_device, dev);
59 list_del(&dev->list);
60 llsec_dev_free(mdev);
61 }
62
63 list_for_each_entry_safe(key, kn, &sec->table.keys, list) {
64 struct mac802154_llsec_key *mkey;
65
66 mkey = container_of(key->key, struct mac802154_llsec_key, key);
67 list_del(&key->list);
68 llsec_key_put(mkey);
453431a5 69 kfree_sensitive(key);
5d637d5a
PB
70 }
71}
72
5d637d5a
PB
73int mac802154_llsec_get_params(struct mac802154_llsec *sec,
74 struct ieee802154_llsec_params *params)
75{
76 read_lock_bh(&sec->lock);
77 *params = sec->params;
78 read_unlock_bh(&sec->lock);
79
80 return 0;
81}
82
83int mac802154_llsec_set_params(struct mac802154_llsec *sec,
84 const struct ieee802154_llsec_params *params,
85 int changed)
86{
87 write_lock_bh(&sec->lock);
88
89 if (changed & IEEE802154_LLSEC_PARAM_ENABLED)
90 sec->params.enabled = params->enabled;
91 if (changed & IEEE802154_LLSEC_PARAM_FRAME_COUNTER)
92 sec->params.frame_counter = params->frame_counter;
93 if (changed & IEEE802154_LLSEC_PARAM_OUT_LEVEL)
94 sec->params.out_level = params->out_level;
95 if (changed & IEEE802154_LLSEC_PARAM_OUT_KEY)
96 sec->params.out_key = params->out_key;
97 if (changed & IEEE802154_LLSEC_PARAM_KEY_SOURCE)
98 sec->params.default_key_source = params->default_key_source;
99 if (changed & IEEE802154_LLSEC_PARAM_PAN_ID)
100 sec->params.pan_id = params->pan_id;
101 if (changed & IEEE802154_LLSEC_PARAM_HWADDR)
102 sec->params.hwaddr = params->hwaddr;
103 if (changed & IEEE802154_LLSEC_PARAM_COORD_HWADDR)
104 sec->params.coord_hwaddr = params->coord_hwaddr;
105 if (changed & IEEE802154_LLSEC_PARAM_COORD_SHORTADDR)
106 sec->params.coord_shortaddr = params->coord_shortaddr;
107
108 write_unlock_bh(&sec->lock);
109
110 return 0;
111}
112
5d637d5a
PB
113static struct mac802154_llsec_key*
114llsec_key_alloc(const struct ieee802154_llsec_key *template)
115{
116 const int authsizes[3] = { 4, 8, 16 };
117 struct mac802154_llsec_key *key;
118 int i;
119
120 key = kzalloc(sizeof(*key), GFP_KERNEL);
121 if (!key)
122 return NULL;
123
124 kref_init(&key->ref);
125 key->key = *template;
126
127 BUILD_BUG_ON(ARRAY_SIZE(authsizes) != ARRAY_SIZE(key->tfm));
128
129 for (i = 0; i < ARRAY_SIZE(key->tfm); i++) {
130 key->tfm[i] = crypto_alloc_aead("ccm(aes)", 0,
131 CRYPTO_ALG_ASYNC);
89eb6d06 132 if (IS_ERR(key->tfm[i]))
5d637d5a
PB
133 goto err_tfm;
134 if (crypto_aead_setkey(key->tfm[i], template->key,
135 IEEE802154_LLSEC_KEY_SIZE))
136 goto err_tfm;
137 if (crypto_aead_setauthsize(key->tfm[i], authsizes[i]))
138 goto err_tfm;
139 }
140
3103f4a7 141 key->tfm0 = crypto_alloc_sync_skcipher("ctr(aes)", 0, 0);
89eb6d06 142 if (IS_ERR(key->tfm0))
5d637d5a
PB
143 goto err_tfm;
144
3103f4a7 145 if (crypto_sync_skcipher_setkey(key->tfm0, template->key,
96953718 146 IEEE802154_LLSEC_KEY_SIZE))
5d637d5a
PB
147 goto err_tfm0;
148
149 return key;
150
151err_tfm0:
3103f4a7 152 crypto_free_sync_skcipher(key->tfm0);
5d637d5a
PB
153err_tfm:
154 for (i = 0; i < ARRAY_SIZE(key->tfm); i++)
1165affd 155 if (!IS_ERR_OR_NULL(key->tfm[i]))
5d637d5a
PB
156 crypto_free_aead(key->tfm[i]);
157
453431a5 158 kfree_sensitive(key);
5d637d5a
PB
159 return NULL;
160}
161
162static void llsec_key_release(struct kref *ref)
163{
164 struct mac802154_llsec_key *key;
165 int i;
166
167 key = container_of(ref, struct mac802154_llsec_key, ref);
168
169 for (i = 0; i < ARRAY_SIZE(key->tfm); i++)
170 crypto_free_aead(key->tfm[i]);
171
3103f4a7 172 crypto_free_sync_skcipher(key->tfm0);
453431a5 173 kfree_sensitive(key);
5d637d5a
PB
174}
175
176static struct mac802154_llsec_key*
177llsec_key_get(struct mac802154_llsec_key *key)
178{
179 kref_get(&key->ref);
180 return key;
181}
182
183static void llsec_key_put(struct mac802154_llsec_key *key)
184{
185 kref_put(&key->ref, llsec_key_release);
186}
187
188static bool llsec_key_id_equal(const struct ieee802154_llsec_key_id *a,
189 const struct ieee802154_llsec_key_id *b)
190{
191 if (a->mode != b->mode)
192 return false;
193
194 if (a->mode == IEEE802154_SCF_KEY_IMPLICIT)
195 return ieee802154_addr_equal(&a->device_addr, &b->device_addr);
196
197 if (a->id != b->id)
198 return false;
199
200 switch (a->mode) {
53819a6c
PB
201 case IEEE802154_SCF_KEY_INDEX:
202 return true;
5d637d5a
PB
203 case IEEE802154_SCF_KEY_SHORT_INDEX:
204 return a->short_source == b->short_source;
205 case IEEE802154_SCF_KEY_HW_INDEX:
206 return a->extended_source == b->extended_source;
207 }
208
209 return false;
210}
211
212int mac802154_llsec_key_add(struct mac802154_llsec *sec,
213 const struct ieee802154_llsec_key_id *id,
214 const struct ieee802154_llsec_key *key)
215{
216 struct mac802154_llsec_key *mkey = NULL;
217 struct ieee802154_llsec_key_entry *pos, *new;
218
219 if (!(key->frame_types & (1 << IEEE802154_FC_TYPE_MAC_CMD)) &&
220 key->cmd_frame_ids)
221 return -EINVAL;
222
223 list_for_each_entry(pos, &sec->table.keys, list) {
224 if (llsec_key_id_equal(&pos->id, id))
225 return -EEXIST;
226
227 if (memcmp(pos->key->key, key->key,
228 IEEE802154_LLSEC_KEY_SIZE))
229 continue;
230
231 mkey = container_of(pos->key, struct mac802154_llsec_key, key);
232
233 /* Don't allow multiple instances of the same AES key to have
234 * different allowed frame types/command frame ids, as this is
235 * not possible in the 802.15.4 PIB.
236 */
237 if (pos->key->frame_types != key->frame_types ||
238 pos->key->cmd_frame_ids != key->cmd_frame_ids)
239 return -EEXIST;
240
241 break;
242 }
243
244 new = kzalloc(sizeof(*new), GFP_KERNEL);
245 if (!new)
246 return -ENOMEM;
247
248 if (!mkey)
249 mkey = llsec_key_alloc(key);
250 else
251 mkey = llsec_key_get(mkey);
252
253 if (!mkey)
254 goto fail;
255
256 new->id = *id;
257 new->key = &mkey->key;
258
259 list_add_rcu(&new->list, &sec->table.keys);
260
261 return 0;
262
263fail:
453431a5 264 kfree_sensitive(new);
5d637d5a
PB
265 return -ENOMEM;
266}
267
268int mac802154_llsec_key_del(struct mac802154_llsec *sec,
269 const struct ieee802154_llsec_key_id *key)
270{
271 struct ieee802154_llsec_key_entry *pos;
272
273 list_for_each_entry(pos, &sec->table.keys, list) {
274 struct mac802154_llsec_key *mkey;
275
276 mkey = container_of(pos->key, struct mac802154_llsec_key, key);
277
278 if (llsec_key_id_equal(&pos->id, key)) {
fff1f59b 279 list_del_rcu(&pos->list);
5d637d5a
PB
280 llsec_key_put(mkey);
281 return 0;
282 }
283 }
284
285 return -ENOENT;
286}
287
5d637d5a
PB
288static bool llsec_dev_use_shortaddr(__le16 short_addr)
289{
290 return short_addr != cpu_to_le16(IEEE802154_ADDR_UNDEF) &&
291 short_addr != cpu_to_le16(0xffff);
292}
293
294static u32 llsec_dev_hash_short(__le16 short_addr, __le16 pan_id)
295{
1db99604 296 return ((__force u16)short_addr) << 16 | (__force u16)pan_id;
5d637d5a
PB
297}
298
299static u64 llsec_dev_hash_long(__le64 hwaddr)
300{
1db99604 301 return (__force u64)hwaddr;
5d637d5a
PB
302}
303
304static struct mac802154_llsec_device*
305llsec_dev_find_short(struct mac802154_llsec *sec, __le16 short_addr,
306 __le16 pan_id)
307{
308 struct mac802154_llsec_device *dev;
309 u32 key = llsec_dev_hash_short(short_addr, pan_id);
310
311 hash_for_each_possible_rcu(sec->devices_short, dev, bucket_s, key) {
312 if (dev->dev.short_addr == short_addr &&
313 dev->dev.pan_id == pan_id)
314 return dev;
315 }
316
317 return NULL;
318}
319
320static struct mac802154_llsec_device*
321llsec_dev_find_long(struct mac802154_llsec *sec, __le64 hwaddr)
322{
323 struct mac802154_llsec_device *dev;
324 u64 key = llsec_dev_hash_long(hwaddr);
325
326 hash_for_each_possible_rcu(sec->devices_hw, dev, bucket_hw, key) {
327 if (dev->dev.hwaddr == hwaddr)
328 return dev;
329 }
330
331 return NULL;
332}
333
334static void llsec_dev_free(struct mac802154_llsec_device *dev)
335{
336 struct ieee802154_llsec_device_key *pos, *pn;
337 struct mac802154_llsec_device_key *devkey;
338
339 list_for_each_entry_safe(pos, pn, &dev->dev.keys, list) {
340 devkey = container_of(pos, struct mac802154_llsec_device_key,
341 devkey);
342
343 list_del(&pos->list);
453431a5 344 kfree_sensitive(devkey);
5d637d5a
PB
345 }
346
453431a5 347 kfree_sensitive(dev);
5d637d5a
PB
348}
349
350int mac802154_llsec_dev_add(struct mac802154_llsec *sec,
351 const struct ieee802154_llsec_device *dev)
352{
353 struct mac802154_llsec_device *entry;
354 u32 skey = llsec_dev_hash_short(dev->short_addr, dev->pan_id);
355 u64 hwkey = llsec_dev_hash_long(dev->hwaddr);
356
357 BUILD_BUG_ON(sizeof(hwkey) != IEEE802154_ADDR_LEN);
358
359 if ((llsec_dev_use_shortaddr(dev->short_addr) &&
360 llsec_dev_find_short(sec, dev->short_addr, dev->pan_id)) ||
361 llsec_dev_find_long(sec, dev->hwaddr))
362 return -EEXIST;
363
364 entry = kmalloc(sizeof(*entry), GFP_KERNEL);
365 if (!entry)
366 return -ENOMEM;
367
368 entry->dev = *dev;
369 spin_lock_init(&entry->lock);
370 INIT_LIST_HEAD(&entry->dev.keys);
371
372 if (llsec_dev_use_shortaddr(dev->short_addr))
373 hash_add_rcu(sec->devices_short, &entry->bucket_s, skey);
374 else
375 INIT_HLIST_NODE(&entry->bucket_s);
376
377 hash_add_rcu(sec->devices_hw, &entry->bucket_hw, hwkey);
378 list_add_tail_rcu(&entry->dev.list, &sec->table.devices);
379
380 return 0;
381}
382
383static void llsec_dev_free_rcu(struct rcu_head *rcu)
384{
385 llsec_dev_free(container_of(rcu, struct mac802154_llsec_device, rcu));
386}
387
388int mac802154_llsec_dev_del(struct mac802154_llsec *sec, __le64 device_addr)
389{
390 struct mac802154_llsec_device *pos;
391
392 pos = llsec_dev_find_long(sec, device_addr);
393 if (!pos)
394 return -ENOENT;
395
396 hash_del_rcu(&pos->bucket_s);
397 hash_del_rcu(&pos->bucket_hw);
cdd38b21 398 list_del_rcu(&pos->dev.list);
5d637d5a
PB
399 call_rcu(&pos->rcu, llsec_dev_free_rcu);
400
401 return 0;
402}
403
5d637d5a
PB
404static struct mac802154_llsec_device_key*
405llsec_devkey_find(struct mac802154_llsec_device *dev,
406 const struct ieee802154_llsec_key_id *key)
407{
408 struct ieee802154_llsec_device_key *devkey;
409
410 list_for_each_entry_rcu(devkey, &dev->dev.keys, list) {
411 if (!llsec_key_id_equal(key, &devkey->key_id))
412 continue;
413
414 return container_of(devkey, struct mac802154_llsec_device_key,
415 devkey);
416 }
417
418 return NULL;
419}
420
421int mac802154_llsec_devkey_add(struct mac802154_llsec *sec,
422 __le64 dev_addr,
423 const struct ieee802154_llsec_device_key *key)
424{
425 struct mac802154_llsec_device *dev;
426 struct mac802154_llsec_device_key *devkey;
427
428 dev = llsec_dev_find_long(sec, dev_addr);
429
430 if (!dev)
431 return -ENOENT;
432
433 if (llsec_devkey_find(dev, &key->key_id))
434 return -EEXIST;
435
436 devkey = kmalloc(sizeof(*devkey), GFP_KERNEL);
437 if (!devkey)
438 return -ENOMEM;
439
440 devkey->devkey = *key;
441 list_add_tail_rcu(&devkey->devkey.list, &dev->dev.keys);
442 return 0;
443}
444
445int mac802154_llsec_devkey_del(struct mac802154_llsec *sec,
446 __le64 dev_addr,
447 const struct ieee802154_llsec_device_key *key)
448{
449 struct mac802154_llsec_device *dev;
450 struct mac802154_llsec_device_key *devkey;
451
452 dev = llsec_dev_find_long(sec, dev_addr);
453
454 if (!dev)
455 return -ENOENT;
456
457 devkey = llsec_devkey_find(dev, &key->key_id);
458 if (!devkey)
459 return -ENOENT;
460
461 list_del_rcu(&devkey->devkey.list);
462 kfree_rcu(devkey, rcu);
463 return 0;
464}
465
5d637d5a
PB
466static struct mac802154_llsec_seclevel*
467llsec_find_seclevel(const struct mac802154_llsec *sec,
468 const struct ieee802154_llsec_seclevel *sl)
469{
470 struct ieee802154_llsec_seclevel *pos;
471
472 list_for_each_entry(pos, &sec->table.security_levels, list) {
473 if (pos->frame_type != sl->frame_type ||
474 (pos->frame_type == IEEE802154_FC_TYPE_MAC_CMD &&
475 pos->cmd_frame_id != sl->cmd_frame_id) ||
476 pos->device_override != sl->device_override ||
477 pos->sec_levels != sl->sec_levels)
478 continue;
479
480 return container_of(pos, struct mac802154_llsec_seclevel,
481 level);
482 }
483
484 return NULL;
485}
486
487int mac802154_llsec_seclevel_add(struct mac802154_llsec *sec,
488 const struct ieee802154_llsec_seclevel *sl)
489{
490 struct mac802154_llsec_seclevel *entry;
491
492 if (llsec_find_seclevel(sec, sl))
493 return -EEXIST;
494
495 entry = kmalloc(sizeof(*entry), GFP_KERNEL);
496 if (!entry)
497 return -ENOMEM;
498
499 entry->level = *sl;
500
501 list_add_tail_rcu(&entry->level.list, &sec->table.security_levels);
502
503 return 0;
504}
505
506int mac802154_llsec_seclevel_del(struct mac802154_llsec *sec,
507 const struct ieee802154_llsec_seclevel *sl)
508{
509 struct mac802154_llsec_seclevel *pos;
510
511 pos = llsec_find_seclevel(sec, sl);
512 if (!pos)
513 return -ENOENT;
514
515 list_del_rcu(&pos->level.list);
516 kfree_rcu(pos, rcu);
517
518 return 0;
519}
03556e4d 520
03556e4d
PB
521static int llsec_recover_addr(struct mac802154_llsec *sec,
522 struct ieee802154_addr *addr)
523{
524 __le16 caddr = sec->params.coord_shortaddr;
4710d806 525
03556e4d
PB
526 addr->pan_id = sec->params.pan_id;
527
528 if (caddr == cpu_to_le16(IEEE802154_ADDR_BROADCAST)) {
529 return -EINVAL;
530 } else if (caddr == cpu_to_le16(IEEE802154_ADDR_UNDEF)) {
531 addr->extended_addr = sec->params.coord_hwaddr;
532 addr->mode = IEEE802154_ADDR_LONG;
533 } else {
534 addr->short_addr = sec->params.coord_shortaddr;
535 addr->mode = IEEE802154_ADDR_SHORT;
536 }
537
538 return 0;
539}
540
541static struct mac802154_llsec_key*
542llsec_lookup_key(struct mac802154_llsec *sec,
543 const struct ieee802154_hdr *hdr,
544 const struct ieee802154_addr *addr,
545 struct ieee802154_llsec_key_id *key_id)
546{
547 struct ieee802154_addr devaddr = *addr;
548 u8 key_id_mode = hdr->sec.key_id_mode;
549 struct ieee802154_llsec_key_entry *key_entry;
550 struct mac802154_llsec_key *key;
551
552 if (key_id_mode == IEEE802154_SCF_KEY_IMPLICIT &&
553 devaddr.mode == IEEE802154_ADDR_NONE) {
554 if (hdr->fc.type == IEEE802154_FC_TYPE_BEACON) {
555 devaddr.extended_addr = sec->params.coord_hwaddr;
556 devaddr.mode = IEEE802154_ADDR_LONG;
557 } else if (llsec_recover_addr(sec, &devaddr) < 0) {
558 return NULL;
559 }
560 }
561
562 list_for_each_entry_rcu(key_entry, &sec->table.keys, list) {
563 const struct ieee802154_llsec_key_id *id = &key_entry->id;
564
565 if (!(key_entry->key->frame_types & BIT(hdr->fc.type)))
566 continue;
567
568 if (id->mode != key_id_mode)
569 continue;
570
571 if (key_id_mode == IEEE802154_SCF_KEY_IMPLICIT) {
572 if (ieee802154_addr_equal(&devaddr, &id->device_addr))
573 goto found;
574 } else {
575 if (id->id != hdr->sec.key_id)
576 continue;
577
578 if ((key_id_mode == IEEE802154_SCF_KEY_INDEX) ||
579 (key_id_mode == IEEE802154_SCF_KEY_SHORT_INDEX &&
580 id->short_source == hdr->sec.short_src) ||
581 (key_id_mode == IEEE802154_SCF_KEY_HW_INDEX &&
582 id->extended_source == hdr->sec.extended_src))
583 goto found;
584 }
585 }
586
587 return NULL;
588
589found:
590 key = container_of(key_entry->key, struct mac802154_llsec_key, key);
591 if (key_id)
592 *key_id = key_entry->id;
593 return llsec_key_get(key);
594}
595
03556e4d
PB
596static void llsec_geniv(u8 iv[16], __le64 addr,
597 const struct ieee802154_sechdr *sec)
598{
599 __be64 addr_bytes = (__force __be64) swab64((__force u64) addr);
600 __be32 frame_counter = (__force __be32) swab32((__force u32) sec->frame_counter);
601
602 iv[0] = 1; /* L' = L - 1 = 1 */
603 memcpy(iv + 1, &addr_bytes, sizeof(addr_bytes));
604 memcpy(iv + 9, &frame_counter, sizeof(frame_counter));
605 iv[13] = sec->level;
606 iv[14] = 0;
607 iv[15] = 1;
608}
609
610static int
611llsec_do_encrypt_unauth(struct sk_buff *skb, const struct mac802154_llsec *sec,
612 const struct ieee802154_hdr *hdr,
613 struct mac802154_llsec_key *key)
614{
615 u8 iv[16];
616 struct scatterlist src;
3103f4a7 617 SYNC_SKCIPHER_REQUEST_ON_STACK(req, key->tfm0);
3e496266
DP
618 int err, datalen;
619 unsigned char *data;
03556e4d
PB
620
621 llsec_geniv(iv, sec->params.hwaddr, &hdr->sec);
3e496266
DP
622 /* Compute data payload offset and data length */
623 data = skb_mac_header(skb) + skb->mac_len;
624 datalen = skb_tail_pointer(skb) - data;
625 sg_init_one(&src, data, datalen);
626
3103f4a7 627 skcipher_request_set_sync_tfm(req, key->tfm0);
96953718 628 skcipher_request_set_callback(req, 0, NULL, NULL);
3e496266 629 skcipher_request_set_crypt(req, &src, &src, datalen, iv);
96953718
HX
630 err = crypto_skcipher_encrypt(req);
631 skcipher_request_zero(req);
632 return err;
03556e4d
PB
633}
634
635static struct crypto_aead*
636llsec_tfm_by_len(struct mac802154_llsec_key *key, int authlen)
637{
638 int i;
639
640 for (i = 0; i < ARRAY_SIZE(key->tfm); i++)
641 if (crypto_aead_authsize(key->tfm[i]) == authlen)
642 return key->tfm[i];
643
644 BUG();
645}
646
647static int
648llsec_do_encrypt_auth(struct sk_buff *skb, const struct mac802154_llsec *sec,
649 const struct ieee802154_hdr *hdr,
650 struct mac802154_llsec_key *key)
651{
652 u8 iv[16];
653 unsigned char *data;
654 int authlen, assoclen, datalen, rc;
25528fda 655 struct scatterlist sg;
03556e4d
PB
656 struct aead_request *req;
657
658 authlen = ieee802154_sechdr_authtag_len(&hdr->sec);
659 llsec_geniv(iv, sec->params.hwaddr, &hdr->sec);
660
661 req = aead_request_alloc(llsec_tfm_by_len(key, authlen), GFP_ATOMIC);
662 if (!req)
663 return -ENOMEM;
664
03556e4d
PB
665 assoclen = skb->mac_len;
666
667 data = skb_mac_header(skb) + skb->mac_len;
668 datalen = skb_tail_pointer(skb) - data;
669
25528fda
HX
670 skb_put(skb, authlen);
671
672 sg_init_one(&sg, skb_mac_header(skb), assoclen + datalen + authlen);
673
674 if (!(hdr->sec.level & IEEE802154_SCF_SECLEVEL_ENC)) {
03556e4d
PB
675 assoclen += datalen;
676 datalen = 0;
677 }
678
03556e4d 679 aead_request_set_callback(req, 0, NULL, NULL);
25528fda
HX
680 aead_request_set_crypt(req, &sg, &sg, datalen, iv);
681 aead_request_set_ad(req, assoclen);
03556e4d
PB
682
683 rc = crypto_aead_encrypt(req);
684
453431a5 685 kfree_sensitive(req);
03556e4d
PB
686
687 return rc;
688}
689
690static int llsec_do_encrypt(struct sk_buff *skb,
691 const struct mac802154_llsec *sec,
692 const struct ieee802154_hdr *hdr,
693 struct mac802154_llsec_key *key)
694{
695 if (hdr->sec.level == IEEE802154_SCF_SECLEVEL_ENC)
696 return llsec_do_encrypt_unauth(skb, sec, hdr, key);
697 else
698 return llsec_do_encrypt_auth(skb, sec, hdr, key);
699}
700
701int mac802154_llsec_encrypt(struct mac802154_llsec *sec, struct sk_buff *skb)
702{
703 struct ieee802154_hdr hdr;
704 int rc, authlen, hlen;
705 struct mac802154_llsec_key *key;
706 u32 frame_ctr;
707
708 hlen = ieee802154_hdr_pull(skb, &hdr);
709
3accf476
MR
710 /* TODO: control frames security support */
711 if (hlen < 0 ||
712 (hdr.fc.type != IEEE802154_FC_TYPE_DATA &&
713 hdr.fc.type != IEEE802154_FC_TYPE_BEACON))
03556e4d
PB
714 return -EINVAL;
715
91f4aa97
DP
716 if (!hdr.fc.security_enabled ||
717 (hdr.sec.level == IEEE802154_SCF_SECLEVEL_NONE)) {
03556e4d
PB
718 skb_push(skb, hlen);
719 return 0;
720 }
721
722 authlen = ieee802154_sechdr_authtag_len(&hdr.sec);
723
724 if (skb->len + hlen + authlen + IEEE802154_MFR_SIZE > IEEE802154_MTU)
725 return -EMSGSIZE;
726
727 rcu_read_lock();
728
729 read_lock_bh(&sec->lock);
730
731 if (!sec->params.enabled) {
732 rc = -EINVAL;
733 goto fail_read;
734 }
735
736 key = llsec_lookup_key(sec, &hdr, &hdr.dest, NULL);
737 if (!key) {
738 rc = -ENOKEY;
739 goto fail_read;
740 }
741
742 read_unlock_bh(&sec->lock);
743
744 write_lock_bh(&sec->lock);
745
746 frame_ctr = be32_to_cpu(sec->params.frame_counter);
747 hdr.sec.frame_counter = cpu_to_le32(frame_ctr);
748 if (frame_ctr == 0xFFFFFFFF) {
749 write_unlock_bh(&sec->lock);
750 llsec_key_put(key);
751 rc = -EOVERFLOW;
752 goto fail;
753 }
754
755 sec->params.frame_counter = cpu_to_be32(frame_ctr + 1);
756
757 write_unlock_bh(&sec->lock);
758
759 rcu_read_unlock();
760
761 skb->mac_len = ieee802154_hdr_push(skb, &hdr);
762 skb_reset_mac_header(skb);
763
764 rc = llsec_do_encrypt(skb, sec, &hdr, key);
765 llsec_key_put(key);
766
62e9c117 767 return rc;
03556e4d
PB
768
769fail_read:
6f3eabcd 770 read_unlock_bh(&sec->lock);
03556e4d
PB
771fail:
772 rcu_read_unlock();
773 return rc;
774}
4c14a2fb 775
4c14a2fb
PB
776static struct mac802154_llsec_device*
777llsec_lookup_dev(struct mac802154_llsec *sec,
778 const struct ieee802154_addr *addr)
779{
780 struct ieee802154_addr devaddr = *addr;
781 struct mac802154_llsec_device *dev = NULL;
782
783 if (devaddr.mode == IEEE802154_ADDR_NONE &&
784 llsec_recover_addr(sec, &devaddr) < 0)
785 return NULL;
786
787 if (devaddr.mode == IEEE802154_ADDR_SHORT) {
788 u32 key = llsec_dev_hash_short(devaddr.short_addr,
789 devaddr.pan_id);
790
791 hash_for_each_possible_rcu(sec->devices_short, dev,
792 bucket_s, key) {
793 if (dev->dev.pan_id == devaddr.pan_id &&
794 dev->dev.short_addr == devaddr.short_addr)
795 return dev;
796 }
797 } else {
798 u64 key = llsec_dev_hash_long(devaddr.extended_addr);
799
800 hash_for_each_possible_rcu(sec->devices_hw, dev,
801 bucket_hw, key) {
802 if (dev->dev.hwaddr == devaddr.extended_addr)
803 return dev;
804 }
805 }
806
807 return NULL;
808}
809
810static int
811llsec_lookup_seclevel(const struct mac802154_llsec *sec,
812 u8 frame_type, u8 cmd_frame_id,
813 struct ieee802154_llsec_seclevel *rlevel)
814{
815 struct ieee802154_llsec_seclevel *level;
816
817 list_for_each_entry_rcu(level, &sec->table.security_levels, list) {
818 if (level->frame_type == frame_type &&
819 (frame_type != IEEE802154_FC_TYPE_MAC_CMD ||
820 level->cmd_frame_id == cmd_frame_id)) {
821 *rlevel = *level;
822 return 0;
823 }
824 }
825
826 return -EINVAL;
827}
828
829static int
830llsec_do_decrypt_unauth(struct sk_buff *skb, const struct mac802154_llsec *sec,
831 const struct ieee802154_hdr *hdr,
832 struct mac802154_llsec_key *key, __le64 dev_addr)
833{
834 u8 iv[16];
835 unsigned char *data;
836 int datalen;
837 struct scatterlist src;
3103f4a7 838 SYNC_SKCIPHER_REQUEST_ON_STACK(req, key->tfm0);
96953718 839 int err;
4c14a2fb
PB
840
841 llsec_geniv(iv, dev_addr, &hdr->sec);
842 data = skb_mac_header(skb) + skb->mac_len;
843 datalen = skb_tail_pointer(skb) - data;
844
845 sg_init_one(&src, data, datalen);
846
3103f4a7 847 skcipher_request_set_sync_tfm(req, key->tfm0);
96953718
HX
848 skcipher_request_set_callback(req, 0, NULL, NULL);
849 skcipher_request_set_crypt(req, &src, &src, datalen, iv);
850
851 err = crypto_skcipher_decrypt(req);
852 skcipher_request_zero(req);
853 return err;
4c14a2fb
PB
854}
855
856static int
857llsec_do_decrypt_auth(struct sk_buff *skb, const struct mac802154_llsec *sec,
858 const struct ieee802154_hdr *hdr,
859 struct mac802154_llsec_key *key, __le64 dev_addr)
860{
861 u8 iv[16];
862 unsigned char *data;
863 int authlen, datalen, assoclen, rc;
25528fda 864 struct scatterlist sg;
4c14a2fb
PB
865 struct aead_request *req;
866
867 authlen = ieee802154_sechdr_authtag_len(&hdr->sec);
868 llsec_geniv(iv, dev_addr, &hdr->sec);
869
870 req = aead_request_alloc(llsec_tfm_by_len(key, authlen), GFP_ATOMIC);
871 if (!req)
872 return -ENOMEM;
873
4c14a2fb
PB
874 assoclen = skb->mac_len;
875
876 data = skb_mac_header(skb) + skb->mac_len;
877 datalen = skb_tail_pointer(skb) - data;
878
25528fda
HX
879 sg_init_one(&sg, skb_mac_header(skb), assoclen + datalen);
880
881 if (!(hdr->sec.level & IEEE802154_SCF_SECLEVEL_ENC)) {
4c14a2fb 882 assoclen += datalen - authlen;
4c14a2fb
PB
883 datalen = authlen;
884 }
885
4c14a2fb 886 aead_request_set_callback(req, 0, NULL, NULL);
25528fda
HX
887 aead_request_set_crypt(req, &sg, &sg, datalen, iv);
888 aead_request_set_ad(req, assoclen);
4c14a2fb
PB
889
890 rc = crypto_aead_decrypt(req);
891
453431a5 892 kfree_sensitive(req);
4c14a2fb
PB
893 skb_trim(skb, skb->len - authlen);
894
895 return rc;
896}
897
898static int
899llsec_do_decrypt(struct sk_buff *skb, const struct mac802154_llsec *sec,
900 const struct ieee802154_hdr *hdr,
901 struct mac802154_llsec_key *key, __le64 dev_addr)
902{
903 if (hdr->sec.level == IEEE802154_SCF_SECLEVEL_ENC)
904 return llsec_do_decrypt_unauth(skb, sec, hdr, key, dev_addr);
905 else
906 return llsec_do_decrypt_auth(skb, sec, hdr, key, dev_addr);
907}
908
f0f77dc6
PB
909static int
910llsec_update_devkey_record(struct mac802154_llsec_device *dev,
911 const struct ieee802154_llsec_key_id *in_key)
912{
913 struct mac802154_llsec_device_key *devkey;
914
915 devkey = llsec_devkey_find(dev, in_key);
916
917 if (!devkey) {
918 struct mac802154_llsec_device_key *next;
919
920 next = kzalloc(sizeof(*devkey), GFP_ATOMIC);
921 if (!next)
922 return -ENOMEM;
923
924 next->devkey.key_id = *in_key;
925
926 spin_lock_bh(&dev->lock);
927
928 devkey = llsec_devkey_find(dev, in_key);
929 if (!devkey)
930 list_add_rcu(&next->devkey.list, &dev->dev.keys);
931 else
453431a5 932 kfree_sensitive(next);
f0f77dc6
PB
933
934 spin_unlock_bh(&dev->lock);
935 }
936
937 return 0;
938}
939
4c14a2fb
PB
940static int
941llsec_update_devkey_info(struct mac802154_llsec_device *dev,
942 const struct ieee802154_llsec_key_id *in_key,
943 u32 frame_counter)
944{
945 struct mac802154_llsec_device_key *devkey = NULL;
946
947 if (dev->dev.key_mode == IEEE802154_LLSEC_DEVKEY_RESTRICT) {
948 devkey = llsec_devkey_find(dev, in_key);
949 if (!devkey)
950 return -ENOENT;
951 }
952
f0f77dc6
PB
953 if (dev->dev.key_mode == IEEE802154_LLSEC_DEVKEY_RECORD) {
954 int rc = llsec_update_devkey_record(dev, in_key);
955
956 if (rc < 0)
957 return rc;
958 }
959
4c14a2fb
PB
960 spin_lock_bh(&dev->lock);
961
962 if ((!devkey && frame_counter < dev->dev.frame_counter) ||
963 (devkey && frame_counter < devkey->devkey.frame_counter)) {
964 spin_unlock_bh(&dev->lock);
965 return -EINVAL;
966 }
967
968 if (devkey)
969 devkey->devkey.frame_counter = frame_counter + 1;
970 else
971 dev->dev.frame_counter = frame_counter + 1;
972
973 spin_unlock_bh(&dev->lock);
974
975 return 0;
976}
977
978int mac802154_llsec_decrypt(struct mac802154_llsec *sec, struct sk_buff *skb)
979{
980 struct ieee802154_hdr hdr;
981 struct mac802154_llsec_key *key;
982 struct ieee802154_llsec_key_id key_id;
983 struct mac802154_llsec_device *dev;
984 struct ieee802154_llsec_seclevel seclevel;
985 int err;
986 __le64 dev_addr;
987 u32 frame_ctr;
988
989 if (ieee802154_hdr_peek(skb, &hdr) < 0)
990 return -EINVAL;
991 if (!hdr.fc.security_enabled)
992 return 0;
993 if (hdr.fc.version == 0)
994 return -EINVAL;
995
996 read_lock_bh(&sec->lock);
997 if (!sec->params.enabled) {
998 read_unlock_bh(&sec->lock);
999 return -EINVAL;
1000 }
1001 read_unlock_bh(&sec->lock);
1002
1003 rcu_read_lock();
1004
1005 key = llsec_lookup_key(sec, &hdr, &hdr.source, &key_id);
1006 if (!key) {
1007 err = -ENOKEY;
1008 goto fail;
1009 }
1010
1011 dev = llsec_lookup_dev(sec, &hdr.source);
1012 if (!dev) {
1013 err = -EINVAL;
1014 goto fail_dev;
1015 }
1016
1017 if (llsec_lookup_seclevel(sec, hdr.fc.type, 0, &seclevel) < 0) {
1018 err = -EINVAL;
1019 goto fail_dev;
1020 }
1021
1022 if (!(seclevel.sec_levels & BIT(hdr.sec.level)) &&
1023 (hdr.sec.level == 0 && seclevel.device_override &&
1024 !dev->dev.seclevel_exempt)) {
1025 err = -EINVAL;
1026 goto fail_dev;
1027 }
1028
1029 frame_ctr = le32_to_cpu(hdr.sec.frame_counter);
1030
1031 if (frame_ctr == 0xffffffff) {
1032 err = -EOVERFLOW;
1033 goto fail_dev;
1034 }
1035
1036 err = llsec_update_devkey_info(dev, &key_id, frame_ctr);
1037 if (err)
1038 goto fail_dev;
1039
1040 dev_addr = dev->dev.hwaddr;
1041
1042 rcu_read_unlock();
1043
1044 err = llsec_do_decrypt(skb, sec, &hdr, key, dev_addr);
1045 llsec_key_put(key);
1046 return err;
1047
1048fail_dev:
1049 llsec_key_put(key);
1050fail:
1051 rcu_read_unlock();
1052 return err;
1053}