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  *
  10  * This program is free software; you can redistribute it and/or modify it
  11  * under the terms of the GNU General Public License as published by the Free
  12  * Software Foundation; either version 2 of the License, or (at your option)
  13  * any later version.
  14  *
  15  * 2004-08-09 Added cipher speed tests (Reyk Floeter <reyk@vantronix.net>)
  16  * 2003-09-14 Rewritten by Kartikey Mahendra Bhatt
  17  *
  18  */
  19
  20 #include <linux/init.h>
  21 #include <linux/module.h>
  22 #include <linux/mm.h>
  23 #include <linux/slab.h>
  24 #include <linux/scatterlist.h>
  25 #include <linux/string.h>
  26 #include <linux/crypto.h>
  27 #include <linux/highmem.h>
  28 #include <linux/moduleparam.h>
  29 #include <linux/jiffies.h>
  30 #include <linux/timex.h>
  31 #include <linux/interrupt.h>
  32 #include "tcrypt.h"
  33
  34 /*
  35  * Need to kmalloc() memory for testing kmap().
  36  */
  37 #define TVMEMSIZE       16384
  38 #define XBUFSIZE        32768
  39
  40 /*
  41  * Indexes into the xbuf to simulate cross-page access.
  42  */
  43 #define IDX1            37
  44 #define IDX2            32400
  45 #define IDX3            1
  46 #define IDX4            8193
  47 #define IDX5            22222
  48 #define IDX6            17101
  49 #define IDX7            27333
  50 #define IDX8            3000
  51
  52 /*
  53 * Used by test_cipher()
  54 */
  55 #define ENCRYPT 1
  56 #define DECRYPT 0
  57 #define MODE_ECB 1
  58 #define MODE_CBC 0
  59
  60 static unsigned int IDX[8] = { IDX1, IDX2, IDX3, IDX4, IDX5, IDX6, IDX7, IDX8 };
  61
  62 /*
  63  * Used by test_cipher_speed()
  64  */
  65 static unsigned int sec;
  66
  67 static int mode;
  68 static char *xbuf;
  69 static char *tvmem;
  70
  71 static char *check[] = {
  72         "des", "md5", "des3_ede", "rot13", "sha1", "sha256", "blowfish",
  73         "twofish", "serpent", "sha384", "sha512", "md4", "aes", "cast6",
  74         "arc4", "michael_mic", "deflate", "crc32c", "tea", "xtea",
  75         "khazad", "wp512", "wp384", "wp256", "tnepres", "xeta", NULL
  76 };
  77
  78 static void hexdump(unsigned char *buf, unsigned int len)
  79 {
  80         while (len--)
  81                 printk("%02x", *buf++);
  82
  83         printk("\n");
  84 }
  85
  86 static void test_hash(char *algo, struct hash_testvec *template,
  87                       unsigned int tcount)
  88 {
  89         unsigned int i, j, k, temp;
  90         struct scatterlist sg[8];
  91         char result[64];
  92         struct crypto_tfm *tfm;
  93         struct hash_testvec *hash_tv;
  94         unsigned int tsize;
  95
  96         printk("\ntesting %s\n", algo);
  97
  98         tsize = sizeof(struct hash_testvec);
  99         tsize *= tcount;
 100
 101         if (tsize > TVMEMSIZE) {
 102                 printk("template (%u) too big for tvmem (%u)\n", tsize, TVMEMSIZE);
 103                 return;
 104         }
 105
 106         memcpy(tvmem, template, tsize);
 107         hash_tv = (void *)tvmem;
 108         tfm = crypto_alloc_tfm(algo, 0);
 109         if (tfm == NULL) {
 110                 printk("failed to load transform for %s\n", algo);
 111                 return;
 112         }
 113
 114         for (i = 0; i < tcount; i++) {
 115                 printk("test %u:\n", i + 1);
 116                 memset(result, 0, 64);
 117
 118                 sg_set_buf(&sg[0], hash_tv[i].plaintext, hash_tv[i].psize);
 119
 120                 crypto_digest_init(tfm);
 121                 if (tfm->crt_u.digest.dit_setkey) {
 122                         crypto_digest_setkey(tfm, hash_tv[i].key,
 123                                              hash_tv[i].ksize);
 124                 }
 125                 crypto_digest_update(tfm, sg, 1);
 126                 crypto_digest_final(tfm, result);
 127
 128                 hexdump(result, crypto_tfm_alg_digestsize(tfm));
 129                 printk("%s\n",
 130                        memcmp(result, hash_tv[i].digest,
 131                               crypto_tfm_alg_digestsize(tfm)) ?
 132                        "fail" : "pass");
 133         }
 134
 135         printk("testing %s across pages\n", algo);
 136
 137         /* setup the dummy buffer first */
 138         memset(xbuf, 0, XBUFSIZE);
 139
 140         j = 0;
 141         for (i = 0; i < tcount; i++) {
 142                 if (hash_tv[i].np) {
 143                         j++;
 144                         printk("test %u:\n", j);
 145                         memset(result, 0, 64);
 146
 147                         temp = 0;
 148                         for (k = 0; k < hash_tv[i].np; k++) {
 149                                 memcpy(&xbuf[IDX[k]],
 150                                        hash_tv[i].plaintext + temp,
 151                                        hash_tv[i].tap[k]);
 152                                 temp += hash_tv[i].tap[k];
 153                                 sg_set_buf(&sg[k], &xbuf[IDX[k]],
 154                                             hash_tv[i].tap[k]);
 155                         }
 156
 157                         crypto_digest_digest(tfm, sg, hash_tv[i].np, result);
 158
 159                         hexdump(result, crypto_tfm_alg_digestsize(tfm));
 160                         printk("%s\n",
 161                                memcmp(result, hash_tv[i].digest,
 162                                       crypto_tfm_alg_digestsize(tfm)) ?
 163                                "fail" : "pass");
 164                 }
 165         }
 166
 167         crypto_free_tfm(tfm);
 168 }
 169
 170
 171 #ifdef CONFIG_CRYPTO_HMAC
 172
 173 static void test_hmac(char *algo, struct hmac_testvec *template,
 174                       unsigned int tcount)
 175 {
 176         unsigned int i, j, k, temp;
 177         struct scatterlist sg[8];
 178         char result[64];
 179         struct crypto_tfm *tfm;
 180         struct hmac_testvec *hmac_tv;
 181         unsigned int tsize, klen;
 182
 183         tfm = crypto_alloc_tfm(algo, 0);
 184         if (tfm == NULL) {
 185                 printk("failed to load transform for %s\n", algo);
 186                 return;
 187         }
 188
 189         printk("\ntesting hmac_%s\n", algo);
 190
 191         tsize = sizeof(struct hmac_testvec);
 192         tsize *= tcount;
 193         if (tsize > TVMEMSIZE) {
 194                 printk("template (%u) too big for tvmem (%u)\n", tsize,
 195                        TVMEMSIZE);
 196                 goto out;
 197         }
 198
 199         memcpy(tvmem, template, tsize);
 200         hmac_tv = (void *)tvmem;
 201
 202         for (i = 0; i < tcount; i++) {
 203                 printk("test %u:\n", i + 1);
 204                 memset(result, 0, sizeof (result));
 205
 206                 klen = hmac_tv[i].ksize;
 207                 sg_set_buf(&sg[0], hmac_tv[i].plaintext, hmac_tv[i].psize);
 208
 209                 crypto_hmac(tfm, hmac_tv[i].key, &klen, sg, 1, result);
 210
 211                 hexdump(result, crypto_tfm_alg_digestsize(tfm));
 212                 printk("%s\n",
 213                        memcmp(result, hmac_tv[i].digest,
 214                               crypto_tfm_alg_digestsize(tfm)) ? "fail" :
 215                        "pass");
 216         }
 217
 218         printk("\ntesting hmac_%s across pages\n", algo);
 219
 220         memset(xbuf, 0, XBUFSIZE);
 221
 222         j = 0;
 223         for (i = 0; i < tcount; i++) {
 224                 if (hmac_tv[i].np) {
 225                         j++;
 226                         printk("test %u:\n",j);
 227                         memset(result, 0, 64);
 228
 229                         temp = 0;
 230                         klen = hmac_tv[i].ksize;
 231                         for (k = 0; k < hmac_tv[i].np; k++) {
 232                                 memcpy(&xbuf[IDX[k]],
 233                                        hmac_tv[i].plaintext + temp,
 234                                        hmac_tv[i].tap[k]);
 235                                 temp += hmac_tv[i].tap[k];
 236                                 sg_set_buf(&sg[k], &xbuf[IDX[k]],
 237                                             hmac_tv[i].tap[k]);
 238                         }
 239
 240                         crypto_hmac(tfm, hmac_tv[i].key, &klen, sg,
 241                                     hmac_tv[i].np, result);
 242                         hexdump(result, crypto_tfm_alg_digestsize(tfm));
 243
 244                         printk("%s\n",
 245                                memcmp(result, hmac_tv[i].digest,
 246                                       crypto_tfm_alg_digestsize(tfm)) ?
 247                                "fail" : "pass");
 248                 }
 249         }
 250 out:
 251         crypto_free_tfm(tfm);
 252 }
 253
 254 #endif  /* CONFIG_CRYPTO_HMAC */
 255
 256 static void test_cipher(char *algo, int mode, int enc,
 257                         struct cipher_testvec *template, unsigned int tcount)
 258 {
 259         unsigned int ret, i, j, k, temp;
 260         unsigned int tsize;
 261         char *q;
 262         struct crypto_tfm *tfm;
 263         char *key;
 264         struct cipher_testvec *cipher_tv;
 265         struct scatterlist sg[8];
 266         const char *e, *m;
 267
 268         if (enc == ENCRYPT)
 269                 e = "encryption";
 270         else
 271                 e = "decryption";
 272         if (mode == MODE_ECB)
 273                 m = "ECB";
 274         else
 275                 m = "CBC";
 276
 277         printk("\ntesting %s %s %s\n", algo, m, e);
 278
 279         tsize = sizeof (struct cipher_testvec);
 280         tsize *= tcount;
 281
 282         if (tsize > TVMEMSIZE) {
 283                 printk("template (%u) too big for tvmem (%u)\n", tsize,
 284                        TVMEMSIZE);
 285                 return;
 286         }
 287
 288         memcpy(tvmem, template, tsize);
 289         cipher_tv = (void *)tvmem;
 290
 291         if (mode)
 292                 tfm = crypto_alloc_tfm(algo, 0);
 293         else
 294                 tfm = crypto_alloc_tfm(algo, CRYPTO_TFM_MODE_CBC);
 295
 296         if (tfm == NULL) {
 297                 printk("failed to load transform for %s %s\n", algo, m);
 298                 return;
 299         }
 300
 301         j = 0;
 302         for (i = 0; i < tcount; i++) {
 303                 if (!(cipher_tv[i].np)) {
 304                         j++;
 305                         printk("test %u (%d bit key):\n",
 306                         j, cipher_tv[i].klen * 8);
 307
 308                         tfm->crt_flags = 0;
 309                         if (cipher_tv[i].wk)
 310                                 tfm->crt_flags |= CRYPTO_TFM_REQ_WEAK_KEY;
 311                         key = cipher_tv[i].key;
 312
 313                         ret = crypto_cipher_setkey(tfm, key, cipher_tv[i].klen);
 314                         if (ret) {
 315                                 printk("setkey() failed flags=%x\n", tfm->crt_flags);
 316
 317                                 if (!cipher_tv[i].fail)
 318                                         goto out;
 319                         }
 320
 321                         sg_set_buf(&sg[0], cipher_tv[i].input,
 322                                    cipher_tv[i].ilen);
 323
 324                         if (!mode) {
 325                                 crypto_cipher_set_iv(tfm, cipher_tv[i].iv,
 326                                         crypto_tfm_alg_ivsize(tfm));
 327                         }
 328
 329                         if (enc)
 330                                 ret = crypto_cipher_encrypt(tfm, sg, sg, cipher_tv[i].ilen);
 331                         else
 332                                 ret = crypto_cipher_decrypt(tfm, sg, sg, cipher_tv[i].ilen);
 333
 334
 335                         if (ret) {
 336                                 printk("%s () failed flags=%x\n", e, tfm->crt_flags);
 337                                 goto out;
 338                         }
 339
 340                         q = kmap(sg[0].page) + sg[0].offset;
 341                         hexdump(q, cipher_tv[i].rlen);
 342
 343                         printk("%s\n",
 344                                memcmp(q, cipher_tv[i].result,
 345                                       cipher_tv[i].rlen) ? "fail" : "pass");
 346                 }
 347         }
 348
 349         printk("\ntesting %s %s %s across pages (chunking)\n", algo, m, e);
 350         memset(xbuf, 0, XBUFSIZE);
 351
 352         j = 0;
 353         for (i = 0; i < tcount; i++) {
 354                 if (cipher_tv[i].np) {
 355                         j++;
 356                         printk("test %u (%d bit key):\n",
 357                         j, cipher_tv[i].klen * 8);
 358
 359                         tfm->crt_flags = 0;
 360                         if (cipher_tv[i].wk)
 361                                 tfm->crt_flags |= CRYPTO_TFM_REQ_WEAK_KEY;
 362                         key = cipher_tv[i].key;
 363
 364                         ret = crypto_cipher_setkey(tfm, key, cipher_tv[i].klen);
 365                         if (ret) {
 366                                 printk("setkey() failed flags=%x\n", tfm->crt_flags);
 367
 368                                 if (!cipher_tv[i].fail)
 369                                         goto out;
 370                         }
 371
 372                         temp = 0;
 373                         for (k = 0; k < cipher_tv[i].np; k++) {
 374                                 memcpy(&xbuf[IDX[k]],
 375                                        cipher_tv[i].input + temp,
 376                                        cipher_tv[i].tap[k]);
 377                                 temp += cipher_tv[i].tap[k];
 378                                 sg_set_buf(&sg[k], &xbuf[IDX[k]],
 379                                            cipher_tv[i].tap[k]);
 380                         }
 381
 382                         if (!mode) {
 383                                 crypto_cipher_set_iv(tfm, cipher_tv[i].iv,
 384                                                 crypto_tfm_alg_ivsize(tfm));
 385                         }
 386
 387                         if (enc)
 388                                 ret = crypto_cipher_encrypt(tfm, sg, sg, cipher_tv[i].ilen);
 389                         else
 390                                 ret = crypto_cipher_decrypt(tfm, sg, sg, cipher_tv[i].ilen);
 391
 392                         if (ret) {
 393                                 printk("%s () failed flags=%x\n", e, tfm->crt_flags);
 394                                 goto out;
 395                         }
 396
 397                         temp = 0;
 398                         for (k = 0; k < cipher_tv[i].np; k++) {
 399                                 printk("page %u\n", k);
 400                                 q = kmap(sg[k].page) + sg[k].offset;
 401                                 hexdump(q, cipher_tv[i].tap[k]);
 402                                 printk("%s\n",
 403                                         memcmp(q, cipher_tv[i].result + temp,
 404                                                 cipher_tv[i].tap[k]) ? "fail" :
 405                                         "pass");
 406                                 temp += cipher_tv[i].tap[k];
 407                         }
 408                 }
 409         }
 410
 411 out:
 412         crypto_free_tfm(tfm);
 413 }
 414
 415 static int test_cipher_jiffies(struct crypto_tfm *tfm, int enc, char *p,
 416                                int blen, int sec)
 417 {
 418         struct scatterlist sg[1];
 419         unsigned long start, end;
 420         int bcount;
 421         int ret;
 422
 423         sg_set_buf(sg, p, blen);
 424
 425         for (start = jiffies, end = start + sec * HZ, bcount = 0;
 426              time_before(jiffies, end); bcount++) {
 427                 if (enc)
 428                         ret = crypto_cipher_encrypt(tfm, sg, sg, blen);
 429                 else
 430                         ret = crypto_cipher_decrypt(tfm, sg, sg, blen);
 431
 432                 if (ret)
 433                         return ret;
 434         }
 435
 436         printk("%d operations in %d seconds (%ld bytes)\n",
 437                bcount, sec, (long)bcount * blen);
 438         return 0;
 439 }
 440
 441 static int test_cipher_cycles(struct crypto_tfm *tfm, int enc, char *p,
 442                               int blen)
 443 {
 444         struct scatterlist sg[1];
 445         unsigned long cycles = 0;
 446         int ret = 0;
 447         int i;
 448
 449         sg_set_buf(sg, p, blen);
 450
 451         local_bh_disable();
 452         local_irq_disable();
 453
 454         /* Warm-up run. */
 455         for (i = 0; i < 4; i++) {
 456                 if (enc)
 457                         ret = crypto_cipher_encrypt(tfm, sg, sg, blen);
 458                 else
 459                         ret = crypto_cipher_decrypt(tfm, sg, sg, blen);
 460
 461                 if (ret)
 462                         goto out;
 463         }
 464
 465         /* The real thing. */
 466         for (i = 0; i < 8; i++) {
 467                 cycles_t start, end;
 468
 469                 start = get_cycles();
 470                 if (enc)
 471                         ret = crypto_cipher_encrypt(tfm, sg, sg, blen);
 472                 else
 473                         ret = crypto_cipher_decrypt(tfm, sg, sg, blen);
 474                 end = get_cycles();
 475
 476                 if (ret)
 477                         goto out;
 478
 479                 cycles += end - start;
 480         }
 481
 482 out:
 483         local_irq_enable();
 484         local_bh_enable();
 485
 486         if (ret == 0)
 487                 printk("1 operation in %lu cycles (%d bytes)\n",
 488                        (cycles + 4) / 8, blen);
 489
 490         return ret;
 491 }
 492
 493 static void test_cipher_speed(char *algo, int mode, int enc, unsigned int sec,
 494                               struct cipher_testvec *template,
 495                               unsigned int tcount, struct cipher_speed *speed)
 496 {
 497         unsigned int ret, i, j, iv_len;
 498         unsigned char *key, *p, iv[128];
 499         struct crypto_tfm *tfm;
 500         const char *e, *m;
 501
 502         if (enc == ENCRYPT)
 503                 e = "encryption";
 504         else
 505                 e = "decryption";
 506         if (mode == MODE_ECB)
 507                 m = "ECB";
 508         else
 509                 m = "CBC";
 510
 511         printk("\ntesting speed of %s %s %s\n", algo, m, e);
 512
 513         if (mode)
 514                 tfm = crypto_alloc_tfm(algo, 0);
 515         else
 516                 tfm = crypto_alloc_tfm(algo, CRYPTO_TFM_MODE_CBC);
 517
 518         if (tfm == NULL) {
 519                 printk("failed to load transform for %s %s\n", algo, m);
 520                 return;
 521         }
 522
 523         for (i = 0; speed[i].klen != 0; i++) {
 524                 if ((speed[i].blen + speed[i].klen) > TVMEMSIZE) {
 525                         printk("template (%u) too big for tvmem (%u)\n",
 526                                speed[i].blen + speed[i].klen, TVMEMSIZE);
 527                         goto out;
 528                 }
 529
 530                 printk("test %u (%d bit key, %d byte blocks): ", i,
 531                        speed[i].klen * 8, speed[i].blen);
 532
 533                 memset(tvmem, 0xff, speed[i].klen + speed[i].blen);
 534
 535                 /* set key, plain text and IV */
 536                 key = (unsigned char *)tvmem;
 537                 for (j = 0; j < tcount; j++) {
 538                         if (template[j].klen == speed[i].klen) {
 539                                 key = template[j].key;
 540                                 break;
 541                         }
 542                 }
 543                 p = (unsigned char *)tvmem + speed[i].klen;
 544
 545                 ret = crypto_cipher_setkey(tfm, key, speed[i].klen);
 546                 if (ret) {
 547                         printk("setkey() failed flags=%x\n", tfm->crt_flags);
 548                         goto out;
 549                 }
 550
 551                 if (!mode) {
 552                         iv_len = crypto_tfm_alg_ivsize(tfm);
 553                         memset(&iv, 0xff, iv_len);
 554                         crypto_cipher_set_iv(tfm, iv, iv_len);
 555                 }
 556
 557                 if (sec)
 558                         ret = test_cipher_jiffies(tfm, enc, p, speed[i].blen,
 559                                                   sec);
 560                 else
 561                         ret = test_cipher_cycles(tfm, enc, p, speed[i].blen);
 562
 563                 if (ret) {
 564                         printk("%s() failed flags=%x\n", e, tfm->crt_flags);
 565                         break;
 566                 }
 567         }
 568
 569 out:
 570         crypto_free_tfm(tfm);
 571 }
 572
 573 static void test_digest_jiffies(struct crypto_tfm *tfm, char *p, int blen,
 574                                 int plen, char *out, int sec)
 575 {
 576         struct scatterlist sg[1];
 577         unsigned long start, end;
 578         int bcount, pcount;
 579
 580         for (start = jiffies, end = start + sec * HZ, bcount = 0;
 581              time_before(jiffies, end); bcount++) {
 582                 crypto_digest_init(tfm);
 583                 for (pcount = 0; pcount < blen; pcount += plen) {
 584                         sg_set_buf(sg, p + pcount, plen);
 585                         crypto_digest_update(tfm, sg, 1);
 586                 }
 587                 /* we assume there is enough space in 'out' for the result */
 588                 crypto_digest_final(tfm, out);
 589         }
 590
 591         printk("%6u opers/sec, %9lu bytes/sec\n",
 592                bcount / sec, ((long)bcount * blen) / sec);
 593
 594         return;
 595 }
 596
 597 static void test_digest_cycles(struct crypto_tfm *tfm, char *p, int blen,
 598                                int plen, char *out)
 599 {
 600         struct scatterlist sg[1];
 601         unsigned long cycles = 0;
 602         int i, pcount;
 603
 604         local_bh_disable();
 605         local_irq_disable();
 606
 607         /* Warm-up run. */
 608         for (i = 0; i < 4; i++) {
 609                 crypto_digest_init(tfm);
 610                 for (pcount = 0; pcount < blen; pcount += plen) {
 611                         sg_set_buf(sg, p + pcount, plen);
 612                         crypto_digest_update(tfm, sg, 1);
 613                 }
 614                 crypto_digest_final(tfm, out);
 615         }
 616
 617         /* The real thing. */
 618         for (i = 0; i < 8; i++) {
 619                 cycles_t start, end;
 620
 621                 crypto_digest_init(tfm);
 622
 623                 start = get_cycles();
 624
 625                 for (pcount = 0; pcount < blen; pcount += plen) {
 626                         sg_set_buf(sg, p + pcount, plen);
 627                         crypto_digest_update(tfm, sg, 1);
 628                 }
 629                 crypto_digest_final(tfm, out);
 630
 631                 end = get_cycles();
 632
 633                 cycles += end - start;
 634         }
 635
 636         local_irq_enable();
 637         local_bh_enable();
 638
 639         printk("%6lu cycles/operation, %4lu cycles/byte\n",
 640                cycles / 8, cycles / (8 * blen));
 641
 642         return;
 643 }
 644
 645 static void test_digest_speed(char *algo, unsigned int sec,
 646                               struct digest_speed *speed)
 647 {
 648         struct crypto_tfm *tfm;
 649         char output[1024];
 650         int i;
 651
 652         printk("\ntesting speed of %s\n", algo);
 653
 654         tfm = crypto_alloc_tfm(algo, 0);
 655
 656         if (tfm == NULL) {
 657                 printk("failed to load transform for %s\n", algo);
 658                 return;
 659         }
 660
 661         if (crypto_tfm_alg_digestsize(tfm) > sizeof(output)) {
 662                 printk("digestsize(%u) > outputbuffer(%zu)\n",
 663                        crypto_tfm_alg_digestsize(tfm), sizeof(output));
 664                 goto out;
 665         }
 666
 667         for (i = 0; speed[i].blen != 0; i++) {
 668                 if (speed[i].blen > TVMEMSIZE) {
 669                         printk("template (%u) too big for tvmem (%u)\n",
 670                                speed[i].blen, TVMEMSIZE);
 671                         goto out;
 672                 }
 673
 674                 printk("test%3u (%5u byte blocks,%5u bytes per update,%4u updates): ",
 675                        i, speed[i].blen, speed[i].plen, speed[i].blen / speed[i].plen);
 676
 677                 memset(tvmem, 0xff, speed[i].blen);
 678
 679                 if (sec)
 680                         test_digest_jiffies(tfm, tvmem, speed[i].blen, speed[i].plen, output, sec);
 681                 else
 682                         test_digest_cycles(tfm, tvmem, speed[i].blen, speed[i].plen, output);
 683         }
 684
 685 out:
 686         crypto_free_tfm(tfm);
 687 }
 688
 689 static void test_deflate(void)
 690 {
 691         unsigned int i;
 692         char result[COMP_BUF_SIZE];
 693         struct crypto_tfm *tfm;
 694         struct comp_testvec *tv;
 695         unsigned int tsize;
 696
 697         printk("\ntesting deflate compression\n");
 698
 699         tsize = sizeof (deflate_comp_tv_template);
 700         if (tsize > TVMEMSIZE) {
 701                 printk("template (%u) too big for tvmem (%u)\n", tsize,
 702                        TVMEMSIZE);
 703                 return;
 704         }
 705
 706         memcpy(tvmem, deflate_comp_tv_template, tsize);
 707         tv = (void *)tvmem;
 708
 709         tfm = crypto_alloc_tfm("deflate", 0);
 710         if (tfm == NULL) {
 711                 printk("failed to load transform for deflate\n");
 712                 return;
 713         }
 714
 715         for (i = 0; i < DEFLATE_COMP_TEST_VECTORS; i++) {
 716                 int ilen, ret, dlen = COMP_BUF_SIZE;
 717
 718                 printk("test %u:\n", i + 1);
 719                 memset(result, 0, sizeof (result));
 720
 721                 ilen = tv[i].inlen;
 722                 ret = crypto_comp_compress(tfm, tv[i].input,
 723                                            ilen, result, &dlen);
 724                 if (ret) {
 725                         printk("fail: ret=%d\n", ret);
 726                         continue;
 727                 }
 728                 hexdump(result, dlen);
 729                 printk("%s (ratio %d:%d)\n",
 730                        memcmp(result, tv[i].output, dlen) ? "fail" : "pass",
 731                        ilen, dlen);
 732         }
 733
 734         printk("\ntesting deflate decompression\n");
 735
 736         tsize = sizeof (deflate_decomp_tv_template);
 737         if (tsize > TVMEMSIZE) {
 738                 printk("template (%u) too big for tvmem (%u)\n", tsize,
 739                        TVMEMSIZE);
 740                 goto out;
 741         }
 742
 743         memcpy(tvmem, deflate_decomp_tv_template, tsize);
 744         tv = (void *)tvmem;
 745
 746         for (i = 0; i < DEFLATE_DECOMP_TEST_VECTORS; i++) {
 747                 int ilen, ret, dlen = COMP_BUF_SIZE;
 748
 749                 printk("test %u:\n", i + 1);
 750                 memset(result, 0, sizeof (result));
 751
 752                 ilen = tv[i].inlen;
 753                 ret = crypto_comp_decompress(tfm, tv[i].input,
 754                                              ilen, result, &dlen);
 755                 if (ret) {
 756                         printk("fail: ret=%d\n", ret);
 757                         continue;
 758                 }
 759                 hexdump(result, dlen);
 760                 printk("%s (ratio %d:%d)\n",
 761                        memcmp(result, tv[i].output, dlen) ? "fail" : "pass",
 762                        ilen, dlen);
 763         }
 764 out:
 765         crypto_free_tfm(tfm);
 766 }
 767
 768 static void test_crc32c(void)
 769 {
 770 #define NUMVEC 6
 771 #define VECSIZE 40
 772
 773         int i, j, pass;
 774         u32 crc;
 775         u8 b, test_vec[NUMVEC][VECSIZE];
 776         static u32 vec_results[NUMVEC] = {
 777                 0x0e2c157f, 0xe980ebf6, 0xde74bded,
 778                 0xd579c862, 0xba979ad0, 0x2b29d913
 779         };
 780         static u32 tot_vec_results = 0x24c5d375;
 781
 782         struct scatterlist sg[NUMVEC];
 783         struct crypto_tfm *tfm;
 784         char *fmtdata = "testing crc32c initialized to %08x: %s\n";
 785 #define SEEDTESTVAL 0xedcba987
 786         u32 seed;
 787
 788         printk("\ntesting crc32c\n");
 789
 790         tfm = crypto_alloc_tfm("crc32c", 0);
 791         if (tfm == NULL) {
 792                 printk("failed to load transform for crc32c\n");
 793                 return;
 794         }
 795
 796         crypto_digest_init(tfm);
 797         crypto_digest_final(tfm, (u8*)&crc);
 798         printk(fmtdata, crc, (crc == 0) ? "pass" : "ERROR");
 799
 800         /*
 801          * stuff test_vec with known values, simple incrementing
 802          * byte values.
 803          */
 804         b = 0;
 805         for (i = 0; i < NUMVEC; i++) {
 806                 for (j = 0; j < VECSIZE; j++)
 807                         test_vec[i][j] = ++b;
 808                 sg_set_buf(&sg[i], test_vec[i], VECSIZE);
 809         }
 810
 811         seed = SEEDTESTVAL;
 812         (void)crypto_digest_setkey(tfm, (const u8*)&seed, sizeof(u32));
 813         crypto_digest_final(tfm, (u8*)&crc);
 814         printk("testing crc32c setkey returns %08x : %s\n", crc, (crc == (SEEDTESTVAL ^ ~(u32)0)) ?
 815                "pass" : "ERROR");
 816
 817         printk("testing crc32c using update/final:\n");
 818
 819         pass = 1;                   /* assume all is well */
 820
 821         for (i = 0; i < NUMVEC; i++) {
 822                 seed = ~(u32)0;
 823                 (void)crypto_digest_setkey(tfm, (const u8*)&seed, sizeof(u32));
 824                 crypto_digest_update(tfm, &sg[i], 1);
 825                 crypto_digest_final(tfm, (u8*)&crc);
 826                 if (crc == vec_results[i]) {
 827                         printk(" %08x:OK", crc);
 828                 } else {
 829                         printk(" %08x:BAD, wanted %08x\n", crc, vec_results[i]);
 830                         pass = 0;
 831                 }
 832         }
 833
 834         printk("\ntesting crc32c using incremental accumulator:\n");
 835         crc = 0;
 836         for (i = 0; i < NUMVEC; i++) {
 837                 seed = (crc ^ ~(u32)0);
 838                 (void)crypto_digest_setkey(tfm, (const u8*)&seed, sizeof(u32));
 839                 crypto_digest_update(tfm, &sg[i], 1);
 840                 crypto_digest_final(tfm, (u8*)&crc);
 841         }
 842         if (crc == tot_vec_results) {
 843                 printk(" %08x:OK", crc);
 844         } else {
 845                 printk(" %08x:BAD, wanted %08x\n", crc, tot_vec_results);
 846                 pass = 0;
 847         }
 848
 849         printk("\ntesting crc32c using digest:\n");
 850         seed = ~(u32)0;
 851         (void)crypto_digest_setkey(tfm, (const u8*)&seed, sizeof(u32));
 852         crypto_digest_digest(tfm, sg, NUMVEC, (u8*)&crc);
 853         if (crc == tot_vec_results) {
 854                 printk(" %08x:OK", crc);
 855         } else {
 856                 printk(" %08x:BAD, wanted %08x\n", crc, tot_vec_results);
 857                 pass = 0;
 858         }
 859
 860         printk("\n%s\n", pass ? "pass" : "ERROR");
 861
 862         crypto_free_tfm(tfm);
 863         printk("crc32c test complete\n");
 864 }
 865
 866 static void test_available(void)
 867 {
 868         char **name = check;
 869
 870         while (*name) {
 871                 printk("alg %s ", *name);
 872                 printk((crypto_alg_available(*name, 0)) ?
 873                         "found\n" : "not found\n");
 874                 name++;
 875         }
 876 }
 877
 878 static void do_test(void)
 879 {
 880         switch (mode) {
 881
 882         case 0:
 883                 test_hash("md5", md5_tv_template, MD5_TEST_VECTORS);
 884
 885                 test_hash("sha1", sha1_tv_template, SHA1_TEST_VECTORS);
 886
 887                 //DES
 888                 test_cipher ("des", MODE_ECB, ENCRYPT, des_enc_tv_template, DES_ENC_TEST_VECTORS);
 889                 test_cipher ("des", MODE_ECB, DECRYPT, des_dec_tv_template, DES_DEC_TEST_VECTORS);
 890                 test_cipher ("des", MODE_CBC, ENCRYPT, des_cbc_enc_tv_template, DES_CBC_ENC_TEST_VECTORS);
 891                 test_cipher ("des", MODE_CBC, DECRYPT, des_cbc_dec_tv_template, DES_CBC_DEC_TEST_VECTORS);
 892
 893                 //DES3_EDE
 894                 test_cipher ("des3_ede", MODE_ECB, ENCRYPT, des3_ede_enc_tv_template, DES3_EDE_ENC_TEST_VECTORS);
 895                 test_cipher ("des3_ede", MODE_ECB, DECRYPT, des3_ede_dec_tv_template, DES3_EDE_DEC_TEST_VECTORS);
 896
 897                 test_hash("md4", md4_tv_template, MD4_TEST_VECTORS);
 898
 899                 test_hash("sha256", sha256_tv_template, SHA256_TEST_VECTORS);
 900
 901                 //BLOWFISH
 902                 test_cipher ("blowfish", MODE_ECB, ENCRYPT, bf_enc_tv_template, BF_ENC_TEST_VECTORS);
 903                 test_cipher ("blowfish", MODE_ECB, DECRYPT, bf_dec_tv_template, BF_DEC_TEST_VECTORS);
 904                 test_cipher ("blowfish", MODE_CBC, ENCRYPT, bf_cbc_enc_tv_template, BF_CBC_ENC_TEST_VECTORS);
 905                 test_cipher ("blowfish", MODE_CBC, DECRYPT, bf_cbc_dec_tv_template, BF_CBC_DEC_TEST_VECTORS);
 906
 907                 //TWOFISH
 908                 test_cipher ("twofish", MODE_ECB, ENCRYPT, tf_enc_tv_template, TF_ENC_TEST_VECTORS);
 909                 test_cipher ("twofish", MODE_ECB, DECRYPT, tf_dec_tv_template, TF_DEC_TEST_VECTORS);
 910                 test_cipher ("twofish", MODE_CBC, ENCRYPT, tf_cbc_enc_tv_template, TF_CBC_ENC_TEST_VECTORS);
 911                 test_cipher ("twofish", MODE_CBC, DECRYPT, tf_cbc_dec_tv_template, TF_CBC_DEC_TEST_VECTORS);
 912
 913                 //SERPENT
 914                 test_cipher ("serpent", MODE_ECB, ENCRYPT, serpent_enc_tv_template, SERPENT_ENC_TEST_VECTORS);
 915                 test_cipher ("serpent", MODE_ECB, DECRYPT, serpent_dec_tv_template, SERPENT_DEC_TEST_VECTORS);
 916
 917                 //TNEPRES
 918                 test_cipher ("tnepres", MODE_ECB, ENCRYPT, tnepres_enc_tv_template, TNEPRES_ENC_TEST_VECTORS);
 919                 test_cipher ("tnepres", MODE_ECB, DECRYPT, tnepres_dec_tv_template, TNEPRES_DEC_TEST_VECTORS);
 920
 921                 //AES
 922                 test_cipher ("aes", MODE_ECB, ENCRYPT, aes_enc_tv_template, AES_ENC_TEST_VECTORS);
 923                 test_cipher ("aes", MODE_ECB, DECRYPT, aes_dec_tv_template, AES_DEC_TEST_VECTORS);
 924                 test_cipher ("aes", MODE_CBC, ENCRYPT, aes_cbc_enc_tv_template, AES_CBC_ENC_TEST_VECTORS);
 925                 test_cipher ("aes", MODE_CBC, DECRYPT, aes_cbc_dec_tv_template, AES_CBC_DEC_TEST_VECTORS);
 926
 927                 //CAST5
 928                 test_cipher ("cast5", MODE_ECB, ENCRYPT, cast5_enc_tv_template, CAST5_ENC_TEST_VECTORS);
 929                 test_cipher ("cast5", MODE_ECB, DECRYPT, cast5_dec_tv_template, CAST5_DEC_TEST_VECTORS);
 930
 931                 //CAST6
 932                 test_cipher ("cast6", MODE_ECB, ENCRYPT, cast6_enc_tv_template, CAST6_ENC_TEST_VECTORS);
 933                 test_cipher ("cast6", MODE_ECB, DECRYPT, cast6_dec_tv_template, CAST6_DEC_TEST_VECTORS);
 934
 935                 //ARC4
 936                 test_cipher ("arc4", MODE_ECB, ENCRYPT, arc4_enc_tv_template, ARC4_ENC_TEST_VECTORS);
 937                 test_cipher ("arc4", MODE_ECB, DECRYPT, arc4_dec_tv_template, ARC4_DEC_TEST_VECTORS);
 938
 939                 //TEA
 940                 test_cipher ("tea", MODE_ECB, ENCRYPT, tea_enc_tv_template, TEA_ENC_TEST_VECTORS);
 941                 test_cipher ("tea", MODE_ECB, DECRYPT, tea_dec_tv_template, TEA_DEC_TEST_VECTORS);
 942
 943
 944                 //XTEA
 945                 test_cipher ("xtea", MODE_ECB, ENCRYPT, xtea_enc_tv_template, XTEA_ENC_TEST_VECTORS);
 946                 test_cipher ("xtea", MODE_ECB, DECRYPT, xtea_dec_tv_template, XTEA_DEC_TEST_VECTORS);
 947
 948                 //KHAZAD
 949                 test_cipher ("khazad", MODE_ECB, ENCRYPT, khazad_enc_tv_template, KHAZAD_ENC_TEST_VECTORS);
 950                 test_cipher ("khazad", MODE_ECB, DECRYPT, khazad_dec_tv_template, KHAZAD_DEC_TEST_VECTORS);
 951
 952                 //ANUBIS
 953                 test_cipher ("anubis", MODE_ECB, ENCRYPT, anubis_enc_tv_template, ANUBIS_ENC_TEST_VECTORS);
 954                 test_cipher ("anubis", MODE_ECB, DECRYPT, anubis_dec_tv_template, ANUBIS_DEC_TEST_VECTORS);
 955                 test_cipher ("anubis", MODE_CBC, ENCRYPT, anubis_cbc_enc_tv_template, ANUBIS_CBC_ENC_TEST_VECTORS);
 956                 test_cipher ("anubis", MODE_CBC, DECRYPT, anubis_cbc_dec_tv_template, ANUBIS_CBC_ENC_TEST_VECTORS);
 957
 958                 //XETA
 959                 test_cipher ("xeta", MODE_ECB, ENCRYPT, xeta_enc_tv_template, XETA_ENC_TEST_VECTORS);
 960                 test_cipher ("xeta", MODE_ECB, DECRYPT, xeta_dec_tv_template, XETA_DEC_TEST_VECTORS);
 961
 962                 test_hash("sha384", sha384_tv_template, SHA384_TEST_VECTORS);
 963                 test_hash("sha512", sha512_tv_template, SHA512_TEST_VECTORS);
 964                 test_hash("wp512", wp512_tv_template, WP512_TEST_VECTORS);
 965                 test_hash("wp384", wp384_tv_template, WP384_TEST_VECTORS);
 966                 test_hash("wp256", wp256_tv_template, WP256_TEST_VECTORS);
 967                 test_hash("tgr192", tgr192_tv_template, TGR192_TEST_VECTORS);
 968                 test_hash("tgr160", tgr160_tv_template, TGR160_TEST_VECTORS);
 969                 test_hash("tgr128", tgr128_tv_template, TGR128_TEST_VECTORS);
 970                 test_deflate();
 971                 test_crc32c();
 972 #ifdef CONFIG_CRYPTO_HMAC
 973                 test_hmac("md5", hmac_md5_tv_template, HMAC_MD5_TEST_VECTORS);
 974                 test_hmac("sha1", hmac_sha1_tv_template, HMAC_SHA1_TEST_VECTORS);
 975                 test_hmac("sha256", hmac_sha256_tv_template, HMAC_SHA256_TEST_VECTORS);
 976 #endif
 977
 978                 test_hash("michael_mic", michael_mic_tv_template, MICHAEL_MIC_TEST_VECTORS);
 979                 break;
 980
 981         case 1:
 982                 test_hash("md5", md5_tv_template, MD5_TEST_VECTORS);
 983                 break;
 984
 985         case 2:
 986                 test_hash("sha1", sha1_tv_template, SHA1_TEST_VECTORS);
 987                 break;
 988
 989         case 3:
 990                 test_cipher ("des", MODE_ECB, ENCRYPT, des_enc_tv_template, DES_ENC_TEST_VECTORS);
 991                 test_cipher ("des", MODE_ECB, DECRYPT, des_dec_tv_template, DES_DEC_TEST_VECTORS);
 992                 test_cipher ("des", MODE_CBC, ENCRYPT, des_cbc_enc_tv_template, DES_CBC_ENC_TEST_VECTORS);
 993                 test_cipher ("des", MODE_CBC, DECRYPT, des_cbc_dec_tv_template, DES_CBC_DEC_TEST_VECTORS);
 994                 break;
 995
 996         case 4:
 997                 test_cipher ("des3_ede", MODE_ECB, ENCRYPT, des3_ede_enc_tv_template, DES3_EDE_ENC_TEST_VECTORS);
 998                 test_cipher ("des3_ede", MODE_ECB, DECRYPT, des3_ede_dec_tv_template, DES3_EDE_DEC_TEST_VECTORS);
 999                 break;
1000
1001         case 5:
1002                 test_hash("md4", md4_tv_template, MD4_TEST_VECTORS);
1003                 break;
1004
1005         case 6:
1006                 test_hash("sha256", sha256_tv_template, SHA256_TEST_VECTORS);
1007                 break;
1008
1009         case 7:
1010                 test_cipher ("blowfish", MODE_ECB, ENCRYPT, bf_enc_tv_template, BF_ENC_TEST_VECTORS);
1011                 test_cipher ("blowfish", MODE_ECB, DECRYPT, bf_dec_tv_template, BF_DEC_TEST_VECTORS);
1012                 test_cipher ("blowfish", MODE_CBC, ENCRYPT, bf_cbc_enc_tv_template, BF_CBC_ENC_TEST_VECTORS);
1013                 test_cipher ("blowfish", MODE_CBC, DECRYPT, bf_cbc_dec_tv_template, BF_CBC_DEC_TEST_VECTORS);
1014                 break;
1015
1016         case 8:
1017                 test_cipher ("twofish", MODE_ECB, ENCRYPT, tf_enc_tv_template, TF_ENC_TEST_VECTORS);
1018                 test_cipher ("twofish", MODE_ECB, DECRYPT, tf_dec_tv_template, TF_DEC_TEST_VECTORS);
1019                 test_cipher ("twofish", MODE_CBC, ENCRYPT, tf_cbc_enc_tv_template, TF_CBC_ENC_TEST_VECTORS);
1020                 test_cipher ("twofish", MODE_CBC, DECRYPT, tf_cbc_dec_tv_template, TF_CBC_DEC_TEST_VECTORS);
1021                 break;
1022
1023         case 9:
1024                 test_cipher ("serpent", MODE_ECB, ENCRYPT, serpent_enc_tv_template, SERPENT_ENC_TEST_VECTORS);
1025                 test_cipher ("serpent", MODE_ECB, DECRYPT, serpent_dec_tv_template, SERPENT_DEC_TEST_VECTORS);
1026                 break;
1027
1028         case 10:
1029                 test_cipher ("aes", MODE_ECB, ENCRYPT, aes_enc_tv_template, AES_ENC_TEST_VECTORS);
1030                 test_cipher ("aes", MODE_ECB, DECRYPT, aes_dec_tv_template, AES_DEC_TEST_VECTORS);
1031                 test_cipher ("aes", MODE_CBC, ENCRYPT, aes_cbc_enc_tv_template, AES_CBC_ENC_TEST_VECTORS);
1032                 test_cipher ("aes", MODE_CBC, DECRYPT, aes_cbc_dec_tv_template, AES_CBC_DEC_TEST_VECTORS);
1033                 break;
1034
1035         case 11:
1036                 test_hash("sha384", sha384_tv_template, SHA384_TEST_VECTORS);
1037                 break;
1038
1039         case 12:
1040                 test_hash("sha512", sha512_tv_template, SHA512_TEST_VECTORS);
1041                 break;
1042
1043         case 13:
1044                 test_deflate();
1045                 break;
1046
1047         case 14:
1048                 test_cipher ("cast5", MODE_ECB, ENCRYPT, cast5_enc_tv_template, CAST5_ENC_TEST_VECTORS);
1049                 test_cipher ("cast5", MODE_ECB, DECRYPT, cast5_dec_tv_template, CAST5_DEC_TEST_VECTORS);
1050                 break;
1051
1052         case 15:
1053                 test_cipher ("cast6", MODE_ECB, ENCRYPT, cast6_enc_tv_template, CAST6_ENC_TEST_VECTORS);
1054                 test_cipher ("cast6", MODE_ECB, DECRYPT, cast6_dec_tv_template, CAST6_DEC_TEST_VECTORS);
1055                 break;
1056
1057         case 16:
1058                 test_cipher ("arc4", MODE_ECB, ENCRYPT, arc4_enc_tv_template, ARC4_ENC_TEST_VECTORS);
1059                 test_cipher ("arc4", MODE_ECB, DECRYPT, arc4_dec_tv_template, ARC4_DEC_TEST_VECTORS);
1060                 break;
1061
1062         case 17:
1063                 test_hash("michael_mic", michael_mic_tv_template, MICHAEL_MIC_TEST_VECTORS);
1064                 break;
1065
1066         case 18:
1067                 test_crc32c();
1068                 break;
1069
1070         case 19:
1071                 test_cipher ("tea", MODE_ECB, ENCRYPT, tea_enc_tv_template, TEA_ENC_TEST_VECTORS);
1072                 test_cipher ("tea", MODE_ECB, DECRYPT, tea_dec_tv_template, TEA_DEC_TEST_VECTORS);
1073                 break;
1074
1075         case 20:
1076                 test_cipher ("xtea", MODE_ECB, ENCRYPT, xtea_enc_tv_template, XTEA_ENC_TEST_VECTORS);
1077                 test_cipher ("xtea", MODE_ECB, DECRYPT, xtea_dec_tv_template, XTEA_DEC_TEST_VECTORS);
1078                 break;
1079
1080         case 21:
1081                 test_cipher ("khazad", MODE_ECB, ENCRYPT, khazad_enc_tv_template, KHAZAD_ENC_TEST_VECTORS);
1082                 test_cipher ("khazad", MODE_ECB, DECRYPT, khazad_dec_tv_template, KHAZAD_DEC_TEST_VECTORS);
1083                 break;
1084
1085         case 22:
1086                 test_hash("wp512", wp512_tv_template, WP512_TEST_VECTORS);
1087                 break;
1088
1089         case 23:
1090                 test_hash("wp384", wp384_tv_template, WP384_TEST_VECTORS);
1091                 break;
1092
1093         case 24:
1094                 test_hash("wp256", wp256_tv_template, WP256_TEST_VECTORS);
1095                 break;
1096
1097         case 25:
1098                 test_cipher ("tnepres", MODE_ECB, ENCRYPT, tnepres_enc_tv_template, TNEPRES_ENC_TEST_VECTORS);
1099                 test_cipher ("tnepres", MODE_ECB, DECRYPT, tnepres_dec_tv_template, TNEPRES_DEC_TEST_VECTORS);
1100                 break;
1101
1102         case 26:
1103                 test_cipher ("anubis", MODE_ECB, ENCRYPT, anubis_enc_tv_template, ANUBIS_ENC_TEST_VECTORS);
1104                 test_cipher ("anubis", MODE_ECB, DECRYPT, anubis_dec_tv_template, ANUBIS_DEC_TEST_VECTORS);
1105                 test_cipher ("anubis", MODE_CBC, ENCRYPT, anubis_cbc_enc_tv_template, ANUBIS_CBC_ENC_TEST_VECTORS);
1106                 test_cipher ("anubis", MODE_CBC, DECRYPT, anubis_cbc_dec_tv_template, ANUBIS_CBC_ENC_TEST_VECTORS);
1107                 break;
1108
1109         case 27:
1110                 test_hash("tgr192", tgr192_tv_template, TGR192_TEST_VECTORS);
1111                 break;
1112
1113         case 28:
1114
1115                 test_hash("tgr160", tgr160_tv_template, TGR160_TEST_VECTORS);
1116                 break;
1117
1118         case 29:
1119                 test_hash("tgr128", tgr128_tv_template, TGR128_TEST_VECTORS);
1120                 break;
1121
1122         case 30:
1123                 test_cipher ("xeta", MODE_ECB, ENCRYPT, xeta_enc_tv_template, XETA_ENC_TEST_VECTORS);
1124                 test_cipher ("xeta", MODE_ECB, DECRYPT, xeta_dec_tv_template, XETA_DEC_TEST_VECTORS);
1125                 break;
1126
1127 #ifdef CONFIG_CRYPTO_HMAC
1128         case 100:
1129                 test_hmac("md5", hmac_md5_tv_template, HMAC_MD5_TEST_VECTORS);
1130                 break;
1131
1132         case 101:
1133                 test_hmac("sha1", hmac_sha1_tv_template, HMAC_SHA1_TEST_VECTORS);
1134                 break;
1135
1136         case 102:
1137                 test_hmac("sha256", hmac_sha256_tv_template, HMAC_SHA256_TEST_VECTORS);
1138                 break;
1139
1140 #endif
1141
1142         case 200:
1143                 test_cipher_speed("aes", MODE_ECB, ENCRYPT, sec, NULL, 0,
1144                                   aes_speed_template);
1145                 test_cipher_speed("aes", MODE_ECB, DECRYPT, sec, NULL, 0,
1146                                   aes_speed_template);
1147                 test_cipher_speed("aes", MODE_CBC, ENCRYPT, sec, NULL, 0,
1148                                   aes_speed_template);
1149                 test_cipher_speed("aes", MODE_CBC, DECRYPT, sec, NULL, 0,
1150                                   aes_speed_template);
1151                 break;
1152
1153         case 201:
1154                 test_cipher_speed("des3_ede", MODE_ECB, ENCRYPT, sec,
1155                                   des3_ede_enc_tv_template,
1156                                   DES3_EDE_ENC_TEST_VECTORS,
1157                                   des3_ede_speed_template);
1158                 test_cipher_speed("des3_ede", MODE_ECB, DECRYPT, sec,
1159                                   des3_ede_dec_tv_template,
1160                                   DES3_EDE_DEC_TEST_VECTORS,
1161                                   des3_ede_speed_template);
1162                 test_cipher_speed("des3_ede", MODE_CBC, ENCRYPT, sec,
1163                                   des3_ede_enc_tv_template,
1164                                   DES3_EDE_ENC_TEST_VECTORS,
1165                                   des3_ede_speed_template);
1166                 test_cipher_speed("des3_ede", MODE_CBC, DECRYPT, sec,
1167                                   des3_ede_dec_tv_template,
1168                                   DES3_EDE_DEC_TEST_VECTORS,
1169                                   des3_ede_speed_template);
1170                 break;
1171
1172         case 202:
1173                 test_cipher_speed("twofish", MODE_ECB, ENCRYPT, sec, NULL, 0,
1174                                   twofish_speed_template);
1175                 test_cipher_speed("twofish", MODE_ECB, DECRYPT, sec, NULL, 0,
1176                                   twofish_speed_template);
1177                 test_cipher_speed("twofish", MODE_CBC, ENCRYPT, sec, NULL, 0,
1178                                   twofish_speed_template);
1179                 test_cipher_speed("twofish", MODE_CBC, DECRYPT, sec, NULL, 0,
1180                                   twofish_speed_template);
1181                 break;
1182
1183         case 203:
1184                 test_cipher_speed("blowfish", MODE_ECB, ENCRYPT, sec, NULL, 0,
1185                                   blowfish_speed_template);
1186                 test_cipher_speed("blowfish", MODE_ECB, DECRYPT, sec, NULL, 0,
1187                                   blowfish_speed_template);
1188                 test_cipher_speed("blowfish", MODE_CBC, ENCRYPT, sec, NULL, 0,
1189                                   blowfish_speed_template);
1190                 test_cipher_speed("blowfish", MODE_CBC, DECRYPT, sec, NULL, 0,
1191                                   blowfish_speed_template);
1192                 break;
1193
1194         case 204:
1195                 test_cipher_speed("des", MODE_ECB, ENCRYPT, sec, NULL, 0,
1196                                   des_speed_template);
1197                 test_cipher_speed("des", MODE_ECB, DECRYPT, sec, NULL, 0,
1198                                   des_speed_template);
1199                 test_cipher_speed("des", MODE_CBC, ENCRYPT, sec, NULL, 0,
1200                                   des_speed_template);
1201                 test_cipher_speed("des", MODE_CBC, DECRYPT, sec, NULL, 0,
1202                                   des_speed_template);
1203                 break;
1204
1205         case 300:
1206                 /* fall through */
1207
1208         case 301:
1209                 test_digest_speed("md4", sec, generic_digest_speed_template);
1210                 if (mode > 300 && mode < 400) break;
1211
1212         case 302:
1213                 test_digest_speed("md5", sec, generic_digest_speed_template);
1214                 if (mode > 300 && mode < 400) break;
1215
1216         case 303:
1217                 test_digest_speed("sha1", sec, generic_digest_speed_template);
1218                 if (mode > 300 && mode < 400) break;
1219
1220         case 304:
1221                 test_digest_speed("sha256", sec, generic_digest_speed_template);
1222                 if (mode > 300 && mode < 400) break;
1223
1224         case 305:
1225                 test_digest_speed("sha384", sec, generic_digest_speed_template);
1226                 if (mode > 300 && mode < 400) break;
1227
1228         case 306:
1229                 test_digest_speed("sha512", sec, generic_digest_speed_template);
1230                 if (mode > 300 && mode < 400) break;
1231
1232         case 307:
1233                 test_digest_speed("wp256", sec, generic_digest_speed_template);
1234                 if (mode > 300 && mode < 400) break;
1235
1236         case 308:
1237                 test_digest_speed("wp384", sec, generic_digest_speed_template);
1238                 if (mode > 300 && mode < 400) break;
1239
1240         case 309:
1241                 test_digest_speed("wp512", sec, generic_digest_speed_template);
1242                 if (mode > 300 && mode < 400) break;
1243
1244         case 310:
1245                 test_digest_speed("tgr128", sec, generic_digest_speed_template);
1246                 if (mode > 300 && mode < 400) break;
1247
1248         case 311:
1249                 test_digest_speed("tgr160", sec, generic_digest_speed_template);
1250                 if (mode > 300 && mode < 400) break;
1251
1252         case 312:
1253                 test_digest_speed("tgr192", sec, generic_digest_speed_template);
1254                 if (mode > 300 && mode < 400) break;
1255
1256         case 399:
1257                 break;
1258
1259         case 1000:
1260                 test_available();
1261                 break;
1262
1263         default:
1264                 /* useful for debugging */
1265                 printk("not testing anything\n");
1266                 break;
1267         }
1268 }
1269
1270 static int __init init(void)
1271 {
1272         tvmem = kmalloc(TVMEMSIZE, GFP_KERNEL);
1273         if (tvmem == NULL)
1274                 return -ENOMEM;
1275
1276         xbuf = kmalloc(XBUFSIZE, GFP_KERNEL);
1277         if (xbuf == NULL) {
1278                 kfree(tvmem);
1279                 return -ENOMEM;
1280         }
1281
1282         do_test();
1283
1284         kfree(xbuf);
1285         kfree(tvmem);
1286
1287         /* We intentionaly return -EAGAIN to prevent keeping
1288          * the module. It does all its work from init()
1289          * and doesn't offer any runtime functionality
1290          * => we don't need it in the memory, do we?
1291          *                                        -- mludvig
1292          */
1293         return -EAGAIN;
1294 }
1295
1296 /*
1297  * If an init function is provided, an exit function must also be provided
1298  * to allow module unload.
1299  */
1300 static void __exit fini(void) { }
1301
1302 module_init(init);
1303 module_exit(fini);
1304
1305 module_param(mode, int, 0);
1306 module_param(sec, uint, 0);
1307 MODULE_PARM_DESC(sec, "Length in seconds of speed tests "
1308                       "(defaults to zero which uses CPU cycles instead)");
1309
1310 MODULE_LICENSE("GPL");
1311 MODULE_DESCRIPTION("Quick & dirty crypto testing module");
1312 MODULE_AUTHOR("James Morris <jmorris@intercode.com.au>");