Add assert to ensure loop termination

Signed-off-by: 胡玮文 <[email protected]>
AcademySoftwareFoundation · Jan 14, 2023 · ffdd48b · ffdd48b
1 parent 55bdfd1
commit ffdd48b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/lib/OpenEXR/ImfMisc.cpp b/src/lib/OpenEXR/ImfMisc.cpp
@@ -22,6 +22,7 @@
 #include <ImfStdIO.h>
 #include <ImfTileDescription.h>
 #include <ImfXdr.h>
+#include <cassert>
 
 OPENEXR_IMF_INTERNAL_NAMESPACE_SOURCE_ENTER
 
@@ -275,6 +276,10 @@ copyIntoFrameBuffer (
     //
     endPtr += xStride;
 
+    // Make sure the copy loop below will terminate.
+    assert ((endPtr - writePtr) / xStride > 0);
+    assert ((endPtr - writePtr) % xStride == 0);
+
     if (fill)
     {
         //
@@ -1533,6 +1538,11 @@ copyFromFrameBuffer (
     char*       localWritePtr = writePtr;
     const char* localReadPtr  = readPtr;
     endPtr += xStride;
+
+    // Make sure the copy loop below will terminate.
+    assert ((endPtr - localReadPtr) / xStride > 0);
+    assert ((endPtr - localReadPtr) % xStride == 0);
+
     //
     // Copy a horizontal row of pixels from a frame
     // buffer to an output file's line or tile buffer.