ffmpeg_dxva2: add a profile check for hevc

1 files changed, 6 insertions, 0 deletions

diff --git a/ffmpeg_dxva2.c b/ffmpeg_dxva2.c
index 43509b464e..09af6c7929 100644
--- a/ffmpeg_dxva2.c
+++ b/ffmpeg_dxva2.c
@@ -637,6 +637,12 @@ int dxva2_init(AVCodecContext *s)
         return AVERROR(EINVAL);
     }
 
+    if (s->codec_id == AV_CODEC_ID_HEVC &&
+        s->profile != FF_PROFILE_HEVC_MAIN && s->profile != FF_PROFILE_HEVC_MAIN_10) {
+        av_log(NULL, loglevel, "Unsupported HEVC profile for DXVA2 HWAccel: %d\n", s->profile);
+        return AVERROR(EINVAL);
+    }
+
     if (ctx->decoder)
         dxva2_destroy_decoder(s);