crypto/xxhash_generic.c

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