[CRYPTO] api: fix writting into unallocated memory in setkey_aligned
[linux-2.6.git] / crypto / cipher.c
1 /*
2  * Cryptographic API.
3  *
4  * Cipher operations.
5  *
6  * Copyright (c) 2002 James Morris <jmorris@intercode.com.au>
7  * Copyright (c) 2005 Herbert Xu <herbert@gondor.apana.org.au>
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
16 #include <linux/kernel.h>
17 #include <linux/crypto.h>
18 #include <linux/errno.h>
19 #include <linux/scatterlist.h>
20 #include <linux/string.h>
21 #include "internal.h"
22
23 static int setkey_unaligned(struct crypto_tfm *tfm, const u8 *key, unsigned int keylen)
24 {
25         struct cipher_alg *cia = &tfm->__crt_alg->cra_cipher;
26         unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
27         int ret;
28         u8 *buffer, *alignbuffer;
29         unsigned long absize;
30
31         absize = keylen + alignmask;
32         buffer = kmalloc(absize, GFP_ATOMIC);
33         if (!buffer)
34                 return -ENOMEM;
35
36         alignbuffer = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1);
37         memcpy(alignbuffer, key, keylen);
38         ret = cia->cia_setkey(tfm, alignbuffer, keylen);
39         memset(alignbuffer, 0, keylen);
40         kfree(buffer);
41         return ret;
42
43 }
44
45 static int setkey(struct crypto_tfm *tfm, const u8 *key, unsigned int keylen)
46 {
47         struct cipher_alg *cia = &tfm->__crt_alg->cra_cipher;
48         unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
49
50         tfm->crt_flags &= ~CRYPTO_TFM_RES_MASK;
51         if (keylen < cia->cia_min_keysize || keylen > cia->cia_max_keysize) {
52                 tfm->crt_flags |= CRYPTO_TFM_RES_BAD_KEY_LEN;
53                 return -EINVAL;
54         }
55
56         if ((unsigned long)key & alignmask)
57                 return setkey_unaligned(tfm, key, keylen);
58
59         return cia->cia_setkey(tfm, key, keylen);
60 }
61
62 static void cipher_crypt_unaligned(void (*fn)(struct crypto_tfm *, u8 *,
63                                               const u8 *),
64                                    struct crypto_tfm *tfm,
65                                    u8 *dst, const u8 *src)
66 {
67         unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
68         unsigned int size = crypto_tfm_alg_blocksize(tfm);
69         u8 buffer[size + alignmask];
70         u8 *tmp = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1);
71
72         memcpy(tmp, src, size);
73         fn(tfm, tmp, tmp);
74         memcpy(dst, tmp, size);
75 }
76
77 static void cipher_encrypt_unaligned(struct crypto_tfm *tfm,
78                                      u8 *dst, const u8 *src)
79 {
80         unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
81         struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
82
83         if (unlikely(((unsigned long)dst | (unsigned long)src) & alignmask)) {
84                 cipher_crypt_unaligned(cipher->cia_encrypt, tfm, dst, src);
85                 return;
86         }
87
88         cipher->cia_encrypt(tfm, dst, src);
89 }
90
91 static void cipher_decrypt_unaligned(struct crypto_tfm *tfm,
92                                      u8 *dst, const u8 *src)
93 {
94         unsigned long alignmask = crypto_tfm_alg_alignmask(tfm);
95         struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
96
97         if (unlikely(((unsigned long)dst | (unsigned long)src) & alignmask)) {
98                 cipher_crypt_unaligned(cipher->cia_decrypt, tfm, dst, src);
99                 return;
100         }
101
102         cipher->cia_decrypt(tfm, dst, src);
103 }
104
105 int crypto_init_cipher_ops(struct crypto_tfm *tfm)
106 {
107         struct cipher_tfm *ops = &tfm->crt_cipher;
108         struct cipher_alg *cipher = &tfm->__crt_alg->cra_cipher;
109
110         ops->cit_setkey = setkey;
111         ops->cit_encrypt_one = crypto_tfm_alg_alignmask(tfm) ?
112                 cipher_encrypt_unaligned : cipher->cia_encrypt;
113         ops->cit_decrypt_one = crypto_tfm_alg_alignmask(tfm) ?
114                 cipher_decrypt_unaligned : cipher->cia_decrypt;
115
116         return 0;
117 }
118
119 void crypto_exit_cipher_ops(struct crypto_tfm *tfm)
120 {
121 }