Add compile-time support for mimalloc (#592)

Microsoft's `mimalloc` is a general purpose allocator with excellent
performance characteristics and, furthermore, it implements defensive
security features to protect against heap vulnerabilities.  See
https://github.com/microsoft/mimalloc.

In initial tests, it appears that it may improve latency at the cost of
~20% memory overhead.

This change primarily enables further testing, since it doesn't seem
worth taking the memory overhead at this point.

Author: olix0r

Cargo.lock

@@ -207,6 +207,15 @@ dependencies = [
  "bitflags",
 ]
 
+[[package]]
+name = "cmake"
+version = "0.1.41"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3c84c596dcf125d6781f58e3f4254677ec2a6d8aa56e8501ac277100990b3229"
+dependencies = [
+ "cc",
+]
+
 [[package]]
 name = "crc"
 version = "1.7.0"
@@ -692,6 +701,15 @@ version = "0.2.65"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "1a31a0627fdf1f6a39ec0dd577e101440b7db22672c0901fe00a9a6fbb5c24e8"
 
+[[package]]
+name = "libmimalloc-sys"
+version = "0.1.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a27252ec1d0c4e0dd6142cbc572da50b363ab56fc334f7aa8fadf295b2e24e74"
+dependencies = [
+ "cmake",
+]
+
 [[package]]
 name = "linked-hash-map"
 version = "0.5.3"
@@ -1146,6 +1164,7 @@ dependencies = [
  "futures 0.3.5",
  "linkerd2-app",
  "linkerd2-signal",
+ "mimalloc",
  "tokio",
  "tracing",
 ]
@@ -1566,6 +1585,15 @@ version = "2.3.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "3728d817d99e5ac407411fa471ff9800a778d88a24685968b36824eaf4bee400"
 
+[[package]]
+name = "mimalloc"
+version = "0.1.19"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "6c52de2069999f01bd26436564dbe7de3a87898feeb7a0d0ff9eb20a05bb7ca0"
+dependencies = [
+ "libmimalloc-sys",
+]
+
 [[package]]
 name = "miniz_oxide"
 version = "0.1.2"

Dockerfile

@@ -37,7 +37,7 @@ ARG PROXY_FEATURES
 
 RUN --mount=type=cache,target=/var/lib/apt/lists \
     --mount=type=cache,target=/var/tmp \
-    apt update && apt install -y time
+    apt update && apt install -y time cmake
 
 WORKDIR /usr/src/linkerd2-proxy
 COPY . .

linkerd2-proxy/Cargo.toml

@@ -11,6 +11,7 @@ mock-orig-dst  = ["linkerd2-app/mock-orig-dst"]
 
 [dependencies]
 futures = { version = "0.3", features = ["compat"] }
+mimalloc = { version = "0.1.19", optional = true }
 linkerd2-app = { path = "../linkerd/app" }
 linkerd2-signal = { path = "../linkerd/signal" }
 tokio = { version = "0.2", features = ["rt-core"] }

linkerd2-proxy/src/main.rs

@@ -8,6 +8,10 @@ use linkerd2_app::{trace, Config};
 use linkerd2_signal as signal;
 pub use tracing::{debug, error, info, warn};
 
+#[cfg(feature = "mimalloc")]
+#[global_allocator]
+static GLOBAL: mimalloc::MiMalloc = mimalloc::MiMalloc;
+
 #[tokio::main(basic_scheduler)]
 async fn main() {
     let trace = trace::init();