1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27#include <linux/kernel.h>
28#include <linux/crypto.h>
29#include <linux/init.h>
30#include <linux/module.h>
31#include <linux/hardirq.h>
32#include <crypto/algapi.h>
33#include <crypto/cryptd.h>
34#include <crypto/ablk_helper.h>
35#include <asm/simd.h>
36
37int ablk_set_key(struct crypto_ablkcipher *tfm, const u8 *key,
38 unsigned int key_len)
39{
40 struct async_helper_ctx *ctx = crypto_ablkcipher_ctx(tfm);
41 struct crypto_ablkcipher *child = &ctx->cryptd_tfm->base;
42 int err;
43
44 crypto_ablkcipher_clear_flags(child, CRYPTO_TFM_REQ_MASK);
45 crypto_ablkcipher_set_flags(child, crypto_ablkcipher_get_flags(tfm)
46 & CRYPTO_TFM_REQ_MASK);
47 err = crypto_ablkcipher_setkey(child, key, key_len);
48 crypto_ablkcipher_set_flags(tfm, crypto_ablkcipher_get_flags(child)
49 & CRYPTO_TFM_RES_MASK);
50 return err;
51}
52EXPORT_SYMBOL_GPL(ablk_set_key);
53
54int __ablk_encrypt(struct ablkcipher_request *req)
55{
56 struct crypto_ablkcipher *tfm = crypto_ablkcipher_reqtfm(req);
57 struct async_helper_ctx *ctx = crypto_ablkcipher_ctx(tfm);
58 struct blkcipher_desc desc;
59
60 desc.tfm = cryptd_ablkcipher_child(ctx->cryptd_tfm);
61 desc.info = req->info;
62 desc.flags = 0;
63
64 return crypto_blkcipher_crt(desc.tfm)->encrypt(
65 &desc, req->dst, req->src, req->nbytes);
66}
67EXPORT_SYMBOL_GPL(__ablk_encrypt);
68
69int ablk_encrypt(struct ablkcipher_request *req)
70{
71 struct crypto_ablkcipher *tfm = crypto_ablkcipher_reqtfm(req);
72 struct async_helper_ctx *ctx = crypto_ablkcipher_ctx(tfm);
73
74 if (!may_use_simd() ||
75 (in_atomic() && cryptd_ablkcipher_queued(ctx->cryptd_tfm))) {
76 struct ablkcipher_request *cryptd_req =
77 ablkcipher_request_ctx(req);
78
79 *cryptd_req = *req;
80 ablkcipher_request_set_tfm(cryptd_req, &ctx->cryptd_tfm->base);
81
82 return crypto_ablkcipher_encrypt(cryptd_req);
83 } else {
84 return __ablk_encrypt(req);
85 }
86}
87EXPORT_SYMBOL_GPL(ablk_encrypt);
88
89int ablk_decrypt(struct ablkcipher_request *req)
90{
91 struct crypto_ablkcipher *tfm = crypto_ablkcipher_reqtfm(req);
92 struct async_helper_ctx *ctx = crypto_ablkcipher_ctx(tfm);
93
94 if (!may_use_simd() ||
95 (in_atomic() && cryptd_ablkcipher_queued(ctx->cryptd_tfm))) {
96 struct ablkcipher_request *cryptd_req =
97 ablkcipher_request_ctx(req);
98
99 *cryptd_req = *req;
100 ablkcipher_request_set_tfm(cryptd_req, &ctx->cryptd_tfm->base);
101
102 return crypto_ablkcipher_decrypt(cryptd_req);
103 } else {
104 struct blkcipher_desc desc;
105
106 desc.tfm = cryptd_ablkcipher_child(ctx->cryptd_tfm);
107 desc.info = req->info;
108 desc.flags = 0;
109
110 return crypto_blkcipher_crt(desc.tfm)->decrypt(
111 &desc, req->dst, req->src, req->nbytes);
112 }
113}
114EXPORT_SYMBOL_GPL(ablk_decrypt);
115
116void ablk_exit(struct crypto_tfm *tfm)
117{
118 struct async_helper_ctx *ctx = crypto_tfm_ctx(tfm);
119
120 cryptd_free_ablkcipher(ctx->cryptd_tfm);
121}
122EXPORT_SYMBOL_GPL(ablk_exit);
123
124int ablk_init_common(struct crypto_tfm *tfm, const char *drv_name)
125{
126 struct async_helper_ctx *ctx = crypto_tfm_ctx(tfm);
127 struct cryptd_ablkcipher *cryptd_tfm;
128
129 cryptd_tfm = cryptd_alloc_ablkcipher(drv_name, CRYPTO_ALG_INTERNAL,
130 CRYPTO_ALG_INTERNAL);
131 if (IS_ERR(cryptd_tfm))
132 return PTR_ERR(cryptd_tfm);
133
134 ctx->cryptd_tfm = cryptd_tfm;
135 tfm->crt_ablkcipher.reqsize = sizeof(struct ablkcipher_request) +
136 crypto_ablkcipher_reqsize(&cryptd_tfm->base);
137
138 return 0;
139}
140EXPORT_SYMBOL_GPL(ablk_init_common);
141
142int ablk_init(struct crypto_tfm *tfm)
143{
144 char drv_name[CRYPTO_MAX_ALG_NAME];
145
146 snprintf(drv_name, sizeof(drv_name), "__driver-%s",
147 crypto_tfm_alg_driver_name(tfm));
148
149 return ablk_init_common(tfm, drv_name);
150}
151EXPORT_SYMBOL_GPL(ablk_init);
152
153MODULE_LICENSE("GPL");
154