FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
utils.c
Go to the documentation of this file.
1 /*
2  * various utility functions for use within FFmpeg
3  * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include <stdarg.h>
23 #include <stdint.h>
24 
25 #include "config.h"
26 
27 #include "libavutil/avassert.h"
28 #include "libavutil/avstring.h"
29 #include "libavutil/dict.h"
30 #include "libavutil/internal.h"
31 #include "libavutil/mathematics.h"
32 #include "libavutil/opt.h"
33 #include "libavutil/parseutils.h"
34 #include "libavutil/pixdesc.h"
35 #include "libavutil/time.h"
37 #include "libavutil/timestamp.h"
38 
39 #include "libavcodec/bytestream.h"
40 #include "libavcodec/internal.h"
41 #include "libavcodec/raw.h"
42 
43 #include "audiointerleave.h"
44 #include "avformat.h"
45 #include "avio_internal.h"
46 #include "id3v2.h"
47 #include "internal.h"
48 #include "metadata.h"
49 #if CONFIG_NETWORK
50 #include "network.h"
51 #endif
52 #include "riff.h"
53 #include "url.h"
54 
55 #include "libavutil/ffversion.h"
56 const char av_format_ffversion[] = "FFmpeg version " FFMPEG_VERSION;
57 
58 /**
59  * @file
60  * various utility functions for use within FFmpeg
61  */
62 
63 unsigned avformat_version(void)
64 {
67 }
68 
69 const char *avformat_configuration(void)
70 {
71  return FFMPEG_CONFIGURATION;
72 }
73 
74 const char *avformat_license(void)
75 {
76 #define LICENSE_PREFIX "libavformat license: "
77  return LICENSE_PREFIX FFMPEG_LICENSE + sizeof(LICENSE_PREFIX) - 1;
78 }
79 
80 #define RELATIVE_TS_BASE (INT64_MAX - (1LL<<48))
81 
82 static int is_relative(int64_t ts) {
83  return ts > (RELATIVE_TS_BASE - (1LL<<48));
84 }
85 
86 /**
87  * Wrap a given time stamp, if there is an indication for an overflow
88  *
89  * @param st stream
90  * @param timestamp the time stamp to wrap
91  * @return resulting time stamp
92  */
93 static int64_t wrap_timestamp(AVStream *st, int64_t timestamp)
94 {
96  st->pts_wrap_reference != AV_NOPTS_VALUE && timestamp != AV_NOPTS_VALUE) {
98  timestamp < st->pts_wrap_reference)
99  return timestamp + (1ULL << st->pts_wrap_bits);
100  else if (st->pts_wrap_behavior == AV_PTS_WRAP_SUB_OFFSET &&
101  timestamp >= st->pts_wrap_reference)
102  return timestamp - (1ULL << st->pts_wrap_bits);
103  }
104  return timestamp;
105 }
106 
107 MAKE_ACCESSORS(AVStream, stream, AVRational, r_frame_rate)
108 MAKE_ACCESSORS(AVStream, stream, char *, recommended_encoder_configuration)
110 MAKE_ACCESSORS(AVFormatContext, format, AVCodec *, audio_codec)
111 MAKE_ACCESSORS(AVFormatContext, format, AVCodec *, subtitle_codec)
112 MAKE_ACCESSORS(AVFormatContext, format, AVCodec *, data_codec)
113 MAKE_ACCESSORS(AVFormatContext, format, int, metadata_header_padding)
114 MAKE_ACCESSORS(AVFormatContext, format, void *, opaque)
115 MAKE_ACCESSORS(AVFormatContext, format, av_format_control_message, control_message_cb)
116 #if FF_API_OLD_OPEN_CALLBACKS
118 MAKE_ACCESSORS(AVFormatContext, format, AVOpenCallback, open_cb)
120 #endif
121 
122 int64_t av_stream_get_end_pts(const AVStream *st)
123 {
124  if (st->priv_pts) {
125  return st->priv_pts->val;
126  } else
127  return AV_NOPTS_VALUE;
128 }
129 
130 struct AVCodecParserContext *av_stream_get_parser(const AVStream *st)
131 {
132  return st->parser;
133 }
134 
135 void av_format_inject_global_side_data(AVFormatContext *s)
136 {
137  int i;
139  for (i = 0; i < s->nb_streams; i++) {
140  AVStream *st = s->streams[i];
141  st->inject_global_side_data = 1;
142  }
143 }
144 
145 int ff_copy_whiteblacklists(AVFormatContext *dst, AVFormatContext *src)
146 {
147  av_assert0(!dst->codec_whitelist &&
148  !dst->format_whitelist &&
149  !dst->protocol_whitelist &&
150  !dst->protocol_blacklist);
151  dst-> codec_whitelist = av_strdup(src->codec_whitelist);
155  if ( (src-> codec_whitelist && !dst-> codec_whitelist)
156  || (src-> format_whitelist && !dst-> format_whitelist)
157  || (src->protocol_whitelist && !dst->protocol_whitelist)
158  || (src->protocol_blacklist && !dst->protocol_blacklist)) {
159  av_log(dst, AV_LOG_ERROR, "Failed to duplicate black/whitelist\n");
160  return AVERROR(ENOMEM);
161  }
162  return 0;
163 }
164 
165 static const AVCodec *find_decoder(AVFormatContext *s, AVStream *st, enum AVCodecID codec_id)
166 {
167 #if FF_API_LAVF_AVCTX
169  if (st->codec->codec)
170  return st->codec->codec;
172 #endif
173 
174  switch (st->codecpar->codec_type) {
175  case AVMEDIA_TYPE_VIDEO:
176  if (s->video_codec) return s->video_codec;
177  break;
178  case AVMEDIA_TYPE_AUDIO:
179  if (s->audio_codec) return s->audio_codec;
180  break;
182  if (s->subtitle_codec) return s->subtitle_codec;
183  break;
184  }
185 
186  return avcodec_find_decoder(codec_id);
187 }
188 
189 int av_format_get_probe_score(const AVFormatContext *s)
190 {
191  return s->probe_score;
192 }
193 
194 /* an arbitrarily chosen "sane" max packet size -- 50M */
195 #define SANE_CHUNK_SIZE (50000000)
196 
198 {
199  if (s->maxsize>= 0) {
200  int64_t remaining= s->maxsize - avio_tell(s);
201  if (remaining < size) {
202  int64_t newsize = avio_size(s);
203  if (!s->maxsize || s->maxsize<newsize)
204  s->maxsize = newsize - !newsize;
205  remaining= s->maxsize - avio_tell(s);
206  remaining= FFMAX(remaining, 0);
207  }
208 
209  if (s->maxsize>= 0 && remaining+1 < size) {
210  av_log(NULL, remaining ? AV_LOG_ERROR : AV_LOG_DEBUG, "Truncating packet of size %d to %"PRId64"\n", size, remaining+1);
211  size = remaining+1;
212  }
213  }
214  return size;
215 }
216 
217 /* Read the data in sane-sized chunks and append to pkt.
218  * Return the number of bytes read or an error. */
220 {
221  int64_t orig_pos = pkt->pos; // av_grow_packet might reset pos
222  int orig_size = pkt->size;
223  int ret;
224 
225  do {
226  int prev_size = pkt->size;
227  int read_size;
228 
229  /* When the caller requests a lot of data, limit it to the amount
230  * left in file or SANE_CHUNK_SIZE when it is not known. */
231  read_size = size;
232  if (read_size > SANE_CHUNK_SIZE/10) {
233  read_size = ffio_limit(s, read_size);
234  // If filesize/maxsize is unknown, limit to SANE_CHUNK_SIZE
235  if (s->maxsize < 0)
236  read_size = FFMIN(read_size, SANE_CHUNK_SIZE);
237  }
238 
239  ret = av_grow_packet(pkt, read_size);
240  if (ret < 0)
241  break;
242 
243  ret = avio_read(s, pkt->data + prev_size, read_size);
244  if (ret != read_size) {
245  av_shrink_packet(pkt, prev_size + FFMAX(ret, 0));
246  break;
247  }
248 
249  size -= read_size;
250  } while (size > 0);
251  if (size > 0)
252  pkt->flags |= AV_PKT_FLAG_CORRUPT;
253 
254  pkt->pos = orig_pos;
255  if (!pkt->size)
256  av_packet_unref(pkt);
257  return pkt->size > orig_size ? pkt->size - orig_size : ret;
258 }
259 
261 {
262  av_init_packet(pkt);
263  pkt->data = NULL;
264  pkt->size = 0;
265  pkt->pos = avio_tell(s);
266 
267  return append_packet_chunked(s, pkt, size);
268 }
269 
271 {
272  if (!pkt->size)
273  return av_get_packet(s, pkt, size);
274  return append_packet_chunked(s, pkt, size);
275 }
276 
277 int av_filename_number_test(const char *filename)
278 {
279  char buf[1024];
280  return filename &&
281  (av_get_frame_filename(buf, sizeof(buf), filename, 1) >= 0);
282 }
283 
284 static int set_codec_from_probe_data(AVFormatContext *s, AVStream *st,
285  AVProbeData *pd)
286 {
287  static const struct {
288  const char *name;
289  enum AVCodecID id;
290  enum AVMediaType type;
291  } fmt_id_type[] = {
303  { "mpegvideo", AV_CODEC_ID_MPEG2VIDEO, AVMEDIA_TYPE_VIDEO },
304  { "truehd", AV_CODEC_ID_TRUEHD, AVMEDIA_TYPE_AUDIO },
305  { 0 }
306  };
307  int score;
308  AVInputFormat *fmt = av_probe_input_format3(pd, 1, &score);
309 
310  if (fmt && st->request_probe <= score) {
311  int i;
312  av_log(s, AV_LOG_DEBUG,
313  "Probe with size=%d, packets=%d detected %s with score=%d\n",
315  fmt->name, score);
316  for (i = 0; fmt_id_type[i].name; i++) {
317  if (!strcmp(fmt->name, fmt_id_type[i].name)) {
318  if (fmt_id_type[i].type != AVMEDIA_TYPE_AUDIO &&
319  st->codecpar->sample_rate)
320  continue;
321  st->codecpar->codec_id = fmt_id_type[i].id;
322  st->codecpar->codec_type = fmt_id_type[i].type;
323  st->internal->need_context_update = 1;
324 #if FF_API_LAVF_AVCTX
326  st->codec->codec_type = st->codecpar->codec_type;
327  st->codec->codec_id = st->codecpar->codec_id;
329 #endif
330  return score;
331  }
332  }
333  }
334  return 0;
335 }
336 
337 /************************************************************/
338 /* input media file */
339 
340 int av_demuxer_open(AVFormatContext *ic) {
341  int err;
342 
343  if (ic->format_whitelist && av_match_list(ic->iformat->name, ic->format_whitelist, ',') <= 0) {
344  av_log(ic, AV_LOG_ERROR, "Format not on whitelist \'%s\'\n", ic->format_whitelist);
345  return AVERROR(EINVAL);
346  }
347 
348  if (ic->iformat->read_header) {
349  err = ic->iformat->read_header(ic);
350  if (err < 0)
351  return err;
352  }
353 
354  if (ic->pb && !ic->internal->data_offset)
355  ic->internal->data_offset = avio_tell(ic->pb);
356 
357  return 0;
358 }
359 
360 /* Open input file and probe the format if necessary. */
361 static int init_input(AVFormatContext *s, const char *filename,
363 {
364  int ret;
365  AVProbeData pd = { filename, NULL, 0 };
366  int score = AVPROBE_SCORE_RETRY;
367 
368  if (s->pb) {
370  if (!s->iformat)
371  return av_probe_input_buffer2(s->pb, &s->iformat, filename,
372  s, 0, s->format_probesize);
373  else if (s->iformat->flags & AVFMT_NOFILE)
374  av_log(s, AV_LOG_WARNING, "Custom AVIOContext makes no sense and "
375  "will be ignored with AVFMT_NOFILE format.\n");
376  return 0;
377  }
378 
379  if ((s->iformat && s->iformat->flags & AVFMT_NOFILE) ||
380  (!s->iformat && (s->iformat = av_probe_input_format2(&pd, 0, &score))))
381  return score;
382 
383  if ((ret = s->io_open(s, &s->pb, filename, AVIO_FLAG_READ | s->avio_flags, options)) < 0)
384  return ret;
385 
386  if (s->iformat)
387  return 0;
388  return av_probe_input_buffer2(s->pb, &s->iformat, filename,
389  s, 0, s->format_probesize);
390 }
391 
392 static int add_to_pktbuf(AVPacketList **packet_buffer, AVPacket *pkt,
393  AVPacketList **plast_pktl, int ref)
394 {
395  AVPacketList *pktl = av_mallocz(sizeof(AVPacketList));
396  int ret;
397 
398  if (!pktl)
399  return AVERROR(ENOMEM);
400 
401  if (ref) {
402  if ((ret = av_packet_ref(&pktl->pkt, pkt)) < 0) {
403  av_free(pktl);
404  return ret;
405  }
406  } else {
407  pktl->pkt = *pkt;
408  }
409 
410  if (*packet_buffer)
411  (*plast_pktl)->next = pktl;
412  else
413  *packet_buffer = pktl;
414 
415  /* Add the packet in the buffered packet list. */
416  *plast_pktl = pktl;
417  return 0;
418 }
419 
420 int avformat_queue_attached_pictures(AVFormatContext *s)
421 {
422  int i, ret;
423  for (i = 0; i < s->nb_streams; i++)
425  s->streams[i]->discard < AVDISCARD_ALL) {
426  if (s->streams[i]->attached_pic.size <= 0) {
428  "Attached picture on stream %d has invalid size, "
429  "ignoring\n", i);
430  continue;
431  }
432 
434  &s->streams[i]->attached_pic,
436  if (ret < 0)
437  return ret;
438  }
439  return 0;
440 }
441 
442 static int update_stream_avctx(AVFormatContext *s)
443 {
444  int i, ret;
445  for (i = 0; i < s->nb_streams; i++) {
446  AVStream *st = s->streams[i];
447 
448  if (!st->internal->need_context_update)
449  continue;
450 
451  /* update internal codec context, for the parser */
453  if (ret < 0)
454  return ret;
455 
456 #if FF_API_LAVF_AVCTX
458  /* update deprecated public codec context */
459  ret = avcodec_parameters_to_context(st->codec, st->codecpar);
460  if (ret < 0)
461  return ret;
463 #endif
464 
465  st->internal->need_context_update = 0;
466  }
467  return 0;
468 }
469 
470 
471 int avformat_open_input(AVFormatContext **ps, const char *filename,
473 {
474  AVFormatContext *s = *ps;
475  int i, ret = 0;
476  AVDictionary *tmp = NULL;
477  ID3v2ExtraMeta *id3v2_extra_meta = NULL;
478 
479  if (!s && !(s = avformat_alloc_context()))
480  return AVERROR(ENOMEM);
481  if (!s->av_class) {
482  av_log(NULL, AV_LOG_ERROR, "Input context has not been properly allocated by avformat_alloc_context() and is not NULL either\n");
483  return AVERROR(EINVAL);
484  }
485  if (fmt)
486  s->iformat = fmt;
487 
488  if (options)
489  av_dict_copy(&tmp, *options, 0);
490 
491  if (s->pb) // must be before any goto fail
493 
494  if ((ret = av_opt_set_dict(s, &tmp)) < 0)
495  goto fail;
496 
497  if ((ret = init_input(s, filename, &tmp)) < 0)
498  goto fail;
499  s->probe_score = ret;
500 
501  if (!s->protocol_whitelist && s->pb && s->pb->protocol_whitelist) {
503  if (!s->protocol_whitelist) {
504  ret = AVERROR(ENOMEM);
505  goto fail;
506  }
507  }
508 
509  if (!s->protocol_blacklist && s->pb && s->pb->protocol_blacklist) {
511  if (!s->protocol_blacklist) {
512  ret = AVERROR(ENOMEM);
513  goto fail;
514  }
515  }
516 
517  if (s->format_whitelist && av_match_list(s->iformat->name, s->format_whitelist, ',') <= 0) {
518  av_log(s, AV_LOG_ERROR, "Format not on whitelist \'%s\'\n", s->format_whitelist);
519  ret = AVERROR(EINVAL);
520  goto fail;
521  }
522 
524 
525  /* Check filename in case an image number is expected. */
526  if (s->iformat->flags & AVFMT_NEEDNUMBER) {
527  if (!av_filename_number_test(filename)) {
528  ret = AVERROR(EINVAL);
529  goto fail;
530  }
531  }
532 
534  av_strlcpy(s->filename, filename ? filename : "", sizeof(s->filename));
535 
536  /* Allocate private data. */
537  if (s->iformat->priv_data_size > 0) {
538  if (!(s->priv_data = av_mallocz(s->iformat->priv_data_size))) {
539  ret = AVERROR(ENOMEM);
540  goto fail;
541  }
542  if (s->iformat->priv_class) {
543  *(const AVClass **) s->priv_data = s->iformat->priv_class;
545  if ((ret = av_opt_set_dict(s->priv_data, &tmp)) < 0)
546  goto fail;
547  }
548  }
549 
550  /* e.g. AVFMT_NOFILE formats will not have a AVIOContext */
551  if (s->pb)
552  ff_id3v2_read(s, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta, 0);
553 
554  if (!(s->flags&AVFMT_FLAG_PRIV_OPT) && s->iformat->read_header)
555  if ((ret = s->iformat->read_header(s)) < 0)
556  goto fail;
557 
558  if (id3v2_extra_meta) {
559  if (!strcmp(s->iformat->name, "mp3") || !strcmp(s->iformat->name, "aac") ||
560  !strcmp(s->iformat->name, "tta")) {
561  if ((ret = ff_id3v2_parse_apic(s, &id3v2_extra_meta)) < 0)
562  goto fail;
563  } else
564  av_log(s, AV_LOG_DEBUG, "demuxer does not support additional id3 data, skipping\n");
565  }
566  ff_id3v2_free_extra_meta(&id3v2_extra_meta);
567 
568  if ((ret = avformat_queue_attached_pictures(s)) < 0)
569  goto fail;
570 
571  if (!(s->flags&AVFMT_FLAG_PRIV_OPT) && s->pb && !s->internal->data_offset)
572  s->internal->data_offset = avio_tell(s->pb);
573 
575 
577 
578  for (i = 0; i < s->nb_streams; i++)
580 
581  if (options) {
582  av_dict_free(options);
583  *options = tmp;
584  }
585  *ps = s;
586  return 0;
587 
588 fail:
589  ff_id3v2_free_extra_meta(&id3v2_extra_meta);
590  av_dict_free(&tmp);
591  if (s->pb && !(s->flags & AVFMT_FLAG_CUSTOM_IO))
592  avio_closep(&s->pb);
594  *ps = NULL;
595  return ret;
596 }
597 
598 /*******************************************************/
599 
600 static void force_codec_ids(AVFormatContext *s, AVStream *st)
601 {
602  switch (st->codecpar->codec_type) {
603  case AVMEDIA_TYPE_VIDEO:
604  if (s->video_codec_id)
605  st->codecpar->codec_id = s->video_codec_id;
606  break;
607  case AVMEDIA_TYPE_AUDIO:
608  if (s->audio_codec_id)
609  st->codecpar->codec_id = s->audio_codec_id;
610  break;
612  if (s->subtitle_codec_id)
614  break;
615  }
616 }
617 
618 static int probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt)
619 {
620  if (st->request_probe>0) {
621  AVProbeData *pd = &st->probe_data;
622  int end;
623  av_log(s, AV_LOG_DEBUG, "probing stream %d pp:%d\n", st->index, st->probe_packets);
624  --st->probe_packets;
625 
626  if (pkt) {
627  uint8_t *new_buf = av_realloc(pd->buf, pd->buf_size+pkt->size+AVPROBE_PADDING_SIZE);
628  if (!new_buf) {
630  "Failed to reallocate probe buffer for stream %d\n",
631  st->index);
632  goto no_packet;
633  }
634  pd->buf = new_buf;
635  memcpy(pd->buf + pd->buf_size, pkt->data, pkt->size);
636  pd->buf_size += pkt->size;
637  memset(pd->buf + pd->buf_size, 0, AVPROBE_PADDING_SIZE);
638  } else {
639 no_packet:
640  st->probe_packets = 0;
641  if (!pd->buf_size) {
643  "nothing to probe for stream %d\n", st->index);
644  }
645  }
646 
648  || st->probe_packets<= 0;
649 
650  if (end || av_log2(pd->buf_size) != av_log2(pd->buf_size - pkt->size)) {
651  int score = set_codec_from_probe_data(s, st, pd);
653  || end) {
654  pd->buf_size = 0;
655  av_freep(&pd->buf);
656  st->request_probe = -1;
657  if (st->codecpar->codec_id != AV_CODEC_ID_NONE) {
658  av_log(s, AV_LOG_DEBUG, "probed stream %d\n", st->index);
659  } else
660  av_log(s, AV_LOG_WARNING, "probed stream %d failed\n", st->index);
661  }
662  force_codec_ids(s, st);
663  }
664  }
665  return 0;
666 }
667 
668 static int update_wrap_reference(AVFormatContext *s, AVStream *st, int stream_index, AVPacket *pkt)
669 {
670  int64_t ref = pkt->dts;
671  int i, pts_wrap_behavior;
672  int64_t pts_wrap_reference;
673  AVProgram *first_program;
674 
675  if (ref == AV_NOPTS_VALUE)
676  ref = pkt->pts;
677  if (st->pts_wrap_reference != AV_NOPTS_VALUE || st->pts_wrap_bits >= 63 || ref == AV_NOPTS_VALUE || !s->correct_ts_overflow)
678  return 0;
679  ref &= (1LL << st->pts_wrap_bits)-1;
680 
681  // reference time stamp should be 60 s before first time stamp
682  pts_wrap_reference = ref - av_rescale(60, st->time_base.den, st->time_base.num);
683  // if first time stamp is not more than 1/8 and 60s before the wrap point, subtract rather than add wrap offset
684  pts_wrap_behavior = (ref < (1LL << st->pts_wrap_bits) - (1LL << st->pts_wrap_bits-3)) ||
685  (ref < (1LL << st->pts_wrap_bits) - av_rescale(60, st->time_base.den, st->time_base.num)) ?
687 
688  first_program = av_find_program_from_stream(s, NULL, stream_index);
689 
690  if (!first_program) {
691  int default_stream_index = av_find_default_stream_index(s);
692  if (s->streams[default_stream_index]->pts_wrap_reference == AV_NOPTS_VALUE) {
693  for (i = 0; i < s->nb_streams; i++) {
695  continue;
696  s->streams[i]->pts_wrap_reference = pts_wrap_reference;
697  s->streams[i]->pts_wrap_behavior = pts_wrap_behavior;
698  }
699  }
700  else {
701  st->pts_wrap_reference = s->streams[default_stream_index]->pts_wrap_reference;
702  st->pts_wrap_behavior = s->streams[default_stream_index]->pts_wrap_behavior;
703  }
704  }
705  else {
706  AVProgram *program = first_program;
707  while (program) {
708  if (program->pts_wrap_reference != AV_NOPTS_VALUE) {
709  pts_wrap_reference = program->pts_wrap_reference;
710  pts_wrap_behavior = program->pts_wrap_behavior;
711  break;
712  }
713  program = av_find_program_from_stream(s, program, stream_index);
714  }
715 
716  // update every program with differing pts_wrap_reference
717  program = first_program;
718  while (program) {
719  if (program->pts_wrap_reference != pts_wrap_reference) {
720  for (i = 0; i<program->nb_stream_indexes; i++) {
721  s->streams[program->stream_index[i]]->pts_wrap_reference = pts_wrap_reference;
722  s->streams[program->stream_index[i]]->pts_wrap_behavior = pts_wrap_behavior;
723  }
724 
725  program->pts_wrap_reference = pts_wrap_reference;
726  program->pts_wrap_behavior = pts_wrap_behavior;
727  }
728  program = av_find_program_from_stream(s, program, stream_index);
729  }
730  }
731  return 1;
732 }
733 
734 int ff_read_packet(AVFormatContext *s, AVPacket *pkt)
735 {
736  int ret, i, err;
737  AVStream *st;
738 
739  for (;;) {
741 
742  if (pktl) {
743  *pkt = pktl->pkt;
744  st = s->streams[pkt->stream_index];
746  if ((err = probe_codec(s, st, NULL)) < 0)
747  return err;
748  if (st->request_probe <= 0) {
749  s->internal->raw_packet_buffer = pktl->next;
751  av_free(pktl);
752  return 0;
753  }
754  }
755 
756  pkt->data = NULL;
757  pkt->size = 0;
758  av_init_packet(pkt);
759  ret = s->iformat->read_packet(s, pkt);
760  if (ret < 0) {
761  /* Some demuxers return FFERROR_REDO when they consume
762  data and discard it (ignored streams, junk, extradata).
763  We must re-call the demuxer to get the real packet. */
764  if (ret == FFERROR_REDO)
765  continue;
766  if (!pktl || ret == AVERROR(EAGAIN))
767  return ret;
768  for (i = 0; i < s->nb_streams; i++) {
769  st = s->streams[i];
770  if (st->probe_packets)
771  if ((err = probe_codec(s, st, NULL)) < 0)
772  return err;
773  av_assert0(st->request_probe <= 0);
774  }
775  continue;
776  }
777 
778  if (!pkt->buf) {
779  AVPacket tmp = { 0 };
780  ret = av_packet_ref(&tmp, pkt);
781  if (ret < 0)
782  return ret;
783  *pkt = tmp;
784  }
785 
786  if ((s->flags & AVFMT_FLAG_DISCARD_CORRUPT) &&
787  (pkt->flags & AV_PKT_FLAG_CORRUPT)) {
789  "Dropped corrupted packet (stream = %d)\n",
790  pkt->stream_index);
791  av_packet_unref(pkt);
792  continue;
793  }
794 
795  if (pkt->stream_index >= (unsigned)s->nb_streams) {
796  av_log(s, AV_LOG_ERROR, "Invalid stream index %d\n", pkt->stream_index);
797  continue;
798  }
799 
800  st = s->streams[pkt->stream_index];
801 
803  // correct first time stamps to negative values
804  if (!is_relative(st->first_dts))
805  st->first_dts = wrap_timestamp(st, st->first_dts);
806  if (!is_relative(st->start_time))
807  st->start_time = wrap_timestamp(st, st->start_time);
808  if (!is_relative(st->cur_dts))
809  st->cur_dts = wrap_timestamp(st, st->cur_dts);
810  }
811 
812  pkt->dts = wrap_timestamp(st, pkt->dts);
813  pkt->pts = wrap_timestamp(st, pkt->pts);
814 
815  force_codec_ids(s, st);
816 
817  /* TODO: audio: time filter; video: frame reordering (pts != dts) */
819  pkt->dts = pkt->pts = av_rescale_q(av_gettime(), AV_TIME_BASE_Q, st->time_base);
820 
821  if (!pktl && st->request_probe <= 0)
822  return ret;
823 
824  err = add_to_pktbuf(&s->internal->raw_packet_buffer, pkt,
826  if (err)
827  return err;
829 
830  if ((err = probe_codec(s, st, pkt)) < 0)
831  return err;
832  }
833 }
834 
835 
836 /**********************************************************/
837 
839 {
840  if (/*avctx->codec_id == AV_CODEC_ID_AAC ||*/
841  avctx->codec_id == AV_CODEC_ID_MP1 ||
842  avctx->codec_id == AV_CODEC_ID_MP2 ||
843  avctx->codec_id == AV_CODEC_ID_MP3/* ||
844  avctx->codec_id == AV_CODEC_ID_CELT*/)
845  return 1;
846  return 0;
847 }
848 
849 /**
850  * Return the frame duration in seconds. Return 0 if not available.
851  */
852 void ff_compute_frame_duration(AVFormatContext *s, int *pnum, int *pden, AVStream *st,
854 {
855  AVRational codec_framerate = s->iformat ? st->internal->avctx->framerate :
857  int frame_size, sample_rate;
858 
859 #if FF_API_LAVF_AVCTX
861  if ((!codec_framerate.den || !codec_framerate.num) && st->codec->time_base.den && st->codec->time_base.num)
862  codec_framerate = av_mul_q(av_inv_q(st->codec->time_base), (AVRational){1, st->codec->ticks_per_frame});
864 #endif
865 
866  *pnum = 0;
867  *pden = 0;
868  switch (st->codecpar->codec_type) {
869  case AVMEDIA_TYPE_VIDEO:
870  if (st->r_frame_rate.num && !pc && s->iformat) {
871  *pnum = st->r_frame_rate.den;
872  *pden = st->r_frame_rate.num;
873  } else if (st->time_base.num * 1000LL > st->time_base.den) {
874  *pnum = st->time_base.num;
875  *pden = st->time_base.den;
876  } else if (codec_framerate.den * 1000LL > codec_framerate.num) {
877  av_assert0(st->internal->avctx->ticks_per_frame);
878  av_reduce(pnum, pden,
879  codec_framerate.den,
880  codec_framerate.num * (int64_t)st->internal->avctx->ticks_per_frame,
881  INT_MAX);
882 
883  if (pc && pc->repeat_pict) {
884  av_assert0(s->iformat); // this may be wrong for interlaced encoding but its not used for that case
885  av_reduce(pnum, pden,
886  (*pnum) * (1LL + pc->repeat_pict),
887  (*pden),
888  INT_MAX);
889  }
890  /* If this codec can be interlaced or progressive then we need
891  * a parser to compute duration of a packet. Thus if we have
892  * no parser in such case leave duration undefined. */
893  if (st->internal->avctx->ticks_per_frame > 1 && !pc)
894  *pnum = *pden = 0;
895  }
896  break;
897  case AVMEDIA_TYPE_AUDIO:
898  if (st->internal->avctx_inited) {
899  frame_size = av_get_audio_frame_duration(st->internal->avctx, pkt->size);
900  sample_rate = st->internal->avctx->sample_rate;
901  } else {
902  frame_size = av_get_audio_frame_duration2(st->codecpar, pkt->size);
903  sample_rate = st->codecpar->sample_rate;
904  }
905  if (frame_size <= 0 || sample_rate <= 0)
906  break;
907  *pnum = frame_size;
908  *pden = sample_rate;
909  break;
910  default:
911  break;
912  }
913 }
914 
915 static int is_intra_only(enum AVCodecID id)
916 {
918  if (!d)
919  return 0;
921  return 0;
922  return 1;
923 }
924 
925 static int has_decode_delay_been_guessed(AVStream *st)
926 {
927  if (st->codecpar->codec_id != AV_CODEC_ID_H264) return 1;
928  if (!st->info) // if we have left find_stream_info then nb_decoded_frames won't increase anymore for stream copy
929  return 1;
930 #if CONFIG_H264_DECODER
931  if (st->internal->avctx->has_b_frames &&
933  return 1;
934 #endif
935  if (st->internal->avctx->has_b_frames<3)
936  return st->nb_decoded_frames >= 7;
937  else if (st->internal->avctx->has_b_frames<4)
938  return st->nb_decoded_frames >= 18;
939  else
940  return st->nb_decoded_frames >= 20;
941 }
942 
943 static AVPacketList *get_next_pkt(AVFormatContext *s, AVStream *st, AVPacketList *pktl)
944 {
945  if (pktl->next)
946  return pktl->next;
947  if (pktl == s->internal->packet_buffer_end)
948  return s->internal->parse_queue;
949  return NULL;
950 }
951 
952 static int64_t select_from_pts_buffer(AVStream *st, int64_t *pts_buffer, int64_t dts) {
953  int onein_oneout = st->codecpar->codec_id != AV_CODEC_ID_H264 &&
955 
956  if(!onein_oneout) {
957  int delay = st->internal->avctx->has_b_frames;
958  int i;
959 
960  if (dts == AV_NOPTS_VALUE) {
961  int64_t best_score = INT64_MAX;
962  for (i = 0; i<delay; i++) {
963  if (st->pts_reorder_error_count[i]) {
964  int64_t score = st->pts_reorder_error[i] / st->pts_reorder_error_count[i];
965  if (score < best_score) {
966  best_score = score;
967  dts = pts_buffer[i];
968  }
969  }
970  }
971  } else {
972  for (i = 0; i<delay; i++) {
973  if (pts_buffer[i] != AV_NOPTS_VALUE) {
974  int64_t diff = FFABS(pts_buffer[i] - dts)
975  + (uint64_t)st->pts_reorder_error[i];
976  diff = FFMAX(diff, st->pts_reorder_error[i]);
977  st->pts_reorder_error[i] = diff;
978  st->pts_reorder_error_count[i]++;
979  if (st->pts_reorder_error_count[i] > 250) {
980  st->pts_reorder_error[i] >>= 1;
981  st->pts_reorder_error_count[i] >>= 1;
982  }
983  }
984  }
985  }
986  }
987 
988  if (dts == AV_NOPTS_VALUE)
989  dts = pts_buffer[0];
990 
991  return dts;
992 }
993 
994 /**
995  * Updates the dts of packets of a stream in pkt_buffer, by re-ordering the pts
996  * of the packets in a window.
997  */
998 static void update_dts_from_pts(AVFormatContext *s, int stream_index,
999  AVPacketList *pkt_buffer)
1000 {
1001  AVStream *st = s->streams[stream_index];
1002  int delay = st->internal->avctx->has_b_frames;
1003  int i;
1004 
1005  int64_t pts_buffer[MAX_REORDER_DELAY+1];
1006 
1007  for (i = 0; i<MAX_REORDER_DELAY+1; i++)
1008  pts_buffer[i] = AV_NOPTS_VALUE;
1009 
1010  for (; pkt_buffer; pkt_buffer = get_next_pkt(s, st, pkt_buffer)) {
1011  if (pkt_buffer->pkt.stream_index != stream_index)
1012  continue;
1013 
1014  if (pkt_buffer->pkt.pts != AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY) {
1015  pts_buffer[0] = pkt_buffer->pkt.pts;
1016  for (i = 0; i<delay && pts_buffer[i] > pts_buffer[i + 1]; i++)
1017  FFSWAP(int64_t, pts_buffer[i], pts_buffer[i + 1]);
1018 
1019  pkt_buffer->pkt.dts = select_from_pts_buffer(st, pts_buffer, pkt_buffer->pkt.dts);
1020  }
1021  }
1022 }
1023 
1024 static void update_initial_timestamps(AVFormatContext *s, int stream_index,
1025  int64_t dts, int64_t pts, AVPacket *pkt)
1026 {
1027  AVStream *st = s->streams[stream_index];
1029  AVPacketList *pktl_it;
1030 
1031  uint64_t shift;
1032 
1033  if (st->first_dts != AV_NOPTS_VALUE ||
1034  dts == AV_NOPTS_VALUE ||
1035  st->cur_dts == AV_NOPTS_VALUE ||
1036  is_relative(dts))
1037  return;
1038 
1039  st->first_dts = dts - (st->cur_dts - RELATIVE_TS_BASE);
1040  st->cur_dts = dts;
1041  shift = (uint64_t)st->first_dts - RELATIVE_TS_BASE;
1042 
1043  if (is_relative(pts))
1044  pts += shift;
1045 
1046  for (pktl_it = pktl; pktl_it; pktl_it = get_next_pkt(s, st, pktl_it)) {
1047  if (pktl_it->pkt.stream_index != stream_index)
1048  continue;
1049  if (is_relative(pktl_it->pkt.pts))
1050  pktl_it->pkt.pts += shift;
1051 
1052  if (is_relative(pktl_it->pkt.dts))
1053  pktl_it->pkt.dts += shift;
1054 
1055  if (st->start_time == AV_NOPTS_VALUE && pktl_it->pkt.pts != AV_NOPTS_VALUE) {
1056  st->start_time = pktl_it->pkt.pts;
1059  }
1060  }
1061 
1063  update_dts_from_pts(s, stream_index, pktl);
1064  }
1065 
1066  if (st->start_time == AV_NOPTS_VALUE) {
1067  st->start_time = pts;
1068  if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO && st->codecpar->sample_rate)
1069  st->start_time += av_rescale_q(st->skip_samples, (AVRational){1, st->codecpar->sample_rate}, st->time_base);
1070  }
1071 }
1072 
1073 static void update_initial_durations(AVFormatContext *s, AVStream *st,
1074  int stream_index, int duration)
1075 {
1077  int64_t cur_dts = RELATIVE_TS_BASE;
1078 
1079  if (st->first_dts != AV_NOPTS_VALUE) {
1081  return;
1083  cur_dts = st->first_dts;
1084  for (; pktl; pktl = get_next_pkt(s, st, pktl)) {
1085  if (pktl->pkt.stream_index == stream_index) {
1086  if (pktl->pkt.pts != pktl->pkt.dts ||
1087  pktl->pkt.dts != AV_NOPTS_VALUE ||
1088  pktl->pkt.duration)
1089  break;
1090  cur_dts -= duration;
1091  }
1092  }
1093  if (pktl && pktl->pkt.dts != st->first_dts) {
1094  av_log(s, AV_LOG_DEBUG, "first_dts %s not matching first dts %s (pts %s, duration %"PRId64") in the queue\n",
1095  av_ts2str(st->first_dts), av_ts2str(pktl->pkt.dts), av_ts2str(pktl->pkt.pts), pktl->pkt.duration);
1096  return;
1097  }
1098  if (!pktl) {
1099  av_log(s, AV_LOG_DEBUG, "first_dts %s but no packet with dts in the queue\n", av_ts2str(st->first_dts));
1100  return;
1101  }
1103  st->first_dts = cur_dts;
1104  } else if (st->cur_dts != RELATIVE_TS_BASE)
1105  return;
1106 
1107  for (; pktl; pktl = get_next_pkt(s, st, pktl)) {
1108  if (pktl->pkt.stream_index != stream_index)
1109  continue;
1110  if (pktl->pkt.pts == pktl->pkt.dts &&
1111  (pktl->pkt.dts == AV_NOPTS_VALUE || pktl->pkt.dts == st->first_dts) &&
1112  !pktl->pkt.duration) {
1113  pktl->pkt.dts = cur_dts;
1114  if (!st->internal->avctx->has_b_frames)
1115  pktl->pkt.pts = cur_dts;
1116 // if (st->codecpar->codec_type != AVMEDIA_TYPE_AUDIO)
1117  pktl->pkt.duration = duration;
1118  } else
1119  break;
1120  cur_dts = pktl->pkt.dts + pktl->pkt.duration;
1121  }
1122  if (!pktl)
1123  st->cur_dts = cur_dts;
1124 }
1125 
1126 static void compute_pkt_fields(AVFormatContext *s, AVStream *st,
1128  int64_t next_dts, int64_t next_pts)
1129 {
1130  int num, den, presentation_delayed, delay, i;
1131  int64_t offset;
1133  int onein_oneout = st->codecpar->codec_id != AV_CODEC_ID_H264 &&
1135 
1136  if (s->flags & AVFMT_FLAG_NOFILLIN)
1137  return;
1138 
1139  if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && pkt->dts != AV_NOPTS_VALUE) {
1140  if (pkt->dts == pkt->pts && st->last_dts_for_order_check != AV_NOPTS_VALUE) {
1141  if (st->last_dts_for_order_check <= pkt->dts) {
1142  st->dts_ordered++;
1143  } else {
1145  "DTS %"PRIi64" < %"PRIi64" out of order\n",
1146  pkt->dts,
1148  st->dts_misordered++;
1149  }
1150  if (st->dts_ordered + st->dts_misordered > 250) {
1151  st->dts_ordered >>= 1;
1152  st->dts_misordered >>= 1;
1153  }
1154  }
1155 
1156  st->last_dts_for_order_check = pkt->dts;
1157  if (st->dts_ordered < 8*st->dts_misordered && pkt->dts == pkt->pts)
1158  pkt->dts = AV_NOPTS_VALUE;
1159  }
1160 
1161  if ((s->flags & AVFMT_FLAG_IGNDTS) && pkt->pts != AV_NOPTS_VALUE)
1162  pkt->dts = AV_NOPTS_VALUE;
1163 
1164  if (pc && pc->pict_type == AV_PICTURE_TYPE_B
1165  && !st->internal->avctx->has_b_frames)
1166  //FIXME Set low_delay = 0 when has_b_frames = 1
1167  st->internal->avctx->has_b_frames = 1;
1168 
1169  /* do we have a video B-frame ? */
1170  delay = st->internal->avctx->has_b_frames;
1171  presentation_delayed = 0;
1172 
1173  /* XXX: need has_b_frame, but cannot get it if the codec is
1174  * not initialized */
1175  if (delay &&
1176  pc && pc->pict_type != AV_PICTURE_TYPE_B)
1177  presentation_delayed = 1;
1178 
1179  if (pkt->pts != AV_NOPTS_VALUE && pkt->dts != AV_NOPTS_VALUE &&
1180  st->pts_wrap_bits < 63 &&
1181  pkt->dts - (1LL << (st->pts_wrap_bits - 1)) > pkt->pts) {
1182  if (is_relative(st->cur_dts) || pkt->dts - (1LL<<(st->pts_wrap_bits - 1)) > st->cur_dts) {
1183  pkt->dts -= 1LL << st->pts_wrap_bits;
1184  } else
1185  pkt->pts += 1LL << st->pts_wrap_bits;
1186  }
1187 
1188  /* Some MPEG-2 in MPEG-PS lack dts (issue #171 / input_file.mpg).
1189  * We take the conservative approach and discard both.
1190  * Note: If this is misbehaving for an H.264 file, then possibly
1191  * presentation_delayed is not set correctly. */
1192  if (delay == 1 && pkt->dts == pkt->pts &&
1193  pkt->dts != AV_NOPTS_VALUE && presentation_delayed) {
1194  av_log(s, AV_LOG_DEBUG, "invalid dts/pts combination %"PRIi64"\n", pkt->dts);
1195  if ( strcmp(s->iformat->name, "mov,mp4,m4a,3gp,3g2,mj2")
1196  && strcmp(s->iformat->name, "flv")) // otherwise we discard correct timestamps for vc1-wmapro.ism
1197  pkt->dts = AV_NOPTS_VALUE;
1198  }
1199 
1200  duration = av_mul_q((AVRational) {pkt->duration, 1}, st->time_base);
1201  if (pkt->duration == 0) {
1202  ff_compute_frame_duration(s, &num, &den, st, pc, pkt);
1203  if (den && num) {
1204  duration = (AVRational) {num, den};
1205  pkt->duration = av_rescale_rnd(1,
1206  num * (int64_t) st->time_base.den,
1207  den * (int64_t) st->time_base.num,
1208  AV_ROUND_DOWN);
1209  }
1210  }
1211 
1212  if (pkt->duration != 0 && (s->internal->packet_buffer || s->internal->parse_queue))
1213  update_initial_durations(s, st, pkt->stream_index, pkt->duration);
1214 
1215  /* Correct timestamps with byte offset if demuxers only have timestamps
1216  * on packet boundaries */
1217  if (pc && st->need_parsing == AVSTREAM_PARSE_TIMESTAMPS && pkt->size) {
1218  /* this will estimate bitrate based on this frame's duration and size */
1219  offset = av_rescale(pc->offset, pkt->duration, pkt->size);
1220  if (pkt->pts != AV_NOPTS_VALUE)
1221  pkt->pts += offset;
1222  if (pkt->dts != AV_NOPTS_VALUE)
1223  pkt->dts += offset;
1224  }
1225 
1226  /* This may be redundant, but it should not hurt. */
1227  if (pkt->dts != AV_NOPTS_VALUE &&
1228  pkt->pts != AV_NOPTS_VALUE &&
1229  pkt->pts > pkt->dts)
1230  presentation_delayed = 1;
1231 
1232  if (s->debug & FF_FDEBUG_TS)
1233  av_log(s, AV_LOG_TRACE,
1234  "IN delayed:%d pts:%s, dts:%s cur_dts:%s st:%d pc:%p duration:%"PRId64" delay:%d onein_oneout:%d\n",
1235  presentation_delayed, av_ts2str(pkt->pts), av_ts2str(pkt->dts), av_ts2str(st->cur_dts),
1236  pkt->stream_index, pc, pkt->duration, delay, onein_oneout);
1237 
1238  /* Interpolate PTS and DTS if they are not present. We skip H264
1239  * currently because delay and has_b_frames are not reliably set. */
1240  if ((delay == 0 || (delay == 1 && pc)) &&
1241  onein_oneout) {
1242  if (presentation_delayed) {
1243  /* DTS = decompression timestamp */
1244  /* PTS = presentation timestamp */
1245  if (pkt->dts == AV_NOPTS_VALUE)
1246  pkt->dts = st->last_IP_pts;
1247  update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts, pkt);
1248  if (pkt->dts == AV_NOPTS_VALUE)
1249  pkt->dts = st->cur_dts;
1250 
1251  /* This is tricky: the dts must be incremented by the duration
1252  * of the frame we are displaying, i.e. the last I- or P-frame. */
1253  if (st->last_IP_duration == 0)
1254  st->last_IP_duration = pkt->duration;
1255  if (pkt->dts != AV_NOPTS_VALUE)
1256  st->cur_dts = pkt->dts + st->last_IP_duration;
1257  if (pkt->dts != AV_NOPTS_VALUE &&
1258  pkt->pts == AV_NOPTS_VALUE &&
1259  st->last_IP_duration > 0 &&
1260  ((uint64_t)st->cur_dts - (uint64_t)next_dts + 1) <= 2 &&
1261  next_dts != next_pts &&
1262  next_pts != AV_NOPTS_VALUE)
1263  pkt->pts = next_dts;
1264 
1265  st->last_IP_duration = pkt->duration;
1266  st->last_IP_pts = pkt->pts;
1267  /* Cannot compute PTS if not present (we can compute it only
1268  * by knowing the future. */
1269  } else if (pkt->pts != AV_NOPTS_VALUE ||
1270  pkt->dts != AV_NOPTS_VALUE ||
1271  pkt->duration ) {
1272 
1273  /* presentation is not delayed : PTS and DTS are the same */
1274  if (pkt->pts == AV_NOPTS_VALUE)
1275  pkt->pts = pkt->dts;
1277  pkt->pts, pkt);
1278  if (pkt->pts == AV_NOPTS_VALUE)
1279  pkt->pts = st->cur_dts;
1280  pkt->dts = pkt->pts;
1281  if (pkt->pts != AV_NOPTS_VALUE)
1282  st->cur_dts = av_add_stable(st->time_base, pkt->pts, duration, 1);
1283  }
1284  }
1285 
1286  if (pkt->pts != AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY) {
1287  st->pts_buffer[0] = pkt->pts;
1288  for (i = 0; i<delay && st->pts_buffer[i] > st->pts_buffer[i + 1]; i++)
1289  FFSWAP(int64_t, st->pts_buffer[i], st->pts_buffer[i + 1]);
1290 
1292  pkt->dts = select_from_pts_buffer(st, st->pts_buffer, pkt->dts);
1293  }
1294  // We skipped it above so we try here.
1295  if (!onein_oneout)
1296  // This should happen on the first packet
1297  update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts, pkt);
1298  if (pkt->dts > st->cur_dts)
1299  st->cur_dts = pkt->dts;
1300 
1301  if (s->debug & FF_FDEBUG_TS)
1302  av_log(s, AV_LOG_TRACE, "OUTdelayed:%d/%d pts:%s, dts:%s cur_dts:%s\n",
1303  presentation_delayed, delay, av_ts2str(pkt->pts), av_ts2str(pkt->dts), av_ts2str(st->cur_dts));
1304 
1305  /* update flags */
1306  if (is_intra_only(st->codecpar->codec_id))
1307  pkt->flags |= AV_PKT_FLAG_KEY;
1308 #if FF_API_CONVERGENCE_DURATION
1310  if (pc)
1313 #endif
1314 }
1315 
1316 static void free_packet_buffer(AVPacketList **pkt_buf, AVPacketList **pkt_buf_end)
1317 {
1318  while (*pkt_buf) {
1319  AVPacketList *pktl = *pkt_buf;
1320  *pkt_buf = pktl->next;
1321  av_packet_unref(&pktl->pkt);
1322  av_freep(&pktl);
1323  }
1324  *pkt_buf_end = NULL;
1325 }
1326 
1327 /**
1328  * Parse a packet, add all split parts to parse_queue.
1329  *
1330  * @param pkt Packet to parse, NULL when flushing the parser at end of stream.
1331  */
1332 static int parse_packet(AVFormatContext *s, AVPacket *pkt, int stream_index)
1333 {
1334  AVPacket out_pkt = { 0 }, flush_pkt = { 0 };
1335  AVStream *st = s->streams[stream_index];
1336  uint8_t *data = pkt ? pkt->data : NULL;
1337  int size = pkt ? pkt->size : 0;
1338  int ret = 0, got_output = 0;
1339 
1340  if (!pkt) {
1342  pkt = &flush_pkt;
1343  got_output = 1;
1344  } else if (!size && st->parser->flags & PARSER_FLAG_COMPLETE_FRAMES) {
1345  // preserve 0-size sync packets
1347  }
1348 
1349  while (size > 0 || (pkt == &flush_pkt && got_output)) {
1350  int len;
1351  int64_t next_pts = pkt->pts;
1352  int64_t next_dts = pkt->dts;
1353 
1354  av_init_packet(&out_pkt);
1355  len = av_parser_parse2(st->parser, st->internal->avctx,
1356  &out_pkt.data, &out_pkt.size, data, size,
1357  pkt->pts, pkt->dts, pkt->pos);
1358 
1359  pkt->pts = pkt->dts = AV_NOPTS_VALUE;
1360  pkt->pos = -1;
1361  /* increment read pointer */
1362  data += len;
1363  size -= len;
1364 
1365  got_output = !!out_pkt.size;
1366 
1367  if (!out_pkt.size)
1368  continue;
1369 
1370  if (pkt->side_data) {
1371  out_pkt.side_data = pkt->side_data;
1372  out_pkt.side_data_elems = pkt->side_data_elems;
1373  pkt->side_data = NULL;
1374  pkt->side_data_elems = 0;
1375  }
1376 
1377  /* set the duration */
1378  out_pkt.duration = (st->parser->flags & PARSER_FLAG_COMPLETE_FRAMES) ? pkt->duration : 0;
1380  if (st->internal->avctx->sample_rate > 0) {
1381  out_pkt.duration =
1383  (AVRational) { 1, st->internal->avctx->sample_rate },
1384  st->time_base,
1385  AV_ROUND_DOWN);
1386  }
1387  }
1388 
1389  out_pkt.stream_index = st->index;
1390  out_pkt.pts = st->parser->pts;
1391  out_pkt.dts = st->parser->dts;
1392  out_pkt.pos = st->parser->pos;
1393 
1395  out_pkt.pos = st->parser->frame_offset;
1396 
1397  if (st->parser->key_frame == 1 ||
1398  (st->parser->key_frame == -1 &&
1400  out_pkt.flags |= AV_PKT_FLAG_KEY;
1401 
1402  if (st->parser->key_frame == -1 && st->parser->pict_type ==AV_PICTURE_TYPE_NONE && (pkt->flags&AV_PKT_FLAG_KEY))
1403  out_pkt.flags |= AV_PKT_FLAG_KEY;
1404 
1405  compute_pkt_fields(s, st, st->parser, &out_pkt, next_dts, next_pts);
1406 
1407  ret = add_to_pktbuf(&s->internal->parse_queue, &out_pkt,
1408  &s->internal->parse_queue_end, 1);
1409  av_packet_unref(&out_pkt);
1410  if (ret < 0)
1411  goto fail;
1412  }
1413 
1414  /* end of the stream => close and free the parser */
1415  if (pkt == &flush_pkt) {
1416  av_parser_close(st->parser);
1417  st->parser = NULL;
1418  }
1419 
1420 fail:
1422  return ret;
1423 }
1424 
1425 static int read_from_packet_buffer(AVPacketList **pkt_buffer,
1426  AVPacketList **pkt_buffer_end,
1427  AVPacket *pkt)
1428 {
1429  AVPacketList *pktl;
1430  av_assert0(*pkt_buffer);
1431  pktl = *pkt_buffer;
1432  *pkt = pktl->pkt;
1433  *pkt_buffer = pktl->next;
1434  if (!pktl->next)
1435  *pkt_buffer_end = NULL;
1436  av_freep(&pktl);
1437  return 0;
1438 }
1439 
1440 static int64_t ts_to_samples(AVStream *st, int64_t ts)
1441 {
1442  return av_rescale(ts, st->time_base.num * st->codecpar->sample_rate, st->time_base.den);
1443 }
1444 
1445 static int read_frame_internal(AVFormatContext *s, AVPacket *pkt)
1446 {
1447  int ret = 0, i, got_packet = 0;
1448  AVDictionary *metadata = NULL;
1449 
1450  av_init_packet(pkt);
1451 
1452  while (!got_packet && !s->internal->parse_queue) {
1453  AVStream *st;
1454  AVPacket cur_pkt;
1455 
1456  /* read next packet */
1457  ret = ff_read_packet(s, &cur_pkt);
1458  if (ret < 0) {
1459  if (ret == AVERROR(EAGAIN))
1460  return ret;
1461  /* flush the parsers */
1462  for (i = 0; i < s->nb_streams; i++) {
1463  st = s->streams[i];
1464  if (st->parser && st->need_parsing)
1465  parse_packet(s, NULL, st->index);
1466  }
1467  /* all remaining packets are now in parse_queue =>
1468  * really terminate parsing */
1469  break;
1470  }
1471  ret = 0;
1472  st = s->streams[cur_pkt.stream_index];
1473 
1474  /* update context if required */
1475  if (st->internal->need_context_update) {
1476  if (avcodec_is_open(st->internal->avctx)) {
1477  av_log(s, AV_LOG_DEBUG, "Demuxer context update while decoder is open, closing and trying to re-open\n");
1479  st->info->found_decoder = 0;
1480  }
1481 
1483  if (ret < 0)
1484  return ret;
1485 
1486  st->internal->need_context_update = 0;
1487  }
1488 
1489  if (cur_pkt.pts != AV_NOPTS_VALUE &&
1490  cur_pkt.dts != AV_NOPTS_VALUE &&
1491  cur_pkt.pts < cur_pkt.dts) {
1493  "Invalid timestamps stream=%d, pts=%s, dts=%s, size=%d\n",
1494  cur_pkt.stream_index,
1495  av_ts2str(cur_pkt.pts),
1496  av_ts2str(cur_pkt.dts),
1497  cur_pkt.size);
1498  }
1499  if (s->debug & FF_FDEBUG_TS)
1500  av_log(s, AV_LOG_DEBUG,
1501  "ff_read_packet stream=%d, pts=%s, dts=%s, size=%d, duration=%"PRId64", flags=%d\n",
1502  cur_pkt.stream_index,
1503  av_ts2str(cur_pkt.pts),
1504  av_ts2str(cur_pkt.dts),
1505  cur_pkt.size, cur_pkt.duration, cur_pkt.flags);
1506 
1507  if (st->need_parsing && !st->parser && !(s->flags & AVFMT_FLAG_NOPARSE)) {
1508  st->parser = av_parser_init(st->codecpar->codec_id);
1509  if (!st->parser) {
1510  av_log(s, AV_LOG_VERBOSE, "parser not found for codec "
1511  "%s, packets or times may be invalid.\n",
1513  /* no parser available: just output the raw packets */
1515  } else if (st->need_parsing == AVSTREAM_PARSE_HEADERS)
1517  else if (st->need_parsing == AVSTREAM_PARSE_FULL_ONCE)
1518  st->parser->flags |= PARSER_FLAG_ONCE;
1519  else if (st->need_parsing == AVSTREAM_PARSE_FULL_RAW)
1521  }
1522 
1523  if (!st->need_parsing || !st->parser) {
1524  /* no parsing needed: we just output the packet as is */
1525  *pkt = cur_pkt;
1527  if ((s->iformat->flags & AVFMT_GENERIC_INDEX) &&
1528  (pkt->flags & AV_PKT_FLAG_KEY) && pkt->dts != AV_NOPTS_VALUE) {
1529  ff_reduce_index(s, st->index);
1530  av_add_index_entry(st, pkt->pos, pkt->dts,
1531  0, 0, AVINDEX_KEYFRAME);
1532  }
1533  got_packet = 1;
1534  } else if (st->discard < AVDISCARD_ALL) {
1535  if ((ret = parse_packet(s, &cur_pkt, cur_pkt.stream_index)) < 0)
1536  return ret;
1538  st->codecpar->bit_rate = st->internal->avctx->bit_rate;
1539  st->codecpar->channels = st->internal->avctx->channels;
1541  st->codecpar->codec_id = st->internal->avctx->codec_id;
1542  } else {
1543  /* free packet */
1544  av_packet_unref(&cur_pkt);
1545  }
1546  if (pkt->flags & AV_PKT_FLAG_KEY)
1547  st->skip_to_keyframe = 0;
1548  if (st->skip_to_keyframe) {
1549  av_packet_unref(&cur_pkt);
1550  if (got_packet) {
1551  *pkt = cur_pkt;
1552  }
1553  got_packet = 0;
1554  }
1555  }
1556 
1557  if (!got_packet && s->internal->parse_queue)
1559 
1560  if (ret >= 0) {
1561  AVStream *st = s->streams[pkt->stream_index];
1562  int discard_padding = 0;
1563  if (st->first_discard_sample && pkt->pts != AV_NOPTS_VALUE) {
1564  int64_t pts = pkt->pts - (is_relative(pkt->pts) ? RELATIVE_TS_BASE : 0);
1565  int64_t sample = ts_to_samples(st, pts);
1566  int duration = ts_to_samples(st, pkt->duration);
1567  int64_t end_sample = sample + duration;
1568  if (duration > 0 && end_sample >= st->first_discard_sample &&
1569  sample < st->last_discard_sample)
1570  discard_padding = FFMIN(end_sample - st->first_discard_sample, duration);
1571  }
1572  if (st->start_skip_samples && (pkt->pts == 0 || pkt->pts == RELATIVE_TS_BASE))
1573  st->skip_samples = st->start_skip_samples;
1574  if (st->skip_samples || discard_padding) {
1576  if (p) {
1577  AV_WL32(p, st->skip_samples);
1578  AV_WL32(p + 4, discard_padding);
1579  av_log(s, AV_LOG_DEBUG, "demuxer injecting skip %d / discard %d\n", st->skip_samples, discard_padding);
1580  }
1581  st->skip_samples = 0;
1582  }
1583 
1584  if (st->inject_global_side_data) {
1585  for (i = 0; i < st->nb_side_data; i++) {
1586  AVPacketSideData *src_sd = &st->side_data[i];
1587  uint8_t *dst_data;
1588 
1589  if (av_packet_get_side_data(pkt, src_sd->type, NULL))
1590  continue;
1591 
1592  dst_data = av_packet_new_side_data(pkt, src_sd->type, src_sd->size);
1593  if (!dst_data) {
1594  av_log(s, AV_LOG_WARNING, "Could not inject global side data\n");
1595  continue;
1596  }
1597 
1598  memcpy(dst_data, src_sd->data, src_sd->size);
1599  }
1600  st->inject_global_side_data = 0;
1601  }
1602 
1603  if (!(s->flags & AVFMT_FLAG_KEEP_SIDE_DATA))
1605  }
1606 
1607  av_opt_get_dict_val(s, "metadata", AV_OPT_SEARCH_CHILDREN, &metadata);
1608  if (metadata) {
1610  av_dict_copy(&s->metadata, metadata, 0);
1611  av_dict_free(&metadata);
1613  }
1614 
1615 #if FF_API_LAVF_AVCTX
1617 #endif
1618 
1619  if (s->debug & FF_FDEBUG_TS)
1620  av_log(s, AV_LOG_DEBUG,
1621  "read_frame_internal stream=%d, pts=%s, dts=%s, "
1622  "size=%d, duration=%"PRId64", flags=%d\n",
1623  pkt->stream_index,
1624  av_ts2str(pkt->pts),
1625  av_ts2str(pkt->dts),
1626  pkt->size, pkt->duration, pkt->flags);
1627 
1628  return ret;
1629 }
1630 
1631 int av_read_frame(AVFormatContext *s, AVPacket *pkt)
1632 {
1633  const int genpts = s->flags & AVFMT_FLAG_GENPTS;
1634  int eof = 0;
1635  int ret;
1636  AVStream *st;
1637 
1638  if (!genpts) {
1639  ret = s->internal->packet_buffer
1641  &s->internal->packet_buffer_end, pkt)
1642  : read_frame_internal(s, pkt);
1643  if (ret < 0)
1644  return ret;
1645  goto return_packet;
1646  }
1647 
1648  for (;;) {
1649  AVPacketList *pktl = s->internal->packet_buffer;
1650 
1651  if (pktl) {
1652  AVPacket *next_pkt = &pktl->pkt;
1653 
1654  if (next_pkt->dts != AV_NOPTS_VALUE) {
1655  int wrap_bits = s->streams[next_pkt->stream_index]->pts_wrap_bits;
1656  // last dts seen for this stream. if any of packets following
1657  // current one had no dts, we will set this to AV_NOPTS_VALUE.
1658  int64_t last_dts = next_pkt->dts;
1659  while (pktl && next_pkt->pts == AV_NOPTS_VALUE) {
1660  if (pktl->pkt.stream_index == next_pkt->stream_index &&
1661  (av_compare_mod(next_pkt->dts, pktl->pkt.dts, 2LL << (wrap_bits - 1)) < 0)) {
1662  if (av_compare_mod(pktl->pkt.pts, pktl->pkt.dts, 2LL << (wrap_bits - 1))) {
1663  // not B-frame
1664  next_pkt->pts = pktl->pkt.dts;
1665  }
1666  if (last_dts != AV_NOPTS_VALUE) {
1667  // Once last dts was set to AV_NOPTS_VALUE, we don't change it.
1668  last_dts = pktl->pkt.dts;
1669  }
1670  }
1671  pktl = pktl->next;
1672  }
1673  if (eof && next_pkt->pts == AV_NOPTS_VALUE && last_dts != AV_NOPTS_VALUE) {
1674  // Fixing the last reference frame had none pts issue (For MXF etc).
1675  // We only do this when
1676  // 1. eof.
1677  // 2. we are not able to resolve a pts value for current packet.
1678  // 3. the packets for this stream at the end of the files had valid dts.
1679  next_pkt->pts = last_dts + next_pkt->duration;
1680  }
1681  pktl = s->internal->packet_buffer;
1682  }
1683 
1684  /* read packet from packet buffer, if there is data */
1685  st = s->streams[next_pkt->stream_index];
1686  if (!(next_pkt->pts == AV_NOPTS_VALUE && st->discard < AVDISCARD_ALL &&
1687  next_pkt->dts != AV_NOPTS_VALUE && !eof)) {
1689  &s->internal->packet_buffer_end, pkt);
1690  goto return_packet;
1691  }
1692  }
1693 
1694  ret = read_frame_internal(s, pkt);
1695  if (ret < 0) {
1696  if (pktl && ret != AVERROR(EAGAIN)) {
1697  eof = 1;
1698  continue;
1699  } else
1700  return ret;
1701  }
1702 
1703  ret = add_to_pktbuf(&s->internal->packet_buffer, pkt,
1704  &s->internal->packet_buffer_end, 1);
1705  av_packet_unref(pkt);
1706  if (ret < 0)
1707  return ret;
1708  }
1709 
1710 return_packet:
1711 
1712  st = s->streams[pkt->stream_index];
1713  if ((s->iformat->flags & AVFMT_GENERIC_INDEX) && pkt->flags & AV_PKT_FLAG_KEY) {
1714  ff_reduce_index(s, st->index);
1715  av_add_index_entry(st, pkt->pos, pkt->dts, 0, 0, AVINDEX_KEYFRAME);
1716  }
1717 
1718  if (is_relative(pkt->dts))
1719  pkt->dts -= RELATIVE_TS_BASE;
1720  if (is_relative(pkt->pts))
1721  pkt->pts -= RELATIVE_TS_BASE;
1722 
1723  return ret;
1724 }
1725 
1726 /* XXX: suppress the packet queue */
1727 static void flush_packet_queue(AVFormatContext *s)
1728 {
1729  if (!s->internal)
1730  return;
1734 
1736 }
1737 
1738 /*******************************************************/
1739 /* seek support */
1740 
1741 int av_find_default_stream_index(AVFormatContext *s)
1742 {
1743  int i;
1744  AVStream *st;
1745  int best_stream = 0;
1746  int best_score = INT_MIN;
1747 
1748  if (s->nb_streams <= 0)
1749  return -1;
1750  for (i = 0; i < s->nb_streams; i++) {
1751  int score = 0;
1752  st = s->streams[i];
1753  if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
1755  score -= 400;
1756  if (st->codecpar->width && st->codecpar->height)
1757  score += 50;
1758  score+= 25;
1759  }
1760  if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
1761  if (st->codecpar->sample_rate)
1762  score += 50;
1763  }
1764  if (st->codec_info_nb_frames)
1765  score += 12;
1766 
1767  if (st->discard != AVDISCARD_ALL)
1768  score += 200;
1769 
1770  if (score > best_score) {
1771  best_score = score;
1772  best_stream = i;
1773  }
1774  }
1775  return best_stream;
1776 }
1777 
1778 /** Flush the frame reader. */
1779 void ff_read_frame_flush(AVFormatContext *s)
1780 {
1781  AVStream *st;
1782  int i, j;
1783 
1784  flush_packet_queue(s);
1785 
1786  /* Reset read state for each stream. */
1787  for (i = 0; i < s->nb_streams; i++) {
1788  st = s->streams[i];
1789 
1790  if (st->parser) {
1791  av_parser_close(st->parser);
1792  st->parser = NULL;
1793  }
1796  if (st->first_dts == AV_NOPTS_VALUE)
1797  st->cur_dts = RELATIVE_TS_BASE;
1798  else
1799  /* We set the current DTS to an unspecified origin. */
1800  st->cur_dts = AV_NOPTS_VALUE;
1801 
1803 
1804  for (j = 0; j < MAX_REORDER_DELAY + 1; j++)
1805  st->pts_buffer[j] = AV_NOPTS_VALUE;
1806 
1808  st->inject_global_side_data = 1;
1809 
1810  st->skip_samples = 0;
1811  }
1812 }
1813 
1814 void ff_update_cur_dts(AVFormatContext *s, AVStream *ref_st, int64_t timestamp)
1815 {
1816  int i;
1817 
1818  for (i = 0; i < s->nb_streams; i++) {
1819  AVStream *st = s->streams[i];
1820 
1821  st->cur_dts =
1822  av_rescale(timestamp,
1823  st->time_base.den * (int64_t) ref_st->time_base.num,
1824  st->time_base.num * (int64_t) ref_st->time_base.den);
1825  }
1826 }
1827 
1828 void ff_reduce_index(AVFormatContext *s, int stream_index)
1829 {
1830  AVStream *st = s->streams[stream_index];
1831  unsigned int max_entries = s->max_index_size / sizeof(AVIndexEntry);
1832 
1833  if ((unsigned) st->nb_index_entries >= max_entries) {
1834  int i;
1835  for (i = 0; 2 * i < st->nb_index_entries; i++)
1836  st->index_entries[i] = st->index_entries[2 * i];
1837  st->nb_index_entries = i;
1838  }
1839 }
1840 
1841 int ff_add_index_entry(AVIndexEntry **index_entries,
1842  int *nb_index_entries,
1843  unsigned int *index_entries_allocated_size,
1844  int64_t pos, int64_t timestamp,
1845  int size, int distance, int flags)
1846 {
1847  AVIndexEntry *entries, *ie;
1848  int index;
1849 
1850  if ((unsigned) *nb_index_entries + 1 >= UINT_MAX / sizeof(AVIndexEntry))
1851  return -1;
1852 
1853  if (timestamp == AV_NOPTS_VALUE)
1854  return AVERROR(EINVAL);
1855 
1856  if (size < 0 || size > 0x3FFFFFFF)
1857  return AVERROR(EINVAL);
1858 
1859  if (is_relative(timestamp)) //FIXME this maintains previous behavior but we should shift by the correct offset once known
1860  timestamp -= RELATIVE_TS_BASE;
1861 
1862  entries = av_fast_realloc(*index_entries,
1863  index_entries_allocated_size,
1864  (*nb_index_entries + 1) *
1865  sizeof(AVIndexEntry));
1866  if (!entries)
1867  return -1;
1868 
1869  *index_entries = entries;
1870 
1871  index = ff_index_search_timestamp(*index_entries, *nb_index_entries,
1872  timestamp, AVSEEK_FLAG_ANY);
1873 
1874  if (index < 0) {
1875  index = (*nb_index_entries)++;
1876  ie = &entries[index];
1877  av_assert0(index == 0 || ie[-1].timestamp < timestamp);
1878  } else {
1879  ie = &entries[index];
1880  if (ie->timestamp != timestamp) {
1881  if (ie->timestamp <= timestamp)
1882  return -1;
1883  memmove(entries + index + 1, entries + index,
1884  sizeof(AVIndexEntry) * (*nb_index_entries - index));
1885  (*nb_index_entries)++;
1886  } else if (ie->pos == pos && distance < ie->min_distance)
1887  // do not reduce the distance
1888  distance = ie->min_distance;
1889  }
1890 
1891  ie->pos = pos;
1892  ie->timestamp = timestamp;
1893  ie->min_distance = distance;
1894  ie->size = size;
1895  ie->flags = flags;
1896 
1897  return index;
1898 }
1899 
1900 int av_add_index_entry(AVStream *st, int64_t pos, int64_t timestamp,
1901  int size, int distance, int flags)
1902 {
1903  timestamp = wrap_timestamp(st, timestamp);
1905  &st->index_entries_allocated_size, pos,
1906  timestamp, size, distance, flags);
1907 }
1908 
1909 int ff_index_search_timestamp(const AVIndexEntry *entries, int nb_entries,
1910  int64_t wanted_timestamp, int flags)
1911 {
1912  int a, b, m;
1913  int64_t timestamp;
1914 
1915  a = -1;
1916  b = nb_entries;
1917 
1918  // Optimize appending index entries at the end.
1919  if (b && entries[b - 1].timestamp < wanted_timestamp)
1920  a = b - 1;
1921 
1922  while (b - a > 1) {
1923  m = (a + b) >> 1;
1924  timestamp = entries[m].timestamp;
1925  if (timestamp >= wanted_timestamp)
1926  b = m;
1927  if (timestamp <= wanted_timestamp)
1928  a = m;
1929  }
1930  m = (flags & AVSEEK_FLAG_BACKWARD) ? a : b;
1931 
1932  if (!(flags & AVSEEK_FLAG_ANY))
1933  while (m >= 0 && m < nb_entries &&
1934  !(entries[m].flags & AVINDEX_KEYFRAME))
1935  m += (flags & AVSEEK_FLAG_BACKWARD) ? -1 : 1;
1936 
1937  if (m == nb_entries)
1938  return -1;
1939  return m;
1940 }
1941 
1942 void ff_configure_buffers_for_index(AVFormatContext *s, int64_t time_tolerance)
1943 {
1944  int ist1, ist2;
1945  int64_t pos_delta = 0;
1946  int64_t skip = 0;
1947  //We could use URLProtocol flags here but as many user applications do not use URLProtocols this would be unreliable
1948  const char *proto = avio_find_protocol_name(s->filename);
1949 
1950  if (!proto) {
1951  av_log(s, AV_LOG_INFO,
1952  "Protocol name not provided, cannot determine if input is local or "
1953  "a network protocol, buffers and access patterns cannot be configured "
1954  "optimally without knowing the protocol\n");
1955  }
1956 
1957  if (proto && !(strcmp(proto, "file") && strcmp(proto, "pipe") && strcmp(proto, "cache")))
1958  return;
1959 
1960  for (ist1 = 0; ist1 < s->nb_streams; ist1++) {
1961  AVStream *st1 = s->streams[ist1];
1962  for (ist2 = 0; ist2 < s->nb_streams; ist2++) {
1963  AVStream *st2 = s->streams[ist2];
1964  int i1, i2;
1965 
1966  if (ist1 == ist2)
1967  continue;
1968 
1969  for (i1 = i2 = 0; i1 < st1->nb_index_entries; i1++) {
1970  AVIndexEntry *e1 = &st1->index_entries[i1];
1971  int64_t e1_pts = av_rescale_q(e1->timestamp, st1->time_base, AV_TIME_BASE_Q);
1972 
1973  skip = FFMAX(skip, e1->size);
1974  for (; i2 < st2->nb_index_entries; i2++) {
1975  AVIndexEntry *e2 = &st2->index_entries[i2];
1976  int64_t e2_pts = av_rescale_q(e2->timestamp, st2->time_base, AV_TIME_BASE_Q);
1977  if (e2_pts - e1_pts < time_tolerance)
1978  continue;
1979  pos_delta = FFMAX(pos_delta, e1->pos - e2->pos);
1980  break;
1981  }
1982  }
1983  }
1984  }
1985 
1986  pos_delta *= 2;
1987  /* XXX This could be adjusted depending on protocol*/
1988  if (s->pb->buffer_size < pos_delta && pos_delta < (1<<24)) {
1989  av_log(s, AV_LOG_VERBOSE, "Reconfiguring buffers to size %"PRId64"\n", pos_delta);
1990  ffio_set_buf_size(s->pb, pos_delta);
1991  s->pb->short_seek_threshold = FFMAX(s->pb->short_seek_threshold, pos_delta/2);
1992  }
1993 
1994  if (skip < (1<<23)) {
1996  }
1997 }
1998 
1999 int av_index_search_timestamp(AVStream *st, int64_t wanted_timestamp, int flags)
2000 {
2002  wanted_timestamp, flags);
2003 }
2004 
2005 static int64_t ff_read_timestamp(AVFormatContext *s, int stream_index, int64_t *ppos, int64_t pos_limit,
2006  int64_t (*read_timestamp)(struct AVFormatContext *, int , int64_t *, int64_t ))
2007 {
2008  int64_t ts = read_timestamp(s, stream_index, ppos, pos_limit);
2009  if (stream_index >= 0)
2010  ts = wrap_timestamp(s->streams[stream_index], ts);
2011  return ts;
2012 }
2013 
2014 int ff_seek_frame_binary(AVFormatContext *s, int stream_index,
2015  int64_t target_ts, int flags)
2016 {
2017  AVInputFormat *avif = s->iformat;
2018  int64_t av_uninit(pos_min), av_uninit(pos_max), pos, pos_limit;
2019  int64_t ts_min, ts_max, ts;
2020  int index;
2021  int64_t ret;
2022  AVStream *st;
2023 
2024  if (stream_index < 0)
2025  return -1;
2026 
2027  av_log(s, AV_LOG_TRACE, "read_seek: %d %s\n", stream_index, av_ts2str(target_ts));
2028 
2029  ts_max =
2030  ts_min = AV_NOPTS_VALUE;
2031  pos_limit = -1; // GCC falsely says it may be uninitialized.
2032 
2033  st = s->streams[stream_index];
2034  if (st->index_entries) {
2035  AVIndexEntry *e;
2036 
2037  /* FIXME: Whole function must be checked for non-keyframe entries in
2038  * index case, especially read_timestamp(). */
2039  index = av_index_search_timestamp(st, target_ts,
2040  flags | AVSEEK_FLAG_BACKWARD);
2041  index = FFMAX(index, 0);
2042  e = &st->index_entries[index];
2043 
2044  if (e->timestamp <= target_ts || e->pos == e->min_distance) {
2045  pos_min = e->pos;
2046  ts_min = e->timestamp;
2047  av_log(s, AV_LOG_TRACE, "using cached pos_min=0x%"PRIx64" dts_min=%s\n",
2048  pos_min, av_ts2str(ts_min));
2049  } else {
2050  av_assert1(index == 0);
2051  }
2052 
2053  index = av_index_search_timestamp(st, target_ts,
2054  flags & ~AVSEEK_FLAG_BACKWARD);
2055  av_assert0(index < st->nb_index_entries);
2056  if (index >= 0) {
2057  e = &st->index_entries[index];
2058  av_assert1(e->timestamp >= target_ts);
2059  pos_max = e->pos;
2060  ts_max = e->timestamp;
2061  pos_limit = pos_max - e->min_distance;
2062  av_log(s, AV_LOG_TRACE, "using cached pos_max=0x%"PRIx64" pos_limit=0x%"PRIx64
2063  " dts_max=%s\n", pos_max, pos_limit, av_ts2str(ts_max));
2064  }
2065  }
2066 
2067  pos = ff_gen_search(s, stream_index, target_ts, pos_min, pos_max, pos_limit,
2068  ts_min, ts_max, flags, &ts, avif->read_timestamp);
2069  if (pos < 0)
2070  return -1;
2071 
2072  /* do the seek */
2073  if ((ret = avio_seek(s->pb, pos, SEEK_SET)) < 0)
2074  return ret;
2075 
2077  ff_update_cur_dts(s, st, ts);
2078 
2079  return 0;
2080 }
2081 
2082 int ff_find_last_ts(AVFormatContext *s, int stream_index, int64_t *ts, int64_t *pos,
2083  int64_t (*read_timestamp)(struct AVFormatContext *, int , int64_t *, int64_t ))
2084 {
2085  int64_t step = 1024;
2086  int64_t limit, ts_max;
2087  int64_t filesize = avio_size(s->pb);
2088  int64_t pos_max = filesize - 1;
2089  do {
2090  limit = pos_max;
2091  pos_max = FFMAX(0, (pos_max) - step);
2092  ts_max = ff_read_timestamp(s, stream_index,
2093  &pos_max, limit, read_timestamp);
2094  step += step;
2095  } while (ts_max == AV_NOPTS_VALUE && 2*limit > step);
2096  if (ts_max == AV_NOPTS_VALUE)
2097  return -1;
2098 
2099  for (;;) {
2100  int64_t tmp_pos = pos_max + 1;
2101  int64_t tmp_ts = ff_read_timestamp(s, stream_index,
2102  &tmp_pos, INT64_MAX, read_timestamp);
2103  if (tmp_ts == AV_NOPTS_VALUE)
2104  break;
2105  av_assert0(tmp_pos > pos_max);
2106  ts_max = tmp_ts;
2107  pos_max = tmp_pos;
2108  if (tmp_pos >= filesize)
2109  break;
2110  }
2111 
2112  if (ts)
2113  *ts = ts_max;
2114  if (pos)
2115  *pos = pos_max;
2116 
2117  return 0;
2118 }
2119 
2120 int64_t ff_gen_search(AVFormatContext *s, int stream_index, int64_t target_ts,
2121  int64_t pos_min, int64_t pos_max, int64_t pos_limit,
2122  int64_t ts_min, int64_t ts_max,
2123  int flags, int64_t *ts_ret,
2124  int64_t (*read_timestamp)(struct AVFormatContext *, int,
2125  int64_t *, int64_t))
2126 {
2127  int64_t pos, ts;
2128  int64_t start_pos;
2129  int no_change;
2130  int ret;
2131 
2132  av_log(s, AV_LOG_TRACE, "gen_seek: %d %s\n", stream_index, av_ts2str(target_ts));
2133 
2134  if (ts_min == AV_NOPTS_VALUE) {
2135  pos_min = s->internal->data_offset;
2136  ts_min = ff_read_timestamp(s, stream_index, &pos_min, INT64_MAX, read_timestamp);
2137  if (ts_min == AV_NOPTS_VALUE)
2138  return -1;
2139  }
2140 
2141  if (ts_min >= target_ts) {
2142  *ts_ret = ts_min;
2143  return pos_min;
2144  }
2145 
2146  if (ts_max == AV_NOPTS_VALUE) {
2147  if ((ret = ff_find_last_ts(s, stream_index, &ts_max, &pos_max, read_timestamp)) < 0)
2148  return ret;
2149  pos_limit = pos_max;
2150  }
2151 
2152  if (ts_max <= target_ts) {
2153  *ts_ret = ts_max;
2154  return pos_max;
2155  }
2156 
2157  av_assert0(ts_min < ts_max);
2158 
2159  no_change = 0;
2160  while (pos_min < pos_limit) {
2161  av_log(s, AV_LOG_TRACE,
2162  "pos_min=0x%"PRIx64" pos_max=0x%"PRIx64" dts_min=%s dts_max=%s\n",
2163  pos_min, pos_max, av_ts2str(ts_min), av_ts2str(ts_max));
2164  av_assert0(pos_limit <= pos_max);
2165 
2166  if (no_change == 0) {
2167  int64_t approximate_keyframe_distance = pos_max - pos_limit;
2168  // interpolate position (better than dichotomy)
2169  pos = av_rescale(target_ts - ts_min, pos_max - pos_min,
2170  ts_max - ts_min) +
2171  pos_min - approximate_keyframe_distance;
2172  } else if (no_change == 1) {
2173  // bisection if interpolation did not change min / max pos last time
2174  pos = (pos_min + pos_limit) >> 1;
2175  } else {
2176  /* linear search if bisection failed, can only happen if there
2177  * are very few or no keyframes between min/max */
2178  pos = pos_min;
2179  }
2180  if (pos <= pos_min)
2181  pos = pos_min + 1;
2182  else if (pos > pos_limit)
2183  pos = pos_limit;
2184  start_pos = pos;
2185 
2186  // May pass pos_limit instead of -1.
2187  ts = ff_read_timestamp(s, stream_index, &pos, INT64_MAX, read_timestamp);
2188  if (pos == pos_max)
2189  no_change++;
2190  else
2191  no_change = 0;
2192  av_log(s, AV_LOG_TRACE, "%"PRId64" %"PRId64" %"PRId64" / %s %s %s"
2193  " target:%s limit:%"PRId64" start:%"PRId64" noc:%d\n",
2194  pos_min, pos, pos_max,
2195  av_ts2str(ts_min), av_ts2str(ts), av_ts2str(ts_max), av_ts2str(target_ts),
2196  pos_limit, start_pos, no_change);
2197  if (ts == AV_NOPTS_VALUE) {
2198  av_log(s, AV_LOG_ERROR, "read_timestamp() failed in the middle\n");
2199  return -1;
2200  }
2201  if (target_ts <= ts) {
2202  pos_limit = start_pos - 1;
2203  pos_max = pos;
2204  ts_max = ts;
2205  }
2206  if (target_ts >= ts) {
2207  pos_min = pos;
2208  ts_min = ts;
2209  }
2210  }
2211 
2212  pos = (flags & AVSEEK_FLAG_BACKWARD) ? pos_min : pos_max;
2213  ts = (flags & AVSEEK_FLAG_BACKWARD) ? ts_min : ts_max;
2214 #if 0
2215  pos_min = pos;
2216  ts_min = ff_read_timestamp(s, stream_index, &pos_min, INT64_MAX, read_timestamp);
2217  pos_min++;
2218  ts_max = ff_read_timestamp(s, stream_index, &pos_min, INT64_MAX, read_timestamp);
2219  av_log(s, AV_LOG_TRACE, "pos=0x%"PRIx64" %s<=%s<=%s\n",
2220  pos, av_ts2str(ts_min), av_ts2str(target_ts), av_ts2str(ts_max));
2221 #endif
2222  *ts_ret = ts;
2223  return pos;
2224 }
2225 
2226 static int seek_frame_byte(AVFormatContext *s, int stream_index,
2227  int64_t pos, int flags)
2228 {
2229  int64_t pos_min, pos_max;
2230 
2231  pos_min = s->internal->data_offset;
2232  pos_max = avio_size(s->pb) - 1;
2233 
2234  if (pos < pos_min)
2235  pos = pos_min;
2236  else if (pos > pos_max)
2237  pos = pos_max;
2238 
2239  avio_seek(s->pb, pos, SEEK_SET);
2240 
2241  s->io_repositioned = 1;
2242 
2243  return 0;
2244 }
2245 
2246 static int seek_frame_generic(AVFormatContext *s, int stream_index,
2247  int64_t timestamp, int flags)
2248 {
2249  int index;
2250  int64_t ret;
2251  AVStream *st;
2252  AVIndexEntry *ie;
2253 
2254  st = s->streams[stream_index];
2255 
2256  index = av_index_search_timestamp(st, timestamp, flags);
2257 
2258  if (index < 0 && st->nb_index_entries &&
2259  timestamp < st->index_entries[0].timestamp)
2260  return -1;
2261 
2262  if (index < 0 || index == st->nb_index_entries - 1) {
2263  AVPacket pkt;
2264  int nonkey = 0;
2265 
2266  if (st->nb_index_entries) {
2268  ie = &st->index_entries[st->nb_index_entries - 1];
2269  if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
2270  return ret;
2271  ff_update_cur_dts(s, st, ie->timestamp);
2272  } else {
2273  if ((ret = avio_seek(s->pb, s->internal->data_offset, SEEK_SET)) < 0)
2274  return ret;
2275  }
2276  for (;;) {
2277  int read_status;
2278  do {
2279  read_status = av_read_frame(s, &pkt);
2280  } while (read_status == AVERROR(EAGAIN));
2281  if (read_status < 0)
2282  break;
2283  if (stream_index == pkt.stream_index && pkt.dts > timestamp) {
2284  if (pkt.flags & AV_PKT_FLAG_KEY) {
2285  av_packet_unref(&pkt);
2286  break;
2287  }
2288  if (nonkey++ > 1000 && st->codecpar->codec_id != AV_CODEC_ID_CDGRAPHICS) {
2289  av_log(s, AV_LOG_ERROR,"seek_frame_generic failed as this stream seems to contain no keyframes after the target timestamp, %d non keyframes found\n", nonkey);
2290  av_packet_unref(&pkt);
2291  break;
2292  }
2293  }
2294  av_packet_unref(&pkt);
2295  }
2296  index = av_index_search_timestamp(st, timestamp, flags);
2297  }
2298  if (index < 0)
2299  return -1;
2300 
2302  if (s->iformat->read_seek)
2303  if (s->iformat->read_seek(s, stream_index, timestamp, flags) >= 0)
2304  return 0;
2305  ie = &st->index_entries[index];
2306  if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
2307  return ret;
2308  ff_update_cur_dts(s, st, ie->timestamp);
2309 
2310  return 0;
2311 }
2312 
2313 static int seek_frame_internal(AVFormatContext *s, int stream_index,
2314  int64_t timestamp, int flags)
2315 {
2316  int ret;
2317  AVStream *st;
2318 
2319  if (flags & AVSEEK_FLAG_BYTE) {
2320  if (s->iformat->flags & AVFMT_NO_BYTE_SEEK)
2321  return -1;
2323  return seek_frame_byte(s, stream_index, timestamp, flags);
2324  }
2325 
2326  if (stream_index < 0) {
2327  stream_index = av_find_default_stream_index(s);
2328  if (stream_index < 0)
2329  return -1;
2330 
2331  st = s->streams[stream_index];
2332  /* timestamp for default must be expressed in AV_TIME_BASE units */
2333  timestamp = av_rescale(timestamp, st->time_base.den,
2334  AV_TIME_BASE * (int64_t) st->time_base.num);
2335  }
2336 
2337  /* first, we try the format specific seek */
2338  if (s->iformat->read_seek) {
2340  ret = s->iformat->read_seek(s, stream_index, timestamp, flags);
2341  } else
2342  ret = -1;
2343  if (ret >= 0)
2344  return 0;
2345 
2346  if (s->iformat->read_timestamp &&
2347  !(s->iformat->flags & AVFMT_NOBINSEARCH)) {
2349  return ff_seek_frame_binary(s, stream_index, timestamp, flags);
2350  } else if (!(s->iformat->flags & AVFMT_NOGENSEARCH)) {
2352  return seek_frame_generic(s, stream_index, timestamp, flags);
2353  } else
2354  return -1;
2355 }
2356 
2357 int av_seek_frame(AVFormatContext *s, int stream_index,
2358  int64_t timestamp, int flags)
2359 {
2360  int ret;
2361 
2362  if (s->iformat->read_seek2 && !s->iformat->read_seek) {
2363  int64_t min_ts = INT64_MIN, max_ts = INT64_MAX;
2364  if ((flags & AVSEEK_FLAG_BACKWARD))
2365  max_ts = timestamp;
2366  else
2367  min_ts = timestamp;
2368  return avformat_seek_file(s, stream_index, min_ts, timestamp, max_ts,
2369  flags & ~AVSEEK_FLAG_BACKWARD);
2370  }
2371 
2372  ret = seek_frame_internal(s, stream_index, timestamp, flags);
2373 
2374  if (ret >= 0)
2376 
2377  return ret;
2378 }
2379 
2380 int avformat_seek_file(AVFormatContext *s, int stream_index, int64_t min_ts,
2381  int64_t ts, int64_t max_ts, int flags)
2382 {
2383  if (min_ts > ts || max_ts < ts)
2384  return -1;
2385  if (stream_index < -1 || stream_index >= (int)s->nb_streams)
2386  return AVERROR(EINVAL);
2387 
2388  if (s->seek2any>0)
2389  flags |= AVSEEK_FLAG_ANY;
2390  flags &= ~AVSEEK_FLAG_BACKWARD;
2391 
2392  if (s->iformat->read_seek2) {
2393  int ret;
2395 
2396  if (stream_index == -1 && s->nb_streams == 1) {
2397  AVRational time_base = s->streams[0]->time_base;
2398  ts = av_rescale_q(ts, AV_TIME_BASE_Q, time_base);
2399  min_ts = av_rescale_rnd(min_ts, time_base.den,
2400  time_base.num * (int64_t)AV_TIME_BASE,
2402  max_ts = av_rescale_rnd(max_ts, time_base.den,
2403  time_base.num * (int64_t)AV_TIME_BASE,
2405  }
2406 
2407  ret = s->iformat->read_seek2(s, stream_index, min_ts,
2408  ts, max_ts, flags);
2409 
2410  if (ret >= 0)
2412  return ret;
2413  }
2414 
2415  if (s->iformat->read_timestamp) {
2416  // try to seek via read_timestamp()
2417  }
2418 
2419  // Fall back on old API if new is not implemented but old is.
2420  // Note the old API has somewhat different semantics.
2421  if (s->iformat->read_seek || 1) {
2422  int dir = (ts - (uint64_t)min_ts > (uint64_t)max_ts - ts ? AVSEEK_FLAG_BACKWARD : 0);
2423  int ret = av_seek_frame(s, stream_index, ts, flags | dir);
2424  if (ret<0 && ts != min_ts && max_ts != ts) {
2425  ret = av_seek_frame(s, stream_index, dir ? max_ts : min_ts, flags | dir);
2426  if (ret >= 0)
2427  ret = av_seek_frame(s, stream_index, ts, flags | (dir^AVSEEK_FLAG_BACKWARD));
2428  }
2429  return ret;
2430  }
2431 
2432  // try some generic seek like seek_frame_generic() but with new ts semantics
2433  return -1; //unreachable
2434 }
2435 
2436 int avformat_flush(AVFormatContext *s)
2437 {
2439  return 0;
2440 }
2441 
2442 /*******************************************************/
2443 
2444 /**
2445  * Return TRUE if the stream has accurate duration in any stream.
2446  *
2447  * @return TRUE if the stream has accurate duration for at least one component.
2448  */
2449 static int has_duration(AVFormatContext *ic)
2450 {
2451  int i;
2452  AVStream *st;
2453 
2454  for (i = 0; i < ic->nb_streams; i++) {
2455  st = ic->streams[i];
2456  if (st->duration != AV_NOPTS_VALUE)
2457  return 1;
2458  }
2459  if (ic->duration != AV_NOPTS_VALUE)
2460  return 1;
2461  return 0;
2462 }
2463 
2464 /**
2465  * Estimate the stream timings from the one of each components.
2466  *
2467  * Also computes the global bitrate if possible.
2468  */
2469 static void update_stream_timings(AVFormatContext *ic)
2470 {
2471  int64_t start_time, start_time1, start_time_text, end_time, end_time1;
2472  int64_t duration, duration1, filesize;
2473  int i;
2474  AVStream *st;
2475  AVProgram *p;
2476 
2477  start_time = INT64_MAX;
2478  start_time_text = INT64_MAX;
2479  end_time = INT64_MIN;
2480  duration = INT64_MIN;
2481  for (i = 0; i < ic->nb_streams; i++) {
2482  st = ic->streams[i];
2483  if (st->start_time != AV_NOPTS_VALUE && st->time_base.den) {
2484  start_time1 = av_rescale_q(st->start_time, st->time_base,
2485  AV_TIME_BASE_Q);
2487  if (start_time1 < start_time_text)
2488  start_time_text = start_time1;
2489  } else
2490  start_time = FFMIN(start_time, start_time1);
2491  end_time1 = av_rescale_q_rnd(st->duration, st->time_base,
2494  if (end_time1 != AV_NOPTS_VALUE) {
2495  end_time1 += start_time1;
2496  end_time = FFMAX(end_time, end_time1);
2497  }
2498  for (p = NULL; (p = av_find_program_from_stream(ic, p, i)); ) {
2499  if (p->start_time == AV_NOPTS_VALUE || p->start_time > start_time1)
2500  p->start_time = start_time1;
2501  if (p->end_time < end_time1)
2502  p->end_time = end_time1;
2503  }
2504  }
2505  if (st->duration != AV_NOPTS_VALUE) {
2506  duration1 = av_rescale_q(st->duration, st->time_base,
2507  AV_TIME_BASE_Q);
2508  duration = FFMAX(duration, duration1);
2509  }
2510  }
2511  if (start_time == INT64_MAX || (start_time > start_time_text && start_time - start_time_text < AV_TIME_BASE))
2512  start_time = start_time_text;
2513  else if (start_time > start_time_text)
2514  av_log(ic, AV_LOG_VERBOSE, "Ignoring outlier non primary stream starttime %f\n", start_time_text / (float)AV_TIME_BASE);
2515 
2516  if (start_time != INT64_MAX) {
2517  ic->start_time = start_time;
2518  if (end_time != INT64_MIN) {
2519  if (ic->nb_programs) {
2520  for (i = 0; i < ic->nb_programs; i++) {
2521  p = ic->programs[i];
2522  if (p->start_time != AV_NOPTS_VALUE && p->end_time > p->start_time)
2523  duration = FFMAX(duration, p->end_time - p->start_time);
2524  }
2525  } else
2526  duration = FFMAX(duration, end_time - start_time);
2527  }
2528  }
2529  if (duration != INT64_MIN && duration > 0 && ic->duration == AV_NOPTS_VALUE) {
2530  ic->duration = duration;
2531  }
2532  if (ic->pb && (filesize = avio_size(ic->pb)) > 0 && ic->duration > 0) {
2533  /* compute the bitrate */
2534  double bitrate = (double) filesize * 8.0 * AV_TIME_BASE /
2535  (double) ic->duration;
2536  if (bitrate >= 0 && bitrate <= INT64_MAX)
2537  ic->bit_rate = bitrate;
2538  }
2539 }
2540 
2541 static void fill_all_stream_timings(AVFormatContext *ic)
2542 {
2543  int i;
2544  AVStream *st;
2545 
2547  for (i = 0; i < ic->nb_streams; i++) {
2548  st = ic->streams[i];
2549  if (st->start_time == AV_NOPTS_VALUE) {
2550  if (ic->start_time != AV_NOPTS_VALUE)
2552  st->time_base);
2553  if (ic->duration != AV_NOPTS_VALUE)
2555  st->time_base);
2556  }
2557  }
2558 }
2559 
2560 static void estimate_timings_from_bit_rate(AVFormatContext *ic)
2561 {
2562  int64_t filesize, duration;
2563  int i, show_warning = 0;
2564  AVStream *st;
2565 
2566  /* if bit_rate is already set, we believe it */
2567  if (ic->bit_rate <= 0) {
2568  int bit_rate = 0;
2569  for (i = 0; i < ic->nb_streams; i++) {
2570  st = ic->streams[i];
2571  if (st->codecpar->bit_rate > 0) {
2572  if (INT_MAX - st->codecpar->bit_rate < bit_rate) {
2573  bit_rate = 0;
2574  break;
2575  }
2576  bit_rate += st->codecpar->bit_rate;
2577  } else if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && st->codec_info_nb_frames > 1) {
2578  // If we have a videostream with packets but without a bitrate
2579  // then consider the sum not known
2580  bit_rate = 0;
2581  break;
2582  }
2583  }
2584  ic->bit_rate = bit_rate;
2585  }
2586 
2587  /* if duration is already set, we believe it */
2588  if (ic->duration == AV_NOPTS_VALUE &&
2589  ic->bit_rate != 0) {
2590  filesize = ic->pb ? avio_size(ic->pb) : 0;
2591  if (filesize > ic->internal->data_offset) {
2592  filesize -= ic->internal->data_offset;
2593  for (i = 0; i < ic->nb_streams; i++) {
2594  st = ic->streams[i];
2595  if ( st->time_base.num <= INT64_MAX / ic->bit_rate
2596  && st->duration == AV_NOPTS_VALUE) {
2597  duration = av_rescale(8 * filesize, st->time_base.den,
2598  ic->bit_rate *
2599  (int64_t) st->time_base.num);
2600  st->duration = duration;
2601  show_warning = 1;
2602  }
2603  }
2604  }
2605  }
2606  if (show_warning)
2607  av_log(ic, AV_LOG_WARNING,
2608  "Estimating duration from bitrate, this may be inaccurate\n");
2609 }
2610 
2611 #define DURATION_MAX_READ_SIZE 250000LL
2612 #define DURATION_MAX_RETRY 6
2613 
2614 /* only usable for MPEG-PS streams */
2615 static void estimate_timings_from_pts(AVFormatContext *ic, int64_t old_offset)
2616 {
2617  AVPacket pkt1, *pkt = &pkt1;
2618  AVStream *st;
2619  int num, den, read_size, i, ret;
2620  int found_duration = 0;
2621  int is_end;
2622  int64_t filesize, offset, duration;
2623  int retry = 0;
2624 
2625  /* flush packet queue */
2626  flush_packet_queue(ic);
2627 
2628  for (i = 0; i < ic->nb_streams; i++) {
2629  st = ic->streams[i];
2630  if (st->start_time == AV_NOPTS_VALUE &&
2631  st->first_dts == AV_NOPTS_VALUE &&
2633  av_log(ic, AV_LOG_WARNING,
2634  "start time for stream %d is not set in estimate_timings_from_pts\n", i);
2635 
2636  if (st->parser) {
2637  av_parser_close(st->parser);
2638  st->parser = NULL;
2639  }
2640  }
2641 
2642  av_opt_set(ic, "skip_changes", "1", AV_OPT_SEARCH_CHILDREN);
2643  /* estimate the end time (duration) */
2644  /* XXX: may need to support wrapping */
2645  filesize = ic->pb ? avio_size(ic->pb) : 0;
2646  do {
2647  is_end = found_duration;
2648  offset = filesize - (DURATION_MAX_READ_SIZE << retry);
2649  if (offset < 0)
2650  offset = 0;
2651 
2652  avio_seek(ic->pb, offset, SEEK_SET);
2653  read_size = 0;
2654  for (;;) {
2655  if (read_size >= DURATION_MAX_READ_SIZE << (FFMAX(retry - 1, 0)))
2656  break;
2657 
2658  do {
2659  ret = ff_read_packet(ic, pkt);
2660  } while (ret == AVERROR(EAGAIN));
2661  if (ret != 0)
2662  break;
2663  read_size += pkt->size;
2664  st = ic->streams[pkt->stream_index];
2665  if (pkt->pts != AV_NOPTS_VALUE &&
2666  (st->start_time != AV_NOPTS_VALUE ||
2667  st->first_dts != AV_NOPTS_VALUE)) {
2668  if (pkt->duration == 0) {
2669  ff_compute_frame_duration(ic, &num, &den, st, st->parser, pkt);
2670  if (den && num) {
2671  pkt->duration = av_rescale_rnd(1,
2672  num * (int64_t) st->time_base.den,
2673  den * (int64_t) st->time_base.num,
2674  AV_ROUND_DOWN);
2675  }
2676  }
2677  duration = pkt->pts + pkt->duration;
2678  found_duration = 1;
2679  if (st->start_time != AV_NOPTS_VALUE)
2680  duration -= st->start_time;
2681  else
2682  duration -= st->first_dts;
2683  if (duration > 0) {
2684  if (st->duration == AV_NOPTS_VALUE || st->info->last_duration<= 0 ||
2685  (st->duration < duration && FFABS(duration - st->info->last_duration) < 60LL*st->time_base.den / st->time_base.num))
2686  st->duration = duration;
2687  st->info->last_duration = duration;
2688  }
2689  }
2690  av_packet_unref(pkt);
2691  }
2692 
2693  /* check if all audio/video streams have valid duration */
2694  if (!is_end) {
2695  is_end = 1;
2696  for (i = 0; i < ic->nb_streams; i++) {
2697  st = ic->streams[i];
2698  switch (st->codecpar->codec_type) {
2699  case AVMEDIA_TYPE_VIDEO:
2700  case AVMEDIA_TYPE_AUDIO:
2701  if (st->duration == AV_NOPTS_VALUE)
2702  is_end = 0;
2703  }
2704  }
2705  }
2706  } while (!is_end &&
2707  offset &&
2708  ++retry <= DURATION_MAX_RETRY);
2709 
2710  av_opt_set(ic, "skip_changes", "0", AV_OPT_SEARCH_CHILDREN);
2711 
2712  /* warn about audio/video streams which duration could not be estimated */
2713  for (i = 0; i < ic->nb_streams; i++) {
2714  st = ic->streams[i];
2715  if (st->duration == AV_NOPTS_VALUE) {
2716  switch (st->codecpar->codec_type) {
2717  case AVMEDIA_TYPE_VIDEO:
2718  case AVMEDIA_TYPE_AUDIO:
2719  if (st->start_time != AV_NOPTS_VALUE || st->first_dts != AV_NOPTS_VALUE) {
2720  av_log(ic, AV_LOG_DEBUG, "stream %d : no PTS found at end of file, duration not set\n", i);
2721  } else
2722  av_log(ic, AV_LOG_DEBUG, "stream %d : no TS found at start of file, duration not set\n", i);
2723  }
2724  }
2725  }
2727 
2728  avio_seek(ic->pb, old_offset, SEEK_SET);
2729  for (i = 0; i < ic->nb_streams; i++) {
2730  int j;
2731 
2732  st = ic->streams[i];
2733  st->cur_dts = st->first_dts;
2736  for (j = 0; j < MAX_REORDER_DELAY + 1; j++)
2737  st->pts_buffer[j] = AV_NOPTS_VALUE;
2738  }
2739 }
2740 
2741 static void estimate_timings(AVFormatContext *ic, int64_t old_offset)
2742 {
2743  int64_t file_size;
2744 
2745  /* get the file size, if possible */
2746  if (ic->iformat->flags & AVFMT_NOFILE) {
2747  file_size = 0;
2748  } else {
2749  file_size = avio_size(ic->pb);
2750  file_size = FFMAX(0, file_size);
2751  }
2752 
2753  if ((!strcmp(ic->iformat->name, "mpeg") ||
2754  !strcmp(ic->iformat->name, "mpegts")) &&
2755  file_size && ic->pb->seekable) {
2756  /* get accurate estimate from the PTSes */
2757  estimate_timings_from_pts(ic, old_offset);
2759  } else if (has_duration(ic)) {
2760  /* at least one component has timings - we use them for all
2761  * the components */
2764  } else {
2765  /* less precise: use bitrate info */
2768  }
2770 
2771  {
2772  int i;
2773  AVStream av_unused *st;
2774  for (i = 0; i < ic->nb_streams; i++) {
2775  st = ic->streams[i];
2776  av_log(ic, AV_LOG_TRACE, "%d: start_time: %0.3f duration: %0.3f\n", i,
2777  (double) st->start_time / AV_TIME_BASE,
2778  (double) st->duration / AV_TIME_BASE);
2779  }
2780  av_log(ic, AV_LOG_TRACE,
2781  "stream: start_time: %0.3f duration: %0.3f bitrate=%"PRId64" kb/s\n",
2782  (double) ic->start_time / AV_TIME_BASE,
2783  (double) ic->duration / AV_TIME_BASE,
2784  (int64_t)ic->bit_rate / 1000);
2785  }
2786 }
2787 
2788 static int has_codec_parameters(AVStream *st, const char **errmsg_ptr)
2789 {
2790  AVCodecContext *avctx = st->internal->avctx;
2791 
2792 #define FAIL(errmsg) do { \
2793  if (errmsg_ptr) \
2794  *errmsg_ptr = errmsg; \
2795  return 0; \
2796  } while (0)
2797 
2798  if ( avctx->codec_id == AV_CODEC_ID_NONE
2799  && avctx->codec_type != AVMEDIA_TYPE_DATA)
2800  FAIL("unknown codec");
2801  switch (avctx->codec_type) {
2802  case AVMEDIA_TYPE_AUDIO:
2803  if (!avctx->frame_size && determinable_frame_size(avctx))
2804  FAIL("unspecified frame size");
2805  if (st->info->found_decoder >= 0 &&
2806  avctx->sample_fmt == AV_SAMPLE_FMT_NONE)
2807  FAIL("unspecified sample format");
2808  if (!avctx->sample_rate)
2809  FAIL("unspecified sample rate");
2810  if (!avctx->channels)
2811  FAIL("unspecified number of channels");
2812  if (st->info->found_decoder >= 0 && !st->nb_decoded_frames && avctx->codec_id == AV_CODEC_ID_DTS)
2813  FAIL("no decodable DTS frames");
2814  break;
2815  case AVMEDIA_TYPE_VIDEO:
2816  if (!avctx->width)
2817  FAIL("unspecified size");
2818  if (st->info->found_decoder >= 0 && avctx->pix_fmt == AV_PIX_FMT_NONE)
2819  FAIL("unspecified pixel format");
2822  FAIL("no frame in rv30/40 and no sar");
2823  break;
2824  case AVMEDIA_TYPE_SUBTITLE:
2825  if (avctx->codec_id == AV_CODEC_ID_HDMV_PGS_SUBTITLE && !avctx->width)
2826  FAIL("unspecified size");
2827  break;
2828  case AVMEDIA_TYPE_DATA:
2829  if (avctx->codec_id == AV_CODEC_ID_NONE) return 1;
2830  }
2831 
2832  return 1;
2833 }
2834 
2835 /* returns 1 or 0 if or if not decoded data was returned, or a negative error */
2836 static int try_decode_frame(AVFormatContext *s, AVStream *st, AVPacket *avpkt,
2838 {
2839  AVCodecContext *avctx = st->internal->avctx;
2840  const AVCodec *codec;
2841  int got_picture = 1, ret = 0;
2843  AVSubtitle subtitle;
2844  AVPacket pkt = *avpkt;
2845  int do_skip_frame = 0;
2846  enum AVDiscard skip_frame;
2847 
2848  if (!frame)
2849  return AVERROR(ENOMEM);
2850 
2851  if (!avcodec_is_open(avctx) &&
2852  st->info->found_decoder <= 0 &&
2853  (st->codecpar->codec_id != -st->info->found_decoder || !st->codecpar->codec_id)) {
2854  AVDictionary *thread_opt = NULL;
2855 
2856  codec = find_decoder(s, st, st->codecpar->codec_id);
2857 
2858  if (!codec) {
2859  st->info->found_decoder = -st->codecpar->codec_id;
2860  ret = -1;
2861  goto fail;
2862  }
2863 
2864  /* Force thread count to 1 since the H.264 decoder will not extract
2865  * SPS and PPS to extradata during multi-threaded decoding. */
2866  av_dict_set(options ? options : &thread_opt, "threads", "1", 0);
2867  if (s->codec_whitelist)
2868  av_dict_set(options ? options : &thread_opt, "codec_whitelist", s->codec_whitelist, 0);
2869  ret = avcodec_open2(avctx, codec, options ? options : &thread_opt);
2870  if (!options)
2871  av_dict_free(&thread_opt);
2872  if (ret < 0) {
2873  st->info->found_decoder = -avctx->codec_id;
2874  goto fail;
2875  }
2876  st->info->found_decoder = 1;
2877  } else if (!st->info->found_decoder)
2878  st->info->found_decoder = 1;
2879 
2880  if (st->info->found_decoder < 0) {
2881  ret = -1;
2882  goto fail;
2883  }
2884 
2886  do_skip_frame = 1;
2887  skip_frame = avctx->skip_frame;
2888  avctx->skip_frame = AVDISCARD_ALL;
2889  }
2890 
2891  while ((pkt.size > 0 || (!pkt.data && got_picture)) &&
2892  ret >= 0 &&
2894  (!st->codec_info_nb_frames &&
2896  got_picture = 0;
2897  if (avctx->codec_type == AVMEDIA_TYPE_VIDEO ||
2898  avctx->codec_type == AVMEDIA_TYPE_AUDIO) {
2899  ret = avcodec_send_packet(avctx, &pkt);
2900  if (ret < 0 && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF)
2901  break;
2902  if (ret >= 0)
2903  pkt.size = 0;
2904  ret = avcodec_receive_frame(avctx, frame);
2905  if (ret >= 0)
2906  got_picture = 1;
2907  if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
2908  ret = 0;
2909  } else if (avctx->codec_type == AVMEDIA_TYPE_SUBTITLE) {
2910  ret = avcodec_decode_subtitle2(avctx, &subtitle,
2911  &got_picture, &pkt);
2912  if (ret >= 0)
2913  pkt.size = 0;
2914  }
2915  if (ret >= 0) {
2916  if (got_picture)
2917  st->nb_decoded_frames++;
2918  ret = got_picture;
2919  }
2920  }
2921 
2922  if (!pkt.data && !got_picture)
2923  ret = -1;
2924 
2925 fail:
2926  if (do_skip_frame) {
2927  avctx->skip_frame = skip_frame;
2928  }
2929 
2930  av_frame_free(&frame);
2931  return ret;
2932 }
2933 
2934 unsigned int ff_codec_get_tag(const AVCodecTag *tags, enum AVCodecID id)
2935 {
2936  while (tags->id != AV_CODEC_ID_NONE) {
2937  if (tags->id == id)
2938  return tags->tag;
2939  tags++;
2940  }
2941  return 0;
2942 }
2943 
2944 enum AVCodecID ff_codec_get_id(const AVCodecTag *tags, unsigned int tag)
2945 {
2946  int i;
2947  for (i = 0; tags[i].id != AV_CODEC_ID_NONE; i++)
2948  if (tag == tags[i].tag)
2949  return tags[i].id;
2950  for (i = 0; tags[i].id != AV_CODEC_ID_NONE; i++)
2951  if (avpriv_toupper4(tag) == avpriv_toupper4(tags[i].tag))
2952  return tags[i].id;
2953  return AV_CODEC_ID_NONE;
2954 }
2955 
2956 enum AVCodecID ff_get_pcm_codec_id(int bps, int flt, int be, int sflags)
2957 {
2958  if (bps <= 0 || bps > 64)
2959  return AV_CODEC_ID_NONE;
2960 
2961  if (flt) {
2962  switch (bps) {
2963  case 32:
2965  case 64:
2967  default:
2968  return AV_CODEC_ID_NONE;
2969  }
2970  } else {
2971  bps += 7;
2972  bps >>= 3;
2973  if (sflags & (1 << (bps - 1))) {
2974  switch (bps) {
2975  case 1:
2976  return AV_CODEC_ID_PCM_S8;
2977  case 2:
2979  case 3:
2981  case 4:
2983  default:
2984  return AV_CODEC_ID_NONE;
2985  }
2986  } else {
2987  switch (bps) {
2988  case 1:
2989  return AV_CODEC_ID_PCM_U8;
2990  case 2:
2992  case 3:
2994  case 4:
2996  default:
2997  return AV_CODEC_ID_NONE;
2998  }
2999  }
3000  }
3001 }
3002 
3003 unsigned int av_codec_get_tag(const AVCodecTag *const *tags, enum AVCodecID id)
3004 {
3005  unsigned int tag;
3006  if (!av_codec_get_tag2(tags, id, &tag))
3007  return 0;
3008  return tag;
3009 }
3010 
3011 int av_codec_get_tag2(const AVCodecTag * const *tags, enum AVCodecID id,
3012  unsigned int *tag)
3013 {
3014  int i;
3015  for (i = 0; tags && tags[i]; i++) {
3016  const AVCodecTag *codec_tags = tags[i];
3017  while (codec_tags->id != AV_CODEC_ID_NONE) {
3018  if (codec_tags->id == id) {
3019  *tag = codec_tags->tag;
3020  return 1;
3021  }
3022  codec_tags++;
3023  }
3024  }
3025  return 0;
3026 }
3027 
3028 enum AVCodecID av_codec_get_id(const AVCodecTag *const *tags, unsigned int tag)
3029 {
3030  int i;
3031  for (i = 0; tags && tags[i]; i++) {
3032  enum AVCodecID id = ff_codec_get_id(tags[i], tag);
3033  if (id != AV_CODEC_ID_NONE)
3034  return id;
3035  }
3036  return AV_CODEC_ID_NONE;
3037 }
3038 
3039 static void compute_chapters_end(AVFormatContext *s)
3040 {
3041  unsigned int i, j;
3042  int64_t max_time = 0;
3043 
3044  if (s->duration > 0)
3045  max_time = s->duration +
3046  ((s->start_time == AV_NOPTS_VALUE) ? 0 : s->start_time);
3047 
3048  for (i = 0; i < s->nb_chapters; i++)
3049  if (s->chapters[i]->end == AV_NOPTS_VALUE) {
3050  AVChapter *ch = s->chapters[i];
3051  int64_t end = max_time ? av_rescale_q(max_time, AV_TIME_BASE_Q,
3052  ch->time_base)
3053  : INT64_MAX;
3054 
3055  for (j = 0; j < s->nb_chapters; j++) {
3056  AVChapter *ch1 = s->chapters[j];
3057  int64_t next_start = av_rescale_q(ch1->start, ch1->time_base,
3058  ch->time_base);
3059  if (j != i && next_start > ch->start && next_start < end)
3060  end = next_start;
3061  }
3062  ch->end = (end == INT64_MAX) ? ch->start : end;
3063  }
3064 }
3065 
3066 static int get_std_framerate(int i)
3067 {
3068  if (i < 30*12)
3069  return (i + 1) * 1001;
3070  i -= 30*12;
3071 
3072  if (i < 30)
3073  return (i + 31) * 1001 * 12;
3074  i -= 30;
3075 
3076  if (i < 3)
3077  return ((const int[]) { 80, 120, 240})[i] * 1001 * 12;
3078 
3079  i -= 3;
3080 
3081  return ((const int[]) { 24, 30, 60, 12, 15, 48 })[i] * 1000 * 12;
3082 }
3083 
3084 /* Is the time base unreliable?
3085  * This is a heuristic to balance between quick acceptance of the values in
3086  * the headers vs. some extra checks.
3087  * Old DivX and Xvid often have nonsense timebases like 1fps or 2fps.
3088  * MPEG-2 commonly misuses field repeat flags to store different framerates.
3089  * And there are "variable" fps files this needs to detect as well. */
3091 {
3092  if (c->time_base.den >= 101LL * c->time_base.num ||
3093  c->time_base.den < 5LL * c->time_base.num ||
3094  // c->codec_tag == AV_RL32("DIVX") ||
3095  // c->codec_tag == AV_RL32("XVID") ||
3096  c->codec_tag == AV_RL32("mp4v") ||
3098  c->codec_id == AV_CODEC_ID_GIF ||
3099  c->codec_id == AV_CODEC_ID_HEVC ||
3100  c->codec_id == AV_CODEC_ID_H264)
3101  return 1;
3102  return 0;
3103 }
3104 
3106 {
3107  int ret;
3108 
3110  par->extradata = NULL;
3111  par->extradata_size = 0;
3112  return AVERROR(EINVAL);
3113  }
3115  if (par->extradata) {
3116  memset(par->extradata + size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
3117  par->extradata_size = size;
3118  ret = 0;
3119  } else {
3120  par->extradata_size = 0;
3121  ret = AVERROR(ENOMEM);
3122  }
3123  return ret;
3124 }
3125 
3126 int ff_get_extradata(AVFormatContext *s, AVCodecParameters *par, AVIOContext *pb, int size)
3127 {
3128  int ret = ff_alloc_extradata(par, size);
3129  if (ret < 0)
3130  return ret;
3131  ret = avio_read(pb, par->extradata, size);
3132  if (ret != size) {
3133  av_freep(&par->extradata);
3134  par->extradata_size = 0;
3135  av_log(s, AV_LOG_ERROR, "Failed to read extradata of size %d\n", size);
3136  return ret < 0 ? ret : AVERROR_INVALIDDATA;
3137  }
3138 
3139  return ret;
3140 }
3141 
3142 int ff_rfps_add_frame(AVFormatContext *ic, AVStream *st, int64_t ts)
3143 {
3144  int i, j;
3145  int64_t last = st->info->last_dts;
3146 
3147  if ( ts != AV_NOPTS_VALUE && last != AV_NOPTS_VALUE && ts > last
3148  && ts - (uint64_t)last < INT64_MAX) {
3149  double dts = (is_relative(ts) ? ts - RELATIVE_TS_BASE : ts) * av_q2d(st->time_base);
3150  int64_t duration = ts - last;
3151 
3152  if (!st->info->duration_error)
3153  st->info->duration_error = av_mallocz(sizeof(st->info->duration_error[0])*2);
3154  if (!st->info->duration_error)
3155  return AVERROR(ENOMEM);
3156 
3157 // if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
3158 // av_log(NULL, AV_LOG_ERROR, "%f\n", dts);
3159  for (i = 0; i<MAX_STD_TIMEBASES; i++) {
3160  if (st->info->duration_error[0][1][i] < 1e10) {
3161  int framerate = get_std_framerate(i);
3162  double sdts = dts*framerate/(1001*12);
3163  for (j= 0; j<2; j++) {
3164  int64_t ticks = llrint(sdts+j*0.5);
3165  double error= sdts - ticks + j*0.5;
3166  st->info->duration_error[j][0][i] += error;
3167  st->info->duration_error[j][1][i] += error*error;
3168  }
3169  }
3170  }
3171  st->info->duration_count++;
3173 
3174  if (st->info->duration_count % 10 == 0) {
3175  int n = st->info->duration_count;
3176  for (i = 0; i<MAX_STD_TIMEBASES; i++) {
3177  if (st->info->duration_error[0][1][i] < 1e10) {
3178  double a0 = st->info->duration_error[0][0][i] / n;
3179  double error0 = st->info->duration_error[0][1][i] / n - a0*a0;
3180  double a1 = st->info->duration_error[1][0][i] / n;
3181  double error1 = st->info->duration_error[1][1][i] / n - a1*a1;
3182  if (error0 > 0.04 && error1 > 0.04) {
3183  st->info->duration_error[0][1][i] = 2e10;
3184  st->info->duration_error[1][1][i] = 2e10;
3185  }
3186  }
3187  }
3188  }
3189 
3190  // ignore the first 4 values, they might have some random jitter
3191  if (st->info->duration_count > 3 && is_relative(ts) == is_relative(last))
3192  st->info->duration_gcd = av_gcd(st->info->duration_gcd, duration);
3193  }
3194  if (ts != AV_NOPTS_VALUE)
3195  st->info->last_dts = ts;
3196 
3197  return 0;
3198 }
3199 
3200 void ff_rfps_calculate(AVFormatContext *ic)
3201 {
3202  int i, j;
3203 
3204  for (i = 0; i < ic->nb_streams; i++) {
3205  AVStream *st = ic->streams[i];
3206 
3208  continue;
3209  // the check for tb_unreliable() is not completely correct, since this is not about handling
3210  // an unreliable/inexact time base, but a time base that is finer than necessary, as e.g.
3211  // ipmovie.c produces.
3212  if (tb_unreliable(st->internal->avctx) && st->info->duration_count > 15 && st->info->duration_gcd > FFMAX(1, st->time_base.den/(500LL*st->time_base.num)) && !st->r_frame_rate.num)
3213  av_reduce(&st->r_frame_rate.num, &st->r_frame_rate.den, st->time_base.den, st->time_base.num * st->info->duration_gcd, INT_MAX);
3214  if (st->info->duration_count>1 && !st->r_frame_rate.num
3215  && tb_unreliable(st->internal->avctx)) {
3216  int num = 0;
3217  double best_error= 0.01;
3218  AVRational ref_rate = st->r_frame_rate.num ? st->r_frame_rate : av_inv_q(st->time_base);
3219 
3220  for (j= 0; j<MAX_STD_TIMEBASES; j++) {
3221  int k;
3222 
3223  if (st->info->codec_info_duration &&
3224  st->info->codec_info_duration*av_q2d(st->time_base) < (1001*11.5)/get_std_framerate(j))
3225  continue;
3226  if (!st->info->codec_info_duration && get_std_framerate(j) < 1001*12)
3227  continue;
3228 
3229  if (av_q2d(st->time_base) * st->info->rfps_duration_sum / st->info->duration_count < (1001*12.0 * 0.8)/get_std_framerate(j))
3230  continue;
3231 
3232  for (k= 0; k<2; k++) {
3233  int n = st->info->duration_count;
3234  double a= st->info->duration_error[k][0][j] / n;
3235  double error= st->info->duration_error[k][1][j]/n - a*a;
3236 
3237  if (error < best_error && best_error> 0.000000001) {
3238  best_error= error;
3239  num = get_std_framerate(j);
3240  }
3241  if (error < 0.02)
3242  av_log(ic, AV_LOG_DEBUG, "rfps: %f %f\n", get_std_framerate(j) / 12.0/1001, error);
3243  }
3244  }
3245  // do not increase frame rate by more than 1 % in order to match a standard rate.
3246  if (num && (!ref_rate.num || (double)num/(12*1001) < 1.01 * av_q2d(ref_rate)))
3247  av_reduce(&st->r_frame_rate.num, &st->r_frame_rate.den, num, 12*1001, INT_MAX);
3248  }
3249  if ( !st->avg_frame_rate.num
3250  && st->r_frame_rate.num && st->info->rfps_duration_sum
3251  && st->info->codec_info_duration <= 0
3252  && st->info->duration_count > 2
3253  && fabs(1.0 / (av_q2d(st->r_frame_rate) * av_q2d(st->time_base)) - st->info->rfps_duration_sum / (double)st->info->duration_count) <= 1.0
3254  ) {
3255  av_log(ic, AV_LOG_DEBUG, "Setting avg frame rate based on r frame rate\n");
3256  st->avg_frame_rate = st->r_frame_rate;
3257  }
3258 
3259  av_freep(&st->info->duration_error);
3260  st->info->last_dts = AV_NOPTS_VALUE;
3261  st->info->duration_count = 0;
3262  st->info->rfps_duration_sum = 0;
3263  }
3264 }
3265 
3267 {
3268  int i, count = 0, ret = 0, j;
3269  int64_t read_size;
3270  AVStream *st;
3271  AVCodecContext *avctx;
3272  AVPacket pkt1, *pkt;
3273  int64_t old_offset = avio_tell(ic->pb);
3274  // new streams might appear, no options for those
3275  int orig_nb_streams = ic->nb_streams;
3276  int flush_codecs;
3277  int64_t max_analyze_duration = ic->max_analyze_duration;
3278  int64_t max_stream_analyze_duration;
3279  int64_t max_subtitle_analyze_duration;
3280  int64_t probesize = ic->probesize;
3281  int eof_reached = 0;
3282 
3283  flush_codecs = probesize > 0;
3284 
3285  av_opt_set(ic, "skip_clear", "1", AV_OPT_SEARCH_CHILDREN);
3286 
3287  max_stream_analyze_duration = max_analyze_duration;
3288  max_subtitle_analyze_duration = max_analyze_duration;
3289  if (!max_analyze_duration) {
3290  max_stream_analyze_duration =
3291  max_analyze_duration = 5*AV_TIME_BASE;
3292  max_subtitle_analyze_duration = 30*AV_TIME_BASE;
3293  if (!strcmp(ic->iformat->name, "flv"))
3294  max_stream_analyze_duration = 90*AV_TIME_BASE;
3295  if (!strcmp(ic->iformat->name, "mpeg") || !strcmp(ic->iformat->name, "mpegts"))
3296  max_stream_analyze_duration = 7*AV_TIME_BASE;
3297  }
3298 
3299  if (ic->pb)
3300  av_log(ic, AV_LOG_DEBUG, "Before avformat_find_stream_info() pos: %"PRId64" bytes read:%"PRId64" seeks:%d\n",
3301  avio_tell(ic->pb), ic->pb->bytes_read, ic->pb->seek_count);
3302 
3303  for (i = 0; i < ic->nb_streams; i++) {
3304  const AVCodec *codec;
3305  AVDictionary *thread_opt = NULL;
3306  st = ic->streams[i];
3307  avctx = st->internal->avctx;
3308 
3309  if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO ||
3311 /* if (!st->time_base.num)
3312  st->time_base = */
3313  if (!avctx->time_base.num)
3314  avctx->time_base = st->time_base;
3315  }
3316  // only for the split stuff
3317  if (!st->parser && !(ic->flags & AVFMT_FLAG_NOPARSE) && st->request_probe <= 0) {
3318  st->parser = av_parser_init(st->codecpar->codec_id);
3319  if (st->parser) {
3320  if (st->need_parsing == AVSTREAM_PARSE_HEADERS) {
3322  } else if (st->need_parsing == AVSTREAM_PARSE_FULL_RAW) {
3324  }
3325  } else if (st->need_parsing) {
3326  av_log(ic, AV_LOG_VERBOSE, "parser not found for codec "
3327  "%s, packets or times may be invalid.\n",
3329  }
3330  }
3331 
3332  /* check if the caller has overridden the codec id */
3333 #if FF_API_LAVF_AVCTX
3335  if (st->codec->codec_id != st->internal->orig_codec_id) {
3336  st->codecpar->codec_id = st->codec->codec_id;
3337  st->codecpar->codec_type = st->codec->codec_type;
3338  st->internal->orig_codec_id = st->codec->codec_id;
3339  }
3341 #endif
3342  if (st->codecpar->codec_id != st->internal->orig_codec_id)
3344 
3345  ret = avcodec_parameters_to_context(avctx, st->codecpar);
3346  if (ret < 0)
3347  goto find_stream_info_err;
3348  if (st->request_probe <= 0)
3349  st->internal->avctx_inited = 1;
3350 
3351  codec = find_decoder(ic, st, st->codecpar->codec_id);
3352 
3353  /* Force thread count to 1 since the H.264 decoder will not extract
3354  * SPS and PPS to extradata during multi-threaded decoding. */
3355  av_dict_set(options ? &options[i] : &thread_opt, "threads", "1", 0);
3356 
3357  if (ic->codec_whitelist)
3358  av_dict_set(options ? &options[i] : &thread_opt, "codec_whitelist", ic->codec_whitelist, 0);
3359 
3360  /* Ensure that subtitle_header is properly set. */
3362  && codec && !avctx->codec) {
3363  if (avcodec_open2(avctx, codec, options ? &options[i] : &thread_opt) < 0)
3364  av_log(ic, AV_LOG_WARNING,
3365  "Failed to open codec in av_find_stream_info\n");
3366  }
3367 
3368  // Try to just open decoders, in case this is enough to get parameters.
3369  if (!has_codec_parameters(st, NULL) && st->request_probe <= 0) {
3370  if (codec && !avctx->codec)
3371  if (avcodec_open2(avctx, codec, options ? &options[i] : &thread_opt) < 0)
3372  av_log(ic, AV_LOG_WARNING,
3373  "Failed to open codec in av_find_stream_info\n");
3374  }
3375  if (!options)
3376  av_dict_free(&thread_opt);
3377  }
3378 
3379  for (i = 0; i < ic->nb_streams; i++) {
3380 #if FF_API_R_FRAME_RATE
3381  ic->streams[i]->info->last_dts = AV_NOPTS_VALUE;
3382 #endif
3385  }
3386 
3387  read_size = 0;
3388  for (;;) {
3389  int analyzed_all_streams;
3391  ret = AVERROR_EXIT;
3392  av_log(ic, AV_LOG_DEBUG, "interrupted\n");
3393  break;
3394  }
3395 
3396  /* check if one codec still needs to be handled */
3397  for (i = 0; i < ic->nb_streams; i++) {
3398  int fps_analyze_framecount = 20;
3399 
3400  st = ic->streams[i];
3401  if (!has_codec_parameters(st, NULL))
3402  break;
3403  /* If the timebase is coarse (like the usual millisecond precision
3404  * of mkv), we need to analyze more frames to reliably arrive at
3405  * the correct fps. */
3406  if (av_q2d(st->time_base) > 0.0005)
3407  fps_analyze_framecount *= 2;
3408  if (!tb_unreliable(st->internal->avctx))
3409  fps_analyze_framecount = 0;
3410  if (ic->fps_probe_size >= 0)
3411  fps_analyze_framecount = ic->fps_probe_size;
3413  fps_analyze_framecount = 0;
3414  /* variable fps and no guess at the real fps */
3415  if (!(st->r_frame_rate.num && st->avg_frame_rate.num) &&
3417  int count = (ic->iformat->flags & AVFMT_NOTIMESTAMPS) ?
3419  st->info->duration_count;
3420  if (count < fps_analyze_framecount)
3421  break;
3422  }
3423  if (st->parser && st->parser->parser->split &&
3424  !st->codecpar->extradata)
3425  break;
3426  if (st->first_dts == AV_NOPTS_VALUE &&
3427  !(ic->iformat->flags & AVFMT_NOTIMESTAMPS) &&
3431  break;
3432  }
3433  analyzed_all_streams = 0;
3434  if (i == ic->nb_streams) {
3435  analyzed_all_streams = 1;
3436  /* NOTE: If the format has no header, then we need to read some
3437  * packets to get most of the streams, so we cannot stop here. */
3438  if (!(ic->ctx_flags & AVFMTCTX_NOHEADER)) {
3439  /* If we found the info for all the codecs, we can stop. */
3440  ret = count;
3441  av_log(ic, AV_LOG_DEBUG, "All info found\n");
3442  flush_codecs = 0;
3443  break;
3444  }
3445  }
3446  /* We did not get all the codec info, but we read too much data. */
3447  if (read_size >= probesize) {
3448  ret = count;
3449  av_log(ic, AV_LOG_DEBUG,
3450  "Probe buffer size limit of %"PRId64" bytes reached\n", probesize);
3451  for (i = 0; i < ic->nb_streams; i++)
3452  if (!ic->streams[i]->r_frame_rate.num &&
3453  ic->streams[i]->info->duration_count <= 1 &&
3455  strcmp(ic->iformat->name, "image2"))
3456  av_log(ic, AV_LOG_WARNING,
3457  "Stream #%d: not enough frames to estimate rate; "
3458  "consider increasing probesize\n", i);
3459  break;
3460  }
3461 
3462  /* NOTE: A new stream can be added there if no header in file
3463  * (AVFMTCTX_NOHEADER). */
3464  ret = read_frame_internal(ic, &pkt1);
3465  if (ret == AVERROR(EAGAIN))
3466  continue;
3467 
3468  if (ret < 0) {
3469  /* EOF or error*/
3470  eof_reached = 1;
3471  break;
3472  }
3473 
3474  pkt = &pkt1;
3475 
3476  if (!(ic->flags & AVFMT_FLAG_NOBUFFER)) {
3477  ret = add_to_pktbuf(&ic->internal->packet_buffer, pkt,
3478  &ic->internal->packet_buffer_end, 0);
3479  if (ret < 0)
3480  goto find_stream_info_err;
3481  }
3482 
3483  st = ic->streams[pkt->stream_index];
3485  read_size += pkt->size;
3486 
3487  avctx = st->internal->avctx;
3488  if (!st->internal->avctx_inited) {
3489  ret = avcodec_parameters_to_context(avctx, st->codecpar);
3490  if (ret < 0)
3491  goto find_stream_info_err;
3492  st->internal->avctx_inited = 1;
3493  }
3494 
3495  if (pkt->dts != AV_NOPTS_VALUE && st->codec_info_nb_frames > 1) {
3496  /* check for non-increasing dts */
3497  if (st->info->fps_last_dts != AV_NOPTS_VALUE &&
3498  st->info->fps_last_dts >= pkt->dts) {
3499  av_log(ic, AV_LOG_DEBUG,
3500  "Non-increasing DTS in stream %d: packet %d with DTS "
3501  "%"PRId64", packet %d with DTS %"PRId64"\n",
3502  st->index, st->info->fps_last_dts_idx,
3504  pkt->dts);
3505  st->info->fps_first_dts =
3507  }
3508  /* Check for a discontinuity in dts. If the difference in dts
3509  * is more than 1000 times the average packet duration in the
3510  * sequence, we treat it as a discontinuity. */
3511  if (st->info->fps_last_dts != AV_NOPTS_VALUE &&
3513  (pkt->dts - st->info->fps_last_dts) / 1000 >
3514  (st->info->fps_last_dts - st->info->fps_first_dts) /
3515  (st->info->fps_last_dts_idx - st->info->fps_first_dts_idx)) {
3516  av_log(ic, AV_LOG_WARNING,
3517  "DTS discontinuity in stream %d: packet %d with DTS "
3518  "%"PRId64", packet %d with DTS %"PRId64"\n",
3519  st->index, st->info->fps_last_dts_idx,
3521  pkt->dts);
3522  st->info->fps_first_dts =
3524  }
3525 
3526  /* update stored dts values */
3527  if (st->info->fps_first_dts == AV_NOPTS_VALUE) {
3528  st->info->fps_first_dts = pkt->dts;
3530  }
3531  st->info->fps_last_dts = pkt->dts;
3533  }
3534  if (st->codec_info_nb_frames>1) {
3535  int64_t t = 0;
3536  int64_t limit;
3537 
3538  if (st->time_base.den > 0)
3540  if (st->avg_frame_rate.num > 0)
3542 
3543  if ( t == 0
3544  && st->codec_info_nb_frames>30
3545  && st->info->fps_first_dts != AV_NOPTS_VALUE
3546  && st->info->fps_last_dts != AV_NOPTS_VALUE)
3548 
3549  if (analyzed_all_streams) limit = max_analyze_duration;
3550  else if (avctx->codec_type == AVMEDIA_TYPE_SUBTITLE) limit = max_subtitle_analyze_duration;
3551  else limit = max_stream_analyze_duration;
3552 
3553  if (t >= limit) {
3554  av_log(ic, AV_LOG_VERBOSE, "max_analyze_duration %"PRId64" reached at %"PRId64" microseconds st:%d\n",
3555  limit,
3556  t, pkt->stream_index);
3557  if (ic->flags & AVFMT_FLAG_NOBUFFER)
3558  av_packet_unref(pkt);
3559  break;
3560  }
3561  if (pkt->duration) {
3562  if (avctx->codec_type == AVMEDIA_TYPE_SUBTITLE && pkt->pts != AV_NOPTS_VALUE && pkt->pts >= st->start_time) {
3563  st->info->codec_info_duration = FFMIN(pkt->pts - st->start_time, st->info->codec_info_duration + pkt->duration);
3564  } else
3565  st->info->codec_info_duration += pkt->duration;
3566  st->info->codec_info_duration_fields += st->parser && st->need_parsing && avctx->ticks_per_frame ==2 ? st->parser->repeat_pict + 1 : 2;
3567  }
3568  }
3569 #if FF_API_R_FRAME_RATE
3571  ff_rfps_add_frame(ic, st, pkt->dts);
3572 #endif
3573  if (st->parser && st->parser->parser->split && !avctx->extradata) {
3574  int i = st->parser->parser->split(avctx, pkt->data, pkt->size);
3575  if (i > 0 && i < FF_MAX_EXTRADATA_SIZE) {
3576  avctx->extradata_size = i;
3577  avctx->extradata = av_mallocz(avctx->extradata_size +
3579  if (!avctx->extradata)
3580  return AVERROR(ENOMEM);
3581  memcpy(avctx->extradata, pkt->data,
3582  avctx->extradata_size);
3583  }
3584  }
3585 
3586  /* If still no information, we try to open the codec and to
3587  * decompress the frame. We try to avoid that in most cases as
3588  * it takes longer and uses more memory. For MPEG-4, we need to
3589  * decompress for QuickTime.
3590  *
3591  * If AV_CODEC_CAP_CHANNEL_CONF is set this will force decoding of at
3592  * least one frame of codec data, this makes sure the codec initializes
3593  * the channel configuration and does not only trust the values from
3594  * the container. */
3595  try_decode_frame(ic, st, pkt,
3596  (options && i < orig_nb_streams) ? &options[i] : NULL);
3597 
3598  if (ic->flags & AVFMT_FLAG_NOBUFFER)
3599  av_packet_unref(pkt);
3600 
3601  st->codec_info_nb_frames++;
3602  count++;
3603  }
3604 
3605  if (eof_reached && ic->internal->packet_buffer) {
3606  int stream_index;
3607  for (stream_index = 0; stream_index < ic->nb_streams; stream_index++) {
3608  // EOF already reached while reading the stream above.
3609  // So continue with reoordering DTS with whatever delay we have.
3610  if (!has_decode_delay_been_guessed(st)) {
3611  update_dts_from_pts(ic, stream_index, ic->internal->packet_buffer);
3612  }
3613  }
3614  }
3615 
3616  if (flush_codecs) {
3617  AVPacket empty_pkt = { 0 };
3618  int err = 0;
3619  av_init_packet(&empty_pkt);
3620 
3621  for (i = 0; i < ic->nb_streams; i++) {
3622 
3623  st = ic->streams[i];
3624 
3625  /* flush the decoders */
3626  if (st->info->found_decoder == 1) {
3627  do {
3628  err = try_decode_frame(ic, st, &empty_pkt,
3629  (options && i < orig_nb_streams)
3630  ? &options[i] : NULL);
3631  } while (err > 0 && !has_codec_parameters(st, NULL));
3632 
3633  if (err < 0) {
3634  av_log(ic, AV_LOG_INFO,
3635  "decoding for stream %d failed\n", st->index);
3636  }
3637  }
3638  }
3639  }
3640 
3641  // close codecs which were opened in try_decode_frame()
3642  for (i = 0; i < ic->nb_streams; i++) {
3643  st = ic->streams[i];
3645  }
3646 
3647  ff_rfps_calculate(ic);
3648 
3649  for (i = 0; i < ic->nb_streams; i++) {
3650  st = ic->streams[i];
3651  avctx = st->internal->avctx;
3652  if (avctx->codec_type == AVMEDIA_TYPE_VIDEO) {
3653  if (avctx->codec_id == AV_CODEC_ID_RAWVIDEO && !avctx->codec_tag && !avctx->bits_per_coded_sample) {
3654  uint32_t tag= avcodec_pix_fmt_to_codec_tag(avctx->pix_fmt);
3656  avctx->codec_tag= tag;
3657  }
3658 
3659  /* estimate average framerate if not set by demuxer */
3660  if (st->info->codec_info_duration_fields &&
3661  !st->avg_frame_rate.num &&
3662  st->info->codec_info_duration) {
3663  int best_fps = 0;
3664  double best_error = 0.01;
3665 
3666  if (st->info->codec_info_duration >= INT64_MAX / st->time_base.num / 2||
3667  st->info->codec_info_duration_fields >= INT64_MAX / st->time_base.den ||
3668  st->info->codec_info_duration < 0)
3669  continue;
3671  st->info->codec_info_duration_fields * (int64_t) st->time_base.den,
3672  st->info->codec_info_duration * 2 * (int64_t) st->time_base.num, 60000);
3673 
3674  /* Round guessed framerate to a "standard" framerate if it's
3675  * within 1% of the original estimate. */
3676  for (j = 0; j < MAX_STD_TIMEBASES; j++) {
3677  AVRational std_fps = { get_std_framerate(j), 12 * 1001 };
3678  double error = fabs(av_q2d(st->avg_frame_rate) /
3679  av_q2d(std_fps) - 1);
3680 
3681  if (error < best_error) {
3682  best_error = error;
3683  best_fps = std_fps.num;
3684  }
3685  }
3686  if (best_fps)
3688  best_fps, 12 * 1001, INT_MAX);
3689  }
3690 
3691  if (!st->r_frame_rate.num) {
3692  if ( avctx->time_base.den * (int64_t) st->time_base.num
3693  <= avctx->time_base.num * avctx->ticks_per_frame * (int64_t) st->time_base.den) {
3694  st->r_frame_rate.num = avctx->time_base.den;
3695  st->r_frame_rate.den = avctx->time_base.num * avctx->ticks_per_frame;
3696  } else {
3697  st->r_frame_rate.num = st->time_base.den;
3698  st->r_frame_rate.den = st->time_base.num;
3699  }
3700  }
3702  AVRational hw_ratio = { avctx->height, avctx->width };
3704  hw_ratio);
3705  }
3706  } else if (avctx->codec_type == AVMEDIA_TYPE_AUDIO) {
3707  if (!avctx->bits_per_coded_sample)
3708  avctx->bits_per_coded_sample =
3710  // set stream disposition based on audio service type
3711  switch (avctx->audio_service_type) {
3714  break;
3717  break;
3720  break;
3723  break;
3726  break;
3727  }
3728  }
3729  }
3730 
3731  if (probesize)
3732  estimate_timings(ic, old_offset);
3733 
3734  av_opt_set(ic, "skip_clear", "0", AV_OPT_SEARCH_CHILDREN);
3735 
3736  if (ret >= 0 && ic->nb_streams)
3737  /* We could not have all the codec parameters before EOF. */
3738  ret = -1;
3739  for (i = 0; i < ic->nb_streams; i++) {
3740  const char *errmsg;
3741  st = ic->streams[i];
3742 
3743  /* if no packet was ever seen, update context now for has_codec_parameters */
3744  if (!st->internal->avctx_inited) {
3746  if (ret < 0)
3747  goto find_stream_info_err;
3748  }
3749  if (!has_codec_parameters(st, &errmsg)) {
3750  char buf[256];
3751  avcodec_string(buf, sizeof(buf), st->internal->avctx, 0);
3752  av_log(ic, AV_LOG_WARNING,
3753  "Could not find codec parameters for stream %d (%s): %s\n"
3754  "Consider increasing the value for the 'analyzeduration' and 'probesize' options\n",
3755  i, buf, errmsg);
3756  } else {
3757  ret = 0;
3758  }
3759  }
3760 
3762 
3763  /* update the stream parameters from the internal codec contexts */
3764  for (i = 0; i < ic->nb_streams; i++) {
3765  st = ic->streams[i];
3766 
3767  if (st->internal->avctx_inited) {
3768  int orig_w = st->codecpar->width;
3769  int orig_h = st->codecpar->height;
3771  if (ret < 0)
3772  goto find_stream_info_err;
3773  // The decoder might reduce the video size by the lowres factor.
3774  if (av_codec_get_lowres(st->internal->avctx) && orig_w) {
3775  st->codecpar->width = orig_w;
3776  st->codecpar->height = orig_h;
3777  }
3778  }
3779 
3780 #if FF_API_LAVF_AVCTX
3782  ret = avcodec_parameters_to_context(st->codec, st->codecpar);
3783  if (ret < 0)
3784  goto find_stream_info_err;
3785 
3786  // The old API (AVStream.codec) "requires" the resolution to be adjusted
3787  // by the lowres factor.
3788  if (av_codec_get_lowres(st->internal->avctx) && st->internal->avctx->width) {
3790  st->codec->width = st->internal->avctx->width;
3791  st->codec->height = st->internal->avctx->height;
3792  st->codec->coded_width = st->internal->avctx->coded_width;
3793  st->codec->coded_height = st->internal->avctx->coded_height;
3794  }
3795 
3796  if (st->codec->codec_tag != MKTAG('t','m','c','d'))
3797  st->codec->time_base = st->internal->avctx->time_base;
3798  st->codec->framerate = st->avg_frame_rate;
3799 
3800  if (st->internal->avctx->subtitle_header) {
3801  st->codec->subtitle_header = av_malloc(st->internal->avctx->subtitle_header_size);
3802  if (!st->codec->subtitle_header)
3803  goto find_stream_info_err;
3804  st->codec->subtitle_header_size = st->internal->avctx->subtitle_header_size;
3805  memcpy(st->codec->subtitle_header, st->internal->avctx->subtitle_header,
3806  st->codec->subtitle_header_size);
3807  }
3809 #endif
3810 
3811  st->internal->avctx_inited = 0;
3812  }
3813 
3814 find_stream_info_err:
3815  for (i = 0; i < ic->nb_streams; i++) {
3816  st = ic->streams[i];
3817  if (st->info)
3818  av_freep(&st->info->duration_error);
3819  av_freep(&ic->streams[i]->info);
3820  }
3821  if (ic->pb)
3822  av_log(ic, AV_LOG_DEBUG, "After avformat_find_stream_info() pos: %"PRId64" bytes read:%"PRId64" seeks:%d frames:%d\n",
3823  avio_tell(ic->pb), ic->pb->bytes_read, ic->pb->seek_count, count);
3824  return ret;
3825 }
3826 
3827 AVProgram *av_find_program_from_stream(AVFormatContext *ic, AVProgram *last, int s)
3828 {
3829  int i, j;
3830 
3831  for (i = 0; i < ic->nb_programs; i++) {
3832  if (ic->programs[i] == last) {
3833  last = NULL;
3834  } else {
3835  if (!last)
3836  for (j = 0; j < ic->programs[i]->nb_stream_indexes; j++)
3837  if (ic->programs[i]->stream_index[j] == s)
3838  return ic->programs[i];
3839  }
3840  }
3841  return NULL;
3842 }
3843 
3844 int av_find_best_stream(AVFormatContext *ic, enum AVMediaType type,
3845  int wanted_stream_nb, int related_stream,
3846  AVCodec **decoder_ret, int flags)
3847 {
3848  int i, nb_streams = ic->nb_streams;
3849  int ret = AVERROR_STREAM_NOT_FOUND, best_count = -1, best_bitrate = -1, best_multiframe = -1, count, bitrate, multiframe;
3850  unsigned *program = NULL;
3851  const AVCodec *decoder = NULL, *best_decoder = NULL;
3852 
3853  if (related_stream >= 0 && wanted_stream_nb < 0) {
3854  AVProgram *p = av_find_program_from_stream(ic, NULL, related_stream);
3855  if (p) {
3856  program = p->stream_index;
3857  nb_streams = p->nb_stream_indexes;
3858  }
3859  }
3860  for (i = 0; i < nb_streams; i++) {
3861  int real_stream_index = program ? program[i] : i;
3862  AVStream *st = ic->streams[real_stream_index];
3863  AVCodecParameters *par = st->codecpar;
3864  if (par->codec_type != type)
3865  continue;
3866  if (wanted_stream_nb >= 0 && real_stream_index != wanted_stream_nb)
3867  continue;
3868  if (wanted_stream_nb != real_stream_index &&
3871  continue;
3872  if (type == AVMEDIA_TYPE_AUDIO && !(par->channels && par->sample_rate))
3873  continue;
3874  if (decoder_ret) {
3875  decoder = find_decoder(ic, st, par->codec_id);
3876  if (!decoder) {
3877  if (ret < 0)
3879  continue;
3880  }
3881  }
3883  bitrate = par->bit_rate;
3884  multiframe = FFMIN(5, count);
3885  if ((best_multiframe > multiframe) ||
3886  (best_multiframe == multiframe && best_bitrate > bitrate) ||
3887  (best_multiframe == multiframe && best_bitrate == bitrate && best_count >= count))
3888  continue;
3889  best_count = count;
3890  best_bitrate = bitrate;
3891  best_multiframe = multiframe;
3892  ret = real_stream_index;
3893  best_decoder = decoder;
3894  if (program && i == nb_streams - 1 && ret < 0) {
3895  program = NULL;
3896  nb_streams = ic->nb_streams;
3897  /* no related stream found, try again with everything */
3898  i = 0;
3899  }
3900  }
3901  if (decoder_ret)
3902  *decoder_ret = (AVCodec*)best_decoder;
3903  return ret;
3904 }
3905 
3906 /*******************************************************/
3907 
3908 int av_read_play(AVFormatContext *s)
3909 {
3910  if (s->iformat->read_play)
3911  return s->iformat->read_play(s);
3912  if (s->pb)
3913  return avio_pause(s->pb, 0);
3914  return AVERROR(ENOSYS);
3915 }
3916 
3917 int av_read_pause(AVFormatContext *s)
3918 {
3919  if (s->iformat->read_pause)
3920  return s->iformat->read_pause(s);
3921  if (s->pb)
3922  return avio_pause(s->pb, 1);
3923  return AVERROR(ENOSYS);
3924 }
3925 
3926 static void free_stream(AVStream **pst)
3927 {
3928  AVStream *st = *pst;
3929  int i;
3930 
3931  if (!st)
3932  return;
3933 
3934  for (i = 0; i < st->nb_side_data; i++)
3935  av_freep(&st->side_data[i].data);
3936  av_freep(&st->side_data);
3937 
3938  if (st->parser)
3939  av_parser_close(st->parser);
3940 
3941  if (st->attached_pic.data)
3943 
3944  if (st->internal) {
3946  }
3947  av_freep(&st->internal);
3948 
3949  av_dict_free(&st->metadata);
3951  av_freep(&st->probe_data.buf);
3952  av_freep(&st->index_entries);
3953 #if FF_API_LAVF_AVCTX
3955  av_freep(&st->codec->extradata);
3956  av_freep(&st->codec->subtitle_header);
3957  av_freep(&st->codec);
3959 #endif
3960  av_freep(&st->priv_data);
3961  if (st->info)
3962  av_freep(&st->info->duration_error);
3963  av_freep(&st->info);
3965  av_freep(&st->priv_pts);
3966 
3967  av_freep(pst);
3968 }
3969 
3970 void ff_free_stream(AVFormatContext *s, AVStream *st)
3971 {
3972  av_assert0(s->nb_streams>0);
3973  av_assert0(s->streams[ s->nb_streams - 1 ] == st);
3974 
3975  free_stream(&s->streams[ --s->nb_streams ]);
3976 }
3977 
3978 void avformat_free_context(AVFormatContext *s)
3979 {
3980  int i;
3981 
3982  if (!s)
3983  return;
3984 
3985  av_opt_free(s);
3986  if (s->iformat && s->iformat->priv_class && s->priv_data)
3987  av_opt_free(s->priv_data);
3988  if (s->oformat && s->oformat->priv_class && s->priv_data)
3989  av_opt_free(s->priv_data);
3990 
3991  for (i = s->nb_streams - 1; i >= 0; i--)
3992  ff_free_stream(s, s->streams[i]);
3993 
3994 
3995  for (i = s->nb_programs - 1; i >= 0; i--) {
3996  av_dict_free(&s->programs[i]->metadata);
3997  av_freep(&s->programs[i]->stream_index);
3998  av_freep(&s->programs[i]);
3999  }
4000  av_freep(&s->programs);
4001  av_freep(&s->priv_data);
4002  while (s->nb_chapters--) {
4004  av_freep(&s->chapters[s->nb_chapters]);
4005  }
4006  av_freep(&s->chapters);
4007  av_dict_free(&s->metadata);
4008  av_freep(&s->streams);
4009  av_freep(&s->internal);
4010  flush_packet_queue(s);
4011  av_free(s);
4012 }
4013 
4014 void avformat_close_input(AVFormatContext **ps)
4015 {
4016  AVFormatContext *s;
4017  AVIOContext *pb;
4018 
4019  if (!ps || !*ps)
4020  return;
4021 
4022  s = *ps;
4023  pb = s->pb;
4024 
4025  if ((s->iformat && strcmp(s->iformat->name, "image2") && s->iformat->flags & AVFMT_NOFILE) ||
4026  (s->flags & AVFMT_FLAG_CUSTOM_IO))
4027  pb = NULL;
4028 
4029  flush_packet_queue(s);
4030 
4031  if (s->iformat)
4032  if (s->iformat->read_close)
4033  s->iformat->read_close(s);
4034 
4036 
4037  *ps = NULL;
4038 
4039  avio_close(pb);
4040 }
4041 
4042 AVStream *avformat_new_stream(AVFormatContext *s, const AVCodec *c)
4043 {
4044  AVStream *st;
4045  int i;
4046  AVStream **streams;
4047 
4048  if (s->nb_streams >= INT_MAX/sizeof(*streams))
4049  return NULL;
4050  streams = av_realloc_array(s->streams, s->nb_streams + 1, sizeof(*streams));
4051  if (!streams)
4052  return NULL;
4053  s->streams = streams;
4054 
4055  st = av_mallocz(sizeof(AVStream));
4056  if (!st)
4057  return NULL;
4058  if (!(st->info = av_mallocz(sizeof(*st->info)))) {
4059  av_free(st);
4060  return NULL;
4061  }
4062  st->info->last_dts = AV_NOPTS_VALUE;
4063 
4064 #if FF_API_LAVF_AVCTX
4066  st->codec = avcodec_alloc_context3(c);
4067  if (!st->codec) {
4068  av_free(st->info);
4069  av_free(st);
4070  return NULL;
4071  }
4073 #endif
4074 
4075  st->internal = av_mallocz(sizeof(*st->internal));
4076  if (!st->internal)
4077  goto fail;
4078 
4080  if (!st->codecpar)
4081  goto fail;
4082 
4084  if (!st->internal->avctx)
4085  goto fail;
4086 
4087  if (s->iformat) {
4088 #if FF_API_LAVF_AVCTX
4090  /* no default bitrate if decoding */
4091  st->codec->bit_rate = 0;
4093 #endif
4094 
4095  /* default pts setting is MPEG-like */
4096  avpriv_set_pts_info(st, 33, 1, 90000);
4097  /* we set the current DTS to 0 so that formats without any timestamps
4098  * but durations get some timestamps, formats with some unknown
4099  * timestamps have their first few packets buffered and the
4100  * timestamps corrected before they are returned to the user */
4101  st->cur_dts = RELATIVE_TS_BASE;
4102  } else {
4103  st->cur_dts = AV_NOPTS_VALUE;
4104  }
4105 
4106  st->index = s->nb_streams;
4107  st->start_time = AV_NOPTS_VALUE;
4108  st->duration = AV_NOPTS_VALUE;
4109  st->first_dts = AV_NOPTS_VALUE;
4113 
4116  for (i = 0; i < MAX_REORDER_DELAY + 1; i++)
4117  st->pts_buffer[i] = AV_NOPTS_VALUE;
4118 
4119  st->sample_aspect_ratio = (AVRational) { 0, 1 };
4120 
4121 #if FF_API_R_FRAME_RATE
4122  st->info->last_dts = AV_NOPTS_VALUE;
4123 #endif
4126 
4128 
4129  st->internal->need_context_update = 1;
4130 
4131  s->streams[s->nb_streams++] = st;
4132  return st;
4133 fail:
4134  free_stream(&st);
4135  return NULL;
4136 }
4137 
4138 AVProgram *av_new_program(AVFormatContext *ac, int id)
4139 {
4140  AVProgram *program = NULL;
4141  int i;
4142 
4143  av_log(ac, AV_LOG_TRACE, "new_program: id=0x%04x\n", id);
4144 
4145  for (i = 0; i < ac->nb_programs; i++)
4146  if (ac->programs[i]->id == id)
4147  program = ac->programs[i];
4148 
4149  if (!program) {
4150  program = av_mallocz(sizeof(AVProgram));
4151  if (!program)
4152  return NULL;
4153  dynarray_add(&ac->programs, &ac->nb_programs, program);
4154  program->discard = AVDISCARD_NONE;
4155  }
4156  program->id = id;
4159 
4160  program->start_time =
4161  program->end_time = AV_NOPTS_VALUE;
4162 
4163  return program;
4164 }
4165 
4166 AVChapter *avpriv_new_chapter(AVFormatContext *s, int id, AVRational time_base,
4167  int64_t start, int64_t end, const char *title)
4168 {
4169  AVChapter *chapter = NULL;
4170  int i;
4171 
4172  if (end != AV_NOPTS_VALUE && start > end) {
4173  av_log(s, AV_LOG_ERROR, "Chapter end time %"PRId64" before start %"PRId64"\n", end, start);
4174  return NULL;
4175  }
4176 
4177  for (i = 0; i < s->nb_chapters; i++)
4178  if (s->chapters[i]->id == id)
4179  chapter = s->chapters[i];
4180 
4181  if (!chapter) {
4182  chapter = av_mallocz(sizeof(AVChapter));
4183  if (!chapter)
4184  return NULL;
4185  dynarray_add(&s->chapters, &s->nb_chapters, chapter);
4186  }
4187  av_dict_set(&chapter->metadata, "title", title, 0);
4188  chapter->id = id;
4189  chapter->time_base = time_base;
4190  chapter->start = start;
4191  chapter->end = end;
4192 
4193  return chapter;
4194 }
4195 
4196 void av_program_add_stream_index(AVFormatContext *ac, int progid, unsigned idx)
4197 {
4198  int i, j;
4199  AVProgram *program = NULL;
4200  void *tmp;
4201 
4202  if (idx >= ac->nb_streams) {
4203  av_log(ac, AV_LOG_ERROR, "stream index %d is not valid\n", idx);
4204  return;
4205  }
4206 
4207  for (i = 0; i < ac->nb_programs; i++) {
4208  if (ac->programs[i]->id != progid)
4209  continue;
4210  program = ac->programs[i];
4211  for (j = 0; j < program->nb_stream_indexes; j++)
4212  if (program->stream_index[j] == idx)
4213  return;
4214 
4215  tmp = av_realloc_array(program->stream_index, program->nb_stream_indexes+1, sizeof(unsigned int));
4216  if (!tmp)
4217  return;
4218  program->stream_index = tmp;
4219  program->stream_index[program->nb_stream_indexes++] = idx;
4220  return;
4221  }
4222 }
4223 
4224 uint64_t ff_ntp_time(void)
4225 {
4226  return (av_gettime() / 1000) * 1000 + NTP_OFFSET_US;
4227 }
4228 
4229 int av_get_frame_filename(char *buf, int buf_size, const char *path, int number)
4230 {
4231  const char *p;
4232  char *q, buf1[20], c;
4233  int nd, len, percentd_found;
4234 
4235  q = buf;
4236  p = path;
4237  percentd_found = 0;
4238  for (;;) {
4239  c = *p++;
4240  if (c == '\0')
4241  break;
4242  if (c == '%') {
4243  do {
4244  nd = 0;
4245  while (av_isdigit(*p))
4246  nd = nd * 10 + *p++ - '0';
4247  c = *p++;
4248  } while (av_isdigit(c));
4249 
4250  switch (c) {
4251  case '%':
4252  goto addchar;
4253  case 'd':
4254  if (percentd_found)
4255  goto fail;
4256  percentd_found = 1;
4257  if (number < 0)
4258  nd += 1;
4259  snprintf(buf1, sizeof(buf1), "%0*d", nd, number);
4260  len = strlen(buf1);
4261  if ((q - buf + len) > buf_size - 1)
4262  goto fail;
4263  memcpy(q, buf1, len);
4264  q += len;
4265  break;
4266  default:
4267  goto fail;
4268  }
4269  } else {
4270 addchar:
4271  if ((q - buf) < buf_size - 1)
4272  *q++ = c;
4273  }
4274  }
4275  if (!percentd_found)
4276  goto fail;
4277  *q = '\0';
4278  return 0;
4279 fail:
4280  *q = '\0';
4281  return -1;
4282 }
4283 
4284 void av_url_split(char *proto, int proto_size,
4285  char *authorization, int authorization_size,
4286  char *hostname, int hostname_size,
4287  int *port_ptr, char *path, int path_size, const char *url)
4288 {
4289  const char *p, *ls, *ls2, *at, *at2, *col, *brk;
4290 
4291  if (port_ptr)
4292  *port_ptr = -1;
4293  if (proto_size > 0)
4294  proto[0] = 0;
4295  if (authorization_size > 0)
4296  authorization[0] = 0;
4297  if (hostname_size > 0)
4298  hostname[0] = 0;
4299  if (path_size > 0)
4300  path[0] = 0;
4301 
4302  /* parse protocol */
4303  if ((p = strchr(url, ':'))) {
4304  av_strlcpy(proto, url, FFMIN(proto_size, p + 1 - url));
4305  p++; /* skip ':' */
4306  if (*p == '/')
4307  p++;
4308  if (*p == '/')
4309  p++;
4310  } else {
4311  /* no protocol means plain filename */
4312  av_strlcpy(path, url, path_size);
4313  return;
4314  }
4315 
4316  /* separate path from hostname */
4317  ls = strchr(p, '/');
4318  ls2 = strchr(p, '?');
4319  if (!ls)
4320  ls = ls2;
4321  else if (ls && ls2)
4322  ls = FFMIN(ls, ls2);
4323  if (ls)
4324  av_strlcpy(path, ls, path_size);
4325  else
4326  ls = &p[strlen(p)]; // XXX
4327 
4328  /* the rest is hostname, use that to parse auth/port */
4329  if (ls != p) {
4330  /* authorization (user[:pass]@hostname) */
4331  at2 = p;
4332  while ((at = strchr(p, '@')) && at < ls) {
4333  av_strlcpy(authorization, at2,
4334  FFMIN(authorization_size, at + 1 - at2));
4335  p = at + 1; /* skip '@' */
4336  }
4337 
4338  if (*p == '[' && (brk = strchr(p, ']')) && brk < ls) {
4339  /* [host]:port */
4340  av_strlcpy(hostname, p + 1,
4341  FFMIN(hostname_size, brk - p));
4342  if (brk[1] == ':' && port_ptr)
4343  *port_ptr = atoi(brk + 2);
4344  } else if ((col = strchr(p, ':')) && col < ls) {
4345  av_strlcpy(hostname, p,
4346  FFMIN(col + 1 - p, hostname_size));
4347  if (port_ptr)
4348  *port_ptr = atoi(col + 1);
4349  } else
4350  av_strlcpy(hostname, p,
4351  FFMIN(ls + 1 - p, hostname_size));
4352  }
4353 }
4354 
4355 char *ff_data_to_hex(char *buff, const uint8_t *src, int s, int lowercase)
4356 {
4357  int i;
4358  static const char hex_table_uc[16] = { '0', '1', '2', '3',
4359  '4', '5', '6', '7',
4360  '8', '9', 'A', 'B',
4361  'C', 'D', 'E', 'F' };
4362  static const char hex_table_lc[16] = { '0', '1', '2', '3',
4363  '4', '5', '6', '7',
4364  '8', '9', 'a', 'b',
4365  'c', 'd', 'e', 'f' };
4366  const char *hex_table = lowercase ? hex_table_lc : hex_table_uc;
4367 
4368  for (i = 0; i < s; i++) {
4369  buff[i * 2] = hex_table[src[i] >> 4];
4370  buff[i * 2 + 1] = hex_table[src[i] & 0xF];
4371  }
4372 
4373  return buff;
4374 }
4375 
4376 int ff_hex_to_data(uint8_t *data, const char *p)
4377 {
4378  int c, len, v;
4379 
4380  len = 0;
4381  v = 1;
4382  for (;;) {
4383  p += strspn(p, SPACE_CHARS);
4384  if (*p == '\0')
4385  break;
4386  c = av_toupper((unsigned char) *p++);
4387  if (c >= '0' && c <= '9')
4388  c = c - '0';
4389  else if (c >= 'A' && c <= 'F')
4390  c = c - 'A' + 10;
4391  else
4392  break;
4393  v = (v << 4) | c;
4394  if (v & 0x100) {
4395  if (data)
4396  data[len] = v;
4397  len++;
4398  v = 1;
4399  }
4400  }
4401  return len;
4402 }
4403 
4404 void avpriv_set_pts_info(AVStream *s, int pts_wrap_bits,
4405  unsigned int pts_num, unsigned int pts_den)
4406 {
4407  AVRational new_tb;
4408  if (av_reduce(&new_tb.num, &new_tb.den, pts_num, pts_den, INT_MAX)) {
4409  if (new_tb.num != pts_num)
4411  "st:%d removing common factor %d from timebase\n",
4412  s->index, pts_num / new_tb.num);
4413  } else
4415  "st:%d has too large timebase, reducing\n", s->index);
4416 
4417  if (new_tb.num <= 0 || new_tb.den <= 0) {
4419  "Ignoring attempt to set invalid timebase %d/%d for st:%d\n",
4420  new_tb.num, new_tb.den,
4421  s->index);
4422  return;
4423  }
4424  s->time_base = new_tb;
4425 #if FF_API_LAVF_AVCTX
4427  av_codec_set_pkt_timebase(s->codec, new_tb);
4429 #endif
4431  s->pts_wrap_bits = pts_wrap_bits;
4432 }
4433 
4434 void ff_parse_key_value(const char *str, ff_parse_key_val_cb callback_get_buf,
4435  void *context)
4436 {
4437  const char *ptr = str;
4438 
4439  /* Parse key=value pairs. */
4440  for (;;) {
4441  const char *key;
4442  char *dest = NULL, *dest_end;
4443  int key_len, dest_len = 0;
4444 
4445  /* Skip whitespace and potential commas. */
4446  while (*ptr && (av_isspace(*ptr) || *ptr == ','))
4447  ptr++;
4448  if (!*ptr)
4449  break;
4450 
4451  key = ptr;
4452 
4453  if (!(ptr = strchr(key, '=')))
4454  break;
4455  ptr++;
4456  key_len = ptr - key;
4457 
4458  callback_get_buf(context, key, key_len, &dest, &dest_len);
4459  dest_end = dest + dest_len - 1;
4460 
4461  if (*ptr == '\"') {
4462  ptr++;
4463  while (*ptr && *ptr != '\"') {
4464  if (*ptr == '\\') {
4465  if (!ptr[1])
4466  break;
4467  if (dest && dest < dest_end)
4468  *dest++ = ptr[1];
4469  ptr += 2;
4470  } else {
4471  if (dest && dest < dest_end)
4472  *dest++ = *ptr;
4473  ptr++;
4474  }
4475  }
4476  if (*ptr == '\"')
4477  ptr++;
4478  } else {
4479  for (; *ptr && !(av_isspace(*ptr) || *ptr == ','); ptr++)
4480  if (dest && dest < dest_end)
4481  *dest++ = *ptr;
4482  }
4483  if (dest)
4484  *dest = 0;
4485  }
4486 }
4487 
4488 int ff_find_stream_index(AVFormatContext *s, int id)
4489 {
4490  int i;
4491  for (i = 0; i < s->nb_streams; i++)
4492  if (s->streams[i]->id == id)
4493  return i;
4494  return -1;
4495 }
4496 
4498  int std_compliance)
4499 {
4500  if (ofmt) {
4501  unsigned int codec_tag;
4502  if (ofmt->query_codec)
4503  return ofmt->query_codec(codec_id, std_compliance);
4504  else if (ofmt->codec_tag)
4505  return !!av_codec_get_tag2(ofmt->codec_tag, codec_id, &codec_tag);
4506  else if (codec_id == ofmt->video_codec ||
4507  codec_id == ofmt->audio_codec ||
4508  codec_id == ofmt->subtitle_codec)
4509  return 1;
4510  }
4511  return AVERROR_PATCHWELCOME;
4512 }
4513 
4515 {
4516 #if CONFIG_NETWORK
4517  int ret;
4519  if ((ret = ff_network_init()) < 0)
4520  return ret;
4521  if ((ret = ff_tls_init()) < 0)
4522  return ret;
4523 #endif
4524  return 0;
4525 }
4526 
4528 {
4529 #if CONFIG_NETWORK
4530  ff_network_close();
4531  ff_tls_deinit();
4533 #endif
4534  return 0;
4535 }
4536 
4538  uint64_t channel_layout, int32_t sample_rate,
4540 {
4541  uint32_t flags = 0;
4542  int size = 4;
4543  uint8_t *data;
4544  if (!pkt)
4545  return AVERROR(EINVAL);
4546  if (channels) {
4547  size += 4;
4549  }
4550  if (channel_layout) {
4551  size += 8;
4553  }
4554  if (sample_rate) {
4555  size += 4;
4557  }
4558  if (width || height) {
4559  size += 8;
4561  }
4563  if (!data)
4564  return AVERROR(ENOMEM);
4565  bytestream_put_le32(&data, flags);
4566  if (channels)
4567  bytestream_put_le32(&data, channels);
4568  if (channel_layout)
4569  bytestream_put_le64(&data, channel_layout);
4570  if (sample_rate)
4571  bytestream_put_le32(&data, sample_rate);
4572  if (width || height) {
4573  bytestream_put_le32(&data, width);
4574  bytestream_put_le32(&data, height);
4575  }
4576  return 0;
4577 }
4578 
4579 AVRational av_guess_sample_aspect_ratio(AVFormatContext *format, AVStream *stream, AVFrame *frame)
4580 {
4581  AVRational undef = {0, 1};
4582  AVRational stream_sample_aspect_ratio = stream ? stream->sample_aspect_ratio : undef;
4583  AVRational codec_sample_aspect_ratio = stream && stream->codecpar ? stream->codecpar->sample_aspect_ratio : undef;
4584  AVRational frame_sample_aspect_ratio = frame ? frame->sample_aspect_ratio : codec_sample_aspect_ratio;
4585 
4586  av_reduce(&stream_sample_aspect_ratio.num, &stream_sample_aspect_ratio.den,
4587  stream_sample_aspect_ratio.num, stream_sample_aspect_ratio.den, INT_MAX);
4588  if (stream_sample_aspect_ratio.num <= 0 || stream_sample_aspect_ratio.den <= 0)
4589  stream_sample_aspect_ratio = undef;
4590 
4591  av_reduce(&frame_sample_aspect_ratio.num, &frame_sample_aspect_ratio.den,
4592  frame_sample_aspect_ratio.num, frame_sample_aspect_ratio.den, INT_MAX);
4593  if (frame_sample_aspect_ratio.num <= 0 || frame_sample_aspect_ratio.den <= 0)
4594  frame_sample_aspect_ratio = undef;
4595 
4596  if (stream_sample_aspect_ratio.num)
4597  return stream_sample_aspect_ratio;
4598  else
4599  return frame_sample_aspect_ratio;
4600 }
4601 
4602 AVRational av_guess_frame_rate(AVFormatContext *format, AVStream *st, AVFrame *frame)
4603 {
4604  AVRational fr = st->r_frame_rate;
4605  AVRational codec_fr = st->internal->avctx->framerate;
4606  AVRational avg_fr = st->avg_frame_rate;
4607 
4608  if (avg_fr.num > 0 && avg_fr.den > 0 && fr.num > 0 && fr.den > 0 &&
4609  av_q2d(avg_fr) < 70 && av_q2d(fr) > 210) {
4610  fr = avg_fr;
4611  }
4612 
4613 
4614  if (st->internal->avctx->ticks_per_frame > 1) {
4615  if ( codec_fr.num > 0 && codec_fr.den > 0 &&
4616  (fr.num == 0 || av_q2d(codec_fr) < av_q2d(fr)*0.7 && fabs(1.0 - av_q2d(av_div_q(avg_fr, fr))) > 0.1))
4617  fr = codec_fr;
4618  }
4619 
4620  return fr;
4621 }
4622 
4623 int avformat_match_stream_specifier(AVFormatContext *s, AVStream *st,
4624  const char *spec)
4625 {
4626  if (*spec <= '9' && *spec >= '0') /* opt:index */
4627  return strtol(spec, NULL, 0) == st->index;
4628  else if (*spec == 'v' || *spec == 'a' || *spec == 's' || *spec == 'd' ||
4629  *spec == 't' || *spec == 'V') { /* opt:[vasdtV] */
4630  enum AVMediaType type;
4631  int nopic = 0;
4632 
4633  switch (*spec++) {
4634  case 'v': type = AVMEDIA_TYPE_VIDEO; break;
4635  case 'a': type = AVMEDIA_TYPE_AUDIO; break;
4636  case 's': type = AVMEDIA_TYPE_SUBTITLE; break;
4637  case 'd': type = AVMEDIA_TYPE_DATA; break;
4638  case 't': type = AVMEDIA_TYPE_ATTACHMENT; break;
4639  case 'V': type = AVMEDIA_TYPE_VIDEO; nopic = 1; break;
4640  default: av_assert0(0);
4641  }
4642 #if FF_API_LAVF_AVCTX
4644  if (type != st->codecpar->codec_type
4645  && (st->codecpar->codec_type != AVMEDIA_TYPE_UNKNOWN || st->codec->codec_type != type))
4646  return 0;
4648 #else
4649  if (type != st->codecpar->codec_type)
4650  return 0;
4651 #endif
4652  if (nopic && (st->disposition & AV_DISPOSITION_ATTACHED_PIC))
4653  return 0;
4654  if (*spec++ == ':') { /* possibly followed by :index */
4655  int i, index = strtol(spec, NULL, 0);
4656  for (i = 0; i < s->nb_streams; i++) {
4657 #if FF_API_LAVF_AVCTX
4659  if ((s->streams[i]->codecpar->codec_type == type
4660  || s->streams[i]->codec->codec_type == type
4661  ) &&
4662  !(nopic && (st->disposition & AV_DISPOSITION_ATTACHED_PIC)) &&
4663  index-- == 0)
4664  return i == st->index;
4666 #else
4667  if ((s->streams[i]->codecpar->codec_type == type) &&
4668  !(nopic && (st->disposition & AV_DISPOSITION_ATTACHED_PIC)) &&
4669  index-- == 0)
4670  return i == st->index;
4671 #endif
4672  }
4673  return 0;
4674  }
4675  return 1;
4676  } else if (*spec == 'p' && *(spec + 1) == ':') {
4677  int prog_id, i, j;
4678  char *endptr;
4679  spec += 2;
4680  prog_id = strtol(spec, &endptr, 0);
4681  for (i = 0; i < s->nb_programs; i++) {
4682  if (s->programs[i]->id != prog_id)
4683  continue;
4684 
4685  if (*endptr++ == ':') {
4686  int stream_idx = strtol(endptr, NULL, 0);
4687  return stream_idx >= 0 &&
4688  stream_idx < s->programs[i]->nb_stream_indexes &&
4689  st->index == s->programs[i]->stream_index[stream_idx];
4690  }
4691 
4692  for (j = 0; j < s->programs[i]->nb_stream_indexes; j++)
4693  if (st->index == s->programs[i]->stream_index[j])
4694  return 1;
4695  }
4696  return 0;
4697  } else if (*spec == '#' ||
4698  (*spec == 'i' && *(spec + 1) == ':')) {
4699  int stream_id;
4700  char *endptr;
4701  spec += 1 + (*spec == 'i');
4702  stream_id = strtol(spec, &endptr, 0);
4703  if (!*endptr)
4704  return stream_id == st->id;
4705  } else if (*spec == 'm' && *(spec + 1) == ':') {
4707  char *key, *val;
4708  int ret;
4709 
4710  spec += 2;
4711  val = strchr(spec, ':');
4712 
4713  key = val ? av_strndup(spec, val - spec) : av_strdup(spec);
4714  if (!key)
4715  return AVERROR(ENOMEM);
4716 
4717  tag = av_dict_get(st->metadata, key, NULL, 0);
4718  if (tag) {
4719  if (!val || !strcmp(tag->value, val + 1))
4720  ret = 1;
4721  else
4722  ret = 0;
4723  } else
4724  ret = 0;
4725 
4726  av_freep(&key);
4727  return ret;
4728  } else if (*spec == 'u') {
4729  AVCodecParameters *par = st->codecpar;
4730 #if FF_API_LAVF_AVCTX
4732  AVCodecContext *codec = st->codec;
4734 #endif
4735  int val;
4736  switch (par->codec_type) {
4737  case AVMEDIA_TYPE_AUDIO:
4738  val = par->sample_rate && par->channels;
4739 #if FF_API_LAVF_AVCTX
4740  val = val || (codec->sample_rate && codec->channels);
4741 #endif
4742  if (par->format == AV_SAMPLE_FMT_NONE
4744  && codec->sample_fmt == AV_SAMPLE_FMT_NONE
4745 #endif
4746  )
4747  return 0;
4748  break;
4749  case AVMEDIA_TYPE_VIDEO:
4750  val = par->width && par->height;
4751 #if FF_API_LAVF_AVCTX
4752  val = val || (codec->width && codec->height);
4753 #endif
4754  if (par->format == AV_PIX_FMT_NONE
4756  && codec->pix_fmt == AV_PIX_FMT_NONE
4757 #endif
4758  )
4759  return 0;
4760  break;
4761  case AVMEDIA_TYPE_UNKNOWN:
4762  val = 0;
4763  break;
4764  default:
4765  val = 1;
4766  break;
4767  }
4768 #if FF_API_LAVF_AVCTX
4769  return (par->codec_id != AV_CODEC_ID_NONE || codec->codec_id != AV_CODEC_ID_NONE) && val != 0;
4770 #else
4771  return par->codec_id != AV_CODEC_ID_NONE && val != 0;
4772 #endif
4773  } else if (!*spec) /* empty specifier, matches everything */
4774  return 1;
4775 
4776  av_log(s, AV_LOG_ERROR, "Invalid stream specifier: %s.\n", spec);
4777  return AVERROR(EINVAL);