rsleigh 0.4.2

SLEIGH (.slaspec) parser and Rust decoder/P-code emitter codegen — Ghidra-compatible disassembly in pure Rust
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

use crate::semantic::meaning::{AttachLiteral, AttachNumber, AttachVarnode, Meaning};
use crate::semantic::{AttachLiteralId, AttachNumberId, AttachVarnodeId, PrintBase, VarnodeId};
use crate::{syntax, SleighError, Span};

use super::{GlobalScope, Sleigh};

impl Sleigh {
    fn attach_meaning_fields(
        &mut self,
        fields: Vec<(String, Span)>,
        meaning: Meaning,
    ) -> Result<(), Box<SleighError>> {
        for (field_name, field_src) in fields.into_iter() {
            match self.get_global(&field_name) {
                Some(GlobalScope::TokenField(token_field)) => {
                    let token_field = self.token_field_mut(token_field);
                    token_field.attach(meaning)?;
                }
                Some(GlobalScope::Context(context)) => {
                    let context = self.context_mut(context);
                    context.attach(meaning)?;
                }
                _ => {
                    return Err(Box::new(SleighError::AttachInvalidVariable(
                        field_src.clone(),
                    )))
                }
            }
        }
        Ok(())
    }
    pub fn attach_meaning(
        &mut self,
        attach: syntax::attach::Attach,
    ) -> Result<(), Box<SleighError>> {
        let syntax::attach::Attach {
            src,
            fields,
            meaning,
        } = attach;
        match meaning {
            syntax::attach::Meaning::Variable(input_vars) => {
                let vars: Box<[(usize, VarnodeId)]> = input_vars
                    .into_iter()
                    .enumerate()
                    .filter_map(|(i, (name, s))| Some((i, name?, s)))
                    .map(
                        |(index, var_name, var_src)| -> Result<_, Box<SleighError>> {
                            let var_id = self
                                .get_global(&var_name)
                                .ok_or_else(|| {
                                    Box::new(SleighError::VarnodeUndefined(var_src.clone()))
                                })?
                                .varnode()
                                .ok_or_else(|| {
                                    Box::new(SleighError::VarnodeInvalid(var_src.clone()))
                                })?;
                            Ok((index, var_id))
                        },
                    )
                    .collect::<Result<_, _>>()?;
                // all varnodes need to have the same size
                let mut var_iter = vars.iter().map(|(_i, var)| self.varnode(*var).len_bytes);
                let varnode_len = var_iter.next().unwrap();
                for var in var_iter {
                    if var != varnode_len {
                        return Err(Box::new(SleighError::VarnodeInvalid(src.clone())));
                    }
                }
                self.attach_varnodes.push(AttachVarnode(vars));
                let meaning = Meaning::Varnode(AttachVarnodeId(self.attach_varnodes.len() - 1));
                self.attach_meaning_fields(fields, meaning)?;
            }
            syntax::attach::Meaning::Name(x) => {
                let names = x
                    .into_iter()
                    .enumerate()
                    .filter_map(|(i, (x, _s))| Some((i, x?)))
                    .collect();
                self.attach_literals.push(AttachLiteral(names));
                let meaning = Meaning::Literal(AttachLiteralId(self.attach_literals.len() - 1));
                self.attach_meaning_fields(fields, meaning)?;
            }
            syntax::attach::Meaning::Number(x) => {
                let values = x
                    .into_iter()
                    .enumerate()
                    .filter_map(|(i, (x, _s))| Some((i, x?)))
                    .collect();
                self.attach_numbers.push(AttachNumber(values));
                let meaning = Meaning::Number(
                    PrintBase::Hex,
                    AttachNumberId(self.attach_numbers.len() - 1),
                );
                self.attach_meaning_fields(fields, meaning)?;
            }
        }
        Ok(())
    }
}