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
12struct xxhash64_tfm_ctx {
13	u64 seed;
14};
15
16struct xxhash64_desc_ctx {
17	struct xxh64_state xxhstate;
18};
19
20static 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		return -EINVAL;
27	tctx->seed = get_unaligned_le64(key);
28	return 0;
29}
30
31static int xxhash64_init(struct shash_desc *desc)
32{
33	struct xxhash64_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm);
34	struct xxhash64_desc_ctx *dctx = shash_desc_ctx(desc);
35
36	xxh64_reset(&dctx->xxhstate, tctx->seed);
37
38	return 0;
39}
40
41static int xxhash64_update(struct shash_desc *desc, const u8 *data,
42			 unsigned int length)
43{
44	struct xxhash64_desc_ctx *dctx = shash_desc_ctx(desc);
45
46	xxh64_update(&dctx->xxhstate, data, length);
47
48	return 0;
49}
50
51static int xxhash64_final(struct shash_desc *desc, u8 *out)
52{
53	struct xxhash64_desc_ctx *dctx = shash_desc_ctx(desc);
54
55	put_unaligned_le64(xxh64_digest(&dctx->xxhstate), out);
56
57	return 0;
58}
59
60static int xxhash64_digest(struct shash_desc *desc, const u8 *data,
61			 unsigned int length, u8 *out)
62{
63	struct xxhash64_tfm_ctx *tctx = crypto_shash_ctx(desc->tfm);
64
65	put_unaligned_le64(xxh64(data, length, tctx->seed), out);
66
67	return 0;
68}
69
70static struct shash_alg alg = {
71	.digestsize	= XXHASH64_DIGEST_SIZE,
72	.setkey		= xxhash64_setkey,
73	.init		= xxhash64_init,
74	.update		= xxhash64_update,
75	.final		= xxhash64_final,
76	.digest		= xxhash64_digest,
77	.descsize	= sizeof(struct xxhash64_desc_ctx),
78	.base		= {
79		.cra_name	 = "xxhash64",
80		.cra_driver_name = "xxhash64-generic",
81		.cra_priority	 = 100,
82		.cra_flags	 = CRYPTO_ALG_OPTIONAL_KEY,
83		.cra_blocksize	 = XXHASH64_BLOCK_SIZE,
84		.cra_ctxsize	 = sizeof(struct xxhash64_tfm_ctx),
85		.cra_module	 = THIS_MODULE,
86	}
87};
88
89static int __init xxhash_mod_init(void)
90{
91	return crypto_register_shash(&alg);
92}
93
94static void __exit xxhash_mod_fini(void)
95{
96	crypto_unregister_shash(&alg);
97}
98
99subsys_initcall(xxhash_mod_init);
100module_exit(xxhash_mod_fini);
101
102MODULE_AUTHOR("Nikolay Borisov <nborisov@suse.com>");
103MODULE_DESCRIPTION("xxhash calculations wrapper for lib/xxhash.c");
104MODULE_LICENSE("GPL");
105MODULE_ALIAS_CRYPTO("xxhash64");
106MODULE_ALIAS_CRYPTO("xxhash64-generic");
107