#[cfg(any(target_os = "linux", target_os = "android"))]
fn has_userfaultfd() -> bool {
    let r = unsafe { libc::syscall(libc::SYS_userfaultfd, libc::O_CLOEXEC | libc::O_NONBLOCK) };
    r == 0
}

#[cfg(not(any(target_os = "linux", target_os = "android")))]
fn has_userfaultfd() -> bool {
    false
}

fn main() {
    let mut build = cc::Build::new();
    let prof = std::env::var("PROFILE").expect("there should be a profile");

    println!("cargo:rerun-if-env-changed=TARGET");
    println!("cargo:rerun-if-env-changed=HOST");
    println!("cargo:rerun-if-env-changed=PROFILE");

    build.include("isoalloc/include");
    build.files([
        "isoalloc/src/iso_alloc.c",
        "isoalloc/src/iso_alloc_util.c",
        "isoalloc/src/iso_alloc_interfaces.c",
        "isoalloc/src/iso_alloc_mem_tags.c",
        "isoalloc/src/iso_alloc_signal.c",
        "isoalloc/src/iso_alloc_random.c",
        "isoalloc/src/iso_alloc_profiler.c",
        "isoalloc/src/iso_alloc_sanity.c",
        "isoalloc/src/iso_alloc_search.c",
        "isoalloc/src/iso_alloc_printf.c",
        "isoalloc/src/libc_hook.c",
        "isoalloc/src/malloc_hook.c",
    ]);

    build.define("SANITIZE_CHUNKS", "1");
    build.define("FUZZ_MODE", "0");
    build.define("MALLOC_HOOK", "1");
    build.define("PERM_FREE_REALLOC", "0");
    build.define("ABORT_NO_ENTROPY", "1");
    build.define("USE_SPINLOCK", "0");
    build.define("STARTUP_MEM_USAGE", "0");
    build.define("HUGE_PAGES", "1");
    build.define("AUTO_CTOR_DTOR", "1");
    build.define("SCHED_GETCPU", "1");

    if cfg!(any(target_os = "linux", target_os = "android")) {
        if cfg!(feature = "userfaultfd") && has_userfaultfd() {
            build.define("UNINIT_READ_SANITY", "1");
        }

        // FIXME: might be a temporary setting before strenghing up this feature
        if cfg!(feature = "mte") {
            build.define("ARM_MTE", "1");
            build.define("DISABLE_CANARY", "1");
            build.define("MEMORY_TAGGING", "0");
        } else if cfg!(feature = "memory_tagging") {
            build.define("ARM_MTE", "0");
            build.define("MEMORY_TAGGING", "1");
            build.define("DISABLE_CANARY", "0");
        } else {
            build.define("DISABLE_CANARY", "0");
        }
    } else {
        build.define("DISABLE_CANARY", "0");
    }

    if cfg!(target_arch = "aarch64") {
        // FIXME: might be a temporary setting before strenghing up this feature
        if cfg!(feature = "neon") {
            build.define("DONT_USE_NEON", "0");
        }
    }

    // FIXME: once runtime options are implemented
    // we can remove some of these
    if cfg!(feature = "sanity") {
        build.define("ALLOC_SANITY", "1");
        build.define("MEMCPY_SANITY", "1");
        build.define("MEMSET_SANITY", "1");
        build.define("_FORTIFY_SOURCE", "0");
    }

    if cfg!(feature = "smallmem") {
        build.define("SMALL_MEM_STARTUP", "1");
    }

    if cfg!(all(feature = "tagging", target_arch = "aarch64")) {
        build.define("MEMORY_TAGGING", "1");
    }

    // unfortunately freebsd's libpthread throws off
    // zone allocations, might need a proper wrapper
    if cfg!(not(any(
        target_os = "freebsd",
        target_os = "ios",
        target_os = "watchos",
        target_os = "tvos",
        feature = "nothread"
    ))) {
        build.define("THREAD_SUPPORT", "1");
        build.flag("-pthread");
    }

    if cfg!(target_vendor = "apple") {
        println!("cargo:rustc-link-lib=framework=Security");
    }

    build.flag("-std=c11");
    build.flag("-Wno-pointer-arith");
    build.flag("-fstrict-aliasing");
    build.flag("-Wno-sign-compare");
    build.flag("-Wno-unused-parameter");
    build.flag("-Wno-attributes");
    build.flag_if_supported("-Wno-gnu-zero-variadic-macro-arguments");
    build.flag_if_supported("-Wno-format-pedantic");
    build.flag_if_supported("-Wno-type-limits");
    build.flag_if_supported("-Wno-old-style-declaration");
    match prof.as_str() {
        "debug" => {
            build.define("RANDOMIZE_FREELIST", "1");
            build.debug(true);
        }
        "release" => {
            build.debug(false);
        }
        _ => (),
    }

    build.static_flag(true);
    build.compile("isoalloc");
}