From 40e53586a38532e9509c260ddf622dcd1970a6b7 Mon Sep 17 00:00:00 2001
From: Alex <enderice2@users.noreply.github.com>
Date: Tue, 20 Dec 2022 06:56:18 +0200
Subject: [PATCH] Test the random number generator

---
 Tests/RandomNumberGenerator.cpp | 67 +++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 Tests/RandomNumberGenerator.cpp

diff --git a/Tests/RandomNumberGenerator.cpp b/Tests/RandomNumberGenerator.cpp
new file mode 100644
index 0000000..f1bcd10
--- /dev/null
+++ b/Tests/RandomNumberGenerator.cpp
@@ -0,0 +1,67 @@
+#ifdef DEBUG
+
+#include <rand.hpp>
+#include <debug.h>
+#include <cpu.hpp>
+
+__constructor void TestRandom()
+{
+    int RDRANDFlag = 0;
+    if (strcmp(CPU::Vendor(), x86_CPUID_VENDOR_AMD) == 0)
+    {
+#if defined(__amd64__)
+        CPU::x64::AMD::CPUID0x1 cpuid1amd;
+#elif defined(__i386__)
+        CPU::x32::AMD::CPUID0x1 cpuid1amd;
+#endif
+#if defined(__amd64__) || defined(__i386__)
+        asmv("cpuid"
+             : "=a"(cpuid1amd.EAX.raw), "=b"(cpuid1amd.EBX.raw), "=c"(cpuid1amd.ECX.raw), "=d"(cpuid1amd.EDX.raw)
+             : "a"(0x1));
+#endif
+        RDRANDFlag = cpuid1amd.ECX.RDRAND;
+    }
+    if (strcmp(CPU::Vendor(), x86_CPUID_VENDOR_INTEL) == 0)
+    {
+#if defined(__amd64__)
+        CPU::x64::Intel::CPUID0x1 cpuid1intel;
+#elif defined(__i386__)
+        CPU::x32::Intel::CPUID0x1 cpuid1intel;
+#endif
+#if defined(__amd64__) || defined(__i386__)
+        asmv("cpuid"
+             : "=a"(cpuid1intel.EAX.raw), "=b"(cpuid1intel.EBX.raw), "=c"(cpuid1intel.ECX.raw), "=d"(cpuid1intel.EDX.raw)
+             : "a"(0x1));
+#endif
+        RDRANDFlag = cpuid1intel.ECX.RDRAND;
+    }
+
+    if (strcmp(CPU::Hypervisor(), x86_CPUID_VENDOR_TCG) == 0)
+        RDRANDFlag = 0;
+
+#if defined(__amd64__) || defined(__i386__)
+    if (RDRANDFlag)
+    {
+        uint64_t RDSEEDValue = 0;
+        asmv("1: rdseed %0; jnc 1b"
+             : "=r"(RDSEEDValue));
+        debug("RDSEED: %ld", RDSEEDValue);
+    }
+#endif
+
+    Random::ChangeSeed(0xdeadbeef);
+    uint16_t Seeds16[16];
+    uint32_t Seeds32[16];
+    uint64_t Seeds64[16];
+    for (uint64_t i = 0; i < 16; i++)
+    {
+        Seeds16[i] = Random::rand16();
+        Seeds32[i] = Random::rand32();
+        Seeds64[i] = Random::rand64();
+    }
+    debug("Random 16: %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld", Seeds16[0], Seeds16[1], Seeds16[2], Seeds16[3], Seeds16[4], Seeds16[5], Seeds16[6], Seeds16[7], Seeds16[8], Seeds16[9], Seeds16[10], Seeds16[11], Seeds16[12], Seeds16[13], Seeds16[14], Seeds16[15]);
+    debug("Random 32: %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld", Seeds32[0], Seeds32[1], Seeds32[2], Seeds32[3], Seeds32[4], Seeds32[5], Seeds32[6], Seeds32[7], Seeds32[8], Seeds32[9], Seeds32[10], Seeds32[11], Seeds32[12], Seeds32[13], Seeds32[14], Seeds32[15]);
+    debug("Random 64: %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld, %ld", Seeds64[0], Seeds64[1], Seeds64[2], Seeds64[3], Seeds64[4], Seeds64[5], Seeds64[6], Seeds64[7], Seeds64[8], Seeds64[9], Seeds64[10], Seeds64[11], Seeds64[12], Seeds64[13], Seeds64[14], Seeds64[15]);
+}
+
+#endif