accessibility: raise RpcFailedToGetPrimaryAppError if failed to get it

doronz88 · Oct 22, 2024 · e1e7445 · e1e7445
1 parent 7e9c3df
commit e1e7445
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/rpcclient/rpcclient/exceptions.py b/src/rpcclient/rpcclient/exceptions.py
@@ -15,6 +15,7 @@ class ResponseNotFoundError(RpcClientException):
 
 class ServerResponseError(RpcClientException):
     """ Server returned error """
+
     def __init__(self, error):
         super().__init__()
         self.error = error
@@ -197,3 +198,8 @@ class RpcXpcSerializationError(RpcXpcError):
 class RpcSetDeveloperModeError(BadReturnValueError):
     """ Failed to set Developer Mode """
     pass
+
+
+class RpcFailedToGetPrimaryAppError(BadReturnValueError):
+    """ Failed to get [AXElement primaryApp] """
+    pass
diff --git a/src/rpcclient/rpcclient/ios/accessibility.py b/src/rpcclient/rpcclient/ios/accessibility.py
@@ -5,7 +5,7 @@
 
 from rpcclient.darwin.symbol import DarwinSymbol
 from rpcclient.exceptions import ElementNotFoundError, FirstElementNotFoundError, LastElementNotFoundError, \
-    MissingLibraryError, RpcAccessibilityTurnedOffError
+    MissingLibraryError, RpcAccessibilityTurnedOffError, RpcFailedToGetPrimaryAppError
 from rpcclient.structs.consts import RTLD_NOW
 
 
@@ -393,7 +393,10 @@ def __init__(self, client):
     def primary_app(self) -> AXElement:
         if not self.enabled:
             raise RpcAccessibilityTurnedOffError()
-        return self.axelement(self._client.symbols.objc_getClass('AXElement').objc_call('primaryApp'))
+        primary_app = self._client.symbols.objc_getClass('AXElement').objc_call('primaryApp')
+        if primary_app == 0:
+            raise RpcFailedToGetPrimaryAppError()
+        return self.axelement(primary_app)
 
     @property
     def enabled(self) -> bool: