Applied-Software/ZLUDA
5
0
Fork 0
mirror of https://github.com/vosen/ZLUDA.git synced 2025-04-22 01:18:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

Clean up slightly

Browse Source
This commit is contained in:
Andrzej Janik
2020-02-10 20:56:09 +01:00
parent 5ec4c5cf72
commit 35caa53c3f
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
notcuda_redirect/src/lib.rs
View File

@ -40,21 +40,19 @@ unsafe extern "system" fn NotCudaLoadLibraryExW(
#[no_mangle]
unsafe extern "system" fn DllMain(_: *const u8, dwReason: u32, _: *const u8) -> i32 {
if dwReason == DLL_PROCESS_ATTACH {
let mut load_lib_ex = LoadLibraryExW as *mut _;
DetourRestoreAfterWith();
DetourTransactionBegin();
DetourUpdateThread(GetCurrentThread());
DetourAttach(
&mut load_lib_ex,
std::mem::transmute(&mut LOAD_LIBRARY_EX),
NotCudaLoadLibraryExW as *mut _,
);
DetourTransactionCommit();
LOAD_LIBRARY_EX = std::mem::transmute(load_lib_ex);
} else if dwReason == DLL_PROCESS_DETACH {
DetourTransactionBegin();
DetourUpdateThread(GetCurrentThread());
DetourDetach(
&mut (LOAD_LIBRARY_EX as *mut _),
std::mem::transmute(&mut LOAD_LIBRARY_EX),
NotCudaLoadLibraryExW as *mut _,
);
DetourTransactionCommit();