subprojects/gst-plugins-good/gst/isomp4/qtdemux.c

   1 /* GStreamer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  * Copyright (C) <2003> David A. Schleef <ds@schleef.org>
   4  * Copyright (C) <2006> Wim Taymans <wim@fluendo.com>
   5  * Copyright (C) <2007> Julien Moutte <julien@fluendo.com>
   6  * Copyright (C) <2009> Tim-Philipp Müller <tim centricular net>
   7  * Copyright (C) <2009> STEricsson <benjamin.gaignard@stericsson.com>
   8  * Copyright (C) <2013> Sreerenj Balachandran <sreerenj.balachandran@intel.com>
   9  * Copyright (C) <2013> Intel Corporation
  10  * Copyright (C) <2014> Centricular Ltd
  11  * Copyright (C) <2015> YouView TV Ltd.
  12  * Copyright (C) <2016> British Broadcasting Corporation
  13  *
  14  * This library is free software; you can redistribute it and/or
  15  * modify it under the terms of the GNU Library General Public
  16  * License as published by the Free Software Foundation; either
  17  * version 2 of the License, or (at your option) any later version.
  18  *
  19  * This library is distributed in the hope that it will be useful,
  20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  22  * Library General Public License for more details.
  23  *
  24  * You should have received a copy of the GNU Library General Public
  25  * License along with this library; if not, write to the
  26  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  27  * Boston, MA 02110-1301, USA.
  28  */
  29
  30 /**
  31  * SECTION:element-qtdemux
  32  * @title: qtdemux
  33  *
  34  * Demuxes a .mov file into raw or compressed audio and/or video streams.
  35  *
  36  * This element supports both push and pull-based scheduling, depending on the
  37  * capabilities of the upstream elements.
  38  *
  39  * ## Example launch line
  40  * |[
  41  * gst-launch-1.0 filesrc location=test.mov ! qtdemux name=demux  demux.audio_0 ! queue ! decodebin ! audioconvert ! audioresample ! autoaudiosink   demux.video_0 ! queue ! decodebin ! videoconvert ! videoscale ! autovideosink
  42  * ]| Play (parse and decode) a .mov file and try to output it to
  43  * an automatically detected soundcard and videosink. If the MOV file contains
  44  * compressed audio or video data, this will only work if you have the
  45  * right decoder elements/plugins installed.
  46  *
  47  */
  48
  49 #ifdef HAVE_CONFIG_H
  50 #include "config.h"
  51 #endif
  52
  53 #include "gst/gst-i18n-plugin.h"
  54
  55 #include <glib/gprintf.h>
  56 #include <gst/base/base.h>
  57 #include <gst/tag/tag.h>
  58 #include <gst/audio/audio.h>
  59 #include <gst/riff/riff.h>
  60 #include <gst/pbutils/pbutils.h>
  61
  62 #include "gstisomp4elements.h"
  63 #include "qtatomparser.h"
  64 #include "qtdemux_types.h"
  65 #include "qtdemux_dump.h"
  66 #include "fourcc.h"
  67 #include "descriptors.h"
  68 #include "qtdemux_lang.h"
  69 #include "qtdemux.h"
  70 #include "qtpalette.h"
  71 #include "qtdemux_tags.h"
  72 #include "qtdemux_tree.h"
  73 #include "qtdemux-webvtt.h"
  74
  75 #include <stdlib.h>
  76 #include <string.h>
  77
  78 #include <math.h>
  79 #include <gst/math-compat.h>
  80
  81 #ifdef HAVE_ZLIB
  82 # include <zlib.h>
  83 #endif
  84
  85 /* max. size considered 'sane' for non-mdat atoms */
  86 #define QTDEMUX_MAX_ATOM_SIZE (32*1024*1024)
  87
  88 /* if the sample index is larger than this, something is likely wrong */
  89 #define QTDEMUX_MAX_SAMPLE_INDEX_SIZE (200*1024*1024)
  90
  91 /* For converting qt creation times to unix epoch times */
  92 #define QTDEMUX_SECONDS_PER_DAY (60 * 60 * 24)
  93 #define QTDEMUX_LEAP_YEARS_FROM_1904_TO_1970 17
  94 #define QTDEMUX_SECONDS_FROM_1904_TO_1970 (((1970 - 1904) * (guint64) 365 + \
  95     QTDEMUX_LEAP_YEARS_FROM_1904_TO_1970) * QTDEMUX_SECONDS_PER_DAY)
  96
  97 #define QTDEMUX_TREE_NODE_FOURCC(n) (QT_FOURCC(((guint8 *) (n)->data) + 4))
  98
  99 #define STREAM_IS_EOS(s) ((s)->time_position == GST_CLOCK_TIME_NONE)
 100
 101 #define ABSDIFF(x, y) ( (x) > (y) ? ((x) - (y)) : ((y) - (x)) )
 102
 103 #define QTDEMUX_STREAM(s) ((QtDemuxStream *)(s))
 104 #define QTDEMUX_N_STREAMS(demux) ((demux)->active_streams->len)
 105 #define QTDEMUX_NTH_STREAM(demux,idx) \
 106    QTDEMUX_STREAM(g_ptr_array_index((demux)->active_streams,idx))
 107 #define QTDEMUX_NTH_OLD_STREAM(demux,idx) \
 108    QTDEMUX_STREAM(g_ptr_array_index((demux)->old_streams,idx))
 109
 110 #define CUR_STREAM(s) (&((s)->stsd_entries[(s)->cur_stsd_entry_index]))
 111
 112 GST_DEBUG_CATEGORY (qtdemux_debug);
 113 #define GST_CAT_DEFAULT qtdemux_debug
 114
 115 typedef struct _QtDemuxCencSampleSetInfo QtDemuxCencSampleSetInfo;
 116 typedef struct _QtDemuxAavdEncryptionInfo QtDemuxAavdEncryptionInfo;
 117
 118 /* Macros for converting to/from timescale */
 119 #define QTSTREAMTIME_TO_GSTTIME(stream, value) (gst_util_uint64_scale((value), GST_SECOND, (stream)->timescale))
 120 #define GSTTIME_TO_QTSTREAMTIME(stream, value) (gst_util_uint64_scale((value), (stream)->timescale, GST_SECOND))
 121
 122 #define QTTIME_TO_GSTTIME(qtdemux, value) (gst_util_uint64_scale((value), GST_SECOND, (qtdemux)->timescale))
 123 #define GSTTIME_TO_QTTIME(qtdemux, value) (gst_util_uint64_scale((value), (qtdemux)->timescale, GST_SECOND))
 124
 125 /* timestamp is the DTS */
 126 #define QTSAMPLE_DTS(stream,sample) (QTSTREAMTIME_TO_GSTTIME((stream), (sample)->timestamp))
 127 /* timestamp + offset + cslg_shift is the outgoing PTS */
 128 #define QTSAMPLE_PTS(stream,sample) (QTSTREAMTIME_TO_GSTTIME((stream), (sample)->timestamp + (stream)->cslg_shift + (sample)->pts_offset))
 129 /* timestamp + offset is the PTS used for internal seek calculations */
 130 #define QTSAMPLE_PTS_NO_CSLG(stream,sample) (QTSTREAMTIME_TO_GSTTIME((stream), (sample)->timestamp + (sample)->pts_offset))
 131 /* timestamp + duration - dts is the duration */
 132 #define QTSAMPLE_DUR_DTS(stream, sample, dts) (QTSTREAMTIME_TO_GSTTIME ((stream), (sample)->timestamp + (sample)->duration) - (dts))
 133
 134 #define QTSAMPLE_KEYFRAME(stream,sample) ((stream)->all_keyframe || (sample)->keyframe)
 135
 136 #define QTDEMUX_EXPOSE_GET_LOCK(demux) (&((demux)->expose_lock))
 137 #define QTDEMUX_EXPOSE_LOCK(demux) G_STMT_START { \
 138     GST_TRACE("Locking from thread %p", g_thread_self()); \
 139     g_mutex_lock (QTDEMUX_EXPOSE_GET_LOCK (demux)); \
 140     GST_TRACE("Locked from thread %p", g_thread_self()); \
 141  } G_STMT_END
 142
 143 #define QTDEMUX_EXPOSE_UNLOCK(demux) G_STMT_START { \
 144     GST_TRACE("Unlocking from thread %p", g_thread_self()); \
 145     g_mutex_unlock (QTDEMUX_EXPOSE_GET_LOCK (demux)); \
 146  } G_STMT_END
 147
 148 /*
 149  * Quicktime has tracks and segments. A track is a continuous piece of
 150  * multimedia content. The track is not always played from start to finish but
 151  * instead, pieces of the track are 'cut out' and played in sequence. This is
 152  * what the segments do.
 153  *
 154  * Inside the track we have keyframes (K) and delta frames. The track has its
 155  * own timing, which starts from 0 and extends to end. The position in the track
 156  * is called the media_time.
 157  *
 158  * The segments now describe the pieces that should be played from this track
 159  * and are basically tuples of media_time/duration/rate entries. We can have
 160  * multiple segments and they are all played after one another. An example:
 161  *
 162  * segment 1: media_time: 1 second, duration: 1 second, rate 1
 163  * segment 2: media_time: 3 second, duration: 2 second, rate 2
 164  *
 165  * To correctly play back this track, one must play: 1 second of media starting
 166  * from media_time 1 followed by 2 seconds of media starting from media_time 3
 167  * at a rate of 2.
 168  *
 169  * Each of the segments will be played at a specific time, the first segment at
 170  * time 0, the second one after the duration of the first one, etc.. Note that
 171  * the time in resulting playback is not identical to the media_time of the
 172  * track anymore.
 173  *
 174  * Visually, assuming the track has 4 second of media_time:
 175  *
 176  *                (a)                   (b)          (c)              (d)
 177  *         .-----------------------------------------------------------.
 178  * track:  | K.....K.........K........K.......K.......K...........K... |
 179  *         '-----------------------------------------------------------'
 180  *         0              1              2              3              4
 181  *           .------------^              ^   .----------^              ^
 182  *          /              .-------------'  /       .------------------'
 183  *         /              /          .-----'       /
 184  *         .--------------.         .--------------.
 185  *         | segment 1    |         | segment 2    |
 186  *         '--------------'         '--------------'
 187  *
 188  * The challenge here is to cut out the right pieces of the track for each of
 189  * the playback segments. This fortunately can easily be done with the SEGMENT
 190  * events of GStreamer.
 191  *
 192  * For playback of segment 1, we need to provide the decoder with the keyframe
 193  * (a), in the above figure, but we must instruct it only to output the decoded
 194  * data between second 1 and 2. We do this with a SEGMENT event for 1 to 2, time
 195  * position set to the time of the segment: 0.
 196  *
 197  * We then proceed to push data from keyframe (a) to frame (b). The decoder
 198  * decodes but clips all before media_time 1.
 199  *
 200  * After finishing a segment, we push out a new SEGMENT event with the clipping
 201  * boundaries of the new data.
 202  *
 203  * This is a good usecase for the GStreamer accumulated SEGMENT events.
 204  */
 205
 206 struct _QtDemuxSegment
 207 {
 208   /* global time and duration, all gst time */
 209   GstClockTime time;
 210   GstClockTime stop_time;
 211   GstClockTime duration;
 212   /* media time of trak, all gst time */
 213   GstClockTime media_start;
 214   GstClockTime media_stop;
 215   gdouble rate;
 216   /* Media start time in trak timescale units */
 217   guint32 trak_media_start;
 218 };
 219
 220 #define QTSEGMENT_IS_EMPTY(s) ((s)->media_start == GST_CLOCK_TIME_NONE)
 221
 222 /* Used with fragmented MP4 files (mfra atom) */
 223 struct _QtDemuxRandomAccessEntry
 224 {
 225   GstClockTime ts;
 226   guint64 moof_offset;
 227 };
 228
 229
 230 /* Contains properties and cryptographic info for a set of samples from a
 231  * track protected using Common Encryption (cenc) */
 232 struct _QtDemuxCencSampleSetInfo
 233 {
 234   GstStructure *default_properties;
 235
 236   /* @crypto_info holds one GstStructure per sample */
 237   GPtrArray *crypto_info;
 238 };
 239
 240 struct _QtDemuxAavdEncryptionInfo
 241 {
 242   GstStructure *default_properties;
 243 };
 244
 245 static const gchar *
 246 qt_demux_state_string (enum QtDemuxState state)
 247 {
 248   switch (state) {
 249     case QTDEMUX_STATE_INITIAL:
 250       return "<INITIAL>";
 251     case QTDEMUX_STATE_HEADER:
 252       return "<HEADER>";
 253     case QTDEMUX_STATE_MOVIE:
 254       return "<MOVIE>";
 255     case QTDEMUX_STATE_BUFFER_MDAT:
 256       return "<BUFFER_MDAT>";
 257     default:
 258       return "<UNKNOWN>";
 259   }
 260 }
 261
 262 static GstFlowReturn qtdemux_add_fragmented_samples (GstQTDemux * qtdemux);
 263
 264 static void gst_qtdemux_check_send_pending_segment (GstQTDemux * demux);
 265
 266 static GstStaticPadTemplate gst_qtdemux_sink_template =
 267     GST_STATIC_PAD_TEMPLATE ("sink",
 268     GST_PAD_SINK,
 269     GST_PAD_ALWAYS,
 270     GST_STATIC_CAPS ("video/quicktime; video/mj2; audio/x-m4a; "
 271         "application/x-3gp")
 272     );
 273
 274 static GstStaticPadTemplate gst_qtdemux_videosrc_template =
 275 GST_STATIC_PAD_TEMPLATE ("video_%u",
 276     GST_PAD_SRC,
 277     GST_PAD_SOMETIMES,
 278     GST_STATIC_CAPS_ANY);
 279
 280 static GstStaticPadTemplate gst_qtdemux_audiosrc_template =
 281 GST_STATIC_PAD_TEMPLATE ("audio_%u",
 282     GST_PAD_SRC,
 283     GST_PAD_SOMETIMES,
 284     GST_STATIC_CAPS_ANY);
 285
 286 static GstStaticPadTemplate gst_qtdemux_subsrc_template =
 287 GST_STATIC_PAD_TEMPLATE ("subtitle_%u",
 288     GST_PAD_SRC,
 289     GST_PAD_SOMETIMES,
 290     GST_STATIC_CAPS_ANY);
 291
 292 #define gst_qtdemux_parent_class parent_class
 293 G_DEFINE_TYPE (GstQTDemux, gst_qtdemux, GST_TYPE_ELEMENT);
 294 GST_ELEMENT_REGISTER_DEFINE_WITH_CODE (qtdemux, "qtdemux",
 295     GST_RANK_PRIMARY, GST_TYPE_QTDEMUX, isomp4_element_init (plugin));
 296
 297 static void gst_qtdemux_dispose (GObject * object);
 298 static void gst_qtdemux_finalize (GObject * object);
 299
 300 static guint32
 301 gst_qtdemux_find_index_linear (GstQTDemux * qtdemux, QtDemuxStream * str,
 302     GstClockTime media_time);
 303 static guint32
 304 gst_qtdemux_find_index_for_given_media_offset_linear (GstQTDemux * qtdemux,
 305     QtDemuxStream * str, gint64 media_offset);
 306
 307 #if 0
 308 static void gst_qtdemux_set_index (GstElement * element, GstIndex * index);
 309 static GstIndex *gst_qtdemux_get_index (GstElement * element);
 310 #endif
 311 static GstStateChangeReturn gst_qtdemux_change_state (GstElement * element,
 312     GstStateChange transition);
 313 static void gst_qtdemux_set_context (GstElement * element,
 314     GstContext * context);
 315 static gboolean qtdemux_sink_activate (GstPad * sinkpad, GstObject * parent);
 316 static gboolean qtdemux_sink_activate_mode (GstPad * sinkpad,
 317     GstObject * parent, GstPadMode mode, gboolean active);
 318
 319 static void gst_qtdemux_loop (GstPad * pad);
 320 static GstFlowReturn gst_qtdemux_chain (GstPad * sinkpad, GstObject * parent,
 321     GstBuffer * inbuf);
 322 static gboolean gst_qtdemux_handle_sink_event (GstPad * pad, GstObject * parent,
 323     GstEvent * event);
 324 static gboolean gst_qtdemux_handle_sink_query (GstPad * pad, GstObject * parent,
 325     GstQuery * query);
 326 static gboolean gst_qtdemux_setcaps (GstQTDemux * qtdemux, GstCaps * caps);
 327 static gboolean gst_qtdemux_configure_stream (GstQTDemux * qtdemux,
 328     QtDemuxStream * stream);
 329 static void gst_qtdemux_stream_check_and_change_stsd_index (GstQTDemux * demux,
 330     QtDemuxStream * stream);
 331 static GstFlowReturn gst_qtdemux_process_adapter (GstQTDemux * demux,
 332     gboolean force);
 333
 334 static void gst_qtdemux_check_seekability (GstQTDemux * demux);
 335
 336 static gboolean qtdemux_parse_moov (GstQTDemux * qtdemux,
 337     const guint8 * buffer, guint length);
 338 static gboolean qtdemux_parse_node (GstQTDemux * qtdemux, GNode * node,
 339     const guint8 * buffer, guint length);
 340 static gboolean qtdemux_parse_tree (GstQTDemux * qtdemux);
 341
 342 static void gst_qtdemux_handle_esds (GstQTDemux * qtdemux,
 343     QtDemuxStream * stream, QtDemuxStreamStsdEntry * entry, GNode * esds,
 344     GstTagList * list);
 345 static GstCaps *qtdemux_video_caps (GstQTDemux * qtdemux,
 346     QtDemuxStream * stream, QtDemuxStreamStsdEntry * entry, guint32 fourcc,
 347     const guint8 * stsd_entry_data, gchar ** codec_name);
 348 static GstCaps *qtdemux_audio_caps (GstQTDemux * qtdemux,
 349     QtDemuxStream * stream, QtDemuxStreamStsdEntry * entry, guint32 fourcc,
 350     const guint8 * data, int len, gchar ** codec_name);
 351 static GstCaps *qtdemux_sub_caps (GstQTDemux * qtdemux, QtDemuxStream * stream,
 352     QtDemuxStreamStsdEntry * entry, guint32 fourcc, const guint8 * data,
 353     gchar ** codec_name);
 354 static GstCaps *qtdemux_generic_caps (GstQTDemux * qtdemux,
 355     QtDemuxStream * stream, QtDemuxStreamStsdEntry * entry, guint32 fourcc,
 356     const guint8 * stsd_entry_data, gchar ** codec_name);
 357
 358 static gboolean qtdemux_parse_samples (GstQTDemux * qtdemux,
 359     QtDemuxStream * stream, guint32 n);
 360 static GstFlowReturn qtdemux_expose_streams (GstQTDemux * qtdemux);
 361 static QtDemuxStream *gst_qtdemux_stream_ref (QtDemuxStream * stream);
 362 static void gst_qtdemux_stream_unref (QtDemuxStream * stream);
 363 static void gst_qtdemux_stream_clear (QtDemuxStream * stream);
 364 static GstFlowReturn qtdemux_prepare_streams (GstQTDemux * qtdemux);
 365 static void qtdemux_do_allocation (QtDemuxStream * stream,
 366     GstQTDemux * qtdemux);
 367 static gboolean gst_qtdemux_activate_segment (GstQTDemux * qtdemux,
 368     QtDemuxStream * stream, guint32 seg_idx, GstClockTime offset);
 369 static gboolean gst_qtdemux_stream_update_segment (GstQTDemux * qtdemux,
 370     QtDemuxStream * stream, gint seg_idx, GstClockTime offset,
 371     GstClockTime * _start, GstClockTime * _stop);
 372 static void gst_qtdemux_send_gap_for_segment (GstQTDemux * demux,
 373     QtDemuxStream * stream, gint segment_index, GstClockTime pos);
 374
 375 static gboolean qtdemux_pull_mfro_mfra (GstQTDemux * qtdemux);
 376 static void check_update_duration (GstQTDemux * qtdemux, GstClockTime duration);
 377
 378 static gchar *qtdemux_uuid_bytes_to_string (gconstpointer uuid_bytes);
 379
 380 static GstStructure *qtdemux_get_cenc_sample_properties (GstQTDemux * qtdemux,
 381     QtDemuxStream * stream, guint sample_index);
 382 static void gst_qtdemux_append_protection_system_id (GstQTDemux * qtdemux,
 383     const gchar * id);
 384 static void qtdemux_gst_structure_free (GstStructure * gststructure);
 385 static void gst_qtdemux_reset (GstQTDemux * qtdemux, gboolean hard);
 386
 387 static void
 388 gst_qtdemux_class_init (GstQTDemuxClass * klass)
 389 {
 390   GObjectClass *gobject_class;
 391   GstElementClass *gstelement_class;
 392
 393   gobject_class = (GObjectClass *) klass;
 394   gstelement_class = (GstElementClass *) klass;
 395
 396   parent_class = g_type_class_peek_parent (klass);
 397
 398   gobject_class->dispose = gst_qtdemux_dispose;
 399   gobject_class->finalize = gst_qtdemux_finalize;
 400
 401   gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_qtdemux_change_state);
 402 #if 0
 403   gstelement_class->set_index = GST_DEBUG_FUNCPTR (gst_qtdemux_set_index);
 404   gstelement_class->get_index = GST_DEBUG_FUNCPTR (gst_qtdemux_get_index);
 405 #endif
 406   gstelement_class->set_context = GST_DEBUG_FUNCPTR (gst_qtdemux_set_context);
 407
 408   gst_tag_register_musicbrainz_tags ();
 409
 410   gst_element_class_add_static_pad_template (gstelement_class,
 411       &gst_qtdemux_sink_template);
 412   gst_element_class_add_static_pad_template (gstelement_class,
 413       &gst_qtdemux_videosrc_template);
 414   gst_element_class_add_static_pad_template (gstelement_class,
 415       &gst_qtdemux_audiosrc_template);
 416   gst_element_class_add_static_pad_template (gstelement_class,
 417       &gst_qtdemux_subsrc_template);
 418   gst_element_class_set_static_metadata (gstelement_class, "QuickTime demuxer",
 419       "Codec/Demuxer",
 420       "Demultiplex a QuickTime file into audio and video streams",
 421       "David Schleef <ds@schleef.org>, Wim Taymans <wim@fluendo.com>");
 422
 423   GST_DEBUG_CATEGORY_INIT (qtdemux_debug, "qtdemux", 0, "qtdemux plugin");
 424   gst_riff_init ();
 425 }
 426
 427 static void
 428 gst_qtdemux_init (GstQTDemux * qtdemux)
 429 {
 430   qtdemux->sinkpad =
 431       gst_pad_new_from_static_template (&gst_qtdemux_sink_template, "sink");
 432   gst_pad_set_activate_function (qtdemux->sinkpad, qtdemux_sink_activate);
 433   gst_pad_set_activatemode_function (qtdemux->sinkpad,
 434       qtdemux_sink_activate_mode);
 435   gst_pad_set_chain_function (qtdemux->sinkpad, gst_qtdemux_chain);
 436   gst_pad_set_event_function (qtdemux->sinkpad, gst_qtdemux_handle_sink_event);
 437   gst_pad_set_query_function (qtdemux->sinkpad, gst_qtdemux_handle_sink_query);
 438   gst_element_add_pad (GST_ELEMENT_CAST (qtdemux), qtdemux->sinkpad);
 439
 440   qtdemux->adapter = gst_adapter_new ();
 441   g_queue_init (&qtdemux->protection_event_queue);
 442   qtdemux->flowcombiner = gst_flow_combiner_new ();
 443   g_mutex_init (&qtdemux->expose_lock);
 444
 445   qtdemux->active_streams = g_ptr_array_new_with_free_func
 446       ((GDestroyNotify) gst_qtdemux_stream_unref);
 447   qtdemux->old_streams = g_ptr_array_new_with_free_func
 448       ((GDestroyNotify) gst_qtdemux_stream_unref);
 449
 450   GST_OBJECT_FLAG_SET (qtdemux, GST_ELEMENT_FLAG_INDEXABLE);
 451
 452   gst_qtdemux_reset (qtdemux, TRUE);
 453 }
 454
 455 static void
 456 gst_qtdemux_finalize (GObject * object)
 457 {
 458   GstQTDemux *qtdemux = GST_QTDEMUX (object);
 459
 460   g_free (qtdemux->redirect_location);
 461
 462   G_OBJECT_CLASS (parent_class)->finalize (object);
 463 }
 464
 465 static void
 466 gst_qtdemux_dispose (GObject * object)
 467 {
 468   GstQTDemux *qtdemux = GST_QTDEMUX (object);
 469
 470   if (qtdemux->adapter) {
 471     g_object_unref (G_OBJECT (qtdemux->adapter));
 472     qtdemux->adapter = NULL;
 473   }
 474   gst_tag_list_unref (qtdemux->tag_list);
 475   gst_flow_combiner_free (qtdemux->flowcombiner);
 476   g_queue_foreach (&qtdemux->protection_event_queue, (GFunc) gst_event_unref,
 477       NULL);
 478   g_queue_clear (&qtdemux->protection_event_queue);
 479
 480   g_free (qtdemux->cenc_aux_info_sizes);
 481   qtdemux->cenc_aux_info_sizes = NULL;
 482   g_mutex_clear (&qtdemux->expose_lock);
 483
 484   g_ptr_array_free (qtdemux->active_streams, TRUE);
 485   g_ptr_array_free (qtdemux->old_streams, TRUE);
 486
 487   G_OBJECT_CLASS (parent_class)->dispose (object);
 488 }
 489
 490 static void
 491 gst_qtdemux_post_no_playable_stream_error (GstQTDemux * qtdemux)
 492 {
 493   if (qtdemux->redirect_location) {
 494     GST_ELEMENT_ERROR_WITH_DETAILS (qtdemux, STREAM, DEMUX,
 495         (_("This file contains no playable streams.")),
 496         ("no known streams found, a redirect message has been posted"),
 497         ("redirect-location", G_TYPE_STRING, qtdemux->redirect_location, NULL));
 498   } else {
 499     GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
 500         (_("This file contains no playable streams.")),
 501         ("no known streams found"));
 502   }
 503 }
 504
 505 static GstBuffer *
 506 _gst_buffer_new_wrapped (gpointer mem, gsize size, GFreeFunc free_func)
 507 {
 508   return gst_buffer_new_wrapped_full (free_func ? 0 : GST_MEMORY_FLAG_READONLY,
 509       mem, size, 0, size, mem, free_func);
 510 }
 511
 512 static GstFlowReturn
 513 gst_qtdemux_pull_atom (GstQTDemux * qtdemux, guint64 offset, guint64 size,
 514     GstBuffer ** buf)
 515 {
 516   GstFlowReturn flow;
 517   GstMapInfo map;
 518   gsize bsize;
 519
 520   if (G_UNLIKELY (size == 0)) {
 521     GstFlowReturn ret;
 522     GstBuffer *tmp = NULL;
 523
 524     ret = gst_qtdemux_pull_atom (qtdemux, offset, sizeof (guint32), &tmp);
 525     if (ret != GST_FLOW_OK)
 526       return ret;
 527
 528     gst_buffer_map (tmp, &map, GST_MAP_READ);
 529     size = QT_UINT32 (map.data);
 530     GST_DEBUG_OBJECT (qtdemux, "size 0x%08" G_GINT64_MODIFIER "x", size);
 531
 532     gst_buffer_unmap (tmp, &map);
 533     gst_buffer_unref (tmp);
 534   }
 535
 536   /* Sanity check: catch bogus sizes (fuzzed/broken files) */
 537   if (G_UNLIKELY (size > QTDEMUX_MAX_ATOM_SIZE)) {
 538     if (qtdemux->state != QTDEMUX_STATE_MOVIE && qtdemux->got_moov) {
 539       /* we're pulling header but already got most interesting bits,
 540        * so never mind the rest (e.g. tags) (that much) */
 541       GST_WARNING_OBJECT (qtdemux, "atom has bogus size %" G_GUINT64_FORMAT,
 542           size);
 543       return GST_FLOW_EOS;
 544     } else {
 545       GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
 546           (_("This file is invalid and cannot be played.")),
 547           ("atom has bogus size %" G_GUINT64_FORMAT, size));
 548       return GST_FLOW_ERROR;
 549     }
 550   }
 551
 552   flow = gst_pad_pull_range (qtdemux->sinkpad, offset, size, buf);
 553
 554   if (G_UNLIKELY (flow != GST_FLOW_OK))
 555     return flow;
 556
 557   bsize = gst_buffer_get_size (*buf);
 558   /* Catch short reads - we don't want any partial atoms */
 559   if (G_UNLIKELY (bsize < size)) {
 560     GST_WARNING_OBJECT (qtdemux,
 561         "short read: %" G_GSIZE_FORMAT " < %" G_GUINT64_FORMAT, bsize, size);
 562     gst_buffer_unref (*buf);
 563     *buf = NULL;
 564     return GST_FLOW_EOS;
 565   }
 566
 567   return flow;
 568 }
 569
 570 #if 1
 571 static gboolean
 572 gst_qtdemux_src_convert (GstQTDemux * qtdemux, GstPad * pad,
 573     GstFormat src_format, gint64 src_value, GstFormat dest_format,
 574     gint64 * dest_value)
 575 {
 576   gboolean res = TRUE;
 577   QtDemuxStream *stream = gst_pad_get_element_private (pad);
 578   gint32 index;
 579
 580   if (stream->subtype != FOURCC_vide) {
 581     res = FALSE;
 582     goto done;
 583   }
 584
 585   switch (src_format) {
 586     case GST_FORMAT_TIME:
 587       switch (dest_format) {
 588         case GST_FORMAT_BYTES:{
 589           index = gst_qtdemux_find_index_linear (qtdemux, stream, src_value);
 590           if (-1 == index) {
 591             res = FALSE;
 592             goto done;
 593           }
 594
 595           *dest_value = stream->samples[index].offset;
 596
 597           GST_DEBUG_OBJECT (qtdemux, "Format Conversion Time->Offset :%"
 598               GST_TIME_FORMAT "->%" G_GUINT64_FORMAT,
 599               GST_TIME_ARGS (src_value), *dest_value);
 600           break;
 601         }
 602         default:
 603           res = FALSE;
 604           break;
 605       }
 606       break;
 607     case GST_FORMAT_BYTES:
 608       switch (dest_format) {
 609         case GST_FORMAT_TIME:{
 610           index =
 611               gst_qtdemux_find_index_for_given_media_offset_linear (qtdemux,
 612               stream, src_value);
 613
 614           if (-1 == index) {
 615             res = FALSE;
 616             goto done;
 617           }
 618
 619           *dest_value =
 620               QTSTREAMTIME_TO_GSTTIME (stream,
 621               stream->samples[index].timestamp);
 622           GST_DEBUG_OBJECT (qtdemux,
 623               "Format Conversion Offset->Time :%" G_GUINT64_FORMAT "->%"
 624               GST_TIME_FORMAT, src_value, GST_TIME_ARGS (*dest_value));
 625           break;
 626         }
 627         default:
 628           res = FALSE;
 629           break;
 630       }
 631       break;
 632     default:
 633       res = FALSE;
 634       break;
 635   }
 636
 637 done:
 638   return res;
 639 }
 640 #endif
 641
 642 static gboolean
 643 gst_qtdemux_get_duration (GstQTDemux * qtdemux, GstClockTime * duration)
 644 {
 645   gboolean res = FALSE;
 646
 647   *duration = GST_CLOCK_TIME_NONE;
 648
 649   if (qtdemux->duration != 0 &&
 650       qtdemux->duration != G_MAXINT64 && qtdemux->timescale != 0) {
 651     *duration = QTTIME_TO_GSTTIME (qtdemux, qtdemux->duration);
 652     res = TRUE;
 653   } else {
 654     *duration = GST_CLOCK_TIME_NONE;
 655   }
 656
 657   return res;
 658 }
 659
 660 static gboolean
 661 gst_qtdemux_handle_src_query (GstPad * pad, GstObject * parent,
 662     GstQuery * query)
 663 {
 664   gboolean res = FALSE;
 665   GstQTDemux *qtdemux = GST_QTDEMUX (parent);
 666
 667   GST_LOG_OBJECT (pad, "%s query", GST_QUERY_TYPE_NAME (query));
 668
 669   switch (GST_QUERY_TYPE (query)) {
 670     case GST_QUERY_POSITION:{
 671       GstFormat fmt;
 672
 673       gst_query_parse_position (query, &fmt, NULL);
 674       if (fmt == GST_FORMAT_TIME
 675           && GST_CLOCK_TIME_IS_VALID (qtdemux->segment.position)) {
 676         gst_query_set_position (query, GST_FORMAT_TIME,
 677             qtdemux->segment.position);
 678         res = TRUE;
 679       }
 680     }
 681       break;
 682     case GST_QUERY_DURATION:{
 683       GstFormat fmt;
 684
 685       gst_query_parse_duration (query, &fmt, NULL);
 686       if (fmt == GST_FORMAT_TIME) {
 687         /* First try to query upstream */
 688         res = gst_pad_query_default (pad, parent, query);
 689         if (!res) {
 690           GstClockTime duration;
 691           if (gst_qtdemux_get_duration (qtdemux, &duration) && duration > 0) {
 692             gst_query_set_duration (query, GST_FORMAT_TIME, duration);
 693             res = TRUE;
 694           }
 695         }
 696       }
 697       break;
 698     }
 699     case GST_QUERY_CONVERT:{
 700       GstFormat src_fmt, dest_fmt;
 701       gint64 src_value, dest_value = 0;
 702
 703       gst_query_parse_convert (query, &src_fmt, &src_value, &dest_fmt, NULL);
 704
 705       res = gst_qtdemux_src_convert (qtdemux, pad,
 706           src_fmt, src_value, dest_fmt, &dest_value);
 707       if (res)
 708         gst_query_set_convert (query, src_fmt, src_value, dest_fmt, dest_value);
 709
 710       break;
 711     }
 712     case GST_QUERY_FORMATS:
 713       gst_query_set_formats (query, 2, GST_FORMAT_TIME, GST_FORMAT_BYTES);
 714       res = TRUE;
 715       break;
 716     case GST_QUERY_SEEKING:{
 717       GstFormat fmt;
 718       gboolean seekable;
 719
 720       gst_query_parse_seeking (query, &fmt, NULL, NULL, NULL);
 721
 722       if (fmt == GST_FORMAT_BYTES) {
 723         /* We always refuse BYTES seeks from downstream */
 724         break;
 725       }
 726
 727       /* try upstream first */
 728       res = gst_pad_query_default (pad, parent, query);
 729
 730       if (!res) {
 731         gst_query_parse_seeking (query, &fmt, NULL, NULL, NULL);
 732         if (fmt == GST_FORMAT_TIME) {
 733           GstClockTime duration;
 734
 735           gst_qtdemux_get_duration (qtdemux, &duration);
 736           seekable = TRUE;
 737           if (!qtdemux->pullbased) {
 738             GstQuery *q;
 739
 740             /* we might be able with help from upstream */
 741             seekable = FALSE;
 742             q = gst_query_new_seeking (GST_FORMAT_BYTES);
 743             if (gst_pad_peer_query (qtdemux->sinkpad, q)) {
 744               gst_query_parse_seeking (q, &fmt, &seekable, NULL, NULL);
 745               GST_LOG_OBJECT (qtdemux, "upstream BYTE seekable %d", seekable);
 746             }
 747             gst_query_unref (q);
 748           }
 749           gst_query_set_seeking (query, GST_FORMAT_TIME, seekable, 0, duration);
 750           res = TRUE;
 751         }
 752       }
 753       break;
 754     }
 755     case GST_QUERY_SEGMENT:
 756     {
 757       GstFormat format;
 758       gint64 start, stop;
 759
 760       format = qtdemux->segment.format;
 761
 762       start =
 763           gst_segment_to_stream_time (&qtdemux->segment, format,
 764           qtdemux->segment.start);
 765       if ((stop = qtdemux->segment.stop) == -1)
 766         stop = qtdemux->segment.duration;
 767       else
 768         stop = gst_segment_to_stream_time (&qtdemux->segment, format, stop);
 769
 770       gst_query_set_segment (query, qtdemux->segment.rate, format, start, stop);
 771       res = TRUE;
 772       break;
 773     }
 774     default:
 775       res = gst_pad_query_default (pad, parent, query);
 776       break;
 777   }
 778
 779   return res;
 780 }
 781
 782 static void
 783 gst_qtdemux_push_tags (GstQTDemux * qtdemux, QtDemuxStream * stream)
 784 {
 785   if (G_LIKELY (stream->pad)) {
 786     GST_DEBUG_OBJECT (qtdemux, "Checking pad %s:%s for tags",
 787         GST_DEBUG_PAD_NAME (stream->pad));
 788
 789     if (!gst_tag_list_is_empty (stream->stream_tags)) {
 790       GST_DEBUG_OBJECT (qtdemux, "Sending tags %" GST_PTR_FORMAT,
 791           stream->stream_tags);
 792       gst_pad_push_event (stream->pad,
 793           gst_event_new_tag (gst_tag_list_ref (stream->stream_tags)));
 794     }
 795
 796     if (G_UNLIKELY (stream->send_global_tags)) {
 797       GST_DEBUG_OBJECT (qtdemux, "Sending global tags %" GST_PTR_FORMAT,
 798           qtdemux->tag_list);
 799       gst_pad_push_event (stream->pad,
 800           gst_event_new_tag (gst_tag_list_ref (qtdemux->tag_list)));
 801       stream->send_global_tags = FALSE;
 802     }
 803   }
 804 }
 805
 806 /* push event on all source pads; takes ownership of the event */
 807 static void
 808 gst_qtdemux_push_event (GstQTDemux * qtdemux, GstEvent * event)
 809 {
 810   gboolean has_valid_stream = FALSE;
 811   GstEventType etype = GST_EVENT_TYPE (event);
 812   guint i;
 813
 814   GST_DEBUG_OBJECT (qtdemux, "pushing %s event on all source pads",
 815       GST_EVENT_TYPE_NAME (event));
 816
 817   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
 818     GstPad *pad;
 819     QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
 820     GST_DEBUG_OBJECT (qtdemux, "pushing on track-id %u", stream->track_id);
 821
 822     if ((pad = stream->pad)) {
 823       has_valid_stream = TRUE;
 824
 825       if (etype == GST_EVENT_EOS) {
 826         /* let's not send twice */
 827         if (stream->sent_eos)
 828           continue;
 829         stream->sent_eos = TRUE;
 830       }
 831
 832       gst_pad_push_event (pad, gst_event_ref (event));
 833     }
 834   }
 835
 836   gst_event_unref (event);
 837
 838   /* if it is EOS and there are no pads, post an error */
 839   if (!has_valid_stream && etype == GST_EVENT_EOS) {
 840     gst_qtdemux_post_no_playable_stream_error (qtdemux);
 841   }
 842 }
 843
 844 typedef struct
 845 {
 846   guint64 media_time;
 847 } FindData;
 848
 849 static gint
 850 find_func (QtDemuxSample * s1, gint64 * media_time, gpointer user_data)
 851 {
 852   if ((gint64) s1->timestamp > *media_time)
 853     return 1;
 854   if ((gint64) s1->timestamp == *media_time)
 855     return 0;
 856
 857   return -1;
 858 }
 859
 860 /* find the index of the sample that includes the data for @media_time using a
 861  * binary search.  Only to be called in optimized cases of linear search below.
 862  *
 863  * Returns the index of the sample with the corresponding *DTS*.
 864  */
 865 static guint32
 866 gst_qtdemux_find_index (GstQTDemux * qtdemux, QtDemuxStream * str,
 867     guint64 media_time)
 868 {
 869   QtDemuxSample *result;
 870   guint32 index;
 871
 872   /* convert media_time to mov format */
 873   media_time =
 874       gst_util_uint64_scale_ceil (media_time, str->timescale, GST_SECOND);
 875
 876   result = gst_util_array_binary_search (str->samples, str->stbl_index + 1,
 877       sizeof (QtDemuxSample), (GCompareDataFunc) find_func,
 878       GST_SEARCH_MODE_BEFORE, &media_time, NULL);
 879
 880   if (G_LIKELY (result))
 881     index = result - str->samples;
 882   else
 883     index = 0;
 884
 885   return index;
 886 }
 887
 888
 889
 890 /* find the index of the sample that includes the data for @media_offset using a
 891  * linear search
 892  *
 893  * Returns the index of the sample.
 894  */
 895 static guint32
 896 gst_qtdemux_find_index_for_given_media_offset_linear (GstQTDemux * qtdemux,
 897     QtDemuxStream * str, gint64 media_offset)
 898 {
 899   QtDemuxSample *result = str->samples;
 900   guint32 index = 0;
 901
 902   if (result == NULL || str->n_samples == 0)
 903     return -1;
 904
 905   if (media_offset == result->offset)
 906     return index;
 907
 908   result++;
 909   while (index < str->n_samples - 1) {
 910     if (!qtdemux_parse_samples (qtdemux, str, index + 1))
 911       goto parse_failed;
 912
 913     if (media_offset < result->offset)
 914       break;
 915
 916     index++;
 917     result++;
 918   }
 919   return index;
 920
 921   /* ERRORS */
 922 parse_failed:
 923   {
 924     GST_LOG_OBJECT (qtdemux, "Parsing of index %u failed!", index + 1);
 925     return -1;
 926   }
 927 }
 928
 929 /* find the index of the sample that includes the data for @media_time using a
 930  * linear search, and keeping in mind that not all samples may have been parsed
 931  * yet.  If possible, it will delegate to binary search.
 932  *
 933  * Returns the index of the sample.
 934  */
 935 static guint32
 936 gst_qtdemux_find_index_linear (GstQTDemux * qtdemux, QtDemuxStream * str,
 937     GstClockTime media_time)
 938 {
 939   guint32 index = 0;
 940   guint64 mov_time;
 941   QtDemuxSample *sample;
 942
 943   /* convert media_time to mov format */
 944   mov_time =
 945       gst_util_uint64_scale_ceil (media_time, str->timescale, GST_SECOND);
 946
 947   sample = str->samples;
 948   if (mov_time == sample->timestamp + sample->pts_offset)
 949     return index;
 950
 951   /* use faster search if requested time in already parsed range */
 952   sample = str->samples + str->stbl_index;
 953   if (str->stbl_index >= 0 && mov_time <= sample->timestamp) {
 954     index = gst_qtdemux_find_index (qtdemux, str, media_time);
 955     sample = str->samples + index;
 956   } else {
 957     while (index < str->n_samples - 1) {
 958       if (!qtdemux_parse_samples (qtdemux, str, index + 1))
 959         goto parse_failed;
 960
 961       sample = str->samples + index + 1;
 962       if (mov_time < sample->timestamp) {
 963         sample = str->samples + index;
 964         break;
 965       }
 966
 967       index++;
 968     }
 969   }
 970
 971   /* sample->timestamp is now <= media_time, need to find the corresponding
 972    * PTS now by looking backwards */
 973   while (index > 0 && sample->timestamp + sample->pts_offset > mov_time) {
 974     index--;
 975     sample = str->samples + index;
 976   }
 977
 978   return index;
 979
 980   /* ERRORS */
 981 parse_failed:
 982   {
 983     GST_LOG_OBJECT (qtdemux, "Parsing of index %u failed!", index + 1);
 984     return -1;
 985   }
 986 }
 987
 988 /* find the index of the keyframe needed to decode the sample at @index
 989  * of stream @str, or of a subsequent keyframe (depending on @next)
 990  *
 991  * Returns the index of the keyframe.
 992  */
 993 static guint32
 994 gst_qtdemux_find_keyframe (GstQTDemux * qtdemux, QtDemuxStream * str,
 995     guint32 index, gboolean next)
 996 {
 997   guint32 new_index = index;
 998
 999   if (index >= str->n_samples) {
1000     new_index = str->n_samples;
1001     goto beach;
1002   }
1003
1004   /* all keyframes, return index */
1005   if (str->all_keyframe) {
1006     new_index = index;
1007     goto beach;
1008   }
1009
1010   /* else search until we have a keyframe */
1011   while (new_index < str->n_samples) {
1012     if (next && !qtdemux_parse_samples (qtdemux, str, new_index))
1013       goto parse_failed;
1014
1015     if (str->samples[new_index].keyframe)
1016       break;
1017
1018     if (new_index == 0)
1019       break;
1020
1021     if (next)
1022       new_index++;
1023     else
1024       new_index--;
1025   }
1026
1027   if (new_index == str->n_samples) {
1028     GST_DEBUG_OBJECT (qtdemux, "no next keyframe");
1029     new_index = -1;
1030   }
1031
1032 beach:
1033   GST_DEBUG_OBJECT (qtdemux, "searching for keyframe index %s index %u "
1034       "gave %u", next ? "after" : "before", index, new_index);
1035
1036   return new_index;
1037
1038   /* ERRORS */
1039 parse_failed:
1040   {
1041     GST_LOG_OBJECT (qtdemux, "Parsing of index %u failed!", new_index);
1042     return -1;
1043   }
1044 }
1045
1046 /* find the segment for @time_position for @stream
1047  *
1048  * Returns the index of the segment containing @time_position.
1049  * Returns the last segment and sets the @eos variable to TRUE
1050  * if the time is beyond the end. @eos may be NULL
1051  */
1052 static guint32
1053 gst_qtdemux_find_segment (GstQTDemux * qtdemux, QtDemuxStream * stream,
1054     GstClockTime time_position)
1055 {
1056   gint i;
1057   guint32 seg_idx;
1058
1059   GST_LOG_OBJECT (stream->pad, "finding segment for %" GST_TIME_FORMAT,
1060       GST_TIME_ARGS (time_position));
1061
1062   seg_idx = -1;
1063   for (i = 0; i < stream->n_segments; i++) {
1064     QtDemuxSegment *segment = &stream->segments[i];
1065
1066     GST_LOG_OBJECT (stream->pad,
1067         "looking at segment %" GST_TIME_FORMAT "-%" GST_TIME_FORMAT,
1068         GST_TIME_ARGS (segment->time), GST_TIME_ARGS (segment->stop_time));
1069
1070     /* For the last segment we include stop_time in the last segment */
1071     if (i < stream->n_segments - 1) {
1072       if (segment->time <= time_position && time_position < segment->stop_time) {
1073         GST_LOG_OBJECT (stream->pad, "segment %d matches", i);
1074         seg_idx = i;
1075         break;
1076       }
1077     } else {
1078       /* Last segment always matches */
1079       seg_idx = i;
1080       break;
1081     }
1082   }
1083   return seg_idx;
1084 }
1085
1086 /* move the stream @str to the sample position @index.
1087  *
1088  * Updates @str->sample_index and marks discontinuity if needed.
1089  */
1090 static void
1091 gst_qtdemux_move_stream (GstQTDemux * qtdemux, QtDemuxStream * str,
1092     guint32 index)
1093 {
1094   /* no change needed */
1095   if (index == str->sample_index)
1096     return;
1097
1098   GST_DEBUG_OBJECT (qtdemux, "moving to sample %u of %u", index,
1099       str->n_samples);
1100
1101   /* position changed, we have a discont */
1102   str->sample_index = index;
1103   str->offset_in_sample = 0;
1104   /* Each time we move in the stream we store the position where we are
1105    * starting from */
1106   str->from_sample = index;
1107   str->discont = TRUE;
1108 }
1109
1110 static void
1111 gst_qtdemux_adjust_seek (GstQTDemux * qtdemux, gint64 desired_time,
1112     gboolean use_sparse, gboolean next, gint64 * key_time, gint64 * key_offset)
1113 {
1114   guint64 min_offset;
1115   gint64 min_byte_offset = -1;
1116   guint i;
1117
1118   min_offset = desired_time;
1119
1120   /* for each stream, find the index of the sample in the segment
1121    * and move back to the previous keyframe. */
1122   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
1123     QtDemuxStream *str;
1124     guint32 index, kindex;
1125     guint32 seg_idx;
1126     GstClockTime media_start;
1127     GstClockTime media_time;
1128     GstClockTime seg_time;
1129     QtDemuxSegment *seg;
1130     gboolean empty_segment = FALSE;
1131
1132     str = QTDEMUX_NTH_STREAM (qtdemux, i);
1133
1134     if (CUR_STREAM (str)->sparse && !use_sparse)
1135       continue;
1136
1137     seg_idx = gst_qtdemux_find_segment (qtdemux, str, desired_time);
1138     GST_DEBUG_OBJECT (qtdemux, "align segment %d", seg_idx);
1139
1140     /* get segment and time in the segment */
1141     seg = &str->segments[seg_idx];
1142     seg_time = (desired_time - seg->time) * seg->rate;
1143
1144     while (QTSEGMENT_IS_EMPTY (seg)) {
1145       seg_time = 0;
1146       empty_segment = TRUE;
1147       GST_DEBUG_OBJECT (str->pad, "Segment %d is empty, moving to next one",
1148           seg_idx);
1149       seg_idx++;
1150       if (seg_idx == str->n_segments)
1151         break;
1152       seg = &str->segments[seg_idx];
1153     }
1154
1155     if (seg_idx == str->n_segments) {
1156       /* FIXME track shouldn't have the last segment as empty, but if it
1157        * happens we better handle it */
1158       continue;
1159     }
1160
1161     /* get the media time in the segment */
1162     media_start = seg->media_start + seg_time;
1163
1164     /* get the index of the sample with media time */
1165     index = gst_qtdemux_find_index_linear (qtdemux, str, media_start);
1166     GST_DEBUG_OBJECT (qtdemux, "sample for %" GST_TIME_FORMAT " at %u"
1167         " at offset %" G_GUINT64_FORMAT " (empty segment: %d)",
1168         GST_TIME_ARGS (media_start), index, str->samples[index].offset,
1169         empty_segment);
1170
1171     /* shift to next frame if we are looking for next keyframe */
1172     if (next && QTSAMPLE_PTS_NO_CSLG (str, &str->samples[index]) < media_start
1173         && index < str->stbl_index)
1174       index++;
1175
1176     if (!empty_segment) {
1177       /* find previous keyframe */
1178       kindex = gst_qtdemux_find_keyframe (qtdemux, str, index, next);
1179
1180       /* we will settle for one before if none found after */
1181       if (next && kindex == -1)
1182         kindex = gst_qtdemux_find_keyframe (qtdemux, str, index, FALSE);
1183
1184       /* Update the requested time whenever a keyframe was found, to make it
1185        * accurate and avoid having the first buffer fall outside of the segment
1186        */
1187       if (kindex != -1) {
1188         index = kindex;
1189
1190         /* get timestamp of keyframe */
1191         media_time = QTSAMPLE_PTS_NO_CSLG (str, &str->samples[kindex]);
1192         GST_DEBUG_OBJECT (qtdemux,
1193             "keyframe at %u with time %" GST_TIME_FORMAT " at offset %"
1194             G_GUINT64_FORMAT, kindex, GST_TIME_ARGS (media_time),
1195             str->samples[kindex].offset);
1196
1197         /* keyframes in the segment get a chance to change the
1198          * desired_offset. keyframes out of the segment are
1199          * ignored. */
1200         if (media_time >= seg->media_start) {
1201           GstClockTime seg_time;
1202
1203           /* this keyframe is inside the segment, convert back to
1204            * segment time */
1205           seg_time = (media_time - seg->media_start) + seg->time;
1206           if ((!next && (seg_time < min_offset)) ||
1207               (next && (seg_time > min_offset)))
1208             min_offset = seg_time;
1209         }
1210       }
1211     }
1212
1213     if (min_byte_offset < 0 || str->samples[index].offset < min_byte_offset)
1214       min_byte_offset = str->samples[index].offset;
1215   }
1216
1217   if (key_time)
1218     *key_time = min_offset;
1219   if (key_offset)
1220     *key_offset = min_byte_offset;
1221 }
1222
1223 static gboolean
1224 gst_qtdemux_convert_seek (GstPad * pad, GstFormat * format,
1225     GstSeekType cur_type, gint64 * cur, GstSeekType stop_type, gint64 * stop)
1226 {
1227   gboolean res;
1228
1229   g_return_val_if_fail (format != NULL, FALSE);
1230   g_return_val_if_fail (cur != NULL, FALSE);
1231   g_return_val_if_fail (stop != NULL, FALSE);
1232
1233   if (*format == GST_FORMAT_TIME)
1234     return TRUE;
1235
1236   res = TRUE;
1237   if (cur_type != GST_SEEK_TYPE_NONE)
1238     res = gst_pad_query_convert (pad, *format, *cur, GST_FORMAT_TIME, cur);
1239   if (res && stop_type != GST_SEEK_TYPE_NONE)
1240     res = gst_pad_query_convert (pad, *format, *stop, GST_FORMAT_TIME, stop);
1241
1242   if (res)
1243     *format = GST_FORMAT_TIME;
1244
1245   return res;
1246 }
1247
1248 /* perform seek in push based mode:
1249    find BYTE position to move to based on time and delegate to upstream
1250 */
1251 static gboolean
1252 gst_qtdemux_do_push_seek (GstQTDemux * qtdemux, GstPad * pad, GstEvent * event)
1253 {
1254   gdouble rate;
1255   GstFormat format;
1256   GstSeekFlags flags;
1257   GstSeekType cur_type, stop_type;
1258   gint64 cur, stop, key_cur;
1259   gboolean res;
1260   gint64 byte_cur;
1261   gint64 original_stop;
1262   guint32 seqnum;
1263
1264   GST_DEBUG_OBJECT (qtdemux, "doing push-based seek");
1265
1266   gst_event_parse_seek (event, &rate, &format, &flags,
1267       &cur_type, &cur, &stop_type, &stop);
1268   seqnum = gst_event_get_seqnum (event);
1269
1270   /* Directly send the instant-rate-change event here before taking the
1271    * stream-lock so that it can be applied as soon as possible */
1272   if (flags & GST_SEEK_FLAG_INSTANT_RATE_CHANGE) {
1273     GstEvent *ev;
1274
1275     /* instant rate change only supported if direction does not change. All
1276      * other requirements are already checked before creating the seek event
1277      * but let's double-check here to be sure */
1278     if ((qtdemux->segment.rate > 0 && rate < 0) ||
1279         (qtdemux->segment.rate < 0 && rate > 0) ||
1280         cur_type != GST_SEEK_TYPE_NONE ||
1281         stop_type != GST_SEEK_TYPE_NONE || (flags & GST_SEEK_FLAG_FLUSH)) {
1282       GST_ERROR_OBJECT (qtdemux,
1283           "Instant rate change seeks only supported in the "
1284           "same direction, without flushing and position change");
1285       return FALSE;
1286     }
1287
1288     ev = gst_event_new_instant_rate_change (rate / qtdemux->segment.rate,
1289         (GstSegmentFlags) flags);
1290     gst_event_set_seqnum (ev, seqnum);
1291     gst_qtdemux_push_event (qtdemux, ev);
1292     return TRUE;
1293   }
1294
1295   /* only forward streaming and seeking is possible */
1296   if (rate <= 0)
1297     goto unsupported_seek;
1298
1299   /* convert to TIME if needed and possible */
1300   if (!gst_qtdemux_convert_seek (pad, &format, cur_type, &cur,
1301           stop_type, &stop))
1302     goto no_format;
1303
1304   /* Upstream seek in bytes will have undefined stop, but qtdemux stores
1305    * the original stop position to use when upstream pushes the new segment
1306    * for this seek */
1307   original_stop = stop;
1308   stop = -1;
1309
1310   /* find reasonable corresponding BYTE position,
1311    * also try to mind about keyframes, since we can not go back a bit for them
1312    * later on */
1313   /* determining @next here based on SNAP_BEFORE/SNAP_AFTER should
1314    * mostly just work, but let's not yet boldly go there  ... */
1315   gst_qtdemux_adjust_seek (qtdemux, cur, FALSE, FALSE, &key_cur, &byte_cur);
1316
1317   if (byte_cur == -1)
1318     goto abort_seek;
1319
1320   GST_DEBUG_OBJECT (qtdemux, "Pushing BYTE seek rate %g, "
1321       "start %" G_GINT64_FORMAT ", stop %" G_GINT64_FORMAT, rate, byte_cur,
1322       stop);
1323
1324   GST_OBJECT_LOCK (qtdemux);
1325   qtdemux->seek_offset = byte_cur;
1326   if (!(flags & GST_SEEK_FLAG_KEY_UNIT)) {
1327     qtdemux->push_seek_start = cur;
1328   } else {
1329     qtdemux->push_seek_start = key_cur;
1330   }
1331
1332   if (stop_type == GST_SEEK_TYPE_NONE) {
1333     qtdemux->push_seek_stop = qtdemux->segment.stop;
1334   } else {
1335     qtdemux->push_seek_stop = original_stop;
1336   }
1337   GST_OBJECT_UNLOCK (qtdemux);
1338
1339   qtdemux->segment_seqnum = seqnum;
1340   /* BYTE seek event */
1341   event = gst_event_new_seek (rate, GST_FORMAT_BYTES, flags, cur_type, byte_cur,
1342       stop_type, stop);
1343   gst_event_set_seqnum (event, seqnum);
1344   res = gst_pad_push_event (qtdemux->sinkpad, event);
1345
1346   return res;
1347
1348   /* ERRORS */
1349 abort_seek:
1350   {
1351     GST_DEBUG_OBJECT (qtdemux, "could not determine byte position to seek to, "
1352         "seek aborted.");
1353     return FALSE;
1354   }
1355 unsupported_seek:
1356   {
1357     GST_DEBUG_OBJECT (qtdemux, "unsupported seek, seek aborted.");
1358     return FALSE;
1359   }
1360 no_format:
1361   {
1362     GST_DEBUG_OBJECT (qtdemux, "unsupported format given, seek aborted.");
1363     return FALSE;
1364   }
1365 }
1366
1367 /* perform the seek.
1368  *
1369  * We set all segment_indexes in the streams to unknown and
1370  * adjust the time_position to the desired position. this is enough
1371  * to trigger a segment switch in the streaming thread to start
1372  * streaming from the desired position.
1373  *
1374  * Keyframe seeking is a little more complicated when dealing with
1375  * segments. Ideally we want to move to the previous keyframe in
1376  * the segment but there might not be a keyframe in the segment. In
1377  * fact, none of the segments could contain a keyframe. We take a
1378  * practical approach: seek to the previous keyframe in the segment,
1379  * if there is none, seek to the beginning of the segment.
1380  *
1381  * Called with STREAM_LOCK
1382  */
1383 static gboolean
1384 gst_qtdemux_perform_seek (GstQTDemux * qtdemux, GstSegment * segment,
1385     guint32 seqnum, GstSeekFlags flags)
1386 {
1387   gint64 desired_offset;
1388   guint i;
1389
1390   desired_offset = segment->position;
1391
1392   GST_DEBUG_OBJECT (qtdemux, "seeking to %" GST_TIME_FORMAT,
1393       GST_TIME_ARGS (desired_offset));
1394
1395   /* may not have enough fragmented info to do this adjustment,
1396    * and we can't scan (and probably should not) at this time with
1397    * possibly flushing upstream */
1398   if ((flags & GST_SEEK_FLAG_KEY_UNIT) && !qtdemux->fragmented) {
1399     gint64 min_offset;
1400     gboolean next, before, after;
1401
1402     before = ! !(flags & GST_SEEK_FLAG_SNAP_BEFORE);
1403     after = ! !(flags & GST_SEEK_FLAG_SNAP_AFTER);
1404     next = after && !before;
1405     if (segment->rate < 0)
1406       next = !next;
1407
1408     gst_qtdemux_adjust_seek (qtdemux, desired_offset, TRUE, next, &min_offset,
1409         NULL);
1410     GST_DEBUG_OBJECT (qtdemux, "keyframe seek, align to %"
1411         GST_TIME_FORMAT, GST_TIME_ARGS (min_offset));
1412     desired_offset = min_offset;
1413   }
1414
1415   /* and set all streams to the final position */
1416   GST_OBJECT_LOCK (qtdemux);
1417   gst_flow_combiner_reset (qtdemux->flowcombiner);
1418   GST_OBJECT_UNLOCK (qtdemux);
1419   qtdemux->segment_seqnum = seqnum;
1420   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
1421     QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
1422
1423     stream->time_position = desired_offset;
1424     stream->accumulated_base = 0;
1425     stream->sample_index = -1;
1426     stream->offset_in_sample = 0;
1427     stream->segment_index = -1;
1428     stream->sent_eos = FALSE;
1429     stream->last_keyframe_dts = GST_CLOCK_TIME_NONE;
1430
1431     if (segment->flags & GST_SEEK_FLAG_FLUSH)
1432       gst_segment_init (&stream->segment, GST_FORMAT_TIME);
1433   }
1434   segment->position = desired_offset;
1435   if (segment->rate >= 0) {
1436     segment->start = desired_offset;
1437     /* We need to update time as we update start in that direction */
1438     segment->time = desired_offset;
1439
1440     /* we stop at the end */
1441     if (segment->stop == -1)
1442       segment->stop = segment->duration;
1443   } else {
1444     segment->stop = desired_offset;
1445   }
1446
1447   if (qtdemux->fragmented)
1448     qtdemux->fragmented_seek_pending = TRUE;
1449
1450   return TRUE;
1451 }
1452
1453 /* do a seek in pull based mode */
1454 static gboolean
1455 gst_qtdemux_do_seek (GstQTDemux * qtdemux, GstPad * pad, GstEvent * event)
1456 {
1457   gdouble rate = 1.0;
1458   GstFormat format;
1459   GstSeekFlags flags;
1460   GstSeekType cur_type, stop_type;
1461   gint64 cur, stop;
1462   gboolean flush, instant_rate_change;
1463   gboolean update;
1464   GstSegment seeksegment;
1465   guint32 seqnum = GST_SEQNUM_INVALID;
1466   GstEvent *flush_event;
1467   gboolean ret;
1468
1469   GST_DEBUG_OBJECT (qtdemux, "doing seek with event");
1470
1471   gst_event_parse_seek (event, &rate, &format, &flags,
1472       &cur_type, &cur, &stop_type, &stop);
1473   seqnum = gst_event_get_seqnum (event);
1474
1475   /* we have to have a format as the segment format. Try to convert
1476    * if not. */
1477   if (!gst_qtdemux_convert_seek (pad, &format, cur_type, &cur,
1478           stop_type, &stop))
1479     goto no_format;
1480
1481   GST_DEBUG_OBJECT (qtdemux, "seek format %s", gst_format_get_name (format));
1482
1483   flush = ! !(flags & GST_SEEK_FLAG_FLUSH);
1484   instant_rate_change = ! !(flags & GST_SEEK_FLAG_INSTANT_RATE_CHANGE);
1485
1486   /* Directly send the instant-rate-change event here before taking the
1487    * stream-lock so that it can be applied as soon as possible */
1488   if (instant_rate_change) {
1489     GstEvent *ev;
1490
1491     /* instant rate change only supported if direction does not change. All
1492      * other requirements are already checked before creating the seek event
1493      * but let's double-check here to be sure */
1494     if ((qtdemux->segment.rate > 0 && rate < 0) ||
1495         (qtdemux->segment.rate < 0 && rate > 0) ||
1496         cur_type != GST_SEEK_TYPE_NONE ||
1497         stop_type != GST_SEEK_TYPE_NONE || flush) {
1498       GST_ERROR_OBJECT (qtdemux,
1499           "Instant rate change seeks only supported in the "
1500           "same direction, without flushing and position change");
1501       return FALSE;
1502     }
1503
1504     ev = gst_event_new_instant_rate_change (rate / qtdemux->segment.rate,
1505         (GstSegmentFlags) flags);
1506     gst_event_set_seqnum (ev, seqnum);
1507     gst_qtdemux_push_event (qtdemux, ev);
1508     return TRUE;
1509   }
1510
1511   /* stop streaming, either by flushing or by pausing the task */
1512   if (flush) {
1513     flush_event = gst_event_new_flush_start ();
1514     if (seqnum != GST_SEQNUM_INVALID)
1515       gst_event_set_seqnum (flush_event, seqnum);
1516     /* unlock upstream pull_range */
1517     gst_pad_push_event (qtdemux->sinkpad, gst_event_ref (flush_event));
1518     /* make sure out loop function exits */
1519     gst_qtdemux_push_event (qtdemux, flush_event);
1520   } else {
1521     /* non flushing seek, pause the task */
1522     gst_pad_pause_task (qtdemux->sinkpad);
1523   }
1524
1525   /* wait for streaming to finish */
1526   GST_PAD_STREAM_LOCK (qtdemux->sinkpad);
1527
1528   /* copy segment, we need this because we still need the old
1529    * segment when we close the current segment. */
1530   memcpy (&seeksegment, &qtdemux->segment, sizeof (GstSegment));
1531
1532   /* configure the segment with the seek variables */
1533   GST_DEBUG_OBJECT (qtdemux, "configuring seek");
1534   if (!gst_segment_do_seek (&seeksegment, rate, format, flags,
1535           cur_type, cur, stop_type, stop, &update)) {
1536     ret = FALSE;
1537     GST_ERROR_OBJECT (qtdemux, "inconsistent seek values, doing nothing");
1538   } else {
1539     /* now do the seek */
1540     ret = gst_qtdemux_perform_seek (qtdemux, &seeksegment, seqnum, flags);
1541   }
1542
1543   /* prepare for streaming again */
1544   if (flush) {
1545     flush_event = gst_event_new_flush_stop (TRUE);
1546     if (seqnum != GST_SEQNUM_INVALID)
1547       gst_event_set_seqnum (flush_event, seqnum);
1548
1549     gst_pad_push_event (qtdemux->sinkpad, gst_event_ref (flush_event));
1550     gst_qtdemux_push_event (qtdemux, flush_event);
1551   }
1552
1553   /* commit the new segment */
1554   memcpy (&qtdemux->segment, &seeksegment, sizeof (GstSegment));
1555
1556   if (qtdemux->segment.flags & GST_SEEK_FLAG_SEGMENT) {
1557     GstMessage *msg = gst_message_new_segment_start (GST_OBJECT_CAST (qtdemux),
1558         qtdemux->segment.format, qtdemux->segment.position);
1559     if (seqnum != GST_SEQNUM_INVALID)
1560       gst_message_set_seqnum (msg, seqnum);
1561     gst_element_post_message (GST_ELEMENT_CAST (qtdemux), msg);
1562   }
1563
1564   /* restart streaming, NEWSEGMENT will be sent from the streaming thread. */
1565   gst_pad_start_task (qtdemux->sinkpad, (GstTaskFunction) gst_qtdemux_loop,
1566       qtdemux->sinkpad, NULL);
1567
1568   GST_PAD_STREAM_UNLOCK (qtdemux->sinkpad);
1569
1570   return ret;
1571
1572   /* ERRORS */
1573 no_format:
1574   {
1575     GST_DEBUG_OBJECT (qtdemux, "unsupported format given, seek aborted.");
1576     return FALSE;
1577   }
1578 }
1579
1580 static gboolean
1581 qtdemux_ensure_index (GstQTDemux * qtdemux)
1582 {
1583   guint i;
1584
1585   GST_DEBUG_OBJECT (qtdemux, "collecting all metadata for all streams");
1586
1587   /* Build complete index */
1588   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
1589     QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
1590
1591     if (!qtdemux_parse_samples (qtdemux, stream, stream->n_samples - 1)) {
1592       GST_LOG_OBJECT (qtdemux,
1593           "Building complete index of track-id %u for seeking failed!",
1594           stream->track_id);
1595       return FALSE;
1596     }
1597   }
1598
1599   return TRUE;
1600 }
1601
1602 static gboolean
1603 gst_qtdemux_handle_src_event (GstPad * pad, GstObject * parent,
1604     GstEvent * event)
1605 {
1606   gboolean res = TRUE;
1607   GstQTDemux *qtdemux = GST_QTDEMUX (parent);
1608
1609   switch (GST_EVENT_TYPE (event)) {
1610     case GST_EVENT_RECONFIGURE:
1611       GST_OBJECT_LOCK (qtdemux);
1612       gst_flow_combiner_reset (qtdemux->flowcombiner);
1613       GST_OBJECT_UNLOCK (qtdemux);
1614       res = gst_pad_event_default (pad, parent, event);
1615       break;
1616     case GST_EVENT_SEEK:
1617     {
1618       GstSeekFlags flags = 0;
1619       GstFormat seek_format;
1620       gboolean instant_rate_change;
1621
1622 #ifndef GST_DISABLE_GST_DEBUG
1623       GstClockTime ts = gst_util_get_timestamp ();
1624 #endif
1625       guint32 seqnum = gst_event_get_seqnum (event);
1626
1627       qtdemux->received_seek = TRUE;
1628
1629       gst_event_parse_seek (event, NULL, &seek_format, &flags, NULL, NULL, NULL,
1630           NULL);
1631       instant_rate_change = ! !(flags & GST_SEEK_FLAG_INSTANT_RATE_CHANGE);
1632
1633       if (seqnum == qtdemux->segment_seqnum) {
1634         GST_LOG_OBJECT (pad,
1635             "Drop duplicated SEEK event seqnum %" G_GUINT32_FORMAT, seqnum);
1636         gst_event_unref (event);
1637         return TRUE;
1638       }
1639
1640       if (qtdemux->upstream_format_is_time && qtdemux->fragmented) {
1641         /* seek should be handled by upstream, we might need to re-download fragments */
1642         GST_DEBUG_OBJECT (qtdemux,
1643             "let upstream handle seek for fragmented playback");
1644         goto upstream;
1645       }
1646
1647       if (seek_format == GST_FORMAT_BYTES) {
1648         GST_DEBUG_OBJECT (pad, "Rejecting seek request in bytes format");
1649         gst_event_unref (event);
1650         return FALSE;
1651       }
1652
1653       gst_event_parse_seek_trickmode_interval (event,
1654           &qtdemux->trickmode_interval);
1655
1656       /* Build complete index for seeking;
1657        * if not a fragmented file at least and we're really doing a seek,
1658        * not just an instant-rate-change */
1659       if (!qtdemux->fragmented && !instant_rate_change) {
1660         if (!qtdemux_ensure_index (qtdemux))
1661           goto index_failed;
1662       }
1663 #ifndef GST_DISABLE_GST_DEBUG
1664       ts = gst_util_get_timestamp () - ts;
1665       GST_INFO_OBJECT (qtdemux,
1666           "Time taken to parse index %" GST_TIME_FORMAT, GST_TIME_ARGS (ts));
1667 #endif
1668       if (qtdemux->pullbased) {
1669         res = gst_qtdemux_do_seek (qtdemux, pad, event);
1670       } else if (gst_pad_push_event (qtdemux->sinkpad, gst_event_ref (event))) {
1671         GST_DEBUG_OBJECT (qtdemux, "Upstream successfully seeked");
1672         res = TRUE;
1673       } else if (qtdemux->state == QTDEMUX_STATE_MOVIE
1674           && QTDEMUX_N_STREAMS (qtdemux)
1675           && !qtdemux->fragmented) {
1676         res = gst_qtdemux_do_push_seek (qtdemux, pad, event);
1677       } else {
1678         GST_DEBUG_OBJECT (qtdemux,
1679             "ignoring seek in push mode in current state");
1680         res = FALSE;
1681       }
1682       gst_event_unref (event);
1683     }
1684       break;
1685     default:
1686     upstream:
1687       res = gst_pad_event_default (pad, parent, event);
1688       break;
1689   }
1690
1691 done:
1692   return res;
1693
1694   /* ERRORS */
1695 index_failed:
1696   {
1697     GST_ERROR_OBJECT (qtdemux, "Index failed");
1698     gst_event_unref (event);
1699     res = FALSE;
1700     goto done;
1701   }
1702 }
1703
1704 /* Find, for each track, the first sample in coding order that has a file offset >= @byte_pos.
1705  *
1706  * If @fw is false, the coding order is explored backwards.
1707  *
1708  * If @set is true, each stream will be moved to its matched sample, or EOS if no matching
1709  * sample is found for that track.
1710  *
1711  * The stream and sample index of the sample with the minimum offset in the direction explored
1712  * (see @fw) is returned in the output parameters @_stream and @_index respectively.
1713  *
1714  * @_time is set to the QTSAMPLE_PTS of the matched sample with the minimum QTSAMPLE_PTS in the
1715  * direction explored, which may not always match the QTSAMPLE_PTS of the sample returned in
1716  * @_stream and @_index. */
1717 static void
1718 gst_qtdemux_find_sample (GstQTDemux * qtdemux, gint64 byte_pos, gboolean fw,
1719     gboolean set, QtDemuxStream ** _stream, gint * _index, gint64 * _time)
1720 {
1721   gint i, index;
1722   gint64 time, min_time;
1723   QtDemuxStream *stream;
1724   gint iter;
1725
1726   min_time = -1;
1727   stream = NULL;
1728   index = -1;
1729
1730   for (iter = 0; iter < QTDEMUX_N_STREAMS (qtdemux); iter++) {
1731     QtDemuxStream *str;
1732     gint inc;
1733     gboolean set_sample;
1734
1735     str = QTDEMUX_NTH_STREAM (qtdemux, iter);
1736     set_sample = !set;
1737
1738     if (fw) {
1739       i = 0;
1740       inc = 1;
1741     } else {
1742       i = str->n_samples - 1;
1743       inc = -1;
1744     }
1745
1746     for (; (i >= 0) && (i < str->n_samples); i += inc) {
1747       if (str->samples[i].size == 0)
1748         continue;
1749
1750       if (fw && (str->samples[i].offset < byte_pos))
1751         continue;
1752
1753       if (!fw && (str->samples[i].offset + str->samples[i].size > byte_pos))
1754         continue;
1755
1756       /* move stream to first available sample */
1757       if (set) {
1758         gst_qtdemux_move_stream (qtdemux, str, i);
1759         set_sample = TRUE;
1760       }
1761
1762       /* avoid index from sparse streams since they might be far away */
1763       if (!CUR_STREAM (str)->sparse) {
1764         /* determine min/max time */
1765         time = QTSAMPLE_PTS (str, &str->samples[i]);
1766         if (min_time == -1 || (!fw && time > min_time) ||
1767             (fw && time < min_time)) {
1768           min_time = time;
1769         }
1770
1771         /* determine stream with leading sample, to get its position */
1772         if (!stream ||
1773             (fw && (str->samples[i].offset < stream->samples[index].offset)) ||
1774             (!fw && (str->samples[i].offset > stream->samples[index].offset))) {
1775           stream = str;
1776           index = i;
1777         }
1778       }
1779       break;
1780     }
1781
1782     /* no sample for this stream, mark eos */
1783     if (!set_sample)
1784       gst_qtdemux_move_stream (qtdemux, str, str->n_samples);
1785   }
1786
1787   if (_time)
1788     *_time = min_time;
1789   if (_stream)
1790     *_stream = stream;
1791   if (_index)
1792     *_index = index;
1793 }
1794
1795 /* Copied from mpegtsbase code */
1796 /* FIXME: replace this function when we add new util function for stream-id creation */
1797 static gchar *
1798 _get_upstream_id (GstQTDemux * demux)
1799 {
1800   gchar *upstream_id = gst_pad_get_stream_id (demux->sinkpad);
1801
1802   if (!upstream_id) {
1803     /* Try to create one from the upstream URI, else use a randome number */
1804     GstQuery *query;
1805     gchar *uri = NULL;
1806
1807     /* Try to generate one from the URI query and
1808      * if it fails take a random number instead */
1809     query = gst_query_new_uri ();
1810     if (gst_element_query (GST_ELEMENT_CAST (demux), query)) {
1811       gst_query_parse_uri (query, &uri);
1812     }
1813
1814     if (uri) {
1815       GChecksum *cs;
1816
1817       /* And then generate an SHA256 sum of the URI */
1818       cs = g_checksum_new (G_CHECKSUM_SHA256);
1819       g_checksum_update (cs, (const guchar *) uri, strlen (uri));
1820       g_free (uri);
1821       upstream_id = g_strdup (g_checksum_get_string (cs));
1822       g_checksum_free (cs);
1823     } else {
1824       /* Just get some random number if the URI query fails */
1825       GST_FIXME_OBJECT (demux, "Creating random stream-id, consider "
1826           "implementing a deterministic way of creating a stream-id");
1827       upstream_id =
1828           g_strdup_printf ("%08x%08x%08x%08x", g_random_int (), g_random_int (),
1829           g_random_int (), g_random_int ());
1830     }
1831
1832     gst_query_unref (query);
1833   }
1834   return upstream_id;
1835 }
1836
1837 static QtDemuxStream *
1838 _create_stream (GstQTDemux * demux, guint32 track_id)
1839 {
1840   QtDemuxStream *stream;
1841   gchar *upstream_id;
1842
1843   stream = g_new0 (QtDemuxStream, 1);
1844   stream->demux = demux;
1845   stream->track_id = track_id;
1846   upstream_id = _get_upstream_id (demux);
1847   stream->stream_id = g_strdup_printf ("%s/%03u", upstream_id, track_id);
1848   g_free (upstream_id);
1849   /* new streams always need a discont */
1850   stream->discont = TRUE;
1851   /* we enable clipping for raw audio/video streams */
1852   stream->need_clip = FALSE;
1853   stream->process_func = NULL;
1854   stream->segment_index = -1;
1855   stream->time_position = 0;
1856   stream->sample_index = -1;
1857   stream->offset_in_sample = 0;
1858   stream->new_stream = TRUE;
1859   stream->multiview_mode = GST_VIDEO_MULTIVIEW_MODE_NONE;
1860   stream->multiview_flags = GST_VIDEO_MULTIVIEW_FLAGS_NONE;
1861   stream->protected = FALSE;
1862   stream->protection_scheme_type = 0;
1863   stream->protection_scheme_version = 0;
1864   stream->protection_scheme_info = NULL;
1865   stream->n_samples_moof = 0;
1866   stream->duration_moof = 0;
1867   stream->duration_last_moof = 0;
1868   stream->alignment = 1;
1869   stream->stream_tags = gst_tag_list_new_empty ();
1870   gst_tag_list_set_scope (stream->stream_tags, GST_TAG_SCOPE_STREAM);
1871   g_queue_init (&stream->protection_scheme_event_queue);
1872   stream->ref_count = 1;
1873   /* consistent default for push based mode */
1874   gst_segment_init (&stream->segment, GST_FORMAT_TIME);
1875   return stream;
1876 }
1877
1878 static gboolean
1879 gst_qtdemux_setcaps (GstQTDemux * demux, GstCaps * caps)
1880 {
1881   GstStructure *structure;
1882   const gchar *variant;
1883   const GstCaps *mediacaps = NULL;
1884
1885   GST_DEBUG_OBJECT (demux, "Sink set caps: %" GST_PTR_FORMAT, caps);
1886
1887   structure = gst_caps_get_structure (caps, 0);
1888   variant = gst_structure_get_string (structure, "variant");
1889
1890   if (variant && strcmp (variant, "mss-fragmented") == 0) {
1891     QtDemuxStream *stream;
1892     const GValue *value;
1893
1894     demux->fragmented = TRUE;
1895     demux->mss_mode = TRUE;
1896
1897     if (QTDEMUX_N_STREAMS (demux) > 1) {
1898       /* can't do this, we can only renegotiate for another mss format */
1899       return FALSE;
1900     }
1901
1902     value = gst_structure_get_value (structure, "media-caps");
1903     /* create stream */
1904     if (value) {
1905       const GValue *timescale_v;
1906
1907       /* TODO update when stream changes during playback */
1908
1909       if (QTDEMUX_N_STREAMS (demux) == 0) {
1910         stream = _create_stream (demux, 1);
1911         g_ptr_array_add (demux->active_streams, stream);
1912         /* mss has no stsd/stsd entry, use id 0 as default */
1913         stream->stsd_entries_length = 1;
1914         stream->stsd_sample_description_id = stream->cur_stsd_entry_index = 0;
1915         stream->stsd_entries = g_new0 (QtDemuxStreamStsdEntry, 1);
1916       } else {
1917         stream = QTDEMUX_NTH_STREAM (demux, 0);
1918       }
1919
1920       timescale_v = gst_structure_get_value (structure, "timescale");
1921       if (timescale_v) {
1922         stream->timescale = g_value_get_uint64 (timescale_v);
1923       } else {
1924         /* default mss timescale */
1925         stream->timescale = 10000000;
1926       }
1927       demux->timescale = stream->timescale;
1928
1929       mediacaps = gst_value_get_caps (value);
1930       if (!CUR_STREAM (stream)->caps
1931           || !gst_caps_is_equal_fixed (mediacaps, CUR_STREAM (stream)->caps)) {
1932         GST_DEBUG_OBJECT (demux, "We have a new caps %" GST_PTR_FORMAT,
1933             mediacaps);
1934         stream->new_caps = TRUE;
1935       }
1936       gst_caps_replace (&CUR_STREAM (stream)->caps, (GstCaps *) mediacaps);
1937       structure = gst_caps_get_structure (mediacaps, 0);
1938       if (g_str_has_prefix (gst_structure_get_name (structure), "video")) {
1939         stream->subtype = FOURCC_vide;
1940
1941         gst_structure_get_int (structure, "width", &CUR_STREAM (stream)->width);
1942         gst_structure_get_int (structure, "height",
1943             &CUR_STREAM (stream)->height);
1944         gst_structure_get_fraction (structure, "framerate",
1945             &CUR_STREAM (stream)->fps_n, &CUR_STREAM (stream)->fps_d);
1946       } else if (g_str_has_prefix (gst_structure_get_name (structure), "audio")) {
1947         gint rate = 0;
1948         stream->subtype = FOURCC_soun;
1949         gst_structure_get_int (structure, "channels",
1950             &CUR_STREAM (stream)->n_channels);
1951         gst_structure_get_int (structure, "rate", &rate);
1952         CUR_STREAM (stream)->rate = rate;
1953       } else if (gst_structure_has_name (structure, "application/x-cenc")) {
1954         if (gst_structure_has_field (structure, "original-media-type")) {
1955           const gchar *media_type =
1956               gst_structure_get_string (structure, "original-media-type");
1957           if (g_str_has_prefix (media_type, "video")) {
1958             stream->subtype = FOURCC_vide;
1959           } else if (g_str_has_prefix (media_type, "audio")) {
1960             stream->subtype = FOURCC_soun;
1961           }
1962         }
1963       }
1964     }
1965     gst_caps_replace (&demux->media_caps, (GstCaps *) mediacaps);
1966   } else {
1967     demux->mss_mode = FALSE;
1968   }
1969
1970   return TRUE;
1971 }
1972
1973 static void
1974 gst_qtdemux_reset (GstQTDemux * qtdemux, gboolean hard)
1975 {
1976   gint i;
1977
1978   GST_DEBUG_OBJECT (qtdemux, "Resetting demux");
1979   gst_pad_stop_task (qtdemux->sinkpad);
1980
1981   if (hard || qtdemux->upstream_format_is_time) {
1982     qtdemux->state = QTDEMUX_STATE_INITIAL;
1983     qtdemux->neededbytes = 16;
1984     qtdemux->todrop = 0;
1985     qtdemux->pullbased = FALSE;
1986     g_clear_pointer (&qtdemux->redirect_location, g_free);
1987     qtdemux->first_mdat = -1;
1988     qtdemux->header_size = 0;
1989     qtdemux->mdatoffset = -1;
1990     qtdemux->restoredata_offset = -1;
1991     if (qtdemux->mdatbuffer)
1992       gst_buffer_unref (qtdemux->mdatbuffer);
1993     if (qtdemux->restoredata_buffer)
1994       gst_buffer_unref (qtdemux->restoredata_buffer);
1995     qtdemux->mdatbuffer = NULL;
1996     qtdemux->restoredata_buffer = NULL;
1997     qtdemux->mdatleft = 0;
1998     qtdemux->mdatsize = 0;
1999     if (qtdemux->comp_brands)
2000       gst_buffer_unref (qtdemux->comp_brands);
2001     qtdemux->comp_brands = NULL;
2002     qtdemux->last_moov_offset = -1;
2003     if (qtdemux->moov_node_compressed) {
2004       g_node_destroy (qtdemux->moov_node_compressed);
2005       if (qtdemux->moov_node)
2006         g_free (qtdemux->moov_node->data);
2007     }
2008     qtdemux->moov_node_compressed = NULL;
2009     if (qtdemux->moov_node)
2010       g_node_destroy (qtdemux->moov_node);
2011     qtdemux->moov_node = NULL;
2012     if (qtdemux->tag_list)
2013       gst_mini_object_unref (GST_MINI_OBJECT_CAST (qtdemux->tag_list));
2014     qtdemux->tag_list = gst_tag_list_new_empty ();
2015     gst_tag_list_set_scope (qtdemux->tag_list, GST_TAG_SCOPE_GLOBAL);
2016 #if 0
2017     if (qtdemux->element_index)
2018       gst_object_unref (qtdemux->element_index);
2019     qtdemux->element_index = NULL;
2020 #endif
2021     qtdemux->major_brand = 0;
2022     qtdemux->upstream_format_is_time = FALSE;
2023     qtdemux->upstream_seekable = FALSE;
2024     qtdemux->upstream_size = 0;
2025
2026     qtdemux->fragment_start = -1;
2027     qtdemux->fragment_start_offset = -1;
2028     qtdemux->duration = 0;
2029     qtdemux->moof_offset = 0;
2030     qtdemux->chapters_track_id = 0;
2031     qtdemux->have_group_id = FALSE;
2032     qtdemux->group_id = G_MAXUINT;
2033
2034     g_queue_foreach (&qtdemux->protection_event_queue, (GFunc) gst_event_unref,
2035         NULL);
2036     g_queue_clear (&qtdemux->protection_event_queue);
2037
2038     qtdemux->received_seek = FALSE;
2039     qtdemux->first_moof_already_parsed = FALSE;
2040   }
2041   qtdemux->offset = 0;
2042   gst_adapter_clear (qtdemux->adapter);
2043   gst_segment_init (&qtdemux->segment, GST_FORMAT_TIME);
2044   qtdemux->need_segment = TRUE;
2045
2046   if (hard) {
2047     qtdemux->segment_seqnum = GST_SEQNUM_INVALID;
2048     qtdemux->trickmode_interval = 0;
2049     g_ptr_array_set_size (qtdemux->active_streams, 0);
2050     g_ptr_array_set_size (qtdemux->old_streams, 0);
2051     qtdemux->n_video_streams = 0;
2052     qtdemux->n_audio_streams = 0;
2053     qtdemux->n_sub_streams = 0;
2054     qtdemux->exposed = FALSE;
2055     qtdemux->fragmented = FALSE;
2056     qtdemux->mss_mode = FALSE;
2057     gst_caps_replace (&qtdemux->media_caps, NULL);
2058     qtdemux->timescale = 0;
2059     qtdemux->got_moov = FALSE;
2060     qtdemux->cenc_aux_info_offset = 0;
2061     qtdemux->cenc_aux_info_sizes = NULL;
2062     qtdemux->cenc_aux_sample_count = 0;
2063     if (qtdemux->protection_system_ids) {
2064       g_ptr_array_free (qtdemux->protection_system_ids, TRUE);
2065       qtdemux->protection_system_ids = NULL;
2066     }
2067     qtdemux->streams_aware = GST_OBJECT_PARENT (qtdemux)
2068         && GST_OBJECT_FLAG_IS_SET (GST_OBJECT_PARENT (qtdemux),
2069         GST_BIN_FLAG_STREAMS_AWARE);
2070
2071     if (qtdemux->preferred_protection_system_id) {
2072       g_free (qtdemux->preferred_protection_system_id);
2073       qtdemux->preferred_protection_system_id = NULL;
2074     }
2075   } else if (qtdemux->mss_mode) {
2076     gst_flow_combiner_reset (qtdemux->flowcombiner);
2077     g_ptr_array_foreach (qtdemux->active_streams,
2078         (GFunc) gst_qtdemux_stream_clear, NULL);
2079   } else {
2080     gst_flow_combiner_reset (qtdemux->flowcombiner);
2081     for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
2082       QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
2083       stream->sent_eos = FALSE;
2084       stream->time_position = 0;
2085       stream->accumulated_base = 0;
2086       stream->last_keyframe_dts = GST_CLOCK_TIME_NONE;
2087     }
2088   }
2089 }
2090
2091
2092 /* Maps the @segment to the qt edts internal segments and pushes
2093  * the corresponding segment event.
2094  *
2095  * If it ends up being at a empty segment, a gap will be pushed and the next
2096  * edts segment will be activated in sequence.
2097  *
2098  * To be used in push-mode only */
2099 static void
2100 gst_qtdemux_map_and_push_segments (GstQTDemux * qtdemux, GstSegment * segment)
2101 {
2102   gint i, iter;
2103
2104   for (iter = 0; iter < QTDEMUX_N_STREAMS (qtdemux); iter++) {
2105     QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, iter);
2106
2107     stream->time_position = segment->start;
2108
2109     /* in push mode we should be guaranteed that we will have empty segments
2110      * at the beginning and then one segment after, other scenarios are not
2111      * supported and are discarded when parsing the edts */
2112     for (i = 0; i < stream->n_segments; i++) {
2113       if (stream->segments[i].stop_time > segment->start) {
2114         /* push the empty segment and move to the next one */
2115         gst_qtdemux_activate_segment (qtdemux, stream, i,
2116             stream->time_position);
2117         if (QTSEGMENT_IS_EMPTY (&stream->segments[i])) {
2118           gst_qtdemux_send_gap_for_segment (qtdemux, stream, i,
2119               stream->time_position);
2120
2121           /* accumulate previous segments */
2122           if (GST_CLOCK_TIME_IS_VALID (stream->segment.stop))
2123             stream->accumulated_base +=
2124                 (stream->segment.stop -
2125                 stream->segment.start) / ABS (stream->segment.rate);
2126           continue;
2127         }
2128
2129         g_assert (i == stream->n_segments - 1);
2130       }
2131     }
2132   }
2133 }
2134
2135 static void
2136 gst_qtdemux_stream_concat (GstQTDemux * qtdemux, GPtrArray * dest,
2137     GPtrArray * src)
2138 {
2139   guint i;
2140   guint len;
2141
2142   len = src->len;
2143
2144   if (len == 0)
2145     return;
2146
2147   for (i = 0; i < len; i++) {
2148     QtDemuxStream *stream = g_ptr_array_index (src, i);
2149
2150 #ifndef GST_DISABLE_GST_DEBUG
2151     GST_DEBUG_OBJECT (qtdemux, "Move stream %p (stream-id %s) to %p",
2152         stream, GST_STR_NULL (stream->stream_id), dest);
2153 #endif
2154     g_ptr_array_add (dest, gst_qtdemux_stream_ref (stream));
2155   }
2156
2157   g_ptr_array_set_size (src, 0);
2158 }
2159
2160 static gboolean
2161 gst_qtdemux_handle_sink_event (GstPad * sinkpad, GstObject * parent,
2162     GstEvent * event)
2163 {
2164   GstQTDemux *demux = GST_QTDEMUX (parent);
2165   gboolean res = TRUE;
2166
2167   GST_LOG_OBJECT (demux, "handling %s event", GST_EVENT_TYPE_NAME (event));
2168
2169   switch (GST_EVENT_TYPE (event)) {
2170     case GST_EVENT_SEGMENT:
2171     {
2172       gint64 offset = 0;
2173       QtDemuxStream *stream;
2174       gint idx;
2175       GstSegment segment;
2176
2177       /* some debug output */
2178       gst_event_copy_segment (event, &segment);
2179       GST_DEBUG_OBJECT (demux, "received newsegment %" GST_SEGMENT_FORMAT,
2180           &segment);
2181
2182       if (segment.format == GST_FORMAT_TIME) {
2183         demux->upstream_format_is_time = TRUE;
2184         demux->segment_seqnum = gst_event_get_seqnum (event);
2185       } else {
2186         GST_DEBUG_OBJECT (demux, "Not storing upstream newsegment, "
2187             "not in time format");
2188
2189         /* chain will send initial newsegment after pads have been added */
2190         if (demux->state != QTDEMUX_STATE_MOVIE || !QTDEMUX_N_STREAMS (demux)) {
2191           GST_DEBUG_OBJECT (demux, "still starting, eating event");
2192           goto exit;
2193         }
2194       }
2195
2196       /* check if this matches a time seek we received previously
2197        * FIXME for backwards compatibility reasons we use the
2198        * seek_offset here to compare. In the future we might want to
2199        * change this to use the seqnum as it uniquely should identify
2200        * the segment that corresponds to the seek. */
2201       GST_DEBUG_OBJECT (demux, "Stored seek offset: %" G_GINT64_FORMAT
2202           ", received segment offset %" G_GINT64_FORMAT,
2203           demux->seek_offset, segment.start);
2204       if (segment.format == GST_FORMAT_BYTES
2205           && demux->seek_offset == segment.start) {
2206         GST_OBJECT_LOCK (demux);
2207         offset = segment.start;
2208
2209         segment.format = GST_FORMAT_TIME;
2210         segment.start = demux->push_seek_start;
2211         segment.stop = demux->push_seek_stop;
2212         GST_DEBUG_OBJECT (demux, "Replaced segment with stored seek "
2213             "segment %" GST_TIME_FORMAT " - %" GST_TIME_FORMAT,
2214             GST_TIME_ARGS (segment.start), GST_TIME_ARGS (segment.stop));
2215         GST_OBJECT_UNLOCK (demux);
2216       }
2217
2218       /* we only expect a BYTE segment, e.g. following a seek */
2219       if (segment.format == GST_FORMAT_BYTES) {
2220         if (GST_CLOCK_TIME_IS_VALID (segment.start)) {
2221           offset = segment.start;
2222
2223           gst_qtdemux_find_sample (demux, segment.start, TRUE, FALSE, NULL,
2224               NULL, (gint64 *) & segment.start);
2225           if ((gint64) segment.start < 0)
2226             segment.start = 0;
2227         }
2228         if (GST_CLOCK_TIME_IS_VALID (segment.stop)) {
2229           gst_qtdemux_find_sample (demux, segment.stop, FALSE, FALSE, NULL,
2230               NULL, (gint64 *) & segment.stop);
2231           /* keyframe seeking should already arrange for start >= stop,
2232            * but make sure in other rare cases */
2233           segment.stop = MAX (segment.stop, segment.start);
2234         }
2235       } else if (segment.format == GST_FORMAT_TIME) {
2236         /* push all data on the adapter before starting this
2237          * new segment */
2238         gst_qtdemux_process_adapter (demux, TRUE);
2239       } else {
2240         GST_DEBUG_OBJECT (demux, "unsupported segment format, ignoring");
2241         goto exit;
2242       }
2243
2244       /* We shouldn't modify upstream driven TIME FORMAT segment */
2245       if (!demux->upstream_format_is_time) {
2246         /* accept upstream's notion of segment and distribute along */
2247         segment.format = GST_FORMAT_TIME;
2248         segment.position = segment.time = segment.start;
2249         segment.duration = demux->segment.duration;
2250         segment.base = gst_segment_to_running_time (&demux->segment,
2251             GST_FORMAT_TIME, demux->segment.position);
2252       }
2253
2254       gst_segment_copy_into (&segment, &demux->segment);
2255       GST_DEBUG_OBJECT (demux, "Pushing newseg %" GST_SEGMENT_FORMAT, &segment);
2256
2257       /* map segment to internal qt segments and push on each stream */
2258       if (QTDEMUX_N_STREAMS (demux)) {
2259         demux->need_segment = TRUE;
2260         gst_qtdemux_check_send_pending_segment (demux);
2261       }
2262
2263       /* clear leftover in current segment, if any */
2264       gst_adapter_clear (demux->adapter);
2265
2266       /* set up streaming thread */
2267       demux->offset = offset;
2268       if (demux->upstream_format_is_time) {
2269         GST_DEBUG_OBJECT (demux, "Upstream is driving in time format, "
2270             "set values to restart reading from a new atom");
2271         demux->neededbytes = 16;
2272         demux->todrop = 0;
2273       } else {
2274         gst_qtdemux_find_sample (demux, offset, TRUE, TRUE, &stream, &idx,
2275             NULL);
2276         if (stream) {
2277           demux->todrop = stream->samples[idx].offset - offset;
2278           demux->neededbytes = demux->todrop + stream->samples[idx].size;
2279         } else {
2280           /* set up for EOS */
2281           demux->neededbytes = -1;
2282           demux->todrop = 0;
2283         }
2284       }
2285     exit:
2286       gst_event_unref (event);
2287       res = TRUE;
2288       goto drop;
2289     }
2290     case GST_EVENT_FLUSH_START:
2291     {
2292       if (gst_event_get_seqnum (event) == demux->offset_seek_seqnum) {
2293         gst_event_unref (event);
2294         goto drop;
2295       }
2296       QTDEMUX_EXPOSE_LOCK (demux);
2297       res = gst_pad_event_default (demux->sinkpad, parent, event);
2298       QTDEMUX_EXPOSE_UNLOCK (demux);
2299       goto drop;
2300     }
2301     case GST_EVENT_FLUSH_STOP:
2302     {
2303       guint64 dur;
2304
2305       dur = demux->segment.duration;
2306       gst_qtdemux_reset (demux, FALSE);
2307       demux->segment.duration = dur;
2308
2309       if (gst_event_get_seqnum (event) == demux->offset_seek_seqnum) {
2310         gst_event_unref (event);
2311         goto drop;
2312       }
2313       break;
2314     }
2315     case GST_EVENT_EOS:
2316       /* If we are in push mode, and get an EOS before we've seen any streams,
2317        * then error out - we have nowhere to send the EOS */
2318       if (!demux->pullbased) {
2319         gint i;
2320         gboolean has_valid_stream = FALSE;
2321         for (i = 0; i < QTDEMUX_N_STREAMS (demux); i++) {
2322           if (QTDEMUX_NTH_STREAM (demux, i)->pad != NULL) {
2323             has_valid_stream = TRUE;
2324             break;
2325           }
2326         }
2327         if (!has_valid_stream)
2328           gst_qtdemux_post_no_playable_stream_error (demux);
2329         else {
2330           GST_DEBUG_OBJECT (demux, "Data still available after EOS: %u",
2331               (guint) gst_adapter_available (demux->adapter));
2332           if (gst_qtdemux_process_adapter (demux, TRUE) != GST_FLOW_OK) {
2333             res = FALSE;
2334           }
2335         }
2336       }
2337       break;
2338     case GST_EVENT_CAPS:{
2339       GstCaps *caps = NULL;
2340
2341       gst_event_parse_caps (event, &caps);
2342       gst_qtdemux_setcaps (demux, caps);
2343       res = TRUE;
2344       gst_event_unref (event);
2345       goto drop;
2346     }
2347     case GST_EVENT_PROTECTION:
2348     {
2349       const gchar *system_id = NULL;
2350
2351       gst_event_parse_protection (event, &system_id, NULL, NULL);
2352       GST_DEBUG_OBJECT (demux, "Received protection event for system ID %s",
2353           system_id);
2354       gst_qtdemux_append_protection_system_id (demux, system_id);
2355       /* save the event for later, for source pads that have not been created */
2356       g_queue_push_tail (&demux->protection_event_queue, gst_event_ref (event));
2357       /* send it to all pads that already exist */
2358       gst_qtdemux_push_event (demux, event);
2359       res = TRUE;
2360       goto drop;
2361     }
2362     case GST_EVENT_STREAM_START:
2363     {
2364       res = TRUE;
2365       gst_event_unref (event);
2366
2367       /* Drain all the buffers */
2368       gst_qtdemux_process_adapter (demux, TRUE);
2369       gst_qtdemux_reset (demux, FALSE);
2370       /* We expect new moov box after new stream-start event */
2371       if (demux->exposed) {
2372         gst_qtdemux_stream_concat (demux,
2373             demux->old_streams, demux->active_streams);
2374       }
2375
2376       goto drop;
2377     }
2378     default:
2379       break;
2380   }
2381
2382   res = gst_pad_event_default (demux->sinkpad, parent, event) & res;
2383
2384 drop:
2385   return res;
2386 }
2387
2388 static gboolean
2389 gst_qtdemux_handle_sink_query (GstPad * pad, GstObject * parent,
2390     GstQuery * query)
2391 {
2392   GstQTDemux *demux = GST_QTDEMUX (parent);
2393   gboolean res = FALSE;
2394
2395   switch (GST_QUERY_TYPE (query)) {
2396     case GST_QUERY_BITRATE:
2397     {
2398       GstClockTime duration;
2399
2400       /* populate demux->upstream_size if not done yet */
2401       gst_qtdemux_check_seekability (demux);
2402
2403       if (demux->upstream_size != -1
2404           && gst_qtdemux_get_duration (demux, &duration)) {
2405         guint bitrate =
2406             gst_util_uint64_scale (8 * demux->upstream_size, GST_SECOND,
2407             duration);
2408
2409         GST_LOG_OBJECT (demux, "bitrate query byte length: %" G_GUINT64_FORMAT
2410             " duration %" GST_TIME_FORMAT " resulting a bitrate of %u",
2411             demux->upstream_size, GST_TIME_ARGS (duration), bitrate);
2412
2413         /* TODO: better results based on ranges/index tables */
2414         gst_query_set_bitrate (query, bitrate);
2415         res = TRUE;
2416       }
2417       break;
2418     }
2419     default:
2420       res = gst_pad_query_default (pad, (GstObject *) demux, query);
2421       break;
2422   }
2423
2424   return res;
2425 }
2426
2427
2428 #if 0
2429 static void
2430 gst_qtdemux_set_index (GstElement * element, GstIndex * index)
2431 {
2432   GstQTDemux *demux = GST_QTDEMUX (element);
2433
2434   GST_OBJECT_LOCK (demux);
2435   if (demux->element_index)
2436     gst_object_unref (demux->element_index);
2437   if (index) {
2438     demux->element_index = gst_object_ref (index);
2439   } else {
2440     demux->element_index = NULL;
2441   }
2442   GST_OBJECT_UNLOCK (demux);
2443   /* object lock might be taken again */
2444   if (index)
2445     gst_index_get_writer_id (index, GST_OBJECT (element), &demux->index_id);
2446   GST_DEBUG_OBJECT (demux, "Set index %" GST_PTR_FORMAT "for writer id %d",
2447       demux->element_index, demux->index_id);
2448 }
2449
2450 static GstIndex *
2451 gst_qtdemux_get_index (GstElement * element)
2452 {
2453   GstIndex *result = NULL;
2454   GstQTDemux *demux = GST_QTDEMUX (element);
2455
2456   GST_OBJECT_LOCK (demux);
2457   if (demux->element_index)
2458     result = gst_object_ref (demux->element_index);
2459   GST_OBJECT_UNLOCK (demux);
2460
2461   GST_DEBUG_OBJECT (demux, "Returning index %" GST_PTR_FORMAT, result);
2462
2463   return result;
2464 }
2465 #endif
2466
2467 static void
2468 gst_qtdemux_stbl_free (QtDemuxStream * stream)
2469 {
2470   g_free ((gpointer) stream->stco.data);
2471   stream->stco.data = NULL;
2472   g_free ((gpointer) stream->stsz.data);
2473   stream->stsz.data = NULL;
2474   g_free ((gpointer) stream->stsc.data);
2475   stream->stsc.data = NULL;
2476   g_free ((gpointer) stream->stts.data);
2477   stream->stts.data = NULL;
2478   g_free ((gpointer) stream->stss.data);
2479   stream->stss.data = NULL;
2480   g_free ((gpointer) stream->stps.data);
2481   stream->stps.data = NULL;
2482   g_free ((gpointer) stream->ctts.data);
2483   stream->ctts.data = NULL;
2484 }
2485
2486 static void
2487 gst_qtdemux_stream_flush_segments_data (QtDemuxStream * stream)
2488 {
2489   g_free (stream->segments);
2490   stream->segments = NULL;
2491   stream->segment_index = -1;
2492   stream->accumulated_base = 0;
2493 }
2494
2495 static void
2496 gst_qtdemux_stream_flush_samples_data (QtDemuxStream * stream)
2497 {
2498   g_free (stream->samples);
2499   stream->samples = NULL;
2500   gst_qtdemux_stbl_free (stream);
2501
2502   /* fragments */
2503   g_free (stream->ra_entries);
2504   stream->ra_entries = NULL;
2505   stream->n_ra_entries = 0;
2506
2507   stream->sample_index = -1;
2508   stream->stbl_index = -1;
2509   stream->n_samples = 0;
2510   stream->time_position = 0;
2511
2512   stream->n_samples_moof = 0;
2513   stream->duration_moof = 0;
2514   stream->duration_last_moof = 0;
2515 }
2516
2517 static void
2518 gst_qtdemux_stream_clear (QtDemuxStream * stream)
2519 {
2520   gint i;
2521   if (stream->allocator)
2522     gst_object_unref (stream->allocator);
2523   while (stream->buffers) {
2524     gst_buffer_unref (GST_BUFFER_CAST (stream->buffers->data));
2525     stream->buffers = g_slist_delete_link (stream->buffers, stream->buffers);
2526   }
2527   for (i = 0; i < stream->stsd_entries_length; i++) {
2528     QtDemuxStreamStsdEntry *entry = &stream->stsd_entries[i];
2529     if (entry->rgb8_palette) {
2530       gst_memory_unref (entry->rgb8_palette);
2531       entry->rgb8_palette = NULL;
2532     }
2533     entry->sparse = FALSE;
2534   }
2535
2536   if (stream->stream_tags)
2537     gst_tag_list_unref (stream->stream_tags);
2538
2539   stream->stream_tags = gst_tag_list_new_empty ();
2540   gst_tag_list_set_scope (stream->stream_tags, GST_TAG_SCOPE_STREAM);
2541   g_free (stream->redirect_uri);
2542   stream->redirect_uri = NULL;
2543   stream->sent_eos = FALSE;
2544   stream->protected = FALSE;
2545   if (stream->protection_scheme_info) {
2546     if (stream->protection_scheme_type == FOURCC_cenc
2547         || stream->protection_scheme_type == FOURCC_cbcs) {
2548       QtDemuxCencSampleSetInfo *info =
2549           (QtDemuxCencSampleSetInfo *) stream->protection_scheme_info;
2550       if (info->default_properties)
2551         gst_structure_free (info->default_properties);
2552       if (info->crypto_info)
2553         g_ptr_array_free (info->crypto_info, TRUE);
2554     }
2555     if (stream->protection_scheme_type == FOURCC_aavd) {
2556       QtDemuxAavdEncryptionInfo *info =
2557           (QtDemuxAavdEncryptionInfo *) stream->protection_scheme_info;
2558       if (info->default_properties)
2559         gst_structure_free (info->default_properties);
2560     }
2561     g_free (stream->protection_scheme_info);
2562     stream->protection_scheme_info = NULL;
2563   }
2564   stream->protection_scheme_type = 0;
2565   stream->protection_scheme_version = 0;
2566   g_queue_foreach (&stream->protection_scheme_event_queue,
2567       (GFunc) gst_event_unref, NULL);
2568   g_queue_clear (&stream->protection_scheme_event_queue);
2569   gst_qtdemux_stream_flush_segments_data (stream);
2570   gst_qtdemux_stream_flush_samples_data (stream);
2571 }
2572
2573 static void
2574 gst_qtdemux_stream_reset (QtDemuxStream * stream)
2575 {
2576   gint i;
2577   gst_qtdemux_stream_clear (stream);
2578   for (i = 0; i < stream->stsd_entries_length; i++) {
2579     QtDemuxStreamStsdEntry *entry = &stream->stsd_entries[i];
2580     if (entry->caps) {
2581       gst_caps_unref (entry->caps);
2582       entry->caps = NULL;
2583     }
2584   }
2585   g_free (stream->stsd_entries);
2586   stream->stsd_entries = NULL;
2587   stream->stsd_entries_length = 0;
2588 }
2589
2590 static QtDemuxStream *
2591 gst_qtdemux_stream_ref (QtDemuxStream * stream)
2592 {
2593   g_atomic_int_add (&stream->ref_count, 1);
2594
2595   return stream;
2596 }
2597
2598 static void
2599 gst_qtdemux_stream_unref (QtDemuxStream * stream)
2600 {
2601   if (g_atomic_int_dec_and_test (&stream->ref_count)) {
2602     gst_qtdemux_stream_reset (stream);
2603     gst_tag_list_unref (stream->stream_tags);
2604     if (stream->pad) {
2605       GstQTDemux *demux = stream->demux;
2606       gst_element_remove_pad (GST_ELEMENT_CAST (demux), stream->pad);
2607       GST_OBJECT_LOCK (demux);
2608       gst_flow_combiner_remove_pad (demux->flowcombiner, stream->pad);
2609       GST_OBJECT_UNLOCK (demux);
2610     }
2611     g_free (stream->stream_id);
2612     g_free (stream);
2613   }
2614 }
2615
2616 static GstStateChangeReturn
2617 gst_qtdemux_change_state (GstElement * element, GstStateChange transition)
2618 {
2619   GstQTDemux *qtdemux = GST_QTDEMUX (element);
2620   GstStateChangeReturn result = GST_STATE_CHANGE_FAILURE;
2621
2622   switch (transition) {
2623     case GST_STATE_CHANGE_READY_TO_PAUSED:
2624       gst_qtdemux_reset (qtdemux, TRUE);
2625       break;
2626     default:
2627       break;
2628   }
2629
2630   result = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
2631
2632   switch (transition) {
2633     case GST_STATE_CHANGE_PAUSED_TO_READY:{
2634       gst_qtdemux_reset (qtdemux, TRUE);
2635       break;
2636     }
2637     default:
2638       break;
2639   }
2640
2641   return result;
2642 }
2643
2644 static void
2645 gst_qtdemux_set_context (GstElement * element, GstContext * context)
2646 {
2647   GstQTDemux *qtdemux = GST_QTDEMUX (element);
2648
2649   g_return_if_fail (GST_IS_CONTEXT (context));
2650
2651   if (gst_context_has_context_type (context,
2652           "drm-preferred-decryption-system-id")) {
2653     const GstStructure *s;
2654
2655     s = gst_context_get_structure (context);
2656     g_free (qtdemux->preferred_protection_system_id);
2657     qtdemux->preferred_protection_system_id =
2658         g_strdup (gst_structure_get_string (s, "decryption-system-id"));
2659     GST_DEBUG_OBJECT (element, "set preferred decryption system to %s",
2660         qtdemux->preferred_protection_system_id);
2661   }
2662
2663   GST_ELEMENT_CLASS (parent_class)->set_context (element, context);
2664 }
2665
2666 static void
2667 qtdemux_parse_ftyp (GstQTDemux * qtdemux, const guint8 * buffer, gint length)
2668 {
2669   /* counts as header data */
2670   qtdemux->header_size += length;
2671
2672   /* only consider at least a sufficiently complete ftyp atom */
2673   if (length >= 20) {
2674     GstBuffer *buf;
2675
2676     qtdemux->major_brand = QT_FOURCC (buffer + 8);
2677     GST_DEBUG_OBJECT (qtdemux, "major brand: %" GST_FOURCC_FORMAT,
2678         GST_FOURCC_ARGS (qtdemux->major_brand));
2679     if (qtdemux->comp_brands)
2680       gst_buffer_unref (qtdemux->comp_brands);
2681     buf = qtdemux->comp_brands = gst_buffer_new_and_alloc (length - 16);
2682     gst_buffer_fill (buf, 0, buffer + 16, length - 16);
2683   }
2684 }
2685
2686 static void
2687 qtdemux_update_default_sample_cenc_settings (GstQTDemux * qtdemux,
2688     QtDemuxCencSampleSetInfo * info, guint32 is_encrypted,
2689     guint32 protection_scheme_type, guint8 iv_size, const guint8 * kid,
2690     guint crypt_byte_block, guint skip_byte_block, guint8 constant_iv_size,
2691     const guint8 * constant_iv)
2692 {
2693   GstBuffer *kid_buf = gst_buffer_new_allocate (NULL, 16, NULL);
2694   gst_buffer_fill (kid_buf, 0, kid, 16);
2695   if (info->default_properties)
2696     gst_structure_free (info->default_properties);
2697   info->default_properties =
2698       gst_structure_new ("application/x-cenc",
2699       "iv_size", G_TYPE_UINT, iv_size,
2700       "encrypted", G_TYPE_BOOLEAN, (is_encrypted == 1),
2701       "kid", GST_TYPE_BUFFER, kid_buf, NULL);
2702   GST_DEBUG_OBJECT (qtdemux, "default sample properties: "
2703       "is_encrypted=%u, iv_size=%u", is_encrypted, iv_size);
2704   gst_buffer_unref (kid_buf);
2705   if (protection_scheme_type == FOURCC_cbcs) {
2706     if (crypt_byte_block != 0 || skip_byte_block != 0) {
2707       gst_structure_set (info->default_properties, "crypt_byte_block",
2708           G_TYPE_UINT, crypt_byte_block, "skip_byte_block", G_TYPE_UINT,
2709           skip_byte_block, NULL);
2710     }
2711     if (constant_iv != NULL) {
2712       GstBuffer *constant_iv_buf =
2713           gst_buffer_new_allocate (NULL, constant_iv_size, NULL);
2714       gst_buffer_fill (constant_iv_buf, 0, constant_iv, constant_iv_size);
2715       gst_structure_set (info->default_properties, "constant_iv_size",
2716           G_TYPE_UINT, constant_iv_size, "iv", GST_TYPE_BUFFER, constant_iv_buf,
2717           NULL);
2718       gst_buffer_unref (constant_iv_buf);
2719     }
2720     gst_structure_set (info->default_properties, "cipher-mode",
2721         G_TYPE_STRING, "cbcs", NULL);
2722   } else {
2723     gst_structure_set (info->default_properties, "cipher-mode",
2724         G_TYPE_STRING, "cenc", NULL);
2725   }
2726 }
2727
2728 static gboolean
2729 qtdemux_update_default_piff_encryption_settings (GstQTDemux * qtdemux,
2730     QtDemuxCencSampleSetInfo * info, GstByteReader * br)
2731 {
2732   guint32 algorithm_id = 0;
2733   const guint8 *kid;
2734   gboolean is_encrypted = TRUE;
2735   guint8 iv_size = 8;
2736
2737   if (!gst_byte_reader_get_uint24_le (br, &algorithm_id)) {
2738     GST_ERROR_OBJECT (qtdemux, "Error getting box's algorithm ID field");
2739     return FALSE;
2740   }
2741
2742   algorithm_id >>= 8;
2743   if (algorithm_id == 0) {
2744     is_encrypted = FALSE;
2745   } else if (algorithm_id == 1) {
2746     GST_DEBUG_OBJECT (qtdemux, "AES 128-bits CTR encrypted stream");
2747   } else if (algorithm_id == 2) {
2748     GST_DEBUG_OBJECT (qtdemux, "AES 128-bits CBC encrypted stream");
2749   }
2750
2751   if (!gst_byte_reader_get_uint8 (br, &iv_size))
2752     return FALSE;
2753
2754   if (!gst_byte_reader_get_data (br, 16, &kid))
2755     return FALSE;
2756
2757   qtdemux_update_default_sample_cenc_settings (qtdemux, info,
2758       is_encrypted, FOURCC_cenc, iv_size, kid, 0, 0, 0, NULL);
2759   gst_structure_set (info->default_properties, "piff_algorithm_id",
2760       G_TYPE_UINT, algorithm_id, NULL);
2761   return TRUE;
2762 }
2763
2764
2765 static void
2766 qtdemux_parse_piff (GstQTDemux * qtdemux, const guint8 * buffer, gint length,
2767     guint offset)
2768 {
2769   GstByteReader br;
2770   guint8 version;
2771   guint32 flags = 0;
2772   guint i;
2773   guint iv_size = 8;
2774   QtDemuxStream *stream;
2775   GstStructure *structure;
2776   QtDemuxCencSampleSetInfo *ss_info = NULL;
2777   const gchar *system_id;
2778   gboolean uses_sub_sample_encryption = FALSE;
2779   guint32 sample_count;
2780
2781   if (QTDEMUX_N_STREAMS (qtdemux) == 0)
2782     return;
2783
2784   stream = QTDEMUX_NTH_STREAM (qtdemux, 0);
2785
2786   structure = gst_caps_get_structure (CUR_STREAM (stream)->caps, 0);
2787   if (!gst_structure_has_name (structure, "application/x-cenc")) {
2788     GST_WARNING_OBJECT (qtdemux,
2789         "Attempting PIFF box parsing on an unencrypted stream.");
2790     return;
2791   }
2792
2793   gst_structure_get (structure, GST_PROTECTION_SYSTEM_ID_CAPS_FIELD,
2794       G_TYPE_STRING, &system_id, NULL);
2795   gst_qtdemux_append_protection_system_id (qtdemux, system_id);
2796
2797   stream->protected = TRUE;
2798   stream->protection_scheme_type = FOURCC_cenc;
2799
2800   if (!stream->protection_scheme_info)
2801     stream->protection_scheme_info = g_new0 (QtDemuxCencSampleSetInfo, 1);
2802
2803   ss_info = (QtDemuxCencSampleSetInfo *) stream->protection_scheme_info;
2804   if (!ss_info->default_properties) {
2805     ss_info->default_properties =
2806         gst_structure_new ("application/x-cenc",
2807         "iv_size", G_TYPE_UINT, iv_size, "encrypted", G_TYPE_BOOLEAN, TRUE,
2808         NULL);
2809
2810   }
2811
2812   if (ss_info->crypto_info) {
2813     GST_LOG_OBJECT (qtdemux, "unreffing existing crypto_info");
2814     g_ptr_array_free (ss_info->crypto_info, TRUE);
2815     ss_info->crypto_info = NULL;
2816   }
2817
2818   /* skip UUID */
2819   gst_byte_reader_init (&br, buffer + offset + 16, length - offset - 16);
2820
2821   if (!gst_byte_reader_get_uint8 (&br, &version)) {
2822     GST_ERROR_OBJECT (qtdemux, "Error getting box's version field");
2823     return;
2824   }
2825
2826   if (!gst_byte_reader_get_uint24_be (&br, &flags)) {
2827     GST_ERROR_OBJECT (qtdemux, "Error getting box's flags field");
2828     return;
2829   }
2830
2831   if ((flags & 0x000001)) {
2832     if (!qtdemux_update_default_piff_encryption_settings (qtdemux, ss_info,
2833             &br))
2834       return;
2835   } else if ((flags & 0x000002)) {
2836     uses_sub_sample_encryption = TRUE;
2837   }
2838
2839   if (!gst_structure_get_uint (ss_info->default_properties, "iv_size",
2840           &iv_size)) {
2841     GST_ERROR_OBJECT (qtdemux, "Error getting encryption IV size field");
2842     return;
2843   }
2844
2845   if (!gst_byte_reader_get_uint32_be (&br, &sample_count)) {
2846     GST_ERROR_OBJECT (qtdemux, "Error getting box's sample count field");
2847     return;
2848   }
2849
2850   ss_info->crypto_info =
2851       g_ptr_array_new_full (sample_count,
2852       (GDestroyNotify) qtdemux_gst_structure_free);
2853
2854   for (i = 0; i < sample_count; ++i) {
2855     GstStructure *properties;
2856     guint8 *data;
2857     GstBuffer *buf;
2858
2859     properties = qtdemux_get_cenc_sample_properties (qtdemux, stream, i);
2860     if (properties == NULL) {
2861       GST_ERROR_OBJECT (qtdemux, "failed to get properties for sample %u", i);
2862       qtdemux->cenc_aux_sample_count = i;
2863       return;
2864     }
2865
2866     if (!gst_byte_reader_dup_data (&br, iv_size, &data)) {
2867       GST_ERROR_OBJECT (qtdemux, "IV data not present for sample %u", i);
2868       gst_structure_free (properties);
2869       qtdemux->cenc_aux_sample_count = i;
2870       return;
2871     }
2872     buf = gst_buffer_new_wrapped (data, iv_size);
2873     gst_structure_set (properties, "iv", GST_TYPE_BUFFER, buf, NULL);
2874     gst_buffer_unref (buf);
2875
2876     if (uses_sub_sample_encryption) {
2877       guint16 n_subsamples;
2878       const GValue *kid_buf_value;
2879
2880       if (!gst_byte_reader_get_uint16_be (&br, &n_subsamples)
2881           || n_subsamples == 0) {
2882         GST_ERROR_OBJECT (qtdemux,
2883             "failed to get subsample count for sample %u", i);
2884         gst_structure_free (properties);
2885         qtdemux->cenc_aux_sample_count = i;
2886         return;
2887       }
2888       GST_LOG_OBJECT (qtdemux, "subsample count: %u", n_subsamples);
2889       if (!gst_byte_reader_dup_data (&br, n_subsamples * 6, &data)) {
2890         GST_ERROR_OBJECT (qtdemux, "failed to get subsample data for sample %u",
2891             i);
2892         gst_structure_free (properties);
2893         qtdemux->cenc_aux_sample_count = i;
2894         return;
2895       }
2896       buf = gst_buffer_new_wrapped (data, n_subsamples * 6);
2897
2898       kid_buf_value =
2899           gst_structure_get_value (ss_info->default_properties, "kid");
2900
2901       gst_structure_set (properties,
2902           "subsample_count", G_TYPE_UINT, n_subsamples,
2903           "subsamples", GST_TYPE_BUFFER, buf, NULL);
2904       gst_structure_set_value (properties, "kid", kid_buf_value);
2905       gst_buffer_unref (buf);
2906     } else {
2907       gst_structure_set (properties, "subsample_count", G_TYPE_UINT, 0, NULL);
2908     }
2909
2910     g_ptr_array_add (ss_info->crypto_info, properties);
2911   }
2912
2913   qtdemux->cenc_aux_sample_count = sample_count;
2914 }
2915
2916 static void
2917 qtdemux_parse_uuid (GstQTDemux * qtdemux, const guint8 * buffer, gint length)
2918 {
2919   static const guint8 xmp_uuid[] = { 0xBE, 0x7A, 0xCF, 0xCB,
2920     0x97, 0xA9, 0x42, 0xE8,
2921     0x9C, 0x71, 0x99, 0x94,
2922     0x91, 0xE3, 0xAF, 0xAC
2923   };
2924   static const guint8 playready_uuid[] = {
2925     0xd0, 0x8a, 0x4f, 0x18, 0x10, 0xf3, 0x4a, 0x82,
2926     0xb6, 0xc8, 0x32, 0xd8, 0xab, 0xa1, 0x83, 0xd3
2927   };
2928
2929   static const guint8 piff_sample_encryption_uuid[] = {
2930     0xa2, 0x39, 0x4f, 0x52, 0x5a, 0x9b, 0x4f, 0x14,
2931     0xa2, 0x44, 0x6c, 0x42, 0x7c, 0x64, 0x8d, 0xf4
2932   };
2933
2934   guint offset;
2935
2936   /* counts as header data */
2937   qtdemux->header_size += length;
2938
2939   offset = (QT_UINT32 (buffer) == 0) ? 16 : 8;
2940
2941   if (length <= offset + 16) {
2942     GST_DEBUG_OBJECT (qtdemux, "uuid atom is too short, skipping");
2943     return;
2944   }
2945
2946   if (memcmp (buffer + offset, xmp_uuid, 16) == 0) {
2947     GstBuffer *buf;
2948     GstTagList *taglist;
2949
2950     buf = _gst_buffer_new_wrapped ((guint8 *) buffer + offset + 16,
2951         length - offset - 16, NULL);
2952     taglist = gst_tag_list_from_xmp_buffer (buf);
2953     gst_buffer_unref (buf);
2954
2955     /* make sure we have a usable taglist */
2956     qtdemux->tag_list = gst_tag_list_make_writable (qtdemux->tag_list);
2957
2958     qtdemux_handle_xmp_taglist (qtdemux, qtdemux->tag_list, taglist);
2959
2960   } else if (memcmp (buffer + offset, playready_uuid, 16) == 0) {
2961     int len;
2962     const gunichar2 *s_utf16;
2963     char *contents;
2964
2965     len = GST_READ_UINT16_LE (buffer + offset + 0x30);
2966     s_utf16 = (const gunichar2 *) (buffer + offset + 0x32);
2967     contents = g_utf16_to_utf8 (s_utf16, len / 2, NULL, NULL, NULL);
2968     GST_ERROR_OBJECT (qtdemux, "contents: %s", contents);
2969
2970     g_free (contents);
2971
2972     GST_ELEMENT_ERROR (qtdemux, STREAM, DECRYPT,
2973         (_("Cannot play stream because it is encrypted with PlayReady DRM.")),
2974         (NULL));
2975   } else if (memcmp (buffer + offset, piff_sample_encryption_uuid, 16) == 0) {
2976     qtdemux_parse_piff (qtdemux, buffer, length, offset);
2977   } else {
2978     GST_DEBUG_OBJECT (qtdemux, "Ignoring unknown uuid: %08x-%08x-%08x-%08x",
2979         GST_READ_UINT32_LE (buffer + offset),
2980         GST_READ_UINT32_LE (buffer + offset + 4),
2981         GST_READ_UINT32_LE (buffer + offset + 8),
2982         GST_READ_UINT32_LE (buffer + offset + 12));
2983   }
2984 }
2985
2986 static void
2987 qtdemux_parse_sidx (GstQTDemux * qtdemux, const guint8 * buffer, gint length)
2988 {
2989   GstSidxParser sidx_parser;
2990   GstIsoffParserResult res;
2991   guint consumed;
2992
2993   gst_isoff_qt_sidx_parser_init (&sidx_parser);
2994
2995   res =
2996       gst_isoff_qt_sidx_parser_add_data (&sidx_parser, buffer, length,
2997       &consumed);
2998   GST_DEBUG_OBJECT (qtdemux, "sidx parse result: %d", res);
2999   if (res == GST_ISOFF_QT_PARSER_DONE) {
3000     check_update_duration (qtdemux, sidx_parser.cumulative_pts);
3001   }
3002   gst_isoff_qt_sidx_parser_clear (&sidx_parser);
3003 }
3004
3005 /* caller verifies at least 8 bytes in buf */
3006 static void
3007 extract_initial_length_and_fourcc (const guint8 * data, guint size,
3008     guint64 * plength, guint32 * pfourcc)
3009 {
3010   guint64 length;
3011   guint32 fourcc;
3012
3013   length = QT_UINT32 (data);
3014   GST_DEBUG ("length 0x%08" G_GINT64_MODIFIER "x", length);
3015   fourcc = QT_FOURCC (data + 4);
3016   GST_DEBUG ("atom type %" GST_FOURCC_FORMAT, GST_FOURCC_ARGS (fourcc));
3017
3018   if (length == 0) {
3019     length = G_MAXUINT64;
3020   } else if (length == 1 && size >= 16) {
3021     /* this means we have an extended size, which is the 64 bit value of
3022      * the next 8 bytes */
3023     length = QT_UINT64 (data + 8);
3024     GST_DEBUG ("length 0x%08" G_GINT64_MODIFIER "x", length);
3025   }
3026
3027   if (plength)
3028     *plength = length;
3029   if (pfourcc)
3030     *pfourcc = fourcc;
3031 }
3032
3033 static gboolean
3034 qtdemux_parse_mehd (GstQTDemux * qtdemux, GstByteReader * br)
3035 {
3036   guint32 version = 0;
3037   GstClockTime duration = 0;
3038
3039   if (!gst_byte_reader_get_uint32_be (br, &version))
3040     goto failed;
3041
3042   version >>= 24;
3043   if (version == 1) {
3044     if (!gst_byte_reader_get_uint64_be (br, &duration))
3045       goto failed;
3046   } else {
3047     guint32 dur = 0;
3048
3049     if (!gst_byte_reader_get_uint32_be (br, &dur))
3050       goto failed;
3051     duration = dur;
3052   }
3053
3054   GST_INFO_OBJECT (qtdemux, "mehd duration: %" G_GUINT64_FORMAT, duration);
3055   qtdemux->duration = duration;
3056
3057   return TRUE;
3058
3059 failed:
3060   {
3061     GST_DEBUG_OBJECT (qtdemux, "parsing mehd failed");
3062     return FALSE;
3063   }
3064 }
3065
3066 static gboolean
3067 qtdemux_parse_trex (GstQTDemux * qtdemux, QtDemuxStream * stream,
3068     guint32 * ds_duration, guint32 * ds_size, guint32 * ds_flags)
3069 {
3070   if (!stream->parsed_trex && qtdemux->moov_node) {
3071     GNode *mvex, *trex;
3072     GstByteReader trex_data;
3073
3074     mvex = qtdemux_tree_get_child_by_type (qtdemux->moov_node, FOURCC_mvex);
3075     if (mvex) {
3076       trex = qtdemux_tree_get_child_by_type_full (mvex, FOURCC_trex,
3077           &trex_data);
3078       while (trex) {
3079         guint32 id = 0, sdi = 0, dur = 0, size = 0, flags = 0;
3080
3081         /* skip version/flags */
3082         if (!gst_byte_reader_skip (&trex_data, 4))
3083           goto next;
3084         if (!gst_byte_reader_get_uint32_be (&trex_data, &id))
3085           goto next;
3086         if (id != stream->track_id)
3087           goto next;
3088         if (!gst_byte_reader_get_uint32_be (&trex_data, &sdi))
3089           goto next;
3090         if (!gst_byte_reader_get_uint32_be (&trex_data, &dur))
3091           goto next;
3092         if (!gst_byte_reader_get_uint32_be (&trex_data, &size))
3093           goto next;
3094         if (!gst_byte_reader_get_uint32_be (&trex_data, &flags))
3095           goto next;
3096
3097         GST_DEBUG_OBJECT (qtdemux, "fragment defaults for stream %d; "
3098             "duration %d,  size %d, flags 0x%x", stream->track_id,
3099             dur, size, flags);
3100
3101         stream->parsed_trex = TRUE;
3102         stream->def_sample_description_index = sdi;
3103         stream->def_sample_duration = dur;
3104         stream->def_sample_size = size;
3105         stream->def_sample_flags = flags;
3106
3107       next:
3108         /* iterate all siblings */
3109         trex = qtdemux_tree_get_sibling_by_type_full (trex, FOURCC_trex,
3110             &trex_data);
3111       }
3112     }
3113   }
3114
3115   *ds_duration = stream->def_sample_duration;
3116   *ds_size = stream->def_sample_size;
3117   *ds_flags = stream->def_sample_flags;
3118
3119   /* even then, above values are better than random ... */
3120   if (G_UNLIKELY (!stream->parsed_trex)) {
3121     GST_WARNING_OBJECT (qtdemux,
3122         "failed to find fragment defaults for stream %d", stream->track_id);
3123     return FALSE;
3124   }
3125
3126   return TRUE;
3127 }
3128
3129 /* This method should be called whenever a more accurate duration might
3130  * have been found. It will update all relevant variables if/where needed
3131  */
3132 static void
3133 check_update_duration (GstQTDemux * qtdemux, GstClockTime duration)
3134 {
3135   guint i;
3136   guint64 movdur;
3137   GstClockTime prevdur;
3138
3139   movdur = GSTTIME_TO_QTTIME (qtdemux, duration);
3140
3141   if (movdur > qtdemux->duration) {
3142     prevdur = QTTIME_TO_GSTTIME (qtdemux, qtdemux->duration);
3143     GST_DEBUG_OBJECT (qtdemux,
3144         "Updating total duration to %" GST_TIME_FORMAT " was %" GST_TIME_FORMAT,
3145         GST_TIME_ARGS (duration), GST_TIME_ARGS (prevdur));
3146     qtdemux->duration = movdur;
3147     GST_DEBUG_OBJECT (qtdemux,
3148         "qtdemux->segment.duration: %" GST_TIME_FORMAT " .stop: %"
3149         GST_TIME_FORMAT, GST_TIME_ARGS (qtdemux->segment.duration),
3150         GST_TIME_ARGS (qtdemux->segment.stop));
3151     if (qtdemux->segment.duration == prevdur) {
3152       /* If the current segment has duration/stop identical to previous duration
3153        * update them also (because they were set at that point in time with
3154        * the wrong duration */
3155       /* We convert the value *from* the timescale version to avoid rounding errors */
3156       GstClockTime fixeddur = QTTIME_TO_GSTTIME (qtdemux, movdur);
3157       GST_DEBUG_OBJECT (qtdemux, "Updated segment.duration and segment.stop");
3158       qtdemux->segment.duration = fixeddur;
3159       qtdemux->segment.stop = fixeddur;
3160     }
3161   }
3162
3163   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
3164     QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
3165
3166     movdur = GSTTIME_TO_QTSTREAMTIME (stream, duration);
3167     if (movdur > stream->duration) {
3168       GST_DEBUG_OBJECT (qtdemux,
3169           "Updating stream #%d duration to %" GST_TIME_FORMAT, i,
3170           GST_TIME_ARGS (duration));
3171       stream->duration = movdur;
3172       /* internal duration tracking state has been updated above, so */
3173       /* preserve an open-ended dummy segment rather than repeatedly updating
3174        * it and spamming downstream accordingly with segment events */
3175       /* also mangle the edit list end time when fragmented with a single edit
3176        * list that may only cover any non-fragmented data */
3177       if ((stream->dummy_segment ||
3178               (qtdemux->fragmented && stream->n_segments == 1)) &&
3179           GST_CLOCK_TIME_IS_VALID (stream->segments[0].duration)) {
3180         /* Update all dummy values to new duration */
3181         stream->segments[0].stop_time = duration;
3182         stream->segments[0].duration = duration;
3183         stream->segments[0].media_stop = duration;
3184
3185         /* let downstream know we possibly have a new stop time */
3186         if (stream->segment_index != -1) {
3187           GstClockTime pos;
3188
3189           if (qtdemux->segment.rate >= 0) {
3190             pos = stream->segment.start;
3191           } else {
3192             pos = stream->segment.stop;
3193           }
3194
3195           gst_qtdemux_stream_update_segment (qtdemux, stream,
3196               stream->segment_index, pos, NULL, NULL);
3197         }
3198       }
3199     }
3200   }
3201 }
3202
3203 static gboolean
3204 qtdemux_parse_trun (GstQTDemux * qtdemux, GstByteReader * trun,
3205     QtDemuxStream * stream, guint32 d_sample_duration, guint32 d_sample_size,
3206     guint32 d_sample_flags, gint64 moof_offset, gint64 moof_length,
3207     gint64 * base_offset, gint64 * running_offset, gint64 decode_ts,
3208     gboolean has_tfdt)
3209 {
3210   GstClockTime gst_ts = GST_CLOCK_TIME_NONE;
3211   guint64 timestamp;
3212   gint32 data_offset = 0;
3213   guint8 version;
3214   guint32 flags = 0, first_flags = 0, samples_count = 0;
3215   gint i;
3216   guint8 *data;
3217   guint entry_size, dur_offset, size_offset, flags_offset = 0, ct_offset = 0;
3218   QtDemuxSample *sample;
3219   gboolean ismv = FALSE;
3220   gint64 initial_offset;
3221   gint32 min_ct = 0;
3222
3223   GST_LOG_OBJECT (qtdemux, "parsing trun track-id %d; "
3224       "default dur %d, size %d, flags 0x%x, base offset %" G_GINT64_FORMAT ", "
3225       "decode ts %" G_GINT64_FORMAT, stream->track_id, d_sample_duration,
3226       d_sample_size, d_sample_flags, *base_offset, decode_ts);
3227
3228   if (stream->pending_seek && moof_offset < stream->pending_seek->moof_offset) {
3229     GST_INFO_OBJECT (stream->pad, "skipping trun before seek target fragment");
3230     return TRUE;
3231   }
3232
3233   /* presence of stss or not can't really tell us much,
3234    * and flags and so on tend to be marginally reliable in these files */
3235   if (stream->subtype == FOURCC_soun) {
3236     GST_DEBUG_OBJECT (qtdemux,
3237         "sound track in fragmented file; marking all keyframes");
3238     stream->all_keyframe = TRUE;
3239   }
3240
3241   if (!gst_byte_reader_get_uint8 (trun, &version) ||
3242       !gst_byte_reader_get_uint24_be (trun, &flags))
3243     goto fail;
3244
3245   if (!gst_byte_reader_get_uint32_be (trun, &samples_count))
3246     goto fail;
3247
3248   if (flags & TR_DATA_OFFSET) {
3249     /* note this is really signed */
3250     if (!gst_byte_reader_get_int32_be (trun, &data_offset))
3251       goto fail;
3252     GST_LOG_OBJECT (qtdemux, "trun data offset %d", data_offset);
3253     /* default base offset = first byte of moof */
3254     if (*base_offset == -1) {
3255       GST_LOG_OBJECT (qtdemux, "base_offset at moof");
3256       *base_offset = moof_offset;
3257     }
3258     *running_offset = *base_offset + data_offset;
3259   } else {
3260     /* if no offset at all, that would mean data starts at moof start,
3261      * which is a bit wrong and is ismv crappy way, so compensate
3262      * assuming data is in mdat following moof */
3263     if (*base_offset == -1) {
3264       *base_offset = moof_offset + moof_length + 8;
3265       GST_LOG_OBJECT (qtdemux, "base_offset assumed in mdat after moof");
3266       ismv = TRUE;
3267     }
3268     if (*running_offset == -1)
3269       *running_offset = *base_offset;
3270   }
3271
3272   GST_LOG_OBJECT (qtdemux, "running offset now %" G_GINT64_FORMAT,
3273       *running_offset);
3274   GST_LOG_OBJECT (qtdemux, "trun offset %d, flags 0x%x, entries %d",
3275       data_offset, flags, samples_count);
3276
3277   if (flags & TR_FIRST_SAMPLE_FLAGS) {
3278     if (G_UNLIKELY (flags & TR_SAMPLE_FLAGS)) {
3279       GST_DEBUG_OBJECT (qtdemux,
3280           "invalid flags; SAMPLE and FIRST_SAMPLE present, discarding latter");
3281       flags ^= TR_FIRST_SAMPLE_FLAGS;
3282     } else {
3283       if (!gst_byte_reader_get_uint32_be (trun, &first_flags))
3284         goto fail;
3285       GST_LOG_OBJECT (qtdemux, "first flags: 0x%x", first_flags);
3286     }
3287   }
3288
3289   /* FIXME ? spec says other bits should also be checked to determine
3290    * entry size (and prefix size for that matter) */
3291   entry_size = 0;
3292   dur_offset = size_offset = 0;
3293   if (flags & TR_SAMPLE_DURATION) {
3294     GST_LOG_OBJECT (qtdemux, "entry duration present");
3295     dur_offset = entry_size;
3296     entry_size += 4;
3297   }
3298   if (flags & TR_SAMPLE_SIZE) {
3299     GST_LOG_OBJECT (qtdemux, "entry size present");
3300     size_offset = entry_size;
3301     entry_size += 4;
3302   }
3303   if (flags & TR_SAMPLE_FLAGS) {
3304     GST_LOG_OBJECT (qtdemux, "entry flags present");
3305     flags_offset = entry_size;
3306     entry_size += 4;
3307   }
3308   if (flags & TR_COMPOSITION_TIME_OFFSETS) {
3309     GST_LOG_OBJECT (qtdemux, "entry ct offset present");
3310     ct_offset = entry_size;
3311     entry_size += 4;
3312   }
3313
3314   if (!qt_atom_parser_has_chunks (trun, samples_count, entry_size))
3315     goto fail;
3316   data = (guint8 *) gst_byte_reader_peek_data_unchecked (trun);
3317
3318   if (stream->n_samples + samples_count >=
3319       QTDEMUX_MAX_SAMPLE_INDEX_SIZE / sizeof (QtDemuxSample))
3320     goto index_too_big;
3321
3322   GST_DEBUG_OBJECT (qtdemux, "allocating n_samples %u * %u (%.2f MB)",
3323       stream->n_samples + samples_count, (guint) sizeof (QtDemuxSample),
3324       (stream->n_samples + samples_count) *
3325       sizeof (QtDemuxSample) / (1024.0 * 1024.0));
3326
3327   /* create a new array of samples if it's the first sample parsed */
3328   if (stream->n_samples == 0) {
3329     g_assert (stream->samples == NULL);
3330     stream->samples = g_try_new0 (QtDemuxSample, samples_count);
3331     /* or try to reallocate it with space enough to insert the new samples */
3332   } else
3333     stream->samples = g_try_renew (QtDemuxSample, stream->samples,
3334         stream->n_samples + samples_count);
3335   if (stream->samples == NULL)
3336     goto out_of_memory;
3337
3338   if (qtdemux->fragment_start != -1) {
3339     timestamp = GSTTIME_TO_QTSTREAMTIME (stream, qtdemux->fragment_start);
3340     qtdemux->fragment_start = -1;
3341   } else {
3342     if (stream->n_samples == 0) {
3343       if (decode_ts > 0) {
3344         timestamp = decode_ts;
3345       } else if (stream->pending_seek != NULL) {
3346         /* if we don't have a timestamp from a tfdt box, we'll use the one
3347          * from the mfra seek table */
3348         GST_INFO_OBJECT (stream->pad, "pending seek ts = %" GST_TIME_FORMAT,
3349             GST_TIME_ARGS (stream->pending_seek->ts));
3350
3351         /* FIXME: this is not fully correct, the timestamp refers to the random
3352          * access sample refered to in the tfra entry, which may not necessarily
3353          * be the first sample in the tfrag/trun (but hopefully/usually is) */
3354         timestamp = GSTTIME_TO_QTSTREAMTIME (stream, stream->pending_seek->ts);
3355       } else {
3356         timestamp = 0;
3357       }
3358
3359       gst_ts = QTSTREAMTIME_TO_GSTTIME (stream, timestamp);
3360       GST_INFO_OBJECT (stream->pad, "first sample ts %" GST_TIME_FORMAT,
3361           GST_TIME_ARGS (gst_ts));
3362     } else {
3363       /* subsequent fragments extend stream */
3364       timestamp =
3365           stream->samples[stream->n_samples - 1].timestamp +
3366           stream->samples[stream->n_samples - 1].duration;
3367
3368       /* If this is a GST_FORMAT_BYTES stream and there's a significant
3369        * difference (1 sec.) between decode_ts and timestamp, prefer the
3370        * former */
3371       if (has_tfdt && !qtdemux->upstream_format_is_time
3372           && ABSDIFF (decode_ts, timestamp) >
3373           MAX (stream->duration_last_moof / 2,
3374               GSTTIME_TO_QTSTREAMTIME (stream, GST_SECOND))) {
3375         GST_INFO_OBJECT (qtdemux,
3376             "decode_ts (%" GST_TIME_FORMAT ") and timestamp (%" GST_TIME_FORMAT
3377             ") are significantly different (more than %" GST_TIME_FORMAT
3378             "), using decode_ts",
3379             GST_TIME_ARGS (QTSTREAMTIME_TO_GSTTIME (stream, decode_ts)),
3380             GST_TIME_ARGS (QTSTREAMTIME_TO_GSTTIME (stream, timestamp)),
3381             GST_TIME_ARGS (QTSTREAMTIME_TO_GSTTIME (stream,
3382                     MAX (stream->duration_last_moof / 2,
3383                         GSTTIME_TO_QTSTREAMTIME (stream, GST_SECOND)))));
3384         timestamp = decode_ts;
3385       }
3386
3387       gst_ts = QTSTREAMTIME_TO_GSTTIME (stream, timestamp);
3388       GST_INFO_OBJECT (qtdemux, "first sample ts %" GST_TIME_FORMAT
3389           " (extends previous samples)", GST_TIME_ARGS (gst_ts));
3390     }
3391   }
3392
3393   initial_offset = *running_offset;
3394
3395   sample = stream->samples + stream->n_samples;
3396   for (i = 0; i < samples_count; i++) {
3397     guint32 dur, size, sflags;
3398     gint32 ct;
3399
3400     /* first read sample data */
3401     if (flags & TR_SAMPLE_DURATION) {
3402       dur = QT_UINT32 (data + dur_offset);
3403     } else {
3404       dur = d_sample_duration;
3405     }
3406     if (flags & TR_SAMPLE_SIZE) {
3407       size = QT_UINT32 (data + size_offset);
3408     } else {
3409       size = d_sample_size;
3410     }
3411     if (flags & TR_FIRST_SAMPLE_FLAGS) {
3412       if (i == 0) {
3413         sflags = first_flags;
3414       } else {
3415         sflags = d_sample_flags;
3416       }
3417     } else if (flags & TR_SAMPLE_FLAGS) {
3418       sflags = QT_UINT32 (data + flags_offset);
3419     } else {
3420       sflags = d_sample_flags;
3421     }
3422
3423     if (flags & TR_COMPOSITION_TIME_OFFSETS) {
3424       /* Read offsets as signed numbers regardless of trun version as very
3425        * high offsets are unlikely and there are files out there that use
3426        * version=0 truns with negative offsets */
3427       ct = QT_UINT32 (data + ct_offset);
3428     } else {
3429       ct = 0;
3430     }
3431     data += entry_size;
3432
3433     /* fill the sample information */
3434     sample->offset = *running_offset;
3435     sample->pts_offset = ct;
3436     sample->size = size;
3437     sample->timestamp = timestamp;
3438     sample->duration = dur;
3439     /* sample-is-difference-sample */
3440     /* ismv seems to use 0x40 for keyframe, 0xc0 for non-keyframe,
3441      * now idea how it relates to bitfield other than massive LE/BE confusion */
3442     sample->keyframe = ismv ? ((sflags & 0xff) == 0x40) : !(sflags & 0x10000);
3443     *running_offset += size;
3444     timestamp += dur;
3445     stream->duration_moof += dur;
3446     sample++;
3447
3448     if (ct < min_ct)
3449       min_ct = ct;
3450   }
3451
3452   /* Shift PTS/DTS to allow for negative composition offsets while keeping
3453    * A/V sync in place. This is similar to the code handling ctts/cslg in the
3454    * non-fragmented case.
3455    */
3456   if (min_ct < 0)
3457     stream->cslg_shift = -min_ct;
3458   else
3459     stream->cslg_shift = 0;
3460
3461   /* Update total duration if needed */
3462   check_update_duration (qtdemux, QTSTREAMTIME_TO_GSTTIME (stream, timestamp));
3463
3464   /* Pre-emptively figure out size of mdat based on trun information.
3465    * If the [mdat] atom is effectivelly read, it will be replaced by the actual
3466    * size, else we will still be able to use this when dealing with gap'ed
3467    * input */
3468   qtdemux->mdatleft = *running_offset - initial_offset;
3469   qtdemux->mdatoffset = initial_offset;
3470   qtdemux->mdatsize = qtdemux->mdatleft;
3471
3472   stream->n_samples += samples_count;
3473   stream->n_samples_moof += samples_count;
3474
3475   if (stream->pending_seek != NULL)
3476     stream->pending_seek = NULL;
3477
3478   return TRUE;
3479
3480 fail:
3481   {
3482     GST_WARNING_OBJECT (qtdemux, "failed to parse trun");
3483     return FALSE;
3484   }
3485 out_of_memory:
3486   {
3487     GST_WARNING_OBJECT (qtdemux, "failed to allocate %d samples",
3488         stream->n_samples);
3489     return FALSE;
3490   }
3491 index_too_big:
3492   {
3493     GST_WARNING_OBJECT (qtdemux, "not allocating index of %d samples, would "
3494         "be larger than %uMB (broken file?)", stream->n_samples,
3495         QTDEMUX_MAX_SAMPLE_INDEX_SIZE >> 20);
3496     return FALSE;
3497   }
3498 }
3499
3500 /* find stream with @id */
3501 static inline QtDemuxStream *
3502 qtdemux_find_stream (GstQTDemux * qtdemux, guint32 id)
3503 {
3504   QtDemuxStream *stream;
3505   gint i;
3506
3507   /* check */
3508   if (G_UNLIKELY (!id)) {
3509     GST_DEBUG_OBJECT (qtdemux, "invalid track id 0");
3510     return NULL;
3511   }
3512
3513   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
3514     stream = QTDEMUX_NTH_STREAM (qtdemux, i);
3515     if (stream->track_id == id)
3516       return stream;
3517   }
3518   if (qtdemux->mss_mode) {
3519     /* mss should have only 1 stream anyway */
3520     return QTDEMUX_NTH_STREAM (qtdemux, 0);
3521   }
3522
3523   return NULL;
3524 }
3525
3526 static gboolean
3527 qtdemux_parse_mfhd (GstQTDemux * qtdemux, GstByteReader * mfhd,
3528     guint32 * fragment_number)
3529 {
3530   if (!gst_byte_reader_skip (mfhd, 4))
3531     goto fail;
3532   if (!gst_byte_reader_get_uint32_be (mfhd, fragment_number))
3533     goto fail;
3534   return TRUE;
3535 fail:
3536   {
3537     GST_WARNING_OBJECT (qtdemux, "Failed to parse mfhd atom");
3538     return FALSE;
3539   }
3540 }
3541
3542 static gboolean
3543 qtdemux_parse_tfhd (GstQTDemux * qtdemux, GstByteReader * tfhd,
3544     QtDemuxStream ** stream, guint32 * default_sample_duration,
3545     guint32 * default_sample_size, guint32 * default_sample_flags,
3546     gint64 * base_offset)
3547 {
3548   guint32 flags = 0;
3549   guint32 track_id = 0;
3550
3551   if (!gst_byte_reader_skip (tfhd, 1) ||
3552       !gst_byte_reader_get_uint24_be (tfhd, &flags))
3553     goto invalid_track;
3554
3555   if (!gst_byte_reader_get_uint32_be (tfhd, &track_id))
3556     goto invalid_track;
3557
3558   *stream = qtdemux_find_stream (qtdemux, track_id);
3559   if (G_UNLIKELY (!*stream))
3560     goto unknown_stream;
3561
3562   if (flags & TF_DEFAULT_BASE_IS_MOOF)
3563     *base_offset = qtdemux->moof_offset;
3564
3565   if (flags & TF_BASE_DATA_OFFSET)
3566     if (!gst_byte_reader_get_uint64_be (tfhd, (guint64 *) base_offset))
3567       goto invalid_track;
3568
3569   /* obtain stream defaults */
3570   qtdemux_parse_trex (qtdemux, *stream,
3571       default_sample_duration, default_sample_size, default_sample_flags);
3572
3573   (*stream)->stsd_sample_description_id =
3574       (*stream)->def_sample_description_index - 1;
3575
3576   if (flags & TF_SAMPLE_DESCRIPTION_INDEX) {
3577     guint32 sample_description_index;
3578     if (!gst_byte_reader_get_uint32_be (tfhd, &sample_description_index))
3579       goto invalid_track;
3580     (*stream)->stsd_sample_description_id = sample_description_index - 1;
3581   }
3582
3583   if (qtdemux->mss_mode) {
3584     /* mss has no stsd entry */
3585     (*stream)->stsd_sample_description_id = 0;
3586   }
3587
3588   if (flags & TF_DEFAULT_SAMPLE_DURATION)
3589     if (!gst_byte_reader_get_uint32_be (tfhd, default_sample_duration))
3590       goto invalid_track;
3591
3592   if (flags & TF_DEFAULT_SAMPLE_SIZE)
3593     if (!gst_byte_reader_get_uint32_be (tfhd, default_sample_size))
3594       goto invalid_track;
3595
3596   if (flags & TF_DEFAULT_SAMPLE_FLAGS)
3597     if (!gst_byte_reader_get_uint32_be (tfhd, default_sample_flags))
3598       goto invalid_track;
3599
3600   return TRUE;
3601
3602 invalid_track:
3603   {
3604     GST_WARNING_OBJECT (qtdemux, "invalid track fragment header");
3605     return FALSE;
3606   }
3607 unknown_stream:
3608   {
3609     GST_DEBUG_OBJECT (qtdemux, "unknown stream (%u) in tfhd", track_id);
3610     return TRUE;
3611   }
3612 }
3613
3614 static gboolean
3615 qtdemux_parse_tfdt (GstQTDemux * qtdemux, GstByteReader * br,
3616     guint64 * decode_time)
3617 {
3618   guint32 version = 0;
3619
3620   if (!gst_byte_reader_get_uint32_be (br, &version))
3621     return FALSE;
3622
3623   version >>= 24;
3624   if (version == 1) {
3625     if (!gst_byte_reader_get_uint64_be (br, decode_time))
3626       goto failed;
3627   } else {
3628     guint32 dec_time = 0;
3629     if (!gst_byte_reader_get_uint32_be (br, &dec_time))
3630       goto failed;
3631     *decode_time = dec_time;
3632   }
3633
3634   GST_INFO_OBJECT (qtdemux, "Track fragment decode time: %" G_GUINT64_FORMAT,
3635       *decode_time);
3636
3637   return TRUE;
3638
3639 failed:
3640   {
3641     GST_DEBUG_OBJECT (qtdemux, "parsing tfdt failed");
3642     return FALSE;
3643   }
3644 }
3645
3646 /* Returns a pointer to a GstStructure containing the properties of
3647  * the stream sample identified by @sample_index. The caller must unref
3648  * the returned object after use. Returns NULL if unsuccessful. */
3649 static GstStructure *
3650 qtdemux_get_cenc_sample_properties (GstQTDemux * qtdemux,
3651     QtDemuxStream * stream, guint sample_index)
3652 {
3653   QtDemuxCencSampleSetInfo *info = NULL;
3654
3655   g_return_val_if_fail (stream != NULL, NULL);
3656   g_return_val_if_fail (stream->protected, NULL);
3657   g_return_val_if_fail (stream->protection_scheme_info != NULL, NULL);
3658
3659   info = (QtDemuxCencSampleSetInfo *) stream->protection_scheme_info;
3660
3661   /* Currently, cenc properties for groups of samples are not supported, so
3662    * simply return a copy of the default sample properties */
3663   return gst_structure_copy (info->default_properties);
3664 }
3665
3666 /* Parses the sizes of sample auxiliary information contained within a stream,
3667  * as given in a saiz box. Returns array of sample_count guint8 size values,
3668  * or NULL on failure */
3669 static guint8 *
3670 qtdemux_parse_saiz (GstQTDemux * qtdemux, QtDemuxStream * stream,
3671     GstByteReader * br, guint32 * sample_count)
3672 {
3673   guint32 flags = 0;
3674   guint8 *info_sizes;
3675   guint8 default_info_size;
3676
3677   g_return_val_if_fail (qtdemux != NULL, NULL);
3678   g_return_val_if_fail (stream != NULL, NULL);
3679   g_return_val_if_fail (br != NULL, NULL);
3680   g_return_val_if_fail (sample_count != NULL, NULL);
3681
3682   if (!gst_byte_reader_get_uint32_be (br, &flags))
3683     return NULL;
3684
3685   if (flags & 0x1) {
3686     /* aux_info_type and aux_info_type_parameter are ignored */
3687     if (!gst_byte_reader_skip (br, 8))
3688       return NULL;
3689   }
3690
3691   if (!gst_byte_reader_get_uint8 (br, &default_info_size))
3692     return NULL;
3693   GST_DEBUG_OBJECT (qtdemux, "default_info_size: %u", default_info_size);
3694
3695   if (!gst_byte_reader_get_uint32_be (br, sample_count))
3696     return NULL;
3697   GST_DEBUG_OBJECT (qtdemux, "sample_count: %u", *sample_count);
3698
3699
3700   if (default_info_size == 0) {
3701     if (!gst_byte_reader_dup_data (br, *sample_count, &info_sizes)) {
3702       return NULL;
3703     }
3704   } else {
3705     info_sizes = g_new (guint8, *sample_count);
3706     memset (info_sizes, default_info_size, *sample_count);
3707   }
3708
3709   return info_sizes;
3710 }
3711
3712 /* Parses the offset of sample auxiliary information contained within a stream,
3713  * as given in a saio box. Returns TRUE if successful; FALSE otherwise. */
3714 static gboolean
3715 qtdemux_parse_saio (GstQTDemux * qtdemux, QtDemuxStream * stream,
3716     GstByteReader * br, guint32 * info_type, guint32 * info_type_parameter,
3717     guint64 * offset)
3718 {
3719   guint8 version = 0;
3720   guint32 flags = 0;
3721   guint32 aux_info_type = 0;
3722   guint32 aux_info_type_parameter = 0;
3723   guint32 entry_count;
3724   guint32 off_32;
3725   guint64 off_64;
3726   const guint8 *aux_info_type_data = NULL;
3727
3728   g_return_val_if_fail (qtdemux != NULL, FALSE);
3729   g_return_val_if_fail (stream != NULL, FALSE);
3730   g_return_val_if_fail (br != NULL, FALSE);
3731   g_return_val_if_fail (offset != NULL, FALSE);
3732
3733   if (!gst_byte_reader_get_uint8 (br, &version))
3734     return FALSE;
3735
3736   if (!gst_byte_reader_get_uint24_be (br, &flags))
3737     return FALSE;
3738
3739   if (flags & 0x1) {
3740
3741     if (!gst_byte_reader_get_data (br, 4, &aux_info_type_data))
3742       return FALSE;
3743     aux_info_type = QT_FOURCC (aux_info_type_data);
3744
3745     if (!gst_byte_reader_get_uint32_be (br, &aux_info_type_parameter))
3746       return FALSE;
3747   } else if (stream->protected) {
3748     aux_info_type = stream->protection_scheme_type;
3749   } else {
3750     aux_info_type = CUR_STREAM (stream)->fourcc;
3751   }
3752
3753   if (info_type)
3754     *info_type = aux_info_type;
3755   if (info_type_parameter)
3756     *info_type_parameter = aux_info_type_parameter;
3757
3758   GST_DEBUG_OBJECT (qtdemux, "aux_info_type: '%" GST_FOURCC_FORMAT "', "
3759       "aux_info_type_parameter:  %#06x",
3760       GST_FOURCC_ARGS (aux_info_type), aux_info_type_parameter);
3761
3762   if (!gst_byte_reader_get_uint32_be (br, &entry_count))
3763     return FALSE;
3764
3765   if (entry_count != 1) {
3766     GST_ERROR_OBJECT (qtdemux, "multiple offsets are not supported");
3767     return FALSE;
3768   }
3769
3770   if (version == 0) {
3771     if (!gst_byte_reader_get_uint32_be (br, &off_32))
3772       return FALSE;
3773     *offset = (guint64) off_32;
3774   } else {
3775     if (!gst_byte_reader_get_uint64_be (br, &off_64))
3776       return FALSE;
3777     *offset = off_64;
3778   }
3779
3780   GST_DEBUG_OBJECT (qtdemux, "offset: %" G_GUINT64_FORMAT, *offset);
3781   return TRUE;
3782 }
3783
3784 static void
3785 qtdemux_gst_structure_free (GstStructure * gststructure)
3786 {
3787   if (gststructure) {
3788     gst_structure_free (gststructure);
3789   }
3790 }
3791
3792 /* Parses auxiliary information relating to samples protected using
3793  * Common Encryption (cenc); the format of this information
3794  * is defined in ISO/IEC 23001-7. Returns TRUE if successful; FALSE
3795  * otherwise. */
3796 static gboolean
3797 qtdemux_parse_cenc_aux_info (GstQTDemux * qtdemux, QtDemuxStream * stream,
3798     GstByteReader * br, guint8 * info_sizes, guint32 sample_count)
3799 {
3800   QtDemuxCencSampleSetInfo *ss_info = NULL;
3801   guint8 size;
3802   gint i;
3803   GPtrArray *old_crypto_info = NULL;
3804   guint old_entries = 0;
3805
3806   g_return_val_if_fail (qtdemux != NULL, FALSE);
3807   g_return_val_if_fail (stream != NULL, FALSE);
3808   g_return_val_if_fail (br != NULL, FALSE);
3809   g_return_val_if_fail (stream->protected, FALSE);
3810   g_return_val_if_fail (stream->protection_scheme_info != NULL, FALSE);
3811
3812   ss_info = (QtDemuxCencSampleSetInfo *) stream->protection_scheme_info;
3813
3814   if (ss_info->crypto_info) {
3815     old_crypto_info = ss_info->crypto_info;
3816     /* Count number of non-null entries remaining at the tail end */
3817     for (i = old_crypto_info->len - 1; i >= 0; i--) {
3818       if (g_ptr_array_index (old_crypto_info, i) == NULL)
3819         break;
3820       old_entries++;
3821     }
3822   }
3823
3824   ss_info->crypto_info =
3825       g_ptr_array_new_full (sample_count + old_entries,
3826       (GDestroyNotify) qtdemux_gst_structure_free);
3827
3828   /* We preserve old entries because we parse the next moof in advance
3829    * of consuming all samples from the previous moof, and otherwise
3830    * we'd discard the corresponding crypto info for the samples
3831    * from the previous fragment. */
3832   if (old_entries) {
3833     GST_DEBUG_OBJECT (qtdemux, "Preserving %d old crypto info entries",
3834         old_entries);
3835     for (i = old_crypto_info->len - old_entries; i < old_crypto_info->len; i++) {
3836       g_ptr_array_add (ss_info->crypto_info, g_ptr_array_index (old_crypto_info,
3837               i));
3838       g_ptr_array_index (old_crypto_info, i) = NULL;
3839     }
3840   }
3841
3842   if (old_crypto_info) {
3843     /* Everything now belongs to the new array */
3844     g_ptr_array_free (old_crypto_info, TRUE);
3845   }
3846
3847   for (i = 0; i < sample_count; ++i) {
3848     GstStructure *properties;
3849     guint16 n_subsamples = 0;
3850     guint8 *data;
3851     guint iv_size;
3852     GstBuffer *buf;
3853     gboolean could_read_iv;
3854
3855     properties = qtdemux_get_cenc_sample_properties (qtdemux, stream, i);
3856     if (properties == NULL) {
3857       GST_ERROR_OBJECT (qtdemux, "failed to get properties for sample %u", i);
3858       return FALSE;
3859     }
3860     if (!gst_structure_get_uint (properties, "iv_size", &iv_size)) {
3861       GST_ERROR_OBJECT (qtdemux, "failed to get iv_size for sample %u", i);
3862       gst_structure_free (properties);
3863       return FALSE;
3864     }
3865     could_read_iv =
3866         iv_size > 0 ? gst_byte_reader_dup_data (br, iv_size, &data) : FALSE;
3867     if (could_read_iv) {
3868       buf = gst_buffer_new_wrapped (data, iv_size);
3869       gst_structure_set (properties, "iv", GST_TYPE_BUFFER, buf, NULL);
3870       gst_buffer_unref (buf);
3871     } else if (stream->protection_scheme_type == FOURCC_cbcs) {
3872       const GValue *constant_iv_size_value =
3873           gst_structure_get_value (properties, "constant_iv_size");
3874       const GValue *constant_iv_value =
3875           gst_structure_get_value (properties, "iv");
3876       if (constant_iv_size_value == NULL || constant_iv_value == NULL) {
3877         GST_ERROR_OBJECT (qtdemux, "failed to get constant_iv");
3878         gst_structure_free (properties);
3879         return FALSE;
3880       }
3881       gst_structure_set_value (properties, "iv_size", constant_iv_size_value);
3882       gst_structure_remove_field (properties, "constant_iv_size");
3883     } else if (stream->protection_scheme_type == FOURCC_cenc) {
3884       GST_ERROR_OBJECT (qtdemux, "failed to get IV for sample %u", i);
3885       gst_structure_free (properties);
3886       return FALSE;
3887     }
3888     size = info_sizes[i];
3889     if (size > iv_size) {
3890       if (!gst_byte_reader_get_uint16_be (br, &n_subsamples)
3891           || !(n_subsamples > 0)) {
3892         gst_structure_free (properties);
3893         GST_ERROR_OBJECT (qtdemux,
3894             "failed to get subsample count for sample %u", i);
3895         return FALSE;
3896       }
3897       GST_LOG_OBJECT (qtdemux, "subsample count: %u", n_subsamples);
3898       if (!gst_byte_reader_dup_data (br, n_subsamples * 6, &data)) {
3899         GST_ERROR_OBJECT (qtdemux, "failed to get subsample data for sample %u",
3900             i);
3901         gst_structure_free (properties);
3902         return FALSE;
3903       }
3904       buf = gst_buffer_new_wrapped (data, n_subsamples * 6);
3905       if (!buf) {
3906         gst_structure_free (properties);
3907         return FALSE;
3908       }
3909       gst_structure_set (properties,
3910           "subsample_count", G_TYPE_UINT, n_subsamples,
3911           "subsamples", GST_TYPE_BUFFER, buf, NULL);
3912       gst_buffer_unref (buf);
3913     } else {
3914       gst_structure_set (properties, "subsample_count", G_TYPE_UINT, 0, NULL);
3915     }
3916     g_ptr_array_add (ss_info->crypto_info, properties);
3917   }
3918   return TRUE;
3919 }
3920
3921 /* Converts a UUID in raw byte form to a string representation, as defined in
3922  * RFC 4122. The caller takes ownership of the returned string and is
3923  * responsible for freeing it after use. */
3924 static gchar *
3925 qtdemux_uuid_bytes_to_string (gconstpointer uuid_bytes)
3926 {
3927   const guint8 *uuid = (const guint8 *) uuid_bytes;
3928
3929   return g_strdup_printf ("%02x%02x%02x%02x-%02x%02x-%02x%02x-"
3930       "%02x%02x-%02x%02x%02x%02x%02x%02x",
3931       uuid[0], uuid[1], uuid[2], uuid[3],
3932       uuid[4], uuid[5], uuid[6], uuid[7],
3933       uuid[8], uuid[9], uuid[10], uuid[11],
3934       uuid[12], uuid[13], uuid[14], uuid[15]);
3935 }
3936
3937 /* Parses a Protection System Specific Header box (pssh), as defined in the
3938  * Common Encryption (cenc) standard (ISO/IEC 23001-7), which contains
3939  * information needed by a specific content protection system in order to
3940  * decrypt cenc-protected tracks. Returns TRUE if successful; FALSE
3941  * otherwise. */
3942 static gboolean
3943 qtdemux_parse_pssh (GstQTDemux * qtdemux, GNode * node)
3944 {
3945   gchar *sysid_string;
3946   guint32 pssh_size = QT_UINT32 (node->data);
3947   GstBuffer *pssh = NULL;
3948   GstEvent *event = NULL;
3949   guint32 parent_box_type;
3950   gint i;
3951
3952   if (G_UNLIKELY (pssh_size < 32U)) {
3953     GST_ERROR_OBJECT (qtdemux, "invalid box size");
3954     return FALSE;
3955   }
3956
3957   sysid_string =
3958       qtdemux_uuid_bytes_to_string ((const guint8 *) node->data + 12);
3959
3960   gst_qtdemux_append_protection_system_id (qtdemux, sysid_string);
3961
3962   pssh = gst_buffer_new_memdup (node->data, pssh_size);
3963   GST_LOG_OBJECT (qtdemux, "cenc pssh size: %" G_GSIZE_FORMAT,
3964       gst_buffer_get_size (pssh));
3965
3966   parent_box_type = QT_FOURCC ((const guint8 *) node->parent->data + 4);
3967
3968   /* Push an event containing the pssh box onto the queues of all streams. */
3969   event = gst_event_new_protection (sysid_string, pssh,
3970       (parent_box_type == FOURCC_moov) ? "isobmff/moov" : "isobmff/moof");
3971   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
3972     QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
3973     GST_TRACE_OBJECT (qtdemux,
3974         "adding protection event for stream %s and system %s",
3975         stream->stream_id, sysid_string);
3976     g_queue_push_tail (&stream->protection_scheme_event_queue,
3977         gst_event_ref (event));
3978   }
3979   g_free (sysid_string);
3980   gst_event_unref (event);
3981   gst_buffer_unref (pssh);
3982   return TRUE;
3983 }
3984
3985 static gboolean
3986 qtdemux_parse_moof (GstQTDemux * qtdemux, const guint8 * buffer, guint length,
3987     guint64 moof_offset, QtDemuxStream * stream)
3988 {
3989   GNode *moof_node, *traf_node, *tfhd_node, *trun_node, *tfdt_node, *mfhd_node;
3990   GNode *uuid_node;
3991   GstByteReader mfhd_data, trun_data, tfhd_data, tfdt_data;
3992   GNode *saiz_node, *saio_node, *pssh_node;
3993   GstByteReader saiz_data, saio_data;
3994   guint32 ds_size = 0, ds_duration = 0, ds_flags = 0;
3995   gint64 base_offset, running_offset;
3996   guint32 frag_num;
3997   GstClockTime min_dts = GST_CLOCK_TIME_NONE;
3998
3999   /* NOTE @stream ignored */
4000
4001   moof_node = g_node_new ((guint8 *) buffer);
4002   qtdemux_parse_node (qtdemux, moof_node, buffer, length);
4003   qtdemux_node_dump (qtdemux, moof_node);
4004
4005   /* Get fragment number from mfhd and check it's valid */
4006   mfhd_node =
4007       qtdemux_tree_get_child_by_type_full (moof_node, FOURCC_mfhd, &mfhd_data);
4008   if (mfhd_node == NULL)
4009     goto missing_mfhd;
4010   if (!qtdemux_parse_mfhd (qtdemux, &mfhd_data, &frag_num))
4011     goto fail;
4012   GST_DEBUG_OBJECT (qtdemux, "Fragment #%d", frag_num);
4013
4014   /* unknown base_offset to start with */
4015   base_offset = running_offset = -1;
4016   traf_node = qtdemux_tree_get_child_by_type (moof_node, FOURCC_traf);
4017   while (traf_node) {
4018     guint64 decode_time = 0;
4019
4020     /* Fragment Header node */
4021     tfhd_node =
4022         qtdemux_tree_get_child_by_type_full (traf_node, FOURCC_tfhd,
4023         &tfhd_data);
4024     if (!tfhd_node)
4025       goto missing_tfhd;
4026     if (!qtdemux_parse_tfhd (qtdemux, &tfhd_data, &stream, &ds_duration,
4027             &ds_size, &ds_flags, &base_offset))
4028       goto missing_tfhd;
4029
4030     /* The following code assumes at most a single set of sample auxiliary
4031      * data in the fragment (consisting of a saiz box and a corresponding saio
4032      * box); in theory, however, there could be multiple sets of sample
4033      * auxiliary data in a fragment. */
4034     saiz_node =
4035         qtdemux_tree_get_child_by_type_full (traf_node, FOURCC_saiz,
4036         &saiz_data);
4037     if (saiz_node) {
4038       guint32 info_type = 0;
4039       guint64 offset = 0;
4040       guint32 info_type_parameter = 0;
4041
4042       g_free (qtdemux->cenc_aux_info_sizes);
4043
4044       qtdemux->cenc_aux_info_sizes =
4045           qtdemux_parse_saiz (qtdemux, stream, &saiz_data,
4046           &qtdemux->cenc_aux_sample_count);
4047       if (qtdemux->cenc_aux_info_sizes == NULL) {
4048         GST_ERROR_OBJECT (qtdemux, "failed to parse saiz box");
4049         goto fail;
4050       }
4051       saio_node =
4052           qtdemux_tree_get_child_by_type_full (traf_node, FOURCC_saio,
4053           &saio_data);
4054       if (!saio_node) {
4055         GST_ERROR_OBJECT (qtdemux, "saiz box without a corresponding saio box");
4056         g_free (qtdemux->cenc_aux_info_sizes);
4057         qtdemux->cenc_aux_info_sizes = NULL;
4058         goto fail;
4059       }
4060
4061       if (G_UNLIKELY (!qtdemux_parse_saio (qtdemux, stream, &saio_data,
4062                   &info_type, &info_type_parameter, &offset))) {
4063         GST_ERROR_OBJECT (qtdemux, "failed to parse saio box");
4064         g_free (qtdemux->cenc_aux_info_sizes);
4065         qtdemux->cenc_aux_info_sizes = NULL;
4066         goto fail;
4067       }
4068       if (base_offset > -1 && base_offset > qtdemux->moof_offset)
4069         offset += (guint64) (base_offset - qtdemux->moof_offset);
4070       if ((info_type == FOURCC_cenc || info_type == FOURCC_cbcs)
4071           && info_type_parameter == 0U) {
4072         GstByteReader br;
4073         if (offset > length) {
4074           GST_DEBUG_OBJECT (qtdemux, "cenc auxiliary info stored out of moof");
4075           qtdemux->cenc_aux_info_offset = offset;
4076         } else {
4077           gst_byte_reader_init (&br, buffer + offset, length - offset);
4078           if (!qtdemux_parse_cenc_aux_info (qtdemux, stream, &br,
4079                   qtdemux->cenc_aux_info_sizes,
4080                   qtdemux->cenc_aux_sample_count)) {
4081             GST_ERROR_OBJECT (qtdemux, "failed to parse cenc auxiliary info");
4082             g_free (qtdemux->cenc_aux_info_sizes);
4083             qtdemux->cenc_aux_info_sizes = NULL;
4084             goto fail;
4085           }
4086         }
4087       }
4088     }
4089
4090     tfdt_node =
4091         qtdemux_tree_get_child_by_type_full (traf_node, FOURCC_tfdt,
4092         &tfdt_data);
4093     if (tfdt_node) {
4094       /* We'll use decode_time to interpolate timestamps
4095        * in case the input timestamps are missing */
4096       qtdemux_parse_tfdt (qtdemux, &tfdt_data, &decode_time);
4097
4098       GST_DEBUG_OBJECT (qtdemux, "decode time %" G_GINT64_FORMAT
4099           " (%" GST_TIME_FORMAT ")", decode_time,
4100           GST_TIME_ARGS (stream ? QTSTREAMTIME_TO_GSTTIME (stream,
4101                   decode_time) : GST_CLOCK_TIME_NONE));
4102
4103       /* Discard the fragment buffer timestamp info to avoid using it.
4104        * Rely on tfdt instead as it is more accurate than the timestamp
4105        * that is fetched from a manifest/playlist and is usually
4106        * less accurate. */
4107       qtdemux->fragment_start = -1;
4108     }
4109
4110     if (G_UNLIKELY (!stream)) {
4111       /* we lost track of offset, we'll need to regain it,
4112        * but can delay complaining until later or avoid doing so altogether */
4113       base_offset = -2;
4114       goto next;
4115     }
4116     if (G_UNLIKELY (base_offset < -1))
4117       goto lost_offset;
4118
4119     min_dts = MIN (min_dts, QTSTREAMTIME_TO_GSTTIME (stream, decode_time));
4120
4121     if (!qtdemux->pullbased) {
4122       /* Sample tables can grow enough to be problematic if the system memory
4123        * is very low (e.g. embedded devices) and the videos very long
4124        * (~8 MiB/hour for 25-30 fps video + typical AAC audio frames).
4125        * Fortunately, we can easily discard them for each new fragment when
4126        * we know qtdemux will not receive seeks outside of the current fragment.
4127        * adaptivedemux honors this assumption.
4128        * This optimization is also useful for applications that use qtdemux as
4129        * a push-based simple demuxer, like Media Source Extensions. */
4130       gst_qtdemux_stream_flush_samples_data (stream);
4131     }
4132
4133     /* initialise moof sample data */
4134     stream->n_samples_moof = 0;
4135     stream->duration_last_moof = stream->duration_moof;
4136     stream->duration_moof = 0;
4137
4138     /* Track Run node */
4139     trun_node =
4140         qtdemux_tree_get_child_by_type_full (traf_node, FOURCC_trun,
4141         &trun_data);
4142     while (trun_node) {
4143       qtdemux_parse_trun (qtdemux, &trun_data, stream,
4144           ds_duration, ds_size, ds_flags, moof_offset, length, &base_offset,
4145           &running_offset, decode_time, (tfdt_node != NULL));
4146       /* iterate all siblings */
4147       trun_node = qtdemux_tree_get_sibling_by_type_full (trun_node, FOURCC_trun,
4148           &trun_data);
4149     }
4150
4151     uuid_node = qtdemux_tree_get_child_by_type (traf_node, FOURCC_uuid);
4152     if (uuid_node) {
4153       guint8 *uuid_buffer = (guint8 *) uuid_node->data;
4154       guint32 box_length = QT_UINT32 (uuid_buffer);
4155
4156       qtdemux_parse_uuid (qtdemux, uuid_buffer, box_length);
4157     }
4158
4159     /* if no new base_offset provided for next traf,
4160      * base is end of current traf */
4161     base_offset = running_offset;
4162     running_offset = -1;
4163
4164     if (stream->n_samples_moof && stream->duration_moof)
4165       stream->new_caps = TRUE;
4166
4167   next:
4168     /* iterate all siblings */
4169     traf_node = qtdemux_tree_get_sibling_by_type (traf_node, FOURCC_traf);
4170   }
4171
4172   /* parse any protection system info */
4173   pssh_node = qtdemux_tree_get_child_by_type (moof_node, FOURCC_pssh);
4174   while (pssh_node) {
4175     GST_LOG_OBJECT (qtdemux, "Parsing pssh box.");
4176     qtdemux_parse_pssh (qtdemux, pssh_node);
4177     pssh_node = qtdemux_tree_get_sibling_by_type (pssh_node, FOURCC_pssh);
4178   }
4179
4180   if (!qtdemux->upstream_format_is_time && !qtdemux->first_moof_already_parsed
4181       && !qtdemux->received_seek && GST_CLOCK_TIME_IS_VALID (min_dts)
4182       && min_dts != 0) {
4183     /* Unless the user has explicitly requested another seek, perform an
4184      * internal seek to the time specified in the tfdt.
4185      *
4186      * This way if the user opens a file where the first tfdt is 1 hour
4187      * into the presentation, they will not have to wait 1 hour for run
4188      * time to catch up and actual playback to start. */
4189     gint i;
4190
4191     GST_DEBUG_OBJECT (qtdemux, "First fragment has a non-zero tfdt, "
4192         "performing an internal seek to %" GST_TIME_FORMAT,
4193         GST_TIME_ARGS (min_dts));
4194
4195     qtdemux->segment.start = min_dts;
4196     qtdemux->segment.time = qtdemux->segment.position = min_dts;
4197
4198     for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
4199       QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
4200       stream->time_position = min_dts;
4201     }
4202
4203     /* Before this code was run a segment was already sent when the moov was
4204      * parsed... which is OK -- some apps (mostly tests) expect a segment to
4205      * be emitted after a moov, and we can emit a second segment anyway for
4206      * special cases like this. */
4207     qtdemux->need_segment = TRUE;
4208   }
4209
4210   qtdemux->first_moof_already_parsed = TRUE;
4211
4212   g_node_destroy (moof_node);
4213   return TRUE;
4214
4215 missing_tfhd:
4216   {
4217     GST_DEBUG_OBJECT (qtdemux, "missing tfhd box");
4218     goto fail;
4219   }
4220 missing_mfhd:
4221   {
4222     GST_DEBUG_OBJECT (qtdemux, "Missing mfhd box");
4223     goto fail;
4224   }
4225 lost_offset:
4226   {
4227     GST_DEBUG_OBJECT (qtdemux, "lost offset");
4228     goto fail;
4229   }
4230 fail:
4231   {
4232     g_node_destroy (moof_node);
4233     GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
4234         (_("This file is corrupt and cannot be played.")), (NULL));
4235     return FALSE;
4236   }
4237 }
4238
4239 #if 0
4240 /* might be used if some day we actually use mfra & co
4241  * for random access to fragments,
4242  * but that will require quite some modifications and much less relying
4243  * on a sample array */
4244 #endif
4245
4246 static gboolean
4247 qtdemux_parse_tfra (GstQTDemux * qtdemux, GNode * tfra_node)
4248 {
4249   QtDemuxStream *stream;
4250   guint32 ver_flags, track_id, len, num_entries, i;
4251   guint value_size, traf_size, trun_size, sample_size;
4252   guint64 time = 0, moof_offset = 0;
4253 #if 0
4254   GstBuffer *buf = NULL;
4255   GstFlowReturn ret;
4256 #endif
4257   GstByteReader tfra;
4258
4259   gst_byte_reader_init (&tfra, tfra_node->data, QT_UINT32 (tfra_node->data));
4260
4261   if (!gst_byte_reader_skip (&tfra, 8))
4262     return FALSE;
4263
4264   if (!gst_byte_reader_get_uint32_be (&tfra, &ver_flags))
4265     return FALSE;
4266
4267   if (!gst_byte_reader_get_uint32_be (&tfra, &track_id)
4268       || !gst_byte_reader_get_uint32_be (&tfra, &len)
4269       || !gst_byte_reader_get_uint32_be (&tfra, &num_entries))
4270     return FALSE;
4271
4272   GST_DEBUG_OBJECT (qtdemux, "parsing tfra box for track id %u", track_id);
4273
4274   stream = qtdemux_find_stream (qtdemux, track_id);
4275   if (stream == NULL)
4276     goto unknown_trackid;
4277
4278   value_size = ((ver_flags >> 24) == 1) ? sizeof (guint64) : sizeof (guint32);
4279   sample_size = (len & 3) + 1;
4280   trun_size = ((len & 12) >> 2) + 1;
4281   traf_size = ((len & 48) >> 4) + 1;
4282
4283   GST_DEBUG_OBJECT (qtdemux, "%u entries, sizes: value %u, traf %u, trun %u, "
4284       "sample %u", num_entries, value_size, traf_size, trun_size, sample_size);
4285
4286   if (num_entries == 0)
4287     goto no_samples;
4288
4289   if (!qt_atom_parser_has_chunks (&tfra, num_entries,
4290           value_size + value_size + traf_size + trun_size + sample_size))
4291     goto corrupt_file;
4292
4293   g_free (stream->ra_entries);
4294   stream->ra_entries = g_new (QtDemuxRandomAccessEntry, num_entries);
4295   stream->n_ra_entries = num_entries;
4296
4297   for (i = 0; i < num_entries; i++) {
4298     qt_atom_parser_get_offset (&tfra, value_size, &time);
4299     qt_atom_parser_get_offset (&tfra, value_size, &moof_offset);
4300     qt_atom_parser_get_uint_with_size_unchecked (&tfra, traf_size);
4301     qt_atom_parser_get_uint_with_size_unchecked (&tfra, trun_size);
4302     qt_atom_parser_get_uint_with_size_unchecked (&tfra, sample_size);
4303
4304     time = QTSTREAMTIME_TO_GSTTIME (stream, time);
4305
4306     GST_LOG_OBJECT (qtdemux, "fragment time: %" GST_TIME_FORMAT ", "
4307         " moof_offset: %" G_GUINT64_FORMAT, GST_TIME_ARGS (time), moof_offset);
4308
4309     stream->ra_entries[i].ts = time;
4310     stream->ra_entries[i].moof_offset = moof_offset;
4311
4312     /* don't want to go through the entire file and read all moofs at startup */
4313 #if 0
4314     ret = gst_qtdemux_pull_atom (qtdemux, moof_offset, 0, &buf);
4315     if (ret != GST_FLOW_OK)
4316       goto corrupt_file;
4317     qtdemux_parse_moof (qtdemux, GST_BUFFER_DATA (buf), GST_BUFFER_SIZE (buf),
4318         moof_offset, stream);
4319     gst_buffer_unref (buf);
4320 #endif
4321   }
4322
4323   check_update_duration (qtdemux, time);
4324
4325   return TRUE;
4326
4327 /* ERRORS */
4328 unknown_trackid:
4329   {
4330     GST_WARNING_OBJECT (qtdemux, "Couldn't find stream for track %u", track_id);
4331     return FALSE;
4332   }
4333 corrupt_file:
4334   {
4335     GST_WARNING_OBJECT (qtdemux, "broken traf box, ignoring");
4336     return FALSE;
4337   }
4338 no_samples:
4339   {
4340     GST_WARNING_OBJECT (qtdemux, "stream has no samples");
4341     return FALSE;
4342   }
4343 }
4344
4345 static gboolean
4346 qtdemux_pull_mfro_mfra (GstQTDemux * qtdemux)
4347 {
4348   GstMapInfo mfro_map = GST_MAP_INFO_INIT;
4349   GstMapInfo mfra_map = GST_MAP_INFO_INIT;
4350   GstBuffer *mfro = NULL, *mfra = NULL;
4351   GstFlowReturn flow;
4352   gboolean ret = FALSE;
4353   GNode *mfra_node, *tfra_node;
4354   guint64 mfra_offset = 0;
4355   guint32 fourcc, mfra_size;
4356   gint64 len;
4357
4358   /* query upstream size in bytes */
4359   if (!gst_pad_peer_query_duration (qtdemux->sinkpad, GST_FORMAT_BYTES, &len))
4360     goto size_query_failed;
4361
4362   /* mfro box should be at the very end of the file */
4363   flow = gst_qtdemux_pull_atom (qtdemux, len - 16, 16, &mfro);
4364   if (flow != GST_FLOW_OK)
4365     goto exit;
4366
4367   gst_buffer_map (mfro, &mfro_map, GST_MAP_READ);
4368
4369   fourcc = QT_FOURCC (mfro_map.data + 4);
4370   if (fourcc != FOURCC_mfro)
4371     goto exit;
4372
4373   GST_INFO_OBJECT (qtdemux, "Found mfro box");
4374   if (mfro_map.size < 16)
4375     goto invalid_mfro_size;
4376
4377   mfra_size = QT_UINT32 (mfro_map.data + 12);
4378   if (mfra_size >= len)
4379     goto invalid_mfra_size;
4380
4381   mfra_offset = len - mfra_size;
4382
4383   GST_INFO_OBJECT (qtdemux, "mfra offset: %" G_GUINT64_FORMAT ", size %u",
4384       mfra_offset, mfra_size);
4385
4386   /* now get and parse mfra box */
4387   flow = gst_qtdemux_pull_atom (qtdemux, mfra_offset, mfra_size, &mfra);
4388   if (flow != GST_FLOW_OK)
4389     goto broken_file;
4390
4391   gst_buffer_map (mfra, &mfra_map, GST_MAP_READ);
4392
4393   mfra_node = g_node_new ((guint8 *) mfra_map.data);
4394   qtdemux_parse_node (qtdemux, mfra_node, mfra_map.data, mfra_map.size);
4395
4396   tfra_node = qtdemux_tree_get_child_by_type (mfra_node, FOURCC_tfra);
4397
4398   while (tfra_node) {
4399     qtdemux_parse_tfra (qtdemux, tfra_node);
4400     /* iterate all siblings */
4401     tfra_node = qtdemux_tree_get_sibling_by_type (tfra_node, FOURCC_tfra);
4402   }
4403   g_node_destroy (mfra_node);
4404
4405   GST_INFO_OBJECT (qtdemux, "parsed movie fragment random access box (mfra)");
4406   ret = TRUE;
4407
4408 exit:
4409
4410   if (mfro) {
4411     if (mfro_map.memory != NULL)
4412       gst_buffer_unmap (mfro, &mfro_map);
4413     gst_buffer_unref (mfro);
4414   }
4415   if (mfra) {
4416     if (mfra_map.memory != NULL)
4417       gst_buffer_unmap (mfra, &mfra_map);
4418     gst_buffer_unref (mfra);
4419   }
4420   return ret;
4421
4422 /* ERRORS */
4423 size_query_failed:
4424   {
4425     GST_WARNING_OBJECT (qtdemux, "could not query upstream size");
4426     goto exit;
4427   }
4428 invalid_mfro_size:
4429   {
4430     GST_WARNING_OBJECT (qtdemux, "mfro size is too small");
4431     goto exit;
4432   }
4433 invalid_mfra_size:
4434   {
4435     GST_WARNING_OBJECT (qtdemux, "mfra_size in mfro box is invalid");
4436     goto exit;
4437   }
4438 broken_file:
4439   {
4440     GST_WARNING_OBJECT (qtdemux, "bogus mfra offset or size, broken file");
4441     goto exit;
4442   }
4443 }
4444
4445 static guint64
4446 add_offset (guint64 offset, guint64 advance)
4447 {
4448   /* Avoid 64-bit overflow by clamping */
4449   if (offset > G_MAXUINT64 - advance)
4450     return G_MAXUINT64;
4451   return offset + advance;
4452 }
4453
4454 static GstFlowReturn
4455 gst_qtdemux_loop_state_header (GstQTDemux * qtdemux)
4456 {
4457   guint64 length = 0;
4458   guint32 fourcc = 0;
4459   GstBuffer *buf = NULL;
4460   GstFlowReturn ret = GST_FLOW_OK;
4461   guint64 cur_offset = qtdemux->offset;
4462   GstMapInfo map;
4463
4464   ret = gst_pad_pull_range (qtdemux->sinkpad, cur_offset, 16, &buf);
4465   if (G_UNLIKELY (ret != GST_FLOW_OK))
4466     goto beach;
4467   gst_buffer_map (buf, &map, GST_MAP_READ);
4468   if (G_LIKELY (map.size >= 8))
4469     extract_initial_length_and_fourcc (map.data, map.size, &length, &fourcc);
4470   gst_buffer_unmap (buf, &map);
4471   gst_buffer_unref (buf);
4472
4473   /* maybe we already got most we needed, so only consider this eof */
4474   if (G_UNLIKELY (length == 0)) {
4475     GST_ELEMENT_WARNING (qtdemux, STREAM, DEMUX,
4476         (_("Invalid atom size.")),
4477         ("Header atom '%" GST_FOURCC_FORMAT "' has empty length",
4478             GST_FOURCC_ARGS (fourcc)));
4479     ret = GST_FLOW_EOS;
4480     goto beach;
4481   }
4482
4483   switch (fourcc) {
4484     case FOURCC_moof:
4485       /* record for later parsing when needed */
4486       if (!qtdemux->moof_offset) {
4487         qtdemux->moof_offset = qtdemux->offset;
4488       }
4489       if (qtdemux_pull_mfro_mfra (qtdemux)) {
4490         /* FIXME */
4491       } else {
4492         qtdemux->offset += length;      /* skip moof and keep going */
4493       }
4494       if (qtdemux->got_moov) {
4495         GST_INFO_OBJECT (qtdemux, "moof header, got moov, done with headers");
4496         ret = GST_FLOW_EOS;
4497         goto beach;
4498       }
4499       break;
4500     case FOURCC_mdat:
4501     case FOURCC_free:
4502     case FOURCC_skip:
4503     case FOURCC_wide:
4504     case FOURCC_PICT:
4505     case FOURCC_pnot:
4506     {
4507       GST_LOG_OBJECT (qtdemux,
4508           "skipping atom '%" GST_FOURCC_FORMAT "' at %" G_GUINT64_FORMAT,
4509           GST_FOURCC_ARGS (fourcc), cur_offset);
4510       qtdemux->offset = add_offset (qtdemux->offset, length);
4511       break;
4512     }
4513     case FOURCC_moov:
4514     {
4515       GstBuffer *moov = NULL;
4516
4517       if (qtdemux->got_moov) {
4518         GST_DEBUG_OBJECT (qtdemux, "Skipping moov atom as we have one already");
4519         qtdemux->offset = add_offset (qtdemux->offset, length);
4520         goto beach;
4521       }
4522
4523       ret = gst_pad_pull_range (qtdemux->sinkpad, cur_offset, length, &moov);
4524       if (ret != GST_FLOW_OK)
4525         goto beach;
4526       gst_buffer_map (moov, &map, GST_MAP_READ);
4527
4528       if (length != map.size) {
4529         /* Some files have a 'moov' atom at the end of the file which contains
4530          * a terminal 'free' atom where the body of the atom is missing.
4531          * Check for, and permit, this special case.
4532          */
4533         if (map.size >= 8) {
4534           guint8 *final_data = map.data + (map.size - 8);
4535           guint32 final_length = QT_UINT32 (final_data);
4536           guint32 final_fourcc = QT_FOURCC (final_data + 4);
4537
4538           if (final_fourcc == FOURCC_free
4539               && map.size + final_length - 8 == length) {
4540             /* Ok, we've found that special case. Allocate a new buffer with
4541              * that free atom actually present. */
4542             GstBuffer *newmoov = gst_buffer_new_and_alloc (length);
4543             gst_buffer_fill (newmoov, 0, map.data, map.size);
4544             gst_buffer_memset (newmoov, map.size, 0, final_length - 8);
4545             gst_buffer_unmap (moov, &map);
4546             gst_buffer_unref (moov);
4547             moov = newmoov;
4548             gst_buffer_map (moov, &map, GST_MAP_READ);
4549           }
4550         }
4551       }
4552
4553       if (length != map.size) {
4554         GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
4555             (_("This file is incomplete and cannot be played.")),
4556             ("We got less than expected (received %" G_GSIZE_FORMAT
4557                 ", wanted %u, offset %" G_GUINT64_FORMAT ")", map.size,
4558                 (guint) length, cur_offset));
4559         gst_buffer_unmap (moov, &map);
4560         gst_buffer_unref (moov);
4561         ret = GST_FLOW_ERROR;
4562         goto beach;
4563       }
4564       qtdemux->offset += length;
4565
4566       qtdemux_parse_moov (qtdemux, map.data, length);
4567       qtdemux_node_dump (qtdemux, qtdemux->moov_node);
4568
4569       qtdemux_parse_tree (qtdemux);
4570       if (qtdemux->moov_node_compressed) {
4571         g_node_destroy (qtdemux->moov_node_compressed);
4572         g_free (qtdemux->moov_node->data);
4573       }
4574       qtdemux->moov_node_compressed = NULL;
4575       g_node_destroy (qtdemux->moov_node);
4576       qtdemux->moov_node = NULL;
4577       gst_buffer_unmap (moov, &map);
4578       gst_buffer_unref (moov);
4579       qtdemux->got_moov = TRUE;
4580
4581       break;
4582     }
4583     case FOURCC_ftyp:
4584     {
4585       GstBuffer *ftyp = NULL;
4586
4587       /* extract major brand; might come in handy for ISO vs QT issues */
4588       ret = gst_qtdemux_pull_atom (qtdemux, cur_offset, length, &ftyp);
4589       if (ret != GST_FLOW_OK)
4590         goto beach;
4591       qtdemux->offset += length;
4592       gst_buffer_map (ftyp, &map, GST_MAP_READ);
4593       qtdemux_parse_ftyp (qtdemux, map.data, map.size);
4594       gst_buffer_unmap (ftyp, &map);
4595       gst_buffer_unref (ftyp);
4596       break;
4597     }
4598     case FOURCC_uuid:
4599     {
4600       GstBuffer *uuid = NULL;
4601
4602       /* uuid are extension atoms */
4603       ret = gst_qtdemux_pull_atom (qtdemux, cur_offset, length, &uuid);
4604       if (ret != GST_FLOW_OK)
4605         goto beach;
4606       qtdemux->offset += length;
4607       gst_buffer_map (uuid, &map, GST_MAP_READ);
4608       qtdemux_parse_uuid (qtdemux, map.data, map.size);
4609       gst_buffer_unmap (uuid, &map);
4610       gst_buffer_unref (uuid);
4611       break;
4612     }
4613     case FOURCC_sidx:
4614     {
4615       GstBuffer *sidx = NULL;
4616       ret = gst_qtdemux_pull_atom (qtdemux, cur_offset, length, &sidx);
4617       if (ret != GST_FLOW_OK)
4618         goto beach;
4619       qtdemux->offset += length;
4620       gst_buffer_map (sidx, &map, GST_MAP_READ);
4621       qtdemux_parse_sidx (qtdemux, map.data, map.size);
4622       gst_buffer_unmap (sidx, &map);
4623       gst_buffer_unref (sidx);
4624       break;
4625     }
4626     default:
4627     {
4628       GstBuffer *unknown = NULL;
4629
4630       GST_LOG_OBJECT (qtdemux,
4631           "unknown %08x '%" GST_FOURCC_FORMAT "' of size %" G_GUINT64_FORMAT
4632           " at %" G_GUINT64_FORMAT, fourcc, GST_FOURCC_ARGS (fourcc), length,
4633           cur_offset);
4634       ret = gst_qtdemux_pull_atom (qtdemux, cur_offset, length, &unknown);
4635       if (ret != GST_FLOW_OK)
4636         goto beach;
4637       gst_buffer_map (unknown, &map, GST_MAP_READ);
4638       GST_MEMDUMP ("Unknown tag", map.data, map.size);
4639       gst_buffer_unmap (unknown, &map);
4640       gst_buffer_unref (unknown);
4641       qtdemux->offset += length;
4642       break;
4643     }
4644   }
4645
4646 beach:
4647   if (ret == GST_FLOW_EOS && (qtdemux->got_moov || qtdemux->media_caps)) {
4648     /* digested all data, show what we have */
4649     qtdemux_prepare_streams (qtdemux);
4650     QTDEMUX_EXPOSE_LOCK (qtdemux);
4651     ret = qtdemux_expose_streams (qtdemux);
4652     QTDEMUX_EXPOSE_UNLOCK (qtdemux);
4653
4654     qtdemux->state = QTDEMUX_STATE_MOVIE;
4655     GST_DEBUG_OBJECT (qtdemux, "switching state to STATE_MOVIE (%d)",
4656         qtdemux->state);
4657     return ret;
4658   }
4659   return ret;
4660 }
4661
4662 /* Seeks to the previous keyframe of the indexed stream and
4663  * aligns other streams with respect to the keyframe timestamp
4664  * of indexed stream. Only called in case of Reverse Playback
4665  */
4666 static GstFlowReturn
4667 gst_qtdemux_seek_to_previous_keyframe (GstQTDemux * qtdemux)
4668 {
4669   guint32 seg_idx = 0, k_index = 0;
4670   guint32 ref_seg_idx, ref_k_index;
4671   GstClockTime k_pos = 0, last_stop = 0;
4672   QtDemuxSegment *seg = NULL;
4673   QtDemuxStream *ref_str = NULL;
4674   guint64 seg_media_start_mov;  /* segment media start time in mov format */
4675   guint64 target_ts;
4676   gint i;
4677
4678   /* Now we choose an arbitrary stream, get the previous keyframe timestamp
4679    * and finally align all the other streams on that timestamp with their
4680    * respective keyframes */
4681   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
4682     QtDemuxStream *str = QTDEMUX_NTH_STREAM (qtdemux, i);
4683
4684     /* No candidate yet, take the first stream */
4685     if (!ref_str) {
4686       ref_str = str;
4687       continue;
4688     }
4689
4690     /* So that stream has a segment, we prefer video streams */
4691     if (str->subtype == FOURCC_vide) {
4692       ref_str = str;
4693       break;
4694     }
4695   }
4696
4697   if (G_UNLIKELY (!ref_str)) {
4698     GST_DEBUG_OBJECT (qtdemux, "couldn't find any stream");
4699     goto eos;
4700   }
4701
4702   if (G_UNLIKELY (!ref_str->from_sample)) {
4703     GST_DEBUG_OBJECT (qtdemux, "reached the beginning of the file");
4704     goto eos;
4705   }
4706
4707   /* So that stream has been playing from from_sample to to_sample. We will
4708    * get the timestamp of the previous sample and search for a keyframe before
4709    * that. For audio streams we do an arbitrary jump in the past (10 samples) */
4710   if (ref_str->subtype == FOURCC_vide) {
4711     k_index = gst_qtdemux_find_keyframe (qtdemux, ref_str,
4712         ref_str->from_sample - 1, FALSE);
4713   } else {
4714     if (ref_str->from_sample >= 10)
4715       k_index = ref_str->from_sample - 10;
4716     else
4717       k_index = 0;
4718   }
4719
4720   target_ts =
4721       ref_str->samples[k_index].timestamp +
4722       ref_str->samples[k_index].pts_offset;
4723
4724   /* get current segment for that stream */
4725   seg = &ref_str->segments[ref_str->segment_index];
4726   /* Use segment start in original timescale for comparisons */
4727   seg_media_start_mov = seg->trak_media_start;
4728
4729   GST_LOG_OBJECT (qtdemux, "keyframe index %u ts %" G_GUINT64_FORMAT
4730       " seg start %" G_GUINT64_FORMAT " %" GST_TIME_FORMAT,
4731       k_index, target_ts, seg_media_start_mov,
4732       GST_TIME_ARGS (seg->media_start));
4733
4734   /* Crawl back through segments to find the one containing this I frame */
4735   while (target_ts < seg_media_start_mov) {
4736     GST_DEBUG_OBJECT (qtdemux,
4737         "keyframe position (sample %u) is out of segment %u " " target %"
4738         G_GUINT64_FORMAT " seg start %" G_GUINT64_FORMAT, k_index,
4739         ref_str->segment_index, target_ts, seg_media_start_mov);
4740
4741     if (G_UNLIKELY (!ref_str->segment_index)) {
4742       /* Reached first segment, let's consider it's EOS */
4743       goto eos;
4744     }
4745     ref_str->segment_index--;
4746     seg = &ref_str->segments[ref_str->segment_index];
4747     /* Use segment start in original timescale for comparisons */
4748     seg_media_start_mov = seg->trak_media_start;
4749   }
4750   /* Calculate time position of the keyframe and where we should stop */
4751   k_pos =
4752       QTSTREAMTIME_TO_GSTTIME (ref_str,
4753       target_ts - seg->trak_media_start) + seg->time;
4754   last_stop =
4755       QTSTREAMTIME_TO_GSTTIME (ref_str,
4756       ref_str->samples[ref_str->from_sample].timestamp -
4757       seg->trak_media_start) + seg->time;
4758
4759   GST_DEBUG_OBJECT (qtdemux, "preferred stream played from sample %u, "
4760       "now going to sample %u (pts %" GST_TIME_FORMAT ")", ref_str->from_sample,
4761       k_index, GST_TIME_ARGS (k_pos));
4762
4763   /* Set last_stop with the keyframe timestamp we pushed of that stream */
4764   qtdemux->segment.position = last_stop;
4765   GST_DEBUG_OBJECT (qtdemux, "last_stop now is %" GST_TIME_FORMAT,
4766       GST_TIME_ARGS (last_stop));
4767
4768   if (G_UNLIKELY (last_stop < qtdemux->segment.start)) {
4769     GST_DEBUG_OBJECT (qtdemux, "reached the beginning of segment");
4770     goto eos;
4771   }
4772
4773   ref_seg_idx = ref_str->segment_index;
4774   ref_k_index = k_index;
4775
4776   /* Align them all on this */
4777   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
4778     guint32 index = 0;
4779     GstClockTime seg_time = 0;
4780     QtDemuxStream *str = QTDEMUX_NTH_STREAM (qtdemux, i);
4781
4782     /* aligning reference stream again might lead to backing up to yet another
4783      * keyframe (due to timestamp rounding issues),
4784      * potentially putting more load on downstream; so let's try to avoid */
4785     if (str == ref_str) {
4786       seg_idx = ref_seg_idx;
4787       seg = &str->segments[seg_idx];
4788       k_index = ref_k_index;
4789       GST_DEBUG_OBJECT (qtdemux, "reference track-id %u segment %d, "
4790           "sample at index %d", str->track_id, ref_str->segment_index, k_index);
4791     } else {
4792       seg_idx = gst_qtdemux_find_segment (qtdemux, str, k_pos);
4793       GST_DEBUG_OBJECT (qtdemux,
4794           "track-id %u align segment %d for keyframe pos %" GST_TIME_FORMAT,
4795           str->track_id, seg_idx, GST_TIME_ARGS (k_pos));
4796
4797       /* get segment and time in the segment */
4798       seg = &str->segments[seg_idx];
4799       seg_time = k_pos - seg->time;
4800
4801       /* get the media time in the segment.
4802        * No adjustment for empty "filler" segments */
4803       if (seg->media_start != GST_CLOCK_TIME_NONE)
4804         seg_time += seg->media_start;
4805
4806       /* get the index of the sample with media time */
4807       index = gst_qtdemux_find_index_linear (qtdemux, str, seg_time);
4808       GST_DEBUG_OBJECT (qtdemux,
4809           "track-id %u sample for %" GST_TIME_FORMAT " at %u", str->track_id,
4810           GST_TIME_ARGS (seg_time), index);
4811
4812       /* find previous keyframe */
4813       k_index = gst_qtdemux_find_keyframe (qtdemux, str, index, FALSE);
4814     }
4815
4816     /* Remember until where we want to go */
4817     str->to_sample = str->from_sample - 1;
4818     /* Define our time position */
4819     target_ts =
4820         str->samples[k_index].timestamp + str->samples[k_index].pts_offset;
4821     str->time_position = QTSTREAMTIME_TO_GSTTIME (str, target_ts) + seg->time;
4822     if (seg->media_start != GST_CLOCK_TIME_NONE)
4823       str->time_position -= seg->media_start;
4824
4825     /* Now seek back in time */
4826     gst_qtdemux_move_stream (qtdemux, str, k_index);
4827     GST_DEBUG_OBJECT (qtdemux, "track-id %u keyframe at %u, time position %"
4828         GST_TIME_FORMAT " playing from sample %u to %u", str->track_id, k_index,
4829         GST_TIME_ARGS (str->time_position), str->from_sample, str->to_sample);
4830   }
4831
4832   return GST_FLOW_OK;
4833
4834 eos:
4835   return GST_FLOW_EOS;
4836 }
4837
4838 /*
4839  * Gets the current qt segment start, stop and position for the
4840  * given time offset. This is used in update_segment()
4841  */
4842 static void
4843 gst_qtdemux_stream_segment_get_boundaries (GstQTDemux * qtdemux,
4844     QtDemuxStream * stream, GstClockTime offset,
4845     GstClockTime * _start, GstClockTime * _stop, GstClockTime * _time)
4846 {
4847   GstClockTime seg_time;
4848   GstClockTime start, stop, time;
4849   QtDemuxSegment *segment;
4850
4851   segment = &stream->segments[stream->segment_index];
4852
4853   /* get time in this segment */
4854   seg_time = (offset - segment->time) * segment->rate;
4855
4856   GST_LOG_OBJECT (stream->pad, "seg_time %" GST_TIME_FORMAT,
4857       GST_TIME_ARGS (seg_time));
4858
4859   if (G_UNLIKELY (seg_time > segment->duration)) {
4860     GST_LOG_OBJECT (stream->pad,
4861         "seg_time > segment->duration %" GST_TIME_FORMAT,
4862         GST_TIME_ARGS (segment->duration));
4863     seg_time = segment->duration;
4864   }
4865
4866   /* qtdemux->segment.stop is in outside-time-realm, whereas
4867    * segment->media_stop is in track-time-realm.
4868    *
4869    * In order to compare the two, we need to bring segment.stop
4870    * into the track-time-realm
4871    *
4872    * FIXME - does this comment still hold? Don't see any conversion here */
4873
4874   stop = qtdemux->segment.stop;
4875   if (stop == GST_CLOCK_TIME_NONE)
4876     stop = qtdemux->segment.duration;
4877   if (stop == GST_CLOCK_TIME_NONE)
4878     stop = segment->media_stop;
4879   else
4880     stop =
4881         MIN (segment->media_stop, stop - segment->time + segment->media_start);
4882
4883   if (G_UNLIKELY (QTSEGMENT_IS_EMPTY (segment))) {
4884     start = segment->time + seg_time;
4885     time = offset;
4886     stop = start - seg_time + segment->duration;
4887   } else if (qtdemux->segment.rate >= 0) {
4888     start = MIN (segment->media_start + seg_time, stop);
4889     time = offset;
4890   } else {
4891     if (segment->media_start >= qtdemux->segment.start) {
4892       time = segment->time;
4893     } else {
4894       time = segment->time + (qtdemux->segment.start - segment->media_start);
4895     }
4896
4897     start = MAX (segment->media_start, qtdemux->segment.start);
4898     stop = MIN (segment->media_start + seg_time, stop);
4899   }
4900
4901   *_start = start;
4902   *_stop = stop;
4903   *_time = time;
4904 }
4905
4906 /*
4907  * Updates the qt segment used for the stream and pushes a new segment event
4908  * downstream on this stream's pad.
4909  */
4910 static gboolean
4911 gst_qtdemux_stream_update_segment (GstQTDemux * qtdemux, QtDemuxStream * stream,
4912     gint seg_idx, GstClockTime offset, GstClockTime * _start,
4913     GstClockTime * _stop)
4914 {
4915   QtDemuxSegment *segment;
4916   GstClockTime start = 0, stop = GST_CLOCK_TIME_NONE, time = 0;
4917   gdouble rate;
4918   GstEvent *event;
4919
4920   /* update the current segment */
4921   stream->segment_index = seg_idx;
4922
4923   /* get the segment */
4924   segment = &stream->segments[seg_idx];
4925
4926   if (G_UNLIKELY (offset < segment->time)) {
4927     GST_WARNING_OBJECT (stream->pad, "offset < segment->time %" GST_TIME_FORMAT,
4928         GST_TIME_ARGS (segment->time));
4929     return FALSE;
4930   }
4931
4932   /* segment lies beyond total indicated duration */
4933   if (G_UNLIKELY (qtdemux->segment.duration != GST_CLOCK_TIME_NONE &&
4934           segment->time > qtdemux->segment.duration)) {
4935     GST_WARNING_OBJECT (stream->pad, "file duration %" GST_TIME_FORMAT
4936         " < segment->time %" GST_TIME_FORMAT,
4937         GST_TIME_ARGS (qtdemux->segment.duration),
4938         GST_TIME_ARGS (segment->time));
4939     return FALSE;
4940   }
4941
4942   gst_qtdemux_stream_segment_get_boundaries (qtdemux, stream, offset,
4943       &start, &stop, &time);
4944
4945   GST_DEBUG_OBJECT (stream->pad, "new segment %d from %" GST_TIME_FORMAT
4946       " to %" GST_TIME_FORMAT ", time %" GST_TIME_FORMAT, seg_idx,
4947       GST_TIME_ARGS (start), GST_TIME_ARGS (stop), GST_TIME_ARGS (time));
4948
4949   /* combine global rate with that of the segment */
4950   rate = segment->rate * qtdemux->segment.rate;
4951
4952   /* Copy flags from main segment */
4953   stream->segment.flags = qtdemux->segment.flags;
4954
4955   /* update the segment values used for clipping */
4956   stream->segment.offset = qtdemux->segment.offset;
4957   stream->segment.base = qtdemux->segment.base + stream->accumulated_base;
4958   stream->segment.applied_rate = qtdemux->segment.applied_rate;
4959   stream->segment.rate = rate;
4960   stream->segment.start = start + QTSTREAMTIME_TO_GSTTIME (stream,
4961       stream->cslg_shift);
4962   if (stop != -1)
4963     stream->segment.stop = stop + QTSTREAMTIME_TO_GSTTIME (stream,
4964         stream->cslg_shift);
4965   else
4966     stream->segment.stop = stop;
4967   stream->segment.time = time;
4968   stream->segment.position = stream->segment.start;
4969
4970   GST_DEBUG_OBJECT (stream->pad, "New segment: %" GST_SEGMENT_FORMAT,
4971       &stream->segment);
4972
4973   /* now prepare and send the segment */
4974   if (stream->pad) {
4975     event = gst_event_new_segment (&stream->segment);
4976     if (qtdemux->segment_seqnum != GST_SEQNUM_INVALID) {
4977       gst_event_set_seqnum (event, qtdemux->segment_seqnum);
4978     }
4979     gst_pad_push_event (stream->pad, event);
4980     /* assume we can send more data now */
4981     GST_PAD_LAST_FLOW_RETURN (stream->pad) = GST_FLOW_OK;
4982     /* clear to send tags on this pad now */
4983     gst_qtdemux_push_tags (qtdemux, stream);
4984   }
4985
4986   if (_start)
4987     *_start = start;
4988   if (_stop)
4989     *_stop = stop;
4990
4991   return TRUE;
4992 }
4993
4994 /* activate the given segment number @seg_idx of @stream at time @offset.
4995  * @offset is an absolute global position over all the segments.
4996  *
4997  * This will push out a NEWSEGMENT event with the right values and
4998  * position the stream index to the first decodable sample before
4999  * @offset.
5000  */
5001 static gboolean
5002 gst_qtdemux_activate_segment (GstQTDemux * qtdemux, QtDemuxStream * stream,
5003     guint32 seg_idx, GstClockTime offset)
5004 {
5005   QtDemuxSegment *segment;
5006   guint32 index, kf_index;
5007   GstClockTime start = 0, stop = GST_CLOCK_TIME_NONE;
5008
5009   GST_LOG_OBJECT (stream->pad, "activate segment %d, offset %" GST_TIME_FORMAT,
5010       seg_idx, GST_TIME_ARGS (offset));
5011
5012   if (!gst_qtdemux_stream_update_segment (qtdemux, stream, seg_idx, offset,
5013           &start, &stop))
5014     return FALSE;
5015
5016   segment = &stream->segments[stream->segment_index];
5017
5018   /* in the fragmented case, we pick a fragment that starts before our
5019    * desired position and rely on downstream to wait for a keyframe
5020    * (FIXME: doesn't seem to work so well with ismv and wmv, as no parser; the
5021    * tfra entries tells us which trun/sample the key unit is in, but we don't
5022    * make use of this additional information at the moment) */
5023   if (qtdemux->fragmented && !qtdemux->fragmented_seek_pending) {
5024     stream->to_sample = G_MAXUINT32;
5025     return TRUE;
5026   } else {
5027     /* well, it will be taken care of below */
5028     qtdemux->fragmented_seek_pending = FALSE;
5029     /* FIXME ideally the do_fragmented_seek can be done right here,
5030      * rather than at loop level
5031      * (which might even allow handling edit lists in a fragmented file) */
5032   }
5033
5034   /* We don't need to look for a sample in push-based */
5035   if (!qtdemux->pullbased)
5036     return TRUE;
5037
5038   /* and move to the keyframe before the indicated media time of the
5039    * segment */
5040   if (G_LIKELY (!QTSEGMENT_IS_EMPTY (segment))) {
5041     if (qtdemux->segment.rate >= 0) {
5042       index = gst_qtdemux_find_index_linear (qtdemux, stream, start);
5043       stream->to_sample = G_MAXUINT32;
5044       GST_DEBUG_OBJECT (stream->pad,
5045           "moving data pointer to %" GST_TIME_FORMAT ", index: %u, pts %"
5046           GST_TIME_FORMAT, GST_TIME_ARGS (start), index,
5047           GST_TIME_ARGS (QTSAMPLE_PTS (stream, &stream->samples[index])));
5048     } else {
5049       index = gst_qtdemux_find_index_linear (qtdemux, stream, stop);
5050       stream->to_sample = index;
5051       GST_DEBUG_OBJECT (stream->pad,
5052           "moving data pointer to %" GST_TIME_FORMAT ", index: %u, pts %"
5053           GST_TIME_FORMAT, GST_TIME_ARGS (stop), index,
5054           GST_TIME_ARGS (QTSAMPLE_PTS (stream, &stream->samples[index])));
5055     }
5056   } else {
5057     GST_DEBUG_OBJECT (stream->pad, "No need to look for keyframe, "
5058         "this is an empty segment");
5059     return TRUE;
5060   }
5061
5062   /* gst_qtdemux_parse_sample () called from gst_qtdemux_find_index_linear ()
5063    * encountered an error and printed a message so we return appropriately */
5064   if (index == -1)
5065     return FALSE;
5066
5067   /* we're at the right spot */
5068   if (index == stream->sample_index) {
5069     GST_DEBUG_OBJECT (stream->pad, "we are at the right index");
5070     return TRUE;
5071   }
5072
5073   /* find keyframe of the target index */
5074   kf_index = gst_qtdemux_find_keyframe (qtdemux, stream, index, FALSE);
5075
5076   /* go back two frames to provide lead-in for non-raw audio decoders */
5077   if (stream->subtype == FOURCC_soun && !stream->need_clip) {
5078     guint32 lead_in = 2;
5079     guint32 old_index = kf_index;
5080     GstStructure *s = gst_caps_get_structure (CUR_STREAM (stream)->caps, 0);
5081
5082     if (gst_structure_has_name (s, "audio/mpeg")) {
5083       gint mpegversion;
5084       if (gst_structure_get_int (s, "mpegversion", &mpegversion)
5085           && mpegversion == 1) {
5086         /* mp3 could need up to 30 frames of lead-in per mpegaudioparse */
5087         lead_in = 30;
5088       }
5089     }
5090
5091     kf_index = MAX (kf_index, lead_in) - lead_in;
5092     if (qtdemux_parse_samples (qtdemux, stream, kf_index)) {
5093       GST_DEBUG_OBJECT (stream->pad,
5094           "Moving backwards %u frames to ensure sufficient sound lead-in",
5095           old_index - kf_index);
5096     } else {
5097       kf_index = old_index;
5098     }
5099   }
5100
5101   /* if we move forwards, we don't have to go back to the previous
5102    * keyframe since we already sent that. We can also just jump to
5103    * the keyframe right before the target index if there is one. */
5104   if (index > stream->sample_index) {
5105     /* moving forwards check if we move past a keyframe */
5106     if (kf_index > stream->sample_index) {
5107       GST_DEBUG_OBJECT (stream->pad,
5108           "moving forwards to keyframe at %u "
5109           "(pts %" GST_TIME_FORMAT " dts %" GST_TIME_FORMAT " )",
5110           kf_index,
5111           GST_TIME_ARGS (QTSAMPLE_PTS (stream, &stream->samples[kf_index])),
5112           GST_TIME_ARGS (QTSAMPLE_DTS (stream, &stream->samples[kf_index])));
5113       gst_qtdemux_move_stream (qtdemux, stream, kf_index);
5114     } else {
5115       GST_DEBUG_OBJECT (stream->pad,
5116           "moving forwards, keyframe at %u "
5117           "(pts %" GST_TIME_FORMAT " dts %" GST_TIME_FORMAT " ) already sent",
5118           kf_index,
5119           GST_TIME_ARGS (QTSAMPLE_PTS (stream, &stream->samples[kf_index])),
5120           GST_TIME_ARGS (QTSAMPLE_DTS (stream, &stream->samples[kf_index])));
5121     }
5122   } else {
5123     GST_DEBUG_OBJECT (stream->pad,
5124         "moving backwards to %sframe at %u "
5125         "(pts %" GST_TIME_FORMAT " dts %" GST_TIME_FORMAT " )",
5126         (stream->subtype == FOURCC_soun) ? "audio " : "key", kf_index,
5127         GST_TIME_ARGS (QTSAMPLE_PTS (stream, &stream->samples[kf_index])),
5128         GST_TIME_ARGS (QTSAMPLE_DTS (stream, &stream->samples[kf_index])));
5129     gst_qtdemux_move_stream (qtdemux, stream, kf_index);
5130   }
5131
5132   return TRUE;
5133 }
5134
5135 /* prepare to get the current sample of @stream, getting essential values.
5136  *
5137  * This function will also prepare and send the segment when needed.
5138  *
5139  * Return FALSE if the stream is EOS.
5140  *
5141  * PULL-BASED
5142  */
5143 static gboolean
5144 gst_qtdemux_prepare_current_sample (GstQTDemux * qtdemux,
5145     QtDemuxStream * stream, gboolean * empty, guint64 * offset, guint * size,
5146     GstClockTime * dts, GstClockTime * pts, GstClockTime * duration,
5147     gboolean * keyframe)
5148 {
5149   QtDemuxSample *sample;
5150   GstClockTime time_position;
5151   guint32 seg_idx;
5152
5153   g_return_val_if_fail (stream != NULL, FALSE);
5154
5155   time_position = stream->time_position;
5156   if (G_UNLIKELY (time_position == GST_CLOCK_TIME_NONE))
5157     goto eos;
5158
5159   seg_idx = stream->segment_index;
5160   if (G_UNLIKELY (seg_idx == -1)) {
5161     /* find segment corresponding to time_position if we are looking
5162      * for a segment. */
5163     seg_idx = gst_qtdemux_find_segment (qtdemux, stream, time_position);
5164   }
5165
5166   /* different segment, activate it, sample_index will be set. */
5167   if (G_UNLIKELY (stream->segment_index != seg_idx))
5168     gst_qtdemux_activate_segment (qtdemux, stream, seg_idx, time_position);
5169
5170   if (G_UNLIKELY (QTSEGMENT_IS_EMPTY (&stream->
5171               segments[stream->segment_index]))) {
5172     QtDemuxSegment *seg = &stream->segments[stream->segment_index];
5173
5174     GST_LOG_OBJECT (qtdemux, "Empty segment activated,"
5175         " prepare empty sample");
5176
5177     *empty = TRUE;
5178     *pts = *dts = time_position;
5179     *duration = seg->duration - (time_position - seg->time);
5180
5181     return TRUE;
5182   }
5183
5184   *empty = FALSE;
5185
5186   if (stream->sample_index == -1)
5187     stream->sample_index = 0;
5188
5189   GST_LOG_OBJECT (qtdemux, "segment active, index = %u of %u",
5190       stream->sample_index, stream->n_samples);
5191
5192   if (G_UNLIKELY (stream->sample_index >= stream->n_samples)) {
5193     if (!qtdemux->fragmented)
5194       goto eos;
5195
5196     GST_INFO_OBJECT (qtdemux, "out of samples, trying to add more");
5197     do {
5198       GstFlowReturn flow;
5199
5200       GST_OBJECT_LOCK (qtdemux);
5201       flow = qtdemux_add_fragmented_samples (qtdemux);
5202       GST_OBJECT_UNLOCK (qtdemux);
5203
5204       if (flow != GST_FLOW_OK)
5205         goto eos;
5206     }
5207     while (stream->sample_index >= stream->n_samples);
5208   }
5209
5210   if (!qtdemux_parse_samples (qtdemux, stream, stream->sample_index)) {
5211     GST_LOG_OBJECT (qtdemux, "Parsing of index %u failed!",
5212         stream->sample_index);
5213     return FALSE;
5214   }
5215
5216   /* now get the info for the sample we're at */
5217   sample = &stream->samples[stream->sample_index];
5218
5219   *dts = QTSAMPLE_DTS (stream, sample);
5220   *pts = QTSAMPLE_PTS (stream, sample);
5221   *offset = sample->offset;
5222   *size = sample->size;
5223   *duration = QTSAMPLE_DUR_DTS (stream, sample, *dts);
5224   *keyframe = QTSAMPLE_KEYFRAME (stream, sample);
5225
5226   return TRUE;
5227
5228   /* special cases */
5229 eos:
5230   {
5231     stream->time_position = GST_CLOCK_TIME_NONE;
5232     return FALSE;
5233   }
5234 }
5235
5236 /* move to the next sample in @stream.
5237  *
5238  * Moves to the next segment when needed.
5239  */
5240 static void
5241 gst_qtdemux_advance_sample (GstQTDemux * qtdemux, QtDemuxStream * stream)
5242 {
5243   QtDemuxSample *sample;
5244   QtDemuxSegment *segment;
5245
5246   /* get current segment */
5247   segment = &stream->segments[stream->segment_index];
5248
5249   if (G_UNLIKELY (QTSEGMENT_IS_EMPTY (segment))) {
5250     GST_DEBUG_OBJECT (qtdemux, "Empty segment, no samples to advance");
5251     goto next_segment;
5252   }
5253
5254   if (G_UNLIKELY (stream->sample_index >= stream->to_sample)) {
5255     /* Mark the stream as EOS */
5256     GST_DEBUG_OBJECT (qtdemux,
5257         "reached max allowed sample %u, mark EOS", stream->to_sample);
5258     stream->time_position = GST_CLOCK_TIME_NONE;
5259     return;
5260   }
5261
5262   /* move to next sample */
5263   stream->sample_index++;
5264   stream->offset_in_sample = 0;
5265
5266   GST_TRACE_OBJECT (qtdemux, "advance to sample %u/%u", stream->sample_index,
5267       stream->n_samples);
5268
5269   /* reached the last sample, we need the next segment */
5270   if (G_UNLIKELY (stream->sample_index >= stream->n_samples))
5271     goto next_segment;
5272
5273   if (!qtdemux_parse_samples (qtdemux, stream, stream->sample_index)) {
5274     GST_LOG_OBJECT (qtdemux, "Parsing of index %u failed!",
5275         stream->sample_index);
5276     return;
5277   }
5278
5279   /* get next sample */
5280   sample = &stream->samples[stream->sample_index];
5281
5282   GST_TRACE_OBJECT (qtdemux, "sample dts %" GST_TIME_FORMAT " media_stop: %"
5283       GST_TIME_FORMAT, GST_TIME_ARGS (QTSAMPLE_DTS (stream, sample)),
5284       GST_TIME_ARGS (segment->media_stop));
5285
5286   /* see if we are past the segment */
5287   if (G_UNLIKELY (QTSAMPLE_DTS (stream, sample) >= segment->media_stop))
5288     goto next_segment;
5289
5290   if (QTSAMPLE_DTS (stream, sample) >= segment->media_start) {
5291     /* inside the segment, update time_position, looks very familiar to
5292      * GStreamer segments, doesn't it? */
5293     stream->time_position =
5294         QTSAMPLE_DTS (stream, sample) - segment->media_start + segment->time;
5295   } else {
5296     /* not yet in segment, time does not yet increment. This means
5297      * that we are still prerolling keyframes to the decoder so it can
5298      * decode the first sample of the segment. */
5299     stream->time_position = segment->time;
5300   }
5301   return;
5302
5303   /* move to the next segment */
5304 next_segment:
5305   {
5306     GST_DEBUG_OBJECT (qtdemux, "segment %d ended ", stream->segment_index);
5307
5308     if (stream->segment_index == stream->n_segments - 1) {
5309       /* are we at the end of the last segment, we're EOS */
5310       stream->time_position = GST_CLOCK_TIME_NONE;
5311     } else {
5312       /* else we're only at the end of the current segment */
5313       stream->time_position = segment->stop_time;
5314     }
5315     /* make sure we select a new segment */
5316
5317     /* accumulate previous segments */
5318     if (GST_CLOCK_TIME_IS_VALID (stream->segment.stop))
5319       stream->accumulated_base +=
5320           (stream->segment.stop -
5321           stream->segment.start) / ABS (stream->segment.rate);
5322
5323     stream->segment_index = -1;
5324   }
5325 }
5326
5327 static void
5328 gst_qtdemux_sync_streams (GstQTDemux * demux)
5329 {
5330   gint i;
5331
5332   if (QTDEMUX_N_STREAMS (demux) <= 1)
5333     return;
5334
5335   for (i = 0; i < QTDEMUX_N_STREAMS (demux); i++) {
5336     QtDemuxStream *stream;
5337     GstClockTime end_time;
5338
5339     stream = QTDEMUX_NTH_STREAM (demux, i);
5340
5341     if (!stream->pad)
5342       continue;
5343
5344     /* TODO advance time on subtitle streams here, if any some day */
5345
5346     /* some clips/trailers may have unbalanced streams at the end,
5347      * so send EOS on shorter stream to prevent stalling others */
5348
5349     /* do not mess with EOS if SEGMENT seeking */
5350     if (demux->segment.flags & GST_SEEK_FLAG_SEGMENT)
5351       continue;
5352
5353     if (demux->pullbased) {
5354       /* loop mode is sample time based */
5355       if (!STREAM_IS_EOS (stream))
5356         continue;
5357     } else {
5358       /* push mode is byte position based */
5359       if (stream->n_samples &&
5360           stream->samples[stream->n_samples - 1].offset >= demux->offset)
5361         continue;
5362     }
5363
5364     if (stream->sent_eos)
5365       continue;
5366
5367     /* only act if some gap */
5368     end_time = stream->segments[stream->n_segments - 1].stop_time;
5369     GST_LOG_OBJECT (demux, "current position: %" GST_TIME_FORMAT
5370         ", stream end: %" GST_TIME_FORMAT,
5371         GST_TIME_ARGS (demux->segment.position), GST_TIME_ARGS (end_time));
5372     if (GST_CLOCK_TIME_IS_VALID (end_time)
5373         && (end_time + 2 * GST_SECOND < demux->segment.position)) {
5374       GstEvent *event;
5375
5376       GST_DEBUG_OBJECT (demux, "sending EOS for stream %s",
5377           GST_PAD_NAME (stream->pad));
5378       stream->sent_eos = TRUE;
5379       event = gst_event_new_eos ();
5380       if (demux->segment_seqnum != GST_SEQNUM_INVALID)
5381         gst_event_set_seqnum (event, demux->segment_seqnum);
5382       gst_pad_push_event (stream->pad, event);
5383     }
5384   }
5385 }
5386
5387 /* EOS and NOT_LINKED need to be combined. This means that we return:
5388  *
5389  *  GST_FLOW_NOT_LINKED: when all pads NOT_LINKED.
5390  *  GST_FLOW_EOS: when all pads EOS or NOT_LINKED.
5391  */
5392 static GstFlowReturn
5393 gst_qtdemux_combine_flows (GstQTDemux * demux, QtDemuxStream * stream,
5394     GstFlowReturn ret)
5395 {
5396   GST_LOG_OBJECT (demux, "flow return: %s", gst_flow_get_name (ret));
5397
5398   if (stream->pad)
5399     ret = gst_flow_combiner_update_pad_flow (demux->flowcombiner, stream->pad,
5400         ret);
5401   else
5402     ret = gst_flow_combiner_update_flow (demux->flowcombiner, ret);
5403
5404   GST_LOG_OBJECT (demux, "combined flow return: %s", gst_flow_get_name (ret));
5405   return ret;
5406 }
5407
5408 /* the input buffer metadata must be writable. Returns NULL when the buffer is
5409  * completely clipped
5410  *
5411  * Should be used only with raw buffers */
5412 static GstBuffer *
5413 gst_qtdemux_clip_buffer (GstQTDemux * qtdemux, QtDemuxStream * stream,
5414     GstBuffer * buf)
5415 {
5416   guint64 start, stop, cstart, cstop, diff;
5417   GstClockTime pts, duration;
5418   gsize size, osize;
5419   gint num_rate, denom_rate;
5420   gint frame_size;
5421   gboolean clip_data;
5422   guint offset;
5423
5424   osize = size = gst_buffer_get_size (buf);
5425   offset = 0;
5426
5427   /* depending on the type, setup the clip parameters */
5428   if (stream->subtype == FOURCC_soun) {
5429     frame_size = CUR_STREAM (stream)->bytes_per_frame;
5430     num_rate = GST_SECOND;
5431     denom_rate = (gint) CUR_STREAM (stream)->rate;
5432     clip_data = TRUE;
5433   } else if (stream->subtype == FOURCC_vide) {
5434     frame_size = size;
5435     num_rate = CUR_STREAM (stream)->fps_n;
5436     denom_rate = CUR_STREAM (stream)->fps_d;
5437     clip_data = FALSE;
5438   } else
5439     goto wrong_type;
5440
5441   if (frame_size <= 0)
5442     goto bad_frame_size;
5443
5444   /* we can only clip if we have a valid pts */
5445   pts = GST_BUFFER_PTS (buf);
5446   if (G_UNLIKELY (!GST_CLOCK_TIME_IS_VALID (pts)))
5447     goto no_pts;
5448
5449   duration = GST_BUFFER_DURATION (buf);
5450
5451   if (G_UNLIKELY (!GST_CLOCK_TIME_IS_VALID (duration))) {
5452     duration =
5453         gst_util_uint64_scale_int (size / frame_size, num_rate, denom_rate);
5454   }
5455
5456   start = pts;
5457   stop = start + duration;
5458
5459   if (G_UNLIKELY (!gst_segment_clip (&stream->segment,
5460               GST_FORMAT_TIME, start, stop, &cstart, &cstop)))
5461     goto clipped;
5462
5463   /* see if some clipping happened */
5464   diff = cstart - start;
5465   if (diff > 0) {
5466     pts += diff;
5467     duration -= diff;
5468
5469     if (clip_data) {
5470       /* bring clipped time to samples and to bytes */
5471       diff = gst_util_uint64_scale_int (diff, denom_rate, num_rate);
5472       diff *= frame_size;
5473
5474       GST_DEBUG_OBJECT (qtdemux,
5475           "clipping start to %" GST_TIME_FORMAT " %"
5476           G_GUINT64_FORMAT " bytes", GST_TIME_ARGS (cstart), diff);
5477
5478       offset = diff;
5479       size -= diff;
5480     }
5481   }
5482   diff = stop - cstop;
5483   if (diff > 0) {
5484     duration -= diff;
5485
5486     if (clip_data) {
5487       /* bring clipped time to samples and then to bytes */
5488       diff = gst_util_uint64_scale_int (diff, denom_rate, num_rate);
5489       diff *= frame_size;
5490       GST_DEBUG_OBJECT (qtdemux,
5491           "clipping stop to %" GST_TIME_FORMAT " %" G_GUINT64_FORMAT
5492           " bytes", GST_TIME_ARGS (cstop), diff);
5493       size -= diff;
5494     }
5495   }
5496
5497   if (offset != 0 || size != osize)
5498     gst_buffer_resize (buf, offset, size);
5499
5500   GST_BUFFER_DTS (buf) = GST_CLOCK_TIME_NONE;
5501   GST_BUFFER_PTS (buf) = pts;
5502   GST_BUFFER_DURATION (buf) = duration;
5503
5504   return buf;
5505
5506   /* dropped buffer */
5507 wrong_type:
5508   {
5509     GST_DEBUG_OBJECT (qtdemux, "unknown stream type");
5510     return buf;
5511   }
5512 bad_frame_size:
5513   {
5514     GST_DEBUG_OBJECT (qtdemux, "bad frame size");
5515     return buf;
5516   }
5517 no_pts:
5518   {
5519     GST_DEBUG_OBJECT (qtdemux, "no pts on buffer");
5520     return buf;
5521   }
5522 clipped:
5523   {
5524     GST_DEBUG_OBJECT (qtdemux, "clipped buffer");
5525     gst_buffer_unref (buf);
5526     return NULL;
5527   }
5528 }
5529
5530 static GstBuffer *
5531 gst_qtdemux_align_buffer (GstQTDemux * demux,
5532     GstBuffer * buffer, gsize alignment)
5533 {
5534   GstMapInfo map;
5535
5536   gst_buffer_map (buffer, &map, GST_MAP_READ);
5537
5538   if (map.size < sizeof (guintptr)) {
5539     gst_buffer_unmap (buffer, &map);
5540     return buffer;
5541   }
5542
5543   if (((guintptr) map.data) & (alignment - 1)) {
5544     GstBuffer *new_buffer;
5545     GstAllocationParams params = { 0, alignment - 1, 0, 0, };
5546
5547     new_buffer = gst_buffer_new_allocate (NULL,
5548         gst_buffer_get_size (buffer), &params);
5549
5550     /* Copy data "by hand", so ensure alignment is kept: */
5551     gst_buffer_fill (new_buffer, 0, map.data, map.size);
5552
5553     gst_buffer_copy_into (new_buffer, buffer, GST_BUFFER_COPY_METADATA, 0, -1);
5554     GST_DEBUG_OBJECT (demux,
5555         "We want output aligned on %" G_GSIZE_FORMAT ", reallocated",
5556         alignment);
5557
5558     gst_buffer_unmap (buffer, &map);
5559     gst_buffer_unref (buffer);
5560
5561     return new_buffer;
5562   }
5563
5564   gst_buffer_unmap (buffer, &map);
5565   return buffer;
5566 }
5567
5568 static guint8 *
5569 convert_to_s334_1a (const guint8 * ccpair, guint8 ccpair_size, guint field,
5570     gsize * res)
5571 {
5572   guint8 *storage;
5573   gsize i;
5574
5575   /* We are converting from pairs to triplets */
5576   *res = ccpair_size / 2 * 3;
5577   storage = g_malloc (*res);
5578   for (i = 0; i * 2 < ccpair_size; i += 1) {
5579     /* FIXME: Use line offset 0 as we simply can't know here */
5580     if (field == 1)
5581       storage[i * 3] = 0x80 | 0x00;
5582     else
5583       storage[i * 3] = 0x00 | 0x00;
5584     storage[i * 3 + 1] = ccpair[i * 2];
5585     storage[i * 3 + 2] = ccpair[i * 2 + 1];
5586   }
5587
5588   return storage;
5589 }
5590
5591 static guint8 *
5592 extract_cc_from_data (QtDemuxStream * stream, const guint8 * data, gsize size,
5593     gsize * cclen)
5594 {
5595   guint8 *res = NULL;
5596   guint32 atom_length, fourcc;
5597   QtDemuxStreamStsdEntry *stsd_entry;
5598
5599   GST_MEMDUMP ("caption atom", data, size);
5600
5601   /* There might be multiple atoms */
5602
5603   *cclen = 0;
5604   if (size < 8)
5605     goto invalid_cdat;
5606   atom_length = QT_UINT32 (data);
5607   fourcc = QT_FOURCC (data + 4);
5608   if (G_UNLIKELY (atom_length > size || atom_length == 8))
5609     goto invalid_cdat;
5610
5611   GST_DEBUG_OBJECT (stream->pad, "here");
5612
5613   /* Check if we have something compatible */
5614   stsd_entry = CUR_STREAM (stream);
5615   switch (stsd_entry->fourcc) {
5616     case FOURCC_c608:{
5617       guint8 *cdat = NULL, *cdt2 = NULL;
5618       gsize cdat_size = 0, cdt2_size = 0;
5619       /* Should be cdat or cdt2 */
5620       if (fourcc != FOURCC_cdat && fourcc != FOURCC_cdt2) {
5621         GST_WARNING_OBJECT (stream->pad,
5622             "Unknown data atom (%" GST_FOURCC_FORMAT ") for CEA608",
5623             GST_FOURCC_ARGS (fourcc));
5624         goto invalid_cdat;
5625       }
5626
5627       /* Convert to S334-1 Annex A byte triplet */
5628       if (fourcc == FOURCC_cdat)
5629         cdat = convert_to_s334_1a (data + 8, atom_length - 8, 1, &cdat_size);
5630       else
5631         cdt2 = convert_to_s334_1a (data + 8, atom_length - 8, 2, &cdt2_size);
5632       GST_DEBUG_OBJECT (stream->pad, "size:%" G_GSIZE_FORMAT " atom_length:%u",
5633           size, atom_length);
5634
5635       /* Check for another atom ? */
5636       if (size > atom_length + 8) {
5637         guint32 new_atom_length = QT_UINT32 (data + atom_length);
5638         if (size >= atom_length + new_atom_length) {
5639           fourcc = QT_FOURCC (data + atom_length + 4);
5640           if (fourcc == FOURCC_cdat) {
5641             if (cdat == NULL)
5642               cdat =
5643                   convert_to_s334_1a (data + atom_length + 8,
5644                   new_atom_length - 8, 1, &cdat_size);
5645             else
5646               GST_WARNING_OBJECT (stream->pad,
5647                   "Got multiple [cdat] atoms in a c608 sample. This is unsupported for now. Please file a bug");
5648           } else {
5649             if (cdt2 == NULL)
5650               cdt2 =
5651                   convert_to_s334_1a (data + atom_length + 8,
5652                   new_atom_length - 8, 2, &cdt2_size);
5653             else
5654               GST_WARNING_OBJECT (stream->pad,
5655                   "Got multiple [cdt2] atoms in a c608 sample. This is unsupported for now. Please file a bug");
5656           }
5657         }
5658       }
5659
5660       *cclen = cdat_size + cdt2_size;
5661       res = g_malloc (*cclen);
5662       if (cdat_size)
5663         memcpy (res, cdat, cdat_size);
5664       if (cdt2_size)
5665         memcpy (res + cdat_size, cdt2, cdt2_size);
5666       g_free (cdat);
5667       g_free (cdt2);
5668     }
5669       break;
5670     case FOURCC_c708:
5671       if (fourcc != FOURCC_ccdp) {
5672         GST_WARNING_OBJECT (stream->pad,
5673             "Unknown data atom (%" GST_FOURCC_FORMAT ") for CEA708",
5674             GST_FOURCC_ARGS (fourcc));
5675         goto invalid_cdat;
5676       }
5677       *cclen = atom_length - 8;
5678       res = g_memdup2 (data + 8, *cclen);
5679       break;
5680     default:
5681       /* Keep this here in case other closed caption formats are added */
5682       g_assert_not_reached ();
5683       break;
5684   }
5685
5686   GST_MEMDUMP ("Output", res, *cclen);
5687   return res;
5688
5689   /* Errors */
5690 invalid_cdat:
5691   GST_WARNING ("[cdat] atom is too small or invalid");
5692   return NULL;
5693 }
5694
5695 /* Handle Closed Caption sample buffers.
5696  * The input buffer metadata must be writable,
5697  * but time/duration etc not yet set and need not be preserved */
5698 static GstBuffer *
5699 gst_qtdemux_process_buffer_clcp (GstQTDemux * qtdemux, QtDemuxStream * stream,
5700     GstBuffer * buf)
5701 {
5702   GstBuffer *outbuf = NULL;
5703   GstMapInfo map;
5704   guint8 *cc;
5705   gsize cclen = 0;
5706
5707   gst_buffer_map (buf, &map, GST_MAP_READ);
5708
5709   /* empty buffer is sent to terminate previous subtitle */
5710   if (map.size <= 2) {
5711     gst_buffer_unmap (buf, &map);
5712     gst_buffer_unref (buf);
5713     return NULL;
5714   }
5715
5716   /* For closed caption, we need to extract the information from the
5717    * [cdat],[cdt2] or [ccdp] atom */
5718   cc = extract_cc_from_data (stream, map.data, map.size, &cclen);
5719   gst_buffer_unmap (buf, &map);
5720   if (cc) {
5721     outbuf = _gst_buffer_new_wrapped (cc, cclen, g_free);
5722     gst_buffer_copy_into (outbuf, buf, GST_BUFFER_COPY_METADATA, 0, -1);
5723   } else {
5724     /* Conversion failed or there's nothing */
5725   }
5726   gst_buffer_unref (buf);
5727
5728   return outbuf;
5729 }
5730
5731 /* DVD subpicture specific sample handling.
5732  * the input buffer metadata must be writable,
5733  * but time/duration etc not yet set and need not be preserved */
5734 static GstBuffer *
5735 gst_qtdemux_process_buffer_dvd (GstQTDemux * qtdemux, QtDemuxStream * stream,
5736     GstBuffer * buf)
5737 {
5738   /* send a one time dvd clut event */
5739   if (stream->pending_event && stream->pad)
5740     gst_pad_push_event (stream->pad, stream->pending_event);
5741   stream->pending_event = NULL;
5742
5743   /* empty buffer is sent to terminate previous subtitle */
5744   if (gst_buffer_get_size (buf) <= 2) {
5745     gst_buffer_unref (buf);
5746     return NULL;
5747   }
5748
5749   /* That's all the processing needed for subpictures */
5750   return buf;
5751 }
5752
5753 /* Timed text formats
5754  * the input buffer metadata must be writable,
5755  * but time/duration etc not yet set and need not be preserved */
5756 static GstBuffer *
5757 gst_qtdemux_process_buffer_text (GstQTDemux * qtdemux, QtDemuxStream * stream,
5758     GstBuffer * buf)
5759 {
5760   GstBuffer *outbuf = NULL;
5761   GstMapInfo map;
5762   guint nsize = 0;
5763   gchar *str;
5764
5765   /* not many cases for now */
5766   if (G_UNLIKELY (stream->subtype != FOURCC_text &&
5767           stream->subtype != FOURCC_sbtl)) {
5768     return buf;
5769   }
5770
5771   gst_buffer_map (buf, &map, GST_MAP_READ);
5772
5773   /* empty buffer is sent to terminate previous subtitle */
5774   if (map.size <= 2) {
5775     gst_buffer_unmap (buf, &map);
5776     gst_buffer_unref (buf);
5777     return NULL;
5778   }
5779
5780   nsize = GST_READ_UINT16_BE (map.data);
5781   nsize = MIN (nsize, map.size - 2);
5782
5783   GST_LOG_OBJECT (qtdemux, "3GPP timed text subtitle: %d/%" G_GSIZE_FORMAT "",
5784       nsize, map.size);
5785
5786   /* takes care of UTF-8 validation or UTF-16 recognition,
5787    * no other encoding expected */
5788   str = gst_tag_freeform_string_to_utf8 ((gchar *) map.data + 2, nsize, NULL);
5789   gst_buffer_unmap (buf, &map);
5790
5791   if (str) {
5792     outbuf = _gst_buffer_new_wrapped (str, strlen (str), g_free);
5793     gst_buffer_copy_into (outbuf, buf, GST_BUFFER_COPY_METADATA, 0, -1);
5794   } else {
5795     /* this should not really happen unless the subtitle is corrupted */
5796   }
5797   gst_buffer_unref (buf);
5798
5799   /* FIXME ? convert optional subsequent style info to markup */
5800
5801   return outbuf;
5802 }
5803
5804 /* WebVTT sample handling according to 14496-30 */
5805 static GstBuffer *
5806 gst_qtdemux_process_buffer_wvtt (GstQTDemux * qtdemux, QtDemuxStream * stream,
5807     GstBuffer * buf)
5808 {
5809   GstBuffer *outbuf = NULL;
5810   GstMapInfo map;
5811
5812   if (!gst_buffer_map (buf, &map, GST_MAP_READ)) {
5813     g_assert_not_reached ();    /* The buffer must be mappable */
5814   }
5815
5816   if (qtdemux_webvtt_is_empty (qtdemux, map.data, map.size)) {
5817     GstEvent *gap = NULL;
5818     /* Push a gap event */
5819     stream->segment.position = GST_BUFFER_PTS (buf);
5820     gap =
5821         gst_event_new_gap (stream->segment.position, GST_BUFFER_DURATION (buf));
5822     gst_pad_push_event (stream->pad, gap);
5823
5824     if (GST_BUFFER_DURATION_IS_VALID (buf))
5825       stream->segment.position += GST_BUFFER_DURATION (buf);
5826   } else {
5827     outbuf =
5828         qtdemux_webvtt_decode (qtdemux, GST_BUFFER_PTS (buf),
5829         GST_BUFFER_DURATION (buf), map.data, map.size);
5830     gst_buffer_copy_into (outbuf, buf, GST_BUFFER_COPY_METADATA, 0, -1);
5831   }
5832
5833   gst_buffer_unmap (buf, &map);
5834   gst_buffer_unref (buf);
5835
5836   return outbuf;
5837 }
5838
5839 static GstFlowReturn
5840 gst_qtdemux_push_buffer (GstQTDemux * qtdemux, QtDemuxStream * stream,
5841     GstBuffer * buf)
5842 {
5843   GstFlowReturn ret = GST_FLOW_OK;
5844   GstClockTime pts, duration;
5845
5846   if (stream->need_clip)
5847     buf = gst_qtdemux_clip_buffer (qtdemux, stream, buf);
5848
5849   if (G_UNLIKELY (buf == NULL))
5850     goto exit;
5851
5852   if (G_UNLIKELY (stream->discont)) {
5853     GST_LOG_OBJECT (qtdemux, "marking discont buffer");
5854     GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_DISCONT);
5855     stream->discont = FALSE;
5856   } else {
5857     GST_BUFFER_FLAG_UNSET (buf, GST_BUFFER_FLAG_DISCONT);
5858   }
5859
5860   GST_LOG_OBJECT (qtdemux,
5861       "Pushing buffer with dts %" GST_TIME_FORMAT ", pts %" GST_TIME_FORMAT
5862       ", duration %" GST_TIME_FORMAT " on pad %s",
5863       GST_TIME_ARGS (GST_BUFFER_DTS (buf)),
5864       GST_TIME_ARGS (GST_BUFFER_PTS (buf)),
5865       GST_TIME_ARGS (GST_BUFFER_DURATION (buf)), GST_PAD_NAME (stream->pad));
5866
5867   if (stream->protected && stream->protection_scheme_type == FOURCC_aavd) {
5868     GstStructure *crypto_info;
5869     QtDemuxAavdEncryptionInfo *info =
5870         (QtDemuxAavdEncryptionInfo *) stream->protection_scheme_info;
5871
5872     crypto_info = gst_structure_copy (info->default_properties);
5873     if (!crypto_info || !gst_buffer_add_protection_meta (buf, crypto_info))
5874       GST_ERROR_OBJECT (qtdemux, "failed to attach aavd metadata to buffer");
5875   }
5876
5877   if (stream->protected && (stream->protection_scheme_type == FOURCC_cenc
5878           || stream->protection_scheme_type == FOURCC_cbcs)) {
5879     GstStructure *crypto_info;
5880     QtDemuxCencSampleSetInfo *info =
5881         (QtDemuxCencSampleSetInfo *) stream->protection_scheme_info;
5882     gint index;
5883     GstEvent *event;
5884
5885     while ((event = g_queue_pop_head (&stream->protection_scheme_event_queue))) {
5886       GST_TRACE_OBJECT (stream->pad, "pushing protection event: %"
5887           GST_PTR_FORMAT, event);
5888       gst_pad_push_event (stream->pad, event);
5889     }
5890
5891     if (info->crypto_info == NULL) {
5892       if (stream->protection_scheme_type == FOURCC_cbcs) {
5893         crypto_info = qtdemux_get_cenc_sample_properties (qtdemux, stream, 0);
5894         if (!crypto_info || !gst_buffer_add_protection_meta (buf, crypto_info)) {
5895           GST_ERROR_OBJECT (qtdemux,
5896               "failed to attach cbcs metadata to buffer");
5897           qtdemux_gst_structure_free (crypto_info);
5898         } else {
5899           GST_TRACE_OBJECT (qtdemux, "added cbcs protection metadata");
5900         }
5901       } else {
5902         GST_DEBUG_OBJECT (qtdemux,
5903             "cenc metadata hasn't been parsed yet, pushing buffer as if it wasn't encrypted");
5904       }
5905     } else {
5906       /* The end of the crypto_info array matches our n_samples position,
5907        * so count backward from there */
5908       index = stream->sample_index - stream->n_samples + info->crypto_info->len;
5909       if (G_LIKELY (index >= 0 && index < info->crypto_info->len)) {
5910         /* steal structure from array */
5911         crypto_info = g_ptr_array_index (info->crypto_info, index);
5912         g_ptr_array_index (info->crypto_info, index) = NULL;
5913         GST_LOG_OBJECT (qtdemux, "attaching cenc metadata [%u/%u]", index,
5914             info->crypto_info->len);
5915         if (!crypto_info || !gst_buffer_add_protection_meta (buf, crypto_info))
5916           GST_ERROR_OBJECT (qtdemux,
5917               "failed to attach cenc metadata to buffer");
5918       } else {
5919         GST_INFO_OBJECT (qtdemux, "No crypto info with index %d and sample %d",
5920             index, stream->sample_index);
5921       }
5922     }
5923   }
5924
5925   if (stream->alignment > 1)
5926     buf = gst_qtdemux_align_buffer (qtdemux, buf, stream->alignment);
5927
5928   pts = GST_BUFFER_PTS (buf);
5929   duration = GST_BUFFER_DURATION (buf);
5930
5931   ret = gst_pad_push (stream->pad, buf);
5932
5933   if (GST_CLOCK_TIME_IS_VALID (pts) && GST_CLOCK_TIME_IS_VALID (duration)) {
5934     /* mark position in stream, we'll need this to know when to send GAP event */
5935     stream->segment.position = pts + duration;
5936   }
5937
5938 exit:
5939
5940   return ret;
5941 }
5942
5943 static GstFlowReturn
5944 gst_qtdemux_split_and_push_buffer (GstQTDemux * qtdemux, QtDemuxStream * stream,
5945     GstBuffer * buf)
5946 {
5947   GstFlowReturn ret = GST_FLOW_OK;
5948
5949   if (stream->subtype == FOURCC_clcp
5950       && CUR_STREAM (stream)->fourcc == FOURCC_c608 && stream->need_split) {
5951     GstMapInfo map;
5952     guint n_output_buffers, n_field1 = 0, n_field2 = 0;
5953     guint n_triplets, i;
5954     guint field1_off = 0, field2_off = 0;
5955
5956     /* We have to split CEA608 buffers so that each outgoing buffer contains
5957      * one byte pair per field according to the framerate of the video track.
5958      *
5959      * If there is only a single byte pair per field we don't have to do
5960      * anything
5961      */
5962
5963     gst_buffer_map (buf, &map, GST_MAP_READ);
5964
5965     n_triplets = map.size / 3;
5966     for (i = 0; i < n_triplets; i++) {
5967       if (map.data[3 * i] & 0x80)
5968         n_field1++;
5969       else
5970         n_field2++;
5971     }
5972
5973     g_assert (n_field1 || n_field2);
5974
5975     /* If there's more than 1 frame we have to split, otherwise we can just
5976      * pass through */
5977     if (n_field1 > 1 || n_field2 > 1) {
5978       n_output_buffers =
5979           gst_util_uint64_scale (GST_BUFFER_DURATION (buf),
5980           CUR_STREAM (stream)->fps_n, GST_SECOND * CUR_STREAM (stream)->fps_d);
5981
5982       for (i = 0; i < n_output_buffers; i++) {
5983         GstBuffer *outbuf =
5984             gst_buffer_new_and_alloc ((n_field1 ? 3 : 0) + (n_field2 ? 3 : 0));
5985         GstMapInfo outmap;
5986         guint8 *outptr;
5987
5988         gst_buffer_map (outbuf, &outmap, GST_MAP_WRITE);
5989         outptr = outmap.data;
5990
5991         if (n_field1) {
5992           gboolean found = FALSE;
5993
5994           while (map.data + field1_off < map.data + map.size) {
5995             if (map.data[field1_off] & 0x80) {
5996               memcpy (outptr, &map.data[field1_off], 3);
5997               field1_off += 3;
5998               found = TRUE;
5999               break;
6000             }
6001             field1_off += 3;
6002           }
6003
6004           if (!found) {
6005             const guint8 empty[] = { 0x80, 0x80, 0x80 };
6006
6007             memcpy (outptr, empty, 3);
6008           }
6009
6010           outptr += 3;
6011         }
6012
6013         if (n_field2) {
6014           gboolean found = FALSE;
6015
6016           while (map.data + field2_off < map.data + map.size) {
6017             if ((map.data[field2_off] & 0x80) == 0) {
6018               memcpy (outptr, &map.data[field2_off], 3);
6019               field2_off += 3;
6020               found = TRUE;
6021               break;
6022             }
6023             field2_off += 3;
6024           }
6025
6026           if (!found) {
6027             const guint8 empty[] = { 0x00, 0x80, 0x80 };
6028
6029             memcpy (outptr, empty, 3);
6030           }
6031
6032           outptr += 3;
6033         }
6034
6035         gst_buffer_unmap (outbuf, &outmap);
6036
6037         GST_BUFFER_PTS (outbuf) =
6038             GST_BUFFER_PTS (buf) + gst_util_uint64_scale (i,
6039             GST_SECOND * CUR_STREAM (stream)->fps_d,
6040             CUR_STREAM (stream)->fps_n);
6041         GST_BUFFER_DURATION (outbuf) =
6042             gst_util_uint64_scale (GST_SECOND, CUR_STREAM (stream)->fps_d,
6043             CUR_STREAM (stream)->fps_n);
6044         GST_BUFFER_OFFSET (outbuf) = -1;
6045         GST_BUFFER_OFFSET_END (outbuf) = -1;
6046
6047         ret = gst_qtdemux_push_buffer (qtdemux, stream, outbuf);
6048
6049         if (ret != GST_FLOW_OK && ret != GST_FLOW_NOT_LINKED)
6050           break;
6051       }
6052       gst_buffer_unmap (buf, &map);
6053       gst_buffer_unref (buf);
6054     } else {
6055       gst_buffer_unmap (buf, &map);
6056       ret = gst_qtdemux_push_buffer (qtdemux, stream, buf);
6057     }
6058   } else {
6059     ret = gst_qtdemux_push_buffer (qtdemux, stream, buf);
6060   }
6061
6062   return ret;
6063 }
6064
6065 /* Sets a buffer's attributes properly and pushes it downstream.
6066  * Also checks for additional actions and custom processing that may
6067  * need to be done first.
6068  */
6069 static GstFlowReturn
6070 gst_qtdemux_decorate_and_push_buffer (GstQTDemux * qtdemux,
6071     QtDemuxStream * stream, GstBuffer * buf,
6072     GstClockTime dts, GstClockTime pts, GstClockTime duration,
6073     gboolean keyframe, GstClockTime position, guint64 byte_position)
6074 {
6075   GstFlowReturn ret = GST_FLOW_OK;
6076
6077   /* offset the timestamps according to the edit list */
6078
6079   if (G_UNLIKELY (CUR_STREAM (stream)->fourcc == FOURCC_rtsp)) {
6080     gchar *url;
6081     GstMapInfo map;
6082
6083     gst_buffer_map (buf, &map, GST_MAP_READ);
6084     url = g_strndup ((gchar *) map.data, map.size);
6085     gst_buffer_unmap (buf, &map);
6086     if (url != NULL && strlen (url) != 0) {
6087       /* we have RTSP redirect now */
6088       g_free (qtdemux->redirect_location);
6089       qtdemux->redirect_location = g_strdup (url);
6090       gst_element_post_message (GST_ELEMENT_CAST (qtdemux),
6091           gst_message_new_element (GST_OBJECT_CAST (qtdemux),
6092               gst_structure_new ("redirect",
6093                   "new-location", G_TYPE_STRING, url, NULL)));
6094     } else {
6095       GST_WARNING_OBJECT (qtdemux, "Redirect URI of stream is empty, not "
6096           "posting");
6097     }
6098     g_free (url);
6099   }
6100
6101   /* position reporting */
6102   if (qtdemux->segment.rate >= 0) {
6103     qtdemux->segment.position = position;
6104     gst_qtdemux_sync_streams (qtdemux);
6105   }
6106
6107   if (G_UNLIKELY (!stream->pad)) {
6108     GST_DEBUG_OBJECT (qtdemux, "No output pad for stream, ignoring");
6109     gst_buffer_unref (buf);
6110     goto exit;
6111   }
6112
6113   /* send out pending buffers */
6114   while (stream->buffers) {
6115     GstBuffer *buffer = (GstBuffer *) stream->buffers->data;
6116
6117     if (G_UNLIKELY (stream->discont)) {
6118       GST_LOG_OBJECT (qtdemux, "marking discont buffer");
6119       GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_DISCONT);
6120       stream->discont = FALSE;
6121     } else {
6122       GST_BUFFER_FLAG_UNSET (buf, GST_BUFFER_FLAG_DISCONT);
6123     }
6124
6125     if (stream->alignment > 1)
6126       buffer = gst_qtdemux_align_buffer (qtdemux, buffer, stream->alignment);
6127     gst_pad_push (stream->pad, buffer);
6128
6129     stream->buffers = g_slist_delete_link (stream->buffers, stream->buffers);
6130   }
6131
6132   /* we're going to modify the metadata */
6133   buf = gst_buffer_make_writable (buf);
6134
6135   GST_BUFFER_DTS (buf) = dts;
6136   GST_BUFFER_PTS (buf) = pts;
6137   GST_BUFFER_DURATION (buf) = duration;
6138   GST_BUFFER_OFFSET (buf) = -1;
6139   GST_BUFFER_OFFSET_END (buf) = -1;
6140
6141   if (G_UNLIKELY (stream->process_func))
6142     buf = stream->process_func (qtdemux, stream, buf);
6143
6144   if (!buf) {
6145     goto exit;
6146   }
6147
6148   if (!keyframe) {
6149     GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_DELTA_UNIT);
6150     stream->on_keyframe = FALSE;
6151   } else {
6152     stream->on_keyframe = TRUE;
6153   }
6154
6155   if (G_UNLIKELY (CUR_STREAM (stream)->rgb8_palette))
6156     gst_buffer_append_memory (buf,
6157         gst_memory_ref (CUR_STREAM (stream)->rgb8_palette));
6158
6159   if (G_UNLIKELY (CUR_STREAM (stream)->padding)) {
6160     gst_buffer_resize (buf, CUR_STREAM (stream)->padding, -1);
6161   }
6162 #if 0
6163   if (G_UNLIKELY (qtdemux->element_index)) {
6164     GstClockTime stream_time;
6165
6166     stream_time =
6167         gst_segment_to_stream_time (&stream->segment, GST_FORMAT_TIME,
6168         timestamp);
6169     if (GST_CLOCK_TIME_IS_VALID (stream_time)) {
6170       GST_LOG_OBJECT (qtdemux,
6171           "adding association %" GST_TIME_FORMAT "-> %"
6172           G_GUINT64_FORMAT, GST_TIME_ARGS (stream_time), byte_position);
6173       gst_index_add_association (qtdemux->element_index,
6174           qtdemux->index_id,
6175           keyframe ? GST_ASSOCIATION_FLAG_KEY_UNIT :
6176           GST_ASSOCIATION_FLAG_DELTA_UNIT, GST_FORMAT_TIME, stream_time,
6177           GST_FORMAT_BYTES, byte_position, NULL);
6178     }
6179   }
6180 #endif
6181
6182   ret = gst_qtdemux_split_and_push_buffer (qtdemux, stream, buf);
6183
6184 exit:
6185   return ret;
6186 }
6187
6188 static const QtDemuxRandomAccessEntry *
6189 gst_qtdemux_stream_seek_fragment (GstQTDemux * qtdemux, QtDemuxStream * stream,
6190     GstClockTime pos, gboolean after)
6191 {
6192   QtDemuxRandomAccessEntry *entries = stream->ra_entries;
6193   guint n_entries = stream->n_ra_entries;
6194   guint i;
6195
6196   /* we assume the table is sorted */
6197   for (i = 0; i < n_entries; ++i) {
6198     if (entries[i].ts > pos)
6199       break;
6200   }
6201
6202   /* FIXME: maybe save first moof_offset somewhere instead, but for now it's
6203    * probably okay to assume that the index lists the very first fragment */
6204   if (i == 0)
6205     return &entries[0];
6206
6207   if (after)
6208     return &entries[i];
6209   else
6210     return &entries[i - 1];
6211 }
6212
6213 static gboolean
6214 gst_qtdemux_do_fragmented_seek (GstQTDemux * qtdemux)
6215 {
6216   const QtDemuxRandomAccessEntry *best_entry = NULL;
6217   gint i;
6218
6219   GST_OBJECT_LOCK (qtdemux);
6220
6221   g_assert (QTDEMUX_N_STREAMS (qtdemux) > 0);
6222
6223   /* first see if we can determine where to go to using mfra,
6224    * before we start clearing things */
6225   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
6226     const QtDemuxRandomAccessEntry *entry;
6227     QtDemuxStream *stream;
6228     gboolean is_audio_or_video;
6229
6230     stream = QTDEMUX_NTH_STREAM (qtdemux, i);
6231
6232     if (stream->ra_entries == NULL)
6233       continue;
6234
6235     if (stream->subtype == FOURCC_vide || stream->subtype == FOURCC_soun)
6236       is_audio_or_video = TRUE;
6237     else
6238       is_audio_or_video = FALSE;
6239
6240     entry =
6241         gst_qtdemux_stream_seek_fragment (qtdemux, stream,
6242         stream->time_position, !is_audio_or_video);
6243
6244     GST_INFO_OBJECT (stream->pad, "%" GST_TIME_FORMAT " at offset "
6245         "%" G_GUINT64_FORMAT, GST_TIME_ARGS (entry->ts), entry->moof_offset);
6246
6247     stream->pending_seek = entry;
6248
6249     /* decide position to jump to just based on audio/video tracks, not subs */
6250     if (!is_audio_or_video)
6251       continue;
6252
6253     if (best_entry == NULL || entry->moof_offset < best_entry->moof_offset)
6254       best_entry = entry;
6255   }
6256
6257   /* no luck, will handle seek otherwise */
6258   if (best_entry == NULL) {
6259     GST_OBJECT_UNLOCK (qtdemux);
6260     return FALSE;
6261   }
6262
6263   /* ok, now we can prepare for processing as of located moof */
6264   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
6265     QtDemuxStream *stream;
6266
6267     stream = QTDEMUX_NTH_STREAM (qtdemux, i);
6268
6269     g_free (stream->samples);
6270     stream->samples = NULL;
6271     stream->n_samples = 0;
6272     stream->stbl_index = -1;    /* no samples have yet been parsed */
6273     stream->sample_index = -1;
6274
6275     if (stream->protection_scheme_info) {
6276       /* Clear out any old cenc crypto info entries as we'll move to a new moof */
6277       if (stream->protection_scheme_type == FOURCC_cenc
6278           || stream->protection_scheme_type == FOURCC_cbcs) {
6279         QtDemuxCencSampleSetInfo *info =
6280             (QtDemuxCencSampleSetInfo *) stream->protection_scheme_info;
6281         if (info->crypto_info) {
6282           g_ptr_array_free (info->crypto_info, TRUE);
6283           info->crypto_info = NULL;
6284         }
6285       }
6286     }
6287   }
6288
6289   GST_INFO_OBJECT (qtdemux, "seek to %" GST_TIME_FORMAT ", best fragment "
6290       "moof offset: %" G_GUINT64_FORMAT ", ts %" GST_TIME_FORMAT,
6291       GST_TIME_ARGS (QTDEMUX_NTH_STREAM (qtdemux, 0)->time_position),
6292       best_entry->moof_offset, GST_TIME_ARGS (best_entry->ts));
6293
6294   qtdemux->moof_offset = best_entry->moof_offset;
6295
6296   qtdemux_add_fragmented_samples (qtdemux);
6297
6298   GST_OBJECT_UNLOCK (qtdemux);
6299   return TRUE;
6300 }
6301
6302 static GstFlowReturn
6303 gst_qtdemux_loop_state_movie (GstQTDemux * qtdemux)
6304 {
6305   GstFlowReturn ret = GST_FLOW_OK;
6306   GstBuffer *buf = NULL;
6307   QtDemuxStream *stream, *target_stream = NULL;
6308   GstClockTime min_time;
6309   guint64 offset = 0;
6310   GstClockTime dts = GST_CLOCK_TIME_NONE;
6311   GstClockTime pts = GST_CLOCK_TIME_NONE;
6312   GstClockTime duration = 0;
6313   gboolean keyframe = FALSE;
6314   guint sample_size = 0;
6315   guint num_samples = 1;
6316   gboolean empty = 0;
6317   guint size;
6318   gint i;
6319
6320   if (qtdemux->fragmented_seek_pending) {
6321     GST_INFO_OBJECT (qtdemux, "pending fragmented seek");
6322     if (gst_qtdemux_do_fragmented_seek (qtdemux)) {
6323       GST_INFO_OBJECT (qtdemux, "fragmented seek done!");
6324       qtdemux->fragmented_seek_pending = FALSE;
6325     } else {
6326       GST_INFO_OBJECT (qtdemux, "fragmented seek still pending");
6327     }
6328   }
6329
6330   /* Figure out the next stream sample to output, min_time is expressed in
6331    * global time and runs over the edit list segments. */
6332   min_time = G_MAXUINT64;
6333   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
6334     GstClockTime position;
6335
6336     stream = QTDEMUX_NTH_STREAM (qtdemux, i);
6337     position = stream->time_position;
6338
6339     if (!GST_CLOCK_TIME_IS_VALID (position))
6340       continue;
6341
6342     if (stream->segment_index != -1) {
6343       QtDemuxSegment *segment = &stream->segments[stream->segment_index];
6344       position += segment->media_start;
6345     }
6346
6347     /* position of -1 is EOS */
6348     if (position < min_time) {
6349       min_time = position;
6350       target_stream = stream;
6351     }
6352   }
6353   /* all are EOS */
6354   if (G_UNLIKELY (target_stream == NULL)) {
6355     GST_DEBUG_OBJECT (qtdemux, "all streams are EOS");
6356     goto eos;
6357   }
6358
6359   /* check for segment end */
6360   if (G_UNLIKELY (qtdemux->segment.stop != -1
6361           && qtdemux->segment.rate >= 0
6362           && qtdemux->segment.stop <= min_time && target_stream->on_keyframe)) {
6363     GST_DEBUG_OBJECT (qtdemux, "we reached the end of our segment.");
6364     target_stream->time_position = GST_CLOCK_TIME_NONE;
6365     goto eos_stream;
6366   }
6367
6368   /* gap events for subtitle streams */
6369   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
6370     stream = QTDEMUX_NTH_STREAM (qtdemux, i);
6371     if (stream->pad) {
6372       GstClockTime gap_threshold;
6373
6374       /* Only send gap events on non-subtitle streams if lagging way behind. */
6375       if (stream->subtype == FOURCC_subp
6376           || stream->subtype == FOURCC_text || stream->subtype == FOURCC_sbtl ||
6377           stream->subtype == FOURCC_wvtt)
6378         gap_threshold = 1 * GST_SECOND;
6379       else
6380         gap_threshold = 3 * GST_SECOND;
6381
6382       /* send gap events until the stream catches up */
6383       /* gaps can only be sent after segment is activated (segment.stop is no longer -1) */
6384       while (GST_CLOCK_TIME_IS_VALID (stream->segment.stop) &&
6385           GST_CLOCK_TIME_IS_VALID (stream->segment.position) &&
6386           stream->segment.position + gap_threshold < min_time) {
6387         GstEvent *gap =
6388             gst_event_new_gap (stream->segment.position, gap_threshold);
6389         gst_pad_push_event (stream->pad, gap);
6390         stream->segment.position += gap_threshold;
6391       }
6392     }
6393   }
6394
6395   stream = target_stream;
6396   /* fetch info for the current sample of this stream */
6397   if (G_UNLIKELY (!gst_qtdemux_prepare_current_sample (qtdemux, stream, &empty,
6398               &offset, &sample_size, &dts, &pts, &duration, &keyframe)))
6399     goto eos_stream;
6400
6401   gst_qtdemux_stream_check_and_change_stsd_index (qtdemux, stream);
6402   if (stream->new_caps) {
6403     gst_qtdemux_configure_stream (qtdemux, stream);
6404     qtdemux_do_allocation (stream, qtdemux);
6405   }
6406
6407   /* If we're doing a keyframe-only trickmode, only push keyframes on video streams */
6408   if (G_UNLIKELY (qtdemux->segment.
6409           flags & GST_SEGMENT_FLAG_TRICKMODE_KEY_UNITS)) {
6410     if (stream->subtype == FOURCC_vide) {
6411       if (!keyframe) {
6412         GST_LOG_OBJECT (qtdemux, "Skipping non-keyframe on track-id %u",
6413             stream->track_id);
6414         goto next;
6415       } else if (qtdemux->trickmode_interval > 0) {
6416         GstClockTimeDiff interval;
6417
6418         if (qtdemux->segment.rate > 0)
6419           interval = stream->time_position - stream->last_keyframe_dts;
6420         else
6421           interval = stream->last_keyframe_dts - stream->time_position;
6422
6423         if (GST_CLOCK_TIME_IS_VALID (stream->last_keyframe_dts)
6424             && interval < qtdemux->trickmode_interval) {
6425           GST_LOG_OBJECT (qtdemux,
6426               "Skipping keyframe within interval on track-id %u",
6427               stream->track_id);
6428           goto next;
6429         } else {
6430           stream->last_keyframe_dts = stream->time_position;
6431         }
6432       }
6433     }
6434   }
6435
6436   GST_DEBUG_OBJECT (qtdemux,
6437       "pushing from track-id %u, empty %d offset %" G_GUINT64_FORMAT
6438       ", size %d, dts=%" GST_TIME_FORMAT ", pts=%" GST_TIME_FORMAT
6439       ", duration %" GST_TIME_FORMAT, stream->track_id, empty, offset,
6440       sample_size, GST_TIME_ARGS (dts), GST_TIME_ARGS (pts),
6441       GST_TIME_ARGS (duration));
6442
6443   if (G_UNLIKELY (empty)) {
6444     /* empty segment, push a gap if there's a second or more
6445      * difference and move to the next one */
6446     if ((pts + duration - stream->segment.position) >= GST_SECOND)
6447       gst_pad_push_event (stream->pad, gst_event_new_gap (pts, duration));
6448     stream->segment.position = pts + duration;
6449     goto next;
6450   }
6451
6452   /* hmm, empty sample, skip and move to next sample */
6453   if (G_UNLIKELY (sample_size <= 0))
6454     goto next;
6455
6456   /* last pushed sample was out of boundary, goto next sample */
6457   if (G_UNLIKELY (GST_PAD_LAST_FLOW_RETURN (stream->pad) == GST_FLOW_EOS))
6458     goto next;
6459
6460   if (stream->max_buffer_size != 0 && sample_size > stream->max_buffer_size) {
6461     GST_DEBUG_OBJECT (qtdemux,
6462         "size %d larger than stream max_buffer_size %d, trimming",
6463         sample_size, stream->max_buffer_size);
6464     size =
6465         MIN (sample_size - stream->offset_in_sample, stream->max_buffer_size);
6466   } else if (stream->min_buffer_size != 0 && stream->offset_in_sample == 0
6467       && sample_size < stream->min_buffer_size) {
6468     guint start_sample_index = stream->sample_index;
6469     guint accumulated_size = sample_size;
6470     guint64 expected_next_offset = offset + sample_size;
6471
6472     GST_DEBUG_OBJECT (qtdemux,
6473         "size %d smaller than stream min_buffer_size %d, combining with the next",
6474         sample_size, stream->min_buffer_size);
6475
6476     while (stream->sample_index < stream->to_sample
6477         && stream->sample_index + 1 < stream->n_samples) {
6478       const QtDemuxSample *next_sample;
6479
6480       /* Increment temporarily */
6481       stream->sample_index++;
6482
6483       /* Failed to parse sample so let's go back to the previous one that was
6484        * still successful */
6485       if (!qtdemux_parse_samples (qtdemux, stream, stream->sample_index)) {
6486         stream->sample_index--;
6487         break;
6488       }
6489
6490       next_sample = &stream->samples[stream->sample_index];
6491
6492       /* Not contiguous with the previous sample so let's go back to the
6493        * previous one that was still successful */
6494       if (next_sample->offset != expected_next_offset) {
6495         stream->sample_index--;
6496         break;
6497       }
6498
6499       accumulated_size += next_sample->size;
6500       expected_next_offset += next_sample->size;
6501       if (accumulated_size >= stream->min_buffer_size)
6502         break;
6503     }
6504
6505     num_samples = stream->sample_index + 1 - start_sample_index;
6506     stream->sample_index = start_sample_index;
6507     GST_DEBUG_OBJECT (qtdemux, "Pulling %u samples of size %u at once",
6508         num_samples, accumulated_size);
6509     size = accumulated_size;
6510   } else {
6511     size = sample_size;
6512   }
6513
6514   if (qtdemux->cenc_aux_info_offset > 0) {
6515     GstMapInfo map;
6516     GstByteReader br;
6517     GstBuffer *aux_info = NULL;
6518
6519     /* pull the data stored before the sample */
6520     ret =
6521         gst_qtdemux_pull_atom (qtdemux, qtdemux->offset,
6522         offset + stream->offset_in_sample - qtdemux->offset, &aux_info);
6523     if (G_UNLIKELY (ret != GST_FLOW_OK))
6524       goto beach;
6525     gst_buffer_map (aux_info, &map, GST_MAP_READ);
6526     GST_DEBUG_OBJECT (qtdemux, "parsing cenc auxiliary info");
6527     gst_byte_reader_init (&br, map.data + 8, map.size);
6528     if (!qtdemux_parse_cenc_aux_info (qtdemux, stream, &br,
6529             qtdemux->cenc_aux_info_sizes, qtdemux->cenc_aux_sample_count)) {
6530       GST_ERROR_OBJECT (qtdemux, "failed to parse cenc auxiliary info");
6531       gst_buffer_unmap (aux_info, &map);
6532       gst_buffer_unref (aux_info);
6533       ret = GST_FLOW_ERROR;
6534       goto beach;
6535     }
6536     gst_buffer_unmap (aux_info, &map);
6537     gst_buffer_unref (aux_info);
6538   }
6539
6540   GST_LOG_OBJECT (qtdemux, "reading %d bytes @ %" G_GUINT64_FORMAT, size,
6541       offset);
6542
6543   if (stream->use_allocator) {
6544     /* if we have a per-stream allocator, use it */
6545     buf = gst_buffer_new_allocate (stream->allocator, size, &stream->params);
6546   }
6547
6548   ret = gst_qtdemux_pull_atom (qtdemux, offset + stream->offset_in_sample,
6549       size, &buf);
6550   if (G_UNLIKELY (ret != GST_FLOW_OK))
6551     goto beach;
6552
6553   /* Update for both splitting and combining of samples */
6554   if (size != sample_size) {
6555     pts += gst_util_uint64_scale_int (GST_SECOND,
6556         stream->offset_in_sample / CUR_STREAM (stream)->bytes_per_frame,
6557         stream->timescale);
6558     dts +=
6559         gst_util_uint64_scale_int (GST_SECOND,
6560         stream->offset_in_sample / CUR_STREAM (stream)->bytes_per_frame,
6561         stream->timescale);
6562     duration =
6563         gst_util_uint64_scale_int (GST_SECOND,
6564         size / CUR_STREAM (stream)->bytes_per_frame, stream->timescale);
6565   }
6566
6567   ret = gst_qtdemux_decorate_and_push_buffer (qtdemux, stream, buf,
6568       dts, pts, duration, keyframe, min_time, offset);
6569
6570   if (size < sample_size) {
6571     QtDemuxSample *sample = &stream->samples[stream->sample_index];
6572     QtDemuxSegment *segment = &stream->segments[stream->segment_index];
6573
6574     GstClockTime time_position = QTSTREAMTIME_TO_GSTTIME (stream,
6575         sample->timestamp +
6576         stream->offset_in_sample / CUR_STREAM (stream)->bytes_per_frame);
6577     if (time_position >= segment->media_start) {
6578       /* inside the segment, update time_position, looks very familiar to
6579        * GStreamer segments, doesn't it? */
6580       stream->time_position = (time_position - segment->media_start) +
6581           segment->time;
6582     } else {
6583       /* not yet in segment, time does not yet increment. This means
6584        * that we are still prerolling keyframes to the decoder so it can
6585        * decode the first sample of the segment. */
6586       stream->time_position = segment->time;
6587     }
6588   } else if (size > sample_size) {
6589     /* Increase to the last sample we already pulled so that advancing
6590      * below brings us to the next sample we need to pull */
6591     stream->sample_index += num_samples - 1;
6592   }
6593
6594   /* combine flows */
6595   GST_OBJECT_LOCK (qtdemux);
6596   ret = gst_qtdemux_combine_flows (qtdemux, stream, ret);
6597   GST_OBJECT_UNLOCK (qtdemux);
6598   /* ignore unlinked, we will not push on the pad anymore and we will EOS when
6599    * we have no more data for the pad to push */
6600   if (ret == GST_FLOW_EOS)
6601     ret = GST_FLOW_OK;
6602
6603   stream->offset_in_sample += size;
6604   if (stream->offset_in_sample >= sample_size) {
6605     gst_qtdemux_advance_sample (qtdemux, stream);
6606   }
6607   goto beach;
6608
6609 next:
6610   gst_qtdemux_advance_sample (qtdemux, stream);
6611
6612 beach:
6613   return ret;
6614
6615   /* special cases */
6616 eos:
6617   {
6618     GST_DEBUG_OBJECT (qtdemux, "No samples left for any streams - EOS");
6619     ret = GST_FLOW_EOS;
6620     goto beach;
6621   }
6622 eos_stream:
6623   {
6624     GST_DEBUG_OBJECT (qtdemux, "No samples left for stream");
6625     /* EOS will be raised if all are EOS */
6626     ret = GST_FLOW_OK;
6627     goto beach;
6628   }
6629 }
6630
6631 static void
6632 gst_qtdemux_loop (GstPad * pad)
6633 {
6634   GstQTDemux *qtdemux;
6635   guint64 cur_offset;
6636   GstFlowReturn ret;
6637
6638   qtdemux = GST_QTDEMUX (gst_pad_get_parent (pad));
6639
6640   cur_offset = qtdemux->offset;
6641   GST_LOG_OBJECT (qtdemux, "loop at position %" G_GUINT64_FORMAT ", state %s",
6642       cur_offset, qt_demux_state_string (qtdemux->state));
6643
6644   switch (qtdemux->state) {
6645     case QTDEMUX_STATE_INITIAL:
6646     case QTDEMUX_STATE_HEADER:
6647       ret = gst_qtdemux_loop_state_header (qtdemux);
6648       break;
6649     case QTDEMUX_STATE_MOVIE:
6650       ret = gst_qtdemux_loop_state_movie (qtdemux);
6651       if (qtdemux->segment.rate < 0 && ret == GST_FLOW_EOS) {
6652         ret = gst_qtdemux_seek_to_previous_keyframe (qtdemux);
6653       }
6654       break;
6655     default:
6656       /* ouch */
6657       goto invalid_state;
6658   }
6659
6660   /* if something went wrong, pause */
6661   if (ret != GST_FLOW_OK)
6662     goto pause;
6663
6664 done:
6665   gst_object_unref (qtdemux);
6666   return;
6667
6668   /* ERRORS */
6669 invalid_state:
6670   {
6671     GST_ELEMENT_ERROR (qtdemux, STREAM, FAILED,
6672         (NULL), ("streaming stopped, invalid state"));
6673     gst_pad_pause_task (pad);
6674     gst_qtdemux_push_event (qtdemux, gst_event_new_eos ());
6675     goto done;
6676   }
6677 pause:
6678   {
6679     const gchar *reason = gst_flow_get_name (ret);
6680
6681     GST_LOG_OBJECT (qtdemux, "pausing task, reason %s", reason);
6682
6683     gst_pad_pause_task (pad);
6684
6685     /* fatal errors need special actions */
6686     /* check EOS */
6687     if (ret == GST_FLOW_EOS) {
6688       if (QTDEMUX_N_STREAMS (qtdemux) == 0) {
6689         /* we have no streams, post an error */
6690         gst_qtdemux_post_no_playable_stream_error (qtdemux);
6691       }
6692       if (qtdemux->segment.flags & GST_SEEK_FLAG_SEGMENT) {
6693         gint64 stop;
6694
6695         if ((stop = qtdemux->segment.stop) == -1)
6696           stop = qtdemux->segment.duration;
6697
6698         if (qtdemux->segment.rate >= 0) {
6699           GstMessage *message;
6700           GstEvent *event;
6701
6702           GST_LOG_OBJECT (qtdemux, "Sending segment done, at end of segment");
6703           message = gst_message_new_segment_done (GST_OBJECT_CAST (qtdemux),
6704               GST_FORMAT_TIME, stop);
6705           event = gst_event_new_segment_done (GST_FORMAT_TIME, stop);
6706           if (qtdemux->segment_seqnum != GST_SEQNUM_INVALID) {
6707             gst_message_set_seqnum (message, qtdemux->segment_seqnum);
6708             gst_event_set_seqnum (event, qtdemux->segment_seqnum);
6709           }
6710           gst_element_post_message (GST_ELEMENT_CAST (qtdemux), message);
6711           gst_qtdemux_push_event (qtdemux, event);
6712         } else {
6713           GstMessage *message;
6714           GstEvent *event;
6715
6716           /*  For Reverse Playback */
6717           GST_LOG_OBJECT (qtdemux, "Sending segment done, at start of segment");
6718           message = gst_message_new_segment_done (GST_OBJECT_CAST (qtdemux),
6719               GST_FORMAT_TIME, qtdemux->segment.start);
6720           event = gst_event_new_segment_done (GST_FORMAT_TIME,
6721               qtdemux->segment.start);
6722           if (qtdemux->segment_seqnum != GST_SEQNUM_INVALID) {
6723             gst_message_set_seqnum (message, qtdemux->segment_seqnum);
6724             gst_event_set_seqnum (event, qtdemux->segment_seqnum);
6725           }
6726           gst_element_post_message (GST_ELEMENT_CAST (qtdemux), message);
6727           gst_qtdemux_push_event (qtdemux, event);
6728         }
6729       } else {
6730         GstEvent *event;
6731
6732         GST_LOG_OBJECT (qtdemux, "Sending EOS at end of segment");
6733         event = gst_event_new_eos ();
6734         if (qtdemux->segment_seqnum != GST_SEQNUM_INVALID)
6735           gst_event_set_seqnum (event, qtdemux->segment_seqnum);
6736         gst_qtdemux_push_event (qtdemux, event);
6737       }
6738     } else if (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_EOS) {
6739       GST_ELEMENT_FLOW_ERROR (qtdemux, ret);
6740       gst_qtdemux_push_event (qtdemux, gst_event_new_eos ());
6741     }
6742     goto done;
6743   }
6744 }
6745
6746 /*
6747  * has_next_entry
6748  *
6749  * Returns if there are samples to be played.
6750  */
6751 static gboolean
6752 has_next_entry (GstQTDemux * demux)
6753 {
6754   QtDemuxStream *stream;
6755   gint i;
6756
6757   GST_DEBUG_OBJECT (demux, "Checking if there are samples not played yet");
6758
6759   for (i = 0; i < QTDEMUX_N_STREAMS (demux); i++) {
6760     stream = QTDEMUX_NTH_STREAM (demux, i);
6761
6762     if (stream->sample_index == -1) {
6763       stream->sample_index = 0;
6764       stream->offset_in_sample = 0;
6765     }
6766
6767     if (stream->sample_index >= stream->n_samples) {
6768       GST_LOG_OBJECT (demux, "track-id %u samples exhausted", stream->track_id);
6769       continue;
6770     }
6771     GST_DEBUG_OBJECT (demux, "Found a sample");
6772     return TRUE;
6773   }
6774
6775   GST_DEBUG_OBJECT (demux, "There wasn't any next sample");
6776   return FALSE;
6777 }
6778
6779 /*
6780  * next_entry_size
6781  *
6782  * Returns the size of the first entry at the current offset.
6783  * If -1, there are none (which means EOS or empty file).
6784  */
6785 static guint64
6786 next_entry_size (GstQTDemux * demux)
6787 {
6788   QtDemuxStream *stream, *target_stream = NULL;
6789   guint64 smalloffs = (guint64) - 1;
6790   QtDemuxSample *sample;
6791   gint i;
6792
6793   GST_LOG_OBJECT (demux, "Finding entry at offset %" G_GUINT64_FORMAT,
6794       demux->offset);
6795
6796   for (i = 0; i < QTDEMUX_N_STREAMS (demux); i++) {
6797     stream = QTDEMUX_NTH_STREAM (demux, i);
6798
6799     if (stream->sample_index == -1) {
6800       stream->sample_index = 0;
6801       stream->offset_in_sample = 0;
6802     }
6803
6804     if (stream->sample_index >= stream->n_samples) {
6805       GST_LOG_OBJECT (demux, "track-id %u samples exhausted", stream->track_id);
6806       continue;
6807     }
6808
6809     if (!qtdemux_parse_samples (demux, stream, stream->sample_index)) {
6810       GST_LOG_OBJECT (demux, "Parsing of index %u from stbl atom failed!",
6811           stream->sample_index);
6812       return -1;
6813     }
6814
6815     sample = &stream->samples[stream->sample_index];
6816
6817     GST_LOG_OBJECT (demux,
6818         "Checking track-id %u (sample_index:%d / offset:%" G_GUINT64_FORMAT
6819         " / size:%" G_GUINT32_FORMAT ")", stream->track_id,
6820         stream->sample_index, sample->offset, sample->size);
6821
6822     if (((smalloffs == -1)
6823             || (sample->offset < smalloffs)) && (sample->size)) {
6824       smalloffs = sample->offset;
6825       target_stream = stream;
6826     }
6827   }
6828
6829   if (!target_stream)
6830     return -1;
6831
6832   GST_LOG_OBJECT (demux,
6833       "track-id %u offset %" G_GUINT64_FORMAT " demux->offset :%"
6834       G_GUINT64_FORMAT, target_stream->track_id, smalloffs, demux->offset);
6835
6836   stream = target_stream;
6837   sample = &stream->samples[stream->sample_index];
6838
6839   if (sample->offset >= demux->offset) {
6840     demux->todrop = sample->offset - demux->offset;
6841     return sample->size + demux->todrop;
6842   }
6843
6844   GST_DEBUG_OBJECT (demux,
6845       "There wasn't any entry at offset %" G_GUINT64_FORMAT, demux->offset);
6846   return -1;
6847 }
6848
6849 static void
6850 gst_qtdemux_post_progress (GstQTDemux * demux, gint num, gint denom)
6851 {
6852   gint perc = (gint) ((gdouble) num * 100.0 / (gdouble) denom);
6853
6854   gst_element_post_message (GST_ELEMENT_CAST (demux),
6855       gst_message_new_element (GST_OBJECT_CAST (demux),
6856           gst_structure_new ("progress", "percent", G_TYPE_INT, perc, NULL)));
6857 }
6858
6859 static gboolean
6860 qtdemux_seek_offset (GstQTDemux * demux, guint64 offset)
6861 {
6862   GstEvent *event;
6863   gboolean res = 0;
6864
6865   GST_DEBUG_OBJECT (demux, "Seeking to %" G_GUINT64_FORMAT, offset);
6866
6867   event =
6868       gst_event_new_seek (1.0, GST_FORMAT_BYTES,
6869       GST_SEEK_FLAG_FLUSH | GST_SEEK_FLAG_ACCURATE, GST_SEEK_TYPE_SET, offset,
6870       GST_SEEK_TYPE_NONE, -1);
6871
6872   /* store seqnum to drop flush events, they don't need to reach downstream */
6873   demux->offset_seek_seqnum = gst_event_get_seqnum (event);
6874   res = gst_pad_push_event (demux->sinkpad, event);
6875   demux->offset_seek_seqnum = GST_SEQNUM_INVALID;
6876
6877   return res;
6878 }
6879
6880 /* check for seekable upstream, above and beyond a mere query */
6881 static void
6882 gst_qtdemux_check_seekability (GstQTDemux * demux)
6883 {
6884   GstQuery *query;
6885   gboolean seekable = FALSE;
6886   gint64 start = -1, stop = -1;
6887
6888   if (demux->upstream_size)
6889     return;
6890
6891   if (demux->upstream_format_is_time)
6892     return;
6893
6894   query = gst_query_new_seeking (GST_FORMAT_BYTES);
6895   if (!gst_pad_peer_query (demux->sinkpad, query)) {
6896     GST_DEBUG_OBJECT (demux, "seeking query failed");
6897     goto done;
6898   }
6899
6900   gst_query_parse_seeking (query, NULL, &seekable, &start, &stop);
6901
6902   /* try harder to query upstream size if we didn't get it the first time */
6903   if (seekable && stop == -1) {
6904     GST_DEBUG_OBJECT (demux, "doing duration query to fix up unset stop");
6905     gst_pad_peer_query_duration (demux->sinkpad, GST_FORMAT_BYTES, &stop);
6906   }
6907
6908   /* if upstream doesn't know the size, it's likely that it's not seekable in
6909    * practice even if it technically may be seekable */
6910   if (seekable && (start != 0 || stop <= start)) {
6911     GST_DEBUG_OBJECT (demux, "seekable but unknown start/stop -> disable");
6912     seekable = FALSE;
6913   }
6914
6915 done:
6916   gst_query_unref (query);
6917
6918   GST_DEBUG_OBJECT (demux, "seekable: %d (%" G_GUINT64_FORMAT " - %"
6919       G_GUINT64_FORMAT ")", seekable, start, stop);
6920   demux->upstream_seekable = seekable;
6921   demux->upstream_size = seekable ? stop : -1;
6922 }
6923
6924 static void
6925 gst_qtdemux_drop_data (GstQTDemux * demux, gint bytes)
6926 {
6927   g_return_if_fail (bytes <= demux->todrop);
6928
6929   GST_LOG_OBJECT (demux, "Dropping %d bytes", bytes);
6930   gst_adapter_flush (demux->adapter, bytes);
6931   demux->neededbytes -= bytes;
6932   demux->offset += bytes;
6933   demux->todrop -= bytes;
6934 }
6935
6936 /* PUSH-MODE only: Send a segment, if not done already. */
6937 static void
6938 gst_qtdemux_check_send_pending_segment (GstQTDemux * demux)
6939 {
6940   if (G_UNLIKELY (demux->need_segment)) {
6941     gint i;
6942
6943     if (!demux->upstream_format_is_time) {
6944       gst_qtdemux_map_and_push_segments (demux, &demux->segment);
6945     } else {
6946       GstEvent *segment_event;
6947       segment_event = gst_event_new_segment (&demux->segment);
6948       if (demux->segment_seqnum != GST_SEQNUM_INVALID)
6949         gst_event_set_seqnum (segment_event, demux->segment_seqnum);
6950       gst_qtdemux_push_event (demux, segment_event);
6951     }
6952
6953     demux->need_segment = FALSE;
6954
6955     /* clear to send tags on all streams */
6956     for (i = 0; i < QTDEMUX_N_STREAMS (demux); i++) {
6957       QtDemuxStream *stream = QTDEMUX_NTH_STREAM (demux, i);
6958       gst_qtdemux_push_tags (demux, stream);
6959       if (CUR_STREAM (stream)->sparse) {
6960         GST_INFO_OBJECT (demux, "Sending gap event on stream %d", i);
6961         gst_pad_push_event (stream->pad,
6962             gst_event_new_gap (stream->segment.position, GST_CLOCK_TIME_NONE));
6963       }
6964     }
6965   }
6966 }
6967
6968 /* Used for push mode only. */
6969 static void
6970 gst_qtdemux_send_gap_for_segment (GstQTDemux * demux,
6971     QtDemuxStream * stream, gint segment_index, GstClockTime pos)
6972 {
6973   GstClockTime ts, dur;
6974
6975   ts = pos;
6976   dur =
6977       stream->segments[segment_index].duration - (pos -
6978       stream->segments[segment_index].time);
6979   stream->time_position += dur;
6980
6981   /* Only gaps with a duration of at least one second are propagated.
6982    * Same workaround as in pull mode.
6983    * (See 2e45926a96ec5298c6ef29bf912e5e6a06dc3e0e) */
6984   if (dur >= GST_SECOND) {
6985     GstEvent *gap;
6986     gap = gst_event_new_gap (ts, dur);
6987
6988     GST_DEBUG_OBJECT (stream->pad, "Pushing gap for empty "
6989         "segment: %" GST_PTR_FORMAT, gap);
6990     gst_pad_push_event (stream->pad, gap);
6991   }
6992 }
6993
6994 static GstFlowReturn
6995 gst_qtdemux_chain (GstPad * sinkpad, GstObject * parent, GstBuffer * inbuf)
6996 {
6997   GstQTDemux *demux;
6998
6999   demux = GST_QTDEMUX (parent);
7000
7001   GST_DEBUG_OBJECT (demux,
7002       "Received buffer pts:%" GST_TIME_FORMAT " dts:%" GST_TIME_FORMAT
7003       " offset:%" G_GUINT64_FORMAT " size:%" G_GSIZE_FORMAT " demux offset:%"
7004       G_GUINT64_FORMAT, GST_TIME_ARGS (GST_BUFFER_PTS (inbuf)),
7005       GST_TIME_ARGS (GST_BUFFER_DTS (inbuf)), GST_BUFFER_OFFSET (inbuf),
7006       gst_buffer_get_size (inbuf), demux->offset);
7007
7008   if (GST_BUFFER_FLAG_IS_SET (inbuf, GST_BUFFER_FLAG_DISCONT)) {
7009     gboolean is_gap_input = FALSE;
7010     gint i;
7011
7012     GST_DEBUG_OBJECT (demux, "Got DISCONT, marking all streams as DISCONT");
7013
7014     for (i = 0; i < QTDEMUX_N_STREAMS (demux); i++) {
7015       QTDEMUX_NTH_STREAM (demux, i)->discont = TRUE;
7016     }
7017
7018     /* Check if we can land back on our feet in the case where upstream is
7019      * handling the seeking/pushing of samples with gaps in between (like
7020      * in the case of trick-mode DASH for example) */
7021     if (demux->upstream_format_is_time
7022         && GST_BUFFER_OFFSET (inbuf) != GST_BUFFER_OFFSET_NONE) {
7023       for (i = 0; i < QTDEMUX_N_STREAMS (demux); i++) {
7024         guint32 res;
7025         QtDemuxStream *stream = QTDEMUX_NTH_STREAM (demux, i);
7026         GST_LOG_OBJECT (demux,
7027             "track-id #%u , checking if offset %" G_GUINT64_FORMAT
7028             " is a sample start", stream->track_id, GST_BUFFER_OFFSET (inbuf));
7029         res =
7030             gst_qtdemux_find_index_for_given_media_offset_linear (demux,
7031             stream, GST_BUFFER_OFFSET (inbuf));
7032         if (res != -1) {
7033           QtDemuxSample *sample = &stream->samples[res];
7034           GST_LOG_OBJECT (demux,
7035               "Checking if sample %d from track-id %u is valid (offset:%"
7036               G_GUINT64_FORMAT " size:%" G_GUINT32_FORMAT ")", res,
7037               stream->track_id, sample->offset, sample->size);
7038           if (sample->offset == GST_BUFFER_OFFSET (inbuf)) {
7039             GST_LOG_OBJECT (demux,
7040                 "new buffer corresponds to a valid sample : %" G_GUINT32_FORMAT,
7041                 res);
7042             is_gap_input = TRUE;
7043             /* We can go back to standard playback mode */
7044             demux->state = QTDEMUX_STATE_MOVIE;
7045             /* Remember which sample this stream is at */
7046             stream->sample_index = res;
7047             /* Finally update all push-based values to the expected values */
7048             demux->neededbytes = stream->samples[res].size;
7049             demux->offset = GST_BUFFER_OFFSET (inbuf);
7050             demux->mdatleft =
7051                 demux->mdatsize - demux->offset + demux->mdatoffset;
7052             demux->todrop = 0;
7053           }
7054         }
7055       }
7056       if (!is_gap_input) {
7057         GST_DEBUG_OBJECT (demux, "Resetting, actual DISCONT");
7058         /* Reset state if it's a real discont */
7059         demux->neededbytes = 16;
7060         demux->state = QTDEMUX_STATE_INITIAL;
7061         demux->offset = GST_BUFFER_OFFSET (inbuf);
7062         gst_adapter_clear (demux->adapter);
7063       }
7064     }
7065     /* Reverse fragmented playback, need to flush all we have before
7066      * consuming a new fragment.
7067      * The samples array have the timestamps calculated by accumulating the
7068      * durations but this won't work for reverse playback of fragments as
7069      * the timestamps of a subsequent fragment should be smaller than the
7070      * previously received one. */
7071     if (!is_gap_input && demux->fragmented && demux->segment.rate < 0) {
7072       gst_qtdemux_process_adapter (demux, TRUE);
7073       g_ptr_array_foreach (demux->active_streams,
7074           (GFunc) gst_qtdemux_stream_flush_samples_data, NULL);
7075     }
7076   }
7077
7078   gst_adapter_push (demux->adapter, inbuf);
7079
7080   GST_DEBUG_OBJECT (demux,
7081       "pushing in inbuf %p, neededbytes:%u, available:%" G_GSIZE_FORMAT, inbuf,
7082       demux->neededbytes, gst_adapter_available (demux->adapter));
7083
7084   return gst_qtdemux_process_adapter (demux, FALSE);
7085 }
7086
7087 static GstFlowReturn
7088 gst_qtdemux_process_adapter (GstQTDemux * demux, gboolean force)
7089 {
7090   GstFlowReturn ret = GST_FLOW_OK;
7091
7092   /* we never really mean to buffer that much */
7093   if (demux->neededbytes == -1) {
7094     goto eos;
7095   }
7096
7097   while (((gst_adapter_available (demux->adapter)) >= demux->neededbytes) &&
7098       (ret == GST_FLOW_OK || (ret == GST_FLOW_NOT_LINKED && force))) {
7099
7100 #ifndef GST_DISABLE_GST_DEBUG
7101     {
7102       guint64 discont_offset, distance_from_discont;
7103
7104       discont_offset = gst_adapter_offset_at_discont (demux->adapter);
7105       distance_from_discont =
7106           gst_adapter_distance_from_discont (demux->adapter);
7107
7108       GST_DEBUG_OBJECT (demux,
7109           "state:%s , demux->neededbytes:%d, demux->offset:%" G_GUINT64_FORMAT
7110           " adapter offset :%" G_GUINT64_FORMAT " (+ %" G_GUINT64_FORMAT
7111           " bytes)", qt_demux_state_string (demux->state), demux->neededbytes,
7112           demux->offset, discont_offset, distance_from_discont);
7113     }
7114 #endif
7115
7116     switch (demux->state) {
7117       case QTDEMUX_STATE_INITIAL:{
7118         const guint8 *data;
7119         guint32 fourcc;
7120         guint64 size;
7121
7122         gst_qtdemux_check_seekability (demux);
7123
7124         data = gst_adapter_map (demux->adapter, demux->neededbytes);
7125
7126         /* get fourcc/length, set neededbytes */
7127         extract_initial_length_and_fourcc ((guint8 *) data, demux->neededbytes,
7128             &size, &fourcc);
7129         gst_adapter_unmap (demux->adapter);
7130         data = NULL;
7131         GST_DEBUG_OBJECT (demux, "Peeking found [%" GST_FOURCC_FORMAT "] "
7132             "size: %" G_GUINT64_FORMAT, GST_FOURCC_ARGS (fourcc), size);
7133         if (size == 0) {
7134           GST_ELEMENT_ERROR (demux, STREAM, DEMUX,
7135               (_("This file is invalid and cannot be played.")),
7136               ("initial atom '%" GST_FOURCC_FORMAT "' has empty length",
7137                   GST_FOURCC_ARGS (fourcc)));
7138           ret = GST_FLOW_ERROR;
7139           break;
7140         }
7141         if (fourcc == FOURCC_mdat) {
7142           gint next_entry = next_entry_size (demux);
7143           if (QTDEMUX_N_STREAMS (demux) > 0 && (next_entry != -1
7144                   || !demux->fragmented)) {
7145             /* we have the headers, start playback */
7146             demux->state = QTDEMUX_STATE_MOVIE;
7147             demux->neededbytes = next_entry;
7148             demux->mdatleft = size;
7149             demux->mdatsize = demux->mdatleft;
7150           } else {
7151             /* no headers yet, try to get them */
7152             guint bs;
7153             gboolean res;
7154             guint64 old, target;
7155
7156           buffer_data:
7157             old = demux->offset;
7158             target = old + size;
7159
7160             /* try to jump over the atom with a seek */
7161             /* only bother if it seems worth doing so,
7162              * and avoids possible upstream/server problems */
7163             if (demux->upstream_seekable &&
7164                 demux->upstream_size > 4 * (1 << 20)) {
7165               res = qtdemux_seek_offset (demux, target);
7166             } else {
7167               GST_DEBUG_OBJECT (demux, "skipping seek");
7168               res = FALSE;
7169             }
7170
7171             if (res) {
7172               GST_DEBUG_OBJECT (demux, "seek success");
7173               /* remember the offset fo the first mdat so we can seek back to it
7174                * after we have the headers */
7175               if (fourcc == FOURCC_mdat && demux->first_mdat == -1) {
7176                 demux->first_mdat = old;
7177                 GST_DEBUG_OBJECT (demux, "first mdat at %" G_GUINT64_FORMAT,
7178                     demux->first_mdat);
7179               }
7180               /* seek worked, continue reading */
7181               demux->offset = target;
7182               demux->neededbytes = 16;
7183               demux->state = QTDEMUX_STATE_INITIAL;
7184             } else {
7185               /* seek failed, need to buffer */
7186               demux->offset = old;
7187               GST_DEBUG_OBJECT (demux, "seek failed/skipped");
7188               /* there may be multiple mdat (or alike) buffers */
7189               /* sanity check */
7190               if (demux->mdatbuffer)
7191                 bs = gst_buffer_get_size (demux->mdatbuffer);
7192               else
7193                 bs = 0;
7194               if (size + bs > 10 * (1 << 20))
7195                 goto no_moov;
7196               demux->state = QTDEMUX_STATE_BUFFER_MDAT;
7197               demux->neededbytes = size;
7198               if (!demux->mdatbuffer)
7199                 demux->mdatoffset = demux->offset;
7200             }
7201           }
7202         } else if (G_UNLIKELY (size > QTDEMUX_MAX_ATOM_SIZE)) {
7203           GST_ELEMENT_ERROR (demux, STREAM, DEMUX,
7204               (_("This file is invalid and cannot be played.")),
7205               ("atom %" GST_FOURCC_FORMAT " has bogus size %" G_GUINT64_FORMAT,
7206                   GST_FOURCC_ARGS (fourcc), size));
7207           ret = GST_FLOW_ERROR;
7208           break;
7209         } else {
7210           /* this means we already started buffering and still no moov header,
7211            * let's continue buffering everything till we get moov */
7212           if (demux->mdatbuffer && !(fourcc == FOURCC_moov
7213                   || fourcc == FOURCC_moof))
7214             goto buffer_data;
7215           demux->neededbytes = size;
7216           demux->state = QTDEMUX_STATE_HEADER;
7217         }
7218         break;
7219       }
7220       case QTDEMUX_STATE_HEADER:{
7221         const guint8 *data;
7222         guint32 fourcc;
7223
7224         GST_DEBUG_OBJECT (demux, "In header");
7225
7226         data = gst_adapter_map (demux->adapter, demux->neededbytes);
7227
7228         /* parse the header */
7229         extract_initial_length_and_fourcc (data, demux->neededbytes, NULL,
7230             &fourcc);
7231         if (fourcc == FOURCC_moov) {
7232           /* in usual fragmented setup we could try to scan for more
7233            * and end up at the the moov (after mdat) again */
7234           if (demux->got_moov && QTDEMUX_N_STREAMS (demux) > 0 &&
7235               (!demux->fragmented
7236                   || demux->last_moov_offset == demux->offset)) {
7237             GST_DEBUG_OBJECT (demux,
7238                 "Skipping moov atom as we have (this) one already");
7239           } else {
7240             GST_DEBUG_OBJECT (demux, "Parsing [moov]");
7241
7242             if (demux->got_moov && demux->fragmented) {
7243               GST_DEBUG_OBJECT (demux,
7244                   "Got a second moov, clean up data from old one");
7245               if (demux->moov_node_compressed) {
7246                 g_node_destroy (demux->moov_node_compressed);
7247                 if (demux->moov_node)
7248                   g_free (demux->moov_node->data);
7249               }
7250               demux->moov_node_compressed = NULL;
7251               if (demux->moov_node)
7252                 g_node_destroy (demux->moov_node);
7253               demux->moov_node = NULL;
7254             }
7255
7256             demux->last_moov_offset = demux->offset;
7257
7258             /* Update streams with new moov */
7259             gst_qtdemux_stream_concat (demux,
7260                 demux->old_streams, demux->active_streams);
7261
7262             qtdemux_parse_moov (demux, data, demux->neededbytes);
7263             qtdemux_node_dump (demux, demux->moov_node);
7264             qtdemux_parse_tree (demux);
7265             qtdemux_prepare_streams (demux);
7266             QTDEMUX_EXPOSE_LOCK (demux);
7267             qtdemux_expose_streams (demux);
7268             QTDEMUX_EXPOSE_UNLOCK (demux);
7269
7270             demux->got_moov = TRUE;
7271
7272             gst_qtdemux_check_send_pending_segment (demux);
7273
7274             if (demux->moov_node_compressed) {
7275               g_node_destroy (demux->moov_node_compressed);
7276               g_free (demux->moov_node->data);
7277             }
7278             demux->moov_node_compressed = NULL;
7279             g_node_destroy (demux->moov_node);
7280             demux->moov_node = NULL;
7281             GST_DEBUG_OBJECT (demux, "Finished parsing the header");
7282           }
7283         } else if (fourcc == FOURCC_moof) {
7284           if ((demux->got_moov || demux->media_caps) && demux->fragmented) {
7285             guint64 dist = 0;
7286             GstClockTime prev_pts;
7287             guint64 prev_offset;
7288             guint64 adapter_discont_offset, adapter_discont_dist;
7289
7290             GST_DEBUG_OBJECT (demux, "Parsing [moof]");
7291
7292             /*
7293              * The timestamp of the moof buffer is relevant as some scenarios
7294              * won't have the initial timestamp in the atoms. Whenever a new
7295              * buffer has started, we get that buffer's PTS and use it as a base
7296              * timestamp for the trun entries.
7297              *
7298              * To keep track of the current buffer timestamp and starting point
7299              * we use gst_adapter_prev_pts that gives us the PTS and the distance
7300              * from the beginning of the buffer, with the distance and demux->offset
7301              * we know if it is still the same buffer or not.
7302              */
7303             prev_pts = gst_adapter_prev_pts (demux->adapter, &dist);
7304             prev_offset = demux->offset - dist;
7305             if (demux->fragment_start_offset == -1
7306                 || prev_offset > demux->fragment_start_offset) {
7307               demux->fragment_start_offset = prev_offset;
7308               demux->fragment_start = prev_pts;
7309               GST_DEBUG_OBJECT (demux,
7310                   "New fragment start found at: %" G_GUINT64_FORMAT " : %"
7311                   GST_TIME_FORMAT, demux->fragment_start_offset,
7312                   GST_TIME_ARGS (demux->fragment_start));
7313             }
7314
7315             /* We can't use prev_offset() here because this would require
7316              * upstream to set consistent and correct offsets on all buffers
7317              * since the discont. Nothing ever did that in the past and we
7318              * would break backwards compatibility here then.
7319              * Instead take the offset we had at the last discont and count
7320              * the bytes from there. This works with old code as there would
7321              * be no discont between moov and moof, and also works with
7322              * adaptivedemux which correctly sets offset and will set the
7323              * DISCONT flag accordingly when needed.
7324              *
7325              * We also only do this for upstream TIME segments as otherwise
7326              * there are potential backwards compatibility problems with
7327              * seeking in PUSH mode and upstream providing inconsistent
7328              * timestamps. */
7329             adapter_discont_offset =
7330                 gst_adapter_offset_at_discont (demux->adapter);
7331             adapter_discont_dist =
7332                 gst_adapter_distance_from_discont (demux->adapter);
7333
7334             GST_DEBUG_OBJECT (demux,
7335                 "demux offset %" G_GUINT64_FORMAT " adapter offset %"
7336                 G_GUINT64_FORMAT " (+ %" G_GUINT64_FORMAT " bytes)",
7337                 demux->offset, adapter_discont_offset, adapter_discont_dist);
7338
7339             if (demux->upstream_format_is_time) {
7340               demux->moof_offset = adapter_discont_offset;
7341               if (demux->moof_offset != GST_BUFFER_OFFSET_NONE)
7342                 demux->moof_offset += adapter_discont_dist;
7343               if (demux->moof_offset == GST_BUFFER_OFFSET_NONE)
7344                 demux->moof_offset = demux->offset;
7345             } else {
7346               demux->moof_offset = demux->offset;
7347             }
7348
7349             if (!qtdemux_parse_moof (demux, data, demux->neededbytes,
7350                     demux->moof_offset, NULL)) {
7351               gst_adapter_unmap (demux->adapter);
7352               ret = GST_FLOW_ERROR;
7353               goto done;
7354             }
7355
7356             /* in MSS we need to expose the pads after the first moof as we won't get a moov */
7357             if (demux->mss_mode && !demux->exposed) {
7358               QTDEMUX_EXPOSE_LOCK (demux);
7359               qtdemux_expose_streams (demux);
7360               QTDEMUX_EXPOSE_UNLOCK (demux);
7361             }
7362
7363             gst_qtdemux_check_send_pending_segment (demux);
7364           } else {
7365             GST_DEBUG_OBJECT (demux, "Discarding [moof]");
7366           }
7367         } else if (fourcc == FOURCC_ftyp) {
7368           GST_DEBUG_OBJECT (demux, "Parsing [ftyp]");
7369           qtdemux_parse_ftyp (demux, data, demux->neededbytes);
7370         } else if (fourcc == FOURCC_uuid) {
7371           GST_DEBUG_OBJECT (demux, "Parsing [uuid]");
7372           qtdemux_parse_uuid (demux, data, demux->neededbytes);
7373         } else if (fourcc == FOURCC_sidx) {
7374           GST_DEBUG_OBJECT (demux, "Parsing [sidx]");
7375           qtdemux_parse_sidx (demux, data, demux->neededbytes);
7376         } else {
7377           switch (fourcc) {
7378             case FOURCC_styp:
7379               /* [styp] is like a [ftyp], but in fragment header. We ignore it for now
7380                * FALLTHROUGH */
7381             case FOURCC_skip:
7382             case FOURCC_free:
7383               /* [free] and [skip] are padding atoms */
7384               GST_DEBUG_OBJECT (demux,
7385                   "Skipping fourcc while parsing header : %" GST_FOURCC_FORMAT,
7386                   GST_FOURCC_ARGS (fourcc));
7387               break;
7388             default:
7389               GST_WARNING_OBJECT (demux,
7390                   "Unknown fourcc while parsing header : %" GST_FOURCC_FORMAT,
7391                   GST_FOURCC_ARGS (fourcc));
7392               /* Let's jump that one and go back to initial state */
7393               break;
7394           }
7395         }
7396         gst_adapter_unmap (demux->adapter);
7397         data = NULL;
7398
7399         if (demux->mdatbuffer && QTDEMUX_N_STREAMS (demux)) {
7400           gsize remaining_data_size = 0;
7401
7402           /* the mdat was before the header */
7403           GST_DEBUG_OBJECT (demux, "We have n_streams:%d and mdatbuffer:%p",
7404               QTDEMUX_N_STREAMS (demux), demux->mdatbuffer);
7405           /* restore our adapter/offset view of things with upstream;
7406            * put preceding buffered data ahead of current moov data.
7407            * This should also handle evil mdat, moov, mdat cases and alike */
7408           gst_adapter_flush (demux->adapter, demux->neededbytes);
7409
7410           /* Store any remaining data after the mdat for later usage */
7411           remaining_data_size = gst_adapter_available (demux->adapter);
7412           if (remaining_data_size > 0) {
7413             g_assert (demux->restoredata_buffer == NULL);
7414             demux->restoredata_buffer =
7415                 gst_adapter_take_buffer (demux->adapter, remaining_data_size);
7416             demux->restoredata_offset = demux->offset + demux->neededbytes;
7417             GST_DEBUG_OBJECT (demux,
7418                 "Stored %" G_GSIZE_FORMAT " post mdat bytes at offset %"
7419                 G_GUINT64_FORMAT, remaining_data_size,
7420                 demux->restoredata_offset);
7421           }
7422
7423           gst_adapter_push (demux->adapter, demux->mdatbuffer);
7424           demux->mdatbuffer = NULL;
7425           demux->offset = demux->mdatoffset;
7426           demux->neededbytes = next_entry_size (demux);
7427           demux->state = QTDEMUX_STATE_MOVIE;
7428           demux->mdatleft = gst_adapter_available (demux->adapter);
7429           demux->mdatsize = demux->mdatleft;
7430         } else {
7431           GST_DEBUG_OBJECT (demux, "Carrying on normally");
7432           gst_adapter_flush (demux->adapter, demux->neededbytes);
7433
7434           /* only go back to the mdat if there are samples to play */
7435           if (demux->got_moov && demux->first_mdat != -1
7436               && has_next_entry (demux)) {
7437             gboolean res;
7438
7439             /* we need to seek back */
7440             res = qtdemux_seek_offset (demux, demux->first_mdat);
7441             if (res) {
7442               demux->offset = demux->first_mdat;
7443             } else {
7444               GST_DEBUG_OBJECT (demux, "Seek back failed");
7445             }
7446           } else {
7447             demux->offset += demux->neededbytes;
7448           }
7449           demux->neededbytes = 16;
7450           demux->state = QTDEMUX_STATE_INITIAL;
7451         }
7452
7453         break;
7454       }
7455       case QTDEMUX_STATE_BUFFER_MDAT:{
7456         GstBuffer *buf;
7457         guint8 fourcc[4];
7458
7459         GST_DEBUG_OBJECT (demux, "Got our buffer at offset %" G_GUINT64_FORMAT,
7460             demux->offset);
7461         buf = gst_adapter_take_buffer (demux->adapter, demux->neededbytes);
7462         gst_buffer_extract (buf, 0, fourcc, 4);
7463         GST_DEBUG_OBJECT (demux, "mdatbuffer starts with %" GST_FOURCC_FORMAT,
7464             GST_FOURCC_ARGS (QT_FOURCC (fourcc)));
7465         if (demux->mdatbuffer)
7466           demux->mdatbuffer = gst_buffer_append (demux->mdatbuffer, buf);
7467         else
7468           demux->mdatbuffer = buf;
7469         demux->offset += demux->neededbytes;
7470         demux->neededbytes = 16;
7471         demux->state = QTDEMUX_STATE_INITIAL;
7472         gst_qtdemux_post_progress (demux, 1, 1);
7473
7474         break;
7475       }
7476       case QTDEMUX_STATE_MOVIE:{
7477         QtDemuxStream *stream = NULL;
7478         QtDemuxSample *sample;
7479         GstClockTime dts, pts, duration;
7480         gboolean keyframe;
7481         gint i;
7482
7483         GST_DEBUG_OBJECT (demux,
7484             "BEGIN // in MOVIE for offset %" G_GUINT64_FORMAT, demux->offset);
7485
7486         if (demux->fragmented) {
7487           GST_DEBUG_OBJECT (demux, "mdat remaining %" G_GUINT64_FORMAT,
7488               demux->mdatleft);
7489           if (G_LIKELY (demux->todrop < demux->mdatleft)) {
7490             /* if needed data starts within this atom,
7491              * then it should not exceed this atom */
7492             if (G_UNLIKELY (demux->neededbytes > demux->mdatleft)) {
7493               GST_ELEMENT_ERROR (demux, STREAM, DEMUX,
7494                   (_("This file is invalid and cannot be played.")),
7495                   ("sample data crosses atom boundary"));
7496               ret = GST_FLOW_ERROR;
7497               break;
7498             }
7499             demux->mdatleft -= demux->neededbytes;
7500           } else {
7501             GST_DEBUG_OBJECT (demux, "data atom emptied; resuming atom scan");
7502             /* so we are dropping more than left in this atom */
7503             gst_qtdemux_drop_data (demux, demux->mdatleft);
7504             demux->mdatleft = 0;
7505
7506             /* need to resume atom parsing so we do not miss any other pieces */
7507             demux->state = QTDEMUX_STATE_INITIAL;
7508             demux->neededbytes = 16;
7509
7510             /* check if there was any stored post mdat data from previous buffers */
7511             if (demux->restoredata_buffer) {
7512               g_assert (gst_adapter_available (demux->adapter) == 0);
7513
7514               gst_adapter_push (demux->adapter, demux->restoredata_buffer);
7515               demux->restoredata_buffer = NULL;
7516               demux->offset = demux->restoredata_offset;
7517             }
7518
7519             break;
7520           }
7521         }
7522
7523         if (demux->todrop) {
7524           if (demux->cenc_aux_info_offset > 0) {
7525             GstByteReader br;
7526             const guint8 *data;
7527
7528             GST_DEBUG_OBJECT (demux, "parsing cenc auxiliary info");
7529             data = gst_adapter_map (demux->adapter, demux->todrop);
7530             gst_byte_reader_init (&br, data + 8, demux->todrop);
7531             if (!qtdemux_parse_cenc_aux_info (demux,
7532                     QTDEMUX_NTH_STREAM (demux, 0), &br,
7533                     demux->cenc_aux_info_sizes, demux->cenc_aux_sample_count)) {
7534               GST_ERROR_OBJECT (demux, "failed to parse cenc auxiliary info");
7535               ret = GST_FLOW_ERROR;
7536               gst_adapter_unmap (demux->adapter);
7537               g_free (demux->cenc_aux_info_sizes);
7538               demux->cenc_aux_info_sizes = NULL;
7539               goto done;
7540             }
7541             demux->cenc_aux_info_offset = 0;
7542             g_free (demux->cenc_aux_info_sizes);
7543             demux->cenc_aux_info_sizes = NULL;
7544             gst_adapter_unmap (demux->adapter);
7545           }
7546           gst_qtdemux_drop_data (demux, demux->todrop);
7547         }
7548
7549         /* first buffer? */
7550         /* initial newsegment sent here after having added pads,
7551          * possible others in sink_event */
7552         gst_qtdemux_check_send_pending_segment (demux);
7553
7554         /* Figure out which stream this packet belongs to */
7555         for (i = 0; i < QTDEMUX_N_STREAMS (demux); i++) {
7556           stream = QTDEMUX_NTH_STREAM (demux, i);
7557           if (stream->sample_index >= stream->n_samples) {
7558             /* reset to be checked below G_UNLIKELY (stream == NULL) */
7559             stream = NULL;
7560             continue;
7561           }
7562           GST_LOG_OBJECT (demux,
7563               "Checking track-id %u (sample_index:%d / offset:%"
7564               G_GUINT64_FORMAT " / size:%d)", stream->track_id,
7565               stream->sample_index,
7566               stream->samples[stream->sample_index].offset,
7567               stream->samples[stream->sample_index].size);
7568
7569           if (stream->samples[stream->sample_index].offset == demux->offset)
7570             break;
7571         }
7572
7573         if (G_UNLIKELY (stream == NULL))
7574           goto unknown_stream;
7575
7576         gst_qtdemux_stream_check_and_change_stsd_index (demux, stream);
7577
7578         if (stream->new_caps) {
7579           gst_qtdemux_configure_stream (demux, stream);
7580         }
7581
7582         /* Put data in a buffer, set timestamps, caps, ... */
7583         sample = &stream->samples[stream->sample_index];
7584
7585         if (G_LIKELY (!(STREAM_IS_EOS (stream)))) {
7586           GST_DEBUG_OBJECT (demux, "stream : %" GST_FOURCC_FORMAT,
7587               GST_FOURCC_ARGS (CUR_STREAM (stream)->fourcc));
7588
7589           dts = QTSAMPLE_DTS (stream, sample);
7590           pts = QTSAMPLE_PTS (stream, sample);
7591           duration = QTSAMPLE_DUR_DTS (stream, sample, dts);
7592           keyframe = QTSAMPLE_KEYFRAME (stream, sample);
7593
7594           /* check for segment end */
7595           if (G_UNLIKELY (demux->segment.stop != -1
7596                   && demux->segment.stop <= pts && stream->on_keyframe)
7597               && !(demux->upstream_format_is_time && demux->segment.rate < 0)) {
7598             GST_DEBUG_OBJECT (demux, "we reached the end of our segment.");
7599             stream->time_position = GST_CLOCK_TIME_NONE;        /* this means EOS */
7600
7601             /* skip this data, stream is EOS */
7602             gst_adapter_flush (demux->adapter, demux->neededbytes);
7603             demux->offset += demux->neededbytes;
7604
7605             /* check if all streams are eos */
7606             ret = GST_FLOW_EOS;
7607             for (i = 0; i < QTDEMUX_N_STREAMS (demux); i++) {
7608               if (!STREAM_IS_EOS (QTDEMUX_NTH_STREAM (demux, i))) {
7609                 ret = GST_FLOW_OK;
7610                 break;
7611               }
7612             }
7613           } else {
7614             GstBuffer *outbuf;
7615
7616             outbuf =
7617                 gst_adapter_take_buffer (demux->adapter, demux->neededbytes);
7618
7619             /* FIXME: should either be an assert or a plain check */
7620             g_return_val_if_fail (outbuf != NULL, GST_FLOW_ERROR);
7621
7622             ret = gst_qtdemux_decorate_and_push_buffer (demux, stream, outbuf,
7623                 dts, pts, duration, keyframe, dts, demux->offset);
7624           }
7625
7626           /* combine flows */
7627           GST_OBJECT_LOCK (demux);
7628           ret = gst_qtdemux_combine_flows (demux, stream, ret);
7629           GST_OBJECT_UNLOCK (demux);
7630         } else {
7631           /* skip this data, stream is EOS */
7632           gst_adapter_flush (demux->adapter, demux->neededbytes);
7633         }
7634
7635         stream->sample_index++;
7636         stream->offset_in_sample = 0;
7637
7638         /* update current offset and figure out size of next buffer */
7639         GST_LOG_OBJECT (demux, "increasing offset %" G_GUINT64_FORMAT " by %u",
7640             demux->offset, demux->neededbytes);
7641         demux->offset += demux->neededbytes;
7642         GST_LOG_OBJECT (demux, "offset is now %" G_GUINT64_FORMAT,
7643             demux->offset);
7644
7645
7646         if (ret == GST_FLOW_EOS) {
7647           GST_DEBUG_OBJECT (demux, "All streams are EOS, signal upstream");
7648           demux->neededbytes = -1;
7649           goto eos;
7650         }
7651
7652         if ((demux->neededbytes = next_entry_size (demux)) == -1) {
7653           if (demux->fragmented) {
7654             GST_DEBUG_OBJECT (demux, "(temporarily) out of fragmented samples");
7655             /* there may be more to follow, only finish this atom */
7656             demux->todrop = demux->mdatleft;
7657             demux->neededbytes = demux->todrop;
7658             break;
7659           }
7660           goto eos;
7661         }
7662         if (ret != GST_FLOW_OK && ret != GST_FLOW_NOT_LINKED) {
7663           goto non_ok_unlinked_flow;
7664         }
7665         break;
7666       }
7667       default:
7668         goto invalid_state;
7669     }
7670   }
7671
7672   /* when buffering movie data, at least show user something is happening */
7673   if (ret == GST_FLOW_OK && demux->state == QTDEMUX_STATE_BUFFER_MDAT &&
7674       gst_adapter_available (demux->adapter) <= demux->neededbytes) {
7675     gst_qtdemux_post_progress (demux, gst_adapter_available (demux->adapter),
7676         demux->neededbytes);
7677   }
7678 done:
7679
7680   return ret;
7681
7682   /* ERRORS */
7683 non_ok_unlinked_flow:
7684   {
7685     GST_DEBUG_OBJECT (demux, "Stopping, combined return flow %s",
7686         gst_flow_get_name (ret));
7687     return ret;
7688   }
7689 unknown_stream:
7690   {
7691     GST_ELEMENT_ERROR (demux, STREAM, FAILED, (NULL), ("unknown stream found"));
7692     ret = GST_FLOW_ERROR;
7693     goto done;
7694   }
7695 eos:
7696   {
7697     GST_DEBUG_OBJECT (demux, "no next entry, EOS");
7698     ret = GST_FLOW_EOS;
7699     goto done;
7700   }
7701 invalid_state:
7702   {
7703     GST_ELEMENT_ERROR (demux, STREAM, FAILED,
7704         (NULL), ("qtdemuxer invalid state %d", demux->state));
7705     ret = GST_FLOW_ERROR;
7706     goto done;
7707   }
7708 no_moov:
7709   {
7710     GST_ELEMENT_ERROR (demux, STREAM, FAILED,
7711         (NULL), ("no 'moov' atom within the first 10 MB"));
7712     ret = GST_FLOW_ERROR;
7713     goto done;
7714   }
7715 }
7716
7717 static gboolean
7718 qtdemux_sink_activate (GstPad * sinkpad, GstObject * parent)
7719 {
7720   GstQuery *query;
7721   gboolean pull_mode;
7722
7723   query = gst_query_new_scheduling ();
7724
7725   if (!gst_pad_peer_query (sinkpad, query)) {
7726     gst_query_unref (query);
7727     goto activate_push;
7728   }
7729
7730   pull_mode = gst_query_has_scheduling_mode_with_flags (query,
7731       GST_PAD_MODE_PULL, GST_SCHEDULING_FLAG_SEEKABLE);
7732   gst_query_unref (query);
7733
7734   if (!pull_mode)
7735     goto activate_push;
7736
7737   GST_DEBUG_OBJECT (sinkpad, "activating pull");
7738   return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PULL, TRUE);
7739
7740 activate_push:
7741   {
7742     GST_DEBUG_OBJECT (sinkpad, "activating push");
7743     return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PUSH, TRUE);
7744   }
7745 }
7746
7747 static gboolean
7748 qtdemux_sink_activate_mode (GstPad * sinkpad, GstObject * parent,
7749     GstPadMode mode, gboolean active)
7750 {
7751   gboolean res;
7752   GstQTDemux *demux = GST_QTDEMUX (parent);
7753
7754   switch (mode) {
7755     case GST_PAD_MODE_PUSH:
7756       demux->pullbased = FALSE;
7757       res = TRUE;
7758       break;
7759     case GST_PAD_MODE_PULL:
7760       if (active) {
7761         demux->pullbased = TRUE;
7762         res = gst_pad_start_task (sinkpad, (GstTaskFunction) gst_qtdemux_loop,
7763             sinkpad, NULL);
7764       } else {
7765         res = gst_pad_stop_task (sinkpad);
7766       }
7767       break;
7768     default:
7769       res = FALSE;
7770       break;
7771   }
7772   return res;
7773 }
7774
7775 #ifdef HAVE_ZLIB
7776 static void *
7777 qtdemux_inflate (void *z_buffer, guint z_length, guint * length)
7778 {
7779   guint8 *buffer;
7780   z_stream z;
7781   int ret;
7782
7783   memset (&z, 0, sizeof (z));
7784   z.zalloc = NULL;
7785   z.zfree = NULL;
7786   z.opaque = NULL;
7787
7788   if ((ret = inflateInit (&z)) != Z_OK) {
7789     GST_ERROR ("inflateInit() returned %d", ret);
7790     return NULL;
7791   }
7792
7793   z.next_in = z_buffer;
7794   z.avail_in = z_length;
7795
7796   buffer = (guint8 *) g_malloc (*length);
7797   z.avail_out = *length;
7798   z.next_out = (Bytef *) buffer;
7799   do {
7800     ret = inflate (&z, Z_NO_FLUSH);
7801     if (ret == Z_STREAM_END) {
7802       break;
7803     } else if (ret != Z_OK) {
7804       GST_WARNING ("inflate() returned %d", ret);
7805       break;
7806     }
7807
7808     *length += 4096;
7809     buffer = (guint8 *) g_realloc (buffer, *length);
7810     z.next_out = (Bytef *) (buffer + z.total_out);
7811     z.avail_out += 4096;
7812   } while (z.avail_in > 0);
7813
7814   if (ret != Z_STREAM_END) {
7815     g_free (buffer);
7816     buffer = NULL;
7817     *length = 0;
7818   } else {
7819     *length = z.total_out;
7820   }
7821
7822   inflateEnd (&z);
7823
7824   return buffer;
7825 }
7826 #endif /* HAVE_ZLIB */
7827
7828 static gboolean
7829 qtdemux_parse_moov (GstQTDemux * qtdemux, const guint8 * buffer, guint length)
7830 {
7831   GNode *cmov;
7832
7833   qtdemux->moov_node = g_node_new ((guint8 *) buffer);
7834
7835   /* counts as header data */
7836   qtdemux->header_size += length;
7837
7838   GST_DEBUG_OBJECT (qtdemux, "parsing 'moov' atom");
7839   qtdemux_parse_node (qtdemux, qtdemux->moov_node, buffer, length);
7840
7841   cmov = qtdemux_tree_get_child_by_type (qtdemux->moov_node, FOURCC_cmov);
7842   if (cmov) {
7843     guint32 method;
7844     GNode *dcom;
7845     GNode *cmvd;
7846     guint32 dcom_len;
7847
7848     dcom = qtdemux_tree_get_child_by_type (cmov, FOURCC_dcom);
7849     cmvd = qtdemux_tree_get_child_by_type (cmov, FOURCC_cmvd);
7850     if (dcom == NULL || cmvd == NULL)
7851       goto invalid_compression;
7852
7853     dcom_len = QT_UINT32 (dcom->data);
7854     if (dcom_len < 12)
7855       goto invalid_compression;
7856
7857     method = QT_FOURCC ((guint8 *) dcom->data + 8);
7858     switch (method) {
7859 #ifdef HAVE_ZLIB
7860       case FOURCC_zlib:{
7861         guint uncompressed_length;
7862         guint compressed_length;
7863         guint8 *buf;
7864         guint32 cmvd_len;
7865
7866         cmvd_len = QT_UINT32 ((guint8 *) cmvd->data);
7867         if (cmvd_len < 12)
7868           goto invalid_compression;
7869
7870         uncompressed_length = QT_UINT32 ((guint8 *) cmvd->data + 8);
7871         compressed_length = cmvd_len - 12;
7872         GST_LOG ("length = %u", uncompressed_length);
7873
7874         buf =
7875             (guint8 *) qtdemux_inflate ((guint8 *) cmvd->data + 12,
7876             compressed_length, &uncompressed_length);
7877
7878         if (buf) {
7879           qtdemux->moov_node_compressed = qtdemux->moov_node;
7880           qtdemux->moov_node = g_node_new (buf);
7881
7882           qtdemux_parse_node (qtdemux, qtdemux->moov_node, buf,
7883               uncompressed_length);
7884         }
7885         break;
7886       }
7887 #endif /* HAVE_ZLIB */
7888       default:
7889         GST_WARNING_OBJECT (qtdemux, "unknown or unhandled header compression "
7890             "type %" GST_FOURCC_FORMAT, GST_FOURCC_ARGS (method));
7891         break;
7892     }
7893   }
7894   return TRUE;
7895
7896   /* ERRORS */
7897 invalid_compression:
7898   {
7899     GST_ERROR_OBJECT (qtdemux, "invalid compressed header");
7900     return FALSE;
7901   }
7902 }
7903
7904 static gboolean
7905 qtdemux_parse_container (GstQTDemux * qtdemux, GNode * node, const guint8 * buf,
7906     const guint8 * end)
7907 {
7908   while (G_UNLIKELY (buf < end)) {
7909     GNode *child;
7910     guint32 len;
7911
7912     if (G_UNLIKELY (buf + 4 > end)) {
7913       GST_LOG_OBJECT (qtdemux, "buffer overrun");
7914       break;
7915     }
7916     len = QT_UINT32 (buf);
7917     if (G_UNLIKELY (len == 0)) {
7918       GST_LOG_OBJECT (qtdemux, "empty container");
7919       break;
7920     }
7921     if (G_UNLIKELY (len < 8)) {
7922       GST_WARNING_OBJECT (qtdemux, "length too short (%d < 8)", len);
7923       break;
7924     }
7925     if (G_UNLIKELY (len > (end - buf))) {
7926       GST_WARNING_OBJECT (qtdemux, "length too long (%d > %d)", len,
7927           (gint) (end - buf));
7928       break;
7929     }
7930
7931     child = g_node_new ((guint8 *) buf);
7932     g_node_append (node, child);
7933     GST_LOG_OBJECT (qtdemux, "adding new node of len %d", len);
7934     qtdemux_parse_node (qtdemux, child, buf, len);
7935
7936     buf += len;
7937   }
7938   return TRUE;
7939 }
7940
7941 static gboolean
7942 qtdemux_parse_theora_extension (GstQTDemux * qtdemux, QtDemuxStream * stream,
7943     GNode * xdxt)
7944 {
7945   int len = QT_UINT32 (xdxt->data);
7946   guint8 *buf = xdxt->data;
7947   guint8 *end = buf + len;
7948   GstBuffer *buffer;
7949
7950   /* skip size and type */
7951   buf += 8;
7952   end -= 8;
7953
7954   while (buf < end) {
7955     gint size;
7956     guint32 type;
7957
7958     size = QT_UINT32 (buf);
7959     type = QT_FOURCC (buf + 4);
7960
7961     GST_LOG_OBJECT (qtdemux, "%p %p", buf, end);
7962
7963     if (buf + size > end || size <= 0)
7964       break;
7965
7966     buf += 8;
7967     size -= 8;
7968
7969     GST_WARNING_OBJECT (qtdemux, "have cookie %" GST_FOURCC_FORMAT,
7970         GST_FOURCC_ARGS (type));
7971
7972     switch (type) {
7973       case FOURCC_tCtH:
7974         buffer = gst_buffer_new_and_alloc (size);
7975         gst_buffer_fill (buffer, 0, buf, size);
7976         stream->buffers = g_slist_append (stream->buffers, buffer);
7977         GST_LOG_OBJECT (qtdemux, "parsing theora header");
7978         break;
7979       case FOURCC_tCt_:
7980         buffer = gst_buffer_new_and_alloc (size);
7981         gst_buffer_fill (buffer, 0, buf, size);
7982         stream->buffers = g_slist_append (stream->buffers, buffer);
7983         GST_LOG_OBJECT (qtdemux, "parsing theora comment");
7984         break;
7985       case FOURCC_tCtC:
7986         buffer = gst_buffer_new_and_alloc (size);
7987         gst_buffer_fill (buffer, 0, buf, size);
7988         stream->buffers = g_slist_append (stream->buffers, buffer);
7989         GST_LOG_OBJECT (qtdemux, "parsing theora codebook");
7990         break;
7991       default:
7992         GST_WARNING_OBJECT (qtdemux,
7993             "unknown theora cookie %" GST_FOURCC_FORMAT,
7994             GST_FOURCC_ARGS (type));
7995         break;
7996     }
7997     buf += size;
7998   }
7999   return TRUE;
8000 }
8001
8002 static gboolean
8003 qtdemux_parse_node (GstQTDemux * qtdemux, GNode * node, const guint8 * buffer,
8004     guint length)
8005 {
8006   guint32 fourcc = 0;
8007   guint32 node_length = 0;
8008   const QtNodeType *type;
8009   const guint8 *end;
8010
8011   GST_LOG_OBJECT (qtdemux, "qtdemux_parse buffer %p length %u", buffer, length);
8012
8013   if (G_UNLIKELY (length < 8))
8014     goto not_enough_data;
8015
8016   node_length = QT_UINT32 (buffer);
8017   fourcc = QT_FOURCC (buffer + 4);
8018
8019   /* ignore empty nodes */
8020   if (G_UNLIKELY (fourcc == 0 || node_length == 8))
8021     return TRUE;
8022
8023   type = qtdemux_type_get (fourcc);
8024
8025   end = buffer + length;
8026
8027   GST_LOG_OBJECT (qtdemux,
8028       "parsing '%" GST_FOURCC_FORMAT "', length=%u, name '%s'",
8029       GST_FOURCC_ARGS (fourcc), node_length, type->name);
8030
8031   if (node_length > length)
8032     goto broken_atom_size;
8033
8034   if (type->flags & QT_FLAG_CONTAINER) {
8035     qtdemux_parse_container (qtdemux, node, buffer + 8, end);
8036   } else {
8037     switch (fourcc) {
8038       case FOURCC_stsd:
8039       {
8040         if (node_length < 20) {
8041           GST_LOG_OBJECT (qtdemux, "skipping small stsd box");
8042           break;
8043         }
8044         GST_DEBUG_OBJECT (qtdemux,
8045             "parsing stsd (sample table, sample description) atom");
8046         /* Skip over 8 byte atom hdr + 1 byte version, 3 bytes flags, 4 byte num_entries */
8047         qtdemux_parse_container (qtdemux, node, buffer + 16, end);
8048         break;
8049       }
8050       case FOURCC_mp4a:
8051       case FOURCC_alac:
8052       case FOURCC_fLaC:
8053       case FOURCC_aavd:
8054       {
8055         guint32 version;
8056         guint32 offset;
8057         guint min_size;
8058
8059         /* also read alac (or whatever) in stead of mp4a in the following,
8060          * since a similar layout is used in other cases as well */
8061         if (fourcc == FOURCC_mp4a)
8062           min_size = 20;
8063         else if (fourcc == FOURCC_fLaC)
8064           min_size = 86;
8065         else
8066           min_size = 40;
8067
8068         /* There are two things we might encounter here: a true mp4a atom, and
8069            an mp4a entry in an stsd atom. The latter is what we're interested
8070            in, and it looks like an atom, but isn't really one. The true mp4a
8071            atom is short, so we detect it based on length here. */
8072         if (length < min_size) {
8073           GST_LOG_OBJECT (qtdemux, "skipping small %" GST_FOURCC_FORMAT " box",
8074               GST_FOURCC_ARGS (fourcc));
8075           break;
8076         }
8077
8078         /* 'version' here is the sound sample description version. Types 0 and
8079            1 are documented in the QTFF reference, but type 2 is not: it's
8080            described in Apple header files instead (struct SoundDescriptionV2
8081            in Movies.h) */
8082         version = QT_UINT16 (buffer + 16);
8083
8084         GST_DEBUG_OBJECT (qtdemux, "%" GST_FOURCC_FORMAT " version 0x%08x",
8085             GST_FOURCC_ARGS (fourcc), version);
8086
8087         /* parse any esds descriptors */
8088         switch (version) {
8089           case 0:
8090             offset = 0x24;
8091             break;
8092           case 1:
8093             offset = 0x34;
8094             break;
8095           case 2:
8096             offset = 0x48;
8097             break;
8098           default:
8099             GST_WARNING_OBJECT (qtdemux,
8100                 "unhandled %" GST_FOURCC_FORMAT " version 0x%08x",
8101                 GST_FOURCC_ARGS (fourcc), version);
8102             offset = 0;
8103             break;
8104         }
8105         if (offset)
8106           qtdemux_parse_container (qtdemux, node, buffer + offset, end);
8107         break;
8108       }
8109       case FOURCC_mp4v:
8110       case FOURCC_MP4V:
8111       case FOURCC_fmp4:
8112       case FOURCC_FMP4:
8113       case FOURCC_apcs:
8114       case FOURCC_apch:
8115       case FOURCC_apcn:
8116       case FOURCC_apco:
8117       case FOURCC_ap4h:
8118       case FOURCC_xvid:
8119       case FOURCC_XVID:
8120       case FOURCC_H264:
8121       case FOURCC_avc1:
8122       case FOURCC_avc3:
8123       case FOURCC_H265:
8124       case FOURCC_hvc1:
8125       case FOURCC_hev1:
8126       case FOURCC_dvh1:
8127       case FOURCC_dvhe:
8128       case FOURCC_mjp2:
8129       case FOURCC_encv:
8130       {
8131         guint32 version;
8132         guint32 str_len;
8133
8134         /* codec_data is contained inside these atoms, which all have
8135          * the same format. */
8136         /* video sample description size is 86 bytes without extension.
8137          * node_length have to be bigger than 86 bytes because video sample
8138          * description can include extensions such as esds, fiel, glbl, etc. */
8139         if (node_length < 86) {
8140           GST_WARNING_OBJECT (qtdemux, "%" GST_FOURCC_FORMAT
8141               " sample description length too short (%u < 86)",
8142               GST_FOURCC_ARGS (fourcc), node_length);
8143           break;
8144         }
8145
8146         GST_DEBUG_OBJECT (qtdemux, "parsing in %" GST_FOURCC_FORMAT,
8147             GST_FOURCC_ARGS (fourcc));
8148
8149         /* version (2 bytes) : this is set to 0, unless a compressor has changed
8150          *              its data format.
8151          * revision level (2 bytes) : must be set to 0. */
8152         version = QT_UINT32 (buffer + 16);
8153         GST_DEBUG_OBJECT (qtdemux, "version %08x", version);
8154
8155         /* compressor name : PASCAL string and informative purposes
8156          * first byte : the number of bytes to be displayed.
8157          *              it has to be less than 32 because it is reserved
8158          *              space of 32 bytes total including itself. */
8159         str_len = QT_UINT8 (buffer + 50);
8160         if (str_len < 32)
8161           GST_DEBUG_OBJECT (qtdemux, "compressorname = %.*s", str_len,
8162               (char *) buffer + 51);
8163         else
8164           GST_WARNING_OBJECT (qtdemux,
8165               "compressorname length too big (%u > 31)", str_len);
8166
8167         GST_MEMDUMP_OBJECT (qtdemux, "video sample description", buffer,
8168             end - buffer);
8169         qtdemux_parse_container (qtdemux, node, buffer + 86, end);
8170         break;
8171       }
8172       case FOURCC_meta:
8173       {
8174         GST_DEBUG_OBJECT (qtdemux, "parsing meta atom");
8175
8176         /* You are reading this correctly. QTFF specifies that the
8177          * metadata atom is a short atom, whereas ISO BMFF specifies
8178          * it's a full atom. But since so many people are doing things
8179          * differently, we actually peek into the atom to see which
8180          * variant it is */
8181         if (length < 16) {
8182           GST_LOG_OBJECT (qtdemux, "skipping small %" GST_FOURCC_FORMAT " box",
8183               GST_FOURCC_ARGS (fourcc));
8184           break;
8185         }
8186         if (QT_FOURCC (buffer + 12) == FOURCC_hdlr) {
8187           /* Variant 1: What QTFF specifies. 'meta' is a short header which
8188            * starts with a 'hdlr' atom */
8189           qtdemux_parse_container (qtdemux, node, buffer + 8, end);
8190         } else if (QT_UINT32 (buffer + 8) == 0x00000000) {
8191           /* Variant 2: What ISO BMFF specifies. 'meta' is a _full_ atom
8192            * with version/flags both set to zero */
8193           qtdemux_parse_container (qtdemux, node, buffer + 12, end);
8194         } else
8195           GST_WARNING_OBJECT (qtdemux, "Unknown 'meta' atom format");
8196         break;
8197       }
8198       case FOURCC_mp4s:
8199       {
8200         GST_MEMDUMP_OBJECT (qtdemux, "mp4s", buffer, end - buffer);
8201         /* Skip 8 byte header, plus 8 byte version + flags + entry_count */
8202         qtdemux_parse_container (qtdemux, node, buffer + 16, end);
8203         break;
8204       }
8205       case FOURCC_XiTh:
8206       {
8207         guint32 version;
8208         guint32 offset;
8209
8210         if (length < 16) {
8211           GST_LOG_OBJECT (qtdemux, "skipping small %" GST_FOURCC_FORMAT " box",
8212               GST_FOURCC_ARGS (fourcc));
8213           break;
8214         }
8215
8216         version = QT_UINT32 (buffer + 12);
8217         GST_DEBUG_OBJECT (qtdemux, "parsing XiTh atom version 0x%08x", version);
8218
8219         switch (version) {
8220           case 0x00000001:
8221             offset = 0x62;
8222             break;
8223           default:
8224             GST_DEBUG_OBJECT (qtdemux, "unknown version 0x%08x", version);
8225             offset = 0;
8226             break;
8227         }
8228         if (offset) {
8229           if (length < offset) {
8230             GST_WARNING_OBJECT (qtdemux,
8231                 "skipping too small %" GST_FOURCC_FORMAT " box",
8232                 GST_FOURCC_ARGS (fourcc));
8233             break;
8234           }
8235           qtdemux_parse_container (qtdemux, node, buffer + offset, end);
8236         }
8237         break;
8238       }
8239       case FOURCC_in24:
8240       {
8241         qtdemux_parse_container (qtdemux, node, buffer + 0x34, end);
8242         break;
8243       }
8244       case FOURCC_uuid:
8245       {
8246         qtdemux_parse_uuid (qtdemux, buffer, end - buffer);
8247         break;
8248       }
8249       case FOURCC_enca:
8250       {
8251         qtdemux_parse_container (qtdemux, node, buffer + 36, end);
8252         break;
8253       }
8254       default:
8255         if (!strcmp (type->name, "unknown"))
8256           GST_MEMDUMP ("Unknown tag", buffer + 4, end - buffer - 4);
8257         break;
8258     }
8259   }
8260   GST_LOG_OBJECT (qtdemux, "parsed '%" GST_FOURCC_FORMAT "'",
8261       GST_FOURCC_ARGS (fourcc));
8262   return TRUE;
8263
8264 /* ERRORS */
8265 not_enough_data:
8266   {
8267     GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
8268         (_("This file is corrupt and cannot be played.")),
8269         ("Not enough data for an atom header, got only %u bytes", length));
8270     return FALSE;
8271   }
8272 broken_atom_size:
8273   {
8274     GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
8275         (_("This file is corrupt and cannot be played.")),
8276         ("Atom '%" GST_FOURCC_FORMAT "' has size of %u bytes, but we have only "
8277             "%u bytes available.", GST_FOURCC_ARGS (fourcc), node_length,
8278             length));
8279     return FALSE;
8280   }
8281 }
8282
8283 static void
8284 qtdemux_do_allocation (QtDemuxStream * stream, GstQTDemux * qtdemux)
8285 {
8286 /* FIXME: This can only reliably work if demuxers have a
8287  * separate streaming thread per srcpad. This should be
8288  * done in a demuxer base class, which integrates parts
8289  * of multiqueue
8290  *
8291  * https://bugzilla.gnome.org/show_bug.cgi?id=701856
8292  */
8293 #if 0
8294   GstQuery *query;
8295
8296   query = gst_query_new_allocation (stream->caps, FALSE);
8297
8298   if (!gst_pad_peer_query (stream->pad, query)) {
8299     /* not a problem, just debug a little */
8300     GST_DEBUG_OBJECT (qtdemux, "peer ALLOCATION query failed");
8301   }
8302
8303   if (stream->allocator)
8304     gst_object_unref (stream->allocator);
8305
8306   if (gst_query_get_n_allocation_params (query) > 0) {
8307     /* try the allocator */
8308     gst_query_parse_nth_allocation_param (query, 0, &stream->allocator,
8309         &stream->params);
8310     stream->use_allocator = TRUE;
8311   } else {
8312     stream->allocator = NULL;
8313     gst_allocation_params_init (&stream->params);
8314     stream->use_allocator = FALSE;
8315   }
8316   gst_query_unref (query);
8317 #endif
8318 }
8319
8320 static gboolean
8321 pad_query (const GValue * item, GValue * value, gpointer user_data)
8322 {
8323   GstPad *pad = g_value_get_object (item);
8324   GstQuery *query = user_data;
8325   gboolean res;
8326
8327   res = gst_pad_peer_query (pad, query);
8328
8329   if (res) {
8330     g_value_set_boolean (value, TRUE);
8331     return FALSE;
8332   }
8333
8334   GST_INFO_OBJECT (pad, "pad peer query failed");
8335   return TRUE;
8336 }
8337
8338 static gboolean
8339 gst_qtdemux_run_query (GstElement * element, GstQuery * query,
8340     GstPadDirection direction)
8341 {
8342   GstIterator *it;
8343   GstIteratorFoldFunction func = pad_query;
8344   GValue res = { 0, };
8345
8346   g_value_init (&res, G_TYPE_BOOLEAN);
8347   g_value_set_boolean (&res, FALSE);
8348
8349   /* Ask neighbor */
8350   if (direction == GST_PAD_SRC)
8351     it = gst_element_iterate_src_pads (element);
8352   else
8353     it = gst_element_iterate_sink_pads (element);
8354
8355   while (gst_iterator_fold (it, func, &res, query) == GST_ITERATOR_RESYNC)
8356     gst_iterator_resync (it);
8357
8358   gst_iterator_free (it);
8359
8360   return g_value_get_boolean (&res);
8361 }
8362
8363 static void
8364 gst_qtdemux_request_protection_context (GstQTDemux * qtdemux,
8365     QtDemuxStream * stream)
8366 {
8367   GstQuery *query;
8368   GstContext *ctxt;
8369   GstElement *element = GST_ELEMENT (qtdemux);
8370   GstStructure *st;
8371   gchar **filtered_sys_ids;
8372   GValue event_list = G_VALUE_INIT;
8373   GList *walk;
8374
8375   /* 1. Check if we already have the context. */
8376   if (qtdemux->preferred_protection_system_id != NULL) {
8377     GST_LOG_OBJECT (element,
8378         "already have the protection context, no need to request it again");
8379     return;
8380   }
8381
8382   g_ptr_array_add (qtdemux->protection_system_ids, NULL);
8383   filtered_sys_ids = gst_protection_filter_systems_by_available_decryptors (
8384       (const gchar **) qtdemux->protection_system_ids->pdata);
8385
8386   g_ptr_array_remove_index (qtdemux->protection_system_ids,
8387       qtdemux->protection_system_ids->len - 1);
8388   GST_TRACE_OBJECT (qtdemux, "detected %u protection systems, we have "
8389       "decryptors for %u of them, running context request",
8390       qtdemux->protection_system_ids->len,
8391       filtered_sys_ids ? g_strv_length (filtered_sys_ids) : 0);
8392
8393
8394   if (stream->protection_scheme_event_queue.length) {
8395     GST_TRACE_OBJECT (qtdemux, "using stream event queue, length %u",
8396         stream->protection_scheme_event_queue.length);
8397     walk = stream->protection_scheme_event_queue.tail;
8398   } else {
8399     GST_TRACE_OBJECT (qtdemux, "using demuxer event queue, length %u",
8400         qtdemux->protection_event_queue.length);
8401     walk = qtdemux->protection_event_queue.tail;
8402   }
8403
8404   g_value_init (&event_list, GST_TYPE_LIST);
8405   for (; walk; walk = g_list_previous (walk)) {
8406     GValue *event_value = g_new0 (GValue, 1);
8407     g_value_init (event_value, GST_TYPE_EVENT);
8408     g_value_set_boxed (event_value, walk->data);
8409     gst_value_list_append_and_take_value (&event_list, event_value);
8410   }
8411
8412   /*  2a) Query downstream with GST_QUERY_CONTEXT for the context and
8413    *      check if downstream already has a context of the specific type
8414    *  2b) Query upstream as above.
8415    */
8416   query = gst_query_new_context ("drm-preferred-decryption-system-id");
8417   st = gst_query_writable_structure (query);
8418   gst_structure_set (st, "track-id", G_TYPE_UINT, stream->track_id,
8419       "available-stream-encryption-systems", G_TYPE_STRV, filtered_sys_ids,
8420       NULL);
8421   gst_structure_set_value (st, "stream-encryption-events", &event_list);
8422   if (gst_qtdemux_run_query (element, query, GST_PAD_SRC)) {
8423     gst_query_parse_context (query, &ctxt);
8424     GST_INFO_OBJECT (element, "found context (%p) in downstream query", ctxt);
8425     gst_element_set_context (element, ctxt);
8426   } else if (gst_qtdemux_run_query (element, query, GST_PAD_SINK)) {
8427     gst_query_parse_context (query, &ctxt);
8428     GST_INFO_OBJECT (element, "found context (%p) in upstream query", ctxt);
8429     gst_element_set_context (element, ctxt);
8430   } else {
8431     /* 3) Post a GST_MESSAGE_NEED_CONTEXT message on the bus with
8432      *    the required context type and afterwards check if a
8433      *    usable context was set now as in 1). The message could
8434      *    be handled by the parent bins of the element and the
8435      *    application.
8436      */
8437     GstMessage *msg;
8438
8439     GST_INFO_OBJECT (element, "posting need context message");
8440     msg = gst_message_new_need_context (GST_OBJECT_CAST (element),
8441         "drm-preferred-decryption-system-id");
8442     st = (GstStructure *) gst_message_get_structure (msg);
8443     gst_structure_set (st, "track-id", G_TYPE_UINT, stream->track_id,
8444         "available-stream-encryption-systems", G_TYPE_STRV, filtered_sys_ids,
8445         NULL);
8446
8447     gst_structure_set_value (st, "stream-encryption-events", &event_list);
8448     gst_element_post_message (element, msg);
8449   }
8450
8451   g_strfreev (filtered_sys_ids);
8452   g_value_unset (&event_list);
8453   gst_query_unref (query);
8454 }
8455
8456 static gboolean
8457 gst_qtdemux_configure_protected_caps (GstQTDemux * qtdemux,
8458     QtDemuxStream * stream)
8459 {
8460   GstStructure *s;
8461   const gchar *selected_system = NULL;
8462
8463   g_return_val_if_fail (qtdemux != NULL, FALSE);
8464   g_return_val_if_fail (stream != NULL, FALSE);
8465   g_return_val_if_fail (gst_caps_get_size (CUR_STREAM (stream)->caps) == 1,
8466       FALSE);
8467
8468   if (stream->protection_scheme_type == FOURCC_aavd) {
8469     s = gst_caps_get_structure (CUR_STREAM (stream)->caps, 0);
8470     if (!gst_structure_has_name (s, "application/x-aavd")) {
8471       gst_structure_set (s,
8472           "original-media-type", G_TYPE_STRING, gst_structure_get_name (s),
8473           NULL);
8474       gst_structure_set_name (s, "application/x-aavd");
8475     }
8476     return TRUE;
8477   }
8478
8479   if (stream->protection_scheme_type != FOURCC_cenc
8480       && stream->protection_scheme_type != FOURCC_cbcs) {
8481     GST_ERROR_OBJECT (qtdemux,
8482         "unsupported protection scheme: %" GST_FOURCC_FORMAT,
8483         GST_FOURCC_ARGS (stream->protection_scheme_type));
8484     return FALSE;
8485   }
8486
8487   s = gst_caps_get_structure (CUR_STREAM (stream)->caps, 0);
8488   if (!gst_structure_has_name (s, "application/x-cenc")) {
8489     gst_structure_set (s,
8490         "original-media-type", G_TYPE_STRING, gst_structure_get_name (s), NULL);
8491     gst_structure_set (s, "cipher-mode", G_TYPE_STRING,
8492         (stream->protection_scheme_type == FOURCC_cbcs) ? "cbcs" : "cenc",
8493         NULL);
8494     gst_structure_set_name (s, "application/x-cenc");
8495   }
8496
8497   if (qtdemux->protection_system_ids == NULL) {
8498     GST_DEBUG_OBJECT (qtdemux, "stream is protected using cenc, but no "
8499         "cenc protection system information has been found, not setting a "
8500         "protection system UUID");
8501     return TRUE;
8502   }
8503
8504   gst_qtdemux_request_protection_context (qtdemux, stream);
8505   if (qtdemux->preferred_protection_system_id != NULL) {
8506     const gchar *preferred_system_array[] =
8507         { qtdemux->preferred_protection_system_id, NULL };
8508
8509     selected_system = gst_protection_select_system (preferred_system_array);
8510
8511     if (selected_system) {
8512       GST_TRACE_OBJECT (qtdemux, "selected preferred system %s",
8513           qtdemux->preferred_protection_system_id);
8514     } else {
8515       GST_WARNING_OBJECT (qtdemux, "could not select preferred system %s "
8516           "because there is no available decryptor",
8517           qtdemux->preferred_protection_system_id);
8518     }
8519   }
8520
8521   if (!selected_system) {
8522     g_ptr_array_add (qtdemux->protection_system_ids, NULL);
8523     selected_system = gst_protection_select_system ((const gchar **)
8524         qtdemux->protection_system_ids->pdata);
8525     g_ptr_array_remove_index (qtdemux->protection_system_ids,
8526         qtdemux->protection_system_ids->len - 1);
8527   }
8528
8529   if (!selected_system) {
8530     GST_ERROR_OBJECT (qtdemux, "stream is protected, but no "
8531         "suitable decryptor element has been found");
8532     return FALSE;
8533   }
8534
8535   GST_DEBUG_OBJECT (qtdemux, "selected protection system is %s",
8536       selected_system);
8537
8538   gst_structure_set (s,
8539       GST_PROTECTION_SYSTEM_ID_CAPS_FIELD, G_TYPE_STRING, selected_system,
8540       NULL);
8541
8542   return TRUE;
8543 }
8544
8545 static gboolean
8546 gst_qtdemux_guess_framerate (GstQTDemux * qtdemux, QtDemuxStream * stream)
8547 {
8548   /* fps is calculated base on the duration of the average framerate since
8549    * qt does not have a fixed framerate. */
8550   gboolean fps_available = TRUE;
8551   guint32 first_duration = 0;
8552
8553   if (stream->n_samples > 0)
8554     first_duration = stream->samples[0].duration;
8555
8556   if ((stream->n_samples == 1 && first_duration == 0)
8557       || (qtdemux->fragmented && stream->n_samples_moof == 1)) {
8558     /* still frame */
8559     CUR_STREAM (stream)->fps_n = 0;
8560     CUR_STREAM (stream)->fps_d = 1;
8561   } else {
8562     if (stream->duration == 0 || stream->n_samples < 2) {
8563       CUR_STREAM (stream)->fps_n = stream->timescale;
8564       CUR_STREAM (stream)->fps_d = 1;
8565       fps_available = FALSE;
8566     } else {
8567       GstClockTime avg_duration;
8568       guint64 duration;
8569       guint32 n_samples;
8570
8571       /* duration and n_samples can be updated for fragmented format
8572        * so, framerate of fragmented format is calculated using data in a moof */
8573       if (qtdemux->fragmented && stream->n_samples_moof > 0
8574           && stream->duration_moof > 0) {
8575         n_samples = stream->n_samples_moof;
8576         duration = stream->duration_moof;
8577       } else {
8578         n_samples = stream->n_samples;
8579         duration = stream->duration;
8580       }
8581
8582       /* Calculate a framerate, ignoring the first sample which is sometimes truncated */
8583       /* stream->duration is guint64, timescale, n_samples are guint32 */
8584       avg_duration =
8585           gst_util_uint64_scale_round (duration -
8586           first_duration, GST_SECOND,
8587           (guint64) (stream->timescale) * (n_samples - 1));
8588
8589       GST_LOG_OBJECT (qtdemux,
8590           "Calculating avg sample duration based on stream (or moof) duration %"
8591           G_GUINT64_FORMAT
8592           " minus first sample %u, leaving %d samples gives %"
8593           GST_TIME_FORMAT, duration, first_duration,
8594           n_samples - 1, GST_TIME_ARGS (avg_duration));
8595
8596       fps_available =
8597           gst_video_guess_framerate (avg_duration,
8598           &CUR_STREAM (stream)->fps_n, &CUR_STREAM (stream)->fps_d);
8599
8600       GST_DEBUG_OBJECT (qtdemux,
8601           "Calculating framerate, timescale %u gave fps_n %d fps_d %d",
8602           stream->timescale, CUR_STREAM (stream)->fps_n,
8603           CUR_STREAM (stream)->fps_d);
8604     }
8605   }
8606
8607   return fps_available;
8608 }
8609
8610 static gboolean
8611 gst_qtdemux_configure_stream (GstQTDemux * qtdemux, QtDemuxStream * stream)
8612 {
8613   if (stream->subtype == FOURCC_vide) {
8614     gboolean fps_available = gst_qtdemux_guess_framerate (qtdemux, stream);
8615
8616     if (CUR_STREAM (stream)->caps) {
8617       CUR_STREAM (stream)->caps =
8618           gst_caps_make_writable (CUR_STREAM (stream)->caps);
8619
8620       if (CUR_STREAM (stream)->width && CUR_STREAM (stream)->height)
8621         gst_caps_set_simple (CUR_STREAM (stream)->caps,
8622             "width", G_TYPE_INT, CUR_STREAM (stream)->width,
8623             "height", G_TYPE_INT, CUR_STREAM (stream)->height, NULL);
8624
8625       /* set framerate if calculated framerate is reliable */
8626       if (fps_available) {
8627         gst_caps_set_simple (CUR_STREAM (stream)->caps,
8628             "framerate", GST_TYPE_FRACTION, CUR_STREAM (stream)->fps_n,
8629             CUR_STREAM (stream)->fps_d, NULL);
8630       }
8631
8632       /* calculate pixel-aspect-ratio using display width and height */
8633       GST_DEBUG_OBJECT (qtdemux,
8634           "video size %dx%d, target display size %dx%d",
8635           CUR_STREAM (stream)->width, CUR_STREAM (stream)->height,
8636           stream->display_width, stream->display_height);
8637       /* qt file might have pasp atom */
8638       if (CUR_STREAM (stream)->par_w > 0 && CUR_STREAM (stream)->par_h > 0) {
8639         GST_DEBUG_OBJECT (qtdemux, "par %d:%d", CUR_STREAM (stream)->par_w,
8640             CUR_STREAM (stream)->par_h);
8641         gst_caps_set_simple (CUR_STREAM (stream)->caps, "pixel-aspect-ratio",
8642             GST_TYPE_FRACTION, CUR_STREAM (stream)->par_w,
8643             CUR_STREAM (stream)->par_h, NULL);
8644       } else if (stream->display_width > 0 && stream->display_height > 0
8645           && CUR_STREAM (stream)->width > 0
8646           && CUR_STREAM (stream)->height > 0) {
8647         gint n, d;
8648
8649         /* calculate the pixel aspect ratio using the display and pixel w/h */
8650         n = stream->display_width * CUR_STREAM (stream)->height;
8651         d = stream->display_height * CUR_STREAM (stream)->width;
8652         if (n == d)
8653           n = d = 1;
8654         GST_DEBUG_OBJECT (qtdemux, "setting PAR to %d/%d", n, d);
8655         CUR_STREAM (stream)->par_w = n;
8656         CUR_STREAM (stream)->par_h = d;
8657         gst_caps_set_simple (CUR_STREAM (stream)->caps, "pixel-aspect-ratio",
8658             GST_TYPE_FRACTION, CUR_STREAM (stream)->par_w,
8659             CUR_STREAM (stream)->par_h, NULL);
8660       }
8661
8662       if (CUR_STREAM (stream)->interlace_mode > 0) {
8663         if (CUR_STREAM (stream)->interlace_mode == 1) {
8664           gst_caps_set_simple (CUR_STREAM (stream)->caps, "interlace-mode",
8665               G_TYPE_STRING, "progressive", NULL);
8666         } else if (CUR_STREAM (stream)->interlace_mode == 2) {
8667           gst_caps_set_simple (CUR_STREAM (stream)->caps, "interlace-mode",
8668               G_TYPE_STRING, "interleaved", NULL);
8669           if (CUR_STREAM (stream)->field_order == 9) {
8670             gst_caps_set_simple (CUR_STREAM (stream)->caps, "field-order",
8671                 G_TYPE_STRING, "top-field-first", NULL);
8672           } else if (CUR_STREAM (stream)->field_order == 14) {
8673             gst_caps_set_simple (CUR_STREAM (stream)->caps, "field-order",
8674                 G_TYPE_STRING, "bottom-field-first", NULL);
8675           }
8676         }
8677       }
8678
8679       /* Create incomplete colorimetry here if needed */
8680       if (CUR_STREAM (stream)->colorimetry.range ||
8681           CUR_STREAM (stream)->colorimetry.matrix ||
8682           CUR_STREAM (stream)->colorimetry.transfer
8683           || CUR_STREAM (stream)->colorimetry.primaries) {
8684         gchar *colorimetry =
8685             gst_video_colorimetry_to_string (&CUR_STREAM (stream)->colorimetry);
8686         gst_caps_set_simple (CUR_STREAM (stream)->caps, "colorimetry",
8687             G_TYPE_STRING, colorimetry, NULL);
8688         g_free (colorimetry);
8689       }
8690
8691       if (stream->multiview_mode != GST_VIDEO_MULTIVIEW_MODE_NONE) {
8692         guint par_w = 1, par_h = 1;
8693
8694         if (CUR_STREAM (stream)->par_w > 0 && CUR_STREAM (stream)->par_h > 0) {
8695           par_w = CUR_STREAM (stream)->par_w;
8696           par_h = CUR_STREAM (stream)->par_h;
8697         }
8698
8699         if (gst_video_multiview_guess_half_aspect (stream->multiview_mode,
8700                 CUR_STREAM (stream)->width, CUR_STREAM (stream)->height, par_w,
8701                 par_h)) {
8702           stream->multiview_flags |= GST_VIDEO_MULTIVIEW_FLAGS_HALF_ASPECT;
8703         }
8704
8705         gst_caps_set_simple (CUR_STREAM (stream)->caps,
8706             "multiview-mode", G_TYPE_STRING,
8707             gst_video_multiview_mode_to_caps_string (stream->multiview_mode),
8708             "multiview-flags", GST_TYPE_VIDEO_MULTIVIEW_FLAGSET,
8709             stream->multiview_flags, GST_FLAG_SET_MASK_EXACT, NULL);
8710       }
8711     }
8712   }
8713
8714   else if (stream->subtype == FOURCC_soun) {
8715     if (CUR_STREAM (stream)->caps) {
8716       CUR_STREAM (stream)->caps =
8717           gst_caps_make_writable (CUR_STREAM (stream)->caps);
8718       if (CUR_STREAM (stream)->rate > 0)
8719         gst_caps_set_simple (CUR_STREAM (stream)->caps,
8720             "rate", G_TYPE_INT, (int) CUR_STREAM (stream)->rate, NULL);
8721       if (CUR_STREAM (stream)->n_channels > 0)
8722         gst_caps_set_simple (CUR_STREAM (stream)->caps,
8723             "channels", G_TYPE_INT, CUR_STREAM (stream)->n_channels, NULL);
8724       if (CUR_STREAM (stream)->n_channels > 2) {
8725         /* FIXME: Need to parse the 'chan' atom to get channel layouts
8726          * correctly; this is just the minimum we can do - assume
8727          * we don't actually have any channel positions. */
8728         gst_caps_set_simple (CUR_STREAM (stream)->caps,
8729             "channel-mask", GST_TYPE_BITMASK, G_GUINT64_CONSTANT (0), NULL);
8730       }
8731     }
8732   }
8733
8734   else if (stream->subtype == FOURCC_clcp && CUR_STREAM (stream)->caps) {
8735     const GstStructure *s;
8736     QtDemuxStream *fps_stream = NULL;
8737     gboolean fps_available = FALSE;
8738
8739     /* CEA608 closed caption tracks are a bit special in that each sample
8740      * can contain CCs for multiple frames, and CCs can be omitted and have to
8741      * be inferred from the duration of the sample then.
8742      *
8743      * As such we take the framerate from the (first) video track here for
8744      * CEA608 as there must be one CC byte pair for every video frame
8745      * according to the spec.
8746      *
8747      * For CEA708 all is fine and there is one sample per frame.
8748      */
8749
8750     s = gst_caps_get_structure (CUR_STREAM (stream)->caps, 0);
8751     if (gst_structure_has_name (s, "closedcaption/x-cea-608")) {
8752       gint i;
8753
8754       for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
8755         QtDemuxStream *tmp = QTDEMUX_NTH_STREAM (qtdemux, i);
8756
8757         if (tmp->subtype == FOURCC_vide) {
8758           fps_stream = tmp;
8759           break;
8760         }
8761       }
8762
8763       if (fps_stream) {
8764         fps_available = gst_qtdemux_guess_framerate (qtdemux, fps_stream);
8765         CUR_STREAM (stream)->fps_n = CUR_STREAM (fps_stream)->fps_n;
8766         CUR_STREAM (stream)->fps_d = CUR_STREAM (fps_stream)->fps_d;
8767       }
8768     } else {
8769       fps_available = gst_qtdemux_guess_framerate (qtdemux, stream);
8770       fps_stream = stream;
8771     }
8772
8773     CUR_STREAM (stream)->caps =
8774         gst_caps_make_writable (CUR_STREAM (stream)->caps);
8775
8776     /* set framerate if calculated framerate is reliable */
8777     if (fps_available) {
8778       gst_caps_set_simple (CUR_STREAM (stream)->caps,
8779           "framerate", GST_TYPE_FRACTION, CUR_STREAM (stream)->fps_n,
8780           CUR_STREAM (stream)->fps_d, NULL);
8781     }
8782   }
8783
8784   if (stream->pad) {
8785     GstCaps *prev_caps = NULL;
8786
8787     GST_PAD_ELEMENT_PRIVATE (stream->pad) = stream;
8788     gst_pad_set_event_function (stream->pad, gst_qtdemux_handle_src_event);
8789     gst_pad_set_query_function (stream->pad, gst_qtdemux_handle_src_query);
8790     gst_pad_set_active (stream->pad, TRUE);
8791
8792     gst_pad_use_fixed_caps (stream->pad);
8793
8794     if (stream->protected) {
8795       if (!gst_qtdemux_configure_protected_caps (qtdemux, stream)) {
8796         GST_ERROR_OBJECT (qtdemux,
8797             "Failed to configure protected stream caps.");
8798         return FALSE;
8799       }
8800     }
8801
8802     GST_DEBUG_OBJECT (qtdemux, "setting caps %" GST_PTR_FORMAT,
8803         CUR_STREAM (stream)->caps);
8804     if (stream->new_stream) {
8805       GstEvent *event;
8806       GstStreamFlags stream_flags = GST_STREAM_FLAG_NONE;
8807
8808       event =
8809           gst_pad_get_sticky_event (qtdemux->sinkpad, GST_EVENT_STREAM_START,
8810           0);
8811       if (event) {
8812         gst_event_parse_stream_flags (event, &stream_flags);
8813         if (gst_event_parse_group_id (event, &qtdemux->group_id))
8814           qtdemux->have_group_id = TRUE;
8815         else
8816           qtdemux->have_group_id = FALSE;
8817         gst_event_unref (event);
8818       } else if (!qtdemux->have_group_id) {
8819         qtdemux->have_group_id = TRUE;
8820         qtdemux->group_id = gst_util_group_id_next ();
8821       }
8822
8823       stream->new_stream = FALSE;
8824       event = gst_event_new_stream_start (stream->stream_id);
8825       if (qtdemux->have_group_id)
8826         gst_event_set_group_id (event, qtdemux->group_id);
8827       if (stream->disabled)
8828         stream_flags |= GST_STREAM_FLAG_UNSELECT;
8829       if (CUR_STREAM (stream)->sparse) {
8830         stream_flags |= GST_STREAM_FLAG_SPARSE;
8831       } else {
8832         stream_flags &= ~GST_STREAM_FLAG_SPARSE;
8833       }
8834       gst_event_set_stream_flags (event, stream_flags);
8835       gst_pad_push_event (stream->pad, event);
8836     }
8837
8838     prev_caps = gst_pad_get_current_caps (stream->pad);
8839
8840     if (CUR_STREAM (stream)->caps) {
8841       if (!prev_caps
8842           || !gst_caps_is_equal_fixed (prev_caps, CUR_STREAM (stream)->caps)) {
8843         GST_DEBUG_OBJECT (qtdemux, "setting caps %" GST_PTR_FORMAT,
8844             CUR_STREAM (stream)->caps);
8845         gst_pad_set_caps (stream->pad, CUR_STREAM (stream)->caps);
8846       } else {
8847         GST_DEBUG_OBJECT (qtdemux, "ignore duplicated caps");
8848       }
8849     } else {
8850       GST_WARNING_OBJECT (qtdemux, "stream without caps");
8851     }
8852
8853     if (prev_caps)
8854       gst_caps_unref (prev_caps);
8855     stream->new_caps = FALSE;
8856   }
8857   return TRUE;
8858 }
8859
8860 static void
8861 gst_qtdemux_stream_check_and_change_stsd_index (GstQTDemux * demux,
8862     QtDemuxStream * stream)
8863 {
8864   if (stream->cur_stsd_entry_index == stream->stsd_sample_description_id)
8865     return;
8866
8867   GST_DEBUG_OBJECT (stream->pad, "Changing stsd index from '%u' to '%u'",
8868       stream->cur_stsd_entry_index, stream->stsd_sample_description_id);
8869   if (G_UNLIKELY (stream->stsd_sample_description_id >=
8870           stream->stsd_entries_length)) {
8871     GST_ELEMENT_ERROR (demux, STREAM, DEMUX,
8872         (_("This file is invalid and cannot be played.")),
8873         ("New sample description id is out of bounds (%d >= %d)",
8874             stream->stsd_sample_description_id, stream->stsd_entries_length));
8875   } else {
8876     stream->cur_stsd_entry_index = stream->stsd_sample_description_id;
8877     stream->new_caps = TRUE;
8878   }
8879 }
8880
8881 static gboolean
8882 gst_qtdemux_add_stream (GstQTDemux * qtdemux,
8883     QtDemuxStream * stream, GstTagList * list)
8884 {
8885   gboolean ret = TRUE;
8886
8887   if (stream->subtype == FOURCC_vide) {
8888     gchar *name = g_strdup_printf ("video_%u", qtdemux->n_video_streams);
8889
8890     stream->pad =
8891         gst_pad_new_from_static_template (&gst_qtdemux_videosrc_template, name);
8892     g_free (name);
8893
8894     if (!gst_qtdemux_configure_stream (qtdemux, stream)) {
8895       gst_object_unref (stream->pad);
8896       stream->pad = NULL;
8897       ret = FALSE;
8898       goto done;
8899     }
8900
8901     qtdemux->n_video_streams++;
8902   } else if (stream->subtype == FOURCC_soun) {
8903     gchar *name = g_strdup_printf ("audio_%u", qtdemux->n_audio_streams);
8904
8905     stream->pad =
8906         gst_pad_new_from_static_template (&gst_qtdemux_audiosrc_template, name);
8907     g_free (name);
8908     if (!gst_qtdemux_configure_stream (qtdemux, stream)) {
8909       gst_object_unref (stream->pad);
8910       stream->pad = NULL;
8911       ret = FALSE;
8912       goto done;
8913     }
8914     qtdemux->n_audio_streams++;
8915   } else if (stream->subtype == FOURCC_strm) {
8916     GST_DEBUG_OBJECT (qtdemux, "stream type, not creating pad");
8917   } else if (stream->subtype == FOURCC_subp || stream->subtype == FOURCC_text
8918       || stream->subtype == FOURCC_sbtl || stream->subtype == FOURCC_subt
8919       || stream->subtype == FOURCC_clcp || stream->subtype == FOURCC_wvtt) {
8920     gchar *name = g_strdup_printf ("subtitle_%u", qtdemux->n_sub_streams);
8921
8922     stream->pad =
8923         gst_pad_new_from_static_template (&gst_qtdemux_subsrc_template, name);
8924     g_free (name);
8925     if (!gst_qtdemux_configure_stream (qtdemux, stream)) {
8926       gst_object_unref (stream->pad);
8927       stream->pad = NULL;
8928       ret = FALSE;
8929       goto done;
8930     }
8931     qtdemux->n_sub_streams++;
8932   } else if (CUR_STREAM (stream)->caps) {
8933     gchar *name = g_strdup_printf ("video_%u", qtdemux->n_video_streams);
8934
8935     stream->pad =
8936         gst_pad_new_from_static_template (&gst_qtdemux_videosrc_template, name);
8937     g_free (name);
8938     if (!gst_qtdemux_configure_stream (qtdemux, stream)) {
8939       gst_object_unref (stream->pad);
8940       stream->pad = NULL;
8941       ret = FALSE;
8942       goto done;
8943     }
8944     qtdemux->n_video_streams++;
8945   } else {
8946     GST_DEBUG_OBJECT (qtdemux, "unknown stream type");
8947     goto done;
8948   }
8949
8950   if (stream->pad) {
8951     GList *l;
8952
8953     GST_DEBUG_OBJECT (qtdemux, "adding pad %s %p to qtdemux %p",
8954         GST_OBJECT_NAME (stream->pad), stream->pad, qtdemux);
8955     gst_element_add_pad (GST_ELEMENT_CAST (qtdemux), stream->pad);
8956     GST_OBJECT_LOCK (qtdemux);
8957     gst_flow_combiner_add_pad (qtdemux->flowcombiner, stream->pad);
8958     GST_OBJECT_UNLOCK (qtdemux);
8959
8960     if (stream->stream_tags)
8961       gst_tag_list_unref (stream->stream_tags);
8962     stream->stream_tags = list;
8963     list = NULL;
8964     /* global tags go on each pad anyway */
8965     stream->send_global_tags = TRUE;
8966     /* send upstream GST_EVENT_PROTECTION events that were received before
8967        this source pad was created */
8968     for (l = qtdemux->protection_event_queue.head; l != NULL; l = l->next)
8969       gst_pad_push_event (stream->pad, gst_event_ref (l->data));
8970   }
8971 done:
8972   if (list)
8973     gst_tag_list_unref (list);
8974   return ret;
8975 }
8976
8977 /* find next atom with @fourcc starting at @offset */
8978 static GstFlowReturn
8979 qtdemux_find_atom (GstQTDemux * qtdemux, guint64 * offset,
8980     guint64 * length, guint32 fourcc)
8981 {
8982   GstFlowReturn ret;
8983   guint32 lfourcc;
8984   GstBuffer *buf;
8985
8986   GST_LOG_OBJECT (qtdemux, "finding fourcc %" GST_FOURCC_FORMAT " at offset %"
8987       G_GUINT64_FORMAT, GST_FOURCC_ARGS (fourcc), *offset);
8988
8989   while (TRUE) {
8990     GstMapInfo map;
8991
8992     buf = NULL;
8993     ret = gst_pad_pull_range (qtdemux->sinkpad, *offset, 16, &buf);
8994     if (G_UNLIKELY (ret != GST_FLOW_OK))
8995       goto locate_failed;
8996     if (G_UNLIKELY (gst_buffer_get_size (buf) != 16)) {
8997       /* likely EOF */
8998       ret = GST_FLOW_EOS;
8999       gst_buffer_unref (buf);
9000       goto locate_failed;
9001     }
9002     gst_buffer_map (buf, &map, GST_MAP_READ);
9003     extract_initial_length_and_fourcc (map.data, 16, length, &lfourcc);
9004     gst_buffer_unmap (buf, &map);
9005     gst_buffer_unref (buf);
9006
9007     if (G_UNLIKELY (*length == 0)) {
9008       GST_DEBUG_OBJECT (qtdemux, "invalid length 0");
9009       ret = GST_FLOW_ERROR;
9010       goto locate_failed;
9011     }
9012
9013     if (lfourcc == fourcc) {
9014       GST_DEBUG_OBJECT (qtdemux, "found '%" GST_FOURCC_FORMAT " at offset %"
9015           G_GUINT64_FORMAT, GST_FOURCC_ARGS (fourcc), *offset);
9016       break;
9017     } else {
9018       GST_LOG_OBJECT (qtdemux,
9019           "skipping atom '%" GST_FOURCC_FORMAT "' at %" G_GUINT64_FORMAT,
9020           GST_FOURCC_ARGS (lfourcc), *offset);
9021       if (*offset == G_MAXUINT64)
9022         goto locate_failed;
9023       *offset += *length;
9024     }
9025   }
9026
9027   return GST_FLOW_OK;
9028
9029 locate_failed:
9030   {
9031     /* might simply have had last one */
9032     GST_DEBUG_OBJECT (qtdemux, "fourcc not found");
9033     return ret;
9034   }
9035 }
9036
9037 /* should only do something in pull mode */
9038 /* call with OBJECT lock */
9039 static GstFlowReturn
9040 qtdemux_add_fragmented_samples (GstQTDemux * qtdemux)
9041 {
9042   guint64 length, offset;
9043   GstBuffer *buf = NULL;
9044   GstFlowReturn ret = GST_FLOW_OK;
9045   GstFlowReturn res = GST_FLOW_OK;
9046   GstMapInfo map;
9047
9048   offset = qtdemux->moof_offset;
9049   GST_DEBUG_OBJECT (qtdemux, "next moof at offset %" G_GUINT64_FORMAT, offset);
9050
9051   if (!offset) {
9052     GST_DEBUG_OBJECT (qtdemux, "no next moof");
9053     return GST_FLOW_EOS;
9054   }
9055
9056   /* best not do pull etc with lock held */
9057   GST_OBJECT_UNLOCK (qtdemux);
9058
9059   ret = qtdemux_find_atom (qtdemux, &offset, &length, FOURCC_moof);
9060   if (ret != GST_FLOW_OK)
9061     goto flow_failed;
9062
9063   ret = gst_qtdemux_pull_atom (qtdemux, offset, length, &buf);
9064   if (G_UNLIKELY (ret != GST_FLOW_OK))
9065     goto flow_failed;
9066   gst_buffer_map (buf, &map, GST_MAP_READ);
9067   if (!qtdemux_parse_moof (qtdemux, map.data, map.size, offset, NULL)) {
9068     gst_buffer_unmap (buf, &map);
9069     gst_buffer_unref (buf);
9070     buf = NULL;
9071     goto parse_failed;
9072   }
9073
9074   gst_buffer_unmap (buf, &map);
9075   gst_buffer_unref (buf);
9076   buf = NULL;
9077
9078   offset += length;
9079   /* look for next moof */
9080   ret = qtdemux_find_atom (qtdemux, &offset, &length, FOURCC_moof);
9081   if (G_UNLIKELY (ret != GST_FLOW_OK))
9082     goto flow_failed;
9083
9084 exit:
9085   GST_OBJECT_LOCK (qtdemux);
9086
9087   qtdemux->moof_offset = offset;
9088
9089   return res;
9090
9091 parse_failed:
9092   {
9093     GST_DEBUG_OBJECT (qtdemux, "failed to parse moof");
9094     offset = 0;
9095     res = GST_FLOW_ERROR;
9096     goto exit;
9097   }
9098 flow_failed:
9099   {
9100     /* maybe upstream temporarily flushing */
9101     if (ret != GST_FLOW_FLUSHING) {
9102       GST_DEBUG_OBJECT (qtdemux, "no next moof");
9103       offset = 0;
9104     } else {
9105       GST_DEBUG_OBJECT (qtdemux, "upstream WRONG_STATE");
9106       /* resume at current position next time */
9107     }
9108     res = ret;
9109     goto exit;
9110   }
9111 }
9112
9113 static void
9114 qtdemux_merge_sample_table (GstQTDemux * qtdemux, QtDemuxStream * stream)
9115 {
9116   guint i;
9117   guint32 num_chunks;
9118   gint32 stts_duration;
9119   GstByteWriter stsc, stts, stsz;
9120
9121   /* Each sample has a different size, which we don't support for merging */
9122   if (stream->sample_size == 0) {
9123     GST_DEBUG_OBJECT (qtdemux,
9124         "Not all samples have the same size, not merging");
9125     return;
9126   }
9127
9128   /* The stream has a ctts table, we don't support that */
9129   if (stream->ctts_present) {
9130     GST_DEBUG_OBJECT (qtdemux, "Have ctts, not merging");
9131     return;
9132   }
9133
9134   /* If there's a sync sample table also ignore this stream */
9135   if (stream->stps_present || stream->stss_present) {
9136     GST_DEBUG_OBJECT (qtdemux, "Have stss/stps, not merging");
9137     return;
9138   }
9139
9140   /* If chunks are considered samples already ignore this stream */
9141   if (stream->chunks_are_samples) {
9142     GST_DEBUG_OBJECT (qtdemux, "Chunks are samples, not merging");
9143     return;
9144   }
9145
9146   /* Require that all samples have the same duration */
9147   if (stream->n_sample_times > 1) {
9148     GST_DEBUG_OBJECT (qtdemux, "Not all samples have the same duration");
9149     return;
9150   }
9151
9152   /* Parse the stts to get the sample duration and number of samples */
9153   gst_byte_reader_skip_unchecked (&stream->stts, 4);
9154   stts_duration = gst_byte_reader_get_uint32_be_unchecked (&stream->stts);
9155
9156   /* Parse the number of chunks from the stco manually because the
9157    * reader is already behind that */
9158   num_chunks = GST_READ_UINT32_BE (stream->stco.data + 4);
9159
9160   GST_DEBUG_OBJECT (qtdemux, "sample_duration %d, num_chunks %u", stts_duration,
9161       num_chunks);
9162
9163   /* Now parse stsc, convert chunks into single samples and generate a
9164    * new stsc, stts and stsz from this information */
9165   gst_byte_writer_init (&stsc);
9166   gst_byte_writer_init (&stts);
9167   gst_byte_writer_init (&stsz);
9168
9169   /* Note: we skip fourccs, size, version, flags and other fields of the new
9170    * atoms as the byte readers with them are already behind that position
9171    * anyway and only update the values of those inside the stream directly.
9172    */
9173   stream->n_sample_times = 0;
9174   stream->n_samples = 0;
9175   for (i = 0; i < stream->n_samples_per_chunk; i++) {
9176     guint j;
9177     guint32 first_chunk, last_chunk, samples_per_chunk, sample_description_id;
9178
9179     first_chunk = gst_byte_reader_get_uint32_be_unchecked (&stream->stsc);
9180     samples_per_chunk = gst_byte_reader_get_uint32_be_unchecked (&stream->stsc);
9181     sample_description_id =
9182         gst_byte_reader_get_uint32_be_unchecked (&stream->stsc);
9183
9184     if (i == stream->n_samples_per_chunk - 1) {
9185       /* +1 because first_chunk is 1-based */
9186       last_chunk = num_chunks + 1;
9187     } else {
9188       last_chunk = gst_byte_reader_peek_uint32_be_unchecked (&stream->stsc);
9189     }
9190
9191     GST_DEBUG_OBJECT (qtdemux,
9192         "Merging first_chunk: %u, last_chunk: %u, samples_per_chunk: %u, sample_description_id: %u",
9193         first_chunk, last_chunk, samples_per_chunk, sample_description_id);
9194
9195     gst_byte_writer_put_uint32_be (&stsc, first_chunk);
9196     /* One sample in this chunk */
9197     gst_byte_writer_put_uint32_be (&stsc, 1);
9198     gst_byte_writer_put_uint32_be (&stsc, sample_description_id);
9199
9200     /* For each chunk write a stts and stsz entry now */
9201     gst_byte_writer_put_uint32_be (&stts, last_chunk - first_chunk);
9202     gst_byte_writer_put_uint32_be (&stts, stts_duration * samples_per_chunk);
9203     for (j = first_chunk; j < last_chunk; j++) {
9204       gst_byte_writer_put_uint32_be (&stsz,
9205           stream->sample_size * samples_per_chunk);
9206     }
9207
9208     stream->n_sample_times += 1;
9209     stream->n_samples += last_chunk - first_chunk;
9210   }
9211
9212   g_assert_cmpint (stream->n_samples, ==, num_chunks);
9213
9214   GST_DEBUG_OBJECT (qtdemux, "Have %u samples and %u sample times",
9215       stream->n_samples, stream->n_sample_times);
9216
9217   /* We don't have a fixed sample size anymore */
9218   stream->sample_size = 0;
9219
9220   /* Free old data for the atoms */
9221   g_free ((gpointer) stream->stsz.data);
9222   stream->stsz.data = NULL;
9223   g_free ((gpointer) stream->stsc.data);
9224   stream->stsc.data = NULL;
9225   g_free ((gpointer) stream->stts.data);
9226   stream->stts.data = NULL;
9227
9228   /* Store new data and replace byte readers */
9229   stream->stsz.size = gst_byte_writer_get_size (&stsz);
9230   stream->stsz.data = gst_byte_writer_reset_and_get_data (&stsz);
9231   gst_byte_reader_init (&stream->stsz, stream->stsz.data, stream->stsz.size);
9232   stream->stts.size = gst_byte_writer_get_size (&stts);
9233   stream->stts.data = gst_byte_writer_reset_and_get_data (&stts);
9234   gst_byte_reader_init (&stream->stts, stream->stts.data, stream->stts.size);
9235   stream->stsc.size = gst_byte_writer_get_size (&stsc);
9236   stream->stsc.data = gst_byte_writer_reset_and_get_data (&stsc);
9237   gst_byte_reader_init (&stream->stsc, stream->stsc.data, stream->stsc.size);
9238 }
9239
9240 /* initialise bytereaders for stbl sub-atoms */
9241 static gboolean
9242 qtdemux_stbl_init (GstQTDemux * qtdemux, QtDemuxStream * stream, GNode * stbl)
9243 {
9244   stream->stbl_index = -1;      /* no samples have yet been parsed */
9245   stream->sample_index = -1;
9246
9247   /* time-to-sample atom */
9248   if (!qtdemux_tree_get_child_by_type_full (stbl, FOURCC_stts, &stream->stts))
9249     goto corrupt_file;
9250
9251   /* copy atom data into a new buffer for later use */
9252   stream->stts.data = g_memdup2 (stream->stts.data, stream->stts.size);
9253
9254   /* skip version + flags */
9255   if (!gst_byte_reader_skip (&stream->stts, 1 + 3) ||
9256       !gst_byte_reader_get_uint32_be (&stream->stts, &stream->n_sample_times))
9257     goto corrupt_file;
9258   GST_LOG_OBJECT (qtdemux, "%u timestamp blocks", stream->n_sample_times);
9259
9260   /* make sure there's enough data */
9261   if (!qt_atom_parser_has_chunks (&stream->stts, stream->n_sample_times, 8)) {
9262     stream->n_sample_times = gst_byte_reader_get_remaining (&stream->stts) / 8;
9263     GST_LOG_OBJECT (qtdemux, "overriding to %u timestamp blocks",
9264         stream->n_sample_times);
9265     if (!stream->n_sample_times)
9266       goto corrupt_file;
9267   }
9268
9269   /* sync sample atom */
9270   stream->stps_present = FALSE;
9271   if ((stream->stss_present =
9272           ! !qtdemux_tree_get_child_by_type_full (stbl, FOURCC_stss,
9273               &stream->stss) ? TRUE : FALSE) == TRUE) {
9274     /* copy atom data into a new buffer for later use */
9275     stream->stss.data = g_memdup2 (stream->stss.data, stream->stss.size);
9276
9277     /* skip version + flags */
9278     if (!gst_byte_reader_skip (&stream->stss, 1 + 3) ||
9279         !gst_byte_reader_get_uint32_be (&stream->stss, &stream->n_sample_syncs))
9280       goto corrupt_file;
9281
9282     if (stream->n_sample_syncs) {
9283       /* make sure there's enough data */
9284       if (!qt_atom_parser_has_chunks (&stream->stss, stream->n_sample_syncs, 4))
9285         goto corrupt_file;
9286     }
9287
9288     /* partial sync sample atom */
9289     if ((stream->stps_present =
9290             ! !qtdemux_tree_get_child_by_type_full (stbl, FOURCC_stps,
9291                 &stream->stps) ? TRUE : FALSE) == TRUE) {
9292       /* copy atom data into a new buffer for later use */
9293       stream->stps.data = g_memdup2 (stream->stps.data, stream->stps.size);
9294
9295       /* skip version + flags */
9296       if (!gst_byte_reader_skip (&stream->stps, 1 + 3) ||
9297           !gst_byte_reader_get_uint32_be (&stream->stps,
9298               &stream->n_sample_partial_syncs))
9299         goto corrupt_file;
9300
9301       /* if there are no entries, the stss table contains the real
9302        * sync samples */
9303       if (stream->n_sample_partial_syncs) {
9304         /* make sure there's enough data */
9305         if (!qt_atom_parser_has_chunks (&stream->stps,
9306                 stream->n_sample_partial_syncs, 4))
9307           goto corrupt_file;
9308       }
9309     }
9310   }
9311
9312   /* sample size */
9313   if (!qtdemux_tree_get_child_by_type_full (stbl, FOURCC_stsz, &stream->stsz))
9314     goto no_samples;
9315
9316   /* copy atom data into a new buffer for later use */
9317   stream->stsz.data = g_memdup2 (stream->stsz.data, stream->stsz.size);
9318
9319   /* skip version + flags */
9320   if (!gst_byte_reader_skip (&stream->stsz, 1 + 3) ||
9321       !gst_byte_reader_get_uint32_be (&stream->stsz, &stream->sample_size))
9322     goto corrupt_file;
9323
9324   if (!gst_byte_reader_get_uint32_be (&stream->stsz, &stream->n_samples))
9325     goto corrupt_file;
9326
9327   if (!stream->n_samples)
9328     goto no_samples;
9329
9330   /* sample-to-chunk atom */
9331   if (!qtdemux_tree_get_child_by_type_full (stbl, FOURCC_stsc, &stream->stsc))
9332     goto corrupt_file;
9333
9334   /* copy atom data into a new buffer for later use */
9335   stream->stsc.data = g_memdup2 (stream->stsc.data, stream->stsc.size);
9336
9337   /* skip version + flags */
9338   if (!gst_byte_reader_skip (&stream->stsc, 1 + 3) ||
9339       !gst_byte_reader_get_uint32_be (&stream->stsc,
9340           &stream->n_samples_per_chunk))
9341     goto corrupt_file;
9342
9343   GST_DEBUG_OBJECT (qtdemux, "n_samples_per_chunk %u",
9344       stream->n_samples_per_chunk);
9345
9346   /* make sure there's enough data */
9347   if (!qt_atom_parser_has_chunks (&stream->stsc, stream->n_samples_per_chunk,
9348           12))
9349     goto corrupt_file;
9350
9351
9352   /* chunk offset */
9353   if (qtdemux_tree_get_child_by_type_full (stbl, FOURCC_stco, &stream->stco))
9354     stream->co_size = sizeof (guint32);
9355   else if (qtdemux_tree_get_child_by_type_full (stbl, FOURCC_co64,
9356           &stream->stco))
9357     stream->co_size = sizeof (guint64);
9358   else
9359     goto corrupt_file;
9360
9361   /* copy atom data into a new buffer for later use */
9362   stream->stco.data = g_memdup2 (stream->stco.data, stream->stco.size);
9363
9364   /* skip version + flags */
9365   if (!gst_byte_reader_skip (&stream->stco, 1 + 3))
9366     goto corrupt_file;
9367
9368   /* chunks_are_samples == TRUE means treat chunks as samples */
9369   stream->chunks_are_samples = stream->sample_size
9370       && !CUR_STREAM (stream)->sampled;
9371   if (stream->chunks_are_samples) {
9372     /* treat chunks as samples */
9373     if (!gst_byte_reader_get_uint32_be (&stream->stco, &stream->n_samples))
9374       goto corrupt_file;
9375   } else {
9376     /* skip number of entries */
9377     if (!gst_byte_reader_skip (&stream->stco, 4))
9378       goto corrupt_file;
9379
9380     /* make sure there are enough data in the stsz atom */
9381     if (!stream->sample_size) {
9382       /* different sizes for each sample */
9383       if (!qt_atom_parser_has_chunks (&stream->stsz, stream->n_samples, 4))
9384         goto corrupt_file;
9385     }
9386   }
9387
9388   /* composition time-to-sample */
9389   if ((stream->ctts_present =
9390           ! !qtdemux_tree_get_child_by_type_full (stbl, FOURCC_ctts,
9391               &stream->ctts) ? TRUE : FALSE) == TRUE) {
9392     GstByteReader cslg = GST_BYTE_READER_INIT (NULL, 0);
9393     guint8 ctts_version;
9394
9395     /* copy atom data into a new buffer for later use */
9396     stream->ctts.data = g_memdup2 (stream->ctts.data, stream->ctts.size);
9397
9398     /* version 1 has signed offsets */
9399     if (!gst_byte_reader_get_uint8 (&stream->ctts, &ctts_version))
9400       goto corrupt_file;
9401
9402     /* flags */
9403     if (!gst_byte_reader_skip (&stream->ctts, 3)
9404         || !gst_byte_reader_get_uint32_be (&stream->ctts,
9405             &stream->n_composition_times))
9406       goto corrupt_file;
9407
9408     /* make sure there's enough data */
9409     if (!qt_atom_parser_has_chunks (&stream->ctts, stream->n_composition_times,
9410             4 + 4))
9411       goto corrupt_file;
9412
9413     /* This is optional, if missing we iterate the ctts */
9414     if (qtdemux_tree_get_child_by_type_full (stbl, FOURCC_cslg, &cslg)) {
9415       guint8 cslg_version;
9416
9417       /* cslg version 1 has 64 bit fields */
9418       if (!gst_byte_reader_get_uint8 (&cslg, &cslg_version))
9419         goto corrupt_file;
9420
9421       /* skip flags */
9422       if (!gst_byte_reader_skip (&cslg, 3))
9423         goto corrupt_file;
9424
9425       if (cslg_version == 0) {
9426         gint32 composition_to_dts_shift;
9427
9428         if (!gst_byte_reader_get_int32_be (&cslg, &composition_to_dts_shift))
9429           goto corrupt_file;
9430
9431         stream->cslg_shift = MAX (0, composition_to_dts_shift);
9432       } else {
9433         gint64 composition_to_dts_shift;
9434
9435         if (!gst_byte_reader_get_int64_be (&cslg, &composition_to_dts_shift))
9436           goto corrupt_file;
9437
9438         stream->cslg_shift = MAX (0, composition_to_dts_shift);
9439       }
9440     } else {
9441       gint32 cslg_least = 0;
9442       guint num_entries, pos;
9443       gint i;
9444
9445       pos = gst_byte_reader_get_pos (&stream->ctts);
9446       num_entries = stream->n_composition_times;
9447
9448       stream->cslg_shift = 0;
9449
9450       for (i = 0; i < num_entries; i++) {
9451         gint32 offset;
9452
9453         gst_byte_reader_skip_unchecked (&stream->ctts, 4);
9454         offset = gst_byte_reader_get_int32_be_unchecked (&stream->ctts);
9455         /* HACK: if sample_offset is larger than 2 * duration, ignore the box.
9456          * slightly inaccurate PTS could be more usable than corrupted one */
9457         if (G_UNLIKELY ((ABS (offset) / 2) > stream->duration)) {
9458           GST_WARNING_OBJECT (qtdemux,
9459               "Ignore corrupted ctts, sample_offset %" G_GINT32_FORMAT
9460               " larger than duration %" G_GUINT64_FORMAT,
9461               offset, stream->duration);
9462
9463           stream->cslg_shift = 0;
9464           stream->ctts_present = FALSE;
9465           goto done;
9466         }
9467
9468         if (offset < cslg_least)
9469           cslg_least = offset;
9470       }
9471
9472       if (cslg_least < 0)
9473         stream->cslg_shift = -cslg_least;
9474       else
9475         stream->cslg_shift = 0;
9476
9477       /* reset the reader so we can generate sample table */
9478       gst_byte_reader_set_pos (&stream->ctts, pos);
9479     }
9480   } else {
9481     /* Ensure the cslg_shift value is consistent so we can use it
9482      * unconditionally to produce TS and Segment */
9483     stream->cslg_shift = 0;
9484   }
9485
9486   /* For raw audio streams especially we might want to merge the samples
9487    * to not output one audio sample per buffer. We're doing this here
9488    * before allocating the sample tables so that from this point onwards
9489    * the number of container samples are static */
9490   if (stream->min_buffer_size > 0) {
9491     qtdemux_merge_sample_table (qtdemux, stream);
9492   }
9493
9494 done:
9495   GST_DEBUG_OBJECT (qtdemux, "allocating n_samples %u * %u (%.2f MB)",
9496       stream->n_samples, (guint) sizeof (QtDemuxSample),
9497       stream->n_samples * sizeof (QtDemuxSample) / (1024.0 * 1024.0));
9498
9499   if (stream->n_samples >=
9500       QTDEMUX_MAX_SAMPLE_INDEX_SIZE / sizeof (QtDemuxSample)) {
9501     GST_WARNING_OBJECT (qtdemux, "not allocating index of %d samples, would "
9502         "be larger than %uMB (broken file?)", stream->n_samples,
9503         QTDEMUX_MAX_SAMPLE_INDEX_SIZE >> 20);
9504     return FALSE;
9505   }
9506
9507   g_assert (stream->samples == NULL);
9508   stream->samples = g_try_new0 (QtDemuxSample, stream->n_samples);
9509   if (!stream->samples) {
9510     GST_WARNING_OBJECT (qtdemux, "failed to allocate %d samples",
9511         stream->n_samples);
9512     return FALSE;
9513   }
9514
9515   return TRUE;
9516
9517 corrupt_file:
9518   {
9519     GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
9520         (_("This file is corrupt and cannot be played.")), (NULL));
9521     return FALSE;
9522   }
9523 no_samples:
9524   {
9525     gst_qtdemux_stbl_free (stream);
9526     if (!qtdemux->fragmented) {
9527       /* not quite good */
9528       GST_WARNING_OBJECT (qtdemux, "stream has no samples");
9529       return FALSE;
9530     } else {
9531       /* may pick up samples elsewhere */
9532       return TRUE;
9533     }
9534   }
9535 }
9536
9537 /* collect samples from the next sample to be parsed up to sample @n for @stream
9538  * by reading the info from @stbl
9539  *
9540  * This code can be executed from both the streaming thread and the seeking
9541  * thread so it takes the object lock to protect itself
9542  */
9543 static gboolean
9544 qtdemux_parse_samples (GstQTDemux * qtdemux, QtDemuxStream * stream, guint32 n)
9545 {
9546   gint i, j, k;
9547   QtDemuxSample *samples, *first, *cur, *last;
9548   guint32 n_samples_per_chunk;
9549   guint32 n_samples;
9550
9551   GST_LOG_OBJECT (qtdemux, "parsing samples for stream fourcc %"
9552       GST_FOURCC_FORMAT ", pad %s",
9553       GST_FOURCC_ARGS (CUR_STREAM (stream)->fourcc),
9554       stream->pad ? GST_PAD_NAME (stream->pad) : "(NULL)");
9555
9556   n_samples = stream->n_samples;
9557
9558   if (n >= n_samples)
9559     goto out_of_samples;
9560
9561   GST_OBJECT_LOCK (qtdemux);
9562   if (n <= stream->stbl_index)
9563     goto already_parsed;
9564
9565   GST_DEBUG_OBJECT (qtdemux, "parsing up to sample %u", n);
9566
9567   if (!stream->stsz.data) {
9568     /* so we already parsed and passed all the moov samples;
9569      * onto fragmented ones */
9570     g_assert (qtdemux->fragmented);
9571     goto done;
9572   }
9573
9574   /* pointer to the sample table */
9575   samples = stream->samples;
9576
9577   /* starts from -1, moves to the next sample index to parse */
9578   stream->stbl_index++;
9579
9580   /* keep track of the first and last sample to fill */
9581   first = &samples[stream->stbl_index];
9582   last = &samples[n];
9583
9584   if (!stream->chunks_are_samples) {
9585     /* set the sample sizes */
9586     if (stream->sample_size == 0) {
9587       /* different sizes for each sample */
9588       for (cur = first; cur <= last; cur++) {
9589         cur->size = gst_byte_reader_get_uint32_be_unchecked (&stream->stsz);
9590         GST_LOG_OBJECT (qtdemux, "sample %d has size %u",
9591             (guint) (cur - samples), cur->size);
9592       }
9593     } else {
9594       /* samples have the same size */
9595       GST_LOG_OBJECT (qtdemux, "all samples have size %u", stream->sample_size);
9596       for (cur = first; cur <= last; cur++)
9597         cur->size = stream->sample_size;
9598     }
9599   }
9600
9601   n_samples_per_chunk = stream->n_samples_per_chunk;
9602   cur = first;
9603
9604   for (i = stream->stsc_index; i < n_samples_per_chunk; i++) {
9605     guint32 last_chunk;
9606
9607     if (stream->stsc_chunk_index >= stream->last_chunk
9608         || stream->stsc_chunk_index < stream->first_chunk) {
9609       stream->first_chunk =
9610           gst_byte_reader_get_uint32_be_unchecked (&stream->stsc);
9611       stream->samples_per_chunk =
9612           gst_byte_reader_get_uint32_be_unchecked (&stream->stsc);
9613       /* starts from 1 */
9614       stream->stsd_sample_description_id =
9615           gst_byte_reader_get_uint32_be_unchecked (&stream->stsc) - 1;
9616
9617       /* chunk numbers are counted from 1 it seems */
9618       if (G_UNLIKELY (stream->first_chunk == 0))
9619         goto corrupt_file;
9620
9621       --stream->first_chunk;
9622
9623       /* the last chunk of each entry is calculated by taking the first chunk
9624        * of the next entry; except if there is no next, where we fake it with
9625        * INT_MAX */
9626       if (G_UNLIKELY (i == (stream->n_samples_per_chunk - 1))) {
9627         stream->last_chunk = G_MAXUINT32;
9628       } else {
9629         stream->last_chunk =
9630             gst_byte_reader_peek_uint32_be_unchecked (&stream->stsc);
9631         if (G_UNLIKELY (stream->last_chunk == 0))
9632           goto corrupt_file;
9633
9634         --stream->last_chunk;
9635       }
9636
9637       GST_LOG_OBJECT (qtdemux,
9638           "entry %d has first_chunk %d, last_chunk %d, samples_per_chunk %d"
9639           "sample desc ID: %d", i, stream->first_chunk, stream->last_chunk,
9640           stream->samples_per_chunk, stream->stsd_sample_description_id);
9641
9642       if (G_UNLIKELY (stream->last_chunk < stream->first_chunk))
9643         goto corrupt_file;
9644
9645       if (stream->last_chunk != G_MAXUINT32) {
9646         if (!qt_atom_parser_peek_sub (&stream->stco,
9647                 stream->first_chunk * stream->co_size,
9648                 (stream->last_chunk - stream->first_chunk) * stream->co_size,
9649                 &stream->co_chunk))
9650           goto corrupt_file;
9651
9652       } else {
9653         stream->co_chunk = stream->stco;
9654         if (!gst_byte_reader_skip (&stream->co_chunk,
9655                 stream->first_chunk * stream->co_size))
9656           goto corrupt_file;
9657       }
9658
9659       stream->stsc_chunk_index = stream->first_chunk;
9660     }
9661
9662     last_chunk = stream->last_chunk;
9663
9664     if (stream->chunks_are_samples) {
9665       cur = &samples[stream->stsc_chunk_index];
9666
9667       for (j = stream->stsc_chunk_index; j < last_chunk; j++) {
9668         if (j > n) {
9669           /* save state */
9670           stream->stsc_chunk_index = j;
9671           goto done;
9672         }
9673
9674         cur->offset =
9675             qt_atom_parser_get_offset_unchecked (&stream->co_chunk,
9676             stream->co_size);
9677
9678         GST_LOG_OBJECT (qtdemux, "Created entry %d with offset "
9679             "%" G_GUINT64_FORMAT, j, cur->offset);
9680
9681         if (CUR_STREAM (stream)->samples_per_frame > 0 &&
9682             CUR_STREAM (stream)->bytes_per_frame > 0) {
9683           cur->size =
9684               (stream->samples_per_chunk * CUR_STREAM (stream)->n_channels) /
9685               CUR_STREAM (stream)->samples_per_frame *
9686               CUR_STREAM (stream)->bytes_per_frame;
9687         } else {
9688           cur->size = stream->samples_per_chunk;
9689         }
9690
9691         GST_DEBUG_OBJECT (qtdemux,
9692             "keyframe sample %d: timestamp %" GST_TIME_FORMAT ", size %u",
9693             j, GST_TIME_ARGS (QTSTREAMTIME_TO_GSTTIME (stream,
9694                     stream->stco_sample_index)), cur->size);
9695
9696         cur->timestamp = stream->stco_sample_index;
9697         cur->duration = stream->samples_per_chunk;
9698         cur->keyframe = TRUE;
9699         cur++;
9700
9701         stream->stco_sample_index += stream->samples_per_chunk;
9702       }
9703       stream->stsc_chunk_index = j;
9704     } else {
9705       for (j = stream->stsc_chunk_index; j < last_chunk; j++) {
9706         guint32 samples_per_chunk;
9707         guint64 chunk_offset;
9708
9709         if (!stream->stsc_sample_index
9710             && !qt_atom_parser_get_offset (&stream->co_chunk, stream->co_size,
9711                 &stream->chunk_offset))
9712           goto corrupt_file;
9713
9714         samples_per_chunk = stream->samples_per_chunk;
9715         chunk_offset = stream->chunk_offset;
9716
9717         for (k = stream->stsc_sample_index; k < samples_per_chunk; k++) {
9718           GST_LOG_OBJECT (qtdemux, "creating entry %d with offset %"
9719               G_GUINT64_FORMAT " and size %d",
9720               (guint) (cur - samples), chunk_offset, cur->size);
9721
9722           cur->offset = chunk_offset;
9723           chunk_offset += cur->size;
9724           cur++;
9725
9726           if (G_UNLIKELY (cur > last)) {
9727             /* save state */
9728             stream->stsc_sample_index = k + 1;
9729             stream->chunk_offset = chunk_offset;
9730             stream->stsc_chunk_index = j;
9731             goto done2;
9732           }
9733         }
9734         stream->stsc_sample_index = 0;
9735       }
9736       stream->stsc_chunk_index = j;
9737     }
9738     stream->stsc_index++;
9739   }
9740
9741   if (stream->chunks_are_samples)
9742     goto ctts;
9743 done2:
9744   {
9745     guint32 n_sample_times;
9746
9747     n_sample_times = stream->n_sample_times;
9748     cur = first;
9749
9750     for (i = stream->stts_index; i < n_sample_times; i++) {
9751       guint32 stts_samples;
9752       gint32 stts_duration;
9753       gint64 stts_time;
9754
9755       if (stream->stts_sample_index >= stream->stts_samples
9756           || !stream->stts_sample_index) {
9757
9758         stream->stts_samples =
9759             gst_byte_reader_get_uint32_be_unchecked (&stream->stts);
9760         stream->stts_duration =
9761             gst_byte_reader_get_uint32_be_unchecked (&stream->stts);
9762
9763         GST_LOG_OBJECT (qtdemux, "block %d, %u timestamps, duration %u",
9764             i, stream->stts_samples, stream->stts_duration);
9765
9766         stream->stts_sample_index = 0;
9767       }
9768
9769       stts_samples = stream->stts_samples;
9770       stts_duration = stream->stts_duration;
9771       stts_time = stream->stts_time;
9772
9773       for (j = stream->stts_sample_index; j < stts_samples; j++) {
9774         GST_DEBUG_OBJECT (qtdemux,
9775             "sample %d: index %d, timestamp %" GST_TIME_FORMAT,
9776             (guint) (cur - samples), j,
9777             GST_TIME_ARGS (QTSTREAMTIME_TO_GSTTIME (stream, stts_time)));
9778
9779         cur->timestamp = stts_time;
9780         cur->duration = stts_duration;
9781
9782         /* avoid 32-bit wrap-around,
9783          * but still mind possible 'negative' duration */
9784         stts_time += (gint64) stts_duration;
9785         cur++;
9786
9787         if (G_UNLIKELY (cur > last)) {
9788           /* save values */
9789           stream->stts_time = stts_time;
9790           stream->stts_sample_index = j + 1;
9791           if (stream->stts_sample_index >= stream->stts_samples)
9792             stream->stts_index++;
9793           goto done3;
9794         }
9795       }
9796       stream->stts_sample_index = 0;
9797       stream->stts_time = stts_time;
9798       stream->stts_index++;
9799     }
9800     /* fill up empty timestamps with the last timestamp, this can happen when
9801      * the last samples do not decode and so we don't have timestamps for them.
9802      * We however look at the last timestamp to estimate the track length so we
9803      * need something in here. */
9804     for (; cur < last; cur++) {
9805       GST_DEBUG_OBJECT (qtdemux,
9806           "fill sample %d: timestamp %" GST_TIME_FORMAT,
9807           (guint) (cur - samples),
9808           GST_TIME_ARGS (QTSTREAMTIME_TO_GSTTIME (stream, stream->stts_time)));
9809       cur->timestamp = stream->stts_time;
9810       cur->duration = -1;
9811     }
9812   }
9813 done3:
9814   {
9815     /* sample sync, can be NULL */
9816     if (stream->stss_present == TRUE) {
9817       guint32 n_sample_syncs;
9818
9819       n_sample_syncs = stream->n_sample_syncs;
9820
9821       if (!n_sample_syncs) {
9822         GST_DEBUG_OBJECT (qtdemux, "all samples are keyframes");
9823         stream->all_keyframe = TRUE;
9824       } else {
9825         for (i = stream->stss_index; i < n_sample_syncs; i++) {
9826           /* note that the first sample is index 1, not 0 */
9827           guint32 index;
9828
9829           index = gst_byte_reader_get_uint32_be_unchecked (&stream->stss);
9830
9831           if (G_LIKELY (index > 0 && index <= n_samples)) {
9832             index -= 1;
9833             samples[index].keyframe = TRUE;
9834             GST_DEBUG_OBJECT (qtdemux, "samples at %u is keyframe", index);
9835             /* and exit if we have enough samples */
9836             if (G_UNLIKELY (index >= n)) {
9837               i++;
9838               break;
9839             }
9840           }
9841         }
9842         /* save state */
9843         stream->stss_index = i;
9844       }
9845
9846       /* stps marks partial sync frames like open GOP I-Frames */
9847       if (stream->stps_present == TRUE) {
9848         guint32 n_sample_partial_syncs;
9849
9850         n_sample_partial_syncs = stream->n_sample_partial_syncs;
9851
9852         /* if there are no entries, the stss table contains the real
9853          * sync samples */
9854         if (n_sample_partial_syncs) {
9855           for (i = stream->stps_index; i < n_sample_partial_syncs; i++) {
9856             /* note that the first sample is index 1, not 0 */
9857             guint32 index;
9858
9859             index = gst_byte_reader_get_uint32_be_unchecked (&stream->stps);
9860
9861             if (G_LIKELY (index > 0 && index <= n_samples)) {
9862               index -= 1;
9863               samples[index].keyframe = TRUE;
9864               GST_DEBUG_OBJECT (qtdemux, "samples at %u is keyframe", index);
9865               /* and exit if we have enough samples */
9866               if (G_UNLIKELY (index >= n)) {
9867                 i++;
9868                 break;
9869               }
9870             }
9871           }
9872           /* save state */
9873           stream->stps_index = i;
9874         }
9875       }
9876     } else {
9877       /* no stss, all samples are keyframes */
9878       stream->all_keyframe = TRUE;
9879       GST_DEBUG_OBJECT (qtdemux, "setting all keyframes");
9880     }
9881   }
9882
9883 ctts:
9884   /* composition time to sample */
9885   if (stream->ctts_present == TRUE) {
9886     guint32 n_composition_times;
9887     guint32 ctts_count;
9888     gint32 ctts_soffset;
9889
9890     /* Fill in the pts_offsets */
9891     cur = first;
9892     n_composition_times = stream->n_composition_times;
9893
9894     for (i = stream->ctts_index; i < n_composition_times; i++) {
9895       if (stream->ctts_sample_index >= stream->ctts_count
9896           || !stream->ctts_sample_index) {
9897         stream->ctts_count =
9898             gst_byte_reader_get_uint32_be_unchecked (&stream->ctts);
9899         stream->ctts_soffset =
9900             gst_byte_reader_get_int32_be_unchecked (&stream->ctts);
9901         stream->ctts_sample_index = 0;
9902       }
9903
9904       ctts_count = stream->ctts_count;
9905       ctts_soffset = stream->ctts_soffset;
9906
9907       for (j = stream->ctts_sample_index; j < ctts_count; j++) {
9908         cur->pts_offset = ctts_soffset;
9909         cur++;
9910
9911         if (G_UNLIKELY (cur > last)) {
9912           /* save state */
9913           stream->ctts_sample_index = j + 1;
9914           goto done;
9915         }
9916       }
9917       stream->ctts_sample_index = 0;
9918       stream->ctts_index++;
9919     }
9920   }
9921 done:
9922   stream->stbl_index = n;
9923   /* if index has been completely parsed, free data that is no-longer needed */
9924   if (n + 1 == stream->n_samples) {
9925     gst_qtdemux_stbl_free (stream);
9926     GST_DEBUG_OBJECT (qtdemux, "parsed all available samples;");
9927     if (qtdemux->pullbased) {
9928       GST_DEBUG_OBJECT (qtdemux, "checking for more samples");
9929       while (n + 1 == stream->n_samples)
9930         if (qtdemux_add_fragmented_samples (qtdemux) != GST_FLOW_OK)
9931           break;
9932     }
9933   }
9934   GST_OBJECT_UNLOCK (qtdemux);
9935
9936   return TRUE;
9937
9938   /* SUCCESS */
9939 already_parsed:
9940   {
9941     GST_LOG_OBJECT (qtdemux,
9942         "Tried to parse up to sample %u but this sample has already been parsed",
9943         n);
9944     /* if fragmented, there may be more */
9945     if (qtdemux->fragmented && n == stream->stbl_index)
9946       goto done;
9947     GST_OBJECT_UNLOCK (qtdemux);
9948     return TRUE;
9949   }
9950   /* ERRORS */
9951 out_of_samples:
9952   {
9953     GST_LOG_OBJECT (qtdemux,
9954         "Tried to parse up to sample %u but there are only %u samples", n + 1,
9955         stream->n_samples);
9956     GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
9957         (_("This file is corrupt and cannot be played.")), (NULL));
9958     return FALSE;
9959   }
9960 corrupt_file:
9961   {
9962     GST_OBJECT_UNLOCK (qtdemux);
9963     GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
9964         (_("This file is corrupt and cannot be played.")), (NULL));
9965     return FALSE;
9966   }
9967 }
9968
9969 /* collect all segment info for @stream.
9970  */
9971 static gboolean
9972 qtdemux_parse_segments (GstQTDemux * qtdemux, QtDemuxStream * stream,
9973     GNode * trak)
9974 {
9975   GNode *edts;
9976   /* accept edts if they contain gaps at start and there is only
9977    * one media segment */
9978   gboolean allow_pushbased_edts = TRUE;
9979   gint media_segments_count = 0;
9980
9981   /* parse and prepare segment info from the edit list */
9982   GST_DEBUG_OBJECT (qtdemux, "looking for edit list container");
9983   stream->n_segments = 0;
9984   stream->segments = NULL;
9985   if ((edts = qtdemux_tree_get_child_by_type (trak, FOURCC_edts))) {
9986     GNode *elst;
9987     gint n_segments;
9988     gint segment_number, entry_size;
9989     guint64 time;
9990     GstClockTime stime;
9991     const guint8 *buffer;
9992     guint8 version;
9993     guint32 size;
9994
9995     GST_DEBUG_OBJECT (qtdemux, "looking for edit list");
9996     if (!(elst = qtdemux_tree_get_child_by_type (edts, FOURCC_elst)))
9997       goto done;
9998
9999     buffer = elst->data;
10000
10001     size = QT_UINT32 (buffer);
10002     /* version, flags, n_segments */
10003     if (size < 16) {
10004       GST_WARNING_OBJECT (qtdemux, "Invalid edit list");
10005       goto done;
10006     }
10007     version = QT_UINT8 (buffer + 8);
10008     entry_size = (version == 1) ? 20 : 12;
10009
10010     n_segments = QT_UINT32 (buffer + 12);
10011
10012     if (n_segments > 100000 || size < 16 + n_segments * entry_size) {
10013       GST_WARNING_OBJECT (qtdemux, "Invalid edit list");
10014       goto done;
10015     }
10016
10017     /* we might allocate a bit too much, at least allocate 1 segment */
10018     stream->segments = g_new (QtDemuxSegment, MAX (n_segments, 1));
10019
10020     /* segments always start from 0 */
10021     time = 0;
10022     stime = 0;
10023     buffer += 16;
10024     for (segment_number = 0; segment_number < n_segments; segment_number++) {
10025       guint64 duration;
10026       guint64 media_time;
10027       gboolean empty_edit = FALSE;
10028       QtDemuxSegment *segment;
10029       guint32 rate_int;
10030       GstClockTime media_start = GST_CLOCK_TIME_NONE;
10031
10032       if (version == 1) {
10033         media_time = QT_UINT64 (buffer + 8);
10034         duration = QT_UINT64 (buffer);
10035         if (media_time == G_MAXUINT64)
10036           empty_edit = TRUE;
10037       } else {
10038         media_time = QT_UINT32 (buffer + 4);
10039         duration = QT_UINT32 (buffer);
10040         if (media_time == G_MAXUINT32)
10041           empty_edit = TRUE;
10042       }
10043
10044       if (!empty_edit)
10045         media_start = QTSTREAMTIME_TO_GSTTIME (stream, media_time);
10046
10047       segment = &stream->segments[segment_number];
10048
10049       /* time and duration expressed in global timescale */
10050       segment->time = stime;
10051       if (duration != 0 || empty_edit) {
10052         /* edge case: empty edits with duration=zero are treated here.
10053          * (files should not have these anyway). */
10054
10055         /* add non scaled values so we don't cause roundoff errors */
10056         time += duration;
10057         stime = QTTIME_TO_GSTTIME (qtdemux, time);
10058         segment->duration = stime - segment->time;
10059       } else {
10060         /* zero duration does not imply media_start == media_stop
10061          * but, only specify media_start. The edit ends with the track. */
10062         stime = segment->duration = GST_CLOCK_TIME_NONE;
10063         /* Don't allow more edits after this one. */
10064         n_segments = segment_number + 1;
10065       }
10066       segment->stop_time = stime;
10067
10068       segment->trak_media_start = media_time;
10069       /* media_time expressed in stream timescale */
10070       if (!empty_edit) {
10071         segment->media_start = media_start;
10072         segment->media_stop = GST_CLOCK_TIME_IS_VALID (segment->duration)
10073             ? segment->media_start + segment->duration : GST_CLOCK_TIME_NONE;
10074         media_segments_count++;
10075       } else {
10076         segment->media_start = GST_CLOCK_TIME_NONE;
10077         segment->media_stop = GST_CLOCK_TIME_NONE;
10078       }
10079       rate_int = QT_UINT32 (buffer + ((version == 1) ? 16 : 8));
10080
10081       if (rate_int <= 1) {
10082         /* 0 is not allowed, some programs write 1 instead of the floating point
10083          * value */
10084         GST_WARNING_OBJECT (qtdemux, "found suspicious rate %" G_GUINT32_FORMAT,
10085             rate_int);
10086         segment->rate = 1;
10087       } else {
10088         segment->rate = rate_int / 65536.0;
10089       }
10090
10091       GST_DEBUG_OBJECT (qtdemux, "created segment %d time %" GST_TIME_FORMAT
10092           ", duration %" GST_TIME_FORMAT ", media_start %" GST_TIME_FORMAT
10093           " (%" G_GUINT64_FORMAT ") , media_stop %" GST_TIME_FORMAT
10094           " stop_time %" GST_TIME_FORMAT " rate %g, (%d) timescale %u",
10095           segment_number, GST_TIME_ARGS (segment->time),
10096           GST_TIME_ARGS (segment->duration),
10097           GST_TIME_ARGS (segment->media_start), media_time,
10098           GST_TIME_ARGS (segment->media_stop),
10099           GST_TIME_ARGS (segment->stop_time), segment->rate, rate_int,
10100           stream->timescale);
10101       if (segment->stop_time > qtdemux->segment.stop &&
10102           !qtdemux->upstream_format_is_time) {
10103         GST_WARNING_OBJECT (qtdemux, "Segment %d "
10104             " extends to %" GST_TIME_FORMAT
10105             " past the end of the declared movie duration %" GST_TIME_FORMAT
10106             " movie segment will be extended", segment_number,
10107             GST_TIME_ARGS (segment->stop_time),
10108             GST_TIME_ARGS (qtdemux->segment.stop));
10109         qtdemux->segment.stop = qtdemux->segment.duration = segment->stop_time;
10110       }
10111
10112       buffer += entry_size;
10113     }
10114     GST_DEBUG_OBJECT (qtdemux, "found %d segments", n_segments);
10115     stream->n_segments = n_segments;
10116     if (media_segments_count != 1)
10117       allow_pushbased_edts = FALSE;
10118   }
10119 done:
10120
10121   /* push based does not handle segments, so act accordingly here,
10122    * and warn if applicable */
10123   if (!qtdemux->pullbased && !allow_pushbased_edts) {
10124     GST_WARNING_OBJECT (qtdemux, "streaming; discarding edit list segments");
10125     /* remove and use default one below, we stream like it anyway */
10126     g_free (stream->segments);
10127     stream->segments = NULL;
10128     stream->n_segments = 0;
10129   }
10130
10131   /* no segments, create one to play the complete trak */
10132   if (stream->n_segments == 0) {
10133     GstClockTime stream_duration =
10134         QTSTREAMTIME_TO_GSTTIME (stream, stream->duration);
10135
10136     if (stream->segments == NULL)
10137       stream->segments = g_new (QtDemuxSegment, 1);
10138
10139     /* represent unknown our way */
10140     if (stream_duration == 0)
10141       stream_duration = GST_CLOCK_TIME_NONE;
10142
10143     stream->segments[0].time = 0;
10144     stream->segments[0].stop_time = stream_duration;
10145     stream->segments[0].duration = stream_duration;
10146     stream->segments[0].media_start = 0;
10147     stream->segments[0].media_stop = stream_duration;
10148     stream->segments[0].rate = 1.0;
10149     stream->segments[0].trak_media_start = 0;
10150
10151     GST_DEBUG_OBJECT (qtdemux, "created dummy segment %" GST_TIME_FORMAT,
10152         GST_TIME_ARGS (stream_duration));
10153     stream->n_segments = 1;
10154     stream->dummy_segment = TRUE;
10155   }
10156   GST_DEBUG_OBJECT (qtdemux, "using %d segments", stream->n_segments);
10157
10158   return TRUE;
10159 }
10160
10161 /*
10162  * Parses the stsd atom of a svq3 trak looking for
10163  * the SMI and gama atoms.
10164  */
10165 static void
10166 qtdemux_parse_svq3_stsd_data (GstQTDemux * qtdemux,
10167     const guint8 * stsd_entry_data, const guint8 ** gamma, GstBuffer ** seqh)
10168 {
10169   const guint8 *_gamma = NULL;
10170   GstBuffer *_seqh = NULL;
10171   const guint8 *stsd_data = stsd_entry_data;
10172   guint32 length = QT_UINT32 (stsd_data);
10173   guint16 version;
10174
10175   if (length < 32) {
10176     GST_WARNING_OBJECT (qtdemux, "stsd too short");
10177     goto end;
10178   }
10179
10180   stsd_data += 16;
10181   length -= 16;
10182   version = QT_UINT16 (stsd_data);
10183   if (version == 3) {
10184     if (length >= 70) {
10185       length -= 70;
10186       stsd_data += 70;
10187       while (length > 8) {
10188         guint32 fourcc, size;
10189         const guint8 *data;
10190         size = QT_UINT32 (stsd_data);
10191         fourcc = QT_FOURCC (stsd_data + 4);
10192         data = stsd_data + 8;
10193
10194         if (size == 0) {
10195           GST_WARNING_OBJECT (qtdemux, "Atom of size 0 found, aborting "
10196               "svq3 atom parsing");
10197           goto end;
10198         }
10199
10200         switch (fourcc) {
10201           case FOURCC_gama:{
10202             if (size == 12) {
10203               _gamma = data;
10204             } else {
10205               GST_WARNING_OBJECT (qtdemux, "Unexpected size %" G_GUINT32_FORMAT
10206                   " for gama atom, expected 12", size);
10207             }
10208             break;
10209           }
10210           case FOURCC_SMI_:{
10211             if (size > 16 && QT_FOURCC (data) == FOURCC_SEQH) {
10212               guint32 seqh_size;
10213               if (_seqh != NULL) {
10214                 GST_WARNING_OBJECT (qtdemux, "Unexpected second SEQH SMI atom "
10215                     " found, ignoring");
10216               } else {
10217                 seqh_size = QT_UINT32 (data + 4);
10218                 if (seqh_size > 0) {
10219                   _seqh = gst_buffer_new_and_alloc (seqh_size);
10220                   gst_buffer_fill (_seqh, 0, data + 8, seqh_size);
10221                 }
10222               }
10223             }
10224             break;
10225           }
10226           default:{
10227             GST_WARNING_OBJECT (qtdemux, "Unhandled atom %" GST_FOURCC_FORMAT
10228                 " in SVQ3 entry in stsd atom", GST_FOURCC_ARGS (fourcc));
10229           }
10230         }
10231
10232         if (size <= length) {
10233           length -= size;
10234           stsd_data += size;
10235         }
10236       }
10237     } else {
10238       GST_WARNING_OBJECT (qtdemux, "SVQ3 entry too short in stsd atom");
10239     }
10240   } else {
10241     GST_WARNING_OBJECT (qtdemux, "Unexpected version for SVQ3 entry %"
10242         G_GUINT16_FORMAT, version);
10243     goto end;
10244   }
10245
10246 end:
10247   if (gamma) {
10248     *gamma = _gamma;
10249   }
10250   if (seqh) {
10251     *seqh = _seqh;
10252   } else if (_seqh) {
10253     gst_buffer_unref (_seqh);
10254   }
10255 }
10256
10257 static gchar *
10258 qtdemux_get_rtsp_uri_from_hndl (GstQTDemux * qtdemux, GNode * minf)
10259 {
10260   GNode *dinf;
10261   GstByteReader dref;
10262   gchar *uri = NULL;
10263
10264   /*
10265    * Get 'dinf', to get its child 'dref', that might contain a 'hndl'
10266    * atom that might contain a 'data' atom with the rtsp uri.
10267    * This case was reported in bug #597497, some info about
10268    * the hndl atom can be found in TN1195
10269    */
10270   dinf = qtdemux_tree_get_child_by_type (minf, FOURCC_dinf);
10271   GST_DEBUG_OBJECT (qtdemux, "Trying to obtain rtsp URI for stream trak");
10272
10273   if (dinf) {
10274     guint32 dref_num_entries = 0;
10275     if (qtdemux_tree_get_child_by_type_full (dinf, FOURCC_dref, &dref) &&
10276         gst_byte_reader_skip (&dref, 4) &&
10277         gst_byte_reader_get_uint32_be (&dref, &dref_num_entries)) {
10278       gint i;
10279
10280       /* search dref entries for hndl atom */
10281       for (i = 0; i < dref_num_entries; i++) {
10282         guint32 size = 0, type;
10283         guint8 string_len = 0;
10284         if (gst_byte_reader_get_uint32_be (&dref, &size) &&
10285             qt_atom_parser_get_fourcc (&dref, &type)) {
10286           if (type == FOURCC_hndl) {
10287             GST_DEBUG_OBJECT (qtdemux, "Found hndl atom");
10288
10289             /* skip data reference handle bytes and the
10290              * following pascal string and some extra 4
10291              * bytes I have no idea what are */
10292             if (!gst_byte_reader_skip (&dref, 4) ||
10293                 !gst_byte_reader_get_uint8 (&dref, &string_len) ||
10294                 !gst_byte_reader_skip (&dref, string_len + 4)) {
10295               GST_WARNING_OBJECT (qtdemux, "Failed to parse hndl atom");
10296               break;
10297             }
10298
10299             /* iterate over the atoms to find the data atom */
10300             while (gst_byte_reader_get_remaining (&dref) >= 8) {
10301               guint32 atom_size;
10302               guint32 atom_type;
10303
10304               if (gst_byte_reader_get_uint32_be (&dref, &atom_size) &&
10305                   qt_atom_parser_get_fourcc (&dref, &atom_type)) {
10306                 if (atom_type == FOURCC_data) {
10307                   const guint8 *uri_aux = NULL;
10308
10309                   /* found the data atom that might contain the rtsp uri */
10310                   GST_DEBUG_OBJECT (qtdemux, "Found data atom inside "
10311                       "hndl atom, interpreting it as an URI");
10312                   if (gst_byte_reader_peek_data (&dref, atom_size - 8,
10313                           &uri_aux)) {
10314                     if (g_strstr_len ((gchar *) uri_aux, 7, "rtsp://") != NULL)
10315                       uri = g_strndup ((gchar *) uri_aux, atom_size - 8);
10316                     else
10317                       GST_WARNING_OBJECT (qtdemux, "Data atom in hndl atom "
10318                           "didn't contain a rtsp address");
10319                   } else {
10320                     GST_WARNING_OBJECT (qtdemux, "Failed to get the data "
10321                         "atom contents");
10322                   }
10323                   break;
10324                 }
10325                 /* skipping to the next entry */
10326                 if (!gst_byte_reader_skip (&dref, atom_size - 8))
10327                   break;
10328               } else {
10329                 GST_WARNING_OBJECT (qtdemux, "Failed to parse hndl child "
10330                     "atom header");
10331                 break;
10332               }
10333             }
10334             break;
10335           }
10336           /* skip to the next entry */
10337           if (!gst_byte_reader_skip (&dref, size - 8))
10338             break;
10339         } else {
10340           GST_WARNING_OBJECT (qtdemux, "Error parsing dref atom");
10341         }
10342       }
10343       GST_DEBUG_OBJECT (qtdemux, "Finished parsing dref atom");
10344     }
10345   }
10346   return uri;
10347 }
10348
10349 #define AMR_NB_ALL_MODES        0x81ff
10350 #define AMR_WB_ALL_MODES        0x83ff
10351 static guint
10352 qtdemux_parse_amr_bitrate (GstBuffer * buf, gboolean wb)
10353 {
10354   /* The 'damr' atom is of the form:
10355    *
10356    * | vendor | decoder_ver | mode_set | mode_change_period | frames/sample |
10357    *    32 b       8 b          16 b           8 b                 8 b
10358    *
10359    * The highest set bit of the first 7 (AMR-NB) or 8 (AMR-WB) bits of mode_set
10360    * represents the highest mode used in the stream (and thus the maximum
10361    * bitrate), with a couple of special cases as seen below.
10362    */
10363
10364   /* Map of frame type ID -> bitrate */
10365   static const guint nb_bitrates[] = {
10366     4750, 5150, 5900, 6700, 7400, 7950, 10200, 12200
10367   };
10368   static const guint wb_bitrates[] = {
10369     6600, 8850, 12650, 14250, 15850, 18250, 19850, 23050, 23850
10370   };
10371   GstMapInfo map;
10372   gsize max_mode;
10373   guint16 mode_set;
10374
10375   gst_buffer_map (buf, &map, GST_MAP_READ);
10376
10377   if (map.size != 0x11) {
10378     GST_DEBUG ("Atom should have size 0x11, not %" G_GSIZE_FORMAT, map.size);
10379     goto bad_data;
10380   }
10381
10382   if (QT_FOURCC (map.data + 4) != FOURCC_damr) {
10383     GST_DEBUG ("Unknown atom in %" GST_FOURCC_FORMAT,
10384         GST_FOURCC_ARGS (QT_UINT32 (map.data + 4)));
10385     goto bad_data;
10386   }
10387
10388   mode_set = QT_UINT16 (map.data + 13);
10389
10390   if (mode_set == (wb ? AMR_WB_ALL_MODES : AMR_NB_ALL_MODES))
10391     max_mode = 7 + (wb ? 1 : 0);
10392   else
10393     /* AMR-NB modes fo from 0-7, and AMR-WB modes go from 0-8 */
10394     max_mode = g_bit_nth_msf ((gulong) mode_set & (wb ? 0x1ff : 0xff), -1);
10395
10396   if (max_mode == -1) {
10397     GST_DEBUG ("No mode indication was found (mode set) = %x",
10398         (guint) mode_set);
10399     goto bad_data;
10400   }
10401
10402   gst_buffer_unmap (buf, &map);
10403   return wb ? wb_bitrates[max_mode] : nb_bitrates[max_mode];
10404
10405 bad_data:
10406   gst_buffer_unmap (buf, &map);
10407   return 0;
10408 }
10409
10410 static gboolean
10411 qtdemux_parse_transformation_matrix (GstQTDemux * qtdemux,
10412     GstByteReader * reader, guint32 * matrix, const gchar * atom)
10413 {
10414   /*
10415    * 9 values of 32 bits (fixed point 16.16, except 2 5 and 8 that are 2.30)
10416    * [0 1 2]
10417    * [3 4 5]
10418    * [6 7 8]
10419    */
10420
10421   if (gst_byte_reader_get_remaining (reader) < 36)
10422     return FALSE;
10423
10424   matrix[0] = gst_byte_reader_get_uint32_be_unchecked (reader);
10425   matrix[1] = gst_byte_reader_get_uint32_be_unchecked (reader);
10426   matrix[2] = gst_byte_reader_get_uint32_be_unchecked (reader);
10427   matrix[3] = gst_byte_reader_get_uint32_be_unchecked (reader);
10428   matrix[4] = gst_byte_reader_get_uint32_be_unchecked (reader);
10429   matrix[5] = gst_byte_reader_get_uint32_be_unchecked (reader);
10430   matrix[6] = gst_byte_reader_get_uint32_be_unchecked (reader);
10431   matrix[7] = gst_byte_reader_get_uint32_be_unchecked (reader);
10432   matrix[8] = gst_byte_reader_get_uint32_be_unchecked (reader);
10433
10434   GST_DEBUG_OBJECT (qtdemux, "Transformation matrix from atom %s", atom);
10435   GST_DEBUG_OBJECT (qtdemux, "%u.%u %u.%u %u.%u", matrix[0] >> 16,
10436       matrix[0] & 0xFFFF, matrix[1] >> 16, matrix[1] & 0xFF, matrix[2] >> 16,
10437       matrix[2] & 0xFF);
10438   GST_DEBUG_OBJECT (qtdemux, "%u.%u %u.%u %u.%u", matrix[3] >> 16,
10439       matrix[3] & 0xFFFF, matrix[4] >> 16, matrix[4] & 0xFF, matrix[5] >> 16,
10440       matrix[5] & 0xFF);
10441   GST_DEBUG_OBJECT (qtdemux, "%u.%u %u.%u %u.%u", matrix[6] >> 16,
10442       matrix[6] & 0xFFFF, matrix[7] >> 16, matrix[7] & 0xFF, matrix[8] >> 16,
10443       matrix[8] & 0xFF);
10444
10445   return TRUE;
10446 }
10447
10448 static void
10449 qtdemux_inspect_transformation_matrix (GstQTDemux * qtdemux,
10450     QtDemuxStream * stream, guint32 * matrix, GstTagList ** taglist)
10451 {
10452
10453 /* [a b c]
10454  * [d e f]
10455  * [g h i]
10456  *
10457  * This macro will only compare value abdegh, it expects cfi to have already
10458  * been checked
10459  */
10460 #define QTCHECK_MATRIX(m,a,b,d,e) ((m)[0] == (a << 16) && (m)[1] == (b << 16) && \
10461                                    (m)[3] == (d << 16) && (m)[4] == (e << 16))
10462
10463   /* only handle the cases where the last column has standard values */
10464   if (matrix[2] == 0 && matrix[5] == 0 && matrix[8] == 1 << 30) {
10465     const gchar *rotation_tag = NULL;
10466
10467     /* no rotation needed */
10468     if (QTCHECK_MATRIX (matrix, 1, 0, 0, 1)) {
10469       /* NOP */
10470     } else if (QTCHECK_MATRIX (matrix, 0, 1, G_MAXUINT16, 0)) {
10471       rotation_tag = "rotate-90";
10472     } else if (QTCHECK_MATRIX (matrix, G_MAXUINT16, 0, 0, G_MAXUINT16)) {
10473       rotation_tag = "rotate-180";
10474     } else if (QTCHECK_MATRIX (matrix, 0, G_MAXUINT16, 1, 0)) {
10475       rotation_tag = "rotate-270";
10476     } else {
10477       GST_FIXME_OBJECT (qtdemux, "Unhandled transformation matrix values");
10478     }
10479
10480     GST_DEBUG_OBJECT (qtdemux, "Transformation matrix rotation %s",
10481         GST_STR_NULL (rotation_tag));
10482     if (rotation_tag != NULL) {
10483       if (*taglist == NULL)
10484         *taglist = gst_tag_list_new_empty ();
10485       gst_tag_list_add (*taglist, GST_TAG_MERGE_REPLACE,
10486           GST_TAG_IMAGE_ORIENTATION, rotation_tag, NULL);
10487     }
10488   } else {
10489     GST_FIXME_OBJECT (qtdemux, "Unhandled transformation matrix values");
10490   }
10491 }
10492
10493 static gboolean
10494 qtdemux_parse_protection_aavd (GstQTDemux * qtdemux,
10495     QtDemuxStream * stream, GNode * container, guint32 * original_fmt)
10496 {
10497   GNode *adrm;
10498   guint32 adrm_size;
10499   GstBuffer *adrm_buf = NULL;
10500   QtDemuxAavdEncryptionInfo *info;
10501
10502   adrm = qtdemux_tree_get_child_by_type (container, FOURCC_adrm);
10503   if (G_UNLIKELY (!adrm)) {
10504     GST_ERROR_OBJECT (qtdemux, "aavd box does not contain mandatory adrm box");
10505     return FALSE;
10506   }
10507   adrm_size = QT_UINT32 (adrm->data);
10508   adrm_buf = gst_buffer_new_memdup (adrm->data, adrm_size);
10509
10510   stream->protection_scheme_type = FOURCC_aavd;
10511
10512   if (!stream->protection_scheme_info)
10513     stream->protection_scheme_info = g_new0 (QtDemuxAavdEncryptionInfo, 1);
10514
10515   info = (QtDemuxAavdEncryptionInfo *) stream->protection_scheme_info;
10516
10517   if (info->default_properties)
10518     gst_structure_free (info->default_properties);
10519   info->default_properties = gst_structure_new ("application/x-aavd",
10520       "encrypted", G_TYPE_BOOLEAN, TRUE,
10521       "adrm", GST_TYPE_BUFFER, adrm_buf, NULL);
10522   gst_buffer_unref (adrm_buf);
10523
10524   *original_fmt = FOURCC_mp4a;
10525   return TRUE;
10526 }
10527
10528 /* Parses the boxes defined in ISO/IEC 14496-12 that enable support for
10529  * protected streams (sinf, frma, schm and schi); if the protection scheme is
10530  * Common Encryption (cenc), the function will also parse the tenc box (defined
10531  * in ISO/IEC 23001-7). @container points to the node that contains these boxes
10532  * (typically an enc[v|a|t|s] sample entry); the function will set
10533  * @original_fmt to the fourcc of the original unencrypted stream format.
10534  * Returns TRUE if successful; FALSE otherwise. */
10535 static gboolean
10536 qtdemux_parse_protection_scheme_info (GstQTDemux * qtdemux,
10537     QtDemuxStream * stream, GNode * container, guint32 * original_fmt)
10538 {
10539   GNode *sinf;
10540   GNode *frma;
10541   GNode *schm;
10542   GNode *schi;
10543   QtDemuxCencSampleSetInfo *info;
10544   GNode *tenc;
10545   const guint8 *tenc_data;
10546
10547   g_return_val_if_fail (qtdemux != NULL, FALSE);
10548   g_return_val_if_fail (stream != NULL, FALSE);
10549   g_return_val_if_fail (container != NULL, FALSE);
10550   g_return_val_if_fail (original_fmt != NULL, FALSE);
10551
10552   sinf = qtdemux_tree_get_child_by_type (container, FOURCC_sinf);
10553   if (G_UNLIKELY (!sinf)) {
10554     if (stream->protection_scheme_type == FOURCC_cenc
10555         || stream->protection_scheme_type == FOURCC_cbcs) {
10556       GST_ERROR_OBJECT (qtdemux, "sinf box does not contain schi box, which is "
10557           "mandatory for Common Encryption");
10558       return FALSE;
10559     }
10560     return TRUE;
10561   }
10562
10563   frma = qtdemux_tree_get_child_by_type (sinf, FOURCC_frma);
10564   if (G_UNLIKELY (!frma)) {
10565     GST_ERROR_OBJECT (qtdemux, "sinf box does not contain mandatory frma box");
10566     return FALSE;
10567   }
10568
10569   *original_fmt = QT_FOURCC ((const guint8 *) frma->data + 8);
10570   GST_DEBUG_OBJECT (qtdemux, "original stream format: '%" GST_FOURCC_FORMAT "'",
10571       GST_FOURCC_ARGS (*original_fmt));
10572
10573   schm = qtdemux_tree_get_child_by_type (sinf, FOURCC_schm);
10574   if (!schm) {
10575     GST_DEBUG_OBJECT (qtdemux, "sinf box does not contain schm box");
10576     return FALSE;
10577   }
10578   stream->protection_scheme_type = QT_FOURCC ((const guint8 *) schm->data + 12);
10579   stream->protection_scheme_version =
10580       QT_UINT32 ((const guint8 *) schm->data + 16);
10581
10582   GST_DEBUG_OBJECT (qtdemux,
10583       "protection_scheme_type: %" GST_FOURCC_FORMAT ", "
10584       "protection_scheme_version: %#010x",
10585       GST_FOURCC_ARGS (stream->protection_scheme_type),
10586       stream->protection_scheme_version);
10587
10588   schi = qtdemux_tree_get_child_by_type (sinf, FOURCC_schi);
10589   if (!schi) {
10590     GST_DEBUG_OBJECT (qtdemux, "sinf box does not contain schi box");
10591     return FALSE;
10592   }
10593   if (stream->protection_scheme_type != FOURCC_cenc &&
10594       stream->protection_scheme_type != FOURCC_piff &&
10595       stream->protection_scheme_type != FOURCC_cbcs) {
10596     GST_ERROR_OBJECT (qtdemux,
10597         "Invalid protection_scheme_type: %" GST_FOURCC_FORMAT,
10598         GST_FOURCC_ARGS (stream->protection_scheme_type));
10599     return FALSE;
10600   }
10601
10602   if (G_UNLIKELY (!stream->protection_scheme_info))
10603     stream->protection_scheme_info =
10604         g_malloc0 (sizeof (QtDemuxCencSampleSetInfo));
10605
10606   info = (QtDemuxCencSampleSetInfo *) stream->protection_scheme_info;
10607
10608   if (stream->protection_scheme_type == FOURCC_cenc
10609       || stream->protection_scheme_type == FOURCC_cbcs) {
10610     guint8 is_encrypted;
10611     guint8 iv_size;
10612     guint8 constant_iv_size = 0;
10613     const guint8 *default_kid;
10614     guint8 crypt_byte_block = 0;
10615     guint8 skip_byte_block = 0;
10616     const guint8 *constant_iv = NULL;
10617
10618     tenc = qtdemux_tree_get_child_by_type (schi, FOURCC_tenc);
10619     if (!tenc) {
10620       GST_ERROR_OBJECT (qtdemux, "schi box does not contain tenc box, "
10621           "which is mandatory for Common Encryption");
10622       return FALSE;
10623     }
10624     tenc_data = (const guint8 *) tenc->data + 12;
10625     is_encrypted = QT_UINT8 (tenc_data + 2);
10626     iv_size = QT_UINT8 (tenc_data + 3);
10627     default_kid = (tenc_data + 4);
10628     if (stream->protection_scheme_type == FOURCC_cbcs) {
10629       guint8 possible_pattern_info;
10630       if (iv_size == 0) {
10631         constant_iv_size = QT_UINT8 (tenc_data + 20);
10632         if (constant_iv_size != 8 && constant_iv_size != 16) {
10633           GST_ERROR_OBJECT (qtdemux,
10634               "constant IV size should be 8 or 16, not %hhu", constant_iv_size);
10635           return FALSE;
10636         }
10637         constant_iv = (tenc_data + 21);
10638       }
10639       possible_pattern_info = QT_UINT8 (tenc_data + 1);
10640       crypt_byte_block = (possible_pattern_info >> 4) & 0x0f;
10641       skip_byte_block = possible_pattern_info & 0x0f;
10642     }
10643     qtdemux_update_default_sample_cenc_settings (qtdemux, info,
10644         is_encrypted, stream->protection_scheme_type, iv_size, default_kid,
10645         crypt_byte_block, skip_byte_block, constant_iv_size, constant_iv);
10646   } else if (stream->protection_scheme_type == FOURCC_piff) {
10647     GstByteReader br;
10648     static const guint8 piff_track_encryption_uuid[] = {
10649       0x89, 0x74, 0xdb, 0xce, 0x7b, 0xe7, 0x4c, 0x51,
10650       0x84, 0xf9, 0x71, 0x48, 0xf9, 0x88, 0x25, 0x54
10651     };
10652
10653     tenc = qtdemux_tree_get_child_by_type (schi, FOURCC_uuid);
10654     if (!tenc) {
10655       GST_ERROR_OBJECT (qtdemux, "schi box does not contain tenc box, "
10656           "which is mandatory for Common Encryption");
10657       return FALSE;
10658     }
10659
10660     tenc_data = (const guint8 *) tenc->data + 8;
10661     if (memcmp (tenc_data, piff_track_encryption_uuid, 16) != 0) {
10662       gchar *box_uuid = qtdemux_uuid_bytes_to_string (tenc_data);
10663       GST_ERROR_OBJECT (qtdemux,
10664           "Unsupported track encryption box with uuid: %s", box_uuid);
10665       g_free (box_uuid);
10666       return FALSE;
10667     }
10668     tenc_data = (const guint8 *) tenc->data + 16 + 12;
10669     gst_byte_reader_init (&br, tenc_data, 20);
10670     if (!qtdemux_update_default_piff_encryption_settings (qtdemux, info, &br)) {
10671       GST_ERROR_OBJECT (qtdemux, "PIFF track box parsing error");
10672       return FALSE;
10673     }
10674     stream->protection_scheme_type = FOURCC_cenc;
10675   }
10676
10677   return TRUE;
10678 }
10679
10680 static gint
10681 qtdemux_track_id_compare_func (QtDemuxStream ** stream1,
10682     QtDemuxStream ** stream2)
10683 {
10684   return (gint) (*stream1)->track_id - (gint) (*stream2)->track_id;
10685 }
10686
10687 static gboolean
10688 qtdemux_parse_stereo_svmi_atom (GstQTDemux * qtdemux, QtDemuxStream * stream,
10689     GNode * stbl)
10690 {
10691   GNode *svmi;
10692
10693   /*parse svmi header if existing */
10694   svmi = qtdemux_tree_get_child_by_type (stbl, FOURCC_svmi);
10695   if (svmi) {
10696     guint len = QT_UINT32 ((guint8 *) svmi->data);
10697     guint32 version = QT_UINT32 ((guint8 *) svmi->data + 8);
10698     if (!version) {
10699       GstVideoMultiviewMode mode = GST_VIDEO_MULTIVIEW_MODE_NONE;
10700       GstVideoMultiviewFlags flags = GST_VIDEO_MULTIVIEW_FLAGS_NONE;
10701       guint8 frame_type, frame_layout;
10702       guint32 stereo_mono_change_count;
10703
10704       if (len < 18)
10705         return FALSE;
10706
10707       /* MPEG-A stereo video */
10708       if (qtdemux->major_brand == FOURCC_ss02)
10709         flags |= GST_VIDEO_MULTIVIEW_FLAGS_MIXED_MONO;
10710
10711       frame_type = QT_UINT8 ((guint8 *) svmi->data + 12);
10712       frame_layout = QT_UINT8 ((guint8 *) svmi->data + 13) & 0x01;
10713       stereo_mono_change_count = QT_UINT32 ((guint8 *) svmi->data + 14);
10714
10715       switch (frame_type) {
10716         case 0:
10717           mode = GST_VIDEO_MULTIVIEW_MODE_SIDE_BY_SIDE;
10718           break;
10719         case 1:
10720           mode = GST_VIDEO_MULTIVIEW_MODE_ROW_INTERLEAVED;
10721           break;
10722         case 2:
10723           mode = GST_VIDEO_MULTIVIEW_MODE_FRAME_BY_FRAME;
10724           break;
10725         case 3:
10726           /* mode 3 is primary/secondary view sequence, ie
10727            * left/right views in separate tracks. See section 7.2
10728            * of ISO/IEC 23000-11:2009 */
10729           /* In the future this might be supported using related
10730            * streams, like an enhancement track - if files like this
10731            * ever exist */
10732           GST_FIXME_OBJECT (qtdemux,
10733               "Implement stereo video in separate streams");
10734       }
10735
10736       if ((frame_layout & 0x1) == 0)
10737         flags |= GST_VIDEO_MULTIVIEW_FLAGS_RIGHT_VIEW_FIRST;
10738
10739       GST_LOG_OBJECT (qtdemux,
10740           "StereoVideo: composition type: %u, is_left_first: %u",
10741           frame_type, frame_layout);
10742
10743       if (stereo_mono_change_count > 1) {
10744         GST_FIXME_OBJECT (qtdemux,
10745             "Mixed-mono flags are not yet supported in qtdemux.");
10746       }
10747
10748       stream->multiview_mode = mode;
10749       stream->multiview_flags = flags;
10750     }
10751   }
10752
10753   return TRUE;
10754 }
10755
10756 /* parse the traks.
10757  * With each track we associate a new QtDemuxStream that contains all the info
10758  * about the trak.
10759  * traks that do not decode to something (like strm traks) will not have a pad.
10760  */
10761 static gboolean
10762 qtdemux_parse_trak (GstQTDemux * qtdemux, GNode * trak)
10763 {
10764   GstByteReader tkhd;
10765   int offset;
10766   GNode *mdia;
10767   GNode *mdhd;
10768   GNode *hdlr;
10769   GNode *minf;
10770   GNode *stbl;
10771   GNode *stsd;
10772   GNode *mp4a;
10773   GNode *mp4v;
10774   GNode *esds;
10775   GNode *tref;
10776   GNode *udta;
10777
10778   QtDemuxStream *stream = NULL;
10779   const guint8 *stsd_data;
10780   const guint8 *stsd_entry_data;
10781   guint remaining_stsd_len;
10782   guint stsd_entry_count;
10783   guint stsd_index;
10784   guint16 lang_code;            /* quicktime lang code or packed iso code */
10785   guint32 version;
10786   guint32 tkhd_flags = 0;
10787   guint8 tkhd_version = 0;
10788   guint32 w = 0, h = 0;
10789   guint value_size, stsd_len, len;
10790   guint32 track_id;
10791   guint32 dummy;
10792
10793   GST_DEBUG_OBJECT (qtdemux, "parse_trak");
10794
10795   if (!qtdemux_tree_get_child_by_type_full (trak, FOURCC_tkhd, &tkhd)
10796       || !gst_byte_reader_get_uint8 (&tkhd, &tkhd_version)
10797       || !gst_byte_reader_get_uint24_be (&tkhd, &tkhd_flags))
10798     goto corrupt_file;
10799
10800   /* pick between 64 or 32 bits */
10801   value_size = tkhd_version == 1 ? 8 : 4;
10802   if (!gst_byte_reader_skip (&tkhd, value_size * 2) ||
10803       !gst_byte_reader_get_uint32_be (&tkhd, &track_id))
10804     goto corrupt_file;
10805
10806   /* Check if current moov has duplicated track_id */
10807   if (qtdemux_find_stream (qtdemux, track_id))
10808     goto existing_stream;
10809
10810   stream = _create_stream (qtdemux, track_id);
10811   stream->stream_tags = gst_tag_list_make_writable (stream->stream_tags);
10812
10813   /* need defaults for fragments */
10814   qtdemux_parse_trex (qtdemux, stream, &dummy, &dummy, &dummy);
10815
10816   if ((tkhd_flags & 1) == 0)
10817     stream->disabled = TRUE;
10818
10819   GST_LOG_OBJECT (qtdemux, "track[tkhd] version/flags/id: 0x%02x/%06x/%u",
10820       tkhd_version, tkhd_flags, stream->track_id);
10821
10822   if (!(mdia = qtdemux_tree_get_child_by_type (trak, FOURCC_mdia)))
10823     goto corrupt_file;
10824
10825   if (!(mdhd = qtdemux_tree_get_child_by_type (mdia, FOURCC_mdhd))) {
10826     /* be nice for some crooked mjp2 files that use mhdr for mdhd */
10827     if (qtdemux->major_brand != FOURCC_mjp2 ||
10828         !(mdhd = qtdemux_tree_get_child_by_type (mdia, FOURCC_mhdr)))
10829       goto corrupt_file;
10830   }
10831
10832   len = QT_UINT32 ((guint8 *) mdhd->data);
10833   version = QT_UINT32 ((guint8 *) mdhd->data + 8);
10834   GST_LOG_OBJECT (qtdemux, "track version/flags: %08x", version);
10835   if (version == 0x01000000) {
10836     if (len < 42)
10837       goto corrupt_file;
10838     stream->timescale = QT_UINT32 ((guint8 *) mdhd->data + 28);
10839     stream->duration = QT_UINT64 ((guint8 *) mdhd->data + 32);
10840     lang_code = QT_UINT16 ((guint8 *) mdhd->data + 40);
10841   } else {
10842     if (len < 30)
10843       goto corrupt_file;
10844     stream->timescale = QT_UINT32 ((guint8 *) mdhd->data + 20);
10845     stream->duration = QT_UINT32 ((guint8 *) mdhd->data + 24);
10846     lang_code = QT_UINT16 ((guint8 *) mdhd->data + 28);
10847   }
10848
10849   if (lang_code < 0x400) {
10850     qtdemux_lang_map_qt_code_to_iso (stream->lang_id, lang_code);
10851   } else if (lang_code == 0x7fff) {
10852     stream->lang_id[0] = 0;     /* unspecified */
10853   } else {
10854     stream->lang_id[0] = 0x60 + ((lang_code >> 10) & 0x1F);
10855     stream->lang_id[1] = 0x60 + ((lang_code >> 5) & 0x1F);
10856     stream->lang_id[2] = 0x60 + (lang_code & 0x1F);
10857     stream->lang_id[3] = 0;
10858   }
10859
10860   GST_LOG_OBJECT (qtdemux, "track timescale: %" G_GUINT32_FORMAT,
10861       stream->timescale);
10862   GST_LOG_OBJECT (qtdemux, "track duration: %" G_GUINT64_FORMAT,
10863       stream->duration);
10864   GST_LOG_OBJECT (qtdemux, "track language code/id: 0x%04x/%s",
10865       lang_code, stream->lang_id);
10866
10867   if (G_UNLIKELY (stream->timescale == 0 || qtdemux->timescale == 0))
10868     goto corrupt_file;
10869
10870   if ((tref = qtdemux_tree_get_child_by_type (trak, FOURCC_tref))) {
10871     /* chapters track reference */
10872     GNode *chap = qtdemux_tree_get_child_by_type (tref, FOURCC_chap);
10873     if (chap) {
10874       gsize length = GST_READ_UINT32_BE (chap->data);
10875       if (qtdemux->chapters_track_id)
10876         GST_FIXME_OBJECT (qtdemux, "Multiple CHAP tracks");
10877
10878       if (length >= 12) {
10879         qtdemux->chapters_track_id =
10880             GST_READ_UINT32_BE ((gint8 *) chap->data + 8);
10881       }
10882     }
10883   }
10884
10885   /* fragmented files may have bogus duration in moov */
10886   if (!qtdemux->fragmented &&
10887       qtdemux->duration != G_MAXINT64 && stream->duration != G_MAXINT32) {
10888     guint64 tdur1, tdur2;
10889
10890     /* don't overflow */
10891     tdur1 = stream->timescale * (guint64) qtdemux->duration;
10892     tdur2 = qtdemux->timescale * (guint64) stream->duration;
10893
10894     /* HACK:
10895      * some of those trailers, nowadays, have prologue images that are
10896      * themselves video tracks as well. I haven't really found a way to
10897      * identify those yet, except for just looking at their duration. */
10898     if (tdur1 != 0 && (tdur2 * 10 / tdur1) < 2) {
10899       GST_WARNING_OBJECT (qtdemux,
10900           "Track shorter than 20%% (%" G_GUINT64_FORMAT "/%" G_GUINT32_FORMAT
10901           " vs. %" G_GUINT64_FORMAT "/%" G_GUINT32_FORMAT ") of the stream "
10902           "found, assuming preview image or something; skipping track",
10903           stream->duration, stream->timescale, qtdemux->duration,
10904           qtdemux->timescale);
10905       gst_qtdemux_stream_unref (stream);
10906       return TRUE;
10907     }
10908   }
10909
10910   if (!(hdlr = qtdemux_tree_get_child_by_type (mdia, FOURCC_hdlr)))
10911     goto corrupt_file;
10912
10913   GST_LOG_OBJECT (qtdemux, "track type: %" GST_FOURCC_FORMAT,
10914       GST_FOURCC_ARGS (QT_FOURCC ((guint8 *) hdlr->data + 12)));
10915
10916   len = QT_UINT32 ((guint8 *) hdlr->data);
10917   if (len >= 20)
10918     stream->subtype = QT_FOURCC ((guint8 *) hdlr->data + 16);
10919   GST_LOG_OBJECT (qtdemux, "track subtype: %" GST_FOURCC_FORMAT,
10920       GST_FOURCC_ARGS (stream->subtype));
10921
10922   if (!(minf = qtdemux_tree_get_child_by_type (mdia, FOURCC_minf)))
10923     goto corrupt_file;
10924
10925   if (!(stbl = qtdemux_tree_get_child_by_type (minf, FOURCC_stbl)))
10926     goto corrupt_file;
10927
10928   /* Parse out svmi (and later st3d/sv3d) atoms */
10929   if (!qtdemux_parse_stereo_svmi_atom (qtdemux, stream, stbl))
10930     goto corrupt_file;
10931
10932   /* parse rest of tkhd */
10933   if (stream->subtype == FOURCC_vide) {
10934     guint32 matrix[9];
10935
10936     /* version 1 uses some 64-bit ints */
10937     if (!gst_byte_reader_skip (&tkhd, 20 + value_size))
10938       goto corrupt_file;
10939
10940     if (!qtdemux_parse_transformation_matrix (qtdemux, &tkhd, matrix, "tkhd"))
10941       goto corrupt_file;
10942
10943     if (!gst_byte_reader_get_uint32_be (&tkhd, &w)
10944         || !gst_byte_reader_get_uint32_be (&tkhd, &h))
10945       goto corrupt_file;
10946
10947     qtdemux_inspect_transformation_matrix (qtdemux, stream, matrix,
10948         &stream->stream_tags);
10949   }
10950
10951   /* parse stsd */
10952   if (!(stsd = qtdemux_tree_get_child_by_type (stbl, FOURCC_stsd)))
10953     goto corrupt_file;
10954   stsd_data = (const guint8 *) stsd->data;
10955
10956   /* stsd should at least have one entry */
10957   stsd_len = QT_UINT32 (stsd_data);
10958   if (stsd_len < 24) {
10959     /* .. but skip stream with empty stsd produced by some Vivotek cameras */
10960     if (stream->subtype == FOURCC_vivo) {
10961       gst_qtdemux_stream_unref (stream);
10962       return TRUE;
10963     } else {
10964       goto corrupt_file;
10965     }
10966   }
10967
10968   stream->stsd_entries_length = stsd_entry_count = QT_UINT32 (stsd_data + 12);
10969   /* each stsd entry must contain at least 8 bytes */
10970   if (stream->stsd_entries_length == 0
10971       || stream->stsd_entries_length > stsd_len / 8) {
10972     stream->stsd_entries_length = 0;
10973     goto corrupt_file;
10974   }
10975   stream->stsd_entries = g_new0 (QtDemuxStreamStsdEntry, stsd_entry_count);
10976   GST_LOG_OBJECT (qtdemux, "stsd len:           %d", stsd_len);
10977   GST_LOG_OBJECT (qtdemux, "stsd entry count:   %u", stsd_entry_count);
10978
10979   stsd_entry_data = stsd_data + 16;
10980   remaining_stsd_len = stsd_len - 16;
10981   for (stsd_index = 0; stsd_index < stsd_entry_count; stsd_index++) {
10982     guint32 fourcc;
10983     gchar *codec = NULL;
10984     QtDemuxStreamStsdEntry *entry = &stream->stsd_entries[stsd_index];
10985
10986     /* and that entry should fit within stsd */
10987     len = QT_UINT32 (stsd_entry_data);
10988     if (len > remaining_stsd_len)
10989       goto corrupt_file;
10990
10991     entry->fourcc = fourcc = QT_FOURCC (stsd_entry_data + 4);
10992     GST_LOG_OBJECT (qtdemux, "stsd type:          %" GST_FOURCC_FORMAT,
10993         GST_FOURCC_ARGS (entry->fourcc));
10994     GST_LOG_OBJECT (qtdemux, "stsd type len:      %d", len);
10995
10996     if ((fourcc == FOURCC_drms) || (fourcc == FOURCC_drmi))
10997       goto error_encrypted;
10998
10999     if (fourcc == FOURCC_aavd) {
11000       if (stream->subtype != FOURCC_soun) {
11001         GST_ERROR_OBJECT (qtdemux,
11002             "Unexpeced stsd type 'aavd' outside 'soun' track");
11003       } else {
11004         /* encrypted audio with sound sample description v0 */
11005         GNode *enc = qtdemux_tree_get_child_by_type (stsd, fourcc);
11006         stream->protected = TRUE;
11007         if (!qtdemux_parse_protection_aavd (qtdemux, stream, enc, &fourcc))
11008           GST_ERROR_OBJECT (qtdemux, "Failed to parse protection scheme info");
11009       }
11010     }
11011
11012     if (fourcc == FOURCC_encv || fourcc == FOURCC_enca) {
11013       /* FIXME this looks wrong, there might be multiple children
11014        * with the same type */
11015       GNode *enc = qtdemux_tree_get_child_by_type (stsd, fourcc);
11016       stream->protected = TRUE;
11017       if (!qtdemux_parse_protection_scheme_info (qtdemux, stream, enc, &fourcc))
11018         GST_ERROR_OBJECT (qtdemux, "Failed to parse protection scheme info");
11019     }
11020
11021     if (stream->subtype == FOURCC_vide) {
11022       GNode *colr;
11023       GNode *fiel;
11024       GNode *pasp;
11025       gboolean gray;
11026       gint depth, palette_size, palette_count;
11027       guint32 *palette_data = NULL;
11028
11029       entry->sampled = TRUE;
11030
11031       stream->display_width = w >> 16;
11032       stream->display_height = h >> 16;
11033
11034       offset = 16;
11035       if (len < 86)             /* TODO verify */
11036         goto corrupt_file;
11037
11038       entry->width = QT_UINT16 (stsd_entry_data + offset + 16);
11039       entry->height = QT_UINT16 (stsd_entry_data + offset + 18);
11040       entry->fps_n = 0;         /* this is filled in later */
11041       entry->fps_d = 0;         /* this is filled in later */
11042       entry->bits_per_sample = QT_UINT16 (stsd_entry_data + offset + 66);
11043       entry->color_table_id = QT_UINT16 (stsd_entry_data + offset + 68);
11044
11045       /* if color_table_id is 0, ctab atom must follow; however some files
11046        * produced by TMPEGEnc have color_table_id = 0 and no ctab atom, so
11047        * if color table is not present we'll correct the value */
11048       if (entry->color_table_id == 0 &&
11049           (len < 90
11050               || QT_FOURCC (stsd_entry_data + offset + 70) != FOURCC_ctab)) {
11051         entry->color_table_id = -1;
11052       }
11053
11054       GST_LOG_OBJECT (qtdemux, "width %d, height %d, bps %d, color table id %d",
11055           entry->width, entry->height, entry->bits_per_sample,
11056           entry->color_table_id);
11057
11058       depth = entry->bits_per_sample;
11059
11060       /* more than 32 bits means grayscale */
11061       gray = (depth > 32);
11062       /* low 32 bits specify the depth  */
11063       depth &= 0x1F;
11064
11065       /* different number of palette entries is determined by depth. */
11066       palette_count = 0;
11067       if ((depth == 1) || (depth == 2) || (depth == 4) || (depth == 8))
11068         palette_count = (1 << depth);
11069       palette_size = palette_count * 4;
11070
11071       if (entry->color_table_id) {
11072         switch (palette_count) {
11073           case 0:
11074             break;
11075           case 2:
11076             palette_data = g_memdup2 (ff_qt_default_palette_2, palette_size);
11077             break;
11078           case 4:
11079             palette_data = g_memdup2 (ff_qt_default_palette_4, palette_size);
11080             break;
11081           case 16:
11082             if (gray)
11083               palette_data =
11084                   g_memdup2 (ff_qt_grayscale_palette_16, palette_size);
11085             else
11086               palette_data = g_memdup2 (ff_qt_default_palette_16, palette_size);
11087             break;
11088           case 256:
11089             if (gray)
11090               palette_data =
11091                   g_memdup2 (ff_qt_grayscale_palette_256, palette_size);
11092             else
11093               palette_data =
11094                   g_memdup2 (ff_qt_default_palette_256, palette_size);
11095             break;
11096           default:
11097             GST_ELEMENT_WARNING (qtdemux, STREAM, DEMUX,
11098                 (_("The video in this file might not play correctly.")),
11099                 ("unsupported palette depth %d", depth));
11100             break;
11101         }
11102       } else {
11103         gint i, j, start, end;
11104
11105         if (len < 94)
11106           goto corrupt_file;
11107
11108         /* read table */
11109         start = QT_UINT32 (stsd_entry_data + offset + 70);
11110         palette_count = QT_UINT16 (stsd_entry_data + offset + 74);
11111         end = QT_UINT16 (stsd_entry_data + offset + 76);
11112
11113         GST_LOG_OBJECT (qtdemux, "start %d, end %d, palette_count %d",
11114             start, end, palette_count);
11115
11116         if (end > 255)
11117           end = 255;
11118         if (start > end)
11119           start = end;
11120
11121         if (len < 94 + (end - start) * 8)
11122           goto corrupt_file;
11123
11124         /* palette is always the same size */
11125         palette_data = g_malloc0 (256 * 4);
11126         palette_size = 256 * 4;
11127
11128         for (j = 0, i = start; i <= end; j++, i++) {
11129           guint32 a, r, g, b;
11130
11131           a = QT_UINT16 (stsd_entry_data + offset + 78 + (j * 8));
11132           r = QT_UINT16 (stsd_entry_data + offset + 80 + (j * 8));
11133           g = QT_UINT16 (stsd_entry_data + offset + 82 + (j * 8));
11134           b = QT_UINT16 (stsd_entry_data + offset + 84 + (j * 8));
11135
11136           palette_data[i] = ((a & 0xff00) << 16) | ((r & 0xff00) << 8) |
11137               (g & 0xff00) | (b >> 8);
11138         }
11139       }
11140
11141       if (entry->caps)
11142         gst_caps_unref (entry->caps);
11143
11144       entry->caps =
11145           qtdemux_video_caps (qtdemux, stream, entry, fourcc, stsd_entry_data,
11146           &codec);
11147       if (G_UNLIKELY (!entry->caps)) {
11148         g_free (palette_data);
11149         goto unknown_stream;
11150       }
11151
11152       if (codec) {
11153         gst_tag_list_add (stream->stream_tags, GST_TAG_MERGE_REPLACE,
11154             GST_TAG_VIDEO_CODEC, codec, NULL);
11155         g_free (codec);
11156         codec = NULL;
11157       }
11158
11159       if (palette_data) {
11160         GstStructure *s;
11161
11162         if (entry->rgb8_palette)
11163           gst_memory_unref (entry->rgb8_palette);
11164         entry->rgb8_palette = gst_memory_new_wrapped (GST_MEMORY_FLAG_READONLY,
11165             palette_data, palette_size, 0, palette_size, palette_data, g_free);
11166
11167         s = gst_caps_get_structure (entry->caps, 0);
11168
11169         /* non-raw video has a palette_data property. raw video has the palette as
11170          * an extra plane that we append to the output buffers before we push
11171          * them*/
11172         if (!gst_structure_has_name (s, "video/x-raw")) {
11173           GstBuffer *palette;
11174
11175           palette = gst_buffer_new ();
11176           gst_buffer_append_memory (palette, entry->rgb8_palette);
11177           entry->rgb8_palette = NULL;
11178
11179           gst_caps_set_simple (entry->caps, "palette_data",
11180               GST_TYPE_BUFFER, palette, NULL);
11181           gst_buffer_unref (palette);
11182         }
11183       } else if (palette_count != 0) {
11184         GST_ELEMENT_WARNING (qtdemux, STREAM, NOT_IMPLEMENTED,
11185             (NULL), ("Unsupported palette depth %d", depth));
11186       }
11187
11188       GST_LOG_OBJECT (qtdemux, "frame count:   %u",
11189           QT_UINT16 (stsd_entry_data + offset + 32));
11190
11191       esds = NULL;
11192       pasp = NULL;
11193       colr = NULL;
11194       fiel = NULL;
11195       /* pick 'the' stsd child */
11196       mp4v = qtdemux_tree_get_child_by_index (stsd, stsd_index);
11197       // We should skip parsing the stsd for non-protected streams if
11198       // the entry doesn't match the fourcc, since they don't change
11199       // format. However, for protected streams we can have partial
11200       // encryption, where parts of the stream are encrypted and parts
11201       // not. For both parts of such streams, we should ensure the
11202       // esds overrides are parsed for both from the stsd.
11203       if (QTDEMUX_TREE_NODE_FOURCC (mp4v) != fourcc) {
11204         if (stream->protected && QTDEMUX_TREE_NODE_FOURCC (mp4v) != FOURCC_encv)
11205           mp4v = NULL;
11206         else if (!stream->protected)
11207           mp4v = NULL;
11208       }
11209
11210       if (mp4v) {
11211         esds = qtdemux_tree_get_child_by_type (mp4v, FOURCC_esds);
11212         pasp = qtdemux_tree_get_child_by_type (mp4v, FOURCC_pasp);
11213         colr = qtdemux_tree_get_child_by_type (mp4v, FOURCC_colr);
11214         fiel = qtdemux_tree_get_child_by_type (mp4v, FOURCC_fiel);
11215       }
11216
11217       if (pasp) {
11218         const guint8 *pasp_data = (const guint8 *) pasp->data;
11219         gint len = QT_UINT32 (pasp_data);
11220
11221         if (len == 16) {
11222           CUR_STREAM (stream)->par_w = QT_UINT32 (pasp_data + 8);
11223           CUR_STREAM (stream)->par_h = QT_UINT32 (pasp_data + 12);
11224         } else {
11225           CUR_STREAM (stream)->par_w = 0;
11226           CUR_STREAM (stream)->par_h = 0;
11227         }
11228       } else {
11229         CUR_STREAM (stream)->par_w = 0;
11230         CUR_STREAM (stream)->par_h = 0;
11231       }
11232
11233       if (fiel) {
11234         const guint8 *fiel_data = (const guint8 *) fiel->data;
11235         gint len = QT_UINT32 (fiel_data);
11236
11237         if (len == 10) {
11238           CUR_STREAM (stream)->interlace_mode = GST_READ_UINT8 (fiel_data + 8);
11239           CUR_STREAM (stream)->field_order = GST_READ_UINT8 (fiel_data + 9);
11240         }
11241       }
11242
11243       if (colr) {
11244         const guint8 *colr_data = (const guint8 *) colr->data;
11245         gint len = QT_UINT32 (colr_data);
11246
11247         if (len == 19 || len == 18) {
11248           guint32 color_type = GST_READ_UINT32_LE (colr_data + 8);
11249
11250           if (color_type == FOURCC_nclx || color_type == FOURCC_nclc) {
11251             guint16 primaries = GST_READ_UINT16_BE (colr_data + 12);
11252             guint16 transfer_function = GST_READ_UINT16_BE (colr_data + 14);
11253             guint16 matrix = GST_READ_UINT16_BE (colr_data + 16);
11254             gboolean full_range = len == 19 ? colr_data[17] >> 7 : FALSE;
11255
11256             CUR_STREAM (stream)->colorimetry.primaries =
11257                 gst_video_color_primaries_from_iso (primaries);
11258             CUR_STREAM (stream)->colorimetry.transfer =
11259                 gst_video_transfer_function_from_iso (transfer_function);
11260             CUR_STREAM (stream)->colorimetry.matrix =
11261                 gst_video_color_matrix_from_iso (matrix);
11262             CUR_STREAM (stream)->colorimetry.range =
11263                 full_range ? GST_VIDEO_COLOR_RANGE_0_255 :
11264                 GST_VIDEO_COLOR_RANGE_16_235;
11265           } else {
11266             GST_DEBUG_OBJECT (qtdemux, "Unsupported color type");
11267           }
11268         } else {
11269           GST_WARNING_OBJECT (qtdemux, "Invalid colr atom size");
11270         }
11271       }
11272
11273       if (esds) {
11274         gst_qtdemux_handle_esds (qtdemux, stream, entry, esds,
11275             stream->stream_tags);
11276       } else {
11277         switch (fourcc) {
11278           case FOURCC_H264:
11279           case FOURCC_avc1:
11280           case FOURCC_avc3:
11281           {
11282             gint len = QT_UINT32 (stsd_entry_data) - 0x56;
11283             const guint8 *avc_data = stsd_entry_data + 0x56;
11284
11285             /* find avcC */
11286             while (len >= 0x8) {
11287               gint size;
11288
11289               if (QT_UINT32 (avc_data) <= len)
11290                 size = QT_UINT32 (avc_data) - 0x8;
11291               else
11292                 size = len - 0x8;
11293
11294               if (size < 1)
11295                 /* No real data, so break out */
11296                 break;
11297
11298               switch (QT_FOURCC (avc_data + 0x4)) {
11299                 case FOURCC_avcC:
11300                 {
11301                   /* parse, if found */
11302                   GstBuffer *buf;
11303
11304                   GST_DEBUG_OBJECT (qtdemux, "found avcC codec_data in stsd");
11305
11306                   /* First 4 bytes are the length of the atom, the next 4 bytes
11307                    * are the fourcc, the next 1 byte is the version, and the
11308                    * subsequent bytes are profile_tier_level structure like data. */
11309                   gst_codec_utils_h264_caps_set_level_and_profile (entry->caps,
11310                       avc_data + 8 + 1, size - 1);
11311                   buf = gst_buffer_new_and_alloc (size);
11312                   gst_buffer_fill (buf, 0, avc_data + 0x8, size);
11313                   gst_caps_set_simple (entry->caps,
11314                       "codec_data", GST_TYPE_BUFFER, buf, NULL);
11315                   gst_buffer_unref (buf);
11316
11317                   break;
11318                 }
11319                 case FOURCC_strf:
11320                 {
11321                   GstBuffer *buf;
11322
11323                   GST_DEBUG_OBJECT (qtdemux, "found strf codec_data in stsd");
11324
11325                   /* First 4 bytes are the length of the atom, the next 4 bytes
11326                    * are the fourcc, next 40 bytes are BITMAPINFOHEADER,
11327                    * next 1 byte is the version, and the
11328                    * subsequent bytes are sequence parameter set like data. */
11329
11330                   size -= 40;   /* we'll be skipping BITMAPINFOHEADER */
11331                   if (size > 1) {
11332                     gst_codec_utils_h264_caps_set_level_and_profile
11333                         (entry->caps, avc_data + 8 + 40 + 1, size - 1);
11334
11335                     buf = gst_buffer_new_and_alloc (size);
11336                     gst_buffer_fill (buf, 0, avc_data + 8 + 40, size);
11337                     gst_caps_set_simple (entry->caps,
11338                         "codec_data", GST_TYPE_BUFFER, buf, NULL);
11339                     gst_buffer_unref (buf);
11340                   }
11341                   break;
11342                 }
11343                 case FOURCC_btrt:
11344                 {
11345                   guint avg_bitrate, max_bitrate;
11346
11347                   /* bufferSizeDB, maxBitrate and avgBitrate - 4 bytes each */
11348                   if (size < 12)
11349                     break;
11350
11351                   max_bitrate = QT_UINT32 (avc_data + 0xc);
11352                   avg_bitrate = QT_UINT32 (avc_data + 0x10);
11353
11354                   if (!max_bitrate && !avg_bitrate)
11355                     break;
11356
11357                   /* Some muxers seem to swap the average and maximum bitrates
11358                    * (I'm looking at you, YouTube), so we swap for sanity. */
11359                   if (max_bitrate > 0 && max_bitrate < avg_bitrate) {
11360                     guint temp = avg_bitrate;
11361
11362                     avg_bitrate = max_bitrate;
11363                     max_bitrate = temp;
11364                   }
11365
11366                   if (max_bitrate > 0 && max_bitrate < G_MAXUINT32) {
11367                     gst_tag_list_add (stream->stream_tags,
11368                         GST_TAG_MERGE_REPLACE, GST_TAG_MAXIMUM_BITRATE,
11369                         max_bitrate, NULL);
11370                   }
11371                   if (avg_bitrate > 0 && avg_bitrate < G_MAXUINT32) {
11372                     gst_tag_list_add (stream->stream_tags,
11373                         GST_TAG_MERGE_REPLACE, GST_TAG_BITRATE, avg_bitrate,
11374                         NULL);
11375                   }
11376
11377                   break;
11378                 }
11379
11380                 default:
11381                   break;
11382               }
11383
11384               len -= size + 8;
11385               avc_data += size + 8;
11386             }
11387
11388             break;
11389           }
11390           case FOURCC_H265:
11391           case FOURCC_hvc1:
11392           case FOURCC_hev1:
11393           case FOURCC_dvh1:
11394           case FOURCC_dvhe:
11395           {
11396             gint len = QT_UINT32 (stsd_entry_data) - 0x56;
11397             const guint8 *hevc_data = stsd_entry_data + 0x56;
11398
11399             /* find hevc */
11400             while (len >= 0x8) {
11401               gint size;
11402
11403               if (QT_UINT32 (hevc_data) <= len)
11404                 size = QT_UINT32 (hevc_data) - 0x8;
11405               else
11406                 size = len - 0x8;
11407
11408               if (size < 1)
11409                 /* No real data, so break out */
11410                 break;
11411
11412               switch (QT_FOURCC (hevc_data + 0x4)) {
11413                 case FOURCC_hvcC:
11414                 {
11415                   /* parse, if found */
11416                   GstBuffer *buf;
11417
11418                   GST_DEBUG_OBJECT (qtdemux, "found hvcC codec_data in stsd");
11419
11420                   /* First 4 bytes are the length of the atom, the next 4 bytes
11421                    * are the fourcc, the next 1 byte is the version, and the
11422                    * subsequent bytes are sequence parameter set like data. */
11423                   gst_codec_utils_h265_caps_set_level_tier_and_profile
11424                       (entry->caps, hevc_data + 8 + 1, size - 1);
11425
11426                   buf = gst_buffer_new_and_alloc (size);
11427                   gst_buffer_fill (buf, 0, hevc_data + 0x8, size);
11428                   gst_caps_set_simple (entry->caps,
11429                       "codec_data", GST_TYPE_BUFFER, buf, NULL);
11430                   gst_buffer_unref (buf);
11431                   break;
11432                 }
11433                 default:
11434                   break;
11435               }
11436               len -= size + 8;
11437               hevc_data += size + 8;
11438             }
11439             break;
11440           }
11441           case FOURCC_mp4v:
11442           case FOURCC_MP4V:
11443           case FOURCC_fmp4:
11444           case FOURCC_FMP4:
11445           case FOURCC_xvid:
11446           case FOURCC_XVID:
11447           {
11448             GNode *glbl;
11449
11450             GST_DEBUG_OBJECT (qtdemux, "found %" GST_FOURCC_FORMAT,
11451                 GST_FOURCC_ARGS (fourcc));
11452
11453             /* codec data might be in glbl extension atom */
11454             glbl = mp4v ?
11455                 qtdemux_tree_get_child_by_type (mp4v, FOURCC_glbl) : NULL;
11456             if (glbl) {
11457               guint8 *data;
11458               GstBuffer *buf;
11459               gint len;
11460
11461               GST_DEBUG_OBJECT (qtdemux, "found glbl data in stsd");
11462               data = glbl->data;
11463               len = QT_UINT32 (data);
11464               if (len > 0x8) {
11465                 len -= 0x8;
11466                 buf = gst_buffer_new_and_alloc (len);
11467                 gst_buffer_fill (buf, 0, data + 8, len);
11468                 gst_caps_set_simple (entry->caps,
11469                     "codec_data", GST_TYPE_BUFFER, buf, NULL);
11470                 gst_buffer_unref (buf);
11471               }
11472             }
11473             break;
11474           }
11475           case FOURCC_mjp2:
11476           {
11477             /* see annex I of the jpeg2000 spec */
11478             GNode *jp2h, *ihdr, *colr, *mjp2, *field, *prefix, *cmap, *cdef;
11479             const guint8 *data;
11480             const gchar *colorspace = NULL;
11481             gint ncomp = 0;
11482             guint32 ncomp_map = 0;
11483             gint32 *comp_map = NULL;
11484             guint32 nchan_def = 0;
11485             gint32 *chan_def = NULL;
11486
11487             GST_DEBUG_OBJECT (qtdemux, "found mjp2");
11488             /* some required atoms */
11489             mjp2 = qtdemux_tree_get_child_by_index (stsd, stsd_index);
11490             if (!mjp2)
11491               break;
11492             jp2h = qtdemux_tree_get_child_by_type (mjp2, FOURCC_jp2h);
11493             if (!jp2h)
11494               break;
11495
11496             /* number of components; redundant with info in codestream, but useful
11497                to a muxer */
11498             ihdr = qtdemux_tree_get_child_by_type (jp2h, FOURCC_ihdr);
11499             if (!ihdr || QT_UINT32 (ihdr->data) != 22)
11500               break;
11501             ncomp = QT_UINT16 (((guint8 *) ihdr->data) + 16);
11502
11503             colr = qtdemux_tree_get_child_by_type (jp2h, FOURCC_colr);
11504             if (!colr)
11505               break;
11506             GST_DEBUG_OBJECT (qtdemux, "found colr");
11507             /* extract colour space info */
11508             if (QT_UINT8 ((guint8 *) colr->data + 8) == 1) {
11509               switch (QT_UINT32 ((guint8 *) colr->data + 11)) {
11510                 case 16:
11511                   colorspace = "sRGB";
11512                   break;
11513                 case 17:
11514                   colorspace = "GRAY";
11515                   break;
11516                 case 18:
11517                   colorspace = "sYUV";
11518                   break;
11519                 default:
11520                   colorspace = NULL;
11521                   break;
11522               }
11523             }
11524             if (!colorspace)
11525               /* colr is required, and only values 16, 17, and 18 are specified,
11526                  so error if we have no colorspace */
11527               break;
11528
11529             /* extract component mapping */
11530             cmap = qtdemux_tree_get_child_by_type (jp2h, FOURCC_cmap);
11531             if (cmap) {
11532               guint32 cmap_len = 0;
11533               int i;
11534               cmap_len = QT_UINT32 (cmap->data);
11535               if (cmap_len >= 8) {
11536                 /* normal box, subtract off header */
11537                 cmap_len -= 8;
11538                 /* cmap: { u16 cmp; u8 mtyp; u8 pcol; }* */
11539                 if (cmap_len % 4 == 0) {
11540                   ncomp_map = (cmap_len / 4);
11541                   comp_map = g_new0 (gint32, ncomp_map);
11542                   for (i = 0; i < ncomp_map; i++) {
11543                     guint16 cmp;
11544                     guint8 mtyp, pcol;
11545                     cmp = QT_UINT16 (((guint8 *) cmap->data) + 8 + i * 4);
11546                     mtyp = QT_UINT8 (((guint8 *) cmap->data) + 8 + i * 4 + 2);
11547                     pcol = QT_UINT8 (((guint8 *) cmap->data) + 8 + i * 4 + 3);
11548                     comp_map[i] = (mtyp << 24) | (pcol << 16) | cmp;
11549                   }
11550                 }
11551               }
11552             }
11553             /* extract channel definitions */
11554             cdef = qtdemux_tree_get_child_by_type (jp2h, FOURCC_cdef);
11555             if (cdef) {
11556               guint32 cdef_len = 0;
11557               int i;
11558               cdef_len = QT_UINT32 (cdef->data);
11559               if (cdef_len >= 10) {
11560                 /* normal box, subtract off header and len */
11561                 cdef_len -= 10;
11562                 /* cdef: u16 n; { u16 cn; u16 typ; u16 asoc; }* */
11563                 if (cdef_len % 6 == 0) {
11564                   nchan_def = (cdef_len / 6);
11565                   chan_def = g_new0 (gint32, nchan_def);
11566                   for (i = 0; i < nchan_def; i++)
11567                     chan_def[i] = -1;
11568                   for (i = 0; i < nchan_def; i++) {
11569                     guint16 cn, typ, asoc;
11570                     cn = QT_UINT16 (((guint8 *) cdef->data) + 10 + i * 6);
11571                     typ = QT_UINT16 (((guint8 *) cdef->data) + 10 + i * 6 + 2);
11572                     asoc = QT_UINT16 (((guint8 *) cdef->data) + 10 + i * 6 + 4);
11573                     if (cn < nchan_def) {
11574                       switch (typ) {
11575                         case 0:
11576                           chan_def[cn] = asoc;
11577                           break;
11578                         case 1:
11579                           chan_def[cn] = 0;     /* alpha */
11580                           break;
11581                         default:
11582                           chan_def[cn] = -typ;
11583                       }
11584                     }
11585                   }
11586                 }
11587               }
11588             }
11589
11590             gst_caps_set_simple (entry->caps,
11591                 "num-components", G_TYPE_INT, ncomp, NULL);
11592             gst_caps_set_simple (entry->caps,
11593                 "colorspace", G_TYPE_STRING, colorspace, NULL);
11594
11595             if (comp_map) {
11596               GValue arr = { 0, };
11597               GValue elt = { 0, };
11598               int i;
11599               g_value_init (&arr, GST_TYPE_ARRAY);
11600               g_value_init (&elt, G_TYPE_INT);
11601               for (i = 0; i < ncomp_map; i++) {
11602                 g_value_set_int (&elt, comp_map[i]);
11603                 gst_value_array_append_value (&arr, &elt);
11604               }
11605               gst_structure_set_value (gst_caps_get_structure (entry->caps, 0),
11606                   "component-map", &arr);
11607               g_value_unset (&elt);
11608               g_value_unset (&arr);
11609               g_free (comp_map);
11610             }
11611
11612             if (chan_def) {
11613               GValue arr = { 0, };
11614               GValue elt = { 0, };
11615               int i;
11616               g_value_init (&arr, GST_TYPE_ARRAY);
11617               g_value_init (&elt, G_TYPE_INT);
11618               for (i = 0; i < nchan_def; i++) {
11619                 g_value_set_int (&elt, chan_def[i]);
11620                 gst_value_array_append_value (&arr, &elt);
11621               }
11622               gst_structure_set_value (gst_caps_get_structure (entry->caps, 0),
11623                   "channel-definitions", &arr);
11624               g_value_unset (&elt);
11625               g_value_unset (&arr);
11626               g_free (chan_def);
11627             }
11628
11629             /* some optional atoms */
11630             field = qtdemux_tree_get_child_by_type (mjp2, FOURCC_fiel);
11631             prefix = qtdemux_tree_get_child_by_type (mjp2, FOURCC_jp2x);
11632
11633             /* indicate possible fields in caps */
11634             if (field) {
11635               data = (guint8 *) field->data + 8;
11636               if (*data != 1)
11637                 gst_caps_set_simple (entry->caps, "fields", G_TYPE_INT,
11638                     (gint) * data, NULL);
11639             }
11640             /* add codec_data if provided */
11641             if (prefix) {
11642               GstBuffer *buf;
11643               gint len;
11644
11645               GST_DEBUG_OBJECT (qtdemux, "found prefix data in stsd");
11646               data = prefix->data;
11647               len = QT_UINT32 (data);
11648               if (len > 0x8) {
11649                 len -= 0x8;
11650                 buf = gst_buffer_new_and_alloc (len);
11651                 gst_buffer_fill (buf, 0, data + 8, len);
11652                 gst_caps_set_simple (entry->caps,
11653                     "codec_data", GST_TYPE_BUFFER, buf, NULL);
11654                 gst_buffer_unref (buf);
11655               }
11656             }
11657             break;
11658           }
11659           case FOURCC_SVQ3:
11660           case FOURCC_VP31:
11661           {
11662             GstBuffer *buf;
11663             GstBuffer *seqh = NULL;
11664             const guint8 *gamma_data = NULL;
11665             gint len = QT_UINT32 (stsd_data);   /* FIXME review - why put the whole stsd in codec data? */
11666
11667             qtdemux_parse_svq3_stsd_data (qtdemux, stsd_entry_data, &gamma_data,
11668                 &seqh);
11669             if (gamma_data) {
11670               gst_caps_set_simple (entry->caps, "applied-gamma", G_TYPE_DOUBLE,
11671                   QT_FP32 (gamma_data), NULL);
11672             }
11673             if (seqh) {
11674               /* sorry for the bad name, but we don't know what this is, other
11675                * than its own fourcc */
11676               gst_caps_set_simple (entry->caps, "seqh", GST_TYPE_BUFFER, seqh,
11677                   NULL);
11678               gst_buffer_unref (seqh);
11679             }
11680
11681             GST_DEBUG_OBJECT (qtdemux, "found codec_data in stsd");
11682             buf = gst_buffer_new_and_alloc (len);
11683             gst_buffer_fill (buf, 0, stsd_data, len);
11684             gst_caps_set_simple (entry->caps,
11685                 "codec_data", GST_TYPE_BUFFER, buf, NULL);
11686             gst_buffer_unref (buf);
11687             break;
11688           }
11689           case FOURCC_jpeg:
11690           {
11691             /* https://developer.apple.com/standards/qtff-2001.pdf,
11692              * page 92, "Video Sample Description", under table 3.1 */
11693             GstByteReader br;
11694
11695             const gint compressor_offset =
11696                 16 + 4 + 4 * 3 + 2 * 2 + 2 * 4 + 4 + 2;
11697             const gint min_size = compressor_offset + 32 + 2 + 2;
11698             GNode *jpeg;
11699             guint32 len;
11700             guint16 color_table_id = 0;
11701             gboolean ok;
11702
11703             GST_DEBUG_OBJECT (qtdemux, "found jpeg");
11704
11705             /* recover information on interlaced/progressive */
11706             jpeg = qtdemux_tree_get_child_by_type (stsd, FOURCC_jpeg);
11707             if (!jpeg)
11708               break;
11709
11710             len = QT_UINT32 (jpeg->data);
11711             GST_DEBUG_OBJECT (qtdemux, "Found jpeg: len %u, need %d", len,
11712                 min_size);
11713             if (len >= min_size) {
11714               gst_byte_reader_init (&br, jpeg->data, len);
11715
11716               gst_byte_reader_skip (&br, compressor_offset + 32 + 2);
11717               gst_byte_reader_get_uint16_le (&br, &color_table_id);
11718               if (color_table_id != 0) {
11719                 /* the spec says there can be concatenated chunks in the data, and we want
11720                  * to find one called field. Walk through them. */
11721                 gint offset = min_size;
11722                 while (offset + 8 < len) {
11723                   guint32 size = 0, tag;
11724                   ok = gst_byte_reader_get_uint32_le (&br, &size);
11725                   ok &= gst_byte_reader_get_uint32_le (&br, &tag);
11726                   if (!ok || size < 8) {
11727                     GST_WARNING_OBJECT (qtdemux,
11728                         "Failed to walk optional chunk list");
11729                     break;
11730                   }
11731                   GST_DEBUG_OBJECT (qtdemux,
11732                       "Found optional %4.4s chunk, size %u",
11733                       (const char *) &tag, size);
11734                   if (tag == FOURCC_fiel) {
11735                     guint8 n_fields = 0, ordering = 0;
11736                     gst_byte_reader_get_uint8 (&br, &n_fields);
11737                     gst_byte_reader_get_uint8 (&br, &ordering);
11738                     if (n_fields == 1 || n_fields == 2) {
11739                       GST_DEBUG_OBJECT (qtdemux,
11740                           "Found fiel tag with %u fields, ordering %u",
11741                           n_fields, ordering);
11742                       if (n_fields == 2)
11743                         gst_caps_set_simple (CUR_STREAM (stream)->caps,
11744                             "interlace-mode", G_TYPE_STRING, "interleaved",
11745                             NULL);
11746                     } else {
11747                       GST_WARNING_OBJECT (qtdemux,
11748                           "Found fiel tag with invalid fields (%u)", n_fields);
11749                     }
11750                   }
11751                   offset += size;
11752                 }
11753               } else {
11754                 GST_DEBUG_OBJECT (qtdemux,
11755                     "Color table ID is 0, not trying to get interlacedness");
11756               }
11757             } else {
11758               GST_WARNING_OBJECT (qtdemux,
11759                   "Length of jpeg chunk is too small, not trying to get interlacedness");
11760             }
11761
11762             break;
11763           }
11764           case FOURCC_rle_:
11765           case FOURCC_WRLE:
11766           {
11767             gst_caps_set_simple (entry->caps,
11768                 "depth", G_TYPE_INT, QT_UINT16 (stsd_entry_data + offset + 66),
11769                 NULL);
11770             break;
11771           }
11772           case FOURCC_XiTh:
11773           {
11774             GNode *xith, *xdxt;
11775
11776             GST_DEBUG_OBJECT (qtdemux, "found XiTh");
11777             xith = qtdemux_tree_get_child_by_index (stsd, stsd_index);
11778             if (!xith)
11779               break;
11780
11781             xdxt = qtdemux_tree_get_child_by_type (xith, FOURCC_XdxT);
11782             if (!xdxt)
11783               break;
11784
11785             GST_DEBUG_OBJECT (qtdemux, "found XdxT node");
11786             /* collect the headers and store them in a stream list so that we can
11787              * send them out first */
11788             qtdemux_parse_theora_extension (qtdemux, stream, xdxt);
11789             break;
11790           }
11791           case FOURCC_ovc1:
11792           {
11793             GNode *ovc1;
11794             guint8 *ovc1_data;
11795             guint ovc1_len;
11796             GstBuffer *buf;
11797
11798             GST_DEBUG_OBJECT (qtdemux, "parse ovc1 header");
11799             ovc1 = qtdemux_tree_get_child_by_index (stsd, stsd_index);
11800             if (!ovc1)
11801               break;
11802             ovc1_data = ovc1->data;
11803             ovc1_len = QT_UINT32 (ovc1_data);
11804             if (ovc1_len <= 198) {
11805               GST_WARNING_OBJECT (qtdemux, "Too small ovc1 header, skipping");
11806               break;
11807             }
11808             buf = gst_buffer_new_and_alloc (ovc1_len - 198);
11809             gst_buffer_fill (buf, 0, ovc1_data + 198, ovc1_len - 198);
11810             gst_caps_set_simple (entry->caps,
11811                 "codec_data", GST_TYPE_BUFFER, buf, NULL);
11812             gst_buffer_unref (buf);
11813             break;
11814           }
11815           case FOURCC_vc_1:
11816           {
11817             gint len = QT_UINT32 (stsd_entry_data) - 0x56;
11818             const guint8 *vc1_data = stsd_entry_data + 0x56;
11819
11820             /* find dvc1 */
11821             while (len >= 8) {
11822               gint size;
11823
11824               if (QT_UINT32 (vc1_data) <= len)
11825                 size = QT_UINT32 (vc1_data) - 8;
11826               else
11827                 size = len - 8;
11828
11829               if (size < 1)
11830                 /* No real data, so break out */
11831                 break;
11832
11833               switch (QT_FOURCC (vc1_data + 0x4)) {
11834                 case GST_MAKE_FOURCC ('d', 'v', 'c', '1'):
11835                 {
11836                   GstBuffer *buf;
11837
11838                   GST_DEBUG_OBJECT (qtdemux, "found dvc1 codec_data in stsd");
11839                   buf = gst_buffer_new_and_alloc (size);
11840                   gst_buffer_fill (buf, 0, vc1_data + 8, size);
11841                   gst_caps_set_simple (entry->caps,
11842                       "codec_data", GST_TYPE_BUFFER, buf, NULL);
11843                   gst_buffer_unref (buf);
11844                   break;
11845                 }
11846                 default:
11847                   break;
11848               }
11849               len -= size + 8;
11850               vc1_data += size + 8;
11851             }
11852             break;
11853           }
11854           case FOURCC_av01:
11855           {
11856             gint len = QT_UINT32 (stsd_entry_data) - 0x56;
11857             const guint8 *av1_data = stsd_entry_data + 0x56;
11858
11859             /* find av1C */
11860             while (len >= 0x8) {
11861               gint size;
11862
11863               if (QT_UINT32 (av1_data) <= len)
11864                 size = QT_UINT32 (av1_data) - 0x8;
11865               else
11866                 size = len - 0x8;
11867
11868               if (size < 1)
11869                 /* No real data, so break out */
11870                 break;
11871
11872               switch (QT_FOURCC (av1_data + 0x4)) {
11873                 case FOURCC_av1C:
11874                 {
11875                   /* parse, if found */
11876                   GstBuffer *buf;
11877                   guint8 pres_delay_field;
11878
11879                   GST_DEBUG_OBJECT (qtdemux,
11880                       "found av1C codec_data in stsd of size %d", size);
11881
11882                   /* not enough data, just ignore and hope for the best */
11883                   if (size < 5)
11884                     break;
11885
11886                   /* Content is:
11887                    * 4 bytes: atom length
11888                    * 4 bytes: fourcc
11889                    * 1 byte: version
11890                    * 3 bytes: flags
11891                    * 3 bits: reserved
11892                    * 1 bits:  initial_presentation_delay_present
11893                    * 4 bits: initial_presentation_delay (if present else reserved
11894                    * rest: OBUs.
11895                    */
11896
11897                   if (av1_data[9] != 0) {
11898                     GST_WARNING ("Unknown version %d of av1C box", av1_data[9]);
11899                     break;
11900                   }
11901
11902                   /* We skip initial_presentation_delay* for now */
11903                   pres_delay_field = *(av1_data + 12);
11904                   if (pres_delay_field & (1 << 5)) {
11905                     gst_caps_set_simple (entry->caps,
11906                         "presentation-delay", G_TYPE_INT,
11907                         (gint) (pres_delay_field & 0x0F) + 1, NULL);
11908                   }
11909                   if (size > 5) {
11910                     buf = gst_buffer_new_and_alloc (size - 5);
11911                     GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_HEADER);
11912                     gst_buffer_fill (buf, 0, av1_data + 13, size - 5);
11913                     gst_caps_set_simple (entry->caps,
11914                         "codec_data", GST_TYPE_BUFFER, buf, NULL);
11915                     gst_buffer_unref (buf);
11916                   }
11917                   break;
11918                 }
11919                 default:
11920                   break;
11921               }
11922
11923               len -= size + 8;
11924               av1_data += size + 8;
11925             }
11926
11927             break;
11928           }
11929
11930             /* TODO: Need to parse vpcC for VP8 codec too.
11931              * Note that VPCodecConfigurationBox (vpcC) is defined for
11932              * vp08, vp09, and vp10 fourcc. */
11933           case FOURCC_vp09:
11934           {
11935             gint len = QT_UINT32 (stsd_entry_data) - 0x56;
11936             const guint8 *vpcc_data = stsd_entry_data + 0x56;
11937
11938             /* find vpcC */
11939             while (len >= 0x8) {
11940               gint size;
11941
11942               if (QT_UINT32 (vpcc_data) <= len)
11943                 size = QT_UINT32 (vpcc_data) - 0x8;
11944               else
11945                 size = len - 0x8;
11946
11947               if (size < 1)
11948                 /* No real data, so break out */
11949                 break;
11950
11951               switch (QT_FOURCC (vpcc_data + 0x4)) {
11952                 case FOURCC_vpcC:
11953                 {
11954                   const gchar *profile_str = NULL;
11955                   const gchar *chroma_format_str = NULL;
11956                   guint8 profile;
11957                   guint8 bitdepth;
11958                   guint8 chroma_format;
11959                   GstVideoColorimetry cinfo;
11960
11961                   /* parse, if found */
11962                   GST_DEBUG_OBJECT (qtdemux,
11963                       "found vp codec_data in stsd of size %d", size);
11964
11965                   /* the meaning of "size" is length of the atom body, excluding
11966                    * atom length and fourcc fields */
11967                   if (size < 12)
11968                     break;
11969
11970                   /* Content is:
11971                    * 4 bytes: atom length
11972                    * 4 bytes: fourcc
11973                    * 1 byte: version
11974                    * 3 bytes: flags
11975                    * 1 byte: profile
11976                    * 1 byte: level
11977                    * 4 bits: bitDepth
11978                    * 3 bits: chromaSubsampling
11979                    * 1 bit: videoFullRangeFlag
11980                    * 1 byte: colourPrimaries
11981                    * 1 byte: transferCharacteristics
11982                    * 1 byte: matrixCoefficients
11983                    * 2 bytes: codecIntializationDataSize (should be zero for vp8 and vp9)
11984                    * rest: codecIntializationData (not used for vp8 and vp9)
11985                    */
11986
11987                   if (vpcc_data[8] != 1) {
11988                     GST_WARNING_OBJECT (qtdemux,
11989                         "unknown vpcC version %d", vpcc_data[8]);
11990                     break;
11991                   }
11992
11993                   profile = vpcc_data[12];
11994                   switch (profile) {
11995                     case 0:
11996                       profile_str = "0";
11997                       break;
11998                     case 1:
11999                       profile_str = "1";
12000                       break;
12001                     case 2:
12002                       profile_str = "2";
12003                       break;
12004                     case 3:
12005                       profile_str = "3";
12006                       break;
12007                     default:
12008                       break;
12009                   }
12010
12011                   if (profile_str) {
12012                     gst_caps_set_simple (entry->caps,
12013                         "profile", G_TYPE_STRING, profile_str, NULL);
12014                   }
12015
12016                   /* skip level, the VP9 spec v0.6 defines only one level atm,
12017                    * but webm spec define various ones. Add level to caps
12018                    * if we really need it then */
12019
12020                   bitdepth = (vpcc_data[14] & 0xf0) >> 4;
12021                   if (bitdepth == 8 || bitdepth == 10 || bitdepth == 12) {
12022                     gst_caps_set_simple (entry->caps,
12023                         "bit-depth-luma", G_TYPE_UINT, bitdepth,
12024                         "bit-depth-chroma", G_TYPE_UINT, bitdepth, NULL);
12025                   }
12026
12027                   chroma_format = (vpcc_data[14] & 0xe) >> 1;
12028                   switch (chroma_format) {
12029                     case 0:
12030                     case 1:
12031                       chroma_format_str = "4:2:0";
12032                       break;
12033                     case 2:
12034                       chroma_format_str = "4:2:2";
12035                       break;
12036                     case 3:
12037                       chroma_format_str = "4:4:4";
12038                       break;
12039                     default:
12040                       break;
12041                   }
12042
12043                   if (chroma_format_str) {
12044                     gst_caps_set_simple (entry->caps,
12045                         "chroma-format", G_TYPE_STRING, chroma_format_str,
12046                         NULL);
12047                   }
12048
12049                   if ((vpcc_data[14] & 0x1) != 0)
12050                     cinfo.range = GST_VIDEO_COLOR_RANGE_0_255;
12051                   else
12052                     cinfo.range = GST_VIDEO_COLOR_RANGE_16_235;
12053                   cinfo.primaries =
12054                       gst_video_color_primaries_from_iso (vpcc_data[15]);
12055                   cinfo.transfer =
12056                       gst_video_transfer_function_from_iso (vpcc_data[16]);
12057                   cinfo.matrix =
12058                       gst_video_color_matrix_from_iso (vpcc_data[17]);
12059
12060                   if (cinfo.primaries != GST_VIDEO_COLOR_PRIMARIES_UNKNOWN &&
12061                       cinfo.transfer != GST_VIDEO_TRANSFER_UNKNOWN &&
12062                       cinfo.matrix != GST_VIDEO_COLOR_MATRIX_UNKNOWN) {
12063                     /* set this only if all values are known, otherwise this
12064                      * might overwrite valid ones parsed from other color box */
12065                     CUR_STREAM (stream)->colorimetry = cinfo;
12066                   }
12067                   break;
12068                 }
12069                 default:
12070                   break;
12071               }
12072
12073               len -= size + 8;
12074               vpcc_data += size + 8;
12075             }
12076
12077             break;
12078           }
12079           default:
12080             break;
12081         }
12082       }
12083
12084       GST_INFO_OBJECT (qtdemux,
12085           "type %" GST_FOURCC_FORMAT " caps %" GST_PTR_FORMAT,
12086           GST_FOURCC_ARGS (fourcc), entry->caps);
12087
12088     } else if (stream->subtype == FOURCC_soun) {
12089       GNode *wave;
12090       int version, samplesize;
12091       guint16 compression_id;
12092       gboolean amrwb = FALSE;
12093
12094       offset = 16;
12095       /* sample description entry (16) + sound sample description v0 (20) */
12096       if (len < 36)
12097         goto corrupt_file;
12098
12099       version = QT_UINT32 (stsd_entry_data + offset);
12100       entry->n_channels = QT_UINT16 (stsd_entry_data + offset + 8);
12101       samplesize = QT_UINT16 (stsd_entry_data + offset + 10);
12102       compression_id = QT_UINT16 (stsd_entry_data + offset + 12);
12103       entry->rate = QT_FP32 (stsd_entry_data + offset + 16);
12104
12105       GST_LOG_OBJECT (qtdemux, "version/rev:      %08x", version);
12106       GST_LOG_OBJECT (qtdemux, "vendor:           %08x",
12107           QT_UINT32 (stsd_entry_data + offset + 4));
12108       GST_LOG_OBJECT (qtdemux, "n_channels:       %d", entry->n_channels);
12109       GST_LOG_OBJECT (qtdemux, "sample_size:      %d", samplesize);
12110       GST_LOG_OBJECT (qtdemux, "compression_id:   %d", compression_id);
12111       GST_LOG_OBJECT (qtdemux, "packet size:      %d",
12112           QT_UINT16 (stsd_entry_data + offset + 14));
12113       GST_LOG_OBJECT (qtdemux, "sample rate:      %g", entry->rate);
12114
12115       if (compression_id == 0xfffe)
12116         entry->sampled = TRUE;
12117
12118       /* first assume uncompressed audio */
12119       entry->bytes_per_sample = samplesize / 8;
12120       entry->samples_per_frame = entry->n_channels;
12121       entry->bytes_per_frame = entry->n_channels * entry->bytes_per_sample;
12122       entry->samples_per_packet = entry->samples_per_frame;
12123       entry->bytes_per_packet = entry->bytes_per_sample;
12124
12125       offset = 36;
12126
12127       if (version == 0x00010000) {
12128         /* sample description entry (16) + sound sample description v1 (20+16) */
12129         if (len < 52)
12130           goto corrupt_file;
12131
12132         /* take information from here over the normal sample description */
12133         entry->samples_per_packet = QT_UINT32 (stsd_entry_data + offset);
12134         entry->bytes_per_packet = QT_UINT32 (stsd_entry_data + offset + 4);
12135         entry->bytes_per_frame = QT_UINT32 (stsd_entry_data + offset + 8);
12136         entry->bytes_per_sample = QT_UINT32 (stsd_entry_data + offset + 12);
12137
12138         GST_LOG_OBJECT (qtdemux, "Sound sample description Version 1");
12139         GST_LOG_OBJECT (qtdemux, "samples/packet:   %d",
12140             entry->samples_per_packet);
12141         GST_LOG_OBJECT (qtdemux, "bytes/packet:     %d",
12142             entry->bytes_per_packet);
12143         GST_LOG_OBJECT (qtdemux, "bytes/frame:      %d",
12144             entry->bytes_per_frame);
12145         GST_LOG_OBJECT (qtdemux, "bytes/sample:     %d",
12146             entry->bytes_per_sample);
12147
12148         if (!entry->sampled && entry->bytes_per_packet) {
12149           entry->samples_per_frame = (entry->bytes_per_frame /
12150               entry->bytes_per_packet) * entry->samples_per_packet;
12151           GST_LOG_OBJECT (qtdemux, "samples/frame:    %d",
12152               entry->samples_per_frame);
12153         }
12154       } else if (version == 0x00020000) {
12155         /* sample description entry (16) + sound sample description v2 (56) */
12156         if (len < 72)
12157           goto corrupt_file;
12158
12159         /* take information from here over the normal sample description */
12160         entry->rate = GST_READ_DOUBLE_BE (stsd_entry_data + offset + 4);
12161         entry->n_channels = QT_UINT32 (stsd_entry_data + offset + 12);
12162         entry->samples_per_frame = entry->n_channels;
12163         entry->bytes_per_sample = QT_UINT32 (stsd_entry_data + offset + 20) / 8;
12164         entry->bytes_per_packet = QT_UINT32 (stsd_entry_data + offset + 28);
12165         entry->samples_per_packet = QT_UINT32 (stsd_entry_data + offset + 32);
12166         entry->bytes_per_frame = entry->bytes_per_sample * entry->n_channels;
12167
12168         GST_LOG_OBJECT (qtdemux, "Sound sample description Version 2");
12169         GST_LOG_OBJECT (qtdemux, "sample rate:        %g", entry->rate);
12170         GST_LOG_OBJECT (qtdemux, "n_channels:         %d", entry->n_channels);
12171         GST_LOG_OBJECT (qtdemux, "bits/channel:       %d",
12172             entry->bytes_per_sample * 8);
12173         GST_LOG_OBJECT (qtdemux, "format flags:       %X",
12174             QT_UINT32 (stsd_entry_data + offset + 24));
12175         GST_LOG_OBJECT (qtdemux, "bytes/packet:       %d",
12176             entry->bytes_per_packet);
12177         GST_LOG_OBJECT (qtdemux, "LPCM frames/packet: %d",
12178             entry->samples_per_packet);
12179       } else if (version != 0x00000) {
12180         GST_WARNING_OBJECT (qtdemux, "unknown audio STSD version %08x",
12181             version);
12182       }
12183
12184       switch (fourcc) {
12185           /* Yes, these have to be hard-coded */
12186         case FOURCC_MAC6:
12187         {
12188           entry->samples_per_packet = 6;
12189           entry->bytes_per_packet = 1;
12190           entry->bytes_per_frame = 1 * entry->n_channels;
12191           entry->bytes_per_sample = 1;
12192           entry->samples_per_frame = 6 * entry->n_channels;
12193           break;
12194         }
12195         case FOURCC_MAC3:
12196         {
12197           entry->samples_per_packet = 3;
12198           entry->bytes_per_packet = 1;
12199           entry->bytes_per_frame = 1 * entry->n_channels;
12200           entry->bytes_per_sample = 1;
12201           entry->samples_per_frame = 3 * entry->n_channels;
12202           break;
12203         }
12204         case FOURCC_ima4:
12205         {
12206           entry->samples_per_packet = 64;
12207           entry->bytes_per_packet = 34;
12208           entry->bytes_per_frame = 34 * entry->n_channels;
12209           entry->bytes_per_sample = 2;
12210           entry->samples_per_frame = 64 * entry->n_channels;
12211           break;
12212         }
12213         case FOURCC_ulaw:
12214         case FOURCC_alaw:
12215         {
12216           entry->samples_per_packet = 1;
12217           entry->bytes_per_packet = 1;
12218           entry->bytes_per_frame = 1 * entry->n_channels;
12219           entry->bytes_per_sample = 1;
12220           entry->samples_per_frame = 1 * entry->n_channels;
12221           break;
12222         }
12223         case FOURCC_agsm:
12224         {
12225           entry->samples_per_packet = 160;
12226           entry->bytes_per_packet = 33;
12227           entry->bytes_per_frame = 33 * entry->n_channels;
12228           entry->bytes_per_sample = 2;
12229           entry->samples_per_frame = 160 * entry->n_channels;
12230           break;
12231         }
12232           /* fix up any invalid header information from above */
12233         case FOURCC_twos:
12234         case FOURCC_sowt:
12235         case FOURCC_raw_:
12236         case FOURCC_lpcm:
12237           /* Sometimes these are set to 0 in the sound sample descriptions so
12238            * let's try to infer useful values from the other information we
12239            * have available */
12240           if (entry->bytes_per_sample == 0)
12241             entry->bytes_per_sample =
12242                 entry->bytes_per_frame / entry->n_channels;
12243           if (entry->bytes_per_sample == 0)
12244             entry->bytes_per_sample = samplesize / 8;
12245
12246           if (entry->bytes_per_frame == 0)
12247             entry->bytes_per_frame =
12248                 entry->bytes_per_sample * entry->n_channels;
12249
12250           if (entry->bytes_per_packet == 0)
12251             entry->bytes_per_packet = entry->bytes_per_sample;
12252
12253           if (entry->samples_per_frame == 0)
12254             entry->samples_per_frame = entry->n_channels;
12255
12256           if (entry->samples_per_packet == 0)
12257             entry->samples_per_packet = entry->samples_per_frame;
12258
12259           break;
12260         case FOURCC_in24:
12261         case FOURCC_in32:
12262         case FOURCC_fl32:
12263         case FOURCC_fl64:
12264         case FOURCC_s16l:{
12265           switch (fourcc) {
12266             case FOURCC_in24:
12267               entry->bytes_per_sample = 3;
12268               break;
12269             case FOURCC_in32:
12270             case FOURCC_fl32:
12271               entry->bytes_per_sample = 4;
12272               break;
12273             case FOURCC_fl64:
12274               entry->bytes_per_sample = 8;
12275               break;
12276             case FOURCC_s16l:
12277               entry->bytes_per_sample = 2;
12278               break;
12279             default:
12280               g_assert_not_reached ();
12281               break;
12282           }
12283           entry->samples_per_frame = entry->n_channels;
12284           entry->bytes_per_frame = entry->n_channels * entry->bytes_per_sample;
12285           entry->samples_per_packet = entry->samples_per_frame;
12286           entry->bytes_per_packet = entry->bytes_per_sample;
12287           break;
12288         }
12289         default:
12290           break;
12291       }
12292
12293       if (entry->caps)
12294         gst_caps_unref (entry->caps);
12295
12296       entry->caps = qtdemux_audio_caps (qtdemux, stream, entry, fourcc,
12297           stsd_entry_data + 32, len - 16, &codec);
12298
12299       switch (fourcc) {
12300         case FOURCC_in24:
12301         case FOURCC_in32:
12302         case FOURCC_fl32:
12303         case FOURCC_fl64:
12304         {
12305           GNode *enda;
12306           GNode *fmt;
12307
12308           fmt = qtdemux_tree_get_child_by_type (stsd, fourcc);
12309
12310           enda = qtdemux_tree_get_child_by_type (fmt, FOURCC_enda);
12311           if (!enda) {
12312             wave = qtdemux_tree_get_child_by_type (fmt, FOURCC_wave);
12313             if (wave)
12314               enda = qtdemux_tree_get_child_by_type (wave, FOURCC_enda);
12315           }
12316           if (enda) {
12317             int enda_value = QT_UINT16 ((guint8 *) enda->data + 8);
12318             const gchar *format_str;
12319
12320             switch (fourcc) {
12321               case FOURCC_in24:
12322                 format_str = (enda_value) ? "S24LE" : "S24BE";
12323                 break;
12324               case FOURCC_in32:
12325                 format_str = (enda_value) ? "S32LE" : "S32BE";
12326                 break;
12327               case FOURCC_fl32:
12328                 format_str = (enda_value) ? "F32LE" : "F32BE";
12329                 break;
12330               case FOURCC_fl64:
12331                 format_str = (enda_value) ? "F64LE" : "F64BE";
12332                 break;
12333               default:
12334                 g_assert_not_reached ();
12335                 break;
12336             }
12337             gst_caps_set_simple (entry->caps,
12338                 "format", G_TYPE_STRING, format_str, NULL);
12339           }
12340           break;
12341         }
12342         case FOURCC_owma:
12343         {
12344           const guint8 *owma_data;
12345           const gchar *codec_name = NULL;
12346           guint owma_len;
12347           GstBuffer *buf;
12348           gint version = 1;
12349           /* from http://msdn.microsoft.com/en-us/library/dd757720(VS.85).aspx */
12350           /* FIXME this should also be gst_riff_strf_auds,
12351            * but the latter one is actually missing bits-per-sample :( */
12352           typedef struct
12353           {
12354             gint16 wFormatTag;
12355             gint16 nChannels;
12356             gint32 nSamplesPerSec;
12357             gint32 nAvgBytesPerSec;
12358             gint16 nBlockAlign;
12359             gint16 wBitsPerSample;
12360             gint16 cbSize;
12361           } WAVEFORMATEX;
12362           WAVEFORMATEX *wfex;
12363
12364           GST_DEBUG_OBJECT (qtdemux, "parse owma");
12365           owma_data = stsd_entry_data;
12366           owma_len = QT_UINT32 (owma_data);
12367           if (owma_len <= 54) {
12368             GST_WARNING_OBJECT (qtdemux, "Too small owma header, skipping");
12369             break;
12370           }
12371           wfex = (WAVEFORMATEX *) (owma_data + 36);
12372           buf = gst_buffer_new_and_alloc (owma_len - 54);
12373           gst_buffer_fill (buf, 0, owma_data + 54, owma_len - 54);
12374           if (wfex->wFormatTag == 0x0161) {
12375             codec_name = "Windows Media Audio";
12376             version = 2;
12377           } else if (wfex->wFormatTag == 0x0162) {
12378             codec_name = "Windows Media Audio 9 Pro";
12379             version = 3;
12380           } else if (wfex->wFormatTag == 0x0163) {
12381             codec_name = "Windows Media Audio 9 Lossless";
12382             /* is that correct? gstffmpegcodecmap.c is missing it, but
12383              * fluendo codec seems to support it */
12384             version = 4;
12385           }
12386
12387           gst_caps_set_simple (entry->caps,
12388               "codec_data", GST_TYPE_BUFFER, buf,
12389               "wmaversion", G_TYPE_INT, version,
12390               "block_align", G_TYPE_INT,
12391               GST_READ_UINT16_LE (&wfex->nBlockAlign), "bitrate", G_TYPE_INT,
12392               GST_READ_UINT32_LE (&wfex->nAvgBytesPerSec), "width", G_TYPE_INT,
12393               GST_READ_UINT16_LE (&wfex->wBitsPerSample), "depth", G_TYPE_INT,
12394               GST_READ_UINT16_LE (&wfex->wBitsPerSample), NULL);
12395           gst_buffer_unref (buf);
12396
12397           if (codec_name) {
12398             g_free (codec);
12399             codec = g_strdup (codec_name);
12400           }
12401           break;
12402         }
12403         case FOURCC_wma_:
12404         {
12405           gint len = QT_UINT32 (stsd_entry_data) - offset;
12406           const guint8 *wfex_data = stsd_entry_data + offset;
12407           const gchar *codec_name = NULL;
12408           gint version = 1;
12409           /* from http://msdn.microsoft.com/en-us/library/dd757720(VS.85).aspx */
12410           /* FIXME this should also be gst_riff_strf_auds,
12411            * but the latter one is actually missing bits-per-sample :( */
12412           typedef struct
12413           {
12414             gint16 wFormatTag;
12415             gint16 nChannels;
12416             gint32 nSamplesPerSec;
12417             gint32 nAvgBytesPerSec;
12418             gint16 nBlockAlign;
12419             gint16 wBitsPerSample;
12420             gint16 cbSize;
12421           } WAVEFORMATEX;
12422           WAVEFORMATEX wfex;
12423
12424           /* FIXME: unify with similar wavformatex parsing code above */
12425           GST_DEBUG_OBJECT (qtdemux, "parse wma, looking for wfex");
12426
12427           /* find wfex */
12428           while (len >= 8) {
12429             gint size;
12430
12431             if (QT_UINT32 (wfex_data) <= len)
12432               size = QT_UINT32 (wfex_data) - 8;
12433             else
12434               size = len - 8;
12435
12436             if (size < 1)
12437               /* No real data, so break out */
12438               break;
12439
12440             switch (QT_FOURCC (wfex_data + 4)) {
12441               case GST_MAKE_FOURCC ('w', 'f', 'e', 'x'):
12442               {
12443                 GST_DEBUG_OBJECT (qtdemux, "found wfex in stsd");
12444
12445                 if (size < 8 + 18)
12446                   break;
12447
12448                 wfex.wFormatTag = GST_READ_UINT16_LE (wfex_data + 8 + 0);
12449                 wfex.nChannels = GST_READ_UINT16_LE (wfex_data + 8 + 2);
12450                 wfex.nSamplesPerSec = GST_READ_UINT32_LE (wfex_data + 8 + 4);
12451                 wfex.nAvgBytesPerSec = GST_READ_UINT32_LE (wfex_data + 8 + 8);
12452                 wfex.nBlockAlign = GST_READ_UINT16_LE (wfex_data + 8 + 12);
12453                 wfex.wBitsPerSample = GST_READ_UINT16_LE (wfex_data + 8 + 14);
12454                 wfex.cbSize = GST_READ_UINT16_LE (wfex_data + 8 + 16);
12455
12456                 GST_LOG_OBJECT (qtdemux, "Found wfex box in stsd:");
12457                 GST_LOG_OBJECT (qtdemux, "FormatTag = 0x%04x, Channels = %u, "
12458                     "SamplesPerSec = %u, AvgBytesPerSec = %u, BlockAlign = %u, "
12459                     "BitsPerSample = %u, Size = %u", wfex.wFormatTag,
12460                     wfex.nChannels, wfex.nSamplesPerSec, wfex.nAvgBytesPerSec,
12461                     wfex.nBlockAlign, wfex.wBitsPerSample, wfex.cbSize);
12462
12463                 if (wfex.wFormatTag == 0x0161) {
12464                   codec_name = "Windows Media Audio";
12465                   version = 2;
12466                 } else if (wfex.wFormatTag == 0x0162) {
12467                   codec_name = "Windows Media Audio 9 Pro";
12468                   version = 3;
12469                 } else if (wfex.wFormatTag == 0x0163) {
12470                   codec_name = "Windows Media Audio 9 Lossless";
12471                   /* is that correct? gstffmpegcodecmap.c is missing it, but
12472                    * fluendo codec seems to support it */
12473                   version = 4;
12474                 }
12475
12476                 gst_caps_set_simple (entry->caps,
12477                     "wmaversion", G_TYPE_INT, version,
12478                     "block_align", G_TYPE_INT, wfex.nBlockAlign,
12479                     "bitrate", G_TYPE_INT, wfex.nAvgBytesPerSec,
12480                     "width", G_TYPE_INT, wfex.wBitsPerSample,
12481                     "depth", G_TYPE_INT, wfex.wBitsPerSample, NULL);
12482
12483                 if (size > wfex.cbSize) {
12484                   GstBuffer *buf;
12485
12486                   buf = gst_buffer_new_and_alloc (size - wfex.cbSize);
12487                   gst_buffer_fill (buf, 0, wfex_data + 8 + wfex.cbSize,
12488                       size - wfex.cbSize);
12489                   gst_caps_set_simple (entry->caps,
12490                       "codec_data", GST_TYPE_BUFFER, buf, NULL);
12491                   gst_buffer_unref (buf);
12492                 } else {
12493                   GST_WARNING_OBJECT (qtdemux, "no codec data");
12494                 }
12495
12496                 if (codec_name) {
12497                   g_free (codec);
12498                   codec = g_strdup (codec_name);
12499                 }
12500                 break;
12501               }
12502               default:
12503                 break;
12504             }
12505             len -= size + 8;
12506             wfex_data += size + 8;
12507           }
12508           break;
12509         }
12510         case FOURCC_opus:
12511         {
12512           const guint8 *dops_data;
12513           guint8 *channel_mapping = NULL;
12514           guint32 rate;
12515           guint8 channels;
12516           guint8 channel_mapping_family;
12517           guint8 stream_count;
12518           guint8 coupled_count;
12519           guint8 i;
12520
12521           version = GST_READ_UINT16_BE (stsd_entry_data + 16);
12522           if (version == 1)
12523             dops_data = stsd_entry_data + 51;
12524           else
12525             dops_data = stsd_entry_data + 35;
12526
12527           channels = GST_READ_UINT8 (dops_data + 10);
12528           rate = GST_READ_UINT32_LE (dops_data + 13);
12529           channel_mapping_family = GST_READ_UINT8 (dops_data + 19);
12530           stream_count = GST_READ_UINT8 (dops_data + 20);
12531           coupled_count = GST_READ_UINT8 (dops_data + 21);
12532
12533           if (channels > 0) {
12534             channel_mapping = g_malloc (channels * sizeof (guint8));
12535             for (i = 0; i < channels; i++)
12536               channel_mapping[i] = GST_READ_UINT8 (dops_data + i + 22);
12537           }
12538
12539           entry->caps = gst_codec_utils_opus_create_caps (rate, channels,
12540               channel_mapping_family, stream_count, coupled_count,
12541               channel_mapping);
12542           break;
12543         }
12544         default:
12545           break;
12546       }
12547
12548       if (codec) {
12549         GstStructure *s;
12550         gint bitrate = 0;
12551
12552         gst_tag_list_add (stream->stream_tags, GST_TAG_MERGE_REPLACE,
12553             GST_TAG_AUDIO_CODEC, codec, NULL);
12554         g_free (codec);
12555         codec = NULL;
12556
12557         /* some bitrate info may have ended up in caps */
12558         s = gst_caps_get_structure (entry->caps, 0);
12559         gst_structure_get_int (s, "bitrate", &bitrate);
12560         if (bitrate > 0)
12561           gst_tag_list_add (stream->stream_tags, GST_TAG_MERGE_REPLACE,
12562               GST_TAG_BITRATE, bitrate, NULL);
12563       }
12564
12565       esds = NULL;
12566       mp4a = qtdemux_tree_get_child_by_index (stsd, stsd_index);
12567       if (QTDEMUX_TREE_NODE_FOURCC (mp4a) != fourcc) {
12568         if (stream->protected) {
12569           if (QTDEMUX_TREE_NODE_FOURCC (mp4a) == FOURCC_aavd) {
12570             esds = qtdemux_tree_get_child_by_type (mp4a, FOURCC_esds);
12571           }
12572           if (QTDEMUX_TREE_NODE_FOURCC (mp4a) != FOURCC_enca) {
12573             mp4a = NULL;
12574           }
12575         } else {
12576           mp4a = NULL;
12577         }
12578       }
12579
12580       wave = NULL;
12581       if (mp4a) {
12582         wave = qtdemux_tree_get_child_by_type (mp4a, FOURCC_wave);
12583         if (wave)
12584           esds = qtdemux_tree_get_child_by_type (wave, FOURCC_esds);
12585         if (!esds)
12586           esds = qtdemux_tree_get_child_by_type (mp4a, FOURCC_esds);
12587       }
12588
12589
12590       /* If the fourcc's bottom 16 bits gives 'sm', then the top
12591          16 bits is a byte-swapped wave-style codec identifier,
12592          and we can find a WAVE header internally to a 'wave' atom here.
12593          This can more clearly be thought of as 'ms' as the top 16 bits, and a
12594          codec id as the bottom 16 bits - but byte-swapped to store in QT (which
12595          is big-endian).
12596        */
12597       if ((fourcc & 0xffff) == (('s' << 8) | 'm')) {
12598         if (len < offset + 20) {
12599           GST_WARNING_OBJECT (qtdemux, "No wave atom in MS-style audio");
12600         } else {
12601           guint32 datalen = QT_UINT32 (stsd_entry_data + offset + 16);
12602           const guint8 *data = stsd_entry_data + offset + 16;
12603           GNode *wavenode;
12604           GNode *waveheadernode;
12605
12606           wavenode = g_node_new ((guint8 *) data);
12607           if (qtdemux_parse_node (qtdemux, wavenode, data, datalen)) {
12608             const guint8 *waveheader;
12609             guint32 headerlen;
12610
12611             waveheadernode = qtdemux_tree_get_child_by_type (wavenode, fourcc);
12612             if (waveheadernode) {
12613               waveheader = (const guint8 *) waveheadernode->data;
12614               headerlen = QT_UINT32 (waveheader);
12615
12616               if (headerlen > 8) {
12617                 gst_riff_strf_auds *header = NULL;
12618                 GstBuffer *headerbuf;
12619                 GstBuffer *extra;
12620
12621                 waveheader += 8;
12622                 headerlen -= 8;
12623
12624                 headerbuf = gst_buffer_new_and_alloc (headerlen);
12625                 gst_buffer_fill (headerbuf, 0, waveheader, headerlen);
12626
12627                 if (gst_riff_parse_strf_auds (GST_ELEMENT_CAST (qtdemux),
12628                         headerbuf, &header, &extra)) {
12629                   gst_caps_unref (entry->caps);
12630                   /* FIXME: Need to do something with the channel reorder map */
12631                   entry->caps =
12632                       gst_riff_create_audio_caps (header->format, NULL, header,
12633                       extra, NULL, NULL, NULL);
12634
12635                   if (extra)
12636                     gst_buffer_unref (extra);
12637                   g_free (header);
12638                 }
12639               }
12640             } else
12641               GST_DEBUG ("Didn't find waveheadernode for this codec");
12642           }
12643           g_node_destroy (wavenode);
12644         }
12645       } else if (esds) {
12646         gst_qtdemux_handle_esds (qtdemux, stream, entry, esds,
12647             stream->stream_tags);
12648       } else {
12649         switch (fourcc) {
12650 #if 0
12651             /* FIXME: what is in the chunk? */
12652           case FOURCC_QDMC:
12653           {
12654             gint len = QT_UINT32 (stsd_data);
12655
12656             /* seems to be always = 116 = 0x74 */
12657             break;
12658           }
12659 #endif
12660           case FOURCC_QDM2:
12661           {
12662             gint len = QT_UINT32 (stsd_entry_data);
12663
12664             if (len > 0x3C) {
12665               GstBuffer *buf = gst_buffer_new_and_alloc (len - 0x3C);
12666
12667               gst_buffer_fill (buf, 0, stsd_entry_data + 0x3C, len - 0x3C);
12668               gst_caps_set_simple (entry->caps,
12669                   "codec_data", GST_TYPE_BUFFER, buf, NULL);
12670               gst_buffer_unref (buf);
12671             }
12672             gst_caps_set_simple (entry->caps,
12673                 "samplesize", G_TYPE_INT, samplesize, NULL);
12674             break;
12675           }
12676           case FOURCC_alac:
12677           {
12678             GNode *alac, *wave = NULL;
12679
12680             /* apparently, m4a has this atom appended directly in the stsd entry,
12681              * while mov has it in a wave atom */
12682             alac = qtdemux_tree_get_child_by_type (stsd, FOURCC_alac);
12683             if (alac) {
12684               /* alac now refers to stsd entry atom */
12685               wave = qtdemux_tree_get_child_by_type (alac, FOURCC_wave);
12686               if (wave)
12687                 alac = qtdemux_tree_get_child_by_type (wave, FOURCC_alac);
12688               else
12689                 alac = qtdemux_tree_get_child_by_type (alac, FOURCC_alac);
12690             }
12691             if (alac) {
12692               const guint8 *alac_data = alac->data;
12693               gint len = QT_UINT32 (alac->data);
12694               GstBuffer *buf;
12695
12696               if (len < 36) {
12697                 GST_DEBUG_OBJECT (qtdemux,
12698                     "discarding alac atom with unexpected len %d", len);
12699               } else {
12700                 /* codec-data contains alac atom size and prefix,
12701                  * ffmpeg likes it that way, not quite gst-ish though ...*/
12702                 buf = gst_buffer_new_and_alloc (len);
12703                 gst_buffer_fill (buf, 0, alac->data, len);
12704                 gst_caps_set_simple (entry->caps,
12705                     "codec_data", GST_TYPE_BUFFER, buf, NULL);
12706                 gst_buffer_unref (buf);
12707
12708                 entry->bytes_per_frame = QT_UINT32 (alac_data + 12);
12709                 entry->n_channels = QT_UINT8 (alac_data + 21);
12710                 entry->rate = QT_UINT32 (alac_data + 32);
12711                 samplesize = QT_UINT8 (alac_data + 16 + 1);
12712               }
12713             }
12714             gst_caps_set_simple (entry->caps,
12715                 "samplesize", G_TYPE_INT, samplesize, NULL);
12716             break;
12717           }
12718           case FOURCC_fLaC:
12719           {
12720             /* The codingname of the sample entry is 'fLaC' */
12721             GNode *flac = qtdemux_tree_get_child_by_type (stsd, FOURCC_fLaC);
12722
12723             if (flac) {
12724               /* The 'dfLa' box is added to the sample entry to convey
12725                  initializing information for the decoder. */
12726               const GNode *dfla =
12727                   qtdemux_tree_get_child_by_type (flac, FOURCC_dfLa);
12728
12729               if (dfla) {
12730                 const guint32 len = QT_UINT32 (dfla->data);
12731
12732                 /* Must contain at least dfLa box header (12),
12733                  * METADATA_BLOCK_HEADER (4), METADATA_BLOCK_STREAMINFO (34) */
12734                 if (len < 50) {
12735                   GST_DEBUG_OBJECT (qtdemux,
12736                       "discarding dfla atom with unexpected len %d", len);
12737                 } else {
12738                   /* skip dfLa header to get the METADATA_BLOCKs */
12739                   const guint8 *metadata_blocks = (guint8 *) dfla->data + 12;
12740                   const guint32 metadata_blocks_len = len - 12;
12741
12742                   gchar *stream_marker = g_strdup ("fLaC");
12743                   GstBuffer *block = gst_buffer_new_wrapped (stream_marker,
12744                       strlen (stream_marker));
12745
12746                   guint32 index = 0;
12747                   guint32 remainder = 0;
12748                   guint32 block_size = 0;
12749                   gboolean is_last = FALSE;
12750
12751                   GValue array = G_VALUE_INIT;
12752                   GValue value = G_VALUE_INIT;
12753
12754                   g_value_init (&array, GST_TYPE_ARRAY);
12755                   g_value_init (&value, GST_TYPE_BUFFER);
12756
12757                   gst_value_set_buffer (&value, block);
12758                   gst_value_array_append_value (&array, &value);
12759                   g_value_reset (&value);
12760
12761                   gst_buffer_unref (block);
12762
12763                   /* check there's at least one METADATA_BLOCK_HEADER's worth
12764                    * of data, and we haven't already finished parsing */
12765                   while (!is_last && ((index + 3) < metadata_blocks_len)) {
12766                     remainder = metadata_blocks_len - index;
12767
12768                     /* add the METADATA_BLOCK_HEADER size to the signalled size */
12769                     block_size = 4 +
12770                         (metadata_blocks[index + 1] << 16) +
12771                         (metadata_blocks[index + 2] << 8) +
12772                         metadata_blocks[index + 3];
12773
12774                     /* be careful not to read off end of box */
12775                     if (block_size > remainder) {
12776                       break;
12777                     }
12778
12779                     is_last = metadata_blocks[index] >> 7;
12780
12781                     block = gst_buffer_new_and_alloc (block_size);
12782
12783                     gst_buffer_fill (block, 0, &metadata_blocks[index],
12784                         block_size);
12785
12786                     gst_value_set_buffer (&value, block);
12787                     gst_value_array_append_value (&array, &value);
12788                     g_value_reset (&value);
12789
12790                     gst_buffer_unref (block);
12791
12792                     index += block_size;
12793                   }
12794
12795                   /* only append the metadata if we successfully read all of it */
12796                   if (is_last) {
12797                     gst_structure_set_value (gst_caps_get_structure (CUR_STREAM
12798                             (stream)->caps, 0), "streamheader", &array);
12799                   } else {
12800                     GST_WARNING_OBJECT (qtdemux,
12801                         "discarding all METADATA_BLOCKs due to invalid "
12802                         "block_size %d at idx %d, rem %d", block_size, index,
12803                         remainder);
12804                   }
12805
12806                   g_value_unset (&value);
12807                   g_value_unset (&array);
12808
12809                   /* The sample rate obtained from the stsd may not be accurate
12810                    * since it cannot represent rates greater than 65535Hz, so
12811                    * override that value with the sample rate from the
12812                    * METADATA_BLOCK_STREAMINFO block */
12813                   CUR_STREAM (stream)->rate =
12814                       (QT_UINT32 (metadata_blocks + 14) >> 12) & 0xFFFFF;
12815                 }
12816               }
12817             }
12818             break;
12819           }
12820           case FOURCC_sawb:
12821             /* Fallthrough! */
12822             amrwb = TRUE;
12823           case FOURCC_samr:
12824           {
12825             gint len = QT_UINT32 (stsd_entry_data);
12826
12827             if (len > 0x24) {
12828               GstBuffer *buf = gst_buffer_new_and_alloc (len - 0x24);
12829               guint bitrate;
12830
12831               gst_buffer_fill (buf, 0, stsd_entry_data + 0x24, len - 0x24);
12832
12833               /* If we have enough data, let's try to get the 'damr' atom. See
12834                * the 3GPP container spec (26.244) for more details. */
12835               if ((len - 0x34) > 8 &&
12836                   (bitrate = qtdemux_parse_amr_bitrate (buf, amrwb))) {
12837                 gst_tag_list_add (stream->stream_tags, GST_TAG_MERGE_REPLACE,
12838                     GST_TAG_MAXIMUM_BITRATE, bitrate, NULL);
12839               }
12840
12841               gst_caps_set_simple (entry->caps,
12842                   "codec_data", GST_TYPE_BUFFER, buf, NULL);
12843               gst_buffer_unref (buf);
12844             }
12845             break;
12846           }
12847           case FOURCC_mp4a:
12848           {
12849             /* mp4a atom withtout ESDS; Attempt to build codec data from atom */
12850             gint len = QT_UINT32 (stsd_entry_data);
12851             guint16 sound_version = 0;
12852             /* FIXME: Can this be determined somehow? There doesn't seem to be
12853              * anything in mp4a atom that specifis compression */
12854             gint profile = 2;
12855             guint16 channels = entry->n_channels;
12856             guint32 time_scale = (guint32) entry->rate;
12857             gint sample_rate_index = -1;
12858
12859             if (len >= 34) {
12860               sound_version = QT_UINT16 (stsd_entry_data + 16);
12861
12862               if (sound_version == 1) {
12863                 channels = QT_UINT16 (stsd_entry_data + 24);
12864                 time_scale = QT_UINT32 (stsd_entry_data + 30);
12865               } else {
12866                 GST_FIXME_OBJECT (qtdemux, "Unhandled mp4a atom version %d",
12867                     sound_version);
12868               }
12869             } else {
12870               GST_DEBUG_OBJECT (qtdemux, "Too small stsd entry data len %d",
12871                   len);
12872             }
12873
12874             sample_rate_index =
12875                 gst_codec_utils_aac_get_index_from_sample_rate (time_scale);
12876             if (sample_rate_index >= 0 && channels > 0) {
12877               guint8 codec_data[2];
12878               GstBuffer *buf;
12879
12880               /* build AAC codec data */
12881               codec_data[0] = profile << 3;
12882               codec_data[0] |= ((sample_rate_index >> 1) & 0x7);
12883               codec_data[1] = (sample_rate_index & 0x01) << 7;
12884               codec_data[1] |= (channels & 0xF) << 3;
12885
12886               buf = gst_buffer_new_and_alloc (2);
12887               gst_buffer_fill (buf, 0, codec_data, 2);
12888               gst_caps_set_simple (entry->caps,
12889                   "codec_data", GST_TYPE_BUFFER, buf, NULL);
12890               gst_buffer_unref (buf);
12891             }
12892             break;
12893           }
12894           case FOURCC_lpcm:
12895           case FOURCC_in24:
12896           case FOURCC_in32:
12897           case FOURCC_fl32:
12898           case FOURCC_fl64:
12899           case FOURCC_s16l:
12900             /* Fully handled elsewhere */
12901             break;
12902           default:
12903             GST_INFO_OBJECT (qtdemux,
12904                 "unhandled type %" GST_FOURCC_FORMAT, GST_FOURCC_ARGS (fourcc));
12905             break;
12906         }
12907       }
12908       GST_INFO_OBJECT (qtdemux,
12909           "type %" GST_FOURCC_FORMAT " caps %" GST_PTR_FORMAT,
12910           GST_FOURCC_ARGS (fourcc), entry->caps);
12911
12912     } else if (stream->subtype == FOURCC_strm) {
12913       if (fourcc == FOURCC_rtsp) {
12914         stream->redirect_uri = qtdemux_get_rtsp_uri_from_hndl (qtdemux, minf);
12915       } else {
12916         GST_INFO_OBJECT (qtdemux, "unhandled stream type %" GST_FOURCC_FORMAT,
12917             GST_FOURCC_ARGS (fourcc));
12918         goto unknown_stream;
12919       }
12920       entry->sampled = TRUE;
12921     } else if (stream->subtype == FOURCC_subp || stream->subtype == FOURCC_text
12922         || stream->subtype == FOURCC_sbtl || stream->subtype == FOURCC_subt
12923         || stream->subtype == FOURCC_clcp || stream->subtype == FOURCC_wvtt) {
12924
12925       entry->sampled = TRUE;
12926       entry->sparse = TRUE;
12927
12928       entry->caps =
12929           qtdemux_sub_caps (qtdemux, stream, entry, fourcc, stsd_entry_data,
12930           &codec);
12931       if (codec) {
12932         gst_tag_list_add (stream->stream_tags, GST_TAG_MERGE_REPLACE,
12933             GST_TAG_SUBTITLE_CODEC, codec, NULL);
12934         g_free (codec);
12935         codec = NULL;
12936       }
12937
12938       /* hunt for sort-of codec data */
12939       switch (fourcc) {
12940         case FOURCC_mp4s:
12941         {
12942           GNode *mp4s = NULL;
12943           GNode *esds = NULL;
12944
12945           /* look for palette in a stsd->mp4s->esds sub-atom */
12946           mp4s = qtdemux_tree_get_child_by_type (stsd, FOURCC_mp4s);
12947           if (mp4s)
12948             esds = qtdemux_tree_get_child_by_type (mp4s, FOURCC_esds);
12949           if (esds == NULL) {
12950             /* Invalid STSD */
12951             GST_LOG_OBJECT (qtdemux, "Skipping invalid stsd: no esds child");
12952             break;
12953           }
12954
12955           gst_qtdemux_handle_esds (qtdemux, stream, entry, esds,
12956               stream->stream_tags);
12957           break;
12958         }
12959         default:
12960           GST_INFO_OBJECT (qtdemux,
12961               "unhandled type %" GST_FOURCC_FORMAT, GST_FOURCC_ARGS (fourcc));
12962           break;
12963       }
12964       GST_INFO_OBJECT (qtdemux,
12965           "type %" GST_FOURCC_FORMAT " caps %" GST_PTR_FORMAT,
12966           GST_FOURCC_ARGS (fourcc), entry->caps);
12967     } else {
12968       /* everything in 1 sample */
12969       entry->sampled = TRUE;
12970
12971       entry->caps =
12972           qtdemux_generic_caps (qtdemux, stream, entry, fourcc, stsd_entry_data,
12973           &codec);
12974
12975       if (entry->caps == NULL)
12976         goto unknown_stream;
12977
12978       if (codec) {
12979         gst_tag_list_add (stream->stream_tags, GST_TAG_MERGE_REPLACE,
12980             GST_TAG_SUBTITLE_CODEC, codec, NULL);
12981         g_free (codec);
12982         codec = NULL;
12983       }
12984     }
12985
12986     /* promote to sampled format */
12987     if (entry->fourcc == FOURCC_samr) {
12988       /* force mono 8000 Hz for AMR */
12989       entry->sampled = TRUE;
12990       entry->n_channels = 1;
12991       entry->rate = 8000;
12992     } else if (entry->fourcc == FOURCC_sawb) {
12993       /* force mono 16000 Hz for AMR-WB */
12994       entry->sampled = TRUE;
12995       entry->n_channels = 1;
12996       entry->rate = 16000;
12997     } else if (entry->fourcc == FOURCC_mp4a) {
12998       entry->sampled = TRUE;
12999     }
13000
13001
13002     stsd_entry_data += len;
13003     remaining_stsd_len -= len;
13004
13005   }
13006
13007   /* collect sample information */
13008   if (!qtdemux_stbl_init (qtdemux, stream, stbl))
13009     goto samples_failed;
13010
13011   if (qtdemux->fragmented) {
13012     guint64 offset;
13013
13014     /* need all moov samples as basis; probably not many if any at all */
13015     /* prevent moof parsing taking of at this time */
13016     offset = qtdemux->moof_offset;
13017     qtdemux->moof_offset = 0;
13018     if (stream->n_samples &&
13019         !qtdemux_parse_samples (qtdemux, stream, stream->n_samples - 1)) {
13020       qtdemux->moof_offset = offset;
13021       goto samples_failed;
13022     }
13023     qtdemux->moof_offset = offset;
13024     /* movie duration more reliable in this case (e.g. mehd) */
13025     if (qtdemux->segment.duration &&
13026         GST_CLOCK_TIME_IS_VALID (qtdemux->segment.duration))
13027       stream->duration =
13028           GSTTIME_TO_QTSTREAMTIME (stream, qtdemux->segment.duration);
13029   }
13030
13031   /* configure segments */
13032   if (!qtdemux_parse_segments (qtdemux, stream, trak))
13033     goto segments_failed;
13034
13035   /* add some language tag, if useful */
13036   if (stream->lang_id[0] != '\0' && strcmp (stream->lang_id, "unk") &&
13037       strcmp (stream->lang_id, "und")) {
13038     const gchar *lang_code;
13039
13040     /* convert ISO 639-2 code to ISO 639-1 */
13041     lang_code = gst_tag_get_language_code (stream->lang_id);
13042     gst_tag_list_add (stream->stream_tags, GST_TAG_MERGE_REPLACE,
13043         GST_TAG_LANGUAGE_CODE, (lang_code) ? lang_code : stream->lang_id, NULL);
13044   }
13045
13046   /* Check for UDTA tags */
13047   if ((udta = qtdemux_tree_get_child_by_type (trak, FOURCC_udta))) {
13048     qtdemux_parse_udta (qtdemux, stream->stream_tags, udta);
13049   }
13050
13051   /* Insert and sort new stream in track-id order.
13052    * This will help in comparing old/new streams during stream update check */
13053   g_ptr_array_add (qtdemux->active_streams, stream);
13054   g_ptr_array_sort (qtdemux->active_streams,
13055       (GCompareFunc) qtdemux_track_id_compare_func);
13056   GST_DEBUG_OBJECT (qtdemux, "n_streams is now %d",
13057       QTDEMUX_N_STREAMS (qtdemux));
13058
13059   return TRUE;
13060
13061 /* ERRORS */
13062 corrupt_file:
13063   {
13064     GST_ELEMENT_ERROR (qtdemux, STREAM, DEMUX,
13065         (_("This file is corrupt and cannot be played.")), (NULL));
13066     if (stream)
13067       gst_qtdemux_stream_unref (stream);
13068     return FALSE;
13069   }
13070 error_encrypted:
13071   {
13072     GST_ELEMENT_ERROR (qtdemux, STREAM, DECRYPT, (NULL), (NULL));
13073     gst_qtdemux_stream_unref (stream);
13074     return FALSE;
13075   }
13076 samples_failed:
13077 segments_failed:
13078   {
13079     /* we posted an error already */
13080     /* free stbl sub-atoms */
13081     gst_qtdemux_stbl_free (stream);
13082     gst_qtdemux_stream_unref (stream);
13083     return FALSE;
13084   }
13085 existing_stream:
13086   {
13087     GST_INFO_OBJECT (qtdemux, "stream with track id %i already exists",
13088         track_id);
13089     return TRUE;
13090   }
13091 unknown_stream:
13092   {
13093     GST_INFO_OBJECT (qtdemux, "unknown subtype %" GST_FOURCC_FORMAT,
13094         GST_FOURCC_ARGS (stream->subtype));
13095     gst_qtdemux_stream_unref (stream);
13096     return TRUE;
13097   }
13098 }
13099
13100 /* If we can estimate the overall bitrate, and don't have information about the
13101  * stream bitrate for exactly one stream, this guesses the stream bitrate as
13102  * the overall bitrate minus the sum of the bitrates of all other streams. This
13103  * should be useful for the common case where we have one audio and one video
13104  * stream and can estimate the bitrate of one, but not the other. */
13105 static void
13106 gst_qtdemux_guess_bitrate (GstQTDemux * qtdemux)
13107 {
13108   QtDemuxStream *stream = NULL;
13109   gint64 size, sys_bitrate, sum_bitrate = 0;
13110   GstClockTime duration;
13111   guint bitrate;
13112   gint i;
13113
13114   if (qtdemux->fragmented)
13115     return;
13116
13117   GST_DEBUG_OBJECT (qtdemux, "Looking for streams with unknown bitrate");
13118
13119   if (!gst_pad_peer_query_duration (qtdemux->sinkpad, GST_FORMAT_BYTES, &size)
13120       || size <= 0) {
13121     GST_DEBUG_OBJECT (qtdemux,
13122         "Size in bytes of the stream not known - bailing");
13123     return;
13124   }
13125
13126   /* Subtract the header size */
13127   GST_DEBUG_OBJECT (qtdemux, "Total size %" G_GINT64_FORMAT ", header size %u",
13128       size, qtdemux->header_size);
13129
13130   if (size < qtdemux->header_size)
13131     return;
13132
13133   size = size - qtdemux->header_size;
13134
13135   if (!gst_qtdemux_get_duration (qtdemux, &duration)) {
13136     GST_DEBUG_OBJECT (qtdemux, "Stream duration not known - bailing");
13137     return;
13138   }
13139
13140   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
13141     QtDemuxStream *str = QTDEMUX_NTH_STREAM (qtdemux, i);
13142     switch (str->subtype) {
13143       case FOURCC_soun:
13144       case FOURCC_vide:
13145         GST_DEBUG_OBJECT (qtdemux, "checking bitrate for %" GST_PTR_FORMAT,
13146             CUR_STREAM (str)->caps);
13147         /* retrieve bitrate, prefer avg then max */
13148         bitrate = 0;
13149         if (str->stream_tags) {
13150           if (gst_tag_list_get_uint (str->stream_tags,
13151                   GST_TAG_MAXIMUM_BITRATE, &bitrate))
13152             GST_DEBUG_OBJECT (qtdemux, "max-bitrate: %u", bitrate);
13153           if (gst_tag_list_get_uint (str->stream_tags,
13154                   GST_TAG_NOMINAL_BITRATE, &bitrate))
13155             GST_DEBUG_OBJECT (qtdemux, "nominal-bitrate: %u", bitrate);
13156           if (gst_tag_list_get_uint (str->stream_tags,
13157                   GST_TAG_BITRATE, &bitrate))
13158             GST_DEBUG_OBJECT (qtdemux, "bitrate: %u", bitrate);
13159         }
13160         if (bitrate)
13161           sum_bitrate += bitrate;
13162         else {
13163           if (stream) {
13164             GST_DEBUG_OBJECT (qtdemux,
13165                 ">1 stream with unknown bitrate - bailing");
13166             return;
13167           } else
13168             stream = str;
13169         }
13170
13171       default:
13172         /* For other subtypes, we assume no significant impact on bitrate */
13173         break;
13174     }
13175   }
13176
13177   if (!stream) {
13178     GST_DEBUG_OBJECT (qtdemux, "All stream bitrates are known");
13179     return;
13180   }
13181
13182   sys_bitrate = gst_util_uint64_scale (size, GST_SECOND * 8, duration);
13183
13184   if (sys_bitrate < sum_bitrate) {
13185     /* This can happen, since sum_bitrate might be derived from maximum
13186      * bitrates and not average bitrates */
13187     GST_DEBUG_OBJECT (qtdemux,
13188         "System bitrate less than sum bitrate - bailing");
13189     return;
13190   }
13191
13192   bitrate = sys_bitrate - sum_bitrate;
13193   GST_DEBUG_OBJECT (qtdemux, "System bitrate = %" G_GINT64_FORMAT
13194       ", Stream bitrate = %u", sys_bitrate, bitrate);
13195
13196   if (!stream->stream_tags)
13197     stream->stream_tags = gst_tag_list_new_empty ();
13198   else
13199     stream->stream_tags = gst_tag_list_make_writable (stream->stream_tags);
13200
13201   gst_tag_list_add (stream->stream_tags, GST_TAG_MERGE_REPLACE,
13202       GST_TAG_BITRATE, bitrate, NULL);
13203 }
13204
13205 static GstFlowReturn
13206 qtdemux_prepare_streams (GstQTDemux * qtdemux)
13207 {
13208   GstFlowReturn ret = GST_FLOW_OK;
13209   gint i;
13210
13211   GST_DEBUG_OBJECT (qtdemux, "prepare %u streams", QTDEMUX_N_STREAMS (qtdemux));
13212
13213   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
13214     QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
13215     guint32 sample_num = 0;
13216
13217     GST_DEBUG_OBJECT (qtdemux, "track-id %u, fourcc %" GST_FOURCC_FORMAT,
13218         stream->track_id, GST_FOURCC_ARGS (CUR_STREAM (stream)->fourcc));
13219
13220     if (qtdemux->fragmented && qtdemux->pullbased) {
13221       /* need all moov samples first */
13222       GST_OBJECT_LOCK (qtdemux);
13223       while (stream->n_samples == 0)
13224         if ((ret = qtdemux_add_fragmented_samples (qtdemux)) != GST_FLOW_OK)
13225           break;
13226       GST_OBJECT_UNLOCK (qtdemux);
13227     } else {
13228       /* discard any stray moof */
13229       qtdemux->moof_offset = 0;
13230     }
13231
13232     /* prepare braking */
13233     if (ret != GST_FLOW_ERROR)
13234       ret = GST_FLOW_OK;
13235
13236     /* in pull mode, we should have parsed some sample info by now;
13237      * and quite some code will not handle no samples.
13238      * in push mode, we'll just have to deal with it */
13239     if (G_UNLIKELY (qtdemux->pullbased && !stream->n_samples)) {
13240       GST_DEBUG_OBJECT (qtdemux, "no samples for stream; discarding");
13241       g_ptr_array_remove_index (qtdemux->active_streams, i);
13242       i--;
13243       continue;
13244     } else if (stream->track_id == qtdemux->chapters_track_id &&
13245         (stream->subtype == FOURCC_text || stream->subtype == FOURCC_sbtl)) {
13246       /* TODO - parse chapters track and expose it as GstToc; For now just ignore it
13247          so that it doesn't look like a subtitle track */
13248       g_ptr_array_remove_index (qtdemux->active_streams, i);
13249       i--;
13250       continue;
13251     }
13252
13253     /* parse the initial sample for use in setting the frame rate cap */
13254     while (sample_num == 0 && sample_num < stream->n_samples) {
13255       if (!qtdemux_parse_samples (qtdemux, stream, sample_num))
13256         break;
13257       ++sample_num;
13258     }
13259   }
13260
13261   return ret;
13262 }
13263
13264 static gboolean
13265 _stream_equal_func (const QtDemuxStream * stream, const gchar * stream_id)
13266 {
13267   return g_strcmp0 (stream->stream_id, stream_id) == 0;
13268 }
13269
13270 static gboolean
13271 qtdemux_is_streams_update (GstQTDemux * qtdemux)
13272 {
13273   gint i;
13274
13275   /* Different length, updated */
13276   if (QTDEMUX_N_STREAMS (qtdemux) != qtdemux->old_streams->len)
13277     return TRUE;
13278
13279   /* streams in list are sorted in track-id order */
13280   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
13281     /* Different stream-id, updated */
13282     if (g_strcmp0 (QTDEMUX_NTH_STREAM (qtdemux, i)->stream_id,
13283             QTDEMUX_NTH_OLD_STREAM (qtdemux, i)->stream_id))
13284       return TRUE;
13285   }
13286
13287   return FALSE;
13288 }
13289
13290 static gboolean
13291 qtdemux_reuse_and_configure_stream (GstQTDemux * qtdemux,
13292     QtDemuxStream * oldstream, QtDemuxStream * newstream)
13293 {
13294   /* Connect old stream's srcpad to new stream */
13295   newstream->pad = oldstream->pad;
13296   oldstream->pad = NULL;
13297
13298   /* unset new_stream to prevent stream-start event, unless we are EOS in which
13299    * case we need to force one through */
13300   newstream->new_stream = GST_PAD_IS_EOS (newstream->pad);
13301
13302   return gst_qtdemux_configure_stream (qtdemux, newstream);
13303 }
13304
13305 static gboolean
13306 qtdemux_update_streams (GstQTDemux * qtdemux)
13307 {
13308   gint i;
13309   g_assert (qtdemux->streams_aware);
13310
13311   /* At below, figure out which stream in active_streams has identical stream-id
13312    * with that of in old_streams. If there is matching stream-id,
13313    * corresponding newstream will not be exposed again,
13314    * but demux will reuse srcpad of matched old stream
13315    *
13316    * active_streams : newly created streams from the latest moov
13317    * old_streams : existing streams (belong to previous moov)
13318    */
13319
13320   for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
13321     QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
13322     QtDemuxStream *oldstream = NULL;
13323     guint target;
13324
13325     GST_DEBUG_OBJECT (qtdemux, "track-id %u, fourcc %" GST_FOURCC_FORMAT,
13326         stream->track_id, GST_FOURCC_ARGS (CUR_STREAM (stream)->fourcc));
13327
13328     if (g_ptr_array_find_with_equal_func (qtdemux->old_streams,
13329             stream->stream_id, (GEqualFunc) _stream_equal_func, &target)) {
13330       oldstream = QTDEMUX_NTH_OLD_STREAM (qtdemux, target);
13331
13332       /* null pad stream cannot be reused */
13333       if (oldstream->pad == NULL)
13334         oldstream = NULL;
13335     }
13336
13337     if (oldstream) {
13338       GST_DEBUG_OBJECT (qtdemux, "Reuse track-id %d", oldstream->track_id);
13339
13340       if (!qtdemux_reuse_and_configure_stream (qtdemux, oldstream, stream))
13341         return FALSE;
13342
13343       /* we don't need to preserve order of old streams */
13344       g_ptr_array_remove_fast (qtdemux->old_streams, oldstream);
13345     } else {
13346       GstTagList *list;
13347
13348       /* now we have all info and can expose */
13349       list = stream->stream_tags;
13350       stream->stream_tags = NULL;
13351       if (!gst_qtdemux_add_stream (qtdemux, stream, list))
13352         return FALSE;
13353     }
13354   }
13355
13356   return TRUE;
13357 }
13358
13359 /* Must be called with expose lock */
13360 static GstFlowReturn
13361 qtdemux_expose_streams (GstQTDemux * qtdemux)
13362 {
13363   gint i;
13364
13365   GST_DEBUG_OBJECT (qtdemux, "exposing streams");
13366
13367   if (!qtdemux_is_streams_update (qtdemux)) {
13368     GST_DEBUG_OBJECT (qtdemux, "Reuse all streams");
13369     for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
13370       QtDemuxStream *new_stream = QTDEMUX_NTH_STREAM (qtdemux, i);
13371       QtDemuxStream *old_stream = QTDEMUX_NTH_OLD_STREAM (qtdemux, i);
13372       if (!qtdemux_reuse_and_configure_stream (qtdemux, old_stream, new_stream))
13373         return GST_FLOW_ERROR;
13374     }
13375
13376     g_ptr_array_set_size (qtdemux->old_streams, 0);
13377     qtdemux->need_segment = TRUE;
13378
13379     return GST_FLOW_OK;
13380   }
13381
13382   if (qtdemux->streams_aware) {
13383     if (!qtdemux_update_streams (qtdemux))
13384       return GST_FLOW_ERROR;
13385   } else {
13386     for (i = 0; i < QTDEMUX_N_STREAMS (qtdemux); i++) {
13387       QtDemuxStream *stream = QTDEMUX_NTH_STREAM (qtdemux, i);
13388       GstTagList *list;
13389
13390       /* now we have all info and can expose */
13391       list = stream->stream_tags;
13392       stream->stream_tags = NULL;
13393       if (!gst_qtdemux_add_stream (qtdemux, stream, list))
13394         return GST_FLOW_ERROR;
13395
13396     }
13397   }
13398
13399   gst_qtdemux_guess_bitrate (qtdemux);
13400
13401   gst_element_no_more_pads (GST_ELEMENT_CAST (qtdemux));
13402
13403   /* If we have still old_streams, it's no more used stream */
13404   for (i = 0; i < qtdemux->old_streams->len; i++) {
13405     QtDemuxStream *stream = QTDEMUX_NTH_OLD_STREAM (qtdemux, i);
13406
13407     if (stream->pad) {
13408       GstEvent *event;
13409
13410       event = gst_event_new_eos ();
13411       if (qtdemux->segment_seqnum)
13412         gst_event_set_seqnum (event, qtdemux->segment_seqnum);
13413
13414       gst_pad_push_event (stream->pad, event);
13415     }
13416   }
13417
13418   g_ptr_array_set_size (qtdemux->old_streams, 0);
13419
13420   /* check if we should post a redirect in case there is a single trak
13421    * and it is a redirecting trak */
13422   if (QTDEMUX_N_STREAMS (qtdemux) == 1 &&
13423       QTDEMUX_NTH_STREAM (qtdemux, 0)->redirect_uri != NULL) {
13424     GstMessage *m;
13425
13426     GST_INFO_OBJECT (qtdemux, "Issuing a redirect due to a single track with "
13427         "an external content");
13428     m = gst_message_new_element (GST_OBJECT_CAST (qtdemux),
13429         gst_structure_new ("redirect",
13430             "new-location", G_TYPE_STRING,
13431             QTDEMUX_NTH_STREAM (qtdemux, 0)->redirect_uri, NULL));
13432     gst_element_post_message (GST_ELEMENT_CAST (qtdemux), m);
13433     g_free (qtdemux->redirect_location);
13434     qtdemux->redirect_location =
13435         g_strdup (QTDEMUX_NTH_STREAM (qtdemux, 0)->redirect_uri);
13436   }
13437
13438   g_ptr_array_foreach (qtdemux->active_streams,
13439       (GFunc) qtdemux_do_allocation, qtdemux);
13440
13441   qtdemux->need_segment = TRUE;
13442
13443   qtdemux->exposed = TRUE;
13444   return GST_FLOW_OK;
13445 }
13446
13447 typedef struct
13448 {
13449   GstStructure *structure;      /* helper for sort function */
13450   gchar *location;
13451   guint min_req_bitrate;
13452   guint min_req_qt_version;
13453 } GstQtReference;
13454
13455 static gint
13456 qtdemux_redirects_sort_func (gconstpointer a, gconstpointer b)
13457 {
13458   GstQtReference *ref_a = (GstQtReference *) a;
13459   GstQtReference *ref_b = (GstQtReference *) b;
13460
13461   if (ref_b->min_req_qt_version != ref_a->min_req_qt_version)
13462     return ref_b->min_req_qt_version - ref_a->min_req_qt_version;
13463
13464   /* known bitrates go before unknown; higher bitrates go first */
13465   return ref_b->min_req_bitrate - ref_a->min_req_bitrate;
13466 }
13467
13468 /* sort the redirects and post a message for the application.
13469  */
13470 static void
13471 qtdemux_process_redirects (GstQTDemux * qtdemux, GList * references)
13472 {
13473   GstQtReference *best;
13474   GstStructure *s;
13475   GstMessage *msg;
13476   GValue list_val = { 0, };
13477   GList *l;
13478
13479   g_assert (references != NULL);
13480
13481   references = g_list_sort (references, qtdemux_redirects_sort_func);
13482
13483   best = (GstQtReference *) references->data;
13484
13485   g_value_init (&list_val, GST_TYPE_LIST);
13486
13487   for (l = references; l != NULL; l = l->next) {
13488     GstQtReference *ref = (GstQtReference *) l->data;
13489     GValue struct_val = { 0, };
13490
13491     ref->structure = gst_structure_new ("redirect",
13492         "new-location", G_TYPE_STRING, ref->location, NULL);
13493
13494     if (ref->min_req_bitrate > 0) {
13495       gst_structure_set (ref->structure, "minimum-bitrate", G_TYPE_INT,
13496           ref->min_req_bitrate, NULL);
13497     }
13498
13499     g_value_init (&struct_val, GST_TYPE_STRUCTURE);
13500     g_value_set_boxed (&struct_val, ref->structure);
13501     gst_value_list_append_value (&list_val, &struct_val);
13502     g_value_unset (&struct_val);
13503     /* don't free anything here yet, since we need best->structure below */
13504   }
13505
13506   g_assert (best != NULL);
13507   s = gst_structure_copy (best->structure);
13508
13509   if (g_list_length (references) > 1) {
13510     gst_structure_set_value (s, "locations", &list_val);
13511   }
13512
13513   g_value_unset (&list_val);
13514
13515   for (l = references; l != NULL; l = l->next) {
13516     GstQtReference *ref = (GstQtReference *) l->data;
13517
13518     gst_structure_free (ref->structure);
13519     g_free (ref->location);
13520     g_free (ref);
13521   }
13522   g_list_free (references);
13523
13524   GST_INFO_OBJECT (qtdemux, "posting redirect message: %" GST_PTR_FORMAT, s);
13525   g_free (qtdemux->redirect_location);
13526   qtdemux->redirect_location =
13527       g_strdup (gst_structure_get_string (s, "new-location"));
13528   msg = gst_message_new_element (GST_OBJECT_CAST (qtdemux), s);
13529   gst_element_post_message (GST_ELEMENT_CAST (qtdemux), msg);
13530 }
13531
13532 /* look for redirect nodes, collect all redirect information and
13533  * process it.
13534  */
13535 static gboolean
13536 qtdemux_parse_redirects (GstQTDemux * qtdemux)
13537 {
13538   GNode *rmra, *rmda, *rdrf;
13539
13540   rmra = qtdemux_tree_get_child_by_type (qtdemux->moov_node, FOURCC_rmra);
13541   if (rmra) {
13542     GList *redirects = NULL;
13543
13544     rmda = qtdemux_tree_get_child_by_type (rmra, FOURCC_rmda);
13545     while (rmda) {
13546       GstQtReference ref = { NULL, NULL, 0, 0 };
13547       GNode *rmdr, *rmvc;
13548
13549       if ((rmdr = qtdemux_tree_get_child_by_type (rmda, FOURCC_rmdr))) {
13550         ref.min_req_bitrate = QT_UINT32 ((guint8 *) rmdr->data + 12);
13551         GST_LOG_OBJECT (qtdemux, "data rate atom, required bitrate = %u",
13552             ref.min_req_bitrate);
13553       }
13554
13555       if ((rmvc = qtdemux_tree_get_child_by_type (rmda, FOURCC_rmvc))) {
13556         guint32 package = QT_FOURCC ((guint8 *) rmvc->data + 12);
13557         guint version = QT_UINT32 ((guint8 *) rmvc->data + 16);
13558
13559 #ifndef GST_DISABLE_GST_DEBUG
13560         guint bitmask = QT_UINT32 ((guint8 *) rmvc->data + 20);
13561 #endif
13562         guint check_type = QT_UINT16 ((guint8 *) rmvc->data + 24);
13563
13564         GST_LOG_OBJECT (qtdemux,
13565             "version check atom [%" GST_FOURCC_FORMAT "], version=0x%08x"
13566             ", mask=%08x, check_type=%u", GST_FOURCC_ARGS (package), version,
13567             bitmask, check_type);
13568         if (package == FOURCC_qtim && check_type == 0) {
13569           ref.min_req_qt_version = version;
13570         }
13571       }
13572
13573       rdrf = qtdemux_tree_get_child_by_type (rmda, FOURCC_rdrf);
13574       if (rdrf) {
13575         guint32 ref_type;
13576         guint8 *ref_data;
13577         guint ref_len;
13578
13579         ref_len = QT_UINT32 ((guint8 *) rdrf->data);
13580         if (ref_len > 20) {
13581           ref_type = QT_FOURCC ((guint8 *) rdrf->data + 12);
13582           ref_data = (guint8 *) rdrf->data + 20;
13583           if (ref_type == FOURCC_alis) {
13584             guint record_len, record_version, fn_len;
13585
13586             if (ref_len > 70) {
13587               /* MacOSX alias record, google for alias-layout.txt */
13588               record_len = QT_UINT16 (ref_data + 4);
13589               record_version = QT_UINT16 (ref_data + 4 + 2);
13590               fn_len = QT_UINT8 (ref_data + 50);
13591               if (record_len > 50 && record_version == 2 && fn_len > 0) {
13592                 ref.location = g_strndup ((gchar *) ref_data + 51, fn_len);
13593               }
13594             } else {
13595               GST_WARNING_OBJECT (qtdemux, "Invalid rdrf/alis size (%u < 70)",
13596                   ref_len);
13597             }
13598           } else if (ref_type == FOURCC_url_) {
13599             ref.location = g_strndup ((gchar *) ref_data, ref_len - 8);
13600           } else {
13601             GST_DEBUG_OBJECT (qtdemux,
13602                 "unknown rdrf reference type %" GST_FOURCC_FORMAT,
13603                 GST_FOURCC_ARGS (ref_type));
13604           }
13605           if (ref.location != NULL) {
13606             GST_INFO_OBJECT (qtdemux, "New location: %s", ref.location);
13607             redirects =
13608                 g_list_prepend (redirects, g_memdup2 (&ref, sizeof (ref)));
13609           } else {
13610             GST_WARNING_OBJECT (qtdemux,
13611                 "Failed to extract redirect location from rdrf atom");
13612           }
13613         } else {
13614           GST_WARNING_OBJECT (qtdemux, "Invalid rdrf size (%u < 20)", ref_len);
13615         }
13616       }
13617
13618       /* look for others */
13619       rmda = qtdemux_tree_get_sibling_by_type (rmda, FOURCC_rmda);
13620     }
13621
13622     if (redirects != NULL) {
13623       qtdemux_process_redirects (qtdemux, redirects);
13624     }
13625   }
13626   return TRUE;
13627 }
13628
13629 static GstTagList *
13630 qtdemux_add_container_format (GstQTDemux * qtdemux, GstTagList * tags)
13631 {
13632   const gchar *fmt;
13633
13634   if (tags == NULL) {
13635     tags = gst_tag_list_new_empty ();
13636     gst_tag_list_set_scope (tags, GST_TAG_SCOPE_GLOBAL);
13637   }
13638
13639   if (qtdemux->major_brand == FOURCC_mjp2)
13640     fmt = "Motion JPEG 2000";
13641   else if ((qtdemux->major_brand & 0xffff) == FOURCC_3g__)
13642     fmt = "3GP";
13643   else if (qtdemux->major_brand == FOURCC_qt__)
13644     fmt = "Quicktime";
13645   else if (qtdemux->fragmented)
13646     fmt = "ISO fMP4";
13647   else
13648     fmt = "ISO MP4/M4A";
13649
13650   GST_LOG_OBJECT (qtdemux, "mapped %" GST_FOURCC_FORMAT " to '%s'",
13651       GST_FOURCC_ARGS (qtdemux->major_brand), fmt);
13652
13653   gst_tag_list_add (tags, GST_TAG_MERGE_REPLACE, GST_TAG_CONTAINER_FORMAT,
13654       fmt, NULL);
13655
13656   return tags;
13657 }
13658
13659 /* we have read the complete moov node now.
13660  * This function parses all of the relevant info, creates the traks and
13661  * prepares all data structures for playback
13662  */
13663 static gboolean
13664 qtdemux_parse_tree (GstQTDemux * qtdemux)
13665 {
13666   GNode *mvhd;
13667   GNode *trak;
13668   GNode *udta;
13669   GNode *mvex;
13670   GNode *pssh;
13671   guint64 creation_time;
13672   GstDateTime *datetime = NULL;
13673   gint version;
13674
13675   /* make sure we have a usable taglist */
13676   qtdemux->tag_list = gst_tag_list_make_writable (qtdemux->tag_list);
13677
13678   mvhd = qtdemux_tree_get_child_by_type (qtdemux->moov_node, FOURCC_mvhd);
13679   if (mvhd == NULL) {
13680     GST_LOG_OBJECT (qtdemux, "No mvhd node found, looking for redirects.");
13681     return qtdemux_parse_redirects (qtdemux);
13682   }
13683
13684   version = QT_UINT8 ((guint8 *) mvhd->data + 8);
13685   if (version == 1) {
13686     creation_time = QT_UINT64 ((guint8 *) mvhd->data + 12);
13687     qtdemux->timescale = QT_UINT32 ((guint8 *) mvhd->data + 28);
13688     qtdemux->duration = QT_UINT64 ((guint8 *) mvhd->data + 32);
13689   } else if (version == 0) {
13690     creation_time = QT_UINT32 ((guint8 *) mvhd->data + 12);
13691     qtdemux->timescale = QT_UINT32 ((guint8 *) mvhd->data + 20);
13692     qtdemux->duration = QT_UINT32 ((guint8 *) mvhd->data + 24);
13693   } else {
13694     GST_WARNING_OBJECT (qtdemux, "Unhandled mvhd version %d", version);
13695     return FALSE;
13696   }
13697
13698   /* Moving qt creation time (secs since 1904) to unix time */
13699   if (creation_time != 0) {
13700     /* Try to use epoch first as it should be faster and more commonly found */
13701     if (creation_time >= QTDEMUX_SECONDS_FROM_1904_TO_1970) {
13702       gint64 now_s;
13703
13704       creation_time -= QTDEMUX_SECONDS_FROM_1904_TO_1970;
13705       /* some data cleansing sanity */
13706       now_s = g_get_real_time () / G_USEC_PER_SEC;
13707       if (now_s + 24 * 3600 < creation_time) {
13708         GST_DEBUG_OBJECT (qtdemux, "discarding bogus future creation time");
13709       } else {
13710         datetime = gst_date_time_new_from_unix_epoch_utc (creation_time);
13711       }
13712     } else {
13713       GDateTime *base_dt = g_date_time_new_utc (1904, 1, 1, 0, 0, 0);
13714       GDateTime *dt, *dt_local;
13715
13716       dt = g_date_time_add_seconds (base_dt, creation_time);
13717       dt_local = g_date_time_to_local (dt);
13718       datetime = gst_date_time_new_from_g_date_time (dt_local);
13719
13720       g_date_time_unref (base_dt);
13721       g_date_time_unref (dt);
13722     }
13723   }
13724   if (datetime) {
13725     /* Use KEEP as explicit tags should have a higher priority than mvhd tag */
13726     gst_tag_list_add (qtdemux->tag_list, GST_TAG_MERGE_KEEP, GST_TAG_DATE_TIME,
13727         datetime, NULL);
13728     gst_date_time_unref (datetime);
13729   }
13730
13731   GST_INFO_OBJECT (qtdemux, "timescale: %u", qtdemux->timescale);
13732   GST_INFO_OBJECT (qtdemux, "duration: %" G_GUINT64_FORMAT, qtdemux->duration);
13733
13734   /* check for fragmented file and get some (default) data */
13735   mvex = qtdemux_tree_get_child_by_type (qtdemux->moov_node, FOURCC_mvex);
13736   if (mvex) {
13737     GNode *mehd;
13738     GstByteReader mehd_data;
13739
13740     /* let track parsing or anyone know weird stuff might happen ... */
13741     qtdemux->fragmented = TRUE;
13742
13743     /* compensate for total duration */
13744     mehd = qtdemux_tree_get_child_by_type_full (mvex, FOURCC_mehd, &mehd_data);
13745     if (mehd)
13746       qtdemux_parse_mehd (qtdemux, &mehd_data);
13747   }
13748
13749   /* Update the movie segment duration, unless it was directly given to us
13750    * by upstream. Otherwise let it as is, as we don't want to mangle the
13751    * duration provided by upstream that may come e.g. from a MPD file. */
13752   if (!qtdemux->upstream_format_is_time) {
13753     GstClockTime duration;
13754     /* set duration in the segment info */
13755     gst_qtdemux_get_duration (qtdemux, &duration);
13756     qtdemux->segment.duration = duration;
13757     /* also do not exceed duration; stop is set that way post seek anyway,
13758      * and segment activation falls back to duration,
13759      * whereas loop only checks stop, so let's align this here as well */
13760     qtdemux->segment.stop = duration;
13761   }
13762
13763   /* parse all traks */
13764   trak = qtdemux_tree_get_child_by_type (qtdemux->moov_node, FOURCC_trak);
13765   while (trak) {
13766     qtdemux_parse_trak (qtdemux, trak);
13767     /* iterate all siblings */
13768     trak = qtdemux_tree_get_sibling_by_type (trak, FOURCC_trak);
13769   }
13770
13771   qtdemux->tag_list = gst_tag_list_make_writable (qtdemux->tag_list);
13772
13773   /* find tags */
13774   udta = qtdemux_tree_get_child_by_type (qtdemux->moov_node, FOURCC_udta);
13775   if (udta) {
13776     qtdemux_parse_udta (qtdemux, qtdemux->tag_list, udta);
13777   } else {
13778     GST_LOG_OBJECT (qtdemux, "No udta node found.");
13779   }
13780
13781   /* maybe also some tags in meta box */
13782   udta = qtdemux_tree_get_child_by_type (qtdemux->moov_node, FOURCC_meta);
13783   if (udta) {
13784     GST_DEBUG_OBJECT (qtdemux, "Parsing meta box for tags.");
13785     qtdemux_parse_udta (qtdemux, qtdemux->tag_list, udta);
13786   } else {
13787     GST_LOG_OBJECT (qtdemux, "No meta node found.");
13788   }
13789
13790   /* parse any protection system info */
13791   pssh = qtdemux_tree_get_child_by_type (qtdemux->moov_node, FOURCC_pssh);
13792   while (pssh) {
13793     GST_LOG_OBJECT (qtdemux, "Parsing pssh box.");
13794     qtdemux_parse_pssh (qtdemux, pssh);
13795     pssh = qtdemux_tree_get_sibling_by_type (pssh, FOURCC_pssh);
13796   }
13797
13798   qtdemux->tag_list = qtdemux_add_container_format (qtdemux, qtdemux->tag_list);
13799
13800   return TRUE;
13801 }
13802
13803 /* taken from ffmpeg */
13804 static int
13805 read_descr_size (guint8 * ptr, guint8 * end, guint8 ** end_out)
13806 {
13807   int count = 4;
13808   int len = 0;
13809
13810   while (count--) {
13811     int c;
13812
13813     if (ptr >= end)
13814       return -1;
13815
13816     c = *ptr++;
13817     len = (len << 7) | (c & 0x7f);
13818     if (!(c & 0x80))
13819       break;
13820   }
13821   *end_out = ptr;
13822   return len;
13823 }
13824
13825 static GList *
13826 parse_xiph_stream_headers (GstQTDemux * qtdemux, gpointer codec_data,
13827     gsize codec_data_size)
13828 {
13829   GList *list = NULL;
13830   guint8 *p = codec_data;
13831   gint i, offset, num_packets;
13832   guint *length, last;
13833
13834   GST_MEMDUMP_OBJECT (qtdemux, "xiph codec data", codec_data, codec_data_size);
13835
13836   if (codec_data == NULL || codec_data_size == 0)
13837     goto error;
13838
13839   /* start of the stream and vorbis audio or theora video, need to
13840    * send the codec_priv data as first three packets */
13841   num_packets = p[0] + 1;
13842   GST_DEBUG_OBJECT (qtdemux,
13843       "%u stream headers, total length=%" G_GSIZE_FORMAT " bytes",
13844       (guint) num_packets, codec_data_size);
13845
13846   /* Let's put some limits, Don't think there even is a xiph codec
13847    * with more than 3-4 headers */
13848   if (G_UNLIKELY (num_packets > 16)) {
13849     GST_WARNING_OBJECT (qtdemux,
13850         "Unlikely number of xiph headers, most likely not valid");
13851     goto error;
13852   }
13853
13854   length = g_alloca (num_packets * sizeof (guint));
13855   last = 0;
13856   offset = 1;
13857
13858   /* first packets, read length values */
13859   for (i = 0; i < num_packets - 1; i++) {
13860     length[i] = 0;
13861     while (offset < codec_data_size) {
13862       length[i] += p[offset];
13863       if (p[offset++] != 0xff)
13864         break;
13865     }
13866     last += length[i];
13867   }
13868   if (offset + last > codec_data_size)
13869     goto error;
13870
13871   /* last packet is the remaining size */
13872   length[i] = codec_data_size - offset - last;
13873
13874   for (i = 0; i < num_packets; i++) {
13875     GstBuffer *hdr;
13876
13877     GST_DEBUG_OBJECT (qtdemux, "buffer %d: %u bytes", i, (guint) length[i]);
13878
13879     if (offset + length[i] > codec_data_size)
13880       goto error;
13881
13882     hdr = gst_buffer_new_memdup (p + offset, length[i]);
13883     list = g_list_append (list, hdr);
13884
13885     offset += length[i];
13886   }
13887
13888   return list;
13889
13890   /* ERRORS */
13891 error:
13892   {
13893     if (list != NULL)
13894       g_list_free_full (list, (GDestroyNotify) gst_buffer_unref);
13895     return NULL;
13896   }
13897
13898 }
13899
13900 /* this can change the codec originally present in @list */
13901 static void
13902 gst_qtdemux_handle_esds (GstQTDemux * qtdemux, QtDemuxStream * stream,
13903     QtDemuxStreamStsdEntry * entry, GNode * esds, GstTagList * list)
13904 {
13905   int len = QT_UINT32 (esds->data);
13906   guint8 *ptr = esds->data;
13907   guint8 *end = ptr + len;
13908   int tag;
13909   guint8 *data_ptr = NULL;
13910   int data_len = 0;
13911   guint8 object_type_id = 0;
13912   guint8 stream_type = 0;
13913   const char *codec_name = NULL;
13914   GstCaps *caps = NULL;
13915
13916   GST_MEMDUMP_OBJECT (qtdemux, "esds", ptr, len);
13917   ptr += 8;
13918   GST_DEBUG_OBJECT (qtdemux, "version/flags = %08x", QT_UINT32 (ptr));
13919   ptr += 4;
13920   while (ptr + 1 < end) {
13921     tag = QT_UINT8 (ptr);
13922     GST_DEBUG_OBJECT (qtdemux, "tag = %02x", tag);
13923     ptr++;
13924     len = read_descr_size (ptr, end, &ptr);
13925     GST_DEBUG_OBJECT (qtdemux, "len = %d", len);
13926
13927     /* Check the stated amount of data is available for reading */
13928     if (len < 0 || ptr + len > end)
13929       break;
13930
13931     switch (tag) {
13932       case ES_DESCRIPTOR_TAG:
13933         GST_DEBUG_OBJECT (qtdemux, "ID 0x%04x", QT_UINT16 (ptr));
13934         GST_DEBUG_OBJECT (qtdemux, "priority 0x%04x", QT_UINT8 (ptr + 2));
13935         ptr += 3;
13936         break;
13937       case DECODER_CONFIG_DESC_TAG:{
13938         guint max_bitrate, avg_bitrate;
13939
13940         object_type_id = QT_UINT8 (ptr);
13941         stream_type = QT_UINT8 (ptr + 1) >> 2;
13942         max_bitrate = QT_UINT32 (ptr + 5);
13943         avg_bitrate = QT_UINT32 (ptr + 9);
13944         GST_DEBUG_OBJECT (qtdemux, "object_type_id %02x", object_type_id);
13945         GST_DEBUG_OBJECT (qtdemux, "stream_type %02x", stream_type);
13946         GST_DEBUG_OBJECT (qtdemux, "buffer_size_db %02x", QT_UINT24 (ptr + 2));
13947         GST_DEBUG_OBJECT (qtdemux, "max bitrate %u", max_bitrate);
13948         GST_DEBUG_OBJECT (qtdemux, "avg bitrate %u", avg_bitrate);
13949         if (max_bitrate > 0 && max_bitrate < G_MAXUINT32) {
13950           gst_tag_list_add (list, GST_TAG_MERGE_REPLACE,
13951               GST_TAG_MAXIMUM_BITRATE, max_bitrate, NULL);
13952         }
13953         if (avg_bitrate > 0 && avg_bitrate < G_MAXUINT32) {
13954           gst_tag_list_add (list, GST_TAG_MERGE_REPLACE, GST_TAG_BITRATE,
13955               avg_bitrate, NULL);
13956         }
13957         ptr += 13;
13958         break;
13959       }
13960       case DECODER_SPECIFIC_INFO_TAG:
13961         GST_MEMDUMP_OBJECT (qtdemux, "data", ptr, len);
13962         if (object_type_id == 0xe0 && len == 0x40) {
13963           guint8 *data;
13964           GstStructure *s;
13965           guint32 clut[16];
13966           gint i;
13967
13968           GST_DEBUG_OBJECT (qtdemux,
13969               "Have VOBSUB palette. Creating palette event");
13970           /* move to decConfigDescr data and read palette */
13971           data = ptr;
13972           for (i = 0; i < 16; i++) {
13973             clut[i] = QT_UINT32 (data);
13974             data += 4;
13975           }
13976
13977           s = gst_structure_new ("application/x-gst-dvd", "event",
13978               G_TYPE_STRING, "dvd-spu-clut-change",
13979               "clut00", G_TYPE_INT, clut[0], "clut01", G_TYPE_INT, clut[1],
13980               "clut02", G_TYPE_INT, clut[2], "clut03", G_TYPE_INT, clut[3],
13981               "clut04", G_TYPE_INT, clut[4], "clut05", G_TYPE_INT, clut[5],
13982               "clut06", G_TYPE_INT, clut[6], "clut07", G_TYPE_INT, clut[7],
13983               "clut08", G_TYPE_INT, clut[8], "clut09", G_TYPE_INT, clut[9],
13984               "clut10", G_TYPE_INT, clut[10], "clut11", G_TYPE_INT, clut[11],
13985               "clut12", G_TYPE_INT, clut[12], "clut13", G_TYPE_INT, clut[13],
13986               "clut14", G_TYPE_INT, clut[14], "clut15", G_TYPE_INT, clut[15],
13987               NULL);
13988
13989           /* store event and trigger custom processing */
13990           stream->pending_event =
13991               gst_event_new_custom (GST_EVENT_CUSTOM_DOWNSTREAM, s);
13992         } else {
13993           /* Generic codec_data handler puts it on the caps */
13994           data_ptr = ptr;
13995           data_len = len;
13996         }
13997
13998         ptr += len;
13999         break;
14000       case SL_CONFIG_DESC_TAG:
14001         GST_DEBUG_OBJECT (qtdemux, "data %02x", QT_UINT8 (ptr));
14002         ptr += 1;
14003         break;
14004       default:
14005         GST_DEBUG_OBJECT (qtdemux, "Unknown/unhandled descriptor tag %02x",
14006             tag);
14007         GST_MEMDUMP_OBJECT (qtdemux, "descriptor data", ptr, len);
14008         ptr += len;
14009         break;
14010     }
14011   }
14012
14013   /* object_type_id in the esds atom in mp4a and mp4v tells us which codec is
14014    * in use, and should also be used to override some other parameters for some
14015    * codecs. */
14016   switch (object_type_id) {
14017     case 0x20:                 /* MPEG-4 */
14018       /* 4 bytes for the visual_object_sequence_start_code and 1 byte for the
14019        * profile_and_level_indication */
14020       if (data_ptr != NULL && data_len >= 5 &&
14021           GST_READ_UINT32_BE (data_ptr) == 0x000001b0) {
14022         gst_codec_utils_mpeg4video_caps_set_level_and_profile (entry->caps,
14023             data_ptr + 4, data_len - 4);
14024       }
14025       break;                    /* Nothing special needed here */
14026     case 0x21:                 /* H.264 */
14027       codec_name = "H.264 / AVC";
14028       caps = gst_caps_new_simple ("video/x-h264",
14029           "stream-format", G_TYPE_STRING, "avc",
14030           "alignment", G_TYPE_STRING, "au", NULL);
14031       break;
14032     case 0x40:                 /* AAC (any) */
14033     case 0x66:                 /* AAC Main */
14034     case 0x67:                 /* AAC LC */
14035     case 0x68:                 /* AAC SSR */
14036       /* Override channels and rate based on the codec_data, as it's often
14037        * wrong. */
14038       /* Only do so for basic setup without HE-AAC extension */
14039       if (data_ptr && data_len == 2) {
14040         guint channels, rate;
14041
14042         channels = gst_codec_utils_aac_get_channels (data_ptr, data_len);
14043         if (channels > 0)
14044           entry->n_channels = channels;
14045
14046         rate = gst_codec_utils_aac_get_sample_rate (data_ptr, data_len);
14047         if (rate > 0)
14048           entry->rate = rate;
14049       }
14050
14051       /* Set level and profile if possible */
14052       if (data_ptr != NULL && data_len >= 2) {
14053         gst_codec_utils_aac_caps_set_level_and_profile (entry->caps,
14054             data_ptr, data_len);
14055       } else {
14056         const gchar *profile_str = NULL;
14057         GstBuffer *buffer;
14058         GstMapInfo map;
14059         guint8 *codec_data;
14060         gint rate_idx, profile;
14061
14062         /* No codec_data, let's invent something.
14063          * FIXME: This is wrong for SBR! */
14064
14065         GST_WARNING_OBJECT (qtdemux, "No codec_data for AAC available");
14066
14067         buffer = gst_buffer_new_and_alloc (2);
14068         gst_buffer_map (buffer, &map, GST_MAP_WRITE);
14069         codec_data = map.data;
14070
14071         rate_idx =
14072             gst_codec_utils_aac_get_index_from_sample_rate (CUR_STREAM
14073             (stream)->rate);
14074
14075         switch (object_type_id) {
14076           case 0x66:
14077             profile_str = "main";
14078             profile = 0;
14079             break;
14080           case 0x67:
14081             profile_str = "lc";
14082             profile = 1;
14083             break;
14084           case 0x68:
14085             profile_str = "ssr";
14086             profile = 2;
14087             break;
14088           default:
14089             profile = 3;
14090             break;
14091         }
14092
14093         codec_data[0] = ((profile + 1) << 3) | ((rate_idx & 0xE) >> 1);
14094         codec_data[1] =
14095             ((rate_idx & 0x1) << 7) | (CUR_STREAM (stream)->n_channels << 3);
14096
14097         gst_buffer_unmap (buffer, &map);
14098         gst_caps_set_simple (CUR_STREAM (stream)->caps, "codec_data",
14099             GST_TYPE_BUFFER, buffer, NULL);
14100         gst_buffer_unref (buffer);
14101
14102         if (profile_str) {
14103           gst_caps_set_simple (CUR_STREAM (stream)->caps, "profile",
14104               G_TYPE_STRING, profile_str, NULL);
14105         }
14106       }
14107       break;
14108     case 0x60:                 /* MPEG-2, various profiles */
14109     case 0x61:
14110     case 0x62:
14111     case 0x63:
14112     case 0x64:
14113     case 0x65:
14114       codec_name = "MPEG-2 video";
14115       caps = gst_caps_new_simple ("video/mpeg",
14116           "mpegversion", G_TYPE_INT, 2,
14117           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
14118       break;
14119     case 0x69:                 /* MPEG-2 BC audio */
14120     case 0x6B:                 /* MPEG-1 audio */
14121       caps = gst_caps_new_simple ("audio/mpeg",
14122           "mpegversion", G_TYPE_INT, 1, NULL);
14123       codec_name = "MPEG-1 audio";
14124       break;
14125     case 0x6A:                 /* MPEG-1 */
14126       codec_name = "MPEG-1 video";
14127       caps = gst_caps_new_simple ("video/mpeg",
14128           "mpegversion", G_TYPE_INT, 1,
14129           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
14130       break;
14131     case 0x6C:                 /* MJPEG */
14132       caps =
14133           gst_caps_new_simple ("image/jpeg", "parsed", G_TYPE_BOOLEAN, TRUE,
14134           NULL);
14135       codec_name = "Motion-JPEG";
14136       break;
14137     case 0x6D:                 /* PNG */
14138       caps =
14139           gst_caps_new_simple ("image/png", "parsed", G_TYPE_BOOLEAN, TRUE,
14140           NULL);
14141       codec_name = "PNG still images";
14142       break;
14143     case 0x6E:                 /* JPEG2000 */
14144       codec_name = "JPEG-2000";
14145       caps = gst_caps_new_simple ("image/x-j2c", "fields", G_TYPE_INT, 1, NULL);
14146       break;
14147     case 0xA4:                 /* Dirac */
14148       codec_name = "Dirac";
14149       caps = gst_caps_new_empty_simple ("video/x-dirac");
14150       break;
14151     case 0xA5:                 /* AC3 */
14152       codec_name = "AC-3 audio";
14153       caps = gst_caps_new_simple ("audio/x-ac3",
14154           "framed", G_TYPE_BOOLEAN, TRUE, NULL);
14155       break;
14156     case 0xA9:                 /* AC3 */
14157       codec_name = "DTS audio";
14158       caps = gst_caps_new_simple ("audio/x-dts",
14159           "framed", G_TYPE_BOOLEAN, TRUE, NULL);
14160       break;
14161     case 0xDD:
14162       if (stream_type == 0x05 && data_ptr) {
14163         GList *headers =
14164             parse_xiph_stream_headers (qtdemux, data_ptr, data_len);
14165         if (headers) {
14166           GList *tmp;
14167           GValue arr_val = G_VALUE_INIT;
14168           GValue buf_val = G_VALUE_INIT;
14169           GstStructure *s;
14170
14171           /* Let's assume it's vorbis if it's an audio stream of type 0xdd and we have codec data that extracts properly */
14172           codec_name = "Vorbis";
14173           caps = gst_caps_new_empty_simple ("audio/x-vorbis");
14174           g_value_init (&arr_val, GST_TYPE_ARRAY);
14175           g_value_init (&buf_val, GST_TYPE_BUFFER);
14176           for (tmp = headers; tmp; tmp = tmp->next) {
14177             g_value_set_boxed (&buf_val, (GstBuffer *) tmp->data);
14178             gst_value_array_append_value (&arr_val, &buf_val);
14179           }
14180           s = gst_caps_get_structure (caps, 0);
14181           gst_structure_take_value (s, "streamheader", &arr_val);
14182           g_value_unset (&buf_val);
14183           g_list_free (headers);
14184
14185           data_ptr = NULL;
14186           data_len = 0;
14187         }
14188       }
14189       break;
14190     case 0xE1:                 /* QCELP */
14191       /* QCELP, the codec_data is a riff tag (little endian) with
14192        * more info (http://ftp.3gpp2.org/TSGC/Working/2003/2003-05-SanDiego/TSG-C-2003-05-San%20Diego/WG1/SWG12/C12-20030512-006%20=%20C12-20030217-015_Draft_Baseline%20Text%20of%20FFMS_R2.doc). */
14193       caps = gst_caps_new_empty_simple ("audio/qcelp");
14194       codec_name = "QCELP";
14195       break;
14196     default:
14197       break;
14198   }
14199
14200   /* If we have a replacement caps, then change our caps for this stream */
14201   if (caps) {
14202     gst_caps_unref (entry->caps);
14203     entry->caps = caps;
14204   }
14205
14206   if (codec_name && list)
14207     gst_tag_list_add (list, GST_TAG_MERGE_REPLACE,
14208         GST_TAG_AUDIO_CODEC, codec_name, NULL);
14209
14210   /* Add the codec_data attribute to caps, if we have it */
14211   if (data_ptr) {
14212     GstBuffer *buffer;
14213
14214     buffer = gst_buffer_new_and_alloc (data_len);
14215     gst_buffer_fill (buffer, 0, data_ptr, data_len);
14216
14217     GST_DEBUG_OBJECT (qtdemux, "setting codec_data from esds");
14218     GST_MEMDUMP_OBJECT (qtdemux, "codec_data from esds", data_ptr, data_len);
14219
14220     gst_caps_set_simple (entry->caps, "codec_data", GST_TYPE_BUFFER,
14221         buffer, NULL);
14222     gst_buffer_unref (buffer);
14223   }
14224
14225 }
14226
14227 static inline GstCaps *
14228 _get_unknown_codec_name (const gchar * type, guint32 fourcc)
14229 {
14230   GstCaps *caps;
14231   guint i;
14232   char *s, fourstr[5];
14233
14234   g_snprintf (fourstr, 5, "%" GST_FOURCC_FORMAT, GST_FOURCC_ARGS (fourcc));
14235   for (i = 0; i < 4; i++) {
14236     if (!g_ascii_isalnum (fourstr[i]))
14237       fourstr[i] = '_';
14238   }
14239   s = g_strdup_printf ("%s/x-gst-fourcc-%s", type, g_strstrip (fourstr));
14240   caps = gst_caps_new_empty_simple (s);
14241   g_free (s);
14242   return caps;
14243 }
14244
14245 #define _codec(name) \
14246   do { \
14247     if (codec_name) { \
14248       *codec_name = g_strdup (name); \
14249     } \
14250   } while (0)
14251
14252 static GstCaps *
14253 qtdemux_video_caps (GstQTDemux * qtdemux, QtDemuxStream * stream,
14254     QtDemuxStreamStsdEntry * entry, guint32 fourcc,
14255     const guint8 * stsd_entry_data, gchar ** codec_name)
14256 {
14257   GstCaps *caps = NULL;
14258   GstVideoFormat format = GST_VIDEO_FORMAT_UNKNOWN;
14259
14260   switch (fourcc) {
14261     case FOURCC_png:
14262       _codec ("PNG still images");
14263       caps = gst_caps_new_empty_simple ("image/png");
14264       break;
14265     case FOURCC_jpeg:
14266       _codec ("JPEG still images");
14267       caps =
14268           gst_caps_new_simple ("image/jpeg", "parsed", G_TYPE_BOOLEAN, TRUE,
14269           NULL);
14270       break;
14271     case GST_MAKE_FOURCC ('m', 'j', 'p', 'a'):
14272     case GST_MAKE_FOURCC ('A', 'V', 'D', 'J'):
14273     case GST_MAKE_FOURCC ('M', 'J', 'P', 'G'):
14274     case GST_MAKE_FOURCC ('d', 'm', 'b', '1'):
14275       _codec ("Motion-JPEG");
14276       caps =
14277           gst_caps_new_simple ("image/jpeg", "parsed", G_TYPE_BOOLEAN, TRUE,
14278           NULL);
14279       break;
14280     case GST_MAKE_FOURCC ('m', 'j', 'p', 'b'):
14281       _codec ("Motion-JPEG format B");
14282       caps = gst_caps_new_empty_simple ("video/x-mjpeg-b");
14283       break;
14284     case FOURCC_mjp2:
14285       _codec ("JPEG-2000");
14286       /* override to what it should be according to spec, avoid palette_data */
14287       entry->bits_per_sample = 24;
14288       caps = gst_caps_new_simple ("image/x-j2c", "fields", G_TYPE_INT, 1, NULL);
14289       break;
14290     case FOURCC_SVQ3:
14291       _codec ("Sorensen video v.3");
14292       caps = gst_caps_new_simple ("video/x-svq",
14293           "svqversion", G_TYPE_INT, 3, NULL);
14294       break;
14295     case GST_MAKE_FOURCC ('s', 'v', 'q', 'i'):
14296     case GST_MAKE_FOURCC ('S', 'V', 'Q', '1'):
14297       _codec ("Sorensen video v.1");
14298       caps = gst_caps_new_simple ("video/x-svq",
14299           "svqversion", G_TYPE_INT, 1, NULL);
14300       break;
14301     case GST_MAKE_FOURCC ('W', 'R', 'A', 'W'):
14302       caps = gst_caps_new_empty_simple ("video/x-raw");
14303       gst_caps_set_simple (caps, "format", G_TYPE_STRING, "RGB8P", NULL);
14304       _codec ("Windows Raw RGB");
14305       stream->alignment = 32;
14306       break;
14307     case FOURCC_raw_:
14308     {
14309       guint16 bps;
14310
14311       bps = QT_UINT16 (stsd_entry_data + 82);
14312       switch (bps) {
14313         case 15:
14314           format = GST_VIDEO_FORMAT_RGB15;
14315           break;
14316         case 16:
14317           format = GST_VIDEO_FORMAT_RGB16;
14318           break;
14319         case 24:
14320           format = GST_VIDEO_FORMAT_RGB;
14321           break;
14322         case 32:
14323           format = GST_VIDEO_FORMAT_ARGB;
14324           break;
14325         default:
14326           /* unknown */
14327           break;
14328       }
14329       break;
14330     }
14331     case GST_MAKE_FOURCC ('y', 'v', '1', '2'):
14332       format = GST_VIDEO_FORMAT_I420;
14333       break;
14334     case GST_MAKE_FOURCC ('y', 'u', 'v', '2'):
14335     case GST_MAKE_FOURCC ('Y', 'u', 'v', '2'):
14336       format = GST_VIDEO_FORMAT_I420;
14337       break;
14338     case FOURCC_2vuy:
14339     case GST_MAKE_FOURCC ('2', 'V', 'u', 'y'):
14340       format = GST_VIDEO_FORMAT_UYVY;
14341       break;
14342     case GST_MAKE_FOURCC ('v', '3', '0', '8'):
14343       format = GST_VIDEO_FORMAT_v308;
14344       break;
14345     case GST_MAKE_FOURCC ('v', '2', '1', '6'):
14346       format = GST_VIDEO_FORMAT_v216;
14347       break;
14348     case FOURCC_v210:
14349       format = GST_VIDEO_FORMAT_v210;
14350       break;
14351     case GST_MAKE_FOURCC ('r', '2', '1', '0'):
14352       format = GST_VIDEO_FORMAT_r210;
14353       break;
14354       /* Packed YUV 4:4:4 10 bit in 32 bits, complex
14355          case GST_MAKE_FOURCC ('v', '4', '1', '0'):
14356          format = GST_VIDEO_FORMAT_v410;
14357          break;
14358        */
14359       /* Packed YUV 4:4:4:4 8 bit in 32 bits
14360        * but different order than AYUV
14361        case GST_MAKE_FOURCC ('v', '4', '0', '8'):
14362        format = GST_VIDEO_FORMAT_v408;
14363        break;
14364        */
14365     case GST_MAKE_FOURCC ('m', 'p', 'e', 'g'):
14366     case GST_MAKE_FOURCC ('m', 'p', 'g', '1'):
14367       _codec ("MPEG-1 video");
14368       caps = gst_caps_new_simple ("video/mpeg", "mpegversion", G_TYPE_INT, 1,
14369           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
14370       break;
14371     case GST_MAKE_FOURCC ('h', 'd', 'v', '1'): /* HDV 720p30 */
14372     case GST_MAKE_FOURCC ('h', 'd', 'v', '2'): /* HDV 1080i60 */
14373     case GST_MAKE_FOURCC ('h', 'd', 'v', '3'): /* HDV 1080i50 */
14374     case GST_MAKE_FOURCC ('h', 'd', 'v', '4'): /* HDV 720p24 */
14375     case GST_MAKE_FOURCC ('h', 'd', 'v', '5'): /* HDV 720p25 */
14376     case GST_MAKE_FOURCC ('h', 'd', 'v', '6'): /* HDV 1080p24 */
14377     case GST_MAKE_FOURCC ('h', 'd', 'v', '7'): /* HDV 1080p25 */
14378     case GST_MAKE_FOURCC ('h', 'd', 'v', '8'): /* HDV 1080p30 */
14379     case GST_MAKE_FOURCC ('h', 'd', 'v', '9'): /* HDV 720p60 */
14380     case GST_MAKE_FOURCC ('h', 'd', 'v', 'a'): /* HDV 720p50 */
14381     case GST_MAKE_FOURCC ('m', 'x', '5', 'n'): /* MPEG2 IMX NTSC 525/60 50mb/s produced by FCP */
14382     case GST_MAKE_FOURCC ('m', 'x', '5', 'p'): /* MPEG2 IMX PAL 625/60 50mb/s produced by FCP */
14383     case GST_MAKE_FOURCC ('m', 'x', '4', 'n'): /* MPEG2 IMX NTSC 525/60 40mb/s produced by FCP */
14384     case GST_MAKE_FOURCC ('m', 'x', '4', 'p'): /* MPEG2 IMX PAL 625/60 40mb/s produced by FCP */
14385     case GST_MAKE_FOURCC ('m', 'x', '3', 'n'): /* MPEG2 IMX NTSC 525/60 30mb/s produced by FCP */
14386     case GST_MAKE_FOURCC ('m', 'x', '3', 'p'): /* MPEG2 IMX PAL 625/50 30mb/s produced by FCP */
14387     case GST_MAKE_FOURCC ('x', 'd', 'v', '1'): /* XDCAM HD 720p30 35Mb/s */
14388     case GST_MAKE_FOURCC ('x', 'd', 'v', '2'): /* XDCAM HD 1080i60 35Mb/s */
14389     case GST_MAKE_FOURCC ('x', 'd', 'v', '3'): /* XDCAM HD 1080i50 35Mb/s */
14390     case GST_MAKE_FOURCC ('x', 'd', 'v', '4'): /* XDCAM HD 720p24 35Mb/s */
14391     case GST_MAKE_FOURCC ('x', 'd', 'v', '5'): /* XDCAM HD 720p25 35Mb/s */
14392     case GST_MAKE_FOURCC ('x', 'd', 'v', '6'): /* XDCAM HD 1080p24 35Mb/s */
14393     case GST_MAKE_FOURCC ('x', 'd', 'v', '7'): /* XDCAM HD 1080p25 35Mb/s */
14394     case GST_MAKE_FOURCC ('x', 'd', 'v', '8'): /* XDCAM HD 1080p30 35Mb/s */
14395     case GST_MAKE_FOURCC ('x', 'd', 'v', '9'): /* XDCAM HD 720p60 35Mb/s */
14396     case GST_MAKE_FOURCC ('x', 'd', 'v', 'a'): /* XDCAM HD 720p50 35Mb/s */
14397     case GST_MAKE_FOURCC ('x', 'd', 'v', 'b'): /* XDCAM EX 1080i60 50Mb/s CBR */
14398     case GST_MAKE_FOURCC ('x', 'd', 'v', 'c'): /* XDCAM EX 1080i50 50Mb/s CBR */
14399     case GST_MAKE_FOURCC ('x', 'd', 'v', 'd'): /* XDCAM HD 1080p24 50Mb/s CBR */
14400     case GST_MAKE_FOURCC ('x', 'd', 'v', 'e'): /* XDCAM HD 1080p25 50Mb/s CBR */
14401     case GST_MAKE_FOURCC ('x', 'd', 'v', 'f'): /* XDCAM HD 1080p30 50Mb/s CBR */
14402     case GST_MAKE_FOURCC ('x', 'd', '5', '1'): /* XDCAM HD422 720p30 50Mb/s CBR */
14403     case GST_MAKE_FOURCC ('x', 'd', '5', '4'): /* XDCAM HD422 720p24 50Mb/s CBR */
14404     case GST_MAKE_FOURCC ('x', 'd', '5', '5'): /* XDCAM HD422 720p25 50Mb/s CBR */
14405     case GST_MAKE_FOURCC ('x', 'd', '5', '9'): /* XDCAM HD422 720p60 50Mb/s CBR */
14406     case GST_MAKE_FOURCC ('x', 'd', '5', 'a'): /* XDCAM HD422 720p50 50Mb/s CBR */
14407     case GST_MAKE_FOURCC ('x', 'd', '5', 'b'): /* XDCAM HD422 1080i50 50Mb/s CBR */
14408     case GST_MAKE_FOURCC ('x', 'd', '5', 'c'): /* XDCAM HD422 1080i50 50Mb/s CBR */
14409     case GST_MAKE_FOURCC ('x', 'd', '5', 'd'): /* XDCAM HD422 1080p24 50Mb/s CBR */
14410     case GST_MAKE_FOURCC ('x', 'd', '5', 'e'): /* XDCAM HD422 1080p25 50Mb/s CBR */
14411     case GST_MAKE_FOURCC ('x', 'd', '5', 'f'): /* XDCAM HD422 1080p30 50Mb/s CBR */
14412     case GST_MAKE_FOURCC ('x', 'd', 'h', 'd'): /* XDCAM HD 540p */
14413     case GST_MAKE_FOURCC ('x', 'd', 'h', '2'): /* XDCAM HD422 540p */
14414     case GST_MAKE_FOURCC ('A', 'V', 'm', 'p'): /* AVID IMX PAL */
14415     case GST_MAKE_FOURCC ('m', 'p', 'g', '2'): /* AVID IMX PAL */
14416     case GST_MAKE_FOURCC ('m', 'p', '2', 'v'): /* AVID IMX PAL */
14417     case GST_MAKE_FOURCC ('m', '2', 'v', '1'):
14418       _codec ("MPEG-2 video");
14419       caps = gst_caps_new_simple ("video/mpeg", "mpegversion", G_TYPE_INT, 2,
14420           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
14421       break;
14422     case GST_MAKE_FOURCC ('g', 'i', 'f', ' '):
14423       _codec ("GIF still images");
14424       caps = gst_caps_new_empty_simple ("image/gif");
14425       break;
14426     case FOURCC_h263:
14427     case GST_MAKE_FOURCC ('H', '2', '6', '3'):
14428     case FOURCC_s263:
14429     case GST_MAKE_FOURCC ('U', '2', '6', '3'):
14430       _codec ("H.263");
14431       /* ffmpeg uses the height/width props, don't know why */
14432       caps = gst_caps_new_simple ("video/x-h263",
14433           "variant", G_TYPE_STRING, "itu", NULL);
14434       break;
14435     case FOURCC_mp4v:
14436     case FOURCC_MP4V:
14437       _codec ("MPEG-4 video");
14438       caps = gst_caps_new_simple ("video/mpeg", "mpegversion", G_TYPE_INT, 4,
14439           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
14440       break;
14441     case GST_MAKE_FOURCC ('3', 'i', 'v', 'd'):
14442     case GST_MAKE_FOURCC ('3', 'I', 'V', 'D'):
14443       _codec ("Microsoft MPEG-4 4.3");  /* FIXME? */
14444       caps = gst_caps_new_simple ("video/x-msmpeg",
14445           "msmpegversion", G_TYPE_INT, 43, NULL);
14446       break;
14447     case GST_MAKE_FOURCC ('D', 'I', 'V', '3'):
14448       _codec ("DivX 3");
14449       caps = gst_caps_new_simple ("video/x-divx",
14450           "divxversion", G_TYPE_INT, 3, NULL);
14451       break;
14452     case GST_MAKE_FOURCC ('D', 'I', 'V', 'X'):
14453     case GST_MAKE_FOURCC ('d', 'i', 'v', 'x'):
14454       _codec ("DivX 4");
14455       caps = gst_caps_new_simple ("video/x-divx",
14456           "divxversion", G_TYPE_INT, 4, NULL);
14457       break;
14458     case GST_MAKE_FOURCC ('D', 'X', '5', '0'):
14459       _codec ("DivX 5");
14460       caps = gst_caps_new_simple ("video/x-divx",
14461           "divxversion", G_TYPE_INT, 5, NULL);
14462       break;
14463
14464     case GST_MAKE_FOURCC ('F', 'F', 'V', '1'):
14465       _codec ("FFV1");
14466       caps = gst_caps_new_simple ("video/x-ffv",
14467           "ffvversion", G_TYPE_INT, 1, NULL);
14468       break;
14469
14470     case GST_MAKE_FOURCC ('3', 'I', 'V', '1'):
14471     case GST_MAKE_FOURCC ('3', 'I', 'V', '2'):
14472     case FOURCC_XVID:
14473     case FOURCC_xvid:
14474     case FOURCC_FMP4:
14475     case FOURCC_fmp4:
14476     case GST_MAKE_FOURCC ('U', 'M', 'P', '4'):
14477       caps = gst_caps_new_simple ("video/mpeg", "mpegversion", G_TYPE_INT, 4,
14478           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
14479       _codec ("MPEG-4");
14480       break;
14481
14482     case GST_MAKE_FOURCC ('c', 'v', 'i', 'd'):
14483       _codec ("Cinepak");
14484       caps = gst_caps_new_empty_simple ("video/x-cinepak");
14485       break;
14486     case GST_MAKE_FOURCC ('q', 'd', 'r', 'w'):
14487       _codec ("Apple QuickDraw");
14488       caps = gst_caps_new_empty_simple ("video/x-qdrw");
14489       break;
14490     case GST_MAKE_FOURCC ('r', 'p', 'z', 'a'):
14491       _codec ("Apple video");
14492       caps = gst_caps_new_empty_simple ("video/x-apple-video");
14493       break;
14494     case FOURCC_H264:
14495     case FOURCC_avc1:
14496     case FOURCC_dva1:
14497       _codec ("H.264 / AVC");
14498       caps = gst_caps_new_simple ("video/x-h264",
14499           "stream-format", G_TYPE_STRING, "avc",
14500           "alignment", G_TYPE_STRING, "au", NULL);
14501       break;
14502     case FOURCC_avc3:
14503     case FOURCC_dvav:
14504       _codec ("H.264 / AVC");
14505       caps = gst_caps_new_simple ("video/x-h264",
14506           "stream-format", G_TYPE_STRING, "avc3",
14507           "alignment", G_TYPE_STRING, "au", NULL);
14508       break;
14509     case FOURCC_H265:
14510     case FOURCC_hvc1:
14511     case FOURCC_dvh1:
14512       _codec ("H.265 / HEVC");
14513       caps = gst_caps_new_simple ("video/x-h265",
14514           "stream-format", G_TYPE_STRING, "hvc1",
14515           "alignment", G_TYPE_STRING, "au", NULL);
14516       break;
14517     case FOURCC_hev1:
14518     case FOURCC_dvhe:
14519       _codec ("H.265 / HEVC");
14520       caps = gst_caps_new_simple ("video/x-h265",
14521           "stream-format", G_TYPE_STRING, "hev1",
14522           "alignment", G_TYPE_STRING, "au", NULL);
14523       break;
14524     case FOURCC_rle_:
14525       _codec ("Run-length encoding");
14526       caps = gst_caps_new_simple ("video/x-rle",
14527           "layout", G_TYPE_STRING, "quicktime", NULL);
14528       break;
14529     case FOURCC_WRLE:
14530       _codec ("Run-length encoding");
14531       caps = gst_caps_new_simple ("video/x-rle",
14532           "layout", G_TYPE_STRING, "microsoft", NULL);
14533       break;
14534     case GST_MAKE_FOURCC ('I', 'V', '3', '2'):
14535     case GST_MAKE_FOURCC ('i', 'v', '3', '2'):
14536       _codec ("Indeo Video 3");
14537       caps = gst_caps_new_simple ("video/x-indeo",
14538           "indeoversion", G_TYPE_INT, 3, NULL);
14539       break;
14540     case GST_MAKE_FOURCC ('I', 'V', '4', '1'):
14541     case GST_MAKE_FOURCC ('i', 'v', '4', '1'):
14542       _codec ("Intel Video 4");
14543       caps = gst_caps_new_simple ("video/x-indeo",
14544           "indeoversion", G_TYPE_INT, 4, NULL);
14545       break;
14546     case FOURCC_dvcp:
14547     case FOURCC_dvc_:
14548     case GST_MAKE_FOURCC ('d', 'v', 's', 'd'):
14549     case GST_MAKE_FOURCC ('D', 'V', 'S', 'D'):
14550     case GST_MAKE_FOURCC ('d', 'v', 'c', 's'):
14551     case GST_MAKE_FOURCC ('D', 'V', 'C', 'S'):
14552     case GST_MAKE_FOURCC ('d', 'v', '2', '5'):
14553     case GST_MAKE_FOURCC ('d', 'v', 'p', 'p'):
14554       _codec ("DV Video");
14555       caps = gst_caps_new_simple ("video/x-dv", "dvversion", G_TYPE_INT, 25,
14556           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
14557       break;
14558     case FOURCC_dv5n:          /* DVCPRO50 NTSC */
14559     case FOURCC_dv5p:          /* DVCPRO50 PAL */
14560       _codec ("DVCPro50 Video");
14561       caps = gst_caps_new_simple ("video/x-dv", "dvversion", G_TYPE_INT, 50,
14562           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
14563       break;
14564     case GST_MAKE_FOURCC ('d', 'v', 'h', '5'): /* DVCPRO HD 50i produced by FCP */
14565     case GST_MAKE_FOURCC ('d', 'v', 'h', '6'): /* DVCPRO HD 60i produced by FCP */
14566       _codec ("DVCProHD Video");
14567       caps = gst_caps_new_simple ("video/x-dv", "dvversion", G_TYPE_INT, 100,
14568           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
14569       break;
14570     case GST_MAKE_FOURCC ('s', 'm', 'c', ' '):
14571       _codec ("Apple Graphics (SMC)");
14572       caps = gst_caps_new_empty_simple ("video/x-smc");
14573       break;
14574     case GST_MAKE_FOURCC ('V', 'P', '3', '1'):
14575       _codec ("VP3");
14576       caps = gst_caps_new_empty_simple ("video/x-vp3");
14577       break;
14578     case GST_MAKE_FOURCC ('V', 'P', '6', 'F'):
14579       _codec ("VP6 Flash");
14580       caps = gst_caps_new_empty_simple ("video/x-vp6-flash");
14581       break;
14582     case FOURCC_XiTh:
14583       _codec ("Theora");
14584       caps = gst_caps_new_empty_simple ("video/x-theora");
14585       /* theora uses one byte of padding in the data stream because it does not
14586        * allow 0 sized packets while theora does */
14587       entry->padding = 1;
14588       break;
14589     case FOURCC_drac:
14590       _codec ("Dirac");
14591       caps = gst_caps_new_empty_simple ("video/x-dirac");
14592       break;
14593     case GST_MAKE_FOURCC ('t', 'i', 'f', 'f'):
14594       _codec ("TIFF still images");
14595       caps = gst_caps_new_empty_simple ("image/tiff");
14596       break;
14597     case GST_MAKE_FOURCC ('i', 'c', 'o', 'd'):
14598       _codec ("Apple Intermediate Codec");
14599       caps = gst_caps_from_string ("video/x-apple-intermediate-codec");
14600       break;
14601     case GST_MAKE_FOURCC ('A', 'V', 'd', 'n'):
14602       _codec ("AVID DNxHD");
14603       caps = gst_caps_from_string ("video/x-dnxhd");
14604       break;
14605     case FOURCC_VP80:
14606     case FOURCC_vp08:
14607       _codec ("On2 VP8");
14608       caps = gst_caps_from_string ("video/x-vp8");
14609       break;
14610     case FOURCC_vp09:
14611       _codec ("Google VP9");
14612       caps = gst_caps_from_string ("video/x-vp9");
14613       break;
14614     case FOURCC_apcs:
14615       _codec ("Apple ProRes LT");
14616       caps =
14617           gst_caps_new_simple ("video/x-prores", "variant", G_TYPE_STRING, "lt",
14618           NULL);
14619       break;
14620     case FOURCC_apch:
14621       _codec ("Apple ProRes HQ");
14622       caps =
14623           gst_caps_new_simple ("video/x-prores", "variant", G_TYPE_STRING, "hq",
14624           NULL);
14625       break;
14626     case FOURCC_apcn:
14627       _codec ("Apple ProRes");
14628       caps =
14629           gst_caps_new_simple ("video/x-prores", "variant", G_TYPE_STRING,
14630           "standard", NULL);
14631       break;
14632     case FOURCC_apco:
14633       _codec ("Apple ProRes Proxy");
14634       caps =
14635           gst_caps_new_simple ("video/x-prores", "variant", G_TYPE_STRING,
14636           "proxy", NULL);
14637       break;
14638     case FOURCC_ap4h:
14639       _codec ("Apple ProRes 4444");
14640       caps =
14641           gst_caps_new_simple ("video/x-prores", "variant", G_TYPE_STRING,
14642           "4444", NULL);
14643
14644       /* 24 bits per sample = an alpha channel is coded but image is always opaque */
14645       if (entry->bits_per_sample > 0) {
14646         gst_caps_set_simple (caps, "depth", G_TYPE_INT, entry->bits_per_sample,
14647             NULL);
14648       }
14649       break;
14650     case FOURCC_ap4x:
14651       _codec ("Apple ProRes 4444 XQ");
14652       caps =
14653           gst_caps_new_simple ("video/x-prores", "variant", G_TYPE_STRING,
14654           "4444xq", NULL);
14655
14656       /* 24 bits per sample = an alpha channel is coded but image is always opaque */
14657       if (entry->bits_per_sample > 0) {
14658         gst_caps_set_simple (caps, "depth", G_TYPE_INT, entry->bits_per_sample,
14659             NULL);
14660       }
14661       break;
14662     case FOURCC_cfhd:
14663       _codec ("GoPro CineForm");
14664       caps = gst_caps_from_string ("video/x-cineform");
14665       break;
14666     case FOURCC_vc_1:
14667     case FOURCC_ovc1:
14668       _codec ("VC-1");
14669       caps = gst_caps_new_simple ("video/x-wmv",
14670           "wmvversion", G_TYPE_INT, 3, "format", G_TYPE_STRING, "WVC1", NULL);
14671       break;
14672     case FOURCC_av01:
14673       _codec ("AV1");
14674       caps = gst_caps_new_empty_simple ("video/x-av1");
14675       break;
14676     case GST_MAKE_FOURCC ('k', 'p', 'c', 'd'):
14677     default:
14678     {
14679       caps = _get_unknown_codec_name ("video", fourcc);
14680       break;
14681     }
14682   }
14683
14684   if (format != GST_VIDEO_FORMAT_UNKNOWN) {
14685     GstVideoInfo info;
14686
14687     gst_video_info_init (&info);
14688     gst_video_info_set_format (&info, format, entry->width, entry->height);
14689
14690     caps = gst_video_info_to_caps (&info);
14691     *codec_name = gst_pb_utils_get_codec_description (caps);
14692
14693     /* enable clipping for raw video streams */
14694     stream->need_clip = TRUE;
14695     stream->alignment = 32;
14696   }
14697
14698   return caps;
14699 }
14700
14701 static guint
14702 round_up_pow2 (guint n)
14703 {
14704   n = n - 1;
14705   n = n | (n >> 1);
14706   n = n | (n >> 2);
14707   n = n | (n >> 4);
14708   n = n | (n >> 8);
14709   n = n | (n >> 16);
14710   return n + 1;
14711 }
14712
14713 static GstCaps *
14714 qtdemux_audio_caps (GstQTDemux * qtdemux, QtDemuxStream * stream,
14715     QtDemuxStreamStsdEntry * entry, guint32 fourcc, const guint8 * data,
14716     int len, gchar ** codec_name)
14717 {
14718   GstCaps *caps;
14719   const GstStructure *s;
14720   const gchar *name;
14721   gint endian = 0;
14722   GstAudioFormat format = 0;
14723   gint depth;
14724
14725   GST_DEBUG_OBJECT (qtdemux, "resolve fourcc 0x%08x", GUINT32_TO_BE (fourcc));
14726
14727   depth = entry->bytes_per_packet * 8;
14728
14729   switch (fourcc) {
14730     case GST_MAKE_FOURCC ('N', 'O', 'N', 'E'):
14731     case FOURCC_raw_:
14732       /* 8-bit audio is unsigned */
14733       if (depth == 8)
14734         format = GST_AUDIO_FORMAT_U8;
14735       /* otherwise it's signed and big-endian just like 'twos' */
14736     case FOURCC_twos:
14737       endian = G_BIG_ENDIAN;
14738       /* fall-through */
14739     case FOURCC_sowt:
14740     {
14741       gchar *str;
14742
14743       if (!endian)
14744         endian = G_LITTLE_ENDIAN;
14745
14746       if (!format)
14747         format = gst_audio_format_build_integer (TRUE, endian, depth, depth);
14748
14749       str = g_strdup_printf ("Raw %d-bit PCM audio", depth);
14750       _codec (str);
14751       g_free (str);
14752
14753       caps = gst_caps_new_simple ("audio/x-raw",
14754           "format", G_TYPE_STRING, gst_audio_format_to_string (format),
14755           "layout", G_TYPE_STRING, "interleaved", NULL);
14756       stream->alignment = GST_ROUND_UP_8 (depth);
14757       stream->alignment = round_up_pow2 (stream->alignment);
14758       break;
14759     }
14760     case FOURCC_fl64:
14761       _codec ("Raw 64-bit floating-point audio");
14762       /* we assume BIG ENDIAN, an enda box will tell us to change this to little
14763        * endian later */
14764       caps = gst_caps_new_simple ("audio/x-raw",
14765           "format", G_TYPE_STRING, "F64BE",
14766           "layout", G_TYPE_STRING, "interleaved", NULL);
14767       stream->alignment = 8;
14768       break;
14769     case FOURCC_fl32:
14770       _codec ("Raw 32-bit floating-point audio");
14771       /* we assume BIG ENDIAN, an enda box will tell us to change this to little
14772        * endian later */
14773       caps = gst_caps_new_simple ("audio/x-raw",
14774           "format", G_TYPE_STRING, "F32BE",
14775           "layout", G_TYPE_STRING, "interleaved", NULL);
14776       stream->alignment = 4;
14777       break;
14778     case FOURCC_in24:
14779       _codec ("Raw 24-bit PCM audio");
14780       /* we assume BIG ENDIAN, an enda box will tell us to change this to little
14781        * endian later */
14782       caps = gst_caps_new_simple ("audio/x-raw",
14783           "format", G_TYPE_STRING, "S24BE",
14784           "layout", G_TYPE_STRING, "interleaved", NULL);
14785       stream->alignment = 4;
14786       break;
14787     case FOURCC_in32:
14788       _codec ("Raw 32-bit PCM audio");
14789       /* we assume BIG ENDIAN, an enda box will tell us to change this to little
14790        * endian later */
14791       caps = gst_caps_new_simple ("audio/x-raw",
14792           "format", G_TYPE_STRING, "S32BE",
14793           "layout", G_TYPE_STRING, "interleaved", NULL);
14794       stream->alignment = 4;
14795       break;
14796     case FOURCC_s16l:
14797       _codec ("Raw 16-bit PCM audio");
14798       caps = gst_caps_new_simple ("audio/x-raw",
14799           "format", G_TYPE_STRING, "S16LE",
14800           "layout", G_TYPE_STRING, "interleaved", NULL);
14801       stream->alignment = 2;
14802       break;
14803     case FOURCC_ulaw:
14804       _codec ("Mu-law audio");
14805       caps = gst_caps_new_empty_simple ("audio/x-mulaw");
14806       break;
14807     case FOURCC_alaw:
14808       _codec ("A-law audio");
14809       caps = gst_caps_new_empty_simple ("audio/x-alaw");
14810       break;
14811     case 0x0200736d:
14812     case 0x6d730002:
14813       _codec ("Microsoft ADPCM");
14814       /* Microsoft ADPCM-ACM code 2 */
14815       caps = gst_caps_new_simple ("audio/x-adpcm",
14816           "layout", G_TYPE_STRING, "microsoft", NULL);
14817       break;
14818     case 0x1100736d:
14819     case 0x6d730011:
14820       _codec ("DVI/IMA ADPCM");
14821       caps = gst_caps_new_simple ("audio/x-adpcm",
14822           "layout", G_TYPE_STRING, "dvi", NULL);
14823       break;
14824     case 0x1700736d:
14825     case 0x6d730017:
14826       _codec ("DVI/Intel IMA ADPCM");
14827       /* FIXME DVI/Intel IMA ADPCM/ACM code 17 */
14828       caps = gst_caps_new_simple ("audio/x-adpcm",
14829           "layout", G_TYPE_STRING, "quicktime", NULL);
14830       break;
14831     case 0x5500736d:
14832     case 0x6d730055:
14833       /* MPEG layer 3, CBR only (pre QT4.1) */
14834     case FOURCC__mp3:
14835     case FOURCC_mp3_:
14836       _codec ("MPEG-1 layer 3");
14837       /* MPEG layer 3, CBR & VBR (QT4.1 and later) */
14838       caps = gst_caps_new_simple ("audio/mpeg", "layer", G_TYPE_INT, 3,
14839           "mpegversion", G_TYPE_INT, 1, NULL);
14840       break;
14841     case GST_MAKE_FOURCC ('.', 'm', 'p', '2'):
14842       _codec ("MPEG-1 layer 2");
14843       /* MPEG layer 2 */
14844       caps = gst_caps_new_simple ("audio/mpeg", "layer", G_TYPE_INT, 2,
14845           "mpegversion", G_TYPE_INT, 1, NULL);
14846       break;
14847     case 0x20736d:
14848     case GST_MAKE_FOURCC ('e', 'c', '-', '3'):
14849       _codec ("EAC-3 audio");
14850       caps = gst_caps_new_simple ("audio/x-eac3",
14851           "framed", G_TYPE_BOOLEAN, TRUE, NULL);
14852       entry->sampled = TRUE;
14853       break;
14854     case GST_MAKE_FOURCC ('s', 'a', 'c', '3'): // Nero Recode
14855     case FOURCC_ac_3:
14856       _codec ("AC-3 audio");
14857       caps = gst_caps_new_simple ("audio/x-ac3",
14858           "framed", G_TYPE_BOOLEAN, TRUE, NULL);
14859       entry->sampled = TRUE;
14860       break;
14861     case GST_MAKE_FOURCC ('d', 't', 's', 'c'):
14862     case GST_MAKE_FOURCC ('D', 'T', 'S', ' '):
14863       _codec ("DTS audio");
14864       caps = gst_caps_new_simple ("audio/x-dts",
14865           "framed", G_TYPE_BOOLEAN, TRUE, NULL);
14866       entry->sampled = TRUE;
14867       break;
14868     case GST_MAKE_FOURCC ('d', 't', 's', 'h'): // DTS-HD
14869     case GST_MAKE_FOURCC ('d', 't', 's', 'l'): // DTS-HD Lossless
14870       _codec ("DTS-HD audio");
14871       caps = gst_caps_new_simple ("audio/x-dts",
14872           "framed", G_TYPE_BOOLEAN, TRUE, NULL);
14873       entry->sampled = TRUE;
14874       break;
14875     case FOURCC_MAC3:
14876       _codec ("MACE-3");
14877       caps = gst_caps_new_simple ("audio/x-mace",
14878           "maceversion", G_TYPE_INT, 3, NULL);
14879       break;
14880     case FOURCC_MAC6:
14881       _codec ("MACE-6");
14882       caps = gst_caps_new_simple ("audio/x-mace",
14883           "maceversion", G_TYPE_INT, 6, NULL);
14884       break;
14885     case GST_MAKE_FOURCC ('O', 'g', 'g', 'V'):
14886       /* ogg/vorbis */
14887       caps = gst_caps_new_empty_simple ("application/ogg");
14888       break;
14889     case GST_MAKE_FOURCC ('d', 'v', 'c', 'a'):
14890       _codec ("DV audio");
14891       caps = gst_caps_new_empty_simple ("audio/x-dv");
14892       break;
14893     case FOURCC_mp4a:
14894       _codec ("MPEG-4 AAC audio");
14895       caps = gst_caps_new_simple ("audio/mpeg",
14896           "mpegversion", G_TYPE_INT, 4, "framed", G_TYPE_BOOLEAN, TRUE,
14897           "stream-format", G_TYPE_STRING, "raw", NULL);
14898       break;
14899     case GST_MAKE_FOURCC ('Q', 'D', 'M', 'C'):
14900       _codec ("QDesign Music");
14901       caps = gst_caps_new_empty_simple ("audio/x-qdm");
14902       break;
14903     case FOURCC_QDM2:
14904       _codec ("QDesign Music v.2");
14905       /* FIXME: QDesign music version 2 (no constant) */
14906       if (FALSE && data) {
14907         caps = gst_caps_new_simple ("audio/x-qdm2",
14908             "framesize", G_TYPE_INT, QT_UINT32 (data + 52),
14909             "bitrate", G_TYPE_INT, QT_UINT32 (data + 40),
14910             "blocksize", G_TYPE_INT, QT_UINT32 (data + 44), NULL);
14911       } else {
14912         caps = gst_caps_new_empty_simple ("audio/x-qdm2");
14913       }
14914       break;
14915     case FOURCC_agsm:
14916       _codec ("GSM audio");
14917       caps = gst_caps_new_empty_simple ("audio/x-gsm");
14918       break;
14919     case FOURCC_samr:
14920       _codec ("AMR audio");
14921       caps = gst_caps_new_empty_simple ("audio/AMR");
14922       break;
14923     case FOURCC_sawb:
14924       _codec ("AMR-WB audio");
14925       caps = gst_caps_new_empty_simple ("audio/AMR-WB");
14926       break;
14927     case FOURCC_ima4:
14928       _codec ("Quicktime IMA ADPCM");
14929       caps = gst_caps_new_simple ("audio/x-adpcm",
14930           "layout", G_TYPE_STRING, "quicktime", NULL);
14931       break;
14932     case FOURCC_alac:
14933       _codec ("Apple lossless audio");
14934       caps = gst_caps_new_empty_simple ("audio/x-alac");
14935       break;
14936     case FOURCC_fLaC:
14937       _codec ("Free Lossless Audio Codec");
14938       caps = gst_caps_new_simple ("audio/x-flac",
14939           "framed", G_TYPE_BOOLEAN, TRUE, NULL);
14940       break;
14941     case GST_MAKE_FOURCC ('Q', 'c', 'l', 'p'):
14942       _codec ("QualComm PureVoice");
14943       caps = gst_caps_from_string ("audio/qcelp");
14944       break;
14945     case FOURCC_wma_:
14946     case FOURCC_owma:
14947       _codec ("WMA");
14948       caps = gst_caps_new_empty_simple ("audio/x-wma");
14949       break;
14950     case FOURCC_opus:
14951       _codec ("Opus");
14952       caps = gst_caps_new_empty_simple ("audio/x-opus");
14953       break;
14954     case FOURCC_lpcm:
14955     {
14956       guint32 flags = 0;
14957       guint32 depth = 0;
14958       guint32 width = 0;
14959       GstAudioFormat format;
14960       enum
14961       {
14962         FLAG_IS_FLOAT = 0x1,
14963         FLAG_IS_BIG_ENDIAN = 0x2,
14964         FLAG_IS_SIGNED = 0x4,
14965         FLAG_IS_PACKED = 0x8,
14966         FLAG_IS_ALIGNED_HIGH = 0x10,
14967         FLAG_IS_NON_INTERLEAVED = 0x20
14968       };
14969       _codec ("Raw LPCM audio");
14970
14971       if (data && len >= 36) {
14972         depth = QT_UINT32 (data + 24);
14973         flags = QT_UINT32 (data + 28);
14974         width = QT_UINT32 (data + 32) * 8 / entry->n_channels;
14975       }
14976       if ((flags & FLAG_IS_FLOAT) == 0) {
14977         if (depth == 0)
14978           depth = 16;
14979         if (width == 0)
14980           width = 16;
14981         if ((flags & FLAG_IS_ALIGNED_HIGH))
14982           depth = width;
14983
14984         format = gst_audio_format_build_integer ((flags & FLAG_IS_SIGNED) ?
14985             TRUE : FALSE, (flags & FLAG_IS_BIG_ENDIAN) ?
14986             G_BIG_ENDIAN : G_LITTLE_ENDIAN, width, depth);
14987         caps = gst_caps_new_simple ("audio/x-raw",
14988             "format", G_TYPE_STRING,
14989             format !=
14990             GST_AUDIO_FORMAT_UNKNOWN ? gst_audio_format_to_string (format) :
14991             "UNKNOWN", "layout", G_TYPE_STRING,
14992             (flags & FLAG_IS_NON_INTERLEAVED) ? "non-interleaved" :
14993             "interleaved", NULL);
14994         stream->alignment = GST_ROUND_UP_8 (depth);
14995         stream->alignment = round_up_pow2 (stream->alignment);
14996       } else {
14997         if (width == 0)
14998           width = 32;
14999         if (width == 64) {
15000           if (flags & FLAG_IS_BIG_ENDIAN)
15001             format = GST_AUDIO_FORMAT_F64BE;
15002           else
15003             format = GST_AUDIO_FORMAT_F64LE;
15004         } else {
15005           if (flags & FLAG_IS_BIG_ENDIAN)
15006             format = GST_AUDIO_FORMAT_F32BE;
15007           else
15008             format = GST_AUDIO_FORMAT_F32LE;
15009         }
15010         caps = gst_caps_new_simple ("audio/x-raw",
15011             "format", G_TYPE_STRING, gst_audio_format_to_string (format),
15012             "layout", G_TYPE_STRING, (flags & FLAG_IS_NON_INTERLEAVED) ?
15013             "non-interleaved" : "interleaved", NULL);
15014         stream->alignment = width / 8;
15015       }
15016       break;
15017     }
15018     case GST_MAKE_FOURCC ('a', 'c', '-', '4'):
15019     {
15020       _codec ("AC4");
15021       caps = gst_caps_new_empty_simple ("audio/x-ac4");
15022       break;
15023     }
15024     case GST_MAKE_FOURCC ('q', 't', 'v', 'r'):
15025       /* ? */
15026     default:
15027     {
15028       caps = _get_unknown_codec_name ("audio", fourcc);
15029       break;
15030     }
15031   }
15032
15033   if (caps) {
15034     GstCaps *templ_caps =
15035         gst_static_pad_template_get_caps (&gst_qtdemux_audiosrc_template);
15036     GstCaps *intersection = gst_caps_intersect (caps, templ_caps);
15037     gst_caps_unref (caps);
15038     gst_caps_unref (templ_caps);
15039     caps = intersection;
15040   }
15041
15042   /* enable clipping for raw audio streams */
15043   s = gst_caps_get_structure (caps, 0);
15044   name = gst_structure_get_name (s);
15045   if (g_str_has_prefix (name, "audio/x-raw")) {
15046     stream->need_clip = TRUE;
15047     stream->min_buffer_size = 1024 * entry->bytes_per_frame;
15048     stream->max_buffer_size = 4096 * entry->bytes_per_frame;
15049     GST_DEBUG ("setting min/max buffer sizes to %d/%d", stream->min_buffer_size,
15050         stream->max_buffer_size);
15051   }
15052   return caps;
15053 }
15054
15055 static GstCaps *
15056 qtdemux_sub_caps (GstQTDemux * qtdemux, QtDemuxStream * stream,
15057     QtDemuxStreamStsdEntry * entry, guint32 fourcc,
15058     const guint8 * stsd_entry_data, gchar ** codec_name)
15059 {
15060   GstCaps *caps;
15061
15062   GST_DEBUG_OBJECT (qtdemux, "resolve fourcc 0x%08x", GUINT32_TO_BE (fourcc));
15063
15064   switch (fourcc) {
15065     case FOURCC_mp4s:
15066       _codec ("DVD subtitle");
15067       caps = gst_caps_new_empty_simple ("subpicture/x-dvd");
15068       stream->process_func = gst_qtdemux_process_buffer_dvd;
15069       break;
15070     case FOURCC_text:
15071       _codec ("Quicktime timed text");
15072       goto text;
15073     case FOURCC_tx3g:
15074       _codec ("3GPP timed text");
15075     text:
15076       caps = gst_caps_new_simple ("text/x-raw", "format", G_TYPE_STRING,
15077           "utf8", NULL);
15078       /* actual text piece needs to be extracted */
15079       stream->process_func = gst_qtdemux_process_buffer_text;
15080       break;
15081     case FOURCC_stpp:
15082       _codec ("XML subtitles");
15083       caps = gst_caps_new_empty_simple ("application/ttml+xml");
15084       break;
15085     case FOURCC_wvtt:
15086     {
15087       GstBuffer *buffer;
15088       const gchar *buf = "WEBVTT\n\n";
15089
15090       _codec ("WebVTT subtitles");
15091       caps = gst_caps_new_empty_simple ("application/x-subtitle-vtt");
15092       stream->process_func = gst_qtdemux_process_buffer_wvtt;
15093
15094       /* FIXME: Parse the vttC atom and get the entire WEBVTT header */
15095       buffer = gst_buffer_new_and_alloc (8);
15096       gst_buffer_fill (buffer, 0, buf, 8);
15097       stream->buffers = g_slist_append (stream->buffers, buffer);
15098
15099       break;
15100     }
15101     case FOURCC_c608:
15102       _codec ("CEA 608 Closed Caption");
15103       caps =
15104           gst_caps_new_simple ("closedcaption/x-cea-608", "format",
15105           G_TYPE_STRING, "s334-1a", NULL);
15106       stream->process_func = gst_qtdemux_process_buffer_clcp;
15107       stream->need_split = TRUE;
15108       break;
15109     case FOURCC_c708:
15110       _codec ("CEA 708 Closed Caption");
15111       caps =
15112           gst_caps_new_simple ("closedcaption/x-cea-708", "format",
15113           G_TYPE_STRING, "cdp", NULL);
15114       stream->process_func = gst_qtdemux_process_buffer_clcp;
15115       break;
15116
15117     default:
15118     {
15119       caps = _get_unknown_codec_name ("text", fourcc);
15120       break;
15121     }
15122   }
15123   return caps;
15124 }
15125
15126 static GstCaps *
15127 qtdemux_generic_caps (GstQTDemux * qtdemux, QtDemuxStream * stream,
15128     QtDemuxStreamStsdEntry * entry, guint32 fourcc,
15129     const guint8 * stsd_entry_data, gchar ** codec_name)
15130 {
15131   GstCaps *caps;
15132
15133   switch (fourcc) {
15134     case FOURCC_m1v:
15135       _codec ("MPEG 1 video");
15136       caps = gst_caps_new_simple ("video/mpeg", "mpegversion", G_TYPE_INT, 1,
15137           "systemstream", G_TYPE_BOOLEAN, FALSE, NULL);
15138       break;
15139     default:
15140       caps = NULL;
15141       break;
15142   }
15143   return caps;
15144 }
15145
15146 static void
15147 gst_qtdemux_append_protection_system_id (GstQTDemux * qtdemux,
15148     const gchar * system_id)
15149 {
15150   gint i;
15151
15152   if (!qtdemux->protection_system_ids)
15153     qtdemux->protection_system_ids =
15154         g_ptr_array_new_with_free_func ((GDestroyNotify) g_free);
15155   /* Check whether we already have an entry for this system ID. */
15156   for (i = 0; i < qtdemux->protection_system_ids->len; ++i) {
15157     const gchar *id = g_ptr_array_index (qtdemux->protection_system_ids, i);
15158     if (g_ascii_strcasecmp (system_id, id) == 0) {
15159       return;
15160     }
15161   }
15162   GST_DEBUG_OBJECT (qtdemux, "Adding cenc protection system ID %s", system_id);
15163   g_ptr_array_add (qtdemux->protection_system_ids, g_ascii_strdown (system_id,
15164           -1));
15165 }