Python3使用PyCryptodome实现AES256支持自定义密钥长度的方法问询

AES256本身要求密钥长度必须为32字节，你看到的在线工具支持任意长度密钥的核心逻辑是：先通过标准密码学密钥派生算法（KDF），将用户输入的任意长度密码转换为符合32字节要求的AES密钥，而非直接使用用户原始输入作为加密密钥。

推荐使用PyCryptodome库内置的PBKDF2HMAC算法实现密钥派生，这是行业通用的安全实现方案，完整实现代码如下：

依赖安装 #

pip install pycryptodome

加密实现 #

from Crypto.Cipher import AES
from Crypto.Protocol.KDF import PBKDF2HMAC
from Crypto.Hash import SHA256
from Crypto.Random import get_random_bytes

def aes256_encrypt(plaintext: str, user_password: str) -> bytes:
    # 生成16字节随机盐，每次加密唯一，无需保密
    salt = get_random_bytes(16)
    # 密钥派生：将用户任意长度密码转换为32字节AES256密钥
    # 迭代次数建议不低于100000，可根据性能需求调整
    key = PBKDF2HMAC(
        password=user_password.encode("utf-8"),
        salt=salt,
        dkLen=32,
        count=100000,
        hmac_hash_module=SHA256
    )
    # 执行AES-EAX加密
    cipher = AES.new(key, AES.MODE_EAX)
    nonce = cipher.nonce
    ciphertext, tag = cipher.encrypt_and_digest(plaintext.encode("utf-8"))
    # 拼接加密结果：盐(16字节) + nonce(16字节) + tag(16字节) + 密文
    return salt + nonce + tag + ciphertext

解密实现 #

def aes256_decrypt(encrypted_data: bytes, user_password: str) -> str:
    # 按加密时的拼接顺序拆分各字段
    salt = encrypted_data[:16]
    nonce = encrypted_data[16:32]
    tag = encrypted_data[32:48]
    ciphertext = encrypted_data[48:]
    # 用相同参数派生密钥
    key = PBKDF2HMAC(
        password=user_password.encode("utf-8"),
        salt=salt,
        dkLen=32,
        count=100000,
        hmac_hash_module=SHA256
    )
    # 执行解密
    cipher = AES.new(key, AES.MODE_EAX, nonce=nonce)
    plaintext = cipher.decrypt_and_verify(ciphertext, tag)
    return plaintext.decode("utf-8")

使用示例 #

# 加密
user_pwd = "用户输入的任意长度密码123@abc"
data = "需要加密的敏感内容"
encrypted = aes256_encrypt(data, user_pwd)
# 解密
decrypted = aes256_decrypt(encrypted, user_pwd)
print(decrypted) # 输出：需要加密的敏感内容

注意事项 #

• 禁止直接对用户输入密码做截断、补0等操作来凑32字节，会大幅降低加密安全性，必须使用标准KDF算法派生密钥
• PBKDF2的迭代次数越高，抗暴力破解能力越强，可根据业务的性能和安全要求灵活调整
• 盐、nonce、tag无需保密，和密文一起存储即可，无需单独加密保存

内容的提问来源于stack exchange，提问作者Nsous