PDFBOX-5872: avoid NPE on image library / allow JPEG2000 library that…

… doesn't support raster, by GÃ¡bor Stefanik; closes #197 git-svn-id: https://svn.apache.org/repos/asf/pdfbox/trunk@1920175 13f79535-47bb-0310-9956-ffa450edef68
apache · Aug 25, 2024 · 107eaef · 107eaef
1 parent e39e166
commit 107eaef
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 6 deletions.
diff --git a/pdfbox/src/main/java/org/apache/pdfbox/filter/DCTFilter.java b/pdfbox/src/main/java/org/apache/pdfbox/filter/DCTFilter.java
@@ -59,7 +59,7 @@ final class DCTFilter extends Filter
     public DecodeResult decode(InputStream encoded, OutputStream decoded, COSDictionary
             parameters, int index, DecodeOptions options) throws IOException
     {
-        ImageReader reader = findImageReader("JPEG", "a suitable JAI I/O image filter is not installed");
+        ImageReader reader = findRasterReader("JPEG", "a suitable JAI I/O image filter is not installed");
         try (ImageInputStream iis = ImageIO.createImageInputStream(encoded))
         {
 

diff --git a/pdfbox/src/main/java/org/apache/pdfbox/filter/Filter.java b/pdfbox/src/main/java/org/apache/pdfbox/filter/Filter.java
@@ -149,7 +149,7 @@ else if (obj != null && !(filter instanceof COSArray || obj instanceof COSArray)
     }
 
     /**
-     * Finds a suitable image raster reader for an image format.
+     * Finds a suitable image reader for an image format.
      *
      * @param formatName The image format to search for.
      * @param errorCause The probably cause if something goes wrong.
@@ -163,11 +163,37 @@ public static final ImageReader findImageReader(String formatName, String errorC
         while (readers.hasNext())
         {
             ImageReader reader = readers.next();
-            if (reader.canReadRaster())
+            if (reader != null)
             {
                 return reader;
             }
-            reader.dispose();
+        }
+        throw new MissingImageReaderException("Cannot read " + formatName + " image: " + errorCause);
+    }
+
+    /**
+     * Finds a suitable image raster reader for an image format.
+     *
+     * @param formatName The image format to search for.
+     * @param errorCause The probably cause if something goes wrong.
+     * @return The image reader for the format.
+     * @throws MissingImageReaderException if no image reader is found.
+     */
+    public static final ImageReader findRasterReader(String formatName, String errorCause)
+            throws MissingImageReaderException
+    {
+        Iterator<ImageReader> readers = ImageIO.getImageReadersByFormatName(formatName);
+        while (readers.hasNext())
+        {
+            ImageReader reader = readers.next();
+            if (reader != null)
+            {
+                if (reader.canReadRaster())
+                {
+                    return reader;
+                }
+                reader.dispose();
+            }
         }
         throw new MissingImageReaderException("Cannot read " + formatName + " image: " + errorCause);
     }

diff --git a/pdfbox/src/main/java/org/apache/pdfbox/pdmodel/graphics/image/JPEGFactory.java b/pdfbox/src/main/java/org/apache/pdfbox/pdmodel/graphics/image/JPEGFactory.java
@@ -148,7 +148,7 @@ private static class Dimensions
     private static Dimensions retrieveDimensions(ByteArrayInputStream stream) throws IOException
     {
         ImageReader reader =
-                Filter.findImageReader("JPEG", "a suitable JAI I/O image filter is not installed");
+                Filter.findRasterReader("JPEG", "a suitable JAI I/O image filter is not installed");
         try (ImageInputStream iis = ImageIO.createImageInputStream(stream))
         {
             reader.setInput(iis);
@@ -169,7 +169,7 @@ private static Dimensions retrieveDimensions(ByteArrayInputStream stream) throws
             catch (IOException ex)
             {
                 LOG.warn("Error reading image metadata, will decode image and use raster size");
-            }            
+            }
 
             // Old method: get from raster (slower)
             ImageIO.setUseCache(false);