xref: /linux/net/tipc/crypto.c (revision 6fdcba32711044c35c0e1b094cbd8f3f0b4472c9)
1 // SPDX-License-Identifier: GPL-2.0
2 /**
3  * net/tipc/crypto.c: TIPC crypto for key handling & packet en/decryption
4  *
5  * Copyright (c) 2019, Ericsson AB
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions are met:
10  *
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer.
13  * 2. Redistributions in binary form must reproduce the above copyright
14  *    notice, this list of conditions and the following disclaimer in the
15  *    documentation and/or other materials provided with the distribution.
16  * 3. Neither the names of the copyright holders nor the names of its
17  *    contributors may be used to endorse or promote products derived from
18  *    this software without specific prior written permission.
19  *
20  * Alternatively, this software may be distributed under the terms of the
21  * GNU General Public License ("GPL") version 2 as published by the Free
22  * Software Foundation.
23  *
24  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
25  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
26  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
27  * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
28  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
29  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
30  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
31  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
32  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
33  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
34  * POSSIBILITY OF SUCH DAMAGE.
35  */
36 
37 #include <crypto/aead.h>
38 #include <crypto/aes.h>
39 #include "crypto.h"
40 
41 #define TIPC_TX_PROBE_LIM	msecs_to_jiffies(1000) /* > 1s */
42 #define TIPC_TX_LASTING_LIM	msecs_to_jiffies(120000) /* 2 mins */
43 #define TIPC_RX_ACTIVE_LIM	msecs_to_jiffies(3000) /* 3s */
44 #define TIPC_RX_PASSIVE_LIM	msecs_to_jiffies(180000) /* 3 mins */
45 #define TIPC_MAX_TFMS_DEF	10
46 #define TIPC_MAX_TFMS_LIM	1000
47 
48 /**
49  * TIPC Key ids
50  */
51 enum {
52 	KEY_UNUSED = 0,
53 	KEY_MIN,
54 	KEY_1 = KEY_MIN,
55 	KEY_2,
56 	KEY_3,
57 	KEY_MAX = KEY_3,
58 };
59 
60 /**
61  * TIPC Crypto statistics
62  */
63 enum {
64 	STAT_OK,
65 	STAT_NOK,
66 	STAT_ASYNC,
67 	STAT_ASYNC_OK,
68 	STAT_ASYNC_NOK,
69 	STAT_BADKEYS, /* tx only */
70 	STAT_BADMSGS = STAT_BADKEYS, /* rx only */
71 	STAT_NOKEYS,
72 	STAT_SWITCHES,
73 
74 	MAX_STATS,
75 };
76 
77 /* TIPC crypto statistics' header */
78 static const char *hstats[MAX_STATS] = {"ok", "nok", "async", "async_ok",
79 					"async_nok", "badmsgs", "nokeys",
80 					"switches"};
81 
82 /* Max TFMs number per key */
83 int sysctl_tipc_max_tfms __read_mostly = TIPC_MAX_TFMS_DEF;
84 
85 /**
86  * struct tipc_key - TIPC keys' status indicator
87  *
88  *         7     6     5     4     3     2     1     0
89  *      +-----+-----+-----+-----+-----+-----+-----+-----+
90  * key: | (reserved)|passive idx| active idx|pending idx|
91  *      +-----+-----+-----+-----+-----+-----+-----+-----+
92  */
93 struct tipc_key {
94 #define KEY_BITS (2)
95 #define KEY_MASK ((1 << KEY_BITS) - 1)
96 	union {
97 		struct {
98 #if defined(__LITTLE_ENDIAN_BITFIELD)
99 			u8 pending:2,
100 			   active:2,
101 			   passive:2, /* rx only */
102 			   reserved:2;
103 #elif defined(__BIG_ENDIAN_BITFIELD)
104 			u8 reserved:2,
105 			   passive:2, /* rx only */
106 			   active:2,
107 			   pending:2;
108 #else
109 #error  "Please fix <asm/byteorder.h>"
110 #endif
111 		} __packed;
112 		u8 keys;
113 	};
114 };
115 
116 /**
117  * struct tipc_tfm - TIPC TFM structure to form a list of TFMs
118  */
119 struct tipc_tfm {
120 	struct crypto_aead *tfm;
121 	struct list_head list;
122 };
123 
124 /**
125  * struct tipc_aead - TIPC AEAD key structure
126  * @tfm_entry: per-cpu pointer to one entry in TFM list
127  * @crypto: TIPC crypto owns this key
128  * @cloned: reference to the source key in case cloning
129  * @users: the number of the key users (TX/RX)
130  * @salt: the key's SALT value
131  * @authsize: authentication tag size (max = 16)
132  * @mode: crypto mode is applied to the key
133  * @hint[]: a hint for user key
134  * @rcu: struct rcu_head
135  * @seqno: the key seqno (cluster scope)
136  * @refcnt: the key reference counter
137  */
138 struct tipc_aead {
139 #define TIPC_AEAD_HINT_LEN (5)
140 	struct tipc_tfm * __percpu *tfm_entry;
141 	struct tipc_crypto *crypto;
142 	struct tipc_aead *cloned;
143 	atomic_t users;
144 	u32 salt;
145 	u8 authsize;
146 	u8 mode;
147 	char hint[TIPC_AEAD_HINT_LEN + 1];
148 	struct rcu_head rcu;
149 
150 	atomic64_t seqno ____cacheline_aligned;
151 	refcount_t refcnt ____cacheline_aligned;
152 
153 } ____cacheline_aligned;
154 
155 /**
156  * struct tipc_crypto_stats - TIPC Crypto statistics
157  */
158 struct tipc_crypto_stats {
159 	unsigned int stat[MAX_STATS];
160 };
161 
162 /**
163  * struct tipc_crypto - TIPC TX/RX crypto structure
164  * @net: struct net
165  * @node: TIPC node (RX)
166  * @aead: array of pointers to AEAD keys for encryption/decryption
167  * @peer_rx_active: replicated peer RX active key index
168  * @key: the key states
169  * @working: the crypto is working or not
170  * @stats: the crypto statistics
171  * @sndnxt: the per-peer sndnxt (TX)
172  * @timer1: general timer 1 (jiffies)
173  * @timer2: general timer 1 (jiffies)
174  * @lock: tipc_key lock
175  */
176 struct tipc_crypto {
177 	struct net *net;
178 	struct tipc_node *node;
179 	struct tipc_aead __rcu *aead[KEY_MAX + 1]; /* key[0] is UNUSED */
180 	atomic_t peer_rx_active;
181 	struct tipc_key key;
182 	u8 working:1;
183 	struct tipc_crypto_stats __percpu *stats;
184 
185 	atomic64_t sndnxt ____cacheline_aligned;
186 	unsigned long timer1;
187 	unsigned long timer2;
188 	spinlock_t lock; /* crypto lock */
189 
190 } ____cacheline_aligned;
191 
192 /* struct tipc_crypto_tx_ctx - TX context for callbacks */
193 struct tipc_crypto_tx_ctx {
194 	struct tipc_aead *aead;
195 	struct tipc_bearer *bearer;
196 	struct tipc_media_addr dst;
197 };
198 
199 /* struct tipc_crypto_rx_ctx - RX context for callbacks */
200 struct tipc_crypto_rx_ctx {
201 	struct tipc_aead *aead;
202 	struct tipc_bearer *bearer;
203 };
204 
205 static struct tipc_aead *tipc_aead_get(struct tipc_aead __rcu *aead);
206 static inline void tipc_aead_put(struct tipc_aead *aead);
207 static void tipc_aead_free(struct rcu_head *rp);
208 static int tipc_aead_users(struct tipc_aead __rcu *aead);
209 static void tipc_aead_users_inc(struct tipc_aead __rcu *aead, int lim);
210 static void tipc_aead_users_dec(struct tipc_aead __rcu *aead, int lim);
211 static void tipc_aead_users_set(struct tipc_aead __rcu *aead, int val);
212 static struct crypto_aead *tipc_aead_tfm_next(struct tipc_aead *aead);
213 static int tipc_aead_init(struct tipc_aead **aead, struct tipc_aead_key *ukey,
214 			  u8 mode);
215 static int tipc_aead_clone(struct tipc_aead **dst, struct tipc_aead *src);
216 static void *tipc_aead_mem_alloc(struct crypto_aead *tfm,
217 				 unsigned int crypto_ctx_size,
218 				 u8 **iv, struct aead_request **req,
219 				 struct scatterlist **sg, int nsg);
220 static int tipc_aead_encrypt(struct tipc_aead *aead, struct sk_buff *skb,
221 			     struct tipc_bearer *b,
222 			     struct tipc_media_addr *dst,
223 			     struct tipc_node *__dnode);
224 static void tipc_aead_encrypt_done(struct crypto_async_request *base, int err);
225 static int tipc_aead_decrypt(struct net *net, struct tipc_aead *aead,
226 			     struct sk_buff *skb, struct tipc_bearer *b);
227 static void tipc_aead_decrypt_done(struct crypto_async_request *base, int err);
228 static inline int tipc_ehdr_size(struct tipc_ehdr *ehdr);
229 static int tipc_ehdr_build(struct net *net, struct tipc_aead *aead,
230 			   u8 tx_key, struct sk_buff *skb,
231 			   struct tipc_crypto *__rx);
232 static inline void tipc_crypto_key_set_state(struct tipc_crypto *c,
233 					     u8 new_passive,
234 					     u8 new_active,
235 					     u8 new_pending);
236 static int tipc_crypto_key_attach(struct tipc_crypto *c,
237 				  struct tipc_aead *aead, u8 pos);
238 static bool tipc_crypto_key_try_align(struct tipc_crypto *rx, u8 new_pending);
239 static struct tipc_aead *tipc_crypto_key_pick_tx(struct tipc_crypto *tx,
240 						 struct tipc_crypto *rx,
241 						 struct sk_buff *skb);
242 static void tipc_crypto_key_synch(struct tipc_crypto *rx, u8 new_rx_active,
243 				  struct tipc_msg *hdr);
244 static int tipc_crypto_key_revoke(struct net *net, u8 tx_key);
245 static void tipc_crypto_rcv_complete(struct net *net, struct tipc_aead *aead,
246 				     struct tipc_bearer *b,
247 				     struct sk_buff **skb, int err);
248 static void tipc_crypto_do_cmd(struct net *net, int cmd);
249 static char *tipc_crypto_key_dump(struct tipc_crypto *c, char *buf);
250 #ifdef TIPC_CRYPTO_DEBUG
251 static char *tipc_key_change_dump(struct tipc_key old, struct tipc_key new,
252 				  char *buf);
253 #endif
254 
255 #define key_next(cur) ((cur) % KEY_MAX + 1)
256 
257 #define tipc_aead_rcu_ptr(rcu_ptr, lock)				\
258 	rcu_dereference_protected((rcu_ptr), lockdep_is_held(lock))
259 
260 #define tipc_aead_rcu_swap(rcu_ptr, ptr, lock)				\
261 	rcu_swap_protected((rcu_ptr), (ptr), lockdep_is_held(lock))
262 
263 #define tipc_aead_rcu_replace(rcu_ptr, ptr, lock)			\
264 do {									\
265 	typeof(rcu_ptr) __tmp = rcu_dereference_protected((rcu_ptr),	\
266 						lockdep_is_held(lock));	\
267 	rcu_assign_pointer((rcu_ptr), (ptr));				\
268 	tipc_aead_put(__tmp);						\
269 } while (0)
270 
271 #define tipc_crypto_key_detach(rcu_ptr, lock)				\
272 	tipc_aead_rcu_replace((rcu_ptr), NULL, lock)
273 
274 /**
275  * tipc_aead_key_validate - Validate a AEAD user key
276  */
277 int tipc_aead_key_validate(struct tipc_aead_key *ukey)
278 {
279 	int keylen;
280 
281 	/* Check if algorithm exists */
282 	if (unlikely(!crypto_has_alg(ukey->alg_name, 0, 0))) {
283 		pr_info("Not found cipher: \"%s\"!\n", ukey->alg_name);
284 		return -ENODEV;
285 	}
286 
287 	/* Currently, we only support the "gcm(aes)" cipher algorithm */
288 	if (strcmp(ukey->alg_name, "gcm(aes)"))
289 		return -ENOTSUPP;
290 
291 	/* Check if key size is correct */
292 	keylen = ukey->keylen - TIPC_AES_GCM_SALT_SIZE;
293 	if (unlikely(keylen != TIPC_AES_GCM_KEY_SIZE_128 &&
294 		     keylen != TIPC_AES_GCM_KEY_SIZE_192 &&
295 		     keylen != TIPC_AES_GCM_KEY_SIZE_256))
296 		return -EINVAL;
297 
298 	return 0;
299 }
300 
301 static struct tipc_aead *tipc_aead_get(struct tipc_aead __rcu *aead)
302 {
303 	struct tipc_aead *tmp;
304 
305 	rcu_read_lock();
306 	tmp = rcu_dereference(aead);
307 	if (unlikely(!tmp || !refcount_inc_not_zero(&tmp->refcnt)))
308 		tmp = NULL;
309 	rcu_read_unlock();
310 
311 	return tmp;
312 }
313 
314 static inline void tipc_aead_put(struct tipc_aead *aead)
315 {
316 	if (aead && refcount_dec_and_test(&aead->refcnt))
317 		call_rcu(&aead->rcu, tipc_aead_free);
318 }
319 
320 /**
321  * tipc_aead_free - Release AEAD key incl. all the TFMs in the list
322  * @rp: rcu head pointer
323  */
324 static void tipc_aead_free(struct rcu_head *rp)
325 {
326 	struct tipc_aead *aead = container_of(rp, struct tipc_aead, rcu);
327 	struct tipc_tfm *tfm_entry, *head, *tmp;
328 
329 	if (aead->cloned) {
330 		tipc_aead_put(aead->cloned);
331 	} else {
332 		head = *this_cpu_ptr(aead->tfm_entry);
333 		list_for_each_entry_safe(tfm_entry, tmp, &head->list, list) {
334 			crypto_free_aead(tfm_entry->tfm);
335 			list_del(&tfm_entry->list);
336 			kfree(tfm_entry);
337 		}
338 		/* Free the head */
339 		crypto_free_aead(head->tfm);
340 		list_del(&head->list);
341 		kfree(head);
342 	}
343 	free_percpu(aead->tfm_entry);
344 	kfree(aead);
345 }
346 
347 static int tipc_aead_users(struct tipc_aead __rcu *aead)
348 {
349 	struct tipc_aead *tmp;
350 	int users = 0;
351 
352 	rcu_read_lock();
353 	tmp = rcu_dereference(aead);
354 	if (tmp)
355 		users = atomic_read(&tmp->users);
356 	rcu_read_unlock();
357 
358 	return users;
359 }
360 
361 static void tipc_aead_users_inc(struct tipc_aead __rcu *aead, int lim)
362 {
363 	struct tipc_aead *tmp;
364 
365 	rcu_read_lock();
366 	tmp = rcu_dereference(aead);
367 	if (tmp)
368 		atomic_add_unless(&tmp->users, 1, lim);
369 	rcu_read_unlock();
370 }
371 
372 static void tipc_aead_users_dec(struct tipc_aead __rcu *aead, int lim)
373 {
374 	struct tipc_aead *tmp;
375 
376 	rcu_read_lock();
377 	tmp = rcu_dereference(aead);
378 	if (tmp)
379 		atomic_add_unless(&rcu_dereference(aead)->users, -1, lim);
380 	rcu_read_unlock();
381 }
382 
383 static void tipc_aead_users_set(struct tipc_aead __rcu *aead, int val)
384 {
385 	struct tipc_aead *tmp;
386 	int cur;
387 
388 	rcu_read_lock();
389 	tmp = rcu_dereference(aead);
390 	if (tmp) {
391 		do {
392 			cur = atomic_read(&tmp->users);
393 			if (cur == val)
394 				break;
395 		} while (atomic_cmpxchg(&tmp->users, cur, val) != cur);
396 	}
397 	rcu_read_unlock();
398 }
399 
400 /**
401  * tipc_aead_tfm_next - Move TFM entry to the next one in list and return it
402  */
403 static struct crypto_aead *tipc_aead_tfm_next(struct tipc_aead *aead)
404 {
405 	struct tipc_tfm **tfm_entry = this_cpu_ptr(aead->tfm_entry);
406 
407 	*tfm_entry = list_next_entry(*tfm_entry, list);
408 	return (*tfm_entry)->tfm;
409 }
410 
411 /**
412  * tipc_aead_init - Initiate TIPC AEAD
413  * @aead: returned new TIPC AEAD key handle pointer
414  * @ukey: pointer to user key data
415  * @mode: the key mode
416  *
417  * Allocate a (list of) new cipher transformation (TFM) with the specific user
418  * key data if valid. The number of the allocated TFMs can be set via the sysfs
419  * "net/tipc/max_tfms" first.
420  * Also, all the other AEAD data are also initialized.
421  *
422  * Return: 0 if the initiation is successful, otherwise: < 0
423  */
424 static int tipc_aead_init(struct tipc_aead **aead, struct tipc_aead_key *ukey,
425 			  u8 mode)
426 {
427 	struct tipc_tfm *tfm_entry, *head;
428 	struct crypto_aead *tfm;
429 	struct tipc_aead *tmp;
430 	int keylen, err, cpu;
431 	int tfm_cnt = 0;
432 
433 	if (unlikely(*aead))
434 		return -EEXIST;
435 
436 	/* Allocate a new AEAD */
437 	tmp = kzalloc(sizeof(*tmp), GFP_ATOMIC);
438 	if (unlikely(!tmp))
439 		return -ENOMEM;
440 
441 	/* The key consists of two parts: [AES-KEY][SALT] */
442 	keylen = ukey->keylen - TIPC_AES_GCM_SALT_SIZE;
443 
444 	/* Allocate per-cpu TFM entry pointer */
445 	tmp->tfm_entry = alloc_percpu(struct tipc_tfm *);
446 	if (!tmp->tfm_entry) {
447 		kzfree(tmp);
448 		return -ENOMEM;
449 	}
450 
451 	/* Make a list of TFMs with the user key data */
452 	do {
453 		tfm = crypto_alloc_aead(ukey->alg_name, 0, 0);
454 		if (IS_ERR(tfm)) {
455 			err = PTR_ERR(tfm);
456 			break;
457 		}
458 
459 		if (unlikely(!tfm_cnt &&
460 			     crypto_aead_ivsize(tfm) != TIPC_AES_GCM_IV_SIZE)) {
461 			crypto_free_aead(tfm);
462 			err = -ENOTSUPP;
463 			break;
464 		}
465 
466 		err = crypto_aead_setauthsize(tfm, TIPC_AES_GCM_TAG_SIZE);
467 		err |= crypto_aead_setkey(tfm, ukey->key, keylen);
468 		if (unlikely(err)) {
469 			crypto_free_aead(tfm);
470 			break;
471 		}
472 
473 		tfm_entry = kmalloc(sizeof(*tfm_entry), GFP_KERNEL);
474 		if (unlikely(!tfm_entry)) {
475 			crypto_free_aead(tfm);
476 			err = -ENOMEM;
477 			break;
478 		}
479 		INIT_LIST_HEAD(&tfm_entry->list);
480 		tfm_entry->tfm = tfm;
481 
482 		/* First entry? */
483 		if (!tfm_cnt) {
484 			head = tfm_entry;
485 			for_each_possible_cpu(cpu) {
486 				*per_cpu_ptr(tmp->tfm_entry, cpu) = head;
487 			}
488 		} else {
489 			list_add_tail(&tfm_entry->list, &head->list);
490 		}
491 
492 	} while (++tfm_cnt < sysctl_tipc_max_tfms);
493 
494 	/* Not any TFM is allocated? */
495 	if (!tfm_cnt) {
496 		free_percpu(tmp->tfm_entry);
497 		kzfree(tmp);
498 		return err;
499 	}
500 
501 	/* Copy some chars from the user key as a hint */
502 	memcpy(tmp->hint, ukey->key, TIPC_AEAD_HINT_LEN);
503 	tmp->hint[TIPC_AEAD_HINT_LEN] = '\0';
504 
505 	/* Initialize the other data */
506 	tmp->mode = mode;
507 	tmp->cloned = NULL;
508 	tmp->authsize = TIPC_AES_GCM_TAG_SIZE;
509 	memcpy(&tmp->salt, ukey->key + keylen, TIPC_AES_GCM_SALT_SIZE);
510 	atomic_set(&tmp->users, 0);
511 	atomic64_set(&tmp->seqno, 0);
512 	refcount_set(&tmp->refcnt, 1);
513 
514 	*aead = tmp;
515 	return 0;
516 }
517 
518 /**
519  * tipc_aead_clone - Clone a TIPC AEAD key
520  * @dst: dest key for the cloning
521  * @src: source key to clone from
522  *
523  * Make a "copy" of the source AEAD key data to the dest, the TFMs list is
524  * common for the keys.
525  * A reference to the source is hold in the "cloned" pointer for the later
526  * freeing purposes.
527  *
528  * Note: this must be done in cluster-key mode only!
529  * Return: 0 in case of success, otherwise < 0
530  */
531 static int tipc_aead_clone(struct tipc_aead **dst, struct tipc_aead *src)
532 {
533 	struct tipc_aead *aead;
534 	int cpu;
535 
536 	if (!src)
537 		return -ENOKEY;
538 
539 	if (src->mode != CLUSTER_KEY)
540 		return -EINVAL;
541 
542 	if (unlikely(*dst))
543 		return -EEXIST;
544 
545 	aead = kzalloc(sizeof(*aead), GFP_ATOMIC);
546 	if (unlikely(!aead))
547 		return -ENOMEM;
548 
549 	aead->tfm_entry = alloc_percpu_gfp(struct tipc_tfm *, GFP_ATOMIC);
550 	if (unlikely(!aead->tfm_entry)) {
551 		kzfree(aead);
552 		return -ENOMEM;
553 	}
554 
555 	for_each_possible_cpu(cpu) {
556 		*per_cpu_ptr(aead->tfm_entry, cpu) =
557 				*per_cpu_ptr(src->tfm_entry, cpu);
558 	}
559 
560 	memcpy(aead->hint, src->hint, sizeof(src->hint));
561 	aead->mode = src->mode;
562 	aead->salt = src->salt;
563 	aead->authsize = src->authsize;
564 	atomic_set(&aead->users, 0);
565 	atomic64_set(&aead->seqno, 0);
566 	refcount_set(&aead->refcnt, 1);
567 
568 	WARN_ON(!refcount_inc_not_zero(&src->refcnt));
569 	aead->cloned = src;
570 
571 	*dst = aead;
572 	return 0;
573 }
574 
575 /**
576  * tipc_aead_mem_alloc - Allocate memory for AEAD request operations
577  * @tfm: cipher handle to be registered with the request
578  * @crypto_ctx_size: size of crypto context for callback
579  * @iv: returned pointer to IV data
580  * @req: returned pointer to AEAD request data
581  * @sg: returned pointer to SG lists
582  * @nsg: number of SG lists to be allocated
583  *
584  * Allocate memory to store the crypto context data, AEAD request, IV and SG
585  * lists, the memory layout is as follows:
586  * crypto_ctx || iv || aead_req || sg[]
587  *
588  * Return: the pointer to the memory areas in case of success, otherwise NULL
589  */
590 static void *tipc_aead_mem_alloc(struct crypto_aead *tfm,
591 				 unsigned int crypto_ctx_size,
592 				 u8 **iv, struct aead_request **req,
593 				 struct scatterlist **sg, int nsg)
594 {
595 	unsigned int iv_size, req_size;
596 	unsigned int len;
597 	u8 *mem;
598 
599 	iv_size = crypto_aead_ivsize(tfm);
600 	req_size = sizeof(**req) + crypto_aead_reqsize(tfm);
601 
602 	len = crypto_ctx_size;
603 	len += iv_size;
604 	len += crypto_aead_alignmask(tfm) & ~(crypto_tfm_ctx_alignment() - 1);
605 	len = ALIGN(len, crypto_tfm_ctx_alignment());
606 	len += req_size;
607 	len = ALIGN(len, __alignof__(struct scatterlist));
608 	len += nsg * sizeof(**sg);
609 
610 	mem = kmalloc(len, GFP_ATOMIC);
611 	if (!mem)
612 		return NULL;
613 
614 	*iv = (u8 *)PTR_ALIGN(mem + crypto_ctx_size,
615 			      crypto_aead_alignmask(tfm) + 1);
616 	*req = (struct aead_request *)PTR_ALIGN(*iv + iv_size,
617 						crypto_tfm_ctx_alignment());
618 	*sg = (struct scatterlist *)PTR_ALIGN((u8 *)*req + req_size,
619 					      __alignof__(struct scatterlist));
620 
621 	return (void *)mem;
622 }
623 
624 /**
625  * tipc_aead_encrypt - Encrypt a message
626  * @aead: TIPC AEAD key for the message encryption
627  * @skb: the input/output skb
628  * @b: TIPC bearer where the message will be delivered after the encryption
629  * @dst: the destination media address
630  * @__dnode: TIPC dest node if "known"
631  *
632  * Return:
633  * 0                   : if the encryption has completed
634  * -EINPROGRESS/-EBUSY : if a callback will be performed
635  * < 0                 : the encryption has failed
636  */
637 static int tipc_aead_encrypt(struct tipc_aead *aead, struct sk_buff *skb,
638 			     struct tipc_bearer *b,
639 			     struct tipc_media_addr *dst,
640 			     struct tipc_node *__dnode)
641 {
642 	struct crypto_aead *tfm = tipc_aead_tfm_next(aead);
643 	struct tipc_crypto_tx_ctx *tx_ctx;
644 	struct aead_request *req;
645 	struct sk_buff *trailer;
646 	struct scatterlist *sg;
647 	struct tipc_ehdr *ehdr;
648 	int ehsz, len, tailen, nsg, rc;
649 	void *ctx;
650 	u32 salt;
651 	u8 *iv;
652 
653 	/* Make sure message len at least 4-byte aligned */
654 	len = ALIGN(skb->len, 4);
655 	tailen = len - skb->len + aead->authsize;
656 
657 	/* Expand skb tail for authentication tag:
658 	 * As for simplicity, we'd have made sure skb having enough tailroom
659 	 * for authentication tag @skb allocation. Even when skb is nonlinear
660 	 * but there is no frag_list, it should be still fine!
661 	 * Otherwise, we must cow it to be a writable buffer with the tailroom.
662 	 */
663 #ifdef TIPC_CRYPTO_DEBUG
664 	SKB_LINEAR_ASSERT(skb);
665 	if (tailen > skb_tailroom(skb)) {
666 		pr_warn("TX: skb tailroom is not enough: %d, requires: %d\n",
667 			skb_tailroom(skb), tailen);
668 	}
669 #endif
670 
671 	if (unlikely(!skb_cloned(skb) && tailen <= skb_tailroom(skb))) {
672 		nsg = 1;
673 		trailer = skb;
674 	} else {
675 		/* TODO: We could avoid skb_cow_data() if skb has no frag_list
676 		 * e.g. by skb_fill_page_desc() to add another page to the skb
677 		 * with the wanted tailen... However, page skbs look not often,
678 		 * so take it easy now!
679 		 * Cloned skbs e.g. from link_xmit() seems no choice though :(
680 		 */
681 		nsg = skb_cow_data(skb, tailen, &trailer);
682 		if (unlikely(nsg < 0)) {
683 			pr_err("TX: skb_cow_data() returned %d\n", nsg);
684 			return nsg;
685 		}
686 	}
687 
688 	pskb_put(skb, trailer, tailen);
689 
690 	/* Allocate memory for the AEAD operation */
691 	ctx = tipc_aead_mem_alloc(tfm, sizeof(*tx_ctx), &iv, &req, &sg, nsg);
692 	if (unlikely(!ctx))
693 		return -ENOMEM;
694 	TIPC_SKB_CB(skb)->crypto_ctx = ctx;
695 
696 	/* Map skb to the sg lists */
697 	sg_init_table(sg, nsg);
698 	rc = skb_to_sgvec(skb, sg, 0, skb->len);
699 	if (unlikely(rc < 0)) {
700 		pr_err("TX: skb_to_sgvec() returned %d, nsg %d!\n", rc, nsg);
701 		goto exit;
702 	}
703 
704 	/* Prepare IV: [SALT (4 octets)][SEQNO (8 octets)]
705 	 * In case we're in cluster-key mode, SALT is varied by xor-ing with
706 	 * the source address (or w0 of id), otherwise with the dest address
707 	 * if dest is known.
708 	 */
709 	ehdr = (struct tipc_ehdr *)skb->data;
710 	salt = aead->salt;
711 	if (aead->mode == CLUSTER_KEY)
712 		salt ^= ehdr->addr; /* __be32 */
713 	else if (__dnode)
714 		salt ^= tipc_node_get_addr(__dnode);
715 	memcpy(iv, &salt, 4);
716 	memcpy(iv + 4, (u8 *)&ehdr->seqno, 8);
717 
718 	/* Prepare request */
719 	ehsz = tipc_ehdr_size(ehdr);
720 	aead_request_set_tfm(req, tfm);
721 	aead_request_set_ad(req, ehsz);
722 	aead_request_set_crypt(req, sg, sg, len - ehsz, iv);
723 
724 	/* Set callback function & data */
725 	aead_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
726 				  tipc_aead_encrypt_done, skb);
727 	tx_ctx = (struct tipc_crypto_tx_ctx *)ctx;
728 	tx_ctx->aead = aead;
729 	tx_ctx->bearer = b;
730 	memcpy(&tx_ctx->dst, dst, sizeof(*dst));
731 
732 	/* Hold bearer */
733 	if (unlikely(!tipc_bearer_hold(b))) {
734 		rc = -ENODEV;
735 		goto exit;
736 	}
737 
738 	/* Now, do encrypt */
739 	rc = crypto_aead_encrypt(req);
740 	if (rc == -EINPROGRESS || rc == -EBUSY)
741 		return rc;
742 
743 	tipc_bearer_put(b);
744 
745 exit:
746 	kfree(ctx);
747 	TIPC_SKB_CB(skb)->crypto_ctx = NULL;
748 	return rc;
749 }
750 
751 static void tipc_aead_encrypt_done(struct crypto_async_request *base, int err)
752 {
753 	struct sk_buff *skb = base->data;
754 	struct tipc_crypto_tx_ctx *tx_ctx = TIPC_SKB_CB(skb)->crypto_ctx;
755 	struct tipc_bearer *b = tx_ctx->bearer;
756 	struct tipc_aead *aead = tx_ctx->aead;
757 	struct tipc_crypto *tx = aead->crypto;
758 	struct net *net = tx->net;
759 
760 	switch (err) {
761 	case 0:
762 		this_cpu_inc(tx->stats->stat[STAT_ASYNC_OK]);
763 		if (likely(test_bit(0, &b->up)))
764 			b->media->send_msg(net, skb, b, &tx_ctx->dst);
765 		else
766 			kfree_skb(skb);
767 		break;
768 	case -EINPROGRESS:
769 		return;
770 	default:
771 		this_cpu_inc(tx->stats->stat[STAT_ASYNC_NOK]);
772 		kfree_skb(skb);
773 		break;
774 	}
775 
776 	kfree(tx_ctx);
777 	tipc_bearer_put(b);
778 	tipc_aead_put(aead);
779 }
780 
781 /**
782  * tipc_aead_decrypt - Decrypt an encrypted message
783  * @net: struct net
784  * @aead: TIPC AEAD for the message decryption
785  * @skb: the input/output skb
786  * @b: TIPC bearer where the message has been received
787  *
788  * Return:
789  * 0                   : if the decryption has completed
790  * -EINPROGRESS/-EBUSY : if a callback will be performed
791  * < 0                 : the decryption has failed
792  */
793 static int tipc_aead_decrypt(struct net *net, struct tipc_aead *aead,
794 			     struct sk_buff *skb, struct tipc_bearer *b)
795 {
796 	struct tipc_crypto_rx_ctx *rx_ctx;
797 	struct aead_request *req;
798 	struct crypto_aead *tfm;
799 	struct sk_buff *unused;
800 	struct scatterlist *sg;
801 	struct tipc_ehdr *ehdr;
802 	int ehsz, nsg, rc;
803 	void *ctx;
804 	u32 salt;
805 	u8 *iv;
806 
807 	if (unlikely(!aead))
808 		return -ENOKEY;
809 
810 	/* Cow skb data if needed */
811 	if (likely(!skb_cloned(skb) &&
812 		   (!skb_is_nonlinear(skb) || !skb_has_frag_list(skb)))) {
813 		nsg = 1 + skb_shinfo(skb)->nr_frags;
814 	} else {
815 		nsg = skb_cow_data(skb, 0, &unused);
816 		if (unlikely(nsg < 0)) {
817 			pr_err("RX: skb_cow_data() returned %d\n", nsg);
818 			return nsg;
819 		}
820 	}
821 
822 	/* Allocate memory for the AEAD operation */
823 	tfm = tipc_aead_tfm_next(aead);
824 	ctx = tipc_aead_mem_alloc(tfm, sizeof(*rx_ctx), &iv, &req, &sg, nsg);
825 	if (unlikely(!ctx))
826 		return -ENOMEM;
827 	TIPC_SKB_CB(skb)->crypto_ctx = ctx;
828 
829 	/* Map skb to the sg lists */
830 	sg_init_table(sg, nsg);
831 	rc = skb_to_sgvec(skb, sg, 0, skb->len);
832 	if (unlikely(rc < 0)) {
833 		pr_err("RX: skb_to_sgvec() returned %d, nsg %d\n", rc, nsg);
834 		goto exit;
835 	}
836 
837 	/* Reconstruct IV: */
838 	ehdr = (struct tipc_ehdr *)skb->data;
839 	salt = aead->salt;
840 	if (aead->mode == CLUSTER_KEY)
841 		salt ^= ehdr->addr; /* __be32 */
842 	else if (ehdr->destined)
843 		salt ^= tipc_own_addr(net);
844 	memcpy(iv, &salt, 4);
845 	memcpy(iv + 4, (u8 *)&ehdr->seqno, 8);
846 
847 	/* Prepare request */
848 	ehsz = tipc_ehdr_size(ehdr);
849 	aead_request_set_tfm(req, tfm);
850 	aead_request_set_ad(req, ehsz);
851 	aead_request_set_crypt(req, sg, sg, skb->len - ehsz, iv);
852 
853 	/* Set callback function & data */
854 	aead_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
855 				  tipc_aead_decrypt_done, skb);
856 	rx_ctx = (struct tipc_crypto_rx_ctx *)ctx;
857 	rx_ctx->aead = aead;
858 	rx_ctx->bearer = b;
859 
860 	/* Hold bearer */
861 	if (unlikely(!tipc_bearer_hold(b))) {
862 		rc = -ENODEV;
863 		goto exit;
864 	}
865 
866 	/* Now, do decrypt */
867 	rc = crypto_aead_decrypt(req);
868 	if (rc == -EINPROGRESS || rc == -EBUSY)
869 		return rc;
870 
871 	tipc_bearer_put(b);
872 
873 exit:
874 	kfree(ctx);
875 	TIPC_SKB_CB(skb)->crypto_ctx = NULL;
876 	return rc;
877 }
878 
879 static void tipc_aead_decrypt_done(struct crypto_async_request *base, int err)
880 {
881 	struct sk_buff *skb = base->data;
882 	struct tipc_crypto_rx_ctx *rx_ctx = TIPC_SKB_CB(skb)->crypto_ctx;
883 	struct tipc_bearer *b = rx_ctx->bearer;
884 	struct tipc_aead *aead = rx_ctx->aead;
885 	struct tipc_crypto_stats __percpu *stats = aead->crypto->stats;
886 	struct net *net = aead->crypto->net;
887 
888 	switch (err) {
889 	case 0:
890 		this_cpu_inc(stats->stat[STAT_ASYNC_OK]);
891 		break;
892 	case -EINPROGRESS:
893 		return;
894 	default:
895 		this_cpu_inc(stats->stat[STAT_ASYNC_NOK]);
896 		break;
897 	}
898 
899 	kfree(rx_ctx);
900 	tipc_crypto_rcv_complete(net, aead, b, &skb, err);
901 	if (likely(skb)) {
902 		if (likely(test_bit(0, &b->up)))
903 			tipc_rcv(net, skb, b);
904 		else
905 			kfree_skb(skb);
906 	}
907 
908 	tipc_bearer_put(b);
909 }
910 
911 static inline int tipc_ehdr_size(struct tipc_ehdr *ehdr)
912 {
913 	return (ehdr->user != LINK_CONFIG) ? EHDR_SIZE : EHDR_CFG_SIZE;
914 }
915 
916 /**
917  * tipc_ehdr_validate - Validate an encryption message
918  * @skb: the message buffer
919  *
920  * Returns "true" if this is a valid encryption message, otherwise "false"
921  */
922 bool tipc_ehdr_validate(struct sk_buff *skb)
923 {
924 	struct tipc_ehdr *ehdr;
925 	int ehsz;
926 
927 	if (unlikely(!pskb_may_pull(skb, EHDR_MIN_SIZE)))
928 		return false;
929 
930 	ehdr = (struct tipc_ehdr *)skb->data;
931 	if (unlikely(ehdr->version != TIPC_EVERSION))
932 		return false;
933 	ehsz = tipc_ehdr_size(ehdr);
934 	if (unlikely(!pskb_may_pull(skb, ehsz)))
935 		return false;
936 	if (unlikely(skb->len <= ehsz + TIPC_AES_GCM_TAG_SIZE))
937 		return false;
938 	if (unlikely(!ehdr->tx_key))
939 		return false;
940 
941 	return true;
942 }
943 
944 /**
945  * tipc_ehdr_build - Build TIPC encryption message header
946  * @net: struct net
947  * @aead: TX AEAD key to be used for the message encryption
948  * @tx_key: key id used for the message encryption
949  * @skb: input/output message skb
950  * @__rx: RX crypto handle if dest is "known"
951  *
952  * Return: the header size if the building is successful, otherwise < 0
953  */
954 static int tipc_ehdr_build(struct net *net, struct tipc_aead *aead,
955 			   u8 tx_key, struct sk_buff *skb,
956 			   struct tipc_crypto *__rx)
957 {
958 	struct tipc_msg *hdr = buf_msg(skb);
959 	struct tipc_ehdr *ehdr;
960 	u32 user = msg_user(hdr);
961 	u64 seqno;
962 	int ehsz;
963 
964 	/* Make room for encryption header */
965 	ehsz = (user != LINK_CONFIG) ? EHDR_SIZE : EHDR_CFG_SIZE;
966 	WARN_ON(skb_headroom(skb) < ehsz);
967 	ehdr = (struct tipc_ehdr *)skb_push(skb, ehsz);
968 
969 	/* Obtain a seqno first:
970 	 * Use the key seqno (= cluster wise) if dest is unknown or we're in
971 	 * cluster key mode, otherwise it's better for a per-peer seqno!
972 	 */
973 	if (!__rx || aead->mode == CLUSTER_KEY)
974 		seqno = atomic64_inc_return(&aead->seqno);
975 	else
976 		seqno = atomic64_inc_return(&__rx->sndnxt);
977 
978 	/* Revoke the key if seqno is wrapped around */
979 	if (unlikely(!seqno))
980 		return tipc_crypto_key_revoke(net, tx_key);
981 
982 	/* Word 1-2 */
983 	ehdr->seqno = cpu_to_be64(seqno);
984 
985 	/* Words 0, 3- */
986 	ehdr->version = TIPC_EVERSION;
987 	ehdr->user = 0;
988 	ehdr->keepalive = 0;
989 	ehdr->tx_key = tx_key;
990 	ehdr->destined = (__rx) ? 1 : 0;
991 	ehdr->rx_key_active = (__rx) ? __rx->key.active : 0;
992 	ehdr->reserved_1 = 0;
993 	ehdr->reserved_2 = 0;
994 
995 	switch (user) {
996 	case LINK_CONFIG:
997 		ehdr->user = LINK_CONFIG;
998 		memcpy(ehdr->id, tipc_own_id(net), NODE_ID_LEN);
999 		break;
1000 	default:
1001 		if (user == LINK_PROTOCOL && msg_type(hdr) == STATE_MSG) {
1002 			ehdr->user = LINK_PROTOCOL;
1003 			ehdr->keepalive = msg_is_keepalive(hdr);
1004 		}
1005 		ehdr->addr = hdr->hdr[3];
1006 		break;
1007 	}
1008 
1009 	return ehsz;
1010 }
1011 
1012 static inline void tipc_crypto_key_set_state(struct tipc_crypto *c,
1013 					     u8 new_passive,
1014 					     u8 new_active,
1015 					     u8 new_pending)
1016 {
1017 #ifdef TIPC_CRYPTO_DEBUG
1018 	struct tipc_key old = c->key;
1019 	char buf[32];
1020 #endif
1021 
1022 	c->key.keys = ((new_passive & KEY_MASK) << (KEY_BITS * 2)) |
1023 		      ((new_active  & KEY_MASK) << (KEY_BITS)) |
1024 		      ((new_pending & KEY_MASK));
1025 
1026 #ifdef TIPC_CRYPTO_DEBUG
1027 	pr_info("%s(%s): key changing %s ::%pS\n",
1028 		(c->node) ? "RX" : "TX",
1029 		(c->node) ? tipc_node_get_id_str(c->node) :
1030 			    tipc_own_id_string(c->net),
1031 		tipc_key_change_dump(old, c->key, buf),
1032 		__builtin_return_address(0));
1033 #endif
1034 }
1035 
1036 /**
1037  * tipc_crypto_key_init - Initiate a new user / AEAD key
1038  * @c: TIPC crypto to which new key is attached
1039  * @ukey: the user key
1040  * @mode: the key mode (CLUSTER_KEY or PER_NODE_KEY)
1041  *
1042  * A new TIPC AEAD key will be allocated and initiated with the specified user
1043  * key, then attached to the TIPC crypto.
1044  *
1045  * Return: new key id in case of success, otherwise: < 0
1046  */
1047 int tipc_crypto_key_init(struct tipc_crypto *c, struct tipc_aead_key *ukey,
1048 			 u8 mode)
1049 {
1050 	struct tipc_aead *aead = NULL;
1051 	int rc = 0;
1052 
1053 	/* Initiate with the new user key */
1054 	rc = tipc_aead_init(&aead, ukey, mode);
1055 
1056 	/* Attach it to the crypto */
1057 	if (likely(!rc)) {
1058 		rc = tipc_crypto_key_attach(c, aead, 0);
1059 		if (rc < 0)
1060 			tipc_aead_free(&aead->rcu);
1061 	}
1062 
1063 	pr_info("%s(%s): key initiating, rc %d!\n",
1064 		(c->node) ? "RX" : "TX",
1065 		(c->node) ? tipc_node_get_id_str(c->node) :
1066 			    tipc_own_id_string(c->net),
1067 		rc);
1068 
1069 	return rc;
1070 }
1071 
1072 /**
1073  * tipc_crypto_key_attach - Attach a new AEAD key to TIPC crypto
1074  * @c: TIPC crypto to which the new AEAD key is attached
1075  * @aead: the new AEAD key pointer
1076  * @pos: desired slot in the crypto key array, = 0 if any!
1077  *
1078  * Return: new key id in case of success, otherwise: -EBUSY
1079  */
1080 static int tipc_crypto_key_attach(struct tipc_crypto *c,
1081 				  struct tipc_aead *aead, u8 pos)
1082 {
1083 	u8 new_pending, new_passive, new_key;
1084 	struct tipc_key key;
1085 	int rc = -EBUSY;
1086 
1087 	spin_lock_bh(&c->lock);
1088 	key = c->key;
1089 	if (key.active && key.passive)
1090 		goto exit;
1091 	if (key.passive && !tipc_aead_users(c->aead[key.passive]))
1092 		goto exit;
1093 	if (key.pending) {
1094 		if (pos)
1095 			goto exit;
1096 		if (tipc_aead_users(c->aead[key.pending]) > 0)
1097 			goto exit;
1098 		/* Replace it */
1099 		new_pending = key.pending;
1100 		new_passive = key.passive;
1101 		new_key = new_pending;
1102 	} else {
1103 		if (pos) {
1104 			if (key.active && pos != key_next(key.active)) {
1105 				new_pending = key.pending;
1106 				new_passive = pos;
1107 				new_key = new_passive;
1108 				goto attach;
1109 			} else if (!key.active && !key.passive) {
1110 				new_pending = pos;
1111 				new_passive = key.passive;
1112 				new_key = new_pending;
1113 				goto attach;
1114 			}
1115 		}
1116 		new_pending = key_next(key.active ?: key.passive);
1117 		new_passive = key.passive;
1118 		new_key = new_pending;
1119 	}
1120 
1121 attach:
1122 	aead->crypto = c;
1123 	tipc_crypto_key_set_state(c, new_passive, key.active, new_pending);
1124 	tipc_aead_rcu_replace(c->aead[new_key], aead, &c->lock);
1125 
1126 	c->working = 1;
1127 	c->timer1 = jiffies;
1128 	c->timer2 = jiffies;
1129 	rc = new_key;
1130 
1131 exit:
1132 	spin_unlock_bh(&c->lock);
1133 	return rc;
1134 }
1135 
1136 void tipc_crypto_key_flush(struct tipc_crypto *c)
1137 {
1138 	int k;
1139 
1140 	spin_lock_bh(&c->lock);
1141 	c->working = 0;
1142 	tipc_crypto_key_set_state(c, 0, 0, 0);
1143 	for (k = KEY_MIN; k <= KEY_MAX; k++)
1144 		tipc_crypto_key_detach(c->aead[k], &c->lock);
1145 	atomic_set(&c->peer_rx_active, 0);
1146 	atomic64_set(&c->sndnxt, 0);
1147 	spin_unlock_bh(&c->lock);
1148 }
1149 
1150 /**
1151  * tipc_crypto_key_try_align - Align RX keys if possible
1152  * @rx: RX crypto handle
1153  * @new_pending: new pending slot if aligned (= TX key from peer)
1154  *
1155  * Peer has used an unknown key slot, this only happens when peer has left and
1156  * rejoned, or we are newcomer.
1157  * That means, there must be no active key but a pending key at unaligned slot.
1158  * If so, we try to move the pending key to the new slot.
1159  * Note: A potential passive key can exist, it will be shifted correspondingly!
1160  *
1161  * Return: "true" if key is successfully aligned, otherwise "false"
1162  */
1163 static bool tipc_crypto_key_try_align(struct tipc_crypto *rx, u8 new_pending)
1164 {
1165 	struct tipc_aead *tmp1, *tmp2 = NULL;
1166 	struct tipc_key key;
1167 	bool aligned = false;
1168 	u8 new_passive = 0;
1169 	int x;
1170 
1171 	spin_lock(&rx->lock);
1172 	key = rx->key;
1173 	if (key.pending == new_pending) {
1174 		aligned = true;
1175 		goto exit;
1176 	}
1177 	if (key.active)
1178 		goto exit;
1179 	if (!key.pending)
1180 		goto exit;
1181 	if (tipc_aead_users(rx->aead[key.pending]) > 0)
1182 		goto exit;
1183 
1184 	/* Try to "isolate" this pending key first */
1185 	tmp1 = tipc_aead_rcu_ptr(rx->aead[key.pending], &rx->lock);
1186 	if (!refcount_dec_if_one(&tmp1->refcnt))
1187 		goto exit;
1188 	rcu_assign_pointer(rx->aead[key.pending], NULL);
1189 
1190 	/* Move passive key if any */
1191 	if (key.passive) {
1192 		tipc_aead_rcu_swap(rx->aead[key.passive], tmp2, &rx->lock);
1193 		x = (key.passive - key.pending + new_pending) % KEY_MAX;
1194 		new_passive = (x <= 0) ? x + KEY_MAX : x;
1195 	}
1196 
1197 	/* Re-allocate the key(s) */
1198 	tipc_crypto_key_set_state(rx, new_passive, 0, new_pending);
1199 	rcu_assign_pointer(rx->aead[new_pending], tmp1);
1200 	if (new_passive)
1201 		rcu_assign_pointer(rx->aead[new_passive], tmp2);
1202 	refcount_set(&tmp1->refcnt, 1);
1203 	aligned = true;
1204 	pr_info("RX(%s): key is aligned!\n", tipc_node_get_id_str(rx->node));
1205 
1206 exit:
1207 	spin_unlock(&rx->lock);
1208 	return aligned;
1209 }
1210 
1211 /**
1212  * tipc_crypto_key_pick_tx - Pick one TX key for message decryption
1213  * @tx: TX crypto handle
1214  * @rx: RX crypto handle (can be NULL)
1215  * @skb: the message skb which will be decrypted later
1216  *
1217  * This function looks up the existing TX keys and pick one which is suitable
1218  * for the message decryption, that must be a cluster key and not used before
1219  * on the same message (i.e. recursive).
1220  *
1221  * Return: the TX AEAD key handle in case of success, otherwise NULL
1222  */
1223 static struct tipc_aead *tipc_crypto_key_pick_tx(struct tipc_crypto *tx,
1224 						 struct tipc_crypto *rx,
1225 						 struct sk_buff *skb)
1226 {
1227 	struct tipc_skb_cb *skb_cb = TIPC_SKB_CB(skb);
1228 	struct tipc_aead *aead = NULL;
1229 	struct tipc_key key = tx->key;
1230 	u8 k, i = 0;
1231 
1232 	/* Initialize data if not yet */
1233 	if (!skb_cb->tx_clone_deferred) {
1234 		skb_cb->tx_clone_deferred = 1;
1235 		memset(&skb_cb->tx_clone_ctx, 0, sizeof(skb_cb->tx_clone_ctx));
1236 	}
1237 
1238 	skb_cb->tx_clone_ctx.rx = rx;
1239 	if (++skb_cb->tx_clone_ctx.recurs > 2)
1240 		return NULL;
1241 
1242 	/* Pick one TX key */
1243 	spin_lock(&tx->lock);
1244 	do {
1245 		k = (i == 0) ? key.pending :
1246 			((i == 1) ? key.active : key.passive);
1247 		if (!k)
1248 			continue;
1249 		aead = tipc_aead_rcu_ptr(tx->aead[k], &tx->lock);
1250 		if (!aead)
1251 			continue;
1252 		if (aead->mode != CLUSTER_KEY ||
1253 		    aead == skb_cb->tx_clone_ctx.last) {
1254 			aead = NULL;
1255 			continue;
1256 		}
1257 		/* Ok, found one cluster key */
1258 		skb_cb->tx_clone_ctx.last = aead;
1259 		WARN_ON(skb->next);
1260 		skb->next = skb_clone(skb, GFP_ATOMIC);
1261 		if (unlikely(!skb->next))
1262 			pr_warn("Failed to clone skb for next round if any\n");
1263 		WARN_ON(!refcount_inc_not_zero(&aead->refcnt));
1264 		break;
1265 	} while (++i < 3);
1266 	spin_unlock(&tx->lock);
1267 
1268 	return aead;
1269 }
1270 
1271 /**
1272  * tipc_crypto_key_synch: Synch own key data according to peer key status
1273  * @rx: RX crypto handle
1274  * @new_rx_active: latest RX active key from peer
1275  * @hdr: TIPCv2 message
1276  *
1277  * This function updates the peer node related data as the peer RX active key
1278  * has changed, so the number of TX keys' users on this node are increased and
1279  * decreased correspondingly.
1280  *
1281  * The "per-peer" sndnxt is also reset when the peer key has switched.
1282  */
1283 static void tipc_crypto_key_synch(struct tipc_crypto *rx, u8 new_rx_active,
1284 				  struct tipc_msg *hdr)
1285 {
1286 	struct net *net = rx->net;
1287 	struct tipc_crypto *tx = tipc_net(net)->crypto_tx;
1288 	u8 cur_rx_active;
1289 
1290 	/* TX might be even not ready yet */
1291 	if (unlikely(!tx->key.active && !tx->key.pending))
1292 		return;
1293 
1294 	cur_rx_active = atomic_read(&rx->peer_rx_active);
1295 	if (likely(cur_rx_active == new_rx_active))
1296 		return;
1297 
1298 	/* Make sure this message destined for this node */
1299 	if (unlikely(msg_short(hdr) ||
1300 		     msg_destnode(hdr) != tipc_own_addr(net)))
1301 		return;
1302 
1303 	/* Peer RX active key has changed, try to update owns' & TX users */
1304 	if (atomic_cmpxchg(&rx->peer_rx_active,
1305 			   cur_rx_active,
1306 			   new_rx_active) == cur_rx_active) {
1307 		if (new_rx_active)
1308 			tipc_aead_users_inc(tx->aead[new_rx_active], INT_MAX);
1309 		if (cur_rx_active)
1310 			tipc_aead_users_dec(tx->aead[cur_rx_active], 0);
1311 
1312 		atomic64_set(&rx->sndnxt, 0);
1313 		/* Mark the point TX key users changed */
1314 		tx->timer1 = jiffies;
1315 
1316 #ifdef TIPC_CRYPTO_DEBUG
1317 		pr_info("TX(%s): key users changed %d-- %d++, peer RX(%s)\n",
1318 			tipc_own_id_string(net), cur_rx_active,
1319 			new_rx_active, tipc_node_get_id_str(rx->node));
1320 #endif
1321 	}
1322 }
1323 
1324 static int tipc_crypto_key_revoke(struct net *net, u8 tx_key)
1325 {
1326 	struct tipc_crypto *tx = tipc_net(net)->crypto_tx;
1327 	struct tipc_key key;
1328 
1329 	spin_lock(&tx->lock);
1330 	key = tx->key;
1331 	WARN_ON(!key.active || tx_key != key.active);
1332 
1333 	/* Free the active key */
1334 	tipc_crypto_key_set_state(tx, key.passive, 0, key.pending);
1335 	tipc_crypto_key_detach(tx->aead[key.active], &tx->lock);
1336 	spin_unlock(&tx->lock);
1337 
1338 	pr_warn("TX(%s): key is revoked!\n", tipc_own_id_string(net));
1339 	return -EKEYREVOKED;
1340 }
1341 
1342 int tipc_crypto_start(struct tipc_crypto **crypto, struct net *net,
1343 		      struct tipc_node *node)
1344 {
1345 	struct tipc_crypto *c;
1346 
1347 	if (*crypto)
1348 		return -EEXIST;
1349 
1350 	/* Allocate crypto */
1351 	c = kzalloc(sizeof(*c), GFP_ATOMIC);
1352 	if (!c)
1353 		return -ENOMEM;
1354 
1355 	/* Allocate statistic structure */
1356 	c->stats = alloc_percpu_gfp(struct tipc_crypto_stats, GFP_ATOMIC);
1357 	if (!c->stats) {
1358 		kzfree(c);
1359 		return -ENOMEM;
1360 	}
1361 
1362 	c->working = 0;
1363 	c->net = net;
1364 	c->node = node;
1365 	tipc_crypto_key_set_state(c, 0, 0, 0);
1366 	atomic_set(&c->peer_rx_active, 0);
1367 	atomic64_set(&c->sndnxt, 0);
1368 	c->timer1 = jiffies;
1369 	c->timer2 = jiffies;
1370 	spin_lock_init(&c->lock);
1371 	*crypto = c;
1372 
1373 	return 0;
1374 }
1375 
1376 void tipc_crypto_stop(struct tipc_crypto **crypto)
1377 {
1378 	struct tipc_crypto *c, *tx, *rx;
1379 	bool is_rx;
1380 	u8 k;
1381 
1382 	if (!*crypto)
1383 		return;
1384 
1385 	rcu_read_lock();
1386 	/* RX stopping? => decrease TX key users if any */
1387 	is_rx = !!((*crypto)->node);
1388 	if (is_rx) {
1389 		rx = *crypto;
1390 		tx = tipc_net(rx->net)->crypto_tx;
1391 		k = atomic_read(&rx->peer_rx_active);
1392 		if (k) {
1393 			tipc_aead_users_dec(tx->aead[k], 0);
1394 			/* Mark the point TX key users changed */
1395 			tx->timer1 = jiffies;
1396 		}
1397 	}
1398 
1399 	/* Release AEAD keys */
1400 	c = *crypto;
1401 	for (k = KEY_MIN; k <= KEY_MAX; k++)
1402 		tipc_aead_put(rcu_dereference(c->aead[k]));
1403 	rcu_read_unlock();
1404 
1405 	pr_warn("%s(%s) has been purged, node left!\n",
1406 		(is_rx) ? "RX" : "TX",
1407 		(is_rx) ? tipc_node_get_id_str((*crypto)->node) :
1408 			  tipc_own_id_string((*crypto)->net));
1409 
1410 	/* Free this crypto statistics */
1411 	free_percpu(c->stats);
1412 
1413 	*crypto = NULL;
1414 	kzfree(c);
1415 }
1416 
1417 void tipc_crypto_timeout(struct tipc_crypto *rx)
1418 {
1419 	struct tipc_net *tn = tipc_net(rx->net);
1420 	struct tipc_crypto *tx = tn->crypto_tx;
1421 	struct tipc_key key;
1422 	u8 new_pending, new_passive;
1423 	int cmd;
1424 
1425 	/* TX key activating:
1426 	 * The pending key (users > 0) -> active
1427 	 * The active key if any (users == 0) -> free
1428 	 */
1429 	spin_lock(&tx->lock);
1430 	key = tx->key;
1431 	if (key.active && tipc_aead_users(tx->aead[key.active]) > 0)
1432 		goto s1;
1433 	if (!key.pending || tipc_aead_users(tx->aead[key.pending]) <= 0)
1434 		goto s1;
1435 	if (time_before(jiffies, tx->timer1 + TIPC_TX_LASTING_LIM))
1436 		goto s1;
1437 
1438 	tipc_crypto_key_set_state(tx, key.passive, key.pending, 0);
1439 	if (key.active)
1440 		tipc_crypto_key_detach(tx->aead[key.active], &tx->lock);
1441 	this_cpu_inc(tx->stats->stat[STAT_SWITCHES]);
1442 	pr_info("TX(%s): key %d is activated!\n", tipc_own_id_string(tx->net),
1443 		key.pending);
1444 
1445 s1:
1446 	spin_unlock(&tx->lock);
1447 
1448 	/* RX key activating:
1449 	 * The pending key (users > 0) -> active
1450 	 * The active key if any -> passive, freed later
1451 	 */
1452 	spin_lock(&rx->lock);
1453 	key = rx->key;
1454 	if (!key.pending || tipc_aead_users(rx->aead[key.pending]) <= 0)
1455 		goto s2;
1456 
1457 	new_pending = (key.passive &&
1458 		       !tipc_aead_users(rx->aead[key.passive])) ?
1459 				       key.passive : 0;
1460 	new_passive = (key.active) ?: ((new_pending) ? 0 : key.passive);
1461 	tipc_crypto_key_set_state(rx, new_passive, key.pending, new_pending);
1462 	this_cpu_inc(rx->stats->stat[STAT_SWITCHES]);
1463 	pr_info("RX(%s): key %d is activated!\n",
1464 		tipc_node_get_id_str(rx->node),	key.pending);
1465 	goto s5;
1466 
1467 s2:
1468 	/* RX key "faulty" switching:
1469 	 * The faulty pending key (users < -30) -> passive
1470 	 * The passive key (users = 0) -> pending
1471 	 * Note: This only happens after RX deactivated - s3!
1472 	 */
1473 	key = rx->key;
1474 	if (!key.pending || tipc_aead_users(rx->aead[key.pending]) > -30)
1475 		goto s3;
1476 	if (!key.passive || tipc_aead_users(rx->aead[key.passive]) != 0)
1477 		goto s3;
1478 
1479 	new_pending = key.passive;
1480 	new_passive = key.pending;
1481 	tipc_crypto_key_set_state(rx, new_passive, key.active, new_pending);
1482 	goto s5;
1483 
1484 s3:
1485 	/* RX key deactivating:
1486 	 * The passive key if any -> pending
1487 	 * The active key -> passive (users = 0) / pending
1488 	 * The pending key if any -> passive (users = 0)
1489 	 */
1490 	key = rx->key;
1491 	if (!key.active)
1492 		goto s4;
1493 	if (time_before(jiffies, rx->timer1 + TIPC_RX_ACTIVE_LIM))
1494 		goto s4;
1495 
1496 	new_pending = (key.passive) ?: key.active;
1497 	new_passive = (key.passive) ? key.active : key.pending;
1498 	tipc_aead_users_set(rx->aead[new_pending], 0);
1499 	if (new_passive)
1500 		tipc_aead_users_set(rx->aead[new_passive], 0);
1501 	tipc_crypto_key_set_state(rx, new_passive, 0, new_pending);
1502 	pr_info("RX(%s): key %d is deactivated!\n",
1503 		tipc_node_get_id_str(rx->node), key.active);
1504 	goto s5;
1505 
1506 s4:
1507 	/* RX key passive -> freed: */
1508 	key = rx->key;
1509 	if (!key.passive || !tipc_aead_users(rx->aead[key.passive]))
1510 		goto s5;
1511 	if (time_before(jiffies, rx->timer2 + TIPC_RX_PASSIVE_LIM))
1512 		goto s5;
1513 
1514 	tipc_crypto_key_set_state(rx, 0, key.active, key.pending);
1515 	tipc_crypto_key_detach(rx->aead[key.passive], &rx->lock);
1516 	pr_info("RX(%s): key %d is freed!\n", tipc_node_get_id_str(rx->node),
1517 		key.passive);
1518 
1519 s5:
1520 	spin_unlock(&rx->lock);
1521 
1522 	/* Limit max_tfms & do debug commands if needed */
1523 	if (likely(sysctl_tipc_max_tfms <= TIPC_MAX_TFMS_LIM))
1524 		return;
1525 
1526 	cmd = sysctl_tipc_max_tfms;
1527 	sysctl_tipc_max_tfms = TIPC_MAX_TFMS_DEF;
1528 	tipc_crypto_do_cmd(rx->net, cmd);
1529 }
1530 
1531 /**
1532  * tipc_crypto_xmit - Build & encrypt TIPC message for xmit
1533  * @net: struct net
1534  * @skb: input/output message skb pointer
1535  * @b: bearer used for xmit later
1536  * @dst: destination media address
1537  * @__dnode: destination node for reference if any
1538  *
1539  * First, build an encryption message header on the top of the message, then
1540  * encrypt the original TIPC message by using the active or pending TX key.
1541  * If the encryption is successful, the encrypted skb is returned directly or
1542  * via the callback.
1543  * Otherwise, the skb is freed!
1544  *
1545  * Return:
1546  * 0                   : the encryption has succeeded (or no encryption)
1547  * -EINPROGRESS/-EBUSY : the encryption is ongoing, a callback will be made
1548  * -ENOKEK             : the encryption has failed due to no key
1549  * -EKEYREVOKED        : the encryption has failed due to key revoked
1550  * -ENOMEM             : the encryption has failed due to no memory
1551  * < 0                 : the encryption has failed due to other reasons
1552  */
1553 int tipc_crypto_xmit(struct net *net, struct sk_buff **skb,
1554 		     struct tipc_bearer *b, struct tipc_media_addr *dst,
1555 		     struct tipc_node *__dnode)
1556 {
1557 	struct tipc_crypto *__rx = tipc_node_crypto_rx(__dnode);
1558 	struct tipc_crypto *tx = tipc_net(net)->crypto_tx;
1559 	struct tipc_crypto_stats __percpu *stats = tx->stats;
1560 	struct tipc_key key = tx->key;
1561 	struct tipc_aead *aead = NULL;
1562 	struct sk_buff *probe;
1563 	int rc = -ENOKEY;
1564 	u8 tx_key;
1565 
1566 	/* No encryption? */
1567 	if (!tx->working)
1568 		return 0;
1569 
1570 	/* Try with the pending key if available and:
1571 	 * 1) This is the only choice (i.e. no active key) or;
1572 	 * 2) Peer has switched to this key (unicast only) or;
1573 	 * 3) It is time to do a pending key probe;
1574 	 */
1575 	if (unlikely(key.pending)) {
1576 		tx_key = key.pending;
1577 		if (!key.active)
1578 			goto encrypt;
1579 		if (__rx && atomic_read(&__rx->peer_rx_active) == tx_key)
1580 			goto encrypt;
1581 		if (TIPC_SKB_CB(*skb)->probe)
1582 			goto encrypt;
1583 		if (!__rx &&
1584 		    time_after(jiffies, tx->timer2 + TIPC_TX_PROBE_LIM)) {
1585 			tx->timer2 = jiffies;
1586 			probe = skb_clone(*skb, GFP_ATOMIC);
1587 			if (probe) {
1588 				TIPC_SKB_CB(probe)->probe = 1;
1589 				tipc_crypto_xmit(net, &probe, b, dst, __dnode);
1590 				if (probe)
1591 					b->media->send_msg(net, probe, b, dst);
1592 			}
1593 		}
1594 	}
1595 	/* Else, use the active key if any */
1596 	if (likely(key.active)) {
1597 		tx_key = key.active;
1598 		goto encrypt;
1599 	}
1600 	goto exit;
1601 
1602 encrypt:
1603 	aead = tipc_aead_get(tx->aead[tx_key]);
1604 	if (unlikely(!aead))
1605 		goto exit;
1606 	rc = tipc_ehdr_build(net, aead, tx_key, *skb, __rx);
1607 	if (likely(rc > 0))
1608 		rc = tipc_aead_encrypt(aead, *skb, b, dst, __dnode);
1609 
1610 exit:
1611 	switch (rc) {
1612 	case 0:
1613 		this_cpu_inc(stats->stat[STAT_OK]);
1614 		break;
1615 	case -EINPROGRESS:
1616 	case -EBUSY:
1617 		this_cpu_inc(stats->stat[STAT_ASYNC]);
1618 		*skb = NULL;
1619 		return rc;
1620 	default:
1621 		this_cpu_inc(stats->stat[STAT_NOK]);
1622 		if (rc == -ENOKEY)
1623 			this_cpu_inc(stats->stat[STAT_NOKEYS]);
1624 		else if (rc == -EKEYREVOKED)
1625 			this_cpu_inc(stats->stat[STAT_BADKEYS]);
1626 		kfree_skb(*skb);
1627 		*skb = NULL;
1628 		break;
1629 	}
1630 
1631 	tipc_aead_put(aead);
1632 	return rc;
1633 }
1634 
1635 /**
1636  * tipc_crypto_rcv - Decrypt an encrypted TIPC message from peer
1637  * @net: struct net
1638  * @rx: RX crypto handle
1639  * @skb: input/output message skb pointer
1640  * @b: bearer where the message has been received
1641  *
1642  * If the decryption is successful, the decrypted skb is returned directly or
1643  * as the callback, the encryption header and auth tag will be trimed out
1644  * before forwarding to tipc_rcv() via the tipc_crypto_rcv_complete().
1645  * Otherwise, the skb will be freed!
1646  * Note: RX key(s) can be re-aligned, or in case of no key suitable, TX
1647  * cluster key(s) can be taken for decryption (- recursive).
1648  *
1649  * Return:
1650  * 0                   : the decryption has successfully completed
1651  * -EINPROGRESS/-EBUSY : the decryption is ongoing, a callback will be made
1652  * -ENOKEY             : the decryption has failed due to no key
1653  * -EBADMSG            : the decryption has failed due to bad message
1654  * -ENOMEM             : the decryption has failed due to no memory
1655  * < 0                 : the decryption has failed due to other reasons
1656  */
1657 int tipc_crypto_rcv(struct net *net, struct tipc_crypto *rx,
1658 		    struct sk_buff **skb, struct tipc_bearer *b)
1659 {
1660 	struct tipc_crypto *tx = tipc_net(net)->crypto_tx;
1661 	struct tipc_crypto_stats __percpu *stats;
1662 	struct tipc_aead *aead = NULL;
1663 	struct tipc_key key;
1664 	int rc = -ENOKEY;
1665 	u8 tx_key = 0;
1666 
1667 	/* New peer?
1668 	 * Let's try with TX key (i.e. cluster mode) & verify the skb first!
1669 	 */
1670 	if (unlikely(!rx))
1671 		goto pick_tx;
1672 
1673 	/* Pick RX key according to TX key, three cases are possible:
1674 	 * 1) The current active key (likely) or;
1675 	 * 2) The pending (new or deactivated) key (if any) or;
1676 	 * 3) The passive or old active key (i.e. users > 0);
1677 	 */
1678 	tx_key = ((struct tipc_ehdr *)(*skb)->data)->tx_key;
1679 	key = rx->key;
1680 	if (likely(tx_key == key.active))
1681 		goto decrypt;
1682 	if (tx_key == key.pending)
1683 		goto decrypt;
1684 	if (tx_key == key.passive) {
1685 		rx->timer2 = jiffies;
1686 		if (tipc_aead_users(rx->aead[key.passive]) > 0)
1687 			goto decrypt;
1688 	}
1689 
1690 	/* Unknown key, let's try to align RX key(s) */
1691 	if (tipc_crypto_key_try_align(rx, tx_key))
1692 		goto decrypt;
1693 
1694 pick_tx:
1695 	/* No key suitable? Try to pick one from TX... */
1696 	aead = tipc_crypto_key_pick_tx(tx, rx, *skb);
1697 	if (aead)
1698 		goto decrypt;
1699 	goto exit;
1700 
1701 decrypt:
1702 	rcu_read_lock();
1703 	if (!aead)
1704 		aead = tipc_aead_get(rx->aead[tx_key]);
1705 	rc = tipc_aead_decrypt(net, aead, *skb, b);
1706 	rcu_read_unlock();
1707 
1708 exit:
1709 	stats = ((rx) ?: tx)->stats;
1710 	switch (rc) {
1711 	case 0:
1712 		this_cpu_inc(stats->stat[STAT_OK]);
1713 		break;
1714 	case -EINPROGRESS:
1715 	case -EBUSY:
1716 		this_cpu_inc(stats->stat[STAT_ASYNC]);
1717 		*skb = NULL;
1718 		return rc;
1719 	default:
1720 		this_cpu_inc(stats->stat[STAT_NOK]);
1721 		if (rc == -ENOKEY) {
1722 			kfree_skb(*skb);
1723 			*skb = NULL;
1724 			if (rx)
1725 				tipc_node_put(rx->node);
1726 			this_cpu_inc(stats->stat[STAT_NOKEYS]);
1727 			return rc;
1728 		} else if (rc == -EBADMSG) {
1729 			this_cpu_inc(stats->stat[STAT_BADMSGS]);
1730 		}
1731 		break;
1732 	}
1733 
1734 	tipc_crypto_rcv_complete(net, aead, b, skb, rc);
1735 	return rc;
1736 }
1737 
1738 static void tipc_crypto_rcv_complete(struct net *net, struct tipc_aead *aead,
1739 				     struct tipc_bearer *b,
1740 				     struct sk_buff **skb, int err)
1741 {
1742 	struct tipc_skb_cb *skb_cb = TIPC_SKB_CB(*skb);
1743 	struct tipc_crypto *rx = aead->crypto;
1744 	struct tipc_aead *tmp = NULL;
1745 	struct tipc_ehdr *ehdr;
1746 	struct tipc_node *n;
1747 	u8 rx_key_active;
1748 	bool destined;
1749 
1750 	/* Is this completed by TX? */
1751 	if (unlikely(!rx->node)) {
1752 		rx = skb_cb->tx_clone_ctx.rx;
1753 #ifdef TIPC_CRYPTO_DEBUG
1754 		pr_info("TX->RX(%s): err %d, aead %p, skb->next %p, flags %x\n",
1755 			(rx) ? tipc_node_get_id_str(rx->node) : "-", err, aead,
1756 			(*skb)->next, skb_cb->flags);
1757 		pr_info("skb_cb [recurs %d, last %p], tx->aead [%p %p %p]\n",
1758 			skb_cb->tx_clone_ctx.recurs, skb_cb->tx_clone_ctx.last,
1759 			aead->crypto->aead[1], aead->crypto->aead[2],
1760 			aead->crypto->aead[3]);
1761 #endif
1762 		if (unlikely(err)) {
1763 			if (err == -EBADMSG && (*skb)->next)
1764 				tipc_rcv(net, (*skb)->next, b);
1765 			goto free_skb;
1766 		}
1767 
1768 		if (likely((*skb)->next)) {
1769 			kfree_skb((*skb)->next);
1770 			(*skb)->next = NULL;
1771 		}
1772 		ehdr = (struct tipc_ehdr *)(*skb)->data;
1773 		if (!rx) {
1774 			WARN_ON(ehdr->user != LINK_CONFIG);
1775 			n = tipc_node_create(net, 0, ehdr->id, 0xffffu, 0,
1776 					     true);
1777 			rx = tipc_node_crypto_rx(n);
1778 			if (unlikely(!rx))
1779 				goto free_skb;
1780 		}
1781 
1782 		/* Skip cloning this time as we had a RX pending key */
1783 		if (rx->key.pending)
1784 			goto rcv;
1785 		if (tipc_aead_clone(&tmp, aead) < 0)
1786 			goto rcv;
1787 		if (tipc_crypto_key_attach(rx, tmp, ehdr->tx_key) < 0) {
1788 			tipc_aead_free(&tmp->rcu);
1789 			goto rcv;
1790 		}
1791 		tipc_aead_put(aead);
1792 		aead = tipc_aead_get(tmp);
1793 	}
1794 
1795 	if (unlikely(err)) {
1796 		tipc_aead_users_dec(aead, INT_MIN);
1797 		goto free_skb;
1798 	}
1799 
1800 	/* Set the RX key's user */
1801 	tipc_aead_users_set(aead, 1);
1802 
1803 rcv:
1804 	/* Mark this point, RX works */
1805 	rx->timer1 = jiffies;
1806 
1807 	/* Remove ehdr & auth. tag prior to tipc_rcv() */
1808 	ehdr = (struct tipc_ehdr *)(*skb)->data;
1809 	destined = ehdr->destined;
1810 	rx_key_active = ehdr->rx_key_active;
1811 	skb_pull(*skb, tipc_ehdr_size(ehdr));
1812 	pskb_trim(*skb, (*skb)->len - aead->authsize);
1813 
1814 	/* Validate TIPCv2 message */
1815 	if (unlikely(!tipc_msg_validate(skb))) {
1816 		pr_err_ratelimited("Packet dropped after decryption!\n");
1817 		goto free_skb;
1818 	}
1819 
1820 	/* Update peer RX active key & TX users */
1821 	if (destined)
1822 		tipc_crypto_key_synch(rx, rx_key_active, buf_msg(*skb));
1823 
1824 	/* Mark skb decrypted */
1825 	skb_cb->decrypted = 1;
1826 
1827 	/* Clear clone cxt if any */
1828 	if (likely(!skb_cb->tx_clone_deferred))
1829 		goto exit;
1830 	skb_cb->tx_clone_deferred = 0;
1831 	memset(&skb_cb->tx_clone_ctx, 0, sizeof(skb_cb->tx_clone_ctx));
1832 	goto exit;
1833 
1834 free_skb:
1835 	kfree_skb(*skb);
1836 	*skb = NULL;
1837 
1838 exit:
1839 	tipc_aead_put(aead);
1840 	if (rx)
1841 		tipc_node_put(rx->node);
1842 }
1843 
1844 static void tipc_crypto_do_cmd(struct net *net, int cmd)
1845 {
1846 	struct tipc_net *tn = tipc_net(net);
1847 	struct tipc_crypto *tx = tn->crypto_tx, *rx;
1848 	struct list_head *p;
1849 	unsigned int stat;
1850 	int i, j, cpu;
1851 	char buf[200];
1852 
1853 	/* Currently only one command is supported */
1854 	switch (cmd) {
1855 	case 0xfff1:
1856 		goto print_stats;
1857 	default:
1858 		return;
1859 	}
1860 
1861 print_stats:
1862 	/* Print a header */
1863 	pr_info("\n=============== TIPC Crypto Statistics ===============\n\n");
1864 
1865 	/* Print key status */
1866 	pr_info("Key status:\n");
1867 	pr_info("TX(%7.7s)\n%s", tipc_own_id_string(net),
1868 		tipc_crypto_key_dump(tx, buf));
1869 
1870 	rcu_read_lock();
1871 	for (p = tn->node_list.next; p != &tn->node_list; p = p->next) {
1872 		rx = tipc_node_crypto_rx_by_list(p);
1873 		pr_info("RX(%7.7s)\n%s", tipc_node_get_id_str(rx->node),
1874 			tipc_crypto_key_dump(rx, buf));
1875 	}
1876 	rcu_read_unlock();
1877 
1878 	/* Print crypto statistics */
1879 	for (i = 0, j = 0; i < MAX_STATS; i++)
1880 		j += scnprintf(buf + j, 200 - j, "|%11s ", hstats[i]);
1881 	pr_info("\nCounter     %s", buf);
1882 
1883 	memset(buf, '-', 115);
1884 	buf[115] = '\0';
1885 	pr_info("%s\n", buf);
1886 
1887 	j = scnprintf(buf, 200, "TX(%7.7s) ", tipc_own_id_string(net));
1888 	for_each_possible_cpu(cpu) {
1889 		for (i = 0; i < MAX_STATS; i++) {
1890 			stat = per_cpu_ptr(tx->stats, cpu)->stat[i];
1891 			j += scnprintf(buf + j, 200 - j, "|%11d ", stat);
1892 		}
1893 		pr_info("%s", buf);
1894 		j = scnprintf(buf, 200, "%12s", " ");
1895 	}
1896 
1897 	rcu_read_lock();
1898 	for (p = tn->node_list.next; p != &tn->node_list; p = p->next) {
1899 		rx = tipc_node_crypto_rx_by_list(p);
1900 		j = scnprintf(buf, 200, "RX(%7.7s) ",
1901 			      tipc_node_get_id_str(rx->node));
1902 		for_each_possible_cpu(cpu) {
1903 			for (i = 0; i < MAX_STATS; i++) {
1904 				stat = per_cpu_ptr(rx->stats, cpu)->stat[i];
1905 				j += scnprintf(buf + j, 200 - j, "|%11d ",
1906 					       stat);
1907 			}
1908 			pr_info("%s", buf);
1909 			j = scnprintf(buf, 200, "%12s", " ");
1910 		}
1911 	}
1912 	rcu_read_unlock();
1913 
1914 	pr_info("\n======================== Done ========================\n");
1915 }
1916 
1917 static char *tipc_crypto_key_dump(struct tipc_crypto *c, char *buf)
1918 {
1919 	struct tipc_key key = c->key;
1920 	struct tipc_aead *aead;
1921 	int k, i = 0;
1922 	char *s;
1923 
1924 	for (k = KEY_MIN; k <= KEY_MAX; k++) {
1925 		if (k == key.passive)
1926 			s = "PAS";
1927 		else if (k == key.active)
1928 			s = "ACT";
1929 		else if (k == key.pending)
1930 			s = "PEN";
1931 		else
1932 			s = "-";
1933 		i += scnprintf(buf + i, 200 - i, "\tKey%d: %s", k, s);
1934 
1935 		rcu_read_lock();
1936 		aead = rcu_dereference(c->aead[k]);
1937 		if (aead)
1938 			i += scnprintf(buf + i, 200 - i,
1939 				       "{\"%s...\", \"%s\"}/%d:%d",
1940 				       aead->hint,
1941 				       (aead->mode == CLUSTER_KEY) ? "c" : "p",
1942 				       atomic_read(&aead->users),
1943 				       refcount_read(&aead->refcnt));
1944 		rcu_read_unlock();
1945 		i += scnprintf(buf + i, 200 - i, "\n");
1946 	}
1947 
1948 	if (c->node)
1949 		i += scnprintf(buf + i, 200 - i, "\tPeer RX active: %d\n",
1950 			       atomic_read(&c->peer_rx_active));
1951 
1952 	return buf;
1953 }
1954 
1955 #ifdef TIPC_CRYPTO_DEBUG
1956 static char *tipc_key_change_dump(struct tipc_key old, struct tipc_key new,
1957 				  char *buf)
1958 {
1959 	struct tipc_key *key = &old;
1960 	int k, i = 0;
1961 	char *s;
1962 
1963 	/* Output format: "[%s %s %s] -> [%s %s %s]", max len = 32 */
1964 again:
1965 	i += scnprintf(buf + i, 32 - i, "[");
1966 	for (k = KEY_MIN; k <= KEY_MAX; k++) {
1967 		if (k == key->passive)
1968 			s = "pas";
1969 		else if (k == key->active)
1970 			s = "act";
1971 		else if (k == key->pending)
1972 			s = "pen";
1973 		else
1974 			s = "-";
1975 		i += scnprintf(buf + i, 32 - i,
1976 			       (k != KEY_MAX) ? "%s " : "%s", s);
1977 	}
1978 	if (key != &new) {
1979 		i += scnprintf(buf + i, 32 - i, "] -> ");
1980 		key = &new;
1981 		goto again;
1982 	}
1983 	i += scnprintf(buf + i, 32 - i, "]");
1984 	return buf;
1985 }
1986 #endif
1987