scxtop 1.1.0

sched_ext scheduler tool for observability
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

// Copyright (c) Meta Platforms, Inc. and affiliates.
//
// This software may be used and distributed according to the terms of the
// GNU General Public License version 2.

use crate::bpf_skel::BpfSkel;
use crate::bpf_stats::BpfStats;

use anyhow::Result;
use libbpf_rs::Link;
use log::debug;

pub struct Tracer<'a> {
    pub skel: BpfSkel<'a>,
    trace_links: Vec<Link>,
}

impl<'a> Tracer<'a> {
    /// Creates a new application.
    pub fn new(skel: BpfSkel<'a>) -> Self {
        let trace_links = vec![];
        Self { skel, trace_links }
    }

    /// Returns the BPF stats for the tracer.
    pub fn stats(&self) -> Result<BpfStats> {
        BpfStats::get_from_skel(&self.skel)
    }

    /// Attaches any BPF programs required for perfetto traces.
    fn attach_trace_progs(&mut self, kprobes: &Vec<String>) -> Result<()> {
        self.trace_links = vec![
            self.skel.progs.on_softirq_entry.attach()?,
            self.skel.progs.on_softirq_exit.attach()?,
            self.skel.progs.on_ipi_send_cpu.attach()?,
        ];

        for kprobe in kprobes {
            self.trace_links.push(
                self.skel
                    .progs
                    .generic_kprobe
                    .attach_kprobe(false, kprobe)?,
            );
        }

        Ok(())
    }

    /// Starts the collection of a trace, does not stop the trace.
    pub fn trace(&mut self, kprobes: &Vec<String>) -> Result<()> {
        self.skel.maps.data_data.as_mut().unwrap().sample_rate = 1;
        self.skel.maps.data_data.as_mut().unwrap().enable_bpf_events = true;
        self.attach_trace_progs(kprobes)?;
        debug!(
            "attached {} trace progs, sample_rate: {}",
            self.trace_links.len(),
            self.skel.maps.data_data.as_ref().unwrap().sample_rate
        );
        Ok(())
    }

    pub fn clear_links(&mut self) -> Result<()> {
        self.trace_links.clear();
        Ok(())
    }
}