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_test processes a NIST CAVP SHA test vector request file and emits
16 // 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 namespace {
27
28 struct TestCtx {
29 std::string hash;
30 };
31
32 }
33
TestSHA(FileTest * t,void * arg)34 static bool TestSHA(FileTest *t, void *arg) {
35 TestCtx *ctx = reinterpret_cast<TestCtx *>(arg);
36
37 const EVP_MD *md = EVP_get_digestbyname(ctx->hash.c_str());
38 if (md == nullptr) {
39 return false;
40 }
41 const size_t md_len = EVP_MD_size(md);
42
43 std::string out_len;
44 if (!t->GetInstruction(&out_len, "L") ||
45 md_len != strtoul(out_len.c_str(), nullptr, 0)) {
46 return false;
47 }
48
49 std::string msg_len_str;
50 std::vector<uint8_t> msg;
51 if (!t->GetAttribute(&msg_len_str, "Len") ||
52 !t->GetBytes(&msg, "Msg")) {
53 return false;
54 }
55
56 size_t msg_len = strtoul(msg_len_str.c_str(), nullptr, 0);
57 if (msg_len % 8 != 0 ||
58 msg_len / 8 > msg.size()) {
59 return false;
60 }
61 msg_len /= 8;
62
63 std::vector<uint8_t> out;
64 out.resize(md_len);
65 unsigned digest_len;
66 if (!EVP_Digest(msg.data(), msg_len, out.data(), &digest_len, md, nullptr) ||
67 digest_len != out.size()) {
68 return false;
69 }
70
71 printf("%s", t->CurrentTestToString().c_str());
72 printf("MD = %s\r\n\r\n", EncodeHex(out.data(), out.size()).c_str());
73
74 return true;
75 }
76
usage(char * arg)77 static int usage(char *arg) {
78 fprintf(stderr, "usage: %s <hash> <test file>\n", arg);
79 return 1;
80 }
81
cavp_sha_test_main(int argc,char ** argv)82 int cavp_sha_test_main(int argc, char **argv) {
83 if (argc != 3) {
84 return usage(argv[0]);
85 }
86
87 TestCtx ctx = {std::string(argv[1])};
88
89 FileTest::Options opts;
90 opts.path = argv[2];
91 opts.callback = TestSHA;
92 opts.arg = &ctx;
93 opts.silent = true;
94 opts.comment_callback = EchoComment;
95 return FileTestMain(opts);
96 }
97