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