Browse Source

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

Signed-off-by: Michael Niedermayer <michaelni@gmx.at>
tags/n2.1
Michael Niedermayer 11 years ago
parent
commit
c4810fbe4f
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      libavformat/utils.c

+ 3
- 2
libavformat/utils.c 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,


Loading…
Cancel
Save