Function generate_proxy_source

src/dll_proxy.rs:18–78

(exports: &[DllExport], forward_target: &str)

}

fn generate_proxy_source(exports: &[DllExport], forward_target: &str) -> String {
    let named: Vec<(usize, &str)> = exports
        .iter()
        .filter_map(|e| e.name.as_deref())
        .enumerate()
        .collect();

    let dll_filename = format!("{}.dll", forward_target);
    let mut s = String::new();

    s.push_str("use std::arch::naked_asm;\n");
    s.push_str("use std::sync::atomic::{AtomicUsize, Ordering};\n\n");

    s.push_str("#[link(name = \"kernel32\")]\n");
    s.push_str("extern \"system\" {\n");
    s.push_str("    #[link_name = \"LoadLibraryA\"]\n");
    s.push_str("    fn rp_load_library(name: *const u8) -> isize;\n");
    s.push_str("    #[link_name = \"GetProcAddress\"]\n");
    s.push_str("    fn rp_get_proc_address(module: isize, name: *const u8) -> usize;\n");
    s.push_str("}\n\n");

    for (i, name) in &named {
        s.push_str(&format!(
            "static RP_ADDR_{}: AtomicUsize = AtomicUsize::new(0); // {}\n",
            i, name
        ));
    }
    if !named.is_empty() {
        s.push('\n');
    }

    s.push_str("pub unsafe fn init() {\n");
    s.push_str(&format!(
        "    let h = rp_load_library(b\"{}\\0\".as_ptr());\n",
        dll_filename
    ));
    s.push_str("    if h == 0 { return; }\n");
    for (i, name) in &named {
        s.push_str(&format!(
            "    RP_ADDR_{}.store(rp_get_proc_address(h, b\"{}\\0\".as_ptr()), Ordering::Release);\n",
            i, name
        ));
    }
    s.push_str("}\n\n");

    for (i, name) in &named {
        s.push_str("#[unsafe(naked)]\n");
        s.push_str(&format!("#[export_name = \"{}\"]\n", name));
        s.push_str(&format!(
            "pub unsafe extern \"system\" fn _rp_fwd_{}() {{\n",
            i
        ));
        s.push_str(&format!(
            "    naked_asm!(\"jmp qword ptr [rip + {{addr}}]\", addr = sym RP_ADDR_{});\n",
            i
        ));
        s.push_str("}\n\n");
    }