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