Account for current pointer being outside of the buffer

Differential Revision: D65118896 Pull Request resolved: #316
pytorch · Nov 6, 2024 · 3c2c129 · 3c2c129
1 parent 8ac81b7
commit 3c2c129
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/torchcodec/decoders/_core/FFMPEGCommon.cpp b/src/torchcodec/decoders/_core/FFMPEGCommon.cpp
@@ -69,6 +69,12 @@ AVIOContext* AVIOBytesContext::getAVIO() {
 int AVIOBytesContext::read(void* opaque, uint8_t* buf, int buf_size) {
   struct AVIOBufferData* bufferData =
       static_cast<struct AVIOBufferData*>(opaque);
+  TORCH_CHECK(
+      bufferData->current <= bufferData->size,
+      "Tried to read outside of the buffer: current=",
+      bufferData->current,
+      ", size=",
+      bufferData->size);
   buf_size = FFMIN(buf_size, bufferData->size - bufferData->current);
   TORCH_CHECK(
       buf_size >= 0,