zach
/
eufy-robovac
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix unpadding of datagram

This commit is contained in:
Luke Bonaccorsi 2023-08-17 12:25:52 +01:00
parent 255f164e6d
commit 4f111af975
1 changed files with 1 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
custom_components/robovac/tuyalocaldiscovery.py
View File

@ -38,20 +38,10 @@ class TuyaLocalDiscovery(asyncio.DatagramProtocol):
cipher = Cipher(algorithms.AES(UDP_KEY), modes.ECB(), default_backend()) cipher = Cipher(algorithms.AES(UDP_KEY), modes.ECB(), default_backend())
decryptor = cipher.decryptor() decryptor = cipher.decryptor()
padded_data = decryptor.update(data) + decryptor.finalize() padded_data = decryptor.update(data) + decryptor.finalize()
data = padded_data[: -ord(data[len(data) - 1 :])] data = padded_data[: -ord(padded_data[len(padded_data) - 1 :])]
except Exception: except Exception:
data = data.decode() data = data.decode()
decoded = json.loads(data) decoded = json.loads(data)
self.discovered_callback(decoded) self.discovered_callback(decoded)
async def discover():
"""Discover and return devices on local network."""
discovery = TuyaDiscovery()
try:
await discovery.start()
await asyncio.sleep(DEFAULT_TIMEOUT)
finally:
discovery.close()
return discovery.devices