X-Git-Url: https://glassweightruler.freedombox.rocks/gitweb/waydroid.git/blobdiff_plain/677710ab36bba687b04268704bd92b37638854d0..d1ba241d92dca786644e52d5de08a77b4e721bf3:/tools/interfaces/IHardware.py

diff --git a/tools/interfaces/IHardware.py b/tools/interfaces/IHardware.py
index f7984db..0338c3f 100644
--- a/tools/interfaces/IHardware.py
+++ b/tools/interfaces/IHardware.py
@@ -12,10 +12,14 @@ TRANSACTION_enableBluetooth = 2
 TRANSACTION_suspend = 3
 TRANSACTION_reboot = 4
 TRANSACTION_upgrade = 5
+TRANSACTION_upgrade2 = 6
 
 def add_service(args, enableNFC, enableBluetooth, suspend, reboot, upgrade):
     helpers.drivers.loadBinderNodes(args)
-    serviceManager = gbinder.ServiceManager("/dev/" + args.BINDER_DRIVER)
+    try:
+        serviceManager = gbinder.ServiceManager("/dev/" + args.BINDER_DRIVER, args.SERVICE_MANAGER_PROTOCOL, args.BINDER_PROTOCOL)
+    except TypeError:
+        serviceManager = gbinder.ServiceManager("/dev/" + args.BINDER_DRIVER)
 
     def response_handler(req, code, flags):
         logging.debug(
@@ -45,6 +49,13 @@ def add_service(args, enableNFC, enableBluetooth, suspend, reboot, upgrade):
             status, arg4 = reader.read_int32()
             upgrade(arg1, arg2, arg3, arg4)
             local_response.append_int32(0)
+        if code == TRANSACTION_upgrade2:
+            arg1 = reader.read_string16()
+            status, arg2 = reader.read_int64()
+            arg3 = reader.read_string16()
+            status, arg4 = reader.read_int64()
+            upgrade(arg1, arg2, arg3, arg4)
+            local_response.append_int32(0)
 
         return local_response, 0