[linux.git] / arch / sparc / crypto / aes_glue.c

// SPDX-License-Identifier: GPL-2.0-only
/* Glue code for AES encryption optimized for sparc64 crypto opcodes.
 *
 * This is based largely upon arch/x86/crypto/aesni-intel_glue.c
 *
 * Copyright (C) 2008, Intel Corp.
 *    Author: Huang Ying <[email protected]>
 *
 * Added RFC4106 AES-GCM support for 128-bit keys under the AEAD
 * interface for 64-bit kernels.
 *    Authors: Adrian Hoban <[email protected]>
 *             Gabriele Paoloni <[email protected]>
 *             Tadeusz Struk ([email protected])
 *             Aidan O'Mahony ([email protected])
 *    Copyright (c) 2010, Intel Corporation.
 */

#define pr_fmt(fmt)	KBUILD_MODNAME ": " fmt

#include <linux/crypto.h>
#include <linux/init.h>
#include <linux/module.h>
#include <linux/mm.h>
#include <linux/types.h>
#include <crypto/algapi.h>
#include <crypto/aes.h>
#include <crypto/internal/skcipher.h>

#include <asm/fpumacro.h>
#include <asm/pstate.h>
#include <asm/elf.h>

#include "opcodes.h"

struct aes_ops {
	void (*encrypt)(const u64 *key, const u32 *input, u32 *output);
	void (*decrypt)(const u64 *key, const u32 *input, u32 *output);
	void (*load_encrypt_keys)(const u64 *key);
	void (*load_decrypt_keys)(const u64 *key);
	void (*ecb_encrypt)(const u64 *key, const u64 *input, u64 *output,
			    unsigned int len);
	void (*ecb_decrypt)(const u64 *key, const u64 *input, u64 *output,
			    unsigned int len);
	void (*cbc_encrypt)(const u64 *key, const u64 *input, u64 *output,
			    unsigned int len, u64 *iv);
	void (*cbc_decrypt)(const u64 *key, const u64 *input, u64 *output,
			    unsigned int len, u64 *iv);
	void (*ctr_crypt)(const u64 *key, const u64 *input, u64 *output,
			  unsigned int len, u64 *iv);
};

struct crypto_sparc64_aes_ctx {
	struct aes_ops *ops;
	u64 key[AES_MAX_KEYLENGTH / sizeof(u64)];
	u32 key_length;
	u32 expanded_key_length;
};

extern void aes_sparc64_encrypt_128(const u64 *key, const u32 *input,
				    u32 *output);
extern void aes_sparc64_encrypt_192(const u64 *key, const u32 *input,
				    u32 *output);
extern void aes_sparc64_encrypt_256(const u64 *key, const u32 *input,
				    u32 *output);

extern void aes_sparc64_decrypt_128(const u64 *key, const u32 *input,
				    u32 *output);
extern void aes_sparc64_decrypt_192(const u64 *key, const u32 *input,
				    u32 *output);
extern void aes_sparc64_decrypt_256(const u64 *key, const u32 *input,
				    u32 *output);

extern void aes_sparc64_load_encrypt_keys_128(const u64 *key);
extern void aes_sparc64_load_encrypt_keys_192(const u64 *key);
extern void aes_sparc64_load_encrypt_keys_256(const u64 *key);

extern void aes_sparc64_load_decrypt_keys_128(const u64 *key);
extern void aes_sparc64_load_decrypt_keys_192(const u64 *key);
extern void aes_sparc64_load_decrypt_keys_256(const u64 *key);

extern void aes_sparc64_ecb_encrypt_128(const u64 *key, const u64 *input,
					u64 *output, unsigned int len);
extern void aes_sparc64_ecb_encrypt_192(const u64 *key, const u64 *input,
					u64 *output, unsigned int len);
extern void aes_sparc64_ecb_encrypt_256(const u64 *key, const u64 *input,
					u64 *output, unsigned int len);

extern void aes_sparc64_ecb_decrypt_128(const u64 *key, const u64 *input,
					u64 *output, unsigned int len);
extern void aes_sparc64_ecb_decrypt_192(const u64 *key, const u64 *input,
					u64 *output, unsigned int len);
extern void aes_sparc64_ecb_decrypt_256(const u64 *key, const u64 *input,
					u64 *output, unsigned int len);

extern void aes_sparc64_cbc_encrypt_128(const u64 *key, const u64 *input,
					u64 *output, unsigned int len,
					u64 *iv);

extern void aes_sparc64_cbc_encrypt_192(const u64 *key, const u64 *input,
					u64 *output, unsigned int len,
					u64 *iv);

extern void aes_sparc64_cbc_encrypt_256(const u64 *key, const u64 *input,
					u64 *output, unsigned int len,
					u64 *iv);

extern void aes_sparc64_cbc_decrypt_128(const u64 *key, const u64 *input,
					u64 *output, unsigned int len,
					u64 *iv);

extern void aes_sparc64_cbc_decrypt_192(const u64 *key, const u64 *input,
					u64 *output, unsigned int len,
					u64 *iv);

extern void aes_sparc64_cbc_decrypt_256(const u64 *key, const u64 *input,
					u64 *output, unsigned int len,
					u64 *iv);

extern void aes_sparc64_ctr_crypt_128(const u64 *key, const u64 *input,
				      u64 *output, unsigned int len,
				      u64 *iv);
extern void aes_sparc64_ctr_crypt_192(const u64 *key, const u64 *input,
				      u64 *output, unsigned int len,
				      u64 *iv);
extern void aes_sparc64_ctr_crypt_256(const u64 *key, const u64 *input,
				      u64 *output, unsigned int len,
				      u64 *iv);

static struct aes_ops aes128_ops = {
	.encrypt		= aes_sparc64_encrypt_128,
	.decrypt		= aes_sparc64_decrypt_128,
	.load_encrypt_keys	= aes_sparc64_load_encrypt_keys_128,
	.load_decrypt_keys	= aes_sparc64_load_decrypt_keys_128,
	.ecb_encrypt		= aes_sparc64_ecb_encrypt_128,
	.ecb_decrypt		= aes_sparc64_ecb_decrypt_128,
	.cbc_encrypt		= aes_sparc64_cbc_encrypt_128,
	.cbc_decrypt		= aes_sparc64_cbc_decrypt_128,
	.ctr_crypt		= aes_sparc64_ctr_crypt_128,
};

static struct aes_ops aes192_ops = {
	.encrypt		= aes_sparc64_encrypt_192,
	.decrypt		= aes_sparc64_decrypt_192,
	.load_encrypt_keys	= aes_sparc64_load_encrypt_keys_192,
	.load_decrypt_keys	= aes_sparc64_load_decrypt_keys_192,
	.ecb_encrypt		= aes_sparc64_ecb_encrypt_192,
	.ecb_decrypt		= aes_sparc64_ecb_decrypt_192,
	.cbc_encrypt		= aes_sparc64_cbc_encrypt_192,
	.cbc_decrypt		= aes_sparc64_cbc_decrypt_192,
	.ctr_crypt		= aes_sparc64_ctr_crypt_192,
};

static struct aes_ops aes256_ops = {
	.encrypt		= aes_sparc64_encrypt_256,
	.decrypt		= aes_sparc64_decrypt_256,
	.load_encrypt_keys	= aes_sparc64_load_encrypt_keys_256,
	.load_decrypt_keys	= aes_sparc64_load_decrypt_keys_256,
	.ecb_encrypt		= aes_sparc64_ecb_encrypt_256,
	.ecb_decrypt		= aes_sparc64_ecb_decrypt_256,
	.cbc_encrypt		= aes_sparc64_cbc_encrypt_256,
	.cbc_decrypt		= aes_sparc64_cbc_decrypt_256,
	.ctr_crypt		= aes_sparc64_ctr_crypt_256,
};

extern void aes_sparc64_key_expand(const u32 *in_key, u64 *output_key,
				   unsigned int key_len);

static int aes_set_key(struct crypto_tfm *tfm, const u8 *in_key,
		       unsigned int key_len)
{
	struct crypto_sparc64_aes_ctx *ctx = crypto_tfm_ctx(tfm);

	switch (key_len) {
	case AES_KEYSIZE_128:
		ctx->expanded_key_length = 0xb0;
		ctx->ops = &aes128_ops;
		break;

	case AES_KEYSIZE_192:
		ctx->expanded_key_length = 0xd0;
		ctx->ops = &aes192_ops;
		break;

	case AES_KEYSIZE_256:
		ctx->expanded_key_length = 0xf0;
		ctx->ops = &aes256_ops;
		break;

	default:
		return -EINVAL;
	}

	aes_sparc64_key_expand((const u32 *)in_key, &ctx->key[0], key_len);
	ctx->key_length = key_len;

	return 0;
}

static int aes_set_key_skcipher(struct crypto_skcipher *tfm, const u8 *in_key,
				unsigned int key_len)
{
	return aes_set_key(crypto_skcipher_tfm(tfm), in_key, key_len);
}

static void crypto_aes_encrypt(struct crypto_tfm *tfm, u8 *dst, const u8 *src)
{
	struct crypto_sparc64_aes_ctx *ctx = crypto_tfm_ctx(tfm);

	ctx->ops->encrypt(&ctx->key[0], (const u32 *) src, (u32 *) dst);
}

static void crypto_aes_decrypt(struct crypto_tfm *tfm, u8 *dst, const u8 *src)
{
	struct crypto_sparc64_aes_ctx *ctx = crypto_tfm_ctx(tfm);

	ctx->ops->decrypt(&ctx->key[0], (const u32 *) src, (u32 *) dst);
}

static int ecb_encrypt(struct skcipher_request *req)
{
	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	struct skcipher_walk walk;
	unsigned int nbytes;
	int err;

	err = skcipher_walk_virt(&walk, req, true);
	if (err)
		return err;

	ctx->ops->load_encrypt_keys(&ctx->key[0]);
	while ((nbytes = walk.nbytes) != 0) {
		ctx->ops->ecb_encrypt(&ctx->key[0], walk.src.virt.addr,
				      walk.dst.virt.addr,
				      round_down(nbytes, AES_BLOCK_SIZE));
		err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
	}
	fprs_write(0);
	return err;
}

static int ecb_decrypt(struct skcipher_request *req)
{
	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	const u64 *key_end;
	struct skcipher_walk walk;
	unsigned int nbytes;
	int err;

	err = skcipher_walk_virt(&walk, req, true);
	if (err)
		return err;

	ctx->ops->load_decrypt_keys(&ctx->key[0]);
	key_end = &ctx->key[ctx->expanded_key_length / sizeof(u64)];
	while ((nbytes = walk.nbytes) != 0) {
		ctx->ops->ecb_decrypt(key_end, walk.src.virt.addr,
				      walk.dst.virt.addr,
				      round_down(nbytes, AES_BLOCK_SIZE));
		err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
	}
	fprs_write(0);

	return err;
}

static int cbc_encrypt(struct skcipher_request *req)
{
	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	struct skcipher_walk walk;
	unsigned int nbytes;
	int err;

	err = skcipher_walk_virt(&walk, req, true);
	if (err)
		return err;

	ctx->ops->load_encrypt_keys(&ctx->key[0]);
	while ((nbytes = walk.nbytes) != 0) {
		ctx->ops->cbc_encrypt(&ctx->key[0], walk.src.virt.addr,
				      walk.dst.virt.addr,
				      round_down(nbytes, AES_BLOCK_SIZE),
				      walk.iv);
		err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
	}
	fprs_write(0);
	return err;
}

static int cbc_decrypt(struct skcipher_request *req)
{
	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	const u64 *key_end;
	struct skcipher_walk walk;
	unsigned int nbytes;
	int err;

	err = skcipher_walk_virt(&walk, req, true);
	if (err)
		return err;

	ctx->ops->load_decrypt_keys(&ctx->key[0]);
	key_end = &ctx->key[ctx->expanded_key_length / sizeof(u64)];
	while ((nbytes = walk.nbytes) != 0) {
		ctx->ops->cbc_decrypt(key_end, walk.src.virt.addr,
				      walk.dst.virt.addr,
				      round_down(nbytes, AES_BLOCK_SIZE),
				      walk.iv);
		err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
	}
	fprs_write(0);

	return err;
}

static void ctr_crypt_final(const struct crypto_sparc64_aes_ctx *ctx,
			    struct skcipher_walk *walk)
{
	u8 *ctrblk = walk->iv;
	u64 keystream[AES_BLOCK_SIZE / sizeof(u64)];
	u8 *src = walk->src.virt.addr;
	u8 *dst = walk->dst.virt.addr;
	unsigned int nbytes = walk->nbytes;

	ctx->ops->ecb_encrypt(&ctx->key[0], (const u64 *)ctrblk,
			      keystream, AES_BLOCK_SIZE);
	crypto_xor_cpy(dst, (u8 *) keystream, src, nbytes);
	crypto_inc(ctrblk, AES_BLOCK_SIZE);
}

static int ctr_crypt(struct skcipher_request *req)
{
	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	struct skcipher_walk walk;
	unsigned int nbytes;
	int err;

	err = skcipher_walk_virt(&walk, req, true);
	if (err)
		return err;

	ctx->ops->load_encrypt_keys(&ctx->key[0]);
	while ((nbytes = walk.nbytes) >= AES_BLOCK_SIZE) {
		ctx->ops->ctr_crypt(&ctx->key[0], walk.src.virt.addr,
				    walk.dst.virt.addr,
				    round_down(nbytes, AES_BLOCK_SIZE),
				    walk.iv);
		err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
	}
	if (walk.nbytes) {
		ctr_crypt_final(ctx, &walk);
		err = skcipher_walk_done(&walk, 0);
	}
	fprs_write(0);
	return err;
}

static struct crypto_alg cipher_alg = {
	.cra_name		= "aes",
	.cra_driver_name	= "aes-sparc64",
	.cra_priority		= SPARC_CR_OPCODE_PRIORITY,
	.cra_flags		= CRYPTO_ALG_TYPE_CIPHER,
	.cra_blocksize		= AES_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct crypto_sparc64_aes_ctx),
	.cra_alignmask		= 3,
	.cra_module		= THIS_MODULE,
	.cra_u	= {
		.cipher	= {
			.cia_min_keysize	= AES_MIN_KEY_SIZE,
			.cia_max_keysize	= AES_MAX_KEY_SIZE,
			.cia_setkey		= aes_set_key,
			.cia_encrypt		= crypto_aes_encrypt,
			.cia_decrypt		= crypto_aes_decrypt
		}
	}
};

static struct skcipher_alg skcipher_algs[] = {
	{
		.base.cra_name		= "ecb(aes)",
		.base.cra_driver_name	= "ecb-aes-sparc64",
		.base.cra_priority	= SPARC_CR_OPCODE_PRIORITY,
		.base.cra_blocksize	= AES_BLOCK_SIZE,
		.base.cra_ctxsize	= sizeof(struct crypto_sparc64_aes_ctx),
		.base.cra_alignmask	= 7,
		.base.cra_module	= THIS_MODULE,
		.min_keysize		= AES_MIN_KEY_SIZE,
		.max_keysize		= AES_MAX_KEY_SIZE,
		.setkey			= aes_set_key_skcipher,
		.encrypt		= ecb_encrypt,
		.decrypt		= ecb_decrypt,
	}, {
		.base.cra_name		= "cbc(aes)",
		.base.cra_driver_name	= "cbc-aes-sparc64",
		.base.cra_priority	= SPARC_CR_OPCODE_PRIORITY,
		.base.cra_blocksize	= AES_BLOCK_SIZE,
		.base.cra_ctxsize	= sizeof(struct crypto_sparc64_aes_ctx),
		.base.cra_alignmask	= 7,
		.base.cra_module	= THIS_MODULE,
		.min_keysize		= AES_MIN_KEY_SIZE,
		.max_keysize		= AES_MAX_KEY_SIZE,
		.ivsize			= AES_BLOCK_SIZE,
		.setkey			= aes_set_key_skcipher,
		.encrypt		= cbc_encrypt,
		.decrypt		= cbc_decrypt,
	}, {
		.base.cra_name		= "ctr(aes)",
		.base.cra_driver_name	= "ctr-aes-sparc64",
		.base.cra_priority	= SPARC_CR_OPCODE_PRIORITY,
		.base.cra_blocksize	= 1,
		.base.cra_ctxsize	= sizeof(struct crypto_sparc64_aes_ctx),
		.base.cra_alignmask	= 7,
		.base.cra_module	= THIS_MODULE,
		.min_keysize		= AES_MIN_KEY_SIZE,
		.max_keysize		= AES_MAX_KEY_SIZE,
		.ivsize			= AES_BLOCK_SIZE,
		.setkey			= aes_set_key_skcipher,
		.encrypt		= ctr_crypt,
		.decrypt		= ctr_crypt,
		.chunksize		= AES_BLOCK_SIZE,
	}
};

static bool __init sparc64_has_aes_opcode(void)
{
	unsigned long cfr;

	if (!(sparc64_elf_hwcap & HWCAP_SPARC_CRYPTO))
		return false;

	__asm__ __volatile__("rd %%asr26, %0" : "=r" (cfr));
	if (!(cfr & CFR_AES))
		return false;

	return true;
}

static int __init aes_sparc64_mod_init(void)
{
	int err;

	if (!sparc64_has_aes_opcode()) {
		pr_info("sparc64 aes opcodes not available.\n");
		return -ENODEV;
	}
	pr_info("Using sparc64 aes opcodes optimized AES implementation\n");
	err = crypto_register_alg(&cipher_alg);
	if (err)
		return err;
	err = crypto_register_skciphers(skcipher_algs,
					ARRAY_SIZE(skcipher_algs));
	if (err)
		crypto_unregister_alg(&cipher_alg);
	return err;
}

static void __exit aes_sparc64_mod_fini(void)
{
	crypto_unregister_alg(&cipher_alg);
	crypto_unregister_skciphers(skcipher_algs, ARRAY_SIZE(skcipher_algs));
}

module_init(aes_sparc64_mod_init);
module_exit(aes_sparc64_mod_fini);

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Rijndael (AES) Cipher Algorithm, sparc64 aes opcode accelerated");

MODULE_ALIAS_CRYPTO("aes");

#include "crop_devid.c"
Commit	Line	Data
09c434b8	1	// SPDX-License-Identifier: GPL-2.0-only
9bf4852d DM	2	/* Glue code for AES encryption optimized for sparc64 crypto opcodes.
	3	*
	4	* This is based largely upon arch/x86/crypto/aesni-intel_glue.c
	5	*
	6	* Copyright (C) 2008, Intel Corp.
	7	* Author: Huang Ying <[email protected]>
	8	*
	9	* Added RFC4106 AES-GCM support for 128-bit keys under the AEAD
	10	* interface for 64-bit kernels.
	11	* Authors: Adrian Hoban <[email protected]>
	12	* Gabriele Paoloni <[email protected]>
	13	* Tadeusz Struk ([email protected])
	14	* Aidan O'Mahony ([email protected])
	15	* Copyright (c) 2010, Intel Corporation.
	16	*/
	17
71741680 DM	18	#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
71741680 DM	19
9bf4852d DM	20	#include <linux/crypto.h>
	21	#include <linux/init.h>
	22	#include <linux/module.h>
	23	#include <linux/mm.h>
	24	#include <linux/types.h>
	25	#include <crypto/algapi.h>
	26	#include <crypto/aes.h>
64db5e74	27	#include <crypto/internal/skcipher.h>
9bf4852d DM	28
	29	#include <asm/fpumacro.h>
	30	#include <asm/pstate.h>
	31	#include <asm/elf.h>
	32
10803624 DM	33	#include "opcodes.h"
10803624 DM	34
0bdcaf74 DM	35	struct aes_ops {
	36	void (encrypt)(const u64 key, const u32 input, u32 output);
	37	void (decrypt)(const u64 key, const u32 input, u32 output);
	38	void (load_encrypt_keys)(const u64 key);
	39	void (load_decrypt_keys)(const u64 key);
	40	void (ecb_encrypt)(const u64 key, const u64 input, u64 output,
	41	unsigned int len);
	42	void (ecb_decrypt)(const u64 key, const u64 input, u64 output,
	43	unsigned int len);
	44	void (cbc_encrypt)(const u64 key, const u64 input, u64 output,
	45	unsigned int len, u64 *iv);
	46	void (cbc_decrypt)(const u64 key, const u64 input, u64 output,
	47	unsigned int len, u64 *iv);
9fd130ec DM	48	void (ctr_crypt)(const u64 key, const u64 input, u64 output,
9fd130ec DM	49	unsigned int len, u64 *iv);
0bdcaf74 DM	50	};
0bdcaf74 DM	51
9bf4852d	52	struct crypto_sparc64_aes_ctx {
0bdcaf74	53	struct aes_ops *ops;
9bf4852d DM	54	u64 key[AES_MAX_KEYLENGTH / sizeof(u64)];
	55	u32 key_length;
	56	u32 expanded_key_length;
	57	};
	58
0bdcaf74 DM	59	extern void aes_sparc64_encrypt_128(const u64 key, const u32 input,
	60	u32 *output);
	61	extern void aes_sparc64_encrypt_192(const u64 key, const u32 input,
	62	u32 *output);
	63	extern void aes_sparc64_encrypt_256(const u64 key, const u32 input,
	64	u32 *output);
	65
	66	extern void aes_sparc64_decrypt_128(const u64 key, const u32 input,
	67	u32 *output);
	68	extern void aes_sparc64_decrypt_192(const u64 key, const u32 input,
	69	u32 *output);
	70	extern void aes_sparc64_decrypt_256(const u64 key, const u32 input,
	71	u32 *output);
	72
	73	extern void aes_sparc64_load_encrypt_keys_128(const u64 *key);
	74	extern void aes_sparc64_load_encrypt_keys_192(const u64 *key);
	75	extern void aes_sparc64_load_encrypt_keys_256(const u64 *key);
	76
	77	extern void aes_sparc64_load_decrypt_keys_128(const u64 *key);
	78	extern void aes_sparc64_load_decrypt_keys_192(const u64 *key);
	79	extern void aes_sparc64_load_decrypt_keys_256(const u64 *key);
	80
	81	extern void aes_sparc64_ecb_encrypt_128(const u64 key, const u64 input,
	82	u64 *output, unsigned int len);
	83	extern void aes_sparc64_ecb_encrypt_192(const u64 key, const u64 input,
	84	u64 *output, unsigned int len);
	85	extern void aes_sparc64_ecb_encrypt_256(const u64 key, const u64 input,
	86	u64 *output, unsigned int len);
	87
	88	extern void aes_sparc64_ecb_decrypt_128(const u64 key, const u64 input,
	89	u64 *output, unsigned int len);
	90	extern void aes_sparc64_ecb_decrypt_192(const u64 key, const u64 input,
	91	u64 *output, unsigned int len);
	92	extern void aes_sparc64_ecb_decrypt_256(const u64 key, const u64 input,
	93	u64 *output, unsigned int len);
	94
	95	extern void aes_sparc64_cbc_encrypt_128(const u64 key, const u64 input,
	96	u64 *output, unsigned int len,
	97	u64 *iv);
	98
	99	extern void aes_sparc64_cbc_encrypt_192(const u64 key, const u64 input,
	100	u64 *output, unsigned int len,
	101	u64 *iv);
	102
	103	extern void aes_sparc64_cbc_encrypt_256(const u64 key, const u64 input,
	104	u64 *output, unsigned int len,
	105	u64 *iv);
	106
	107	extern void aes_sparc64_cbc_decrypt_128(const u64 key, const u64 input,
	108	u64 *output, unsigned int len,
	109	u64 *iv);
	110
	111	extern void aes_sparc64_cbc_decrypt_192(const u64 key, const u64 input,
	112	u64 *output, unsigned int len,
	113	u64 *iv);
	114
	115	extern void aes_sparc64_cbc_decrypt_256(const u64 key, const u64 input,
	116	u64 *output, unsigned int len,
	117	u64 *iv);
	118
9fd130ec DM	119	extern void aes_sparc64_ctr_crypt_128(const u64 key, const u64 input,
	120	u64 *output, unsigned int len,
	121	u64 *iv);
	122	extern void aes_sparc64_ctr_crypt_192(const u64 key, const u64 input,
	123	u64 *output, unsigned int len,
	124	u64 *iv);
	125	extern void aes_sparc64_ctr_crypt_256(const u64 key, const u64 input,
	126	u64 *output, unsigned int len,
	127	u64 *iv);
	128
756382cb	129	static struct aes_ops aes128_ops = {
0bdcaf74 DM	130	.encrypt = aes_sparc64_encrypt_128,
	131	.decrypt = aes_sparc64_decrypt_128,
	132	.load_encrypt_keys = aes_sparc64_load_encrypt_keys_128,
	133	.load_decrypt_keys = aes_sparc64_load_decrypt_keys_128,
	134	.ecb_encrypt = aes_sparc64_ecb_encrypt_128,
	135	.ecb_decrypt = aes_sparc64_ecb_decrypt_128,
	136	.cbc_encrypt = aes_sparc64_cbc_encrypt_128,
	137	.cbc_decrypt = aes_sparc64_cbc_decrypt_128,
9fd130ec	138	.ctr_crypt = aes_sparc64_ctr_crypt_128,
0bdcaf74 DM	139	};
0bdcaf74 DM	140
756382cb	141	static struct aes_ops aes192_ops = {
0bdcaf74 DM	142	.encrypt = aes_sparc64_encrypt_192,
	143	.decrypt = aes_sparc64_decrypt_192,
	144	.load_encrypt_keys = aes_sparc64_load_encrypt_keys_192,
	145	.load_decrypt_keys = aes_sparc64_load_decrypt_keys_192,
	146	.ecb_encrypt = aes_sparc64_ecb_encrypt_192,
	147	.ecb_decrypt = aes_sparc64_ecb_decrypt_192,
	148	.cbc_encrypt = aes_sparc64_cbc_encrypt_192,
	149	.cbc_decrypt = aes_sparc64_cbc_decrypt_192,
9fd130ec	150	.ctr_crypt = aes_sparc64_ctr_crypt_192,
0bdcaf74 DM	151	};
0bdcaf74 DM	152
756382cb	153	static struct aes_ops aes256_ops = {
0bdcaf74 DM	154	.encrypt = aes_sparc64_encrypt_256,
	155	.decrypt = aes_sparc64_decrypt_256,
	156	.load_encrypt_keys = aes_sparc64_load_encrypt_keys_256,
	157	.load_decrypt_keys = aes_sparc64_load_decrypt_keys_256,
	158	.ecb_encrypt = aes_sparc64_ecb_encrypt_256,
	159	.ecb_decrypt = aes_sparc64_ecb_decrypt_256,
	160	.cbc_encrypt = aes_sparc64_cbc_encrypt_256,
	161	.cbc_decrypt = aes_sparc64_cbc_decrypt_256,
9fd130ec	162	.ctr_crypt = aes_sparc64_ctr_crypt_256,
0bdcaf74 DM	163	};
0bdcaf74 DM	164
9bf4852d DM	165	extern void aes_sparc64_key_expand(const u32 in_key, u64 output_key,
	166	unsigned int key_len);
	167
	168	static int aes_set_key(struct crypto_tfm tfm, const u8 in_key,
	169	unsigned int key_len)
	170	{
	171	struct crypto_sparc64_aes_ctx *ctx = crypto_tfm_ctx(tfm);
9bf4852d DM	172
	173	switch (key_len) {
	174	case AES_KEYSIZE_128:
	175	ctx->expanded_key_length = 0xb0;
0bdcaf74	176	ctx->ops = &aes128_ops;
9bf4852d DM	177	break;
	178
	179	case AES_KEYSIZE_192:
	180	ctx->expanded_key_length = 0xd0;
0bdcaf74	181	ctx->ops = &aes192_ops;
9bf4852d DM	182	break;
	183
	184	case AES_KEYSIZE_256:
	185	ctx->expanded_key_length = 0xf0;
0bdcaf74	186	ctx->ops = &aes256_ops;
9bf4852d DM	187	break;
	188
	189	default:
9bf4852d DM	190	return -EINVAL;
	191	}
	192
	193	aes_sparc64_key_expand((const u32 *)in_key, &ctx->key[0], key_len);
	194	ctx->key_length = key_len;
	195
	196	return 0;
	197	}
	198
64db5e74 EB	199	static int aes_set_key_skcipher(struct crypto_skcipher tfm, const u8 in_key,
	200	unsigned int key_len)
	201	{
	202	return aes_set_key(crypto_skcipher_tfm(tfm), in_key, key_len);
	203	}
	204
724ecd3c	205	static void crypto_aes_encrypt(struct crypto_tfm tfm, u8 dst, const u8 *src)
9bf4852d DM	206	{
	207	struct crypto_sparc64_aes_ctx *ctx = crypto_tfm_ctx(tfm);
	208
0bdcaf74	209	ctx->ops->encrypt(&ctx->key[0], (const u32 ) src, (u32 ) dst);
9bf4852d DM	210	}
9bf4852d DM	211
724ecd3c	212	static void crypto_aes_decrypt(struct crypto_tfm tfm, u8 dst, const u8 *src)
9bf4852d DM	213	{
	214	struct crypto_sparc64_aes_ctx *ctx = crypto_tfm_ctx(tfm);
	215
0bdcaf74	216	ctx->ops->decrypt(&ctx->key[0], (const u32 ) src, (u32 ) dst);
9bf4852d DM	217	}
9bf4852d DM	218
64db5e74	219	static int ecb_encrypt(struct skcipher_request *req)
9bf4852d	220	{
64db5e74 EB	221	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	222	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	223	struct skcipher_walk walk;
	224	unsigned int nbytes;
9bf4852d DM	225	int err;
9bf4852d DM	226
64db5e74 EB	227	err = skcipher_walk_virt(&walk, req, true);
	228	if (err)
	229	return err;
9bf4852d	230
0bdcaf74	231	ctx->ops->load_encrypt_keys(&ctx->key[0]);
64db5e74 EB	232	while ((nbytes = walk.nbytes) != 0) {
	233	ctx->ops->ecb_encrypt(&ctx->key[0], walk.src.virt.addr,
	234	walk.dst.virt.addr,
	235	round_down(nbytes, AES_BLOCK_SIZE));
	236	err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
9bf4852d DM	237	}
	238	fprs_write(0);
	239	return err;
	240	}
	241
64db5e74	242	static int ecb_decrypt(struct skcipher_request *req)
9bf4852d	243	{
64db5e74 EB	244	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	245	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	246	const u64 *key_end;
	247	struct skcipher_walk walk;
	248	unsigned int nbytes;
9bf4852d DM	249	int err;
9bf4852d DM	250
64db5e74 EB	251	err = skcipher_walk_virt(&walk, req, true);
	252	if (err)
	253	return err;
9bf4852d	254
0bdcaf74	255	ctx->ops->load_decrypt_keys(&ctx->key[0]);
9bf4852d	256	key_end = &ctx->key[ctx->expanded_key_length / sizeof(u64)];
64db5e74 EB	257	while ((nbytes = walk.nbytes) != 0) {
	258	ctx->ops->ecb_decrypt(key_end, walk.src.virt.addr,
	259	walk.dst.virt.addr,
	260	round_down(nbytes, AES_BLOCK_SIZE));
	261	err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
9bf4852d DM	262	}
	263	fprs_write(0);
	264
	265	return err;
	266	}
	267
64db5e74	268	static int cbc_encrypt(struct skcipher_request *req)
9bf4852d	269	{
64db5e74 EB	270	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	271	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	272	struct skcipher_walk walk;
	273	unsigned int nbytes;
9bf4852d DM	274	int err;
9bf4852d DM	275
64db5e74 EB	276	err = skcipher_walk_virt(&walk, req, true);
	277	if (err)
	278	return err;
9bf4852d	279
0bdcaf74	280	ctx->ops->load_encrypt_keys(&ctx->key[0]);
64db5e74 EB	281	while ((nbytes = walk.nbytes) != 0) {
	282	ctx->ops->cbc_encrypt(&ctx->key[0], walk.src.virt.addr,
	283	walk.dst.virt.addr,
	284	round_down(nbytes, AES_BLOCK_SIZE),
	285	walk.iv);
	286	err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
9bf4852d DM	287	}
	288	fprs_write(0);
	289	return err;
	290	}
	291
64db5e74	292	static int cbc_decrypt(struct skcipher_request *req)
9bf4852d	293	{
64db5e74 EB	294	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	295	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	296	const u64 *key_end;
	297	struct skcipher_walk walk;
	298	unsigned int nbytes;
9bf4852d DM	299	int err;
9bf4852d DM	300
64db5e74 EB	301	err = skcipher_walk_virt(&walk, req, true);
	302	if (err)
	303	return err;
9bf4852d	304
0bdcaf74	305	ctx->ops->load_decrypt_keys(&ctx->key[0]);
9bf4852d	306	key_end = &ctx->key[ctx->expanded_key_length / sizeof(u64)];
64db5e74 EB	307	while ((nbytes = walk.nbytes) != 0) {
	308	ctx->ops->cbc_decrypt(key_end, walk.src.virt.addr,
	309	walk.dst.virt.addr,
	310	round_down(nbytes, AES_BLOCK_SIZE),
	311	walk.iv);
	312	err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
9bf4852d DM	313	}
	314	fprs_write(0);
	315
	316	return err;
	317	}
	318
64db5e74 EB	319	static void ctr_crypt_final(const struct crypto_sparc64_aes_ctx *ctx,
64db5e74 EB	320	struct skcipher_walk *walk)
a8d97cef DM	321	{
	322	u8 *ctrblk = walk->iv;
	323	u64 keystream[AES_BLOCK_SIZE / sizeof(u64)];
	324	u8 *src = walk->src.virt.addr;
	325	u8 *dst = walk->dst.virt.addr;
	326	unsigned int nbytes = walk->nbytes;
	327
	328	ctx->ops->ecb_encrypt(&ctx->key[0], (const u64 *)ctrblk,
	329	keystream, AES_BLOCK_SIZE);
45fe93df	330	crypto_xor_cpy(dst, (u8 *) keystream, src, nbytes);
a8d97cef DM	331	crypto_inc(ctrblk, AES_BLOCK_SIZE);
	332	}
	333
64db5e74	334	static int ctr_crypt(struct skcipher_request *req)
9fd130ec	335	{
64db5e74 EB	336	struct crypto_skcipher *tfm = crypto_skcipher_reqtfm(req);
	337	const struct crypto_sparc64_aes_ctx *ctx = crypto_skcipher_ctx(tfm);
	338	struct skcipher_walk walk;
	339	unsigned int nbytes;
9fd130ec DM	340	int err;
9fd130ec DM	341
64db5e74 EB	342	err = skcipher_walk_virt(&walk, req, true);
	343	if (err)
	344	return err;
9fd130ec DM	345
9fd130ec DM	346	ctx->ops->load_encrypt_keys(&ctx->key[0]);
a8d97cef	347	while ((nbytes = walk.nbytes) >= AES_BLOCK_SIZE) {
64db5e74 EB	348	ctx->ops->ctr_crypt(&ctx->key[0], walk.src.virt.addr,
	349	walk.dst.virt.addr,
	350	round_down(nbytes, AES_BLOCK_SIZE),
	351	walk.iv);
	352	err = skcipher_walk_done(&walk, nbytes % AES_BLOCK_SIZE);
9fd130ec	353	}
a8d97cef DM	354	if (walk.nbytes) {
a8d97cef DM	355	ctr_crypt_final(ctx, &walk);
64db5e74	356	err = skcipher_walk_done(&walk, 0);
a8d97cef	357	}
9fd130ec DM	358	fprs_write(0);
	359	return err;
	360	}
	361
64db5e74	362	static struct crypto_alg cipher_alg = {
9bf4852d DM	363	.cra_name = "aes",
9bf4852d DM	364	.cra_driver_name = "aes-sparc64",
10803624	365	.cra_priority = SPARC_CR_OPCODE_PRIORITY,
9bf4852d DM	366	.cra_flags = CRYPTO_ALG_TYPE_CIPHER,
	367	.cra_blocksize = AES_BLOCK_SIZE,
	368	.cra_ctxsize = sizeof(struct crypto_sparc64_aes_ctx),
	369	.cra_alignmask = 3,
	370	.cra_module = THIS_MODULE,
	371	.cra_u = {
	372	.cipher = {
	373	.cia_min_keysize = AES_MIN_KEY_SIZE,
	374	.cia_max_keysize = AES_MAX_KEY_SIZE,
	375	.cia_setkey = aes_set_key,
724ecd3c AB	376	.cia_encrypt = crypto_aes_encrypt,
724ecd3c AB	377	.cia_decrypt = crypto_aes_decrypt
9bf4852d DM	378	}
9bf4852d DM	379	}
64db5e74 EB	380	};
	381
	382	static struct skcipher_alg skcipher_algs[] = {
	383	{
	384	.base.cra_name = "ecb(aes)",
	385	.base.cra_driver_name = "ecb-aes-sparc64",
	386	.base.cra_priority = SPARC_CR_OPCODE_PRIORITY,
	387	.base.cra_blocksize = AES_BLOCK_SIZE,
	388	.base.cra_ctxsize = sizeof(struct crypto_sparc64_aes_ctx),
	389	.base.cra_alignmask = 7,
	390	.base.cra_module = THIS_MODULE,
	391	.min_keysize = AES_MIN_KEY_SIZE,
	392	.max_keysize = AES_MAX_KEY_SIZE,
	393	.setkey = aes_set_key_skcipher,
	394	.encrypt = ecb_encrypt,
	395	.decrypt = ecb_decrypt,
	396	}, {
	397	.base.cra_name = "cbc(aes)",
	398	.base.cra_driver_name = "cbc-aes-sparc64",
	399	.base.cra_priority = SPARC_CR_OPCODE_PRIORITY,
	400	.base.cra_blocksize = AES_BLOCK_SIZE,
	401	.base.cra_ctxsize = sizeof(struct crypto_sparc64_aes_ctx),
	402	.base.cra_alignmask = 7,
	403	.base.cra_module = THIS_MODULE,
	404	.min_keysize = AES_MIN_KEY_SIZE,
	405	.max_keysize = AES_MAX_KEY_SIZE,
	406	.ivsize = AES_BLOCK_SIZE,
	407	.setkey = aes_set_key_skcipher,
	408	.encrypt = cbc_encrypt,
	409	.decrypt = cbc_decrypt,
	410	}, {
	411	.base.cra_name = "ctr(aes)",
	412	.base.cra_driver_name = "ctr-aes-sparc64",
	413	.base.cra_priority = SPARC_CR_OPCODE_PRIORITY,
	414	.base.cra_blocksize = 1,
	415	.base.cra_ctxsize = sizeof(struct crypto_sparc64_aes_ctx),
	416	.base.cra_alignmask = 7,
	417	.base.cra_module = THIS_MODULE,
	418	.min_keysize = AES_MIN_KEY_SIZE,
	419	.max_keysize = AES_MAX_KEY_SIZE,
	420	.ivsize = AES_BLOCK_SIZE,
	421	.setkey = aes_set_key_skcipher,
	422	.encrypt = ctr_crypt,
	423	.decrypt = ctr_crypt,
	424	.chunksize = AES_BLOCK_SIZE,
	425	}
	426	};
9bf4852d DM	427
	428	static bool __init sparc64_has_aes_opcode(void)
	429	{
	430	unsigned long cfr;
	431
	432	if (!(sparc64_elf_hwcap & HWCAP_SPARC_CRYPTO))
	433	return false;
	434
	435	__asm__ __volatile__("rd %%asr26, %0" : "=r" (cfr));
	436	if (!(cfr & CFR_AES))
	437	return false;
	438
	439	return true;
	440	}
	441
	442	static int __init aes_sparc64_mod_init(void)
	443	{
64db5e74 EB	444	int err;
	445
	446	if (!sparc64_has_aes_opcode()) {
	447	pr_info("sparc64 aes opcodes not available.\n");
	448	return -ENODEV;
9bf4852d	449	}
64db5e74 EB	450	pr_info("Using sparc64 aes opcodes optimized AES implementation\n");
	451	err = crypto_register_alg(&cipher_alg);
	452	if (err)
	453	return err;
	454	err = crypto_register_skciphers(skcipher_algs,
	455	ARRAY_SIZE(skcipher_algs));
	456	if (err)
	457	crypto_unregister_alg(&cipher_alg);
	458	return err;
9bf4852d DM	459	}
	460
	461	static void __exit aes_sparc64_mod_fini(void)
	462	{
64db5e74 EB	463	crypto_unregister_alg(&cipher_alg);
64db5e74 EB	464	crypto_unregister_skciphers(skcipher_algs, ARRAY_SIZE(skcipher_algs));
9bf4852d DM	465	}
	466
	467	module_init(aes_sparc64_mod_init);
	468	module_exit(aes_sparc64_mod_fini);
	469
	470	MODULE_LICENSE("GPL");
b0126417	471	MODULE_DESCRIPTION("Rijndael (AES) Cipher Algorithm, sparc64 aes opcode accelerated");
9bf4852d	472
5d26a105	473	MODULE_ALIAS_CRYPTO("aes");
226f7cea DM	474
226f7cea DM	475	#include "crop_devid.c"