From 48763aa77e6375e065d7ee9dabbedd4b1be500d8 Mon Sep 17 00:00:00 2001
From: "J. Nick Koston" <nick@koston.org>
Date: Thu, 25 Jan 2024 21:56:43 -1000
Subject: [PATCH] Use hashlib for klap since its faster

---
 kasa/klaptransport.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/kasa/klaptransport.py b/kasa/klaptransport.py
index cd0e3de6..66052f59 100644
--- a/kasa/klaptransport.py
+++ b/kasa/klaptransport.py
@@ -467,13 +467,9 @@ class KlapEncryptionSession:
         padder = padding.PKCS7(128).padder()
         padded_data = padder.update(msg) + padder.finalize()
         ciphertext = encryptor.update(padded_data) + encryptor.finalize()
-
-        digest = hashes.Hash(hashes.SHA256())
-        digest.update(
+        signature = hashlib.sha256(
             self._sig + self._seq.to_bytes(4, "big", signed=True) + ciphertext
-        )
-        signature = digest.finalize()
-
+        ).digest()
         return (signature + ciphertext, self._seq)
 
     def decrypt(self, msg):