gst/realmedia/rmdemux.c

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