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