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..88e21f62c
--- /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;
+   }
+
+size_t 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 1;
+      }
+
+   return 0;
+   }
+
+}
+
+size_t test_rngs()
+   {
+   std::ifstream vec(TEST_DATA_DIR "/x931.vec");
+
+   return run_tests_bb(vec, "RNG", "Out", true,
+             [](std::map<std::string, std::string> m) -> size_t
+             {
+             return x931_test(m["RNG"], m["IKM"], m["Out"], to_u32bit(m["L"]));
+             });
+   }