forked from AbleOS/ableos
29 lines
728 B
Rust
29 lines
728 B
Rust
|
extern crate proc_macro;
|
||
|
|
extern crate quote;
|
||
|
|
extern crate syn;
|
||
|
|
use {
|
||
|
|
proc_macro::TokenStream,
|
||
|
|
quote::quote,
|
||
|
|
syn::{parse_macro_input, ItemFn}
|
||
|
|
};
|
||
|
|
|
||
|
|
#[proc_macro_attribute]
|
||
|
|
pub fn ktest(_attr: TokenStream, item: TokenStream) -> TokenStream {
|
||
|
|
let input = parse_macro_input!(item as ItemFn);
|
||
|
|
let test_name = &input.sig.ident;
|
||
|
|
let static_var_name = syn::Ident::new(
|
||
|
|
&format!("__ktest_{}", test_name),
|
||
|
|
test_name.span(),
|
||
|
|
);
|
||
|
|
let out = quote! {
|
||
|
|
// #[cfg(feature = "ktest")]
|
||
|
|
#input
|
||
|
|
|
||
|
|
// #[cfg(feature = "ktest")]
|
||
|
|
#[unsafe(link_section = ".note.ktest")]
|
||
|
|
#[used]
|
||
|
|
pub static #static_var_name: fn() = #test_name;
|
||
|
|
};
|
||
|
|
|
||
|
|
TokenStream::from(out)
|
||
|
|
}
|