[fio.git] / crc / sha3.c

/*
 * Cryptographic API.
 *
 * SHA-3, as specified in
 * http://nvlpubs.nist.gov/nistpubs/FIPS/NIST.FIPS.202.pdf
 *
 * SHA-3 code by Jeff Garzik <jeff@garzik.org>
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by the Free
 * Software Foundation; either version 2 of the License, or (at your option)•
 * any later version.
 *
 */
#include <string.h>

#include "../os/os.h"

#include "sha3.h"

#define KECCAK_ROUNDS 24

#define ROTL64(x, y) (((x) << (y)) | ((x) >> (64 - (y))))

static const uint64_t keccakf_rndc[24] = {
	0x0000000000000001ULL, 0x0000000000008082ULL, 0x800000000000808aULL,
	0x8000000080008000ULL, 0x000000000000808bULL, 0x0000000080000001ULL,
	0x8000000080008081ULL, 0x8000000000008009ULL, 0x000000000000008aULL,
	0x0000000000000088ULL, 0x0000000080008009ULL, 0x000000008000000aULL,
	0x000000008000808bULL, 0x800000000000008bULL, 0x8000000000008089ULL,
	0x8000000000008003ULL, 0x8000000000008002ULL, 0x8000000000000080ULL,
	0x000000000000800aULL, 0x800000008000000aULL, 0x8000000080008081ULL,
	0x8000000000008080ULL, 0x0000000080000001ULL, 0x8000000080008008ULL
};

static const int keccakf_rotc[24] = {
	1,  3,  6,  10, 15, 21, 28, 36, 45, 55, 2,  14,
	27, 41, 56, 8,  25, 43, 62, 18, 39, 61, 20, 44
};

static const int keccakf_piln[24] = {
	10, 7,  11, 17, 18, 3, 5,  16, 8,  21, 24, 4,
	15, 23, 19, 13, 12, 2, 20, 14, 22, 9,  6,  1
};

/* update the state with given number of rounds */

static void keccakf(uint64_t st[25])
{
	int i, j, round;
	uint64_t t, bc[5];

	for (round = 0; round < KECCAK_ROUNDS; round++) {

		/* Theta */
		for (i = 0; i < 5; i++)
			bc[i] = st[i] ^ st[i + 5] ^ st[i + 10] ^ st[i + 15]
				^ st[i + 20];

		for (i = 0; i < 5; i++) {
			t = bc[(i + 4) % 5] ^ ROTL64(bc[(i + 1) % 5], 1);
			for (j = 0; j < 25; j += 5)
				st[j + i] ^= t;
		}

		/* Rho Pi */
		t = st[1];
		for (i = 0; i < 24; i++) {
			j = keccakf_piln[i];
			bc[0] = st[j];
			st[j] = ROTL64(t, keccakf_rotc[i]);
			t = bc[0];
		}

		/* Chi */
		for (j = 0; j < 25; j += 5) {
			for (i = 0; i < 5; i++)
				bc[i] = st[j + i];
			for (i = 0; i < 5; i++)
				st[j + i] ^= (~bc[(i + 1) % 5]) &
					     bc[(i + 2) % 5];
		}

		/* Iota */
		st[0] ^= keccakf_rndc[round];
	}
}

static void fio_sha3_init(struct fio_sha3_ctx *sctx, unsigned int digest_sz)
{
	memset(sctx->st, 0, sizeof(sctx->st));
	sctx->md_len = digest_sz;
	sctx->rsiz = 200 - 2 * digest_sz;
	sctx->rsizw = sctx->rsiz / 8;
	sctx->partial = 0;
	memset(sctx->buf, 0, sizeof(sctx->buf));
}

void fio_sha3_224_init(struct fio_sha3_ctx *sctx)
{
	fio_sha3_init(sctx, SHA3_224_DIGEST_SIZE);
}

void fio_sha3_256_init(struct fio_sha3_ctx *sctx)
{
	fio_sha3_init(sctx, SHA3_256_DIGEST_SIZE);
}

void fio_sha3_384_init(struct fio_sha3_ctx *sctx)
{
	fio_sha3_init(sctx, SHA3_384_DIGEST_SIZE);
}

void fio_sha3_512_init(struct fio_sha3_ctx *sctx)
{
	fio_sha3_init(sctx, SHA3_512_DIGEST_SIZE);
}

int fio_sha3_update(struct fio_sha3_ctx *sctx, const uint8_t *data,
		    unsigned int len)
{
	unsigned int done;
	const uint8_t *src;

	done = 0;
	src = data;

	if ((sctx->partial + len) > (sctx->rsiz - 1)) {
		if (sctx->partial) {
			done = -sctx->partial;
			memcpy(sctx->buf + sctx->partial, data,
			       done + sctx->rsiz);
			src = sctx->buf;
		}

		do {
			unsigned int i;

			for (i = 0; i < sctx->rsizw; i++)
				sctx->st[i] ^= ((uint64_t *) src)[i];
			keccakf(sctx->st);

			done += sctx->rsiz;
			src = data + done;
		} while (done + (sctx->rsiz - 1) < len);

		sctx->partial = 0;
	}
	memcpy(sctx->buf + sctx->partial, src, len - done);
	sctx->partial += (len - done);

	return 0;
}

void fio_sha3_final(struct fio_sha3_ctx *sctx)
{
	unsigned int i, inlen = sctx->partial;

	sctx->buf[inlen++] = 0x06;
	memset(sctx->buf + inlen, 0, sctx->rsiz - inlen);
	sctx->buf[sctx->rsiz - 1] |= 0x80;

	for (i = 0; i < sctx->rsizw; i++)
		sctx->st[i] ^= ((uint64_t *) sctx->buf)[i];

	keccakf(sctx->st);

	for (i = 0; i < sctx->rsizw; i++)
		sctx->st[i] = cpu_to_le64(sctx->st[i]);

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