[FFmpeg-devel] [PATCH 1/2] avcodec/cuviddec: fix null pointer dereference

[FFmpeg-devel] [PATCH 1/2] avcodec/cuviddec: fix null pointer dereference

[Zhao Zhili]

From: Zhao Zhili <zhilizhao@tencent.com>

It can happened on error path of cuvid_decode_init().
---
 libavcodec/cuviddec.c | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/libavcodec/cuviddec.c b/libavcodec/cuviddec.c
index fc9c0515fe..2f6bbdc7d8 100644
--- a/libavcodec/cuviddec.c
+++ b/libavcodec/cuviddec.c
@@ -655,12 +655,13 @@ error:
 static av_cold int cuvid_decode_end(AVCodecContext *avctx)
 {
     CuvidContext *ctx = avctx->priv_data;
-    AVHWDeviceContext *device_ctx = (AVHWDeviceContext *)ctx->hwdevice->data;
-    AVCUDADeviceContext *device_hwctx = device_ctx->hwctx;
-    CUcontext dummy, cuda_ctx = device_hwctx->cuda_ctx;
+    AVHWDeviceContext *device_ctx = ctx->hwdevice ? (AVHWDeviceContext *)ctx->hwdevice->data : NULL;
+    AVCUDADeviceContext *device_hwctx = device_ctx ? device_ctx->hwctx : NULL;
+    CUcontext dummy, cuda_ctx = device_hwctx ? device_hwctx->cuda_ctx : NULL;
 
     av_fifo_freep2(&ctx->frame_queue);
 
+    if (cuda_ctx) {
     ctx->cudl->cuCtxPushCurrent(cuda_ctx);
 
     if (ctx->cuparser)
@@ -670,6 +671,7 @@ static av_cold int cuvid_decode_end(AVCodecContext *avctx)
         ctx->cvdl->cuvidDestroyDecoder(ctx->cudecoder);
 
     ctx->cudl->cuCtxPopCurrent(&dummy);
+    }
 
     ctx->cudl = NULL;
 
-- 
2.34.1

_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel

To unsubscribe, visit link above, or email
ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".