crypto: tcrypt - Add speed tests for async hashing
[linux-2.6.git] / crypto / tcrypt.c
1 /*
2  * Quick & dirty crypto testing module.
3  *
4  * This will only exist until we have a better testing mechanism
5  * (e.g. a char device).
6  *
7  * Copyright (c) 2002 James Morris <jmorris@intercode.com.au>
8  * Copyright (c) 2002 Jean-Francois Dive <jef@linuxbe.org>
9  * Copyright (c) 2007 Nokia Siemens Networks
10  *
11  * This program is free software; you can redistribute it and/or modify it
12  * under the terms of the GNU General Public License as published by the Free
13  * Software Foundation; either version 2 of the License, or (at your option)
14  * any later version.
15  *
16  */
17
18 #include <crypto/hash.h>
19 #include <linux/err.h>
20 #include <linux/init.h>
21 #include <linux/gfp.h>
22 #include <linux/module.h>
23 #include <linux/scatterlist.h>
24 #include <linux/string.h>
25 #include <linux/moduleparam.h>
26 #include <linux/jiffies.h>
27 #include <linux/timex.h>
28 #include <linux/interrupt.h>
29 #include "tcrypt.h"
30 #include "internal.h"
31
32 /*
33  * Need slab memory for testing (size in number of pages).
34  */
35 #define TVMEMSIZE       4
36
37 /*
38 * Used by test_cipher_speed()
39 */
40 #define ENCRYPT 1
41 #define DECRYPT 0
42
43 /*
44  * Used by test_cipher_speed()
45  */
46 static unsigned int sec;
47
48 static char *alg = NULL;
49 static u32 type;
50 static u32 mask;
51 static int mode;
52 static char *tvmem[TVMEMSIZE];
53
54 static char *check[] = {
55         "des", "md5", "des3_ede", "rot13", "sha1", "sha224", "sha256",
56         "blowfish", "twofish", "serpent", "sha384", "sha512", "md4", "aes",
57         "cast6", "arc4", "michael_mic", "deflate", "crc32c", "tea", "xtea",
58         "khazad", "wp512", "wp384", "wp256", "tnepres", "xeta",  "fcrypt",
59         "camellia", "seed", "salsa20", "rmd128", "rmd160", "rmd256", "rmd320",
60         "lzo", "cts", "zlib", NULL
61 };
62
63 static int test_cipher_jiffies(struct blkcipher_desc *desc, int enc,
64                                struct scatterlist *sg, int blen, int sec)
65 {
66         unsigned long start, end;
67         int bcount;
68         int ret;
69
70         for (start = jiffies, end = start + sec * HZ, bcount = 0;
71              time_before(jiffies, end); bcount++) {
72                 if (enc)
73                         ret = crypto_blkcipher_encrypt(desc, sg, sg, blen);
74                 else
75                         ret = crypto_blkcipher_decrypt(desc, sg, sg, blen);
76
77                 if (ret)
78                         return ret;
79         }
80
81         printk("%d operations in %d seconds (%ld bytes)\n",
82                bcount, sec, (long)bcount * blen);
83         return 0;
84 }
85
86 static int test_cipher_cycles(struct blkcipher_desc *desc, int enc,
87                               struct scatterlist *sg, int blen)
88 {
89         unsigned long cycles = 0;
90         int ret = 0;
91         int i;
92
93         local_bh_disable();
94         local_irq_disable();
95
96         /* Warm-up run. */
97         for (i = 0; i < 4; i++) {
98                 if (enc)
99                         ret = crypto_blkcipher_encrypt(desc, sg, sg, blen);
100                 else
101                         ret = crypto_blkcipher_decrypt(desc, sg, sg, blen);
102
103                 if (ret)
104                         goto out;
105         }
106
107         /* The real thing. */
108         for (i = 0; i < 8; i++) {
109                 cycles_t start, end;
110
111                 start = get_cycles();
112                 if (enc)
113                         ret = crypto_blkcipher_encrypt(desc, sg, sg, blen);
114                 else
115                         ret = crypto_blkcipher_decrypt(desc, sg, sg, blen);
116                 end = get_cycles();
117
118                 if (ret)
119                         goto out;
120
121                 cycles += end - start;
122         }
123
124 out:
125         local_irq_enable();
126         local_bh_enable();
127
128         if (ret == 0)
129                 printk("1 operation in %lu cycles (%d bytes)\n",
130                        (cycles + 4) / 8, blen);
131
132         return ret;
133 }
134
135 static u32 block_sizes[] = { 16, 64, 256, 1024, 8192, 0 };
136
137 static void test_cipher_speed(const char *algo, int enc, unsigned int sec,
138                               struct cipher_speed_template *template,
139                               unsigned int tcount, u8 *keysize)
140 {
141         unsigned int ret, i, j, iv_len;
142         const char *key, iv[128];
143         struct crypto_blkcipher *tfm;
144         struct blkcipher_desc desc;
145         const char *e;
146         u32 *b_size;
147
148         if (enc == ENCRYPT)
149                 e = "encryption";
150         else
151                 e = "decryption";
152
153         printk("\ntesting speed of %s %s\n", algo, e);
154
155         tfm = crypto_alloc_blkcipher(algo, 0, CRYPTO_ALG_ASYNC);
156
157         if (IS_ERR(tfm)) {
158                 printk("failed to load transform for %s: %ld\n", algo,
159                        PTR_ERR(tfm));
160                 return;
161         }
162         desc.tfm = tfm;
163         desc.flags = 0;
164
165         i = 0;
166         do {
167
168                 b_size = block_sizes;
169                 do {
170                         struct scatterlist sg[TVMEMSIZE];
171
172                         if ((*keysize + *b_size) > TVMEMSIZE * PAGE_SIZE) {
173                                 printk("template (%u) too big for "
174                                        "tvmem (%lu)\n", *keysize + *b_size,
175                                        TVMEMSIZE * PAGE_SIZE);
176                                 goto out;
177                         }
178
179                         printk("test %u (%d bit key, %d byte blocks): ", i,
180                                         *keysize * 8, *b_size);
181
182                         memset(tvmem[0], 0xff, PAGE_SIZE);
183
184                         /* set key, plain text and IV */
185                         key = tvmem[0];
186                         for (j = 0; j < tcount; j++) {
187                                 if (template[j].klen == *keysize) {
188                                         key = template[j].key;
189                                         break;
190                                 }
191                         }
192
193                         ret = crypto_blkcipher_setkey(tfm, key, *keysize);
194                         if (ret) {
195                                 printk("setkey() failed flags=%x\n",
196                                                 crypto_blkcipher_get_flags(tfm));
197                                 goto out;
198                         }
199
200                         sg_init_table(sg, TVMEMSIZE);
201                         sg_set_buf(sg, tvmem[0] + *keysize,
202                                    PAGE_SIZE - *keysize);
203                         for (j = 1; j < TVMEMSIZE; j++) {
204                                 sg_set_buf(sg + j, tvmem[j], PAGE_SIZE);
205                                 memset (tvmem[j], 0xff, PAGE_SIZE);
206                         }
207
208                         iv_len = crypto_blkcipher_ivsize(tfm);
209                         if (iv_len) {
210                                 memset(&iv, 0xff, iv_len);
211                                 crypto_blkcipher_set_iv(tfm, iv, iv_len);
212                         }
213
214                         if (sec)
215                                 ret = test_cipher_jiffies(&desc, enc, sg,
216                                                           *b_size, sec);
217                         else
218                                 ret = test_cipher_cycles(&desc, enc, sg,
219                                                          *b_size);
220
221                         if (ret) {
222                                 printk("%s() failed flags=%x\n", e, desc.flags);
223                                 break;
224                         }
225                         b_size++;
226                         i++;
227                 } while (*b_size);
228                 keysize++;
229         } while (*keysize);
230
231 out:
232         crypto_free_blkcipher(tfm);
233 }
234
235 static int test_hash_jiffies_digest(struct hash_desc *desc,
236                                     struct scatterlist *sg, int blen,
237                                     char *out, int sec)
238 {
239         unsigned long start, end;
240         int bcount;
241         int ret;
242
243         for (start = jiffies, end = start + sec * HZ, bcount = 0;
244              time_before(jiffies, end); bcount++) {
245                 ret = crypto_hash_digest(desc, sg, blen, out);
246                 if (ret)
247                         return ret;
248         }
249
250         printk("%6u opers/sec, %9lu bytes/sec\n",
251                bcount / sec, ((long)bcount * blen) / sec);
252
253         return 0;
254 }
255
256 static int test_hash_jiffies(struct hash_desc *desc, struct scatterlist *sg,
257                              int blen, int plen, char *out, int sec)
258 {
259         unsigned long start, end;
260         int bcount, pcount;
261         int ret;
262
263         if (plen == blen)
264                 return test_hash_jiffies_digest(desc, sg, blen, out, sec);
265
266         for (start = jiffies, end = start + sec * HZ, bcount = 0;
267              time_before(jiffies, end); bcount++) {
268                 ret = crypto_hash_init(desc);
269                 if (ret)
270                         return ret;
271                 for (pcount = 0; pcount < blen; pcount += plen) {
272                         ret = crypto_hash_update(desc, sg, plen);
273                         if (ret)
274                                 return ret;
275                 }
276                 /* we assume there is enough space in 'out' for the result */
277                 ret = crypto_hash_final(desc, out);
278                 if (ret)
279                         return ret;
280         }
281
282         printk("%6u opers/sec, %9lu bytes/sec\n",
283                bcount / sec, ((long)bcount * blen) / sec);
284
285         return 0;
286 }
287
288 static int test_hash_cycles_digest(struct hash_desc *desc,
289                                    struct scatterlist *sg, int blen, char *out)
290 {
291         unsigned long cycles = 0;
292         int i;
293         int ret;
294
295         local_bh_disable();
296         local_irq_disable();
297
298         /* Warm-up run. */
299         for (i = 0; i < 4; i++) {
300                 ret = crypto_hash_digest(desc, sg, blen, out);
301                 if (ret)
302                         goto out;
303         }
304
305         /* The real thing. */
306         for (i = 0; i < 8; i++) {
307                 cycles_t start, end;
308
309                 start = get_cycles();
310
311                 ret = crypto_hash_digest(desc, sg, blen, out);
312                 if (ret)
313                         goto out;
314
315                 end = get_cycles();
316
317                 cycles += end - start;
318         }
319
320 out:
321         local_irq_enable();
322         local_bh_enable();
323
324         if (ret)
325                 return ret;
326
327         printk("%6lu cycles/operation, %4lu cycles/byte\n",
328                cycles / 8, cycles / (8 * blen));
329
330         return 0;
331 }
332
333 static int test_hash_cycles(struct hash_desc *desc, struct scatterlist *sg,
334                             int blen, int plen, char *out)
335 {
336         unsigned long cycles = 0;
337         int i, pcount;
338         int ret;
339
340         if (plen == blen)
341                 return test_hash_cycles_digest(desc, sg, blen, out);
342
343         local_bh_disable();
344         local_irq_disable();
345
346         /* Warm-up run. */
347         for (i = 0; i < 4; i++) {
348                 ret = crypto_hash_init(desc);
349                 if (ret)
350                         goto out;
351                 for (pcount = 0; pcount < blen; pcount += plen) {
352                         ret = crypto_hash_update(desc, sg, plen);
353                         if (ret)
354                                 goto out;
355                 }
356                 ret = crypto_hash_final(desc, out);
357                 if (ret)
358                         goto out;
359         }
360
361         /* The real thing. */
362         for (i = 0; i < 8; i++) {
363                 cycles_t start, end;
364
365                 start = get_cycles();
366
367                 ret = crypto_hash_init(desc);
368                 if (ret)
369                         goto out;
370                 for (pcount = 0; pcount < blen; pcount += plen) {
371                         ret = crypto_hash_update(desc, sg, plen);
372                         if (ret)
373                                 goto out;
374                 }
375                 ret = crypto_hash_final(desc, out);
376                 if (ret)
377                         goto out;
378
379                 end = get_cycles();
380
381                 cycles += end - start;
382         }
383
384 out:
385         local_irq_enable();
386         local_bh_enable();
387
388         if (ret)
389                 return ret;
390
391         printk("%6lu cycles/operation, %4lu cycles/byte\n",
392                cycles / 8, cycles / (8 * blen));
393
394         return 0;
395 }
396
397 static void test_hash_sg_init(struct scatterlist *sg)
398 {
399         int i;
400
401         sg_init_table(sg, TVMEMSIZE);
402         for (i = 0; i < TVMEMSIZE; i++) {
403                 sg_set_buf(sg + i, tvmem[i], PAGE_SIZE);
404                 memset(tvmem[i], 0xff, PAGE_SIZE);
405         }
406 }
407
408 static void test_hash_speed(const char *algo, unsigned int sec,
409                             struct hash_speed *speed)
410 {
411         struct scatterlist sg[TVMEMSIZE];
412         struct crypto_hash *tfm;
413         struct hash_desc desc;
414         static char output[1024];
415         int i;
416         int ret;
417
418         printk(KERN_INFO "\ntesting speed of %s\n", algo);
419
420         tfm = crypto_alloc_hash(algo, 0, CRYPTO_ALG_ASYNC);
421
422         if (IS_ERR(tfm)) {
423                 printk(KERN_ERR "failed to load transform for %s: %ld\n", algo,
424                        PTR_ERR(tfm));
425                 return;
426         }
427
428         desc.tfm = tfm;
429         desc.flags = 0;
430
431         if (crypto_hash_digestsize(tfm) > sizeof(output)) {
432                 printk(KERN_ERR "digestsize(%u) > outputbuffer(%zu)\n",
433                        crypto_hash_digestsize(tfm), sizeof(output));
434                 goto out;
435         }
436
437         test_hash_sg_init(sg);
438         for (i = 0; speed[i].blen != 0; i++) {
439                 if (speed[i].blen > TVMEMSIZE * PAGE_SIZE) {
440                         printk(KERN_ERR
441                                "template (%u) too big for tvmem (%lu)\n",
442                                speed[i].blen, TVMEMSIZE * PAGE_SIZE);
443                         goto out;
444                 }
445
446                 if (speed[i].klen)
447                         crypto_hash_setkey(tfm, tvmem[0], speed[i].klen);
448
449                 printk(KERN_INFO "test%3u "
450                        "(%5u byte blocks,%5u bytes per update,%4u updates): ",
451                        i, speed[i].blen, speed[i].plen, speed[i].blen / speed[i].plen);
452
453                 if (sec)
454                         ret = test_hash_jiffies(&desc, sg, speed[i].blen,
455                                                 speed[i].plen, output, sec);
456                 else
457                         ret = test_hash_cycles(&desc, sg, speed[i].blen,
458                                                speed[i].plen, output);
459
460                 if (ret) {
461                         printk(KERN_ERR "hashing failed ret=%d\n", ret);
462                         break;
463                 }
464         }
465
466 out:
467         crypto_free_hash(tfm);
468 }
469
470 struct tcrypt_result {
471         struct completion completion;
472         int err;
473 };
474
475 static void tcrypt_complete(struct crypto_async_request *req, int err)
476 {
477         struct tcrypt_result *res = req->data;
478
479         if (err == -EINPROGRESS)
480                 return;
481
482         res->err = err;
483         complete(&res->completion);
484 }
485
486 static inline int do_one_ahash_op(struct ahash_request *req, int ret)
487 {
488         if (ret == -EINPROGRESS || ret == -EBUSY) {
489                 struct tcrypt_result *tr = req->base.data;
490
491                 ret = wait_for_completion_interruptible(&tr->completion);
492                 if (!ret)
493                         ret = tr->err;
494                 INIT_COMPLETION(tr->completion);
495         }
496         return ret;
497 }
498
499 static int test_ahash_jiffies_digest(struct ahash_request *req, int blen,
500                                      char *out, int sec)
501 {
502         unsigned long start, end;
503         int bcount;
504         int ret;
505
506         for (start = jiffies, end = start + sec * HZ, bcount = 0;
507              time_before(jiffies, end); bcount++) {
508                 ret = do_one_ahash_op(req, crypto_ahash_digest(req));
509                 if (ret)
510                         return ret;
511         }
512
513         printk("%6u opers/sec, %9lu bytes/sec\n",
514                bcount / sec, ((long)bcount * blen) / sec);
515
516         return 0;
517 }
518
519 static int test_ahash_jiffies(struct ahash_request *req, int blen,
520                               int plen, char *out, int sec)
521 {
522         unsigned long start, end;
523         int bcount, pcount;
524         int ret;
525
526         if (plen == blen)
527                 return test_ahash_jiffies_digest(req, blen, out, sec);
528
529         for (start = jiffies, end = start + sec * HZ, bcount = 0;
530              time_before(jiffies, end); bcount++) {
531                 ret = crypto_ahash_init(req);
532                 if (ret)
533                         return ret;
534                 for (pcount = 0; pcount < blen; pcount += plen) {
535                         ret = do_one_ahash_op(req, crypto_ahash_update(req));
536                         if (ret)
537                                 return ret;
538                 }
539                 /* we assume there is enough space in 'out' for the result */
540                 ret = do_one_ahash_op(req, crypto_ahash_final(req));
541                 if (ret)
542                         return ret;
543         }
544
545         pr_cont("%6u opers/sec, %9lu bytes/sec\n",
546                 bcount / sec, ((long)bcount * blen) / sec);
547
548         return 0;
549 }
550
551 static int test_ahash_cycles_digest(struct ahash_request *req, int blen,
552                                     char *out)
553 {
554         unsigned long cycles = 0;
555         int ret, i;
556
557         /* Warm-up run. */
558         for (i = 0; i < 4; i++) {
559                 ret = do_one_ahash_op(req, crypto_ahash_digest(req));
560                 if (ret)
561                         goto out;
562         }
563
564         /* The real thing. */
565         for (i = 0; i < 8; i++) {
566                 cycles_t start, end;
567
568                 start = get_cycles();
569
570                 ret = do_one_ahash_op(req, crypto_ahash_digest(req));
571                 if (ret)
572                         goto out;
573
574                 end = get_cycles();
575
576                 cycles += end - start;
577         }
578
579 out:
580         if (ret)
581                 return ret;
582
583         pr_cont("%6lu cycles/operation, %4lu cycles/byte\n",
584                 cycles / 8, cycles / (8 * blen));
585
586         return 0;
587 }
588
589 static int test_ahash_cycles(struct ahash_request *req, int blen,
590                              int plen, char *out)
591 {
592         unsigned long cycles = 0;
593         int i, pcount, ret;
594
595         if (plen == blen)
596                 return test_ahash_cycles_digest(req, blen, out);
597
598         /* Warm-up run. */
599         for (i = 0; i < 4; i++) {
600                 ret = crypto_ahash_init(req);
601                 if (ret)
602                         goto out;
603                 for (pcount = 0; pcount < blen; pcount += plen) {
604                         ret = do_one_ahash_op(req, crypto_ahash_update(req));
605                         if (ret)
606                                 goto out;
607                 }
608                 ret = do_one_ahash_op(req, crypto_ahash_final(req));
609                 if (ret)
610                         goto out;
611         }
612
613         /* The real thing. */
614         for (i = 0; i < 8; i++) {
615                 cycles_t start, end;
616
617                 start = get_cycles();
618
619                 ret = crypto_ahash_init(req);
620                 if (ret)
621                         goto out;
622                 for (pcount = 0; pcount < blen; pcount += plen) {
623                         ret = do_one_ahash_op(req, crypto_ahash_update(req));
624                         if (ret)
625                                 goto out;
626                 }
627                 ret = do_one_ahash_op(req, crypto_ahash_final(req));
628                 if (ret)
629                         goto out;
630
631                 end = get_cycles();
632
633                 cycles += end - start;
634         }
635
636 out:
637         if (ret)
638                 return ret;
639
640         pr_cont("%6lu cycles/operation, %4lu cycles/byte\n",
641                 cycles / 8, cycles / (8 * blen));
642
643         return 0;
644 }
645
646 static void test_ahash_speed(const char *algo, unsigned int sec,
647                              struct hash_speed *speed)
648 {
649         struct scatterlist sg[TVMEMSIZE];
650         struct tcrypt_result tresult;
651         struct ahash_request *req;
652         struct crypto_ahash *tfm;
653         static char output[1024];
654         int i, ret;
655
656         printk(KERN_INFO "\ntesting speed of async %s\n", algo);
657
658         tfm = crypto_alloc_ahash(algo, 0, 0);
659         if (IS_ERR(tfm)) {
660                 pr_err("failed to load transform for %s: %ld\n",
661                        algo, PTR_ERR(tfm));
662                 return;
663         }
664
665         if (crypto_ahash_digestsize(tfm) > sizeof(output)) {
666                 pr_err("digestsize(%u) > outputbuffer(%zu)\n",
667                        crypto_ahash_digestsize(tfm), sizeof(output));
668                 goto out;
669         }
670
671         test_hash_sg_init(sg);
672         req = ahash_request_alloc(tfm, GFP_KERNEL);
673         if (!req) {
674                 pr_err("ahash request allocation failure\n");
675                 goto out;
676         }
677
678         init_completion(&tresult.completion);
679         ahash_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
680                                    tcrypt_complete, &tresult);
681
682         for (i = 0; speed[i].blen != 0; i++) {
683                 if (speed[i].blen > TVMEMSIZE * PAGE_SIZE) {
684                         pr_err("template (%u) too big for tvmem (%lu)\n",
685                                speed[i].blen, TVMEMSIZE * PAGE_SIZE);
686                         break;
687                 }
688
689                 pr_info("test%3u "
690                         "(%5u byte blocks,%5u bytes per update,%4u updates): ",
691                         i, speed[i].blen, speed[i].plen, speed[i].blen / speed[i].plen);
692
693                 ahash_request_set_crypt(req, sg, output, speed[i].plen);
694
695                 if (sec)
696                         ret = test_ahash_jiffies(req, speed[i].blen,
697                                                  speed[i].plen, output, sec);
698                 else
699                         ret = test_ahash_cycles(req, speed[i].blen,
700                                                 speed[i].plen, output);
701
702                 if (ret) {
703                         pr_err("hashing failed ret=%d\n", ret);
704                         break;
705                 }
706         }
707
708         ahash_request_free(req);
709
710 out:
711         crypto_free_ahash(tfm);
712 }
713
714 static void test_available(void)
715 {
716         char **name = check;
717
718         while (*name) {
719                 printk("alg %s ", *name);
720                 printk(crypto_has_alg(*name, 0, 0) ?
721                        "found\n" : "not found\n");
722                 name++;
723         }
724 }
725
726 static inline int tcrypt_test(const char *alg)
727 {
728         int ret;
729
730         ret = alg_test(alg, alg, 0, 0);
731         /* non-fips algs return -EINVAL in fips mode */
732         if (fips_enabled && ret == -EINVAL)
733                 ret = 0;
734         return ret;
735 }
736
737 static int do_test(int m)
738 {
739         int i;
740         int ret = 0;
741
742         switch (m) {
743         case 0:
744                 for (i = 1; i < 200; i++)
745                         ret += do_test(i);
746                 break;
747
748         case 1:
749                 ret += tcrypt_test("md5");
750                 break;
751
752         case 2:
753                 ret += tcrypt_test("sha1");
754                 break;
755
756         case 3:
757                 ret += tcrypt_test("ecb(des)");
758                 ret += tcrypt_test("cbc(des)");
759                 break;
760
761         case 4:
762                 ret += tcrypt_test("ecb(des3_ede)");
763                 ret += tcrypt_test("cbc(des3_ede)");
764                 break;
765
766         case 5:
767                 ret += tcrypt_test("md4");
768                 break;
769
770         case 6:
771                 ret += tcrypt_test("sha256");
772                 break;
773
774         case 7:
775                 ret += tcrypt_test("ecb(blowfish)");
776                 ret += tcrypt_test("cbc(blowfish)");
777                 break;
778
779         case 8:
780                 ret += tcrypt_test("ecb(twofish)");
781                 ret += tcrypt_test("cbc(twofish)");
782                 break;
783
784         case 9:
785                 ret += tcrypt_test("ecb(serpent)");
786                 break;
787
788         case 10:
789                 ret += tcrypt_test("ecb(aes)");
790                 ret += tcrypt_test("cbc(aes)");
791                 ret += tcrypt_test("lrw(aes)");
792                 ret += tcrypt_test("xts(aes)");
793                 ret += tcrypt_test("ctr(aes)");
794                 ret += tcrypt_test("rfc3686(ctr(aes))");
795                 break;
796
797         case 11:
798                 ret += tcrypt_test("sha384");
799                 break;
800
801         case 12:
802                 ret += tcrypt_test("sha512");
803                 break;
804
805         case 13:
806                 ret += tcrypt_test("deflate");
807                 break;
808
809         case 14:
810                 ret += tcrypt_test("ecb(cast5)");
811                 break;
812
813         case 15:
814                 ret += tcrypt_test("ecb(cast6)");
815                 break;
816
817         case 16:
818                 ret += tcrypt_test("ecb(arc4)");
819                 break;
820
821         case 17:
822                 ret += tcrypt_test("michael_mic");
823                 break;
824
825         case 18:
826                 ret += tcrypt_test("crc32c");
827                 break;
828
829         case 19:
830                 ret += tcrypt_test("ecb(tea)");
831                 break;
832
833         case 20:
834                 ret += tcrypt_test("ecb(xtea)");
835                 break;
836
837         case 21:
838                 ret += tcrypt_test("ecb(khazad)");
839                 break;
840
841         case 22:
842                 ret += tcrypt_test("wp512");
843                 break;
844
845         case 23:
846                 ret += tcrypt_test("wp384");
847                 break;
848
849         case 24:
850                 ret += tcrypt_test("wp256");
851                 break;
852
853         case 25:
854                 ret += tcrypt_test("ecb(tnepres)");
855                 break;
856
857         case 26:
858                 ret += tcrypt_test("ecb(anubis)");
859                 ret += tcrypt_test("cbc(anubis)");
860                 break;
861
862         case 27:
863                 ret += tcrypt_test("tgr192");
864                 break;
865
866         case 28:
867
868                 ret += tcrypt_test("tgr160");
869                 break;
870
871         case 29:
872                 ret += tcrypt_test("tgr128");
873                 break;
874
875         case 30:
876                 ret += tcrypt_test("ecb(xeta)");
877                 break;
878
879         case 31:
880                 ret += tcrypt_test("pcbc(fcrypt)");
881                 break;
882
883         case 32:
884                 ret += tcrypt_test("ecb(camellia)");
885                 ret += tcrypt_test("cbc(camellia)");
886                 break;
887         case 33:
888                 ret += tcrypt_test("sha224");
889                 break;
890
891         case 34:
892                 ret += tcrypt_test("salsa20");
893                 break;
894
895         case 35:
896                 ret += tcrypt_test("gcm(aes)");
897                 break;
898
899         case 36:
900                 ret += tcrypt_test("lzo");
901                 break;
902
903         case 37:
904                 ret += tcrypt_test("ccm(aes)");
905                 break;
906
907         case 38:
908                 ret += tcrypt_test("cts(cbc(aes))");
909                 break;
910
911         case 39:
912                 ret += tcrypt_test("rmd128");
913                 break;
914
915         case 40:
916                 ret += tcrypt_test("rmd160");
917                 break;
918
919         case 41:
920                 ret += tcrypt_test("rmd256");
921                 break;
922
923         case 42:
924                 ret += tcrypt_test("rmd320");
925                 break;
926
927         case 43:
928                 ret += tcrypt_test("ecb(seed)");
929                 break;
930
931         case 44:
932                 ret += tcrypt_test("zlib");
933                 break;
934
935         case 45:
936                 ret += tcrypt_test("rfc4309(ccm(aes))");
937                 break;
938
939         case 100:
940                 ret += tcrypt_test("hmac(md5)");
941                 break;
942
943         case 101:
944                 ret += tcrypt_test("hmac(sha1)");
945                 break;
946
947         case 102:
948                 ret += tcrypt_test("hmac(sha256)");
949                 break;
950
951         case 103:
952                 ret += tcrypt_test("hmac(sha384)");
953                 break;
954
955         case 104:
956                 ret += tcrypt_test("hmac(sha512)");
957                 break;
958
959         case 105:
960                 ret += tcrypt_test("hmac(sha224)");
961                 break;
962
963         case 106:
964                 ret += tcrypt_test("xcbc(aes)");
965                 break;
966
967         case 107:
968                 ret += tcrypt_test("hmac(rmd128)");
969                 break;
970
971         case 108:
972                 ret += tcrypt_test("hmac(rmd160)");
973                 break;
974
975         case 109:
976                 ret += tcrypt_test("vmac(aes)");
977                 break;
978
979         case 150:
980                 ret += tcrypt_test("ansi_cprng");
981                 break;
982
983         case 200:
984                 test_cipher_speed("ecb(aes)", ENCRYPT, sec, NULL, 0,
985                                 speed_template_16_24_32);
986                 test_cipher_speed("ecb(aes)", DECRYPT, sec, NULL, 0,
987                                 speed_template_16_24_32);
988                 test_cipher_speed("cbc(aes)", ENCRYPT, sec, NULL, 0,
989                                 speed_template_16_24_32);
990                 test_cipher_speed("cbc(aes)", DECRYPT, sec, NULL, 0,
991                                 speed_template_16_24_32);
992                 test_cipher_speed("lrw(aes)", ENCRYPT, sec, NULL, 0,
993                                 speed_template_32_40_48);
994                 test_cipher_speed("lrw(aes)", DECRYPT, sec, NULL, 0,
995                                 speed_template_32_40_48);
996                 test_cipher_speed("xts(aes)", ENCRYPT, sec, NULL, 0,
997                                 speed_template_32_48_64);
998                 test_cipher_speed("xts(aes)", DECRYPT, sec, NULL, 0,
999                                 speed_template_32_48_64);
1000                 break;
1001
1002         case 201:
1003                 test_cipher_speed("ecb(des3_ede)", ENCRYPT, sec,
1004                                 des3_speed_template, DES3_SPEED_VECTORS,
1005                                 speed_template_24);
1006                 test_cipher_speed("ecb(des3_ede)", DECRYPT, sec,
1007                                 des3_speed_template, DES3_SPEED_VECTORS,
1008                                 speed_template_24);
1009                 test_cipher_speed("cbc(des3_ede)", ENCRYPT, sec,
1010                                 des3_speed_template, DES3_SPEED_VECTORS,
1011                                 speed_template_24);
1012                 test_cipher_speed("cbc(des3_ede)", DECRYPT, sec,
1013                                 des3_speed_template, DES3_SPEED_VECTORS,
1014                                 speed_template_24);
1015                 break;
1016
1017         case 202:
1018                 test_cipher_speed("ecb(twofish)", ENCRYPT, sec, NULL, 0,
1019                                 speed_template_16_24_32);
1020                 test_cipher_speed("ecb(twofish)", DECRYPT, sec, NULL, 0,
1021                                 speed_template_16_24_32);
1022                 test_cipher_speed("cbc(twofish)", ENCRYPT, sec, NULL, 0,
1023                                 speed_template_16_24_32);
1024                 test_cipher_speed("cbc(twofish)", DECRYPT, sec, NULL, 0,
1025                                 speed_template_16_24_32);
1026                 break;
1027
1028         case 203:
1029                 test_cipher_speed("ecb(blowfish)", ENCRYPT, sec, NULL, 0,
1030                                   speed_template_8_32);
1031                 test_cipher_speed("ecb(blowfish)", DECRYPT, sec, NULL, 0,
1032                                   speed_template_8_32);
1033                 test_cipher_speed("cbc(blowfish)", ENCRYPT, sec, NULL, 0,
1034                                   speed_template_8_32);
1035                 test_cipher_speed("cbc(blowfish)", DECRYPT, sec, NULL, 0,
1036                                   speed_template_8_32);
1037                 break;
1038
1039         case 204:
1040                 test_cipher_speed("ecb(des)", ENCRYPT, sec, NULL, 0,
1041                                   speed_template_8);
1042                 test_cipher_speed("ecb(des)", DECRYPT, sec, NULL, 0,
1043                                   speed_template_8);
1044                 test_cipher_speed("cbc(des)", ENCRYPT, sec, NULL, 0,
1045                                   speed_template_8);
1046                 test_cipher_speed("cbc(des)", DECRYPT, sec, NULL, 0,
1047                                   speed_template_8);
1048                 break;
1049
1050         case 205:
1051                 test_cipher_speed("ecb(camellia)", ENCRYPT, sec, NULL, 0,
1052                                 speed_template_16_24_32);
1053                 test_cipher_speed("ecb(camellia)", DECRYPT, sec, NULL, 0,
1054                                 speed_template_16_24_32);
1055                 test_cipher_speed("cbc(camellia)", ENCRYPT, sec, NULL, 0,
1056                                 speed_template_16_24_32);
1057                 test_cipher_speed("cbc(camellia)", DECRYPT, sec, NULL, 0,
1058                                 speed_template_16_24_32);
1059                 break;
1060
1061         case 206:
1062                 test_cipher_speed("salsa20", ENCRYPT, sec, NULL, 0,
1063                                   speed_template_16_32);
1064                 break;
1065
1066         case 300:
1067                 /* fall through */
1068
1069         case 301:
1070                 test_hash_speed("md4", sec, generic_hash_speed_template);
1071                 if (mode > 300 && mode < 400) break;
1072
1073         case 302:
1074                 test_hash_speed("md5", sec, generic_hash_speed_template);
1075                 if (mode > 300 && mode < 400) break;
1076
1077         case 303:
1078                 test_hash_speed("sha1", sec, generic_hash_speed_template);
1079                 if (mode > 300 && mode < 400) break;
1080
1081         case 304:
1082                 test_hash_speed("sha256", sec, generic_hash_speed_template);
1083                 if (mode > 300 && mode < 400) break;
1084
1085         case 305:
1086                 test_hash_speed("sha384", sec, generic_hash_speed_template);
1087                 if (mode > 300 && mode < 400) break;
1088
1089         case 306:
1090                 test_hash_speed("sha512", sec, generic_hash_speed_template);
1091                 if (mode > 300 && mode < 400) break;
1092
1093         case 307:
1094                 test_hash_speed("wp256", sec, generic_hash_speed_template);
1095                 if (mode > 300 && mode < 400) break;
1096
1097         case 308:
1098                 test_hash_speed("wp384", sec, generic_hash_speed_template);
1099                 if (mode > 300 && mode < 400) break;
1100
1101         case 309:
1102                 test_hash_speed("wp512", sec, generic_hash_speed_template);
1103                 if (mode > 300 && mode < 400) break;
1104
1105         case 310:
1106                 test_hash_speed("tgr128", sec, generic_hash_speed_template);
1107                 if (mode > 300 && mode < 400) break;
1108
1109         case 311:
1110                 test_hash_speed("tgr160", sec, generic_hash_speed_template);
1111                 if (mode > 300 && mode < 400) break;
1112
1113         case 312:
1114                 test_hash_speed("tgr192", sec, generic_hash_speed_template);
1115                 if (mode > 300 && mode < 400) break;
1116
1117         case 313:
1118                 test_hash_speed("sha224", sec, generic_hash_speed_template);
1119                 if (mode > 300 && mode < 400) break;
1120
1121         case 314:
1122                 test_hash_speed("rmd128", sec, generic_hash_speed_template);
1123                 if (mode > 300 && mode < 400) break;
1124
1125         case 315:
1126                 test_hash_speed("rmd160", sec, generic_hash_speed_template);
1127                 if (mode > 300 && mode < 400) break;
1128
1129         case 316:
1130                 test_hash_speed("rmd256", sec, generic_hash_speed_template);
1131                 if (mode > 300 && mode < 400) break;
1132
1133         case 317:
1134                 test_hash_speed("rmd320", sec, generic_hash_speed_template);
1135                 if (mode > 300 && mode < 400) break;
1136
1137         case 318:
1138                 test_hash_speed("ghash-generic", sec, hash_speed_template_16);
1139                 if (mode > 300 && mode < 400) break;
1140
1141         case 399:
1142                 break;
1143
1144         case 400:
1145                 /* fall through */
1146
1147         case 401:
1148                 test_ahash_speed("md4", sec, generic_hash_speed_template);
1149                 if (mode > 400 && mode < 500) break;
1150
1151         case 402:
1152                 test_ahash_speed("md5", sec, generic_hash_speed_template);
1153                 if (mode > 400 && mode < 500) break;
1154
1155         case 403:
1156                 test_ahash_speed("sha1", sec, generic_hash_speed_template);
1157                 if (mode > 400 && mode < 500) break;
1158
1159         case 404:
1160                 test_ahash_speed("sha256", sec, generic_hash_speed_template);
1161                 if (mode > 400 && mode < 500) break;
1162
1163         case 405:
1164                 test_ahash_speed("sha384", sec, generic_hash_speed_template);
1165                 if (mode > 400 && mode < 500) break;
1166
1167         case 406:
1168                 test_ahash_speed("sha512", sec, generic_hash_speed_template);
1169                 if (mode > 400 && mode < 500) break;
1170
1171         case 407:
1172                 test_ahash_speed("wp256", sec, generic_hash_speed_template);
1173                 if (mode > 400 && mode < 500) break;
1174
1175         case 408:
1176                 test_ahash_speed("wp384", sec, generic_hash_speed_template);
1177                 if (mode > 400 && mode < 500) break;
1178
1179         case 409:
1180                 test_ahash_speed("wp512", sec, generic_hash_speed_template);
1181                 if (mode > 400 && mode < 500) break;
1182
1183         case 410:
1184                 test_ahash_speed("tgr128", sec, generic_hash_speed_template);
1185                 if (mode > 400 && mode < 500) break;
1186
1187         case 411:
1188                 test_ahash_speed("tgr160", sec, generic_hash_speed_template);
1189                 if (mode > 400 && mode < 500) break;
1190
1191         case 412:
1192                 test_ahash_speed("tgr192", sec, generic_hash_speed_template);
1193                 if (mode > 400 && mode < 500) break;
1194
1195         case 413:
1196                 test_ahash_speed("sha224", sec, generic_hash_speed_template);
1197                 if (mode > 400 && mode < 500) break;
1198
1199         case 414:
1200                 test_ahash_speed("rmd128", sec, generic_hash_speed_template);
1201                 if (mode > 400 && mode < 500) break;
1202
1203         case 415:
1204                 test_ahash_speed("rmd160", sec, generic_hash_speed_template);
1205                 if (mode > 400 && mode < 500) break;
1206
1207         case 416:
1208                 test_ahash_speed("rmd256", sec, generic_hash_speed_template);
1209                 if (mode > 400 && mode < 500) break;
1210
1211         case 417:
1212                 test_ahash_speed("rmd320", sec, generic_hash_speed_template);
1213                 if (mode > 400 && mode < 500) break;
1214
1215         case 499:
1216                 break;
1217
1218         case 1000:
1219                 test_available();
1220                 break;
1221         }
1222
1223         return ret;
1224 }
1225
1226 static int do_alg_test(const char *alg, u32 type, u32 mask)
1227 {
1228         return crypto_has_alg(alg, type, mask ?: CRYPTO_ALG_TYPE_MASK) ?
1229                0 : -ENOENT;
1230 }
1231
1232 static int __init tcrypt_mod_init(void)
1233 {
1234         int err = -ENOMEM;
1235         int i;
1236
1237         for (i = 0; i < TVMEMSIZE; i++) {
1238                 tvmem[i] = (void *)__get_free_page(GFP_KERNEL);
1239                 if (!tvmem[i])
1240                         goto err_free_tv;
1241         }
1242
1243         if (alg)
1244                 err = do_alg_test(alg, type, mask);
1245         else
1246                 err = do_test(mode);
1247
1248         if (err) {
1249                 printk(KERN_ERR "tcrypt: one or more tests failed!\n");
1250                 goto err_free_tv;
1251         }
1252
1253         /* We intentionaly return -EAGAIN to prevent keeping the module,
1254          * unless we're running in fips mode. It does all its work from
1255          * init() and doesn't offer any runtime functionality, but in
1256          * the fips case, checking for a successful load is helpful.
1257          * => we don't need it in the memory, do we?
1258          *                                        -- mludvig
1259          */
1260         if (!fips_enabled)
1261                 err = -EAGAIN;
1262
1263 err_free_tv:
1264         for (i = 0; i < TVMEMSIZE && tvmem[i]; i++)
1265                 free_page((unsigned long)tvmem[i]);
1266
1267         return err;
1268 }
1269
1270 /*
1271  * If an init function is provided, an exit function must also be provided
1272  * to allow module unload.
1273  */
1274 static void __exit tcrypt_mod_fini(void) { }
1275
1276 module_init(tcrypt_mod_init);
1277 module_exit(tcrypt_mod_fini);
1278
1279 module_param(alg, charp, 0);
1280 module_param(type, uint, 0);
1281 module_param(mask, uint, 0);
1282 module_param(mode, int, 0);
1283 module_param(sec, uint, 0);
1284 MODULE_PARM_DESC(sec, "Length in seconds of speed tests "
1285                       "(defaults to zero which uses CPU cycles instead)");
1286
1287 MODULE_LICENSE("GPL");
1288 MODULE_DESCRIPTION("Quick & dirty crypto testing module");
1289 MODULE_AUTHOR("James Morris <jmorris@intercode.com.au>");