aya/test/integration-ebpf/src/uprobe_cookie.rs

#![no_std]
#![no_main]

use aya_ebpf::{
    helpers,
    macros::{map, uprobe},
    maps::RingBuf,
    programs::ProbeContext,
    EbpfContext,
};

#[map]
static RING_BUF: RingBuf = RingBuf::with_byte_size(0, 0);

#[uprobe]
pub fn uprobe_cookie(ctx: ProbeContext) {
    let cookie = unsafe { helpers::bpf_get_attach_cookie(ctx.as_ptr()) };
    let cookie_bytes = cookie.to_le_bytes();
    let _res = RING_BUF.output(&cookie_bytes, 0);
}

#[cfg(not(test))]
#[panic_handler]
fn panic(_info: &core::panic::PanicInfo) -> ! {
    loop {}
}