crypto: tcrypt - add xts(serpent) tests
[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  * 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                 ret += tcrypt_test("lrw(twofish)");
994                 break;
995
996         case 9:
997                 ret += tcrypt_test("ecb(serpent)");
998                 ret += tcrypt_test("cbc(serpent)");
999                 ret += tcrypt_test("ctr(serpent)");
1000                 ret += tcrypt_test("lrw(serpent)");
1001                 ret += tcrypt_test("xts(serpent)");
1002                 break;
1003
1004         case 10:
1005                 ret += tcrypt_test("ecb(aes)");
1006                 ret += tcrypt_test("cbc(aes)");
1007                 ret += tcrypt_test("lrw(aes)");
1008                 ret += tcrypt_test("xts(aes)");
1009                 ret += tcrypt_test("ctr(aes)");
1010                 ret += tcrypt_test("rfc3686(ctr(aes))");
1011                 break;
1012
1013         case 11:
1014                 ret += tcrypt_test("sha384");
1015                 break;
1016
1017         case 12:
1018                 ret += tcrypt_test("sha512");
1019                 break;
1020
1021         case 13:
1022                 ret += tcrypt_test("deflate");
1023                 break;
1024
1025         case 14:
1026                 ret += tcrypt_test("ecb(cast5)");
1027                 break;
1028
1029         case 15:
1030                 ret += tcrypt_test("ecb(cast6)");
1031                 break;
1032
1033         case 16:
1034                 ret += tcrypt_test("ecb(arc4)");
1035                 break;
1036
1037         case 17:
1038                 ret += tcrypt_test("michael_mic");
1039                 break;
1040
1041         case 18:
1042                 ret += tcrypt_test("crc32c");
1043                 break;
1044
1045         case 19:
1046                 ret += tcrypt_test("ecb(tea)");
1047                 break;
1048
1049         case 20:
1050                 ret += tcrypt_test("ecb(xtea)");
1051                 break;
1052
1053         case 21:
1054                 ret += tcrypt_test("ecb(khazad)");
1055                 break;
1056
1057         case 22:
1058                 ret += tcrypt_test("wp512");
1059                 break;
1060
1061         case 23:
1062                 ret += tcrypt_test("wp384");
1063                 break;
1064
1065         case 24:
1066                 ret += tcrypt_test("wp256");
1067                 break;
1068
1069         case 25:
1070                 ret += tcrypt_test("ecb(tnepres)");
1071                 break;
1072
1073         case 26:
1074                 ret += tcrypt_test("ecb(anubis)");
1075                 ret += tcrypt_test("cbc(anubis)");
1076                 break;
1077
1078         case 27:
1079                 ret += tcrypt_test("tgr192");
1080                 break;
1081
1082         case 28:
1083
1084                 ret += tcrypt_test("tgr160");
1085                 break;
1086
1087         case 29:
1088                 ret += tcrypt_test("tgr128");
1089                 break;
1090
1091         case 30:
1092                 ret += tcrypt_test("ecb(xeta)");
1093                 break;
1094
1095         case 31:
1096                 ret += tcrypt_test("pcbc(fcrypt)");
1097                 break;
1098
1099         case 32:
1100                 ret += tcrypt_test("ecb(camellia)");
1101                 ret += tcrypt_test("cbc(camellia)");
1102                 break;
1103         case 33:
1104                 ret += tcrypt_test("sha224");
1105                 break;
1106
1107         case 34:
1108                 ret += tcrypt_test("salsa20");
1109                 break;
1110
1111         case 35:
1112                 ret += tcrypt_test("gcm(aes)");
1113                 break;
1114
1115         case 36:
1116                 ret += tcrypt_test("lzo");
1117                 break;
1118
1119         case 37:
1120                 ret += tcrypt_test("ccm(aes)");
1121                 break;
1122
1123         case 38:
1124                 ret += tcrypt_test("cts(cbc(aes))");
1125                 break;
1126
1127         case 39:
1128                 ret += tcrypt_test("rmd128");
1129                 break;
1130
1131         case 40:
1132                 ret += tcrypt_test("rmd160");
1133                 break;
1134
1135         case 41:
1136                 ret += tcrypt_test("rmd256");
1137                 break;
1138
1139         case 42:
1140                 ret += tcrypt_test("rmd320");
1141                 break;
1142
1143         case 43:
1144                 ret += tcrypt_test("ecb(seed)");
1145                 break;
1146
1147         case 44:
1148                 ret += tcrypt_test("zlib");
1149                 break;
1150
1151         case 45:
1152                 ret += tcrypt_test("rfc4309(ccm(aes))");
1153                 break;
1154
1155         case 100:
1156                 ret += tcrypt_test("hmac(md5)");
1157                 break;
1158
1159         case 101:
1160                 ret += tcrypt_test("hmac(sha1)");
1161                 break;
1162
1163         case 102:
1164                 ret += tcrypt_test("hmac(sha256)");
1165                 break;
1166
1167         case 103:
1168                 ret += tcrypt_test("hmac(sha384)");
1169                 break;
1170
1171         case 104:
1172                 ret += tcrypt_test("hmac(sha512)");
1173                 break;
1174
1175         case 105:
1176                 ret += tcrypt_test("hmac(sha224)");
1177                 break;
1178
1179         case 106:
1180                 ret += tcrypt_test("xcbc(aes)");
1181                 break;
1182
1183         case 107:
1184                 ret += tcrypt_test("hmac(rmd128)");
1185                 break;
1186
1187         case 108:
1188                 ret += tcrypt_test("hmac(rmd160)");
1189                 break;
1190
1191         case 109:
1192                 ret += tcrypt_test("vmac(aes)");
1193                 break;
1194
1195         case 150:
1196                 ret += tcrypt_test("ansi_cprng");
1197                 break;
1198
1199         case 151:
1200                 ret += tcrypt_test("rfc4106(gcm(aes))");
1201                 break;
1202
1203         case 200:
1204                 test_cipher_speed("ecb(aes)", ENCRYPT, sec, NULL, 0,
1205                                 speed_template_16_24_32);
1206                 test_cipher_speed("ecb(aes)", DECRYPT, sec, NULL, 0,
1207                                 speed_template_16_24_32);
1208                 test_cipher_speed("cbc(aes)", ENCRYPT, sec, NULL, 0,
1209                                 speed_template_16_24_32);
1210                 test_cipher_speed("cbc(aes)", DECRYPT, sec, NULL, 0,
1211                                 speed_template_16_24_32);
1212                 test_cipher_speed("lrw(aes)", ENCRYPT, sec, NULL, 0,
1213                                 speed_template_32_40_48);
1214                 test_cipher_speed("lrw(aes)", DECRYPT, sec, NULL, 0,
1215                                 speed_template_32_40_48);
1216                 test_cipher_speed("xts(aes)", ENCRYPT, sec, NULL, 0,
1217                                 speed_template_32_48_64);
1218                 test_cipher_speed("xts(aes)", DECRYPT, sec, NULL, 0,
1219                                 speed_template_32_48_64);
1220                 test_cipher_speed("ctr(aes)", ENCRYPT, sec, NULL, 0,
1221                                 speed_template_16_24_32);
1222                 test_cipher_speed("ctr(aes)", DECRYPT, sec, NULL, 0,
1223                                 speed_template_16_24_32);
1224                 break;
1225
1226         case 201:
1227                 test_cipher_speed("ecb(des3_ede)", ENCRYPT, sec,
1228                                 des3_speed_template, DES3_SPEED_VECTORS,
1229                                 speed_template_24);
1230                 test_cipher_speed("ecb(des3_ede)", DECRYPT, sec,
1231                                 des3_speed_template, DES3_SPEED_VECTORS,
1232                                 speed_template_24);
1233                 test_cipher_speed("cbc(des3_ede)", ENCRYPT, sec,
1234                                 des3_speed_template, DES3_SPEED_VECTORS,
1235                                 speed_template_24);
1236                 test_cipher_speed("cbc(des3_ede)", DECRYPT, sec,
1237                                 des3_speed_template, DES3_SPEED_VECTORS,
1238                                 speed_template_24);
1239                 break;
1240
1241         case 202:
1242                 test_cipher_speed("ecb(twofish)", ENCRYPT, sec, NULL, 0,
1243                                 speed_template_16_24_32);
1244                 test_cipher_speed("ecb(twofish)", DECRYPT, sec, NULL, 0,
1245                                 speed_template_16_24_32);
1246                 test_cipher_speed("cbc(twofish)", ENCRYPT, sec, NULL, 0,
1247                                 speed_template_16_24_32);
1248                 test_cipher_speed("cbc(twofish)", DECRYPT, sec, NULL, 0,
1249                                 speed_template_16_24_32);
1250                 test_cipher_speed("ctr(twofish)", ENCRYPT, sec, NULL, 0,
1251                                 speed_template_16_24_32);
1252                 test_cipher_speed("ctr(twofish)", DECRYPT, sec, NULL, 0,
1253                                 speed_template_16_24_32);
1254                 test_cipher_speed("lrw(twofish)", ENCRYPT, sec, NULL, 0,
1255                                 speed_template_32_40_48);
1256                 test_cipher_speed("lrw(twofish)", DECRYPT, sec, NULL, 0,
1257                                 speed_template_32_40_48);
1258                 break;
1259
1260         case 203:
1261                 test_cipher_speed("ecb(blowfish)", ENCRYPT, sec, NULL, 0,
1262                                   speed_template_8_32);
1263                 test_cipher_speed("ecb(blowfish)", DECRYPT, sec, NULL, 0,
1264                                   speed_template_8_32);
1265                 test_cipher_speed("cbc(blowfish)", ENCRYPT, sec, NULL, 0,
1266                                   speed_template_8_32);
1267                 test_cipher_speed("cbc(blowfish)", DECRYPT, sec, NULL, 0,
1268                                   speed_template_8_32);
1269                 test_cipher_speed("ctr(blowfish)", ENCRYPT, sec, NULL, 0,
1270                                   speed_template_8_32);
1271                 test_cipher_speed("ctr(blowfish)", DECRYPT, sec, NULL, 0,
1272                                   speed_template_8_32);
1273                 break;
1274
1275         case 204:
1276                 test_cipher_speed("ecb(des)", ENCRYPT, sec, NULL, 0,
1277                                   speed_template_8);
1278                 test_cipher_speed("ecb(des)", DECRYPT, sec, NULL, 0,
1279                                   speed_template_8);
1280                 test_cipher_speed("cbc(des)", ENCRYPT, sec, NULL, 0,
1281                                   speed_template_8);
1282                 test_cipher_speed("cbc(des)", DECRYPT, sec, NULL, 0,
1283                                   speed_template_8);
1284                 break;
1285
1286         case 205:
1287                 test_cipher_speed("ecb(camellia)", ENCRYPT, sec, NULL, 0,
1288                                 speed_template_16_24_32);
1289                 test_cipher_speed("ecb(camellia)", DECRYPT, sec, NULL, 0,
1290                                 speed_template_16_24_32);
1291                 test_cipher_speed("cbc(camellia)", ENCRYPT, sec, NULL, 0,
1292                                 speed_template_16_24_32);
1293                 test_cipher_speed("cbc(camellia)", DECRYPT, sec, NULL, 0,
1294                                 speed_template_16_24_32);
1295                 break;
1296
1297         case 206:
1298                 test_cipher_speed("salsa20", ENCRYPT, sec, NULL, 0,
1299                                   speed_template_16_32);
1300                 break;
1301
1302         case 207:
1303                 test_cipher_speed("ecb(serpent)", ENCRYPT, sec, NULL, 0,
1304                                   speed_template_16_32);
1305                 test_cipher_speed("ecb(serpent)", DECRYPT, sec, NULL, 0,
1306                                   speed_template_16_32);
1307                 test_cipher_speed("cbc(serpent)", ENCRYPT, sec, NULL, 0,
1308                                   speed_template_16_32);
1309                 test_cipher_speed("cbc(serpent)", DECRYPT, sec, NULL, 0,
1310                                   speed_template_16_32);
1311                 test_cipher_speed("ctr(serpent)", ENCRYPT, sec, NULL, 0,
1312                                   speed_template_16_32);
1313                 test_cipher_speed("ctr(serpent)", DECRYPT, sec, NULL, 0,
1314                                   speed_template_16_32);
1315                 test_cipher_speed("lrw(serpent)", ENCRYPT, sec, NULL, 0,
1316                                   speed_template_32_48);
1317                 test_cipher_speed("lrw(serpent)", DECRYPT, sec, NULL, 0,
1318                                   speed_template_32_48);
1319                 test_cipher_speed("xts(serpent)", ENCRYPT, sec, NULL, 0,
1320                                   speed_template_32_64);
1321                 test_cipher_speed("xts(serpent)", DECRYPT, sec, NULL, 0,
1322                                   speed_template_32_64);
1323                 break;
1324
1325         case 300:
1326                 /* fall through */
1327
1328         case 301:
1329                 test_hash_speed("md4", sec, generic_hash_speed_template);
1330                 if (mode > 300 && mode < 400) break;
1331
1332         case 302:
1333                 test_hash_speed("md5", sec, generic_hash_speed_template);
1334                 if (mode > 300 && mode < 400) break;
1335
1336         case 303:
1337                 test_hash_speed("sha1", sec, generic_hash_speed_template);
1338                 if (mode > 300 && mode < 400) break;
1339
1340         case 304:
1341                 test_hash_speed("sha256", sec, generic_hash_speed_template);
1342                 if (mode > 300 && mode < 400) break;
1343
1344         case 305:
1345                 test_hash_speed("sha384", sec, generic_hash_speed_template);
1346                 if (mode > 300 && mode < 400) break;
1347
1348         case 306:
1349                 test_hash_speed("sha512", sec, generic_hash_speed_template);
1350                 if (mode > 300 && mode < 400) break;
1351
1352         case 307:
1353                 test_hash_speed("wp256", sec, generic_hash_speed_template);
1354                 if (mode > 300 && mode < 400) break;
1355
1356         case 308:
1357                 test_hash_speed("wp384", sec, generic_hash_speed_template);
1358                 if (mode > 300 && mode < 400) break;
1359
1360         case 309:
1361                 test_hash_speed("wp512", sec, generic_hash_speed_template);
1362                 if (mode > 300 && mode < 400) break;
1363
1364         case 310:
1365                 test_hash_speed("tgr128", sec, generic_hash_speed_template);
1366                 if (mode > 300 && mode < 400) break;
1367
1368         case 311:
1369                 test_hash_speed("tgr160", sec, generic_hash_speed_template);
1370                 if (mode > 300 && mode < 400) break;
1371
1372         case 312:
1373                 test_hash_speed("tgr192", sec, generic_hash_speed_template);
1374                 if (mode > 300 && mode < 400) break;
1375
1376         case 313:
1377                 test_hash_speed("sha224", sec, generic_hash_speed_template);
1378                 if (mode > 300 && mode < 400) break;
1379
1380         case 314:
1381                 test_hash_speed("rmd128", sec, generic_hash_speed_template);
1382                 if (mode > 300 && mode < 400) break;
1383
1384         case 315:
1385                 test_hash_speed("rmd160", sec, generic_hash_speed_template);
1386                 if (mode > 300 && mode < 400) break;
1387
1388         case 316:
1389                 test_hash_speed("rmd256", sec, generic_hash_speed_template);
1390                 if (mode > 300 && mode < 400) break;
1391
1392         case 317:
1393                 test_hash_speed("rmd320", sec, generic_hash_speed_template);
1394                 if (mode > 300 && mode < 400) break;
1395
1396         case 318:
1397                 test_hash_speed("ghash-generic", sec, hash_speed_template_16);
1398                 if (mode > 300 && mode < 400) break;
1399
1400         case 399:
1401                 break;
1402
1403         case 400:
1404                 /* fall through */
1405
1406         case 401:
1407                 test_ahash_speed("md4", sec, generic_hash_speed_template);
1408                 if (mode > 400 && mode < 500) break;
1409
1410         case 402:
1411                 test_ahash_speed("md5", sec, generic_hash_speed_template);
1412                 if (mode > 400 && mode < 500) break;
1413
1414         case 403:
1415                 test_ahash_speed("sha1", sec, generic_hash_speed_template);
1416                 if (mode > 400 && mode < 500) break;
1417
1418         case 404:
1419                 test_ahash_speed("sha256", sec, generic_hash_speed_template);
1420                 if (mode > 400 && mode < 500) break;
1421
1422         case 405:
1423                 test_ahash_speed("sha384", sec, generic_hash_speed_template);
1424                 if (mode > 400 && mode < 500) break;
1425
1426         case 406:
1427                 test_ahash_speed("sha512", sec, generic_hash_speed_template);
1428                 if (mode > 400 && mode < 500) break;
1429
1430         case 407:
1431                 test_ahash_speed("wp256", sec, generic_hash_speed_template);
1432                 if (mode > 400 && mode < 500) break;
1433
1434         case 408:
1435                 test_ahash_speed("wp384", sec, generic_hash_speed_template);
1436                 if (mode > 400 && mode < 500) break;
1437
1438         case 409:
1439                 test_ahash_speed("wp512", sec, generic_hash_speed_template);
1440                 if (mode > 400 && mode < 500) break;
1441
1442         case 410:
1443                 test_ahash_speed("tgr128", sec, generic_hash_speed_template);
1444                 if (mode > 400 && mode < 500) break;
1445
1446         case 411:
1447                 test_ahash_speed("tgr160", sec, generic_hash_speed_template);
1448                 if (mode > 400 && mode < 500) break;
1449
1450         case 412:
1451                 test_ahash_speed("tgr192", sec, generic_hash_speed_template);
1452                 if (mode > 400 && mode < 500) break;
1453
1454         case 413:
1455                 test_ahash_speed("sha224", sec, generic_hash_speed_template);
1456                 if (mode > 400 && mode < 500) break;
1457
1458         case 414:
1459                 test_ahash_speed("rmd128", sec, generic_hash_speed_template);
1460                 if (mode > 400 && mode < 500) break;
1461
1462         case 415:
1463                 test_ahash_speed("rmd160", sec, generic_hash_speed_template);
1464                 if (mode > 400 && mode < 500) break;
1465
1466         case 416:
1467                 test_ahash_speed("rmd256", sec, generic_hash_speed_template);
1468                 if (mode > 400 && mode < 500) break;
1469
1470         case 417:
1471                 test_ahash_speed("rmd320", sec, generic_hash_speed_template);
1472                 if (mode > 400 && mode < 500) break;
1473
1474         case 499:
1475                 break;
1476
1477         case 500:
1478                 test_acipher_speed("ecb(aes)", ENCRYPT, sec, NULL, 0,
1479                                    speed_template_16_24_32);
1480                 test_acipher_speed("ecb(aes)", DECRYPT, sec, NULL, 0,
1481                                    speed_template_16_24_32);
1482                 test_acipher_speed("cbc(aes)", ENCRYPT, sec, NULL, 0,
1483                                    speed_template_16_24_32);
1484                 test_acipher_speed("cbc(aes)", DECRYPT, sec, NULL, 0,
1485                                    speed_template_16_24_32);
1486                 test_acipher_speed("lrw(aes)", ENCRYPT, sec, NULL, 0,
1487                                    speed_template_32_40_48);
1488                 test_acipher_speed("lrw(aes)", DECRYPT, sec, NULL, 0,
1489                                    speed_template_32_40_48);
1490                 test_acipher_speed("xts(aes)", ENCRYPT, sec, NULL, 0,
1491                                    speed_template_32_48_64);
1492                 test_acipher_speed("xts(aes)", DECRYPT, sec, NULL, 0,
1493                                    speed_template_32_48_64);
1494                 test_acipher_speed("ctr(aes)", ENCRYPT, sec, NULL, 0,
1495                                    speed_template_16_24_32);
1496                 test_acipher_speed("ctr(aes)", DECRYPT, sec, NULL, 0,
1497                                    speed_template_16_24_32);
1498                 break;
1499
1500         case 501:
1501                 test_acipher_speed("ecb(des3_ede)", ENCRYPT, sec,
1502                                    des3_speed_template, DES3_SPEED_VECTORS,
1503                                    speed_template_24);
1504                 test_acipher_speed("ecb(des3_ede)", DECRYPT, sec,
1505                                    des3_speed_template, DES3_SPEED_VECTORS,
1506                                    speed_template_24);
1507                 test_acipher_speed("cbc(des3_ede)", ENCRYPT, sec,
1508                                    des3_speed_template, DES3_SPEED_VECTORS,
1509                                    speed_template_24);
1510                 test_acipher_speed("cbc(des3_ede)", DECRYPT, sec,
1511                                    des3_speed_template, DES3_SPEED_VECTORS,
1512                                    speed_template_24);
1513                 break;
1514
1515         case 502:
1516                 test_acipher_speed("ecb(des)", ENCRYPT, sec, NULL, 0,
1517                                    speed_template_8);
1518                 test_acipher_speed("ecb(des)", DECRYPT, sec, NULL, 0,
1519                                    speed_template_8);
1520                 test_acipher_speed("cbc(des)", ENCRYPT, sec, NULL, 0,
1521                                    speed_template_8);
1522                 test_acipher_speed("cbc(des)", DECRYPT, sec, NULL, 0,
1523                                    speed_template_8);
1524                 break;
1525
1526         case 503:
1527                 test_acipher_speed("ecb(serpent)", ENCRYPT, sec, NULL, 0,
1528                                    speed_template_16_32);
1529                 test_acipher_speed("ecb(serpent)", DECRYPT, sec, NULL, 0,
1530                                    speed_template_16_32);
1531                 test_acipher_speed("cbc(serpent)", ENCRYPT, sec, NULL, 0,
1532                                    speed_template_16_32);
1533                 test_acipher_speed("cbc(serpent)", DECRYPT, sec, NULL, 0,
1534                                    speed_template_16_32);
1535                 test_acipher_speed("ctr(serpent)", ENCRYPT, sec, NULL, 0,
1536                                    speed_template_16_32);
1537                 test_acipher_speed("ctr(serpent)", DECRYPT, sec, NULL, 0,
1538                                    speed_template_16_32);
1539                 test_acipher_speed("lrw(serpent)", ENCRYPT, sec, NULL, 0,
1540                                    speed_template_32_48);
1541                 test_acipher_speed("lrw(serpent)", DECRYPT, sec, NULL, 0,
1542                                    speed_template_32_48);
1543                 test_acipher_speed("xts(serpent)", ENCRYPT, sec, NULL, 0,
1544                                    speed_template_32_64);
1545                 test_acipher_speed("xts(serpent)", DECRYPT, sec, NULL, 0,
1546                                    speed_template_32_64);
1547                 break;
1548
1549         case 1000:
1550                 test_available();
1551                 break;
1552         }
1553
1554         return ret;
1555 }
1556
1557 static int do_alg_test(const char *alg, u32 type, u32 mask)
1558 {
1559         return crypto_has_alg(alg, type, mask ?: CRYPTO_ALG_TYPE_MASK) ?
1560                0 : -ENOENT;
1561 }
1562
1563 static int __init tcrypt_mod_init(void)
1564 {
1565         int err = -ENOMEM;
1566         int i;
1567
1568         for (i = 0; i < TVMEMSIZE; i++) {
1569                 tvmem[i] = (void *)__get_free_page(GFP_KERNEL);
1570                 if (!tvmem[i])
1571                         goto err_free_tv;
1572         }
1573
1574         if (alg)
1575                 err = do_alg_test(alg, type, mask);
1576         else
1577                 err = do_test(mode);
1578
1579         if (err) {
1580                 printk(KERN_ERR "tcrypt: one or more tests failed!\n");
1581                 goto err_free_tv;
1582         }
1583
1584         /* We intentionaly return -EAGAIN to prevent keeping the module,
1585          * unless we're running in fips mode. It does all its work from
1586          * init() and doesn't offer any runtime functionality, but in
1587          * the fips case, checking for a successful load is helpful.
1588          * => we don't need it in the memory, do we?
1589          *                                        -- mludvig
1590          */
1591         if (!fips_enabled)
1592                 err = -EAGAIN;
1593
1594 err_free_tv:
1595         for (i = 0; i < TVMEMSIZE && tvmem[i]; i++)
1596                 free_page((unsigned long)tvmem[i]);
1597
1598         return err;
1599 }
1600
1601 /*
1602  * If an init function is provided, an exit function must also be provided
1603  * to allow module unload.
1604  */
1605 static void __exit tcrypt_mod_fini(void) { }
1606
1607 module_init(tcrypt_mod_init);
1608 module_exit(tcrypt_mod_fini);
1609
1610 module_param(alg, charp, 0);
1611 module_param(type, uint, 0);
1612 module_param(mask, uint, 0);
1613 module_param(mode, int, 0);
1614 module_param(sec, uint, 0);
1615 MODULE_PARM_DESC(sec, "Length in seconds of speed tests "
1616                       "(defaults to zero which uses CPU cycles instead)");
1617
1618 MODULE_LICENSE("GPL");
1619 MODULE_DESCRIPTION("Quick & dirty crypto testing module");
1620 MODULE_AUTHOR("James Morris <jmorris@intercode.com.au>");