unshell/ush-obfuscate/src/proc_impl_switcher.rs

/// Call some other function
macro_rules! passtrough {
    ($name:tt, $ref:expr) => {
        pub fn $name(input: TokenStream) -> TokenStream {
            $ref(input)
        }
    };
}

/// Just return the underlying string
macro_rules! unwrap_string {
    ($func:tt) => {
        pub fn $func(input: TokenStream) -> TokenStream {
            let input = parse_macro_input!(input as LitStr);

            (quote::quote! {
                #input
            })
            .into()
        }
    };
}

/// Delete the content
macro_rules! delete {
    ($func:tt) => {
        pub fn $func(_: TokenStream) -> TokenStream {
            (quote::quote! {}).into()
        }
    };
}

#[cfg(all(not(feature = "obfuscate_aes"), not(feature = "obfuscate_ref")))]
pub mod proc_impl {
    use proc_macro::TokenStream;
    use syn::{LitStr, parse_macro_input};

    unwrap_string!(xor);
    delete!(junk_asm);

    unwrap_string!(sym);
    unwrap_string!(sym_fn);
}

#[cfg(feature = "obfuscate_aes")]
pub mod proc_impl {
    use proc_macro::TokenStream;

    passtrough!(xor, crate::obfuscate::xor);
    passtrough!(junk_asm, crate::obfuscate::junk_asm);

    passtrough!(sym, crate::symbolic_aes::aes_str);
    passtrough!(sym_fn, crate::symbolic_aes::aes_fn_name);
}

#[cfg(feature = "obfuscate_ref")]
pub mod proc_impl {
    use proc_macro::TokenStream;
    use syn::{LitStr, parse_macro_input};

    unwrap_string!(xor);
    delete!(junk_asm);

    passtrough!(sym, crate::symbolic_ref::sym_ref);
    unwrap_string!(sym_fn);
}