1/*
2 * Copyright (c) 2001, 2017, 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 4420318 8183341
27 * @summary Checks that an IllegalStateException is thrown by getNumImages(true)
28 *          when seekForwardOnly is true
29 * @modules java.desktop/com.sun.imageio.plugins.gif
30 *          java.desktop/com.sun.imageio.plugins.jpeg
31 *          java.desktop/com.sun.imageio.plugins.png
32 */
33
34import java.io.File;
35import java.io.IOException;
36import java.nio.file.Files;
37
38import javax.imageio.ImageIO;
39import javax.imageio.ImageReader;
40import javax.imageio.stream.ImageInputStream;
41
42import com.sun.imageio.plugins.gif.GIFImageReader;
43import com.sun.imageio.plugins.jpeg.JPEGImageReader;
44import com.sun.imageio.plugins.png.PNGImageReader;
45
46public class AllowSearch {
47    private static void test(ImageReader reader, String format)
48        throws IOException {
49        boolean gotISE = false;
50        File f = null;
51        ImageInputStream stream = null;
52        try {
53            f = File.createTempFile("imageio", ".tmp");
54            stream = ImageIO.createImageInputStream(f);
55            reader.setInput(stream, true);
56
57            try {
58                int numImages = reader.getNumImages(true);
59            } catch (IOException ioe) {
60                gotISE = false;
61            } catch (IllegalStateException ise) {
62                gotISE = true;
63            }
64        } finally {
65            if (stream != null) {
66                stream.close();
67            }
68
69            reader.dispose();
70
71            if (f != null) {
72                Files.delete(f.toPath());
73            }
74        }
75
76        if (!gotISE) {
77            throw new RuntimeException("Failed to get desired exception for " +
78                                       format + " reader!");
79        }
80    }
81
82    public static void main(String[] args) throws IOException {
83        ImageReader gifReader = new GIFImageReader(null);
84        ImageReader jpegReader = new JPEGImageReader(null);
85        ImageReader pngReader = new PNGImageReader(null);
86
87        test(gifReader, "GIF");
88        test(jpegReader, "JPEG");
89        test(pngReader, "PNG");
90    }
91}
92