crypto/nhpoly1305.c - linux/kernel/git/brauner/linux - Git at Google

 // SPDX-License-Identifier: GPL-2.0
 /*
  * NHPoly1305 - ε-almost-∆-universal hash function for Adiantum
  *
  * Copyright 2018 Google LLC
  */

 /*
  * "NHPoly1305" is the main component of Adiantum hashing.
  * Specifically, it is the calculation
  *
  *	H_L ← Poly1305_{K_L}(NH_{K_N}(pad_{128}(L)))
  *
  * from the procedure in section 6.4 of the Adiantum paper [1].  It is an
  * ε-almost-∆-universal (ε-∆U) hash function for equal-length inputs over
  * Z/(2^{128}Z), where the "∆" operation is addition.  It hashes 1024-byte
  * chunks of the input with the NH hash function [2], reducing the input length
  * by 32x.  The resulting NH digests are evaluated as a polynomial in
  * GF(2^{130}-5), like in the Poly1305 MAC [3].  Note that the polynomial
  * evaluation by itself would suffice to achieve the ε-∆U property; NH is used
  * for performance since it's over twice as fast as Poly1305.
  *
  * This is *not* a cryptographic hash function; do not use it as such!
  *
  * [1] Adiantum: length-preserving encryption for entry-level processors
  *     (https://eprint.iacr.org/2018/720.pdf)
  * [2] UMAC: Fast and Secure Message Authentication
  *     (https://fastcrypto.org/umac/umac_proc.pdf)
  * [3] The Poly1305-AES message-authentication code
  *     (https://cr.yp.to/mac/poly1305-20050329.pdf)
  */

 #include <asm/unaligned.h>
 #include <crypto/algapi.h>
 #include <crypto/internal/hash.h>
 #include <crypto/internal/poly1305.h>
 #include <crypto/nhpoly1305.h>
 #include <linux/crypto.h>
 #include <linux/kernel.h>
 #include <linux/module.h>

 static void nh_generic(const u32 *key, const u8 *message, size_t message_len,
 		       __le64 hash[NH_NUM_PASSES])
 {
 	u64 sums[4] = { 0, 0, 0, 0 };

 	BUILD_BUG_ON(NH_PAIR_STRIDE != 2);
 	BUILD_BUG_ON(NH_NUM_PASSES != 4);

 	while (message_len) {
 		u32 m0 = get_unaligned_le32(message + 0);
 		u32 m1 = get_unaligned_le32(message + 4);
 		u32 m2 = get_unaligned_le32(message + 8);
 		u32 m3 = get_unaligned_le32(message + 12);

 		sums[0] += (u64)(u32)(m0 + key[ 0]) * (u32)(m2 + key[ 2]);
 		sums[1] += (u64)(u32)(m0 + key[ 4]) * (u32)(m2 + key[ 6]);
 		sums[2] += (u64)(u32)(m0 + key[ 8]) * (u32)(m2 + key[10]);
 		sums[3] += (u64)(u32)(m0 + key[12]) * (u32)(m2 + key[14]);
 		sums[0] += (u64)(u32)(m1 + key[ 1]) * (u32)(m3 + key[ 3]);
 		sums[1] += (u64)(u32)(m1 + key[ 5]) * (u32)(m3 + key[ 7]);
 		sums[2] += (u64)(u32)(m1 + key[ 9]) * (u32)(m3 + key[11]);
 		sums[3] += (u64)(u32)(m1 + key[13]) * (u32)(m3 + key[15]);
 		key += NH_MESSAGE_UNIT / sizeof(key[0]);
 		message += NH_MESSAGE_UNIT;
 		message_len -= NH_MESSAGE_UNIT;
 	}

 	hash[0] = cpu_to_le64(sums[0]);
 	hash[1] = cpu_to_le64(sums[1]);
 	hash[2] = cpu_to_le64(sums[2]);
 	hash[3] = cpu_to_le64(sums[3]);
 }

 /* Pass the next NH hash value through Poly1305 */
 static void process_nh_hash_value(struct nhpoly1305_state *state,
 				  const struct nhpoly1305_key *key)
 {
 	BUILD_BUG_ON(NH_HASH_BYTES % POLY1305_BLOCK_SIZE != 0);

 	poly1305_core_blocks(&state->poly_state, &key->poly_key, state->nh_hash,
 			     NH_HASH_BYTES / POLY1305_BLOCK_SIZE, 1);
 }

 /*
  * Feed the next portion of the source data, as a whole number of 16-byte
  * "NH message units", through NH and Poly1305.  Each NH hash is taken over
  * 1024 bytes, except possibly the final one which is taken over a multiple of
  * 16 bytes up to 1024.  Also, in the case where data is passed in misaligned
  * chunks, we combine partial hashes; the end result is the same either way.
  */
 static void nhpoly1305_units(struct nhpoly1305_state *state,
 			     const struct nhpoly1305_key *key,
 			     const u8 *src, unsigned int srclen, nh_t nh_fn)
 {
 	do {
 		unsigned int bytes;

 		if (state->nh_remaining == 0) {
 			/* Starting a new NH message */
 			bytes = min_t(unsigned int, srclen, NH_MESSAGE_BYTES);
 			nh_fn(key->nh_key, src, bytes, state->nh_hash);
 			state->nh_remaining = NH_MESSAGE_BYTES - bytes;
 		} else {
 			/* Continuing a previous NH message */
 			__le64 tmp_hash[NH_NUM_PASSES];
 			unsigned int pos;
 			int i;

 			pos = NH_MESSAGE_BYTES - state->nh_remaining;
 			bytes = min(srclen, state->nh_remaining);
 			nh_fn(&key->nh_key[pos / 4], src, bytes, tmp_hash);
 			for (i = 0; i < NH_NUM_PASSES; i++)
 				le64_add_cpu(&state->nh_hash[i],
 					     le64_to_cpu(tmp_hash[i]));
 			state->nh_remaining -= bytes;
 		}
 		if (state->nh_remaining == 0)
 			process_nh_hash_value(state, key);
 		src += bytes;
 		srclen -= bytes;
 	} while (srclen);
 }

 int crypto_nhpoly1305_setkey(struct crypto_shash *tfm,
 			     const u8 *key, unsigned int keylen)
 {
 	struct nhpoly1305_key *ctx = crypto_shash_ctx(tfm);
 	int i;

 	if (keylen != NHPOLY1305_KEY_SIZE)
 		return -EINVAL;

 	poly1305_core_setkey(&ctx->poly_key, key);
 	key += POLY1305_BLOCK_SIZE;

 	for (i = 0; i < NH_KEY_WORDS; i++)
 		ctx->nh_key[i] = get_unaligned_le32(key + i * sizeof(u32));

 	return 0;
 }
 EXPORT_SYMBOL(crypto_nhpoly1305_setkey);

 int crypto_nhpoly1305_init(struct shash_desc *desc)
 {
 	struct nhpoly1305_state *state = shash_desc_ctx(desc);

 	poly1305_core_init(&state->poly_state);
 	state->buflen = 0;
 	state->nh_remaining = 0;
 	return 0;
 }
 EXPORT_SYMBOL(crypto_nhpoly1305_init);

 int crypto_nhpoly1305_update_helper(struct shash_desc *desc,
 				    const u8 *src, unsigned int srclen,
 				    nh_t nh_fn)
 {
 	struct nhpoly1305_state *state = shash_desc_ctx(desc);
 	const struct nhpoly1305_key *key = crypto_shash_ctx(desc->tfm);
 	unsigned int bytes;

 	if (state->buflen) {
 		bytes = min(srclen, (int)NH_MESSAGE_UNIT - state->buflen);
 		memcpy(&state->buffer[state->buflen], src, bytes);
 		state->buflen += bytes;
 		if (state->buflen < NH_MESSAGE_UNIT)
 			return 0;
 		nhpoly1305_units(state, key, state->buffer, NH_MESSAGE_UNIT,
 				 nh_fn);
 		state->buflen = 0;
 		src += bytes;
 		srclen -= bytes;
 	}

 	if (srclen >= NH_MESSAGE_UNIT) {
 		bytes = round_down(srclen, NH_MESSAGE_UNIT);
 		nhpoly1305_units(state, key, src, bytes, nh_fn);
 		src += bytes;
 		srclen -= bytes;
 	}

 	if (srclen) {
 		memcpy(state->buffer, src, srclen);
 		state->buflen = srclen;
 	}
 	return 0;
 }
 EXPORT_SYMBOL(crypto_nhpoly1305_update_helper);

 int crypto_nhpoly1305_update(struct shash_desc *desc,
 			     const u8 *src, unsigned int srclen)
 {
 	return crypto_nhpoly1305_update_helper(desc, src, srclen, nh_generic);
 }
 EXPORT_SYMBOL(crypto_nhpoly1305_update);

 int crypto_nhpoly1305_final_helper(struct shash_desc *desc, u8 *dst, nh_t nh_fn)
 {
 	struct nhpoly1305_state *state = shash_desc_ctx(desc);
 	const struct nhpoly1305_key *key = crypto_shash_ctx(desc->tfm);

 	if (state->buflen) {
 		memset(&state->buffer[state->buflen], 0,
 		       NH_MESSAGE_UNIT - state->buflen);
 		nhpoly1305_units(state, key, state->buffer, NH_MESSAGE_UNIT,
 				 nh_fn);
 	}

 	if (state->nh_remaining)
 		process_nh_hash_value(state, key);

 	poly1305_core_emit(&state->poly_state, NULL, dst);
 	return 0;
 }
 EXPORT_SYMBOL(crypto_nhpoly1305_final_helper);

 int crypto_nhpoly1305_final(struct shash_desc *desc, u8 *dst)
 {
 	return crypto_nhpoly1305_final_helper(desc, dst, nh_generic);
 }
 EXPORT_SYMBOL(crypto_nhpoly1305_final);

 static struct shash_alg nhpoly1305_alg = {
 	.base.cra_name		= "nhpoly1305",
 	.base.cra_driver_name	= "nhpoly1305-generic",
 	.base.cra_priority	= 100,
 	.base.cra_ctxsize	= sizeof(struct nhpoly1305_key),
 	.base.cra_module	= THIS_MODULE,
 	.digestsize		= POLY1305_DIGEST_SIZE,
 	.init			= crypto_nhpoly1305_init,
 	.update			= crypto_nhpoly1305_update,
 	.final			= crypto_nhpoly1305_final,
 	.setkey			= crypto_nhpoly1305_setkey,
 	.descsize		= sizeof(struct nhpoly1305_state),
 };

 static int __init nhpoly1305_mod_init(void)
 {
 	return crypto_register_shash(&nhpoly1305_alg);
 }

 static void __exit nhpoly1305_mod_exit(void)
 {
 	crypto_unregister_shash(&nhpoly1305_alg);
 }

 subsys_initcall(nhpoly1305_mod_init);
 module_exit(nhpoly1305_mod_exit);

 MODULE_DESCRIPTION("NHPoly1305 ε-almost-∆-universal hash function");
 MODULE_LICENSE("GPL v2");
 MODULE_AUTHOR("Eric Biggers <ebiggers@google.com>");
 MODULE_ALIAS_CRYPTO("nhpoly1305");
 MODULE_ALIAS_CRYPTO("nhpoly1305-generic");
	// SPDX-License-Identifier: GPL-2.0
	/*
	* NHPoly1305 - ε-almost-∆-universal hash function for Adiantum
	*
	* Copyright 2018 Google LLC
	*/

	/*
	* "NHPoly1305" is the main component of Adiantum hashing.
	* Specifically, it is the calculation
	*
	* H_L ← Poly1305_{K_L}(NH_{K_N}(pad_{128}(L)))
	*
	* from the procedure in section 6.4 of the Adiantum paper [1]. It is an
	* ε-almost-∆-universal (ε-∆U) hash function for equal-length inputs over
	* Z/(2^{128}Z), where the "∆" operation is addition. It hashes 1024-byte
	* chunks of the input with the NH hash function [2], reducing the input length
	* by 32x. The resulting NH digests are evaluated as a polynomial in
	* GF(2^{130}-5), like in the Poly1305 MAC [3]. Note that the polynomial
	* evaluation by itself would suffice to achieve the ε-∆U property; NH is used
	* for performance since it's over twice as fast as Poly1305.
	*
	* This is not a cryptographic hash function; do not use it as such!
	*
	* [1] Adiantum: length-preserving encryption for entry-level processors
	* (https://eprint.iacr.org/2018/720.pdf)
	* [2] UMAC: Fast and Secure Message Authentication
	* (https://fastcrypto.org/umac/umac_proc.pdf)
	* [3] The Poly1305-AES message-authentication code
	* (https://cr.yp.to/mac/poly1305-20050329.pdf)
	*/

	#include <asm/unaligned.h>
	#include <crypto/algapi.h>
	#include <crypto/internal/hash.h>
	#include <crypto/internal/poly1305.h>
	#include <crypto/nhpoly1305.h>
	#include <linux/crypto.h>
	#include <linux/kernel.h>
	#include <linux/module.h>

	static void nh_generic(const u32 key, const u8 message, size_t message_len,
	__le64 hash[NH_NUM_PASSES])
	{
	u64 sums[4] = { 0, 0, 0, 0 };

	BUILD_BUG_ON(NH_PAIR_STRIDE != 2);
	BUILD_BUG_ON(NH_NUM_PASSES != 4);

	while (message_len) {
	u32 m0 = get_unaligned_le32(message + 0);
	u32 m1 = get_unaligned_le32(message + 4);
	u32 m2 = get_unaligned_le32(message + 8);
	u32 m3 = get_unaligned_le32(message + 12);

	sums[0] += (u64)(u32)(m0 + key[ 0]) * (u32)(m2 + key[ 2]);
	sums[1] += (u64)(u32)(m0 + key[ 4]) * (u32)(m2 + key[ 6]);
	sums[2] += (u64)(u32)(m0 + key[ 8]) * (u32)(m2 + key[10]);
	sums[3] += (u64)(u32)(m0 + key[12]) * (u32)(m2 + key[14]);
	sums[0] += (u64)(u32)(m1 + key[ 1]) * (u32)(m3 + key[ 3]);
	sums[1] += (u64)(u32)(m1 + key[ 5]) * (u32)(m3 + key[ 7]);
	sums[2] += (u64)(u32)(m1 + key[ 9]) * (u32)(m3 + key[11]);
	sums[3] += (u64)(u32)(m1 + key[13]) * (u32)(m3 + key[15]);
	key += NH_MESSAGE_UNIT / sizeof(key[0]);
	message += NH_MESSAGE_UNIT;
	message_len -= NH_MESSAGE_UNIT;
	}

	hash[0] = cpu_to_le64(sums[0]);
	hash[1] = cpu_to_le64(sums[1]);
	hash[2] = cpu_to_le64(sums[2]);
	hash[3] = cpu_to_le64(sums[3]);
	}

	/* Pass the next NH hash value through Poly1305 */
	static void process_nh_hash_value(struct nhpoly1305_state *state,
	const struct nhpoly1305_key *key)
	{
	BUILD_BUG_ON(NH_HASH_BYTES % POLY1305_BLOCK_SIZE != 0);

	poly1305_core_blocks(&state->poly_state, &key->poly_key, state->nh_hash,
	NH_HASH_BYTES / POLY1305_BLOCK_SIZE, 1);
	}

	/*
	* Feed the next portion of the source data, as a whole number of 16-byte
	* "NH message units", through NH and Poly1305. Each NH hash is taken over
	* 1024 bytes, except possibly the final one which is taken over a multiple of
	* 16 bytes up to 1024. Also, in the case where data is passed in misaligned
	* chunks, we combine partial hashes; the end result is the same either way.
	*/
	static void nhpoly1305_units(struct nhpoly1305_state *state,
	const struct nhpoly1305_key *key,
	const u8 *src, unsigned int srclen, nh_t nh_fn)
	{
	do {
	unsigned int bytes;

	if (state->nh_remaining == 0) {
	/* Starting a new NH message */
	bytes = min_t(unsigned int, srclen, NH_MESSAGE_BYTES);
	nh_fn(key->nh_key, src, bytes, state->nh_hash);
	state->nh_remaining = NH_MESSAGE_BYTES - bytes;
	} else {
	/* Continuing a previous NH message */
	__le64 tmp_hash[NH_NUM_PASSES];
	unsigned int pos;
	int i;

	pos = NH_MESSAGE_BYTES - state->nh_remaining;
	bytes = min(srclen, state->nh_remaining);
	nh_fn(&key->nh_key[pos / 4], src, bytes, tmp_hash);
	for (i = 0; i < NH_NUM_PASSES; i++)
	le64_add_cpu(&state->nh_hash[i],
	le64_to_cpu(tmp_hash[i]));
	state->nh_remaining -= bytes;
	}
	if (state->nh_remaining == 0)
	process_nh_hash_value(state, key);
	src += bytes;
	srclen -= bytes;
	} while (srclen);
	}

	int crypto_nhpoly1305_setkey(struct crypto_shash *tfm,
	const u8 *key, unsigned int keylen)
	{
	struct nhpoly1305_key *ctx = crypto_shash_ctx(tfm);
	int i;

	if (keylen != NHPOLY1305_KEY_SIZE)
	return -EINVAL;

	poly1305_core_setkey(&ctx->poly_key, key);
	key += POLY1305_BLOCK_SIZE;

	for (i = 0; i < NH_KEY_WORDS; i++)
	ctx->nh_key[i] = get_unaligned_le32(key + i * sizeof(u32));

	return 0;
	}
	EXPORT_SYMBOL(crypto_nhpoly1305_setkey);

	int crypto_nhpoly1305_init(struct shash_desc *desc)
	{
	struct nhpoly1305_state *state = shash_desc_ctx(desc);

	poly1305_core_init(&state->poly_state);
	state->buflen = 0;
	state->nh_remaining = 0;
	return 0;
	}
	EXPORT_SYMBOL(crypto_nhpoly1305_init);

	int crypto_nhpoly1305_update_helper(struct shash_desc *desc,
	const u8 *src, unsigned int srclen,
	nh_t nh_fn)
	{
	struct nhpoly1305_state *state = shash_desc_ctx(desc);
	const struct nhpoly1305_key *key = crypto_shash_ctx(desc->tfm);
	unsigned int bytes;

	if (state->buflen) {
	bytes = min(srclen, (int)NH_MESSAGE_UNIT - state->buflen);
	memcpy(&state->buffer[state->buflen], src, bytes);
	state->buflen += bytes;
	if (state->buflen < NH_MESSAGE_UNIT)
	return 0;
	nhpoly1305_units(state, key, state->buffer, NH_MESSAGE_UNIT,
	nh_fn);
	state->buflen = 0;
	src += bytes;
	srclen -= bytes;
	}

	if (srclen >= NH_MESSAGE_UNIT) {
	bytes = round_down(srclen, NH_MESSAGE_UNIT);
	nhpoly1305_units(state, key, src, bytes, nh_fn);
	src += bytes;
	srclen -= bytes;
	}

	if (srclen) {
	memcpy(state->buffer, src, srclen);
	state->buflen = srclen;
	}
	return 0;
	}
	EXPORT_SYMBOL(crypto_nhpoly1305_update_helper);

	int crypto_nhpoly1305_update(struct shash_desc *desc,
	const u8 *src, unsigned int srclen)
	{
	return crypto_nhpoly1305_update_helper(desc, src, srclen, nh_generic);
	}
	EXPORT_SYMBOL(crypto_nhpoly1305_update);

	int crypto_nhpoly1305_final_helper(struct shash_desc desc, u8 dst, nh_t nh_fn)
	{
	struct nhpoly1305_state *state = shash_desc_ctx(desc);
	const struct nhpoly1305_key *key = crypto_shash_ctx(desc->tfm);

	if (state->buflen) {
	memset(&state->buffer[state->buflen], 0,
	NH_MESSAGE_UNIT - state->buflen);
	nhpoly1305_units(state, key, state->buffer, NH_MESSAGE_UNIT,
	nh_fn);
	}

	if (state->nh_remaining)
	process_nh_hash_value(state, key);

	poly1305_core_emit(&state->poly_state, NULL, dst);
	return 0;
	}
	EXPORT_SYMBOL(crypto_nhpoly1305_final_helper);

	int crypto_nhpoly1305_final(struct shash_desc desc, u8 dst)
	{
	return crypto_nhpoly1305_final_helper(desc, dst, nh_generic);
	}
	EXPORT_SYMBOL(crypto_nhpoly1305_final);

	static struct shash_alg nhpoly1305_alg = {
	.base.cra_name = "nhpoly1305",
	.base.cra_driver_name = "nhpoly1305-generic",
	.base.cra_priority = 100,
	.base.cra_ctxsize = sizeof(struct nhpoly1305_key),
	.base.cra_module = THIS_MODULE,
	.digestsize = POLY1305_DIGEST_SIZE,
	.init = crypto_nhpoly1305_init,
	.update = crypto_nhpoly1305_update,
	.final = crypto_nhpoly1305_final,
	.setkey = crypto_nhpoly1305_setkey,
	.descsize = sizeof(struct nhpoly1305_state),
	};

	static int __init nhpoly1305_mod_init(void)
	{
	return crypto_register_shash(&nhpoly1305_alg);
	}

	static void __exit nhpoly1305_mod_exit(void)
	{
	crypto_unregister_shash(&nhpoly1305_alg);
	}

	subsys_initcall(nhpoly1305_mod_init);
	module_exit(nhpoly1305_mod_exit);

	MODULE_DESCRIPTION("NHPoly1305 ε-almost-∆-universal hash function");
	MODULE_LICENSE("GPL v2");
	MODULE_AUTHOR("Eric Biggers <ebiggers@google.com>");
	MODULE_ALIAS_CRYPTO("nhpoly1305");
	MODULE_ALIAS_CRYPTO("nhpoly1305-generic");