Add support for symbolic relocations

1 files changed, 24 insertions, 0 deletions

diff --git a/binaryview.cpp b/binaryview.cpp
index 6c4418b8..03821071 100644
--- a/binaryview.cpp
+++ b/binaryview.cpp
@@ -449,6 +449,7 @@ BinaryView::BinaryView(const std::string& typeName, FileMetadata* file, BinaryVi
 	view.getAddressSize = GetAddressSizeCallback;
 	view.save = SaveCallback;
 	view.defineRelocation = DefineRelocationCallback;
+	view.defineSymbolRelocation = DefineSymbolRelocationCallback;
 	m_file = file;
 	AddRefForRegistration();
 	m_object = BNCreateCustomBinaryView(typeName.c_str(), m_file->GetObject(),
@@ -621,6 +622,17 @@ void BinaryView::DefineRelocationCallback(void* ctxt, BNArchitecture* arch, BNRe
 }
 
 
+void BinaryView::DefineSymbolRelocationCallback(void* ctxt, BNArchitecture* arch, BNRelocationInfo* info, BNSymbol* sym,
+	uint64_t reloc)
+{
+	BinaryView* view = (BinaryView*)ctxt;
+	BNRelocationInfo curInfo = *info;
+	Architecture* curArch = new CoreArchitecture(arch);
+	Ref<Symbol> curSymbol = new Symbol(sym);
+	return view->PerformDefineRelocation(curArch, curInfo, curSymbol, reloc);
+}
+
+
 bool BinaryView::PerformIsValidOffset(uint64_t offset)
 {
 	uint8_t val;
@@ -700,6 +712,12 @@ void BinaryView::PerformDefineRelocation(Architecture* arch, BNRelocationInfo& i
 }
 
 
+void BinaryView::PerformDefineRelocation(Architecture* arch, BNRelocationInfo& info, Ref<Symbol> target, uint64_t reloc)
+{
+	DefineRelocation(arch, info, target, reloc);
+}
+
+
 void BinaryView::NotifyDataWritten(uint64_t offset, size_t len)
 {
 	BNNotifyDataWritten(m_object, offset, len);
@@ -885,6 +903,12 @@ void BinaryView::DefineRelocation(Architecture* arch, BNRelocationInfo& info, ui
 }
 
 
+void BinaryView::DefineRelocation(Architecture* arch, BNRelocationInfo& info, Ref<Symbol> target, uint64_t reloc)
+{
+	BNDefineSymbolRelocation(m_object, arch->GetObject(), &info, target->GetObject(), reloc);
+}
+
+
 void BinaryView::RegisterNotification(BinaryDataNotification* notify)
 {
 	BNRegisterDataNotification(m_object, notify->GetCallbacks());