allow to use snmalloc

Author: zh-jq

Cargo.lock

@@ -191,6 +191,7 @@ libc = "0.2.182"
 rustix = { version = "1.1", default-features = false }
 windows-sys = "0.61"
 socket2 = { version = "0.6.4", features = ["all"] }
+snmalloc = { package = "snmalloc-rs", version = "0.7", features = ["libc-api"] }
 #
 serde = "1.0"
 yaml-rust = { package = "yaml-rust2", version = "0.11" }

Cargo.toml

@@ -41,6 +41,7 @@ fastrand.workspace = true
 base64.workspace = true
 hickory-net.workspace = true
 hickory-proto.workspace = true
+snmalloc = { workspace = true, optional = true }
 vey-runtime.workspace = true
 vey-std-ext.workspace = true
 vey-types = { workspace = true, features = ["openssl", "rustls"] }
@@ -68,6 +69,8 @@ jemalloc = ["dep:vey-jemalloc"]
 mimalloc = ["dep:vey-mimalloc"]
 vendored-mimalloc = ["mimalloc", "vey-mimalloc?/vendored"]
 secure-vendored-mimalloc = ["mimalloc", "vey-mimalloc?/secure"]
+snmalloc = ["dep:snmalloc"]
+secure-snmalloc = ["snmalloc", "snmalloc/check"]
 quic = ["vey-types/quic", "vey-socks/quic", "vey-io-ext/quic", "vey-hickory-client/quic", "dep:quinn", "dep:h3", "dep:h3-quinn"]
 rustls-ring = ["vey-rustls-provider/rustls-ring", "vey-types/rustls-ring", "quinn?/rustls-ring"]
 rustls-aws-lc = ["vey-rustls-provider/rustls-aws-lc", "vey-types/rustls-aws-lc", "quinn?/rustls-aws-lc-rs"]

vey-bench/Cargo.toml

@@ -35,6 +35,8 @@ pub fn print_version() {
             }
         } else if #[cfg(feature = "mimalloc")] {
             println!("mimalloc {}", vey_mimalloc::lib_version());
+        } else if #[cfg(feature = "snmalloc")] {
+            println!("snmalloc");
         } else {
             println!("system");
         }

vey-bench/src/build.rs

@@ -22,6 +22,9 @@ cfg_if::cfg_if! {
     } else if #[cfg(feature="mimalloc")] {
         #[global_allocator]
         static GLOBAL: vey_mimalloc::Mimalloc = vey_mimalloc::Mimalloc;
+    } else if #[cfg(feature="snmalloc")] {
+        #[global_allocator]
+        static GLOBAL: snmalloc::SnMalloc = snmalloc::SnMalloc;
     }
 }
 

vey-bench/src/main.rs

@@ -18,6 +18,7 @@ openssl-sys.workspace = true # for openssl variant detection
 tokio = { workspace = true, features = ["macros", "net", "io-util", "time", "signal"] }
 kanal = { workspace = true, features = ["async"] }
 yaml-rust.workspace = true
+snmalloc = { workspace = true, optional = true }
 vey-std-ext.workspace = true
 vey-types.workspace = true
 vey-yaml = { workspace = true, features = ["histogram", "openssl"] }
@@ -39,6 +40,8 @@ jemalloc = ["dep:vey-jemalloc", "vey-daemon/jemalloc"]
 mimalloc = ["dep:vey-mimalloc", "vey-daemon/mimalloc"]
 vendored-mimalloc = ["mimalloc", "vey-mimalloc?/vendored"]
 secure-vendored-mimalloc = ["mimalloc", "vey-mimalloc?/secure"]
+snmalloc = ["dep:snmalloc"]
+secure-snmalloc = ["snmalloc", "snmalloc/check"]
 vendored-openssl = ["openssl/vendored"]
 vendored-tongsuo = ["openssl/tongsuo", "vey-cert-agent/tongsuo"]
 vendored-boringssl = ["openssl/boringssl"]

vey-dcgen/Cargo.toml

@@ -31,6 +31,8 @@ pub fn print_version(verbose_level: u8) {
                 }
             } else if #[cfg(feature = "mimalloc")] {
                 println!("mimalloc {}", vey_mimalloc::lib_version());
+            } else if #[cfg(feature = "snmalloc")] {
+                println!("snmalloc");
             } else {
                 println!("system");
             }

vey-dcgen/src/build.rs

@@ -16,6 +16,9 @@ cfg_if::cfg_if! {
     } else if #[cfg(feature="mimalloc")] {
         #[global_allocator]
         static GLOBAL: vey_mimalloc::Mimalloc = vey_mimalloc::Mimalloc;
+    } else if #[cfg(feature="snmalloc")] {
+        #[global_allocator]
+        static GLOBAL: snmalloc::SnMalloc = snmalloc::SnMalloc;
     }
 }
 

vey-dcgen/src/main.rs

@@ -41,6 +41,7 @@ uuid.workspace = true
 bitflags.workspace = true
 kanal = { workspace = true, features = ["async"] }
 rustc-hash.workspace = true
+snmalloc = { workspace = true, optional = true }
 vey-codec = { workspace = true, features = ["tls"] }
 vey-daemon = { workspace = true, features = ["event-log"] }
 vey-histogram.workspace = true
@@ -69,6 +70,8 @@ jemalloc = ["dep:vey-jemalloc", "vey-daemon/jemalloc"]
 mimalloc = ["dep:vey-mimalloc", "vey-daemon/mimalloc"]
 vendored-mimalloc = ["mimalloc", "vey-mimalloc?/vendored"]
 secure-vendored-mimalloc = ["mimalloc", "vey-mimalloc?/secure"]
+snmalloc = ["dep:snmalloc"]
+secure-snmalloc = ["snmalloc", "snmalloc/check"]
 ebpf = ["vey-daemon/ebpf"]
 vendored-libbpf = ["ebpf", "vey-reuseport/vendored-libbpf"]
 vendored-ebpf = ["ebpf", "vey-reuseport/vendored"]

vey-gateway/Cargo.toml

@@ -38,6 +38,8 @@ pub(crate) fn print_version(verbose_level: u8) {
                 }
             } else if #[cfg(feature = "mimalloc")] {
                 println!("mimalloc {}", vey_mimalloc::lib_version());
+            } else if #[cfg(feature = "snmalloc")] {
+                println!("snmalloc");
             } else {
                 println!("system");
             }