File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff 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."""
You can’t perform that action at this time.
0 commit comments