python實現RSA加解密

python實現rsa加解密

概念網上很多，不再贅述，直接上代碼。

import rsa
import base64
# 1、公私鑰由服務端生成（agent配置文件是服務端生成）
# 2、對稱加密key必須是16位字符串，採用公鑰加密，私鑰解密
# 3、body數據採用key進行res對稱加密

class MyRsa(object):
    # 生成公私鑰
    def genkey(self):
        (pubkey, privkey) = rsa.newkeys(1024)

        with open('public.pem', 'w+') as f:
            f.write(pubkey.save_pkcs1().decode())

        with open('private.pem', 'w+') as f:
            f.write(privkey.save_pkcs1().decode())

    @staticmethod
    def encrypt(key):
        with open('public.pem', 'rb') as publickfile:
            p = publickfile.read()
        pubkey = rsa.PublicKey.load_pkcs1(p)

        encrypt_key = rsa.encrypt(key.encode('utf-8'), pubkey)

        encrypt_key = base64.standard_b64encode(encrypt_key)

        return encrypt_key

    @staticmethod
    def decrypt(encrypt_key):
        with open('private.pem', 'rb') as privatefile:
            p = privatefile.read()
        privkey = rsa.PrivateKey.load_pkcs1(p)

        encrypt_key = base64.standard_b64decode(encrypt_key)

        key = rsa.decrypt(encrypt_key, privkey)

        return key


if __name__ == '__main__':
    myrsa = MyRsa()
    a = myrsa.encrypt("1234567890abcdef")
    print("加密後", a.decode('utf-8'))
    b = myrsa.decrypt(a)
    print("解密後", b.decode('utf-8'))