1 /* 2 * Copyright (c) 2013, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 /** 25 * @test 26 * @bug 8020983 27 * @summary Test verifies that jpeg writer instances are collected 28 * even if destroy() or reset() methods is not invoked. 29 * 30 * @run main JpegWriterLeakTest 31 */ 32 33 import java.awt.Color; 34 import java.awt.Graphics2D; 35 import java.awt.image.BufferedImage; 36 import java.io.ByteArrayOutputStream; 37 import java.io.IOException; 38 import java.lang.ref.Reference; 39 import java.lang.ref.ReferenceQueue; 40 import java.lang.ref.WeakReference; 41 import java.util.ArrayList; 42 import java.util.Random; 43 import javax.imageio.ImageIO; 44 import javax.imageio.ImageWriter; 45 import javax.imageio.stream.ImageOutputStream; 46 47 public class JpegWriterLeakTest { 48 49 public static void main(String[] args) { 50 final ReferenceQueue<ImageWriter> queue = new ReferenceQueue<ImageWriter>(); 51 final ArrayList<Reference<? extends ImageWriter>> refs = new ArrayList<Reference<? extends ImageWriter>>(); 52 53 int count = 2; 54 55 do { 56 ImageWriter writer = 57 ImageIO.getImageWritersByFormatName("jpeg").next(); 58 59 final WeakReference<? extends ImageWriter> ref = 60 new WeakReference<ImageWriter>(writer, queue); 61 62 refs.add(ref); 63 64 65 try { 66 final ImageOutputStream os = 67 ImageIO.createImageOutputStream(new ByteArrayOutputStream()); 68 writer.setOutput(os); 69 70 writer.write(getImage()); 71 72 73 // NB: dispose() or reset() workarounds the problem. 74 } catch (IOException e) { 75 } finally { 76 writer = null; 77 } 78 count--; 79 } while (count > 0); 80 81 82 System.out.println("Wait for GC..."); 83 84 final long testTimeOut = 60000L; 85 86 final long startTime = System.currentTimeMillis(); 87 88 while (!refs.isEmpty()) { 89 // check for the test timeout 90 final long now = System.currentTimeMillis(); 91 92 if (now - startTime > testTimeOut) { 93 System.out.println(); 94 throw new RuntimeException("Test FAILED."); 95 } 96 97 System.gc(); 98 99 try { 100 System.out.print("."); 101 Thread.sleep(1000); 102 } catch (InterruptedException e) { 103 }; 104 105 Reference<? extends ImageWriter> r = queue.poll(); 106 if (r != null) { 107 System.out.println("Got reference: " + r); 108 refs.remove(r); 109 } 110 } 111 System.out.println("Test PASSED."); 112 } 113 114 private static BufferedImage getImage() { 115 int width = 2500; 116 int height = new Random().nextInt(2500) + 1; 117 BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB); 118 119 Graphics2D g = image.createGraphics(); 120 g.setColor(Color.blue); 121 g.fillRect(0, 0, width, height); 122 123 return image; 124 } 125 }