1
0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2025-03-23 04:24:35 +02:00

avformat/utils: fix av_probe_input_buffer2() so it returns the probe score

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
This commit is contained in:
Michael Niedermayer 2013-08-29 01:07:57 +02:00
parent 7528fdcda3
commit c4810fbe4f

View File

@ -322,6 +322,7 @@ int av_probe_input_buffer2(AVIOContext *pb, AVInputFormat **fmt,
unsigned char *buf = NULL; unsigned char *buf = NULL;
uint8_t *mime_type; uint8_t *mime_type;
int ret = 0, probe_size, buf_offset = 0; int ret = 0, probe_size, buf_offset = 0;
int score = 0;
if (!max_probe_size) { if (!max_probe_size) {
max_probe_size = PROBE_BUF_MAX; max_probe_size = PROBE_BUF_MAX;
@ -346,12 +347,12 @@ int av_probe_input_buffer2(AVIOContext *pb, AVInputFormat **fmt,
for(probe_size= PROBE_BUF_MIN; probe_size<=max_probe_size && !*fmt; for(probe_size= PROBE_BUF_MIN; probe_size<=max_probe_size && !*fmt;
probe_size = FFMIN(probe_size<<1, FFMAX(max_probe_size, probe_size+1))) { probe_size = FFMIN(probe_size<<1, FFMAX(max_probe_size, probe_size+1))) {
int score = probe_size < max_probe_size ? AVPROBE_SCORE_RETRY : 0;
void *buftmp; void *buftmp;
if (probe_size < offset) { if (probe_size < offset) {
continue; continue;
} }
score = probe_size < max_probe_size ? AVPROBE_SCORE_RETRY : 0;
/* read probe data */ /* read probe data */
buftmp = av_realloc(buf, probe_size + AVPROBE_PADDING_SIZE); buftmp = av_realloc(buf, probe_size + AVPROBE_PADDING_SIZE);
@ -392,7 +393,7 @@ int av_probe_input_buffer2(AVIOContext *pb, AVInputFormat **fmt,
/* rewind. reuse probe buffer to avoid seeking */ /* rewind. reuse probe buffer to avoid seeking */
ret = ffio_rewind_with_probe_data(pb, &buf, pd.buf_size); ret = ffio_rewind_with_probe_data(pb, &buf, pd.buf_size);
return ret; return ret < 0 ? ret : score;
} }
int av_probe_input_buffer(AVIOContext *pb, AVInputFormat **fmt, int av_probe_input_buffer(AVIOContext *pb, AVInputFormat **fmt,