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