virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-28 20:53:54 +02:00
Code Issues Releases Activity

lavf: Make probe_codec return an error code

Browse Source 
This allows handling errors from av_realloc properly.

Signed-off-by: Martin Storsjö <martin@martin.st>
This commit is contained in:
Alexandra Khirnova 2013-09-12 10:39:22 +02:00 committed by Martin Storsjö
parent 1daea5232f
commit 68b4677420
1 changed files with 14 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
libavformat/utils.c
View File

@ -449,7 +449,7 @@ fail:
/*******************************************************/ /*******************************************************/
static void probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt) static int probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt)
{ {
if(st->codec->codec_id == AV_CODEC_ID_PROBE){ if(st->codec->codec_id == AV_CODEC_ID_PROBE){
AVProbeData *pd = &st->probe_data; AVProbeData *pd = &st->probe_data;
@ -457,7 +457,10 @@ static void probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt)
--st->probe_packets; --st->probe_packets;
if (pkt) { if (pkt) {
pd->buf = av_realloc(pd->buf, pd->buf_size+pkt->size+AVPROBE_PADDING_SIZE); int err;
if ((err = av_reallocp(&pd->buf, pd->buf_size + pkt->size +
AVPROBE_PADDING_SIZE)) < 0)
return err;
memcpy(pd->buf+pd->buf_size, pkt->data, pkt->size); memcpy(pd->buf+pd->buf_size, pkt->data, pkt->size);
pd->buf_size += pkt->size; pd->buf_size += pkt->size;
memset(pd->buf+pd->buf_size, 0, AVPROBE_PADDING_SIZE); memset(pd->buf+pd->buf_size, 0, AVPROBE_PADDING_SIZE);
@ -466,7 +469,7 @@ static void probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt)
if (!pd->buf_size) { if (!pd->buf_size) {
av_log(s, AV_LOG_ERROR, "nothing to probe for stream %d\n", av_log(s, AV_LOG_ERROR, "nothing to probe for stream %d\n",
st->index); st->index);
return; return 0;
} }
} }
@ -480,11 +483,12 @@ static void probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt)
} }
} }
} }
return 0;
} }
int ff_read_packet(AVFormatContext *s, AVPacket *pkt) int ff_read_packet(AVFormatContext *s, AVPacket *pkt)
{ {
int ret, i; int ret, i, err;
AVStream *st; AVStream *st;
for(;;){ for(;;){
@ -497,7 +501,8 @@ int ff_read_packet(AVFormatContext *s, AVPacket *pkt)
s->raw_packet_buffer_remaining_size < pkt->size) { s->raw_packet_buffer_remaining_size < pkt->size) {
AVProbeData *pd; AVProbeData *pd;
if (st->probe_packets) { if (st->probe_packets) {
probe_codec(s, st, NULL); if ((err = probe_codec(s, st, NULL)) < 0)
return err;
} }
pd = &st->probe_data; pd = &st->probe_data;
av_freep(&pd->buf); av_freep(&pd->buf);
@ -519,7 +524,8 @@ int ff_read_packet(AVFormatContext *s, AVPacket *pkt)
for (i = 0; i < s->nb_streams; i++) { for (i = 0; i < s->nb_streams; i++) {
st = s->streams[i]; st = s->streams[i];
if (st->probe_packets) { if (st->probe_packets) {
probe_codec(s, st, NULL); if ((err = probe_codec(s, st, NULL)) < 0)
return err;
} }
} }
continue; continue;
@ -555,7 +561,8 @@ int ff_read_packet(AVFormatContext *s, AVPacket *pkt)
add_to_pktbuf(&s->raw_packet_buffer, pkt, &s->raw_packet_buffer_end); add_to_pktbuf(&s->raw_packet_buffer, pkt, &s->raw_packet_buffer_end);
s->raw_packet_buffer_remaining_size -= pkt->size; s->raw_packet_buffer_remaining_size -= pkt->size;
probe_codec(s, st, pkt); if ((err = probe_codec(s, st, pkt)) < 0)
return err;
} }
} }