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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
use binaryninjacore_sys::*;
use std::{ffi::CStr, result};
use crate::architecture::CoreArchitecture;
use crate::string::{BnStrCompatible, BnString};
use crate::types::Type;
use crate::rc::*;
pub type Result<R> = result::Result<R, ()>;
pub fn demangle_gnu3<S: BnStrCompatible>(
arch: &CoreArchitecture,
mangled_name: S,
simplify: bool,
) -> Result<(Option<Ref<Type>>, Vec<String>)> {
let mangled_name_bwn = mangled_name.as_bytes_with_nul();
let mangled_name_ptr = mangled_name_bwn.as_ref();
let mut out_type: *mut BNType = unsafe { std::mem::zeroed() };
let mut out_name: *mut *mut std::os::raw::c_char = unsafe { std::mem::zeroed() };
let mut out_size: usize = 0;
let res = unsafe {
BNDemangleGNU3(
arch.0,
mangled_name_ptr.as_ptr() as *const i8,
&mut out_type,
&mut out_name,
&mut out_size,
simplify,
)
};
if !res || out_size == 0 {
let cstr = match CStr::from_bytes_with_nul(mangled_name_ptr) {
Ok(cstr) => cstr,
Err(_) => {
log::error!("demangle_gnu3: failed to parse mangled name");
return Err(());
}
};
return Ok((None, vec![cstr.to_string_lossy().into_owned()]));
}
let out_type = match out_type.is_null() {
true => {
log::debug!("demangle_gnu3: out_type is NULL");
None
}
false => Some(unsafe { Type::ref_from_raw(out_type) }),
};
if out_name.is_null() {
log::error!("demangle_gnu3: out_name is NULL");
return Err(());
}
let names = unsafe { ArrayGuard::<BnString>::new(out_name, out_size, ()) }
.iter()
.map(|name| name.to_string())
.collect();
unsafe { BNFreeDemangledName(&mut out_name, out_size) };
Ok((out_type, names))
}
pub fn demangle_ms<S: BnStrCompatible>(
arch: &CoreArchitecture,
mangled_name: S,
simplify: bool,
) -> Result<(Option<Ref<Type>>, Vec<String>)> {
let mangled_name_bwn = mangled_name.as_bytes_with_nul();
let mangled_name_ptr = mangled_name_bwn.as_ref();
let mut out_type: *mut BNType = unsafe { std::mem::zeroed() };
let mut out_name: *mut *mut std::os::raw::c_char = unsafe { std::mem::zeroed() };
let mut out_size: usize = 0;
let res = unsafe {
BNDemangleMS(
arch.0,
mangled_name_ptr.as_ptr() as *const i8,
&mut out_type,
&mut out_name,
&mut out_size,
simplify,
)
};
if !res || out_size == 0 {
let cstr = match CStr::from_bytes_with_nul(mangled_name_ptr) {
Ok(cstr) => cstr,
Err(_) => {
log::error!("demangle_ms: failed to parse mangled name");
return Err(());
}
};
return Ok((None, vec![cstr.to_string_lossy().into_owned()]));
}
let out_type = match out_type.is_null() {
true => {
log::debug!("demangle_ms: out_type is NULL");
None
}
false => Some(unsafe { Type::ref_from_raw(out_type) }),
};
if out_name.is_null() {
log::error!("demangle_ms: out_name is NULL");
return Err(());
}
let names = unsafe { ArrayGuard::<BnString>::new(out_name, out_size, ()) }
.iter()
.map(|name| name.to_string())
.collect();
unsafe { BNFreeDemangledName(&mut out_name, out_size) };
Ok((out_type, names))
}
|
