Home | History | Annotate | Download | only in fipstools 
      1 /* Copyright (c) 2017, Google Inc.
      2  *
      3  * Permission to use, copy, modify, and/or distribute this software for any
      4  * purpose with or without fee is hereby granted, provided that the above
      5  * copyright notice and this permission notice appear in all copies.
      6  *
      7  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
      8  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
      9  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
     10  * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
     11  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
     12  * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
     13  * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */
     14 
     15 // cavp_hmac_test processes a NIST CAVP HMAC test vector request file and emits
     16 // the corresponding response.
     17 
     18 #include <stdlib.h>
     19 
     20 #include <openssl/crypto.h>
     21 #include <openssl/hmac.h>
     22 
     23 #include "../crypto/test/file_test.h"
     24 #include "cavp_test_util.h"
     25 
     26 
     27 static bool TestHMAC(FileTest *t, void *arg) {
     28   std::string md_len_str;
     29   if (!t->GetInstruction(&md_len_str, "L")) {
     30     return false;
     31   }
     32   const size_t md_len = strtoul(md_len_str.c_str(), nullptr, 0);
     33 
     34   const EVP_MD *md;
     35   switch (md_len) {
     36     case 20:
     37       md = EVP_sha1();
     38       break;
     39     case 28:
     40       md = EVP_sha224();
     41       break;
     42     case 32:
     43       md = EVP_sha256();
     44       break;
     45     case 48:
     46       md = EVP_sha384();
     47       break;
     48     case 64:
     49       md = EVP_sha512();
     50       break;
     51     default:
     52       return false;
     53   }
     54 
     55   std::string count_str, k_len_str, t_len_str;
     56   std::vector<uint8_t> key, msg;
     57   if (!t->GetAttribute(&count_str, "Count") ||
     58       !t->GetAttribute(&k_len_str, "Klen") ||
     59       !t->GetAttribute(&t_len_str, "Tlen") ||
     60       !t->GetBytes(&key, "Key") ||
     61       !t->GetBytes(&msg, "Msg")) {
     62     return false;
     63   }
     64 
     65   size_t k_len = strtoul(k_len_str.c_str(), nullptr, 0);
     66   size_t t_len = strtoul(t_len_str.c_str(), nullptr, 0);
     67   if (key.size() < k_len) {
     68     return false;
     69   }
     70   unsigned out_len;
     71   uint8_t out[EVP_MAX_MD_SIZE];
     72   if (HMAC(md, key.data(), k_len, msg.data(), msg.size(), out, &out_len) ==
     73       NULL) {
     74     return false;
     75   }
     76 
     77   if (out_len < t_len) {
     78     return false;
     79   }
     80 
     81   printf("%s", t->CurrentTestToString().c_str());
     82   printf("Mac = %s\r\n\r\n", EncodeHex(out, t_len).c_str());
     83 
     84   return true;
     85 }
     86 
     87 static int usage(char *arg) {
     88   fprintf(stderr, "usage: %s <test file>\n", arg);
     89   return 1;
     90 }
     91 
     92 int cavp_hmac_test_main(int argc, char **argv) {
     93   if (argc != 2) {
     94     return usage(argv[0]);
     95   }
     96 
     97   FileTest::Options opts;
     98   opts.path = argv[1];
     99   opts.callback = TestHMAC;
    100   opts.silent = true;
    101   opts.comment_callback = EchoComment;
    102   return FileTestMain(opts);
    103 }
    104