Python使用pycryptodome库来进行AES加密解密

飞起来fly呀

已于 2024-11-29 10:38:47 修改

阅读量918

点赞数 6

分类专栏： Python 文章标签： python

于 2024-09-19 12:06:30 首次发布

本文链接：https://blog.csdn.net/qq_42771102/article/details/142356874

版权

Python 专栏收录该内容

22 篇文章

订阅专栏

在现代通信和数据存储中，加密技术是保障数据安全的核心手段。AES（Advanced Encryption Standard）是一种对称加密算法，广泛应用于各种信息安全领域。Python提供了丰富的加密库，其中PyCryptodome是一个功能强大且常用的库，它支持多种加密算法和模式。以下指南将详细介绍如何在Python中使用PyCryptodome库进行AES加密和解密。

一、安装PyCryptodome库

要使用PyCryptodome库，首先需要确保它已安装在你的Python环境中。可以通过pip进行安装：

pip install pycryptodome

二、使用PyCryptodome进行AES加密与解密

使用AES加密时，一个安全的密钥和随机的nonce（即"number used once"）是确保加密过程安全的关键。下面将介绍如何使用AES-GCM模式进行操作，该模式提供了加密和数据完整性验证功能。

1. 创建密钥派生函数

为了安全地使用密码作为密钥，首先需要通过哈希函数将其转换为固定长度的字节串。这里使用SHA-256将密码派生为32字节的密钥：

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

# 密钥派生函数
def get_key(password):
    hasher = hashlib.sha256()
    hasher.update(password)
    return hasher.digest()

2. 加密函数

在AES-GCM模式中，加密需要一个随机产生的nonce和密钥。使用PyCryptodome库进行加密如下：

def encrypt_AES_GCM(msg, password):
    # 生成随机的nonce，长度通常为12字节
    nonce = get_random_bytes(12)
    # 生成密钥
    key = get_key(password)
    # 创建AES-GCM加密器
    cipher = AES.new(key, AES.MODE_GCM, nonce=nonce)
    # 加密并生成校验标签
    ciphertext, tag = cipher.encrypt_and_digest(msg.encode('utf-8'))
    # 返回密文、nonce和校验标签
    return (ciphertext, nonce, tag)

3. 解密函数

解密过程使用与加密相同的密钥和nonce，并通过校验标签验证数据的完整性：

def decrypt_AES_GCM(ciphertext, nonce, tag, password):
    # 生成密钥
    key = get_key(password)
    # 创建AES-GCM解密器
    cipher = AES.new(key, AES.MODE_GCM, nonce=nonce)
    # 解密并验证
    plaintext = cipher.decrypt_and_verify(ciphertext, tag)
    # 返回解密后的消息
    return plaintext.decode('utf-8')

4. 测试加密和解密过程

# 示例密码
password = b'secret_password_123'

# 示例消息
msg = 'Hello, world!'

# 加密消息
ciphertext, nonce, tag = encrypt_AES_GCM(msg, password)
print(f"密文: {ciphertext.hex()}")

# 解密消息
plaintext = decrypt_AES_GCM(ciphertext, nonce, tag, password)
print(f"解密后的消息: {plaintext}")