fix: error catch and typing (#133)

humbertogontijo · Oct 29, 2023 · 171c302 · 171c302
1 parent 316fc0d
commit 171c302
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/roborock/api.py b/roborock/api.py
@@ -293,7 +293,10 @@ def on_message_received(self, messages: list[RoborockMessage]) -> None:
                     payload = data.payload[0:24]
                     [endpoint, _, request_id, _] = struct.unpack("<8s8sH6s", payload)
                     if endpoint.decode().startswith(self._endpoint):
-                        decrypted = Utils.decrypt_cbc(data.payload[24:], self._nonce)
+                        try:
+                            decrypted = Utils.decrypt_cbc(data.payload[24:], self._nonce)
+                        except ValueError as err:
+                            raise RoborockException("Failed to decode %s for %s", data.payload, data.protocol) from err
                         decompressed = Utils.decompress(decrypted)
                         queue = self._waiting_queue.get(request_id)
                         if queue:

diff --git a/roborock/py.typed b/roborock/py.typed