aya-gen: add bindgen module and utils to write to file

pull/1/head
Alessandro Decina 4 years ago
parent 16975e87d8
commit 99447bcc0d

@ -0,0 +1,13 @@
use bindgen::{self, Builder, EnumVariation};
pub fn builder() -> Builder {
let bindgen = bindgen::builder()
.use_core()
.ctypes_prefix("::aya_bpf_cty")
.layout_tests(false)
.clang_arg("-Wno-unknown-attributes")
.default_enum_style(EnumVariation::Consts)
.prepend_enum_name(false);
bindgen
}

@ -1,9 +1,9 @@
use std::{io, path::Path, process::Command, str::from_utf8}; use std::{io, path::Path, process::Command, str::from_utf8};
use bindgen::builder;
use thiserror::Error; use thiserror::Error;
use crate::{ use crate::{
bindgen,
getters::{generate_getters_for_items, probe_read_getter}, getters::{generate_getters_for_items, probe_read_getter},
rustfmt, rustfmt,
}; };
@ -28,11 +28,7 @@ pub fn generate<T: AsRef<str>>(
types: &[T], types: &[T],
probe_read_getters: bool, probe_read_getters: bool,
) -> Result<String, Error> { ) -> Result<String, Error> {
let mut bindgen = builder() let mut bindgen = bindgen::builder();
.use_core()
.ctypes_prefix("::aya_bpf_cty")
.layout_tests(false)
.clang_arg("-Wno-unknown-attributes");
let c_header = c_header_from_btf(btf_file)?; let c_header = c_header_from_btf(btf_file)?;
bindgen = bindgen.header_contents("kernel_types.h", &c_header); bindgen = bindgen.header_contents("kernel_types.h", &c_header);

@ -1,3 +1,19 @@
use std::{
fs::File,
io::{self, Write},
path::Path,
};
pub mod bindgen;
pub mod btf_types; pub mod btf_types;
pub mod getters; pub mod getters;
pub mod rustfmt; pub mod rustfmt;
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)?)
}

Loading…
Cancel
Save