segment: reorder seg_write_header allocation
authorLuca Barbato <lu_zero@gentoo.org>
Fri, 27 Apr 2012 18:09:30 +0000 (11:09 -0700)
committerLuca Barbato <lu_zero@gentoo.org>
Fri, 27 Apr 2012 21:03:43 +0000 (14:03 -0700)
As pointed by Paul B Mahol <onemda@gmail.com> the previous code could
lead to null pointer dereference.

libavformat/segment.c

index 1af412a..8274792 100644 (file)
@@ -113,10 +113,15 @@ static int seg_write_header(AVFormatContext *s)
     seg->offset_time = 0;
     seg->recording_time = seg->time * 1000000;
 
+    oc = avformat_alloc_context();
+
+    if (!oc)
+        return AVERROR(ENOMEM);
+
     if (seg->list)
         if ((ret = avio_open2(&seg->pb, seg->list, AVIO_FLAG_WRITE,
                               &s->interrupt_callback, NULL)) < 0)
-            return ret;
+            goto fail;
 
     for (i = 0; i< s->nb_streams; i++)
         seg->has_video +=
@@ -127,13 +132,6 @@ static int seg_write_header(AVFormatContext *s)
                "More than a single video stream present, "
                "expect issues decoding it.\n");
 
-    oc = avformat_alloc_context();
-
-    if (!oc) {
-        ret = AVERROR(ENOMEM);
-        goto fail;
-    }
-
     oc->oformat = av_guess_format(seg->format, s->filename, NULL);
 
     if (!oc->oformat) {