From 35caa53c3f185842f3625e5f32d7cb16b6ba646f Mon Sep 17 00:00:00 2001 From: Andrzej Janik Date: Mon, 10 Feb 2020 20:56:09 +0100 Subject: [PATCH] Clean up slightly --- notcuda_redirect/src/lib.rs | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/notcuda_redirect/src/lib.rs b/notcuda_redirect/src/lib.rs index 0eab6ed..49b3bbe 100644 --- a/notcuda_redirect/src/lib.rs +++ b/notcuda_redirect/src/lib.rs @@ -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();