mirror of
https://github.com/raspberrypi/linux.git
synced 2025-12-08 02:49:48 +00:00
00fccd3ecc
This derive macro implements `Zeroable` for structs & unions precisely
if all fields also implement `Zeroable` and does nothing otherwise. The
plain `Zeroable` derive macro instead errors when it cannot derive
`Zeroable` safely. The `MaybeZeroable` derive macro is useful in cases
where manual checking is infeasible such as with the bindings crate.
Move the zeroable generics parsing into a standalone function in order
to avoid code duplication between the two derive macros.
Link: 1165cdad1a
Signed-off-by: Benno Lossin <benno.lossin@proton.me>
102 lines
3.4 KiB
Rust
102 lines
3.4 KiB
Rust
// SPDX-License-Identifier: GPL-2.0
|
|
|
|
#[cfg(not(kernel))]
|
|
use proc_macro2 as proc_macro;
|
|
|
|
use crate::helpers::{parse_generics, Generics};
|
|
use proc_macro::{TokenStream, TokenTree};
|
|
|
|
pub(crate) fn parse_zeroable_derive_input(
|
|
input: TokenStream,
|
|
) -> (
|
|
Vec<TokenTree>,
|
|
Vec<TokenTree>,
|
|
Vec<TokenTree>,
|
|
Option<TokenTree>,
|
|
) {
|
|
let (
|
|
Generics {
|
|
impl_generics,
|
|
decl_generics: _,
|
|
ty_generics,
|
|
},
|
|
mut rest,
|
|
) = parse_generics(input);
|
|
// This should be the body of the struct `{...}`.
|
|
let last = rest.pop();
|
|
// Now we insert `Zeroable` as a bound for every generic parameter in `impl_generics`.
|
|
let mut new_impl_generics = Vec::with_capacity(impl_generics.len());
|
|
// Are we inside of a generic where we want to add `Zeroable`?
|
|
let mut in_generic = !impl_generics.is_empty();
|
|
// Have we already inserted `Zeroable`?
|
|
let mut inserted = false;
|
|
// Level of `<>` nestings.
|
|
let mut nested = 0;
|
|
for tt in impl_generics {
|
|
match &tt {
|
|
// If we find a `,`, then we have finished a generic/constant/lifetime parameter.
|
|
TokenTree::Punct(p) if nested == 0 && p.as_char() == ',' => {
|
|
if in_generic && !inserted {
|
|
new_impl_generics.extend(quote! { : ::pin_init::Zeroable });
|
|
}
|
|
in_generic = true;
|
|
inserted = false;
|
|
new_impl_generics.push(tt);
|
|
}
|
|
// If we find `'`, then we are entering a lifetime.
|
|
TokenTree::Punct(p) if nested == 0 && p.as_char() == '\'' => {
|
|
in_generic = false;
|
|
new_impl_generics.push(tt);
|
|
}
|
|
TokenTree::Punct(p) if nested == 0 && p.as_char() == ':' => {
|
|
new_impl_generics.push(tt);
|
|
if in_generic {
|
|
new_impl_generics.extend(quote! { ::pin_init::Zeroable + });
|
|
inserted = true;
|
|
}
|
|
}
|
|
TokenTree::Punct(p) if p.as_char() == '<' => {
|
|
nested += 1;
|
|
new_impl_generics.push(tt);
|
|
}
|
|
TokenTree::Punct(p) if p.as_char() == '>' => {
|
|
assert!(nested > 0);
|
|
nested -= 1;
|
|
new_impl_generics.push(tt);
|
|
}
|
|
_ => new_impl_generics.push(tt),
|
|
}
|
|
}
|
|
assert_eq!(nested, 0);
|
|
if in_generic && !inserted {
|
|
new_impl_generics.extend(quote! { : ::pin_init::Zeroable });
|
|
}
|
|
(rest, new_impl_generics, ty_generics, last)
|
|
}
|
|
|
|
pub(crate) fn derive(input: TokenStream) -> TokenStream {
|
|
let (rest, new_impl_generics, ty_generics, last) = parse_zeroable_derive_input(input);
|
|
quote! {
|
|
::pin_init::__derive_zeroable!(
|
|
parse_input:
|
|
@sig(#(#rest)*),
|
|
@impl_generics(#(#new_impl_generics)*),
|
|
@ty_generics(#(#ty_generics)*),
|
|
@body(#last),
|
|
);
|
|
}
|
|
}
|
|
|
|
pub(crate) fn maybe_derive(input: TokenStream) -> TokenStream {
|
|
let (rest, new_impl_generics, ty_generics, last) = parse_zeroable_derive_input(input);
|
|
quote! {
|
|
::pin_init::__maybe_derive_zeroable!(
|
|
parse_input:
|
|
@sig(#(#rest)*),
|
|
@impl_generics(#(#new_impl_generics)*),
|
|
@ty_generics(#(#ty_generics)*),
|
|
@body(#last),
|
|
);
|
|
}
|
|
}
|