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