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 asmlinkage void __aes_arm64_decrypt(u32 *rk, u8 *out, const u8 *in, int rounds);
14
aes_arm64_encrypt(struct crypto_tfm * tfm,u8 * out,const u8 * in)15 static void aes_arm64_encrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
16 {
17 struct crypto_aes_ctx *ctx = crypto_tfm_ctx(tfm);
18 int rounds = 6 + ctx->key_length / 4;
19
20 __aes_arm64_encrypt(ctx->key_enc, out, in, rounds);
21 }
22
aes_arm64_decrypt(struct crypto_tfm * tfm,u8 * out,const u8 * in)23 static void aes_arm64_decrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
24 {
25 struct crypto_aes_ctx *ctx = crypto_tfm_ctx(tfm);
26 int rounds = 6 + ctx->key_length / 4;
27
28 __aes_arm64_decrypt(ctx->key_dec, out, in, rounds);
29 }
30
31 static struct crypto_alg aes_alg = {
32 .cra_name = "aes",
33 .cra_driver_name = "aes-arm64",
34 .cra_priority = 200,
35 .cra_flags = CRYPTO_ALG_TYPE_CIPHER,
36 .cra_blocksize = AES_BLOCK_SIZE,
37 .cra_ctxsize = sizeof(struct crypto_aes_ctx),
38 .cra_module = THIS_MODULE,
39
40 .cra_cipher.cia_min_keysize = AES_MIN_KEY_SIZE,
41 .cra_cipher.cia_max_keysize = AES_MAX_KEY_SIZE,
42 .cra_cipher.cia_setkey = crypto_aes_set_key,
43 .cra_cipher.cia_encrypt = aes_arm64_encrypt,
44 .cra_cipher.cia_decrypt = aes_arm64_decrypt
45 };
46
aes_init(void)47 static int __init aes_init(void)
48 {
49 return crypto_register_alg(&aes_alg);
50 }
51
aes_fini(void)52 static void __exit aes_fini(void)
53 {
54 crypto_unregister_alg(&aes_alg);
55 }
56
57 module_init(aes_init);
58 module_exit(aes_fini);
59
60 MODULE_DESCRIPTION("Scalar AES cipher for arm64");
61 MODULE_AUTHOR("Ard Biesheuvel <ard.biesheuvel@linaro.org>");
62 MODULE_LICENSE("GPL v2");
63 MODULE_ALIAS_CRYPTO("aes");
64