FastDivert/build.rs at master · one-api/FastDivert · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
/*
 * Copyright (c) 2026 github.com/one-api. All rights reserved.
 * Licensed under AGPLv3 (https://www.gnu.org/licenses/agpl-3.0.html) or a commercial license.
 * See: https://github.com/one-api/FastDivert#license
 */

use bindgen;
use std::env;
use std::path::PathBuf;

fn main() -> Result<(), wdk_build::ConfigError> {
    let sdk_version = "10.0.26100.0";
    let include_base = format!(
        r"C:\Program Files (x86)\Windows Kits\10\Include\{}",
        sdk_version
    );

    // 1. Build NDIS/WFP Bindings
    let ndis_bindings = bindgen::Builder::default()
        .header("src/wdk_ext/wrapper.h")
        .use_core()
        .clang_arg("-D_AMD64_")
        .clang_arg("-DNTDDI_VERSION=0x0A000000")
        .clang_arg("-D_WIN32_WINNT=0x0A00")
        .clang_arg("-D_KERNEL_MODE")
        .clang_arg("-fms-extensions")
        .clang_arg("-fms-compatibility")
        .clang_arg("-Wno-microsoft-enum-forward-reference")
        .clang_arg(format!("-I{}/km", include_base))
        .clang_arg(format!("-I{}/km/ndis", include_base))
        .clang_arg(format!("-I{}/shared", include_base))
        .clang_arg(format!("-I{}/um", include_base))
        .clang_arg(format!("-I{}/ndis", include_base))
        .clang_args(&["-target", "x86_64-pc-windows-msvc"])
        .allowlist_function("Ndis.*")
        .allowlist_function("Fwps.*")
        .allowlist_function("Fwpm.*")
        .allowlist_type("FWPM_CALLOUT.*")
        .allowlist_type("FWPS_.*")
        .allowlist_type("FWPM.*")
        .allowlist_type("MDL")
        .allowlist_type("PMDL")
        .allowlist_var("FWPS_.*")
        .allowlist_var("FWPM.*")
        .allowlist_var("NET_.*")
        .allowlist_var("NDIS_.*")
        .allowlist_var("FWP_.*")
        .blocklist_type("GUID")
        .blocklist_type("HANDLE")
        .blocklist_type("PMDL")
        .manually_drop_union(".*")
        .layout_tests(false)
        .derive_default(true)
        .generate()
        .expect("failed to generate NDIS bindings");

    // 2. Build Minifilter (FltMgr) Bindings
    let flt_bindings = bindgen::Builder::default()
        .header("src/wdk_ext/flt_wrapper.h")
        .use_core()
        .clang_arg("-D_AMD64_")
        .clang_arg("-DNTDDI_VERSION=0x0A000000")
        .clang_arg("-D_WIN32_WINNT=0x0A00")
        .clang_arg("-D_KERNEL_MODE")
        .clang_arg("-fms-extensions")
        .clang_arg("-fms-compatibility")
        .clang_arg("-Wno-microsoft-enum-forward-reference")
        .clang_arg(format!("-I{}/km", include_base))
        .clang_arg(format!("-I{}/shared", include_base))
        .clang_arg(format!("-I{}/um", include_base))
        .clang_args(&["-target", "x86_64-pc-windows-msvc"])
        .allowlist_function("Flt.*")
        .allowlist_type("FLT_.*")
        .allowlist_type("PFLT_.*")
        .allowlist_var("FLT_.*")
        .blocklist_type("GUID")
        .blocklist_type("HANDLE")
        .blocklist_type("PMDL")
        .layout_tests(false)
        .derive_default(true)
        .generate()
        .expect("failed to generate Minifilter bindings");

    let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
    ndis_bindings
        .write_to_file(out_path.join("ndis_bindings.rs"))
        .ok();
    flt_bindings
        .write_to_file(out_path.join("flt_bindings.rs"))
        .ok();

    println!("cargo:rustc-link-lib=ntoskrnl");
    println!("cargo:rustc-link-lib=fwpkclnt");
    println!("cargo:rustc-link-lib=ndis");
    println!("cargo:rustc-link-lib=uuid");
    println!("cargo:rustc-link-lib=wdm");
    println!("cargo:rustc-link-lib=FltMgr");

    wdk_build::configure_wdk_binary_build()
}