sanitizer_macros 0.2.1

Derive macro for the sanitizer crate
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47

use crate::arg::ArgBuilder;
use crate::codegen::sanitizers::PathOrList;
use crate::sanitizer::SanitizerError;
use proc_macro2::TokenStream;
use quote::quote;

#[macro_use]
macro_rules! sanitizer_with_arg {
    ( $sanitizer : expr, $body : expr ) => {
        if $sanitizer.has_args() {
            $body
        } else {
            Err(SanitizerError::Only64BitInt)
        }
    };
}

pub fn get_int_sanitizers(sanitizer: &PathOrList) -> Result<TokenStream, SanitizerError> {
    match sanitizer.to_string().as_str() {
        "clamp" => {
            sanitizer_with_arg!(sanitizer, {
                if sanitizer.get_args().len() == 2 {
                    let arg_one = ArgBuilder::int(sanitizer.get_args().args[0].as_str());
                    let arg_two = ArgBuilder::int(sanitizer.get_args().args[1].as_str());
                    Ok(quote! {
                        clamp(#arg_one, #arg_two)
                    })
                } else {
                    Err(SanitizerError::WrongArguments)
                }
            })
        }
        "custom" => {
            sanitizer_with_arg!(sanitizer, {
                if sanitizer.get_args().len() == 1 {
                    let arg_one = ArgBuilder::ident(sanitizer.get_args().args[0].as_str());
                    Ok(quote! {
                        call(#arg_one)
                    })
                } else {
                    Err(SanitizerError::WrongArguments)
                }
            })
        }
        _ => Err(SanitizerError::InvalidSanitizer),
    }
}