1// SPDX-License-Identifier: GPL-2.0-or-later
2/*
3 * Crypto API support for BLAKE2b
4 *
5 * Copyright 2025 Google LLC
6 */
7#include <crypto/blake2b.h>
8#include <crypto/internal/hash.h>
9#include <linux/kernel.h>
10#include <linux/module.h>
11
12struct blake2b_tfm_ctx {
13 unsigned int keylen;
14 u8 key[BLAKE2B_KEY_SIZE];
15};
16
17static int crypto_blake2b_setkey(struct crypto_shash *tfm,
18 const u8 *key, unsigned int keylen)
19{
20 struct blake2b_tfm_ctx *tctx = crypto_shash_ctx(tfm);
21
22 if (keylen > BLAKE2B_KEY_SIZE)
23 return -EINVAL;
24 memcpy(tctx->key, key, keylen);
25 tctx->keylen = keylen;
26 return 0;
27}
28
29#define BLAKE2B_CTX(desc) ((struct blake2b_ctx *)shash_desc_ctx(desc))
30
31static int crypto_blake2b_init(struct shash_desc *desc)
32{
33 const struct blake2b_tfm_ctx *tctx = crypto_shash_ctx(tfm: desc->tfm);
34 unsigned int digestsize = crypto_shash_digestsize(tfm: desc->tfm);
35
36 blake2b_init_key(BLAKE2B_CTX(desc), outlen: digestsize,
37 key: tctx->key, keylen: tctx->keylen);
38 return 0;
39}
40
41static int crypto_blake2b_update(struct shash_desc *desc,
42 const u8 *data, unsigned int len)
43{
44 blake2b_update(BLAKE2B_CTX(desc), in: data, inlen: len);
45 return 0;
46}
47
48static int crypto_blake2b_final(struct shash_desc *desc, u8 *out)
49{
50 blake2b_final(BLAKE2B_CTX(desc), out);
51 return 0;
52}
53
54static int crypto_blake2b_digest(struct shash_desc *desc,
55 const u8 *data, unsigned int len, u8 *out)
56{
57 const struct blake2b_tfm_ctx *tctx = crypto_shash_ctx(tfm: desc->tfm);
58 unsigned int digestsize = crypto_shash_digestsize(tfm: desc->tfm);
59
60 blake2b(key: tctx->key, keylen: tctx->keylen, in: data, inlen: len, out, outlen: digestsize);
61 return 0;
62}
63
64#define BLAKE2B_ALG(name, digest_size) \
65 { \
66 .base.cra_name = name, \
67 .base.cra_driver_name = name "-lib", \
68 .base.cra_priority = 300, \
69 .base.cra_flags = CRYPTO_ALG_OPTIONAL_KEY, \
70 .base.cra_blocksize = BLAKE2B_BLOCK_SIZE, \
71 .base.cra_ctxsize = sizeof(struct blake2b_tfm_ctx), \
72 .base.cra_module = THIS_MODULE, \
73 .digestsize = digest_size, \
74 .setkey = crypto_blake2b_setkey, \
75 .init = crypto_blake2b_init, \
76 .update = crypto_blake2b_update, \
77 .final = crypto_blake2b_final, \
78 .digest = crypto_blake2b_digest, \
79 .descsize = sizeof(struct blake2b_ctx), \
80 }
81
82static struct shash_alg algs[] = {
83 BLAKE2B_ALG("blake2b-160", BLAKE2B_160_HASH_SIZE),
84 BLAKE2B_ALG("blake2b-256", BLAKE2B_256_HASH_SIZE),
85 BLAKE2B_ALG("blake2b-384", BLAKE2B_384_HASH_SIZE),
86 BLAKE2B_ALG("blake2b-512", BLAKE2B_512_HASH_SIZE),
87};
88
89static int __init crypto_blake2b_mod_init(void)
90{
91 return crypto_register_shashes(algs, ARRAY_SIZE(algs));
92}
93module_init(crypto_blake2b_mod_init);
94
95static void __exit crypto_blake2b_mod_exit(void)
96{
97 crypto_unregister_shashes(algs, ARRAY_SIZE(algs));
98}
99module_exit(crypto_blake2b_mod_exit);
100
101MODULE_LICENSE("GPL");
102MODULE_DESCRIPTION("Crypto API support for BLAKE2b");
103
104MODULE_ALIAS_CRYPTO("blake2b-160");
105MODULE_ALIAS_CRYPTO("blake2b-160-lib");
106MODULE_ALIAS_CRYPTO("blake2b-256");
107MODULE_ALIAS_CRYPTO("blake2b-256-lib");
108MODULE_ALIAS_CRYPTO("blake2b-384");
109MODULE_ALIAS_CRYPTO("blake2b-384-lib");
110MODULE_ALIAS_CRYPTO("blake2b-512");
111MODULE_ALIAS_CRYPTO("blake2b-512-lib");
112

source code of linux/crypto/blake2b.c