]> git.sesse.net Git - ffmpeg/blob - libavformat/utils.c
Merge remote-tracking branch 'qatar/master'
[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 /* #define DEBUG */
23
24 #include "avformat.h"
25 #include "avio_internal.h"
26 #include "internal.h"
27 #include "libavcodec/internal.h"
28 #include "libavcodec/raw.h"
29 #include "libavcodec/bytestream.h"
30 #include "libavutil/avassert.h"
31 #include "libavutil/opt.h"
32 #include "libavutil/dict.h"
33 #include "libavutil/pixdesc.h"
34 #include "metadata.h"
35 #include "id3v2.h"
36 #include "libavutil/avassert.h"
37 #include "libavutil/avstring.h"
38 #include "libavutil/mathematics.h"
39 #include "libavutil/parseutils.h"
40 #include "libavutil/time.h"
41 #include "libavutil/timestamp.h"
42 #include "riff.h"
43 #include "audiointerleave.h"
44 #include "url.h"
45 #include <stdarg.h>
46 #if CONFIG_NETWORK
47 #include "network.h"
48 #endif
49
50 #undef NDEBUG
51 #include <assert.h>
52
53 /**
54  * @file
55  * various utility functions for use within FFmpeg
56  */
57
58 unsigned avformat_version(void)
59 {
60     av_assert0(LIBAVFORMAT_VERSION_MICRO >= 100);
61     return LIBAVFORMAT_VERSION_INT;
62 }
63
64 const char *avformat_configuration(void)
65 {
66     return FFMPEG_CONFIGURATION;
67 }
68
69 const char *avformat_license(void)
70 {
71 #define LICENSE_PREFIX "libavformat license: "
72     return LICENSE_PREFIX FFMPEG_LICENSE + sizeof(LICENSE_PREFIX) - 1;
73 }
74
75 #define RELATIVE_TS_BASE (INT64_MAX - (1LL<<48))
76
77 static int is_relative(int64_t ts) {
78     return ts > (RELATIVE_TS_BASE - (1LL<<48));
79 }
80
81 /** head of registered input format linked list */
82 static AVInputFormat *first_iformat = NULL;
83 /** head of registered output format linked list */
84 static AVOutputFormat *first_oformat = NULL;
85
86 AVInputFormat  *av_iformat_next(AVInputFormat  *f)
87 {
88     if(f) return f->next;
89     else  return first_iformat;
90 }
91
92 AVOutputFormat *av_oformat_next(AVOutputFormat *f)
93 {
94     if(f) return f->next;
95     else  return first_oformat;
96 }
97
98 void av_register_input_format(AVInputFormat *format)
99 {
100     AVInputFormat **p;
101     p = &first_iformat;
102     while (*p != NULL) p = &(*p)->next;
103     *p = format;
104     format->next = NULL;
105 }
106
107 void av_register_output_format(AVOutputFormat *format)
108 {
109     AVOutputFormat **p;
110     p = &first_oformat;
111     while (*p != NULL) p = &(*p)->next;
112     *p = format;
113     format->next = NULL;
114 }
115
116 int av_match_ext(const char *filename, const char *extensions)
117 {
118     const char *ext, *p;
119     char ext1[32], *q;
120
121     if(!filename)
122         return 0;
123
124     ext = strrchr(filename, '.');
125     if (ext) {
126         ext++;
127         p = extensions;
128         for(;;) {
129             q = ext1;
130             while (*p != '\0' && *p != ',' && q-ext1<sizeof(ext1)-1)
131                 *q++ = *p++;
132             *q = '\0';
133             if (!av_strcasecmp(ext1, ext))
134                 return 1;
135             if (*p == '\0')
136                 break;
137             p++;
138         }
139     }
140     return 0;
141 }
142
143 static int match_format(const char *name, const char *names)
144 {
145     const char *p;
146     int len, namelen;
147
148     if (!name || !names)
149         return 0;
150
151     namelen = strlen(name);
152     while ((p = strchr(names, ','))) {
153         len = FFMAX(p - names, namelen);
154         if (!av_strncasecmp(name, names, len))
155             return 1;
156         names = p+1;
157     }
158     return !av_strcasecmp(name, names);
159 }
160
161 AVOutputFormat *av_guess_format(const char *short_name, const char *filename,
162                                 const char *mime_type)
163 {
164     AVOutputFormat *fmt = NULL, *fmt_found;
165     int score_max, score;
166
167     /* specific test for image sequences */
168 #if CONFIG_IMAGE2_MUXER
169     if (!short_name && filename &&
170         av_filename_number_test(filename) &&
171         ff_guess_image2_codec(filename) != AV_CODEC_ID_NONE) {
172         return av_guess_format("image2", NULL, NULL);
173     }
174 #endif
175     /* Find the proper file type. */
176     fmt_found = NULL;
177     score_max = 0;
178     while ((fmt = av_oformat_next(fmt))) {
179         score = 0;
180         if (fmt->name && short_name && match_format(short_name, fmt->name))
181             score += 100;
182         if (fmt->mime_type && mime_type && !strcmp(fmt->mime_type, mime_type))
183             score += 10;
184         if (filename && fmt->extensions &&
185             av_match_ext(filename, fmt->extensions)) {
186             score += 5;
187         }
188         if (score > score_max) {
189             score_max = score;
190             fmt_found = fmt;
191         }
192     }
193     return fmt_found;
194 }
195
196 enum AVCodecID av_guess_codec(AVOutputFormat *fmt, const char *short_name,
197                             const char *filename, const char *mime_type, enum AVMediaType type){
198     if(type == AVMEDIA_TYPE_VIDEO){
199         enum AVCodecID codec_id= AV_CODEC_ID_NONE;
200
201 #if CONFIG_IMAGE2_MUXER
202         if(!strcmp(fmt->name, "image2") || !strcmp(fmt->name, "image2pipe")){
203             codec_id= ff_guess_image2_codec(filename);
204         }
205 #endif
206         if(codec_id == AV_CODEC_ID_NONE)
207             codec_id= fmt->video_codec;
208         return codec_id;
209     }else if(type == AVMEDIA_TYPE_AUDIO)
210         return fmt->audio_codec;
211     else if (type == AVMEDIA_TYPE_SUBTITLE)
212         return fmt->subtitle_codec;
213     else
214         return AV_CODEC_ID_NONE;
215 }
216
217 AVInputFormat *av_find_input_format(const char *short_name)
218 {
219     AVInputFormat *fmt = NULL;
220     while ((fmt = av_iformat_next(fmt))) {
221         if (match_format(short_name, fmt->name))
222             return fmt;
223     }
224     return NULL;
225 }
226
227 int ffio_limit(AVIOContext *s, int size)
228 {
229     if(s->maxsize>=0){
230         int64_t remaining= s->maxsize - avio_tell(s);
231         if(remaining < size){
232             int64_t newsize= avio_size(s);
233             if(!s->maxsize || s->maxsize<newsize)
234                 s->maxsize= newsize - !newsize;
235             remaining= s->maxsize - avio_tell(s);
236             remaining= FFMAX(remaining, 0);
237         }
238
239         if(s->maxsize>=0 && remaining+1 < size){
240             av_log(NULL, remaining ? AV_LOG_ERROR : AV_LOG_DEBUG, "Truncating packet of size %d to %"PRId64"\n", size, remaining+1);
241             size= remaining+1;
242         }
243     }
244     return size;
245 }
246
247 int av_get_packet(AVIOContext *s, AVPacket *pkt, int size)
248 {
249     int ret;
250     int orig_size = size;
251     size= ffio_limit(s, size);
252
253     ret= av_new_packet(pkt, size);
254
255     if(ret<0)
256         return ret;
257
258     pkt->pos= avio_tell(s);
259
260     ret= avio_read(s, pkt->data, size);
261     if(ret<=0)
262         av_free_packet(pkt);
263     else
264         av_shrink_packet(pkt, ret);
265     if (pkt->size < orig_size)
266         pkt->flags |= AV_PKT_FLAG_CORRUPT;
267
268     return ret;
269 }
270
271 int av_append_packet(AVIOContext *s, AVPacket *pkt, int size)
272 {
273     int ret;
274     int old_size;
275     if (!pkt->size)
276         return av_get_packet(s, pkt, size);
277     old_size = pkt->size;
278     ret = av_grow_packet(pkt, size);
279     if (ret < 0)
280         return ret;
281     ret = avio_read(s, pkt->data + old_size, size);
282     av_shrink_packet(pkt, old_size + FFMAX(ret, 0));
283     return ret;
284 }
285
286
287 int av_filename_number_test(const char *filename)
288 {
289     char buf[1024];
290     return filename && (av_get_frame_filename(buf, sizeof(buf), filename, 1)>=0);
291 }
292
293 AVInputFormat *av_probe_input_format3(AVProbeData *pd, int is_opened, int *score_ret)
294 {
295     AVProbeData lpd = *pd;
296     AVInputFormat *fmt1 = NULL, *fmt;
297     int score, nodat = 0, score_max=0;
298     const static uint8_t zerobuffer[AVPROBE_PADDING_SIZE];
299
300     if (!lpd.buf)
301         lpd.buf = zerobuffer;
302
303     if (lpd.buf_size > 10 && ff_id3v2_match(lpd.buf, ID3v2_DEFAULT_MAGIC)) {
304         int id3len = ff_id3v2_tag_len(lpd.buf);
305         if (lpd.buf_size > id3len + 16) {
306             lpd.buf += id3len;
307             lpd.buf_size -= id3len;
308         }else
309             nodat = 1;
310     }
311
312     fmt = NULL;
313     while ((fmt1 = av_iformat_next(fmt1))) {
314         if (!is_opened == !(fmt1->flags & AVFMT_NOFILE))
315             continue;
316         score = 0;
317         if (fmt1->read_probe) {
318             score = fmt1->read_probe(&lpd);
319             if(fmt1->extensions && av_match_ext(lpd.filename, fmt1->extensions))
320                 score = FFMAX(score, nodat ? AVPROBE_SCORE_MAX/4-1 : 1);
321         } else if (fmt1->extensions) {
322             if (av_match_ext(lpd.filename, fmt1->extensions)) {
323                 score = 50;
324             }
325         }
326         if (score > score_max) {
327             score_max = score;
328             fmt = fmt1;
329         }else if (score == score_max)
330             fmt = NULL;
331     }
332     *score_ret= score_max;
333
334     return fmt;
335 }
336
337 AVInputFormat *av_probe_input_format2(AVProbeData *pd, int is_opened, int *score_max)
338 {
339     int score_ret;
340     AVInputFormat *fmt= av_probe_input_format3(pd, is_opened, &score_ret);
341     if(score_ret > *score_max){
342         *score_max= score_ret;
343         return fmt;
344     }else
345         return NULL;
346 }
347
348 AVInputFormat *av_probe_input_format(AVProbeData *pd, int is_opened){
349     int score=0;
350     return av_probe_input_format2(pd, is_opened, &score);
351 }
352
353 static int set_codec_from_probe_data(AVFormatContext *s, AVStream *st, AVProbeData *pd)
354 {
355     static const struct {
356         const char *name; enum AVCodecID id; enum AVMediaType type;
357     } fmt_id_type[] = {
358         { "aac"      , AV_CODEC_ID_AAC       , AVMEDIA_TYPE_AUDIO },
359         { "ac3"      , AV_CODEC_ID_AC3       , AVMEDIA_TYPE_AUDIO },
360         { "dts"      , AV_CODEC_ID_DTS       , AVMEDIA_TYPE_AUDIO },
361         { "eac3"     , AV_CODEC_ID_EAC3      , AVMEDIA_TYPE_AUDIO },
362         { "h264"     , AV_CODEC_ID_H264      , AVMEDIA_TYPE_VIDEO },
363         { "loas"     , AV_CODEC_ID_AAC_LATM  , AVMEDIA_TYPE_AUDIO },
364         { "m4v"      , AV_CODEC_ID_MPEG4     , AVMEDIA_TYPE_VIDEO },
365         { "mp3"      , AV_CODEC_ID_MP3       , AVMEDIA_TYPE_AUDIO },
366         { "mpegvideo", AV_CODEC_ID_MPEG2VIDEO, AVMEDIA_TYPE_VIDEO },
367         { 0 }
368     };
369     int score;
370     AVInputFormat *fmt = av_probe_input_format3(pd, 1, &score);
371
372     if (fmt && st->request_probe <= score) {
373         int i;
374         av_log(s, AV_LOG_DEBUG, "Probe with size=%d, packets=%d detected %s with score=%d\n",
375                pd->buf_size, MAX_PROBE_PACKETS - st->probe_packets, fmt->name, score);
376         for (i = 0; fmt_id_type[i].name; i++) {
377             if (!strcmp(fmt->name, fmt_id_type[i].name)) {
378                 st->codec->codec_id   = fmt_id_type[i].id;
379                 st->codec->codec_type = fmt_id_type[i].type;
380                 break;
381             }
382         }
383     }
384     return score;
385 }
386
387 /************************************************************/
388 /* input media file */
389
390 int av_demuxer_open(AVFormatContext *ic){
391     int err;
392
393     if (ic->iformat->read_header) {
394         err = ic->iformat->read_header(ic);
395         if (err < 0)
396             return err;
397     }
398
399     if (ic->pb && !ic->data_offset)
400         ic->data_offset = avio_tell(ic->pb);
401
402     return 0;
403 }
404
405
406 /** size of probe buffer, for guessing file type from file contents */
407 #define PROBE_BUF_MIN 2048
408 #define PROBE_BUF_MAX (1<<20)
409
410 int av_probe_input_buffer(AVIOContext *pb, AVInputFormat **fmt,
411                           const char *filename, void *logctx,
412                           unsigned int offset, unsigned int max_probe_size)
413 {
414     AVProbeData pd = { filename ? filename : "", NULL, -offset };
415     unsigned char *buf = NULL;
416     int ret = 0, probe_size;
417
418     if (!max_probe_size) {
419         max_probe_size = PROBE_BUF_MAX;
420     } else if (max_probe_size > PROBE_BUF_MAX) {
421         max_probe_size = PROBE_BUF_MAX;
422     } else if (max_probe_size < PROBE_BUF_MIN) {
423         return AVERROR(EINVAL);
424     }
425
426     if (offset >= max_probe_size) {
427         return AVERROR(EINVAL);
428     }
429
430     for(probe_size= PROBE_BUF_MIN; probe_size<=max_probe_size && !*fmt;
431         probe_size = FFMIN(probe_size<<1, FFMAX(max_probe_size, probe_size+1))) {
432         int score = probe_size < max_probe_size ? AVPROBE_SCORE_RETRY : 0;
433         int buf_offset = (probe_size == PROBE_BUF_MIN) ? 0 : probe_size>>1;
434         void *buftmp;
435
436         if (probe_size < offset) {
437             continue;
438         }
439
440         /* read probe data */
441         buftmp = av_realloc(buf, probe_size + AVPROBE_PADDING_SIZE);
442         if(!buftmp){
443             av_free(buf);
444             return AVERROR(ENOMEM);
445         }
446         buf=buftmp;
447         if ((ret = avio_read(pb, buf + buf_offset, probe_size - buf_offset)) < 0) {
448             /* fail if error was not end of file, otherwise, lower score */
449             if (ret != AVERROR_EOF) {
450                 av_free(buf);
451                 return ret;
452             }
453             score = 0;
454             ret = 0;            /* error was end of file, nothing read */
455         }
456         pd.buf_size += ret;
457         pd.buf = &buf[offset];
458
459         memset(pd.buf + pd.buf_size, 0, AVPROBE_PADDING_SIZE);
460
461         /* guess file format */
462         *fmt = av_probe_input_format2(&pd, 1, &score);
463         if(*fmt){
464             if(score <= AVPROBE_SCORE_RETRY){ //this can only be true in the last iteration
465                 av_log(logctx, AV_LOG_WARNING, "Format %s detected only with low score of %d, misdetection possible!\n", (*fmt)->name, score);
466             }else
467                 av_log(logctx, AV_LOG_DEBUG, "Format %s probed with size=%d and score=%d\n", (*fmt)->name, probe_size, score);
468         }
469     }
470
471     if (!*fmt) {
472         av_free(buf);
473         return AVERROR_INVALIDDATA;
474     }
475
476     /* rewind. reuse probe buffer to avoid seeking */
477     ret = ffio_rewind_with_probe_data(pb, &buf, pd.buf_size);
478
479     return ret;
480 }
481
482 /* open input file and probe the format if necessary */
483 static int init_input(AVFormatContext *s, const char *filename, AVDictionary **options)
484 {
485     int ret;
486     AVProbeData pd = {filename, NULL, 0};
487     int score = AVPROBE_SCORE_RETRY;
488
489     if (s->pb) {
490         s->flags |= AVFMT_FLAG_CUSTOM_IO;
491         if (!s->iformat)
492             return av_probe_input_buffer(s->pb, &s->iformat, filename, s, 0, s->probesize);
493         else if (s->iformat->flags & AVFMT_NOFILE)
494             av_log(s, AV_LOG_WARNING, "Custom AVIOContext makes no sense and "
495                                       "will be ignored with AVFMT_NOFILE format.\n");
496         return 0;
497     }
498
499     if ( (s->iformat && s->iformat->flags & AVFMT_NOFILE) ||
500         (!s->iformat && (s->iformat = av_probe_input_format2(&pd, 0, &score))))
501         return 0;
502
503     if ((ret = avio_open2(&s->pb, filename, AVIO_FLAG_READ | s->avio_flags,
504                           &s->interrupt_callback, options)) < 0)
505         return ret;
506     if (s->iformat)
507         return 0;
508     return av_probe_input_buffer(s->pb, &s->iformat, filename, s, 0, s->probesize);
509 }
510
511 static AVPacket *add_to_pktbuf(AVPacketList **packet_buffer, AVPacket *pkt,
512                                AVPacketList **plast_pktl){
513     AVPacketList *pktl = av_mallocz(sizeof(AVPacketList));
514     if (!pktl)
515         return NULL;
516
517     if (*packet_buffer)
518         (*plast_pktl)->next = pktl;
519     else
520         *packet_buffer = pktl;
521
522     /* add the packet in the buffered packet list */
523     *plast_pktl = pktl;
524     pktl->pkt= *pkt;
525     return &pktl->pkt;
526 }
527
528 void avformat_queue_attached_pictures(AVFormatContext *s)
529 {
530     int i;
531     for (i = 0; i < s->nb_streams; i++)
532         if (s->streams[i]->disposition & AV_DISPOSITION_ATTACHED_PIC &&
533             s->streams[i]->discard < AVDISCARD_ALL) {
534             AVPacket copy = s->streams[i]->attached_pic;
535             copy.destruct = NULL;
536             add_to_pktbuf(&s->raw_packet_buffer, &copy, &s->raw_packet_buffer_end);
537         }
538 }
539
540 int avformat_open_input(AVFormatContext **ps, const char *filename, AVInputFormat *fmt, AVDictionary **options)
541 {
542     AVFormatContext *s = *ps;
543     int ret = 0;
544     AVDictionary *tmp = NULL;
545     ID3v2ExtraMeta *id3v2_extra_meta = NULL;
546
547     if (!s && !(s = avformat_alloc_context()))
548         return AVERROR(ENOMEM);
549     if (!s->av_class){
550         av_log(NULL, AV_LOG_ERROR, "Input context has not been properly allocated by avformat_alloc_context() and is not NULL either\n");
551         return AVERROR(EINVAL);
552     }
553     if (fmt)
554         s->iformat = fmt;
555
556     if (options)
557         av_dict_copy(&tmp, *options, 0);
558
559     if ((ret = av_opt_set_dict(s, &tmp)) < 0)
560         goto fail;
561
562     if ((ret = init_input(s, filename, &tmp)) < 0)
563         goto fail;
564     avio_skip(s->pb, s->skip_initial_bytes);
565
566     /* check filename in case an image number is expected */
567     if (s->iformat->flags & AVFMT_NEEDNUMBER) {
568         if (!av_filename_number_test(filename)) {
569             ret = AVERROR(EINVAL);
570             goto fail;
571         }
572     }
573
574     s->duration = s->start_time = AV_NOPTS_VALUE;
575     av_strlcpy(s->filename, filename ? filename : "", sizeof(s->filename));
576
577     /* allocate private data */
578     if (s->iformat->priv_data_size > 0) {
579         if (!(s->priv_data = av_mallocz(s->iformat->priv_data_size))) {
580             ret = AVERROR(ENOMEM);
581             goto fail;
582         }
583         if (s->iformat->priv_class) {
584             *(const AVClass**)s->priv_data = s->iformat->priv_class;
585             av_opt_set_defaults(s->priv_data);
586             if ((ret = av_opt_set_dict(s->priv_data, &tmp)) < 0)
587                 goto fail;
588         }
589     }
590
591     /* e.g. AVFMT_NOFILE formats will not have a AVIOContext */
592     if (s->pb)
593         ff_id3v2_read(s, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta);
594
595     if (!(s->flags&AVFMT_FLAG_PRIV_OPT) && s->iformat->read_header)
596         if ((ret = s->iformat->read_header(s)) < 0)
597             goto fail;
598
599     if (id3v2_extra_meta) {
600         if (!strcmp(s->iformat->name, "mp3")) {
601             if((ret = ff_id3v2_parse_apic(s, &id3v2_extra_meta)) < 0)
602                 goto fail;
603         } else
604             av_log(s, AV_LOG_DEBUG, "demuxer does not support additional id3 data, skipping\n");
605     }
606     ff_id3v2_free_extra_meta(&id3v2_extra_meta);
607
608     avformat_queue_attached_pictures(s);
609
610     if (!(s->flags&AVFMT_FLAG_PRIV_OPT) && s->pb && !s->data_offset)
611         s->data_offset = avio_tell(s->pb);
612
613     s->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
614
615     if (options) {
616         av_dict_free(options);
617         *options = tmp;
618     }
619     *ps = s;
620     return 0;
621
622 fail:
623     ff_id3v2_free_extra_meta(&id3v2_extra_meta);
624     av_dict_free(&tmp);
625     if (s->pb && !(s->flags & AVFMT_FLAG_CUSTOM_IO))
626         avio_close(s->pb);
627     avformat_free_context(s);
628     *ps = NULL;
629     return ret;
630 }
631
632 /*******************************************************/
633
634 static void probe_codec(AVFormatContext *s, AVStream *st, const AVPacket *pkt)
635 {
636     if(st->request_probe>0){
637         AVProbeData *pd = &st->probe_data;
638         int end;
639         av_log(s, AV_LOG_DEBUG, "probing stream %d pp:%d\n", st->index, st->probe_packets);
640         --st->probe_packets;
641
642         if (pkt) {
643             uint8_t *new_buf = av_realloc(pd->buf, pd->buf_size+pkt->size+AVPROBE_PADDING_SIZE);
644             if(!new_buf)
645                 goto no_packet;
646             pd->buf = new_buf;
647             memcpy(pd->buf+pd->buf_size, pkt->data, pkt->size);
648             pd->buf_size += pkt->size;
649             memset(pd->buf+pd->buf_size, 0, AVPROBE_PADDING_SIZE);
650         } else {
651 no_packet:
652             st->probe_packets = 0;
653             if (!pd->buf_size) {
654                 av_log(s, AV_LOG_WARNING, "nothing to probe for stream %d\n",
655                        st->index);
656             }
657         }
658
659         end=    s->raw_packet_buffer_remaining_size <= 0
660                 || st->probe_packets<=0;
661
662         if(end || av_log2(pd->buf_size) != av_log2(pd->buf_size - pkt->size)){
663             int score= set_codec_from_probe_data(s, st, pd);
664             if(    (st->codec->codec_id != AV_CODEC_ID_NONE && score > AVPROBE_SCORE_RETRY)
665                 || end){
666                 pd->buf_size=0;
667                 av_freep(&pd->buf);
668                 st->request_probe= -1;
669                 if(st->codec->codec_id != AV_CODEC_ID_NONE){
670                     av_log(s, AV_LOG_DEBUG, "probed stream %d\n", st->index);
671                 }else
672                     av_log(s, AV_LOG_WARNING, "probed stream %d failed\n", st->index);
673             }
674         }
675     }
676 }
677
678 static void force_codec_ids(AVFormatContext *s, AVStream *st)
679 {
680     switch(st->codec->codec_type){
681     case AVMEDIA_TYPE_VIDEO:
682         if(s->video_codec_id)   st->codec->codec_id= s->video_codec_id;
683         break;
684     case AVMEDIA_TYPE_AUDIO:
685         if(s->audio_codec_id)   st->codec->codec_id= s->audio_codec_id;
686         break;
687     case AVMEDIA_TYPE_SUBTITLE:
688         if(s->subtitle_codec_id)st->codec->codec_id= s->subtitle_codec_id;
689         break;
690     }
691 }
692
693 int ff_read_packet(AVFormatContext *s, AVPacket *pkt)
694 {
695     int ret, i;
696     AVStream *st;
697
698     for(;;){
699         AVPacketList *pktl = s->raw_packet_buffer;
700
701         if (pktl) {
702             *pkt = pktl->pkt;
703             st = s->streams[pkt->stream_index];
704             if(st->request_probe <= 0){
705                 s->raw_packet_buffer = pktl->next;
706                 s->raw_packet_buffer_remaining_size += pkt->size;
707                 av_free(pktl);
708                 return 0;
709             }
710         }
711
712         pkt->data = NULL;
713         pkt->size = 0;
714         av_init_packet(pkt);
715         ret= s->iformat->read_packet(s, pkt);
716         if (ret < 0) {
717             if (!pktl || ret == AVERROR(EAGAIN))
718                 return ret;
719             for (i = 0; i < s->nb_streams; i++) {
720                 st = s->streams[i];
721                 if (st->probe_packets) {
722                     probe_codec(s, st, NULL);
723                 }
724                 av_assert0(st->request_probe <= 0);
725             }
726             continue;
727         }
728
729         if ((s->flags & AVFMT_FLAG_DISCARD_CORRUPT) &&
730             (pkt->flags & AV_PKT_FLAG_CORRUPT)) {
731             av_log(s, AV_LOG_WARNING,
732                    "Dropped corrupted packet (stream = %d)\n",
733                    pkt->stream_index);
734             av_free_packet(pkt);
735             continue;
736         }
737
738         if(!(s->flags & AVFMT_FLAG_KEEP_SIDE_DATA))
739             av_packet_merge_side_data(pkt);
740
741         if(pkt->stream_index >= (unsigned)s->nb_streams){
742             av_log(s, AV_LOG_ERROR, "Invalid stream index %d\n", pkt->stream_index);
743             continue;
744         }
745
746         st= s->streams[pkt->stream_index];
747
748         force_codec_ids(s, st);
749
750         /* TODO: audio: time filter; video: frame reordering (pts != dts) */
751         if (s->use_wallclock_as_timestamps)
752             pkt->dts = pkt->pts = av_rescale_q(av_gettime(), AV_TIME_BASE_Q, st->time_base);
753
754         if(!pktl && st->request_probe <= 0)
755             return ret;
756
757         add_to_pktbuf(&s->raw_packet_buffer, pkt, &s->raw_packet_buffer_end);
758         s->raw_packet_buffer_remaining_size -= pkt->size;
759
760         probe_codec(s, st, pkt);
761     }
762 }
763
764 #if FF_API_READ_PACKET
765 int av_read_packet(AVFormatContext *s, AVPacket *pkt)
766 {
767     return ff_read_packet(s, pkt);
768 }
769 #endif
770
771
772 /**********************************************************/
773
774 static int determinable_frame_size(AVCodecContext *avctx)
775 {
776     if (/*avctx->codec_id == AV_CODEC_ID_AAC ||*/
777         avctx->codec_id == AV_CODEC_ID_MP1 ||
778         avctx->codec_id == AV_CODEC_ID_MP2 ||
779         avctx->codec_id == AV_CODEC_ID_MP3/* ||
780         avctx->codec_id == AV_CODEC_ID_CELT*/)
781         return 1;
782     return 0;
783 }
784
785 /**
786  * Get the number of samples of an audio frame. Return -1 on error.
787  */
788 int ff_get_audio_frame_size(AVCodecContext *enc, int size, int mux)
789 {
790     int frame_size;
791
792     /* give frame_size priority if demuxing */
793     if (!mux && enc->frame_size > 1)
794         return enc->frame_size;
795
796     if ((frame_size = av_get_audio_frame_duration(enc, size)) > 0)
797         return frame_size;
798
799     /* fallback to using frame_size if muxing */
800     if (enc->frame_size > 1)
801         return enc->frame_size;
802
803     return -1;
804 }
805
806
807 /**
808  * Return the frame duration in seconds. Return 0 if not available.
809  */
810 void ff_compute_frame_duration(int *pnum, int *pden, AVStream *st,
811                                AVCodecParserContext *pc, AVPacket *pkt)
812 {
813     int frame_size;
814
815     *pnum = 0;
816     *pden = 0;
817     switch(st->codec->codec_type) {
818     case AVMEDIA_TYPE_VIDEO:
819         if (st->r_frame_rate.num && !pc) {
820             *pnum = st->r_frame_rate.den;
821             *pden = st->r_frame_rate.num;
822         } else if(st->time_base.num*1000LL > st->time_base.den) {
823             *pnum = st->time_base.num;
824             *pden = st->time_base.den;
825         }else if(st->codec->time_base.num*1000LL > st->codec->time_base.den){
826             *pnum = st->codec->time_base.num;
827             *pden = st->codec->time_base.den;
828             if (pc && pc->repeat_pict) {
829                 *pnum = (*pnum) * (1 + pc->repeat_pict);
830             }
831             //If this codec can be interlaced or progressive then we need a parser to compute duration of a packet
832             //Thus if we have no parser in such case leave duration undefined.
833             if(st->codec->ticks_per_frame>1 && !pc){
834                 *pnum = *pden = 0;
835             }
836         }
837         break;
838     case AVMEDIA_TYPE_AUDIO:
839         frame_size = ff_get_audio_frame_size(st->codec, pkt->size, 0);
840         if (frame_size <= 0 || st->codec->sample_rate <= 0)
841             break;
842         *pnum = frame_size;
843         *pden = st->codec->sample_rate;
844         break;
845     default:
846         break;
847     }
848 }
849
850 static int is_intra_only(AVCodecContext *enc){
851     const AVCodecDescriptor *desc;
852
853     if(enc->codec_type != AVMEDIA_TYPE_VIDEO)
854         return 1;
855
856     desc = av_codec_get_codec_descriptor(enc);
857     if (!desc) {
858         desc = avcodec_descriptor_get(enc->codec_id);
859         av_codec_set_codec_descriptor(enc, desc);
860     }
861     if (desc)
862         return !!(desc->props & AV_CODEC_PROP_INTRA_ONLY);
863     return 0;
864 }
865
866 static int has_decode_delay_been_guessed(AVStream *st)
867 {
868     if(st->codec->codec_id != AV_CODEC_ID_H264) return 1;
869     if(!st->info) // if we have left find_stream_info then nb_decoded_frames wont increase anymore for stream copy
870         return 1;
871 #if CONFIG_H264_DECODER
872     if(st->codec->has_b_frames &&
873        avpriv_h264_has_num_reorder_frames(st->codec) == st->codec->has_b_frames)
874         return 1;
875 #endif
876     if(st->codec->has_b_frames<3)
877         return st->nb_decoded_frames >= 7;
878     else if(st->codec->has_b_frames<4)
879         return st->nb_decoded_frames >= 18;
880     else
881         return st->nb_decoded_frames >= 20;
882 }
883
884 static AVPacketList *get_next_pkt(AVFormatContext *s, AVStream *st, AVPacketList *pktl)
885 {
886     if (pktl->next)
887         return pktl->next;
888     if (pktl == s->parse_queue_end)
889         return s->packet_buffer;
890     return NULL;
891 }
892
893 static void update_initial_timestamps(AVFormatContext *s, int stream_index,
894                                       int64_t dts, int64_t pts)
895 {
896     AVStream *st= s->streams[stream_index];
897     AVPacketList *pktl= s->parse_queue ? s->parse_queue : s->packet_buffer;
898     int64_t pts_buffer[MAX_REORDER_DELAY+1];
899     int64_t shift;
900     int i, delay;
901
902     if(st->first_dts != AV_NOPTS_VALUE || dts == AV_NOPTS_VALUE || st->cur_dts == AV_NOPTS_VALUE || is_relative(dts))
903         return;
904
905     delay = st->codec->has_b_frames;
906     st->first_dts= dts - (st->cur_dts - RELATIVE_TS_BASE);
907     st->cur_dts= dts;
908     shift = st->first_dts - RELATIVE_TS_BASE;
909
910     for (i=0; i<MAX_REORDER_DELAY+1; i++)
911         pts_buffer[i] = AV_NOPTS_VALUE;
912
913     if (is_relative(pts))
914         pts += shift;
915
916     for(; pktl; pktl= get_next_pkt(s, st, pktl)){
917         if(pktl->pkt.stream_index != stream_index)
918             continue;
919         if(is_relative(pktl->pkt.pts))
920             pktl->pkt.pts += shift;
921
922         if(is_relative(pktl->pkt.dts))
923             pktl->pkt.dts += shift;
924
925         if(st->start_time == AV_NOPTS_VALUE && pktl->pkt.pts != AV_NOPTS_VALUE)
926             st->start_time= pktl->pkt.pts;
927
928         if(pktl->pkt.pts != AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY && has_decode_delay_been_guessed(st)){
929             pts_buffer[0]= pktl->pkt.pts;
930             for(i=0; i<delay && pts_buffer[i] > pts_buffer[i+1]; i++)
931                 FFSWAP(int64_t, pts_buffer[i], pts_buffer[i+1]);
932             if(pktl->pkt.dts == AV_NOPTS_VALUE)
933                 pktl->pkt.dts= pts_buffer[0];
934         }
935     }
936     if (st->start_time == AV_NOPTS_VALUE)
937         st->start_time = pts;
938 }
939
940 static void update_initial_durations(AVFormatContext *s, AVStream *st,
941                                      int stream_index, int duration)
942 {
943     AVPacketList *pktl= s->parse_queue ? s->parse_queue : s->packet_buffer;
944     int64_t cur_dts= RELATIVE_TS_BASE;
945
946     if(st->first_dts != AV_NOPTS_VALUE){
947         cur_dts= st->first_dts;
948         for(; pktl; pktl= get_next_pkt(s, st, pktl)){
949             if(pktl->pkt.stream_index == stream_index){
950                 if(pktl->pkt.pts != pktl->pkt.dts || pktl->pkt.dts != AV_NOPTS_VALUE || pktl->pkt.duration)
951                     break;
952                 cur_dts -= duration;
953             }
954         }
955         if(pktl && pktl->pkt.dts != st->first_dts) {
956             av_log(s, AV_LOG_DEBUG, "first_dts %s not matching first dts %s in que\n", av_ts2str(st->first_dts), av_ts2str(pktl->pkt.dts));
957             return;
958         }
959         if(!pktl) {
960             av_log(s, AV_LOG_DEBUG, "first_dts %s but no packet with dts in ques\n", av_ts2str(st->first_dts));
961             return;
962         }
963         pktl= s->parse_queue ? s->parse_queue : s->packet_buffer;
964         st->first_dts = cur_dts;
965     }else if(st->cur_dts != RELATIVE_TS_BASE)
966         return;
967
968     for(; pktl; pktl= get_next_pkt(s, st, pktl)){
969         if(pktl->pkt.stream_index != stream_index)
970             continue;
971         if(pktl->pkt.pts == pktl->pkt.dts && (pktl->pkt.dts == AV_NOPTS_VALUE || pktl->pkt.dts == st->first_dts)
972            && !pktl->pkt.duration){
973             pktl->pkt.dts= cur_dts;
974             if(!st->codec->has_b_frames)
975                 pktl->pkt.pts= cur_dts;
976 //            if (st->codec->codec_type != AVMEDIA_TYPE_AUDIO)
977                 pktl->pkt.duration = duration;
978         }else
979             break;
980         cur_dts = pktl->pkt.dts + pktl->pkt.duration;
981     }
982     if(!pktl)
983         st->cur_dts= cur_dts;
984 }
985
986 static void compute_pkt_fields(AVFormatContext *s, AVStream *st,
987                                AVCodecParserContext *pc, AVPacket *pkt)
988 {
989     int num, den, presentation_delayed, delay, i;
990     int64_t offset;
991
992     if (s->flags & AVFMT_FLAG_NOFILLIN)
993         return;
994
995     if((s->flags & AVFMT_FLAG_IGNDTS) && pkt->pts != AV_NOPTS_VALUE)
996         pkt->dts= AV_NOPTS_VALUE;
997
998     if (st->codec->codec_id != AV_CODEC_ID_H264 && pc && pc->pict_type == AV_PICTURE_TYPE_B)
999         //FIXME Set low_delay = 0 when has_b_frames = 1
1000         st->codec->has_b_frames = 1;
1001
1002     /* do we have a video B-frame ? */
1003     delay= st->codec->has_b_frames;
1004     presentation_delayed = 0;
1005
1006     /* XXX: need has_b_frame, but cannot get it if the codec is
1007         not initialized */
1008     if (delay &&
1009         pc && pc->pict_type != AV_PICTURE_TYPE_B)
1010         presentation_delayed = 1;
1011
1012     if(pkt->pts != AV_NOPTS_VALUE && pkt->dts != AV_NOPTS_VALUE && st->pts_wrap_bits<63 && pkt->dts - (1LL<<(st->pts_wrap_bits-1)) > pkt->pts){
1013         if(is_relative(st->cur_dts) || pkt->dts - (1LL<<(st->pts_wrap_bits-1)) > st->cur_dts) {
1014             pkt->dts -= 1LL<<st->pts_wrap_bits;
1015         } else
1016             pkt->pts += 1LL<<st->pts_wrap_bits;
1017     }
1018
1019     // some mpeg2 in mpeg-ps lack dts (issue171 / input_file.mpg)
1020     // we take the conservative approach and discard both
1021     // Note, if this is misbehaving for a H.264 file then possibly presentation_delayed is not set correctly.
1022     if(delay==1 && pkt->dts == pkt->pts && pkt->dts != AV_NOPTS_VALUE && presentation_delayed){
1023         av_log(s, AV_LOG_DEBUG, "invalid dts/pts combination %"PRIi64"\n", pkt->dts);
1024         if(strcmp(s->iformat->name, "mov,mp4,m4a,3gp,3g2,mj2")) // otherwise we discard correct timestamps for vc1-wmapro.ism
1025             pkt->dts= AV_NOPTS_VALUE;
1026     }
1027
1028     if (pkt->duration == 0) {
1029         ff_compute_frame_duration(&num, &den, st, pc, pkt);
1030         if (den && num) {
1031             pkt->duration = av_rescale_rnd(1, num * (int64_t)st->time_base.den, den * (int64_t)st->time_base.num, AV_ROUND_DOWN);
1032         }
1033     }
1034     if(pkt->duration != 0 && (s->packet_buffer || s->parse_queue))
1035         update_initial_durations(s, st, pkt->stream_index, pkt->duration);
1036
1037     /* correct timestamps with byte offset if demuxers only have timestamps
1038        on packet boundaries */
1039     if(pc && st->need_parsing == AVSTREAM_PARSE_TIMESTAMPS && pkt->size){
1040         /* this will estimate bitrate based on this frame's duration and size */
1041         offset = av_rescale(pc->offset, pkt->duration, pkt->size);
1042         if(pkt->pts != AV_NOPTS_VALUE)
1043             pkt->pts += offset;
1044         if(pkt->dts != AV_NOPTS_VALUE)
1045             pkt->dts += offset;
1046     }
1047
1048     if (pc && pc->dts_sync_point >= 0) {
1049         // we have synchronization info from the parser
1050         int64_t den = st->codec->time_base.den * (int64_t) st->time_base.num;
1051         if (den > 0) {
1052             int64_t num = st->codec->time_base.num * (int64_t) st->time_base.den;
1053             if (pkt->dts != AV_NOPTS_VALUE) {
1054                 // got DTS from the stream, update reference timestamp
1055                 st->reference_dts = pkt->dts - pc->dts_ref_dts_delta * num / den;
1056                 pkt->pts = pkt->dts + pc->pts_dts_delta * num / den;
1057             } else if (st->reference_dts != AV_NOPTS_VALUE) {
1058                 // compute DTS based on reference timestamp
1059                 pkt->dts = st->reference_dts + pc->dts_ref_dts_delta * num / den;
1060                 pkt->pts = pkt->dts + pc->pts_dts_delta * num / den;
1061             }
1062             if (pc->dts_sync_point > 0)
1063                 st->reference_dts = pkt->dts; // new reference
1064         }
1065     }
1066
1067     /* This may be redundant, but it should not hurt. */
1068     if(pkt->dts != AV_NOPTS_VALUE && pkt->pts != AV_NOPTS_VALUE && pkt->pts > pkt->dts)
1069         presentation_delayed = 1;
1070
1071     av_dlog(NULL, "IN delayed:%d pts:%s, dts:%s cur_dts:%s st:%d pc:%p duration:%d\n",
1072            presentation_delayed, av_ts2str(pkt->pts), av_ts2str(pkt->dts), av_ts2str(st->cur_dts), pkt->stream_index, pc, pkt->duration);
1073     /* interpolate PTS and DTS if they are not present */
1074     //We skip H264 currently because delay and has_b_frames are not reliably set
1075     if((delay==0 || (delay==1 && pc)) && st->codec->codec_id != AV_CODEC_ID_H264){
1076         if (presentation_delayed) {
1077             /* DTS = decompression timestamp */
1078             /* PTS = presentation timestamp */
1079             if (pkt->dts == AV_NOPTS_VALUE)
1080                 pkt->dts = st->last_IP_pts;
1081             update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts);
1082             if (pkt->dts == AV_NOPTS_VALUE)
1083                 pkt->dts = st->cur_dts;
1084
1085             /* this is tricky: the dts must be incremented by the duration
1086             of the frame we are displaying, i.e. the last I- or P-frame */
1087             if (st->last_IP_duration == 0)
1088                 st->last_IP_duration = pkt->duration;
1089             if(pkt->dts != AV_NOPTS_VALUE)
1090                 st->cur_dts = pkt->dts + st->last_IP_duration;
1091             st->last_IP_duration  = pkt->duration;
1092             st->last_IP_pts= pkt->pts;
1093             /* cannot compute PTS if not present (we can compute it only
1094             by knowing the future */
1095         } else if (pkt->pts != AV_NOPTS_VALUE ||
1096                    pkt->dts != AV_NOPTS_VALUE ||
1097                    pkt->duration                ) {
1098             int duration = pkt->duration;
1099
1100             if(st->cur_dts != AV_NOPTS_VALUE && pkt->pts != AV_NOPTS_VALUE && duration){
1101                 int64_t old_diff= FFABS(st->cur_dts - duration - pkt->pts);
1102                 int64_t new_diff= FFABS(st->cur_dts - pkt->pts);
1103                 if(   old_diff < new_diff && old_diff < (duration>>3)
1104                    && st->codec->codec_type == AVMEDIA_TYPE_VIDEO
1105                    && (!strcmp(s->iformat->name, "mpeg") ||
1106                        !strcmp(s->iformat->name, "mpegts"))){
1107                     pkt->pts += duration;
1108                     av_log(s, AV_LOG_WARNING, "Adjusting PTS forward\n");
1109 //                    av_log(NULL, AV_LOG_DEBUG, "id:%d old:%"PRId64" new:%"PRId64" dur:%d cur:%s size:%d\n",
1110 //                           pkt->stream_index, old_diff, new_diff, pkt->duration, av_ts2str(st->cur_dts), pkt->size);
1111                 }
1112             }
1113
1114             /* presentation is not delayed : PTS and DTS are the same */
1115             if (pkt->pts == AV_NOPTS_VALUE)
1116                 pkt->pts = pkt->dts;
1117             update_initial_timestamps(s, pkt->stream_index, pkt->pts,
1118                                       pkt->pts);
1119             if (pkt->pts == AV_NOPTS_VALUE)
1120                 pkt->pts = st->cur_dts;
1121             pkt->dts = pkt->pts;
1122             if (pkt->pts != AV_NOPTS_VALUE)
1123                 st->cur_dts = pkt->pts + duration;
1124         }
1125     }
1126
1127     if(pkt->pts != AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY && has_decode_delay_been_guessed(st)){
1128         st->pts_buffer[0]= pkt->pts;
1129         for(i=0; i<delay && st->pts_buffer[i] > st->pts_buffer[i+1]; i++)
1130             FFSWAP(int64_t, st->pts_buffer[i], st->pts_buffer[i+1]);
1131         if(pkt->dts == AV_NOPTS_VALUE)
1132             pkt->dts= st->pts_buffer[0];
1133     }
1134     if(st->codec->codec_id == AV_CODEC_ID_H264){ // we skipped it above so we try here
1135         update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts); // this should happen on the first packet
1136     }
1137     if(pkt->dts > st->cur_dts)
1138         st->cur_dts = pkt->dts;
1139
1140     av_dlog(NULL, "OUTdelayed:%d/%d pts:%s, dts:%s cur_dts:%s\n",
1141             presentation_delayed, delay, av_ts2str(pkt->pts), av_ts2str(pkt->dts), av_ts2str(st->cur_dts));
1142
1143     /* update flags */
1144     if (is_intra_only(st->codec))
1145         pkt->flags |= AV_PKT_FLAG_KEY;
1146     if (pc)
1147         pkt->convergence_duration = pc->convergence_duration;
1148 }
1149
1150 static void free_packet_buffer(AVPacketList **pkt_buf, AVPacketList **pkt_buf_end)
1151 {
1152     while (*pkt_buf) {
1153         AVPacketList *pktl = *pkt_buf;
1154         *pkt_buf = pktl->next;
1155         av_free_packet(&pktl->pkt);
1156         av_freep(&pktl);
1157     }
1158     *pkt_buf_end = NULL;
1159 }
1160
1161 /**
1162  * Parse a packet, add all split parts to parse_queue
1163  *
1164  * @param pkt packet to parse, NULL when flushing the parser at end of stream
1165  */
1166 static int parse_packet(AVFormatContext *s, AVPacket *pkt, int stream_index)
1167 {
1168     AVPacket out_pkt = { 0 }, flush_pkt = { 0 };
1169     AVStream     *st = s->streams[stream_index];
1170     uint8_t    *data = pkt ? pkt->data : NULL;
1171     int         size = pkt ? pkt->size : 0;
1172     int ret = 0, got_output = 0;
1173
1174     if (!pkt) {
1175         av_init_packet(&flush_pkt);
1176         pkt = &flush_pkt;
1177         got_output = 1;
1178     } else if (!size && st->parser->flags & PARSER_FLAG_COMPLETE_FRAMES) {
1179         // preserve 0-size sync packets
1180         compute_pkt_fields(s, st, st->parser, pkt);
1181     }
1182
1183     while (size > 0 || (pkt == &flush_pkt && got_output)) {
1184         int len;
1185
1186         av_init_packet(&out_pkt);
1187         len = av_parser_parse2(st->parser,  st->codec,
1188                                &out_pkt.data, &out_pkt.size, data, size,
1189                                pkt->pts, pkt->dts, pkt->pos);
1190
1191         pkt->pts = pkt->dts = AV_NOPTS_VALUE;
1192         pkt->pos = -1;
1193         /* increment read pointer */
1194         data += len;
1195         size -= len;
1196
1197         got_output = !!out_pkt.size;
1198
1199         if (!out_pkt.size)
1200             continue;
1201
1202         /* set the duration */
1203         out_pkt.duration = 0;
1204         if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
1205             if (st->codec->sample_rate > 0) {
1206                 out_pkt.duration = av_rescale_q_rnd(st->parser->duration,
1207                                                     (AVRational){ 1, st->codec->sample_rate },
1208                                                     st->time_base,
1209                                                     AV_ROUND_DOWN);
1210             }
1211         } else if (st->codec->time_base.num != 0 &&
1212                    st->codec->time_base.den != 0) {
1213             out_pkt.duration = av_rescale_q_rnd(st->parser->duration,
1214                                                 st->codec->time_base,
1215                                                 st->time_base,
1216                                                 AV_ROUND_DOWN);
1217         }
1218
1219         out_pkt.stream_index = st->index;
1220         out_pkt.pts = st->parser->pts;
1221         out_pkt.dts = st->parser->dts;
1222         out_pkt.pos = st->parser->pos;
1223
1224         if(st->need_parsing == AVSTREAM_PARSE_FULL_RAW)
1225             out_pkt.pos = st->parser->frame_offset;
1226
1227         if (st->parser->key_frame == 1 ||
1228             (st->parser->key_frame == -1 &&
1229              st->parser->pict_type == AV_PICTURE_TYPE_I))
1230             out_pkt.flags |= AV_PKT_FLAG_KEY;
1231
1232         if(st->parser->key_frame == -1 && st->parser->pict_type==AV_PICTURE_TYPE_NONE && (pkt->flags&AV_PKT_FLAG_KEY))
1233             out_pkt.flags |= AV_PKT_FLAG_KEY;
1234
1235         compute_pkt_fields(s, st, st->parser, &out_pkt);
1236
1237         if (out_pkt.data == pkt->data && out_pkt.size == pkt->size) {
1238             out_pkt.destruct = pkt->destruct;
1239             pkt->destruct = NULL;
1240         }
1241         if ((ret = av_dup_packet(&out_pkt)) < 0)
1242             goto fail;
1243
1244         if (!add_to_pktbuf(&s->parse_queue, &out_pkt, &s->parse_queue_end)) {
1245             av_free_packet(&out_pkt);
1246             ret = AVERROR(ENOMEM);
1247             goto fail;
1248         }
1249     }
1250
1251
1252     /* end of the stream => close and free the parser */
1253     if (pkt == &flush_pkt) {
1254         av_parser_close(st->parser);
1255         st->parser = NULL;
1256     }
1257
1258 fail:
1259     av_free_packet(pkt);
1260     return ret;
1261 }
1262
1263 static int read_from_packet_buffer(AVPacketList **pkt_buffer,
1264                                    AVPacketList **pkt_buffer_end,
1265                                    AVPacket      *pkt)
1266 {
1267     AVPacketList *pktl;
1268     av_assert0(*pkt_buffer);
1269     pktl = *pkt_buffer;
1270     *pkt = pktl->pkt;
1271     *pkt_buffer = pktl->next;
1272     if (!pktl->next)
1273         *pkt_buffer_end = NULL;
1274     av_freep(&pktl);
1275     return 0;
1276 }
1277
1278 static int read_frame_internal(AVFormatContext *s, AVPacket *pkt)
1279 {
1280     int ret = 0, i, got_packet = 0;
1281
1282     av_init_packet(pkt);
1283
1284     while (!got_packet && !s->parse_queue) {
1285         AVStream *st;
1286         AVPacket cur_pkt;
1287
1288         /* read next packet */
1289         ret = ff_read_packet(s, &cur_pkt);
1290         if (ret < 0) {
1291             if (ret == AVERROR(EAGAIN))
1292                 return ret;
1293             /* flush the parsers */
1294             for(i = 0; i < s->nb_streams; i++) {
1295                 st = s->streams[i];
1296                 if (st->parser && st->need_parsing)
1297                     parse_packet(s, NULL, st->index);
1298             }
1299             /* all remaining packets are now in parse_queue =>
1300              * really terminate parsing */
1301             break;
1302         }
1303         ret = 0;
1304         st  = s->streams[cur_pkt.stream_index];
1305
1306         if (cur_pkt.pts != AV_NOPTS_VALUE &&
1307             cur_pkt.dts != AV_NOPTS_VALUE &&
1308             cur_pkt.pts < cur_pkt.dts) {
1309             av_log(s, AV_LOG_WARNING, "Invalid timestamps stream=%d, pts=%s, dts=%s, size=%d\n",
1310                    cur_pkt.stream_index,
1311                    av_ts2str(cur_pkt.pts),
1312                    av_ts2str(cur_pkt.dts),
1313                    cur_pkt.size);
1314         }
1315         if (s->debug & FF_FDEBUG_TS)
1316             av_log(s, AV_LOG_DEBUG, "ff_read_packet stream=%d, pts=%s, dts=%s, size=%d, duration=%d, flags=%d\n",
1317                    cur_pkt.stream_index,
1318                    av_ts2str(cur_pkt.pts),
1319                    av_ts2str(cur_pkt.dts),
1320                    cur_pkt.size,
1321                    cur_pkt.duration,
1322                    cur_pkt.flags);
1323
1324         if (st->need_parsing && !st->parser && !(s->flags & AVFMT_FLAG_NOPARSE)) {
1325             st->parser = av_parser_init(st->codec->codec_id);
1326             if (!st->parser) {
1327                 av_log(s, AV_LOG_VERBOSE, "parser not found for codec "
1328                        "%s, packets or times may be invalid.\n",
1329                        avcodec_get_name(st->codec->codec_id));
1330                 /* no parser available: just output the raw packets */
1331                 st->need_parsing = AVSTREAM_PARSE_NONE;
1332             } else if(st->need_parsing == AVSTREAM_PARSE_HEADERS) {
1333                 st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
1334             } else if(st->need_parsing == AVSTREAM_PARSE_FULL_ONCE) {
1335                 st->parser->flags |= PARSER_FLAG_ONCE;
1336             } else if(st->need_parsing == AVSTREAM_PARSE_FULL_RAW) {
1337                 st->parser->flags |= PARSER_FLAG_USE_CODEC_TS;
1338             }
1339         }
1340
1341         if (!st->need_parsing || !st->parser) {
1342             /* no parsing needed: we just output the packet as is */
1343             *pkt = cur_pkt;
1344             compute_pkt_fields(s, st, NULL, pkt);
1345             if ((s->iformat->flags & AVFMT_GENERIC_INDEX) &&
1346                 (pkt->flags & AV_PKT_FLAG_KEY) && pkt->dts != AV_NOPTS_VALUE) {
1347                 ff_reduce_index(s, st->index);
1348                 av_add_index_entry(st, pkt->pos, pkt->dts, 0, 0, AVINDEX_KEYFRAME);
1349             }
1350             got_packet = 1;
1351         } else if (st->discard < AVDISCARD_ALL) {
1352             if ((ret = parse_packet(s, &cur_pkt, cur_pkt.stream_index)) < 0)
1353                 return ret;
1354         } else {
1355             /* free packet */
1356             av_free_packet(&cur_pkt);
1357         }
1358         if (pkt->flags & AV_PKT_FLAG_KEY)
1359             st->skip_to_keyframe = 0;
1360         if (st->skip_to_keyframe) {
1361             av_free_packet(&cur_pkt);
1362             got_packet = 0;
1363         }
1364     }
1365
1366     if (!got_packet && s->parse_queue)
1367         ret = read_from_packet_buffer(&s->parse_queue, &s->parse_queue_end, pkt);
1368
1369     if(s->debug & FF_FDEBUG_TS)
1370         av_log(s, AV_LOG_DEBUG, "read_frame_internal stream=%d, pts=%s, dts=%s, size=%d, duration=%d, flags=%d\n",
1371             pkt->stream_index,
1372             av_ts2str(pkt->pts),
1373             av_ts2str(pkt->dts),
1374             pkt->size,
1375             pkt->duration,
1376             pkt->flags);
1377
1378     return ret;
1379 }
1380
1381 int av_read_frame(AVFormatContext *s, AVPacket *pkt)
1382 {
1383     const int genpts = s->flags & AVFMT_FLAG_GENPTS;
1384     int          eof = 0;
1385     int ret;
1386     AVStream *st;
1387
1388     if (!genpts) {
1389         ret = s->packet_buffer ?
1390             read_from_packet_buffer(&s->packet_buffer, &s->packet_buffer_end, pkt) :
1391             read_frame_internal(s, pkt);
1392         if (ret < 0)
1393             return ret;
1394         goto return_packet;
1395     }
1396
1397     for (;;) {
1398         AVPacketList *pktl = s->packet_buffer;
1399
1400         if (pktl) {
1401             AVPacket *next_pkt = &pktl->pkt;
1402
1403             if (next_pkt->dts != AV_NOPTS_VALUE) {
1404                 int wrap_bits = s->streams[next_pkt->stream_index]->pts_wrap_bits;
1405                 // last dts seen for this stream. if any of packets following
1406                 // current one had no dts, we will set this to AV_NOPTS_VALUE.
1407                 int64_t last_dts = next_pkt->dts;
1408                 while (pktl && next_pkt->pts == AV_NOPTS_VALUE) {
1409                     if (pktl->pkt.stream_index == next_pkt->stream_index &&
1410                         (av_compare_mod(next_pkt->dts, pktl->pkt.dts, 2LL << (wrap_bits - 1)) < 0)) {
1411                         if (av_compare_mod(pktl->pkt.pts, pktl->pkt.dts, 2LL << (wrap_bits - 1))) { //not b frame
1412                             next_pkt->pts = pktl->pkt.dts;
1413                         }
1414                         if (last_dts != AV_NOPTS_VALUE) {
1415                             // Once last dts was set to AV_NOPTS_VALUE, we don't change it.
1416                             last_dts = pktl->pkt.dts;
1417                         }
1418                     }
1419                     pktl = pktl->next;
1420                 }
1421                 if (eof && next_pkt->pts == AV_NOPTS_VALUE && last_dts != AV_NOPTS_VALUE) {
1422                     // Fixing the last reference frame had none pts issue (For MXF etc).
1423                     // We only do this when
1424                     // 1. eof.
1425                     // 2. we are not able to resolve a pts value for current packet.
1426                     // 3. the packets for this stream at the end of the files had valid dts.
1427                     next_pkt->pts = last_dts + next_pkt->duration;
1428                 }
1429                 pktl = s->packet_buffer;
1430             }
1431
1432             /* read packet from packet buffer, if there is data */
1433             if (!(next_pkt->pts == AV_NOPTS_VALUE &&
1434                   next_pkt->dts != AV_NOPTS_VALUE && !eof)) {
1435                 ret = read_from_packet_buffer(&s->packet_buffer,
1436                                                &s->packet_buffer_end, pkt);
1437                 goto return_packet;
1438             }
1439         }
1440
1441         ret = read_frame_internal(s, pkt);
1442         if (ret < 0) {
1443             if (pktl && ret != AVERROR(EAGAIN)) {
1444                 eof = 1;
1445                 continue;
1446             } else
1447                 return ret;
1448         }
1449
1450         if (av_dup_packet(add_to_pktbuf(&s->packet_buffer, pkt,
1451                           &s->packet_buffer_end)) < 0)
1452             return AVERROR(ENOMEM);
1453     }
1454
1455 return_packet:
1456
1457     st = s->streams[pkt->stream_index];
1458     if (st->skip_samples) {
1459         uint8_t *p = av_packet_new_side_data(pkt, AV_PKT_DATA_SKIP_SAMPLES, 10);
1460         AV_WL32(p, st->skip_samples);
1461         av_log(s, AV_LOG_DEBUG, "demuxer injecting skip %d\n", st->skip_samples);
1462         st->skip_samples = 0;
1463     }
1464
1465     if ((s->iformat->flags & AVFMT_GENERIC_INDEX) && pkt->flags & AV_PKT_FLAG_KEY) {
1466         ff_reduce_index(s, st->index);
1467         av_add_index_entry(st, pkt->pos, pkt->dts, 0, 0, AVINDEX_KEYFRAME);
1468     }
1469
1470     if (is_relative(pkt->dts))
1471         pkt->dts -= RELATIVE_TS_BASE;
1472     if (is_relative(pkt->pts))
1473         pkt->pts -= RELATIVE_TS_BASE;
1474
1475     return ret;
1476 }
1477
1478 /* XXX: suppress the packet queue */
1479 static void flush_packet_queue(AVFormatContext *s)
1480 {
1481     free_packet_buffer(&s->parse_queue,       &s->parse_queue_end);
1482     free_packet_buffer(&s->packet_buffer,     &s->packet_buffer_end);
1483     free_packet_buffer(&s->raw_packet_buffer, &s->raw_packet_buffer_end);
1484
1485     s->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
1486 }
1487
1488 /*******************************************************/
1489 /* seek support */
1490
1491 int av_find_default_stream_index(AVFormatContext *s)
1492 {
1493     int first_audio_index = -1;
1494     int i;
1495     AVStream *st;
1496
1497     if (s->nb_streams <= 0)
1498         return -1;
1499     for(i = 0; i < s->nb_streams; i++) {
1500         st = s->streams[i];
1501         if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO &&
1502             !(st->disposition & AV_DISPOSITION_ATTACHED_PIC)) {
1503             return i;
1504         }
1505         if (first_audio_index < 0 && st->codec->codec_type == AVMEDIA_TYPE_AUDIO)
1506             first_audio_index = i;
1507     }
1508     return first_audio_index >= 0 ? first_audio_index : 0;
1509 }
1510
1511 /**
1512  * Flush the frame reader.
1513  */
1514 void ff_read_frame_flush(AVFormatContext *s)
1515 {
1516     AVStream *st;
1517     int i, j;
1518
1519     flush_packet_queue(s);
1520
1521     /* for each stream, reset read state */
1522     for(i = 0; i < s->nb_streams; i++) {
1523         st = s->streams[i];
1524
1525         if (st->parser) {
1526             av_parser_close(st->parser);
1527             st->parser = NULL;
1528         }
1529         st->last_IP_pts = AV_NOPTS_VALUE;
1530         if(st->first_dts == AV_NOPTS_VALUE) st->cur_dts = RELATIVE_TS_BASE;
1531         else                                st->cur_dts = AV_NOPTS_VALUE; /* we set the current DTS to an unspecified origin */
1532         st->reference_dts = AV_NOPTS_VALUE;
1533
1534         st->probe_packets = MAX_PROBE_PACKETS;
1535
1536         for(j=0; j<MAX_REORDER_DELAY+1; j++)
1537             st->pts_buffer[j]= AV_NOPTS_VALUE;
1538     }
1539 }
1540
1541 void ff_update_cur_dts(AVFormatContext *s, AVStream *ref_st, int64_t timestamp)
1542 {
1543     int i;
1544
1545     for(i = 0; i < s->nb_streams; i++) {
1546         AVStream *st = s->streams[i];
1547
1548         st->cur_dts = av_rescale(timestamp,
1549                                  st->time_base.den * (int64_t)ref_st->time_base.num,
1550                                  st->time_base.num * (int64_t)ref_st->time_base.den);
1551     }
1552 }
1553
1554 void ff_reduce_index(AVFormatContext *s, int stream_index)
1555 {
1556     AVStream *st= s->streams[stream_index];
1557     unsigned int max_entries= s->max_index_size / sizeof(AVIndexEntry);
1558
1559     if((unsigned)st->nb_index_entries >= max_entries){
1560         int i;
1561         for(i=0; 2*i<st->nb_index_entries; i++)
1562             st->index_entries[i]= st->index_entries[2*i];
1563         st->nb_index_entries= i;
1564     }
1565 }
1566
1567 int ff_add_index_entry(AVIndexEntry **index_entries,
1568                        int *nb_index_entries,
1569                        unsigned int *index_entries_allocated_size,
1570                        int64_t pos, int64_t timestamp, int size, int distance, int flags)
1571 {
1572     AVIndexEntry *entries, *ie;
1573     int index;
1574
1575     if((unsigned)*nb_index_entries + 1 >= UINT_MAX / sizeof(AVIndexEntry))
1576         return -1;
1577
1578     if(timestamp == AV_NOPTS_VALUE)
1579         return AVERROR(EINVAL);
1580
1581     if (is_relative(timestamp)) //FIXME this maintains previous behavior but we should shift by the correct offset once known
1582         timestamp -= RELATIVE_TS_BASE;
1583
1584     entries = av_fast_realloc(*index_entries,
1585                               index_entries_allocated_size,
1586                               (*nb_index_entries + 1) *
1587                               sizeof(AVIndexEntry));
1588     if(!entries)
1589         return -1;
1590
1591     *index_entries= entries;
1592
1593     index= ff_index_search_timestamp(*index_entries, *nb_index_entries, timestamp, AVSEEK_FLAG_ANY);
1594
1595     if(index<0){
1596         index= (*nb_index_entries)++;
1597         ie= &entries[index];
1598         assert(index==0 || ie[-1].timestamp < timestamp);
1599     }else{
1600         ie= &entries[index];
1601         if(ie->timestamp != timestamp){
1602             if(ie->timestamp <= timestamp)
1603                 return -1;
1604             memmove(entries + index + 1, entries + index, sizeof(AVIndexEntry)*(*nb_index_entries - index));
1605             (*nb_index_entries)++;
1606         }else if(ie->pos == pos && distance < ie->min_distance) //do not reduce the distance
1607             distance= ie->min_distance;
1608     }
1609
1610     ie->pos = pos;
1611     ie->timestamp = timestamp;
1612     ie->min_distance= distance;
1613     ie->size= size;
1614     ie->flags = flags;
1615
1616     return index;
1617 }
1618
1619 int av_add_index_entry(AVStream *st,
1620                        int64_t pos, int64_t timestamp, int size, int distance, int flags)
1621 {
1622     return ff_add_index_entry(&st->index_entries, &st->nb_index_entries,
1623                               &st->index_entries_allocated_size, pos,
1624                               timestamp, size, distance, flags);
1625 }
1626
1627 int ff_index_search_timestamp(const AVIndexEntry *entries, int nb_entries,
1628                               int64_t wanted_timestamp, int flags)
1629 {
1630     int a, b, m;
1631     int64_t timestamp;
1632
1633     a = - 1;
1634     b = nb_entries;
1635
1636     //optimize appending index entries at the end
1637     if(b && entries[b-1].timestamp < wanted_timestamp)
1638         a= b-1;
1639
1640     while (b - a > 1) {
1641         m = (a + b) >> 1;
1642         timestamp = entries[m].timestamp;
1643         if(timestamp >= wanted_timestamp)
1644             b = m;
1645         if(timestamp <= wanted_timestamp)
1646             a = m;
1647     }
1648     m= (flags & AVSEEK_FLAG_BACKWARD) ? a : b;
1649
1650     if(!(flags & AVSEEK_FLAG_ANY)){
1651         while(m>=0 && m<nb_entries && !(entries[m].flags & AVINDEX_KEYFRAME)){
1652             m += (flags & AVSEEK_FLAG_BACKWARD) ? -1 : 1;
1653         }
1654     }
1655
1656     if(m == nb_entries)
1657         return -1;
1658     return  m;
1659 }
1660
1661 int av_index_search_timestamp(AVStream *st, int64_t wanted_timestamp,
1662                               int flags)
1663 {
1664     return ff_index_search_timestamp(st->index_entries, st->nb_index_entries,
1665                                      wanted_timestamp, flags);
1666 }
1667
1668 int ff_seek_frame_binary(AVFormatContext *s, int stream_index, int64_t target_ts, int flags)
1669 {
1670     AVInputFormat *avif= s->iformat;
1671     int64_t av_uninit(pos_min), av_uninit(pos_max), pos, pos_limit;
1672     int64_t ts_min, ts_max, ts;
1673     int index;
1674     int64_t ret;
1675     AVStream *st;
1676
1677     if (stream_index < 0)
1678         return -1;
1679
1680     av_dlog(s, "read_seek: %d %s\n", stream_index, av_ts2str(target_ts));
1681
1682     ts_max=
1683     ts_min= AV_NOPTS_VALUE;
1684     pos_limit= -1; //gcc falsely says it may be uninitialized
1685
1686     st= s->streams[stream_index];
1687     if(st->index_entries){
1688         AVIndexEntry *e;
1689
1690         index= av_index_search_timestamp(st, target_ts, flags | AVSEEK_FLAG_BACKWARD); //FIXME whole func must be checked for non-keyframe entries in index case, especially read_timestamp()
1691         index= FFMAX(index, 0);
1692         e= &st->index_entries[index];
1693
1694         if(e->timestamp <= target_ts || e->pos == e->min_distance){
1695             pos_min= e->pos;
1696             ts_min= e->timestamp;
1697             av_dlog(s, "using cached pos_min=0x%"PRIx64" dts_min=%s\n",
1698                     pos_min, av_ts2str(ts_min));
1699         }else{
1700             assert(index==0);
1701         }
1702
1703         index= av_index_search_timestamp(st, target_ts, flags & ~AVSEEK_FLAG_BACKWARD);
1704         assert(index < st->nb_index_entries);
1705         if(index >= 0){
1706             e= &st->index_entries[index];
1707             assert(e->timestamp >= target_ts);
1708             pos_max= e->pos;
1709             ts_max= e->timestamp;
1710             pos_limit= pos_max - e->min_distance;
1711             av_dlog(s, "using cached pos_max=0x%"PRIx64" pos_limit=0x%"PRIx64" dts_max=%s\n",
1712                     pos_max, pos_limit, av_ts2str(ts_max));
1713         }
1714     }
1715
1716     pos= ff_gen_search(s, stream_index, target_ts, pos_min, pos_max, pos_limit, ts_min, ts_max, flags, &ts, avif->read_timestamp);
1717     if(pos<0)
1718         return -1;
1719
1720     /* do the seek */
1721     if ((ret = avio_seek(s->pb, pos, SEEK_SET)) < 0)
1722         return ret;
1723
1724     ff_read_frame_flush(s);
1725     ff_update_cur_dts(s, st, ts);
1726
1727     return 0;
1728 }
1729
1730 int64_t ff_gen_search(AVFormatContext *s, int stream_index, int64_t target_ts,
1731                       int64_t pos_min, int64_t pos_max, int64_t pos_limit,
1732                       int64_t ts_min, int64_t ts_max, int flags, int64_t *ts_ret,
1733                       int64_t (*read_timestamp)(struct AVFormatContext *, int , int64_t *, int64_t ))
1734 {
1735     int64_t pos, ts;
1736     int64_t start_pos, filesize;
1737     int no_change;
1738
1739     av_dlog(s, "gen_seek: %d %s\n", stream_index, av_ts2str(target_ts));
1740
1741     if(ts_min == AV_NOPTS_VALUE){
1742         pos_min = s->data_offset;
1743         ts_min = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
1744         if (ts_min == AV_NOPTS_VALUE)
1745             return -1;
1746     }
1747
1748     if(ts_min >= target_ts){
1749         *ts_ret= ts_min;
1750         return pos_min;
1751     }
1752
1753     if(ts_max == AV_NOPTS_VALUE){
1754         int step= 1024;
1755         filesize = avio_size(s->pb);
1756         pos_max = filesize - 1;
1757         do{
1758             pos_max -= step;
1759             ts_max = read_timestamp(s, stream_index, &pos_max, pos_max + step);
1760             step += step;
1761         }while(ts_max == AV_NOPTS_VALUE && pos_max >= step);
1762         if (ts_max == AV_NOPTS_VALUE)
1763             return -1;
1764
1765         for(;;){
1766             int64_t tmp_pos= pos_max + 1;
1767             int64_t tmp_ts= read_timestamp(s, stream_index, &tmp_pos, INT64_MAX);
1768             if(tmp_ts == AV_NOPTS_VALUE)
1769                 break;
1770             ts_max= tmp_ts;
1771             pos_max= tmp_pos;
1772             if(tmp_pos >= filesize)
1773                 break;
1774         }
1775         pos_limit= pos_max;
1776     }
1777
1778     if(ts_max <= target_ts){
1779         *ts_ret= ts_max;
1780         return pos_max;
1781     }
1782
1783     if(ts_min > ts_max){
1784         return -1;
1785     }else if(ts_min == ts_max){
1786         pos_limit= pos_min;
1787     }
1788
1789     no_change=0;
1790     while (pos_min < pos_limit) {
1791         av_dlog(s, "pos_min=0x%"PRIx64" pos_max=0x%"PRIx64" dts_min=%s dts_max=%s\n",
1792                 pos_min, pos_max, av_ts2str(ts_min), av_ts2str(ts_max));
1793         assert(pos_limit <= pos_max);
1794
1795         if(no_change==0){
1796             int64_t approximate_keyframe_distance= pos_max - pos_limit;
1797             // interpolate position (better than dichotomy)
1798             pos = av_rescale(target_ts - ts_min, pos_max - pos_min, ts_max - ts_min)
1799                 + pos_min - approximate_keyframe_distance;
1800         }else if(no_change==1){
1801             // bisection, if interpolation failed to change min or max pos last time
1802             pos = (pos_min + pos_limit)>>1;
1803         }else{
1804             /* linear search if bisection failed, can only happen if there
1805                are very few or no keyframes between min/max */
1806             pos=pos_min;
1807         }
1808         if(pos <= pos_min)
1809             pos= pos_min + 1;
1810         else if(pos > pos_limit)
1811             pos= pos_limit;
1812         start_pos= pos;
1813
1814         ts = read_timestamp(s, stream_index, &pos, INT64_MAX); //may pass pos_limit instead of -1
1815         if(pos == pos_max)
1816             no_change++;
1817         else
1818             no_change=0;
1819         av_dlog(s, "%"PRId64" %"PRId64" %"PRId64" / %s %s %s target:%s limit:%"PRId64" start:%"PRId64" noc:%d\n",
1820                 pos_min, pos, pos_max,
1821                 av_ts2str(ts_min), av_ts2str(ts), av_ts2str(ts_max), av_ts2str(target_ts),
1822                 pos_limit, start_pos, no_change);
1823         if(ts == AV_NOPTS_VALUE){
1824             av_log(s, AV_LOG_ERROR, "read_timestamp() failed in the middle\n");
1825             return -1;
1826         }
1827         assert(ts != AV_NOPTS_VALUE);
1828         if (target_ts <= ts) {
1829             pos_limit = start_pos - 1;
1830             pos_max = pos;
1831             ts_max = ts;
1832         }
1833         if (target_ts >= ts) {
1834             pos_min = pos;
1835             ts_min = ts;
1836         }
1837     }
1838
1839     pos = (flags & AVSEEK_FLAG_BACKWARD) ? pos_min : pos_max;
1840     ts  = (flags & AVSEEK_FLAG_BACKWARD) ?  ts_min :  ts_max;
1841 #if 0
1842     pos_min = pos;
1843     ts_min = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
1844     pos_min++;
1845     ts_max = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
1846     av_dlog(s, "pos=0x%"PRIx64" %s<=%s<=%s\n",
1847             pos, av_ts2str(ts_min), av_ts2str(target_ts), av_ts2str(ts_max));
1848 #endif
1849     *ts_ret= ts;
1850     return pos;
1851 }
1852
1853 static int seek_frame_byte(AVFormatContext *s, int stream_index, int64_t pos, int flags){
1854     int64_t pos_min, pos_max;
1855
1856     pos_min = s->data_offset;
1857     pos_max = avio_size(s->pb) - 1;
1858
1859     if     (pos < pos_min) pos= pos_min;
1860     else if(pos > pos_max) pos= pos_max;
1861
1862     avio_seek(s->pb, pos, SEEK_SET);
1863
1864     return 0;
1865 }
1866
1867 static int seek_frame_generic(AVFormatContext *s,
1868                                  int stream_index, int64_t timestamp, int flags)
1869 {
1870     int index;
1871     int64_t ret;
1872     AVStream *st;
1873     AVIndexEntry *ie;
1874
1875     st = s->streams[stream_index];
1876
1877     index = av_index_search_timestamp(st, timestamp, flags);
1878
1879     if(index < 0 && st->nb_index_entries && timestamp < st->index_entries[0].timestamp)
1880         return -1;
1881
1882     if(index < 0 || index==st->nb_index_entries-1){
1883         AVPacket pkt;
1884         int nonkey=0;
1885
1886         if(st->nb_index_entries){
1887             assert(st->index_entries);
1888             ie= &st->index_entries[st->nb_index_entries-1];
1889             if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
1890                 return ret;
1891             ff_update_cur_dts(s, st, ie->timestamp);
1892         }else{
1893             if ((ret = avio_seek(s->pb, s->data_offset, SEEK_SET)) < 0)
1894                 return ret;
1895         }
1896         for (;;) {
1897             int read_status;
1898             do{
1899                 read_status = av_read_frame(s, &pkt);
1900             } while (read_status == AVERROR(EAGAIN));
1901             if (read_status < 0)
1902                 break;
1903             av_free_packet(&pkt);
1904             if(stream_index == pkt.stream_index && pkt.dts > timestamp){
1905                 if(pkt.flags & AV_PKT_FLAG_KEY)
1906                     break;
1907                 if(nonkey++ > 1000 && st->codec->codec_id != AV_CODEC_ID_CDGRAPHICS){
1908                     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);
1909                     break;
1910                 }
1911             }
1912         }
1913         index = av_index_search_timestamp(st, timestamp, flags);
1914     }
1915     if (index < 0)
1916         return -1;
1917
1918     ff_read_frame_flush(s);
1919     if (s->iformat->read_seek){
1920         if(s->iformat->read_seek(s, stream_index, timestamp, flags) >= 0)
1921             return 0;
1922     }
1923     ie = &st->index_entries[index];
1924     if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
1925         return ret;
1926     ff_update_cur_dts(s, st, ie->timestamp);
1927
1928     return 0;
1929 }
1930
1931 static int seek_frame_internal(AVFormatContext *s, int stream_index,
1932                                int64_t timestamp, int flags)
1933 {
1934     int ret;
1935     AVStream *st;
1936
1937     if (flags & AVSEEK_FLAG_BYTE) {
1938         if (s->iformat->flags & AVFMT_NO_BYTE_SEEK)
1939             return -1;
1940         ff_read_frame_flush(s);
1941         return seek_frame_byte(s, stream_index, timestamp, flags);
1942     }
1943
1944     if(stream_index < 0){
1945         stream_index= av_find_default_stream_index(s);
1946         if(stream_index < 0)
1947             return -1;
1948
1949         st= s->streams[stream_index];
1950         /* timestamp for default must be expressed in AV_TIME_BASE units */
1951         timestamp = av_rescale(timestamp, st->time_base.den, AV_TIME_BASE * (int64_t)st->time_base.num);
1952     }
1953
1954     /* first, we try the format specific seek */
1955     if (s->iformat->read_seek) {
1956         ff_read_frame_flush(s);
1957         ret = s->iformat->read_seek(s, stream_index, timestamp, flags);
1958     } else
1959         ret = -1;
1960     if (ret >= 0) {
1961         return 0;
1962     }
1963
1964     if (s->iformat->read_timestamp && !(s->iformat->flags & AVFMT_NOBINSEARCH)) {
1965         ff_read_frame_flush(s);
1966         return ff_seek_frame_binary(s, stream_index, timestamp, flags);
1967     } else if (!(s->iformat->flags & AVFMT_NOGENSEARCH)) {
1968         ff_read_frame_flush(s);
1969         return seek_frame_generic(s, stream_index, timestamp, flags);
1970     }
1971     else
1972         return -1;
1973 }
1974
1975 int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
1976 {
1977     int ret = seek_frame_internal(s, stream_index, timestamp, flags);
1978
1979     if (ret >= 0)
1980         avformat_queue_attached_pictures(s);
1981
1982     return ret;
1983 }
1984
1985 int avformat_seek_file(AVFormatContext *s, int stream_index, int64_t min_ts, int64_t ts, int64_t max_ts, int flags)
1986 {
1987     if(min_ts > ts || max_ts < ts)
1988         return -1;
1989
1990     if (s->iformat->read_seek2) {
1991         int ret;
1992         ff_read_frame_flush(s);
1993         ret = s->iformat->read_seek2(s, stream_index, min_ts, ts, max_ts, flags);
1994
1995         if (ret >= 0)
1996             avformat_queue_attached_pictures(s);
1997         return ret;
1998     }
1999
2000     if(s->iformat->read_timestamp){
2001         //try to seek via read_timestamp()
2002     }
2003
2004     //Fallback to old API if new is not implemented but old is
2005     //Note the old has somewhat different semantics
2006     if (s->iformat->read_seek || 1) {
2007         int dir = (ts - (uint64_t)min_ts > (uint64_t)max_ts - ts ? AVSEEK_FLAG_BACKWARD : 0);
2008         int ret = av_seek_frame(s, stream_index, ts, flags | dir);
2009         if (ret<0 && ts != min_ts && max_ts != ts) {
2010             ret = av_seek_frame(s, stream_index, dir ? max_ts : min_ts, flags | dir);
2011             if (ret >= 0)
2012                 ret = av_seek_frame(s, stream_index, ts, flags | (dir^AVSEEK_FLAG_BACKWARD));
2013         }
2014         return ret;
2015     }
2016
2017     // try some generic seek like seek_frame_generic() but with new ts semantics
2018 }
2019
2020 /*******************************************************/
2021
2022 /**
2023  * Return TRUE if the stream has accurate duration in any stream.
2024  *
2025  * @return TRUE if the stream has accurate duration for at least one component.
2026  */
2027 static int has_duration(AVFormatContext *ic)
2028 {
2029     int i;
2030     AVStream *st;
2031
2032     for(i = 0;i < ic->nb_streams; i++) {
2033         st = ic->streams[i];
2034         if (st->duration != AV_NOPTS_VALUE)
2035             return 1;
2036     }
2037     if (ic->duration != AV_NOPTS_VALUE)
2038         return 1;
2039     return 0;
2040 }
2041
2042 /**
2043  * Estimate the stream timings from the one of each components.
2044  *
2045  * Also computes the global bitrate if possible.
2046  */
2047 static void update_stream_timings(AVFormatContext *ic)
2048 {
2049     int64_t start_time, start_time1, start_time_text, end_time, end_time1;
2050     int64_t duration, duration1, filesize;
2051     int i;
2052     AVStream *st;
2053     AVProgram *p;
2054
2055     start_time = INT64_MAX;
2056     start_time_text = INT64_MAX;
2057     end_time = INT64_MIN;
2058     duration = INT64_MIN;
2059     for(i = 0;i < ic->nb_streams; i++) {
2060         st = ic->streams[i];
2061         if (st->start_time != AV_NOPTS_VALUE && st->time_base.den) {
2062             start_time1= av_rescale_q(st->start_time, st->time_base, AV_TIME_BASE_Q);
2063             if (st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE || st->codec->codec_type == AVMEDIA_TYPE_DATA) {
2064                 if (start_time1 < start_time_text)
2065                     start_time_text = start_time1;
2066             } else
2067                 start_time = FFMIN(start_time, start_time1);
2068             end_time1 = AV_NOPTS_VALUE;
2069             if (st->duration != AV_NOPTS_VALUE) {
2070                 end_time1 = start_time1
2071                           + av_rescale_q(st->duration, st->time_base, AV_TIME_BASE_Q);
2072                 end_time = FFMAX(end_time, end_time1);
2073             }
2074             for(p = NULL; (p = av_find_program_from_stream(ic, p, i)); ){
2075                 if(p->start_time == AV_NOPTS_VALUE || p->start_time > start_time1)
2076                     p->start_time = start_time1;
2077                 if(p->end_time < end_time1)
2078                     p->end_time = end_time1;
2079             }
2080         }
2081         if (st->duration != AV_NOPTS_VALUE) {
2082             duration1 = av_rescale_q(st->duration, st->time_base, AV_TIME_BASE_Q);
2083             duration = FFMAX(duration, duration1);
2084         }
2085     }
2086     if (start_time == INT64_MAX || (start_time > start_time_text && start_time - start_time_text < AV_TIME_BASE))
2087         start_time = start_time_text;
2088     else if(start_time > start_time_text)
2089         av_log(ic, AV_LOG_VERBOSE, "Ignoring outlier non primary stream starttime %f\n", start_time_text / (float)AV_TIME_BASE);
2090
2091     if (start_time != INT64_MAX) {
2092         ic->start_time = start_time;
2093         if (end_time != INT64_MIN) {
2094             if (ic->nb_programs) {
2095                 for (i=0; i<ic->nb_programs; i++) {
2096                     p = ic->programs[i];
2097                     if(p->start_time != AV_NOPTS_VALUE && p->end_time > p->start_time)
2098                         duration = FFMAX(duration, p->end_time - p->start_time);
2099                 }
2100             } else
2101                 duration = FFMAX(duration, end_time - start_time);
2102         }
2103     }
2104     if (duration != INT64_MIN && duration > 0 && ic->duration == AV_NOPTS_VALUE) {
2105         ic->duration = duration;
2106     }
2107         if (ic->pb && (filesize = avio_size(ic->pb)) > 0 && ic->duration != AV_NOPTS_VALUE) {
2108             /* compute the bitrate */
2109             ic->bit_rate = (double)filesize * 8.0 * AV_TIME_BASE /
2110                 (double)ic->duration;
2111         }
2112 }
2113
2114 static void fill_all_stream_timings(AVFormatContext *ic)
2115 {
2116     int i;
2117     AVStream *st;
2118
2119     update_stream_timings(ic);
2120     for(i = 0;i < ic->nb_streams; i++) {
2121         st = ic->streams[i];
2122         if (st->start_time == AV_NOPTS_VALUE) {
2123             if(ic->start_time != AV_NOPTS_VALUE)
2124                 st->start_time = av_rescale_q(ic->start_time, AV_TIME_BASE_Q, st->time_base);
2125             if(ic->duration != AV_NOPTS_VALUE)
2126                 st->duration = av_rescale_q(ic->duration, AV_TIME_BASE_Q, st->time_base);
2127         }
2128     }
2129 }
2130
2131 static void estimate_timings_from_bit_rate(AVFormatContext *ic)
2132 {
2133     int64_t filesize, duration;
2134     int bit_rate, i;
2135     AVStream *st;
2136
2137     /* if bit_rate is already set, we believe it */
2138     if (ic->bit_rate <= 0) {
2139         bit_rate = 0;
2140         for(i=0;i<ic->nb_streams;i++) {
2141             st = ic->streams[i];
2142             if (st->codec->bit_rate > 0)
2143             bit_rate += st->codec->bit_rate;
2144         }
2145         ic->bit_rate = bit_rate;
2146     }
2147
2148     /* if duration is already set, we believe it */
2149     if (ic->duration == AV_NOPTS_VALUE &&
2150         ic->bit_rate != 0) {
2151         filesize = ic->pb ? avio_size(ic->pb) : 0;
2152         if (filesize > 0) {
2153             for(i = 0; i < ic->nb_streams; i++) {
2154                 st = ic->streams[i];
2155                 duration= av_rescale(8*filesize, st->time_base.den, ic->bit_rate*(int64_t)st->time_base.num);
2156                 if (st->duration == AV_NOPTS_VALUE)
2157                     st->duration = duration;
2158             }
2159         }
2160     }
2161 }
2162
2163 #define DURATION_MAX_READ_SIZE 250000LL
2164 #define DURATION_MAX_RETRY 4
2165
2166 /* only usable for MPEG-PS streams */
2167 static void estimate_timings_from_pts(AVFormatContext *ic, int64_t old_offset)
2168 {
2169     AVPacket pkt1, *pkt = &pkt1;
2170     AVStream *st;
2171     int read_size, i, ret;
2172     int64_t end_time;
2173     int64_t filesize, offset, duration;
2174     int retry=0;
2175
2176     /* flush packet queue */
2177     flush_packet_queue(ic);
2178
2179     for (i=0; i<ic->nb_streams; i++) {
2180         st = ic->streams[i];
2181         if (st->start_time == AV_NOPTS_VALUE && st->first_dts == AV_NOPTS_VALUE)
2182             av_log(st->codec, AV_LOG_WARNING, "start time is not set in estimate_timings_from_pts\n");
2183
2184         if (st->parser) {
2185             av_parser_close(st->parser);
2186             st->parser= NULL;
2187         }
2188     }
2189
2190     /* estimate the end time (duration) */
2191     /* XXX: may need to support wrapping */
2192     filesize = ic->pb ? avio_size(ic->pb) : 0;
2193     end_time = AV_NOPTS_VALUE;
2194     do{
2195         offset = filesize - (DURATION_MAX_READ_SIZE<<retry);
2196         if (offset < 0)
2197             offset = 0;
2198
2199         avio_seek(ic->pb, offset, SEEK_SET);
2200         read_size = 0;
2201         for(;;) {
2202             if (read_size >= DURATION_MAX_READ_SIZE<<(FFMAX(retry-1,0)))
2203                 break;
2204
2205             do {
2206                 ret = ff_read_packet(ic, pkt);
2207             } while(ret == AVERROR(EAGAIN));
2208             if (ret != 0)
2209                 break;
2210             read_size += pkt->size;
2211             st = ic->streams[pkt->stream_index];
2212             if (pkt->pts != AV_NOPTS_VALUE &&
2213                 (st->start_time != AV_NOPTS_VALUE ||
2214                  st->first_dts  != AV_NOPTS_VALUE)) {
2215                 duration = end_time = pkt->pts;
2216                 if (st->start_time != AV_NOPTS_VALUE)
2217                     duration -= st->start_time;
2218                 else
2219                     duration -= st->first_dts;
2220                 if (duration < 0)
2221                     duration += 1LL<<st->pts_wrap_bits;
2222                 if (duration > 0) {
2223                     if (st->duration == AV_NOPTS_VALUE || st->duration < duration)
2224                         st->duration = duration;
2225                 }
2226             }
2227             av_free_packet(pkt);
2228         }
2229     }while(   end_time==AV_NOPTS_VALUE
2230            && filesize > (DURATION_MAX_READ_SIZE<<retry)
2231            && ++retry <= DURATION_MAX_RETRY);
2232
2233     fill_all_stream_timings(ic);
2234
2235     avio_seek(ic->pb, old_offset, SEEK_SET);
2236     for (i=0; i<ic->nb_streams; i++) {
2237         st= ic->streams[i];
2238         st->cur_dts= st->first_dts;
2239         st->last_IP_pts = AV_NOPTS_VALUE;
2240         st->reference_dts = AV_NOPTS_VALUE;
2241     }
2242 }
2243
2244 static void estimate_timings(AVFormatContext *ic, int64_t old_offset)
2245 {
2246     int64_t file_size;
2247
2248     /* get the file size, if possible */
2249     if (ic->iformat->flags & AVFMT_NOFILE) {
2250         file_size = 0;
2251     } else {
2252         file_size = avio_size(ic->pb);
2253         file_size = FFMAX(0, file_size);
2254     }
2255
2256     if ((!strcmp(ic->iformat->name, "mpeg") ||
2257          !strcmp(ic->iformat->name, "mpegts")) &&
2258         file_size && ic->pb->seekable) {
2259         /* get accurate estimate from the PTSes */
2260         estimate_timings_from_pts(ic, old_offset);
2261         ic->duration_estimation_method = AVFMT_DURATION_FROM_PTS;
2262     } else if (has_duration(ic)) {
2263         /* at least one component has timings - we use them for all
2264            the components */
2265         fill_all_stream_timings(ic);
2266         ic->duration_estimation_method = AVFMT_DURATION_FROM_STREAM;
2267     } else {
2268         av_log(ic, AV_LOG_WARNING, "Estimating duration from bitrate, this may be inaccurate\n");
2269         /* less precise: use bitrate info */
2270         estimate_timings_from_bit_rate(ic);
2271         ic->duration_estimation_method = AVFMT_DURATION_FROM_BITRATE;
2272     }
2273     update_stream_timings(ic);
2274
2275     {
2276         int i;
2277         AVStream av_unused *st;
2278         for(i = 0;i < ic->nb_streams; i++) {
2279             st = ic->streams[i];
2280             av_dlog(ic, "%d: start_time: %0.3f duration: %0.3f\n", i,
2281                     (double) st->start_time / AV_TIME_BASE,
2282                     (double) st->duration   / AV_TIME_BASE);
2283         }
2284         av_dlog(ic, "stream: start_time: %0.3f duration: %0.3f bitrate=%d kb/s\n",
2285                 (double) ic->start_time / AV_TIME_BASE,
2286                 (double) ic->duration   / AV_TIME_BASE,
2287                 ic->bit_rate / 1000);
2288     }
2289 }
2290
2291 static int has_codec_parameters(AVStream *st, const char **errmsg_ptr)
2292 {
2293     AVCodecContext *avctx = st->codec;
2294
2295 #define FAIL(errmsg) do {                                         \
2296         if (errmsg_ptr)                                           \
2297             *errmsg_ptr = errmsg;                                 \
2298         return 0;                                                 \
2299     } while (0)
2300
2301     switch (avctx->codec_type) {
2302     case AVMEDIA_TYPE_AUDIO:
2303         if (!avctx->frame_size && determinable_frame_size(avctx))
2304             FAIL("unspecified frame size");
2305         if (st->info->found_decoder >= 0 && avctx->sample_fmt == AV_SAMPLE_FMT_NONE)
2306             FAIL("unspecified sample format");
2307         if (!avctx->sample_rate)
2308             FAIL("unspecified sample rate");
2309         if (!avctx->channels)
2310             FAIL("unspecified number of channels");
2311         if (st->info->found_decoder >= 0 && !st->nb_decoded_frames && avctx->codec_id == AV_CODEC_ID_DTS)
2312             FAIL("no decodable DTS frames");
2313         break;
2314     case AVMEDIA_TYPE_VIDEO:
2315         if (!avctx->width)
2316             FAIL("unspecified size");
2317         if (st->info->found_decoder >= 0 && avctx->pix_fmt == AV_PIX_FMT_NONE)
2318             FAIL("unspecified pixel format");
2319         break;
2320     case AVMEDIA_TYPE_SUBTITLE:
2321         if (avctx->codec_id == AV_CODEC_ID_HDMV_PGS_SUBTITLE && !avctx->width)
2322             FAIL("unspecified size");
2323         break;
2324     case AVMEDIA_TYPE_DATA:
2325         if(avctx->codec_id == AV_CODEC_ID_NONE) return 1;
2326     }
2327
2328     if (avctx->codec_id == AV_CODEC_ID_NONE)
2329         FAIL("unknown codec");
2330     return 1;
2331 }
2332
2333 /* returns 1 or 0 if or if not decoded data was returned, or a negative error */
2334 static int try_decode_frame(AVStream *st, AVPacket *avpkt, AVDictionary **options)
2335 {
2336     const AVCodec *codec;
2337     int got_picture = 1, ret = 0;
2338     AVFrame *frame = avcodec_alloc_frame();
2339     AVSubtitle subtitle;
2340     AVPacket pkt = *avpkt;
2341
2342     if (!frame)
2343         return AVERROR(ENOMEM);
2344
2345     if (!avcodec_is_open(st->codec) && !st->info->found_decoder) {
2346         AVDictionary *thread_opt = NULL;
2347
2348         codec = st->codec->codec ? st->codec->codec :
2349                                    avcodec_find_decoder(st->codec->codec_id);
2350
2351         if (!codec) {
2352             st->info->found_decoder = -1;
2353             ret = -1;
2354             goto fail;
2355         }
2356
2357         /* force thread count to 1 since the h264 decoder will not extract SPS
2358          *  and PPS to extradata during multi-threaded decoding */
2359         av_dict_set(options ? options : &thread_opt, "threads", "1", 0);
2360         ret = avcodec_open2(st->codec, codec, options ? options : &thread_opt);
2361         if (!options)
2362             av_dict_free(&thread_opt);
2363         if (ret < 0) {
2364             st->info->found_decoder = -1;
2365             goto fail;
2366         }
2367         st->info->found_decoder = 1;
2368     } else if (!st->info->found_decoder)
2369         st->info->found_decoder = 1;
2370
2371     if (st->info->found_decoder < 0) {
2372         ret = -1;
2373         goto fail;
2374     }
2375
2376     while ((pkt.size > 0 || (!pkt.data && got_picture)) &&
2377            ret >= 0 &&
2378            (!has_codec_parameters(st, NULL)   ||
2379            !has_decode_delay_been_guessed(st) ||
2380            (!st->codec_info_nb_frames && st->codec->codec->capabilities & CODEC_CAP_CHANNEL_CONF))) {
2381         got_picture = 0;
2382         avcodec_get_frame_defaults(frame);
2383         switch(st->codec->codec_type) {
2384         case AVMEDIA_TYPE_VIDEO:
2385             ret = avcodec_decode_video2(st->codec, frame,
2386                                         &got_picture, &pkt);
2387             break;
2388         case AVMEDIA_TYPE_AUDIO:
2389             ret = avcodec_decode_audio4(st->codec, frame, &got_picture, &pkt);
2390             break;
2391         case AVMEDIA_TYPE_SUBTITLE:
2392             ret = avcodec_decode_subtitle2(st->codec, &subtitle,
2393                                            &got_picture, &pkt);
2394             ret = pkt.size;
2395             break;
2396         default:
2397             break;
2398         }
2399         if (ret >= 0) {
2400             if (got_picture)
2401                 st->nb_decoded_frames++;
2402             pkt.data += ret;
2403             pkt.size -= ret;
2404             ret       = got_picture;
2405         }
2406     }
2407
2408     if(!pkt.data && !got_picture)
2409         ret = -1;
2410
2411 fail:
2412     avcodec_free_frame(&frame);
2413     return ret;
2414 }
2415
2416 unsigned int ff_codec_get_tag(const AVCodecTag *tags, enum AVCodecID id)
2417 {
2418     while (tags->id != AV_CODEC_ID_NONE) {
2419         if (tags->id == id)
2420             return tags->tag;
2421         tags++;
2422     }
2423     return 0;
2424 }
2425
2426 enum AVCodecID ff_codec_get_id(const AVCodecTag *tags, unsigned int tag)
2427 {
2428     int i;
2429     for(i=0; tags[i].id != AV_CODEC_ID_NONE;i++) {
2430         if(tag == tags[i].tag)
2431             return tags[i].id;
2432     }
2433     for(i=0; tags[i].id != AV_CODEC_ID_NONE; i++) {
2434         if (avpriv_toupper4(tag) == avpriv_toupper4(tags[i].tag))
2435             return tags[i].id;
2436     }
2437     return AV_CODEC_ID_NONE;
2438 }
2439
2440 unsigned int av_codec_get_tag(const AVCodecTag * const *tags, enum AVCodecID id)
2441 {
2442     int i;
2443     for(i=0; tags && tags[i]; i++){
2444         int tag= ff_codec_get_tag(tags[i], id);
2445         if(tag) return tag;
2446     }
2447     return 0;
2448 }
2449
2450 enum AVCodecID av_codec_get_id(const AVCodecTag * const *tags, unsigned int tag)
2451 {
2452     int i;
2453     for(i=0; tags && tags[i]; i++){
2454         enum AVCodecID id= ff_codec_get_id(tags[i], tag);
2455         if(id!=AV_CODEC_ID_NONE) return id;
2456     }
2457     return AV_CODEC_ID_NONE;
2458 }
2459
2460 static void compute_chapters_end(AVFormatContext *s)
2461 {
2462     unsigned int i, j;
2463     int64_t max_time = s->duration + ((s->start_time == AV_NOPTS_VALUE) ? 0 : s->start_time);
2464
2465     for (i = 0; i < s->nb_chapters; i++)
2466         if (s->chapters[i]->end == AV_NOPTS_VALUE) {
2467             AVChapter *ch = s->chapters[i];
2468             int64_t   end = max_time ? av_rescale_q(max_time, AV_TIME_BASE_Q, ch->time_base)
2469                                      : INT64_MAX;
2470
2471             for (j = 0; j < s->nb_chapters; j++) {
2472                 AVChapter *ch1 = s->chapters[j];
2473                 int64_t next_start = av_rescale_q(ch1->start, ch1->time_base, ch->time_base);
2474                 if (j != i && next_start > ch->start && next_start < end)
2475                     end = next_start;
2476             }
2477             ch->end = (end == INT64_MAX) ? ch->start : end;
2478         }
2479 }
2480
2481 static int get_std_framerate(int i){
2482     if(i<60*12) return (i+1)*1001;
2483     else        return ((const int[]){24,30,60,12,15,48})[i-60*12]*1000*12;
2484 }
2485
2486 /*
2487  * Is the time base unreliable.
2488  * This is a heuristic to balance between quick acceptance of the values in
2489  * the headers vs. some extra checks.
2490  * Old DivX and Xvid often have nonsense timebases like 1fps or 2fps.
2491  * MPEG-2 commonly misuses field repeat flags to store different framerates.
2492  * And there are "variable" fps files this needs to detect as well.
2493  */
2494 static int tb_unreliable(AVCodecContext *c){
2495     if(   c->time_base.den >= 101L*c->time_base.num
2496        || c->time_base.den <    5L*c->time_base.num
2497 /*       || c->codec_tag == AV_RL32("DIVX")
2498        || c->codec_tag == AV_RL32("XVID")*/
2499        || c->codec_tag == AV_RL32("mp4v")
2500        || c->codec_id == AV_CODEC_ID_MPEG2VIDEO
2501        || c->codec_id == AV_CODEC_ID_H264
2502        )
2503         return 1;
2504     return 0;
2505 }
2506
2507 #if FF_API_FORMAT_PARAMETERS
2508 int av_find_stream_info(AVFormatContext *ic)
2509 {
2510     return avformat_find_stream_info(ic, NULL);
2511 }
2512 #endif
2513
2514 int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
2515 {
2516     int i, count, ret, read_size, j;
2517     AVStream *st;
2518     AVPacket pkt1, *pkt;
2519     int64_t old_offset = avio_tell(ic->pb);
2520     int orig_nb_streams = ic->nb_streams;        // new streams might appear, no options for those
2521     int flush_codecs = ic->probesize > 0;
2522
2523     if(ic->pb)
2524         av_log(ic, AV_LOG_DEBUG, "File position before avformat_find_stream_info() is %"PRId64"\n", avio_tell(ic->pb));
2525
2526     for(i=0;i<ic->nb_streams;i++) {
2527         const AVCodec *codec;
2528         AVDictionary *thread_opt = NULL;
2529         st = ic->streams[i];
2530
2531         if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO ||
2532             st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE) {
2533 /*            if(!st->time_base.num)
2534                 st->time_base= */
2535             if(!st->codec->time_base.num)
2536                 st->codec->time_base= st->time_base;
2537         }
2538         //only for the split stuff
2539         if (!st->parser && !(ic->flags & AVFMT_FLAG_NOPARSE)) {
2540             st->parser = av_parser_init(st->codec->codec_id);
2541             if(st->parser){
2542                 if(st->need_parsing == AVSTREAM_PARSE_HEADERS){
2543                     st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
2544                 } else if(st->need_parsing == AVSTREAM_PARSE_FULL_RAW) {
2545                     st->parser->flags |= PARSER_FLAG_USE_CODEC_TS;
2546                 }
2547             } else if (st->need_parsing) {
2548                 av_log(ic, AV_LOG_VERBOSE, "parser not found for codec "
2549                        "%s, packets or times may be invalid.\n",
2550                        avcodec_get_name(st->codec->codec_id));
2551             }
2552         }
2553         codec = st->codec->codec ? st->codec->codec :
2554                                    avcodec_find_decoder(st->codec->codec_id);
2555
2556         /* force thread count to 1 since the h264 decoder will not extract SPS
2557          *  and PPS to extradata during multi-threaded decoding */
2558         av_dict_set(options ? &options[i] : &thread_opt, "threads", "1", 0);
2559
2560         /* Ensure that subtitle_header is properly set. */
2561         if (st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE
2562             && codec && !st->codec->codec)
2563             avcodec_open2(st->codec, codec, options ? &options[i]
2564                               : &thread_opt);
2565
2566         //try to just open decoders, in case this is enough to get parameters
2567         if (!has_codec_parameters(st, NULL) && st->request_probe <= 0) {
2568             if (codec && !st->codec->codec)
2569                 avcodec_open2(st->codec, codec, options ? &options[i]
2570                               : &thread_opt);
2571         }
2572         if (!options)
2573             av_dict_free(&thread_opt);
2574     }
2575
2576     for (i=0; i<ic->nb_streams; i++) {
2577 #if FF_API_R_FRAME_RATE
2578         ic->streams[i]->info->last_dts = AV_NOPTS_VALUE;
2579 #endif
2580         ic->streams[i]->info->fps_first_dts = AV_NOPTS_VALUE;
2581         ic->streams[i]->info->fps_last_dts  = AV_NOPTS_VALUE;
2582     }
2583
2584     count = 0;
2585     read_size = 0;
2586     for(;;) {
2587         if (ff_check_interrupt(&ic->interrupt_callback)){
2588             ret= AVERROR_EXIT;
2589             av_log(ic, AV_LOG_DEBUG, "interrupted\n");
2590             break;
2591         }
2592
2593         /* check if one codec still needs to be handled */
2594         for(i=0;i<ic->nb_streams;i++) {
2595             int fps_analyze_framecount = 20;
2596
2597             st = ic->streams[i];
2598             if (!has_codec_parameters(st, NULL))
2599                 break;
2600             /* if the timebase is coarse (like the usual millisecond precision
2601                of mkv), we need to analyze more frames to reliably arrive at
2602                the correct fps */
2603             if (av_q2d(st->time_base) > 0.0005)
2604                 fps_analyze_framecount *= 2;
2605             if (ic->fps_probe_size >= 0)
2606                 fps_analyze_framecount = ic->fps_probe_size;
2607             /* variable fps and no guess at the real fps */
2608             if(   tb_unreliable(st->codec) && !(st->r_frame_rate.num && st->avg_frame_rate.num)
2609                && st->info->duration_count < fps_analyze_framecount
2610                && st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
2611                 break;
2612             if(st->parser && st->parser->parser->split && !st->codec->extradata)
2613                 break;
2614             if (st->first_dts == AV_NOPTS_VALUE &&
2615                 (st->codec->codec_type == AVMEDIA_TYPE_VIDEO ||
2616                  st->codec->codec_type == AVMEDIA_TYPE_AUDIO))
2617                 break;
2618         }
2619         if (i == ic->nb_streams) {
2620             /* NOTE: if the format has no header, then we need to read
2621                some packets to get most of the streams, so we cannot
2622                stop here */
2623             if (!(ic->ctx_flags & AVFMTCTX_NOHEADER)) {
2624                 /* if we found the info for all the codecs, we can stop */
2625                 ret = count;
2626                 av_log(ic, AV_LOG_DEBUG, "All info found\n");
2627                 flush_codecs = 0;
2628                 break;
2629             }
2630         }
2631         /* we did not get all the codec info, but we read too much data */
2632         if (read_size >= ic->probesize) {
2633             ret = count;
2634             av_log(ic, AV_LOG_DEBUG, "Probe buffer size limit %d reached\n", ic->probesize);
2635             for (i = 0; i < ic->nb_streams; i++)
2636                 if (!ic->streams[i]->r_frame_rate.num &&
2637                     ic->streams[i]->info->duration_count <= 1)
2638                     av_log(ic, AV_LOG_WARNING,
2639                            "Stream #%d: not enough frames to estimate rate; "
2640                            "consider increasing probesize\n", i);
2641             break;
2642         }
2643
2644         /* NOTE: a new stream can be added there if no header in file
2645            (AVFMTCTX_NOHEADER) */
2646         ret = read_frame_internal(ic, &pkt1);
2647         if (ret == AVERROR(EAGAIN))
2648             continue;
2649
2650         if (ret < 0) {
2651             /* EOF or error*/
2652             break;
2653         }
2654
2655         if (ic->flags & AVFMT_FLAG_NOBUFFER) {
2656             pkt = &pkt1;
2657         } else {
2658             pkt = add_to_pktbuf(&ic->packet_buffer, &pkt1,
2659                                 &ic->packet_buffer_end);
2660             if ((ret = av_dup_packet(pkt)) < 0)
2661                 goto find_stream_info_err;
2662         }
2663
2664         read_size += pkt->size;
2665
2666         st = ic->streams[pkt->stream_index];
2667         if (pkt->dts != AV_NOPTS_VALUE && st->codec_info_nb_frames > 1) {
2668             /* check for non-increasing dts */
2669             if (st->info->fps_last_dts != AV_NOPTS_VALUE &&
2670                 st->info->fps_last_dts >= pkt->dts) {
2671                 av_log(ic, AV_LOG_DEBUG, "Non-increasing DTS in stream %d: "
2672                        "packet %d with DTS %"PRId64", packet %d with DTS "
2673                        "%"PRId64"\n", st->index, st->info->fps_last_dts_idx,
2674                        st->info->fps_last_dts, st->codec_info_nb_frames, pkt->dts);
2675                 st->info->fps_first_dts = st->info->fps_last_dts = AV_NOPTS_VALUE;
2676             }
2677             /* check for a discontinuity in dts - if the difference in dts
2678              * is more than 1000 times the average packet duration in the sequence,
2679              * we treat it as a discontinuity */
2680             if (st->info->fps_last_dts != AV_NOPTS_VALUE &&
2681                 st->info->fps_last_dts_idx > st->info->fps_first_dts_idx &&
2682                 (pkt->dts - st->info->fps_last_dts) / 1000 >
2683                 (st->info->fps_last_dts - st->info->fps_first_dts) / (st->info->fps_last_dts_idx - st->info->fps_first_dts_idx)) {
2684                 av_log(ic, AV_LOG_WARNING, "DTS discontinuity in stream %d: "
2685                        "packet %d with DTS %"PRId64", packet %d with DTS "
2686                        "%"PRId64"\n", st->index, st->info->fps_last_dts_idx,
2687                        st->info->fps_last_dts, st->codec_info_nb_frames, pkt->dts);
2688                 st->info->fps_first_dts = st->info->fps_last_dts = AV_NOPTS_VALUE;
2689             }
2690
2691             /* update stored dts values */
2692             if (st->info->fps_first_dts == AV_NOPTS_VALUE) {
2693                 st->info->fps_first_dts     = pkt->dts;
2694                 st->info->fps_first_dts_idx = st->codec_info_nb_frames;
2695             }
2696             st->info->fps_last_dts = pkt->dts;
2697             st->info->fps_last_dts_idx = st->codec_info_nb_frames;
2698         }
2699         if (st->codec_info_nb_frames>1) {
2700             int64_t t=0;
2701             if (st->time_base.den > 0)
2702                 t = av_rescale_q(st->info->codec_info_duration, st->time_base, AV_TIME_BASE_Q);
2703             if (st->avg_frame_rate.num > 0)
2704                 t = FFMAX(t, av_rescale_q(st->codec_info_nb_frames, av_inv_q(st->avg_frame_rate), AV_TIME_BASE_Q));
2705
2706             if (t >= ic->max_analyze_duration) {
2707                 av_log(ic, AV_LOG_WARNING, "max_analyze_duration %d reached at %"PRId64"\n", ic->max_analyze_duration, t);
2708                 break;
2709             }
2710             if (pkt->duration) {
2711                 st->info->codec_info_duration        += pkt->duration;
2712                 st->info->codec_info_duration_fields += st->parser && st->codec->ticks_per_frame==2 ? st->parser->repeat_pict + 1 : 2;
2713             }
2714         }
2715 #if FF_API_R_FRAME_RATE
2716         {
2717             int64_t last = st->info->last_dts;
2718
2719             if(pkt->dts != AV_NOPTS_VALUE && last != AV_NOPTS_VALUE && pkt->dts > last){
2720                 double dts= (is_relative(pkt->dts) ?  pkt->dts - RELATIVE_TS_BASE : pkt->dts) * av_q2d(st->time_base);
2721                 int64_t duration= pkt->dts - last;
2722
2723 //                 if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
2724 //                     av_log(NULL, AV_LOG_ERROR, "%f\n", dts);
2725                 for (i=0; i<FF_ARRAY_ELEMS(st->info->duration_error[0][0]); i++) {
2726                     int framerate= get_std_framerate(i);
2727                     double sdts= dts*framerate/(1001*12);
2728                     for(j=0; j<2; j++){
2729                         int64_t ticks= llrint(sdts+j*0.5);
2730                         double error= sdts - ticks + j*0.5;
2731                         st->info->duration_error[j][0][i] += error;
2732                         st->info->duration_error[j][1][i] += error*error;
2733                     }
2734                 }
2735                 st->info->duration_count++;
2736                 // ignore the first 4 values, they might have some random jitter
2737                 if (st->info->duration_count > 3 && is_relative(pkt->dts) == is_relative(last))
2738                     st->info->duration_gcd = av_gcd(st->info->duration_gcd, duration);
2739             }
2740             if (pkt->dts != AV_NOPTS_VALUE)
2741                 st->info->last_dts = pkt->dts;
2742         }
2743 #endif
2744         if(st->parser && st->parser->parser->split && !st->codec->extradata){
2745             int i= st->parser->parser->split(st->codec, pkt->data, pkt->size);
2746             if (i > 0 && i < FF_MAX_EXTRADATA_SIZE) {
2747                 st->codec->extradata_size= i;
2748                 st->codec->extradata= av_malloc(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
2749                 if (!st->codec->extradata)
2750                     return AVERROR(ENOMEM);
2751                 memcpy(st->codec->extradata, pkt->data, st->codec->extradata_size);
2752                 memset(st->codec->extradata + i, 0, FF_INPUT_BUFFER_PADDING_SIZE);
2753             }
2754         }
2755
2756         /* if still no information, we try to open the codec and to
2757            decompress the frame. We try to avoid that in most cases as
2758            it takes longer and uses more memory. For MPEG-4, we need to
2759            decompress for QuickTime.
2760
2761            If CODEC_CAP_CHANNEL_CONF is set this will force decoding of at
2762            least one frame of codec data, this makes sure the codec initializes
2763            the channel configuration and does not only trust the values from the container.
2764         */
2765         try_decode_frame(st, pkt, (options && i < orig_nb_streams ) ? &options[i] : NULL);
2766
2767         st->codec_info_nb_frames++;
2768         count++;
2769     }
2770
2771     if (flush_codecs) {
2772         AVPacket empty_pkt = { 0 };
2773         int err = 0;
2774         av_init_packet(&empty_pkt);
2775
2776         ret = -1; /* we could not have all the codec parameters before EOF */
2777         for(i=0;i<ic->nb_streams;i++) {
2778             const char *errmsg;
2779
2780             st = ic->streams[i];
2781
2782             /* flush the decoders */
2783             if (st->info->found_decoder == 1) {
2784                 do {
2785                     err = try_decode_frame(st, &empty_pkt,
2786                                             (options && i < orig_nb_streams) ?
2787                                             &options[i] : NULL);
2788                 } while (err > 0 && !has_codec_parameters(st, NULL));
2789
2790                 if (err < 0) {
2791                     av_log(ic, AV_LOG_INFO,
2792                         "decoding for stream %d failed\n", st->index);
2793                 }
2794             }
2795
2796             if (!has_codec_parameters(st, &errmsg)) {
2797                 char buf[256];
2798                 avcodec_string(buf, sizeof(buf), st->codec, 0);
2799                 av_log(ic, AV_LOG_WARNING,
2800                        "Could not find codec parameters for stream %d (%s): %s\n"
2801                        "Consider increasing the value for the 'analyzeduration' and 'probesize' options\n",
2802                        i, buf, errmsg);
2803             } else {
2804                 ret = 0;
2805             }
2806         }
2807     }
2808
2809     // close codecs which were opened in try_decode_frame()
2810     for(i=0;i<ic->nb_streams;i++) {
2811         st = ic->streams[i];
2812         avcodec_close(st->codec);
2813     }
2814     for(i=0;i<ic->nb_streams;i++) {
2815         st = ic->streams[i];
2816         if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
2817             if(st->codec->codec_id == AV_CODEC_ID_RAWVIDEO && !st->codec->codec_tag && !st->codec->bits_per_coded_sample){
2818                 uint32_t tag= avcodec_pix_fmt_to_codec_tag(st->codec->pix_fmt);
2819                 if(ff_find_pix_fmt(ff_raw_pix_fmt_tags, tag) == st->codec->pix_fmt)
2820                     st->codec->codec_tag= tag;
2821             }
2822
2823             /* estimate average framerate if not set by demuxer */
2824             if (st->info->codec_info_duration_fields && !st->avg_frame_rate.num && st->info->codec_info_duration) {
2825                 int      best_fps = 0;
2826                 double best_error = 0.01;
2827
2828                 av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den,
2829                           st->info->codec_info_duration_fields*(int64_t)st->time_base.den,
2830                           st->info->codec_info_duration*2*(int64_t)st->time_base.num, 60000);
2831
2832                 /* round guessed framerate to a "standard" framerate if it's
2833                  * within 1% of the original estimate*/
2834                 for (j = 1; j < MAX_STD_TIMEBASES; j++) {
2835                     AVRational std_fps = { get_std_framerate(j), 12*1001 };
2836                     double error = fabs(av_q2d(st->avg_frame_rate) / av_q2d(std_fps) - 1);
2837
2838                     if (error < best_error) {
2839                         best_error = error;
2840                         best_fps   = std_fps.num;
2841                     }
2842                 }
2843                 if (best_fps) {
2844                     av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den,
2845                               best_fps, 12*1001, INT_MAX);
2846                 }
2847             }
2848             // the check for tb_unreliable() is not completely correct, since this is not about handling
2849             // a unreliable/inexact time base, but a time base that is finer than necessary, as e.g.
2850             // ipmovie.c produces.
2851             if (tb_unreliable(st->codec) && 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)
2852                 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);
2853             if (st->info->duration_count && !st->r_frame_rate.num
2854                 && tb_unreliable(st->codec)) {
2855                 int num = 0;
2856                 double best_error= 0.01;
2857
2858                 for (j=0; j<FF_ARRAY_ELEMS(st->info->duration_error[0][0]); j++) {
2859                     int k;
2860
2861                     if(st->info->codec_info_duration && st->info->codec_info_duration*av_q2d(st->time_base) < (1001*12.0)/get_std_framerate(j))
2862                         continue;
2863                     if(!st->info->codec_info_duration && 1.0 < (1001*12.0)/get_std_framerate(j))
2864                         continue;
2865                     for(k=0; k<2; k++){
2866                         int n= st->info->duration_count;
2867                         double a= st->info->duration_error[k][0][j] / n;
2868                         double error= st->info->duration_error[k][1][j]/n - a*a;
2869
2870                         if(error < best_error && best_error> 0.000000001){
2871                             best_error= error;
2872                             num = get_std_framerate(j);
2873                         }
2874                         if(error < 0.02)
2875                             av_log(NULL, AV_LOG_DEBUG, "rfps: %f %f\n", get_std_framerate(j) / 12.0/1001, error);
2876                     }
2877                 }
2878                 // do not increase frame rate by more than 1 % in order to match a standard rate.
2879                 if (num && (!st->r_frame_rate.num || (double)num/(12*1001) < 1.01 * av_q2d(st->r_frame_rate)))
2880                     av_reduce(&st->r_frame_rate.num, &st->r_frame_rate.den, num, 12*1001, INT_MAX);
2881             }
2882
2883             if (!st->r_frame_rate.num){
2884                 if(    st->codec->time_base.den * (int64_t)st->time_base.num
2885                     <= st->codec->time_base.num * st->codec->ticks_per_frame * (int64_t)st->time_base.den){
2886                     st->r_frame_rate.num = st->codec->time_base.den;
2887                     st->r_frame_rate.den = st->codec->time_base.num * st->codec->ticks_per_frame;
2888                 }else{
2889                     st->r_frame_rate.num = st->time_base.den;
2890                     st->r_frame_rate.den = st->time_base.num;
2891                 }
2892             }
2893         }else if(st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
2894             if(!st->codec->bits_per_coded_sample)
2895                 st->codec->bits_per_coded_sample= av_get_bits_per_sample(st->codec->codec_id);
2896             // set stream disposition based on audio service type
2897             switch (st->codec->audio_service_type) {
2898             case AV_AUDIO_SERVICE_TYPE_EFFECTS:
2899                 st->disposition = AV_DISPOSITION_CLEAN_EFFECTS;    break;
2900             case AV_AUDIO_SERVICE_TYPE_VISUALLY_IMPAIRED:
2901                 st->disposition = AV_DISPOSITION_VISUAL_IMPAIRED;  break;
2902             case AV_AUDIO_SERVICE_TYPE_HEARING_IMPAIRED:
2903                 st->disposition = AV_DISPOSITION_HEARING_IMPAIRED; break;
2904             case AV_AUDIO_SERVICE_TYPE_COMMENTARY:
2905                 st->disposition = AV_DISPOSITION_COMMENT;          break;
2906             case AV_AUDIO_SERVICE_TYPE_KARAOKE:
2907                 st->disposition = AV_DISPOSITION_KARAOKE;          break;
2908             }
2909         }
2910     }
2911
2912     if(ic->probesize)
2913     estimate_timings(ic, old_offset);
2914
2915     compute_chapters_end(ic);
2916
2917  find_stream_info_err:
2918     for (i=0; i < ic->nb_streams; i++) {
2919         if (ic->streams[i]->codec)
2920             ic->streams[i]->codec->thread_count = 0;
2921         av_freep(&ic->streams[i]->info);
2922     }
2923     if(ic->pb)
2924         av_log(ic, AV_LOG_DEBUG, "File position after avformat_find_stream_info() is %"PRId64"\n", avio_tell(ic->pb));
2925     return ret;
2926 }
2927
2928 AVProgram *av_find_program_from_stream(AVFormatContext *ic, AVProgram *last, int s)
2929 {
2930     int i, j;
2931
2932     for (i = 0; i < ic->nb_programs; i++) {
2933         if (ic->programs[i] == last) {
2934             last = NULL;
2935         } else {
2936             if (!last)
2937                 for (j = 0; j < ic->programs[i]->nb_stream_indexes; j++)
2938                     if (ic->programs[i]->stream_index[j] == s)
2939                         return ic->programs[i];
2940         }
2941     }
2942     return NULL;
2943 }
2944
2945 int av_find_best_stream(AVFormatContext *ic,
2946                         enum AVMediaType type,
2947                         int wanted_stream_nb,
2948                         int related_stream,
2949                         AVCodec **decoder_ret,
2950                         int flags)
2951 {
2952     int i, nb_streams = ic->nb_streams;
2953     int ret = AVERROR_STREAM_NOT_FOUND, best_count = -1, best_bitrate = -1, best_multiframe = -1, count, bitrate, multiframe;
2954     unsigned *program = NULL;
2955     AVCodec *decoder = NULL, *best_decoder = NULL;
2956
2957     if (related_stream >= 0 && wanted_stream_nb < 0) {
2958         AVProgram *p = av_find_program_from_stream(ic, NULL, related_stream);
2959         if (p) {
2960             program = p->stream_index;
2961             nb_streams = p->nb_stream_indexes;
2962         }
2963     }
2964     for (i = 0; i < nb_streams; i++) {
2965         int real_stream_index = program ? program[i] : i;
2966         AVStream *st = ic->streams[real_stream_index];
2967         AVCodecContext *avctx = st->codec;
2968         if (avctx->codec_type != type)
2969             continue;
2970         if (wanted_stream_nb >= 0 && real_stream_index != wanted_stream_nb)
2971             continue;
2972         if (st->disposition & (AV_DISPOSITION_HEARING_IMPAIRED|AV_DISPOSITION_VISUAL_IMPAIRED))
2973             continue;
2974         if (decoder_ret) {
2975             decoder = avcodec_find_decoder(st->codec->codec_id);
2976             if (!decoder) {
2977                 if (ret < 0)
2978                     ret = AVERROR_DECODER_NOT_FOUND;
2979                 continue;
2980             }
2981         }
2982         count = st->codec_info_nb_frames;
2983         bitrate = avctx->bit_rate;
2984         multiframe = FFMIN(5, count);
2985         if ((best_multiframe >  multiframe) ||
2986             (best_multiframe == multiframe && best_bitrate >  bitrate) ||
2987             (best_multiframe == multiframe && best_bitrate == bitrate && best_count >= count))
2988             continue;
2989         best_count = count;
2990         best_bitrate = bitrate;
2991         best_multiframe = multiframe;
2992         ret = real_stream_index;
2993         best_decoder = decoder;
2994         if (program && i == nb_streams - 1 && ret < 0) {
2995             program = NULL;
2996             nb_streams = ic->nb_streams;
2997             i = 0; /* no related stream found, try again with everything */
2998         }
2999     }
3000     if (decoder_ret)
3001         *decoder_ret = best_decoder;
3002     return ret;
3003 }
3004
3005 /*******************************************************/
3006
3007 int av_read_play(AVFormatContext *s)
3008 {
3009     if (s->iformat->read_play)
3010         return s->iformat->read_play(s);
3011     if (s->pb)
3012         return avio_pause(s->pb, 0);
3013     return AVERROR(ENOSYS);
3014 }
3015
3016 int av_read_pause(AVFormatContext *s)
3017 {
3018     if (s->iformat->read_pause)
3019         return s->iformat->read_pause(s);
3020     if (s->pb)
3021         return avio_pause(s->pb, 1);
3022     return AVERROR(ENOSYS);
3023 }
3024
3025 void ff_free_stream(AVFormatContext *s, AVStream *st){
3026     av_assert0(s->nb_streams>0);
3027     av_assert0(s->streams[ s->nb_streams-1 ] == st);
3028
3029     if (st->parser) {
3030         av_parser_close(st->parser);
3031     }
3032     if (st->attached_pic.data)
3033         av_free_packet(&st->attached_pic);
3034     av_dict_free(&st->metadata);
3035     av_freep(&st->index_entries);
3036     av_freep(&st->codec->extradata);
3037     av_freep(&st->codec->subtitle_header);
3038     av_freep(&st->codec);
3039     av_freep(&st->priv_data);
3040     av_freep(&st->info);
3041     av_freep(&st->probe_data.buf);
3042     av_freep(&s->streams[ --s->nb_streams ]);
3043 }
3044
3045 void avformat_free_context(AVFormatContext *s)
3046 {
3047     int i;
3048
3049     if (!s)
3050         return;
3051
3052     av_opt_free(s);
3053     if (s->iformat && s->iformat->priv_class && s->priv_data)
3054         av_opt_free(s->priv_data);
3055
3056     for(i=s->nb_streams-1; i>=0; i--) {
3057         ff_free_stream(s, s->streams[i]);
3058     }
3059     for(i=s->nb_programs-1; i>=0; i--) {
3060         av_dict_free(&s->programs[i]->metadata);
3061         av_freep(&s->programs[i]->stream_index);
3062         av_freep(&s->programs[i]);
3063     }
3064     av_freep(&s->programs);
3065     av_freep(&s->priv_data);
3066     while(s->nb_chapters--) {
3067         av_dict_free(&s->chapters[s->nb_chapters]->metadata);
3068         av_freep(&s->chapters[s->nb_chapters]);
3069     }
3070     av_freep(&s->chapters);
3071     av_dict_free(&s->metadata);
3072     av_freep(&s->streams);
3073     av_free(s);
3074 }
3075
3076 #if FF_API_CLOSE_INPUT_FILE
3077 void av_close_input_file(AVFormatContext *s)
3078 {
3079     avformat_close_input(&s);
3080 }
3081 #endif
3082
3083 void avformat_close_input(AVFormatContext **ps)
3084 {
3085     AVFormatContext *s = *ps;
3086     AVIOContext *pb = s->pb;
3087
3088     if ((s->iformat && s->iformat->flags & AVFMT_NOFILE) ||
3089         (s->flags & AVFMT_FLAG_CUSTOM_IO))
3090         pb = NULL;
3091
3092     flush_packet_queue(s);
3093
3094     if (s->iformat) {
3095         if (s->iformat->read_close)
3096             s->iformat->read_close(s);
3097     }
3098
3099     avformat_free_context(s);
3100
3101     *ps = NULL;
3102
3103     avio_close(pb);
3104 }
3105
3106 #if FF_API_NEW_STREAM
3107 AVStream *av_new_stream(AVFormatContext *s, int id)
3108 {
3109     AVStream *st = avformat_new_stream(s, NULL);
3110     if (st)
3111         st->id = id;
3112     return st;
3113 }
3114 #endif
3115
3116 AVStream *avformat_new_stream(AVFormatContext *s, const AVCodec *c)
3117 {
3118     AVStream *st;
3119     int i;
3120     AVStream **streams;
3121
3122     if (s->nb_streams >= INT_MAX/sizeof(*streams))
3123         return NULL;
3124     streams = av_realloc(s->streams, (s->nb_streams + 1) * sizeof(*streams));
3125     if (!streams)
3126         return NULL;
3127     s->streams = streams;
3128
3129     st = av_mallocz(sizeof(AVStream));
3130     if (!st)
3131         return NULL;
3132     if (!(st->info = av_mallocz(sizeof(*st->info)))) {
3133         av_free(st);
3134         return NULL;
3135     }
3136     st->info->last_dts = AV_NOPTS_VALUE;
3137
3138     st->codec = avcodec_alloc_context3(c);
3139     if (s->iformat) {
3140         /* no default bitrate if decoding */
3141         st->codec->bit_rate = 0;
3142     }
3143     st->index = s->nb_streams;
3144     st->start_time = AV_NOPTS_VALUE;
3145     st->duration = AV_NOPTS_VALUE;
3146         /* we set the current DTS to 0 so that formats without any timestamps
3147            but durations get some timestamps, formats with some unknown
3148            timestamps have their first few packets buffered and the
3149            timestamps corrected before they are returned to the user */
3150     st->cur_dts = s->iformat ? RELATIVE_TS_BASE : 0;
3151     st->first_dts = AV_NOPTS_VALUE;
3152     st->probe_packets = MAX_PROBE_PACKETS;
3153
3154     /* default pts setting is MPEG-like */
3155     avpriv_set_pts_info(st, 33, 1, 90000);
3156     st->last_IP_pts = AV_NOPTS_VALUE;
3157     for(i=0; i<MAX_REORDER_DELAY+1; i++)
3158         st->pts_buffer[i]= AV_NOPTS_VALUE;
3159     st->reference_dts = AV_NOPTS_VALUE;
3160
3161     st->sample_aspect_ratio = (AVRational){0,1};
3162
3163 #if FF_API_R_FRAME_RATE
3164     st->info->last_dts      = AV_NOPTS_VALUE;
3165 #endif
3166     st->info->fps_first_dts = AV_NOPTS_VALUE;
3167     st->info->fps_last_dts  = AV_NOPTS_VALUE;
3168
3169     s->streams[s->nb_streams++] = st;
3170     return st;
3171 }
3172
3173 AVProgram *av_new_program(AVFormatContext *ac, int id)
3174 {
3175     AVProgram *program=NULL;
3176     int i;
3177
3178     av_dlog(ac, "new_program: id=0x%04x\n", id);
3179
3180     for(i=0; i<ac->nb_programs; i++)
3181         if(ac->programs[i]->id == id)
3182             program = ac->programs[i];
3183
3184     if(!program){
3185         program = av_mallocz(sizeof(AVProgram));
3186         if (!program)
3187             return NULL;
3188         dynarray_add(&ac->programs, &ac->nb_programs, program);
3189         program->discard = AVDISCARD_NONE;
3190     }
3191     program->id = id;
3192
3193     program->start_time =
3194     program->end_time   = AV_NOPTS_VALUE;
3195
3196     return program;
3197 }
3198
3199 AVChapter *avpriv_new_chapter(AVFormatContext *s, int id, AVRational time_base, int64_t start, int64_t end, const char *title)
3200 {
3201     AVChapter *chapter = NULL;
3202     int i;
3203
3204     for(i=0; i<s->nb_chapters; i++)
3205         if(s->chapters[i]->id == id)
3206             chapter = s->chapters[i];
3207
3208     if(!chapter){
3209         chapter= av_mallocz(sizeof(AVChapter));
3210         if(!chapter)
3211             return NULL;
3212         dynarray_add(&s->chapters, &s->nb_chapters, chapter);
3213     }
3214     av_dict_set(&chapter->metadata, "title", title, 0);
3215     chapter->id    = id;
3216     chapter->time_base= time_base;
3217     chapter->start = start;
3218     chapter->end   = end;
3219
3220     return chapter;
3221 }
3222
3223 void ff_program_add_stream_index(AVFormatContext *ac, int progid, unsigned int idx)
3224 {
3225     int i, j;
3226     AVProgram *program=NULL;
3227     void *tmp;
3228
3229     if (idx >= ac->nb_streams) {
3230         av_log(ac, AV_LOG_ERROR, "stream index %d is not valid\n", idx);
3231         return;
3232     }
3233
3234     for(i=0; i<ac->nb_programs; i++){
3235         if(ac->programs[i]->id != progid)
3236             continue;
3237         program = ac->programs[i];
3238         for(j=0; j<program->nb_stream_indexes; j++)
3239             if(program->stream_index[j] == idx)
3240                 return;
3241
3242         tmp = av_realloc(program->stream_index, sizeof(unsigned int)*(program->nb_stream_indexes+1));
3243         if(!tmp)
3244             return;
3245         program->stream_index = tmp;
3246         program->stream_index[program->nb_stream_indexes++] = idx;
3247         return;
3248     }
3249 }
3250
3251 static void print_fps(double d, const char *postfix){
3252     uint64_t v= lrintf(d*100);
3253     if     (v% 100      ) av_log(NULL, AV_LOG_INFO, ", %3.2f %s", d, postfix);
3254     else if(v%(100*1000)) av_log(NULL, AV_LOG_INFO, ", %1.0f %s", d, postfix);
3255     else                  av_log(NULL, AV_LOG_INFO, ", %1.0fk %s", d/1000, postfix);
3256 }
3257
3258 static void dump_metadata(void *ctx, AVDictionary *m, const char *indent)
3259 {
3260     if(m && !(av_dict_count(m) == 1 && av_dict_get(m, "language", NULL, 0))){
3261         AVDictionaryEntry *tag=NULL;
3262
3263         av_log(ctx, AV_LOG_INFO, "%sMetadata:\n", indent);
3264         while((tag=av_dict_get(m, "", tag, AV_DICT_IGNORE_SUFFIX))) {
3265             if(strcmp("language", tag->key)){
3266                 const char *p = tag->value;
3267                 av_log(ctx, AV_LOG_INFO, "%s  %-16s: ", indent, tag->key);
3268                 while(*p) {
3269                     char tmp[256];
3270                     size_t len = strcspn(p, "\x8\xa\xb\xc\xd");
3271                     av_strlcpy(tmp, p, FFMIN(sizeof(tmp), len+1));
3272                     av_log(ctx, AV_LOG_INFO, "%s", tmp);
3273                     p += len;
3274                     if (*p == 0xd) av_log(ctx, AV_LOG_INFO, " ");
3275                     if (*p == 0xa) av_log(ctx, AV_LOG_INFO, "\n%s  %-16s: ", indent, "");
3276                     if (*p) p++;
3277                 }
3278                 av_log(ctx, AV_LOG_INFO, "\n");
3279             }
3280         }
3281     }
3282 }
3283
3284 /* "user interface" functions */
3285 static void dump_stream_format(AVFormatContext *ic, int i, int index, int is_output)
3286 {
3287     char buf[256];
3288     int flags = (is_output ? ic->oformat->flags : ic->iformat->flags);
3289     AVStream *st = ic->streams[i];
3290     int g = av_gcd(st->time_base.num, st->time_base.den);
3291     AVDictionaryEntry *lang = av_dict_get(st->metadata, "language", NULL, 0);
3292     avcodec_string(buf, sizeof(buf), st->codec, is_output);
3293     av_log(NULL, AV_LOG_INFO, "    Stream #%d:%d", index, i);
3294     /* the pid is an important information, so we display it */
3295     /* XXX: add a generic system */
3296     if (flags & AVFMT_SHOW_IDS)
3297         av_log(NULL, AV_LOG_INFO, "[0x%x]", st->id);
3298     if (lang)
3299         av_log(NULL, AV_LOG_INFO, "(%s)", lang->value);
3300     av_log(NULL, AV_LOG_DEBUG, ", %d, %d/%d", st->codec_info_nb_frames, st->time_base.num/g, st->time_base.den/g);
3301     av_log(NULL, AV_LOG_INFO, ": %s", buf);
3302     if (st->sample_aspect_ratio.num && // default
3303         av_cmp_q(st->sample_aspect_ratio, st->codec->sample_aspect_ratio)) {
3304         AVRational display_aspect_ratio;
3305         av_reduce(&display_aspect_ratio.num, &display_aspect_ratio.den,
3306                   st->codec->width*st->sample_aspect_ratio.num,
3307                   st->codec->height*st->sample_aspect_ratio.den,
3308                   1024*1024);
3309         av_log(NULL, AV_LOG_INFO, ", SAR %d:%d DAR %d:%d",
3310                  st->sample_aspect_ratio.num, st->sample_aspect_ratio.den,
3311                  display_aspect_ratio.num, display_aspect_ratio.den);
3312     }
3313     if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO){
3314         if(st->avg_frame_rate.den && st->avg_frame_rate.num)
3315             print_fps(av_q2d(st->avg_frame_rate), "fps");
3316 #if FF_API_R_FRAME_RATE
3317         if(st->r_frame_rate.den && st->r_frame_rate.num)
3318             print_fps(av_q2d(st->r_frame_rate), "tbr");
3319 #endif
3320         if(st->time_base.den && st->time_base.num)
3321             print_fps(1/av_q2d(st->time_base), "tbn");
3322         if(st->codec->time_base.den && st->codec->time_base.num)
3323             print_fps(1/av_q2d(st->codec->time_base), "tbc");
3324     }
3325     if (st->disposition & AV_DISPOSITION_DEFAULT)
3326         av_log(NULL, AV_LOG_INFO, " (default)");
3327     if (st->disposition & AV_DISPOSITION_DUB)
3328         av_log(NULL, AV_LOG_INFO, " (dub)");
3329     if (st->disposition & AV_DISPOSITION_ORIGINAL)
3330         av_log(NULL, AV_LOG_INFO, " (original)");
3331     if (st->disposition & AV_DISPOSITION_COMMENT)
3332         av_log(NULL, AV_LOG_INFO, " (comment)");
3333     if (st->disposition & AV_DISPOSITION_LYRICS)
3334         av_log(NULL, AV_LOG_INFO, " (lyrics)");
3335     if (st->disposition & AV_DISPOSITION_KARAOKE)
3336         av_log(NULL, AV_LOG_INFO, " (karaoke)");
3337     if (st->disposition & AV_DISPOSITION_FORCED)
3338         av_log(NULL, AV_LOG_INFO, " (forced)");
3339     if (st->disposition & AV_DISPOSITION_HEARING_IMPAIRED)
3340         av_log(NULL, AV_LOG_INFO, " (hearing impaired)");
3341     if (st->disposition & AV_DISPOSITION_VISUAL_IMPAIRED)
3342         av_log(NULL, AV_LOG_INFO, " (visual impaired)");
3343     if (st->disposition & AV_DISPOSITION_CLEAN_EFFECTS)
3344         av_log(NULL, AV_LOG_INFO, " (clean effects)");
3345     av_log(NULL, AV_LOG_INFO, "\n");
3346     dump_metadata(NULL, st->metadata, "    ");
3347 }
3348
3349 void av_dump_format(AVFormatContext *ic,
3350                     int index,
3351                     const char *url,
3352                     int is_output)
3353 {
3354     int i;
3355     uint8_t *printed = ic->nb_streams ? av_mallocz(ic->nb_streams) : NULL;
3356     if (ic->nb_streams && !printed)
3357         return;
3358
3359     av_log(NULL, AV_LOG_INFO, "%s #%d, %s, %s '%s':\n",
3360             is_output ? "Output" : "Input",
3361             index,
3362             is_output ? ic->oformat->name : ic->iformat->name,
3363             is_output ? "to" : "from", url);
3364     dump_metadata(NULL, ic->metadata, "  ");
3365     if (!is_output) {
3366         av_log(NULL, AV_LOG_INFO, "  Duration: ");
3367         if (ic->duration != AV_NOPTS_VALUE) {
3368             int hours, mins, secs, us;
3369             int64_t duration = ic->duration + 5000;
3370             secs = duration / AV_TIME_BASE;
3371             us = duration % AV_TIME_BASE;
3372             mins = secs / 60;
3373             secs %= 60;
3374             hours = mins / 60;
3375             mins %= 60;
3376             av_log(NULL, AV_LOG_INFO, "%02d:%02d:%02d.%02d", hours, mins, secs,
3377                    (100 * us) / AV_TIME_BASE);
3378         } else {
3379             av_log(NULL, AV_LOG_INFO, "N/A");
3380         }
3381         if (ic->start_time != AV_NOPTS_VALUE) {
3382             int secs, us;
3383             av_log(NULL, AV_LOG_INFO, ", start: ");
3384             secs = ic->start_time / AV_TIME_BASE;
3385             us = abs(ic->start_time % AV_TIME_BASE);
3386             av_log(NULL, AV_LOG_INFO, "%d.%06d",
3387                    secs, (int)av_rescale(us, 1000000, AV_TIME_BASE));
3388         }
3389         av_log(NULL, AV_LOG_INFO, ", bitrate: ");
3390         if (ic->bit_rate) {
3391             av_log(NULL, AV_LOG_INFO,"%d kb/s", ic->bit_rate / 1000);
3392         } else {
3393             av_log(NULL, AV_LOG_INFO, "N/A");
3394         }
3395         av_log(NULL, AV_LOG_INFO, "\n");
3396     }
3397     for (i = 0; i < ic->nb_chapters; i++) {
3398         AVChapter *ch = ic->chapters[i];
3399         av_log(NULL, AV_LOG_INFO, "    Chapter #%d.%d: ", index, i);
3400         av_log(NULL, AV_LOG_INFO, "start %f, ", ch->start * av_q2d(ch->time_base));
3401         av_log(NULL, AV_LOG_INFO, "end %f\n",   ch->end   * av_q2d(ch->time_base));
3402
3403         dump_metadata(NULL, ch->metadata, "    ");
3404     }
3405     if(ic->nb_programs) {
3406         int j, k, total = 0;
3407         for(j=0; j<ic->nb_programs; j++) {
3408             AVDictionaryEntry *name = av_dict_get(ic->programs[j]->metadata,
3409                                                   "name", NULL, 0);
3410             av_log(NULL, AV_LOG_INFO, "  Program %d %s\n", ic->programs[j]->id,
3411                    name ? name->value : "");
3412             dump_metadata(NULL, ic->programs[j]->metadata, "    ");
3413             for(k=0; k<ic->programs[j]->nb_stream_indexes; k++) {
3414                 dump_stream_format(ic, ic->programs[j]->stream_index[k], index, is_output);
3415                 printed[ic->programs[j]->stream_index[k]] = 1;
3416             }
3417             total += ic->programs[j]->nb_stream_indexes;
3418         }
3419         if (total < ic->nb_streams)
3420             av_log(NULL, AV_LOG_INFO, "  No Program\n");
3421     }
3422     for(i=0;i<ic->nb_streams;i++)
3423         if (!printed[i])
3424             dump_stream_format(ic, i, index, is_output);
3425
3426     av_free(printed);
3427 }
3428
3429 #if FF_API_AV_GETTIME && CONFIG_SHARED && HAVE_SYMVER
3430 FF_SYMVER(int64_t, av_gettime, (void), "LIBAVFORMAT_54")
3431 {
3432     return av_gettime();
3433 }
3434 #endif
3435
3436 uint64_t ff_ntp_time(void)
3437 {
3438   return (av_gettime() / 1000) * 1000 + NTP_OFFSET_US;
3439 }
3440
3441 int av_get_frame_filename(char *buf, int buf_size,
3442                           const char *path, int number)
3443 {
3444     const char *p;
3445     char *q, buf1[20], c;
3446     int nd, len, percentd_found;
3447
3448     q = buf;
3449     p = path;
3450     percentd_found = 0;
3451     for(;;) {
3452         c = *p++;
3453         if (c == '\0')
3454             break;
3455         if (c == '%') {
3456             do {
3457                 nd = 0;
3458                 while (isdigit(*p)) {
3459                     nd = nd * 10 + *p++ - '0';
3460                 }
3461                 c = *p++;
3462             } while (isdigit(c));
3463
3464             switch(c) {
3465             case '%':
3466                 goto addchar;
3467             case 'd':
3468                 if (percentd_found)
3469                     goto fail;
3470                 percentd_found = 1;
3471                 snprintf(buf1, sizeof(buf1), "%0*d", nd, number);
3472                 len = strlen(buf1);
3473                 if ((q - buf + len) > buf_size - 1)
3474                     goto fail;
3475                 memcpy(q, buf1, len);
3476                 q += len;
3477                 break;
3478             default:
3479                 goto fail;
3480             }
3481         } else {
3482         addchar:
3483             if ((q - buf) < buf_size - 1)
3484                 *q++ = c;
3485         }
3486     }
3487     if (!percentd_found)
3488         goto fail;
3489     *q = '\0';
3490     return 0;
3491  fail:
3492     *q = '\0';
3493     return -1;
3494 }
3495
3496 static void hex_dump_internal(void *avcl, FILE *f, int level,
3497                               const uint8_t *buf, int size)
3498 {
3499     int len, i, j, c;
3500 #undef fprintf
3501 #define PRINT(...) do { if (!f) av_log(avcl, level, __VA_ARGS__); else fprintf(f, __VA_ARGS__); } while(0)
3502
3503     for(i=0;i<size;i+=16) {
3504         len = size - i;
3505         if (len > 16)
3506             len = 16;
3507         PRINT("%08x ", i);
3508         for(j=0;j<16;j++) {
3509             if (j < len)
3510                 PRINT(" %02x", buf[i+j]);
3511             else
3512                 PRINT("   ");
3513         }
3514         PRINT(" ");
3515         for(j=0;j<len;j++) {
3516             c = buf[i+j];
3517             if (c < ' ' || c > '~')
3518                 c = '.';
3519             PRINT("%c", c);
3520         }
3521         PRINT("\n");
3522     }
3523 #undef PRINT
3524 }
3525
3526 void av_hex_dump(FILE *f, const uint8_t *buf, int size)
3527 {
3528     hex_dump_internal(NULL, f, 0, buf, size);
3529 }
3530
3531 void av_hex_dump_log(void *avcl, int level, const uint8_t *buf, int size)
3532 {
3533     hex_dump_internal(avcl, NULL, level, buf, size);
3534 }
3535
3536 static void pkt_dump_internal(void *avcl, FILE *f, int level, AVPacket *pkt, int dump_payload, AVRational time_base)
3537 {
3538 #undef fprintf
3539 #define PRINT(...) do { if (!f) av_log(avcl, level, __VA_ARGS__); else fprintf(f, __VA_ARGS__); } while(0)
3540     PRINT("stream #%d:\n", pkt->stream_index);
3541     PRINT("  keyframe=%d\n", ((pkt->flags & AV_PKT_FLAG_KEY) != 0));
3542     PRINT("  duration=%0.3f\n", pkt->duration * av_q2d(time_base));
3543     /* DTS is _always_ valid after av_read_frame() */
3544     PRINT("  dts=");
3545     if (pkt->dts == AV_NOPTS_VALUE)
3546         PRINT("N/A");
3547     else
3548         PRINT("%0.3f", pkt->dts * av_q2d(time_base));
3549     /* PTS may not be known if B-frames are present. */
3550     PRINT("  pts=");
3551     if (pkt->pts == AV_NOPTS_VALUE)
3552         PRINT("N/A");
3553     else
3554         PRINT("%0.3f", pkt->pts * av_q2d(time_base));
3555     PRINT("\n");
3556     PRINT("  size=%d\n", pkt->size);
3557 #undef PRINT
3558     if (dump_payload)
3559         av_hex_dump(f, pkt->data, pkt->size);
3560 }
3561
3562 #if FF_API_PKT_DUMP
3563 void av_pkt_dump(FILE *f, AVPacket *pkt, int dump_payload)
3564 {
3565     AVRational tb = { 1, AV_TIME_BASE };
3566     pkt_dump_internal(NULL, f, 0, pkt, dump_payload, tb);
3567 }
3568 #endif
3569
3570 void av_pkt_dump2(FILE *f, AVPacket *pkt, int dump_payload, AVStream *st)
3571 {
3572     pkt_dump_internal(NULL, f, 0, pkt, dump_payload, st->time_base);
3573 }
3574
3575 #if FF_API_PKT_DUMP
3576 void av_pkt_dump_log(void *avcl, int level, AVPacket *pkt, int dump_payload)
3577 {
3578     AVRational tb = { 1, AV_TIME_BASE };
3579     pkt_dump_internal(avcl, NULL, level, pkt, dump_payload, tb);
3580 }
3581 #endif
3582
3583 void av_pkt_dump_log2(void *avcl, int level, AVPacket *pkt, int dump_payload,
3584                       AVStream *st)
3585 {
3586     pkt_dump_internal(avcl, NULL, level, pkt, dump_payload, st->time_base);
3587 }
3588
3589 void av_url_split(char *proto, int proto_size,
3590                   char *authorization, int authorization_size,
3591                   char *hostname, int hostname_size,
3592                   int *port_ptr,
3593                   char *path, int path_size,
3594                   const char *url)
3595 {
3596     const char *p, *ls, *ls2, *at, *at2, *col, *brk;
3597
3598     if (port_ptr)               *port_ptr = -1;
3599     if (proto_size > 0)         proto[0] = 0;
3600     if (authorization_size > 0) authorization[0] = 0;
3601     if (hostname_size > 0)      hostname[0] = 0;
3602     if (path_size > 0)          path[0] = 0;
3603
3604     /* parse protocol */
3605     if ((p = strchr(url, ':'))) {
3606         av_strlcpy(proto, url, FFMIN(proto_size, p + 1 - url));
3607         p++; /* skip ':' */
3608         if (*p == '/') p++;
3609         if (*p == '/') p++;
3610     } else {
3611         /* no protocol means plain filename */
3612         av_strlcpy(path, url, path_size);
3613         return;
3614     }
3615
3616     /* separate path from hostname */
3617     ls = strchr(p, '/');
3618     ls2 = strchr(p, '?');
3619     if(!ls)
3620         ls = ls2;
3621     else if (ls && ls2)
3622         ls = FFMIN(ls, ls2);
3623     if(ls)
3624         av_strlcpy(path, ls, path_size);
3625     else
3626         ls = &p[strlen(p)]; // XXX
3627
3628     /* the rest is hostname, use that to parse auth/port */
3629     if (ls != p) {
3630         /* authorization (user[:pass]@hostname) */
3631         at2 = p;
3632         while ((at = strchr(p, '@')) && at < ls) {
3633             av_strlcpy(authorization, at2,
3634                        FFMIN(authorization_size, at + 1 - at2));
3635             p = at + 1; /* skip '@' */
3636         }
3637
3638         if (*p == '[' && (brk = strchr(p, ']')) && brk < ls) {
3639             /* [host]:port */
3640             av_strlcpy(hostname, p + 1,
3641                        FFMIN(hostname_size, brk - p));
3642             if (brk[1] == ':' && port_ptr)
3643                 *port_ptr = atoi(brk + 2);
3644         } else if ((col = strchr(p, ':')) && col < ls) {
3645             av_strlcpy(hostname, p,
3646                        FFMIN(col + 1 - p, hostname_size));
3647             if (port_ptr) *port_ptr = atoi(col + 1);
3648         } else
3649             av_strlcpy(hostname, p,
3650                        FFMIN(ls + 1 - p, hostname_size));
3651     }
3652 }
3653
3654 char *ff_data_to_hex(char *buff, const uint8_t *src, int s, int lowercase)
3655 {
3656     int i;
3657     static const char hex_table_uc[16] = { '0', '1', '2', '3',
3658                                            '4', '5', '6', '7',
3659                                            '8', '9', 'A', 'B',
3660                                            'C', 'D', 'E', 'F' };
3661     static const char hex_table_lc[16] = { '0', '1', '2', '3',
3662                                            '4', '5', '6', '7',
3663                                            '8', '9', 'a', 'b',
3664                                            'c', 'd', 'e', 'f' };
3665     const char *hex_table = lowercase ? hex_table_lc : hex_table_uc;
3666
3667     for(i = 0; i < s; i++) {
3668         buff[i * 2]     = hex_table[src[i] >> 4];
3669         buff[i * 2 + 1] = hex_table[src[i] & 0xF];
3670     }
3671
3672     return buff;
3673 }
3674
3675 int ff_hex_to_data(uint8_t *data, const char *p)
3676 {
3677     int c, len, v;
3678
3679     len = 0;
3680     v = 1;
3681     for (;;) {
3682         p += strspn(p, SPACE_CHARS);
3683         if (*p == '\0')
3684             break;
3685         c = toupper((unsigned char) *p++);
3686         if (c >= '0' && c <= '9')
3687             c = c - '0';
3688         else if (c >= 'A' && c <= 'F')
3689             c = c - 'A' + 10;
3690         else
3691             break;
3692         v = (v << 4) | c;
3693         if (v & 0x100) {
3694             if (data)
3695                 data[len] = v;
3696             len++;
3697             v = 1;
3698         }
3699     }
3700     return len;
3701 }
3702
3703 #if FF_API_SET_PTS_INFO
3704 void av_set_pts_info(AVStream *s, int pts_wrap_bits,
3705                      unsigned int pts_num, unsigned int pts_den)
3706 {
3707     avpriv_set_pts_info(s, pts_wrap_bits, pts_num, pts_den);
3708 }
3709 #endif
3710
3711 void avpriv_set_pts_info(AVStream *s, int pts_wrap_bits,
3712                          unsigned int pts_num, unsigned int pts_den)
3713 {
3714     AVRational new_tb;
3715     if(av_reduce(&new_tb.num, &new_tb.den, pts_num, pts_den, INT_MAX)){
3716         if(new_tb.num != pts_num)
3717             av_log(NULL, AV_LOG_DEBUG, "st:%d removing common factor %d from timebase\n", s->index, pts_num/new_tb.num);
3718     }else
3719         av_log(NULL, AV_LOG_WARNING, "st:%d has too large timebase, reducing\n", s->index);
3720
3721     if(new_tb.num <= 0 || new_tb.den <= 0) {
3722         av_log(NULL, AV_LOG_ERROR, "Ignoring attempt to set invalid timebase %d/%d for st:%d\n", new_tb.num, new_tb.den, s->index);
3723         return;
3724     }
3725     s->time_base = new_tb;
3726     av_codec_set_pkt_timebase(s->codec, new_tb);
3727     s->pts_wrap_bits = pts_wrap_bits;
3728 }
3729
3730 int ff_url_join(char *str, int size, const char *proto,
3731                 const char *authorization, const char *hostname,
3732                 int port, const char *fmt, ...)
3733 {
3734 #if CONFIG_NETWORK
3735     struct addrinfo hints = { 0 }, *ai;
3736 #endif
3737
3738     str[0] = '\0';
3739     if (proto)
3740         av_strlcatf(str, size, "%s://", proto);
3741     if (authorization && authorization[0])
3742         av_strlcatf(str, size, "%s@", authorization);
3743 #if CONFIG_NETWORK && defined(AF_INET6)
3744     /* Determine if hostname is a numerical IPv6 address,
3745      * properly escape it within [] in that case. */
3746     hints.ai_flags = AI_NUMERICHOST;
3747     if (!getaddrinfo(hostname, NULL, &hints, &ai)) {
3748         if (ai->ai_family == AF_INET6) {
3749             av_strlcat(str, "[", size);
3750             av_strlcat(str, hostname, size);
3751             av_strlcat(str, "]", size);
3752         } else {
3753             av_strlcat(str, hostname, size);
3754         }
3755         freeaddrinfo(ai);
3756     } else
3757 #endif
3758         /* Not an IPv6 address, just output the plain string. */
3759         av_strlcat(str, hostname, size);
3760
3761     if (port >= 0)
3762         av_strlcatf(str, size, ":%d", port);
3763     if (fmt) {
3764         va_list vl;
3765         int len = strlen(str);
3766
3767         va_start(vl, fmt);
3768         vsnprintf(str + len, size > len ? size - len : 0, fmt, vl);
3769         va_end(vl);
3770     }
3771     return strlen(str);
3772 }
3773
3774 int ff_write_chained(AVFormatContext *dst, int dst_stream, AVPacket *pkt,
3775                      AVFormatContext *src)
3776 {
3777     AVPacket local_pkt;
3778
3779     local_pkt = *pkt;
3780     local_pkt.stream_index = dst_stream;
3781     if (pkt->pts != AV_NOPTS_VALUE)
3782         local_pkt.pts = av_rescale_q(pkt->pts,
3783                                      src->streams[pkt->stream_index]->time_base,
3784                                      dst->streams[dst_stream]->time_base);
3785     if (pkt->dts != AV_NOPTS_VALUE)
3786         local_pkt.dts = av_rescale_q(pkt->dts,
3787                                      src->streams[pkt->stream_index]->time_base,
3788                                      dst->streams[dst_stream]->time_base);
3789     return av_write_frame(dst, &local_pkt);
3790 }
3791
3792 void ff_parse_key_value(const char *str, ff_parse_key_val_cb callback_get_buf,
3793                         void *context)
3794 {
3795     const char *ptr = str;
3796
3797     /* Parse key=value pairs. */
3798     for (;;) {
3799         const char *key;
3800         char *dest = NULL, *dest_end;
3801         int key_len, dest_len = 0;
3802
3803         /* Skip whitespace and potential commas. */
3804         while (*ptr && (isspace(*ptr) || *ptr == ','))
3805             ptr++;
3806         if (!*ptr)
3807             break;
3808
3809         key = ptr;
3810
3811         if (!(ptr = strchr(key, '=')))
3812             break;
3813         ptr++;
3814         key_len = ptr - key;
3815
3816         callback_get_buf(context, key, key_len, &dest, &dest_len);
3817         dest_end = dest + dest_len - 1;
3818
3819         if (*ptr == '\"') {
3820             ptr++;
3821             while (*ptr && *ptr != '\"') {
3822                 if (*ptr == '\\') {
3823                     if (!ptr[1])
3824                         break;
3825                     if (dest && dest < dest_end)
3826                         *dest++ = ptr[1];
3827                     ptr += 2;
3828                 } else {
3829                     if (dest && dest < dest_end)
3830                         *dest++ = *ptr;
3831                     ptr++;
3832                 }
3833             }
3834             if (*ptr == '\"')
3835                 ptr++;
3836         } else {
3837             for (; *ptr && !(isspace(*ptr) || *ptr == ','); ptr++)
3838                 if (dest && dest < dest_end)
3839                     *dest++ = *ptr;
3840         }
3841         if (dest)
3842             *dest = 0;
3843     }
3844 }
3845
3846 int ff_find_stream_index(AVFormatContext *s, int id)
3847 {
3848     int i;
3849     for (i = 0; i < s->nb_streams; i++) {
3850         if (s->streams[i]->id == id)
3851             return i;
3852     }
3853     return -1;
3854 }
3855
3856 void ff_make_absolute_url(char *buf, int size, const char *base,
3857                           const char *rel)
3858 {
3859     char *sep, *path_query;
3860     /* Absolute path, relative to the current server */
3861     if (base && strstr(base, "://") && rel[0] == '/') {
3862         if (base != buf)
3863             av_strlcpy(buf, base, size);
3864         sep = strstr(buf, "://");
3865         if (sep) {
3866             /* Take scheme from base url */
3867             if (rel[1] == '/') {
3868                 sep[1] = '\0';
3869             } else {
3870                 /* Take scheme and host from base url */
3871                 sep += 3;
3872                 sep = strchr(sep, '/');
3873                 if (sep)
3874                     *sep = '\0';
3875             }
3876         }
3877         av_strlcat(buf, rel, size);
3878         return;
3879     }
3880     /* If rel actually is an absolute url, just copy it */
3881     if (!base || strstr(rel, "://") || rel[0] == '/') {
3882         av_strlcpy(buf, rel, size);
3883         return;
3884     }
3885     if (base != buf)
3886         av_strlcpy(buf, base, size);
3887
3888     /* Strip off any query string from base */
3889     path_query = strchr(buf, '?');
3890     if (path_query != NULL)
3891         *path_query = '\0';
3892
3893     /* Is relative path just a new query part? */
3894     if (rel[0] == '?') {
3895         av_strlcat(buf, rel, size);
3896         return;
3897     }
3898
3899     /* Remove the file name from the base url */
3900     sep = strrchr(buf, '/');
3901     if (sep)
3902         sep[1] = '\0';
3903     else
3904         buf[0] = '\0';
3905     while (av_strstart(rel, "../", NULL) && sep) {
3906         /* Remove the path delimiter at the end */
3907         sep[0] = '\0';
3908         sep = strrchr(buf, '/');
3909         /* If the next directory name to pop off is "..", break here */
3910         if (!strcmp(sep ? &sep[1] : buf, "..")) {
3911             /* Readd the slash we just removed */
3912             av_strlcat(buf, "/", size);
3913             break;
3914         }
3915         /* Cut off the directory name */
3916         if (sep)
3917             sep[1] = '\0';
3918         else
3919             buf[0] = '\0';
3920         rel += 3;
3921     }
3922     av_strlcat(buf, rel, size);
3923 }
3924
3925 int64_t ff_iso8601_to_unix_time(const char *datestr)
3926 {
3927     struct tm time1 = {0}, time2 = {0};
3928     char *ret1, *ret2;
3929     ret1 = av_small_strptime(datestr, "%Y - %m - %d %H:%M:%S", &time1);
3930     ret2 = av_small_strptime(datestr, "%Y - %m - %dT%H:%M:%S", &time2);
3931     if (ret2 && !ret1)
3932         return av_timegm(&time2);
3933     else
3934         return av_timegm(&time1);
3935 }
3936
3937 int avformat_query_codec(AVOutputFormat *ofmt, enum AVCodecID codec_id, int std_compliance)
3938 {
3939     if (ofmt) {
3940         if (ofmt->query_codec)
3941             return ofmt->query_codec(codec_id, std_compliance);
3942         else if (ofmt->codec_tag)
3943             return !!av_codec_get_tag(ofmt->codec_tag, codec_id);
3944         else if (codec_id == ofmt->video_codec || codec_id == ofmt->audio_codec ||
3945                  codec_id == ofmt->subtitle_codec)
3946             return 1;
3947     }
3948     return AVERROR_PATCHWELCOME;
3949 }
3950
3951 int avformat_network_init(void)
3952 {
3953 #if CONFIG_NETWORK
3954     int ret;
3955     ff_network_inited_globally = 1;
3956     if ((ret = ff_network_init()) < 0)
3957         return ret;
3958     ff_tls_init();
3959 #endif
3960     return 0;
3961 }
3962
3963 int avformat_network_deinit(void)
3964 {
3965 #if CONFIG_NETWORK
3966     ff_network_close();
3967     ff_tls_deinit();
3968 #endif
3969     return 0;
3970 }
3971
3972 int ff_add_param_change(AVPacket *pkt, int32_t channels,
3973                         uint64_t channel_layout, int32_t sample_rate,
3974                         int32_t width, int32_t height)
3975 {
3976     uint32_t flags = 0;
3977     int size = 4;
3978     uint8_t *data;
3979     if (!pkt)
3980         return AVERROR(EINVAL);
3981     if (channels) {
3982         size += 4;
3983         flags |= AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_COUNT;
3984     }
3985     if (channel_layout) {
3986         size += 8;
3987         flags |= AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_LAYOUT;
3988     }
3989     if (sample_rate) {
3990         size += 4;
3991         flags |= AV_SIDE_DATA_PARAM_CHANGE_SAMPLE_RATE;
3992     }
3993     if (width || height) {
3994         size += 8;
3995         flags |= AV_SIDE_DATA_PARAM_CHANGE_DIMENSIONS;
3996     }
3997     data = av_packet_new_side_data(pkt, AV_PKT_DATA_PARAM_CHANGE, size);
3998     if (!data)
3999         return AVERROR(ENOMEM);
4000     bytestream_put_le32(&data, flags);
4001     if (channels)
4002         bytestream_put_le32(&data, channels);
4003     if (channel_layout)
4004         bytestream_put_le64(&data, channel_layout);
4005     if (sample_rate)
4006         bytestream_put_le32(&data, sample_rate);
4007     if (width || height) {
4008         bytestream_put_le32(&data, width);
4009         bytestream_put_le32(&data, height);
4010     }
4011     return 0;
4012 }
4013
4014 const struct AVCodecTag *avformat_get_riff_video_tags(void)
4015 {
4016     return ff_codec_bmp_tags;
4017 }
4018 const struct AVCodecTag *avformat_get_riff_audio_tags(void)
4019 {
4020     return ff_codec_wav_tags;
4021 }
4022
4023 AVRational av_guess_sample_aspect_ratio(AVFormatContext *format, AVStream *stream, AVFrame *frame)
4024 {
4025     AVRational undef = {0, 1};
4026     AVRational stream_sample_aspect_ratio = stream ? stream->sample_aspect_ratio : undef;
4027     AVRational codec_sample_aspect_ratio  = stream && stream->codec ? stream->codec->sample_aspect_ratio : undef;
4028     AVRational frame_sample_aspect_ratio  = frame  ? frame->sample_aspect_ratio  : codec_sample_aspect_ratio;
4029
4030     av_reduce(&stream_sample_aspect_ratio.num, &stream_sample_aspect_ratio.den,
4031                stream_sample_aspect_ratio.num,  stream_sample_aspect_ratio.den, INT_MAX);
4032     if (stream_sample_aspect_ratio.num <= 0 || stream_sample_aspect_ratio.den <= 0)
4033         stream_sample_aspect_ratio = undef;
4034
4035     av_reduce(&frame_sample_aspect_ratio.num, &frame_sample_aspect_ratio.den,
4036                frame_sample_aspect_ratio.num,  frame_sample_aspect_ratio.den, INT_MAX);
4037     if (frame_sample_aspect_ratio.num <= 0 || frame_sample_aspect_ratio.den <= 0)
4038         frame_sample_aspect_ratio = undef;
4039
4040     if (stream_sample_aspect_ratio.num)
4041         return stream_sample_aspect_ratio;
4042     else
4043         return frame_sample_aspect_ratio;
4044 }
4045
4046 int avformat_match_stream_specifier(AVFormatContext *s, AVStream *st,
4047                                     const char *spec)
4048 {
4049     if (*spec <= '9' && *spec >= '0') /* opt:index */
4050         return strtol(spec, NULL, 0) == st->index;
4051     else if (*spec == 'v' || *spec == 'a' || *spec == 's' || *spec == 'd' ||
4052              *spec == 't') { /* opt:[vasdt] */
4053         enum AVMediaType type;
4054
4055         switch (*spec++) {
4056         case 'v': type = AVMEDIA_TYPE_VIDEO;      break;
4057         case 'a': type = AVMEDIA_TYPE_AUDIO;      break;
4058         case 's': type = AVMEDIA_TYPE_SUBTITLE;   break;
4059         case 'd': type = AVMEDIA_TYPE_DATA;       break;
4060         case 't': type = AVMEDIA_TYPE_ATTACHMENT; break;
4061         default:  av_assert0(0);
4062         }
4063         if (type != st->codec->codec_type)
4064             return 0;
4065         if (*spec++ == ':') { /* possibly followed by :index */
4066             int i, index = strtol(spec, NULL, 0);
4067             for (i = 0; i < s->nb_streams; i++)
4068                 if (s->streams[i]->codec->codec_type == type && index-- == 0)
4069                    return i == st->index;
4070             return 0;
4071         }
4072         return 1;
4073     } else if (*spec == 'p' && *(spec + 1) == ':') {
4074         int prog_id, i, j;
4075         char *endptr;
4076         spec += 2;
4077         prog_id = strtol(spec, &endptr, 0);
4078         for (i = 0; i < s->nb_programs; i++) {
4079             if (s->programs[i]->id != prog_id)
4080                 continue;
4081
4082             if (*endptr++ == ':') {
4083                 int stream_idx = strtol(endptr, NULL, 0);
4084                 return stream_idx >= 0 &&
4085                     stream_idx < s->programs[i]->nb_stream_indexes &&
4086                     st->index == s->programs[i]->stream_index[stream_idx];
4087             }
4088
4089             for (j = 0; j < s->programs[i]->nb_stream_indexes; j++)
4090                 if (st->index == s->programs[i]->stream_index[j])
4091                     return 1;
4092         }
4093         return 0;
4094     } else if (*spec == '#') {
4095         int sid;
4096         char *endptr;
4097         sid = strtol(spec + 1, &endptr, 0);
4098         if (!*endptr)
4099             return st->id == sid;
4100     } else if (!*spec) /* empty specifier, matches everything */
4101         return 1;
4102
4103     av_log(s, AV_LOG_ERROR, "Invalid stream specifier: %s.\n", spec);
4104     return AVERROR(EINVAL);
4105 }