reset keys after each connection

This commit is contained in:
Patrick762 committed 2025-06-17 21:13:07 +02:00
1 parent 206d22c50b
commit 998362a5e6
2 files changed
+7

No files matched your search

@@ -175,6 +175,9 @@ class DeviceReader:
if not parsed_data: if not parsed_data:
return None return None
# Reset Encryption keys
self.encryption.reset()
return parsed_data return parsed_data
async def _async_send_command(self, command: ReadHoldingRegisters) -> bytes: async def _async_send_command(self, command: ReadHoldingRegisters) -> bytes:
@@ -283,3 +283,7 @@ class BluettiEncryption:
if self.secure_aes_key is None if self.secure_aes_key is None
else (self.secure_aes_key, None) else (self.secure_aes_key, None)
) )
def reset(self):
self.peer_pubkey = None
self.secure_aes_key = None