[thirdparty/linux.git] / crypto / xxhash_generic.c

// SPDX-License-Identifier: GPL-2.0

#include <crypto/internal/hash.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/xxhash.h>
#include <asm/unaligned.h>

#define XXHASH64_BLOCK_SIZE	32
#define XXHASH64_DIGEST_SIZE	8

struct xxhash64_tfm_ctx {
	u64 seed;
};

struct xxhash64_desc_ctx {
	struct xxh64_state xxhstate;
};

static int xxhash64_setkey(struct crypto_shash *tfm, const u8 *key,
			 unsigned int keylen)
{
	struct xxhash64_tfm_ctx *tctx = crypto_shash_ctx(tfm);

	if (keylen != sizeof(tctx->seed)) {
		crypto_shash_set_flags(tfm, CRYPTO_TFM_RES_BAD_KEY_LEN);
		return -EINVAL;
	}
	tctx->seed = get_unaligned_le64(key);
	return 0;
}

static int xxhash64_init(struct shash_desc *desc)
{
	struct xxhash64_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm);
	struct xxhash64_desc_ctx *dctx = shash_desc_ctx(desc);

	xxh64_reset(&dctx->xxhstate, tctx->seed);

	return 0;
}

static int xxhash64_update(struct shash_desc *desc, const u8 *data,
			 unsigned int length)
{
	struct xxhash64_desc_ctx *dctx = shash_desc_ctx(desc);

	xxh64_update(&dctx->xxhstate, data, length);

	return 0;
}

static int xxhash64_final(struct shash_desc *desc, u8 *out)
{
	struct xxhash64_desc_ctx *dctx = shash_desc_ctx(desc);

	put_unaligned_le64(xxh64_digest(&dctx->xxhstate), out);

	return 0;
}

static int xxhash64_digest(struct shash_desc *desc, const u8 *data,
			 unsigned int length, u8 *out)
{
	struct xxhash64_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm);

	put_unaligned_le64(xxh64(data, length, tctx->seed), out);

	return 0;
}

static struct shash_alg alg = {
	.digestsize	= XXHASH64_DIGEST_SIZE,
	.setkey		= xxhash64_setkey,
	.init		= xxhash64_init,
	.update		= xxhash64_update,
	.final		= xxhash64_final,
	.digest		= xxhash64_digest,
	.descsize	= sizeof(struct xxhash64_desc_ctx),
	.base		= {
		.cra_name	 = "xxhash64",
		.cra_driver_name = "xxhash64-generic",
		.cra_priority	 = 100,
		.cra_flags	 = CRYPTO_ALG_OPTIONAL_KEY,
		.cra_blocksize	 = XXHASH64_BLOCK_SIZE,
		.cra_ctxsize	 = sizeof(struct xxhash64_tfm_ctx),
		.cra_module	 = THIS_MODULE,
	}
};

static int __init xxhash_mod_init(void)
{
	return crypto_register_shash(&alg);
}

static void __exit xxhash_mod_fini(void)
{
	crypto_unregister_shash(&alg);
}

subsys_initcall(xxhash_mod_init);
module_exit(xxhash_mod_fini);

MODULE_AUTHOR("Nikolay Borisov <nborisov@suse.com>");
MODULE_DESCRIPTION("xxhash calculations wrapper for lib/xxhash.c");
MODULE_LICENSE("GPL");
MODULE_ALIAS_CRYPTO("xxhash64");
MODULE_ALIAS_CRYPTO("xxhash64-generic");
Commit	Line	Data
67882e76 NB	1	// SPDX-License-Identifier: GPL-2.0
	2
	3	#include <crypto/internal/hash.h>
	4	#include <linux/init.h>
	5	#include <linux/module.h>
	6	#include <linux/xxhash.h>
	7	#include <asm/unaligned.h>
	8
	9	#define XXHASH64_BLOCK_SIZE 32
	10	#define XXHASH64_DIGEST_SIZE 8
	11
	12	struct xxhash64_tfm_ctx {
	13	u64 seed;
	14	};
	15
	16	struct xxhash64_desc_ctx {
	17	struct xxh64_state xxhstate;
	18	};
	19
	20	static int xxhash64_setkey(struct crypto_shash tfm, const u8 key,
	21	unsigned int keylen)
	22	{
	23	struct xxhash64_tfm_ctx *tctx = crypto_shash_ctx(tfm);
	24
	25	if (keylen != sizeof(tctx->seed)) {
	26	crypto_shash_set_flags(tfm, CRYPTO_TFM_RES_BAD_KEY_LEN);
	27	return -EINVAL;
	28	}
	29	tctx->seed = get_unaligned_le64(key);
	30	return 0;
	31	}
	32
	33	static int xxhash64_init(struct shash_desc *desc)
	34	{
	35	struct xxhash64_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm);
	36	struct xxhash64_desc_ctx *dctx = shash_desc_ctx(desc);
	37
	38	xxh64_reset(&dctx->xxhstate, tctx->seed);
	39
	40	return 0;
	41	}
	42
	43	static int xxhash64_update(struct shash_desc desc, const u8 data,
	44	unsigned int length)
	45	{
	46	struct xxhash64_desc_ctx *dctx = shash_desc_ctx(desc);
	47
	48	xxh64_update(&dctx->xxhstate, data, length);
	49
	50	return 0;
	51	}
	52
	53	static int xxhash64_final(struct shash_desc desc, u8 out)
	54	{
	55	struct xxhash64_desc_ctx *dctx = shash_desc_ctx(desc);
	56
	57	put_unaligned_le64(xxh64_digest(&dctx->xxhstate), out);
	58
	59	return 0;
	60	}
	61
	62	static int xxhash64_digest(struct shash_desc desc, const u8 data,
	63	unsigned int length, u8 *out)
	64	{
65	struct xxhash64_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm);
66
67	put_unaligned_le64(xxh64(data, length, tctx->seed), out);
68
69	return 0;
70	}
71
72	static struct shash_alg alg = {
73	.digestsize = XXHASH64_DIGEST_SIZE,
74	.setkey = xxhash64_setkey,
75	.init = xxhash64_init,
76	.update = xxhash64_update,
77	.final = xxhash64_final,
78	.digest = xxhash64_digest,
79	.descsize = sizeof(struct xxhash64_desc_ctx),
80	.base = {
81	.cra_name = "xxhash64",
82	.cra_driver_name = "xxhash64-generic",
83	.cra_priority = 100,
84	.cra_flags = CRYPTO_ALG_OPTIONAL_KEY,
85	.cra_blocksize = XXHASH64_BLOCK_SIZE,
86	.cra_ctxsize = sizeof(struct xxhash64_tfm_ctx),
87	.cra_module = THIS_MODULE,
88	}
89	};
90
91	static int __init xxhash_mod_init(void)
92	{
93	return crypto_register_shash(&alg);
94	}
95
96	static void __exit xxhash_mod_fini(void)
97	{
98	crypto_unregister_shash(&alg);
99	}
100
101	subsys_initcall(xxhash_mod_init);
102	module_exit(xxhash_mod_fini);
103
104	MODULE_AUTHOR("Nikolay Borisov <nborisov@suse.com>");
105	MODULE_DESCRIPTION("xxhash calculations wrapper for lib/xxhash.c");
106	MODULE_LICENSE("GPL");
107	MODULE_ALIAS_CRYPTO("xxhash64");
108	MODULE_ALIAS_CRYPTO("xxhash64-generic");