windows_win/raw/
module.rs

1//! Provides functions to interact with modules.
2
3use core::ptr;
4
5use crate::sys::*;
6use crate::utils::{self, Result};
7
8#[macro_export]
9///Converts ident to module address.
10///
11///Mostly to be used with module's functions.
12macro_rules! module_to_addr {
13    ($mod:ident) => (&$mod as *const _ as *const u16)
14}
15
16///Retrieves module handle by using address inside.
17///
18///Underhood it uses flags `GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS | GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT`
19///Due to that produced HMODULE must not be passed to `FreeLibrary`
20///
21///Use macro `module_to_addr!` to convert local function into module address.
22pub fn get_module_handle_from_addr(module_addr: LPCWSTR) -> Result<HMODULE> {
23    //GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS | GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT
24    const FLAGS: DWORD = 0x00000004 | 0x00000002;
25    let mut result: HMODULE = ptr::null_mut();
26
27    unsafe {
28        if GetModuleHandleExW(FLAGS, module_addr, &mut result as *mut HMODULE) == 0 {
29            return Err(utils::get_last_error());
30        }
31    }
32
33    Ok(result)
34}
35
36///Retrieves file name of module
37pub fn get_module_name(module: HMODULE) -> Result<String> {
38    let buf_len = MAX_PATH as u32;
39    let mut result: Vec<u16> = vec![0; buf_len as usize];
40    let text_ptr = result.as_mut_ptr() as LPWSTR;
41
42    let buf_len = unsafe { GetModuleFileNameW(module, text_ptr, buf_len) };
43
44    if buf_len == 0 {
45        Err(utils::get_last_error())
46    }
47    else {
48        Ok(String::from_utf16_lossy(&result[..buf_len as usize]))
49    }
50}
windows_win/raw/module.rs

windows_win/raw/
module.rs
