From 68f9dfa5cc2e876207ca38affbbdaa0d6714d654 Mon Sep 17 00:00:00 2001 From: Timo Rothenpieler Date: Tue, 10 Oct 2023 17:34:47 +0200 Subject: [PATCH] avcodec/nvdec_hevc: fail to initialize on unsupported profiles --- libavcodec/nvdec_hevc.c | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/libavcodec/nvdec_hevc.c b/libavcodec/nvdec_hevc.c index 91fb9f5384..b83d5edcf9 100644 --- a/libavcodec/nvdec_hevc.c +++ b/libavcodec/nvdec_hevc.c @@ -306,6 +306,15 @@ static int nvdec_hevc_frame_params(AVCodecContext *avctx, static int nvdec_hevc_decode_init(AVCodecContext *avctx) { NVDECContext *ctx = avctx->internal->hwaccel_priv_data; ctx->supports_444 = 1; + + if (avctx->profile != AV_PROFILE_HEVC_MAIN && + avctx->profile != AV_PROFILE_HEVC_MAIN_10 && + avctx->profile != AV_PROFILE_HEVC_MAIN_STILL_PICTURE && + avctx->profile != AV_PROFILE_HEVC_REXT) { + av_log(avctx, AV_LOG_ERROR, "Unsupported HEVC profile: %d\n", avctx->profile); + return AVERROR(ENOTSUP); + } + return ff_nvdec_decode_init(avctx); }