[SharedCache] Add the ability to manually trigger Objective-C processing

A problem with processing Objective-C sections at the time a library is loaded is that some of the references within those sections may refer to unload sections. This results in things like selectors not being correctly typed and named.

This commit provides a way for users to manually trigger Objective-C parsing against sections for a specific library or all libraries, via the API. Combined with the previous commit a user can use the API to batch load a number of libraries and skip Objective-C processing for each one and then run it across the entire of the DSC once they are all loaded.

Further improvement would be to provide a way to trigger Objective-C processing through the UI.

2 files changed, 44 insertions, 0 deletions

diff --git a/view/sharedcache/api/python/_sharedcachecore.py b/view/sharedcache/api/python/_sharedcachecore.py
index bd9e0764..5f31697e 100644
--- a/view/sharedcache/api/python/_sharedcachecore.py
+++ b/view/sharedcache/api/python/_sharedcachecore.py
@@ -582,6 +582,44 @@ def BNDSCViewLoadSectionAtAddress(
 
 
 # -------------------------------------------------------
+# _BNDSCViewProcessAllObjCSections
+
+_BNDSCViewProcessAllObjCSections = core.BNDSCViewProcessAllObjCSections
+_BNDSCViewProcessAllObjCSections.restype = None
+_BNDSCViewProcessAllObjCSections.argtypes = [
+		ctypes.POINTER(BNSharedCache),
+	]
+
+
+# noinspection PyPep8Naming
+def BNDSCViewProcessAllObjCSections(
+		cache: ctypes.POINTER(BNSharedCache)
+		) -> None:
+	return _BNDSCViewProcessAllObjCSections(cache)
+
+
+# -------------------------------------------------------
+# _BNDSCViewProcessObjCSectionsForImageWithInstallName
+
+_BNDSCViewProcessObjCSectionsForImageWithInstallName = core.BNDSCViewProcessObjCSectionsForImageWithInstallName
+_BNDSCViewProcessObjCSectionsForImageWithInstallName.restype = None
+_BNDSCViewProcessObjCSectionsForImageWithInstallName.argtypes = [
+		ctypes.POINTER(BNSharedCache),
+		ctypes.c_char_p,
+		ctypes.c_bool,
+	]
+
+
+# noinspection PyPep8Naming
+def BNDSCViewProcessObjCSectionsForImageWithInstallName(
+		cache: ctypes.POINTER(BNSharedCache), 
+		name: Optional[str], 
+		deallocName: bool
+		) -> None:
+	return _BNDSCViewProcessObjCSectionsForImageWithInstallName(cache, cstr(name), deallocName)
+
+
+# -------------------------------------------------------
 # _BNFreeSharedCacheReference
 
 _BNFreeSharedCacheReference = core.BNFreeSharedCacheReference
diff --git a/view/sharedcache/api/python/sharedcache.py b/view/sharedcache/api/python/sharedcache.py
index 7902c76a..e0e18bde 100644
--- a/view/sharedcache/api/python/sharedcache.py
+++ b/view/sharedcache/api/python/sharedcache.py
@@ -117,6 +117,12 @@ class SharedCache:
 	def load_image_containing_address(self, addr, skipObjC = False):
 		return sccore.BNDSCViewLoadImageContainingAddress(self.handle, addr, skipObjC)
 
+	def process_objc_sections_for_image_with_install_name(self, installName):
+		return sccore.BNDSCViewProcessObjCSectionsForImageWithInstallName(self.handle, installName, False)
+
+	def process_all_objc_sections(self):
+		return sccore.BNDSCViewProcessAllObjCSections(self.handle)
+
 	@property
 	def caches(self):
 		count = ctypes.c_ulonglong()