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 <memory> 61 #include <string> 62 #include <vector> 63 64 #include <openssl/crypto.h> 65 #include <openssl/digest.h> 66 #include <openssl/hmac.h> 67 68 #include "../test/file_test.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 unsigned expected_mac_len = EVP_MD_size(digest); 108 std::unique_ptr<uint8_t[]> mac(new uint8_t[expected_mac_len]); 109 unsigned mac_len; 110 if (nullptr == HMAC(digest, key.data(), key.size(), input.data(), 111 input.size(), mac.get(), &mac_len) || 112 mac_len != expected_mac_len || 113 !t->ExpectBytesEqual(output.data(), output.size(), mac.get(), mac_len)) { 114 t->PrintLine("One-shot API failed."); 115 return false; 116 } 117 118 // Test using HMAC_CTX. 119 bssl::ScopedHMAC_CTX ctx; 120 if (!HMAC_Init_ex(ctx.get(), key.data(), key.size(), digest, nullptr) || 121 !HMAC_Update(ctx.get(), input.data(), input.size()) || 122 !HMAC_Final(ctx.get(), mac.get(), &mac_len) || 123 mac_len != expected_mac_len || 124 !t->ExpectBytesEqual(output.data(), output.size(), mac.get(), mac_len)) { 125 t->PrintLine("HMAC_CTX failed."); 126 return false; 127 } 128 129 // Test that an HMAC_CTX may be reset with the same key. 130 if (!HMAC_Init_ex(ctx.get(), nullptr, 0, digest, nullptr) || 131 !HMAC_Update(ctx.get(), input.data(), input.size()) || 132 !HMAC_Final(ctx.get(), mac.get(), &mac_len) || 133 mac_len != expected_mac_len || 134 !t->ExpectBytesEqual(output.data(), output.size(), mac.get(), mac_len)) { 135 t->PrintLine("HMAC_CTX with reset failed."); 136 return false; 137 } 138 139 // Test feeding the input in byte by byte. 140 if (!HMAC_Init_ex(ctx.get(), nullptr, 0, nullptr, nullptr)) { 141 t->PrintLine("HMAC_CTX streaming failed."); 142 return false; 143 } 144 for (size_t i = 0; i < input.size(); i++) { 145 if (!HMAC_Update(ctx.get(), &input[i], 1)) { 146 t->PrintLine("HMAC_CTX streaming failed."); 147 return false; 148 } 149 } 150 if (!HMAC_Final(ctx.get(), mac.get(), &mac_len) || 151 mac_len != expected_mac_len || 152 !t->ExpectBytesEqual(output.data(), output.size(), mac.get(), mac_len)) { 153 t->PrintLine("HMAC_CTX streaming failed."); 154 return false; 155 } 156 157 return true; 158 } 159 160 int main(int argc, char *argv[]) { 161 CRYPTO_library_init(); 162 163 if (argc != 2) { 164 fprintf(stderr, "%s <test file.txt>\n", argv[0]); 165 return 1; 166 } 167 168 return FileTestMain(TestHMAC, nullptr, argv[1]); 169 } 170