Python python rsa 加密问题

yiyu · 2021年08月16日 · 最后由 yiyu 回复于 2021年08月18日 · 4145 次阅读

python rsa 使用 java 接口返的 pubkey 进行密码的加密 提示 AttributeError: 'str' object has no attribute 'n' 是长度问题么

共收到 4 条回复 时间 点赞

把 rsaEncrypt 里的参数名改一下,不要用 str

import rsa
import base64
from common.FILE_PATH import PRIVATE_KEY_FILE_PATH,PUBLIC_KEY_FILE_PATH

class RsaSign(object):

    @classmethod
    def to_encrypt(cls, msg, pub_key=None):
        """
        非对称加密
        :param msg: 待加密字符串或者字节
        :param pub_key: 公钥
        :return: 密文
        """
        if isinstance(msg, str):            # 如果msg为字符串, 则转化为字节类型
            msg = msg.encode('utf-8')
        elif isinstance(msg, bytes):        # 如果msg为字节类型, 则无需处理
            pass
        else:                               # 否则抛出异常
            raise TypeError('msg必须为字符串或者字节类型!')

        if not pub_key:                     # 如果pub_key为空, 则使用全局公钥
            with open(PUBLIC_KEY_FILE_PATH, 'rb') as file:   # 读取公钥
                pub_key = file.read()

        elif isinstance(pub_key, str):      # 如果pub_key为字符串, 则转化为字节类型
            pub_key = pub_key.encode('utf-8')
        elif isinstance(pub_key, bytes):    # 如果msg为字节类型, 则无需处理
            pass
        else:                               # 否则抛出异常
            raise TypeError('pub_key必须为None、字符串或者字节类型!')
        #后端开发密钥格式为pkcs8格式公钥,'-----BEGIN PUBLIC KEY-----' 用load_pkcs1_openssl_pem()这个方法
        #pkcs1格式公钥用load_pkcs1()这个方法
        public_key_obj = rsa.PublicKey.load_pkcs1_openssl_pem(pub_key)  # 创建 PublicKey 对象

        cryto_msg = rsa.encrypt(msg, public_key_obj)  # 生成加密文本
        cipher_base64 = base64.b64encode(cryto_msg)   # 将加密文本转化为 base64 编码

        return cipher_base64.decode()   # 将字节类型的 base64 编码转化为字符串类型


    @classmethod
    def to_decrypt(cls, msg, pri_key=None):  # 用私钥解密
        if isinstance(msg, str):  # 如果msg为字符串, 则转化为字节类型
            msg = msg.encode('utf-8')
        elif isinstance(msg, bytes):  # 如果msg为字节类型, 则无需处理
            pass
        else:  # 否则抛出异常
            raise TypeError('msg必须为字符串或者字节类型!')

        if not pri_key:  # 如果pub_key为空, 则使用全局公钥
            with open(PRIVATE_KEY_FILE_PATH, 'rb') as file:  # 读取公钥
                pri_key = file.read()

        elif isinstance(pri_key, str):  # 如果pub_key为字符串, 则转化为字节类型
            pri_key = pri_key.encode('utf-8')
        elif isinstance(pri_key, bytes):  # 如果msg为字节类型, 则无需处理
            pass
        else:  # 否则抛出异常
            raise TypeError('pub_key必须为None、字符串或者字节类型!')
            # python只支持pkcs1格式的私钥解密,需要使用openssl将pkcs8转为pkcs1格式
        #https://blog.csdn.net/sanpangouba/article/details/100997735
        privateKey_key_obj = rsa.PrivateKey.load_pkcs1(pri_key)  # 创建 PublicKey 对象

        msg=base64.b64decode(msg) #base64 解码

        decrypt_msg = rsa.decrypt(msg, privateKey_key_obj)  # 生成加密文本

        return decrypt_msg.decode('utf-8')  # 将字节类型的文本转化为字符串类型

    @classmethod
    def create_keys(cls):  # 生成公钥和私钥
        # 设置1024位长度
        (pubkey, privkey) = rsa.newkeys(1024)
        #密钥格式为pkcs1格式, -----BEGIN RSA PUBLIC KEY-----
        pub = pubkey.save_pkcs1()
        with open(PUBLIC_KEY_FILE_PATH, 'wb+')as f:
            f.write(pub)
        pri = privkey.save_pkcs1()
        with open(PRIVATE_KEY_FILE_PATH, 'wb+')as f:
            f.write(pri)

    @classmethod
    def generate_sign(cls, message):
        """
        生成sign
        :param message: message, 代加密内容, 为str类型
        :return: 加密后字符串
        """
        sign = cls.to_encrypt(message)  # 生成sign
        return sign

    @classmethod
    def decrypt_sign(cls,message):
        """
        解密sign
        :param message: message, 加密内容, 为str类型
        :return: 解密后字符串
        """
        sign = cls.to_decrypt(message)  # 解密sign
        return sign




if __name__ == '__main__':
    msg='F5^t~#6H'
    aaaa=RsaSign.generate_sign(msg)
    print(aaaa)

试一下将公钥保存成文件试试

-----BEGIN PUBLIC KEY-----
MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDIvvJN1o5dsjSx2c8Ju0W5nbuli
ZuU/xJ00julQ2r4VJYnYRCzlLjYaWJvL60HJ9DiEeTo6LzgM7cq5PM9aqI6sLo8T1
zuUJ7qYWYAdH1tCHXnXT9tRJ/e+TT7scv5H7mOViao28Ne+5dawVvSZtODwUT+fbS
vpk9IUYJ+13jvRwIDABCD
-----END PUBLIC KEY-----
yiyu #4 · 2021年08月18日 Author

解决了 原因是接口给的公钥是 str 类型的 与 python rsa 的公钥要求类型不一致 处理完就好了

yiyu 关闭了讨论 08月18日 09:39
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册