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_sha_monte_test processes a NIST CAVP SHA-Monte test vector request file
     16 // and emits the corresponding response.
     17 
     18 #include <stdlib.h>
     19 
     20 #include <openssl/crypto.h>
     21 #include <openssl/digest.h>
     22 
     23 #include "../crypto/test/file_test.h"
     24 #include "cavp_test_util.h"
     25 
     26 
     27 namespace {
     28 
     29 struct TestCtx {
     30   std::string hash;
     31 };
     32 
     33 }
     34 
     35 static bool TestSHAMonte(FileTest *t, void *arg) {
     36   TestCtx *ctx = reinterpret_cast<TestCtx *>(arg);
     37 
     38   const EVP_MD *md = EVP_get_digestbyname(ctx->hash.c_str());
     39   if (md == nullptr) {
     40     return false;
     41   }
     42   const size_t md_len = EVP_MD_size(md);
     43 
     44   std::string out_len;
     45   if (!t->GetInstruction(&out_len, "L") ||
     46       md_len != strtoul(out_len.c_str(), nullptr, 0)) {
     47     return false;
     48   }
     49 
     50   std::vector<uint8_t> seed;
     51   if (!t->GetBytes(&seed, "Seed") ||
     52       seed.size() != md_len) {
     53     return false;
     54   }
     55 
     56   std::vector<uint8_t> out = seed;
     57 
     58   printf("%s\r\n", t->CurrentTestToString().c_str());
     59 
     60   for (int count = 0; count < 100; count++) {
     61     std::vector<uint8_t> msg;
     62     msg.insert(msg.end(), out.begin(), out.end());
     63     msg.insert(msg.end(), out.begin(), out.end());
     64     msg.insert(msg.end(), out.begin(), out.end());
     65     for (int i = 0; i < 1000; i++) {
     66       unsigned digest_len;
     67       if (!EVP_Digest(msg.data(), msg.size(), out.data(), &digest_len, md,
     68                       nullptr) ||
     69           digest_len != out.size()) {
     70         return false;
     71       }
     72 
     73       msg.erase(msg.begin(), msg.begin() + out.size());
     74       msg.insert(msg.end(), out.begin(), out.end());
     75     }
     76     printf("COUNT = %d\r\n", count);
     77     printf("MD = %s\r\n\r\n", EncodeHex(out.data(), out.size()).c_str());
     78   }
     79 
     80   return true;
     81 }
     82 
     83 static int usage(char *arg) {
     84   fprintf(stderr, "usage: %s <hash> <test file>\n", arg);
     85   return 1;
     86 }
     87 
     88 int cavp_sha_monte_test_main(int argc, char **argv) {
     89   if (argc != 3) {
     90     return usage(argv[0]);
     91   }
     92 
     93   TestCtx ctx = {std::string(argv[1])};
     94 
     95   FileTest::Options opts;
     96   opts.path = argv[2];
     97   opts.callback = TestSHAMonte;
     98   opts.arg = &ctx;
     99   opts.silent = true;
    100   opts.comment_callback = EchoComment;
    101   return FileTestMain(opts);
    102 }
    103