admin/0xflux-rust_shellcode
1
0
Fork 0
mirror of https://github.com/0xflux/rust_shellcode synced 2026-06-08 10:13:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

tidy up

Browse Source
This commit is contained in:
flux
2024-06-16 19:40:25 +01:00
parent 1610935352
commit 1bb4c08589
2 changed files with 63 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files
shellcode/src/resolver.rs
+8 -8
View File
@@ -58,7 +58,7 @@ pub extern "system" fn get_function_from_exports(dll_base: *const c_void, needle
// cast the dos header
let dos_header = &*(dll_base as *const IMAGE_DOS_HEADER);
if (*dos_header).e_magic != IMAGE_DOS_SIGNATURE {
return 0x12345678 as *const c_void;
return 0x87654321 as *const c_void;
}
// get us to the export directory
@@ -76,14 +76,14 @@ pub extern "system" fn get_function_from_exports(dll_base: *const c_void, needle
// iterate through each export, looking for our chosen
for i in 0..number_of_names {
let name_rva_p: *const DWORD = (dll_base as *const u8).offset((addr_of_names + i * 4) as isize) as *const _;
let name_index_p: *const WORD = (dll_base as *const u8).offset((addr_of_ords + i * 2) as isize) as *const _;
let name_index = name_index_p.as_ref().unwrap();
let mut off: u32 = (4 * name_index) as u32;
off = off + addr_of_funcs;
let func_rva: *const DWORD = (dll_base as *const u8).offset(off as _) as *const _;
let p_name_rva: *const DWORD = (dll_base as *const u8).offset((addr_of_names + i * 4) as isize) as *const _;
let p_name_index: *const WORD = (dll_base as *const u8).offset((addr_of_ords + i * 2) as isize) as *const _;
let name_index = p_name_index.as_ref().unwrap();
let mut offset: u32 = (4 * name_index) as u32;
offset = offset + addr_of_funcs;
let func_rva: *const DWORD = (dll_base as *const u8).offset(offset as _) as *const _;
let name_rva = name_rva_p.as_ref().unwrap();
let name_rva = p_name_rva.as_ref().unwrap();
let curr_name = (dll_base as *const u8).offset(*name_rva as isize);
if *curr_name == 0 {