undefect. CWE-407 — 63 sites patched across 27 ecosystems
Authors: russell@unturf.com · brackishbert@gmail.com · foxhop.net · TimeHexOn.com Patches, unit tests, benchmarks, whitepaper, and outreach briefs. Public domain — no copyright claimed. Use freely.
This commit is contained in:
commit
0a580b313d
70422 changed files with 17213626 additions and 0 deletions
228
test/jdk/javax/imageio/plugins/bmp/Write3ByteBgrTest.java
Normal file
228
test/jdk/javax/imageio/plugins/bmp/Write3ByteBgrTest.java
Normal file
|
|
@ -0,0 +1,228 @@
|
|||
/*
|
||||
* Copyright (c) 2003, 2017, Oracle and/or its affiliates. All rights reserved.
|
||||
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
||||
*
|
||||
* This code is free software; you can redistribute it and/or modify it
|
||||
* under the terms of the GNU General Public License version 2 only, as
|
||||
* published by the Free Software Foundation.
|
||||
*
|
||||
* This code is distributed in the hope that it will be useful, but WITHOUT
|
||||
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||||
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
||||
* version 2 for more details (a copy is included in the LICENSE file that
|
||||
* accompanied this code).
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License version
|
||||
* 2 along with this work; if not, write to the Free Software Foundation,
|
||||
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
||||
*
|
||||
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
|
||||
* or visit www.oracle.com if you need additional information or have any
|
||||
* questions.
|
||||
*/
|
||||
|
||||
/*
|
||||
* @test
|
||||
* @bug 4892194 8014427
|
||||
* @summary Test checks that we able to encode TYPE_3BYTE_BGR images to bmp
|
||||
* format. Test failed if ArrayIndexOutOfBoundsException will be thrown
|
||||
* or pixel colors will be changed by the writing/reading.
|
||||
*/
|
||||
|
||||
import java.awt.Color;
|
||||
import java.awt.Dimension;
|
||||
import java.awt.Graphics;
|
||||
import java.awt.Graphics2D;
|
||||
import java.awt.color.ColorSpace;
|
||||
import java.awt.image.BufferedImage;
|
||||
import java.awt.image.ColorModel;
|
||||
import java.awt.image.ComponentColorModel;
|
||||
import java.awt.image.DataBuffer;
|
||||
import java.awt.image.Raster;
|
||||
import java.awt.image.WritableRaster;
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.File;
|
||||
import java.io.FileOutputStream;
|
||||
import java.io.IOException;
|
||||
|
||||
import javax.imageio.IIOImage;
|
||||
import javax.imageio.ImageIO;
|
||||
import javax.imageio.ImageWriter;
|
||||
import javax.imageio.stream.ImageOutputStream;
|
||||
import javax.swing.JComponent;
|
||||
import javax.swing.JFrame;
|
||||
|
||||
public class Write3ByteBgrTest {
|
||||
private static int width = 100;
|
||||
private static int height = 100;
|
||||
private static Color color = new Color(0x10, 0x20, 0x30);
|
||||
|
||||
static int bufferedImageType[] = {
|
||||
BufferedImage.TYPE_CUSTOM,
|
||||
BufferedImage.TYPE_BYTE_BINARY,
|
||||
BufferedImage.TYPE_3BYTE_BGR
|
||||
};
|
||||
|
||||
static String bufferedImageStringType[] = {
|
||||
"BufferedImage.TYPE_CUSTOM: test for BandedSampleModel",
|
||||
"BufferedImage.TYPE_BYTE_BINARY",
|
||||
"BufferedImage.TYPE_3BYTE_BGR"
|
||||
};
|
||||
|
||||
private static String writingFormat = "BMP";
|
||||
private static ImageWriter writer = (ImageWriter)ImageIO.getImageWritersByFormatName(writingFormat).next();
|
||||
private int type;
|
||||
|
||||
public static void main(String[] args) {
|
||||
|
||||
//int i = 0;
|
||||
for(int i=0; i<bufferedImageType.length; i++) {
|
||||
Write3ByteBgrTest t1 = new Write3ByteBgrTest(bufferedImageType[i]);
|
||||
|
||||
System.out.println("\n\nImage test for type " + bufferedImageStringType[i]);
|
||||
t1.doImageTest();
|
||||
}
|
||||
}
|
||||
|
||||
private Write3ByteBgrTest(int type) {
|
||||
this.type = type;
|
||||
}
|
||||
|
||||
private void doImageTest() {
|
||||
try {
|
||||
BufferedImage src = createTestImage(type);
|
||||
BufferedImage dst = writeImage(src);
|
||||
|
||||
compareImages(src, dst);
|
||||
} catch (ArrayIndexOutOfBoundsException e) {
|
||||
throw new RuntimeException("Test failed: index out of array bounds!");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
private void compareImages(BufferedImage src, BufferedImage dst) {
|
||||
Object dstPixel = dst.getRaster().getDataElements(width/2, height/2, null);
|
||||
Object srcPixel = src.getRaster().getDataElements(width/2, height/2, null);
|
||||
|
||||
if ( (src.getColorModel().getRed(srcPixel) != dst.getColorModel().getRed(dstPixel)) ||
|
||||
(src.getColorModel().getGreen(srcPixel) != dst.getColorModel().getGreen(dstPixel)) ||
|
||||
(src.getColorModel().getBlue(srcPixel) != dst.getColorModel().getBlue(dstPixel)) ||
|
||||
(src.getColorModel().getAlpha(srcPixel) != dst.getColorModel().getAlpha(dstPixel)) ) {
|
||||
|
||||
showPixel(src, width/2, height/2);
|
||||
showPixel(dst, width/2, height/2);
|
||||
|
||||
showRes(dst, src);
|
||||
throw new RuntimeException(
|
||||
"Colors are different: " +
|
||||
Integer.toHexString(src.getColorModel().getRGB(srcPixel)) + " and " +
|
||||
Integer.toHexString(dst.getColorModel().getRGB(dstPixel)));
|
||||
}
|
||||
}
|
||||
|
||||
private BufferedImage writeImage(BufferedImage src) {
|
||||
try {
|
||||
BufferedImage dst = null;
|
||||
if (!writer.getOriginatingProvider().canEncodeImage(src)) {
|
||||
throw new RuntimeException(writingFormat+" writer does not support the image type "+type);
|
||||
}
|
||||
System.out.println(writingFormat+" writer claims it can encode the image "+type);
|
||||
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
||||
ImageOutputStream ios = ImageIO.createImageOutputStream(baos);
|
||||
writer.setOutput(ios);
|
||||
IIOImage img = new IIOImage(src.getRaster(), null, null);
|
||||
writer.write(img);
|
||||
ios.close();
|
||||
baos.close();
|
||||
|
||||
// save to file
|
||||
File f = new File("test"+src.getType()+".bmp");
|
||||
FileOutputStream fos = new FileOutputStream(f);
|
||||
fos.write(baos.toByteArray());
|
||||
fos.close();
|
||||
|
||||
|
||||
ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
|
||||
dst = ImageIO.read(bais);
|
||||
return dst;
|
||||
} catch (IOException e) {
|
||||
e.printStackTrace();
|
||||
throw new RuntimeException(e);
|
||||
}
|
||||
}
|
||||
|
||||
private static void showPixel(BufferedImage src, int x, int y) {
|
||||
System.out.println("Img is " + src);
|
||||
Object p = src.getRaster().getDataElements(x, y, null);
|
||||
System.out.println("RGB: " +
|
||||
Integer.toHexString(src.getColorModel().getRGB(p)));
|
||||
System.out.println("Red: " +
|
||||
Integer.toHexString(src.getColorModel().getRed(p)));
|
||||
System.out.println("Green: " +
|
||||
Integer.toHexString(src.getColorModel().getGreen(p)));
|
||||
System.out.println("Blue: " +
|
||||
Integer.toHexString(src.getColorModel().getBlue(p)));
|
||||
System.out.println("Alpha: " +
|
||||
Integer.toHexString(src.getColorModel().getAlpha(p)));
|
||||
}
|
||||
|
||||
private static BufferedImage createTestImage(int type) {
|
||||
return createTestImage(type, color);
|
||||
}
|
||||
|
||||
private static BufferedImage createTestImage(int type, Color c) {
|
||||
BufferedImage i = null;
|
||||
if (type == BufferedImage.TYPE_CUSTOM) {
|
||||
WritableRaster wr = Raster.createBandedRaster(
|
||||
DataBuffer.TYPE_BYTE,
|
||||
width, height,
|
||||
width, // scanlineStride
|
||||
new int[] { 0, 1, 2},// bankIndices[],
|
||||
new int[] { 1, 2, 0},// bankOffsets[],
|
||||
null);
|
||||
|
||||
ColorSpace cs = ColorSpace.getInstance(ColorSpace.CS_sRGB);
|
||||
|
||||
ColorModel cm = new ComponentColorModel(cs,
|
||||
new int[] { 8, 8, 8},
|
||||
false,
|
||||
false,
|
||||
ColorModel.OPAQUE,
|
||||
DataBuffer.TYPE_BYTE);
|
||||
i = new BufferedImage(cm, wr, false, null);
|
||||
} else {
|
||||
i = new BufferedImage(width, height, type);
|
||||
}
|
||||
|
||||
Graphics2D g = i.createGraphics();
|
||||
|
||||
g.setColor(c);
|
||||
g.fillRect(0, 0, width, height);
|
||||
g.setColor(Color.white);
|
||||
g.drawRect(10,10, width-20, height-20);
|
||||
|
||||
return i;
|
||||
}
|
||||
|
||||
private static void showRes(final BufferedImage src,
|
||||
final BufferedImage dst)
|
||||
{
|
||||
final int w = src.getWidth()+ dst.getWidth();
|
||||
final int h = Math.max(src.getHeight(), dst.getHeight());
|
||||
|
||||
JFrame f = new JFrame("Test results");
|
||||
f.getContentPane().add( new JComponent() {
|
||||
public Dimension getPreferredSize() {
|
||||
return new Dimension(w,h);
|
||||
}
|
||||
|
||||
public void paintComponent(Graphics g) {
|
||||
g.drawImage(src,0,0, null);
|
||||
g.drawImage(dst, src.getWidth(),0, null);
|
||||
}
|
||||
});
|
||||
f.pack();
|
||||
f.setVisible(true);
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue