X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/1375eb51603466b723ab7dd1ca4194ee5d662f75..965940833071025bf0d386f4a9c70a5258453dbd:/lunaix-os/hal/rng/rngx86.c?ds=inline

diff --git a/lunaix-os/hal/rng/rngx86.c b/lunaix-os/hal/rng/rngx86.c
index 5c12ca6..0b8a586 100644
--- a/lunaix-os/hal/rng/rngx86.c
+++ b/lunaix-os/hal/rng/rngx86.c
@@ -1,6 +1,8 @@
 #include <lunaix/device.h>
 #include <lunaix/mm/page.h>
 
+#include <klibc/string.h>
+
 static inline void
 rng_fill(void* data, size_t len)
 {
@@ -24,18 +26,25 @@ __rand_rd_pg(struct device* dev, void* buf, size_t offset)
 static int
 __rand_rd(struct device* dev, void* buf, size_t offset, size_t len)
 {
-    rng_fill(buf, len);
+    if (unlikely(len < 4)) {
+        int tmp_buf = 0;
+        rng_fill(&tmp_buf, 4);
+        memcpy(buf, &tmp_buf, len);
+    } else {
+        rng_fill(buf, len);
+    }
     return len;
 }
 
 int
 pdev_randdev_init(struct device_def* devdef)
 {
+    // FIXME add check on cpuid for presence of rdrand
     struct device* devrand = device_allocseq(NULL, NULL);
     devrand->ops.read = __rand_rd;
     devrand->ops.read_page = __rand_rd_pg;
 
-    device_register(devrand, &devdef->class, "rand");
+    register_device(devrand, &devdef->class, "rand");
 
     return 0;
 }