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