diff options
author | Marton Balint <cus@passwd.hu> | 2016-06-11 21:18:40 +0300 |
---|---|---|
committer | Marton Balint <cus@passwd.hu> | 2016-06-20 01:27:26 +0300 |
commit | 517fe64406d4b0f5ae2b1428f17063ac7a1643aa (patch) | |
tree | 7223b80899d26844a5e1e0fc0fe6eab2ddb3e8eb | |
parent | b577d4218385667921ed01ed5785dcb37a459b9f (diff) |
avformat/mux: do not call write_header multiple times if it fails the first time
Signed-off-by: Marton Balint <cus@passwd.hu>
-rw-r--r-- | libavformat/internal.h | 1 | ||||
-rw-r--r-- | libavformat/mux.c | 5 |
2 files changed, 4 insertions, 2 deletions
diff --git a/libavformat/internal.h b/libavformat/internal.h index 40ba0899b5..b6c2020386 100644 --- a/libavformat/internal.h +++ b/libavformat/internal.h @@ -124,6 +124,7 @@ struct AVFormatInternal { * Whether or not a header has already been written */ int header_written; + int write_header_ret; }; struct AVStreamInternal { diff --git a/libavformat/mux.c b/libavformat/mux.c index 08ed940874..b1b65a1e96 100644 --- a/libavformat/mux.c +++ b/libavformat/mux.c @@ -479,6 +479,7 @@ static int write_header_internal(AVFormatContext *s) int ret = s->oformat->write_header(s); if (ret >= 0 && s->pb && s->pb->error < 0) ret = s->pb->error; + s->internal->write_header_ret = ret; if (ret < 0) return ret; if (s->flush_packets && s->pb && s->pb->error >= 0 && s->flags & AVFMT_FLAG_FLUSH_PACKETS) @@ -713,7 +714,7 @@ static int write_packet(AVFormatContext *s, AVPacket *pkt) did_split = av_packet_split_side_data(pkt); if (!s->internal->header_written) { - ret = write_header_internal(s); + ret = s->internal->write_header_ret ? s->internal->write_header_ret : write_header_internal(s); if (ret < 0) goto fail; } @@ -1152,7 +1153,7 @@ int av_write_trailer(AVFormatContext *s) } if (!s->internal->header_written) { - ret = write_header_internal(s); + ret = s->internal->write_header_ret ? s->internal->write_header_ret : write_header_internal(s); if (ret < 0) goto fail; } |