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.
56 lines
1.4 KiB
Rust
56 lines
1.4 KiB
Rust
use std::time::Duration;
|
|
|
|
use aya::{include_bytes_aligned, programs::UProbe, Bpf};
|
|
|
|
#[test]
|
|
fn relocations() {
|
|
let bpf = load_and_attach(
|
|
"test_64_32_call_relocs",
|
|
include_bytes_aligned!("../../../target/bpfel-unknown-none/release/relocations"),
|
|
);
|
|
|
|
trigger_relocations_program();
|
|
std::thread::sleep(Duration::from_millis(100));
|
|
|
|
let m = aya::maps::Array::<_, u64>::try_from(bpf.map("RESULTS").unwrap()).unwrap();
|
|
assert_eq!(m.get(&0, 0).unwrap(), 1);
|
|
assert_eq!(m.get(&1, 0).unwrap(), 2);
|
|
assert_eq!(m.get(&2, 0).unwrap(), 3);
|
|
}
|
|
|
|
#[test]
|
|
fn text_64_64_reloc() {
|
|
let mut bpf = load_and_attach("test_text_64_64_reloc", integration_test::TEXT_64_64_RELOC);
|
|
|
|
let mut m = aya::maps::Array::<_, u64>::try_from(bpf.map_mut("RESULTS").unwrap()).unwrap();
|
|
m.set(0, 1, 0).unwrap();
|
|
m.set(1, 2, 0).unwrap();
|
|
|
|
trigger_relocations_program();
|
|
std::thread::sleep(Duration::from_millis(100));
|
|
|
|
assert_eq!(m.get(&0, 0).unwrap(), 2);
|
|
assert_eq!(m.get(&1, 0).unwrap(), 3);
|
|
}
|
|
|
|
fn load_and_attach(name: &str, bytes: &[u8]) -> Bpf {
|
|
let mut bpf = Bpf::load(bytes).unwrap();
|
|
|
|
let prog: &mut UProbe = bpf.program_mut(name).unwrap().try_into().unwrap();
|
|
prog.load().unwrap();
|
|
|
|
prog.attach(
|
|
Some("trigger_relocations_program"),
|
|
0,
|
|
"/proc/self/exe",
|
|
None,
|
|
)
|
|
.unwrap();
|
|
|
|
bpf
|
|
}
|
|
|
|
#[no_mangle]
|
|
#[inline(never)]
|
|
pub extern "C" fn trigger_relocations_program() {}
|