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