1 | // SPDX-License-Identifier: GPL-2.0-or-later |
2 | /* |
3 | * Shared crypto simd helpers |
4 | * |
5 | * Copyright (c) 2012 Jussi Kivilinna <jussi.kivilinna@mbnet.fi> |
6 | * Copyright (c) 2016 Herbert Xu <herbert@gondor.apana.org.au> |
7 | * Copyright (c) 2019 Google LLC |
8 | * |
9 | * Based on aesni-intel_glue.c by: |
10 | * Copyright (C) 2008, Intel Corp. |
11 | * Author: Huang Ying <ying.huang@intel.com> |
12 | */ |
13 | |
14 | /* |
15 | * Shared crypto SIMD helpers. These functions dynamically create and register |
16 | * an skcipher or AEAD algorithm that wraps another, internal algorithm. The |
17 | * wrapper ensures that the internal algorithm is only executed in a context |
18 | * where SIMD instructions are usable, i.e. where may_use_simd() returns true. |
19 | * If SIMD is already usable, the wrapper directly calls the internal algorithm. |
20 | * Otherwise it defers execution to a workqueue via cryptd. |
21 | * |
22 | * This is an alternative to the internal algorithm implementing a fallback for |
23 | * the !may_use_simd() case itself. |
24 | * |
25 | * Note that the wrapper algorithm is asynchronous, i.e. it has the |
26 | * CRYPTO_ALG_ASYNC flag set. Therefore it won't be found by users who |
27 | * explicitly allocate a synchronous algorithm. |
28 | */ |
29 | |
30 | #include <crypto/cryptd.h> |
31 | #include <crypto/internal/aead.h> |
32 | #include <crypto/internal/simd.h> |
33 | #include <crypto/internal/skcipher.h> |
34 | #include <linux/kernel.h> |
35 | #include <linux/module.h> |
36 | #include <linux/preempt.h> |
37 | #include <asm/simd.h> |
38 | |
39 | /* skcipher support */ |
40 | |
41 | struct simd_skcipher_alg { |
42 | const char *ialg_name; |
43 | struct skcipher_alg alg; |
44 | }; |
45 | |
46 | struct simd_skcipher_ctx { |
47 | struct cryptd_skcipher *cryptd_tfm; |
48 | }; |
49 | |
50 | static int simd_skcipher_setkey(struct crypto_skcipher *tfm, const u8 *key, |
51 | unsigned int key_len) |
52 | { |
53 | struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm); |
54 | struct crypto_skcipher *child = &ctx->cryptd_tfm->base; |
55 | |
56 | crypto_skcipher_clear_flags(tfm: child, CRYPTO_TFM_REQ_MASK); |
57 | crypto_skcipher_set_flags(tfm: child, flags: crypto_skcipher_get_flags(tfm) & |
58 | CRYPTO_TFM_REQ_MASK); |
59 | return crypto_skcipher_setkey(tfm: child, key, keylen: key_len); |
60 | } |
61 | |
62 | static int simd_skcipher_encrypt(struct skcipher_request *req) |
63 | { |
64 | struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req); |
65 | struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm); |
66 | struct skcipher_request *subreq; |
67 | struct crypto_skcipher *child; |
68 | |
69 | subreq = skcipher_request_ctx(req); |
70 | *subreq = *req; |
71 | |
72 | if (!crypto_simd_usable() || |
73 | (in_atomic() && cryptd_skcipher_queued(tfm: ctx->cryptd_tfm))) |
74 | child = &ctx->cryptd_tfm->base; |
75 | else |
76 | child = cryptd_skcipher_child(tfm: ctx->cryptd_tfm); |
77 | |
78 | skcipher_request_set_tfm(req: subreq, tfm: child); |
79 | |
80 | return crypto_skcipher_encrypt(req: subreq); |
81 | } |
82 | |
83 | static int simd_skcipher_decrypt(struct skcipher_request *req) |
84 | { |
85 | struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req); |
86 | struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm); |
87 | struct skcipher_request *subreq; |
88 | struct crypto_skcipher *child; |
89 | |
90 | subreq = skcipher_request_ctx(req); |
91 | *subreq = *req; |
92 | |
93 | if (!crypto_simd_usable() || |
94 | (in_atomic() && cryptd_skcipher_queued(tfm: ctx->cryptd_tfm))) |
95 | child = &ctx->cryptd_tfm->base; |
96 | else |
97 | child = cryptd_skcipher_child(tfm: ctx->cryptd_tfm); |
98 | |
99 | skcipher_request_set_tfm(req: subreq, tfm: child); |
100 | |
101 | return crypto_skcipher_decrypt(req: subreq); |
102 | } |
103 | |
104 | static void simd_skcipher_exit(struct crypto_skcipher *tfm) |
105 | { |
106 | struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm); |
107 | |
108 | cryptd_free_skcipher(tfm: ctx->cryptd_tfm); |
109 | } |
110 | |
111 | static int simd_skcipher_init(struct crypto_skcipher *tfm) |
112 | { |
113 | struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm); |
114 | struct cryptd_skcipher *cryptd_tfm; |
115 | struct simd_skcipher_alg *salg; |
116 | struct skcipher_alg *alg; |
117 | unsigned reqsize; |
118 | |
119 | alg = crypto_skcipher_alg(tfm); |
120 | salg = container_of(alg, struct simd_skcipher_alg, alg); |
121 | |
122 | cryptd_tfm = cryptd_alloc_skcipher(alg_name: salg->ialg_name, |
123 | CRYPTO_ALG_INTERNAL, |
124 | CRYPTO_ALG_INTERNAL); |
125 | if (IS_ERR(ptr: cryptd_tfm)) |
126 | return PTR_ERR(ptr: cryptd_tfm); |
127 | |
128 | ctx->cryptd_tfm = cryptd_tfm; |
129 | |
130 | reqsize = crypto_skcipher_reqsize(tfm: cryptd_skcipher_child(tfm: cryptd_tfm)); |
131 | reqsize = max(reqsize, crypto_skcipher_reqsize(&cryptd_tfm->base)); |
132 | reqsize += sizeof(struct skcipher_request); |
133 | |
134 | crypto_skcipher_set_reqsize(skcipher: tfm, reqsize); |
135 | |
136 | return 0; |
137 | } |
138 | |
139 | struct simd_skcipher_alg *simd_skcipher_create_compat(const char *algname, |
140 | const char *drvname, |
141 | const char *basename) |
142 | { |
143 | struct simd_skcipher_alg *salg; |
144 | struct crypto_skcipher *tfm; |
145 | struct skcipher_alg *ialg; |
146 | struct skcipher_alg *alg; |
147 | int err; |
148 | |
149 | tfm = crypto_alloc_skcipher(alg_name: basename, CRYPTO_ALG_INTERNAL, |
150 | CRYPTO_ALG_INTERNAL | CRYPTO_ALG_ASYNC); |
151 | if (IS_ERR(ptr: tfm)) |
152 | return ERR_CAST(ptr: tfm); |
153 | |
154 | ialg = crypto_skcipher_alg(tfm); |
155 | |
156 | salg = kzalloc(size: sizeof(*salg), GFP_KERNEL); |
157 | if (!salg) { |
158 | salg = ERR_PTR(error: -ENOMEM); |
159 | goto out_put_tfm; |
160 | } |
161 | |
162 | salg->ialg_name = basename; |
163 | alg = &salg->alg; |
164 | |
165 | err = -ENAMETOOLONG; |
166 | if (snprintf(buf: alg->base.cra_name, CRYPTO_MAX_ALG_NAME, fmt: "%s" , algname) >= |
167 | CRYPTO_MAX_ALG_NAME) |
168 | goto out_free_salg; |
169 | |
170 | if (snprintf(buf: alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, fmt: "%s" , |
171 | drvname) >= CRYPTO_MAX_ALG_NAME) |
172 | goto out_free_salg; |
173 | |
174 | alg->base.cra_flags = CRYPTO_ALG_ASYNC | |
175 | (ialg->base.cra_flags & CRYPTO_ALG_INHERITED_FLAGS); |
176 | alg->base.cra_priority = ialg->base.cra_priority; |
177 | alg->base.cra_blocksize = ialg->base.cra_blocksize; |
178 | alg->base.cra_alignmask = ialg->base.cra_alignmask; |
179 | alg->base.cra_module = ialg->base.cra_module; |
180 | alg->base.cra_ctxsize = sizeof(struct simd_skcipher_ctx); |
181 | |
182 | alg->ivsize = ialg->ivsize; |
183 | alg->chunksize = ialg->chunksize; |
184 | alg->min_keysize = ialg->min_keysize; |
185 | alg->max_keysize = ialg->max_keysize; |
186 | |
187 | alg->init = simd_skcipher_init; |
188 | alg->exit = simd_skcipher_exit; |
189 | |
190 | alg->setkey = simd_skcipher_setkey; |
191 | alg->encrypt = simd_skcipher_encrypt; |
192 | alg->decrypt = simd_skcipher_decrypt; |
193 | |
194 | err = crypto_register_skcipher(alg); |
195 | if (err) |
196 | goto out_free_salg; |
197 | |
198 | out_put_tfm: |
199 | crypto_free_skcipher(tfm); |
200 | return salg; |
201 | |
202 | out_free_salg: |
203 | kfree(objp: salg); |
204 | salg = ERR_PTR(error: err); |
205 | goto out_put_tfm; |
206 | } |
207 | EXPORT_SYMBOL_GPL(simd_skcipher_create_compat); |
208 | |
209 | struct simd_skcipher_alg *simd_skcipher_create(const char *algname, |
210 | const char *basename) |
211 | { |
212 | char drvname[CRYPTO_MAX_ALG_NAME]; |
213 | |
214 | if (snprintf(buf: drvname, CRYPTO_MAX_ALG_NAME, fmt: "simd-%s" , basename) >= |
215 | CRYPTO_MAX_ALG_NAME) |
216 | return ERR_PTR(error: -ENAMETOOLONG); |
217 | |
218 | return simd_skcipher_create_compat(algname, drvname, basename); |
219 | } |
220 | EXPORT_SYMBOL_GPL(simd_skcipher_create); |
221 | |
222 | void simd_skcipher_free(struct simd_skcipher_alg *salg) |
223 | { |
224 | crypto_unregister_skcipher(alg: &salg->alg); |
225 | kfree(objp: salg); |
226 | } |
227 | EXPORT_SYMBOL_GPL(simd_skcipher_free); |
228 | |
229 | int simd_register_skciphers_compat(struct skcipher_alg *algs, int count, |
230 | struct simd_skcipher_alg **simd_algs) |
231 | { |
232 | int err; |
233 | int i; |
234 | const char *algname; |
235 | const char *drvname; |
236 | const char *basename; |
237 | struct simd_skcipher_alg *simd; |
238 | |
239 | err = crypto_register_skciphers(algs, count); |
240 | if (err) |
241 | return err; |
242 | |
243 | for (i = 0; i < count; i++) { |
244 | WARN_ON(strncmp(algs[i].base.cra_name, "__" , 2)); |
245 | WARN_ON(strncmp(algs[i].base.cra_driver_name, "__" , 2)); |
246 | algname = algs[i].base.cra_name + 2; |
247 | drvname = algs[i].base.cra_driver_name + 2; |
248 | basename = algs[i].base.cra_driver_name; |
249 | simd = simd_skcipher_create_compat(algname, drvname, basename); |
250 | err = PTR_ERR(ptr: simd); |
251 | if (IS_ERR(ptr: simd)) |
252 | goto err_unregister; |
253 | simd_algs[i] = simd; |
254 | } |
255 | return 0; |
256 | |
257 | err_unregister: |
258 | simd_unregister_skciphers(algs, count, simd_algs); |
259 | return err; |
260 | } |
261 | EXPORT_SYMBOL_GPL(simd_register_skciphers_compat); |
262 | |
263 | void simd_unregister_skciphers(struct skcipher_alg *algs, int count, |
264 | struct simd_skcipher_alg **simd_algs) |
265 | { |
266 | int i; |
267 | |
268 | crypto_unregister_skciphers(algs, count); |
269 | |
270 | for (i = 0; i < count; i++) { |
271 | if (simd_algs[i]) { |
272 | simd_skcipher_free(simd_algs[i]); |
273 | simd_algs[i] = NULL; |
274 | } |
275 | } |
276 | } |
277 | EXPORT_SYMBOL_GPL(simd_unregister_skciphers); |
278 | |
279 | /* AEAD support */ |
280 | |
281 | struct simd_aead_alg { |
282 | const char *ialg_name; |
283 | struct aead_alg alg; |
284 | }; |
285 | |
286 | struct simd_aead_ctx { |
287 | struct cryptd_aead *cryptd_tfm; |
288 | }; |
289 | |
290 | static int simd_aead_setkey(struct crypto_aead *tfm, const u8 *key, |
291 | unsigned int key_len) |
292 | { |
293 | struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm); |
294 | struct crypto_aead *child = &ctx->cryptd_tfm->base; |
295 | |
296 | crypto_aead_clear_flags(tfm: child, CRYPTO_TFM_REQ_MASK); |
297 | crypto_aead_set_flags(tfm: child, flags: crypto_aead_get_flags(tfm) & |
298 | CRYPTO_TFM_REQ_MASK); |
299 | return crypto_aead_setkey(tfm: child, key, keylen: key_len); |
300 | } |
301 | |
302 | static int simd_aead_setauthsize(struct crypto_aead *tfm, unsigned int authsize) |
303 | { |
304 | struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm); |
305 | struct crypto_aead *child = &ctx->cryptd_tfm->base; |
306 | |
307 | return crypto_aead_setauthsize(tfm: child, authsize); |
308 | } |
309 | |
310 | static int simd_aead_encrypt(struct aead_request *req) |
311 | { |
312 | struct crypto_aead *tfm = crypto_aead_reqtfm(req); |
313 | struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm); |
314 | struct aead_request *subreq; |
315 | struct crypto_aead *child; |
316 | |
317 | subreq = aead_request_ctx(req); |
318 | *subreq = *req; |
319 | |
320 | if (!crypto_simd_usable() || |
321 | (in_atomic() && cryptd_aead_queued(tfm: ctx->cryptd_tfm))) |
322 | child = &ctx->cryptd_tfm->base; |
323 | else |
324 | child = cryptd_aead_child(tfm: ctx->cryptd_tfm); |
325 | |
326 | aead_request_set_tfm(req: subreq, tfm: child); |
327 | |
328 | return crypto_aead_encrypt(req: subreq); |
329 | } |
330 | |
331 | static int simd_aead_decrypt(struct aead_request *req) |
332 | { |
333 | struct crypto_aead *tfm = crypto_aead_reqtfm(req); |
334 | struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm); |
335 | struct aead_request *subreq; |
336 | struct crypto_aead *child; |
337 | |
338 | subreq = aead_request_ctx(req); |
339 | *subreq = *req; |
340 | |
341 | if (!crypto_simd_usable() || |
342 | (in_atomic() && cryptd_aead_queued(tfm: ctx->cryptd_tfm))) |
343 | child = &ctx->cryptd_tfm->base; |
344 | else |
345 | child = cryptd_aead_child(tfm: ctx->cryptd_tfm); |
346 | |
347 | aead_request_set_tfm(req: subreq, tfm: child); |
348 | |
349 | return crypto_aead_decrypt(req: subreq); |
350 | } |
351 | |
352 | static void simd_aead_exit(struct crypto_aead *tfm) |
353 | { |
354 | struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm); |
355 | |
356 | cryptd_free_aead(tfm: ctx->cryptd_tfm); |
357 | } |
358 | |
359 | static int simd_aead_init(struct crypto_aead *tfm) |
360 | { |
361 | struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm); |
362 | struct cryptd_aead *cryptd_tfm; |
363 | struct simd_aead_alg *salg; |
364 | struct aead_alg *alg; |
365 | unsigned reqsize; |
366 | |
367 | alg = crypto_aead_alg(tfm); |
368 | salg = container_of(alg, struct simd_aead_alg, alg); |
369 | |
370 | cryptd_tfm = cryptd_alloc_aead(alg_name: salg->ialg_name, CRYPTO_ALG_INTERNAL, |
371 | CRYPTO_ALG_INTERNAL); |
372 | if (IS_ERR(ptr: cryptd_tfm)) |
373 | return PTR_ERR(ptr: cryptd_tfm); |
374 | |
375 | ctx->cryptd_tfm = cryptd_tfm; |
376 | |
377 | reqsize = crypto_aead_reqsize(tfm: cryptd_aead_child(tfm: cryptd_tfm)); |
378 | reqsize = max(reqsize, crypto_aead_reqsize(&cryptd_tfm->base)); |
379 | reqsize += sizeof(struct aead_request); |
380 | |
381 | crypto_aead_set_reqsize(aead: tfm, reqsize); |
382 | |
383 | return 0; |
384 | } |
385 | |
386 | struct simd_aead_alg *simd_aead_create_compat(const char *algname, |
387 | const char *drvname, |
388 | const char *basename) |
389 | { |
390 | struct simd_aead_alg *salg; |
391 | struct crypto_aead *tfm; |
392 | struct aead_alg *ialg; |
393 | struct aead_alg *alg; |
394 | int err; |
395 | |
396 | tfm = crypto_alloc_aead(alg_name: basename, CRYPTO_ALG_INTERNAL, |
397 | CRYPTO_ALG_INTERNAL | CRYPTO_ALG_ASYNC); |
398 | if (IS_ERR(ptr: tfm)) |
399 | return ERR_CAST(ptr: tfm); |
400 | |
401 | ialg = crypto_aead_alg(tfm); |
402 | |
403 | salg = kzalloc(size: sizeof(*salg), GFP_KERNEL); |
404 | if (!salg) { |
405 | salg = ERR_PTR(error: -ENOMEM); |
406 | goto out_put_tfm; |
407 | } |
408 | |
409 | salg->ialg_name = basename; |
410 | alg = &salg->alg; |
411 | |
412 | err = -ENAMETOOLONG; |
413 | if (snprintf(buf: alg->base.cra_name, CRYPTO_MAX_ALG_NAME, fmt: "%s" , algname) >= |
414 | CRYPTO_MAX_ALG_NAME) |
415 | goto out_free_salg; |
416 | |
417 | if (snprintf(buf: alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, fmt: "%s" , |
418 | drvname) >= CRYPTO_MAX_ALG_NAME) |
419 | goto out_free_salg; |
420 | |
421 | alg->base.cra_flags = CRYPTO_ALG_ASYNC | |
422 | (ialg->base.cra_flags & CRYPTO_ALG_INHERITED_FLAGS); |
423 | alg->base.cra_priority = ialg->base.cra_priority; |
424 | alg->base.cra_blocksize = ialg->base.cra_blocksize; |
425 | alg->base.cra_alignmask = ialg->base.cra_alignmask; |
426 | alg->base.cra_module = ialg->base.cra_module; |
427 | alg->base.cra_ctxsize = sizeof(struct simd_aead_ctx); |
428 | |
429 | alg->ivsize = ialg->ivsize; |
430 | alg->maxauthsize = ialg->maxauthsize; |
431 | alg->chunksize = ialg->chunksize; |
432 | |
433 | alg->init = simd_aead_init; |
434 | alg->exit = simd_aead_exit; |
435 | |
436 | alg->setkey = simd_aead_setkey; |
437 | alg->setauthsize = simd_aead_setauthsize; |
438 | alg->encrypt = simd_aead_encrypt; |
439 | alg->decrypt = simd_aead_decrypt; |
440 | |
441 | err = crypto_register_aead(alg); |
442 | if (err) |
443 | goto out_free_salg; |
444 | |
445 | out_put_tfm: |
446 | crypto_free_aead(tfm); |
447 | return salg; |
448 | |
449 | out_free_salg: |
450 | kfree(objp: salg); |
451 | salg = ERR_PTR(error: err); |
452 | goto out_put_tfm; |
453 | } |
454 | EXPORT_SYMBOL_GPL(simd_aead_create_compat); |
455 | |
456 | struct simd_aead_alg *simd_aead_create(const char *algname, |
457 | const char *basename) |
458 | { |
459 | char drvname[CRYPTO_MAX_ALG_NAME]; |
460 | |
461 | if (snprintf(buf: drvname, CRYPTO_MAX_ALG_NAME, fmt: "simd-%s" , basename) >= |
462 | CRYPTO_MAX_ALG_NAME) |
463 | return ERR_PTR(error: -ENAMETOOLONG); |
464 | |
465 | return simd_aead_create_compat(algname, drvname, basename); |
466 | } |
467 | EXPORT_SYMBOL_GPL(simd_aead_create); |
468 | |
469 | void simd_aead_free(struct simd_aead_alg *salg) |
470 | { |
471 | crypto_unregister_aead(alg: &salg->alg); |
472 | kfree(objp: salg); |
473 | } |
474 | EXPORT_SYMBOL_GPL(simd_aead_free); |
475 | |
476 | int simd_register_aeads_compat(struct aead_alg *algs, int count, |
477 | struct simd_aead_alg **simd_algs) |
478 | { |
479 | int err; |
480 | int i; |
481 | const char *algname; |
482 | const char *drvname; |
483 | const char *basename; |
484 | struct simd_aead_alg *simd; |
485 | |
486 | err = crypto_register_aeads(algs, count); |
487 | if (err) |
488 | return err; |
489 | |
490 | for (i = 0; i < count; i++) { |
491 | WARN_ON(strncmp(algs[i].base.cra_name, "__" , 2)); |
492 | WARN_ON(strncmp(algs[i].base.cra_driver_name, "__" , 2)); |
493 | algname = algs[i].base.cra_name + 2; |
494 | drvname = algs[i].base.cra_driver_name + 2; |
495 | basename = algs[i].base.cra_driver_name; |
496 | simd = simd_aead_create_compat(algname, drvname, basename); |
497 | err = PTR_ERR(ptr: simd); |
498 | if (IS_ERR(ptr: simd)) |
499 | goto err_unregister; |
500 | simd_algs[i] = simd; |
501 | } |
502 | return 0; |
503 | |
504 | err_unregister: |
505 | simd_unregister_aeads(algs, count, simd_algs); |
506 | return err; |
507 | } |
508 | EXPORT_SYMBOL_GPL(simd_register_aeads_compat); |
509 | |
510 | void simd_unregister_aeads(struct aead_alg *algs, int count, |
511 | struct simd_aead_alg **simd_algs) |
512 | { |
513 | int i; |
514 | |
515 | crypto_unregister_aeads(algs, count); |
516 | |
517 | for (i = 0; i < count; i++) { |
518 | if (simd_algs[i]) { |
519 | simd_aead_free(simd_algs[i]); |
520 | simd_algs[i] = NULL; |
521 | } |
522 | } |
523 | } |
524 | EXPORT_SYMBOL_GPL(simd_unregister_aeads); |
525 | |
526 | MODULE_LICENSE("GPL" ); |
527 | |