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