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.
43 lines
1.6 KiB
Rust
43 lines
1.6 KiB
Rust
use procfs::KernelVersion;
|
|
|
|
use aya::{
|
|
include_bytes_aligned,
|
|
programs::{Extension, Xdp, XdpFlags},
|
|
Bpf, BpfLoader,
|
|
};
|
|
|
|
#[test]
|
|
fn xdp() {
|
|
let kernel_version = KernelVersion::current().unwrap();
|
|
if kernel_version < KernelVersion::new(5, 18, 0) {
|
|
eprintln!("skipping test on kernel {kernel_version:?}, support for BPF_F_XDP_HAS_FRAGS was added in 5.18.0; see https://github.com/torvalds/linux/commit/c2f2cdb");
|
|
return;
|
|
}
|
|
|
|
let bytes = include_bytes_aligned!("../../../target/bpfel-unknown-none/release/pass");
|
|
let mut bpf = Bpf::load(bytes).unwrap();
|
|
let dispatcher: &mut Xdp = bpf.program_mut("pass").unwrap().try_into().unwrap();
|
|
dispatcher.load().unwrap();
|
|
dispatcher.attach("lo", XdpFlags::default()).unwrap();
|
|
}
|
|
|
|
#[test]
|
|
fn extension() {
|
|
let kernel_version = KernelVersion::current().unwrap();
|
|
if kernel_version < KernelVersion::new(5, 9, 0) {
|
|
eprintln!("skipping test on kernel {kernel_version:?}, XDP uses netlink");
|
|
return;
|
|
}
|
|
let main_bytes =
|
|
include_bytes_aligned!("../../../target/bpfel-unknown-none/release/main.bpf.o");
|
|
let mut bpf = Bpf::load(main_bytes).unwrap();
|
|
let pass: &mut Xdp = bpf.program_mut("pass").unwrap().try_into().unwrap();
|
|
pass.load().unwrap();
|
|
pass.attach("lo", XdpFlags::default()).unwrap();
|
|
|
|
let ext_bytes = include_bytes_aligned!("../../../target/bpfel-unknown-none/release/ext.bpf.o");
|
|
let mut bpf = BpfLoader::new().extension("drop").load(ext_bytes).unwrap();
|
|
let drop_: &mut Extension = bpf.program_mut("drop").unwrap().try_into().unwrap();
|
|
drop_.load(pass.fd().unwrap(), "xdp_pass").unwrap();
|
|
}
|