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

pub extern crate serde;
pub extern crate serde_json;

pub mod account;
pub mod startup;
pub mod sys;
pub mod transfer;
pub mod stake;
pub mod types;
pub mod utils;
pub mod contract;

#[macro_export]
macro_rules! log {
    () => ($crate::utils::log(""));
    ($fmt:expr) => ($crate::utils::log(&format!($fmt)));
    ($fmt:expr, $($arg:tt)*) => ($crate::utils::log(&format!($fmt, $($arg)*)));
}

#[macro_export]
macro_rules! processor_entry {
    // entry: fn (tag: &str, payload: &[u8]) -> Result<(), ProcessError>
    ($entry:ident) => {
        #[allow(unused_imports)]
        use $crate::types::*;

        #[no_mangle]
        pub extern "C" fn process() -> InternalProcessResult {
            fn assert_entry_type<
                F: Fn($crate::startup::TransactionContext) -> ProcessResult<()>,
            >(
                _: F,
            ) {
            }
            assert_entry_type($entry);

            let result: ProcessResult<()> = $crate::startup::TransactionContext::initial_read()
                .and_then(|context| $entry(context));

            if let Err(ref e) = result {
                match *e {
                    ProcessError::Ignore => {}
                    _ => {
                        log!("Got an error while processing transaction: {:?}", e);
                    }
                }
            }

            result.into()
        }
    };
}

#[cfg(test)]
#[allow(private_no_mangle_fns)]
#[allow(dead_code)]
mod compile_test {
    use startup::TransactionContext;
    fn process_transaction(_: TransactionContext) -> Result<(), ProcessError> {
        Err(ProcessError::Generic)
    }
    processor_entry!(process_transaction);
}