/* * This library is licensed under 2 different licenses and you * can choose to use it under the terms of either one of them. The * two licenses are the MPL 1.1 and the LGPL. * * MPL: * * The contents of this file are subject to the Mozilla Public License * Version 1.1 (the "License"); you may not use this file except in * compliance with the License. You may obtain a copy of the License at * http://www.mozilla.org/MPL/. * * Software distributed under the License is distributed on an "AS IS" * basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See the * License for the specific language governing rights and limitations * under the License. * * LGPL: * * 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. * * The Original Code is Fluendo MPEG Demuxer plugin. * * The Initial Developer of the Original Code is Fluendo, S.A. * Portions created by Fluendo, S.L. are Copyright (C) 2005,2006,2007,2008,2009 * Fluendo, S.A. All Rights Reserved. * * Contributor(s): Wim Taymans */ #ifdef HAVE_CONFIG_H #include "config.h" #endif #include #include #include #include "gstmpegdefs.h" #include "gstmpegtsdemux.h" #include "flutspatinfo.h" #include "flutspmtinfo.h" GST_DEBUG_CATEGORY_STATIC (gstmpegtsdemux_debug); #define GST_CAT_DEFAULT (gstmpegtsdemux_debug) /* elementfactory information */ #ifndef __always_inline #if (__GNUC__ > 3) || (__GNUC__ == 3 && __GNUC_MINOR__ >= 1) #define __always_inline inline __attribute__((always_inline)) #else #define __always_inline inline #endif #endif #ifndef DISABLE_INLINE #define FORCE_INLINE __always_inline #else #define FORCE_INLINE #endif /* MPEG2Demux signals and args */ enum { /* FILL ME */ LAST_SIGNAL }; #define DEFAULT_PROP_ES_PIDS "" #define DEFAULT_PROP_CHECK_CRC TRUE #define DEFAULT_PROP_PROGRAM_NUMBER -1 /* latency in mseconds */ #define TS_LATENCY 700 enum { PROP_0, PROP_ES_PIDS, PROP_CHECK_CRC, PROP_PROGRAM_NUMBER, PROP_PAT_INFO, PROP_PMT_INFO, }; #define GSTTIME_TO_BYTES(time) \ ((time != -1) ? gst_util_uint64_scale (MAX(0,(gint64) ((time))), \ demux->bitrate, GST_SECOND) : -1) #define BYTES_TO_GSTTIME(bytes) \ ((bytes != -1) ? (gst_util_uint64_scale (bytes, GST_SECOND, \ demux->bitrate)) : -1) #define VIDEO_CAPS \ GST_STATIC_CAPS (\ "video/mpeg, " \ "mpegversion = (int) { 1, 2, 4 }, " \ "systemstream = (boolean) FALSE; " \ "video/x-h264;" \ "video/x-dirac;" \ "video/x-wmv," \ "wmvversion = (int) 3, " \ "format = (fourcc) WVC1" \ ) #define AUDIO_CAPS \ GST_STATIC_CAPS ( \ "audio/mpeg, " \ "mpegversion = (int) { 1, 4 };" \ "audio/x-lpcm, " \ "width = (int) { 16, 20, 24 }, " \ "rate = (int) { 48000, 96000 }, " \ "channels = (int) [ 1, 8 ], " \ "dynamic_range = (int) [ 0, 255 ], " \ "emphasis = (boolean) { FALSE, TRUE }, " \ "mute = (boolean) { FALSE, TRUE }; " \ "audio/x-ac3; audio/x-eac3;" \ "audio/x-dts;" \ "audio/x-private-ts-lpcm" \ ) /* Can also use the subpicture pads for text subtitles? */ #define SUBPICTURE_CAPS \ GST_STATIC_CAPS ("subpicture/x-pgs; video/x-dvd-subpicture") static GstStaticPadTemplate sink_template = GST_STATIC_PAD_TEMPLATE ("sink", GST_PAD_SINK, GST_PAD_ALWAYS, GST_STATIC_CAPS ("video/mpegts") ); static GstStaticPadTemplate video_template = GST_STATIC_PAD_TEMPLATE ("video_%04x", GST_PAD_SRC, GST_PAD_SOMETIMES, VIDEO_CAPS); static GstStaticPadTemplate audio_template = GST_STATIC_PAD_TEMPLATE ("audio_%04x", GST_PAD_SRC, GST_PAD_SOMETIMES, AUDIO_CAPS); static GstStaticPadTemplate subpicture_template = GST_STATIC_PAD_TEMPLATE ("subpicture_%04x", GST_PAD_SRC, GST_PAD_SOMETIMES, SUBPICTURE_CAPS); static GstStaticPadTemplate private_template = GST_STATIC_PAD_TEMPLATE ("private_%04x", GST_PAD_SRC, GST_PAD_SOMETIMES, GST_STATIC_CAPS_ANY); static void gst_mpegts_demux_base_init (GstMpegTSDemuxClass * klass); static void gst_mpegts_demux_class_init (GstMpegTSDemuxClass * klass); static void gst_mpegts_demux_init (GstMpegTSDemux * demux); static void gst_mpegts_demux_finalize (GstMpegTSDemux * demux); static void gst_mpegts_demux_reset (GstMpegTSDemux * demux); //static void gst_mpegts_demux_remove_pads (GstMpegTSDemux * demux); static void gst_mpegts_demux_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec); static void gst_mpegts_demux_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec); static gboolean gst_mpegts_demux_is_PMT (GstMpegTSDemux * demux, guint16 PID); static gboolean gst_mpegts_demux_sink_event (GstPad * pad, GstEvent * event); static gboolean gst_mpegts_demux_src_event (GstPad * pad, GstEvent * event); static GstFlowReturn gst_mpegts_demux_chain (GstPad * pad, GstBuffer * buffer); static gboolean gst_mpegts_demux_sink_setcaps (GstPad * pad, GstCaps * caps); static GstClock *gst_mpegts_demux_provide_clock (GstElement * element); static gboolean gst_mpegts_demux_src_pad_query (GstPad * pad, GstQuery * query); static const GstQueryType *gst_mpegts_demux_src_pad_query_type (GstPad * pad); static GstStateChangeReturn gst_mpegts_demux_change_state (GstElement * element, GstStateChange transition); static MpegTsPmtInfo *mpegts_demux_build_pmt_info (GstMpegTSDemux * demux, guint16 pmt_pid); static GstElementClass *parent_class = NULL; /*static guint gst_mpegts_demux_signals[LAST_SIGNAL] = { 0 };*/ GType gst_mpegts_demux_get_type (void) { static GType mpegts_demux_type = 0; if (G_UNLIKELY (!mpegts_demux_type)) { static const GTypeInfo mpegts_demux_info = { sizeof (GstMpegTSDemuxClass), (GBaseInitFunc) gst_mpegts_demux_base_init, NULL, (GClassInitFunc) gst_mpegts_demux_class_init, NULL, NULL, sizeof (GstMpegTSDemux), 0, (GInstanceInitFunc) gst_mpegts_demux_init, }; mpegts_demux_type = g_type_register_static (GST_TYPE_ELEMENT, "GstMpegTSDemux", &mpegts_demux_info, 0); GST_DEBUG_CATEGORY_INIT (gstmpegtsdemux_debug, "mpegtsdemux", 0, "MPEG program stream demultiplexer element"); } return mpegts_demux_type; } static void gst_mpegts_demux_base_init (GstMpegTSDemuxClass * klass) { GstElementClass *element_class = GST_ELEMENT_CLASS (klass); klass->sink_template = gst_static_pad_template_get (&sink_template); klass->video_template = gst_static_pad_template_get (&video_template); klass->audio_template = gst_static_pad_template_get (&audio_template); klass->subpicture_template = gst_static_pad_template_get (&subpicture_template); klass->private_template = gst_static_pad_template_get (&private_template); gst_element_class_add_pad_template (element_class, klass->video_template); gst_element_class_add_pad_template (element_class, klass->audio_template); gst_element_class_add_pad_template (element_class, klass->subpicture_template); gst_element_class_add_pad_template (element_class, klass->private_template); gst_element_class_add_pad_template (element_class, klass->sink_template); gst_element_class_set_details_simple (element_class, "The Fluendo MPEG Transport stream demuxer", "Codec/Demuxer", "Demultiplexes MPEG2 Transport Streams", "Wim Taymans "); } static void gst_mpegts_demux_class_init (GstMpegTSDemuxClass * klass) { GObjectClass *gobject_class; GstElementClass *gstelement_class; parent_class = g_type_class_peek_parent (klass); gobject_class = (GObjectClass *) klass; gstelement_class = (GstElementClass *) klass; gobject_class->finalize = (GObjectFinalizeFunc) gst_mpegts_demux_finalize; gobject_class->set_property = gst_mpegts_demux_set_property; gobject_class->get_property = gst_mpegts_demux_get_property; g_object_class_install_property (gobject_class, PROP_ES_PIDS, g_param_spec_string ("es-pids", "Colon separated list of PIDs containing Elementary Streams", "PIDs to treat as Elementary Streams in the absence of a PMT, " "eg 0x10:0x11:0x20", DEFAULT_PROP_ES_PIDS, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); g_object_class_install_property (gobject_class, PROP_CHECK_CRC, g_param_spec_boolean ("check-crc", "Check CRC", "Enable CRC checking", DEFAULT_PROP_CHECK_CRC, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); g_object_class_install_property (gobject_class, PROP_PROGRAM_NUMBER, g_param_spec_int ("program-number", "Program Number", "Program number to demux for (-1 to ignore)", -1, G_MAXINT, DEFAULT_PROP_PROGRAM_NUMBER, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS)); g_object_class_install_property (gobject_class, PROP_PAT_INFO, g_param_spec_value_array ("pat-info", "GValueArray containing GObjects with properties", "Array of GObjects containing information from the TS PAT " "about all programs listed in the current Program Association " "Table (PAT)", g_param_spec_object ("flu-pat-streaminfo", "FluPATStreamInfo", "Fluendo TS Demuxer PAT Stream info object", MPEGTS_TYPE_PAT_INFO, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS), G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)); g_object_class_install_property (gobject_class, PROP_PMT_INFO, g_param_spec_object ("pmt-info", "Information about the current program", "GObject with properties containing information from the TS PMT " "about the currently selected program and its streams", MPEGTS_TYPE_PMT_INFO, G_PARAM_READABLE | G_PARAM_STATIC_STRINGS)); gstelement_class->change_state = gst_mpegts_demux_change_state; gstelement_class->provide_clock = gst_mpegts_demux_provide_clock; } static void gst_mpegts_demux_init (GstMpegTSDemux * demux) { GstMpegTSDemuxClass *klass = GST_MPEGTS_DEMUX_GET_CLASS (demux); demux->streams = g_malloc0 (sizeof (GstMpegTSStream *) * (MPEGTS_MAX_PID + 1)); demux->sinkpad = gst_pad_new_from_template (klass->sink_template, "sink"); gst_pad_set_chain_function (demux->sinkpad, gst_mpegts_demux_chain); gst_pad_set_event_function (demux->sinkpad, gst_mpegts_demux_sink_event); gst_pad_set_setcaps_function (demux->sinkpad, gst_mpegts_demux_sink_setcaps); gst_element_add_pad (GST_ELEMENT (demux), demux->sinkpad); demux->elementary_pids = NULL; demux->nb_elementary_pids = 0; demux->check_crc = DEFAULT_PROP_CHECK_CRC; demux->program_number = DEFAULT_PROP_PROGRAM_NUMBER; demux->sync_lut = NULL; demux->sync_lut_len = 0; demux->bitrate = -1; demux->num_packets = 0; demux->pcr[0] = -1; demux->pcr[1] = -1; demux->cache_duration = GST_CLOCK_TIME_NONE; demux->base_pts = GST_CLOCK_TIME_NONE; } static void gst_mpegts_demux_finalize (GstMpegTSDemux * demux) { gst_mpegts_demux_reset (demux); g_free (demux->streams); G_OBJECT_CLASS (parent_class)->finalize (G_OBJECT (demux)); } static void gst_mpegts_demux_reset (GstMpegTSDemux * demux) { /* Clean up the streams and pads we allocated */ gint i; for (i = 0; i < MPEGTS_MAX_PID + 1; i++) { GstMpegTSStream *stream = demux->streams[i]; if (stream != NULL) { if (stream->pad) gst_element_remove_pad (GST_ELEMENT_CAST (demux), stream->pad); if (stream->ES_info) gst_mpeg_descriptor_free (stream->ES_info); if (stream->PMT.entries) g_array_free (stream->PMT.entries, TRUE); if (stream->PMT.program_info) gst_mpeg_descriptor_free (stream->PMT.program_info); if (stream->PAT.entries) g_array_free (stream->PAT.entries, TRUE); switch (stream->PID_type) { case PID_TYPE_ELEMENTARY: gst_pes_filter_uninit (&stream->filter); break; case PID_TYPE_PROGRAM_ASSOCIATION: case PID_TYPE_CONDITIONAL_ACCESS: case PID_TYPE_PROGRAM_MAP: gst_section_filter_uninit (&stream->section_filter); break; } if (stream->pes_buffer) { gst_buffer_unref (stream->pes_buffer); stream->pes_buffer = NULL; } g_free (stream); demux->streams[i] = NULL; } } if (demux->clock) { g_object_unref (demux->clock); demux->clock = NULL; } } #if 0 static void gst_mpegts_demux_remove_pads (GstMpegTSDemux * demux) { /* remove pads we added in preparation for adding new ones */ /* FIXME: instead of walking all streams, we should retain a list only * of streams that have added pads */ gint i; if (demux->need_no_more_pads) { gst_element_no_more_pads ((GstElement *) demux); demux->need_no_more_pads = FALSE; } for (i = 0; i < MPEGTS_MAX_PID + 1; i++) { GstMpegTSStream *stream = demux->streams[i]; if (stream != NULL) { if (GST_IS_PAD (stream->pad)) { gst_pad_push_event (stream->pad, gst_event_new_eos ()); gst_element_remove_pad (GST_ELEMENT_CAST (demux), stream->pad); } stream->pad = NULL; if (stream->PID_type == PID_TYPE_ELEMENTARY) gst_pes_filter_drain (&stream->filter); } } } #endif static guint32 crc_tab[256] = { 0x00000000, 0x04c11db7, 0x09823b6e, 0x0d4326d9, 0x130476dc, 0x17c56b6b, 0x1a864db2, 0x1e475005, 0x2608edb8, 0x22c9f00f, 0x2f8ad6d6, 0x2b4bcb61, 0x350c9b64, 0x31cd86d3, 0x3c8ea00a, 0x384fbdbd, 0x4c11db70, 0x48d0c6c7, 0x4593e01e, 0x4152fda9, 0x5f15adac, 0x5bd4b01b, 0x569796c2, 0x52568b75, 0x6a1936c8, 0x6ed82b7f, 0x639b0da6, 0x675a1011, 0x791d4014, 0x7ddc5da3, 0x709f7b7a, 0x745e66cd, 0x9823b6e0, 0x9ce2ab57, 0x91a18d8e, 0x95609039, 0x8b27c03c, 0x8fe6dd8b, 0x82a5fb52, 0x8664e6e5, 0xbe2b5b58, 0xbaea46ef, 0xb7a96036, 0xb3687d81, 0xad2f2d84, 0xa9ee3033, 0xa4ad16ea, 0xa06c0b5d, 0xd4326d90, 0xd0f37027, 0xddb056fe, 0xd9714b49, 0xc7361b4c, 0xc3f706fb, 0xceb42022, 0xca753d95, 0xf23a8028, 0xf6fb9d9f, 0xfbb8bb46, 0xff79a6f1, 0xe13ef6f4, 0xe5ffeb43, 0xe8bccd9a, 0xec7dd02d, 0x34867077, 0x30476dc0, 0x3d044b19, 0x39c556ae, 0x278206ab, 0x23431b1c, 0x2e003dc5, 0x2ac12072, 0x128e9dcf, 0x164f8078, 0x1b0ca6a1, 0x1fcdbb16, 0x018aeb13, 0x054bf6a4, 0x0808d07d, 0x0cc9cdca, 0x7897ab07, 0x7c56b6b0, 0x71159069, 0x75d48dde, 0x6b93dddb, 0x6f52c06c, 0x6211e6b5, 0x66d0fb02, 0x5e9f46bf, 0x5a5e5b08, 0x571d7dd1, 0x53dc6066, 0x4d9b3063, 0x495a2dd4, 0x44190b0d, 0x40d816ba, 0xaca5c697, 0xa864db20, 0xa527fdf9, 0xa1e6e04e, 0xbfa1b04b, 0xbb60adfc, 0xb6238b25, 0xb2e29692, 0x8aad2b2f, 0x8e6c3698, 0x832f1041, 0x87ee0df6, 0x99a95df3, 0x9d684044, 0x902b669d, 0x94ea7b2a, 0xe0b41de7, 0xe4750050, 0xe9362689, 0xedf73b3e, 0xf3b06b3b, 0xf771768c, 0xfa325055, 0xfef34de2, 0xc6bcf05f, 0xc27dede8, 0xcf3ecb31, 0xcbffd686, 0xd5b88683, 0xd1799b34, 0xdc3abded, 0xd8fba05a, 0x690ce0ee, 0x6dcdfd59, 0x608edb80, 0x644fc637, 0x7a089632, 0x7ec98b85, 0x738aad5c, 0x774bb0eb, 0x4f040d56, 0x4bc510e1, 0x46863638, 0x42472b8f, 0x5c007b8a, 0x58c1663d, 0x558240e4, 0x51435d53, 0x251d3b9e, 0x21dc2629, 0x2c9f00f0, 0x285e1d47, 0x36194d42, 0x32d850f5, 0x3f9b762c, 0x3b5a6b9b, 0x0315d626, 0x07d4cb91, 0x0a97ed48, 0x0e56f0ff, 0x1011a0fa, 0x14d0bd4d, 0x19939b94, 0x1d528623, 0xf12f560e, 0xf5ee4bb9, 0xf8ad6d60, 0xfc6c70d7, 0xe22b20d2, 0xe6ea3d65, 0xeba91bbc, 0xef68060b, 0xd727bbb6, 0xd3e6a601, 0xdea580d8, 0xda649d6f, 0xc423cd6a, 0xc0e2d0dd, 0xcda1f604, 0xc960ebb3, 0xbd3e8d7e, 0xb9ff90c9, 0xb4bcb610, 0xb07daba7, 0xae3afba2, 0xaafbe615, 0xa7b8c0cc, 0xa379dd7b, 0x9b3660c6, 0x9ff77d71, 0x92b45ba8, 0x9675461f, 0x8832161a, 0x8cf30bad, 0x81b02d74, 0x857130c3, 0x5d8a9099, 0x594b8d2e, 0x5408abf7, 0x50c9b640, 0x4e8ee645, 0x4a4ffbf2, 0x470cdd2b, 0x43cdc09c, 0x7b827d21, 0x7f436096, 0x7200464f, 0x76c15bf8, 0x68860bfd, 0x6c47164a, 0x61043093, 0x65c52d24, 0x119b4be9, 0x155a565e, 0x18197087, 0x1cd86d30, 0x029f3d35, 0x065e2082, 0x0b1d065b, 0x0fdc1bec, 0x3793a651, 0x3352bbe6, 0x3e119d3f, 0x3ad08088, 0x2497d08d, 0x2056cd3a, 0x2d15ebe3, 0x29d4f654, 0xc5a92679, 0xc1683bce, 0xcc2b1d17, 0xc8ea00a0, 0xd6ad50a5, 0xd26c4d12, 0xdf2f6bcb, 0xdbee767c, 0xe3a1cbc1, 0xe760d676, 0xea23f0af, 0xeee2ed18, 0xf0a5bd1d, 0xf464a0aa, 0xf9278673, 0xfde69bc4, 0x89b8fd09, 0x8d79e0be, 0x803ac667, 0x84fbdbd0, 0x9abc8bd5, 0x9e7d9662, 0x933eb0bb, 0x97ffad0c, 0xafb010b1, 0xab710d06, 0xa6322bdf, 0xa2f33668, 0xbcb4666d, 0xb8757bda, 0xb5365d03, 0xb1f740b4 }; /*This function fills the value of negotiated packetsize at sinkpad*/ static gboolean gst_mpegts_demux_sink_setcaps (GstPad * pad, GstCaps * caps) { GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad)); GstStructure *structure = NULL; structure = gst_caps_get_structure (caps, 0); GST_DEBUG_OBJECT (demux, "setcaps called with %" GST_PTR_FORMAT, caps); if (!gst_structure_get_int (structure, "packetsize", &demux->packetsize)) { GST_DEBUG_OBJECT (demux, "packetsize parameter not found in sink caps"); } gst_object_unref (demux); return TRUE; } static FORCE_INLINE guint32 gst_mpegts_demux_calc_crc32 (guint8 * data, guint datalen) { gint i; guint32 crc = 0xffffffff; for (i = 0; i < datalen; i++) { crc = (crc << 8) ^ crc_tab[((crc >> 24) ^ *data++) & 0xff]; } return crc; } static FORCE_INLINE gboolean gst_mpegts_is_dirac_stream (GstMpegTSStream * stream) { gboolean is_dirac = FALSE; if (stream->stream_type != ST_VIDEO_DIRAC) return FALSE; if (stream->ES_info != NULL) { guint8 *dirac_desc; /* Check for a Registration Descriptor to confirm this is dirac */ dirac_desc = gst_mpeg_descriptor_find (stream->ES_info, DESC_REGISTRATION); if (dirac_desc != NULL && DESC_LENGTH (dirac_desc) >= 4) { if (DESC_REGISTRATION_format_identifier (dirac_desc) == 0x64726163) { /* 'drac' in hex */ is_dirac = TRUE; } } else { /* Check for old mapping as originally specified too */ dirac_desc = gst_mpeg_descriptor_find (stream->ES_info, DESC_DIRAC_TC_PRIVATE); if (dirac_desc != NULL && DESC_LENGTH (dirac_desc) == 0) is_dirac = TRUE; } } return is_dirac; } static FORCE_INLINE gboolean gst_mpegts_stream_is_video (GstMpegTSStream * stream) { switch (stream->stream_type) { case ST_VIDEO_MPEG1: case ST_VIDEO_MPEG2: case ST_VIDEO_MPEG4: case ST_VIDEO_H264: return TRUE; case ST_VIDEO_DIRAC: return gst_mpegts_is_dirac_stream (stream); } return FALSE; } static gboolean gst_mpegts_demux_is_reserved_PID (GstMpegTSDemux * demux, guint16 PID) { return (PID >= PID_RESERVED_FIRST) && (PID < PID_RESERVED_LAST); } /* This function assumes that provided PID never will be greater than * MPEGTS_MAX_PID (13 bits), this is currently guaranteed as everywhere in * the code recovered PID at maximum is 13 bits long. */ static FORCE_INLINE GstMpegTSStream * gst_mpegts_demux_get_stream_for_PID (GstMpegTSDemux * demux, guint16 PID) { GstMpegTSStream *stream = NULL; stream = demux->streams[PID]; if (G_UNLIKELY (stream == NULL)) { stream = g_new0 (GstMpegTSStream, 1); stream->demux = demux; stream->PID = PID; stream->pad = NULL; stream->base_PCR = -1; stream->last_PCR = -1; stream->last_PCR_difference = -1; stream->PMT.version_number = -1; stream->PAT.version_number = -1; stream->PMT_pid = MPEGTS_MAX_PID + 1; stream->flags |= MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN; stream->pes_buffer_in_sync = FALSE; switch (PID) { /* check for fixed mapping */ case PID_PROGRAM_ASSOCIATION_TABLE: stream->PID_type = PID_TYPE_PROGRAM_ASSOCIATION; /* initialise section filter */ gst_section_filter_init (&stream->section_filter); break; case PID_CONDITIONAL_ACCESS_TABLE: stream->PID_type = PID_TYPE_CONDITIONAL_ACCESS; /* initialise section filter */ gst_section_filter_init (&stream->section_filter); break; case PID_NULL_PACKET: stream->PID_type = PID_TYPE_NULL_PACKET; break; default: /* mark reserved PIDs */ if (gst_mpegts_demux_is_reserved_PID (demux, PID)) { stream->PID_type = PID_TYPE_RESERVED; } else { /* check if PMT found in PAT */ if (gst_mpegts_demux_is_PMT (demux, PID)) { stream->PID_type = PID_TYPE_PROGRAM_MAP; /* initialise section filter */ gst_section_filter_init (&stream->section_filter); } else stream->PID_type = PID_TYPE_UNKNOWN; } break; } GST_DEBUG_OBJECT (demux, "creating stream %p for PID 0x%04x, PID_type %d", stream, PID, stream->PID_type); demux->streams[PID] = stream; } return stream; } static gboolean gst_mpegts_demux_fill_stream (GstMpegTSStream * stream, guint8 id, guint8 stream_type) { GstPadTemplate *template; gchar *name; GstMpegTSDemuxClass *klass; GstMpegTSDemux *demux; GstCaps *caps; if (stream->stream_type && stream->stream_type != stream_type) goto wrong_type; demux = stream->demux; klass = GST_MPEGTS_DEMUX_GET_CLASS (demux); name = NULL; template = NULL; caps = NULL; switch (stream_type) { case ST_VIDEO_MPEG1: case ST_VIDEO_MPEG2: template = klass->video_template; name = g_strdup_printf ("video_%04x", stream->PID); caps = gst_caps_new_simple ("video/mpeg", "mpegversion", G_TYPE_INT, stream_type == ST_VIDEO_MPEG1 ? 1 : 2, "systemstream", G_TYPE_BOOLEAN, FALSE, NULL); break; case ST_AUDIO_MPEG1: case ST_AUDIO_MPEG2: template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/mpeg", "mpegversion", G_TYPE_INT, 1, NULL); break; case ST_PRIVATE_DATA: /* check if there is an AC3 descriptor associated with this stream * from the PMT */ if (gst_mpeg_descriptor_find (stream->ES_info, DESC_DVB_AC3)) { template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/x-ac3", NULL); } else if (gst_mpeg_descriptor_find (stream->ES_info, DESC_DVB_ENHANCED_AC3)) { template = klass->private_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/x-eac3", NULL); } else if (gst_mpeg_descriptor_find (stream->ES_info, DESC_DVB_TELETEXT)) { template = klass->private_template; name = g_strdup_printf ("private_%04x", stream->PID); caps = gst_caps_new_simple ("private/teletext", NULL); } else if (gst_mpeg_descriptor_find (stream->ES_info, DESC_DVB_SUBTITLING)) { template = klass->private_template; name = g_strdup_printf ("private_%04x", stream->PID); caps = gst_caps_new_simple ("subpicture/x-dvb", NULL); } break; case ST_HDV_AUX_V: template = klass->private_template; name = g_strdup_printf ("private_%04x", stream->PID); caps = gst_caps_new_simple ("hdv/aux-v", NULL); break; case ST_HDV_AUX_A: template = klass->private_template; name = g_strdup_printf ("private_%04x", stream->PID); caps = gst_caps_new_simple ("hdv/aux-a", NULL); break; case ST_PRIVATE_SECTIONS: case ST_MHEG: case ST_DSMCC: break; case ST_AUDIO_AAC: template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/mpeg", "mpegversion", G_TYPE_INT, 4, NULL); break; case ST_VIDEO_MPEG4: template = klass->video_template; name = g_strdup_printf ("video_%04x", stream->PID); caps = gst_caps_new_simple ("video/mpeg", "mpegversion", G_TYPE_INT, 4, "systemstream", G_TYPE_BOOLEAN, FALSE, NULL); break; case ST_VIDEO_H264: template = klass->video_template; name = g_strdup_printf ("video_%04x", stream->PID); caps = gst_caps_new_simple ("video/x-h264", NULL); break; case ST_VIDEO_DIRAC: if (gst_mpegts_is_dirac_stream (stream)) { template = klass->video_template; name = g_strdup_printf ("video_%04x", stream->PID); caps = gst_caps_new_simple ("video/x-dirac", NULL); } break; case ST_PRIVATE_EA: /* Try to detect a VC1 stream */ { guint8 *desc = NULL; if (stream->ES_info) desc = gst_mpeg_descriptor_find (stream->ES_info, DESC_REGISTRATION); if (!(desc && DESC_REGISTRATION_format_identifier (desc) == DRF_ID_VC1)) { GST_WARNING ("0xea private stream type found but no descriptor " "for VC1. Assuming plain VC1."); } template = klass->video_template; name = g_strdup_printf ("video_%04x", stream->PID); caps = gst_caps_new_simple ("video/x-wmv", "wmvversion", G_TYPE_INT, 3, "format", GST_TYPE_FOURCC, GST_MAKE_FOURCC ('W', 'V', 'C', '1'), NULL); break; } case ST_BD_AUDIO_AC3: { GstMpegTSStream *PMT_stream = gst_mpegts_demux_get_stream_for_PID (stream->demux, stream->PMT_pid); GstMPEGDescriptor *program_info = PMT_stream->PMT.program_info; guint8 *desc = NULL; if (program_info) desc = gst_mpeg_descriptor_find (program_info, DESC_REGISTRATION); if (desc && DESC_REGISTRATION_format_identifier (desc) == DRF_ID_HDMV) { template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/x-eac3", NULL); } else if (stream->ES_info && gst_mpeg_descriptor_find (stream->ES_info, DESC_DVB_ENHANCED_AC3)) { template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/x-eac3", NULL); } else { if (!stream->ES_info || !gst_mpeg_descriptor_find (stream->ES_info, DESC_DVB_AC3)) { GST_WARNING ("AC3 stream type found but no corresponding " "descriptor to differentiate between AC3 and EAC3. " "Assuming plain AC3."); } template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/x-ac3", NULL); } break; } case ST_BD_AUDIO_EAC3: template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/x-eac3", NULL); break; case ST_PS_AUDIO_DTS: case ST_BD_AUDIO_DTS: case ST_BD_AUDIO_DTS_HD: case ST_BD_AUDIO_DTS_HD_MASTER_AUDIO: template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/x-dts", NULL); break; case ST_PS_AUDIO_LPCM: template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/x-lpcm", NULL); break; case ST_BD_AUDIO_LPCM: template = klass->audio_template; name = g_strdup_printf ("audio_%04x", stream->PID); caps = gst_caps_new_simple ("audio/x-private-ts-lpcm", NULL); break; case ST_PS_DVD_SUBPICTURE: template = klass->subpicture_template; name = g_strdup_printf ("subpicture_%04x", stream->PID); caps = gst_caps_new_simple ("video/x-dvd-subpicture", NULL); break; case ST_BD_PGS_SUBPICTURE: template = klass->subpicture_template; name = g_strdup_printf ("subpicture_%04x", stream->PID); caps = gst_caps_new_simple ("subpicture/x-pgs", NULL); break; default: break; } if (name == NULL || template == NULL || caps == NULL) return FALSE; stream->stream_type = stream_type; stream->id = id; GST_DEBUG ("creating new pad %s", name); stream->pad = gst_pad_new_from_template (template, name); gst_pad_use_fixed_caps (stream->pad); gst_pad_set_caps (stream->pad, caps); gst_caps_unref (caps); gst_pad_set_query_function (stream->pad, GST_DEBUG_FUNCPTR (gst_mpegts_demux_src_pad_query)); gst_pad_set_query_type_function (stream->pad, GST_DEBUG_FUNCPTR (gst_mpegts_demux_src_pad_query_type)); gst_pad_set_event_function (stream->pad, GST_DEBUG_FUNCPTR (gst_mpegts_demux_src_event)); g_free (name); return TRUE; wrong_type: { return FALSE; } } static FORCE_INLINE gboolean mpegts_is_elem_pid (GstMpegTSDemux * demux, guint16 PID) { int i; /* check if it's in our partial ts pid list */ for (i = 0; i < demux->nb_elementary_pids; i++) { if (demux->elementary_pids[i] == PID) { return TRUE; } } return FALSE; } static gboolean gst_mpegts_demux_setup_base_pts (GstMpegTSDemux * demux, gint64 pts) { GstMpegTSStream *PCR_stream; GstMpegTSStream *PMT_stream; guint64 base_PCR; /* for the reference start time we need to consult the PCR_PID of the * current PMT */ if (demux->current_PMT == 0) goto no_pmt_stream; PMT_stream = demux->streams[demux->current_PMT]; if (PMT_stream == NULL) goto no_pmt_stream; PCR_stream = demux->streams[PMT_stream->PMT.PCR_PID]; if (PCR_stream == NULL) goto no_pcr_stream; if (PCR_stream->base_PCR == -1) { GST_DEBUG_OBJECT (demux, "no base PCR, using last PCR %" G_GUINT64_FORMAT, PCR_stream->last_PCR); PCR_stream->base_PCR = PCR_stream->last_PCR; } else { GST_DEBUG_OBJECT (demux, "using base PCR %" G_GUINT64_FORMAT, PCR_stream->base_PCR); } if (PCR_stream->last_PCR == -1) { GST_DEBUG_OBJECT (demux, "no last PCR, using PTS %" G_GUINT64_FORMAT, pts); PCR_stream->base_PCR = pts; PCR_stream->last_PCR = pts; } base_PCR = PCR_stream->base_PCR; demux->base_pts = MPEGTIME_TO_GSTTIME (base_PCR); if (demux->base_pts == GST_CLOCK_TIME_NONE) return FALSE; return TRUE; no_pmt_stream: { GST_DEBUG_OBJECT (demux, "no PMT stream found"); return FALSE; } no_pcr_stream: { GST_DEBUG_OBJECT (demux, "no PCR stream found"); return FALSE; } } static gboolean gst_mpegts_demux_send_new_segment (GstMpegTSDemux * demux, GstMpegTSStream * stream, gint64 pts) { GstClockTime time; /* base_pts needs to have been set up by a call to * gst_mpegts_demux_setup_base_pts() before calling this function */ if (demux->base_pts == GST_CLOCK_TIME_NONE) goto no_base_time; time = demux->base_pts; GST_DEBUG_OBJECT (demux, "segment PTS to time: %" GST_TIME_FORMAT, GST_TIME_ARGS (time)); if (demux->clock && demux->clock_base == GST_CLOCK_TIME_NONE) { demux->clock_base = gst_clock_get_time (demux->clock); gst_clock_set_calibration (demux->clock, gst_clock_get_internal_time (demux->clock), demux->clock_base, 1, 1); } gst_pad_push_event (stream->pad, gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_TIME, time, -1, 0)); return TRUE; /* ERRORS */ no_base_time: { /* check if it's in our partial ts pid list */ if (mpegts_is_elem_pid (demux, stream->PID)) { GST_DEBUG_OBJECT (demux, "Elementary PID, using pts %" G_GUINT64_FORMAT, pts); time = MPEGTIME_TO_GSTTIME (pts) + stream->base_time; GST_DEBUG_OBJECT (demux, "segment PTS to (%" G_GUINT64_FORMAT ") time: %" G_GUINT64_FORMAT, pts, time); gst_pad_push_event (stream->pad, gst_event_new_new_segment (FALSE, 1.0, GST_FORMAT_TIME, time, -1, 0)); return TRUE; } } return FALSE; } static void gst_mpegts_demux_send_tags_for_stream (GstMpegTSDemux * demux, GstMpegTSStream * stream) { GstTagList *list = NULL; gint i; if (stream->ES_info) { static const guint8 lang_descs[] = { DESC_ISO_639_LANGUAGE, DESC_DVB_SUBTITLING }; for (i = 0; i < G_N_ELEMENTS (lang_descs); i++) { guint8 *iso639_languages = gst_mpeg_descriptor_find (stream->ES_info, lang_descs[i]); if (iso639_languages) { if (DESC_ISO_639_LANGUAGE_codes_n (iso639_languages)) { const gchar *lc; gchar lang_code[4]; gchar *language_n; language_n = (gchar *) DESC_ISO_639_LANGUAGE_language_code_nth (iso639_languages, 0); lang_code[0] = language_n[0]; lang_code[1] = language_n[1]; lang_code[2] = language_n[2]; lang_code[3] = 0; if (!list) list = gst_tag_list_new (); /* descriptor contains ISO 639-2 code, we want the ISO 639-1 code */ lc = gst_tag_get_language_code (lang_code); gst_tag_list_add (list, GST_TAG_MERGE_REPLACE, GST_TAG_LANGUAGE_CODE, (lc) ? lc : lang_code, NULL); } } } } if (list) { GST_DEBUG_OBJECT (demux, "Sending tags %p for pad %s:%s", list, GST_DEBUG_PAD_NAME (stream->pad)); gst_element_found_tags_for_pad (GST_ELEMENT (demux), stream->pad, list); } } static GstFlowReturn gst_mpegts_demux_combine_flows (GstMpegTSDemux * demux, GstMpegTSStream * stream, GstFlowReturn ret) { gint i; /* store the value */ stream->last_ret = ret; /* if it's success we can return the value right away */ if (ret == GST_FLOW_OK) goto done; /* any other error that is not-linked can be returned right * away */ if (ret != GST_FLOW_NOT_LINKED) goto done; /* only return NOT_LINKED if all other pads returned NOT_LINKED */ for (i = 0; i < MPEGTS_MAX_PID + 1; i++) { if (!(stream = demux->streams[i])) continue; /* some other return value (must be SUCCESS but we can return * other values as well) */ ret = stream->last_ret; if (ret != GST_FLOW_NOT_LINKED) goto done; } /* if we get here, all other pads were unlinked and we return * NOT_LINKED then */ done: return ret; } static GstFlowReturn gst_mpegts_demux_data_cb (GstPESFilter * filter, gboolean first, GstBuffer * buffer, GstMpegTSStream * stream) { GstMpegTSDemux *demux; GstFlowReturn ret; GstPad *srcpad; gint64 pts; GstClockTime time; demux = stream->demux; srcpad = stream->pad; GST_DEBUG_OBJECT (demux, "got data on PID 0x%04x", stream->PID); if (first && filter->pts != -1) { pts = filter->pts; time = MPEGTIME_TO_GSTTIME (pts) + stream->base_time; if ((stream->last_time > 0 && stream->last_time < time && time - stream->last_time > GST_SECOND * 60 * 10) || (stream->last_time > time && stream->last_time - time > GST_SECOND * 60 * 10)) { /* check first to see if we're in middle of detecting a discont in PCR. * if we are we're not sure what timestamp the buffer should have, best * to drop. */ if (stream->PMT_pid <= MPEGTS_MAX_PID && demux->streams[stream->PMT_pid] && demux->streams[demux->streams[stream->PMT_pid]->PMT.PCR_PID] && demux->streams[demux->streams[stream->PMT_pid]->PMT.PCR_PID]-> discont_PCR) { GST_WARNING_OBJECT (demux, "middle of discont, dropping"); goto bad_timestamp; } /* check for wraparounds */ else if (stream->last_time > 0 && time < stream->last_time && stream->last_time - time > MPEGTIME_TO_GSTTIME (G_MAXUINT32)) { /* wrap around occurred */ if (stream->base_time + MPEGTIME_TO_GSTTIME ((guint64) (1) << 33) + MPEGTIME_TO_GSTTIME (pts) > stream->last_time + GST_SECOND * 60 * 10) { GST_DEBUG_OBJECT (demux, "looks like we have a corrupt packet because its pts is a lot lower than" " the previous pts but not a wraparound"); goto bad_timestamp; } /* wraparound has occured but before we have detected in the pcr, * so check we're actually getting pcr's...if we are, don't update * the base time..just set the time and last_time correctly */ if (stream->PMT_pid <= MPEGTS_MAX_PID && demux->streams[stream->PMT_pid] && demux->streams[demux->streams[stream->PMT_pid]->PMT.PCR_PID] && demux->streams[demux->streams[stream->PMT_pid]->PMT.PCR_PID]-> last_PCR > 0) { GST_DEBUG_OBJECT (demux, "timestamps wrapped before noticed in PCR"); time = MPEGTIME_TO_GSTTIME (pts) + stream->base_time + MPEGTIME_TO_GSTTIME ((guint64) (1) << 33); stream->last_time = time; } else { stream->base_time = stream->base_time + MPEGTIME_TO_GSTTIME ((guint64) (1) << 33); time = MPEGTIME_TO_GSTTIME (pts) + stream->base_time; GST_DEBUG_OBJECT (demux, "timestamps wrapped around, compensating with new base time: %" GST_TIME_FORMAT "last time: %" GST_TIME_FORMAT " time: %" GST_TIME_FORMAT, GST_TIME_ARGS (stream->base_time), GST_TIME_ARGS (stream->last_time), GST_TIME_ARGS (time)); stream->last_time = time; } } else if (stream->last_time > 0 && time > stream->last_time && time - stream->last_time > MPEGTIME_TO_GSTTIME (G_MAXUINT32) && stream->base_time > 0) { /* had a previous wrap around */ if (time - MPEGTIME_TO_GSTTIME ((guint64) (1) << 33) + GST_SECOND * 60 * 10 < stream->last_time) { GST_DEBUG_OBJECT (demux, "looks like we have a corrupt packet because its pts is a lot higher than" " the previous pts but not because of a wraparound or pcr discont"); goto bad_timestamp; } if (ABS ((time - MPEGTIME_TO_GSTTIME ((guint64) (1) << 33)) - stream->last_time) < GST_SECOND) { GST_DEBUG_OBJECT (demux, "timestamps wrapped around earlier but we have an out of pts: %" G_GUINT64_FORMAT ", as %" GST_TIME_FORMAT " translated to: %" GST_TIME_FORMAT " and last_time of %" GST_TIME_FORMAT, pts, GST_TIME_ARGS (time), GST_TIME_ARGS (time - MPEGTIME_TO_GSTTIME ((guint64) (1) << 33)), GST_TIME_ARGS (stream->last_time)); time = time - MPEGTIME_TO_GSTTIME ((guint64) (1) << 33); } else { GST_DEBUG_OBJECT (demux, "timestamp may have wrapped around recently but not sure and pts" " is very different, dropping it timestamp of this packet: %" GST_TIME_FORMAT " compared to last timestamp: %" GST_TIME_FORMAT, GST_TIME_ARGS (time - MPEGTIME_TO_GSTTIME ((guint64) (1) << (33))), GST_TIME_ARGS (stream->last_time)); goto bad_timestamp; } } else { /* we must have a corrupt packet */ GST_WARNING_OBJECT (demux, "looks like we have a corrupt packet because" " its timestamp is buggered timestamp: %" GST_TIME_FORMAT " compared to" " last timestamp: %" GST_TIME_FORMAT, GST_TIME_ARGS (time), GST_TIME_ARGS (stream->last_time)); goto bad_timestamp; } } else { /* do not set last_time if a packet with pts from before wrap around arrived after the wrap around occured */ stream->last_time = time; } } else { time = GST_CLOCK_TIME_NONE; pts = -1; } GST_DEBUG_OBJECT (demux, "setting PTS to (%" G_GUINT64_FORMAT ") time: %" GST_TIME_FORMAT " on buffer %p first buffer: %d base_time: %" GST_TIME_FORMAT, pts, GST_TIME_ARGS (time), buffer, first, GST_TIME_ARGS (stream->base_time)); GST_BUFFER_TIMESTAMP (buffer) = time; /* check if we have a pad already */ if (srcpad == NULL) { /* When adding a stream, require either a valid base PCR, or a valid PTS */ if (!gst_mpegts_demux_setup_base_pts (demux, pts)) goto bad_timestamp; /* fill in the last bits of the stream */ /* if no stream type, then assume it based on the PES start code, * needed for partial ts streams without PMT */ if (G_UNLIKELY (stream->flags & MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN)) { if ((filter->start_code & 0xFFFFFFF0) == PACKET_VIDEO_START_CODE) { /* it is mpeg2 video */ stream->stream_type = ST_VIDEO_MPEG2; stream->flags &= ~MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN; stream->flags |= MPEGTS_STREAM_FLAG_IS_VIDEO; GST_DEBUG_OBJECT (demux, "Found stream 0x%04x without PMT with video " "start_code. Treating as video", stream->PID); } else if ((filter->start_code & 0xFFFFFFE0) == PACKET_AUDIO_START_CODE) { /* it is mpeg audio */ stream->stream_type = ST_AUDIO_MPEG2; stream->flags &= ~MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN; GST_DEBUG_OBJECT (demux, "Found stream 0x%04x without PMT with audio " "start_code. Treating as audio", stream->PID); } else { GST_LOG_OBJECT (demux, "Stream start code on pid 0x%04x is: 0x%x", stream->PID, filter->start_code); } } if (!gst_mpegts_demux_fill_stream (stream, filter->id, stream->stream_type)) goto unknown_type; GST_DEBUG_OBJECT (demux, "New stream 0x%04x of type 0x%02x with caps %" GST_PTR_FORMAT, stream->PID, stream->stream_type, GST_PAD_CAPS (stream->pad)); srcpad = stream->pad; /* activate and add */ gst_pad_set_active (srcpad, TRUE); gst_element_add_pad (GST_ELEMENT_CAST (demux), srcpad); demux->need_no_more_pads = TRUE; stream->discont = TRUE; /* send new_segment */ gst_mpegts_demux_send_new_segment (demux, stream, pts); /* send tags */ gst_mpegts_demux_send_tags_for_stream (demux, stream); } GST_DEBUG_OBJECT (srcpad, "pushing buffer"); gst_buffer_set_caps (buffer, GST_PAD_CAPS (srcpad)); if (stream->discont) { GST_BUFFER_FLAG_SET (buffer, GST_BUFFER_FLAG_DISCONT); stream->discont = FALSE; } ret = gst_pad_push (srcpad, buffer); ret = gst_mpegts_demux_combine_flows (demux, stream, ret); return ret; /* ERROR */ unknown_type: { GST_DEBUG_OBJECT (demux, "got unknown stream id 0x%02x, type 0x%02x", filter->id, stream->stream_type); gst_buffer_unref (buffer); return gst_mpegts_demux_combine_flows (demux, stream, GST_FLOW_NOT_LINKED); } bad_timestamp: { gst_buffer_unref (buffer); return gst_mpegts_demux_combine_flows (demux, stream, GST_FLOW_OK); } } static void gst_mpegts_demux_resync_cb (GstPESFilter * filter, GstMpegTSStream * stream) { /* does nothing for now */ } /* * CA_section() { * table_id 8 uimsbf == 0x01 * section_syntax_indicator 1 bslbf == 1 * '0' 1 bslbf == 0 * reserved 2 bslbf * section_length 12 uimsbf == 00xxxxx... * reserved 18 bslbf * version_number 5 uimsbf * current_next_indicator 1 bslbf * section_number 8 uimsbf * last_section_number 8 uimsbf * for (i=0; idemux; GST_DEBUG_OBJECT (demux, "parsing CA section"); return TRUE; } static void gst_mpegts_activate_pmt (GstMpegTSDemux * demux, GstMpegTSStream * stream) { GST_DEBUG_OBJECT (demux, "activating PMT 0x%08x", stream->PID); /* gst_mpegts_demux_remove_pads (demux); */ demux->current_PMT = stream->PID; /* PMT has been updated, signal the change */ if (demux->current_PMT == stream->PID) g_object_notify ((GObject *) (demux), "pmt-info"); } /* * TS_program_map_section() { * table_id 8 uimsbf == 0x02 * section_syntax_indicator 1 bslbf == 1 * '0' 1 bslbf == 0 * reserved 2 bslbf * section_length 12 uimsbf == 00xxxxx... * program_number 16 uimsbf * reserved 2 bslbf * version_number 5 uimsbf * current_next_indicator 1 bslbf * section_number 8 uimsbf * last_section_number 8 uimsbf * reserved 3 bslbf * PCR_PID 13 uimsbf * reserved 4 bslbf * program_info_length 12 uimsbf == 00xxxxx... * for (i=0; idemux; if (G_UNLIKELY (*data++ != 0x02)) goto wrong_id; if ((data[0] & 0xc0) != 0x80) goto wrong_sync; if ((data[0] & 0x0c) != 0x00) goto wrong_seclen; data += 2; if (demux->check_crc) if (G_UNLIKELY (gst_mpegts_demux_calc_crc32 (data - 3, datalen) != 0)) goto wrong_crc; GST_LOG_OBJECT (demux, "PMT section_length: %d", datalen - 3); PMT = &stream->PMT; /* check if version number changed */ version_number = (data[2] & 0x3e) >> 1; GST_LOG_OBJECT (demux, "PMT version_number: %d", version_number); current_next_indicator = (data[2] & 0x01); GST_LOG_OBJECT (demux, "PMT current_next_indicator %d", current_next_indicator); if (current_next_indicator == 0) goto not_yet_applicable; program_number = GST_READ_UINT16_BE (data); if (demux->program_number != -1 && demux->program_number != program_number) { goto wrong_program_number; } if (demux->program_number == -1) { GST_INFO_OBJECT (demux, "No program number set, so using first parsed PMT" "'s program number: %d", program_number); demux->program_number = program_number; } if (version_number == PMT->version_number) goto same_version; PMT->version_number = version_number; PMT->current_next_indicator = current_next_indicator; stream->PMT.program_number = program_number; data += 3; GST_DEBUG_OBJECT (demux, "PMT program_number: %d", PMT->program_number); PMT->section_number = *data++; GST_DEBUG_OBJECT (demux, "PMT section_number: %d", PMT->section_number); PMT->last_section_number = *data++; GST_DEBUG_OBJECT (demux, "PMT last_section_number: %d", PMT->last_section_number); PMT->PCR_PID = GST_READ_UINT16_BE (data); PMT->PCR_PID &= 0x1fff; data += 2; GST_DEBUG_OBJECT (demux, "PMT PCR_PID: 0x%04x", PMT->PCR_PID); /* create or get stream, not much we can say about it except that when we get * a data stream and we need a PCR, we can use the stream to get/store the * base_PCR. */ gst_mpegts_demux_get_stream_for_PID (demux, PMT->PCR_PID); if ((data[0] & 0x0c) != 0x00) goto wrong_pilen; PMT->program_info_length = GST_READ_UINT16_BE (data); PMT->program_info_length &= 0x0fff; /* FIXME: validate value of program_info_length */ data += 2; /* FIXME: validate value of program_info_length, before using */ /* parse descriptor */ if (G_UNLIKELY (PMT->program_info)) gst_mpeg_descriptor_free (PMT->program_info); PMT->program_info = gst_mpeg_descriptor_parse (data, PMT->program_info_length); /* skip descriptor */ data += PMT->program_info_length; GST_DEBUG_OBJECT (demux, "PMT program_info_length: %d", PMT->program_info_length); entries = datalen - 3 - PMT->program_info_length - 9 - 4; if (G_UNLIKELY (PMT->entries)) g_array_free (PMT->entries, TRUE); PMT->entries = g_array_new (FALSE, TRUE, sizeof (GstMpegTSPMTEntry)); while (entries > 0) { GstMpegTSPMTEntry entry; GstMpegTSStream *ES_stream; guint8 stream_type; guint16 ES_info_length; stream_type = *data++; entry.PID = GST_READ_UINT16_BE (data); entry.PID &= 0x1fff; data += 2; if ((data[0] & 0x0c) != 0x00) goto wrong_esilen; ES_info_length = GST_READ_UINT16_BE (data); ES_info_length &= 0x0fff; data += 2; /* get/create elementary stream */ ES_stream = gst_mpegts_demux_get_stream_for_PID (demux, entry.PID); /* check if PID unknown */ if (ES_stream->PID_type == PID_TYPE_UNKNOWN) { /* set as elementary */ ES_stream->PID_type = PID_TYPE_ELEMENTARY; /* set stream type */ /* hack for ITV HD (sid 10510, video pid 3401 */ if (program_number == 10510 && entry.PID == 3401 && stream_type == ST_PRIVATE_DATA) stream_type = ST_VIDEO_H264; ES_stream->stream_type = stream_type; ES_stream->flags &= ~MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN; /* init base and last time */ ES_stream->base_time = 0; ES_stream->last_time = 0; /* parse descriptor */ ES_stream->ES_info = gst_mpeg_descriptor_parse (data, ES_info_length); if (stream_type == ST_PRIVATE_SECTIONS) { /* not really an ES, so use section filter not pes filter */ /* initialise section filter */ GstCaps *caps; gst_section_filter_init (&ES_stream->section_filter); ES_stream->PID_type = PID_TYPE_PRIVATE_SECTION; ES_stream->pad = gst_pad_new_from_static_template (&private_template, g_strdup_printf ("private_%04x", entry.PID)); gst_pad_set_active (ES_stream->pad, TRUE); caps = gst_caps_new_simple ("application/x-mpegts-private-section", NULL); gst_pad_use_fixed_caps (ES_stream->pad); gst_pad_set_caps (ES_stream->pad, caps); gst_caps_unref (caps); gst_element_add_pad (GST_ELEMENT_CAST (demux), ES_stream->pad); } else { /* Recognise video streams based on stream_type */ if (gst_mpegts_stream_is_video (ES_stream)) ES_stream->flags |= MPEGTS_STREAM_FLAG_IS_VIDEO; /* set adaptor */ GST_LOG ("Initializing PES filter for PID %u", ES_stream->PID); gst_pes_filter_init (&ES_stream->filter, NULL, NULL); if (ES_stream->stream_type == ST_PRIVATE_DATA) { guint8 *dvb_sub_desc = gst_mpeg_descriptor_find (ES_stream->ES_info, DESC_DVB_SUBTITLING); /* enable gather PES for DVB subtitles since the dvbsuboverlay * expects complete PES packets */ if (dvb_sub_desc) { /* FIXME: There's another place where pes filters could get * initialized. Might need similar temporary hack there as well */ ES_stream->filter.gather_pes = TRUE; } } gst_pes_filter_set_callbacks (&ES_stream->filter, (GstPESFilterData) gst_mpegts_demux_data_cb, (GstPESFilterResync) gst_mpegts_demux_resync_cb, ES_stream); if (ES_stream->flags & MPEGTS_STREAM_FLAG_IS_VIDEO) ES_stream->filter.allow_unbounded = TRUE; ES_stream->PMT_pid = stream->PID; } } /* skip descriptor */ data += ES_info_length; GST_DEBUG_OBJECT (demux, " PMT stream_type: %02x, PID: 0x%04x (ES_info_len %d)", stream_type, entry.PID, ES_info_length); g_array_append_val (PMT->entries, entry); entries -= 5 + ES_info_length; } CRC = GST_READ_UINT32_BE (data); GST_DEBUG_OBJECT (demux, "PMT CRC: 0x%08x", CRC); if (demux->program_number == -1) { /* No program specified, take the first PMT */ if (demux->current_PMT == 0 || demux->current_PMT == stream->PID) gst_mpegts_activate_pmt (demux, stream); } else { /* Program specified, activate this if it matches */ if (demux->program_number == PMT->program_number) gst_mpegts_activate_pmt (demux, stream); } return TRUE; /* ERRORS */ wrong_crc: { GST_DEBUG_OBJECT (demux, "wrong crc"); return FALSE; } same_version: { GST_DEBUG_OBJECT (demux, "same version as existing PMT"); return TRUE; } wrong_program_number: { GST_DEBUG_OBJECT (demux, "PMT is for program number we don't care about"); return TRUE; } not_yet_applicable: { GST_DEBUG_OBJECT (demux, "Ignoring PMT with current_next_indicator = 0"); return TRUE; } wrong_id: { GST_DEBUG_OBJECT (demux, "expected table_id == 0, got 0x%02x", data[0]); return FALSE; } wrong_sync: { GST_DEBUG_OBJECT (demux, "expected sync 10, got %02x", data[0]); return FALSE; } wrong_seclen: { GST_DEBUG_OBJECT (demux, "first two bits of section length must be 0, got %02x", data[0]); return FALSE; } wrong_pilen: { GST_DEBUG_OBJECT (demux, "first two bits of program_info length must be 0, got %02x", data[0]); return FALSE; } wrong_esilen: { GST_DEBUG_OBJECT (demux, "first two bits of ES_info length must be 0, got %02x", data[0]); g_array_free (stream->PMT.entries, TRUE); stream->PMT.entries = NULL; gst_mpeg_descriptor_free (stream->PMT.program_info); stream->PMT.program_info = NULL; return FALSE; } } /* * private_section() { * table_id 8 uimsbf * section_syntax_indicator 1 bslbf * private_indicator 1 bslbf * reserved 2 bslbf * private_section_length 12 uimsbf * if (section_syntax_indicator == '0') { * for ( i=0;idemux; if (demux->check_crc) if (gst_mpegts_demux_calc_crc32 (data, datalen) != 0) goto wrong_crc; /* just dump this down the pad */ buffer = gst_buffer_new_and_alloc (datalen); memcpy (buffer->data, data, datalen); gst_pad_push (stream->pad, buffer); GST_DEBUG_OBJECT (demux, "parsing private section"); return TRUE; wrong_crc: { GST_DEBUG_OBJECT (demux, "wrong crc"); return FALSE; } } /* * adaptation_field() { * adaptation_field_length 8 uimsbf * if(adaptation_field_length >0) { * discontinuity_indicator 1 bslbf * random_access_indicator 1 bslbf * elementary_stream_priority_indicator 1 bslbf * PCR_flag 1 bslbf * OPCR_flag 1 bslbf * splicing_point_flag 1 bslbf * transport_private_data_flag 1 bslbf * adaptation_field_extension_flag 1 bslbf * if(PCR_flag == '1') { * program_clock_reference_base 33 uimsbf * reserved 6 bslbf * program_clock_reference_extension 9 uimsbf * } * if(OPCR_flag == '1') { * original_program_clock_reference_base 33 uimsbf * reserved 6 bslbf * original_program_clock_reference_extension 9 uimsbf * } * if (splicing_point_flag == '1') { * splice_countdown 8 tcimsbf * } * if(transport_private_data_flag == '1') { * transport_private_data_length 8 uimsbf * for (i=0; idemux; data_end = ((guint8 *) data) + data_len; length = *data++; if (G_UNLIKELY (length > data_len)) goto wrong_length; GST_DEBUG_OBJECT (demux, "parsing adaptation field, length %d", length); if (length > 0) { guint8 flags = *data++; GST_LOG_OBJECT (demux, "flags 0x%02x", flags); /* discontinuity flag */ if (flags & 0x80) { GST_DEBUG_OBJECT (demux, "discontinuity flag set"); } /* PCR_flag */ if (flags & 0x10) { guint32 pcr1; guint16 pcr2; guint64 pcr, pcr_ext; gboolean valid_pcr = TRUE; pcr1 = GST_READ_UINT32_BE (data); pcr2 = GST_READ_UINT16_BE (data + 4); pcr = ((guint64) pcr1) << 1; pcr |= (pcr2 & 0x8000) >> 15; pcr_ext = (pcr2 & 0x01ff); if (pcr_ext) pcr = (pcr * 300 + pcr_ext % 300) / 300; GST_DEBUG_OBJECT (demux, "have PCR %" G_GUINT64_FORMAT "(%" GST_TIME_FORMAT ") on PID 0x%04x " "and last pcr is %" G_GUINT64_FORMAT " (%" GST_TIME_FORMAT ")", pcr, GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (pcr)), stream->PID, stream->last_PCR, GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (stream->last_PCR))); /* pcr has been converted into units of 90Khz ticks * so assume discont if last pcr was > 900000 (10 second) lower */ if (stream->last_PCR != -1 && (pcr - stream->last_PCR > 900000 || pcr < stream->last_PCR)) { GstClockTimeDiff base_time_difference; GST_DEBUG_OBJECT (demux, "looks like we have a discont, this pcr should really be approx: %" G_GUINT64_FORMAT, stream->last_PCR + stream->last_PCR_difference); if (stream->discont_PCR == FALSE) { if (pcr > stream->last_PCR) { base_time_difference = -MPEGTIME_TO_GSTTIME ((pcr - (stream->last_PCR + stream->last_PCR_difference))); } else { base_time_difference = MPEGTIME_TO_GSTTIME ((stream->last_PCR + stream->last_PCR_difference) - pcr); } stream->discont_PCR = TRUE; stream->discont_difference = base_time_difference; valid_pcr = FALSE; } else { GstClockTimeDiff base_time_difference; /* need to update all pmt streams in case this pcr is pcr * for multiple programs */ int j; gboolean *pmts_checked = (gboolean *) & demux->pmts_checked; memset (pmts_checked, 0, sizeof (gboolean) * (MPEGTS_MAX_PID + 1)); for (j = 0; j < MPEGTS_MAX_PID + 1; j++) { if (demux->streams[j] && demux->streams[j]->PMT_pid <= MPEGTS_MAX_PID) { if (!pmts_checked[demux->streams[j]->PMT_pid]) { /* check if this is correct pcr for pmt */ if (demux->streams[demux->streams[j]->PMT_pid] && stream->PID == demux->streams[demux->streams[j]->PMT_pid]->PMT.PCR_PID) { /* checking the pcr discont is similar this second time * if similar, update the es pids * if not, assume it's a false discont due to corruption * or other */ if (pcr > stream->last_PCR) { base_time_difference = -MPEGTIME_TO_GSTTIME ((pcr - (stream->last_PCR + stream->last_PCR_difference))); } else { base_time_difference = MPEGTIME_TO_GSTTIME ((stream->last_PCR + stream->last_PCR_difference) - pcr); } if ((base_time_difference - stream->discont_difference > 0 && base_time_difference - stream->discont_difference < GST_SECOND * 10) || (stream->discont_difference - base_time_difference > 0 && stream->discont_difference - base_time_difference < GST_SECOND * 10)) { pmt_stream = demux->streams[demux->streams[j]->PMT_pid]; GST_DEBUG_OBJECT (demux, "Updating base_time on all es " "pids belonging to PMT 0x%02x", stream->PMT_pid); for (i = 0; i < pmt_stream->PMT.entries->len; i++) { GstMpegTSPMTEntry *cur_entry = &g_array_index (pmt_stream->PMT.entries, GstMpegTSPMTEntry, i); GST_DEBUG_OBJECT (demux, "Updating base time on " "pid 0x%02x by %" G_GINT64_FORMAT, cur_entry->PID, stream->discont_difference); if (cur_entry->PID <= MPEGTS_MAX_PID && demux->streams[cur_entry->PID]) { demux->streams[cur_entry->PID]->base_time += stream->discont_difference; } } } else { GST_DEBUG_OBJECT (demux, "last PCR discont looked to be " "bogus: previous discont difference %" G_GINT64_FORMAT " now %" G_GINT64_FORMAT, stream->discont_difference, base_time_difference); valid_pcr = FALSE; } } } pmts_checked[demux->streams[j]->PMT_pid] = TRUE; } } stream->discont_PCR = FALSE; stream->discont_difference = 0; } } else if (stream->last_PCR != -1) { if (stream->discont_PCR) { GST_DEBUG_OBJECT (demux, "last PCR discont looked to be bogus"); stream->discont_PCR = FALSE; stream->discont_difference = 0; } stream->last_PCR_difference = pcr - stream->last_PCR; } GST_DEBUG_OBJECT (demux, "valid pcr: %d last PCR difference: %" G_GUINT64_FORMAT, valid_pcr, stream->last_PCR_difference); if (valid_pcr) { GstMpegTSStream *PMT_stream = demux->streams[demux->current_PMT]; if (PMT_stream && PMT_stream->PMT.PCR_PID == stream->PID) { if (demux->pcr[0] == -1) { GST_DEBUG ("RECORDING pcr[0]:%" G_GUINT64_FORMAT, pcr); demux->pcr[0] = pcr; demux->num_packets = 0; } /* Considering a difference of 1 sec ie 90000 ticks */ else if (G_UNLIKELY (demux->pcr[1] == -1 && ((pcr - demux->pcr[0]) >= 90000))) { GST_DEBUG ("RECORDING pcr[1]:%" G_GUINT64_FORMAT, pcr); demux->pcr[1] = pcr; } } stream->last_PCR = pcr; if (demux->clock && demux->clock_base != GST_CLOCK_TIME_NONE) { gdouble r_squared; GstMpegTSStream *PMT_stream; /* for the reference start time we need to consult the PCR_PID of the * current PMT */ PMT_stream = demux->streams[demux->current_PMT]; if (PMT_stream->PMT.PCR_PID == stream->PID) { GST_LOG_OBJECT (demux, "internal %" GST_TIME_FORMAT " observation %" GST_TIME_FORMAT " pcr: %" G_GUINT64_FORMAT " base_pcr: %" G_GUINT64_FORMAT "pid: %d", GST_TIME_ARGS (gst_clock_get_internal_time (demux->clock)), GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (pcr) - MPEGTIME_TO_GSTTIME (stream->base_PCR) + stream->base_time + demux->clock_base), pcr, stream->base_PCR, stream->PID); gst_clock_add_observation (demux->clock, gst_clock_get_internal_time (demux->clock), demux->clock_base + stream->base_time + MPEGTIME_TO_GSTTIME (pcr) - MPEGTIME_TO_GSTTIME (stream->base_PCR), &r_squared); } } } data += 6; } /* OPCR_flag */ if (flags & 0x08) { guint32 opcr1; guint16 opcr2; guint64 opcr, opcr_ext; opcr1 = GST_READ_UINT32_BE (data); opcr2 = GST_READ_UINT16_BE (data + 4); opcr = ((guint64) opcr1) << 1; opcr |= (opcr2 & 0x8000) >> 15; opcr_ext = (opcr2 & 0x01ff); if (opcr_ext) opcr = (opcr * 300 + opcr_ext % 300) / 300; GST_DEBUG_OBJECT (demux, "have OPCR %" G_GUINT64_FORMAT " on PID 0x%04x", opcr, stream->PID); stream->last_OPCR = opcr; data += 6; } /* splicing_point_flag */ if (flags & 0x04) { guint8 splice_countdown; splice_countdown = *data++; GST_DEBUG_OBJECT (demux, "have splicing point, countdown %d", splice_countdown); } /* transport_private_data_flag */ if (flags & 0x02) { guint8 plength = *data++; if (data + plength > data_end) goto private_data_too_large; GST_DEBUG_OBJECT (demux, "have private data, length: %d", plength); data += plength; } /* adaptation_field_extension_flag */ if (flags & 0x01) { GST_DEBUG_OBJECT (demux, "have field extension"); } } *consumed = length + 1; return TRUE; /* ERRORS */ wrong_length: { GST_DEBUG_OBJECT (demux, "length %d > %d", length, data_len); return FALSE; } private_data_too_large: { GST_DEBUG_OBJECT (demux, "have too large a private data length"); return FALSE; } } /* * program_association_section() { * table_id 8 uimsbf == 0x00 * section_syntax_indicator 1 bslbf == 1 * '0' 1 bslbf == 0 * reserved 2 bslbf * section_length 12 uimsbf == 00xxxxx... * transport_stream_id 16 uimsbf * reserved 2 bslbf * version_number 5 uimsbf * current_next_indicator 1 bslbf * section_number 8 uimsbf * last_section_number 8 uimsbf * for (i=0; idemux; if (datalen < 8) return FALSE; if (*data++ != 0x00) goto wrong_id; if ((data[0] & 0xc0) != 0x80) goto wrong_sync; if (G_UNLIKELY ((data[0] & 0x0c) != 0x00)) goto wrong_seclen; data += 2; GST_DEBUG_OBJECT (demux, "PAT section_length: %d", datalen - 3); if (demux->check_crc) if (gst_mpegts_demux_calc_crc32 (data - 3, datalen) != 0) goto wrong_crc; PAT = &stream->PAT; version_number = (data[2] & 0x3e) >> 1; GST_DEBUG_OBJECT (demux, "PAT version_number: %d", version_number); if (G_UNLIKELY (version_number == PAT->version_number)) goto same_version; current_next_indicator = (data[2] & 0x01); GST_DEBUG_OBJECT (demux, "PAT current_next_indicator %d", current_next_indicator); if (current_next_indicator == 0) goto not_yet_applicable; PAT->version_number = version_number; PAT->current_next_indicator = current_next_indicator; PAT->transport_stream_id = GST_READ_UINT16_BE (data); data += 3; GST_DEBUG_OBJECT (demux, "PAT stream_id: %d", PAT->transport_stream_id); PAT->section_number = *data++; PAT->last_section_number = *data++; GST_DEBUG_OBJECT (demux, "PAT current_next_indicator: %d", PAT->current_next_indicator); GST_DEBUG_OBJECT (demux, "PAT section_number: %d", PAT->section_number); GST_DEBUG_OBJECT (demux, "PAT last_section_number: %d", PAT->last_section_number); /* 5 bytes after section length and a 4 bytes CRC, * the rest is 4 byte entries */ entries = (datalen - 3 - 9) / 4; if (PAT->entries) g_array_free (PAT->entries, TRUE); PAT->entries = g_array_sized_new (FALSE, TRUE, sizeof (GstMpegTSPATEntry), entries); while (entries--) { GstMpegTSPATEntry entry; GstMpegTSStream *PMT_stream; entry.program_number = GST_READ_UINT16_BE (data); data += 2; entry.PID = GST_READ_UINT16_BE (data); entry.PID &= 0x1fff; data += 2; /* get/create stream for PMT */ PMT_stream = gst_mpegts_demux_get_stream_for_PID (demux, entry.PID); if (PMT_stream->PID_type != PID_TYPE_PROGRAM_MAP) { /* set as program map */ PMT_stream->PID_type = PID_TYPE_PROGRAM_MAP; /* initialise section filter */ gst_section_filter_init (&PMT_stream->section_filter); } g_array_append_val (PAT->entries, entry); GST_DEBUG_OBJECT (demux, " PAT program: %d, PID 0x%04x", entry.program_number, entry.PID); } CRC = GST_READ_UINT32_BE (data); GST_DEBUG_OBJECT (demux, "PAT CRC: 0x%08x", CRC); /* PAT has been updated, signal the change */ g_object_notify ((GObject *) (demux), "pat-info"); return TRUE; /* ERRORS */ wrong_crc: { GST_DEBUG_OBJECT (demux, "wrong crc"); return FALSE; } same_version: { GST_DEBUG_OBJECT (demux, "same version as existing PAT"); return TRUE; } not_yet_applicable: { GST_DEBUG_OBJECT (demux, "Ignoring PAT with current_next_indicator = 0"); return TRUE; } wrong_id: { GST_DEBUG_OBJECT (demux, "expected table_id == 0, got %02x", data[0]); return FALSE; } wrong_sync: { GST_DEBUG_OBJECT (demux, "expected sync 10, got %02x", data[0]); return FALSE; } wrong_seclen: { GST_DEBUG_OBJECT (demux, "first two bits of section length must be 0, got %02x", data[0]); return FALSE; } } static gboolean gst_mpegts_demux_is_PMT (GstMpegTSDemux * demux, guint16 PID) { GstMpegTSStream *stream; GstMpegTSPAT *PAT; gint i; /* get the PAT */ stream = demux->streams[PID_PROGRAM_ASSOCIATION_TABLE]; if (stream == NULL || stream->PAT.entries == NULL) return FALSE; PAT = &stream->PAT; for (i = 0; i < PAT->entries->len; i++) { GstMpegTSPATEntry *entry; entry = &g_array_index (PAT->entries, GstMpegTSPATEntry, i); if (!entry) continue; if (entry->PID == PID) return TRUE; } return FALSE; } static FORCE_INLINE GstFlowReturn gst_mpegts_stream_pes_buffer_flush (GstMpegTSStream * stream, gboolean discard) { GstFlowReturn ret = GST_FLOW_OK; if (stream->pes_buffer) { if (discard) { gst_buffer_unref (stream->pes_buffer); stream->pes_buffer_in_sync = FALSE; } else { GST_BUFFER_SIZE (stream->pes_buffer) = stream->pes_buffer_used; ret = gst_pes_filter_push (&stream->filter, stream->pes_buffer); if (ret == GST_FLOW_LOST_SYNC) stream->pes_buffer_in_sync = FALSE; } stream->pes_buffer = NULL; } return ret; } static FORCE_INLINE GstFlowReturn gst_mpegts_stream_pes_buffer_push (GstMpegTSStream * stream, const guint8 * in_data, guint in_size) { GstFlowReturn ret = GST_FLOW_OK; guint8 *out_data; if (G_UNLIKELY (stream->pes_buffer && stream->pes_buffer_used + in_size > stream->pes_buffer_size)) { GST_DEBUG ("stream with PID 0x%04x have PES buffer full at %u bytes." " Flushing and growing the buffer", stream->PID, stream->pes_buffer_size); stream->pes_buffer_overflow = TRUE; if (stream->pes_buffer_size < (MPEGTS_MAX_PES_BUFFER_SIZE >> 1)) stream->pes_buffer_size <<= 1; ret = gst_mpegts_stream_pes_buffer_flush (stream, FALSE); if (ret == GST_FLOW_LOST_SYNC) goto done; } if (G_UNLIKELY (!stream->pes_buffer)) { /* set initial size of PES buffer */ if (G_UNLIKELY (stream->pes_buffer_size == 0)) stream->pes_buffer_size = MPEGTS_MIN_PES_BUFFER_SIZE; stream->pes_buffer = gst_buffer_new_and_alloc (stream->pes_buffer_size); stream->pes_buffer_used = 0; } out_data = GST_BUFFER_DATA (stream->pes_buffer) + stream->pes_buffer_used; memcpy (out_data, in_data, in_size); stream->pes_buffer_used += in_size; done: return ret; } static FORCE_INLINE GstFlowReturn gst_mpegts_demux_pes_buffer_flush (GstMpegTSDemux * demux, gboolean discard) { gint i; GstFlowReturn ret = GST_FLOW_OK; for (i = 0; i < MPEGTS_MAX_PID + 1; i++) { GstMpegTSStream *stream = demux->streams[i]; if (stream && stream->pad) { gst_mpegts_stream_pes_buffer_flush (stream, discard); stream->pes_buffer_in_sync = FALSE; } } return ret; } static FORCE_INLINE GstFlowReturn gst_mpegts_demux_push_fragment (GstMpegTSStream * stream, const guint8 * in_data, guint in_size) { GstFlowReturn ret; GstBuffer *es_buf = gst_buffer_new_and_alloc (in_size); memcpy (GST_BUFFER_DATA (es_buf), in_data, in_size); ret = gst_pes_filter_push (&stream->filter, es_buf); /* If PES filter return ok then PES fragment buffering * can be enabled */ if (ret == GST_FLOW_OK) stream->pes_buffer_in_sync = TRUE; else if (ret == GST_FLOW_LOST_SYNC) stream->pes_buffer_in_sync = FALSE; return ret; } /* * transport_packet(){ * sync_byte 8 bslbf == 0x47 * transport_error_indicator 1 bslbf * payload_unit_start_indicator 1 bslbf * transport _priority 1 bslbf * PID 13 uimsbf * transport_scrambling_control 2 bslbf * adaptation_field_control 2 bslbf * continuity_counter 4 uimsbf * if(adaptation_field_control=='10' || adaptation_field_control=='11'){ * adaptation_field() * } * if(adaptation_field_control=='01' || adaptation_field_control=='11') { * for (i=0;iPID; transport_scrambling_control = (data[2] & 0xc0) >> 6; adaptation_field_control = (data[2] & 0x30) >> 4; continuity_counter = data[2] & 0x0f; data += 3; datalen -= 3; GST_LOG_OBJECT (demux, "afc 0x%x, pusi %d, PID 0x%04x datalen %u", adaptation_field_control, payload_unit_start_indicator, PID, datalen); ret = GST_FLOW_OK; /* packets with adaptation_field_control == 0 must be skipped */ if (adaptation_field_control == 0) goto skip; /* parse adaption field if any */ if (adaptation_field_control & 0x2) { guint consumed; if (!gst_mpegts_demux_parse_adaptation_field (stream, data, datalen, &consumed)) goto done; if (datalen <= consumed) goto too_small; data += consumed; datalen -= consumed; GST_LOG_OBJECT (demux, "consumed: %u datalen: %u", consumed, datalen); } /* If this packet has a payload, handle it */ if (adaptation_field_control & 0x1) { GST_LOG_OBJECT (demux, "Packet payload %d bytes, PID 0x%04x", datalen, PID); /* For unknown streams, check if the PID is in the partial PIDs * list as an elementary stream and override the type if so */ if (G_UNLIKELY (stream->PID_type == PID_TYPE_UNKNOWN)) { if (mpegts_is_elem_pid (demux, PID)) { GST_DEBUG_OBJECT (demux, "PID 0x%04x is an elementary stream in the PID list", PID); stream->PID_type = PID_TYPE_ELEMENTARY; stream->flags |= MPEGTS_STREAM_FLAG_STREAM_TYPE_UNKNOWN; stream->base_time = 0; stream->last_time = 0; /* Clear any existing descriptor */ if (stream->ES_info) { gst_mpeg_descriptor_free (stream->ES_info); stream->ES_info = NULL; } /* Initialise our PES filter */ GST_LOG ("Initializing PES filter for PID %u", stream->PID); gst_pes_filter_init (&stream->filter, NULL, NULL); gst_pes_filter_set_callbacks (&stream->filter, (GstPESFilterData) gst_mpegts_demux_data_cb, (GstPESFilterResync) gst_mpegts_demux_resync_cb, stream); } } /* now parse based on the stream type */ switch (stream->PID_type) { case PID_TYPE_PROGRAM_ASSOCIATION: case PID_TYPE_CONDITIONAL_ACCESS: case PID_TYPE_PROGRAM_MAP: case PID_TYPE_PRIVATE_SECTION: { GstBuffer *sec_buf; guint8 *section_data; guint16 section_length; guint8 pointer; /* do stuff with our section */ if (payload_unit_start_indicator) { pointer = *data++; datalen -= 1; if (pointer >= datalen) { GST_DEBUG_OBJECT (demux, "pointer: 0x%02x too large", pointer); return GST_FLOW_OK; } data += pointer; datalen -= pointer; } /* FIXME: try to use data directly instead of creating a buffer and pushing in into adapter at section filter */ sec_buf = gst_buffer_new_and_alloc (datalen); memcpy (GST_BUFFER_DATA (sec_buf), data, datalen); if (gst_section_filter_push (&stream->section_filter, payload_unit_start_indicator, continuity_counter, sec_buf)) { GST_DEBUG_OBJECT (demux, "section finished"); /* section ready */ section_length = stream->section_filter.section_length; section_data = (guint8 *) gst_adapter_peek (stream->section_filter.adapter, section_length + 3); switch (stream->PID_type) { case PID_TYPE_PROGRAM_ASSOCIATION: gst_mpegts_stream_parse_pat (stream, section_data, section_length + 3); break; case PID_TYPE_CONDITIONAL_ACCESS: gst_mpegts_stream_parse_cat (stream, section_data, section_length + 3); break; case PID_TYPE_PROGRAM_MAP: gst_mpegts_stream_parse_pmt (stream, section_data, section_length + 3); break; case PID_TYPE_PRIVATE_SECTION: gst_mpegts_stream_parse_private_section (stream, section_data, section_length + 3); break; } gst_section_filter_clear (&stream->section_filter); } else { /* section still going, don't parse left */ GST_DEBUG_OBJECT (demux, "section still going for PID 0x%04x", PID); } break; } case PID_TYPE_NULL_PACKET: GST_DEBUG_OBJECT (demux, "skipping PID 0x%04x, type 0x%04x (NULL packet)", PID, stream->PID_type); break; case PID_TYPE_UNKNOWN: GST_DEBUG_OBJECT (demux, "skipping unknown PID 0x%04x, type 0x%04x", PID, stream->PID_type); break; case PID_TYPE_ELEMENTARY: { if (payload_unit_start_indicator) { GST_DEBUG_OBJECT (demux, "new PES start for PID 0x%04x, used %u " "bytes of %u bytes in the PES buffer", PID, stream->pes_buffer_used, stream->pes_buffer_size); /* Flush buffered PES data */ gst_mpegts_stream_pes_buffer_flush (stream, FALSE); gst_pes_filter_drain (&stream->filter); /* Resize the buffer to half if no overflow detected and * had been used less than half of it */ if (stream->pes_buffer_overflow == FALSE && stream->pes_buffer_used < (stream->pes_buffer_size >> 1)) { stream->pes_buffer_size >>= 1; if (stream->pes_buffer_size < MPEGTS_MIN_PES_BUFFER_SIZE) stream->pes_buffer_size = MPEGTS_MIN_PES_BUFFER_SIZE; GST_DEBUG_OBJECT (demux, "PES buffer size reduced to %u bytes", stream->pes_buffer_size); } /* mark the stream not in sync to give a chance on PES filter to * detect lost sync */ stream->pes_buffer_in_sync = FALSE; stream->pes_buffer_overflow = FALSE; } GST_LOG_OBJECT (demux, "Elementary packet of size %u for PID 0x%04x", datalen, PID); if (datalen > 0) { if (!stream->pes_buffer_in_sync) { /* Push the first fragment to PES filter to have a chance to * detect GST_FLOW_LOST_SYNC. */ GST_LOG_OBJECT (demux, "fragment directly pushed to PES filter"); ret = gst_mpegts_demux_push_fragment (stream, data, datalen); } else { /* Otherwhise we buffer the PES fragment */ ret = gst_mpegts_stream_pes_buffer_push (stream, data, datalen); /* If sync is lost here is due a pes_buffer_flush and we can try * to resync in the PES filter with the current fragment */ if (ret == GST_FLOW_LOST_SYNC) { GST_LOG_OBJECT (demux, "resync, fragment pushed to PES filter"); ret = gst_mpegts_demux_push_fragment (stream, data, datalen); } } break; } else { GST_WARNING_OBJECT (demux, "overflow of datalen: %u so skipping", datalen); return GST_FLOW_OK; } } } } done: return ret; skip: { GST_DEBUG_OBJECT (demux, "skipping, adaptation_field_control == 0"); return GST_FLOW_OK; } too_small: { GST_DEBUG_OBJECT (demux, "skipping, adaptation_field consumed all data"); return GST_FLOW_OK; } } static FORCE_INLINE GstFlowReturn gst_mpegts_demux_parse_transport_packet (GstMpegTSDemux * demux, const guint8 * data) { GstFlowReturn ret = GST_FLOW_OK; guint16 PID; GstMpegTSStream *stream; /* skip sync byte */ data++; /* get PID */ PID = ((data[0] & 0x1f) << 8) | data[1]; /* Skip NULL packets */ if (G_UNLIKELY (PID == 0x1fff)) goto beach; /* get the stream. */ stream = gst_mpegts_demux_get_stream_for_PID (demux, PID); /* parse the stream */ ret = gst_mpegts_demux_parse_stream (demux, stream, data, MPEGTS_NORMAL_TS_PACKETSIZE - 1); if (demux->pcr[1] != -1 && demux->bitrate == -1) { guint64 bitrate; GST_DEBUG_OBJECT (demux, "pcr[0]:%" G_GUINT64_FORMAT, demux->pcr[0]); GST_DEBUG_OBJECT (demux, "pcr[1]:%" G_GUINT64_FORMAT, demux->pcr[1]); GST_DEBUG_OBJECT (demux, "diff in time %" GST_TIME_FORMAT, GST_TIME_ARGS (MPEGTIME_TO_GSTTIME (demux->pcr[1] - demux->pcr[0]))); GST_DEBUG_OBJECT (demux, "stream->last_PCR_difference: %" G_GUINT64_FORMAT ", demux->num_packets %" G_GUINT64_FORMAT, demux->pcr[1] - demux->pcr[0], demux->num_packets); bitrate = gst_util_uint64_scale (GST_SECOND, MPEGTS_NORMAL_TS_PACKETSIZE * demux->num_packets, MPEGTIME_TO_GSTTIME (demux->pcr[1] - demux->pcr[0])); /* somehow... I doubt a bitrate below one packet per second is valid */ if (bitrate > MPEGTS_NORMAL_TS_PACKETSIZE - 1) { demux->bitrate = bitrate; GST_DEBUG_OBJECT (demux, "bitrate is %" G_GINT64_FORMAT " bytes per second", demux->bitrate); } else { GST_WARNING_OBJECT (demux, "Couldn't compute valid bitrate, recomputing"); demux->pcr[0] = demux->pcr[1] = -1; demux->num_packets = -1; } } beach: demux->num_packets++; return ret; /* ERRORS */ } static gboolean gst_mpegts_demux_handle_seek_push (GstMpegTSDemux * demux, GstEvent * event) { gboolean res = FALSE; gdouble rate; GstFormat format; GstSeekFlags flags; GstSeekType start_type, stop_type; gint64 start, stop, bstart, bstop; GstEvent *bevent; gst_event_parse_seek (event, &rate, &format, &flags, &start_type, &start, &stop_type, &stop); GST_DEBUG_OBJECT (demux, "seek event, rate: %f start: %" GST_TIME_FORMAT " stop: %" GST_TIME_FORMAT, rate, GST_TIME_ARGS (start), GST_TIME_ARGS (stop)); if (format == GST_FORMAT_BYTES) { GST_DEBUG_OBJECT (demux, "seek not supported on format %d", format); goto beach; } GST_DEBUG_OBJECT (demux, "seek - trying directly upstream first"); /* first try original format seek */ res = gst_pad_push_event (demux->sinkpad, gst_event_ref (event)); if (res == TRUE) goto beach; GST_DEBUG_OBJECT (demux, "seek - no upstream"); if (format != GST_FORMAT_TIME) { /* From here down, we only support time based seeks */ GST_DEBUG_OBJECT (demux, "seek not supported on format %d", format); goto beach; } /* We need to convert to byte based seek and we need a scr_rate for that. */ if (demux->bitrate == -1) { GST_DEBUG_OBJECT (demux, "seek not possible, no bitrate"); goto beach; } GST_DEBUG_OBJECT (demux, "try with bitrate"); bstart = GSTTIME_TO_BYTES (start); bstop = GSTTIME_TO_BYTES (stop); GST_DEBUG_OBJECT (demux, "in bytes bstart %" G_GINT64_FORMAT " bstop %" G_GINT64_FORMAT, bstart, bstop); bevent = gst_event_new_seek (rate, GST_FORMAT_BYTES, flags, start_type, bstart, stop_type, bstop); res = gst_pad_push_event (demux->sinkpad, bevent); beach: gst_event_unref (event); return res; } static gboolean gst_mpegts_demux_src_event (GstPad * pad, GstEvent * event) { GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad)); gboolean res = FALSE; GST_DEBUG_OBJECT (demux, "got event %s", gst_event_type_get_name (GST_EVENT_TYPE (event))); switch (GST_EVENT_TYPE (event)) { case GST_EVENT_SEEK: res = gst_mpegts_demux_handle_seek_push (demux, event); break; default: res = gst_pad_push_event (demux->sinkpad, event); break; } gst_object_unref (demux); return res; } static void gst_mpegts_demux_flush (GstMpegTSDemux * demux, gboolean discard) { gint i; GstMpegTSStream *PCR_stream; GstMpegTSStream *PMT_stream; GST_DEBUG_OBJECT (demux, "flushing MPEG TS demuxer (discard %d)", discard); /* Start by flushing internal buffers */ gst_mpegts_demux_pes_buffer_flush (demux, discard); /* Clear adapter */ gst_adapter_clear (demux->adapter); /* Try resetting the last_PCR value as we will have a discont */ if (demux->current_PMT == 0) goto beach; PMT_stream = demux->streams[demux->current_PMT]; if (PMT_stream == NULL) goto beach; PCR_stream = demux->streams[PMT_stream->PMT.PCR_PID]; if (PCR_stream == NULL) goto beach; PCR_stream->last_PCR = -1; /* Reset last time of all streams */ for (i = 0; i < MPEGTS_MAX_PID + 1; i++) { GstMpegTSStream *stream = demux->streams[i]; if (stream) { stream->last_time = 0; stream->discont = TRUE; } } beach: return; } static gboolean gst_mpegts_demux_send_event (GstMpegTSDemux * demux, GstEvent * event) { gint i; gboolean have_stream = FALSE, res = TRUE; for (i = 0; i < MPEGTS_MAX_PID + 1; i++) { GstMpegTSStream *stream = demux->streams[i]; if (stream && stream->pad) { res &= gst_pad_push_event (stream->pad, gst_event_ref (event)); have_stream = TRUE; } } gst_event_unref (event); return have_stream; } static gboolean gst_mpegts_demux_sink_event (GstPad * pad, GstEvent * event) { GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad)); gboolean res = FALSE; GST_DEBUG_OBJECT (demux, "got event %s", gst_event_type_get_name (GST_EVENT_TYPE (event))); switch (GST_EVENT_TYPE (event)) { case GST_EVENT_FLUSH_START: res = gst_mpegts_demux_send_event (demux, event); break; case GST_EVENT_FLUSH_STOP: gst_adapter_clear (demux->adapter); gst_mpegts_demux_flush (demux, TRUE); res = gst_mpegts_demux_send_event (demux, event); break; case GST_EVENT_EOS: gst_mpegts_demux_flush (demux, FALSE); /* Send the EOS event on each stream */ if (!(res = gst_mpegts_demux_send_event (demux, event))) { /* we have no streams */ GST_ELEMENT_ERROR (demux, STREAM, TYPE_NOT_FOUND, (NULL), ("No valid streams found at EOS")); } break; case GST_EVENT_NEWSEGMENT: { gboolean update; gdouble rate; GstFormat format; gint64 start, stop, time; gst_event_parse_new_segment (event, &update, &rate, &format, &start, &stop, &time); gst_event_unref (event); GST_INFO_OBJECT (demux, "received new segment: rate %g " "format %d, start: %" G_GINT64_FORMAT ", stop: %" G_GINT64_FORMAT ", time: %" G_GINT64_FORMAT, rate, format, start, stop, time); if (format == GST_FORMAT_BYTES && demux->bitrate != -1) { gint64 tstart = 0, tstop = 0, pos = 0; if (demux->base_pts != GST_CLOCK_TIME_NONE) { tstart = tstop = demux->base_pts; } tstart += BYTES_TO_GSTTIME (start); tstop += BYTES_TO_GSTTIME (stop); pos = BYTES_TO_GSTTIME (time); event = gst_event_new_new_segment (update, rate, GST_FORMAT_TIME, tstart, tstop, pos); GST_DEBUG_OBJECT (demux, "pushing time newsegment from %" GST_TIME_FORMAT " to %" GST_TIME_FORMAT " pos %" GST_TIME_FORMAT, GST_TIME_ARGS (tstart), GST_TIME_ARGS (tstop), GST_TIME_ARGS (pos)); res = gst_mpegts_demux_send_event (demux, event); } break; } default: res = gst_mpegts_demux_send_event (demux, event); break; } gst_object_unref (demux); return res; } static gboolean gst_mpegts_demux_provides_clock (GstElement * element) { GstMpegTSDemux *demux; GstQuery *query; gboolean is_live = FALSE; GstPad *peer; demux = GST_MPEGTS_DEMUX (element); query = gst_query_new_latency (); peer = gst_pad_get_peer (demux->sinkpad); if (peer) { if (gst_pad_query (peer, query)) gst_query_parse_latency (query, &is_live, NULL, NULL); gst_object_unref (peer); } gst_query_unref (query); return is_live; } static GstClock * gst_mpegts_demux_provide_clock (GstElement * element) { GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (element); if (gst_mpegts_demux_provides_clock (element)) { if (demux->clock == NULL) { demux->clock = g_object_new (GST_TYPE_SYSTEM_CLOCK, "name", "MpegTSClock", NULL); demux->clock_base = GST_CLOCK_TIME_NONE; } return gst_object_ref (demux->clock); } return NULL; } static const GstQueryType * gst_mpegts_demux_src_pad_query_type (GstPad * pad) { static const GstQueryType types[] = { GST_QUERY_LATENCY, GST_QUERY_DURATION, GST_QUERY_SEEKING, 0 }; return types; } static gboolean gst_mpegts_demux_src_pad_query (GstPad * pad, GstQuery * query) { GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad)); gboolean res = FALSE; GstPad *peer; switch (GST_QUERY_TYPE (query)) { case GST_QUERY_LATENCY: { peer = gst_pad_get_peer (demux->sinkpad); if (peer) { res = gst_pad_query (peer, query); if (res) { gboolean is_live; GstClockTime min_latency, max_latency; gst_query_parse_latency (query, &is_live, &min_latency, &max_latency); if (is_live) { min_latency += TS_LATENCY * GST_MSECOND; if (max_latency != GST_CLOCK_TIME_NONE) max_latency += TS_LATENCY * GST_MSECOND; } gst_query_set_latency (query, is_live, min_latency, max_latency); } gst_object_unref (peer); } break; } case GST_QUERY_DURATION: { GstFormat format; GstPad *peer; gst_query_parse_duration (query, &format, NULL); /* Try query upstream first */ peer = gst_pad_get_peer (demux->sinkpad); if (peer) { res = gst_pad_query (peer, query); /* Try doing something with that query if it failed */ if (!res && format == GST_FORMAT_TIME && demux->bitrate != -1) { /* Try using cache first */ if (GST_CLOCK_TIME_IS_VALID (demux->cache_duration)) { GST_LOG_OBJECT (demux, "replying duration query from cache %" GST_TIME_FORMAT, GST_TIME_ARGS (demux->cache_duration)); gst_query_set_duration (query, GST_FORMAT_TIME, demux->cache_duration); res = TRUE; } else { /* Query upstream and approximate */ GstQuery *bquery = gst_query_new_duration (GST_FORMAT_BYTES); gint64 duration = 0; /* Query peer for duration in bytes */ res = gst_pad_query (peer, bquery); if (res) { /* Convert to time format */ gst_query_parse_duration (bquery, &format, &duration); GST_DEBUG_OBJECT (demux, "query on peer pad reported bytes %" G_GUINT64_FORMAT, duration); demux->cache_duration = BYTES_TO_GSTTIME (duration); GST_DEBUG_OBJECT (demux, "converted to time %" GST_TIME_FORMAT, GST_TIME_ARGS (demux->cache_duration)); gst_query_set_duration (query, GST_FORMAT_TIME, demux->cache_duration); } gst_query_unref (bquery); } } else { GST_WARNING_OBJECT (demux, "unsupported query format or no bitrate " "yet to approximate duration from bytes"); } gst_object_unref (peer); } break; } case GST_QUERY_SEEKING:{ GstFormat fmt; gst_query_parse_seeking (query, &fmt, NULL, NULL, NULL); if (fmt == GST_FORMAT_BYTES) { /* Seeking in BYTES format not supported at all */ gst_query_set_seeking (query, fmt, FALSE, -1, -1); } else { GstQuery *peerquery; gboolean seekable; /* Then ask upstream */ res = gst_pad_peer_query (demux->sinkpad, query); if (res) { /* If upstream can handle seeks we're done, if it * can't we still have our TIME->BYTES conversion seek */ gst_query_parse_seeking (query, NULL, &seekable, NULL, NULL); if (seekable || fmt != GST_FORMAT_TIME) goto beach; } /* We can't say anything about seekability if we didn't * have a second PCR yet because the bitrate is calculated * from this */ if (demux->bitrate == -1 && demux->pcr[1] == -1) goto beach; /* We can seek if upstream supports BYTES seeks and we * have a bitrate */ peerquery = gst_query_new_seeking (GST_FORMAT_BYTES); res = gst_pad_peer_query (demux->sinkpad, peerquery); if (!res || demux->bitrate == -1) { gst_query_set_seeking (query, fmt, FALSE, -1, -1); } else { gst_query_parse_seeking (peerquery, NULL, &seekable, NULL, NULL); if (seekable) gst_query_set_seeking (query, GST_FORMAT_TIME, TRUE, 0, -1); else gst_query_set_seeking (query, fmt, FALSE, -1, -1); } gst_query_unref (peerquery); res = TRUE; } break; } default: res = gst_pad_query_default (pad, query); break; } beach: gst_object_unref (demux); return res; } static FORCE_INLINE gint is_mpegts_sync (const guint8 * in_data, const guint8 * end_data, guint packetsize) { guint ret = 0; if (G_LIKELY (IS_MPEGTS_SYNC (in_data))) return 100; if (in_data + packetsize < end_data - 5) { if (G_LIKELY (IS_MPEGTS_SYNC (in_data + packetsize))) ret += 50; } if (in_data[0] == 0x47) { ret += 25; if ((in_data[1] & 0x80) == 0x00) ret += 10; if ((in_data[3] & 0x10) == 0x10) ret += 5; } return ret; } static inline void gst_mpegts_demux_detect_packet_size (GstMpegTSDemux * demux, guint len) { guint i, packetsize = 0; for (i = 1; i < len; i++) { packetsize = demux->sync_lut[i] - demux->sync_lut[i - 1]; if (packetsize == MPEGTS_NORMAL_TS_PACKETSIZE || packetsize == MPEGTS_M2TS_TS_PACKETSIZE || packetsize == MPEGTS_DVB_ASI_TS_PACKETSIZE || packetsize == MPEGTS_ATSC_TS_PACKETSIZE) goto done; else packetsize = 0; } done: demux->packetsize = (packetsize ? packetsize : MPEGTS_NORMAL_TS_PACKETSIZE); GST_DEBUG_OBJECT (demux, "packet_size set to %d bytes", demux->packetsize); } static FORCE_INLINE guint gst_mpegts_demux_sync_scan (GstMpegTSDemux * demux, const guint8 * in_data, guint size, guint * flush) { guint sync_count = 0; const guint8 *end_scan = in_data + size - demux->packetsize; guint8 *ptr_data = (guint8 *) in_data; guint packetsize = (demux->packetsize ? demux->packetsize : MPEGTS_NORMAL_TS_PACKETSIZE); /* Check if the LUT table is big enough */ if (G_UNLIKELY (demux->sync_lut_len < (size / packetsize))) { demux->sync_lut_len = size / packetsize; if (demux->sync_lut) g_free (demux->sync_lut); demux->sync_lut = g_new0 (guint8 *, demux->sync_lut_len); GST_DEBUG_OBJECT (demux, "created sync LUT table with %u entries", demux->sync_lut_len); } while (ptr_data <= end_scan && sync_count < demux->sync_lut_len) { /* if sync code is found try to store it in the LUT */ guint chance = is_mpegts_sync (ptr_data, end_scan, packetsize); if (G_LIKELY (chance > 50)) { /* skip paketsize bytes and try find next */ guint8 *next_sync = ptr_data + packetsize; if (next_sync < end_scan) { demux->sync_lut[sync_count] = ptr_data; sync_count++; ptr_data += packetsize; } else goto done; } else { ptr_data++; } } done: if (G_UNLIKELY (!demux->packetsize)) gst_mpegts_demux_detect_packet_size (demux, sync_count); *flush = MIN (ptr_data - in_data, size); return sync_count; } static GstFlowReturn gst_mpegts_demux_chain (GstPad * pad, GstBuffer * buffer) { GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (gst_pad_get_parent (pad)); GstFlowReturn ret = GST_FLOW_OK; const guint8 *data; guint avail; guint flush = 0; gint i; guint sync_count; if (GST_BUFFER_IS_DISCONT (buffer)) { gst_mpegts_demux_flush (demux, FALSE); } /* first push the new buffer into the adapter */ gst_adapter_push (demux->adapter, buffer); /* check if there's enough data to parse a packet */ avail = gst_adapter_available (demux->adapter); if (G_UNLIKELY (avail < demux->packetsize)) goto done; /* recover all data from adapter */ data = gst_adapter_peek (demux->adapter, avail); /* scan for sync codes */ sync_count = gst_mpegts_demux_sync_scan (demux, data, avail, &flush); /* process all packets */ for (i = 0; i < sync_count; i++) { ret = gst_mpegts_demux_parse_transport_packet (demux, demux->sync_lut[i]); if (G_UNLIKELY (ret == GST_FLOW_LOST_SYNC || ret == GST_FLOW_NEED_MORE_DATA)) { ret = GST_FLOW_OK; continue; } if (G_UNLIKELY (ret != GST_FLOW_OK)) { flush = demux->sync_lut[i] - data + demux->packetsize; flush = MIN (avail, flush); goto done; } } done: /* flush processed data */ if (flush) { GST_DEBUG_OBJECT (demux, "flushing %d/%d", flush, avail); gst_adapter_flush (demux->adapter, flush); } gst_object_unref (demux); return ret; } static GstStateChangeReturn gst_mpegts_demux_change_state (GstElement * element, GstStateChange transition) { GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (element); GstStateChangeReturn result; switch (transition) { case GST_STATE_CHANGE_NULL_TO_READY: demux->adapter = gst_adapter_new (); break; case GST_STATE_CHANGE_READY_TO_PAUSED: break; default: break; } result = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition); switch (transition) { case GST_STATE_CHANGE_PAUSED_TO_READY: gst_mpegts_demux_reset (demux); break; case GST_STATE_CHANGE_READY_TO_NULL: g_object_unref (demux->adapter); if (demux->sync_lut) g_free (demux->sync_lut); demux->sync_lut = NULL; demux->sync_lut_len = 0; break; default: break; } return result; } static GValueArray * mpegts_demux_build_pat_info (GstMpegTSDemux * demux) { GValueArray *vals = NULL; GstMpegTSPAT *PAT; gint i; g_return_val_if_fail (demux->streams[0] != NULL, NULL); g_return_val_if_fail (demux->streams[0]->PID_type == PID_TYPE_PROGRAM_ASSOCIATION, NULL); PAT = &(demux->streams[0]->PAT); vals = g_value_array_new (PAT->entries->len); for (i = 0; i < PAT->entries->len; i++) { GstMpegTSPATEntry *cur_entry = &g_array_index (PAT->entries, GstMpegTSPATEntry, i); GValue v = { 0, }; MpegTsPatInfo *info_obj; info_obj = mpegts_pat_info_new (cur_entry->program_number, cur_entry->PID); g_value_init (&v, G_TYPE_OBJECT); g_value_take_object (&v, info_obj); g_value_array_append (vals, &v); g_value_unset (&v); } return vals; } static MpegTsPmtInfo * mpegts_demux_build_pmt_info (GstMpegTSDemux * demux, guint16 pmt_pid) { MpegTsPmtInfo *info_obj; GstMpegTSPMT *PMT; gint i; g_return_val_if_fail (demux->streams[pmt_pid] != NULL, NULL); g_return_val_if_fail (demux->streams[pmt_pid]->PID_type == PID_TYPE_PROGRAM_MAP, NULL); PMT = &(demux->streams[pmt_pid]->PMT); info_obj = mpegts_pmt_info_new (PMT->program_number, PMT->PCR_PID, PMT->version_number); for (i = 0; i < PMT->entries->len; i++) { GstMpegTSStream *stream; MpegTsPmtStreamInfo *stream_info; GstMpegTSPMTEntry *cur_entry = &g_array_index (PMT->entries, GstMpegTSPMTEntry, i); stream = demux->streams[cur_entry->PID]; stream_info = mpegts_pmt_stream_info_new (cur_entry->PID, stream->stream_type); if (stream->ES_info) { int i; /* add languages */ guint8 *iso639_languages = gst_mpeg_descriptor_find (stream->ES_info, DESC_ISO_639_LANGUAGE); if (iso639_languages) { for (i = 0; i < DESC_ISO_639_LANGUAGE_codes_n (iso639_languages); i++) { gchar *language_n = (gchar *) DESC_ISO_639_LANGUAGE_language_code_nth (iso639_languages, i); mpegts_pmt_stream_info_add_language (stream_info, g_strndup (language_n, 3)); } } for (i = 0; i < gst_mpeg_descriptor_n_desc (stream->ES_info); ++i) { guint8 *desc = gst_mpeg_descriptor_nth (stream->ES_info, i); /* add the whole descriptor, tag + length + DESC_LENGTH bytes */ mpegts_pmt_stream_info_add_descriptor (stream_info, (gchar *) desc, 2 + DESC_LENGTH (desc)); } } mpegts_pmt_info_add_stream (info_obj, stream_info); } return info_obj; } static void gst_mpegts_demux_set_property (GObject * object, guint prop_id, const GValue * value, GParamSpec * pspec) { GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (object); gchar **pids; guint num_pids; int i; switch (prop_id) { case PROP_ES_PIDS: pids = g_strsplit (g_value_get_string (value), ":", -1); num_pids = g_strv_length (pids); if (num_pids > 0) { demux->elementary_pids = g_new0 (guint16, num_pids); demux->nb_elementary_pids = num_pids; for (i = 0; i < num_pids; i++) { demux->elementary_pids[i] = strtol (pids[i], NULL, 0); GST_INFO ("partial TS ES pid %d", demux->elementary_pids[i]); } } g_strfreev (pids); break; case PROP_CHECK_CRC: demux->check_crc = g_value_get_boolean (value); break; case PROP_PROGRAM_NUMBER: demux->program_number = g_value_get_int (value); break; default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } static void gst_mpegts_demux_get_property (GObject * object, guint prop_id, GValue * value, GParamSpec * pspec) { GstMpegTSDemux *demux = GST_MPEGTS_DEMUX (object); int i; switch (prop_id) { case PROP_ES_PIDS: if (demux->nb_elementary_pids == 0) { g_value_set_static_string (value, ""); } else { GString *ts_pids; ts_pids = g_string_sized_new (32); /* FIXME: align with property description which uses hex numbers? */ g_string_append_printf (ts_pids, "%d", demux->elementary_pids[0]); for (i = 1; i < demux->nb_elementary_pids; i++) { g_string_append_printf (ts_pids, ":%d", demux->elementary_pids[i]); } g_value_take_string (value, g_string_free (ts_pids, FALSE)); } break; case PROP_CHECK_CRC: g_value_set_boolean (value, demux->check_crc); break; case PROP_PROGRAM_NUMBER: g_value_set_int (value, demux->program_number); break; case PROP_PAT_INFO: { if (demux->streams[0] != NULL) { g_value_take_boxed (value, mpegts_demux_build_pat_info (demux)); } break; } case PROP_PMT_INFO: { if (demux->current_PMT != 0 && demux->streams[demux->current_PMT] != NULL) { g_value_take_object (value, mpegts_demux_build_pmt_info (demux, demux->current_PMT)); } break; } default: G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec); break; } } gboolean gst_mpegts_demux_plugin_init (GstPlugin * plugin) { if (!gst_element_register (plugin, "mpegtsdemux", GST_RANK_PRIMARY, GST_TYPE_MPEGTS_DEMUX)) return FALSE; return TRUE; }