i8051-disassembler 0.20.0

An i8051 disassembler, useful for structured disassembly.
Documentation 
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

use std::{
    collections::{BTreeMap, HashMap},
    ops,
};

use crate::address::{AddressSpace, AddressValue, PhysicalAddr};

#[derive(Default)]
pub(crate) struct ImplicitLabels {
    labels: BTreeMap<AddressSpace, Labels>,
}

impl ops::Deref for ImplicitLabels {
    type Target = BTreeMap<AddressSpace, Labels>;

    fn deref(&self) -> &Self::Target {
        &self.labels
    }
}

#[derive(Default)]
pub(crate) struct Labels {
    labels: BTreeMap<AddressValue, String>,
}

impl ops::Deref for Labels {
    type Target = BTreeMap<AddressValue, String>;

    fn deref(&self) -> &Self::Target {
        &self.labels
    }
}

#[derive(Default)]
pub(crate) struct LabelCollector {
    labels: HashMap<PhysicalAddr, Option<String>>,
}

impl LabelCollector {
    pub fn collect(&mut self, space: AddressSpace, addr: AddressValue, label: Option<String>) {
        self.labels.insert(
            PhysicalAddr {
                space,
                offset: addr,
            },
            label,
        );
    }

    pub fn into_implicit_labels(self) -> ImplicitLabels {
        let mut implicit_labels = BTreeMap::new();
        for (addr, ref_name) in self.labels {
            if let Some(ref_name) = ref_name {
                implicit_labels
                    .entry(addr.space)
                    .or_insert_with(Labels::default)
                    .labels
                    .insert(addr.offset, ref_name);
            } else {
                implicit_labels
                    .entry(addr.space)
                    .or_insert_with(Labels::default)
                    .labels
                    .insert(addr.offset, format!("loc_{addr:04X}", addr = addr.offset));
            }
        }
        ImplicitLabels {
            labels: implicit_labels,
        }
    }
}