python 3 pycrypto iv должен иметь длину 16 байт - PullRequest
Новая
       8

python 3 pycrypto iv должен иметь длину 16 байт

0 голосов
/ 10 января 2019

, поэтому я пытался создать программу шифрования AES на основе руководства github pycrypto ссылка на github , однако, когда я иду на декодирование, появляется ошибка: 

Traceback (most recent call last):
File "/home/pi/Desktop/aes/newAES.py", line 24, in <module>
    print(decrypt(key,msg,iv))
File "/home/pi/Desktop/aes/newAES.py", line 13, in decrypt
    cipher = AES.new(key,AES.MODE_CFB)
File "/usr/lib/python3/dist-packages/Crypto/Cipher/AES.py", line 94, in new
    return AESCipher(key, *args, **kwargs)
File "/usr/lib/python3/dist-packages/Crypto/Cipher/AES.py", line 59, in __init__
blockalgo.BlockAlgo.__init__(self, _AES, key, *args, **kwargs)
File "/usr/lib/python3/dist-packages/Crypto/Cipher/blockalgo.py", line 141, in __init__
    self._cipher = factory.new(key, *args, **kwargs)
ValueError: IV must be 16 bytes long

мой код: 

from Crypto.Cipher import AES
from Crypto.Random import get_random_bytes
def encrypt(key,msg):
    if key == 0:
        key=get_random_bytes(16)
        print("key: "+key)
    iv = get_random_bytes(16)
    print('iv: '+str(iv))
    cipher = AES.new(key,AES.MODE_CFB,iv)
    ciphertext= cipher.decrypt(msg)
    return("your encrypted message: "+str(ciphertext))
def decrypt(key,ciphertext,iv):
    cipher = AES.new(key,AES.MODE_CFB)
    msg = cipher.decrypt(ciphertext)
ed = input('(e)ncrypt or (d)ecrypt: ')
if ed=='e':
    key = input('16 digit key: ')
    msg = input('message: ')
    print(encrypt(key,msg))
elif ed =='d':
    key = input('16 digit key: ')
    iv = bytes(input('iv: '),'utf-8')
    msg = bytes(input('encrypted message:'),'utf-8')
    print(decrypt(key,msg,iv))

Буду признателен за любую помощь в решении этой проблемы, надеюсь, это не глупая ошибка

1 Ответ

0 голосов
/ 11 января 2019

Проблема с iv состоит в том, что он состоит из случайных байтов , но он считывается в вашу программу как строка . Вызов bytes для этой строки не делает то, что вы ожидаете. 

>>> iv = b'\xba\x0eyO8\x17\xcf\x97=\xf2&l34#('
>>> siv = str(iv)
>>> siv
"b'\\xba\\x0eyO8\\x17\\xcf\\x97=\\xf2&l34#('"   # Note 'b' is part of the string
>>> biv = bytes(siv, 'utf-8')
>>> biv
b"b'\\xba\\x0eyO8\\x17\\xcf\\x97=\\xf2&l34#('"  # Now there are two 'b's!

Вы можете решить эту проблему, используя ast.literal_eval: 

>>> ast.literal_eval(siv)
b'\xba\x0eyO8\x17\xcf\x97=\xf2&l34#('

Вот рабочая версия вашего кода - я убрал необходимость копировать / вставлять iv, но те же замечания относительно ввода байтов применимы и к зашифрованному тексту. 

import ast

from Crypto.Cipher import AES
from Crypto.Random import get_random_bytes


def encrypt(key, msg):
    iv = get_random_bytes(16)
    cipher = AES.new(key, AES.MODE_CFB, iv)
    ciphertext = cipher.encrypt(msg)    # Use the right method here
    return iv + ciphertext


def decrypt(key, ciphertext):
    iv = ciphertext[:16]
    ciphertext = ciphertext[16:]
    cipher = AES.new(key, AES.MODE_CFB, iv)
    msg = cipher.decrypt(ciphertext)
    return msg.decode("utf-8")


if __name__ == "__main__":
    ed = input("(e)ncrypt or (d)ecrypt: ")
    if ed == "e":
        key = input("16 digit key: ")
        msg = input("message: ")
        print("Encrypted message: ", encrypt(key, msg))
    elif ed == "d":
        key = input("16 digit key: ")
        smsg = input("encrypted message: ")
        msg = ast.literal_eval(smsg)
        print("Decrypted message: ", decrypt(key, msg))

Код в действии: 

(e)ncrypt or (d)ecrypt: e
16 digit key: abcdabcdabcdabcd
message: Spam, spam, spam
Encrypted message:  b'\xa4?\xa9RI>\x1f\xb5*\xb2,NWN\x0c\xfd"yB|\x1f\x82\x96\xd5\xb4\xd4\x1d&\x8bM\xdb\x07'

(e)ncrypt or (d)ecrypt: d
16 digit key: abcdabcdabcdabcd
encrypted message: b'\xa4?\xa9RI>\x1f\xb5*\xb2,NWN\x0c\xfd"yB|\x1f\x82\x96\xd5\xb4\xd4\x1d&\x8bM\xdb\x07'
Decrypted message:  Spam, spam, spam
...