Fix crash on loading EXR 2.0 input

KhronosGroup · Jan 6, 2025 · ea1d50e · ea1d50e
1 parent 3038f7c
commit ea1d50e
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/tools/imageio/exr.imageio/exrinput.cc b/tools/imageio/exr.imageio/exrinput.cc
@@ -37,7 +37,9 @@ class ExrInput final : public ImageInput {
     }
     ~ExrInput() {
         // FreeEXRVersion(&version); // No need to call, no such function
-        FreeEXRImage(&image);
+        if (image.width != 0 && image.height != 0) {
+            FreeEXRImage(&image);
+        }
         FreeEXRHeader(&header);
         FreeEXRErrorMessage(err);
     }
@@ -223,6 +225,16 @@ void ExrInput::readImage(void* outputBuffer, size_t bufferByteCount,
                     "Requested format conversion from the input type is not supported."));
     }
 
+    // Load image version
+    EXRVersion exr_version;
+    ec = ParseEXRVersionFromMemory(&exr_version, exrBuffer.data(), exrBuffer.size());
+    if (ec != TINYEXR_SUCCESS)
+        throw std::runtime_error(
+            fmt::format("EXR load error: {} - {}.", ec, "Failed to parse EXR version"));
+    if (exr_version.multipart || exr_version.non_image)
+        throw std::runtime_error(
+            fmt::format("EXR load error: {}.", "Unsupported EXR version (2.0)"));
+
     // Load image data
     ec = LoadEXRImageFromMemory(&image, &header, exrBuffer.data(), exrBuffer.size(), &err);
     if (ec != TINYEXR_SUCCESS)