summaryrefslogtreecommitdiffstats
path: root/demux/demux_lavf.c
diff options
context:
space:
mode:
Diffstat (limited to 'demux/demux_lavf.c')
-rw-r--r--demux/demux_lavf.c64
1 files changed, 33 insertions, 31 deletions
diff --git a/demux/demux_lavf.c b/demux/demux_lavf.c
index a482f736c6..efe8edd43c 100644
--- a/demux/demux_lavf.c
+++ b/demux/demux_lavf.c
@@ -47,7 +47,7 @@
#include "core/m_option.h"
#define INITIAL_PROBE_SIZE STREAM_BUFFER_SIZE
-#define PROBE_BUF_SIZE (2 * 1024 * 1024)
+#define PROBE_BUF_SIZE FFMIN(STREAM_MAX_BUFFER_SIZE, 2 * 1024 * 1024)
#define OPT_BASE_STRUCT struct MPOpts
@@ -190,18 +190,15 @@ static int lavf_check_file(demuxer_t *demuxer)
{
struct MPOpts *opts = demuxer->opts;
struct lavfdopts *lavfdopts = &opts->lavfdopts;
- AVProbeData avpd;
+ struct stream *s = demuxer->stream;
lavf_priv_t *priv;
- int probe_data_size = 0;
- int read_size = INITIAL_PROBE_SIZE;
- int score;
assert(!demuxer->priv);
demuxer->priv = talloc_zero(NULL, lavf_priv_t);
priv = demuxer->priv;
priv->autoselect_sub = -1;
- priv->filename = demuxer->stream->url;
+ priv->filename = s->url;
if (!priv->filename) {
priv->filename = "mp:unknown";
mp_msg(MSGT_DEMUX, MSGL_WARN, "Stream url is not set!\n");
@@ -210,7 +207,7 @@ static int lavf_check_file(demuxer_t *demuxer)
priv->filename = remove_prefix(priv->filename, prefixes);
char *avdevice_format = NULL;
- if (demuxer->stream->type == STREAMTYPE_AVDEVICE) {
+ if (s->type == STREAMTYPE_AVDEVICE) {
// always require filename in the form "format:filename"
char *sep = strchr(priv->filename, ':');
if (!sep) {
@@ -237,7 +234,7 @@ static int lavf_check_file(demuxer_t *demuxer)
const char *format = lavfdopts->format;
if (!format)
- format = demuxer->stream->lavf_type;
+ format = s->lavf_type;
if (!format)
format = avdevice_format;
if (format) {
@@ -262,38 +259,43 @@ static int lavf_check_file(demuxer_t *demuxer)
if (lavfdopts->probescore)
min_probe = lavfdopts->probescore;
- avpd.buf = av_mallocz(FFMAX(BIO_BUFFER_SIZE, PROBE_BUF_SIZE) +
- FF_INPUT_BUFFER_PADDING_SIZE);
- do {
- read_size = stream_read(demuxer->stream, avpd.buf + probe_data_size,
- read_size);
- if (read_size < 0) {
- av_free(avpd.buf);
- return 0;
- }
- probe_data_size += read_size;
- avpd.filename = priv->filename;
- avpd.buf_size = probe_data_size;
+ AVProbeData avpd = {
+ .filename = priv->filename,
+ .buf_size = 0,
+ .buf = av_mallocz(PROBE_BUF_SIZE + FF_INPUT_BUFFER_PADDING_SIZE),
+ };
+
+ while (avpd.buf_size < PROBE_BUF_SIZE) {
+ int nsize = av_clip(avpd.buf_size * 2, INITIAL_PROBE_SIZE,
+ PROBE_BUF_SIZE);
+ int read_size = stream_read(s, avpd.buf + avpd.buf_size,
+ nsize - avpd.buf_size);
+ if (read_size <= 0)
+ break;
- score = 0;
- priv->avif = av_probe_input_format2(&avpd, probe_data_size > 0, &score);
+ avpd.buf_size += read_size;
+
+ int score = 0;
+ priv->avif = av_probe_input_format2(&avpd, avpd.buf_size > 0, &score);
if (priv->avif) {
mp_msg(MSGT_HEADER, MSGL_V, "Found '%s' at score=%d size=%d.\n",
- priv->avif->name, score, probe_data_size);
- }
+ priv->avif->name, score, avpd.buf_size);
- if (priv->avif && score >= min_probe)
- break;
- if (priv->avif && expected_format) {
- if (strcmp(priv->avif->name, expected_format) == 0 &&
- score >= expected_format_probescore)
+ if (score >= min_probe)
break;
+
+ if (expected_format) {
+ if (strcmp(priv->avif->name, expected_format) == 0 &&
+ score >= expected_format_probescore)
+ break;
+ }
}
priv->avif = NULL;
- read_size = FFMIN(2 * read_size, PROBE_BUF_SIZE - probe_data_size);
- } while (read_size > 0 && probe_data_size < PROBE_BUF_SIZE);
+ }
+
+ stream_unread_buffer(s, avpd.buf, avpd.buf_size);
av_free(avpd.buf);
if (!priv->avif) {