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