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(child, CRYPTO_TFM_REQ_MASK); 57 crypto_skcipher_set_flags(child, crypto_skcipher_get_flags(tfm) & 58 CRYPTO_TFM_REQ_MASK); 59 return crypto_skcipher_setkey(child, key, 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(ctx->cryptd_tfm))) 74 child = &ctx->cryptd_tfm->base; 75 else 76 child = cryptd_skcipher_child(ctx->cryptd_tfm); 77 78 skcipher_request_set_tfm(subreq, child); 79 80 return crypto_skcipher_encrypt(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(ctx->cryptd_tfm))) 95 child = &ctx->cryptd_tfm->base; 96 else 97 child = cryptd_skcipher_child(ctx->cryptd_tfm); 98 99 skcipher_request_set_tfm(subreq, child); 100 101 return crypto_skcipher_decrypt(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(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(salg->ialg_name, 123 CRYPTO_ALG_INTERNAL, 124 CRYPTO_ALG_INTERNAL); 125 if (IS_ERR(cryptd_tfm)) 126 return PTR_ERR(cryptd_tfm); 127 128 ctx->cryptd_tfm = cryptd_tfm; 129 130 reqsize = crypto_skcipher_reqsize(cryptd_skcipher_child(cryptd_tfm)); 131 reqsize = max(reqsize, crypto_skcipher_reqsize(&cryptd_tfm->base)); 132 reqsize += sizeof(struct skcipher_request); 133 134 crypto_skcipher_set_reqsize(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(basename, CRYPTO_ALG_INTERNAL, 150 CRYPTO_ALG_INTERNAL | CRYPTO_ALG_ASYNC); 151 if (IS_ERR(tfm)) 152 return ERR_CAST(tfm); 153 154 ialg = crypto_skcipher_alg(tfm); 155 156 salg = kzalloc(sizeof(*salg), GFP_KERNEL); 157 if (!salg) { 158 salg = ERR_PTR(-ENOMEM); 159 goto out_put_tfm; 160 } 161 162 salg->ialg_name = basename; 163 alg = &salg->alg; 164 165 err = -ENAMETOOLONG; 166 if (snprintf(alg->base.cra_name, CRYPTO_MAX_ALG_NAME, "%s", algname) >= 167 CRYPTO_MAX_ALG_NAME) 168 goto out_free_salg; 169 170 if (snprintf(alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, "%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(salg); 204 salg = ERR_PTR(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(drvname, CRYPTO_MAX_ALG_NAME, "simd-%s", basename) >= 215 CRYPTO_MAX_ALG_NAME) 216 return ERR_PTR(-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(&salg->alg); 225 kfree(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(simd); 251 if (IS_ERR(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(child, CRYPTO_TFM_REQ_MASK); 297 crypto_aead_set_flags(child, crypto_aead_get_flags(tfm) & 298 CRYPTO_TFM_REQ_MASK); 299 return crypto_aead_setkey(child, key, 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(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(ctx->cryptd_tfm))) 322 child = &ctx->cryptd_tfm->base; 323 else 324 child = cryptd_aead_child(ctx->cryptd_tfm); 325 326 aead_request_set_tfm(subreq, child); 327 328 return crypto_aead_encrypt(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(ctx->cryptd_tfm))) 343 child = &ctx->cryptd_tfm->base; 344 else 345 child = cryptd_aead_child(ctx->cryptd_tfm); 346 347 aead_request_set_tfm(subreq, child); 348 349 return crypto_aead_decrypt(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(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(salg->ialg_name, CRYPTO_ALG_INTERNAL, 371 CRYPTO_ALG_INTERNAL); 372 if (IS_ERR(cryptd_tfm)) 373 return PTR_ERR(cryptd_tfm); 374 375 ctx->cryptd_tfm = cryptd_tfm; 376 377 reqsize = crypto_aead_reqsize(cryptd_aead_child(cryptd_tfm)); 378 reqsize = max(reqsize, crypto_aead_reqsize(&cryptd_tfm->base)); 379 reqsize += sizeof(struct aead_request); 380 381 crypto_aead_set_reqsize(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(basename, CRYPTO_ALG_INTERNAL, 397 CRYPTO_ALG_INTERNAL | CRYPTO_ALG_ASYNC); 398 if (IS_ERR(tfm)) 399 return ERR_CAST(tfm); 400 401 ialg = crypto_aead_alg(tfm); 402 403 salg = kzalloc(sizeof(*salg), GFP_KERNEL); 404 if (!salg) { 405 salg = ERR_PTR(-ENOMEM); 406 goto out_put_tfm; 407 } 408 409 salg->ialg_name = basename; 410 alg = &salg->alg; 411 412 err = -ENAMETOOLONG; 413 if (snprintf(alg->base.cra_name, CRYPTO_MAX_ALG_NAME, "%s", algname) >= 414 CRYPTO_MAX_ALG_NAME) 415 goto out_free_salg; 416 417 if (snprintf(alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, "%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(salg); 451 salg = ERR_PTR(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(drvname, CRYPTO_MAX_ALG_NAME, "simd-%s", basename) >= 462 CRYPTO_MAX_ALG_NAME) 463 return ERR_PTR(-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(&salg->alg); 472 kfree(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(simd); 498 if (IS_ERR(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_DESCRIPTION("Shared crypto SIMD helpers"); 527 MODULE_LICENSE("GPL"); 528