feat(memtrack): support tcmalloc

Author: not-matthias

crates/memtrack/src/allocators/dynamic.rs

@@ -46,6 +46,25 @@ fn get_allocator_paths(lib: &AllocatorKind) -> &'static [&'static str] {
             // NixOS
             "/nix/store/*mimalloc*/lib/libmimalloc.so*",
         ],
+        AllocatorKind::Tcmalloc => &[
+            // gperftools tcmalloc variants
+            // Debian, Ubuntu: Standard Linux multiarch paths
+            "/lib/*-linux-gnu/libtcmalloc.so*",
+            "/lib/*-linux-gnu/libtcmalloc_minimal.so*",
+            "/lib/*-linux-gnu/libtcmalloc_debug.so*",
+            "/lib/*-linux-gnu/libtcmalloc_and_profiler.so*",
+            "/usr/lib/*-linux-gnu/libtcmalloc.so*",
+            "/usr/lib/*-linux-gnu/libtcmalloc_minimal.so*",
+            "/usr/lib/*-linux-gnu/libtcmalloc_debug.so*",
+            "/usr/lib/*-linux-gnu/libtcmalloc_and_profiler.so*",
+            // RHEL, Fedora, CentOS, Arch
+            "/lib*/libtcmalloc*.so*",
+            "/usr/lib*/libtcmalloc*.so*",
+            "/usr/local/lib*/libtcmalloc*.so*",
+            // NixOS
+            "/nix/store/*tcmalloc*/lib/libtcmalloc*.so*",
+            "/nix/store/*gperftools*/lib/libtcmalloc*.so*",
+        ],
     }
 }
 

crates/memtrack/src/allocators/mod.rs

@@ -19,8 +19,18 @@ pub enum AllocatorKind {
     Jemalloc,
     /// mimalloc - Microsoft's allocator
     Mimalloc,
+    /// TCMalloc - Google's thread-caching malloc
+    ///
+    /// Two variants exist:
+    /// - **gperftools** (github.com/gperftools/gperftools): Original ~2005 release.
+    ///   Exports both standard symbols (malloc/free) AND tc_* prefixed symbols.
+    /// - **google/tcmalloc** (github.com/google/tcmalloc): Modern ~2020 rewrite.
+    ///   Exports ONLY standard symbols (malloc/free/etc.) - no tc_* prefix.
+    ///
+    /// We'll always try to attach to both the standard and `tc_*` API. If the newer rewrite is
+    /// used, we'll only attach to the standard API.
+    Tcmalloc,
     // Future allocators:
-    // Tcmalloc,
     // Hoard,
     // Rpmalloc,
 }
@@ -33,6 +43,7 @@ impl AllocatorKind {
         &[
             AllocatorKind::Jemalloc,
             AllocatorKind::Mimalloc,
+            AllocatorKind::Tcmalloc,
             AllocatorKind::LibCpp,
             AllocatorKind::Libc,
         ]
@@ -45,23 +56,14 @@ impl AllocatorKind {
             AllocatorKind::LibCpp => "libc++",
             AllocatorKind::Jemalloc => "jemalloc",
             AllocatorKind::Mimalloc => "mimalloc",
+            AllocatorKind::Tcmalloc => "tcmalloc",
         }
     }
 
     /// Returns true if this allocator is required (must be found).
     pub fn is_required(&self) -> bool {
         matches!(self, AllocatorKind::Libc)
     }
-
-    /// Returns the symbol names used to detect this allocator in binaries.
-    pub fn symbols(&self) -> &'static [&'static str] {
-        match self {
-            AllocatorKind::Libc => &["malloc", "free"],
-            AllocatorKind::LibCpp => &["_Znwm", "_Znam", "_ZdlPv", "_ZdaPv"],
-            AllocatorKind::Jemalloc => &["_rjem_malloc", "je_malloc", "je_malloc_default"],
-            AllocatorKind::Mimalloc => &["mi_malloc_aligned", "mi_malloc", "mi_free"],
-        }
-    }
 }
 
 /// Discovered allocator library with its kind and path.

crates/memtrack/src/allocators/static_linked.rs

@@ -4,6 +4,19 @@ use std::path::{Path, PathBuf};
 
 use crate::allocators::{AllocatorKind, AllocatorLib};
 
+impl AllocatorKind {
+    /// Returns the symbol names used to detect this allocator in binaries.
+    pub fn symbols(&self) -> &'static [&'static str] {
+        match self {
+            AllocatorKind::Libc => &["malloc", "free"],
+            AllocatorKind::LibCpp => &["_Znwm", "_Znam", "_ZdlPv", "_ZdaPv"],
+            AllocatorKind::Jemalloc => &["_rjem_malloc", "je_malloc", "je_malloc_default"],
+            AllocatorKind::Mimalloc => &["mi_malloc_aligned", "mi_malloc", "mi_free"],
+            AllocatorKind::Tcmalloc => &["tc_malloc", "tc_free", "tc_version"],
+        }
+    }
+}
+
 /// Walk upward and downward from current directory to find build directories.
 /// Returns all found build directories in order of preference.
 fn find_build_dirs() -> Vec<PathBuf> {

crates/memtrack/src/ebpf/memtrack.rs

@@ -291,6 +291,11 @@ impl MemtrackBpf {
                 let _ = self.attach_libcpp_probes(lib_path);
                 self.attach_mimalloc_probes(lib_path)
             }
+            AllocatorKind::Tcmalloc => {
+                // Try C++ operators (tcmalloc exports these for C++ programs)
+                let _ = self.attach_libcpp_probes(lib_path);
+                self.attach_tcmalloc_probes(lib_path)
+            }
         }
     }
 
@@ -405,6 +410,22 @@ impl MemtrackBpf {
 
         Ok(())
     }
+
+    /// Attach TCMalloc probes ( tc_* API).
+    ///
+    /// See:
+    /// - https://github.com/google/tcmalloc/blob/master/docs/reference.md
+    /// - https://github.com/gperftools/gperftools/blob/a47243150ec41097602730ff8779fafcc172d1fb/src/tcmalloc.cc#L178-L190
+
+    fn attach_tcmalloc_probes(&mut self, lib_path: &Path) -> Result<()> {
+        self.attach_standard_probes(lib_path, &["tc_"], &[])?;
+
+        self.try_attach_free(lib_path, "free_sized");
+        self.try_attach_free(lib_path, "free_aligned_sized");
+        self.try_attach_free(lib_path, "sdallocx");
+
+        Ok(())
+    }
     attach_tracepoint!(sched_fork);
 
     pub fn attach_tracepoints(&mut self) -> Result<()> {