/*
* (C) 2014,2015 Jack Lloyd
*
* Botan is released under the Simplified BSD License (see license.txt)
*/

#include "tests.h"

#include <botan/hex.h>
#include <botan/rng.h>
#include <botan/symkey.h>
#include <botan/internal/filesystem.h>
#include <botan/internal/fmt.h>
#include <botan/internal/loadstor.h>
#include <botan/internal/parsing.h>
#include <botan/internal/stl_util.h>
#include <botan/internal/target_info.h>
#include <chrono>
#include <deque>
#include <fstream>
#include <iomanip>
#include <set>
#include <sstream>
#include <unordered_set>

#if defined(BOTAN_HAS_BIGINT)
   #include <botan/bigint.h>
#endif

#if defined(BOTAN_HAS_CPUID)
   #include <botan/internal/cpuid.h>
#endif

#if defined(BOTAN_HAS_LEGACY_EC_POINT)
   #include <botan/ec_point.h>
#endif

#if defined(BOTAN_TARGET_OS_HAS_POSIX1)
   #include <stdlib.h>
   #include <unistd.h>
#endif

#if defined(BOTAN_TARGET_OS_HAS_FILESYSTEM)
   #include <version>
   #if defined(__cpp_lib_filesystem)
      #include <filesystem>
   #endif
#endif

#if defined(BOTAN_HAS_ECC_GROUP)
   #include <botan/ec_group.h>
#endif

namespace Botan_Tests {

Test_Error::Test_Error(std::string_view what) : std::runtime_error(std::string(what)) {}

Test::Test() = default;

Test::~Test() = default;

Test::Result::Result(std::string_view who) : m_who(who), m_timestamp(Test::timestamp()) {}

Test::Result::Result(std::string_view who, const std::vector<Result>& downstream_results) : Result(who) {
   for(const auto& result : downstream_results) {
      merge(result, true /* ignore non-matching test names */);
   }
}

void Test::Result::merge(const Result& other, bool ignore_test_name) {
   if(who() != other.who()) {
      if(!ignore_test_name) {
         throw Test_Error("Merging tests from different sources");
      }

      // When deliberately merging results with different names, the code location is
      // likely inconsistent and must be discarded.
      m_where.reset();
   } else {
      m_where = other.m_where;
   }

   m_timestamp = std::min(m_timestamp, other.m_timestamp);
   m_ns_taken += other.m_ns_taken;
   m_tests_passed += other.m_tests_passed;
   m_fail_log.insert(m_fail_log.end(), other.m_fail_log.begin(), other.m_fail_log.end());
   m_log.insert(m_log.end(), other.m_log.begin(), other.m_log.end());
}

void Test::Result::start_timer() {
   if(m_started == 0) {
      m_started = Test::timestamp();
   }
}

void Test::Result::end_timer() {
   if(m_started > 0) {
      m_ns_taken += Test::timestamp() - m_started;
      m_started = 0;
   }
}

void Test::Result::test_note(std::string_view note, std::span<const uint8_t> context) {
   return test_note(note, Botan::hex_encode(context));
}

void Test::Result::test_note(std::string_view note, std::string_view context) {
   m_log.emplace_back(Botan::fmt("{} {}: {}", who(), note, context));
}

void Test::Result::test_note(std::string_view note) {
   m_log.emplace_back(Botan::fmt("{} {}", who(), note));
}

void Test::Result::note_missing(std::string_view whatever_sv) {
   static std::set<std::string> s_already_seen;

   const std::string whatever(whatever_sv);
   if(!s_already_seen.contains(whatever)) {
      test_note(Botan::fmt("Skipping tests due to missing {}", whatever));
      s_already_seen.insert(whatever);
   }
}

void Test::Result::require(std::string_view what, bool expr) {
   if(!test_is_true(what, expr)) {
      throw Test_Aborted(Botan::fmt("Test aborted, because required condition was not met: {}", what));
   }
}

Test::Result::ThrowExpectations::~ThrowExpectations() {
   BOTAN_ASSERT_NOMSG(m_consumed);
}

void Test::Result::ThrowExpectations::assert_that_success_is_not_expected() const {
   BOTAN_ASSERT_NOMSG(!m_expect_success);
}

Test::Result::ThrowExpectations& Test::Result::ThrowExpectations::expect_success() {
   BOTAN_ASSERT_NOMSG(!m_expected_message && !m_expected_exception_check_fn);
   m_expect_success = true;
   return *this;
}

Test::Result::ThrowExpectations& Test::Result::ThrowExpectations::expect_message(std::string_view message) {
   assert_that_success_is_not_expected();
   m_expected_message = message;
   return *this;
}

bool Test::Result::ThrowExpectations::check(std::string_view test_name, Test::Result& result) {
   m_consumed = true;

   try {
      m_fn();
      if(!m_expect_success) {
         return result.test_failure(Botan::fmt("{} failed to throw expected exception", test_name));
      }
   } catch(const std::exception& ex) {
      if(m_expect_success) {
         return result.test_failure(Botan::fmt("{} threw unexpected exception: {}", test_name, ex.what()));
      }
      if(m_expected_exception_check_fn && !m_expected_exception_check_fn(std::current_exception())) {
         return result.test_failure(Botan::fmt("{} threw unexpected exception: {}", test_name, ex.what()));
      }
      if(m_expected_message.has_value() && m_expected_message.value() != ex.what()) {
         return result.test_failure(Botan::fmt("{} threw exception with unexpected message (expected {} got {})",
                                               test_name,
                                               m_expected_message.value(),
                                               ex.what()));
      }
   } catch(...) {
      if(m_expect_success || m_expected_exception_check_fn || m_expected_message.has_value()) {
         return result.test_failure(Botan::fmt("{} threw unexpected unknown exception", test_name));
      }
   }

   return result.test_success();
}

bool Test::Result::test_throws(std::string_view what, std::function<void()> fn) {
   return ThrowExpectations(std::move(fn)).check(what, *this);
}

bool Test::Result::test_throws(std::string_view what, std::string_view expected, std::function<void()> fn) {
   return ThrowExpectations(std::move(fn)).expect_message(expected).check(what, *this);
}

bool Test::Result::test_no_throw(std::string_view what, std::function<void()> fn) {
   return ThrowExpectations(std::move(fn)).expect_success().check(what, *this);
}

bool Test::Result::test_success(std::string_view note) {
   if(Test::options().log_success()) {
      test_note(note);
   }
   ++m_tests_passed;
   return true;
}

bool Test::Result::test_failure(std::string_view what, std::string_view error) {
   return test_failure(Botan::fmt("{} {} with error {}", who(), what, error));
}

void Test::Result::test_failure(std::string_view what, const uint8_t buf[], size_t buf_len) {
   return test_failure(what, {buf, buf_len});
}

void Test::Result::test_failure(std::string_view what, std::span<const uint8_t> context) {
   test_failure(Botan::fmt("{} {} with value {}", who(), what, Botan::hex_encode(context)));
}

bool Test::Result::test_failure(const char* err) {
   return test_failure(std::string_view(err));
}

bool Test::Result::test_failure(std::string_view err) {
   return test_failure(std::string(err));
}

bool Test::Result::test_failure(std::string err) {
   m_fail_log.push_back(std::move(err));

   if(Test::options().abort_on_first_fail() && m_who != "Failing Test") {
      std::abort();
   }
   return false;
}

namespace {

bool same_contents(std::span<const uint8_t> x, std::span<const uint8_t> y) {
   if(x.size() != y.size()) {
      return false;
   }
   if(x.empty()) {
      return true;
   }

   return std::memcmp(x.data(), y.data(), x.size()) == 0;
}

}  // namespace

bool Test::Result::test_bin_ne(std::string_view what,
                               std::span<const uint8_t> produced,
                               std::span<const uint8_t> expected) {
   if(produced.size() == expected.size() && same_contents(produced, expected)) {
      return test_failure(Botan::fmt("{} {} produced matching bytes", who(), what));
   }
   return test_success();
}

bool Test::Result::test_bin_eq(std::string_view what,
                               std::span<const uint8_t> produced,
                               std::string_view expected_hex) {
   const std::vector<uint8_t> expected = Botan::hex_decode(expected_hex);
   return test_bin_eq(what, produced, expected);
}

bool Test::Result::test_bin_eq(std::string_view what,
                               std::span<const uint8_t> produced,
                               std::span<const uint8_t> expected) {
   if(same_contents(produced, expected)) {
      return test_success();
   }

   std::ostringstream err;

   err << who();

   err << " unexpected result for " << what;

   if(produced.size() != expected.size()) {
      err << " produced " << produced.size() << " bytes expected " << expected.size();
   }

   std::vector<uint8_t> xor_diff(std::min(produced.size(), expected.size()));
   size_t bytes_different = 0;

   for(size_t i = 0; i != xor_diff.size(); ++i) {
      xor_diff[i] = produced[i] ^ expected[i];
      if(xor_diff[i] > 0) {
         bytes_different++;
      }
   }

   err << "\nProduced: " << Botan::hex_encode(produced) << "\nExpected: " << Botan::hex_encode(expected);

   if(bytes_different > 0) {
      err << "\nXOR Diff: " << Botan::hex_encode(xor_diff);
   }

   return test_failure(err.str());
}

bool Test::Result::test_str_not_empty(std::string_view what, std::string_view produced) {
   if(produced.empty()) {
      return test_failure(what, "was empty");
   } else {
      return test_success();
   }
}

bool Test::Result::test_str_eq(std::string_view what, std::string_view produced, std::string_view expected) {
   if(produced == expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: '{}' == '{}'", who(), what, produced, expected));
   }
}

bool Test::Result::test_str_ne(std::string_view what, std::string_view str1, std::string_view str2) {
   if(str1 != str2) {
      return test_success(Botan::fmt("{} != {}", str1, str2));
   } else {
      return test_failure(Botan::fmt("{} {} unexpectedly produced matching strings {}", who(), what, str1));
   }
}

bool Test::Result::test_i16_eq(std::string_view what, int16_t produced, int16_t expected) {
   return test_i32_eq(what, produced, expected);
}

bool Test::Result::test_i32_eq(std::string_view what, int32_t produced, int32_t expected) {
   if(produced == expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: {} == {}", who(), what, produced, expected));
   }
}

bool Test::Result::test_u8_eq(uint8_t produced, uint8_t expected) {
   return test_sz_eq("comparison", produced, expected);
}

bool Test::Result::test_u8_eq(std::string_view what, uint8_t produced, uint8_t expected) {
   return test_sz_eq(what, produced, expected);
}

bool Test::Result::test_u16_eq(uint16_t produced, uint16_t expected) {
   return test_sz_eq("comparison", produced, expected);
}

bool Test::Result::test_u16_eq(std::string_view what, uint16_t produced, uint16_t expected) {
   return test_sz_eq(what, produced, expected);
}

bool Test::Result::test_u32_eq(uint32_t produced, uint32_t expected) {
   return test_sz_eq("comparison", produced, expected);
}

bool Test::Result::test_u32_eq(std::string_view what, uint32_t produced, uint32_t expected) {
   return test_sz_eq(what, produced, expected);
}

bool Test::Result::test_u64_eq(uint64_t produced, uint64_t expected) {
   return test_u64_eq("comparison", produced, expected);
}

bool Test::Result::test_u64_eq(std::string_view what, uint64_t produced, uint64_t expected) {
   if(produced == expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: {} == {}", who(), what, produced, expected));
   }
}

bool Test::Result::test_u64_lt(std::string_view what, uint64_t produced, uint64_t expected) {
   if(produced < expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: {} < {}", who(), what, produced, expected));
   }
}

bool Test::Result::test_sz_eq(std::string_view what, size_t produced, size_t expected) {
   if(produced == expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: {} == {}", who(), what, produced, expected));
   }
}

bool Test::Result::test_sz_ne(std::string_view what, size_t produced, size_t expected) {
   if(produced != expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: {} != {}", who(), what, produced, expected));
   }
}

bool Test::Result::test_sz_lt(std::string_view what, size_t produced, size_t expected) {
   if(produced < expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: {} < {}", who(), what, produced, expected));
   }
}

bool Test::Result::test_sz_lte(std::string_view what, size_t produced, size_t expected) {
   if(produced <= expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: {} <= {}", who(), what, produced, expected));
   }
}

bool Test::Result::test_sz_gt(std::string_view what, size_t produced, size_t expected) {
   if(produced > expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: {} > {}", who(), what, produced, expected));
   }
}

bool Test::Result::test_sz_gte(std::string_view what, size_t produced, size_t expected) {
   if(produced >= expected) {
      return test_success();
   } else {
      return test_failure(Botan::fmt("Assertion failure in {} {}: {} >= {}", who(), what, produced, expected));
   }
}

bool Test::Result::test_opt_u8_eq(std::string_view what,
                                  std::optional<uint8_t> produced,
                                  std::optional<uint8_t> expected) {
   if(produced.has_value() and !expected.has_value()) {
      return test_failure(Botan::fmt("Assertion {} produced value {} but nullopt was expected", what, *produced));
   } else if(!produced.has_value() && expected.has_value()) {
      return test_failure(Botan::fmt("Assertion {} produced nullopt but {} was expected", what, *expected));
   } else if(produced.has_value() && expected.has_value()) {
      return test_u8_eq(what, *produced, *expected);
   } else {
      return test_success();
   }
}

bool Test::Result::test_opt_u64_eq(std::string_view what,
                                   std::optional<uint64_t> produced,
                                   std::optional<uint64_t> expected) {
   if(produced.has_value() and !expected.has_value()) {
      return test_failure(Botan::fmt("Assertion {} produced value {} but nullopt was expected", what, *produced));
   } else if(!produced.has_value() && expected.has_value()) {
      return test_failure(Botan::fmt("Assertion {} produced nullopt but {} was expected", what, *expected));
   } else if(produced.has_value() && expected.has_value()) {
      return test_u64_eq(what, *produced, *expected);
   } else {
      return test_success();
   }
}

#if defined(BOTAN_HAS_BIGINT)
bool Test::Result::test_bn_eq(std::string_view what, const BigInt& produced, const BigInt& expected) {
   if(produced == expected) {
      return test_success();
   } else {
      std::ostringstream err;
      err << who() << " " << what << " produced " << produced << " != expected value " << expected;
      return test_failure(err.str());
   }
}

bool Test::Result::test_bn_ne(std::string_view what, const BigInt& produced, const BigInt& expected) {
   if(produced != expected) {
      return test_success();
   } else {
      std::ostringstream err;
      err << who() << " " << what << " produced " << produced << " prohibited value";
      return test_failure(err.str());
   }
}
#endif

bool Test::Result::test_bool_eq(std::string_view what, bool produced, bool expected) {
   if(produced == expected) {
      return test_success();
   } else {
      if(expected == true) {
         return test_failure(Botan::fmt("Assertion failure in {}, {} was unexpectedly false", who(), what));
      } else {
         return test_failure(Botan::fmt("Assertion failure in {}, {} was unexpectedly true", who(), what));
      }
   }
}

bool Test::Result::test_is_true(std::string_view what, bool produced) {
   return test_bool_eq(what, produced, true);
}

bool Test::Result::test_is_false(std::string_view what, bool produced) {
   return test_bool_eq(what, produced, false);
}

bool Test::Result::test_rc_ok(std::string_view func, int rc) {
   if(rc != 0) {
      std::ostringstream err;
      err << m_who << " " << func << " unexpectedly failed with error code " << rc;
      return test_failure(err.str());
   }

   return test_success();
}

bool Test::Result::test_rc_fail(std::string_view func, std::string_view why, int rc) {
   if(rc == 0) {
      std::ostringstream err;
      err << m_who << " call to " << func << " unexpectedly succeeded expecting failure because " << why;
      return test_failure(err.str());
   }

   return test_success();
}

bool Test::Result::test_rc_init(std::string_view func, int rc) {
   if(rc == 0) {
      return test_success();
   } else {
      std::ostringstream msg;
      msg << m_who;
      msg << " " << func;

      // -40 is BOTAN_FFI_ERROR_NOT_IMPLEMENTED
      if(rc == -40) {
         msg << " returned not implemented";
      } else {
         msg << " unexpectedly failed with error code " << rc;
      }

      if(rc == -40) {
         this->test_note(msg.str());
      } else {
         this->test_failure(msg.str());
      }
      return false;
   }
}

bool Test::Result::test_rc(std::string_view func, int rc, int expected) {
   if(expected != rc) {
      std::ostringstream err;
      err << m_who;
      err << " call to " << func << " unexpectedly returned " << rc;
      err << " but expecting " << expected;
      return test_failure(err.str());
   }

   return test_success();
}

void Test::initialize(std::string test_name, CodeLocation location) {
   m_test_name = std::move(test_name);
   m_registration_location = location;
}

Botan::RandomNumberGenerator& Test::rng() const {
   if(!m_test_rng) {
      m_test_rng = Test::new_rng(m_test_name);
   }

   return *m_test_rng;
}

std::optional<std::string> Test::supported_ec_group_name(std::vector<std::string> preferred_groups) {
#if defined(BOTAN_HAS_ECC_GROUP)
   if(preferred_groups.empty()) {
      preferred_groups = {
         "secp256r1",
         "brainpool256r1",
         "secp384r1",
         "brainpool384r1",
         "secp521r1",
         "brainpool512r1",
      };
   }

   for(const auto& group : preferred_groups) {
      if(Botan::EC_Group::supports_named_group(group)) {
         return group;
      }
   }
#else
   BOTAN_UNUSED(preferred_groups);
#endif

   return std::nullopt;
}

std::vector<uint8_t> Test::mutate_vec(const std::vector<uint8_t>& v,
                                      Botan::RandomNumberGenerator& rng,
                                      bool maybe_resize,
                                      size_t min_offset) {
   std::vector<uint8_t> r = v;

   if(maybe_resize && (r.empty() || rng.next_byte() < 32)) {
      // TODO: occasionally truncate, insert at random index
      const size_t add = 1 + (rng.next_byte() % 16);
      r.resize(r.size() + add);
      rng.randomize(&r[r.size() - add], add);
   }

   if(r.size() > min_offset) {
      const size_t offset = std::max<size_t>(min_offset, rng.next_byte() % r.size());
      const uint8_t perturb = rng.next_nonzero_byte();
      r[offset] ^= perturb;
   }

   return r;
}

std::vector<std::string> Test::possible_providers(const std::string& /*alg*/) {
   return Test::provider_filter({"base"});
}

//static
std::string Test::format_time(uint64_t nanoseconds) {
   std::ostringstream o;

   if(nanoseconds > 1000000000) {
      o << std::setprecision(2) << std::fixed << nanoseconds / 1000000000.0 << " sec";
   } else {
      o << std::setprecision(2) << std::fixed << nanoseconds / 1000000.0 << " msec";
   }

   return o.str();
}

// TODO: this should move to `StdoutReporter`
std::string Test::Result::result_string() const {
   const bool verbose = Test::options().verbose();

   if(tests_run() == 0 && !verbose) {
      return "";
   }

   std::ostringstream report;

   report << who() << " ran ";

   if(tests_run() == 0) {
      report << "ZERO";
   } else {
      report << tests_run();
   }
   report << " tests";

   if(m_ns_taken > 0) {
      report << " in " << format_time(m_ns_taken);
   }

   if(tests_failed() > 0) {
      report << " " << tests_failed() << " FAILED";
   } else {
      report << " all ok";
   }

   report << "\n";

   for(size_t i = 0; i != m_fail_log.size(); ++i) {
      report << "Failure " << (i + 1) << ": " << m_fail_log[i];
      if(m_where) {
         report << " (at " << m_where->path << ":" << m_where->line << ")";
      }
      report << "\n";
   }

   if(!m_fail_log.empty() || tests_run() == 0 || verbose) {
      for(size_t i = 0; i != m_log.size(); ++i) {
         report << "Note " << (i + 1) << ": " << m_log[i] << "\n";
      }
   }

   return report.str();
}

namespace {

class Test_Registry {
   public:
      static Test_Registry& instance() {
         static Test_Registry registry;
         return registry;
      }

      void register_test(const std::string& category,
                         const std::string& name,
                         bool smoke_test,
                         bool needs_serialization,
                         std::function<std::unique_ptr<Test>()> maker_fn) {
         if(m_tests.contains(name)) {
            throw Test_Error("Duplicate registration of test '" + name + "'");
         }

         if(m_tests.contains(category)) {
            throw Test_Error("'" + category + "' cannot be used as category, test exists");
         }

         if(m_categories.contains(name)) {
            throw Test_Error("'" + name + "' cannot be used as test name, category exists");
         }

         if(smoke_test) {
            m_smoke_tests.push_back(name);
         }

         if(needs_serialization) {
            m_mutexed_tests.push_back(name);
         }

         m_tests.emplace(name, std::move(maker_fn));
         m_categories.emplace(category, name);
      }

      std::unique_ptr<Test> get_test(const std::string& test_name) const {
         auto i = m_tests.find(test_name);
         if(i != m_tests.end()) {
            return i->second();
         }
         return nullptr;
      }

      std::vector<std::string> registered_tests() const {
         std::vector<std::string> s;
         s.reserve(m_tests.size());
         for(auto&& i : m_tests) {
            s.push_back(i.first);
         }
         return s;
      }

      std::vector<std::string> registered_test_categories() const {
         std::set<std::string> s;
         for(auto&& i : m_categories) {
            s.insert(i.first);
         }
         return std::vector<std::string>(s.begin(), s.end());
      }

      std::vector<std::string> filter_registered_tests(const std::vector<std::string>& requested,
                                                       const std::vector<std::string>& to_be_skipped) {
         std::vector<std::string> result;

         std::set<std::string> to_be_skipped_set(to_be_skipped.begin(), to_be_skipped.end());
         // TODO: this is O(n^2), but we have a relatively small number of tests.
         auto insert_if_not_exists_and_not_skipped = [&](const std::string& test_name) {
            if(!Botan::value_exists(result, test_name) && !to_be_skipped_set.contains(test_name)) {
               result.push_back(test_name);
            }
         };

         if(requested.empty()) {
            /*
            If nothing was requested on the command line, run everything. First
            run the "essentials" to smoke test, then everything else in
            alphabetical order.
            */
            result = m_smoke_tests;
            for(const auto& [test_name, _] : m_tests) {
               insert_if_not_exists_and_not_skipped(test_name);
            }
         } else {
            for(const auto& r : requested) {
               if(m_tests.contains(r)) {
                  insert_if_not_exists_and_not_skipped(r);
               } else if(auto elems = m_categories.equal_range(r); elems.first != m_categories.end()) {
                  for(; elems.first != elems.second; ++elems.first) {
                     insert_if_not_exists_and_not_skipped(elems.first->second);
                  }
               } else {
                  throw Test_Error("Unknown test suite or category: " + r);
               }
            }
         }

         return result;
      }

      bool needs_serialization(const std::string& test_name) const {
         return Botan::value_exists(m_mutexed_tests, test_name);
      }

   private:
      Test_Registry() = default;

   private:
      std::map<std::string, std::function<std::unique_ptr<Test>()>> m_tests;
      std::multimap<std::string, std::string> m_categories;
      std::vector<std::string> m_smoke_tests;
      std::vector<std::string> m_mutexed_tests;
};

}  // namespace

// static Test:: functions

//static
void Test::register_test(const std::string& category,
                         const std::string& name,
                         bool smoke_test,
                         bool needs_serialization,
                         std::function<std::unique_ptr<Test>()> maker_fn) {
   Test_Registry::instance().register_test(category, name, smoke_test, needs_serialization, std::move(maker_fn));
}

//static
uint64_t Test::timestamp() {
   auto now = std::chrono::system_clock::now().time_since_epoch();
   return std::chrono::duration_cast<std::chrono::nanoseconds>(now).count();
}

//static
std::vector<Test::Result> Test::flatten_result_lists(std::vector<std::vector<Test::Result>> result_lists) {
   std::vector<Test::Result> results;
   for(auto& result_list : result_lists) {
      for(auto& result : result_list) {
         results.emplace_back(std::move(result));
      }
   }
   return results;
}

//static
std::vector<std::string> Test::registered_tests() {
   return Test_Registry::instance().registered_tests();
}

//static
std::vector<std::string> Test::registered_test_categories() {
   return Test_Registry::instance().registered_test_categories();
}

//static
std::unique_ptr<Test> Test::get_test(const std::string& test_name) {
   return Test_Registry::instance().get_test(test_name);
}

//static
bool Test::test_needs_serialization(const std::string& test_name) {
   return Test_Registry::instance().needs_serialization(test_name);
}

//static
std::vector<std::string> Test::filter_registered_tests(const std::vector<std::string>& requested,
                                                       const std::vector<std::string>& to_be_skipped) {
   return Test_Registry::instance().filter_registered_tests(requested, to_be_skipped);
}

//static
std::string Test::temp_file_name(const std::string& basename) {
   // TODO add a --tmp-dir option to the tests to specify where these files go

#if defined(BOTAN_TARGET_OS_HAS_POSIX1)

   // POSIX only calls for 6 'X' chars but OpenBSD allows arbitrary amount
   std::string mkstemp_basename = "/tmp/" + basename + ".XXXXXXXXXX";

   const int fd = ::mkstemp(mkstemp_basename.data());

   // error
   if(fd < 0) {
      return "";
   }

   ::close(fd);

   return mkstemp_basename;
#else
   // For now just create the temp in the current working directory
   return basename;
#endif
}

bool Test::copy_file(const std::string& from, const std::string& to) {
#if defined(BOTAN_TARGET_OS_HAS_FILESYSTEM) && defined(__cpp_lib_filesystem)
   std::error_code ec;  // don't throw, just return false on error
   return std::filesystem::copy_file(from, to, std::filesystem::copy_options::overwrite_existing, ec);
#else
   // TODO: implement fallbacks to POSIX or WIN32
   // ... but then again: it's 2023 and we're using C++20 :o)
   BOTAN_UNUSED(from, to);
   throw Botan::No_Filesystem_Access();
#endif
}

std::string Test::read_data_file(const std::string& path) {
   const std::string fsname = Test::data_file(path);
   std::ifstream file(fsname.c_str());
   if(!file.good()) {
      throw Test_Error("Error reading from " + fsname);
   }

   return std::string((std::istreambuf_iterator<char>(file)), std::istreambuf_iterator<char>());
}

std::vector<uint8_t> Test::read_binary_data_file(const std::string& path) {
   const std::string fsname = Test::data_file(path);
   std::ifstream file(fsname.c_str(), std::ios::binary);
   if(!file.good()) {
      throw Test_Error("Error reading from " + fsname);
   }

   std::vector<uint8_t> contents;

   while(file.good()) {
      std::vector<uint8_t> buf(4096);
      file.read(reinterpret_cast<char*>(buf.data()), buf.size());
      const size_t got = static_cast<size_t>(file.gcount());

      if(got == 0 && file.eof()) {
         break;
      }

      contents.insert(contents.end(), buf.data(), buf.data() + got);
   }

   return contents;
}

// static member variables of Test

// NOLINTNEXTLINE(*-avoid-non-const-global-variables)
Test_Options Test::m_opts;
// NOLINTNEXTLINE(*-avoid-non-const-global-variables)
std::string Test::m_test_rng_seed;

//static
void Test::set_test_options(const Test_Options& opts) {
   m_opts = opts;
}

namespace {

/*
* This is a fast, simple, deterministic PRNG that's used for running
* the tests. It is not intended to be cryptographically secure.
*/
class Testsuite_RNG final : public Botan::RandomNumberGenerator {
   public:
      std::string name() const override { return "Testsuite_RNG"; }

      void clear() override { m_x = 0; }

      bool accepts_input() const override { return true; }

      bool is_seeded() const override { return true; }

      void fill_bytes_with_input(std::span<uint8_t> output, std::span<const uint8_t> input) override {
         for(const auto byte : input) {
            mix(byte);
         }

         for(auto& byte : output) {
            byte = mix();
         }
      }

      Testsuite_RNG(std::string_view seed, std::string_view test_name) : m_x(0) {
         for(const char c : seed) {
            this->mix(static_cast<uint8_t>(c));
         }
         for(const char c : test_name) {
            this->mix(static_cast<uint8_t>(c));
         }
      }

   private:
      uint8_t mix(uint8_t input = 0) {
         m_x ^= input;
         m_x *= 0xF2E16957;
         m_x += 0xE50B590F;
         return static_cast<uint8_t>(m_x >> 27);
      }

      uint64_t m_x;
};

}  // namespace

//static
void Test::set_test_rng_seed(std::span<const uint8_t> seed, size_t epoch) {
   m_test_rng_seed = Botan::fmt("seed={} epoch={}", Botan::hex_encode(seed), epoch);
}

//static
std::unique_ptr<Botan::RandomNumberGenerator> Test::new_rng(std::string_view test_name) {
   return std::make_unique<Testsuite_RNG>(m_test_rng_seed, test_name);
}

//static
std::shared_ptr<Botan::RandomNumberGenerator> Test::new_shared_rng(std::string_view test_name) {
   return std::make_shared<Testsuite_RNG>(m_test_rng_seed, test_name);
}

//static
std::string Test::data_file(const std::string& file) {
   return options().data_dir() + "/" + file;
}

//static
std::string Test::data_dir(const std::string& subdir) {
   return options().data_dir() + "/" + subdir;
}

//static
std::vector<std::string> Test::files_in_data_dir(const std::string& subdir) {
   auto fs = Botan::get_files_recursive(options().data_dir() + "/" + subdir);
   if(fs.empty()) {
      throw Test_Error("Test::files_in_data_dir encountered empty subdir " + subdir);
   }
   return fs;
}

//static
std::string Test::data_file_as_temporary_copy(const std::string& what) {
   auto tmp_basename = what;
   std::replace(tmp_basename.begin(), tmp_basename.end(), '/', '_');
   auto temp_file = temp_file_name("tmp-" + tmp_basename);
   if(temp_file.empty()) {
      return "";
   }
   if(!Test::copy_file(data_file(what), temp_file)) {
      return "";
   }
   return temp_file;
}

//static
std::vector<std::string> Test::provider_filter(const std::vector<std::string>& providers) {
   if(m_opts.provider().empty()) {
      return providers;
   }
   for(auto&& provider : providers) {
      if(provider == m_opts.provider()) {
         return std::vector<std::string>{provider};
      }
   }
   return std::vector<std::string>{};
}

std::string Test::random_password(Botan::RandomNumberGenerator& rng) {
   const size_t len = 1 + rng.next_byte() % 32;
   return Botan::hex_encode(rng.random_vec(len));
}

size_t Test::random_index(Botan::RandomNumberGenerator& rng, size_t max) {
   return Botan::load_be(rng.random_array<8>()) % max;
}

void VarMap::clear() {
   m_vars.clear();
}

namespace {

bool varmap_pair_lt(const std::pair<std::string, std::string>& kv, std::string_view k) {
   return kv.first < k;
}

}  // namespace

bool VarMap::has_key(std::string_view key) const {
   return get_opt_var(key).has_value();
}

void VarMap::add(std::string_view key, std::string_view value) {
   auto i = std::lower_bound(m_vars.begin(), m_vars.end(), key, varmap_pair_lt);

   if(i != m_vars.end() && i->first == key) {
      i->second = value;
   } else {
      m_vars.emplace(i, key, value);
   }
}

std::optional<std::string> VarMap::get_opt_var(std::string_view key) const {
   auto i = std::lower_bound(m_vars.begin(), m_vars.end(), key, varmap_pair_lt);

   if(i != m_vars.end() && i->first == key) {
      return i->second;
   } else {
      return {};
   }
}

const std::string& VarMap::get_req_var(std::string_view key) const {
   auto i = std::lower_bound(m_vars.begin(), m_vars.end(), key, varmap_pair_lt);

   if(i != m_vars.end() && i->first == key) {
      return i->second;
   } else {
      throw Test_Error(Botan::fmt("Test missing variable '{}'", key));
   }
}

std::string VarMap::get_req_str(std::string_view key) const {
   return std::string(get_req_var(key));
}

std::vector<std::vector<uint8_t>> VarMap::get_req_bin_list(std::string_view key) const {
   const auto& var = get_req_var(key);

   std::vector<std::vector<uint8_t>> bin_list;

   for(auto&& part : Botan::split_on(var, ',')) {
      try {
         bin_list.push_back(Botan::hex_decode(part));
      } catch(std::exception& e) {
         std::ostringstream oss;
         oss << "Bad input '" << part << "'"
             << " in binary list key " << key << " - " << e.what();
         throw Test_Error(oss.str());
      }
   }

   return bin_list;
}

std::vector<uint8_t> VarMap::get_req_bin(std::string_view key) const {
   const auto& var = get_req_var(key);

   try {
      return Botan::hex_decode(var);
   } catch(std::exception& e) {
      throw Test_Error(Botan::fmt("Bad hex input '{}' for key '{}' err '{}'", var, key, e.what()));
   }
}

std::string VarMap::get_opt_str(std::string_view key, std::string_view def_value) const {
   if(auto v = get_opt_var(key)) {
      return *v;
   } else {
      return std::string(def_value);
   }
}

bool VarMap::get_req_bool(std::string_view key) const {
   const auto& var = get_req_var(key);

   if(var == "true") {
      return true;
   } else if(var == "false") {
      return false;
   } else {
      throw Test_Error(Botan::fmt("Invalid boolean '{}' for key '{}'", var, key));
   }
}

size_t VarMap::get_req_sz(std::string_view key) const {
   return Botan::to_u32bit(get_req_var(key));
}

uint8_t VarMap::get_req_u8(std::string_view key) const {
   const size_t s = this->get_req_sz(key);
   if(s > 256) {
      throw Test_Error(Botan::fmt("Invalid value for '{}' expected uint8_t but got '{}'", key, s));
   }
   return static_cast<uint8_t>(s);
}

uint32_t VarMap::get_req_u32(std::string_view key) const {
   return static_cast<uint32_t>(get_req_sz(key));
}

uint64_t VarMap::get_req_u64(std::string_view key) const {
   const auto& var = get_req_var(key);
   try {
      return std::stoull(var);
   } catch(std::exception&) {
      throw Test_Error("Invalid u64 value '" + var + "'");
   }
}

size_t VarMap::get_opt_sz(std::string_view key, const size_t def_value) const {
   if(auto v = get_opt_var(key)) {
      return Botan::to_u32bit(*v);
   } else {
      return def_value;
   }
}

uint64_t VarMap::get_opt_u64(std::string_view key, const uint64_t def_value) const {
   if(auto v = get_opt_var(key)) {
      try {
         return std::stoull(*v);
      } catch(std::exception&) {
         throw Test_Error("Invalid u64 value '" + *v + "'");
      }
   } else {
      return def_value;
   }
}

std::vector<uint8_t> VarMap::get_opt_bin(std::string_view key) const {
   if(auto v = get_opt_var(key)) {
      try {
         return Botan::hex_decode(*v);
      } catch(std::exception&) {
         throw Test_Error(Botan::fmt("Invalid hex for key '{}' got '{}'", key, *v));
      }
   } else {
      return {};
   };
}

#if defined(BOTAN_HAS_BIGINT)
Botan::BigInt VarMap::get_req_bn(std::string_view key) const {
   const auto& var = get_req_var(key);

   try {
      return Botan::BigInt(var);
   } catch(std::exception&) {
      throw Test_Error(Botan::fmt("Invalid BigInt for key '{}' got '{}'", key, var));
   }
}

Botan::BigInt VarMap::get_opt_bn(std::string_view key, const Botan::BigInt& def_value) const {
   if(auto v = get_opt_var(key)) {
      try {
         return Botan::BigInt(*v);
      } catch(std::exception&) {
         throw Test_Error(Botan::fmt("Invalid BigInt for key '{}' got '{}'", key, *v));
      }
   } else {
      return def_value;
   }
}
#endif

class Text_Based_Test::Text_Based_Test_Data {
   public:
      Text_Based_Test_Data(const std::string& data_src,
                           const std::string& required_keys_str,
                           const std::string& optional_keys_str) :
            m_data_src(data_src) {
         if(required_keys_str.empty()) {
            throw Test_Error("Invalid test spec");
         }

         m_required_keys = Botan::split_on(required_keys_str, ',');
         std::vector<std::string> optional_keys = Botan::split_on(optional_keys_str, ',');

         m_all_keys.insert(m_required_keys.begin(), m_required_keys.end());
         m_all_keys.insert(optional_keys.begin(), optional_keys.end());
         m_output_key = m_required_keys.at(m_required_keys.size() - 1);
      }

      std::string get_next_line();

      const std::string& current_source_name() const { return m_cur_src_name; }

      bool known_key(const std::string& key) const;

      const std::vector<std::string>& required_keys() const { return m_required_keys; }

      const std::string& output_key() const { return m_output_key; }

      const std::vector<std::string>& cpu_flags() const { return m_cpu_flags; }

      void set_cpu_flags(std::vector<std::string> flags) { m_cpu_flags = std::move(flags); }

      const std::string& initial_data_src_name() const { return m_data_src; }

   private:
      std::string m_data_src;
      std::vector<std::string> m_required_keys;
      std::unordered_set<std::string> m_all_keys;
      std::string m_output_key;

      bool m_first = true;
      std::unique_ptr<std::istream> m_cur;
      std::string m_cur_src_name;
      std::deque<std::string> m_srcs;
      std::vector<std::string> m_cpu_flags;
};

Text_Based_Test::Text_Based_Test(const std::string& data_src,
                                 const std::string& required_keys,
                                 const std::string& optional_keys) :
      m_data(std::make_unique<Text_Based_Test_Data>(data_src, required_keys, optional_keys)) {}

Text_Based_Test::~Text_Based_Test() = default;

std::string Text_Based_Test::Text_Based_Test_Data::get_next_line() {
   while(true) {
      if(m_cur == nullptr || m_cur->good() == false) {
         if(m_srcs.empty()) {
            if(m_first) {
               if(m_data_src.ends_with(".vec")) {
                  m_srcs.push_back(Test::data_file(m_data_src));
               } else {
                  const auto fs = Test::files_in_data_dir(m_data_src);
                  m_srcs.assign(fs.begin(), fs.end());
                  if(m_srcs.empty()) {
                     throw Test_Error("Error reading test data dir " + m_data_src);
                  }
               }

               m_first = false;
            } else {
               return "";  // done
            }
         }

         m_cur = std::make_unique<std::ifstream>(m_srcs[0]);
         m_cur_src_name = m_srcs[0];

#if defined(BOTAN_HAS_CPUID)
         // Reinit cpuid on new file if needed
         if(m_cpu_flags.empty() == false) {
            m_cpu_flags.clear();
            Botan::CPUID::initialize();
         }
#endif

         if(!m_cur->good()) {
            throw Test_Error("Could not open input file '" + m_cur_src_name);
         }

         m_srcs.pop_front();
      }

      while(m_cur->good()) {
         std::string line;
         std::getline(*m_cur, line);

         if(line.empty()) {
            continue;
         }

         if(line[0] == '#') {
            if(line.starts_with("#test ")) {
               return line;
            } else {
               continue;
            }
         }

         return line;
      }
   }
}

bool Text_Based_Test::Text_Based_Test_Data::known_key(const std::string& key) const {
   return m_all_keys.contains(key);
}

namespace {

// strips leading and trailing but not internal whitespace
std::string strip_ws(const std::string& in) {
   const char* whitespace = " ";

   const auto first_c = in.find_first_not_of(whitespace);
   if(first_c == std::string::npos) {
      return "";
   }

   const auto last_c = in.find_last_not_of(whitespace);

   return in.substr(first_c, last_c - first_c + 1);
}

std::vector<std::string> parse_cpuid_bits(const std::vector<std::string>& tok) {
   std::vector<std::string> bits;

#if defined(BOTAN_HAS_CPUID)
   for(size_t i = 1; i < tok.size(); ++i) {
      if(auto bit = Botan::CPUID::bit_from_string(tok[i])) {
         bits.push_back(bit->to_string());
      }
   }
#else
   BOTAN_UNUSED(tok);
#endif

   return bits;
}

}  // namespace

bool Text_Based_Test::skip_this_test(const std::string& /*header*/, const VarMap& /*vars*/) {
   return false;
}

std::vector<Test::Result> Text_Based_Test::run() {
   std::vector<Test::Result> results;

   std::string header;
   std::string header_or_name = m_data->initial_data_src_name();
   VarMap vars;
   size_t test_cnt = 0;

   while(true) {
      const std::string line = m_data->get_next_line();
      if(line.empty()) {
         // EOF
         break;
      }

      if(line.starts_with("#test ")) {
         std::vector<std::string> pragma_tokens = Botan::split_on(line.substr(6), ' ');

         if(pragma_tokens.empty()) {
            throw Test_Error("Empty pragma found in " + m_data->current_source_name());
         }

         if(pragma_tokens[0] != "cpuid") {
            throw Test_Error("Unknown test pragma '" + line + "' in " + m_data->current_source_name());
         }

         if(!Test_Registry::instance().needs_serialization(this->test_name())) {
            throw Test_Error(Botan::fmt("'{}' used cpuid control but is not serialized", this->test_name()));
         }

         m_data->set_cpu_flags(parse_cpuid_bits(pragma_tokens));

         continue;
      } else if(line[0] == '#') {
         throw Test_Error("Unknown test pragma '" + line + "' in " + m_data->current_source_name());
      }

      if(line[0] == '[' && line[line.size() - 1] == ']') {
         header = line.substr(1, line.size() - 2);
         header_or_name = header;
         test_cnt = 0;
         vars.clear();
         continue;
      }

      const std::string test_id = "test " + std::to_string(test_cnt);

      auto equal_i = line.find_first_of('=');

      if(equal_i == std::string::npos) {
         results.push_back(Test::Result::Failure(header_or_name, "invalid input '" + line + "'"));
         continue;
      }

      const std::string key = strip_ws(std::string(line.begin(), line.begin() + equal_i - 1));
      const std::string val = strip_ws(std::string(line.begin() + equal_i + 1, line.end()));

      if(!m_data->known_key(key)) {
         auto r = Test::Result::Failure(header_or_name, Botan::fmt("{} failed unknown key {}", test_id, key));
         results.push_back(r);
      }

      vars.add(key, val);

      if(key == m_data->output_key()) {
         try {
            for(const auto& req_key : m_data->required_keys()) {
               if(!vars.has_key(req_key)) {
                  auto r =
                     Test::Result::Failure(header_or_name, Botan::fmt("{} missing required key {}", test_id, req_key));
                  results.push_back(r);
               }
            }

            if(skip_this_test(header, vars)) {
               continue;
            }

            ++test_cnt;

            const uint64_t start = Test::timestamp();

            Test::Result result = run_one_test(header, vars);
#if defined(BOTAN_HAS_CPUID)
            if(!m_data->cpu_flags().empty()) {
               for(const auto& cpuid_str : m_data->cpu_flags()) {
                  if(const auto bit = Botan::CPUID::Feature::from_string(cpuid_str)) {
                     if(Botan::CPUID::has(*bit)) {
                        Botan::CPUID::clear_cpuid_bit(*bit);
                        // now re-run the test
                        result.merge(run_one_test(header, vars));
                     }
                  }
               }
               Botan::CPUID::initialize();
            }
#endif
            result.set_ns_consumed(Test::timestamp() - start);

            if(result.tests_failed() > 0) {
               std::ostringstream oss;
               oss << "Test # " << test_cnt << " ";
               if(!header.empty()) {
                  oss << header << " ";
               }
               oss << "failed ";

               for(const auto& k : m_data->required_keys()) {
                  oss << k << "=" << vars.get_req_str(k) << " ";
               }

               result.test_note(oss.str());
            }
            results.push_back(result);
         } catch(std::exception& e) {
            std::ostringstream oss;
            oss << "Test # " << test_cnt << " ";
            if(!header.empty()) {
               oss << header << " ";
            }

            for(const auto& k : m_data->required_keys()) {
               oss << k << "=" << vars.get_req_str(k) << " ";
            }

            oss << "failed with exception '" << e.what() << "'";

            results.push_back(Test::Result::Failure(header_or_name, oss.str()));
         }

         if(clear_between_callbacks()) {
            vars.clear();
         }
      }
   }

   if(results.empty()) {
      return results;
   }

   try {
      std::vector<Test::Result> final_tests = run_final_tests();
      results.insert(results.end(), final_tests.begin(), final_tests.end());
   } catch(std::exception& e) {
      results.push_back(Test::Result::Failure(header_or_name, "run_final_tests exception " + std::string(e.what())));
   }

   return results;
}

}  // namespace Botan_Tests
