crypto: blake2s - generic C library implementation and selftest

/* SPDX-License-Identifier: GPL-2.0 OR MIT */

/*

 * Copyright (C) 2015-2019 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.

 */

#ifndef BLAKE2S_H

#define BLAKE2S_H

#include <linux/types.h>

#include <linux/kernel.h>

#include <linux/string.h>

#include <asm/bug.h>

enum blake2s_lengths {

	BLAKE2S_BLOCK_SIZE = 64,

	BLAKE2S_HASH_SIZE = 32,

	BLAKE2S_KEY_SIZE = 32,

	BLAKE2S_128_HASH_SIZE = 16,

	BLAKE2S_160_HASH_SIZE = 20,

	BLAKE2S_224_HASH_SIZE = 28,

	BLAKE2S_256_HASH_SIZE = 32,

};

struct blake2s_state {

	u32 h[8];

	u32 t[2];

	u32 f[2];

	u8 buf[BLAKE2S_BLOCK_SIZE];

	unsigned int buflen;

	unsigned int outlen;

};

enum blake2s_iv {

	BLAKE2S_IV0 = 0x6A09E667UL,

	BLAKE2S_IV1 = 0xBB67AE85UL,

	BLAKE2S_IV2 = 0x3C6EF372UL,

	BLAKE2S_IV3 = 0xA54FF53AUL,

	BLAKE2S_IV4 = 0x510E527FUL,

	BLAKE2S_IV5 = 0x9B05688CUL,

	BLAKE2S_IV6 = 0x1F83D9ABUL,

	BLAKE2S_IV7 = 0x5BE0CD19UL,

};

void blake2s_update(struct blake2s_state *state, const u8 *in, size_t inlen);

void blake2s_final(struct blake2s_state *state, u8 *out);

static inline void blake2s_init_param(struct blake2s_state *state,

				      const u32 param)

{

	*state = (struct blake2s_state){{

		BLAKE2S_IV0 ^ param,

		BLAKE2S_IV1,

		BLAKE2S_IV2,

		BLAKE2S_IV3,

		BLAKE2S_IV4,

		BLAKE2S_IV5,

		BLAKE2S_IV6,

		BLAKE2S_IV7,

	}};

}

static inline void blake2s_init(struct blake2s_state *state,

				const size_t outlen)

{

	blake2s_init_param(state, 0x01010000 | outlen);

	state->outlen = outlen;

}

static inline void blake2s_init_key(struct blake2s_state *state,

				    const size_t outlen, const void *key,

				    const size_t keylen)

{

	WARN_ON(IS_ENABLED(DEBUG) && (!outlen || outlen > BLAKE2S_HASH_SIZE ||

		!key || !keylen || keylen > BLAKE2S_KEY_SIZE));

	blake2s_init_param(state, 0x01010000 | keylen << 8 | outlen);

	memcpy(state->buf, key, keylen);

	state->buflen = BLAKE2S_BLOCK_SIZE;

	state->outlen = outlen;

}

static inline void blake2s(u8 *out, const u8 *in, const u8 *key,

			   const size_t outlen, const size_t inlen,

			   const size_t keylen)

{

	struct blake2s_state state;

	WARN_ON(IS_ENABLED(DEBUG) && ((!in && inlen > 0) || !out || !outlen ||

		outlen > BLAKE2S_HASH_SIZE || keylen > BLAKE2S_KEY_SIZE ||

		(!key && keylen)));

	if (keylen)

		blake2s_init_key(&state, outlen, key, keylen);

	else

		blake2s_init(&state, outlen);

	blake2s_update(&state, in, inlen);

	blake2s_final(&state, out);

}

void blake2s256_hmac(u8 *out, const u8 *in, const u8 *key, const size_t inlen,

		     const size_t keylen);

#endif /* BLAKE2S_H */

/* SPDX-License-Identifier: GPL-2.0 OR MIT */

#ifndef BLAKE2S_INTERNAL_H

#define BLAKE2S_INTERNAL_H

#include <crypto/blake2s.h>

void blake2s_compress_generic(struct blake2s_state *state,const u8 *block,

			      size_t nblocks, const u32 inc);

void blake2s_compress_arch(struct blake2s_state *state,const u8 *block,

			   size_t nblocks, const u32 inc);

static inline void blake2s_set_lastblock(struct blake2s_state *state)

{

	state->f[0] = -1;

}

#endif /* BLAKE2S_INTERNAL_H */

obj-$(CONFIG_CRYPTO_LIB_SHA256) += libsha256.o

libsha256-y := sha256.o

obj-y += libblake2s.o

libblake2s-y += blake2s.o blake2s-generic.o

ifneq ($(CONFIG_CRYPTO_MANAGER_DISABLE_TESTS),y)

libblake2s-y += blake2s-selftest.o

endif

// SPDX-License-Identifier: GPL-2.0 OR MIT

/*

 * Copyright (C) 2015-2019 Jason A. Donenfeld <Jason@zx2c4.com>. All Rights Reserved.

 *

 * This is an implementation of the BLAKE2s hash and PRF functions.

 *

 * Information: https://blake2.net/

 *

 */

#include <crypto/internal/blake2s.h>

#include <linux/types.h>

#include <linux/string.h>

#include <linux/kernel.h>

#include <linux/module.h>

#include <linux/init.h>

#include <linux/bug.h>

#include <asm/unaligned.h>

static const u8 blake2s_sigma[10][16] = {

	{ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15 },

	{ 14, 10, 4, 8, 9, 15, 13, 6, 1, 12, 0, 2, 11, 7, 5, 3 },

	{ 11, 8, 12, 0, 5, 2, 15, 13, 10, 14, 3, 6, 7, 1, 9, 4 },

	{ 7, 9, 3, 1, 13, 12, 11, 14, 2, 6, 5, 10, 4, 0, 15, 8 },

	{ 9, 0, 5, 7, 2, 4, 10, 15, 14, 1, 11, 12, 6, 8, 3, 13 },

	{ 2, 12, 6, 10, 0, 11, 8, 3, 4, 13, 7, 5, 15, 14, 1, 9 },

	{ 12, 5, 1, 15, 14, 13, 4, 10, 0, 7, 6, 3, 9, 2, 8, 11 },

	{ 13, 11, 7, 14, 12, 1, 3, 9, 5, 0, 15, 4, 8, 6, 2, 10 },

	{ 6, 15, 14, 9, 11, 3, 0, 8, 12, 2, 13, 7, 1, 4, 10, 5 },

	{ 10, 2, 8, 4, 7, 6, 1, 5, 15, 11, 9, 14, 3, 12, 13, 0 },

};

static inline void blake2s_increment_counter(struct blake2s_state *state,

					     const u32 inc)

{

	state->t[0] += inc;

	state->t[1] += (state->t[0] < inc);

}

void blake2s_compress_generic(struct blake2s_state *state,const u8 *block,

			      size_t nblocks, const u32 inc)

{

	u32 m[16];

	u32 v[16];

	int i;

	WARN_ON(IS_ENABLED(DEBUG) &&

		(nblocks > 1 && inc != BLAKE2S_BLOCK_SIZE));

	while (nblocks > 0) {

		blake2s_increment_counter(state, inc);

		memcpy(m, block, BLAKE2S_BLOCK_SIZE);

		le32_to_cpu_array(m, ARRAY_SIZE(m));

		memcpy(v, state->h, 32);

		v[ 8] = BLAKE2S_IV0;

		v[ 9] = BLAKE2S_IV1;

		v[10] = BLAKE2S_IV2;

		v[11] = BLAKE2S_IV3;

		v[12] = BLAKE2S_IV4 ^ state->t[0];

		v[13] = BLAKE2S_IV5 ^ state->t[1];

		v[14] = BLAKE2S_IV6 ^ state->f[0];

		v[15] = BLAKE2S_IV7 ^ state->f[1];

#define G(r, i, a, b, c, d) do { \

	a += b + m[blake2s_sigma[r][2 * i + 0]]; \

	d = ror32(d ^ a, 16); \

	c += d; \

	b = ror32(b ^ c, 12); \

	a += b + m[blake2s_sigma[r][2 * i + 1]]; \

	d = ror32(d ^ a, 8); \

	c += d; \

	b = ror32(b ^ c, 7); \

} while (0)

#define ROUND(r) do { \

	G(r, 0, v[0], v[ 4], v[ 8], v[12]); \

	G(r, 1, v[1], v[ 5], v[ 9], v[13]); \

	G(r, 2, v[2], v[ 6], v[10], v[14]); \

	G(r, 3, v[3], v[ 7], v[11], v[15]); \

	G(r, 4, v[0], v[ 5], v[10], v[15]); \

	G(r, 5, v[1], v[ 6], v[11], v[12]); \

	G(r, 6, v[2], v[ 7], v[ 8], v[13]); \

	G(r, 7, v[3], v[ 4], v[ 9], v[14]); \

} while (0)

		ROUND(0);

		ROUND(1);

		ROUND(2);

		ROUND(3);

		ROUND(4);

		ROUND(5);

		ROUND(6);

		ROUND(7);

		ROUND(8);

		ROUND(9);

#undef G

#undef ROUND

		for (i = 0; i < 8; ++i)

			state->h[i] ^= v[i] ^ v[i + 8];

		block += BLAKE2S_BLOCK_SIZE;

		--nblocks;

	}

}

EXPORT_SYMBOL(blake2s_compress_generic);

MODULE_LICENSE("GPL v2");

MODULE_DESCRIPTION("BLAKE2s hash function");

MODULE_AUTHOR("Jason A. Donenfeld <Jason@zx2c4.com>");