evan/protomask
1
Fork 0
Code

putting together bpf loader

Browse Source
This commit is contained in:
Evan Pratten 2023-07-27 15:09:55 -04:00
parent c1dec2ffc8
commit 9e9bd5a2b8
3 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
protomask/Cargo.toml
View File

@ -17,3 +17,4 @@ tokio = { version = "1.29.1", features = ["macros", "rt-multi-thread", "sync"] }
clap = { version = "4.3.11", features = ["derive"] }
aya = { git = "https://github.com/aya-rs/aya", rev = "f334cbd86ee8c83b29940ee80f0ab80a6d58e5d1", features = ["async_tokio"] }
aya-log = { git = "https://github.com/aya-rs/aya", rev = "f334cbd86ee8c83b29940ee80f0ab80a6d58e5d1" }
cfg-if = "1.0.0"

15
protomask/src/bpf.rs Normal file
View File

@ -0,0 +1,15 @@
use aya::include_bytes_aligned;
use aya::Bpf;
use aya::BpfError;
use cfg_if::cfg_if;
#[cfg(all(target_os = "linux", target_arch = "x86_64", debug_assertions))]
pub fn load_bpf() -> Result<Bpf, BpfError> {
Bpf::load(&include_bytes_aligned!("../../target/bpfel-unknown-none/debug/protomask-ebpf"))
}
#[cfg(all(target_os = "linux", target_arch = "x86_64", not(debug_assertions)))]
pub fn load_bpf() -> Result<Bpf, BpfError> {
Bpf::load(&include_bytes_aligned!("../../target/bpfel-unknown-none/release/protomask-ebpf"))
}

1
protomask/src/main.rs
View File

@ -1,3 +1,4 @@
mod bpf;
#[tokio::main]
pub async fn main(){