[FFmpeg-cvslog] avcodec/cuviddec: check for av_buffer_ref() failure

[James Almer]

ffmpeg | branch: master | James Almer <jamrial at gmail.com> | Sun Dec 13 14:57:19 2020 -0300| [eadf7e3a59e89a8e26b28608a9617736b334ce85] | committer: James Almer

avcodec/cuviddec: check for av_buffer_ref() failure

Signed-off-by: James Almer <jamrial at gmail.com>

> http://git.videolan.org/gitweb.cgi/ffmpeg.git/?a=commit;h=eadf7e3a59e89a8e26b28608a9617736b334ce85
---

 libavcodec/cuviddec.c | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/libavcodec/cuviddec.c b/libavcodec/cuviddec.c
index 61d7f36c79..331851231f 100644
--- a/libavcodec/cuviddec.c
+++ b/libavcodec/cuviddec.c
@@ -553,6 +553,12 @@ static int cuvid_output_frame(AVCodecContext *avctx, AVFrame *frame)
 
             tmp_frame->format        = AV_PIX_FMT_CUDA;
             tmp_frame->hw_frames_ctx = av_buffer_ref(ctx->hwframe);
+            if (!tmp_frame->hw_frames_ctx) {
+                ret = AVERROR(ENOMEM);
+                av_frame_free(&tmp_frame);
+                goto error;
+            }
+
             tmp_frame->width         = avctx->width;
             tmp_frame->height        = avctx->height;