pip install pycryptodome

from Crypto.Cipher import AES
from Crypto.Util.Padding import pad, unpad
from Crypto.Random import get_random_bytes
import base64

# 生成符合AES要求的密鑰（16/24/32字節(jié)）
key = get_random_bytes(16)  # AES-128

def aes_ecb_encrypt(plaintext):
    cipher = AES.new(key, AES.MODE_ECB)
    padded_data = pad(plaintext.encode(), AES.block_size)  # PKCS7填充
    ciphertext = cipher.encrypt(padded_data)
    return base64.b64encode(ciphertext).decode()

def aes_ecb_decrypt(ciphertext):
    cipher = AES.new(key, AES.MODE_ECB)
    decoded_data = base64.b64decode(ciphertext)
    decrypted = cipher.decrypt(decoded_data)
    return unpad(decrypted, AES.block_size).decode()

# 測試用例
original_text = "Hello World!" * 3  # 創(chuàng)建多塊數(shù)據(jù)
encrypted = aes_ecb_encrypt(original_text)
decrypted = aes_ecb_decrypt(encrypted)

print(f"原始數(shù)據(jù): {original_text}")
print(f"加密結果: {encrypted[:20]}...")  # 截取部分密文
print(f"解密結果: {decrypted}")

# 圖像加密偽代碼
with open("image.png", "rb") as f:
    image_data = f.read()
    
# ECB模式加密
ecb_cipher = AES.new(key, AES.MODE_ECB)
encrypted_image = ecb_cipher.encrypt(pad(image_data, 16))

# CBC模式加密（需IV）
iv = get_random_bytes(16)
cbc_cipher = AES.new(key, AES.MODE_CBC, iv)
encrypted_image_cbc = cbc_cipher.encrypt(pad(image_data, 16))