[thirdparty/linux.git] / crypto / simd.c

// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Shared crypto simd helpers
 *
 * Copyright (c) 2012 Jussi Kivilinna <jussi.kivilinna@mbnet.fi>
 * Copyright (c) 2016 Herbert Xu <herbert@gondor.apana.org.au>
 * Copyright (c) 2019 Google LLC
 *
 * Based on aesni-intel_glue.c by:
 *  Copyright (C) 2008, Intel Corp.
 *    Author: Huang Ying <ying.huang@intel.com>
 */

/*
 * Shared crypto SIMD helpers.  These functions dynamically create and register
 * an skcipher or AEAD algorithm that wraps another, internal algorithm.  The
 * wrapper ensures that the internal algorithm is only executed in a context
 * where SIMD instructions are usable, i.e. where may_use_simd() returns true.
 * If SIMD is already usable, the wrapper directly calls the internal algorithm.
 * Otherwise it defers execution to a workqueue via cryptd.
 *
 * This is an alternative to the internal algorithm implementing a fallback for
 * the !may_use_simd() case itself.
 *
 * Note that the wrapper algorithm is asynchronous, i.e. it has the
 * CRYPTO_ALG_ASYNC flag set.  Therefore it won't be found by users who
 * explicitly allocate a synchronous algorithm.
 */

#include <crypto/cryptd.h>
#include <crypto/internal/aead.h>
#include <crypto/internal/simd.h>
#include <crypto/internal/skcipher.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/preempt.h>
#include <asm/simd.h>

/* skcipher support */

struct simd_skcipher_alg {
	const char *ialg_name;
	struct skcipher_alg alg;
};

struct simd_skcipher_ctx {
	struct cryptd_skcipher *cryptd_tfm;
};

static int simd_skcipher_setkey(struct crypto_skcipher *tfm, const u8 *key,
				unsigned int key_len)
{
	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);
	struct crypto_skcipher *child = &ctx->cryptd_tfm->base;

	crypto_skcipher_clear_flags(child, CRYPTO_TFM_REQ_MASK);
	crypto_skcipher_set_flags(child, crypto_skcipher_get_flags(tfm) &
					 CRYPTO_TFM_REQ_MASK);
	return crypto_skcipher_setkey(child, key, key_len);
}

static int simd_skcipher_encrypt(struct skcipher_request *req)
{
	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);
	struct skcipher_request *subreq;
	struct crypto_skcipher *child;

	subreq = skcipher_request_ctx(req);
	*subreq = *req;

	if (!crypto_simd_usable() ||
	    (in_atomic() && cryptd_skcipher_queued(ctx->cryptd_tfm)))
		child = &ctx->cryptd_tfm->base;
	else
		child = cryptd_skcipher_child(ctx->cryptd_tfm);

	skcipher_request_set_tfm(subreq, child);

	return crypto_skcipher_encrypt(subreq);
}

static int simd_skcipher_decrypt(struct skcipher_request *req)
{
	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);
	struct skcipher_request *subreq;
	struct crypto_skcipher *child;

	subreq = skcipher_request_ctx(req);
	*subreq = *req;

	if (!crypto_simd_usable() ||
	    (in_atomic() && cryptd_skcipher_queued(ctx->cryptd_tfm)))
		child = &ctx->cryptd_tfm->base;
	else
		child = cryptd_skcipher_child(ctx->cryptd_tfm);

	skcipher_request_set_tfm(subreq, child);

	return crypto_skcipher_decrypt(subreq);
}

static void simd_skcipher_exit(struct crypto_skcipher *tfm)
{
	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);

	cryptd_free_skcipher(ctx->cryptd_tfm);
}

static int simd_skcipher_init(struct crypto_skcipher *tfm)
{
	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);
	struct cryptd_skcipher *cryptd_tfm;
	struct simd_skcipher_alg *salg;
	struct skcipher_alg *alg;
	unsigned reqsize;

	alg = crypto_skcipher_alg(tfm);
	salg = container_of(alg, struct simd_skcipher_alg, alg);

	cryptd_tfm = cryptd_alloc_skcipher(salg->ialg_name,
					   CRYPTO_ALG_INTERNAL,
					   CRYPTO_ALG_INTERNAL);
	if (IS_ERR(cryptd_tfm))
		return PTR_ERR(cryptd_tfm);

	ctx->cryptd_tfm = cryptd_tfm;

	reqsize = crypto_skcipher_reqsize(cryptd_skcipher_child(cryptd_tfm));
	reqsize = max(reqsize, crypto_skcipher_reqsize(&cryptd_tfm->base));
	reqsize += sizeof(struct skcipher_request);

	crypto_skcipher_set_reqsize(tfm, reqsize);

	return 0;
}

struct simd_skcipher_alg *simd_skcipher_create_compat(struct skcipher_alg *ialg,
						      const char *algname,
						      const char *drvname,
						      const char *basename)
{
	struct simd_skcipher_alg *salg;
	struct skcipher_alg *alg;
	int err;

	salg = kzalloc(sizeof(*salg), GFP_KERNEL);
	if (!salg) {
		salg = ERR_PTR(-ENOMEM);
		goto out;
	}

	salg->ialg_name = basename;
	alg = &salg->alg;

	err = -ENAMETOOLONG;
	if (snprintf(alg->base.cra_name, CRYPTO_MAX_ALG_NAME, "%s", algname) >=
	    CRYPTO_MAX_ALG_NAME)
		goto out_free_salg;

	if (snprintf(alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, "%s",
		     drvname) >= CRYPTO_MAX_ALG_NAME)
		goto out_free_salg;

	alg->base.cra_flags = CRYPTO_ALG_ASYNC |
		(ialg->base.cra_flags & CRYPTO_ALG_INHERITED_FLAGS);
	alg->base.cra_priority = ialg->base.cra_priority;
	alg->base.cra_blocksize = ialg->base.cra_blocksize;
	alg->base.cra_alignmask = ialg->base.cra_alignmask;
	alg->base.cra_module = ialg->base.cra_module;
	alg->base.cra_ctxsize = sizeof(struct simd_skcipher_ctx);

	alg->ivsize = ialg->ivsize;
	alg->chunksize = ialg->chunksize;
	alg->min_keysize = ialg->min_keysize;
	alg->max_keysize = ialg->max_keysize;

	alg->init = simd_skcipher_init;
	alg->exit = simd_skcipher_exit;

	alg->setkey = simd_skcipher_setkey;
	alg->encrypt = simd_skcipher_encrypt;
	alg->decrypt = simd_skcipher_decrypt;

	err = crypto_register_skcipher(alg);
	if (err)
		goto out_free_salg;

out:
	return salg;

out_free_salg:
	kfree(salg);
	salg = ERR_PTR(err);
	goto out;
}
EXPORT_SYMBOL_GPL(simd_skcipher_create_compat);

void simd_skcipher_free(struct simd_skcipher_alg *salg)
{
	crypto_unregister_skcipher(&salg->alg);
	kfree(salg);
}
EXPORT_SYMBOL_GPL(simd_skcipher_free);

int simd_register_skciphers_compat(struct skcipher_alg *algs, int count,
				   struct simd_skcipher_alg **simd_algs)
{
	int err;
	int i;
	const char *algname;
	const char *drvname;
	const char *basename;
	struct simd_skcipher_alg *simd;

	err = crypto_register_skciphers(algs, count);
	if (err)
		return err;

	for (i = 0; i < count; i++) {
		WARN_ON(strncmp(algs[i].base.cra_name, "__", 2));
		WARN_ON(strncmp(algs[i].base.cra_driver_name, "__", 2));
		algname = algs[i].base.cra_name + 2;
		drvname = algs[i].base.cra_driver_name + 2;
		basename = algs[i].base.cra_driver_name;
		simd = simd_skcipher_create_compat(algs + i, algname, drvname, basename);
		err = PTR_ERR(simd);
		if (IS_ERR(simd))
			goto err_unregister;
		simd_algs[i] = simd;
	}
	return 0;

err_unregister:
	simd_unregister_skciphers(algs, count, simd_algs);
	return err;
}
EXPORT_SYMBOL_GPL(simd_register_skciphers_compat);

void simd_unregister_skciphers(struct skcipher_alg *algs, int count,
			       struct simd_skcipher_alg **simd_algs)
{
	int i;

	crypto_unregister_skciphers(algs, count);

	for (i = 0; i < count; i++) {
		if (simd_algs[i]) {
			simd_skcipher_free(simd_algs[i]);
			simd_algs[i] = NULL;
		}
	}
}
EXPORT_SYMBOL_GPL(simd_unregister_skciphers);

/* AEAD support */

struct simd_aead_alg {
	const char *ialg_name;
	struct aead_alg alg;
};

struct simd_aead_ctx {
	struct cryptd_aead *cryptd_tfm;
};

static int simd_aead_setkey(struct crypto_aead *tfm, const u8 *key,
				unsigned int key_len)
{
	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	struct crypto_aead *child = &ctx->cryptd_tfm->base;

	crypto_aead_clear_flags(child, CRYPTO_TFM_REQ_MASK);
	crypto_aead_set_flags(child, crypto_aead_get_flags(tfm) &
				     CRYPTO_TFM_REQ_MASK);
	return crypto_aead_setkey(child, key, key_len);
}

static int simd_aead_setauthsize(struct crypto_aead *tfm, unsigned int authsize)
{
	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	struct crypto_aead *child = &ctx->cryptd_tfm->base;

	return crypto_aead_setauthsize(child, authsize);
}

static int simd_aead_encrypt(struct aead_request *req)
{
	struct crypto_aead *tfm = crypto_aead_reqtfm(req);
	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	struct aead_request *subreq;
	struct crypto_aead *child;

	subreq = aead_request_ctx(req);
	*subreq = *req;

	if (!crypto_simd_usable() ||
	    (in_atomic() && cryptd_aead_queued(ctx->cryptd_tfm)))
		child = &ctx->cryptd_tfm->base;
	else
		child = cryptd_aead_child(ctx->cryptd_tfm);

	aead_request_set_tfm(subreq, child);

	return crypto_aead_encrypt(subreq);
}

static int simd_aead_decrypt(struct aead_request *req)
{
	struct crypto_aead *tfm = crypto_aead_reqtfm(req);
	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	struct aead_request *subreq;
	struct crypto_aead *child;

	subreq = aead_request_ctx(req);
	*subreq = *req;

	if (!crypto_simd_usable() ||
	    (in_atomic() && cryptd_aead_queued(ctx->cryptd_tfm)))
		child = &ctx->cryptd_tfm->base;
	else
		child = cryptd_aead_child(ctx->cryptd_tfm);

	aead_request_set_tfm(subreq, child);

	return crypto_aead_decrypt(subreq);
}

static void simd_aead_exit(struct crypto_aead *tfm)
{
	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);

	cryptd_free_aead(ctx->cryptd_tfm);
}

static int simd_aead_init(struct crypto_aead *tfm)
{
	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	struct cryptd_aead *cryptd_tfm;
	struct simd_aead_alg *salg;
	struct aead_alg *alg;
	unsigned reqsize;

	alg = crypto_aead_alg(tfm);
	salg = container_of(alg, struct simd_aead_alg, alg);

	cryptd_tfm = cryptd_alloc_aead(salg->ialg_name, CRYPTO_ALG_INTERNAL,
				       CRYPTO_ALG_INTERNAL);
	if (IS_ERR(cryptd_tfm))
		return PTR_ERR(cryptd_tfm);

	ctx->cryptd_tfm = cryptd_tfm;

	reqsize = crypto_aead_reqsize(cryptd_aead_child(cryptd_tfm));
	reqsize = max(reqsize, crypto_aead_reqsize(&cryptd_tfm->base));
	reqsize += sizeof(struct aead_request);

	crypto_aead_set_reqsize(tfm, reqsize);

	return 0;
}

static struct simd_aead_alg *simd_aead_create_compat(struct aead_alg *ialg,
						     const char *algname,
						     const char *drvname,
						     const char *basename)
{
	struct simd_aead_alg *salg;
	struct aead_alg *alg;
	int err;

	salg = kzalloc(sizeof(*salg), GFP_KERNEL);
	if (!salg) {
		salg = ERR_PTR(-ENOMEM);
		goto out;
	}

	salg->ialg_name = basename;
	alg = &salg->alg;

	err = -ENAMETOOLONG;
	if (snprintf(alg->base.cra_name, CRYPTO_MAX_ALG_NAME, "%s", algname) >=
	    CRYPTO_MAX_ALG_NAME)
		goto out_free_salg;

	if (snprintf(alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, "%s",
		     drvname) >= CRYPTO_MAX_ALG_NAME)
		goto out_free_salg;

	alg->base.cra_flags = CRYPTO_ALG_ASYNC |
		(ialg->base.cra_flags & CRYPTO_ALG_INHERITED_FLAGS);
	alg->base.cra_priority = ialg->base.cra_priority;
	alg->base.cra_blocksize = ialg->base.cra_blocksize;
	alg->base.cra_alignmask = ialg->base.cra_alignmask;
	alg->base.cra_module = ialg->base.cra_module;
	alg->base.cra_ctxsize = sizeof(struct simd_aead_ctx);

	alg->ivsize = ialg->ivsize;
	alg->maxauthsize = ialg->maxauthsize;
	alg->chunksize = ialg->chunksize;

	alg->init = simd_aead_init;
	alg->exit = simd_aead_exit;

	alg->setkey = simd_aead_setkey;
	alg->setauthsize = simd_aead_setauthsize;
	alg->encrypt = simd_aead_encrypt;
	alg->decrypt = simd_aead_decrypt;

	err = crypto_register_aead(alg);
	if (err)
		goto out_free_salg;

out:
	return salg;

out_free_salg:
	kfree(salg);
	salg = ERR_PTR(err);
	goto out;
}

static void simd_aead_free(struct simd_aead_alg *salg)
{
	crypto_unregister_aead(&salg->alg);
	kfree(salg);
}

int simd_register_aeads_compat(struct aead_alg *algs, int count,
			       struct simd_aead_alg **simd_algs)
{
	int err;
	int i;
	const char *algname;
	const char *drvname;
	const char *basename;
	struct simd_aead_alg *simd;

	err = crypto_register_aeads(algs, count);
	if (err)
		return err;

	for (i = 0; i < count; i++) {
		WARN_ON(strncmp(algs[i].base.cra_name, "__", 2));
		WARN_ON(strncmp(algs[i].base.cra_driver_name, "__", 2));
		algname = algs[i].base.cra_name + 2;
		drvname = algs[i].base.cra_driver_name + 2;
		basename = algs[i].base.cra_driver_name;
		simd = simd_aead_create_compat(algs + i, algname, drvname, basename);
		err = PTR_ERR(simd);
		if (IS_ERR(simd))
			goto err_unregister;
		simd_algs[i] = simd;
	}
	return 0;

err_unregister:
	simd_unregister_aeads(algs, count, simd_algs);
	return err;
}
EXPORT_SYMBOL_GPL(simd_register_aeads_compat);

void simd_unregister_aeads(struct aead_alg *algs, int count,
			   struct simd_aead_alg **simd_algs)
{
	int i;

	crypto_unregister_aeads(algs, count);

	for (i = 0; i < count; i++) {
		if (simd_algs[i]) {
			simd_aead_free(simd_algs[i]);
			simd_algs[i] = NULL;
		}
	}
}
EXPORT_SYMBOL_GPL(simd_unregister_aeads);

MODULE_DESCRIPTION("Shared crypto SIMD helpers");
MODULE_LICENSE("GPL");
Commit	Line	Data
1ccea77e	1	// SPDX-License-Identifier: GPL-2.0-or-later
266d0516 HX	2	/*
	3	* Shared crypto simd helpers
	4	*
	5	* Copyright (c) 2012 Jussi Kivilinna <jussi.kivilinna@mbnet.fi>
	6	* Copyright (c) 2016 Herbert Xu <herbert@gondor.apana.org.au>
1661131a	7	* Copyright (c) 2019 Google LLC
266d0516 HX	8	*
	9	* Based on aesni-intel_glue.c by:
	10	* Copyright (C) 2008, Intel Corp.
	11	* Author: Huang Ying <ying.huang@intel.com>
1661131a EB	12	*/
	13
	14	/*
	15	* Shared crypto SIMD helpers. These functions dynamically create and register
	16	* an skcipher or AEAD algorithm that wraps another, internal algorithm. The
	17	* wrapper ensures that the internal algorithm is only executed in a context
	18	* where SIMD instructions are usable, i.e. where may_use_simd() returns true.
	19	* If SIMD is already usable, the wrapper directly calls the internal algorithm.
	20	* Otherwise it defers execution to a workqueue via cryptd.
266d0516	21	*
1661131a EB	22	* This is an alternative to the internal algorithm implementing a fallback for
	23	* the !may_use_simd() case itself.
	24	*
	25	* Note that the wrapper algorithm is asynchronous, i.e. it has the
	26	* CRYPTO_ALG_ASYNC flag set. Therefore it won't be found by users who
	27	* explicitly allocate a synchronous algorithm.
266d0516 HX	28	*/
	29
	30	#include <crypto/cryptd.h>
1661131a	31	#include <crypto/internal/aead.h>
266d0516 HX	32	#include <crypto/internal/simd.h>
	33	#include <crypto/internal/skcipher.h>
	34	#include <linux/kernel.h>
	35	#include <linux/module.h>
	36	#include <linux/preempt.h>
	37	#include <asm/simd.h>
	38
1661131a EB	39	/* skcipher support */
1661131a EB	40
266d0516 HX	41	struct simd_skcipher_alg {
	42	const char *ialg_name;
	43	struct skcipher_alg alg;
	44	};
	45
	46	struct simd_skcipher_ctx {
	47	struct cryptd_skcipher *cryptd_tfm;
	48	};
	49
	50	static int simd_skcipher_setkey(struct crypto_skcipher tfm, const u8 key,
	51	unsigned int key_len)
	52	{
	53	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);
	54	struct crypto_skcipher *child = &ctx->cryptd_tfm->base;
266d0516 HX	55
	56	crypto_skcipher_clear_flags(child, CRYPTO_TFM_REQ_MASK);
	57	crypto_skcipher_set_flags(child, crypto_skcipher_get_flags(tfm) &
	58	CRYPTO_TFM_REQ_MASK);
af5034e8	59	return crypto_skcipher_setkey(child, key, key_len);
266d0516 HX	60	}
	61
	62	static int simd_skcipher_encrypt(struct skcipher_request *req)
	63	{
	64	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	65	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);
	66	struct skcipher_request *subreq;
	67	struct crypto_skcipher *child;
	68
	69	subreq = skcipher_request_ctx(req);
	70	subreq = req;
	71
8b8d91d4	72	if (!crypto_simd_usable() \|\|
266d0516 HX	73	(in_atomic() && cryptd_skcipher_queued(ctx->cryptd_tfm)))
	74	child = &ctx->cryptd_tfm->base;
	75	else
	76	child = cryptd_skcipher_child(ctx->cryptd_tfm);
	77
	78	skcipher_request_set_tfm(subreq, child);
	79
	80	return crypto_skcipher_encrypt(subreq);
	81	}
	82
	83	static int simd_skcipher_decrypt(struct skcipher_request *req)
	84	{
	85	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	86	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);
	87	struct skcipher_request *subreq;
	88	struct crypto_skcipher *child;
	89
	90	subreq = skcipher_request_ctx(req);
	91	subreq = req;
	92
8b8d91d4	93	if (!crypto_simd_usable() \|\|
266d0516 HX	94	(in_atomic() && cryptd_skcipher_queued(ctx->cryptd_tfm)))
	95	child = &ctx->cryptd_tfm->base;
	96	else
	97	child = cryptd_skcipher_child(ctx->cryptd_tfm);
	98
	99	skcipher_request_set_tfm(subreq, child);
	100
	101	return crypto_skcipher_decrypt(subreq);
	102	}
	103
	104	static void simd_skcipher_exit(struct crypto_skcipher *tfm)
	105	{
	106	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);
	107
	108	cryptd_free_skcipher(ctx->cryptd_tfm);
	109	}
	110
	111	static int simd_skcipher_init(struct crypto_skcipher *tfm)
	112	{
	113	struct simd_skcipher_ctx *ctx = crypto_skcipher_ctx(tfm);
	114	struct cryptd_skcipher *cryptd_tfm;
	115	struct simd_skcipher_alg *salg;
	116	struct skcipher_alg *alg;
	117	unsigned reqsize;
	118
	119	alg = crypto_skcipher_alg(tfm);
	120	salg = container_of(alg, struct simd_skcipher_alg, alg);
	121
	122	cryptd_tfm = cryptd_alloc_skcipher(salg->ialg_name,
	123	CRYPTO_ALG_INTERNAL,
	124	CRYPTO_ALG_INTERNAL);
	125	if (IS_ERR(cryptd_tfm))
	126	return PTR_ERR(cryptd_tfm);
	127
	128	ctx->cryptd_tfm = cryptd_tfm;
	129
508a1c4d AB	130	reqsize = crypto_skcipher_reqsize(cryptd_skcipher_child(cryptd_tfm));
	131	reqsize = max(reqsize, crypto_skcipher_reqsize(&cryptd_tfm->base));
	132	reqsize += sizeof(struct skcipher_request);
266d0516 HX	133
	134	crypto_skcipher_set_reqsize(tfm, reqsize);
	135
	136	return 0;
	137	}
	138
3c44d31c HX	139	struct simd_skcipher_alg simd_skcipher_create_compat(struct skcipher_alg ialg,
3c44d31c HX	140	const char *algname,
266d0516 HX	141	const char *drvname,
	142	const char *basename)
	143	{
	144	struct simd_skcipher_alg *salg;
266d0516 HX	145	struct skcipher_alg *alg;
	146	int err;
	147
266d0516 HX	148	salg = kzalloc(sizeof(*salg), GFP_KERNEL);
	149	if (!salg) {
	150	salg = ERR_PTR(-ENOMEM);
3c44d31c	151	goto out;
266d0516 HX	152	}
	153
	154	salg->ialg_name = basename;
	155	alg = &salg->alg;
	156
	157	err = -ENAMETOOLONG;
	158	if (snprintf(alg->base.cra_name, CRYPTO_MAX_ALG_NAME, "%s", algname) >=
	159	CRYPTO_MAX_ALG_NAME)
	160	goto out_free_salg;
	161
	162	if (snprintf(alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, "%s",
	163	drvname) >= CRYPTO_MAX_ALG_NAME)
	164	goto out_free_salg;
	165
7bcb2c99 EB	166	alg->base.cra_flags = CRYPTO_ALG_ASYNC \|
7bcb2c99 EB	167	(ialg->base.cra_flags & CRYPTO_ALG_INHERITED_FLAGS);
266d0516 HX	168	alg->base.cra_priority = ialg->base.cra_priority;
	169	alg->base.cra_blocksize = ialg->base.cra_blocksize;
	170	alg->base.cra_alignmask = ialg->base.cra_alignmask;
	171	alg->base.cra_module = ialg->base.cra_module;
	172	alg->base.cra_ctxsize = sizeof(struct simd_skcipher_ctx);
	173
	174	alg->ivsize = ialg->ivsize;
	175	alg->chunksize = ialg->chunksize;
	176	alg->min_keysize = ialg->min_keysize;
	177	alg->max_keysize = ialg->max_keysize;
	178
	179	alg->init = simd_skcipher_init;
	180	alg->exit = simd_skcipher_exit;
	181
	182	alg->setkey = simd_skcipher_setkey;
	183	alg->encrypt = simd_skcipher_encrypt;
	184	alg->decrypt = simd_skcipher_decrypt;
	185
	186	err = crypto_register_skcipher(alg);
	187	if (err)
	188	goto out_free_salg;
	189
3c44d31c	190	out:
266d0516 HX	191	return salg;
	192
	193	out_free_salg:
	194	kfree(salg);
	195	salg = ERR_PTR(err);
3c44d31c	196	goto out;
266d0516 HX	197	}
	198	EXPORT_SYMBOL_GPL(simd_skcipher_create_compat);
	199
266d0516 HX	200	void simd_skcipher_free(struct simd_skcipher_alg *salg)
	201	{
	202	crypto_unregister_skcipher(&salg->alg);
	203	kfree(salg);
	204	}
	205	EXPORT_SYMBOL_GPL(simd_skcipher_free);
	206
d14f0a1f EB	207	int simd_register_skciphers_compat(struct skcipher_alg *algs, int count,
	208	struct simd_skcipher_alg **simd_algs)
	209	{
	210	int err;
	211	int i;
	212	const char *algname;
	213	const char *drvname;
	214	const char *basename;
	215	struct simd_skcipher_alg *simd;
	216
	217	err = crypto_register_skciphers(algs, count);
	218	if (err)
	219	return err;
	220
	221	for (i = 0; i < count; i++) {
	222	WARN_ON(strncmp(algs[i].base.cra_name, "__", 2));
	223	WARN_ON(strncmp(algs[i].base.cra_driver_name, "__", 2));
	224	algname = algs[i].base.cra_name + 2;
	225	drvname = algs[i].base.cra_driver_name + 2;
	226	basename = algs[i].base.cra_driver_name;
3c44d31c	227	simd = simd_skcipher_create_compat(algs + i, algname, drvname, basename);
d14f0a1f EB	228	err = PTR_ERR(simd);
	229	if (IS_ERR(simd))
	230	goto err_unregister;
	231	simd_algs[i] = simd;
	232	}
	233	return 0;
	234
	235	err_unregister:
	236	simd_unregister_skciphers(algs, count, simd_algs);
	237	return err;
	238	}
	239	EXPORT_SYMBOL_GPL(simd_register_skciphers_compat);
	240
	241	void simd_unregister_skciphers(struct skcipher_alg *algs, int count,
	242	struct simd_skcipher_alg **simd_algs)
	243	{
	244	int i;
	245
	246	crypto_unregister_skciphers(algs, count);
	247
	248	for (i = 0; i < count; i++) {
	249	if (simd_algs[i]) {
	250	simd_skcipher_free(simd_algs[i]);
	251	simd_algs[i] = NULL;
	252	}
	253	}
	254	}
	255	EXPORT_SYMBOL_GPL(simd_unregister_skciphers);
	256
1661131a EB	257	/* AEAD support */
	258
	259	struct simd_aead_alg {
	260	const char *ialg_name;
	261	struct aead_alg alg;
	262	};
	263
	264	struct simd_aead_ctx {
	265	struct cryptd_aead *cryptd_tfm;
	266	};
	267
	268	static int simd_aead_setkey(struct crypto_aead tfm, const u8 key,
	269	unsigned int key_len)
	270	{
	271	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	272	struct crypto_aead *child = &ctx->cryptd_tfm->base;
1661131a EB	273
	274	crypto_aead_clear_flags(child, CRYPTO_TFM_REQ_MASK);
	275	crypto_aead_set_flags(child, crypto_aead_get_flags(tfm) &
	276	CRYPTO_TFM_REQ_MASK);
af5034e8	277	return crypto_aead_setkey(child, key, key_len);
1661131a EB	278	}
	279
	280	static int simd_aead_setauthsize(struct crypto_aead *tfm, unsigned int authsize)
	281	{
	282	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	283	struct crypto_aead *child = &ctx->cryptd_tfm->base;
	284
	285	return crypto_aead_setauthsize(child, authsize);
	286	}
	287
	288	static int simd_aead_encrypt(struct aead_request *req)
	289	{
	290	struct crypto_aead *tfm = crypto_aead_reqtfm(req);
	291	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	292	struct aead_request *subreq;
	293	struct crypto_aead *child;
	294
	295	subreq = aead_request_ctx(req);
	296	subreq = req;
	297
8b8d91d4	298	if (!crypto_simd_usable() \|\|
1661131a EB	299	(in_atomic() && cryptd_aead_queued(ctx->cryptd_tfm)))
	300	child = &ctx->cryptd_tfm->base;
	301	else
	302	child = cryptd_aead_child(ctx->cryptd_tfm);
	303
	304	aead_request_set_tfm(subreq, child);
	305
	306	return crypto_aead_encrypt(subreq);
	307	}
	308
	309	static int simd_aead_decrypt(struct aead_request *req)
	310	{
	311	struct crypto_aead *tfm = crypto_aead_reqtfm(req);
	312	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	313	struct aead_request *subreq;
	314	struct crypto_aead *child;
	315
	316	subreq = aead_request_ctx(req);
	317	subreq = req;
	318
8b8d91d4	319	if (!crypto_simd_usable() \|\|
1661131a EB	320	(in_atomic() && cryptd_aead_queued(ctx->cryptd_tfm)))
	321	child = &ctx->cryptd_tfm->base;
	322	else
	323	child = cryptd_aead_child(ctx->cryptd_tfm);
	324
	325	aead_request_set_tfm(subreq, child);
	326
	327	return crypto_aead_decrypt(subreq);
	328	}
	329
	330	static void simd_aead_exit(struct crypto_aead *tfm)
	331	{
	332	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	333
	334	cryptd_free_aead(ctx->cryptd_tfm);
	335	}
	336
	337	static int simd_aead_init(struct crypto_aead *tfm)
	338	{
	339	struct simd_aead_ctx *ctx = crypto_aead_ctx(tfm);
	340	struct cryptd_aead *cryptd_tfm;
	341	struct simd_aead_alg *salg;
	342	struct aead_alg *alg;
	343	unsigned reqsize;
	344
	345	alg = crypto_aead_alg(tfm);
	346	salg = container_of(alg, struct simd_aead_alg, alg);
	347
	348	cryptd_tfm = cryptd_alloc_aead(salg->ialg_name, CRYPTO_ALG_INTERNAL,
	349	CRYPTO_ALG_INTERNAL);
	350	if (IS_ERR(cryptd_tfm))
	351	return PTR_ERR(cryptd_tfm);
	352
	353	ctx->cryptd_tfm = cryptd_tfm;
	354
	355	reqsize = crypto_aead_reqsize(cryptd_aead_child(cryptd_tfm));
	356	reqsize = max(reqsize, crypto_aead_reqsize(&cryptd_tfm->base));
	357	reqsize += sizeof(struct aead_request);
	358
	359	crypto_aead_set_reqsize(tfm, reqsize);
	360
	361	return 0;
	362	}
	363
3c44d31c HX	364	static struct simd_aead_alg simd_aead_create_compat(struct aead_alg ialg,
	365	const char *algname,
	366	const char *drvname,
	367	const char *basename)
1661131a EB	368	{
1661131a EB	369	struct simd_aead_alg *salg;
1661131a EB	370	struct aead_alg *alg;
	371	int err;
	372
1661131a EB	373	salg = kzalloc(sizeof(*salg), GFP_KERNEL);
	374	if (!salg) {
	375	salg = ERR_PTR(-ENOMEM);
3c44d31c	376	goto out;
1661131a EB	377	}
	378
	379	salg->ialg_name = basename;
	380	alg = &salg->alg;
	381
	382	err = -ENAMETOOLONG;
	383	if (snprintf(alg->base.cra_name, CRYPTO_MAX_ALG_NAME, "%s", algname) >=
	384	CRYPTO_MAX_ALG_NAME)
	385	goto out_free_salg;
	386
	387	if (snprintf(alg->base.cra_driver_name, CRYPTO_MAX_ALG_NAME, "%s",
	388	drvname) >= CRYPTO_MAX_ALG_NAME)
	389	goto out_free_salg;
	390
7bcb2c99 EB	391	alg->base.cra_flags = CRYPTO_ALG_ASYNC \|
7bcb2c99 EB	392	(ialg->base.cra_flags & CRYPTO_ALG_INHERITED_FLAGS);
1661131a EB	393	alg->base.cra_priority = ialg->base.cra_priority;
	394	alg->base.cra_blocksize = ialg->base.cra_blocksize;
	395	alg->base.cra_alignmask = ialg->base.cra_alignmask;
	396	alg->base.cra_module = ialg->base.cra_module;
	397	alg->base.cra_ctxsize = sizeof(struct simd_aead_ctx);
	398
	399	alg->ivsize = ialg->ivsize;
	400	alg->maxauthsize = ialg->maxauthsize;
	401	alg->chunksize = ialg->chunksize;
	402
	403	alg->init = simd_aead_init;
	404	alg->exit = simd_aead_exit;
	405
	406	alg->setkey = simd_aead_setkey;
	407	alg->setauthsize = simd_aead_setauthsize;
	408	alg->encrypt = simd_aead_encrypt;
	409	alg->decrypt = simd_aead_decrypt;
	410
	411	err = crypto_register_aead(alg);
	412	if (err)
	413	goto out_free_salg;
	414
3c44d31c	415	out:
1661131a EB	416	return salg;
	417
	418	out_free_salg:
	419	kfree(salg);
	420	salg = ERR_PTR(err);
3c44d31c	421	goto out;
1661131a	422	}
1661131a	423
3c44d31c	424	static void simd_aead_free(struct simd_aead_alg *salg)
1661131a EB	425	{
	426	crypto_unregister_aead(&salg->alg);
	427	kfree(salg);
	428	}
1661131a EB	429
	430	int simd_register_aeads_compat(struct aead_alg *algs, int count,
	431	struct simd_aead_alg **simd_algs)
	432	{
	433	int err;
	434	int i;
	435	const char *algname;
	436	const char *drvname;
	437	const char *basename;
	438	struct simd_aead_alg *simd;
	439
	440	err = crypto_register_aeads(algs, count);
	441	if (err)
	442	return err;
	443
	444	for (i = 0; i < count; i++) {
	445	WARN_ON(strncmp(algs[i].base.cra_name, "__", 2));
	446	WARN_ON(strncmp(algs[i].base.cra_driver_name, "__", 2));
	447	algname = algs[i].base.cra_name + 2;
	448	drvname = algs[i].base.cra_driver_name + 2;
	449	basename = algs[i].base.cra_driver_name;
3c44d31c	450	simd = simd_aead_create_compat(algs + i, algname, drvname, basename);
1661131a EB	451	err = PTR_ERR(simd);
	452	if (IS_ERR(simd))
	453	goto err_unregister;
	454	simd_algs[i] = simd;
	455	}
	456	return 0;
	457
	458	err_unregister:
	459	simd_unregister_aeads(algs, count, simd_algs);
	460	return err;
	461	}
	462	EXPORT_SYMBOL_GPL(simd_register_aeads_compat);
	463
	464	void simd_unregister_aeads(struct aead_alg *algs, int count,
	465	struct simd_aead_alg **simd_algs)
	466	{
	467	int i;
	468
	469	crypto_unregister_aeads(algs, count);
	470
	471	for (i = 0; i < count; i++) {
	472	if (simd_algs[i]) {
	473	simd_aead_free(simd_algs[i]);
	474	simd_algs[i] = NULL;
	475	}
	476	}
	477	}
	478	EXPORT_SYMBOL_GPL(simd_unregister_aeads);
	479
7c699fe9	480	MODULE_DESCRIPTION("Shared crypto SIMD helpers");
266d0516	481	MODULE_LICENSE("GPL");