Pycrypto AES加密抛出两个错误

时间:2017-03-14 15:28:10

标签: python encryption cryptography aes pycrypto

我创建了一个小脚本来加密和解密文件,但文件是零字节,并且没有创建解密文件

代码:

from hashlib import md5
from Crypto import Random
from Crypto.Cipher import AES
import os
from Crypto import *

def encrypt(in_file, out_file, key, iv):
    bs = AES.block_size
    cipher = AES.new(key, AES.MODE_CBC, iv)
    finished = False
    while not finished:
        chunk = in_file.read(1024 * bs)
        if len(chunk) == 0 or len(chunk) % bs != 0:
            padding_length = bs - (len(chunk) % bs)
            chunk += bytes(padding_length) * chr(padding_length)
            finished = True
    out_file.write(cipher.encrypt(chunk))

def decrypt(in_file, out_file, key, iv):
    bs = AES.block_size
    cipher = AES.new(key, AES.MODE_CBC, iv)
    next_chunk = ''
    finished = False
    while not finished:
        chunk, next_chunk = next_chunk, cipher.decrypt(in_file.read(1024 * bs))
    if len(next_chunk) == 0:
        padding_length = ord(chunk[-1])
        if padding_length < 1 or padding_length > bs:
            raise ValueError("bad decrypt pad (%d)" % padding_length)
        if chunk[-padding_length:] != (padding_length * chr(padding_length)):
            raise ValueError("bad decrypt")
        chunk = chunk[:-padding_length]
        finished = True
    out_file.write(chunk)


in_file = open('C:/Users/saeed/Desktop/ImportantFolder/arrest.jpg', 'rb')
out_file = open('C:/Users/saeed/Desktop/ImportantFolder/arrest_enc.jpg', 'wb')
key = os.urandom(32)
iv = os.urandom(16)
encrypt(in_file, out_file, key, iv)
in_file.close()
out_file.close()
print('Encrypted!')

in_file = open('C:/Users/saeed/Desktop/ImportantFolder/arrest_enc.jpg', 'rb')
out_file = open('C:/Users/saeed/Desktop/ImportantFolder/arrest_dec.jpg', 'wb')
decrypt(in_file, out_file, key, iv)
in_file.close()
out_file.close()
print('Decrypted!')

加密的文件_enc已创建但是为零字节,并且在代码停止时无法解密。它在加密过程中陷入困境,从不打印加密!错误:

Traceback (most recent call last):
  File "C:/Users/saeed/IdeaProjects/encryptor/encryption.py", line 41, in <module>

encrypt(in_file, out_file, key, iv)

File "C:/Users/saeed/IdeaProjects/encryptor/encryption.py", line 15, in 
encrypt

chunk += padding_length * chr(padding_length)
TypeError: can't concat bytes to str

这是什么意思,我该如何解决?

1 个答案:

答案 0 :(得分:0)

从文件中读取时,读取数据的类型为bytes类型,而不是str类型。如果您改为使用chunk += bytes( padding_length ),则填充将附加到块中。