still need to pad for hap

postlund · Jan 26, 2024 · 860a767 · 860a767
1 parent 4e7e375
commit 860a767
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyatv/support/chacha20.py b/pyatv/support/chacha20.py
@@ -48,7 +48,7 @@ def encrypt(
         if nonce is None:
             nonce = self.out_nonce
             self._out_counter += 1
-        if len(nonce) < NONCE_LENGTH:
+        elif len(nonce) < NONCE_LENGTH:
             nonce = b"\x00" * (NONCE_LENGTH - len(nonce)) + nonce
         return self._enc_out.encrypt(nonce, data, aad)
 
@@ -59,6 +59,6 @@ def decrypt(
         if nonce is None:
             nonce = self.in_nonce
             self._in_counter += 1
-        if len(nonce) < NONCE_LENGTH:
+        elif len(nonce) < NONCE_LENGTH:
             nonce = b"\x00" * (NONCE_LENGTH - len(nonce)) + nonce
         return self._enc_in.decrypt(nonce, data, aad)