yanguangshaonian
/
aya
mirror of https://github.com/aya-rs/aya
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
aya/aya-gen/src/lib.rs

20 lines
443 B
Rust
Raw Normal View History Unescape Escape

aya-gen: add bindgen module and utils to write to file
use std::{
fs::File,
io::{self, Write},
path::Path,
};
pub mod bindgen;
Add aya-gen aya-gen can be used to generate bindings for kernel types, eg: aya-gen btf-types ethhdr --probe-read-getters Will generate: // ... #[repr(C)] #[derive(Debug, Copy, Clone)] pub struct ethhdr { pub h_dest: [::aya_bpf_cty::c_uchar; 6usize], pub h_source: [::aya_bpf_cty::c_uchar; 6usize], pub h_proto: __be16, } impl ethhdr { pub fn h_dest(&self) -> Option<[::aya_bpf_cty::c_uchar; 6usize]> { unsafe { ::aya_bpf::helpers::bpf_probe_read(&self.h_dest) }.ok() } pub fn h_source(&self) -> Option<[::aya_bpf_cty::c_uchar; 6usize]> { unsafe { ::aya_bpf::helpers::bpf_probe_read(&self.h_source) }.ok() } pub fn h_proto(&self) -> Option<__be16> { unsafe { ::aya_bpf::helpers::bpf_probe_read(&self.h_proto) }.ok() } }
pub mod btf_types;
pub mod getters;
pub mod rustfmt;
aya-gen: add bindgen module and utils to write to file
pub fn write_to_file<T: AsRef<Path>>(path: T, code: &str) -> Result<(), io::Error> {
let mut file = File::create(path)?;
file.write_all(code.as_bytes())
}
pub fn write_to_file_fmt<T: AsRef<Path>>(path: T, code: &str) -> Result<(), io::Error> {
write_to_file(path, &rustfmt::format(code)?)
}