diff mbox series

[FFmpeg-devel,v1,09/10] return value check for init_get_bits in vaapi_mpeg2.c

Message ID PAXP193MB1262341838298629A7EDE419B6F99@PAXP193MB1262.EURP193.PROD.OUTLOOK.COM
State New
Headers show
Series [FFmpeg-devel,v1,01/10] return value check for init_get_bits in wmv2dec.c | expand

Checks

Context Check Description
andriy/x86_make fail Make failed
andriy/PPC64_make warning Make failed

Commit Message

Maryam Ebrahimzadeh Aug. 12, 2021, 4:54 a.m. UTC 
---
 libavcodec/vaapi_mpeg2.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)
diff mbox series

Patch

diff --git a/libavcodec/vaapi_mpeg2.c b/libavcodec/vaapi_mpeg2.c
index 26e0cd827c..8e3903212a 100644
--- a/libavcodec/vaapi_mpeg2.c
+++ b/libavcodec/vaapi_mpeg2.c
@@ -136,9 +136,12 @@  static int vaapi_mpeg2_decode_slice(AVCodecContext *avctx, const uint8_t *buffer
     GetBitContext gb;
     uint32_t quantiser_scale_code, intra_slice_flag, macroblock_offset;
     int err;
+    int ret;
 
     /* Determine macroblock_offset */
-    init_get_bits(&gb, buffer, 8 * size);
+    ret = init_get_bits8(&gb, buffer, size);
+    if (ret < 0 )
+        return ret;
     if (get_bits_long(&gb, 32) >> 8 != 1) /* start code */
         return AVERROR_INVALIDDATA;
     quantiser_scale_code = get_bits(&gb, 5);