Skip to content

Latest commit

 

History

History
133 lines (106 loc) · 3.44 KB

README.md

File metadata and controls

133 lines (106 loc) · 3.44 KB
Raw

forward-dll

生成 DLL 转发的函数。

使用方法——静态转发

先在 Cargo.toml 中添加依赖：

[build-dependencies]
forward-dll = "0.1.11"

在 build.rs 中添加如下代码：

use forward_dll::forward_dll;

fn main() {
  forward_dll("C:\\Windows\\System32\\version.dll").unwrap();
}

使用方法——自定义静态转发

use forward_dll::forward_dll_with_exports;
forward_dll_with_exports(
    "C:\\Windows\\system32\\version.dll",
    &[
        (1, "GetFileVersionInfoA"),
        (2, "GetFileVersionInfoByHandle"),
        (3, "GetFileVersionInfoExA"),
        (4, "GetFileVersionInfoExW"),
        (5, "GetFileVersionInfoSizeA"),
        (6, "GetFileVersionInfoSizeExA"),
        (7, "GetFileVersionInfoSizeExW"),
        (8, "GetFileVersionInfoSizeW"),
        (9, "GetFileVersionInfoW"),
        (10, "VerFindFileA"),
        (11, "VerFindFileW"),
        (12, "VerInstallFileA"),
        (13, "VerInstallFileW"),
        (14, "VerLanguageNameA"),
        (15, "VerLanguageNameW"),
        (16, "VerQueryValueA"),
        (17, "VerQueryValueW"),
    ],
)
.unwrap();

使用方法——动态转发

use forward_dll::ForwardModule;

#[derive(ForwardModule)]
#[forward(target = "C:\\Windows\\system32\\version.dll")]
pub struct VersionModule;

const VERSION_LIB: VersionModule = VersionModule;

#[no_mangle]
pub extern "system" fn DllMain(_inst: isize, reason: u32, _: *const u8) -> u32 {
    if reason == 1 {
        println!("==> version.dll loaded");
        VERSION_LIB.init().unwrap();
        println!("==> version.dll initialized");
    }
    1
}

注意，#[forward(target = "path/of/your/dll")] 中的路径，应在编译期可以访问到（过程宏会读取这个文件并提取出导出表），如果这个路径为相对路径，则应相对于 Cargo.toml 所在的目录。

限制

  • 动态转发不支持设置 ordinal，更不支持转发仅导出 ordinal 的符号。

    这是因为 Rust 目前不支持设置导出 ordinal，如果需要仅导出 ordinal 或设置 ordinal 需要通过 build.rs 修改链接参数来实现，而这个如何与过程宏协作是较为麻烦的问题。如果有这方面需要可以探讨下怎么实现。

v0.1.5 及之前版本的使用方法

forward_dll::forward_dll!(
  "C:\\Windows\\system32\\version.dll",
  DLL_VERSION_FORWARDER,
  GetFileVersionInfoA
  GetFileVersionInfoByHandle
  GetFileVersionInfoExA
  GetFileVersionInfoExW
  GetFileVersionInfoSizeA
  GetFileVersionInfoSizeExA
  GetFileVersionInfoSizeExW
  GetFileVersionInfoSizeW
  GetFileVersionInfoW
  VerFindFileA
  VerFindFileW
  VerInstallFileA
  VerInstallFileW
  VerLanguageNameA
  VerLanguageNameW
  VerQueryValueA
  VerQueryValueW
);

#[no_mangle]
pub extern "system" fn DllMain(_inst: isize, reason: u32, _: *const u8) -> u32 {
  if reason == 1 {
    // 这里要自行持有底层的 version.dll 的句柄，防止被释放。
    let _ = forward_dll::utils::load_library("C:\\Windows\\system32\\version.dll");
    // 调用 forward_all 方法，建立导出函数与目标函数之间的映射关系。
    let _ = unsafe { DLL_VERSION_FORWARDER.forward_all() };
  }
  1
}

运行 example

警告：运行的时候会发出声音（调用了 winmm.dll 中的 PlaySoundW）。

cargo build -p version -p winmm
cargo run -p just-call-version

License

MIT

Copyright (c) 2022-present, hamflx