[FFmpeg-devel] [PATCH 1/2] avformat/hnm: Check for extradata allocation failure

[Paul B Mahol]

lgtm

On 3/21/20, Andreas Rheinhardt <andreas.rheinhardt at gmail.com> wrote:
> and also add padding to it; moreover, don't use memcpy to write one byte
> to extradata.
>
> Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt at gmail.com>
> ---
>  libavformat/hnm.c | 7 ++++---
>  1 file changed, 4 insertions(+), 3 deletions(-)
>
> diff --git a/libavformat/hnm.c b/libavformat/hnm.c
> index 40ef5c06ee..31221553a4 100644
> --- a/libavformat/hnm.c
> +++ b/libavformat/hnm.c
> @@ -70,6 +70,7 @@ static int hnm_read_header(AVFormatContext *s)
>      Hnm4DemuxContext *hnm = s->priv_data;
>      AVIOContext *pb = s->pb;
>      AVStream *vst;
> +    int ret;
>
>      /* default context members */
>      hnm->pts = 0;
> @@ -113,10 +114,10 @@ static int hnm_read_header(AVFormatContext *s)
>      vst->codecpar->codec_tag  = 0;
>      vst->codecpar->width      = hnm->width;
>      vst->codecpar->height     = hnm->height;
> -    vst->codecpar->extradata  = av_mallocz(1);
> +    if ((ret = ff_alloc_extradata(vst->codecpar, 1)) < 0)
> +        return ret;
>
> -    vst->codecpar->extradata_size = 1;
> -    memcpy(vst->codecpar->extradata, &hnm->version, 1);
> +    vst->codecpar->extradata[0] = hnm->version;
>
>      vst->start_time = 0;
>
> --
> 2.20.1
>
> _______________________________________________
> ffmpeg-devel mailing list
> ffmpeg-devel at ffmpeg.org
> https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
>
> To unsubscribe, visit link above, or email
> ffmpeg-devel-request at ffmpeg.org with subject "unsubscribe".