1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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
33enum blake2b_constant {
34 BLAKE2B_BLOCKBYTES = 128,
35 BLAKE2B_KEYBYTES = 64,
36};
37
38struct 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
46static const u64 blake2b_IV[8] = {
47 0x6a09e667f3bcc908ULL, 0xbb67ae8584caa73bULL,
48 0x3c6ef372fe94f82bULL, 0xa54ff53a5f1d36f1ULL,
49 0x510e527fade682d1ULL, 0x9b05688c2b3e6c1fULL,
50 0x1f83d9abfb41bd6bULL, 0x5be0cd19137e2179ULL
51};
52
53static 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
68static 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
98static 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
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
142struct blake2b_tfm_ctx {
143 u8 key[BLAKE2B_KEYBYTES];
144 unsigned int keylen;
145};
146
147static 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
161static 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
171 state->h[0] ^= 0x01010000 | tctx->keylen << 8 | digestsize;
172
173 if (tctx->keylen) {
174
175
176
177
178 memcpy(state->buf, tctx->key, tctx->keylen);
179 state->buflen = BLAKE2B_BLOCKBYTES;
180 }
181 return 0;
182}
183
184static 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
197 memcpy(state->buf + left, in, fill);
198 blake2b_increment_counter(state, BLAKE2B_BLOCKBYTES);
199
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
216static 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
224 state->f[0] = (u64)-1;
225
226 memset(state->buf + state->buflen, 0, BLAKE2B_BLOCKBYTES - state->buflen);
227 blake2b_compress(state, state->buf);
228
229
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
237static 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
297static int __init blake2b_mod_init(void)
298{
299 return crypto_register_shashes(blake2b_algs, ARRAY_SIZE(blake2b_algs));
300}
301
302static void __exit blake2b_mod_fini(void)
303{
304 crypto_unregister_shashes(blake2b_algs, ARRAY_SIZE(blake2b_algs));
305}
306
307subsys_initcall(blake2b_mod_init);
308module_exit(blake2b_mod_fini);
309
310MODULE_AUTHOR("David Sterba <kdave@kernel.org>");
311MODULE_DESCRIPTION("BLAKE2b generic implementation");
312MODULE_LICENSE("GPL");
313MODULE_ALIAS_CRYPTO("blake2b-160");
314MODULE_ALIAS_CRYPTO("blake2b-160-generic");
315MODULE_ALIAS_CRYPTO("blake2b-256");
316MODULE_ALIAS_CRYPTO("blake2b-256-generic");
317MODULE_ALIAS_CRYPTO("blake2b-384");
318MODULE_ALIAS_CRYPTO("blake2b-384-generic");
319MODULE_ALIAS_CRYPTO("blake2b-512");
320MODULE_ALIAS_CRYPTO("blake2b-512-generic");
321
