[FFmpeg-devel] [PATCH v2] avcodec/h264_metadata_bsf: remove AUDs at any position

[Gyan Doshi]

On 2023-02-04 03:36 pm, Gyan Doshi wrote:
> Some files, likely due to faulty packetization or muxing, can have AUDs
> at other positions besides the head unit of a packet. Remove these too.
Plan to push in 24h if no more comments.

> ---
>   libavcodec/h264_metadata_bsf.c | 13 +++++++------
>   1 file changed, 7 insertions(+), 6 deletions(-)
>
> diff --git a/libavcodec/h264_metadata_bsf.c b/libavcodec/h264_metadata_bsf.c
> index d318bf0cee..b9cfeaba94 100644
> --- a/libavcodec/h264_metadata_bsf.c
> +++ b/libavcodec/h264_metadata_bsf.c
> @@ -469,12 +469,13 @@ static int h264_metadata_update_fragment(AVBSFContext *bsf, AVPacket *pkt,
>       H264MetadataContext *ctx = bsf->priv_data;
>       int err, i, has_sps, seek_point;
>   
> -    // If an AUD is present, it must be the first NAL unit.
> -    if (au->nb_units && au->units[0].type == H264_NAL_AUD) {
> -        if (ctx->aud == BSF_ELEMENT_REMOVE)
> -            ff_cbs_delete_unit(au, 0);
> -    } else {
> -        if (pkt && ctx->aud == BSF_ELEMENT_INSERT) {
> +    if (ctx->aud == BSF_ELEMENT_REMOVE) {
> +        for (i = au->nb_units - 1; i >= 0; i--) {
> +            if (au->units[i].type == H264_NAL_AUD)
> +                ff_cbs_delete_unit(au, i);
> +        }
> +    } else if (ctx->aud == BSF_ELEMENT_INSERT) {
> +        if (pkt) {
>               err = h264_metadata_insert_aud(bsf, au);
>               if (err < 0)
>                   return err;