/* Copyright (c) 2017, Google Inc.
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
 * SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
 * OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
 * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */

#include <openssl/rand.h>

#include <openssl/crypto.h>

#include "internal.h"
#include "../test/test_util.h"
#include "../test/file_test.h"


static bool TestCTRDRBG(FileTest *t, void *arg) {
  std::vector<uint8_t> seed, personalisation, reseed, ai_reseed, ai1, ai2,
    expected;
  if (!t->GetBytes(&seed, "EntropyInput") ||
      !t->GetBytes(&personalisation, "PersonalizationString") ||
      !t->GetBytes(&reseed, "EntropyInputReseed") ||
      !t->GetBytes(&ai_reseed, "AdditionalInputReseed") ||
      !t->GetBytes(&ai1, "AdditionalInput1") ||
      !t->GetBytes(&ai2, "AdditionalInput2") ||
      !t->GetBytes(&expected, "ReturnedBits")) {
    t->PrintLine("missing value");
    return false;
  }

  if (seed.size() != CTR_DRBG_ENTROPY_LEN ||
      reseed.size() != CTR_DRBG_ENTROPY_LEN) {
    t->PrintLine("bad seed length");
    return false;
  }

  CTR_DRBG_STATE drbg;
  CTR_DRBG_init(&drbg, seed.data(),
                personalisation.size() > 0 ? personalisation.data() : nullptr,
                personalisation.size());
  CTR_DRBG_reseed(&drbg, reseed.data(),
                  ai_reseed.size() > 0 ? ai_reseed.data() : nullptr,
                  ai_reseed.size());

  std::vector<uint8_t> out;
  out.resize(expected.size());

  CTR_DRBG_generate(&drbg, out.data(), out.size(),
                    ai1.size() > 0 ? ai1.data() : nullptr, ai1.size());
  CTR_DRBG_generate(&drbg, out.data(), out.size(),
                    ai2.size() > 0 ? ai2.data() : nullptr, ai2.size());

  return t->ExpectBytesEqual(expected.data(), expected.size(), out.data(),
                             out.size());
}

int main(int argc, char **argv) {
  CRYPTO_library_init();

  if (argc != 2) {
    fprintf(stderr, "%s <test file>\n", argv[0]);
    return 1;
  }

  return FileTestMain(TestCTRDRBG, nullptr, argv[1]);
}