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 <crypto/algapi.h>
10#include <linux/module.h>
11
12asmlinkage void __aes_arm64_encrypt(u32 *rk, u8 *out, const u8 *in, int rounds);
13asmlinkage void __aes_arm64_decrypt(u32 *rk, u8 *out, const u8 *in, int rounds);
14
15static 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
23static 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
31static 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
47static int __init aes_init(void)
48{
49	return crypto_register_alg(&aes_alg);
50}
51
52static void __exit aes_fini(void)
53{
54	crypto_unregister_alg(&aes_alg);
55}
56
57module_init(aes_init);
58module_exit(aes_fini);
59
60MODULE_DESCRIPTION("Scalar AES cipher for arm64");
61MODULE_AUTHOR("Ard Biesheuvel <ard.biesheuvel@linaro.org>");
62MODULE_LICENSE("GPL v2");
63MODULE_ALIAS_CRYPTO("aes");
64