avcodec/msvideo1enc: Add check for av_malloc

Add the check for the return value of the av_malloc in order to avoid
NULL pointer deference.

Fixes: e4be3485af ("MS Video 1 encoder")
Signed-off-by: Jiasheng Jiang <jiasheng@iscas.ac.cn>
This commit is contained in:
Jiasheng Jiang 2023-04-17 10:32:51 +08:00 committed by James Almer
parent f8d0b13de5
commit b9ba036680
1 changed files with 2 additions and 0 deletions

View File

@ -84,6 +84,8 @@ static int encode_frame(AVCodecContext *avctx, AVPacket *pkt,
if(!c->prev)
c->prev = av_malloc(avctx->width * 3 * (avctx->height + 3));
if (!c->prev)
return AVERROR(ENOMEM);
prevptr = c->prev + avctx->width * 3 * (FFALIGN(avctx->height, 4) - 1);
src = (const uint16_t*)(p->data[0] + p->linesize[0]*(FFALIGN(avctx->height, 4) - 1));
if(c->keyint >= avctx->keyint_min)