Skip to content

Commit 85eed20

Browse files
committed
Use hashlib for klap since its faster
1 parent c318303 commit 85eed20

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

kasa/klaptransport.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -467,13 +467,9 @@ def encrypt(self, msg):
467467
padder = padding.PKCS7(128).padder()
468468
padded_data = padder.update(msg) + padder.finalize()
469469
ciphertext = encryptor.update(padded_data) + encryptor.finalize()
470-
471-
digest = hashes.Hash(hashes.SHA256())
472-
digest.update(
470+
signature = hashlib.sha256(
473471
self._sig + self._seq.to_bytes(4, "big", signed=True) + ciphertext
474-
)
475-
signature = digest.finalize()
476-
472+
).digest()
477473
return (signature + ciphertext, self._seq)
478474

479475
def decrypt(self, msg):

0 commit comments

Comments
 (0)