diff --git a/vpxenc.c b/vpxenc.c index 6c13cd1bdaf7f6fae6facb2823a3537399f004d1..39256b6657dfe8093b916df3058e1fa9625a1a11 100755 --- a/vpxenc.c +++ b/vpxenc.c @@ -236,7 +236,13 @@ void stats_write(stats_io_t *stats, const void *pkt, size_t len) stats->buf_ptr = new_ptr + (stats->buf_ptr - (char *)stats->buf.buf); stats->buf.buf = new_ptr; stats->buf_alloc_sz = new_sz; - } /* else ... */ + } + else + { + fprintf(stderr, + "\nFailed to realloc firstpass stats buffer.\n"); + exit(EXIT_FAILURE); + } } memcpy(stats->buf_ptr, pkt, len); @@ -698,10 +704,18 @@ write_webm_block(EbmlGlobal *glob, /* Save a cue point if this is a keyframe. */ if(is_keyframe) { - struct cue_entry *cue; + struct cue_entry *cue, *new_cue_list; + + new_cue_list = realloc(glob->cue_list, + (glob->cues+1) * sizeof(struct cue_entry)); + if(new_cue_list) + glob->cue_list = new_cue_list; + else + { + fprintf(stderr, "\nFailed to realloc cue list.\n"); + exit(EXIT_FAILURE); + } - glob->cue_list = realloc(glob->cue_list, - (glob->cues+1) * sizeof(struct cue_entry)); cue = &glob->cue_list[glob->cues]; cue->time = glob->cluster_timecode; cue->loc = glob->cluster_pos;