feat: check for perf compression and enable multi-events conditionally

Author: GuillaumeLagrange

src/executor/wall_time/perf/mod.rs

@@ -18,6 +18,7 @@ use crate::run::UnwindingMode;
 use anyhow::Context;
 use fifo::PerfFifo;
 use libc::pid_t;
+use perf_executable::get_compression_flags;
 use perf_executable::get_event_flags;
 use perf_map::ProcessSymbols;
 use runner_shared::artifacts::ArtifactExt;
@@ -137,10 +138,15 @@ impl PerfRunner {
         if !is_codspeed_debug_enabled() {
             perf_wrapper_builder.arg("--quiet");
         }
-        // Add events flag if all required events are available
-        if let Some(events_flag) = get_event_flags(&working_perf_executable)? {
-            perf_wrapper_builder.arg(events_flag);
+        // Add compression if available
+        if let Some(compression_flags) = get_compression_flags(&working_perf_executable)? {
+            perf_wrapper_builder.arg(compression_flags);
+            // Add events flag if all required events are available
+            if let Some(events_flag) = get_event_flags(&working_perf_executable)? {
+                perf_wrapper_builder.arg(events_flag);
+            }
         }
+
         perf_wrapper_builder.args([
             "--timestamp",
             // Required for matching the markers and URIs to the samples.
@@ -149,7 +155,6 @@ impl PerfRunner {
             "--freq=997", // Use a prime number to avoid synchronization with periodic tasks
             "--delay=-1",
             "-g",
-            "--compression-level=3", // 3 is a widely adopted default level (AWS Athena, Python, ...)
             "--user-callchains",
             &format!("--call-graph={cg_mode}"),
             &format!(

src/executor/wall_time/perf/perf_executable.rs

@@ -1,4 +1,5 @@
 use crate::prelude::*;
+use std::path::Path;
 
 use std::{ffi::OsString, process::Command};
 
@@ -108,3 +109,29 @@ pub fn get_event_flags(perf_executable: &OsString) -> anyhow::Result<Option<Stri
     );
     Ok(Some(format!("-e {{{}}}", perf_events.join(","))))
 }
+
+pub fn get_compression_flags<S: AsRef<Path>>(perf_executable: S) -> Result<Option<String>> {
+    let output = Command::new(perf_executable.as_ref())
+        .arg("version")
+        .arg("--build-options")
+        .output()
+        .context("Failed to run perf version --build-options")?;
+
+    let stdout = String::from_utf8_lossy(&output.stdout);
+    debug!("Perf version build options:\n{stdout}");
+
+    // Look for zstd compression support in the build options
+    // Expected format: "                  zstd: [ on  ]  # HAVE_ZSTD_SUPPORT"
+    let has_zstd = stdout
+        .lines()
+        .any(|line| line.to_lowercase().contains("zstd: [ on"));
+
+    if has_zstd {
+        debug!("perf supports zstd compression");
+        // 3 is a widely adopted default level (AWS Athena, Python, ...)
+        Ok(Some("--compression-level=3".to_string()))
+    } else {
+        debug!("perf does not support zstd compression");
+        Ok(None)
+    }
+}