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