533 lines
20 KiB
Diff
533 lines
20 KiB
Diff
|
From 2f4e49b95baaecaae86f93c9ab803500db48c4cc Mon Sep 17 00:00:00 2001
|
||
|
From: Kimball Thurston <kdt3rd@gmail.com>
|
||
|
Date: Sat, 3 Feb 2024 20:00:53 +1300
|
||
|
Subject: [PATCH 1/4] enable deep file checks for core
|
||
|
|
||
|
Signed-off-by: Kimball Thurston <kdt3rd@gmail.com>
|
||
|
---
|
||
|
src/lib/OpenEXRUtil/ImfCheckFile.cpp | 181 ++++++++++++++++++++++-----
|
||
|
1 file changed, 148 insertions(+), 33 deletions(-)
|
||
|
|
||
|
diff --git a/src/lib/OpenEXRUtil/ImfCheckFile.cpp b/src/lib/OpenEXRUtil/ImfCheckFile.cpp
|
||
|
index 8c4fbe84..8946e88d 100644
|
||
|
--- a/src/lib/OpenEXRUtil/ImfCheckFile.cpp
|
||
|
+++ b/src/lib/OpenEXRUtil/ImfCheckFile.cpp
|
||
|
@@ -1172,11 +1172,83 @@ runChecks(T& source,bool reduceMemory,bool reduceTime)
|
||
|
return threw;
|
||
|
}
|
||
|
|
||
|
+static exr_result_t
|
||
|
+realloc_deepdata(exr_decode_pipeline_t* decode)
|
||
|
+{
|
||
|
+ int32_t w = decode->chunk.width;
|
||
|
+ int32_t h = decode->chunk.height;
|
||
|
+ uint64_t totsamps = 0, bytes = 0;
|
||
|
+ const int32_t *sampbuffer = decode->sample_count_table;
|
||
|
+ std::vector<uint8_t>* ud = static_cast<std::vector<uint8_t>*>(
|
||
|
+ decode->decoding_user_data);
|
||
|
+
|
||
|
+ if ( ! ud )
|
||
|
+ {
|
||
|
+ for (int c = 0; c < decode->channel_count; c++)
|
||
|
+ {
|
||
|
+ exr_coding_channel_info_t& outc = decode->channels[c];
|
||
|
+ outc.decode_to_ptr = NULL;
|
||
|
+ outc.user_pixel_stride = outc.user_bytes_per_element;
|
||
|
+ outc.user_line_stride = 0;
|
||
|
+ }
|
||
|
+ return EXR_ERR_SUCCESS;
|
||
|
+ }
|
||
|
+
|
||
|
+ if ((decode->decode_flags &
|
||
|
+ EXR_DECODE_SAMPLE_COUNTS_AS_INDIVIDUAL))
|
||
|
+ {
|
||
|
+ for (int32_t y = 0; y < h; ++y)
|
||
|
+ {
|
||
|
+ for (int x = 0; x < w; ++x)
|
||
|
+ totsamps += sampbuffer[x];
|
||
|
+ sampbuffer += w;
|
||
|
+ }
|
||
|
+ }
|
||
|
+ else
|
||
|
+ {
|
||
|
+ for (int32_t y = 0; y < h; ++y)
|
||
|
+ totsamps += sampbuffer[y*w + w - 1];
|
||
|
+ }
|
||
|
+
|
||
|
+ for (int c = 0; c < decode->channel_count; c++)
|
||
|
+ {
|
||
|
+ exr_coding_channel_info_t& outc = decode->channels[c];
|
||
|
+ bytes += totsamps * outc.user_bytes_per_element;
|
||
|
+ }
|
||
|
+
|
||
|
+ if (bytes >= gMaxBytesPerDeepScanline * h)
|
||
|
+ {
|
||
|
+ for (int c = 0; c < decode->channel_count; c++)
|
||
|
+ {
|
||
|
+ exr_coding_channel_info_t& outc = decode->channels[c];
|
||
|
+ outc.decode_to_ptr = NULL;
|
||
|
+ outc.user_pixel_stride = outc.user_bytes_per_element;
|
||
|
+ outc.user_line_stride = 0;
|
||
|
+ }
|
||
|
+ return EXR_ERR_SUCCESS;
|
||
|
+ }
|
||
|
+
|
||
|
+ if (ud->size () < bytes)
|
||
|
+ ud->resize (bytes);
|
||
|
+
|
||
|
+ uint8_t* dptr = &((*ud)[0]);
|
||
|
+ for (int c = 0; c < decode->channel_count; c++)
|
||
|
+ {
|
||
|
+ exr_coding_channel_info_t& outc = decode->channels[c];
|
||
|
+ outc.decode_to_ptr = dptr;
|
||
|
+ outc.user_pixel_stride = outc.user_bytes_per_element;
|
||
|
+ outc.user_line_stride = 0;
|
||
|
+
|
||
|
+ dptr += totsamps * (uint64_t) outc.user_bytes_per_element;
|
||
|
+ }
|
||
|
+ return EXR_ERR_SUCCESS;
|
||
|
+}
|
||
|
+
|
||
|
////////////////////////////////////////
|
||
|
|
||
|
bool readCoreScanlinePart(exr_context_t f, int part, bool reduceMemory, bool reduceTime)
|
||
|
{
|
||
|
- exr_result_t rv;
|
||
|
+ exr_result_t rv, frv;
|
||
|
exr_attr_box2i_t datawin;
|
||
|
rv = exr_get_data_window (f, part, &datawin);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
@@ -1194,6 +1266,8 @@ bool readCoreScanlinePart(exr_context_t f, int part, bool reduceMemory, bool red
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
return true;
|
||
|
|
||
|
+ frv = rv;
|
||
|
+
|
||
|
for (uint64_t chunk = 0; chunk < height; chunk += lines_per_chunk)
|
||
|
{
|
||
|
exr_chunk_info_t cinfo = { 0 };
|
||
|
@@ -1202,6 +1276,7 @@ bool readCoreScanlinePart(exr_context_t f, int part, bool reduceMemory, bool red
|
||
|
rv = exr_read_scanline_chunk_info (f, part, y, &cinfo);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
if (reduceTime)
|
||
|
break;
|
||
|
continue;
|
||
|
@@ -1224,22 +1299,33 @@ bool readCoreScanlinePart(exr_context_t f, int part, bool reduceMemory, bool red
|
||
|
bytes += width * (uint64_t)outc.user_bytes_per_element * (uint64_t)lines_per_chunk;
|
||
|
}
|
||
|
|
||
|
- // TODO: check we are supposed to multiple by lines per chunk above
|
||
|
doread = true;
|
||
|
- if (reduceMemory && bytes >= gMaxBytesPerScanline)
|
||
|
- doread = false;
|
||
|
+ if (cinfo.type == EXR_STORAGE_DEEP_SCANLINE)
|
||
|
+ {
|
||
|
+ decoder.decoding_user_data = &imgdata;
|
||
|
+ decoder.realloc_nonimage_data_fn = &realloc_deepdata;
|
||
|
+ }
|
||
|
+ else
|
||
|
+ {
|
||
|
+ if (reduceMemory && bytes >= gMaxBytesPerScanline)
|
||
|
+ doread = false;
|
||
|
|
||
|
- if (doread)
|
||
|
- imgdata.resize( bytes );
|
||
|
+ if (doread)
|
||
|
+ imgdata.resize (bytes);
|
||
|
+ }
|
||
|
rv = exr_decoding_choose_default_routines (f, part, &decoder);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
+ {
|
||
|
+ frv = rv;
|
||
|
break;
|
||
|
+ }
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
rv = exr_decoding_update (f, part, &cinfo, &decoder);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
if (reduceTime)
|
||
|
break;
|
||
|
continue;
|
||
|
@@ -1248,19 +1334,25 @@ bool readCoreScanlinePart(exr_context_t f, int part, bool reduceMemory, bool red
|
||
|
|
||
|
if (doread)
|
||
|
{
|
||
|
- uint8_t *dptr = &(imgdata[0]);
|
||
|
- for (int c = 0; c < decoder.channel_count; c++)
|
||
|
+ if (cinfo.type != EXR_STORAGE_DEEP_SCANLINE)
|
||
|
{
|
||
|
- exr_coding_channel_info_t & outc = decoder.channels[c];
|
||
|
- outc.decode_to_ptr = dptr;
|
||
|
- outc.user_pixel_stride = outc.user_bytes_per_element;
|
||
|
- outc.user_line_stride = outc.user_pixel_stride * width;
|
||
|
- dptr += width * (uint64_t)outc.user_bytes_per_element * (uint64_t)lines_per_chunk;
|
||
|
+ uint8_t* dptr = &(imgdata[0]);
|
||
|
+ for (int c = 0; c < decoder.channel_count; c++)
|
||
|
+ {
|
||
|
+ exr_coding_channel_info_t& outc = decoder.channels[c];
|
||
|
+ outc.decode_to_ptr = dptr;
|
||
|
+ outc.user_pixel_stride = outc.user_bytes_per_element;
|
||
|
+ outc.user_line_stride = outc.user_pixel_stride * width;
|
||
|
+
|
||
|
+ dptr += width * (uint64_t) outc.user_bytes_per_element *
|
||
|
+ (uint64_t) lines_per_chunk;
|
||
|
+ }
|
||
|
}
|
||
|
|
||
|
rv = exr_decoding_run (f, part, &decoder);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
if (reduceTime)
|
||
|
break;
|
||
|
}
|
||
|
@@ -1269,14 +1361,14 @@ bool readCoreScanlinePart(exr_context_t f, int part, bool reduceMemory, bool red
|
||
|
|
||
|
exr_decoding_destroy (f, &decoder);
|
||
|
|
||
|
- return (rv != EXR_ERR_SUCCESS);
|
||
|
+ return (frv != EXR_ERR_SUCCESS);
|
||
|
}
|
||
|
|
||
|
////////////////////////////////////////
|
||
|
|
||
|
bool readCoreTiledPart(exr_context_t f, int part, bool reduceMemory, bool reduceTime)
|
||
|
{
|
||
|
- exr_result_t rv;
|
||
|
+ exr_result_t rv, frv;
|
||
|
|
||
|
exr_attr_box2i_t datawin;
|
||
|
rv = exr_get_data_window (f, part, &datawin);
|
||
|
@@ -1296,6 +1388,7 @@ bool readCoreTiledPart(exr_context_t f, int part, bool reduceMemory, bool reduce
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
return true;
|
||
|
|
||
|
+ frv = rv;
|
||
|
bool keepgoing = true;
|
||
|
for (int32_t ylevel = 0; keepgoing && ylevel < levelsy; ++ylevel )
|
||
|
{
|
||
|
@@ -1305,6 +1398,7 @@ bool readCoreTiledPart(exr_context_t f, int part, bool reduceMemory, bool reduce
|
||
|
rv = exr_get_level_sizes (f, part, xlevel, ylevel, &levw, &levh);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
if (reduceTime)
|
||
|
{
|
||
|
keepgoing = false;
|
||
|
@@ -1317,6 +1411,7 @@ bool readCoreTiledPart(exr_context_t f, int part, bool reduceMemory, bool reduce
|
||
|
rv = exr_get_tile_sizes (f, part, xlevel, ylevel, &curtw, &curth);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
if (reduceTime)
|
||
|
{
|
||
|
keepgoing = false;
|
||
|
@@ -1343,6 +1438,7 @@ bool readCoreTiledPart(exr_context_t f, int part, bool reduceMemory, bool reduce
|
||
|
rv = exr_read_tile_chunk_info (f, part, tx, ty, xlevel, ylevel, &cinfo);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
if (reduceTime)
|
||
|
{
|
||
|
keepgoing = false;
|
||
|
@@ -1356,6 +1452,7 @@ bool readCoreTiledPart(exr_context_t f, int part, bool reduceMemory, bool reduce
|
||
|
rv = exr_decoding_initialize (f, part, &cinfo, &decoder);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
keepgoing = false;
|
||
|
break;
|
||
|
}
|
||
|
@@ -1372,14 +1469,24 @@ bool readCoreTiledPart(exr_context_t f, int part, bool reduceMemory, bool reduce
|
||
|
}
|
||
|
|
||
|
doread = true;
|
||
|
- if (reduceMemory && bytes >= gMaxTileBytes)
|
||
|
- doread = false;
|
||
|
+ if (cinfo.type == EXR_STORAGE_DEEP_TILED)
|
||
|
+ {
|
||
|
+ decoder.decoding_user_data = &tiledata;
|
||
|
+ decoder.realloc_nonimage_data_fn = &realloc_deepdata;
|
||
|
+ }
|
||
|
+ else
|
||
|
+ {
|
||
|
+ if (reduceMemory && bytes >= gMaxTileBytes)
|
||
|
+ doread = false;
|
||
|
|
||
|
- if (doread)
|
||
|
- tiledata.resize( bytes );
|
||
|
- rv = exr_decoding_choose_default_routines (f, part, &decoder);
|
||
|
+ if (doread)
|
||
|
+ tiledata.resize (bytes);
|
||
|
+ }
|
||
|
+ rv = exr_decoding_choose_default_routines (
|
||
|
+ f, part, &decoder);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
keepgoing = false;
|
||
|
break;
|
||
|
}
|
||
|
@@ -1389,6 +1496,7 @@ bool readCoreTiledPart(exr_context_t f, int part, bool reduceMemory, bool reduce
|
||
|
rv = exr_decoding_update (f, part, &cinfo, &decoder);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
if (reduceTime)
|
||
|
{
|
||
|
keepgoing = false;
|
||
|
@@ -1400,19 +1508,28 @@ bool readCoreTiledPart(exr_context_t f, int part, bool reduceMemory, bool reduce
|
||
|
|
||
|
if (doread)
|
||
|
{
|
||
|
- uint8_t *dptr = &(tiledata[0]);
|
||
|
- for (int c = 0; c < decoder.channel_count; c++)
|
||
|
+ if (cinfo.type != EXR_STORAGE_DEEP_TILED)
|
||
|
{
|
||
|
- exr_coding_channel_info_t & outc = decoder.channels[c];
|
||
|
- outc.decode_to_ptr = dptr;
|
||
|
- outc.user_pixel_stride = outc.user_bytes_per_element;
|
||
|
- outc.user_line_stride = outc.user_pixel_stride * curtw;
|
||
|
- dptr += (uint64_t)curtw * (uint64_t)outc.user_bytes_per_element * (uint64_t)curth;
|
||
|
+ uint8_t* dptr = &(tiledata[0]);
|
||
|
+ for (int c = 0; c < decoder.channel_count; c++)
|
||
|
+ {
|
||
|
+ exr_coding_channel_info_t& outc =
|
||
|
+ decoder.channels[c];
|
||
|
+ outc.decode_to_ptr = dptr;
|
||
|
+ outc.user_pixel_stride =
|
||
|
+ outc.user_bytes_per_element;
|
||
|
+ outc.user_line_stride =
|
||
|
+ outc.user_pixel_stride * curtw;
|
||
|
+ dptr += (uint64_t) curtw *
|
||
|
+ (uint64_t) outc.user_bytes_per_element *
|
||
|
+ (uint64_t) curth;
|
||
|
+ }
|
||
|
}
|
||
|
|
||
|
rv = exr_decoding_run (f, part, &decoder);
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
{
|
||
|
+ frv = rv;
|
||
|
if (reduceTime)
|
||
|
{
|
||
|
keepgoing = false;
|
||
|
@@ -1448,16 +1565,14 @@ bool checkCoreFile(exr_context_t f, bool reduceMemory, bool reduceTime)
|
||
|
if (rv != EXR_ERR_SUCCESS)
|
||
|
return true;
|
||
|
|
||
|
- // TODO: Need to fill this in
|
||
|
- if (store == EXR_STORAGE_DEEP_SCANLINE || store == EXR_STORAGE_DEEP_TILED)
|
||
|
- continue;
|
||
|
-
|
||
|
- if (store == EXR_STORAGE_SCANLINE)
|
||
|
+ if (store == EXR_STORAGE_SCANLINE ||
|
||
|
+ store == EXR_STORAGE_DEEP_SCANLINE)
|
||
|
{
|
||
|
if ( readCoreScanlinePart (f, p, reduceMemory, reduceTime) )
|
||
|
return true;
|
||
|
}
|
||
|
- else if (store == EXR_STORAGE_TILED)
|
||
|
+ else if (store == EXR_STORAGE_TILED ||
|
||
|
+ store == EXR_STORAGE_DEEP_TILED)
|
||
|
{
|
||
|
if ( readCoreTiledPart (f, p, reduceMemory, reduceTime) )
|
||
|
return true;
|
||
|
--
|
||
|
2.43.0
|
||
|
|
||
|
|
||
|
From 166ace3ef06a2a4fc16bb58f42204629c9273798 Mon Sep 17 00:00:00 2001
|
||
|
From: Kimball Thurston <kdt3rd@gmail.com>
|
||
|
Date: Sat, 3 Feb 2024 20:01:33 +1300
|
||
|
Subject: [PATCH 2/4] fix possible int overflow
|
||
|
|
||
|
Signed-off-by: Kimball Thurston <kdt3rd@gmail.com>
|
||
|
---
|
||
|
src/lib/OpenEXRCore/unpack.c | 9 ++++++---
|
||
|
1 file changed, 6 insertions(+), 3 deletions(-)
|
||
|
|
||
|
diff --git a/src/lib/OpenEXRCore/unpack.c b/src/lib/OpenEXRCore/unpack.c
|
||
|
index 9ecb729c..80990e0d 100644
|
||
|
--- a/src/lib/OpenEXRCore/unpack.c
|
||
|
+++ b/src/lib/OpenEXRCore/unpack.c
|
||
|
@@ -1205,9 +1205,10 @@ generic_unpack_deep_pointers (exr_decode_pipeline_t* decode)
|
||
|
if (outpix)
|
||
|
{
|
||
|
uint8_t* cdata = outpix;
|
||
|
+
|
||
|
UNPACK_SAMPLES (samps)
|
||
|
}
|
||
|
- srcbuffer += bpc * samps;
|
||
|
+ srcbuffer += ((size_t) bpc) * ((size_t) samps);
|
||
|
}
|
||
|
}
|
||
|
sampbuffer += w;
|
||
|
@@ -1251,12 +1252,14 @@ generic_unpack_deep (exr_decode_pipeline_t* decode)
|
||
|
}
|
||
|
else
|
||
|
prevsamps = sampbuffer[w - 1];
|
||
|
+
|
||
|
srcbuffer += ((size_t) bpc) * ((size_t) prevsamps);
|
||
|
|
||
|
if (incr_tot) totsamps += (size_t) prevsamps;
|
||
|
|
||
|
continue;
|
||
|
}
|
||
|
+
|
||
|
cdata += totsamps * ((size_t) ubpc);
|
||
|
|
||
|
for (int x = 0; x < w; ++x)
|
||
|
@@ -1272,7 +1275,7 @@ generic_unpack_deep (exr_decode_pipeline_t* decode)
|
||
|
|
||
|
UNPACK_SAMPLES (samps)
|
||
|
|
||
|
- srcbuffer += bpc * samps;
|
||
|
+ srcbuffer += ((size_t) bpc) * ((size_t) samps);
|
||
|
if (incr_tot) totsamps += (size_t) samps;
|
||
|
}
|
||
|
}
|
||
|
@@ -1310,7 +1313,7 @@ internal_exr_match_decode (
|
||
|
|
||
|
if (isdeep)
|
||
|
{
|
||
|
- if ((decode->decode_flags & EXR_DECODE_SAMPLE_COUNTS_AS_INDIVIDUAL))
|
||
|
+ if ((decode->decode_flags & EXR_DECODE_NON_IMAGE_DATA_AS_POINTERS))
|
||
|
return &generic_unpack_deep_pointers;
|
||
|
return &generic_unpack_deep;
|
||
|
}
|
||
|
--
|
||
|
2.43.0
|
||
|
|
||
|
|
||
|
From d1cea15375da4da642defa91f7f4b78e251af95d Mon Sep 17 00:00:00 2001
|
||
|
From: Kimball Thurston <kdt3rd@gmail.com>
|
||
|
Date: Sat, 3 Feb 2024 20:04:16 +1300
|
||
|
Subject: [PATCH 3/4] fix validation of deep sample counts
|
||
|
|
||
|
Addresses CVE-2023-5841, fixing sample count check to not only check
|
||
|
against 0 but previous sample as well.
|
||
|
|
||
|
Signed-off-by: Kimball Thurston <kdt3rd@gmail.com>
|
||
|
---
|
||
|
src/lib/OpenEXRCore/decoding.c | 37 +++++++++++++++++++++-------------
|
||
|
1 file changed, 23 insertions(+), 14 deletions(-)
|
||
|
|
||
|
diff --git a/src/lib/OpenEXRCore/decoding.c b/src/lib/OpenEXRCore/decoding.c
|
||
|
index e9e8f7f9..65f0f2f8 100644
|
||
|
--- a/src/lib/OpenEXRCore/decoding.c
|
||
|
+++ b/src/lib/OpenEXRCore/decoding.c
|
||
|
@@ -290,6 +290,9 @@ default_decompress_chunk (exr_decode_pipeline_t* decode)
|
||
|
{
|
||
|
uint64_t sampsize =
|
||
|
(((uint64_t) decode->chunk.width) * ((uint64_t) decode->chunk.height));
|
||
|
+
|
||
|
+ if ((decode->decode_flags & EXR_DECODE_SAMPLE_COUNTS_AS_INDIVIDUAL))
|
||
|
+ sampsize += 1;
|
||
|
sampsize *= sizeof (int32_t);
|
||
|
|
||
|
rv = decompress_data (
|
||
|
@@ -342,7 +345,7 @@ unpack_sample_table (
|
||
|
exr_result_t rv = EXR_ERR_SUCCESS;
|
||
|
int32_t w = decode->chunk.width;
|
||
|
int32_t h = decode->chunk.height;
|
||
|
- int32_t totsamp = 0;
|
||
|
+ uint64_t totsamp = 0;
|
||
|
int32_t* samptable = decode->sample_count_table;
|
||
|
size_t combSampSize = 0;
|
||
|
|
||
|
@@ -353,38 +356,44 @@ unpack_sample_table (
|
||
|
{
|
||
|
for (int32_t y = 0; y < h; ++y)
|
||
|
{
|
||
|
+ int32_t *cursampline = samptable + y * w;
|
||
|
int32_t prevsamp = 0;
|
||
|
for (int32_t x = 0; x < w; ++x)
|
||
|
{
|
||
|
int32_t nsamps =
|
||
|
- (int32_t) one_to_native32 ((uint32_t) samptable[y * w + x]);
|
||
|
- if (nsamps < 0) return EXR_ERR_INVALID_SAMPLE_DATA;
|
||
|
- samptable[y * w + x] = nsamps - prevsamp;
|
||
|
- prevsamp = nsamps;
|
||
|
+ (int32_t) one_to_native32 ((uint32_t) cursampline[x]);
|
||
|
+ if (nsamps < prevsamp) return EXR_ERR_INVALID_SAMPLE_DATA;
|
||
|
+
|
||
|
+ cursampline[x] = nsamps - prevsamp;
|
||
|
+ prevsamp = nsamps;
|
||
|
}
|
||
|
- totsamp += prevsamp;
|
||
|
+ totsamp += (uint64_t)prevsamp;
|
||
|
}
|
||
|
- samptable[w * h] = totsamp;
|
||
|
+ if (totsamp >= (uint64_t)INT32_MAX)
|
||
|
+ return EXR_ERR_INVALID_SAMPLE_DATA;
|
||
|
+ samptable[w * h] = (int32_t)totsamp;
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
for (int32_t y = 0; y < h; ++y)
|
||
|
{
|
||
|
+ int32_t *cursampline = samptable + y * w;
|
||
|
int32_t prevsamp = 0;
|
||
|
for (int32_t x = 0; x < w; ++x)
|
||
|
{
|
||
|
int32_t nsamps =
|
||
|
- (int32_t) one_to_native32 ((uint32_t) samptable[y * w + x]);
|
||
|
- if (nsamps < 0) return EXR_ERR_INVALID_SAMPLE_DATA;
|
||
|
- samptable[y * w + x] = nsamps;
|
||
|
- prevsamp = nsamps;
|
||
|
+ (int32_t) one_to_native32 ((uint32_t) cursampline[x]);
|
||
|
+ if (nsamps < prevsamp) return EXR_ERR_INVALID_SAMPLE_DATA;
|
||
|
+
|
||
|
+ cursampline[x] = nsamps;
|
||
|
+ prevsamp = nsamps;
|
||
|
}
|
||
|
- totsamp += prevsamp;
|
||
|
+
|
||
|
+ totsamp += (uint64_t)prevsamp;
|
||
|
}
|
||
|
}
|
||
|
|
||
|
- if (totsamp < 0 ||
|
||
|
- (((uint64_t) totsamp) * combSampSize) > decode->chunk.unpacked_size)
|
||
|
+ if ((totsamp * combSampSize) > decode->chunk.unpacked_size)
|
||
|
{
|
||
|
rv = pctxt->report_error (
|
||
|
pctxt, EXR_ERR_INVALID_SAMPLE_DATA, "Corrupt sample count table");
|
||
|
--
|
||
|
2.43.0
|
||
|
|
||
|
|
||
|
From 9ad0e7c8a271df1d7a2ae4722fb5f0d48d191b62 Mon Sep 17 00:00:00 2001
|
||
|
From: Kimball Thurston <kdt3rd@gmail.com>
|
||
|
Date: Sat, 3 Feb 2024 20:07:49 +1300
|
||
|
Subject: [PATCH 4/4] add clarifying comment
|
||
|
|
||
|
Signed-off-by: Kimball Thurston <kdt3rd@gmail.com>
|
||
|
---
|
||
|
src/lib/OpenEXRUtil/ImfCheckFile.cpp | 3 +++
|
||
|
1 file changed, 3 insertions(+)
|
||
|
|
||
|
diff --git a/src/lib/OpenEXRUtil/ImfCheckFile.cpp b/src/lib/OpenEXRUtil/ImfCheckFile.cpp
|
||
|
index 8946e88d..0058f627 100644
|
||
|
--- a/src/lib/OpenEXRUtil/ImfCheckFile.cpp
|
||
|
+++ b/src/lib/OpenEXRUtil/ImfCheckFile.cpp
|
||
|
@@ -1172,6 +1172,9 @@ runChecks(T& source,bool reduceMemory,bool reduceTime)
|
||
|
return threw;
|
||
|
}
|
||
|
|
||
|
+// This is not entirely needed in that the chunk info has the
|
||
|
+// total unpacked_size field which can be used for allocation
|
||
|
+// but this adds an additional point to use when debugging issues.
|
||
|
static exr_result_t
|
||
|
realloc_deepdata(exr_decode_pipeline_t* decode)
|
||
|
{
|
||
|
--
|
||
|
2.43.0
|
||
|
|