1 // SPDX-License-Identifier: GPL-2.0-only
2 /*
3  * Scalar AES core transform
4  *
5  * Copyright (C) 2017 Linaro Ltd <ard.biesheuvel@linaro.org>
6  */
7 
8 #include <crypto/aes.h>
9 #include <linux/crypto.h>
10 #include <linux/module.h>
11 
12 asmlinkage void __aes_arm64_encrypt(u32 *rk, u8 *out, const u8 *in, int rounds);
13 EXPORT_SYMBOL(__aes_arm64_encrypt);
14 
15 asmlinkage void __aes_arm64_decrypt(u32 *rk, u8 *out, const u8 *in, int rounds);
16 EXPORT_SYMBOL(__aes_arm64_decrypt);
17 
18 static void aes_encrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
19 {
20 	struct crypto_aes_ctx *ctx = crypto_tfm_ctx(tfm);
21 	int rounds = 6 + ctx->key_length / 4;
22 
23 	__aes_arm64_encrypt(ctx->key_enc, out, in, rounds);
24 }
25 
26 static void aes_decrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
27 {
28 	struct crypto_aes_ctx *ctx = crypto_tfm_ctx(tfm);
29 	int rounds = 6 + ctx->key_length / 4;
30 
31 	__aes_arm64_decrypt(ctx->key_dec, out, in, rounds);
32 }
33 
34 static struct crypto_alg aes_alg = {
35 	.cra_name			= "aes",
36 	.cra_driver_name		= "aes-arm64",
37 	.cra_priority			= 200,
38 	.cra_flags			= CRYPTO_ALG_TYPE_CIPHER,
39 	.cra_blocksize			= AES_BLOCK_SIZE,
40 	.cra_ctxsize			= sizeof(struct crypto_aes_ctx),
41 	.cra_module			= THIS_MODULE,
42 
43 	.cra_cipher.cia_min_keysize	= AES_MIN_KEY_SIZE,
44 	.cra_cipher.cia_max_keysize	= AES_MAX_KEY_SIZE,
45 	.cra_cipher.cia_setkey		= crypto_aes_set_key,
46 	.cra_cipher.cia_encrypt		= aes_encrypt,
47 	.cra_cipher.cia_decrypt		= aes_decrypt
48 };
49 
50 static int __init aes_init(void)
51 {
52 	return crypto_register_alg(&aes_alg);
53 }
54 
55 static void __exit aes_fini(void)
56 {
57 	crypto_unregister_alg(&aes_alg);
58 }
59 
60 module_init(aes_init);
61 module_exit(aes_fini);
62 
63 MODULE_DESCRIPTION("Scalar AES cipher for arm64");
64 MODULE_AUTHOR("Ard Biesheuvel <ard.biesheuvel@linaro.org>");
65 MODULE_LICENSE("GPL v2");
66 MODULE_ALIAS_CRYPTO("aes");
67