Add stage0_commands config option, bump version to 0.10.0

README.md

@@ -39,6 +39,8 @@ chaudloader has some development options which can be enabled to aid with mod de
 
 -   `developer_mode` (type: boolean, default: `false`): Enables developer mode. Required to be `true` in order to use any of the other development options.
 -   `enable_hook_guards` (type: boolean, default: `false`): Enables hook guards.
+-   `stage0_commands` (type: string array, default: `[]`): List of shell commands to run during stage0 before the game's entry point. If any of the commands returns a nonzero exit code, loading is aborted. The following variables can be used in commands:
+    -   `%PID%`: Replaced by the game's process ID.
 
 ## For developers
 

chaudloader/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "chaudloader"
-version = "0.9.2"
+version = "0.10.0"
 edition = "2021"
 
 [lib]
@@ -33,3 +33,4 @@ maud = "0.25"
 url = "2"
 opener = "0.6"
 proc-macro2 = "^1.0.60"
+run_script = "0.10.1"

chaudloader/src/config.rs

@@ -15,6 +15,7 @@ pub struct Config {
     // Secret options
     pub developer_mode: Option<bool>,
     pub enable_hook_guards: Option<bool>, // requires developer_mode
+    pub stage0_commands: Option<std::collections::BTreeSet<String>>, // requires developer_mode
 }
 
 const CONFIG_FILE_NAME: &str = "chaudloader.toml";

chaudloader/src/hooks/stage0.rs

@@ -224,6 +224,18 @@ pub unsafe fn install() -> Result<(), anyhow::Error> {
 
     let config = config::load()?;
 
+    if config.developer_mode == Some(true) {
+        for cmd in config.stage0_commands.iter().flatten() {
+            let (code, _, _) = run_script::run_script!(
+                cmd
+                    .replace("%PID%", &std::process::id().to_string())
+                )?;
+            if code != 0 {
+                log::error!("Command {cmd} exited with code {code}");
+            }
+        }
+    }
+
     unsafe {
         // Hook GetProcAddress to avoid ntdll.dll hooks being installed
         GetProcAddressForCaller