heretek 0.6.0

GDB TUI Dashboard for the understanding of vast knowledge
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

use crate::mi::parse_asm_insns_values;
use crate::register::RegisterStorage;
use crate::{State, Written};

/// `MIResponse::ExecResult`, key: "asm_insns"
pub fn recv_exec_result_asm_insns(state: &mut State, asm: &String) {
    if state.written.is_empty() {
        return;
    }
    let last_written = state.written.pop_front().unwrap();
    // TODO: change to match
    if let Written::AsmAtPc = last_written {
        state.asm = parse_asm_insns_values(asm).clone();
    }
    if let Written::SymbolAtAddrRegister((base_reg, _n)) = &last_written {
        for RegisterStorage { name: _, register, deref } in state.registers.iter_mut() {
            if let Some(reg) = register {
                if reg.number == *base_reg {
                    let new_asms = parse_asm_insns_values(asm);
                    if !new_asms.is_empty() {
                        if let Some(func_name) = &new_asms[0].func_name {
                            deref.final_assembly = format!(
                                "{}+{} ({})",
                                func_name.to_owned(),
                                new_asms[0].offset,
                                new_asms[0].inst
                            );
                        } else {
                            deref.final_assembly = new_asms[0].inst.to_owned();
                        }
                    }
                }
            }
        }
    }
    if let Written::SymbolAtAddrStack(deref) = last_written {
        let key = u64::from_str_radix(&deref, 16).unwrap();
        if let Some(deref) = state.stack.get_mut(&key) {
            let new_asms = parse_asm_insns_values(asm);
            if !new_asms.is_empty() {
                // Try and show func_name, otherwise asm
                if let Some(func_name) = &new_asms[0].func_name {
                    deref.final_assembly = format!(
                        "{}+{} ({})",
                        func_name.to_owned(),
                        new_asms[0].offset,
                        new_asms[0].inst
                    );
                } else {
                    deref.final_assembly = new_asms[0].inst.to_owned();
                }
            }
        }
    }
}