1 files changed, 106 insertions, 0 deletions
diff --git a/zluda_inject/src/bin.rs b/zluda_inject/src/bin.rs
new file mode 100644
index 0000000..e021a41
--- /dev/null
+++ b/zluda_inject/src/bin.rs
@@ -0,0 +1,106 @@
+extern crate clap;
+#[macro_use]
+extern crate guid;
+extern crate detours_sys;
+extern crate winapi;
+
+use std::error::Error;
+use std::ffi::OsStr;
+use std::mem;
+use std::os::windows::ffi::OsStrExt;
+use std::ptr;
+
+use winapi::um::processthreadsapi::{GetExitCodeProcess, ResumeThread};
+use winapi::um::synchapi::WaitForSingleObject;
+use winapi::um::winbase::{INFINITE, WAIT_FAILED};
+
+use clap::{App, AppSettings, Arg};
+
+#[macro_use]
+mod win;
+
+fn main() -> Result<(), Box<dyn Error>> {
+    let matches = App::new("ZLUDA injector")
+        .setting(AppSettings::TrailingVarArg)
+        .arg(
+            Arg::with_name("EXE")
+                .help("Path to the executable to be injected with ZLUDA")
+                .required(true),
+        )
+        .arg(
+            Arg::with_name("ARGS")
+                .multiple(true)
+                .help("Arguments that will be passed to <EXE>"),
+        )
+        .get_matches();
+    let exe = matches.value_of_os("EXE").unwrap();
+    let args: Vec<&OsStr> = matches
+        .values_of_os("ARGS")
+        .map(|x| x.collect())
+        .unwrap_or_else(|| Vec::new());
+    let mut cmd_line = Vec::<u16>::with_capacity(exe.len() + 2);
+    cmd_line.push('\"' as u16);
+    copy_to(exe, &mut cmd_line);
+    cmd_line.push('\"' as u16);
+    cmd_line.push(' ' as u16);
+    args.split_last().map(|(last_arg, args)| {
+        for arg in args {
+            cmd_line.reserve(arg.len());
+            copy_to(arg, &mut cmd_line);
+            cmd_line.push(' ' as u16);
+        }
+        copy_to(last_arg, &mut cmd_line);
+    });
+
+    cmd_line.push(0);
+    let mut startup_info = unsafe { mem::zeroed::<detours_sys::_STARTUPINFOW>() };
+    let mut proc_info = unsafe { mem::zeroed::<detours_sys::_PROCESS_INFORMATION>() };
+    os_call!(
+        detours_sys::DetourCreateProcessWithDllExW(
+            ptr::null(),
+            cmd_line.as_mut_ptr(),
+            ptr::null_mut(),
+            ptr::null_mut(),
+            0,
+            0,
+            ptr::null_mut(),
+            ptr::null(),
+            &mut startup_info as *mut _,
+            &mut proc_info as *mut _,
+            "zluda_redirect.dll\0".as_ptr() as *const i8,
+            Option::None
+        ),
+        |x| x != 0
+    );
+    let mut exe_path = std::env::current_dir()?
+        .as_os_str()
+        .encode_wide()
+        .collect::<Vec<_>>();
+    let guid = guid! {"C225FC0C-00D7-40B8-935A-7E342A9344C1"};
+    os_call!(
+        detours_sys::DetourCopyPayloadToProcess(
+            proc_info.hProcess,
+            mem::transmute(&guid),
+            exe_path.as_mut_ptr() as *mut _,
+            (exe_path.len() * mem::size_of::<u16>()) as u32
+        ),
+        |x| x != 0
+    );
+    os_call!(ResumeThread(proc_info.hThread), |x| x as i32 != -1);
+    os_call!(WaitForSingleObject(proc_info.hProcess, INFINITE), |x| x
+        != WAIT_FAILED);
+    let mut child_exit_code: u32 = 0;
+    os_call!(
+        GetExitCodeProcess(proc_info.hProcess, &mut child_exit_code as *mut _),
+        |x| x != 0
+    );
+    std::process::exit(child_exit_code as i32)
+}
+
+fn copy_to(from: &OsStr, to: &mut Vec<u16>) {
+    for x in from.encode_wide() {
+        to.push(x);
+    }
+}
+
+//