如何在 Python 中将字节数组转换为位图图像？答案

【问题标题】：How do I convert byte array to bitmap image in Python?如何在 Python 中将字节数组转换为位图图像？
【发布时间】：2018-12-09 22:26:03
【问题描述】：

我正在尝试创建一个简单的密码，该密码与位图图像进行异或密钥。

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

如何将字节数组转换回位图图像？任何帮助将不胜感激。

【问题讨论】：

您的xor 函数返回一个列表，因此您不能将encrypted_image 和decrypted_image 作为参数传递给io.BytesIO()。也许将 xor 函数更改为 return bytearray([ a ^ b for a,b in zip(x,cycle(y))]) 可能会有所帮助。
感谢 myrmica。当我将解密的字节数组转换为位图图像时，您的建议有所帮助。但是，我无法将加密的字节数组转换为位图图像。我希望得到一张白噪声图像，并且能够将白噪声图像解密回原始位图。知道该怎么做吗？
BMP 文件的标头包含您不得使用密码修改的信息。 xor 函数中应该有类似 return x[:54] + bytearray([a^b for a, b in zip(x[54:], cycle(y))]) 的内容。但实际偏移量可能会有所不同，因为有different versions of BMP。

标签： python arrays bitmapimage

【解决方案1】：

我更深入地研究了如何使用 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 标头的大小。

（链接：original 和 encrypted 图片）

【讨论】：

【解决方案2】：

更新了将纯图像转换为密码图像并返回纯图像的脚本。

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 的帮助

【讨论】：