use std::env;
use std::path::Path;
fn main() {
let mut build = cc::Build::new();
let version = if env::var("CARGO_FEATURE_V3").is_ok() {
"v3"
} else {
"v2"
};
let cargo_manifest_dir = env::var("CARGO_MANIFEST_DIR").expect("CARGO_MANIFEST_DIR not set");
let include_dir = Path::new(&cargo_manifest_dir)
.join("c_src/mimalloc/")
.join(version)
.join("include")
.to_str()
.expect("include path is not valid UTF-8")
.to_string();
println!("cargo:INCLUDE_DIR={include_dir}");
build.include(format!("c_src/mimalloc/{version}/include"));
build.include(format!("c_src/mimalloc/{version}/src"));
build.file(format!("c_src/mimalloc/{version}/src/static.c"));
let target_os = env::var("CARGO_CFG_TARGET_OS").expect("target_os not defined!");
let target_family = env::var("CARGO_CFG_TARGET_FAMILY").expect("target_family not defined!");
let target_vendor = env::var("CARGO_CFG_TARGET_VENDOR").expect("target_vendor not defined!");
let target_arch = env::var("CARGO_CFG_TARGET_ARCH").expect("target_arch not defined!");
if target_family != "windows" {
build.flag("-Wno-error=date-time");
}
if env::var_os("CARGO_FEATURE_OVERRIDE").is_some() {
if target_family != "windows" {
build.define("MI_MALLOC_OVERRIDE", None);
}
if target_vendor == "apple" {
build.define("MI_OSX_ZONE", Some("1"));
build.define("MI_OSX_INTERPOSE", Some("1"));
}
}
if env::var_os("CARGO_FEATURE_SECURE").is_some() {
build.define("MI_SECURE", "4");
}
let dynamic_tls = env::var("CARGO_FEATURE_LOCAL_DYNAMIC_TLS").is_ok();
if target_family == "unix" && target_os != "haiku" {
if dynamic_tls {
build.flag_if_supported("-ftls-model=local-dynamic");
} else {
build.flag_if_supported("-ftls-model=initial-exec");
}
}
if (target_os == "linux" || target_os == "android")
&& env::var_os("CARGO_FEATURE_NO_THP").is_some()
{
build.define("MI_NO_THP", "1");
}
if env::var_os("CARGO_FEATURE_DEBUG").is_some()
|| (env::var_os("CARGO_FEATURE_DEBUG_IN_DEBUG").is_some() && cfg!(debug_assertions))
{
build.define("MI_DEBUG", "3");
build.define("MI_SHOW_ERRORS", "1");
} else {
build.define("MI_DEBUG", "0");
}
if build.get_compiler().is_like_msvc() {
build.cpp(true);
}
build.compile("mimalloc");
if target_os == "linux" && target_arch == "arm" {
let atomic_name = env::var("DEP_ATOMIC").unwrap_or("atomic".to_owned());
println!("cargo:rustc-link-lib={atomic_name}");
}
if target_os == "windows" {
let libs = ["psapi", "shell32", "user32", "advapi32", "bcrypt"];
for lib in libs {
println!("cargo:rustc-link-lib={lib}");
}
}
}