1 /* Copyright (C) 1995-1998 Eric Young (eay (at) cryptsoft.com) 2 * All rights reserved. 3 * 4 * This package is an SSL implementation written 5 * by Eric Young (eay (at) cryptsoft.com). 6 * The implementation was written so as to conform with Netscapes SSL. 7 * 8 * This library is free for commercial and non-commercial use as long as 9 * the following conditions are aheared to. The following conditions 10 * apply to all code found in this distribution, be it the RC4, RSA, 11 * lhash, DES, etc., code; not just the SSL code. The SSL documentation 12 * included with this distribution is covered by the same copyright terms 13 * except that the holder is Tim Hudson (tjh (at) cryptsoft.com). 14 * 15 * Copyright remains Eric Young's, and as such any Copyright notices in 16 * the code are not to be removed. 17 * If this package is used in a product, Eric Young should be given attribution 18 * as the author of the parts of the library used. 19 * This can be in the form of a textual message at program startup or 20 * in documentation (online or textual) provided with the package. 21 * 22 * Redistribution and use in source and binary forms, with or without 23 * modification, are permitted provided that the following conditions 24 * are met: 25 * 1. Redistributions of source code must retain the copyright 26 * notice, this list of conditions and the following disclaimer. 27 * 2. Redistributions in binary form must reproduce the above copyright 28 * notice, this list of conditions and the following disclaimer in the 29 * documentation and/or other materials provided with the distribution. 30 * 3. All advertising materials mentioning features or use of this software 31 * must display the following acknowledgement: 32 * "This product includes cryptographic software written by 33 * Eric Young (eay (at) cryptsoft.com)" 34 * The word 'cryptographic' can be left out if the rouines from the library 35 * being used are not cryptographic related :-). 36 * 4. If you include any Windows specific code (or a derivative thereof) from 37 * the apps directory (application code) you must include an acknowledgement: 38 * "This product includes software written by Tim Hudson (tjh (at) cryptsoft.com)" 39 * 40 * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND 41 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 42 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 43 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE 44 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL 45 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS 46 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) 47 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT 48 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 49 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 50 * SUCH DAMAGE. 51 * 52 * The licence and distribution terms for any publically available version or 53 * derivative of this code cannot be changed. i.e. this code cannot simply be 54 * copied and put under another distribution licence 55 * [including the GNU Public Licence.] */ 56 57 #include <stdio.h> 58 #include <string.h> 59 60 #include <string> 61 #include <vector> 62 63 #include <openssl/crypto.h> 64 #include <openssl/digest.h> 65 #include <openssl/hmac.h> 66 67 #include "../test/file_test.h" 68 #include "../test/scoped_types.h" 69 70 71 static const EVP_MD *GetDigest(const std::string &name) { 72 if (name == "MD5") { 73 return EVP_md5(); 74 } else if (name == "SHA1") { 75 return EVP_sha1(); 76 } else if (name == "SHA224") { 77 return EVP_sha224(); 78 } else if (name == "SHA256") { 79 return EVP_sha256(); 80 } else if (name == "SHA384") { 81 return EVP_sha384(); 82 } else if (name == "SHA512") { 83 return EVP_sha512(); 84 } 85 return nullptr; 86 } 87 88 static bool TestHMAC(FileTest *t, void *arg) { 89 std::string digest_str; 90 if (!t->GetAttribute(&digest_str, "HMAC")) { 91 return false; 92 } 93 const EVP_MD *digest = GetDigest(digest_str); 94 if (digest == nullptr) { 95 t->PrintLine("Unknown digest '%s'", digest_str.c_str()); 96 return false; 97 } 98 99 std::vector<uint8_t> key, input, output; 100 if (!t->GetBytes(&key, "Key") || 101 !t->GetBytes(&input, "Input") || 102 !t->GetBytes(&output, "Output")) { 103 return false; 104 } 105 106 // Test using the one-shot API. 107 uint8_t mac[EVP_MAX_MD_SIZE]; 108 unsigned mac_len; 109 if (nullptr == HMAC(digest, key.data(), key.size(), input.data(), 110 input.size(), mac, &mac_len) || 111 !t->ExpectBytesEqual(output.data(), output.size(), mac, mac_len)) { 112 t->PrintLine("One-shot API failed."); 113 return false; 114 } 115 116 // Test using HMAC_CTX. 117 ScopedHMAC_CTX ctx; 118 if (!HMAC_Init_ex(ctx.get(), key.data(), key.size(), digest, nullptr) || 119 !HMAC_Update(ctx.get(), input.data(), input.size()) || 120 !HMAC_Final(ctx.get(), mac, &mac_len) || 121 !t->ExpectBytesEqual(output.data(), output.size(), mac, mac_len)) { 122 t->PrintLine("HMAC_CTX failed."); 123 return false; 124 } 125 126 // Test that an HMAC_CTX may be reset with the same key. 127 if (!HMAC_Init_ex(ctx.get(), nullptr, 0, digest, nullptr) || 128 !HMAC_Update(ctx.get(), input.data(), input.size()) || 129 !HMAC_Final(ctx.get(), mac, &mac_len) || 130 !t->ExpectBytesEqual(output.data(), output.size(), mac, mac_len)) { 131 t->PrintLine("HMAC_CTX with reset failed."); 132 return false; 133 } 134 135 // Test feeding the input in byte by byte. 136 if (!HMAC_Init_ex(ctx.get(), nullptr, 0, nullptr, nullptr)) { 137 t->PrintLine("HMAC_CTX streaming failed."); 138 return false; 139 } 140 for (size_t i = 0; i < input.size(); i++) { 141 if (!HMAC_Update(ctx.get(), &input[i], 1)) { 142 t->PrintLine("HMAC_CTX streaming failed."); 143 return false; 144 } 145 } 146 if (!HMAC_Final(ctx.get(), mac, &mac_len) || 147 !t->ExpectBytesEqual(output.data(), output.size(), mac, mac_len)) { 148 t->PrintLine("HMAC_CTX streaming failed."); 149 return false; 150 } 151 152 return true; 153 } 154 155 int main(int argc, char *argv[]) { 156 CRYPTO_library_init(); 157 158 if (argc != 2) { 159 fprintf(stderr, "%s <test file.txt>\n", argv[0]); 160 return 1; 161 } 162 163 return FileTestMain(TestHMAC, nullptr, argv[1]); 164 } 165
