pfm 0.1.2

High-level wrapper of libpfm4
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

use super::util::pfm_err_description;
use libc::{prctl, PR_TASK_PERF_EVENTS_DISABLE, PR_TASK_PERF_EVENTS_ENABLE};
use pfm_sys::{pfm_initialize, PFM_SUCCESS};

#[derive(Default)]
pub struct Perfmon {
    initialized: bool,
}

impl Perfmon {
    /// Initialize perfmon
    pub fn initialize(&mut self) -> Result<(), String> {
        let errno = unsafe { pfm_initialize() };
        if errno == PFM_SUCCESS {
            self.initialized = true;
            Ok(())
        } else {
            Err(pfm_err_description(errno))
        }
    }

    /// Enable all counters on the calling process
    pub fn enable(&self) {
        unsafe {
            prctl(PR_TASK_PERF_EVENTS_ENABLE);
        }
    }

    /// Disable all counters on the calling process
    pub fn disable(&self) {
        unsafe {
            prctl(PR_TASK_PERF_EVENTS_DISABLE);
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    #[test]
    fn test_initialization() {
        let mut perfmon: Perfmon = Default::default();
        assert!(!perfmon.initialized);
        perfmon.initialize().expect("Perfmon failed to initialize");
        assert!(perfmon.initialized);
    }
}