Реализация AES OFB в python с pycryptodome - PullRequest
Новая
       17

Реализация AES OFB в python с pycryptodome

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

Я пытаюсь реализовать режим шифрования AES OFB с использованием библиотеки pycryptodome, и у меня возникают проблемы с пониманием того, что вводить в шифр. Я знаю, что в pycryptodome уже реализовано OFB, но мне нужно взять простой текст, разбить его на байты, применить соответствующие шаги OFB, чтобы зашифровать его с помощью шифра AES в режиме ECB, а затем расшифровать.

Как работают байтовые строки, такие как b'\x16\xa8W\xed.)\xc4\xb8x\xd6\xcf\x7f\xf3\xe3;^' в Python?

Мне нужно взять такую ​​строку байтов, зашифровать ее, затем разбить ее пополам и XOR с 8 байтами простого текста. Самый простой способ для меня понять и сделать это - зашифровать IV (байтовую строку сверху) с помощью AES, затем преобразовать его и простой текст в двоичный файл и переписать их, а затем преобразовать обратно в байтовые строки. Как я могу это сделать? 

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

plainText = b"Lorem ipsum dolor sit amet, consectetur adipiscing e"
key = b"ANAAREMEREAAAAAA"
iv = get_random_bytes(32)

print("iv: ", iv)

cipher = AES.new(key, AES.MODE_ECB)
toXor = cipher.encrypt(iv)

print("toXor: ", toXor)

"""
toXorF=first half of toXor
ivS=second half of iv

cipherText=toXorF xored to first 8 bytes of plainText
iv=ivS + toXorS
"""

Вывод отпечатков: 

iv:  b"v'xg;\xd7h\xfc\xf2\xa4[\r\xee]J\x1eu\xa5\xe68\xa3a\xde\x02(\xc1\xe0\xc9z\x0f\xc3n"
toXor:  b'\x97\xbex\xfc\xb6\xbb\x85\xccZ\xc4\xe4\x9d\xd6M\xf2\xd7\xb7\xbf\xd0\xbe\xa5A\xd6\xee\x07U\xe0S\x7f\x88\xea\xcd'

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

1 Ответ

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

Я применил то, что предложил t.m.adam, и это последний код, который отлично работает. 

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

plainText = b"Lorem ipsum dolor sit amet, consectetur adipiscing e"
key = b"ANAAREMEREAAAAAA"


def ofbEnc(plainText, key):
    pos = 0
    cipherTextChunks = []
    iv = get_random_bytes(16)
    originalIV = iv
    cipher = AES.new(key, AES.MODE_ECB)
    if len(plainText) % 16 != 0:
        plainText += b"1"
    while len(plainText) % 16 != 0:
        plainText += b"0"
    while pos + 16 <= len(plainText):
        toXor = cipher.encrypt(iv)
        nextPos = pos + 16
        toEnc = plainText[pos:nextPos]
        cipherText = bytes([toXor[i] ^ toEnc[i] for i in range(16)])
        cipherTextChunks.append(cipherText)
        pos += 16
        iv = toXor
    return (originalIV, cipherTextChunks)


def ofbDec(cipherTextChunks, key, iv):
    plainText = b""
    cipher = AES.new(key, AES.MODE_ECB)
    for chunk in cipherTextChunks:
        toXor = cipher.encrypt(iv)
        plainText += bytes([toXor[i] ^ chunk[i] for i in range(15)])
        iv = toXor
    while plainText[-1] == 48:
        plainText = plainText[0:-1]
    if plainText[-1] == 49:
        plainText = plainText[0:-1]
    return plainText


iv, result = ofbEnc(plainText, key)
print(iv, result)

plain = ofbDec(result, key, iv)
print(plain)
...