Merge pull request #150 from msmarks/fix_request_count_overflow

fix self._request_count overflow
georgezhao2010 · Dec 22, 2022 · 06fc4b5 · 06fc4b5
2 parents 7680f9a + 22061b3
commit 06fc4b5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/custom_components/midea_ac_lan/midea/core/security.py b/custom_components/midea_ac_lan/midea/core/security.py
@@ -88,6 +88,8 @@ def encode_8370(self, data, msgtype):
         header += bytearray([0x20, padding << 4 | msgtype])
         data = self._request_count.to_bytes(2, "big") + data
         self._request_count += 1
+        if self._request_count > 0xFFFF:
+            self._request_count = 0
         if msgtype in (MSGTYPE_ENCRYPTED_RESPONSE, MSGTYPE_ENCRYPTED_REQUEST):
             sign = sha256(header + data).digest()
             data = self.aes_cbc_encrypt(raw=data, key=self._tcp_key) + sign