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