har0ke/FFmpeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

avfilter/overlay_cuda: check av_buffer_ref result

Browse Source
This commit is contained in:
Timo Rothenpieler 2021-04-11 10:15:47 +02:00
parent 0fcaa97739
commit 85713d85a7
1 changed files with 4 additions and 0 deletions

4
libavfilter/vf_overlay_cuda.c

@ -346,6 +346,8 @@ static int overlay_cuda_config_output(AVFilterLink *outlink)
// initialize // initialize
ctx->hw_device_ctx = av_buffer_ref(frames_ctx->device_ref); ctx->hw_device_ctx = av_buffer_ref(frames_ctx->device_ref);
if (!ctx->hw_device_ctx)
return AVERROR(ENOMEM);
ctx->hwctx = ((AVHWDeviceContext*)ctx->hw_device_ctx->data)->hwctx; ctx->hwctx = ((AVHWDeviceContext*)ctx->hw_device_ctx->data)->hwctx;
cuda_ctx = ctx->hwctx->cuda_ctx; cuda_ctx = ctx->hwctx->cuda_ctx;
@ -354,6 +356,8 @@ static int overlay_cuda_config_output(AVFilterLink *outlink)
ctx->cu_stream = ctx->hwctx->stream; ctx->cu_stream = ctx->hwctx->stream;
outlink->hw_frames_ctx = av_buffer_ref(inlink->hw_frames_ctx); outlink->hw_frames_ctx = av_buffer_ref(inlink->hw_frames_ctx);
if (!outlink->hw_frames_ctx)
return AVERROR(ENOMEM);
// load functions // load functions