avcodec/mjpegbdec: use init_get_bits8()

sandflow · Sep 21, 2022 · b9d71cd · b9d71cd
1 parent 6a150fc
commit b9d71cd
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/libavcodec/mjpegbdec.c b/libavcodec/mjpegbdec.c
@@ -64,10 +64,8 @@ static int mjpegb_decode_frame(AVCodecContext *avctx, AVFrame *rframe,
     s->restart_count = 0;
     s->mjpb_skiptosod = 0;
 
-    if (buf_end - buf_ptr >= 1 << 28)
-        return AVERROR_INVALIDDATA;
-
-    init_get_bits(&hgb, buf_ptr, /*buf_size*/(buf_end - buf_ptr)*8);
+    if ((ret = init_get_bits8(&hgb, buf_ptr, /*buf_size*/(buf_end - buf_ptr))) < 0)
+        return ret;
 
     skip_bits(&hgb, 32); /* reserved zeros */