yumoqing/apppublic
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2025-08-14 15:24:37 +08:00
parent 3ac0d1a668
commit 20eb1b19d3
1 changed files with 10 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
appPublic/aes.py
View File

@ -13,14 +13,23 @@ def unpad(data: bytes) -> bytes:
unpadder = padding.PKCS7(128).unpadder() unpadder = padding.PKCS7(128).unpadder()
return unpadder.update(data) + unpadder.finalize() 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: def aes_encrypt_ecb(key: bytes, plaintext: str) -> bytes:
key = len32(key)
cipher = Cipher(algorithms.AES(key), modes.ECB(), backend=default_backend()) cipher = Cipher(algorithms.AES(key), modes.ECB(), backend=default_backend())
encryptor = cipher.encryptor() encryptor = cipher.encryptor()
padded_data = pad(plaintext.encode()) padded_data = pad(plaintext.encode())
return encryptor.update(padded_data) + encryptor.finalize() return encryptor.update(padded_data) + encryptor.finalize()
def aes_decrypt_ecb(key: bytes, ciphertext: bytes) -> str: def aes_decrypt_ecb(key: bytes, ciphertext: bytes) -> str:
key = len32(key)
cipher = Cipher(algorithms.AES(key), modes.ECB(), backend=default_backend()) cipher = Cipher(algorithms.AES(key), modes.ECB(), backend=default_backend())
decryptor = cipher.decryptor() decryptor = cipher.decryptor()
padded_plain = decryptor.update(ciphertext) + decryptor.finalize() padded_plain = decryptor.update(ciphertext) + decryptor.finalize()