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
use crate::architecture::CoreArchitecture;
use crate::function::Function;
use crate::rc::{CoreArrayProvider, CoreArrayWrapper, CoreOwnedArrayProvider, Ref};
use binaryninjacore_sys::{BNFreeCodeReferences, BNFreeDataReferences, BNReferenceSource};
use std::mem::ManuallyDrop;
#[derive(Debug)]
pub struct CodeReference {
arch: CoreArchitecture,
func: ManuallyDrop<Ref<Function>>,
pub address: u64,
}
pub struct DataReference {
pub address: u64,
}
impl CodeReference {
pub(crate) unsafe fn new(handle: &BNReferenceSource) -> Self {
let func = ManuallyDrop::new(Function::from_raw(handle.func));
let arch = CoreArchitecture::from_raw(handle.arch);
let address = handle.addr;
Self {
func,
arch,
address,
}
}
}
impl<'a> CodeReference {
pub fn function(&'a self) -> &'a Function {
self.func.as_ref()
}
pub fn architecture(&self) -> CoreArchitecture {
self.arch
}
}
impl CoreArrayProvider for CodeReference {
type Raw = BNReferenceSource;
type Context = ();
}
unsafe impl CoreOwnedArrayProvider for CodeReference {
unsafe fn free(raw: *mut Self::Raw, count: usize, _context: &Self::Context) {
BNFreeCodeReferences(raw, count)
}
}
unsafe impl<'a> CoreArrayWrapper<'a> for CodeReference {
type Wrapped = CodeReference;
unsafe fn wrap_raw(raw: &'a Self::Raw, _context: &'a Self::Context) -> Self::Wrapped {
CodeReference::new(raw)
}
}
impl CoreArrayProvider for DataReference {
type Raw = u64;
type Context = ();
}
unsafe impl CoreOwnedArrayProvider for DataReference {
unsafe fn free(raw: *mut Self::Raw, _count: usize, _context: &Self::Context) {
BNFreeDataReferences(raw)
}
}
unsafe impl<'a> CoreArrayWrapper<'a> for DataReference {
type Wrapped = DataReference;
unsafe fn wrap_raw(raw: &'a Self::Raw, _context: &'a Self::Context) -> Self::Wrapped {
DataReference { address: *raw }
}
}