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

use nom::combinator::{cut, map};
use nom::multi::many0;
use nom::sequence::{pair, preceded, terminated, tuple};
use nom::IResult;

use crate::preprocessor::token::Token;
use crate::syntax::parser::{ident, this_ident};
use crate::syntax::BitRangeLsbMsb;
use crate::{SleighError, Span};

use super::TokenFieldAttribute;

#[derive(Clone, Debug)]
pub struct Context {
    pub varnode_name: String,
    pub varnode_span: Span,
    pub fields: Vec<ContextField>,
}

impl Context {
    pub fn parse(input: &[Token]) -> IResult<&[Token], Self, Box<SleighError>> {
        map(
            preceded(
                this_ident("context"),
                cut(pair(ident, many0(ContextField::parse))),
            ),
            |((varnode_name, varnode_span), fields)| Context {
                varnode_name,
                varnode_span: varnode_span.clone(),
                fields,
            },
        )(input)
    }
}

#[derive(Clone, Debug)]
pub struct ContextField {
    pub src: Span,
    pub name: String,
    pub range: BitRangeLsbMsb,
    pub attributes: Vec<ContextFieldAttribute>,
}

impl ContextField {
    pub fn parse(input: &[Token]) -> IResult<&[Token], Self, Box<SleighError>> {
        map(
            tuple((
                terminated(ident, tag!("=")),
                BitRangeLsbMsb::parse,
                many0(ContextFieldAttribute::parse),
            )),
            |((name, name_src), range, attributes)| Self {
                src: name_src.clone(),
                name,
                range,
                attributes,
            },
        )(input)
    }
}

#[derive(Clone, Copy, Debug)]
pub enum ContextFieldAttribute {
    Token(TokenFieldAttribute),
    Noflow,
}

impl ContextFieldAttribute {
    pub(crate) fn from_str(name: &str) -> Option<Self> {
        TokenFieldAttribute::from_str(name)
            .map(Self::Token)
            .or(match name {
                "noflow" => Some(Self::Noflow),
                _ => None,
            })
    }
    fn parse(input_ori: &[Token]) -> IResult<&[Token], Self, Box<SleighError>> {
        let (input, (att, location)) = ident(input_ori)?;
        Self::from_str(&att)
            .map(|att| (input, att))
            .ok_or(nom::Err::Error(Box::new(SleighError::StatementInvalid(
                location.clone(),
            ))))
    }
}