feat: add snmalloc support

Cargo.toml

@@ -604,9 +604,11 @@ tempfile = "3.8"
 test-fuzz = "6"
 rstest = "0.23.0"
 
+# allocators
 tikv-jemalloc-ctl = "0.6"
 tikv-jemallocator = "0.6"
 tracy-client = "0.17.3"
+snmalloc-rs = { version = "0.3.7", features = ["build_cc"] }
 
 # [patch.crates-io]
 # alloy-consensus = { git = "https://github.com/alloy-rs/alloy", rev = "5492e40" }

bin/reth/Cargo.toml

@@ -118,6 +118,12 @@ jemalloc-prof = [
 ]
 tracy-allocator = ["reth-cli-util/tracy-allocator"]
 
+# Because jemalloc is default and preferred over snmalloc when both features are
+# enabled, `--no-default-features` should be used when enabling snmalloc or
+# snmalloc-native.
+snmalloc = ["reth-cli-util/snmalloc"]
+snmalloc-native = ["reth-cli-util/snmalloc-native"]
+
 min-error-logs = ["tracing/release_max_level_error"]
 min-warn-logs = ["tracing/release_max_level_warn"]
 min-info-logs = ["tracing/release_max_level_info"]

crates/cli/util/Cargo.toml

@@ -30,10 +30,21 @@ tracy-client = { workspace = true, optional = true, features = ["demangle"] }
 
 [target.'cfg(unix)'.dependencies]
 tikv-jemallocator = { workspace = true, optional = true }
+snmalloc-rs = { workspace = true, optional = true }
 libc = "0.2"
 
 [features]
 jemalloc = ["dep:tikv-jemallocator"]
+
+# Enables jemalloc profiling features
 jemalloc-prof = ["jemalloc", "tikv-jemallocator?/profiling"]
 
+# Wraps the selected allocator in the tracy profiling allocator
 tracy-allocator = ["dep:tracy-client"]
+
+snmalloc = ["dep:snmalloc-rs"]
+
+# Enables the snmalloc-rs `native-cpu` feature, which optimizes snmalloc for the
+# native CPU of the host machine. Not sure why this feature is not derived from
+# RUSTFLAGS or enabled when `target-cpu=native`.
+snmalloc-native = ["snmalloc", "snmalloc-rs/native-cpu"]

crates/cli/util/src/allocator.rs

@@ -1,14 +1,27 @@
 //! Custom allocator implementation.
+//!
+//! We provide support for jemalloc and snmalloc on unix systems, and prefer jemalloc if both are
+//! enabled.
 
-// We use jemalloc for performance reasons.
+// We provide jemalloc allocator support, alongside snmalloc. If both features are enabled, jemalloc
+// is prioritized.
 cfg_if::cfg_if! {
     if #[cfg(all(feature = "jemalloc", unix))] {
         type AllocatorInner = tikv_jemallocator::Jemalloc;
+    } else if #[cfg(all(feature = "snmalloc", unix))] {
+        type AllocatorInner = snmalloc_rs::SnMalloc;
     } else {
         type AllocatorInner = std::alloc::System;
     }
 }
 
+// This is to prevent clippy unused warnings when we do `--all-features`
+cfg_if::cfg_if! {
+    if #[cfg(all(feature = "snmalloc", feature = "jemalloc", unix))] {
+        use snmalloc_rs as _;
+    }
+}
+
 cfg_if::cfg_if! {
     if #[cfg(feature = "tracy-allocator")] {
         type AllocatorWrapper = tracy_client::ProfiledAllocator<AllocatorInner>;