[GitHub/mt8127/android_kernel_alcatel_ttab.git] / crypto / cipher.c

/*
 * Cryptographic API.
 *
 * Cipher operations.
 *
 * Copyright (c) 2002 James Morris <jmorris@intercode.com.au>
 * Copyright (c) 2005 Herbert Xu <herbert@gondor.apana.org.au>
 *
 * 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/compiler.h>
#include <linux/kernel.h>
#include <linux/crypto.h>
#include <linux/errno.h>
#include <linux/mm.h>
#include <linux/slab.h>
#include <linux/string.h>
#include <asm/scatterlist.h>
#include "internal.h"
#include "scatterwalk.h"

static inline void xor_64(u8 *a, const u8 *b)
{
	((u32 *)a)[0] ^= ((u32 *)b)[0];
	((u32 *)a)[1] ^= ((u32 *)b)[1];
}

static inline void xor_128(u8 *a, const u8 *b)
{
	((u32 *)a)[0] ^= ((u32 *)b)[0];
	((u32 *)a)[1] ^= ((u32 *)b)[1];
	((u32 *)a)[2] ^= ((u32 *)b)[2];
	((u32 *)a)[3] ^= ((u32 *)b)[3];
}

static unsigned int crypt_slow(const struct cipher_desc *desc,
			       struct scatter_walk *in,
			       struct scatter_walk *out, unsigned int bsize)
{
	unsigned long alignmask = crypto_tfm_alg_alignmask(desc->tfm);
	u8 buffer[bsize * 2 + alignmask];
	u8 *src = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1);
	u8 *dst = src + bsize;
	unsigned int n;

	n = scatterwalk_copychunks(src, in, bsize, 0);
	scatterwalk_advance(in, n);

	desc->prfn(desc, dst, src, bsize);

	n = scatterwalk_copychunks(dst, out, bsize, 1);
	scatterwalk_advance(out, n);

	return bsize;
}

static inline unsigned int crypt_fast(const struct cipher_desc *desc,
				      struct scatter_walk *in,
				      struct scatter_walk *out,
				      unsigned int nbytes, u8 *tmp)
{
	u8 *src, *dst;

	src = in->data;
	dst = scatterwalk_samebuf(in, out) ? src : out->data;

	if (tmp) {
		memcpy(tmp, in->data, nbytes);
		src = tmp;
		dst = tmp;
	}

	nbytes = desc->prfn(desc, dst, src, nbytes);

	if (tmp)
		memcpy(out->data, tmp, nbytes);

	scatterwalk_advance(in, nbytes);
	scatterwalk_advance(out, nbytes);

	return nbytes;
}

/* 
 * Generic encrypt/decrypt wrapper for ciphers, handles operations across
 * multiple page boundaries by using temporary blocks.  In user context,
 * the kernel is given a chance to schedule us once per page.
 */
static int crypt(const struct cipher_desc *desc,
		 struct scatterlist *dst,
		 struct scatterlist *src,
		 unsigned int nbytes)
{
	struct scatter_walk walk_in, walk_out;
	struct crypto_tfm *tfm = desc->tfm;
	const unsigned int bsize = crypto_tfm_alg_blocksize(tfm);
	unsigned int alignmask = crypto_tfm_alg_alignmask(tfm);
	unsigned long buffer = 0;

	if (!nbytes)
		return 0;

	if (nbytes % bsize) {
		tfm->crt_flags |= CRYPTO_TFM_RES_BAD_BLOCK_LEN;
		return -EINVAL;
	}

	scatterwalk_start(&walk_in, src);
	scatterwalk_start(&walk_out, dst);

	for(;;) {
		unsigned int n = nbytes;
		u8 *tmp = NULL;

		if (!scatterwalk_aligned(&walk_in, alignmask) ||
		    !scatterwalk_aligned(&walk_out, alignmask)) {
			if (!buffer) {
				buffer = __get_free_page(GFP_ATOMIC);
				if (!buffer)
					n = 0;
			}
			tmp = (u8 *)buffer;
		}

		scatterwalk_map(&walk_in, 0);
		scatterwalk_map(&walk_out, 1);

		n = scatterwalk_clamp(&walk_in, n);
		n = scatterwalk_clamp(&walk_out, n);

		if (likely(n >= bsize))
			n = crypt_fast(desc, &walk_in, &walk_out, n, tmp);
		else
			n = crypt_slow(desc, &walk_in, &walk_out, bsize);

		nbytes -= n;

		scatterwalk_done(&walk_in, 0, nbytes);
		scatterwalk_done(&walk_out, 1, nbytes);

		if (!nbytes)
			break;

		crypto_yield(tfm->crt_flags);
	}

	if (buffer)
		free_page(buffer);

	return 0;
}

static int crypt_iv_unaligned(struct cipher_desc *desc,
			      struct scatterlist *dst,
			      struct scatterlist *src,
			      unsigned int nbytes)
{
	struct crypto_tfm *tfm = desc->tfm;
	unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
	u8 *iv = desc->info;

	if (unlikely(((unsigned long)iv & alignmask))) {
		unsigned int ivsize = tfm->crt_cipher.cit_ivsize;
		u8 buffer[ivsize + alignmask];
		u8 *tmp = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1);
		int err;

		desc->info = memcpy(tmp, iv, ivsize);
		err = crypt(desc, dst, src, nbytes);
		memcpy(iv, tmp, ivsize);

		return err;
	}

	return crypt(desc, dst, src, nbytes);
}

static unsigned int cbc_process_encrypt(const struct cipher_desc *desc,
					u8 *dst, const u8 *src,
					unsigned int nbytes)
{
	struct crypto_tfm *tfm = desc->tfm;
	void (*xor)(u8 *, const u8 *) = tfm->crt_u.cipher.cit_xor_block;
	int bsize = crypto_tfm_alg_blocksize(tfm);

	void (*fn)(struct crypto_tfm *, u8 *, const u8 *) = desc->crfn;
	u8 *iv = desc->info;
	unsigned int done = 0;

	nbytes -= bsize;

	do {
		xor(iv, src);
		fn(tfm, dst, iv);
		memcpy(iv, dst, bsize);

		src += bsize;
		dst += bsize;
	} while ((done += bsize) <= nbytes);

	return done;
}

static unsigned int cbc_process_decrypt(const struct cipher_desc *desc,
					u8 *dst, const u8 *src,
					unsigned int nbytes)
{
	struct crypto_tfm *tfm = desc->tfm;
	void (*xor)(u8 *, const u8 *) = tfm->crt_u.cipher.cit_xor_block;
	int bsize = crypto_tfm_alg_blocksize(tfm);
	unsigned long alignmask = crypto_tfm_alg_alignmask(desc->tfm);

	u8 stack[src == dst ? bsize + alignmask : 0];
	u8 *buf = (u8 *)ALIGN((unsigned long)stack, alignmask + 1);
	u8 **dst_p = src == dst ? &buf : &dst;

	void (*fn)(struct crypto_tfm *, u8 *, const u8 *) = desc->crfn;
	u8 *iv = desc->info;
	unsigned int done = 0;

	nbytes -= bsize;

	do {
		u8 *tmp_dst = *dst_p;

		fn(tfm, tmp_dst, src);
		xor(tmp_dst, iv);
		memcpy(iv, src, bsize);
		if (tmp_dst != dst)
			memcpy(dst, tmp_dst, bsize);

		src += bsize;
		dst += bsize;
	} while ((done += bsize) <= nbytes);

	return done;
}

static unsigned int ecb_process(const struct cipher_desc *desc, u8 *dst,
				const u8 *src, unsigned int nbytes)
{
	struct crypto_tfm *tfm = desc->tfm;
	int bsize = crypto_tfm_alg_blocksize(tfm);
	void (*fn)(struct crypto_tfm *, u8 *, const u8 *) = desc->crfn;
	unsigned int done = 0;

	nbytes -= bsize;

	do {
		fn(tfm, dst, src);

		src += bsize;
		dst += bsize;
	} while ((done += bsize) <= nbytes);

	return done;
}

static int setkey(struct crypto_tfm *tfm, const u8 *key, unsigned int keylen)
{
	struct cipher_alg *cia = &tfm->__crt_alg->cra_cipher;
	
	tfm->crt_flags &= ~CRYPTO_TFM_RES_MASK;
	if (keylen < cia->cia_min_keysize || keylen > cia->cia_max_keysize) {
		tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN;
		return -EINVAL;
	} else
		return cia->cia_setkey(tfm, key, keylen);
}

static int ecb_encrypt(struct crypto_tfm *tfm,
		       struct scatterlist *dst,
                       struct scatterlist *src, unsigned int nbytes)
{
	struct cipher_desc desc;
	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;

	desc.tfm = tfm;
	desc.crfn = cipher->cia_encrypt;
	desc.prfn = cipher->cia_encrypt_ecb ?: ecb_process;

	return crypt(&desc, dst, src, nbytes);
}

static int ecb_decrypt(struct crypto_tfm *tfm,
                       struct scatterlist *dst,
                       struct scatterlist *src,
		       unsigned int nbytes)
{
	struct cipher_desc desc;
	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;

	desc.tfm = tfm;
	desc.crfn = cipher->cia_decrypt;
	desc.prfn = cipher->cia_decrypt_ecb ?: ecb_process;

	return crypt(&desc, dst, src, nbytes);
}

static int cbc_encrypt(struct crypto_tfm *tfm,
                       struct scatterlist *dst,
                       struct scatterlist *src,
		       unsigned int nbytes)
{
	struct cipher_desc desc;
	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;

	desc.tfm = tfm;
	desc.crfn = cipher->cia_encrypt;
	desc.prfn = cipher->cia_encrypt_cbc ?: cbc_process_encrypt;
	desc.info = tfm->crt_cipher.cit_iv;

	return crypt(&desc, dst, src, nbytes);
}

static int cbc_encrypt_iv(struct crypto_tfm *tfm,
                          struct scatterlist *dst,
                          struct scatterlist *src,
                          unsigned int nbytes, u8 *iv)
{
	struct cipher_desc desc;
	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;

	desc.tfm = tfm;
	desc.crfn = cipher->cia_encrypt;
	desc.prfn = cipher->cia_encrypt_cbc ?: cbc_process_encrypt;
	desc.info = iv;

	return crypt_iv_unaligned(&desc, dst, src, nbytes);
}

static int cbc_decrypt(struct crypto_tfm *tfm,
                       struct scatterlist *dst,
                       struct scatterlist *src,
		       unsigned int nbytes)
{
	struct cipher_desc desc;
	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;

	desc.tfm = tfm;
	desc.crfn = cipher->cia_decrypt;
	desc.prfn = cipher->cia_decrypt_cbc ?: cbc_process_decrypt;
	desc.info = tfm->crt_cipher.cit_iv;

	return crypt(&desc, dst, src, nbytes);
}

static int cbc_decrypt_iv(struct crypto_tfm *tfm,
                          struct scatterlist *dst,
                          struct scatterlist *src,
                          unsigned int nbytes, u8 *iv)
{
	struct cipher_desc desc;
	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;

	desc.tfm = tfm;
	desc.crfn = cipher->cia_decrypt;
	desc.prfn = cipher->cia_decrypt_cbc ?: cbc_process_decrypt;
	desc.info = iv;

	return crypt_iv_unaligned(&desc, dst, src, nbytes);
}

static int nocrypt(struct crypto_tfm *tfm,
                   struct scatterlist *dst,
                   struct scatterlist *src,
		   unsigned int nbytes)
{
	return -ENOSYS;
}

static int nocrypt_iv(struct crypto_tfm *tfm,
                      struct scatterlist *dst,
                      struct scatterlist *src,
                      unsigned int nbytes, u8 *iv)
{
	return -ENOSYS;
}

int crypto_init_cipher_flags(struct crypto_tfm *tfm, u32 flags)
{
	u32 mode = flags & CRYPTO_TFM_MODE_MASK;
	tfm->crt_cipher.cit_mode = mode ? mode : CRYPTO_TFM_MODE_ECB;
	return 0;
}

int crypto_init_cipher_ops(struct crypto_tfm *tfm)
{
	int ret = 0;
	struct cipher_tfm *ops = &tfm->crt_cipher;

	ops->cit_setkey = setkey;

	switch (tfm->crt_cipher.cit_mode) {
	case CRYPTO_TFM_MODE_ECB:
		ops->cit_encrypt = ecb_encrypt;
		ops->cit_decrypt = ecb_decrypt;
		ops->cit_encrypt_iv = nocrypt_iv;
		ops->cit_decrypt_iv = nocrypt_iv;
		break;
		
	case CRYPTO_TFM_MODE_CBC:
		ops->cit_encrypt = cbc_encrypt;
		ops->cit_decrypt = cbc_decrypt;
		ops->cit_encrypt_iv = cbc_encrypt_iv;
		ops->cit_decrypt_iv = cbc_decrypt_iv;
		break;
		
	case CRYPTO_TFM_MODE_CFB:
		ops->cit_encrypt = nocrypt;
		ops->cit_decrypt = nocrypt;
		ops->cit_encrypt_iv = nocrypt_iv;
		ops->cit_decrypt_iv = nocrypt_iv;
		break;
	
	case CRYPTO_TFM_MODE_CTR:
		ops->cit_encrypt = nocrypt;
		ops->cit_decrypt = nocrypt;
		ops->cit_encrypt_iv = nocrypt_iv;
		ops->cit_decrypt_iv = nocrypt_iv;
		break;

	default:
		BUG();
	}
	
	if (ops->cit_mode == CRYPTO_TFM_MODE_CBC) {
		unsigned long align;
		unsigned long addr;
	    	
	    	switch (crypto_tfm_alg_blocksize(tfm)) {
	    	case 8:
	    		ops->cit_xor_block = xor_64;
	    		break;
	    		
	    	case 16:
	    		ops->cit_xor_block = xor_128;
	    		break;
	    		
	    	default:
	    		printk(KERN_WARNING "%s: block size %u not supported\n",
	    		       crypto_tfm_alg_name(tfm),
	    		       crypto_tfm_alg_blocksize(tfm));
	    		ret = -EINVAL;
	    		goto out;
	    	}
	    	
		ops->cit_ivsize = crypto_tfm_alg_blocksize(tfm);
		align = crypto_tfm_alg_alignmask(tfm) + 1;
		addr = (unsigned long)crypto_tfm_ctx(tfm);
		addr = ALIGN(addr, align);
		addr += ALIGN(tfm->__crt_alg->cra_ctxsize, align);
		ops->cit_iv = (void *)addr;
	}

out:	
	return ret;
}

void crypto_exit_cipher_ops(struct crypto_tfm *tfm)
{
}
Commit	Line	Data
1da177e4 LT	1	/*
	2	* Cryptographic API.
	3	*
	4	* Cipher operations.
	5	*
	6	* Copyright (c) 2002 James Morris <jmorris@intercode.com.au>
c774e93e	7	* Copyright (c) 2005 Herbert Xu <herbert@gondor.apana.org.au>
1da177e4 LT	8	*
	9	* This program is free software; you can redistribute it and/or modify it
	10	* under the terms of the GNU General Public License as published by the Free
	11	* Software Foundation; either version 2 of the License, or (at your option)
	12	* any later version.
	13	*
	14	*/
	15	#include <linux/compiler.h>
	16	#include <linux/kernel.h>
	17	#include <linux/crypto.h>
	18	#include <linux/errno.h>
	19	#include <linux/mm.h>
	20	#include <linux/slab.h>
	21	#include <linux/string.h>
	22	#include <asm/scatterlist.h>
	23	#include "internal.h"
	24	#include "scatterwalk.h"
	25
1da177e4 LT	26	static inline void xor_64(u8 a, const u8 b)
	27	{
	28	((u32 )a)[0] ^= ((u32 )b)[0];
	29	((u32 )a)[1] ^= ((u32 )b)[1];
	30	}
	31
	32	static inline void xor_128(u8 a, const u8 b)
	33	{
	34	((u32 )a)[0] ^= ((u32 )b)[0];
	35	((u32 )a)[1] ^= ((u32 )b)[1];
	36	((u32 )a)[2] ^= ((u32 )b)[2];
	37	((u32 )a)[3] ^= ((u32 )b)[3];
	38	}
c774e93e HX	39
	40	static unsigned int crypt_slow(const struct cipher_desc *desc,
	41	struct scatter_walk *in,
	42	struct scatter_walk *out, unsigned int bsize)
1da177e4	43	{
9d853c37	44	unsigned long alignmask = crypto_tfm_alg_alignmask(desc->tfm);
95477377 HX	45	u8 buffer[bsize * 2 + alignmask];
	46	u8 src = (u8 )ALIGN((unsigned long)buffer, alignmask + 1);
	47	u8 *dst = src + bsize;
c774e93e	48	unsigned int n;
1da177e4	49
c774e93e HX	50	n = scatterwalk_copychunks(src, in, bsize, 0);
c774e93e HX	51	scatterwalk_advance(in, n);
1da177e4	52
c774e93e	53	desc->prfn(desc, dst, src, bsize);
1da177e4	54
c774e93e HX	55	n = scatterwalk_copychunks(dst, out, bsize, 1);
c774e93e HX	56	scatterwalk_advance(out, n);
1da177e4	57
c774e93e	58	return bsize;
1da177e4 LT	59	}
1da177e4 LT	60
c774e93e HX	61	static inline unsigned int crypt_fast(const struct cipher_desc *desc,
	62	struct scatter_walk *in,
	63	struct scatter_walk *out,
95477377	64	unsigned int nbytes, u8 *tmp)
1da177e4	65	{
c774e93e HX	66	u8 src, dst;
	67
	68	src = in->data;
	69	dst = scatterwalk_samebuf(in, out) ? src : out->data;
	70
95477377 HX	71	if (tmp) {
	72	memcpy(tmp, in->data, nbytes);
	73	src = tmp;
	74	dst = tmp;
	75	}
	76
c774e93e HX	77	nbytes = desc->prfn(desc, dst, src, nbytes);
c774e93e HX	78
95477377 HX	79	if (tmp)
	80	memcpy(out->data, tmp, nbytes);
	81
c774e93e HX	82	scatterwalk_advance(in, nbytes);
c774e93e HX	83	scatterwalk_advance(out, nbytes);
1da177e4	84
c774e93e	85	return nbytes;
1da177e4 LT	86	}
	87
	88	/*
	89	* Generic encrypt/decrypt wrapper for ciphers, handles operations across
	90	* multiple page boundaries by using temporary blocks. In user context,
c774e93e	91	* the kernel is given a chance to schedule us once per page.
1da177e4	92	*/
c774e93e	93	static int crypt(const struct cipher_desc *desc,
1da177e4 LT	94	struct scatterlist *dst,
1da177e4 LT	95	struct scatterlist *src,
c774e93e	96	unsigned int nbytes)
1da177e4 LT	97	{
1da177e4 LT	98	struct scatter_walk walk_in, walk_out;
c774e93e	99	struct crypto_tfm *tfm = desc->tfm;
1da177e4	100	const unsigned int bsize = crypto_tfm_alg_blocksize(tfm);
fbdae9f3	101	unsigned int alignmask = crypto_tfm_alg_alignmask(tfm);
95477377	102	unsigned long buffer = 0;
1da177e4 LT	103
	104	if (!nbytes)
	105	return 0;
	106
	107	if (nbytes % bsize) {
	108	tfm->crt_flags \|= CRYPTO_TFM_RES_BAD_BLOCK_LEN;
	109	return -EINVAL;
	110	}
	111
	112	scatterwalk_start(&walk_in, src);
	113	scatterwalk_start(&walk_out, dst);
	114
	115	for(;;) {
95477377 HX	116	unsigned int n = nbytes;
	117	u8 *tmp = NULL;
	118
	119	if (!scatterwalk_aligned(&walk_in, alignmask) \|\|
	120	!scatterwalk_aligned(&walk_out, alignmask)) {
	121	if (!buffer) {
	122	buffer = __get_free_page(GFP_ATOMIC);
	123	if (!buffer)
	124	n = 0;
	125	}
	126	tmp = (u8 *)buffer;
	127	}
1da177e4 LT	128
	129	scatterwalk_map(&walk_in, 0);
	130	scatterwalk_map(&walk_out, 1);
	131
95477377	132	n = scatterwalk_clamp(&walk_in, n);
c774e93e	133	n = scatterwalk_clamp(&walk_out, n);
1da177e4	134
c774e93e	135	if (likely(n >= bsize))
95477377	136	n = crypt_fast(desc, &walk_in, &walk_out, n, tmp);
c774e93e HX	137	else
c774e93e HX	138	n = crypt_slow(desc, &walk_in, &walk_out, bsize);
1da177e4	139
c774e93e	140	nbytes -= n;
1da177e4 LT	141
	142	scatterwalk_done(&walk_in, 0, nbytes);
	143	scatterwalk_done(&walk_out, 1, nbytes);
	144
	145	if (!nbytes)
95477377	146	break;
1da177e4	147
8f21cf0d	148	crypto_yield(tfm->crt_flags);
1da177e4	149	}
95477377 HX	150
	151	if (buffer)
	152	free_page(buffer);
	153
	154	return 0;
1da177e4 LT	155	}
1da177e4 LT	156
915e8561 HX	157	static int crypt_iv_unaligned(struct cipher_desc *desc,
	158	struct scatterlist *dst,
	159	struct scatterlist *src,
	160	unsigned int nbytes)
	161	{
	162	struct crypto_tfm *tfm = desc->tfm;
9d853c37	163	unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
915e8561 HX	164	u8 *iv = desc->info;
	165
	166	if (unlikely(((unsigned long)iv & alignmask))) {
	167	unsigned int ivsize = tfm->crt_cipher.cit_ivsize;
	168	u8 buffer[ivsize + alignmask];
	169	u8 tmp = (u8 )ALIGN((unsigned long)buffer, alignmask + 1);
	170	int err;
	171
	172	desc->info = memcpy(tmp, iv, ivsize);
	173	err = crypt(desc, dst, src, nbytes);
	174	memcpy(iv, tmp, ivsize);
	175
	176	return err;
	177	}
	178
	179	return crypt(desc, dst, src, nbytes);
	180	}
	181
c774e93e HX	182	static unsigned int cbc_process_encrypt(const struct cipher_desc *desc,
	183	u8 dst, const u8 src,
	184	unsigned int nbytes)
1da177e4	185	{
c774e93e HX	186	struct crypto_tfm *tfm = desc->tfm;
	187	void (xor)(u8 , const u8 *) = tfm->crt_u.cipher.cit_xor_block;
	188	int bsize = crypto_tfm_alg_blocksize(tfm);
	189
6c2bb98b	190	void (fn)(struct crypto_tfm , u8 , const u8 ) = desc->crfn;
c774e93e HX	191	u8 *iv = desc->info;
	192	unsigned int done = 0;
	193
fe2d5295 HX	194	nbytes -= bsize;
fe2d5295 HX	195
c774e93e HX	196	do {
c774e93e HX	197	xor(iv, src);
6c2bb98b	198	fn(tfm, dst, iv);
c774e93e	199	memcpy(iv, dst, bsize);
1da177e4	200
c774e93e HX	201	src += bsize;
c774e93e HX	202	dst += bsize;
fe2d5295	203	} while ((done += bsize) <= nbytes);
c774e93e HX	204
c774e93e HX	205	return done;
1da177e4 LT	206	}
1da177e4 LT	207
c774e93e HX	208	static unsigned int cbc_process_decrypt(const struct cipher_desc *desc,
	209	u8 dst, const u8 src,
	210	unsigned int nbytes)
1da177e4	211	{
c774e93e HX	212	struct crypto_tfm *tfm = desc->tfm;
	213	void (xor)(u8 , const u8 *) = tfm->crt_u.cipher.cit_xor_block;
	214	int bsize = crypto_tfm_alg_blocksize(tfm);
827c3911	215	unsigned long alignmask = crypto_tfm_alg_alignmask(desc->tfm);
c774e93e	216
827c3911 HX	217	u8 stack[src == dst ? bsize + alignmask : 0];
827c3911 HX	218	u8 buf = (u8 )ALIGN((unsigned long)stack, alignmask + 1);
c774e93e HX	219	u8 **dst_p = src == dst ? &buf : &dst;
c774e93e HX	220
6c2bb98b	221	void (fn)(struct crypto_tfm , u8 , const u8 ) = desc->crfn;
c774e93e HX	222	u8 *iv = desc->info;
	223	unsigned int done = 0;
	224
fe2d5295 HX	225	nbytes -= bsize;
fe2d5295 HX	226
c774e93e HX	227	do {
c774e93e HX	228	u8 tmp_dst = dst_p;
1da177e4	229
6c2bb98b	230	fn(tfm, tmp_dst, src);
c774e93e HX	231	xor(tmp_dst, iv);
	232	memcpy(iv, src, bsize);
	233	if (tmp_dst != dst)
	234	memcpy(dst, tmp_dst, bsize);
	235
	236	src += bsize;
	237	dst += bsize;
fe2d5295	238	} while ((done += bsize) <= nbytes);
c774e93e HX	239
c774e93e HX	240	return done;
1da177e4 LT	241	}
1da177e4 LT	242
c774e93e HX	243	static unsigned int ecb_process(const struct cipher_desc desc, u8 dst,
c774e93e HX	244	const u8 *src, unsigned int nbytes)
1da177e4	245	{
c774e93e HX	246	struct crypto_tfm *tfm = desc->tfm;
c774e93e HX	247	int bsize = crypto_tfm_alg_blocksize(tfm);
6c2bb98b	248	void (fn)(struct crypto_tfm , u8 , const u8 ) = desc->crfn;
c774e93e HX	249	unsigned int done = 0;
c774e93e HX	250
fe2d5295 HX	251	nbytes -= bsize;
fe2d5295 HX	252
c774e93e	253	do {
6c2bb98b	254	fn(tfm, dst, src);
c774e93e HX	255
	256	src += bsize;
	257	dst += bsize;
fe2d5295	258	} while ((done += bsize) <= nbytes);
c774e93e HX	259
c774e93e HX	260	return done;
1da177e4 LT	261	}
	262
	263	static int setkey(struct crypto_tfm tfm, const u8 key, unsigned int keylen)
	264	{
	265	struct cipher_alg *cia = &tfm->__crt_alg->cra_cipher;
	266
560c06ae	267	tfm->crt_flags &= ~CRYPTO_TFM_RES_MASK;
1da177e4 LT	268	if (keylen < cia->cia_min_keysize \|\| keylen > cia->cia_max_keysize) {
	269	tfm->crt_flags \|= CRYPTO_TFM_RES_BAD_KEY_LEN;
	270	return -EINVAL;
	271	} else
560c06ae	272	return cia->cia_setkey(tfm, key, keylen);
1da177e4 LT	273	}
	274
	275	static int ecb_encrypt(struct crypto_tfm *tfm,
	276	struct scatterlist *dst,
	277	struct scatterlist *src, unsigned int nbytes)
	278	{
c774e93e	279	struct cipher_desc desc;
40725181	280	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	281
c774e93e HX	282	desc.tfm = tfm;
40725181 HX	283	desc.crfn = cipher->cia_encrypt;
40725181 HX	284	desc.prfn = cipher->cia_encrypt_ecb ?: ecb_process;
c774e93e HX	285
c774e93e HX	286	return crypt(&desc, dst, src, nbytes);
1da177e4 LT	287	}
	288
	289	static int ecb_decrypt(struct crypto_tfm *tfm,
	290	struct scatterlist *dst,
	291	struct scatterlist *src,
	292	unsigned int nbytes)
	293	{
c774e93e	294	struct cipher_desc desc;
40725181	295	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	296
c774e93e HX	297	desc.tfm = tfm;
40725181 HX	298	desc.crfn = cipher->cia_decrypt;
40725181 HX	299	desc.prfn = cipher->cia_decrypt_ecb ?: ecb_process;
c774e93e HX	300
c774e93e HX	301	return crypt(&desc, dst, src, nbytes);
1da177e4 LT	302	}
	303
	304	static int cbc_encrypt(struct crypto_tfm *tfm,
	305	struct scatterlist *dst,
	306	struct scatterlist *src,
	307	unsigned int nbytes)
	308	{
c774e93e	309	struct cipher_desc desc;
40725181	310	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	311
c774e93e HX	312	desc.tfm = tfm;
40725181 HX	313	desc.crfn = cipher->cia_encrypt;
40725181 HX	314	desc.prfn = cipher->cia_encrypt_cbc ?: cbc_process_encrypt;
c774e93e HX	315	desc.info = tfm->crt_cipher.cit_iv;
	316
	317	return crypt(&desc, dst, src, nbytes);
1da177e4 LT	318	}
	319
	320	static int cbc_encrypt_iv(struct crypto_tfm *tfm,
	321	struct scatterlist *dst,
	322	struct scatterlist *src,
	323	unsigned int nbytes, u8 *iv)
	324	{
c774e93e	325	struct cipher_desc desc;
40725181	326	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	327
c774e93e HX	328	desc.tfm = tfm;
40725181 HX	329	desc.crfn = cipher->cia_encrypt;
40725181 HX	330	desc.prfn = cipher->cia_encrypt_cbc ?: cbc_process_encrypt;
c774e93e HX	331	desc.info = iv;
c774e93e HX	332
915e8561	333	return crypt_iv_unaligned(&desc, dst, src, nbytes);
1da177e4 LT	334	}
	335
	336	static int cbc_decrypt(struct crypto_tfm *tfm,
	337	struct scatterlist *dst,
	338	struct scatterlist *src,
	339	unsigned int nbytes)
	340	{
c774e93e	341	struct cipher_desc desc;
40725181	342	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	343
c774e93e HX	344	desc.tfm = tfm;
40725181 HX	345	desc.crfn = cipher->cia_decrypt;
40725181 HX	346	desc.prfn = cipher->cia_decrypt_cbc ?: cbc_process_decrypt;
c774e93e HX	347	desc.info = tfm->crt_cipher.cit_iv;
	348
	349	return crypt(&desc, dst, src, nbytes);
1da177e4 LT	350	}
	351
	352	static int cbc_decrypt_iv(struct crypto_tfm *tfm,
	353	struct scatterlist *dst,
	354	struct scatterlist *src,
	355	unsigned int nbytes, u8 *iv)
	356	{
c774e93e	357	struct cipher_desc desc;
40725181	358	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	359
c774e93e HX	360	desc.tfm = tfm;
40725181 HX	361	desc.crfn = cipher->cia_decrypt;
40725181 HX	362	desc.prfn = cipher->cia_decrypt_cbc ?: cbc_process_decrypt;
c774e93e HX	363	desc.info = iv;
c774e93e HX	364
915e8561	365	return crypt_iv_unaligned(&desc, dst, src, nbytes);
1da177e4 LT	366	}
	367
	368	static int nocrypt(struct crypto_tfm *tfm,
	369	struct scatterlist *dst,
	370	struct scatterlist *src,
	371	unsigned int nbytes)
	372	{
	373	return -ENOSYS;
	374	}
	375
	376	static int nocrypt_iv(struct crypto_tfm *tfm,
	377	struct scatterlist *dst,
	378	struct scatterlist *src,
	379	unsigned int nbytes, u8 *iv)
	380	{
	381	return -ENOSYS;
	382	}
	383
	384	int crypto_init_cipher_flags(struct crypto_tfm *tfm, u32 flags)
	385	{
	386	u32 mode = flags & CRYPTO_TFM_MODE_MASK;
1da177e4	387	tfm->crt_cipher.cit_mode = mode ? mode : CRYPTO_TFM_MODE_ECB;
1da177e4 LT	388	return 0;
	389	}
	390
	391	int crypto_init_cipher_ops(struct crypto_tfm *tfm)
	392	{
	393	int ret = 0;
	394	struct cipher_tfm *ops = &tfm->crt_cipher;
	395
	396	ops->cit_setkey = setkey;
	397
	398	switch (tfm->crt_cipher.cit_mode) {
	399	case CRYPTO_TFM_MODE_ECB:
	400	ops->cit_encrypt = ecb_encrypt;
	401	ops->cit_decrypt = ecb_decrypt;
df89820e HX	402	ops->cit_encrypt_iv = nocrypt_iv;
df89820e HX	403	ops->cit_decrypt_iv = nocrypt_iv;
1da177e4 LT	404	break;
	405
	406	case CRYPTO_TFM_MODE_CBC:
	407	ops->cit_encrypt = cbc_encrypt;
	408	ops->cit_decrypt = cbc_decrypt;
	409	ops->cit_encrypt_iv = cbc_encrypt_iv;
	410	ops->cit_decrypt_iv = cbc_decrypt_iv;
	411	break;
	412
	413	case CRYPTO_TFM_MODE_CFB:
	414	ops->cit_encrypt = nocrypt;
	415	ops->cit_decrypt = nocrypt;
	416	ops->cit_encrypt_iv = nocrypt_iv;
	417	ops->cit_decrypt_iv = nocrypt_iv;
	418	break;
	419
	420	case CRYPTO_TFM_MODE_CTR:
	421	ops->cit_encrypt = nocrypt;
	422	ops->cit_decrypt = nocrypt;
	423	ops->cit_encrypt_iv = nocrypt_iv;
	424	ops->cit_decrypt_iv = nocrypt_iv;
	425	break;
	426
	427	default:
	428	BUG();
	429	}
	430
	431	if (ops->cit_mode == CRYPTO_TFM_MODE_CBC) {
9d853c37	432	unsigned long align;
fbdae9f3	433	unsigned long addr;
1da177e4 LT	434
	435	switch (crypto_tfm_alg_blocksize(tfm)) {
	436	case 8:
	437	ops->cit_xor_block = xor_64;
	438	break;
	439
	440	case 16:
	441	ops->cit_xor_block = xor_128;
	442	break;
	443
	444	default:
	445	printk(KERN_WARNING "%s: block size %u not supported\n",
	446	crypto_tfm_alg_name(tfm),
	447	crypto_tfm_alg_blocksize(tfm));
	448	ret = -EINVAL;
	449	goto out;
	450	}
	451
	452	ops->cit_ivsize = crypto_tfm_alg_blocksize(tfm);
fbdae9f3 HX	453	align = crypto_tfm_alg_alignmask(tfm) + 1;
	454	addr = (unsigned long)crypto_tfm_ctx(tfm);
	455	addr = ALIGN(addr, align);
	456	addr += ALIGN(tfm->__crt_alg->cra_ctxsize, align);
	457	ops->cit_iv = (void *)addr;
1da177e4 LT	458	}
	459
	460	out:
	461	return ret;
	462	}
	463
	464	void crypto_exit_cipher_ops(struct crypto_tfm *tfm)
	465	{
1da177e4	466	}