1/* SPDX-License-Identifier: GPL-2.0-only */
2/*
3 * sha2-ce-core.S - core SHA-224/SHA-256 transform using v8 Crypto Extensions
4 *
5 * Copyright (C) 2014 Linaro Ltd <ard.biesheuvel@linaro.org>
6 */
7
8#include <linux/linkage.h>
9#include <asm/assembler.h>
10
11 .text
12 .arch armv8-a+crypto
13
14 dga .req q20
15 dgav .req v20
16 dgb .req q21
17 dgbv .req v21
18
19 t0 .req v22
20 t1 .req v23
21
22 dg0q .req q24
23 dg0v .req v24
24 dg1q .req q25
25 dg1v .req v25
26 dg2q .req q26
27 dg2v .req v26
28
29 .macro add_only, ev, rc, s0
30 mov dg2v.16b, dg0v.16b
31 .ifeq \ev
32 add t1.4s, v\s0\().4s, \rc\().4s
33 sha256h dg0q, dg1q, t0.4s
34 sha256h2 dg1q, dg2q, t0.4s
35 .else
36 .ifnb \s0
37 add t0.4s, v\s0\().4s, \rc\().4s
38 .endif
39 sha256h dg0q, dg1q, t1.4s
40 sha256h2 dg1q, dg2q, t1.4s
41 .endif
42 .endm
43
44 .macro add_update, ev, rc, s0, s1, s2, s3
45 sha256su0 v\s0\().4s, v\s1\().4s
46 add_only \ev, \rc, \s1
47 sha256su1 v\s0\().4s, v\s2\().4s, v\s3\().4s
48 .endm
49
50 /*
51 * The SHA-256 round constants
52 */
53 .section ".rodata", "a"
54 .align 4
55.Lsha2_rcon:
56 .word 0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5
57 .word 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5
58 .word 0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3
59 .word 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174
60 .word 0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc
61 .word 0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da
62 .word 0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7
63 .word 0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967
64 .word 0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13
65 .word 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85
66 .word 0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3
67 .word 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070
68 .word 0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5
69 .word 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3
70 .word 0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208
71 .word 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2
72
73 /*
74 * int __sha256_ce_transform(struct sha256_ce_state *sst, u8 const *src,
75 * int blocks)
76 */
77 .text
78SYM_FUNC_START(__sha256_ce_transform)
79 /* load round constants */
80 adr_l x8, .Lsha2_rcon
81 ld1 { v0.4s- v3.4s}, [x8], #64
82 ld1 { v4.4s- v7.4s}, [x8], #64
83 ld1 { v8.4s-v11.4s}, [x8], #64
84 ld1 {v12.4s-v15.4s}, [x8]
85
86 /* load state */
87 ld1 {dgav.4s, dgbv.4s}, [x0]
88
89 /* load sha256_ce_state::finalize */
90 ldr_l w4, sha256_ce_offsetof_finalize, x4
91 ldr w4, [x0, x4]
92
93 /* load input */
940: ld1 {v16.4s-v19.4s}, [x1], #64
95 sub w2, w2, #1
96
97CPU_LE( rev32 v16.16b, v16.16b )
98CPU_LE( rev32 v17.16b, v17.16b )
99CPU_LE( rev32 v18.16b, v18.16b )
100CPU_LE( rev32 v19.16b, v19.16b )
101
1021: add t0.4s, v16.4s, v0.4s
103 mov dg0v.16b, dgav.16b
104 mov dg1v.16b, dgbv.16b
105
106 add_update 0, v1, 16, 17, 18, 19
107 add_update 1, v2, 17, 18, 19, 16
108 add_update 0, v3, 18, 19, 16, 17
109 add_update 1, v4, 19, 16, 17, 18
110
111 add_update 0, v5, 16, 17, 18, 19
112 add_update 1, v6, 17, 18, 19, 16
113 add_update 0, v7, 18, 19, 16, 17
114 add_update 1, v8, 19, 16, 17, 18
115
116 add_update 0, v9, 16, 17, 18, 19
117 add_update 1, v10, 17, 18, 19, 16
118 add_update 0, v11, 18, 19, 16, 17
119 add_update 1, v12, 19, 16, 17, 18
120
121 add_only 0, v13, 17
122 add_only 1, v14, 18
123 add_only 0, v15, 19
124 add_only 1
125
126 /* update state */
127 add dgav.4s, dgav.4s, dg0v.4s
128 add dgbv.4s, dgbv.4s, dg1v.4s
129
130 /* handled all input blocks? */
131 cbz w2, 2f
132 cond_yield 3f, x5, x6
133 b 0b
134
135 /*
136 * Final block: add padding and total bit count.
137 * Skip if the input size was not a round multiple of the block size,
138 * the padding is handled by the C code in that case.
139 */
1402: cbz x4, 3f
141 ldr_l w4, sha256_ce_offsetof_count, x4
142 ldr x4, [x0, x4]
143 movi v17.2d, #0
144 mov x8, #0x80000000
145 movi v18.2d, #0
146 ror x7, x4, #29 // ror(lsl(x4, 3), 32)
147 fmov d16, x8
148 mov x4, #0
149 mov v19.d[0], xzr
150 mov v19.d[1], x7
151 b 1b
152
153 /* store new state */
1543: st1 {dgav.4s, dgbv.4s}, [x0]
155 mov w0, w2
156 ret
157SYM_FUNC_END(__sha256_ce_transform)
158

source code of linux/arch/arm64/crypto/sha2-ce-core.S