asf: fix x-wmv format in caps
[platform/upstream/gst-plugins-ugly.git] / gst / asfdemux / gstasfdemux.c
1 /* GStreamer ASF/WMV/WMA demuxer
2  * Copyright (C) 1999 Erik Walthinsen <omega@cse.ogi.edu>
3  * Copyright (C) 2006-2009 Tim-Philipp Müller <tim centricular net>
4  *
5  * This library is free software; you can redistribute it and/or
6  * modify it under the terms of the GNU Library General Public
7  * License as published by the Free Software Foundation; either
8  * version 2 of the License, or (at your option) any later version.
9  *
10  * This library is distributed in the hope that it will be useful,
11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * Library General Public License for more details.
14  *
15  * You should have received a copy of the GNU Library General Public
16  * License along with this library; if not, write to the
17  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18  * Boston, MA 02111-1307, USA.
19  */
20
21 /* TODO:
22  *
23  * - _loop():
24  *   stop if at end of segment if != end of file, ie. demux->segment.stop
25  *
26  * - fix packet parsing:
27  *   there's something wrong with timestamps for packets with keyframes,
28  *   and durations too.
29  */
30
31 #ifdef HAVE_CONFIG_H
32 #include "config.h"
33 #endif
34
35 #include <gst/gstutils.h>
36 #include <gst/base/gstbytereader.h>
37 #include <gst/riff/riff-media.h>
38 #include <gst/tag/tag.h>
39 #include <gst/gst-i18n-plugin.h>
40 #include <stdio.h>
41 #include <stdlib.h>
42 #include <string.h>
43
44 #include "gstasfdemux.h"
45 #include "asfheaders.h"
46 #include "asfpacket.h"
47
48 static GstStaticPadTemplate gst_asf_demux_sink_template =
49 GST_STATIC_PAD_TEMPLATE ("sink",
50     GST_PAD_SINK,
51     GST_PAD_ALWAYS,
52     GST_STATIC_CAPS ("video/x-ms-asf")
53     );
54
55 static GstStaticPadTemplate audio_src_template =
56 GST_STATIC_PAD_TEMPLATE ("audio_%u",
57     GST_PAD_SRC,
58     GST_PAD_SOMETIMES,
59     GST_STATIC_CAPS_ANY);
60
61 static GstStaticPadTemplate video_src_template =
62 GST_STATIC_PAD_TEMPLATE ("video_%u",
63     GST_PAD_SRC,
64     GST_PAD_SOMETIMES,
65     GST_STATIC_CAPS_ANY);
66
67 /* size of an ASF object header, ie. GUID (16 bytes) + object size (8 bytes) */
68 #define ASF_OBJECT_HEADER_SIZE  (16+8)
69
70 /* FIXME: get rid of this */
71 /* abuse this GstFlowReturn enum for internal usage */
72 #define ASF_FLOW_NEED_MORE_DATA  99
73
74 #define gst_asf_get_flow_name(flow)    \
75   (flow == ASF_FLOW_NEED_MORE_DATA) ?  \
76   "need-more-data" : gst_flow_get_name (flow)
77
78 GST_DEBUG_CATEGORY (asfdemux_dbg);
79
80 static GstStateChangeReturn gst_asf_demux_change_state (GstElement * element,
81     GstStateChange transition);
82 static gboolean gst_asf_demux_element_send_event (GstElement * element,
83     GstEvent * event);
84 static gboolean gst_asf_demux_send_event_unlocked (GstASFDemux * demux,
85     GstEvent * event);
86 static gboolean gst_asf_demux_handle_src_query (GstPad * pad,
87     GstObject * parent, GstQuery * query);
88 static GstFlowReturn gst_asf_demux_chain (GstPad * pad, GstObject * parent,
89     GstBuffer * buf);
90 static gboolean gst_asf_demux_sink_event (GstPad * pad, GstObject * parent,
91     GstEvent * event);
92 static GstFlowReturn gst_asf_demux_process_object (GstASFDemux * demux,
93     guint8 ** p_data, guint64 * p_size);
94 static gboolean gst_asf_demux_activate (GstPad * sinkpad, GstObject * parent);
95 static gboolean gst_asf_demux_activate_mode (GstPad * sinkpad,
96     GstObject * parent, GstPadMode mode, gboolean active);
97 static void gst_asf_demux_loop (GstASFDemux * demux);
98 static void
99 gst_asf_demux_process_queued_extended_stream_objects (GstASFDemux * demux);
100 static gboolean gst_asf_demux_pull_headers (GstASFDemux * demux);
101 static void gst_asf_demux_pull_indices (GstASFDemux * demux);
102 static void gst_asf_demux_reset_stream_state_after_discont (GstASFDemux * asf);
103 static gboolean
104 gst_asf_demux_parse_data_object_start (GstASFDemux * demux, guint8 * data);
105 static void gst_asf_demux_descramble_buffer (GstASFDemux * demux,
106     AsfStream * stream, GstBuffer ** p_buffer);
107 static void gst_asf_demux_activate_stream (GstASFDemux * demux,
108     AsfStream * stream);
109 static GstStructure *gst_asf_demux_get_metadata_for_stream (GstASFDemux * d,
110     guint stream_num);
111 static GstFlowReturn gst_asf_demux_push_complete_payloads (GstASFDemux * demux,
112     gboolean force);
113
114 #define gst_asf_demux_parent_class parent_class
115 G_DEFINE_TYPE (GstASFDemux, gst_asf_demux, GST_TYPE_ELEMENT);
116
117 static void
118 gst_asf_demux_class_init (GstASFDemuxClass * klass)
119 {
120   GstElementClass *gstelement_class;
121
122   gstelement_class = (GstElementClass *) klass;
123
124   gst_element_class_set_static_metadata (gstelement_class, "ASF Demuxer",
125       "Codec/Demuxer",
126       "Demultiplexes ASF Streams", "Owen Fraser-Green <owen@discobabe.net>");
127
128   gst_element_class_add_pad_template (gstelement_class,
129       gst_static_pad_template_get (&audio_src_template));
130   gst_element_class_add_pad_template (gstelement_class,
131       gst_static_pad_template_get (&video_src_template));
132   gst_element_class_add_pad_template (gstelement_class,
133       gst_static_pad_template_get (&gst_asf_demux_sink_template));
134
135   gstelement_class->change_state =
136       GST_DEBUG_FUNCPTR (gst_asf_demux_change_state);
137   gstelement_class->send_event =
138       GST_DEBUG_FUNCPTR (gst_asf_demux_element_send_event);
139 }
140
141 static void
142 gst_asf_demux_free_stream (GstASFDemux * demux, AsfStream * stream)
143 {
144   gst_caps_replace (&stream->caps, NULL);
145   if (stream->pending_tags) {
146     gst_tag_list_free (stream->pending_tags);
147     stream->pending_tags = NULL;
148   }
149   if (stream->pad) {
150     if (stream->active)
151       gst_element_remove_pad (GST_ELEMENT_CAST (demux), stream->pad);
152     else
153       gst_object_unref (stream->pad);
154     stream->pad = NULL;
155   }
156
157   while (stream->payloads->len > 0) {
158     AsfPayload *payload;
159     guint last;
160
161     last = stream->payloads->len - 1;
162     payload = &g_array_index (stream->payloads, AsfPayload, last);
163     gst_buffer_replace (&payload->buf, NULL);
164     g_array_remove_index (stream->payloads, last);
165   }
166   if (stream->payloads) {
167     g_array_free (stream->payloads, TRUE);
168     stream->payloads = NULL;
169   }
170   if (stream->ext_props.valid) {
171     g_free (stream->ext_props.payload_extensions);
172     stream->ext_props.payload_extensions = NULL;
173   }
174 }
175
176 static void
177 gst_asf_demux_reset (GstASFDemux * demux, gboolean chain_reset)
178 {
179   GST_LOG_OBJECT (demux, "resetting");
180
181   gst_segment_init (&demux->segment, GST_FORMAT_UNDEFINED);
182   demux->segment_running = FALSE;
183   if (demux->adapter && !chain_reset) {
184     gst_adapter_clear (demux->adapter);
185     g_object_unref (demux->adapter);
186     demux->adapter = NULL;
187   }
188   if (demux->taglist) {
189     gst_tag_list_free (demux->taglist);
190     demux->taglist = NULL;
191   }
192   if (demux->metadata) {
193     gst_caps_unref (demux->metadata);
194     demux->metadata = NULL;
195   }
196   if (demux->global_metadata) {
197     gst_structure_free (demux->global_metadata);
198     demux->global_metadata = NULL;
199   }
200
201   demux->state = GST_ASF_DEMUX_STATE_HEADER;
202   g_free (demux->objpath);
203   demux->objpath = NULL;
204   g_strfreev (demux->languages);
205   demux->languages = NULL;
206   demux->num_languages = 0;
207   g_slist_foreach (demux->ext_stream_props, (GFunc) gst_mini_object_unref,
208       NULL);
209   g_slist_free (demux->ext_stream_props);
210   demux->ext_stream_props = NULL;
211
212   while (demux->old_num_streams > 0) {
213     gst_asf_demux_free_stream (demux,
214         &demux->old_stream[demux->old_num_streams - 1]);
215     --demux->old_num_streams;
216   }
217   memset (demux->old_stream, 0, sizeof (demux->old_stream));
218   demux->old_num_streams = 0;
219
220   /* when resetting for a new chained asf, we don't want to remove the pads
221    * before adding the new ones */
222   if (chain_reset) {
223     memcpy (demux->old_stream, demux->stream, sizeof (demux->stream));
224     demux->old_num_streams = demux->num_streams;
225     demux->num_streams = 0;
226   }
227
228   while (demux->num_streams > 0) {
229     gst_asf_demux_free_stream (demux, &demux->stream[demux->num_streams - 1]);
230     --demux->num_streams;
231   }
232   memset (demux->stream, 0, sizeof (demux->stream));
233   if (!chain_reset) {
234     /* do not remove those for not adding pads with same name */
235     demux->num_audio_streams = 0;
236     demux->num_video_streams = 0;
237   }
238   demux->num_streams = 0;
239   demux->activated_streams = FALSE;
240   demux->first_ts = GST_CLOCK_TIME_NONE;
241   demux->segment_ts = GST_CLOCK_TIME_NONE;
242   demux->in_gap = 0;
243   if (!chain_reset)
244     gst_segment_init (&demux->in_segment, GST_FORMAT_UNDEFINED);
245   demux->state = GST_ASF_DEMUX_STATE_HEADER;
246   demux->seekable = FALSE;
247   demux->broadcast = FALSE;
248   demux->sidx_interval = 0;
249   demux->sidx_num_entries = 0;
250   g_free (demux->sidx_entries);
251   demux->sidx_entries = NULL;
252
253   demux->speed_packets = 1;
254
255   if (chain_reset) {
256     GST_LOG_OBJECT (demux, "Restarting");
257     gst_segment_init (&demux->segment, GST_FORMAT_TIME);
258     demux->need_newsegment = TRUE;
259     demux->segment_running = FALSE;
260     demux->accurate = FALSE;
261     demux->metadata = gst_caps_new_empty ();
262     demux->global_metadata = gst_structure_new_empty ("metadata");
263     demux->data_size = 0;
264     demux->data_offset = 0;
265     demux->index_offset = 0;
266   } else {
267     demux->base_offset = 0;
268   }
269 }
270
271 static void
272 gst_asf_demux_init (GstASFDemux * demux)
273 {
274   demux->sinkpad =
275       gst_pad_new_from_static_template (&gst_asf_demux_sink_template, "sink");
276   gst_pad_set_chain_function (demux->sinkpad,
277       GST_DEBUG_FUNCPTR (gst_asf_demux_chain));
278   gst_pad_set_event_function (demux->sinkpad,
279       GST_DEBUG_FUNCPTR (gst_asf_demux_sink_event));
280   gst_pad_set_activate_function (demux->sinkpad,
281       GST_DEBUG_FUNCPTR (gst_asf_demux_activate));
282   gst_pad_set_activatemode_function (demux->sinkpad,
283       GST_DEBUG_FUNCPTR (gst_asf_demux_activate_mode));
284   gst_element_add_pad (GST_ELEMENT (demux), demux->sinkpad);
285
286   /* set initial state */
287   gst_asf_demux_reset (demux, FALSE);
288 }
289
290 static gboolean
291 gst_asf_demux_activate (GstPad * sinkpad, GstObject * parent)
292 {
293   GstQuery *query;
294   gboolean pull_mode;
295
296   query = gst_query_new_scheduling ();
297
298   if (!gst_pad_peer_query (sinkpad, query)) {
299     gst_query_unref (query);
300     goto activate_push;
301   }
302
303   pull_mode = gst_query_has_scheduling_mode (query, GST_PAD_MODE_PULL);
304   gst_query_unref (query);
305
306   if (!pull_mode)
307     goto activate_push;
308
309   GST_DEBUG_OBJECT (sinkpad, "activating pull");
310   return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PULL, TRUE);
311
312 activate_push:
313   {
314     GST_DEBUG_OBJECT (sinkpad, "activating push");
315     return gst_pad_activate_mode (sinkpad, GST_PAD_MODE_PUSH, TRUE);
316   }
317 }
318
319 static gboolean
320 gst_asf_demux_activate_mode (GstPad * sinkpad, GstObject * parent,
321     GstPadMode mode, gboolean active)
322 {
323   gboolean res;
324   GstASFDemux *demux;
325
326   demux = GST_ASF_DEMUX (parent);
327
328   switch (mode) {
329     case GST_PAD_MODE_PUSH:
330       demux->state = GST_ASF_DEMUX_STATE_HEADER;
331       demux->streaming = TRUE;
332       res = TRUE;
333       break;
334     case GST_PAD_MODE_PULL:
335       if (active) {
336         demux->state = GST_ASF_DEMUX_STATE_HEADER;
337         demux->streaming = FALSE;
338
339         res = gst_pad_start_task (sinkpad, (GstTaskFunction) gst_asf_demux_loop,
340             demux, NULL);
341       } else {
342         res = gst_pad_stop_task (sinkpad);
343       }
344       break;
345     default:
346       res = FALSE;
347       break;
348   }
349   return res;
350 }
351
352 static gboolean
353 gst_asf_demux_sink_event (GstPad * pad, GstObject * parent, GstEvent * event)
354 {
355   GstASFDemux *demux;
356   gboolean ret = TRUE;
357
358   demux = GST_ASF_DEMUX (parent);
359
360   GST_LOG_OBJECT (demux, "handling %s event", GST_EVENT_TYPE_NAME (event));
361   switch (GST_EVENT_TYPE (event)) {
362     case GST_EVENT_SEGMENT:{
363       const GstSegment *segment;
364
365       gst_event_parse_segment (event, &segment);
366
367       if (segment->format == GST_FORMAT_BYTES) {
368         if (demux->packet_size && segment->start > demux->data_offset)
369           demux->packet = (segment->start - demux->data_offset) /
370               demux->packet_size;
371         else
372           demux->packet = 0;
373       } else if (segment->format == GST_FORMAT_TIME) {
374         /* do not know packet position, not really a problem */
375         demux->packet = -1;
376       } else {
377         GST_WARNING_OBJECT (demux, "unsupported newsegment format, ignoring");
378         gst_event_unref (event);
379         break;
380       }
381
382       /* record upstream segment for interpolation */
383       if (segment->format != demux->in_segment.format)
384         gst_segment_init (&demux->in_segment, GST_FORMAT_UNDEFINED);
385       gst_segment_copy_into (segment, &demux->in_segment);
386
387       /* in either case, clear some state and generate newsegment later on */
388       GST_OBJECT_LOCK (demux);
389       demux->segment_ts = GST_CLOCK_TIME_NONE;
390       demux->in_gap = GST_CLOCK_TIME_NONE;
391       demux->need_newsegment = TRUE;
392       gst_asf_demux_reset_stream_state_after_discont (demux);
393       GST_OBJECT_UNLOCK (demux);
394
395       gst_event_unref (event);
396       break;
397     }
398     case GST_EVENT_EOS:{
399       GstFlowReturn flow;
400
401       if (demux->state == GST_ASF_DEMUX_STATE_HEADER) {
402         GST_ELEMENT_ERROR (demux, STREAM, DEMUX,
403             (_("This stream contains no data.")),
404             ("got eos and didn't receive a complete header object"));
405         break;
406       }
407       flow = gst_asf_demux_push_complete_payloads (demux, TRUE);
408       if (flow < GST_FLOW_EOS || flow == GST_FLOW_NOT_LINKED) {
409         GST_ELEMENT_ERROR (demux, STREAM, FAILED,
410             (_("Internal data stream error.")),
411             ("streaming stopped, reason %s", gst_flow_get_name (flow)));
412         break;
413       }
414
415       GST_OBJECT_LOCK (demux);
416       gst_adapter_clear (demux->adapter);
417       GST_OBJECT_UNLOCK (demux);
418       gst_asf_demux_send_event_unlocked (demux, event);
419       break;
420     }
421
422     case GST_EVENT_FLUSH_STOP:
423       GST_OBJECT_LOCK (demux);
424       gst_asf_demux_reset_stream_state_after_discont (demux);
425       GST_OBJECT_UNLOCK (demux);
426       gst_asf_demux_send_event_unlocked (demux, event);
427       /* upon activation, latency is no longer introduced, e.g. after seek */
428       if (demux->activated_streams)
429         demux->latency = 0;
430       break;
431
432     default:
433       ret = gst_pad_event_default (pad, parent, event);
434       break;
435   }
436
437   return ret;
438 }
439
440 static gboolean
441 gst_asf_demux_seek_index_lookup (GstASFDemux * demux, guint * packet,
442     GstClockTime seek_time, GstClockTime * p_idx_time, guint * speed,
443     gboolean next, gboolean * eos)
444 {
445   GstClockTime idx_time;
446   guint idx;
447
448   if (eos)
449     *eos = FALSE;
450
451   if (G_UNLIKELY (demux->sidx_num_entries == 0 || demux->sidx_interval == 0))
452     return FALSE;
453
454   idx = (guint) ((seek_time + demux->preroll) / demux->sidx_interval);
455
456   if (next) {
457     /* if we want the next keyframe, we have to go forward till we find
458        a different packet number */
459     guint idx2 = idx;
460     if (idx >= demux->sidx_num_entries - 1) {
461       /* If we get here, we're asking for next keyframe after the last one. There isn't one. */
462       if (eos)
463         *eos = TRUE;
464       return FALSE;
465     }
466     for (idx2 = idx + 1; idx2 < demux->sidx_num_entries; ++idx2) {
467       if (demux->sidx_entries[idx].packet != demux->sidx_entries[idx2].packet) {
468         idx = idx2;
469         break;
470       }
471     }
472   }
473
474   if (G_UNLIKELY (idx >= demux->sidx_num_entries)) {
475     if (eos)
476       *eos = TRUE;
477     return FALSE;
478   }
479
480   *packet = demux->sidx_entries[idx].packet;
481   if (speed)
482     *speed = demux->sidx_entries[idx].count;
483
484   /* so we get closer to the actual time of the packet ... actually, let's not
485    * do this, since we throw away superfluous payloads before the seek position
486    * anyway; this way, our key unit seek 'snap resolution' is a bit better
487    * (ie. same as index resolution) */
488   /*
489      while (idx > 0 && demux->sidx_entries[idx-1] == demux->sidx_entries[idx])
490      --idx;
491    */
492
493   idx_time = demux->sidx_interval * idx;
494   if (G_LIKELY (idx_time >= demux->preroll))
495     idx_time -= demux->preroll;
496
497   GST_DEBUG_OBJECT (demux, "%" GST_TIME_FORMAT " => packet %u at %"
498       GST_TIME_FORMAT, GST_TIME_ARGS (seek_time), *packet,
499       GST_TIME_ARGS (idx_time));
500
501   if (G_LIKELY (p_idx_time))
502     *p_idx_time = idx_time;
503
504   return TRUE;
505 }
506
507 static void
508 gst_asf_demux_reset_stream_state_after_discont (GstASFDemux * demux)
509 {
510   guint n;
511
512   gst_adapter_clear (demux->adapter);
513
514   GST_DEBUG_OBJECT (demux, "reset stream state");
515
516   for (n = 0; n < demux->num_streams; n++) {
517     demux->stream[n].discont = TRUE;
518     demux->stream[n].last_flow = GST_FLOW_OK;
519
520     while (demux->stream[n].payloads->len > 0) {
521       AsfPayload *payload;
522       guint last;
523
524       last = demux->stream[n].payloads->len - 1;
525       payload = &g_array_index (demux->stream[n].payloads, AsfPayload, last);
526       gst_buffer_replace (&payload->buf, NULL);
527       g_array_remove_index (demux->stream[n].payloads, last);
528     }
529   }
530 }
531
532 static void
533 gst_asf_demux_mark_discont (GstASFDemux * demux)
534 {
535   guint n;
536
537   GST_DEBUG_OBJECT (demux, "Mark stream discont");
538
539   for (n = 0; n < demux->num_streams; n++)
540     demux->stream[n].discont = TRUE;
541 }
542
543 /* do a seek in push based mode */
544 static gboolean
545 gst_asf_demux_handle_seek_push (GstASFDemux * demux, GstEvent * event)
546 {
547   gdouble rate;
548   GstFormat format;
549   GstSeekFlags flags;
550   GstSeekType cur_type, stop_type;
551   gint64 cur, stop;
552   guint packet;
553   gboolean res;
554
555   gst_event_parse_seek (event, &rate, &format, &flags, &cur_type, &cur,
556       &stop_type, &stop);
557
558   stop_type = GST_SEEK_TYPE_NONE;
559   stop = -1;
560
561   GST_DEBUG_OBJECT (demux, "seeking to %" GST_TIME_FORMAT, GST_TIME_ARGS (cur));
562
563   /* determine packet, by index or by estimation */
564   if (!gst_asf_demux_seek_index_lookup (demux, &packet, cur, NULL, NULL, FALSE,
565           NULL)) {
566     packet =
567         (guint) gst_util_uint64_scale (demux->num_packets, cur,
568         demux->play_time);
569   }
570
571   if (packet > demux->num_packets) {
572     GST_DEBUG_OBJECT (demux, "could not determine packet to seek to, "
573         "seek aborted.");
574     return FALSE;
575   }
576
577   GST_DEBUG_OBJECT (demux, "seeking to packet %d", packet);
578
579   cur = demux->data_offset + (packet * demux->packet_size);
580
581   GST_DEBUG_OBJECT (demux, "Pushing BYTE seek rate %g, "
582       "start %" G_GINT64_FORMAT ", stop %" G_GINT64_FORMAT, rate, cur, stop);
583   /* BYTE seek event */
584   event = gst_event_new_seek (rate, GST_FORMAT_BYTES, flags, cur_type, cur,
585       stop_type, stop);
586   res = gst_pad_push_event (demux->sinkpad, event);
587
588   return res;
589 }
590
591 static gboolean
592 gst_asf_demux_handle_seek_event (GstASFDemux * demux, GstEvent * event)
593 {
594   GstClockTime idx_time;
595   GstSegment segment;
596   GstSeekFlags flags;
597   GstSeekType cur_type, stop_type;
598   GstFormat format;
599   gboolean only_need_update;
600   gboolean keyunit_sync, after, before, next;
601   gboolean flush;
602   gdouble rate;
603   gint64 cur, stop;
604   gint64 seek_time;
605   guint packet, speed_count = 1;
606   gboolean eos;
607
608   if (G_UNLIKELY (demux->seekable == FALSE || demux->packet_size == 0 ||
609           demux->num_packets == 0 || demux->play_time == 0)) {
610     GST_LOG_OBJECT (demux, "stream is not seekable");
611     return FALSE;
612   }
613
614   if (G_UNLIKELY (!demux->activated_streams)) {
615     GST_LOG_OBJECT (demux, "streams not yet activated, ignoring seek");
616     return FALSE;
617   }
618
619   gst_event_parse_seek (event, &rate, &format, &flags, &cur_type, &cur,
620       &stop_type, &stop);
621
622   if (G_UNLIKELY (format != GST_FORMAT_TIME)) {
623     GST_LOG_OBJECT (demux, "seeking is only supported in TIME format");
624     return FALSE;
625   }
626
627   if (G_UNLIKELY (rate <= 0.0)) {
628     GST_LOG_OBJECT (demux, "backward playback is not supported yet");
629     return FALSE;
630   }
631
632   flush = ((flags & GST_SEEK_FLAG_FLUSH) == GST_SEEK_FLAG_FLUSH);
633   demux->accurate =
634       ((flags & GST_SEEK_FLAG_ACCURATE) == GST_SEEK_FLAG_ACCURATE);
635   keyunit_sync = ((flags & GST_SEEK_FLAG_KEY_UNIT) == GST_SEEK_FLAG_KEY_UNIT);
636   after = ((flags & GST_SEEK_FLAG_SNAP_AFTER) == GST_SEEK_FLAG_SNAP_AFTER);
637   before = ((flags & GST_SEEK_FLAG_SNAP_BEFORE) == GST_SEEK_FLAG_SNAP_BEFORE);
638   next = after && !before;
639
640   if (G_UNLIKELY (demux->streaming)) {
641     /* support it safely needs more segment handling, e.g. closing etc */
642     if (!flush) {
643       GST_LOG_OBJECT (demux, "streaming; non-flushing seek not supported");
644       return FALSE;
645     }
646     /* we can (re)construct the start later on, but not the end */
647     if (stop_type != GST_SEEK_TYPE_NONE) {
648       GST_LOG_OBJECT (demux, "streaming; end type must be NONE");
649       return FALSE;
650     }
651     gst_event_ref (event);
652     /* upstream might handle TIME seek, e.g. mms or rtsp,
653      * or not, e.g. http, then we give it a hand */
654     if (!gst_pad_push_event (demux->sinkpad, event))
655       return gst_asf_demux_handle_seek_push (demux, event);
656     else
657       return TRUE;
658   }
659
660   /* unlock the streaming thread */
661   if (G_LIKELY (flush)) {
662     gst_pad_push_event (demux->sinkpad, gst_event_new_flush_start ());
663     gst_asf_demux_send_event_unlocked (demux, gst_event_new_flush_start ());
664   } else {
665     gst_pad_pause_task (demux->sinkpad);
666   }
667
668   /* grab the stream lock so that streaming cannot continue, for
669    * non flushing seeks when the element is in PAUSED this could block
670    * forever */
671   GST_PAD_STREAM_LOCK (demux->sinkpad);
672
673   /* we now can stop flushing, since we have the stream lock now */
674   gst_pad_push_event (demux->sinkpad, gst_event_new_flush_stop (TRUE));
675
676   if (G_LIKELY (flush))
677     gst_asf_demux_send_event_unlocked (demux, gst_event_new_flush_stop (TRUE));
678
679   /* operating on copy of segment until we know the seek worked */
680   segment = demux->segment;
681
682   if (G_UNLIKELY (demux->segment_running && !flush)) {
683     GstSegment newsegment;
684     GstEvent *newseg;
685
686     /* create the segment event to close the current segment */
687     gst_segment_copy_into (&segment, &newsegment);
688     newseg = gst_event_new_segment (&newsegment);
689
690     gst_asf_demux_send_event_unlocked (demux, newseg);
691   }
692
693   gst_segment_do_seek (&segment, rate, format, flags, cur_type,
694       cur, stop_type, stop, &only_need_update);
695
696   GST_DEBUG_OBJECT (demux, "seeking to time %" GST_TIME_FORMAT ", segment: "
697       "%" GST_SEGMENT_FORMAT, GST_TIME_ARGS (segment.start), &segment);
698
699   seek_time = segment.start;
700
701   /* FIXME: should check the KEY_UNIT flag; need to adjust position to
702    * real start of data and segment_start to indexed time for key unit seek*/
703   if (G_UNLIKELY (!gst_asf_demux_seek_index_lookup (demux, &packet, seek_time,
704               &idx_time, &speed_count, next, &eos))) {
705     gint64 offset;
706
707     if (eos) {
708       demux->packet = demux->num_packets;
709       goto skip;
710     }
711
712     /* First try to query our source to see if it can convert for us. This is
713        the case when our source is an mms stream, notice that in this case
714        gstmms will do a time based seek to get the byte offset, this is not a
715        problem as the seek to this offset needs to happen anway. */
716     if (gst_pad_peer_query_convert (demux->sinkpad, GST_FORMAT_TIME, seek_time,
717             GST_FORMAT_BYTES, &offset)) {
718       packet = (offset - demux->data_offset) / demux->packet_size;
719       GST_LOG_OBJECT (demux, "convert %" GST_TIME_FORMAT
720           " to bytes query result: %" G_GINT64_FORMAT ", data_ofset: %"
721           G_GINT64_FORMAT ", packet_size: %u," " resulting packet: %u\n",
722           GST_TIME_ARGS (seek_time), offset, demux->data_offset,
723           demux->packet_size, packet);
724     } else {
725       /* FIXME: For streams containing video, seek to an earlier position in
726        * the hope of hitting a keyframe and let the sinks throw away the stuff
727        * before the segment start. For audio-only this is unnecessary as every
728        * frame is 'key'. */
729       if (flush && (demux->accurate || (keyunit_sync && !next))
730           && demux->num_video_streams > 0) {
731         seek_time -= 5 * GST_SECOND;
732         if (seek_time < 0)
733           seek_time = 0;
734       }
735
736       packet = (guint) gst_util_uint64_scale (demux->num_packets,
737           seek_time, demux->play_time);
738
739       if (packet > demux->num_packets)
740         packet = demux->num_packets;
741     }
742   } else {
743     if (G_LIKELY (keyunit_sync)) {
744       GST_DEBUG_OBJECT (demux, "key unit seek, adjust seek_time = %"
745           GST_TIME_FORMAT " to index_time = %" GST_TIME_FORMAT,
746           GST_TIME_ARGS (seek_time), GST_TIME_ARGS (idx_time));
747       segment.start = idx_time;
748       segment.position = idx_time;
749       segment.time = idx_time;
750     }
751   }
752
753   GST_DEBUG_OBJECT (demux, "seeking to packet %u (%d)", packet, speed_count);
754
755   GST_OBJECT_LOCK (demux);
756   demux->segment = segment;
757   demux->packet = packet;
758   demux->need_newsegment = TRUE;
759   demux->speed_packets = speed_count;
760   gst_asf_demux_reset_stream_state_after_discont (demux);
761   GST_OBJECT_UNLOCK (demux);
762
763 skip:
764   /* restart our task since it might have been stopped when we did the flush */
765   gst_pad_start_task (demux->sinkpad, (GstTaskFunction) gst_asf_demux_loop,
766       demux, NULL);
767
768   /* streaming can continue now */
769   GST_PAD_STREAM_UNLOCK (demux->sinkpad);
770
771   return TRUE;
772 }
773
774 static gboolean
775 gst_asf_demux_handle_src_event (GstPad * pad, GstObject * parent,
776     GstEvent * event)
777 {
778   GstASFDemux *demux;
779   gboolean ret;
780
781   demux = GST_ASF_DEMUX (parent);
782
783   switch (GST_EVENT_TYPE (event)) {
784     case GST_EVENT_SEEK:
785       GST_LOG_OBJECT (pad, "seek event");
786       ret = gst_asf_demux_handle_seek_event (demux, event);
787       gst_event_unref (event);
788       break;
789     case GST_EVENT_QOS:
790     case GST_EVENT_NAVIGATION:
791       /* just drop these two silently */
792       gst_event_unref (event);
793       ret = FALSE;
794       break;
795     default:
796       GST_LOG_OBJECT (pad, "%s event", GST_EVENT_TYPE_NAME (event));
797       ret = gst_pad_event_default (pad, parent, event);
798       break;
799   }
800
801   return ret;
802 }
803
804 static inline guint32
805 gst_asf_demux_identify_guid (const ASFGuidHash * guids, ASFGuid * guid)
806 {
807   guint32 ret;
808
809   ret = gst_asf_identify_guid (guids, guid);
810
811   GST_LOG ("%s  0x%08x-0x%08x-0x%08x-0x%08x",
812       gst_asf_get_guid_nick (guids, ret),
813       guid->v1, guid->v2, guid->v3, guid->v4);
814
815   return ret;
816 }
817
818 typedef struct
819 {
820   AsfObjectID id;
821   guint64 size;
822 } AsfObject;
823
824
825 /* expect is true when the user is expeting an object,
826  * when false, it will give no warnings if the object
827  * is not identified
828  */
829 static gboolean
830 asf_demux_peek_object (GstASFDemux * demux, const guint8 * data,
831     guint data_len, AsfObject * object, gboolean expect)
832 {
833   ASFGuid guid;
834
835   if (data_len < ASF_OBJECT_HEADER_SIZE)
836     return FALSE;
837
838   guid.v1 = GST_READ_UINT32_LE (data + 0);
839   guid.v2 = GST_READ_UINT32_LE (data + 4);
840   guid.v3 = GST_READ_UINT32_LE (data + 8);
841   guid.v4 = GST_READ_UINT32_LE (data + 12);
842
843   object->size = GST_READ_UINT64_LE (data + 16);
844
845   /* FIXME: make asf_demux_identify_object_guid() */
846   object->id = gst_asf_demux_identify_guid (asf_object_guids, &guid);
847   if (object->id == ASF_OBJ_UNDEFINED && expect) {
848     GST_WARNING_OBJECT (demux, "Unknown object %08x-%08x-%08x-%08x",
849         guid.v1, guid.v2, guid.v3, guid.v4);
850   }
851
852   return TRUE;
853 }
854
855 static void
856 gst_asf_demux_release_old_pads (GstASFDemux * demux)
857 {
858   GST_DEBUG_OBJECT (demux, "Releasing old pads");
859
860   while (demux->old_num_streams > 0) {
861     gst_pad_push_event (demux->old_stream[demux->old_num_streams - 1].pad,
862         gst_event_new_eos ());
863     gst_asf_demux_free_stream (demux,
864         &demux->old_stream[demux->old_num_streams - 1]);
865     --demux->old_num_streams;
866   }
867   memset (demux->old_stream, 0, sizeof (demux->old_stream));
868   demux->old_num_streams = 0;
869 }
870
871 static GstFlowReturn
872 gst_asf_demux_chain_headers (GstASFDemux * demux)
873 {
874   GstFlowReturn flow;
875   AsfObject obj;
876   guint8 *header_data, *data = NULL;
877   const guint8 *cdata = NULL;
878   guint64 header_size;
879
880   cdata = (guint8 *) gst_adapter_map (demux->adapter, ASF_OBJECT_HEADER_SIZE);
881   if (cdata == NULL)
882     goto need_more_data;
883
884   asf_demux_peek_object (demux, cdata, ASF_OBJECT_HEADER_SIZE, &obj, TRUE);
885   if (obj.id != ASF_OBJ_HEADER)
886     goto wrong_type;
887
888   GST_LOG_OBJECT (demux, "header size = %u", (guint) obj.size);
889
890   /* + 50 for non-packet data at beginning of ASF_OBJ_DATA */
891   if (gst_adapter_available (demux->adapter) < obj.size + 50)
892     goto need_more_data;
893
894   data = gst_adapter_take (demux->adapter, obj.size + 50);
895
896   header_data = data;
897   header_size = obj.size;
898   flow = gst_asf_demux_process_object (demux, &header_data, &header_size);
899   if (flow != GST_FLOW_OK)
900     goto parse_failed;
901
902   /* calculate where the packet data starts */
903   demux->data_offset = obj.size + 50;
904
905   /* now parse the beginning of the ASF_OBJ_DATA object */
906   if (!gst_asf_demux_parse_data_object_start (demux, data + obj.size))
907     goto wrong_type;
908
909   if (demux->num_streams == 0)
910     goto no_streams;
911
912   g_free (data);
913   return GST_FLOW_OK;
914
915 /* NON-FATAL */
916 need_more_data:
917   {
918     GST_LOG_OBJECT (demux, "not enough data in adapter yet");
919     return GST_FLOW_OK;
920   }
921
922 /* ERRORS */
923 wrong_type:
924   {
925     GST_ELEMENT_ERROR (demux, STREAM, WRONG_TYPE, (NULL),
926         ("This doesn't seem to be an ASF file"));
927     g_free (data);
928     return GST_FLOW_ERROR;
929   }
930 no_streams:
931 parse_failed:
932   {
933     GST_ELEMENT_ERROR (demux, STREAM, DEMUX, (NULL),
934         ("header parsing failed, or no streams found, flow = %s",
935             gst_flow_get_name (flow)));
936     g_free (data);
937     return GST_FLOW_ERROR;
938   }
939 }
940
941 static GstFlowReturn
942 gst_asf_demux_aggregate_flow_return (GstASFDemux * demux, AsfStream * stream,
943     GstFlowReturn flow)
944 {
945   int i;
946
947   GST_DEBUG_OBJECT (demux, "Aggregating");
948
949   /* Store the value */
950   stream->last_flow = flow;
951
952   /* any other error that is not not-linked can be returned right away */
953   if (flow != GST_FLOW_NOT_LINKED)
954     goto done;
955
956   for (i = 0; i < demux->num_streams; i++) {
957     if (demux->stream[i].active) {
958       flow = demux->stream[i].last_flow;
959       GST_DEBUG_OBJECT (demux, "Aggregating: flow %i return %s", i,
960           gst_flow_get_name (flow));
961       if (flow != GST_FLOW_NOT_LINKED)
962         goto done;
963     }
964   }
965
966   /* If we got here, then all our active streams are not linked */
967 done:
968   return flow;
969 }
970
971 static gboolean
972 gst_asf_demux_pull_data (GstASFDemux * demux, guint64 offset, guint size,
973     GstBuffer ** p_buf, GstFlowReturn * p_flow)
974 {
975   gsize buffer_size;
976   GstFlowReturn flow;
977
978   GST_LOG_OBJECT (demux, "pulling buffer at %" G_GUINT64_FORMAT "+%u",
979       offset, size);
980
981   flow = gst_pad_pull_range (demux->sinkpad, offset, size, p_buf);
982
983   if (G_LIKELY (p_flow))
984     *p_flow = flow;
985
986   if (G_UNLIKELY (flow != GST_FLOW_OK)) {
987     GST_DEBUG_OBJECT (demux, "flow %s pulling buffer at %" G_GUINT64_FORMAT
988         "+%u", gst_flow_get_name (flow), offset, size);
989     *p_buf = NULL;
990     return FALSE;
991   }
992
993   g_assert (*p_buf != NULL);
994
995   buffer_size = gst_buffer_get_size (*p_buf);
996   if (G_UNLIKELY (buffer_size < size)) {
997     GST_DEBUG_OBJECT (demux, "short read pulling buffer at %" G_GUINT64_FORMAT
998         "+%u (got only %" G_GSIZE_FORMAT " bytes)", offset, size, buffer_size);
999     gst_buffer_unref (*p_buf);
1000     if (G_LIKELY (p_flow))
1001       *p_flow = GST_FLOW_EOS;
1002     *p_buf = NULL;
1003     return FALSE;
1004   }
1005
1006   return TRUE;
1007 }
1008
1009 static void
1010 gst_asf_demux_pull_indices (GstASFDemux * demux)
1011 {
1012   GstBuffer *buf = NULL;
1013   guint64 offset;
1014   guint num_read = 0;
1015
1016   offset = demux->index_offset;
1017
1018   if (G_UNLIKELY (offset == 0)) {
1019     GST_DEBUG_OBJECT (demux, "can't read indices, don't know index offset");
1020     return;
1021   }
1022
1023   while (gst_asf_demux_pull_data (demux, offset, 16 + 8, &buf, NULL)) {
1024     GstFlowReturn flow;
1025     AsfObject obj;
1026     GstMapInfo map;
1027     guint8 *bufdata;
1028
1029     gst_buffer_map (buf, &map, GST_MAP_READ);
1030     g_assert (map.size >= 16 + 8);
1031     asf_demux_peek_object (demux, map.data, 16 + 8, &obj, TRUE);
1032     gst_buffer_unmap (buf, &map);
1033     gst_buffer_replace (&buf, NULL);
1034
1035     /* check for sanity */
1036     if (G_UNLIKELY (obj.size > (5 * 1024 * 1024))) {
1037       GST_DEBUG_OBJECT (demux, "implausible index object size, bailing out");
1038       break;
1039     }
1040
1041     if (G_UNLIKELY (!gst_asf_demux_pull_data (demux, offset, obj.size, &buf,
1042                 NULL)))
1043       break;
1044
1045     GST_LOG_OBJECT (demux, "index object at offset 0x%" G_GINT64_MODIFIER "X"
1046         ", size %u", offset, (guint) obj.size);
1047
1048     offset += obj.size;         /* increase before _process_object changes it */
1049
1050     gst_buffer_map (buf, &map, GST_MAP_READ);
1051     g_assert (map.size >= obj.size);
1052     bufdata = (guint8 *) map.data;
1053     flow = gst_asf_demux_process_object (demux, &bufdata, &obj.size);
1054     gst_buffer_unmap (buf, &map);
1055     gst_buffer_replace (&buf, NULL);
1056
1057     if (G_UNLIKELY (flow != GST_FLOW_OK))
1058       break;
1059
1060     ++num_read;
1061   }
1062   GST_DEBUG_OBJECT (demux, "read %u index objects", num_read);
1063 }
1064
1065 static gboolean
1066 gst_asf_demux_parse_data_object_start (GstASFDemux * demux, guint8 * data)
1067 {
1068   AsfObject obj;
1069
1070   asf_demux_peek_object (demux, data, 50, &obj, TRUE);
1071   if (obj.id != ASF_OBJ_DATA) {
1072     GST_WARNING_OBJECT (demux, "headers not followed by a DATA object");
1073     return FALSE;
1074   }
1075
1076   demux->state = GST_ASF_DEMUX_STATE_DATA;
1077
1078   if (!demux->broadcast && obj.size > 50) {
1079     demux->data_size = obj.size - 50;
1080     /* CHECKME: for at least one file this is off by +158 bytes?! */
1081     demux->index_offset = demux->data_offset + demux->data_size;
1082   } else {
1083     demux->data_size = 0;
1084     demux->index_offset = 0;
1085   }
1086
1087   demux->packet = 0;
1088
1089   if (!demux->broadcast) {
1090     /* skip object header (24 bytes) and file GUID (16 bytes) */
1091     demux->num_packets = GST_READ_UINT64_LE (data + (16 + 8) + 16);
1092   } else {
1093     demux->num_packets = 0;
1094   }
1095
1096   if (demux->num_packets == 0)
1097     demux->seekable = FALSE;
1098
1099   /* fallback in the unlikely case that headers are inconsistent, can't hurt */
1100   if (demux->data_size == 0 && demux->num_packets > 0) {
1101     demux->data_size = demux->num_packets * demux->packet_size;
1102     demux->index_offset = demux->data_offset + demux->data_size;
1103   }
1104
1105   /* process pending stream objects and create pads for those */
1106   gst_asf_demux_process_queued_extended_stream_objects (demux);
1107
1108   GST_INFO_OBJECT (demux, "Stream has %" G_GUINT64_FORMAT " packets, "
1109       "data_offset=%" G_GINT64_FORMAT ", data_size=%" G_GINT64_FORMAT
1110       ", index_offset=%" G_GUINT64_FORMAT, demux->num_packets,
1111       demux->data_offset, demux->data_size, demux->index_offset);
1112
1113   return TRUE;
1114 }
1115
1116 static gboolean
1117 gst_asf_demux_pull_headers (GstASFDemux * demux)
1118 {
1119   GstFlowReturn flow;
1120   AsfObject obj;
1121   GstBuffer *buf = NULL;
1122   guint64 size;
1123   GstMapInfo map;
1124   guint8 *bufdata;
1125
1126   GST_LOG_OBJECT (demux, "reading headers");
1127
1128   /* pull HEADER object header, so we know its size */
1129   if (!gst_asf_demux_pull_data (demux, demux->base_offset, 16 + 8, &buf, NULL))
1130     goto read_failed;
1131
1132   gst_buffer_map (buf, &map, GST_MAP_READ);
1133   g_assert (map.size >= 16 + 8);
1134   asf_demux_peek_object (demux, map.data, 16 + 8, &obj, TRUE);
1135   gst_buffer_unmap (buf, &map);
1136   gst_buffer_replace (&buf, NULL);
1137
1138   if (obj.id != ASF_OBJ_HEADER)
1139     goto wrong_type;
1140
1141   GST_LOG_OBJECT (demux, "header size = %u", (guint) obj.size);
1142
1143   /* pull HEADER object */
1144   if (!gst_asf_demux_pull_data (demux, demux->base_offset, obj.size, &buf,
1145           NULL))
1146     goto read_failed;
1147
1148   size = obj.size;              /* don't want obj.size changed */
1149   gst_buffer_map (buf, &map, GST_MAP_READ);
1150   g_assert (map.size >= size);
1151   bufdata = (guint8 *) map.data;
1152   flow = gst_asf_demux_process_object (demux, &bufdata, &size);
1153   gst_buffer_unmap (buf, &map);
1154   gst_buffer_replace (&buf, NULL);
1155
1156   if (flow != GST_FLOW_OK) {
1157     GST_WARNING_OBJECT (demux, "process_object: %s", gst_flow_get_name (flow));
1158     goto parse_failed;
1159   }
1160
1161   /* calculate where the packet data starts */
1162   demux->data_offset = demux->base_offset + obj.size + 50;
1163
1164   /* now pull beginning of DATA object before packet data */
1165   if (!gst_asf_demux_pull_data (demux, demux->base_offset + obj.size, 50, &buf,
1166           NULL))
1167     goto read_failed;
1168
1169   gst_buffer_map (buf, &map, GST_MAP_READ);
1170   g_assert (map.size >= size);
1171   bufdata = (guint8 *) map.data;
1172   if (!gst_asf_demux_parse_data_object_start (demux, bufdata))
1173     goto wrong_type;
1174
1175   if (demux->num_streams == 0)
1176     goto no_streams;
1177
1178   gst_buffer_unmap (buf, &map);
1179   gst_buffer_replace (&buf, NULL);
1180
1181   return TRUE;
1182
1183 /* ERRORS */
1184 wrong_type:
1185   {
1186     if (buf != NULL) {
1187       gst_buffer_unmap (buf, &map);
1188       gst_buffer_replace (&buf, NULL);
1189     }
1190     GST_ELEMENT_ERROR (demux, STREAM, WRONG_TYPE, (NULL),
1191         ("This doesn't seem to be an ASF file"));
1192     return FALSE;
1193   }
1194
1195 no_streams:
1196 read_failed:
1197 parse_failed:
1198   {
1199     if (buf)
1200       gst_buffer_unmap (buf, &map);
1201     gst_buffer_replace (&buf, NULL);
1202     GST_ELEMENT_ERROR (demux, STREAM, DEMUX, (NULL), (NULL));
1203     return FALSE;
1204   }
1205 }
1206
1207 static gboolean
1208 all_streams_prerolled (GstASFDemux * demux)
1209 {
1210   GstClockTime preroll_time;
1211   guint i, num_no_data = 0;
1212
1213   /* Allow at least 500ms of preroll_time  */
1214   preroll_time = MAX (demux->preroll, 500 * GST_MSECOND);
1215
1216   /* returns TRUE as long as there isn't a stream which (a) has data queued
1217    * and (b) the timestamp of last piece of data queued is < demux->preroll
1218    * AND there is at least one other stream with data queued */
1219   for (i = 0; i < demux->num_streams; ++i) {
1220     AsfPayload *last_payload;
1221     AsfStream *stream;
1222     guint last_idx;
1223
1224     stream = &demux->stream[i];
1225     if (G_UNLIKELY (stream->payloads->len == 0)) {
1226       ++num_no_data;
1227       GST_LOG_OBJECT (stream->pad, "no data queued");
1228       continue;
1229     }
1230
1231     last_idx = stream->payloads->len - 1;
1232     last_payload = &g_array_index (stream->payloads, AsfPayload, last_idx);
1233
1234     GST_LOG_OBJECT (stream->pad, "checking if %" GST_TIME_FORMAT " > %"
1235         GST_TIME_FORMAT, GST_TIME_ARGS (last_payload->ts),
1236         GST_TIME_ARGS (preroll_time));
1237     if (G_UNLIKELY (last_payload->ts <= preroll_time)) {
1238       GST_LOG_OBJECT (stream->pad, "not beyond preroll point yet");
1239       return FALSE;
1240     }
1241   }
1242
1243   if (G_UNLIKELY (num_no_data == demux->num_streams))
1244     return FALSE;
1245
1246   return TRUE;
1247 }
1248
1249 #if 0
1250 static gboolean
1251 gst_asf_demux_have_mutually_exclusive_active_stream (GstASFDemux * demux,
1252     AsfStream * stream)
1253 {
1254   GSList *l;
1255
1256   for (l = demux->mut_ex_streams; l != NULL; l = l->next) {
1257     guint8 *mes;
1258
1259     /* check for each mutual exclusion group whether it affects this stream */
1260     for (mes = (guint8 *) l->data; mes != NULL && *mes != 0xff; ++mes) {
1261       if (*mes == stream->id) {
1262         /* we are in this group; let's check if we've already activated streams
1263          * that are in the same group (and hence mutually exclusive to this
1264          * one) */
1265         for (mes = (guint8 *) l->data; mes != NULL && *mes != 0xff; ++mes) {
1266           guint i;
1267
1268           for (i = 0; i < demux->num_streams; ++i) {
1269             if (demux->stream[i].id == *mes && demux->stream[i].active) {
1270               GST_LOG_OBJECT (demux, "stream with ID %d is mutually exclusive "
1271                   "to already active stream with ID %d", stream->id,
1272                   demux->stream[i].id);
1273               return TRUE;
1274             }
1275           }
1276         }
1277         /* we can only be in this group once, let's break out and move on to
1278          * the next mutual exclusion group */
1279         break;
1280       }
1281     }
1282   }
1283
1284   return FALSE;
1285 }
1286 #endif
1287
1288 static gboolean
1289 gst_asf_demux_check_activate_streams (GstASFDemux * demux, gboolean force)
1290 {
1291   guint i;
1292
1293   if (demux->activated_streams)
1294     return TRUE;
1295
1296   if (!all_streams_prerolled (demux) && !force) {
1297     GST_DEBUG_OBJECT (demux, "not all streams with data beyond preroll yet");
1298     return FALSE;
1299   }
1300
1301   for (i = 0; i < demux->num_streams; ++i) {
1302     AsfStream *stream = &demux->stream[i];
1303
1304     if (stream->payloads->len > 0) {
1305       /* we don't check mutual exclusion stuff here; either we have data for
1306        * a stream, then we active it, or we don't, then we'll ignore it */
1307       GST_LOG_OBJECT (stream->pad, "is prerolled - activate!");
1308       gst_asf_demux_activate_stream (demux, stream);
1309     } else {
1310       GST_LOG_OBJECT (stream->pad, "no data, ignoring stream");
1311     }
1312   }
1313
1314   gst_asf_demux_release_old_pads (demux);
1315
1316   demux->activated_streams = TRUE;
1317   GST_LOG_OBJECT (demux, "signalling no more pads");
1318   gst_element_no_more_pads (GST_ELEMENT (demux));
1319   return TRUE;
1320 }
1321
1322 /* returns the stream that has a complete payload with the lowest timestamp
1323  * queued, or NULL (we push things by timestamp because during the internal
1324  * prerolling we might accumulate more data then the external queues can take,
1325  * so we'd lock up if we pushed all accumulated data for stream N in one go) */
1326 static AsfStream *
1327 gst_asf_demux_find_stream_with_complete_payload (GstASFDemux * demux)
1328 {
1329   AsfPayload *best_payload = NULL;
1330   AsfStream *best_stream = NULL;
1331   guint i;
1332
1333   for (i = 0; i < demux->num_streams; ++i) {
1334     AsfStream *stream;
1335
1336     stream = &demux->stream[i];
1337
1338     /* Don't push any data until we have at least one payload that falls within
1339      * the current segment. This way we can remove out-of-segment payloads that
1340      * don't need to be decoded after a seek, sending only data from the
1341      * keyframe directly before our segment start */
1342     if (stream->payloads->len > 0) {
1343       AsfPayload *payload;
1344       guint last_idx;
1345
1346       last_idx = stream->payloads->len - 1;
1347       payload = &g_array_index (stream->payloads, AsfPayload, last_idx);
1348       if (G_UNLIKELY (GST_CLOCK_TIME_IS_VALID (payload->ts) &&
1349               (payload->ts < demux->segment.start))) {
1350         if (G_UNLIKELY ((!demux->accurate) && payload->keyframe)) {
1351           GST_DEBUG_OBJECT (stream->pad,
1352               "Found keyframe, updating segment start to %" GST_TIME_FORMAT,
1353               GST_TIME_ARGS (payload->ts));
1354           demux->segment.start = payload->ts;
1355           demux->segment.time = payload->ts;
1356         } else {
1357           GST_DEBUG_OBJECT (stream->pad, "Last queued payload has timestamp %"
1358               GST_TIME_FORMAT " which is before our segment start %"
1359               GST_TIME_FORMAT ", not pushing yet", GST_TIME_ARGS (payload->ts),
1360               GST_TIME_ARGS (demux->segment.start));
1361           continue;
1362         }
1363       }
1364
1365       /* Now see if there's a complete payload queued for this stream */
1366
1367       payload = &g_array_index (stream->payloads, AsfPayload, 0);
1368       if (!gst_asf_payload_is_complete (payload))
1369         continue;
1370
1371       /* ... and whether its timestamp is lower than the current best */
1372       if (best_stream == NULL || best_payload->ts > payload->ts) {
1373         best_stream = stream;
1374         best_payload = payload;
1375       }
1376     }
1377   }
1378
1379   return best_stream;
1380 }
1381
1382 static GstFlowReturn
1383 gst_asf_demux_push_complete_payloads (GstASFDemux * demux, gboolean force)
1384 {
1385   AsfStream *stream;
1386   GstFlowReturn ret = GST_FLOW_OK;
1387
1388   if (G_UNLIKELY (!demux->activated_streams)) {
1389     if (!gst_asf_demux_check_activate_streams (demux, force))
1390       return GST_FLOW_OK;
1391     /* streams are now activated */
1392   }
1393
1394   /* wait until we had a chance to "lock on" some payload's timestamp */
1395   if (G_UNLIKELY (demux->need_newsegment
1396           && !GST_CLOCK_TIME_IS_VALID (demux->segment_ts)))
1397     return GST_FLOW_OK;
1398
1399   while ((stream = gst_asf_demux_find_stream_with_complete_payload (demux))) {
1400     AsfPayload *payload;
1401
1402     payload = &g_array_index (stream->payloads, AsfPayload, 0);
1403
1404     /* do we need to send a newsegment event */
1405     if ((G_UNLIKELY (demux->need_newsegment))) {
1406
1407       /* safe default if insufficient upstream info */
1408       if (!GST_CLOCK_TIME_IS_VALID (demux->in_gap))
1409         demux->in_gap = 0;
1410
1411       if (demux->segment.stop == GST_CLOCK_TIME_NONE &&
1412           demux->segment.duration > 0) {
1413         /* slight HACK; prevent clipping of last bit */
1414         demux->segment.stop = demux->segment.duration + demux->in_gap;
1415       }
1416
1417       /* FIXME : only if ACCURATE ! */
1418       if (G_LIKELY (!demux->accurate
1419               && (GST_CLOCK_TIME_IS_VALID (payload->ts)))) {
1420         GST_DEBUG ("Adjusting newsegment start to %" GST_TIME_FORMAT,
1421             GST_TIME_ARGS (payload->ts));
1422         demux->segment.start = payload->ts;
1423         demux->segment.time = payload->ts;
1424       }
1425
1426       GST_DEBUG_OBJECT (demux, "sending new-segment event %" GST_SEGMENT_FORMAT,
1427           &demux->segment);
1428
1429       /* note: we fix up all timestamps to start from 0, so this should be ok */
1430       gst_asf_demux_send_event_unlocked (demux,
1431           gst_event_new_segment (&demux->segment));
1432
1433       /* now post any global tags we may have found */
1434       if (demux->taglist == NULL)
1435         demux->taglist = gst_tag_list_new_empty ();
1436
1437       gst_tag_list_add (demux->taglist, GST_TAG_MERGE_REPLACE,
1438           GST_TAG_CONTAINER_FORMAT, "ASF", NULL);
1439
1440       GST_DEBUG_OBJECT (demux, "global tags: %" GST_PTR_FORMAT, demux->taglist);
1441       gst_asf_demux_send_event_unlocked (demux,
1442           gst_event_new_tag ("GstDemuxer", demux->taglist));
1443       demux->taglist = NULL;
1444
1445       demux->need_newsegment = FALSE;
1446       demux->segment_running = TRUE;
1447     }
1448
1449     /* Do we have tags pending for this stream? */
1450     if (G_UNLIKELY (stream->pending_tags)) {
1451       GST_LOG_OBJECT (stream->pad, "%" GST_PTR_FORMAT, stream->pending_tags);
1452       gst_pad_push_event (stream->pad,
1453           gst_event_new_tag ("GstDemuxer", stream->pending_tags));
1454       stream->pending_tags = NULL;
1455     }
1456
1457     /* We have the whole packet now so we should push the packet to
1458      * the src pad now. First though we should check if we need to do
1459      * descrambling */
1460     if (G_UNLIKELY (demux->span > 1)) {
1461       gst_asf_demux_descramble_buffer (demux, stream, &payload->buf);
1462     }
1463
1464     payload->buf = gst_buffer_make_writable (payload->buf);
1465
1466     if (G_LIKELY (!payload->keyframe)) {
1467       GST_BUFFER_FLAG_SET (payload->buf, GST_BUFFER_FLAG_DELTA_UNIT);
1468     }
1469
1470     if (G_UNLIKELY (stream->discont)) {
1471       GST_DEBUG_OBJECT (stream->pad, "marking DISCONT on stream");
1472       GST_BUFFER_FLAG_SET (payload->buf, GST_BUFFER_FLAG_DISCONT);
1473       stream->discont = FALSE;
1474     }
1475
1476     if (G_UNLIKELY (stream->is_video && payload->par_x && payload->par_y &&
1477             (payload->par_x != stream->par_x) &&
1478             (payload->par_y != stream->par_y))) {
1479       GST_DEBUG ("Updating PAR (%d/%d => %d/%d)",
1480           stream->par_x, stream->par_y, payload->par_x, payload->par_y);
1481       stream->par_x = payload->par_x;
1482       stream->par_y = payload->par_y;
1483       stream->caps = gst_caps_make_writable (stream->caps);
1484       gst_caps_set_simple (stream->caps, "pixel-aspect-ratio",
1485           GST_TYPE_FRACTION, stream->par_x, stream->par_y, NULL);
1486       gst_pad_set_caps (stream->pad, stream->caps);
1487     }
1488
1489     if (G_UNLIKELY (stream->interlaced != payload->interlaced)) {
1490       GST_DEBUG ("Updating interlaced status (%d => %d)", stream->interlaced,
1491           payload->interlaced);
1492       stream->interlaced = payload->interlaced;
1493       stream->caps = gst_caps_make_writable (stream->caps);
1494       gst_caps_set_simple (stream->caps, "interlace-mode", G_TYPE_BOOLEAN,
1495           (stream->interlaced ? "mixed" : "progressive"), NULL);
1496       gst_pad_set_caps (stream->pad, stream->caps);
1497     }
1498
1499     /* (sort of) interpolate timestamps using upstream "frame of reference",
1500      * typically useful for live src, but might (unavoidably) mess with
1501      * position reporting if a live src is playing not so live content
1502      * (e.g. rtspsrc taking some time to fall back to tcp) */
1503     GST_BUFFER_TIMESTAMP (payload->buf) = payload->ts + demux->in_gap;
1504     if (payload->duration == GST_CLOCK_TIME_NONE
1505         && stream->ext_props.avg_time_per_frame != 0)
1506       GST_BUFFER_DURATION (payload->buf) =
1507           stream->ext_props.avg_time_per_frame * 100;
1508     else
1509       GST_BUFFER_DURATION (payload->buf) = payload->duration;
1510
1511     /* FIXME: we should really set durations on buffers if we can */
1512
1513     GST_LOG_OBJECT (stream->pad, "pushing buffer, ts=%" GST_TIME_FORMAT
1514         ", dur=%" GST_TIME_FORMAT " size=%" G_GSIZE_FORMAT,
1515         GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (payload->buf)),
1516         GST_TIME_ARGS (GST_BUFFER_DURATION (payload->buf)),
1517         gst_buffer_get_size (payload->buf));
1518
1519     ret = gst_pad_push (stream->pad, payload->buf);
1520     ret = gst_asf_demux_aggregate_flow_return (demux, stream, ret);
1521     payload->buf = NULL;
1522     g_array_remove_index (stream->payloads, 0);
1523
1524     /* Break out as soon as we have an issue */
1525     if (G_UNLIKELY (ret != GST_FLOW_OK))
1526       break;
1527   }
1528
1529   return ret;
1530 }
1531
1532 static gboolean
1533 gst_asf_demux_check_buffer_is_header (GstASFDemux * demux, GstBuffer * buf)
1534 {
1535   AsfObject obj;
1536   GstMapInfo map;
1537   g_assert (buf != NULL);
1538
1539   GST_LOG_OBJECT (demux, "Checking if buffer is a header");
1540
1541   gst_buffer_map (buf, &map, GST_MAP_READ);
1542
1543   /* we return false on buffer too small */
1544   if (map.size < ASF_OBJECT_HEADER_SIZE) {
1545     gst_buffer_unmap (buf, &map);
1546     return FALSE;
1547   }
1548
1549   /* check if it is a header */
1550   asf_demux_peek_object (demux, map.data, ASF_OBJECT_HEADER_SIZE, &obj, TRUE);
1551   gst_buffer_unmap (buf, &map);
1552   if (obj.id == ASF_OBJ_HEADER) {
1553     return TRUE;
1554   }
1555   return FALSE;
1556 }
1557
1558 static gboolean
1559 gst_asf_demux_check_chained_asf (GstASFDemux * demux)
1560 {
1561   guint64 off = demux->data_offset + (demux->packet * demux->packet_size);
1562   GstFlowReturn ret = GST_FLOW_OK;
1563   GstBuffer *buf = NULL;
1564   gboolean header = FALSE;
1565
1566   /* TODO maybe we should skip index objects after the data and look
1567    * further for a new header */
1568   if (gst_asf_demux_pull_data (demux, off, ASF_OBJECT_HEADER_SIZE, &buf, &ret)) {
1569     g_assert (buf != NULL);
1570     /* check if it is a header */
1571     if (gst_asf_demux_check_buffer_is_header (demux, buf)) {
1572       GST_DEBUG_OBJECT (demux, "new base offset: %" G_GUINT64_FORMAT, off);
1573       demux->base_offset = off;
1574       header = TRUE;
1575     }
1576
1577     gst_buffer_unref (buf);
1578   }
1579
1580   return header;
1581 }
1582
1583 static void
1584 gst_asf_demux_loop (GstASFDemux * demux)
1585 {
1586   GstFlowReturn flow = GST_FLOW_OK;
1587   GstBuffer *buf = NULL;
1588   guint64 off;
1589   gboolean sent_eos = FALSE;
1590
1591   if (G_UNLIKELY (demux->state == GST_ASF_DEMUX_STATE_HEADER)) {
1592     if (!gst_asf_demux_pull_headers (demux)) {
1593       flow = GST_FLOW_ERROR;
1594       goto pause;
1595     }
1596
1597     gst_asf_demux_pull_indices (demux);
1598   }
1599
1600   g_assert (demux->state == GST_ASF_DEMUX_STATE_DATA);
1601
1602   if (G_UNLIKELY (demux->num_packets != 0
1603           && demux->packet >= demux->num_packets))
1604     goto eos;
1605
1606   GST_LOG_OBJECT (demux, "packet %u/%u", (guint) demux->packet + 1,
1607       (guint) demux->num_packets);
1608
1609   off = demux->data_offset + (demux->packet * demux->packet_size);
1610
1611   if (G_UNLIKELY (!gst_asf_demux_pull_data (demux, off,
1612               demux->packet_size * demux->speed_packets, &buf, &flow))) {
1613     GST_DEBUG_OBJECT (demux, "got flow %s", gst_flow_get_name (flow));
1614     if (flow == GST_FLOW_EOS)
1615       goto eos;
1616     else if (flow == GST_FLOW_FLUSHING) {
1617       GST_DEBUG_OBJECT (demux, "Not fatal");
1618       goto pause;
1619     } else
1620       goto read_failed;
1621   }
1622
1623   if (G_LIKELY (demux->speed_packets == 1)) {
1624     GstAsfDemuxParsePacketError err;
1625     err = gst_asf_demux_parse_packet (demux, buf);
1626     if (G_UNLIKELY (err != GST_ASF_DEMUX_PARSE_PACKET_ERROR_NONE)) {
1627       /* when we don't know when the data object ends, we should check
1628        * for a chained asf */
1629       if (demux->num_packets == 0) {
1630         if (gst_asf_demux_check_buffer_is_header (demux, buf)) {
1631           GST_INFO_OBJECT (demux, "Chained asf found");
1632           demux->base_offset = off;
1633           gst_asf_demux_reset (demux, TRUE);
1634           gst_buffer_unref (buf);
1635           return;
1636         }
1637       }
1638       /* FIXME: We should tally up fatal errors and error out only
1639        * after a few broken packets in a row? */
1640
1641       GST_INFO_OBJECT (demux, "Ignoring recoverable parse error");
1642       gst_buffer_unref (buf);
1643       ++demux->packet;
1644       return;
1645     }
1646
1647     flow = gst_asf_demux_push_complete_payloads (demux, FALSE);
1648
1649     ++demux->packet;
1650
1651   } else {
1652     guint n;
1653     for (n = 0; n < demux->speed_packets; n++) {
1654       GstBuffer *sub;
1655       GstAsfDemuxParsePacketError err;
1656
1657       sub =
1658           gst_buffer_copy_region (buf, GST_BUFFER_COPY_ALL,
1659           n * demux->packet_size, demux->packet_size);
1660       err = gst_asf_demux_parse_packet (demux, sub);
1661       if (G_UNLIKELY (err != GST_ASF_DEMUX_PARSE_PACKET_ERROR_NONE)) {
1662         /* when we don't know when the data object ends, we should check
1663          * for a chained asf */
1664         if (demux->num_packets == 0) {
1665           if (gst_asf_demux_check_buffer_is_header (demux, sub)) {
1666             GST_INFO_OBJECT (demux, "Chained asf found");
1667             demux->base_offset = off + n * demux->packet_size;
1668             gst_asf_demux_reset (demux, TRUE);
1669             gst_buffer_unref (sub);
1670             gst_buffer_unref (buf);
1671             return;
1672           }
1673         }
1674         /* FIXME: We should tally up fatal errors and error out only
1675          * after a few broken packets in a row? */
1676
1677         GST_INFO_OBJECT (demux, "Ignoring recoverable parse error");
1678         flow = GST_FLOW_OK;
1679       }
1680
1681       gst_buffer_unref (sub);
1682
1683       if (err == GST_ASF_DEMUX_PARSE_PACKET_ERROR_NONE)
1684         flow = gst_asf_demux_push_complete_payloads (demux, FALSE);
1685
1686       ++demux->packet;
1687
1688     }
1689
1690     /* reset speed pull */
1691     demux->speed_packets = 1;
1692   }
1693
1694   gst_buffer_unref (buf);
1695
1696   if (G_UNLIKELY (demux->num_packets > 0
1697           && demux->packet >= demux->num_packets)) {
1698     GST_LOG_OBJECT (demux, "reached EOS");
1699     goto eos;
1700   }
1701
1702   if (G_UNLIKELY (flow != GST_FLOW_OK)) {
1703     GST_DEBUG_OBJECT (demux, "pushing complete payloads failed");
1704     goto pause;
1705   }
1706
1707   /* check if we're at the end of the configured segment */
1708   /* FIXME: check if segment end reached etc. */
1709
1710   return;
1711
1712 eos:
1713   {
1714     /* if we haven't activated our streams yet, this might be because we have
1715      * less data queued than required for preroll; force stream activation and
1716      * send any pending payloads before sending EOS */
1717     if (!demux->activated_streams)
1718       gst_asf_demux_push_complete_payloads (demux, TRUE);
1719
1720     /* we want to push an eos or post a segment-done in any case */
1721     if (demux->segment.flags & GST_SEEK_FLAG_SEGMENT) {
1722       gint64 stop;
1723
1724       /* for segment playback we need to post when (in stream time)
1725        * we stopped, this is either stop (when set) or the duration. */
1726       if ((stop = demux->segment.stop) == -1)
1727         stop = demux->segment.duration;
1728
1729       GST_INFO_OBJECT (demux, "Posting segment-done, at end of segment");
1730       gst_element_post_message (GST_ELEMENT_CAST (demux),
1731           gst_message_new_segment_done (GST_OBJECT (demux), GST_FORMAT_TIME,
1732               stop));
1733     } else if (flow != GST_FLOW_EOS) {
1734       /* check if we have a chained asf, in case, we don't eos yet */
1735       if (gst_asf_demux_check_chained_asf (demux)) {
1736         GST_INFO_OBJECT (demux, "Chained ASF starting");
1737         gst_asf_demux_reset (demux, TRUE);
1738         return;
1739       }
1740     }
1741     /* normal playback, send EOS to all linked pads */
1742     GST_INFO_OBJECT (demux, "Sending EOS, at end of stream");
1743     gst_asf_demux_send_event_unlocked (demux, gst_event_new_eos ());
1744     sent_eos = TRUE;
1745     /* ... and fall through to pause */
1746   }
1747 pause:
1748   {
1749     GST_DEBUG_OBJECT (demux, "pausing task, flow return: %s",
1750         gst_flow_get_name (flow));
1751     demux->segment_running = FALSE;
1752     gst_pad_pause_task (demux->sinkpad);
1753
1754     /* For the error cases (not EOS) */
1755     if (!sent_eos) {
1756       if (flow == GST_FLOW_EOS)
1757         gst_asf_demux_send_event_unlocked (demux, gst_event_new_eos ());
1758       else if (flow < GST_FLOW_EOS || flow == GST_FLOW_NOT_LINKED) {
1759         /* Post an error. Hopefully something else already has, but if not... */
1760         GST_ELEMENT_ERROR (demux, STREAM, FAILED,
1761             (_("Internal data stream error.")),
1762             ("streaming stopped, reason %s", gst_flow_get_name (flow)));
1763       }
1764     }
1765     return;
1766   }
1767
1768 /* ERRORS */
1769 read_failed:
1770   {
1771     GST_DEBUG_OBJECT (demux, "Read failed, doh");
1772     gst_asf_demux_send_event_unlocked (demux, gst_event_new_eos ());
1773     flow = GST_FLOW_EOS;
1774     goto pause;
1775   }
1776 #if 0
1777   /* See FIXMEs above */
1778 parse_error:
1779   {
1780     gst_buffer_unref (buf);
1781     GST_ELEMENT_ERROR (demux, STREAM, DEMUX, (NULL),
1782         ("Error parsing ASF packet %u", (guint) demux->packet));
1783     gst_asf_demux_send_event_unlocked (demux, gst_event_new_eos ());
1784     flow = GST_FLOW_ERROR;
1785     goto pause;
1786   }
1787 #endif
1788 }
1789
1790 #define GST_ASF_DEMUX_CHECK_HEADER_YES       0
1791 #define GST_ASF_DEMUX_CHECK_HEADER_NO        1
1792 #define GST_ASF_DEMUX_CHECK_HEADER_NEED_DATA 2
1793
1794 static gint
1795 gst_asf_demux_check_header (GstASFDemux * demux)
1796 {
1797   AsfObject obj;
1798   guint8 *cdata = (guint8 *) gst_adapter_map (demux->adapter,
1799       ASF_OBJECT_HEADER_SIZE);
1800   if (cdata == NULL)            /* need more data */
1801     return GST_ASF_DEMUX_CHECK_HEADER_NEED_DATA;
1802
1803   asf_demux_peek_object (demux, cdata, ASF_OBJECT_HEADER_SIZE, &obj, FALSE);
1804   if (obj.id != ASF_OBJ_HEADER) {
1805     return GST_ASF_DEMUX_CHECK_HEADER_NO;
1806   } else {
1807     return GST_ASF_DEMUX_CHECK_HEADER_YES;
1808   }
1809 }
1810
1811 static GstFlowReturn
1812 gst_asf_demux_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
1813 {
1814   GstFlowReturn ret = GST_FLOW_OK;
1815   GstASFDemux *demux;
1816
1817   demux = GST_ASF_DEMUX (parent);
1818
1819   GST_LOG_OBJECT (demux,
1820       "buffer: size=%" G_GSIZE_FORMAT ", offset=%" G_GINT64_FORMAT ", time=%"
1821       GST_TIME_FORMAT, gst_buffer_get_size (buf), GST_BUFFER_OFFSET (buf),
1822       GST_TIME_ARGS (GST_BUFFER_TIMESTAMP (buf)));
1823
1824   if (G_UNLIKELY (GST_BUFFER_IS_DISCONT (buf))) {
1825     GST_DEBUG_OBJECT (demux, "received DISCONT");
1826     gst_asf_demux_mark_discont (demux);
1827   }
1828
1829   if (G_UNLIKELY ((!GST_CLOCK_TIME_IS_VALID (demux->in_gap) &&
1830               GST_BUFFER_TIMESTAMP_IS_VALID (buf)))) {
1831     demux->in_gap = GST_BUFFER_TIMESTAMP (buf) - demux->in_segment.start;
1832     GST_DEBUG_OBJECT (demux, "upstream segment start %" GST_TIME_FORMAT
1833         ", interpolation gap: %" GST_TIME_FORMAT,
1834         GST_TIME_ARGS (demux->in_segment.start), GST_TIME_ARGS (demux->in_gap));
1835   }
1836
1837   gst_adapter_push (demux->adapter, buf);
1838
1839   switch (demux->state) {
1840     case GST_ASF_DEMUX_STATE_INDEX:{
1841       gint result = gst_asf_demux_check_header (demux);
1842       if (result == GST_ASF_DEMUX_CHECK_HEADER_NEED_DATA)       /* need more data */
1843         break;
1844
1845       if (result == GST_ASF_DEMUX_CHECK_HEADER_NO) {
1846         /* we don't care about this, probably an index */
1847         /* TODO maybe would be smarter to skip all the indices
1848          * until we got a new header or EOS to decide */
1849         GST_LOG_OBJECT (demux, "Received index object, its EOS");
1850         goto eos;
1851       } else {
1852         GST_INFO_OBJECT (demux, "Chained asf starting");
1853         /* cleanup and get ready for a chained asf */
1854         gst_asf_demux_reset (demux, TRUE);
1855         /* fall through */
1856       }
1857     }
1858     case GST_ASF_DEMUX_STATE_HEADER:{
1859       ret = gst_asf_demux_chain_headers (demux);
1860       if (demux->state != GST_ASF_DEMUX_STATE_DATA)
1861         break;
1862       /* otherwise fall through */
1863     }
1864     case GST_ASF_DEMUX_STATE_DATA:
1865     {
1866       guint64 data_size;
1867
1868       data_size = demux->packet_size;
1869
1870       while (gst_adapter_available (demux->adapter) >= data_size) {
1871         GstBuffer *buf;
1872         GstAsfDemuxParsePacketError err;
1873
1874         /* we don't know the length of the stream
1875          * check for a chained asf everytime */
1876         if (demux->num_packets == 0) {
1877           gint result = gst_asf_demux_check_header (demux);
1878
1879           if (result == GST_ASF_DEMUX_CHECK_HEADER_YES) {
1880             GST_INFO_OBJECT (demux, "Chained asf starting");
1881             /* cleanup and get ready for a chained asf */
1882             gst_asf_demux_reset (demux, TRUE);
1883             break;
1884           }
1885         } else if (G_UNLIKELY (demux->num_packets != 0 && demux->packet >= 0
1886                 && demux->packet >= demux->num_packets)) {
1887           /* do not overshoot data section when streaming */
1888           break;
1889         }
1890
1891         buf = gst_adapter_take_buffer (demux->adapter, data_size);
1892
1893         /* FIXME: We should tally up fatal errors and error out only
1894          * after a few broken packets in a row? */
1895         err = gst_asf_demux_parse_packet (demux, buf);
1896
1897         gst_buffer_unref (buf);
1898
1899         if (G_LIKELY (err == GST_ASF_DEMUX_PARSE_PACKET_ERROR_NONE))
1900           ret = gst_asf_demux_push_complete_payloads (demux, FALSE);
1901         else
1902           GST_WARNING_OBJECT (demux, "Parse error");
1903
1904         if (demux->packet >= 0)
1905           ++demux->packet;
1906       }
1907       if (G_UNLIKELY (demux->num_packets != 0 && demux->packet >= 0
1908               && demux->packet >= demux->num_packets)) {
1909         demux->state = GST_ASF_DEMUX_STATE_INDEX;
1910       }
1911       break;
1912     }
1913     default:
1914       g_assert_not_reached ();
1915   }
1916
1917 done:
1918   if (ret != GST_FLOW_OK)
1919     GST_DEBUG_OBJECT (demux, "flow: %s", gst_flow_get_name (ret));
1920
1921   return ret;
1922
1923 eos:
1924   {
1925     GST_DEBUG_OBJECT (demux, "Handled last packet, setting EOS");
1926     ret = GST_FLOW_EOS;
1927     goto done;
1928   }
1929 }
1930
1931 static inline gboolean
1932 gst_asf_demux_skip_bytes (guint num_bytes, guint8 ** p_data, guint64 * p_size)
1933 {
1934   if (*p_size < num_bytes)
1935     return FALSE;
1936
1937   *p_data += num_bytes;
1938   *p_size -= num_bytes;
1939   return TRUE;
1940 }
1941
1942 static inline guint8
1943 gst_asf_demux_get_uint8 (guint8 ** p_data, guint64 * p_size)
1944 {
1945   guint8 ret;
1946
1947   g_assert (*p_size >= 1);
1948   ret = GST_READ_UINT8 (*p_data);
1949   *p_data += sizeof (guint8);
1950   *p_size -= sizeof (guint8);
1951   return ret;
1952 }
1953
1954 static inline guint16
1955 gst_asf_demux_get_uint16 (guint8 ** p_data, guint64 * p_size)
1956 {
1957   guint16 ret;
1958
1959   g_assert (*p_size >= 2);
1960   ret = GST_READ_UINT16_LE (*p_data);
1961   *p_data += sizeof (guint16);
1962   *p_size -= sizeof (guint16);
1963   return ret;
1964 }
1965
1966 static inline guint32
1967 gst_asf_demux_get_uint32 (guint8 ** p_data, guint64 * p_size)
1968 {
1969   guint32 ret;
1970
1971   g_assert (*p_size >= 4);
1972   ret = GST_READ_UINT32_LE (*p_data);
1973   *p_data += sizeof (guint32);
1974   *p_size -= sizeof (guint32);
1975   return ret;
1976 }
1977
1978 static inline guint64
1979 gst_asf_demux_get_uint64 (guint8 ** p_data, guint64 * p_size)
1980 {
1981   guint64 ret;
1982
1983   g_assert (*p_size >= 8);
1984   ret = GST_READ_UINT64_LE (*p_data);
1985   *p_data += sizeof (guint64);
1986   *p_size -= sizeof (guint64);
1987   return ret;
1988 }
1989
1990 static inline guint32
1991 gst_asf_demux_get_var_length (guint8 type, guint8 ** p_data, guint64 * p_size)
1992 {
1993   switch (type) {
1994     case 0:
1995       return 0;
1996
1997     case 1:
1998       g_assert (*p_size >= 1);
1999       return gst_asf_demux_get_uint8 (p_data, p_size);
2000
2001     case 2:
2002       g_assert (*p_size >= 2);
2003       return gst_asf_demux_get_uint16 (p_data, p_size);
2004
2005     case 3:
2006       g_assert (*p_size >= 4);
2007       return gst_asf_demux_get_uint32 (p_data, p_size);
2008
2009     default:
2010       g_assert_not_reached ();
2011       break;
2012   }
2013   return 0;
2014 }
2015
2016 static gboolean
2017 gst_asf_demux_get_buffer (GstBuffer ** p_buf, guint num_bytes_to_read,
2018     guint8 ** p_data, guint64 * p_size)
2019 {
2020   *p_buf = NULL;
2021
2022   if (*p_size < num_bytes_to_read)
2023     return FALSE;
2024
2025   *p_buf = gst_buffer_new_and_alloc (num_bytes_to_read);
2026   gst_buffer_fill (*p_buf, 0, *p_data, num_bytes_to_read);
2027
2028   *p_data += num_bytes_to_read;
2029   *p_size -= num_bytes_to_read;
2030
2031   return TRUE;
2032 }
2033
2034 static gboolean
2035 gst_asf_demux_get_bytes (guint8 ** p_buf, guint num_bytes_to_read,
2036     guint8 ** p_data, guint64 * p_size)
2037 {
2038   *p_buf = NULL;
2039
2040   if (*p_size < num_bytes_to_read)
2041     return FALSE;
2042
2043   *p_buf = g_memdup (*p_data, num_bytes_to_read);
2044   *p_data += num_bytes_to_read;
2045   *p_size -= num_bytes_to_read;
2046   return TRUE;
2047 }
2048
2049 static gboolean
2050 gst_asf_demux_get_string (gchar ** p_str, guint16 * p_strlen,
2051     guint8 ** p_data, guint64 * p_size)
2052 {
2053   guint16 s_length;
2054   guint8 *s;
2055
2056   *p_str = NULL;
2057
2058   if (*p_size < 2)
2059     return FALSE;
2060
2061   s_length = gst_asf_demux_get_uint16 (p_data, p_size);
2062
2063   if (p_strlen)
2064     *p_strlen = s_length;
2065
2066   if (s_length == 0) {
2067     GST_WARNING ("zero-length string");
2068     *p_str = g_strdup ("");
2069     return TRUE;
2070   }
2071
2072   if (!gst_asf_demux_get_bytes (&s, s_length, p_data, p_size))
2073     return FALSE;
2074
2075   g_assert (s != NULL);
2076
2077   /* just because They don't exist doesn't
2078    * mean They are not out to get you ... */
2079   if (s[s_length - 1] != '\0') {
2080     s = g_realloc (s, s_length + 1);
2081     s[s_length] = '\0';
2082   }
2083
2084   *p_str = (gchar *) s;
2085   return TRUE;
2086 }
2087
2088
2089 static void
2090 gst_asf_demux_get_guid (ASFGuid * guid, guint8 ** p_data, guint64 * p_size)
2091 {
2092   g_assert (*p_size >= 4 * sizeof (guint32));
2093
2094   guid->v1 = gst_asf_demux_get_uint32 (p_data, p_size);
2095   guid->v2 = gst_asf_demux_get_uint32 (p_data, p_size);
2096   guid->v3 = gst_asf_demux_get_uint32 (p_data, p_size);
2097   guid->v4 = gst_asf_demux_get_uint32 (p_data, p_size);
2098 }
2099
2100 static gboolean
2101 gst_asf_demux_get_stream_audio (asf_stream_audio * audio, guint8 ** p_data,
2102     guint64 * p_size)
2103 {
2104   if (*p_size < (2 + 2 + 4 + 4 + 2 + 2 + 2))
2105     return FALSE;
2106
2107   /* WAVEFORMATEX Structure */
2108   audio->codec_tag = gst_asf_demux_get_uint16 (p_data, p_size);
2109   audio->channels = gst_asf_demux_get_uint16 (p_data, p_size);
2110   audio->sample_rate = gst_asf_demux_get_uint32 (p_data, p_size);
2111   audio->byte_rate = gst_asf_demux_get_uint32 (p_data, p_size);
2112   audio->block_align = gst_asf_demux_get_uint16 (p_data, p_size);
2113   audio->word_size = gst_asf_demux_get_uint16 (p_data, p_size);
2114   /* Codec specific data size */
2115   audio->size = gst_asf_demux_get_uint16 (p_data, p_size);
2116   return TRUE;
2117 }
2118
2119 static gboolean
2120 gst_asf_demux_get_stream_video (asf_stream_video * video, guint8 ** p_data,
2121     guint64 * p_size)
2122 {
2123   if (*p_size < (4 + 4 + 1 + 2))
2124     return FALSE;
2125
2126   video->width = gst_asf_demux_get_uint32 (p_data, p_size);
2127   video->height = gst_asf_demux_get_uint32 (p_data, p_size);
2128   video->unknown = gst_asf_demux_get_uint8 (p_data, p_size);
2129   video->size = gst_asf_demux_get_uint16 (p_data, p_size);
2130   return TRUE;
2131 }
2132
2133 static gboolean
2134 gst_asf_demux_get_stream_video_format (asf_stream_video_format * fmt,
2135     guint8 ** p_data, guint64 * p_size)
2136 {
2137   if (*p_size < (4 + 4 + 4 + 2 + 2 + 4 + 4 + 4 + 4 + 4 + 4))
2138     return FALSE;
2139
2140   fmt->size = gst_asf_demux_get_uint32 (p_data, p_size);
2141   fmt->width = gst_asf_demux_get_uint32 (p_data, p_size);
2142   fmt->height = gst_asf_demux_get_uint32 (p_data, p_size);
2143   fmt->planes = gst_asf_demux_get_uint16 (p_data, p_size);
2144   fmt->depth = gst_asf_demux_get_uint16 (p_data, p_size);
2145   fmt->tag = gst_asf_demux_get_uint32 (p_data, p_size);
2146   fmt->image_size = gst_asf_demux_get_uint32 (p_data, p_size);
2147   fmt->xpels_meter = gst_asf_demux_get_uint32 (p_data, p_size);
2148   fmt->ypels_meter = gst_asf_demux_get_uint32 (p_data, p_size);
2149   fmt->num_colors = gst_asf_demux_get_uint32 (p_data, p_size);
2150   fmt->imp_colors = gst_asf_demux_get_uint32 (p_data, p_size);
2151   return TRUE;
2152 }
2153
2154 AsfStream *
2155 gst_asf_demux_get_stream (GstASFDemux * demux, guint16 id)
2156 {
2157   guint i;
2158
2159   for (i = 0; i < demux->num_streams; i++) {
2160     if (demux->stream[i].id == id)
2161       return &demux->stream[i];
2162   }
2163
2164   GST_WARNING ("Segment found for undefined stream: (%d)", id);
2165   return NULL;
2166 }
2167
2168 static void
2169 gst_asf_demux_setup_pad (GstASFDemux * demux, GstPad * src_pad,
2170     GstCaps * caps, guint16 id, gboolean is_video, GstTagList * tags)
2171 {
2172   AsfStream *stream;
2173
2174   gst_pad_use_fixed_caps (src_pad);
2175   gst_pad_set_caps (src_pad, caps);
2176
2177   gst_pad_set_event_function (src_pad,
2178       GST_DEBUG_FUNCPTR (gst_asf_demux_handle_src_event));
2179   gst_pad_set_query_function (src_pad,
2180       GST_DEBUG_FUNCPTR (gst_asf_demux_handle_src_query));
2181
2182   stream = &demux->stream[demux->num_streams];
2183   stream->caps = caps;
2184   stream->pad = src_pad;
2185   stream->id = id;
2186   stream->fps_known = !is_video;        /* bit hacky for audio */
2187   stream->is_video = is_video;
2188   stream->pending_tags = tags;
2189   stream->discont = TRUE;
2190   if (is_video) {
2191     GstStructure *st;
2192     gint par_x, par_y;
2193     st = gst_caps_get_structure (caps, 0);
2194     if (gst_structure_get_fraction (st, "pixel-aspect-ratio", &par_x, &par_y) &&
2195         par_x > 0 && par_y > 0) {
2196       GST_DEBUG ("PAR %d/%d", par_x, par_y);
2197       stream->par_x = par_x;
2198       stream->par_y = par_y;
2199     }
2200   }
2201
2202   stream->payloads = g_array_new (FALSE, FALSE, sizeof (AsfPayload));
2203
2204   GST_INFO ("Created pad %s for stream %u with caps %" GST_PTR_FORMAT,
2205       GST_PAD_NAME (src_pad), demux->num_streams, caps);
2206
2207   ++demux->num_streams;
2208
2209   stream->active = FALSE;
2210 }
2211
2212 static void
2213 gst_asf_demux_add_audio_stream (GstASFDemux * demux,
2214     asf_stream_audio * audio, guint16 id, guint8 ** p_data, guint64 * p_size)
2215 {
2216   GstTagList *tags = NULL;
2217   GstBuffer *extradata = NULL;
2218   GstPad *src_pad;
2219   GstCaps *caps;
2220   guint16 size_left = 0;
2221   gchar *codec_name = NULL;
2222   gchar *name = NULL;
2223
2224   size_left = audio->size;
2225
2226   /* Create the audio pad */
2227   name = g_strdup_printf ("audio_%u", demux->num_audio_streams);
2228
2229   src_pad = gst_pad_new_from_static_template (&audio_src_template, name);
2230   g_free (name);
2231
2232   /* Swallow up any left over data and set up the 
2233    * standard properties from the header info */
2234   if (size_left) {
2235     GST_INFO_OBJECT (demux, "Audio header contains %d bytes of "
2236         "codec specific data", size_left);
2237
2238     g_assert (size_left <= *p_size);
2239     gst_asf_demux_get_buffer (&extradata, size_left, p_data, p_size);
2240   }
2241
2242   /* asf_stream_audio is the same as gst_riff_strf_auds, but with an
2243    * additional two bytes indicating extradata. */
2244   /* FIXME: Handle the channel reorder map here */
2245   caps = gst_riff_create_audio_caps (audio->codec_tag, NULL,
2246       (gst_riff_strf_auds *) audio, extradata, NULL, &codec_name, NULL);
2247
2248   if (caps == NULL) {
2249     caps = gst_caps_new_simple ("audio/x-asf-unknown", "codec_id",
2250         G_TYPE_INT, (gint) audio->codec_tag, NULL);
2251   }
2252
2253   /* Informing about that audio format we just added */
2254   if (codec_name) {
2255     tags = gst_tag_list_new (GST_TAG_AUDIO_CODEC, codec_name, NULL);
2256     g_free (codec_name);
2257   }
2258
2259   if (extradata)
2260     gst_buffer_unref (extradata);
2261
2262   GST_INFO ("Adding audio stream #%u, id %u codec %u (0x%04x), tags=%"
2263       GST_PTR_FORMAT, demux->num_audio_streams, id, audio->codec_tag,
2264       audio->codec_tag, tags);
2265
2266   ++demux->num_audio_streams;
2267
2268   gst_asf_demux_setup_pad (demux, src_pad, caps, id, FALSE, tags);
2269 }
2270
2271 static void
2272 gst_asf_demux_add_video_stream (GstASFDemux * demux,
2273     asf_stream_video_format * video, guint16 id,
2274     guint8 ** p_data, guint64 * p_size)
2275 {
2276   GstTagList *tags = NULL;
2277   GstBuffer *extradata = NULL;
2278   GstPad *src_pad;
2279   GstCaps *caps;
2280   gchar *str;
2281   gchar *name = NULL;
2282   gchar *codec_name = NULL;
2283   gint size_left = video->size - 40;
2284
2285   /* Create the video pad */
2286   name = g_strdup_printf ("video_%u", demux->num_video_streams);
2287   src_pad = gst_pad_new_from_static_template (&video_src_template, name);
2288   g_free (name);
2289
2290   /* Now try some gstreamer formatted MIME types (from gst_avi_demux_strf_vids) */
2291   if (size_left) {
2292     GST_LOG ("Video header has %d bytes of codec specific data", size_left);
2293     g_assert (size_left <= *p_size);
2294     gst_asf_demux_get_buffer (&extradata, size_left, p_data, p_size);
2295   }
2296
2297   GST_DEBUG ("video codec %" GST_FOURCC_FORMAT, GST_FOURCC_ARGS (video->tag));
2298
2299   /* yes, asf_stream_video_format and gst_riff_strf_vids are the same */
2300   caps = gst_riff_create_video_caps (video->tag, NULL,
2301       (gst_riff_strf_vids *) video, extradata, NULL, &codec_name);
2302
2303   if (caps == NULL) {
2304     caps = gst_caps_new_simple ("video/x-asf-unknown", "fourcc",
2305         G_TYPE_UINT, video->tag, NULL);
2306   } else {
2307     GstStructure *s;
2308     gint ax, ay;
2309
2310     s = gst_asf_demux_get_metadata_for_stream (demux, id);
2311     if (gst_structure_get_int (s, "AspectRatioX", &ax) &&
2312         gst_structure_get_int (s, "AspectRatioY", &ay) && (ax > 0 && ay > 0)) {
2313       gst_caps_set_simple (caps, "pixel-aspect-ratio", GST_TYPE_FRACTION,
2314           ax, ay, NULL);
2315
2316     } else {
2317       guint ax, ay;
2318       /* retry with the global metadata */
2319       GST_DEBUG ("Retrying with global metadata %" GST_PTR_FORMAT,
2320           demux->global_metadata);
2321       s = demux->global_metadata;
2322       if (gst_structure_get_uint (s, "AspectRatioX", &ax) &&
2323           gst_structure_get_uint (s, "AspectRatioY", &ay)) {
2324         GST_DEBUG ("ax:%d, ay:%d", ax, ay);
2325         if (ax > 0 && ay > 0)
2326           gst_caps_set_simple (caps, "pixel-aspect-ratio", GST_TYPE_FRACTION,
2327               ax, ay, NULL);
2328       }
2329     }
2330     s = gst_caps_get_structure (caps, 0);
2331     gst_structure_remove_field (s, "framerate");
2332   }
2333
2334   /* add fourcc format to caps, some proprietary decoders seem to need it */
2335   str = g_strdup_printf ("%" GST_FOURCC_FORMAT, GST_FOURCC_ARGS (video->tag));
2336   gst_caps_set_simple (caps, "format", G_TYPE_STRING, str, NULL);
2337   g_free (str);
2338
2339   if (codec_name) {
2340     tags = gst_tag_list_new (GST_TAG_VIDEO_CODEC, codec_name, NULL);
2341     g_free (codec_name);
2342   }
2343
2344   if (extradata)
2345     gst_buffer_unref (extradata);
2346
2347   GST_INFO ("Adding video stream #%u, id %u, codec %"
2348       GST_FOURCC_FORMAT " (0x%08x)", demux->num_video_streams, id,
2349       GST_FOURCC_ARGS (video->tag), video->tag);
2350
2351   ++demux->num_video_streams;
2352
2353   gst_asf_demux_setup_pad (demux, src_pad, caps, id, TRUE, tags);
2354 }
2355
2356 static void
2357 gst_asf_demux_activate_stream (GstASFDemux * demux, AsfStream * stream)
2358 {
2359   if (!stream->active) {
2360     GST_INFO_OBJECT (demux, "Activating stream %2u, pad %s, caps %"
2361         GST_PTR_FORMAT, stream->id, GST_PAD_NAME (stream->pad), stream->caps);
2362     gst_pad_set_active (stream->pad, TRUE);
2363     gst_pad_set_caps (stream->pad, stream->caps);
2364     gst_element_add_pad (GST_ELEMENT_CAST (demux), stream->pad);
2365     stream->active = TRUE;
2366   }
2367 }
2368
2369 static AsfStream *
2370 gst_asf_demux_parse_stream_object (GstASFDemux * demux, guint8 * data,
2371     guint64 size)
2372 {
2373   AsfCorrectionType correction_type;
2374   AsfStreamType stream_type;
2375   GstClockTime time_offset;
2376   gboolean is_encrypted G_GNUC_UNUSED;
2377   guint16 stream_id;
2378   guint16 flags;
2379   ASFGuid guid;
2380   guint stream_specific_size;
2381   guint type_specific_size G_GNUC_UNUSED;
2382   guint unknown G_GNUC_UNUSED;
2383
2384   /* Get the rest of the header's header */
2385   if (size < (16 + 16 + 8 + 4 + 4 + 2 + 4))
2386     goto not_enough_data;
2387
2388   gst_asf_demux_get_guid (&guid, &data, &size);
2389   stream_type = gst_asf_demux_identify_guid (asf_stream_guids, &guid);
2390
2391   gst_asf_demux_get_guid (&guid, &data, &size);
2392   correction_type = gst_asf_demux_identify_guid (asf_correction_guids, &guid);
2393
2394   time_offset = gst_asf_demux_get_uint64 (&data, &size) * 100;
2395
2396   type_specific_size = gst_asf_demux_get_uint32 (&data, &size);
2397   stream_specific_size = gst_asf_demux_get_uint32 (&data, &size);
2398
2399   flags = gst_asf_demux_get_uint16 (&data, &size);
2400   stream_id = flags & 0x7f;
2401   is_encrypted = ! !((flags & 0x8000) << 15);
2402   unknown = gst_asf_demux_get_uint32 (&data, &size);
2403
2404   GST_DEBUG_OBJECT (demux, "Found stream %u, time_offset=%" GST_TIME_FORMAT,
2405       stream_id, GST_TIME_ARGS (time_offset));
2406
2407   switch (stream_type) {
2408     case ASF_STREAM_AUDIO:{
2409       asf_stream_audio audio_object;
2410
2411       if (!gst_asf_demux_get_stream_audio (&audio_object, &data, &size))
2412         goto not_enough_data;
2413
2414       GST_INFO ("Object is an audio stream with %u bytes of additional data",
2415           audio_object.size);
2416
2417       gst_asf_demux_add_audio_stream (demux, &audio_object, stream_id,
2418           &data, &size);
2419
2420       switch (correction_type) {
2421         case ASF_CORRECTION_ON:{
2422           guint span, packet_size, chunk_size, data_size, silence_data;
2423
2424           GST_INFO ("Using error correction");
2425
2426           if (size < (1 + 2 + 2 + 2 + 1))
2427             goto not_enough_data;
2428
2429           span = gst_asf_demux_get_uint8 (&data, &size);
2430           packet_size = gst_asf_demux_get_uint16 (&data, &size);
2431           chunk_size = gst_asf_demux_get_uint16 (&data, &size);
2432           data_size = gst_asf_demux_get_uint16 (&data, &size);
2433           silence_data = gst_asf_demux_get_uint8 (&data, &size);
2434
2435           /* FIXME: shouldn't this be per-stream? */
2436           demux->span = span;
2437
2438           GST_DEBUG_OBJECT (demux, "Descrambling ps:%u cs:%u ds:%u s:%u sd:%u",
2439               packet_size, chunk_size, data_size, span, silence_data);
2440
2441           if (demux->span > 1) {
2442             if (chunk_size == 0 || ((packet_size / chunk_size) <= 1)) {
2443               /* Disable descrambling */
2444               demux->span = 0;
2445             } else {
2446               /* FIXME: this else branch was added for
2447                * weird_al_yankovic - the saga begins.asf */
2448               demux->ds_packet_size = packet_size;
2449               demux->ds_chunk_size = chunk_size;
2450             }
2451           } else {
2452             /* Descambling is enabled */
2453             demux->ds_packet_size = packet_size;
2454             demux->ds_chunk_size = chunk_size;
2455           }
2456 #if 0
2457           /* Now skip the rest of the silence data */
2458           if (data_size > 1)
2459             gst_bytestream_flush (demux->bs, data_size - 1);
2460 #else
2461           /* FIXME: CHECKME. And why -1? */
2462           if (data_size > 1) {
2463             if (!gst_asf_demux_skip_bytes (data_size - 1, &data, &size)) {
2464               goto not_enough_data;
2465             }
2466           }
2467 #endif
2468           break;
2469         }
2470         case ASF_CORRECTION_OFF:{
2471           GST_INFO ("Error correction off");
2472           if (!gst_asf_demux_skip_bytes (stream_specific_size, &data, &size))
2473             goto not_enough_data;
2474           break;
2475         }
2476         default:
2477           GST_ELEMENT_ERROR (demux, STREAM, DEMUX, (NULL),
2478               ("Audio stream using unknown error correction"));
2479           return NULL;
2480       }
2481
2482       break;
2483     }
2484
2485     case ASF_STREAM_VIDEO:{
2486       asf_stream_video_format video_format_object;
2487       asf_stream_video video_object;
2488       guint16 vsize;
2489
2490       if (!gst_asf_demux_get_stream_video (&video_object, &data, &size))
2491         goto not_enough_data;
2492
2493       vsize = video_object.size - 40;   /* Byte order gets offset by single byte */
2494
2495       GST_INFO ("object is a video stream with %u bytes of "
2496           "additional data", vsize);
2497
2498       if (!gst_asf_demux_get_stream_video_format (&video_format_object,
2499               &data, &size)) {
2500         goto not_enough_data;
2501       }
2502
2503       gst_asf_demux_add_video_stream (demux, &video_format_object, stream_id,
2504           &data, &size);
2505
2506       break;
2507     }
2508
2509     default:
2510       GST_WARNING_OBJECT (demux, "Unknown stream type for stream %u",
2511           stream_id);
2512       break;
2513   }
2514
2515   return gst_asf_demux_get_stream (demux, stream_id);
2516
2517 not_enough_data:
2518   {
2519     GST_WARNING_OBJECT (demux, "Unexpected end of data parsing stream object");
2520     /* we'll error out later if we found no streams */
2521     return NULL;
2522   }
2523 }
2524
2525 static const gchar *
2526 gst_asf_demux_get_gst_tag_from_tag_name (const gchar * name_utf8)
2527 {
2528   const struct
2529   {
2530     const gchar *asf_name;
2531     const gchar *gst_name;
2532   } tags[] = {
2533     {
2534     "WM/Genre", GST_TAG_GENRE}, {
2535     "WM/AlbumTitle", GST_TAG_ALBUM}, {
2536     "WM/AlbumArtist", GST_TAG_ARTIST}, {
2537     "WM/Picture", GST_TAG_IMAGE}, {
2538     "WM/Track", GST_TAG_TRACK_NUMBER}, {
2539     "WM/TrackNumber", GST_TAG_TRACK_NUMBER}, {
2540     "WM/Year", GST_TAG_DATE}
2541     /* { "WM/Composer", GST_TAG_COMPOSER } */
2542   };
2543   gsize out;
2544   guint i;
2545
2546   if (name_utf8 == NULL) {
2547     GST_WARNING ("Failed to convert name to UTF8, skipping");
2548     return NULL;
2549   }
2550
2551   out = strlen (name_utf8);
2552
2553   for (i = 0; i < G_N_ELEMENTS (tags); ++i) {
2554     if (strncmp (tags[i].asf_name, name_utf8, out) == 0) {
2555       GST_LOG ("map tagname '%s' -> '%s'", name_utf8, tags[i].gst_name);
2556       return tags[i].gst_name;
2557     }
2558   }
2559
2560   return NULL;
2561 }
2562
2563 /* gst_asf_demux_add_global_tags() takes ownership of taglist! */
2564 static void
2565 gst_asf_demux_add_global_tags (GstASFDemux * demux, GstTagList * taglist)
2566 {
2567   GstTagList *t;
2568
2569   GST_DEBUG_OBJECT (demux, "adding global tags: %" GST_PTR_FORMAT, taglist);
2570
2571   if (taglist == NULL)
2572     return;
2573
2574   if (gst_tag_list_is_empty (taglist)) {
2575     gst_tag_list_free (taglist);
2576     return;
2577   }
2578
2579   t = gst_tag_list_merge (demux->taglist, taglist, GST_TAG_MERGE_APPEND);
2580   if (demux->taglist)
2581     gst_tag_list_free (demux->taglist);
2582   gst_tag_list_free (taglist);
2583   demux->taglist = t;
2584   GST_LOG_OBJECT (demux, "global tags now: %" GST_PTR_FORMAT, demux->taglist);
2585 }
2586
2587 #define ASF_DEMUX_DATA_TYPE_UTF16LE_STRING  0
2588 #define ASF_DEMUX_DATA_TYPE_BYTE_ARRAY      1
2589 #define ASF_DEMUX_DATA_TYPE_DWORD           3
2590
2591 static void
2592 asf_demux_parse_picture_tag (GstTagList * tags, const guint8 * tag_data,
2593     guint tag_data_len)
2594 {
2595   GstByteReader r;
2596   const guint8 *img_data = NULL;
2597   guint32 img_data_len = 0;
2598   guint8 pic_type = 0;
2599
2600   gst_byte_reader_init (&r, tag_data, tag_data_len);
2601
2602   /* skip mime type string (we don't trust it and do our own typefinding),
2603    * and also skip the description string, since we don't use it */
2604   if (!gst_byte_reader_get_uint8 (&r, &pic_type) ||
2605       !gst_byte_reader_get_uint32_le (&r, &img_data_len) ||
2606       !gst_byte_reader_skip_string_utf16 (&r) ||
2607       !gst_byte_reader_skip_string_utf16 (&r) ||
2608       !gst_byte_reader_get_data (&r, img_data_len, &img_data)) {
2609     goto not_enough_data;
2610   }
2611
2612
2613   if (!gst_tag_list_add_id3_image (tags, img_data, img_data_len, pic_type))
2614     GST_DEBUG ("failed to add image extracted from WM/Picture tag to taglist");
2615
2616   return;
2617
2618 not_enough_data:
2619   {
2620     GST_DEBUG ("Failed to read WM/Picture tag: not enough data");
2621     GST_MEMDUMP ("WM/Picture data", tag_data, tag_data_len);
2622     return;
2623   }
2624 }
2625
2626 /* Extended Content Description Object */
2627 static GstFlowReturn
2628 gst_asf_demux_process_ext_content_desc (GstASFDemux * demux, guint8 * data,
2629     guint64 size)
2630 {
2631   /* Other known (and unused) 'text/unicode' metadata available :
2632    *
2633    *   WM/Lyrics =
2634    *   WM/MediaPrimaryClassID = {D1607DBC-E323-4BE2-86A1-48A42A28441E}
2635    *   WMFSDKVersion = 9.00.00.2980
2636    *   WMFSDKNeeded = 0.0.0.0000
2637    *   WM/UniqueFileIdentifier = AMGa_id=R    15334;AMGp_id=P     5149;AMGt_id=T  2324984
2638    *   WM/Publisher = 4AD
2639    *   WM/Provider = AMG
2640    *   WM/ProviderRating = 8
2641    *   WM/ProviderStyle = Rock (similar to WM/Genre)
2642    *   WM/GenreID (similar to WM/Genre)
2643    *   WM/TrackNumber (same as WM/Track but as a string)
2644    *
2645    * Other known (and unused) 'non-text' metadata available :
2646    *
2647    *   WM/EncodingTime
2648    *   WM/MCDI
2649    *   IsVBR
2650    *
2651    * We might want to read WM/TrackNumber and use atoi() if we don't have
2652    * WM/Track
2653    */
2654
2655   GstTagList *taglist;
2656   guint16 blockcount, i;
2657
2658   GST_INFO_OBJECT (demux, "object is an extended content description");
2659
2660   taglist = gst_tag_list_new_empty ();
2661
2662   /* Content Descriptor Count */
2663   if (size < 2)
2664     goto not_enough_data;
2665
2666   blockcount = gst_asf_demux_get_uint16 (&data, &size);
2667
2668   for (i = 1; i <= blockcount; ++i) {
2669     const gchar *gst_tag_name;
2670     guint16 datatype;
2671     guint16 value_len;
2672     guint16 name_len;
2673     GValue tag_value = { 0, };
2674     gsize in, out;
2675     gchar *name;
2676     gchar *name_utf8 = NULL;
2677     gchar *value;
2678
2679     /* Descriptor */
2680     if (!gst_asf_demux_get_string (&name, &name_len, &data, &size))
2681       goto not_enough_data;
2682
2683     if (size < 2) {
2684       g_free (name);
2685       goto not_enough_data;
2686     }
2687     /* Descriptor Value Data Type */
2688     datatype = gst_asf_demux_get_uint16 (&data, &size);
2689
2690     /* Descriptor Value (not really a string, but same thing reading-wise) */
2691     if (!gst_asf_demux_get_string (&value, &value_len, &data, &size)) {
2692       g_free (name);
2693       goto not_enough_data;
2694     }
2695
2696     name_utf8 =
2697         g_convert (name, name_len, "UTF-8", "UTF-16LE", &in, &out, NULL);
2698
2699     if (name_utf8 != NULL) {
2700       GST_DEBUG ("Found tag/metadata %s", name_utf8);
2701
2702       gst_tag_name = gst_asf_demux_get_gst_tag_from_tag_name (name_utf8);
2703       GST_DEBUG ("gst_tag_name %s", GST_STR_NULL (gst_tag_name));
2704
2705       switch (datatype) {
2706         case ASF_DEMUX_DATA_TYPE_UTF16LE_STRING:{
2707           gchar *value_utf8;
2708
2709           value_utf8 = g_convert (value, value_len, "UTF-8", "UTF-16LE",
2710               &in, &out, NULL);
2711
2712           /* get rid of tags with empty value */
2713           if (value_utf8 != NULL && *value_utf8 != '\0') {
2714             GST_DEBUG ("string value %s", value_utf8);
2715
2716             value_utf8[out] = '\0';
2717
2718             if (gst_tag_name != NULL) {
2719               if (strcmp (gst_tag_name, GST_TAG_DATE) == 0) {
2720                 guint year = atoi (value_utf8);
2721
2722                 if (year > 0) {
2723                   /* FIXME: really want a GDateTime with just the year field */
2724                   g_value_init (&tag_value, G_TYPE_DATE);
2725                   g_value_take_boxed (&tag_value, g_date_new_dmy (1, 1, year));
2726                 }
2727               } else if (strcmp (gst_tag_name, GST_TAG_GENRE) == 0) {
2728                 guint id3v1_genre_id;
2729                 const gchar *genre_str;
2730
2731                 if (sscanf (value_utf8, "(%u)", &id3v1_genre_id) == 1 &&
2732                     ((genre_str = gst_tag_id3_genre_get (id3v1_genre_id)))) {
2733                   GST_DEBUG ("Genre: %s -> %s", value_utf8, genre_str);
2734                   g_free (value_utf8);
2735                   value_utf8 = g_strdup (genre_str);
2736                 }
2737               } else {
2738                 GType tag_type;
2739
2740                 /* convert tag from string to other type if required */
2741                 tag_type = gst_tag_get_type (gst_tag_name);
2742                 g_value_init (&tag_value, tag_type);
2743                 if (!gst_value_deserialize (&tag_value, value_utf8)) {
2744                   GValue from_val = { 0, };
2745
2746                   g_value_init (&from_val, G_TYPE_STRING);
2747                   g_value_set_string (&from_val, value_utf8);
2748                   if (!g_value_transform (&from_val, &tag_value)) {
2749                     GST_WARNING_OBJECT (demux,
2750                         "Could not transform string tag to " "%s tag type %s",
2751                         gst_tag_name, g_type_name (tag_type));
2752                     g_value_unset (&tag_value);
2753                   }
2754                   g_value_unset (&from_val);
2755                 }
2756               }
2757             } else {
2758               /* metadata ! */
2759               GST_DEBUG ("Setting metadata");
2760               g_value_init (&tag_value, G_TYPE_STRING);
2761               g_value_set_string (&tag_value, value_utf8);
2762             }
2763           } else if (value_utf8 == NULL) {
2764             GST_WARNING ("Failed to convert string value to UTF8, skipping");
2765           } else {
2766             GST_DEBUG ("Skipping empty string value for %s",
2767                 GST_STR_NULL (gst_tag_name));
2768           }
2769           g_free (value_utf8);
2770           break;
2771         }
2772         case ASF_DEMUX_DATA_TYPE_BYTE_ARRAY:{
2773           if (gst_tag_name) {
2774             if (!g_str_equal (gst_tag_name, GST_TAG_IMAGE)) {
2775               GST_FIXME ("Unhandled byte array tag %s",
2776                   GST_STR_NULL (gst_tag_name));
2777               break;
2778             } else {
2779               asf_demux_parse_picture_tag (taglist, (guint8 *) value,
2780                   value_len);
2781             }
2782           }
2783           break;
2784         }
2785         case ASF_DEMUX_DATA_TYPE_DWORD:{
2786           guint uint_val = GST_READ_UINT32_LE (value);
2787
2788           /* this is the track number */
2789           g_value_init (&tag_value, G_TYPE_UINT);
2790
2791           /* WM/Track counts from 0 */
2792           if (!strcmp (name_utf8, "WM/Track"))
2793             ++uint_val;
2794
2795           g_value_set_uint (&tag_value, uint_val);
2796           break;
2797         }
2798         default:{
2799           GST_DEBUG ("Skipping tag %s of type %d", gst_tag_name, datatype);
2800           break;
2801         }
2802       }
2803
2804       if (G_IS_VALUE (&tag_value)) {
2805         if (gst_tag_name) {
2806           GstTagMergeMode merge_mode = GST_TAG_MERGE_APPEND;
2807
2808           /* WM/TrackNumber is more reliable than WM/Track, since the latter
2809            * is supposed to have a 0 base but is often wrongly written to start
2810            * from 1 as well, so prefer WM/TrackNumber when we have it: either
2811            * replace the value added earlier from WM/Track or put it first in
2812            * the list, so that it will get picked up by _get_uint() */
2813           if (strcmp (name_utf8, "WM/TrackNumber") == 0)
2814             merge_mode = GST_TAG_MERGE_REPLACE;
2815
2816           gst_tag_list_add_values (taglist, merge_mode, gst_tag_name,
2817               &tag_value, NULL);
2818         } else {
2819           GST_DEBUG ("Setting global metadata %s", name_utf8);
2820           gst_structure_set_value (demux->global_metadata, name_utf8,
2821               &tag_value);
2822         }
2823
2824         g_value_unset (&tag_value);
2825       }
2826     }
2827
2828     g_free (name);
2829     g_free (value);
2830     g_free (name_utf8);
2831   }
2832
2833   gst_asf_demux_add_global_tags (demux, taglist);
2834
2835   return GST_FLOW_OK;
2836
2837   /* Errors */
2838 not_enough_data:
2839   {
2840     GST_WARNING ("Unexpected end of data parsing ext content desc object");
2841     gst_tag_list_free (taglist);
2842     return GST_FLOW_OK;         /* not really fatal */
2843   }
2844 }
2845
2846 static GstStructure *
2847 gst_asf_demux_get_metadata_for_stream (GstASFDemux * demux, guint stream_num)
2848 {
2849   gchar sname[32];
2850   guint i;
2851
2852   g_snprintf (sname, sizeof (sname), "stream-%u", stream_num);
2853
2854   for (i = 0; i < gst_caps_get_size (demux->metadata); ++i) {
2855     GstStructure *s;
2856
2857     s = gst_caps_get_structure (demux->metadata, i);
2858     if (gst_structure_has_name (s, sname))
2859       return s;
2860   }
2861
2862   gst_caps_append_structure (demux->metadata, gst_structure_new_empty (sname));
2863
2864   /* try lookup again; demux->metadata took ownership of the structure, so we
2865    * can't really make any assumptions about what happened to it, so we can't
2866    * just return it directly after appending it */
2867   return gst_asf_demux_get_metadata_for_stream (demux, stream_num);
2868 }
2869
2870 static GstFlowReturn
2871 gst_asf_demux_process_metadata (GstASFDemux * demux, guint8 * data,
2872     guint64 size)
2873 {
2874   guint16 blockcount, i;
2875
2876   GST_INFO_OBJECT (demux, "object is a metadata object");
2877
2878   /* Content Descriptor Count */
2879   if (size < 2)
2880     goto not_enough_data;
2881
2882   blockcount = gst_asf_demux_get_uint16 (&data, &size);
2883
2884   for (i = 0; i < blockcount; ++i) {
2885     GstStructure *s;
2886     guint16 stream_num, name_len, data_type, lang_idx G_GNUC_UNUSED;
2887     guint32 data_len, ival;
2888     gchar *name_utf8;
2889
2890     if (size < (2 + 2 + 2 + 2 + 4))
2891       goto not_enough_data;
2892
2893     lang_idx = gst_asf_demux_get_uint16 (&data, &size);
2894     stream_num = gst_asf_demux_get_uint16 (&data, &size);
2895     name_len = gst_asf_demux_get_uint16 (&data, &size);
2896     data_type = gst_asf_demux_get_uint16 (&data, &size);
2897     data_len = gst_asf_demux_get_uint32 (&data, &size);
2898
2899     if (size < name_len + data_len)
2900       goto not_enough_data;
2901
2902     /* convert name to UTF-8 */
2903     name_utf8 = g_convert ((gchar *) data, name_len, "UTF-8", "UTF-16LE",
2904         NULL, NULL, NULL);
2905     gst_asf_demux_skip_bytes (name_len, &data, &size);
2906
2907     if (name_utf8 == NULL) {
2908       GST_WARNING ("Failed to convert value name to UTF8, skipping");
2909       gst_asf_demux_skip_bytes (data_len, &data, &size);
2910       continue;
2911     }
2912
2913     if (data_type != ASF_DEMUX_DATA_TYPE_DWORD) {
2914       gst_asf_demux_skip_bytes (data_len, &data, &size);
2915       g_free (name_utf8);
2916       continue;
2917     }
2918
2919     /* read DWORD */
2920     if (size < 4) {
2921       g_free (name_utf8);
2922       goto not_enough_data;
2923     }
2924
2925     ival = gst_asf_demux_get_uint32 (&data, &size);
2926
2927     /* skip anything else there may be, just in case */
2928     gst_asf_demux_skip_bytes (data_len - 4, &data, &size);
2929
2930     s = gst_asf_demux_get_metadata_for_stream (demux, stream_num);
2931     gst_structure_set (s, name_utf8, G_TYPE_INT, ival, NULL);
2932     g_free (name_utf8);
2933   }
2934
2935   GST_INFO_OBJECT (demux, "metadata = %" GST_PTR_FORMAT, demux->metadata);
2936   return GST_FLOW_OK;
2937
2938   /* Errors */
2939 not_enough_data:
2940   {
2941     GST_WARNING ("Unexpected end of data parsing metadata object");
2942     return GST_FLOW_OK;         /* not really fatal */
2943   }
2944 }
2945
2946 static GstFlowReturn
2947 gst_asf_demux_process_header (GstASFDemux * demux, guint8 * data, guint64 size)
2948 {
2949   GstFlowReturn ret = GST_FLOW_OK;
2950   guint32 i, num_objects;
2951   guint8 unknown G_GNUC_UNUSED;
2952
2953   /* Get the rest of the header's header */
2954   if (size < (4 + 1 + 1))
2955     goto not_enough_data;
2956
2957   num_objects = gst_asf_demux_get_uint32 (&data, &size);
2958   unknown = gst_asf_demux_get_uint8 (&data, &size);
2959   unknown = gst_asf_demux_get_uint8 (&data, &size);
2960
2961   GST_INFO_OBJECT (demux, "object is a header with %u parts", num_objects);
2962
2963   /* Loop through the header's objects, processing those */
2964   for (i = 0; i < num_objects; ++i) {
2965     GST_INFO_OBJECT (demux, "reading header part %u", i);
2966     ret = gst_asf_demux_process_object (demux, &data, &size);
2967     if (ret != GST_FLOW_OK) {
2968       GST_WARNING ("process_object returned %s", gst_asf_get_flow_name (ret));
2969       break;
2970     }
2971   }
2972
2973   return ret;
2974
2975 not_enough_data:
2976   {
2977     GST_ELEMENT_ERROR (demux, STREAM, DEMUX, (NULL),
2978         ("short read parsing HEADER object"));
2979     return GST_FLOW_ERROR;
2980   }
2981 }
2982
2983 static GstFlowReturn
2984 gst_asf_demux_process_file (GstASFDemux * demux, guint8 * data, guint64 size)
2985 {
2986   guint64 creation_time G_GNUC_UNUSED;
2987   guint64 file_size G_GNUC_UNUSED;
2988   guint64 send_time G_GNUC_UNUSED;
2989   guint64 packets_count, play_time, preroll;
2990   guint32 flags, min_pktsize, max_pktsize, min_bitrate G_GNUC_UNUSED;
2991
2992   if (size < (16 + 8 + 8 + 8 + 8 + 8 + 8 + 4 + 4 + 4 + 4))
2993     goto not_enough_data;
2994
2995   gst_asf_demux_skip_bytes (16, &data, &size);  /* skip GUID */
2996   file_size = gst_asf_demux_get_uint64 (&data, &size);
2997   creation_time = gst_asf_demux_get_uint64 (&data, &size);
2998   packets_count = gst_asf_demux_get_uint64 (&data, &size);
2999   play_time = gst_asf_demux_get_uint64 (&data, &size);
3000   send_time = gst_asf_demux_get_uint64 (&data, &size);
3001   preroll = gst_asf_demux_get_uint64 (&data, &size);
3002   flags = gst_asf_demux_get_uint32 (&data, &size);
3003   min_pktsize = gst_asf_demux_get_uint32 (&data, &size);
3004   max_pktsize = gst_asf_demux_get_uint32 (&data, &size);
3005   min_bitrate = gst_asf_demux_get_uint32 (&data, &size);
3006
3007   demux->broadcast = ! !(flags & 0x01);
3008   demux->seekable = ! !(flags & 0x02);
3009
3010   GST_DEBUG_OBJECT (demux, "min_pktsize = %u", min_pktsize);
3011   GST_DEBUG_OBJECT (demux, "flags::broadcast = %d", demux->broadcast);
3012   GST_DEBUG_OBJECT (demux, "flags::seekable  = %d", demux->seekable);
3013
3014   if (demux->broadcast) {
3015     /* these fields are invalid if the broadcast flag is set */
3016     play_time = 0;
3017     file_size = 0;
3018   }
3019
3020   if (min_pktsize != max_pktsize)
3021     goto non_fixed_packet_size;
3022
3023   demux->packet_size = max_pktsize;
3024
3025   /* FIXME: do we need send_time as well? what is it? */
3026   if ((play_time * 100) >= (preroll * GST_MSECOND))
3027     demux->play_time = (play_time * 100) - (preroll * GST_MSECOND);
3028   else
3029     demux->play_time = 0;
3030
3031   demux->preroll = preroll * GST_MSECOND;
3032
3033   /* initial latency */
3034   demux->latency = demux->preroll;
3035
3036   if (demux->play_time == 0)
3037     demux->seekable = FALSE;
3038
3039   GST_DEBUG_OBJECT (demux, "play_time %" GST_TIME_FORMAT,
3040       GST_TIME_ARGS (demux->play_time));
3041   GST_DEBUG_OBJECT (demux, "preroll   %" GST_TIME_FORMAT,
3042       GST_TIME_ARGS (demux->preroll));
3043
3044   if (demux->play_time > 0) {
3045     demux->segment.duration = demux->play_time;
3046   }
3047
3048   GST_INFO ("object is a file with %" G_GUINT64_FORMAT " data packets",
3049       packets_count);
3050   GST_INFO ("preroll = %" G_GUINT64_FORMAT, demux->preroll);
3051
3052   return GST_FLOW_OK;
3053
3054 /* ERRORS */
3055 non_fixed_packet_size:
3056   {
3057     GST_ELEMENT_ERROR (demux, STREAM, DEMUX, (NULL),
3058         ("packet size must be fixed"));
3059     return GST_FLOW_ERROR;
3060   }
3061 not_enough_data:
3062   {
3063     GST_ELEMENT_ERROR (demux, STREAM, DEMUX, (NULL),
3064         ("short read parsing FILE object"));
3065     return GST_FLOW_ERROR;
3066   }
3067 }
3068
3069 /* Content Description Object */
3070 static GstFlowReturn
3071 gst_asf_demux_process_comment (GstASFDemux * demux, guint8 * data, guint64 size)
3072 {
3073   struct
3074   {
3075     const gchar *gst_tag;
3076     guint16 val_length;
3077     gchar *val_utf8;
3078   } tags[5] = {
3079     {
3080     GST_TAG_TITLE, 0, NULL}, {
3081     GST_TAG_ARTIST, 0, NULL}, {
3082     GST_TAG_COPYRIGHT, 0, NULL}, {
3083     GST_TAG_DESCRIPTION, 0, NULL}, {
3084     GST_TAG_COMMENT, 0, NULL}
3085   };
3086   GstTagList *taglist;
3087   GValue value = { 0 };
3088   gsize in, out;
3089   gint i = -1;
3090
3091   GST_INFO_OBJECT (demux, "object is a comment");
3092
3093   if (size < (2 + 2 + 2 + 2 + 2))
3094     goto not_enough_data;
3095
3096   tags[0].val_length = gst_asf_demux_get_uint16 (&data, &size);
3097   tags[1].val_length = gst_asf_demux_get_uint16 (&data, &size);
3098   tags[2].val_length = gst_asf_demux_get_uint16 (&data, &size);
3099   tags[3].val_length = gst_asf_demux_get_uint16 (&data, &size);
3100   tags[4].val_length = gst_asf_demux_get_uint16 (&data, &size);
3101
3102   GST_DEBUG_OBJECT (demux, "Comment lengths: title=%d author=%d copyright=%d "
3103       "description=%d rating=%d", tags[0].val_length, tags[1].val_length,
3104       tags[2].val_length, tags[3].val_length, tags[4].val_length);
3105
3106   for (i = 0; i < G_N_ELEMENTS (tags); ++i) {
3107     if (size < tags[i].val_length)
3108       goto not_enough_data;
3109
3110     /* might be just '/0', '/0'... */
3111     if (tags[i].val_length > 2 && tags[i].val_length % 2 == 0) {
3112       /* convert to UTF-8 */
3113       tags[i].val_utf8 = g_convert ((gchar *) data, tags[i].val_length,
3114           "UTF-8", "UTF-16LE", &in, &out, NULL);
3115     }
3116     gst_asf_demux_skip_bytes (tags[i].val_length, &data, &size);
3117   }
3118
3119   /* parse metadata into taglist */
3120   taglist = gst_tag_list_new_empty ();
3121   g_value_init (&value, G_TYPE_STRING);
3122   for (i = 0; i < G_N_ELEMENTS (tags); ++i) {
3123     if (tags[i].val_utf8 && strlen (tags[i].val_utf8) > 0 && tags[i].gst_tag) {
3124       g_value_set_string (&value, tags[i].val_utf8);
3125       gst_tag_list_add_values (taglist, GST_TAG_MERGE_APPEND,
3126           tags[i].gst_tag, &value, NULL);
3127     }
3128   }
3129   g_value_unset (&value);
3130
3131   gst_asf_demux_add_global_tags (demux, taglist);
3132
3133   for (i = 0; i < G_N_ELEMENTS (tags); ++i)
3134     g_free (tags[i].val_utf8);
3135
3136   return GST_FLOW_OK;
3137
3138 not_enough_data:
3139   {
3140     GST_WARNING_OBJECT (demux, "unexpectedly short of data while processing "
3141         "comment tag section %d, skipping comment object", i);
3142     for (i = 0; i < G_N_ELEMENTS (tags); i++)
3143       g_free (tags[i].val_utf8);
3144     return GST_FLOW_OK;         /* not really fatal */
3145   }
3146 }
3147
3148 static GstFlowReturn
3149 gst_asf_demux_process_bitrate_props_object (GstASFDemux * demux, guint8 * data,
3150     guint64 size)
3151 {
3152   guint16 num_streams, i;
3153   AsfStream *stream;
3154
3155   if (size < 2)
3156     goto not_enough_data;
3157
3158   num_streams = gst_asf_demux_get_uint16 (&data, &size);
3159
3160   GST_INFO ("object is a bitrate properties object with %u streams",
3161       num_streams);
3162
3163   if (size < (num_streams * (2 + 4)))
3164     goto not_enough_data;
3165
3166   for (i = 0; i < num_streams; ++i) {
3167     guint32 bitrate;
3168     guint16 stream_id;
3169
3170     stream_id = gst_asf_demux_get_uint16 (&data, &size);
3171     bitrate = gst_asf_demux_get_uint32 (&data, &size);
3172
3173     if (stream_id < GST_ASF_DEMUX_NUM_STREAM_IDS) {
3174       GST_DEBUG_OBJECT (demux, "bitrate of stream %u = %u", stream_id, bitrate);
3175       stream = gst_asf_demux_get_stream (demux, stream_id);
3176       if (stream) {
3177         if (stream->pending_tags == NULL) {
3178           stream->pending_tags =
3179               gst_tag_list_new (GST_TAG_BITRATE, bitrate, NULL);
3180         }
3181       } else {
3182         GST_WARNING_OBJECT (demux, "Stream id %u wasn't found", stream_id);
3183       }
3184     } else {
3185       GST_WARNING ("stream id %u is too large", stream_id);
3186     }
3187   }
3188
3189   return GST_FLOW_OK;
3190
3191 not_enough_data:
3192   {
3193     GST_WARNING_OBJECT (demux, "short read parsing bitrate props object!");
3194     return GST_FLOW_OK;         /* not really fatal */
3195   }
3196 }
3197
3198 static GstFlowReturn
3199 gst_asf_demux_process_header_ext (GstASFDemux * demux, guint8 * data,
3200     guint64 size)
3201 {
3202   GstFlowReturn ret = GST_FLOW_OK;
3203   guint64 hdr_size;
3204
3205   /* Get the rest of the header's header */
3206   if (size < (16 + 2 + 4))
3207     goto not_enough_data;
3208
3209   /* skip GUID and two other bytes */
3210   gst_asf_demux_skip_bytes (16 + 2, &data, &size);
3211   hdr_size = gst_asf_demux_get_uint32 (&data, &size);
3212
3213   GST_INFO ("extended header object with a size of %u bytes", (guint) size);
3214
3215   /* FIXME: does data_size include the rest of the header that we have read? */
3216   if (hdr_size > size)
3217     goto not_enough_data;
3218
3219   while (hdr_size > 0) {
3220     ret = gst_asf_demux_process_object (demux, &data, &hdr_size);
3221     if (ret != GST_FLOW_OK)
3222       break;
3223   }
3224
3225   return ret;
3226
3227 not_enough_data:
3228   {
3229     GST_ELEMENT_ERROR (demux, STREAM, DEMUX, (NULL),
3230         ("short read parsing extended header object"));
3231     return GST_FLOW_ERROR;
3232   }
3233 }
3234
3235 static GstFlowReturn
3236 gst_asf_demux_process_language_list (GstASFDemux * demux, guint8 * data,
3237     guint64 size)
3238 {
3239   guint i;
3240
3241   if (size < 2)
3242     goto not_enough_data;
3243
3244   if (demux->languages) {
3245     GST_WARNING ("More than one LANGUAGE_LIST object in stream");
3246     g_strfreev (demux->languages);
3247     demux->languages = NULL;
3248     demux->num_languages = 0;
3249   }
3250
3251   demux->num_languages = gst_asf_demux_get_uint16 (&data, &size);
3252   GST_LOG ("%u languages:", demux->num_languages);
3253
3254   demux->languages = g_new0 (gchar *, demux->num_languages + 1);
3255   for (i = 0; i < demux->num_languages; ++i) {
3256     guint8 len, *lang_data = NULL;
3257
3258     if (size < 1)
3259       goto not_enough_data;
3260     len = gst_asf_demux_get_uint8 (&data, &size);
3261     if (gst_asf_demux_get_bytes (&lang_data, len, &data, &size)) {
3262       gchar *utf8;
3263
3264       utf8 = g_convert ((gchar *) lang_data, len, "UTF-8", "UTF-16LE", NULL,
3265           NULL, NULL);
3266
3267       /* truncate "en-us" etc. to just "en" */
3268       if (utf8 && strlen (utf8) >= 5 && (utf8[2] == '-' || utf8[2] == '_')) {
3269         utf8[2] = '\0';
3270       }
3271       GST_DEBUG ("[%u] %s", i, GST_STR_NULL (utf8));
3272       demux->languages[i] = utf8;
3273       g_free (lang_data);
3274     } else {
3275       goto not_enough_data;
3276     }
3277   }
3278
3279   return GST_FLOW_OK;
3280
3281 not_enough_data:
3282   {
3283     GST_WARNING_OBJECT (demux, "short read parsing language list object!");
3284     g_free (demux->languages);
3285     demux->languages = NULL;
3286     return GST_FLOW_OK;         /* not fatal */
3287   }
3288 }
3289
3290 static GstFlowReturn
3291 gst_asf_demux_process_simple_index (GstASFDemux * demux, guint8 * data,
3292     guint64 size)
3293 {
3294   GstClockTime interval;
3295   guint32 count, i;
3296
3297   if (size < (16 + 8 + 4 + 4))
3298     goto not_enough_data;
3299
3300   /* skip file id */
3301   gst_asf_demux_skip_bytes (16, &data, &size);
3302   interval = gst_asf_demux_get_uint64 (&data, &size) * (GstClockTime) 100;
3303   gst_asf_demux_skip_bytes (4, &data, &size);
3304   count = gst_asf_demux_get_uint32 (&data, &size);
3305   if (count > 0) {
3306     demux->sidx_interval = interval;
3307     demux->sidx_num_entries = count;
3308     g_free (demux->sidx_entries);
3309     demux->sidx_entries = g_new0 (AsfSimpleIndexEntry, count);
3310
3311     for (i = 0; i < count; ++i) {
3312       if (G_UNLIKELY (size <= 6))
3313         break;
3314       demux->sidx_entries[i].packet = gst_asf_demux_get_uint32 (&data, &size);
3315       demux->sidx_entries[i].count = gst_asf_demux_get_uint16 (&data, &size);
3316       GST_LOG_OBJECT (demux, "%" GST_TIME_FORMAT " = packet %4u  count : %2d",
3317           GST_TIME_ARGS (i * interval), demux->sidx_entries[i].packet,
3318           demux->sidx_entries[i].count);
3319     }
3320   } else {
3321     GST_DEBUG_OBJECT (demux, "simple index object with 0 entries");
3322   }
3323
3324   return GST_FLOW_OK;
3325
3326 not_enough_data:
3327   {
3328     GST_WARNING_OBJECT (demux, "short read parsing simple index object!");
3329     return GST_FLOW_OK;         /* not fatal */
3330   }
3331 }
3332
3333 static GstFlowReturn
3334 gst_asf_demux_process_advanced_mutual_exclusion (GstASFDemux * demux,
3335     guint8 * data, guint64 size)
3336 {
3337   ASFGuid guid;
3338   guint16 num, i;
3339   guint8 *mes;
3340
3341   if (size < 16 + 2 + (2 * 2))
3342     goto not_enough_data;
3343
3344   gst_asf_demux_get_guid (&guid, &data, &size);
3345   num = gst_asf_demux_get_uint16 (&data, &size);
3346
3347   if (num < 2) {
3348     GST_WARNING_OBJECT (demux, "nonsensical mutually exclusive streams count");
3349     return GST_FLOW_OK;
3350   }
3351
3352   if (size < (num * sizeof (guint16)))
3353     goto not_enough_data;
3354
3355   /* read mutually exclusive stream numbers */
3356   mes = g_new (guint8, num + 1);
3357   for (i = 0; i < num; ++i) {
3358     mes[i] = gst_asf_demux_get_uint16 (&data, &size) & 0x7f;
3359     GST_LOG_OBJECT (demux, "mutually exclusive: stream #%d", mes[i]);
3360   }
3361
3362   /* add terminator so we can easily get the count or know when to stop */
3363   mes[i] = (guint8) - 1;
3364
3365   demux->mut_ex_streams = g_slist_append (demux->mut_ex_streams, mes);
3366
3367   return GST_FLOW_OK;
3368
3369   /* Errors */
3370 not_enough_data:
3371   {
3372     GST_WARNING_OBJECT (demux, "short read parsing advanced mutual exclusion");
3373     return GST_FLOW_OK;         /* not absolutely fatal */
3374   }
3375 }
3376
3377 static GstFlowReturn
3378 gst_asf_demux_process_ext_stream_props (GstASFDemux * demux, guint8 * data,
3379     guint64 size)
3380 {
3381   AsfStreamExtProps esp;
3382   AsfStream *stream = NULL;
3383   AsfObject stream_obj;
3384   guint16 stream_name_count;
3385   guint16 num_payload_ext;
3386   guint64 len;
3387   guint8 *stream_obj_data = NULL;
3388   guint8 *data_start;
3389   guint obj_size;
3390   guint i, stream_num;
3391
3392   data_start = data;
3393   obj_size = (guint) size;
3394
3395   if (size < 64)
3396     goto not_enough_data;
3397
3398   esp.valid = TRUE;
3399   esp.start_time = gst_asf_demux_get_uint64 (&data, &size) * GST_MSECOND;
3400   esp.end_time = gst_asf_demux_get_uint64 (&data, &size) * GST_MSECOND;
3401   esp.data_bitrate = gst_asf_demux_get_uint32 (&data, &size);
3402   esp.buffer_size = gst_asf_demux_get_uint32 (&data, &size);
3403   esp.intial_buf_fullness = gst_asf_demux_get_uint32 (&data, &size);
3404   esp.data_bitrate2 = gst_asf_demux_get_uint32 (&data, &size);
3405   esp.buffer_size2 = gst_asf_demux_get_uint32 (&data, &size);
3406   esp.intial_buf_fullness2 = gst_asf_demux_get_uint32 (&data, &size);
3407   esp.max_obj_size = gst_asf_demux_get_uint32 (&data, &size);
3408   esp.flags = gst_asf_demux_get_uint32 (&data, &size);
3409   stream_num = gst_asf_demux_get_uint16 (&data, &size);
3410   esp.lang_idx = gst_asf_demux_get_uint16 (&data, &size);
3411   esp.avg_time_per_frame = gst_asf_demux_get_uint64 (&data, &size);
3412   stream_name_count = gst_asf_demux_get_uint16 (&data, &size);
3413   num_payload_ext = gst_asf_demux_get_uint16 (&data, &size);
3414
3415   GST_INFO ("start_time             = %" GST_TIME_FORMAT,
3416       GST_TIME_ARGS (esp.start_time));
3417   GST_INFO ("end_time               = %" GST_TIME_FORMAT,
3418       GST_TIME_ARGS (esp.end_time));
3419   GST_INFO ("flags                  = %08x", esp.flags);
3420   GST_INFO ("average time per frame = %" GST_TIME_FORMAT,
3421       GST_TIME_ARGS (esp.avg_time_per_frame * 100));
3422   GST_INFO ("stream number          = %u", stream_num);
3423   GST_INFO ("stream language ID idx = %u (%s)", esp.lang_idx,
3424       (esp.lang_idx < demux->num_languages) ?
3425       GST_STR_NULL (demux->languages[esp.lang_idx]) : "??");
3426   GST_INFO ("stream name count      = %u", stream_name_count);
3427
3428   /* read stream names */
3429   for (i = 0; i < stream_name_count; ++i) {
3430     guint16 stream_lang_idx G_GNUC_UNUSED;
3431     gchar *stream_name = NULL;
3432
3433     if (size < 2)
3434       goto not_enough_data;
3435     stream_lang_idx = gst_asf_demux_get_uint16 (&data, &size);
3436     if (!gst_asf_demux_get_string (&stream_name, NULL, &data, &size))
3437       goto not_enough_data;
3438     GST_INFO ("stream name %d: %s", i, GST_STR_NULL (stream_name));
3439     g_free (stream_name);       /* TODO: store names in struct */
3440   }
3441
3442   /* read payload extension systems stuff */
3443   GST_LOG ("payload extension systems count = %u", num_payload_ext);
3444
3445   if (num_payload_ext > 0)
3446     esp.payload_extensions = g_new0 (AsfPayloadExtension, num_payload_ext + 1);
3447   else
3448     esp.payload_extensions = NULL;
3449
3450   for (i = 0; i < num_payload_ext; ++i) {
3451     AsfPayloadExtension ext;
3452     ASFGuid ext_guid;
3453     guint32 sys_info_len;
3454
3455     if (size < 16 + 2 + 4)
3456       goto not_enough_data;
3457
3458     gst_asf_demux_get_guid (&ext_guid, &data, &size);
3459     ext.id = gst_asf_demux_identify_guid (asf_payload_ext_guids, &ext_guid);
3460     ext.len = gst_asf_demux_get_uint16 (&data, &size);
3461
3462     sys_info_len = gst_asf_demux_get_uint32 (&data, &size);
3463     GST_LOG ("payload systems info len = %u", sys_info_len);
3464     if (!gst_asf_demux_skip_bytes (sys_info_len, &data, &size))
3465       goto not_enough_data;
3466
3467     esp.payload_extensions[i] = ext;
3468   }
3469
3470   GST_LOG ("bytes read: %u/%u", (guint) (data - data_start), obj_size);
3471
3472   /* there might be an optional STREAM_INFO object here now; if not, we
3473    * should have parsed the corresponding stream info object already (since
3474    * we are parsing the extended stream properties objects delayed) */
3475   if (size == 0) {
3476     stream = gst_asf_demux_get_stream (demux, stream_num);
3477     goto done;
3478   }
3479
3480   /* get size of the stream object */
3481   if (!asf_demux_peek_object (demux, data, size, &stream_obj, TRUE))
3482     goto not_enough_data;
3483
3484   if (stream_obj.id != ASF_OBJ_STREAM)
3485     goto expected_stream_object;
3486
3487   if (stream_obj.size < ASF_OBJECT_HEADER_SIZE ||
3488       stream_obj.size > (10 * 1024 * 1024))
3489     goto not_enough_data;
3490
3491   gst_asf_demux_skip_bytes (ASF_OBJECT_HEADER_SIZE, &data, &size);
3492
3493   /* process this stream object later after all the other 'normal' ones
3494    * have been processed (since the others are more important/non-hidden) */
3495   len = stream_obj.size - ASF_OBJECT_HEADER_SIZE;
3496   if (!gst_asf_demux_get_bytes (&stream_obj_data, len, &data, &size))
3497     goto not_enough_data;
3498
3499   /* parse stream object */
3500   stream = gst_asf_demux_parse_stream_object (demux, stream_obj_data, len);
3501   g_free (stream_obj_data);
3502
3503 done:
3504
3505   if (stream) {
3506     stream->ext_props = esp;
3507
3508     /* try to set the framerate */
3509     if (stream->is_video && stream->caps) {
3510       GValue framerate = { 0 };
3511       GstStructure *s;
3512       gint num, denom;
3513
3514       g_value_init (&framerate, GST_TYPE_FRACTION);
3515
3516       num = GST_SECOND / 100;
3517       denom = esp.avg_time_per_frame;
3518       if (denom == 0) {
3519         /* avoid division by 0, assume 25/1 framerate */
3520         denom = GST_SECOND / 2500;
3521       }
3522
3523       gst_value_set_fraction (&framerate, num, denom);
3524
3525       stream->caps = gst_caps_make_writable (stream->caps);
3526       s = gst_caps_get_structure (stream->caps, 0);
3527       gst_structure_set_value (s, "framerate", &framerate);
3528       g_value_unset (&framerate);
3529       GST_DEBUG_OBJECT (demux, "setting framerate of %d/%d = %f",
3530           num, denom, ((gdouble) num) / denom);
3531     }
3532
3533     /* add language info now if we have it */
3534     if (stream->ext_props.lang_idx < demux->num_languages) {
3535       if (stream->pending_tags == NULL)
3536         stream->pending_tags = gst_tag_list_new_empty ();
3537       GST_LOG_OBJECT (demux, "stream %u has language '%s'", stream->id,
3538           demux->languages[stream->ext_props.lang_idx]);
3539       gst_tag_list_add (stream->pending_tags, GST_TAG_MERGE_APPEND,
3540           GST_TAG_LANGUAGE_CODE, demux->languages[stream->ext_props.lang_idx],
3541           NULL);
3542     }
3543   } else {
3544     GST_WARNING_OBJECT (demux, "Ext. stream properties for unknown stream");
3545   }
3546
3547   return GST_FLOW_OK;
3548
3549   /* Errors */
3550 not_enough_data:
3551   {
3552     GST_WARNING_OBJECT (demux, "short read parsing ext stream props object!");
3553     return GST_FLOW_OK;         /* not absolutely fatal */
3554   }
3555 expected_stream_object:
3556   {
3557     GST_WARNING_OBJECT (demux, "error parsing extended stream properties "
3558         "object: expected embedded stream object, but got %s object instead!",
3559         gst_asf_get_guid_nick (asf_object_guids, stream_obj.id));
3560     return GST_FLOW_OK;         /* not absolutely fatal */
3561   }
3562 }
3563
3564 static const gchar *
3565 gst_asf_demux_push_obj (GstASFDemux * demux, guint32 obj_id)
3566 {
3567   const gchar *nick;
3568
3569   nick = gst_asf_get_guid_nick (asf_object_guids, obj_id);
3570   if (g_str_has_prefix (nick, "ASF_OBJ_"))
3571     nick += strlen ("ASF_OBJ_");
3572
3573   if (demux->objpath == NULL) {
3574     demux->objpath = g_strdup (nick);
3575   } else {
3576     gchar *newpath;
3577
3578     newpath = g_strdup_printf ("%s/%s", demux->objpath, nick);
3579     g_free (demux->objpath);
3580     demux->objpath = newpath;
3581   }
3582
3583   return (const gchar *) demux->objpath;
3584 }
3585
3586 static void
3587 gst_asf_demux_pop_obj (GstASFDemux * demux)
3588 {
3589   gchar *s;
3590
3591   if ((s = g_strrstr (demux->objpath, "/"))) {
3592     *s = '\0';
3593   } else {
3594     g_free (demux->objpath);
3595     demux->objpath = NULL;
3596   }
3597 }
3598
3599 static void
3600 gst_asf_demux_process_queued_extended_stream_objects (GstASFDemux * demux)
3601 {
3602   GSList *l;
3603   guint i;
3604
3605   /* Parse the queued extended stream property objects and add the info
3606    * to the existing streams or add the new embedded streams, but without
3607    * activating them yet */
3608   GST_LOG_OBJECT (demux, "%u queued extended stream properties objects",
3609       g_slist_length (demux->ext_stream_props));
3610
3611   for (l = demux->ext_stream_props, i = 0; l != NULL; l = l->next, ++i) {
3612     GstBuffer *buf = GST_BUFFER (l->data);
3613     GstMapInfo map;
3614
3615     gst_buffer_map (buf, &map, GST_MAP_READ);
3616
3617     GST_LOG_OBJECT (demux, "parsing ext. stream properties object #%u", i);
3618     gst_asf_demux_process_ext_stream_props (demux, map.data, map.size);
3619     gst_buffer_unmap (buf, &map);
3620     gst_buffer_unref (buf);
3621   }
3622   g_slist_free (demux->ext_stream_props);
3623   demux->ext_stream_props = NULL;
3624 }
3625
3626 #if 0
3627 static void
3628 gst_asf_demux_activate_ext_props_streams (GstASFDemux * demux)
3629 {
3630   guint i, j;
3631
3632   for (i = 0; i < demux->num_streams; ++i) {
3633     AsfStream *stream;
3634     gboolean is_hidden;
3635     GSList *x;
3636
3637     stream = &demux->stream[i];
3638
3639     GST_LOG_OBJECT (demux, "checking  stream %2u", stream->id);
3640
3641     if (stream->active) {
3642       GST_LOG_OBJECT (demux, "stream %2u is already activated", stream->id);
3643       continue;
3644     }
3645
3646     is_hidden = FALSE;
3647     for (x = demux->mut_ex_streams; x != NULL; x = x->next) {
3648       guint8 *mes;
3649
3650       /* check for each mutual exclusion whether it affects this stream */
3651       for (mes = (guint8 *) x->data; mes != NULL && *mes != 0xff; ++mes) {
3652         if (*mes == stream->id) {
3653           /* if yes, check if we've already added streams that are mutually
3654            * exclusive with the stream we're about to add */
3655           for (mes = (guint8 *) x->data; mes != NULL && *mes != 0xff; ++mes) {
3656             for (j = 0; j < demux->num_streams; ++j) {
3657               /* if the broadcast flag is set, assume the hidden streams aren't
3658                * actually streamed and hide them (or playbin won't work right),
3659                * otherwise assume their data is available */
3660               if (demux->stream[j].id == *mes && demux->broadcast) {
3661                 is_hidden = TRUE;
3662                 GST_LOG_OBJECT (demux, "broadcast stream ID %d to be added is "
3663                     "mutually exclusive with already existing stream ID %d, "
3664                     "hiding stream", stream->id, demux->stream[j].id);
3665                 goto next;
3666               }
3667             }
3668           }
3669           break;
3670         }
3671       }
3672     }
3673
3674   next:
3675
3676     /* FIXME: we should do stream activation based on preroll data in
3677      * streaming mode too */
3678     if (demux->streaming && !is_hidden)
3679       gst_asf_demux_activate_stream (demux, stream);
3680   }
3681 }
3682 #endif
3683
3684 static GstFlowReturn
3685 gst_asf_demux_process_object (GstASFDemux * demux, guint8 ** p_data,
3686     guint64 * p_size)
3687 {
3688   GstFlowReturn ret = GST_FLOW_OK;
3689   AsfObject obj;
3690   guint64 obj_data_size;
3691
3692   if (*p_size < ASF_OBJECT_HEADER_SIZE)
3693     return ASF_FLOW_NEED_MORE_DATA;
3694
3695   asf_demux_peek_object (demux, *p_data, ASF_OBJECT_HEADER_SIZE, &obj, TRUE);
3696   gst_asf_demux_skip_bytes (ASF_OBJECT_HEADER_SIZE, p_data, p_size);
3697
3698   obj_data_size = obj.size - ASF_OBJECT_HEADER_SIZE;
3699
3700   if (*p_size < obj_data_size)
3701     return ASF_FLOW_NEED_MORE_DATA;
3702
3703   gst_asf_demux_push_obj (demux, obj.id);
3704
3705   GST_INFO ("%s: size %" G_GUINT64_FORMAT, demux->objpath, obj.size);
3706
3707   switch (obj.id) {
3708     case ASF_OBJ_STREAM:
3709       gst_asf_demux_parse_stream_object (demux, *p_data, obj_data_size);
3710       ret = GST_FLOW_OK;
3711       break;
3712     case ASF_OBJ_FILE:
3713       ret = gst_asf_demux_process_file (demux, *p_data, obj_data_size);
3714       break;
3715     case ASF_OBJ_HEADER:
3716       ret = gst_asf_demux_process_header (demux, *p_data, obj_data_size);
3717       break;
3718     case ASF_OBJ_COMMENT:
3719       ret = gst_asf_demux_process_comment (demux, *p_data, obj_data_size);
3720       break;
3721     case ASF_OBJ_HEAD1:
3722       ret = gst_asf_demux_process_header_ext (demux, *p_data, obj_data_size);
3723       break;
3724     case ASF_OBJ_BITRATE_PROPS:
3725       ret =
3726           gst_asf_demux_process_bitrate_props_object (demux, *p_data,
3727           obj_data_size);
3728       break;
3729     case ASF_OBJ_EXT_CONTENT_DESC:
3730       ret =
3731           gst_asf_demux_process_ext_content_desc (demux, *p_data,
3732           obj_data_size);
3733       break;
3734     case ASF_OBJ_METADATA_OBJECT:
3735       ret = gst_asf_demux_process_metadata (demux, *p_data, obj_data_size);
3736       break;
3737     case ASF_OBJ_EXTENDED_STREAM_PROPS:{
3738       GstBuffer *buf;
3739
3740       /* process these later, we might not have parsed the corresponding
3741        * stream object yet */
3742       GST_LOG ("%s: queued for later parsing", demux->objpath);
3743       buf = gst_buffer_new_and_alloc (obj_data_size);
3744       gst_buffer_fill (buf, 0, *p_data, obj_data_size);
3745       demux->ext_stream_props = g_slist_append (demux->ext_stream_props, buf);
3746       ret = GST_FLOW_OK;
3747       break;
3748     }
3749     case ASF_OBJ_LANGUAGE_LIST:
3750       ret = gst_asf_demux_process_language_list (demux, *p_data, obj_data_size);
3751       break;
3752     case ASF_OBJ_ADVANCED_MUTUAL_EXCLUSION:
3753       ret = gst_asf_demux_process_advanced_mutual_exclusion (demux, *p_data,
3754           obj_data_size);
3755       break;
3756     case ASF_OBJ_SIMPLE_INDEX:
3757       ret = gst_asf_demux_process_simple_index (demux, *p_data, obj_data_size);
3758       break;
3759     case ASF_OBJ_CONTENT_ENCRYPTION:
3760     case ASF_OBJ_EXT_CONTENT_ENCRYPTION:
3761     case ASF_OBJ_DIGITAL_SIGNATURE_OBJECT:
3762     case ASF_OBJ_UNKNOWN_ENCRYPTION_OBJECT:
3763       goto error_encrypted;
3764     case ASF_OBJ_CONCEAL_NONE:
3765     case ASF_OBJ_HEAD2:
3766     case ASF_OBJ_UNDEFINED:
3767     case ASF_OBJ_CODEC_COMMENT:
3768     case ASF_OBJ_INDEX:
3769     case ASF_OBJ_PADDING:
3770     case ASF_OBJ_BITRATE_MUTEX:
3771     case ASF_OBJ_COMPATIBILITY:
3772     case ASF_OBJ_INDEX_PLACEHOLDER:
3773     case ASF_OBJ_INDEX_PARAMETERS:
3774     case ASF_OBJ_STREAM_PRIORITIZATION:
3775     case ASF_OBJ_SCRIPT_COMMAND:
3776     default:
3777       /* Unknown/unhandled object, skip it and hope for the best */
3778       GST_INFO ("%s: skipping object", demux->objpath);
3779       ret = GST_FLOW_OK;
3780       break;
3781   }
3782
3783   /* this can't fail, we checked the number of bytes available before */
3784   gst_asf_demux_skip_bytes (obj_data_size, p_data, p_size);
3785
3786   GST_LOG ("%s: ret = %s", demux->objpath, gst_asf_get_flow_name (ret));
3787
3788   gst_asf_demux_pop_obj (demux);
3789
3790   return ret;
3791
3792 /* ERRORS */
3793 error_encrypted:
3794   {
3795     GST_ELEMENT_ERROR (demux, STREAM, DECRYPT, (NULL), (NULL));
3796     return GST_FLOW_ERROR;
3797   }
3798 }
3799
3800 static void
3801 gst_asf_demux_descramble_buffer (GstASFDemux * demux, AsfStream * stream,
3802     GstBuffer ** p_buffer)
3803 {
3804   GstBuffer *descrambled_buffer;
3805   GstBuffer *scrambled_buffer;
3806   GstBuffer *sub_buffer;
3807   guint offset;
3808   guint off;
3809   guint row;
3810   guint col;
3811   guint idx;
3812
3813   /* descrambled_buffer is initialised in the first iteration */
3814   descrambled_buffer = NULL;
3815   scrambled_buffer = *p_buffer;
3816
3817   if (gst_buffer_get_size (scrambled_buffer) <
3818       demux->ds_packet_size * demux->span)
3819     return;
3820
3821   for (offset = 0; offset < gst_buffer_get_size (scrambled_buffer);
3822       offset += demux->ds_chunk_size) {
3823     off = offset / demux->ds_chunk_size;
3824     row = off / demux->span;
3825     col = off % demux->span;
3826     idx = row + col * demux->ds_packet_size / demux->ds_chunk_size;
3827     GST_DEBUG ("idx=%u, row=%u, col=%u, off=%u, ds_chunk_size=%u", idx, row,
3828         col, off, demux->ds_chunk_size);
3829     GST_DEBUG ("scrambled buffer size=%" G_GSIZE_FORMAT
3830         ", span=%u, packet_size=%u", gst_buffer_get_size (scrambled_buffer),
3831         demux->span, demux->ds_packet_size);
3832     GST_DEBUG ("gst_buffer_get_size (scrambled_buffer) = %" G_GSIZE_FORMAT,
3833         gst_buffer_get_size (scrambled_buffer));
3834     sub_buffer =
3835         gst_buffer_copy_region (scrambled_buffer, GST_BUFFER_COPY_NONE,
3836         idx * demux->ds_chunk_size, demux->ds_chunk_size);
3837     if (!offset) {
3838       descrambled_buffer = sub_buffer;
3839     } else {
3840       descrambled_buffer = gst_buffer_append (descrambled_buffer, sub_buffer);
3841     }
3842   }
3843
3844   GST_BUFFER_TIMESTAMP (descrambled_buffer) =
3845       GST_BUFFER_TIMESTAMP (scrambled_buffer);
3846   GST_BUFFER_DURATION (descrambled_buffer) =
3847       GST_BUFFER_DURATION (scrambled_buffer);
3848   GST_BUFFER_OFFSET (descrambled_buffer) = GST_BUFFER_OFFSET (scrambled_buffer);
3849   GST_BUFFER_OFFSET_END (descrambled_buffer) =
3850       GST_BUFFER_OFFSET_END (scrambled_buffer);
3851
3852   /* FIXME/CHECK: do we need to transfer buffer flags here too? */
3853
3854   gst_buffer_unref (scrambled_buffer);
3855   *p_buffer = descrambled_buffer;
3856 }
3857
3858 static gboolean
3859 gst_asf_demux_element_send_event (GstElement * element, GstEvent * event)
3860 {
3861   GstASFDemux *demux = GST_ASF_DEMUX (element);
3862   gint i;
3863
3864   GST_DEBUG ("handling element event of type %s", GST_EVENT_TYPE_NAME (event));
3865
3866   for (i = 0; i < demux->num_streams; ++i) {
3867     gst_event_ref (event);
3868     if (gst_asf_demux_handle_src_event (demux->stream[i].pad,
3869             GST_OBJECT_CAST (element), event)) {
3870       gst_event_unref (event);
3871       return TRUE;
3872     }
3873   }
3874
3875   gst_event_unref (event);
3876   return FALSE;
3877 }
3878
3879 /* takes ownership of the passed event */
3880 static gboolean
3881 gst_asf_demux_send_event_unlocked (GstASFDemux * demux, GstEvent * event)
3882 {
3883   gboolean ret = TRUE;
3884   gint i;
3885
3886   GST_DEBUG_OBJECT (demux, "sending %s event to all source pads",
3887       GST_EVENT_TYPE_NAME (event));
3888
3889   for (i = 0; i < demux->num_streams; ++i) {
3890     gst_event_ref (event);
3891     ret &= gst_pad_push_event (demux->stream[i].pad, event);
3892   }
3893   gst_event_unref (event);
3894   return ret;
3895 }
3896
3897 static gboolean
3898 gst_asf_demux_handle_src_query (GstPad * pad, GstObject * parent,
3899     GstQuery * query)
3900 {
3901   GstASFDemux *demux;
3902   gboolean res = FALSE;
3903
3904   demux = GST_ASF_DEMUX (parent);
3905
3906   GST_DEBUG ("handling %s query",
3907       gst_query_type_get_name (GST_QUERY_TYPE (query)));
3908
3909   switch (GST_QUERY_TYPE (query)) {
3910     case GST_QUERY_DURATION:
3911     {
3912       GstFormat format;
3913
3914       gst_query_parse_duration (query, &format, NULL);
3915
3916       if (format != GST_FORMAT_TIME) {
3917         GST_LOG ("only support duration queries in TIME format");
3918         break;
3919       }
3920
3921       GST_OBJECT_LOCK (demux);
3922
3923       if (demux->segment.duration != GST_CLOCK_TIME_NONE) {
3924         GST_LOG ("returning duration: %" GST_TIME_FORMAT,
3925             GST_TIME_ARGS (demux->segment.duration));
3926
3927         gst_query_set_duration (query, GST_FORMAT_TIME,
3928             demux->segment.duration);
3929
3930         res = TRUE;
3931       } else {
3932         GST_LOG ("duration not known yet");
3933       }
3934
3935       GST_OBJECT_UNLOCK (demux);
3936       break;
3937     }
3938
3939     case GST_QUERY_POSITION:{
3940       GstFormat format;
3941
3942       gst_query_parse_position (query, &format, NULL);
3943
3944       if (format != GST_FORMAT_TIME) {
3945         GST_LOG ("only support position queries in TIME format");
3946         break;
3947       }
3948
3949       GST_OBJECT_LOCK (demux);
3950
3951       if (demux->segment.position != GST_CLOCK_TIME_NONE) {
3952         GST_LOG ("returning position: %" GST_TIME_FORMAT,
3953             GST_TIME_ARGS (demux->segment.position));
3954
3955         gst_query_set_position (query, GST_FORMAT_TIME,
3956             demux->segment.position);
3957
3958         res = TRUE;
3959       } else {
3960         GST_LOG ("position not known yet");
3961       }
3962
3963       GST_OBJECT_UNLOCK (demux);
3964       break;
3965     }
3966
3967     case GST_QUERY_SEEKING:{
3968       GstFormat format;
3969
3970       gst_query_parse_seeking (query, &format, NULL, NULL, NULL);
3971       if (format == GST_FORMAT_TIME) {
3972         gint64 duration;
3973
3974         GST_OBJECT_LOCK (demux);
3975         duration = demux->segment.duration;
3976         GST_OBJECT_UNLOCK (demux);
3977
3978         if (!demux->streaming || !demux->seekable) {
3979           gst_query_set_seeking (query, GST_FORMAT_TIME, demux->seekable, 0,
3980               duration);
3981           res = TRUE;
3982         } else {
3983           GstFormat fmt;
3984           gboolean seekable;
3985
3986           /* try downstream first in TIME */
3987           res = gst_pad_query_default (pad, parent, query);
3988
3989           gst_query_parse_seeking (query, &fmt, &seekable, NULL, NULL);
3990           GST_LOG_OBJECT (demux, "upstream %s seekable %d",
3991               GST_STR_NULL (gst_format_get_name (fmt)), seekable);
3992           /* if no luck, maybe in BYTES */
3993           if (!seekable || fmt != GST_FORMAT_TIME) {
3994             GstQuery *q;
3995
3996             q = gst_query_new_seeking (GST_FORMAT_BYTES);
3997             if ((res = gst_pad_peer_query (demux->sinkpad, q))) {
3998               gst_query_parse_seeking (q, &fmt, &seekable, NULL, NULL);
3999               GST_LOG_OBJECT (demux, "upstream %s seekable %d",
4000                   GST_STR_NULL (gst_format_get_name (fmt)), seekable);
4001               if (fmt != GST_FORMAT_BYTES)
4002                 seekable = FALSE;
4003             }
4004             gst_query_unref (q);
4005             gst_query_set_seeking (query, GST_FORMAT_TIME, seekable, 0,
4006                 duration);
4007             res = TRUE;
4008           }
4009         }
4010       } else
4011         GST_LOG_OBJECT (demux, "only support seeking in TIME format");
4012       break;
4013     }
4014
4015     case GST_QUERY_LATENCY:
4016     {
4017       gboolean live;
4018       GstClockTime min, max;
4019
4020       /* preroll delay does not matter in non-live pipeline,
4021        * but we might end up in a live (rtsp) one ... */
4022
4023       /* first forward */
4024       res = gst_pad_query_default (pad, parent, query);
4025       if (!res)
4026         break;
4027
4028       gst_query_parse_latency (query, &live, &min, &max);
4029
4030       GST_DEBUG_OBJECT (demux, "Peer latency: live %d, min %"
4031           GST_TIME_FORMAT " max %" GST_TIME_FORMAT, live,
4032           GST_TIME_ARGS (min), GST_TIME_ARGS (max));
4033
4034       GST_OBJECT_LOCK (demux);
4035       if (min != -1)
4036         min += demux->latency;
4037       if (max != -1)
4038         max += demux->latency;
4039       GST_OBJECT_UNLOCK (demux);
4040
4041       gst_query_set_latency (query, live, min, max);
4042       break;
4043     }
4044     default:
4045       res = gst_pad_query_default (pad, parent, query);
4046       break;
4047   }
4048
4049   return res;
4050 }
4051
4052 static GstStateChangeReturn
4053 gst_asf_demux_change_state (GstElement * element, GstStateChange transition)
4054 {
4055   GstASFDemux *demux = GST_ASF_DEMUX (element);
4056   GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;
4057
4058   switch (transition) {
4059     case GST_STATE_CHANGE_NULL_TO_READY:{
4060       gst_segment_init (&demux->segment, GST_FORMAT_TIME);
4061       demux->need_newsegment = TRUE;
4062       demux->segment_running = FALSE;
4063       demux->accurate = FALSE;
4064       demux->adapter = gst_adapter_new ();
4065       demux->metadata = gst_caps_new_empty ();
4066       demux->global_metadata = gst_structure_new_empty ("metadata");
4067       demux->data_size = 0;
4068       demux->data_offset = 0;
4069       demux->index_offset = 0;
4070       demux->base_offset = 0;
4071       break;
4072     }
4073     default:
4074       break;
4075   }
4076
4077   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
4078   if (ret == GST_STATE_CHANGE_FAILURE)
4079     return ret;
4080
4081   switch (transition) {
4082     case GST_STATE_CHANGE_PAUSED_TO_READY:
4083     case GST_STATE_CHANGE_READY_TO_NULL:
4084       gst_asf_demux_reset (demux, FALSE);
4085       break;
4086     default:
4087       break;
4088   }
4089
4090   return ret;
4091 }