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