Mirrored_Repos/python-kasa
1
0
Fork 0
mirror of https://github.com/python-kasa/python-kasa.git synced 2024-12-23 03:33:35 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Use hashlib for klap since its faster (#711)

Browse Source
This commit is contained in:
J. Nick Koston 2024-01-25 23:33:18 -10:00 committed by GitHub
parent 0d0f56414c
commit fcd4883645
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
kasa/klaptransport.py
View File

@ -467,13 +467,9 @@ class KlapEncryptionSession:
padder = padding.PKCS7(128).padder() padder = padding.PKCS7(128).padder()
padded_data = padder.update(msg) + padder.finalize() padded_data = padder.update(msg) + padder.finalize()
ciphertext = encryptor.update(padded_data) + encryptor.finalize() ciphertext = encryptor.update(padded_data) + encryptor.finalize()
signature = hashlib.sha256(
digest = hashes.Hash(hashes.SHA256())
digest.update(
self._sig + self._seq.to_bytes(4, "big", signed=True) + ciphertext self._sig + self._seq.to_bytes(4, "big", signed=True) + ciphertext
) ).digest()
signature = digest.finalize()
return (signature + ciphertext, self._seq) return (signature + ciphertext, self._seq)
def decrypt(self, msg): def decrypt(self, msg):