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