代码收藏家技术教程 2024-06-29

Python实现RSA算法加密字符串的基础教程

在现代数据安全中，加密算法起着至关重要的作用。特别是非对称加密算法，如RSA（Rivest-Shamir-Adleman），广泛应用于保护敏感信息的传输。本文将详细介绍如何使用RSA算法加密和解密字符串，包括生成密钥对、加密和解密消息的完整代码实现。

什么是RSA加密？

RSA加密是一种非对称加密算法，这意味着它使用一对密钥：公钥和私钥。公钥用于加密数据，私钥用于解密数据。非对称加密的主要优势在于，公钥可以公开发布，而私钥必须安全保存。这种方法确保了即使加密密钥被广泛传播，只有持有解密密钥的人才能读取加密信息。

安装必要的库

在开始之前，确保你的Python环境中已经安装了PyCryptodome库，这是一个功能强大的加密库，可以实现各种加密算法，包括RSA。

pip install pycryptodome

生成RSA密钥对

首先，我们需要生成一对RSA密钥，包括一个公钥和一个私钥。公钥用于加密消息，而私钥用于解密消息。

from Crypto.PublicKey import RSA

def generate_key_pair():
    """
    生成RSA密钥对
    """
    key = RSA.generate(2048)  # 生成2048位的RSA密钥
    private_key = key.export_key()  # 导出私钥
    public_key = key.publickey().export_key()  # 导出公钥
    return private_key, public_key

# 生成密钥对
private_key, public_key = generate_key_pair()

加密字符串

使用公钥加密字符串，我们需要导入公钥并创建一个加密对象。然后，使用该对象对字符串进行加密。

from Crypto.Cipher import PKCS1_OAEP

def encrypt_string(message, public_key):
    """
    使用RSA公钥加密字符串
    """
    key = RSA.import_key(public_key)  # 导入公钥
    cipher_rsa = PKCS1_OAEP.new(key)  # 创建用于加密的PKCS1_OAEP对象
    ciphertext = cipher_rsa.encrypt(message.encode())  # 加密消息
    return ciphertext

# 自定义待加密的字符串
custom_message = "Hello, this is a secret message!"
encrypted_message = encrypt_string(custom_message, public_key)

查看加密后的密文

加密后的密文通常是二进制数据，为了便于查看和传输，我们可以将其转换为Base64编码。

import base64

# 将密文转换为Base64编码
encrypted_message_base64 = base64.b64encode(encrypted_message).decode('utf-8')
print("Encrypted message (Base64):", encrypted_message_base64)

解密字符串

使用私钥解密密文，需要导入私钥并创建一个解密对象。然后，使用该对象对密文进行解密。

def decrypt_string(ciphertext, private_key):
    """
    使用RSA私钥解密字符串
    """
    key = RSA.import_key(private_key)  # 导入私钥
    cipher_rsa = PKCS1_OAEP.new(key)  # 创建用于解密的PKCS1_OAEP对象
    message = cipher_rsa.decrypt(ciphertext).decode()  # 解密消息
    return message

# 解密密文
decrypted_message = decrypt_string(encrypted_message, private_key)
print("Decrypted message:", decrypted_message)

完整示例代码

以下是完整的示例代码，展示了如何生成RSA密钥对、加密字符串、查看加密后的密文以及解密字符串。

from Crypto.PublicKey import RSA
from Crypto.Cipher import PKCS1_OAEP
import base64

def generate_key_pair():
    """
    生成RSA密钥对
    """
    key = RSA.generate(2048)
    private_key = key.export_key()
    public_key = key.publickey().export_key()
    return private_key, public_key

def encrypt_string(message, public_key):
    """
    使用RSA公钥加密字符串
    """
    key = RSA.import_key(public_key)
    cipher_rsa = PKCS1_OAEP.new(key)
    ciphertext = cipher_rsa.encrypt(message.encode())
    return ciphertext

def decrypt_string(ciphertext, private_key):
    """
    使用RSA私钥解密字符串
    """
    key = RSA.import_key(private_key)
    cipher_rsa = PKCS1_OAEP.new(key)
    message = cipher_rsa.decrypt(ciphertext).decode()
    return message

if __name__ == "__main__":
    # 生成RSA密钥对
    private_key, public_key = generate_key_pair()
    
    # 自定义密钥和待加密的字符串
    custom_message = "Hello, this is a secret message!"
    custom_public_key = public_key  # 使用生成的公钥进行加密
    
    # 加密字符串
    encrypted_message = encrypt_string(custom_message, custom_public_key)
    
    # 将密文转换为Base64编码，以便更容易查看
    encrypted_message_base64 = base64.b64encode(encrypted_message).decode('utf-8')
    print("Encrypted message (Base64):", encrypted_message_base64)
    
    # 解密字符串
    decrypted_message = decrypt_string(encrypted_message, private_key)
    print("Decrypted message:", decrypted_message)