libmpdemux/demuxer.c

   1 /*
   2  * DEMUXER v2.5
   3  *
   4  * This file is part of MPlayer.
   5  *
   6  * MPlayer is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * MPlayer is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  19  */
  20
  21 #include <stdio.h>
  22 #include <stdlib.h>
  23 #include <string.h>
  24 #include <unistd.h>
  25
  26 #include <sys/types.h>
  27 #include <sys/stat.h>
  28
  29 #include "config.h"
  30 #include "options.h"
  31 #include "talloc.h"
  32 #include "mp_msg.h"
  33 #include "m_config.h"
  34
  35 #include "libvo/fastmemcpy.h"
  36
  37 #include "stream/stream.h"
  38 #include "demuxer.h"
  39 #include "stheader.h"
  40 #include "mf.h"
  41
  42 #include "libaf/af_format.h"
  43 #include "libmpcodecs/dec_teletext.h"
  44 #include "libmpcodecs/vd_ffmpeg.h"
  45
  46 #ifdef CONFIG_FFMPEG
  47 #include "libavcodec/avcodec.h"
  48 #if MP_INPUT_BUFFER_PADDING_SIZE < FF_INPUT_BUFFER_PADDING_SIZE
  49 #error MP_INPUT_BUFFER_PADDING_SIZE is too small!
  50 #endif
  51 #endif
  52
  53 static void clear_parser(sh_common_t *sh);
  54
  55 // Demuxer list
  56 extern const struct demuxer_desc demuxer_desc_edl;
  57 extern const demuxer_desc_t demuxer_desc_rawaudio;
  58 extern const demuxer_desc_t demuxer_desc_rawvideo;
  59 extern const demuxer_desc_t demuxer_desc_tv;
  60 extern const demuxer_desc_t demuxer_desc_mf;
  61 extern const demuxer_desc_t demuxer_desc_avi;
  62 extern const demuxer_desc_t demuxer_desc_y4m;
  63 extern const demuxer_desc_t demuxer_desc_asf;
  64 extern const demuxer_desc_t demuxer_desc_real;
  65 extern const demuxer_desc_t demuxer_desc_smjpeg;
  66 extern const demuxer_desc_t demuxer_desc_matroska;
  67 extern const demuxer_desc_t demuxer_desc_realaudio;
  68 extern const demuxer_desc_t demuxer_desc_vqf;
  69 extern const demuxer_desc_t demuxer_desc_mov;
  70 extern const demuxer_desc_t demuxer_desc_vivo;
  71 extern const demuxer_desc_t demuxer_desc_fli;
  72 extern const demuxer_desc_t demuxer_desc_film;
  73 extern const demuxer_desc_t demuxer_desc_roq;
  74 extern const demuxer_desc_t demuxer_desc_gif;
  75 extern const demuxer_desc_t demuxer_desc_ogg;
  76 extern const demuxer_desc_t demuxer_desc_avs;
  77 extern const demuxer_desc_t demuxer_desc_pva;
  78 extern const demuxer_desc_t demuxer_desc_nsv;
  79 extern const demuxer_desc_t demuxer_desc_mpeg_ts;
  80 extern const demuxer_desc_t demuxer_desc_lmlm4;
  81 extern const demuxer_desc_t demuxer_desc_mpeg_ps;
  82 extern const demuxer_desc_t demuxer_desc_mpeg_pes;
  83 extern const demuxer_desc_t demuxer_desc_mpeg_es;
  84 extern const demuxer_desc_t demuxer_desc_mpeg_gxf;
  85 extern const demuxer_desc_t demuxer_desc_mpeg4_es;
  86 extern const demuxer_desc_t demuxer_desc_h264_es;
  87 extern const demuxer_desc_t demuxer_desc_rawdv;
  88 extern const demuxer_desc_t demuxer_desc_mpc;
  89 extern const demuxer_desc_t demuxer_desc_audio;
  90 extern const demuxer_desc_t demuxer_desc_xmms;
  91 extern const demuxer_desc_t demuxer_desc_mpeg_ty;
  92 extern const demuxer_desc_t demuxer_desc_rtp;
  93 extern const demuxer_desc_t demuxer_desc_rtp_nemesi;
  94 extern const demuxer_desc_t demuxer_desc_lavf;
  95 extern const demuxer_desc_t demuxer_desc_lavf_preferred;
  96 extern const demuxer_desc_t demuxer_desc_aac;
  97 extern const demuxer_desc_t demuxer_desc_nut;
  98 extern const demuxer_desc_t demuxer_desc_mng;
  99
 100 /* Please do not add any new demuxers here. If you want to implement a new
 101  * demuxer, add it to libavformat, except for wrappers around external
 102  * libraries and demuxers requiring binary support. */
 103
 104 const demuxer_desc_t *const demuxer_list[] = {
 105     &demuxer_desc_edl,
 106     &demuxer_desc_rawaudio,
 107     &demuxer_desc_rawvideo,
 108 #ifdef CONFIG_TV
 109     &demuxer_desc_tv,
 110 #endif
 111     &demuxer_desc_mf,
 112 #ifdef CONFIG_FFMPEG
 113     &demuxer_desc_lavf_preferred,
 114 #endif
 115     &demuxer_desc_avi,
 116     &demuxer_desc_y4m,
 117     &demuxer_desc_asf,
 118     &demuxer_desc_nsv,
 119     &demuxer_desc_real,
 120     &demuxer_desc_smjpeg,
 121     &demuxer_desc_matroska,
 122     &demuxer_desc_realaudio,
 123     &demuxer_desc_vqf,
 124     &demuxer_desc_mov,
 125     &demuxer_desc_vivo,
 126     &demuxer_desc_fli,
 127     &demuxer_desc_film,
 128     &demuxer_desc_roq,
 129 #ifdef CONFIG_GIF
 130     &demuxer_desc_gif,
 131 #endif
 132 #ifdef CONFIG_OGGVORBIS
 133     &demuxer_desc_ogg,
 134 #endif
 135 #ifdef CONFIG_WIN32DLL
 136     &demuxer_desc_avs,
 137 #endif
 138     &demuxer_desc_pva,
 139     &demuxer_desc_mpeg_ts,
 140     &demuxer_desc_lmlm4,
 141     &demuxer_desc_mpeg_ps,
 142     &demuxer_desc_mpeg_pes,
 143     &demuxer_desc_mpeg_es,
 144     &demuxer_desc_mpeg_gxf,
 145     &demuxer_desc_mpeg4_es,
 146     &demuxer_desc_h264_es,
 147     &demuxer_desc_audio,
 148     &demuxer_desc_mpeg_ty,
 149 #ifdef CONFIG_LIVE555
 150     &demuxer_desc_rtp,
 151 #endif
 152 #ifdef CONFIG_LIBNEMESI
 153     &demuxer_desc_rtp_nemesi,
 154 #endif
 155 #ifdef CONFIG_FFMPEG
 156     &demuxer_desc_lavf,
 157 #endif
 158 #ifdef CONFIG_MUSEPACK
 159     &demuxer_desc_mpc,
 160 #endif
 161 #ifdef CONFIG_LIBDV095
 162     &demuxer_desc_rawdv,
 163 #endif
 164     &demuxer_desc_aac,
 165 #ifdef CONFIG_LIBNUT
 166     &demuxer_desc_nut,
 167 #endif
 168 #ifdef CONFIG_XMMS
 169     &demuxer_desc_xmms,
 170 #endif
 171 #ifdef CONFIG_MNG
 172     &demuxer_desc_mng,
 173 #endif
 174     /* Please do not add any new demuxers here. If you want to implement a new
 175      * demuxer, add it to libavformat, except for wrappers around external
 176      * libraries and demuxers requiring binary support. */
 177     NULL
 178 };
 179
 180 struct demux_packet *new_demux_packet(size_t len)
 181 {
 182     if (len > 1000000000) {
 183         mp_msg(MSGT_DEMUXER, MSGL_FATAL, "Attempt to allocate demux packet "
 184                "over 1 GB!\n");
 185         abort();
 186     }
 187     struct demux_packet *dp = malloc(sizeof(struct demux_packet));
 188     dp->len = len;
 189     dp->next = NULL;
 190     dp->pts = MP_NOPTS_VALUE;
 191     dp->duration = -1;
 192     dp->stream_pts = MP_NOPTS_VALUE;
 193     dp->pos = 0;
 194     dp->flags = 0;
 195     dp->refcount = 1;
 196     dp->master = NULL;
 197     dp->buffer = NULL;
 198     dp->buffer = malloc(len + MP_INPUT_BUFFER_PADDING_SIZE);
 199     if (!dp->buffer) {
 200         mp_msg(MSGT_DEMUXER, MSGL_FATAL, "Memory allocation failure!\n");
 201         abort();
 202     }
 203     memset(dp->buffer + len, 0, 8);
 204     return dp;
 205 }
 206
 207 void resize_demux_packet(struct demux_packet *dp, size_t len)
 208 {
 209     if (len > 1000000000) {
 210         mp_msg(MSGT_DEMUXER, MSGL_FATAL, "Attempt to realloc demux packet "
 211                "over 1 GB!\n");
 212         abort();
 213     }
 214     dp->buffer = realloc(dp->buffer, len + MP_INPUT_BUFFER_PADDING_SIZE);
 215     if (!dp->buffer) {
 216         mp_msg(MSGT_DEMUXER, MSGL_FATAL, "Memory allocation failure!\n");
 217         abort();
 218     }
 219     memset(dp->buffer + len, 0, 8);
 220     dp->len = len;
 221 }
 222
 223 struct demux_packet *clone_demux_packet(struct demux_packet *pack)
 224 {
 225     struct demux_packet *dp = malloc(sizeof(struct demux_packet));
 226     while (pack->master)
 227         pack = pack->master;  // find the master
 228     memcpy(dp, pack, sizeof(struct demux_packet));
 229     dp->next = NULL;
 230     dp->refcount = 0;
 231     dp->master = pack;
 232     pack->refcount++;
 233     return dp;
 234 }
 235
 236 void free_demux_packet(struct demux_packet *dp)
 237 {
 238     if (dp->master == NULL) {  //dp is a master packet
 239         dp->refcount--;
 240         if (dp->refcount == 0) {
 241             free(dp->buffer);
 242             free(dp);
 243         }
 244         return;
 245     }
 246     // dp is a clone:
 247     free_demux_packet(dp->master);
 248     free(dp);
 249 }
 250
 251 static void free_demuxer_stream(struct demux_stream *ds)
 252 {
 253     ds_free_packs(ds);
 254     free(ds);
 255 }
 256
 257 static struct demux_stream *new_demuxer_stream(struct demuxer *demuxer, int id)
 258 {
 259     demux_stream_t *ds = malloc(sizeof(demux_stream_t));
 260     *ds = (demux_stream_t){
 261         .id = id,
 262         .demuxer = demuxer,
 263         .asf_seq = -1,
 264     };
 265     return ds;
 266 }
 267
 268
 269 /**
 270  * Get demuxer description structure for a given demuxer type
 271  *
 272  * @param file_format    type of the demuxer
 273  * @return               structure for the demuxer, NULL if not found
 274  */
 275 static const demuxer_desc_t *get_demuxer_desc_from_type(int file_format)
 276 {
 277     int i;
 278
 279     for (i = 0; demuxer_list[i]; i++)
 280         if (file_format == demuxer_list[i]->type)
 281             return demuxer_list[i];
 282
 283     return NULL;
 284 }
 285
 286
 287 demuxer_t *new_demuxer(struct MPOpts *opts, stream_t *stream, int type,
 288                        int a_id, int v_id, int s_id, char *filename)
 289 {
 290     struct demuxer *d = talloc_zero(NULL, struct demuxer);
 291     d->stream = stream;
 292     d->stream_pts = MP_NOPTS_VALUE;
 293     d->reference_clock = MP_NOPTS_VALUE;
 294     d->movi_start = stream->start_pos;
 295     d->movi_end = stream->end_pos;
 296     d->seekable = 1;
 297     d->synced = 0;
 298     d->filepos = -1;
 299     d->audio = new_demuxer_stream(d, a_id);
 300     d->video = new_demuxer_stream(d, v_id);
 301     d->sub = new_demuxer_stream(d, s_id);
 302     d->type = type;
 303     d->opts = opts;
 304     if (type)
 305         if (!(d->desc = get_demuxer_desc_from_type(type)))
 306             mp_msg(MSGT_DEMUXER, MSGL_ERR,
 307                    "BUG! Invalid demuxer type in new_demuxer(), "
 308                    "big troubles ahead.");
 309     if (filename) // Filename hack for avs_check_file
 310         d->filename = strdup(filename);
 311     stream_seek(stream, stream->start_pos);
 312     return d;
 313 }
 314
 315 const char *sh_sub_type2str(int type)
 316 {
 317     switch (type) {
 318     case 't': return "text";
 319     case 'm': return "movtext";
 320     case 'a': return "ass";
 321     case 'v': return "vobsub";
 322     case 'x': return "xsub";
 323     case 'b': return "dvb";
 324     case 'd': return "dvb-teletext";
 325     case 'p': return "hdmv pgs";
 326     }
 327     return "unknown";
 328 }
 329
 330 sh_sub_t *new_sh_sub_sid(demuxer_t *demuxer, int id, int sid)
 331 {
 332     if (id > MAX_S_STREAMS - 1 || id < 0) {
 333         mp_msg(MSGT_DEMUXER, MSGL_WARN,
 334                "Requested sub stream id overflow (%d > %d)\n", id,
 335                MAX_S_STREAMS);
 336         return NULL;
 337     }
 338     if (demuxer->s_streams[id])
 339         mp_msg(MSGT_DEMUXER, MSGL_WARN, "Sub stream %i redefined\n", id);
 340     else {
 341         struct sh_sub *sh = talloc_zero(demuxer, struct sh_sub);
 342         demuxer->s_streams[id] = sh;
 343         sh->sid = sid;
 344         sh->opts = demuxer->opts;
 345         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_SUBTITLE_ID=%d\n", sid);
 346     }
 347     return demuxer->s_streams[id];
 348 }
 349
 350 struct sh_sub *new_sh_sub_sid_lang(struct demuxer *demuxer, int id, int sid,
 351                                    const char *lang)
 352 {
 353     struct sh_sub *sh = new_sh_sub_sid(demuxer, id, sid);
 354     if (lang && lang[0] && strcmp(lang, "und")) {
 355         sh->lang = talloc_strdup(sh, lang);
 356         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_SID_%d_LANG=%s\n", sid, lang);
 357     }
 358     return sh;
 359 }
 360
 361 static void free_sh_sub(sh_sub_t *sh)
 362 {
 363     mp_msg(MSGT_DEMUXER, MSGL_DBG2, "DEMUXER: freeing sh_sub at %p\n", sh);
 364     free(sh->extradata);
 365 #ifdef CONFIG_FFMPEG
 366     clear_parser((sh_common_t *)sh);
 367 #endif
 368     talloc_free(sh);
 369 }
 370
 371 sh_audio_t *new_sh_audio_aid(demuxer_t *demuxer, int id, int aid)
 372 {
 373     if (id > MAX_A_STREAMS - 1 || id < 0) {
 374         mp_msg(MSGT_DEMUXER, MSGL_WARN,
 375                "Requested audio stream id overflow (%d > %d)\n", id,
 376                MAX_A_STREAMS);
 377         return NULL;
 378     }
 379     if (demuxer->a_streams[id]) {
 380         mp_tmsg(MSGT_DEMUXER, MSGL_WARN, "WARNING: Audio stream header %d redefined.\n", id);
 381     } else {
 382         mp_tmsg(MSGT_DEMUXER, MSGL_V, "==> Found audio stream: %d\n", id);
 383         struct sh_audio *sh = talloc_zero(demuxer, struct sh_audio);
 384         demuxer->a_streams[id] = sh;
 385         sh->aid = aid;
 386         sh->ds = demuxer->audio;
 387         // set some defaults
 388         sh->samplesize = 2;
 389         sh->sample_format = AF_FORMAT_S16_NE;
 390         sh->opts = demuxer->opts;
 391         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_AUDIO_ID=%d\n", aid);
 392     }
 393     return demuxer->a_streams[id];
 394 }
 395
 396 void free_sh_audio(demuxer_t *demuxer, int id)
 397 {
 398     sh_audio_t *sh = demuxer->a_streams[id];
 399     demuxer->a_streams[id] = NULL;
 400     mp_msg(MSGT_DEMUXER, MSGL_DBG2, "DEMUXER: freeing sh_audio at %p\n", sh);
 401     free(sh->wf);
 402     free(sh->codecdata);
 403 #ifdef CONFIG_FFMPEG
 404     clear_parser((sh_common_t *)sh);
 405 #endif
 406     talloc_free(sh);
 407 }
 408
 409 sh_video_t *new_sh_video_vid(demuxer_t *demuxer, int id, int vid)
 410 {
 411     if (id > MAX_V_STREAMS - 1 || id < 0) {
 412         mp_msg(MSGT_DEMUXER, MSGL_WARN,
 413                "Requested video stream id overflow (%d > %d)\n", id,
 414                MAX_V_STREAMS);
 415         return NULL;
 416     }
 417     if (demuxer->v_streams[id])
 418         mp_tmsg(MSGT_DEMUXER, MSGL_WARN, "WARNING: Video stream header %d redefined.\n", id);
 419     else {
 420         mp_tmsg(MSGT_DEMUXER, MSGL_V, "==> Found video stream: %d\n", id);
 421         struct sh_video *sh = talloc_zero(demuxer, struct sh_video);
 422         demuxer->v_streams[id] = sh;
 423         sh->vid = vid;
 424         sh->ds = demuxer->video;
 425         sh->opts = demuxer->opts;
 426         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_VIDEO_ID=%d\n", vid);
 427     }
 428     return demuxer->v_streams[id];
 429 }
 430
 431 void free_sh_video(sh_video_t *sh)
 432 {
 433     mp_msg(MSGT_DEMUXER, MSGL_DBG2, "DEMUXER: freeing sh_video at %p\n", sh);
 434     free(sh->bih);
 435 #ifdef CONFIG_FFMPEG
 436     clear_parser((sh_common_t *)sh);
 437 #endif
 438     talloc_free(sh);
 439 }
 440
 441 void free_demuxer(demuxer_t *demuxer)
 442 {
 443     int i;
 444     mp_msg(MSGT_DEMUXER, MSGL_DBG2, "DEMUXER: freeing %s demuxer at %p\n",
 445            demuxer->desc->shortdesc, demuxer);
 446     if (demuxer->desc->close)
 447         demuxer->desc->close(demuxer);
 448     // Very ugly hack to make it behave like old implementation
 449     if (demuxer->desc->type == DEMUXER_TYPE_DEMUXERS)
 450         goto skip_streamfree;
 451     // free streams:
 452     for (i = 0; i < MAX_A_STREAMS; i++)
 453         if (demuxer->a_streams[i])
 454             free_sh_audio(demuxer, i);
 455     for (i = 0; i < MAX_V_STREAMS; i++)
 456         if (demuxer->v_streams[i])
 457             free_sh_video(demuxer->v_streams[i]);
 458     for (i = 0; i < MAX_S_STREAMS; i++)
 459         if (demuxer->s_streams[i])
 460             free_sh_sub(demuxer->s_streams[i]);
 461     // free demuxers:
 462     free_demuxer_stream(demuxer->audio);
 463     free_demuxer_stream(demuxer->video);
 464     free_demuxer_stream(demuxer->sub);
 465  skip_streamfree:
 466     free(demuxer->filename);
 467     if (demuxer->teletext)
 468         teletext_control(demuxer->teletext, TV_VBI_CONTROL_STOP, NULL);
 469     talloc_free(demuxer);
 470 }
 471
 472
 473 void ds_add_packet(demux_stream_t *ds, demux_packet_t *dp)
 474 {
 475     // append packet to DS stream:
 476     ++ds->packs;
 477     ds->bytes += dp->len;
 478     if (ds->last) {
 479         // next packet in stream
 480         ds->last->next = dp;
 481         ds->last = dp;
 482     } else {
 483         // first packet in stream
 484         ds->first = ds->last = dp;
 485     }
 486     mp_dbg(MSGT_DEMUXER, MSGL_DBG2,
 487            "DEMUX: Append packet to %s, len=%d  pts=%5.3f  pos=%u  [packs: A=%d V=%d]\n",
 488            (ds == ds->demuxer->audio) ? "d_audio" : "d_video", dp->len,
 489            dp->pts, (unsigned int) dp->pos, ds->demuxer->audio->packs,
 490            ds->demuxer->video->packs);
 491 }
 492
 493 #ifdef CONFIG_FFMPEG
 494 static void allocate_parser(AVCodecContext **avctx, AVCodecParserContext **parser, unsigned format)
 495 {
 496     enum CodecID codec_id = CODEC_ID_NONE;
 497
 498     init_avcodec();
 499
 500     switch (format) {
 501 #if LIBAVCODEC_VERSION_INT >= AV_VERSION_INT(52, 94, 0)
 502     case MKTAG('M', 'P', '4', 'L'):
 503         codec_id = CODEC_ID_AAC_LATM;
 504         break;
 505 #endif
 506     case 0x2000:
 507     case 0x332D6361:
 508     case 0x332D4341:
 509     case 0x20736D:
 510     case MKTAG('s', 'a', 'c', '3'):
 511         codec_id = CODEC_ID_AC3;
 512         break;
 513     case MKTAG('d', 'n', 'e', 't'):
 514         // DNET/byte-swapped AC-3 - there is no parser for that yet
 515         //codec_id = CODEC_ID_DNET;
 516         break;
 517     case MKTAG('E', 'A', 'C', '3'):
 518         codec_id = CODEC_ID_EAC3;
 519         break;
 520     case 0x2001:
 521     case 0x86:
 522         codec_id = CODEC_ID_DTS;
 523         break;
 524     case MKTAG('f', 'L', 'a', 'C'):
 525         codec_id = CODEC_ID_FLAC;
 526         break;
 527     case MKTAG('M', 'L', 'P', ' '):
 528         codec_id = CODEC_ID_MLP;
 529         break;
 530     case 0x55:
 531     case 0x5500736d:
 532     case 0x55005354:
 533     case MKTAG('.', 'm', 'p', '3'):
 534     case MKTAG('M', 'P', '3', ' '):
 535     case MKTAG('L', 'A', 'M', 'E'):
 536         codec_id = CODEC_ID_MP3;
 537         break;
 538     case 0x50:
 539     case 0x5000736d:
 540     case MKTAG('.', 'm', 'p', '2'):
 541     case MKTAG('.', 'm', 'p', '1'):
 542         codec_id = CODEC_ID_MP2;
 543         break;
 544     case MKTAG('T', 'R', 'H', 'D'):
 545         codec_id = CODEC_ID_TRUEHD;
 546         break;
 547     }
 548     if (codec_id != CODEC_ID_NONE) {
 549         *avctx = avcodec_alloc_context();
 550         if (!*avctx)
 551             return;
 552         *parser = av_parser_init(codec_id);
 553         if (!*parser)
 554             av_freep(avctx);
 555     }
 556 }
 557
 558 static void get_parser(sh_common_t *sh, AVCodecContext **avctx, AVCodecParserContext **parser)
 559 {
 560     *avctx  = NULL;
 561     *parser = NULL;
 562
 563     if (!sh || !sh->needs_parsing)
 564         return;
 565
 566     *avctx  = sh->avctx;
 567     *parser = sh->parser;
 568     if (*parser)
 569         return;
 570
 571     allocate_parser(avctx, parser, sh->format);
 572     sh->avctx  = *avctx;
 573     sh->parser = *parser;
 574 }
 575
 576 int ds_parse(demux_stream_t *ds, uint8_t **buffer, int *len, double pts, off_t pos)
 577 {
 578     AVCodecContext *avctx;
 579     AVCodecParserContext *parser;
 580     get_parser(ds->sh, &avctx, &parser);
 581     if (!parser)
 582         return *len;
 583     return av_parser_parse2(parser, avctx, buffer, len, *buffer, *len, pts, pts, pos);
 584 }
 585
 586 static void clear_parser(sh_common_t *sh)
 587 {
 588     av_parser_close(sh->parser);
 589     sh->parser = NULL;
 590     av_freep(&sh->avctx);
 591 }
 592
 593 void ds_clear_parser(demux_stream_t *ds)
 594 {
 595     if (!ds->sh)
 596         return;
 597     clear_parser(ds->sh);
 598 }
 599 #endif
 600
 601 void ds_read_packet(demux_stream_t *ds, stream_t *stream, int len,
 602                     double pts, off_t pos, int flags)
 603 {
 604     demux_packet_t *dp = new_demux_packet(len);
 605     len = stream_read(stream, dp->buffer, len);
 606     resize_demux_packet(dp, len);
 607     dp->pts = pts;
 608     dp->pos = pos;
 609     dp->flags = flags;
 610     // append packet to DS stream:
 611     ds_add_packet(ds, dp);
 612 }
 613
 614 // return value:
 615 //     0 = EOF or no stream found or invalid type
 616 //     1 = successfully read a packet
 617
 618 int demux_fill_buffer(demuxer_t *demux, demux_stream_t *ds)
 619 {
 620     // Note: parameter 'ds' can be NULL!
 621     return demux->desc->fill_buffer(demux, ds);
 622 }
 623
 624 // return value:
 625 //     0 = EOF
 626 //     1 = successful
 627 int ds_fill_buffer(demux_stream_t *ds)
 628 {
 629     demuxer_t *demux = ds->demuxer;
 630     if (ds->current)
 631         free_demux_packet(ds->current);
 632     ds->current = NULL;
 633     mp_dbg(MSGT_DEMUXER, MSGL_DBG3, "ds_fill_buffer (%s) called\n",
 634            ds == demux->audio ? "d_audio" : ds == demux->video ? "d_video" :
 635            ds == demux->sub   ? "d_sub"   : "unknown");
 636     while (1) {
 637         if (ds->packs) {
 638             demux_packet_t *p = ds->first;
 639             // copy useful data:
 640             ds->buffer = p->buffer;
 641             ds->buffer_pos = 0;
 642             ds->buffer_size = p->len;
 643             ds->pos = p->pos;
 644             ds->dpos += p->len; // !!!
 645             ++ds->pack_no;
 646             if (p->pts != MP_NOPTS_VALUE) {
 647                 ds->pts = p->pts;
 648                 ds->pts_bytes = 0;
 649             }
 650             ds->pts_bytes += p->len;    // !!!
 651             if (p->stream_pts != MP_NOPTS_VALUE)
 652                 demux->stream_pts = p->stream_pts;
 653             ds->flags = p->flags;
 654             // unlink packet:
 655             ds->bytes -= p->len;
 656             ds->current = p;
 657             ds->first = p->next;
 658             if (!ds->first)
 659                 ds->last = NULL;
 660             --ds->packs;
 661             /* The code below can set ds->eof to 1 when another stream runs
 662              * out of buffer space. That makes sense because in that situation
 663              * the calling code should not count on being able to demux more
 664              * packets from this stream.
 665              * If however the situation improves and we're called again
 666              * despite the eof flag then it's better to clear it to avoid
 667              * weird behavior. */
 668             ds->eof = 0;
 669             return 1;
 670         }
 671
 672 #define MaybeNI _("Maybe you are playing a non-interleaved stream/file or the codec failed?\n" \
 673                 "For AVI files, try to force non-interleaved mode with the -ni option.\n")
 674
 675         if (demux->audio->packs >= MAX_PACKS
 676             || demux->audio->bytes >= MAX_PACK_BYTES) {
 677             mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "\nToo many audio packets in the buffer: (%d in %d bytes).\n",
 678                    demux->audio->packs, demux->audio->bytes);
 679             mp_tmsg(MSGT_DEMUXER, MSGL_HINT, MaybeNI);
 680             break;
 681         }
 682         if (demux->video->packs >= MAX_PACKS
 683             || demux->video->bytes >= MAX_PACK_BYTES) {
 684             mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "\nToo many video packets in the buffer: (%d in %d bytes).\n",
 685                    demux->video->packs, demux->video->bytes);
 686             mp_tmsg(MSGT_DEMUXER, MSGL_HINT, MaybeNI);
 687             break;
 688         }
 689         if (!demux_fill_buffer(demux, ds)) {
 690             mp_dbg(MSGT_DEMUXER, MSGL_DBG2,
 691                    "ds_fill_buffer()->demux_fill_buffer() failed\n");
 692             break; // EOF
 693         }
 694     }
 695     ds->buffer_pos = ds->buffer_size = 0;
 696     ds->buffer = NULL;
 697     mp_msg(MSGT_DEMUXER, MSGL_V,
 698            "ds_fill_buffer: EOF reached (stream: %s)  \n",
 699            ds == demux->audio ? "audio" : "video");
 700     ds->eof = 1;
 701     return 0;
 702 }
 703
 704 int demux_read_data(demux_stream_t *ds, unsigned char *mem, int len)
 705 {
 706     int x;
 707     int bytes = 0;
 708     while (len > 0) {
 709         x = ds->buffer_size - ds->buffer_pos;
 710         if (x == 0) {
 711             if (!ds_fill_buffer(ds))
 712                 return bytes;
 713         } else {
 714             if (x > len)
 715                 x = len;
 716             if (mem)
 717                 fast_memcpy(mem + bytes, &ds->buffer[ds->buffer_pos], x);
 718             bytes += x;
 719             len -= x;
 720             ds->buffer_pos += x;
 721         }
 722     }
 723     return bytes;
 724 }
 725
 726 /**
 727  * \brief read data until the given 3-byte pattern is encountered, up to maxlen
 728  * \param mem memory to read data into, may be NULL to discard data
 729  * \param maxlen maximum number of bytes to read
 730  * \param read number of bytes actually read
 731  * \param pattern pattern to search for (lowest 8 bits are ignored)
 732  * \return whether pattern was found
 733  */
 734 int demux_pattern_3(demux_stream_t *ds, unsigned char *mem, int maxlen,
 735                     int *read, uint32_t pattern)
 736 {
 737     register uint32_t head = 0xffffff00;
 738     register uint32_t pat = pattern & 0xffffff00;
 739     int total_len = 0;
 740     do {
 741         register unsigned char *ds_buf = &ds->buffer[ds->buffer_size];
 742         int len = ds->buffer_size - ds->buffer_pos;
 743         register long pos = -len;
 744         if (unlikely(pos >= 0)) { // buffer is empty
 745             ds_fill_buffer(ds);
 746             continue;
 747         }
 748         do {
 749             head |= ds_buf[pos];
 750             head <<= 8;
 751         } while (++pos && head != pat);
 752         len += pos;
 753         if (total_len + len > maxlen)
 754             len = maxlen - total_len;
 755         len = demux_read_data(ds, mem ? &mem[total_len] : NULL, len);
 756         total_len += len;
 757     } while ((head != pat || total_len < 3) && total_len < maxlen && !ds->eof);
 758     if (read)
 759         *read = total_len;
 760     return total_len >= 3 && head == pat;
 761 }
 762
 763 void ds_free_packs(demux_stream_t *ds)
 764 {
 765     demux_packet_t *dp = ds->first;
 766     while (dp) {
 767         demux_packet_t *dn = dp->next;
 768         free_demux_packet(dp);
 769         dp = dn;
 770     }
 771     if (ds->asf_packet) {
 772         // free unfinished .asf fragments:
 773         free(ds->asf_packet->buffer);
 774         free(ds->asf_packet);
 775         ds->asf_packet = NULL;
 776     }
 777     ds->first = ds->last = NULL;
 778     ds->packs = 0; // !!!!!
 779     ds->bytes = 0;
 780     if (ds->current)
 781         free_demux_packet(ds->current);
 782     ds->current = NULL;
 783     ds->buffer = NULL;
 784     ds->buffer_pos = ds->buffer_size;
 785     ds->pts = 0;
 786     ds->pts_bytes = 0;
 787 }
 788
 789 int ds_get_packet(demux_stream_t *ds, unsigned char **start)
 790 {
 791     int len;
 792     if (ds->buffer_pos >= ds->buffer_size) {
 793         if (!ds_fill_buffer(ds)) {
 794             // EOF
 795             *start = NULL;
 796             return -1;
 797         }
 798     }
 799     len = ds->buffer_size - ds->buffer_pos;
 800     *start = &ds->buffer[ds->buffer_pos];
 801     ds->buffer_pos += len;
 802     return len;
 803 }
 804
 805 int ds_get_packet_pts(demux_stream_t *ds, unsigned char **start, double *pts)
 806 {
 807     int len;
 808     *pts = MP_NOPTS_VALUE;
 809     len = ds_get_packet(ds, start);
 810     if (len < 0)
 811         return len;
 812     // Return pts unless this read starts from the middle of a packet
 813     if (len == ds->buffer_pos)
 814         *pts = ds->current->pts;
 815     return len;
 816 }
 817
 818 int ds_get_packet_sub(demux_stream_t *ds, unsigned char **start)
 819 {
 820     int len;
 821     if (ds->buffer_pos >= ds->buffer_size) {
 822         *start = NULL;
 823         if (!ds->packs)
 824             return -1;  // no sub
 825         if (!ds_fill_buffer(ds))
 826             return -1;  // EOF
 827     }
 828     len = ds->buffer_size - ds->buffer_pos;
 829     *start = &ds->buffer[ds->buffer_pos];
 830     ds->buffer_pos += len;
 831     return len;
 832 }
 833
 834 double ds_get_next_pts(demux_stream_t *ds)
 835 {
 836     demuxer_t *demux = ds->demuxer;
 837     // if we have not read from the "current" packet, consider it
 838     // as the next, otherwise we never get the pts for the first packet.
 839     while (!ds->first && (!ds->current || ds->buffer_pos)) {
 840         if (demux->audio->packs >= MAX_PACKS
 841             || demux->audio->bytes >= MAX_PACK_BYTES) {
 842             mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "\nToo many audio packets in the buffer: (%d in %d bytes).\n",
 843                    demux->audio->packs, demux->audio->bytes);
 844             mp_tmsg(MSGT_DEMUXER, MSGL_HINT, MaybeNI);
 845             return MP_NOPTS_VALUE;
 846         }
 847         if (demux->video->packs >= MAX_PACKS
 848             || demux->video->bytes >= MAX_PACK_BYTES) {
 849             mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "\nToo many video packets in the buffer: (%d in %d bytes).\n",
 850                    demux->video->packs, demux->video->bytes);
 851             mp_tmsg(MSGT_DEMUXER, MSGL_HINT, MaybeNI);
 852             return MP_NOPTS_VALUE;
 853         }
 854         if (!demux_fill_buffer(demux, ds))
 855             return MP_NOPTS_VALUE;
 856     }
 857     // take pts from "current" if we never read from it.
 858     if (ds->current && !ds->buffer_pos)
 859         return ds->current->pts;
 860     return ds->first->pts;
 861 }
 862
 863 // ====================================================================
 864
 865 void demuxer_help(void)
 866 {
 867     int i;
 868
 869     mp_msg(MSGT_DEMUXER, MSGL_INFO, "Available demuxers:\n");
 870     mp_msg(MSGT_DEMUXER, MSGL_INFO, " demuxer:   info:  (comment)\n");
 871     mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_DEMUXERS\n");
 872     for (i = 0; demuxer_list[i]; i++) {
 873         if (demuxer_list[i]->type >= DEMUXER_TYPE_END)  // internal type
 874             continue;
 875         if (demuxer_list[i]->comment && strlen(demuxer_list[i]->comment))
 876             mp_msg(MSGT_DEMUXER, MSGL_INFO, "%10s  %s (%s)\n",
 877                    demuxer_list[i]->name, demuxer_list[i]->info,
 878                    demuxer_list[i]->comment);
 879         else
 880             mp_msg(MSGT_DEMUXER, MSGL_INFO, "%10s  %s\n",
 881                    demuxer_list[i]->name, demuxer_list[i]->info);
 882     }
 883 }
 884
 885
 886 /**
 887  * Get demuxer type for a given demuxer name
 888  *
 889  * @param demuxer_name    string with demuxer name of demuxer number
 890  * @param force           will be set if demuxer should be forced.
 891  *                        May be NULL.
 892  * @return                DEMUXER_TYPE_xxx, -1 if error or not found
 893  */
 894 static int get_demuxer_type_from_name(char *demuxer_name, int *force)
 895 {
 896     if (!demuxer_name || !demuxer_name[0])
 897         return DEMUXER_TYPE_UNKNOWN;
 898     if (force)
 899         *force = demuxer_name[0] == '+';
 900     if (demuxer_name[0] == '+')
 901         demuxer_name = &demuxer_name[1];
 902     for (int i = 0; demuxer_list[i]; i++) {
 903         if (demuxer_list[i]->type >= DEMUXER_TYPE_END)
 904             // Can't select special demuxers from commandline
 905             continue;
 906         if (strcmp(demuxer_name, demuxer_list[i]->name) == 0)
 907             return demuxer_list[i]->type;
 908     }
 909
 910     return -1;
 911 }
 912
 913 static struct demuxer *open_given_type(struct MPOpts *opts,
 914                                        const struct demuxer_desc *desc,
 915                                        struct stream *stream, bool force,
 916                                        int audio_id, int video_id, int sub_id,
 917                                        char *filename)
 918 {
 919     struct demuxer *demuxer;
 920     int fformat;
 921     demuxer = new_demuxer(opts, stream, desc->type, audio_id,
 922                           video_id, sub_id, filename);
 923     if (desc->check_file)
 924         fformat = desc->check_file(demuxer);
 925     else
 926         fformat = desc->type;
 927     if (force)
 928         fformat = desc->type;
 929     if (fformat == 0)
 930         goto fail;
 931     if (fformat == desc->type) {
 932         if (demuxer->filetype)
 933             mp_tmsg(MSGT_DEMUXER, MSGL_INFO, "Detected file format: %s (%s)\n",
 934                     demuxer->filetype, desc->shortdesc);
 935         else
 936             mp_tmsg(MSGT_DEMUXER, MSGL_INFO, "Detected file format: %s\n",
 937                     desc->shortdesc);
 938         if (demuxer->desc->open) {
 939             struct demuxer *demux2 = demuxer->desc->open(demuxer);
 940             if (!demux2) {
 941                 mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "Opening as detected format "
 942                         "\"%s\" failed.\n", desc->shortdesc);
 943                 goto fail;
 944             }
 945             /* At least demux_mov can return a demux_demuxers instance
 946              * from open() instead of the original fed in. */
 947             demuxer = demux2;
 948         }
 949         demuxer->file_format = fformat;
 950         return demuxer;
 951     } else {
 952         // demux_mov can return playlist instead of mov
 953         if (fformat == DEMUXER_TYPE_PLAYLIST)
 954             return demuxer; // handled in mplayer.c
 955         /* Internal MPEG PS demuxer check can return other MPEG subtypes
 956          * which don't have their own checks; recurse to try opening as
 957          * the returned type instead. */
 958         free_demuxer(demuxer);
 959         desc = get_demuxer_desc_from_type(fformat);
 960         if (!desc) {
 961             mp_msg(MSGT_DEMUXER, MSGL_ERR,
 962                    "BUG: recursion to nonexistent file format\n");
 963             return NULL;
 964         }
 965         return open_given_type(opts, desc, stream, false, audio_id,
 966                                video_id, sub_id, filename);
 967     }
 968  fail:
 969     free_demuxer(demuxer);
 970     return NULL;
 971 }
 972
 973 static struct demuxer *demux_open_stream(struct MPOpts *opts,
 974                                          struct stream *stream,
 975                                          int file_format, bool force,
 976                                          int audio_id, int video_id, int sub_id,
 977                                          char *filename)
 978 {
 979     struct demuxer *demuxer = NULL;
 980     const struct demuxer_desc *desc;
 981
 982     // If somebody requested a demuxer check it
 983     if (file_format) {
 984         desc = get_demuxer_desc_from_type(file_format);
 985         if (!desc)
 986             // should only happen with obsolete -demuxer 99 numeric format
 987             return NULL;
 988         demuxer = open_given_type(opts, desc, stream, force, audio_id,
 989                                   video_id, sub_id, filename);
 990         if (demuxer)
 991             goto dmx_open;
 992         return NULL;
 993     }
 994
 995     // Test demuxers with safe file checks
 996     for (int i = 0; (desc = demuxer_list[i]); i++) {
 997         if (desc->safe_check) {
 998             demuxer = open_given_type(opts, desc, stream, false, audio_id,
 999                                       video_id, sub_id, filename);
1000             if (demuxer)
1001                 goto dmx_open;
1002         }
1003     }
1004
1005     // Ok. We're over the stable detectable fileformats, the next ones are
1006     // a bit fuzzy. So by default (extension_parsing==1) try extension-based
1007     // detection first:
1008     if (filename && opts->extension_parsing == 1) {
1009         desc = get_demuxer_desc_from_type(demuxer_type_by_filename(filename));
1010         if (desc)
1011             demuxer = open_given_type(opts, desc, stream, false, audio_id,
1012                                       video_id, sub_id, filename);
1013         if (demuxer)
1014             goto dmx_open;
1015     }
1016
1017     // Finally try detection for demuxers with unsafe checks
1018     for (int i = 0; (desc = demuxer_list[i]); i++) {
1019         if (!desc->safe_check && desc->check_file) {
1020             demuxer = open_given_type(opts, desc, stream, false, audio_id,
1021                                       video_id, sub_id, filename);
1022             if (demuxer)
1023                 goto dmx_open;
1024         }
1025     }
1026
1027     return NULL;
1028
1029  dmx_open:
1030
1031     if (demuxer->type == DEMUXER_TYPE_PLAYLIST)
1032         return demuxer;
1033
1034     struct sh_video *sh_video = demuxer->video->sh;
1035     if (sh_video && sh_video->bih) {
1036         int biComp = le2me_32(sh_video->bih->biCompression);
1037         mp_msg(MSGT_DEMUX, MSGL_INFO,
1038                "VIDEO:  [%.4s]  %dx%d  %dbpp  %5.3f fps  %5.1f kbps (%4.1f kbyte/s)\n",
1039                (char *) &biComp, sh_video->bih->biWidth,
1040                sh_video->bih->biHeight, sh_video->bih->biBitCount,
1041                sh_video->fps, sh_video->i_bps * 0.008f,
1042                sh_video->i_bps / 1024.0f);
1043     }
1044     return demuxer;
1045 }
1046
1047 demuxer_t *demux_open(struct MPOpts *opts, stream_t *vs, int file_format,
1048                       int audio_id, int video_id, int dvdsub_id,
1049                       char *filename)
1050 {
1051     stream_t *as = NULL, *ss = NULL;
1052     demuxer_t *vd, *ad = NULL, *sd = NULL;
1053     demuxer_t *res;
1054     int afmt = DEMUXER_TYPE_UNKNOWN, sfmt = DEMUXER_TYPE_UNKNOWN;
1055     int demuxer_type;
1056     int audio_demuxer_type = 0, sub_demuxer_type = 0;
1057     int demuxer_force = 0, audio_demuxer_force = 0, sub_demuxer_force = 0;
1058
1059     if ((demuxer_type =
1060          get_demuxer_type_from_name(opts->demuxer_name, &demuxer_force)) < 0) {
1061         mp_msg(MSGT_DEMUXER, MSGL_ERR, "-demuxer %s does not exist.\n",
1062                opts->demuxer_name);
1063         return NULL;
1064     }
1065     if ((audio_demuxer_type =
1066          get_demuxer_type_from_name(opts->audio_demuxer_name,
1067                                     &audio_demuxer_force)) < 0) {
1068         mp_msg(MSGT_DEMUXER, MSGL_ERR, "-audio-demuxer %s does not exist.\n",
1069                opts->audio_demuxer_name);
1070         if (opts->audio_stream)
1071             return NULL;
1072     }
1073     if ((sub_demuxer_type =
1074          get_demuxer_type_from_name(opts->sub_demuxer_name,
1075                                     &sub_demuxer_force)) < 0) {
1076         mp_msg(MSGT_DEMUXER, MSGL_ERR, "-sub-demuxer %s does not exist.\n",
1077                opts->sub_demuxer_name);
1078         if (opts->sub_stream)
1079             return NULL;
1080     }
1081
1082     if (opts->audio_stream) {
1083         as = open_stream(opts->audio_stream, 0, &afmt);
1084         if (!as) {
1085             mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "Cannot open audio stream: %s\n",
1086                    opts->audio_stream);
1087             return NULL;
1088         }
1089         if (opts->audio_stream_cache) {
1090             if (!stream_enable_cache
1091                 (as, opts->audio_stream_cache * 1024,
1092                  opts->audio_stream_cache * 1024 *
1093                             (opts->stream_cache_min_percent / 100.0),
1094                  opts->audio_stream_cache * 1024 *
1095                             (opts->stream_cache_seek_min_percent / 100.0))) {
1096                 free_stream(as);
1097                 mp_msg(MSGT_DEMUXER, MSGL_ERR,
1098                        "Can't enable audio stream cache\n");
1099                 return NULL;
1100             }
1101         }
1102     }
1103     if (opts->sub_stream) {
1104         ss = open_stream(opts->sub_stream, 0, &sfmt);
1105         if (!ss) {
1106             mp_tmsg(MSGT_DEMUXER, MSGL_ERR, "Cannot open subtitle stream: %s\n",
1107                    opts->sub_stream);
1108             return NULL;
1109         }
1110     }
1111
1112     vd = demux_open_stream(opts, vs, demuxer_type ? demuxer_type : file_format,
1113                            demuxer_force, opts->audio_stream ? -2 : audio_id,
1114                            video_id, opts->sub_stream ? -2 : dvdsub_id, filename);
1115     if (!vd) {
1116         if (as)
1117             free_stream(as);
1118         if (ss)
1119             free_stream(ss);
1120         return NULL;
1121     }
1122     if (as) {
1123         ad = demux_open_stream(opts, as,
1124                                audio_demuxer_type ? audio_demuxer_type : afmt,
1125                                audio_demuxer_force, audio_id, -2, -2,
1126                                opts->audio_stream);
1127         if (!ad) {
1128             mp_tmsg(MSGT_DEMUXER, MSGL_WARN, "Failed to open audio demuxer: %s\n",
1129                    opts->audio_stream);
1130             free_stream(as);
1131         } else if (ad->audio->sh
1132                    && ((sh_audio_t *) ad->audio->sh)->format == 0x55) // MP3
1133             opts->hr_mp3_seek = 1;    // Enable high res seeking
1134     }
1135     if (ss) {
1136         sd = demux_open_stream(opts, ss,
1137                                sub_demuxer_type ? sub_demuxer_type : sfmt,
1138                                sub_demuxer_force, -2, -2, dvdsub_id,
1139                                opts->sub_stream);
1140         if (!sd) {
1141             mp_tmsg(MSGT_DEMUXER, MSGL_WARN,
1142                    "Failed to open subtitle demuxer: %s\n", opts->sub_stream);
1143             free_stream(ss);
1144         }
1145     }
1146
1147     if (ad && sd)
1148         res = new_demuxers_demuxer(vd, ad, sd);
1149     else if (ad)
1150         res = new_demuxers_demuxer(vd, ad, vd);
1151     else if (sd)
1152         res = new_demuxers_demuxer(vd, vd, sd);
1153     else
1154         res = vd;
1155
1156     opts->correct_pts = opts->user_correct_pts;
1157     if (opts->correct_pts < 0)
1158         opts->correct_pts =
1159             demux_control(vd ? vd : res, DEMUXER_CTRL_CORRECT_PTS,
1160                           NULL) == DEMUXER_CTRL_OK;
1161     return res;
1162 }
1163
1164
1165 void demux_flush(demuxer_t *demuxer)
1166 {
1167     ds_free_packs(demuxer->video);
1168     ds_free_packs(demuxer->audio);
1169     ds_free_packs(demuxer->sub);
1170 }
1171
1172 int demux_seek(demuxer_t *demuxer, float rel_seek_secs, float audio_delay,
1173                int flags)
1174 {
1175     if (!demuxer->seekable) {
1176         if (demuxer->file_format == DEMUXER_TYPE_AVI)
1177             mp_tmsg(MSGT_SEEK, MSGL_WARN, "Cannot seek in raw AVI streams. (Index required, try with the -idx switch.)\n");
1178 #ifdef CONFIG_TV
1179         else if (demuxer->file_format == DEMUXER_TYPE_TV)
1180             mp_tmsg(MSGT_SEEK, MSGL_WARN, "TV input is not seekable! (Seeking will probably be for changing channels ;)\n");
1181 #endif
1182         else
1183             mp_tmsg(MSGT_SEEK, MSGL_WARN, "Cannot seek in this file.\n");
1184         return 0;
1185     }
1186     // clear demux buffers:
1187     demux_flush(demuxer);
1188     demuxer->video->eof = 0;
1189     demuxer->audio->eof = 0;
1190     demuxer->sub->eof = 0;
1191
1192     /* HACK: assume any demuxer used with these streams can cope with
1193      * the stream layer suddenly seeking to a different position under it
1194      * (nothing actually implements DEMUXER_CTRL_RESYNC now).
1195      */
1196     struct stream *stream = demuxer->stream;
1197     if (stream->type == STREAMTYPE_DVD || stream->type == STREAMTYPE_DVDNAV) {
1198         double pts;
1199
1200         if (flags & SEEK_ABSOLUTE)
1201             pts = 0.0f;
1202         else {
1203             if (demuxer->stream_pts == MP_NOPTS_VALUE)
1204                 goto dmx_seek;
1205             pts = demuxer->stream_pts;
1206         }
1207
1208         if (flags & SEEK_FACTOR) {
1209             double tmp = 0;
1210             if (stream_control(demuxer->stream, STREAM_CTRL_GET_TIME_LENGTH,
1211                                &tmp) == STREAM_UNSUPPORTED)
1212                 goto dmx_seek;
1213             pts += tmp * rel_seek_secs;
1214         } else
1215             pts += rel_seek_secs;
1216
1217         if (stream_control(demuxer->stream, STREAM_CTRL_SEEK_TO_TIME, &pts)
1218             != STREAM_UNSUPPORTED) {
1219             demux_control(demuxer, DEMUXER_CTRL_RESYNC, NULL);
1220             return 1;
1221         }
1222     }
1223
1224   dmx_seek:
1225     if (demuxer->desc->seek)
1226         demuxer->desc->seek(demuxer, rel_seek_secs, audio_delay, flags);
1227
1228     return 1;
1229 }
1230
1231 int demux_info_add(demuxer_t *demuxer, const char *opt, const char *param)
1232 {
1233     return demux_info_add_bstr(demuxer, BSTR(opt), BSTR(param));
1234 }
1235
1236 int demux_info_add_bstr(demuxer_t *demuxer, struct bstr opt, struct bstr param)
1237 {
1238     char **info = demuxer->info;
1239     int n = 0;
1240
1241
1242     for (n = 0; info && info[2 * n] != NULL; n++) {
1243         if (!bstrcasecmp(opt, BSTR(info[2*n]))) {
1244             if (!bstrcmp(param, BSTR(info[2*n + 1]))) {
1245                 mp_msg(MSGT_DEMUX, MSGL_V, "Demuxer info %.*s set to unchanged value %.*s\n",
1246                        BSTR_P(opt), BSTR_P(param));
1247                 return 0;
1248             }
1249             mp_tmsg(MSGT_DEMUX, MSGL_INFO, "Demuxer info %.*s changed to %.*s\n",
1250                     BSTR_P(opt), BSTR_P(param));
1251             talloc_free(info[2*n + 1]);
1252             info[2*n + 1] = talloc_strndup(demuxer->info, param.start, param.len);
1253             return 0;
1254         }
1255     }
1256
1257     info = demuxer->info = talloc_realloc(demuxer, info, char *, 2 * (n + 2));
1258     info[2*n]     = talloc_strndup(demuxer->info, opt.start,   opt.len);
1259     info[2*n + 1] = talloc_strndup(demuxer->info, param.start, param.len);
1260     memset(&info[2 * (n + 1)], 0, 2 * sizeof(char *));
1261
1262     return 1;
1263 }
1264
1265 int demux_info_print(demuxer_t *demuxer)
1266 {
1267     char **info = demuxer->info;
1268     int n;
1269
1270     if (!info)
1271         return 0;
1272
1273     mp_tmsg(MSGT_DEMUX, MSGL_INFO, "Clip info:\n");
1274     for (n = 0; info[2 * n] != NULL; n++) {
1275         mp_msg(MSGT_DEMUX, MSGL_INFO, " %s: %s\n", info[2 * n],
1276                info[2 * n + 1]);
1277         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CLIP_INFO_NAME%d=%s\n", n,
1278                info[2 * n]);
1279         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CLIP_INFO_VALUE%d=%s\n", n,
1280                info[2 * n + 1]);
1281     }
1282     mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CLIP_INFO_N=%d\n", n);
1283
1284     return 0;
1285 }
1286
1287 char *demux_info_get(demuxer_t *demuxer, const char *opt)
1288 {
1289     int i;
1290     char **info = demuxer->info;
1291
1292     for (i = 0; info && info[2 * i] != NULL; i++) {
1293         if (!strcasecmp(opt, info[2 * i]))
1294             return info[2 * i + 1];
1295     }
1296
1297     return NULL;
1298 }
1299
1300 int demux_control(demuxer_t *demuxer, int cmd, void *arg)
1301 {
1302
1303     if (demuxer->desc->control)
1304         return demuxer->desc->control(demuxer, cmd, arg);
1305
1306     return DEMUXER_CTRL_NOTIMPL;
1307 }
1308
1309 int demuxer_switch_audio(demuxer_t *demuxer, int index)
1310 {
1311     int res = demux_control(demuxer, DEMUXER_CTRL_SWITCH_AUDIO, &index);
1312     if (res == DEMUXER_CTRL_NOTIMPL) {
1313         struct sh_audio *sh_audio = demuxer->audio->sh;
1314         return sh_audio ? sh_audio->aid : -2;
1315     }
1316     if (demuxer->audio->id >= 0) {
1317         struct sh_audio *sh_audio = demuxer->a_streams[demuxer->audio->id];
1318         demuxer->audio->sh = sh_audio;
1319         index = sh_audio->aid; // internal MPEG demuxers don't set it right
1320     }
1321     else
1322         demuxer->audio->sh = NULL;
1323     return index;
1324 }
1325
1326 int demuxer_switch_video(demuxer_t *demuxer, int index)
1327 {
1328     int res = demux_control(demuxer, DEMUXER_CTRL_SWITCH_VIDEO, &index);
1329     if (res == DEMUXER_CTRL_NOTIMPL) {
1330         struct sh_video *sh_video = demuxer->video->sh;
1331         return sh_video ? sh_video->vid : -2;
1332     }
1333     if (demuxer->video->id >= 0) {
1334         struct sh_video *sh_video = demuxer->v_streams[demuxer->video->id];
1335         demuxer->video->sh = sh_video;
1336         index = sh_video->vid; // internal MPEG demuxers don't set it right
1337     } else
1338         demuxer->video->sh = NULL;
1339     return index;
1340 }
1341
1342 int demuxer_add_attachment(demuxer_t *demuxer, struct bstr name,
1343                            struct bstr type, struct bstr data)
1344 {
1345     if (!(demuxer->num_attachments % 32))
1346         demuxer->attachments = talloc_realloc(demuxer, demuxer->attachments,
1347                                               struct demux_attachment,
1348                                               demuxer->num_attachments + 32);
1349
1350     struct demux_attachment *att =
1351         demuxer->attachments + demuxer->num_attachments;
1352     att->name = talloc_strndup(demuxer->attachments, name.start, name.len);
1353     att->type = talloc_strndup(demuxer->attachments, type.start, type.len);
1354     att->data = talloc_size(demuxer->attachments, data.len);
1355     memcpy(att->data, data.start, data.len);
1356     att->data_size = data.len;
1357
1358     return demuxer->num_attachments++;
1359 }
1360
1361 int demuxer_add_chapter(demuxer_t *demuxer, struct bstr name,
1362                         uint64_t start, uint64_t end)
1363 {
1364     if (!(demuxer->num_chapters % 32))
1365         demuxer->chapters = talloc_realloc(demuxer, demuxer->chapters,
1366                                            struct demux_chapter,
1367                                            demuxer->num_chapters + 32);
1368
1369     demuxer->chapters[demuxer->num_chapters].start = start;
1370     demuxer->chapters[demuxer->num_chapters].end = end;
1371     demuxer->chapters[demuxer->num_chapters].name = name.len ?
1372         talloc_strndup(demuxer->chapters, name.start, name.len) :
1373         talloc_strdup(demuxer->chapters, mp_gtext("unknown"));
1374
1375     mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CHAPTER_ID=%d\n", demuxer->num_chapters);
1376     mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CHAPTER_%d_START=%"PRIu64"\n", demuxer->num_chapters, start / 1000000);
1377     if (end)
1378         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CHAPTER_%d_END=%"PRIu64"\n", demuxer->num_chapters, end / 1000000);
1379     if (name.start)
1380         mp_msg(MSGT_IDENTIFY, MSGL_INFO, "ID_CHAPTER_%d_NAME=%.*s\n", demuxer->num_chapters, BSTR_P(name));
1381
1382     return demuxer->num_chapters++;
1383 }
1384
1385 /**
1386  * \brief demuxer_seek_chapter() seeks to a chapter in two possible ways:
1387  *        either using the demuxer->chapters structure set by the demuxer
1388  *        or asking help to the stream layer (e.g. dvd)
1389  * \param chapter - chapter number wished - 0-based
1390  * \param seek_pts set by the function to the pts to seek to (if demuxer->chapters is set)
1391  * \param chapter_name name of chapter found (set by this function is param is not null)
1392  * \return -1 on error, current chapter if successful
1393  */
1394
1395 int demuxer_seek_chapter(demuxer_t *demuxer, int chapter, double *seek_pts,
1396                          char **chapter_name)
1397 {
1398     int ris;
1399
1400     if (!demuxer->num_chapters || !demuxer->chapters) {
1401         demux_flush(demuxer);
1402
1403         ris = stream_control(demuxer->stream, STREAM_CTRL_SEEK_TO_CHAPTER,
1404                              &chapter);
1405         if (ris != STREAM_UNSUPPORTED)
1406             demux_control(demuxer, DEMUXER_CTRL_RESYNC, NULL);
1407
1408         // exit status may be ok, but main() doesn't have to seek itself
1409         // (because e.g. dvds depend on sectors, not on pts)
1410         *seek_pts = -1.0;
1411
1412         if (chapter_name) {
1413             *chapter_name = NULL;
1414             int num_chapters;
1415             if (stream_control(demuxer->stream, STREAM_CTRL_GET_NUM_CHAPTERS,
1416                                &num_chapters) == STREAM_UNSUPPORTED)
1417                 num_chapters = 0;
1418             if (num_chapters) {
1419                 *chapter_name = talloc_size(NULL, 16);
1420                 sprintf(*chapter_name, " of %3d", num_chapters);
1421             }
1422         }
1423
1424         return ris != STREAM_UNSUPPORTED ? chapter : -1;
1425     } else {  // chapters structure is set in the demuxer
1426         if (chapter >= demuxer->num_chapters)
1427             return -1;
1428         if (chapter < 0)
1429             chapter = 0;
1430
1431         *seek_pts = demuxer->chapters[chapter].start / 1e9;
1432
1433         if (chapter_name)
1434             *chapter_name = talloc_strdup(NULL, demuxer->chapters[chapter].name);
1435
1436         return chapter;
1437     }
1438 }
1439
1440 int demuxer_get_current_chapter(demuxer_t *demuxer, double time_now)
1441 {
1442     int chapter = -2;
1443     if (!demuxer->num_chapters || !demuxer->chapters) {
1444         if (stream_control(demuxer->stream, STREAM_CTRL_GET_CURRENT_CHAPTER,
1445                            &chapter) == STREAM_UNSUPPORTED)
1446             chapter = -2;
1447     } else {
1448         uint64_t now = time_now * 1e9 + 0.5;
1449         for (chapter = demuxer->num_chapters - 1; chapter >= 0; --chapter) {
1450             if (demuxer->chapters[chapter].start <= now)
1451                 break;
1452         }
1453     }
1454     return chapter;
1455 }
1456
1457 char *demuxer_chapter_name(demuxer_t *demuxer, int chapter)
1458 {
1459     if (demuxer->num_chapters && demuxer->chapters) {
1460         if (chapter >= 0 && chapter < demuxer->num_chapters
1461             && demuxer->chapters[chapter].name)
1462             return strdup(demuxer->chapters[chapter].name);
1463     }
1464     return NULL;
1465 }
1466
1467 char *demuxer_chapter_display_name(demuxer_t *demuxer, int chapter)
1468 {
1469     char *chapter_name = demuxer_chapter_name(demuxer, chapter);
1470     if (chapter_name) {
1471         char *tmp = malloc(strlen(chapter_name) + 14);
1472         snprintf(tmp, 63, "(%d) %s", chapter + 1, chapter_name);
1473         free(chapter_name);
1474         return tmp;
1475     } else {
1476         int chapter_num = demuxer_chapter_count(demuxer);
1477         char tmp[30];
1478         if (chapter_num <= 0)
1479             sprintf(tmp, "(%d)", chapter + 1);
1480         else
1481             sprintf(tmp, "(%d) of %d", chapter + 1, chapter_num);
1482         return strdup(tmp);
1483     }
1484 }
1485
1486 float demuxer_chapter_time(demuxer_t *demuxer, int chapter, float *end)
1487 {
1488     if (demuxer->num_chapters && demuxer->chapters && chapter >= 0
1489         && chapter < demuxer->num_chapters) {
1490         if (end)
1491             *end = demuxer->chapters[chapter].end / 1e9;
1492         return demuxer->chapters[chapter].start / 1e9;
1493     }
1494     return -1.0;
1495 }
1496
1497 int demuxer_chapter_count(demuxer_t *demuxer)
1498 {
1499     if (!demuxer->num_chapters || !demuxer->chapters) {
1500         int num_chapters = 0;
1501         if (stream_control(demuxer->stream, STREAM_CTRL_GET_NUM_CHAPTERS,
1502                            &num_chapters) == STREAM_UNSUPPORTED)
1503             num_chapters = 0;
1504         return num_chapters;
1505     } else
1506         return demuxer->num_chapters;
1507 }
1508
1509 int demuxer_angles_count(demuxer_t *demuxer)
1510 {
1511     int ris, angles = -1;
1512
1513     ris = stream_control(demuxer->stream, STREAM_CTRL_GET_NUM_ANGLES, &angles);
1514     if (ris == STREAM_UNSUPPORTED)
1515         return -1;
1516     return angles;
1517 }
1518
1519 int demuxer_get_current_angle(demuxer_t *demuxer)
1520 {
1521     int ris, curr_angle = -1;
1522     ris = stream_control(demuxer->stream, STREAM_CTRL_GET_ANGLE, &curr_angle);
1523     if (ris == STREAM_UNSUPPORTED)
1524         return -1;
1525     return curr_angle;
1526 }
1527
1528
1529 int demuxer_set_angle(demuxer_t *demuxer, int angle)
1530 {
1531     int ris, angles = -1;
1532
1533     angles = demuxer_angles_count(demuxer);
1534     if ((angles < 1) || (angle > angles))
1535         return -1;
1536
1537     demux_flush(demuxer);
1538
1539     ris = stream_control(demuxer->stream, STREAM_CTRL_SET_ANGLE, &angle);
1540     if (ris == STREAM_UNSUPPORTED)
1541         return -1;
1542
1543     demux_control(demuxer, DEMUXER_CTRL_RESYNC, NULL);
1544
1545     return angle;
1546 }
1547
1548 int demuxer_audio_track_by_lang_and_default(struct demuxer *d, char **langt)
1549 {
1550     int n = 0;
1551     while (1) {
1552         char *lang = langt ? langt[n++] : NULL;
1553         int id = -1;
1554         for (int i = 0; i < MAX_A_STREAMS; i++) {
1555             struct sh_audio *sh = d->a_streams[i];
1556             if (sh && (!lang || sh->lang && !strcmp(lang, sh->lang))) {
1557                 if (sh->default_track)
1558                     return sh->aid;
1559                 if (id < 0)
1560                     id = sh->aid;
1561             }
1562         }
1563         if (id >= 0)
1564             return id;
1565         if (!lang)
1566             return -1;
1567     }
1568 }
1569
1570 int demuxer_sub_track_by_lang_and_default(struct demuxer *d, char **langt)
1571 {
1572     int n = 0;
1573     while (1) {
1574         char *lang = langt ? langt[n++] : NULL;
1575         int id = -1;
1576         for (int i = 0; i < MAX_S_STREAMS; i++) {
1577             struct sh_sub *sh = d->s_streams[i];
1578             if (sh && (!lang || sh->lang && !strcmp(lang, sh->lang))) {
1579                 if (sh->default_track)
1580                     return sh->sid;
1581                 if (id < 0)
1582                     id = sh->sid;
1583             }
1584         }
1585         if (!lang)
1586             return -1;
1587         if (id >= 0)
1588             return id;
1589     }
1590 }