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