core: Use same RNG algorithm as avmplus

Author: zjk261

Cargo.lock

@@ -69,6 +69,12 @@ either = "1.15.0"
 chardetng = "0.1.17"
 tracy-client = { version = "0.18.2", optional = true, default-features = false }
 
+[target.'cfg(target_os = "windows")'.dependencies]
+windows-sys = { version = "0.61.2", features = ["Win32_System_Performance"] }
+
+[target.'cfg(target_os = "macos")'.dependencies]
+mach2 = "0.5.0"
+
 [target.'cfg(not(target_family = "wasm"))'.dependencies.futures]
 workspace = true
 

core/Cargo.toml

@@ -18,7 +18,6 @@ use crate::vminterface::Instantiator;
 use crate::{avm_error, avm_warn};
 use gc_arena::{Gc, Mutation};
 use indexmap::IndexMap;
-use rand::Rng;
 use ruffle_macros::istr;
 use std::borrow::Cow;
 use std::cell::Cell;
@@ -1831,7 +1830,7 @@ impl<'a, 'gc> Activation<'a, 'gc> {
         // The max value is clamped to the range [0, 2^31 - 1).
         let max = self.context.avm1.pop().coerce_to_f64(self)? as i32;
         let result = if max > 0 {
-            self.context.rng.random_range(0..max)
+            self.context.rng.generate_random_number() % max
         } else {
             0
         };

core/src/avm1/activation.rs

@@ -3,7 +3,6 @@ use crate::avm1::error::Error;
 use crate::avm1::property_decl::{DeclContext, Declaration};
 use crate::avm1::{Object, Value};
 
-use rand::Rng;
 use std::f64::consts;
 
 macro_rules! wrap_std {
@@ -161,7 +160,7 @@ pub fn random<'gc>(
     // See https://github.com/adobe/avmplus/blob/858d034a3bd3a54d9b70909386435cf4aec81d21/core/MathUtils.cpp#L1731C24-L1731C44
     // This generated a restricted set of 'f64' values, which some SWFs implicitly rely on.
     const MAX_VAL: u32 = 0x7FFFFFFF;
-    let rand = activation.context.rng.random_range(0..MAX_VAL);
+    let rand = activation.context.rng.generate_random_number();
     Ok(((rand as f64) / (MAX_VAL as f64 + 1f64)).into())
 }
 

core/src/avm1/globals/math.rs

@@ -7,7 +7,6 @@ use crate::avm2::parameters::ParametersExt;
 use crate::avm2::value::Value;
 use crate::avm2::{ClassObject, Error};
 use num_traits::ToPrimitive;
-use rand::Rng;
 
 macro_rules! wrap_std {
     ($name:ident, $std:expr) => {
@@ -159,6 +158,6 @@ pub fn random<'gc>(
     // See https://github.com/adobe/avmplus/blob/858d034a3bd3a54d9b70909386435cf4aec81d21/core/MathUtils.cpp#L1731C24-L1731C44
     // This generated a restricted set of 'f64' values, which some SWFs implicitly rely on.
     const MAX_VAL: u32 = 0x7FFFFFFF;
-    let rand = activation.context.rng.random_range(0..MAX_VAL);
+    let rand = activation.context.rng.generate_random_number();
     Ok(((rand as f64) / (MAX_VAL as f64 + 1f64)).into())
 }

core/src/avm2/globals/math.rs

@@ -0,0 +1,98 @@
+#[cfg(windows)]
+use windows_sys::Win32::System::Performance::QueryPerformanceCounter;
+
+#[cfg(target_os = "macos")]
+use mach2::mach_time::mach_absolute_time;
+
+#[cfg(not(any(windows, target_os = "macos")))]
+use crate::locale::get_current_date_time;
+
+// https://github.com/adobe/avmplus/blob/858d034a3bd3a54d9b70909386435cf4aec81d21/core/MathUtils.cpp#L1546
+const C1: i32 = 1376312589;
+const C2: i32 = 789221;
+const C3: i32 = 15731;
+const K_RANDOM_PURE_MAX: i32 = 0x7FFFFFFF;
+
+#[derive(Debug, Clone, Copy)]
+pub struct AvmRng {
+    u_value: u32,
+    u_xor_mask: u32,
+}
+
+impl Default for AvmRng {
+    fn default() -> Self {
+        Self {
+            u_value: 0, // 0 means uninitialized
+            u_xor_mask: 0,
+        }
+    }
+}
+
+impl AvmRng {
+    pub fn init_with_seed(&mut self, seed: u32) {
+        self.u_value = seed;
+        self.u_xor_mask = 0x48000000;
+    }
+
+    fn random_fast_next(&mut self) -> i32 {
+        if (self.u_value & 1) != 0 {
+            self.u_value = (self.u_value >> 1) ^ self.u_xor_mask;
+        } else {
+            self.u_value >>= 1;
+        }
+        self.u_value as i32
+    }
+
+    fn random_pure_hasher(&self, mut i_seed: i32) -> i32 {
+        i_seed = ((i_seed << 13) ^ i_seed).wrapping_sub(i_seed >> 21);
+
+        let mut i_result = i_seed.wrapping_mul(i_seed);
+        i_result = i_result.wrapping_mul(C3);
+        i_result = i_result.wrapping_add(C2);
+        i_result = i_result.wrapping_mul(i_seed);
+        i_result = i_result.wrapping_add(C1);
+        i_result = i_result & K_RANDOM_PURE_MAX;
+
+        i_result = i_result.wrapping_add(i_seed);
+
+        i_result = ((i_result << 13) ^ i_result).wrapping_sub(i_result >> 21);
+
+        i_result
+    }
+
+    pub fn generate_random_number(&mut self) -> i32 {
+        // In avmplus, RNG is initialized on first use.
+        if self.u_value == 0 {
+            let seed = get_seed();
+            self.init_with_seed(seed);
+        }
+
+        let mut a_num = self.random_fast_next();
+
+        a_num = self.random_pure_hasher(a_num.wrapping_mul(71));
+
+        a_num & K_RANDOM_PURE_MAX
+    }
+}
+
+// https://github.com/adobe-flash/avmplus/blob/65a05927767f3735db37823eebf7d743531f5d37/VMPI/WinPortUtils.cpp#L305
+#[cfg(windows)]
+fn get_seed() -> u32 {
+    let mut value: i64 = 0;
+    unsafe {
+        QueryPerformanceCounter(&mut value);
+    }
+    value as u32
+}
+
+// https://github.com/adobe-flash/avmplus/blob/65a05927767f3735db37823eebf7d743531f5d37/VMPI/MacPortUtils.cpp#L31
+#[cfg(target_os = "macos")]
+fn get_seed() -> u32 {
+    unsafe { mach_absolute_time() as u32 }
+}
+
+// https://github.com/adobe-flash/avmplus/blob/65a05927767f3735db37823eebf7d743531f5d37/VMPI/PosixSpecificUtils.cpp#L18
+#[cfg(not(any(windows, target_os = "macos")))]
+fn get_seed() -> u32 {
+    get_current_date_time().timestamp_micros() as u32
+}

core/src/avm_rng.rs

@@ -8,6 +8,7 @@ use crate::avm1::{Object as Avm1Object, Value as Avm1Value};
 use crate::avm2::api_version::ApiVersion;
 use crate::avm2::Activation as Avm2Activation;
 use crate::avm2::{Avm2, LoaderInfoObject, Object as Avm2Object, SoundChannelObject};
+use crate::avm_rng::AvmRng;
 use crate::backend::{
     audio::{AudioBackend, AudioManager, SoundHandle, SoundInstanceHandle},
     log::LogBackend,
@@ -43,7 +44,6 @@ use crate::PlayerMode;
 use async_channel::Sender;
 use core::fmt;
 use gc_arena::{Collect, Mutation};
-use rand::rngs::SmallRng;
 use ruffle_render::backend::{BitmapCacheEntry, RenderBackend};
 use ruffle_render::commands::{CommandHandler, CommandList};
 use ruffle_render::transform::TransformStack;
@@ -119,7 +119,7 @@ pub struct UpdateContext<'gc> {
     pub video: &'gc mut dyn VideoBackend,
 
     /// The RNG, used by the AVM `RandomNumber` opcode, `Math.random(),` and `random()`.
-    pub rng: &'gc mut SmallRng,
+    pub rng: &'gc mut AvmRng,
 
     /// The current player's stage (including all loaded levels)
     pub stage: Stage<'gc>,

core/src/context.rs

@@ -15,6 +15,7 @@ extern crate num_derive;
 #[macro_use]
 mod avm1;
 mod avm2;
+mod avm_rng;
 mod binary_data;
 pub mod bitmap;
 pub mod buffer;

core/src/lib.rs

@@ -38,7 +38,6 @@ use crate::library::Library;
 use crate::limits::ExecutionLimit;
 use crate::loader::{LoadBehavior, LoadManager};
 use crate::local_connection::LocalConnections;
-use crate::locale::get_current_date_time;
 use crate::net_connection::NetConnections;
 use crate::orphan_manager::OrphanManager;
 use crate::prelude::*;
@@ -53,8 +52,8 @@ use crate::vminterface::Instantiator;
 use crate::DefaultFont;
 use async_channel::Sender;
 use gc_arena::lock::GcRefLock;
+use crate::avm_rng::AvmRng;
 use gc_arena::{Collect, DynamicRootSet, Mutation, Rootable};
-use rand::{rngs::SmallRng, SeedableRng};
 use ruffle_macros::istr;
 use ruffle_render::backend::{null::NullRenderer, RenderBackend, ViewportDimensions};
 use ruffle_render::commands::CommandList;
@@ -315,7 +314,7 @@ pub struct Player {
 
     transform_stack: TransformStack,
 
-    rng: SmallRng,
+    rng: AvmRng,
 
     gc_arena: Rc<RefCell<GcArena>>,
 
@@ -2949,7 +2948,7 @@ impl PlayerBuilder {
                 mouse_cursor_needs_check: false,
 
                 // Misc. state
-                rng: SmallRng::seed_from_u64(get_current_date_time().timestamp_millis() as u64),
+                rng: AvmRng::default(),
                 system: SystemProperties::new(language),
                 page_url: self.page_url.clone(),
                 transform_stack: TransformStack::new(),