avformat/mp3enc: flush buffered packets if referencing fails

Reviewed-by: Michael Niedermayer <michael@niedermayer.cc>
Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
James Almer 2017-10-06 15:55:09 -03:00
parent 2a31ad7d60
commit cc5b7601f7

View File

@ -515,20 +515,15 @@ static int mp3_write_packet(AVFormatContext *s, AVPacket *pkt)
if (mp3->pics_to_write) {
/* buffer audio packets until we get all the pictures */
AVPacketList *pktl = av_mallocz(sizeof(*pktl));
int ret;
if (!pktl) {
if (!pktl || av_packet_ref(&pktl->pkt, pkt) < 0) {
av_freep(&pktl);
av_log(s, AV_LOG_WARNING, "Not enough memory to buffer audio. Skipping picture streams\n");
mp3->pics_to_write = 0;
mp3_queue_flush(s);
return mp3_write_audio_packet(s, pkt);
}
ret = av_packet_ref(&pktl->pkt, pkt);
if (ret < 0) {
av_freep(&pktl);
return ret;
}
if (mp3->queue_end)
mp3->queue_end->next = pktl;
else