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-test/src/tests/args.rs

35 lines
1008 B
Rust

use aya::{
include_bytes_aligned,
programs::{FEntry, KProbe},
Bpf, Btf,
};
use super::{integration_test, IntegrationTest};
#[integration_test]
fn kprobe_args() {
let bytes = include_bytes_aligned!("../../../../target/bpfel-unknown-none/debug/args");
let mut bpf = Bpf::load(bytes).unwrap();
let kprobe_vfs_write: &mut KProbe = bpf
.program_mut("kprobe_vfs_write")
.unwrap()
.try_into()
.unwrap();
kprobe_vfs_write.load().unwrap();
kprobe_vfs_write.attach("vfs_write", 0).unwrap();
}
#[integration_test]
fn fentry_args() {
let bytes = include_bytes_aligned!("../../../../target/bpfel-unknown-none/debug/args");
let mut bpf = Bpf::load(bytes).unwrap();
let fentry_vfs_write: &mut FEntry = bpf
.program_mut("fentry_vfs_write")
.unwrap()
.try_into()
.unwrap();
let btf = Btf::from_sys_fs().unwrap();
fentry_vfs_write.load("vfs_write", &btf).unwrap();
fentry_vfs_write.attach().unwrap();
}