Random.int added

docs/_docs/user-guide/eldritch.md

@@ -695,6 +695,12 @@ The random library is designed to enable generation of cryptogrphically secure r
 
 The <b>random.bool</b> method returns a randomly sourced boolean value.
 
+### random.int
+
+`random.int(min: i32, max: i32) -> i32`
+
+The <b>random.int</b> method returns randomly generated integer value between a specified range. The range is inclusive on the minimum and exclusive on the maximum.
+
 ### random.string
 
 `random.string(length: uint, charset: Optional<str>) -> str`

implants/lib/eldritch/src/random/int_impl.rs

@@ -0,0 +1,50 @@
+use anyhow::Result;
+use rand::Rng;
+use rand_chacha::rand_core::SeedableRng;
+
+pub fn int(min: i32, max: i32) -> Result<i32> {
+    if min >= max {
+        return Err(anyhow::anyhow!("Invalid range"));
+    }
+    let mut rng = rand_chacha::ChaCha20Rng::from_entropy();
+    Ok(rng.gen_range(min..max))
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    const NUM_ITERATION: i32 = 50000;
+    const MIN_VALUE: i32 = 0;
+    const MAX_VALUE: i32 = 1000;
+
+    #[test]
+    fn test_random_int() -> anyhow::Result<()> {
+        let random_number = int(MIN_VALUE, MAX_VALUE)?;
+        assert!(random_number >= MIN_VALUE && random_number < MAX_VALUE);
+        Ok(())
+    }
+
+    #[test]
+    fn test_random_int_uniform() -> anyhow::Result<()> {
+        let mut counts = vec![0; MAX_VALUE as usize];
+        for _ in 0..NUM_ITERATION {
+            let random_number = int(MIN_VALUE, MAX_VALUE)?;
+            counts[random_number as usize] += 1;
+        }
+
+        let lower_bound = 0.90 * NUM_ITERATION as f64 / MAX_VALUE as f64;
+        let upper_bound = 1.10 * NUM_ITERATION as f64 / MAX_VALUE as f64;
+
+        for count in counts {
+            assert!(
+                count as f64 >= lower_bound && count as f64 <= upper_bound,
+                "Count {} is not within the acceptable bounds of ({},{})",
+                count,
+                lower_bound,
+                upper_bound
+            );
+        }
+        Ok(())
+    }
+}

implants/lib/eldritch/src/random/mod.rs

@@ -1,4 +1,5 @@
 mod bool_impl;
+mod int_impl;
 mod string_impl;
 
 use starlark::{environment::MethodsBuilder, starlark_module, values::starlark_value};
@@ -21,6 +22,11 @@ fn methods(builder: &mut MethodsBuilder) {
         bool_impl::bool()
     }
 
+    #[allow(unused_variables)]
+    fn int<'v>(this: &RandomLibrary, min: i32, max: i32) -> anyhow::Result<i32> {
+        int_impl::int(min, max)
+    }
+
     #[allow(unused_variables)]
     fn string<'v>(this: &RandomLibrary, length: u64, charset: Option<String>) -> anyhow::Result<String> {
         string_impl::string(length, charset)