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
241
test/jdk/javax/imageio/plugins/bmp/BMPPluginTest.java
Normal file
241
test/jdk/javax/imageio/plugins/bmp/BMPPluginTest.java
Normal file
|
|
@ -0,0 +1,241 @@
|
|||
/*
|
||||
* 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 4641872 4892194
|
||||
* @summary Tests writing and reading abilities of BMP plugin
|
||||
*/
|
||||
|
||||
import java.awt.Color;
|
||||
import java.awt.Graphics2D;
|
||||
import java.awt.image.BufferedImage;
|
||||
import java.awt.image.ColorModel;
|
||||
import java.awt.image.Raster;
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.File;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.FileOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.util.Iterator;
|
||||
|
||||
import javax.imageio.IIOException;
|
||||
import javax.imageio.IIOImage;
|
||||
import javax.imageio.ImageIO;
|
||||
import javax.imageio.ImageReader;
|
||||
import javax.imageio.ImageTypeSpecifier;
|
||||
import javax.imageio.ImageWriteParam;
|
||||
import javax.imageio.ImageWriter;
|
||||
import javax.imageio.metadata.IIOMetadata;
|
||||
import javax.imageio.spi.ImageWriterSpi;
|
||||
|
||||
public class BMPPluginTest {
|
||||
|
||||
private static final int[] types = {
|
||||
BufferedImage.TYPE_INT_RGB, // = 1;
|
||||
BufferedImage.TYPE_INT_ARGB, // = 2;
|
||||
BufferedImage.TYPE_INT_ARGB_PRE, // = 3;
|
||||
BufferedImage.TYPE_INT_BGR, // = 4;
|
||||
BufferedImage.TYPE_3BYTE_BGR, // = 5;
|
||||
BufferedImage.TYPE_4BYTE_ABGR, // = 6;
|
||||
BufferedImage.TYPE_4BYTE_ABGR_PRE, // 7
|
||||
BufferedImage.TYPE_USHORT_565_RGB, // 8
|
||||
BufferedImage.TYPE_USHORT_555_RGB, // 9
|
||||
BufferedImage.TYPE_BYTE_GRAY, // 10
|
||||
BufferedImage.TYPE_USHORT_GRAY, //11
|
||||
BufferedImage.TYPE_BYTE_BINARY, //12
|
||||
BufferedImage.TYPE_BYTE_INDEXED //13
|
||||
};
|
||||
|
||||
private static String format = "BMP";
|
||||
|
||||
private static ImageReader ir = null;
|
||||
private static ImageWriter iw = null;
|
||||
private BufferedImage img;
|
||||
private ImageWriteParam param;
|
||||
private ByteArrayOutputStream baos;
|
||||
|
||||
private static void init() {
|
||||
|
||||
Iterator i = ImageIO.getImageWritersByFormatName(format);
|
||||
if (!i.hasNext()) {
|
||||
throw new RuntimeException("No available ImageWrites for "+format+" format!");
|
||||
}
|
||||
iw = (ImageWriter)i.next();
|
||||
|
||||
i = ImageIO.getImageReadersByFormatName(format);
|
||||
if (!i.hasNext()) {
|
||||
throw new RuntimeException("No available ImageReaders for " +format+" format!");
|
||||
}
|
||||
|
||||
ir = (ImageReader)i.next();
|
||||
}
|
||||
|
||||
public static void main(String[] args) {
|
||||
if (args.length > 0) {
|
||||
format = args[0];
|
||||
System.out.println("Test format " + format);
|
||||
}
|
||||
|
||||
init();
|
||||
System.out.println("IR="+ir);
|
||||
System.out.println("IW="+iw);
|
||||
ImageIO.setUseCache(false);
|
||||
|
||||
for (int i=0; i<types.length; i++) {
|
||||
boolean bPassed = true;
|
||||
Object reason = null;
|
||||
|
||||
try {
|
||||
|
||||
BufferedImage image = createTestImage(types[i]);
|
||||
|
||||
ImageWriteParam param = iw.getDefaultWriteParam();
|
||||
|
||||
BMPPluginTest t = new BMPPluginTest(image, param);
|
||||
boolean res = false;
|
||||
res = t.test();
|
||||
if (!res) {
|
||||
bPassed = false;
|
||||
reason = new String("Null result");
|
||||
}
|
||||
} catch (IllegalArgumentException ex) {
|
||||
System.out.println("Expected exception type was caught: " + ex);
|
||||
|
||||
} catch (Throwable ex ) {
|
||||
System.out.println("FAILED");
|
||||
ex.printStackTrace();
|
||||
bPassed = false;
|
||||
reason = ex;
|
||||
throw new RuntimeException("Test for type " + types[i] + " FAILED due to exception");
|
||||
}
|
||||
/*
|
||||
System.out.println("Type " + types[i] + " result: " +
|
||||
(bPassed ? "PASSED" : "FAILED") +
|
||||
((reason != null) ? (" Reason: " + reason) : ""));
|
||||
*/
|
||||
System.out.println("Test for type " + types[i] + " PASSED");
|
||||
}
|
||||
|
||||
System.out.println("END OF TEST");
|
||||
}
|
||||
|
||||
public BMPPluginTest(BufferedImage img, ImageWriteParam param) {
|
||||
|
||||
this.img = img;
|
||||
this.param = param;
|
||||
baos = new ByteArrayOutputStream();
|
||||
}
|
||||
|
||||
public boolean test() throws IIOException, IOException {
|
||||
|
||||
ir.reset();
|
||||
iw.reset();
|
||||
|
||||
String[] suffixes = iw.getOriginatingProvider().getFileSuffixes();
|
||||
|
||||
IIOMetadata md = iw.getDefaultImageMetadata(new ImageTypeSpecifier(img), param);
|
||||
|
||||
System.out.println("Image type " + img.getType());
|
||||
|
||||
ImageWriterSpi spi = iw.getOriginatingProvider();
|
||||
boolean bCanEncode = spi.canEncodeImage(img);
|
||||
|
||||
System.out.println("Can encode image? " + (bCanEncode ? "YES" : "NO"));
|
||||
if (!bCanEncode) {
|
||||
return true;
|
||||
}
|
||||
IIOImage iio_img = new IIOImage(img, null, md);
|
||||
|
||||
String fname = "test"+img.getType()+"."+suffixes[0];
|
||||
|
||||
iw.setOutput(ImageIO.createImageOutputStream(new FileOutputStream(new File(fname))));
|
||||
System.out.print("write image ... ");
|
||||
iw.write(iio_img);
|
||||
System.out.println("OK");
|
||||
System.out.print("read image ... ");
|
||||
|
||||
byte[] ba_image = baos.toByteArray();
|
||||
|
||||
ByteArrayInputStream bais = new ByteArrayInputStream(ba_image);
|
||||
|
||||
ir.setInput(ImageIO.createImageInputStream(new FileInputStream(new File(fname))));
|
||||
|
||||
BufferedImage res = ir.read(0);
|
||||
System.out.println("OK");
|
||||
|
||||
System.out.print("compare images ... ");
|
||||
boolean r = compare(img,res);
|
||||
System.out.println(r?"OK":"FAILED");
|
||||
return r;
|
||||
}
|
||||
|
||||
private boolean compare(BufferedImage in, BufferedImage out) {
|
||||
int width = in.getWidth();
|
||||
int height = in.getHeight();
|
||||
if (out.getWidth() != width || out.getHeight() != height) {
|
||||
throw new RuntimeException("Dimensions changed!");
|
||||
}
|
||||
|
||||
Raster oldras = in.getRaster();
|
||||
ColorModel oldcm = in.getColorModel();
|
||||
Raster newras = out.getRaster();
|
||||
ColorModel newcm = out.getColorModel();
|
||||
|
||||
for (int j = 0; j < height; j++) {
|
||||
for (int i = 0; i < width; i++) {
|
||||
Object oldpixel = oldras.getDataElements(i, j, null);
|
||||
int oldrgb = oldcm.getRGB(oldpixel);
|
||||
int oldalpha = oldcm.getAlpha(oldpixel);
|
||||
|
||||
Object newpixel = newras.getDataElements(i, j, null);
|
||||
int newrgb = newcm.getRGB(newpixel);
|
||||
int newalpha = newcm.getAlpha(newpixel);
|
||||
|
||||
if (newrgb != oldrgb ||
|
||||
newalpha != oldalpha) {
|
||||
throw new RuntimeException("Pixels differ at " + i +
|
||||
", " + j);
|
||||
}
|
||||
}
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
private static BufferedImage createTestImage(int type) throws IOException {
|
||||
|
||||
int w = 200;
|
||||
int h = 200;
|
||||
BufferedImage b = new BufferedImage(w, h, type);
|
||||
Graphics2D g = b.createGraphics();
|
||||
g.setColor(Color.white);
|
||||
g.fillRect(0,0, w, h);
|
||||
g.setColor(Color.black);
|
||||
g.fillOval(10, 10, w -20, h-20);
|
||||
|
||||
return b;
|
||||
}
|
||||
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue