Safer approach

Author: squadgazzz

Dockerfile

@@ -60,6 +60,13 @@ ENTRYPOINT [ "solvers" ]
 
 # Extract Binary
 FROM intermediate
+
+# Configure jemalloc profiling (only affects services built with jemalloc-profiling feature)
+# prof:true - Enable profiling
+# prof_active:true - Start profiling immediately
+# lg_prof_sample:19 - Sample every 2^19 bytes (~524KB)
+ENV MALLOC_CONF="prof:true,prof_active:true,lg_prof_sample:19"
+
 RUN apt-get update && \
     apt-get install -y build-essential cmake git zlib1g-dev libelf-dev libdw-dev libboost-dev libboost-iostreams-dev libboost-program-options-dev libboost-system-dev libboost-filesystem-dev libunwind-dev libzstd-dev git netcat-openbsd
 RUN git clone https://invent.kde.org/sdk/heaptrack.git /heaptrack && \

crates/observe/src/heap_dump_handler.rs

@@ -89,12 +89,17 @@ async fn generate_and_stream_dump(socket: &mut UnixStream) {
     tracing::info!("generating heap dump via jemalloc_pprof");
 
     // Access the global profiling controller
-    let prof_ctl = match jemalloc_pprof::PROF_CTL.as_ref() {
-        Some(ctl) => ctl,
-        None => {
+    // Catch panic if jemalloc profiling is not available
+    let prof_ctl = match std::panic::catch_unwind(|| jemalloc_pprof::PROF_CTL.as_ref()) {
+        Ok(Some(ctl)) => ctl,
+        Ok(None) => {
             tracing::error!("jemalloc profiling not initialized");
             return;
         }
+        Err(_) => {
+            tracing::error!("jemalloc profiling not available - service not built with jemalloc-profiling feature or allocator not configured");
+            return;
+        }
     };
 
     let mut prof_ctl = prof_ctl.lock().await;