diff --git a/src/common/mod.rs b/src/common/mod.rs index 60b5aa7..5cb3a48 100644 --- a/src/common/mod.rs +++ b/src/common/mod.rs @@ -27,6 +27,7 @@ pub static mut MENU_STRUCT: consts::TrainingModpackMenu = consts::TrainingModpac pub static mut MENU: &'static mut consts::TrainingModpackMenu = unsafe { &mut MENU_STRUCT }; pub static mut FIGHTER_MANAGER_ADDR: usize = 0; +pub static mut STAGE_MANAGER_ADDR: usize = 0; extern "C" { #[link_name = "\u{1}_ZN3app9smashball16is_training_modeEv"] diff --git a/src/training/mod.rs b/src/training/mod.rs index f8bd29b..8f35e86 100644 --- a/src/training/mod.rs +++ b/src/training/mod.rs @@ -1,4 +1,4 @@ -use crate::common::FIGHTER_MANAGER_ADDR; +use crate::common::{FIGHTER_MANAGER_ADDR, STAGE_MANAGER_ADDR}; use crate::hitbox_visualizer; use skyline::nn::ro::LookupSymbol; use smash::app::{self, lua_bind::*}; @@ -203,6 +203,13 @@ pub fn training_mods() { .as_bytes() .as_ptr(), ); + + LookupSymbol( + &mut STAGE_MANAGER_ADDR, + "_ZN3lib9SingletonIN3app12StageManagerEE9instance_E\u{0}" + .as_bytes() + .as_ptr(), + ); } skyline::install_hooks!(