cifs: no need to use rcu_assign_pointer on immutable keys
[linux-2.6] / crypto / testmgr.c
1 /*
2  * Algorithm testing framework and tests.
3  *
4  * Copyright (c) 2002 James Morris <jmorris@intercode.com.au>
5  * Copyright (c) 2002 Jean-Francois Dive <jef@linuxbe.org>
6  * Copyright (c) 2007 Nokia Siemens Networks
7  * Copyright (c) 2008 Herbert Xu <herbert@gondor.apana.org.au>
8  *
9  * This program is free software; you can redistribute it and/or modify it
10  * under the terms of the GNU General Public License as published by the Free
11  * Software Foundation; either version 2 of the License, or (at your option)
12  * any later version.
13  *
14  */
15
16 #include <crypto/hash.h>
17 #include <linux/err.h>
18 #include <linux/module.h>
19 #include <linux/scatterlist.h>
20 #include <linux/slab.h>
21 #include <linux/string.h>
22
23 #include "internal.h"
24 #include "testmgr.h"
25
26 /*
27  * Need slab memory for testing (size in number of pages).
28  */
29 #define XBUFSIZE        8
30
31 /*
32  * Indexes into the xbuf to simulate cross-page access.
33  */
34 #define IDX1            32
35 #define IDX2            32400
36 #define IDX3            1
37 #define IDX4            8193
38 #define IDX5            22222
39 #define IDX6            17101
40 #define IDX7            27333
41 #define IDX8            3000
42
43 /*
44 * Used by test_cipher()
45 */
46 #define ENCRYPT 1
47 #define DECRYPT 0
48
49 struct tcrypt_result {
50         struct completion completion;
51         int err;
52 };
53
54 struct aead_test_suite {
55         struct {
56                 struct aead_testvec *vecs;
57                 unsigned int count;
58         } enc, dec;
59 };
60
61 struct cipher_test_suite {
62         struct {
63                 struct cipher_testvec *vecs;
64                 unsigned int count;
65         } enc, dec;
66 };
67
68 struct comp_test_suite {
69         struct {
70                 struct comp_testvec *vecs;
71                 unsigned int count;
72         } comp, decomp;
73 };
74
75 struct pcomp_test_suite {
76         struct {
77                 struct pcomp_testvec *vecs;
78                 unsigned int count;
79         } comp, decomp;
80 };
81
82 struct hash_test_suite {
83         struct hash_testvec *vecs;
84         unsigned int count;
85 };
86
87 struct alg_test_desc {
88         const char *alg;
89         int (*test)(const struct alg_test_desc *desc, const char *driver,
90                     u32 type, u32 mask);
91
92         union {
93                 struct aead_test_suite aead;
94                 struct cipher_test_suite cipher;
95                 struct comp_test_suite comp;
96                 struct pcomp_test_suite pcomp;
97                 struct hash_test_suite hash;
98         } suite;
99 };
100
101 static unsigned int IDX[8] = { IDX1, IDX2, IDX3, IDX4, IDX5, IDX6, IDX7, IDX8 };
102
103 static char *xbuf[XBUFSIZE];
104 static char *axbuf[XBUFSIZE];
105
106 static void hexdump(unsigned char *buf, unsigned int len)
107 {
108         print_hex_dump(KERN_CONT, "", DUMP_PREFIX_OFFSET,
109                         16, 1,
110                         buf, len, false);
111 }
112
113 static void tcrypt_complete(struct crypto_async_request *req, int err)
114 {
115         struct tcrypt_result *res = req->data;
116
117         if (err == -EINPROGRESS)
118                 return;
119
120         res->err = err;
121         complete(&res->completion);
122 }
123
124 static int test_hash(struct crypto_ahash *tfm, struct hash_testvec *template,
125                      unsigned int tcount)
126 {
127         const char *algo = crypto_tfm_alg_driver_name(crypto_ahash_tfm(tfm));
128         unsigned int i, j, k, temp;
129         struct scatterlist sg[8];
130         char result[64];
131         struct ahash_request *req;
132         struct tcrypt_result tresult;
133         int ret;
134         void *hash_buff;
135
136         init_completion(&tresult.completion);
137
138         req = ahash_request_alloc(tfm, GFP_KERNEL);
139         if (!req) {
140                 printk(KERN_ERR "alg: hash: Failed to allocate request for "
141                        "%s\n", algo);
142                 ret = -ENOMEM;
143                 goto out_noreq;
144         }
145         ahash_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
146                                    tcrypt_complete, &tresult);
147
148         for (i = 0; i < tcount; i++) {
149                 memset(result, 0, 64);
150
151                 hash_buff = xbuf[0];
152
153                 memcpy(hash_buff, template[i].plaintext, template[i].psize);
154                 sg_init_one(&sg[0], hash_buff, template[i].psize);
155
156                 if (template[i].ksize) {
157                         crypto_ahash_clear_flags(tfm, ~0);
158                         ret = crypto_ahash_setkey(tfm, template[i].key,
159                                                   template[i].ksize);
160                         if (ret) {
161                                 printk(KERN_ERR "alg: hash: setkey failed on "
162                                        "test %d for %s: ret=%d\n", i + 1, algo,
163                                        -ret);
164                                 goto out;
165                         }
166                 }
167
168                 ahash_request_set_crypt(req, sg, result, template[i].psize);
169                 ret = crypto_ahash_digest(req);
170                 switch (ret) {
171                 case 0:
172                         break;
173                 case -EINPROGRESS:
174                 case -EBUSY:
175                         ret = wait_for_completion_interruptible(
176                                 &tresult.completion);
177                         if (!ret && !(ret = tresult.err)) {
178                                 INIT_COMPLETION(tresult.completion);
179                                 break;
180                         }
181                         /* fall through */
182                 default:
183                         printk(KERN_ERR "alg: hash: digest failed on test %d "
184                                "for %s: ret=%d\n", i + 1, algo, -ret);
185                         goto out;
186                 }
187
188                 if (memcmp(result, template[i].digest,
189                            crypto_ahash_digestsize(tfm))) {
190                         printk(KERN_ERR "alg: hash: Test %d failed for %s\n",
191                                i + 1, algo);
192                         hexdump(result, crypto_ahash_digestsize(tfm));
193                         ret = -EINVAL;
194                         goto out;
195                 }
196         }
197
198         j = 0;
199         for (i = 0; i < tcount; i++) {
200                 if (template[i].np) {
201                         j++;
202                         memset(result, 0, 64);
203
204                         temp = 0;
205                         sg_init_table(sg, template[i].np);
206                         for (k = 0; k < template[i].np; k++) {
207                                 sg_set_buf(&sg[k],
208                                            memcpy(xbuf[IDX[k] >> PAGE_SHIFT] +
209                                                   offset_in_page(IDX[k]),
210                                                   template[i].plaintext + temp,
211                                                   template[i].tap[k]),
212                                            template[i].tap[k]);
213                                 temp += template[i].tap[k];
214                         }
215
216                         if (template[i].ksize) {
217                                 crypto_ahash_clear_flags(tfm, ~0);
218                                 ret = crypto_ahash_setkey(tfm, template[i].key,
219                                                           template[i].ksize);
220
221                                 if (ret) {
222                                         printk(KERN_ERR "alg: hash: setkey "
223                                                "failed on chunking test %d "
224                                                "for %s: ret=%d\n", j, algo,
225                                                -ret);
226                                         goto out;
227                                 }
228                         }
229
230                         ahash_request_set_crypt(req, sg, result,
231                                                 template[i].psize);
232                         ret = crypto_ahash_digest(req);
233                         switch (ret) {
234                         case 0:
235                                 break;
236                         case -EINPROGRESS:
237                         case -EBUSY:
238                                 ret = wait_for_completion_interruptible(
239                                         &tresult.completion);
240                                 if (!ret && !(ret = tresult.err)) {
241                                         INIT_COMPLETION(tresult.completion);
242                                         break;
243                                 }
244                                 /* fall through */
245                         default:
246                                 printk(KERN_ERR "alg: hash: digest failed "
247                                        "on chunking test %d for %s: "
248                                        "ret=%d\n", j, algo, -ret);
249                                 goto out;
250                         }
251
252                         if (memcmp(result, template[i].digest,
253                                    crypto_ahash_digestsize(tfm))) {
254                                 printk(KERN_ERR "alg: hash: Chunking test %d "
255                                        "failed for %s\n", j, algo);
256                                 hexdump(result, crypto_ahash_digestsize(tfm));
257                                 ret = -EINVAL;
258                                 goto out;
259                         }
260                 }
261         }
262
263         ret = 0;
264
265 out:
266         ahash_request_free(req);
267 out_noreq:
268         return ret;
269 }
270
271 static int test_aead(struct crypto_aead *tfm, int enc,
272                      struct aead_testvec *template, unsigned int tcount)
273 {
274         const char *algo = crypto_tfm_alg_driver_name(crypto_aead_tfm(tfm));
275         unsigned int i, j, k, n, temp;
276         int ret = 0;
277         char *q;
278         char *key;
279         struct aead_request *req;
280         struct scatterlist sg[8];
281         struct scatterlist asg[8];
282         const char *e;
283         struct tcrypt_result result;
284         unsigned int authsize;
285         void *input;
286         void *assoc;
287         char iv[MAX_IVLEN];
288
289         if (enc == ENCRYPT)
290                 e = "encryption";
291         else
292                 e = "decryption";
293
294         init_completion(&result.completion);
295
296         req = aead_request_alloc(tfm, GFP_KERNEL);
297         if (!req) {
298                 printk(KERN_ERR "alg: aead: Failed to allocate request for "
299                        "%s\n", algo);
300                 ret = -ENOMEM;
301                 goto out;
302         }
303
304         aead_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
305                                   tcrypt_complete, &result);
306
307         for (i = 0, j = 0; i < tcount; i++) {
308                 if (!template[i].np) {
309                         j++;
310
311                         /* some tepmplates have no input data but they will
312                          * touch input
313                          */
314                         input = xbuf[0];
315                         assoc = axbuf[0];
316
317                         memcpy(input, template[i].input, template[i].ilen);
318                         memcpy(assoc, template[i].assoc, template[i].alen);
319                         if (template[i].iv)
320                                 memcpy(iv, template[i].iv, MAX_IVLEN);
321                         else
322                                 memset(iv, 0, MAX_IVLEN);
323
324                         crypto_aead_clear_flags(tfm, ~0);
325                         if (template[i].wk)
326                                 crypto_aead_set_flags(
327                                         tfm, CRYPTO_TFM_REQ_WEAK_KEY);
328
329                         key = template[i].key;
330
331                         ret = crypto_aead_setkey(tfm, key,
332                                                  template[i].klen);
333                         if (!ret == template[i].fail) {
334                                 printk(KERN_ERR "alg: aead: setkey failed on "
335                                        "test %d for %s: flags=%x\n", j, algo,
336                                        crypto_aead_get_flags(tfm));
337                                 goto out;
338                         } else if (ret)
339                                 continue;
340
341                         authsize = abs(template[i].rlen - template[i].ilen);
342                         ret = crypto_aead_setauthsize(tfm, authsize);
343                         if (ret) {
344                                 printk(KERN_ERR "alg: aead: Failed to set "
345                                        "authsize to %u on test %d for %s\n",
346                                        authsize, j, algo);
347                                 goto out;
348                         }
349
350                         sg_init_one(&sg[0], input,
351                                     template[i].ilen + (enc ? authsize : 0));
352
353                         sg_init_one(&asg[0], assoc, template[i].alen);
354
355                         aead_request_set_crypt(req, sg, sg,
356                                                template[i].ilen, iv);
357
358                         aead_request_set_assoc(req, asg, template[i].alen);
359
360                         ret = enc ?
361                                 crypto_aead_encrypt(req) :
362                                 crypto_aead_decrypt(req);
363
364                         switch (ret) {
365                         case 0:
366                                 break;
367                         case -EINPROGRESS:
368                         case -EBUSY:
369                                 ret = wait_for_completion_interruptible(
370                                         &result.completion);
371                                 if (!ret && !(ret = result.err)) {
372                                         INIT_COMPLETION(result.completion);
373                                         break;
374                                 }
375                                 /* fall through */
376                         default:
377                                 printk(KERN_ERR "alg: aead: %s failed on test "
378                                        "%d for %s: ret=%d\n", e, j, algo, -ret);
379                                 goto out;
380                         }
381
382                         q = input;
383                         if (memcmp(q, template[i].result, template[i].rlen)) {
384                                 printk(KERN_ERR "alg: aead: Test %d failed on "
385                                        "%s for %s\n", j, e, algo);
386                                 hexdump(q, template[i].rlen);
387                                 ret = -EINVAL;
388                                 goto out;
389                         }
390                 }
391         }
392
393         for (i = 0, j = 0; i < tcount; i++) {
394                 if (template[i].np) {
395                         j++;
396
397                         if (template[i].iv)
398                                 memcpy(iv, template[i].iv, MAX_IVLEN);
399                         else
400                                 memset(iv, 0, MAX_IVLEN);
401
402                         crypto_aead_clear_flags(tfm, ~0);
403                         if (template[i].wk)
404                                 crypto_aead_set_flags(
405                                         tfm, CRYPTO_TFM_REQ_WEAK_KEY);
406                         key = template[i].key;
407
408                         ret = crypto_aead_setkey(tfm, key, template[i].klen);
409                         if (!ret == template[i].fail) {
410                                 printk(KERN_ERR "alg: aead: setkey failed on "
411                                        "chunk test %d for %s: flags=%x\n", j,
412                                        algo, crypto_aead_get_flags(tfm));
413                                 goto out;
414                         } else if (ret)
415                                 continue;
416
417                         authsize = abs(template[i].rlen - template[i].ilen);
418
419                         ret = -EINVAL;
420                         sg_init_table(sg, template[i].np);
421                         for (k = 0, temp = 0; k < template[i].np; k++) {
422                                 if (WARN_ON(offset_in_page(IDX[k]) +
423                                             template[i].tap[k] > PAGE_SIZE))
424                                         goto out;
425
426                                 q = xbuf[IDX[k] >> PAGE_SHIFT] +
427                                     offset_in_page(IDX[k]);
428
429                                 memcpy(q, template[i].input + temp,
430                                        template[i].tap[k]);
431
432                                 n = template[i].tap[k];
433                                 if (k == template[i].np - 1 && enc)
434                                         n += authsize;
435                                 if (offset_in_page(q) + n < PAGE_SIZE)
436                                         q[n] = 0;
437
438                                 sg_set_buf(&sg[k], q, template[i].tap[k]);
439                                 temp += template[i].tap[k];
440                         }
441
442                         ret = crypto_aead_setauthsize(tfm, authsize);
443                         if (ret) {
444                                 printk(KERN_ERR "alg: aead: Failed to set "
445                                        "authsize to %u on chunk test %d for "
446                                        "%s\n", authsize, j, algo);
447                                 goto out;
448                         }
449
450                         if (enc) {
451                                 if (WARN_ON(sg[k - 1].offset +
452                                             sg[k - 1].length + authsize >
453                                             PAGE_SIZE)) {
454                                         ret = -EINVAL;
455                                         goto out;
456                                 }
457
458                                 sg[k - 1].length += authsize;
459                         }
460
461                         sg_init_table(asg, template[i].anp);
462                         for (k = 0, temp = 0; k < template[i].anp; k++) {
463                                 sg_set_buf(&asg[k],
464                                            memcpy(axbuf[IDX[k] >> PAGE_SHIFT] +
465                                                   offset_in_page(IDX[k]),
466                                                   template[i].assoc + temp,
467                                                   template[i].atap[k]),
468                                            template[i].atap[k]);
469                                 temp += template[i].atap[k];
470                         }
471
472                         aead_request_set_crypt(req, sg, sg,
473                                                template[i].ilen,
474                                                iv);
475
476                         aead_request_set_assoc(req, asg, template[i].alen);
477
478                         ret = enc ?
479                                 crypto_aead_encrypt(req) :
480                                 crypto_aead_decrypt(req);
481
482                         switch (ret) {
483                         case 0:
484                                 break;
485                         case -EINPROGRESS:
486                         case -EBUSY:
487                                 ret = wait_for_completion_interruptible(
488                                         &result.completion);
489                                 if (!ret && !(ret = result.err)) {
490                                         INIT_COMPLETION(result.completion);
491                                         break;
492                                 }
493                                 /* fall through */
494                         default:
495                                 printk(KERN_ERR "alg: aead: %s failed on "
496                                        "chunk test %d for %s: ret=%d\n", e, j,
497                                        algo, -ret);
498                                 goto out;
499                         }
500
501                         ret = -EINVAL;
502                         for (k = 0, temp = 0; k < template[i].np; k++) {
503                                 q = xbuf[IDX[k] >> PAGE_SHIFT] +
504                                     offset_in_page(IDX[k]);
505
506                                 n = template[i].tap[k];
507                                 if (k == template[i].np - 1)
508                                         n += enc ? authsize : -authsize;
509
510                                 if (memcmp(q, template[i].result + temp, n)) {
511                                         printk(KERN_ERR "alg: aead: Chunk "
512                                                "test %d failed on %s at page "
513                                                "%u for %s\n", j, e, k, algo);
514                                         hexdump(q, n);
515                                         goto out;
516                                 }
517
518                                 q += n;
519                                 if (k == template[i].np - 1 && !enc) {
520                                         if (memcmp(q, template[i].input +
521                                                       temp + n, authsize))
522                                                 n = authsize;
523                                         else
524                                                 n = 0;
525                                 } else {
526                                         for (n = 0; offset_in_page(q + n) &&
527                                                     q[n]; n++)
528                                                 ;
529                                 }
530                                 if (n) {
531                                         printk(KERN_ERR "alg: aead: Result "
532                                                "buffer corruption in chunk "
533                                                "test %d on %s at page %u for "
534                                                "%s: %u bytes:\n", j, e, k,
535                                                algo, n);
536                                         hexdump(q, n);
537                                         goto out;
538                                 }
539
540                                 temp += template[i].tap[k];
541                         }
542                 }
543         }
544
545         ret = 0;
546
547 out:
548         aead_request_free(req);
549         return ret;
550 }
551
552 static int test_cipher(struct crypto_cipher *tfm, int enc,
553                        struct cipher_testvec *template, unsigned int tcount)
554 {
555         const char *algo = crypto_tfm_alg_driver_name(crypto_cipher_tfm(tfm));
556         unsigned int i, j, k;
557         int ret;
558         char *q;
559         const char *e;
560         void *data;
561
562         if (enc == ENCRYPT)
563                 e = "encryption";
564         else
565                 e = "decryption";
566
567         j = 0;
568         for (i = 0; i < tcount; i++) {
569                 if (template[i].np)
570                         continue;
571
572                 j++;
573
574                 data = xbuf[0];
575                 memcpy(data, template[i].input, template[i].ilen);
576
577                 crypto_cipher_clear_flags(tfm, ~0);
578                 if (template[i].wk)
579                         crypto_cipher_set_flags(tfm, CRYPTO_TFM_REQ_WEAK_KEY);
580
581                 ret = crypto_cipher_setkey(tfm, template[i].key,
582                                            template[i].klen);
583                 if (!ret == template[i].fail) {
584                         printk(KERN_ERR "alg: cipher: setkey failed "
585                                "on test %d for %s: flags=%x\n", j,
586                                algo, crypto_cipher_get_flags(tfm));
587                         goto out;
588                 } else if (ret)
589                         continue;
590
591                 for (k = 0; k < template[i].ilen;
592                      k += crypto_cipher_blocksize(tfm)) {
593                         if (enc)
594                                 crypto_cipher_encrypt_one(tfm, data + k,
595                                                           data + k);
596                         else
597                                 crypto_cipher_decrypt_one(tfm, data + k,
598                                                           data + k);
599                 }
600
601                 q = data;
602                 if (memcmp(q, template[i].result, template[i].rlen)) {
603                         printk(KERN_ERR "alg: cipher: Test %d failed "
604                                "on %s for %s\n", j, e, algo);
605                         hexdump(q, template[i].rlen);
606                         ret = -EINVAL;
607                         goto out;
608                 }
609         }
610
611         ret = 0;
612
613 out:
614         return ret;
615 }
616
617 static int test_skcipher(struct crypto_ablkcipher *tfm, int enc,
618                          struct cipher_testvec *template, unsigned int tcount)
619 {
620         const char *algo =
621                 crypto_tfm_alg_driver_name(crypto_ablkcipher_tfm(tfm));
622         unsigned int i, j, k, n, temp;
623         int ret;
624         char *q;
625         struct ablkcipher_request *req;
626         struct scatterlist sg[8];
627         const char *e;
628         struct tcrypt_result result;
629         void *data;
630         char iv[MAX_IVLEN];
631
632         if (enc == ENCRYPT)
633                 e = "encryption";
634         else
635                 e = "decryption";
636
637         init_completion(&result.completion);
638
639         req = ablkcipher_request_alloc(tfm, GFP_KERNEL);
640         if (!req) {
641                 printk(KERN_ERR "alg: skcipher: Failed to allocate request "
642                        "for %s\n", algo);
643                 ret = -ENOMEM;
644                 goto out;
645         }
646
647         ablkcipher_request_set_callback(req, CRYPTO_TFM_REQ_MAY_BACKLOG,
648                                         tcrypt_complete, &result);
649
650         j = 0;
651         for (i = 0; i < tcount; i++) {
652                 if (template[i].iv)
653                         memcpy(iv, template[i].iv, MAX_IVLEN);
654                 else
655                         memset(iv, 0, MAX_IVLEN);
656
657                 if (!(template[i].np)) {
658                         j++;
659
660                         data = xbuf[0];
661                         memcpy(data, template[i].input, template[i].ilen);
662
663                         crypto_ablkcipher_clear_flags(tfm, ~0);
664                         if (template[i].wk)
665                                 crypto_ablkcipher_set_flags(
666                                         tfm, CRYPTO_TFM_REQ_WEAK_KEY);
667
668                         ret = crypto_ablkcipher_setkey(tfm, template[i].key,
669                                                        template[i].klen);
670                         if (!ret == template[i].fail) {
671                                 printk(KERN_ERR "alg: skcipher: setkey failed "
672                                        "on test %d for %s: flags=%x\n", j,
673                                        algo, crypto_ablkcipher_get_flags(tfm));
674                                 goto out;
675                         } else if (ret)
676                                 continue;
677
678                         sg_init_one(&sg[0], data, template[i].ilen);
679
680                         ablkcipher_request_set_crypt(req, sg, sg,
681                                                      template[i].ilen, iv);
682                         ret = enc ?
683                                 crypto_ablkcipher_encrypt(req) :
684                                 crypto_ablkcipher_decrypt(req);
685
686                         switch (ret) {
687                         case 0:
688                                 break;
689                         case -EINPROGRESS:
690                         case -EBUSY:
691                                 ret = wait_for_completion_interruptible(
692                                         &result.completion);
693                                 if (!ret && !((ret = result.err))) {
694                                         INIT_COMPLETION(result.completion);
695                                         break;
696                                 }
697                                 /* fall through */
698                         default:
699                                 printk(KERN_ERR "alg: skcipher: %s failed on "
700                                        "test %d for %s: ret=%d\n", e, j, algo,
701                                        -ret);
702                                 goto out;
703                         }
704
705                         q = data;
706                         if (memcmp(q, template[i].result, template[i].rlen)) {
707                                 printk(KERN_ERR "alg: skcipher: Test %d "
708                                        "failed on %s for %s\n", j, e, algo);
709                                 hexdump(q, template[i].rlen);
710                                 ret = -EINVAL;
711                                 goto out;
712                         }
713                 }
714         }
715
716         j = 0;
717         for (i = 0; i < tcount; i++) {
718
719                 if (template[i].iv)
720                         memcpy(iv, template[i].iv, MAX_IVLEN);
721                 else
722                         memset(iv, 0, MAX_IVLEN);
723
724                 if (template[i].np) {
725                         j++;
726
727                         crypto_ablkcipher_clear_flags(tfm, ~0);
728                         if (template[i].wk)
729                                 crypto_ablkcipher_set_flags(
730                                         tfm, CRYPTO_TFM_REQ_WEAK_KEY);
731
732                         ret = crypto_ablkcipher_setkey(tfm, template[i].key,
733                                                        template[i].klen);
734                         if (!ret == template[i].fail) {
735                                 printk(KERN_ERR "alg: skcipher: setkey failed "
736                                        "on chunk test %d for %s: flags=%x\n",
737                                        j, algo,
738                                        crypto_ablkcipher_get_flags(tfm));
739                                 goto out;
740                         } else if (ret)
741                                 continue;
742
743                         temp = 0;
744                         ret = -EINVAL;
745                         sg_init_table(sg, template[i].np);
746                         for (k = 0; k < template[i].np; k++) {
747                                 if (WARN_ON(offset_in_page(IDX[k]) +
748                                             template[i].tap[k] > PAGE_SIZE))
749                                         goto out;
750
751                                 q = xbuf[IDX[k] >> PAGE_SHIFT] +
752                                     offset_in_page(IDX[k]);
753
754                                 memcpy(q, template[i].input + temp,
755                                        template[i].tap[k]);
756
757                                 if (offset_in_page(q) + template[i].tap[k] <
758                                     PAGE_SIZE)
759                                         q[template[i].tap[k]] = 0;
760
761                                 sg_set_buf(&sg[k], q, template[i].tap[k]);
762
763                                 temp += template[i].tap[k];
764                         }
765
766                         ablkcipher_request_set_crypt(req, sg, sg,
767                                         template[i].ilen, iv);
768
769                         ret = enc ?
770                                 crypto_ablkcipher_encrypt(req) :
771                                 crypto_ablkcipher_decrypt(req);
772
773                         switch (ret) {
774                         case 0:
775                                 break;
776                         case -EINPROGRESS:
777                         case -EBUSY:
778                                 ret = wait_for_completion_interruptible(
779                                         &result.completion);
780                                 if (!ret && !((ret = result.err))) {
781                                         INIT_COMPLETION(result.completion);
782                                         break;
783                                 }
784                                 /* fall through */
785                         default:
786                                 printk(KERN_ERR "alg: skcipher: %s failed on "
787                                        "chunk test %d for %s: ret=%d\n", e, j,
788                                        algo, -ret);
789                                 goto out;
790                         }
791
792                         temp = 0;
793                         ret = -EINVAL;
794                         for (k = 0; k < template[i].np; k++) {
795                                 q = xbuf[IDX[k] >> PAGE_SHIFT] +
796                                     offset_in_page(IDX[k]);
797
798                                 if (memcmp(q, template[i].result + temp,
799                                            template[i].tap[k])) {
800                                         printk(KERN_ERR "alg: skcipher: Chunk "
801                                                "test %d failed on %s at page "
802                                                "%u for %s\n", j, e, k, algo);
803                                         hexdump(q, template[i].tap[k]);
804                                         goto out;
805                                 }
806
807                                 q += template[i].tap[k];
808                                 for (n = 0; offset_in_page(q + n) && q[n]; n++)
809                                         ;
810                                 if (n) {
811                                         printk(KERN_ERR "alg: skcipher: "
812                                                "Result buffer corruption in "
813                                                "chunk test %d on %s at page "
814                                                "%u for %s: %u bytes:\n", j, e,
815                                                k, algo, n);
816                                         hexdump(q, n);
817                                         goto out;
818                                 }
819                                 temp += template[i].tap[k];
820                         }
821                 }
822         }
823
824         ret = 0;
825
826 out:
827         ablkcipher_request_free(req);
828         return ret;
829 }
830
831 static int test_comp(struct crypto_comp *tfm, struct comp_testvec *ctemplate,
832                      struct comp_testvec *dtemplate, int ctcount, int dtcount)
833 {
834         const char *algo = crypto_tfm_alg_driver_name(crypto_comp_tfm(tfm));
835         unsigned int i;
836         char result[COMP_BUF_SIZE];
837         int ret;
838
839         for (i = 0; i < ctcount; i++) {
840                 int ilen, dlen = COMP_BUF_SIZE;
841
842                 memset(result, 0, sizeof (result));
843
844                 ilen = ctemplate[i].inlen;
845                 ret = crypto_comp_compress(tfm, ctemplate[i].input,
846                                            ilen, result, &dlen);
847                 if (ret) {
848                         printk(KERN_ERR "alg: comp: compression failed "
849                                "on test %d for %s: ret=%d\n", i + 1, algo,
850                                -ret);
851                         goto out;
852                 }
853
854                 if (dlen != ctemplate[i].outlen) {
855                         printk(KERN_ERR "alg: comp: Compression test %d "
856                                "failed for %s: output len = %d\n", i + 1, algo,
857                                dlen);
858                         ret = -EINVAL;
859                         goto out;
860                 }
861
862                 if (memcmp(result, ctemplate[i].output, dlen)) {
863                         printk(KERN_ERR "alg: comp: Compression test %d "
864                                "failed for %s\n", i + 1, algo);
865                         hexdump(result, dlen);
866                         ret = -EINVAL;
867                         goto out;
868                 }
869         }
870
871         for (i = 0; i < dtcount; i++) {
872                 int ilen, dlen = COMP_BUF_SIZE;
873
874                 memset(result, 0, sizeof (result));
875
876                 ilen = dtemplate[i].inlen;
877                 ret = crypto_comp_decompress(tfm, dtemplate[i].input,
878                                              ilen, result, &dlen);
879                 if (ret) {
880                         printk(KERN_ERR "alg: comp: decompression failed "
881                                "on test %d for %s: ret=%d\n", i + 1, algo,
882                                -ret);
883                         goto out;
884                 }
885
886                 if (dlen != dtemplate[i].outlen) {
887                         printk(KERN_ERR "alg: comp: Decompression test %d "
888                                "failed for %s: output len = %d\n", i + 1, algo,
889                                dlen);
890                         ret = -EINVAL;
891                         goto out;
892                 }
893
894                 if (memcmp(result, dtemplate[i].output, dlen)) {
895                         printk(KERN_ERR "alg: comp: Decompression test %d "
896                                "failed for %s\n", i + 1, algo);
897                         hexdump(result, dlen);
898                         ret = -EINVAL;
899                         goto out;
900                 }
901         }
902
903         ret = 0;
904
905 out:
906         return ret;
907 }
908
909 static int test_pcomp(struct crypto_pcomp *tfm,
910                       struct pcomp_testvec *ctemplate,
911                       struct pcomp_testvec *dtemplate, int ctcount,
912                       int dtcount)
913 {
914         const char *algo = crypto_tfm_alg_driver_name(crypto_pcomp_tfm(tfm));
915         unsigned int i;
916         char result[COMP_BUF_SIZE];
917         int error;
918
919         for (i = 0; i < ctcount; i++) {
920                 struct comp_request req;
921
922                 error = crypto_compress_setup(tfm, ctemplate[i].params,
923                                               ctemplate[i].paramsize);
924                 if (error) {
925                         pr_err("alg: pcomp: compression setup failed on test "
926                                "%d for %s: error=%d\n", i + 1, algo, error);
927                         return error;
928                 }
929
930                 error = crypto_compress_init(tfm);
931                 if (error) {
932                         pr_err("alg: pcomp: compression init failed on test "
933                                "%d for %s: error=%d\n", i + 1, algo, error);
934                         return error;
935                 }
936
937                 memset(result, 0, sizeof(result));
938
939                 req.next_in = ctemplate[i].input;
940                 req.avail_in = ctemplate[i].inlen / 2;
941                 req.next_out = result;
942                 req.avail_out = ctemplate[i].outlen / 2;
943
944                 error = crypto_compress_update(tfm, &req);
945                 if (error && (error != -EAGAIN || req.avail_in)) {
946                         pr_err("alg: pcomp: compression update failed on test "
947                                "%d for %s: error=%d\n", i + 1, algo, error);
948                         return error;
949                 }
950
951                 /* Add remaining input data */
952                 req.avail_in += (ctemplate[i].inlen + 1) / 2;
953
954                 error = crypto_compress_update(tfm, &req);
955                 if (error && (error != -EAGAIN || req.avail_in)) {
956                         pr_err("alg: pcomp: compression update failed on test "
957                                "%d for %s: error=%d\n", i + 1, algo, error);
958                         return error;
959                 }
960
961                 /* Provide remaining output space */
962                 req.avail_out += COMP_BUF_SIZE - ctemplate[i].outlen / 2;
963
964                 error = crypto_compress_final(tfm, &req);
965                 if (error) {
966                         pr_err("alg: pcomp: compression final failed on test "
967                                "%d for %s: error=%d\n", i + 1, algo, error);
968                         return error;
969                 }
970
971                 if (COMP_BUF_SIZE - req.avail_out != ctemplate[i].outlen) {
972                         pr_err("alg: comp: Compression test %d failed for %s: "
973                                "output len = %d (expected %d)\n", i + 1, algo,
974                                COMP_BUF_SIZE - req.avail_out,
975                                ctemplate[i].outlen);
976                         return -EINVAL;
977                 }
978
979                 if (memcmp(result, ctemplate[i].output, ctemplate[i].outlen)) {
980                         pr_err("alg: pcomp: Compression test %d failed for "
981                                "%s\n", i + 1, algo);
982                         hexdump(result, ctemplate[i].outlen);
983                         return -EINVAL;
984                 }
985         }
986
987         for (i = 0; i < dtcount; i++) {
988                 struct comp_request req;
989
990                 error = crypto_decompress_setup(tfm, dtemplate[i].params,
991                                                 dtemplate[i].paramsize);
992                 if (error) {
993                         pr_err("alg: pcomp: decompression setup failed on "
994                                "test %d for %s: error=%d\n", i + 1, algo,
995                                error);
996                         return error;
997                 }
998
999                 error = crypto_decompress_init(tfm);
1000                 if (error) {
1001                         pr_err("alg: pcomp: decompression init failed on test "
1002                                "%d for %s: error=%d\n", i + 1, algo, error);
1003                         return error;
1004                 }
1005
1006                 memset(result, 0, sizeof(result));
1007
1008                 req.next_in = dtemplate[i].input;
1009                 req.avail_in = dtemplate[i].inlen / 2;
1010                 req.next_out = result;
1011                 req.avail_out = dtemplate[i].outlen / 2;
1012
1013                 error = crypto_decompress_update(tfm, &req);
1014                 if (error  && (error != -EAGAIN || req.avail_in)) {
1015                         pr_err("alg: pcomp: decompression update failed on "
1016                                "test %d for %s: error=%d\n", i + 1, algo,
1017                                error);
1018                         return error;
1019                 }
1020
1021                 /* Add remaining input data */
1022                 req.avail_in += (dtemplate[i].inlen + 1) / 2;
1023
1024                 error = crypto_decompress_update(tfm, &req);
1025                 if (error  && (error != -EAGAIN || req.avail_in)) {
1026                         pr_err("alg: pcomp: decompression update failed on "
1027                                "test %d for %s: error=%d\n", i + 1, algo,
1028                                error);
1029                         return error;
1030                 }
1031
1032                 /* Provide remaining output space */
1033                 req.avail_out += COMP_BUF_SIZE - dtemplate[i].outlen / 2;
1034
1035                 error = crypto_decompress_final(tfm, &req);
1036                 if (error  && (error != -EAGAIN || req.avail_in)) {
1037                         pr_err("alg: pcomp: decompression final failed on "
1038                                "test %d for %s: error=%d\n", i + 1, algo,
1039                                error);
1040                         return error;
1041                 }
1042
1043                 if (COMP_BUF_SIZE - req.avail_out != dtemplate[i].outlen) {
1044                         pr_err("alg: comp: Decompression test %d failed for "
1045                                "%s: output len = %d (expected %d)\n", i + 1,
1046                                algo, COMP_BUF_SIZE - req.avail_out,
1047                                dtemplate[i].outlen);
1048                         return -EINVAL;
1049                 }
1050
1051                 if (memcmp(result, dtemplate[i].output, dtemplate[i].outlen)) {
1052                         pr_err("alg: pcomp: Decompression test %d failed for "
1053                                "%s\n", i + 1, algo);
1054                         hexdump(result, dtemplate[i].outlen);
1055                         return -EINVAL;
1056                 }
1057         }
1058
1059         return 0;
1060 }
1061
1062 static int alg_test_aead(const struct alg_test_desc *desc, const char *driver,
1063                          u32 type, u32 mask)
1064 {
1065         struct crypto_aead *tfm;
1066         int err = 0;
1067
1068         tfm = crypto_alloc_aead(driver, type, mask);
1069         if (IS_ERR(tfm)) {
1070                 printk(KERN_ERR "alg: aead: Failed to load transform for %s: "
1071                        "%ld\n", driver, PTR_ERR(tfm));
1072                 return PTR_ERR(tfm);
1073         }
1074
1075         if (desc->suite.aead.enc.vecs) {
1076                 err = test_aead(tfm, ENCRYPT, desc->suite.aead.enc.vecs,
1077                                 desc->suite.aead.enc.count);
1078                 if (err)
1079                         goto out;
1080         }
1081
1082         if (!err && desc->suite.aead.dec.vecs)
1083                 err = test_aead(tfm, DECRYPT, desc->suite.aead.dec.vecs,
1084                                 desc->suite.aead.dec.count);
1085
1086 out:
1087         crypto_free_aead(tfm);
1088         return err;
1089 }
1090
1091 static int alg_test_cipher(const struct alg_test_desc *desc,
1092                            const char *driver, u32 type, u32 mask)
1093 {
1094         struct crypto_cipher *tfm;
1095         int err = 0;
1096
1097         tfm = crypto_alloc_cipher(driver, type, mask);
1098         if (IS_ERR(tfm)) {
1099                 printk(KERN_ERR "alg: cipher: Failed to load transform for "
1100                        "%s: %ld\n", driver, PTR_ERR(tfm));
1101                 return PTR_ERR(tfm);
1102         }
1103
1104         if (desc->suite.cipher.enc.vecs) {
1105                 err = test_cipher(tfm, ENCRYPT, desc->suite.cipher.enc.vecs,
1106                                   desc->suite.cipher.enc.count);
1107                 if (err)
1108                         goto out;
1109         }
1110
1111         if (desc->suite.cipher.dec.vecs)
1112                 err = test_cipher(tfm, DECRYPT, desc->suite.cipher.dec.vecs,
1113                                   desc->suite.cipher.dec.count);
1114
1115 out:
1116         crypto_free_cipher(tfm);
1117         return err;
1118 }
1119
1120 static int alg_test_skcipher(const struct alg_test_desc *desc,
1121                              const char *driver, u32 type, u32 mask)
1122 {
1123         struct crypto_ablkcipher *tfm;
1124         int err = 0;
1125
1126         tfm = crypto_alloc_ablkcipher(driver, type, mask);
1127         if (IS_ERR(tfm)) {
1128                 printk(KERN_ERR "alg: skcipher: Failed to load transform for "
1129                        "%s: %ld\n", driver, PTR_ERR(tfm));
1130                 return PTR_ERR(tfm);
1131         }
1132
1133         if (desc->suite.cipher.enc.vecs) {
1134                 err = test_skcipher(tfm, ENCRYPT, desc->suite.cipher.enc.vecs,
1135                                     desc->suite.cipher.enc.count);
1136                 if (err)
1137                         goto out;
1138         }
1139
1140         if (desc->suite.cipher.dec.vecs)
1141                 err = test_skcipher(tfm, DECRYPT, desc->suite.cipher.dec.vecs,
1142                                     desc->suite.cipher.dec.count);
1143
1144 out:
1145         crypto_free_ablkcipher(tfm);
1146         return err;
1147 }
1148
1149 static int alg_test_comp(const struct alg_test_desc *desc, const char *driver,
1150                          u32 type, u32 mask)
1151 {
1152         struct crypto_comp *tfm;
1153         int err;
1154
1155         tfm = crypto_alloc_comp(driver, type, mask);
1156         if (IS_ERR(tfm)) {
1157                 printk(KERN_ERR "alg: comp: Failed to load transform for %s: "
1158                        "%ld\n", driver, PTR_ERR(tfm));
1159                 return PTR_ERR(tfm);
1160         }
1161
1162         err = test_comp(tfm, desc->suite.comp.comp.vecs,
1163                         desc->suite.comp.decomp.vecs,
1164                         desc->suite.comp.comp.count,
1165                         desc->suite.comp.decomp.count);
1166
1167         crypto_free_comp(tfm);
1168         return err;
1169 }
1170
1171 static int alg_test_pcomp(const struct alg_test_desc *desc, const char *driver,
1172                           u32 type, u32 mask)
1173 {
1174         struct crypto_pcomp *tfm;
1175         int err;
1176
1177         tfm = crypto_alloc_pcomp(driver, type, mask);
1178         if (IS_ERR(tfm)) {
1179                 pr_err("alg: pcomp: Failed to load transform for %s: %ld\n",
1180                        driver, PTR_ERR(tfm));
1181                 return PTR_ERR(tfm);
1182         }
1183
1184         err = test_pcomp(tfm, desc->suite.pcomp.comp.vecs,
1185                          desc->suite.pcomp.decomp.vecs,
1186                          desc->suite.pcomp.comp.count,
1187                          desc->suite.pcomp.decomp.count);
1188
1189         crypto_free_pcomp(tfm);
1190         return err;
1191 }
1192
1193 static int alg_test_hash(const struct alg_test_desc *desc, const char *driver,
1194                          u32 type, u32 mask)
1195 {
1196         struct crypto_ahash *tfm;
1197         int err;
1198
1199         tfm = crypto_alloc_ahash(driver, type, mask);
1200         if (IS_ERR(tfm)) {
1201                 printk(KERN_ERR "alg: hash: Failed to load transform for %s: "
1202                        "%ld\n", driver, PTR_ERR(tfm));
1203                 return PTR_ERR(tfm);
1204         }
1205
1206         err = test_hash(tfm, desc->suite.hash.vecs, desc->suite.hash.count);
1207
1208         crypto_free_ahash(tfm);
1209         return err;
1210 }
1211
1212 static int alg_test_crc32c(const struct alg_test_desc *desc,
1213                            const char *driver, u32 type, u32 mask)
1214 {
1215         struct crypto_shash *tfm;
1216         u32 val;
1217         int err;
1218
1219         err = alg_test_hash(desc, driver, type, mask);
1220         if (err)
1221                 goto out;
1222
1223         tfm = crypto_alloc_shash(driver, type, mask);
1224         if (IS_ERR(tfm)) {
1225                 printk(KERN_ERR "alg: crc32c: Failed to load transform for %s: "
1226                        "%ld\n", driver, PTR_ERR(tfm));
1227                 err = PTR_ERR(tfm);
1228                 goto out;
1229         }
1230
1231         do {
1232                 struct {
1233                         struct shash_desc shash;
1234                         char ctx[crypto_shash_descsize(tfm)];
1235                 } sdesc;
1236
1237                 sdesc.shash.tfm = tfm;
1238                 sdesc.shash.flags = 0;
1239
1240                 *(u32 *)sdesc.ctx = le32_to_cpu(420553207);
1241                 err = crypto_shash_final(&sdesc.shash, (u8 *)&val);
1242                 if (err) {
1243                         printk(KERN_ERR "alg: crc32c: Operation failed for "
1244                                "%s: %d\n", driver, err);
1245                         break;
1246                 }
1247
1248                 if (val != ~420553207) {
1249                         printk(KERN_ERR "alg: crc32c: Test failed for %s: "
1250                                "%d\n", driver, val);
1251                         err = -EINVAL;
1252                 }
1253         } while (0);
1254
1255         crypto_free_shash(tfm);
1256
1257 out:
1258         return err;
1259 }
1260
1261 /* Please keep this list sorted by algorithm name. */
1262 static const struct alg_test_desc alg_test_descs[] = {
1263         {
1264                 .alg = "cbc(aes)",
1265                 .test = alg_test_skcipher,
1266                 .suite = {
1267                         .cipher = {
1268                                 .enc = {
1269                                         .vecs = aes_cbc_enc_tv_template,
1270                                         .count = AES_CBC_ENC_TEST_VECTORS
1271                                 },
1272                                 .dec = {
1273                                         .vecs = aes_cbc_dec_tv_template,
1274                                         .count = AES_CBC_DEC_TEST_VECTORS
1275                                 }
1276                         }
1277                 }
1278         }, {
1279                 .alg = "cbc(anubis)",
1280                 .test = alg_test_skcipher,
1281                 .suite = {
1282                         .cipher = {
1283                                 .enc = {
1284                                         .vecs = anubis_cbc_enc_tv_template,
1285                                         .count = ANUBIS_CBC_ENC_TEST_VECTORS
1286                                 },
1287                                 .dec = {
1288                                         .vecs = anubis_cbc_dec_tv_template,
1289                                         .count = ANUBIS_CBC_DEC_TEST_VECTORS
1290                                 }
1291                         }
1292                 }
1293         }, {
1294                 .alg = "cbc(blowfish)",
1295                 .test = alg_test_skcipher,
1296                 .suite = {
1297                         .cipher = {
1298                                 .enc = {
1299                                         .vecs = bf_cbc_enc_tv_template,
1300                                         .count = BF_CBC_ENC_TEST_VECTORS
1301                                 },
1302                                 .dec = {
1303                                         .vecs = bf_cbc_dec_tv_template,
1304                                         .count = BF_CBC_DEC_TEST_VECTORS
1305                                 }
1306                         }
1307                 }
1308         }, {
1309                 .alg = "cbc(camellia)",
1310                 .test = alg_test_skcipher,
1311                 .suite = {
1312                         .cipher = {
1313                                 .enc = {
1314                                         .vecs = camellia_cbc_enc_tv_template,
1315                                         .count = CAMELLIA_CBC_ENC_TEST_VECTORS
1316                                 },
1317                                 .dec = {
1318                                         .vecs = camellia_cbc_dec_tv_template,
1319                                         .count = CAMELLIA_CBC_DEC_TEST_VECTORS
1320                                 }
1321                         }
1322                 }
1323         }, {
1324                 .alg = "cbc(des)",
1325                 .test = alg_test_skcipher,
1326                 .suite = {
1327                         .cipher = {
1328                                 .enc = {
1329                                         .vecs = des_cbc_enc_tv_template,
1330                                         .count = DES_CBC_ENC_TEST_VECTORS
1331                                 },
1332                                 .dec = {
1333                                         .vecs = des_cbc_dec_tv_template,
1334                                         .count = DES_CBC_DEC_TEST_VECTORS
1335                                 }
1336                         }
1337                 }
1338         }, {
1339                 .alg = "cbc(des3_ede)",
1340                 .test = alg_test_skcipher,
1341                 .suite = {
1342                         .cipher = {
1343                                 .enc = {
1344                                         .vecs = des3_ede_cbc_enc_tv_template,
1345                                         .count = DES3_EDE_CBC_ENC_TEST_VECTORS
1346                                 },
1347                                 .dec = {
1348                                         .vecs = des3_ede_cbc_dec_tv_template,
1349                                         .count = DES3_EDE_CBC_DEC_TEST_VECTORS
1350                                 }
1351                         }
1352                 }
1353         }, {
1354                 .alg = "cbc(twofish)",
1355                 .test = alg_test_skcipher,
1356                 .suite = {
1357                         .cipher = {
1358                                 .enc = {
1359                                         .vecs = tf_cbc_enc_tv_template,
1360                                         .count = TF_CBC_ENC_TEST_VECTORS
1361                                 },
1362                                 .dec = {
1363                                         .vecs = tf_cbc_dec_tv_template,
1364                                         .count = TF_CBC_DEC_TEST_VECTORS
1365                                 }
1366                         }
1367                 }
1368         }, {
1369                 .alg = "ccm(aes)",
1370                 .test = alg_test_aead,
1371                 .suite = {
1372                         .aead = {
1373                                 .enc = {
1374                                         .vecs = aes_ccm_enc_tv_template,
1375                                         .count = AES_CCM_ENC_TEST_VECTORS
1376                                 },
1377                                 .dec = {
1378                                         .vecs = aes_ccm_dec_tv_template,
1379                                         .count = AES_CCM_DEC_TEST_VECTORS
1380                                 }
1381                         }
1382                 }
1383         }, {
1384                 .alg = "crc32c",
1385                 .test = alg_test_crc32c,
1386                 .suite = {
1387                         .hash = {
1388                                 .vecs = crc32c_tv_template,
1389                                 .count = CRC32C_TEST_VECTORS
1390                         }
1391                 }
1392         }, {
1393                 .alg = "cts(cbc(aes))",
1394                 .test = alg_test_skcipher,
1395                 .suite = {
1396                         .cipher = {
1397                                 .enc = {
1398                                         .vecs = cts_mode_enc_tv_template,
1399                                         .count = CTS_MODE_ENC_TEST_VECTORS
1400                                 },
1401                                 .dec = {
1402                                         .vecs = cts_mode_dec_tv_template,
1403                                         .count = CTS_MODE_DEC_TEST_VECTORS
1404                                 }
1405                         }
1406                 }
1407         }, {
1408                 .alg = "deflate",
1409                 .test = alg_test_comp,
1410                 .suite = {
1411                         .comp = {
1412                                 .comp = {
1413                                         .vecs = deflate_comp_tv_template,
1414                                         .count = DEFLATE_COMP_TEST_VECTORS
1415                                 },
1416                                 .decomp = {
1417                                         .vecs = deflate_decomp_tv_template,
1418                                         .count = DEFLATE_DECOMP_TEST_VECTORS
1419                                 }
1420                         }
1421                 }
1422         }, {
1423                 .alg = "ecb(aes)",
1424                 .test = alg_test_skcipher,
1425                 .suite = {
1426                         .cipher = {
1427                                 .enc = {
1428                                         .vecs = aes_enc_tv_template,
1429                                         .count = AES_ENC_TEST_VECTORS
1430                                 },
1431                                 .dec = {
1432                                         .vecs = aes_dec_tv_template,
1433                                         .count = AES_DEC_TEST_VECTORS
1434                                 }
1435                         }
1436                 }
1437         }, {
1438                 .alg = "ecb(anubis)",
1439                 .test = alg_test_skcipher,
1440                 .suite = {
1441                         .cipher = {
1442                                 .enc = {
1443                                         .vecs = anubis_enc_tv_template,
1444                                         .count = ANUBIS_ENC_TEST_VECTORS
1445                                 },
1446                                 .dec = {
1447                                         .vecs = anubis_dec_tv_template,
1448                                         .count = ANUBIS_DEC_TEST_VECTORS
1449                                 }
1450                         }
1451                 }
1452         }, {
1453                 .alg = "ecb(arc4)",
1454                 .test = alg_test_skcipher,
1455                 .suite = {
1456                         .cipher = {
1457                                 .enc = {
1458                                         .vecs = arc4_enc_tv_template,
1459                                         .count = ARC4_ENC_TEST_VECTORS
1460                                 },
1461                                 .dec = {
1462                                         .vecs = arc4_dec_tv_template,
1463                                         .count = ARC4_DEC_TEST_VECTORS
1464                                 }
1465                         }
1466                 }
1467         }, {
1468                 .alg = "ecb(blowfish)",
1469                 .test = alg_test_skcipher,
1470                 .suite = {
1471                         .cipher = {
1472                                 .enc = {
1473                                         .vecs = bf_enc_tv_template,
1474                                         .count = BF_ENC_TEST_VECTORS
1475                                 },
1476                                 .dec = {
1477                                         .vecs = bf_dec_tv_template,
1478                                         .count = BF_DEC_TEST_VECTORS
1479                                 }
1480                         }
1481                 }
1482         }, {
1483                 .alg = "ecb(camellia)",
1484                 .test = alg_test_skcipher,
1485                 .suite = {
1486                         .cipher = {
1487                                 .enc = {
1488                                         .vecs = camellia_enc_tv_template,
1489                                         .count = CAMELLIA_ENC_TEST_VECTORS
1490                                 },
1491                                 .dec = {
1492                                         .vecs = camellia_dec_tv_template,
1493                                         .count = CAMELLIA_DEC_TEST_VECTORS
1494                                 }
1495                         }
1496                 }
1497         }, {
1498                 .alg = "ecb(cast5)",
1499                 .test = alg_test_skcipher,
1500                 .suite = {
1501                         .cipher = {
1502                                 .enc = {
1503                                         .vecs = cast5_enc_tv_template,
1504                                         .count = CAST5_ENC_TEST_VECTORS
1505                                 },
1506                                 .dec = {
1507                                         .vecs = cast5_dec_tv_template,
1508                                         .count = CAST5_DEC_TEST_VECTORS
1509                                 }
1510                         }
1511                 }
1512         }, {
1513                 .alg = "ecb(cast6)",
1514                 .test = alg_test_skcipher,
1515                 .suite = {
1516                         .cipher = {
1517                                 .enc = {
1518                                         .vecs = cast6_enc_tv_template,
1519                                         .count = CAST6_ENC_TEST_VECTORS
1520                                 },
1521                                 .dec = {
1522                                         .vecs = cast6_dec_tv_template,
1523                                         .count = CAST6_DEC_TEST_VECTORS
1524                                 }
1525                         }
1526                 }
1527         }, {
1528                 .alg = "ecb(des)",
1529                 .test = alg_test_skcipher,
1530                 .suite = {
1531                         .cipher = {
1532                                 .enc = {
1533                                         .vecs = des_enc_tv_template,
1534                                         .count = DES_ENC_TEST_VECTORS
1535                                 },
1536                                 .dec = {
1537                                         .vecs = des_dec_tv_template,
1538                                         .count = DES_DEC_TEST_VECTORS
1539                                 }
1540                         }
1541                 }
1542         }, {
1543                 .alg = "ecb(des3_ede)",
1544                 .test = alg_test_skcipher,
1545                 .suite = {
1546                         .cipher = {
1547                                 .enc = {
1548                                         .vecs = des3_ede_enc_tv_template,
1549                                         .count = DES3_EDE_ENC_TEST_VECTORS
1550                                 },
1551                                 .dec = {
1552                                         .vecs = des3_ede_dec_tv_template,
1553                                         .count = DES3_EDE_DEC_TEST_VECTORS
1554                                 }
1555                         }
1556                 }
1557         }, {
1558                 .alg = "ecb(khazad)",
1559                 .test = alg_test_skcipher,
1560                 .suite = {
1561                         .cipher = {
1562                                 .enc = {
1563                                         .vecs = khazad_enc_tv_template,
1564                                         .count = KHAZAD_ENC_TEST_VECTORS
1565                                 },
1566                                 .dec = {
1567                                         .vecs = khazad_dec_tv_template,
1568                                         .count = KHAZAD_DEC_TEST_VECTORS
1569                                 }
1570                         }
1571                 }
1572         }, {
1573                 .alg = "ecb(seed)",
1574                 .test = alg_test_skcipher,
1575                 .suite = {
1576                         .cipher = {
1577                                 .enc = {
1578                                         .vecs = seed_enc_tv_template,
1579                                         .count = SEED_ENC_TEST_VECTORS
1580                                 },
1581                                 .dec = {
1582                                         .vecs = seed_dec_tv_template,
1583                                         .count = SEED_DEC_TEST_VECTORS
1584                                 }
1585                         }
1586                 }
1587         }, {
1588                 .alg = "ecb(serpent)",
1589                 .test = alg_test_skcipher,
1590                 .suite = {
1591                         .cipher = {
1592                                 .enc = {
1593                                         .vecs = serpent_enc_tv_template,
1594                                         .count = SERPENT_ENC_TEST_VECTORS
1595                                 },
1596                                 .dec = {
1597                                         .vecs = serpent_dec_tv_template,
1598                                         .count = SERPENT_DEC_TEST_VECTORS
1599                                 }
1600                         }
1601                 }
1602         }, {
1603                 .alg = "ecb(tea)",
1604                 .test = alg_test_skcipher,
1605                 .suite = {
1606                         .cipher = {
1607                                 .enc = {
1608                                         .vecs = tea_enc_tv_template,
1609                                         .count = TEA_ENC_TEST_VECTORS
1610                                 },
1611                                 .dec = {
1612                                         .vecs = tea_dec_tv_template,
1613                                         .count = TEA_DEC_TEST_VECTORS
1614                                 }
1615                         }
1616                 }
1617         }, {
1618                 .alg = "ecb(tnepres)",
1619                 .test = alg_test_skcipher,
1620                 .suite = {
1621                         .cipher = {
1622                                 .enc = {
1623                                         .vecs = tnepres_enc_tv_template,
1624                                         .count = TNEPRES_ENC_TEST_VECTORS
1625                                 },
1626                                 .dec = {
1627                                         .vecs = tnepres_dec_tv_template,
1628                                         .count = TNEPRES_DEC_TEST_VECTORS
1629                                 }
1630                         }
1631                 }
1632         }, {
1633                 .alg = "ecb(twofish)",
1634                 .test = alg_test_skcipher,
1635                 .suite = {
1636                         .cipher = {
1637                                 .enc = {
1638                                         .vecs = tf_enc_tv_template,
1639                                         .count = TF_ENC_TEST_VECTORS
1640                                 },
1641                                 .dec = {
1642                                         .vecs = tf_dec_tv_template,
1643                                         .count = TF_DEC_TEST_VECTORS
1644                                 }
1645                         }
1646                 }
1647         }, {
1648                 .alg = "ecb(xeta)",
1649                 .test = alg_test_skcipher,
1650                 .suite = {
1651                         .cipher = {
1652                                 .enc = {
1653                                         .vecs = xeta_enc_tv_template,
1654                                         .count = XETA_ENC_TEST_VECTORS
1655                                 },
1656                                 .dec = {
1657                                         .vecs = xeta_dec_tv_template,
1658                                         .count = XETA_DEC_TEST_VECTORS
1659                                 }
1660                         }
1661                 }
1662         }, {
1663                 .alg = "ecb(xtea)",
1664                 .test = alg_test_skcipher,
1665                 .suite = {
1666                         .cipher = {
1667                                 .enc = {
1668                                         .vecs = xtea_enc_tv_template,
1669                                         .count = XTEA_ENC_TEST_VECTORS
1670                                 },
1671                                 .dec = {
1672                                         .vecs = xtea_dec_tv_template,
1673                                         .count = XTEA_DEC_TEST_VECTORS
1674                                 }
1675                         }
1676                 }
1677         }, {
1678                 .alg = "gcm(aes)",
1679                 .test = alg_test_aead,
1680                 .suite = {
1681                         .aead = {
1682                                 .enc = {
1683                                         .vecs = aes_gcm_enc_tv_template,
1684                                         .count = AES_GCM_ENC_TEST_VECTORS
1685                                 },
1686                                 .dec = {
1687                                         .vecs = aes_gcm_dec_tv_template,
1688                                         .count = AES_GCM_DEC_TEST_VECTORS
1689                                 }
1690                         }
1691                 }
1692         }, {
1693                 .alg = "hmac(md5)",
1694                 .test = alg_test_hash,
1695                 .suite = {
1696                         .hash = {
1697                                 .vecs = hmac_md5_tv_template,
1698                                 .count = HMAC_MD5_TEST_VECTORS
1699                         }
1700                 }
1701         }, {
1702                 .alg = "hmac(rmd128)",
1703                 .test = alg_test_hash,
1704                 .suite = {
1705                         .hash = {
1706                                 .vecs = hmac_rmd128_tv_template,
1707                                 .count = HMAC_RMD128_TEST_VECTORS
1708                         }
1709                 }
1710         }, {
1711                 .alg = "hmac(rmd160)",
1712                 .test = alg_test_hash,
1713                 .suite = {
1714                         .hash = {
1715                                 .vecs = hmac_rmd160_tv_template,
1716                                 .count = HMAC_RMD160_TEST_VECTORS
1717                         }
1718                 }
1719         }, {
1720                 .alg = "hmac(sha1)",
1721                 .test = alg_test_hash,
1722                 .suite = {
1723                         .hash = {
1724                                 .vecs = hmac_sha1_tv_template,
1725                                 .count = HMAC_SHA1_TEST_VECTORS
1726                         }
1727                 }
1728         }, {
1729                 .alg = "hmac(sha224)",
1730                 .test = alg_test_hash,
1731                 .suite = {
1732                         .hash = {
1733                                 .vecs = hmac_sha224_tv_template,
1734                                 .count = HMAC_SHA224_TEST_VECTORS
1735                         }
1736                 }
1737         }, {
1738                 .alg = "hmac(sha256)",
1739                 .test = alg_test_hash,
1740                 .suite = {
1741                         .hash = {
1742                                 .vecs = hmac_sha256_tv_template,
1743                                 .count = HMAC_SHA256_TEST_VECTORS
1744                         }
1745                 }
1746         }, {
1747                 .alg = "hmac(sha384)",
1748                 .test = alg_test_hash,
1749                 .suite = {
1750                         .hash = {
1751                                 .vecs = hmac_sha384_tv_template,
1752                                 .count = HMAC_SHA384_TEST_VECTORS
1753                         }
1754                 }
1755         }, {
1756                 .alg = "hmac(sha512)",
1757                 .test = alg_test_hash,
1758                 .suite = {
1759                         .hash = {
1760                                 .vecs = hmac_sha512_tv_template,
1761                                 .count = HMAC_SHA512_TEST_VECTORS
1762                         }
1763                 }
1764         }, {
1765                 .alg = "lrw(aes)",
1766                 .test = alg_test_skcipher,
1767                 .suite = {
1768                         .cipher = {
1769                                 .enc = {
1770                                         .vecs = aes_lrw_enc_tv_template,
1771                                         .count = AES_LRW_ENC_TEST_VECTORS
1772                                 },
1773                                 .dec = {
1774                                         .vecs = aes_lrw_dec_tv_template,
1775                                         .count = AES_LRW_DEC_TEST_VECTORS
1776                                 }
1777                         }
1778                 }
1779         }, {
1780                 .alg = "lzo",
1781                 .test = alg_test_comp,
1782                 .suite = {
1783                         .comp = {
1784                                 .comp = {
1785                                         .vecs = lzo_comp_tv_template,
1786                                         .count = LZO_COMP_TEST_VECTORS
1787                                 },
1788                                 .decomp = {
1789                                         .vecs = lzo_decomp_tv_template,
1790                                         .count = LZO_DECOMP_TEST_VECTORS
1791                                 }
1792                         }
1793                 }
1794         }, {
1795                 .alg = "md4",
1796                 .test = alg_test_hash,
1797                 .suite = {
1798                         .hash = {
1799                                 .vecs = md4_tv_template,
1800                                 .count = MD4_TEST_VECTORS
1801                         }
1802                 }
1803         }, {
1804                 .alg = "md5",
1805                 .test = alg_test_hash,
1806                 .suite = {
1807                         .hash = {
1808                                 .vecs = md5_tv_template,
1809                                 .count = MD5_TEST_VECTORS
1810                         }
1811                 }
1812         }, {
1813                 .alg = "michael_mic",
1814                 .test = alg_test_hash,
1815                 .suite = {
1816                         .hash = {
1817                                 .vecs = michael_mic_tv_template,
1818                                 .count = MICHAEL_MIC_TEST_VECTORS
1819                         }
1820                 }
1821         }, {
1822                 .alg = "pcbc(fcrypt)",
1823                 .test = alg_test_skcipher,
1824                 .suite = {
1825                         .cipher = {
1826                                 .enc = {
1827                                         .vecs = fcrypt_pcbc_enc_tv_template,
1828                                         .count = FCRYPT_ENC_TEST_VECTORS
1829                                 },
1830                                 .dec = {
1831                                         .vecs = fcrypt_pcbc_dec_tv_template,
1832                                         .count = FCRYPT_DEC_TEST_VECTORS
1833                                 }
1834                         }
1835                 }
1836         }, {
1837                 .alg = "rfc3686(ctr(aes))",
1838                 .test = alg_test_skcipher,
1839                 .suite = {
1840                         .cipher = {
1841                                 .enc = {
1842                                         .vecs = aes_ctr_enc_tv_template,
1843                                         .count = AES_CTR_ENC_TEST_VECTORS
1844                                 },
1845                                 .dec = {
1846                                         .vecs = aes_ctr_dec_tv_template,
1847                                         .count = AES_CTR_DEC_TEST_VECTORS
1848                                 }
1849                         }
1850                 }
1851         }, {
1852                 .alg = "rmd128",
1853                 .test = alg_test_hash,
1854                 .suite = {
1855                         .hash = {
1856                                 .vecs = rmd128_tv_template,
1857                                 .count = RMD128_TEST_VECTORS
1858                         }
1859                 }
1860         }, {
1861                 .alg = "rmd160",
1862                 .test = alg_test_hash,
1863                 .suite = {
1864                         .hash = {
1865                                 .vecs = rmd160_tv_template,
1866                                 .count = RMD160_TEST_VECTORS
1867                         }
1868                 }
1869         }, {
1870                 .alg = "rmd256",
1871                 .test = alg_test_hash,
1872                 .suite = {
1873                         .hash = {
1874                                 .vecs = rmd256_tv_template,
1875                                 .count = RMD256_TEST_VECTORS
1876                         }
1877                 }
1878         }, {
1879                 .alg = "rmd320",
1880                 .test = alg_test_hash,
1881                 .suite = {
1882                         .hash = {
1883                                 .vecs = rmd320_tv_template,
1884                                 .count = RMD320_TEST_VECTORS
1885                         }
1886                 }
1887         }, {
1888                 .alg = "salsa20",
1889                 .test = alg_test_skcipher,
1890                 .suite = {
1891                         .cipher = {
1892                                 .enc = {
1893                                         .vecs = salsa20_stream_enc_tv_template,
1894                                         .count = SALSA20_STREAM_ENC_TEST_VECTORS
1895                                 }
1896                         }
1897                 }
1898         }, {
1899                 .alg = "sha1",
1900                 .test = alg_test_hash,
1901                 .suite = {
1902                         .hash = {
1903                                 .vecs = sha1_tv_template,
1904                                 .count = SHA1_TEST_VECTORS
1905                         }
1906                 }
1907         }, {
1908                 .alg = "sha224",
1909                 .test = alg_test_hash,
1910                 .suite = {
1911                         .hash = {
1912                                 .vecs = sha224_tv_template,
1913                                 .count = SHA224_TEST_VECTORS
1914                         }
1915                 }
1916         }, {
1917                 .alg = "sha256",
1918                 .test = alg_test_hash,
1919                 .suite = {
1920                         .hash = {
1921                                 .vecs = sha256_tv_template,
1922                                 .count = SHA256_TEST_VECTORS
1923                         }
1924                 }
1925         }, {
1926                 .alg = "sha384",
1927                 .test = alg_test_hash,
1928                 .suite = {
1929                         .hash = {
1930                                 .vecs = sha384_tv_template,
1931                                 .count = SHA384_TEST_VECTORS
1932                         }
1933                 }
1934         }, {
1935                 .alg = "sha512",
1936                 .test = alg_test_hash,
1937                 .suite = {
1938                         .hash = {
1939                                 .vecs = sha512_tv_template,
1940                                 .count = SHA512_TEST_VECTORS
1941                         }
1942                 }
1943         }, {
1944                 .alg = "tgr128",
1945                 .test = alg_test_hash,
1946                 .suite = {
1947                         .hash = {
1948                                 .vecs = tgr128_tv_template,
1949                                 .count = TGR128_TEST_VECTORS
1950                         }
1951                 }
1952         }, {
1953                 .alg = "tgr160",
1954                 .test = alg_test_hash,
1955                 .suite = {
1956                         .hash = {
1957                                 .vecs = tgr160_tv_template,
1958                                 .count = TGR160_TEST_VECTORS
1959                         }
1960                 }
1961         }, {
1962                 .alg = "tgr192",
1963                 .test = alg_test_hash,
1964                 .suite = {
1965                         .hash = {
1966                                 .vecs = tgr192_tv_template,
1967                                 .count = TGR192_TEST_VECTORS
1968                         }
1969                 }
1970         }, {
1971                 .alg = "wp256",
1972                 .test = alg_test_hash,
1973                 .suite = {
1974                         .hash = {
1975                                 .vecs = wp256_tv_template,
1976                                 .count = WP256_TEST_VECTORS
1977                         }
1978                 }
1979         }, {
1980                 .alg = "wp384",
1981                 .test = alg_test_hash,
1982                 .suite = {
1983                         .hash = {
1984                                 .vecs = wp384_tv_template,
1985                                 .count = WP384_TEST_VECTORS
1986                         }
1987                 }
1988         }, {
1989                 .alg = "wp512",
1990                 .test = alg_test_hash,
1991                 .suite = {
1992                         .hash = {
1993                                 .vecs = wp512_tv_template,
1994                                 .count = WP512_TEST_VECTORS
1995                         }
1996                 }
1997         }, {
1998                 .alg = "xcbc(aes)",
1999                 .test = alg_test_hash,
2000                 .suite = {
2001                         .hash = {
2002                                 .vecs = aes_xcbc128_tv_template,
2003                                 .count = XCBC_AES_TEST_VECTORS
2004                         }
2005                 }
2006         }, {
2007                 .alg = "xts(aes)",
2008                 .test = alg_test_skcipher,
2009                 .suite = {
2010                         .cipher = {
2011                                 .enc = {
2012                                         .vecs = aes_xts_enc_tv_template,
2013                                         .count = AES_XTS_ENC_TEST_VECTORS
2014                                 },
2015                                 .dec = {
2016                                         .vecs = aes_xts_dec_tv_template,
2017                                         .count = AES_XTS_DEC_TEST_VECTORS
2018                                 }
2019                         }
2020                 }
2021         }, {
2022                 .alg = "zlib",
2023                 .test = alg_test_pcomp,
2024                 .suite = {
2025                         .pcomp = {
2026                                 .comp = {
2027                                         .vecs = zlib_comp_tv_template,
2028                                         .count = ZLIB_COMP_TEST_VECTORS
2029                                 },
2030                                 .decomp = {
2031                                         .vecs = zlib_decomp_tv_template,
2032                                         .count = ZLIB_DECOMP_TEST_VECTORS
2033                                 }
2034                         }
2035                 }
2036         }
2037 };
2038
2039 static int alg_find_test(const char *alg)
2040 {
2041         int start = 0;
2042         int end = ARRAY_SIZE(alg_test_descs);
2043
2044         while (start < end) {
2045                 int i = (start + end) / 2;
2046                 int diff = strcmp(alg_test_descs[i].alg, alg);
2047
2048                 if (diff > 0) {
2049                         end = i;
2050                         continue;
2051                 }
2052
2053                 if (diff < 0) {
2054                         start = i + 1;
2055                         continue;
2056                 }
2057
2058                 return i;
2059         }
2060
2061         return -1;
2062 }
2063
2064 int alg_test(const char *driver, const char *alg, u32 type, u32 mask)
2065 {
2066         int i;
2067         int rc;
2068
2069         if ((type & CRYPTO_ALG_TYPE_MASK) == CRYPTO_ALG_TYPE_CIPHER) {
2070                 char nalg[CRYPTO_MAX_ALG_NAME];
2071
2072                 if (snprintf(nalg, sizeof(nalg), "ecb(%s)", alg) >=
2073                     sizeof(nalg))
2074                         return -ENAMETOOLONG;
2075
2076                 i = alg_find_test(nalg);
2077                 if (i < 0)
2078                         goto notest;
2079
2080                 return alg_test_cipher(alg_test_descs + i, driver, type, mask);
2081         }
2082
2083         i = alg_find_test(alg);
2084         if (i < 0)
2085                 goto notest;
2086
2087         rc = alg_test_descs[i].test(alg_test_descs + i, driver,
2088                                       type, mask);
2089         if (fips_enabled && rc)
2090                 panic("%s: %s alg self test failed in fips mode!\n", driver, alg);
2091
2092         return rc;
2093
2094 notest:
2095         printk(KERN_INFO "alg: No test for %s (%s)\n", alg, driver);
2096         return 0;
2097 }
2098 EXPORT_SYMBOL_GPL(alg_test);
2099
2100 int __init testmgr_init(void)
2101 {
2102         int i;
2103
2104         for (i = 0; i < XBUFSIZE; i++) {
2105                 xbuf[i] = (void *)__get_free_page(GFP_KERNEL);
2106                 if (!xbuf[i])
2107                         goto err_free_xbuf;
2108         }
2109
2110         for (i = 0; i < XBUFSIZE; i++) {
2111                 axbuf[i] = (void *)__get_free_page(GFP_KERNEL);
2112                 if (!axbuf[i])
2113                         goto err_free_axbuf;
2114         }
2115
2116         return 0;
2117
2118 err_free_axbuf:
2119         for (i = 0; i < XBUFSIZE && axbuf[i]; i++)
2120                 free_page((unsigned long)axbuf[i]);
2121 err_free_xbuf:
2122         for (i = 0; i < XBUFSIZE && xbuf[i]; i++)
2123                 free_page((unsigned long)xbuf[i]);
2124
2125         return -ENOMEM;
2126 }
2127
2128 void testmgr_exit(void)
2129 {
2130         int i;
2131
2132         for (i = 0; i < XBUFSIZE; i++)
2133                 free_page((unsigned long)axbuf[i]);
2134         for (i = 0; i < XBUFSIZE; i++)
2135                 free_page((unsigned long)xbuf[i]);
2136 }