mirror of https://github.com/aya-rs/aya
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
45 lines
1.1 KiB
Rust
45 lines
1.1 KiB
Rust
use aya::{
|
|
Ebpf,
|
|
maps::{Array, HashMap},
|
|
programs::UProbe,
|
|
};
|
|
|
|
#[unsafe(no_mangle)]
|
|
#[inline(never)]
|
|
extern "C" fn trigger_ebpf_program_maps_disjoint() {
|
|
core::hint::black_box(trigger_ebpf_program_maps_disjoint);
|
|
}
|
|
|
|
#[test_log::test]
|
|
fn test_maps_disjoint() {
|
|
let mut bpf: Ebpf = Ebpf::load(crate::MAP_TEST).unwrap();
|
|
let prog: &mut UProbe = bpf
|
|
.program_mut("simple_prog_mut")
|
|
.unwrap()
|
|
.try_into()
|
|
.unwrap();
|
|
|
|
prog.load().unwrap();
|
|
prog.attach(
|
|
"trigger_ebpf_program_maps_disjoint",
|
|
"/proc/self/exe",
|
|
None,
|
|
None,
|
|
)
|
|
.unwrap();
|
|
|
|
let [foo, bar, baz] = bpf.maps_disjoint_mut(["FOO", "BAR", "BAZ"]);
|
|
|
|
let mut foo: Array<_, u32> = Array::try_from(foo.unwrap()).unwrap();
|
|
let mut bar: HashMap<_, u32, u8> = HashMap::try_from(bar.unwrap()).unwrap();
|
|
assert!(baz.is_none());
|
|
|
|
foo.set(0, 5, 0).unwrap();
|
|
bar.insert(0, 10, 0).unwrap();
|
|
|
|
trigger_ebpf_program_maps_disjoint();
|
|
|
|
assert_eq!(foo.get(&0, 0).unwrap(), 6);
|
|
assert_eq!(bar.get(&0, 0).unwrap(), 11);
|
|
}
|