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.
53 lines
1.2 KiB
Rust
53 lines
1.2 KiB
Rust
use aya::{
|
|
include_bytes_aligned,
|
|
maps::{AsyncPerfEventArray, HashMap},
|
|
programs::UProbe,
|
|
util::online_cpus,
|
|
Bpf,
|
|
};
|
|
use aya_log::BpfLogger;
|
|
use bytes::BytesMut;
|
|
use log::warn;
|
|
use tokio::task;
|
|
|
|
use crate::STACK_ARGUMENT;
|
|
|
|
#[no_mangle]
|
|
#[inline(never)]
|
|
pub extern "C" fn trigger_stack_argument(
|
|
a_0: u64,
|
|
a_1: u64,
|
|
a_2: u64,
|
|
a_3: u64,
|
|
a_4: u64,
|
|
a_5: u64,
|
|
//in x86_64, from arg6, stack_argument would be used
|
|
a_6: u64,
|
|
a_7: i64,
|
|
) {
|
|
}
|
|
|
|
#[tokio::test]
|
|
async fn stack_argument() {
|
|
event_logger::init();
|
|
let mut bpf = Bpf::load(crate::STACK_ARGUMENT).unwrap();
|
|
|
|
let prog: &mut UProbe = bpf
|
|
.program_mut("test_stack_argument")
|
|
.unwrap()
|
|
.try_into()
|
|
.unwrap();
|
|
prog.load().unwrap();
|
|
prog.attach(Some("trigger_stack_argument"), 0, "/proc/self/exe", None)
|
|
.unwrap();
|
|
let mut args_map = HashMap::try_from(bpf.take_map("ARGS").unwrap())?;
|
|
trigger_stack_argument(0, 1, 2, 3, 4, 5, 6, 7);
|
|
|
|
tokio::time::sleep(std::time::Duration::from_millis(100)).await;
|
|
assert_eq!(args_map.keys().count(), 8);
|
|
for iter in args_map.iter() {
|
|
let iter_v = iter.unwrap();
|
|
assert_eq!(iter_v.0, iter_v.1);
|
|
}
|
|
}
|