lamda 7.76__tar.gz → 7.85__tar.gz

{lamda-7.76 → lamda-7.85}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lamda
-Version: 7.76
+Version: 7.85
 Summary: Android reverse engineering & automation framework (Client API)
 Home-page: https://github.com/firerpa/lamda
 Author: rev1si0n

{lamda-7.76 → lamda-7.85}/lamda/__init__.py

@@ -2,4 +2,4 @@
 #
 # Distributed under MIT license.
 # See file LICENSE for detail or copy at https://opensource.org/licenses/MIT
-__version__ = "7.76"
+__version__ = "7.85"

{lamda-7.76 → lamda-7.85}/lamda/client.py

@@ -57,6 +57,8 @@ logger.addHandler(handler)
 
 sys.path.append(joinpath(dirname(__file__)))
 sys.path.append(joinpath(dirname(__file__), "rpc"))
+# use native resolver to support mDNS
+os.environ["GRPC_DNS_RESOLVER"] = "native"
 
 protos, services = grpc.protos_and_services("services.proto")
 __all__ = [
@@ -179,6 +181,8 @@ def contain(a, b):
                 b.right <= a.right])
 
 def equal(a, b):
+    if not isinstance(b, protos.Bound):
+        return False
     return all([b.top == a.top,
                 b.left == a.left,
                 b.bottom == a.bottom,
@@ -194,6 +198,8 @@ Corner = protos.Corner
 Direction = protos.Direction
 GproxyType = protos.GproxyType
 GrantType = protos.GrantType
+ScriptRuntime = protos.ScriptRuntime
+DataEncode = protos.DataEncode
 
 Group = protos.Group
 Key = protos.Key
@@ -256,6 +262,9 @@ TouchSequence.appendUp = touchSequenceAppendUp
 TouchSequence.__getitem__ = touchSequenceIndexer
 TouchSequence.__iter__ = touchSequenceIter
 
+HookRpcRequest = protos.HookRpcRequest
+HookRpcResponse = protos.HookRpcResponse
+
 Bound.width = property(width)
 Bound.height = property(height)
 
@@ -380,13 +389,14 @@ class GrpcRemoteExceptionInterceptor(ClientInterceptor):
 
 
 class ObjectUiAutomatorOpStub:
-    def __init__(self, stub, selector):
+    def __init__(self, caller, selector):
         """
         UiAutomator 子接口，用来模拟出实例的意味
         """
         self._selector = selector
         self.selector = Selector(**selector)
-        self.stub = stub
+        self.stub = caller.stub
+        self.caller = caller
     def __str__(self):
         selector = ", ".join(["{}={}".format(k, v) \
                         for k, v in self._selector.items()])
@@ -489,6 +499,72 @@ class ObjectUiAutomatorOpStub:
         req = protos.SelectorOnlyRequest(selector=self.selector)
         r = self.stub.selectorObjInfoOfAllInstances(req)
         return r.objects
+    def all_instances(self):
+        """
+        获取选择器选中的所有元素控件
+        """
+        return list(self)
+    def _new_object(self, **kwargs):
+        selector = copy.deepcopy(self._selector)
+        selector.update(**kwargs)
+        instance = self.caller(**selector)
+        return instance
+    def text(self, txt):
+        return self._new_object(text=txt)
+    def resourceId(self, name):
+        return self._new_object(resourceId=name)
+    def description(self, desc):
+        return self._new_object(description=desc)
+    def packageName(self, name):
+        return self._new_object(packageName=name)
+    def className(self, name):
+        return self._new_object(className=name)
+    def textContains(self, needle):
+        return self._new_object(textContains=needle)
+    def descriptionContains(self, needle):
+        return self._new_object(descriptionContains=needle)
+    def textStartsWith(self, needle):
+        return self._new_object(textStartsWith=needle)
+    def descriptionStartsWith(self, needle):
+        return self._new_object(descriptionStartsWith=needle)
+    def textMatches(self, match):
+        return self._new_object(textMatches=match)
+    def descriptionMatches(self, match):
+        return self._new_object(descriptionMatches=match)
+    def resourceIdMatches(self, match):
+        return self._new_object(resourceIdMatches=match)
+    def packageNameMatches(self, match):
+        return self._new_object(packageNameMatches=match)
+    def classNameMatches(self, match):
+        return self._new_object(classNameMatches=match)
+    def checkable(self, value):
+        return self._new_object(checkable=value)
+    def clickable(self, value):
+        return self._new_object(clickable=value)
+    def focusable(self, value):
+        return self._new_object(focusable=value)
+    def scrollable(self, value):
+        return self._new_object(scrollable=value)
+    def longClickable(self, value):
+        return self._new_object(longClickable=value)
+    def enabled(self, value):
+        return self._new_object(enabled=value)
+    def checked(self, value):
+        return self._new_object(checked=value)
+    def focused(self, value):
+        return self._new_object(focused=value)
+    def selected(self, value):
+        return self._new_object(selected=value)
+    def index(self, idx):
+        return self._new_object(index=idx)
+    def instance(self, idx):
+        return self._new_object(instance=idx)
+    def __iter__(self):
+        """
+        遍历所有符合选择器条件的元素实例
+        """
+        yield from [self.instance(i) for i in \
+                            range(self.count())]
     def count(self):
         """
         获取选择器选中控件的数量
@@ -957,7 +1033,25 @@ class UiAutomatorStub(BaseServiceStub):
         r = self.stub.waitForIdle(protos.Integer(value=timeout))
         return r.value
     def __call__(self, **kwargs):
-        return ObjectUiAutomatorOpStub(self.stub, kwargs)
+        return ObjectUiAutomatorOpStub(self, kwargs)
+
+
+class AppScriptRpcInterface(object):
+    def __init__(self, stub, application,
+                                    name):
+        self.application = application
+        self.stub = stub
+        self.name = name
+    def __call__(self, *args):
+        call_args = dict()
+        call_args["method"] = self.name
+        call_args["args"] = args
+        req = HookRpcRequest()
+        req.package = self.application.applicationId
+        req.callinfo = json.dumps(call_args)
+        result = self.stub.callScript(req)
+        r = json.loads(result.callresult)
+        return r
 
 
 class ApplicationOpStub:
@@ -1114,6 +1208,51 @@ class ApplicationOpStub:
         req.user = self.user
         r = self.stub.isInstalled(req)
         return r.value
+    def attach_script(self, script, runtime=ScriptRuntime.RUNTIME_QJS,
+                                                    emit="",
+                                encode=DataEncode.DATA_ENCODE_NONE,
+                                standup=5):
+        """
+        向应用注入持久化 Hook 脚本
+        """
+        s = isinstance(script, str)
+        script = script.encode() if s else script
+        req = protos.HookRequest()
+        req.package     = self.applicationId
+        req.script      = script
+        req.runtime     = runtime
+        req.standup     = standup
+        req.destination = emit
+        req.encode      = encode
+        r = self.stub.attachScript(req)
+        return r.value
+    def detach_script(self):
+        """
+        移除注入应用的 Hook 脚本
+        """
+        req = protos.String(value=self.applicationId)
+        r = self.stub.detachScript(req)
+        return r.value
+    def is_attached_script(self):
+        """
+        检查使用在此应用注入了 Hook 脚本
+        """
+        req = protos.String(value=self.applicationId)
+        r = self.stub.isScriptAttached(req)
+        return r.value
+    def is_script_alive(self):
+        """
+        检查应用中的 Hook 脚本是否正常
+        """
+        req = protos.String(value=self.applicationId)
+        r = self.stub.isScriptAlive(req)
+        return r.value
+    def __getattr__(self, name):
+        """
+        调用注入应用 Hook 脚本的导出方法
+        """
+        return AppScriptRpcInterface(self.stub, self,
+                                            name)
 
 
 class ApplicationStub(BaseServiceStub):
@@ -1974,6 +2113,14 @@ class Device(object):
                                         session=None):
         self.certificate = certificate
         self.server = "{0}:{1}".format(host, port)
+        policy = dict()
+        policy["maxAttempts"] = 5
+        policy["retryableStatusCodes"] = ["UNAVAILABLE"]
+        policy["backoffMultiplier"] = 2
+        policy["initialBackoff"] = "0.5s"
+        policy["maxBackoff"] = "15s"
+        config = json.dumps(dict(methodConfig=[{"name": [{}],
+                                 "retryPolicy": policy,}]))
         if certificate is not None:
             with open(certificate, "rb") as fd:
                 key, crt, ca = self._parse_certdata(fd.read())
@@ -1983,10 +2130,14 @@ class Device(object):
             self._chan = grpc.secure_channel(self.server, creds,
                     options=(("grpc.ssl_target_name_override",
                                 self._parse_cname(crt)),
+                             ("grpc.service_config", config),
                              ("grpc.enable_http_proxy",
                                 0)))
         else:
-            self._chan = grpc.insecure_channel(self.server)
+            self._chan = grpc.insecure_channel(self.server,
+                    options=(("grpc.service_config", config),
+                            ("grpc.enable_http_proxy", 0))
+            )
         session = session or uuid.uuid4().hex
         interceptors = [ClientSessionMetadataInterceptor(session),
                         GrpcRemoteExceptionInterceptor(),

{lamda-7.76 → lamda-7.85}/lamda/rpc/application.proto

@@ -13,6 +13,16 @@ enum GrantType {
                 GRANT_IGNORE      = 2;
 }
 
+enum DataEncode {
+        DATA_ENCODE_NONE  = 0;
+        DATA_ENCODE_ZLIB  = 1;
+}
+
+enum ScriptRuntime {
+        RUNTIME_QJS        = 0;
+        RUNTIME_V8         = 1;
+}
+
 message ApplicationRequest {
         string name             = 1;
         string permission       = 2;
@@ -83,3 +93,22 @@ message ApplicationProcesses {
 message ApplicationPkgNames {
         repeated string names   = 1;
 }
+
+message HookRequest {
+        string package          = 1;
+        bytes script            = 2;
+        ScriptRuntime runtime   = 3;
+        string destination      = 4;
+        DataEncode encode       = 5;
+        uint32 standup          = 6;
+}
+
+message HookRpcRequest {
+        string package          = 1;
+        string callinfo         = 2;
+}
+
+message HookRpcResponse {
+        string package          = 1;
+        string callresult       = 2;
+}

{lamda-7.76 → lamda-7.85}/lamda/rpc/services.proto

@@ -50,6 +50,12 @@ service Application {
     rpc addToDozeModeWhiteList(ApplicationRequest) returns (Boolean) {}
     rpc removeFromDozeModeWhiteList(ApplicationRequest) returns (Boolean) {}
     rpc getIdentifierByLabel(String) returns (String) {}
+
+    rpc callScript(HookRpcRequest) returns (HookRpcResponse) {}
+    rpc isScriptAlive(String) returns (Boolean) {}
+    rpc isScriptAttached(String) returns (Boolean) {}
+    rpc attachScript(HookRequest) returns (Boolean) {}
+    rpc detachScript(String) returns (Boolean) {}
 }
 
 service Debug {

{lamda-7.76 → lamda-7.85}/lamda.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: lamda
-Version: 7.76
+Version: 7.85
 Summary: Android reverse engineering & automation framework (Client API)
 Home-page: https://github.com/firerpa/lamda
 Author: rev1si0n