crypto: shash - Fix zero-length shash ahash digest crash
[pandora-kernel.git] / crypto / shash.c
1 /*
2  * Synchronous Cryptographic Hash operations.
3  *
4  * Copyright (c) 2008 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/scatterwalk.h>
14 #include <crypto/internal/hash.h>
15 #include <linux/err.h>
16 #include <linux/kernel.h>
17 #include <linux/module.h>
18 #include <linux/slab.h>
19 #include <linux/seq_file.h>
20 #include <linux/cryptouser.h>
21 #include <net/netlink.h>
22
23 #include "internal.h"
24
25 static const struct crypto_type crypto_shash_type;
26
27 static int shash_no_setkey(struct crypto_shash *tfm, const u8 *key,
28                            unsigned int keylen)
29 {
30         return -ENOSYS;
31 }
32
33 static int shash_setkey_unaligned(struct crypto_shash *tfm, const u8 *key,
34                                   unsigned int keylen)
35 {
36         struct shash_alg *shash = crypto_shash_alg(tfm);
37         unsigned long alignmask = crypto_shash_alignmask(tfm);
38         unsigned long absize;
39         u8 *buffer, *alignbuffer;
40         int err;
41
42         absize = keylen + (alignmask & ~(crypto_tfm_ctx_alignment() - 1));
43         buffer = kmalloc(absize, GFP_KERNEL);
44         if (!buffer)
45                 return -ENOMEM;
46
47         alignbuffer = (u8 *)ALIGN((unsigned long)buffer, alignmask + 1);
48         memcpy(alignbuffer, key, keylen);
49         err = shash->setkey(tfm, alignbuffer, keylen);
50         kzfree(buffer);
51         return err;
52 }
53
54 int crypto_shash_setkey(struct crypto_shash *tfm, const u8 *key,
55                         unsigned int keylen)
56 {
57         struct shash_alg *shash = crypto_shash_alg(tfm);
58         unsigned long alignmask = crypto_shash_alignmask(tfm);
59
60         if ((unsigned long)key & alignmask)
61                 return shash_setkey_unaligned(tfm, key, keylen);
62
63         return shash->setkey(tfm, key, keylen);
64 }
65 EXPORT_SYMBOL_GPL(crypto_shash_setkey);
66
67 static inline unsigned int shash_align_buffer_size(unsigned len,
68                                                    unsigned long mask)
69 {
70         return len + (mask & ~(__alignof__(u8 __attribute__ ((aligned))) - 1));
71 }
72
73 static int shash_update_unaligned(struct shash_desc *desc, const u8 *data,
74                                   unsigned int len)
75 {
76         struct crypto_shash *tfm = desc->tfm;
77         struct shash_alg *shash = crypto_shash_alg(tfm);
78         unsigned long alignmask = crypto_shash_alignmask(tfm);
79         unsigned int unaligned_len = alignmask + 1 -
80                                      ((unsigned long)data & alignmask);
81         u8 ubuf[shash_align_buffer_size(unaligned_len, alignmask)]
82                 __attribute__ ((aligned));
83         u8 *buf = PTR_ALIGN(&ubuf[0], alignmask + 1);
84         int err;
85
86         if (unaligned_len > len)
87                 unaligned_len = len;
88
89         memcpy(buf, data, unaligned_len);
90         err = shash->update(desc, buf, unaligned_len);
91         memset(buf, 0, unaligned_len);
92
93         return err ?:
94                shash->update(desc, data + unaligned_len, len - unaligned_len);
95 }
96
97 int crypto_shash_update(struct shash_desc *desc, const u8 *data,
98                         unsigned int len)
99 {
100         struct crypto_shash *tfm = desc->tfm;
101         struct shash_alg *shash = crypto_shash_alg(tfm);
102         unsigned long alignmask = crypto_shash_alignmask(tfm);
103
104         if ((unsigned long)data & alignmask)
105                 return shash_update_unaligned(desc, data, len);
106
107         return shash->update(desc, data, len);
108 }
109 EXPORT_SYMBOL_GPL(crypto_shash_update);
110
111 static int shash_final_unaligned(struct shash_desc *desc, u8 *out)
112 {
113         struct crypto_shash *tfm = desc->tfm;
114         unsigned long alignmask = crypto_shash_alignmask(tfm);
115         struct shash_alg *shash = crypto_shash_alg(tfm);
116         unsigned int ds = crypto_shash_digestsize(tfm);
117         u8 ubuf[shash_align_buffer_size(ds, alignmask)]
118                 __attribute__ ((aligned));
119         u8 *buf = PTR_ALIGN(&ubuf[0], alignmask + 1);
120         int err;
121
122         err = shash->final(desc, buf);
123         if (err)
124                 goto out;
125
126         memcpy(out, buf, ds);
127
128 out:
129         memset(buf, 0, ds);
130         return err;
131 }
132
133 int crypto_shash_final(struct shash_desc *desc, u8 *out)
134 {
135         struct crypto_shash *tfm = desc->tfm;
136         struct shash_alg *shash = crypto_shash_alg(tfm);
137         unsigned long alignmask = crypto_shash_alignmask(tfm);
138
139         if ((unsigned long)out & alignmask)
140                 return shash_final_unaligned(desc, out);
141
142         return shash->final(desc, out);
143 }
144 EXPORT_SYMBOL_GPL(crypto_shash_final);
145
146 static int shash_finup_unaligned(struct shash_desc *desc, const u8 *data,
147                                  unsigned int len, u8 *out)
148 {
149         return crypto_shash_update(desc, data, len) ?:
150                crypto_shash_final(desc, out);
151 }
152
153 int crypto_shash_finup(struct shash_desc *desc, const u8 *data,
154                        unsigned int len, u8 *out)
155 {
156         struct crypto_shash *tfm = desc->tfm;
157         struct shash_alg *shash = crypto_shash_alg(tfm);
158         unsigned long alignmask = crypto_shash_alignmask(tfm);
159
160         if (((unsigned long)data | (unsigned long)out) & alignmask)
161                 return shash_finup_unaligned(desc, data, len, out);
162
163         return shash->finup(desc, data, len, out);
164 }
165 EXPORT_SYMBOL_GPL(crypto_shash_finup);
166
167 static int shash_digest_unaligned(struct shash_desc *desc, const u8 *data,
168                                   unsigned int len, u8 *out)
169 {
170         return crypto_shash_init(desc) ?:
171                crypto_shash_finup(desc, data, len, out);
172 }
173
174 int crypto_shash_digest(struct shash_desc *desc, const u8 *data,
175                         unsigned int len, u8 *out)
176 {
177         struct crypto_shash *tfm = desc->tfm;
178         struct shash_alg *shash = crypto_shash_alg(tfm);
179         unsigned long alignmask = crypto_shash_alignmask(tfm);
180
181         if (((unsigned long)data | (unsigned long)out) & alignmask)
182                 return shash_digest_unaligned(desc, data, len, out);
183
184         return shash->digest(desc, data, len, out);
185 }
186 EXPORT_SYMBOL_GPL(crypto_shash_digest);
187
188 static int shash_default_export(struct shash_desc *desc, void *out)
189 {
190         memcpy(out, shash_desc_ctx(desc), crypto_shash_descsize(desc->tfm));
191         return 0;
192 }
193
194 static int shash_default_import(struct shash_desc *desc, const void *in)
195 {
196         memcpy(shash_desc_ctx(desc), in, crypto_shash_descsize(desc->tfm));
197         return 0;
198 }
199
200 static int shash_async_setkey(struct crypto_ahash *tfm, const u8 *key,
201                               unsigned int keylen)
202 {
203         struct crypto_shash **ctx = crypto_ahash_ctx(tfm);
204
205         return crypto_shash_setkey(*ctx, key, keylen);
206 }
207
208 static int shash_async_init(struct ahash_request *req)
209 {
210         struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
211         struct shash_desc *desc = ahash_request_ctx(req);
212
213         desc->tfm = *ctx;
214         desc->flags = req->base.flags;
215
216         return crypto_shash_init(desc);
217 }
218
219 int shash_ahash_update(struct ahash_request *req, struct shash_desc *desc)
220 {
221         struct crypto_hash_walk walk;
222         int nbytes;
223
224         for (nbytes = crypto_hash_walk_first(req, &walk); nbytes > 0;
225              nbytes = crypto_hash_walk_done(&walk, nbytes))
226                 nbytes = crypto_shash_update(desc, walk.data, nbytes);
227
228         return nbytes;
229 }
230 EXPORT_SYMBOL_GPL(shash_ahash_update);
231
232 static int shash_async_update(struct ahash_request *req)
233 {
234         return shash_ahash_update(req, ahash_request_ctx(req));
235 }
236
237 static int shash_async_final(struct ahash_request *req)
238 {
239         return crypto_shash_final(ahash_request_ctx(req), req->result);
240 }
241
242 int shash_ahash_finup(struct ahash_request *req, struct shash_desc *desc)
243 {
244         struct crypto_hash_walk walk;
245         int nbytes;
246
247         nbytes = crypto_hash_walk_first(req, &walk);
248         if (!nbytes)
249                 return crypto_shash_final(desc, req->result);
250
251         do {
252                 nbytes = crypto_hash_walk_last(&walk) ?
253                          crypto_shash_finup(desc, walk.data, nbytes,
254                                             req->result) :
255                          crypto_shash_update(desc, walk.data, nbytes);
256                 nbytes = crypto_hash_walk_done(&walk, nbytes);
257         } while (nbytes > 0);
258
259         return nbytes;
260 }
261 EXPORT_SYMBOL_GPL(shash_ahash_finup);
262
263 static int shash_async_finup(struct ahash_request *req)
264 {
265         struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
266         struct shash_desc *desc = ahash_request_ctx(req);
267
268         desc->tfm = *ctx;
269         desc->flags = req->base.flags;
270
271         return shash_ahash_finup(req, desc);
272 }
273
274 int shash_ahash_digest(struct ahash_request *req, struct shash_desc *desc)
275 {
276         unsigned int nbytes = req->nbytes;
277         struct scatterlist *sg;
278         unsigned int offset;
279         int err;
280
281         if (nbytes &&
282             (sg = req->src, offset = sg->offset,
283              nbytes < min(sg->length, ((unsigned int)(PAGE_SIZE)) - offset))) {
284                 void *data;
285
286                 data = crypto_kmap(sg_page(sg), 0);
287                 err = crypto_shash_digest(desc, data + offset, nbytes,
288                                           req->result);
289                 crypto_kunmap(data, 0);
290                 crypto_yield(desc->flags);
291         } else
292                 err = crypto_shash_init(desc) ?:
293                       shash_ahash_finup(req, desc);
294
295         return err;
296 }
297 EXPORT_SYMBOL_GPL(shash_ahash_digest);
298
299 static int shash_async_digest(struct ahash_request *req)
300 {
301         struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
302         struct shash_desc *desc = ahash_request_ctx(req);
303
304         desc->tfm = *ctx;
305         desc->flags = req->base.flags;
306
307         return shash_ahash_digest(req, desc);
308 }
309
310 static int shash_async_export(struct ahash_request *req, void *out)
311 {
312         return crypto_shash_export(ahash_request_ctx(req), out);
313 }
314
315 static int shash_async_import(struct ahash_request *req, const void *in)
316 {
317         struct crypto_shash **ctx = crypto_ahash_ctx(crypto_ahash_reqtfm(req));
318         struct shash_desc *desc = ahash_request_ctx(req);
319
320         desc->tfm = *ctx;
321         desc->flags = req->base.flags;
322
323         return crypto_shash_import(desc, in);
324 }
325
326 static void crypto_exit_shash_ops_async(struct crypto_tfm *tfm)
327 {
328         struct crypto_shash **ctx = crypto_tfm_ctx(tfm);
329
330         crypto_free_shash(*ctx);
331 }
332
333 int crypto_init_shash_ops_async(struct crypto_tfm *tfm)
334 {
335         struct crypto_alg *calg = tfm->__crt_alg;
336         struct shash_alg *alg = __crypto_shash_alg(calg);
337         struct crypto_ahash *crt = __crypto_ahash_cast(tfm);
338         struct crypto_shash **ctx = crypto_tfm_ctx(tfm);
339         struct crypto_shash *shash;
340
341         if (!crypto_mod_get(calg))
342                 return -EAGAIN;
343
344         shash = crypto_create_tfm(calg, &crypto_shash_type);
345         if (IS_ERR(shash)) {
346                 crypto_mod_put(calg);
347                 return PTR_ERR(shash);
348         }
349
350         *ctx = shash;
351         tfm->exit = crypto_exit_shash_ops_async;
352
353         crt->init = shash_async_init;
354         crt->update = shash_async_update;
355         crt->final = shash_async_final;
356         crt->finup = shash_async_finup;
357         crt->digest = shash_async_digest;
358         crt->setkey = shash_async_setkey;
359
360         crt->has_setkey = alg->setkey != shash_no_setkey;
361
362         if (alg->export)
363                 crt->export = shash_async_export;
364         if (alg->import)
365                 crt->import = shash_async_import;
366
367         crt->reqsize = sizeof(struct shash_desc) + crypto_shash_descsize(shash);
368
369         return 0;
370 }
371
372 static int shash_compat_setkey(struct crypto_hash *tfm, const u8 *key,
373                                unsigned int keylen)
374 {
375         struct shash_desc **descp = crypto_hash_ctx(tfm);
376         struct shash_desc *desc = *descp;
377
378         return crypto_shash_setkey(desc->tfm, key, keylen);
379 }
380
381 static int shash_compat_init(struct hash_desc *hdesc)
382 {
383         struct shash_desc **descp = crypto_hash_ctx(hdesc->tfm);
384         struct shash_desc *desc = *descp;
385
386         desc->flags = hdesc->flags;
387
388         return crypto_shash_init(desc);
389 }
390
391 static int shash_compat_update(struct hash_desc *hdesc, struct scatterlist *sg,
392                                unsigned int len)
393 {
394         struct shash_desc **descp = crypto_hash_ctx(hdesc->tfm);
395         struct shash_desc *desc = *descp;
396         struct crypto_hash_walk walk;
397         int nbytes;
398
399         for (nbytes = crypto_hash_walk_first_compat(hdesc, &walk, sg, len);
400              nbytes > 0; nbytes = crypto_hash_walk_done(&walk, nbytes))
401                 nbytes = crypto_shash_update(desc, walk.data, nbytes);
402
403         return nbytes;
404 }
405
406 static int shash_compat_final(struct hash_desc *hdesc, u8 *out)
407 {
408         struct shash_desc **descp = crypto_hash_ctx(hdesc->tfm);
409
410         return crypto_shash_final(*descp, out);
411 }
412
413 static int shash_compat_digest(struct hash_desc *hdesc, struct scatterlist *sg,
414                                unsigned int nbytes, u8 *out)
415 {
416         unsigned int offset = sg->offset;
417         int err;
418
419         if (nbytes < min(sg->length, ((unsigned int)(PAGE_SIZE)) - offset)) {
420                 struct shash_desc **descp = crypto_hash_ctx(hdesc->tfm);
421                 struct shash_desc *desc = *descp;
422                 void *data;
423
424                 desc->flags = hdesc->flags;
425
426                 data = crypto_kmap(sg_page(sg), 0);
427                 err = crypto_shash_digest(desc, data + offset, nbytes, out);
428                 crypto_kunmap(data, 0);
429                 crypto_yield(desc->flags);
430                 goto out;
431         }
432
433         err = shash_compat_init(hdesc);
434         if (err)
435                 goto out;
436
437         err = shash_compat_update(hdesc, sg, nbytes);
438         if (err)
439                 goto out;
440
441         err = shash_compat_final(hdesc, out);
442
443 out:
444         return err;
445 }
446
447 static void crypto_exit_shash_ops_compat(struct crypto_tfm *tfm)
448 {
449         struct shash_desc **descp = crypto_tfm_ctx(tfm);
450         struct shash_desc *desc = *descp;
451
452         crypto_free_shash(desc->tfm);
453         kzfree(desc);
454 }
455
456 static int crypto_init_shash_ops_compat(struct crypto_tfm *tfm)
457 {
458         struct hash_tfm *crt = &tfm->crt_hash;
459         struct crypto_alg *calg = tfm->__crt_alg;
460         struct shash_alg *alg = __crypto_shash_alg(calg);
461         struct shash_desc **descp = crypto_tfm_ctx(tfm);
462         struct crypto_shash *shash;
463         struct shash_desc *desc;
464
465         if (!crypto_mod_get(calg))
466                 return -EAGAIN;
467
468         shash = crypto_create_tfm(calg, &crypto_shash_type);
469         if (IS_ERR(shash)) {
470                 crypto_mod_put(calg);
471                 return PTR_ERR(shash);
472         }
473
474         desc = kmalloc(sizeof(*desc) + crypto_shash_descsize(shash),
475                        GFP_KERNEL);
476         if (!desc) {
477                 crypto_free_shash(shash);
478                 return -ENOMEM;
479         }
480
481         *descp = desc;
482         desc->tfm = shash;
483         tfm->exit = crypto_exit_shash_ops_compat;
484
485         crt->init = shash_compat_init;
486         crt->update = shash_compat_update;
487         crt->final  = shash_compat_final;
488         crt->digest = shash_compat_digest;
489         crt->setkey = shash_compat_setkey;
490
491         crt->digestsize = alg->digestsize;
492
493         return 0;
494 }
495
496 static int crypto_init_shash_ops(struct crypto_tfm *tfm, u32 type, u32 mask)
497 {
498         switch (mask & CRYPTO_ALG_TYPE_MASK) {
499         case CRYPTO_ALG_TYPE_HASH_MASK:
500                 return crypto_init_shash_ops_compat(tfm);
501         }
502
503         return -EINVAL;
504 }
505
506 static unsigned int crypto_shash_ctxsize(struct crypto_alg *alg, u32 type,
507                                          u32 mask)
508 {
509         switch (mask & CRYPTO_ALG_TYPE_MASK) {
510         case CRYPTO_ALG_TYPE_HASH_MASK:
511                 return sizeof(struct shash_desc *);
512         }
513
514         return 0;
515 }
516
517 static int crypto_shash_init_tfm(struct crypto_tfm *tfm)
518 {
519         struct crypto_shash *hash = __crypto_shash_cast(tfm);
520
521         hash->descsize = crypto_shash_alg(hash)->descsize;
522         return 0;
523 }
524
525 static unsigned int crypto_shash_extsize(struct crypto_alg *alg)
526 {
527         return alg->cra_ctxsize;
528 }
529
530 #ifdef CONFIG_NET
531 static int crypto_shash_report(struct sk_buff *skb, struct crypto_alg *alg)
532 {
533         struct crypto_report_hash rhash;
534         struct shash_alg *salg = __crypto_shash_alg(alg);
535
536         strncpy(rhash.type, "shash", sizeof(rhash.type));
537
538         rhash.blocksize = alg->cra_blocksize;
539         rhash.digestsize = salg->digestsize;
540
541         NLA_PUT(skb, CRYPTOCFGA_REPORT_HASH,
542                 sizeof(struct crypto_report_hash), &rhash);
543
544         return 0;
545
546 nla_put_failure:
547         return -EMSGSIZE;
548 }
549 #else
550 static int crypto_shash_report(struct sk_buff *skb, struct crypto_alg *alg)
551 {
552         return -ENOSYS;
553 }
554 #endif
555
556 static void crypto_shash_show(struct seq_file *m, struct crypto_alg *alg)
557         __attribute__ ((unused));
558 static void crypto_shash_show(struct seq_file *m, struct crypto_alg *alg)
559 {
560         struct shash_alg *salg = __crypto_shash_alg(alg);
561
562         seq_printf(m, "type         : shash\n");
563         seq_printf(m, "blocksize    : %u\n", alg->cra_blocksize);
564         seq_printf(m, "digestsize   : %u\n", salg->digestsize);
565 }
566
567 static const struct crypto_type crypto_shash_type = {
568         .ctxsize = crypto_shash_ctxsize,
569         .extsize = crypto_shash_extsize,
570         .init = crypto_init_shash_ops,
571         .init_tfm = crypto_shash_init_tfm,
572 #ifdef CONFIG_PROC_FS
573         .show = crypto_shash_show,
574 #endif
575         .report = crypto_shash_report,
576         .maskclear = ~CRYPTO_ALG_TYPE_MASK,
577         .maskset = CRYPTO_ALG_TYPE_MASK,
578         .type = CRYPTO_ALG_TYPE_SHASH,
579         .tfmsize = offsetof(struct crypto_shash, base),
580 };
581
582 struct crypto_shash *crypto_alloc_shash(const char *alg_name, u32 type,
583                                         u32 mask)
584 {
585         return crypto_alloc_tfm(alg_name, &crypto_shash_type, type, mask);
586 }
587 EXPORT_SYMBOL_GPL(crypto_alloc_shash);
588
589 static int shash_prepare_alg(struct shash_alg *alg)
590 {
591         struct crypto_alg *base = &alg->base;
592
593         if (alg->digestsize > PAGE_SIZE / 8 ||
594             alg->descsize > PAGE_SIZE / 8 ||
595             alg->statesize > PAGE_SIZE / 8)
596                 return -EINVAL;
597
598         base->cra_type = &crypto_shash_type;
599         base->cra_flags &= ~CRYPTO_ALG_TYPE_MASK;
600         base->cra_flags |= CRYPTO_ALG_TYPE_SHASH;
601
602         if (!alg->finup)
603                 alg->finup = shash_finup_unaligned;
604         if (!alg->digest)
605                 alg->digest = shash_digest_unaligned;
606         if (!alg->export) {
607                 alg->export = shash_default_export;
608                 alg->import = shash_default_import;
609                 alg->statesize = alg->descsize;
610         }
611         if (!alg->setkey)
612                 alg->setkey = shash_no_setkey;
613
614         return 0;
615 }
616
617 int crypto_register_shash(struct shash_alg *alg)
618 {
619         struct crypto_alg *base = &alg->base;
620         int err;
621
622         err = shash_prepare_alg(alg);
623         if (err)
624                 return err;
625
626         return crypto_register_alg(base);
627 }
628 EXPORT_SYMBOL_GPL(crypto_register_shash);
629
630 int crypto_unregister_shash(struct shash_alg *alg)
631 {
632         return crypto_unregister_alg(&alg->base);
633 }
634 EXPORT_SYMBOL_GPL(crypto_unregister_shash);
635
636 int shash_register_instance(struct crypto_template *tmpl,
637                             struct shash_instance *inst)
638 {
639         int err;
640
641         err = shash_prepare_alg(&inst->alg);
642         if (err)
643                 return err;
644
645         return crypto_register_instance(tmpl, shash_crypto_instance(inst));
646 }
647 EXPORT_SYMBOL_GPL(shash_register_instance);
648
649 void shash_free_instance(struct crypto_instance *inst)
650 {
651         crypto_drop_spawn(crypto_instance_ctx(inst));
652         kfree(shash_instance(inst));
653 }
654 EXPORT_SYMBOL_GPL(shash_free_instance);
655
656 int crypto_init_shash_spawn(struct crypto_shash_spawn *spawn,
657                             struct shash_alg *alg,
658                             struct crypto_instance *inst)
659 {
660         return crypto_init_spawn2(&spawn->base, &alg->base, inst,
661                                   &crypto_shash_type);
662 }
663 EXPORT_SYMBOL_GPL(crypto_init_shash_spawn);
664
665 struct shash_alg *shash_attr_alg(struct rtattr *rta, u32 type, u32 mask)
666 {
667         struct crypto_alg *alg;
668
669         alg = crypto_attr_alg2(rta, &crypto_shash_type, type, mask);
670         return IS_ERR(alg) ? ERR_CAST(alg) :
671                container_of(alg, struct shash_alg, base);
672 }
673 EXPORT_SYMBOL_GPL(shash_attr_alg);
674
675 MODULE_LICENSE("GPL");
676 MODULE_DESCRIPTION("Synchronous cryptographic hash type");