RSA加密算法实现

RSA加密

import base64
import rsa


class RSAendecrypt(object):
    """
    generate public and private key (生成公钥和私钥)
    encrypt passwd (加密数据)
    decrypt passwd (解密数据)
    """

    def __init__(self, filepath=None):
        # 给定参数filepath init filepath 获取公钥或私钥存储路径
        if filepath is None:
            filepath = './'
        self.filepath = filepath

    def generatekey(self):
        # public and private keys 生成公钥和私钥到指定的路径
        # 生成密钥对
        pub, pri = rsa.newkeys(1024)
        # 使用python rsa模块生成的公/私钥均为pkcs1格式
        pub_key = pub.save_pkcs1().decode()
        pri_key = pri.save_pkcs1().decode()
        #导入密钥
        with open(self.filepath + 'private.pem', 'w') as f:
            f.write(pri_key)
        with open(self.filepath + 'public.pem', 'w') as f:
            f.write(pub_key)

    def encrypt(self, enstr):
        #encrypt string 加密数据 返回加密后的值
        with open(self.filepath + 'public.pem', 'r') as f:
            result = f.read()
        # 解析公钥赋值
        pub = rsa.PublicKey.load_pkcs1(result)
        endata = rsa.encrypt(enstr.encode('utf-8'), pub)
        return base64.b64encode(endata).decode('utf-8')

    def decrypt(self, destr):
        #decrypt string 解密数据 返回被加密的值
        with open(self.filepath + 'private.pem', 'r') as f:
            result = f.read()
        # 解析私钥赋值
        pri = rsa.PrivateKey.load_pkcs1(result)
        dedata = rsa.decrypt(base64.b64decode(destr), pri)
        return dedata.decode('utf-8')


if __name__ == '__main__':
    # 定义一个字符串
    mingwen = input("明文：")
    # 创建密钥生成对象
    rsadate = RSAendecrypt()
    # 生成公钥和私钥
    rsadate.generatekey()
    # 加密数据
    encry = rsadate.encrypt(mingwen)
    print(encry)
    # 解密数据
    decry = rsadate.decrypt(encry)
    print(decry)