X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=libavformat%2Fapngenc.c;h=0c40be297fcb83be31e91be6b5d54d5d3ba9ef9c;hb=f17eea883a6fc8dbae71fe445ab0c7e925ac2953;hp=e5e8aa998f0618b212d1e64c37cb2c9d3de3fab8;hpb=a478ae39388bbe534889f7007f1d4b769224343e;p=ffmpeg

diff --git a/libavformat/apngenc.c b/libavformat/apngenc.c
index e5e8aa998f0..0c40be297fc 100644
--- a/libavformat/apngenc.c
+++ b/libavformat/apngenc.c
@@ -81,6 +81,7 @@ static void apng_write_chunk(AVIOContext *io_context, uint32_t tag,
 static int apng_write_header(AVFormatContext *format_context)
 {
     APNGMuxContext *apng = format_context->priv_data;
+    AVCodecParameters *par = format_context->streams[0]->codecpar;
 
     if (format_context->nb_streams != 1 ||
         format_context->streams[0]->codecpar->codec_type != AVMEDIA_TYPE_VIDEO ||
@@ -101,6 +102,14 @@ static int apng_write_header(AVFormatContext *format_context)
     avio_wb64(format_context->pb, PNGSIG);
     // Remaining headers are written when they are copied from the encoder
 
+    if (par->extradata_size) {
+        apng->extra_data = av_mallocz(par->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
+        if (!apng->extra_data)
+            return AVERROR(ENOMEM);
+        apng->extra_data_size = par->extradata_size;
+        memcpy(apng->extra_data, par->extradata, par->extradata_size);
+    }
+
     return 0;
 }