Add GetAddressSize method to Platform

Author: zznop

binaryninjaapi.h

@@ -15451,6 +15451,7 @@ namespace BinaryNinja {
 		static void InitViewCallback(void* ctxt, BNBinaryView* view);
 		static uint32_t* GetGlobalRegistersCallback(void* ctxt, size_t* count);
 		static void FreeRegisterListCallback(void* ctxt, uint32_t* regs, size_t count);
+		static size_t GetAddressSizeCallback(void* ctxt);
 		static BNType* GetGlobalRegisterTypeCallback(void* ctxt, uint32_t reg);
 		static void AdjustTypeParserInputCallback(
 			void* ctxt,
@@ -15635,6 +15636,12 @@ namespace BinaryNinja {
 		 */
 		virtual Ref<Type> GetGlobalRegisterType(uint32_t reg);
 
+		/*! Get the address size for this platform
+
+			\return The address size for this platform
+		*/
+		virtual size_t GetAddressSize() const;
+
 		/*! Modify the input passed to the Type Parser with Platform-specific features.
 
 			\param[in] parser Type Parser instance
@@ -15762,6 +15769,7 @@ namespace BinaryNinja {
 			std::vector<std::string>& arguments,
 			std::vector<std::pair<std::string, std::string>>& sourceFiles
 		) override;
+		virtual size_t GetAddressSize() const override;
 	};
 
 	/*!

binaryninjacore.h

@@ -1926,6 +1926,8 @@ extern "C"
 
 		BNType* (*getGlobalRegisterType)(void* ctxt, uint32_t reg);
 
+		size_t (*getAddressSize)(void* ctxt);
+
 		void (*adjustTypeParserInput)(
 			void* ctxt,
 			BNTypeParser* parser,
@@ -6940,6 +6942,7 @@ extern "C"
 
 	BINARYNINJACOREAPI uint32_t* BNGetPlatformGlobalRegisters(BNPlatform* platform, size_t* count);
 	BINARYNINJACOREAPI BNType* BNGetPlatformGlobalRegisterType(BNPlatform* platform, uint32_t reg);
+	BINARYNINJACOREAPI size_t BNGetPlatformAddressSize(BNPlatform* platform);
 	BINARYNINJACOREAPI void BNPlatformAdjustTypeParserInput(
 		BNPlatform* platform,
 		BNTypeParser* parser,

platform.cpp

@@ -41,6 +41,7 @@ Platform::Platform(Architecture* arch, const string& name)
 	plat.viewInit = InitViewCallback;
 	plat.getGlobalRegisters = GetGlobalRegistersCallback;
 	plat.freeRegisterList = FreeRegisterListCallback;
+	plat.getAddressSize = GetAddressSizeCallback;
 	plat.getGlobalRegisterType = GetGlobalRegisterTypeCallback;
 	plat.adjustTypeParserInput = AdjustTypeParserInputCallback;
 	plat.freeTypeParserInput = FreeTypeParserInputCallback;
@@ -59,6 +60,7 @@ Platform::Platform(Architecture* arch, const string& name, const string& typeFil
 	plat.getGlobalRegisters = GetGlobalRegistersCallback;
 	plat.freeRegisterList = FreeRegisterListCallback;
 	plat.getGlobalRegisterType = GetGlobalRegisterTypeCallback;
+	plat.getAddressSize = GetAddressSizeCallback;
 	plat.adjustTypeParserInput = AdjustTypeParserInputCallback;
 	plat.freeTypeParserInput = FreeTypeParserInputCallback;
 	plat.getFallbackEnabled = GetFallbackEnabledCallback;
@@ -193,6 +195,14 @@ BNType* Platform::GetGlobalRegisterTypeCallback(void* ctxt, uint32_t reg)
 	return BNNewTypeReference(result->GetObject());
 }
 
+
+size_t Platform::GetAddressSizeCallback(void* ctxt)
+{
+	CallbackRef<Platform> plat(ctxt);
+	return plat->GetAddressSize();
+}
+
+
 bool Platform::GetFallbackEnabledCallback(void* ctxt)
 {
 	CallbackRef<Platform> plat(ctxt);
@@ -424,6 +434,12 @@ bool Platform::GetFallbackEnabled()
 }
 
 
+size_t Platform::GetAddressSize() const
+{
+	return GetArchitecture()->GetAddressSize();
+}
+
+
 std::vector<uint32_t> CorePlatform::GetGlobalRegisters()
 {
 	size_t count;
@@ -448,6 +464,12 @@ Ref<Type> CorePlatform::GetGlobalRegisterType(uint32_t reg)
 }
 
 
+size_t CorePlatform::GetAddressSize() const
+{
+	return BNGetPlatformAddressSize(m_object);
+}
+
+
 void Platform::AdjustTypeParserInput(
 	Ref<TypeParser> parser,
 	vector<string>& arguments,

python/platform.py

@@ -85,6 +85,7 @@ def __init__(self, arch: Optional['architecture.Architecture'] = None, handle=No
 			self._cb.getGlobalRegisters = self._cb.getGlobalRegisters.__class__(self._get_global_regs)
 			self._cb.freeRegisterList = self._cb.freeRegisterList.__class__(self._free_register_list)
 			self._cb.getGlobalRegisterType = self._cb.getGlobalRegisterType.__class__(self._get_global_reg_type)
+			self._cb.getAddressSize = self._cb.getAddressSize.__class__(lambda ctxt: self._get_address_size)
 			self._cb.adjustTypeParserInput = self._cb.adjustTypeParserInput.__class__(self._adjust_type_parser_input)
 			self._cb.freeTypeParserInput = self._cb.freeTypeParserInput.__class__(self._free_type_parser_input)
 			self._pending_reg_lists = {}
@@ -168,6 +169,13 @@ def _get_global_reg_type(self, ctxt, reg):
 			log_error(traceback.format_exc())
 			return None
 
+	def _get_address_size(self, ctxt):
+		try:
+			return core.BNGetPlatformAddressSize(self.handle)
+		except:
+			log_error(traceback.format_exc())
+			return None
+
 	def _adjust_type_parser_input(
 			self,
 			ctxt,