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