1 // SPDX-License-Identifier: (GPL-2.0-only OR Apache-2.0)
2 /*
3 * BLAKE2b reference source code package - reference C implementations
4 *
5 * Copyright 2012, Samuel Neves <sneves@dei.uc.pt>. You may use this under the
6 * terms of the CC0, the OpenSSL Licence, or the Apache Public License 2.0, at
7 * your option. The terms of these licenses can be found at:
8 *
9 * - CC0 1.0 Universal : http://creativecommons.org/publicdomain/zero/1.0
10 * - OpenSSL license : https://www.openssl.org/source/license.html
11 * - Apache 2.0 : https://www.apache.org/licenses/LICENSE-2.0
12 *
13 * More information about the BLAKE2 hash function can be found at
14 * https://blake2.net.
15 *
16 * Note: the original sources have been modified for inclusion in linux kernel
17 * in terms of coding style, using generic helpers and simplifications of error
18 * handling.
19 */
20
21 #include <asm/unaligned.h>
22 #include <linux/module.h>
23 #include <linux/string.h>
24 #include <linux/kernel.h>
25 #include <linux/bitops.h>
26 #include <crypto/internal/hash.h>
27
28 #define BLAKE2B_160_DIGEST_SIZE (160 / 8)
29 #define BLAKE2B_256_DIGEST_SIZE (256 / 8)
30 #define BLAKE2B_384_DIGEST_SIZE (384 / 8)
31 #define BLAKE2B_512_DIGEST_SIZE (512 / 8)
32
33 enum blake2b_constant {
34 BLAKE2B_BLOCKBYTES = 128,
35 BLAKE2B_KEYBYTES = 64,
36 };
37
38 struct blake2b_state {
39 u64 h[8];
40 u64 t[2];
41 u64 f[2];
42 u8 buf[BLAKE2B_BLOCKBYTES];
43 size_t buflen;
44 };
45
46 static const u64 blake2b_IV[8] = {
47 0x6a09e667f3bcc908ULL, 0xbb67ae8584caa73bULL,
48 0x3c6ef372fe94f82bULL, 0xa54ff53a5f1d36f1ULL,
49 0x510e527fade682d1ULL, 0x9b05688c2b3e6c1fULL,
50 0x1f83d9abfb41bd6bULL, 0x5be0cd19137e2179ULL
51 };
52
53 static const u8 blake2b_sigma[12][16] = {
54 { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 },
55 { 14, 10, 4, 8, 9, 15, 13, 6, 1, 12, 0, 2, 11, 7, 5, 3 },
56 { 11, 8, 12, 0, 5, 2, 15, 13, 10, 14, 3, 6, 7, 1, 9, 4 },
57 { 7, 9, 3, 1, 13, 12, 11, 14, 2, 6, 5, 10, 4, 0, 15, 8 },
58 { 9, 0, 5, 7, 2, 4, 10, 15, 14, 1, 11, 12, 6, 8, 3, 13 },
59 { 2, 12, 6, 10, 0, 11, 8, 3, 4, 13, 7, 5, 15, 14, 1, 9 },
60 { 12, 5, 1, 15, 14, 13, 4, 10, 0, 7, 6, 3, 9, 2, 8, 11 },
61 { 13, 11, 7, 14, 12, 1, 3, 9, 5, 0, 15, 4, 8, 6, 2, 10 },
62 { 6, 15, 14, 9, 11, 3, 0, 8, 12, 2, 13, 7, 1, 4, 10, 5 },
63 { 10, 2, 8, 4, 7, 6, 1, 5, 15, 11, 9, 14, 3, 12, 13, 0 },
64 { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 },
65 { 14, 10, 4, 8, 9, 15, 13, 6, 1, 12, 0, 2, 11, 7, 5, 3 }
66 };
67
blake2b_increment_counter(struct blake2b_state * S,const u64 inc)68 static void blake2b_increment_counter(struct blake2b_state *S, const u64 inc)
69 {
70 S->t[0] += inc;
71 S->t[1] += (S->t[0] < inc);
72 }
73
74 #define G(r,i,a,b,c,d) \
75 do { \
76 a = a + b + m[blake2b_sigma[r][2*i+0]]; \
77 d = ror64(d ^ a, 32); \
78 c = c + d; \
79 b = ror64(b ^ c, 24); \
80 a = a + b + m[blake2b_sigma[r][2*i+1]]; \
81 d = ror64(d ^ a, 16); \
82 c = c + d; \
83 b = ror64(b ^ c, 63); \
84 } while (0)
85
86 #define ROUND(r) \
87 do { \
88 G(r,0,v[ 0],v[ 4],v[ 8],v[12]); \
89 G(r,1,v[ 1],v[ 5],v[ 9],v[13]); \
90 G(r,2,v[ 2],v[ 6],v[10],v[14]); \
91 G(r,3,v[ 3],v[ 7],v[11],v[15]); \
92 G(r,4,v[ 0],v[ 5],v[10],v[15]); \
93 G(r,5,v[ 1],v[ 6],v[11],v[12]); \
94 G(r,6,v[ 2],v[ 7],v[ 8],v[13]); \
95 G(r,7,v[ 3],v[ 4],v[ 9],v[14]); \
96 } while (0)
97
blake2b_compress(struct blake2b_state * S,const u8 block[BLAKE2B_BLOCKBYTES])98 static void blake2b_compress(struct blake2b_state *S,
99 const u8 block[BLAKE2B_BLOCKBYTES])
100 {
101 u64 m[16];
102 u64 v[16];
103 size_t i;
104
105 for (i = 0; i < 16; ++i)
106 m[i] = get_unaligned_le64(block + i * sizeof(m[i]));
107
108 for (i = 0; i < 8; ++i)
109 v[i] = S->h[i];
110
111 v[ 8] = blake2b_IV[0];
112 v[ 9] = blake2b_IV[1];
113 v[10] = blake2b_IV[2];
114 v[11] = blake2b_IV[3];
115 v[12] = blake2b_IV[4] ^ S->t[0];
116 v[13] = blake2b_IV[5] ^ S->t[1];
117 v[14] = blake2b_IV[6] ^ S->f[0];
118 v[15] = blake2b_IV[7] ^ S->f[1];
119
120 ROUND(0);
121 ROUND(1);
122 ROUND(2);
123 ROUND(3);
124 ROUND(4);
125 ROUND(5);
126 ROUND(6);
127 ROUND(7);
128 ROUND(8);
129 ROUND(9);
130 ROUND(10);
131 ROUND(11);
132 #ifdef CONFIG_CC_IS_CLANG
133 #pragma nounroll /* https://bugs.llvm.org/show_bug.cgi?id=45803 */
134 #endif
135 for (i = 0; i < 8; ++i)
136 S->h[i] = S->h[i] ^ v[i] ^ v[i + 8];
137 }
138
139 #undef G
140 #undef ROUND
141
142 struct blake2b_tfm_ctx {
143 u8 key[BLAKE2B_KEYBYTES];
144 unsigned int keylen;
145 };
146
blake2b_setkey(struct crypto_shash * tfm,const u8 * key,unsigned int keylen)147 static int blake2b_setkey(struct crypto_shash *tfm, const u8 *key,
148 unsigned int keylen)
149 {
150 struct blake2b_tfm_ctx *tctx = crypto_shash_ctx(tfm);
151
152 if (keylen == 0 || keylen > BLAKE2B_KEYBYTES)
153 return -EINVAL;
154
155 memcpy(tctx->key, key, keylen);
156 tctx->keylen = keylen;
157
158 return 0;
159 }
160
blake2b_init(struct shash_desc * desc)161 static int blake2b_init(struct shash_desc *desc)
162 {
163 struct blake2b_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm);
164 struct blake2b_state *state = shash_desc_ctx(desc);
165 const int digestsize = crypto_shash_digestsize(desc->tfm);
166
167 memset(state, 0, sizeof(*state));
168 memcpy(state->h, blake2b_IV, sizeof(state->h));
169
170 /* Parameter block is all zeros except index 0, no xor for 1..7 */
171 state->h[0] ^= 0x01010000 | tctx->keylen << 8 | digestsize;
172
173 if (tctx->keylen) {
174 /*
175 * Prefill the buffer with the key, next call to _update or
176 * _final will process it
177 */
178 memcpy(state->buf, tctx->key, tctx->keylen);
179 state->buflen = BLAKE2B_BLOCKBYTES;
180 }
181 return 0;
182 }
183
blake2b_update(struct shash_desc * desc,const u8 * in,unsigned int inlen)184 static int blake2b_update(struct shash_desc *desc, const u8 *in,
185 unsigned int inlen)
186 {
187 struct blake2b_state *state = shash_desc_ctx(desc);
188 const size_t left = state->buflen;
189 const size_t fill = BLAKE2B_BLOCKBYTES - left;
190
191 if (!inlen)
192 return 0;
193
194 if (inlen > fill) {
195 state->buflen = 0;
196 /* Fill buffer */
197 memcpy(state->buf + left, in, fill);
198 blake2b_increment_counter(state, BLAKE2B_BLOCKBYTES);
199 /* Compress */
200 blake2b_compress(state, state->buf);
201 in += fill;
202 inlen -= fill;
203 while (inlen > BLAKE2B_BLOCKBYTES) {
204 blake2b_increment_counter(state, BLAKE2B_BLOCKBYTES);
205 blake2b_compress(state, in);
206 in += BLAKE2B_BLOCKBYTES;
207 inlen -= BLAKE2B_BLOCKBYTES;
208 }
209 }
210 memcpy(state->buf + state->buflen, in, inlen);
211 state->buflen += inlen;
212
213 return 0;
214 }
215
blake2b_final(struct shash_desc * desc,u8 * out)216 static int blake2b_final(struct shash_desc *desc, u8 *out)
217 {
218 struct blake2b_state *state = shash_desc_ctx(desc);
219 const int digestsize = crypto_shash_digestsize(desc->tfm);
220 size_t i;
221
222 blake2b_increment_counter(state, state->buflen);
223 /* Set last block */
224 state->f[0] = (u64)-1;
225 /* Padding */
226 memset(state->buf + state->buflen, 0, BLAKE2B_BLOCKBYTES - state->buflen);
227 blake2b_compress(state, state->buf);
228
229 /* Avoid temporary buffer and switch the internal output to LE order */
230 for (i = 0; i < ARRAY_SIZE(state->h); i++)
231 __cpu_to_le64s(&state->h[i]);
232
233 memcpy(out, state->h, digestsize);
234 return 0;
235 }
236
237 static struct shash_alg blake2b_algs[] = {
238 {
239 .base.cra_name = "blake2b-160",
240 .base.cra_driver_name = "blake2b-160-generic",
241 .base.cra_priority = 100,
242 .base.cra_flags = CRYPTO_ALG_OPTIONAL_KEY,
243 .base.cra_blocksize = BLAKE2B_BLOCKBYTES,
244 .base.cra_ctxsize = sizeof(struct blake2b_tfm_ctx),
245 .base.cra_module = THIS_MODULE,
246 .digestsize = BLAKE2B_160_DIGEST_SIZE,
247 .setkey = blake2b_setkey,
248 .init = blake2b_init,
249 .update = blake2b_update,
250 .final = blake2b_final,
251 .descsize = sizeof(struct blake2b_state),
252 }, {
253 .base.cra_name = "blake2b-256",
254 .base.cra_driver_name = "blake2b-256-generic",
255 .base.cra_priority = 100,
256 .base.cra_flags = CRYPTO_ALG_OPTIONAL_KEY,
257 .base.cra_blocksize = BLAKE2B_BLOCKBYTES,
258 .base.cra_ctxsize = sizeof(struct blake2b_tfm_ctx),
259 .base.cra_module = THIS_MODULE,
260 .digestsize = BLAKE2B_256_DIGEST_SIZE,
261 .setkey = blake2b_setkey,
262 .init = blake2b_init,
263 .update = blake2b_update,
264 .final = blake2b_final,
265 .descsize = sizeof(struct blake2b_state),
266 }, {
267 .base.cra_name = "blake2b-384",
268 .base.cra_driver_name = "blake2b-384-generic",
269 .base.cra_priority = 100,
270 .base.cra_flags = CRYPTO_ALG_OPTIONAL_KEY,
271 .base.cra_blocksize = BLAKE2B_BLOCKBYTES,
272 .base.cra_ctxsize = sizeof(struct blake2b_tfm_ctx),
273 .base.cra_module = THIS_MODULE,
274 .digestsize = BLAKE2B_384_DIGEST_SIZE,
275 .setkey = blake2b_setkey,
276 .init = blake2b_init,
277 .update = blake2b_update,
278 .final = blake2b_final,
279 .descsize = sizeof(struct blake2b_state),
280 }, {
281 .base.cra_name = "blake2b-512",
282 .base.cra_driver_name = "blake2b-512-generic",
283 .base.cra_priority = 100,
284 .base.cra_flags = CRYPTO_ALG_OPTIONAL_KEY,
285 .base.cra_blocksize = BLAKE2B_BLOCKBYTES,
286 .base.cra_ctxsize = sizeof(struct blake2b_tfm_ctx),
287 .base.cra_module = THIS_MODULE,
288 .digestsize = BLAKE2B_512_DIGEST_SIZE,
289 .setkey = blake2b_setkey,
290 .init = blake2b_init,
291 .update = blake2b_update,
292 .final = blake2b_final,
293 .descsize = sizeof(struct blake2b_state),
294 }
295 };
296
blake2b_mod_init(void)297 static int __init blake2b_mod_init(void)
298 {
299 return crypto_register_shashes(blake2b_algs, ARRAY_SIZE(blake2b_algs));
300 }
301
blake2b_mod_fini(void)302 static void __exit blake2b_mod_fini(void)
303 {
304 crypto_unregister_shashes(blake2b_algs, ARRAY_SIZE(blake2b_algs));
305 }
306
307 subsys_initcall(blake2b_mod_init);
308 module_exit(blake2b_mod_fini);
309
310 MODULE_AUTHOR("David Sterba <kdave@kernel.org>");
311 MODULE_DESCRIPTION("BLAKE2b generic implementation");
312 MODULE_LICENSE("GPL");
313 MODULE_ALIAS_CRYPTO("blake2b-160");
314 MODULE_ALIAS_CRYPTO("blake2b-160-generic");
315 MODULE_ALIAS_CRYPTO("blake2b-256");
316 MODULE_ALIAS_CRYPTO("blake2b-256-generic");
317 MODULE_ALIAS_CRYPTO("blake2b-384");
318 MODULE_ALIAS_CRYPTO("blake2b-384-generic");
319 MODULE_ALIAS_CRYPTO("blake2b-512");
320 MODULE_ALIAS_CRYPTO("blake2b-512-generic");
321
