Skip to main content

trident_derive_flow_executor/
lib.rs

1use proc_macro::TokenStream;
2use quote::ToTokens;
3use syn::parse_macro_input;
4use syn::ItemFn;
5use syn::ItemImpl;
6use trident_syn::parser::trident_flow_executor::parse_trident_flow_executor;
7
8/// Marks a method to be executed as part of the fuzzing flow
9#[proc_macro_attribute]
10pub fn flow(_attr: TokenStream, item: TokenStream) -> TokenStream {
11    let input_fn = parse_macro_input!(item as ItemFn);
12    quote::quote!(#input_fn).into()
13}
14
15/// Marks a method to run once before any flow methods
16#[proc_macro_attribute]
17pub fn init(_attr: TokenStream, item: TokenStream) -> TokenStream {
18    let input_fn = parse_macro_input!(item as ItemFn);
19    quote::quote!(#input_fn).into()
20}
21
22/// Marks a method to run once after all flow methods
23#[proc_macro_attribute]
24pub fn end(_attr: TokenStream, item: TokenStream) -> TokenStream {
25    let input_fn = parse_macro_input!(item as ItemFn);
26    quote::quote!(#input_fn).into()
27}
28
29/// Implements the flow executor for a struct
30#[proc_macro_attribute]
31pub fn flow_executor(_attr: TokenStream, item: TokenStream) -> TokenStream {
32    let input = parse_macro_input!(item as ItemImpl);
33
34    match parse_trident_flow_executor(&input) {
35        Ok(executor) => executor.to_token_stream().into(),
36        Err(err) => err.to_compile_error().into(),
37    }
38}