Python实现RSA算法的公钥加密私钥解密和私钥签名公钥验签

#!/usr/bin/env python3
# coding=utf-8

"""
Created by Johnny 2020/7/8 14:10
"""
import rsa
import base64


class RsaUtil:
    def __init__(self, privateKey, publicKey, data):
        if privateKey and publicKey:
            self.privateKey = rsa.PrivateKey.load_pkcs1(privateKey)
            self.publicKey = rsa.PublicKey.load_pkcs1(publicKey)
        else:
            (self.publicKey, self.privateKey) = rsa.newkeys(1024)
        self.data = data.encode("utf8")
        print("解密前长度： {} \n解密前内容： {}".format(len(data), data))

    def encrypt(self):
        """
        公钥加密，public.pem 要求 PKCS1 格式
        :return: encryptData 加密后内容
        """
        length = len(self.data)
        length_max = 117
        encryptDataText = []
        for i in range(0, length, length_max):
            encryptDataText.append(
                rsa.encrypt(self.data[i:i + length_max], self.publicKey))  # str类型，需要进行encode()转化为bytes
        encryptData = b''.join(encryptDataText)
        encryptedData = base64.b64encode(encryptData)
        print("加密后内容：", encryptedData.decode())
        return encryptData

    def decrypt(self, encryptData):
        """
        私钥解密，private.pem 要求 PKCS1 格式
        :param encryptData: RSA公钥加密后内容
        :return: decryptData 解密后内容
        """
        length = len(encryptData)
        length_max = 128
        decryptDataText = []
        for i in range(0, length, length_max):
            decryptDataText.append(
                rsa.decrypt(encryptData[i:i + length_max], self.privateKey).decode())  # bytes类型，需要进行decode()转化为str
        decryptData = ''.join(decryptDataText)
        print("解密后长度： {} \n解密后内容： {}".format(len(decryptData), decryptData))

    def sign(self):
        # hash_method: the hash method used on the message. Use 'MD5', 'SHA-1','SHA-224', SHA-256', 'SHA-384' or 'SHA-512'.
        return rsa.sign(self.data, self.privateKey, 'MD5')

    def verify(self, signature):
        return bool(rsa.verify(self.data, signature, self.publicKey))


if __name__ == "__main__":
    privateKey = '''-----BEGIN RSA PRIVATE KEY-----
MIICXgIBAAKBgQDZAzJb0n62WqMKQUFBdIBUc8Ld8NKuK1nrd9xXVrqt/UwXQlYn
MuGc8M1+c4rhRMZHcG1a4RBwUZBjQSWFSf9RdYAMHdyncmiHeTcAExZJC8jN8DrR
arbcJqPPPFPSsCMoRh9mxZESLJPikJjUCEdZvBYKXbMtiW5y3eefR6U2WQIDAQAB
AoGBAL1NtZM11sUZ4ZmjfNotV3jUFovmdNHsDR+DylkB1gzKpaKwgljlYLu3r3p8
Lgz+InzVDP+2ztE7xVlfzewstaNtRF/P32DI1J+zkK8tvW9jJ1Qj3kBIBeS6adn2
iWeMzcA4hNSekNPj3OXl8ZlsQHcwM+U0WoJV6t6nHF3dMMyBAkEA7KzVGFW5CgBn
OLITLbtMCpWgLeL7Cz5ZVZ/0bWOQ8L4Tl2h64XmPCLFWlmIWN1o8ndncfrb7r2BG
Y1QJcaNiyQJBAOq7XEuB9TMwXl6L8YdY18Ejve9TrTy8B9m9b++SeYYpKmrQrGxX
KOpSY6CV3W04fTdnv3GSeMD1wwqC3oUC7xECQQDAREd41WrU7S7tp/xckmNb1eGi
ZVp779Ky9JakptYAPOm9fmsU8KN59FbbJCPYI75Kncm6Rvx/pD6KQqLJZmnBAkEA
uLeqYM0rHRZCHRr5fa4fUyECVbS+jh3V+7ZEwP2+XiJE+/usxDEuxH8DYZqtvkaG
2zPshr5iAk8kJkBoRbnSUQJAbS97Id1Beq/rejivApjKTP2lCfkOj4TbluNspiec
rs7Eac1FTIFOwD+6tMG3K7nuRQ1UB9Cltjy15XW8MmYHRA==
-----END RSA PRIVATE KEY-----

'''
    publicKey = '''-----BEGIN RSA PUBLIC KEY-----
MIGJAoGBANkDMlvSfrZaowpBQUF0gFRzwt3w0q4rWet33FdWuq39TBdCVicy4Zzw
zX5ziuFExkdwbVrhEHBRkGNBJYVJ/1F1gAwd3KdyaId5NwATFkkLyM3wOtFqttwm
o888U9KwIyhGH2bFkRIsk+KQmNQIR1m8Fgpdsy2JbnLd559HpTZZAgMBAAE=
-----END RSA PUBLIC KEY-----
'''
    # publicKey = privateKey = None
    data = 'nihao@456'
    RsaUtil = RsaUtil(privateKey, publicKey, data)
    encryptData = RsaUtil.encrypt()
    RsaUtil.decrypt(encryptData)
    signature = RsaUtil.sign()
    print(str(signature))
    result = RsaUtil.verify(signature)
    print("验签：", result)