2 * Copyright (c) 2020 Doc.ai and/or its affiliates.
3 * Copyright (c) 2012 Samuel Neves <sneves@dei.uc.pt>.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at:
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 More information about the BLAKE2 hash function can be found at
25 #include <wireguard/blake/blake2s.h>
26 #include "blake2-impl.h"
28 static const uint32_t blake2s_IV[8] = {
29 0x6A09E667UL, 0xBB67AE85UL, 0x3C6EF372UL, 0xA54FF53AUL,
30 0x510E527FUL, 0x9B05688CUL, 0x1F83D9ABUL, 0x5BE0CD19UL
33 static const uint8_t blake2s_sigma[10][16] = {
34 {0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15},
35 {14, 10, 4, 8, 9, 15, 13, 6, 1, 12, 0, 2, 11, 7, 5, 3},
36 {11, 8, 12, 0, 5, 2, 15, 13, 10, 14, 3, 6, 7, 1, 9, 4},
37 {7, 9, 3, 1, 13, 12, 11, 14, 2, 6, 5, 10, 4, 0, 15, 8},
38 {9, 0, 5, 7, 2, 4, 10, 15, 14, 1, 11, 12, 6, 8, 3, 13},
39 {2, 12, 6, 10, 0, 11, 8, 3, 4, 13, 7, 5, 15, 14, 1, 9},
40 {12, 5, 1, 15, 14, 13, 4, 10, 0, 7, 6, 3, 9, 2, 8, 11},
41 {13, 11, 7, 14, 12, 1, 3, 9, 5, 0, 15, 4, 8, 6, 2, 10},
42 {6, 15, 14, 9, 11, 3, 0, 8, 12, 2, 13, 7, 1, 4, 10, 5},
43 {10, 2, 8, 4, 7, 6, 1, 5, 15, 11, 9, 14, 3, 12, 13, 0},
47 blake2s_set_lastnode (blake2s_state_t * S)
49 S->f[1] = (uint32_t) - 1;
52 /* Some helper functions, not necessarily useful */
54 blake2s_is_lastblock (const blake2s_state_t * S)
60 blake2s_set_lastblock (blake2s_state_t * S)
63 blake2s_set_lastnode (S);
65 S->f[0] = (uint32_t) - 1;
69 blake2s_increment_counter (blake2s_state_t * S, const uint32_t inc)
72 S->t[1] += (S->t[0] < inc);
76 blake2s_init0 (blake2s_state_t * S)
79 memset (S, 0, sizeof (blake2s_state_t));
81 for (i = 0; i < 8; ++i)
82 S->h[i] = blake2s_IV[i];
85 /* init2 xors IV with input parameter block */
87 blake2s_init_param (blake2s_state_t * S, const blake2s_param_t * P)
89 const unsigned char *p = (const unsigned char *) (P);
94 /* IV XOR ParamBlock */
95 for (i = 0; i < 8; ++i)
96 S->h[i] ^= load32 (&p[i * 4]);
98 S->outlen = P->digest_length;
103 /* Sequential blake2s initialization */
105 blake2s_init (blake2s_state_t * S, size_t outlen)
107 blake2s_param_t P[1];
109 /* Move interval verification here? */
110 if ((!outlen) || (outlen > BLAKE2S_OUT_BYTES))
113 P->digest_length = (uint8_t) outlen;
117 store32 (&P->leaf_length, 0);
118 store32 (&P->node_offset, 0);
119 store16 (&P->xof_length, 0);
122 /* memset(P->reserved, 0, sizeof(P->reserved) ); */
123 memset (P->salt, 0, sizeof (P->salt));
124 memset (P->personal, 0, sizeof (P->personal));
125 return blake2s_init_param (S, P);
129 blake2s_init_key (blake2s_state_t * S, size_t outlen, const void *key,
132 blake2s_param_t P[1];
134 if ((!outlen) || (outlen > BLAKE2S_OUT_BYTES))
137 if (!key || !keylen || keylen > BLAKE2S_KEY_BYTES)
140 P->digest_length = (uint8_t) outlen;
141 P->key_length = (uint8_t) keylen;
144 store32 (&P->leaf_length, 0);
145 store32 (&P->node_offset, 0);
146 store16 (&P->xof_length, 0);
149 /* memset(P->reserved, 0, sizeof(P->reserved) ); */
150 memset (P->salt, 0, sizeof (P->salt));
151 memset (P->personal, 0, sizeof (P->personal));
153 if (blake2s_init_param (S, P) < 0)
157 uint8_t block[BLAKE2S_BLOCK_BYTES];
158 memset (block, 0, BLAKE2S_BLOCK_BYTES);
159 memcpy (block, key, keylen);
160 blake2s_update (S, block, BLAKE2S_BLOCK_BYTES);
161 secure_zero_memory (block, BLAKE2S_BLOCK_BYTES); /* Burn the key from stack */
166 #define G(r,i,a,b,c,d) \
168 a = a + b + m[blake2s_sigma[r][2*i+0]]; \
169 d = rotr32(d ^ a, 16); \
171 b = rotr32(b ^ c, 12); \
172 a = a + b + m[blake2s_sigma[r][2*i+1]]; \
173 d = rotr32(d ^ a, 8); \
175 b = rotr32(b ^ c, 7); \
180 G(r,0,v[ 0],v[ 4],v[ 8],v[12]); \
181 G(r,1,v[ 1],v[ 5],v[ 9],v[13]); \
182 G(r,2,v[ 2],v[ 6],v[10],v[14]); \
183 G(r,3,v[ 3],v[ 7],v[11],v[15]); \
184 G(r,4,v[ 0],v[ 5],v[10],v[15]); \
185 G(r,5,v[ 1],v[ 6],v[11],v[12]); \
186 G(r,6,v[ 2],v[ 7],v[ 8],v[13]); \
187 G(r,7,v[ 3],v[ 4],v[ 9],v[14]); \
191 blake2s_compress (blake2s_state_t * S, const uint8_t in[BLAKE2S_BLOCK_BYTES])
197 for (i = 0; i < 16; ++i)
199 m[i] = load32 (in + i * sizeof (m[i]));
202 for (i = 0; i < 8; ++i)
207 v[8] = blake2s_IV[0];
208 v[9] = blake2s_IV[1];
209 v[10] = blake2s_IV[2];
210 v[11] = blake2s_IV[3];
211 v[12] = S->t[0] ^ blake2s_IV[4];
212 v[13] = S->t[1] ^ blake2s_IV[5];
213 v[14] = S->f[0] ^ blake2s_IV[6];
214 v[15] = S->f[1] ^ blake2s_IV[7];
227 for (i = 0; i < 8; ++i)
229 S->h[i] = S->h[i] ^ v[i] ^ v[i + 8];
237 blake2s_update (blake2s_state_t * S, const void *pin, size_t inlen)
239 const unsigned char *in = (const unsigned char *) pin;
242 size_t left = S->buflen;
243 size_t fill = BLAKE2S_BLOCK_BYTES - left;
247 memcpy (S->buf + left, in, fill); /* Fill buffer */
248 blake2s_increment_counter (S, BLAKE2S_BLOCK_BYTES);
249 blake2s_compress (S, S->buf); /* Compress */
252 while (inlen > BLAKE2S_BLOCK_BYTES)
254 blake2s_increment_counter (S, BLAKE2S_BLOCK_BYTES);
255 blake2s_compress (S, in);
256 in += BLAKE2S_BLOCK_BYTES;
257 inlen -= BLAKE2S_BLOCK_BYTES;
260 memcpy (S->buf + S->buflen, in, inlen);
267 blake2s_final (blake2s_state_t * S, void *out, size_t outlen)
269 uint8_t buffer[BLAKE2S_OUT_BYTES] = { 0 };
272 if (out == NULL || outlen < S->outlen)
275 if (blake2s_is_lastblock (S))
278 blake2s_increment_counter (S, (uint32_t) S->buflen);
279 blake2s_set_lastblock (S);
280 memset (S->buf + S->buflen, 0, BLAKE2S_BLOCK_BYTES - S->buflen); /* Padding */
281 blake2s_compress (S, S->buf);
283 for (i = 0; i < 8; ++i) /* Output full hash to temp buffer */
284 store32 (buffer + sizeof (S->h[i]) * i, S->h[i]);
286 memcpy (out, buffer, outlen);
287 secure_zero_memory (buffer, sizeof (buffer));
292 blake2s (void *out, size_t outlen, const void *in, size_t inlen,
293 const void *key, size_t keylen)
295 blake2s_state_t S[1];
297 /* Verify parameters */
298 if (NULL == in && inlen > 0)
304 if (NULL == key && keylen > 0)
307 if (!outlen || outlen > BLAKE2S_OUT_BYTES)
310 if (keylen > BLAKE2S_KEY_BYTES)
315 if (blake2s_init_key (S, outlen, key, keylen) < 0)
320 if (blake2s_init (S, outlen) < 0)
324 blake2s_update (S, (const uint8_t *) in, inlen);
325 blake2s_final (S, out, outlen);
330 * fd.io coding-style-patch-verification: ON
333 * eval: (c-set-style "gnu")