python 3.5.1 エラーへの対処

Question

###前提・実現したいこと
PyCryptoのRSAを使って暗号化&復号化をしたいのですが以下のエラーの原因がわからず困っています。

###発生している問題・エラーメッセージ

AttributeError: 'bytes' object has no attribute '_randfunc'

###該当のソースコード

from Crypto.Cipher import PKCS1_v1_5
from Crypto.PublicKey import RSA

def encryption(data):
    rsa = RSA.generate(2048)
    pub_key = rsa.publickey().exportKey()
    cipher = PKCS1_v1_5.new(pub_key)
    ciphertext = cipher.encrypt(data)

def decryption(ciphertext):
    rsa = RSA.generate(2048)
    pri_key = rsa.exportKey()
    cipher = PKCS1_v1_5.new(pri_key)
    orig = cipher.decrypt(ciphertext)


if __name__ == "__main__":
    data = b'message'
    print(encryption(data))
    ciphertext = encryption(data)
    print(decryption(ciphertext))

###試したこと
エラーで指定されている箇所を確認したのですが解決方法はわかりませんでした。

###補足情報(言語/FW/ツール等のバージョンなど)
python3.5.1, pycrypto2.6.1を使っています。

Answer 1

newにはRSA key objectを入力しなければならないらしいです。

https://www.dlitz.net/software/pycrypto/api/2.6/Crypto.Cipher.PKCS1_v1_5-module.html#new

from Crypto.Cipher import PKCS1_v1_5
from Crypto.PublicKey import RSA


def encryption(data):
    rsa = RSA.generate(2048)
    pub_key = rsa.publickey()  # ここを変更
    cipher = PKCS1_v1_5.new(pub_key)
    ciphertext = cipher.encrypt(data)
    return ciphertext


if __name__ == "__main__":
    data = b'message'
    print(encryption(data))

Comments

[YouheiSakurai]

ちなみに '_randfunc'が珍しく感じたのでググったらhttps://stackoverflow.com/a/24792738を見つけてドキュメンテーションを探してこの答えにたどり着きました。

[YusukeKanai]

回答していただきありがとうございます。