gst/realmedia/rmdemux.c

   1 /* GStreamer RealMedia demuxer
   2  * Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
   3  * Copyright (C) <2003> David A. Schleef <ds@schleef.org>
   4  * Copyright (C) <2004> Stephane Loeuillet <gstreamer@leroutier.net>
   5  * Copyright (C) <2005> Owen Fraser-Green <owen@discobabe.net>
   6  * Copyright (C) <2005> Michael Smith <fluendo.com>
   7  * Copyright (C) <2006> Wim Taymans <wim@fluendo.com>
   8  * Copyright (C) <2006> Tim-Philipp Müller <tim centricular net>
   9  * Copyright (C) <2007> Wim Taymans <wim.taymans@gmail.com>
  10  *
  11  * This library is free software; you can redistribute it and/or
  12  * modify it under the terms of the GNU Library General Public
  13  * License as published by the Free Software Foundation; either
  14  * version 2 of the License, or (at your option) any later version.
  15  *
  16  * This library is distributed in the hope that it will be useful,
  17  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  19  * Library General Public License for more details.
  20  *
  21  * You should have received a copy of the GNU Library General Public
  22  * License along with this library; if not, write to the
  23  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
  24  * Boston, MA 02110-1301, USA.
  25  */
  26
  27 #ifdef HAVE_CONFIG_H
  28 #  include "config.h"
  29 #endif
  30
  31 #include "rmdemux.h"
  32 #include "rmutils.h"
  33
  34 #include <string.h>
  35 #include <ctype.h>
  36
  37 #define RMDEMUX_GUINT32_GET(a)  GST_READ_UINT32_BE(a)
  38 #define RMDEMUX_GUINT16_GET(a)  GST_READ_UINT16_BE(a)
  39 #define RMDEMUX_FOURCC_GET(a)   GST_READ_UINT32_LE(a)
  40 #define HEADER_SIZE 10
  41 #define DATA_SIZE 8
  42
  43 #define MAX_FRAGS 256
  44
  45 static const guint8 sipr_subpk_size[4] = { 29, 19, 37, 20 };
  46
  47 typedef struct _GstRMDemuxIndex GstRMDemuxIndex;
  48
  49 struct _GstRMDemuxStream
  50 {
  51   guint32 subtype;
  52   guint32 fourcc;
  53   guint32 subformat;
  54   guint32 format;
  55
  56   int id;
  57   GstPad *pad;
  58   gboolean discont;
  59   int timescale;
  60
  61   int sample_index;
  62   GstRMDemuxIndex *index;
  63   int index_length;
  64   gint framerate_numerator;
  65   gint framerate_denominator;
  66   guint32 seek_offset;
  67
  68   guint16 width;
  69   guint16 height;
  70   guint16 flavor;
  71   guint16 rate;                 /* samplerate         */
  72   guint16 n_channels;           /* channels           */
  73   guint16 sample_width;         /* bits_per_sample    */
  74   guint16 leaf_size;            /* subpacket_size     */
  75   guint32 packet_size;          /* coded_frame_size   */
  76   guint16 version;
  77   guint32 extra_data_size;      /* codec_data_length  */
  78   guint8 *extra_data;           /* extras             */
  79   guint32 bitrate;
  80
  81   gboolean needs_descrambling;
  82   guint subpackets_needed;      /* subpackets needed for descrambling    */
  83   GPtrArray *subpackets;        /* array containing subpacket GstBuffers */
  84
  85   /* Variables needed for fixing timestamps. */
  86   GstClockTime next_ts, last_ts;
  87   guint16 next_seq, last_seq;
  88
  89   gint frag_seqnum;
  90   gint frag_subseq;
  91   guint frag_length;
  92   guint frag_current;
  93   guint frag_count;
  94   guint frag_offset[MAX_FRAGS];
  95   GstAdapter *adapter;
  96
  97   GstTagList *pending_tags;
  98 };
  99
 100 struct _GstRMDemuxIndex
 101 {
 102   guint32 offset;
 103   GstClockTime timestamp;
 104 };
 105
 106 static GstStaticPadTemplate gst_rmdemux_sink_template =
 107 GST_STATIC_PAD_TEMPLATE ("sink",
 108     GST_PAD_SINK,
 109     GST_PAD_ALWAYS,
 110     GST_STATIC_CAPS ("application/vnd.rn-realmedia")
 111     );
 112
 113 static GstStaticPadTemplate gst_rmdemux_videosrc_template =
 114 GST_STATIC_PAD_TEMPLATE ("video_%u",
 115     GST_PAD_SRC,
 116     GST_PAD_SOMETIMES,
 117     GST_STATIC_CAPS_ANY);
 118
 119 static GstStaticPadTemplate gst_rmdemux_audiosrc_template =
 120 GST_STATIC_PAD_TEMPLATE ("audio_%u",
 121     GST_PAD_SRC,
 122     GST_PAD_SOMETIMES,
 123     GST_STATIC_CAPS_ANY);
 124
 125 GST_DEBUG_CATEGORY_STATIC (rmdemux_debug);
 126 #define GST_CAT_DEFAULT rmdemux_debug
 127
 128 static GstElementClass *parent_class = NULL;
 129
 130 static void gst_rmdemux_class_init (GstRMDemuxClass * klass);
 131 static void gst_rmdemux_base_init (GstRMDemuxClass * klass);
 132 static void gst_rmdemux_init (GstRMDemux * rmdemux);
 133 static void gst_rmdemux_finalize (GObject * object);
 134 static GstStateChangeReturn gst_rmdemux_change_state (GstElement * element,
 135     GstStateChange transition);
 136 static GstFlowReturn gst_rmdemux_chain (GstPad * pad, GstObject * parent,
 137     GstBuffer * buffer);
 138 static void gst_rmdemux_loop (GstPad * pad);
 139 static gboolean gst_rmdemux_sink_activate (GstPad * sinkpad,
 140     GstObject * parent);
 141 static gboolean gst_rmdemux_sink_activate_mode (GstPad * sinkpad,
 142     GstObject * parent, GstPadMode mode, gboolean active);
 143 static gboolean gst_rmdemux_sink_event (GstPad * pad, GstObject * parent,
 144     GstEvent * event);
 145 static gboolean gst_rmdemux_src_event (GstPad * pad, GstObject * parent,
 146     GstEvent * event);
 147 static void gst_rmdemux_send_event (GstRMDemux * rmdemux, GstEvent * event);
 148 static gboolean gst_rmdemux_src_query (GstPad * pad, GstObject * parent,
 149     GstQuery * query);
 150 static gboolean gst_rmdemux_perform_seek (GstRMDemux * rmdemux,
 151     GstEvent * event);
 152
 153 static void gst_rmdemux_parse__rmf (GstRMDemux * rmdemux, const guint8 * data,
 154     int length);
 155 static void gst_rmdemux_parse_prop (GstRMDemux * rmdemux, const guint8 * data,
 156     int length);
 157 static void gst_rmdemux_parse_mdpr (GstRMDemux * rmdemux,
 158     const guint8 * data, int length);
 159 static guint gst_rmdemux_parse_indx (GstRMDemux * rmdemux, const guint8 * data,
 160     int length);
 161 static void gst_rmdemux_parse_data (GstRMDemux * rmdemux, const guint8 * data,
 162     int length);
 163 static void gst_rmdemux_parse_cont (GstRMDemux * rmdemux, const guint8 * data,
 164     int length);
 165 static GstFlowReturn gst_rmdemux_parse_packet (GstRMDemux * rmdemux,
 166     GstBuffer * in, guint16 version);
 167 static void gst_rmdemux_parse_indx_data (GstRMDemux * rmdemux,
 168     const guint8 * data, int length);
 169 static void gst_rmdemux_stream_clear_cached_subpackets (GstRMDemux * rmdemux,
 170     GstRMDemuxStream * stream);
 171 static GstRMDemuxStream *gst_rmdemux_get_stream_by_id (GstRMDemux * rmdemux,
 172     int id);
 173
 174 static GType
 175 gst_rmdemux_get_type (void)
 176 {
 177   static GType rmdemux_type = 0;
 178
 179   if (!rmdemux_type) {
 180     static const GTypeInfo rmdemux_info = {
 181       sizeof (GstRMDemuxClass),
 182       (GBaseInitFunc) gst_rmdemux_base_init, NULL,
 183       (GClassInitFunc) gst_rmdemux_class_init,
 184       NULL, NULL, sizeof (GstRMDemux), 0,
 185       (GInstanceInitFunc) gst_rmdemux_init,
 186     };
 187
 188     rmdemux_type =
 189         g_type_register_static (GST_TYPE_ELEMENT, "GstRMDemux", &rmdemux_info,
 190         0);
 191   }
 192   return rmdemux_type;
 193 }
 194
 195 static void
 196 gst_rmdemux_base_init (GstRMDemuxClass * klass)
 197 {
 198   GstElementClass *element_class = GST_ELEMENT_CLASS (klass);
 199
 200   gst_element_class_add_static_pad_template (element_class,
 201       &gst_rmdemux_sink_template);
 202   gst_element_class_add_static_pad_template (element_class,
 203       &gst_rmdemux_videosrc_template);
 204   gst_element_class_add_static_pad_template (element_class,
 205       &gst_rmdemux_audiosrc_template);
 206   gst_element_class_set_static_metadata (element_class, "RealMedia Demuxer",
 207       "Codec/Demuxer",
 208       "Demultiplex a RealMedia file into audio and video streams",
 209       "David Schleef <ds@schleef.org>");
 210 }
 211
 212 static void
 213 gst_rmdemux_class_init (GstRMDemuxClass * klass)
 214 {
 215   GObjectClass *gobject_class;
 216   GstElementClass *gstelement_class;
 217
 218   gobject_class = (GObjectClass *) klass;
 219   gstelement_class = (GstElementClass *) klass;
 220
 221   parent_class = g_type_class_peek_parent (klass);
 222
 223   gstelement_class->change_state = GST_DEBUG_FUNCPTR (gst_rmdemux_change_state);
 224
 225   GST_DEBUG_CATEGORY_INIT (rmdemux_debug, "rmdemux",
 226       0, "Demuxer for Realmedia streams");
 227
 228   gobject_class->finalize = gst_rmdemux_finalize;
 229 }
 230
 231 static void
 232 gst_rmdemux_finalize (GObject * object)
 233 {
 234   GstRMDemux *rmdemux = GST_RMDEMUX (object);
 235
 236   if (rmdemux->adapter) {
 237     g_object_unref (rmdemux->adapter);
 238     rmdemux->adapter = NULL;
 239   }
 240   if (rmdemux->flowcombiner) {
 241     gst_flow_combiner_free (rmdemux->flowcombiner);
 242     rmdemux->flowcombiner = NULL;
 243   }
 244
 245   GST_CALL_PARENT (G_OBJECT_CLASS, finalize, (object));
 246 }
 247
 248 static void
 249 gst_rmdemux_init (GstRMDemux * rmdemux)
 250 {
 251   rmdemux->sinkpad =
 252       gst_pad_new_from_static_template (&gst_rmdemux_sink_template, "sink");
 253   gst_pad_set_event_function (rmdemux->sinkpad,
 254       GST_DEBUG_FUNCPTR (gst_rmdemux_sink_event));
 255   gst_pad_set_chain_function (rmdemux->sinkpad,
 256       GST_DEBUG_FUNCPTR (gst_rmdemux_chain));
 257   gst_pad_set_activate_function (rmdemux->sinkpad,
 258       GST_DEBUG_FUNCPTR (gst_rmdemux_sink_activate));
 259   gst_pad_set_activatemode_function (rmdemux->sinkpad,
 260       GST_DEBUG_FUNCPTR (gst_rmdemux_sink_activate_mode));
 261
 262   gst_element_add_pad (GST_ELEMENT (rmdemux), rmdemux->sinkpad);
 263
 264   rmdemux->adapter = gst_adapter_new ();
 265   rmdemux->first_ts = GST_CLOCK_TIME_NONE;
 266   rmdemux->base_ts = GST_CLOCK_TIME_NONE;
 267   rmdemux->need_newsegment = TRUE;
 268   rmdemux->have_group_id = FALSE;
 269   rmdemux->group_id = G_MAXUINT;
 270   rmdemux->flowcombiner = gst_flow_combiner_new ();
 271
 272   gst_rm_utils_run_tests ();
 273 }
 274
 275 static gboolean
 276 gst_rmdemux_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
 277 {
 278   gboolean ret;
 279
 280   switch (GST_EVENT_TYPE (event)) {
 281     case GST_EVENT_SEGMENT:
 282       gst_event_unref (event);
 283       ret = TRUE;
 284       break;
 285     default:
 286       ret = gst_pad_event_default (pad, parent, event);
 287       break;
 288   }
 289   return ret;
 290 }
 291
 292 static gboolean
 293 gst_rmdemux_src_event (GstPad * pad, GstObject * parent, GstEvent * event)
 294 {
 295   gboolean ret = TRUE;
 296
 297   GstRMDemux *rmdemux = GST_RMDEMUX (parent);
 298
 299   GST_LOG_OBJECT (rmdemux, "handling src event");
 300
 301   switch (GST_EVENT_TYPE (event)) {
 302     case GST_EVENT_SEEK:
 303     {
 304       gboolean running;
 305
 306       GST_LOG_OBJECT (rmdemux, "Event on src: SEEK");
 307       /* can't seek if we are not seekable, FIXME could pass the
 308        * seek query upstream after converting it to bytes using
 309        * the average bitrate of the stream. */
 310       if (!rmdemux->seekable) {
 311         ret = FALSE;
 312         GST_DEBUG ("seek on non seekable stream");
 313         goto done_unref;
 314       }
 315
 316       GST_OBJECT_LOCK (rmdemux);
 317       /* check if we can do the seek now */
 318       running = rmdemux->running;
 319       GST_OBJECT_UNLOCK (rmdemux);
 320
 321       /* now do the seek */
 322       if (running) {
 323         ret = gst_rmdemux_perform_seek (rmdemux, event);
 324       } else
 325         ret = TRUE;
 326
 327       gst_event_unref (event);
 328       break;
 329     }
 330     default:
 331       GST_LOG_OBJECT (rmdemux, "Event on src: type=%d", GST_EVENT_TYPE (event));
 332       ret = gst_pad_event_default (pad, parent, event);
 333       break;
 334   }
 335
 336   return ret;
 337
 338 done_unref:
 339   GST_DEBUG ("error handling event");
 340   gst_event_unref (event);
 341   return ret;
 342 }
 343
 344 /* Validate that this looks like a reasonable point to seek to */
 345 static gboolean
 346 gst_rmdemux_validate_offset (GstRMDemux * rmdemux)
 347 {
 348   GstBuffer *buffer;
 349   GstFlowReturn flowret;
 350   guint16 version, length;
 351   gboolean ret = TRUE;
 352   GstMapInfo map;
 353
 354   buffer = NULL;
 355   flowret = gst_pad_pull_range (rmdemux->sinkpad, rmdemux->offset, 4, &buffer);
 356
 357   if (flowret != GST_FLOW_OK) {
 358     GST_DEBUG_OBJECT (rmdemux, "Failed to pull data at offset %d",
 359         rmdemux->offset);
 360     return FALSE;
 361   }
 362   /* TODO: Can we also be seeking to a 'DATA' chunk header? Check this.
 363    * Also, for the case we currently handle, can we check any more? It's pretty
 364    * sucky to not be validating a little more heavily than this... */
 365   /* This should now be the start of a data packet header. That begins with
 366    * a 2-byte 'version' field, which has to be 0 or 1, then a length. I'm not
 367    * certain what values are valid for length, but it must always be at least
 368    * 4 bytes, and we can check that it won't take us past our known total size
 369    */
 370
 371   gst_buffer_map (buffer, &map, GST_MAP_READ);
 372   version = RMDEMUX_GUINT16_GET (map.data);
 373   if (version != 0 && version != 1) {
 374     GST_DEBUG_OBJECT (rmdemux, "Expected version 0 or 1, got %d",
 375         (int) version);
 376     ret = FALSE;
 377   }
 378
 379   length = RMDEMUX_GUINT16_GET (map.data + 2);
 380   /* TODO: Also check against total stream length */
 381   if (length < 4) {
 382     GST_DEBUG_OBJECT (rmdemux, "Expected length >= 4, got %d", (int) length);
 383     ret = FALSE;
 384   }
 385   gst_buffer_unmap (buffer, &map);
 386
 387   if (ret) {
 388     rmdemux->offset += 4;
 389     gst_adapter_clear (rmdemux->adapter);
 390     gst_adapter_push (rmdemux->adapter, buffer);
 391   } else {
 392     GST_WARNING_OBJECT (rmdemux, "Failed to validate seek offset at %d",
 393         rmdemux->offset);
 394     gst_buffer_unref (buffer);
 395   }
 396
 397   return ret;
 398 }
 399
 400 static gboolean
 401 find_seek_offset_bytes (GstRMDemux * rmdemux, guint target)
 402 {
 403   int i;
 404   GSList *cur;
 405   gboolean ret = FALSE;
 406
 407   for (cur = rmdemux->streams; cur; cur = cur->next) {
 408     GstRMDemuxStream *stream = cur->data;
 409
 410     /* Search backwards through this stream's index until we find the first
 411      * timestamp before our target time */
 412     for (i = stream->index_length - 1; i >= 0; i--) {
 413       if (stream->index[i].offset <= target) {
 414         /* Set the seek_offset for the stream so we don't bother parsing it
 415          * until we've passed that point */
 416         stream->seek_offset = stream->index[i].offset;
 417         rmdemux->offset = stream->index[i].offset;
 418         ret = TRUE;
 419         break;
 420       }
 421     }
 422   }
 423   return ret;
 424 }
 425
 426 static gboolean
 427 find_seek_offset_time (GstRMDemux * rmdemux, GstClockTime time)
 428 {
 429   int i, n_stream;
 430   gboolean ret = FALSE;
 431   GSList *cur;
 432   GstClockTime earliest = GST_CLOCK_TIME_NONE;
 433
 434   n_stream = 0;
 435   for (cur = rmdemux->streams; cur; cur = cur->next, n_stream++) {
 436     GstRMDemuxStream *stream = cur->data;
 437
 438     /* Search backwards through this stream's index until we find the first
 439      * timestamp before our target time */
 440     for (i = stream->index_length - 1; i >= 0; i--) {
 441       if (stream->index[i].timestamp <= time) {
 442         /* Set the seek_offset for the stream so we don't bother parsing it
 443          * until we've passed that point */
 444         stream->seek_offset = stream->index[i].offset;
 445
 446         /* If it's also the earliest timestamp we've seen of all streams, then
 447          * that's our target!
 448          */
 449         if (earliest == GST_CLOCK_TIME_NONE ||
 450             stream->index[i].timestamp < earliest) {
 451           earliest = stream->index[i].timestamp;
 452           rmdemux->offset = stream->index[i].offset;
 453           GST_DEBUG_OBJECT (rmdemux,
 454               "We're looking for %" GST_TIME_FORMAT
 455               " and we found that stream %d has the latest index at %"
 456               GST_TIME_FORMAT, GST_TIME_ARGS (rmdemux->segment.start), n_stream,
 457               GST_TIME_ARGS (earliest));
 458         }
 459
 460         ret = TRUE;
 461
 462         break;
 463       }
 464     }
 465     stream->discont = TRUE;
 466   }
 467   return ret;
 468 }
 469
 470 static gboolean
 471 gst_rmdemux_perform_seek (GstRMDemux * rmdemux, GstEvent * event)
 472 {
 473   gboolean validated;
 474   gboolean ret = TRUE;
 475   gboolean flush;
 476   GstFormat format;
 477   gdouble rate;
 478   GstSeekFlags flags;
 479   GstSeekType cur_type, stop_type;
 480   gint64 cur, stop;
 481   gboolean update;
 482
 483   if (event) {
 484     GST_DEBUG_OBJECT (rmdemux, "seek with event");
 485
 486     gst_event_parse_seek (event, &rate, &format, &flags,
 487         &cur_type, &cur, &stop_type, &stop);
 488
 489     /* we can only seek on time */
 490     if (format != GST_FORMAT_TIME) {
 491       GST_DEBUG_OBJECT (rmdemux, "can only seek on TIME");
 492       goto error;
 493     }
 494     /* cannot yet do backwards playback */
 495     if (rate <= 0.0) {
 496       GST_DEBUG_OBJECT (rmdemux, "can only seek with positive rate, not %lf",
 497           rate);
 498       goto error;
 499     }
 500   } else {
 501     GST_DEBUG_OBJECT (rmdemux, "seek without event");
 502
 503     flags = 0;
 504     rate = 1.0;
 505   }
 506
 507   GST_DEBUG_OBJECT (rmdemux, "seek, rate %g", rate);
 508
 509   flush = flags & GST_SEEK_FLAG_FLUSH;
 510
 511   /* first step is to unlock the streaming thread if it is
 512    * blocked in a chain call, we do this by starting the flush. */
 513   if (flush) {
 514     gst_pad_push_event (rmdemux->sinkpad, gst_event_new_flush_start ());
 515     gst_rmdemux_send_event (rmdemux, gst_event_new_flush_start ());
 516   } else {
 517     gst_pad_pause_task (rmdemux->sinkpad);
 518   }
 519
 520   GST_LOG_OBJECT (rmdemux, "Done starting flushes");
 521
 522   /* now grab the stream lock so that streaming cannot continue, for
 523    * non flushing seeks when the element is in PAUSED this could block
 524    * forever. */
 525   GST_PAD_STREAM_LOCK (rmdemux->sinkpad);
 526
 527   GST_LOG_OBJECT (rmdemux, "Took streamlock");
 528
 529   if (event) {
 530     gst_segment_do_seek (&rmdemux->segment, rate, format, flags,
 531         cur_type, cur, stop_type, stop, &update);
 532   }
 533
 534   GST_DEBUG_OBJECT (rmdemux, "segment positions set to %" GST_TIME_FORMAT "-%"
 535       GST_TIME_FORMAT, GST_TIME_ARGS (rmdemux->segment.start),
 536       GST_TIME_ARGS (rmdemux->segment.stop));
 537
 538   /* we need to stop flushing on the sinkpad as we're going to use it
 539    * next. We can do this as we have the STREAM lock now. */
 540   gst_pad_push_event (rmdemux->sinkpad, gst_event_new_flush_stop (TRUE));
 541
 542   GST_LOG_OBJECT (rmdemux, "Pushed FLUSH_STOP event");
 543
 544   /* For each stream, find the first index offset equal to or before our seek
 545    * target. Of these, find the smallest offset. That's where we seek to.
 546    *
 547    * Then we pull 4 bytes from that offset, and validate that we've seeked to a
 548    * what looks like a plausible packet.
 549    * If that fails, restart, with the seek target set to one less than the
 550    * offset we just tried. If we run out of places to try, treat that as a fatal
 551    * error.
 552    */
 553   if (!find_seek_offset_time (rmdemux, rmdemux->segment.position)) {
 554     GST_LOG_OBJECT (rmdemux, "Failed to find seek offset by time");
 555     ret = FALSE;
 556     goto done;
 557   }
 558
 559   GST_LOG_OBJECT (rmdemux, "Validating offset %u", rmdemux->offset);
 560   validated = gst_rmdemux_validate_offset (rmdemux);
 561   while (!validated) {
 562     GST_INFO_OBJECT (rmdemux, "Failed to validate offset at %u",
 563         rmdemux->offset);
 564     if (!find_seek_offset_bytes (rmdemux, rmdemux->offset - 1)) {
 565       ret = FALSE;
 566       goto done;
 567     }
 568     validated = gst_rmdemux_validate_offset (rmdemux);
 569   }
 570
 571   GST_LOG_OBJECT (rmdemux, "Found final offset. Excellent!");
 572
 573   /* now we have a new position, prepare for streaming again */
 574   {
 575     /* Reset the demuxer state */
 576     rmdemux->state = RMDEMUX_STATE_DATA_PACKET;
 577
 578     if (flush)
 579       gst_rmdemux_send_event (rmdemux, gst_event_new_flush_stop (TRUE));
 580
 581     /* must send newsegment event from streaming thread, so just set flag */
 582     rmdemux->need_newsegment = TRUE;
 583
 584     /* notify start of new segment */
 585     if (rmdemux->segment.flags & GST_SEEK_FLAG_SEGMENT) {
 586       gst_element_post_message (GST_ELEMENT_CAST (rmdemux),
 587           gst_message_new_segment_start (GST_OBJECT_CAST (rmdemux),
 588               GST_FORMAT_TIME, rmdemux->segment.position));
 589     }
 590
 591     /* restart our task since it might have been stopped when we did the
 592      * flush. */
 593     gst_pad_start_task (rmdemux->sinkpad, (GstTaskFunction) gst_rmdemux_loop,
 594         rmdemux->sinkpad, NULL);
 595   }
 596
 597 done:
 598   /* streaming can continue now */
 599   GST_PAD_STREAM_UNLOCK (rmdemux->sinkpad);
 600
 601   return ret;
 602
 603 error:
 604   {
 605     GST_DEBUG_OBJECT (rmdemux, "seek failed");
 606     return FALSE;
 607   }
 608 }
 609
 610
 611 static gboolean
 612 gst_rmdemux_src_query (GstPad * pad, GstObject * parent, GstQuery * query)
 613 {
 614   gboolean res = FALSE;
 615   GstRMDemux *rmdemux;
 616
 617   rmdemux = GST_RMDEMUX (parent);
 618
 619   switch (GST_QUERY_TYPE (query)) {
 620     case GST_QUERY_POSITION:
 621       GST_DEBUG_OBJECT (rmdemux, "Position query: no idea from demuxer!");
 622       break;
 623     case GST_QUERY_DURATION:{
 624       GstFormat fmt;
 625
 626       gst_query_parse_duration (query, &fmt, NULL);
 627       if (fmt == GST_FORMAT_TIME) {
 628         GST_OBJECT_LOCK (rmdemux);
 629         if (G_LIKELY (rmdemux->running)) {
 630           gst_query_set_duration (query, GST_FORMAT_TIME, rmdemux->duration);
 631           GST_DEBUG_OBJECT (rmdemux, "duration set to %" GST_TIME_FORMAT,
 632               GST_TIME_ARGS (rmdemux->duration));
 633           res = TRUE;
 634         }
 635         GST_OBJECT_UNLOCK (rmdemux);
 636       }
 637       break;
 638     }
 639     case GST_QUERY_SEEKING:{
 640       GstFormat fmt;
 641
 642       gst_query_parse_seeking (query, &fmt, NULL, NULL, NULL);
 643       if (fmt == GST_FORMAT_TIME) {
 644         GST_OBJECT_LOCK (rmdemux);
 645         if (G_LIKELY (rmdemux->running)) {
 646           gst_query_set_seeking (query, GST_FORMAT_TIME, rmdemux->seekable,
 647               0, rmdemux->duration);
 648           res = TRUE;
 649         }
 650         GST_OBJECT_UNLOCK (rmdemux);
 651       }
 652       break;
 653     }
 654     case GST_QUERY_SEGMENT:
 655     {
 656       GstFormat format;
 657       gint64 start, stop;
 658
 659       format = rmdemux->segment.format;
 660
 661       start =
 662           gst_segment_to_stream_time (&rmdemux->segment, format,
 663           rmdemux->segment.start);
 664       if ((stop = rmdemux->segment.stop) == -1)
 665         stop = rmdemux->segment.duration;
 666       else
 667         stop = gst_segment_to_stream_time (&rmdemux->segment, format, stop);
 668
 669       gst_query_set_segment (query, rmdemux->segment.rate, format, start, stop);
 670       res = TRUE;
 671       break;
 672     }
 673     default:
 674       res = gst_pad_query_default (pad, parent, query);
 675       break;
 676   }
 677
 678   return res;
 679 }
 680
 681 static void
 682 gst_rmdemux_free_stream (GstRMDemux * rmdemux, GstRMDemuxStream * stream)
 683 {
 684   g_object_unref (stream->adapter);
 685   gst_rmdemux_stream_clear_cached_subpackets (rmdemux, stream);
 686   if (stream->pending_tags)
 687     gst_tag_list_unref (stream->pending_tags);
 688   if (stream->subpackets)
 689     g_ptr_array_free (stream->subpackets, TRUE);
 690   g_free (stream->index);
 691   g_free (stream);
 692 }
 693
 694 static void
 695 gst_rmdemux_reset (GstRMDemux * rmdemux)
 696 {
 697   GSList *cur;
 698
 699   GST_OBJECT_LOCK (rmdemux);
 700   rmdemux->running = FALSE;
 701   GST_OBJECT_UNLOCK (rmdemux);
 702
 703   for (cur = rmdemux->streams; cur; cur = cur->next) {
 704     GstRMDemuxStream *stream = cur->data;
 705
 706     gst_flow_combiner_remove_pad (rmdemux->flowcombiner, stream->pad);
 707     gst_element_remove_pad (GST_ELEMENT (rmdemux), stream->pad);
 708     gst_rmdemux_free_stream (rmdemux, stream);
 709   }
 710   g_slist_free (rmdemux->streams);
 711   rmdemux->streams = NULL;
 712   rmdemux->n_audio_streams = 0;
 713   rmdemux->n_video_streams = 0;
 714
 715   if (rmdemux->pending_tags != NULL) {
 716     gst_tag_list_unref (rmdemux->pending_tags);
 717     rmdemux->pending_tags = NULL;
 718   }
 719
 720   gst_adapter_clear (rmdemux->adapter);
 721   rmdemux->state = RMDEMUX_STATE_HEADER;
 722   rmdemux->have_pads = FALSE;
 723
 724   gst_segment_init (&rmdemux->segment, GST_FORMAT_UNDEFINED);
 725   rmdemux->first_ts = GST_CLOCK_TIME_NONE;
 726   rmdemux->base_ts = GST_CLOCK_TIME_NONE;
 727   rmdemux->need_newsegment = TRUE;
 728
 729   rmdemux->have_group_id = FALSE;
 730   rmdemux->group_id = G_MAXUINT;
 731 }
 732
 733 static GstStateChangeReturn
 734 gst_rmdemux_change_state (GstElement * element, GstStateChange transition)
 735 {
 736   GstRMDemux *rmdemux = GST_RMDEMUX (element);
 737   GstStateChangeReturn res;
 738
 739   switch (transition) {
 740     case GST_STATE_CHANGE_NULL_TO_READY:
 741       break;
 742     case GST_STATE_CHANGE_READY_TO_PAUSED:
 743       rmdemux->state = RMDEMUX_STATE_HEADER;
 744       rmdemux->have_pads = FALSE;
 745       gst_segment_init (&rmdemux->segment, GST_FORMAT_TIME);
 746       rmdemux->running = FALSE;
 747       break;
 748     case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
 749       break;
 750     default:
 751       break;
 752   }
 753
 754   res = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
 755
 756   switch (transition) {
 757     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
 758       break;
 759     case GST_STATE_CHANGE_PAUSED_TO_READY:{
 760       gst_rmdemux_reset (rmdemux);
 761       break;
 762     }
 763     case GST_STATE_CHANGE_READY_TO_NULL:
 764       break;
 765     default:
 766       break;
 767   }
 768
 769   return res;
 770 }
 771
 772 /* this function is called when the pad is activated and should start
 773  * processing data.
 774  *
 775  * We check if we can do random access to decide if we work push or
 776  * pull based.
 777  */
 778 static gboolean
 779 gst_rmdemux_sink_activate (GstPad * sinkpad, GstObject * parent)
 780 {
 781   GstQuery *query;
 782   gboolean pull_mode;
 783
 784   query = gst_query_new_scheduling ();
 785
 786   if (!gst_pad_peer_query (sinkpad, query)) {
 787     gst_query_unref (query);
 788     goto activate_push;
 789   }
 790
 791   pull_mode = gst_query_has_scheduling_mode_with_flags (query,
 792       GST_PAD_MODE_PULL, GST_SCHEDULING_FLAG_SEEKABLE);
 793   gst_query_unref (query);
 794
 795   if (!pull_mode)
 796     goto activate_push;
 797
 798   GST_DEBUG_OBJECT (sinkpad, "activating pull");
 799   return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PULL, TRUE);
 800
 801 activate_push:
 802   {
 803     GST_DEBUG_OBJECT (sinkpad, "activating push");
 804     return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PUSH, TRUE);
 805   }
 806 }
 807
 808 static gboolean
 809 gst_rmdemux_sink_activate_mode (GstPad * sinkpad, GstObject * parent,
 810     GstPadMode mode, gboolean active)
 811 {
 812   gboolean res;
 813   GstRMDemux *demux;
 814
 815   demux = GST_RMDEMUX (parent);
 816
 817   switch (mode) {
 818     case GST_PAD_MODE_PUSH:
 819       demux->seekable = FALSE;
 820       demux->running = active;
 821       res = TRUE;
 822       break;
 823     case GST_PAD_MODE_PULL:
 824       if (active) {
 825         demux->seekable = TRUE;
 826         demux->offset = 0;
 827         demux->loop_state = RMDEMUX_LOOP_STATE_HEADER;
 828         demux->data_offset = G_MAXUINT;
 829         res =
 830             gst_pad_start_task (sinkpad, (GstTaskFunction) gst_rmdemux_loop,
 831             sinkpad, NULL);
 832       } else {
 833         res = gst_pad_stop_task (sinkpad);
 834       }
 835       break;
 836     default:
 837       res = FALSE;
 838       break;
 839   }
 840   return res;
 841 }
 842
 843
 844 /* random access mode - just pass over to our chain function */
 845 static void
 846 gst_rmdemux_loop (GstPad * pad)
 847 {
 848   GstRMDemux *rmdemux;
 849   GstBuffer *buffer;
 850   GstFlowReturn ret = GST_FLOW_OK;
 851   guint size;
 852
 853   rmdemux = GST_RMDEMUX (GST_PAD_PARENT (pad));
 854
 855   GST_LOG_OBJECT (rmdemux, "loop with state=%d and offset=0x%x",
 856       rmdemux->loop_state, rmdemux->offset);
 857
 858   switch (rmdemux->state) {
 859     case RMDEMUX_STATE_HEADER:
 860       size = HEADER_SIZE;
 861       break;
 862     case RMDEMUX_STATE_HEADER_DATA:
 863       size = DATA_SIZE;
 864       break;
 865     case RMDEMUX_STATE_DATA_PACKET:
 866       size = rmdemux->avg_packet_size;
 867       break;
 868     case RMDEMUX_STATE_EOS:
 869       GST_LOG_OBJECT (rmdemux, "At EOS, pausing task");
 870       ret = GST_FLOW_EOS;
 871       goto need_pause;
 872     default:
 873       GST_LOG_OBJECT (rmdemux, "Default: requires %d bytes (state is %d)",
 874           (int) rmdemux->size, rmdemux->state);
 875       size = rmdemux->size;
 876   }
 877
 878   buffer = NULL;
 879   ret = gst_pad_pull_range (pad, rmdemux->offset, size, &buffer);
 880   if (ret != GST_FLOW_OK) {
 881     if (rmdemux->offset == rmdemux->index_offset) {
 882       /* The index isn't available so forget about it */
 883       rmdemux->loop_state = RMDEMUX_LOOP_STATE_DATA;
 884       rmdemux->offset = rmdemux->data_offset;
 885       GST_OBJECT_LOCK (rmdemux);
 886       rmdemux->running = TRUE;
 887       rmdemux->seekable = FALSE;
 888       GST_OBJECT_UNLOCK (rmdemux);
 889       return;
 890     } else {
 891       GST_DEBUG_OBJECT (rmdemux, "Unable to pull %d bytes at offset 0x%08x "
 892           "(pull_range returned flow %s, state is %d)", (gint) size,
 893           rmdemux->offset, gst_flow_get_name (ret), GST_STATE (rmdemux));
 894       goto need_pause;
 895     }
 896   }
 897
 898   size = gst_buffer_get_size (buffer);
 899
 900   /* Defer to the chain function */
 901   ret = gst_rmdemux_chain (pad, GST_OBJECT_CAST (rmdemux), buffer);
 902   if (ret != GST_FLOW_OK) {
 903     GST_DEBUG_OBJECT (rmdemux, "Chain flow failed at offset 0x%08x",
 904         rmdemux->offset);
 905     goto need_pause;
 906   }
 907
 908   rmdemux->offset += size;
 909
 910   switch (rmdemux->loop_state) {
 911     case RMDEMUX_LOOP_STATE_HEADER:
 912       if (rmdemux->offset >= rmdemux->data_offset) {
 913         /* It's the end of the header */
 914         rmdemux->loop_state = RMDEMUX_LOOP_STATE_INDEX;
 915         rmdemux->offset = rmdemux->index_offset;
 916       }
 917       break;
 918     case RMDEMUX_LOOP_STATE_INDEX:
 919       if (rmdemux->state == RMDEMUX_STATE_HEADER) {
 920         if (rmdemux->index_offset == 0) {
 921           /* We've read the last index */
 922           rmdemux->loop_state = RMDEMUX_LOOP_STATE_DATA;
 923           rmdemux->offset = rmdemux->data_offset;
 924           GST_OBJECT_LOCK (rmdemux);
 925           rmdemux->running = TRUE;
 926           GST_OBJECT_UNLOCK (rmdemux);
 927         } else {
 928           /* Get the next index */
 929           rmdemux->offset = rmdemux->index_offset;
 930         }
 931       }
 932       break;
 933     case RMDEMUX_LOOP_STATE_DATA:
 934       break;
 935   }
 936
 937   return;
 938
 939   /* ERRORS */
 940 need_pause:
 941   {
 942     const gchar *reason = gst_flow_get_name (ret);
 943
 944     GST_LOG_OBJECT (rmdemux, "pausing task, reason %s", reason);
 945     rmdemux->segment_running = FALSE;
 946     gst_pad_pause_task (rmdemux->sinkpad);
 947
 948     if (ret == GST_FLOW_EOS) {
 949       /* perform EOS logic */
 950       if (rmdemux->segment.flags & GST_SEEK_FLAG_SEGMENT) {
 951         gint64 stop;
 952
 953         /* for segment playback we need to post when (in stream time)
 954          * we stopped, this is either stop (when set) or the duration. */
 955         if ((stop = rmdemux->segment.stop) == -1)
 956           stop = rmdemux->segment.duration;
 957
 958         GST_LOG_OBJECT (rmdemux, "Sending segment done, at end of segment");
 959         gst_element_post_message (GST_ELEMENT (rmdemux),
 960             gst_message_new_segment_done (GST_OBJECT (rmdemux),
 961                 GST_FORMAT_TIME, stop));
 962         gst_rmdemux_send_event (rmdemux,
 963             gst_event_new_segment_done (GST_FORMAT_TIME, stop));
 964       } else {
 965         /* normal playback, send EOS to all linked pads */
 966         GST_LOG_OBJECT (rmdemux, "Sending EOS, at end of stream");
 967         gst_rmdemux_send_event (rmdemux, gst_event_new_eos ());
 968       }
 969     } else if (ret == GST_FLOW_NOT_LINKED || ret < GST_FLOW_EOS) {
 970       GST_ELEMENT_ERROR (rmdemux, STREAM, FAILED,
 971           (NULL), ("stream stopped, reason %s", reason));
 972       gst_rmdemux_send_event (rmdemux, gst_event_new_eos ());
 973     }
 974     return;
 975   }
 976 }
 977
 978 static gboolean
 979 gst_rmdemux_fourcc_isplausible (guint32 fourcc)
 980 {
 981   int i;
 982
 983   for (i = 0; i < 4; i++) {
 984     if (!isprint ((int) ((unsigned char *) (&fourcc))[i])) {
 985       return FALSE;
 986     }
 987   }
 988   return TRUE;
 989 }
 990
 991 static GstFlowReturn
 992 gst_rmdemux_chain (GstPad * pad, GstObject * parent, GstBuffer * buffer)
 993 {
 994   GstFlowReturn ret = GST_FLOW_OK;
 995   const guint8 *data;
 996   guint16 version;
 997   guint avail;
 998
 999   GstRMDemux *rmdemux = GST_RMDEMUX (parent);
1000
1001   if (rmdemux->base_ts == -1) {
1002     if (GST_BUFFER_DTS_IS_VALID (buffer))
1003       rmdemux->base_ts = GST_BUFFER_DTS (buffer);
1004     else
1005       rmdemux->base_ts = GST_BUFFER_PTS (buffer);
1006
1007     GST_LOG_OBJECT (rmdemux, "base_ts %" GST_TIME_FORMAT,
1008         GST_TIME_ARGS (rmdemux->base_ts));
1009   }
1010
1011   gst_adapter_push (rmdemux->adapter, buffer);
1012
1013   GST_LOG_OBJECT (rmdemux, "Chaining buffer of size %" G_GSIZE_FORMAT,
1014       gst_buffer_get_size (buffer));
1015
1016   while (TRUE) {
1017     avail = gst_adapter_available (rmdemux->adapter);
1018
1019     GST_LOG_OBJECT (rmdemux, "looping in chain, avail %u", avail);
1020     switch (rmdemux->state) {
1021       case RMDEMUX_STATE_HEADER:
1022       {
1023         if (gst_adapter_available (rmdemux->adapter) < HEADER_SIZE)
1024           goto unlock;
1025
1026         data = gst_adapter_map (rmdemux->adapter, HEADER_SIZE);
1027
1028         rmdemux->object_id = RMDEMUX_FOURCC_GET (data + 0);
1029         rmdemux->size = RMDEMUX_GUINT32_GET (data + 4) - HEADER_SIZE;
1030         rmdemux->object_version = RMDEMUX_GUINT16_GET (data + 8);
1031
1032         /* Sanity-check. We assume that the FOURCC is printable ASCII */
1033         if (!gst_rmdemux_fourcc_isplausible (rmdemux->object_id)) {
1034           /* Failed. Remain in HEADER state, try again... We flush only
1035            * the actual FOURCC, not the entire header, because we could
1036            * need to resync anywhere at all... really, this should never
1037            * happen. */
1038           GST_WARNING_OBJECT (rmdemux, "Bogus looking header, unprintable "
1039               "FOURCC");
1040           gst_adapter_unmap (rmdemux->adapter);
1041           gst_adapter_flush (rmdemux->adapter, 4);
1042
1043           break;
1044         }
1045
1046         GST_LOG_OBJECT (rmdemux, "header found with object_id=%"
1047             GST_FOURCC_FORMAT
1048             " size=%08x object_version=%d",
1049             GST_FOURCC_ARGS (rmdemux->object_id), rmdemux->size,
1050             rmdemux->object_version);
1051
1052         gst_adapter_unmap (rmdemux->adapter);
1053         gst_adapter_flush (rmdemux->adapter, HEADER_SIZE);
1054
1055         switch (rmdemux->object_id) {
1056           case GST_MAKE_FOURCC ('.', 'R', 'M', 'F'):
1057             rmdemux->state = RMDEMUX_STATE_HEADER_RMF;
1058             break;
1059           case GST_MAKE_FOURCC ('P', 'R', 'O', 'P'):
1060             rmdemux->state = RMDEMUX_STATE_HEADER_PROP;
1061             break;
1062           case GST_MAKE_FOURCC ('M', 'D', 'P', 'R'):
1063             rmdemux->state = RMDEMUX_STATE_HEADER_MDPR;
1064             break;
1065           case GST_MAKE_FOURCC ('I', 'N', 'D', 'X'):
1066             rmdemux->state = RMDEMUX_STATE_HEADER_INDX;
1067             break;
1068           case GST_MAKE_FOURCC ('D', 'A', 'T', 'A'):
1069             rmdemux->state = RMDEMUX_STATE_HEADER_DATA;
1070             break;
1071           case GST_MAKE_FOURCC ('C', 'O', 'N', 'T'):
1072             rmdemux->state = RMDEMUX_STATE_HEADER_CONT;
1073             break;
1074           default:
1075             rmdemux->state = RMDEMUX_STATE_HEADER_UNKNOWN;
1076             break;
1077         }
1078         break;
1079       }
1080       case RMDEMUX_STATE_HEADER_UNKNOWN:
1081       {
1082         if (gst_adapter_available (rmdemux->adapter) < rmdemux->size)
1083           goto unlock;
1084
1085         GST_WARNING_OBJECT (rmdemux, "Unknown object_id %" GST_FOURCC_FORMAT,
1086             GST_FOURCC_ARGS (rmdemux->object_id));
1087
1088         gst_adapter_flush (rmdemux->adapter, rmdemux->size);
1089         rmdemux->state = RMDEMUX_STATE_HEADER;
1090         break;
1091       }
1092       case RMDEMUX_STATE_HEADER_RMF:
1093       {
1094         if (gst_adapter_available (rmdemux->adapter) < rmdemux->size)
1095           goto unlock;
1096
1097         if ((rmdemux->object_version == 0) || (rmdemux->object_version == 1)) {
1098           data = gst_adapter_map (rmdemux->adapter, rmdemux->size);
1099           gst_rmdemux_parse__rmf (rmdemux, data, rmdemux->size);
1100           gst_adapter_unmap (rmdemux->adapter);
1101           gst_adapter_flush (rmdemux->adapter, rmdemux->size);
1102         } else {
1103           gst_adapter_flush (rmdemux->adapter, rmdemux->size);
1104         }
1105         rmdemux->state = RMDEMUX_STATE_HEADER;
1106         break;
1107       }
1108       case RMDEMUX_STATE_HEADER_PROP:
1109       {
1110         if (gst_adapter_available (rmdemux->adapter) < rmdemux->size)
1111           goto unlock;
1112
1113         data = gst_adapter_map (rmdemux->adapter, rmdemux->size);
1114         gst_rmdemux_parse_prop (rmdemux, data, rmdemux->size);
1115         gst_adapter_unmap (rmdemux->adapter);
1116         gst_adapter_flush (rmdemux->adapter, rmdemux->size);
1117
1118         rmdemux->state = RMDEMUX_STATE_HEADER;
1119         break;
1120       }
1121       case RMDEMUX_STATE_HEADER_MDPR:
1122       {
1123         if (gst_adapter_available (rmdemux->adapter) < rmdemux->size)
1124           goto unlock;
1125
1126         data = gst_adapter_map (rmdemux->adapter, rmdemux->size);
1127         gst_rmdemux_parse_mdpr (rmdemux, data, rmdemux->size);
1128         gst_adapter_unmap (rmdemux->adapter);
1129         gst_adapter_flush (rmdemux->adapter, rmdemux->size);
1130
1131         rmdemux->state = RMDEMUX_STATE_HEADER;
1132         break;
1133       }
1134       case RMDEMUX_STATE_HEADER_CONT:
1135       {
1136         if (gst_adapter_available (rmdemux->adapter) < rmdemux->size)
1137           goto unlock;
1138
1139         data = gst_adapter_map (rmdemux->adapter, rmdemux->size);
1140         gst_rmdemux_parse_cont (rmdemux, data, rmdemux->size);
1141         gst_adapter_unmap (rmdemux->adapter);
1142         gst_adapter_flush (rmdemux->adapter, rmdemux->size);
1143
1144         rmdemux->state = RMDEMUX_STATE_HEADER;
1145         break;
1146       }
1147       case RMDEMUX_STATE_HEADER_DATA:
1148       {
1149         /* If we haven't already done so then signal there are no more pads */
1150         if (!rmdemux->have_pads) {
1151           GST_LOG_OBJECT (rmdemux, "no more pads");
1152           gst_element_no_more_pads (GST_ELEMENT (rmdemux));
1153           rmdemux->have_pads = TRUE;
1154         }
1155
1156         /* The actual header is only 8 bytes */
1157         rmdemux->size = DATA_SIZE;
1158         GST_LOG_OBJECT (rmdemux, "data available %" G_GSIZE_FORMAT,
1159             gst_adapter_available (rmdemux->adapter));
1160         if (gst_adapter_available (rmdemux->adapter) < rmdemux->size)
1161           goto unlock;
1162
1163         data = gst_adapter_map (rmdemux->adapter, rmdemux->size);
1164         gst_rmdemux_parse_data (rmdemux, data, rmdemux->size);
1165         gst_adapter_unmap (rmdemux->adapter);
1166         gst_adapter_flush (rmdemux->adapter, rmdemux->size);
1167
1168         rmdemux->state = RMDEMUX_STATE_DATA_PACKET;
1169         break;
1170       }
1171       case RMDEMUX_STATE_HEADER_INDX:
1172       {
1173         if (gst_adapter_available (rmdemux->adapter) < rmdemux->size)
1174           goto unlock;
1175
1176         data = gst_adapter_map (rmdemux->adapter, rmdemux->size);
1177         rmdemux->size = gst_rmdemux_parse_indx (rmdemux, data, rmdemux->size);
1178         /* Only flush the header */
1179         gst_adapter_unmap (rmdemux->adapter);
1180         gst_adapter_flush (rmdemux->adapter, HEADER_SIZE);
1181
1182         rmdemux->state = RMDEMUX_STATE_INDX_DATA;
1183         break;
1184       }
1185       case RMDEMUX_STATE_INDX_DATA:
1186       {
1187         /* There's not always an data to get... */
1188         if (rmdemux->size > 0) {
1189           if (gst_adapter_available (rmdemux->adapter) < rmdemux->size)
1190             goto unlock;
1191
1192           data = gst_adapter_map (rmdemux->adapter, rmdemux->size);
1193           gst_rmdemux_parse_indx_data (rmdemux, data, rmdemux->size);
1194           gst_adapter_unmap (rmdemux->adapter);
1195           gst_adapter_flush (rmdemux->adapter, rmdemux->size);
1196         }
1197
1198         rmdemux->state = RMDEMUX_STATE_HEADER;
1199         break;
1200       }
1201       case RMDEMUX_STATE_DATA_PACKET:
1202       {
1203         guint8 header[4];
1204
1205         if (gst_adapter_available (rmdemux->adapter) < 2)
1206           goto unlock;
1207
1208         gst_adapter_copy (rmdemux->adapter, header, 0, 2);
1209         version = RMDEMUX_GUINT16_GET (header);
1210         GST_LOG_OBJECT (rmdemux, "Data packet with version=%d", version);
1211
1212         if (version == 0 || version == 1) {
1213           guint16 length;
1214
1215           if (gst_adapter_available (rmdemux->adapter) < 4)
1216             goto unlock;
1217
1218           gst_adapter_copy (rmdemux->adapter, header, 0, 4);
1219
1220           length = RMDEMUX_GUINT16_GET (header + 2);
1221           GST_LOG_OBJECT (rmdemux, "Got length %d", length);
1222
1223           if (length < 4) {
1224             GST_LOG_OBJECT (rmdemux, "length too small, dropping");
1225             /* Invalid, just drop it */
1226             gst_adapter_flush (rmdemux->adapter, 4);
1227           } else {
1228             GstBuffer *buffer;
1229
1230             avail = gst_adapter_available (rmdemux->adapter);
1231             if (avail < length)
1232               goto unlock;
1233
1234             GST_LOG_OBJECT (rmdemux, "we have %u available and we needed %d",
1235                 avail, length);
1236
1237             /* flush version and length */
1238             gst_adapter_flush (rmdemux->adapter, 4);
1239             length -= 4;
1240
1241             buffer = gst_adapter_take_buffer (rmdemux->adapter, length);
1242
1243             ret = gst_rmdemux_parse_packet (rmdemux, buffer, version);
1244             rmdemux->chunk_index++;
1245           }
1246
1247           if (rmdemux->chunk_index == rmdemux->n_chunks || length == 0)
1248             rmdemux->state = RMDEMUX_STATE_HEADER;
1249         } else {
1250           /* Stream done */
1251           gst_adapter_flush (rmdemux->adapter, 2);
1252
1253           if (rmdemux->data_offset == 0) {
1254             GST_LOG_OBJECT (rmdemux,
1255                 "No further data, internal demux state EOS");
1256             rmdemux->state = RMDEMUX_STATE_EOS;
1257           } else
1258             rmdemux->state = RMDEMUX_STATE_HEADER;
1259         }
1260         break;
1261       }
1262       case RMDEMUX_STATE_EOS:
1263         gst_rmdemux_send_event (rmdemux, gst_event_new_eos ());
1264         goto unlock;
1265       default:
1266         GST_WARNING_OBJECT (rmdemux, "Unhandled state %d", rmdemux->state);
1267         goto unlock;
1268     }
1269   }
1270
1271 unlock:
1272   return ret;
1273 }
1274
1275 static GstRMDemuxStream *
1276 gst_rmdemux_get_stream_by_id (GstRMDemux * rmdemux, int id)
1277 {
1278   GSList *cur;
1279
1280   for (cur = rmdemux->streams; cur; cur = cur->next) {
1281     GstRMDemuxStream *stream = cur->data;
1282
1283     if (stream->id == id) {
1284       return stream;
1285     }
1286   }
1287
1288   return NULL;
1289 }
1290
1291 static void
1292 gst_rmdemux_send_event (GstRMDemux * rmdemux, GstEvent * event)
1293 {
1294   GSList *cur;
1295
1296   for (cur = rmdemux->streams; cur; cur = cur->next) {
1297     GstRMDemuxStream *stream = cur->data;
1298
1299     GST_DEBUG_OBJECT (rmdemux, "Pushing %s event on pad %s",
1300         GST_EVENT_TYPE_NAME (event), GST_PAD_NAME (stream->pad));
1301
1302     switch (GST_EVENT_TYPE (event)) {
1303       case GST_EVENT_FLUSH_STOP:
1304         stream->last_ts = -1;
1305         stream->next_ts = -1;
1306         stream->last_seq = -1;
1307         stream->next_seq = -1;
1308         break;
1309       default:
1310         break;
1311     }
1312     gst_event_ref (event);
1313     gst_pad_push_event (stream->pad, event);
1314   }
1315   gst_event_unref (event);
1316 }
1317
1318 static void
1319 gst_rmdemux_add_stream (GstRMDemux * rmdemux, GstRMDemuxStream * stream)
1320 {
1321   GstCaps *stream_caps = NULL;
1322   const gchar *codec_tag = NULL;
1323   gchar *codec_name = NULL;
1324   gchar *stream_id;
1325   int version = 0;
1326
1327   if (stream->subtype == GST_RMDEMUX_STREAM_VIDEO) {
1328     char *name = g_strdup_printf ("video_%u", rmdemux->n_video_streams);
1329
1330     stream->pad =
1331         gst_pad_new_from_static_template (&gst_rmdemux_videosrc_template, name);
1332     g_free (name);
1333
1334     codec_tag = GST_TAG_VIDEO_CODEC;
1335
1336     switch (stream->fourcc) {
1337       case GST_RM_VDO_RV10:
1338         version = 1;
1339         break;
1340       case GST_RM_VDO_RV20:
1341         version = 2;
1342         break;
1343       case GST_RM_VDO_RV30:
1344         version = 3;
1345         break;
1346       case GST_RM_VDO_RV40:
1347         version = 4;
1348         break;
1349       default:
1350         stream_caps = gst_caps_new_simple ("video/x-unknown-fourcc",
1351             "fourcc", G_TYPE_UINT, stream->fourcc, NULL);
1352         GST_WARNING_OBJECT (rmdemux,
1353             "Unknown video FOURCC code \"%" GST_FOURCC_FORMAT "\" (%08x)",
1354             GST_FOURCC_ARGS (stream->fourcc), stream->fourcc);
1355     }
1356
1357     if (version) {
1358       stream_caps =
1359           gst_caps_new_simple ("video/x-pn-realvideo", "rmversion", G_TYPE_INT,
1360           (int) version,
1361           "format", G_TYPE_INT,
1362           (int) stream->format,
1363           "subformat", G_TYPE_INT, (int) stream->subformat, NULL);
1364     }
1365
1366     if (stream_caps) {
1367       gst_caps_set_simple (stream_caps,
1368           "width", G_TYPE_INT, stream->width,
1369           "height", G_TYPE_INT, stream->height,
1370           "framerate", GST_TYPE_FRACTION, stream->framerate_numerator,
1371           stream->framerate_denominator, NULL);
1372     }
1373     rmdemux->n_video_streams++;
1374
1375   } else if (stream->subtype == GST_RMDEMUX_STREAM_AUDIO) {
1376     char *name = g_strdup_printf ("audio_%u", rmdemux->n_audio_streams);
1377
1378     stream->pad =
1379         gst_pad_new_from_static_template (&gst_rmdemux_audiosrc_template, name);
1380     GST_LOG_OBJECT (rmdemux, "Created audio pad \"%s\"", name);
1381     g_free (name);
1382
1383     codec_tag = GST_TAG_AUDIO_CODEC;
1384
1385     switch (stream->fourcc) {
1386         /* Older RealAudio Codecs */
1387       case GST_RM_AUD_14_4:
1388         version = 1;
1389         break;
1390
1391       case GST_RM_AUD_28_8:
1392         version = 2;
1393         break;
1394
1395         /* DolbyNet (Dolby AC3, low bitrate) */
1396       case GST_RM_AUD_DNET:
1397         stream_caps =
1398             gst_caps_new_simple ("audio/x-ac3", "rate", G_TYPE_INT,
1399             (int) stream->rate, NULL);
1400         stream->needs_descrambling = TRUE;
1401         stream->subpackets_needed = 1;
1402         stream->subpackets = NULL;
1403         break;
1404
1405         /* MPEG-4 based */
1406       case GST_RM_AUD_RAAC:
1407       case GST_RM_AUD_RACP:
1408         stream_caps =
1409             gst_caps_new_simple ("audio/mpeg", "mpegversion", G_TYPE_INT,
1410             (int) 4, "framed", G_TYPE_BOOLEAN, TRUE, NULL);
1411         if (stream->extra_data_size > 0) {
1412           /* strip off an unknown byte in the extra data */
1413           stream->extra_data_size--;
1414           stream->extra_data++;
1415         }
1416         stream->needs_descrambling = TRUE;
1417         stream->subpackets_needed = 1;
1418         stream->subpackets = NULL;
1419         break;
1420
1421         /* Sony ATRAC3 */
1422       case GST_RM_AUD_ATRC:
1423         stream_caps = gst_caps_new_empty_simple ("audio/x-vnd.sony.atrac3");
1424         stream->needs_descrambling = TRUE;
1425         stream->subpackets_needed = stream->height;
1426         stream->subpackets = NULL;
1427         break;
1428
1429         /* RealAudio G2 audio */
1430       case GST_RM_AUD_COOK:
1431         version = 8;
1432         stream->needs_descrambling = TRUE;
1433         stream->subpackets_needed = stream->height;
1434         stream->subpackets = NULL;
1435         break;
1436
1437         /* RALF is lossless */
1438       case GST_RM_AUD_RALF:
1439         GST_DEBUG_OBJECT (rmdemux, "RALF");
1440         stream_caps = gst_caps_new_empty_simple ("audio/x-ralf-mpeg4-generic");
1441         break;
1442
1443       case GST_RM_AUD_SIPR:
1444
1445         if (stream->flavor > 3) {
1446           GST_WARNING_OBJECT (rmdemux, "bad SIPR flavor %d, freeing it",
1447               stream->flavor);
1448           g_object_unref (stream->pad);
1449           gst_rmdemux_free_stream (rmdemux, stream);
1450           goto beach;
1451         }
1452
1453         GST_DEBUG_OBJECT (rmdemux, "SIPR");
1454         stream_caps = gst_caps_new_empty_simple ("audio/x-sipro");
1455         stream->needs_descrambling = TRUE;
1456         stream->subpackets_needed = stream->height;
1457         stream->subpackets = NULL;
1458         stream->leaf_size = sipr_subpk_size[stream->flavor];
1459
1460         break;
1461
1462       default:
1463         stream_caps = gst_caps_new_simple ("video/x-unknown-fourcc",
1464             "fourcc", G_TYPE_UINT, stream->fourcc, NULL);
1465         GST_WARNING_OBJECT (rmdemux,
1466             "Unknown audio FOURCC code \"%" GST_FOURCC_FORMAT "\" (%08x)",
1467             GST_FOURCC_ARGS (stream->fourcc), stream->fourcc);
1468         break;
1469     }
1470
1471     if (version) {
1472       stream_caps =
1473           gst_caps_new_simple ("audio/x-pn-realaudio", "raversion", G_TYPE_INT,
1474           (int) version, NULL);
1475     }
1476
1477     if (stream_caps) {
1478       gst_caps_set_simple (stream_caps,
1479           "flavor", G_TYPE_INT, (int) stream->flavor,
1480           "rate", G_TYPE_INT, (int) stream->rate,
1481           "channels", G_TYPE_INT, (int) stream->n_channels,
1482           "width", G_TYPE_INT, (int) stream->sample_width,
1483           "leaf_size", G_TYPE_INT, (int) stream->leaf_size,
1484           "packet_size", G_TYPE_INT, (int) stream->packet_size,
1485           "bitrate", G_TYPE_INT, (int) stream->bitrate,
1486           "height", G_TYPE_INT, (int) stream->height, NULL);
1487     }
1488     rmdemux->n_audio_streams++;
1489   } else {
1490     GST_WARNING_OBJECT (rmdemux, "not adding stream of type %d, freeing it",
1491         stream->subtype);
1492     gst_rmdemux_free_stream (rmdemux, stream);
1493     goto beach;
1494   }
1495
1496   GST_PAD_ELEMENT_PRIVATE (stream->pad) = stream;
1497   rmdemux->streams = g_slist_append (rmdemux->streams, stream);
1498   GST_LOG_OBJECT (rmdemux, "n_streams is now %d",
1499       g_slist_length (rmdemux->streams));
1500
1501   GST_LOG ("stream->pad = %p, stream_caps = %" GST_PTR_FORMAT, stream->pad,
1502       stream_caps);
1503
1504   if (stream->pad && stream_caps) {
1505     GstEvent *event;
1506
1507     GST_LOG_OBJECT (rmdemux, "%d bytes of extra data for stream %s",
1508         stream->extra_data_size, GST_PAD_NAME (stream->pad));
1509
1510     /* add codec_data if there is any */
1511     if (stream->extra_data_size > 0) {
1512       GstBuffer *buffer;
1513
1514       buffer = gst_buffer_new_and_alloc (stream->extra_data_size);
1515       gst_buffer_fill (buffer, 0, stream->extra_data, stream->extra_data_size);
1516
1517       gst_caps_set_simple (stream_caps, "codec_data", GST_TYPE_BUFFER,
1518           buffer, NULL);
1519
1520       gst_buffer_unref (buffer);
1521     }
1522
1523     gst_pad_use_fixed_caps (stream->pad);
1524
1525     gst_pad_set_event_function (stream->pad,
1526         GST_DEBUG_FUNCPTR (gst_rmdemux_src_event));
1527     gst_pad_set_query_function (stream->pad,
1528         GST_DEBUG_FUNCPTR (gst_rmdemux_src_query));
1529
1530     GST_DEBUG_OBJECT (rmdemux, "adding pad %s with caps %" GST_PTR_FORMAT
1531         ", stream_id=%d", GST_PAD_NAME (stream->pad), stream_caps, stream->id);
1532     gst_pad_set_active (stream->pad, TRUE);
1533
1534     stream_id =
1535         gst_pad_create_stream_id_printf (stream->pad,
1536         GST_ELEMENT_CAST (rmdemux), "%03u", stream->id);
1537
1538     event =
1539         gst_pad_get_sticky_event (rmdemux->sinkpad, GST_EVENT_STREAM_START, 0);
1540     if (event) {
1541       if (gst_event_parse_group_id (event, &rmdemux->group_id))
1542         rmdemux->have_group_id = TRUE;
1543       else
1544         rmdemux->have_group_id = FALSE;
1545       gst_event_unref (event);
1546     } else if (!rmdemux->have_group_id) {
1547       rmdemux->have_group_id = TRUE;
1548       rmdemux->group_id = gst_util_group_id_next ();
1549     }
1550
1551     event = gst_event_new_stream_start (stream_id);
1552     if (rmdemux->have_group_id)
1553       gst_event_set_group_id (event, rmdemux->group_id);
1554
1555     gst_pad_push_event (stream->pad, event);
1556     g_free (stream_id);
1557
1558     gst_pad_set_caps (stream->pad, stream_caps);
1559
1560     codec_name = gst_pb_utils_get_codec_description (stream_caps);
1561
1562     /* save for later, we must send the tags after the newsegment event */
1563     if (codec_tag != NULL && codec_name != NULL) {
1564       if (stream->pending_tags == NULL)
1565         stream->pending_tags = gst_tag_list_new_empty ();
1566       gst_tag_list_add (stream->pending_tags, GST_TAG_MERGE_KEEP,
1567           codec_tag, codec_name, NULL);
1568       g_free (codec_name);
1569     }
1570     gst_element_add_pad (GST_ELEMENT_CAST (rmdemux), stream->pad);
1571     gst_flow_combiner_add_pad (rmdemux->flowcombiner, stream->pad);
1572   }
1573
1574 beach:
1575
1576   if (stream_caps)
1577     gst_caps_unref (stream_caps);
1578 }
1579
1580 static int
1581 re_skip_pascal_string (const guint8 * ptr)
1582 {
1583   int length;
1584
1585   length = ptr[0];
1586
1587   return length + 1;
1588 }
1589
1590 static void
1591 gst_rmdemux_parse__rmf (GstRMDemux * rmdemux, const guint8 * data, int length)
1592 {
1593   GST_LOG_OBJECT (rmdemux, "file_version: %d", RMDEMUX_GUINT32_GET (data));
1594   GST_LOG_OBJECT (rmdemux, "num_headers: %d", RMDEMUX_GUINT32_GET (data + 4));
1595 }
1596
1597 static void
1598 gst_rmdemux_parse_prop (GstRMDemux * rmdemux, const guint8 * data, int length)
1599 {
1600   GST_LOG_OBJECT (rmdemux, "max bitrate: %d", RMDEMUX_GUINT32_GET (data));
1601   GST_LOG_OBJECT (rmdemux, "avg bitrate: %d", RMDEMUX_GUINT32_GET (data + 4));
1602   GST_LOG_OBJECT (rmdemux, "max packet size: %d",
1603       RMDEMUX_GUINT32_GET (data + 8));
1604   rmdemux->avg_packet_size = RMDEMUX_GUINT32_GET (data + 12);
1605   GST_LOG_OBJECT (rmdemux, "avg packet size: %d", rmdemux->avg_packet_size);
1606   rmdemux->num_packets = RMDEMUX_GUINT32_GET (data + 16);
1607   GST_LOG_OBJECT (rmdemux, "number of packets: %d", rmdemux->num_packets);
1608
1609   GST_LOG_OBJECT (rmdemux, "duration: %d", RMDEMUX_GUINT32_GET (data + 20));
1610   rmdemux->duration = RMDEMUX_GUINT32_GET (data + 20) * GST_MSECOND;
1611
1612   GST_LOG_OBJECT (rmdemux, "preroll: %d", RMDEMUX_GUINT32_GET (data + 24));
1613   rmdemux->index_offset = RMDEMUX_GUINT32_GET (data + 28);
1614   GST_LOG_OBJECT (rmdemux, "offset of INDX section: 0x%08x",
1615       rmdemux->index_offset);
1616   rmdemux->data_offset = RMDEMUX_GUINT32_GET (data + 32);
1617   GST_LOG_OBJECT (rmdemux, "offset of DATA section: 0x%08x",
1618       rmdemux->data_offset);
1619   GST_LOG_OBJECT (rmdemux, "n streams: %d", RMDEMUX_GUINT16_GET (data + 36));
1620   GST_LOG_OBJECT (rmdemux, "flags: 0x%04x", RMDEMUX_GUINT16_GET (data + 38));
1621 }
1622
1623 static void
1624 gst_rmdemux_parse_mdpr (GstRMDemux * rmdemux, const guint8 * data, int length)
1625 {
1626   GstRMDemuxStream *stream;
1627   char *stream1_type_string;
1628   char *stream2_type_string;
1629   guint str_len = 0;
1630   int stream_type;
1631   int offset;
1632   guint32 max_bitrate;
1633   guint32 avg_bitrate;
1634
1635   stream = g_new0 (GstRMDemuxStream, 1);
1636
1637   stream->id = RMDEMUX_GUINT16_GET (data);
1638   stream->index = NULL;
1639   stream->seek_offset = 0;
1640   stream->last_ts = -1;
1641   stream->next_ts = -1;
1642   stream->discont = TRUE;
1643   stream->adapter = gst_adapter_new ();
1644   GST_LOG_OBJECT (rmdemux, "stream_number=%d", stream->id);
1645
1646   /* parse the bitrates */
1647   max_bitrate = RMDEMUX_GUINT32_GET (data + 2);
1648   avg_bitrate = RMDEMUX_GUINT32_GET (data + 6);
1649   stream->bitrate = avg_bitrate;
1650   GST_LOG_OBJECT (rmdemux, "Stream max bitrate=%u", max_bitrate);
1651   GST_LOG_OBJECT (rmdemux, "Stream avg bitrate=%u", avg_bitrate);
1652   if (max_bitrate != 0) {
1653     if (stream->pending_tags == NULL)
1654       stream->pending_tags = gst_tag_list_new_empty ();
1655     gst_tag_list_add (stream->pending_tags, GST_TAG_MERGE_REPLACE,
1656         GST_TAG_MAXIMUM_BITRATE, max_bitrate, NULL);
1657   }
1658   if (avg_bitrate != 0) {
1659     if (stream->pending_tags == NULL)
1660       stream->pending_tags = gst_tag_list_new_empty ();
1661     gst_tag_list_add (stream->pending_tags, GST_TAG_MERGE_REPLACE,
1662         GST_TAG_BITRATE, avg_bitrate, NULL);
1663   }
1664
1665   offset = 30;
1666   stream1_type_string = gst_rm_utils_read_string8 (data + offset,
1667       length - offset, &str_len);
1668   offset += str_len;
1669   stream2_type_string = gst_rm_utils_read_string8 (data + offset,
1670       length - offset, &str_len);
1671   offset += str_len;
1672
1673   /* stream1_type_string for audio and video stream is a "put_whatever_you_want" field :
1674    * observed values :
1675    * - "[The ]Video/Audio Stream" (File produced by an official Real encoder)
1676    * - "RealVideoPremierePlugIn-VIDEO/AUDIO" (File produced by Abobe Premiere)
1677    *
1678    * so, we should not rely on it to know which stream type it is
1679    */
1680
1681   GST_LOG_OBJECT (rmdemux, "stream type: %s", stream1_type_string);
1682   GST_LOG_OBJECT (rmdemux, "MIME type=%s", stream2_type_string);
1683
1684   if (strcmp (stream2_type_string, "video/x-pn-realvideo") == 0) {
1685     stream_type = GST_RMDEMUX_STREAM_VIDEO;
1686   } else if (strcmp (stream2_type_string,
1687           "video/x-pn-multirate-realvideo") == 0) {
1688     stream_type = GST_RMDEMUX_STREAM_VIDEO;
1689   } else if (strcmp (stream2_type_string, "audio/x-pn-realaudio") == 0) {
1690     stream_type = GST_RMDEMUX_STREAM_AUDIO;
1691   } else if (strcmp (stream2_type_string,
1692           "audio/x-pn-multirate-realaudio") == 0) {
1693     stream_type = GST_RMDEMUX_STREAM_AUDIO;
1694   } else if (strcmp (stream2_type_string,
1695           "audio/x-pn-multirate-realaudio-live") == 0) {
1696     stream_type = GST_RMDEMUX_STREAM_AUDIO;
1697   } else if (strcmp (stream2_type_string, "audio/x-ralf-mpeg4-generic") == 0) {
1698     /* Another audio type found in the real testsuite */
1699     stream_type = GST_RMDEMUX_STREAM_AUDIO;
1700   } else if (strcmp (stream1_type_string, "") == 0 &&
1701       strcmp (stream2_type_string, "logical-fileinfo") == 0) {
1702     stream_type = GST_RMDEMUX_STREAM_FILEINFO;
1703   } else {
1704     stream_type = GST_RMDEMUX_STREAM_UNKNOWN;
1705     GST_WARNING_OBJECT (rmdemux, "unknown stream type \"%s\",\"%s\"",
1706         stream1_type_string, stream2_type_string);
1707   }
1708   g_free (stream1_type_string);
1709   g_free (stream2_type_string);
1710
1711   offset += 4;
1712
1713   stream->subtype = stream_type;
1714   switch (stream_type) {
1715
1716     case GST_RMDEMUX_STREAM_VIDEO:
1717       /* RV10/RV20/RV30/RV40 => video/x-pn-realvideo, version=1,2,3,4 */
1718       stream->fourcc = RMDEMUX_FOURCC_GET (data + offset + 8);
1719       stream->width = RMDEMUX_GUINT16_GET (data + offset + 12);
1720       stream->height = RMDEMUX_GUINT16_GET (data + offset + 14);
1721       stream->rate = RMDEMUX_GUINT16_GET (data + offset + 16);
1722       stream->subformat = RMDEMUX_GUINT32_GET (data + offset + 26);
1723       stream->format = RMDEMUX_GUINT32_GET (data + offset + 30);
1724       stream->extra_data_size = length - (offset + 26);
1725       stream->extra_data = (guint8 *) data + offset + 26;
1726       /* Natural way to represent framerates here requires unsigned 32 bit
1727        * numerator, which we don't have. For the nasty case, approximate...
1728        */
1729       {
1730         guint32 numerator = RMDEMUX_GUINT16_GET (data + offset + 22) * 65536 +
1731             RMDEMUX_GUINT16_GET (data + offset + 24);
1732         if (numerator > G_MAXINT) {
1733           stream->framerate_numerator = (gint) (numerator >> 1);
1734           stream->framerate_denominator = 32768;
1735         } else {
1736           stream->framerate_numerator = (gint) numerator;
1737           stream->framerate_denominator = 65536;
1738         }
1739       }
1740
1741       GST_DEBUG_OBJECT (rmdemux,
1742           "Video stream with fourcc=%" GST_FOURCC_FORMAT
1743           " width=%d height=%d rate=%d framerate=%d/%d subformat=%x format=%x extra_data_size=%d",
1744           GST_FOURCC_ARGS (stream->fourcc), stream->width, stream->height,
1745           stream->rate, stream->framerate_numerator,
1746           stream->framerate_denominator, stream->subformat, stream->format,
1747           stream->extra_data_size);
1748       break;
1749     case GST_RMDEMUX_STREAM_AUDIO:{
1750       stream->version = RMDEMUX_GUINT16_GET (data + offset + 4);
1751       GST_INFO ("stream version = %u", stream->version);
1752       switch (stream->version) {
1753         case 3:
1754           stream->fourcc = GST_RM_AUD_14_4;
1755           stream->packet_size = 20;
1756           stream->rate = 8000;
1757           stream->n_channels = 1;
1758           stream->sample_width = 16;
1759           stream->flavor = 1;
1760           stream->leaf_size = 0;
1761           stream->height = 0;
1762           break;
1763         case 4:
1764           stream->flavor = RMDEMUX_GUINT16_GET (data + offset + 22);
1765           stream->packet_size = RMDEMUX_GUINT32_GET (data + offset + 24);
1766           /* stream->frame_size = RMDEMUX_GUINT32_GET (data + offset + 42); */
1767           stream->leaf_size = RMDEMUX_GUINT16_GET (data + offset + 44);
1768           stream->height = RMDEMUX_GUINT16_GET (data + offset + 40);
1769           stream->rate = RMDEMUX_GUINT16_GET (data + offset + 48);
1770           stream->sample_width = RMDEMUX_GUINT16_GET (data + offset + 52);
1771           stream->n_channels = RMDEMUX_GUINT16_GET (data + offset + 54);
1772           stream->fourcc = RMDEMUX_FOURCC_GET (data + offset + 62);
1773           stream->extra_data_size = RMDEMUX_GUINT32_GET (data + offset + 69);
1774           GST_DEBUG_OBJECT (rmdemux, "%u bytes of extra codec data",
1775               stream->extra_data_size);
1776           if (length - (offset + 73) >= stream->extra_data_size) {
1777             stream->extra_data = (guint8 *) data + offset + 73;
1778           } else {
1779             GST_WARNING_OBJECT (rmdemux, "codec data runs beyond MDPR chunk");
1780             stream->extra_data_size = 0;
1781           }
1782           break;
1783         case 5:
1784           stream->flavor = RMDEMUX_GUINT16_GET (data + offset + 22);
1785           stream->packet_size = RMDEMUX_GUINT32_GET (data + offset + 24);
1786           /* stream->frame_size = RMDEMUX_GUINT32_GET (data + offset + 42); */
1787           stream->leaf_size = RMDEMUX_GUINT16_GET (data + offset + 44);
1788           stream->height = RMDEMUX_GUINT16_GET (data + offset + 40);
1789           stream->rate = RMDEMUX_GUINT16_GET (data + offset + 54);
1790           stream->sample_width = RMDEMUX_GUINT16_GET (data + offset + 58);
1791           stream->n_channels = RMDEMUX_GUINT16_GET (data + offset + 60);
1792           stream->fourcc = RMDEMUX_FOURCC_GET (data + offset + 66);
1793           stream->extra_data_size = RMDEMUX_GUINT32_GET (data + offset + 74);
1794           GST_DEBUG_OBJECT (rmdemux, "%u bytes of extra codec data",
1795               stream->extra_data_size);
1796           if (length - (offset + 78) >= stream->extra_data_size) {
1797             stream->extra_data = (guint8 *) data + offset + 78;
1798           } else {
1799             GST_WARNING_OBJECT (rmdemux, "codec data runs beyond MDPR chunk");
1800             stream->extra_data_size = 0;
1801           }
1802           break;
1803         default:{
1804           GST_WARNING_OBJECT (rmdemux, "Unhandled audio stream version %d",
1805               stream->version);
1806           break;
1807         }
1808       }
1809       /*  14_4, 28_8, cook, dnet, sipr, raac, racp, ralf, atrc */
1810       GST_DEBUG_OBJECT (rmdemux,
1811           "Audio stream with rate=%d sample_width=%d n_channels=%d",
1812           stream->rate, stream->sample_width, stream->n_channels);
1813
1814       break;
1815     }
1816     case GST_RMDEMUX_STREAM_FILEINFO:
1817     {
1818       int element_nb;
1819
1820       /* Length of this section */
1821       GST_DEBUG_OBJECT (rmdemux, "length2: 0x%08x",
1822           RMDEMUX_GUINT32_GET (data + offset));
1823       offset += 4;
1824
1825       /* Unknown : 00 00 00 00 */
1826       offset += 4;
1827
1828       /* Number of variables that would follow (loop iterations) */
1829       element_nb = RMDEMUX_GUINT32_GET (data + offset);
1830       offset += 4;
1831
1832       while (element_nb) {
1833         /* Category Id : 00 00 00 XX 00 00 */
1834         offset += 6;
1835
1836         /* Variable Name */
1837         offset += re_skip_pascal_string (data + offset);
1838
1839         /* Variable Value Type */
1840         /*   00 00 00 00 00 => integer/boolean, preceded by length */
1841         /*   00 00 00 02 00 => pascal string, preceded by length, no trailing \0 */
1842         offset += 5;
1843
1844         /* Variable Value */
1845         offset += re_skip_pascal_string (data + offset);
1846
1847         element_nb--;
1848       }
1849     }
1850       break;
1851     case GST_RMDEMUX_STREAM_UNKNOWN:
1852     default:
1853       break;
1854   }
1855
1856   gst_rmdemux_add_stream (rmdemux, stream);
1857 }
1858
1859 static guint
1860 gst_rmdemux_parse_indx (GstRMDemux * rmdemux, const guint8 * data, int length)
1861 {
1862   int n;
1863   int id;
1864
1865   n = RMDEMUX_GUINT32_GET (data);
1866   id = RMDEMUX_GUINT16_GET (data + 4);
1867   rmdemux->index_offset = RMDEMUX_GUINT32_GET (data + 6);
1868
1869   GST_DEBUG_OBJECT (rmdemux, "Number of indices=%d Stream ID=%d length=%d", n,
1870       id, length);
1871
1872   /* Point to the next index_stream */
1873   rmdemux->index_stream = gst_rmdemux_get_stream_by_id (rmdemux, id);
1874
1875   /* Return the length of the index */
1876   return 14 * n;
1877 }
1878
1879 static void
1880 gst_rmdemux_parse_indx_data (GstRMDemux * rmdemux, const guint8 * data,
1881     int length)
1882 {
1883   int i;
1884   int n;
1885   GstRMDemuxIndex *index;
1886
1887   /* The number of index records */
1888   n = length / 14;
1889
1890   if (rmdemux->index_stream == NULL)
1891     return;
1892
1893   /* don't parse the index a second time when operating pull-based and
1894    * reaching the end of the file */
1895   if (rmdemux->index_stream->index_length > 0) {
1896     GST_DEBUG_OBJECT (rmdemux, "Already have an index for this stream");
1897     return;
1898   }
1899
1900   index = g_malloc (sizeof (GstRMDemuxIndex) * n);
1901   rmdemux->index_stream->index = index;
1902   rmdemux->index_stream->index_length = n;
1903
1904   for (i = 0; i < n; i++) {
1905     index[i].timestamp = RMDEMUX_GUINT32_GET (data + 2) * GST_MSECOND;
1906     index[i].offset = RMDEMUX_GUINT32_GET (data + 6);
1907
1908     GST_DEBUG_OBJECT (rmdemux, "Index found for timestamp=%f (at offset=%x)",
1909         gst_guint64_to_gdouble (index[i].timestamp) / GST_SECOND,
1910         index[i].offset);
1911     data += 14;
1912   }
1913 }
1914
1915 static void
1916 gst_rmdemux_parse_data (GstRMDemux * rmdemux, const guint8 * data, int length)
1917 {
1918   rmdemux->n_chunks = RMDEMUX_GUINT32_GET (data);
1919   rmdemux->data_offset = RMDEMUX_GUINT32_GET (data + 4);
1920   rmdemux->chunk_index = 0;
1921   GST_DEBUG_OBJECT (rmdemux, "Data chunk found with %d packets "
1922       "(next data at 0x%08x)", rmdemux->n_chunks, rmdemux->data_offset);
1923 }
1924
1925 static void
1926 gst_rmdemux_parse_cont (GstRMDemux * rmdemux, const guint8 * data, int length)
1927 {
1928   GstTagList *tags;
1929
1930   tags = gst_rm_utils_read_tags (data, length, gst_rm_utils_read_string16);
1931
1932   if (tags) {
1933     GstTagList *old_tags = rmdemux->pending_tags;
1934
1935     GST_LOG_OBJECT (rmdemux, "tags: %" GST_PTR_FORMAT, tags);
1936
1937     rmdemux->pending_tags =
1938         gst_tag_list_merge (old_tags, tags, GST_TAG_MERGE_APPEND);
1939
1940     gst_tag_list_unref (tags);
1941     if (old_tags)
1942       gst_tag_list_unref (old_tags);
1943
1944     gst_tag_list_set_scope (rmdemux->pending_tags, GST_TAG_SCOPE_GLOBAL);
1945   }
1946 }
1947
1948 static void
1949 gst_rmdemux_stream_clear_cached_subpackets (GstRMDemux * rmdemux,
1950     GstRMDemuxStream * stream)
1951 {
1952   if (stream->subpackets == NULL || stream->subpackets->len == 0)
1953     return;
1954
1955   GST_DEBUG_OBJECT (rmdemux, "discarding %u previously collected subpackets",
1956       stream->subpackets->len);
1957   g_ptr_array_foreach (stream->subpackets, (GFunc) gst_mini_object_unref, NULL);
1958   g_ptr_array_set_size (stream->subpackets, 0);
1959 }
1960
1961 static GstFlowReturn
1962 gst_rmdemux_descramble_audio (GstRMDemux * rmdemux, GstRMDemuxStream * stream)
1963 {
1964   GstFlowReturn ret = GST_FLOW_ERROR;
1965   GstBuffer *outbuf;
1966   GstMapInfo outmap;
1967   guint packet_size = stream->packet_size;
1968   guint height = stream->subpackets->len;
1969   guint leaf_size = stream->leaf_size;
1970   guint p, x;
1971
1972   g_assert (stream->height == height);
1973
1974   GST_LOG ("packet_size = %u, leaf_size = %u, height= %u", packet_size,
1975       leaf_size, height);
1976
1977   outbuf = gst_buffer_new_and_alloc (height * packet_size);
1978   gst_buffer_map (outbuf, &outmap, GST_MAP_WRITE);
1979
1980   for (p = 0; p < height; ++p) {
1981     GstBuffer *b = g_ptr_array_index (stream->subpackets, p);
1982     GstMapInfo map;
1983
1984     gst_buffer_map (b, &map, GST_MAP_READ);
1985
1986     if (p == 0) {
1987       GST_BUFFER_PTS (outbuf) = GST_BUFFER_PTS (b);
1988       GST_BUFFER_DTS (outbuf) = GST_BUFFER_DTS (b);
1989     }
1990
1991     for (x = 0; x < packet_size / leaf_size; ++x) {
1992       guint idx;
1993
1994       idx = height * x + ((height + 1) / 2) * (p % 2) + (p / 2);
1995
1996       /* GST_LOG ("%3u => %3u", (height * p) + x, idx); */
1997       memcpy (outmap.data + leaf_size * idx, map.data + leaf_size * x,
1998           leaf_size);
1999     }
2000     gst_buffer_unmap (b, &map);
2001   }
2002   gst_buffer_unmap (outbuf, &outmap);
2003
2004   /* some decoders, such as realaudiodec, need to be fed in packet units */
2005   for (p = 0; p < height; ++p) {
2006     GstBuffer *subbuf;
2007
2008     subbuf =
2009         gst_buffer_copy_region (outbuf, GST_BUFFER_COPY_ALL, p * packet_size,
2010         packet_size);
2011
2012     GST_LOG_OBJECT (rmdemux, "pushing buffer dts %" GST_TIME_FORMAT ", pts %"
2013         GST_TIME_FORMAT, GST_TIME_ARGS (GST_BUFFER_DTS (subbuf)),
2014         GST_TIME_ARGS (GST_BUFFER_PTS (subbuf)));
2015
2016     if (stream->discont) {
2017       GST_BUFFER_FLAG_SET (subbuf, GST_BUFFER_FLAG_DISCONT);
2018       stream->discont = FALSE;
2019     }
2020
2021     ret = gst_pad_push (stream->pad, subbuf);
2022     if (ret != GST_FLOW_OK)
2023       break;
2024   }
2025
2026   gst_buffer_unref (outbuf);
2027
2028   gst_rmdemux_stream_clear_cached_subpackets (rmdemux, stream);
2029
2030   return ret;
2031 }
2032
2033 static GstFlowReturn
2034 gst_rmdemux_descramble_dnet_audio (GstRMDemux * rmdemux,
2035     GstRMDemuxStream * stream)
2036 {
2037   GstBuffer *buf;
2038
2039   buf = g_ptr_array_index (stream->subpackets, 0);
2040   g_ptr_array_index (stream->subpackets, 0) = NULL;
2041   g_ptr_array_set_size (stream->subpackets, 0);
2042
2043   buf = gst_rm_utils_descramble_dnet_buffer (buf);
2044
2045   if (stream->discont) {
2046     GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_DISCONT);
2047     stream->discont = FALSE;
2048   }
2049   return gst_pad_push (stream->pad, buf);
2050 }
2051
2052 static GstFlowReturn
2053 gst_rmdemux_descramble_mp4a_audio (GstRMDemux * rmdemux,
2054     GstRMDemuxStream * stream)
2055 {
2056   GstFlowReturn res;
2057   GstBuffer *buf, *outbuf;
2058   guint frames, index, i;
2059   GstMapInfo map;
2060   GstClockTime timestamp;
2061
2062   res = GST_FLOW_OK;
2063
2064   buf = g_ptr_array_index (stream->subpackets, 0);
2065   g_ptr_array_index (stream->subpackets, 0) = NULL;
2066   g_ptr_array_set_size (stream->subpackets, 0);
2067
2068   gst_buffer_map (buf, &map, GST_MAP_READ);
2069   timestamp = GST_BUFFER_PTS (buf);
2070
2071   frames = (map.data[1] & 0xf0) >> 4;
2072   index = 2 * frames + 2;
2073
2074   for (i = 0; i < frames; i++) {
2075     guint len = (map.data[i * 2 + 2] << 8) | map.data[i * 2 + 3];
2076
2077     outbuf = gst_buffer_copy_region (buf, GST_BUFFER_COPY_ALL, index, len);
2078     if (i == 0) {
2079       GST_BUFFER_PTS (outbuf) = timestamp;
2080       GST_BUFFER_DTS (outbuf) = timestamp;
2081     }
2082
2083     index += len;
2084
2085     if (stream->discont) {
2086       GST_BUFFER_FLAG_SET (buf, GST_BUFFER_FLAG_DISCONT);
2087       stream->discont = FALSE;
2088     }
2089     res = gst_pad_push (stream->pad, outbuf);
2090     if (res != GST_FLOW_OK)
2091       break;
2092   }
2093   gst_buffer_unmap (buf, &map);
2094   gst_buffer_unref (buf);
2095   return res;
2096 }
2097
2098 static GstFlowReturn
2099 gst_rmdemux_descramble_sipr_audio (GstRMDemux * rmdemux,
2100     GstRMDemuxStream * stream)
2101 {
2102   GstFlowReturn ret;
2103   GstBuffer *outbuf;
2104   GstMapInfo outmap;
2105   guint packet_size = stream->packet_size;
2106   guint height = stream->subpackets->len;
2107   guint p;
2108
2109   g_assert (stream->height == height);
2110
2111   GST_LOG ("packet_size = %u, leaf_size = %u, height= %u", packet_size,
2112       stream->leaf_size, height);
2113
2114   outbuf = gst_buffer_new_and_alloc (height * packet_size);
2115   gst_buffer_map (outbuf, &outmap, GST_MAP_WRITE);
2116
2117   for (p = 0; p < height; ++p) {
2118     GstBuffer *b = g_ptr_array_index (stream->subpackets, p);
2119
2120     if (p == 0) {
2121       GST_BUFFER_DTS (outbuf) = GST_BUFFER_DTS (b);
2122       GST_BUFFER_PTS (outbuf) = GST_BUFFER_PTS (b);
2123     }
2124
2125     gst_buffer_extract (b, 0, outmap.data + packet_size * p, packet_size);
2126   }
2127   gst_buffer_unmap (outbuf, &outmap);
2128
2129   GST_LOG_OBJECT (rmdemux, "pushing buffer dts %" GST_TIME_FORMAT ", pts %"
2130       GST_TIME_FORMAT, GST_TIME_ARGS (GST_BUFFER_DTS (outbuf)),
2131       GST_TIME_ARGS (GST_BUFFER_PTS (outbuf)));
2132
2133   if (stream->discont) {
2134     GST_BUFFER_FLAG_SET (outbuf, GST_BUFFER_FLAG_DISCONT);
2135     stream->discont = FALSE;
2136   }
2137
2138   outbuf = gst_rm_utils_descramble_sipr_buffer (outbuf);
2139
2140   ret = gst_pad_push (stream->pad, outbuf);
2141
2142   gst_rmdemux_stream_clear_cached_subpackets (rmdemux, stream);
2143
2144   return ret;
2145 }
2146
2147 static GstFlowReturn
2148 gst_rmdemux_handle_scrambled_packet (GstRMDemux * rmdemux,
2149     GstRMDemuxStream * stream, GstBuffer * buf, gboolean keyframe)
2150 {
2151   GstFlowReturn ret;
2152
2153   if (stream->subpackets == NULL)
2154     stream->subpackets = g_ptr_array_sized_new (stream->subpackets_needed);
2155
2156   GST_LOG ("Got subpacket %u/%u, len=%" G_GSIZE_FORMAT ", key=%d",
2157       stream->subpackets->len + 1, stream->subpackets_needed,
2158       gst_buffer_get_size (buf), keyframe);
2159
2160   if (keyframe && stream->subpackets->len > 0) {
2161     gst_rmdemux_stream_clear_cached_subpackets (rmdemux, stream);
2162   }
2163
2164   g_ptr_array_add (stream->subpackets, buf);
2165
2166   if (stream->subpackets->len < stream->subpackets_needed)
2167     return GST_FLOW_OK;
2168
2169   g_assert (stream->subpackets->len >= 1);
2170
2171   switch (stream->fourcc) {
2172     case GST_RM_AUD_DNET:
2173       ret = gst_rmdemux_descramble_dnet_audio (rmdemux, stream);
2174       break;
2175     case GST_RM_AUD_COOK:
2176     case GST_RM_AUD_ATRC:
2177       ret = gst_rmdemux_descramble_audio (rmdemux, stream);
2178       break;
2179     case GST_RM_AUD_RAAC:
2180     case GST_RM_AUD_RACP:
2181       ret = gst_rmdemux_descramble_mp4a_audio (rmdemux, stream);
2182       break;
2183     case GST_RM_AUD_SIPR:
2184       ret = gst_rmdemux_descramble_sipr_audio (rmdemux, stream);
2185       break;
2186     default:
2187       ret = GST_FLOW_ERROR;
2188       g_assert_not_reached ();
2189   }
2190
2191   return ret;
2192 }
2193
2194 #define PARSE_NUMBER(data, size, number, label) \
2195 G_STMT_START {                                  \
2196   if (size < 2)                                 \
2197     goto label;                                 \
2198   number = GST_READ_UINT16_BE (data);           \
2199   if (!(number & 0xc000)) {                     \
2200     if (size < 4)                               \
2201       goto label;                               \
2202     number = GST_READ_UINT32_BE (data);         \
2203     data += 4;                                  \
2204     size -= 4;                                  \
2205   } else {                                      \
2206     number &= 0x3fff;                           \
2207     data += 2;                                  \
2208     size -= 2;                                  \
2209   }                                             \
2210 } G_STMT_END
2211
2212 static GstFlowReturn
2213 gst_rmdemux_parse_video_packet (GstRMDemux * rmdemux, GstRMDemuxStream * stream,
2214     GstBuffer * in, guint offset, guint16 version,
2215     GstClockTime timestamp, gboolean key)
2216 {
2217   GstFlowReturn ret;
2218   GstMapInfo map;
2219   const guint8 *data;
2220   gsize size;
2221
2222   gst_buffer_map (in, &map, GST_MAP_READ);
2223
2224   data = map.data + offset;
2225   size = map.size - offset;
2226
2227   /* if size <= 2, we want this method to return the same GstFlowReturn as it
2228    * was previously for that given stream. */
2229   ret = GST_PAD_LAST_FLOW_RETURN (stream->pad);
2230
2231   while (size > 2) {
2232     guint8 pkg_header;
2233     guint pkg_offset;
2234     guint pkg_length;
2235     guint pkg_subseq = 0, pkg_seqnum = G_MAXUINT;
2236     guint fragment_size;
2237     GstBuffer *fragment;
2238
2239     pkg_header = *data++;
2240     size--;
2241
2242     /* packet header
2243      * bit 7: 1=last block in block chain
2244      * bit 6: 1=short header (only one block?)
2245      */
2246     if ((pkg_header & 0xc0) == 0x40) {
2247       /* skip unknown byte */
2248       data++;
2249       size--;
2250       pkg_offset = 0;
2251       pkg_length = size;
2252     } else {
2253       if ((pkg_header & 0x40) == 0) {
2254         pkg_subseq = (*data++) & 0x7f;
2255         size--;
2256       } else {
2257         pkg_subseq = 0;
2258       }
2259
2260       /* length */
2261       PARSE_NUMBER (data, size, pkg_length, not_enough_data);
2262
2263       /* offset */
2264       PARSE_NUMBER (data, size, pkg_offset, not_enough_data);
2265
2266       /* seqnum */
2267       if (size < 1)
2268         goto not_enough_data;
2269
2270       pkg_seqnum = *data++;
2271       size--;
2272     }
2273
2274     GST_DEBUG_OBJECT (rmdemux,
2275         "seq %d, subseq %d, offset %d, length %d, size %" G_GSIZE_FORMAT
2276         ", header %02x", pkg_seqnum, pkg_subseq, pkg_offset, pkg_length, size,
2277         pkg_header);
2278
2279     /* calc size of fragment */
2280     if ((pkg_header & 0xc0) == 0x80) {
2281       fragment_size = pkg_offset;
2282     } else {
2283       if ((pkg_header & 0xc0) == 0)
2284         fragment_size = size;
2285       else
2286         fragment_size = pkg_length;
2287     }
2288     GST_DEBUG_OBJECT (rmdemux, "fragment size %d", fragment_size);
2289
2290     /* get the fragment */
2291     fragment =
2292         gst_buffer_copy_region (in, GST_BUFFER_COPY_ALL, data - map.data,
2293         fragment_size);
2294
2295     if (pkg_subseq == 1) {
2296       GST_DEBUG_OBJECT (rmdemux, "start new fragment");
2297       gst_adapter_clear (stream->adapter);
2298       stream->frag_current = 0;
2299       stream->frag_count = 0;
2300       stream->frag_length = pkg_length;
2301     } else if (pkg_subseq == 0) {
2302       GST_DEBUG_OBJECT (rmdemux, "non fragmented packet");
2303       stream->frag_current = 0;
2304       stream->frag_count = 0;
2305       stream->frag_length = fragment_size;
2306     }
2307
2308     /* put fragment in adapter */
2309     gst_adapter_push (stream->adapter, fragment);
2310     stream->frag_offset[stream->frag_count] = stream->frag_current;
2311     stream->frag_current += fragment_size;
2312     stream->frag_count++;
2313
2314     if (stream->frag_count > MAX_FRAGS)
2315       goto too_many_fragments;
2316
2317     GST_DEBUG_OBJECT (rmdemux, "stored fragment in adapter %d/%d",
2318         stream->frag_current, stream->frag_length);
2319
2320     /* flush fragment when complete */
2321     if (stream->frag_current >= stream->frag_length) {
2322       GstBuffer *out;
2323       GstMapInfo outmap;
2324       guint8 *outdata;
2325       guint header_size;
2326       gint i, avail;
2327
2328       /* calculate header size, which is:
2329        * 1 byte for the number of fragments - 1
2330        * for each fragment:
2331        *   4 bytes 0x00000001 little endian
2332        *   4 bytes fragment offset
2333        *
2334        * This is also the matroska header for realvideo, the decoder needs the
2335        * fragment offsets, both in ffmpeg and real .so, so we just give it that
2336        * in front of the data.
2337        */
2338       header_size = 1 + (8 * (stream->frag_count));
2339
2340       GST_DEBUG_OBJECT (rmdemux,
2341           "fragmented completed. count %d, header_size %u", stream->frag_count,
2342           header_size);
2343
2344       avail = gst_adapter_available (stream->adapter);
2345
2346       out = gst_buffer_new_and_alloc (header_size + avail);
2347       gst_buffer_map (out, &outmap, GST_MAP_WRITE);
2348       outdata = outmap.data;
2349
2350       /* create header */
2351       *outdata++ = stream->frag_count - 1;
2352       for (i = 0; i < stream->frag_count; i++) {
2353         GST_WRITE_UINT32_LE (outdata, 0x00000001);
2354         outdata += 4;
2355         GST_WRITE_UINT32_LE (outdata, stream->frag_offset[i]);
2356         outdata += 4;
2357       }
2358
2359       /* copy packet data after the header now */
2360       gst_adapter_copy (stream->adapter, outdata, 0, avail);
2361       gst_adapter_flush (stream->adapter, avail);
2362
2363       stream->frag_current = 0;
2364       stream->frag_count = 0;
2365       stream->frag_length = 0;
2366
2367       if (timestamp != -1) {
2368         if (rmdemux->first_ts != -1 && timestamp > rmdemux->first_ts)
2369           timestamp -= rmdemux->first_ts;
2370         else
2371           timestamp = 0;
2372
2373         if (rmdemux->base_ts != -1)
2374           timestamp += rmdemux->base_ts;
2375       }
2376       gst_buffer_unmap (out, &outmap);
2377
2378       /* video has DTS */
2379       GST_BUFFER_DTS (out) = timestamp;
2380       GST_BUFFER_PTS (out) = GST_CLOCK_TIME_NONE;
2381
2382       GST_LOG_OBJECT (rmdemux, "pushing timestamp %" GST_TIME_FORMAT,
2383           GST_TIME_ARGS (timestamp));
2384
2385       if (stream->discont) {
2386         GST_BUFFER_FLAG_SET (out, GST_BUFFER_FLAG_DISCONT);
2387         stream->discont = FALSE;
2388       }
2389
2390       if (!key) {
2391         GST_BUFFER_FLAG_SET (out, GST_BUFFER_FLAG_DELTA_UNIT);
2392       }
2393
2394       ret = gst_pad_push (stream->pad, out);
2395       ret = gst_flow_combiner_update_flow (rmdemux->flowcombiner, ret);
2396       if (ret != GST_FLOW_OK)
2397         break;
2398
2399       timestamp = GST_CLOCK_TIME_NONE;
2400     }
2401     data += fragment_size;
2402     size -= fragment_size;
2403   }
2404   GST_DEBUG_OBJECT (rmdemux, "%" G_GSIZE_FORMAT " bytes left", size);
2405
2406 done:
2407   gst_buffer_unmap (in, &map);
2408   gst_buffer_unref (in);
2409
2410   return ret;
2411
2412   /* ERRORS */
2413 not_enough_data:
2414   {
2415     GST_ELEMENT_WARNING (rmdemux, STREAM, DECODE, ("Skipping bad packet."),
2416         (NULL));
2417     ret = GST_FLOW_OK;
2418     goto done;
2419   }
2420 too_many_fragments:
2421   {
2422     GST_ELEMENT_ERROR (rmdemux, STREAM, DECODE,
2423         ("Got more fragments (%u) than can be handled (%u)",
2424             stream->frag_count, MAX_FRAGS), (NULL));
2425     ret = GST_FLOW_ERROR;
2426     goto done;
2427   }
2428 }
2429
2430 static GstFlowReturn
2431 gst_rmdemux_parse_audio_packet (GstRMDemux * rmdemux, GstRMDemuxStream * stream,
2432     GstBuffer * in, guint offset, guint16 version,
2433     GstClockTime timestamp, gboolean key)
2434 {
2435   GstFlowReturn ret;
2436   GstBuffer *buffer;
2437
2438   buffer = gst_buffer_copy_region (in, GST_BUFFER_COPY_MEMORY, offset, -1);
2439
2440   if (rmdemux->first_ts != -1 && timestamp > rmdemux->first_ts)
2441     timestamp -= rmdemux->first_ts;
2442   else
2443     timestamp = 0;
2444
2445   if (rmdemux->base_ts != -1)
2446     timestamp += rmdemux->base_ts;
2447
2448   GST_BUFFER_PTS (buffer) = timestamp;
2449   GST_BUFFER_DTS (buffer) = timestamp;
2450
2451   if (stream->needs_descrambling) {
2452     GST_LOG_OBJECT (rmdemux, "descramble timestamp %" GST_TIME_FORMAT,
2453         GST_TIME_ARGS (timestamp));
2454     ret = gst_rmdemux_handle_scrambled_packet (rmdemux, stream, buffer, key);
2455   } else {
2456     GST_LOG_OBJECT (rmdemux,
2457         "Pushing buffer of size %" G_GSIZE_FORMAT ", timestamp %"
2458         GST_TIME_FORMAT "to pad %s", gst_buffer_get_size (buffer),
2459         GST_TIME_ARGS (timestamp), GST_PAD_NAME (stream->pad));
2460
2461     if (stream->discont) {
2462       GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_DISCONT);
2463       stream->discont = FALSE;
2464     }
2465     ret = gst_pad_push (stream->pad, buffer);
2466   }
2467
2468   gst_buffer_unref (in);
2469
2470   return ret;
2471 }
2472
2473 static GstFlowReturn
2474 gst_rmdemux_parse_packet (GstRMDemux * rmdemux, GstBuffer * in, guint16 version)
2475 {
2476   guint16 id;
2477   GstRMDemuxStream *stream;
2478   gsize size, offset;
2479   GstFlowReturn cret, ret;
2480   GstClockTime timestamp;
2481   gboolean key;
2482   GstMapInfo map;
2483   guint8 *data;
2484   guint8 flags;
2485   guint32 ts;
2486
2487   gst_buffer_map (in, &map, GST_MAP_READ);
2488   data = map.data;
2489   size = map.size;
2490
2491   /* stream number */
2492   id = RMDEMUX_GUINT16_GET (data);
2493
2494   stream = gst_rmdemux_get_stream_by_id (rmdemux, id);
2495   if (!stream || !stream->pad)
2496     goto unknown_stream;
2497
2498   /* timestamp in Msec */
2499   ts = RMDEMUX_GUINT32_GET (data + 2);
2500   timestamp = ts * GST_MSECOND;
2501
2502   rmdemux->segment.position = timestamp;
2503
2504   GST_LOG_OBJECT (rmdemux, "Parsing a packet for stream=%d, timestamp=%"
2505       GST_TIME_FORMAT ", size %" G_GSIZE_FORMAT ", version=%d, ts=%u", id,
2506       GST_TIME_ARGS (timestamp), size, version, ts);
2507
2508   if (rmdemux->first_ts == GST_CLOCK_TIME_NONE) {
2509     GST_DEBUG_OBJECT (rmdemux, "First timestamp: %" GST_TIME_FORMAT,
2510         GST_TIME_ARGS (timestamp));
2511     rmdemux->first_ts = timestamp;
2512   }
2513
2514   /* skip stream_id and timestamp */
2515   data += (2 + 4);
2516   size -= (2 + 4);
2517
2518   /* get flags */
2519   flags = GST_READ_UINT8 (data + 1);
2520
2521   data += 2;
2522   size -= 2;
2523
2524   /* version 1 has an extra byte */
2525   if (version == 1) {
2526     data += 1;
2527     size -= 1;
2528   }
2529   offset = data - map.data;
2530   gst_buffer_unmap (in, &map);
2531
2532   key = (flags & 0x02) != 0;
2533   GST_DEBUG_OBJECT (rmdemux, "flags %d, Keyframe %d", flags, key);
2534
2535   if (rmdemux->need_newsegment) {
2536     GstEvent *event;
2537
2538     event = gst_event_new_segment (&rmdemux->segment);
2539
2540     GST_DEBUG_OBJECT (rmdemux, "sending NEWSEGMENT event, segment.start= %"
2541         GST_TIME_FORMAT, GST_TIME_ARGS (rmdemux->segment.start));
2542
2543     gst_rmdemux_send_event (rmdemux, event);
2544     rmdemux->need_newsegment = FALSE;
2545
2546     if (rmdemux->pending_tags != NULL) {
2547       gst_rmdemux_send_event (rmdemux,
2548           gst_event_new_tag (rmdemux->pending_tags));
2549       rmdemux->pending_tags = NULL;
2550     }
2551   }
2552
2553   if (stream->pending_tags != NULL) {
2554     GST_LOG_OBJECT (stream->pad, "tags %" GST_PTR_FORMAT, stream->pending_tags);
2555     gst_pad_push_event (stream->pad, gst_event_new_tag (stream->pending_tags));
2556     stream->pending_tags = NULL;
2557   }
2558
2559   if ((rmdemux->offset + size) <= stream->seek_offset) {
2560     GST_DEBUG_OBJECT (rmdemux,
2561         "Stream %d is skipping: seek_offset=%d, offset=%d, size=%"
2562         G_GSIZE_FORMAT, stream->id, stream->seek_offset, rmdemux->offset, size);
2563     cret = GST_FLOW_OK;
2564     gst_buffer_unref (in);
2565     goto beach;
2566   }
2567
2568   /* do special headers */
2569   if (stream->subtype == GST_RMDEMUX_STREAM_VIDEO) {
2570     ret =
2571         gst_rmdemux_parse_video_packet (rmdemux, stream, in, offset,
2572         version, timestamp, key);
2573   } else if (stream->subtype == GST_RMDEMUX_STREAM_AUDIO) {
2574     ret =
2575         gst_rmdemux_parse_audio_packet (rmdemux, stream, in, offset,
2576         version, timestamp, key);
2577   } else {
2578     gst_buffer_unref (in);
2579     ret = GST_FLOW_OK;
2580   }
2581
2582   cret = gst_flow_combiner_update_pad_flow (rmdemux->flowcombiner, stream->pad,
2583       ret);
2584
2585 beach:
2586   return cret;
2587
2588   /* ERRORS */
2589 unknown_stream:
2590   {
2591     GST_WARNING_OBJECT (rmdemux, "No stream for stream id %d in parsing "
2592         "data packet", id);
2593     gst_buffer_unmap (in, &map);
2594     gst_buffer_unref (in);
2595     return GST_FLOW_OK;
2596   }
2597 }
2598
2599 gboolean
2600 gst_rmdemux_plugin_init (GstPlugin * plugin)
2601 {
2602   return gst_element_register (plugin, "rmdemux",
2603       GST_RANK_PRIMARY, GST_TYPE_RMDEMUX);
2604 }