mirror of https://github.com/aya-rs/aya
Dave Tucker
2 years ago
parent
a0473548ca
commit
9a33b6b654
@ -1,151 +0,0 @@
|
||||
use indexmap::IndexMap;
|
||||
use proc_macro2::{Span, TokenStream};
|
||||
use quote::{quote, TokenStreamExt};
|
||||
use syn::{
|
||||
self, Fields, FieldsNamed, Generics, Ident, Item, ItemStruct, ItemUnion, Path, Type, TypePath,
|
||||
Visibility,
|
||||
};
|
||||
|
||||
pub struct GetterList<'a> {
|
||||
slf: Ident,
|
||||
item_fields: IndexMap<Ident, (&'a Item, &'a FieldsNamed)>,
|
||||
}
|
||||
|
||||
impl<'a> GetterList<'a> {
|
||||
pub fn new(items: &'a [Item]) -> GetterList<'a> {
|
||||
let item_fields = items
|
||||
.iter()
|
||||
.filter_map(|item| {
|
||||
unpack_item(item).map(|(ident, _generics, fields)| (ident.clone(), (item, fields)))
|
||||
})
|
||||
.collect();
|
||||
GetterList {
|
||||
slf: Ident::new("self", Span::call_site()),
|
||||
item_fields,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn iter(&self) -> impl Iterator<Item = (&'a Item, Vec<Getter<'_>>)> {
|
||||
self.item_fields
|
||||
.values()
|
||||
.map(move |(item, fields)| (*item, self.getters(&self.slf, fields)))
|
||||
}
|
||||
|
||||
fn getters(&self, ident: &'a Ident, fields: &'a FieldsNamed) -> Vec<Getter<'a>> {
|
||||
let mut getters = Vec::new();
|
||||
for field in &fields.named {
|
||||
if let Visibility::Inherited = field.vis {
|
||||
continue;
|
||||
}
|
||||
|
||||
let field_ident = field.ident.as_ref().unwrap();
|
||||
let field_s = field_ident.to_string();
|
||||
|
||||
// FIXME: bindgen generates fields named `_bitfield_N` for bitfields. If a type T has
|
||||
// two or more unions with bitfields, the getters for the bitfields - generated in impl
|
||||
// T - will clash. To avoid that we skip getters for bitfields altogether for now.
|
||||
// See sk_reuseport_md for an example where the clash happens.
|
||||
if field_s.starts_with("_bitfield") {
|
||||
continue;
|
||||
}
|
||||
|
||||
if field_s.starts_with("__bindgen_anon") {
|
||||
let field_ty_ident = match &field.ty {
|
||||
Type::Path(TypePath {
|
||||
path: Path { segments, .. },
|
||||
..
|
||||
}) => &segments.first().unwrap().ident,
|
||||
_ => panic!(),
|
||||
};
|
||||
let sub_fields = self
|
||||
.item_fields
|
||||
.get(field_ty_ident)
|
||||
.unwrap_or_else(|| panic!("{}", field_ty_ident.to_string()))
|
||||
.1;
|
||||
getters.extend(self.getters(field_ident, sub_fields).into_iter().map(
|
||||
|mut getter| {
|
||||
getter.prefix.insert(0, ident);
|
||||
getter
|
||||
},
|
||||
));
|
||||
} else {
|
||||
getters.push(Getter {
|
||||
ident: field_ident,
|
||||
prefix: vec![ident],
|
||||
ty: &field.ty,
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
getters
|
||||
}
|
||||
}
|
||||
|
||||
pub fn generate_getters_for_items(
|
||||
items: &[Item],
|
||||
gen_getter: impl Fn(&Getter<'_>) -> TokenStream,
|
||||
) -> TokenStream {
|
||||
let mut tokens = TokenStream::new();
|
||||
tokens.append_all(GetterList::new(items).iter().map(|(item, getters)| {
|
||||
let getters = getters.iter().map(&gen_getter);
|
||||
let (ident, generics, _) = unpack_item(item).unwrap();
|
||||
quote! {
|
||||
impl#generics #ident#generics {
|
||||
#(#getters)*
|
||||
}
|
||||
}
|
||||
}));
|
||||
|
||||
tokens
|
||||
}
|
||||
|
||||
pub fn read_getter(getter: &Getter<'_>, read_fn: &Path) -> TokenStream {
|
||||
let ident = getter.ident;
|
||||
let ty = getter.ty;
|
||||
let prefix = &getter.prefix;
|
||||
match ty {
|
||||
Type::Ptr(_) => {
|
||||
quote! {
|
||||
pub fn #ident(&self) -> Option<#ty> {
|
||||
let v = unsafe { #read_fn(&#(#prefix).*.#ident) }.ok()?;
|
||||
if v.is_null() {
|
||||
None
|
||||
} else {
|
||||
Some(v)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
_ => {
|
||||
quote! {
|
||||
pub fn #ident(&self) -> Option<#ty> {
|
||||
unsafe { #read_fn(&#(#prefix).*.#ident) }.ok()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
pub struct Getter<'a> {
|
||||
pub ident: &'a Ident,
|
||||
pub prefix: Vec<&'a Ident>,
|
||||
pub ty: &'a Type,
|
||||
}
|
||||
|
||||
fn unpack_item(item: &Item) -> Option<(&Ident, &Generics, &FieldsNamed)> {
|
||||
match item {
|
||||
Item::Struct(ItemStruct {
|
||||
ident,
|
||||
generics,
|
||||
fields: Fields::Named(fields),
|
||||
..
|
||||
})
|
||||
| Item::Union(ItemUnion {
|
||||
ident,
|
||||
generics,
|
||||
fields,
|
||||
..
|
||||
}) => Some((ident, generics, fields)),
|
||||
_ => None,
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue