[英]Python AES Decryption Routine (Code Help)
I developed a code based on information available online regarding an AES Encryption and Decryption routine. 我根据在线上有关AES加密和解密例程的信息开发了代码。
Language: Python 2.7.x 语言:Python 2.7.x
Complete Code - 完整代码-
#!/usr/bin/python
import sys, os
import hashlib
import base64
from Crypto.Cipher import AES
## Variables in computation.
IV = u'1234567890123456'
BLOCK_SIZE = 32
INTERRUPT = u'\u0001'
PAD = u'\u0000'
SECRET = os.urandom(32)
filename=sys.argv[1]
def AddPadding(data, interrupt, pad, block_size):
new_data = ''.join([data, interrupt])
new_data_len = len(new_data)
remaining_len = block_size - new_data_len
to_pad_len = remaining_len % block_size
pad_string = pad * to_pad_len
return ''.join([new_data, pad_string])
def StripPadding(data, interrupt, pad):
return data.rstrip(pad).rstrip(interrupt)
def encAES(cipher_code, file_data):
data_padded = AddPadding(file_data, INTERRUPT, PAD, BLOCK_SIZE)
encrypted = cipher_code.encrypt(data_padded)
return encrypted
def decAES(cipher_code, file_data):
decrypted = cipher_code.decrypt(file_data)
return StripPadding(decrypted, INTERRUPT, PAD)
def FileSave(fwname, fwdata):
f = open(fwname, 'w')
f.write(fwdata)
f.close
def FileRead(frname):
f = open(frname, 'rb')
frdata = f.read()
return frdata
cipher = AES.new(SECRET, AES.MODE_CBC, IV)
## Encryption
data2encrypt = base64.b64encode(FileRead(filename))
encrypted_data = encAES(cipher, data2encrypt)
encrypted_content = base64.b64encode(encrypted_data)
encrypted_filename = "enc_"+filename
FileSave(encrypted_filename, encrypted_content)
print "Encryption complete. File saved as: "+ encrypted_filename
## Decryption
data2decrypt = base64.b64decode(FileRead(encrypted_filename))
decrypted_data = decAES(cipher, data2decrypt)
decrypted_content = base64.b64decode(decrypted_data)
decrypted_filename = "dec_"+filename
FileSave(decrypted_filename, decrypted_content)
print "Decryption complete. File saved as: "+ decrypted_filename
Now, the encryption routine is working fine but Decryption Routine is giving an error - 现在,加密例程可以正常工作,但是解密例程给出了错误-
Commandline - python test.py sample.txt 命令行-python test.py sample.txt
ERROR: 错误:
Traceback (most recent call last):
File "test.py", line 65, in <module>
decrypted_data = decAES(cipher, data2decrypt)
File "test.py", line 38, in decAES
return StripPadding(decrypted, INTERRUPT, PAD)
File "test.py", line 29, in StripPadding
return data.rstrip(pad).rstrip(interrupt)
UnicodeDecodeError: 'ascii' codec can't decode byte 0xe6 in position 3: ordinal not in range(128)
What can be the possible workaround? 可能的解决方法是什么?
You are concatenating a byte string with a Unicode value, triggering an automatic decode of the bytestring. 您正在使用Unicode值连接字节字符串,从而触发该字节字符串的自动解码。 This fails, as your decrypted text is not decodable as ASCII.
这将失败,因为您解密后的文本无法解码为ASCII。
Don't use Unicode INTERRUPT
and PAD
values here; 在此不要使用Unicode
INTERRUPT
和PAD
值; you are not reading Unicode data from the file here anyway: 无论如何,您不是从文件中读取Unicode数据:
INTERRUPT = '\1'
PAD = '\0'
You'll have to create a new instance of the AES
object to decrypt; 您必须创建一个新的
AES
对象实例才能解密。 you cannot reuse the object you used for encrypting as it's IV state has been altered by the encryption: 您不能重复使用用于加密的对象,因为其IV状态已被加密更改:
decrypt_cipher = AES.new(SECRET, AES.MODE_CBC, IV)
decrypted_data = decAES(decrypt_cipher, data2decrypt)
With those changes your code works and can encrypt and again decrypt the data. 通过这些更改,您的代码可以工作,并且可以加密并再次解密数据。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.