kamp/python-broadlink
1
0
Fork 0
mirror of https://github.com/mjg59/python-broadlink.git synced 2024-11-21 22:51:41 +01:00
Code

Fix encryption errors (#505)

Browse Source
This commit is contained in:
Felipe Martins Diel 2021-01-11 17:51:31 -03:00
parent 448e3026df
commit b33bbdbbcf
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
broadlink/device.py
View File

@ -154,18 +154,18 @@ class device:
def update_aes(self, key: bytes) -> None: def update_aes(self, key: bytes) -> None:
"""Update AES.""" """Update AES."""
self.aes = Cipher( self.aes = Cipher(
algorithms.AES(key), modes.CBC(self.iv), backend=default_backend() algorithms.AES(bytes(key)), modes.CBC(self.iv), backend=default_backend()
) )
def encrypt(self, payload: bytes) -> bytes: def encrypt(self, payload: bytes) -> bytes:
"""Encrypt the payload.""" """Encrypt the payload."""
encryptor = self.aes.encryptor() encryptor = self.aes.encryptor()
return encryptor.update(payload) + encryptor.finalize() return encryptor.update(bytes(payload)) + encryptor.finalize()
def decrypt(self, payload: bytes) -> bytes: def decrypt(self, payload: bytes) -> bytes:
"""Decrypt the payload.""" """Decrypt the payload."""
decryptor = self.aes.decryptor() decryptor = self.aes.decryptor()
return decryptor.update(payload) + decryptor.finalize() return decryptor.update(bytes(payload)) + decryptor.finalize()
def auth(self) -> bool: def auth(self) -> bool:
"""Authenticate to the device.""" """Authenticate to the device."""