| /* Copyright (C) 1995-1998 Eric Young (eay@cryptsoft.com) |
| * All rights reserved. |
| * |
| * This package is an SSL implementation written |
| * by Eric Young (eay@cryptsoft.com). |
| * The implementation was written so as to conform with Netscapes SSL. |
| * |
| * This library is free for commercial and non-commercial use as long as |
| * the following conditions are aheared to. The following conditions |
| * apply to all code found in this distribution, be it the RC4, RSA, |
| * lhash, DES, etc., code; not just the SSL code. The SSL documentation |
| * included with this distribution is covered by the same copyright terms |
| * except that the holder is Tim Hudson (tjh@cryptsoft.com). |
| * |
| * Copyright remains Eric Young's, and as such any Copyright notices in |
| * the code are not to be removed. |
| * If this package is used in a product, Eric Young should be given attribution |
| * as the author of the parts of the library used. |
| * This can be in the form of a textual message at program startup or |
| * in documentation (online or textual) provided with the package. |
| * |
| * Redistribution and use in source and binary forms, with or without |
| * modification, are permitted provided that the following conditions |
| * are met: |
| * 1. Redistributions of source code must retain the copyright |
| * notice, this list of conditions and the following disclaimer. |
| * 2. Redistributions in binary form must reproduce the above copyright |
| * notice, this list of conditions and the following disclaimer in the |
| * documentation and/or other materials provided with the distribution. |
| * 3. All advertising materials mentioning features or use of this software |
| * must display the following acknowledgement: |
| * "This product includes cryptographic software written by |
| * Eric Young (eay@cryptsoft.com)" |
| * The word 'cryptographic' can be left out if the rouines from the library |
| * being used are not cryptographic related :-). |
| * 4. If you include any Windows specific code (or a derivative thereof) from |
| * the apps directory (application code) you must include an acknowledgement: |
| * "This product includes software written by Tim Hudson (tjh@cryptsoft.com)" |
| * |
| * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND |
| * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE |
| * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
| * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
| * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
| * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
| * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
| * SUCH DAMAGE. |
| * |
| * The licence and distribution terms for any publically available version or |
| * derivative of this code cannot be changed. i.e. this code cannot simply be |
| * copied and put under another distribution licence |
| * [including the GNU Public Licence.] */ |
| |
| #include <stdio.h> |
| #include <string.h> |
| |
| #include <string> |
| #include <vector> |
| |
| #include <openssl/crypto.h> |
| #include <openssl/digest.h> |
| #include <openssl/hmac.h> |
| |
| #include "../test/file_test.h" |
| #include "../test/scoped_types.h" |
| |
| |
| static const EVP_MD *GetDigest(const std::string &name) { |
| if (name == "MD5") { |
| return EVP_md5(); |
| } else if (name == "SHA1") { |
| return EVP_sha1(); |
| } else if (name == "SHA224") { |
| return EVP_sha224(); |
| } else if (name == "SHA256") { |
| return EVP_sha256(); |
| } else if (name == "SHA384") { |
| return EVP_sha384(); |
| } else if (name == "SHA512") { |
| return EVP_sha512(); |
| } |
| return nullptr; |
| } |
| |
| static bool TestHMAC(FileTest *t, void *arg) { |
| std::string digest_str; |
| if (!t->GetAttribute(&digest_str, "HMAC")) { |
| return false; |
| } |
| const EVP_MD *digest = GetDigest(digest_str); |
| if (digest == nullptr) { |
| t->PrintLine("Unknown digest '%s'", digest_str.c_str()); |
| return false; |
| } |
| |
| std::vector<uint8_t> key, input, output; |
| if (!t->GetBytes(&key, "Key") || |
| !t->GetBytes(&input, "Input") || |
| !t->GetBytes(&output, "Output")) { |
| return false; |
| } |
| |
| // Test using the one-shot API. |
| unsigned expected_mac_len = EVP_MD_size(digest); |
| std::unique_ptr<uint8_t[]> mac(new uint8_t[expected_mac_len]); |
| unsigned mac_len; |
| if (nullptr == HMAC(digest, key.data(), key.size(), input.data(), |
| input.size(), mac.get(), &mac_len) || |
| mac_len != expected_mac_len || |
| !t->ExpectBytesEqual(output.data(), output.size(), mac.get(), mac_len)) { |
| t->PrintLine("One-shot API failed."); |
| return false; |
| } |
| |
| // Test using HMAC_CTX. |
| ScopedHMAC_CTX ctx; |
| if (!HMAC_Init_ex(ctx.get(), key.data(), key.size(), digest, nullptr) || |
| !HMAC_Update(ctx.get(), input.data(), input.size()) || |
| !HMAC_Final(ctx.get(), mac.get(), &mac_len) || |
| mac_len != expected_mac_len || |
| !t->ExpectBytesEqual(output.data(), output.size(), mac.get(), mac_len)) { |
| t->PrintLine("HMAC_CTX failed."); |
| return false; |
| } |
| |
| // Test that an HMAC_CTX may be reset with the same key. |
| if (!HMAC_Init_ex(ctx.get(), nullptr, 0, digest, nullptr) || |
| !HMAC_Update(ctx.get(), input.data(), input.size()) || |
| !HMAC_Final(ctx.get(), mac.get(), &mac_len) || |
| mac_len != expected_mac_len || |
| !t->ExpectBytesEqual(output.data(), output.size(), mac.get(), mac_len)) { |
| t->PrintLine("HMAC_CTX with reset failed."); |
| return false; |
| } |
| |
| // Test feeding the input in byte by byte. |
| if (!HMAC_Init_ex(ctx.get(), nullptr, 0, nullptr, nullptr)) { |
| t->PrintLine("HMAC_CTX streaming failed."); |
| return false; |
| } |
| for (size_t i = 0; i < input.size(); i++) { |
| if (!HMAC_Update(ctx.get(), &input[i], 1)) { |
| t->PrintLine("HMAC_CTX streaming failed."); |
| return false; |
| } |
| } |
| if (!HMAC_Final(ctx.get(), mac.get(), &mac_len) || |
| mac_len != expected_mac_len || |
| !t->ExpectBytesEqual(output.data(), output.size(), mac.get(), mac_len)) { |
| t->PrintLine("HMAC_CTX streaming failed."); |
| return false; |
| } |
| |
| return true; |
| } |
| |
| int main(int argc, char *argv[]) { |
| CRYPTO_library_init(); |
| |
| if (argc != 2) { |
| fprintf(stderr, "%s <test file.txt>\n", argv[0]); |
| return 1; |
| } |
| |
| return FileTestMain(TestHMAC, nullptr, argv[1]); |
| } |