probe 0.5.2

Static instrumentation probes
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

#![cfg(any(target_os = "linux", target_os = "android"))]

use probe::probe;
use std::env;
use std::process::Command;

#[test]
fn check_notes() {
    // First let's create probes with and without arguments
    probe!(test, foo);
    probe!(test, bar, 42);

    // Now make sure readelf can find "stapsdt" ELF notes in this test executable
    let test_exe = env::current_exe().unwrap();
    let output = Command::new("readelf")
        .arg("-n")
        .arg(&test_exe)
        .output()
        .unwrap();
    assert!(output.status.success());

    for error in String::from_utf8_lossy(&output.stderr).lines() {
        if error.contains("Warning: Gap in build notes detected") {
            continue;
        }
        panic!("{}", error);
    }

    let count = String::from_utf8_lossy(&output.stdout)
        .lines()
        .filter(|line| line.contains("NT_STAPSDT"))
        .count();
    assert_eq!(count, 2);
}

#[test]
fn check_base() {
    // No need to create further probes. The ones introduced by check_notes are
    // enough.
    // Make sure readelf can find ".stapsdt.base" ELF section in the executable
    let test_exe = env::current_exe().unwrap();
    let output = Command::new("readelf")
        .arg("-S")
        .arg(&test_exe)
        .output()
        .unwrap();
    assert!(output.status.success());

    let count = String::from_utf8_lossy(&output.stdout)
        .lines()
        .filter(|line| line.contains("stapsdt.base"))
        .count();
    assert_eq!(count, 1);
}