将字节数组转换为png

Question

我使用以下代码从图像中获取了一个字节数组。

String path = "/home/mypc/Desktop/Steganography/image.png";
File file = new File(path);
BufferedImage bfimage = ImageIO.read(file);

ByteArrayOutputStream baos = new ByteArrayOutputStream();
ImageIO.write(bfimage, "png", baos);
baos.flush();
byte[] img_in_bytes = baos.toByteArray();
baos.close();

然后，我使用以下代码将这些字节转换回png图像。

BufferedImage final_img = ImageIO.read(new ByteArrayInputStream(img_in_bytes));
File output_file = new File("Stegano2.png");
ImageIO.write(final_img, "png", output_file);

如果我只执行这段代码就完全没问题了。但是如果我尝试修改中间的一些字节，就像这样：

    Insert_number_to_image(image_in_bytes,  10);

我的方法"Inset_number_to_image“是这样的：

static void Insert_number_to_image(byte[] image, int size){     
    byte[] size_in_byte = new byte[4];
    size_in_byte[0] = (byte)(size >>> 0);
    size_in_byte[1] = (byte)(size >>> 8);
    size_in_byte[2] = (byte)(size >>> 16);
    size_in_byte[3] = (byte)(size >>> 24);

    byte temp;
    int count = 0;

    for(int i=0; i<4; i++)
    {
        for(int j=0; j<8; j++)
        {
            temp = size_in_byte[i];
            temp = (byte)(temp >>> j);
            temp = (byte)(temp & 1);
            if(temp == 1)
            {
                image[count] = (byte)(image[count] | 1);
            }
            else if(temp == 0)
            {
                image[count] = (byte)(image[count] & (byte)(~(1)));
            }
            count++;
        }
    }
}

然后，当我使用上面提到的相同代码将修改后的字节数组保存为png图像时，我得到了这个错误：

Exception in thread "main" java.lang.IllegalArgumentException: image == null!
    at javax.imageio.ImageTypeSpecifier.createFromRenderedImage(ImageTypeSpecifier.java:925)
    at javax.imageio.ImageIO.getWriter(ImageIO.java:1591)
    at javax.imageio.ImageIO.write(ImageIO.java:1520)
    at Steganography.main(Steganography.java:211)

Answer 1

您使用的是PNG图像的原始字节流。PNG is a compressed format，其中字节流不直接反映任何像素值，甚至更改一个字节都可能不可逆转地损坏文件。

相反，您希望使用以下命令将像素数据提取到字节数组中

byte[] pixels = ((DataBufferByte) img.getRaster().getDataBuffer()).getData();

现在您可以随心所欲地修改像素值。当您准备好将其保存回文件时，通过将像素数组放入DataBufferByte对象并将其传递给WriteableRaster (然后使用它来创建BufferedImage )来执行convert the pixel byte array to a BufferedImage。

您的方法适用于原始字节流不直接表示像素的格式，例如BMP格式。但是，即使这样，您也必须跳过前几个字节，以避免损坏报头。