regulator: check for init_data on registration
[linux-2.6] / crypto / cbc.c
1 /*
2  * CBC: Cipher Block Chaining mode
3  *
4  * Copyright (c) 2006 Herbert Xu <herbert@gondor.apana.org.au>
5  *
6  * This program is free software; you can redistribute it and/or modify it
7  * under the terms of the GNU General Public License as published by the Free
8  * Software Foundation; either version 2 of the License, or (at your option)
9  * any later version.
10  *
11  */
12
13 #include <crypto/algapi.h>
14 #include <linux/err.h>
15 #include <linux/init.h>
16 #include <linux/kernel.h>
17 #include <linux/log2.h>
18 #include <linux/module.h>
19 #include <linux/scatterlist.h>
20 #include <linux/slab.h>
21
22 struct crypto_cbc_ctx {
23         struct crypto_cipher *child;
24 };
25
26 static int crypto_cbc_setkey(struct crypto_tfm *parent, const u8 *key,
27                              unsigned int keylen)
28 {
29         struct crypto_cbc_ctx *ctx = crypto_tfm_ctx(parent);
30         struct crypto_cipher *child = ctx->child;
31         int err;
32
33         crypto_cipher_clear_flags(child, CRYPTO_TFM_REQ_MASK);
34         crypto_cipher_set_flags(child, crypto_tfm_get_flags(parent) &
35                                        CRYPTO_TFM_REQ_MASK);
36         err = crypto_cipher_setkey(child, key, keylen);
37         crypto_tfm_set_flags(parent, crypto_cipher_get_flags(child) &
38                                      CRYPTO_TFM_RES_MASK);
39         return err;
40 }
41
42 static int crypto_cbc_encrypt_segment(struct blkcipher_desc *desc,
43                                       struct blkcipher_walk *walk,
44                                       struct crypto_cipher *tfm)
45 {
46         void (*fn)(struct crypto_tfm *, u8 *, const u8 *) =
47                 crypto_cipher_alg(tfm)->cia_encrypt;
48         int bsize = crypto_cipher_blocksize(tfm);
49         unsigned int nbytes = walk->nbytes;
50         u8 *src = walk->src.virt.addr;
51         u8 *dst = walk->dst.virt.addr;
52         u8 *iv = walk->iv;
53
54         do {
55                 crypto_xor(iv, src, bsize);
56                 fn(crypto_cipher_tfm(tfm), dst, iv);
57                 memcpy(iv, dst, bsize);
58
59                 src += bsize;
60                 dst += bsize;
61         } while ((nbytes -= bsize) >= bsize);
62
63         return nbytes;
64 }
65
66 static int crypto_cbc_encrypt_inplace(struct blkcipher_desc *desc,
67                                       struct blkcipher_walk *walk,
68                                       struct crypto_cipher *tfm)
69 {
70         void (*fn)(struct crypto_tfm *, u8 *, const u8 *) =
71                 crypto_cipher_alg(tfm)->cia_encrypt;
72         int bsize = crypto_cipher_blocksize(tfm);
73         unsigned int nbytes = walk->nbytes;
74         u8 *src = walk->src.virt.addr;
75         u8 *iv = walk->iv;
76
77         do {
78                 crypto_xor(src, iv, bsize);
79                 fn(crypto_cipher_tfm(tfm), src, src);
80                 iv = src;
81
82                 src += bsize;
83         } while ((nbytes -= bsize) >= bsize);
84
85         memcpy(walk->iv, iv, bsize);
86
87         return nbytes;
88 }
89
90 static int crypto_cbc_encrypt(struct blkcipher_desc *desc,
91                               struct scatterlist *dst, struct scatterlist *src,
92                               unsigned int nbytes)
93 {
94         struct blkcipher_walk walk;
95         struct crypto_blkcipher *tfm = desc->tfm;
96         struct crypto_cbc_ctx *ctx = crypto_blkcipher_ctx(tfm);
97         struct crypto_cipher *child = ctx->child;
98         int err;
99
100         blkcipher_walk_init(&walk, dst, src, nbytes);
101         err = blkcipher_walk_virt(desc, &walk);
102
103         while ((nbytes = walk.nbytes)) {
104                 if (walk.src.virt.addr == walk.dst.virt.addr)
105                         nbytes = crypto_cbc_encrypt_inplace(desc, &walk, child);
106                 else
107                         nbytes = crypto_cbc_encrypt_segment(desc, &walk, child);
108                 err = blkcipher_walk_done(desc, &walk, nbytes);
109         }
110
111         return err;
112 }
113
114 static int crypto_cbc_decrypt_segment(struct blkcipher_desc *desc,
115                                       struct blkcipher_walk *walk,
116                                       struct crypto_cipher *tfm)
117 {
118         void (*fn)(struct crypto_tfm *, u8 *, const u8 *) =
119                 crypto_cipher_alg(tfm)->cia_decrypt;
120         int bsize = crypto_cipher_blocksize(tfm);
121         unsigned int nbytes = walk->nbytes;
122         u8 *src = walk->src.virt.addr;
123         u8 *dst = walk->dst.virt.addr;
124         u8 *iv = walk->iv;
125
126         do {
127                 fn(crypto_cipher_tfm(tfm), dst, src);
128                 crypto_xor(dst, iv, bsize);
129                 iv = src;
130
131                 src += bsize;
132                 dst += bsize;
133         } while ((nbytes -= bsize) >= bsize);
134
135         memcpy(walk->iv, iv, bsize);
136
137         return nbytes;
138 }
139
140 static int crypto_cbc_decrypt_inplace(struct blkcipher_desc *desc,
141                                       struct blkcipher_walk *walk,
142                                       struct crypto_cipher *tfm)
143 {
144         void (*fn)(struct crypto_tfm *, u8 *, const u8 *) =
145                 crypto_cipher_alg(tfm)->cia_decrypt;
146         int bsize = crypto_cipher_blocksize(tfm);
147         unsigned int nbytes = walk->nbytes;
148         u8 *src = walk->src.virt.addr;
149         u8 last_iv[bsize];
150
151         /* Start of the last block. */
152         src += nbytes - (nbytes & (bsize - 1)) - bsize;
153         memcpy(last_iv, src, bsize);
154
155         for (;;) {
156                 fn(crypto_cipher_tfm(tfm), src, src);
157                 if ((nbytes -= bsize) < bsize)
158                         break;
159                 crypto_xor(src, src - bsize, bsize);
160                 src -= bsize;
161         }
162
163         crypto_xor(src, walk->iv, bsize);
164         memcpy(walk->iv, last_iv, bsize);
165
166         return nbytes;
167 }
168
169 static int crypto_cbc_decrypt(struct blkcipher_desc *desc,
170                               struct scatterlist *dst, struct scatterlist *src,
171                               unsigned int nbytes)
172 {
173         struct blkcipher_walk walk;
174         struct crypto_blkcipher *tfm = desc->tfm;
175         struct crypto_cbc_ctx *ctx = crypto_blkcipher_ctx(tfm);
176         struct crypto_cipher *child = ctx->child;
177         int err;
178
179         blkcipher_walk_init(&walk, dst, src, nbytes);
180         err = blkcipher_walk_virt(desc, &walk);
181
182         while ((nbytes = walk.nbytes)) {
183                 if (walk.src.virt.addr == walk.dst.virt.addr)
184                         nbytes = crypto_cbc_decrypt_inplace(desc, &walk, child);
185                 else
186                         nbytes = crypto_cbc_decrypt_segment(desc, &walk, child);
187                 err = blkcipher_walk_done(desc, &walk, nbytes);
188         }
189
190         return err;
191 }
192
193 static int crypto_cbc_init_tfm(struct crypto_tfm *tfm)
194 {
195         struct crypto_instance *inst = (void *)tfm->__crt_alg;
196         struct crypto_spawn *spawn = crypto_instance_ctx(inst);
197         struct crypto_cbc_ctx *ctx = crypto_tfm_ctx(tfm);
198         struct crypto_cipher *cipher;
199
200         cipher = crypto_spawn_cipher(spawn);
201         if (IS_ERR(cipher))
202                 return PTR_ERR(cipher);
203
204         ctx->child = cipher;
205         return 0;
206 }
207
208 static void crypto_cbc_exit_tfm(struct crypto_tfm *tfm)
209 {
210         struct crypto_cbc_ctx *ctx = crypto_tfm_ctx(tfm);
211         crypto_free_cipher(ctx->child);
212 }
213
214 static struct crypto_instance *crypto_cbc_alloc(struct rtattr **tb)
215 {
216         struct crypto_instance *inst;
217         struct crypto_alg *alg;
218         int err;
219
220         err = crypto_check_attr_type(tb, CRYPTO_ALG_TYPE_BLKCIPHER);
221         if (err)
222                 return ERR_PTR(err);
223
224         alg = crypto_get_attr_alg(tb, CRYPTO_ALG_TYPE_CIPHER,
225                                   CRYPTO_ALG_TYPE_MASK);
226         if (IS_ERR(alg))
227                 return ERR_CAST(alg);
228
229         inst = ERR_PTR(-EINVAL);
230         if (!is_power_of_2(alg->cra_blocksize))
231                 goto out_put_alg;
232
233         inst = crypto_alloc_instance("cbc", alg);
234         if (IS_ERR(inst))
235                 goto out_put_alg;
236
237         inst->alg.cra_flags = CRYPTO_ALG_TYPE_BLKCIPHER;
238         inst->alg.cra_priority = alg->cra_priority;
239         inst->alg.cra_blocksize = alg->cra_blocksize;
240         inst->alg.cra_alignmask = alg->cra_alignmask;
241         inst->alg.cra_type = &crypto_blkcipher_type;
242
243         /* We access the data as u32s when xoring. */
244         inst->alg.cra_alignmask |= __alignof__(u32) - 1;
245
246         inst->alg.cra_blkcipher.ivsize = alg->cra_blocksize;
247         inst->alg.cra_blkcipher.min_keysize = alg->cra_cipher.cia_min_keysize;
248         inst->alg.cra_blkcipher.max_keysize = alg->cra_cipher.cia_max_keysize;
249
250         inst->alg.cra_ctxsize = sizeof(struct crypto_cbc_ctx);
251
252         inst->alg.cra_init = crypto_cbc_init_tfm;
253         inst->alg.cra_exit = crypto_cbc_exit_tfm;
254
255         inst->alg.cra_blkcipher.setkey = crypto_cbc_setkey;
256         inst->alg.cra_blkcipher.encrypt = crypto_cbc_encrypt;
257         inst->alg.cra_blkcipher.decrypt = crypto_cbc_decrypt;
258
259 out_put_alg:
260         crypto_mod_put(alg);
261         return inst;
262 }
263
264 static void crypto_cbc_free(struct crypto_instance *inst)
265 {
266         crypto_drop_spawn(crypto_instance_ctx(inst));
267         kfree(inst);
268 }
269
270 static struct crypto_template crypto_cbc_tmpl = {
271         .name = "cbc",
272         .alloc = crypto_cbc_alloc,
273         .free = crypto_cbc_free,
274         .module = THIS_MODULE,
275 };
276
277 static int __init crypto_cbc_module_init(void)
278 {
279         return crypto_register_template(&crypto_cbc_tmpl);
280 }
281
282 static void __exit crypto_cbc_module_exit(void)
283 {
284         crypto_unregister_template(&crypto_cbc_tmpl);
285 }
286
287 module_init(crypto_cbc_module_init);
288 module_exit(crypto_cbc_module_exit);
289
290 MODULE_LICENSE("GPL");
291 MODULE_DESCRIPTION("CBC block cipher algorithm");