Make dumper compatible with older versions of CUDA

1 files changed, 31 insertions, 0 deletions

diff --git a/zluda_dump/src/os_win.rs b/zluda_dump/src/os_win.rs
index 7e411ac..70a2b42 100644
--- a/zluda_dump/src/os_win.rs
+++ b/zluda_dump/src/os_win.rs
@@ -5,9 +5,11 @@ use std::{
     ptr,

 };

 

+use std::os::windows::io::AsRawHandle;

 use wchar::wch_c;

 use winapi::{

     shared::minwindef::HMODULE,

+    um::debugapi::OutputDebugStringA,

     um::libloaderapi::{GetProcAddress, LoadLibraryW},

 };

 

@@ -66,3 +68,32 @@ unsafe fn get_non_detoured_load_library(
 pub unsafe fn get_proc_address(handle: *mut c_void, func: &CStr) -> *mut c_void {

     GetProcAddress(handle as *mut _, func.as_ptr()) as *mut _

 }

+

+#[macro_export]

+macro_rules! os_log {

+    ($format:tt) => {

+        {

+            use crate::os::__log_impl;

+            __log_impl(format!($format));

+        }

+    };

+    ($format:tt, $($obj: expr),+) => {

+        {

+            use crate::os::__log_impl;

+            __log_impl(format!($format, $($obj,)+));

+        }

+    };

+}

+

+pub fn __log_impl(s: String) {

+    let log_to_stderr = std::io::stderr().as_raw_handle() != ptr::null_mut();

+    if log_to_stderr {

+        eprintln!("[ZLUDA_DUMP] {}\n", s);

+    } else {

+        let mut win_str = String::with_capacity("[ZLUDA_DUMP] ".len() + s.len() + 2);

+        win_str.push_str("[ZLUDA_DUMP] ");

+        win_str.push_str(&s);

+        win_str.push_str("\n\0");

+        unsafe { OutputDebugStringA(win_str.as_ptr() as *const _) };

+    }

+}