xref: /linux-6.15/lib/crypto/sha256.c (revision 63642d5c)
1 // SPDX-License-Identifier: GPL-2.0-or-later
2 /*
3  * SHA-256, as specified in
4  * http://csrc.nist.gov/groups/STM/cavp/documents/shs/sha256-384-512.pdf
5  *
6  * SHA-256 code by Jean-Luc Cooke <[email protected]>.
7  *
8  * Copyright (c) Jean-Luc Cooke <[email protected]>
9  * Copyright (c) Andrew McDonald <[email protected]>
10  * Copyright (c) 2002 James Morris <[email protected]>
11  * Copyright (c) 2014 Red Hat Inc.
12  */
13 
14 #include <linux/bitops.h>
15 #include <linux/export.h>
16 #include <linux/module.h>
17 #include <linux/string.h>
18 #include <crypto/sha.h>
19 #include <asm/unaligned.h>
20 
21 static const u32 SHA256_K[] = {
22 	0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5,
23 	0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5,
24 	0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3,
25 	0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174,
26 	0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc,
27 	0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da,
28 	0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7,
29 	0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967,
30 	0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13,
31 	0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85,
32 	0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3,
33 	0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070,
34 	0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5,
35 	0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3,
36 	0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208,
37 	0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2,
38 };
39 
40 static inline u32 Ch(u32 x, u32 y, u32 z)
41 {
42 	return z ^ (x & (y ^ z));
43 }
44 
45 static inline u32 Maj(u32 x, u32 y, u32 z)
46 {
47 	return (x & y) | (z & (x | y));
48 }
49 
50 #define e0(x)       (ror32(x, 2) ^ ror32(x, 13) ^ ror32(x, 22))
51 #define e1(x)       (ror32(x, 6) ^ ror32(x, 11) ^ ror32(x, 25))
52 #define s0(x)       (ror32(x, 7) ^ ror32(x, 18) ^ (x >> 3))
53 #define s1(x)       (ror32(x, 17) ^ ror32(x, 19) ^ (x >> 10))
54 
55 static inline void LOAD_OP(int I, u32 *W, const u8 *input)
56 {
57 	W[I] = get_unaligned_be32((__u32 *)input + I);
58 }
59 
60 static inline void BLEND_OP(int I, u32 *W)
61 {
62 	W[I] = s1(W[I-2]) + W[I-7] + s0(W[I-15]) + W[I-16];
63 }
64 
65 #define SHA256_ROUND(i, a, b, c, d, e, f, g, h) do {		\
66 	u32 t1, t2;						\
67 	t1 = h + e1(e) + Ch(e, f, g) + SHA256_K[i] + W[i];	\
68 	t2 = e0(a) + Maj(a, b, c);				\
69 	d += t1;						\
70 	h = t1 + t2;						\
71 } while (0)
72 
73 static void sha256_transform(u32 *state, const u8 *input, u32 *W)
74 {
75 	u32 a, b, c, d, e, f, g, h;
76 	int i;
77 
78 	/* load the input */
79 	for (i = 0; i < 16; i++)
80 		LOAD_OP(i, W, input);
81 
82 	/* now blend */
83 	for (i = 16; i < 64; i++)
84 		BLEND_OP(i, W);
85 
86 	/* load the state into our registers */
87 	a = state[0];  b = state[1];  c = state[2];  d = state[3];
88 	e = state[4];  f = state[5];  g = state[6];  h = state[7];
89 
90 	/* now iterate */
91 	for (i = 0; i < 64; i += 8) {
92 		SHA256_ROUND(i + 0, a, b, c, d, e, f, g, h);
93 		SHA256_ROUND(i + 1, h, a, b, c, d, e, f, g);
94 		SHA256_ROUND(i + 2, g, h, a, b, c, d, e, f);
95 		SHA256_ROUND(i + 3, f, g, h, a, b, c, d, e);
96 		SHA256_ROUND(i + 4, e, f, g, h, a, b, c, d);
97 		SHA256_ROUND(i + 5, d, e, f, g, h, a, b, c);
98 		SHA256_ROUND(i + 6, c, d, e, f, g, h, a, b);
99 		SHA256_ROUND(i + 7, b, c, d, e, f, g, h, a);
100 	}
101 
102 	state[0] += a; state[1] += b; state[2] += c; state[3] += d;
103 	state[4] += e; state[5] += f; state[6] += g; state[7] += h;
104 }
105 
106 void sha256_update(struct sha256_state *sctx, const u8 *data, unsigned int len)
107 {
108 	unsigned int partial, done;
109 	const u8 *src;
110 	u32 W[64];
111 
112 	partial = sctx->count & 0x3f;
113 	sctx->count += len;
114 	done = 0;
115 	src = data;
116 
117 	if ((partial + len) > 63) {
118 		if (partial) {
119 			done = -partial;
120 			memcpy(sctx->buf + partial, data, done + 64);
121 			src = sctx->buf;
122 		}
123 
124 		do {
125 			sha256_transform(sctx->state, src, W);
126 			done += 64;
127 			src = data + done;
128 		} while (done + 63 < len);
129 
130 		memzero_explicit(W, sizeof(W));
131 
132 		partial = 0;
133 	}
134 	memcpy(sctx->buf + partial, src, len - done);
135 }
136 EXPORT_SYMBOL(sha256_update);
137 
138 void sha224_update(struct sha256_state *sctx, const u8 *data, unsigned int len)
139 {
140 	sha256_update(sctx, data, len);
141 }
142 EXPORT_SYMBOL(sha224_update);
143 
144 static void __sha256_final(struct sha256_state *sctx, u8 *out, int digest_words)
145 {
146 	__be32 *dst = (__be32 *)out;
147 	__be64 bits;
148 	unsigned int index, pad_len;
149 	int i;
150 	static const u8 padding[64] = { 0x80, };
151 
152 	/* Save number of bits */
153 	bits = cpu_to_be64(sctx->count << 3);
154 
155 	/* Pad out to 56 mod 64. */
156 	index = sctx->count & 0x3f;
157 	pad_len = (index < 56) ? (56 - index) : ((64+56) - index);
158 	sha256_update(sctx, padding, pad_len);
159 
160 	/* Append length (before padding) */
161 	sha256_update(sctx, (const u8 *)&bits, sizeof(bits));
162 
163 	/* Store state in digest */
164 	for (i = 0; i < digest_words; i++)
165 		put_unaligned_be32(sctx->state[i], &dst[i]);
166 
167 	/* Zeroize sensitive information. */
168 	memzero_explicit(sctx, sizeof(*sctx));
169 }
170 
171 void sha256_final(struct sha256_state *sctx, u8 *out)
172 {
173 	__sha256_final(sctx, out, 8);
174 }
175 EXPORT_SYMBOL(sha256_final);
176 
177 void sha224_final(struct sha256_state *sctx, u8 *out)
178 {
179 	__sha256_final(sctx, out, 7);
180 }
181 EXPORT_SYMBOL(sha224_final);
182 
183 void sha256(const u8 *data, unsigned int len, u8 *out)
184 {
185 	struct sha256_state sctx;
186 
187 	sha256_init(&sctx);
188 	sha256_update(&sctx, data, len);
189 	sha256_final(&sctx, out);
190 }
191 EXPORT_SYMBOL(sha256);
192 
193 MODULE_LICENSE("GPL");
194