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

#![allow(clippy::all)]
#![allow(non_upper_case_globals)]
#![allow(non_camel_case_types)]
#![allow(non_snake_case)]

#[cfg(enabled)] //Only include the bindings if build.rs says to enabled them
include!("libstapsdt.rs");

#[cfg(enabled)]
#[cfg(test)]
mod tests {
    use super::*;

    use std::ffi::CString;
    use std::os::raw::*;

    #[test]
    fn smoke_test() {
        unsafe {
            let provider = providerInit(CString::new("foo").unwrap().as_ptr());
            assert!(!provider.is_null());

            let probe1 = providerAddProbe(provider, CString::new("probe1").unwrap().as_ptr(), 0);
            assert!(!probe1.is_null());

            let probe2 = providerAddProbe(
                provider,
                CString::new("probe2").unwrap().as_ptr(),
                6,
                ArgType_t_uint64,
                ArgType_t_uint16,
                ArgType_t_uint32,
                ArgType_t_uint8,
                ArgType_t_int64,
                ArgType_t_int32,
            );
            assert!(!probe2.is_null());

            //Probes are not loaded yet, so they definitely should not be enabled
            assert_eq!(0, probeIsEnabled(probe1));
            assert_eq!(0, probeIsEnabled(probe2));

            assert_eq!(0, providerLoad(provider));

            for _ in 0..100 {
                probeFire(probe1);
                probeFire(
                    probe2,
                    6 as c_ulonglong,
                    7 as c_ushort as c_uint, // hurray weird C variadic conventions
                    8 as c_ulong,
                    9 as c_uchar as c_uint, // hurray weird C variadic conventions
                    -10 as c_longlong,
                    -11 as c_long,
                );
            }

            assert_eq!(0, providerUnload(provider));

            providerDestroy(provider);
        }
    }
}