1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
|
/*
* (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;
}
std::unique_ptr<Botan::HMAC_DRBG> rng(new 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_TEST("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&, 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("chacha_rng", ChaCha_RNG_Tests);
#endif
}
}
|
