[mirror_ubuntu-bionic-kernel.git] / arch / x86 / crypto / serpent_avx2_glue.c

/*
 * Glue Code for x86_64/AVX2 assembler optimized version of Serpent
 *
 * Copyright © 2012-2013 Jussi Kivilinna <jussi.kivilinna@mbnet.fi>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 */

#include <linux/module.h>
#include <linux/types.h>
#include <linux/crypto.h>
#include <linux/err.h>
#include <crypto/ablk_helper.h>
#include <crypto/algapi.h>
#include <crypto/ctr.h>
#include <crypto/lrw.h>
#include <crypto/xts.h>
#include <crypto/serpent.h>
#include <asm/fpu/api.h>
#include <asm/crypto/serpent-avx.h>
#include <asm/crypto/glue_helper.h>

#define SERPENT_AVX2_PARALLEL_BLOCKS 16

/* 16-way AVX2 parallel cipher functions */
asmlinkage void serpent_ecb_enc_16way(struct serpent_ctx *ctx, u8 *dst,
				      const u8 *src);
asmlinkage void serpent_ecb_dec_16way(struct serpent_ctx *ctx, u8 *dst,
				      const u8 *src);
asmlinkage void serpent_cbc_dec_16way(void *ctx, u128 *dst, const u128 *src);

asmlinkage void serpent_ctr_16way(void *ctx, u128 *dst, const u128 *src,
				  le128 *iv);
asmlinkage void serpent_xts_enc_16way(struct serpent_ctx *ctx, u8 *dst,
				      const u8 *src, le128 *iv);
asmlinkage void serpent_xts_dec_16way(struct serpent_ctx *ctx, u8 *dst,
				      const u8 *src, le128 *iv);

static const struct common_glue_ctx serpent_enc = {
	.num_funcs = 3,
	.fpu_blocks_limit = 8,

	.funcs = { {
		.num_blocks = 16,
		.fn_u = { .ecb = GLUE_FUNC_CAST(serpent_ecb_enc_16way) }
	}, {
		.num_blocks = 8,
		.fn_u = { .ecb = GLUE_FUNC_CAST(serpent_ecb_enc_8way_avx) }
	}, {
		.num_blocks = 1,
		.fn_u = { .ecb = GLUE_FUNC_CAST(__serpent_encrypt) }
	} }
};

static const struct common_glue_ctx serpent_ctr = {
	.num_funcs = 3,
	.fpu_blocks_limit = 8,

	.funcs = { {
		.num_blocks = 16,
		.fn_u = { .ctr = GLUE_CTR_FUNC_CAST(serpent_ctr_16way) }
	},  {
		.num_blocks = 8,
		.fn_u = { .ctr = GLUE_CTR_FUNC_CAST(serpent_ctr_8way_avx) }
	}, {
		.num_blocks = 1,
		.fn_u = { .ctr = GLUE_CTR_FUNC_CAST(__serpent_crypt_ctr) }
	} }
};

static const struct common_glue_ctx serpent_enc_xts = {
	.num_funcs = 3,
	.fpu_blocks_limit = 8,

	.funcs = { {
		.num_blocks = 16,
		.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_enc_16way) }
	}, {
		.num_blocks = 8,
		.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_enc_8way_avx) }
	}, {
		.num_blocks = 1,
		.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_enc) }
	} }
};

static const struct common_glue_ctx serpent_dec = {
	.num_funcs = 3,
	.fpu_blocks_limit = 8,

	.funcs = { {
		.num_blocks = 16,
		.fn_u = { .ecb = GLUE_FUNC_CAST(serpent_ecb_dec_16way) }
	}, {
		.num_blocks = 8,
		.fn_u = { .ecb = GLUE_FUNC_CAST(serpent_ecb_dec_8way_avx) }
	}, {
		.num_blocks = 1,
		.fn_u = { .ecb = GLUE_FUNC_CAST(__serpent_decrypt) }
	} }
};

static const struct common_glue_ctx serpent_dec_cbc = {
	.num_funcs = 3,
	.fpu_blocks_limit = 8,

	.funcs = { {
		.num_blocks = 16,
		.fn_u = { .cbc = GLUE_CBC_FUNC_CAST(serpent_cbc_dec_16way) }
	}, {
		.num_blocks = 8,
		.fn_u = { .cbc = GLUE_CBC_FUNC_CAST(serpent_cbc_dec_8way_avx) }
	}, {
		.num_blocks = 1,
		.fn_u = { .cbc = GLUE_CBC_FUNC_CAST(__serpent_decrypt) }
	} }
};

static const struct common_glue_ctx serpent_dec_xts = {
	.num_funcs = 3,
	.fpu_blocks_limit = 8,

	.funcs = { {
		.num_blocks = 16,
		.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_dec_16way) }
	}, {
		.num_blocks = 8,
		.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_dec_8way_avx) }
	}, {
		.num_blocks = 1,
		.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_dec) }
	} }
};

static int ecb_encrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
		       struct scatterlist *src, unsigned int nbytes)
{
	return glue_ecb_crypt_128bit(&serpent_enc, desc, dst, src, nbytes);
}

static int ecb_decrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
		       struct scatterlist *src, unsigned int nbytes)
{
	return glue_ecb_crypt_128bit(&serpent_dec, desc, dst, src, nbytes);
}

static int cbc_encrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
		       struct scatterlist *src, unsigned int nbytes)
{
	return glue_cbc_encrypt_128bit(GLUE_FUNC_CAST(__serpent_encrypt), desc,
				       dst, src, nbytes);
}

static int cbc_decrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
		       struct scatterlist *src, unsigned int nbytes)
{
	return glue_cbc_decrypt_128bit(&serpent_dec_cbc, desc, dst, src,
				       nbytes);
}

static int ctr_crypt(struct blkcipher_desc *desc, struct scatterlist *dst,
		     struct scatterlist *src, unsigned int nbytes)
{
	return glue_ctr_crypt_128bit(&serpent_ctr, desc, dst, src, nbytes);
}

static inline bool serpent_fpu_begin(bool fpu_enabled, unsigned int nbytes)
{
	/* since reusing AVX functions, starts using FPU at 8 parallel blocks */
	return glue_fpu_begin(SERPENT_BLOCK_SIZE, 8, NULL, fpu_enabled, nbytes);
}

static inline void serpent_fpu_end(bool fpu_enabled)
{
	glue_fpu_end(fpu_enabled);
}

struct crypt_priv {
	struct serpent_ctx *ctx;
	bool fpu_enabled;
};

static void encrypt_callback(void *priv, u8 *srcdst, unsigned int nbytes)
{
	const unsigned int bsize = SERPENT_BLOCK_SIZE;
	struct crypt_priv *ctx = priv;
	int i;

	ctx->fpu_enabled = serpent_fpu_begin(ctx->fpu_enabled, nbytes);

	if (nbytes >= SERPENT_AVX2_PARALLEL_BLOCKS * bsize) {
		serpent_ecb_enc_16way(ctx->ctx, srcdst, srcdst);
		srcdst += bsize * SERPENT_AVX2_PARALLEL_BLOCKS;
		nbytes -= bsize * SERPENT_AVX2_PARALLEL_BLOCKS;
	}

	while (nbytes >= SERPENT_PARALLEL_BLOCKS * bsize) {
		serpent_ecb_enc_8way_avx(ctx->ctx, srcdst, srcdst);
		srcdst += bsize * SERPENT_PARALLEL_BLOCKS;
		nbytes -= bsize * SERPENT_PARALLEL_BLOCKS;
	}

	for (i = 0; i < nbytes / bsize; i++, srcdst += bsize)
		__serpent_encrypt(ctx->ctx, srcdst, srcdst);
}

static void decrypt_callback(void *priv, u8 *srcdst, unsigned int nbytes)
{
	const unsigned int bsize = SERPENT_BLOCK_SIZE;
	struct crypt_priv *ctx = priv;
	int i;

	ctx->fpu_enabled = serpent_fpu_begin(ctx->fpu_enabled, nbytes);

	if (nbytes >= SERPENT_AVX2_PARALLEL_BLOCKS * bsize) {
		serpent_ecb_dec_16way(ctx->ctx, srcdst, srcdst);
		srcdst += bsize * SERPENT_AVX2_PARALLEL_BLOCKS;
		nbytes -= bsize * SERPENT_AVX2_PARALLEL_BLOCKS;
	}

	while (nbytes >= SERPENT_PARALLEL_BLOCKS * bsize) {
		serpent_ecb_dec_8way_avx(ctx->ctx, srcdst, srcdst);
		srcdst += bsize * SERPENT_PARALLEL_BLOCKS;
		nbytes -= bsize * SERPENT_PARALLEL_BLOCKS;
	}

	for (i = 0; i < nbytes / bsize; i++, srcdst += bsize)
		__serpent_decrypt(ctx->ctx, srcdst, srcdst);
}

static int lrw_encrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
		       struct scatterlist *src, unsigned int nbytes)
{
	struct serpent_lrw_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
	be128 buf[SERPENT_AVX2_PARALLEL_BLOCKS];
	struct crypt_priv crypt_ctx = {
		.ctx = &ctx->serpent_ctx,
		.fpu_enabled = false,
	};
	struct lrw_crypt_req req = {
		.tbuf = buf,
		.tbuflen = sizeof(buf),

		.table_ctx = &ctx->lrw_table,
		.crypt_ctx = &crypt_ctx,
		.crypt_fn = encrypt_callback,
	};
	int ret;

	desc->flags &= ~CRYPTO_TFM_REQ_MAY_SLEEP;
	ret = lrw_crypt(desc, dst, src, nbytes, &req);
	serpent_fpu_end(crypt_ctx.fpu_enabled);

	return ret;
}

static int lrw_decrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
		       struct scatterlist *src, unsigned int nbytes)
{
	struct serpent_lrw_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
	be128 buf[SERPENT_AVX2_PARALLEL_BLOCKS];
	struct crypt_priv crypt_ctx = {
		.ctx = &ctx->serpent_ctx,
		.fpu_enabled = false,
	};
	struct lrw_crypt_req req = {
		.tbuf = buf,
		.tbuflen = sizeof(buf),

		.table_ctx = &ctx->lrw_table,
		.crypt_ctx = &crypt_ctx,
		.crypt_fn = decrypt_callback,
	};
	int ret;

	desc->flags &= ~CRYPTO_TFM_REQ_MAY_SLEEP;
	ret = lrw_crypt(desc, dst, src, nbytes, &req);
	serpent_fpu_end(crypt_ctx.fpu_enabled);

	return ret;
}

static int xts_encrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
		       struct scatterlist *src, unsigned int nbytes)
{
	struct serpent_xts_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);

	return glue_xts_crypt_128bit(&serpent_enc_xts, desc, dst, src, nbytes,
				     XTS_TWEAK_CAST(__serpent_encrypt),
				     &ctx->tweak_ctx, &ctx->crypt_ctx);
}

static int xts_decrypt(struct blkcipher_desc *desc, struct scatterlist *dst,
		       struct scatterlist *src, unsigned int nbytes)
{
	struct serpent_xts_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);

	return glue_xts_crypt_128bit(&serpent_dec_xts, desc, dst, src, nbytes,
				     XTS_TWEAK_CAST(__serpent_encrypt),
				     &ctx->tweak_ctx, &ctx->crypt_ctx);
}

static struct crypto_alg srp_algs[10] = { {
	.cra_name		= "__ecb-serpent-avx2",
	.cra_driver_name	= "__driver-ecb-serpent-avx2",
	.cra_priority		= 0,
	.cra_flags		= CRYPTO_ALG_TYPE_BLKCIPHER |
				  CRYPTO_ALG_INTERNAL,
	.cra_blocksize		= SERPENT_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct serpent_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_blkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[0].cra_list),
	.cra_u = {
		.blkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE,
			.max_keysize	= SERPENT_MAX_KEY_SIZE,
			.setkey		= serpent_setkey,
			.encrypt	= ecb_encrypt,
			.decrypt	= ecb_decrypt,
		},
	},
}, {
	.cra_name		= "__cbc-serpent-avx2",
	.cra_driver_name	= "__driver-cbc-serpent-avx2",
	.cra_priority		= 0,
	.cra_flags		= CRYPTO_ALG_TYPE_BLKCIPHER |
				  CRYPTO_ALG_INTERNAL,
	.cra_blocksize		= SERPENT_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct serpent_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_blkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[1].cra_list),
	.cra_u = {
		.blkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE,
			.max_keysize	= SERPENT_MAX_KEY_SIZE,
			.setkey		= serpent_setkey,
			.encrypt	= cbc_encrypt,
			.decrypt	= cbc_decrypt,
		},
	},
}, {
	.cra_name		= "__ctr-serpent-avx2",
	.cra_driver_name	= "__driver-ctr-serpent-avx2",
	.cra_priority		= 0,
	.cra_flags		= CRYPTO_ALG_TYPE_BLKCIPHER |
				  CRYPTO_ALG_INTERNAL,
	.cra_blocksize		= 1,
	.cra_ctxsize		= sizeof(struct serpent_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_blkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[2].cra_list),
	.cra_u = {
		.blkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE,
			.max_keysize	= SERPENT_MAX_KEY_SIZE,
			.ivsize		= SERPENT_BLOCK_SIZE,
			.setkey		= serpent_setkey,
			.encrypt	= ctr_crypt,
			.decrypt	= ctr_crypt,
		},
	},
}, {
	.cra_name		= "__lrw-serpent-avx2",
	.cra_driver_name	= "__driver-lrw-serpent-avx2",
	.cra_priority		= 0,
	.cra_flags		= CRYPTO_ALG_TYPE_BLKCIPHER |
				  CRYPTO_ALG_INTERNAL,
	.cra_blocksize		= SERPENT_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct serpent_lrw_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_blkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[3].cra_list),
	.cra_exit		= lrw_serpent_exit_tfm,
	.cra_u = {
		.blkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE +
					  SERPENT_BLOCK_SIZE,
			.max_keysize	= SERPENT_MAX_KEY_SIZE +
					  SERPENT_BLOCK_SIZE,
			.ivsize		= SERPENT_BLOCK_SIZE,
			.setkey		= lrw_serpent_setkey,
			.encrypt	= lrw_encrypt,
			.decrypt	= lrw_decrypt,
		},
	},
}, {
	.cra_name		= "__xts-serpent-avx2",
	.cra_driver_name	= "__driver-xts-serpent-avx2",
	.cra_priority		= 0,
	.cra_flags		= CRYPTO_ALG_TYPE_BLKCIPHER |
				  CRYPTO_ALG_INTERNAL,
	.cra_blocksize		= SERPENT_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct serpent_xts_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_blkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[4].cra_list),
	.cra_u = {
		.blkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE * 2,
			.max_keysize	= SERPENT_MAX_KEY_SIZE * 2,
			.ivsize		= SERPENT_BLOCK_SIZE,
			.setkey		= xts_serpent_setkey,
			.encrypt	= xts_encrypt,
			.decrypt	= xts_decrypt,
		},
	},
}, {
	.cra_name		= "ecb(serpent)",
	.cra_driver_name	= "ecb-serpent-avx2",
	.cra_priority		= 600,
	.cra_flags		= CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
	.cra_blocksize		= SERPENT_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct async_helper_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_ablkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[5].cra_list),
	.cra_init		= ablk_init,
	.cra_exit		= ablk_exit,
	.cra_u = {
		.ablkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE,
			.max_keysize	= SERPENT_MAX_KEY_SIZE,
			.setkey		= ablk_set_key,
			.encrypt	= ablk_encrypt,
			.decrypt	= ablk_decrypt,
		},
	},
}, {
	.cra_name		= "cbc(serpent)",
	.cra_driver_name	= "cbc-serpent-avx2",
	.cra_priority		= 600,
	.cra_flags		= CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
	.cra_blocksize		= SERPENT_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct async_helper_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_ablkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[6].cra_list),
	.cra_init		= ablk_init,
	.cra_exit		= ablk_exit,
	.cra_u = {
		.ablkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE,
			.max_keysize	= SERPENT_MAX_KEY_SIZE,
			.ivsize		= SERPENT_BLOCK_SIZE,
			.setkey		= ablk_set_key,
			.encrypt	= __ablk_encrypt,
			.decrypt	= ablk_decrypt,
		},
	},
}, {
	.cra_name		= "ctr(serpent)",
	.cra_driver_name	= "ctr-serpent-avx2",
	.cra_priority		= 600,
	.cra_flags		= CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
	.cra_blocksize		= 1,
	.cra_ctxsize		= sizeof(struct async_helper_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_ablkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[7].cra_list),
	.cra_init		= ablk_init,
	.cra_exit		= ablk_exit,
	.cra_u = {
		.ablkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE,
			.max_keysize	= SERPENT_MAX_KEY_SIZE,
			.ivsize		= SERPENT_BLOCK_SIZE,
			.setkey		= ablk_set_key,
			.encrypt	= ablk_encrypt,
			.decrypt	= ablk_encrypt,
			.geniv		= "chainiv",
		},
	},
}, {
	.cra_name		= "lrw(serpent)",
	.cra_driver_name	= "lrw-serpent-avx2",
	.cra_priority		= 600,
	.cra_flags		= CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
	.cra_blocksize		= SERPENT_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct async_helper_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_ablkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[8].cra_list),
	.cra_init		= ablk_init,
	.cra_exit		= ablk_exit,
	.cra_u = {
		.ablkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE +
					  SERPENT_BLOCK_SIZE,
			.max_keysize	= SERPENT_MAX_KEY_SIZE +
					  SERPENT_BLOCK_SIZE,
			.ivsize		= SERPENT_BLOCK_SIZE,
			.setkey		= ablk_set_key,
			.encrypt	= ablk_encrypt,
			.decrypt	= ablk_decrypt,
		},
	},
}, {
	.cra_name		= "xts(serpent)",
	.cra_driver_name	= "xts-serpent-avx2",
	.cra_priority		= 600,
	.cra_flags		= CRYPTO_ALG_TYPE_ABLKCIPHER | CRYPTO_ALG_ASYNC,
	.cra_blocksize		= SERPENT_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct async_helper_ctx),
	.cra_alignmask		= 0,
	.cra_type		= &crypto_ablkcipher_type,
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(srp_algs[9].cra_list),
	.cra_init		= ablk_init,
	.cra_exit		= ablk_exit,
	.cra_u = {
		.ablkcipher = {
			.min_keysize	= SERPENT_MIN_KEY_SIZE * 2,
			.max_keysize	= SERPENT_MAX_KEY_SIZE * 2,
			.ivsize		= SERPENT_BLOCK_SIZE,
			.setkey		= ablk_set_key,
			.encrypt	= ablk_encrypt,
			.decrypt	= ablk_decrypt,
		},
	},
} };

static int __init init(void)
{
	const char *feature_name;

	if (!cpu_has_avx2 || !cpu_has_osxsave) {
		pr_info("AVX2 instructions are not detected.\n");
		return -ENODEV;
	}
	if (!cpu_has_xfeatures(XFEATURE_MASK_SSE | XFEATURE_MASK_YMM,
				&feature_name)) {
		pr_info("CPU feature '%s' is not supported.\n", feature_name);
		return -ENODEV;
	}

	return crypto_register_algs(srp_algs, ARRAY_SIZE(srp_algs));
}

static void __exit fini(void)
{
	crypto_unregister_algs(srp_algs, ARRAY_SIZE(srp_algs));
}

module_init(init);
module_exit(fini);

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Serpent Cipher Algorithm, AVX2 optimized");
MODULE_ALIAS_CRYPTO("serpent");
MODULE_ALIAS_CRYPTO("serpent-asm");
Commit	Line	Data
56d76c96 JK	1	/*
	2	* Glue Code for x86_64/AVX2 assembler optimized version of Serpent
	3	*
	4	* Copyright © 2012-2013 Jussi Kivilinna <jussi.kivilinna@mbnet.fi>
	5	*
	6	* This program is free software; you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License as published by
	8	* the Free Software Foundation; either version 2 of the License, or
	9	* (at your option) any later version.
	10	*
	11	*/
	12
	13	#include <linux/module.h>
	14	#include <linux/types.h>
	15	#include <linux/crypto.h>
	16	#include <linux/err.h>
801201aa	17	#include <crypto/ablk_helper.h>
56d76c96 JK	18	#include <crypto/algapi.h>
	19	#include <crypto/ctr.h>
	20	#include <crypto/lrw.h>
	21	#include <crypto/xts.h>
	22	#include <crypto/serpent.h>
534ff06e	23	#include <asm/fpu/api.h>
56d76c96	24	#include <asm/crypto/serpent-avx.h>
56d76c96 JK	25	#include <asm/crypto/glue_helper.h>
	26
	27	#define SERPENT_AVX2_PARALLEL_BLOCKS 16
	28
	29	/* 16-way AVX2 parallel cipher functions */
	30	asmlinkage void serpent_ecb_enc_16way(struct serpent_ctx ctx, u8 dst,
	31	const u8 *src);
	32	asmlinkage void serpent_ecb_dec_16way(struct serpent_ctx ctx, u8 dst,
	33	const u8 *src);
	34	asmlinkage void serpent_cbc_dec_16way(void ctx, u128 dst, const u128 *src);
	35
	36	asmlinkage void serpent_ctr_16way(void ctx, u128 dst, const u128 *src,
	37	le128 *iv);
	38	asmlinkage void serpent_xts_enc_16way(struct serpent_ctx ctx, u8 dst,
	39	const u8 src, le128 iv);
	40	asmlinkage void serpent_xts_dec_16way(struct serpent_ctx ctx, u8 dst,
	41	const u8 src, le128 iv);
	42
	43	static const struct common_glue_ctx serpent_enc = {
	44	.num_funcs = 3,
	45	.fpu_blocks_limit = 8,
	46
	47	.funcs = { {
	48	.num_blocks = 16,
	49	.fn_u = { .ecb = GLUE_FUNC_CAST(serpent_ecb_enc_16way) }
	50	}, {
	51	.num_blocks = 8,
	52	.fn_u = { .ecb = GLUE_FUNC_CAST(serpent_ecb_enc_8way_avx) }
	53	}, {
	54	.num_blocks = 1,
	55	.fn_u = { .ecb = GLUE_FUNC_CAST(__serpent_encrypt) }
	56	} }
	57	};
	58
	59	static const struct common_glue_ctx serpent_ctr = {
	60	.num_funcs = 3,
	61	.fpu_blocks_limit = 8,
	62
	63	.funcs = { {
	64	.num_blocks = 16,
	65	.fn_u = { .ctr = GLUE_CTR_FUNC_CAST(serpent_ctr_16way) }
	66	}, {
	67	.num_blocks = 8,
	68	.fn_u = { .ctr = GLUE_CTR_FUNC_CAST(serpent_ctr_8way_avx) }
	69	}, {
	70	.num_blocks = 1,
	71	.fn_u = { .ctr = GLUE_CTR_FUNC_CAST(__serpent_crypt_ctr) }
	72	} }
	73	};
	74
	75	static const struct common_glue_ctx serpent_enc_xts = {
	76	.num_funcs = 3,
	77	.fpu_blocks_limit = 8,
	78
	79	.funcs = { {
	80	.num_blocks = 16,
	81	.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_enc_16way) }
	82	}, {
	83	.num_blocks = 8,
	84	.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_enc_8way_avx) }
	85	}, {
	86	.num_blocks = 1,
	87	.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_enc) }
	88	} }
89	};
90
91	static const struct common_glue_ctx serpent_dec = {
92	.num_funcs = 3,
93	.fpu_blocks_limit = 8,
94
95	.funcs = { {
96	.num_blocks = 16,
97	.fn_u = { .ecb = GLUE_FUNC_CAST(serpent_ecb_dec_16way) }
98	}, {
99	.num_blocks = 8,
100	.fn_u = { .ecb = GLUE_FUNC_CAST(serpent_ecb_dec_8way_avx) }
101	}, {
102	.num_blocks = 1,
103	.fn_u = { .ecb = GLUE_FUNC_CAST(__serpent_decrypt) }
104	} }
105	};
106
107	static const struct common_glue_ctx serpent_dec_cbc = {
108	.num_funcs = 3,
109	.fpu_blocks_limit = 8,
110
111	.funcs = { {
112	.num_blocks = 16,
113	.fn_u = { .cbc = GLUE_CBC_FUNC_CAST(serpent_cbc_dec_16way) }
114	}, {
115	.num_blocks = 8,
116	.fn_u = { .cbc = GLUE_CBC_FUNC_CAST(serpent_cbc_dec_8way_avx) }
117	}, {
118	.num_blocks = 1,
119	.fn_u = { .cbc = GLUE_CBC_FUNC_CAST(__serpent_decrypt) }
120	} }
121	};
122
123	static const struct common_glue_ctx serpent_dec_xts = {
124	.num_funcs = 3,
125	.fpu_blocks_limit = 8,
126
127	.funcs = { {
128	.num_blocks = 16,
129	.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_dec_16way) }
130	}, {
131	.num_blocks = 8,
132	.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_dec_8way_avx) }
133	}, {
134	.num_blocks = 1,
135	.fn_u = { .xts = GLUE_XTS_FUNC_CAST(serpent_xts_dec) }
136	} }
137	};
138
139	static int ecb_encrypt(struct blkcipher_desc desc, struct scatterlist dst,
140	struct scatterlist *src, unsigned int nbytes)
141	{
142	return glue_ecb_crypt_128bit(&serpent_enc, desc, dst, src, nbytes);
143	}
144
145	static int ecb_decrypt(struct blkcipher_desc desc, struct scatterlist dst,
146	struct scatterlist *src, unsigned int nbytes)
147	{
148	return glue_ecb_crypt_128bit(&serpent_dec, desc, dst, src, nbytes);
149	}
150
151	static int cbc_encrypt(struct blkcipher_desc desc, struct scatterlist dst,
152	struct scatterlist *src, unsigned int nbytes)
153	{
154	return glue_cbc_encrypt_128bit(GLUE_FUNC_CAST(__serpent_encrypt), desc,
155	dst, src, nbytes);
156	}
157
158	static int cbc_decrypt(struct blkcipher_desc desc, struct scatterlist dst,
159	struct scatterlist *src, unsigned int nbytes)
160	{
161	return glue_cbc_decrypt_128bit(&serpent_dec_cbc, desc, dst, src,
162	nbytes);
163	}
164
165	static int ctr_crypt(struct blkcipher_desc desc, struct scatterlist dst,
166	struct scatterlist *src, unsigned int nbytes)
167	{
168	return glue_ctr_crypt_128bit(&serpent_ctr, desc, dst, src, nbytes);
169	}
170
171	static inline bool serpent_fpu_begin(bool fpu_enabled, unsigned int nbytes)
172	{
173	/* since reusing AVX functions, starts using FPU at 8 parallel blocks */
174	return glue_fpu_begin(SERPENT_BLOCK_SIZE, 8, NULL, fpu_enabled, nbytes);
175	}
176
177	static inline void serpent_fpu_end(bool fpu_enabled)
178	{
179	glue_fpu_end(fpu_enabled);
180	}
181
182	struct crypt_priv {
183	struct serpent_ctx *ctx;
184	bool fpu_enabled;
185	};
186
187	static void encrypt_callback(void priv, u8 srcdst, unsigned int nbytes)
188	{
189	const unsigned int bsize = SERPENT_BLOCK_SIZE;
190	struct crypt_priv *ctx = priv;
191	int i;
192
193	ctx->fpu_enabled = serpent_fpu_begin(ctx->fpu_enabled, nbytes);
194
195	if (nbytes >= SERPENT_AVX2_PARALLEL_BLOCKS * bsize) {
196	serpent_ecb_enc_16way(ctx->ctx, srcdst, srcdst);
197	srcdst += bsize * SERPENT_AVX2_PARALLEL_BLOCKS;
198	nbytes -= bsize * SERPENT_AVX2_PARALLEL_BLOCKS;
199	}
200
201	while (nbytes >= SERPENT_PARALLEL_BLOCKS * bsize) {
202	serpent_ecb_enc_8way_avx(ctx->ctx, srcdst, srcdst);
203	srcdst += bsize * SERPENT_PARALLEL_BLOCKS;
204	nbytes -= bsize * SERPENT_PARALLEL_BLOCKS;
205	}
206
207	for (i = 0; i < nbytes / bsize; i++, srcdst += bsize)
208	__serpent_encrypt(ctx->ctx, srcdst, srcdst);
209	}
210
211	static void decrypt_callback(void priv, u8 srcdst, unsigned int nbytes)
212	{
213	const unsigned int bsize = SERPENT_BLOCK_SIZE;
214	struct crypt_priv *ctx = priv;
215	int i;
216
217	ctx->fpu_enabled = serpent_fpu_begin(ctx->fpu_enabled, nbytes);
218
219	if (nbytes >= SERPENT_AVX2_PARALLEL_BLOCKS * bsize) {
220	serpent_ecb_dec_16way(ctx->ctx, srcdst, srcdst);
221	srcdst += bsize * SERPENT_AVX2_PARALLEL_BLOCKS;
222	nbytes -= bsize * SERPENT_AVX2_PARALLEL_BLOCKS;
223	}
224
225	while (nbytes >= SERPENT_PARALLEL_BLOCKS * bsize) {
226	serpent_ecb_dec_8way_avx(ctx->ctx, srcdst, srcdst);
227	srcdst += bsize * SERPENT_PARALLEL_BLOCKS;
228	nbytes -= bsize * SERPENT_PARALLEL_BLOCKS;
229	}
230
231	for (i = 0; i < nbytes / bsize; i++, srcdst += bsize)
232	__serpent_decrypt(ctx->ctx, srcdst, srcdst);
233	}
234
235	static int lrw_encrypt(struct blkcipher_desc desc, struct scatterlist dst,
236	struct scatterlist *src, unsigned int nbytes)
237	{
238	struct serpent_lrw_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
239	be128 buf[SERPENT_AVX2_PARALLEL_BLOCKS];
240	struct crypt_priv crypt_ctx = {
241	.ctx = &ctx->serpent_ctx,
242	.fpu_enabled = false,
243	};
244	struct lrw_crypt_req req = {
245	.tbuf = buf,
246	.tbuflen = sizeof(buf),
247
248	.table_ctx = &ctx->lrw_table,
249	.crypt_ctx = &crypt_ctx,
250	.crypt_fn = encrypt_callback,
251	};
252	int ret;
253
254	desc->flags &= ~CRYPTO_TFM_REQ_MAY_SLEEP;
255	ret = lrw_crypt(desc, dst, src, nbytes, &req);
256	serpent_fpu_end(crypt_ctx.fpu_enabled);
257
258	return ret;
259	}
260
261	static int lrw_decrypt(struct blkcipher_desc desc, struct scatterlist dst,
262	struct scatterlist *src, unsigned int nbytes)
263	{
264	struct serpent_lrw_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
265	be128 buf[SERPENT_AVX2_PARALLEL_BLOCKS];
266	struct crypt_priv crypt_ctx = {
267	.ctx = &ctx->serpent_ctx,
268	.fpu_enabled = false,
269	};
270	struct lrw_crypt_req req = {
271	.tbuf = buf,
272	.tbuflen = sizeof(buf),
273
274	.table_ctx = &ctx->lrw_table,
275	.crypt_ctx = &crypt_ctx,
276	.crypt_fn = decrypt_callback,
277	};
278	int ret;
279
280	desc->flags &= ~CRYPTO_TFM_REQ_MAY_SLEEP;
281	ret = lrw_crypt(desc, dst, src, nbytes, &req);
282	serpent_fpu_end(crypt_ctx.fpu_enabled);
283
284	return ret;
285	}
286
287	static int xts_encrypt(struct blkcipher_desc desc, struct scatterlist dst,
288	struct scatterlist *src, unsigned int nbytes)
289	{
290	struct serpent_xts_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
291
292	return glue_xts_crypt_128bit(&serpent_enc_xts, desc, dst, src, nbytes,
293	XTS_TWEAK_CAST(__serpent_encrypt),
294	&ctx->tweak_ctx, &ctx->crypt_ctx);
295	}
296
297	static int xts_decrypt(struct blkcipher_desc desc, struct scatterlist dst,
298	struct scatterlist *src, unsigned int nbytes)
299	{
300	struct serpent_xts_ctx *ctx = crypto_blkcipher_ctx(desc->tfm);
301
302	return glue_xts_crypt_128bit(&serpent_dec_xts, desc, dst, src, nbytes,
303	XTS_TWEAK_CAST(__serpent_encrypt),
304	&ctx->tweak_ctx, &ctx->crypt_ctx);
305	}
306
307	static struct crypto_alg srp_algs[10] = { {
308	.cra_name = "__ecb-serpent-avx2",
309	.cra_driver_name = "__driver-ecb-serpent-avx2",
310	.cra_priority = 0,
f82419ac SM	311	.cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER \|
f82419ac SM	312	CRYPTO_ALG_INTERNAL,
56d76c96 JK	313	.cra_blocksize = SERPENT_BLOCK_SIZE,
	314	.cra_ctxsize = sizeof(struct serpent_ctx),
	315	.cra_alignmask = 0,
	316	.cra_type = &crypto_blkcipher_type,
	317	.cra_module = THIS_MODULE,
	318	.cra_list = LIST_HEAD_INIT(srp_algs[0].cra_list),
	319	.cra_u = {
	320	.blkcipher = {
	321	.min_keysize = SERPENT_MIN_KEY_SIZE,
	322	.max_keysize = SERPENT_MAX_KEY_SIZE,
	323	.setkey = serpent_setkey,
	324	.encrypt = ecb_encrypt,
	325	.decrypt = ecb_decrypt,
	326	},
	327	},
	328	}, {
	329	.cra_name = "__cbc-serpent-avx2",
	330	.cra_driver_name = "__driver-cbc-serpent-avx2",
	331	.cra_priority = 0,
f82419ac SM	332	.cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER \|
f82419ac SM	333	CRYPTO_ALG_INTERNAL,
56d76c96 JK	334	.cra_blocksize = SERPENT_BLOCK_SIZE,
	335	.cra_ctxsize = sizeof(struct serpent_ctx),
	336	.cra_alignmask = 0,
	337	.cra_type = &crypto_blkcipher_type,
	338	.cra_module = THIS_MODULE,
	339	.cra_list = LIST_HEAD_INIT(srp_algs[1].cra_list),
	340	.cra_u = {
	341	.blkcipher = {
	342	.min_keysize = SERPENT_MIN_KEY_SIZE,
	343	.max_keysize = SERPENT_MAX_KEY_SIZE,
	344	.setkey = serpent_setkey,
	345	.encrypt = cbc_encrypt,
	346	.decrypt = cbc_decrypt,
	347	},
	348	},
	349	}, {
	350	.cra_name = "__ctr-serpent-avx2",
	351	.cra_driver_name = "__driver-ctr-serpent-avx2",
	352	.cra_priority = 0,
f82419ac SM	353	.cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER \|
f82419ac SM	354	CRYPTO_ALG_INTERNAL,
56d76c96 JK	355	.cra_blocksize = 1,
	356	.cra_ctxsize = sizeof(struct serpent_ctx),
	357	.cra_alignmask = 0,
	358	.cra_type = &crypto_blkcipher_type,
	359	.cra_module = THIS_MODULE,
	360	.cra_list = LIST_HEAD_INIT(srp_algs[2].cra_list),
	361	.cra_u = {
	362	.blkcipher = {
	363	.min_keysize = SERPENT_MIN_KEY_SIZE,
	364	.max_keysize = SERPENT_MAX_KEY_SIZE,
	365	.ivsize = SERPENT_BLOCK_SIZE,
	366	.setkey = serpent_setkey,
	367	.encrypt = ctr_crypt,
	368	.decrypt = ctr_crypt,
	369	},
	370	},
	371	}, {
	372	.cra_name = "__lrw-serpent-avx2",
	373	.cra_driver_name = "__driver-lrw-serpent-avx2",
	374	.cra_priority = 0,
f82419ac SM	375	.cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER \|
f82419ac SM	376	CRYPTO_ALG_INTERNAL,
56d76c96 JK	377	.cra_blocksize = SERPENT_BLOCK_SIZE,
	378	.cra_ctxsize = sizeof(struct serpent_lrw_ctx),
	379	.cra_alignmask = 0,
	380	.cra_type = &crypto_blkcipher_type,
	381	.cra_module = THIS_MODULE,
	382	.cra_list = LIST_HEAD_INIT(srp_algs[3].cra_list),
	383	.cra_exit = lrw_serpent_exit_tfm,
	384	.cra_u = {
	385	.blkcipher = {
	386	.min_keysize = SERPENT_MIN_KEY_SIZE +
	387	SERPENT_BLOCK_SIZE,
	388	.max_keysize = SERPENT_MAX_KEY_SIZE +
	389	SERPENT_BLOCK_SIZE,
	390	.ivsize = SERPENT_BLOCK_SIZE,
	391	.setkey = lrw_serpent_setkey,
	392	.encrypt = lrw_encrypt,
	393	.decrypt = lrw_decrypt,
	394	},
	395	},
	396	}, {
	397	.cra_name = "__xts-serpent-avx2",
	398	.cra_driver_name = "__driver-xts-serpent-avx2",
	399	.cra_priority = 0,
f82419ac SM	400	.cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER \|
f82419ac SM	401	CRYPTO_ALG_INTERNAL,
56d76c96 JK	402	.cra_blocksize = SERPENT_BLOCK_SIZE,
	403	.cra_ctxsize = sizeof(struct serpent_xts_ctx),
	404	.cra_alignmask = 0,
	405	.cra_type = &crypto_blkcipher_type,
	406	.cra_module = THIS_MODULE,
	407	.cra_list = LIST_HEAD_INIT(srp_algs[4].cra_list),
	408	.cra_u = {
	409	.blkcipher = {
	410	.min_keysize = SERPENT_MIN_KEY_SIZE * 2,
	411	.max_keysize = SERPENT_MAX_KEY_SIZE * 2,
	412	.ivsize = SERPENT_BLOCK_SIZE,
	413	.setkey = xts_serpent_setkey,
	414	.encrypt = xts_encrypt,
	415	.decrypt = xts_decrypt,
	416	},
	417	},
	418	}, {
	419	.cra_name = "ecb(serpent)",
	420	.cra_driver_name = "ecb-serpent-avx2",
	421	.cra_priority = 600,
	422	.cra_flags = CRYPTO_ALG_TYPE_ABLKCIPHER \| CRYPTO_ALG_ASYNC,
	423	.cra_blocksize = SERPENT_BLOCK_SIZE,
	424	.cra_ctxsize = sizeof(struct async_helper_ctx),
	425	.cra_alignmask = 0,
	426	.cra_type = &crypto_ablkcipher_type,
	427	.cra_module = THIS_MODULE,
	428	.cra_list = LIST_HEAD_INIT(srp_algs[5].cra_list),
	429	.cra_init = ablk_init,
	430	.cra_exit = ablk_exit,
	431	.cra_u = {
	432	.ablkcipher = {
	433	.min_keysize = SERPENT_MIN_KEY_SIZE,
	434	.max_keysize = SERPENT_MAX_KEY_SIZE,
	435	.setkey = ablk_set_key,
	436	.encrypt = ablk_encrypt,
	437	.decrypt = ablk_decrypt,
	438	},
	439	},
	440	}, {
	441	.cra_name = "cbc(serpent)",
	442	.cra_driver_name = "cbc-serpent-avx2",
	443	.cra_priority = 600,
	444	.cra_flags = CRYPTO_ALG_TYPE_ABLKCIPHER \| CRYPTO_ALG_ASYNC,
	445	.cra_blocksize = SERPENT_BLOCK_SIZE,
	446	.cra_ctxsize = sizeof(struct async_helper_ctx),
	447	.cra_alignmask = 0,
	448	.cra_type = &crypto_ablkcipher_type,
	449	.cra_module = THIS_MODULE,
	450	.cra_list = LIST_HEAD_INIT(srp_algs[6].cra_list),
	451	.cra_init = ablk_init,
	452	.cra_exit = ablk_exit,
	453	.cra_u = {
	454	.ablkcipher = {
	455	.min_keysize = SERPENT_MIN_KEY_SIZE,
	456	.max_keysize = SERPENT_MAX_KEY_SIZE,
	457	.ivsize = SERPENT_BLOCK_SIZE,
	458	.setkey = ablk_set_key,
	459	.encrypt = __ablk_encrypt,
	460	.decrypt = ablk_decrypt,
	461	},
	462	},
	463	}, {
	464	.cra_name = "ctr(serpent)",
	465	.cra_driver_name = "ctr-serpent-avx2",
466	.cra_priority = 600,
467	.cra_flags = CRYPTO_ALG_TYPE_ABLKCIPHER \| CRYPTO_ALG_ASYNC,
468	.cra_blocksize = 1,
469	.cra_ctxsize = sizeof(struct async_helper_ctx),
470	.cra_alignmask = 0,
471	.cra_type = &crypto_ablkcipher_type,
472	.cra_module = THIS_MODULE,
473	.cra_list = LIST_HEAD_INIT(srp_algs[7].cra_list),
474	.cra_init = ablk_init,
475	.cra_exit = ablk_exit,
476	.cra_u = {
477	.ablkcipher = {
478	.min_keysize = SERPENT_MIN_KEY_SIZE,
479	.max_keysize = SERPENT_MAX_KEY_SIZE,
480	.ivsize = SERPENT_BLOCK_SIZE,
481	.setkey = ablk_set_key,
482	.encrypt = ablk_encrypt,
483	.decrypt = ablk_encrypt,
484	.geniv = "chainiv",
485	},
486	},
487	}, {
488	.cra_name = "lrw(serpent)",
489	.cra_driver_name = "lrw-serpent-avx2",
490	.cra_priority = 600,
491	.cra_flags = CRYPTO_ALG_TYPE_ABLKCIPHER \| CRYPTO_ALG_ASYNC,
492	.cra_blocksize = SERPENT_BLOCK_SIZE,
493	.cra_ctxsize = sizeof(struct async_helper_ctx),
494	.cra_alignmask = 0,
495	.cra_type = &crypto_ablkcipher_type,
496	.cra_module = THIS_MODULE,
497	.cra_list = LIST_HEAD_INIT(srp_algs[8].cra_list),
498	.cra_init = ablk_init,
499	.cra_exit = ablk_exit,
500	.cra_u = {
501	.ablkcipher = {
502	.min_keysize = SERPENT_MIN_KEY_SIZE +
503	SERPENT_BLOCK_SIZE,
504	.max_keysize = SERPENT_MAX_KEY_SIZE +
505	SERPENT_BLOCK_SIZE,
506	.ivsize = SERPENT_BLOCK_SIZE,
507	.setkey = ablk_set_key,
508	.encrypt = ablk_encrypt,
509	.decrypt = ablk_decrypt,
510	},
511	},
512	}, {
513	.cra_name = "xts(serpent)",
514	.cra_driver_name = "xts-serpent-avx2",
515	.cra_priority = 600,
516	.cra_flags = CRYPTO_ALG_TYPE_ABLKCIPHER \| CRYPTO_ALG_ASYNC,
517	.cra_blocksize = SERPENT_BLOCK_SIZE,
518	.cra_ctxsize = sizeof(struct async_helper_ctx),
519	.cra_alignmask = 0,
520	.cra_type = &crypto_ablkcipher_type,
521	.cra_module = THIS_MODULE,
522	.cra_list = LIST_HEAD_INIT(srp_algs[9].cra_list),
523	.cra_init = ablk_init,
524	.cra_exit = ablk_exit,
525	.cra_u = {
526	.ablkcipher = {
527	.min_keysize = SERPENT_MIN_KEY_SIZE * 2,
528	.max_keysize = SERPENT_MAX_KEY_SIZE * 2,
529	.ivsize = SERPENT_BLOCK_SIZE,
530	.setkey = ablk_set_key,
531	.encrypt = ablk_encrypt,
532	.decrypt = ablk_decrypt,
533	},
534	},
535	} };
536
537	static int __init init(void)
538	{
534ff06e	539	const char *feature_name;
56d76c96	540
b54b4bbb IM	541	if (!cpu_has_avx2 \|\| !cpu_has_osxsave) {
	542	pr_info("AVX2 instructions are not detected.\n");
	543	return -ENODEV;
	544	}
d91cab78 DH	545	if (!cpu_has_xfeatures(XFEATURE_MASK_SSE \| XFEATURE_MASK_YMM,
d91cab78 DH	546	&feature_name)) {
534ff06e	547	pr_info("CPU feature '%s' is not supported.\n", feature_name);
56d76c96 JK	548	return -ENODEV;
	549	}
	550
	551	return crypto_register_algs(srp_algs, ARRAY_SIZE(srp_algs));
	552	}
	553
	554	static void __exit fini(void)
	555	{
	556	crypto_unregister_algs(srp_algs, ARRAY_SIZE(srp_algs));
	557	}
	558
	559	module_init(init);
	560	module_exit(fini);
	561
	562	MODULE_LICENSE("GPL");
	563	MODULE_DESCRIPTION("Serpent Cipher Algorithm, AVX2 optimized");
5d26a105 KC	564	MODULE_ALIAS_CRYPTO("serpent");
5d26a105 KC	565	MODULE_ALIAS_CRYPTO("serpent-asm");