Revision control
Copy as Markdown
Other Tools
/*
* (C) 2014,2015,2017 Jack Lloyd
* (C) 2016 René Korthaus, Rohde & Schwarz Cybersecurity
*
* Botan is released under the Simplified BSD License (see license.txt)
*/
#include "tests.h"
#if defined(BOTAN_HAS_HMAC_DRBG)
#include <botan/hmac_drbg.h>
#endif
#if defined(BOTAN_HAS_CHACHA_RNG)
#include <botan/chacha_rng.h>
#endif
namespace Botan_Tests {
namespace {
#if defined(BOTAN_HAS_HMAC_DRBG)
class HMAC_DRBG_Tests final : public Text_Based_Test {
public:
HMAC_DRBG_Tests() :
Text_Based_Test(
"rng/hmac_drbg.vec", "EntropyInput,EntropyInputReseed,Out", "AdditionalInput1,AdditionalInput2") {}
Test::Result run_one_test(const std::string& algo, const VarMap& vars) override {
const std::vector<uint8_t> seed_input = vars.get_req_bin("EntropyInput");
const std::vector<uint8_t> reseed_input = vars.get_req_bin("EntropyInputReseed");
const std::vector<uint8_t> expected = vars.get_req_bin("Out");
const std::vector<uint8_t> ad1 = vars.get_opt_bin("AdditionalInput1");
const std::vector<uint8_t> ad2 = vars.get_opt_bin("AdditionalInput2");
Test::Result result("HMAC_DRBG(" + algo + ")");
auto mac = Botan::MessageAuthenticationCode::create("HMAC(" + algo + ")");
if(!mac) {
result.note_missing("HMAC(" + algo + ")");
return result;
}
auto rng = std::make_unique<Botan::HMAC_DRBG>(std::move(mac));
rng->initialize_with(seed_input.data(), seed_input.size());
// now reseed
rng->add_entropy(reseed_input.data(), reseed_input.size());
std::vector<uint8_t> out(expected.size());
// first block is discarded
rng->randomize_with_input(out.data(), out.size(), ad1.data(), ad1.size());
rng->randomize_with_input(out.data(), out.size(), ad2.data(), ad2.size());
result.test_eq("rng", out, expected);
return result;
}
};
BOTAN_REGISTER_SMOKE_TEST("rng", "hmac_drbg", HMAC_DRBG_Tests);
#endif
#if defined(BOTAN_HAS_CHACHA_RNG)
class ChaCha_RNG_Tests final : public Text_Based_Test {
public:
ChaCha_RNG_Tests() :
Text_Based_Test(
"rng/chacha_rng.vec", "EntropyInput,EntropyInputReseed,Out", "AdditionalInput1,AdditionalInput2") {}
Test::Result run_one_test(const std::string& /*header*/, const VarMap& vars) override {
const std::vector<uint8_t> seed_input = vars.get_req_bin("EntropyInput");
const std::vector<uint8_t> reseed_input = vars.get_req_bin("EntropyInputReseed");
const std::vector<uint8_t> expected = vars.get_req_bin("Out");
const std::vector<uint8_t> ad1 = vars.get_opt_bin("AdditionalInput1");
const std::vector<uint8_t> ad2 = vars.get_opt_bin("AdditionalInput2");
Test::Result result("ChaCha_RNG");
Botan::ChaCha_RNG rng;
rng.initialize_with(seed_input.data(), seed_input.size());
// now reseed
rng.add_entropy(reseed_input.data(), reseed_input.size());
std::vector<uint8_t> out(expected.size());
// first block is discarded
rng.randomize_with_input(out.data(), out.size(), ad1.data(), ad1.size());
rng.randomize_with_input(out.data(), out.size(), ad2.data(), ad2.size());
result.test_eq("rng", out, expected);
return result;
}
};
BOTAN_REGISTER_TEST("rng", "chacha_rng", ChaCha_RNG_Tests);
#endif
} // namespace
} // namespace Botan_Tests