[Decode]Fix mpeg2 PSIRT issue (#5555)

Co-authored-by: Wu, Stella <[email protected]>
intel · Jul 27, 2023 · 15c8f90 · 15c8f90
1 parent a9fb0bc
commit 15c8f90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/_studio/shared/umc/codec/mpeg2_dec/src/umc_mpeg2_decoder.cpp b/_studio/shared/umc/codec/mpeg2_dec/src/umc_mpeg2_decoder.cpp
@@ -587,7 +587,7 @@ namespace UMC_MPEG2_DECODER
             uint8_t* picExtBegin = RawHeaderIterator::FindStartCode(data.begin + prefix_size + bitStream.BytesDecoded(), data.end); // Find begining of extension
             MFX_CHECK(picExtBegin, UMC::UMC_ERR_INVALID_STREAM);
 
-            bitStream.Reset(picExtBegin + prefix_size, (uint32_t)(data.end - picExtBegin));
+            bitStream.Reset(picExtBegin + prefix_size, (uint32_t)(data.end - picExtBegin - prefix_size + 1));
             bitStream.Seek(8 + 4); // skip data and extension types
             bitStream.GetPictureExtensionHeader(*picExtHdr.get()); // decode extension