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
|
pub mod container;
pub mod function;
pub mod guid;
pub mod type_reference;
pub use function::*;
pub use guid::*;
pub use type_reference::*;
use binaryninja::binary_view::BinaryView;
use binaryninja::function::Function as BNFunction;
use binaryninja::object_destructor::{register_object_destructor, ObjectDestructor};
use binaryninja::rc::Guard;
use binaryninja::rc::Ref as BNRef;
use std::hash::{DefaultHasher, Hash, Hasher};
pub fn register_cache_destructor() {
let destructor = register_object_destructor(CacheDestructor);
std::mem::forget(destructor);
}
/// A unique view ID, used for caching.
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq, PartialOrd, Ord)]
pub struct ViewID(u64);
impl From<&BinaryView> for ViewID {
fn from(value: &BinaryView) -> Self {
let mut hasher = DefaultHasher::new();
hasher.write_u64(value.original_image_base());
hasher.write_usize(value.file().session_id().0);
Self(hasher.finish())
}
}
impl From<BNRef<BinaryView>> for ViewID {
fn from(value: BNRef<BinaryView>) -> Self {
Self::from(value.as_ref())
}
}
impl From<Guard<'_, BinaryView>> for ViewID {
fn from(value: Guard<'_, BinaryView>) -> Self {
Self::from(value.as_ref())
}
}
/// A unique function ID, used for caching.
#[derive(Copy, Clone, Debug, Hash, PartialEq, Eq, PartialOrd, Ord)]
pub struct FunctionID(u64);
impl From<&BNFunction> for FunctionID {
fn from(value: &BNFunction) -> Self {
let mut hasher = DefaultHasher::new();
hasher.write_u64(value.start());
hasher.write_u64(value.lowest_address());
hasher.write_u64(value.highest_address());
Self(hasher.finish())
}
}
impl From<BNRef<BNFunction>> for FunctionID {
fn from(value: BNRef<BNFunction>) -> Self {
Self::from(value.as_ref())
}
}
impl From<Guard<'_, BNFunction>> for FunctionID {
fn from(value: Guard<'_, BNFunction>) -> Self {
Self::from(value.as_ref())
}
}
pub struct CacheDestructor;
impl ObjectDestructor for CacheDestructor {
fn destruct_view(&self, view: &BinaryView) {
clear_type_ref_cache(view);
tracing::debug!("Removed WARP caches for {}", view.file());
}
}
|
