Function apply_proxy

src/puzzle.rs:241–284

(order: &Order, folder: &Path)

}

fn apply_proxy(order: &Order, folder: &Path) {
    let proxy_path = order.proxy_dll.as_ref().unwrap();
    let exports = pe_parser::parse_exports(proxy_path).unwrap_or_else(|e| {
        eprintln!("[-] Failed to parse proxy DLL exports: {}", e);
        exit(1);
    });

    if exports.is_empty() {
        eprintln!("[-] Warning: proxy DLL has no exports");
    }

    let stem = pe_parser::dll_stem(proxy_path);
    let proxy_output = dll_proxy::generate_proxy(&exports, &stem);

    let src_dir = folder.join("src");
    fs::write(src_dir.join("proxy.rs"), &proxy_output.proxy_source)
        .expect("Failed to write proxy.rs");

    let lib_rs_path = src_dir.join("lib.rs");
    let existing = fs::read_to_string(&lib_rs_path).expect("Failed to read lib.rs");

    // Insert `mod proxy;` after inner attributes (#![...]) to avoid breaking them
    let mut inner_attr_end = 0;
    for line in existing.lines() {
        let trimmed = line.trim();
        if trimmed.starts_with("#!") || trimmed.is_empty() {
            inner_attr_end += line.len() + 1; // +1 for newline
        } else {
            break;
        }
    }
    let updated = format!(
        "{}\n#[allow(non_upper_case_globals, non_snake_case)]\nmod proxy;\n{}",
        &existing[..inner_attr_end.min(existing.len())].trim_end(),
        &existing[inner_attr_end.min(existing.len())..]
    );
    fs::write(&lib_rs_path, updated).expect("Failed to update lib.rs with mod proxy");

    println!(
        "[+] DLL proxying: {} exports forwarded. Rename the original DLL to '{}'",
        exports.len(),
        proxy_output.original_dll_name
    );
}

pub fn assemble(order: Order) -> PathBuf {
    println!("[+] Assembling Rust code..");