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
|
use binaryninja::{
binaryview::{BinaryView, BinaryViewExt},
command::register,
disassembly::{DisassemblyTextLine, InstructionTextToken, InstructionTextTokenContents},
flowgraph::{BranchType, EdgePenStyle, EdgeStyle, FlowGraph, FlowGraphNode, ThemeColor},
string::BnString,
};
fn test_graph(view: &BinaryView) {
let graph = FlowGraph::new();
let disassembly_lines_a = vec![DisassemblyTextLine::from(vec![
InstructionTextToken::new(BnString::new("Li"), InstructionTextTokenContents::Text),
InstructionTextToken::new(BnString::new("ne"), InstructionTextTokenContents::Text),
InstructionTextToken::new(BnString::new(" 1"), InstructionTextTokenContents::Text),
])];
let node_a = FlowGraphNode::new(&graph);
node_a.set_disassembly_lines(&disassembly_lines_a);
let node_b = FlowGraphNode::new(&graph);
let disassembly_lines_b = vec![DisassemblyTextLine::from(&vec!["Li", "ne", " 2"])];
node_b.set_disassembly_lines(&disassembly_lines_b);
let node_c = FlowGraphNode::new(&graph);
node_c.set_lines(vec!["Line 3", "Line 4", "Line 5"]);
graph.append(&node_a);
graph.append(&node_b);
graph.append(&node_c);
let edge = EdgeStyle::new(EdgePenStyle::DashDotDotLine, 2, ThemeColor::AddressColor);
node_a.add_outgoing_edge(BranchType::UserDefinedBranch, &node_b, &edge);
node_a.add_outgoing_edge(
BranchType::UnconditionalBranch,
&node_c,
&EdgeStyle::default(),
);
view.show_graph_report("Rust Graph Title", graph);
}
#[no_mangle]
pub extern "C" fn UIPluginInit() -> bool {
register(
"Rust Graph Test Title",
"Rust Graph Test Description",
test_graph,
);
true
}
|
