from Crypto.PublicKey import RSA
from Crypto.Cipher import PKCS1_OAEP
from Crypto.Random import get_random_bytes
 
# 生成RSA密钥对
def generate_rsa_keypair(bits=2048):
    key = RSA.generate(bits)
    private_key = key.export_key()
    public_key = key.publickey().export_key()
    return private_key, public_key
 
# 使用公钥加密消息
def encrypt_message(public_key, message):
    recipient_key = RSA.import_key(public_key)
    cipher_rsa = PKCS1_OAEP.new(recipient_key)
    encrypted_msg = cipher_rsa.encrypt(message.encode('utf-8'))
    return encrypted_msg
 
# 使用私钥解密消息
def decrypt_message(private_key, encrypted_msg):
    private_key = RSA.import_key(private_key)
    cipher_rsa = PKCS1_OAEP.new(private_key)
    decrypted_msg = cipher_rsa.decrypt(encrypted_msg)
    return decrypted_msg.decode('utf-8')
 
# 主程序
if __name__ == "__main__":
    # 生成RSA密钥对
    private_key, public_key = generate_rsa_keypair()
    print("私钥（不建议打印在终端中）:")
    print(private_key.decode('utf-8'))  # 注意：这里仅用于演示，实际使用中应安全存储私钥
    print("公钥:")
    print(public_key.decode('utf-8'))
 
    # 要加密的消息
    message = "Hello, RSA Encryption!"
    print(f"原始消息: {message}")
 
    # 加密消息
    encrypted_message = encrypt_message(public_key, message)
    print(f"加密后的消息（十六进制）: {encrypted_message.hex()}")
 
    # 解密消息
    decrypted_message = decrypt_message(private_key, encrypted_message)
    print(f"解密后的消息: {decrypted_message}")
 
    # 注意：在实际应用中，加密和解密过程通常在不同的设备或程序中执行，
    # 并且私钥应该被安全地存储和访问。