It (unfortunately) involves an allocation and can therefore fail.
Signed-off-by: Andreas Rheinhardt <[email protected]>
---
I don't have any nvidia hardware, so all these patches are untested
(apart from ensuring that they compile without creating new warnings).
libavcodec/nvdec.c | 7 +++++--
1 file changed, 5 insertions(+), 2 deletions(-)
diff --git a/libavcodec/nvdec.c b/libavcodec/nvdec.c
index edff46d310..15665b83bb 100644
--- a/libavcodec/nvdec.c
+++ b/libavcodec/nvdec.c
@@ -532,8 +532,11 @@ static int nvdec_retrieve_data(void *logctx, AVFrame
*frame)
}
unmap_data->idx = cf->idx;
- unmap_data->idx_ref = av_buffer_ref(cf->idx_ref);
- unmap_data->decoder_ref = av_buffer_ref(cf->decoder_ref);
+ if (!(unmap_data->idx_ref = av_buffer_ref(cf->idx_ref)) ||
+ !(unmap_data->decoder_ref = av_buffer_ref(cf->decoder_ref))) {
+ ret = AVERROR(ENOMEM);
+ goto copy_fail;
+ }
av_pix_fmt_get_chroma_sub_sample(hwctx->sw_format, &shift_h, &shift_v);
for (i = 0; frame->linesize[i]; i++) {
--
2.34.1
_______________________________________________
ffmpeg-devel mailing list
[email protected]
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
To unsubscribe, visit link above, or email
[email protected] with subject "unsubscribe".
