tests/test-aes.c

   1 /*
   2  * Test program for AES
   3  * Copyright (c) 2003-2012, Jouni Malinen <j@w1.fi>
   4  *
   5  * This software may be distributed under the terms of the BSD license.
   6  * See README for more details.
   7  */
   8
   9 #include "includes.h"
  10
  11 #include "common.h"
  12 #include "crypto/crypto.h"
  13 #include "crypto/aes_wrap.h"
  14
  15 #define BLOCK_SIZE 16
  16
  17 static void test_aes_perf(void)
  18 {
  19 #if 0 /* this did not seem to work with new compiler?! */
  20 #ifdef __i386__
  21 #define rdtscll(val) \
  22      __asm__ __volatile__("rdtsc" : "=A" (val))
  23         const int num_iters = 10;
  24         int i;
  25         unsigned int start, end;
  26         u8 key[16], pt[16], ct[16];
  27         void *ctx;
  28
  29         printf("keySetupEnc:");
  30         for (i = 0; i < num_iters; i++) {
  31                 rdtscll(start);
  32                 ctx = aes_encrypt_init(key, 16);
  33                 rdtscll(end);
  34                 aes_encrypt_deinit(ctx);
  35                 printf(" %d", end - start);
  36         }
  37         printf("\n");
  38
  39         printf("Encrypt:");
  40         ctx = aes_encrypt_init(key, 16);
  41         for (i = 0; i < num_iters; i++) {
  42                 rdtscll(start);
  43                 aes_encrypt(ctx, pt, ct);
  44                 rdtscll(end);
  45                 printf(" %d", end - start);
  46         }
  47         aes_encrypt_deinit(ctx);
  48         printf("\n");
  49 #endif /* __i386__ */
  50 #endif
  51 }
  52
  53
  54 /*
  55  * GCM test vectors from
  56  * http://csrc.nist.gov/groups/ST/toolkit/BCM/documents/proposedmodes/gcm/gcm-spec.pdf
  57  */
  58 struct gcm_test_vector {
  59         char *k;
  60         char *p;
  61         char *aad;
  62         char *iv;
  63         char *c;
  64         char *t;
  65 };
  66
  67 static const struct gcm_test_vector gcm_tests[] = {
  68         {
  69                 /* Test Case 1 */
  70                 "00000000000000000000000000000000",
  71                 "",
  72                 "",
  73                 "000000000000000000000000",
  74                 "",
  75                 "58e2fccefa7e3061367f1d57a4e7455a"
  76         },
  77         {
  78                 /* Test Case 2 */
  79                 "00000000000000000000000000000000",
  80                 "00000000000000000000000000000000",
  81                 "",
  82                 "000000000000000000000000",
  83                 "0388dace60b6a392f328c2b971b2fe78",
  84                 "ab6e47d42cec13bdf53a67b21257bddf"
  85         },
  86         {
  87                 /* Test Case 3 */
  88                 "feffe9928665731c6d6a8f9467308308",
  89                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b391aafd255",
  90                 "",
  91                 "cafebabefacedbaddecaf888",
  92                 "42831ec2217774244b7221b784d0d49ce3aa212f2c02a4e035c17e2329aca12e21d514b25466931c7d8f6a5aac84aa051ba30b396a0aac973d58e091473f5985",
  93                 "4d5c2af327cd64a62cf35abd2ba6fab4"
  94         },
  95         {
  96                 /* Test Case 4 */
  97                 "feffe9928665731c6d6a8f9467308308",
  98                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39",
  99                 "feedfacedeadbeeffeedfacedeadbeefabaddad2",
 100                 "cafebabefacedbaddecaf888",
 101                 "42831ec2217774244b7221b784d0d49ce3aa212f2c02a4e035c17e2329aca12e21d514b25466931c7d8f6a5aac84aa051ba30b396a0aac973d58e091",
 102                 "5bc94fbc3221a5db94fae95ae7121a47"
 103         },
 104         {
 105                 /* Test Case 5 */
 106                 "feffe9928665731c6d6a8f9467308308",
 107                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39",
 108                 "feedfacedeadbeeffeedfacedeadbeefabaddad2",
 109                 "cafebabefacedbad",
 110                 "61353b4c2806934a777ff51fa22a4755699b2a714fcdc6f83766e5f97b6c742373806900e49f24b22b097544d4896b424989b5e1ebac0f07c23f4598",
 111                 "3612d2e79e3b0785561be14aaca2fccb"
 112         },
 113         {
 114                 /* Test Case 6 */
 115                 "feffe9928665731c6d6a8f9467308308",
 116                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39",
 117                 "feedfacedeadbeeffeedfacedeadbeefabaddad2",
 118                 "9313225df88406e555909c5aff5269aa6a7a9538534f7da1e4c303d2a318a728c3c0c95156809539fcf0e2429a6b525416aedbf5a0de6a57a637b39b",
 119                 "8ce24998625615b603a033aca13fb894be9112a5c3a211a8ba262a3cca7e2ca701e4a9a4fba43c90ccdcb281d48c7c6fd62875d2aca417034c34aee5",
 120                 "619cc5aefffe0bfa462af43c1699d050"
 121         },
 122         {
 123                 /* Test Case 7 */
 124                 "000000000000000000000000000000000000000000000000",
 125                 "",
 126                 "",
 127                 "000000000000000000000000",
 128                 "",
 129                 "cd33b28ac773f74ba00ed1f312572435"
 130         },
 131         {
 132                 /* Test Case 8 */
 133                 "000000000000000000000000000000000000000000000000",
 134                 "00000000000000000000000000000000",
 135                 "",
 136                 "000000000000000000000000",
 137                 "98e7247c07f0fe411c267e4384b0f600",
 138                 "2ff58d80033927ab8ef4d4587514f0fb"
 139         },
 140         {
 141                 /* Test Case 9 */
 142                 "feffe9928665731c6d6a8f9467308308feffe9928665731c",
 143                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b391aafd255",
 144                 "",
 145                 "cafebabefacedbaddecaf888",
 146                 "3980ca0b3c00e841eb06fac4872a2757859e1ceaa6efd984628593b40ca1e19c7d773d00c144c525ac619d18c84a3f4718e2448b2fe324d9ccda2710acade256",
 147                 "9924a7c8587336bfb118024db8674a14"
 148         },
 149         {
 150                 /* Test Case 10 */
 151                 "feffe9928665731c6d6a8f9467308308feffe9928665731c",
 152                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39",
 153                 "feedfacedeadbeeffeedfacedeadbeefabaddad2",
 154                 "cafebabefacedbaddecaf888",
 155                 "3980ca0b3c00e841eb06fac4872a2757859e1ceaa6efd984628593b40ca1e19c7d773d00c144c525ac619d18c84a3f4718e2448b2fe324d9ccda2710",
 156                 "2519498e80f1478f37ba55bd6d27618c"
 157         },
 158         {
 159                 /* Test Case 11 */
 160                 "feffe9928665731c6d6a8f9467308308feffe9928665731c",
 161                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39",
 162                 "feedfacedeadbeeffeedfacedeadbeefabaddad2",
 163                 "cafebabefacedbad",
 164                 "0f10f599ae14a154ed24b36e25324db8c566632ef2bbb34f8347280fc4507057fddc29df9a471f75c66541d4d4dad1c9e93a19a58e8b473fa0f062f7",
 165                 "65dcc57fcf623a24094fcca40d3533f8"
 166         },
 167         {
 168                 /* Test Case 12 */
 169                 "feffe9928665731c6d6a8f9467308308feffe9928665731c",
 170                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39",
 171                 "feedfacedeadbeeffeedfacedeadbeefabaddad2",
 172                 "9313225df88406e555909c5aff5269aa6a7a9538534f7da1e4c303d2a318a728c3c0c95156809539fcf0e2429a6b525416aedbf5a0de6a57a637b39b",
 173                 "d27e88681ce3243c4830165a8fdcf9ff1de9a1d8e6b447ef6ef7b79828666e4581e79012af34ddd9e2f037589b292db3e67c036745fa22e7e9b7373b",
 174                 "dcf566ff291c25bbb8568fc3d376a6d9"
 175         },
 176         {
 177                 /* Test Case 13 */
 178                 "0000000000000000000000000000000000000000000000000000000000000000",
 179                 "",
 180                 "",
 181                 "000000000000000000000000",
 182                 "",
 183                 "530f8afbc74536b9a963b4f1c4cb738b"
 184         },
 185         {
 186                 /* Test Case 14 */
 187                 "0000000000000000000000000000000000000000000000000000000000000000",
 188                 "00000000000000000000000000000000",
 189                 "",
 190                 "000000000000000000000000",
 191                 "cea7403d4d606b6e074ec5d3baf39d18",
 192                 "d0d1c8a799996bf0265b98b5d48ab919"
 193         },
 194         {
 195                 /* Test Case 15 */
 196                 "feffe9928665731c6d6a8f9467308308feffe9928665731c6d6a8f9467308308",
 197                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b391aafd255",
 198                 "",
 199                 "cafebabefacedbaddecaf888",
 200                 "522dc1f099567d07f47f37a32a84427d643a8cdcbfe5c0c97598a2bd2555d1aa8cb08e48590dbb3da7b08b1056828838c5f61e6393ba7a0abcc9f662898015ad",
 201                 "b094dac5d93471bdec1a502270e3cc6c"
 202         },
 203         {
 204                 /* Test Case 16 */
 205                 "feffe9928665731c6d6a8f9467308308feffe9928665731c6d6a8f9467308308",
 206                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39",
 207                 "feedfacedeadbeeffeedfacedeadbeefabaddad2",
 208                 "cafebabefacedbaddecaf888",
 209                 "522dc1f099567d07f47f37a32a84427d643a8cdcbfe5c0c97598a2bd2555d1aa8cb08e48590dbb3da7b08b1056828838c5f61e6393ba7a0abcc9f662",
 210                 "76fc6ece0f4e1768cddf8853bb2d551b"
 211         },
 212         {
 213                 /* Test Case 17 */
 214                 "feffe9928665731c6d6a8f9467308308feffe9928665731c6d6a8f9467308308",
 215                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39",
 216                 "feedfacedeadbeeffeedfacedeadbeefabaddad2",
 217                 "cafebabefacedbad",
 218                 "c3762df1ca787d32ae47c13bf19844cbaf1ae14d0b976afac52ff7d79bba9de0feb582d33934a4f0954cc2363bc73f7862ac430e64abe499f47c9b1f",
 219                 "3a337dbf46a792c45e454913fe2ea8f2"
 220         },
 221         {
 222                 /* Test Case 18 */
 223                 "feffe9928665731c6d6a8f9467308308feffe9928665731c6d6a8f9467308308",
 224                 "d9313225f88406e5a55909c5aff5269a86a7a9531534f7da2e4c303d8a318a721c3c0c95956809532fcf0e2449a6b525b16aedf5aa0de657ba637b39",
 225                 "feedfacedeadbeeffeedfacedeadbeefabaddad2",
 226                 "9313225df88406e555909c5aff5269aa6a7a9538534f7da1e4c303d2a318a728c3c0c95156809539fcf0e2429a6b525416aedbf5a0de6a57a637b39b",
 227                 "5a8def2f0c9e53f1f75d7853659e2a20eeb2b22aafde6419a058ab4f6f746bf40fc0c3b780f244452da3ebf1c5d82cdea2418997200ef82e44ae7e3f",
 228                 "a44a8266ee1c8eb0c8b5d4cf5ae9f19a"
 229         }
 230 };
 231
 232
 233 static int test_gcm(void)
 234 {
 235         int ret = 0;
 236         int i;
 237         u8 k[32], aad[32], iv[64], t[16], tag[16];
 238         u8 p[64], c[64], tmp[64];
 239         size_t k_len, p_len, aad_len, iv_len;
 240
 241         for (i = 0; i < ARRAY_SIZE(gcm_tests); i++) {
 242                 const struct gcm_test_vector *tc = &gcm_tests[i];
 243
 244                 k_len = os_strlen(tc->k) / 2;
 245                 if (hexstr2bin(tc->k, k, k_len)) {
 246                         printf("Invalid GCM test vector %d (k)\n", i);
 247                         ret++;
 248                         continue;
 249                 }
 250
 251                 p_len = os_strlen(tc->p) / 2;
 252                 if (hexstr2bin(tc->p, p, p_len)) {
 253                         printf("Invalid GCM test vector %d (p)\n", i);
 254                         ret++;
 255                         continue;
 256                 }
 257
 258                 aad_len = os_strlen(tc->aad) / 2;
 259                 if (hexstr2bin(tc->aad, aad, aad_len)) {
 260                         printf("Invalid GCM test vector %d (aad)\n", i);
 261                         ret++;
 262                         continue;
 263                 }
 264
 265                 iv_len = os_strlen(tc->iv) / 2;
 266                 if (hexstr2bin(tc->iv, iv, iv_len)) {
 267                         printf("Invalid GCM test vector %d (iv)\n", i);
 268                         ret++;
 269                         continue;
 270                 }
 271
 272                 if (hexstr2bin(tc->c, c, p_len)) {
 273                         printf("Invalid GCM test vector %d (c)\n", i);
 274                         ret++;
 275                         continue;
 276                 }
 277
 278                 if (hexstr2bin(tc->t, t, sizeof(t))) {
 279                         printf("Invalid GCM test vector %d (t)\n", i);
 280                         ret++;
 281                         continue;
 282                 }
 283
 284                 if (aes_gcm_ae(k, k_len, iv, iv_len, p, p_len, aad, aad_len,
 285                                tmp, tag) < 0) {
 286                         printf("GCM-AE failed (test case %d)\n", i);
 287                         ret++;
 288                         continue;
 289                 }
 290
 291                 if (os_memcmp(c, tmp, p_len) != 0) {
 292                         printf("GCM-AE mismatch (test case %d)\n", i);
 293                         ret++;
 294                 }
 295
 296                 if (os_memcmp(tag, t, sizeof(tag)) != 0) {
 297                         printf("GCM-AE tag mismatch (test case %d)\n", i);
 298                         ret++;
 299                 }
 300
 301                 if (p_len == 0) {
 302                         if (aes_gmac(k, k_len, iv, iv_len, aad, aad_len, tag) <
 303                             0) {
 304                                 printf("GMAC failed (test case %d)\n", i);
 305                                 ret++;
 306                                 continue;
 307                         }
 308
 309                         if (os_memcmp(tag, t, sizeof(tag)) != 0) {
 310                                 printf("GMAC tag mismatch (test case %d)\n", i);
 311                                 ret++;
 312                         }
 313                 }
 314
 315                 if (aes_gcm_ad(k, k_len, iv, iv_len, c, p_len, aad, aad_len,
 316                                t, tmp) < 0) {
 317                         printf("GCM-AD failed (test case %d)\n", i);
 318                         ret++;
 319                         continue;
 320                 }
 321
 322                 if (os_memcmp(p, tmp, p_len) != 0) {
 323                         printf("GCM-AD mismatch (test case %d)\n", i);
 324                         ret++;
 325                 }
 326         }
 327
 328         return ret;
 329 }
 330
 331
 332 static int test_nist_key_wrap_ae(const char *fname)
 333 {
 334         FILE *f;
 335         int ret = 0;
 336         char buf[15000], *pos, *pos2;
 337         u8 bin[2000], k[32], p[1024], c[1024 + 8], result[1024 + 8];
 338         size_t bin_len, k_len = 0, p_len = 0, c_len = 0;
 339         int ok = 0;
 340
 341         printf("NIST KW AE tests from %s\n", fname);
 342
 343         f = fopen(fname, "r");
 344         if (f == NULL) {
 345                 printf("%s does not exist - cannot validate test vectors\n",
 346                        fname);
 347                 return 1;
 348         }
 349
 350         while (fgets(buf, sizeof(buf), f)) {
 351                 if (buf[0] == '#')
 352                         continue;
 353                 pos = os_strchr(buf, '=');
 354                 if (pos == NULL)
 355                         continue;
 356                 pos2 = pos - 1;
 357                 while (pos2 >= buf && *pos2 == ' ')
 358                         *pos2-- = '\0';
 359                 *pos++ = '\0';
 360                 while (*pos == ' ')
 361                         *pos++ = '\0';
 362                 pos2 = os_strchr(pos, '\r');
 363                 if (!pos2)
 364                         pos2 = os_strchr(pos, '\n');
 365                 if (pos2)
 366                         *pos2 = '\0';
 367                 else
 368                         pos2 = pos + os_strlen(pos);
 369
 370                 if (buf[0] == '[') {
 371                         printf("%s = %s\n", buf, pos);
 372                         continue;
 373                 }
 374
 375                 if (os_strcmp(buf, "COUNT") == 0) {
 376                         printf("Test %s - ", pos);
 377                         continue;
 378                 }
 379
 380                 bin_len = os_strlen(pos);
 381                 if (bin_len > sizeof(bin) * 2) {
 382                         printf("Too long binary data (%s)\n", buf);
 383                         return 1;
 384                 }
 385                 if (bin_len & 0x01) {
 386                         printf("Odd number of hexstring values (%s)\n",
 387                                 buf);
 388                         return 1;
 389                 }
 390                 bin_len /= 2;
 391                 if (hexstr2bin(pos, bin, bin_len) < 0) {
 392                         printf("Invalid hex string '%s' (%s)\n", pos, buf);
 393                         return 1;
 394                 }
 395
 396                 if (os_strcmp(buf, "K") == 0) {
 397                         if (bin_len > sizeof(k)) {
 398                                 printf("Too long K (%u)\n", (unsigned) bin_len);
 399                                 return 1;
 400                         }
 401                         os_memcpy(k, bin, bin_len);
 402                         k_len = bin_len;
 403                         continue;
 404                 }
 405
 406                 if (os_strcmp(buf, "P") == 0) {
 407                         if (bin_len > sizeof(p)) {
 408                                 printf("Too long P (%u)\n", (unsigned) bin_len);
 409                                 return 1;
 410                         }
 411                         os_memcpy(p, bin, bin_len);
 412                         p_len = bin_len;
 413                         continue;
 414                 }
 415
 416                 if (os_strcmp(buf, "C") != 0) {
 417                         printf("Unexpected field '%s'\n", buf);
 418                         continue;
 419                 }
 420
 421                 if (bin_len > sizeof(c)) {
 422                         printf("Too long C (%u)\n", (unsigned) bin_len);
 423                         return 1;
 424                 }
 425                 os_memcpy(c, bin, bin_len);
 426                 c_len = bin_len;
 427
 428                 if (p_len % 8 != 0 || c_len % 8 != 0 || c_len - p_len != 8) {
 429                         printf("invalid parameter length (p_len=%u c_len=%u)\n",
 430                                (unsigned) p_len, (unsigned) c_len);
 431                         continue;
 432                 }
 433
 434                 if (aes_wrap(k, k_len, p_len / 8, p, result)) {
 435                         printf("aes_wrap() failed\n");
 436                         ret++;
 437                         continue;
 438                 }
 439
 440                 if (os_memcmp(c, result, c_len) == 0) {
 441                         printf("OK\n");
 442                         ok++;
 443                 } else {
 444                         printf("FAIL\n");
 445                         ret++;
 446                 }
 447         }
 448
 449         fclose(f);
 450
 451         if (ret)
 452                 printf("Test case failed\n");
 453         else
 454                 printf("%d test vectors OK\n", ok);
 455
 456         return ret;
 457 }
 458
 459
 460 static int test_nist_key_wrap_ad(const char *fname)
 461 {
 462         FILE *f;
 463         int ret = 0;
 464         char buf[15000], *pos, *pos2;
 465         u8 bin[2000], k[32], p[1024], c[1024 + 8], result[1024 + 8];
 466         size_t bin_len, k_len = 0, p_len = 0, c_len = 0;
 467         int ok = 0;
 468         int fail;
 469
 470         printf("NIST KW AD tests from %s\n", fname);
 471
 472         f = fopen(fname, "r");
 473         if (f == NULL) {
 474                 printf("%s does not exist - cannot validate test vectors\n",
 475                        fname);
 476                 return 1;
 477         }
 478
 479         while (fgets(buf, sizeof(buf), f)) {
 480                 if (buf[0] == '#')
 481                         continue;
 482                 fail = 0;
 483                 pos = os_strchr(buf, '=');
 484                 if (pos == NULL) {
 485                         if (os_strncmp(buf, "FAIL", 4) == 0) {
 486                                 fail = 1;
 487                                 goto skip_val_parse;
 488                         }
 489                         continue;
 490                 }
 491                 pos2 = pos - 1;
 492                 while (pos2 >= buf && *pos2 == ' ')
 493                         *pos2-- = '\0';
 494                 *pos++ = '\0';
 495                 while (*pos == ' ')
 496                         *pos++ = '\0';
 497                 pos2 = os_strchr(pos, '\r');
 498                 if (!pos2)
 499                         pos2 = os_strchr(pos, '\n');
 500                 if (pos2)
 501                         *pos2 = '\0';
 502                 else
 503                         pos2 = pos + os_strlen(pos);
 504
 505                 if (buf[0] == '[') {
 506                         printf("%s = %s\n", buf, pos);
 507                         continue;
 508                 }
 509
 510                 if (os_strcmp(buf, "COUNT") == 0) {
 511                         printf("Test %s - ", pos);
 512                         continue;
 513                 }
 514
 515                 bin_len = os_strlen(pos);
 516                 if (bin_len > sizeof(bin) * 2) {
 517                         printf("Too long binary data (%s)\n", buf);
 518                         return 1;
 519                 }
 520                 if (bin_len & 0x01) {
 521                         printf("Odd number of hexstring values (%s)\n",
 522                                 buf);
 523                         return 1;
 524                 }
 525                 bin_len /= 2;
 526                 if (hexstr2bin(pos, bin, bin_len) < 0) {
 527                         printf("Invalid hex string '%s' (%s)\n", pos, buf);
 528                         return 1;
 529                 }
 530
 531                 if (os_strcmp(buf, "K") == 0) {
 532                         if (bin_len > sizeof(k)) {
 533                                 printf("Too long K (%u)\n", (unsigned) bin_len);
 534                                 return 1;
 535                         }
 536                         os_memcpy(k, bin, bin_len);
 537                         k_len = bin_len;
 538                         continue;
 539                 }
 540
 541                 if (os_strcmp(buf, "C") == 0) {
 542                         if (bin_len > sizeof(c)) {
 543                                 printf("Too long C (%u)\n", (unsigned) bin_len);
 544                                 return 1;
 545                         }
 546                         os_memcpy(c, bin, bin_len);
 547                         c_len = bin_len;
 548                         continue;
 549                 }
 550
 551         skip_val_parse:
 552                 if (!fail) {
 553                         if (os_strcmp(buf, "P") != 0) {
 554                                 printf("Unexpected field '%s'\n", buf);
 555                                 continue;
 556                         }
 557
 558                         if (bin_len > sizeof(p)) {
 559                                 printf("Too long P (%u)\n", (unsigned) bin_len);
 560                                 return 1;
 561                         }
 562                         os_memcpy(p, bin, bin_len);
 563                         p_len = bin_len;
 564
 565                         if (p_len % 8 != 0 || c_len % 8 != 0 ||
 566                             c_len - p_len != 8) {
 567                                 printf("invalid parameter length (p_len=%u c_len=%u)\n",
 568                                        (unsigned) p_len, (unsigned) c_len);
 569                                 continue;
 570                         }
 571                 }
 572
 573                 if (aes_unwrap(k, k_len, (c_len / 8) - 1, c, result)) {
 574                         if (fail) {
 575                                 printf("OK (fail reported)\n");
 576                                 ok++;
 577                                 continue;
 578                         }
 579                         printf("aes_unwrap() failed\n");
 580                         ret++;
 581                         continue;
 582                 }
 583
 584                 if (fail) {
 585                         printf("FAIL (mismatch not reported)\n");
 586                         ret++;
 587                 } else if (os_memcmp(p, result, p_len) == 0) {
 588                         printf("OK\n");
 589                         ok++;
 590                 } else {
 591                         printf("FAIL\n");
 592                         ret++;
 593                 }
 594         }
 595
 596         fclose(f);
 597
 598         if (ret)
 599                 printf("Test case failed\n");
 600         else
 601                 printf("%d test vectors OK\n", ok);
 602
 603         return ret;
 604 }
 605
 606
 607 int main(int argc, char *argv[])
 608 {
 609         int ret = 0;
 610
 611         if (argc >= 3 && os_strcmp(argv[1], "NIST-KW-AE") == 0)
 612                 ret += test_nist_key_wrap_ae(argv[2]);
 613         else if (argc >= 3 && os_strcmp(argv[1], "NIST-KW-AD") == 0)
 614                 ret += test_nist_key_wrap_ad(argv[2]);
 615
 616         test_aes_perf();
 617
 618         ret += test_gcm();
 619
 620         if (ret)
 621                 printf("FAILED!\n");
 622
 623         return ret;
 624 }