Use secure allocator for RNG state

ElementsProject · Jan 17, 2019 · f2e60ca · f2e60ca
1 parent cddb31b
commit f2e60ca
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/random.cpp b/src/random.cpp
@@ -19,6 +19,8 @@
 #include <chrono>
 #include <thread>
 
+#include <support/allocators/secure.h>
+
 #ifndef WIN32
 #include <fcntl.h>
 #include <sys/time.h>
@@ -351,8 +353,8 @@ RNGState& GetRNGState() noexcept
 {
  // This C++11 idiom relies on the guarantee that static variable are initialized
  // on first call, even when multiple parallel calls are permitted.
- static std::unique_ptr<RNGState> g_rng{new RNGState()};
- return *g_rng;
+ static std::vector<RNGState, secure_allocator<RNGState>> g_rng(1);
+ return g_rng[0];
 }
 }