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