[FFmpeg-devel] [PATCH v2 24/27] avcodec/rkmppdec: Check av_buffer_ref()

[Andreas Rheinhardt]

Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt at outlook.com>
---
 libavcodec/rkmppdec.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/libavcodec/rkmppdec.c b/libavcodec/rkmppdec.c
index 6889545b20..4e14d09c7c 100644
--- a/libavcodec/rkmppdec.c
+++ b/libavcodec/rkmppdec.c
@@ -450,6 +450,10 @@ static int rkmpp_retrieve_frame(AVCodecContext *avctx, AVFrame *frame)
             // MPP decoder needs to be closed only when all frames have been released.
             framecontext = (RKMPPFrameContext *)framecontextref->data;
             framecontext->decoder_ref = av_buffer_ref(rk_context->decoder_ref);
+            if (!framecontext->decoder_ref) {
+                ret = AVERROR(ENOMEM);
+                goto fail;
+            }
             framecontext->frame = mppframe;
 
             frame->data[0]  = (uint8_t *)desc;
-- 
2.40.1