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