162306a36Sopenharmony_ci// SPDX-License-Identifier: GPL-2.0-only
262306a36Sopenharmony_ci/*
362306a36Sopenharmony_ci * Scalar AES core transform
462306a36Sopenharmony_ci *
562306a36Sopenharmony_ci * Copyright (C) 2017 Linaro Ltd.
662306a36Sopenharmony_ci * Author: Ard Biesheuvel <ard.biesheuvel@linaro.org>
762306a36Sopenharmony_ci */
862306a36Sopenharmony_ci
962306a36Sopenharmony_ci#include <crypto/aes.h>
1062306a36Sopenharmony_ci#include <crypto/algapi.h>
1162306a36Sopenharmony_ci#include <linux/module.h>
1262306a36Sopenharmony_ci
1362306a36Sopenharmony_ciasmlinkage void __aes_arm_encrypt(u32 *rk, int rounds, const u8 *in, u8 *out);
1462306a36Sopenharmony_ciasmlinkage void __aes_arm_decrypt(u32 *rk, int rounds, const u8 *in, u8 *out);
1562306a36Sopenharmony_ci
1662306a36Sopenharmony_cistatic void aes_arm_encrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
1762306a36Sopenharmony_ci{
1862306a36Sopenharmony_ci	struct crypto_aes_ctx *ctx = crypto_tfm_ctx(tfm);
1962306a36Sopenharmony_ci	int rounds = 6 + ctx->key_length / 4;
2062306a36Sopenharmony_ci
2162306a36Sopenharmony_ci	__aes_arm_encrypt(ctx->key_enc, rounds, in, out);
2262306a36Sopenharmony_ci}
2362306a36Sopenharmony_ci
2462306a36Sopenharmony_cistatic void aes_arm_decrypt(struct crypto_tfm *tfm, u8 *out, const u8 *in)
2562306a36Sopenharmony_ci{
2662306a36Sopenharmony_ci	struct crypto_aes_ctx *ctx = crypto_tfm_ctx(tfm);
2762306a36Sopenharmony_ci	int rounds = 6 + ctx->key_length / 4;
2862306a36Sopenharmony_ci
2962306a36Sopenharmony_ci	__aes_arm_decrypt(ctx->key_dec, rounds, in, out);
3062306a36Sopenharmony_ci}
3162306a36Sopenharmony_ci
3262306a36Sopenharmony_cistatic struct crypto_alg aes_alg = {
3362306a36Sopenharmony_ci	.cra_name			= "aes",
3462306a36Sopenharmony_ci	.cra_driver_name		= "aes-arm",
3562306a36Sopenharmony_ci	.cra_priority			= 200,
3662306a36Sopenharmony_ci	.cra_flags			= CRYPTO_ALG_TYPE_CIPHER,
3762306a36Sopenharmony_ci	.cra_blocksize			= AES_BLOCK_SIZE,
3862306a36Sopenharmony_ci	.cra_ctxsize			= sizeof(struct crypto_aes_ctx),
3962306a36Sopenharmony_ci	.cra_module			= THIS_MODULE,
4062306a36Sopenharmony_ci
4162306a36Sopenharmony_ci	.cra_cipher.cia_min_keysize	= AES_MIN_KEY_SIZE,
4262306a36Sopenharmony_ci	.cra_cipher.cia_max_keysize	= AES_MAX_KEY_SIZE,
4362306a36Sopenharmony_ci	.cra_cipher.cia_setkey		= crypto_aes_set_key,
4462306a36Sopenharmony_ci	.cra_cipher.cia_encrypt		= aes_arm_encrypt,
4562306a36Sopenharmony_ci	.cra_cipher.cia_decrypt		= aes_arm_decrypt,
4662306a36Sopenharmony_ci
4762306a36Sopenharmony_ci#ifndef CONFIG_HAVE_EFFICIENT_UNALIGNED_ACCESS
4862306a36Sopenharmony_ci	.cra_alignmask			= 3,
4962306a36Sopenharmony_ci#endif
5062306a36Sopenharmony_ci};
5162306a36Sopenharmony_ci
5262306a36Sopenharmony_cistatic int __init aes_init(void)
5362306a36Sopenharmony_ci{
5462306a36Sopenharmony_ci	return crypto_register_alg(&aes_alg);
5562306a36Sopenharmony_ci}
5662306a36Sopenharmony_ci
5762306a36Sopenharmony_cistatic void __exit aes_fini(void)
5862306a36Sopenharmony_ci{
5962306a36Sopenharmony_ci	crypto_unregister_alg(&aes_alg);
6062306a36Sopenharmony_ci}
6162306a36Sopenharmony_ci
6262306a36Sopenharmony_cimodule_init(aes_init);
6362306a36Sopenharmony_cimodule_exit(aes_fini);
6462306a36Sopenharmony_ci
6562306a36Sopenharmony_ciMODULE_DESCRIPTION("Scalar AES cipher for ARM");
6662306a36Sopenharmony_ciMODULE_AUTHOR("Ard Biesheuvel <ard.biesheuvel@linaro.org>");
6762306a36Sopenharmony_ciMODULE_LICENSE("GPL v2");
6862306a36Sopenharmony_ciMODULE_ALIAS_CRYPTO("aes");
69