Merge pull request #2950 from multics69/lavd-ftrace-futex

scx_utils, scx_lavd: Harden the futex tracing.

Author: multics69

rust/scx_utils/src/compat.rs

@@ -214,9 +214,31 @@ pub fn debugfs_mount() -> Result<std::path::PathBuf> {
     mounts.into_iter().next().context("No debugfs mount found")
 }
 
+pub fn tracer_available(tracer: &str) -> Result<bool> {
+    let base_path = tracefs_mount().unwrap_or_else(|_| debugfs_mount().unwrap().join("tracing"));
+    let file = match std::fs::File::open(base_path.join("available_tracers")) {
+        Ok(f) => f,
+        Err(_) => return Ok(false),
+    };
+    let reader = std::io::BufReader::new(file);
+
+    for line in reader.lines() {
+        for tc in line.unwrap().split_whitespace() {
+            if tracer == tc {
+                return Ok(true);
+            }
+        }
+    }
+
+    Ok(false)
+}
+
 pub fn tracepoint_exists(tracepoint: &str) -> Result<bool> {
     let base_path = tracefs_mount().unwrap_or_else(|_| debugfs_mount().unwrap().join("tracing"));
-    let file = std::fs::File::open(base_path.join("available_events"))?;
+    let file = match std::fs::File::open(base_path.join("available_events")) {
+        Ok(f) => f,
+        Err(_) => return Ok(false),
+    };
     let reader = std::io::BufReader::new(file);
 
     for line in reader.lines() {

scheds/rust/scx_lavd/src/main.rs

@@ -455,6 +455,11 @@ impl<'a> Scheduler<'a> {
             ("futex_unlock_pi", &skel.progs.fexit_futex_unlock_pi),
         ];
 
+        if compat::tracer_available("function_graph")? == false {
+            info!("Ftrace is not enabled in the kernel.");
+            return Ok(false);
+        }
+
         compat::cond_kprobes_enable(ftraces)
     }