Initial commit: SUB-OS 64-Bit Production Monolithic Operating System with LazyBox userland suite

crypto/prng.c

@@ -0,0 +1,43 @@
+#include <crypto/crypto.h>
+#include <arch/x86_64/pit.h>
+
+static uint64_t prng_s[2] = { 0x8a5cd789635d2dffULL, 0x121fd2155c472f96ULL };
+
+void prng_seed(uint64_t seed1, uint64_t seed2) {
+    if (seed1 == 0 && seed2 == 0) {
+        seed1 = 0x8a5cd789635d2dffULL ^ pit_get_ticks();
+        seed2 = 0x121fd2155c472f96ULL ^ (pit_get_ticks() << 32);
+    }
+    prng_s[0] = seed1;
+    prng_s[1] = seed2;
+}
+
+uint64_t prng_rand64(void) {
+    uint64_t s1 = prng_s[0];
+    const uint64_t s0 = prng_s[1];
+    prng_s[0] = s0;
+    s1 ^= s1 << 23;
+    prng_s[1] = s1 ^ s0 ^ (s1 >> 17) ^ (s0 >> 26);
+    return prng_s[1] + s0;
+}
+
+uint32_t prng_rand32(void) {
+    return (uint32_t)(prng_rand64() & 0xFFFFFFFF);
+}
+
+void prng_get_bytes(void* buf, size_t len) {
+    uint8_t* p = (uint8_t*)buf;
+    while (len >= 8) {
+        uint64_t r = prng_rand64();
+        *(uint64_t*)p = r;
+        p += 8;
+        len -= 8;
+    }
+    if (len > 0) {
+        uint64_t r = prng_rand64();
+        uint8_t* r_bytes = (uint8_t*)&r;
+        for (size_t i = 0; i < len; i++) {
+            p[i] = r_bytes[i];
+        }
+    }
+}