hardware_manager: Prevent race-condition in upgrade check

[waydroid.git] / tools / interfaces / IHardware.py

diff --git a/tools/interfaces/IHardware.py b/tools/interfaces/IHardware.py
index 9e1f2fe58d274e48d41dd93671c13d7265741bf5..69539f189a97c12ad38addaed962e844aa160c2f 100644 (file)
--- a/tools/interfaces/IHardware.py
+++ b/tools/interfaces/IHardware.py
@@ -12,6 +12,7 @@ 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)
@@ -30,24 +31,33 @@ def add_service(args, enableNFC, enableBluetooth, suspend, reboot, upgrade):
             ret = enableNFC(arg1 != 0)
             local_response.append_int32(0)
             local_response.append_int32(ret)
-        if code == TRANSACTION_enableBluetooth:
+        elif code == TRANSACTION_enableBluetooth:
             status, arg1 = reader.read_int32()
             ret = enableBluetooth(arg1 != 0)
             local_response.append_int32(0)
             local_response.append_int32(ret)
-        if code == TRANSACTION_suspend:
+        elif code == TRANSACTION_suspend:
             suspend()
             local_response.append_int32(0)
-        if code == TRANSACTION_reboot:
+        elif code == TRANSACTION_reboot:
             reboot()
             local_response.append_int32(0)
-        if code == TRANSACTION_upgrade:
+        elif code == TRANSACTION_upgrade:
             arg1 = reader.read_string16()
             status, arg2 = reader.read_int32()
             arg3 = reader.read_string16()
             status, arg4 = reader.read_int32()
             upgrade(arg1, arg2, arg3, arg4)
             local_response.append_int32(0)
+        elif 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)
+        else:
+            return local_response, -99999 # Some error unknown to binder to force a RemoteException
 
         return local_response, 0