msvc

Author: Brooooooklyn

Cargo.toml

@@ -25,6 +25,7 @@ travis-ci = { repository = "purpleprotocol/mimalloc_rust" }
 libmimalloc-sys2 = { path = "libmimalloc-sys", version = "0.1.47", default-features = false }
 
 [features]
+asm = ["libmimalloc-sys2/asm"]
 default = []
 secure = ["libmimalloc-sys2/secure"]
 override = ["libmimalloc-sys2/override"]

libmimalloc-sys/Cargo.toml

@@ -22,12 +22,13 @@ cty = { version = "0.2", optional = true }
 libc = "0.2"
 
 [build-dependencies]
+cc = "1.0"
 cmake = "0.1"
 
 [features]
 # For debug purposes
 etw = []
-
+asm = []
 secure = []
 override = []
 extended = ["cty"]

libmimalloc-sys/build.rs

@@ -19,12 +19,21 @@ fn main() {
     let target_env = env::var("CARGO_CFG_TARGET_ENV").expect("target_env not defined!");
     let profile = env::var("PROFILE").expect("profile not defined!");
 
+    if target_os == "windows" && target_env == "msvc" {
+        build_mimalloc_win();
+        return;
+    }
+
     if env::var_os("CARGO_FEATURE_OVERRIDE").is_some() {
         cmake_config.define("MI_OVERRIDE", "ON");
     } else {
         cmake_config.define("MI_OVERRIDE", "OFF");
     }
 
+    if env::var_os("CARGO_FEATURE_ASM").is_some() {
+        cmake_config.define("MI_SEE_ASM", "ON");
+    }
+
     if env::var_os("CARGO_FEATURE_SKIP_COLLECT_ON_EXIT").is_some() {
         cmake_config.define("MI_SKIP_COLLECT_ON_EXIT", "ON");
     }
@@ -42,8 +51,7 @@ fn main() {
         cmake_config.define("MI_OPT_ARCH", "OFF");
     }
 
-    // it's complicated to link ucrt in debug mode on windows
-    if profile == "debug" && target_env != "msvc" {
+    if profile == "debug" {
         cmake_config
             .define("MI_DEBUG_FULL", "ON")
             .define("MI_SHOW_ERRORS", "ON");
@@ -83,35 +91,9 @@ fn main() {
         cmake_config.define("MI_DEBUG_FULL", "OFF");
     }
 
-    if target_env == "msvc" {
-        cmake_config
-            .define("MI_USE_CXX", "ON")
-            // always turn off debug full and show errors on msvc
-            .define("MI_DEBUG_FULL", "OFF")
-            .define("MI_SHOW_ERRORS", "OFF")
-            .profile("release")
-            .static_crt(false);
-        // Link with libs needed on Windows
-        // https://github.com/microsoft/mimalloc/blob/af21001f7a65eafb8fb16460b018ebf9d75e2ad8/CMakeLists.txt#L487
-        // https://github.com/rust-lang/rust/issues/139352
-        let libs = ["psapi", "shell32", "user32", "advapi32", "bcrypt"];
-
-        for lib in libs {
-            println!("cargo:rustc-link-lib={}", lib);
-        }
-    }
-
     let dst = cmake_config.build();
 
-    if target_os == "windows" {
-        println!(
-            "cargo:rustc-link-search=native={}/build/Release",
-            dst.display(),
-        );
-        println!("cargo:rustc-link-search=native={}/build", dst.display());
-    } else {
-        println!("cargo:rustc-link-search=native={}/build", dst.display());
-    }
+    println!("cargo:rustc-link-search=native={}/build", dst.display());
 
     println!("cargo:rustc-link-lib=static={}", mimalloc_base_name);
 
@@ -124,3 +106,55 @@ fn main() {
         println!("cargo:rustc-link-lib={}", atomic_name);
     }
 }
+
+fn build_mimalloc_win() {
+    use std::env;
+
+    let mut build = cc::Build::new();
+    let target_arch = env::var("CARGO_CFG_TARGET_ARCH").expect("target_arch not defined!");
+
+    build
+        .include("./c_src/mimalloc/include")
+        .include("./c_src/mimalloc/src")
+        .file("./c_src/mimalloc/src/static.c")
+        .define("MI_BUILD_SHARED", "0")
+        .cpp(false)
+        .warnings(false)
+        .flag_if_supported("-w");
+
+    if env::var_os("CARGO_FEATURE_SECURE").is_some() {
+        build.define("MI_SECURE", "4");
+    }
+
+    if env::var_os("CARGO_FEATURE_ASM").is_some() {
+        build.flag_if_supported("-save-temps");
+    }
+
+    if env::var_os("CARGO_FEATURE_NO_OPT_ARCH").is_none() && target_arch == "arm64" {
+        build.flag_if_supported("/arch:armv8.1");
+    }
+
+    if env::var_os("CARGO_FEATURE_SKIP_COLLECT_ON_EXIT").is_some() {
+        build.define("MI_SKIP_COLLECT_ON_EXIT", "1");
+    }
+
+    // Remove heavy debug assertions etc
+    let profile = std::env::var("PROFILE").unwrap();
+    match profile.as_str() {
+        "debug" => build.define("MI_DEBUG_FULL", "3"),
+        "release" => build.define("MI_DEBUG_FULL", "0").define("MI_DEBUG", "0"),
+        _ => build.define("MI_DEBUG_FULL", "3"),
+    };
+
+    if build.get_compiler().is_like_msvc() {
+        build.cpp(true);
+    }
+
+    const LIBS: [&str; 5] = ["psapi", "shell32", "user32", "advapi32", "bcrypt"];
+
+    for lib in LIBS {
+        println!("cargo:rustc-link-lib={}", lib);
+    }
+
+    build.compile("mimalloc_safe_static");
+}