Я пытаюсь проверить, создайте простой шифр, который имеет ключ xor с растровым изображением.
import os, io, hashlib
from PIL import Image
from array import array
from itertools import cycle
key = "aaaabbbb"
def generate_keys(key):
round_keys = hashlib.md5(key).digest()
return bytearray(round_keys)
def readimage(path):
with open(path, "rb") as f:
return bytearray(f.read())
def generate_output_image(input_image, filename_out):
output_image = Image.open(io.BytesIO(input_image))
output_image.save(filename_out)
def xor(x,y):
return [ a ^ b for a,b in zip(x,cycle(y))]
round_keys = generate_keys(key)
input_image = readimage("lena512.bmp")
encrypted_image = xor(input_image, round_keys)
generate_output_image(encrypted_image, "lena_encrypted.bmp")
input_image = readimage("lena_encrypted.bmp");
decrypted_image = xor(input_image, round_keys)
generate_output_image(decrypted_image, "lena_decrypted.bmp")
Но при запуске скрипта я получаю следующую ошибку:
Traceback (most recent call last):
File "test.py", line 26, in <module>
generate_output_image(encrypted_image, "lena_encrypted.bmp")
File "test.py", line 17, in generate_output_image
output_image = Image.open(io.BytesIO(input_image))
TypeError: 'list' does not have the buffer interface
Как преобразовать массив байтов в растровое изображение? Любая помощь будет оценена с благодарностью.
Я немного больше посмотрел, как более элегантно извлекать/обновлять данные изображения с помощью инструментов, включенных в PIL/Pillow, и сделал этот простой рабочий пример:
from PIL import BmpImagePlugin
import hashlib
from itertools import cycle
keys = hashlib.md5(b"aaaabbbb").digest()
input_image = BmpImagePlugin.BmpImageFile("img/tea.bmp")
# extract pure image data as bytes
image_data = input_image.tobytes()
# encrypt
image_data = bytes(a^b for a, b in zip(image_data, cycle(keys)))
# create new image, update with encrypted data and save
output_image = input_image.copy()
output_image.frombytes(image_data)
output_image.save("img/tea-encrypted.bmp")
Таким образом, вам не нужно беспокоиться о размере заголовка BMP.
(ссылки: оригинал и зашифрованное изображение)
Обновленный скрипт, который преобразует обычное изображение в зашифрованное изображение и обратно в обычное изображение.
import os, io, hashlib, math, binascii
from PIL import Image
from array import array
from itertools import cycle
user_key = "aaaabbbb"
test_image = "lena512.bmp"
def generate_keys(key):
round_keys = hashlib.md5(key).digest()
return bytearray(round_keys)
def readimage(path):
with open(path, "rb") as f:
return bytearray(f.read())
def generate_output_image(input_image):
output_image = Image.open(io.BytesIO(input_image))
output_image.save(test_image)
def xor(x,y):
return x[:54] + bytearray([a^b for a, b in zip(x[54:], cycle(y))])
round_keys = generate_keys(user_key)
input_image = readimage(test_image)
encrypted_image = xor(input_image, round_keys)
generate_output_image(encrypted_image)
input_image = readimage(test_image);
decrypted_image = xor(input_image, round_keys)
generate_output_image(decrypted_image)
Спасибо myrmica за вашу помощь
xor
возвращает список, поэтому вы не можете передатьencrypted_image
иdecrypted_image
в качестве аргументаio.BytesIO()
. Возможно, может помочь изменение функцииxor
дляreturn bytearray([ a ^ b for a,b in zip(x,cycle(y))])
.