[people/ms/linux.git] / crypto / shash.c

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Synchronous Cryptographic Hash operations.
 *
 * Copyright (c) 2008 Herbert Xu <herbert@gondor.apana.org.au>
 */

#include <crypto/scatterwalk.h>
#include <crypto/internal/hash.h>
#include <linux/err.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/seq_file.h>
#include <linux/cryptouser.h>
#include <net/netlink.h>
#include <linux/compiler.h>

#include "internal.h"

static const struct crypto_type crypto_shash_type;

static int shash_no_setkey(struct crypto_shash *tfm, const u8 *key,
			   unsigned int keylen)
{
	return -ENOSYS;
}

/*
 * Check whether an shash algorithm has a setkey function.
 *
 * For CFI compatibility, this must not be an inline function.  This is because
 * when CFI is enabled, modules won't get the same address for shash_no_setkey
 * (if it were exported, which inlining would require) as the core kernel will.
 */
bool crypto_shash_alg_has_setkey(struct shash_alg *alg)
{
	return alg->setkey != shash_no_setkey;
}
EXPORT_SYMBOL_GPL(crypto_shash_alg_has_setkey);

static int shash_setkey_unaligned(struct crypto_shash *tfm, const u8 *key,
				  unsigned int keylen)
{
	struct shash_alg *shash = crypto_shash_alg(tfm);
	unsigned long alignmask = crypto_shash_alignmask(tfm);
	unsigned long absize;
	u8 *buffer, *alignbuffer;
	int err;

	absize = keylen + (alignmask & ~(crypto_tfm_ctx_alignment() - 1));
	buffer = kmalloc(absize, GFP_ATOMIC);
	if (!buffer)
		return -ENOMEM;

	alignbuffer = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1);
	memcpy(alignbuffer, key, keylen);
	err = shash->setkey(tfm, alignbuffer, keylen);
	kfree_sensitive(buffer);
	return err;
}

static void shash_set_needkey(struct crypto_shash *tfm, struct shash_alg *alg)
{
	if (crypto_shash_alg_needs_key(alg))
		crypto_shash_set_flags(tfm, CRYPTO_TFM_NEED_KEY);
}

int crypto_shash_setkey(struct crypto_shash *tfm, const u8 *key,
			unsigned int keylen)
{
	struct shash_alg *shash = crypto_shash_alg(tfm);
	unsigned long alignmask = crypto_shash_alignmask(tfm);
	int err;

	if ((unsigned long)key & alignmask)
		err = shash_setkey_unaligned(tfm, key, keylen);
	else
		err = shash->setkey(tfm, key, keylen);

	if (unlikely(err)) {
		shash_set_needkey(tfm, shash);
		return err;
	}

	crypto_shash_clear_flags(tfm, CRYPTO_TFM_NEED_KEY);
	return 0;
}
EXPORT_SYMBOL_GPL(crypto_shash_setkey);

static int shash_update_unaligned(struct shash_desc *desc, const u8 *data,
				  unsigned int len)
{
	struct crypto_shash *tfm = desc->tfm;
	struct shash_alg *shash = crypto_shash_alg(tfm);
	unsigned long alignmask = crypto_shash_alignmask(tfm);
	unsigned int unaligned_len = alignmask + 1 -
				     ((unsigned long)data & alignmask);
	/*
	 * We cannot count on __aligned() working for large values:
	 * https://patchwork.kernel.org/patch/9507697/
	 */
	u8 ubuf[MAX_ALGAPI_ALIGNMASK * 2];
	u8 *buf = PTR_ALIGN(&ubuf[0], alignmask + 1);
	int err;

	if (WARN_ON(buf + unaligned_len > ubuf + sizeof(ubuf)))
		return -EINVAL;

	if (unaligned_len > len)
		unaligned_len = len;

	memcpy(buf, data, unaligned_len);
	err = shash->update(desc, buf, unaligned_len);
	memset(buf, 0, unaligned_len);

	return err ?:
	       shash->update(desc, data + unaligned_len, len - unaligned_len);
}

int crypto_shash_update(struct shash_desc *desc, const u8 *data,
			unsigned int len)
{
	struct crypto_shash *tfm = desc->tfm;
	struct shash_alg *shash = crypto_shash_alg(tfm);
	unsigned long alignmask = crypto_shash_alignmask(tfm);

	if ((unsigned long)data & alignmask)
		return shash_update_unaligned(desc, data, len);

	return shash->update(desc, data, len);
}
EXPORT_SYMBOL_GPL(crypto_shash_update);

static int shash_final_unaligned(struct shash_desc *desc, u8 *out)
{
	struct crypto_shash *tfm = desc->tfm;
	unsigned long alignmask = crypto_shash_alignmask(tfm);
	struct shash_alg *shash = crypto_shash_alg(tfm);
	unsigned int ds = crypto_shash_digestsize(tfm);
	/*
	 * We cannot count on __aligned() working for large values:
	 * https://patchwork.kernel.org/patch/9507697/
	 */
	u8 ubuf[MAX_ALGAPI_ALIGNMASK + HASH_MAX_DIGESTSIZE];
	u8 *buf = PTR_ALIGN(&ubuf[0], alignmask + 1);
	int err;

	if (WARN_ON(buf + ds > ubuf + sizeof(ubuf)))
		return -EINVAL;

	err = shash->final(desc, buf);
	if (err)
		goto out;

	memcpy(out, buf, ds);

out:
	memset(buf, 0, ds);
	return err;
}

int crypto_shash_final(struct shash_desc *desc, u8 *out)
{
	struct crypto_shash *tfm = desc->tfm;
	struct shash_alg *shash = crypto_shash_alg(tfm);
	unsigned long alignmask = crypto_shash_alignmask(tfm);

	if ((unsigned long)out & alignmask)
		return shash_final_unaligned(desc, out);

	return shash->final(desc, out);
}
EXPORT_SYMBOL_GPL(crypto_shash_final);

static int shash_finup_unaligned(struct shash_desc *desc, const u8 *data,
				 unsigned int len, u8 *out)
{
	return crypto_shash_update(desc, data, len) ?:
	       crypto_shash_final(desc, out);
}

int crypto_shash_finup(struct shash_desc *desc, const u8 *data,
		       unsigned int len, u8 *out)
{
	struct crypto_shash *tfm = desc->tfm;
	struct shash_alg *shash = crypto_shash_alg(tfm);
	unsigned long alignmask = crypto_shash_alignmask(tfm);

	if (((unsigned long)data | (unsigned long)out) & alignmask)
		return shash_finup_unaligned(desc, data, len, out);

	return shash->finup(desc, data, len, out);
}
EXPORT_SYMBOL_GPL(crypto_shash_finup);

static int shash_digest_unaligned(struct shash_desc *desc, const u8 *data,
				  unsigned int len, u8 *out)
{
	return crypto_shash_init(desc) ?:
	       crypto_shash_finup(desc, data, len, out);
}

int crypto_shash_digest(struct shash_desc *desc, const u8 *data,
			unsigned int len, u8 *out)
{
	struct crypto_shash *tfm = desc->tfm;
	struct shash_alg *shash = crypto_shash_alg(tfm);
	unsigned long alignmask = crypto_shash_alignmask(tfm);

	if (crypto_shash_get_flags(tfm) & CRYPTO_TFM_NEED_KEY)
		return -ENOKEY;

	if (((unsigned long)data | (unsigned long)out) & alignmask)
		return shash_digest_unaligned(desc, data, len, out);

	return shash->digest(desc, data, len, out);
}
EXPORT_SYMBOL_GPL(crypto_shash_digest);

int crypto_shash_tfm_digest(struct crypto_shash *tfm, const u8 *data,
			    unsigned int len, u8 *out)
{
	SHASH_DESC_ON_STACK(desc, tfm);
	int err;

	desc->tfm = tfm;

	err = crypto_shash_digest(desc, data, len, out);

	shash_desc_zero(desc);

	return err;
}
EXPORT_SYMBOL_GPL(crypto_shash_tfm_digest);

static int shash_default_export(struct shash_desc *desc, void *out)
{
	memcpy(out, shash_desc_ctx(desc), crypto_shash_descsize(desc->tfm));
	return 0;
}

static int shash_default_import(struct shash_desc *desc, const void *in)
{
	memcpy(shash_desc_ctx(desc), in, crypto_shash_descsize(desc->tfm));
	return 0;
}

static int shash_async_setkey(struct crypto_ahash *tfm, const u8 *key,
			      unsigned int keylen)
{
	struct crypto_shash **ctx = crypto_ahash_ctx(tfm);

	return crypto_shash_setkey(*ctx, key, keylen);
}

static int shash_async_init(struct ahash_request *req)
{
	struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
	struct shash_desc *desc = ahash_request_ctx(req);

	desc->tfm = *ctx;

	return crypto_shash_init(desc);
}

int shash_ahash_update(struct ahash_request *req, struct shash_desc *desc)
{
	struct crypto_hash_walk walk;
	int nbytes;

	for (nbytes = crypto_hash_walk_first(req, &walk); nbytes > 0;
	     nbytes = crypto_hash_walk_done(&walk, nbytes))
		nbytes = crypto_shash_update(desc, walk.data, nbytes);

	return nbytes;
}
EXPORT_SYMBOL_GPL(shash_ahash_update);

static int shash_async_update(struct ahash_request *req)
{
	return shash_ahash_update(req, ahash_request_ctx(req));
}

static int shash_async_final(struct ahash_request *req)
{
	return crypto_shash_final(ahash_request_ctx(req), req->result);
}

int shash_ahash_finup(struct ahash_request *req, struct shash_desc *desc)
{
	struct crypto_hash_walk walk;
	int nbytes;

	nbytes = crypto_hash_walk_first(req, &walk);
	if (!nbytes)
		return crypto_shash_final(desc, req->result);

	do {
		nbytes = crypto_hash_walk_last(&walk) ?
			 crypto_shash_finup(desc, walk.data, nbytes,
					    req->result) :
			 crypto_shash_update(desc, walk.data, nbytes);
		nbytes = crypto_hash_walk_done(&walk, nbytes);
	} while (nbytes > 0);

	return nbytes;
}
EXPORT_SYMBOL_GPL(shash_ahash_finup);

static int shash_async_finup(struct ahash_request *req)
{
	struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
	struct shash_desc *desc = ahash_request_ctx(req);

	desc->tfm = *ctx;

	return shash_ahash_finup(req, desc);
}

int shash_ahash_digest(struct ahash_request *req, struct shash_desc *desc)
{
	unsigned int nbytes = req->nbytes;
	struct scatterlist *sg;
	unsigned int offset;
	int err;

	if (nbytes &&
	    (sg = req->src, offset = sg->offset,
	     nbytes <= min(sg->length, ((unsigned int)(PAGE_SIZE)) - offset))) {
		void *data;

		data = kmap_atomic(sg_page(sg));
		err = crypto_shash_digest(desc, data + offset, nbytes,
					  req->result);
		kunmap_atomic(data);
	} else
		err = crypto_shash_init(desc) ?:
		      shash_ahash_finup(req, desc);

	return err;
}
EXPORT_SYMBOL_GPL(shash_ahash_digest);

static int shash_async_digest(struct ahash_request *req)
{
	struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
	struct shash_desc *desc = ahash_request_ctx(req);

	desc->tfm = *ctx;

	return shash_ahash_digest(req, desc);
}

static int shash_async_export(struct ahash_request *req, void *out)
{
	return crypto_shash_export(ahash_request_ctx(req), out);
}

static int shash_async_import(struct ahash_request *req, const void *in)
{
	struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
	struct shash_desc *desc = ahash_request_ctx(req);

	desc->tfm = *ctx;

	return crypto_shash_import(desc, in);
}

static void crypto_exit_shash_ops_async(struct crypto_tfm *tfm)
{
	struct crypto_shash **ctx = crypto_tfm_ctx(tfm);

	crypto_free_shash(*ctx);
}

int crypto_init_shash_ops_async(struct crypto_tfm *tfm)
{
	struct crypto_alg *calg = tfm->__crt_alg;
	struct shash_alg *alg = __crypto_shash_alg(calg);
	struct crypto_ahash *crt = __crypto_ahash_cast(tfm);
	struct crypto_shash **ctx = crypto_tfm_ctx(tfm);
	struct crypto_shash *shash;

	if (!crypto_mod_get(calg))
		return -EAGAIN;

	shash = crypto_create_tfm(calg, &crypto_shash_type);
	if (IS_ERR(shash)) {
		crypto_mod_put(calg);
		return PTR_ERR(shash);
	}

	*ctx = shash;
	tfm->exit = crypto_exit_shash_ops_async;

	crt->init = shash_async_init;
	crt->update = shash_async_update;
	crt->final = shash_async_final;
	crt->finup = shash_async_finup;
	crt->digest = shash_async_digest;
	if (crypto_shash_alg_has_setkey(alg))
		crt->setkey = shash_async_setkey;

	crypto_ahash_set_flags(crt, crypto_shash_get_flags(shash) &
				    CRYPTO_TFM_NEED_KEY);

	crt->export = shash_async_export;
	crt->import = shash_async_import;

	crt->reqsize = sizeof(struct shash_desc) + crypto_shash_descsize(shash);

	return 0;
}

static void crypto_shash_exit_tfm(struct crypto_tfm *tfm)
{
	struct crypto_shash *hash = __crypto_shash_cast(tfm);
	struct shash_alg *alg = crypto_shash_alg(hash);

	alg->exit_tfm(hash);
}

static int crypto_shash_init_tfm(struct crypto_tfm *tfm)
{
	struct crypto_shash *hash = __crypto_shash_cast(tfm);
	struct shash_alg *alg = crypto_shash_alg(hash);
	int err;

	hash->descsize = alg->descsize;

	shash_set_needkey(hash, alg);

	if (alg->exit_tfm)
		tfm->exit = crypto_shash_exit_tfm;

	if (!alg->init_tfm)
		return 0;

	err = alg->init_tfm(hash);
	if (err)
		return err;

	/* ->init_tfm() may have increased the descsize. */
	if (WARN_ON_ONCE(hash->descsize > HASH_MAX_DESCSIZE)) {
		if (alg->exit_tfm)
			alg->exit_tfm(hash);
		return -EINVAL;
	}

	return 0;
}

static void crypto_shash_free_instance(struct crypto_instance *inst)
{
	struct shash_instance *shash = shash_instance(inst);

	shash->free(shash);
}

#ifdef CONFIG_NET
static int crypto_shash_report(struct sk_buff *skb, struct crypto_alg *alg)
{
	struct crypto_report_hash rhash;
	struct shash_alg *salg = __crypto_shash_alg(alg);

	memset(&rhash, 0, sizeof(rhash));

	strscpy(rhash.type, "shash", sizeof(rhash.type));

	rhash.blocksize = alg->cra_blocksize;
	rhash.digestsize = salg->digestsize;

	return nla_put(skb, CRYPTOCFGA_REPORT_HASH, sizeof(rhash), &rhash);
}
#else
static int crypto_shash_report(struct sk_buff *skb, struct crypto_alg *alg)
{
	return -ENOSYS;
}
#endif

static void crypto_shash_show(struct seq_file *m, struct crypto_alg *alg)
	__maybe_unused;
static void crypto_shash_show(struct seq_file *m, struct crypto_alg *alg)
{
	struct shash_alg *salg = __crypto_shash_alg(alg);

	seq_printf(m, "type         : shash\n");
	seq_printf(m, "blocksize    : %u\n", alg->cra_blocksize);
	seq_printf(m, "digestsize   : %u\n", salg->digestsize);
}

static const struct crypto_type crypto_shash_type = {
	.extsize = crypto_alg_extsize,
	.init_tfm = crypto_shash_init_tfm,
	.free = crypto_shash_free_instance,
#ifdef CONFIG_PROC_FS
	.show = crypto_shash_show,
#endif
	.report = crypto_shash_report,
	.maskclear = ~CRYPTO_ALG_TYPE_MASK,
	.maskset = CRYPTO_ALG_TYPE_MASK,
	.type = CRYPTO_ALG_TYPE_SHASH,
	.tfmsize = offsetof(struct crypto_shash, base),
};

int crypto_grab_shash(struct crypto_shash_spawn *spawn,
		      struct crypto_instance *inst,
		      const char *name, u32 type, u32 mask)
{
	spawn->base.frontend = &crypto_shash_type;
	return crypto_grab_spawn(&spawn->base, inst, name, type, mask);
}
EXPORT_SYMBOL_GPL(crypto_grab_shash);

struct crypto_shash *crypto_alloc_shash(const char *alg_name, u32 type,
					u32 mask)
{
	return crypto_alloc_tfm(alg_name, &crypto_shash_type, type, mask);
}
EXPORT_SYMBOL_GPL(crypto_alloc_shash);

int crypto_has_shash(const char *alg_name, u32 type, u32 mask)
{
	return crypto_type_has_alg(alg_name, &crypto_shash_type, type, mask);
}
EXPORT_SYMBOL_GPL(crypto_has_shash);

static int shash_prepare_alg(struct shash_alg *alg)
{
	struct crypto_alg *base = &alg->base;

	if (alg->digestsize > HASH_MAX_DIGESTSIZE ||
	    alg->descsize > HASH_MAX_DESCSIZE ||
	    alg->statesize > HASH_MAX_STATESIZE)
		return -EINVAL;

	if ((alg->export && !alg->import) || (alg->import && !alg->export))
		return -EINVAL;

	base->cra_type = &crypto_shash_type;
	base->cra_flags &= ~CRYPTO_ALG_TYPE_MASK;
	base->cra_flags |= CRYPTO_ALG_TYPE_SHASH;

	if (!alg->finup)
		alg->finup = shash_finup_unaligned;
	if (!alg->digest)
		alg->digest = shash_digest_unaligned;
	if (!alg->export) {
		alg->export = shash_default_export;
		alg->import = shash_default_import;
		alg->statesize = alg->descsize;
	}
	if (!alg->setkey)
		alg->setkey = shash_no_setkey;

	return 0;
}

int crypto_register_shash(struct shash_alg *alg)
{
	struct crypto_alg *base = &alg->base;
	int err;

	err = shash_prepare_alg(alg);
	if (err)
		return err;

	return crypto_register_alg(base);
}
EXPORT_SYMBOL_GPL(crypto_register_shash);

void crypto_unregister_shash(struct shash_alg *alg)
{
	crypto_unregister_alg(&alg->base);
}
EXPORT_SYMBOL_GPL(crypto_unregister_shash);

int crypto_register_shashes(struct shash_alg *algs, int count)
{
	int i, ret;

	for (i = 0; i < count; i++) {
		ret = crypto_register_shash(&algs[i]);
		if (ret)
			goto err;
	}

	return 0;

err:
	for (--i; i >= 0; --i)
		crypto_unregister_shash(&algs[i]);

	return ret;
}
EXPORT_SYMBOL_GPL(crypto_register_shashes);

void crypto_unregister_shashes(struct shash_alg *algs, int count)
{
	int i;

	for (i = count - 1; i >= 0; --i)
		crypto_unregister_shash(&algs[i]);
}
EXPORT_SYMBOL_GPL(crypto_unregister_shashes);

int shash_register_instance(struct crypto_template *tmpl,
			    struct shash_instance *inst)
{
	int err;

	if (WARN_ON(!inst->free))
		return -EINVAL;

	err = shash_prepare_alg(&inst->alg);
	if (err)
		return err;

	return crypto_register_instance(tmpl, shash_crypto_instance(inst));
}
EXPORT_SYMBOL_GPL(shash_register_instance);

void shash_free_singlespawn_instance(struct shash_instance *inst)
{
	crypto_drop_spawn(shash_instance_ctx(inst));
	kfree(inst);
}
EXPORT_SYMBOL_GPL(shash_free_singlespawn_instance);

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Synchronous cryptographic hash type");
Commit	Line	Data
2874c5fd	1	// SPDX-License-Identifier: GPL-2.0-or-later
7b5a080b HX	2	/*
	3	* Synchronous Cryptographic Hash operations.
	4	*
	5	* Copyright (c) 2008 Herbert Xu <herbert@gondor.apana.org.au>
7b5a080b HX	6	*/
7b5a080b HX	7
3b2f6df0	8	#include <crypto/scatterwalk.h>
7b5a080b HX	9	#include <crypto/internal/hash.h>
	10	#include <linux/err.h>
	11	#include <linux/kernel.h>
	12	#include <linux/module.h>
	13	#include <linux/slab.h>
	14	#include <linux/seq_file.h>
f4d663ce SK	15	#include <linux/cryptouser.h>
f4d663ce SK	16	#include <net/netlink.h>
d8c34b94	17	#include <linux/compiler.h>
7b5a080b	18
3b2f6df0 HX	19	#include "internal.h"
3b2f6df0 HX	20
3f683d61 HX	21	static const struct crypto_type crypto_shash_type;
3f683d61 HX	22
22ca9f4a AB	23	static int shash_no_setkey(struct crypto_shash tfm, const u8 key,
22ca9f4a AB	24	unsigned int keylen)
57cfe44b HX	25	{
	26	return -ENOSYS;
	27	}
22ca9f4a AB	28
	29	/*
	30	* Check whether an shash algorithm has a setkey function.
	31	*
	32	* For CFI compatibility, this must not be an inline function. This is because
	33	* when CFI is enabled, modules won't get the same address for shash_no_setkey
	34	* (if it were exported, which inlining would require) as the core kernel will.
	35	*/
	36	bool crypto_shash_alg_has_setkey(struct shash_alg *alg)
	37	{
	38	return alg->setkey != shash_no_setkey;
	39	}
	40	EXPORT_SYMBOL_GPL(crypto_shash_alg_has_setkey);
57cfe44b	41
7b5a080b HX	42	static int shash_setkey_unaligned(struct crypto_shash tfm, const u8 key,
	43	unsigned int keylen)
	44	{
	45	struct shash_alg *shash = crypto_shash_alg(tfm);
	46	unsigned long alignmask = crypto_shash_alignmask(tfm);
	47	unsigned long absize;
	48	u8 buffer, alignbuffer;
	49	int err;
	50
18eb8ea6	51	absize = keylen + (alignmask & ~(crypto_tfm_ctx_alignment() - 1));
9039f3ef	52	buffer = kmalloc(absize, GFP_ATOMIC);
7b5a080b HX	53	if (!buffer)
	54	return -ENOMEM;
	55
	56	alignbuffer = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1);
	57	memcpy(alignbuffer, key, keylen);
	58	err = shash->setkey(tfm, alignbuffer, keylen);
453431a5	59	kfree_sensitive(buffer);
7b5a080b HX	60	return err;
	61	}
	62
ba7d7433 EB	63	static void shash_set_needkey(struct crypto_shash tfm, struct shash_alg alg)
ba7d7433 EB	64	{
c2881789	65	if (crypto_shash_alg_needs_key(alg))
ba7d7433 EB	66	crypto_shash_set_flags(tfm, CRYPTO_TFM_NEED_KEY);
	67	}
	68
7b5a080b HX	69	int crypto_shash_setkey(struct crypto_shash tfm, const u8 key,
	70	unsigned int keylen)
	71	{
	72	struct shash_alg *shash = crypto_shash_alg(tfm);
	73	unsigned long alignmask = crypto_shash_alignmask(tfm);
9fa68f62	74	int err;
7b5a080b HX	75
7b5a080b HX	76	if ((unsigned long)key & alignmask)
9fa68f62 EB	77	err = shash_setkey_unaligned(tfm, key, keylen);
	78	else
	79	err = shash->setkey(tfm, key, keylen);
	80
ba7d7433 EB	81	if (unlikely(err)) {
ba7d7433 EB	82	shash_set_needkey(tfm, shash);
9fa68f62	83	return err;
ba7d7433	84	}
7b5a080b	85
9fa68f62 EB	86	crypto_shash_clear_flags(tfm, CRYPTO_TFM_NEED_KEY);
9fa68f62 EB	87	return 0;
7b5a080b HX	88	}
	89	EXPORT_SYMBOL_GPL(crypto_shash_setkey);
	90
7b5a080b HX	91	static int shash_update_unaligned(struct shash_desc desc, const u8 data,
	92	unsigned int len)
	93	{
	94	struct crypto_shash *tfm = desc->tfm;
	95	struct shash_alg *shash = crypto_shash_alg(tfm);
	96	unsigned long alignmask = crypto_shash_alignmask(tfm);
	97	unsigned int unaligned_len = alignmask + 1 -
	98	((unsigned long)data & alignmask);
f3569fd6 KC	99	/*
	100	* We cannot count on __aligned() working for large values:
	101	* https://patchwork.kernel.org/patch/9507697/
	102	*/
	103	u8 ubuf[MAX_ALGAPI_ALIGNMASK * 2];
0e2d3a12	104	u8 *buf = PTR_ALIGN(&ubuf[0], alignmask + 1);
8c32c516	105	int err;
7b5a080b	106
f3569fd6 KC	107	if (WARN_ON(buf + unaligned_len > ubuf + sizeof(ubuf)))
	108	return -EINVAL;
	109
f4f68993 YS	110	if (unaligned_len > len)
	111	unaligned_len = len;
	112
7b5a080b	113	memcpy(buf, data, unaligned_len);
8c32c516 HX	114	err = shash->update(desc, buf, unaligned_len);
8c32c516 HX	115	memset(buf, 0, unaligned_len);
7b5a080b	116
8c32c516	117	return err ?:
7b5a080b HX	118	shash->update(desc, data + unaligned_len, len - unaligned_len);
	119	}
	120
	121	int crypto_shash_update(struct shash_desc desc, const u8 data,
	122	unsigned int len)
	123	{
	124	struct crypto_shash *tfm = desc->tfm;
	125	struct shash_alg *shash = crypto_shash_alg(tfm);
	126	unsigned long alignmask = crypto_shash_alignmask(tfm);
	127
	128	if ((unsigned long)data & alignmask)
	129	return shash_update_unaligned(desc, data, len);
	130
	131	return shash->update(desc, data, len);
	132	}
	133	EXPORT_SYMBOL_GPL(crypto_shash_update);
	134
	135	static int shash_final_unaligned(struct shash_desc desc, u8 out)
	136	{
	137	struct crypto_shash *tfm = desc->tfm;
	138	unsigned long alignmask = crypto_shash_alignmask(tfm);
	139	struct shash_alg *shash = crypto_shash_alg(tfm);
	140	unsigned int ds = crypto_shash_digestsize(tfm);
f3569fd6 KC	141	/*
	142	* We cannot count on __aligned() working for large values:
	143	* https://patchwork.kernel.org/patch/9507697/
	144	*/
	145	u8 ubuf[MAX_ALGAPI_ALIGNMASK + HASH_MAX_DIGESTSIZE];
0e2d3a12	146	u8 *buf = PTR_ALIGN(&ubuf[0], alignmask + 1);
7b5a080b HX	147	int err;
7b5a080b HX	148
f3569fd6 KC	149	if (WARN_ON(buf + ds > ubuf + sizeof(ubuf)))
	150	return -EINVAL;
	151
7b5a080b	152	err = shash->final(desc, buf);
8c32c516 HX	153	if (err)
	154	goto out;
	155
7b5a080b	156	memcpy(out, buf, ds);
8c32c516 HX	157
	158	out:
	159	memset(buf, 0, ds);
7b5a080b HX	160	return err;
	161	}
	162
	163	int crypto_shash_final(struct shash_desc desc, u8 out)
	164	{
	165	struct crypto_shash *tfm = desc->tfm;
	166	struct shash_alg *shash = crypto_shash_alg(tfm);
	167	unsigned long alignmask = crypto_shash_alignmask(tfm);
	168
	169	if ((unsigned long)out & alignmask)
	170	return shash_final_unaligned(desc, out);
	171
	172	return shash->final(desc, out);
	173	}
	174	EXPORT_SYMBOL_GPL(crypto_shash_final);
	175
	176	static int shash_finup_unaligned(struct shash_desc desc, const u8 data,
	177	unsigned int len, u8 *out)
	178	{
	179	return crypto_shash_update(desc, data, len) ?:
	180	crypto_shash_final(desc, out);
	181	}
	182
	183	int crypto_shash_finup(struct shash_desc desc, const u8 data,
	184	unsigned int len, u8 *out)
	185	{
	186	struct crypto_shash *tfm = desc->tfm;
	187	struct shash_alg *shash = crypto_shash_alg(tfm);
	188	unsigned long alignmask = crypto_shash_alignmask(tfm);
	189
8267adab	190	if (((unsigned long)data \| (unsigned long)out) & alignmask)
7b5a080b HX	191	return shash_finup_unaligned(desc, data, len, out);
	192
	193	return shash->finup(desc, data, len, out);
	194	}
	195	EXPORT_SYMBOL_GPL(crypto_shash_finup);
	196
	197	static int shash_digest_unaligned(struct shash_desc desc, const u8 data,
	198	unsigned int len, u8 *out)
	199	{
	200	return crypto_shash_init(desc) ?:
f88ad8de	201	crypto_shash_finup(desc, data, len, out);
7b5a080b HX	202	}
	203
	204	int crypto_shash_digest(struct shash_desc desc, const u8 data,
	205	unsigned int len, u8 *out)
	206	{
	207	struct crypto_shash *tfm = desc->tfm;
	208	struct shash_alg *shash = crypto_shash_alg(tfm);
	209	unsigned long alignmask = crypto_shash_alignmask(tfm);
	210
9fa68f62 EB	211	if (crypto_shash_get_flags(tfm) & CRYPTO_TFM_NEED_KEY)
	212	return -ENOKEY;
	213
8267adab	214	if (((unsigned long)data \| (unsigned long)out) & alignmask)
7b5a080b HX	215	return shash_digest_unaligned(desc, data, len, out);
	216
	217	return shash->digest(desc, data, len, out);
	218	}
	219	EXPORT_SYMBOL_GPL(crypto_shash_digest);
	220
822a98b8 EB	221	int crypto_shash_tfm_digest(struct crypto_shash tfm, const u8 data,
	222	unsigned int len, u8 *out)
	223	{
	224	SHASH_DESC_ON_STACK(desc, tfm);
	225	int err;
	226
	227	desc->tfm = tfm;
	228
	229	err = crypto_shash_digest(desc, data, len, out);
	230
	231	shash_desc_zero(desc);
	232
	233	return err;
	234	}
	235	EXPORT_SYMBOL_GPL(crypto_shash_tfm_digest);
	236
f592682f	237	static int shash_default_export(struct shash_desc desc, void out)
dec8b786	238	{
f592682f HX	239	memcpy(out, shash_desc_ctx(desc), crypto_shash_descsize(desc->tfm));
f592682f HX	240	return 0;
99d27e1c	241	}
dec8b786	242
f592682f	243	static int shash_default_import(struct shash_desc desc, const void in)
99d27e1c	244	{
f592682f HX	245	memcpy(shash_desc_ctx(desc), in, crypto_shash_descsize(desc->tfm));
f592682f HX	246	return 0;
dec8b786	247	}
dec8b786	248
3b2f6df0 HX	249	static int shash_async_setkey(struct crypto_ahash tfm, const u8 key,
	250	unsigned int keylen)
	251	{
	252	struct crypto_shash **ctx = crypto_ahash_ctx(tfm);
	253
	254	return crypto_shash_setkey(*ctx, key, keylen);
	255	}
	256
	257	static int shash_async_init(struct ahash_request *req)
	258	{
	259	struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
	260	struct shash_desc *desc = ahash_request_ctx(req);
	261
	262	desc->tfm = *ctx;
3b2f6df0 HX	263
	264	return crypto_shash_init(desc);
	265	}
	266
7eddf95e	267	int shash_ahash_update(struct ahash_request req, struct shash_desc desc)
3b2f6df0	268	{
3b2f6df0 HX	269	struct crypto_hash_walk walk;
	270	int nbytes;
	271
	272	for (nbytes = crypto_hash_walk_first(req, &walk); nbytes > 0;
	273	nbytes = crypto_hash_walk_done(&walk, nbytes))
	274	nbytes = crypto_shash_update(desc, walk.data, nbytes);
	275
	276	return nbytes;
	277	}
7eddf95e HX	278	EXPORT_SYMBOL_GPL(shash_ahash_update);
	279
	280	static int shash_async_update(struct ahash_request *req)
	281	{
	282	return shash_ahash_update(req, ahash_request_ctx(req));
	283	}
3b2f6df0 HX	284
	285	static int shash_async_final(struct ahash_request *req)
	286	{
	287	return crypto_shash_final(ahash_request_ctx(req), req->result);
	288	}
	289
66f6ce5e HX	290	int shash_ahash_finup(struct ahash_request req, struct shash_desc desc)
	291	{
	292	struct crypto_hash_walk walk;
	293	int nbytes;
	294
cbc86b91 HX	295	nbytes = crypto_hash_walk_first(req, &walk);
	296	if (!nbytes)
	297	return crypto_shash_final(desc, req->result);
	298
	299	do {
66f6ce5e HX	300	nbytes = crypto_hash_walk_last(&walk) ?
	301	crypto_shash_finup(desc, walk.data, nbytes,
	302	req->result) :
	303	crypto_shash_update(desc, walk.data, nbytes);
cbc86b91 HX	304	nbytes = crypto_hash_walk_done(&walk, nbytes);
cbc86b91 HX	305	} while (nbytes > 0);
66f6ce5e HX	306
	307	return nbytes;
	308	}
	309	EXPORT_SYMBOL_GPL(shash_ahash_finup);
	310
	311	static int shash_async_finup(struct ahash_request *req)
	312	{
	313	struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
	314	struct shash_desc *desc = ahash_request_ctx(req);
	315
	316	desc->tfm = *ctx;
66f6ce5e HX	317
	318	return shash_ahash_finup(req, desc);
	319	}
	320
7eddf95e	321	int shash_ahash_digest(struct ahash_request req, struct shash_desc desc)
3b2f6df0	322	{
3b2f6df0	323	unsigned int nbytes = req->nbytes;
b61907bb HX	324	struct scatterlist *sg;
b61907bb HX	325	unsigned int offset;
3b2f6df0 HX	326	int err;
3b2f6df0 HX	327
b61907bb HX	328	if (nbytes &&
b61907bb HX	329	(sg = req->src, offset = sg->offset,
67cb60e4	330	nbytes <= min(sg->length, ((unsigned int)(PAGE_SIZE)) - offset))) {
3b2f6df0 HX	331	void *data;
3b2f6df0 HX	332
f0dfc0b0	333	data = kmap_atomic(sg_page(sg));
3b2f6df0 HX	334	err = crypto_shash_digest(desc, data + offset, nbytes,
3b2f6df0 HX	335	req->result);
f0dfc0b0	336	kunmap_atomic(data);
7eddf95e HX	337	} else
7eddf95e HX	338	err = crypto_shash_init(desc) ?:
66f6ce5e	339	shash_ahash_finup(req, desc);
3b2f6df0	340
7eddf95e HX	341	return err;
	342	}
	343	EXPORT_SYMBOL_GPL(shash_ahash_digest);
3b2f6df0	344
7eddf95e HX	345	static int shash_async_digest(struct ahash_request *req)
	346	{
	347	struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
	348	struct shash_desc *desc = ahash_request_ctx(req);
3b2f6df0	349
7eddf95e	350	desc->tfm = *ctx;
3b2f6df0	351
7eddf95e	352	return shash_ahash_digest(req, desc);
3b2f6df0 HX	353	}
3b2f6df0 HX	354
66f6ce5e HX	355	static int shash_async_export(struct ahash_request req, void out)
	356	{
	357	return crypto_shash_export(ahash_request_ctx(req), out);
	358	}
	359
	360	static int shash_async_import(struct ahash_request req, const void in)
	361	{
90246e79 HX	362	struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
	363	struct shash_desc *desc = ahash_request_ctx(req);
	364
	365	desc->tfm = *ctx;
90246e79 HX	366
90246e79 HX	367	return crypto_shash_import(desc, in);
66f6ce5e HX	368	}
66f6ce5e HX	369
3b2f6df0 HX	370	static void crypto_exit_shash_ops_async(struct crypto_tfm *tfm)
	371	{
	372	struct crypto_shash **ctx = crypto_tfm_ctx(tfm);
	373
	374	crypto_free_shash(*ctx);
	375	}
	376
88056ec3	377	int crypto_init_shash_ops_async(struct crypto_tfm *tfm)
3b2f6df0 HX	378	{
3b2f6df0 HX	379	struct crypto_alg *calg = tfm->__crt_alg;
66f6ce5e	380	struct shash_alg *alg = __crypto_shash_alg(calg);
88056ec3	381	struct crypto_ahash *crt = __crypto_ahash_cast(tfm);
3b2f6df0 HX	382	struct crypto_shash **ctx = crypto_tfm_ctx(tfm);
	383	struct crypto_shash *shash;
	384
	385	if (!crypto_mod_get(calg))
	386	return -EAGAIN;
	387
3f683d61	388	shash = crypto_create_tfm(calg, &crypto_shash_type);
3b2f6df0 HX	389	if (IS_ERR(shash)) {
	390	crypto_mod_put(calg);
	391	return PTR_ERR(shash);
	392	}
	393
	394	*ctx = shash;
	395	tfm->exit = crypto_exit_shash_ops_async;
	396
	397	crt->init = shash_async_init;
	398	crt->update = shash_async_update;
66f6ce5e HX	399	crt->final = shash_async_final;
66f6ce5e HX	400	crt->finup = shash_async_finup;
3b2f6df0	401	crt->digest = shash_async_digest;
ba7d7433 EB	402	if (crypto_shash_alg_has_setkey(alg))
ba7d7433 EB	403	crt->setkey = shash_async_setkey;
00420a65	404
9fa68f62 EB	405	crypto_ahash_set_flags(crt, crypto_shash_get_flags(shash) &
9fa68f62 EB	406	CRYPTO_TFM_NEED_KEY);
66f6ce5e	407
2b091e32 EB	408	crt->export = shash_async_export;
2b091e32 EB	409	crt->import = shash_async_import;
3b2f6df0	410
3b2f6df0 HX	411	crt->reqsize = sizeof(struct shash_desc) + crypto_shash_descsize(shash);
	412
	413	return 0;
	414	}
	415
fbce6be5 HX	416	static void crypto_shash_exit_tfm(struct crypto_tfm *tfm)
	417	{
	418	struct crypto_shash *hash = __crypto_shash_cast(tfm);
	419	struct shash_alg *alg = crypto_shash_alg(hash);
	420
	421	alg->exit_tfm(hash);
	422	}
	423
2ca33da1	424	static int crypto_shash_init_tfm(struct crypto_tfm *tfm)
7b5a080b	425	{
113adefc	426	struct crypto_shash *hash = __crypto_shash_cast(tfm);
9fa68f62	427	struct shash_alg *alg = crypto_shash_alg(hash);
fbce6be5	428	int err;
9fa68f62 EB	429
	430	hash->descsize = alg->descsize;
	431
ba7d7433	432	shash_set_needkey(hash, alg);
113adefc	433
fbce6be5 HX	434	if (alg->exit_tfm)
	435	tfm->exit = crypto_shash_exit_tfm;
	436
	437	if (!alg->init_tfm)
	438	return 0;
	439
	440	err = alg->init_tfm(hash);
	441	if (err)
	442	return err;
	443
	444	/* ->init_tfm() may have increased the descsize. */
	445	if (WARN_ON_ONCE(hash->descsize > HASH_MAX_DESCSIZE)) {
	446	if (alg->exit_tfm)
	447	alg->exit_tfm(hash);
	448	return -EINVAL;
	449	}
	450
7b5a080b HX	451	return 0;
	452	}
	453
48fb3e57 EB	454	static void crypto_shash_free_instance(struct crypto_instance *inst)
	455	{
	456	struct shash_instance *shash = shash_instance(inst);
	457
48fb3e57 EB	458	shash->free(shash);
	459	}
	460
3acc8473	461	#ifdef CONFIG_NET
f4d663ce SK	462	static int crypto_shash_report(struct sk_buff skb, struct crypto_alg alg)
	463	{
	464	struct crypto_report_hash rhash;
	465	struct shash_alg *salg = __crypto_shash_alg(alg);
	466
37db69e0 EB	467	memset(&rhash, 0, sizeof(rhash));
	468
	469	strscpy(rhash.type, "shash", sizeof(rhash.type));
9a5467bf	470
f4d663ce SK	471	rhash.blocksize = alg->cra_blocksize;
	472	rhash.digestsize = salg->digestsize;
	473
37db69e0	474	return nla_put(skb, CRYPTOCFGA_REPORT_HASH, sizeof(rhash), &rhash);
f4d663ce	475	}
3acc8473 HX	476	#else
	477	static int crypto_shash_report(struct sk_buff skb, struct crypto_alg alg)
	478	{
	479	return -ENOSYS;
	480	}
	481	#endif
f4d663ce	482
7b5a080b	483	static void crypto_shash_show(struct seq_file m, struct crypto_alg alg)
d8c34b94	484	__maybe_unused;
7b5a080b HX	485	static void crypto_shash_show(struct seq_file m, struct crypto_alg alg)
	486	{
	487	struct shash_alg *salg = __crypto_shash_alg(alg);
	488
	489	seq_printf(m, "type : shash\n");
	490	seq_printf(m, "blocksize : %u\n", alg->cra_blocksize);
	491	seq_printf(m, "digestsize : %u\n", salg->digestsize);
7b5a080b HX	492	}
	493
	494	static const struct crypto_type crypto_shash_type = {
ac611680	495	.extsize = crypto_alg_extsize,
7b5a080b	496	.init_tfm = crypto_shash_init_tfm,
48fb3e57	497	.free = crypto_shash_free_instance,
7b5a080b HX	498	#ifdef CONFIG_PROC_FS
	499	.show = crypto_shash_show,
	500	#endif
f4d663ce	501	.report = crypto_shash_report,
7b5a080b HX	502	.maskclear = ~CRYPTO_ALG_TYPE_MASK,
	503	.maskset = CRYPTO_ALG_TYPE_MASK,
	504	.type = CRYPTO_ALG_TYPE_SHASH,
	505	.tfmsize = offsetof(struct crypto_shash, base),
	506	};
	507
fdfad1ff EB	508	int crypto_grab_shash(struct crypto_shash_spawn *spawn,
	509	struct crypto_instance *inst,
	510	const char *name, u32 type, u32 mask)
	511	{
	512	spawn->base.frontend = &crypto_shash_type;
	513	return crypto_grab_spawn(&spawn->base, inst, name, type, mask);
	514	}
	515	EXPORT_SYMBOL_GPL(crypto_grab_shash);
	516
7b5a080b HX	517	struct crypto_shash crypto_alloc_shash(const char alg_name, u32 type,
	518	u32 mask)
	519	{
3f683d61	520	return crypto_alloc_tfm(alg_name, &crypto_shash_type, type, mask);
7b5a080b HX	521	}
	522	EXPORT_SYMBOL_GPL(crypto_alloc_shash);
	523
85cc4243 HR	524	int crypto_has_shash(const char *alg_name, u32 type, u32 mask)
	525	{
	526	return crypto_type_has_alg(alg_name, &crypto_shash_type, type, mask);
	527	}
	528	EXPORT_SYMBOL_GPL(crypto_has_shash);
	529
619a6ebd	530	static int shash_prepare_alg(struct shash_alg *alg)
7b5a080b HX	531	{
	532	struct crypto_alg *base = &alg->base;
	533
b68a7ec1 KC	534	if (alg->digestsize > HASH_MAX_DIGESTSIZE \|\|
	535	alg->descsize > HASH_MAX_DESCSIZE \|\|
	536	alg->statesize > HASH_MAX_STATESIZE)
7b5a080b HX	537	return -EINVAL;
7b5a080b HX	538
41a2e94f EB	539	if ((alg->export && !alg->import) \|\| (alg->import && !alg->export))
	540	return -EINVAL;
	541
7b5a080b HX	542	base->cra_type = &crypto_shash_type;
	543	base->cra_flags &= ~CRYPTO_ALG_TYPE_MASK;
	544	base->cra_flags \|= CRYPTO_ALG_TYPE_SHASH;
99d27e1c	545
8267adab HX	546	if (!alg->finup)
	547	alg->finup = shash_finup_unaligned;
	548	if (!alg->digest)
	549	alg->digest = shash_digest_unaligned;
f592682f HX	550	if (!alg->export) {
	551	alg->export = shash_default_export;
	552	alg->import = shash_default_import;
	553	alg->statesize = alg->descsize;
	554	}
57cfe44b HX	555	if (!alg->setkey)
57cfe44b HX	556	alg->setkey = shash_no_setkey;
99d27e1c	557
619a6ebd HX	558	return 0;
	559	}
	560
	561	int crypto_register_shash(struct shash_alg *alg)
	562	{
	563	struct crypto_alg *base = &alg->base;
	564	int err;
	565
	566	err = shash_prepare_alg(alg);
	567	if (err)
	568	return err;
7b5a080b HX	569
	570	return crypto_register_alg(base);
	571	}
	572	EXPORT_SYMBOL_GPL(crypto_register_shash);
	573
c6d633a9	574	void crypto_unregister_shash(struct shash_alg *alg)
7b5a080b	575	{
c6d633a9	576	crypto_unregister_alg(&alg->base);
7b5a080b HX	577	}
	578	EXPORT_SYMBOL_GPL(crypto_unregister_shash);
	579
50fc3e8d JK	580	int crypto_register_shashes(struct shash_alg *algs, int count)
	581	{
	582	int i, ret;
	583
	584	for (i = 0; i < count; i++) {
	585	ret = crypto_register_shash(&algs[i]);
	586	if (ret)
	587	goto err;
	588	}
	589
	590	return 0;
	591
	592	err:
	593	for (--i; i >= 0; --i)
	594	crypto_unregister_shash(&algs[i]);
	595
	596	return ret;
	597	}
	598	EXPORT_SYMBOL_GPL(crypto_register_shashes);
	599
c6d633a9	600	void crypto_unregister_shashes(struct shash_alg *algs, int count)
50fc3e8d	601	{
c6d633a9	602	int i;
50fc3e8d	603
c6d633a9 EB	604	for (i = count - 1; i >= 0; --i)
c6d633a9 EB	605	crypto_unregister_shash(&algs[i]);
50fc3e8d JK	606	}
	607	EXPORT_SYMBOL_GPL(crypto_unregister_shashes);
	608
619a6ebd HX	609	int shash_register_instance(struct crypto_template *tmpl,
	610	struct shash_instance *inst)
	611	{
	612	int err;
	613
d4fdc2df EB	614	if (WARN_ON(!inst->free))
	615	return -EINVAL;
	616
619a6ebd HX	617	err = shash_prepare_alg(&inst->alg);
	618	if (err)
	619	return err;
	620
	621	return crypto_register_instance(tmpl, shash_crypto_instance(inst));
	622	}
	623	EXPORT_SYMBOL_GPL(shash_register_instance);
	624
a39c66cc	625	void shash_free_singlespawn_instance(struct shash_instance *inst)
2e4fddd8	626	{
a39c66cc EB	627	crypto_drop_spawn(shash_instance_ctx(inst));
a39c66cc EB	628	kfree(inst);
2e4fddd8	629	}
a39c66cc	630	EXPORT_SYMBOL_GPL(shash_free_singlespawn_instance);
2e4fddd8	631
7b5a080b HX	632	MODULE_LICENSE("GPL");
7b5a080b HX	633	MODULE_DESCRIPTION("Synchronous cryptographic hash type");