[GitHub/mt8127/android_kernel_alcatel_ttab.git] / arch / arm / crypto / aes_glue.c

/*
 * Glue Code for the asm optimized version of the AES Cipher Algorithm
 */

#include <linux/module.h>
#include <linux/crypto.h>
#include <crypto/aes.h>

#define AES_MAXNR 14

typedef struct {
	unsigned int rd_key[4 *(AES_MAXNR + 1)];
	int rounds;
} AES_KEY;

struct AES_CTX {
	AES_KEY enc_key;
	AES_KEY dec_key;
};

asmlinkage void AES_encrypt(const u8 *in, u8 *out, AES_KEY *ctx);
asmlinkage void AES_decrypt(const u8 *in, u8 *out, AES_KEY *ctx);
asmlinkage int private_AES_set_decrypt_key(const unsigned char *userKey, const int bits, AES_KEY *key);
asmlinkage int private_AES_set_encrypt_key(const unsigned char *userKey, const int bits, AES_KEY *key);

static void aes_encrypt(struct crypto_tfm *tfm, u8 *dst, const u8 *src)
{
	struct AES_CTX *ctx = crypto_tfm_ctx(tfm);
	AES_encrypt(src, dst, &ctx->enc_key);
}

static void aes_decrypt(struct crypto_tfm *tfm, u8 *dst, const u8 *src)
{
	struct AES_CTX *ctx = crypto_tfm_ctx(tfm);
	AES_decrypt(src, dst, &ctx->dec_key);
}

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

	switch (key_len) {
	case AES_KEYSIZE_128:
		key_len = 128;
		break;
	case AES_KEYSIZE_192:
		key_len = 192;
		break;
	case AES_KEYSIZE_256:
		key_len = 256;
		break;
	default:
		tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN;
		return -EINVAL;
	}

	if (private_AES_set_encrypt_key(in_key, key_len, &ctx->enc_key) == -1) {
		tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN;
		return -EINVAL;
	}
	/* private_AES_set_decrypt_key expects an encryption key as input */
	ctx->dec_key = ctx->enc_key;
	if (private_AES_set_decrypt_key(in_key, key_len, &ctx->dec_key) == -1) {
		tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN;
		return -EINVAL;
	}
	return 0;
}

static struct crypto_alg aes_alg = {
	.cra_name		= "aes",
	.cra_driver_name	= "aes-asm",
	.cra_priority		= 200,
	.cra_flags		= CRYPTO_ALG_TYPE_CIPHER,
	.cra_blocksize		= AES_BLOCK_SIZE,
	.cra_ctxsize		= sizeof(struct AES_CTX),
	.cra_module		= THIS_MODULE,
	.cra_list		= LIST_HEAD_INIT(aes_alg.cra_list),
	.cra_u	= {
		.cipher	= {
			.cia_min_keysize	= AES_MIN_KEY_SIZE,
			.cia_max_keysize	= AES_MAX_KEY_SIZE,
			.cia_setkey			= aes_set_key,
			.cia_encrypt		= aes_encrypt,
			.cia_decrypt		= aes_decrypt
		}
	}
};

static int __init aes_init(void)
{
	return crypto_register_alg(&aes_alg);
}

static void __exit aes_fini(void)
{
	crypto_unregister_alg(&aes_alg);
}

module_init(aes_init);
module_exit(aes_fini);

MODULE_DESCRIPTION("Rijndael (AES) Cipher Algorithm (ASM)");
MODULE_LICENSE("GPL");
MODULE_ALIAS("aes");
MODULE_ALIAS("aes-asm");
MODULE_AUTHOR("David McCullough <ucdevel@gmail.com>");
Commit	Line	Data
f0be44f4 DM	1	/*
	2	* Glue Code for the asm optimized version of the AES Cipher Algorithm
	3	*/
	4
	5	#include <linux/module.h>
	6	#include <linux/crypto.h>
	7	#include <crypto/aes.h>
	8
	9	#define AES_MAXNR 14
	10
	11	typedef struct {
	12	unsigned int rd_key[4 *(AES_MAXNR + 1)];
	13	int rounds;
	14	} AES_KEY;
	15
	16	struct AES_CTX {
	17	AES_KEY enc_key;
	18	AES_KEY dec_key;
	19	};
	20
	21	asmlinkage void AES_encrypt(const u8 in, u8 out, AES_KEY *ctx);
	22	asmlinkage void AES_decrypt(const u8 in, u8 out, AES_KEY *ctx);
	23	asmlinkage int private_AES_set_decrypt_key(const unsigned char userKey, const int bits, AES_KEY key);
	24	asmlinkage int private_AES_set_encrypt_key(const unsigned char userKey, const int bits, AES_KEY key);
	25
	26	static void aes_encrypt(struct crypto_tfm tfm, u8 dst, const u8 *src)
	27	{
	28	struct AES_CTX *ctx = crypto_tfm_ctx(tfm);
	29	AES_encrypt(src, dst, &ctx->enc_key);
	30	}
	31
	32	static void aes_decrypt(struct crypto_tfm tfm, u8 dst, const u8 *src)
	33	{
	34	struct AES_CTX *ctx = crypto_tfm_ctx(tfm);
	35	AES_decrypt(src, dst, &ctx->dec_key);
	36	}
	37
	38	static int aes_set_key(struct crypto_tfm tfm, const u8 in_key,
	39	unsigned int key_len)
	40	{
	41	struct AES_CTX *ctx = crypto_tfm_ctx(tfm);
	42
	43	switch (key_len) {
	44	case AES_KEYSIZE_128:
	45	key_len = 128;
	46	break;
	47	case AES_KEYSIZE_192:
	48	key_len = 192;
	49	break;
	50	case AES_KEYSIZE_256:
	51	key_len = 256;
	52	break;
	53	default:
	54	tfm->crt_flags \|= CRYPTO_TFM_RES_BAD_KEY_LEN;
	55	return -EINVAL;
	56	}
	57
	58	if (private_AES_set_encrypt_key(in_key, key_len, &ctx->enc_key) == -1) {
	59	tfm->crt_flags \|= CRYPTO_TFM_RES_BAD_KEY_LEN;
	60	return -EINVAL;
	61	}
	62	/* private_AES_set_decrypt_key expects an encryption key as input */
	63	ctx->dec_key = ctx->enc_key;
	64	if (private_AES_set_decrypt_key(in_key, key_len, &ctx->dec_key) == -1) {
65	tfm->crt_flags \|= CRYPTO_TFM_RES_BAD_KEY_LEN;
66	return -EINVAL;
67	}
68	return 0;
69	}
70
71	static struct crypto_alg aes_alg = {
72	.cra_name = "aes",
73	.cra_driver_name = "aes-asm",
74	.cra_priority = 200,
75	.cra_flags = CRYPTO_ALG_TYPE_CIPHER,
76	.cra_blocksize = AES_BLOCK_SIZE,
77	.cra_ctxsize = sizeof(struct AES_CTX),
78	.cra_module = THIS_MODULE,
79	.cra_list = LIST_HEAD_INIT(aes_alg.cra_list),
80	.cra_u = {
81	.cipher = {
82	.cia_min_keysize = AES_MIN_KEY_SIZE,
83	.cia_max_keysize = AES_MAX_KEY_SIZE,
84	.cia_setkey = aes_set_key,
85	.cia_encrypt = aes_encrypt,
86	.cia_decrypt = aes_decrypt
87	}
88	}
89	};
90
91	static int __init aes_init(void)
92	{
93	return crypto_register_alg(&aes_alg);
94	}
95
96	static void __exit aes_fini(void)
97	{
98	crypto_unregister_alg(&aes_alg);
99	}
100
101	module_init(aes_init);
102	module_exit(aes_fini);
103
104	MODULE_DESCRIPTION("Rijndael (AES) Cipher Algorithm (ASM)");
105	MODULE_LICENSE("GPL");
106	MODULE_ALIAS("aes");
107	MODULE_ALIAS("aes-asm");
108	MODULE_AUTHOR("David McCullough <ucdevel@gmail.com>");