python 实现AES 算法

以下是一个用Python实现AES算法的示例代码：

from Crypto.Cipher import AES
from Crypto.Random import get_random_bytes

def pad(data):
    length = 16 - (len(data) % 16)
    return data + bytes([length])*length

def unpad(data):
    return data[:-data[-1]]

def encrypt(key, data):
    cipher = AES.new(key, AES.MODE_CBC)
    ciphertext = cipher.encrypt(pad(data))
    return cipher.iv + ciphertext

def decrypt(key, data):
    iv = data[:16]
    data = data[16:]
    cipher = AES.new(key, AES.MODE_CBC, iv)
    plaintext = cipher.decrypt(data)
    return unpad(plaintext)

key = get_random_bytes(16)
data = b'Hello, AES!'

encrypted_data = encrypt(key, data)
print('Encrypted data:', encrypted_data)

decrypted_data = decrypt(key, encrypted_data)
print('Decrypted data:', decrypted_data)

在这个示例中，我们使用Crypto.Cipher模块中的AES实现了加密和解密的函数。pad函数用于填充数据，使其长度为16的倍数。unpad函数用于去除填充数据。encrypt函数使用AES算法进行加密并返回密文。decrypt函数使用AES算法进行解密并返回明文。

在示例中，我们生成了一个16字节的随机密钥，并定义了一个待加密的数据。使用encrypt函数对数据进行加密，并打印出密文。然后使用decrypt函数对密文进行解密，并打印出明文。

请注意，AES算法是对称加密算法，所以加密和解密使用相同的密钥。在实际应用中，应该妥善保管密钥，以确保数据的安全性。