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
|
use log::{info, LevelFilter};
use binaryninja::architecture::CoreArchitecture;
use binaryninja::binaryview::BinaryView;
use binaryninja::{command, logger};
use binaryninja::command::Command;
use binaryninja::demangle::{Demangler, CustomDemangler};
use binaryninja::rc::Ref;
use binaryninja::types::{QualifiedName, Type};
struct TestDemangler;
impl CustomDemangler for TestDemangler {
fn is_mangled_string(&self, name: &str) -> bool {
name == "test_name" || name == "test_name2"
}
fn demangle(&self, _arch: &CoreArchitecture, name: &str, _view: Option<Ref<BinaryView>>) -> Result<(Option<Ref<Type>>, QualifiedName), ()> {
match name {
"test_name" => Ok((Some(Type::bool()), QualifiedName::from(vec!["test_name"]))),
"test_name2" => Ok((None, QualifiedName::from(vec!["test_name2", "aaa"]))),
_ => Err(()),
}
}
}
struct DemangleCommand;
impl Command for DemangleCommand {
fn action(&self, view: &BinaryView) {
for d in Demangler::list().iter() {
info!("{}", d.name());
info!("{}", d.is_mangled_string("__ZN1AC2Ei"));
info!("{:?}", d.demangle(
&CoreArchitecture::by_name("x86_64").expect("x86 exists"),
"__ZN1AC2Ei",
Some(view)
));
info!("{:?}", d.demangle(
&CoreArchitecture::by_name("x86_64").expect("x86 exists"),
"test_name",
None
));
info!("{:?}", d.demangle(
&CoreArchitecture::by_name("x86_64").expect("x86 exists"),
"test_name2",
None
));
}
}
fn valid(&self, _view: &BinaryView) -> bool {
true
}
}
#[no_mangle]
pub extern "C" fn CorePluginInit() -> bool {
let _ = logger::init(LevelFilter::Info);
Demangler::register("Test", TestDemangler {});
command::register("Demangle Test", "Test", DemangleCommand {});
true
}
|
