Skip to content

Commit 4dfda47

Browse files
committed
Fix encryption errors
1 parent 5ebfb1f commit 4dfda47

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

broadlink/device.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -154,18 +154,18 @@ def __str__(self):
154154
def update_aes(self, key: bytes) -> None:
155155
"""Update AES."""
156156
self.aes = Cipher(
157-
algorithms.AES(key), modes.CBC(self.iv), backend=default_backend()
157+
algorithms.AES(bytes(key)), modes.CBC(self.iv), backend=default_backend()
158158
)
159159

160160
def encrypt(self, payload: bytes) -> bytes:
161161
"""Encrypt the payload."""
162162
encryptor = self.aes.encryptor()
163-
return encryptor.update(payload) + encryptor.finalize()
163+
return encryptor.update(bytes(payload)) + encryptor.finalize()
164164

165165
def decrypt(self, payload: bytes) -> bytes:
166166
"""Decrypt the payload."""
167167
decryptor = self.aes.decryptor()
168-
return decryptor.update(payload) + decryptor.finalize()
168+
return decryptor.update(bytes(payload)) + decryptor.finalize()
169169

170170
def auth(self) -> bool:
171171
"""Authenticate to the device."""

0 commit comments

Comments
 (0)