Question

完整的代码如下：

from Crypto.Protocol.KDF import scrypt
from Crypto.Cipher import AES
from Crypto.Random import get_random_bytes

class transmitter():

    def __init__(self):

        self.random_password = None
        self.message_plain = True
        self.key = None
        self.salt = None

        self.password_option()
        self.text_option()
        self.encrypt()


    def password_option(self):

        while ( self.random_password == None ):

            random = input("\nDo you want to generate a random symmetric key? (y/n)\n\n>>> ").strip().lower()

            if random == "y":
                self.random_password = True
                self.random()

            elif random == "n":
                self.random_password = False
                self.random()

            else:
                pass

    def text_option(self):

        if self.message_plain:

            question = input("\nHow will you enter your message?\n\n[1] file\n\n[2] directly in the program\n\n>>> ").strip()

            if question == "1":
                path = input("\nEnter the file path\n\n>>> ")
                name = path.split("\\")[-1]
                with open(name,mode = "r") as self.message_plain:
                    self.message_plain.read().encode("utf-8")

            elif question == "2":
                self.message_plain = input("\nEnter your message\n\n>>> ").strip()
                self.message_plain = self.message_plain.encode("utf-8")


    def random(self):

        if self.random_password:
            password = "password".encode("utf-8")
            self.salt = get_random_bytes(16)
            self.key = scrypt(password, self.salt, 16, N=2**14, r=8, p=1)

        else:
            password = input("\nEnter your password\n\n>>> ").strip()
            self.salt = get_random_bytes(16)
            self.key = scrypt(password.encode("utf-8"), self.salt, 16, N=2**14, r=8, p=1)


    def encrypt(self):

        cipher = AES.new(self.key,AES.MODE_GCM)
        cipher.update(b"header")
        cipher_text,tag_mac = cipher.encrypt_and_digest(self.message_plain)
        transmitted_message = cipher_text,tag_mac,self.salt,cipher.nonce
        Receptor(transmitted_message)

class receiver():

    def __init__(self,received_message):
        # nonce = aes_cipher.nonce
        self.cipher_text,self.tag_mac,self.salt,self.nonce = received_message
        self.decrypt(self.cipher_text,self.tag_mac,self.salt,self.nonce)

    def decrypt(self,cipher_text,tag_mac,salt,nonce):

        try:
            password = input("\nEnter your password\n\n>>> ").strip()
            decryption_key = scrypt(password.encode("utf-8"), salt, 16, N=2**14, r=8, p=1)
            cipher = AES.new(decryption_key,AES.MODE_GCM,nonce)
            cipher.update(b"header")
            plain_text = cipher.decrypt_and_verify(cipher_text,tag_mac)
            plain_text = plain_text.decode("utf-8")
            print(f"\nText -> {plain_text}\n")

        except ValueError:
            print("\nAn error has occurred..\n")

if __name__ == '__main__':
    init = transmitter()

错误如下：

Traceback (most recent call last):
  File ".\crypto.py", line 109, in <module>
    init = Emisor()
  File ".\crypto.py", line 16, in __init__
    self.encrypt()
  File ".\crypto.py", line 74, in encrypt
    cipher_text,tag_mac = cipher.encrypt_and_digest(self.message_plain)
  File "C:\Users\EQUIPO\AppData\Local\Programs\Python\Python37-32\lib\site-packages\Crypto\Cipher\_mode_gcm.py", line 547, in encryp
t_and_digest
    return self.encrypt(plaintext, output=output), self.digest()
  File "C:\Users\EQUIPO\AppData\Local\Programs\Python\Python37-32\lib\site-packages\Crypto\Cipher\_mode_gcm.py", line 374, in encryp
t
    ciphertext = self._cipher.encrypt(plaintext, output=output)
  File "C:\Users\EQUIPO\AppData\Local\Programs\Python\Python37-32\lib\site-packages\Crypto\Cipher\_mode_ctr.py", line 189, in encryp
t
    ciphertext = create_string_buffer(len(plaintext))
TypeError: object of type '_io.TextIOWrapper' has no len()

我已经尝试了一切，但老实说，我不知道该怎么办。有谁知道可能出什么问题了？

基本上，我想做的是将读取特定文件的结果保存在变量中。这样我就可以对其进行加密。幸运的是，其余代码运行良好，只有出现错误的那一部分代码（我在问题中提出的第一部分）。

Answer 1

您需要更改此内容

with open(name, mode = "r") as self.message_plain:
    self.message_plain.read().encode("utf-8")

对此：

with open(name, mode="r") as input_file:
    self.message_plain = input_file.read().encode("utf-8")

第一个with块在功能上等效于此：

self.message_plain = open(name, mode = "r")
self.message_plain.read().encode("utf-8")
self.message_plain.close()

这没有意义，因为它只是将self.message_plain变成文件对象，而没有保存实际内容read()。

第二个with块在功能上等效于此：

input_file = open(name, mode = "r")
self.message_plain = input_file.read().encode("utf-8")
input_file.close()

这更有意义。

基本上，您错误地使用了with statement，因为您将self.message_plain的类型更改为<class '_io.TextIOWrapper'>。您可以通过在type(self.message_plain)语句之后/之外打印with来进行检查。但是encrypt_and_digest方法期望一个类似字节的序列：

>>> help(cipher.encrypt_and_digest)
Help on method encrypt_and_digest in module Crypto.Cipher._mode_gcm:

encrypt_and_digest(plaintext, output=None) method of Crypto.Cipher._mode_gcm.GcmMode instance
    Perform encrypt() and digest() in one step.

    :Parameters:
      plaintext : bytes/bytearray/memoryview
        The piece of data to encrypt.
...

我还要建议您改进编码风格。

Put spaces after commas。

代替此：

self.cipher_text,self.tag_mac,self.salt,self.nonce = received_message

这样写：

self.cipher_text, self.tag_mac, self.salt, self.nonce = received_message

Class names should use the CapWords convention. 代替：
```
class transmitter
```
将其更改为
```
class Transmitter
```
将变量初始化为您期望的相同类型。

代替此：
```
self.message_plain = True
```
将其初始化为空字符串（""或（None）。

TypeError：“ _ io.TextIOWrapper”类型的对象没有len（）

1 个答案: