9 releases

0.1.16 Sep 22, 2023
0.1.15 Sep 21, 2023
0.1.11 Apr 3, 2023

#50 in #forward

Download history 8/week @ 2024-07-01 9/week @ 2024-07-08 10/week @ 2024-07-15 88/week @ 2024-07-22 3/week @ 2024-07-29 7/week @ 2024-09-02 8/week @ 2024-09-09 4/week @ 2024-09-16 37/week @ 2024-09-23 11/week @ 2024-09-30 3/week @ 2024-10-07 8/week @ 2024-10-14

59 downloads per month
Used in forward-dll

MIT license

10KB
92 lines

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 需要通过 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

Dependencies

~2.5MB
~54K SLoC