Use QueueUserAPC to run before main

trumank · Nov 3, 2023 · 8b537ee · 8b537ee
1 parent c209759
commit 8b537ee
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/hook/src/lib.rs b/hook/src/lib.rs
@@ -6,11 +6,12 @@ use std::{
 use anyhow::{bail, Context, Result};
 use patternsleuth::{resolvers::resolve_self, scanner::Pattern, PatternConfig};
 use retour::static_detour;
-use windows::{
-    Win32::Foundation::*,
-    Win32::System::{
+use windows::Win32::{
+    Foundation::HMODULE,
+    System::{
         Memory::{VirtualProtect, PAGE_EXECUTE_READWRITE, PAGE_PROTECTION_FLAGS},
         SystemServices::*,
+        Threading::{GetCurrentThread, QueueUserAPC},
     },
 };
 
@@ -36,7 +37,7 @@ extern "system" fn DllMain(dll_module: HMODULE, call_reason: u32, _: *mut ()) ->
     unsafe {
         match call_reason {
             DLL_PROCESS_ATTACH => {
-                std::thread::spawn(|| patch().ok());
+                QueueUserAPC(Some(init), GetCurrentThread(), 0);
             }
             DLL_PROCESS_DETACH => (),
             _ => (),
@@ -69,6 +70,10 @@ impl DRGInstallationType {
     }
 }
 
+unsafe extern "system" fn init(_: usize) {
+    patch().ok();
+}
+
 unsafe fn patch() -> Result<()> {
     let pak_path = std::env::current_exe()
         .ok()