/* 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"
#include "../test/stl_compat.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.
  uint8_t mac[EVP_MAX_MD_SIZE];
  unsigned mac_len;
  if (nullptr == HMAC(digest, bssl::vector_data(&key), key.size(),
                      bssl::vector_data(&input), input.size(), mac,
                      &mac_len) ||
      !t->ExpectBytesEqual(bssl::vector_data(&output), output.size(), mac,
                           mac_len)) {
    t->PrintLine("One-shot API failed.");
    return false;
  }

  // Test using HMAC_CTX.
  ScopedHMAC_CTX ctx;
  if (!HMAC_Init_ex(ctx.get(), bssl::vector_data(&key), key.size(), digest,
                    nullptr) ||
      !HMAC_Update(ctx.get(), bssl::vector_data(&input), input.size()) ||
      !HMAC_Final(ctx.get(), mac, &mac_len) ||
      !t->ExpectBytesEqual(bssl::vector_data(&output), output.size(), mac,
                           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(), bssl::vector_data(&input), input.size()) ||
      !HMAC_Final(ctx.get(), mac, &mac_len) ||
      !t->ExpectBytesEqual(bssl::vector_data(&output), output.size(), mac,
                           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, &mac_len) ||
      !t->ExpectBytesEqual(bssl::vector_data(&output), output.size(), mac,
                           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]);
}
