From 20eb1b19d30bd504ff63ca8c2baa9201e0a755d6 Mon Sep 17 00:00:00 2001 From: yumoqing Date: Thu, 14 Aug 2025 15:24:37 +0800 Subject: [PATCH] bugfix --- appPublic/aes.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/appPublic/aes.py b/appPublic/aes.py index 276761d..9f85207 100644 --- a/appPublic/aes.py +++ b/appPublic/aes.py @@ -13,14 +13,23 @@ def unpad(data: bytes) -> bytes: unpadder = padding.PKCS7(128).unpadder() return unpadder.update(data) + unpadder.finalize() +def len32(key): + cnt = len(key) + if cnt < 32: + key += b'*' * (32 - cnt) + elif cnt > 32: + key = key[:32] + return key + def aes_encrypt_ecb(key: bytes, plaintext: str) -> bytes: - + key = len32(key) cipher = Cipher(algorithms.AES(key), modes.ECB(), backend=default_backend()) encryptor = cipher.encryptor() padded_data = pad(plaintext.encode()) return encryptor.update(padded_data) + encryptor.finalize() def aes_decrypt_ecb(key: bytes, ciphertext: bytes) -> str: + key = len32(key) cipher = Cipher(algorithms.AES(key), modes.ECB(), backend=default_backend()) decryptor = cipher.decryptor() padded_plain = decryptor.update(ciphertext) + decryptor.finalize()