diff options
author | Jack Lloyd <[email protected]> | 2017-10-02 22:03:07 -0400 |
---|---|---|
committer | Jack Lloyd <[email protected]> | 2017-10-02 22:45:20 -0400 |
commit | 40ec27f8a12e18e7eb17df4ff5874f3bb66987c5 (patch) | |
tree | 6fdf489a79eea085dd0eb3837e424571bb7e1396 | |
parent | c24bcc2fc88b66e9b58375de70917cd4d9793a35 (diff) |
Remove protected data from Fixed_Output_RNG in test code
-rw-r--r-- | src/tests/test_rng.h | 46 |
1 files changed, 15 insertions, 31 deletions
diff --git a/src/tests/test_rng.h b/src/tests/test_rng.h index 77ae2ba1a..9ac93577b 100644 --- a/src/tests/test_rng.h +++ b/src/tests/test_rng.h @@ -29,18 +29,6 @@ class Fixed_Output_RNG : public Botan::RandomNumberGenerator return !m_buf.empty(); } - virtual uint8_t random() - { - if(!is_seeded()) - { - throw Test_Error("Fixed output RNG ran out of bytes, test bug?"); - } - - uint8_t out = m_buf.front(); - m_buf.pop_front(); - return out; - } - size_t reseed(Botan::Entropy_Sources&, size_t, std::chrono::milliseconds) override @@ -81,11 +69,19 @@ class Fixed_Output_RNG : public Botan::RandomNumberGenerator Fixed_Output_RNG() = default; protected: - size_t remaining() const + uint8_t random() { - return m_buf.size(); + if(m_buf.empty()) + { + throw Test_Error("Fixed output RNG ran out of bytes, test bug?"); + } + + uint8_t out = m_buf.front(); + m_buf.pop_front(); + return out; } + private: std::deque<uint8_t> m_buf; }; @@ -98,19 +94,7 @@ class Fixed_Output_Position_RNG final : public Fixed_Output_RNG public: bool is_seeded() const override { - return !m_buf.empty() || Test::rng().is_seeded(); - } - - uint8_t random() override - { - if(m_buf.empty()) - { - throw Test_Error("Fixed output RNG ran out of bytes, test bug?"); - } - - uint8_t out = m_buf.front(); - m_buf.pop_front(); - return out; + return Fixed_Output_RNG::is_seeded() || Test::rng().is_seeded(); } void randomize(uint8_t out[], size_t len) override @@ -142,17 +126,17 @@ class Fixed_Output_Position_RNG final : public Fixed_Output_RNG return "Fixed_Output_Position_RNG"; } - explicit Fixed_Output_Position_RNG(const std::vector<uint8_t>& in, uint32_t pos) + explicit Fixed_Output_Position_RNG(const std::vector<uint8_t>& in, size_t pos) : Fixed_Output_RNG(in) , m_pos(pos) {} - explicit Fixed_Output_Position_RNG(const std::string& in_str, uint32_t pos) + explicit Fixed_Output_Position_RNG(const std::string& in_str, size_t pos) : Fixed_Output_RNG(in_str) , m_pos(pos) {} private: - uint32_t m_pos = 0; - uint32_t m_requests = 0; + size_t m_pos = 0; + size_t m_requests = 0; }; class SeedCapturing_RNG final : public Botan::RandomNumberGenerator |