xref: /openbmc/linux/crypto/cryptd.c (revision f3a8b664)
1 /*
2  * Software async crypto daemon.
3  *
4  * Copyright (c) 2006 Herbert Xu <herbert@gondor.apana.org.au>
5  *
6  * Added AEAD support to cryptd.
7  *    Authors: Tadeusz Struk (tadeusz.struk@intel.com)
8  *             Adrian Hoban <adrian.hoban@intel.com>
9  *             Gabriele Paoloni <gabriele.paoloni@intel.com>
10  *             Aidan O'Mahony (aidan.o.mahony@intel.com)
11  *    Copyright (c) 2010, Intel Corporation.
12  *
13  * This program is free software; you can redistribute it and/or modify it
14  * under the terms of the GNU General Public License as published by the Free
15  * Software Foundation; either version 2 of the License, or (at your option)
16  * any later version.
17  *
18  */
19 
20 #include <crypto/algapi.h>
21 #include <crypto/internal/hash.h>
22 #include <crypto/internal/aead.h>
23 #include <crypto/cryptd.h>
24 #include <crypto/crypto_wq.h>
25 #include <linux/atomic.h>
26 #include <linux/err.h>
27 #include <linux/init.h>
28 #include <linux/kernel.h>
29 #include <linux/list.h>
30 #include <linux/module.h>
31 #include <linux/scatterlist.h>
32 #include <linux/sched.h>
33 #include <linux/slab.h>
34 
35 #define CRYPTD_MAX_CPU_QLEN 1000
36 
37 struct cryptd_cpu_queue {
38 	struct crypto_queue queue;
39 	struct work_struct work;
40 };
41 
42 struct cryptd_queue {
43 	struct cryptd_cpu_queue __percpu *cpu_queue;
44 };
45 
46 struct cryptd_instance_ctx {
47 	struct crypto_spawn spawn;
48 	struct cryptd_queue *queue;
49 };
50 
51 struct hashd_instance_ctx {
52 	struct crypto_shash_spawn spawn;
53 	struct cryptd_queue *queue;
54 };
55 
56 struct aead_instance_ctx {
57 	struct crypto_aead_spawn aead_spawn;
58 	struct cryptd_queue *queue;
59 };
60 
61 struct cryptd_blkcipher_ctx {
62 	atomic_t refcnt;
63 	struct crypto_blkcipher *child;
64 };
65 
66 struct cryptd_blkcipher_request_ctx {
67 	crypto_completion_t complete;
68 };
69 
70 struct cryptd_hash_ctx {
71 	atomic_t refcnt;
72 	struct crypto_shash *child;
73 };
74 
75 struct cryptd_hash_request_ctx {
76 	crypto_completion_t complete;
77 	struct shash_desc desc;
78 };
79 
80 struct cryptd_aead_ctx {
81 	atomic_t refcnt;
82 	struct crypto_aead *child;
83 };
84 
85 struct cryptd_aead_request_ctx {
86 	crypto_completion_t complete;
87 };
88 
89 static void cryptd_queue_worker(struct work_struct *work);
90 
91 static int cryptd_init_queue(struct cryptd_queue *queue,
92 			     unsigned int max_cpu_qlen)
93 {
94 	int cpu;
95 	struct cryptd_cpu_queue *cpu_queue;
96 
97 	queue->cpu_queue = alloc_percpu(struct cryptd_cpu_queue);
98 	if (!queue->cpu_queue)
99 		return -ENOMEM;
100 	for_each_possible_cpu(cpu) {
101 		cpu_queue = per_cpu_ptr(queue->cpu_queue, cpu);
102 		crypto_init_queue(&cpu_queue->queue, max_cpu_qlen);
103 		INIT_WORK(&cpu_queue->work, cryptd_queue_worker);
104 	}
105 	return 0;
106 }
107 
108 static void cryptd_fini_queue(struct cryptd_queue *queue)
109 {
110 	int cpu;
111 	struct cryptd_cpu_queue *cpu_queue;
112 
113 	for_each_possible_cpu(cpu) {
114 		cpu_queue = per_cpu_ptr(queue->cpu_queue, cpu);
115 		BUG_ON(cpu_queue->queue.qlen);
116 	}
117 	free_percpu(queue->cpu_queue);
118 }
119 
120 static int cryptd_enqueue_request(struct cryptd_queue *queue,
121 				  struct crypto_async_request *request)
122 {
123 	int cpu, err;
124 	struct cryptd_cpu_queue *cpu_queue;
125 	struct crypto_tfm *tfm;
126 	atomic_t *refcnt;
127 	bool may_backlog;
128 
129 	cpu = get_cpu();
130 	cpu_queue = this_cpu_ptr(queue->cpu_queue);
131 	err = crypto_enqueue_request(&cpu_queue->queue, request);
132 
133 	refcnt = crypto_tfm_ctx(request->tfm);
134 	may_backlog = request->flags & CRYPTO_TFM_REQ_MAY_BACKLOG;
135 
136 	if (err == -EBUSY && !may_backlog)
137 		goto out_put_cpu;
138 
139 	queue_work_on(cpu, kcrypto_wq, &cpu_queue->work);
140 
141 	if (!atomic_read(refcnt))
142 		goto out_put_cpu;
143 
144 	tfm = request->tfm;
145 	atomic_inc(refcnt);
146 
147 out_put_cpu:
148 	put_cpu();
149 
150 	return err;
151 }
152 
153 /* Called in workqueue context, do one real cryption work (via
154  * req->complete) and reschedule itself if there are more work to
155  * do. */
156 static void cryptd_queue_worker(struct work_struct *work)
157 {
158 	struct cryptd_cpu_queue *cpu_queue;
159 	struct crypto_async_request *req, *backlog;
160 
161 	cpu_queue = container_of(work, struct cryptd_cpu_queue, work);
162 	/*
163 	 * Only handle one request at a time to avoid hogging crypto workqueue.
164 	 * preempt_disable/enable is used to prevent being preempted by
165 	 * cryptd_enqueue_request(). local_bh_disable/enable is used to prevent
166 	 * cryptd_enqueue_request() being accessed from software interrupts.
167 	 */
168 	local_bh_disable();
169 	preempt_disable();
170 	backlog = crypto_get_backlog(&cpu_queue->queue);
171 	req = crypto_dequeue_request(&cpu_queue->queue);
172 	preempt_enable();
173 	local_bh_enable();
174 
175 	if (!req)
176 		return;
177 
178 	if (backlog)
179 		backlog->complete(backlog, -EINPROGRESS);
180 	req->complete(req, 0);
181 
182 	if (cpu_queue->queue.qlen)
183 		queue_work(kcrypto_wq, &cpu_queue->work);
184 }
185 
186 static inline struct cryptd_queue *cryptd_get_queue(struct crypto_tfm *tfm)
187 {
188 	struct crypto_instance *inst = crypto_tfm_alg_instance(tfm);
189 	struct cryptd_instance_ctx *ictx = crypto_instance_ctx(inst);
190 	return ictx->queue;
191 }
192 
193 static inline void cryptd_check_internal(struct rtattr **tb, u32 *type,
194 					 u32 *mask)
195 {
196 	struct crypto_attr_type *algt;
197 
198 	algt = crypto_get_attr_type(tb);
199 	if (IS_ERR(algt))
200 		return;
201 
202 	*type |= algt->type & CRYPTO_ALG_INTERNAL;
203 	*mask |= algt->mask & CRYPTO_ALG_INTERNAL;
204 }
205 
206 static int cryptd_blkcipher_setkey(struct crypto_ablkcipher *parent,
207 				   const u8 *key, unsigned int keylen)
208 {
209 	struct cryptd_blkcipher_ctx *ctx = crypto_ablkcipher_ctx(parent);
210 	struct crypto_blkcipher *child = ctx->child;
211 	int err;
212 
213 	crypto_blkcipher_clear_flags(child, CRYPTO_TFM_REQ_MASK);
214 	crypto_blkcipher_set_flags(child, crypto_ablkcipher_get_flags(parent) &
215 					  CRYPTO_TFM_REQ_MASK);
216 	err = crypto_blkcipher_setkey(child, key, keylen);
217 	crypto_ablkcipher_set_flags(parent, crypto_blkcipher_get_flags(child) &
218 					    CRYPTO_TFM_RES_MASK);
219 	return err;
220 }
221 
222 static void cryptd_blkcipher_crypt(struct ablkcipher_request *req,
223 				   struct crypto_blkcipher *child,
224 				   int err,
225 				   int (*crypt)(struct blkcipher_desc *desc,
226 						struct scatterlist *dst,
227 						struct scatterlist *src,
228 						unsigned int len))
229 {
230 	struct cryptd_blkcipher_request_ctx *rctx;
231 	struct cryptd_blkcipher_ctx *ctx;
232 	struct crypto_ablkcipher *tfm;
233 	struct blkcipher_desc desc;
234 	int refcnt;
235 
236 	rctx = ablkcipher_request_ctx(req);
237 
238 	if (unlikely(err == -EINPROGRESS))
239 		goto out;
240 
241 	desc.tfm = child;
242 	desc.info = req->info;
243 	desc.flags = CRYPTO_TFM_REQ_MAY_SLEEP;
244 
245 	err = crypt(&desc, req->dst, req->src, req->nbytes);
246 
247 	req->base.complete = rctx->complete;
248 
249 out:
250 	tfm = crypto_ablkcipher_reqtfm(req);
251 	ctx = crypto_ablkcipher_ctx(tfm);
252 	refcnt = atomic_read(&ctx->refcnt);
253 
254 	local_bh_disable();
255 	rctx->complete(&req->base, err);
256 	local_bh_enable();
257 
258 	if (err != -EINPROGRESS && refcnt && atomic_dec_and_test(&ctx->refcnt))
259 		crypto_free_ablkcipher(tfm);
260 }
261 
262 static void cryptd_blkcipher_encrypt(struct crypto_async_request *req, int err)
263 {
264 	struct cryptd_blkcipher_ctx *ctx = crypto_tfm_ctx(req->tfm);
265 	struct crypto_blkcipher *child = ctx->child;
266 
267 	cryptd_blkcipher_crypt(ablkcipher_request_cast(req), child, err,
268 			       crypto_blkcipher_crt(child)->encrypt);
269 }
270 
271 static void cryptd_blkcipher_decrypt(struct crypto_async_request *req, int err)
272 {
273 	struct cryptd_blkcipher_ctx *ctx = crypto_tfm_ctx(req->tfm);
274 	struct crypto_blkcipher *child = ctx->child;
275 
276 	cryptd_blkcipher_crypt(ablkcipher_request_cast(req), child, err,
277 			       crypto_blkcipher_crt(child)->decrypt);
278 }
279 
280 static int cryptd_blkcipher_enqueue(struct ablkcipher_request *req,
281 				    crypto_completion_t compl)
282 {
283 	struct cryptd_blkcipher_request_ctx *rctx = ablkcipher_request_ctx(req);
284 	struct crypto_ablkcipher *tfm = crypto_ablkcipher_reqtfm(req);
285 	struct cryptd_queue *queue;
286 
287 	queue = cryptd_get_queue(crypto_ablkcipher_tfm(tfm));
288 	rctx->complete = req->base.complete;
289 	req->base.complete = compl;
290 
291 	return cryptd_enqueue_request(queue, &req->base);
292 }
293 
294 static int cryptd_blkcipher_encrypt_enqueue(struct ablkcipher_request *req)
295 {
296 	return cryptd_blkcipher_enqueue(req, cryptd_blkcipher_encrypt);
297 }
298 
299 static int cryptd_blkcipher_decrypt_enqueue(struct ablkcipher_request *req)
300 {
301 	return cryptd_blkcipher_enqueue(req, cryptd_blkcipher_decrypt);
302 }
303 
304 static int cryptd_blkcipher_init_tfm(struct crypto_tfm *tfm)
305 {
306 	struct crypto_instance *inst = crypto_tfm_alg_instance(tfm);
307 	struct cryptd_instance_ctx *ictx = crypto_instance_ctx(inst);
308 	struct crypto_spawn *spawn = &ictx->spawn;
309 	struct cryptd_blkcipher_ctx *ctx = crypto_tfm_ctx(tfm);
310 	struct crypto_blkcipher *cipher;
311 
312 	cipher = crypto_spawn_blkcipher(spawn);
313 	if (IS_ERR(cipher))
314 		return PTR_ERR(cipher);
315 
316 	ctx->child = cipher;
317 	tfm->crt_ablkcipher.reqsize =
318 		sizeof(struct cryptd_blkcipher_request_ctx);
319 	return 0;
320 }
321 
322 static void cryptd_blkcipher_exit_tfm(struct crypto_tfm *tfm)
323 {
324 	struct cryptd_blkcipher_ctx *ctx = crypto_tfm_ctx(tfm);
325 
326 	crypto_free_blkcipher(ctx->child);
327 }
328 
329 static int cryptd_init_instance(struct crypto_instance *inst,
330 				struct crypto_alg *alg)
331 {
332 	if (snprintf(inst->alg.cra_driver_name, CRYPTO_MAX_ALG_NAME,
333 		     "cryptd(%s)",
334 		     alg->cra_driver_name) >= CRYPTO_MAX_ALG_NAME)
335 		return -ENAMETOOLONG;
336 
337 	memcpy(inst->alg.cra_name, alg->cra_name, CRYPTO_MAX_ALG_NAME);
338 
339 	inst->alg.cra_priority = alg->cra_priority + 50;
340 	inst->alg.cra_blocksize = alg->cra_blocksize;
341 	inst->alg.cra_alignmask = alg->cra_alignmask;
342 
343 	return 0;
344 }
345 
346 static void *cryptd_alloc_instance(struct crypto_alg *alg, unsigned int head,
347 				   unsigned int tail)
348 {
349 	char *p;
350 	struct crypto_instance *inst;
351 	int err;
352 
353 	p = kzalloc(head + sizeof(*inst) + tail, GFP_KERNEL);
354 	if (!p)
355 		return ERR_PTR(-ENOMEM);
356 
357 	inst = (void *)(p + head);
358 
359 	err = cryptd_init_instance(inst, alg);
360 	if (err)
361 		goto out_free_inst;
362 
363 out:
364 	return p;
365 
366 out_free_inst:
367 	kfree(p);
368 	p = ERR_PTR(err);
369 	goto out;
370 }
371 
372 static int cryptd_create_blkcipher(struct crypto_template *tmpl,
373 				   struct rtattr **tb,
374 				   struct cryptd_queue *queue)
375 {
376 	struct cryptd_instance_ctx *ctx;
377 	struct crypto_instance *inst;
378 	struct crypto_alg *alg;
379 	u32 type = CRYPTO_ALG_TYPE_BLKCIPHER;
380 	u32 mask = CRYPTO_ALG_TYPE_MASK;
381 	int err;
382 
383 	cryptd_check_internal(tb, &type, &mask);
384 
385 	alg = crypto_get_attr_alg(tb, type, mask);
386 	if (IS_ERR(alg))
387 		return PTR_ERR(alg);
388 
389 	inst = cryptd_alloc_instance(alg, 0, sizeof(*ctx));
390 	err = PTR_ERR(inst);
391 	if (IS_ERR(inst))
392 		goto out_put_alg;
393 
394 	ctx = crypto_instance_ctx(inst);
395 	ctx->queue = queue;
396 
397 	err = crypto_init_spawn(&ctx->spawn, alg, inst,
398 				CRYPTO_ALG_TYPE_MASK | CRYPTO_ALG_ASYNC);
399 	if (err)
400 		goto out_free_inst;
401 
402 	type = CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC;
403 	if (alg->cra_flags & CRYPTO_ALG_INTERNAL)
404 		type |= CRYPTO_ALG_INTERNAL;
405 	inst->alg.cra_flags = type;
406 	inst->alg.cra_type = &crypto_ablkcipher_type;
407 
408 	inst->alg.cra_ablkcipher.ivsize = alg->cra_blkcipher.ivsize;
409 	inst->alg.cra_ablkcipher.min_keysize = alg->cra_blkcipher.min_keysize;
410 	inst->alg.cra_ablkcipher.max_keysize = alg->cra_blkcipher.max_keysize;
411 
412 	inst->alg.cra_ablkcipher.geniv = alg->cra_blkcipher.geniv;
413 
414 	inst->alg.cra_ctxsize = sizeof(struct cryptd_blkcipher_ctx);
415 
416 	inst->alg.cra_init = cryptd_blkcipher_init_tfm;
417 	inst->alg.cra_exit = cryptd_blkcipher_exit_tfm;
418 
419 	inst->alg.cra_ablkcipher.setkey = cryptd_blkcipher_setkey;
420 	inst->alg.cra_ablkcipher.encrypt = cryptd_blkcipher_encrypt_enqueue;
421 	inst->alg.cra_ablkcipher.decrypt = cryptd_blkcipher_decrypt_enqueue;
422 
423 	err = crypto_register_instance(tmpl, inst);
424 	if (err) {
425 		crypto_drop_spawn(&ctx->spawn);
426 out_free_inst:
427 		kfree(inst);
428 	}
429 
430 out_put_alg:
431 	crypto_mod_put(alg);
432 	return err;
433 }
434 
435 static int cryptd_hash_init_tfm(struct crypto_tfm *tfm)
436 {
437 	struct crypto_instance *inst = crypto_tfm_alg_instance(tfm);
438 	struct hashd_instance_ctx *ictx = crypto_instance_ctx(inst);
439 	struct crypto_shash_spawn *spawn = &ictx->spawn;
440 	struct cryptd_hash_ctx *ctx = crypto_tfm_ctx(tfm);
441 	struct crypto_shash *hash;
442 
443 	hash = crypto_spawn_shash(spawn);
444 	if (IS_ERR(hash))
445 		return PTR_ERR(hash);
446 
447 	ctx->child = hash;
448 	crypto_ahash_set_reqsize(__crypto_ahash_cast(tfm),
449 				 sizeof(struct cryptd_hash_request_ctx) +
450 				 crypto_shash_descsize(hash));
451 	return 0;
452 }
453 
454 static void cryptd_hash_exit_tfm(struct crypto_tfm *tfm)
455 {
456 	struct cryptd_hash_ctx *ctx = crypto_tfm_ctx(tfm);
457 
458 	crypto_free_shash(ctx->child);
459 }
460 
461 static int cryptd_hash_setkey(struct crypto_ahash *parent,
462 				   const u8 *key, unsigned int keylen)
463 {
464 	struct cryptd_hash_ctx *ctx   = crypto_ahash_ctx(parent);
465 	struct crypto_shash *child = ctx->child;
466 	int err;
467 
468 	crypto_shash_clear_flags(child, CRYPTO_TFM_REQ_MASK);
469 	crypto_shash_set_flags(child, crypto_ahash_get_flags(parent) &
470 				      CRYPTO_TFM_REQ_MASK);
471 	err = crypto_shash_setkey(child, key, keylen);
472 	crypto_ahash_set_flags(parent, crypto_shash_get_flags(child) &
473 				       CRYPTO_TFM_RES_MASK);
474 	return err;
475 }
476 
477 static int cryptd_hash_enqueue(struct ahash_request *req,
478 				crypto_completion_t compl)
479 {
480 	struct cryptd_hash_request_ctx *rctx = ahash_request_ctx(req);
481 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
482 	struct cryptd_queue *queue =
483 		cryptd_get_queue(crypto_ahash_tfm(tfm));
484 
485 	rctx->complete = req->base.complete;
486 	req->base.complete = compl;
487 
488 	return cryptd_enqueue_request(queue, &req->base);
489 }
490 
491 static void cryptd_hash_complete(struct ahash_request *req, int err)
492 {
493 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
494 	struct cryptd_hash_ctx *ctx = crypto_ahash_ctx(tfm);
495 	struct cryptd_hash_request_ctx *rctx = ahash_request_ctx(req);
496 	int refcnt = atomic_read(&ctx->refcnt);
497 
498 	local_bh_disable();
499 	rctx->complete(&req->base, err);
500 	local_bh_enable();
501 
502 	if (err != -EINPROGRESS && refcnt && atomic_dec_and_test(&ctx->refcnt))
503 		crypto_free_ahash(tfm);
504 }
505 
506 static void cryptd_hash_init(struct crypto_async_request *req_async, int err)
507 {
508 	struct cryptd_hash_ctx *ctx = crypto_tfm_ctx(req_async->tfm);
509 	struct crypto_shash *child = ctx->child;
510 	struct ahash_request *req = ahash_request_cast(req_async);
511 	struct cryptd_hash_request_ctx *rctx = ahash_request_ctx(req);
512 	struct shash_desc *desc = &rctx->desc;
513 
514 	if (unlikely(err == -EINPROGRESS))
515 		goto out;
516 
517 	desc->tfm = child;
518 	desc->flags = CRYPTO_TFM_REQ_MAY_SLEEP;
519 
520 	err = crypto_shash_init(desc);
521 
522 	req->base.complete = rctx->complete;
523 
524 out:
525 	cryptd_hash_complete(req, err);
526 }
527 
528 static int cryptd_hash_init_enqueue(struct ahash_request *req)
529 {
530 	return cryptd_hash_enqueue(req, cryptd_hash_init);
531 }
532 
533 static void cryptd_hash_update(struct crypto_async_request *req_async, int err)
534 {
535 	struct ahash_request *req = ahash_request_cast(req_async);
536 	struct cryptd_hash_request_ctx *rctx;
537 
538 	rctx = ahash_request_ctx(req);
539 
540 	if (unlikely(err == -EINPROGRESS))
541 		goto out;
542 
543 	err = shash_ahash_update(req, &rctx->desc);
544 
545 	req->base.complete = rctx->complete;
546 
547 out:
548 	cryptd_hash_complete(req, err);
549 }
550 
551 static int cryptd_hash_update_enqueue(struct ahash_request *req)
552 {
553 	return cryptd_hash_enqueue(req, cryptd_hash_update);
554 }
555 
556 static void cryptd_hash_final(struct crypto_async_request *req_async, int err)
557 {
558 	struct ahash_request *req = ahash_request_cast(req_async);
559 	struct cryptd_hash_request_ctx *rctx = ahash_request_ctx(req);
560 
561 	if (unlikely(err == -EINPROGRESS))
562 		goto out;
563 
564 	err = crypto_shash_final(&rctx->desc, req->result);
565 
566 	req->base.complete = rctx->complete;
567 
568 out:
569 	cryptd_hash_complete(req, err);
570 }
571 
572 static int cryptd_hash_final_enqueue(struct ahash_request *req)
573 {
574 	return cryptd_hash_enqueue(req, cryptd_hash_final);
575 }
576 
577 static void cryptd_hash_finup(struct crypto_async_request *req_async, int err)
578 {
579 	struct ahash_request *req = ahash_request_cast(req_async);
580 	struct cryptd_hash_request_ctx *rctx = ahash_request_ctx(req);
581 
582 	if (unlikely(err == -EINPROGRESS))
583 		goto out;
584 
585 	err = shash_ahash_finup(req, &rctx->desc);
586 
587 	req->base.complete = rctx->complete;
588 
589 out:
590 	cryptd_hash_complete(req, err);
591 }
592 
593 static int cryptd_hash_finup_enqueue(struct ahash_request *req)
594 {
595 	return cryptd_hash_enqueue(req, cryptd_hash_finup);
596 }
597 
598 static void cryptd_hash_digest(struct crypto_async_request *req_async, int err)
599 {
600 	struct cryptd_hash_ctx *ctx = crypto_tfm_ctx(req_async->tfm);
601 	struct crypto_shash *child = ctx->child;
602 	struct ahash_request *req = ahash_request_cast(req_async);
603 	struct cryptd_hash_request_ctx *rctx = ahash_request_ctx(req);
604 	struct shash_desc *desc = &rctx->desc;
605 
606 	if (unlikely(err == -EINPROGRESS))
607 		goto out;
608 
609 	desc->tfm = child;
610 	desc->flags = CRYPTO_TFM_REQ_MAY_SLEEP;
611 
612 	err = shash_ahash_digest(req, desc);
613 
614 	req->base.complete = rctx->complete;
615 
616 out:
617 	cryptd_hash_complete(req, err);
618 }
619 
620 static int cryptd_hash_digest_enqueue(struct ahash_request *req)
621 {
622 	return cryptd_hash_enqueue(req, cryptd_hash_digest);
623 }
624 
625 static int cryptd_hash_export(struct ahash_request *req, void *out)
626 {
627 	struct cryptd_hash_request_ctx *rctx = ahash_request_ctx(req);
628 
629 	return crypto_shash_export(&rctx->desc, out);
630 }
631 
632 static int cryptd_hash_import(struct ahash_request *req, const void *in)
633 {
634 	struct crypto_ahash *tfm = crypto_ahash_reqtfm(req);
635 	struct cryptd_hash_ctx *ctx = crypto_ahash_ctx(tfm);
636 	struct shash_desc *desc = cryptd_shash_desc(req);
637 
638 	desc->tfm = ctx->child;
639 	desc->flags = req->base.flags;
640 
641 	return crypto_shash_import(desc, in);
642 }
643 
644 static int cryptd_create_hash(struct crypto_template *tmpl, struct rtattr **tb,
645 			      struct cryptd_queue *queue)
646 {
647 	struct hashd_instance_ctx *ctx;
648 	struct ahash_instance *inst;
649 	struct shash_alg *salg;
650 	struct crypto_alg *alg;
651 	u32 type = 0;
652 	u32 mask = 0;
653 	int err;
654 
655 	cryptd_check_internal(tb, &type, &mask);
656 
657 	salg = shash_attr_alg(tb[1], type, mask);
658 	if (IS_ERR(salg))
659 		return PTR_ERR(salg);
660 
661 	alg = &salg->base;
662 	inst = cryptd_alloc_instance(alg, ahash_instance_headroom(),
663 				     sizeof(*ctx));
664 	err = PTR_ERR(inst);
665 	if (IS_ERR(inst))
666 		goto out_put_alg;
667 
668 	ctx = ahash_instance_ctx(inst);
669 	ctx->queue = queue;
670 
671 	err = crypto_init_shash_spawn(&ctx->spawn, salg,
672 				      ahash_crypto_instance(inst));
673 	if (err)
674 		goto out_free_inst;
675 
676 	type = CRYPTO_ALG_ASYNC;
677 	if (alg->cra_flags & CRYPTO_ALG_INTERNAL)
678 		type |= CRYPTO_ALG_INTERNAL;
679 	inst->alg.halg.base.cra_flags = type;
680 
681 	inst->alg.halg.digestsize = salg->digestsize;
682 	inst->alg.halg.statesize = salg->statesize;
683 	inst->alg.halg.base.cra_ctxsize = sizeof(struct cryptd_hash_ctx);
684 
685 	inst->alg.halg.base.cra_init = cryptd_hash_init_tfm;
686 	inst->alg.halg.base.cra_exit = cryptd_hash_exit_tfm;
687 
688 	inst->alg.init   = cryptd_hash_init_enqueue;
689 	inst->alg.update = cryptd_hash_update_enqueue;
690 	inst->alg.final  = cryptd_hash_final_enqueue;
691 	inst->alg.finup  = cryptd_hash_finup_enqueue;
692 	inst->alg.export = cryptd_hash_export;
693 	inst->alg.import = cryptd_hash_import;
694 	inst->alg.setkey = cryptd_hash_setkey;
695 	inst->alg.digest = cryptd_hash_digest_enqueue;
696 
697 	err = ahash_register_instance(tmpl, inst);
698 	if (err) {
699 		crypto_drop_shash(&ctx->spawn);
700 out_free_inst:
701 		kfree(inst);
702 	}
703 
704 out_put_alg:
705 	crypto_mod_put(alg);
706 	return err;
707 }
708 
709 static int cryptd_aead_setkey(struct crypto_aead *parent,
710 			      const u8 *key, unsigned int keylen)
711 {
712 	struct cryptd_aead_ctx *ctx = crypto_aead_ctx(parent);
713 	struct crypto_aead *child = ctx->child;
714 
715 	return crypto_aead_setkey(child, key, keylen);
716 }
717 
718 static int cryptd_aead_setauthsize(struct crypto_aead *parent,
719 				   unsigned int authsize)
720 {
721 	struct cryptd_aead_ctx *ctx = crypto_aead_ctx(parent);
722 	struct crypto_aead *child = ctx->child;
723 
724 	return crypto_aead_setauthsize(child, authsize);
725 }
726 
727 static void cryptd_aead_crypt(struct aead_request *req,
728 			struct crypto_aead *child,
729 			int err,
730 			int (*crypt)(struct aead_request *req))
731 {
732 	struct cryptd_aead_request_ctx *rctx;
733 	struct cryptd_aead_ctx *ctx;
734 	crypto_completion_t compl;
735 	struct crypto_aead *tfm;
736 	int refcnt;
737 
738 	rctx = aead_request_ctx(req);
739 	compl = rctx->complete;
740 
741 	tfm = crypto_aead_reqtfm(req);
742 
743 	if (unlikely(err == -EINPROGRESS))
744 		goto out;
745 	aead_request_set_tfm(req, child);
746 	err = crypt( req );
747 
748 out:
749 	ctx = crypto_aead_ctx(tfm);
750 	refcnt = atomic_read(&ctx->refcnt);
751 
752 	local_bh_disable();
753 	compl(&req->base, err);
754 	local_bh_enable();
755 
756 	if (err != -EINPROGRESS && refcnt && atomic_dec_and_test(&ctx->refcnt))
757 		crypto_free_aead(tfm);
758 }
759 
760 static void cryptd_aead_encrypt(struct crypto_async_request *areq, int err)
761 {
762 	struct cryptd_aead_ctx *ctx = crypto_tfm_ctx(areq->tfm);
763 	struct crypto_aead *child = ctx->child;
764 	struct aead_request *req;
765 
766 	req = container_of(areq, struct aead_request, base);
767 	cryptd_aead_crypt(req, child, err, crypto_aead_alg(child)->encrypt);
768 }
769 
770 static void cryptd_aead_decrypt(struct crypto_async_request *areq, int err)
771 {
772 	struct cryptd_aead_ctx *ctx = crypto_tfm_ctx(areq->tfm);
773 	struct crypto_aead *child = ctx->child;
774 	struct aead_request *req;
775 
776 	req = container_of(areq, struct aead_request, base);
777 	cryptd_aead_crypt(req, child, err, crypto_aead_alg(child)->decrypt);
778 }
779 
780 static int cryptd_aead_enqueue(struct aead_request *req,
781 				    crypto_completion_t compl)
782 {
783 	struct cryptd_aead_request_ctx *rctx = aead_request_ctx(req);
784 	struct crypto_aead *tfm = crypto_aead_reqtfm(req);
785 	struct cryptd_queue *queue = cryptd_get_queue(crypto_aead_tfm(tfm));
786 
787 	rctx->complete = req->base.complete;
788 	req->base.complete = compl;
789 	return cryptd_enqueue_request(queue, &req->base);
790 }
791 
792 static int cryptd_aead_encrypt_enqueue(struct aead_request *req)
793 {
794 	return cryptd_aead_enqueue(req, cryptd_aead_encrypt );
795 }
796 
797 static int cryptd_aead_decrypt_enqueue(struct aead_request *req)
798 {
799 	return cryptd_aead_enqueue(req, cryptd_aead_decrypt );
800 }
801 
802 static int cryptd_aead_init_tfm(struct crypto_aead *tfm)
803 {
804 	struct aead_instance *inst = aead_alg_instance(tfm);
805 	struct aead_instance_ctx *ictx = aead_instance_ctx(inst);
806 	struct crypto_aead_spawn *spawn = &ictx->aead_spawn;
807 	struct cryptd_aead_ctx *ctx = crypto_aead_ctx(tfm);
808 	struct crypto_aead *cipher;
809 
810 	cipher = crypto_spawn_aead(spawn);
811 	if (IS_ERR(cipher))
812 		return PTR_ERR(cipher);
813 
814 	ctx->child = cipher;
815 	crypto_aead_set_reqsize(
816 		tfm, max((unsigned)sizeof(struct cryptd_aead_request_ctx),
817 			 crypto_aead_reqsize(cipher)));
818 	return 0;
819 }
820 
821 static void cryptd_aead_exit_tfm(struct crypto_aead *tfm)
822 {
823 	struct cryptd_aead_ctx *ctx = crypto_aead_ctx(tfm);
824 	crypto_free_aead(ctx->child);
825 }
826 
827 static int cryptd_create_aead(struct crypto_template *tmpl,
828 		              struct rtattr **tb,
829 			      struct cryptd_queue *queue)
830 {
831 	struct aead_instance_ctx *ctx;
832 	struct aead_instance *inst;
833 	struct aead_alg *alg;
834 	const char *name;
835 	u32 type = 0;
836 	u32 mask = CRYPTO_ALG_ASYNC;
837 	int err;
838 
839 	cryptd_check_internal(tb, &type, &mask);
840 
841 	name = crypto_attr_alg_name(tb[1]);
842 	if (IS_ERR(name))
843 		return PTR_ERR(name);
844 
845 	inst = kzalloc(sizeof(*inst) + sizeof(*ctx), GFP_KERNEL);
846 	if (!inst)
847 		return -ENOMEM;
848 
849 	ctx = aead_instance_ctx(inst);
850 	ctx->queue = queue;
851 
852 	crypto_set_aead_spawn(&ctx->aead_spawn, aead_crypto_instance(inst));
853 	err = crypto_grab_aead(&ctx->aead_spawn, name, type, mask);
854 	if (err)
855 		goto out_free_inst;
856 
857 	alg = crypto_spawn_aead_alg(&ctx->aead_spawn);
858 	err = cryptd_init_instance(aead_crypto_instance(inst), &alg->base);
859 	if (err)
860 		goto out_drop_aead;
861 
862 	inst->alg.base.cra_flags = CRYPTO_ALG_ASYNC |
863 				   (alg->base.cra_flags & CRYPTO_ALG_INTERNAL);
864 	inst->alg.base.cra_ctxsize = sizeof(struct cryptd_aead_ctx);
865 
866 	inst->alg.ivsize = crypto_aead_alg_ivsize(alg);
867 	inst->alg.maxauthsize = crypto_aead_alg_maxauthsize(alg);
868 
869 	inst->alg.init = cryptd_aead_init_tfm;
870 	inst->alg.exit = cryptd_aead_exit_tfm;
871 	inst->alg.setkey = cryptd_aead_setkey;
872 	inst->alg.setauthsize = cryptd_aead_setauthsize;
873 	inst->alg.encrypt = cryptd_aead_encrypt_enqueue;
874 	inst->alg.decrypt = cryptd_aead_decrypt_enqueue;
875 
876 	err = aead_register_instance(tmpl, inst);
877 	if (err) {
878 out_drop_aead:
879 		crypto_drop_aead(&ctx->aead_spawn);
880 out_free_inst:
881 		kfree(inst);
882 	}
883 	return err;
884 }
885 
886 static struct cryptd_queue queue;
887 
888 static int cryptd_create(struct crypto_template *tmpl, struct rtattr **tb)
889 {
890 	struct crypto_attr_type *algt;
891 
892 	algt = crypto_get_attr_type(tb);
893 	if (IS_ERR(algt))
894 		return PTR_ERR(algt);
895 
896 	switch (algt->type & algt->mask & CRYPTO_ALG_TYPE_MASK) {
897 	case CRYPTO_ALG_TYPE_BLKCIPHER:
898 		return cryptd_create_blkcipher(tmpl, tb, &queue);
899 	case CRYPTO_ALG_TYPE_DIGEST:
900 		return cryptd_create_hash(tmpl, tb, &queue);
901 	case CRYPTO_ALG_TYPE_AEAD:
902 		return cryptd_create_aead(tmpl, tb, &queue);
903 	}
904 
905 	return -EINVAL;
906 }
907 
908 static void cryptd_free(struct crypto_instance *inst)
909 {
910 	struct cryptd_instance_ctx *ctx = crypto_instance_ctx(inst);
911 	struct hashd_instance_ctx *hctx = crypto_instance_ctx(inst);
912 	struct aead_instance_ctx *aead_ctx = crypto_instance_ctx(inst);
913 
914 	switch (inst->alg.cra_flags & CRYPTO_ALG_TYPE_MASK) {
915 	case CRYPTO_ALG_TYPE_AHASH:
916 		crypto_drop_shash(&hctx->spawn);
917 		kfree(ahash_instance(inst));
918 		return;
919 	case CRYPTO_ALG_TYPE_AEAD:
920 		crypto_drop_aead(&aead_ctx->aead_spawn);
921 		kfree(aead_instance(inst));
922 		return;
923 	default:
924 		crypto_drop_spawn(&ctx->spawn);
925 		kfree(inst);
926 	}
927 }
928 
929 static struct crypto_template cryptd_tmpl = {
930 	.name = "cryptd",
931 	.create = cryptd_create,
932 	.free = cryptd_free,
933 	.module = THIS_MODULE,
934 };
935 
936 struct cryptd_ablkcipher *cryptd_alloc_ablkcipher(const char *alg_name,
937 						  u32 type, u32 mask)
938 {
939 	char cryptd_alg_name[CRYPTO_MAX_ALG_NAME];
940 	struct cryptd_blkcipher_ctx *ctx;
941 	struct crypto_tfm *tfm;
942 
943 	if (snprintf(cryptd_alg_name, CRYPTO_MAX_ALG_NAME,
944 		     "cryptd(%s)", alg_name) >= CRYPTO_MAX_ALG_NAME)
945 		return ERR_PTR(-EINVAL);
946 	type = crypto_skcipher_type(type);
947 	mask &= ~CRYPTO_ALG_TYPE_MASK;
948 	mask |= (CRYPTO_ALG_GENIV | CRYPTO_ALG_TYPE_BLKCIPHER_MASK);
949 	tfm = crypto_alloc_base(cryptd_alg_name, type, mask);
950 	if (IS_ERR(tfm))
951 		return ERR_CAST(tfm);
952 	if (tfm->__crt_alg->cra_module != THIS_MODULE) {
953 		crypto_free_tfm(tfm);
954 		return ERR_PTR(-EINVAL);
955 	}
956 
957 	ctx = crypto_tfm_ctx(tfm);
958 	atomic_set(&ctx->refcnt, 1);
959 
960 	return __cryptd_ablkcipher_cast(__crypto_ablkcipher_cast(tfm));
961 }
962 EXPORT_SYMBOL_GPL(cryptd_alloc_ablkcipher);
963 
964 struct crypto_blkcipher *cryptd_ablkcipher_child(struct cryptd_ablkcipher *tfm)
965 {
966 	struct cryptd_blkcipher_ctx *ctx = crypto_ablkcipher_ctx(&tfm->base);
967 	return ctx->child;
968 }
969 EXPORT_SYMBOL_GPL(cryptd_ablkcipher_child);
970 
971 bool cryptd_ablkcipher_queued(struct cryptd_ablkcipher *tfm)
972 {
973 	struct cryptd_blkcipher_ctx *ctx = crypto_ablkcipher_ctx(&tfm->base);
974 
975 	return atomic_read(&ctx->refcnt) - 1;
976 }
977 EXPORT_SYMBOL_GPL(cryptd_ablkcipher_queued);
978 
979 void cryptd_free_ablkcipher(struct cryptd_ablkcipher *tfm)
980 {
981 	struct cryptd_blkcipher_ctx *ctx = crypto_ablkcipher_ctx(&tfm->base);
982 
983 	if (atomic_dec_and_test(&ctx->refcnt))
984 		crypto_free_ablkcipher(&tfm->base);
985 }
986 EXPORT_SYMBOL_GPL(cryptd_free_ablkcipher);
987 
988 struct cryptd_ahash *cryptd_alloc_ahash(const char *alg_name,
989 					u32 type, u32 mask)
990 {
991 	char cryptd_alg_name[CRYPTO_MAX_ALG_NAME];
992 	struct cryptd_hash_ctx *ctx;
993 	struct crypto_ahash *tfm;
994 
995 	if (snprintf(cryptd_alg_name, CRYPTO_MAX_ALG_NAME,
996 		     "cryptd(%s)", alg_name) >= CRYPTO_MAX_ALG_NAME)
997 		return ERR_PTR(-EINVAL);
998 	tfm = crypto_alloc_ahash(cryptd_alg_name, type, mask);
999 	if (IS_ERR(tfm))
1000 		return ERR_CAST(tfm);
1001 	if (tfm->base.__crt_alg->cra_module != THIS_MODULE) {
1002 		crypto_free_ahash(tfm);
1003 		return ERR_PTR(-EINVAL);
1004 	}
1005 
1006 	ctx = crypto_ahash_ctx(tfm);
1007 	atomic_set(&ctx->refcnt, 1);
1008 
1009 	return __cryptd_ahash_cast(tfm);
1010 }
1011 EXPORT_SYMBOL_GPL(cryptd_alloc_ahash);
1012 
1013 struct crypto_shash *cryptd_ahash_child(struct cryptd_ahash *tfm)
1014 {
1015 	struct cryptd_hash_ctx *ctx = crypto_ahash_ctx(&tfm->base);
1016 
1017 	return ctx->child;
1018 }
1019 EXPORT_SYMBOL_GPL(cryptd_ahash_child);
1020 
1021 struct shash_desc *cryptd_shash_desc(struct ahash_request *req)
1022 {
1023 	struct cryptd_hash_request_ctx *rctx = ahash_request_ctx(req);
1024 	return &rctx->desc;
1025 }
1026 EXPORT_SYMBOL_GPL(cryptd_shash_desc);
1027 
1028 bool cryptd_ahash_queued(struct cryptd_ahash *tfm)
1029 {
1030 	struct cryptd_hash_ctx *ctx = crypto_ahash_ctx(&tfm->base);
1031 
1032 	return atomic_read(&ctx->refcnt) - 1;
1033 }
1034 EXPORT_SYMBOL_GPL(cryptd_ahash_queued);
1035 
1036 void cryptd_free_ahash(struct cryptd_ahash *tfm)
1037 {
1038 	struct cryptd_hash_ctx *ctx = crypto_ahash_ctx(&tfm->base);
1039 
1040 	if (atomic_dec_and_test(&ctx->refcnt))
1041 		crypto_free_ahash(&tfm->base);
1042 }
1043 EXPORT_SYMBOL_GPL(cryptd_free_ahash);
1044 
1045 struct cryptd_aead *cryptd_alloc_aead(const char *alg_name,
1046 						  u32 type, u32 mask)
1047 {
1048 	char cryptd_alg_name[CRYPTO_MAX_ALG_NAME];
1049 	struct cryptd_aead_ctx *ctx;
1050 	struct crypto_aead *tfm;
1051 
1052 	if (snprintf(cryptd_alg_name, CRYPTO_MAX_ALG_NAME,
1053 		     "cryptd(%s)", alg_name) >= CRYPTO_MAX_ALG_NAME)
1054 		return ERR_PTR(-EINVAL);
1055 	tfm = crypto_alloc_aead(cryptd_alg_name, type, mask);
1056 	if (IS_ERR(tfm))
1057 		return ERR_CAST(tfm);
1058 	if (tfm->base.__crt_alg->cra_module != THIS_MODULE) {
1059 		crypto_free_aead(tfm);
1060 		return ERR_PTR(-EINVAL);
1061 	}
1062 
1063 	ctx = crypto_aead_ctx(tfm);
1064 	atomic_set(&ctx->refcnt, 1);
1065 
1066 	return __cryptd_aead_cast(tfm);
1067 }
1068 EXPORT_SYMBOL_GPL(cryptd_alloc_aead);
1069 
1070 struct crypto_aead *cryptd_aead_child(struct cryptd_aead *tfm)
1071 {
1072 	struct cryptd_aead_ctx *ctx;
1073 	ctx = crypto_aead_ctx(&tfm->base);
1074 	return ctx->child;
1075 }
1076 EXPORT_SYMBOL_GPL(cryptd_aead_child);
1077 
1078 bool cryptd_aead_queued(struct cryptd_aead *tfm)
1079 {
1080 	struct cryptd_aead_ctx *ctx = crypto_aead_ctx(&tfm->base);
1081 
1082 	return atomic_read(&ctx->refcnt) - 1;
1083 }
1084 EXPORT_SYMBOL_GPL(cryptd_aead_queued);
1085 
1086 void cryptd_free_aead(struct cryptd_aead *tfm)
1087 {
1088 	struct cryptd_aead_ctx *ctx = crypto_aead_ctx(&tfm->base);
1089 
1090 	if (atomic_dec_and_test(&ctx->refcnt))
1091 		crypto_free_aead(&tfm->base);
1092 }
1093 EXPORT_SYMBOL_GPL(cryptd_free_aead);
1094 
1095 static int __init cryptd_init(void)
1096 {
1097 	int err;
1098 
1099 	err = cryptd_init_queue(&queue, CRYPTD_MAX_CPU_QLEN);
1100 	if (err)
1101 		return err;
1102 
1103 	err = crypto_register_template(&cryptd_tmpl);
1104 	if (err)
1105 		cryptd_fini_queue(&queue);
1106 
1107 	return err;
1108 }
1109 
1110 static void __exit cryptd_exit(void)
1111 {
1112 	cryptd_fini_queue(&queue);
1113 	crypto_unregister_template(&cryptd_tmpl);
1114 }
1115 
1116 subsys_initcall(cryptd_init);
1117 module_exit(cryptd_exit);
1118 
1119 MODULE_LICENSE("GPL");
1120 MODULE_DESCRIPTION("Software async crypto daemon");
1121 MODULE_ALIAS_CRYPTO("cryptd");
1122