Initial commit

arm9/source/rng/ThreadSafeRandomGenerator.h

@@ -0,0 +1,41 @@
+#pragma once
+#include <memory>
+#include <libtwl/rtos/rtosMutex.h>
+#include "RandomGenerator.h"
+
+template <class TGenerator>
+class ThreadSafeRandomGenerator : public RandomGenerator
+{
+public:
+    template <typename ...Args>
+    explicit ThreadSafeRandomGenerator(Args && ...args)
+        : _generator(std::forward<Args>(args)...)
+    {
+        rtos_createMutex(&_mutex);
+    }
+
+    void Seed(u64 seed) override
+    {
+        rtos_lockMutex(&_mutex);
+        {
+            _generator.Seed(seed);
+        }
+        rtos_unlockMutex(&_mutex);
+    }
+
+protected:
+    u32 Next() override
+    {
+        u32 result;
+        rtos_lockMutex(&_mutex);
+        {
+            result = ((RandomGenerator&)_generator).Next();
+        }
+        rtos_unlockMutex(&_mutex);
+        return result;
+    }
+
+private:
+    rtos_mutex_t _mutex;
+    TGenerator _generator;
+};