diff options
| -rw-r--r-- | zluda_dump/src/os_unix.rs | 18 | ||||
| -rw-r--r-- | zluda_dump/src/os_win.rs | 11 |
2 files changed, 25 insertions, 4 deletions
diff --git a/zluda_dump/src/os_unix.rs b/zluda_dump/src/os_unix.rs index affa841..49e1825 100644 --- a/zluda_dump/src/os_unix.rs +++ b/zluda_dump/src/os_unix.rs @@ -1,5 +1,6 @@ use crate::cuda::CUuuid;
use std::ffi::{c_void, CStr};
+use std::mem;
const NVCUDA_DEFAULT_PATH: &'static [u8] = b"/usr/lib/x86_64-linux-gnu/libcuda.so.1\0";
@@ -36,23 +37,34 @@ pub fn get_thunk( guid: *const CUuuid,
idx: usize,
) -> *const c_void {
- use std::mem;
-
use dynasmrt::{dynasm, DynasmApi};
let mut ops = dynasmrt::x86::Assembler::new().unwrap();
let start = ops.offset();
+ // Let's hope there's never more than 6 arguments
dynasm!(ops
; .arch x64
+ ; push rbp
+ ; mov rbp, rsp
; push rdi
; push rsi
+ ; push rdx
+ ; push rcx
+ ; push r8
+ ; push r9
; mov rdi, QWORD guid as i64
; mov rsi, QWORD idx as i64
; mov rax, QWORD report_fn as i64
; call rax
+ ; pop r9
+ ; pop r8
+ ; pop rcx
+ ; pop rdx
; pop rsi
; pop rdi
; mov rax, QWORD original_fn as i64
- ; jmp rax
+ ; call rax
+ ; pop rbp
+ ; ret
; int 3
);
let exe_buf = ops.finalize().unwrap();
diff --git a/zluda_dump/src/os_win.rs b/zluda_dump/src/os_win.rs index 1617aa5..0cd8f3d 100644 --- a/zluda_dump/src/os_win.rs +++ b/zluda_dump/src/os_win.rs @@ -137,18 +137,27 @@ pub fn get_thunk( use dynasmrt::{dynasm, DynasmApi};
let mut ops = dynasmrt::x86::Assembler::new().unwrap();
let start = ops.offset();
+ // Let's hope there's never more than 4 arguments
dynasm!(ops
; .arch x64
+ ; push rbp
+ ; mov rbp, rsp
; push rcx
; push rdx
+ ; push r8
+ ; push r9
; mov rcx, QWORD guid as i64
; mov rdx, QWORD idx as i64
; mov rax, QWORD report_fn as i64
; call rax
+ ; pop r9
+ ; pop r8
; pop rdx
; pop rcx
; mov rax, QWORD original_fn as i64
- ; jmp rax
+ ; call rax
+ ; pop rbp
+ ; ret
; int 3
);
let exe_buf = ops.finalize().unwrap();
|