2a8a726c5e15294a91c48e24a35e5bf5c46c8509
[akaros.git] / kern / lib / random / sha2.c
1 /*      $OpenBSD: sha2.c,v 1.6 2004/05/03 02:57:36 millert Exp $        */
2
3 /*
4  * FILE:        sha2.c
5  * AUTHOR:      Aaron D. Gifford <me@aarongifford.com>
6  *
7  * Copyright (c) 2000-2001, Aaron D. Gifford
8  * All rights reserved.
9  *
10  * Redistribution and use in source and binary forms, with or without
11  * modification, are permitted provided that the following conditions
12  * are met:
13  * 1. Redistributions of source code must retain the above copyright
14  *        notice, this list of conditions and the following disclaimer.
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *        notice, this list of conditions and the following disclaimer in the
17  *        documentation and/or other materials provided with the distribution.
18  * 3. Neither the name of the copyright holder nor the names of contributors
19  *        may be used to endorse or promote products derived from this software
20  *        without specific prior written permission.
21  *
22  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTOR(S) ``AS IS'' AND
23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25  * ARE DISCLAIMED.      IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTOR(S) BE LIABLE
26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32  * SUCH DAMAGE.
33  *
34  * $From: sha2.c,v 1.1 2001/11/08 00:01:51 adg Exp adg $
35  *
36  * contrib/pgcrypto/sha2.c
37  */
38
39 #include "u.h"
40 #include <libc.h>
41 #include "sha2.h"
42
43
44 /*** SHA-256/512 Various Length Definitions ***********************/
45 enum{
46         SHA256ShortBlockLength  = (SHA256BlockLength - 8),
47         SHA512ShortBlockLength  = (SHA512_block_length - 16)
48 };
49
50 /*
51  * Macro for incrementally adding the unsigned 64-bit integer n to the
52  * unsigned 128-bit integer (represented using a two-element array of
53  * 64-bit words):
54  */
55 #define ADDINC128(w,n)  { \
56         (w)[0] += (uint64_t)(n); \
57         if ((w)[0] < (n)) { \
58                 (w)[1]++; \
59         } \
60 }
61
62 /*** THE SIX LOGICAL FUNCTIONS ****************************************/
63 /*
64  * Bit shifting and rotation (used by the six SHA-XYZ logical functions:
65  *
66  *   NOTE:  The naming of R and S appears backwards here (R is a SHIFT and
67  *   S is a ROTATION) because the SHA-256/384/512 description document
68  *   (see http://csrc.nist.gov/cryptval/shs/sha256-384-512.pdf) uses this
69  *   same "backwards" definition.
70  */
71 /* Shift-right (used in SHA-256, SHA-384, and SHA-512): */
72 #define R(b,x)          ((x) >> (b))
73 /* 32-bit Rotate-right (used in SHA-256): */
74 #define S32(b,x)        (((x) >> (b)) | ((x) << (32 - (b))))
75 /* 64-bit Rotate-right (used in SHA-384 and SHA-512): */
76 #define S64(b,x)        (((x) >> (b)) | ((x) << (64 - (b))))
77
78 /* Two of six logical functions used in SHA-256, SHA-384, and SHA-512: */
79 #define Ch(x,y,z)       (((x) & (y)) ^ ((~(x)) & (z)))
80 #define Maj(x,y,z)      (((x) & (y)) ^ ((x) & (z)) ^ ((y) & (z)))
81
82 /* Four of six logical functions used in SHA-256: */
83 #define Sigma0_256(x)   (S32(2,  (x)) ^ S32(13, (x)) ^ S32(22, (x)))
84 #define Sigma1_256(x)   (S32(6,  (x)) ^ S32(11, (x)) ^ S32(25, (x)))
85 #define sigma0_256(x)   (S32(7,  (x)) ^ S32(18, (x)) ^ R(3 ,   (x)))
86 #define sigma1_256(x)   (S32(17, (x)) ^ S32(19, (x)) ^ R(10,   (x)))
87
88 /* Four of six logical functions used in SHA-384 and SHA-512: */
89 #define Sigma0_512(x)   (S64(28, (x)) ^ S64(34, (x)) ^ S64(39, (x)))
90 #define Sigma1_512(x)   (S64(14, (x)) ^ S64(18, (x)) ^ S64(41, (x)))
91 #define sigma0_512(x)   (S64( 1, (x)) ^ S64( 8, (x)) ^ R( 7,   (x)))
92 #define sigma1_512(x)   (S64(19, (x)) ^ S64(61, (x)) ^ R( 6,   (x)))
93
94 /*** INTERNAL FUNCTION PROTOTYPES *************************************/
95 /* NOTE: These should not be accessed directly from outside this
96  * library -- they are intended for private internal visibility/use
97  * only.
98  */
99 static void SHA512_Last(SHA512Ctx*);
100 static void SHA256_Transform(SHA256Ctx*, const uint32_t*);
101 static void SHA512_Transform(SHA512Ctx*, const uint64_t*);
102
103 /*** SHA-XYZ INITIAL HASH VALUES AND CONSTANTS ************************/
104 /* Hash constant words K for SHA-256: */
105 const uint32_t K256[64] = {
106         0x428a2f98UL, 0x71374491UL, 0xb5c0fbcfUL, 0xe9b5dba5UL,
107         0x3956c25bUL, 0x59f111f1UL, 0x923f82a4UL, 0xab1c5ed5UL,
108         0xd807aa98UL, 0x12835b01UL, 0x243185beUL, 0x550c7dc3UL,
109         0x72be5d74UL, 0x80deb1feUL, 0x9bdc06a7UL, 0xc19bf174UL,
110         0xe49b69c1UL, 0xefbe4786UL, 0x0fc19dc6UL, 0x240ca1ccUL,
111         0x2de92c6fUL, 0x4a7484aaUL, 0x5cb0a9dcUL, 0x76f988daUL,
112         0x983e5152UL, 0xa831c66dUL, 0xb00327c8UL, 0xbf597fc7UL,
113         0xc6e00bf3UL, 0xd5a79147UL, 0x06ca6351UL, 0x14292967UL,
114         0x27b70a85UL, 0x2e1b2138UL, 0x4d2c6dfcUL, 0x53380d13UL,
115         0x650a7354UL, 0x766a0abbUL, 0x81c2c92eUL, 0x92722c85UL,
116         0xa2bfe8a1UL, 0xa81a664bUL, 0xc24b8b70UL, 0xc76c51a3UL,
117         0xd192e819UL, 0xd6990624UL, 0xf40e3585UL, 0x106aa070UL,
118         0x19a4c116UL, 0x1e376c08UL, 0x2748774cUL, 0x34b0bcb5UL,
119         0x391c0cb3UL, 0x4ed8aa4aUL, 0x5b9cca4fUL, 0x682e6ff3UL,
120         0x748f82eeUL, 0x78a5636fUL, 0x84c87814UL, 0x8cc70208UL,
121         0x90befffaUL, 0xa4506cebUL, 0xbef9a3f7UL, 0xc67178f2UL
122 };
123
124 /* Initial hash value H for SHA-224: */
125 const uint32_t sha224_initial_hash_value[8] = {
126         0xc1059ed8UL,
127         0x367cd507UL,
128         0x3070dd17UL,
129         0xf70e5939UL,
130         0xffc00b31UL,
131         0x68581511UL,
132         0x64f98fa7UL,
133         0xbefa4fa4UL
134 };
135
136 /* Initial hash value H for SHA-256: */
137 static const uint32_t sha256_initial_hash_value[8] = {
138         0x6a09e667UL,
139         0xbb67ae85UL,
140         0x3c6ef372UL,
141         0xa54ff53aUL,
142         0x510e527fUL,
143         0x9b05688cUL,
144         0x1f83d9abUL,
145         0x5be0cd19UL
146 };
147
148 /* Hash constant words K for SHA-384 and SHA-512: */
149 static const uint64_t K512[80] = {
150         0x428a2f98d728ae22ULL, 0x7137449123ef65cdULL,
151         0xb5c0fbcfec4d3b2fULL, 0xe9b5dba58189dbbcULL,
152         0x3956c25bf348b538ULL, 0x59f111f1b605d019ULL,
153         0x923f82a4af194f9bULL, 0xab1c5ed5da6d8118ULL,
154         0xd807aa98a3030242ULL, 0x12835b0145706fbeULL,
155         0x243185be4ee4b28cULL, 0x550c7dc3d5ffb4e2ULL,
156         0x72be5d74f27b896fULL, 0x80deb1fe3b1696b1ULL,
157         0x9bdc06a725c71235ULL, 0xc19bf174cf692694ULL,
158         0xe49b69c19ef14ad2ULL, 0xefbe4786384f25e3ULL,
159         0x0fc19dc68b8cd5b5ULL, 0x240ca1cc77ac9c65ULL,
160         0x2de92c6f592b0275ULL, 0x4a7484aa6ea6e483ULL,
161         0x5cb0a9dcbd41fbd4ULL, 0x76f988da831153b5ULL,
162         0x983e5152ee66dfabULL, 0xa831c66d2db43210ULL,
163         0xb00327c898fb213fULL, 0xbf597fc7beef0ee4ULL,
164         0xc6e00bf33da88fc2ULL, 0xd5a79147930aa725ULL,
165         0x06ca6351e003826fULL, 0x142929670a0e6e70ULL,
166         0x27b70a8546d22ffcULL, 0x2e1b21385c26c926ULL,
167         0x4d2c6dfc5ac42aedULL, 0x53380d139d95b3dfULL,
168         0x650a73548baf63deULL, 0x766a0abb3c77b2a8ULL,
169         0x81c2c92e47edaee6ULL, 0x92722c851482353bULL,
170         0xa2bfe8a14cf10364ULL, 0xa81a664bbc423001ULL,
171         0xc24b8b70d0f89791ULL, 0xc76c51a30654be30ULL,
172         0xd192e819d6ef5218ULL, 0xd69906245565a910ULL,
173         0xf40e35855771202aULL, 0x106aa07032bbd1b8ULL,
174         0x19a4c116b8d2d0c8ULL, 0x1e376c085141ab53ULL,
175         0x2748774cdf8eeb99ULL, 0x34b0bcb5e19b48a8ULL,
176         0x391c0cb3c5c95a63ULL, 0x4ed8aa4ae3418acbULL,
177         0x5b9cca4f7763e373ULL, 0x682e6ff3d6b2b8a3ULL,
178         0x748f82ee5defb2fcULL, 0x78a5636f43172f60ULL,
179         0x84c87814a1f0ab72ULL, 0x8cc702081a6439ecULL,
180         0x90befffa23631e28ULL, 0xa4506cebde82bde9ULL,
181         0xbef9a3f7b2c67915ULL, 0xc67178f2e372532bULL,
182         0xca273eceea26619cULL, 0xd186b8c721c0c207ULL,
183         0xeada7dd6cde0eb1eULL, 0xf57d4f7fee6ed178ULL,
184         0x06f067aa72176fbaULL, 0x0a637dc5a2c898a6ULL,
185         0x113f9804bef90daeULL, 0x1b710b35131c471bULL,
186         0x28db77f523047d84ULL, 0x32caab7b40c72493ULL,
187         0x3c9ebe0a15c9bebcULL, 0x431d67c49c100d4cULL,
188         0x4cc5d4becb3e42b6ULL, 0x597f299cfc657e2aULL,
189         0x5fcb6fab3ad6faecULL, 0x6c44198c4a475817ULL
190 };
191
192 /* Initial hash value H for SHA-384 */
193 static const uint64_t sha384_initial_hash_value[8] = {
194         0xcbbb9d5dc1059ed8ULL,
195         0x629a292a367cd507ULL,
196         0x9159015a3070dd17ULL,
197         0x152fecd8f70e5939ULL,
198         0x67332667ffc00b31ULL,
199         0x8eb44a8768581511ULL,
200         0xdb0c2e0d64f98fa7ULL,
201         0x47b5481dbefa4fa4ULL
202 };
203
204 /* Initial hash value H for SHA-512 */
205 static const uint64_t sha512_initial_hash_value[8] = {
206         0x6a09e667f3bcc908ULL,
207         0xbb67ae8584caa73bULL,
208         0x3c6ef372fe94f82bULL,
209         0xa54ff53a5f1d36f1ULL,
210         0x510e527fade682d1ULL,
211         0x9b05688c2b3e6c1fULL,
212         0x1f83d9abfb41bd6bULL,
213         0x5be0cd19137e2179ULL
214 };
215
216
217 /*** SHA-256: *********************************************************/
218 void
219 SHA256_Init(SHA256Ctx *context)
220 {
221         if (context == nil)
222                 return;
223         memmove(context->state, sha256_initial_hash_value, SHA256DigestLength);
224         memset(context->buffer, 0, SHA256BlockLength);
225         context->bitcount = 0;
226 }
227 static void
228 SHA256_Transform(SHA256Ctx *context, const uint32_t *data)
229 {
230         uint32_t                a,
231                                 b,
232                                 c,
233                                 d,
234                                 e,
235                                 f,
236                                 g,
237                                 h,
238                                 s0,
239                                 s1;
240         uint32_t                T1,
241                                 T2,
242                            *W256;
243         int                     j;
244
245         W256 = (uint32_t *) context->buffer;
246
247         /* Initialize registers with the prev. intermediate value */
248         a = context->state[0];
249         b = context->state[1];
250         c = context->state[2];
251         d = context->state[3];
252         e = context->state[4];
253         f = context->state[5];
254         g = context->state[6];
255         h = context->state[7];
256
257         j = 0;
258         do
259         {
260                 W256[j] = (uint32_t) data[3] | ((uint32_t) data[2] << 8) |
261                         ((uint32_t) data[1] << 16) | ((uint32_t) data[0] << 24);
262                 data += 4;
263                 /* Apply the SHA-256 compression function to update a..h */
264                 T1 = h + Sigma1_256(e) + Ch(e, f, g) + K256[j] + W256[j];
265                 T2 = Sigma0_256(a) + Maj(a, b, c);
266                 h = g;
267                 g = f;
268                 f = e;
269                 e = d + T1;
270                 d = c;
271                 c = b;
272                 b = a;
273                 a = T1 + T2;
274
275                 j++;
276         } while (j < 16);
277
278         do
279         {
280                 /* Part of the message block expansion: */
281                 s0 = W256[(j + 1) & 0x0f];
282                 s0 = sigma0_256(s0);
283                 s1 = W256[(j + 14) & 0x0f];
284                 s1 = sigma1_256(s1);
285
286                 /* Apply the SHA-256 compression function to update a..h */
287                 T1 = h + Sigma1_256(e) + Ch(e, f, g) + K256[j] +
288                         (W256[j & 0x0f] += s1 + W256[(j + 9) & 0x0f] + s0);
289                 T2 = Sigma0_256(a) + Maj(a, b, c);
290                 h = g;
291                 g = f;
292                 f = e;
293                 e = d + T1;
294                 d = c;
295                 c = b;
296                 b = a;
297                 a = T1 + T2;
298
299                 j++;
300         } while (j < 64);
301
302         /* Compute the current intermediate hash value */
303         context->state[0] += a;
304         context->state[1] += b;
305         context->state[2] += c;
306         context->state[3] += d;
307         context->state[4] += e;
308         context->state[5] += f;
309         context->state[6] += g;
310         context->state[7] += h;
311
312         /* Clean up */
313         a = b = c = d = e = f = g = h = T1 = T2 = 0;
314 }
315
316 void
317 SHA256_Update(SHA256Ctx *context, const uint8_t *data, size_t len)
318 {
319         size_t          freespace,
320                                 usedspace;
321
322         /* Calling with no data is valid (we do nothing) */
323         if (len == 0)
324                 return;
325
326         usedspace = (context->bitcount >> 3) % SHA256BlockLength;
327         if (usedspace > 0)
328         {
329                 /* Calculate how much free space is available in the buffer */
330                 freespace = SHA256BlockLength - usedspace;
331
332                 if (len >= freespace)
333                 {
334                         /* Fill the buffer completely and process it */
335                         memmove(&context->buffer[usedspace], data, freespace);
336                         context->bitcount += freespace << 3;
337                         len -= freespace;
338                         data += freespace;
339                         SHA256_Transform(context, (uint32_t*)context->buffer);
340                 }
341                 else
342                 {
343                         /* The buffer is not yet full */
344                         memmove(&context->buffer[usedspace], data, len);
345                         context->bitcount += len << 3;
346                         /* Clean up: */
347                         usedspace = freespace = 0;
348                         return;
349                 }
350         }
351         while (len >= SHA256BlockLength)
352         {
353                 /* Process as many complete blocks as we can */
354                 SHA256_Transform(context, (const uint32_t*)data);
355                 context->bitcount += SHA256BlockLength << 3;
356                 len -= SHA256BlockLength;
357                 data += SHA256BlockLength;
358         }
359         if (len > 0)
360         {
361                 /* There's left-overs, so save 'em */
362                 memmove(context->buffer, data, len);
363                 context->bitcount += len << 3;
364         }
365         /* Clean up: */
366         usedspace = freespace = 0;
367 }
368
369 static void
370 SHA256_Last(SHA256Ctx *context)
371 {
372         unsigned int usedspace;
373
374         usedspace = (context->bitcount >> 3) % SHA256BlockLength;
375
376         if (usedspace > 0)
377         {
378                 /* Begin padding with a 1 bit: */
379                 context->buffer[usedspace++] = 0x80;
380
381                 if (usedspace <= SHA256ShortBlockLength)
382                 {
383                         /* Set-up for the last transform: */
384                         memset(&context->buffer[usedspace], 0, SHA256ShortBlockLength - usedspace);
385                 }
386                 else
387                 {
388                         if (usedspace < SHA256BlockLength)
389                         {
390                                 memset(&context->buffer[usedspace], 0, SHA256BlockLength - usedspace);
391                         }
392                         /* Do second-to-last transform: */
393                         SHA256_Transform(context, (uint32_t*)context->buffer);
394
395                         /* And set-up for the last transform: */
396                         memset(context->buffer, 0, SHA256ShortBlockLength);
397                 }
398         }
399         else
400         {
401                 /* Set-up for the last transform: */
402                 memset(context->buffer, 0, SHA256ShortBlockLength);
403
404                 /* Begin padding with a 1 bit: */
405                 *context->buffer = 0x80;
406         }
407         /* Set the bit count: */
408         *(uint64_t *) &context->buffer[SHA256ShortBlockLength] = context->bitcount;
409
410         /* Final transform: */
411         SHA256_Transform(context, (uint32_t*)context->buffer);
412 }
413
414 void
415 SHA256_Final(uint8_t digest[], SHA256Ctx *context)
416 {
417         /* If no digest buffer is passed, we don't bother doing this: */
418         if (digest != nil)
419         {
420                 SHA256_Last(context);
421
422                 memmove(digest, context->state, SHA256DigestLength);
423         }
424
425         /* Clean up state data: */
426         memset(context, 0, sizeof(*context));
427 }
428
429
430 /*** SHA-512: *********************************************************/
431 void
432 SHA512_Init(SHA512Ctx *context)
433 {
434         if (context == nil)
435                 return;
436         memmove(context->state, sha512_initial_hash_value, SHA512DigestLength);
437         memset(context->buffer, 0, SHA512_block_length);
438         context->bitcount[0] = context->bitcount[1] = 0;
439 }
440
441 static void
442 SHA512_Transform(SHA512Ctx *context, const uint64_t *data)
443 {
444         uint64_t                a,
445                                 b,
446                                 c,
447                                 d,
448                                 e,
449                                 f,
450                                 g,
451                                 h,
452                                 s0,
453                                 s1;
454         uint64_t                T1,
455                                 T2,
456                            *W512 = (uint64_t *) context->buffer;
457         int                     j;
458
459         /* Initialize registers with the prev. intermediate value */
460         a = context->state[0];
461         b = context->state[1];
462         c = context->state[2];
463         d = context->state[3];
464         e = context->state[4];
465         f = context->state[5];
466         g = context->state[6];
467         h = context->state[7];
468
469         j = 0;
470         do
471         {
472                 W512[j] = (uint64_t) data[7] | ((uint64_t) data[6] << 8) |
473                         ((uint64_t) data[5] << 16) | ((uint64_t) data[4] << 24) |
474                         ((uint64_t) data[3] << 32) | ((uint64_t) data[2] << 40) |
475                         ((uint64_t) data[1] << 48) | ((uint64_t) data[0] << 56);
476                 data += 8;
477                 /* Apply the SHA-512 compression function to update a..h */
478                 T1 = h + Sigma1_512(e) + Ch(e, f, g) + K512[j] + W512[j];
479                 T2 = Sigma0_512(a) + Maj(a, b, c);
480                 h = g;
481                 g = f;
482                 f = e;
483                 e = d + T1;
484                 d = c;
485                 c = b;
486                 b = a;
487                 a = T1 + T2;
488
489                 j++;
490         } while (j < 16);
491
492         do
493         {
494                 /* Part of the message block expansion: */
495                 s0 = W512[(j + 1) & 0x0f];
496                 s0 = sigma0_512(s0);
497                 s1 = W512[(j + 14) & 0x0f];
498                 s1 = sigma1_512(s1);
499
500                 /* Apply the SHA-512 compression function to update a..h */
501                 T1 = h + Sigma1_512(e) + Ch(e, f, g) + K512[j] +
502                         (W512[j & 0x0f] += s1 + W512[(j + 9) & 0x0f] + s0);
503                 T2 = Sigma0_512(a) + Maj(a, b, c);
504                 h = g;
505                 g = f;
506                 f = e;
507                 e = d + T1;
508                 d = c;
509                 c = b;
510                 b = a;
511                 a = T1 + T2;
512
513                 j++;
514         } while (j < 80);
515
516         /* Compute the current intermediate hash value */
517         context->state[0] += a;
518         context->state[1] += b;
519         context->state[2] += c;
520         context->state[3] += d;
521         context->state[4] += e;
522         context->state[5] += f;
523         context->state[6] += g;
524         context->state[7] += h;
525
526         /* Clean up */
527         a = b = c = d = e = f = g = h = T1 = T2 = 0;
528 }
529
530 void
531 SHA512_Update(SHA512Ctx *context, const uint8_t *data, size_t len)
532 {
533         size_t          freespace,
534                                 usedspace;
535
536         /* Calling with no data is valid (we do nothing) */
537         if (len == 0)
538                 return;
539
540         usedspace = (context->bitcount[0] >> 3) % SHA512_block_length;
541         if (usedspace > 0)
542         {
543                 /* Calculate how much free space is available in the buffer */
544                 freespace = SHA512_block_length - usedspace;
545
546                 if (len >= freespace)
547                 {
548                         /* Fill the buffer completely and process it */
549                         memmove(&context->buffer[usedspace], data, freespace);
550                         ADDINC128(context->bitcount, freespace << 3);
551                         len -= freespace;
552                         data += freespace;
553                         SHA512_Transform(context, (uint64_t*)context->buffer);
554                 }
555                 else
556                 {
557                         /* The buffer is not yet full */
558                         memmove(&context->buffer[usedspace], data, len);
559                         ADDINC128(context->bitcount, len << 3);
560                         /* Clean up: */
561                         usedspace = freespace = 0;
562                         return;
563                 }
564         }
565         while (len >= SHA512_block_length)
566         {
567                 /* Process as many complete blocks as we can */
568                 SHA512_Transform(context, (const uint64_t*)data);
569                 ADDINC128(context->bitcount, SHA512_block_length << 3);
570                 len -= SHA512_block_length;
571                 data += SHA512_block_length;
572         }
573         if (len > 0)
574         {
575                 /* There's left-overs, so save 'em */
576                 memmove(context->buffer, data, len);
577                 ADDINC128(context->bitcount, len << 3);
578         }
579         /* Clean up: */
580         usedspace = freespace = 0;
581 }
582
583 static void
584 SHA512_Last(SHA512Ctx *context)
585 {
586         unsigned int usedspace;
587
588         usedspace = (context->bitcount[0] >> 3) % SHA512_block_length;
589
590         if (usedspace > 0)
591         {
592                 /* Begin padding with a 1 bit: */
593                 context->buffer[usedspace++] = 0x80;
594
595                 if (usedspace <= SHA512ShortBlockLength)
596                 {
597                         /* Set-up for the last transform: */
598                         memset(&context->buffer[usedspace], 0, SHA512ShortBlockLength - usedspace);
599                 }
600                 else
601                 {
602                         if (usedspace < SHA512_block_length)
603                         {
604                                 memset(&context->buffer[usedspace], 0, SHA512_block_length - usedspace);
605                         }
606                         /* Do second-to-last transform: */
607                         SHA512_Transform(context, (uint64_t*)context->buffer);
608
609                         /* And set-up for the last transform: */
610                         memset(context->buffer, 0, SHA512_block_length - 2);
611                 }
612         }
613         else
614         {
615                 /* Prepare for final transform: */
616                 memset(context->buffer, 0, SHA512ShortBlockLength);
617
618                 /* Begin padding with a 1 bit: */
619                 *context->buffer = 0x80;
620         }
621         /* Store the length of input data (in bits): */
622         *(uint64_t *) &context->buffer[SHA512ShortBlockLength] = context->bitcount[1];
623         *(uint64_t *) &context->buffer[SHA512ShortBlockLength + 8] = context->bitcount[0];
624
625         /* Final transform: */
626         SHA512_Transform(context, (uint64_t*)context->buffer);
627 }
628
629 void
630 SHA512_Final(uint8_t digest[], SHA512Ctx *context)
631 {
632         /* If no digest buffer is passed, we don't bother doing this: */
633         if (digest != nil)
634         {
635                 SHA512_Last(context);
636
637                 /* Save the hash data for output: */
638                 memmove(digest, context->state, SHA512DigestLength);
639         }
640
641         /* Zero out state data */
642         memset(context, 0, sizeof(*context));
643 }
644
645
646 /*** SHA-384: *********************************************************/
647 void
648 SHA384_Init(SHA384Ctx *context)
649 {
650         if (context == nil)
651                 return;
652         memmove(context->state, sha384_initial_hash_value, SHA512DigestLength);
653         memset(context->buffer, 0, SHA384BlockLength);
654         context->bitcount[0] = context->bitcount[1] = 0;
655 }
656
657 void
658 SHA384_Update(SHA384Ctx *context, const uint8_t *data, size_t len)
659 {
660         SHA512_Update((SHA512Ctx *) context, data, len);
661 }
662
663 void
664 SHA384_Final(uint8_t digest[], SHA384Ctx *context)
665 {
666         /* If no digest buffer is passed, we don't bother doing this: */
667         if (digest != nil)
668         {
669                 SHA512_Last((SHA512Ctx *) context);
670
671                 /* Save the hash data for output: */
672                 memmove(digest, context->state, SHA384DigestLength);
673         }
674
675         /* Zero out state data */
676         memset(context, 0, sizeof(*context));
677 }
678
679 /*** SHA-224: *********************************************************/
680 void
681 SHA224_Init(SHA224Ctx *context)
682 {
683         if (context == nil)
684                 return;
685         memmove(context->state, sha224_initial_hash_value, SHA256DigestLength);
686         memset(context->buffer, 0, SHA256BlockLength);
687         context->bitcount = 0;
688 }
689
690 void
691 SHA224_Update(SHA224Ctx *context, const uint8_t *data, size_t len)
692 {
693         SHA256_Update((SHA256Ctx *) context, data, len);
694 }
695
696 void
697 SHA224_Final(uint8_t digest[], SHA224Ctx *context)
698 {
699         /* If no digest buffer is passed, we don't bother doing this: */
700         if (digest != nil)
701         {
702                 SHA256_Last(context);
703
704                 memmove(digest, context->state, SHA224DigestLength);
705         }
706
707         /* Clean up state data: */
708         memset(context, 0, sizeof(*context));
709 }