mirrors/gstreamer
mirrors/gstreamer
1
1
Fork 0
mirror of https://gitlab.freedesktop.org/gstreamer/gstreamer.git synced 2024-11-14 13:21:28 +00:00
Code Issues 1 Projects Releases Wiki Activity
gstreamer/gst/asfdemux/gstasfdemux.c
Ronald S. Bultje 8ca520d43a gst/asfdemux/gstasfdemux.c: Fix wrong order of reading of optional bytes (#165290). 
Original commit message from CVS:
* gst/asfdemux/gstasfdemux.c: (gst_asf_demux_handle_data):
Fix wrong order of reading of optional bytes (#165290).
2005-02-08 13:38:11 +00:00

1869 lines
54 KiB
C
Raw Blame History

/* GStreamer
* Copyright (C) <1999> Erik Walthinsen <omega@cse.ogi.edu>
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Library General Public
* License as published by the Free Software Foundation; either
* version 2 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Library General Public License for more details.
*
* You should have received a copy of the GNU Library General Public
* License along with this library; if not, write to the
* Free Software Foundation, Inc., 59 Temple Place - Suite 330,
* Boston, MA 02111-1307, USA.
*/
#ifdef HAVE_CONFIG_H
#include "config.h"
#endif
#include <gst/gstutils.h>
#include <gst/riff/riff-media.h>
#include <string.h>
#include "gstasfdemux.h"
#include "asfheaders.h"
static GstStaticPadTemplate gst_asf_demux_sink_template =
GST_STATIC_PAD_TEMPLATE ("sink",
GST_PAD_SINK,
GST_PAD_ALWAYS,
GST_STATIC_CAPS ("video/x-ms-asf")
);
GST_DEBUG_CATEGORY_STATIC (asf_debug);
#define GST_CAT_DEFAULT asf_debug
static void gst_asf_demux_base_init (gpointer g_class);
static void gst_asf_demux_class_init (GstASFDemuxClass * klass);
static void gst_asf_demux_init (GstASFDemux * asf_demux);
static gboolean gst_asf_demux_send_event (GstElement * element,
GstEvent * event);
static void gst_asf_demux_loop (GstElement * element);
static gboolean gst_asf_demux_handle_data (GstASFDemux * asf_demux);
static gboolean gst_asf_demux_process_object (GstASFDemux * asf_demux);
static void gst_asf_demux_get_property (GObject * object,
guint prop_id, GValue * value, GParamSpec * pspec);
static guint32 gst_asf_demux_identify_guid (GstASFDemux * asf_demux,
ASFGuidHash * guids, ASFGuid * guid_raw);
static gboolean gst_asf_demux_process_chunk (GstASFDemux * asf_demux,
asf_packet_info * packet_info, asf_segment_info * segment_info);
static const GstEventMask *gst_asf_demux_get_src_event_mask (GstPad * pad);
static gboolean gst_asf_demux_handle_sink_event (GstASFDemux * asf_demux,
GstEvent * event, guint32 remaining);
static gboolean gst_asf_demux_handle_src_event (GstPad * pad, GstEvent * event);
static const GstFormat *gst_asf_demux_get_src_formats (GstPad * pad);
static const GstQueryType *gst_asf_demux_get_src_query_types (GstPad * pad);
static gboolean gst_asf_demux_handle_src_query (GstPad * pad,
GstQueryType type, GstFormat * format, gint64 * value);
static gboolean gst_asf_demux_add_video_stream (GstASFDemux * asf_demux,
asf_stream_video_format * video_format, guint16 id);
static gboolean gst_asf_demux_add_audio_stream (GstASFDemux * asf_demux,
asf_stream_audio * audio, guint16 id);
static gboolean gst_asf_demux_setup_pad (GstASFDemux * asf_demux,
GstPad * src_pad, GstCaps * caps, guint16 id);
static GstElementStateReturn gst_asf_demux_change_state (GstElement * element);
static GstPadTemplate *videosrctempl, *audiosrctempl;
static GstElementClass *parent_class = NULL;
GType
gst_asf_demux_get_type (void)
{
static GType asf_demux_type = 0;
if (!asf_demux_type) {
static const GTypeInfo asf_demux_info = {
sizeof (GstASFDemuxClass),
gst_asf_demux_base_init,
NULL,
(GClassInitFunc) gst_asf_demux_class_init,
NULL,
NULL,
sizeof (GstASFDemux),
0,
(GInstanceInitFunc) gst_asf_demux_init,
};
asf_demux_type =
g_type_register_static (GST_TYPE_ELEMENT, "GstASFDemux",
&asf_demux_info, 0);
GST_DEBUG_CATEGORY_INIT (asf_debug, "asfdemux", 0, "asf demuxer element");
}
return asf_demux_type;
}
static void
gst_asf_demux_base_init (gpointer g_class)
{
GstElementClass *element_class = GST_ELEMENT_CLASS (g_class);
static GstElementDetails gst_asf_demux_details = {
"ASF Demuxer",
"Codec/Demuxer",
"Demultiplexes ASF Streams",
"Owen Fraser-Green <owen@discobabe.net>"
};
GstCaps *audcaps = gst_riff_create_audio_template_caps (),
*vidcaps = gst_riff_create_video_template_caps ();
audiosrctempl = gst_pad_template_new ("audio_%02d",
GST_PAD_SRC, GST_PAD_SOMETIMES, audcaps);
videosrctempl = gst_pad_template_new ("video_%02d",
GST_PAD_SRC, GST_PAD_SOMETIMES, vidcaps);
gst_element_class_add_pad_template (element_class, audiosrctempl);
gst_element_class_add_pad_template (element_class, videosrctempl);
gst_element_class_add_pad_template (element_class,
gst_static_pad_template_get (&gst_asf_demux_sink_template));
gst_element_class_set_details (element_class, &gst_asf_demux_details);
}
static void
gst_asf_demux_class_init (GstASFDemuxClass * klass)
{
GObjectClass *gobject_class;
GstElementClass *gstelement_class;
gobject_class = (GObjectClass *) klass;
gstelement_class = (GstElementClass *) klass;
parent_class = g_type_class_ref (GST_TYPE_ELEMENT);
gobject_class->get_property = gst_asf_demux_get_property;
gstelement_class->change_state = gst_asf_demux_change_state;
gstelement_class->send_event = gst_asf_demux_send_event;
}
static void
gst_asf_demux_init (GstASFDemux * asf_demux)
{
guint i;
asf_demux->sinkpad =
gst_pad_new_from_template (gst_static_pad_template_get
(&gst_asf_demux_sink_template), "sink");
gst_element_add_pad (GST_ELEMENT (asf_demux), asf_demux->sinkpad);
gst_element_set_loop_function (GST_ELEMENT (asf_demux), gst_asf_demux_loop);
/* We should zero everything to be on the safe side */
for (i = 0; i < GST_ASF_DEMUX_NUM_VIDEO_PADS; i++) {
asf_demux->video_pad[i] = NULL;
asf_demux->video_PTS[i] = 0;
}
for (i = 0; i < GST_ASF_DEMUX_NUM_AUDIO_PADS; i++) {
asf_demux->audio_pad[i] = NULL;
asf_demux->audio_PTS[i] = 0;
}
asf_demux->num_audio_streams = 0;
asf_demux->num_video_streams = 0;
asf_demux->num_streams = 0;
asf_demux->taglist = NULL;
asf_demux->state = GST_ASF_DEMUX_STATE_HEADER;
asf_demux->seek_pending = GST_CLOCK_TIME_NONE;
asf_demux->seek_discont = FALSE;
GST_FLAG_SET (asf_demux, GST_ELEMENT_EVENT_AWARE);
}
static gboolean
gst_asf_demux_send_event (GstElement * element, GstEvent * event)
{
const GList *pads;
pads = gst_element_get_pad_list (element);
while (pads) {
GstPad *pad = GST_PAD (pads->data);
if (GST_PAD_DIRECTION (pad) == GST_PAD_SRC) {
/* we ref the event here as we might have to try again if the event
* failed on this pad */
gst_event_ref (event);
if (gst_asf_demux_handle_src_event (pad, event)) {
gst_event_unref (event);
return TRUE;
}
}
pads = g_list_next (pads);
}
gst_event_unref (event);
return FALSE;
}
static void
gst_asf_demux_loop (GstElement * element)
{
GstASFDemux *asf_demux;
g_return_if_fail (element != NULL);
g_return_if_fail (GST_IS_ASF_DEMUX (element));
asf_demux = GST_ASF_DEMUX (element);
/* this is basically an infinite loop */
switch (asf_demux->state) {
case GST_ASF_DEMUX_STATE_HEADER:
gst_asf_demux_process_object (asf_demux);
break;
case GST_ASF_DEMUX_STATE_DATA:{
guint64 start_off = gst_bytestream_tell (asf_demux->bs), end_off;
/* make sure a full packet is actually available */
if (asf_demux->packet_size != (guint32) - 1)
do {
guint32 remaining;
GstEvent *event;
guint got_bytes;
guint8 *data;
if ((got_bytes = gst_bytestream_peek_bytes (asf_demux->bs, &data,
asf_demux->packet_size)) == asf_demux->packet_size)
break;
gst_bytestream_get_status (asf_demux->bs, &remaining, &event);
if (!gst_asf_demux_handle_sink_event (asf_demux, event, remaining))
return;
} while (1);
/* now handle data */
gst_asf_demux_handle_data (asf_demux);
/* align by packet size */
end_off = gst_bytestream_tell (asf_demux->bs);
if (asf_demux->packet_size != (guint32) - 1 &&
end_off - start_off < asf_demux->packet_size) {
gst_bytestream_flush_fast (asf_demux->bs,
asf_demux->packet_size - (end_off - start_off));
}
break;
}
case GST_ASF_DEMUX_STATE_EOS:
gst_pad_event_default (asf_demux->sinkpad, gst_event_new (GST_EVENT_EOS));
break;
default:
g_assert_not_reached ();
break;
}
}
static guint32
_read_var_length (GstASFDemux * asf_demux, guint8 type, guint32 * rsize)
{
guint32 got_bytes;
guint8 *var;
guint32 ret = 0;
GstByteStream *bs = asf_demux->bs;
if (type == 0) {
return 0;
}
got_bytes = gst_bytestream_peek_bytes (bs, &var, 4);
do {
guint32 remaining;
GstEvent *event;
if ((got_bytes = gst_bytestream_peek_bytes (bs, &var, 4)) == 4)
break;
gst_bytestream_get_status (bs, &remaining, &event);
if (!gst_asf_demux_handle_sink_event (asf_demux, event, remaining))
return FALSE;
} while (1);
switch (type) {
case 1:
ret = (GST_READ_UINT32_LE (var)) & 0xff;
gst_bytestream_flush (bs, 1);
*rsize += 1;
break;
case 2:
ret = (GST_READ_UINT32_LE (var)) & 0xffff;
gst_bytestream_flush (bs, 2);
*rsize += 2;
break;
case 3:
ret = GST_READ_UINT32_LE (var);
gst_bytestream_flush (bs, 4);
*rsize += 4;
break;
}
return ret;
}
#define READ_UINT_BITS_FUNCTION(bits) \
static gboolean \
_read_uint ## bits (GstASFDemux *asf_demux, guint ## bits *ret) \
{ \
GstEvent *event; \
guint32 remaining; \
guint8* data; \
\
g_return_val_if_fail (ret != NULL, FALSE); \
\
do { \
if (gst_bytestream_peek_bytes (asf_demux->bs, &data, bits / 8) == bits / 8) { \
*ret = GST_READ_UINT ## bits ## _LE (data); \
gst_bytestream_flush (asf_demux->bs, bits / 8); \
return TRUE; \
} \
gst_bytestream_get_status (asf_demux->bs, &remaining, &event); \
} while (gst_asf_demux_handle_sink_event (asf_demux, event, remaining)); \
\
return FALSE; \
}
#define GST_READ_UINT8_LE(x) GST_READ_UINT8(x)
READ_UINT_BITS_FUNCTION (8);
READ_UINT_BITS_FUNCTION (16);
READ_UINT_BITS_FUNCTION (32);
READ_UINT_BITS_FUNCTION (64);
static gboolean
_read_guid (GstASFDemux * asf_demux, ASFGuid * guid)
{
return (_read_uint32 (asf_demux, &guid->v1) &&
_read_uint32 (asf_demux, &guid->v2) &&
_read_uint32 (asf_demux, &guid->v3) &&
_read_uint32 (asf_demux, &guid->v4));
}
static gboolean
_read_obj_file (GstASFDemux * asf_demux, asf_obj_file * object)
{
return (_read_guid (asf_demux, &object->file_id) &&
_read_uint64 (asf_demux, &object->file_size) &&
_read_uint64 (asf_demux, &object->creation_time) &&
_read_uint64 (asf_demux, &object->packets_count) &&
_read_uint64 (asf_demux, &object->play_time) &&
_read_uint64 (asf_demux, &object->send_time) &&
_read_uint64 (asf_demux, &object->preroll) &&
_read_uint32 (asf_demux, &object->flags) &&
_read_uint32 (asf_demux, &object->min_pktsize) &&
_read_uint32 (asf_demux, &object->max_pktsize) &&
_read_uint32 (asf_demux, &object->min_bitrate));
}
static gboolean
_read_bitrate_record (GstASFDemux * asf_demux, asf_bitrate_record * record)
{
return (_read_uint16 (asf_demux, &record->stream_id) &&
_read_uint32 (asf_demux, &record->bitrate));
}
static gboolean
_read_obj_comment (GstASFDemux * asf_demux, asf_obj_comment * comment)
{
return (_read_uint16 (asf_demux, &comment->title_length) &&
_read_uint16 (asf_demux, &comment->author_length) &&
_read_uint16 (asf_demux, &comment->copyright_length) &&
_read_uint16 (asf_demux, &comment->description_length) &&
_read_uint16 (asf_demux, &comment->rating_length));
}
static gboolean
_read_obj_header (GstASFDemux * asf_demux, asf_obj_header * header)
{
return (_read_uint32 (asf_demux, &header->num_objects) &&
_read_uint8 (asf_demux, &header->unknown1) &&
_read_uint8 (asf_demux, &header->unknown2));
}
static gboolean
_read_obj_header_ext (GstASFDemux * asf_demux, asf_obj_header_ext * header_ext)
{
return (_read_guid (asf_demux, &header_ext->reserved1) &&
_read_uint16 (asf_demux, &header_ext->reserved2) &&
_read_uint32 (asf_demux, &header_ext->data_size));
}
static gboolean
_read_obj_stream (GstASFDemux * asf_demux, asf_obj_stream * stream)
{
return (_read_guid (asf_demux, &stream->type) &&
_read_guid (asf_demux, &stream->correction) &&
_read_uint64 (asf_demux, &stream->unknown1) &&
_read_uint32 (asf_demux, &stream->type_specific_size) &&
_read_uint32 (asf_demux, &stream->stream_specific_size) &&
_read_uint16 (asf_demux, &stream->id) &&
_read_uint32 (asf_demux, &stream->unknown2));
}
static gboolean
_read_replicated_data (GstASFDemux * asf_demux, asf_replicated_data * rep)
{
return (_read_uint32 (asf_demux, &rep->object_size) &&
_read_uint32 (asf_demux, &rep->frag_timestamp));
}
static gboolean
_read_obj_data (GstASFDemux * asf_demux, asf_obj_data * object)
{
return (_read_guid (asf_demux, &object->file_id) &&
_read_uint64 (asf_demux, &object->packets) &&
_read_uint8 (asf_demux, &object->unknown1) &&
/*_read_uint8 (asf_demux, &object->unknown2) && */
_read_uint8 (asf_demux, &object->correction));
}
static gboolean
_read_obj_data_correction (GstASFDemux * asf_demux,
asf_obj_data_correction * object)
{
return (_read_uint8 (asf_demux, &object->type) &&
_read_uint8 (asf_demux, &object->cycle));
}
static gboolean
_read_obj_data_packet (GstASFDemux * asf_demux, asf_obj_data_packet * object)
{
return (_read_uint8 (asf_demux, &object->flags) &&
_read_uint8 (asf_demux, &object->property));
}
static gboolean
_read_stream_audio (GstASFDemux * asf_demux, asf_stream_audio * audio)
{
/* WAVEFORMATEX Structure */
return (_read_uint16 (asf_demux, &audio->codec_tag) && _read_uint16 (asf_demux, &audio->channels) && _read_uint32 (asf_demux, &audio->sample_rate) && _read_uint32 (asf_demux, &audio->byte_rate) && _read_uint16 (asf_demux, &audio->block_align) && _read_uint16 (asf_demux, &audio->word_size) && _read_uint16 (asf_demux, &audio->size)); /* Codec specific data size */
}
static gboolean
_read_stream_correction (GstASFDemux * asf_demux,
asf_stream_correction * object)
{
return (_read_uint8 (asf_demux, &object->span) &&
_read_uint16 (asf_demux, &object->packet_size) &&
_read_uint16 (asf_demux, &object->chunk_size) &&
_read_uint16 (asf_demux, &object->data_size) &&
_read_uint8 (asf_demux, &object->silence_data));
}
static gboolean
_read_stream_video (GstASFDemux * asf_demux, asf_stream_video * video)
{
return (_read_uint32 (asf_demux, &video->width) &&
_read_uint32 (asf_demux, &video->height) &&
_read_uint8 (asf_demux, &video->unknown) &&
_read_uint16 (asf_demux, &video->size));
}
static gboolean
_read_stream_video_format (GstASFDemux * asf_demux,
asf_stream_video_format * fmt)
{
return (_read_uint32 (asf_demux, &fmt->size) &&
_read_uint32 (asf_demux, &fmt->width) &&
_read_uint32 (asf_demux, &fmt->height) &&
_read_uint16 (asf_demux, &fmt->planes) &&
_read_uint16 (asf_demux, &fmt->depth) &&
_read_uint32 (asf_demux, &fmt->tag) &&
_read_uint32 (asf_demux, &fmt->image_size) &&
_read_uint32 (asf_demux, &fmt->xpels_meter) &&
_read_uint32 (asf_demux, &fmt->ypels_meter) &&
_read_uint32 (asf_demux, &fmt->num_colors) &&
_read_uint32 (asf_demux, &fmt->imp_colors));
}
static gboolean
gst_asf_demux_process_file (GstASFDemux * asf_demux, guint64 * obj_size)
{
asf_obj_file object;
/* Get the rest of the header's header */
_read_obj_file (asf_demux, &object);
if (object.min_pktsize == object.max_pktsize)
asf_demux->packet_size = object.max_pktsize;
else {
asf_demux->packet_size = (guint32) - 1;
GST_WARNING_OBJECT (asf_demux, "Non-const packet size, seeking disabled");
}
asf_demux->play_time = (guint64) object.play_time * (GST_SECOND / 10000000);
asf_demux->preroll = object.preroll;
GST_INFO ("Object is a file with %" G_GUINT64_FORMAT " data packets",
object.packets_count);
return TRUE;
}
static gboolean
gst_asf_demux_process_bitrate_props_object (GstASFDemux * asf_demux,
guint64 * obj_size)
{
guint16 num_streams;
guint8 stream_id;
guint16 i;
asf_bitrate_record bitrate_record;
if (!_read_uint16 (asf_demux, &num_streams))
return FALSE;
GST_INFO ("Object is a bitrate properties object with %u streams.",
num_streams);
for (i = 0; i < num_streams; i++) {
_read_bitrate_record (asf_demux, &bitrate_record);
stream_id = bitrate_record.stream_id & 0x7f;
asf_demux->bitrate[stream_id] = bitrate_record.bitrate;
}
return TRUE;
}
static void
gst_asf_demux_commit_taglist (GstASFDemux * asf_demux, GstTagList * taglist)
{
GstElement *element = GST_ELEMENT (asf_demux);
GstEvent *event;
const GList *padlist;
/* emit signal */
gst_element_found_tags (element, taglist);
/* save internally */
if (!asf_demux->taglist)
asf_demux->taglist = gst_tag_list_copy (taglist);
else {
GstTagList *t;
t = gst_tag_list_merge (asf_demux->taglist, taglist, GST_TAG_MERGE_APPEND);
gst_tag_list_free (asf_demux->taglist);
asf_demux->taglist = t;
}
/* pads */
event = gst_event_new_tag (taglist);
for (padlist = gst_element_get_pad_list (element);
padlist != NULL; padlist = padlist->next) {
if (GST_PAD_IS_SRC (padlist->data) && GST_PAD_IS_USABLE (padlist->data)) {
gst_event_ref (event);
gst_pad_push (GST_PAD (padlist->data), GST_DATA (event));
}
}
gst_event_unref (event);
}
/* Content Description Object */
static gboolean
gst_asf_demux_process_comment (GstASFDemux * asf_demux, guint64 * obj_size)
{
asf_obj_comment object;
GstByteStream *bs = asf_demux->bs;
gchar *utf8_comments[5] = { NULL, NULL, NULL, NULL, NULL };
guchar *data;
const gchar *tags[5] = { GST_TAG_TITLE, GST_TAG_ARTIST, GST_TAG_COPYRIGHT,
GST_TAG_COMMENT, NULL /* ? */
};
guint16 *lengths = (guint16 *) & object;
gint i;
gsize in, out;
GstTagList *taglist;
GValue value = { 0 };
gboolean have_tags = FALSE;
GST_INFO ("Object is a comment.");
/* Get the rest of the comment's header */
_read_obj_comment (asf_demux, &object);
GST_DEBUG
("Comment lengths: title=%d author=%d copyright=%d description=%d rating=%d",
object.title_length, object.author_length, object.copyright_length,
object.description_length, object.rating_length);
for (i = 0; i < 5; i++) {
/* might be just '/0', '/0'... */
if (lengths[i] > 2 && lengths[i] % 2 == 0) {
if (gst_bytestream_peek_bytes (bs, &data, lengths[i]) != lengths[i])
goto fail;
/* convert to UTF-8 */
utf8_comments[i] = g_convert (data, lengths[i],
"UTF-8", "UTF-16LE", &in, &out, NULL);
gst_bytestream_flush_fast (bs, lengths[i]);
} else {
if (lengths[i] > 0 && !gst_bytestream_flush (bs, lengths[i]))
goto fail;
}
}
/* parse metadata into taglist */
taglist = gst_tag_list_new ();
g_value_init (&value, G_TYPE_STRING);
for (i = 0; i < 5; i++) {
if (utf8_comments[i] && tags[i]) {
have_tags = TRUE;
g_value_set_string (&value, utf8_comments[i]);
gst_tag_list_add_values (taglist, GST_TAG_MERGE_APPEND,
tags[i], &value, NULL);
g_free (utf8_comments[i]);
}
}
g_value_unset (&value);
if (have_tags) {
gst_asf_demux_commit_taglist (asf_demux, taglist);
} else {
gst_tag_list_free (taglist);
}
return TRUE;
fail:
for (i = 0; i < 5; i++)
g_free (utf8_comments[i]);
return FALSE;
}
/* Extended Content Description Object */
static gboolean
gst_asf_demux_process_ext_content_desc (GstASFDemux * asf_demux,
guint64 * obj_size)
{
GstByteStream *bs = asf_demux->bs;
guint16 blockcount, i;
/*
Other known (and unused) 'text/unicode' metadata available :
WM/Lyrics =
WM/MediaPrimaryClassID = {D1607DBC-E323-4BE2-86A1-48A42A28441E}
WMFSDKVersion = 9.00.00.2980
WMFSDKNeeded = 0.0.0.0000
WM/UniqueFileIdentifier = AMGa_id=R 15334;AMGp_id=P 5149;AMGt_id=T 2324984
WM/Publisher = 4AD
WM/Provider = AMG
WM/ProviderRating = 8
WM/ProviderStyle = Rock (similar to WM/Genre)
WM/GenreID (similar to WM/Genre)
Other known (and unused) 'non-text' metadata available :
WM/Track (same as WM/TrackNumber but starts at 0)
WM/EncodingTime
WM/MCDI
IsVBR
*/
const guchar *tags[] = { GST_TAG_GENRE, GST_TAG_ALBUM, GST_TAG_ARTIST, GST_TAG_TRACK_NUMBER, GST_TAG_DATE, NULL }; // GST_TAG_COMPOSER
const guchar *tags_label[] = { "WM/Genre", "WM/AlbumTitle", "WM/AlbumArtist", "WM/TrackNumber", "WM/Year", NULL }; // "WM/Composer"
GstTagList *taglist;
GValue tag_value = { 0 };
gboolean have_tags = FALSE;
guint8 *name = NULL;
GST_INFO ("Object is an extended content description.");
taglist = gst_tag_list_new ();
/* Content Descriptors Count */
if (!_read_uint16 (asf_demux, &blockcount))
goto fail;
for (i = 1; i <= blockcount; i++) {
guint16 name_length;
guint16 datatype;
guint16 value_length;
guint8 *tmpname;
guint8 *tmpvalue, *value;
/* Descriptor Name Length */
if (!_read_uint16 (asf_demux, &name_length))
goto fail;
/* Descriptor Name */
name = g_malloc (name_length);
if (gst_bytestream_peek_bytes (bs, &tmpname, name_length) != name_length)
goto fail;
name = g_memdup (tmpname, name_length);
gst_bytestream_flush_fast (bs, name_length);
/* Descriptor Value Data Type */
if (!_read_uint16 (asf_demux, &datatype))
goto fail;
/* Descriptor Value Length */
if (!_read_uint16 (asf_demux, &value_length))
goto fail;
/* Descriptor Value */
if (gst_bytestream_peek_bytes (bs, &tmpvalue, value_length) != value_length)
goto fail;
value = g_memdup (tmpvalue, value_length);
gst_bytestream_flush_fast (bs, value_length);
/* 0000 = Unicode String OR 0003 = DWORD */
if ((datatype == 0) || (datatype == 3)) {
gsize in, out;
guchar tag;
guint16 j = 0;
/* convert to UTF-8 */
name =
g_convert (name, name_length, "UTF-8", "UTF-16LE", &in, &out, NULL);
name[out] = 0;
while (tags_label[j] != NULL) {
if (!strcmp (name, tags_label[j])) {
tag = j;
/* 0000 = UTF-16 String */
if (datatype == 0) {
/* convert to UTF-8 */
value =
g_convert (value, value_length, "UTF-8", "UTF-16LE", &in, &out,
NULL);
value[out] = 0;
/* get rid of tags with empty value */
if (strlen (value)) {
g_value_init (&tag_value, G_TYPE_STRING);
g_value_set_string (&tag_value, value);
}
}
/* 0003 = DWORD */
if (datatype == 3) {
g_value_init (&tag_value, G_TYPE_INT);
g_value_set_int (&tag_value, GUINT32_FROM_LE ((guint32) * value));
}
if (G_IS_VALUE (&tag_value)) {
gst_tag_list_add_values (taglist, GST_TAG_MERGE_APPEND, tags[tag],
&tag_value, NULL);
g_value_unset (&tag_value);
have_tags = TRUE;
}
}
j++;
}
}
g_free (name);
g_free (value);
}
if (have_tags) {
gst_asf_demux_commit_taglist (asf_demux, taglist);
} else {
gst_tag_list_free (taglist);
}
return TRUE;
fail:
if (name) {
g_free (name);
};
return FALSE;
}
static gboolean
gst_asf_demux_process_header (GstASFDemux * asf_demux, guint64 * obj_size)
{
asf_obj_header object;
guint32 i;
/* Get the rest of the header's header */
_read_obj_header (asf_demux, &object);
GST_INFO ("Object is a header with %u parts", object.num_objects);
/* Loop through the header's objects, processing those */
for (i = 0; i < object.num_objects; i++) {
if (!gst_asf_demux_process_object (asf_demux)) {
return FALSE;
}
}
return TRUE;
}
static gboolean
gst_asf_demux_process_header_ext (GstASFDemux * asf_demux, guint64 * obj_size)
{
asf_obj_header_ext object;
guint64 original_offset;
/* Get the rest of the header's header */
_read_obj_header_ext (asf_demux, &object);
GST_INFO ("Object is an extended header with a size of %u bytes",
object.data_size);
original_offset = asf_demux->bs->offset;
while ((asf_demux->bs->offset - original_offset) < object.data_size) {
if (!gst_asf_demux_process_object (asf_demux))
return FALSE;
}
return TRUE;
}
static gboolean
gst_asf_demux_process_segment (GstASFDemux * asf_demux,
asf_packet_info * packet_info)
{
guint8 byte;
gboolean key_frame;
guint32 replic_size;
guint8 time_delta;
guint32 time_start;
guint32 frag_size;
guint32 rsize;
asf_segment_info segment_info;
_read_uint8 (asf_demux, &byte);
rsize = 1;
segment_info.stream_number = byte & 0x7f;
key_frame = (byte & 0x80) >> 7;
GST_INFO ("Processing segment for stream %u", segment_info.stream_number);
segment_info.sequence =
_read_var_length (asf_demux, packet_info->seqtype, &rsize);
segment_info.frag_offset =
_read_var_length (asf_demux, packet_info->fragoffsettype, &rsize);
replic_size =
_read_var_length (asf_demux, packet_info->replicsizetype, &rsize);
GST_DEBUG ("sequence = %x, frag_offset = %x, replic_size = %x",
segment_info.sequence, segment_info.frag_offset, replic_size);
if (replic_size > 1) {
asf_replicated_data replicated_data_header;
segment_info.compressed = FALSE;
/* It's uncompressed with replic data */
if (replic_size < 8) {
GST_ELEMENT_ERROR (asf_demux, STREAM, DEMUX, (NULL),
("The payload has replicated data but the size is less than 8"));
return FALSE;
}
_read_replicated_data (asf_demux, &replicated_data_header);
segment_info.frag_timestamp = replicated_data_header.frag_timestamp;
segment_info.segment_size = replicated_data_header.object_size;
if (replic_size > 8) {
gst_bytestream_flush (asf_demux->bs, replic_size - 8);
}
rsize += replic_size;
} else {
if (replic_size == 1) {
/* It's compressed */
segment_info.compressed = TRUE;
_read_uint8 (asf_demux, &time_delta);
rsize++;
GST_DEBUG ("time_delta %u", time_delta);
} else {
segment_info.compressed = FALSE;
}
time_start = segment_info.frag_offset;
segment_info.frag_offset = 0;
segment_info.frag_timestamp = asf_demux->timestamp;
}
GST_DEBUG ("multiple = %u compressed = %u", packet_info->multiple,
segment_info.compressed);
if (packet_info->multiple) {
frag_size = _read_var_length (asf_demux, packet_info->segsizetype, &rsize);
} else {
frag_size = packet_info->size_left - rsize;
}
packet_info->size_left -= rsize;
GST_DEBUG ("size left = %u frag size = %u rsize = %u", packet_info->size_left,
frag_size, rsize);
if (segment_info.compressed) {
while (frag_size > 0) {
_read_uint8 (asf_demux, &byte);
packet_info->size_left--;
segment_info.chunk_size = byte;
segment_info.segment_size = segment_info.chunk_size;
if (segment_info.chunk_size > packet_info->size_left) {
GST_ELEMENT_ERROR (asf_demux, STREAM, DEMUX, (NULL),
("Payload chunk overruns packet size."));
return FALSE;
}
if (!gst_asf_demux_process_chunk (asf_demux, packet_info, &segment_info))
return FALSE;
if (segment_info.chunk_size < frag_size)
frag_size -= segment_info.chunk_size + 1;
else {
GST_ELEMENT_ERROR (asf_demux, STREAM, TOO_LAZY,
("Invalid data in stream"),
("Invalid fragment size indicator in segment"));
return FALSE;
}
}
} else {
segment_info.chunk_size = frag_size;
if (!gst_asf_demux_process_chunk (asf_demux, packet_info, &segment_info))
return FALSE;
}
return TRUE;
}
static gboolean
gst_asf_demux_process_data (GstASFDemux * asf_demux, guint64 * obj_size)
{
asf_obj_data object;
/* Get the rest of the header */
_read_obj_data (asf_demux, &object);
GST_INFO ("Object is data with %" G_GUINT64_FORMAT " packets",
object.packets);
gst_element_no_more_pads (GST_ELEMENT (asf_demux));
asf_demux->state = GST_ASF_DEMUX_STATE_DATA;
asf_demux->packet = 0;
asf_demux->num_packets = object.packets;
asf_demux->data_size = *obj_size;
asf_demux->data_offset = gst_bytestream_tell (asf_demux->bs);
return TRUE;
}
static gboolean
gst_asf_demux_handle_data (GstASFDemux * asf_demux)
{
asf_obj_data_packet packet_properties_object;
gboolean correction;
guint8 buf;
guint32 sequence;
guint32 packet_length;
guint16 duration;
guint8 segment;
guint8 segments;
guint8 flags;
guint8 property;
asf_packet_info packet_info;
guint32 rsize;
gint n;
/* handle seek, if any */
if (GST_CLOCK_TIME_IS_VALID (asf_demux->seek_pending) &&
asf_demux->packet_size != 0) {
guint64 packet_seek = asf_demux->num_packets *
asf_demux->seek_pending / asf_demux->play_time;
if (packet_seek > asf_demux->num_packets)
packet_seek = asf_demux->num_packets;
if (gst_bytestream_seek (asf_demux->bs, packet_seek *
asf_demux->packet_size + asf_demux->data_offset,
GST_SEEK_METHOD_SET)) {
asf_demux->packet = packet_seek;
for (n = 0; n < asf_demux->num_streams; n++) {
if (asf_demux->stream[n].frag_offset > 0) {
gst_buffer_unref (asf_demux->stream[n].payload);
asf_demux->stream[n].frag_offset = 0;
}
}
asf_demux->seek_discont = TRUE;
}
asf_demux->seek_pending = GST_CLOCK_TIME_NONE;
}
GST_INFO ("Process packet");
if (asf_demux->packet++ >= asf_demux->num_packets) {
GstEvent *event;
guint32 remaining;
gst_bytestream_flush (asf_demux->bs, 0xFFFFFF);
gst_bytestream_get_status (asf_demux->bs, &remaining, &event);
if (!event || GST_EVENT_TYPE (event) != GST_EVENT_EOS)
g_warning ("No EOS");
if (event)
gst_event_unref (event);
return gst_asf_demux_handle_sink_event (asf_demux,
gst_event_new (GST_EVENT_EOS), 0);
}
_read_uint8 (asf_demux, &buf);
rsize = 1;
if (buf & 0x80) {
asf_obj_data_correction correction_object;
/* Uses error correction */
correction = TRUE;
GST_DEBUG ("Data has error correction (%x)", buf);
_read_obj_data_correction (asf_demux, &correction_object);
rsize += 2;
}
/* Read the packet flags */
_read_obj_data_packet (asf_demux, &packet_properties_object);
rsize += 2;
flags = packet_properties_object.flags;
property = packet_properties_object.property;
packet_length = _read_var_length (asf_demux, (flags >> 5) & 0x03, &rsize);
if (packet_length == 0)
packet_length = asf_demux->packet_size;
packet_info.multiple = flags & 0x01;
sequence = _read_var_length (asf_demux, (flags >> 1) & 0x03, &rsize);
packet_info.padsize =
_read_var_length (asf_demux, (flags >> 3) & 0x03, &rsize);
GST_DEBUG ("Multiple = %u, Sequence = %u, Padsize = %u, Packet length = %u",
packet_info.multiple, sequence, packet_info.padsize, packet_length);
/* Read the property flags */
packet_info.replicsizetype = property & 0x03;
packet_info.fragoffsettype = (property >> 2) & 0x03;
packet_info.seqtype = (property >> 4) & 0x03;
_read_uint32 (asf_demux, &asf_demux->timestamp);
_read_uint16 (asf_demux, &duration);
rsize += 6;
GST_DEBUG ("Timestamp = %x, Duration = %x", asf_demux->timestamp, duration);
if (packet_info.multiple) {
/* There are multiple payloads */
_read_uint8 (asf_demux, &buf);
rsize++;
packet_info.segsizetype = (buf >> 6) & 0x03;
segments = buf & 0x3f;
} else {
packet_info.segsizetype = 2;
segments = 1;
}
packet_info.size_left = packet_length - packet_info.padsize - rsize;
GST_DEBUG ("rsize: %u size left: %u", rsize, packet_info.size_left);
for (segment = 0; segment < segments; segment++) {
if (!gst_asf_demux_process_segment (asf_demux, &packet_info))
return FALSE;
}
/* Skip the padding */
if (packet_info.padsize > 0)
gst_bytestream_flush (asf_demux->bs, packet_info.padsize);
GST_DEBUG ("Remaining size left: %u", packet_info.size_left);
if (packet_info.size_left > 0)
gst_bytestream_flush (asf_demux->bs, packet_info.size_left);
return TRUE;
}
static gboolean
gst_asf_demux_process_stream (GstASFDemux * asf_demux, guint64 * obj_size)
{
asf_obj_stream object;
guint32 stream_id;
guint32 correction;
asf_stream_audio audio_object;
asf_stream_correction correction_object;
asf_stream_video video_object;
asf_stream_video_format video_format_object;
guint16 size;
guint16 id;
/* Get the rest of the header's header */
_read_obj_stream (asf_demux, &object);
/* Identify the stream type */
stream_id =
gst_asf_demux_identify_guid (asf_demux, asf_stream_guids, &(object.type));
correction =
gst_asf_demux_identify_guid (asf_demux, asf_correction_guids,
&(object.correction));
id = object.id;
switch (stream_id) {
case ASF_STREAM_AUDIO:
_read_stream_audio (asf_demux, &audio_object);
size = audio_object.size;
GST_INFO ("Object is an audio stream with %u bytes of additional data.",
size);
if (!gst_asf_demux_add_audio_stream (asf_demux, &audio_object, id))
return FALSE;
switch (correction) {
case ASF_CORRECTION_ON:
GST_INFO ("Using error correction");
_read_stream_correction (asf_demux, &correction_object);
asf_demux->span = correction_object.span;
GST_DEBUG ("Descrambling: ps:%d cs:%d ds:%d s:%d sd:%d",
correction_object.packet_size, correction_object.chunk_size,
correction_object.data_size, (guint) correction_object.span,
(guint) correction_object.silence_data);
if (asf_demux->span > 1) {
if (!correction_object.chunk_size
|| ((correction_object.packet_size /
correction_object.chunk_size) <= 1))
/* Disable descrambling */
asf_demux->span = 0;
} else {
/* Descambling is enabled */
asf_demux->ds_packet_size = correction_object.packet_size;
asf_demux->ds_chunk_size = correction_object.chunk_size;
}
/* Now skip the rest of the silence data */
if (correction_object.data_size > 1)
gst_bytestream_flush (asf_demux->bs,
correction_object.data_size - 1);
break;
case ASF_CORRECTION_OFF:
GST_INFO ("Error correction off");
gst_bytestream_flush (asf_demux->bs, object.stream_specific_size);
break;
default:
GST_ELEMENT_ERROR (asf_demux, STREAM, DEMUX, (NULL),
("Audio stream using unknown error correction"));
return FALSE;
}
break;
case ASF_STREAM_VIDEO:
_read_stream_video (asf_demux, &video_object);
size = video_object.size - 40; /* Byte order gets offset by single byte */
GST_INFO ("Object is a video stream with %u bytes of additional data.",
size);
_read_stream_video_format (asf_demux, &video_format_object);
if (!gst_asf_demux_add_video_stream (asf_demux, &video_format_object, id))
return FALSE;
break;
default:
GST_ELEMENT_ERROR (asf_demux, STREAM, WRONG_TYPE, (NULL),
("unknown asf stream (id %08x)", (guint) stream_id));
return FALSE;
}
return TRUE;
}
static gboolean
gst_asf_demux_skip_object (GstASFDemux * asf_demux, guint64 * obj_size)
{
GstByteStream *bs = asf_demux->bs;
GST_INFO ("Skipping object...");
gst_bytestream_flush (bs, *obj_size - 24);
return TRUE;
}
static inline gboolean
_read_object_header (GstASFDemux * asf_demux, guint32 * obj_id,
guint64 * obj_size)
{
ASFGuid guid;
/* First get the GUID */
if (!_read_guid (asf_demux, &guid))
return FALSE;
*obj_id = gst_asf_demux_identify_guid (asf_demux, asf_object_guids, &guid);
if (!_read_uint64 (asf_demux, obj_size))
return FALSE;
if (*obj_id == ASF_OBJ_UNDEFINED) {
GST_WARNING_OBJECT (asf_demux,
"Could not identify object (0x%08x/0x%08x/0x%08x/0x%08x) with size=%llu",
guid.v1, guid.v2, guid.v3, guid.v4, *obj_size);
return TRUE;
}
return TRUE;
}
static gboolean
gst_asf_demux_process_object (GstASFDemux * asf_demux)
{
guint32 obj_id;
guint64 obj_size;
if (!_read_object_header (asf_demux, &obj_id, &obj_size)) {
GST_DEBUG (" ***** Error reading object at filepos %" G_GUINT64_FORMAT
" (EOS?)\n", /**filepos*/ gst_bytestream_tell (asf_demux->bs));
gst_asf_demux_handle_sink_event (asf_demux, gst_event_new (GST_EVENT_EOS),
0);
return FALSE;
}
GST_INFO ("Found object %u with size %" G_GUINT64_FORMAT, obj_id, obj_size);
switch (obj_id) {
case ASF_OBJ_STREAM:
return gst_asf_demux_process_stream (asf_demux, &obj_size);
case ASF_OBJ_DATA:
gst_asf_demux_process_data (asf_demux, &obj_size);
/* This is the last object */
return FALSE;
case ASF_OBJ_FILE:
return gst_asf_demux_process_file (asf_demux, &obj_size);
case ASF_OBJ_HEADER:
return gst_asf_demux_process_header (asf_demux, &obj_size);
case ASF_OBJ_COMMENT:
return gst_asf_demux_process_comment (asf_demux, &obj_size);
case ASF_OBJ_HEAD1:
return gst_asf_demux_process_header_ext (asf_demux, &obj_size);
case ASF_OBJ_BITRATE_PROPS:
return gst_asf_demux_process_bitrate_props_object (asf_demux, &obj_size);
case ASF_OBJ_EXT_CONTENT_DESC:
return gst_asf_demux_process_ext_content_desc (asf_demux, &obj_size);
case ASF_OBJ_CONCEAL_NONE:
case ASF_OBJ_HEAD2:
case ASF_OBJ_UNDEFINED:
case ASF_OBJ_CODEC_COMMENT:
case ASF_OBJ_INDEX:
case ASF_OBJ_PADDING:
case ASF_OBJ_BITRATE_MUTEX:
case ASF_OBJ_LANGUAGE_LIST:
case ASF_OBJ_METADATA_OBJECT:
case ASF_OBJ_EXTENDED_STREAM_PROPS:
default:
/* unknown/unhandled object read, just ignore it, we hate fatal errors */
return gst_asf_demux_skip_object (asf_demux, &obj_size);
}
return TRUE;
}
static asf_stream_context *
gst_asf_demux_get_stream (GstASFDemux * asf_demux, guint16 id)
{
guint8 i;
asf_stream_context *stream;
for (i = 0; i < asf_demux->num_streams; i++) {
stream = &asf_demux->stream[i];
if (stream->id == id) {
/* We've found the one with the matching id */
return &asf_demux->stream[i];
}
}
/* Base case if we haven't found one at all */
GST_WARNING_OBJECT (asf_demux,
"Segment found for undefined stream: (%d)", id);
return NULL;
}
static inline void
gst_asf_demux_descramble_segment (GstASFDemux * asf_demux,
asf_segment_info * segment_info, asf_stream_context * stream)
{
GstBuffer *scrambled_buffer;
GstBuffer *descrambled_buffer;
GstBuffer *sub_buffer;
guint offset;
guint off;
guint row;
guint col;
guint idx;
/* descrambled_buffer is initialised in the first iteration */
descrambled_buffer = NULL;
scrambled_buffer = stream->payload;
offset = 0;
for (offset = 0; offset < segment_info->segment_size;
offset += asf_demux->ds_chunk_size) {
off = offset / asf_demux->ds_chunk_size;
row = off / asf_demux->span;
col = off % asf_demux->span;
idx = row + col * asf_demux->ds_packet_size / asf_demux->ds_chunk_size;
sub_buffer =
gst_buffer_create_sub (scrambled_buffer, idx * asf_demux->ds_chunk_size,
asf_demux->ds_chunk_size);
if (!offset) {
descrambled_buffer = sub_buffer;
} else {
GstBuffer *newbuf;
newbuf = gst_buffer_merge (descrambled_buffer, sub_buffer);
gst_buffer_unref (sub_buffer);
gst_buffer_unref (descrambled_buffer);
descrambled_buffer = newbuf;
}
}
stream->payload = descrambled_buffer;
gst_buffer_unref (scrambled_buffer);
}
static gboolean
gst_asf_demux_process_chunk (GstASFDemux * asf_demux,
asf_packet_info * packet_info, asf_segment_info * segment_info)
{
asf_stream_context *stream;
guint32 got_bytes;
GstByteStream *bs = asf_demux->bs;
GstBuffer *buffer;
if (!(stream =
gst_asf_demux_get_stream (asf_demux, segment_info->stream_number)))
goto done;
GST_DEBUG ("Processing chunk of size %u (fo = %d)", segment_info->chunk_size,
stream->frag_offset);
if (segment_info->frag_offset == 0) {
/* new packet */
stream->sequence = segment_info->sequence;
asf_demux->pts = segment_info->frag_timestamp - asf_demux->preroll;
got_bytes = gst_bytestream_peek (bs, &buffer, segment_info->chunk_size);
if (got_bytes == 0)
goto done;
GST_DEBUG ("BUFFER: Copied stream to buffer (%p - %d)", buffer,
GST_BUFFER_REFCOUNT_VALUE (buffer));
stream->payload = buffer;
} else {
GST_DEBUG
("segment_info->sequence = %d, stream->sequence = %d, segment_info->frag_offset = %d, stream->frag_offset = %d",
segment_info->sequence, stream->sequence, segment_info->frag_offset,
stream->frag_offset);
if (segment_info->sequence == stream->sequence &&
segment_info->frag_offset == stream->frag_offset) {
GstBuffer *new_buffer;
/* continuing packet */
GST_INFO ("A continuation packet");
got_bytes = gst_bytestream_peek (bs, &buffer, segment_info->chunk_size);
if (got_bytes == 0)
goto done;
GST_DEBUG ("Copied stream to buffer (%p - %d)", buffer,
GST_BUFFER_REFCOUNT_VALUE (buffer));
new_buffer = gst_buffer_merge (stream->payload, buffer);
GST_DEBUG
("BUFFER: Merged new_buffer (%p - %d) from stream->payload(%p - %d) and buffer (%p - %d)",
new_buffer, GST_BUFFER_REFCOUNT_VALUE (new_buffer), stream->payload,
GST_BUFFER_REFCOUNT_VALUE (stream->payload), buffer,
GST_BUFFER_REFCOUNT_VALUE (buffer));
gst_buffer_unref (stream->payload);
gst_buffer_unref (buffer);
stream->payload = new_buffer;
} else {
/* cannot continue current packet: free it */
if (stream->frag_offset != 0) {
/* cannot create new packet */
GST_DEBUG ("BUFFER: Freeing stream->payload (%p)", stream->payload);
gst_buffer_unref (stream->payload);
gst_bytestream_flush (bs, segment_info->chunk_size);
packet_info->size_left -= segment_info->chunk_size;
stream->frag_offset = 0;
}
asf_demux->pts = segment_info->frag_timestamp - asf_demux->preroll;
goto done;
//return TRUE;
//} else {
/* create new packet */
//stream->sequence = segment_info->sequence;
//}
}
}
stream->frag_offset += segment_info->chunk_size;
GST_DEBUG ("frag_offset = %d segment_size = %d ", stream->frag_offset,
segment_info->segment_size);
if (stream->frag_offset < segment_info->segment_size) {
/* We don't have the whole packet yet */
} else if (got_bytes < segment_info->chunk_size) {
/* We didn't get the entire chunk, so don't push it down the pipeline. */
gst_buffer_unref (stream->payload);
stream->payload = NULL;
stream->frag_offset = 0;
} else {
/* We have the whole packet now so we should push the packet to
the src pad now. First though we should check if we need to do
descrambling */
if (asf_demux->span > 1) {
gst_asf_demux_descramble_segment (asf_demux, segment_info, stream);
}
if (GST_PAD_IS_USABLE (stream->pad)) {
GST_DEBUG ("New buffer is at: %p size: %u",
GST_BUFFER_DATA (stream->payload), GST_BUFFER_SIZE (stream->payload));
GST_BUFFER_TIMESTAMP (stream->payload) =
(GST_SECOND / 1000) * asf_demux->pts;
/*!!! Should handle flush events here? */
GST_DEBUG ("Sending stream %d of size %d", stream->id,
segment_info->chunk_size);
GST_INFO ("Pushing pad");
if (asf_demux->seek_discont) {
if (asf_demux->seek_flush) {
gst_pad_event_default (asf_demux->sinkpad,
gst_event_new (GST_EVENT_FLUSH));
}
gst_pad_event_default (asf_demux->sinkpad,
gst_event_new_discontinuous (FALSE, GST_FORMAT_TIME,
GST_BUFFER_TIMESTAMP (stream->payload), GST_FORMAT_UNDEFINED));
asf_demux->seek_discont = FALSE;
}
gst_pad_push (stream->pad, GST_DATA (stream->payload));
} else {
gst_buffer_unref (stream->payload);
stream->payload = NULL;
}
stream->frag_offset = 0;
}
done:
gst_bytestream_flush (bs, segment_info->chunk_size);
packet_info->size_left -= segment_info->chunk_size;
return TRUE;
}
/*
* Event stuff
*/
static gboolean
gst_asf_demux_handle_sink_event (GstASFDemux * asf_demux,
GstEvent * event, guint32 remaining)
{
gboolean ret = TRUE;
if (!event) {
GST_ELEMENT_ERROR (asf_demux, RESOURCE, READ, (NULL), (NULL));
return FALSE;
}
switch (GST_EVENT_TYPE (event)) {
case GST_EVENT_EOS:
gst_pad_event_default (asf_demux->sinkpad, event);
return FALSE;
case GST_EVENT_DISCONTINUOUS:
{
gint i;
GstEvent *discont;
for (i = 0; i < asf_demux->num_streams; i++) {
asf_stream_context *stream = &asf_demux->stream[i];
if (GST_PAD_IS_USABLE (stream->pad)) {
GST_DEBUG ("sending discont on stream %d with %" GST_TIME_FORMAT
" + %" GST_TIME_FORMAT " = %" GST_TIME_FORMAT,
i, GST_TIME_ARGS (asf_demux->last_seek),
GST_TIME_ARGS (stream->delay),
GST_TIME_ARGS (asf_demux->last_seek + stream->delay));
discont =
gst_event_new_discontinuous (FALSE, GST_FORMAT_TIME,
asf_demux->last_seek + stream->delay, NULL);
gst_pad_push (stream->pad, GST_DATA (discont));
}
}
break;
}
case GST_EVENT_FLUSH:
GST_WARNING_OBJECT (asf_demux, "flush event");
break;
default:
GST_WARNING_OBJECT (asf_demux, "unhandled event %d",
GST_EVENT_TYPE (event));
ret = FALSE;
break;
}
gst_event_unref (event);
return ret;
}
static gboolean
gst_asf_demux_handle_src_event (GstPad * pad, GstEvent * event)
{
gboolean res = TRUE;
GstASFDemux *asf_demux = GST_ASF_DEMUX (gst_pad_get_parent (pad));
GST_DEBUG ("asfdemux: handle_src_event");
switch (GST_EVENT_TYPE (event)) {
case GST_EVENT_SEEK:
switch (GST_EVENT_SEEK_FORMAT (event)) {
case GST_FORMAT_BYTES:
case GST_FORMAT_DEFAULT:
res = FALSE;
break;
case GST_FORMAT_TIME:
asf_demux->seek_pending = GST_EVENT_SEEK_OFFSET (event);
asf_demux->seek_flush = (GST_EVENT_SEEK_FLAGS (event) &
GST_SEEK_FLAG_FLUSH) ? TRUE : FALSE;
break;
default:
res = FALSE;
break;
}
break;
default:
res = FALSE;
break;
}
return res;
}
static const GstEventMask *
gst_asf_demux_get_src_event_mask (GstPad * pad)
{
static const GstEventMask masks[] = {
{GST_EVENT_SEEK, GST_SEEK_METHOD_SET | GST_SEEK_FLAG_KEY_UNIT},
{0,}
};
return masks;
}
static const GstFormat *
gst_asf_demux_get_src_formats (GstPad * pad)
{
static const GstFormat formats[] = {
GST_FORMAT_TIME,
0
};
return formats;
}
static const GstQueryType *
gst_asf_demux_get_src_query_types (GstPad * pad)
{
static const GstQueryType types[] = {
GST_QUERY_TOTAL,
GST_QUERY_POSITION,
0
};
return types;
}
static gboolean
gst_asf_demux_handle_src_query (GstPad * pad,
GstQueryType type, GstFormat * format, gint64 * value)
{
GstASFDemux *asf_demux;
gboolean res = TRUE;
asf_demux = GST_ASF_DEMUX (gst_pad_get_parent (pad));
switch (type) {
case GST_QUERY_TOTAL:
switch (*format) {
case GST_FORMAT_DEFAULT:
*format = GST_FORMAT_TIME;
/* fall through */
case GST_FORMAT_TIME:
*value = asf_demux->play_time;
break;
default:
res = FALSE;
break;
}
break;
case GST_QUERY_POSITION:
switch (*format) {
case GST_FORMAT_DEFAULT:
*format = GST_FORMAT_TIME;
/* fall through */
case GST_FORMAT_TIME:
*value = (GST_SECOND / 1000) * asf_demux->timestamp;
break;
default:
res = FALSE;
break;
}
break;
default:
res = FALSE;
break;
}
return res;
}
static void
gst_asf_demux_get_property (GObject * object,
guint prop_id, GValue * value, GParamSpec * pspec)
{
GstASFDemux *src;
g_return_if_fail (GST_IS_ASF_DEMUX (object));
src = GST_ASF_DEMUX (object);
switch (prop_id) {
default:
break;
}
}
static GstElementStateReturn
gst_asf_demux_change_state (GstElement * element)
{
GstASFDemux *asf_demux = GST_ASF_DEMUX (element);
gint i;
switch (GST_STATE_TRANSITION (element)) {
case GST_STATE_READY_TO_PAUSED:
asf_demux->bs = gst_bytestream_new (asf_demux->sinkpad);
asf_demux->last_seek = 0;
break;
case GST_STATE_PAUSED_TO_READY:
gst_bytestream_destroy (asf_demux->bs);
for (i = 0; i < GST_ASF_DEMUX_NUM_VIDEO_PADS; i++) {
asf_demux->video_PTS[i] = 0;
}
for (i = 0; i < GST_ASF_DEMUX_NUM_AUDIO_PADS; i++) {
asf_demux->audio_PTS[i] = 0;
}
if (asf_demux->taglist) {
gst_tag_list_free (asf_demux->taglist);
asf_demux->taglist = NULL;
}
asf_demux->state = GST_ASF_DEMUX_STATE_HEADER;
asf_demux->seek_pending = GST_CLOCK_TIME_NONE;
asf_demux->seek_discont = FALSE;
break;
case GST_STATE_READY_TO_NULL:
break;
default:
break;
}
return GST_ELEMENT_CLASS (parent_class)->change_state (element);
}
static guint32
gst_asf_demux_identify_guid (GstASFDemux * asf_demux,
ASFGuidHash * guids, ASFGuid * guid)
{
guint32 i;
GST_LOG_OBJECT (asf_demux, "identifying 0x%08x/0x%08x/0x%08x/0x%08x",
guid->v1, guid->v2, guid->v3, guid->v4);
i = 0;
while (guids[i].obj_id != ASF_OBJ_UNDEFINED) {
if (guids[i].guid.v1 == guid->v1 &&
guids[i].guid.v2 == guid->v2 &&
guids[i].guid.v3 == guid->v3 && guids[i].guid.v4 == guid->v4) {
return guids[i].obj_id;
}
i++;
}
/* The base case if none is found */
return ASF_OBJ_UNDEFINED;
}
static gboolean
gst_asf_demux_add_audio_stream (GstASFDemux * asf_demux,
asf_stream_audio * audio, guint16 id)
{
GstPad *src_pad;
GstCaps *caps;
gchar *name = NULL;
guint16 size_left = 0;
char *codec_name = NULL;
GstTagList *list = gst_tag_list_new ();
GstBuffer *extradata = NULL;
size_left = audio->size;
/* Create the audio pad */
name = g_strdup_printf ("audio_%02d", asf_demux->num_audio_streams);
src_pad = gst_pad_new_from_template (audiosrctempl, name);
g_free (name);
gst_pad_use_explicit_caps (src_pad);
/* Swallow up any left over data and set up the standard properties from the header info */
if (size_left) {
GST_WARNING_OBJECT (asf_demux,
"asfdemux: Audio header contains %d bytes of codec specific data",
size_left);
gst_bytestream_peek (asf_demux->bs, &extradata, size_left);
gst_bytestream_flush (asf_demux->bs, size_left);
}
/* asf_stream_audio is the same as gst_riff_strf_auds, but with an
* additional two bytes indicating extradata. */
caps = gst_riff_create_audio_caps_with_data (audio->codec_tag,
NULL, (gst_riff_strf_auds *) audio, extradata, NULL, &codec_name);
/* Informing about that audio format we just added */
gst_tag_list_add (list, GST_TAG_MERGE_APPEND, GST_TAG_AUDIO_CODEC,
codec_name, NULL);
gst_element_found_tags (GST_ELEMENT (asf_demux), list);
g_free (codec_name);
if (extradata)
gst_buffer_unref (extradata);
GST_INFO ("Adding audio stream %u codec %u (0x%x)",
asf_demux->num_video_streams, audio->codec_tag, audio->codec_tag);
asf_demux->num_audio_streams++;
if (!gst_asf_demux_setup_pad (asf_demux, src_pad, caps, id))
return FALSE;
gst_pad_push (src_pad, GST_DATA (gst_event_new_tag (list)));
return TRUE;
}
static gboolean
gst_asf_demux_add_video_stream (GstASFDemux * asf_demux,
asf_stream_video_format * video, guint16 id)
{
GstPad *src_pad;
GstCaps *caps;
gchar *name = NULL;
char *codec_name = NULL;
GstTagList *list = gst_tag_list_new ();
gint size_left = video->size - 40;
GstBuffer *extradata = NULL;
/* Create the audio pad */
name = g_strdup_printf ("video_%02d", asf_demux->num_video_streams);
src_pad = gst_pad_new_from_template (videosrctempl, name);
g_free (name);
/* Now try some gstreamer formatted MIME types (from gst_avi_demux_strf_vids) */
if (size_left) {
GST_LOG_OBJECT (asf_demux,
"asfdemux: Video header contains %d bytes of codec specific data",
size_left);
gst_bytestream_peek (asf_demux->bs, &extradata, size_left);
gst_bytestream_flush (asf_demux->bs, size_left);
}
/* yes, asf_stream_video_format and gst_riff_strf_vids are the same */
caps = gst_riff_create_video_caps_with_data (video->tag, NULL,
(gst_riff_strf_vids *) video, extradata, NULL, &codec_name);
gst_caps_set_simple (caps, "framerate", G_TYPE_DOUBLE, 25.0, NULL);
gst_tag_list_add (list, GST_TAG_MERGE_APPEND, GST_TAG_VIDEO_CODEC,
codec_name, NULL);
gst_element_found_tags (GST_ELEMENT (asf_demux), list);
g_free (codec_name);
if (extradata)
gst_buffer_unref (extradata);
GST_INFO ("Adding video stream %u codec " GST_FOURCC_FORMAT " (0x%08x)",
asf_demux->num_video_streams, GST_FOURCC_ARGS (video->tag), video->tag);
asf_demux->num_video_streams++;
if (!gst_asf_demux_setup_pad (asf_demux, src_pad, caps, id))
return FALSE;
gst_pad_push (src_pad, GST_DATA (gst_event_new_tag (list)));
return TRUE;
}
static gboolean
gst_asf_demux_setup_pad (GstASFDemux * asf_demux,
GstPad * src_pad, GstCaps * caps, guint16 id)
{
asf_stream_context *stream;
gst_pad_use_explicit_caps (src_pad);
gst_pad_set_explicit_caps (src_pad, caps);
gst_pad_set_formats_function (src_pad, gst_asf_demux_get_src_formats);
gst_pad_set_event_mask_function (src_pad, gst_asf_demux_get_src_event_mask);
gst_pad_set_event_function (src_pad, gst_asf_demux_handle_src_event);
gst_pad_set_query_type_function (src_pad, gst_asf_demux_get_src_query_types);
gst_pad_set_query_function (src_pad, gst_asf_demux_handle_src_query);
stream = &asf_demux->stream[asf_demux->num_streams];
stream->pad = src_pad;
stream->id = id;
stream->frag_offset = 0;
stream->sequence = 0;
stream->delay = 0LL;
gst_pad_set_element_private (src_pad, stream);
GST_INFO ("Adding pad for stream %u", asf_demux->num_streams);
asf_demux->num_streams++;
gst_element_add_pad (GST_ELEMENT (asf_demux), src_pad);
if (asf_demux->taglist) {
gst_pad_push (src_pad,
GST_DATA (gst_event_new_tag (gst_tag_list_copy (asf_demux->taglist))));
}
return TRUE;
}
Reference in a new issue View git blame Copy permalink