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.
aya/test/integration-ebpf/src/relocations.rs

47 lines
998 B
Rust

#![no_std]
#![no_main]
use core::hint;
use aya_bpf::{
macros::{map, uprobe},
maps::Array,
programs::ProbeContext,
};
#[map]
static mut RESULTS: Array<u64> = Array::with_max_entries(3, 0);
#[uprobe]
pub fn test_64_32_call_relocs(_ctx: ProbeContext) {
// this will link set_result and do a forward call
set_result(0, hint::black_box(1));
// set_result is already linked, this will just do the forward call
set_result(1, hint::black_box(2));
// this will link set_result_backward after set_result. Then will do a
// backward call to set_result.
set_result_backward(2, hint::black_box(3));
}
#[inline(never)]
fn set_result(index: u32, value: u64) {
unsafe {
if let Some(v) = RESULTS.get_ptr_mut(index) {
*v = value;
}
}
}
#[inline(never)]
fn set_result_backward(index: u32, value: u64) {
set_result(index, value);
}
#[cfg(not(test))]
#[panic_handler]
fn panic(_info: &core::panic::PanicInfo) -> ! {
loop {}
}