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
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
extern crate proc_macro;
extern crate syn;
extern crate timebomb;
use proc_macro::TokenStream;
use quote::quote;
use syn::parse_macro_input;
#[proc_macro_attribute]
pub fn timeout(attr: TokenStream, item: TokenStream) -> TokenStream {
let input = syn::parse_macro_input!(item as syn::ItemFn);
let attribute_args = parse_macro_input!(attr as syn::AttributeArgs);
let name = &input.sig.ident;
let body = &input.block;
let time_ms = get_timeout(&attribute_args);
assert_other_timeouts(&input);
let result = quote! {
fn #name() {
timebomb::timeout_ms(|| {
#body
}, #time_ms);
}
};
result.into()
}
fn assert_other_timeouts(input: &syn::ItemFn) {
for attribute in &input.attrs {
let meta = attribute.parse_meta();
match meta {
Ok(m) => match m {
syn::Meta::Path(p) => {
let identifier = p.get_ident().expect("Expected identifier!");
if identifier == "timeout" {
panic!("Timeout attribute is only allowed once");
}
}
syn::Meta::List(ml) => {
let identifier = ml.path.get_ident().expect("Expected identifier!");
if identifier == "timeout" {
panic!("Timeout attribute is only allowed once");
}
}
syn::Meta::NameValue(nv) => {
let identifier = nv.path.get_ident().expect("Expected identifier!");
if identifier == "timeout" {
panic!("Timeout attribute is only allowed once");
}
}
},
Err(e) => panic!("Could not determine meta data. Error {}.", e),
}
}
}
fn get_timeout(attribute_args: &syn::AttributeArgs) -> u32 {
if attribute_args.len() > 1 {
panic!("Only one integer expected. Example: #[timeout(10)]");
}
match &attribute_args[0] {
syn::NestedMeta::Meta(_) => {
panic!("Integer expected. Example: #[timeout(10)]");
}
syn::NestedMeta::Lit(lit) => match lit {
syn::Lit::Int(int) => return int.base10_parse::<u32>().expect("Integer expected"),
_ => {
panic!("Integer as timeout in ms expected. Example: #[timeout(10)]");
}
},
}
}