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"); 109