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