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	 * size_t __sha256_ce_transform(struct sha256_block_state *state,
75	 *				const u8 *data, size_t nblocks);
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 input */
900:	ld1		{v16.4s-v19.4s}, [x1], #64
91	sub		x2, x2, #1
92
93CPU_LE(	rev32		v16.16b, v16.16b	)
94CPU_LE(	rev32		v17.16b, v17.16b	)
95CPU_LE(	rev32		v18.16b, v18.16b	)
96CPU_LE(	rev32		v19.16b, v19.16b	)
97
98	add		t0.4s, v16.4s, v0.4s
99	mov		dg0v.16b, dgav.16b
100	mov		dg1v.16b, dgbv.16b
101
102	add_update	0,  v1, 16, 17, 18, 19
103	add_update	1,  v2, 17, 18, 19, 16
104	add_update	0,  v3, 18, 19, 16, 17
105	add_update	1,  v4, 19, 16, 17, 18
106
107	add_update	0,  v5, 16, 17, 18, 19
108	add_update	1,  v6, 17, 18, 19, 16
109	add_update	0,  v7, 18, 19, 16, 17
110	add_update	1,  v8, 19, 16, 17, 18
111
112	add_update	0,  v9, 16, 17, 18, 19
113	add_update	1, v10, 17, 18, 19, 16
114	add_update	0, v11, 18, 19, 16, 17
115	add_update	1, v12, 19, 16, 17, 18
116
117	add_only	0, v13, 17
118	add_only	1, v14, 18
119	add_only	0, v15, 19
120	add_only	1
121
122	/* update state */
123	add		dgav.4s, dgav.4s, dg0v.4s
124	add		dgbv.4s, dgbv.4s, dg1v.4s
125
126	/* return early if voluntary preemption is needed */
127	cond_yield	1f, x5, x6
128
129	/* handled all input blocks? */
130	cbnz		x2, 0b
131
132	/* store new state */
1331:	st1		{dgav.4s, dgbv.4s}, [x0]
134	mov		x0, x2
135	ret
136SYM_FUNC_END(__sha256_ce_transform)
137