1 files changed, 2 insertions, 2 deletions
diff --git a/libavformat/dfa.c b/libavformat/dfa.c
index 5799d983cf..12516afdc5 100644
--- a/libavformat/dfa.c
+++ b/libavformat/dfa.c
@@ -64,8 +64,8 @@ static int dfa_read_header(AVFormatContext *s)
     avio_skip(pb, 128 - 16); // padding
     st->duration = frames;
 
-    st->codec->extradata = av_malloc(2);
-    st->codec->extradata_size = 2;
+    if (ff_alloc_extradata(st->codec, 2))
+        return AVERROR(ENOMEM);
     AV_WL16(st->codec->extradata, version);
     if (version == 0x100)
         st->sample_aspect_ratio = (AVRational){2, 1};