Python3: AES 对称加密

雨点打透心脏的1/2处 2021-09-21 23:12 413阅读 0赞

本文链接: [https://blog.csdn.net/xietansheng/article/details/115559278][https_blog.csdn.net_xietansheng_article_details_115559278]

[Python3 学习笔记（目录）][Python3]

`pycryptodome`模块文档: [https://pycryptodome.readthedocs.io/en/latest/][https_pycryptodome.readthedocs.io_en_latest]

Python 的 AES 加密/解密使用第三方模块`pycryptodome`, 安装模块:

pip3 install pycryptodome

`pycryptodome`模块的 AES 加密的密钥和明文的字节长度均必须为 16 的倍数，如果不足 16 的倍数，可以在末尾添加值为 0 的字节（即`b"\x00"`）。如果原明文加密前末尾添加了`b"\x00"`，则解密后得到的明文末尾也会有，需要手动去掉。

AES 加密/解密代码示例:

from Crypto.Cipher import AES
    import base64
    
    
    def add_to_16x(bs: bytes):
        """ 在字节序列末尾加字节 0 值至长度为 16 的倍数 """
        new_bs = bytearray(bs)
        while len(new_bs) % 16 != 0:
            new_bs.append(0)
        return new_bs
    
    
    def remove_end_0(bs: bytes):
        """ 移除字节序列末尾的 0 值（b"\x00" 即值为 0 的 bytes） """
        return bs.rstrip(b"\x00")
    
    
    # 密钥 Key
    key = "123456".encode()
    
    # 待加密的明文
    plaintext = "hello 你好".encode()
    
    # 创建密码器, 指定 密钥 和 模式
    aes = AES.new(add_to_16x(key), AES.MODE_ECB)
    
    
    """ 加密 """
    
    # 加密明文, 返回加密结果（bytes类型）
    ciphertext = aes.encrypt(add_to_16x(plaintext))
    
    # 加密结果为 bytes 类型
    print(ciphertext)                       # b'4\xe4>\x85\x08\xcd<U\x8d\xf7d\xa9x\xca\x9e}'
    
    # 把 bytes 类型编码为 base64
    print(base64.b64encode(ciphertext))     # b'NOQ+hQjNPFWN92SpeMqefQ=='
    
    
    """ 解密 """
    
    # 解密密文, 返回解密结果（bytes类型）
    plaintext = aes.decrypt(ciphertext)
    
    # 解密后得到的 bytes 末尾可能有无效的 0 值, 移除掉
    plaintext = remove_end_0(plaintext)
    
    # bytes 类型的解密结果
    print(plaintext)                        # b'hello \xe4\xbd\xa0\xe5\xa5\xbd'
    
    # 把 bytes 解码为原明文字符串
    print(plaintext.decode())               # "hello 你好"

[https_blog.csdn.net_xietansheng_article_details_115559278]: https://blog.csdn.net/xietansheng/article/details/115559278
[Python3]: https://blog.csdn.net/xietansheng/article/details/109266600
[https_pycryptodome.readthedocs.io_en_latest]: https://pycryptodome.readthedocs.io/en/latest/