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