• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * Cryptographic API.
3  *
4  * SHA-3, as specified in
5  * http://nvlpubs.nist.gov/nistpubs/FIPS/NIST.FIPS.202.pdf
6  *
7  * SHA-3 code by Jeff Garzik <jeff@garzik.org>
8  *
9  * This program is free software; you can redistribute it and/or modify it
10  * under the terms of the GNU General Public License as published by the Free
11  * Software Foundation; either version 2 of the License, or (at your option)•
12  * any later version.
13  *
14  */
15 #include <string.h>
16 #include <inttypes.h>
17 
18 #include "../os/os.h"
19 
20 #include "sha3.h"
21 
22 #define KECCAK_ROUNDS 24
23 
24 #define ROTL64(x, y) (((x) << (y)) | ((x) >> (64 - (y))))
25 
26 static const uint64_t keccakf_rndc[24] = {
27 	0x0000000000000001ULL, 0x0000000000008082ULL, 0x800000000000808aULL,
28 	0x8000000080008000ULL, 0x000000000000808bULL, 0x0000000080000001ULL,
29 	0x8000000080008081ULL, 0x8000000000008009ULL, 0x000000000000008aULL,
30 	0x0000000000000088ULL, 0x0000000080008009ULL, 0x000000008000000aULL,
31 	0x000000008000808bULL, 0x800000000000008bULL, 0x8000000000008089ULL,
32 	0x8000000000008003ULL, 0x8000000000008002ULL, 0x8000000000000080ULL,
33 	0x000000000000800aULL, 0x800000008000000aULL, 0x8000000080008081ULL,
34 	0x8000000000008080ULL, 0x0000000080000001ULL, 0x8000000080008008ULL
35 };
36 
37 static const int keccakf_rotc[24] = {
38 	1,  3,  6,  10, 15, 21, 28, 36, 45, 55, 2,  14,
39 	27, 41, 56, 8,  25, 43, 62, 18, 39, 61, 20, 44
40 };
41 
42 static const int keccakf_piln[24] = {
43 	10, 7,  11, 17, 18, 3, 5,  16, 8,  21, 24, 4,
44 	15, 23, 19, 13, 12, 2, 20, 14, 22, 9,  6,  1
45 };
46 
47 /* update the state with given number of rounds */
48 
keccakf(uint64_t st[25])49 static void keccakf(uint64_t st[25])
50 {
51 	int i, j, round;
52 	uint64_t t, bc[5];
53 
54 	for (round = 0; round < KECCAK_ROUNDS; round++) {
55 
56 		/* Theta */
57 		for (i = 0; i < 5; i++)
58 			bc[i] = st[i] ^ st[i + 5] ^ st[i + 10] ^ st[i + 15]
59 				^ st[i + 20];
60 
61 		for (i = 0; i < 5; i++) {
62 			t = bc[(i + 4) % 5] ^ ROTL64(bc[(i + 1) % 5], 1);
63 			for (j = 0; j < 25; j += 5)
64 				st[j + i] ^= t;
65 		}
66 
67 		/* Rho Pi */
68 		t = st[1];
69 		for (i = 0; i < 24; i++) {
70 			j = keccakf_piln[i];
71 			bc[0] = st[j];
72 			st[j] = ROTL64(t, keccakf_rotc[i]);
73 			t = bc[0];
74 		}
75 
76 		/* Chi */
77 		for (j = 0; j < 25; j += 5) {
78 			for (i = 0; i < 5; i++)
79 				bc[i] = st[j + i];
80 			for (i = 0; i < 5; i++)
81 				st[j + i] ^= (~bc[(i + 1) % 5]) &
82 					     bc[(i + 2) % 5];
83 		}
84 
85 		/* Iota */
86 		st[0] ^= keccakf_rndc[round];
87 	}
88 }
89 
fio_sha3_init(struct fio_sha3_ctx * sctx,unsigned int digest_sz)90 static void fio_sha3_init(struct fio_sha3_ctx *sctx, unsigned int digest_sz)
91 {
92 	memset(sctx->st, 0, sizeof(sctx->st));
93 	sctx->md_len = digest_sz;
94 	sctx->rsiz = 200 - 2 * digest_sz;
95 	sctx->rsizw = sctx->rsiz / 8;
96 	sctx->partial = 0;
97 	memset(sctx->buf, 0, sizeof(sctx->buf));
98 }
99 
fio_sha3_224_init(struct fio_sha3_ctx * sctx)100 void fio_sha3_224_init(struct fio_sha3_ctx *sctx)
101 {
102 	fio_sha3_init(sctx, SHA3_224_DIGEST_SIZE);
103 }
104 
fio_sha3_256_init(struct fio_sha3_ctx * sctx)105 void fio_sha3_256_init(struct fio_sha3_ctx *sctx)
106 {
107 	fio_sha3_init(sctx, SHA3_256_DIGEST_SIZE);
108 }
109 
fio_sha3_384_init(struct fio_sha3_ctx * sctx)110 void fio_sha3_384_init(struct fio_sha3_ctx *sctx)
111 {
112 	fio_sha3_init(sctx, SHA3_384_DIGEST_SIZE);
113 }
114 
fio_sha3_512_init(struct fio_sha3_ctx * sctx)115 void fio_sha3_512_init(struct fio_sha3_ctx *sctx)
116 {
117 	fio_sha3_init(sctx, SHA3_512_DIGEST_SIZE);
118 }
119 
fio_sha3_update(struct fio_sha3_ctx * sctx,const uint8_t * data,unsigned int len)120 int fio_sha3_update(struct fio_sha3_ctx *sctx, const uint8_t *data,
121 		    unsigned int len)
122 {
123 	unsigned int done;
124 	const uint8_t *src;
125 
126 	done = 0;
127 	src = data;
128 
129 	if ((sctx->partial + len) > (sctx->rsiz - 1)) {
130 		if (sctx->partial) {
131 			done = -sctx->partial;
132 			memcpy(sctx->buf + sctx->partial, data,
133 			       done + sctx->rsiz);
134 			src = sctx->buf;
135 		}
136 
137 		do {
138 			unsigned int i;
139 
140 			for (i = 0; i < sctx->rsizw; i++)
141 				sctx->st[i] ^= ((uint64_t *) src)[i];
142 			keccakf(sctx->st);
143 
144 			done += sctx->rsiz;
145 			src = data + done;
146 		} while (done + (sctx->rsiz - 1) < len);
147 
148 		sctx->partial = 0;
149 	}
150 	memcpy(sctx->buf + sctx->partial, src, len - done);
151 	sctx->partial += (len - done);
152 
153 	return 0;
154 }
155 
fio_sha3_final(struct fio_sha3_ctx * sctx)156 void fio_sha3_final(struct fio_sha3_ctx *sctx)
157 {
158 	unsigned int i, inlen = sctx->partial;
159 
160 	sctx->buf[inlen++] = 0x06;
161 	memset(sctx->buf + inlen, 0, sctx->rsiz - inlen);
162 	sctx->buf[sctx->rsiz - 1] |= 0x80;
163 
164 	for (i = 0; i < sctx->rsizw; i++)
165 		sctx->st[i] ^= ((uint64_t *) sctx->buf)[i];
166 
167 	keccakf(sctx->st);
168 
169 	for (i = 0; i < sctx->rsizw; i++)
170 		sctx->st[i] = cpu_to_le64(sctx->st[i]);
171 
172 	memcpy(sctx->sha, sctx->st, sctx->md_len);
173 }
174