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
|
use binaryninja::binary_view::BinaryView;
use binaryninja::debuginfo::*;
use binaryninja::headless::Session;
use binaryninja::types::{MemberAccess, MemberScope, StructureBuilder, Type, TypeBuilder};
use std::path::PathBuf;
use std::sync::atomic::{AtomicBool, Ordering};
static TEST_PARSER_ENABLED: AtomicBool = AtomicBool::new(true);
struct TestDebugInfoParser;
impl CustomDebugInfoParser for TestDebugInfoParser {
fn is_valid(&self, _view: &BinaryView) -> bool {
TEST_PARSER_ENABLED.load(Ordering::SeqCst)
}
fn parse_info(
&self,
debug_info: &mut DebugInfo,
_view: &BinaryView,
_debug_file: &BinaryView,
_progress: Box<dyn Fn(usize, usize) -> Result<(), ()>>,
) -> bool {
let test_type = TypeBuilder::int(4, true).finalize();
let test_struct = StructureBuilder::new()
.append(
&test_type,
"myfield",
MemberAccess::PublicAccess,
MemberScope::NoScope,
)
.finalize();
let new_type = TypeBuilder::structure(&test_struct).finalize();
debug_info.add_type("test_dbg", &new_type, &[]);
let func_type = Type::function(&test_type, vec![], true);
let test_func = DebugFunctionInfo::new(
None,
None,
Some("test_func".to_string()),
Some(func_type),
Some(0x3b440),
None,
vec![],
vec![],
);
debug_info.add_function(&test_func);
true
}
}
#[test]
fn test_debug_info() {
let _session = Session::new().expect("Failed to initialize session");
let out_dir = env!("OUT_DIR").parse::<PathBuf>().unwrap();
// Register test parser
DebugInfoParser::register("test", TestDebugInfoParser);
// Make sure it exists.
let _parser = DebugInfoParser::from_name("test").expect("Debug info test parser exists");
{
let view = binaryninja::load(out_dir.join("atox.obj")).expect("Failed to create view");
view.type_by_name("test_dbg")
.expect("Debug info test type exists");
let func = view
.function_at(&view.default_platform().unwrap(), 0x3b440)
.expect("Debug info test function exists");
assert_eq!(func.symbol().raw_name().to_string_lossy(), "test_func");
view.file().close();
}
// Disable the parser for other tests, so we don't introduce any unwanted behavior.
TEST_PARSER_ENABLED.store(false, Ordering::SeqCst);
}
|
