bsalign_sys 0.1.0

Rust bindings for the bsalign C library
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

fn complie() {
    println!("cargo:rustc-link-lib=z");
    println!("cargo:rustc-link-lib=pthread");
    println!("cargo:rustc-link-lib=m");
    println!("cargo:rustc-link-lib=rt");

    let debug = std::env::var("CARGO_CFG_DEBUG").unwrap_or_else(|_| "false".to_string()) == "true";

    let mut b = cc::Build::new();
    b.file("./wrapper.c")
        .include("./")
        .flag("-std=gnu89")
        .flag("-Wall")
        .flag("-Wextra")
        .flag("-Wno-unused-parameter")
        .flag("-Wno-unused-function")
        .flag("-Wno-unused-variable")
        .flag("-Wno-unused-but-set-variable")
        .flag("-DNDEBUG")
        .flag("-D_GNU_SOURCE")
        .flag("-D_FILE_OFFSET_BITS=64")
        .flag("-mpopcnt")
        .flag("-DVERSION=\"1.2.1\"")
        .static_flag(true)
        .shared_flag(true)
        .warnings(false);

    if debug {
        b.debug(true);
        b.opt_level(0);
    } else {
        b.debug(false);
        b.opt_level(3);
    }

    let target = std::env::var("TARGET").unwrap();
    if target.contains("x86_64") {
        if target.contains("aarch64") || target.contains("arm") {
            // raise error
            panic!("bsalign unsupport aarch64/arm cpu arch.")
        }
        b.flag("-msse4.2");
    } else {
        // raise error
        panic!("bsalign unsupport non-x86_64 cpu arch.")
    }

    b.compile("bsalign");
}

fn check_bsalign_repo() {
    let project_dir = std::env::var("CARGO_MANIFEST_DIR").unwrap();
    // check if bsalign repo is exist
    let bsalign_header_path = format!("{}/bsalign/bsalign.h", project_dir);
    if std::path::Path::new(&bsalign_header_path).exists() {
        return;
    }
    let status = std::process::Command::new("git")
        .args(&["submodule", "update", "--init", "--recursive"])
        .current_dir(&project_dir)
        .status()
        .expect("Failed to update submodule");
    if !status.success() {
        panic!("Failed to update submodule");
    }
    println!("cargo:rerun-if-changed=.gitmodules");
}

fn generate() {
    bindgen::Builder::default()
        .header("./wrapper.h")
        .clang_arg("-std=gnu89")
        .blocklist_var("FP_NAN")
        .blocklist_var("FP_INFINITE")
        .blocklist_var("FP_ZERO")
        .blocklist_var("FP_SUBNORMAL")
        .blocklist_var("FP_NORMAL")
        .derive_default(true)
        .no_default("BSPOAPar")
        .derive_debug(true)
        .wrap_static_fns(false)
        .wrap_unsafe_ops(true)
        .fit_macro_constants(true)
        .dynamic_link_require_all(true)
        .disable_header_comment()
        .generate()
        .expect("Unable to generate bindings")
        .write_to_file("src/bindings.rs")
        .expect("Couldn't write bindings!");
}

fn main() {
    println!("cargo:rerun-if-changed=wrapper.h");
    check_bsalign_repo();
    complie();
    generate();
}