Skip to content

Commit

Permalink
merge bitcoin#29390: speedup bip324_cipher.py unit test
Browse files Browse the repository at this point in the history
  • Loading branch information
kwvg committed Oct 24, 2024
1 parent 8850bea commit cfde560
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions test/functional/test_framework/crypto/bip324_cipher.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,8 @@ def pad16(x):

def aead_chacha20_poly1305_encrypt(key, nonce, aad, plaintext):
"""Encrypt a plaintext using ChaCha20Poly1305."""
if plaintext is None:
return None
ret = bytearray()
msg_len = len(plaintext)
for i in range((msg_len + 63) // 64):
Expand All @@ -42,7 +44,7 @@ def aead_chacha20_poly1305_encrypt(key, nonce, aad, plaintext):

def aead_chacha20_poly1305_decrypt(key, nonce, aad, ciphertext):
"""Decrypt a ChaCha20Poly1305 ciphertext."""
if len(ciphertext) < 16:
if ciphertext is None or len(ciphertext) < 16:
return None
msg_len = len(ciphertext) - 16
poly1305 = Poly1305(chacha20_block(key, nonce, 0)[:32])
Expand Down Expand Up @@ -191,11 +193,11 @@ def test_fschacha20poly1305aead(self):
dec_aead = FSChaCha20Poly1305(key)

for _ in range(msg_idx):
enc_aead.encrypt(b"", b"")
enc_aead.encrypt(b"", None)
ciphertext = enc_aead.encrypt(aad, plain)
self.assertEqual(hex_cipher, ciphertext.hex())

for _ in range(msg_idx):
dec_aead.decrypt(b"", bytes(16))
dec_aead.decrypt(b"", None)
plaintext = dec_aead.decrypt(aad, ciphertext)
self.assertEqual(plain, plaintext)

0 comments on commit cfde560

Please sign in to comment.