py9b/transport/xiaomi: add recover_keys method

2019-10-21 11:39:21 +02:00 · 2019-10-21 11:39:21 +02:00 · 8a6b8731ad
parent af23f5fe6f
commit 8a6b8731ad
1 changed files with 7 additions and 0 deletions
--- a/py9b/transport/xiaomi.py
+++ b/py9b/transport/xiaomi.py
@ -122,5 +122,12 @@ class XiaomiTransport(BT):
        k = self.keys
        return bytearray([b ^ (k[i] if i < len(k) else 0) for i, b in enumerate(data)])

+    def recover_keys(self):
+        req = BasePacket(src=BT.HOST, dst=BT.BMS, cmd=0x01,
+                         arg=0x50, data=bytearray([0x20]))
+        self.send(req)
+        resp = self.recv()
+        self.keys += resp.data[9:]
+

 __all__ = ["XiaomiTransport"]