bat-cli 0.13.1

Blockchain Auditor Toolkit (BAT)
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
104
105
106
107
108
109
110
111
112
113
114
115
116
117

use solar_parse::{ast, interface::Session};

use crate::batbelt::evm::types::{EvmFunction, EvmMutability, EvmParam, EvmVisibility};

use super::evm_file_parser::{
    extract_source_by_lines, span_to_end_line, span_to_line, type_to_string,
};

/// Parse an ItemFunction AST node into our EvmFunction type.
pub fn parse_function_definition(
    sess: &Session,
    func: &ast::ItemFunction<'_>,
    contract_name: &str,
    source: &str,
) -> EvmFunction {
    let name = func
        .header
        .name
        .map(|n| n.as_str().to_string())
        .unwrap_or_else(|| match func.kind {
            ast::FunctionKind::Constructor => "constructor".to_string(),
            ast::FunctionKind::Fallback => "fallback".to_string(),
            ast::FunctionKind::Receive => "receive".to_string(),
            ast::FunctionKind::Modifier => "modifier".to_string(),
            ast::FunctionKind::Function => "unnamed".to_string(),
        });

    let visibility = func
        .header
        .visibility
        .as_ref()
        .map(|v| match v.data {
            ast::Visibility::External => EvmVisibility::External,
            ast::Visibility::Public => EvmVisibility::Public,
            ast::Visibility::Internal => EvmVisibility::Internal,
            ast::Visibility::Private => EvmVisibility::Private,
        })
        .unwrap_or(EvmVisibility::Internal);

    let mutability = func
        .header
        .state_mutability
        .as_ref()
        .map(|m| match m.data {
            ast::StateMutability::Pure => EvmMutability::Pure,
            ast::StateMutability::View => EvmMutability::View,
            ast::StateMutability::Payable => EvmMutability::Payable,
            ast::StateMutability::NonPayable => EvmMutability::NonPayable,
        })
        .unwrap_or(EvmMutability::NonPayable);

    let modifiers: Vec<String> = func
        .header
        .modifiers
        .iter()
        .map(|m| m.name.last().as_str().to_string())
        .collect();

    let params: Vec<EvmParam> = func
        .header
        .parameters
        .iter()
        .map(|p| parse_parameter(sess, p))
        .collect();

    let returns: Vec<EvmParam> = func
        .header
        .returns
        .as_ref()
        .map(|r| r.iter().map(|p| parse_parameter(sess, p)).collect())
        .unwrap_or_default();

    let body_source = func
        .body
        .as_ref()
        .map(|block| {
            extract_source_by_lines(
                source,
                span_to_line(sess, block.span),
                span_to_end_line(sess, block.span),
            )
        })
        .unwrap_or_default();

    // Use full function span: from header start to body end
    let full_span = func.header.span.to(func.body_span);
    let line = span_to_line(sess, full_span);
    let end_line = span_to_end_line(sess, full_span);

    EvmFunction {
        name,
        contract_name: contract_name.to_string(),
        visibility,
        mutability,
        modifiers,
        params,
        returns,
        body_source,
        line,
        end_line,
        is_constructor: func.kind == ast::FunctionKind::Constructor,
        is_fallback: func.kind == ast::FunctionKind::Fallback,
        is_receive: func.kind == ast::FunctionKind::Receive,
    }
}

fn parse_parameter(sess: &Session, p: &ast::VariableDefinition<'_>) -> EvmParam {
    let name = p.name.map(|n| n.as_str().to_string()).unwrap_or_default();
    let type_name = type_to_string(sess, &p.ty);
    let storage_location = p.data_location.map(|s| s.to_str().to_string());

    EvmParam {
        name,
        type_name,
        storage_location,
    }
}