perf-event-open 0.5.1

Full-featured support for the `perf_event_open` syscall.
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150

use super::{RecordId, SampleType, Task};
use crate::ffi::deref_offset;

/// Process exited.
///
/// Can be enabled by [`ExtraRecord::task`][crate::config::ExtraRecord::task].
///
/// # Examples
///
/// ```rust
/// use std::sync::atomic::{AtomicBool, Ordering};
/// use std::sync::mpsc::channel;
/// use std::thread;
///
/// use perf_event_open::config::{Cpu, Opts, Proc};
/// use perf_event_open::count::Counter;
/// use perf_event_open::event::sw::Software;
/// # use perf_event_open::sample::record::Record;
///
/// static WAIT: AtomicBool = AtomicBool::new(true);
///
/// let (tid_tx, tid_rx) = channel();
/// let handle = thread::spawn(move || {
///     tid_tx.send(unsafe { libc::gettid() }).unwrap();
///     while WAIT.load(Ordering::Relaxed) {
///         std::hint::spin_loop();
///     }
///     thread::spawn(|| {}); // Fork here.
/// });
///
/// let event = Software::Dummy;
/// let target = (Proc(tid_rx.recv().unwrap() as _), Cpu::ALL);
///
/// let mut opts = Opts::default();
/// opts.extra_record.task = true;
///
/// let counter = Counter::new(event, target, opts).unwrap();
/// let sampler = counter.sampler(5).unwrap();
///
/// counter.enable().unwrap();
/// WAIT.store(false, Ordering::Relaxed);
/// handle.join().unwrap(); // Exit here.
///
/// # let mut vec = vec![];
/// for it in sampler.iter() {
///     println!("{:-?}", it);
///     # vec.push(it);
/// }
/// # assert!(vec.iter().any(|(_, it)| matches!(it, Record::Fork(_))));
/// # assert!(vec.iter().any(|(_, it)| matches!(it, Record::Exit(_))));
/// ```
#[derive(Clone, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct Exit {
    /// Record IDs.
    pub record_id: Option<RecordId>,

    /// Task info.
    pub task: Task,
    /// Parent task info.
    pub parent_task: Task,
    /// Timestamp.
    pub time: u64,
}

impl Exit {
    pub(crate) unsafe fn from_ptr(mut ptr: *const u8, sample_id_all: Option<SampleType>) -> Self {
        // https://github.com/torvalds/linux/blob/v6.13/include/uapi/linux/perf_event.h#L912
        // struct {
        //     struct perf_event_header header;
        //     u32 pid, ppid;
        //     u32 tid, ptid;
        //     u64 time;
        //     struct sample_id sample_id;
        // };

        let pid = deref_offset(&mut ptr);
        let ppid = deref_offset(&mut ptr);
        let tid = deref_offset(&mut ptr);
        let ptid = deref_offset(&mut ptr);

        // https://github.com/torvalds/linux/blob/v6.13/kernel/events/core.c#L8428
        let time = deref_offset(&mut ptr);
        let record_id = sample_id_all.map(|SampleType(ty)| RecordId::from_ptr(ptr, ty));

        let task = Task { pid, tid };
        let parent_task = Task {
            pid: ppid,
            tid: ptid,
        };

        Self {
            record_id,
            task,
            parent_task,
            time,
        }
    }
}

super::from!(Exit);

super::debug!(Exit {
    {record_id?},
    {task},
    {parent_task},
    {time},
});

/// Process forked.
///
/// See [`Exit`] for examples.
///
/// Can be enabled by [`ExtraRecord::task`][crate::config::ExtraRecord::task].
#[derive(Clone, PartialEq, Eq, Hash)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct Fork {
    /// Record IDs.
    pub record_id: Option<RecordId>,

    /// Task info.
    pub task: Task,
    /// Parent task info.
    pub parent_task: Task,
    /// Timestamp.
    pub time: u64,
}

impl Fork {
    pub(crate) unsafe fn from_ptr(ptr: *const u8, sample_id_all: Option<SampleType>) -> Self {
        // https://github.com/torvalds/linux/blob/v6.13/kernel/events/core.c#L8423
        let layout = Exit::from_ptr(ptr, sample_id_all);

        Self {
            record_id: layout.record_id,
            task: layout.task,
            parent_task: layout.parent_task,
            time: layout.time,
        }
    }
}

super::from!(Fork);

super::debug!(Fork {
    {record_id?},
    {task},
    {parent_task},
    {time},
});