Add Test::random_index test utility

randombit · Jan 20, 2025 · fa5c704 · fa5c704
1 parent 1dc31ad
commit fa5c704
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/src/tests/tests.cpp b/src/tests/tests.cpp
@@ -10,6 +10,7 @@
 #include <botan/internal/cpuid.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 <fstream>
@@ -842,6 +843,10 @@ std::string Test::random_password(Botan::RandomNumberGenerator& rng) {
    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;
+}
+
 std::vector<std::vector<uint8_t>> VarMap::get_req_bin_list(const std::string& key) const {
    auto i = m_vars.find(key);
    if(i == m_vars.end()) {

diff --git a/src/tests/tests.h b/src/tests/tests.h
@@ -670,6 +670,7 @@ class Test {
       static std::shared_ptr<Botan::RandomNumberGenerator> new_shared_rng(std::string_view test_name);
 
       static std::string random_password(Botan::RandomNumberGenerator& rng);
+      static size_t random_index(Botan::RandomNumberGenerator& rng, size_t max);
       static uint64_t timestamp();  // nanoseconds arbitrary epoch
 
       static std::vector<Test::Result> flatten_result_lists(std::vector<std::vector<Test::Result>> result_lists);