From 308429e124b97337a768839c1d5091900e974e7e Mon Sep 17 00:00:00 2001 From: Michael Niedermayer Date: Tue, 9 Dec 2014 16:10:55 +0100 Subject: avformat/wavenc: check return value of strftime() Fixes CID1257006 Signed-off-by: Michael Niedermayer --- libavformat/wavenc.c | 17 ++++++++++++----- 1 file changed, 12 insertions(+), 5 deletions(-) (limited to 'libavformat/wavenc.c') diff --git a/libavformat/wavenc.c b/libavformat/wavenc.c index bce4876976..2345fc594a 100644 --- a/libavformat/wavenc.c +++ b/libavformat/wavenc.c @@ -252,7 +252,7 @@ static void peak_write_frame(AVFormatContext *s) wav->peak_num_frames++; } -static void peak_write_chunk(AVFormatContext *s) +static int peak_write_chunk(AVFormatContext *s) { WAVMuxContext *wav = s->priv_data; AVIOContext *pb = s->pb; @@ -272,8 +272,12 @@ static void peak_write_chunk(AVFormatContext *s) av_log(s, AV_LOG_INFO, "Writing local time and date to Peak Envelope Chunk\n"); now0 = av_gettime(); now_secs = now0 / 1000000; - strftime(timestamp, sizeof(timestamp), "%Y:%m:%d:%H:%M:%S:", localtime_r(&now_secs, &tmpbuf)); - av_strlcatf(timestamp, sizeof(timestamp), "%03d", (int)((now0 / 1000) % 1000)); + if (strftime(timestamp, sizeof(timestamp), "%Y:%m:%d:%H:%M:%S:", localtime_r(&now_secs, &tmpbuf))) { + av_strlcatf(timestamp, sizeof(timestamp), "%03d", (int)((now0 / 1000) % 1000)); + } else { + av_log(s, AV_LOG_ERROR, "Failed to write timestamp\n"); + return -1; + } } avio_wl32(pb, 1); /* version */ @@ -293,6 +297,8 @@ static void peak_write_chunk(AVFormatContext *s) if (!wav->data) wav->data = peak; + + return 0; } static int wav_write_header(AVFormatContext *s) @@ -414,6 +420,7 @@ static int wav_write_trailer(AVFormatContext *s) int64_t file_size, data_size; int64_t number_of_samples = 0; int rf64 = 0; + int ret = 0; avio_flush(pb); @@ -424,7 +431,7 @@ static int wav_write_trailer(AVFormatContext *s) } if (wav->write_peak && wav->peak_output) { - peak_write_chunk(s); + ret = peak_write_chunk(s); avio_flush(pb); } @@ -485,7 +492,7 @@ static int wav_write_trailer(AVFormatContext *s) if (wav->write_peak) peak_free_buffers(s); - return 0; + return ret; } #define OFFSET(x) offsetof(WAVMuxContext, x) -- cgit v1.2.3