gst/realmedia/rmdemux.c

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