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
113
114
115
116
117
|
// Copyright (c) 2015-2017 Vector 35 LLC
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to
// deal in the Software without restriction, including without limitation the
// rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
// FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
// IN THE SOFTWARE.
#include "binaryninjaapi.h"
using namespace std;
using namespace BinaryNinja;
RelocationHandler::RelocationHandler(BNRelocationHandler* handler)
{
m_object = handler;
}
RelocationHandler::RelocationHandler()
{
BNCustomRelocationHandler handler;
handler.context = this;
handler.freeObject = FreeCallback;
handler.getRelocationInfo = GetRelocationInfoCallback;
handler.applyRelocation = ApplyRelocationCallback;
AddRefForRegistration();
m_object = BNCreateRelocationHandler(&handler);
}
void RelocationHandler::FreeCallback(void* ctxt)
{
RelocationHandler* handler = (RelocationHandler*)ctxt;
handler->ReleaseForRegistration();
}
bool RelocationHandler::GetRelocationInfoCallback(void* ctxt, BNBinaryView* view, BNArchitecture* arch,
BNRelocationInfo* result, size_t resultCount)
{
RelocationHandler* handler = (RelocationHandler*)ctxt;
Ref<BinaryView> viewObj = new BinaryView(BNNewViewReference(view));
Ref<Architecture> archObj = new CoreArchitecture(arch);
if (!result)
return false;
vector<BNRelocationInfo> resultVector(&result[0], &result[resultCount]);
bool success = handler->GetRelocationInfo(viewObj, archObj, resultVector);
for (size_t i = 0; i < resultCount; i++)
result[i] = resultVector[i];
return success;
}
bool RelocationHandler::ApplyRelocationCallback(void* ctxt, BNBinaryView* view, BNArchitecture* arch, BNRelocation* reloc,
uint8_t* dest, size_t len)
{
RelocationHandler* handler = (RelocationHandler*)ctxt;
Ref<Architecture> archObj = new CoreArchitecture(arch);
Ref<BinaryView> viewObj = new BinaryView(BNNewViewReference(view));
Ref<Relocation> relocObj = new Relocation(BNNewRelocationReference(reloc));
return handler->ApplyRelocation(viewObj, archObj, relocObj, dest, len);
}
bool RelocationHandler::GetRelocationInfo(Ref<BinaryView> view, Ref<Architecture> arch, std::vector<BNRelocationInfo>& result)
{
(void)view;
(void)arch;
(void)result;
return false;
}
bool RelocationHandler::ApplyRelocation(Ref<BinaryView> view, Ref<Architecture> arch, Ref<Relocation> reloc, uint8_t* dest,
size_t len)
{
return BNRelocationHandlerDefaultApplyRelocation(m_object, view->GetObject(), arch->GetObject(), BNNewRelocationReference(reloc->GetObject()), dest, len);
}
CoreRelocationHandler::CoreRelocationHandler(BNRelocationHandler* handler) : RelocationHandler(handler)
{
}
bool CoreRelocationHandler::ApplyRelocation(Ref<BinaryView> view, Ref<Architecture> arch, Ref<Relocation> reloc, uint8_t* dest,
size_t len)
{
return BNRelocationHandlerApplyRelocation(m_object, view->GetObject(), arch->GetObject(), BNNewRelocationReference(reloc->GetObject()), dest, len);
}
bool CoreRelocationHandler::GetRelocationInfo(Ref<BinaryView> view, Ref<Architecture> arch, std::vector<BNRelocationInfo>& result)
{
BNRelocationInfo* results = new BNRelocationInfo[result.size()];
for (size_t i = 0; i < result.size(); i++)
results[i] = result[i];
bool status = BNRelocationHandlerGetRelocationInfo(m_object, view->GetObject(), arch->GetObject(), results,
result.size());
for (size_t i = 0; i < result.size(); i++)
result[i] = results[i];
return status;
}
|
