From 72baf48863cffec16e41c281c7c4835fed79cced Mon Sep 17 00:00:00 2001
From: pedrocarlo <pedro.muniz.carlo@gmail.com>
Date: Tue, 21 Oct 2025 11:44:40 -0300
Subject: [PATCH] add random generation in simulator IO

---
 simulator/runner/io.rs        | 8 ++++++--
 simulator/runner/memory/io.rs | 8 ++++++--
 whopper/io.rs                 | 5 +++++
 3 files changed, 17 insertions(+), 4 deletions(-)

diff --git a/simulator/runner/io.rs b/simulator/runner/io.rs
index c5c38f928..baf4d9e98 100644
--- a/simulator/runner/io.rs
+++ b/simulator/runner/io.rs
@@ -3,7 +3,7 @@ use std::{
     sync::Arc,
 };
 
-use rand::{RngCore, SeedableRng};
+use rand::{Rng, RngCore, SeedableRng};
 use rand_chacha::ChaCha8Rng;
 use turso_core::{Clock, IO, Instant, OpenFlags, PlatformIO, Result};
 
@@ -136,6 +136,10 @@ impl IO for SimulatorIO {
     }
 
     fn generate_random_number(&self) -> i64 {
-        self.rng.borrow_mut().next_u64() as i64
+        self.rng.borrow_mut().random()
+    }
+
+    fn fill_bytes(&self, dest: &mut [u8]) {
+        self.rng.borrow_mut().fill_bytes(dest);
     }
 }
diff --git a/simulator/runner/memory/io.rs b/simulator/runner/memory/io.rs
index 975f0d7ce..fc406e7c1 100644
--- a/simulator/runner/memory/io.rs
+++ b/simulator/runner/memory/io.rs
@@ -3,7 +3,7 @@ use std::sync::Arc;
 
 use indexmap::IndexMap;
 use parking_lot::Mutex;
-use rand::{RngCore, SeedableRng};
+use rand::{Rng, RngCore, SeedableRng};
 use rand_chacha::ChaCha8Rng;
 use turso_core::{Clock, Completion, IO, Instant, OpenFlags, Result};
 
@@ -269,7 +269,11 @@ impl IO for MemorySimIO {
     }
 
     fn generate_random_number(&self) -> i64 {
-        self.rng.borrow_mut().next_u64() as i64
+        self.rng.borrow_mut().random()
+    }
+
+    fn fill_bytes(&self, dest: &mut [u8]) {
+        self.rng.borrow_mut().fill_bytes(dest);
     }
 
     fn remove_file(&self, path: &str) -> Result<()> {
diff --git a/whopper/io.rs b/whopper/io.rs
index 5b9da7b3e..9f8c8a872 100644
--- a/whopper/io.rs
+++ b/whopper/io.rs
@@ -142,6 +142,11 @@ impl IO for SimulatorIO {
         let mut rng = self.rng.lock().unwrap();
         rng.next_u64() as i64
     }
+
+    fn fill_bytes(&self, dest: &mut [u8]) {
+        let mut rng = self.rng.lock().unwrap();
+        rng.fill_bytes(dest);
+    }
 }
 
 const MAX_FILE_SIZE: usize = 1 << 33; // 8 GiB