> From: Fu, Linjie <linjie...@intel.com>
> Sent: Tuesday, March 31, 2020 23:34
> To: ffmpeg-devel@ffmpeg.org
> Cc: Fu, Linjie <linjie...@intel.com>
> Subject: [PATCH] lavc/vaapi_encode: add FF_CODEC_CAP_INIT_CLEANUP
> caps for encoders
>
> ff_vaapi_encode_close() is not enough to free the resources like cbs
> if initialization failure happens after codec->configure (except for
> vp8/vp9).
>
> We need to call avctx->codec->close() to deallocate, otherwise memory
> leak happens.
>
> Add FF_CODEC_CAP_INIT_CLEANUP for vaapi encoders and deallocate the
> resources at free_and_end inside avcodec_open2().
>
> Signed-off-by: Linjie Fu <linjie...@intel.com>
> ---
> libavcodec/vaapi_encode.c | 1 -
> libavcodec/vaapi_encode_h264.c | 1 +
> libavcodec/vaapi_encode_h265.c | 1 +
> libavcodec/vaapi_encode_mjpeg.c | 1 +
> libavcodec/vaapi_encode_mpeg2.c | 1 +
> libavcodec/vaapi_encode_vp8.c | 1 +
> libavcodec/vaapi_encode_vp9.c | 1 +
> 7 files changed, 6 insertions(+), 1 deletion(-)
>
> diff --git a/libavcodec/vaapi_encode.c b/libavcodec/vaapi_encode.c
> index 8ff720e..d371033 100644
> --- a/libavcodec/vaapi_encode.c
> +++ b/libavcodec/vaapi_encode.c
> @@ -2361,7 +2361,6 @@ av_cold int ff_vaapi_encode_init(AVCodecContext
> *avctx)
> return 0;
>
> fail:
> - ff_vaapi_encode_close(avctx);
> return err;
> }
>
> diff --git a/libavcodec/vaapi_encode_h264.c
> b/libavcodec/vaapi_encode_h264.c
> index f4965d8..6a86905 100644
> --- a/libavcodec/vaapi_encode_h264.c
> +++ b/libavcodec/vaapi_encode_h264.c
> @@ -1356,6 +1356,7 @@ AVCodec ff_h264_vaapi_encoder = {
> .close = &vaapi_encode_h264_close,
> .priv_class = &vaapi_encode_h264_class,
> .capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_HARDWARE,
> + .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
> .defaults = vaapi_encode_h264_defaults,
> .pix_fmts = (const enum AVPixelFormat[]) {
> AV_PIX_FMT_VAAPI,
> diff --git a/libavcodec/vaapi_encode_h265.c
> b/libavcodec/vaapi_encode_h265.c
> index 97dc5a7..4c24f7c 100644
> --- a/libavcodec/vaapi_encode_h265.c
> +++ b/libavcodec/vaapi_encode_h265.c
> @@ -1292,6 +1292,7 @@ AVCodec ff_hevc_vaapi_encoder = {
> .close = &vaapi_encode_h265_close,
> .priv_class = &vaapi_encode_h265_class,
> .capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_HARDWARE,
> + .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
> .defaults = vaapi_encode_h265_defaults,
> .pix_fmts = (const enum AVPixelFormat[]) {
> AV_PIX_FMT_VAAPI,
> diff --git a/libavcodec/vaapi_encode_mjpeg.c
> b/libavcodec/vaapi_encode_mjpeg.c
> index bd029cc..c469e70 100644
> --- a/libavcodec/vaapi_encode_mjpeg.c
> +++ b/libavcodec/vaapi_encode_mjpeg.c
> @@ -565,6 +565,7 @@ AVCodec ff_mjpeg_vaapi_encoder = {
> .priv_class = &vaapi_encode_mjpeg_class,
> .capabilities = AV_CODEC_CAP_HARDWARE |
> AV_CODEC_CAP_INTRA_ONLY,
> + .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
> .defaults = vaapi_encode_mjpeg_defaults,
> .pix_fmts = (const enum AVPixelFormat[]) {
> AV_PIX_FMT_VAAPI,
> diff --git a/libavcodec/vaapi_encode_mpeg2.c
> b/libavcodec/vaapi_encode_mpeg2.c
> index bac9ea1..55f9289 100644
> --- a/libavcodec/vaapi_encode_mpeg2.c
> +++ b/libavcodec/vaapi_encode_mpeg2.c
> @@ -702,6 +702,7 @@ AVCodec ff_mpeg2_vaapi_encoder = {
> .close = &vaapi_encode_mpeg2_close,
> .priv_class = &vaapi_encode_mpeg2_class,
> .capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_HARDWARE,
> + .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
> .defaults = vaapi_encode_mpeg2_defaults,
> .pix_fmts = (const enum AVPixelFormat[]) {
> AV_PIX_FMT_VAAPI,
> diff --git a/libavcodec/vaapi_encode_vp8.c
> b/libavcodec/vaapi_encode_vp8.c
> index 6e7bf9d..d8fb031 100644
> --- a/libavcodec/vaapi_encode_vp8.c
> +++ b/libavcodec/vaapi_encode_vp8.c
> @@ -257,6 +257,7 @@ AVCodec ff_vp8_vaapi_encoder = {
> .close = &ff_vaapi_encode_close,
> .priv_class = &vaapi_encode_vp8_class,
> .capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_HARDWARE,
> + .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
> .defaults = vaapi_encode_vp8_defaults,
> .pix_fmts = (const enum AVPixelFormat[]) {
> AV_PIX_FMT_VAAPI,
> diff --git a/libavcodec/vaapi_encode_vp9.c
> b/libavcodec/vaapi_encode_vp9.c
> index d7f415d..ea19470 100644
> --- a/libavcodec/vaapi_encode_vp9.c
> +++ b/libavcodec/vaapi_encode_vp9.c
> @@ -291,6 +291,7 @@ AVCodec ff_vp9_vaapi_encoder = {
> .close = &ff_vaapi_encode_close,
> .priv_class = &vaapi_encode_vp9_class,
> .capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_HARDWARE,
> + .caps_internal = FF_CODEC_CAP_INIT_CLEANUP,
> .defaults = vaapi_encode_vp9_defaults,
> .pix_fmts = (const enum AVPixelFormat[]) {
> AV_PIX_FMT_VAAPI,
> --
Ping for review, this patch helps to fix the potential memory leaks.
_______________________________________________
ffmpeg-devel mailing list
ffmpeg-devel@ffmpeg.org
https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
To unsubscribe, visit link above, or email
ffmpeg-devel-requ...@ffmpeg.org with subject "unsubscribe".
