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