from cryptography.hazmat.backends import default_backend
from cryptography.hazmat.primitives.asymmetric import rsa
from cryptography.hazmat.primitives import serialization
from cryptography.hazmat.primitives import hashes
from cryptography.hazmat.primitives.asymmetric import padding


def generate_rsa_keys():
    # private_key = rsa.generate_private_key(
    #     public_exponent=65537,
    #     key_size=2048,
    #     backend=default_backend()
    # )
    # public_key = private_key.public_key()
    #
    # # 序列化公钥和私钥
    # pem = public_key.public_bytes(
    #     encoding=serialization.Encoding.PEM,
    #     format=serialization.PublicFormat.SubjectPublicKeyInfo
    # )
    # pem_private = private_key.private_bytes(
    #     encoding=serialization.Encoding.PEM,
    #     format=serialization.PrivateFormat.TraditionalOpenSSL,
    #     encryption_algorithm=serialization.NoEncryption()
    # )

    pem = b'-----BEGIN PUBLIC KEY-----\nMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA02zpy+F92JitiI0oaeSU\njV+P7WrYwH3lbylD5Q20DKe/7wOsL/lawq2/Z9RWjgg+K51SC92JkXunybGv/aqf\nckx29P9BG9A+OhW9pPFdd25voCQgE8htL/I5IdEIGyS8iTBzmnUICcDCT59BPC2F\nlT0UlErbaLFKvr9RqTmPp8EcdWBh0gwQcBqhiojVwRndzbEKXpsCqIWgcEGxQaYz\nQEZcOiU7B94os822347K1aWJBHQeAhAtQ4a1BCBkEaFasAny3h7hBit1p3N3BMUx\nhdbiESBFoI4QVM+hOOnVQ7Na7KM+N/Rk2HsFlMzZJdYiaDf45WvLvlWgJknE3zXb\nsQIDAQAB\n-----END PUBLIC KEY-----\n'
    # print(pem)
    pem_private = b'-----BEGIN RSA PRIVATE KEY-----\nMIIEpQIBAAKCAQEA02zpy+F92JitiI0oaeSUjV+P7WrYwH3lbylD5Q20DKe/7wOs\nL/lawq2/Z9RWjgg+K51SC92JkXunybGv/aqfckx29P9BG9A+OhW9pPFdd25voCQg\nE8htL/I5IdEIGyS8iTBzmnUICcDCT59BPC2FlT0UlErbaLFKvr9RqTmPp8EcdWBh\n0gwQcBqhiojVwRndzbEKXpsCqIWgcEGxQaYzQEZcOiU7B94os822347K1aWJBHQe\nAhAtQ4a1BCBkEaFasAny3h7hBit1p3N3BMUxhdbiESBFoI4QVM+hOOnVQ7Na7KM+\nN/Rk2HsFlMzZJdYiaDf45WvLvlWgJknE3zXbsQIDAQABAoIBAALGCkAkNmYtxJMG\nxQm0l7VOFGGJNgw6apTdffhGiH7lAXnVesALXaMxq9HlIXiOB6DDWNTyhFA+Bsq4\n08X4Y8yYT7QxmA1RG0vgRqNgDjmXn4vChZA04CfRnenBaAGRJw8GrXGQiPk1x+bL\nAExVjYvSaA909vFl6ATeFihp6qZI9meSRfB2lrURY2jY9Qu2nECfxxFpVa7bFlBW\n8KVD+RAwN7pZKqTNak7UMmw2/U1TbBol78PhIstwT28XMgDjTyR+rA2x8BXbvg62\nb28kT2Gd/bzw/ZVLuBL/zOsl3BbPgqhDWLOsLYk3FXjAbumPNo2/1sgzu+PgBvzg\n2DSufQUCgYEA69S5asMSNjlB6xTn6DYGe0s8ej1McgLB2V1URnmPmijgNwFZyYN2\nJ0R3ggA6axHLTGIVwWxngWCku9xHUP19jGntQuG7e0wYK9uvW+MTX6ePthqd5tTy\n/4+4/JnaUtKKf1w8TfWlgcY/pYFjRYGzSoQwZwM3LH8sTqNvHr/4rsUCgYEA5YHa\n/uD2rNVgM1/beD/FKIZ3VaNiCkzzl5pVRALva0M8L5+z2SWzYXbtnhUtEIqKiGS4\ndczZOqPzDtjnHZ/YdpEz+Q0j0PtDB466KWaz0qJkPx05fGlnSzcE82T9IG2OjXgi\nKopA1INFXnyzABpPUyzKJbWvftUu3igpTqWex/0CgYEAu6mIbquSN3HywZXeQQuU\nttIU6xUSWvtUtJqu8HHVUxBJ85twjom9INw72UhDNGZxh3PZ6hRQfUShchiE6678\nFBtKwVkNrrzL00K61XzqMskM9BVUJ5a3U6RTyTbI4/VEXrrQBh1+ubk/dn1yuKHQ\nQXDVb3PAmPkOEfm9+BM9Wh0CgYEA3yI6LZXAnlaxuiEXOrkdUYIyYYOAnS0lo558\nCayk428yA8dMTvF1jmvXJ9vEvZFEkMkEAhvQk0+0zxmY0deH1/biReA9kz/GcoGU\ngw4EBOS3fBlIu6mqsJjkalMkOrPpZaeH6D4UO8kI1417Zr0tpFsmijUzU6emcn0v\nc/RYBG0CgYEA4/H4LILR3cmehLm3O10ukvXjCc1ED34bLQf1WGTPEjM2bL2faXB2\nFJSXMsssrMFoAHrtrMa8TPGov3SshIqHrj9OCJ/S1GIwseUXOJl/hSECmnuyCe7K\n7FKaBPy21+vPeu2QWSZNiNR3vEuLaXfEa4MDmfaBTwAs7AhNuD4MLIc=\n-----END RSA PRIVATE KEY-----\n'
    # print(pem_private)
    print("*"*90)
    return pem, pem_private


def rsa_encrypt(plaintext, public_key_pem):
    public_key = serialization.load_pem_public_key(
        public_key_pem,
        backend=default_backend()
    )
    encrypted = public_key.encrypt(
        plaintext.encode(),
        padding.OAEP(
            mgf=padding.MGF1(algorithm=hashes.SHA256()),
            algorithm=hashes.SHA256(),
            label=None
        )
    )
    return encrypted


def rsa_decrypt(ciphertext, private_key_pem):
    private_key = serialization.load_pem_private_key(
        private_key_pem,
        password=None,
        backend=default_backend()
    )
    decrypted = private_key.decrypt(
        ciphertext,
        padding.OAEP(
            mgf=padding.MGF1(algorithm=hashes.SHA256()),
            algorithm=hashes.SHA256(),
            label=None
        )
    )
    return decrypted.decode()


# 使用示例
public_key_pem, private_key_pem = generate_rsa_keys()
plaintext = "Hello, RSA!"
ciphertext = rsa_encrypt(plaintext, public_key_pem)
print(ciphertext)
jiemi = rsa_decrypt(ciphertext, private_key_pem)
print(jiemi)