[FFmpeg-devel] [PATCH] avcodec/rv34: Add check for av_malloc

[James Almer]

On 4/16/2023 11:32 PM, Jiasheng Jiang wrote:
> 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 at iscas.ac.cn>
> ---
>   libavcodec/rv34.c | 2 ++
>   1 file changed, 2 insertions(+)
> 
> diff --git a/libavcodec/rv34.c b/libavcodec/rv34.c
> index 45f64666f8..51f18147af 100644
> --- a/libavcodec/rv34.c
> +++ b/libavcodec/rv34.c
> @@ -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;

Applied, thanks.