diff options
author | lloyd <[email protected]> | 2014-01-01 21:20:55 +0000 |
---|---|---|
committer | lloyd <[email protected]> | 2014-01-01 21:20:55 +0000 |
commit | 197dc467dec28a04c3b2f30da7cef122dfbb13e9 (patch) | |
tree | cdbd3ddaec051c72f0a757db461973d90c37b97a /src/tests/test_rng.cpp | |
parent | 62faac373c07cfe10bc8c309e89ebdd30d8e5eaa (diff) |
Shuffle things around. Add NIST X.509 test to build.
Diffstat (limited to 'src/tests/test_rng.cpp')
-rw-r--r-- | src/tests/test_rng.cpp | 63 |
1 files changed, 63 insertions, 0 deletions
diff --git a/src/tests/test_rng.cpp b/src/tests/test_rng.cpp new file mode 100644 index 000000000..99b2fbf8f --- /dev/null +++ b/src/tests/test_rng.cpp @@ -0,0 +1,63 @@ +#include "tests.h" +#include "test_rng.h" + +#include <botan/libstate.h> +#include <botan/x931_rng.h> +#include <botan/aes.h> +#include <botan/des.h> +#include <botan/hex.h> +#include <iostream> +#include <fstream> +#include <deque> + +using namespace Botan; + +namespace { + +RandomNumberGenerator* get_x931(const std::string& algo, const std::string& ikm_hex) + { + const auto ikm = hex_decode(ikm_hex); + + if(algo == "X9.31-RNG(TripleDES)") + return new ANSI_X931_RNG(new TripleDES, new Fixed_Output_RNG(ikm)); + else if(algo == "X9.31-RNG(AES-128)") + return new ANSI_X931_RNG(new AES_128, new Fixed_Output_RNG(ikm)); + else if(algo == "X9.31-RNG(AES-192)") + return new ANSI_X931_RNG(new AES_192, new Fixed_Output_RNG(ikm)); + else if(algo == "X9.31-RNG(AES-256)") + return new ANSI_X931_RNG(new AES_256, new Fixed_Output_RNG(ikm)); + + return nullptr; + } + +bool x931_test(const std::string& algo, + const std::string& ikm, + const std::string& out, + size_t L) + { + std::unique_ptr<RandomNumberGenerator> x931(get_x931(algo, ikm)); + x931->reseed(0); + + const std::string got = hex_encode(x931->random_vec(L)); + + if(got != out) + { + std::cout << "X9.31 " << got << " != " << out << "\n"; + return false; + } + + return true; + } + +} + +size_t test_rngs() + { + std::ifstream vec(CHECKS_DIR "/x931.vec"); + + return run_tests_bb(vec, "RNG", "Out", true, + [](std::map<std::string, std::string> m) -> bool + { + return x931_test(m["RNG"], m["IKM"], m["Out"], to_u32bit(m["L"])); + }); + } |