[deliverable/linux.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;

	scatterwalk_copychunks(src, in, bsize, 0);
	desc->prfn(desc, dst, src, bsize);
	scatterwalk_copychunks(dst, out, bsize, 1);

	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;
	u8 *real_src, *real_dst;

	real_src = scatterwalk_map(in, 0);
	real_dst = scatterwalk_map(out, 1);

	src = real_src;
	dst = scatterwalk_samebuf(in, out) ? src : real_dst;

	if (tmp) {
		memcpy(tmp, src, nbytes);
		src = tmp;
		dst = tmp;
	}

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

	if (tmp)
		memcpy(real_dst, tmp, nbytes);

	scatterwalk_unmap(real_src, 0);
	scatterwalk_unmap(real_dst, 1);

	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;
		}

		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;
}

static void cipher_crypt_unaligned(void (*fn)(struct crypto_tfm *, u8 *,
					      const u8 *),
				   struct crypto_tfm *tfm,
				   u8 *dst, const u8 *src)
{
	unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
	unsigned int size = crypto_tfm_alg_blocksize(tfm);
	u8 buffer[size + alignmask];
	u8 *tmp = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1);

	memcpy(tmp, src, size);
	fn(tfm, tmp, tmp);
	memcpy(dst, tmp, size);
}

static void cipher_encrypt_unaligned(struct crypto_tfm *tfm,
				     u8 *dst, const u8 *src)
{
	unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;

	if (unlikely(((unsigned long)dst | (unsigned long)src) & alignmask)) {
		cipher_crypt_unaligned(cipher->cia_encrypt, tfm, dst, src);
		return;
	}

	cipher->cia_encrypt(tfm, dst, src);
}

static void cipher_decrypt_unaligned(struct crypto_tfm *tfm,
				     u8 *dst, const u8 *src)
{
	unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;

	if (unlikely(((unsigned long)dst | (unsigned long)src) & alignmask)) {
		cipher_crypt_unaligned(cipher->cia_decrypt, tfm, dst, src);
		return;
	}

	cipher->cia_decrypt(tfm, dst, src);
}

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

	ops->cit_setkey = setkey;
	ops->cit_encrypt_one = crypto_tfm_alg_alignmask(tfm) ?
		cipher_encrypt_unaligned : cipher->cia_encrypt;
	ops->cit_decrypt_one = crypto_tfm_alg_alignmask(tfm) ?
		cipher_decrypt_unaligned : cipher->cia_decrypt;

	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;
1da177e4	48
5c64097a	49	scatterwalk_copychunks(src, in, bsize, 0);
c774e93e	50	desc->prfn(desc, dst, src, bsize);
5c64097a	51	scatterwalk_copychunks(dst, out, bsize, 1);
1da177e4	52
c774e93e	53	return bsize;
1da177e4 LT	54	}
1da177e4 LT	55
c774e93e HX	56	static inline unsigned int crypt_fast(const struct cipher_desc *desc,
	57	struct scatter_walk *in,
	58	struct scatter_walk *out,
95477377	59	unsigned int nbytes, u8 *tmp)
1da177e4	60	{
c774e93e	61	u8 src, dst;
5c64097a HX	62	u8 real_src, real_dst;
	63
	64	real_src = scatterwalk_map(in, 0);
	65	real_dst = scatterwalk_map(out, 1);
c774e93e	66
5c64097a HX	67	src = real_src;
5c64097a HX	68	dst = scatterwalk_samebuf(in, out) ? src : real_dst;
c774e93e	69
95477377	70	if (tmp) {
5c64097a	71	memcpy(tmp, src, nbytes);
95477377 HX	72	src = tmp;
	73	dst = tmp;
	74	}
	75
c774e93e HX	76	nbytes = desc->prfn(desc, dst, src, nbytes);
c774e93e HX	77
95477377	78	if (tmp)
5c64097a HX	79	memcpy(real_dst, tmp, nbytes);
	80
	81	scatterwalk_unmap(real_src, 0);
	82	scatterwalk_unmap(real_dst, 1);
95477377	83
c774e93e HX	84	scatterwalk_advance(in, nbytes);
c774e93e HX	85	scatterwalk_advance(out, nbytes);
1da177e4	86
c774e93e	87	return nbytes;
1da177e4 LT	88	}
	89
	90	/*
	91	* Generic encrypt/decrypt wrapper for ciphers, handles operations across
	92	* multiple page boundaries by using temporary blocks. In user context,
c774e93e	93	* the kernel is given a chance to schedule us once per page.
1da177e4	94	*/
c774e93e	95	static int crypt(const struct cipher_desc *desc,
1da177e4 LT	96	struct scatterlist *dst,
1da177e4 LT	97	struct scatterlist *src,
c774e93e	98	unsigned int nbytes)
1da177e4 LT	99	{
1da177e4 LT	100	struct scatter_walk walk_in, walk_out;
c774e93e	101	struct crypto_tfm *tfm = desc->tfm;
1da177e4	102	const unsigned int bsize = crypto_tfm_alg_blocksize(tfm);
fbdae9f3	103	unsigned int alignmask = crypto_tfm_alg_alignmask(tfm);
95477377	104	unsigned long buffer = 0;
1da177e4 LT	105
	106	if (!nbytes)
	107	return 0;
	108
	109	if (nbytes % bsize) {
	110	tfm->crt_flags \|= CRYPTO_TFM_RES_BAD_BLOCK_LEN;
	111	return -EINVAL;
	112	}
	113
	114	scatterwalk_start(&walk_in, src);
	115	scatterwalk_start(&walk_out, dst);
	116
	117	for(;;) {
95477377 HX	118	unsigned int n = nbytes;
	119	u8 *tmp = NULL;
	120
	121	if (!scatterwalk_aligned(&walk_in, alignmask) \|\|
	122	!scatterwalk_aligned(&walk_out, alignmask)) {
	123	if (!buffer) {
	124	buffer = __get_free_page(GFP_ATOMIC);
	125	if (!buffer)
	126	n = 0;
	127	}
	128	tmp = (u8 *)buffer;
	129	}
1da177e4	130
95477377	131	n = scatterwalk_clamp(&walk_in, n);
c774e93e	132	n = scatterwalk_clamp(&walk_out, n);
1da177e4	133
c774e93e	134	if (likely(n >= bsize))
95477377	135	n = crypt_fast(desc, &walk_in, &walk_out, n, tmp);
c774e93e HX	136	else
c774e93e HX	137	n = crypt_slow(desc, &walk_in, &walk_out, bsize);
1da177e4	138
c774e93e	139	nbytes -= n;
1da177e4 LT	140
	141	scatterwalk_done(&walk_in, 0, nbytes);
	142	scatterwalk_done(&walk_out, 1, nbytes);
	143
	144	if (!nbytes)
95477377	145	break;
1da177e4	146
8f21cf0d	147	crypto_yield(tfm->crt_flags);
1da177e4	148	}
95477377 HX	149
	150	if (buffer)
	151	free_page(buffer);
	152
	153	return 0;
1da177e4 LT	154	}
1da177e4 LT	155
915e8561 HX	156	static int crypt_iv_unaligned(struct cipher_desc *desc,
	157	struct scatterlist *dst,
	158	struct scatterlist *src,
	159	unsigned int nbytes)
	160	{
	161	struct crypto_tfm *tfm = desc->tfm;
9d853c37	162	unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
915e8561 HX	163	u8 *iv = desc->info;
	164
	165	if (unlikely(((unsigned long)iv & alignmask))) {
	166	unsigned int ivsize = tfm->crt_cipher.cit_ivsize;
	167	u8 buffer[ivsize + alignmask];
	168	u8 tmp = (u8 )ALIGN((unsigned long)buffer, alignmask + 1);
	169	int err;
	170
	171	desc->info = memcpy(tmp, iv, ivsize);
	172	err = crypt(desc, dst, src, nbytes);
	173	memcpy(iv, tmp, ivsize);
	174
	175	return err;
	176	}
	177
	178	return crypt(desc, dst, src, nbytes);
	179	}
	180
c774e93e HX	181	static unsigned int cbc_process_encrypt(const struct cipher_desc *desc,
	182	u8 dst, const u8 src,
	183	unsigned int nbytes)
1da177e4	184	{
c774e93e HX	185	struct crypto_tfm *tfm = desc->tfm;
	186	void (xor)(u8 , const u8 *) = tfm->crt_u.cipher.cit_xor_block;
	187	int bsize = crypto_tfm_alg_blocksize(tfm);
	188
6c2bb98b	189	void (fn)(struct crypto_tfm , u8 , const u8 ) = desc->crfn;
c774e93e HX	190	u8 *iv = desc->info;
	191	unsigned int done = 0;
	192
fe2d5295 HX	193	nbytes -= bsize;
fe2d5295 HX	194
c774e93e HX	195	do {
c774e93e HX	196	xor(iv, src);
6c2bb98b	197	fn(tfm, dst, iv);
c774e93e	198	memcpy(iv, dst, bsize);
1da177e4	199
c774e93e HX	200	src += bsize;
c774e93e HX	201	dst += bsize;
fe2d5295	202	} while ((done += bsize) <= nbytes);
c774e93e HX	203
c774e93e HX	204	return done;
1da177e4 LT	205	}
1da177e4 LT	206
c774e93e HX	207	static unsigned int cbc_process_decrypt(const struct cipher_desc *desc,
	208	u8 dst, const u8 src,
	209	unsigned int nbytes)
1da177e4	210	{
c774e93e HX	211	struct crypto_tfm *tfm = desc->tfm;
	212	void (xor)(u8 , const u8 *) = tfm->crt_u.cipher.cit_xor_block;
	213	int bsize = crypto_tfm_alg_blocksize(tfm);
827c3911	214	unsigned long alignmask = crypto_tfm_alg_alignmask(desc->tfm);
c774e93e	215
827c3911 HX	216	u8 stack[src == dst ? bsize + alignmask : 0];
827c3911 HX	217	u8 buf = (u8 )ALIGN((unsigned long)stack, alignmask + 1);
c774e93e HX	218	u8 **dst_p = src == dst ? &buf : &dst;
c774e93e HX	219
6c2bb98b	220	void (fn)(struct crypto_tfm , u8 , const u8 ) = desc->crfn;
c774e93e HX	221	u8 *iv = desc->info;
	222	unsigned int done = 0;
	223
fe2d5295 HX	224	nbytes -= bsize;
fe2d5295 HX	225
c774e93e HX	226	do {
c774e93e HX	227	u8 tmp_dst = dst_p;
1da177e4	228
6c2bb98b	229	fn(tfm, tmp_dst, src);
c774e93e HX	230	xor(tmp_dst, iv);
	231	memcpy(iv, src, bsize);
	232	if (tmp_dst != dst)
	233	memcpy(dst, tmp_dst, bsize);
	234
	235	src += bsize;
	236	dst += bsize;
fe2d5295	237	} while ((done += bsize) <= nbytes);
c774e93e HX	238
c774e93e HX	239	return done;
1da177e4 LT	240	}
1da177e4 LT	241
c774e93e HX	242	static unsigned int ecb_process(const struct cipher_desc desc, u8 dst,
c774e93e HX	243	const u8 *src, unsigned int nbytes)
1da177e4	244	{
c774e93e HX	245	struct crypto_tfm *tfm = desc->tfm;
c774e93e HX	246	int bsize = crypto_tfm_alg_blocksize(tfm);
6c2bb98b	247	void (fn)(struct crypto_tfm , u8 , const u8 ) = desc->crfn;
c774e93e HX	248	unsigned int done = 0;
c774e93e HX	249
fe2d5295 HX	250	nbytes -= bsize;
fe2d5295 HX	251
c774e93e	252	do {
6c2bb98b	253	fn(tfm, dst, src);
c774e93e HX	254
	255	src += bsize;
	256	dst += bsize;
fe2d5295	257	} while ((done += bsize) <= nbytes);
c774e93e HX	258
c774e93e HX	259	return done;
1da177e4 LT	260	}
	261
	262	static int setkey(struct crypto_tfm tfm, const u8 key, unsigned int keylen)
	263	{
	264	struct cipher_alg *cia = &tfm->__crt_alg->cra_cipher;
	265
560c06ae	266	tfm->crt_flags &= ~CRYPTO_TFM_RES_MASK;
1da177e4 LT	267	if (keylen < cia->cia_min_keysize \|\| keylen > cia->cia_max_keysize) {
	268	tfm->crt_flags \|= CRYPTO_TFM_RES_BAD_KEY_LEN;
	269	return -EINVAL;
	270	} else
560c06ae	271	return cia->cia_setkey(tfm, key, keylen);
1da177e4 LT	272	}
	273
	274	static int ecb_encrypt(struct crypto_tfm *tfm,
	275	struct scatterlist *dst,
	276	struct scatterlist *src, unsigned int nbytes)
	277	{
c774e93e	278	struct cipher_desc desc;
40725181	279	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	280
c774e93e HX	281	desc.tfm = tfm;
40725181 HX	282	desc.crfn = cipher->cia_encrypt;
40725181 HX	283	desc.prfn = cipher->cia_encrypt_ecb ?: ecb_process;
c774e93e HX	284
c774e93e HX	285	return crypt(&desc, dst, src, nbytes);
1da177e4 LT	286	}
	287
	288	static int ecb_decrypt(struct crypto_tfm *tfm,
	289	struct scatterlist *dst,
	290	struct scatterlist *src,
	291	unsigned int nbytes)
	292	{
c774e93e	293	struct cipher_desc desc;
40725181	294	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	295
c774e93e HX	296	desc.tfm = tfm;
40725181 HX	297	desc.crfn = cipher->cia_decrypt;
40725181 HX	298	desc.prfn = cipher->cia_decrypt_ecb ?: ecb_process;
c774e93e HX	299
c774e93e HX	300	return crypt(&desc, dst, src, nbytes);
1da177e4 LT	301	}
	302
	303	static int cbc_encrypt(struct crypto_tfm *tfm,
	304	struct scatterlist *dst,
	305	struct scatterlist *src,
	306	unsigned int nbytes)
	307	{
c774e93e	308	struct cipher_desc desc;
40725181	309	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	310
c774e93e HX	311	desc.tfm = tfm;
40725181 HX	312	desc.crfn = cipher->cia_encrypt;
40725181 HX	313	desc.prfn = cipher->cia_encrypt_cbc ?: cbc_process_encrypt;
c774e93e HX	314	desc.info = tfm->crt_cipher.cit_iv;
	315
	316	return crypt(&desc, dst, src, nbytes);
1da177e4 LT	317	}
	318
	319	static int cbc_encrypt_iv(struct crypto_tfm *tfm,
	320	struct scatterlist *dst,
	321	struct scatterlist *src,
	322	unsigned int nbytes, u8 *iv)
	323	{
c774e93e	324	struct cipher_desc desc;
40725181	325	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	326
c774e93e HX	327	desc.tfm = tfm;
40725181 HX	328	desc.crfn = cipher->cia_encrypt;
40725181 HX	329	desc.prfn = cipher->cia_encrypt_cbc ?: cbc_process_encrypt;
c774e93e HX	330	desc.info = iv;
c774e93e HX	331
915e8561	332	return crypt_iv_unaligned(&desc, dst, src, nbytes);
1da177e4 LT	333	}
	334
	335	static int cbc_decrypt(struct crypto_tfm *tfm,
	336	struct scatterlist *dst,
	337	struct scatterlist *src,
	338	unsigned int nbytes)
	339	{
c774e93e	340	struct cipher_desc desc;
40725181	341	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	342
c774e93e HX	343	desc.tfm = tfm;
40725181 HX	344	desc.crfn = cipher->cia_decrypt;
40725181 HX	345	desc.prfn = cipher->cia_decrypt_cbc ?: cbc_process_decrypt;
c774e93e HX	346	desc.info = tfm->crt_cipher.cit_iv;
	347
	348	return crypt(&desc, dst, src, nbytes);
1da177e4 LT	349	}
	350
	351	static int cbc_decrypt_iv(struct crypto_tfm *tfm,
	352	struct scatterlist *dst,
	353	struct scatterlist *src,
	354	unsigned int nbytes, u8 *iv)
	355	{
c774e93e	356	struct cipher_desc desc;
40725181	357	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
c774e93e HX	358
c774e93e HX	359	desc.tfm = tfm;
40725181 HX	360	desc.crfn = cipher->cia_decrypt;
40725181 HX	361	desc.prfn = cipher->cia_decrypt_cbc ?: cbc_process_decrypt;
c774e93e HX	362	desc.info = iv;
c774e93e HX	363
915e8561	364	return crypt_iv_unaligned(&desc, dst, src, nbytes);
1da177e4 LT	365	}
	366
	367	static int nocrypt(struct crypto_tfm *tfm,
	368	struct scatterlist *dst,
	369	struct scatterlist *src,
	370	unsigned int nbytes)
	371	{
	372	return -ENOSYS;
	373	}
	374
	375	static int nocrypt_iv(struct crypto_tfm *tfm,
	376	struct scatterlist *dst,
	377	struct scatterlist *src,
	378	unsigned int nbytes, u8 *iv)
	379	{
	380	return -ENOSYS;
	381	}
	382
	383	int crypto_init_cipher_flags(struct crypto_tfm *tfm, u32 flags)
	384	{
	385	u32 mode = flags & CRYPTO_TFM_MODE_MASK;
1da177e4	386	tfm->crt_cipher.cit_mode = mode ? mode : CRYPTO_TFM_MODE_ECB;
1da177e4 LT	387	return 0;
	388	}
	389
f28776a3 HX	390	static void cipher_crypt_unaligned(void (fn)(struct crypto_tfm , u8 *,
	391	const u8 *),
	392	struct crypto_tfm *tfm,
	393	u8 dst, const u8 src)
	394	{
	395	unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
	396	unsigned int size = crypto_tfm_alg_blocksize(tfm);
	397	u8 buffer[size + alignmask];
	398	u8 tmp = (u8 )ALIGN((unsigned long)buffer, alignmask + 1);
	399
	400	memcpy(tmp, src, size);
	401	fn(tfm, tmp, tmp);
	402	memcpy(dst, tmp, size);
	403	}
	404
	405	static void cipher_encrypt_unaligned(struct crypto_tfm *tfm,
	406	u8 dst, const u8 src)
	407	{
	408	unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
	409	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
	410
	411	if (unlikely(((unsigned long)dst \| (unsigned long)src) & alignmask)) {
	412	cipher_crypt_unaligned(cipher->cia_encrypt, tfm, dst, src);
	413	return;
	414	}
	415
	416	cipher->cia_encrypt(tfm, dst, src);
	417	}
	418
	419	static void cipher_decrypt_unaligned(struct crypto_tfm *tfm,
	420	u8 dst, const u8 src)
	421	{
	422	unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
	423	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
	424
	425	if (unlikely(((unsigned long)dst \| (unsigned long)src) & alignmask)) {
	426	cipher_crypt_unaligned(cipher->cia_decrypt, tfm, dst, src);
	427	return;
	428	}
	429
	430	cipher->cia_decrypt(tfm, dst, src);
	431	}
	432
1da177e4 LT	433	int crypto_init_cipher_ops(struct crypto_tfm *tfm)
	434	{
	435	int ret = 0;
	436	struct cipher_tfm *ops = &tfm->crt_cipher;
f28776a3	437	struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
1da177e4 LT	438
1da177e4 LT	439	ops->cit_setkey = setkey;
f28776a3 HX	440	ops->cit_encrypt_one = crypto_tfm_alg_alignmask(tfm) ?
	441	cipher_encrypt_unaligned : cipher->cia_encrypt;
	442	ops->cit_decrypt_one = crypto_tfm_alg_alignmask(tfm) ?
	443	cipher_decrypt_unaligned : cipher->cia_decrypt;
1da177e4 LT	444
	445	switch (tfm->crt_cipher.cit_mode) {
	446	case CRYPTO_TFM_MODE_ECB:
	447	ops->cit_encrypt = ecb_encrypt;
	448	ops->cit_decrypt = ecb_decrypt;
df89820e HX	449	ops->cit_encrypt_iv = nocrypt_iv;
df89820e HX	450	ops->cit_decrypt_iv = nocrypt_iv;
1da177e4 LT	451	break;
	452
	453	case CRYPTO_TFM_MODE_CBC:
	454	ops->cit_encrypt = cbc_encrypt;
	455	ops->cit_decrypt = cbc_decrypt;
	456	ops->cit_encrypt_iv = cbc_encrypt_iv;
	457	ops->cit_decrypt_iv = cbc_decrypt_iv;
	458	break;
	459
	460	case CRYPTO_TFM_MODE_CFB:
	461	ops->cit_encrypt = nocrypt;
	462	ops->cit_decrypt = nocrypt;
	463	ops->cit_encrypt_iv = nocrypt_iv;
	464	ops->cit_decrypt_iv = nocrypt_iv;
	465	break;
	466
	467	case CRYPTO_TFM_MODE_CTR:
	468	ops->cit_encrypt = nocrypt;
	469	ops->cit_decrypt = nocrypt;
	470	ops->cit_encrypt_iv = nocrypt_iv;
	471	ops->cit_decrypt_iv = nocrypt_iv;
	472	break;
	473
	474	default:
	475	BUG();
	476	}
	477
	478	if (ops->cit_mode == CRYPTO_TFM_MODE_CBC) {
9d853c37	479	unsigned long align;
fbdae9f3	480	unsigned long addr;
1da177e4 LT	481
	482	switch (crypto_tfm_alg_blocksize(tfm)) {
	483	case 8:
	484	ops->cit_xor_block = xor_64;
	485	break;
	486
	487	case 16:
	488	ops->cit_xor_block = xor_128;
	489	break;
	490
	491	default:
	492	printk(KERN_WARNING "%s: block size %u not supported\n",
	493	crypto_tfm_alg_name(tfm),
	494	crypto_tfm_alg_blocksize(tfm));
	495	ret = -EINVAL;
	496	goto out;
	497	}
	498
	499	ops->cit_ivsize = crypto_tfm_alg_blocksize(tfm);
fbdae9f3 HX	500	align = crypto_tfm_alg_alignmask(tfm) + 1;
	501	addr = (unsigned long)crypto_tfm_ctx(tfm);
	502	addr = ALIGN(addr, align);
	503	addr += ALIGN(tfm->__crt_alg->cra_ctxsize, align);
	504	ops->cit_iv = (void *)addr;
1da177e4 LT	505	}
	506
	507	out:
	508	return ret;
	509	}
	510
	511	void crypto_exit_cipher_ops(struct crypto_tfm *tfm)
	512	{
1da177e4	513	}