|
| 1 | +// SPDX-License-Identifier: (LGPL-2.1 OR BSD-2-Clause) |
| 2 | + |
| 3 | +use core::time::Duration; |
| 4 | +use std::ffi::c_int; |
| 5 | +use std::mem::MaybeUninit; |
| 6 | + |
| 7 | +use plain::Plain; |
| 8 | + |
| 9 | +use libbpf_rs::skel::OpenSkel; |
| 10 | +use libbpf_rs::skel::Skel; |
| 11 | +use libbpf_rs::skel::SkelBuilder; |
| 12 | +use libbpf_rs::Result; |
| 13 | + |
| 14 | +mod ringbuf_multi { |
| 15 | + include!(concat!( |
| 16 | + env!("CARGO_MANIFEST_DIR"), |
| 17 | + "/src/bpf/ringbuf_multi.skel.rs" |
| 18 | + )); |
| 19 | +} |
| 20 | + |
| 21 | +use ringbuf_multi::*; |
| 22 | + |
| 23 | +unsafe impl Plain for types::sample {} |
| 24 | + |
| 25 | +fn process_sample(ring: c_int, data: &[u8]) -> i32 { |
| 26 | + let s = plain::from_bytes::<types::sample>(data).unwrap(); |
| 27 | + |
| 28 | + match s.seq { |
| 29 | + 0 => { |
| 30 | + assert_eq!(ring, 1); |
| 31 | + assert_eq!(s.value, 333); |
| 32 | + 0 |
| 33 | + } |
| 34 | + 1 => { |
| 35 | + assert_eq!(ring, 2); |
| 36 | + assert_eq!(s.value, 777); |
| 37 | + 0 |
| 38 | + } |
| 39 | + _ => unreachable!(), |
| 40 | + } |
| 41 | +} |
| 42 | + |
| 43 | +fn main() -> Result<()> { |
| 44 | + let skel_builder = RingbufMultiSkelBuilder::default(); |
| 45 | + let mut open_object = MaybeUninit::uninit(); |
| 46 | + let open_skel = skel_builder.open(&mut open_object)?; |
| 47 | + let mut skel = open_skel.load()?; |
| 48 | + |
| 49 | + // Only trigger BPF program for current process. |
| 50 | + let pid = unsafe { libc::getpid() }; |
| 51 | + skel.maps.bss_data.pid = pid; |
| 52 | + |
| 53 | + let mut builder = libbpf_rs::RingBufferBuilder::new(); |
| 54 | + builder |
| 55 | + .add(&skel.maps.ringbuf1, |data| process_sample(1, data)) |
| 56 | + .expect("failed to add ringbuf"); |
| 57 | + builder |
| 58 | + .add(&skel.maps.ringbuf2, |data| process_sample(2, data)) |
| 59 | + .expect("failed to add ringbuf"); |
| 60 | + let ringbuf = builder.build().unwrap(); |
| 61 | + |
| 62 | + let () = skel.attach()?; |
| 63 | + |
| 64 | + // trigger few samples, some will be skipped |
| 65 | + skel.maps.bss_data.target_ring = 0; |
| 66 | + skel.maps.bss_data.value = 333; |
| 67 | + let _pgid = unsafe { libc::getpgid(pid) }; |
| 68 | + |
| 69 | + // skipped, no ringbuf in slot 1 |
| 70 | + skel.maps.bss_data.target_ring = 1; |
| 71 | + skel.maps.bss_data.value = 555; |
| 72 | + let _pgid = unsafe { libc::getpgid(pid) }; |
| 73 | + |
| 74 | + skel.maps.bss_data.target_ring = 2; |
| 75 | + skel.maps.bss_data.value = 777; |
| 76 | + let _pgid = unsafe { libc::getpgid(pid) }; |
| 77 | + |
| 78 | + // poll for samples, should get 2 ringbufs back |
| 79 | + let n = ringbuf.poll_raw(Duration::MAX); |
| 80 | + assert_eq!(n, 2); |
| 81 | + println!("successfully polled {n} samples"); |
| 82 | + |
| 83 | + // expect extra polling to return nothing |
| 84 | + let n = ringbuf.poll_raw(Duration::from_secs(0)); |
| 85 | + assert!(n == 0, "{n}"); |
| 86 | + Ok(()) |
| 87 | +} |
0 commit comments