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