blob: 636a8856bed8a113bbbcddff43af10398edc9a42 (
plain)
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
|
#include "binaryninjaapi.h"
using namespace BinaryNinja;
using namespace std;
BNBinaryView* BinaryViewType::CreateCallback(void* ctxt, BNBinaryView* data)
{
BinaryViewType* type = (BinaryViewType*)ctxt;
Ref<BinaryView> view = new CoreBinaryView(BNNewViewReference(data));
Ref<BinaryView> result = type->Create(view);
return BNNewViewReference(result->GetViewObject());
}
bool BinaryViewType::IsValidCallback(void* ctxt, BNBinaryView* data)
{
BinaryViewType* type = (BinaryViewType*)ctxt;
Ref<BinaryView> view = new CoreBinaryView(BNNewViewReference(data));
return type->IsTypeValidForData(view);
}
BinaryViewType::BinaryViewType(BNBinaryViewType* type): m_type(type)
{
}
BinaryViewType::BinaryViewType(const string& name, const string& longName):
m_type(nullptr), m_nameForRegister(name), m_longNameForRegister(longName)
{
}
BinaryViewType::~BinaryViewType()
{
BNFreeBinaryViewType(m_type);
}
void BinaryViewType::Register(BinaryViewType* type)
{
BNCustomBinaryViewType callbacks;
callbacks.context = type;
callbacks.create = CreateCallback;
callbacks.isValidForData = IsValidCallback;
type->m_type = BNRegisterBinaryViewType(type->m_nameForRegister.c_str(),
type->m_longNameForRegister.c_str(), &callbacks);
}
Ref<BinaryViewType> BinaryViewType::GetByName(const string& name)
{
BNBinaryViewType* type = BNGetBinaryViewTypeByName(name.c_str());
if (!type)
return nullptr;
return new CoreBinaryViewType(type);
}
vector<Ref<BinaryViewType>> BinaryViewType::GetViewTypesForData(BinaryView* data)
{
BNBinaryViewType** types;
size_t count;
types = BNGetBinaryViewTypesForData(data->GetViewObject(), &count);
vector<Ref<BinaryViewType>> result;
for (size_t i = 0; i < count; i++)
result.push_back(new CoreBinaryViewType(BNNewViewTypeReference(types[i])));
BNFreeBinaryViewTypeList(types, count);
return result;
}
string BinaryViewType::GetName()
{
char* contents = BNGetBinaryViewTypeName(m_type);
string result = contents;
BNFreeString(contents);
return result;
}
string BinaryViewType::GetLongName()
{
char* contents = BNGetBinaryViewTypeLongName(m_type);
string result = contents;
BNFreeString(contents);
return result;
}
CoreBinaryViewType::CoreBinaryViewType(BNBinaryViewType* type): BinaryViewType(type)
{
}
BinaryView* CoreBinaryViewType::Create(BinaryView* data)
{
BNBinaryView* view = BNCreateBinaryViewOfType(m_type, data->GetViewObject());
if (!view)
return nullptr;
return new CoreBinaryView(view);
}
bool CoreBinaryViewType::IsTypeValidForData(BinaryView* data)
{
return BNIsBinaryViewTypeValidForData(m_type, data->GetViewObject());
}
|
