avcodec/rv34: Add check for av_malloc

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

Fixes: b86ab38137 ("Add weighted motion compensation for RV40 B-frames")
Signed-off-by: Jiasheng Jiang <jiasheng@iscas.ac.cn>
This commit is contained in:
Jiasheng Jiang 2023-04-17 10:32:12 +08:00 committed by James Almer
parent a3f49168dc
commit f8d0b13de5
1 changed files with 2 additions and 0 deletions

View File

@ -1692,6 +1692,8 @@ int ff_rv34_decode_frame(AVCodecContext *avctx, AVFrame *pict,
int i;
r->tmp_b_block_base = av_malloc(s->linesize * 48);
if (!r->tmp_b_block_base)
return AVERROR(ENOMEM);
for (i = 0; i < 2; i++)
r->tmp_b_block_y[i] = r->tmp_b_block_base
+ i * 16 * s->linesize;