Python进阶数据加解密

最新推荐文章于 2024-11-14 18:36:15 发布

程序媛幂幂

最新推荐文章于 2024-11-14 18:36:15 发布

阅读量998

点赞数 25

文章标签： python 开发语言

本文链接：https://blog.csdn.net/Trb601012/article/details/135929642

版权

数据安全是永远必须考虑的重点，而数据加解密就是数据安全始终绕不过去的安全港湾！

谁能解开我的密码？

本文讲述：

利用非对称密钥RSA和对称密钥AES算法对传输数据进行加密

实现原理：

发送方本地保存有RSA公钥，发送数据前，利用AES算法对传输数据明文进行加密，利用RSA算法对AES的密钥进行加密
将密钥的密文和数据的密文打包到一块进行传输
接收方本地保存有RSA密钥，接收数据后，对传输密文进行拆分，利用RSA算法对AES密钥进行解密，然后利用AES算法还原数据明文

下面是本文核心部分，直接上代码

from Cryptodome.Cipher import AES, PKCS1_OAEP
from Cryptodome.PublicKey import RSA
from Cryptodome.Random import get_random_bytes
from io import BytesIO

import base64
import zlib

# 生成RSA公钥和密钥
def generate_key():
    new_key = RSA.generate_key(2048)
    private_key = new_key.exportKey()
    public_key = new_key.public_key().exportKey()

    with open('key.private', 'wb') as f:
        f.write(private_key)

    with open('key.public', 'wb') as f:
        f.write(public_key)

def get_rsa_cipher_by_key(keytype):
    with open(f'key.{keytype}') as f:
        key = f.read()
    rsakey = RSA.importKey(key)
    return (PKCS1_OAEP.new(rsakey), rsakey.size_in_bytes())
# 加密：利用AES加密数据，同时用RSA对AES密钥进行加密，将加密后的部分合并，base64编码后传输
def encrypt(plaintext):
    compressed_text = zlib.compress(plaintext)
    session_key = get_random_bytes(16)

    cipher_aes = AES.new(session_key, AES.MODE_EAX)
    ciphertext, tag = cipher_aes.encrypt_and_digest(compressed_text)
    cipher_rsa, _ = get_rsa_cipher_by_key('public')

    encrypted_session_key = cipher_rsa.encrypt(session_key)
    msg_payload = encrypted_session_key + cipher_aes.nonce + tag + ciphertext
    encrypted = base64.encodebytes(msg_payload)
    return encrypted

# 解密：base64解码密文，用bytesIO对象拆分出密钥和数据部分，读取本地的RSA密钥，用RSA解密出AES的密钥，再用AES解密出明文数据
def decrypt(encrypted):
    encrypted_bytes = BytesIO(base64.decodebytes(encrypted))
    cipher_rsa, keysize_in_bytes = get_rsa_cipher_by_key('private')

    encrypted_session_key = encrypted_bytes.read(keysize_in_bytes)
    nonce = encrypted_bytes.read(16)
    tag = encrypted_bytes.read(16)
    ciphertext = encrypted_bytes.read()

    session_key = cipher_rsa.decrypt(encrypted_session_key)
    cipher_aes = AES.new(session_key, AES.MODE_EAX, nonce)
    decrypted = cipher_aes.decrypt_and_verify(ciphertext, tag)

    plaintext = zlib.decompress(decrypted)
    return plaintext
# 下面测试加解密效果
if __name__ == "__main__":
    generate_key()
    plaintext = b"This is a secrete message, right?"
    print(decrypt(encrypt(plaintext)))