feat[python]: Support for custom RelocationHandler and

BinaryView.define_relocation

Author: nullableVoidPtr

python/architecture.py

@@ -42,6 +42,7 @@
 from . import variable
 from . import basicblock
 from . import log
+from . import relocation
 
 RegisterIndex = NewType('RegisterIndex', int)
 RegisterStackIndex = NewType('RegisterStackIndex', int)
@@ -2590,6 +2591,28 @@ def skip_and_return_value(self, data: bytes, addr: int, value: int) -> Optional[
 		"""
 		return NotImplemented
 
+	def register_relocation_handler(self, view_name: str, handler: 'relocation.RelocationHandler') -> None:
+		"""
+		``register_relocation_handler`` registers a new relocation handler for the Architecture.
+
+		:param RelocationHandler handler: RelocationHandler object to be registered
+		:rtype: None
+		"""
+		core.BNArchitectureRegisterRelocationHandler(self.handle, core.cstr(view_name), handler.handle)
+
+	def get_relocation_handler(self, view_name: str) -> Optional['relocation.RelocationHandler']:
+		"""
+		``get_relocation_handler`` gets a relocation handler by its view name for the Architecture.
+
+		:param str view_name: the name of the corresponding view
+		:rtype: Optional['relocation.RelocationHandler']
+		"""
+		handler_handle = core.BNArchitectureGetRelocationHandler(self.handle, core.cstr(view_name))
+		if handler_handle is None:
+			return None
+
+		return relocation.RelocationHandler(handler_handle)
+
 	def register_calling_convention(self, cc: 'callingconvention.CallingConvention') -> None:
 		"""
 		``register_calling_convention`` registers a new calling convention for the Architecture.

python/binaryview.py

@@ -82,6 +82,7 @@
 from . import externallibrary
 from . import undo
 from . import stringrecognizer
+from . import relocation
 
 
 PathType = Union[str, os.PathLike]
@@ -2124,101 +2125,6 @@ def data(self, value: str) -> None:
 		core.BNTagSetData(self.handle, value)
 
 
-@dataclass
-class RelocationInfo:
-	type: RelocationType
-	pc_relative: bool
-	base_relative: bool
-	base: int
-	size: int
-	truncate_size: int
-	native_type: int
-	addend: int
-	has_sign: bool
-	implicit_addend: bool
-	external: bool
-	symbol_index: int
-	section_index: int
-	address: int
-	target: int
-	data_relocation: bool
-
-	def __init__(self, info: core.BNRelocationInfo):
-		self.type = RelocationType(info.type)
-		self.pc_relative = info.pcRelative
-		self.base_relative = info.baseRelative
-		self.base = info.base
-		self.size = info.size
-		self.truncate_size = info.truncateSize
-		self.native_type = info.nativeType
-		self.addend = info.addend
-		self.has_sign = info.hasSign
-		self.implicit_addend = info.implicitAddend
-		self.external = info.external
-		self.symbol_index = info.symbolIndex
-		self.section_index = info.sectionIndex
-		self.address = info.address
-		self.target = info.target
-		self.data_relocation = info.dataRelocation
-
-
-class Relocation:
-	def __init__(self, handle: core.BNRelocationHandle):
-		self.handle = handle
-
-	def __del__(self):
-		if core is not None:
-			core.BNFreeRelocation(self.handle)
-
-	def __repr__(self):
-		if self.symbol is None:
-			return f"<Relocation: {self.target:#x}>"
-		try:
-			return f"<Relocation: \"{self.symbol.full_name}\" @ {self.target:#x}>"
-		except UnicodeDecodeError:
-			return f"<Relocation: \"{self.symbol.raw_bytes}\" @ {self.target:#x}>"
-
-	def __eq__(self, other):
-		if not isinstance(other, self.__class__):
-			return NotImplemented
-		return ctypes.addressof(self.handle.contents) == ctypes.addressof(other.handle.contents)
-
-	def __ne__(self, other):
-		if not isinstance(other, self.__class__):
-			return NotImplemented
-		return not (self == other)
-
-	def __hash__(self):
-		return hash(ctypes.addressof(self.handle.contents))
-
-	@property
-	def info(self) -> RelocationInfo:
-		return RelocationInfo(core.BNRelocationGetInfo(self.handle))
-
-	@property
-	def arch(self) -> Optional['architecture.Architecture']:
-		"""The architecture associated with the :py:class:`Relocation` (read/write)"""
-		core_arch = core.BNRelocationGetArchitecture(self.handle)
-		return architecture.CoreArchitecture._from_cache(handle=core_arch)
-
-	@property
-	def target(self) -> int:
-		"""Where the reloc needs to point to"""
-		return core.BNRelocationGetTarget(self.handle)
-
-	@property
-	def reloc(self) -> int:
-		"""The actual pointer that needs to be relocated"""
-		return core.BNRelocationGetReloc(self.handle)
-
-	@property
-	def symbol(self) -> Optional['_types.CoreSymbol']:
-		core_symbol = core.BNRelocationGetSymbol(self.handle)
-		if core_symbol is None:
-			return None
-		return _types.CoreSymbol(core_symbol)
-
-
 class _BinaryViewAssociatedDataStore(associateddatastore._AssociatedDataStore):
 	_defaults = {}
 
@@ -4020,7 +3926,19 @@ def max_function_size_for_analysis(self) -> int:
 	def max_function_size_for_analysis(self, size: int) -> None:
 		core.BNSetMaxFunctionSizeForAnalysis(self.handle, size)
 
-	def relocations_at(self, addr: int) -> List[Relocation]:
+	def define_relocation(self, info: 'relocation.RelocationInfo', target: 'int | _types.Symbol', reloc: int, arch: Optional['architecture.Architecture'] = None):
+		if arch is None:
+			if self.arch is None:
+				raise Exception("Can not define relocation with no Architecture specified")
+			arch = self.arch
+
+		_info = info._to_core_struct()
+		if isinstance(target, int):
+			core.BNDefineRelocation(self.handle, arch.handle, _info, target, reloc)
+		else:
+			core.BNDefineSymbolRelocation(self.handle, arch.handle, _info, target.handle, reloc)
+
+	def relocations_at(self, addr: int) -> List['relocation.Relocation']:
 		"""List of relocations for a given address"""
 		count = ctypes.c_ulonglong()
 		relocs = core.BNGetRelocationsAt(self.handle, addr, count)
@@ -4030,7 +3948,7 @@ def relocations_at(self, addr: int) -> List[Relocation]:
 			for i in range(0, count.value):
 				reloc_handle = core.BNNewRelocationReference(relocs[i])
 				assert reloc_handle is not None, "core.BNNewRelocationReference is not None"
-				result.append(Relocation(reloc_handle))
+				result.append(relocation.Relocation(reloc_handle))
 			return result
 		finally:
 			core.BNFreeRelocationList(relocs, count)