Fix format string vulnerability detected by -Wformat-security.
[libav.git] / libavformat / utils.c
CommitLineData
de6d9b64 1/*
2912e87a 2 * various utility functions for use within Libav
19720f15 3 * Copyright (c) 2000, 2001, 2002 Fabrice Bellard
de6d9b64 4 *
2912e87a 5 * This file is part of Libav.
b78e7197 6 *
2912e87a 7 * Libav is free software; you can redistribute it and/or
19720f15
FB
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
b78e7197 10 * version 2.1 of the License, or (at your option) any later version.
de6d9b64 11 *
2912e87a 12 * Libav is distributed in the hope that it will be useful,
de6d9b64 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19720f15
FB
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
de6d9b64 16 *
19720f15 17 * You should have received a copy of the GNU Lesser General Public
2912e87a 18 * License along with Libav; if not, write to the Free Software
5509bffa 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
de6d9b64 20 */
79f43a8c
SS
21
22/* #define DEBUG */
23
8be1c656 24#include "avformat.h"
f1ef2cd9 25#include "avio_internal.h"
f1c80e35 26#include "internal.h"
603a5f04 27#include "libavcodec/internal.h"
3e79c2ad 28#include "libavcodec/bytestream.h"
6ed04040 29#include "libavutil/opt.h"
d2d67e42 30#include "libavutil/dict.h"
603b8bc2 31#include "libavutil/pixdesc.h"
06a7bd9a 32#include "metadata.h"
6612d8cf 33#include "id3v2.h"
6450599e 34#include "libavutil/avassert.h"
245976da 35#include "libavutil/avstring.h"
0ebcdf5c 36#include "libavutil/mathematics.h"
4a835416 37#include "libavutil/parseutils.h"
45da8124 38#include "riff.h"
c26e58e3 39#include "audiointerleave.h"
5cec8971 40#include "url.h"
82e4ac2c
RP
41#include <sys/time.h>
42#include <time.h>
780d7897
MS
43#include <stdarg.h>
44#if CONFIG_NETWORK
45#include "network.h"
46#endif
c5510dd6 47
b754978a
MN
48#undef NDEBUG
49#include <assert.h>
50
e36bdf8b 51/**
ba87f080 52 * @file
2912e87a 53 * various utility functions for use within Libav
e36bdf8b
DK
54 */
55
c97429e2
SS
56unsigned avformat_version(void)
57{
58 return LIBAVFORMAT_VERSION_INT;
59}
60
41600690 61const char *avformat_configuration(void)
c1736936 62{
29ba0911 63 return LIBAV_CONFIGURATION;
c1736936
DB
64}
65
41600690 66const char *avformat_license(void)
c1736936
DB
67{
68#define LICENSE_PREFIX "libavformat license: "
a03be6e1 69 return LICENSE_PREFIX LIBAV_LICENSE + sizeof(LICENSE_PREFIX) - 1;
c1736936
DB
70}
71
8163c870
SS
72/* fraction handling */
73
74/**
75 * f = val + (num / den) + 0.5.
76 *
77 * 'num' is normalized so that it is such as 0 <= num < den.
78 *
79 * @param f fractional number
80 * @param val integer value
81 * @param num must be >= 0
82 * @param den must be >= 1
83 */
d3bb7191 84static void frac_init(AVFrac *f, int64_t val, int64_t num, int64_t den)
8163c870
SS
85{
86 num += (den >> 1);
87 if (num >= den) {
88 val += num / den;
89 num = num % den;
90 }
91 f->val = val;
92 f->num = num;
93 f->den = den;
94}
95
96/**
97 * Fractional addition to f: f = f + (incr / f->den).
98 *
99 * @param f fractional number
100 * @param incr increment, can be positive or negative
101 */
d3bb7191 102static void frac_add(AVFrac *f, int64_t incr)
8163c870
SS
103{
104 int64_t num, den;
105
106 num = f->num + incr;
107 den = f->den;
108 if (num < 0) {
109 f->val += num / den;
110 num = num % den;
111 if (num < 0) {
112 num += den;
113 f->val--;
114 }
115 } else if (num >= den) {
116 f->val += num / den;
117 num = num % den;
118 }
119 f->num = num;
120}
f3a30e3a 121
a85736f2 122/** head of registered input format linked list */
b4c5acab 123static AVInputFormat *first_iformat = NULL;
a85736f2 124/** head of registered output format linked list */
b4c5acab 125static AVOutputFormat *first_oformat = NULL;
de6d9b64 126
84be6e72
MN
127AVInputFormat *av_iformat_next(AVInputFormat *f)
128{
129 if(f) return f->next;
130 else return first_iformat;
131}
132
133AVOutputFormat *av_oformat_next(AVOutputFormat *f)
134{
135 if(f) return f->next;
136 else return first_oformat;
137}
138
b9a281db 139void av_register_input_format(AVInputFormat *format)
de6d9b64 140{
b9a281db
FB
141 AVInputFormat **p;
142 p = &first_iformat;
143 while (*p != NULL) p = &(*p)->next;
144 *p = format;
145 format->next = NULL;
146}
147
148void av_register_output_format(AVOutputFormat *format)
149{
150 AVOutputFormat **p;
151 p = &first_oformat;
de6d9b64
FB
152 while (*p != NULL) p = &(*p)->next;
153 *p = format;
154 format->next = NULL;
155}
156
8eb631fa
SS
157int av_match_ext(const char *filename, const char *extensions)
158{
de6d9b64
FB
159 const char *ext, *p;
160 char ext1[32], *q;
161
293ed23f
MN
162 if(!filename)
163 return 0;
115329f1 164
de6d9b64
FB
165 ext = strrchr(filename, '.');
166 if (ext) {
167 ext++;
168 p = extensions;
169 for(;;) {
170 q = ext1;
115329f1 171 while (*p != '\0' && *p != ',' && q-ext1<sizeof(ext1)-1)
de6d9b64
FB
172 *q++ = *p++;
173 *q = '\0';
bb3244de 174 if (!av_strcasecmp(ext1, ext))
de6d9b64 175 return 1;
115329f1 176 if (*p == '\0')
de6d9b64
FB
177 break;
178 p++;
179 }
180 }
181 return 0;
182}
183
2323ac01
BC
184static int match_format(const char *name, const char *names)
185{
186 const char *p;
187 int len, namelen;
188
189 if (!name || !names)
190 return 0;
191
192 namelen = strlen(name);
193 while ((p = strchr(names, ','))) {
194 len = FFMAX(p - names, namelen);
bb3244de 195 if (!av_strncasecmp(name, names, len))
2323ac01
BC
196 return 1;
197 names = p+1;
198 }
bb3244de 199 return !av_strcasecmp(name, names);
2323ac01
BC
200}
201
a1f547b9
SS
202AVOutputFormat *av_guess_format(const char *short_name, const char *filename,
203 const char *mime_type)
204{
27323146 205 AVOutputFormat *fmt = NULL, *fmt_found;
de6d9b64
FB
206 int score_max, score;
207
87a0a681 208 /* specific test for image sequences */
b250f9c6 209#if CONFIG_IMAGE2_MUXER
115329f1 210 if (!short_name && filename &&
5c07cf53 211 av_filename_number_test(filename) &&
a9bf9d8e 212 ff_guess_image2_codec(filename) != CODEC_ID_NONE) {
0f52ef1a 213 return av_guess_format("image2", NULL, NULL);
5b6d5596 214 }
ff70e601 215#endif
a85736f2 216 /* Find the proper file type. */
de6d9b64
FB
217 fmt_found = NULL;
218 score_max = 0;
27323146 219 while ((fmt = av_oformat_next(fmt))) {
de6d9b64
FB
220 score = 0;
221 if (fmt->name && short_name && !strcmp(fmt->name, short_name))
222 score += 100;
223 if (fmt->mime_type && mime_type && !strcmp(fmt->mime_type, mime_type))
224 score += 10;
115329f1 225 if (filename && fmt->extensions &&
aa13b573 226 av_match_ext(filename, fmt->extensions)) {
de6d9b64
FB
227 score += 5;
228 }
229 if (score > score_max) {
230 score_max = score;
231 fmt_found = fmt;
232 }
de6d9b64
FB
233 }
234 return fmt_found;
115329f1 235}
de6d9b64 236
115329f1 237enum CodecID av_guess_codec(AVOutputFormat *fmt, const char *short_name,
72415b2a
SS
238 const char *filename, const char *mime_type, enum AVMediaType type){
239 if(type == AVMEDIA_TYPE_VIDEO){
5b6d5596
MN
240 enum CodecID codec_id= CODEC_ID_NONE;
241
b250f9c6 242#if CONFIG_IMAGE2_MUXER
ae214ac3 243 if(!strcmp(fmt->name, "image2") || !strcmp(fmt->name, "image2pipe")){
a9bf9d8e 244 codec_id= ff_guess_image2_codec(filename);
5b6d5596 245 }
ff70e601 246#endif
5b6d5596
MN
247 if(codec_id == CODEC_ID_NONE)
248 codec_id= fmt->video_codec;
249 return codec_id;
72415b2a 250 }else if(type == AVMEDIA_TYPE_AUDIO)
5b6d5596 251 return fmt->audio_codec;
118ccae0
AJ
252 else if (type == AVMEDIA_TYPE_SUBTITLE)
253 return fmt->subtitle_codec;
5b6d5596
MN
254 else
255 return CODEC_ID_NONE;
256}
257
b9a281db
FB
258AVInputFormat *av_find_input_format(const char *short_name)
259{
27323146
AK
260 AVInputFormat *fmt = NULL;
261 while ((fmt = av_iformat_next(fmt))) {
2323ac01 262 if (match_format(short_name, fmt->name))
b9a281db
FB
263 return fmt;
264 }
265 return NULL;
266}
267
de6d9b64 268
ae628ec1 269int av_get_packet(AVIOContext *s, AVPacket *pkt, int size)
2692067a
MN
270{
271 int ret= av_new_packet(pkt, size);
272
273 if(ret<0)
274 return ret;
275
a2704c97 276 pkt->pos= avio_tell(s);
2692067a 277
b7effd4e 278 ret= avio_read(s, pkt->data, size);
2692067a
MN
279 if(ret<=0)
280 av_free_packet(pkt);
281 else
feb993e5 282 av_shrink_packet(pkt, ret);
2692067a
MN
283
284 return ret;
285}
286
ae628ec1 287int av_append_packet(AVIOContext *s, AVPacket *pkt, int size)
6bfc2683
RD
288{
289 int ret;
290 int old_size;
291 if (!pkt->size)
292 return av_get_packet(s, pkt, size);
293 old_size = pkt->size;
294 ret = av_grow_packet(pkt, size);
295 if (ret < 0)
296 return ret;
b7effd4e 297 ret = avio_read(s, pkt->data + old_size, size);
6bfc2683
RD
298 av_shrink_packet(pkt, old_size + FFMAX(ret, 0));
299 return ret;
300}
301
fb2758c8 302
5c07cf53 303int av_filename_number_test(const char *filename)
b9a281db
FB
304{
305 char buf[1024];
5c07cf53 306 return filename && (av_get_frame_filename(buf, sizeof(buf), filename, 1)>=0);
b9a281db
FB
307}
308
8e2ee182 309AVInputFormat *av_probe_input_format2(AVProbeData *pd, int is_opened, int *score_max)
b9a281db 310{
6612d8cf 311 AVProbeData lpd = *pd;
27323146 312 AVInputFormat *fmt1 = NULL, *fmt;
56a10009 313 int score, id3 = 0;
b9a281db 314
6612d8cf
RD
315 if (lpd.buf_size > 10 && ff_id3v2_match(lpd.buf, ID3v2_DEFAULT_MAGIC)) {
316 int id3len = ff_id3v2_tag_len(lpd.buf);
317 if (lpd.buf_size > id3len + 16) {
318 lpd.buf += id3len;
319 lpd.buf_size -= id3len;
320 }
56a10009 321 id3 = 1;
6612d8cf
RD
322 }
323
b9a281db 324 fmt = NULL;
27323146 325 while ((fmt1 = av_iformat_next(fmt1))) {
b8e705ec 326 if (!is_opened == !(fmt1->flags & AVFMT_NOFILE))
b9a281db
FB
327 continue;
328 score = 0;
a8dbe951 329 if (fmt1->read_probe) {
6612d8cf 330 score = fmt1->read_probe(&lpd);
a8dbe951 331 } else if (fmt1->extensions) {
6612d8cf 332 if (av_match_ext(lpd.filename, fmt1->extensions)) {
b9a281db
FB
333 score = 50;
334 }
115329f1 335 }
79750486
MN
336 if (score > *score_max) {
337 *score_max = score;
b9a281db 338 fmt = fmt1;
4b3cca36
MN
339 }else if (score == *score_max)
340 fmt = NULL;
b9a281db 341 }
56a10009
AK
342
343 /* a hack for files with huge id3v2 tags -- try to guess by file extension. */
7a773d4d 344 if (!fmt && is_opened && *score_max < AVPROBE_SCORE_MAX/4) {
56a10009
AK
345 while ((fmt = av_iformat_next(fmt)))
346 if (fmt->extensions && av_match_ext(lpd.filename, fmt->extensions)) {
347 *score_max = AVPROBE_SCORE_MAX/4;
348 break;
349 }
350 }
351
61856d06
AC
352 if (!fmt && id3 && *score_max < AVPROBE_SCORE_MAX/4-1) {
353 while ((fmt = av_iformat_next(fmt)))
354 if (fmt->extensions && av_match_ext("mp3", fmt->extensions)) {
355 *score_max = AVPROBE_SCORE_MAX/4-1;
356 break;
357 }
358 }
359
b9a281db
FB
360 return fmt;
361}
362
79750486
MN
363AVInputFormat *av_probe_input_format(AVProbeData *pd, int is_opened){
364 int score=0;
365 return av_probe_input_format2(pd, is_opened, &score);
366}
367
db46c4e1 368static int set_codec_from_probe_data(AVFormatContext *s, AVStream *st, AVProbeData *pd, int score)
64cd3108 369{
cf5b33d9
SS
370 static const struct {
371 const char *name; enum CodecID id; enum AVMediaType type;
372 } fmt_id_type[] = {
373 { "aac" , CODEC_ID_AAC , AVMEDIA_TYPE_AUDIO },
374 { "ac3" , CODEC_ID_AC3 , AVMEDIA_TYPE_AUDIO },
375 { "dts" , CODEC_ID_DTS , AVMEDIA_TYPE_AUDIO },
376 { "eac3" , CODEC_ID_EAC3 , AVMEDIA_TYPE_AUDIO },
377 { "h264" , CODEC_ID_H264 , AVMEDIA_TYPE_VIDEO },
378 { "m4v" , CODEC_ID_MPEG4 , AVMEDIA_TYPE_VIDEO },
379 { "mp3" , CODEC_ID_MP3 , AVMEDIA_TYPE_AUDIO },
380 { "mpegvideo", CODEC_ID_MPEG2VIDEO, AVMEDIA_TYPE_VIDEO },
381 { 0 }
382 };
383 AVInputFormat *fmt = av_probe_input_format2(pd, 1, &score);
64cd3108
MN
384
385 if (fmt) {
cf5b33d9 386 int i;
db46c4e1
BC
387 av_log(s, AV_LOG_DEBUG, "Probe with size=%d, packets=%d detected %s with score=%d\n",
388 pd->buf_size, MAX_PROBE_PACKETS - st->probe_packets, fmt->name, score);
cf5b33d9
SS
389 for (i = 0; fmt_id_type[i].name; i++) {
390 if (!strcmp(fmt->name, fmt_id_type[i].name)) {
391 st->codec->codec_id = fmt_id_type[i].id;
392 st->codec->codec_type = fmt_id_type[i].type;
393 break;
394 }
f1588ed5 395 }
64cd3108
MN
396 }
397 return !!fmt;
398}
399
b9a281db
FB
400/************************************************************/
401/* input media file */
96baaa6a 402
a85736f2 403/** size of probe buffer, for guessing file type from file contents */
a877eedc 404#define PROBE_BUF_MIN 2048
329b1e75 405#define PROBE_BUF_MAX (1<<20)
b9a281db 406
ae628ec1 407int av_probe_input_buffer(AVIOContext *pb, AVInputFormat **fmt,
eadd495d
MG
408 const char *filename, void *logctx,
409 unsigned int offset, unsigned int max_probe_size)
410{
411 AVProbeData pd = { filename ? filename : "", NULL, -offset };
412 unsigned char *buf = NULL;
413 int ret = 0, probe_size;
414
415 if (!max_probe_size) {
416 max_probe_size = PROBE_BUF_MAX;
417 } else if (max_probe_size > PROBE_BUF_MAX) {
418 max_probe_size = PROBE_BUF_MAX;
419 } else if (max_probe_size < PROBE_BUF_MIN) {
420 return AVERROR(EINVAL);
421 }
422
423 if (offset >= max_probe_size) {
424 return AVERROR(EINVAL);
425 }
426
5ef953e8 427 for(probe_size= PROBE_BUF_MIN; probe_size<=max_probe_size && !*fmt;
532aa889 428 probe_size = FFMIN(probe_size<<1, FFMAX(max_probe_size, probe_size+1))) {
5ef953e8 429 int score = probe_size < max_probe_size ? AVPROBE_SCORE_MAX/4 : 0;
eadd495d
MG
430 int buf_offset = (probe_size == PROBE_BUF_MIN) ? 0 : probe_size>>1;
431
432 if (probe_size < offset) {
433 continue;
434 }
435
436 /* read probe data */
437 buf = av_realloc(buf, probe_size + AVPROBE_PADDING_SIZE);
b7effd4e 438 if ((ret = avio_read(pb, buf + buf_offset, probe_size - buf_offset)) < 0) {
eadd495d
MG
439 /* fail if error was not end of file, otherwise, lower score */
440 if (ret != AVERROR_EOF) {
441 av_free(buf);
442 return ret;
443 }
444 score = 0;
c7f625ee 445 ret = 0; /* error was end of file, nothing read */
eadd495d
MG
446 }
447 pd.buf_size += ret;
448 pd.buf = &buf[offset];
449
450 memset(pd.buf + pd.buf_size, 0, AVPROBE_PADDING_SIZE);
451
452 /* guess file format */
453 *fmt = av_probe_input_format2(&pd, 1, &score);
454 if(*fmt){
455 if(score <= AVPROBE_SCORE_MAX/4){ //this can only be true in the last iteration
456 av_log(logctx, AV_LOG_WARNING, "Format detected only with low score of %d, misdetection possible!\n", score);
457 }else
458 av_log(logctx, AV_LOG_DEBUG, "Probed with size=%d and score=%d\n", probe_size, score);
459 }
460 }
461
532aa889 462 if (!*fmt) {
01d91b9b 463 av_free(buf);
532aa889
MG
464 return AVERROR_INVALIDDATA;
465 }
466
01d91b9b 467 /* rewind. reuse probe buffer to avoid seeking */
f1ef2cd9 468 if ((ret = ffio_rewind_with_probe_data(pb, buf, pd.buf_size)) < 0)
01d91b9b 469 av_free(buf);
eadd495d 470
01d91b9b 471 return ret;
eadd495d
MG
472}
473
05e84c95 474/* open input file and probe the format if necessary */
32caa7b1 475static int init_input(AVFormatContext *s, const char *filename, AVDictionary **options)
05e84c95
AK
476{
477 int ret;
478 AVProbeData pd = {filename, NULL, 0};
479
480 if (s->pb) {
481 s->flags |= AVFMT_FLAG_CUSTOM_IO;
482 if (!s->iformat)
483 return av_probe_input_buffer(s->pb, &s->iformat, filename, s, 0, 0);
484 else if (s->iformat->flags & AVFMT_NOFILE)
485 return AVERROR(EINVAL);
486 return 0;
487 }
488
489 if ( (s->iformat && s->iformat->flags & AVFMT_NOFILE) ||
490 (!s->iformat && (s->iformat = av_probe_input_format(&pd, 0))))
491 return 0;
492
9d77a8fa 493 if ((ret = avio_open2(&s->pb, filename, AVIO_FLAG_READ,
32caa7b1 494 &s->interrupt_callback, options)) < 0)
7e6029f9 495 return ret;
05e84c95
AK
496 if (s->iformat)
497 return 0;
498 return av_probe_input_buffer(s->pb, &s->iformat, filename, s, 0, 0);
499}
500
dd2a4bcf
AK
501static AVPacket *add_to_pktbuf(AVPacketList **packet_buffer, AVPacket *pkt,
502 AVPacketList **plast_pktl){
503 AVPacketList *pktl = av_mallocz(sizeof(AVPacketList));
504 if (!pktl)
505 return NULL;
506
507 if (*packet_buffer)
508 (*plast_pktl)->next = pktl;
509 else
510 *packet_buffer = pktl;
511
512 /* add the packet in the buffered packet list */
513 *plast_pktl = pktl;
514 pktl->pkt= *pkt;
515 return &pktl->pkt;
516}
517
05e84c95
AK
518int avformat_open_input(AVFormatContext **ps, const char *filename, AVInputFormat *fmt, AVDictionary **options)
519{
520 AVFormatContext *s = *ps;
dd2a4bcf 521 int i, ret = 0;
05e84c95 522 AVDictionary *tmp = NULL;
079ea6ca 523 ID3v2ExtraMeta *id3v2_extra_meta = NULL;
05e84c95
AK
524
525 if (!s && !(s = avformat_alloc_context()))
526 return AVERROR(ENOMEM);
527 if (fmt)
528 s->iformat = fmt;
529
530 if (options)
531 av_dict_copy(&tmp, *options, 0);
532
533 if ((ret = av_opt_set_dict(s, &tmp)) < 0)
534 goto fail;
535
32caa7b1 536 if ((ret = init_input(s, filename, &tmp)) < 0)
05e84c95
AK
537 goto fail;
538
539 /* check filename in case an image number is expected */
540 if (s->iformat->flags & AVFMT_NEEDNUMBER) {
541 if (!av_filename_number_test(filename)) {
542 ret = AVERROR(EINVAL);
543 goto fail;
544 }
545 }
546
547 s->duration = s->start_time = AV_NOPTS_VALUE;
548 av_strlcpy(s->filename, filename, sizeof(s->filename));
549
550 /* allocate private data */
551 if (s->iformat->priv_data_size > 0) {
552 if (!(s->priv_data = av_mallocz(s->iformat->priv_data_size))) {
553 ret = AVERROR(ENOMEM);
554 goto fail;
555 }
556 if (s->iformat->priv_class) {
557 *(const AVClass**)s->priv_data = s->iformat->priv_class;
558 av_opt_set_defaults(s->priv_data);
559 if ((ret = av_opt_set_dict(s->priv_data, &tmp)) < 0)
560 goto fail;
561 }
562 }
563
564 /* e.g. AVFMT_NOFILE formats will not have a AVIOContext */
565 if (s->pb)
393fd0d8 566 ff_id3v2_read(s, ID3v2_DEFAULT_MAGIC, &id3v2_extra_meta);
05e84c95
AK
567
568 if (s->iformat->read_header)
6e9651d1 569 if ((ret = s->iformat->read_header(s)) < 0)
05e84c95
AK
570 goto fail;
571
079ea6ca
AK
572 if (id3v2_extra_meta &&
573 (ret = ff_id3v2_parse_apic(s, &id3v2_extra_meta)) < 0)
574 goto fail;
575 ff_id3v2_free_extra_meta(&id3v2_extra_meta);
576
dd2a4bcf
AK
577 /* queue attached pictures */
578 for (i = 0; i < s->nb_streams; i++)
579 if (s->streams[i]->disposition & AV_DISPOSITION_ATTACHED_PIC) {
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
05e84c95
AK
585 if (s->pb && !s->data_offset)
586 s->data_offset = avio_tell(s->pb);
587
588 s->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
589
590 if (options) {
591 av_dict_free(options);
592 *options = tmp;
593 }
594 *ps = s;
595 return 0;
596
597fail:
079ea6ca 598 ff_id3v2_free_extra_meta(&id3v2_extra_meta);
05e84c95
AK
599 av_dict_free(&tmp);
600 if (s->pb && !(s->flags & AVFMT_FLAG_CUSTOM_IO))
601 avio_close(s->pb);
602 avformat_free_context(s);
603 *ps = NULL;
604 return ret;
605}
606
da24c5e3
FB
607/*******************************************************/
608
de6d9b64
FB
609int av_read_packet(AVFormatContext *s, AVPacket *pkt)
610{
86cb7e33 611 int ret, i;
62600469 612 AVStream *st;
0bef08e5
MN
613
614 for(;;){
615 AVPacketList *pktl = s->raw_packet_buffer;
616
617 if (pktl) {
618 *pkt = pktl->pkt;
86cb7e33 619 if(s->streams[pkt->stream_index]->codec->codec_id != CODEC_ID_PROBE ||
af122d6a
BC
620 !s->streams[pkt->stream_index]->probe_packets ||
621 s->raw_packet_buffer_remaining_size < pkt->size){
622 AVProbeData *pd = &s->streams[pkt->stream_index]->probe_data;
623 av_freep(&pd->buf);
624 pd->buf_size = 0;
0bef08e5 625 s->raw_packet_buffer = pktl->next;
af122d6a 626 s->raw_packet_buffer_remaining_size += pkt->size;
0bef08e5
MN
627 av_free(pktl);
628 return 0;
629 }
630 }
631
55823964
MN
632 av_init_packet(pkt);
633 ret= s->iformat->read_packet(s, pkt);
86cb7e33
BC
634 if (ret < 0) {
635 if (!pktl || ret == AVERROR(EAGAIN))
636 return ret;
637 for (i = 0; i < s->nb_streams; i++)
638 s->streams[i]->probe_packets = 0;
639 continue;
640 }
73e8e8db
ZK
641
642 if ((s->flags & AVFMT_FLAG_DISCARD_CORRUPT) &&
643 (pkt->flags & AV_PKT_FLAG_CORRUPT)) {
644 av_log(s, AV_LOG_WARNING,
645 "Dropped corrupted packet (stream = %d)\n",
646 pkt->stream_index);
055a141e 647 av_free_packet(pkt);
73e8e8db
ZK
648 continue;
649 }
650
55823964 651 st= s->streams[pkt->stream_index];
62600469 652
55823964 653 switch(st->codec->codec_type){
72415b2a 654 case AVMEDIA_TYPE_VIDEO:
55823964
MN
655 if(s->video_codec_id) st->codec->codec_id= s->video_codec_id;
656 break;
72415b2a 657 case AVMEDIA_TYPE_AUDIO:
55823964
MN
658 if(s->audio_codec_id) st->codec->codec_id= s->audio_codec_id;
659 break;
72415b2a 660 case AVMEDIA_TYPE_SUBTITLE:
55823964
MN
661 if(s->subtitle_codec_id)st->codec->codec_id= s->subtitle_codec_id;
662 break;
663 }
62600469 664
86cb7e33
BC
665 if(!pktl && (st->codec->codec_id != CODEC_ID_PROBE ||
666 !st->probe_packets))
744b4c02
MN
667 return ret;
668
5c5b1731 669 add_to_pktbuf(&s->raw_packet_buffer, pkt, &s->raw_packet_buffer_end);
af122d6a 670 s->raw_packet_buffer_remaining_size -= pkt->size;
744b4c02 671
0bef08e5
MN
672 if(st->codec->codec_id == CODEC_ID_PROBE){
673 AVProbeData *pd = &st->probe_data;
0e500e0d 674 av_log(s, AV_LOG_DEBUG, "probing stream %d\n", st->index);
86cb7e33
BC
675 --st->probe_packets;
676
0bef08e5
MN
677 pd->buf = av_realloc(pd->buf, pd->buf_size+pkt->size+AVPROBE_PADDING_SIZE);
678 memcpy(pd->buf+pd->buf_size, pkt->data, pkt->size);
679 pd->buf_size += pkt->size;
680 memset(pd->buf+pd->buf_size, 0, AVPROBE_PADDING_SIZE);
681
942de2f4 682 if(av_log2(pd->buf_size) != av_log2(pd->buf_size - pkt->size)){
da9cea77 683 //FIXME we do not reduce score to 0 for the case of running out of buffer space in bytes
0f1f4816 684 set_codec_from_probe_data(s, st, pd, st->probe_packets > 0 ? AVPROBE_SCORE_MAX/4 : 0);
25d3fb73
MN
685 if(st->codec->codec_id != CODEC_ID_PROBE){
686 pd->buf_size=0;
687 av_freep(&pd->buf);
0e500e0d 688 av_log(s, AV_LOG_DEBUG, "probed stream %d\n", st->index);
25d3fb73 689 }
942de2f4 690 }
0bef08e5
MN
691 }
692 }
fb2758c8
FB
693}
694
695/**********************************************************/
696
e36bdf8b 697/**
a85736f2 698 * Get the number of samples of an audio frame. Return -1 on error.
e36bdf8b 699 */
fb2758c8
FB
700static int get_audio_frame_size(AVCodecContext *enc, int size)
701{
702 int frame_size;
703
704 if (enc->frame_size <= 1) {
ac3e1834
BC
705 int bits_per_sample = av_get_bits_per_sample(enc->codec_id);
706
707 if (bits_per_sample) {
fb2758c8
FB
708 if (enc->channels == 0)
709 return -1;
f1b163e0 710 frame_size = (size << 3) / (bits_per_sample * enc->channels);
ac3e1834 711 } else {
fb2758c8
FB
712 /* used for example by ADPCM codecs */
713 if (enc->bit_rate == 0)
714 return -1;
db5dc02b 715 frame_size = ((int64_t)size * 8 * enc->sample_rate) / enc->bit_rate;
fb2758c8
FB
716 }
717 } else {
718 frame_size = enc->frame_size;
719 }
720 return frame_size;
721}
722
723
e36bdf8b 724/**
a85736f2 725 * Return the frame duration in seconds. Return 0 if not available.
e36bdf8b 726 */
115329f1 727static void compute_frame_duration(int *pnum, int *pden, AVStream *st,
fb2758c8
FB
728 AVCodecParserContext *pc, AVPacket *pkt)
729{
730 int frame_size;
731
732 *pnum = 0;
733 *pden = 0;
01f4895c 734 switch(st->codec->codec_type) {
72415b2a 735 case AVMEDIA_TYPE_VIDEO:
20922325
AK
736 if (st->r_frame_rate.num) {
737 *pnum = st->r_frame_rate.den;
738 *pden = st->r_frame_rate.num;
739 } else if(st->time_base.num*1000LL > st->time_base.den) {
c0df9d75
MN
740 *pnum = st->time_base.num;
741 *pden = st->time_base.den;
01f4895c
MN
742 }else if(st->codec->time_base.num*1000LL > st->codec->time_base.den){
743 *pnum = st->codec->time_base.num;
744 *pden = st->codec->time_base.den;
327c4076 745 if (pc && pc->repeat_pict) {
810c451b 746 *pnum = (*pnum) * (1 + pc->repeat_pict);
327c4076 747 }
497431a5
MN
748 //If this codec can be interlaced or progressive then we need a parser to compute duration of a packet
749 //Thus if we have no parser in such case leave duration undefined.
750 if(st->codec->ticks_per_frame>1 && !pc){
751 *pnum = *pden = 0;
752 }
fb2758c8
FB
753 }
754 break;
72415b2a 755 case AVMEDIA_TYPE_AUDIO:
01f4895c 756 frame_size = get_audio_frame_size(st->codec, pkt->size);
6cbce636 757 if (frame_size <= 0 || st->codec->sample_rate <= 0)
fb2758c8
FB
758 break;
759 *pnum = frame_size;
01f4895c 760 *pden = st->codec->sample_rate;
fb2758c8
FB
761 break;
762 default:
763 break;
764 }
765}
766
5ba7c3d7 767static int is_intra_only(AVCodecContext *enc){
72415b2a 768 if(enc->codec_type == AVMEDIA_TYPE_AUDIO){
5ba7c3d7 769 return 1;
72415b2a 770 }else if(enc->codec_type == AVMEDIA_TYPE_VIDEO){
5ba7c3d7
MN
771 switch(enc->codec_id){
772 case CODEC_ID_MJPEG:
773 case CODEC_ID_MJPEGB:
774 case CODEC_ID_LJPEG:
5cc2530f 775 case CODEC_ID_PRORES:
5ba7c3d7
MN
776 case CODEC_ID_RAWVIDEO:
777 case CODEC_ID_DVVIDEO:
778 case CODEC_ID_HUFFYUV:
f37b9768 779 case CODEC_ID_FFVHUFF:
5ba7c3d7
MN
780 case CODEC_ID_ASV1:
781 case CODEC_ID_ASV2:
782 case CODEC_ID_VCR1:
b774fdd7 783 case CODEC_ID_DNXHD:
aa915625 784 case CODEC_ID_JPEG2000:
5ba7c3d7
MN
785 return 1;
786 default: break;
787 }
788 }
789 return 0;
790}
791
9fcbcca6
NB
792static void update_initial_timestamps(AVFormatContext *s, int stream_index,
793 int64_t dts, int64_t pts)
794{
82583548
MN
795 AVStream *st= s->streams[stream_index];
796 AVPacketList *pktl= s->packet_buffer;
797
7efeb73a 798 if(st->first_dts != AV_NOPTS_VALUE || dts == AV_NOPTS_VALUE || st->cur_dts == AV_NOPTS_VALUE)
82583548
MN
799 return;
800
801 st->first_dts= dts - st->cur_dts;
802 st->cur_dts= dts;
803
804 for(; pktl; pktl= pktl->next){
805 if(pktl->pkt.stream_index != stream_index)
806 continue;
807 //FIXME think more about this check
808 if(pktl->pkt.pts != AV_NOPTS_VALUE && pktl->pkt.pts == pktl->pkt.dts)
809 pktl->pkt.pts += st->first_dts;
810
811 if(pktl->pkt.dts != AV_NOPTS_VALUE)
812 pktl->pkt.dts += st->first_dts;
48a59dfe
MN
813
814 if(st->start_time == AV_NOPTS_VALUE && pktl->pkt.pts != AV_NOPTS_VALUE)
815 st->start_time= pktl->pkt.pts;
82583548 816 }
9fcbcca6
NB
817 if (st->start_time == AV_NOPTS_VALUE)
818 st->start_time = pts;
82583548
MN
819}
820
83a9db42
MN
821static void update_initial_durations(AVFormatContext *s, AVStream *st, AVPacket *pkt)
822{
823 AVPacketList *pktl= s->packet_buffer;
820ad60c
MN
824 int64_t cur_dts= 0;
825
826 if(st->first_dts != AV_NOPTS_VALUE){
827 cur_dts= st->first_dts;
828 for(; pktl; pktl= pktl->next){
829 if(pktl->pkt.stream_index == pkt->stream_index){
830 if(pktl->pkt.pts != pktl->pkt.dts || pktl->pkt.dts != AV_NOPTS_VALUE || pktl->pkt.duration)
831 break;
832 cur_dts -= pkt->duration;
833 }
834 }
835 pktl= s->packet_buffer;
836 st->first_dts = cur_dts;
837 }else if(st->cur_dts)
838 return;
83a9db42
MN
839
840 for(; pktl; pktl= pktl->next){
841 if(pktl->pkt.stream_index != pkt->stream_index)
842 continue;
91acf9a8
MN
843 if(pktl->pkt.pts == pktl->pkt.dts && pktl->pkt.dts == AV_NOPTS_VALUE
844 && !pktl->pkt.duration){
820ad60c 845 pktl->pkt.dts= cur_dts;
5853423c 846 if(!st->codec->has_b_frames)
820ad60c
MN
847 pktl->pkt.pts= cur_dts;
848 cur_dts += pkt->duration;
83a9db42
MN
849 pktl->pkt.duration= pkt->duration;
850 }else
851 break;
852 }
820ad60c
MN
853 if(st->first_dts == AV_NOPTS_VALUE)
854 st->cur_dts= cur_dts;
83a9db42
MN
855}
856
115329f1 857static void compute_pkt_fields(AVFormatContext *s, AVStream *st,
fb2758c8
FB
858 AVCodecParserContext *pc, AVPacket *pkt)
859{
d9e1efb7 860 int num, den, presentation_delayed, delay, i;
a74008a4 861 int64_t offset;
115329f1 862
fe8344a2
MN
863 if (s->flags & AVFMT_FLAG_NOFILLIN)
864 return;
865
c55806e3
MN
866 if((s->flags & AVFMT_FLAG_IGNDTS) && pkt->pts != AV_NOPTS_VALUE)
867 pkt->dts= AV_NOPTS_VALUE;
868
f781f748
MN
869 /* do we have a video B-frame ? */
870 delay= st->codec->has_b_frames;
871 presentation_delayed = 0;
37b00b47 872
f781f748
MN
873 /* XXX: need has_b_frame, but cannot get it if the codec is
874 not initialized */
875 if (delay &&
975a1447 876 pc && pc->pict_type != AV_PICTURE_TYPE_B)
f781f748
MN
877 presentation_delayed = 1;
878
10a7571b
MN
879 if(pkt->pts != AV_NOPTS_VALUE && pkt->dts != AV_NOPTS_VALUE && pkt->dts > pkt->pts && st->pts_wrap_bits<63
880 /*&& pkt->dts-(1LL<<st->pts_wrap_bits) < pkt->pts*/){
881 pkt->dts -= 1LL<<st->pts_wrap_bits;
882 }
883
9806f846
MN
884 // some mpeg2 in mpeg-ps lack dts (issue171 / input_file.mpg)
885 // we take the conservative approach and discard both
886 // Note, if this is misbehaving for a H.264 file then possibly presentation_delayed is not set correctly.
887 if(delay==1 && pkt->dts == pkt->pts && pkt->dts != AV_NOPTS_VALUE && presentation_delayed){
b0a18c2f 888 av_log(s, AV_LOG_DEBUG, "invalid dts/pts combination\n");
9806f846
MN
889 pkt->dts= pkt->pts= AV_NOPTS_VALUE;
890 }
891
fb2758c8 892 if (pkt->duration == 0) {
3c895fc0 893 compute_frame_duration(&num, &den, st, pc, pkt);
fb2758c8 894 if (den && num) {
0e1f78f9 895 pkt->duration = av_rescale_rnd(1, num * (int64_t)st->time_base.den, den * (int64_t)st->time_base.num, AV_ROUND_DOWN);
83a9db42 896
820ad60c 897 if(pkt->duration != 0 && s->packet_buffer)
83a9db42 898 update_initial_durations(s, st, pkt);
fb2758c8
FB
899 }
900 }
901
a85736f2
DB
902 /* correct timestamps with byte offset if demuxers only have timestamps
903 on packet boundaries */
a74008a4
JP
904 if(pc && st->need_parsing == AVSTREAM_PARSE_TIMESTAMPS && pkt->size){
905 /* this will estimate bitrate based on this frame's duration and size */
906 offset = av_rescale(pc->offset, pkt->duration, pkt->size);
907 if(pkt->pts != AV_NOPTS_VALUE)
908 pkt->pts += offset;
909 if(pkt->dts != AV_NOPTS_VALUE)
910 pkt->dts += offset;
911 }
912
27ca0a79
IS
913 if (pc && pc->dts_sync_point >= 0) {
914 // we have synchronization info from the parser
915 int64_t den = st->codec->time_base.den * (int64_t) st->time_base.num;
916 if (den > 0) {
917 int64_t num = st->codec->time_base.num * (int64_t) st->time_base.den;
918 if (pkt->dts != AV_NOPTS_VALUE) {
919 // got DTS from the stream, update reference timestamp
920 st->reference_dts = pkt->dts - pc->dts_ref_dts_delta * num / den;
921 pkt->pts = pkt->dts + pc->pts_dts_delta * num / den;
922 } else if (st->reference_dts != AV_NOPTS_VALUE) {
923 // compute DTS based on reference timestamp
924 pkt->dts = st->reference_dts + pc->dts_ref_dts_delta * num / den;
925 pkt->pts = pkt->dts + pc->pts_dts_delta * num / den;
926 }
927 if (pc->dts_sync_point > 0)
928 st->reference_dts = pkt->dts; // new reference
929 }
930 }
931
755bfeab 932 /* This may be redundant, but it should not hurt. */
7e4baa66
MN
933 if(pkt->dts != AV_NOPTS_VALUE && pkt->pts != AV_NOPTS_VALUE && pkt->pts > pkt->dts)
934 presentation_delayed = 1;
115329f1 935
949b1a13 936// av_log(NULL, AV_LOG_DEBUG, "IN delayed:%d pts:%"PRId64", dts:%"PRId64" cur_dts:%"PRId64" st:%d pc:%p\n", presentation_delayed, pkt->pts, pkt->dts, st->cur_dts, pkt->stream_index, pc);
fb2758c8 937 /* interpolate PTS and DTS if they are not present */
9e6c124a
MN
938 //We skip H264 currently because delay and has_b_frames are not reliably set
939 if((delay==0 || (delay==1 && pc)) && st->codec->codec_id != CODEC_ID_H264){
7e4baa66 940 if (presentation_delayed) {
a85736f2
DB
941 /* DTS = decompression timestamp */
942 /* PTS = presentation timestamp */
7e4baa66 943 if (pkt->dts == AV_NOPTS_VALUE)
635fbcb1 944 pkt->dts = st->last_IP_pts;
9fcbcca6 945 update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts);
7e4baa66
MN
946 if (pkt->dts == AV_NOPTS_VALUE)
947 pkt->dts = st->cur_dts;
948
949 /* this is tricky: the dts must be incremented by the duration
a85736f2 950 of the frame we are displaying, i.e. the last I- or P-frame */
635fbcb1
MN
951 if (st->last_IP_duration == 0)
952 st->last_IP_duration = pkt->duration;
7efeb73a 953 if(pkt->dts != AV_NOPTS_VALUE)
cdb5af79 954 st->cur_dts = pkt->dts + st->last_IP_duration;
635fbcb1
MN
955 st->last_IP_duration = pkt->duration;
956 st->last_IP_pts= pkt->pts;
7e4baa66 957 /* cannot compute PTS if not present (we can compute it only
a85736f2 958 by knowing the future */
028d6f3e 959 } else if(pkt->pts != AV_NOPTS_VALUE || pkt->dts != AV_NOPTS_VALUE || pkt->duration){
7e4baa66
MN
960 if(pkt->pts != AV_NOPTS_VALUE && pkt->duration){
961 int64_t old_diff= FFABS(st->cur_dts - pkt->duration - pkt->pts);
962 int64_t new_diff= FFABS(st->cur_dts - pkt->pts);
963 if(old_diff < new_diff && old_diff < (pkt->duration>>3)){
964 pkt->pts += pkt->duration;
965 // av_log(NULL, AV_LOG_DEBUG, "id:%d old:%"PRId64" new:%"PRId64" dur:%d cur:%"PRId64" size:%d\n", pkt->stream_index, old_diff, new_diff, pkt->duration, st->cur_dts, pkt->size);
966 }
90bb394d 967 }
115329f1 968
7e4baa66
MN
969 /* presentation is not delayed : PTS and DTS are the same */
970 if(pkt->pts == AV_NOPTS_VALUE)
971 pkt->pts = pkt->dts;
9fcbcca6 972 update_initial_timestamps(s, pkt->stream_index, pkt->pts, pkt->pts);
7e4baa66
MN
973 if(pkt->pts == AV_NOPTS_VALUE)
974 pkt->pts = st->cur_dts;
975 pkt->dts = pkt->pts;
7efeb73a 976 if(pkt->pts != AV_NOPTS_VALUE)
cdb5af79 977 st->cur_dts = pkt->pts + pkt->duration;
7e4baa66 978 }
fb2758c8 979 }
d9e1efb7 980
cb5b96cd 981 if(pkt->pts != AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY){
d9e1efb7 982 st->pts_buffer[0]= pkt->pts;
d9e1efb7
MN
983 for(i=0; i<delay && st->pts_buffer[i] > st->pts_buffer[i+1]; i++)
984 FFSWAP(int64_t, st->pts_buffer[i], st->pts_buffer[i+1]);
985 if(pkt->dts == AV_NOPTS_VALUE)
986 pkt->dts= st->pts_buffer[0];
da9cea77 987 if(st->codec->codec_id == CODEC_ID_H264){ // we skipped it above so we try here
9fcbcca6 988 update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts); // this should happen on the first packet
82583548 989 }
d9e1efb7
MN
990 if(pkt->dts > st->cur_dts)
991 st->cur_dts = pkt->dts;
992 }
993
994// av_log(NULL, AV_LOG_ERROR, "OUTdelayed:%d/%d pts:%"PRId64", dts:%"PRId64" cur_dts:%"PRId64"\n", presentation_delayed, delay, pkt->pts, pkt->dts, st->cur_dts);
115329f1 995
fb2758c8 996 /* update flags */
3a1d3588 997 if(is_intra_only(st->codec))
cc947f04 998 pkt->flags |= AV_PKT_FLAG_KEY;
b1fa4942
IS
999 if (pc)
1000 pkt->convergence_duration = pc->convergence_duration;
fb2758c8
FB
1001}
1002
fb2758c8 1003
d3bb7191 1004static int read_frame_internal(AVFormatContext *s, AVPacket *pkt)
fb2758c8
FB
1005{
1006 AVStream *st;
37353960 1007 int len, ret, i;
fb2758c8 1008
b237eb80
MN
1009 av_init_packet(pkt);
1010
fb2758c8
FB
1011 for(;;) {
1012 /* select current input stream component */
1013 st = s->cur_st;
1014 if (st) {
90ad92b3 1015 if (!st->need_parsing || !st->parser) {
fb2758c8
FB
1016 /* no parsing needed: we just output the packet as is */
1017 /* raw data support */
3a41c2f7 1018 *pkt = st->cur_pkt; st->cur_pkt.data= NULL;
fb2758c8
FB
1019 compute_pkt_fields(s, st, NULL, pkt);
1020 s->cur_st = NULL;
74f5ae84 1021 if ((s->iformat->flags & AVFMT_GENERIC_INDEX) &&
cc947f04 1022 (pkt->flags & AV_PKT_FLAG_KEY) && pkt->dts != AV_NOPTS_VALUE) {
74f5ae84
RD
1023 ff_reduce_index(s, st->index);
1024 av_add_index_entry(st, pkt->pos, pkt->dts, 0, 0, AVINDEX_KEYFRAME);
1025 }
434cab9e 1026 break;
3a41c2f7 1027 } else if (st->cur_len > 0 && st->discard < AVDISCARD_ALL) {
61c23c15
IS
1028 len = av_parser_parse2(st->parser, st->codec, &pkt->data, &pkt->size,
1029 st->cur_ptr, st->cur_len,
1030 st->cur_pkt.pts, st->cur_pkt.dts,
1031 st->cur_pkt.pos);
3a41c2f7
MN
1032 st->cur_pkt.pts = AV_NOPTS_VALUE;
1033 st->cur_pkt.dts = AV_NOPTS_VALUE;
fb2758c8 1034 /* increment read pointer */
3a41c2f7
MN
1035 st->cur_ptr += len;
1036 st->cur_len -= len;
115329f1 1037
fb2758c8
FB
1038 /* return packet if any */
1039 if (pkt->size) {
37353960 1040 got_packet:
fb2758c8 1041 pkt->duration = 0;
e9cda853
JR
1042 if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
1043 if (st->codec->sample_rate > 0) {
1044 pkt->duration = av_rescale_q_rnd(st->parser->duration,
1045 (AVRational){ 1, st->codec->sample_rate },
1046 st->time_base,
1047 AV_ROUND_DOWN);
1048 }
1049 } else if (st->codec->time_base.num != 0 &&
1050 st->codec->time_base.den != 0) {
1051 pkt->duration = av_rescale_q_rnd(st->parser->duration,
1052 st->codec->time_base,
1053 st->time_base,
1054 AV_ROUND_DOWN);
1055 }
fb2758c8 1056 pkt->stream_index = st->index;
6ec87caa
FB
1057 pkt->pts = st->parser->pts;
1058 pkt->dts = st->parser->dts;
61c23c15 1059 pkt->pos = st->parser->pos;
655b24c0
AK
1060 if (st->parser->key_frame == 1 ||
1061 (st->parser->key_frame == -1 &&
1062 st->parser->pict_type == AV_PICTURE_TYPE_I))
1063 pkt->flags |= AV_PKT_FLAG_KEY;
a5266a47
MN
1064 if(pkt->data == st->cur_pkt.data && pkt->size == st->cur_pkt.size){
1065 s->cur_st = NULL;
1066 pkt->destruct= st->cur_pkt.destruct;
22d78b05 1067 st->cur_pkt.destruct= NULL;
a5266a47
MN
1068 st->cur_pkt.data = NULL;
1069 assert(st->cur_len == 0);
1070 }else{
7e6029f9 1071 pkt->destruct = NULL;
a5266a47 1072 }
fb2758c8 1073 compute_pkt_fields(s, st, st->parser, pkt);
e9b78eeb 1074
cc947f04 1075 if((s->iformat->flags & AVFMT_GENERIC_INDEX) && pkt->flags & AV_PKT_FLAG_KEY){
3dea63bd 1076 ff_reduce_index(s, st->index);
e9b78eeb
MN
1077 av_add_index_entry(st, st->parser->frame_offset, pkt->dts,
1078 0, 0, AVINDEX_KEYFRAME);
1079 }
1080
434cab9e 1081 break;
fb2758c8
FB
1082 }
1083 } else {
bcbecff1 1084 /* free packet */
3a41c2f7 1085 av_free_packet(&st->cur_pkt);
fb2758c8
FB
1086 s->cur_st = NULL;
1087 }
1088 } else {
3a41c2f7 1089 AVPacket cur_pkt;
fb2758c8 1090 /* read next packet */
3a41c2f7 1091 ret = av_read_packet(s, &cur_pkt);
37353960 1092 if (ret < 0) {
8fa36ae0 1093 if (ret == AVERROR(EAGAIN))
37353960
FB
1094 return ret;
1095 /* return the last frames, if any */
1096 for(i = 0; i < s->nb_streams; i++) {
1097 st = s->streams[i];
90ad92b3 1098 if (st->parser && st->need_parsing) {
61c23c15 1099 av_parser_parse2(st->parser, st->codec,
115329f1
DB
1100 &pkt->data, &pkt->size,
1101 NULL, 0,
61c23c15
IS
1102 AV_NOPTS_VALUE, AV_NOPTS_VALUE,
1103 AV_NOPTS_VALUE);
37353960
FB
1104 if (pkt->size)
1105 goto got_packet;
1106 }
1107 }
a85736f2 1108 /* no more packets: really terminate parsing */
fb2758c8 1109 return ret;
37353960 1110 }
3a41c2f7
MN
1111 st = s->streams[cur_pkt.stream_index];
1112 st->cur_pkt= cur_pkt;
115329f1 1113
3a41c2f7
MN
1114 if(st->cur_pkt.pts != AV_NOPTS_VALUE &&
1115 st->cur_pkt.dts != AV_NOPTS_VALUE &&
1116 st->cur_pkt.pts < st->cur_pkt.dts){
b18a4ab2 1117 av_log(s, AV_LOG_WARNING, "Invalid timestamps stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d\n",
3a41c2f7
MN
1118 st->cur_pkt.stream_index,
1119 st->cur_pkt.pts,
1120 st->cur_pkt.dts,
1121 st->cur_pkt.size);
1122// av_free_packet(&st->cur_pkt);
b18a4ab2
MN
1123// return -1;
1124 }
1125
45b2b05f 1126 if(s->debug & FF_FDEBUG_TS)
050ba6f4 1127 av_log(s, AV_LOG_DEBUG, "av_read_packet stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d, duration=%d, flags=%d\n",
3a41c2f7
MN
1128 st->cur_pkt.stream_index,
1129 st->cur_pkt.pts,
1130 st->cur_pkt.dts,
1131 st->cur_pkt.size,
050ba6f4 1132 st->cur_pkt.duration,
3a41c2f7 1133 st->cur_pkt.flags);
fb2758c8 1134
fb2758c8 1135 s->cur_st = st;
3a41c2f7
MN
1136 st->cur_ptr = st->cur_pkt.data;
1137 st->cur_len = st->cur_pkt.size;
fe8344a2 1138 if (st->need_parsing && !st->parser && !(s->flags & AVFMT_FLAG_NOPARSE)) {
01f4895c 1139 st->parser = av_parser_init(st->codec->codec_id);
fb2758c8 1140 if (!st->parser) {
a85736f2 1141 /* no parser available: just output the raw packets */
57004ff1
AJ
1142 st->need_parsing = AVSTREAM_PARSE_NONE;
1143 }else if(st->need_parsing == AVSTREAM_PARSE_HEADERS){
7cbaa7ba 1144 st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
74a6df59
AC
1145 }else if(st->need_parsing == AVSTREAM_PARSE_FULL_ONCE){
1146 st->parser->flags |= PARSER_FLAG_ONCE;
fb2758c8
FB
1147 }
1148 }
1149 }
1150 }
45b2b05f 1151 if(s->debug & FF_FDEBUG_TS)
d3bb7191 1152 av_log(s, AV_LOG_DEBUG, "read_frame_internal stream=%d, pts=%"PRId64", dts=%"PRId64", size=%d, duration=%d, flags=%d\n",
434cab9e
MN
1153 pkt->stream_index,
1154 pkt->pts,
1155 pkt->dts,
3041a4a1 1156 pkt->size,
050ba6f4 1157 pkt->duration,
3041a4a1 1158 pkt->flags);
434cab9e
MN
1159
1160 return 0;
fb2758c8
FB
1161}
1162
6450599e
AK
1163static int read_from_packet_buffer(AVFormatContext *s, AVPacket *pkt)
1164{
1165 AVPacketList *pktl = s->packet_buffer;
1166 av_assert0(pktl);
1167 *pkt = pktl->pkt;
1168 s->packet_buffer = pktl->next;
1169 av_freep(&pktl);
1170 return 0;
1171}
1172
fb2758c8
FB
1173int av_read_frame(AVFormatContext *s, AVPacket *pkt)
1174{
f9b9dd87
AK
1175 const int genpts = s->flags & AVFMT_FLAG_GENPTS;
1176 int eof = 0;
30bc6613 1177
6450599e
AK
1178 if (!genpts)
1179 return s->packet_buffer ? read_from_packet_buffer(s, pkt) :
1180 read_frame_internal(s, pkt);
1181
f9b9dd87 1182 for (;;) {
6450599e 1183 int ret;
f9b9dd87 1184 AVPacketList *pktl = s->packet_buffer;
6450599e 1185
30bc6613 1186 if (pktl) {
f9b9dd87 1187 AVPacket *next_pkt = &pktl->pkt;
30bc6613 1188
6450599e 1189 if (next_pkt->dts != AV_NOPTS_VALUE) {
5be5d28c 1190 int wrap_bits = s->streams[next_pkt->stream_index]->pts_wrap_bits;
f9b9dd87
AK
1191 while (pktl && next_pkt->pts == AV_NOPTS_VALUE) {
1192 if (pktl->pkt.stream_index == next_pkt->stream_index &&
1193 (av_compare_mod(next_pkt->dts, pktl->pkt.dts, 2LL << (wrap_bits - 1)) < 0) &&
1194 av_compare_mod(pktl->pkt.pts, pktl->pkt.dts, 2LL << (wrap_bits - 1))) { //not b frame
1195 next_pkt->pts = pktl->pkt.dts;
30bc6613 1196 }
f9b9dd87 1197 pktl = pktl->next;
30bc6613
MN
1198 }
1199 pktl = s->packet_buffer;
1200 }
115329f1 1201
6450599e
AK
1202 /* read packet from packet buffer, if there is data */
1203 if (!(next_pkt->pts == AV_NOPTS_VALUE &&
1204 next_pkt->dts != AV_NOPTS_VALUE && !eof))
1205 return read_from_packet_buffer(s, pkt);
30bc6613 1206 }
115329f1 1207
6450599e
AK
1208 ret = read_frame_internal(s, pkt);
1209 if (ret < 0) {
1210 if (pktl && ret != AVERROR(EAGAIN)) {
1211 eof = 1;
1212 continue;
1213 } else
1214 return ret;
30bc6613 1215 }
6450599e
AK
1216
1217 if (av_dup_packet(add_to_pktbuf(&s->packet_buffer, pkt,
1218 &s->packet_buffer_end)) < 0)
1219 return AVERROR(ENOMEM);
fb2758c8
FB
1220 }
1221}
1222
1223/* XXX: suppress the packet queue */
1224static void flush_packet_queue(AVFormatContext *s)
1225{
1226 AVPacketList *pktl;
1227
1228 for(;;) {
1229 pktl = s->packet_buffer;
115329f1 1230 if (!pktl)
fb2758c8
FB
1231 break;
1232 s->packet_buffer = pktl->next;
1233 av_free_packet(&pktl->pkt);
1234 av_free(pktl);
b9a281db 1235 }
86cb7e33
BC
1236 while(s->raw_packet_buffer){
1237 pktl = s->raw_packet_buffer;
1238 s->raw_packet_buffer = pktl->next;
1239 av_free_packet(&pktl->pkt);
1240 av_free(pktl);
1241 }
42831b46
MN
1242 s->packet_buffer_end=
1243 s->raw_packet_buffer_end= NULL;
af122d6a 1244 s->raw_packet_buffer_remaining_size = RAW_PACKET_BUFFER_SIZE;
b9a281db
FB
1245}
1246
da24c5e3 1247/*******************************************************/
fb2758c8
FB
1248/* seek support */
1249
b754978a
MN
1250int av_find_default_stream_index(AVFormatContext *s)
1251{
ca162a50 1252 int first_audio_index = -1;
b754978a
MN
1253 int i;
1254 AVStream *st;
1255
1256 if (s->nb_streams <= 0)
1257 return -1;
1258 for(i = 0; i < s->nb_streams; i++) {
1259 st = s->streams[i];
72415b2a 1260 if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
b754978a
MN
1261 return i;
1262 }
72415b2a 1263 if (first_audio_index < 0 && st->codec->codec_type == AVMEDIA_TYPE_AUDIO)
ca162a50 1264 first_audio_index = i;
b754978a 1265 }
ca162a50 1266 return first_audio_index >= 0 ? first_audio_index : 0;
b754978a
MN
1267}
1268
e36bdf8b
DK
1269/**
1270 * Flush the frame reader.
1271 */
972ffe62 1272void ff_read_frame_flush(AVFormatContext *s)
fb2758c8
FB
1273{
1274 AVStream *st;
106fa129 1275 int i, j;
fb2758c8
FB
1276
1277 flush_packet_queue(s);
1278
3a41c2f7 1279 s->cur_st = NULL;
115329f1 1280
fb2758c8
FB
1281 /* for each stream, reset read state */
1282 for(i = 0; i < s->nb_streams; i++) {
1283 st = s->streams[i];
115329f1 1284
fb2758c8
FB
1285 if (st->parser) {
1286 av_parser_close(st->parser);
1287 st->parser = NULL;
3a41c2f7 1288 av_free_packet(&st->cur_pkt);
fb2758c8 1289 }
635fbcb1 1290 st->last_IP_pts = AV_NOPTS_VALUE;
a843d1ff 1291 st->cur_dts = AV_NOPTS_VALUE; /* we set the current DTS to an unspecified origin */
27ca0a79 1292 st->reference_dts = AV_NOPTS_VALUE;
3a41c2f7
MN
1293 /* fail safe */
1294 st->cur_ptr = NULL;
1295 st->cur_len = 0;
86cb7e33
BC
1296
1297 st->probe_packets = MAX_PROBE_PACKETS;
106fa129
JS
1298
1299 for(j=0; j<MAX_REORDER_DELAY+1; j++)
1300 st->pts_buffer[j]= AV_NOPTS_VALUE;
fb2758c8
FB
1301 }
1302}
1303
a2faa951
AK
1304void ff_update_cur_dts(AVFormatContext *s, AVStream *ref_st, int64_t timestamp)
1305{
8bcb147f
MN
1306 int i;
1307
1308 for(i = 0; i < s->nb_streams; i++) {
1a1dc611 1309 AVStream *st = s->streams[i];
8bcb147f 1310
115329f1 1311 st->cur_dts = av_rescale(timestamp,
1a1dc611
NK
1312 st->time_base.den * (int64_t)ref_st->time_base.num,
1313 st->time_base.num * (int64_t)ref_st->time_base.den);
8bcb147f
MN
1314 }
1315}
1316
3dea63bd
PK
1317void ff_reduce_index(AVFormatContext *s, int stream_index)
1318{
1319 AVStream *st= s->streams[stream_index];
1320 unsigned int max_entries= s->max_index_size / sizeof(AVIndexEntry);
1321
1322 if((unsigned)st->nb_index_entries >= max_entries){
1323 int i;
1324 for(i=0; 2*i<st->nb_index_entries; i++)
1325 st->index_entries[i]= st->index_entries[2*i];
1326 st->nb_index_entries= i;
1327 }
1328}
1329
e6fb5a4f
PR
1330int ff_add_index_entry(AVIndexEntry **index_entries,
1331 int *nb_index_entries,
1332 unsigned int *index_entries_allocated_size,
1333 int64_t pos, int64_t timestamp, int size, int distance, int flags)
fb2758c8
FB
1334{
1335 AVIndexEntry *entries, *ie;
b754978a 1336 int index;
115329f1 1337
e6fb5a4f 1338 if((unsigned)*nb_index_entries + 1 >= UINT_MAX / sizeof(AVIndexEntry))
568e18b1 1339 return -1;
115329f1 1340
e6fb5a4f
PR
1341 entries = av_fast_realloc(*index_entries,
1342 index_entries_allocated_size,
1343 (*nb_index_entries + 1) *
fb2758c8 1344 sizeof(AVIndexEntry));
568e18b1
MN
1345 if(!entries)
1346 return -1;
1347
e6fb5a4f 1348 *index_entries= entries;
b754978a 1349
e6fb5a4f 1350 index= ff_index_search_timestamp(*index_entries, *nb_index_entries, timestamp, AVSEEK_FLAG_ANY);
b754978a 1351
3ba1438d 1352 if(index<0){
e6fb5a4f 1353 index= (*nb_index_entries)++;
3e9245a9 1354 ie= &entries[index];
3ba1438d
MN
1355 assert(index==0 || ie[-1].timestamp < timestamp);
1356 }else{
1357 ie= &entries[index];
1358 if(ie->timestamp != timestamp){
528c2c73
MN
1359 if(ie->timestamp <= timestamp)
1360 return -1;
e6fb5a4f
PR
1361 memmove(entries + index + 1, entries + index, sizeof(AVIndexEntry)*(*nb_index_entries - index));
1362 (*nb_index_entries)++;
755bfeab 1363 }else if(ie->pos == pos && distance < ie->min_distance) //do not reduce the distance
3ba1438d 1364 distance= ie->min_distance;
3e9245a9 1365 }
3ba1438d 1366
b754978a
MN
1367 ie->pos = pos;
1368 ie->timestamp = timestamp;
3e9245a9 1369 ie->min_distance= distance;
30a43f2d 1370 ie->size= size;
b754978a 1371 ie->flags = flags;
115329f1 1372
3e9245a9 1373 return index;
fb2758c8
FB
1374}
1375
e6fb5a4f
PR
1376int av_add_index_entry(AVStream *st,
1377 int64_t pos, int64_t timestamp, int size, int distance, int flags)
1378{
1379 return ff_add_index_entry(&st->index_entries, &st->nb_index_entries,
1380 &st->index_entries_allocated_size, pos,
1381 timestamp, size, distance, flags);
1382}
1383
1384int ff_index_search_timestamp(const AVIndexEntry *entries, int nb_entries,
1385 int64_t wanted_timestamp, int flags)
fb2758c8
FB
1386{
1387 int a, b, m;
1388 int64_t timestamp;
1389
3ba1438d
MN
1390 a = - 1;
1391 b = nb_entries;
b754978a 1392
67c10de7
MN
1393 //optimize appending index entries at the end
1394 if(b && entries[b-1].timestamp < wanted_timestamp)
1395 a= b-1;
1396
3ba1438d
MN
1397 while (b - a > 1) {
1398 m = (a + b) >> 1;
fb2758c8 1399 timestamp = entries[m].timestamp;
3ba1438d
MN
1400 if(timestamp >= wanted_timestamp)
1401 b = m;
1402 if(timestamp <= wanted_timestamp)
b754978a 1403 a = m;
fb2758c8 1404 }
27a5fe5f 1405 m= (flags & AVSEEK_FLAG_BACKWARD) ? a : b;
115329f1 1406
27a5fe5f
MN
1407 if(!(flags & AVSEEK_FLAG_ANY)){
1408 while(m>=0 && m<nb_entries && !(entries[m].flags & AVINDEX_KEYFRAME)){
1409 m += (flags & AVSEEK_FLAG_BACKWARD) ? -1 : 1;
1410 }
1411 }
3ba1438d 1412
115329f1 1413 if(m == nb_entries)
3ba1438d
MN
1414 return -1;
1415 return m;
fb2758c8
FB
1416}
1417
e6fb5a4f
PR
1418int av_index_search_timestamp(AVStream *st, int64_t wanted_timestamp,
1419 int flags)
1420{
1421 return ff_index_search_timestamp(st->index_entries, st->nb_index_entries,
1422 wanted_timestamp, flags);
1423}
1424
a2faa951
AK
1425int ff_seek_frame_binary(AVFormatContext *s, int stream_index, int64_t target_ts, int flags)
1426{
8d14a25c 1427 AVInputFormat *avif= s->iformat;
e6586575 1428 int64_t av_uninit(pos_min), av_uninit(pos_max), pos, pos_limit;
8d14a25c 1429 int64_t ts_min, ts_max, ts;
89ddd2a9 1430 int index;
b593f7fd 1431 int64_t ret;
8d14a25c
MN
1432 AVStream *st;
1433
cdd5034f
MN
1434 if (stream_index < 0)
1435 return -1;
115329f1 1436
919d7a34 1437 av_dlog(s, "read_seek: %d %"PRId64"\n", stream_index, target_ts);
8d14a25c
MN
1438
1439 ts_max=
1440 ts_min= AV_NOPTS_VALUE;
90b5b51e 1441 pos_limit= -1; //gcc falsely says it may be uninitialized
8d14a25c
MN
1442
1443 st= s->streams[stream_index];
1444 if(st->index_entries){
1445 AVIndexEntry *e;
1446
a85736f2 1447 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()
3ba1438d 1448 index= FFMAX(index, 0);
8d14a25c
MN
1449 e= &st->index_entries[index];
1450
1451 if(e->timestamp <= target_ts || e->pos == e->min_distance){
1452 pos_min= e->pos;
1453 ts_min= e->timestamp;
919d7a34
DB
1454 av_dlog(s, "using cached pos_min=0x%"PRIx64" dts_min=%"PRId64"\n",
1455 pos_min,ts_min);
8d14a25c
MN
1456 }else{
1457 assert(index==0);
1458 }
115329f1
DB
1459
1460 index= av_index_search_timestamp(st, target_ts, flags & ~AVSEEK_FLAG_BACKWARD);
27a5fe5f
MN
1461 assert(index < st->nb_index_entries);
1462 if(index >= 0){
8d14a25c
MN
1463 e= &st->index_entries[index];
1464 assert(e->timestamp >= target_ts);
1465 pos_max= e->pos;
1466 ts_max= e->timestamp;
1467 pos_limit= pos_max - e->min_distance;
919d7a34
DB
1468 av_dlog(s, "using cached pos_max=0x%"PRIx64" pos_limit=0x%"PRIx64" dts_max=%"PRId64"\n",
1469 pos_max,pos_limit, ts_max);
8d14a25c
MN
1470 }
1471 }
1472
a2faa951 1473 pos= ff_gen_search(s, stream_index, target_ts, pos_min, pos_max, pos_limit, ts_min, ts_max, flags, &ts, avif->read_timestamp);
89ddd2a9
MN
1474 if(pos<0)
1475 return -1;
1476
1477 /* do the seek */
6b4aa5da 1478 if ((ret = avio_seek(s->pb, pos, SEEK_SET)) < 0)
b593f7fd 1479 return ret;
89ddd2a9 1480
a2faa951 1481 ff_update_cur_dts(s, st, ts);
89ddd2a9
MN
1482
1483 return 0;
1484}
1485
a2faa951
AK
1486int64_t ff_gen_search(AVFormatContext *s, int stream_index, int64_t target_ts,
1487 int64_t pos_min, int64_t pos_max, int64_t pos_limit,
1488 int64_t ts_min, int64_t ts_max, int flags, int64_t *ts_ret,
1489 int64_t (*read_timestamp)(struct AVFormatContext *, int , int64_t *, int64_t ))
1490{
89ddd2a9
MN
1491 int64_t pos, ts;
1492 int64_t start_pos, filesize;
1493 int no_change;
1494
919d7a34 1495 av_dlog(s, "gen_seek: %d %"PRId64"\n", stream_index, target_ts);
89ddd2a9 1496
8d14a25c
MN
1497 if(ts_min == AV_NOPTS_VALUE){
1498 pos_min = s->data_offset;
89ddd2a9 1499 ts_min = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
8d14a25c
MN
1500 if (ts_min == AV_NOPTS_VALUE)
1501 return -1;
1502 }
1503
1504 if(ts_max == AV_NOPTS_VALUE){
1505 int step= 1024;
76aa876e 1506 filesize = avio_size(s->pb);
6fd93ce2 1507 pos_max = filesize - 1;
8d14a25c
MN
1508 do{
1509 pos_max -= step;
89ddd2a9 1510 ts_max = read_timestamp(s, stream_index, &pos_max, pos_max + step);
8d14a25c
MN
1511 step += step;
1512 }while(ts_max == AV_NOPTS_VALUE && pos_max >= step);
1513 if (ts_max == AV_NOPTS_VALUE)
1514 return -1;
115329f1 1515
8d14a25c
MN
1516 for(;;){
1517 int64_t tmp_pos= pos_max + 1;
89ddd2a9 1518 int64_t tmp_ts= read_timestamp(s, stream_index, &tmp_pos, INT64_MAX);
8d14a25c
MN
1519 if(tmp_ts == AV_NOPTS_VALUE)
1520 break;
1521 ts_max= tmp_ts;
1522 pos_max= tmp_pos;
6fd93ce2
KA
1523 if(tmp_pos >= filesize)
1524 break;
8d14a25c
MN
1525 }
1526 pos_limit= pos_max;
1527 }
1528
53f7c43f
MN
1529 if(ts_min > ts_max){
1530 return -1;
1531 }else if(ts_min == ts_max){
1532 pos_limit= pos_min;
1533 }
1534
8d14a25c
MN
1535 no_change=0;
1536 while (pos_min < pos_limit) {
919d7a34
DB
1537 av_dlog(s, "pos_min=0x%"PRIx64" pos_max=0x%"PRIx64" dts_min=%"PRId64" dts_max=%"PRId64"\n",
1538 pos_min, pos_max, ts_min, ts_max);
8d14a25c
MN
1539 assert(pos_limit <= pos_max);
1540
1541 if(no_change==0){
1542 int64_t approximate_keyframe_distance= pos_max - pos_limit;
1543 // interpolate position (better than dichotomy)
3ba1438d
MN
1544 pos = av_rescale(target_ts - ts_min, pos_max - pos_min, ts_max - ts_min)
1545 + pos_min - approximate_keyframe_distance;
8d14a25c
MN
1546 }else if(no_change==1){
1547 // bisection, if interpolation failed to change min or max pos last time
1548 pos = (pos_min + pos_limit)>>1;
1549 }else{
a85736f2
DB
1550 /* linear search if bisection failed, can only happen if there
1551 are very few or no keyframes between min/max */
8d14a25c
MN
1552 pos=pos_min;
1553 }
1554 if(pos <= pos_min)
1555 pos= pos_min + 1;
1556 else if(pos > pos_limit)
1557 pos= pos_limit;
1558 start_pos= pos;
1559
89ddd2a9 1560 ts = read_timestamp(s, stream_index, &pos, INT64_MAX); //may pass pos_limit instead of -1
8d14a25c
MN
1561 if(pos == pos_max)
1562 no_change++;
1563 else
1564 no_change=0;
919d7a34
DB
1565 av_dlog(s, "%"PRId64" %"PRId64" %"PRId64" / %"PRId64" %"PRId64" %"PRId64" target:%"PRId64" limit:%"PRId64" start:%"PRId64" noc:%d\n",
1566 pos_min, pos, pos_max, ts_min, ts, ts_max, target_ts,
1567 pos_limit, start_pos, no_change);
db2a0e22
MN
1568 if(ts == AV_NOPTS_VALUE){
1569 av_log(s, AV_LOG_ERROR, "read_timestamp() failed in the middle\n");
1570 return -1;
1571 }
8d14a25c 1572 assert(ts != AV_NOPTS_VALUE);
3ba1438d 1573 if (target_ts <= ts) {
8d14a25c
MN
1574 pos_limit = start_pos - 1;
1575 pos_max = pos;
1576 ts_max = ts;
3ba1438d
MN
1577 }
1578 if (target_ts >= ts) {
8d14a25c
MN
1579 pos_min = pos;
1580 ts_min = ts;
8d14a25c
MN
1581 }
1582 }
115329f1 1583
3ba1438d
MN
1584 pos = (flags & AVSEEK_FLAG_BACKWARD) ? pos_min : pos_max;
1585 ts = (flags & AVSEEK_FLAG_BACKWARD) ? ts_min : ts_max;
8d14a25c 1586 pos_min = pos;
89ddd2a9 1587 ts_min = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
8d14a25c 1588 pos_min++;
89ddd2a9 1589 ts_max = read_timestamp(s, stream_index, &pos_min, INT64_MAX);
919d7a34
DB
1590 av_dlog(s, "pos=0x%"PRIx64" %"PRId64"<=%"PRId64"<=%"PRId64"\n",
1591 pos, ts_min, target_ts, ts_max);
89ddd2a9
MN
1592 *ts_ret= ts;
1593 return pos;
8d14a25c
MN
1594}
1595
d3bb7191 1596static int seek_frame_byte(AVFormatContext *s, int stream_index, int64_t pos, int flags){
3ba1438d
MN
1597 int64_t pos_min, pos_max;
1598#if 0
1599 AVStream *st;
1600
1601 if (stream_index < 0)
1602 return -1;
1603
1604 st= s->streams[stream_index];
1605#endif
1606
1607 pos_min = s->data_offset;
76aa876e 1608 pos_max = avio_size(s->pb) - 1;
3ba1438d
MN
1609
1610 if (pos < pos_min) pos= pos_min;
1611 else if(pos > pos_max) pos= pos_max;
1612
6b4aa5da 1613 avio_seek(s->pb, pos, SEEK_SET);
3ba1438d
MN
1614
1615#if 0
8bcb147f 1616 av_update_cur_dts(s, st, ts);
3ba1438d
MN
1617#endif
1618 return 0;
1619}
1620
d3bb7191 1621static int seek_frame_generic(AVFormatContext *s,
3ba1438d 1622 int stream_index, int64_t timestamp, int flags)
fb2758c8 1623{
6659b32a
SS
1624 int index;
1625 int64_t ret;
fb2758c8
FB
1626 AVStream *st;
1627 AVIndexEntry *ie;
1628
fb2758c8 1629 st = s->streams[stream_index];
e9b78eeb 1630
27a5fe5f 1631 index = av_index_search_timestamp(st, timestamp, flags);
e9b78eeb 1632
8c3b161e
MN
1633 if(index < 0 && st->nb_index_entries && timestamp < st->index_entries[0].timestamp)
1634 return -1;
1635
b5a3541d 1636 if(index < 0 || index==st->nb_index_entries-1){
e9b78eeb
MN
1637 AVPacket pkt;
1638
5e5c9086
MN
1639 if(st->nb_index_entries){
1640 assert(st->index_entries);
e9b78eeb 1641 ie= &st->index_entries[st->nb_index_entries-1];
6b4aa5da 1642 if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
aaec4e03 1643 return ret;
a2faa951 1644 ff_update_cur_dts(s, st, ie->timestamp);
aaec4e03 1645 }else{
6b4aa5da 1646 if ((ret = avio_seek(s->pb, s->data_offset, SEEK_SET)) < 0)
aaec4e03
BC
1647 return ret;
1648 }
940173d4 1649 for (;;) {
4439caa4 1650 int read_status;
cda6902d 1651 do{
4439caa4
AC
1652 read_status = av_read_frame(s, &pkt);
1653 } while (read_status == AVERROR(EAGAIN));
1654 if (read_status < 0)
e9b78eeb
MN
1655 break;
1656 av_free_packet(&pkt);
1657 if(stream_index == pkt.stream_index){
cc947f04 1658 if((pkt.flags & AV_PKT_FLAG_KEY) && pkt.dts > timestamp)
e9b78eeb
MN
1659 break;
1660 }
1661 }
1662 index = av_index_search_timestamp(st, timestamp, flags);
1663 }
fb2758c8
FB
1664 if (index < 0)
1665 return -1;
1666
972ffe62 1667 ff_read_frame_flush(s);
e9b78eeb
MN
1668 if (s->iformat->read_seek){
1669 if(s->iformat->read_seek(s, stream_index, timestamp, flags) >= 0)
1670 return 0;
1671 }
1672 ie = &st->index_entries[index];
6b4aa5da 1673 if ((ret = avio_seek(s->pb, ie->pos, SEEK_SET)) < 0)
aaec4e03 1674 return ret;
a2faa951 1675 ff_update_cur_dts(s, st, ie->timestamp);
cdd5034f 1676
fb2758c8
FB
1677 return 0;
1678}
1679
3ba1438d 1680int av_seek_frame(AVFormatContext *s, int stream_index, int64_t timestamp, int flags)
fb2758c8
FB
1681{
1682 int ret;
cdd5034f 1683 AVStream *st;
115329f1 1684
0041cdba 1685 if (flags & AVSEEK_FLAG_BYTE) {
b631fba9
JR
1686 if (s->iformat->flags & AVFMT_NO_BYTE_SEEK)
1687 return -1;
0041cdba 1688 ff_read_frame_flush(s);
d3bb7191 1689 return seek_frame_byte(s, stream_index, timestamp, flags);
0041cdba 1690 }
115329f1 1691
cdd5034f
MN
1692 if(stream_index < 0){
1693 stream_index= av_find_default_stream_index(s);
1694 if(stream_index < 0)
1695 return -1;
115329f1 1696
3ba1438d 1697 st= s->streams[stream_index];
7e6029f9 1698 /* timestamp for default must be expressed in AV_TIME_BASE units */
3ba1438d 1699 timestamp = av_rescale(timestamp, st->time_base.den, AV_TIME_BASE * (int64_t)st->time_base.num);
cdd5034f 1700 }
cdd5034f 1701
fb2758c8 1702 /* first, we try the format specific seek */
0041cdba
JR
1703 if (s->iformat->read_seek) {
1704 ff_read_frame_flush(s);
3ba1438d 1705 ret = s->iformat->read_seek(s, stream_index, timestamp, flags);
0041cdba 1706 } else
fb2758c8
FB
1707 ret = -1;
1708 if (ret >= 0) {
1709 return 0;
1710 }
8d14a25c 1711
0041cdba
JR
1712 if (s->iformat->read_timestamp && !(s->iformat->flags & AVFMT_NOBINSEARCH)) {
1713 ff_read_frame_flush(s);
a2faa951 1714 return ff_seek_frame_binary(s, stream_index, timestamp, flags);
0041cdba
JR
1715 } else if (!(s->iformat->flags & AVFMT_NOGENSEARCH)) {
1716 ff_read_frame_flush(s);
d3bb7191 1717 return seek_frame_generic(s, stream_index, timestamp, flags);
0041cdba 1718 }
69fa2396
VP
1719 else
1720 return -1;
fb2758c8
FB
1721}
1722
32d88592
MN
1723int avformat_seek_file(AVFormatContext *s, int stream_index, int64_t min_ts, int64_t ts, int64_t max_ts, int flags)
1724{
1725 if(min_ts > ts || max_ts < ts)
1726 return -1;
1727
0041cdba
JR
1728 if (s->iformat->read_seek2) {
1729 ff_read_frame_flush(s);
32d88592 1730 return s->iformat->read_seek2(s, stream_index, min_ts, ts, max_ts, flags);
0041cdba 1731 }
32d88592
MN
1732
1733 if(s->iformat->read_timestamp){
1734 //try to seek via read_timestamp()
1735 }
1736
1737 //Fallback to old API if new is not implemented but old is
1738 //Note the old has somewat different sematics
1739 if(s->iformat->read_seek || 1)
85b4230f 1740 return av_seek_frame(s, stream_index, ts, flags | (ts - min_ts > (uint64_t)(max_ts - ts) ? AVSEEK_FLAG_BACKWARD : 0));
32d88592 1741
d3bb7191 1742 // try some generic seek like seek_frame_generic() but with new ts semantics
32d88592
MN
1743}
1744
fb2758c8 1745/*******************************************************/
12f996ed 1746
e36bdf8b 1747/**
49bd8e4b 1748 * Return TRUE if the stream has accurate duration in any stream.
e36bdf8b 1749 *
c1e8b678 1750 * @return TRUE if the stream has accurate duration for at least one component.
e36bdf8b 1751 */
d3bb7191 1752static int has_duration(AVFormatContext *ic)
12f996ed
FB
1753{
1754 int i;
1755 AVStream *st;
1756
1757 for(i = 0;i < ic->nb_streams; i++) {
1758 st = ic->streams[i];
c1e8b678 1759 if (st->duration != AV_NOPTS_VALUE)
12f996ed
FB
1760 return 1;
1761 }
1762 return 0;
1763}
1764
e36bdf8b
DK
1765/**
1766 * Estimate the stream timings from the one of each components.
1767 *
1768 * Also computes the global bitrate if possible.
1769 */
d3bb7191 1770static void update_stream_timings(AVFormatContext *ic)
12f996ed 1771{
c0df9d75 1772 int64_t start_time, start_time1, end_time, end_time1;
c10731e7 1773 int64_t duration, duration1, filesize;
12f996ed
FB
1774 int i;
1775 AVStream *st;
1776
f27a7268
MR
1777 start_time = INT64_MAX;
1778 end_time = INT64_MIN;
c1e8b678 1779 duration = INT64_MIN;
12f996ed
FB
1780 for(i = 0;i < ic->nb_streams; i++) {
1781 st = ic->streams[i];
7f938dd3 1782 if (st->start_time != AV_NOPTS_VALUE && st->time_base.den) {
c0df9d75 1783 start_time1= av_rescale_q(st->start_time, st->time_base, AV_TIME_BASE_Q);
a7503430 1784 start_time = FFMIN(start_time, start_time1);
12f996ed 1785 if (st->duration != AV_NOPTS_VALUE) {
c0df9d75
MN
1786 end_time1 = start_time1
1787 + av_rescale_q(st->duration, st->time_base, AV_TIME_BASE_Q);
a7503430 1788 end_time = FFMAX(end_time, end_time1);
12f996ed
FB
1789 }
1790 }
c1e8b678
NB
1791 if (st->duration != AV_NOPTS_VALUE) {
1792 duration1 = av_rescale_q(st->duration, st->time_base, AV_TIME_BASE_Q);
a7503430 1793 duration = FFMAX(duration, duration1);
c1e8b678 1794 }
12f996ed 1795 }
f27a7268 1796 if (start_time != INT64_MAX) {
12f996ed 1797 ic->start_time = start_time;
a7503430
AK
1798 if (end_time != INT64_MIN)
1799 duration = FFMAX(duration, end_time - start_time);
c1e8b678
NB
1800 }
1801 if (duration != INT64_MIN) {
1802 ic->duration = duration;
c10731e7 1803 if (ic->pb && (filesize = avio_size(ic->pb)) > 0) {
a85736f2 1804 /* compute the bitrate */
c10731e7 1805 ic->bit_rate = (double)filesize * 8.0 * AV_TIME_BASE /
c1e8b678 1806 (double)ic->duration;
12f996ed
FB
1807 }
1808 }
12f996ed
FB
1809}
1810
1811static void fill_all_stream_timings(AVFormatContext *ic)
1812{
1813 int i;
1814 AVStream *st;
1815
d3bb7191 1816 update_stream_timings(ic);
12f996ed
FB
1817 for(i = 0;i < ic->nb_streams; i++) {
1818 st = ic->streams[i];
1819 if (st->start_time == AV_NOPTS_VALUE) {
c0df9d75
MN
1820 if(ic->start_time != AV_NOPTS_VALUE)
1821 st->start_time = av_rescale_q(ic->start_time, AV_TIME_BASE_Q, st->time_base);
1822 if(ic->duration != AV_NOPTS_VALUE)
1823 st->duration = av_rescale_q(ic->duration, AV_TIME_BASE_Q, st->time_base);
12f996ed
FB
1824 }
1825 }
1826}
1827
d3bb7191 1828static void estimate_timings_from_bit_rate(AVFormatContext *ic)
12f996ed
FB
1829{
1830 int64_t filesize, duration;
1831 int bit_rate, i;
1832 AVStream *st;
1833
1834 /* if bit_rate is already set, we believe it */
9100d4d6 1835 if (ic->bit_rate <= 0) {
12f996ed
FB
1836 bit_rate = 0;
1837 for(i=0;i<ic->nb_streams;i++) {
1838 st = ic->streams[i];
9100d4d6 1839 if (st->codec->bit_rate > 0)
01f4895c 1840 bit_rate += st->codec->bit_rate;
12f996ed
FB
1841 }
1842 ic->bit_rate = bit_rate;
1843 }
1844
1845 /* if duration is already set, we believe it */
115329f1 1846 if (ic->duration == AV_NOPTS_VALUE &&
c10731e7
AK
1847 ic->bit_rate != 0) {
1848 filesize = ic->pb ? avio_size(ic->pb) : 0;
12f996ed 1849 if (filesize > 0) {
12f996ed
FB
1850 for(i = 0; i < ic->nb_streams; i++) {
1851 st = ic->streams[i];
c0df9d75 1852 duration= av_rescale(8*filesize, st->time_base.den, ic->bit_rate*(int64_t)st->time_base.num);
9f32041d 1853 if (st->duration == AV_NOPTS_VALUE)
12f996ed 1854 st->duration = duration;
12f996ed
FB
1855 }
1856 }
1857 }
1858}
1859
12f996ed 1860#define DURATION_MAX_READ_SIZE 250000
411ff322 1861#define DURATION_MAX_RETRY 3
12f996ed
FB
1862
1863/* only usable for MPEG-PS streams */
d3bb7191 1864static void estimate_timings_from_pts(AVFormatContext *ic, int64_t old_offset)
12f996ed
FB
1865{
1866 AVPacket pkt1, *pkt = &pkt1;
1867 AVStream *st;
1868 int read_size, i, ret;
3e4318bf 1869 int64_t end_time;
12f996ed 1870 int64_t filesize, offset, duration;
411ff322 1871 int retry=0;
115329f1 1872
578688fa
LA
1873 ic->cur_st = NULL;
1874
1875 /* flush packet queue */
1876 flush_packet_queue(ic);
1877
e99179de 1878 for (i=0; i<ic->nb_streams; i++) {
578688fa 1879 st = ic->streams[i];
3e4318bf 1880 if (st->start_time == AV_NOPTS_VALUE && st->first_dts == AV_NOPTS_VALUE)
d3bb7191 1881 av_log(st->codec, AV_LOG_WARNING, "start time is not set in estimate_timings_from_pts\n");
cc20679a 1882
578688fa
LA
1883 if (st->parser) {
1884 av_parser_close(st->parser);
1885 st->parser= NULL;
3a41c2f7 1886 av_free_packet(&st->cur_pkt);
578688fa
LA
1887 }
1888 }
115329f1 1889
12f996ed
FB
1890 /* estimate the end time (duration) */
1891 /* XXX: may need to support wrapping */
c10731e7 1892 filesize = ic->pb ? avio_size(ic->pb) : 0;
411ff322
MN
1893 end_time = AV_NOPTS_VALUE;
1894 do{
7e6029f9
AC
1895 offset = filesize - (DURATION_MAX_READ_SIZE<<retry);
1896 if (offset < 0)
1897 offset = 0;
12f996ed 1898
7e6029f9
AC
1899 avio_seek(ic->pb, offset, SEEK_SET);
1900 read_size = 0;
1901 for(;;) {
1902 if (read_size >= DURATION_MAX_READ_SIZE<<(FFMAX(retry-1,0)))
1903 break;
115329f1 1904
7e6029f9
AC
1905 do {
1906 ret = av_read_packet(ic, pkt);
1907 } while(ret == AVERROR(EAGAIN));
1908 if (ret != 0)
1909 break;
1910 read_size += pkt->size;
1911 st = ic->streams[pkt->stream_index];
1912 if (pkt->pts != AV_NOPTS_VALUE &&
1913 (st->start_time != AV_NOPTS_VALUE ||
1914 st->first_dts != AV_NOPTS_VALUE)) {
1915 duration = end_time = pkt->pts;
1916 if (st->start_time != AV_NOPTS_VALUE)
1917 duration -= st->start_time;
1918 else
1919 duration -= st->first_dts;
1920 if (duration < 0)
1921 duration += 1LL<<st->pts_wrap_bits;
1922 if (duration > 0) {
1923 if (st->duration == AV_NOPTS_VALUE || st->duration < duration)
1924 st->duration = duration;
1925 }
12f996ed 1926 }
7e6029f9 1927 av_free_packet(pkt);
12f996ed 1928 }
411ff322
MN
1929 }while( end_time==AV_NOPTS_VALUE
1930 && filesize > (DURATION_MAX_READ_SIZE<<retry)
1931 && ++retry <= DURATION_MAX_RETRY);
115329f1 1932
c0df9d75 1933 fill_all_stream_timings(ic);
12f996ed 1934
6b4aa5da 1935 avio_seek(ic->pb, old_offset, SEEK_SET);
e99179de 1936 for (i=0; i<ic->nb_streams; i++) {
df886e7e
MN
1937 st= ic->streams[i];
1938 st->cur_dts= st->first_dts;
635fbcb1 1939 st->last_IP_pts = AV_NOPTS_VALUE;
a8fd2f4e 1940 st->reference_dts = AV_NOPTS_VALUE;
df886e7e 1941 }
12f996ed
FB
1942}
1943
d3bb7191 1944static void estimate_timings(AVFormatContext *ic, int64_t old_offset)
12f996ed 1945{
12f996ed
FB
1946 int64_t file_size;
1947
1948 /* get the file size, if possible */
1949 if (ic->iformat->flags & AVFMT_NOFILE) {
1950 file_size = 0;
1951 } else {
76aa876e 1952 file_size = avio_size(ic->pb);
a7503430 1953 file_size = FFMAX(0, file_size);
12f996ed 1954 }
12f996ed 1955
ff70e601
MR
1956 if ((!strcmp(ic->iformat->name, "mpeg") ||
1957 !strcmp(ic->iformat->name, "mpegts")) &&
8978feda 1958 file_size && ic->pb->seekable) {
12f996ed 1959 /* get accurate estimate from the PTSes */
d3bb7191
AK
1960 estimate_timings_from_pts(ic, old_offset);
1961 } else if (has_duration(ic)) {
a85736f2 1962 /* at least one component has timings - we use them for all
12f996ed
FB
1963 the components */
1964 fill_all_stream_timings(ic);
1965 } else {
77ac76a3 1966 av_log(ic, AV_LOG_WARNING, "Estimating duration from bitrate, this may be inaccurate\n");
a85736f2 1967 /* less precise: use bitrate info */
d3bb7191 1968 estimate_timings_from_bit_rate(ic);
12f996ed 1969 }
d3bb7191 1970 update_stream_timings(ic);
12f996ed 1971
12f996ed
FB
1972 {
1973 int i;
5e1166b3 1974 AVStream av_unused *st;
12f996ed
FB
1975 for(i = 0;i < ic->nb_streams; i++) {
1976 st = ic->streams[i];
045dd4b9
DB
1977 av_dlog(ic, "%d: start_time: %0.3f duration: %0.3f\n", i,
1978 (double) st->start_time / AV_TIME_BASE,
1979 (double) st->duration / AV_TIME_BASE);
12f996ed 1980 }
045dd4b9
DB
1981 av_dlog(ic, "stream: start_time: %0.3f duration: %0.3f bitrate=%d kb/s\n",
1982 (double) ic->start_time / AV_TIME_BASE,
1983 (double) ic->duration / AV_TIME_BASE,
1984 ic->bit_rate / 1000);
12f996ed 1985 }
12f996ed
FB
1986}
1987
25dfda7f 1988static int has_codec_parameters(AVCodecContext *avctx)
b9a281db
FB
1989{
1990 int val;
25dfda7f 1991 switch (avctx->codec_type) {
72415b2a 1992 case AVMEDIA_TYPE_AUDIO:
25dfda7f
SS
1993 val = avctx->sample_rate && avctx->channels && avctx->sample_fmt != AV_SAMPLE_FMT_NONE;
1994 if (!avctx->frame_size &&
5602a464 1995 (avctx->codec_id == CODEC_ID_AAC ||
25dfda7f
SS
1996 avctx->codec_id == CODEC_ID_MP1 ||
1997 avctx->codec_id == CODEC_ID_MP2 ||
1998 avctx->codec_id == CODEC_ID_MP3 ||
4ca59d19 1999 avctx->codec_id == CODEC_ID_CELT))
4d35bf74 2000 return 0;
b9a281db 2001 break;
72415b2a 2002 case AVMEDIA_TYPE_VIDEO:
25dfda7f 2003 val = avctx->width && avctx->pix_fmt != PIX_FMT_NONE;
b9a281db
FB
2004 break;
2005 default:
2006 val = 1;
2007 break;
2008 }
25dfda7f 2009 return avctx->codec_id != CODEC_ID_NONE && val != 0;
b9a281db
FB
2010}
2011
ae447836
BC
2012static int has_decode_delay_been_guessed(AVStream *st)
2013{
2014 return st->codec->codec_id != CODEC_ID_H264 ||
38a4be3f 2015 st->info->nb_decoded_frames >= 6;
ae447836
BC
2016}
2017
b3461c29 2018/* returns 1 or 0 if or if not decoded data was returned, or a negative error */
a67c061e 2019static int try_decode_frame(AVStream *st, AVPacket *avpkt, AVDictionary **options)
fb2758c8 2020{
fb2758c8 2021 AVCodec *codec;
212fd3a1 2022 int got_picture = 1, ret = 0;
fb2758c8 2023 AVFrame picture;
f08e54e8 2024 AVPacket pkt = *avpkt;
115329f1 2025
af08d9ae 2026 if (!avcodec_is_open(st->codec)) {
59297ad6
JG
2027 AVDictionary *thread_opt = NULL;
2028
bc901998
AK
2029 codec = st->codec->codec ? st->codec->codec :
2030 avcodec_find_decoder(st->codec->codec_id);
2031
e472ea34
BC
2032 if (!codec)
2033 return -1;
59297ad6
JG
2034
2035 /* force thread count to 1 since the h264 decoder will not extract SPS
2036 * and PPS to extradata during multi-threaded decoding */
2037 av_dict_set(options ? options : &thread_opt, "threads", "1", 0);
2038 ret = avcodec_open2(st->codec, codec, options ? options : &thread_opt);
2039 if (!options)
2040 av_dict_free(&thread_opt);
e472ea34
BC
2041 if (ret < 0)
2042 return ret;
2043 }
644a9262 2044
212fd3a1
AK
2045 while ((pkt.size > 0 || (!pkt.data && got_picture)) &&
2046 ret >= 0 &&
f08e54e8
JR
2047 (!has_codec_parameters(st->codec) ||
2048 !has_decode_delay_been_guessed(st) ||
2049 (!st->codec_info_nb_frames && st->codec->codec->capabilities & CODEC_CAP_CHANNEL_CONF))) {
2050 got_picture = 0;
2051 avcodec_get_frame_defaults(&picture);
e472ea34 2052 switch(st->codec->codec_type) {
72415b2a 2053 case AVMEDIA_TYPE_VIDEO:
e472ea34 2054 ret = avcodec_decode_video2(st->codec, &picture,
f08e54e8 2055 &got_picture, &pkt);
e472ea34 2056 break;
72415b2a 2057 case AVMEDIA_TYPE_AUDIO:
f08e54e8 2058 ret = avcodec_decode_audio4(st->codec, &picture, &got_picture, &pkt);
e472ea34
BC
2059 break;
2060 default:
2061 break;
2062 }
f08e54e8
JR
2063 if (ret >= 0) {
2064 if (got_picture)
2065 st->info->nb_decoded_frames++;
2066 pkt.data += ret;
2067 pkt.size -= ret;
b3461c29 2068 ret = got_picture;
f08e54e8 2069 }
fb2758c8 2070 }
fb2758c8
FB
2071 return ret;
2072}
2073
83c27079 2074unsigned int ff_codec_get_tag(const AVCodecTag *tags, enum CodecID id)
45da8124
AJ
2075{
2076 while (tags->id != CODEC_ID_NONE) {
2077 if (tags->id == id)
2078 return tags->tag;
2079 tags++;
2080 }
2081 return 0;
2082}
2083
1a40491e 2084enum CodecID ff_codec_get_id(const AVCodecTag *tags, unsigned int tag)
45da8124 2085{
41415d28
MN
2086 int i;
2087 for(i=0; tags[i].id != CODEC_ID_NONE;i++) {
2088 if(tag == tags[i].tag)
2089 return tags[i].id;
2090 }
2091 for(i=0; tags[i].id != CODEC_ID_NONE; i++) {
0842d589 2092 if (avpriv_toupper4(tag) == avpriv_toupper4(tags[i].tag))
41415d28 2093 return tags[i].id;
45da8124
AJ
2094 }
2095 return CODEC_ID_NONE;
2096}
2097
15545a09 2098unsigned int av_codec_get_tag(const AVCodecTag * const *tags, enum CodecID id)
45da8124
AJ
2099{
2100 int i;
2101 for(i=0; tags && tags[i]; i++){
1a40491e 2102 int tag= ff_codec_get_tag(tags[i], id);
45da8124
AJ
2103 if(tag) return tag;
2104 }
2105 return 0;
2106}
2107
15545a09 2108enum CodecID av_codec_get_id(const AVCodecTag * const *tags, unsigned int tag)
45da8124
AJ
2109{
2110 int i;
2111 for(i=0; tags && tags[i]; i++){
1a40491e 2112 enum CodecID id= ff_codec_get_id(tags[i], tag);
45da8124
AJ
2113 if(id!=CODEC_ID_NONE) return id;
2114 }
2115 return CODEC_ID_NONE;
2116}
2117
c2c3dedf
AJ
2118static void compute_chapters_end(AVFormatContext *s)
2119{
ab11317c 2120 unsigned int i, j;
656566d7 2121 int64_t max_time = s->duration + ((s->start_time == AV_NOPTS_VALUE) ? 0 : s->start_time);
c2c3dedf 2122
ab11317c 2123 for (i = 0; i < s->nb_chapters; i++)
c2c3dedf 2124 if (s->chapters[i]->end == AV_NOPTS_VALUE) {
ab11317c
AK
2125 AVChapter *ch = s->chapters[i];
2126 int64_t end = max_time ? av_rescale_q(max_time, AV_TIME_BASE_Q, ch->time_base)
2127 : INT64_MAX;
2128
2129 for (j = 0; j < s->nb_chapters; j++) {
2130 AVChapter *ch1 = s->chapters[j];
2131 int64_t next_start = av_rescale_q(ch1->start, ch1->time_base, ch->time_base);
2132 if (j != i && next_start > ch->start && next_start < end)
2133 end = next_start;
2134 }
2135 ch->end = (end == INT64_MAX) ? ch->start : end;
c2c3dedf 2136 }
c2c3dedf
AJ
2137}
2138
4d43cbcc
MN
2139static int get_std_framerate(int i){
2140 if(i<60*12) return i*1001;
aecf157e 2141 else return ((const int[]){24,30,60,12,15})[i-60*12]*1000*12;
4d43cbcc
MN
2142}
2143
945208ca
MN
2144/*
2145 * Is the time base unreliable.
2146 * This is a heuristic to balance between quick acceptance of the values in
2147 * the headers vs. some extra checks.
a85736f2
DB
2148 * Old DivX and Xvid often have nonsense timebases like 1fps or 2fps.
2149 * MPEG-2 commonly misuses field repeat flags to store different framerates.
945208ca
MN
2150 * And there are "variable" fps files this needs to detect as well.
2151 */
2152static int tb_unreliable(AVCodecContext *c){
2153 if( c->time_base.den >= 101L*c->time_base.num
2154 || c->time_base.den < 5L*c->time_base.num
2bb6eba2
AJ
2155/* || c->codec_tag == AV_RL32("DIVX")
2156 || c->codec_tag == AV_RL32("XVID")*/
7f123e7f
MN
2157 || c->codec_id == CODEC_ID_MPEG2VIDEO
2158 || c->codec_id == CODEC_ID_H264
2159 )
945208ca
MN
2160 return 1;
2161 return 0;
2162}
2163
a67c061e
AK
2164int avformat_find_stream_info(AVFormatContext *ic, AVDictionary **options)
2165{
9f75260e 2166 int i, count, ret, read_size, j;
b9a281db 2167 AVStream *st;
fb2758c8 2168 AVPacket pkt1, *pkt;
a2704c97 2169 int64_t old_offset = avio_tell(ic->pb);
a67c061e 2170 int orig_nb_streams = ic->nb_streams; // new streams might appear, no options for those
0cbff027 2171
c0df9d75 2172 for(i=0;i<ic->nb_streams;i++) {
62784e37 2173 AVCodec *codec;
59297ad6 2174 AVDictionary *thread_opt = NULL;
c0df9d75 2175 st = ic->streams[i];
dafaef2f 2176
90ad92b3 2177 //only for the split stuff
fe8344a2 2178 if (!st->parser && !(ic->flags & AVFMT_FLAG_NOPARSE)) {
01f4895c 2179 st->parser = av_parser_init(st->codec->codec_id);
57004ff1 2180 if(st->need_parsing == AVSTREAM_PARSE_HEADERS && st->parser){
7cbaa7ba
MN
2181 st->parser->flags |= PARSER_FLAG_COMPLETE_FRAMES;
2182 }
90ad92b3 2183 }
bc901998
AK
2184 codec = st->codec->codec ? st->codec->codec :
2185 avcodec_find_decoder(st->codec->codec_id);
62784e37 2186
59297ad6
JG
2187 /* force thread count to 1 since the h264 decoder will not extract SPS
2188 * and PPS to extradata during multi-threaded decoding */
2189 av_dict_set(options ? &options[i] : &thread_opt, "threads", "1", 0);
2190
cb2c971d
AJ
2191 /* Ensure that subtitle_header is properly set. */
2192 if (st->codec->codec_type == AVMEDIA_TYPE_SUBTITLE
2193 && codec && !st->codec->codec)
59297ad6
JG
2194 avcodec_open2(st->codec, codec, options ? &options[i]
2195 : &thread_opt);
cb2c971d 2196
43e4d57f
MN
2197 //try to just open decoders, in case this is enough to get parameters
2198 if(!has_codec_parameters(st->codec)){
cb2c971d 2199 if (codec && !st->codec->codec)
59297ad6
JG
2200 avcodec_open2(st->codec, codec, options ? &options[i]
2201 : &thread_opt);
43e4d57f 2202 }
59297ad6
JG
2203 if (!options)
2204 av_dict_free(&thread_opt);
c0df9d75
MN
2205 }
2206
feb2440c 2207 for (i=0; i<ic->nb_streams; i++) {
fd0368e7 2208 ic->streams[i]->info->last_dts = AV_NOPTS_VALUE;
15bc38e5 2209 }
115329f1 2210
b9a281db
FB
2211 count = 0;
2212 read_size = 0;
b9a281db 2213 for(;;) {
9957cdbf 2214 if (ff_check_interrupt(&ic->interrupt_callback)){
c76374c6 2215 ret= AVERROR_EXIT;
71ee6515 2216 av_log(ic, AV_LOG_DEBUG, "interrupted\n");
1d14361d
MN
2217 break;
2218 }
2219
b9a281db
FB
2220 /* check if one codec still needs to be handled */
2221 for(i=0;i<ic->nb_streams;i++) {
7c152a45
AH
2222 int fps_analyze_framecount = 20;
2223
b9a281db 2224 st = ic->streams[i];
01f4895c 2225 if (!has_codec_parameters(st->codec))
b9a281db 2226 break;
7c152a45
AH
2227 /* if the timebase is coarse (like the usual millisecond precision
2228 of mkv), we need to analyze more frames to reliably arrive at
2229 the correct fps */
2230 if (av_q2d(st->time_base) > 0.0005)
2231 fps_analyze_framecount *= 2;
30315a8d
AC
2232 if (ic->fps_probe_size >= 0)
2233 fps_analyze_framecount = ic->fps_probe_size;
3e76d1b5 2234 /* variable fps and no guess at the real fps */
0e5f33f2 2235 if( tb_unreliable(st->codec) && !(st->r_frame_rate.num && st->avg_frame_rate.num)
7c152a45
AH
2236 && st->info->duration_count < fps_analyze_framecount
2237 && st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
3e76d1b5 2238 break;
01f4895c 2239 if(st->parser && st->parser->parser->split && !st->codec->extradata)
90ad92b3 2240 break;
82583548
MN
2241 if(st->first_dts == AV_NOPTS_VALUE)
2242 break;
b9a281db
FB
2243 }
2244 if (i == ic->nb_streams) {
2245 /* NOTE: if the format has no header, then we need to read
2246 some packets to get most of the streams, so we cannot
2247 stop here */
fb2758c8 2248 if (!(ic->ctx_flags & AVFMTCTX_NOHEADER)) {
b9a281db
FB
2249 /* if we found the info for all the codecs, we can stop */
2250 ret = count;
71ee6515 2251 av_log(ic, AV_LOG_DEBUG, "All info found\n");
b9a281db
FB
2252 break;
2253 }
5fb83c38 2254 }
35eab0c0 2255 /* we did not get all the codec info, but we read too much data */
57011a13 2256 if (read_size >= ic->probesize) {
35eab0c0 2257 ret = count;
9bbe9a0d 2258 av_log(ic, AV_LOG_DEBUG, "Probe buffer size limit %d reached\n", ic->probesize);
35eab0c0
MN
2259 break;
2260 }
b9a281db 2261
fb2758c8
FB
2262 /* NOTE: a new stream can be added there if no header in file
2263 (AVFMTCTX_NOHEADER) */
d3bb7191 2264 ret = read_frame_internal(ic, &pkt1);
59ca3955
AF
2265 if (ret == AVERROR(EAGAIN))
2266 continue;
2267
2268 if (ret < 0) {
212fd3a1
AK
2269 /* EOF or error*/
2270 AVPacket empty_pkt = { 0 };
2271 int err;
2272 av_init_packet(&empty_pkt);
2273
fb2758c8 2274 ret = -1; /* we could not have all the codec parameters before EOF */
e19456e3
MN
2275 for(i=0;i<ic->nb_streams;i++) {
2276 st = ic->streams[i];
212fd3a1
AK
2277
2278 /* flush the decoders */
b3461c29
JG
2279 do {
2280 err = try_decode_frame(st, &empty_pkt,
2281 (options && i < orig_nb_streams) ?
2282 &options[i] : NULL);
2283 } while (err > 0 && !has_codec_parameters(st->codec));
2284
2285 if (err < 0) {
2286 av_log(ic, AV_LOG_WARNING,
2287 "decoding for stream %d failed\n", st->index);
2288 } else if (!has_codec_parameters(st->codec)){
305ee50f
MN
2289 char buf[256];
2290 avcodec_string(buf, sizeof(buf), st->codec, 0);
b3461c29
JG
2291 av_log(ic, AV_LOG_WARNING,
2292 "Could not find codec parameters (%s)\n", buf);
344a18c3
MR
2293 } else {
2294 ret = 0;
305ee50f 2295 }
e19456e3 2296 }
fb2758c8
FB
2297 break;
2298 }
2299
5c5b1731 2300 pkt= add_to_pktbuf(&ic->packet_buffer, &pkt1, &ic->packet_buffer_end);
fd0368e7
AJ
2301 if ((ret = av_dup_packet(pkt)) < 0)
2302 goto find_stream_info_err;
b9a281db 2303
fb2758c8 2304 read_size += pkt->size;
b9a281db
FB
2305
2306 st = ic->streams[pkt->stream_index];
fd0368e7
AJ
2307 if (st->codec_info_nb_frames>1) {
2308 if (st->time_base.den > 0 && av_rescale_q(st->info->codec_info_duration, st->time_base, AV_TIME_BASE_Q) >= ic->max_analyze_duration) {
657eca1f 2309 av_log(ic, AV_LOG_WARNING, "max_analyze_duration reached\n");
3a560188 2310 break;
71ee6515 2311 }
fd0368e7 2312 st->info->codec_info_duration += pkt->duration;
3a560188 2313 }
cefe0607 2314 {
fd0368e7 2315 int64_t last = st->info->last_dts;
3c150d16 2316
a31e9f68
MR
2317 if(pkt->dts != AV_NOPTS_VALUE && last != AV_NOPTS_VALUE && pkt->dts > last){
2318 int64_t duration= pkt->dts - last;
4d43cbcc
MN
2319 double dur= duration * av_q2d(st->time_base);
2320
72415b2a 2321// if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
4d43cbcc 2322// av_log(NULL, AV_LOG_ERROR, "%f\n", dur);
fd0368e7
AJ
2323 if (st->info->duration_count < 2)
2324 memset(st->info->duration_error, 0, sizeof(st->info->duration_error));
2325 for (i=1; i<FF_ARRAY_ELEMS(st->info->duration_error); i++) {
69c262d1
MN
2326 int framerate= get_std_framerate(i);
2327 int ticks= lrintf(dur*framerate/(1001*12));
52767d89 2328 double error = dur - (double)ticks*1001*12 / framerate;
fd0368e7 2329 st->info->duration_error[i] += error*error;
69c262d1 2330 }
fd0368e7 2331 st->info->duration_count++;
85142724 2332 // ignore the first 4 values, they might have some random jitter
fd0368e7
AJ
2333 if (st->info->duration_count > 3)
2334 st->info->duration_gcd = av_gcd(st->info->duration_gcd, duration);
15bc38e5 2335 }
fd0368e7
AJ
2336 if (last == AV_NOPTS_VALUE || st->info->duration_count <= 1)
2337 st->info->last_dts = pkt->dts;
15bc38e5 2338 }
01f4895c
MN
2339 if(st->parser && st->parser->parser->split && !st->codec->extradata){
2340 int i= st->parser->parser->split(st->codec, pkt->data, pkt->size);
4df30f71 2341 if (i > 0 && i < FF_MAX_EXTRADATA_SIZE) {
01f4895c 2342 st->codec->extradata_size= i;
62c52121 2343 st->codec->extradata= av_malloc(st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
4df30f71
AC
2344 if (!st->codec->extradata)
2345 return AVERROR(ENOMEM);
01f4895c 2346 memcpy(st->codec->extradata, pkt->data, st->codec->extradata_size);
62c52121 2347 memset(st->codec->extradata + i, 0, FF_INPUT_BUFFER_PADDING_SIZE);
90ad92b3
MN
2348 }
2349 }
115329f1 2350
fb2758c8
FB
2351 /* if still no information, we try to open the codec and to
2352 decompress the frame. We try to avoid that in most cases as
a85736f2 2353 it takes longer and uses more memory. For MPEG-4, we need to
dafaef2f
BL
2354 decompress for QuickTime.
2355
2356 If CODEC_CAP_CHANNEL_CONF is set this will force decoding of at
2357 least one frame of codec data, this makes sure the codec initializes
2358 the channel configuration and does not only trust the values from the container.
2359 */
212fd3a1 2360 try_decode_frame(st, pkt, (options && i < orig_nb_streams ) ? &options[i] : NULL);
115329f1 2361
9d3fdf20 2362 st->codec_info_nb_frames++;
b9a281db
FB
2363 count++;
2364 }
2365
bd107136 2366 // close codecs which were opened in try_decode_frame()
43c0040a
MN
2367 for(i=0;i<ic->nb_streams;i++) {
2368 st = ic->streams[i];
af08d9ae 2369 avcodec_close(st->codec);
43c0040a 2370 }
b9a281db
FB
2371 for(i=0;i<ic->nb_streams;i++) {
2372 st = ic->streams[i];
fd0368e7 2373 if (st->codec_info_nb_frames>2 && !st->avg_frame_rate.num && st->info->codec_info_duration)
02b398ef 2374 av_reduce(&st->avg_frame_rate.num, &st->avg_frame_rate.den,
6c6e6ef5 2375 (st->codec_info_nb_frames-2)*(int64_t)st->time_base.den,
fd0368e7 2376 st->info->codec_info_duration*(int64_t)st->time_base.num, 60000);
72415b2a 2377 if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
85142724
RD
2378 // the check for tb_unreliable() is not completely correct, since this is not about handling
2379 // a unreliable/inexact time base, but a time base that is finer than necessary, as e.g.
2380 // ipmovie.c produces.
fd0368e7
AJ
2381 if (tb_unreliable(st->codec) && st->info->duration_count > 15 && st->info->duration_gcd > 1 && !st->r_frame_rate.num)
2382 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);
2383 if (st->info->duration_count && !st->r_frame_rate.num
945208ca 2384 && tb_unreliable(st->codec) /*&&
a85736f2 2385 //FIXME we should not special-case MPEG-2, but this needs testing with non-MPEG-2 ...
fd0368e7 2386 st->time_base.num*duration_sum[i]/st->info->duration_count*101LL > st->time_base.den*/){
fe02d9e7 2387 int num = 0;
4d43cbcc 2388 double best_error= 2*av_q2d(st->time_base);
fd0368e7 2389 best_error = best_error*best_error*st->info->duration_count*1000*12*30;
4d43cbcc 2390
fd0368e7
AJ
2391 for (j=1; j<FF_ARRAY_ELEMS(st->info->duration_error); j++) {
2392 double error = st->info->duration_error[j] * get_std_framerate(j);
72415b2a 2393// if(st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
4d43cbcc 2394// av_log(NULL, AV_LOG_ERROR, "%f %f\n", get_std_framerate(j) / 12.0/1001, error);
9f75260e
MN
2395 if(error < best_error){
2396 best_error= error;
fe02d9e7 2397 num = get_std_framerate(j);
9f75260e 2398 }
3c150d16 2399 }
fe02d9e7
RD
2400 // do not increase frame rate by more than 1 % in order to match a standard rate.
2401 if (num && (!st->r_frame_rate.num || (double)num/(12*1001) < 1.01 * av_q2d(st->r_frame_rate)))
2402 av_reduce(&st->r_frame_rate.num, &st->r_frame_rate.den, num, 12*1001, INT_MAX);
15bc38e5 2403 }
72415b2a 2404 }else if(st->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
dd1c8f3e
LA
2405 if(!st->codec->bits_per_coded_sample)
2406 st->codec->bits_per_coded_sample= av_get_bits_per_sample(st->codec->codec_id);
c70a6a41
JR
2407 // set stream disposition based on audio service type
2408 switch (st->codec->audio_service_type) {
2409 case AV_AUDIO_SERVICE_TYPE_EFFECTS:
2410 st->disposition = AV_DISPOSITION_CLEAN_EFFECTS; break;
2411 case AV_AUDIO_SERVICE_TYPE_VISUALLY_IMPAIRED:
2412 st->disposition = AV_DISPOSITION_VISUAL_IMPAIRED; break;
2413 case AV_AUDIO_SERVICE_TYPE_HEARING_IMPAIRED:
2414 st->disposition = AV_DISPOSITION_HEARING_IMPAIRED; break;
2415 case AV_AUDIO_SERVICE_TYPE_COMMENTARY:
2416 st->disposition = AV_DISPOSITION_COMMENT; break;
2417 case AV_AUDIO_SERVICE_TYPE_KARAOKE:
2418 st->disposition = AV_DISPOSITION_KARAOKE; break;
2419 }
b9a281db 2420 }
de6d9b64 2421 }
b9a281db 2422
d3bb7191 2423 estimate_timings(ic, old_offset);
6fea687e 2424
c2c3dedf
AJ
2425 compute_chapters_end(ic);
2426
e928649b 2427#if 0
a85736f2 2428 /* correct DTS for B-frame streams with no timestamps */
e928649b
MN
2429 for(i=0;i<ic->nb_streams;i++) {
2430 st = ic->streams[i];
72415b2a 2431 if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
e928649b
MN
2432 if(b-frames){
2433 ppktl = &ic->packet_buffer;
2434 while(ppkt1){
2435 if(ppkt1->stream_index != i)
2436 continue;
2437 if(ppkt1->pkt->dts < 0)
2438 break;
2439 if(ppkt1->pkt->pts != AV_NOPTS_VALUE)
2440 break;
2441 ppkt1->pkt->dts -= delta;
2442 ppkt1= ppkt1->next;
2443 }
2444 if(ppkt1)
2445 continue;
2446 st->cur_dts -= delta;
2447 }
2448 }
2449 }
2450#endif
0cbff027 2451
fd0368e7 2452 find_stream_info_err:
e4e30256
JG
2453 for (i=0; i < ic->nb_streams; i++) {
2454 if (ic->streams[i]->codec)
2455 ic->streams[i]->codec->thread_count = 0;
fd0368e7 2456 av_freep(&ic->streams[i]->info);
e4e30256 2457 }
b9a281db 2458 return ret;
de6d9b64
FB
2459}
2460
9128ae08
NG
2461static AVProgram *find_program_from_stream(AVFormatContext *ic, int s)
2462{
2463 int i, j;
2464
2465 for (i = 0; i < ic->nb_programs; i++)
2466 for (j = 0; j < ic->programs[i]->nb_stream_indexes; j++)
2467 if (ic->programs[i]->stream_index[j] == s)
2468 return ic->programs[i];
2469 return NULL;
2470}
2471
2472int av_find_best_stream(AVFormatContext *ic,
2473 enum AVMediaType type,
2474 int wanted_stream_nb,
2475 int related_stream,
2476 AVCodec **decoder_ret,
2477 int flags)
2478{
2c715816 2479 int i, nb_streams = ic->nb_streams;
9128ae08
NG
2480 int ret = AVERROR_STREAM_NOT_FOUND, best_count = -1;
2481 unsigned *program = NULL;
2482 AVCodec *decoder = NULL, *best_decoder = NULL;
2483
2484 if (related_stream >= 0 && wanted_stream_nb < 0) {
2485 AVProgram *p = find_program_from_stream(ic, related_stream);
2486 if (p) {
2487 program = p->stream_index;
2488 nb_streams = p->nb_stream_indexes;
2489 }
2490 }
2491 for (i = 0; i < nb_streams; i++) {
2c715816
MB
2492 int real_stream_index = program ? program[i] : i;
2493 AVStream *st = ic->streams[real_stream_index];
9128ae08
NG
2494 AVCodecContext *avctx = st->codec;
2495 if (avctx->codec_type != type)
2496 continue;
2c715816 2497 if (wanted_stream_nb >= 0 && real_stream_index != wanted_stream_nb)
9128ae08 2498 continue;
52091491
PR
2499 if (st->disposition & (AV_DISPOSITION_HEARING_IMPAIRED|AV_DISPOSITION_VISUAL_IMPAIRED))
2500 continue;
9128ae08 2501 if (decoder_ret) {
6b35a795 2502 decoder = avcodec_find_decoder(st->codec->codec_id);
9128ae08
NG
2503 if (!decoder) {
2504 if (ret < 0)
2505 ret = AVERROR_DECODER_NOT_FOUND;
2506 continue;
2507 }
2508 }
2509 if (best_count >= st->codec_info_nb_frames)
2510 continue;
2511 best_count = st->codec_info_nb_frames;
2c715816 2512 ret = real_stream_index;
9128ae08
NG
2513 best_decoder = decoder;
2514 if (program && i == nb_streams - 1 && ret < 0) {
2515 program = NULL;
2516 nb_streams = ic->nb_streams;
2517 i = 0; /* no related stream found, try again with everything */
2518 }
2519 }
2520 if (decoder_ret)
2521 *decoder_ret = best_decoder;
2522 return ret;
2523}
2524
fb2758c8
FB
2525/*******************************************************/
2526
fb2758c8
FB
2527int av_read_play(AVFormatContext *s)
2528{
fa13095a
BA
2529 if (s->iformat->read_play)
2530 return s->iformat->read_play(s);
fd2982a0 2531 if (s->pb)
ff1ec0c3 2532 return avio_pause(s->pb, 0);
fa13095a 2533 return AVERROR(ENOSYS);
fb2758c8
FB
2534}
2535
fb2758c8
FB
2536int av_read_pause(AVFormatContext *s)
2537{
fa13095a
BA
2538 if (s->iformat->read_pause)
2539 return s->iformat->read_pause(s);
fd2982a0 2540 if (s->pb)
ff1ec0c3 2541 return avio_pause(s->pb, 1);
fa13095a 2542 return AVERROR(ENOSYS);
fb2758c8
FB
2543}
2544
f124b087
MS
2545void avformat_free_context(AVFormatContext *s)
2546{
2547 int i;
2548 AVStream *st;
2549
36773283 2550 av_opt_free(s);
dbaba52e 2551 if (s->iformat && s->iformat->priv_class && s->priv_data)
36773283
AK
2552 av_opt_free(s->priv_data);
2553
de6d9b64 2554 for(i=0;i<s->nb_streams;i++) {
da24c5e3
FB
2555 /* free all data in a stream component */
2556 st = s->streams[i];
fb2758c8
FB
2557 if (st->parser) {
2558 av_parser_close(st->parser);
3a41c2f7 2559 av_free_packet(&st->cur_pkt);
de6d9b64 2560 }
dd2a4bcf
AK
2561 if (st->attached_pic.data)
2562 av_free_packet(&st->attached_pic);
d2d67e42 2563 av_dict_free(&st->metadata);
fb2758c8 2564 av_free(st->index_entries);
a5e9102b 2565 av_free(st->codec->extradata);
cb2c971d 2566 av_free(st->codec->subtitle_header);
01f4895c 2567 av_free(st->codec);
ade8d8b9 2568 av_free(st->priv_data);
fd0368e7 2569 av_free(st->info);
fb2758c8 2570 av_free(st);
de6d9b64 2571 }
15afa396 2572 for(i=s->nb_programs-1; i>=0; i--) {
d2d67e42 2573 av_dict_free(&s->programs[i]->metadata);
526efa10 2574 av_freep(&s->programs[i]->stream_index);
15afa396
NS
2575 av_freep(&s->programs[i]);
2576 }
ceedda75 2577 av_freep(&s->programs);
a8dbe951 2578 av_freep(&s->priv_data);
7c8202cc 2579 while(s->nb_chapters--) {
d2d67e42 2580 av_dict_free(&s->chapters[s->nb_chapters]->metadata);
7c8202cc 2581 av_free(s->chapters[s->nb_chapters]);
79d7836a
AK
2582 }
2583 av_freep(&s->chapters);
d2d67e42 2584 av_dict_free(&s->metadata);
7bbb67d5 2585 av_freep(&s->streams);
1ea4f593 2586 av_free(s);
de6d9b64
FB
2587}
2588
52660454 2589#if FF_API_CLOSE_INPUT_FILE
2506fd54
RD
2590void av_close_input_file(AVFormatContext *s)
2591{
52660454
AK
2592 avformat_close_input(&s);
2593}
2594#endif
2595
2596void avformat_close_input(AVFormatContext **ps)
2597{
2598 AVFormatContext *s = *ps;
05e84c95
AK
2599 AVIOContext *pb = (s->iformat->flags & AVFMT_NOFILE) || (s->flags & AVFMT_FLAG_CUSTOM_IO) ?
2600 NULL : s->pb;
3a7f7678
AK
2601 flush_packet_queue(s);
2602 if (s->iformat->read_close)
2603 s->iformat->read_close(s);
2604 avformat_free_context(s);
52660454 2605 *ps = NULL;
2506fd54 2606 if (pb)
22a3212e 2607 avio_close(pb);
2506fd54
RD
2608}
2609
569129a6
AK
2610AVStream *avformat_new_stream(AVFormatContext *s, AVCodec *c)
2611{
b9a281db 2612 AVStream *st;
504ee036 2613 int i;
38aab35f 2614 AVStream **streams;
b9a281db 2615
38aab35f
AJ
2616 if (s->nb_streams >= INT_MAX/sizeof(*streams))
2617 return NULL;
2618 streams = av_realloc(s->streams, (s->nb_streams + 1) * sizeof(*streams));
2619 if (!streams)
2620 return NULL;
2621 s->streams = streams;
b9a281db
FB
2622
2623 st = av_mallocz(sizeof(AVStream));
2624 if (!st)
2625 return NULL;
fd0368e7
AJ
2626 if (!(st->info = av_mallocz(sizeof(*st->info)))) {
2627 av_free(st);
2628 return NULL;
2629 }
115329f1 2630
569129a6 2631 st->codec = avcodec_alloc_context3(c);
48091512
FB
2632 if (s->iformat) {
2633 /* no default bitrate if decoding */
01f4895c 2634 st->codec->bit_rate = 0;
48091512 2635 }
b9a281db 2636 st->index = s->nb_streams;
12f996ed
FB
2637 st->start_time = AV_NOPTS_VALUE;
2638 st->duration = AV_NOPTS_VALUE;
6d77d9ac
MN
2639 /* we set the current DTS to 0 so that formats without any timestamps
2640 but durations get some timestamps, formats with some unknown
2641 timestamps have their first few packets buffered and the
2642 timestamps corrected before they are returned to the user */
2643 st->cur_dts = 0;
82583548 2644 st->first_dts = AV_NOPTS_VALUE;
86cb7e33 2645 st->probe_packets = MAX_PROBE_PACKETS;
9ee91c2f 2646
a85736f2 2647 /* default pts setting is MPEG-like */
c3f9ebf7 2648 avpriv_set_pts_info(st, 33, 1, 90000);
635fbcb1 2649 st->last_IP_pts = AV_NOPTS_VALUE;
504ee036
MN
2650 for(i=0; i<MAX_REORDER_DELAY+1; i++)
2651 st->pts_buffer[i]= AV_NOPTS_VALUE;
27ca0a79 2652 st->reference_dts = AV_NOPTS_VALUE;
9ee91c2f 2653
c30a4489
AJ
2654 st->sample_aspect_ratio = (AVRational){0,1};
2655
b9a281db
FB
2656 s->streams[s->nb_streams++] = st;
2657 return st;
2658}
2659
15afa396
NS
2660AVProgram *av_new_program(AVFormatContext *ac, int id)
2661{
2662 AVProgram *program=NULL;
2663 int i;
2664
919d7a34 2665 av_dlog(ac, "new_program: id=0x%04x\n", id);
15afa396
NS
2666
2667 for(i=0; i<ac->nb_programs; i++)
2668 if(ac->programs[i]->id == id)
2669 program = ac->programs[i];
2670
2671 if(!program){
2672 program = av_mallocz(sizeof(AVProgram));
2673 if (!program)
2674 return NULL;
2675 dynarray_add(&ac->programs, &ac->nb_programs, program);
2676 program->discard = AVDISCARD_NONE;
2677 }
2678 program->id = id;
2679
2680 return program;
2681}
2682
1fa395e4 2683AVChapter *avpriv_new_chapter(AVFormatContext *s, int id, AVRational time_base, int64_t start, int64_t end, const char *title)
79d7836a 2684{
7a2a3e8e
MN
2685 AVChapter *chapter = NULL;
2686 int i;
2687
7c8202cc 2688 for(i=0; i<s->nb_chapters; i++)
7a2a3e8e
MN
2689 if(s->chapters[i]->id == id)
2690 chapter = s->chapters[i];
2691
2692 if(!chapter){
2693 chapter= av_mallocz(sizeof(AVChapter));
6b43e2c7 2694 if(!chapter)
5c37f43a 2695 return NULL;
7c8202cc 2696 dynarray_add(&s->chapters, &s->nb_chapters, chapter);
7a2a3e8e 2697 }
d2d67e42 2698 av_dict_set(&chapter->metadata, "title", title, 0);
7a2a3e8e 2699 chapter->id = id;
abd2256d 2700 chapter->time_base= time_base;
79d7836a 2701 chapter->start = start;
747fb6c6 2702 chapter->end = end;
79d7836a 2703
5c37f43a 2704 return chapter;
79d7836a 2705}
15afa396 2706
b9a281db
FB
2707/************************************************************/
2708/* output media file */
de6d9b64 2709
698f4cc7
FL
2710static int validate_codec_tag(AVFormatContext *s, AVStream *st)
2711{
2712 const AVCodecTag *avctag;
2713 int n;
2714 enum CodecID id = CODEC_ID_NONE;
2715 unsigned int tag = 0;
2716
2717 /**
2718 * Check that tag + id is in the table
2719 * If neither is in the table -> OK
2720 * If tag is in the table with another id -> FAIL
2721 * If id is in the table with another tag -> FAIL unless strict < normal
2722 */
2723 for (n = 0; s->oformat->codec_tag[n]; n++) {
2724 avctag = s->oformat->codec_tag[n];
2725 while (avctag->id != CODEC_ID_NONE) {
0842d589 2726 if (avpriv_toupper4(avctag->tag) == avpriv_toupper4(st->codec->codec_tag)) {
698f4cc7
FL
2727 id = avctag->id;
2728 if (id == st->codec->codec_id)
2729 return 1;
2730 }
2731 if (avctag->id == st->codec->codec_id)
2732 tag = avctag->tag;
2733 avctag++;
2734 }
2735 }
2736 if (id != CODEC_ID_NONE)
2737 return 0;
2738 if (tag && (st->codec->strict_std_compliance >= FF_COMPLIANCE_NORMAL))
2739 return 0;
2740 return 1;
2741}
2742
25de5958
AK
2743int avformat_write_header(AVFormatContext *s, AVDictionary **options)
2744{
2745 int ret = 0, i;
1e51d801 2746 AVStream *st;
25de5958
AK
2747 AVDictionary *tmp = NULL;
2748
2749 if (options)
2750 av_dict_copy(&tmp, *options, 0);
2751 if ((ret = av_opt_set_dict(s, &tmp)) < 0)
2752 goto fail;
1e51d801 2753
9450118b 2754 // some sanity checks
bb62d5c1 2755 if (s->nb_streams == 0 && !(s->oformat->flags & AVFMT_NOSTREAMS)) {
094a63f1 2756 av_log(s, AV_LOG_ERROR, "no streams\n");
25de5958
AK
2757 ret = AVERROR(EINVAL);
2758 goto fail;
094a63f1
TH
2759 }
2760
9450118b
MN
2761 for(i=0;i<s->nb_streams;i++) {
2762 st = s->streams[i];
2763
2764 switch (st->codec->codec_type) {
72415b2a 2765 case AVMEDIA_TYPE_AUDIO:
9450118b
MN
2766 if(st->codec->sample_rate<=0){
2767 av_log(s, AV_LOG_ERROR, "sample rate not set\n");
25de5958
AK
2768 ret = AVERROR(EINVAL);
2769 goto fail;
9450118b 2770 }
bf912a48
BC
2771 if(!st->codec->block_align)
2772 st->codec->block_align = st->codec->channels *
2773 av_get_bits_per_sample(st->codec->codec_id) >> 3;
9450118b 2774 break;
72415b2a 2775 case AVMEDIA_TYPE_VIDEO:
9450118b
MN
2776 if(st->codec->time_base.num<=0 || st->codec->time_base.den<=0){ //FIXME audio too?
2777 av_log(s, AV_LOG_ERROR, "time base not set\n");
25de5958
AK
2778 ret = AVERROR(EINVAL);
2779 goto fail;
9450118b 2780 }
ab5a0175 2781 if((st->codec->width<=0 || st->codec->height<=0) && !(s->oformat->flags & AVFMT_NODIMENSIONS)){
9450118b 2782 av_log(s, AV_LOG_ERROR, "dimensions not set\n");
25de5958
AK
2783 ret = AVERROR(EINVAL);
2784 goto fail;
9450118b 2785 }
0354ddb7 2786 if(av_cmp_q(st->sample_aspect_ratio, st->codec->sample_aspect_ratio)){
e3cc6172
AU
2787 av_log(s, AV_LOG_ERROR, "Aspect ratio mismatch between muxer "
2788 "(%d/%d) and encoder layer (%d/%d)\n",
420df8b7
RC
2789 st->sample_aspect_ratio.num, st->sample_aspect_ratio.den,
2790 st->codec->sample_aspect_ratio.num,
2791 st->codec->sample_aspect_ratio.den);
25de5958
AK
2792 ret = AVERROR(EINVAL);
2793 goto fail;
0354ddb7 2794 }
9450118b
MN
2795 break;
2796 }
5ecfa9f5
MN
2797
2798 if(s->oformat->codec_tag){
7686ab07
MN
2799 if(st->codec->codec_tag && st->codec->codec_id == CODEC_ID_RAWVIDEO && av_codec_get_tag(s->oformat->codec_tag, st->codec->codec_id) == 0 && !validate_codec_tag(s, st)){
2800 //the current rawvideo encoding system ends up setting the wrong codec_tag for avi, we override it here
2801 st->codec->codec_tag= 0;
2802 }
5ecfa9f5 2803 if(st->codec->codec_tag){
698f4cc7 2804 if (!validate_codec_tag(s, st)) {
b603ab8d
SS
2805 char tagbuf[32];
2806 av_get_codec_tag_string(tagbuf, sizeof(tagbuf), st->codec->codec_tag);
698f4cc7 2807 av_log(s, AV_LOG_ERROR,
d2064fd4
BC
2808 "Tag %s/0x%08x incompatible with output codec id '%d'\n",
2809 tagbuf, st->codec->codec_tag, st->codec->codec_id);
25de5958
AK
2810 ret = AVERROR_INVALIDDATA;
2811 goto fail;
698f4cc7 2812 }
5ecfa9f5
MN
2813 }else
2814 st->codec->codec_tag= av_codec_get_tag(s->oformat->codec_tag, st->codec->codec_id);
2815 }
d5cce0a4
AC
2816
2817 if(s->oformat->flags & AVFMT_GLOBALHEADER &&
2818 !(st->codec->flags & CODEC_FLAG_GLOBAL_HEADER))
2819 av_log(s, AV_LOG_WARNING, "Codec for stream %d does not use global headers but container format requires global headers\n", i);
9450118b
MN
2820 }
2821
8fae2df5 2822 if (!s->priv_data && s->oformat->priv_data_size > 0) {
c6efa4b5 2823 s->priv_data = av_mallocz(s->oformat->priv_data_size);
25de5958
AK
2824 if (!s->priv_data) {
2825 ret = AVERROR(ENOMEM);
2826 goto fail;
2827 }
2828 if (s->oformat->priv_class) {
2829 *(const AVClass**)s->priv_data= s->oformat->priv_class;
2830 av_opt_set_defaults(s->priv_data);
2831 if ((ret = av_opt_set_dict(s->priv_data, &tmp)) < 0)
2832 goto fail;
2833 }
8fae2df5 2834 }
c6efa4b5 2835
ed7694d8 2836 /* set muxer identification string */
bb62d5c1 2837 if (s->nb_streams && !(s->streams[0]->codec->flags & CODEC_FLAG_BITEXACT)) {
d2d67e42 2838 av_dict_set(&s->metadata, "encoder", LIBAVFORMAT_IDENT, 0);
ed7694d8
AK
2839 }
2840
31e11451
MN
2841 if(s->oformat->write_header){
2842 ret = s->oformat->write_header(s);
2843 if (ret < 0)
25de5958 2844 goto fail;
31e11451 2845 }
1e51d801
FB
2846
2847 /* init PTS generation */
2848 for(i=0;i<s->nb_streams;i++) {
f0ff20a1 2849 int64_t den = AV_NOPTS_VALUE;
1e51d801
FB
2850 st = s->streams[i];
2851
01f4895c 2852 switch (st->codec->codec_type) {
72415b2a 2853 case AVMEDIA_TYPE_AUDIO:
f0ff20a1 2854 den = (int64_t)st->time_base.num * st->codec->sample_rate;
1e51d801 2855 break;
72415b2a 2856 case AVMEDIA_TYPE_VIDEO:
f0ff20a1 2857 den = (int64_t)st->time_base.num * st->codec->time_base.den;
1e51d801
FB
2858 break;
2859 default:
2860 break;
2861 }
f0ff20a1 2862 if (den != AV_NOPTS_VALUE) {
25de5958
AK
2863 if (den <= 0) {
2864 ret = AVERROR_INVALIDDATA;
2865 goto fail;
2866 }
d3bb7191 2867 frac_init(&st->pts, 0, 0, den);
f0ff20a1 2868 }
1e51d801 2869 }
25de5958
AK
2870
2871 if (options) {
2872 av_dict_free(options);
2873 *options = tmp;
2874 }
1e51d801 2875 return 0;
25de5958
AK
2876fail:
2877 av_dict_free(&tmp);
2878 return ret;
b9a281db
FB
2879}
2880
3c895fc0 2881//FIXME merge with compute_pkt_fields
66765b59 2882static int compute_pkt_fields2(AVFormatContext *s, AVStream *st, AVPacket *pkt){
504ee036
MN
2883 int delay = FFMAX(st->codec->has_b_frames, !!st->codec->max_b_frames);
2884 int num, den, frame_size, i;
b0c7f5a9 2885
4ad0693e 2886 av_dlog(s, "compute_pkt_fields2: pts:%"PRId64" dts:%"PRId64" cur_dts:%"PRId64" b:%d size:%d st:%d\n",
79f43a8c 2887 pkt->pts, pkt->dts, st->cur_dts, delay, pkt->size, pkt->stream_index);
115329f1 2888
e928649b 2889/* if(pkt->pts == AV_NOPTS_VALUE && pkt->dts == AV_NOPTS_VALUE)
996bbdbf 2890 return AVERROR(EINVAL);*/
115329f1 2891
e928649b 2892 /* duration field */
3c895fc0
MN
2893 if (pkt->duration == 0) {
2894 compute_frame_duration(&num, &den, st, NULL, pkt);
2895 if (den && num) {
3797c74b 2896 pkt->duration = av_rescale(1, num * (int64_t)st->time_base.den * st->codec->ticks_per_frame, den * (int64_t)st->time_base.num);
3c895fc0
MN
2897 }
2898 }
e928649b 2899
6e1aa0f3
MN
2900 if(pkt->pts == AV_NOPTS_VALUE && pkt->dts != AV_NOPTS_VALUE && delay==0)
2901 pkt->pts= pkt->dts;
2902
e928649b 2903 //XXX/FIXME this is a temporary hack until all encoders output pts
504ee036 2904 if((pkt->pts == 0 || pkt->pts == AV_NOPTS_VALUE) && pkt->dts == AV_NOPTS_VALUE && !delay){
e928649b
MN
2905 pkt->dts=
2906// pkt->pts= st->cur_dts;
2907 pkt->pts= st->pts.val;
2908 }
2909
115329f1 2910 //calculate dts from pts
cb5b96cd 2911 if(pkt->pts != AV_NOPTS_VALUE && pkt->dts == AV_NOPTS_VALUE && delay <= MAX_REORDER_DELAY){
504ee036
MN
2912 st->pts_buffer[0]= pkt->pts;
2913 for(i=1; i<delay+1 && st->pts_buffer[i] == AV_NOPTS_VALUE; i++)
663322c1 2914 st->pts_buffer[i]= pkt->pts + (i-delay-1) * pkt->duration;
504ee036 2915 for(i=0; i<delay && st->pts_buffer[i] > st->pts_buffer[i+1]; i++)
1345f4ed 2916 FFSWAP(int64_t, st->pts_buffer[i], st->pts_buffer[i+1]);
504ee036
MN
2917
2918 pkt->dts= st->pts_buffer[0];
e928649b 2919 }
115329f1 2920
5edea431 2921 if(st->cur_dts && st->cur_dts != AV_NOPTS_VALUE && st->cur_dts >= pkt->dts){
66765b59 2922 av_log(s, AV_LOG_ERROR,
440d761e 2923 "Application provided invalid, non monotonically increasing dts to muxer in stream %d: %"PRId64" >= %"PRId64"\n",
66765b59 2924 st->index, st->cur_dts, pkt->dts);
996bbdbf 2925 return AVERROR(EINVAL);
5edea431
MN
2926 }
2927 if(pkt->dts != AV_NOPTS_VALUE && pkt->pts != AV_NOPTS_VALUE && pkt->pts < pkt->dts){
440d761e 2928 av_log(s, AV_LOG_ERROR, "pts < dts in stream %d\n", st->index);
996bbdbf 2929 return AVERROR(EINVAL);
5edea431
MN
2930 }
2931
66765b59 2932// av_log(s, AV_LOG_DEBUG, "av_write_frame: pts2:%"PRId64" dts2:%"PRId64"\n", pkt->pts, pkt->dts);
e928649b
MN
2933 st->cur_dts= pkt->dts;
2934 st->pts.val= pkt->dts;
2935
1e51d801 2936 /* update pts */
01f4895c 2937 switch (st->codec->codec_type) {
72415b2a 2938 case AVMEDIA_TYPE_AUDIO:
01f4895c 2939 frame_size = get_audio_frame_size(st->codec, pkt->size);
6d8f985e 2940
a85736f2
DB
2941 /* HACK/FIXME, we skip the initial 0 size packets as they are most
2942 likely equal to the encoder delay, but it would be better if we
2943 had the real timestamps from the encoder */
e928649b 2944 if (frame_size >= 0 && (pkt->size || st->pts.num!=st->pts.den>>1 || st->pts.val)) {
d3bb7191 2945 frac_add(&st->pts, (int64_t)st->time_base.den * frame_size);
7feb950a 2946 }
1e51d801 2947 break;
72415b2a 2948 case AVMEDIA_TYPE_VIDEO:
d3bb7191 2949 frac_add(&st->pts, (int64_t)st->time_base.den * st->codec->time_base.num);
1e51d801
FB
2950 break;
2951 default:
2952 break;
2953 }
5edea431 2954 return 0;
3c895fc0
MN
2955}
2956
3c895fc0
MN
2957int av_write_frame(AVFormatContext *s, AVPacket *pkt)
2958{
f1caf01d
MS
2959 int ret;
2960
2961 if (!pkt) {
2962 if (s->oformat->flags & AVFMT_ALLOW_FLUSH)
2963 return s->oformat->write_packet(s, pkt);
2964 return 1;
2965 }
2966
2967 ret = compute_pkt_fields2(s, s->streams[pkt->stream_index], pkt);
576ae256 2968
494bbf58 2969 if(ret<0 && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))
5edea431 2970 return ret;
115329f1 2971
576ae256 2972 ret= s->oformat->write_packet(s, pkt);
1c6d2b7d
AK
2973
2974 if (ret >= 0)
2975 s->streams[pkt->stream_index]->nb_frames++;
576ae256 2976 return ret;
3c895fc0
MN
2977}
2978
ccf0071d
BC
2979void ff_interleave_add_packet(AVFormatContext *s, AVPacket *pkt,
2980 int (*compare)(AVFormatContext *, AVPacket *, AVPacket *))
2981{
2982 AVPacketList **next_point, *this_pktl;
2983
2984 this_pktl = av_mallocz(sizeof(AVPacketList));
2985 this_pktl->pkt= *pkt;
d2e63e8b
RD
2986 pkt->destruct= NULL; // do not free original but only the copy
2987 av_dup_packet(&this_pktl->pkt); // duplicate the packet if it uses non-alloced memory
ccf0071d 2988
e07b882b
MN
2989 if(s->streams[pkt->stream_index]->last_in_packet_buffer){
2990 next_point = &(s->streams[pkt->stream_index]->last_in_packet_buffer->next);
2991 }else
03555376 2992 next_point = &s->packet_buffer;
e07b882b
MN
2993
2994 if(*next_point){
2995 if(compare(s, &s->packet_buffer_end->pkt, pkt)){
2996 while(!compare(s, &(*next_point)->pkt, pkt)){
8a6c7a52
MN
2997 next_point= &(*next_point)->next;
2998 }
e07b882b 2999 goto next_non_null;
8a6c7a52
MN
3000 }else{
3001 next_point = &(s->packet_buffer_end->next);
e07b882b 3002 }
ddce56ef 3003 }
8a6c7a52 3004 assert(!*next_point);
ddce56ef 3005
8a6c7a52 3006 s->packet_buffer_end= this_pktl;
e07b882b 3007next_non_null:
ddce56ef 3008
e07b882b 3009 this_pktl->next= *next_point;
ddce56ef 3010
e07b882b
MN
3011 s->streams[pkt->stream_index]->last_in_packet_buffer=
3012 *next_point= this_pktl;
ccf0071d
BC
3013}
3014
101e1f6f 3015static int ff_interleave_compare_dts(AVFormatContext *s, AVPacket *next, AVPacket *pkt)
ccf0071d
BC
3016{
3017 AVStream *st = s->streams[ pkt ->stream_index];
3018 AVStream *st2= s->streams[ next->stream_index];
ecc29730
VS
3019 int comp = av_compare_ts(next->dts, st2->time_base, pkt->dts,
3020 st->time_base);
3021
3022 if (comp == 0)
96573c0d 3023 return pkt->stream_index < next->stream_index;
ecc29730 3024 return comp > 0;
ccf0071d
BC
3025}
3026
f21c0b4c 3027int av_interleave_packet_per_dts(AVFormatContext *s, AVPacket *out, AVPacket *pkt, int flush){
ccf0071d 3028 AVPacketList *pktl;
fe2d6fe2 3029 int stream_count=0;
ddce56ef 3030 int i;
fe2d6fe2
MN
3031
3032 if(pkt){
ccf0071d 3033 ff_interleave_add_packet(s, pkt, ff_interleave_compare_dts);
fe2d6fe2 3034 }
115329f1 3035
ddce56ef 3036 for(i=0; i < s->nb_streams; i++)
e07b882b 3037 stream_count+= !!s->streams[i]->last_in_packet_buffer;
115329f1 3038
6919e54c 3039 if(stream_count && (s->nb_streams == stream_count || flush)){
fe2d6fe2
MN
3040 pktl= s->packet_buffer;
3041 *out= pktl->pkt;
115329f1
DB
3042
3043 s->packet_buffer= pktl->next;
ddce56ef
MN
3044 if(!s->packet_buffer)
3045 s->packet_buffer_end= NULL;
3046
e07b882b
MN
3047 if(s->streams[out->stream_index]->last_in_packet_buffer == pktl)
3048 s->streams[out->stream_index]->last_in_packet_buffer= NULL;
fe2d6fe2
MN
3049 av_freep(&pktl);
3050 return 1;
3051 }else{
3052 av_init_packet(out);
3053 return 0;
3054 }
3055}
3056
3057/**
49bd8e4b 3058 * Interleave an AVPacket correctly so it can be muxed.
fe2d6fe2
MN
3059 * @param out the interleaved packet will be output here
3060 * @param in the input packet
3061 * @param flush 1 if no further packets are available as input and all
3062 * remaining packets should be output
115329f1 3063 * @return 1 if a packet was output, 0 if no packet could be output,
d9526386 3064 * < 0 if an error occurred
fe2d6fe2 3065 */
d3bb7191 3066static int interleave_packet(AVFormatContext *s, AVPacket *out, AVPacket *in, int flush){
aa2e4bb0
AK
3067 if (s->oformat->interleave_packet) {
3068 int ret = s->oformat->interleave_packet(s, out, in, flush);
3069 if (in)
3070 av_free_packet(in);
3071 return ret;
3072 } else
fe2d6fe2
MN
3073 return av_interleave_packet_per_dts(s, out, in, flush);
3074}
3075
3c895fc0 3076int av_interleaved_write_frame(AVFormatContext *s, AVPacket *pkt){
3c895fc0 3077 AVStream *st= s->streams[ pkt->stream_index];
68d875ad 3078 int ret;
3c895fc0 3079
6f824977 3080 //FIXME/XXX/HACK drop zero sized packets
72415b2a 3081 if(st->codec->codec_type == AVMEDIA_TYPE_AUDIO && pkt->size==0)
6f824977 3082 return 0;
3ce16b30 3083
70abc323
SS
3084 av_dlog(s, "av_interleaved_write_frame size:%d dts:%"PRId64" pts:%"PRId64"\n",
3085 pkt->size, pkt->dts, pkt->pts);
68d875ad
SS
3086 if((ret = compute_pkt_fields2(s, st, pkt)) < 0 && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))
3087 return ret;
115329f1 3088
4f0f9bdd 3089 if(pkt->dts == AV_NOPTS_VALUE && !(s->oformat->flags & AVFMT_NOTIMESTAMPS))
68d875ad 3090 return AVERROR(EINVAL);
3c895fc0 3091
fe2d6fe2
MN
3092 for(;;){
3093 AVPacket opkt;
d3bb7191 3094 int ret= interleave_packet(s, &opkt, pkt, 0);
fe2d6fe2
MN
3095 if(ret<=0) //FIXME cleanup needed for ret<0 ?
3096 return ret;
115329f1 3097
fe2d6fe2 3098 ret= s->oformat->write_packet(s, &opkt);
1c6d2b7d
AK
3099 if (ret >= 0)
3100 s->streams[opkt.stream_index]->nb_frames++;
115329f1 3101
fe2d6fe2
MN
3102 av_free_packet(&opkt);
3103 pkt= NULL;
115329f1 3104
3c895fc0
MN
3105 if(ret<0)
3106 return ret;
3107 }
b9a281db
FB
3108}
3109
b9a281db
FB
3110int av_write_trailer(AVFormatContext *s)
3111{
c40a3a42 3112 int ret, i;
115329f1 3113
fe2d6fe2
MN
3114 for(;;){
3115 AVPacket pkt;
d3bb7191 3116 ret= interleave_packet(s, &pkt, NULL, 1);
fe2d6fe2 3117 if(ret<0) //FIXME cleanup needed for ret<0 ?