diff --git a/libavcodec/pngdec.c b/libavcodec/pngdec.c index aad07055ad..2e70f7f056 100644 --- a/libavcodec/pngdec.c +++ b/libavcodec/pngdec.c @@ -618,6 +618,8 @@ static int decode_idat_chunk(AVCodecContext *avctx, PNGDecContext *s, int ret; size_t byte_depth = s->bit_depth > 8 ? 2 : 1; + if (!p) + return AVERROR_INVALIDDATA; if (!(s->hdr_state & PNG_IHDR)) { av_log(avctx, AV_LOG_ERROR, "IDAT without IHDR\n"); return AVERROR_INVALIDDATA; @@ -1226,6 +1228,10 @@ skip_tag: } exit_loop: + if (!p) + return AVERROR_INVALIDDATA; + + if (s->bits_per_pixel <= 4) handle_small_bpp(s, p); @@ -1362,7 +1368,7 @@ static int decode_frame_apng(AVCodecContext *avctx, s->zstream.zfree = ff_png_zfree; bytestream2_init(&s->gb, avctx->extradata, avctx->extradata_size); - if ((ret = decode_frame_common(avctx, s, p, avpkt)) < 0) + if ((ret = decode_frame_common(avctx, s, NULL, avpkt)) < 0) goto end; }