hc
2024-12-19 9370bb92b2d16684ee45cf24e879c93c509162da
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/* SPDX-License-Identifier: GPL-2.0-only */
/*
 * Copyright (C) 2013 - 2017 Linaro Ltd <ard.biesheuvel@linaro.org>
 */
 
#include <linux/linkage.h>
#include <asm/assembler.h>
 
   .arch        armv8-a+crypto
 
SYM_FUNC_START(__aes_ce_encrypt)
   sub        w3, w3, #2
   ld1        {v0.16b}, [x2]
   ld1        {v1.4s}, [x0], #16
   cmp        w3, #10
   bmi        0f
   bne        3f
   mov        v3.16b, v1.16b
   b        2f
0:    mov        v2.16b, v1.16b
   ld1        {v3.4s}, [x0], #16
1:    aese        v0.16b, v2.16b
   aesmc        v0.16b, v0.16b
2:    ld1        {v1.4s}, [x0], #16
   aese        v0.16b, v3.16b
   aesmc        v0.16b, v0.16b
3:    ld1        {v2.4s}, [x0], #16
   subs        w3, w3, #3
   aese        v0.16b, v1.16b
   aesmc        v0.16b, v0.16b
   ld1        {v3.4s}, [x0], #16
   bpl        1b
   aese        v0.16b, v2.16b
   eor        v0.16b, v0.16b, v3.16b
   st1        {v0.16b}, [x1]
   ret
SYM_FUNC_END(__aes_ce_encrypt)
 
SYM_FUNC_START(__aes_ce_decrypt)
   sub        w3, w3, #2
   ld1        {v0.16b}, [x2]
   ld1        {v1.4s}, [x0], #16
   cmp        w3, #10
   bmi        0f
   bne        3f
   mov        v3.16b, v1.16b
   b        2f
0:    mov        v2.16b, v1.16b
   ld1        {v3.4s}, [x0], #16
1:    aesd        v0.16b, v2.16b
   aesimc        v0.16b, v0.16b
2:    ld1        {v1.4s}, [x0], #16
   aesd        v0.16b, v3.16b
   aesimc        v0.16b, v0.16b
3:    ld1        {v2.4s}, [x0], #16
   subs        w3, w3, #3
   aesd        v0.16b, v1.16b
   aesimc        v0.16b, v0.16b
   ld1        {v3.4s}, [x0], #16
   bpl        1b
   aesd        v0.16b, v2.16b
   eor        v0.16b, v0.16b, v3.16b
   st1        {v0.16b}, [x1]
   ret
SYM_FUNC_END(__aes_ce_decrypt)
 
/*
 * __aes_ce_sub() - use the aese instruction to perform the AES sbox
 *                  substitution on each byte in 'input'
 */
SYM_FUNC_START(__aes_ce_sub)
   dup        v1.4s, w0
   movi        v0.16b, #0
   aese        v0.16b, v1.16b
   umov        w0, v0.s[0]
   ret
SYM_FUNC_END(__aes_ce_sub)
 
SYM_FUNC_START(__aes_ce_invert)
   ld1        {v0.4s}, [x1]
   aesimc        v1.16b, v0.16b
   st1        {v1.4s}, [x0]
   ret
SYM_FUNC_END(__aes_ce_invert)