avutil/hwcontext_cuda: Only handle CUDA hardware frames

30b28f9a · Timo Rothenpieler · 741565a1 · 30b28f9a
Commit 30b28f9a authored Apr 06, 2020 by Timo Rothenpieler
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

hwcontext_cuda.c libavutil/hwcontext_cuda.c +4 -0

No files found.
--- a/libavutil/hwcontext_cuda.c
+++ b/libavutil/hwcontext_cuda.c
@@ -212,6 +212,10 @@ static int cuda_transfer_data(AVHWFramesContext *ctx, AVFrame *dst,
    CUcontext dummy;
    int i, ret;

+    if ((src->hw_frames_ctx && ((AVHWFramesContext*)src->hw_frames_ctx->data)->format != AV_PIX_FMT_CUDA) ||
+        (dst->hw_frames_ctx && ((AVHWFramesContext*)dst->hw_frames_ctx->data)->format != AV_PIX_FMT_CUDA))
+        return AVERROR(ENOSYS);
+
    ret = CHECK_CU(cu->cuCtxPushCurrent(hwctx->cuda_ctx));
    if (ret < 0)
        return ret;