# Copyright (c) 2015-2021 Vector 35 Inc # # 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. import ctypes from typing import Generator, List, Union, Mapping, Tuple, Optional from dataclasses import dataclass import uuid from abc import abstractmethod # Binary Ninja components from . import _binaryninjacore as core from .enums import (StructureVariant, SymbolType, SymbolBinding, TypeClass, NamedTypeReferenceClass, ReferenceType, VariableSourceType, TypeReferenceType, MemberAccess, MemberScope) from . import callingconvention from . import function as _function from . import variable from . import architecture from . import types from . import binaryview from . import platform as _platform from . import typelibrary QualifiedNameType = Union[List[str], str, 'QualifiedName', List[bytes]] BoolWithConfidenceType = Union[bool, 'BoolWithConfidence'] SizeWithConfidenceType = Union[int, 'SizeWithConfidence'] OffsetWithConfidenceType = Union[int, 'OffsetWithConfidence'] ParamsType = Union[List['Type'], List['FunctionParameter'], List[Tuple['Type', str]]] MembersType = Union[List['StructureMember'], List[Tuple['Type', str]]] EnumMembersType = Union[List[Tuple[str,int]], List[str], List['EnumerationMember']] SomeType = Union['TypeBuilder', 'Type'] TypeContainer = Union['binaryview.BinaryView', 'typelibrary.TypeLibrary'] NameSpaceType = Optional[Union[str, List[str], 'NameSpace']] # The following are needed to prevent the type checker from getting # confused as we have member functions in `Type` named the same thing _int = int _bool = bool MemberName = str MemberIndex = int MemberOffset = int class TypeCreateException(ValueError): pass class QualifiedName: def __init__(self, name:QualifiedNameType=[]): self._name:List[str] = [] if isinstance(name, str): self._name = [name] elif isinstance(name, self.__class__): self._name = name._name elif isinstance(name, (list, tuple)): for i in name: if isinstance(i, bytes): self._name.append(i.decode("utf-8")) else: self._name.append(str(i)) def __str__(self): return "::".join(self.name) def __repr__(self): return repr(str(self)) def __len__(self): return len(self.name) def __eq__(self, other): if isinstance(other, str): return str(self) == other elif isinstance(other, list): return self.name == other elif isinstance(other, self.__class__): return self.name == other.name return NotImplemented def __ne__(self, other): if isinstance(other, str): return str(self) != other elif isinstance(other, list): return self.name != other elif isinstance(other, self.__class__): return self.name != other.name return NotImplemented def __lt__(self, other): if isinstance(other, self.__class__): return self.name < other.name return NotImplemented def __le__(self, other): if isinstance(other, self.__class__): return self.name <= other.name return NotImplemented def __gt__(self, other): if isinstance(other, self.__class__): return self.name > other.name return NotImplemented def __ge__(self, other): if isinstance(other, self.__class__): return self.name >= other.name return NotImplemented def __cmp__(self, other): if not isinstance(other, self.__class__): return NotImplemented if self == other: return 0 if self < other: return -1 return 1 def __hash__(self): return hash(str(self)) def __getitem__(self, key): return self.name[key] def __iter__(self): return iter(self.name) def _to_core_struct(self) -> core.BNQualifiedName: result = core.BNQualifiedName() name_list = (ctypes.c_char_p * len(self.name))() for i in range(0, len(self.name)): name_list[i] = self.name[i].encode("utf-8") result.name = name_list result.nameCount = len(self.name) return result @staticmethod def _from_core_struct(name): result = [] for i in range(0, name.nameCount): result.append(name.name[i].decode("utf-8")) return QualifiedName(result) @property def name(self) -> List[str]: return self._name @name.setter def name(self, value:List[str]) -> None: self._name = value @dataclass(frozen=True) class TypeReferenceSource: name:QualifiedName offset:int ref_type:TypeReferenceType def __repr__(self): if self.ref_type == TypeReferenceType.DirectTypeReferenceType: s = 'direct' elif self.ref_type == TypeReferenceType.IndirectTypeReferenceType: s = 'indirect' else: s = 'unknown' return '' % (self.name, self.offset, s) class NameSpace(QualifiedName): def __str__(self): return ":".join(self.name) def _to_core_struct(self) -> core.BNNameSpace: result = core.BNNameSpace() name_list = (ctypes.c_char_p * len(self.name))() for i in range(0, len(self.name)): name_list[i] = self.name[i].encode('charmap') result.name = name_list result.nameCount = len(self.name) return result @staticmethod def _from_core_struct(name): result = [] for i in range(0, name.nameCount): result.append(name.name[i].decode("utf-8")) return NameSpace(result) @staticmethod def get_core_struct(name:Optional[Union[str, List[str], 'NameSpace']]) -> Optional[core.BNNameSpace]: if name is None: return None if isinstance(name, NameSpace): return name._to_core_struct() else: return NameSpace(name)._to_core_struct() class CoreSymbol: def __init__(self, handle:core.BNSymbolHandle): self._handle = handle def __del__(self): if core is not None: core.BNFreeSymbol(self._handle) def __repr__(self): return f"<{self.type.name}: \"{self.full_name}\" @ {self.address:#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 type(self) -> SymbolType: """Symbol type (read-only)""" return SymbolType(core.BNGetSymbolType(self._handle)) @property def binding(self) -> SymbolBinding: """Symbol binding (read-only)""" return SymbolBinding(core.BNGetSymbolBinding(self._handle)) @property def namespace(self) -> 'NameSpace': """Symbol namespace (read-only)""" ns = core.BNGetSymbolNameSpace(self._handle) result = NameSpace._from_core_struct(ns) core.BNFreeNameSpace(ns) return result @property def name(self) -> str: """Symbol name (read-only)""" return core.BNGetSymbolRawName(self._handle) @property def short_name(self) -> str: """Symbol short name (read-only)""" return core.BNGetSymbolShortName(self._handle) @property def full_name(self) -> str: """Symbol full name (read-only)""" return core.BNGetSymbolFullName(self._handle) @property def raw_name(self) -> str: """Symbol raw name (read-only)""" return core.BNGetSymbolRawName(self._handle) @property def address(self) -> int: """Symbol address (read-only)""" return core.BNGetSymbolAddress(self._handle) @property def ordinal(self) -> int: """Symbol ordinal (read-only)""" return core.BNGetSymbolOrdinal(self._handle) @property def auto(self) -> bool: return core.BNIsSymbolAutoDefined(self._handle) @property def handle(self): return self._handle class Symbol(CoreSymbol): """ Symbols are defined as one of the following types: =========================== ============================================================== SymbolType Description =========================== ============================================================== FunctionSymbol Symbol for function that exists in the current binary ImportAddressSymbol Symbol defined in the Import Address Table ImportedFunctionSymbol Symbol for a function that is not defined in the current binary DataSymbol Symbol for data in the current binary ImportedDataSymbol Symbol for data that is not defined in the current binary ExternalSymbol Symbols for data and code that reside outside the BinaryView LibraryFunctionSymbol Symbols for external functions outside the library =========================== ============================================================== """ def __init__(self, sym_type, addr, short_name, full_name=None, raw_name=None, binding=None, namespace=None, ordinal=0): if isinstance(sym_type, str): sym_type = SymbolType[sym_type] if full_name is None: full_name = short_name if raw_name is None: raw_name = full_name if binding is None: binding = SymbolBinding.NoBinding _namespace = NameSpace.get_core_struct(namespace) _handle = core.BNCreateSymbol(sym_type, short_name, full_name, raw_name, addr, binding, _namespace, ordinal) assert _handle is not None, "core.BNCreateSymbol return None" super(Symbol, self).__init__(_handle) @dataclass class FunctionParameter: type:SomeType name:str = "" location:Optional['variable.VariableNameAndType'] = None def __repr__(self): if (self.location is not None) and (self.location.name != self.name): return "%s %s%s @ %s" % (self.type.immutable_copy().get_string_before_name(), self.name, self.type.immutable_copy().get_string_after_name(), self.location.name) return "%s %s%s" % (self.type.immutable_copy().get_string_before_name(), self.name, self.type.immutable_copy().get_string_after_name()) def immutable_copy(self) -> 'FunctionParameter': return FunctionParameter(self.type.immutable_copy(), self.name, self.location) def mutable_copy(self) -> 'FunctionParameter': return FunctionParameter(self.type.mutable_copy(), self.name, self.location) @dataclass(frozen=True) class OffsetWithConfidence: value:int confidence:int=core.max_confidence def __int__(self): return self.value def __eq__(self, other): if not isinstance(other, self.__class__): return self.value == int(other) else: return (self.value, self.confidence) == (other.value, other.confidence) def __ne__(self, other): return not (self == other) def __gt__(self, other): return self.value > int(other) def __le__(self, other): return self.value <= int(other) def __ge__(self, other): return self.value >= int(other) def __lt__(self, other): return self.value < int(other) def _to_core_struct(self) -> core.BNOffsetWithConfidence: result = core.BNOffsetWithConfidence() result.value = self.value result.confidence = self.confidence return result @classmethod def from_core_struct(cls, core_struct:core.BNOffsetWithConfidence) -> 'OffsetWithConfidence': return cls(core_struct.value, core_struct.confidence) @staticmethod def get_core_struct(value:OffsetWithConfidenceType) -> core.BNOffsetWithConfidence: if isinstance(value, OffsetWithConfidence): return value._to_core_struct() else: return OffsetWithConfidence(value)._to_core_struct() @dataclass(frozen=True) class BoolWithConfidence: value:bool confidence:int=core.max_confidence def __eq__(self, other): if not isinstance(other, self.__class__): return self.value == bool(other) else: return (self.value, self.confidence) == (other.value, other.confidence) def __ne__(self, other): return not (self == other) def __gt__(self, other): return self.confidence > other.confidence def __le__(self, other): return self.confidence <= other.confidence def __ge__(self, other): return self.confidence >= other.confidence def __lt__(self, other): return self.confidence < other.confidence def __bool__(self): return self.value def _to_core_struct(self) -> core.BNBoolWithConfidence: result = core.BNBoolWithConfidence() result.value = self.value result.confidence = self.confidence return result @classmethod def from_core_struct(cls, core_struct:core.BNBoolWithConfidence) -> 'BoolWithConfidence': return cls(core_struct.value, core_struct.confidence) @staticmethod def get_core_struct(value:BoolWithConfidenceType) -> core.BNBoolWithConfidence: if isinstance(value, BoolWithConfidence): return value._to_core_struct() else: return BoolWithConfidence(value)._to_core_struct() @dataclass(frozen=True) class SizeWithConfidence: value:int confidence:int=core.max_confidence def __int__(self): return self.value def __eq__(self, other): if not isinstance(other, self.__class__): return self.value == int(other) else: return (self.value, self.confidence) == (other.value, other.confidence) def __ne__(self, other): return not (self == other) def __gt__(self, other): return self.value > int(other) def __le__(self, other): return self.value <= int(other) def __ge__(self, other): return self.value >= int(other) def __lt__(self, other): return self.value < int(other) def _to_core_struct(self) -> core.BNSizeWithConfidence: result = core.BNSizeWithConfidence() result.value = self.value result.confidence = self.confidence return result @classmethod def from_core_struct(cls, core_struct:core.BNSizeWithConfidence) -> 'SizeWithConfidence': return cls(core_struct.value, core_struct.confidence) @staticmethod def get_core_struct(value:SizeWithConfidenceType) -> core.BNSizeWithConfidence: if isinstance(value, SizeWithConfidence): return value._to_core_struct() else: return SizeWithConfidence(value)._to_core_struct() @dataclass class MutableTypeBuilder: type:'TypeBuilder' container:TypeContainer name:QualifiedName platform:Optional['_platform.Platform'] confidence:int user:bool = True def __enter__(self): return self.type def __exit__(self, type, value, traceback): if isinstance(self.container, binaryview.BinaryView): if self.user: self.container.define_user_type(self.name, self.type.immutable_copy()) else: type_id = types.Type.generate_auto_type_id(str(uuid.uuid4()), str(self.name)) self.container.define_type(type_id, self.name, self.type.immutable_copy()) else: self.container.add_named_type(self.name, self.type.immutable_copy()) class TypeBuilder: def __init__(self, handle:core.BNTypeBuilderHandle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence): assert isinstance(handle, core.BNTypeBuilderHandle), "handle isn't an instance of BNTypeBuilderHandle" self._handle = handle self.platform = platform self.confidence = confidence def __del__(self): if core is not None: core.BNFreeTypeBuilder(self._handle) @property def handle(self) -> core.BNTypeHandle: return self.immutable_copy().handle def _to_core_struct(self) -> core.BNTypeWithConfidence: type_conf = core.BNTypeWithConfidence() type_conf.type = self.handle type_conf.confidence = self.confidence return type_conf def immutable_copy(self): Types = { TypeClass.VoidTypeClass:VoidType, TypeClass.BoolTypeClass:BoolType, TypeClass.IntegerTypeClass:IntegerType, TypeClass.FloatTypeClass:FloatType, TypeClass.PointerTypeClass:PointerType, TypeClass.ArrayTypeClass:ArrayType, TypeClass.FunctionTypeClass:FunctionType, TypeClass.WideCharTypeClass:WideCharType, # TypeClass.StructureTypeClass:StructureType, # TypeClass.EnumerationTypeClass:EnumerationType, # TypeClass.NamedTypeReferenceClass:NamedTypeReferenceType, } return Types[self.type_class](self.finalized, self.platform, self.confidence) def mutable_copy(self) -> 'TypeBuilder': return self @classmethod def create(cls): _ = cls return NotImplemented @classmethod def builder(cls, container:TypeContainer, name:'QualifiedName', user:bool=True, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'MutableTypeBuilder': return MutableTypeBuilder(cls.create(), container, name, platform, confidence, user) @staticmethod def void() -> 'VoidBuilder': return VoidBuilder.create() @staticmethod def bool() -> 'BoolBuilder': return BoolBuilder.create() @staticmethod def char(alternate_name:str="") -> 'CharBuilder': return CharBuilder.create(alternate_name) @staticmethod def int(width:_int, sign:BoolWithConfidenceType=BoolWithConfidence(True), altname:str="") -> 'IntegerBuilder': """ ``int`` class method for creating an int Type. :param int width: width of the integer in bytes :param bool sign: optional variable representing signedness :param str altname: alternate name for type """ return IntegerBuilder.create(width, sign, altname) @staticmethod def float(width:_int, altname:str="") -> 'FloatBuilder': """ ``float`` class method for creating floating point Types. :param int width: width of the floating point number in bytes :param str altname: alternate name for type """ return FloatBuilder.create(width, altname) @staticmethod def wide_char(width:_int, altname:str="") -> 'WideCharBuilder': """ ``wide_char`` class method for creating wide char Types. :param int width: width of the wide character in bytes :param str altname: alternate name for type """ return WideCharBuilder.create(width, altname) @staticmethod def named_type_from_type(name:QualifiedName, type_class:Optional[NamedTypeReferenceClass]=None) -> 'NamedTypeReferenceBuilder': return NamedTypeReferenceBuilder.named_type_from_type(name, type_class) @staticmethod def named_type_from_type_and_id(type_id:str, name:QualifiedName, type:Optional['Type']=None) -> 'NamedTypeReferenceBuilder': return NamedTypeReferenceBuilder.named_type_from_type_and_id(type_id, name, type) @staticmethod def generate_named_type_reference(guid:str, name:QualifiedName) -> 'NamedTypeReferenceBuilder': return NamedTypeReferenceBuilder.generate_named_type_reference(guid, name) @staticmethod def named_type_from_registered_type(view:'binaryview.BinaryView', name:QualifiedName) -> 'NamedTypeReferenceBuilder': return NamedTypeReferenceBuilder.named_type_from_registered_type(view, name) @staticmethod def pointer(arch:'architecture.Architecture', type:'Type', const:BoolWithConfidenceType=BoolWithConfidence(False), volatile:BoolWithConfidenceType=BoolWithConfidence(False), ref_type:ReferenceType=ReferenceType.PointerReferenceType) -> 'PointerBuilder': if arch is not None: width = arch.address_size if width is None: raise TypeCreateException("Must specify either an architecture or a width to create a pointer") return PointerBuilder.create(type, width, arch, const, volatile, ref_type) @staticmethod def pointer_of_width(width:_int, type:'Type', const:BoolWithConfidenceType=BoolWithConfidence(False), volatile:BoolWithConfidenceType=BoolWithConfidence(False), ref_type:ReferenceType=ReferenceType.PointerReferenceType) -> 'PointerBuilder': return PointerBuilder.create(type, width, None, const, volatile, ref_type) @staticmethod def array(type:'Type', count:_int) -> 'ArrayBuilder': return ArrayBuilder.create(type, count) @staticmethod def function(ret:Optional['Type'], params:ParamsType=[], calling_convention:'callingconvention.CallingConvention'=None, variable_arguments:BoolWithConfidenceType=BoolWithConfidence(False), stack_adjust:SizeWithConfidenceType=0) -> 'FunctionBuilder': """ ``function`` class method for creating an function Type. :param Type ret: return Type of the function :param params: list of parameter Types :type params: list(Type) :param CallingConvention calling_convention: optional argument for the function calling convention :param bool variable_arguments: optional boolean, true if the function has a variable number of arguments """ if ret is None: ret = Type.void() return FunctionBuilder.create(ret, calling_convention, params, variable_arguments, stack_adjust) @staticmethod def structure(members:MembersType=[], packed:_bool=False, type:StructureVariant=StructureVariant.StructStructureType) -> 'StructureBuilder': return StructureBuilder.create(members, type=type, packed=packed) @staticmethod def enumeration(arch:Optional['architecture.Architecture']=None, members:List[EnumMembersType]=[], width:Optional[_int]=None, sign:BoolWithConfidenceType=BoolWithConfidence(False)) -> 'EnumerationBuilder': return EnumerationBuilder.create(members, width, arch, sign) @staticmethod def named_type_reference(type_class:NamedTypeReferenceClass, name:QualifiedName, type_id:Optional[str]=None, alignment:_int=1, width:_int=0) -> 'NamedTypeReferenceBuilder': return NamedTypeReferenceBuilder.create(type_class, type_id, name, width, alignment) @property def width(self) -> _int: return core.BNGetTypeBuilderWidth(self._handle) def __len__(self): return self.width @property def finalized(self): type_handle = core.BNFinalizeTypeBuilder(self._handle) assert type_handle is not None, "core.BNFinalizeTypeBuilder returned None" type_handle = core.BNNewTypeReference(type_handle) assert type_handle is not None, "core.BNNewTypeReference returned None" return type_handle @property def const(self) -> BoolWithConfidence: """Whether type is const (read/write)""" result = core.BNIsTypeBuilderConst(self._handle) return BoolWithConfidence(result.value, confidence = result.confidence) @const.setter def const(self, value:BoolWithConfidenceType) -> None: # type: ignore We explicitly allow 'set' type to be different than 'get' type core.BNTypeBuilderSetConst(self._handle, BoolWithConfidence.get_core_struct(value)) @property def volatile(self) -> BoolWithConfidence: """Whether type is volatile (read/write)""" result = core.BNIsTypeBuilderVolatile(self._handle) return BoolWithConfidence(result.value, confidence = result.confidence) @volatile.setter def volatile(self, value:BoolWithConfidenceType) -> None: # type: ignore We explicitly allow 'set' type to be different than 'get' type core.BNTypeBuilderSetVolatile(self._handle, BoolWithConfidence.get_core_struct(value)) @property def alignment(self) -> _int: return core.BNGetTypeBuilderAlignment(self._handle) @property def child(self) -> 'Type': type_conf = core.BNGetTypeBuilderChildType(self._handle) assert type_conf is not None, "core.BNGetTypeBuilderChildType returned None" handle = core.BNNewTypeReference(type_conf.type) assert handle is not None, "core.BNNewTypeReference returned None" return Type.create(handle, self.platform, type_conf.confidence) @child.setter def child(self, value:SomeType) -> None: # type: ignore return core.BNTypeBuilderSetChildType(self._handle, value.immutable_copy()._to_core_struct()) @property def alternate_name(self) -> str: return core.BNGetTypeBuilderAlternateName(self._handle) @alternate_name.setter def alternate_name(self, name:str) -> None: return core.BNTypeBuilderSetAlternateName(self._handle, name) @property def type_class(self) -> TypeClass: return TypeClass(core.BNGetTypeBuilderClass(self._handle)) class VoidBuilder(TypeBuilder): @classmethod def create(cls, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'VoidBuilder': handle = core.BNCreateVoidTypeBuilder() assert handle is not None, "core.BNCreateVoidTypeBuilder returned None" return cls(handle, platform, confidence) class BoolBuilder(TypeBuilder): @classmethod def create(cls, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'BoolBuilder': handle = core.BNCreateBoolTypeBuilder() assert handle is not None, "core.BNCreateBoolTypeBuilder returned None" return cls(handle, platform, confidence) class IntegerBuilder(TypeBuilder): @classmethod def create(cls, width:int, sign:BoolWithConfidenceType=True, alternate_name:str="", platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'IntegerBuilder': _sign = BoolWithConfidence.get_core_struct(sign) handle = core.BNCreateIntegerTypeBuilder(width, _sign, alternate_name) assert handle is not None, "core.BNCreateIntegerTypeBuilder returned None" return cls(handle, platform, confidence) @property def signed(self) -> BoolWithConfidence: """Whether type is signed (read/write)""" result = core.BNIsTypeBuilderSigned(self._handle) return BoolWithConfidence(result.value, confidence = result.confidence) @signed.setter def signed(self, value:BoolWithConfidenceType) -> None: # type: ignore core.BNTypeBuilderSetSigned(self._handle, BoolWithConfidence.get_core_struct(value)) class CharBuilder(IntegerBuilder): @classmethod def create(cls, alternate_name:str="", platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'CharBuilder': handle = core.BNCreateIntegerTypeBuilder(1, False, alternate_name) assert handle is not None, "BNCreateIntegerTypeBuilder returned None" return cls(handle, platform, confidence) class FloatBuilder(TypeBuilder): @classmethod def create(cls, width:int, alternate_name:str="", platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'FloatBuilder': handle = core.BNCreateFloatTypeBuilder(width, alternate_name) assert handle is not None, "core.BNCreateFloatTypeBuilder returned None" return cls(handle, platform, confidence) class WideCharBuilder(TypeBuilder): @classmethod def create(cls, width:int, alternate_name:str="", platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'WideCharBuilder': handle = core.BNCreateWideCharTypeBuilder(width, alternate_name) assert handle is not None, "core.BNCreateWideCharTypeBuilder returned None" return cls(handle, platform, confidence) class PointerBuilder(TypeBuilder): @classmethod def create(cls, type:'Type', width:int=4, arch:Optional['architecture.Architecture']=None, const:BoolWithConfidenceType=False, volatile:BoolWithConfidenceType=False, ref_type:ReferenceType=ReferenceType.PointerReferenceType, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'PointerBuilder': assert width is not None or arch is not None, "Must specify either a width or architecture when creating a pointer" _width = width if arch is not None: _width = arch.address_size _const = BoolWithConfidence.get_core_struct(const) _volatile = BoolWithConfidence.get_core_struct(volatile) handle = core.BNCreatePointerTypeBuilderOfWidth(_width, type.handle, _const, _volatile, ref_type) assert handle is not None, "BNCreatePointerTypeBuilderOfWidth returned None" return cls(handle, platform, confidence) @property def target(self) -> 'TypeBuilder': return self.immutable_target.mutable_copy() @property def immutable_target(self) -> 'Type': return self.child class ArrayBuilder(TypeBuilder): @classmethod def create(cls, type:SomeType, element_count:int, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'ArrayBuilder': handle = core.BNCreateArrayTypeBuilder(type.handle, element_count) assert handle is not None, "BNCreateArrayTypeBuilder returned None" return cls(handle, platform, confidence) @property def count(self) -> int: return core.BNGetTypeBuilderElementCount(self._handle) @property def element_type(self): return self.child class FunctionBuilder(TypeBuilder): @classmethod def create(cls, return_type:SomeType, calling_convention:Optional['callingconvention.CallingConvention']=None, params:ParamsType=[], var_args:BoolWithConfidenceType=False, stack_adjust:SizeWithConfidenceType=0, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'FunctionBuilder': param_buf = FunctionBuilder._to_core_struct(params) ret_conf = return_type._to_core_struct() conv_conf = core.BNCallingConventionWithConfidence() if calling_convention is None: conv_conf.convention = None conv_conf.confidence = 0 else: conv_conf.convention = calling_convention.handle conv_conf.confidence = calling_convention.confidence vararg_conf = BoolWithConfidence.get_core_struct(var_args) stack_adjust_conf = SizeWithConfidence.get_core_struct(stack_adjust) handle = core.BNCreateFunctionTypeBuilder(ret_conf, conv_conf, param_buf, len(params), vararg_conf, stack_adjust_conf) assert handle is not None, "BNCreateFunctionTypeBuilder returned None" return cls(handle, platform, confidence) @property def immutable_return_value(self) -> 'Type': return self.child @property def return_value(self) -> TypeBuilder: return self.child.mutable_copy() @return_value.setter def return_value(self, value:SomeType) -> None: # type: ignore self.child = value def append(self, type:Union[SomeType, FunctionParameter], name:str=""): params = self.parameters if isinstance(type, FunctionParameter): self.parameters.append(type.mutable_copy()) else: self.parameters.append(FunctionParameter(type.mutable_copy(), name)) self.params = params @property def calling_convention(self) -> 'callingconvention.CallingConvention': return callingconvention.CallingConvention(core.BNNewCallingConventionReference(core.BNGetTypeBuilderCallingConvention(self._handle))) @property def can_return(self) -> bool: return core.BNFunctionTypeBuilderCanReturn(self._handle) @can_return.setter def can_return(self, value:BoolWithConfidenceType) -> None: # type: ignore core.BNSetFunctionTypeBuilderCanReturn(self._handle, BoolWithConfidence.get_core_struct(value)) @property def stack_adjust(self) -> OffsetWithConfidence: return OffsetWithConfidence.from_core_struct(core.BNGetTypeBuilderStackAdjustment(self._handle)) @property def parameters(self) -> List[FunctionParameter]: """Type parameters list (read-only)""" count = ctypes.c_ulonglong() params = core.BNGetTypeBuilderParameters(self._handle, count) assert params is not None, "core.BNGetTypeBuilderParameters returned None" result = [] for i in range(0, count.value): param_type = Type.create(core.BNNewTypeReference(params[i].type), platform = self.platform, confidence = params[i].typeConfidence) if params[i].defaultLocation: param_location = None else: name = params[i].name if (params[i].location.type == VariableSourceType.RegisterVariableSourceType) and (self.platform is not None): name = self.platform.arch.get_reg_name(params[i].location.storage) elif params[i].location.type == VariableSourceType.StackVariableSourceType: name = "arg_%x" % params[i].location.storage param_location = variable.VariableNameAndType(params[i].location.type, params[i].location.index, params[i].location.storage, name, param_type) result.append(FunctionParameter(param_type, params[i].name, param_location)) core.BNFreeTypeParameterList(params, count.value) return result @property def variable_arguments(self) -> BoolWithConfidence: return BoolWithConfidence.from_core_struct(core.BNTypeBuilderHasVariableArguments(self._handle)) @staticmethod def _to_core_struct(params:ParamsType): param_buf = (core.BNFunctionParameter * len(params))() for i, param in enumerate(params): core_param = param_buf[i] if isinstance(param, (Type, TypeBuilder)): assert param.handle is not None, "Attempting to construct function parameter without properly constructed type" core_param.name = "" core_param.type = param.handle core_param.typeConfidence = param.confidence core_param.defaultLocation = True elif isinstance(param, FunctionParameter): assert param.type is not None, "Attempting to construct function parameter without properly constructed type" core_param.name = param.name core_param.type = param.type.handle core_param.typeConfidence = param.type.confidence if param.location is None: core_param.defaultLocation = True else: core_param.defaultLocation = False core_param.location.type = param.location.source_type core_param.location.index = param.location.index core_param.location.storage = param.location.storage elif isinstance(param, tuple): name, _type = param assert isinstance(name, str), f"Conversion from unsupported function parameter type {type(param)}" assert isinstance(_type, (Type, TypeBuilder)), f"Conversion from unsupported function parameter type {type(param)}" core_param.name = name core_param.type = _type.handle core_param.typeConfidence = _type.confidence core_param.defaultLocation = True else: assert False, f"Conversion from unsupported function parameter type {type(param)}" return param_buf @parameters.setter def parameters(self, params:List[FunctionParameter]) -> None: core.BNSetFunctionTypeBuilderParameters(self._handle, FunctionBuilder._to_core_struct(params), len(params)) @dataclass class StructureMember: type:'Type' name:str offset:int access:MemberAccess = MemberAccess.NoAccess scope:MemberScope = MemberScope.NoScope def __repr__(self): if len(self.name) == 0: return f"" return f"<{self.type.get_string_before_name()} {self.name}{self.type.get_string_after_name()}" + \ f", offset {self.offset:#x}>" def __len__(self): return len(self.type) class StructureBuilder(TypeBuilder): def __init__(self, handle:core.BNTypeBuilderHandle, builder_handle:core.BNStructureBuilderHandle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence): super(StructureBuilder, self).__init__(handle, platform, confidence) assert builder_handle is not None, "Can't instantiate Structure with builder_handle set to None" self.builder_handle = builder_handle @classmethod def create(cls, members:MembersType=[], type:StructureVariant=StructureVariant.StructStructureType, packed:bool=False, width:Optional[int]=None, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'StructureBuilder': structure_builder_handle = core.BNCreateStructureBuilderWithOptions(type, packed) assert structure_builder_handle is not None, "core.BNCreateStructureBuilderWithOptions returned None" if width is not None: core.BNSetStructureBuilderWidth(structure_builder_handle, width) for member in members: if isinstance(member, Tuple): _type, _name = member core.BNAddStructureBuilderMember(structure_builder_handle, _type._to_core_struct(), _name, MemberAccess.NoAccess, MemberScope.NoScope) elif isinstance(member, StructureMember): core.BNAddStructureBuilderMemberAtOffset(structure_builder_handle, member.type._to_core_struct(), member.name, member.offset, False, member.access, member.scope) type_builder_handle = core.BNCreateStructureTypeBuilderWithBuilder(structure_builder_handle) assert type_builder_handle is not None, "core.BNCreateStructureTypeBuilderWithBuilder returned None" return cls(type_builder_handle, structure_builder_handle, platform, confidence) def immutable_copy(self) -> 'StructureType': assert self.builder_handle is not None structure_handle = core.BNFinalizeStructureBuilder(self.builder_handle) assert structure_handle is not None, "core.BNFinalizeStructureBuilder returned None" handle = core.BNCreateStructureType(structure_handle) assert handle is not None, "core.BNCreateStructureType returned None" return StructureType(handle, self.platform, self.confidence) @property def members(self) -> List[StructureMember]: """Structure member list (read-only)""" count = ctypes.c_ulonglong() members = core.BNGetStructureBuilderMembers(self.builder_handle, count) assert members is not None, "core.BNGetStructureBuilderMembers returned None" try: result = [] for i in range(0, count.value): t = Type.create(core.BNNewTypeReference(members[i].type), confidence=members[i].typeConfidence) result.append(StructureMember(t, members[i].name, members[i].offset, MemberAccess(members[i].access), MemberScope(members[i].scope))) return result finally: core.BNFreeStructureMemberList(members, count.value) @members.setter def members(self, members:List[StructureMember]) -> None: for i in range(len(self.members)): core.BNRemoveStructureBuilderMember(i) for member in members: core.BNAddStructureBuilderMember(self.builder_handle, member.type._to_core_struct(), member.name, member.access, member.scope) @property def packed(self) -> bool: return core.BNIsStructureBuilderPacked(self.builder_handle) @packed.setter def packed(self, value:bool) -> None: core.BNSetStructureBuilderPacked(self.builder_handle, value) @property def alignment(self) -> int: return core.BNGetStructureBuilderAlignment(self.builder_handle) @alignment.setter def alignment(self, value:int) -> None: core.BNSetStructureBuilderAlignment(self.builder_handle, value) @property def width(self) -> int: return core.BNGetStructureBuilderWidth(self.builder_handle) @width.setter def width(self, value:int) -> None: core.BNSetStructureBuilderWidth(self.builder_handle, value) @property def union(self) -> bool: return core.BNIsStructureBuilderUnion(self.builder_handle) @property def type(self) -> StructureVariant: return StructureVariant(core.BNGetStructureBuilderType(self.builder_handle)) @type.setter def type(self, value:StructureVariant) -> None: core.BNSetStructureBuilderType(self.builder_handle, value) def __repr__(self): return f"" def __getitem__(self, name:str) -> Optional[StructureMember]: member = core.BNGetStructureBuilderMemberByName(self._handle, name) if member is None: return None try: return StructureMember(Type(core.BNNewTypeReference(member.contents.type), confidence=member.contents.typeConfidence), member.contents.name, member.contents.offset) finally: core.BNFreeStructureMember(member) def __iter__(self) -> Generator[StructureMember, None, None]: for member in self.members: yield member def __len__(self) -> int: return self.width def member_at_offset(self, offset:int) -> Optional[StructureMember]: for member in self.members: if member.offset == offset: return member return None def index_by_name(self, name:MemberName) -> Optional[MemberIndex]: for i, member in enumerate(self.members): if member.name == name: return i return None def index_by_offset(self, offset:MemberOffset) -> Optional[MemberIndex]: for i, member in enumerate(self.members): if member.offset == offset: return i return None def index_from(self, index:Optional[MemberIndex]=None, name:Optional[MemberName]=None, offset:Optional[MemberOffset]=None) -> MemberIndex: if index is not None: if index >= len(self.members): raise IndexError("list index out of range") elif name is not None: index = self.index_by_name(name) if index is None: raise ValueError(f"StructureMember {name} doesn't exist") elif offset is not None: index = self.index_by_offset(offset) if index is None: raise ValueError(f"No StructureMember at offset exists") else: raise ValueError("One of the following must") return index def replace(self, index:int, type:SomeType, name:str="", overwrite_existing:bool=True): core.BNReplaceStructureBuilderMember(self.builder_handle, index, type._to_core_struct(), name, overwrite_existing) def remove(self, index:int): core.BNRemoveStructureBuilderMember(self.builder_handle, index) def insert(self, offset:int, type:SomeType, name:str="", overwrite_existing:bool=True, access:MemberAccess=MemberAccess.NoAccess, scope:MemberScope=MemberScope.NoScope): core.BNAddStructureBuilderMemberAtOffset(self.builder_handle, type._to_core_struct(), name, offset, overwrite_existing, access, scope) def append(self, type:SomeType, name:MemberName="", access:MemberAccess=MemberAccess.NoAccess, scope:MemberScope=MemberScope.NoScope) -> 'StructureBuilder': # appends a member at the end of the structure growing the structure core.BNAddStructureBuilderMember(self.builder_handle, type._to_core_struct(), name, access, scope) return self def add_member_at_offset(self, name:MemberName, type:SomeType, offset:MemberOffset, overwrite_existing:bool=True, access:MemberAccess=MemberAccess.NoAccess, scope:MemberScope=MemberScope.NoScope) -> 'StructureBuilder': # Adds structure member to the given offset optionally clearing any members within the range offset-offset+len(type) core.BNAddStructureBuilderMemberAtOffset(self.builder_handle, type._to_core_struct(), name, offset, overwrite_existing, access, scope) return self @dataclass(frozen=True) class EnumerationMember: name:str value:Optional[int] def __repr__(self): value = f"{self.value:#x}" if self.value is not None else "auto()" return f"<{self.name} = {value}>" def __int__(self) -> Optional[int]: return self.value class EnumerationBuilder(TypeBuilder): def __init__(self, handle:core.BNTypeBuilderHandle, enum_builder_handle:core.BNEnumerationBuilderHandle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence): super(EnumerationBuilder, self).__init__(handle, platform, confidence) self.enum_builder_handle = enum_builder_handle @staticmethod def _enum_handle_from_members(members): enum_builder_handle = core.BNCreateEnumerationBuilder() for member in members: if member.value is None: core.BNAddEnumerationBuilderMember(enum_builder_handle, member.name) else: core.BNAddEnumerationBuilderMemberWithValue(enum_builder_handle, member.name, member.value) @classmethod def create(cls, members=List[EnumMembersType], width:Optional[int]=None, arch:Optional['architecture.Architecture']=None, sign:BoolWithConfidenceType=False, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'EnumerationBuilder': _width = width if arch is not None: _width = arch.address_size if _width is None: _width = 4 _sign = BoolWithConfidence.get_core_struct(sign) enum_builder_handle = core.BNCreateEnumerationBuilder() assert enum_builder_handle is not None, "core.BNCreateEnumerationBuilder returned None" EnumerationBuilder.add_members(enum_builder_handle, members) type_builder_handle = core.BNCreateEnumerationTypeBuilderWithBuilder(None, enum_builder_handle, _width, _sign) assert type_builder_handle is not None, "core.BNCreateEnumerationTypeBuilderWithBuilder returned None" return cls(type_builder_handle, enum_builder_handle, platform, confidence) def immutable_copy(self) -> 'EnumerationType': enum_handle = core.BNFinalizeEnumerationBuilder(self.enum_builder_handle) assert enum_handle is not None, "core.BNFinalizeEnumerationBuilder returned None" handle = core.BNCreateEnumerationType(None, enum_handle, self.width, self.signed) assert handle is not None, "core.BNCreateEnumerationType returned None" return EnumerationType(handle, self.platform, self.confidence) def __repr__(self): return "" % repr(self.members) @property def signed(self) -> BoolWithConfidence: """Whether type is signed (read/write)""" result = core.BNIsTypeBuilderSigned(self._handle) return BoolWithConfidence(result.value, confidence = result.confidence) @signed.setter def signed(self, value:BoolWithConfidenceType) -> None: # type: ignore _value = BoolWithConfidence.get_core_struct(value) core.BNTypeBuilderSetSigned(self._handle, _value) @property def members(self) -> List[EnumerationMember]: """Enumeration member list (read-only)""" count = ctypes.c_ulonglong() members = core.BNGetEnumerationBuilderMembers(self.enum_builder_handle, count) assert members is not None, "core.BNGetEnumerationBuilderMembers returned None" result = [] try: for i in range(count.value): result.append(EnumerationMember(members[i].name, members[i].value if not members[i].isDefault else None)) return result finally: core.BNFreeEnumerationMemberList(members, count.value) @members.setter def members(self, members:List[EnumMembersType]) -> None: # type: ignore EnumerationBuilder.add_members(self.enum_builder_handle, members) @staticmethod def add_members(enum_builder_handle, members:List[EnumMembersType]): for i, member in enumerate(members): value = None if isinstance(member, Tuple): name, value = member elif isinstance(member, EnumerationMember): name = member.name value = member.value elif isinstance(member, str): name = member else: assert False, "Unhandled type for EnumerationMember" if value is None: core.BNAddEnumerationBuilderMember(enum_builder_handle, name) else: core.BNAddEnumerationBuilderMemberWithValue(enum_builder_handle, name, value) def append(self, name:str, value:Optional[int]=None) -> 'EnumerationBuilder': if value is None: core.BNAddEnumerationBuilderMember(self.enum_builder_handle, name) else: core.BNAddEnumerationBuilderMemberWithValue(self.enum_builder_handle, name, value) return self def remove(self, i:int) -> 'EnumerationBuilder': core.BNRemoveEnumerationBuilderMember(self.enum_builder_handle, i) return self def replace(self, i:int, name:str, value:Optional[int]=None) -> 'EnumerationBuilder': core.BNReplaceEnumerationBuilderMember(self.enum_builder_handle, i, name, value) return self def __iter__(self) -> Generator[EnumerationMember, None, None]: for i, member in enumerate(self.members): if member.value is None: yield EnumerationMember(member.name, i) else: yield member def __getitem__(self, value:Union[str, int, slice]): if isinstance(value, str): for member in self.members: if member.name == value: return member return None elif isinstance(value, int): return self.members[value] elif isinstance(value, slice): # not combined with the previous check due to pyright bug return self.members[value] else: raise ValueError(f"Incompatible type {type(value)} for __getitem__") def __setitem__(self, item, value): if isinstance(item, str): for i, member in enumerate(self.members): if member.name == item: self.replace(i, member.name, value) elif isinstance(item, int) and isinstance(value, EnumerationMember): self.replace(item, value.name, value.value) else: assert False, "Invalid type for Enumeration.__setitem__" class NamedTypeReferenceBuilder(TypeBuilder): def __init__(self, handle:core.BNTypeBuilderHandle, ntr_builder_handle:core.BNNamedTypeReferenceBuilderHandle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence): assert ntr_builder_handle is not None, "Failed to construct NameTypeReference" assert handle is not None, "Failed to construct NameTypeReference" assert isinstance(ntr_builder_handle, core.BNNamedTypeReferenceBuilderHandle), "Failed to construct NameTypeReference" super(NamedTypeReferenceBuilder, self).__init__(handle, platform, confidence) self.ntr_builder_handle = ntr_builder_handle @classmethod def create(cls, type_class:NamedTypeReferenceClass=NamedTypeReferenceClass.UnknownNamedTypeClass, type_id:Optional[str]=None, name:QualifiedName=QualifiedName(""), width:int=0, align:int=1, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'NamedTypeReferenceBuilder': ntr_builder_handle = core.BNCreateNamedTypeBuilder(type_class, type_id, name._to_core_struct()) assert ntr_builder_handle is not None, "core.BNCreateNamedTypeBuilder returned None" type_builder_handle = core.BNCreateNamedTypeReferenceBuilderWithBuilder(ntr_builder_handle, width, align) assert type_builder_handle is not None, "core.BNCreateNamedTypeReferenceBuilderWithBuilder returned None" return cls(type_builder_handle, ntr_builder_handle, platform, confidence) def immutable_copy(self) -> 'NamedTypeReferenceType': ntr_handle = core.BNFinalizeNamedTypeReferenceBuilder(self.ntr_builder_handle) assert ntr_handle is not None, "core.BNFinalizeEnumerationBuilder returned None" handle = core.BNCreateNamedTypeReference(ntr_handle, self.width, self.alignment) assert handle is not None, "core.BNCreateEnumerationType returned None" return NamedTypeReferenceType(handle, self.platform, self.confidence) @property def name(self) -> QualifiedName: return QualifiedName._from_core_struct(core.BNGetTypeReferenceBuilderName(self.ntr_builder_handle)) @property def id(self) -> str: return core.BNGetTypeReferenceBuilderId(self.ntr_builder_handle) @property def type_id(self) -> str: return core.BNGetTypeReferenceBuilderId(self.ntr_builder_handle) @property def named_type_class(self) -> NamedTypeReferenceClass: return NamedTypeReferenceClass(core.BNGetTypeReferenceBuilderClass(self.ntr_builder_handle)) @staticmethod def named_type(named_type:'NamedTypeReferenceBuilder', width:int=0, align:int=1) -> 'NamedTypeReferenceBuilder': return NamedTypeReferenceBuilder.create(named_type.named_type_class, named_type.id, named_type.name, width, align) @staticmethod def named_type_from_type_and_id(type_id:str, name:QualifiedName, type:Optional['Type']) -> 'NamedTypeReferenceBuilder': if type is None: return NamedTypeReferenceBuilder.create(NamedTypeReferenceClass.UnknownNamedTypeClass, type_id, name) elif type.type_class == TypeClass.StructureTypeClass: if type.structure_type == StructureVariant.StructStructureType: return NamedTypeReferenceBuilder.create(NamedTypeReferenceClass.StructNamedTypeClass, type_id, name) elif type.structure_type == StructureVariant.UnionStructureType: return NamedTypeReferenceBuilder.create(NamedTypeReferenceClass.UnionNamedTypeClass, type_id, name) else: return NamedTypeReferenceBuilder.create(NamedTypeReferenceClass.ClassNamedTypeClass, type_id, name) elif type.type_class == TypeClass.EnumerationTypeClass: return NamedTypeReferenceBuilder.create(NamedTypeReferenceClass.EnumNamedTypeClass, type_id, name) else: return NamedTypeReferenceBuilder.create(NamedTypeReferenceClass.TypedefNamedTypeClass, type_id, name) @staticmethod def named_type_from_type(name:QualifiedName, type_class:Optional[NamedTypeReferenceClass]=None) -> 'NamedTypeReferenceBuilder': if type_class is None: return NamedTypeReferenceBuilder.create(NamedTypeReferenceClass.UnknownNamedTypeClass, str(uuid.uuid4()), name) else: return NamedTypeReferenceBuilder.create(NamedTypeReferenceClass.TypedefNamedTypeClass, str(uuid.uuid4()), name) @staticmethod def named_type_from_registered_type(view:'binaryview.BinaryView', name:QualifiedName) -> 'NamedTypeReferenceBuilder': type = view.get_type_by_name(name) if type is None: raise TypeCreateException(f"Unable to find type named {name}") return NamedTypeReferenceBuilder.named_type_from_type_and_id(type_id=str(uuid.uuid4()), name=name, type=type) def __repr__(self): if self.named_type_class == NamedTypeReferenceClass.TypedefNamedTypeClass: return f"" if self.named_type_class == NamedTypeReferenceClass.StructNamedTypeClass: return f"" if self.named_type_class == NamedTypeReferenceClass.UnionNamedTypeClass: return f"" if self.named_type_class == NamedTypeReferenceClass.EnumNamedTypeClass: return f"" return "" class Type: """ ``class Type`` allows you to interact with the Binary Ninja type system. Note that the ``repr`` and ``str`` handlers respond differently on type objects. Other related functions that may be helpful include: :py:meth:`parse_type_string ` :py:meth:`parse_types_from_source ` :py:meth:`parse_types_from_source_file ` """ def __init__(self, handle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence): assert isinstance(handle.contents, core.BNType), "Attempting to create mutable Type" self._handle = handle self._confidence = confidence self._platform = platform @classmethod def create(cls, handle=core.BNTypeHandle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'Type': assert handle is not None, "Passed a handle which is None" assert isinstance(handle, core.BNTypeHandle) type_class = TypeClass(core.BNGetTypeClass(handle)) return Types[type_class](handle, platform, confidence) def __del__(self): if core is not None: core.BNFreeType(self._handle) def __repr__(self): if self._confidence < core.max_confidence: return f"" return f"" def __str__(self): platform = None if self._platform is not None: platform = self._platform.handle return core.BNGetTypeString(self._handle, platform) def __len__(self): return self.width def __eq__(self, other): if not isinstance(other, self.__class__): return NotImplemented return core.BNTypesEqual(self._handle, other._handle) def __ne__(self, other): if not isinstance(other, self.__class__): return NotImplemented return core.BNTypesNotEqual(self._handle, other._handle) @property def handle(self): return self._handle @property def type_class(self) -> TypeClass: """Type class (read-only)""" return TypeClass(core.BNGetTypeClass(self._handle)) @property def width(self) -> int: """Type width (read-only)""" return core.BNGetTypeWidth(self._handle) @property def alignment(self) -> int: """Type alignment (read-only)""" return core.BNGetTypeAlignment(self._handle) @property def offset(self) -> int: """Offset into structure (read-only)""" return core.BNGetTypeOffset(self._handle) @property def altname(self) -> str: """Alternative name for the type object""" return core.BNGetTypeAlternateName(self._handle) def _to_core_struct(self) -> core.BNTypeWithConfidence: type_conf = core.BNTypeWithConfidence() type_conf.type = self._handle type_conf.confidence = self.confidence return type_conf def get_string_before_name(self) -> str: platform = None if self._platform is not None: platform = self._platform.handle return core.BNGetTypeStringBeforeName(self._handle, platform) def get_string_after_name(self) -> str: platform = None if self._platform is not None: platform = self._platform.handle return core.BNGetTypeStringAfterName(self._handle, platform) @property def tokens(self) -> List['_function.InstructionTextToken']: """Type string as a list of tokens (read-only)""" return self.get_tokens() def get_tokens(self, base_confidence = core.max_confidence) -> List['_function.InstructionTextToken']: count = ctypes.c_ulonglong() platform = None if self._platform is not None: platform = self._platform.handle tokens = core.BNGetTypeTokens(self._handle, platform, base_confidence, count) assert tokens is not None, "core.BNGetTypeTokens returned None" result = _function.InstructionTextToken._from_core_struct(tokens, count.value) core.BNFreeInstructionText(tokens, count.value) return result def get_tokens_before_name(self, base_confidence = core.max_confidence) -> List['_function.InstructionTextToken']: count = ctypes.c_ulonglong() platform = None if self._platform is not None: platform = self._platform.handle tokens = core.BNGetTypeTokensBeforeName(self._handle, platform, base_confidence, count) assert tokens is not None, "core.BNGetTypeTokensBeforeName returned None" result = _function.InstructionTextToken._from_core_struct(tokens, count.value) core.BNFreeInstructionText(tokens, count.value) return result def get_tokens_after_name(self, base_confidence = core.max_confidence) -> List['_function.InstructionTextToken']: count = ctypes.c_ulonglong() platform = None if self._platform is not None: platform = self._platform.handle tokens = core.BNGetTypeTokensAfterName(self._handle, platform, base_confidence, count) assert tokens is not None, "core.BNGetTypeTokensAfterName returned None" result = _function.InstructionTextToken._from_core_struct(tokens, count.value) core.BNFreeInstructionText(tokens, count.value) return result def with_confidence(self, confidence) -> 'Type': return Type.create(handle = core.BNNewTypeReference(self._handle), platform = self._platform, confidence = confidence) @property def confidence(self) -> _int: return self._confidence @confidence.setter def confidence(self, value:_int) -> None: self._confidence = value @property def platform(self) -> Optional['_platform.Platform']: return self._platform @platform.setter def platform(self, value:'_platform.Platform') -> None: self._platform = value def mutable_copy(self) -> 'TypeBuilder': TypeBuilders = { TypeClass.VoidTypeClass:VoidBuilder, TypeClass.BoolTypeClass:BoolBuilder, TypeClass.IntegerTypeClass:IntegerBuilder, TypeClass.FloatTypeClass:FloatBuilder, TypeClass.PointerTypeClass:PointerBuilder, TypeClass.ArrayTypeClass:ArrayBuilder, TypeClass.FunctionTypeClass:FunctionBuilder, TypeClass.WideCharTypeClass:WideCharBuilder, # TypeClass.StructureTypeClass:Structure, # TypeClass.EnumerationTypeClass:Enumeration, # TypeClass.NamedTypeReferenceClass:NamedTypeReference, } builder_handle = core.BNCreateTypeBuilderFromType(self._handle) assert builder_handle is not None, "core.BNCreateTypeBuilderFromType returned None" return TypeBuilders[self.type_class](builder_handle, self.platform, self.confidence) def immutable_copy(self) -> 'Type': return self def get_builder(self, bv:'binaryview.BinaryView') -> 'MutableTypeBuilder': return MutableTypeBuilder(self.mutable_copy(), bv, self.name, self.platform, self._confidence) @staticmethod def builder(bv:'binaryview.BinaryView', name:Optional[QualifiedName]=None, id:Optional[str]=None, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'MutableTypeBuilder': type = None if name is None and id is None: raise TypeCreateException("Must specify either a name or id to create a builder object") if name is None and id is not None: type = bv.get_type_by_id(id) if type is None: raise TypeCreateException("failed to look up type by id") assert isinstance(type, NamedTypeReferenceType) registered_name = type.registered_name if registered_name is None: raise TypeCreateException("Registered name for type is None") name = registered_name.name if name is None: raise TypeCreateException("Name for registered name is None") elif name is not None: type = bv.get_type_by_name(name) if type is None: raise TypeCreateException("failed to look up type by name") assert type is not None assert name is not None return MutableTypeBuilder(type.mutable_copy(), bv, name, platform, confidence) def with_replaced_structure(self, from_struct, to_struct): return Type.create(handle = core.BNTypeWithReplacedStructure(self._handle, from_struct.handle, to_struct.handle)) def with_replaced_enumeration(self, from_enum, to_enum): return Type.create(handle = core.BNTypeWithReplacedEnumeration(self._handle, from_enum.handle, to_enum.handle)) def with_replaced_named_type_reference(self, from_ref, to_ref): return Type.create(handle = core.BNTypeWithReplacedNamedTypeReference(self._handle, from_ref.handle, to_ref.handle)) @staticmethod def void() -> 'VoidType': return VoidType.create() @staticmethod def bool() -> 'BoolType': return BoolType.create() @staticmethod def char(alternate_name:str="") -> 'CharType': return CharType.create(alternate_name) @staticmethod def int(width:_int, sign:BoolWithConfidenceType=True, alternate_name:str="") -> 'IntegerType': """ ``int`` class method for creating an int Type. :param int width: width of the integer in bytes :param bool sign: optional variable representing signedness :param str alternate_name: alternate name for type """ return IntegerType.create(width, sign, alternate_name) @staticmethod def float(width:_int, alternate_name:str="") -> 'FloatType': """ ``float`` class method for creating floating point Types. :param int width: width of the floating point number in bytes :param str alternate_name: alternate name for type """ return FloatType.create(width, alternate_name) @staticmethod def wide_char(width:_int, alternate_name:str="") -> 'WideCharType': """ ``wide_char`` class method for creating wide char Types. :param int width: width of the wide character in bytes :param str alternate_name: alternate name for type """ return WideCharType.create(width=width, alternate_name=alternate_name) @staticmethod def structure_type(structure:'StructureBuilder') -> 'StructureType': result = structure.immutable_copy() assert isinstance(result, StructureType) return result @staticmethod def named_type(named_type:'NamedTypeReferenceBuilder') -> 'NamedTypeReferenceType': result = named_type.immutable_copy() assert isinstance(result, NamedTypeReferenceType) return result @staticmethod def named_type_from_type(name:QualifiedName, type:'Type') -> 'NamedTypeReferenceType': return NamedTypeReferenceType.create_from_type(name, type) @staticmethod def named_type_from_type_and_id(type_id:str, name:QualifiedName, type:Optional['Type']=None) -> 'NamedTypeReferenceType': return NamedTypeReferenceType.create_from_type(name, type, type_id) @staticmethod def generate_named_type_reference(guid:str, name:QualifiedName) -> 'NamedTypeReferenceType': return NamedTypeReferenceType.create(NamedTypeReferenceClass.TypedefNamedTypeClass, guid, name) @staticmethod def named_type_from_registered_type(view:'binaryview.BinaryView', name:QualifiedName) -> 'NamedTypeReferenceType': return NamedTypeReferenceType.create_from_registered_type(view, name) @staticmethod def enumeration_type(arch, enum:'EnumerationBuilder', width:_int=None, sign:_bool=False) -> 'EnumerationType': return EnumerationType.create(enum.members, enum.width, arch, enum.signed) @staticmethod def pointer(arch:'architecture.Architecture', type:'Type', const:BoolWithConfidenceType=BoolWithConfidence(False), volatile:BoolWithConfidenceType=BoolWithConfidence(False), ref_type:ReferenceType=ReferenceType.PointerReferenceType, width:_int=None) -> 'PointerType': if arch is not None: width = arch.address_size if width is None: raise TypeCreateException("Must specify either an architecture or a width to create a pointer") return PointerType.create_with_width(width, type, const, volatile, ref_type) @staticmethod def pointer_of_width(width:_int, type:'Type', const:BoolWithConfidenceType=False, volatile:BoolWithConfidenceType=False, ref_type:ReferenceType=ReferenceType.PointerReferenceType) -> 'PointerType': return PointerType.create_with_width(width, type, const, volatile, ref_type) @staticmethod def array(type:'Type', count:_int) -> 'ArrayType': return ArrayType.create(type, count) @staticmethod def function(ret:Optional['Type'], params:ParamsType=[], calling_convention:'callingconvention.CallingConvention'=None, variable_arguments:BoolWithConfidenceType=False, stack_adjust:OffsetWithConfidence=OffsetWithConfidence(0)) -> 'FunctionType': """ ``function`` class method for creating an function Type. :param Type ret: return Type of the function :param params: list of parameter Types :type params: list(Type) :param CallingConvention calling_convention: optional argument for the function calling convention :param bool variable_arguments: optional boolean, true if the function has a variable number of arguments """ return FunctionType.create(ret, params, calling_convention, variable_arguments, stack_adjust) @staticmethod def from_core_struct(core_type:core.BNType): return Type.create(core.BNNewTypeReference(core_type)) @staticmethod def structure(members:MembersType=[], packed:_bool=False, type:StructureVariant=StructureVariant.StructStructureType) -> 'StructureType': return StructureType.create(members, packed, type) @staticmethod def enumeration(arch:Optional['architecture.Architecture']=None, members:List[EnumMembersType]=[], width:Optional[_int]=None, sign:BoolWithConfidenceType=False) -> 'EnumerationType': return EnumerationType.create(members, width, arch, sign) @staticmethod def named_type_reference(type_class:NamedTypeReferenceClass, name:QualifiedName, type_id:Optional[str]=None, alignment:_int=1, width:_int=0): return NamedTypeReferenceType.create(type_class, type_id, name, alignment, width) @property @abstractmethod def name(self) -> QualifiedName: raise NotImplementedError("Name not implemented for this type") @staticmethod def generate_auto_type_id(source, name:str) -> str: _name = QualifiedName(name)._to_core_struct() return core.BNGenerateAutoTypeId(source, _name) @staticmethod def generate_auto_demangled_type_id(name:str) -> str: _name = QualifiedName(name)._to_core_struct() return core.BNGenerateAutoDemangledTypeId(_name) @staticmethod def get_auto_demangled_type_id_source() -> str: return core.BNGetAutoDemangledTypeIdSource() @property def registered_name(self) -> Optional['NamedTypeReferenceType']: """Name of type registered to binary view, if any (read-only)""" ntr_handle = core.BNGetRegisteredTypeName(self._handle) if ntr_handle is None: return None return NamedTypeReferenceType(self._handle, self.platform, self.confidence, ntr_handle) @property def const(self): """Whether type is const (read/write)""" result = core.BNIsTypeConst(self._handle) return BoolWithConfidence(result.value, confidence = result.confidence) @property def volatile(self): """Whether type is volatile (read/write)""" result = core.BNIsTypeVolatile(self._handle) return BoolWithConfidence(result.value, confidence = result.confidence) @dataclass(frozen=True) class RegisterStackAdjustmentWithConfidence: value:int confidence:int=core.max_confidence def __int__(self): return self.value class VoidType(Type): @classmethod def create(cls, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'VoidType': core_void = core.BNCreateVoidType() assert core_void is not None, "core.BNCreateVoidType returned None" return cls(core.BNNewTypeReference(core_void), platform, confidence) class BoolType(Type): @classmethod def create(cls, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'BoolType': handle = core.BNCreateBoolType() assert handle is not None, "core.BNCreateBoolType returned None" return cls(core.BNNewTypeReference(handle), platform, confidence) class IntegerType(Type): def __init__(self, handle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence): super(IntegerType, self).__init__(handle, platform, confidence) @classmethod def create(cls, width:int, sign:BoolWithConfidenceType=True, alternate_name:str="", platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'IntegerType': _sign = BoolWithConfidence.get_core_struct(sign) handle = core.BNCreateIntegerType(width, _sign, alternate_name) assert handle is not None, "core.BNCreateIntegerType returned None" return cls(core.BNNewTypeReference(handle), platform, confidence) @property def signed(self) -> BoolWithConfidence: """Whether type is signed (read-only)""" return BoolWithConfidence.from_core_struct(core.BNIsTypeSigned(self._handle)) class CharType(IntegerType): @classmethod def create(cls, altname:str="char", platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'CharType': return cls(IntegerType.create(1, True, altname).handle, platform, confidence) class FloatType(Type): @classmethod def create(cls, width:int, altname:str="", platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'FloatType': """ ``float`` class method for creating floating point Types. :param int width: width of the floating point number in bytes :param str altname: alternate name for type """ core_float = core.BNCreateFloatType(width, altname) assert core_float is not None, "core.BNCreateFloatType returned None" return cls(core.BNNewTypeReference(core_float), platform, confidence) class StructureType(Type): def __init__(self, handle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence): assert handle is not None, "Attempted to create EnumerationType with handle which is None" super(StructureType, self).__init__(handle, platform, confidence) struct_handle = core.BNGetTypeStructure(handle) assert struct_handle is not None, "core.BNGetTypeStructure returned None" self.struct_handle = struct_handle @classmethod def create(cls, members:MembersType=[], packed:bool=False, type:StructureVariant=StructureVariant.StructStructureType, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'StructureType': builder = core.BNCreateStructureBuilderWithOptions(type, packed) assert builder is not None, "core.BNCreateStructureBuilder returned None" for member in members: if isinstance(member, Tuple): _type, _name = member core.BNAddStructureBuilderMember(builder, _type._to_core_struct(), _name, MemberAccess.NoAccess, MemberScope.NoScope) elif isinstance(member, StructureMember): core.BNAddStructureBuilderMemberAtOffset(builder, member.type._to_core_struct(), member.name, member.offset, False, member.access, member.scope) core_struct = core.BNFinalizeStructureBuilder(builder) assert core_struct is not None, "core.BNFinalizeStructureBuilder returned None" core_type = core.BNCreateStructureType(core_struct) assert core_type is not None, "core.BNCreateStructureType returned None" return cls(core.BNNewTypeReference(core_type), platform, confidence) def mutable_copy(self) -> 'StructureBuilder': type_builder_handle = core.BNCreateTypeBuilderFromType(self._handle) assert type_builder_handle is not None, "core.BNCreateTypeBuilderFromType returned None" structure_handle = core.BNGetTypeStructure(self._handle) assert structure_handle is not None, "core.BNGetTypeStructure returned None" structure_builder_handle = core.BNCreateStructureBuilderFromStructure(structure_handle) assert structure_builder_handle is not None, "core.BNCreateStructureBuilderFromStructure returned None" return StructureBuilder(type_builder_handle, structure_builder_handle, self.platform, self.confidence) @classmethod def from_core_struct(cls, structure:core.BNStructure) -> 'StructureType': return cls(core.BNNewTypeReference(core.BNCreateStructureType(structure))) def __del__(self): if core is not None: core.BNFreeStructure(self.struct_handle) def __eq__(self, other): if not isinstance(other, self.__class__): return NotImplemented assert other._handle is not None return ctypes.addressof(self.struct_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.struct_handle.contents)) def __getitem__(self, name:str) -> StructureMember: member = None try: member = core.BNGetStructureMemberByName(self.struct_handle, name) if member is None: raise ValueError(f"Member {name} is not part of structure") return StructureMember(Type.create(core.BNNewTypeReference(member.contents.type), confidence=member.contents.typeConfidence), member.contents.name, member.contents.offset) finally: if member is not None: core.BNFreeStructureMember(member) def member_at_offset(self, offset:int) -> StructureMember: member = None try: member = core.BNGetStructureMemberAtOffset(self.struct_handle, offset, None) if member is None: raise ValueError(f"No member exists a offset {offset}") return StructureMember(Type.create(core.BNNewTypeReference(member.contents.type), confidence=member.contents.typeConfidence), member.contents.name, member.contents.offset) finally: core.BNFreeStructureMember(member) @property def members(self): """Structure member list (read-only)""" count = ctypes.c_ulonglong() members = core.BNGetStructureMembers(self.struct_handle, count) assert members is not None, "core.BNGetStructureMembers returned None" try: result = [] for i in range(0, count.value): result.append(StructureMember(Type.create(core.BNNewTypeReference(members[i].type), confidence=members[i].typeConfidence), members[i].name, members[i].offset)) finally: core.BNFreeStructureMemberList(members, count.value) return result @property def width(self): """Structure width""" return core.BNGetStructureWidth(self.struct_handle) @property def alignment(self): """Structure alignment""" return core.BNGetStructureAlignment(self.struct_handle) @property def packed(self): return core.BNIsStructurePacked(self.struct_handle) @property def type(self) -> StructureVariant: return StructureVariant(core.BNGetStructureType(self.struct_handle)) def with_replaced_structure(self, from_struct, to_struct) -> 'StructureType': return StructureType(core.BNStructureWithReplacedStructure(self.struct_handle, from_struct.handle, to_struct.handle)) def with_replaced_enumeration(self, from_enum, to_enum) -> 'StructureType': return StructureType(core.BNStructureWithReplacedEnumeration(self.struct_handle, from_enum.handle, to_enum.handle)) def with_replaced_named_type_reference(self, from_ref, to_ref) -> 'StructureType': return StructureType(core.BNStructureWithReplacedNamedTypeReference(self.struct_handle, from_ref.handle, to_ref.handle)) def generate_named_type_reference(self, guid:str, name:QualifiedName): if self.type == StructureVariant.StructStructureType: ntr_type = NamedTypeReferenceClass.StructNamedTypeClass elif self.type == StructureVariant.UnionStructureType: ntr_type = NamedTypeReferenceClass.UnionNamedTypeClass else: ntr_type = NamedTypeReferenceClass.ClassNamedTypeClass return NamedTypeReferenceType.create(ntr_type, guid, name, self.alignment, self.width, self.platform, self.confidence) class EnumerationType(IntegerType): def __init__(self, handle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence): assert handle is not None, "Attempted to create EnumerationType without handle" super(EnumerationType, self).__init__(handle, platform, confidence) enum_handle = core.BNGetTypeEnumeration(handle) core.BNNewEnumerationReference(enum_handle) assert enum_handle is not None, "core.BNGetTypeEnumeration returned None" self.enum_handle = enum_handle @classmethod def create(cls, members=List[EnumMembersType], width:Optional[int]=None, arch:Optional['architecture.Architecture']=None, sign:BoolWithConfidenceType=False, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'EnumerationType': if width is None: if arch is None: raise ValueError("One of the following parameters must not be None: (arch, width)") width = arch.default_int_size if width == 0: raise ValueError("enumeration width must not be 0") builder = core.BNCreateEnumerationBuilder() assert builder is not None, "core.BNCreateEnumerationType returned None" EnumerationBuilder.add_members(builder, members) core_enum = core.BNFinalizeEnumerationBuilder(builder) assert core_enum is not None, "core.BNFinalizeEnumerationBuilder returned None" core.BNFreeEnumerationBuilder(builder) core_type = core.BNCreateEnumerationTypeOfWidth(core_enum, width, BoolWithConfidence.get_core_struct(sign)) assert core_type is not None, "core.BNCreateEnumerationTypeOfWidth returned None" return cls(core.BNNewTypeReference(core_type), platform, confidence) def mutable_copy(self) -> 'EnumerationBuilder': type_builder_handle = core.BNCreateTypeBuilderFromType(self._handle) assert type_builder_handle is not None, "core.BNCreateTypeBuilderFromType returned None" enumeration_handle = core.BNGetTypeEnumeration(self._handle) assert enumeration_handle is not None, "core.BNGetTypeStructure returned None" enumeration_builder_handle = core.BNCreateEnumerationTypeBuilder(enumeration_handle) assert enumeration_builder_handle is not None, "core.BNCreateEnumerationTypeBuilder returned None" return EnumerationBuilder(type_builder_handle, enumeration_builder_handle, self.platform, self.confidence) def __del__(self): if core is not None: core.BNFreeEnumeration(self.enum_handle) # def __repr__(self): # return "" % repr(self.members) def __eq__(self, other): if not isinstance(other, self.__class__): return NotImplemented return ctypes.addressof(self.enum_handle.contents) == ctypes.addressof(other.enum_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.enum_handle.contents)) @property def members(self): """Enumeration member list (read-only)""" count = ctypes.c_ulonglong() members = core.BNGetEnumerationMembers(self.enum_handle, count) assert members is not None, "core.BNGetEnumerationMembers returned None" result = [] for i in range(0, count.value): result.append(EnumerationMember(members[i].name, members[i].value)) core.BNFreeEnumerationMemberList(members, count.value) return result def generate_named_type_reference(self, guid:str, name:QualifiedName): ntr_type = NamedTypeReferenceClass.EnumNamedTypeClass return NamedTypeReferenceType.create(ntr_type, guid, name, platform=self.platform, confidence=self.confidence) class PointerType(Type): @property def ref_type(self) -> ReferenceType: return core.BNTypeGetReferenceType(self._handle) @classmethod def create(cls, arch:'architecture.Architecture', type:SomeType, const:BoolWithConfidenceType=False, volatile:BoolWithConfidenceType=False, ref_type:ReferenceType=ReferenceType.PointerReferenceType, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'PointerType': return cls.create_with_width(arch.address_size, type, const, volatile, ref_type, platform, confidence) @staticmethod def from_bools(const:BoolWithConfidenceType, volatile:BoolWithConfidenceType) -> Tuple[BoolWithConfidence, BoolWithConfidence]: _const = const if const is None: _const = BoolWithConfidence(False, confidence = 0) elif isinstance(const, bool): _const = BoolWithConfidence(const) if not isinstance(_const, BoolWithConfidence): raise ValueError(f"unhandled type {type(const)} for 'const' argument") _volatile = volatile if volatile is None: _volatile = BoolWithConfidence(False, confidence = 0) elif isinstance(volatile, bool): _volatile = BoolWithConfidence(volatile) if not isinstance(_volatile, BoolWithConfidence): raise ValueError(f"unhandled type {type(volatile)} for 'volatile' argument") return (_const, _volatile) @classmethod def create_with_width(cls, width:int, type:SomeType, const:BoolWithConfidenceType=False, volatile:BoolWithConfidenceType=False, ref_type:ReferenceType=None, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'PointerType': _const, _volatile = PointerType.from_bools(const, volatile) type = type.immutable_copy() if ref_type is None: ref_type = ReferenceType.PointerReferenceType type_conf = type._to_core_struct() core_type = core.BNCreatePointerTypeOfWidth(width, type_conf, _const._to_core_struct(), _volatile._to_core_struct(), ref_type) assert core_type is not None, "core.BNCreatePointerTypeOfWidth returned None" return cls(core.BNNewTypeReference(core_type), platform, confidence) @property def target(self) -> Type: """Target (read-only)""" result = core.BNGetChildType(self._handle) assert result is not None, "core.BNGetChildType returned None" return Type.create(core.BNNewTypeReference(result.type), self._platform, result.confidence) class ArrayType(Type): @classmethod def create(cls, element_type:Type, count:int, platform:'_platform.Platform'=None, confidence:int=core.max_confidence): type_conf = element_type._to_core_struct() core_array = core.BNCreateArrayType(type_conf, count) assert core_array is not None, "core.BNCreateArrayType returned None" return cls(core.BNNewTypeReference(core_array)) @property def count(self): """Type count (read-only)""" return core.BNGetTypeElementCount(self._handle) @property def element_type(self) -> Type: result = core.BNGetChildType(self._handle) assert result is not None, "core.BNGetChildType returned None" return Type.create(core.BNNewTypeReference(result.type), self._platform, result.confidence) class FunctionType(Type): @classmethod def create(cls, ret:Optional[Type]=None, params:ParamsType=[], calling_convention:'callingconvention.CallingConvention'=None, variable_arguments:BoolWithConfidenceType=BoolWithConfidence(False), stack_adjust:OffsetWithConfidence=OffsetWithConfidence(0), platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'FunctionType': if ret is None: ret = VoidType.create() param_buf = FunctionBuilder._to_core_struct(params) ret_conf = ret._to_core_struct() conv_conf = core.BNCallingConventionWithConfidence() if calling_convention is None: conv_conf.convention = None conv_conf.confidence = 0 else: conv_conf.convention = calling_convention.handle conv_conf.confidence = calling_convention.confidence _variable_arguments = BoolWithConfidence.get_core_struct(variable_arguments) _stack_adjust = OffsetWithConfidence.get_core_struct(stack_adjust) func_type = core.BNCreateFunctionType(ret_conf, conv_conf, param_buf, len(params), _variable_arguments, _stack_adjust) assert func_type is not None, f"core.BNCreateFunctionType returned None {ret_conf} {conv_conf} {param_buf} {_variable_arguments} {_stack_adjust}" return cls(core.BNNewTypeReference(func_type), platform, confidence) @property def stack_adjustment(self) -> OffsetWithConfidence: """Stack adjustment for function (read-only)""" result = core.BNGetTypeStackAdjustment(self._handle) return OffsetWithConfidence(result.value, confidence = result.confidence) @property def return_value(self) -> Type: """Return value (read-only)""" result = core.BNGetChildType(self._handle) if result is None: return Type.void() return Type.create(core.BNNewTypeReference(result.type), platform = self._platform, confidence = result.confidence) @property def calling_convention(self) -> Optional[callingconvention.CallingConvention]: """Calling convention (read-only)""" result = core.BNGetTypeCallingConvention(self._handle) if not result.convention: return None return callingconvention.CallingConvention(None, handle = result.convention, confidence = result.confidence) @property def parameters(self) -> List[FunctionParameter]: """Type parameters list (read-only)""" count = ctypes.c_ulonglong() params = core.BNGetTypeParameters(self._handle, count) assert params is not None, "core.BNGetTypeParameters returned None" result = [] for i in range(0, count.value): param_type = Type.create(core.BNNewTypeReference(params[i].type), platform = self._platform, confidence = params[i].typeConfidence) if params[i].defaultLocation: param_location = None else: name = params[i].name if (params[i].location.type == VariableSourceType.RegisterVariableSourceType) and (self._platform is not None): name = self._platform.arch.get_reg_name(params[i].location.storage) elif params[i].location.type == VariableSourceType.StackVariableSourceType: name = "arg_%x" % params[i].location.storage param_location = variable.VariableNameAndType(params[i].location.type, params[i].location.index, params[i].location.storage, name, param_type) result.append(FunctionParameter(param_type, params[i].name, param_location)) core.BNFreeTypeParameterList(params, count.value) return result @property def has_variable_arguments(self) -> BoolWithConfidence: """Whether type has variable arguments (read-only)""" result = core.BNTypeHasVariableArguments(self._handle) return BoolWithConfidence(result.value, confidence = result.confidence) @property def can_return(self) -> BoolWithConfidence: """Whether type can return""" result = core.BNFunctionTypeCanReturn(self._handle) return BoolWithConfidence(result.value, confidence = result.confidence) class NamedTypeReferenceType(Type): def __init__(self, handle, platform:'_platform.Platform'=None, confidence:int=core.max_confidence, ntr_handle=None): assert handle is not None, "Attempting to create NamedTypeReferenceType handle which is None" super(NamedTypeReferenceType, self).__init__(handle, platform, confidence) if ntr_handle is None: ntr_handle = core.BNGetTypeNamedTypeReference(handle) assert ntr_handle is not None, "core.BNGetTypeNamedTypeReference returned None" self.ntr_handle = ntr_handle def mutable_copy(self): type_builder_handle = core.BNCreateTypeBuilderFromType(self._handle) assert type_builder_handle is not None, "core.BNCreateTypeBuilderFromType returned None" ntr_builder_handle = core.BNCreateNamedTypeBuilder(self.named_type_class, self.type_id, self.name._to_core_struct()) assert ntr_builder_handle is not None, "core.BNCreateNamedTypeBuilder returned None" return NamedTypeReferenceBuilder(type_builder_handle, ntr_builder_handle, self.platform, self.confidence) @classmethod def create(cls, named_type_class:NamedTypeReferenceClass, guid:Optional[str], name:QualifiedName, alignment:int=0, width:int=0, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'NamedTypeReferenceType': _guid = guid if guid is None: _guid = str(uuid.uuid4()) _name = QualifiedName(name)._to_core_struct() core_ntr = core.BNCreateNamedType(named_type_class, _guid, _name) assert core_ntr is not None, "core.BNCreateNamedType returned None" core_type = core.BNCreateNamedTypeReference(core_ntr, width, alignment) assert core_type is not None, "core.BNCreateNamedTypeReference returned None" return cls(core.BNNewTypeReference(core_type), platform, confidence) @classmethod def create_from_type(cls, name:QualifiedName, type:Optional[Type], guid:Optional[str]=None, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'NamedTypeReferenceType': _guid = guid if _guid is None: _guid = str(uuid.uuid4()) if type is None: return cls.create(NamedTypeReferenceClass.UnknownNamedTypeClass, _guid, name) else: return type.generate_named_type_reference(_guid, name) @classmethod def create_from_registered_type(cls, view:'binaryview.BinaryView', name:QualifiedName, platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'NamedTypeReferenceType': _name = QualifiedName(name)._to_core_struct() core_type = core.BNCreateNamedTypeReferenceFromType(view.handle, _name) assert core_type is not None, "core.BNCreateNamedTypeReferenceFromType returned None" return cls(core.BNNewTypeReference(core_type), platform, confidence) def __del__(self): if core is not None: core.BNFreeNamedTypeReference(self.ntr_handle) def __repr__(self): if self.named_type_class == NamedTypeReferenceClass.TypedefNamedTypeClass: return f"" if self.named_type_class == NamedTypeReferenceClass.StructNamedTypeClass: return f"" if self.named_type_class == NamedTypeReferenceClass.UnionNamedTypeClass: return f"" if self.named_type_class == NamedTypeReferenceClass.EnumNamedTypeClass: return f"" return "" def __str__(self): name = self.registered_name if name is None: name = "" else: name = " " + str(name.name) return f"{self.get_string_before_name()}{name}{self.get_string_after_name()}" def __eq__(self, other): if not isinstance(other, self.__class__): return NotImplemented assert other._handle is not None return ctypes.addressof(self.ntr_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.ntr_handle.contents)) @property def named_type_class(self) -> NamedTypeReferenceClass: return NamedTypeReferenceClass(core.BNGetTypeReferenceClass(self.ntr_handle)) @property def type_id(self) -> str: return core.BNGetTypeReferenceId(self.ntr_handle) @property def name(self) -> QualifiedName: name = core.BNGetTypeReferenceName(self.ntr_handle) result = QualifiedName._from_core_struct(name) core.BNFreeQualifiedName(name) return result @staticmethod def generate_auto_type_ref(type_class, source, name): type_id = Type.generate_auto_type_id(source, name) return NamedTypeReferenceType.create(type_class, type_id, name) @staticmethod def generate_auto_demangled_type_ref(type_class, name): type_id = Type.generate_auto_demangled_type_id(name) return NamedTypeReferenceType.create(type_class, type_id, name) def _target_helper(self, bv:'binaryview.BinaryView', type_ids=set()) -> Optional[Type]: t = bv.get_type_by_id(self.type_id) if t is None: return None if isinstance(t, NamedTypeReferenceType): if t.type_id in type_ids: raise TypeError("Can't get target for recursively defined type") type_ids.add(t) return t._target_helper(bv, type_ids) else: return t def target(self, bv:'binaryview.BinaryView') -> Optional[Type]: """Returns the type pointed to by the current type :param bv: The BinaryView in which this type is defined. :type bv: binaryview.BinaryView :return: The type this NamedTypeReference is referencing :rtype: Optional[Type] """ return self._target_helper(bv) class WideCharType(Type): @classmethod def create(cls, width:int, alternate_name:str="", platform:'_platform.Platform'=None, confidence:int=core.max_confidence) -> 'WideCharType': """ ``wide_char`` class method for creating wide char Types. :param int width: width of the wide character in bytes :param str alternate_name: alternate name for type """ core_type = core.BNCreateWideCharType(width, alternate_name) assert core_type is not None, "core.BNCreateWideCharType returned None" return cls(core.BNNewTypeReference(core_type), platform, confidence) Types = { TypeClass.VoidTypeClass:VoidType, TypeClass.BoolTypeClass:BoolType, TypeClass.IntegerTypeClass:IntegerType, TypeClass.FloatTypeClass:FloatType, TypeClass.StructureTypeClass:StructureType, TypeClass.EnumerationTypeClass:EnumerationType, TypeClass.PointerTypeClass:PointerType, TypeClass.ArrayTypeClass:ArrayType, TypeClass.FunctionTypeClass:FunctionType, TypeClass.NamedTypeReferenceClass:NamedTypeReferenceType, TypeClass.WideCharTypeClass:WideCharType, } @dataclass(frozen=True) class RegisterSet: regs:List['architecture.RegisterName'] confidence:int=core.max_confidence def __iter__(self) -> Generator['architecture.RegisterName', None, None]: for reg in self.regs: yield reg def __getitem__(self, idx): return self.regs[idx] def __len__(self): return len(self.regs) def with_confidence(self, confidence): return RegisterSet(list(self.regs), confidence=confidence) @dataclass(frozen=True) class TypeParserResult: types:Mapping[QualifiedName, Type] variables:Mapping[QualifiedName, Type] functions:Mapping[QualifiedName, Type] def __repr__(self): return f"" def preprocess_source(source:str, filename:str=None, include_dirs:List[str]=[]) -> Tuple[Optional[str], str]: """ ``preprocess_source`` run the C preprocessor on the given source or source filename. :param str source: source to pre-process :param str filename: optional filename to pre-process :param include_dirs: list of string directories to use as include directories. :type include_dirs: list(str) :return: returns a tuple of (preprocessed_source, error_string) :rtype: tuple(str,str) :Example: >>> source = "#define TEN 10\\nint x[TEN];\\n" >>> preprocess_source(source) ('#line 1 "input"\\n\\n#line 2 "input"\\n int x [ 10 ] ;\\n', '') >>> """ if filename is None: filename = "input" dir_buf = (ctypes.c_char_p * len(include_dirs))() for i in range(0, len(include_dirs)): dir_buf[i] = include_dirs[i].encode('charmap') output = ctypes.c_char_p() errors = ctypes.c_char_p() result = core.BNPreprocessSource(source, filename, output, errors, dir_buf, len(include_dirs)) assert output.value is not None assert errors.value is not None output_str = output.value.decode('utf-8') error_str = errors.value.decode('utf-8') core.free_string(output) core.free_string(errors) if result: return (output_str, error_str) return (None, error_str) @dataclass(frozen=True) class TypeFieldReference: func:Optional['_function.Function'] arch:Optional['architecture.Architecture'] address:int size:int incomingType:Optional[Type] def __repr__(self): if self.arch: return f"" else: return f""