avformat/utils: fix memory leak in avformat_free_context
[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(const 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, const 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, const 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 static const AVCodec *find_probe_decoder(AVFormatContext *s, const AVStream *st, enum AVCodecID codec_id)
190 {
191     const AVCodec *codec;
192
193 #if CONFIG_H264_DECODER
194     /* Other parts of the code assume this decoder to be used for h264,
195      * so force it if possible. */
196     if (codec_id == AV_CODEC_ID_H264)
197         return avcodec_find_decoder_by_name("h264");
198 #endif
199
200     codec = find_decoder(s, st, codec_id);
201     if (!codec)
202         return NULL;
203
204     if (codec->capabilities & AV_CODEC_CAP_AVOID_PROBING) {
205         const AVCodec *probe_codec = NULL;
206         while (probe_codec = av_codec_next(probe_codec)) {
207             if (probe_codec->id == codec_id &&
208                     av_codec_is_decoder(probe_codec) &&
209                     !(probe_codec->capabilities & (AV_CODEC_CAP_AVOID_PROBING | AV_CODEC_CAP_EXPERIMENTAL))) {
210                 return probe_codec;
211             }
212         }
213     }
214
215     return codec;
216 }
217
218 int av_format_get_probe_score(const AVFormatContext *s)
219 {
220     return s->probe_score;
221 }
222
223 /* an arbitrarily chosen "sane" max packet size -- 50M */
224 #define SANE_CHUNK_SIZE (50000000)
225
226 int ffio_limit(AVIOContext *s, int size)
227 {
228     if (s->maxsize>= 0) {
229         int64_t remaining= s->maxsize - avio_tell(s);
230         if (remaining < size) {
231             int64_t newsize = avio_size(s);
232             if (!s->maxsize || s->maxsize<newsize)
233                 s->maxsize = newsize - !newsize;
234             remaining= s->maxsize - avio_tell(s);
235             remaining= FFMAX(remaining, 0);
236         }
237
238         if (s->maxsize>= 0 && remaining+1 < size) {
239             av_log(NULL, remaining ? AV_LOG_ERROR : AV_LOG_DEBUG, "Truncating packet of size %d to %"PRId64"\n", size, remaining+1);
240             size = remaining+1;
241         }
242     }
243     return size;
244 }
245
246 /* Read the data in sane-sized chunks and append to pkt.
247  * Return the number of bytes read or an error. */
248 static int append_packet_chunked(AVIOContext *s, AVPacket *pkt, int size)
249 {
250     int64_t orig_pos   = pkt->pos; // av_grow_packet might reset pos
251     int orig_size      = pkt->size;
252     int ret;
253
254     do {
255         int prev_size = pkt->size;
256         int read_size;
257
258         /* When the caller requests a lot of data, limit it to the amount
259          * left in file or SANE_CHUNK_SIZE when it is not known. */
260         read_size = size;
261         if (read_size > SANE_CHUNK_SIZE/10) {
262             read_size = ffio_limit(s, read_size);
263             // If filesize/maxsize is unknown, limit to SANE_CHUNK_SIZE
264             if (s->maxsize < 0)
265                 read_size = FFMIN(read_size, SANE_CHUNK_SIZE);
266         }
267
268         ret = av_grow_packet(pkt, read_size);
269         if (ret < 0)
270             break;
271
272         ret = avio_read(s, pkt->data + prev_size, read_size);
273         if (ret != read_size) {
274             av_shrink_packet(pkt, prev_size + FFMAX(ret, 0));
275             break;
276         }
277
278         size -= read_size;
279     } while (size > 0);
280     if (size > 0)
281         pkt->flags |= AV_PKT_FLAG_CORRUPT;
282
283     pkt->pos = orig_pos;
284     if (!pkt->size)
285         av_packet_unref(pkt);
286     return pkt->size > orig_size ? pkt->size - orig_size : ret;
287 }
288
289 int av_get_packet(AVIOContext *s, AVPacket *pkt, int size)
290 {
291     av_init_packet(pkt);
292     pkt->data = NULL;
293     pkt->size = 0;
294     pkt->pos  = avio_tell(s);
295
296     return append_packet_chunked(s, pkt, size);
297 }
298
299 int av_append_packet(AVIOContext *s, AVPacket *pkt, int size)
300 {
301     if (!pkt->size)
302         return av_get_packet(s, pkt, size);
303     return append_packet_chunked(s, pkt, size);
304 }
305
306 int av_filename_number_test(const char *filename)
307 {
308     char buf[1024];
309     return filename &&
310            (av_get_frame_filename(buf, sizeof(buf), filename, 1) >= 0);
311 }
312
313 static int set_codec_from_probe_data(AVFormatContext *s, AVStream *st,
314                                      AVProbeData *pd)
315 {
316     static const struct {
317         const char *name;
318         enum AVCodecID id;
319         enum AVMediaType type;
320     } fmt_id_type[] = {
321         { "aac",       AV_CODEC_ID_AAC,        AVMEDIA_TYPE_AUDIO },
322         { "ac3",       AV_CODEC_ID_AC3,        AVMEDIA_TYPE_AUDIO },
323         { "dts",       AV_CODEC_ID_DTS,        AVMEDIA_TYPE_AUDIO },
324         { "dvbsub",    AV_CODEC_ID_DVB_SUBTITLE,AVMEDIA_TYPE_SUBTITLE },
325         { "dvbtxt",    AV_CODEC_ID_DVB_TELETEXT,AVMEDIA_TYPE_SUBTITLE },
326         { "eac3",      AV_CODEC_ID_EAC3,       AVMEDIA_TYPE_AUDIO },
327         { "h264",      AV_CODEC_ID_H264,       AVMEDIA_TYPE_VIDEO },
328         { "hevc",      AV_CODEC_ID_HEVC,       AVMEDIA_TYPE_VIDEO },
329         { "loas",      AV_CODEC_ID_AAC_LATM,   AVMEDIA_TYPE_AUDIO },
330         { "m4v",       AV_CODEC_ID_MPEG4,      AVMEDIA_TYPE_VIDEO },
331         { "mjpeg_2000",AV_CODEC_ID_JPEG2000,   AVMEDIA_TYPE_VIDEO },
332         { "mp3",       AV_CODEC_ID_MP3,        AVMEDIA_TYPE_AUDIO },
333         { "mpegvideo", AV_CODEC_ID_MPEG2VIDEO, AVMEDIA_TYPE_VIDEO },
334         { "truehd",    AV_CODEC_ID_TRUEHD,     AVMEDIA_TYPE_AUDIO },
335         { 0 }
336     };
337     int score;
338     AVInputFormat *fmt = av_probe_input_format3(pd, 1, &score);
339
340     if (fmt) {
341         int i;
342         av_log(s, AV_LOG_DEBUG,
343                "Probe with size=%d, packets=%d detected %s with score=%d\n",
344                pd->buf_size, MAX_PROBE_PACKETS - st->probe_packets,
345                fmt->name, score);
346         for (i = 0; fmt_id_type[i].name; i++) {
347             if (!strcmp(fmt->name, fmt_id_type[i].name)) {
348                 if (fmt_id_type[i].type != AVMEDIA_TYPE_AUDIO &&
349                     st->codecpar->sample_rate)
350                     continue;
351                 if (st->request_probe > score &&
352                     st->codecpar->codec_id != fmt_id_type[i].id)
353                     continue;
354                 st->codecpar->codec_id   = fmt_id_type[i].id;
355                 st->codecpar->codec_type = fmt_id_type[i].type;
356                 st->internal->need_context_update = 1;
357 #if FF_API_LAVF_AVCTX
358 FF_DISABLE_DEPRECATION_WARNINGS
359                 st->codec->codec_type = st->codecpar->codec_type;
360                 st->codec->codec_id   = st->codecpar->codec_id;
361 FF_ENABLE_DEPRECATION_WARNINGS
362 #endif
363                 return score;
364             }
365         }
366     }
367     return 0;
368 }
369
370 /************************************************************/
371 /* input media file */
372
373 int av_demuxer_open(AVFormatContext *ic) {
374     int err;
375
376     if (ic->format_whitelist && av_match_list(ic->iformat->name, ic->format_whitelist, ',') <= 0) {
377         av_log(ic, AV_LOG_ERROR, "Format not on whitelist \'%s\'\n", ic->format_whitelist);
378         return AVERROR(EINVAL);
379     }
380
381     if (ic->iformat->read_header) {
382         err = ic->iformat->read_header(ic);
383         if (err < 0)
384             return err;
385     }
386
387     if (ic->pb && !ic->internal->data_offset)
388         ic->internal->data_offset = avio_tell(ic->pb);
389
390     return 0;
391 }
392
393 /* Open input file and probe the format if necessary. */
394 static int init_input(AVFormatContext *s, const char *filename,
395                       AVDictionary **options)
396 {
397     int ret;
398     AVProbeData pd = { filename, NULL, 0 };
399     int score = AVPROBE_SCORE_RETRY;
400
401     if (s->pb) {
402         s->flags |= AVFMT_FLAG_CUSTOM_IO;
403         if (!s->iformat)
404             return av_probe_input_buffer2(s->pb, &s->iformat, filename,
405                                          s, 0, s->format_probesize);
406         else if (s->iformat->flags & AVFMT_NOFILE)
407             av_log(s, AV_LOG_WARNING, "Custom AVIOContext makes no sense and "
408                                       "will be ignored with AVFMT_NOFILE format.\n");
409         return 0;
410     }
411
412     if ((s->iformat && s->iformat->flags & AVFMT_NOFILE) ||
413         (!s->iformat && (s->iformat = av_probe_input_format2(&pd, 0, &score))))
414         return score;
415
416     if ((ret = s->io_open(s, &s->pb, filename, AVIO_FLAG_READ | s->avio_flags, options)) < 0)
417         return ret;
418
419     if (s->iformat)
420         return 0;
421     return av_probe_input_buffer2(s->pb, &s->iformat, filename,
422                                  s, 0, s->format_probesize);
423 }
424
425 static int add_to_pktbuf(AVPacketList **packet_buffer, AVPacket *pkt,
426                          AVPacketList **plast_pktl, int ref)
427 {
428     AVPacketList *pktl = av_mallocz(sizeof(AVPacketList));
429     int ret;
430
431     if (!pktl)
432         return AVERROR(ENOMEM);
433
434     if (ref) {
435         if ((ret = av_packet_ref(&pktl->pkt, pkt)) < 0) {
436             av_free(pktl);
437             return ret;
438         }
439     } else {
440         pktl->pkt = *pkt;
441     }
442
443     if (*packet_buffer)
444         (*plast_pktl)->next = pktl;
445     else
446         *packet_buffer = pktl;
447
448     /* Add the packet in the buffered packet list. */
449     *plast_pktl = pktl;
450     return 0;
451 }
452
453 int avformat_queue_attached_pictures(AVFormatContext *s)
454 {
455     int i, ret;
456     for (i = 0; i < s->nb_streams; i++)
457         if (s->streams[i]->disposition & AV_DISPOSITION_ATTACHED_PIC &&
458             s->streams[i]->discard < AVDISCARD_ALL) {
459             if (s->streams[i]->attached_pic.size <= 0) {
460                 av_log(s, AV_LOG_WARNING,
461                     "Attached picture on stream %d has invalid size, "
462                     "ignoring\n", i);
463                 continue;
464             }
465
466             ret = add_to_pktbuf(&s->internal->raw_packet_buffer,
467                                 &s->streams[i]->attached_pic,
468                                 &s->internal->raw_packet_buffer_end, 1);
469             if (ret < 0)
470                 return ret;
471         }
472     return 0;
473 }
474
475 static int update_stream_avctx(AVFormatContext *s)
476 {
477     int i, ret;
478     for (i = 0; i < s->nb_streams; i++) {
479         AVStream *st = s->streams[i];
480
481         if (!st->internal->need_context_update)
482             continue;
483
484         /* close parser, because it depends on the codec */
485         if (st->parser && st->internal->avctx->codec_id != st->codecpar->codec_id) {
486             av_parser_close(st->parser);
487             st->parser = NULL;
488         }
489
490         /* update internal codec context, for the parser */
491         ret = avcodec_parameters_to_context(st->internal->avctx, st->codecpar);
492         if (ret < 0)
493             return ret;
494
495 #if FF_API_LAVF_AVCTX
496 FF_DISABLE_DEPRECATION_WARNINGS
497         /* update deprecated public codec context */
498         ret = avcodec_parameters_to_context(st->codec, st->codecpar);
499         if (ret < 0)
500             return ret;
501 FF_ENABLE_DEPRECATION_WARNINGS
502 #endif
503
504         st->internal->need_context_update = 0;
505     }
506     return 0;
507 }
508
509
510 int avformat_open_input(AVFormatContext **ps, const char *filename,
511                         AVInputFormat *fmt, AVDictionary **options)
512 {
513     AVFormatContext *s = *ps;
514     int i, ret = 0;
515     AVDictionary *tmp = NULL;
516     ID3v2ExtraMeta *id3v2_extra_meta = NULL;
517
518     if (!s && !(s = avformat_alloc_context()))
519         return AVERROR(ENOMEM);
520     if (!s->av_class) {
521         av_log(NULL, AV_LOG_ERROR, "Input context has not been properly allocated by avformat_alloc_context() and is not NULL either\n");
522         return AVERROR(EINVAL);
523     }
524     if (fmt)
525         s->iformat = fmt;
526
527     if (options)
528         av_dict_copy(&tmp, *options, 0);
529
530     if (s->pb) // must be before any goto fail
531         s->flags |= AVFMT_FLAG_CUSTOM_IO;
532
533     if ((ret = av_opt_set_dict(s, &tmp)) < 0)
534         goto fail;
535
536     av_strlcpy(s->filename, filename ? filename : "", sizeof(s->filename));
537     if ((ret = init_input(s, filename, &tmp)) < 0)
538         goto fail;
539     s->probe_score = ret;
540
541     if (!s->protocol_whitelist && s->pb && s->pb->protocol_whitelist) {
542         s->protocol_whitelist = av_strdup(s->pb->protocol_whitelist);
543         if (!s->protocol_whitelist) {
544             ret = AVERROR(ENOMEM);
545             goto fail;
546         }
547     }
548
549     if (!s->protocol_blacklist && s->pb && s->pb->protocol_blacklist) {
550         s->protocol_blacklist = av_strdup(s->pb->protocol_blacklist);
551         if (!s->protocol_blacklist) {
552             ret = AVERROR(ENOMEM);
553             goto fail;
554         }
555     }
556
557     if (s->format_whitelist && av_match_list(s->iformat->name, s->format_whitelist, ',') <= 0) {
558         av_log(s, AV_LOG_ERROR, "Format not on whitelist \'%s\'\n", s->format_whitelist);
559         ret = AVERROR(EINVAL);
560         goto fail;
561     }
562
563     avio_skip(s->pb, s->skip_initial_bytes);
564
565     /* Check filename in case an image number is expected. */
566     if (s->iformat->flags & AVFMT_NEEDNUMBER) {
567         if (!av_filename_number_test(filename)) {
568             ret = AVERROR(EINVAL);
569             goto fail;
570         }
571     }
572
573     s->duration = s->start_time = AV_NOPTS_VALUE;
574
575     /* Allocate private data. */
576     if (s->iformat->priv_data_size > 0) {
577         if (!(s->priv_data = av_mallocz(s->iformat->priv_data_size))) {
578             ret = AVERROR(ENOMEM);
579             goto fail;
580         }
581         if (s->iformat->priv_class) {
582             *(const AVClass **) s->priv_data = s->iformat->priv_class;
583             av_opt_set_defaults(s->priv_data);
584             if ((ret = av_opt_set_dict(s->priv_data, &tmp)) < 0)
585                 goto fail;
586         }
587     }
588
589     /* e.g. AVFMT_NOFILE formats will not have a AVIOContext */
590     if (s->pb)
591         ff_id3v2_read_dict(s->pb, &s->internal->id3v2_meta, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta);
592
593
594     if (!(s->flags&AVFMT_FLAG_PRIV_OPT) && s->iformat->read_header)
595         if ((ret = s->iformat->read_header(s)) < 0)
596             goto fail;
597
598     if (!s->metadata) {
599         s->metadata = s->internal->id3v2_meta;
600         s->internal->id3v2_meta = NULL;
601     } else if (s->internal->id3v2_meta) {
602         int level = AV_LOG_WARNING;
603         if (s->error_recognition & AV_EF_COMPLIANT)
604             level = AV_LOG_ERROR;
605         av_log(s, level, "Discarding ID3 tags because more suitable tags were found.\n");
606         av_dict_free(&s->internal->id3v2_meta);
607         if (s->error_recognition & AV_EF_EXPLODE)
608             return AVERROR_INVALIDDATA;
609     }
610
611     if (id3v2_extra_meta) {
612         if (!strcmp(s->iformat->name, "mp3") || !strcmp(s->iformat->name, "aac") ||
613             !strcmp(s->iformat->name, "tta")) {
614             if ((ret = ff_id3v2_parse_apic(s, &id3v2_extra_meta)) < 0)
615                 goto fail;
616         } else
617             av_log(s, AV_LOG_DEBUG, "demuxer does not support additional id3 data, skipping\n");
618     }
619     ff_id3v2_free_extra_meta(&id3v2_extra_meta);
620
621     if ((ret = avformat_queue_attached_pictures(s)) < 0)
622         goto fail;
623
624     if (!(s->flags&AVFMT_FLAG_PRIV_OPT) && s->pb && !s->internal->data_offset)
625         s->internal->data_offset = avio_tell(s->pb);
626
627     s->internal->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
628
629     update_stream_avctx(s);
630
631     for (i = 0; i < s->nb_streams; i++)
632         s->streams[i]->internal->orig_codec_id = s->streams[i]->codecpar->codec_id;
633
634     if (options) {
635         av_dict_free(options);
636         *options = tmp;
637     }
638     *ps = s;
639     return 0;
640
641 fail:
642     ff_id3v2_free_extra_meta(&id3v2_extra_meta);
643     av_dict_free(&tmp);
644     if (s->pb && !(s->flags & AVFMT_FLAG_CUSTOM_IO))
645         avio_closep(&s->pb);
646     avformat_free_context(s);
647     *ps = NULL;
648     return ret;
649 }
650
651 /*******************************************************/
652
653 static void force_codec_ids(AVFormatContext *s, AVStream *st)
654 {
655     switch (st->codecpar->codec_type) {
656     case AVMEDIA_TYPE_VIDEO:
657         if (s->video_codec_id)
658             st->codecpar->codec_id = s->video_codec_id;
659         break;
660     case AVMEDIA_TYPE_AUDIO:
661         if (s->audio_codec_id)
662             st->codecpar->codec_id = s->audio_codec_id;
663         break;
664     case AVMEDIA_TYPE_SUBTITLE:
665         if (s->subtitle_codec_id)
666             st->codecpar->codec_id = s->subtitle_codec_id;
667         break;
668     case AVMEDIA_TYPE_DATA:
669         if (s->data_codec_id)
670             st->codecpar->codec_id = s->data_codec_id;
671         break;
672     }
673 }
674
675 static int probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt)
676 {
677     if (st->request_probe>0) {
678         AVProbeData *pd = &st->probe_data;
679         int end;
680         av_log(s, AV_LOG_DEBUG, "probing stream %d pp:%d\n", st->index, st->probe_packets);
681         --st->probe_packets;
682
683         if (pkt) {
684             uint8_t *new_buf = av_realloc(pd->buf, pd->buf_size+pkt->size+AVPROBE_PADDING_SIZE);
685             if (!new_buf) {
686                 av_log(s, AV_LOG_WARNING,
687                        "Failed to reallocate probe buffer for stream %d\n",
688                        st->index);
689                 goto no_packet;
690             }
691             pd->buf = new_buf;
692             memcpy(pd->buf + pd->buf_size, pkt->data, pkt->size);
693             pd->buf_size += pkt->size;
694             memset(pd->buf + pd->buf_size, 0, AVPROBE_PADDING_SIZE);
695         } else {
696 no_packet:
697             st->probe_packets = 0;
698             if (!pd->buf_size) {
699                 av_log(s, AV_LOG_WARNING,
700                        "nothing to probe for stream %d\n", st->index);
701             }
702         }
703
704         end=    s->internal->raw_packet_buffer_remaining_size <= 0
705                 || st->probe_packets<= 0;
706
707         if (end || av_log2(pd->buf_size) != av_log2(pd->buf_size - pkt->size)) {
708             int score = set_codec_from_probe_data(s, st, pd);
709             if (    (st->codecpar->codec_id != AV_CODEC_ID_NONE && score > AVPROBE_SCORE_STREAM_RETRY)
710                 || end) {
711                 pd->buf_size = 0;
712                 av_freep(&pd->buf);
713                 st->request_probe = -1;
714                 if (st->codecpar->codec_id != AV_CODEC_ID_NONE) {
715                     av_log(s, AV_LOG_DEBUG, "probed stream %d\n", st->index);
716                 } else
717                     av_log(s, AV_LOG_WARNING, "probed stream %d failed\n", st->index);
718             }
719             force_codec_ids(s, st);
720         }
721     }
722     return 0;
723 }
724
725 static int update_wrap_reference(AVFormatContext *s, AVStream *st, int stream_index, AVPacket *pkt)
726 {
727     int64_t ref = pkt->dts;
728     int i, pts_wrap_behavior;
729     int64_t pts_wrap_reference;
730     AVProgram *first_program;
731
732     if (ref == AV_NOPTS_VALUE)
733         ref = pkt->pts;
734     if (st->pts_wrap_reference != AV_NOPTS_VALUE || st->pts_wrap_bits >= 63 || ref == AV_NOPTS_VALUE || !s->correct_ts_overflow)
735         return 0;
736     ref &= (1LL << st->pts_wrap_bits)-1;
737
738     // reference time stamp should be 60 s before first time stamp
739     pts_wrap_reference = ref - av_rescale(60, st->time_base.den, st->time_base.num);
740     // if first time stamp is not more than 1/8 and 60s before the wrap point, subtract rather than add wrap offset
741     pts_wrap_behavior = (ref < (1LL << st->pts_wrap_bits) - (1LL << st->pts_wrap_bits-3)) ||
742         (ref < (1LL << st->pts_wrap_bits) - av_rescale(60, st->time_base.den, st->time_base.num)) ?
743         AV_PTS_WRAP_ADD_OFFSET : AV_PTS_WRAP_SUB_OFFSET;
744
745     first_program = av_find_program_from_stream(s, NULL, stream_index);
746
747     if (!first_program) {
748         int default_stream_index = av_find_default_stream_index(s);
749         if (s->streams[default_stream_index]->pts_wrap_reference == AV_NOPTS_VALUE) {
750             for (i = 0; i < s->nb_streams; i++) {
751                 if (av_find_program_from_stream(s, NULL, i))
752                     continue;
753                 s->streams[i]->pts_wrap_reference = pts_wrap_reference;
754                 s->streams[i]->pts_wrap_behavior = pts_wrap_behavior;
755             }
756         }
757         else {
758             st->pts_wrap_reference = s->streams[default_stream_index]->pts_wrap_reference;
759             st->pts_wrap_behavior = s->streams[default_stream_index]->pts_wrap_behavior;
760         }
761     }
762     else {
763         AVProgram *program = first_program;
764         while (program) {
765             if (program->pts_wrap_reference != AV_NOPTS_VALUE) {
766                 pts_wrap_reference = program->pts_wrap_reference;
767                 pts_wrap_behavior = program->pts_wrap_behavior;
768                 break;
769             }
770             program = av_find_program_from_stream(s, program, stream_index);
771         }
772
773         // update every program with differing pts_wrap_reference
774         program = first_program;
775         while (program) {
776             if (program->pts_wrap_reference != pts_wrap_reference) {
777                 for (i = 0; i<program->nb_stream_indexes; i++) {
778                     s->streams[program->stream_index[i]]->pts_wrap_reference = pts_wrap_reference;
779                     s->streams[program->stream_index[i]]->pts_wrap_behavior = pts_wrap_behavior;
780                 }
781
782                 program->pts_wrap_reference = pts_wrap_reference;
783                 program->pts_wrap_behavior = pts_wrap_behavior;
784             }
785             program = av_find_program_from_stream(s, program, stream_index);
786         }
787     }
788     return 1;
789 }
790
791 int ff_read_packet(AVFormatContext *s, AVPacket *pkt)
792 {
793     int ret, i, err;
794     AVStream *st;
795
796     for (;;) {
797         AVPacketList *pktl = s->internal->raw_packet_buffer;
798
799         if (pktl) {
800             *pkt = pktl->pkt;
801             st   = s->streams[pkt->stream_index];
802             if (s->internal->raw_packet_buffer_remaining_size <= 0)
803                 if ((err = probe_codec(s, st, NULL)) < 0)
804                     return err;
805             if (st->request_probe <= 0) {
806                 s->internal->raw_packet_buffer                 = pktl->next;
807                 s->internal->raw_packet_buffer_remaining_size += pkt->size;
808                 av_free(pktl);
809                 return 0;
810             }
811         }
812
813         pkt->data = NULL;
814         pkt->size = 0;
815         av_init_packet(pkt);
816         ret = s->iformat->read_packet(s, pkt);
817         if (ret < 0) {
818             /* Some demuxers return FFERROR_REDO when they consume
819                data and discard it (ignored streams, junk, extradata).
820                We must re-call the demuxer to get the real packet. */
821             if (ret == FFERROR_REDO)
822                 continue;
823             if (!pktl || ret == AVERROR(EAGAIN))
824                 return ret;
825             for (i = 0; i < s->nb_streams; i++) {
826                 st = s->streams[i];
827                 if (st->probe_packets || st->request_probe > 0)
828                     if ((err = probe_codec(s, st, NULL)) < 0)
829                         return err;
830                 av_assert0(st->request_probe <= 0);
831             }
832             continue;
833         }
834
835         if (!pkt->buf) {
836             AVPacket tmp = { 0 };
837             ret = av_packet_ref(&tmp, pkt);
838             if (ret < 0)
839                 return ret;
840             *pkt = tmp;
841         }
842
843         if ((s->flags & AVFMT_FLAG_DISCARD_CORRUPT) &&
844             (pkt->flags & AV_PKT_FLAG_CORRUPT)) {
845             av_log(s, AV_LOG_WARNING,
846                    "Dropped corrupted packet (stream = %d)\n",
847                    pkt->stream_index);
848             av_packet_unref(pkt);
849             continue;
850         }
851
852         if (pkt->stream_index >= (unsigned)s->nb_streams) {
853             av_log(s, AV_LOG_ERROR, "Invalid stream index %d\n", pkt->stream_index);
854             continue;
855         }
856
857         st = s->streams[pkt->stream_index];
858
859         if (update_wrap_reference(s, st, pkt->stream_index, pkt) && st->pts_wrap_behavior == AV_PTS_WRAP_SUB_OFFSET) {
860             // correct first time stamps to negative values
861             if (!is_relative(st->first_dts))
862                 st->first_dts = wrap_timestamp(st, st->first_dts);
863             if (!is_relative(st->start_time))
864                 st->start_time = wrap_timestamp(st, st->start_time);
865             if (!is_relative(st->cur_dts))
866                 st->cur_dts = wrap_timestamp(st, st->cur_dts);
867         }
868
869         pkt->dts = wrap_timestamp(st, pkt->dts);
870         pkt->pts = wrap_timestamp(st, pkt->pts);
871
872         force_codec_ids(s, st);
873
874         /* TODO: audio: time filter; video: frame reordering (pts != dts) */
875         if (s->use_wallclock_as_timestamps)
876             pkt->dts = pkt->pts = av_rescale_q(av_gettime(), AV_TIME_BASE_Q, st->time_base);
877
878         if (!pktl && st->request_probe <= 0)
879             return ret;
880
881         err = add_to_pktbuf(&s->internal->raw_packet_buffer, pkt,
882                             &s->internal->raw_packet_buffer_end, 0);
883         if (err)
884             return err;
885         s->internal->raw_packet_buffer_remaining_size -= pkt->size;
886
887         if ((err = probe_codec(s, st, pkt)) < 0)
888             return err;
889     }
890 }
891
892
893 /**********************************************************/
894
895 static int determinable_frame_size(AVCodecContext *avctx)
896 {
897     switch(avctx->codec_id) {
898     case AV_CODEC_ID_MP1:
899     case AV_CODEC_ID_MP2:
900     case AV_CODEC_ID_MP3:
901         return 1;
902     }
903
904     return 0;
905 }
906
907 /**
908  * Return the frame duration in seconds. Return 0 if not available.
909  */
910 void ff_compute_frame_duration(AVFormatContext *s, int *pnum, int *pden, AVStream *st,
911                                AVCodecParserContext *pc, AVPacket *pkt)
912 {
913     AVRational codec_framerate = s->iformat ? st->internal->avctx->framerate :
914                                               av_mul_q(av_inv_q(st->internal->avctx->time_base), (AVRational){1, st->internal->avctx->ticks_per_frame});
915     int frame_size, sample_rate;
916
917 #if FF_API_LAVF_AVCTX
918 FF_DISABLE_DEPRECATION_WARNINGS
919     if ((!codec_framerate.den || !codec_framerate.num) && st->codec->time_base.den && st->codec->time_base.num)
920         codec_framerate = av_mul_q(av_inv_q(st->codec->time_base), (AVRational){1, st->codec->ticks_per_frame});
921 FF_ENABLE_DEPRECATION_WARNINGS
922 #endif
923
924     *pnum = 0;
925     *pden = 0;
926     switch (st->codecpar->codec_type) {
927     case AVMEDIA_TYPE_VIDEO:
928         if (st->r_frame_rate.num && !pc && s->iformat) {
929             *pnum = st->r_frame_rate.den;
930             *pden = st->r_frame_rate.num;
931         } else if (st->time_base.num * 1000LL > st->time_base.den) {
932             *pnum = st->time_base.num;
933             *pden = st->time_base.den;
934         } else if (codec_framerate.den * 1000LL > codec_framerate.num) {
935             av_assert0(st->internal->avctx->ticks_per_frame);
936             av_reduce(pnum, pden,
937                       codec_framerate.den,
938                       codec_framerate.num * (int64_t)st->internal->avctx->ticks_per_frame,
939                       INT_MAX);
940
941             if (pc && pc->repeat_pict) {
942                 av_assert0(s->iformat); // this may be wrong for interlaced encoding but its not used for that case
943                 av_reduce(pnum, pden,
944                           (*pnum) * (1LL + pc->repeat_pict),
945                           (*pden),
946                           INT_MAX);
947             }
948             /* If this codec can be interlaced or progressive then we need
949              * a parser to compute duration of a packet. Thus if we have
950              * no parser in such case leave duration undefined. */
951             if (st->internal->avctx->ticks_per_frame > 1 && !pc)
952                 *pnum = *pden = 0;
953         }
954         break;
955     case AVMEDIA_TYPE_AUDIO:
956         if (st->internal->avctx_inited) {
957             frame_size = av_get_audio_frame_duration(st->internal->avctx, pkt->size);
958             sample_rate = st->internal->avctx->sample_rate;
959         } else {
960             frame_size = av_get_audio_frame_duration2(st->codecpar, pkt->size);
961             sample_rate = st->codecpar->sample_rate;
962         }
963         if (frame_size <= 0 || sample_rate <= 0)
964             break;
965         *pnum = frame_size;
966         *pden = sample_rate;
967         break;
968     default:
969         break;
970     }
971 }
972
973 static int is_intra_only(enum AVCodecID id)
974 {
975     const AVCodecDescriptor *d = avcodec_descriptor_get(id);
976     if (!d)
977         return 0;
978     if (d->type == AVMEDIA_TYPE_VIDEO && !(d->props & AV_CODEC_PROP_INTRA_ONLY))
979         return 0;
980     return 1;
981 }
982
983 static int has_decode_delay_been_guessed(AVStream *st)
984 {
985     if (st->codecpar->codec_id != AV_CODEC_ID_H264) return 1;
986     if (!st->info) // if we have left find_stream_info then nb_decoded_frames won't increase anymore for stream copy
987         return 1;
988 #if CONFIG_H264_DECODER
989     if (st->internal->avctx->has_b_frames &&
990        avpriv_h264_has_num_reorder_frames(st->internal->avctx) == st->internal->avctx->has_b_frames)
991         return 1;
992 #endif
993     if (st->internal->avctx->has_b_frames<3)
994         return st->nb_decoded_frames >= 7;
995     else if (st->internal->avctx->has_b_frames<4)
996         return st->nb_decoded_frames >= 18;
997     else
998         return st->nb_decoded_frames >= 20;
999 }
1000
1001 static AVPacketList *get_next_pkt(AVFormatContext *s, AVStream *st, AVPacketList *pktl)
1002 {
1003     if (pktl->next)
1004         return pktl->next;
1005     if (pktl == s->internal->packet_buffer_end)
1006         return s->internal->parse_queue;
1007     return NULL;
1008 }
1009
1010 static int64_t select_from_pts_buffer(AVStream *st, int64_t *pts_buffer, int64_t dts) {
1011     int onein_oneout = st->codecpar->codec_id != AV_CODEC_ID_H264 &&
1012                        st->codecpar->codec_id != AV_CODEC_ID_HEVC;
1013
1014     if(!onein_oneout) {
1015         int delay = st->internal->avctx->has_b_frames;
1016         int i;
1017
1018         if (dts == AV_NOPTS_VALUE) {
1019             int64_t best_score = INT64_MAX;
1020             for (i = 0; i<delay; i++) {
1021                 if (st->pts_reorder_error_count[i]) {
1022                     int64_t score = st->pts_reorder_error[i] / st->pts_reorder_error_count[i];
1023                     if (score < best_score) {
1024                         best_score = score;
1025                         dts = pts_buffer[i];
1026                     }
1027                 }
1028             }
1029         } else {
1030             for (i = 0; i<delay; i++) {
1031                 if (pts_buffer[i] != AV_NOPTS_VALUE) {
1032                     int64_t diff =  FFABS(pts_buffer[i] - dts)
1033                                     + (uint64_t)st->pts_reorder_error[i];
1034                     diff = FFMAX(diff, st->pts_reorder_error[i]);
1035                     st->pts_reorder_error[i] = diff;
1036                     st->pts_reorder_error_count[i]++;
1037                     if (st->pts_reorder_error_count[i] > 250) {
1038                         st->pts_reorder_error[i] >>= 1;
1039                         st->pts_reorder_error_count[i] >>= 1;
1040                     }
1041                 }
1042             }
1043         }
1044     }
1045
1046     if (dts == AV_NOPTS_VALUE)
1047         dts = pts_buffer[0];
1048
1049     return dts;
1050 }
1051
1052 /**
1053  * Updates the dts of packets of a stream in pkt_buffer, by re-ordering the pts
1054  * of the packets in a window.
1055  */
1056 static void update_dts_from_pts(AVFormatContext *s, int stream_index,
1057                                 AVPacketList *pkt_buffer)
1058 {
1059     AVStream *st       = s->streams[stream_index];
1060     int delay          = st->internal->avctx->has_b_frames;
1061     int i;
1062
1063     int64_t pts_buffer[MAX_REORDER_DELAY+1];
1064
1065     for (i = 0; i<MAX_REORDER_DELAY+1; i++)
1066         pts_buffer[i] = AV_NOPTS_VALUE;
1067
1068     for (; pkt_buffer; pkt_buffer = get_next_pkt(s, st, pkt_buffer)) {
1069         if (pkt_buffer->pkt.stream_index != stream_index)
1070             continue;
1071
1072         if (pkt_buffer->pkt.pts != AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY) {
1073             pts_buffer[0] = pkt_buffer->pkt.pts;
1074             for (i = 0; i<delay && pts_buffer[i] > pts_buffer[i + 1]; i++)
1075                 FFSWAP(int64_t, pts_buffer[i], pts_buffer[i + 1]);
1076
1077             pkt_buffer->pkt.dts = select_from_pts_buffer(st, pts_buffer, pkt_buffer->pkt.dts);
1078         }
1079     }
1080 }
1081
1082 static void update_initial_timestamps(AVFormatContext *s, int stream_index,
1083                                       int64_t dts, int64_t pts, AVPacket *pkt)
1084 {
1085     AVStream *st       = s->streams[stream_index];
1086     AVPacketList *pktl = s->internal->packet_buffer ? s->internal->packet_buffer : s->internal->parse_queue;
1087     AVPacketList *pktl_it;
1088
1089     uint64_t shift;
1090
1091     if (st->first_dts != AV_NOPTS_VALUE ||
1092         dts           == AV_NOPTS_VALUE ||
1093         st->cur_dts   == AV_NOPTS_VALUE ||
1094         is_relative(dts))
1095         return;
1096
1097     st->first_dts = dts - (st->cur_dts - RELATIVE_TS_BASE);
1098     st->cur_dts   = dts;
1099     shift         = (uint64_t)st->first_dts - RELATIVE_TS_BASE;
1100
1101     if (is_relative(pts))
1102         pts += shift;
1103
1104     for (pktl_it = pktl; pktl_it; pktl_it = get_next_pkt(s, st, pktl_it)) {
1105         if (pktl_it->pkt.stream_index != stream_index)
1106             continue;
1107         if (is_relative(pktl_it->pkt.pts))
1108             pktl_it->pkt.pts += shift;
1109
1110         if (is_relative(pktl_it->pkt.dts))
1111             pktl_it->pkt.dts += shift;
1112
1113         if (st->start_time == AV_NOPTS_VALUE && pktl_it->pkt.pts != AV_NOPTS_VALUE) {
1114             st->start_time = pktl_it->pkt.pts;
1115             if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO && st->codecpar->sample_rate)
1116                 st->start_time += av_rescale_q(st->skip_samples, (AVRational){1, st->codecpar->sample_rate}, st->time_base);
1117         }
1118     }
1119
1120     if (has_decode_delay_been_guessed(st)) {
1121         update_dts_from_pts(s, stream_index, pktl);
1122     }
1123
1124     if (st->start_time == AV_NOPTS_VALUE) {
1125         st->start_time = pts;
1126         if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO && st->codecpar->sample_rate)
1127             st->start_time += av_rescale_q(st->skip_samples, (AVRational){1, st->codecpar->sample_rate}, st->time_base);
1128     }
1129 }
1130
1131 static void update_initial_durations(AVFormatContext *s, AVStream *st,
1132                                      int stream_index, int duration)
1133 {
1134     AVPacketList *pktl = s->internal->packet_buffer ? s->internal->packet_buffer : s->internal->parse_queue;
1135     int64_t cur_dts    = RELATIVE_TS_BASE;
1136
1137     if (st->first_dts != AV_NOPTS_VALUE) {
1138         if (st->update_initial_durations_done)
1139             return;
1140         st->update_initial_durations_done = 1;
1141         cur_dts = st->first_dts;
1142         for (; pktl; pktl = get_next_pkt(s, st, pktl)) {
1143             if (pktl->pkt.stream_index == stream_index) {
1144                 if (pktl->pkt.pts != pktl->pkt.dts  ||
1145                     pktl->pkt.dts != AV_NOPTS_VALUE ||
1146                     pktl->pkt.duration)
1147                     break;
1148                 cur_dts -= duration;
1149             }
1150         }
1151         if (pktl && pktl->pkt.dts != st->first_dts) {
1152             av_log(s, AV_LOG_DEBUG, "first_dts %s not matching first dts %s (pts %s, duration %"PRId64") in the queue\n",
1153                    av_ts2str(st->first_dts), av_ts2str(pktl->pkt.dts), av_ts2str(pktl->pkt.pts), pktl->pkt.duration);
1154             return;
1155         }
1156         if (!pktl) {
1157             av_log(s, AV_LOG_DEBUG, "first_dts %s but no packet with dts in the queue\n", av_ts2str(st->first_dts));
1158             return;
1159         }
1160         pktl          = s->internal->packet_buffer ? s->internal->packet_buffer : s->internal->parse_queue;
1161         st->first_dts = cur_dts;
1162     } else if (st->cur_dts != RELATIVE_TS_BASE)
1163         return;
1164
1165     for (; pktl; pktl = get_next_pkt(s, st, pktl)) {
1166         if (pktl->pkt.stream_index != stream_index)
1167             continue;
1168         if ((pktl->pkt.pts == pktl->pkt.dts ||
1169              pktl->pkt.pts == AV_NOPTS_VALUE) &&
1170             (pktl->pkt.dts == AV_NOPTS_VALUE ||
1171              pktl->pkt.dts == st->first_dts ||
1172              pktl->pkt.dts == RELATIVE_TS_BASE) &&
1173             !pktl->pkt.duration) {
1174             pktl->pkt.dts = cur_dts;
1175             if (!st->internal->avctx->has_b_frames)
1176                 pktl->pkt.pts = cur_dts;
1177 //            if (st->codecpar->codec_type != AVMEDIA_TYPE_AUDIO)
1178                 pktl->pkt.duration = duration;
1179         } else
1180             break;
1181         cur_dts = pktl->pkt.dts + pktl->pkt.duration;
1182     }
1183     if (!pktl)
1184         st->cur_dts = cur_dts;
1185 }
1186
1187 static void compute_pkt_fields(AVFormatContext *s, AVStream *st,
1188                                AVCodecParserContext *pc, AVPacket *pkt,
1189                                int64_t next_dts, int64_t next_pts)
1190 {
1191     int num, den, presentation_delayed, delay, i;
1192     int64_t offset;
1193     AVRational duration;
1194     int onein_oneout = st->codecpar->codec_id != AV_CODEC_ID_H264 &&
1195                        st->codecpar->codec_id != AV_CODEC_ID_HEVC;
1196
1197     if (s->flags & AVFMT_FLAG_NOFILLIN)
1198         return;
1199
1200     if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && pkt->dts != AV_NOPTS_VALUE) {
1201         if (pkt->dts == pkt->pts && st->last_dts_for_order_check != AV_NOPTS_VALUE) {
1202             if (st->last_dts_for_order_check <= pkt->dts) {
1203                 st->dts_ordered++;
1204             } else {
1205                 av_log(s, st->dts_misordered ? AV_LOG_DEBUG : AV_LOG_WARNING,
1206                        "DTS %"PRIi64" < %"PRIi64" out of order\n",
1207                        pkt->dts,
1208                        st->last_dts_for_order_check);
1209                 st->dts_misordered++;
1210             }
1211             if (st->dts_ordered + st->dts_misordered > 250) {
1212                 st->dts_ordered    >>= 1;
1213                 st->dts_misordered >>= 1;
1214             }
1215         }
1216
1217         st->last_dts_for_order_check = pkt->dts;
1218         if (st->dts_ordered < 8*st->dts_misordered && pkt->dts == pkt->pts)
1219             pkt->dts = AV_NOPTS_VALUE;
1220     }
1221
1222     if ((s->flags & AVFMT_FLAG_IGNDTS) && pkt->pts != AV_NOPTS_VALUE)
1223         pkt->dts = AV_NOPTS_VALUE;
1224
1225     if (pc && pc->pict_type == AV_PICTURE_TYPE_B
1226         && !st->internal->avctx->has_b_frames)
1227         //FIXME Set low_delay = 0 when has_b_frames = 1
1228         st->internal->avctx->has_b_frames = 1;
1229
1230     /* do we have a video B-frame ? */
1231     delay = st->internal->avctx->has_b_frames;
1232     presentation_delayed = 0;
1233
1234     /* XXX: need has_b_frame, but cannot get it if the codec is
1235      *  not initialized */
1236     if (delay &&
1237         pc && pc->pict_type != AV_PICTURE_TYPE_B)
1238         presentation_delayed = 1;
1239
1240     if (pkt->pts != AV_NOPTS_VALUE && pkt->dts != AV_NOPTS_VALUE &&
1241         st->pts_wrap_bits < 63 &&
1242         pkt->dts - (1LL << (st->pts_wrap_bits - 1)) > pkt->pts) {
1243         if (is_relative(st->cur_dts) || pkt->dts - (1LL<<(st->pts_wrap_bits - 1)) > st->cur_dts) {
1244             pkt->dts -= 1LL << st->pts_wrap_bits;
1245         } else
1246             pkt->pts += 1LL << st->pts_wrap_bits;
1247     }
1248
1249     /* Some MPEG-2 in MPEG-PS lack dts (issue #171 / input_file.mpg).
1250      * We take the conservative approach and discard both.
1251      * Note: If this is misbehaving for an H.264 file, then possibly
1252      * presentation_delayed is not set correctly. */
1253     if (delay == 1 && pkt->dts == pkt->pts &&
1254         pkt->dts != AV_NOPTS_VALUE && presentation_delayed) {
1255         av_log(s, AV_LOG_DEBUG, "invalid dts/pts combination %"PRIi64"\n", pkt->dts);
1256         if (    strcmp(s->iformat->name, "mov,mp4,m4a,3gp,3g2,mj2")
1257              && strcmp(s->iformat->name, "flv")) // otherwise we discard correct timestamps for vc1-wmapro.ism
1258             pkt->dts = AV_NOPTS_VALUE;
1259     }
1260
1261     duration = av_mul_q((AVRational) {pkt->duration, 1}, st->time_base);
1262     if (pkt->duration == 0) {
1263         ff_compute_frame_duration(s, &num, &den, st, pc, pkt);
1264         if (den && num) {
1265             duration = (AVRational) {num, den};
1266             pkt->duration = av_rescale_rnd(1,
1267                                            num * (int64_t) st->time_base.den,
1268                                            den * (int64_t) st->time_base.num,
1269                                            AV_ROUND_DOWN);
1270         }
1271     }
1272
1273     if (pkt->duration != 0 && (s->internal->packet_buffer || s->internal->parse_queue))
1274         update_initial_durations(s, st, pkt->stream_index, pkt->duration);
1275
1276     /* Correct timestamps with byte offset if demuxers only have timestamps
1277      * on packet boundaries */
1278     if (pc && st->need_parsing == AVSTREAM_PARSE_TIMESTAMPS && pkt->size) {
1279         /* this will estimate bitrate based on this frame's duration and size */
1280         offset = av_rescale(pc->offset, pkt->duration, pkt->size);
1281         if (pkt->pts != AV_NOPTS_VALUE)
1282             pkt->pts += offset;
1283         if (pkt->dts != AV_NOPTS_VALUE)
1284             pkt->dts += offset;
1285     }
1286
1287     /* This may be redundant, but it should not hurt. */
1288     if (pkt->dts != AV_NOPTS_VALUE &&
1289         pkt->pts != AV_NOPTS_VALUE &&
1290         pkt->pts > pkt->dts)
1291         presentation_delayed = 1;
1292
1293     if (s->debug & FF_FDEBUG_TS)
1294         av_log(s, AV_LOG_TRACE,
1295             "IN delayed:%d pts:%s, dts:%s cur_dts:%s st:%d pc:%p duration:%"PRId64" delay:%d onein_oneout:%d\n",
1296             presentation_delayed, av_ts2str(pkt->pts), av_ts2str(pkt->dts), av_ts2str(st->cur_dts),
1297             pkt->stream_index, pc, pkt->duration, delay, onein_oneout);
1298
1299     /* Interpolate PTS and DTS if they are not present. We skip H264
1300      * currently because delay and has_b_frames are not reliably set. */
1301     if ((delay == 0 || (delay == 1 && pc)) &&
1302         onein_oneout) {
1303         if (presentation_delayed) {
1304             /* DTS = decompression timestamp */
1305             /* PTS = presentation timestamp */
1306             if (pkt->dts == AV_NOPTS_VALUE)
1307                 pkt->dts = st->last_IP_pts;
1308             update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts, pkt);
1309             if (pkt->dts == AV_NOPTS_VALUE)
1310                 pkt->dts = st->cur_dts;
1311
1312             /* This is tricky: the dts must be incremented by the duration
1313              * of the frame we are displaying, i.e. the last I- or P-frame. */
1314             if (st->last_IP_duration == 0)
1315                 st->last_IP_duration = pkt->duration;
1316             if (pkt->dts != AV_NOPTS_VALUE)
1317                 st->cur_dts = pkt->dts + st->last_IP_duration;
1318             if (pkt->dts != AV_NOPTS_VALUE &&
1319                 pkt->pts == AV_NOPTS_VALUE &&
1320                 st->last_IP_duration > 0 &&
1321                 ((uint64_t)st->cur_dts - (uint64_t)next_dts + 1) <= 2 &&
1322                 next_dts != next_pts &&
1323                 next_pts != AV_NOPTS_VALUE)
1324                 pkt->pts = next_dts;
1325
1326             st->last_IP_duration = pkt->duration;
1327             st->last_IP_pts      = pkt->pts;
1328             /* Cannot compute PTS if not present (we can compute it only
1329              * by knowing the future. */
1330         } else if (pkt->pts != AV_NOPTS_VALUE ||
1331                    pkt->dts != AV_NOPTS_VALUE ||
1332                    pkt->duration                ) {
1333
1334             /* presentation is not delayed : PTS and DTS are the same */
1335             if (pkt->pts == AV_NOPTS_VALUE)
1336                 pkt->pts = pkt->dts;
1337             update_initial_timestamps(s, pkt->stream_index, pkt->pts,
1338                                       pkt->pts, pkt);
1339             if (pkt->pts == AV_NOPTS_VALUE)
1340                 pkt->pts = st->cur_dts;
1341             pkt->dts = pkt->pts;
1342             if (pkt->pts != AV_NOPTS_VALUE)
1343                 st->cur_dts = av_add_stable(st->time_base, pkt->pts, duration, 1);
1344         }
1345     }
1346
1347     if (pkt->pts != AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY) {
1348         st->pts_buffer[0] = pkt->pts;
1349         for (i = 0; i<delay && st->pts_buffer[i] > st->pts_buffer[i + 1]; i++)
1350             FFSWAP(int64_t, st->pts_buffer[i], st->pts_buffer[i + 1]);
1351
1352         if(has_decode_delay_been_guessed(st))
1353             pkt->dts = select_from_pts_buffer(st, st->pts_buffer, pkt->dts);
1354     }
1355     // We skipped it above so we try here.
1356     if (!onein_oneout)
1357         // This should happen on the first packet
1358         update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts, pkt);
1359     if (pkt->dts > st->cur_dts)
1360         st->cur_dts = pkt->dts;
1361
1362     if (s->debug & FF_FDEBUG_TS)
1363         av_log(s, AV_LOG_TRACE, "OUTdelayed:%d/%d pts:%s, dts:%s cur_dts:%s\n",
1364             presentation_delayed, delay, av_ts2str(pkt->pts), av_ts2str(pkt->dts), av_ts2str(st->cur_dts));
1365
1366     /* update flags */
1367     if (is_intra_only(st->codecpar->codec_id))
1368         pkt->flags |= AV_PKT_FLAG_KEY;
1369 #if FF_API_CONVERGENCE_DURATION
1370 FF_DISABLE_DEPRECATION_WARNINGS
1371     if (pc)
1372         pkt->convergence_duration = pc->convergence_duration;
1373 FF_ENABLE_DEPRECATION_WARNINGS
1374 #endif
1375 }
1376
1377 static void free_packet_buffer(AVPacketList **pkt_buf, AVPacketList **pkt_buf_end)
1378 {
1379     while (*pkt_buf) {
1380         AVPacketList *pktl = *pkt_buf;
1381         *pkt_buf = pktl->next;
1382         av_packet_unref(&pktl->pkt);
1383         av_freep(&pktl);
1384     }
1385     *pkt_buf_end = NULL;
1386 }
1387
1388 /**
1389  * Parse a packet, add all split parts to parse_queue.
1390  *
1391  * @param pkt Packet to parse, NULL when flushing the parser at end of stream.
1392  */
1393 static int parse_packet(AVFormatContext *s, AVPacket *pkt, int stream_index)
1394 {
1395     AVPacket out_pkt = { 0 }, flush_pkt = { 0 };
1396     AVStream *st = s->streams[stream_index];
1397     uint8_t *data = pkt ? pkt->data : NULL;
1398     int size      = pkt ? pkt->size : 0;
1399     int ret = 0, got_output = 0;
1400
1401     if (!pkt) {
1402         av_init_packet(&flush_pkt);
1403         pkt        = &flush_pkt;
1404         got_output = 1;
1405     } else if (!size && st->parser->flags & PARSER_FLAG_COMPLETE_FRAMES) {
1406         // preserve 0-size sync packets
1407         compute_pkt_fields(s, st, st->parser, pkt, AV_NOPTS_VALUE, AV_NOPTS_VALUE);
1408     }
1409
1410     while (size > 0 || (pkt == &flush_pkt && got_output)) {
1411         int len;
1412         int64_t next_pts = pkt->pts;
1413         int64_t next_dts = pkt->dts;
1414
1415         av_init_packet(&out_pkt);
1416         len = av_parser_parse2(st->parser, st->internal->avctx,
1417                                &out_pkt.data, &out_pkt.size, data, size,
1418                                pkt->pts, pkt->dts, pkt->pos);
1419
1420         pkt->pts = pkt->dts = AV_NOPTS_VALUE;
1421         pkt->pos = -1;
1422         /* increment read pointer */
1423         data += len;
1424         size -= len;
1425
1426         got_output = !!out_pkt.size;
1427
1428         if (!out_pkt.size)
1429             continue;
1430
1431         if (pkt->side_data) {
1432             out_pkt.side_data       = pkt->side_data;
1433             out_pkt.side_data_elems = pkt->side_data_elems;
1434             pkt->side_data          = NULL;
1435             pkt->side_data_elems    = 0;
1436         }
1437
1438         /* set the duration */
1439         out_pkt.duration = (st->parser->flags & PARSER_FLAG_COMPLETE_FRAMES) ? pkt->duration : 0;
1440         if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
1441             if (st->internal->avctx->sample_rate > 0) {
1442                 out_pkt.duration =
1443                     av_rescale_q_rnd(st->parser->duration,
1444                                      (AVRational) { 1, st->internal->avctx->sample_rate },
1445                                      st->time_base,
1446                                      AV_ROUND_DOWN);
1447             }
1448         }
1449
1450         out_pkt.stream_index = st->index;
1451         out_pkt.pts          = st->parser->pts;
1452         out_pkt.dts          = st->parser->dts;
1453         out_pkt.pos          = st->parser->pos;
1454
1455         if (st->need_parsing == AVSTREAM_PARSE_FULL_RAW)
1456             out_pkt.pos = st->parser->frame_offset;
1457
1458         if (st->parser->key_frame == 1 ||
1459             (st->parser->key_frame == -1 &&
1460              st->parser->pict_type == AV_PICTURE_TYPE_I))
1461             out_pkt.flags |= AV_PKT_FLAG_KEY;
1462
1463         if (st->parser->key_frame == -1 && st->parser->pict_type ==AV_PICTURE_TYPE_NONE && (pkt->flags&AV_PKT_FLAG_KEY))
1464             out_pkt.flags |= AV_PKT_FLAG_KEY;
1465
1466         compute_pkt_fields(s, st, st->parser, &out_pkt, next_dts, next_pts);
1467
1468         ret = add_to_pktbuf(&s->internal->parse_queue, &out_pkt,
1469                             &s->internal->parse_queue_end, 1);
1470         av_packet_unref(&out_pkt);
1471         if (ret < 0)
1472             goto fail;
1473     }
1474
1475     /* end of the stream => close and free the parser */
1476     if (pkt == &flush_pkt) {
1477         av_parser_close(st->parser);
1478         st->parser = NULL;
1479     }
1480
1481 fail:
1482     av_packet_unref(pkt);
1483     return ret;
1484 }
1485
1486 static int read_from_packet_buffer(AVPacketList **pkt_buffer,
1487                                    AVPacketList **pkt_buffer_end,
1488                                    AVPacket      *pkt)
1489 {
1490     AVPacketList *pktl;
1491     av_assert0(*pkt_buffer);
1492     pktl        = *pkt_buffer;
1493     *pkt        = pktl->pkt;
1494     *pkt_buffer = pktl->next;
1495     if (!pktl->next)
1496         *pkt_buffer_end = NULL;
1497     av_freep(&pktl);
1498     return 0;
1499 }
1500
1501 static int64_t ts_to_samples(AVStream *st, int64_t ts)
1502 {
1503     return av_rescale(ts, st->time_base.num * st->codecpar->sample_rate, st->time_base.den);
1504 }
1505
1506 static int read_frame_internal(AVFormatContext *s, AVPacket *pkt)
1507 {
1508     int ret = 0, i, got_packet = 0;
1509     AVDictionary *metadata = NULL;
1510
1511     av_init_packet(pkt);
1512
1513     while (!got_packet && !s->internal->parse_queue) {
1514         AVStream *st;
1515         AVPacket cur_pkt;
1516
1517         /* read next packet */
1518         ret = ff_read_packet(s, &cur_pkt);
1519         if (ret < 0) {
1520             if (ret == AVERROR(EAGAIN))
1521                 return ret;
1522             /* flush the parsers */
1523             for (i = 0; i < s->nb_streams; i++) {
1524                 st = s->streams[i];
1525                 if (st->parser && st->need_parsing)
1526                     parse_packet(s, NULL, st->index);
1527             }
1528             /* all remaining packets are now in parse_queue =>
1529              * really terminate parsing */
1530             break;
1531         }
1532         ret = 0;
1533         st  = s->streams[cur_pkt.stream_index];
1534
1535         /* update context if required */
1536         if (st->internal->need_context_update) {
1537             if (avcodec_is_open(st->internal->avctx)) {
1538                 av_log(s, AV_LOG_DEBUG, "Demuxer context update while decoder is open, closing and trying to re-open\n");
1539                 avcodec_close(st->internal->avctx);
1540                 st->info->found_decoder = 0;
1541             }
1542
1543             /* close parser, because it depends on the codec */
1544             if (st->parser && st->internal->avctx->codec_id != st->codecpar->codec_id) {
1545                 av_parser_close(st->parser);
1546                 st->parser = NULL;
1547             }
1548
1549             ret = avcodec_parameters_to_context(st->internal->avctx, st->codecpar);
1550             if (ret < 0)
1551                 return ret;
1552
1553 #if FF_API_LAVF_AVCTX
1554 FF_DISABLE_DEPRECATION_WARNINGS
1555             /* update deprecated public codec context */
1556             ret = avcodec_parameters_to_context(st->codec, st->codecpar);
1557             if (ret < 0)
1558                 return ret;
1559 FF_ENABLE_DEPRECATION_WARNINGS
1560 #endif
1561
1562             st->internal->need_context_update = 0;
1563         }
1564
1565         if (cur_pkt.pts != AV_NOPTS_VALUE &&
1566             cur_pkt.dts != AV_NOPTS_VALUE &&
1567             cur_pkt.pts < cur_pkt.dts) {
1568             av_log(s, AV_LOG_WARNING,
1569                    "Invalid timestamps stream=%d, pts=%s, dts=%s, size=%d\n",
1570                    cur_pkt.stream_index,
1571                    av_ts2str(cur_pkt.pts),
1572                    av_ts2str(cur_pkt.dts),
1573                    cur_pkt.size);
1574         }
1575         if (s->debug & FF_FDEBUG_TS)
1576             av_log(s, AV_LOG_DEBUG,
1577                    "ff_read_packet stream=%d, pts=%s, dts=%s, size=%d, duration=%"PRId64", flags=%d\n",
1578                    cur_pkt.stream_index,
1579                    av_ts2str(cur_pkt.pts),
1580                    av_ts2str(cur_pkt.dts),
1581                    cur_pkt.size, cur_pkt.duration, cur_pkt.flags);
1582
1583         if (st->need_parsing && !st->parser && !(s->flags & AVFMT_FLAG_NOPARSE)) {
1584             st->parser = av_parser_init(st->codecpar->codec_id);
1585             if (!st->parser) {
1586                 av_log(s, AV_LOG_VERBOSE, "parser not found for codec "
1587                        "%s, packets or times may be invalid.\n",
1588                        avcodec_get_name(st->codecpar->codec_id));
1589                 /* no parser available: just output the raw packets */
1590                 st->need_parsing = AVSTREAM_PARSE_NONE;
1591             } else if (st->need_parsing == AVSTREAM_PARSE_HEADERS)
1592                 st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
1593             else if (st->need_parsing == AVSTREAM_PARSE_FULL_ONCE)
1594                 st->parser->flags |= PARSER_FLAG_ONCE;
1595             else if (st->need_parsing == AVSTREAM_PARSE_FULL_RAW)
1596                 st->parser->flags |= PARSER_FLAG_USE_CODEC_TS;
1597         }
1598
1599         if (!st->need_parsing || !st->parser) {
1600             /* no parsing needed: we just output the packet as is */
1601             *pkt = cur_pkt;
1602             compute_pkt_fields(s, st, NULL, pkt, AV_NOPTS_VALUE, AV_NOPTS_VALUE);
1603             if ((s->iformat->flags & AVFMT_GENERIC_INDEX) &&
1604                 (pkt->flags & AV_PKT_FLAG_KEY) && pkt->dts != AV_NOPTS_VALUE) {
1605                 ff_reduce_index(s, st->index);
1606                 av_add_index_entry(st, pkt->pos, pkt->dts,
1607                                    0, 0, AVINDEX_KEYFRAME);
1608             }
1609             got_packet = 1;
1610         } else if (st->discard < AVDISCARD_ALL) {
1611             if ((ret = parse_packet(s, &cur_pkt, cur_pkt.stream_index)) < 0)
1612                 return ret;
1613             st->codecpar->sample_rate = st->internal->avctx->sample_rate;
1614             st->codecpar->bit_rate = st->internal->avctx->bit_rate;
1615             st->codecpar->channels = st->internal->avctx->channels;
1616             st->codecpar->channel_layout = st->internal->avctx->channel_layout;
1617             st->codecpar->codec_id = st->internal->avctx->codec_id;
1618         } else {
1619             /* free packet */
1620             av_packet_unref(&cur_pkt);
1621         }
1622         if (pkt->flags & AV_PKT_FLAG_KEY)
1623             st->skip_to_keyframe = 0;
1624         if (st->skip_to_keyframe) {
1625             av_packet_unref(&cur_pkt);
1626             if (got_packet) {
1627                 *pkt = cur_pkt;
1628             }
1629             got_packet = 0;
1630         }
1631     }
1632
1633     if (!got_packet && s->internal->parse_queue)
1634         ret = read_from_packet_buffer(&s->internal->parse_queue, &s->internal->parse_queue_end, pkt);
1635
1636     if (ret >= 0) {
1637         AVStream *st = s->streams[pkt->stream_index];
1638         int discard_padding = 0;
1639         if (st->first_discard_sample && pkt->pts != AV_NOPTS_VALUE) {
1640             int64_t pts = pkt->pts - (is_relative(pkt->pts) ? RELATIVE_TS_BASE : 0);
1641             int64_t sample = ts_to_samples(st, pts);
1642             int duration = ts_to_samples(st, pkt->duration);
1643             int64_t end_sample = sample + duration;
1644             if (duration > 0 && end_sample >= st->first_discard_sample &&
1645                 sample < st->last_discard_sample)
1646                 discard_padding = FFMIN(end_sample - st->first_discard_sample, duration);
1647         }
1648         if (st->start_skip_samples && (pkt->pts == 0 || pkt->pts == RELATIVE_TS_BASE))
1649             st->skip_samples = st->start_skip_samples;
1650         if (st->skip_samples || discard_padding) {
1651             uint8_t *p = av_packet_new_side_data(pkt, AV_PKT_DATA_SKIP_SAMPLES, 10);
1652             if (p) {
1653                 AV_WL32(p, st->skip_samples);
1654                 AV_WL32(p + 4, discard_padding);
1655                 av_log(s, AV_LOG_DEBUG, "demuxer injecting skip %d / discard %d\n", st->skip_samples, discard_padding);
1656             }
1657             st->skip_samples = 0;
1658         }
1659
1660         if (st->inject_global_side_data) {
1661             for (i = 0; i < st->nb_side_data; i++) {
1662                 AVPacketSideData *src_sd = &st->side_data[i];
1663                 uint8_t *dst_data;
1664
1665                 if (av_packet_get_side_data(pkt, src_sd->type, NULL))
1666                     continue;
1667
1668                 dst_data = av_packet_new_side_data(pkt, src_sd->type, src_sd->size);
1669                 if (!dst_data) {
1670                     av_log(s, AV_LOG_WARNING, "Could not inject global side data\n");
1671                     continue;
1672                 }
1673
1674                 memcpy(dst_data, src_sd->data, src_sd->size);
1675             }
1676             st->inject_global_side_data = 0;
1677         }
1678
1679 #if FF_API_LAVF_MERGE_SD
1680 FF_DISABLE_DEPRECATION_WARNINGS
1681         if (!(s->flags & AVFMT_FLAG_KEEP_SIDE_DATA))
1682             av_packet_merge_side_data(pkt);
1683 FF_ENABLE_DEPRECATION_WARNINGS
1684 #endif
1685     }
1686
1687     av_opt_get_dict_val(s, "metadata", AV_OPT_SEARCH_CHILDREN, &metadata);
1688     if (metadata) {
1689         s->event_flags |= AVFMT_EVENT_FLAG_METADATA_UPDATED;
1690         av_dict_copy(&s->metadata, metadata, 0);
1691         av_dict_free(&metadata);
1692         av_opt_set_dict_val(s, "metadata", NULL, AV_OPT_SEARCH_CHILDREN);
1693     }
1694
1695 #if FF_API_LAVF_AVCTX
1696     update_stream_avctx(s);
1697 #endif
1698
1699     if (s->debug & FF_FDEBUG_TS)
1700         av_log(s, AV_LOG_DEBUG,
1701                "read_frame_internal stream=%d, pts=%s, dts=%s, "
1702                "size=%d, duration=%"PRId64", flags=%d\n",
1703                pkt->stream_index,
1704                av_ts2str(pkt->pts),
1705                av_ts2str(pkt->dts),
1706                pkt->size, pkt->duration, pkt->flags);
1707
1708     return ret;
1709 }
1710
1711 int av_read_frame(AVFormatContext *s, AVPacket *pkt)
1712 {
1713     const int genpts = s->flags & AVFMT_FLAG_GENPTS;
1714     int eof = 0;
1715     int ret;
1716     AVStream *st;
1717
1718     if (!genpts) {
1719         ret = s->internal->packet_buffer
1720               ? read_from_packet_buffer(&s->internal->packet_buffer,
1721                                         &s->internal->packet_buffer_end, pkt)
1722               : read_frame_internal(s, pkt);
1723         if (ret < 0)
1724             return ret;
1725         goto return_packet;
1726     }
1727
1728     for (;;) {
1729         AVPacketList *pktl = s->internal->packet_buffer;
1730
1731         if (pktl) {
1732             AVPacket *next_pkt = &pktl->pkt;
1733
1734             if (next_pkt->dts != AV_NOPTS_VALUE) {
1735                 int wrap_bits = s->streams[next_pkt->stream_index]->pts_wrap_bits;
1736                 // last dts seen for this stream. if any of packets following
1737                 // current one had no dts, we will set this to AV_NOPTS_VALUE.
1738                 int64_t last_dts = next_pkt->dts;
1739                 while (pktl && next_pkt->pts == AV_NOPTS_VALUE) {
1740                     if (pktl->pkt.stream_index == next_pkt->stream_index &&
1741                         (av_compare_mod(next_pkt->dts, pktl->pkt.dts, 2LL << (wrap_bits - 1)) < 0)) {
1742                         if (av_compare_mod(pktl->pkt.pts, pktl->pkt.dts, 2LL << (wrap_bits - 1))) {
1743                             // not B-frame
1744                             next_pkt->pts = pktl->pkt.dts;
1745                         }
1746                         if (last_dts != AV_NOPTS_VALUE) {
1747                             // Once last dts was set to AV_NOPTS_VALUE, we don't change it.
1748                             last_dts = pktl->pkt.dts;
1749                         }
1750                     }
1751                     pktl = pktl->next;
1752                 }
1753                 if (eof && next_pkt->pts == AV_NOPTS_VALUE && last_dts != AV_NOPTS_VALUE) {
1754                     // Fixing the last reference frame had none pts issue (For MXF etc).
1755                     // We only do this when
1756                     // 1. eof.
1757                     // 2. we are not able to resolve a pts value for current packet.
1758                     // 3. the packets for this stream at the end of the files had valid dts.
1759                     next_pkt->pts = last_dts + next_pkt->duration;
1760                 }
1761                 pktl = s->internal->packet_buffer;
1762             }
1763
1764             /* read packet from packet buffer, if there is data */
1765             st = s->streams[next_pkt->stream_index];
1766             if (!(next_pkt->pts == AV_NOPTS_VALUE && st->discard < AVDISCARD_ALL &&
1767                   next_pkt->dts != AV_NOPTS_VALUE && !eof)) {
1768                 ret = read_from_packet_buffer(&s->internal->packet_buffer,
1769                                                &s->internal->packet_buffer_end, pkt);
1770                 goto return_packet;
1771             }
1772         }
1773
1774         ret = read_frame_internal(s, pkt);
1775         if (ret < 0) {
1776             if (pktl && ret != AVERROR(EAGAIN)) {
1777                 eof = 1;
1778                 continue;
1779             } else
1780                 return ret;
1781         }
1782
1783         ret = add_to_pktbuf(&s->internal->packet_buffer, pkt,
1784                             &s->internal->packet_buffer_end, 1);
1785         av_packet_unref(pkt);
1786         if (ret < 0)
1787             return ret;
1788     }
1789
1790 return_packet:
1791
1792     st = s->streams[pkt->stream_index];
1793     if ((s->iformat->flags & AVFMT_GENERIC_INDEX) && pkt->flags & AV_PKT_FLAG_KEY) {
1794         ff_reduce_index(s, st->index);
1795         av_add_index_entry(st, pkt->pos, pkt->dts, 0, 0, AVINDEX_KEYFRAME);
1796     }
1797
1798     if (is_relative(pkt->dts))
1799         pkt->dts -= RELATIVE_TS_BASE;
1800     if (is_relative(pkt->pts))
1801         pkt->pts -= RELATIVE_TS_BASE;
1802
1803     return ret;
1804 }
1805
1806 /* XXX: suppress the packet queue */
1807 static void flush_packet_queue(AVFormatContext *s)
1808 {
1809     if (!s->internal)
1810         return;
1811     free_packet_buffer(&s->internal->parse_queue,       &s->internal->parse_queue_end);
1812     free_packet_buffer(&s->internal->packet_buffer,     &s->internal->packet_buffer_end);
1813     free_packet_buffer(&s->internal->raw_packet_buffer, &s->internal->raw_packet_buffer_end);
1814
1815     s->internal->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
1816 }
1817
1818 /*******************************************************/
1819 /* seek support */
1820
1821 int av_find_default_stream_index(AVFormatContext *s)
1822 {
1823     int i;
1824     AVStream *st;
1825     int best_stream = 0;
1826     int best_score = INT_MIN;
1827
1828     if (s->nb_streams <= 0)
1829         return -1;
1830     for (i = 0; i < s->nb_streams; i++) {
1831         int score = 0;
1832         st = s->streams[i];
1833         if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO) {
1834             if (st->disposition & AV_DISPOSITION_ATTACHED_PIC)
1835                 score -= 400;
1836             if (st->codecpar->width && st->codecpar->height)
1837                 score += 50;
1838             score+= 25;
1839         }
1840         if (st->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
1841             if (st->codecpar->sample_rate)
1842                 score += 50;
1843         }
1844         if (st->codec_info_nb_frames)
1845             score += 12;
1846
1847         if (st->discard != AVDISCARD_ALL)
1848             score += 200;
1849
1850         if (score > best_score) {
1851             best_score = score;
1852             best_stream = i;
1853         }
1854     }
1855     return best_stream;
1856 }
1857
1858 /** Flush the frame reader. */
1859 void ff_read_frame_flush(AVFormatContext *s)
1860 {
1861     AVStream *st;
1862     int i, j;
1863
1864     flush_packet_queue(s);
1865
1866     /* Reset read state for each stream. */
1867     for (i = 0; i < s->nb_streams; i++) {
1868         st = s->streams[i];
1869
1870         if (st->parser) {
1871             av_parser_close(st->parser);
1872             st->parser = NULL;
1873         }
1874         st->last_IP_pts = AV_NOPTS_VALUE;
1875         st->last_dts_for_order_check = AV_NOPTS_VALUE;
1876         if (st->first_dts == AV_NOPTS_VALUE)
1877             st->cur_dts = RELATIVE_TS_BASE;
1878         else
1879             /* We set the current DTS to an unspecified origin. */
1880             st->cur_dts = AV_NOPTS_VALUE;
1881
1882         st->probe_packets = MAX_PROBE_PACKETS;
1883
1884         for (j = 0; j < MAX_REORDER_DELAY + 1; j++)
1885             st->pts_buffer[j] = AV_NOPTS_VALUE;
1886
1887         if (s->internal->inject_global_side_data)
1888             st->inject_global_side_data = 1;
1889
1890         st->skip_samples = 0;
1891     }
1892 }
1893
1894 void ff_update_cur_dts(AVFormatContext *s, AVStream *ref_st, int64_t timestamp)
1895 {
1896     int i;
1897
1898     for (i = 0; i < s->nb_streams; i++) {
1899         AVStream *st = s->streams[i];
1900
1901         st->cur_dts =
1902             av_rescale(timestamp,
1903                        st->time_base.den * (int64_t) ref_st->time_base.num,
1904                        st->time_base.num * (int64_t) ref_st->time_base.den);
1905     }
1906 }
1907
1908 void ff_reduce_index(AVFormatContext *s, int stream_index)
1909 {
1910     AVStream *st             = s->streams[stream_index];
1911     unsigned int max_entries = s->max_index_size / sizeof(AVIndexEntry);
1912
1913     if ((unsigned) st->nb_index_entries >= max_entries) {
1914         int i;
1915         for (i = 0; 2 * i < st->nb_index_entries; i++)
1916             st->index_entries[i] = st->index_entries[2 * i];
1917         st->nb_index_entries = i;
1918     }
1919 }
1920
1921 int ff_add_index_entry(AVIndexEntry **index_entries,
1922                        int *nb_index_entries,
1923                        unsigned int *index_entries_allocated_size,
1924                        int64_t pos, int64_t timestamp,
1925                        int size, int distance, int flags)
1926 {
1927     AVIndexEntry *entries, *ie;
1928     int index;
1929
1930     if ((unsigned) *nb_index_entries + 1 >= UINT_MAX / sizeof(AVIndexEntry))
1931         return -1;
1932
1933     if (timestamp == AV_NOPTS_VALUE)
1934         return AVERROR(EINVAL);
1935
1936     if (size < 0 || size > 0x3FFFFFFF)
1937         return AVERROR(EINVAL);
1938
1939     if (is_relative(timestamp)) //FIXME this maintains previous behavior but we should shift by the correct offset once known
1940         timestamp -= RELATIVE_TS_BASE;
1941
1942     entries = av_fast_realloc(*index_entries,
1943                               index_entries_allocated_size,
1944                               (*nb_index_entries + 1) *
1945                               sizeof(AVIndexEntry));
1946     if (!entries)
1947         return -1;
1948
1949     *index_entries = entries;
1950
1951     index = ff_index_search_timestamp(*index_entries, *nb_index_entries,
1952                                       timestamp, AVSEEK_FLAG_ANY);
1953
1954     if (index < 0) {
1955         index = (*nb_index_entries)++;
1956         ie    = &entries[index];
1957         av_assert0(index == 0 || ie[-1].timestamp < timestamp);
1958     } else {
1959         ie = &entries[index];
1960         if (ie->timestamp != timestamp) {
1961             if (ie->timestamp <= timestamp)
1962                 return -1;
1963             memmove(entries + index + 1, entries + index,
1964                     sizeof(AVIndexEntry) * (*nb_index_entries - index));
1965             (*nb_index_entries)++;
1966         } else if (ie->pos == pos && distance < ie->min_distance)
1967             // do not reduce the distance
1968             distance = ie->min_distance;
1969     }
1970
1971     ie->pos          = pos;
1972     ie->timestamp    = timestamp;
1973     ie->min_distance = distance;
1974     ie->size         = size;
1975     ie->flags        = flags;
1976
1977     return index;
1978 }
1979
1980 int av_add_index_entry(AVStream *st, int64_t pos, int64_t timestamp,
1981                        int size, int distance, int flags)
1982 {
1983     timestamp = wrap_timestamp(st, timestamp);
1984     return ff_add_index_entry(&st->index_entries, &st->nb_index_entries,
1985                               &st->index_entries_allocated_size, pos,
1986                               timestamp, size, distance, flags);
1987 }
1988
1989 int ff_index_search_timestamp(const AVIndexEntry *entries, int nb_entries,
1990                               int64_t wanted_timestamp, int flags)
1991 {
1992     int a, b, m;
1993     int64_t timestamp;
1994
1995     a = -1;
1996     b = nb_entries;
1997
1998     // Optimize appending index entries at the end.
1999     if (b && entries[b - 1].timestamp < wanted_timestamp)
2000         a = b - 1;
2001
2002     while (b - a > 1) {
2003         m         = (a + b) >> 1;
2004
2005         // Search for the next non-discarded packet.
2006         while ((entries[m].flags & AVINDEX_DISCARD_FRAME) && m < b && m < nb_entries - 1) {
2007             m++;
2008             if (m == b && entries[m].timestamp >= wanted_timestamp) {
2009                 m = b - 1;
2010                 break;
2011             }
2012         }
2013
2014         timestamp = entries[m].timestamp;
2015         if (timestamp >= wanted_timestamp)
2016             b = m;
2017         if (timestamp <= wanted_timestamp)
2018             a = m;
2019     }
2020     m = (flags & AVSEEK_FLAG_BACKWARD) ? a : b;
2021
2022     if (!(flags & AVSEEK_FLAG_ANY))
2023         while (m >= 0 && m < nb_entries &&
2024                !(entries[m].flags & AVINDEX_KEYFRAME))
2025             m += (flags & AVSEEK_FLAG_BACKWARD) ? -1 : 1;
2026
2027     if (m == nb_entries)
2028         return -1;
2029     return m;
2030 }
2031
2032 void ff_configure_buffers_for_index(AVFormatContext *s, int64_t time_tolerance)
2033 {
2034     int ist1, ist2;
2035     int64_t pos_delta = 0;
2036     int64_t skip = 0;
2037     //We could use URLProtocol flags here but as many user applications do not use URLProtocols this would be unreliable
2038     const char *proto = avio_find_protocol_name(s->filename);
2039
2040     if (!proto) {
2041         av_log(s, AV_LOG_INFO,
2042                "Protocol name not provided, cannot determine if input is local or "
2043                "a network protocol, buffers and access patterns cannot be configured "
2044                "optimally without knowing the protocol\n");
2045     }
2046
2047     if (proto && !(strcmp(proto, "file") && strcmp(proto, "pipe") && strcmp(proto, "cache")))
2048         return;
2049
2050     for (ist1 = 0; ist1 < s->nb_streams; ist1++) {
2051         AVStream *st1 = s->streams[ist1];
2052         for (ist2 = 0; ist2 < s->nb_streams; ist2++) {
2053             AVStream *st2 = s->streams[ist2];
2054             int i1, i2;
2055
2056             if (ist1 == ist2)
2057                 continue;
2058
2059             for (i1 = i2 = 0; i1 < st1->nb_index_entries; i1++) {
2060                 AVIndexEntry *e1 = &st1->index_entries[i1];
2061                 int64_t e1_pts = av_rescale_q(e1->timestamp, st1->time_base, AV_TIME_BASE_Q);
2062
2063                 skip = FFMAX(skip, e1->size);
2064                 for (; i2 < st2->nb_index_entries; i2++) {
2065                     AVIndexEntry *e2 = &st2->index_entries[i2];
2066                     int64_t e2_pts = av_rescale_q(e2->timestamp, st2->time_base, AV_TIME_BASE_Q);
2067                     if (e2_pts - e1_pts < time_tolerance)
2068                         continue;
2069                     pos_delta = FFMAX(pos_delta, e1->pos - e2->pos);
2070                     break;
2071                 }
2072             }
2073         }
2074     }
2075
2076     pos_delta *= 2;
2077     /* XXX This could be adjusted depending on protocol*/
2078     if (s->pb->buffer_size < pos_delta && pos_delta < (1<<24)) {
2079         av_log(s, AV_LOG_VERBOSE, "Reconfiguring buffers to size %"PRId64"\n", pos_delta);
2080         ffio_set_buf_size(s->pb, pos_delta);
2081         s->pb->short_seek_threshold = FFMAX(s->pb->short_seek_threshold, pos_delta/2);
2082     }
2083
2084     if (skip < (1<<23)) {
2085         s->pb->short_seek_threshold = FFMAX(s->pb->short_seek_threshold, skip);
2086     }
2087 }
2088
2089 int av_index_search_timestamp(AVStream *st, int64_t wanted_timestamp, int flags)
2090 {
2091     return ff_index_search_timestamp(st->index_entries, st->nb_index_entries,
2092                                      wanted_timestamp, flags);
2093 }
2094
2095 static int64_t ff_read_timestamp(AVFormatContext *s, int stream_index, int64_t *ppos, int64_t pos_limit,
2096                                  int64_t (*read_timestamp)(struct AVFormatContext *, int , int64_t *, int64_t ))
2097 {
2098     int64_t ts = read_timestamp(s, stream_index, ppos, pos_limit);
2099     if (stream_index >= 0)
2100         ts = wrap_timestamp(s->streams[stream_index], ts);
2101     return ts;
2102 }
2103
2104 int ff_seek_frame_binary(AVFormatContext *s, int stream_index,
2105                          int64_t target_ts, int flags)
2106 {
2107     AVInputFormat *avif = s->iformat;
2108     int64_t av_uninit(pos_min), av_uninit(pos_max), pos, pos_limit;
2109     int64_t ts_min, ts_max, ts;
2110     int index;
2111     int64_t ret;
2112     AVStream *st;
2113
2114     if (stream_index < 0)
2115         return -1;
2116
2117     av_log(s, AV_LOG_TRACE, "read_seek: %d %s\n", stream_index, av_ts2str(target_ts));
2118
2119     ts_max =
2120     ts_min = AV_NOPTS_VALUE;
2121     pos_limit = -1; // GCC falsely says it may be uninitialized.
2122
2123     st = s->streams[stream_index];
2124     if (st->index_entries) {
2125         AVIndexEntry *e;
2126
2127         /* FIXME: Whole function must be checked for non-keyframe entries in
2128          * index case, especially read_timestamp(). */
2129         index = av_index_search_timestamp(st, target_ts,
2130                                           flags | AVSEEK_FLAG_BACKWARD);
2131         index = FFMAX(index, 0);
2132         e     = &st->index_entries[index];
2133
2134         if (e->timestamp <= target_ts || e->pos == e->min_distance) {
2135             pos_min = e->pos;
2136             ts_min  = e->timestamp;
2137             av_log(s, AV_LOG_TRACE, "using cached pos_min=0x%"PRIx64" dts_min=%s\n",
2138                     pos_min, av_ts2str(ts_min));
2139         } else {
2140             av_assert1(index == 0);
2141         }
2142
2143         index = av_index_search_timestamp(st, target_ts,
2144                                           flags & ~AVSEEK_FLAG_BACKWARD);
2145         av_assert0(index < st->nb_index_entries);
2146         if (index >= 0) {
2147             e = &st->index_entries[index];
2148             av_assert1(e->timestamp >= target_ts);
2149             pos_max   = e->pos;
2150             ts_max    = e->timestamp;
2151             pos_limit = pos_max - e->min_distance;
2152             av_log(s, AV_LOG_TRACE, "using cached pos_max=0x%"PRIx64" pos_limit=0x%"PRIx64
2153                     " dts_max=%s\n", pos_max, pos_limit, av_ts2str(ts_max));
2154         }
2155     }
2156
2157     pos = ff_gen_search(s, stream_index, target_ts, pos_min, pos_max, pos_limit,
2158                         ts_min, ts_max, flags, &ts, avif->read_timestamp);
2159     if (pos < 0)
2160         return -1;
2161
2162     /* do the seek */
2163     if ((ret = avio_seek(s->pb, pos, SEEK_SET)) < 0)
2164         return ret;
2165
2166     ff_read_frame_flush(s);
2167     ff_update_cur_dts(s, st, ts);
2168
2169     return 0;
2170 }
2171
2172 int ff_find_last_ts(AVFormatContext *s, int stream_index, int64_t *ts, int64_t *pos,
2173                     int64_t (*read_timestamp)(struct AVFormatContext *, int , int64_t *, int64_t ))
2174 {
2175     int64_t step = 1024;
2176     int64_t limit, ts_max;
2177     int64_t filesize = avio_size(s->pb);
2178     int64_t pos_max  = filesize - 1;
2179     do {
2180         limit = pos_max;
2181         pos_max = FFMAX(0, (pos_max) - step);
2182         ts_max  = ff_read_timestamp(s, stream_index,
2183                                     &pos_max, limit, read_timestamp);
2184         step   += step;
2185     } while (ts_max == AV_NOPTS_VALUE && 2*limit > step);
2186     if (ts_max == AV_NOPTS_VALUE)
2187         return -1;
2188
2189     for (;;) {
2190         int64_t tmp_pos = pos_max + 1;
2191         int64_t tmp_ts  = ff_read_timestamp(s, stream_index,
2192                                             &tmp_pos, INT64_MAX, read_timestamp);
2193         if (tmp_ts == AV_NOPTS_VALUE)
2194             break;
2195         av_assert0(tmp_pos > pos_max);
2196         ts_max  = tmp_ts;
2197         pos_max = tmp_pos;
2198         if (tmp_pos >= filesize)
2199             break;
2200     }
2201
2202     if (ts)
2203         *ts = ts_max;
2204     if (pos)
2205         *pos = pos_max;
2206
2207     return 0;
2208 }
2209
2210 int64_t ff_gen_search(AVFormatContext *s, int stream_index, int64_t target_ts,
2211                       int64_t pos_min, int64_t pos_max, int64_t pos_limit,
2212                       int64_t ts_min, int64_t ts_max,
2213                       int flags, int64_t *ts_ret,
2214                       int64_t (*read_timestamp)(struct AVFormatContext *, int,
2215                                                 int64_t *, int64_t))
2216 {
2217     int64_t pos, ts;
2218     int64_t start_pos;
2219     int no_change;
2220     int ret;
2221
2222     av_log(s, AV_LOG_TRACE, "gen_seek: %d %s\n", stream_index, av_ts2str(target_ts));
2223
2224     if (ts_min == AV_NOPTS_VALUE) {
2225         pos_min = s->internal->data_offset;
2226         ts_min  = ff_read_timestamp(s, stream_index, &pos_min, INT64_MAX, read_timestamp);
2227         if (ts_min == AV_NOPTS_VALUE)
2228             return -1;
2229     }
2230
2231     if (ts_min >= target_ts) {
2232         *ts_ret = ts_min;
2233         return pos_min;
2234     }
2235
2236     if (ts_max == AV_NOPTS_VALUE) {
2237         if ((ret = ff_find_last_ts(s, stream_index, &ts_max, &pos_max, read_timestamp)) < 0)
2238             return ret;
2239         pos_limit = pos_max;
2240     }
2241
2242     if (ts_max <= target_ts) {
2243         *ts_ret = ts_max;
2244         return pos_max;
2245     }
2246
2247     av_assert0(ts_min < ts_max);
2248
2249     no_change = 0;
2250     while (pos_min < pos_limit) {
2251         av_log(s, AV_LOG_TRACE,
2252                 "pos_min=0x%"PRIx64" pos_max=0x%"PRIx64" dts_min=%s dts_max=%s\n",
2253                 pos_min, pos_max, av_ts2str(ts_min), av_ts2str(ts_max));
2254         av_assert0(pos_limit <= pos_max);
2255
2256         if (no_change == 0) {
2257             int64_t approximate_keyframe_distance = pos_max - pos_limit;
2258             // interpolate position (better than dichotomy)
2259             pos = av_rescale(target_ts - ts_min, pos_max - pos_min,
2260                              ts_max - ts_min) +
2261                   pos_min - approximate_keyframe_distance;
2262         } else if (no_change == 1) {
2263             // bisection if interpolation did not change min / max pos last time
2264             pos = (pos_min + pos_limit) >> 1;
2265         } else {
2266             /* linear search if bisection failed, can only happen if there
2267              * are very few or no keyframes between min/max */
2268             pos = pos_min;
2269         }
2270         if (pos <= pos_min)
2271             pos = pos_min + 1;
2272         else if (pos > pos_limit)
2273             pos = pos_limit;
2274         start_pos = pos;
2275
2276         // May pass pos_limit instead of -1.
2277         ts = ff_read_timestamp(s, stream_index, &pos, INT64_MAX, read_timestamp);
2278         if (pos == pos_max)
2279             no_change++;
2280         else
2281             no_change = 0;
2282         av_log(s, AV_LOG_TRACE, "%"PRId64" %"PRId64" %"PRId64" / %s %s %s"
2283                 " target:%s limit:%"PRId64" start:%"PRId64" noc:%d\n",
2284                 pos_min, pos, pos_max,
2285                 av_ts2str(ts_min), av_ts2str(ts), av_ts2str(ts_max), av_ts2str(target_ts),
2286                 pos_limit, start_pos, no_change);
2287         if (ts == AV_NOPTS_VALUE) {
2288             av_log(s, AV_LOG_ERROR, "read_timestamp() failed in the middle\n");
2289             return -1;
2290         }
2291         if (target_ts <= ts) {
2292             pos_limit = start_pos - 1;
2293             pos_max   = pos;
2294             ts_max    = ts;
2295         }
2296         if (target_ts >= ts) {
2297             pos_min = pos;
2298             ts_min  = ts;
2299         }
2300     }
2301
2302     pos     = (flags & AVSEEK_FLAG_BACKWARD) ? pos_min : pos_max;
2303     ts      = (flags & AVSEEK_FLAG_BACKWARD) ? ts_min  : ts_max;
2304 #if 0
2305     pos_min = pos;
2306     ts_min  = ff_read_timestamp(s, stream_index, &pos_min, INT64_MAX, read_timestamp);
2307     pos_min++;
2308     ts_max = ff_read_timestamp(s, stream_index, &pos_min, INT64_MAX, read_timestamp);
2309     av_log(s, AV_LOG_TRACE, "pos=0x%"PRIx64" %s<=%s<=%s\n",
2310             pos, av_ts2str(ts_min), av_ts2str(target_ts), av_ts2str(ts_max));
2311 #endif
2312     *ts_ret = ts;
2313     return pos;
2314 }
2315
2316 static int seek_frame_byte(AVFormatContext *s, int stream_index,
2317                            int64_t pos, int flags)
2318 {
2319     int64_t pos_min, pos_max;
2320
2321     pos_min = s->internal->data_offset;
2322     pos_max = avio_size(s->pb) - 1;
2323
2324     if (pos < pos_min)
2325         pos = pos_min;
2326     else if (pos > pos_max)
2327         pos = pos_max;
2328
2329     avio_seek(s->pb, pos, SEEK_SET);
2330
2331     s->io_repositioned = 1;
2332
2333     return 0;
2334 }
2335
2336 static int seek_frame_generic(AVFormatContext *s, int stream_index,
2337                               int64_t timestamp, int flags)
2338 {
2339     int index;
2340     int64_t ret;
2341     AVStream *st;
2342     AVIndexEntry *ie;
2343
2344     st = s->streams[stream_index];
2345
2346     index = av_index_search_timestamp(st, timestamp, flags);
2347
2348     if (index < 0 && st->nb_index_entries &&
2349         timestamp < st->index_entries[0].timestamp)
2350         return -1;
2351
2352     if (index < 0 || index == st->nb_index_entries - 1) {
2353         AVPacket pkt;
2354         int nonkey = 0;
2355
2356         if (st->nb_index_entries) {
2357             av_assert0(st->index_entries);
2358             ie = &st->index_entries[st->nb_index_entries - 1];
2359             if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
2360                 return ret;
2361             ff_update_cur_dts(s, st, ie->timestamp);
2362         } else {
2363             if ((ret = avio_seek(s->pb, s->internal->data_offset, SEEK_SET)) < 0)
2364                 return ret;
2365         }
2366         for (;;) {
2367             int read_status;
2368             do {
2369                 read_status = av_read_frame(s, &pkt);
2370             } while (read_status == AVERROR(EAGAIN));
2371             if (read_status < 0)
2372                 break;
2373             if (stream_index == pkt.stream_index && pkt.dts > timestamp) {
2374                 if (pkt.flags & AV_PKT_FLAG_KEY) {
2375                     av_packet_unref(&pkt);
2376                     break;
2377                 }
2378                 if (nonkey++ > 1000 && st->codecpar->codec_id != AV_CODEC_ID_CDGRAPHICS) {
2379                     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);
2380                     av_packet_unref(&pkt);
2381                     break;
2382                 }
2383             }
2384             av_packet_unref(&pkt);
2385         }
2386         index = av_index_search_timestamp(st, timestamp, flags);
2387     }
2388     if (index < 0)
2389         return -1;
2390
2391     ff_read_frame_flush(s);
2392     if (s->iformat->read_seek)
2393         if (s->iformat->read_seek(s, stream_index, timestamp, flags) >= 0)
2394             return 0;
2395     ie = &st->index_entries[index];
2396     if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
2397         return ret;
2398     ff_update_cur_dts(s, st, ie->timestamp);
2399
2400     return 0;
2401 }
2402
2403 static int seek_frame_internal(AVFormatContext *s, int stream_index,
2404                                int64_t timestamp, int flags)
2405 {
2406     int ret;
2407     AVStream *st;
2408
2409     if (flags & AVSEEK_FLAG_BYTE) {
2410         if (s->iformat->flags & AVFMT_NO_BYTE_SEEK)
2411             return -1;
2412         ff_read_frame_flush(s);
2413         return seek_frame_byte(s, stream_index, timestamp, flags);
2414     }
2415
2416     if (stream_index < 0) {
2417         stream_index = av_find_default_stream_index(s);
2418         if (stream_index < 0)
2419             return -1;
2420
2421         st = s->streams[stream_index];
2422         /* timestamp for default must be expressed in AV_TIME_BASE units */
2423         timestamp = av_rescale(timestamp, st->time_base.den,
2424                                AV_TIME_BASE * (int64_t) st->time_base.num);
2425     }
2426
2427     /* first, we try the format specific seek */
2428     if (s->iformat->read_seek) {
2429         ff_read_frame_flush(s);
2430         ret = s->iformat->read_seek(s, stream_index, timestamp, flags);
2431     } else
2432         ret = -1;
2433     if (ret >= 0)
2434         return 0;
2435
2436     if (s->iformat->read_timestamp &&
2437         !(s->iformat->flags & AVFMT_NOBINSEARCH)) {
2438         ff_read_frame_flush(s);
2439         return ff_seek_frame_binary(s, stream_index, timestamp, flags);
2440     } else if (!(s->iformat->flags & AVFMT_NOGENSEARCH)) {
2441         ff_read_frame_flush(s);
2442         return seek_frame_generic(s, stream_index, timestamp, flags);
2443     } else
2444         return -1;
2445 }
2446
2447 int av_seek_frame(AVFormatContext *s, int stream_index,
2448                   int64_t timestamp, int flags)
2449 {
2450     int ret;
2451
2452     if (s->iformat->read_seek2 && !s->iformat->read_seek) {
2453         int64_t min_ts = INT64_MIN, max_ts = INT64_MAX;
2454         if ((flags & AVSEEK_FLAG_BACKWARD))
2455             max_ts = timestamp;
2456         else
2457             min_ts = timestamp;
2458         return avformat_seek_file(s, stream_index, min_ts, timestamp, max_ts,
2459                                   flags & ~AVSEEK_FLAG_BACKWARD);
2460     }
2461
2462     ret = seek_frame_internal(s, stream_index, timestamp, flags);
2463
2464     if (ret >= 0)
2465         ret = avformat_queue_attached_pictures(s);
2466
2467     return ret;
2468 }
2469
2470 int avformat_seek_file(AVFormatContext *s, int stream_index, int64_t min_ts,
2471                        int64_t ts, int64_t max_ts, int flags)
2472 {
2473     if (min_ts > ts || max_ts < ts)
2474         return -1;
2475     if (stream_index < -1 || stream_index >= (int)s->nb_streams)
2476         return AVERROR(EINVAL);
2477
2478     if (s->seek2any>0)
2479         flags |= AVSEEK_FLAG_ANY;
2480     flags &= ~AVSEEK_FLAG_BACKWARD;
2481
2482     if (s->iformat->read_seek2) {
2483         int ret;
2484         ff_read_frame_flush(s);
2485
2486         if (stream_index == -1 && s->nb_streams == 1) {
2487             AVRational time_base = s->streams[0]->time_base;
2488             ts = av_rescale_q(ts, AV_TIME_BASE_Q, time_base);
2489             min_ts = av_rescale_rnd(min_ts, time_base.den,
2490                                     time_base.num * (int64_t)AV_TIME_BASE,
2491                                     AV_ROUND_UP   | AV_ROUND_PASS_MINMAX);
2492             max_ts = av_rescale_rnd(max_ts, time_base.den,
2493                                     time_base.num * (int64_t)AV_TIME_BASE,
2494                                     AV_ROUND_DOWN | AV_ROUND_PASS_MINMAX);
2495             stream_index = 0;
2496         }
2497
2498         ret = s->iformat->read_seek2(s, stream_index, min_ts,
2499                                      ts, max_ts, flags);
2500
2501         if (ret >= 0)
2502             ret = avformat_queue_attached_pictures(s);
2503         return ret;
2504     }
2505
2506     if (s->iformat->read_timestamp) {
2507         // try to seek via read_timestamp()
2508     }
2509
2510     // Fall back on old API if new is not implemented but old is.
2511     // Note the old API has somewhat different semantics.
2512     if (s->iformat->read_seek || 1) {
2513         int dir = (ts - (uint64_t)min_ts > (uint64_t)max_ts - ts ? AVSEEK_FLAG_BACKWARD : 0);
2514         int ret = av_seek_frame(s, stream_index, ts, flags | dir);
2515         if (ret<0 && ts != min_ts && max_ts != ts) {
2516             ret = av_seek_frame(s, stream_index, dir ? max_ts : min_ts, flags | dir);
2517             if (ret >= 0)
2518                 ret = av_seek_frame(s, stream_index, ts, flags | (dir^AVSEEK_FLAG_BACKWARD));
2519         }
2520         return ret;
2521     }
2522
2523     // try some generic seek like seek_frame_generic() but with new ts semantics
2524     return -1; //unreachable
2525 }
2526
2527 int avformat_flush(AVFormatContext *s)
2528 {
2529     ff_read_frame_flush(s);
2530     return 0;
2531 }
2532
2533 /*******************************************************/
2534
2535 /**
2536  * Return TRUE if the stream has accurate duration in any stream.
2537  *
2538  * @return TRUE if the stream has accurate duration for at least one component.
2539  */
2540 static int has_duration(AVFormatContext *ic)
2541 {
2542     int i;
2543     AVStream *st;
2544
2545     for (i = 0; i < ic->nb_streams; i++) {
2546         st = ic->streams[i];
2547         if (st->duration != AV_NOPTS_VALUE)
2548             return 1;
2549     }
2550     if (ic->duration != AV_NOPTS_VALUE)
2551         return 1;
2552     return 0;
2553 }
2554
2555 /**
2556  * Estimate the stream timings from the one of each components.
2557  *
2558  * Also computes the global bitrate if possible.
2559  */
2560 static void update_stream_timings(AVFormatContext *ic)
2561 {
2562     int64_t start_time, start_time1, start_time_text, end_time, end_time1, end_time_text;
2563     int64_t duration, duration1, filesize;
2564     int i;
2565     AVStream *st;
2566     AVProgram *p;
2567
2568     start_time = INT64_MAX;
2569     start_time_text = INT64_MAX;
2570     end_time   = INT64_MIN;
2571     end_time_text   = INT64_MIN;
2572     duration   = INT64_MIN;
2573     for (i = 0; i < ic->nb_streams; i++) {
2574         st = ic->streams[i];
2575         if (st->start_time != AV_NOPTS_VALUE && st->time_base.den) {
2576             start_time1 = av_rescale_q(st->start_time, st->time_base,
2577                                        AV_TIME_BASE_Q);
2578             if (st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE || st->codecpar->codec_type == AVMEDIA_TYPE_DATA) {
2579                 if (start_time1 < start_time_text)
2580                     start_time_text = start_time1;
2581             } else
2582                 start_time = FFMIN(start_time, start_time1);
2583             end_time1 = av_rescale_q_rnd(st->duration, st->time_base,
2584                                          AV_TIME_BASE_Q,
2585                                          AV_ROUND_NEAR_INF|AV_ROUND_PASS_MINMAX);
2586             if (end_time1 != AV_NOPTS_VALUE && (end_time1 > 0 ? start_time1 <= INT64_MAX - end_time1 : start_time1 >= INT64_MIN - end_time1)) {
2587                 end_time1 += start_time1;
2588                 if (st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE || st->codecpar->codec_type == AVMEDIA_TYPE_DATA)
2589                     end_time_text = FFMAX(end_time_text, end_time1);
2590                 else
2591                     end_time = FFMAX(end_time, end_time1);
2592             }
2593             for (p = NULL; (p = av_find_program_from_stream(ic, p, i)); ) {
2594                 if (p->start_time == AV_NOPTS_VALUE || p->start_time > start_time1)
2595                     p->start_time = start_time1;
2596                 if (p->end_time < end_time1)
2597                     p->end_time = end_time1;
2598             }
2599         }
2600         if (st->duration != AV_NOPTS_VALUE) {
2601             duration1 = av_rescale_q(st->duration, st->time_base,
2602                                      AV_TIME_BASE_Q);
2603             duration  = FFMAX(duration, duration1);
2604         }
2605     }
2606     if (start_time == INT64_MAX || (start_time > start_time_text && start_time - start_time_text < AV_TIME_BASE))
2607         start_time = start_time_text;
2608     else if (start_time > start_time_text)
2609         av_log(ic, AV_LOG_VERBOSE, "Ignoring outlier non primary stream starttime %f\n", start_time_text / (float)AV_TIME_BASE);
2610
2611     if (end_time == INT64_MIN || (end_time < end_time_text && end_time_text - end_time < AV_TIME_BASE)) {
2612         end_time = end_time_text;
2613     } else if (end_time < end_time_text) {
2614         av_log(ic, AV_LOG_VERBOSE, "Ignoring outlier non primary stream endtime %f\n", end_time_text / (float)AV_TIME_BASE);
2615     }
2616
2617     if (start_time != INT64_MAX) {
2618         ic->start_time = start_time;
2619         if (end_time != INT64_MIN) {
2620             if (ic->nb_programs > 1) {
2621                 for (i = 0; i < ic->nb_programs; i++) {
2622                     p = ic->programs[i];
2623                     if (p->start_time != AV_NOPTS_VALUE &&
2624                         p->end_time > p->start_time &&
2625                         p->end_time - (uint64_t)p->start_time <= INT64_MAX)
2626                         duration = FFMAX(duration, p->end_time - p->start_time);
2627                 }
2628             } else if (end_time >= start_time && end_time - (uint64_t)start_time <= INT64_MAX) {
2629                 duration = FFMAX(duration, end_time - start_time);
2630             }
2631         }
2632     }
2633     if (duration != INT64_MIN && duration > 0 && ic->duration == AV_NOPTS_VALUE) {
2634         ic->duration = duration;
2635     }
2636     if (ic->pb && (filesize = avio_size(ic->pb)) > 0 && ic->duration > 0) {
2637         /* compute the bitrate */
2638         double bitrate = (double) filesize * 8.0 * AV_TIME_BASE /
2639                          (double) ic->duration;
2640         if (bitrate >= 0 && bitrate <= INT64_MAX)
2641             ic->bit_rate = bitrate;
2642     }
2643 }
2644
2645 static void fill_all_stream_timings(AVFormatContext *ic)
2646 {
2647     int i;
2648     AVStream *st;
2649
2650     update_stream_timings(ic);
2651     for (i = 0; i < ic->nb_streams; i++) {
2652         st = ic->streams[i];
2653         if (st->start_time == AV_NOPTS_VALUE) {
2654             if (ic->start_time != AV_NOPTS_VALUE)
2655                 st->start_time = av_rescale_q(ic->start_time, AV_TIME_BASE_Q,
2656                                               st->time_base);
2657             if (ic->duration != AV_NOPTS_VALUE)
2658                 st->duration = av_rescale_q(ic->duration, AV_TIME_BASE_Q,
2659                                             st->time_base);
2660         }
2661     }
2662 }
2663
2664 static void estimate_timings_from_bit_rate(AVFormatContext *ic)
2665 {
2666     int64_t filesize, duration;
2667     int i, show_warning = 0;
2668     AVStream *st;
2669
2670     /* if bit_rate is already set, we believe it */
2671     if (ic->bit_rate <= 0) {
2672         int64_t bit_rate = 0;
2673         for (i = 0; i < ic->nb_streams; i++) {
2674             st = ic->streams[i];
2675             if (st->codecpar->bit_rate <= 0 && st->internal->avctx->bit_rate > 0)
2676                 st->codecpar->bit_rate = st->internal->avctx->bit_rate;
2677             if (st->codecpar->bit_rate > 0) {
2678                 if (INT64_MAX - st->codecpar->bit_rate < bit_rate) {
2679                     bit_rate = 0;
2680                     break;
2681                 }
2682                 bit_rate += st->codecpar->bit_rate;
2683             } else if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO && st->codec_info_nb_frames > 1) {
2684                 // If we have a videostream with packets but without a bitrate
2685                 // then consider the sum not known
2686                 bit_rate = 0;
2687                 break;
2688             }
2689         }
2690         ic->bit_rate = bit_rate;
2691     }
2692
2693     /* if duration is already set, we believe it */
2694     if (ic->duration == AV_NOPTS_VALUE &&
2695         ic->bit_rate != 0) {
2696         filesize = ic->pb ? avio_size(ic->pb) : 0;
2697         if (filesize > ic->internal->data_offset) {
2698             filesize -= ic->internal->data_offset;
2699             for (i = 0; i < ic->nb_streams; i++) {
2700                 st      = ic->streams[i];
2701                 if (   st->time_base.num <= INT64_MAX / ic->bit_rate
2702                     && st->duration == AV_NOPTS_VALUE) {
2703                     duration = av_rescale(8 * filesize, st->time_base.den,
2704                                           ic->bit_rate *
2705                                           (int64_t) st->time_base.num);
2706                     st->duration = duration;
2707                     show_warning = 1;
2708                 }
2709             }
2710         }
2711     }
2712     if (show_warning)
2713         av_log(ic, AV_LOG_WARNING,
2714                "Estimating duration from bitrate, this may be inaccurate\n");
2715 }
2716
2717 #define DURATION_MAX_READ_SIZE 250000LL
2718 #define DURATION_MAX_RETRY 6
2719
2720 /* only usable for MPEG-PS streams */
2721 static void estimate_timings_from_pts(AVFormatContext *ic, int64_t old_offset)
2722 {
2723     AVPacket pkt1, *pkt = &pkt1;
2724     AVStream *st;
2725     int num, den, read_size, i, ret;
2726     int found_duration = 0;
2727     int is_end;
2728     int64_t filesize, offset, duration;
2729     int retry = 0;
2730
2731     /* flush packet queue */
2732     flush_packet_queue(ic);
2733
2734     for (i = 0; i < ic->nb_streams; i++) {
2735         st = ic->streams[i];
2736         if (st->start_time == AV_NOPTS_VALUE &&
2737             st->first_dts == AV_NOPTS_VALUE &&
2738             st->codecpar->codec_type != AVMEDIA_TYPE_UNKNOWN)
2739             av_log(ic, AV_LOG_WARNING,
2740                    "start time for stream %d is not set in estimate_timings_from_pts\n", i);
2741
2742         if (st->parser) {
2743             av_parser_close(st->parser);
2744             st->parser = NULL;
2745         }
2746     }
2747
2748     av_opt_set(ic, "skip_changes", "1", AV_OPT_SEARCH_CHILDREN);
2749     /* estimate the end time (duration) */
2750     /* XXX: may need to support wrapping */
2751     filesize = ic->pb ? avio_size(ic->pb) : 0;
2752     do {
2753         is_end = found_duration;
2754         offset = filesize - (DURATION_MAX_READ_SIZE << retry);
2755         if (offset < 0)
2756             offset = 0;
2757
2758         avio_seek(ic->pb, offset, SEEK_SET);
2759         read_size = 0;
2760         for (;;) {
2761             if (read_size >= DURATION_MAX_READ_SIZE << (FFMAX(retry - 1, 0)))
2762                 break;
2763
2764             do {
2765                 ret = ff_read_packet(ic, pkt);
2766             } while (ret == AVERROR(EAGAIN));
2767             if (ret != 0)
2768                 break;
2769             read_size += pkt->size;
2770             st         = ic->streams[pkt->stream_index];
2771             if (pkt->pts != AV_NOPTS_VALUE &&
2772                 (st->start_time != AV_NOPTS_VALUE ||
2773                  st->first_dts  != AV_NOPTS_VALUE)) {
2774                 if (pkt->duration == 0) {
2775                     ff_compute_frame_duration(ic, &num, &den, st, st->parser, pkt);
2776                     if (den && num) {
2777                         pkt->duration = av_rescale_rnd(1,
2778                                            num * (int64_t) st->time_base.den,
2779                                            den * (int64_t) st->time_base.num,
2780                                            AV_ROUND_DOWN);
2781                     }
2782                 }
2783                 duration = pkt->pts + pkt->duration;
2784                 found_duration = 1;
2785                 if (st->start_time != AV_NOPTS_VALUE)
2786                     duration -= st->start_time;
2787                 else
2788                     duration -= st->first_dts;
2789                 if (duration > 0) {
2790                     if (st->duration == AV_NOPTS_VALUE || st->info->last_duration<= 0 ||
2791                         (st->duration < duration && FFABS(duration - st->info->last_duration) < 60LL*st->time_base.den / st->time_base.num))
2792                         st->duration = duration;
2793                     st->info->last_duration = duration;
2794                 }
2795             }
2796             av_packet_unref(pkt);
2797         }
2798
2799         /* check if all audio/video streams have valid duration */
2800         if (!is_end) {
2801             is_end = 1;
2802             for (i = 0; i < ic->nb_streams; i++) {
2803                 st = ic->streams[i];
2804                 switch (st->codecpar->codec_type) {
2805                     case AVMEDIA_TYPE_VIDEO:
2806                     case AVMEDIA_TYPE_AUDIO:
2807                         if (st->duration == AV_NOPTS_VALUE)
2808                             is_end = 0;
2809                 }
2810             }
2811         }
2812     } while (!is_end &&
2813              offset &&
2814              ++retry <= DURATION_MAX_RETRY);
2815
2816     av_opt_set(ic, "skip_changes", "0", AV_OPT_SEARCH_CHILDREN);
2817
2818     /* warn about audio/video streams which duration could not be estimated */
2819     for (i = 0; i < ic->nb_streams; i++) {
2820         st = ic->streams[i];
2821         if (st->duration == AV_NOPTS_VALUE) {
2822             switch (st->codecpar->codec_type) {
2823             case AVMEDIA_TYPE_VIDEO:
2824             case AVMEDIA_TYPE_AUDIO:
2825                 if (st->start_time != AV_NOPTS_VALUE || st->first_dts  != AV_NOPTS_VALUE) {
2826                     av_log(ic, AV_LOG_DEBUG, "stream %d : no PTS found at end of file, duration not set\n", i);
2827                 } else
2828                     av_log(ic, AV_LOG_DEBUG, "stream %d : no TS found at start of file, duration not set\n", i);
2829             }
2830         }
2831     }
2832     fill_all_stream_timings(ic);
2833
2834     avio_seek(ic->pb, old_offset, SEEK_SET);
2835     for (i = 0; i < ic->nb_streams; i++) {
2836         int j;
2837
2838         st              = ic->streams[i];
2839         st->cur_dts     = st->first_dts;
2840         st->last_IP_pts = AV_NOPTS_VALUE;
2841         st->last_dts_for_order_check = AV_NOPTS_VALUE;
2842         for (j = 0; j < MAX_REORDER_DELAY + 1; j++)
2843             st->pts_buffer[j] = AV_NOPTS_VALUE;
2844     }
2845 }
2846
2847 static void estimate_timings(AVFormatContext *ic, int64_t old_offset)
2848 {
2849     int64_t file_size;
2850
2851     /* get the file size, if possible */
2852     if (ic->iformat->flags & AVFMT_NOFILE) {
2853         file_size = 0;
2854     } else {
2855         file_size = avio_size(ic->pb);
2856         file_size = FFMAX(0, file_size);
2857     }
2858
2859     if ((!strcmp(ic->iformat->name, "mpeg") ||
2860          !strcmp(ic->iformat->name, "mpegts")) &&
2861         file_size && (ic->pb->seekable & AVIO_SEEKABLE_NORMAL)) {
2862         /* get accurate estimate from the PTSes */
2863         estimate_timings_from_pts(ic, old_offset);
2864         ic->duration_estimation_method = AVFMT_DURATION_FROM_PTS;
2865     } else if (has_duration(ic)) {
2866         /* at least one component has timings - we use them for all
2867          * the components */
2868         fill_all_stream_timings(ic);
2869         ic->duration_estimation_method = AVFMT_DURATION_FROM_STREAM;
2870     } else {
2871         /* less precise: use bitrate info */
2872         estimate_timings_from_bit_rate(ic);
2873         ic->duration_estimation_method = AVFMT_DURATION_FROM_BITRATE;
2874     }
2875     update_stream_timings(ic);
2876
2877     {
2878         int i;
2879         AVStream av_unused *st;
2880         for (i = 0; i < ic->nb_streams; i++) {
2881             st = ic->streams[i];
2882             av_log(ic, AV_LOG_TRACE, "stream %d: start_time: %0.3f duration: %0.3f\n", i,
2883                    (double) st->start_time * av_q2d(st->time_base),
2884                    (double) st->duration   * av_q2d(st->time_base));
2885         }
2886         av_log(ic, AV_LOG_TRACE,
2887                 "format: start_time: %0.3f duration: %0.3f bitrate=%"PRId64" kb/s\n",
2888                 (double) ic->start_time / AV_TIME_BASE,
2889                 (double) ic->duration   / AV_TIME_BASE,
2890                 (int64_t)ic->bit_rate / 1000);
2891     }
2892 }
2893
2894 static int has_codec_parameters(AVStream *st, const char **errmsg_ptr)
2895 {
2896     AVCodecContext *avctx = st->internal->avctx;
2897
2898 #define FAIL(errmsg) do {                                         \
2899         if (errmsg_ptr)                                           \
2900             *errmsg_ptr = errmsg;                                 \
2901         return 0;                                                 \
2902     } while (0)
2903
2904     if (   avctx->codec_id == AV_CODEC_ID_NONE
2905         && avctx->codec_type != AVMEDIA_TYPE_DATA)
2906         FAIL("unknown codec");
2907     switch (avctx->codec_type) {
2908     case AVMEDIA_TYPE_AUDIO:
2909         if (!avctx->frame_size && determinable_frame_size(avctx))
2910             FAIL("unspecified frame size");
2911         if (st->info->found_decoder >= 0 &&
2912             avctx->sample_fmt == AV_SAMPLE_FMT_NONE)
2913             FAIL("unspecified sample format");
2914         if (!avctx->sample_rate)
2915             FAIL("unspecified sample rate");
2916         if (!avctx->channels)
2917             FAIL("unspecified number of channels");
2918         if (st->info->found_decoder >= 0 && !st->nb_decoded_frames && avctx->codec_id == AV_CODEC_ID_DTS)
2919             FAIL("no decodable DTS frames");
2920         break;
2921     case AVMEDIA_TYPE_VIDEO:
2922         if (!avctx->width)
2923             FAIL("unspecified size");
2924         if (st->info->found_decoder >= 0 && avctx->pix_fmt == AV_PIX_FMT_NONE)
2925             FAIL("unspecified pixel format");
2926         if (st->codecpar->codec_id == AV_CODEC_ID_RV30 || st->codecpar->codec_id == AV_CODEC_ID_RV40)
2927             if (!st->sample_aspect_ratio.num && !st->codecpar->sample_aspect_ratio.num && !st->codec_info_nb_frames)
2928                 FAIL("no frame in rv30/40 and no sar");
2929         break;
2930     case AVMEDIA_TYPE_SUBTITLE:
2931         if (avctx->codec_id == AV_CODEC_ID_HDMV_PGS_SUBTITLE && !avctx->width)
2932             FAIL("unspecified size");
2933         break;
2934     case AVMEDIA_TYPE_DATA:
2935         if (avctx->codec_id == AV_CODEC_ID_NONE) return 1;
2936     }
2937
2938     return 1;
2939 }
2940
2941 /* returns 1 or 0 if or if not decoded data was returned, or a negative error */
2942 static int try_decode_frame(AVFormatContext *s, AVStream *st, AVPacket *avpkt,
2943                             AVDictionary **options)
2944 {
2945     AVCodecContext *avctx = st->internal->avctx;
2946     const AVCodec *codec;
2947     int got_picture = 1, ret = 0;
2948     AVFrame *frame = av_frame_alloc();
2949     AVSubtitle subtitle;
2950     AVPacket pkt = *avpkt;
2951     int do_skip_frame = 0;
2952     enum AVDiscard skip_frame;
2953
2954     if (!frame)
2955         return AVERROR(ENOMEM);
2956
2957     if (!avcodec_is_open(avctx) &&
2958         st->info->found_decoder <= 0 &&
2959         (st->codecpar->codec_id != -st->info->found_decoder || !st->codecpar->codec_id)) {
2960         AVDictionary *thread_opt = NULL;
2961
2962         codec = find_probe_decoder(s, st, st->codecpar->codec_id);
2963
2964         if (!codec) {
2965             st->info->found_decoder = -st->codecpar->codec_id;
2966             ret                     = -1;
2967             goto fail;
2968         }
2969
2970         /* Force thread count to 1 since the H.264 decoder will not extract
2971          * SPS and PPS to extradata during multi-threaded decoding. */
2972         av_dict_set(options ? options : &thread_opt, "threads", "1", 0);
2973         if (s->codec_whitelist)
2974             av_dict_set(options ? options : &thread_opt, "codec_whitelist", s->codec_whitelist, 0);
2975         ret = avcodec_open2(avctx, codec, options ? options : &thread_opt);
2976         if (!options)
2977             av_dict_free(&thread_opt);
2978         if (ret < 0) {
2979             st->info->found_decoder = -avctx->codec_id;
2980             goto fail;
2981         }
2982         st->info->found_decoder = 1;
2983     } else if (!st->info->found_decoder)
2984         st->info->found_decoder = 1;
2985
2986     if (st->info->found_decoder < 0) {
2987         ret = -1;
2988         goto fail;
2989     }
2990
2991     if (avpriv_codec_get_cap_skip_frame_fill_param(avctx->codec)) {
2992         do_skip_frame = 1;
2993         skip_frame = avctx->skip_frame;
2994         avctx->skip_frame = AVDISCARD_ALL;
2995     }
2996
2997     while ((pkt.size > 0 || (!pkt.data && got_picture)) &&
2998            ret >= 0 &&
2999            (!has_codec_parameters(st, NULL) || !has_decode_delay_been_guessed(st) ||
3000             (!st->codec_info_nb_frames &&
3001              (avctx->codec->capabilities & AV_CODEC_CAP_CHANNEL_CONF)))) {
3002         got_picture = 0;
3003         if (avctx->codec_type == AVMEDIA_TYPE_VIDEO ||
3004             avctx->codec_type == AVMEDIA_TYPE_AUDIO) {
3005             ret = avcodec_send_packet(avctx, &pkt);
3006             if (ret < 0 && ret != AVERROR(EAGAIN) && ret != AVERROR_EOF)
3007                 break;
3008             if (ret >= 0)
3009                 pkt.size = 0;
3010             ret = avcodec_receive_frame(avctx, frame);
3011             if (ret >= 0)
3012                 got_picture = 1;
3013             if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF)
3014                 ret = 0;
3015         } else if (avctx->codec_type == AVMEDIA_TYPE_SUBTITLE) {
3016             ret = avcodec_decode_subtitle2(avctx, &subtitle,
3017                                            &got_picture, &pkt);
3018             if (ret >= 0)
3019                 pkt.size = 0;
3020         }
3021         if (ret >= 0) {
3022             if (got_picture)
3023                 st->nb_decoded_frames++;
3024             ret       = got_picture;
3025         }
3026     }
3027
3028     if (!pkt.data && !got_picture)
3029         ret = -1;
3030
3031 fail:
3032     if (do_skip_frame) {
3033         avctx->skip_frame = skip_frame;
3034     }
3035
3036     av_frame_free(&frame);
3037     return ret;
3038 }
3039
3040 unsigned int ff_codec_get_tag(const AVCodecTag *tags, enum AVCodecID id)
3041 {
3042     while (tags->id != AV_CODEC_ID_NONE) {
3043         if (tags->id == id)
3044             return tags->tag;
3045         tags++;
3046     }
3047     return 0;
3048 }
3049
3050 enum AVCodecID ff_codec_get_id(const AVCodecTag *tags, unsigned int tag)
3051 {
3052     int i;
3053     for (i = 0; tags[i].id != AV_CODEC_ID_NONE; i++)
3054         if (tag == tags[i].tag)
3055             return tags[i].id;
3056     for (i = 0; tags[i].id != AV_CODEC_ID_NONE; i++)
3057         if (avpriv_toupper4(tag) == avpriv_toupper4(tags[i].tag))
3058             return tags[i].id;
3059     return AV_CODEC_ID_NONE;
3060 }
3061
3062 enum AVCodecID ff_get_pcm_codec_id(int bps, int flt, int be, int sflags)
3063 {
3064     if (bps <= 0 || bps > 64)
3065         return AV_CODEC_ID_NONE;
3066
3067     if (flt) {
3068         switch (bps) {
3069         case 32:
3070             return be ? AV_CODEC_ID_PCM_F32BE : AV_CODEC_ID_PCM_F32LE;
3071         case 64:
3072             return be ? AV_CODEC_ID_PCM_F64BE : AV_CODEC_ID_PCM_F64LE;
3073         default:
3074             return AV_CODEC_ID_NONE;
3075         }
3076     } else {
3077         bps  += 7;
3078         bps >>= 3;
3079         if (sflags & (1 << (bps - 1))) {
3080             switch (bps) {
3081             case 1:
3082                 return AV_CODEC_ID_PCM_S8;
3083             case 2:
3084                 return be ? AV_CODEC_ID_PCM_S16BE : AV_CODEC_ID_PCM_S16LE;
3085             case 3:
3086                 return be ? AV_CODEC_ID_PCM_S24BE : AV_CODEC_ID_PCM_S24LE;
3087             case 4:
3088                 return be ? AV_CODEC_ID_PCM_S32BE : AV_CODEC_ID_PCM_S32LE;
3089             case 8:
3090                 return be ? AV_CODEC_ID_PCM_S64BE : AV_CODEC_ID_PCM_S64LE;
3091             default:
3092                 return AV_CODEC_ID_NONE;
3093             }
3094         } else {
3095             switch (bps) {
3096             case 1:
3097                 return AV_CODEC_ID_PCM_U8;
3098             case 2:
3099                 return be ? AV_CODEC_ID_PCM_U16BE : AV_CODEC_ID_PCM_U16LE;
3100             case 3:
3101                 return be ? AV_CODEC_ID_PCM_U24BE : AV_CODEC_ID_PCM_U24LE;
3102             case 4:
3103                 return be ? AV_CODEC_ID_PCM_U32BE : AV_CODEC_ID_PCM_U32LE;
3104             default:
3105                 return AV_CODEC_ID_NONE;
3106             }
3107         }
3108     }
3109 }
3110
3111 unsigned int av_codec_get_tag(const AVCodecTag *const *tags, enum AVCodecID id)
3112 {
3113     unsigned int tag;
3114     if (!av_codec_get_tag2(tags, id, &tag))
3115         return 0;
3116     return tag;
3117 }
3118
3119 int av_codec_get_tag2(const AVCodecTag * const *tags, enum AVCodecID id,
3120                       unsigned int *tag)
3121 {
3122     int i;
3123     for (i = 0; tags && tags[i]; i++) {
3124         const AVCodecTag *codec_tags = tags[i];
3125         while (codec_tags->id != AV_CODEC_ID_NONE) {
3126             if (codec_tags->id == id) {
3127                 *tag = codec_tags->tag;
3128                 return 1;
3129             }
3130             codec_tags++;
3131         }
3132     }
3133     return 0;
3134 }
3135
3136 enum AVCodecID av_codec_get_id(const AVCodecTag *const *tags, unsigned int tag)
3137 {
3138     int i;
3139     for (i = 0; tags && tags[i]; i++) {
3140         enum AVCodecID id = ff_codec_get_id(tags[i], tag);
3141         if (id != AV_CODEC_ID_NONE)
3142             return id;
3143     }
3144     return AV_CODEC_ID_NONE;
3145 }
3146
3147 static void compute_chapters_end(AVFormatContext *s)
3148 {
3149     unsigned int i, j;
3150     int64_t max_time = 0;
3151
3152     if (s->duration > 0 && s->start_time < INT64_MAX - s->duration)
3153         max_time = s->duration +
3154                        ((s->start_time == AV_NOPTS_VALUE) ? 0 : s->start_time);
3155
3156     for (i = 0; i < s->nb_chapters; i++)
3157         if (s->chapters[i]->end == AV_NOPTS_VALUE) {
3158             AVChapter *ch = s->chapters[i];
3159             int64_t end = max_time ? av_rescale_q(max_time, AV_TIME_BASE_Q,
3160                                                   ch->time_base)
3161                                    : INT64_MAX;
3162
3163             for (j = 0; j < s->nb_chapters; j++) {
3164                 AVChapter *ch1     = s->chapters[j];
3165                 int64_t next_start = av_rescale_q(ch1->start, ch1->time_base,
3166                                                   ch->time_base);
3167                 if (j != i && next_start > ch->start && next_start < end)
3168                     end = next_start;
3169             }
3170             ch->end = (end == INT64_MAX) ? ch->start : end;
3171         }
3172 }
3173
3174 static int get_std_framerate(int i)
3175 {
3176     if (i < 30*12)
3177         return (i + 1) * 1001;
3178     i -= 30*12;
3179
3180     if (i < 30)
3181         return (i + 31) * 1001 * 12;
3182     i -= 30;
3183
3184     if (i < 3)
3185         return ((const int[]) { 80, 120, 240})[i] * 1001 * 12;
3186
3187     i -= 3;
3188
3189     return ((const int[]) { 24, 30, 60, 12, 15, 48 })[i] * 1000 * 12;
3190 }
3191
3192 /* Is the time base unreliable?
3193  * This is a heuristic to balance between quick acceptance of the values in
3194  * the headers vs. some extra checks.
3195  * Old DivX and Xvid often have nonsense timebases like 1fps or 2fps.
3196  * MPEG-2 commonly misuses field repeat flags to store different framerates.
3197  * And there are "variable" fps files this needs to detect as well. */
3198 static int tb_unreliable(AVCodecContext *c)
3199 {
3200     if (c->time_base.den >= 101LL * c->time_base.num ||
3201         c->time_base.den <    5LL * c->time_base.num ||
3202         // c->codec_tag == AV_RL32("DIVX") ||
3203         // c->codec_tag == AV_RL32("XVID") ||
3204         c->codec_tag == AV_RL32("mp4v") ||
3205         c->codec_id == AV_CODEC_ID_MPEG2VIDEO ||
3206         c->codec_id == AV_CODEC_ID_GIF ||
3207         c->codec_id == AV_CODEC_ID_HEVC ||
3208         c->codec_id == AV_CODEC_ID_H264)
3209         return 1;
3210     return 0;
3211 }
3212
3213 int ff_alloc_extradata(AVCodecParameters *par, int size)
3214 {
3215     int ret;
3216
3217     if (size < 0 || size >= INT32_MAX - AV_INPUT_BUFFER_PADDING_SIZE) {
3218         par->extradata = NULL;
3219         par->extradata_size = 0;
3220         return AVERROR(EINVAL);
3221     }
3222     par->extradata = av_malloc(size + AV_INPUT_BUFFER_PADDING_SIZE);
3223     if (par->extradata) {
3224         memset(par->extradata + size, 0, AV_INPUT_BUFFER_PADDING_SIZE);
3225         par->extradata_size = size;
3226         ret = 0;
3227     } else {
3228         par->extradata_size = 0;
3229         ret = AVERROR(ENOMEM);
3230     }
3231     return ret;
3232 }
3233
3234 int ff_get_extradata(AVFormatContext *s, AVCodecParameters *par, AVIOContext *pb, int size)
3235 {
3236     int ret = ff_alloc_extradata(par, size);
3237     if (ret < 0)
3238         return ret;
3239     ret = avio_read(pb, par->extradata, size);
3240     if (ret != size) {
3241         av_freep(&par->extradata);
3242         par->extradata_size = 0;
3243         av_log(s, AV_LOG_ERROR, "Failed to read extradata of size %d\n", size);
3244         return ret < 0 ? ret : AVERROR_INVALIDDATA;
3245     }
3246
3247     return ret;
3248 }
3249
3250 int ff_rfps_add_frame(AVFormatContext *ic, AVStream *st, int64_t ts)
3251 {
3252     int i, j;
3253     int64_t last = st->info->last_dts;
3254
3255     if (   ts != AV_NOPTS_VALUE && last != AV_NOPTS_VALUE && ts > last
3256        && ts - (uint64_t)last < INT64_MAX) {
3257         double dts = (is_relative(ts) ?  ts - RELATIVE_TS_BASE : ts) * av_q2d(st->time_base);
3258         int64_t duration = ts - last;
3259
3260         if (!st->info->duration_error)
3261             st->info->duration_error = av_mallocz(sizeof(st->info->duration_error[0])*2);
3262         if (!st->info->duration_error)
3263             return AVERROR(ENOMEM);
3264
3265 //         if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
3266 //             av_log(NULL, AV_LOG_ERROR, "%f\n", dts);
3267         for (i = 0; i<MAX_STD_TIMEBASES; i++) {
3268             if (st->info->duration_error[0][1][i] < 1e10) {
3269                 int framerate = get_std_framerate(i);
3270                 double sdts = dts*framerate/(1001*12);
3271                 for (j= 0; j<2; j++) {
3272                     int64_t ticks = llrint(sdts+j*0.5);
3273                     double error= sdts - ticks + j*0.5;
3274                     st->info->duration_error[j][0][i] += error;
3275                     st->info->duration_error[j][1][i] += error*error;
3276                 }
3277             }
3278         }
3279         st->info->duration_count++;
3280         st->info->rfps_duration_sum += duration;
3281
3282         if (st->info->duration_count % 10 == 0) {
3283             int n = st->info->duration_count;
3284             for (i = 0; i<MAX_STD_TIMEBASES; i++) {
3285                 if (st->info->duration_error[0][1][i] < 1e10) {
3286                     double a0     = st->info->duration_error[0][0][i] / n;
3287                     double error0 = st->info->duration_error[0][1][i] / n - a0*a0;
3288                     double a1     = st->info->duration_error[1][0][i] / n;
3289                     double error1 = st->info->duration_error[1][1][i] / n - a1*a1;
3290                     if (error0 > 0.04 && error1 > 0.04) {
3291                         st->info->duration_error[0][1][i] = 2e10;
3292                         st->info->duration_error[1][1][i] = 2e10;
3293                     }
3294                 }
3295             }
3296         }
3297
3298         // ignore the first 4 values, they might have some random jitter
3299         if (st->info->duration_count > 3 && is_relative(ts) == is_relative(last))
3300             st->info->duration_gcd = av_gcd(st->info->duration_gcd, duration);
3301     }
3302     if (ts != AV_NOPTS_VALUE)
3303         st->info->last_dts = ts;
3304
3305     return 0;
3306 }
3307
3308 void ff_rfps_calculate(AVFormatContext *ic)
3309 {
3310     int i, j;
3311
3312     for (i = 0; i < ic->nb_streams; i++) {
3313         AVStream *st = ic->streams[i];
3314
3315         if (st->codecpar->codec_type != AVMEDIA_TYPE_VIDEO)
3316             continue;
3317         // the check for tb_unreliable() is not completely correct, since this is not about handling
3318         // an unreliable/inexact time base, but a time base that is finer than necessary, as e.g.
3319         // ipmovie.c produces.
3320         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)
3321             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);
3322         if (st->info->duration_count>1 && !st->r_frame_rate.num
3323             && tb_unreliable(st->internal->avctx)) {
3324             int num = 0;
3325             double best_error= 0.01;
3326             AVRational ref_rate = st->r_frame_rate.num ? st->r_frame_rate : av_inv_q(st->time_base);
3327
3328             for (j= 0; j<MAX_STD_TIMEBASES; j++) {
3329                 int k;
3330
3331                 if (st->info->codec_info_duration &&
3332                     st->info->codec_info_duration*av_q2d(st->time_base) < (1001*11.5)/get_std_framerate(j))
3333                     continue;
3334                 if (!st->info->codec_info_duration && get_std_framerate(j) < 1001*12)
3335                     continue;
3336
3337                 if (av_q2d(st->time_base) * st->info->rfps_duration_sum / st->info->duration_count < (1001*12.0 * 0.8)/get_std_framerate(j))
3338                     continue;
3339
3340                 for (k= 0; k<2; k++) {
3341                     int n = st->info->duration_count;
3342                     double a= st->info->duration_error[k][0][j] / n;
3343                     double error= st->info->duration_error[k][1][j]/n - a*a;
3344
3345                     if (error < best_error && best_error> 0.000000001) {
3346                         best_error= error;
3347                         num = get_std_framerate(j);
3348                     }
3349                     if (error < 0.02)
3350                         av_log(ic, AV_LOG_DEBUG, "rfps: %f %f\n", get_std_framerate(j) / 12.0/1001, error);
3351                 }
3352             }
3353             // do not increase frame rate by more than 1 % in order to match a standard rate.
3354             if (num && (!ref_rate.num || (double)num/(12*1001) < 1.01 * av_q2d(ref_rate)))
3355                 av_reduce(&st->r_frame_rate.num, &st->r_frame_rate.den, num, 12*1001, INT_MAX);
3356         }
3357         if (   !st->avg_frame_rate.num
3358             && st->r_frame_rate.num && st->info->rfps_duration_sum
3359             && st->info->codec_info_duration <= 0
3360             && st->info->duration_count > 2
3361             && 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
3362             ) {
3363             av_log(ic, AV_LOG_DEBUG, "Setting avg frame rate based on r frame rate\n");
3364             st->avg_frame_rate = st->r_frame_rate;
3365         }
3366
3367         av_freep(&st->info->duration_error);
3368         st->info->last_dts = AV_NOPTS_VALUE;
3369         st->info->duration_count = 0;
3370         st->info->rfps_duration_sum = 0;
3371     }
3372 }
3373
3374 static int extract_extradata_check(AVStream *st)
3375 {
3376     const AVBitStreamFilter *f;
3377
3378     f = av_bsf_get_by_name("extract_extradata");
3379     if (!f)
3380         return 0;
3381
3382     if (f->codec_ids) {
3383         const enum AVCodecID *ids;
3384         for (ids = f->codec_ids; *ids != AV_CODEC_ID_NONE; ids++)
3385             if (*ids == st->codecpar->codec_id)
3386                 return 1;
3387     }
3388
3389     return 0;
3390 }
3391
3392 static int extract_extradata_init(AVStream *st)
3393 {
3394     AVStreamInternal *i = st->internal;
3395     const AVBitStreamFilter *f;
3396     int ret;
3397
3398     f = av_bsf_get_by_name("extract_extradata");
3399     if (!f)
3400         goto finish;
3401
3402     /* check that the codec id is supported */
3403     ret = extract_extradata_check(st);
3404     if (!ret)
3405         goto finish;
3406
3407     i->extract_extradata.pkt = av_packet_alloc();
3408     if (!i->extract_extradata.pkt)
3409         return AVERROR(ENOMEM);
3410
3411     ret = av_bsf_alloc(f, &i->extract_extradata.bsf);
3412     if (ret < 0)
3413         goto fail;
3414
3415     ret = avcodec_parameters_copy(i->extract_extradata.bsf->par_in,
3416                                   st->codecpar);
3417     if (ret < 0)
3418         goto fail;
3419
3420     i->extract_extradata.bsf->time_base_in = st->time_base;
3421
3422     /* if init fails here, we assume extracting extradata is just not
3423      * supported for this codec, so we return success */
3424     ret = av_bsf_init(i->extract_extradata.bsf);
3425     if (ret < 0) {
3426         av_bsf_free(&i->extract_extradata.bsf);
3427         ret = 0;
3428     }
3429
3430 finish:
3431     i->extract_extradata.inited = 1;
3432
3433     return 0;
3434 fail:
3435     av_bsf_free(&i->extract_extradata.bsf);
3436     av_packet_free(&i->extract_extradata.pkt);
3437     return ret;
3438 }
3439
3440 static int extract_extradata(AVStream *st, AVPacket *pkt)
3441 {
3442     AVStreamInternal *i = st->internal;
3443     AVPacket *pkt_ref;
3444     int ret;
3445
3446     if (!i->extract_extradata.inited) {
3447         ret = extract_extradata_init(st);
3448         if (ret < 0)
3449             return ret;
3450     }
3451
3452     if (i->extract_extradata.inited && !i->extract_extradata.bsf)
3453         return 0;
3454
3455     pkt_ref = i->extract_extradata.pkt;
3456     ret = av_packet_ref(pkt_ref, pkt);
3457     if (ret < 0)
3458         return ret;
3459
3460     ret = av_bsf_send_packet(i->extract_extradata.bsf, pkt_ref);
3461     if (ret < 0) {
3462         av_packet_unref(pkt_ref);
3463         return ret;
3464     }
3465
3466     while (ret >= 0 && !i->avctx->extradata) {
3467         int extradata_size;
3468         uint8_t *extradata;
3469
3470         ret = av_bsf_receive_packet(i->extract_extradata.bsf, pkt_ref);
3471         if (ret < 0) {
3472             if (ret != AVERROR(EAGAIN) && ret != AVERROR_EOF)
3473                 return ret;
3474             continue;
3475         }
3476
3477         extradata = av_packet_get_side_data(pkt_ref, AV_PKT_DATA_NEW_EXTRADATA,
3478                                             &extradata_size);
3479
3480         if (extradata) {
3481             i->avctx->extradata = av_mallocz(extradata_size + AV_INPUT_BUFFER_PADDING_SIZE);
3482             if (!i->avctx->extradata) {
3483                 av_packet_unref(pkt_ref);
3484                 return AVERROR(ENOMEM);
3485             }
3486             memcpy(i->avctx->extradata, extradata, extradata_size);
3487             i->avctx->extradata_size = extradata_size;
3488         }
3489         av_packet_unref(pkt_ref);
3490     }
3491
3492     return 0;
3493 }
3494
3495 int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
3496 {
3497     int i, count = 0, ret = 0, j;
3498     int64_t read_size;
3499     AVStream *st;
3500     AVCodecContext *avctx;
3501     AVPacket pkt1, *pkt;
3502     int64_t old_offset  = avio_tell(ic->pb);
3503     // new streams might appear, no options for those
3504     int orig_nb_streams = ic->nb_streams;
3505     int flush_codecs;
3506     int64_t max_analyze_duration = ic->max_analyze_duration;
3507     int64_t max_stream_analyze_duration;
3508     int64_t max_subtitle_analyze_duration;
3509     int64_t probesize = ic->probesize;
3510     int eof_reached = 0;
3511     int *missing_streams = av_opt_ptr(ic->iformat->priv_class, ic->priv_data, "missing_streams");
3512
3513     flush_codecs = probesize > 0;
3514
3515     av_opt_set(ic, "skip_clear", "1", AV_OPT_SEARCH_CHILDREN);
3516
3517     max_stream_analyze_duration = max_analyze_duration;
3518     max_subtitle_analyze_duration = max_analyze_duration;
3519     if (!max_analyze_duration) {
3520         max_stream_analyze_duration =
3521         max_analyze_duration        = 5*AV_TIME_BASE;
3522         max_subtitle_analyze_duration = 30*AV_TIME_BASE;
3523         if (!strcmp(ic->iformat->name, "flv"))
3524             max_stream_analyze_duration = 90*AV_TIME_BASE;
3525         if (!strcmp(ic->iformat->name, "mpeg") || !strcmp(ic->iformat->name, "mpegts"))
3526             max_stream_analyze_duration = 7*AV_TIME_BASE;
3527     }
3528
3529     if (ic->pb)
3530         av_log(ic, AV_LOG_DEBUG, "Before avformat_find_stream_info() pos: %"PRId64" bytes read:%"PRId64" seeks:%d nb_streams:%d\n",
3531                avio_tell(ic->pb), ic->pb->bytes_read, ic->pb->seek_count, ic->nb_streams);
3532
3533     for (i = 0; i < ic->nb_streams; i++) {
3534         const AVCodec *codec;
3535         AVDictionary *thread_opt = NULL;
3536         st = ic->streams[i];
3537         avctx = st->internal->avctx;
3538
3539         if (st->codecpar->codec_type == AVMEDIA_TYPE_VIDEO ||
3540             st->codecpar->codec_type == AVMEDIA_TYPE_SUBTITLE) {
3541 /*            if (!st->time_base.num)
3542                 st->time_base = */
3543             if (!avctx->time_base.num)
3544                 avctx->time_base = st->time_base;
3545         }
3546
3547         /* check if the caller has overridden the codec id */
3548 #if FF_API_LAVF_AVCTX
3549 FF_DISABLE_DEPRECATION_WARNINGS
3550         if (st->codec->codec_id != st->internal->orig_codec_id) {
3551             st->codecpar->codec_id   = st->codec->codec_id;
3552             st->codecpar->codec_type = st->codec->codec_type;
3553             st->internal->orig_codec_id = st->codec->codec_id;
3554         }
3555 FF_ENABLE_DEPRECATION_WARNINGS
3556 #endif
3557