47f5ba0b11d5780a4e740273ece6198883416a24
[ffmpeg.git] / libavformat / utils.c
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"
36 #include "libavutil/time_internal.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 {
65     av_assert0(LIBAVFORMAT_VERSION_MICRO >= 100);
66     return LIBAVFORMAT_VERSION_INT;
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 {
95     if (st->pts_wrap_behavior != AV_PTS_WRAP_IGNORE &&
96         st->pts_wrap_reference != AV_NOPTS_VALUE && timestamp != AV_NOPTS_VALUE) {
97         if (st->pts_wrap_behavior == AV_PTS_WRAP_ADD_OFFSET &&
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)
109 MAKE_ACCESSORS(AVFormatContext, format, AVCodec *, video_codec)
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
117 FF_DISABLE_DEPRECATION_WARNINGS
118 MAKE_ACCESSORS(AVFormatContext, format, AVOpenCallback, open_cb)
119 FF_ENABLE_DEPRECATION_WARNINGS
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;
138     s->internal->inject_global_side_data = 1;
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);
152     dst->format_whitelist = av_strdup(src->format_whitelist);
153     dst->protocol_whitelist = av_strdup(src->protocol_whitelist);
154     dst->protocol_blacklist = av_strdup(src->protocol_blacklist);
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
168 FF_DISABLE_DEPRECATION_WARNINGS
169     if (st->codec->codec)
170         return st->codec->codec;
171 FF_ENABLE_DEPRECATION_WARNINGS
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;
181     case AVMEDIA_TYPE_SUBTITLE:
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
197 int ffio_limit(AVIOContext *s, int size)
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. */
219 static int append_packet_chunked(AVIOContext *s, AVPacket *pkt, int size)
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
260 int av_get_packet(AVIOContext *s, AVPacket *pkt, int size)
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
270 int av_append_packet(AVIOContext *s, AVPacket *pkt, int size)
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[] = {
292         { "aac",       AV_CODEC_ID_AAC,        AVMEDIA_TYPE_AUDIO },
293         { "ac3",       AV_CODEC_ID_AC3,        AVMEDIA_TYPE_AUDIO },
294         { "dts",       AV_CODEC_ID_DTS,        AVMEDIA_TYPE_AUDIO },
295         { "dvbsub",    AV_CODEC_ID_DVB_SUBTITLE,AVMEDIA_TYPE_SUBTITLE },
296         { "dvbtxt",    AV_CODEC_ID_DVB_TELETEXT,AVMEDIA_TYPE_SUBTITLE },
297         { "eac3",      AV_CODEC_ID_EAC3,       AVMEDIA_TYPE_AUDIO },
298         { "h264",      AV_CODEC_ID_H264,       AVMEDIA_TYPE_VIDEO },
299         { "hevc",      AV_CODEC_ID_HEVC,       AVMEDIA_TYPE_VIDEO },
300         { "loas",      AV_CODEC_ID_AAC_LATM,   AVMEDIA_TYPE_AUDIO },
301         { "m4v",       AV_CODEC_ID_MPEG4,      AVMEDIA_TYPE_VIDEO },
302         { "mp3",       AV_CODEC_ID_MP3,        AVMEDIA_TYPE_AUDIO },
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",
314                pd->buf_size, MAX_PROBE_PACKETS - st->probe_packets,
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
325 FF_DISABLE_DEPRECATION_WARNINGS
326                 st->codec->codec_type = st->codecpar->codec_type;
327                 st->codec->codec_id   = st->codecpar->codec_id;
328 FF_ENABLE_DEPRECATION_WARNINGS
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,
362                       AVDictionary **options)
363 {
364     int ret;
365     AVProbeData pd = { filename, NULL, 0 };
366     int score = AVPROBE_SCORE_RETRY;
367
368     if (s->pb) {
369         s->flags |= AVFMT_FLAG_CUSTOM_IO;
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++)
424         if (s->streams[i]->disposition & AV_DISPOSITION_ATTACHED_PIC &&
425             s->streams[i]->discard < AVDISCARD_ALL) {
426             if (s->streams[i]->attached_pic.size <= 0) {
427                 av_log(s, AV_LOG_WARNING,
428                     "Attached picture on stream %d has invalid size, "
429                     "ignoring\n", i);
430                 continue;
431             }
432
433             ret = add_to_pktbuf(&s->internal->raw_packet_buffer,
434                                 &s->streams[i]->attached_pic,
435                                 &s->internal->raw_packet_buffer_end, 1);
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 */
452         ret = avcodec_parameters_to_context(st->internal->avctx, st->codecpar);
453         if (ret < 0)
454             return ret;
455
456 #if FF_API_LAVF_AVCTX
457 FF_DISABLE_DEPRECATION_WARNINGS
458         /* update deprecated public codec context */
459         ret = avcodec_parameters_to_context(st->codec, st->codecpar);
460         if (ret < 0)
461             return ret;
462 FF_ENABLE_DEPRECATION_WARNINGS
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,
472                         AVInputFormat *fmt, AVDictionary **options)
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
492         s->flags |= AVFMT_FLAG_CUSTOM_IO;
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) {
502         s->protocol_whitelist = av_strdup(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) {
510         s->protocol_blacklist = av_strdup(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
523     avio_skip(s->pb, s->skip_initial_bytes);
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
533     s->duration = s->start_time = AV_NOPTS_VALUE;
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;
544             av_opt_set_defaults(s->priv_data);
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
574     s->internal->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
575
576     update_stream_avctx(s);
577
578     for (i = 0; i < s->nb_streams; i++)
579         s->streams[i]->internal->orig_codec_id = s->streams[i]->codecpar->codec_id;
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);
593     avformat_free_context(s);
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;
611     case AVMEDIA_TYPE_SUBTITLE:
612         if (s->subtitle_codec_id)
613             st->codecpar->codec_id = 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) {
629                 av_log(s, AV_LOG_WARNING,
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) {
642                 av_log(s, AV_LOG_WARNING,
643                        "nothing to probe for stream %d\n", st->index);
644             }
645         }
646
647         end=    s->internal->raw_packet_buffer_remaining_size <= 0
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);
652             if (    (st->codecpar->codec_id != AV_CODEC_ID_NONE && score > AVPROBE_SCORE_STREAM_RETRY)
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)) ?
686         AV_PTS_WRAP_ADD_OFFSET : AV_PTS_WRAP_SUB_OFFSET;
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++) {
694                 if (av_find_program_from_stream(s, NULL, 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 (;;) {
740         AVPacketList *pktl = s->internal->raw_packet_buffer;
741
742         if (pktl) {
743             *pkt = pktl->pkt;
744             st   = s->streams[pkt->stream_index];
745             if (s->internal->raw_packet_buffer_remaining_size <= 0)
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;
750                 s->internal->raw_packet_buffer_remaining_size += pkt->size;
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)) {
788             av_log(s, AV_LOG_WARNING,
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
802         if (update_wrap_reference(s, st, pkt->stream_index, pkt) && st->pts_wrap_behavior == AV_PTS_WRAP_SUB_OFFSET) {
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) */
818         if (s->use_wallclock_as_timestamps)
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,
825                             &s->internal->raw_packet_buffer_end, 0);
826         if (err)
827             return err;
828         s->internal->raw_packet_buffer_remaining_size -= pkt->size;
829
830         if ((err = probe_codec(s, st, pkt)) < 0)
831             return err;
832     }
833 }
834
835
836 /**********************************************************/
837
838 static int determinable_frame_size(AVCodecContext *avctx)
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,
853                                AVCodecParserContext *pc, AVPacket *pkt)
854 {
855     AVRational codec_framerate = s->iformat ? st->internal->avctx->framerate :
856                                               av_mul_q(av_inv_q(st->internal->avctx->time_base), (AVRational){1, st->internal->avctx->ticks_per_frame});
857     int frame_size, sample_rate;
858
859 #if FF_API_LAVF_AVCTX
860 FF_DISABLE_DEPRECATION_WARNINGS
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});
863 FF_ENABLE_DEPRECATION_WARNINGS
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 {
917     const AVCodecDescriptor *d = avcodec_descriptor_get(id);
918     if (!d)
919         return 0;
920     if (d->type == AVMEDIA_TYPE_VIDEO && !(d->props & AV_CODEC_PROP_INTRA_ONLY))
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 &&
932        avpriv_h264_has_num_reorder_frames(st->internal->avctx) == 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 &&
954                        st->codecpar->codec_id != AV_CODEC_ID_HEVC;
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];
1028     AVPacketList *pktl = s->internal->packet_buffer ? s->internal->packet_buffer : s->internal->parse_queue;
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;
1057             if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO && st->codecpar->sample_rate)
1058                 st->start_time += av_rescale_q(st->skip_samples, (AVRational){1, st->codecpar->sample_rate}, st->time_base);
1059         }
1060     }
1061
1062     if (has_decode_delay_been_guessed(st)) {
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 {
1076     AVPacketList *pktl = s->internal->packet_buffer ? s->internal->packet_buffer : s->internal->parse_queue;
1077     int64_t cur_dts    = RELATIVE_TS_BASE;
1078
1079     if (st->first_dts != AV_NOPTS_VALUE) {
1080         if (st->update_initial_durations_done)
1081             return;
1082         st->update_initial_durations_done = 1;
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         }
1102         pktl          = s->internal->packet_buffer ? s->internal->packet_buffer : s->internal->parse_queue;
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,
1127                                AVCodecParserContext *pc, AVPacket *pkt,
1128                                int64_t next_dts, int64_t next_pts)
1129 {
1130     int num, den, presentation_delayed, delay, i;
1131     int64_t offset;
1132     AVRational duration;
1133     int onein_oneout = st->codecpar->codec_id != AV_CODEC_ID_H264 &&
1134                        st->codecpar->codec_id != AV_CODEC_ID_HEVC;
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 {
1144                 av_log(s, st->dts_misordered ? AV_LOG_DEBUG : AV_LOG_WARNING,
1145                        "DTS %"PRIi64" < %"PRIi64" out of order\n",
1146                        pkt->dts,
1147                        st->last_dts_for_order_check);
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;
1276             update_initial_timestamps(s, pkt->stream_index, pkt->pts,
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
1291         if(has_decode_delay_been_guessed(st))
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
1309 FF_DISABLE_DEPRECATION_WARNINGS
1310     if (pc)
1311         pkt->convergence_duration = pc->convergence_duration;
1312 FF_ENABLE_DEPRECATION_WARNINGS
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) {
1341         av_init_packet(&flush_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
1346         compute_pkt_fields(s, st, st->parser, pkt, AV_NOPTS_VALUE, AV_NOPTS_VALUE);
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;
1379         if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
1380             if (st->internal->avctx->sample_rate > 0) {
1381                 out_pkt.duration =
1382                     av_rescale_q_rnd(st->parser->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
1394         if (st->need_parsing == AVSTREAM_PARSE_FULL_RAW)
1395             out_pkt.pos = st->parser->frame_offset;
1396
1397         if (st->parser->key_frame == 1 ||
1398             (st->parser->key_frame == -1 &&
1399              st->parser->pict_type == AV_PICTURE_TYPE_I))
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:
1421     av_packet_unref(pkt);
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");
1478                 avcodec_close(st->internal->avctx);
1479                 st->info->found_decoder = 0;
1480             }
1481
1482             ret = avcodec_parameters_to_context(st->internal->avctx, st->codecpar);
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) {
1492             av_log(s, AV_LOG_WARNING,
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",
1512                        avcodec_get_name(st->codecpar->codec_id));
1513                 /* no parser available: just output the raw packets */
1514                 st->need_parsing = AVSTREAM_PARSE_NONE;
1515             } else if (st->need_parsing == AVSTREAM_PARSE_HEADERS)
1516                 st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
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)
1520                 st->parser->flags |= PARSER_FLAG_USE_CODEC_TS;
1521         }
1522
1523         if (!st->need_parsing || !st->parser) {
1524             /* no parsing needed: we just output the packet as is */
1525             *pkt = cur_pkt;
1526             compute_pkt_fields(s, st, NULL, pkt, AV_NOPTS_VALUE, AV_NOPTS_VALUE);
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;
1537             st->codecpar->sample_rate = st->internal->avctx->sample_rate;
1538             st->codecpar->bit_rate = st->internal->avctx->bit_rate;
1539             st->codecpar->channels = st->internal->avctx->channels;
1540             st->codecpar->channel_layout = st->internal->avctx->channel_layout;
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)
1558         ret = read_from_packet_buffer(&s->internal->parse_queue, &s->internal->parse_queue_end, pkt);
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) {
1575             uint8_t *p = av_packet_new_side_data(pkt, AV_PKT_DATA_SKIP_SAMPLES, 10);
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))
1604             av_packet_merge_side_data(pkt);
1605     }
1606
1607     av_opt_get_dict_val(s, "metadata", AV_OPT_SEARCH_CHILDREN, &metadata);
1608     if (metadata) {
1609         s->event_flags |= AVFMT_EVENT_FLAG_METADATA_UPDATED;
1610         av_dict_copy(&s->metadata, metadata, 0);
1611         av_dict_free(&metadata);
1612         av_opt_set_dict_val(s, "metadata", NULL, AV_OPT_SEARCH_CHILDREN);
1613     }
1614
1615 #if FF_API_LAVF_AVCTX
1616     update_stream_avctx(s);
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
1640               ? read_from_packet_buffer(&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)) {
1688                 ret = read_from_packet_buffer(&s->internal->packet_buffer,
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;
1731     free_packet_buffer(&s->internal->parse_queue,       &s->internal->parse_queue_end);
1732     free_packet_buffer(&s->internal->packet_buffer,     &s->internal->packet_buffer_end);
1733     free_packet_buffer(&s->internal->raw_packet_buffer, &s->internal->raw_packet_buffer_end);
1734
1735     s->internal->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
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) {
1754             if (st->disposition & AV_DISPOSITION_ATTACHED_PIC)
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         }
1794         st->last_IP_pts = AV_NOPTS_VALUE;
1795         st->last_dts_for_order_check = AV_NOPTS_VALUE;
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
1802         st->probe_packets = MAX_PROBE_PACKETS;
1803
1804         for (j = 0; j < MAX_REORDER_DELAY + 1; j++)
1805             st->pts_buffer[j] = AV_NOPTS_VALUE;
1806
1807         if (s->internal->inject_global_side_data)
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);
1904     return ff_add_index_entry(&st->index_entries, &st->nb_index_entries,
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)) {
1995         s->pb->short_seek_threshold = FFMAX(s->pb->short_seek_threshold, skip);
1996     }
1997 }
1998
1999 int av_index_search_timestamp(AVStream *st, int64_t wanted_timestamp, int flags)
2000 {
2001     return ff_index_search_timestamp(st->index_entries, st->nb_index_entries,
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
2076     ff_read_frame_flush(s);
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) {
2267             av_assert0(st->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
2301     ff_read_frame_flush(s);
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;
2322         ff_read_frame_flush(s);
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) {
2339         ff_read_frame_flush(s);
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)) {
2348         ff_read_frame_flush(s);
2349         return ff_seek_frame_binary(s, stream_index, timestamp, flags);
2350     } else if (!(s->iformat->flags & AVFMT_NOGENSEARCH)) {
2351         ff_read_frame_flush(s);
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)
2375         ret = avformat_queue_attached_pictures(s);
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;
2394         ff_read_frame_flush(s);
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,
2401                                     AV_ROUND_UP   | AV_ROUND_PASS_MINMAX);
2402             max_ts = av_rescale_rnd(max_ts, time_base.den,
2403                                     time_base.num * (int64_t)AV_TIME_BASE,
2404                                     AV_ROUND_DOWN | AV_ROUND_PASS_MINMAX);
2405         }
2406
2407         ret = s->iformat->read_seek2(s, stream_index, min_ts,
2408                                      ts, max_ts, flags);
2409
2410         if (ret >= 0)
2411             ret = avformat_queue_attached_pictures(s);
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 {
2438     ff_read_frame_flush(s);
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);
2486             if (st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE || st->codecpar->codec_type == AVMEDIA_TYPE_DATA) {
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,
2492                                          AV_TIME_BASE_Q,
2493                                          AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX);
2494             if (end_time1 != AV_NOPTS_VALUE && start_time1 <= INT64_MAX - end_time1) {
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
2546     update_stream_timings(ic);
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)
2551                 st->start_time = av_rescale_q(ic->start_time, AV_TIME_BASE_Q,
2552                                               st->time_base);
2553             if (ic->duration != AV_NOPTS_VALUE)
2554                 st->duration = av_rescale_q(ic->duration, AV_TIME_BASE_Q,
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 &&
2632             st->codecpar->codec_type != AVMEDIA_TYPE_UNKNOWN)
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     }
2726     fill_all_stream_timings(ic);
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;
2734         st->last_IP_pts = AV_NOPTS_VALUE;
2735         st->last_dts_for_order_check = AV_NOPTS_VALUE;
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);
2758         ic->duration_estimation_method = AVFMT_DURATION_FROM_PTS;
2759     } else if (has_duration(ic)) {
2760         /* at least one component has timings - we use them for all
2761          * the components */
2762         fill_all_stream_timings(ic);
2763         ic->duration_estimation_method = AVFMT_DURATION_FROM_STREAM;
2764     } else {
2765         /* less precise: use bitrate info */
2766         estimate_timings_from_bit_rate(ic);
2767         ic->duration_estimation_method = AVFMT_DURATION_FROM_BITRATE;
2768     }
2769     update_stream_timings(ic);
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");
2820         if (st->codecpar->codec_id == AV_CODEC_ID_RV30 || st->codecpar->codec_id == AV_CODEC_ID_RV40)
2821             if (!st->sample_aspect_ratio.num && !st->codecpar->sample_aspect_ratio.num && !st->codec_info_nb_frames)
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,
2837                             AVDictionary **options)
2838 {
2839     AVCodecContext *avctx = st->internal->avctx;
2840     const AVCodec *codec;
2841     int got_picture = 1, ret = 0;
2842     AVFrame *frame = av_frame_alloc();
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
2885     if (avpriv_codec_get_cap_skip_frame_fill_param(avctx->codec)) {
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 &&
2893            (!has_codec_parameters(st, NULL) || !has_decode_delay_been_guessed(st) ||
2894             (!st->codec_info_nb_frames &&
2895              (avctx->codec->capabilities & AV_CODEC_CAP_CHANNEL_CONF)))) {
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:
2964             return be ? AV_CODEC_ID_PCM_F32BE : AV_CODEC_ID_PCM_F32LE;
2965         case 64:
2966             return be ? AV_CODEC_ID_PCM_F64BE : AV_CODEC_ID_PCM_F64LE;
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:
2978                 return be ? AV_CODEC_ID_PCM_S16BE : AV_CODEC_ID_PCM_S16LE;
2979             case 3:
2980                 return be ? AV_CODEC_ID_PCM_S24BE : AV_CODEC_ID_PCM_S24LE;
2981             case 4:
2982                 return be ? AV_CODEC_ID_PCM_S32BE : AV_CODEC_ID_PCM_S32LE;
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:
2991                 return be ? AV_CODEC_ID_PCM_U16BE : AV_CODEC_ID_PCM_U16LE;
2992             case 3:
2993                 return be ? AV_CODEC_ID_PCM_U24BE : AV_CODEC_ID_PCM_U24LE;
2994             case 4:
2995                 return be ? AV_CODEC_ID_PCM_U32BE : AV_CODEC_ID_PCM_U32LE;
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 && s->start_time < INT64_MAX - s->duration)
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. */
3090 static int tb_unreliable(AVCodecContext *c)
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") ||
3097         c->codec_id == AV_CODEC_ID_MPEG2VIDEO ||
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
3105 int ff_alloc_extradata(AVCodecParameters *par, int size)
3106 {
3107     int ret;
3108
3109     if (size < 0 || size >= INT32_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
3110         par->extradata = NULL;
3111         par->extradata_size = 0;
3112         return AVERROR(EINVAL);
3113     }
3114     par->extradata = av_malloc(size + AV_INPUT_BUFFER_PADDING_SIZE);
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++;
3172         st->info->rfps_duration_sum += duration;
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
3207         if (st->codecpar->codec_type != AVMEDIA_TYPE_VIDEO)
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
3266 int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
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 ||
3310             st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE) {
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) {
3321                     st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
3322                 } else if (st->need_parsing == AVSTREAM_PARSE_FULL_RAW) {
3323                     st->parser->flags |= PARSER_FLAG_USE_CODEC_TS;
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",
3328                        avcodec_get_name(st->codecpar->codec_id));
3329             }
3330         }
3331
3332         /* check if the caller has overridden the codec id */
3333 #if FF_API_LAVF_AVCTX
3334 FF_DISABLE_DEPRECATION_WARNINGS
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         }
3340 FF_ENABLE_DEPRECATION_WARNINGS
3341 #endif
3342         if (st->codecpar->codec_id != st->internal->orig_codec_id)
3343             st->internal->orig_codec_id = st->codecpar->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. */
3361         if (st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE
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
3383         ic->streams[i]->info->fps_first_dts = AV_NOPTS_VALUE;
3384         ic->streams[i]->info->fps_last_dts  = AV_NOPTS_VALUE;
3385     }
3386
3387     read_size = 0;
3388     for (;;) {
3389         int analyzed_all_streams;
3390         if (ff_check_interrupt(&ic->interrupt_callback)) {
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;
3412             if (st->disposition & AV_DISPOSITION_ATTACHED_PIC)
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) &&
3416                 st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
3417                 int count = (ic->iformat->flags & AVFMT_NOTIMESTAMPS) ?
3418                     st->info->codec_info_duration_fields/2 :
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) &&
3428                 st->codec_info_nb_frames < ((st->disposition & AV_DISPOSITION_ATTACHED_PIC) ? 1 : ic->max_ts_probe) &&
3429                 (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO ||
3430                  st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO))
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 &&
3454                     ic->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO &&
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];
3484         if (!(st->disposition & AV_DISPOSITION_ATTACHED_PIC))
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,
3503                        st->info->fps_last_dts, st->codec_info_nb_frames,
3504                        pkt->dts);
3505                 st->info->fps_first_dts =
3506                 st->info->fps_last_dts  = AV_NOPTS_VALUE;
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 &&
3512                 st->info->fps_last_dts_idx > st->info->fps_first_dts_idx &&
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,
3520                        st->info->fps_last_dts, st->codec_info_nb_frames,
3521                        pkt->dts);
3522                 st->info->fps_first_dts =
3523                 st->info->fps_last_dts  = AV_NOPTS_VALUE;
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;
3529                 st->info->fps_first_dts_idx = st->codec_info_nb_frames;
3530             }
3531             st->info->fps_last_dts     = pkt->dts;
3532             st->info->fps_last_dts_idx = st->codec_info_nb_frames;
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)
3539                 t = av_rescale_q(st->info->codec_info_duration, st->time_base, AV_TIME_BASE_Q);
3540             if (st->avg_frame_rate.num > 0)
3541                 t = FFMAX(t, av_rescale_q(st->codec_info_nb_frames, av_inv_q(st->avg_frame_rate), AV_TIME_BASE_Q));
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)
3547                 t = FFMAX(t, av_rescale_q(st->info->fps_last_dts - st->info->fps_first_dts, st->time_base, AV_TIME_BASE_Q));
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
3570         if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
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 +
3578                                                    AV_INPUT_BUFFER_PADDING_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];
3644         avcodec_close(st->internal->avctx);
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);
3655                 if (avpriv_find_pix_fmt(avpriv_get_raw_pix_fmt_tags(), 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;
3670                 av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den,
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)
3687                     av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den,
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             }
3701             if (st->display_aspect_ratio.num && st->display_aspect_ratio.den) {
3702                 AVRational hw_ratio = { avctx->height, avctx->width };
3703                 st->sample_aspect_ratio = av_mul_q(st->display_aspect_ratio,
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 =
3709                     av_get_bits_per_sample(avctx->codec_id);
3710             // set stream disposition based on audio service type
3711             switch (avctx->audio_service_type) {
3712             case AV_AUDIO_SERVICE_TYPE_EFFECTS:
3713                 st->disposition = AV_DISPOSITION_CLEAN_EFFECTS;
3714                 break;
3715             case AV_AUDIO_SERVICE_TYPE_VISUALLY_IMPAIRED:
3716                 st->disposition = AV_DISPOSITION_VISUAL_IMPAIRED;
3717                 break;
3718             case AV_AUDIO_SERVICE_TYPE_HEARING_IMPAIRED:
3719                 st->disposition = AV_DISPOSITION_HEARING_IMPAIRED;
3720                 break;
3721             case AV_AUDIO_SERVICE_TYPE_COMMENTARY:
3722                 st->disposition = AV_DISPOSITION_COMMENT;
3723                 break;
3724             case AV_AUDIO_SERVICE_TYPE_KARAOKE:
3725                 st->disposition = AV_DISPOSITION_KARAOKE;
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) {
3745             ret = avcodec_parameters_to_context(st->internal->avctx, st->codecpar);
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
3761     compute_chapters_end(ic);
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;
3770             ret = avcodec_parameters_from_context(st->codecpar, st->internal->avctx);
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
3781 FF_DISABLE_DEPRECATION_WARNINGS
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) {
3789             av_codec_set_lowres(st->codec, av_codec_get_lowres(st->internal->avctx));
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         }
3808 FF_ENABLE_DEPRECATION_WARNINGS
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 &&
3869             st->disposition & (AV_DISPOSITION_HEARING_IMPAIRED |
3870                                AV_DISPOSITION_VISUAL_IMPAIRED))
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)
3878                     ret = AVERROR_DECODER_NOT_FOUND;
3879                 continue;
3880             }
3881         }
3882         count = st->codec_info_nb_frames;
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)
3942         av_packet_unref(&st->attached_pic);
3943
3944     if (st->internal) {
3945         avcodec_free_context(&st->internal->avctx);
3946     }
3947     av_freep(&st->internal);
3948
3949     av_dict_free(&st->metadata);
3950     avcodec_parameters_free(&st->codecpar);
3951     av_freep(&st->probe_data.buf);
3952     av_freep(&st->index_entries);
3953 #if FF_API_LAVF_AVCTX
3954 FF_DISABLE_DEPRECATION_WARNINGS
3955     av_freep(&st->codec->extradata);
3956     av_freep(&st->codec->subtitle_header);
3957     av_freep(&st->codec);
3958 FF_ENABLE_DEPRECATION_WARNINGS
3959 #endif
3960     av_freep(&st->priv_data);
3961     if (st->info)
3962         av_freep(&st->info->duration_error);
3963     av_freep(&st->info);
3964     av_freep(&st->recommended_encoder_configuration);
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--) {
4003         av_dict_free(&s->chapters[s->nb_chapters]->metadata);
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
4035     avformat_free_context(s);
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
4065 FF_DISABLE_DEPRECATION_WARNINGS
4066     st->codec = avcodec_alloc_context3(c);
4067     if (!st->codec) {
4068         av_free(st->info);
4069         av_free(st);
4070         return NULL;
4071     }
4072 FF_ENABLE_DEPRECATION_WARNINGS
4073 #endif
4074
4075     st->internal = av_mallocz(sizeof(*st->internal));
4076     if (!st->internal)
4077         goto fail;
4078
4079     st->codecpar = avcodec_parameters_alloc();
4080     if (!st->codecpar)
4081         goto fail;
4082
4083     st->internal->avctx = avcodec_alloc_context3(NULL);
4084     if (!st->internal->avctx)
4085         goto fail;
4086
4087     if (s->iformat) {
4088 #if FF_API_LAVF_AVCTX
4089 FF_DISABLE_DEPRECATION_WARNINGS
4090         /* no default bitrate if decoding */
4091         st->codec->bit_rate = 0;
4092 FF_ENABLE_DEPRECATION_WARNINGS
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;
4110     st->probe_packets = MAX_PROBE_PACKETS;
4111     st->pts_wrap_reference = AV_NOPTS_VALUE;
4112     st->pts_wrap_behavior = AV_PTS_WRAP_IGNORE;
4113
4114     st->last_IP_pts = AV_NOPTS_VALUE;
4115     st->last_dts_for_order_check = AV_NOPTS_VALUE;
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
4124     st->info->fps_first_dts = AV_NOPTS_VALUE;
4125     st->info->fps_last_dts  = AV_NOPTS_VALUE;
4126
4127     st->inject_global_side_data = s->internal->inject_global_side_data;
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