gst/sdp/gstsdpsrc.c - imx-gst-plugins-bad - Git at Google

 /*
  * Copyright (C) 2016 Sebastian Dröge <sebastian@centricular.com>
  *
  * 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., 51 Franklin St, Fifth Floor,
  * Boston, MA 02110-1301, USA.
  */

 #ifdef HAVE_CONFIG_H
 #include "config.h"
 #endif

 #include "gstsdpsrc.h"
 #include <gst/app/app.h>
 #include <string.h>

 GST_DEBUG_CATEGORY_STATIC (sdp_src_debug);
 #define GST_CAT_DEFAULT sdp_src_debug

 static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("stream_%u",
     GST_PAD_SRC,
     GST_PAD_SOMETIMES,
     GST_STATIC_CAPS ("application/x-rtp"));

 enum
 {
   PROP_0,
   PROP_LOCATION,
   PROP_SDP
 };

 static void gst_sdp_src_handler_init (gpointer g_iface, gpointer iface_data);

 #define gst_sdp_src_parent_class parent_class
 G_DEFINE_TYPE_WITH_CODE (GstSdpSrc, gst_sdp_src, GST_TYPE_BIN,
     G_IMPLEMENT_INTERFACE (GST_TYPE_URI_HANDLER, gst_sdp_src_handler_init));

 static void
 gst_sdp_src_finalize (GObject * object)
 {
   GstSdpSrc *self = GST_SDP_SRC_CAST (object);

   if (self->sdp_buffer)
     gst_buffer_unref (self->sdp_buffer);
   g_free (self->location);
   g_free (self->sdp);

   G_OBJECT_CLASS (parent_class)->finalize (object);
 }

 static void
 gst_sdp_src_get_property (GObject * object, guint prop_id,
     GValue * value, GParamSpec * pspec)
 {
   GstSdpSrc *self = GST_SDP_SRC_CAST (object);

   switch (prop_id) {
     case PROP_LOCATION:
       g_value_set_string (value, self->location);
       break;
     case PROP_SDP:
       g_value_set_string (value, self->sdp);
       break;
     default:
       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
       break;
   }
 }

 static void
 gst_sdp_src_set_property (GObject * object, guint prop_id,
     const GValue * value, GParamSpec * pspec)
 {
   GstSdpSrc *self = GST_SDP_SRC_CAST (object);

   switch (prop_id) {
     case PROP_LOCATION:
       g_free (self->location);
       self->location = g_value_dup_string (value);
       break;
     case PROP_SDP:
       g_free (self->sdp);
       self->sdp = g_value_dup_string (value);
       break;
     default:
       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
       break;
   }
 }

 static void
 pad_added_cb (GstElement * element, GstPad * pad, gpointer user_data)
 {
   GstSdpSrc *self = GST_SDP_SRC_CAST (user_data);
   GstPad *ghost;

   ghost =
       gst_ghost_pad_new_from_template (GST_PAD_NAME (pad), pad,
       gst_static_pad_template_get (&src_template));
   gst_pad_set_active (ghost, TRUE);
   gst_element_add_pad (GST_ELEMENT_CAST (self), ghost);
 }

 static void
 pad_removed_cb (GstElement * element, GstPad * pad, gpointer user_data)
 {
   GstSdpSrc *self = GST_SDP_SRC_CAST (user_data);
   GstPad *peer;

   peer = gst_pad_get_peer (pad);
   if (peer) {
     GstPad *ghost =
         GST_PAD_CAST (gst_proxy_pad_get_internal (GST_PROXY_PAD (peer)));

     if (ghost) {
       gst_ghost_pad_set_target (GST_GHOST_PAD_CAST (ghost), NULL);
       gst_element_remove_pad (GST_ELEMENT_CAST (self), ghost);
       gst_object_unref (ghost);
     }

     gst_object_unref (peer);
   }
 }

 static void
 no_more_pads_cb (GstElement * element, gpointer user_data)
 {
   gst_element_no_more_pads (GST_ELEMENT_CAST (user_data));
 }

 static void
 remove_pad (const GValue * item, gpointer user_data)
 {
   GstElement *self = user_data;
   GstPad *pad = g_value_get_object (item);

   gst_element_remove_pad (self, pad);
 }

 static GstStateChangeReturn
 gst_sdp_src_change_state (GstElement * element, GstStateChange transition)
 {
   GstSdpSrc *self = GST_SDP_SRC_CAST (element);
   GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;

   switch (transition) {
     case GST_STATE_CHANGE_NULL_TO_READY:
       GST_OBJECT_LOCK (self);
       if (self->sdp_buffer)
         gst_buffer_unref (self->sdp_buffer);
       self->sdp_buffer = NULL;

       if (self->location && strcmp (self->location, "sdp://") != 0) {
         /* Do nothing */
       } else if (self->sdp) {
         self->sdp_buffer =
             gst_buffer_new_wrapped (self->sdp, strlen (self->sdp) + 1);
       } else {
         ret = GST_STATE_CHANGE_FAILURE;
       }
       GST_OBJECT_UNLOCK (self);

       if (ret != GST_STATE_CHANGE_FAILURE) {
         if (self->sdp_buffer) {
           GstCaps *caps = gst_caps_new_empty_simple ("application/sdp");

           self->src = gst_element_factory_make ("appsrc", NULL);
           g_object_set (self->src, "caps", caps, "emit-signals", FALSE, NULL);
           gst_caps_unref (caps);
         } else {
           self->src = gst_element_factory_make ("filesrc", NULL);
           g_object_set (self->src, "location", self->location + 6, NULL);
         }

         self->demux = gst_element_factory_make ("sdpdemux", NULL);
         g_signal_connect (self->demux, "pad-added", G_CALLBACK (pad_added_cb),
             self);
         g_signal_connect (self->demux, "pad-removed",
             G_CALLBACK (pad_removed_cb), self);
         g_signal_connect (self->demux, "no-more-pads",
             G_CALLBACK (no_more_pads_cb), self);
         gst_bin_add_many (GST_BIN_CAST (self), self->src, self->demux, NULL);
         gst_element_link_pads (self->src, "src", self->demux, "sink");
       }
       break;
     default:
       break;
   }

   if (ret == GST_STATE_CHANGE_FAILURE)
     return ret;
   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
   if (ret == GST_STATE_CHANGE_FAILURE)
     return ret;

   switch (transition) {
     case GST_STATE_CHANGE_READY_TO_NULL:{
       GstIterator *it;

       it = gst_element_iterate_src_pads (GST_ELEMENT_CAST (self));
       while (gst_iterator_foreach (it, remove_pad, self) == GST_ITERATOR_RESYNC)
         gst_iterator_resync (it);
       gst_iterator_free (it);

       if (self->src) {
         gst_bin_remove (GST_BIN_CAST (self), self->src);
         self->src = NULL;
       }
       if (self->demux) {
         gst_bin_remove (GST_BIN_CAST (self), self->demux);
         self->demux = NULL;
       }
       break;
     }
     case GST_STATE_CHANGE_READY_TO_PAUSED:
       if (ret != GST_STATE_CHANGE_FAILURE)
         ret = GST_STATE_CHANGE_NO_PREROLL;
       if (self->sdp_buffer) {
         if (gst_app_src_push_buffer (GST_APP_SRC_CAST (self->src),
                 gst_buffer_ref (self->sdp_buffer)) != GST_FLOW_OK)
           ret = GST_STATE_CHANGE_FAILURE;
         else
           gst_app_src_end_of_stream (GST_APP_SRC_CAST (self->src));
       }
       break;
     default:
       break;
   }

   return ret;
 }

 static void
 gst_sdp_src_class_init (GstSdpSrcClass * klass)
 {
   GObjectClass *gobject_class = (GObjectClass *) klass;
   GstElementClass *element_class = (GstElementClass *) klass;

   GST_DEBUG_CATEGORY_INIT (sdp_src_debug, "sdpsrc", 0, "SDP Source");

   gobject_class->finalize = gst_sdp_src_finalize;
   gobject_class->set_property = gst_sdp_src_set_property;
   gobject_class->get_property = gst_sdp_src_get_property;

   g_object_class_install_property (gobject_class, PROP_LOCATION,
       g_param_spec_string ("location",
           "Location",
           "URI to SDP file (sdp:///path/to/file)", NULL,
           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));

   g_object_class_install_property (gobject_class, PROP_SDP,
       g_param_spec_string ("sdp",
           "SDP",
           "SDP description used instead of location", NULL,
           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));

   gst_element_class_add_pad_template (element_class,
       gst_static_pad_template_get (&src_template));

   gst_element_class_set_static_metadata (element_class, "SDP Source",
       "Source/Network/RTP",
       "Stream RTP based on an SDP",
       "Sebastian Dröge <sebastian@centricular.com>");

   element_class->change_state = GST_DEBUG_FUNCPTR (gst_sdp_src_change_state);
 }

 static void
 gst_sdp_src_init (GstSdpSrc * self)
 {
 }

 static GstURIType
 gst_sdp_src_get_uri_type (GType type)
 {
   return GST_URI_SRC;
 }

 static const gchar *const *
 gst_sdp_src_get_protocols (GType type)
 {
   static const gchar *protocols[] = { "sdp", 0 };

   return protocols;
 }

 static gchar *
 gst_sdp_src_get_uri (GstURIHandler * handler)
 {
   gchar *uri = NULL;

   g_object_get (handler, "location", &uri, NULL);

   return uri;
 }

 static gboolean
 gst_sdp_src_set_uri (GstURIHandler * handler, const gchar * uri,
     GError ** error)
 {
   if (uri && !g_str_has_prefix (uri, "sdp://")) {
     g_set_error (error, GST_URI_ERROR, GST_URI_ERROR_BAD_URI,
         "Invalid SDP URI");
     return FALSE;
   }

   g_object_set (handler, "location", uri, NULL);

   return TRUE;
 }

 static void
 gst_sdp_src_handler_init (gpointer g_iface, gpointer iface_data)
 {
   GstURIHandlerInterface *iface = (GstURIHandlerInterface *) g_iface;

   iface->get_type = gst_sdp_src_get_uri_type;
   iface->get_protocols = gst_sdp_src_get_protocols;
   iface->get_uri = gst_sdp_src_get_uri;
   iface->set_uri = gst_sdp_src_set_uri;
 }
	/*
	* Copyright (C) 2016 Sebastian Dröge <sebastian@centricular.com>
	*
	* 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., 51 Franklin St, Fifth Floor,
	* Boston, MA 02110-1301, USA.
	*/

	#ifdef HAVE_CONFIG_H
	#include "config.h"
	#endif

	#include "gstsdpsrc.h"
	#include <gst/app/app.h>
	#include <string.h>

	GST_DEBUG_CATEGORY_STATIC (sdp_src_debug);
	#define GST_CAT_DEFAULT sdp_src_debug

	static GstStaticPadTemplate src_template = GST_STATIC_PAD_TEMPLATE ("stream_%u",
	GST_PAD_SRC,
	GST_PAD_SOMETIMES,
	GST_STATIC_CAPS ("application/x-rtp"));

	enum
	{
	PROP_0,
	PROP_LOCATION,
	PROP_SDP
	};

	static void gst_sdp_src_handler_init (gpointer g_iface, gpointer iface_data);

	#define gst_sdp_src_parent_class parent_class
	G_DEFINE_TYPE_WITH_CODE (GstSdpSrc, gst_sdp_src, GST_TYPE_BIN,
	G_IMPLEMENT_INTERFACE (GST_TYPE_URI_HANDLER, gst_sdp_src_handler_init));

	static void
	gst_sdp_src_finalize (GObject * object)
	{
	GstSdpSrc *self = GST_SDP_SRC_CAST (object);

	if (self->sdp_buffer)
	gst_buffer_unref (self->sdp_buffer);
	g_free (self->location);
	g_free (self->sdp);

	G_OBJECT_CLASS (parent_class)->finalize (object);
	}

	static void
	gst_sdp_src_get_property (GObject * object, guint prop_id,
	GValue * value, GParamSpec * pspec)
	{
	GstSdpSrc *self = GST_SDP_SRC_CAST (object);

	switch (prop_id) {
	case PROP_LOCATION:
	g_value_set_string (value, self->location);
	break;
	case PROP_SDP:
	g_value_set_string (value, self->sdp);
	break;
	default:
	G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
	break;
	}
	}

	static void
	gst_sdp_src_set_property (GObject * object, guint prop_id,
	const GValue * value, GParamSpec * pspec)
	{
	GstSdpSrc *self = GST_SDP_SRC_CAST (object);

	switch (prop_id) {
	case PROP_LOCATION:
	g_free (self->location);
	self->location = g_value_dup_string (value);
	break;
	case PROP_SDP:
	g_free (self->sdp);
	self->sdp = g_value_dup_string (value);
	break;
	default:
	G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
	break;
	}
	}

	static void
	pad_added_cb (GstElement * element, GstPad * pad, gpointer user_data)
	{
	GstSdpSrc *self = GST_SDP_SRC_CAST (user_data);
	GstPad *ghost;

	ghost =
	gst_ghost_pad_new_from_template (GST_PAD_NAME (pad), pad,
	gst_static_pad_template_get (&src_template));
	gst_pad_set_active (ghost, TRUE);
	gst_element_add_pad (GST_ELEMENT_CAST (self), ghost);
	}

	static void
	pad_removed_cb (GstElement * element, GstPad * pad, gpointer user_data)
	{
	GstSdpSrc *self = GST_SDP_SRC_CAST (user_data);
	GstPad *peer;

	peer = gst_pad_get_peer (pad);
	if (peer) {
	GstPad *ghost =
	GST_PAD_CAST (gst_proxy_pad_get_internal (GST_PROXY_PAD (peer)));

	if (ghost) {
	gst_ghost_pad_set_target (GST_GHOST_PAD_CAST (ghost), NULL);
	gst_element_remove_pad (GST_ELEMENT_CAST (self), ghost);
	gst_object_unref (ghost);
	}

	gst_object_unref (peer);
	}
	}

	static void
	no_more_pads_cb (GstElement * element, gpointer user_data)
	{
	gst_element_no_more_pads (GST_ELEMENT_CAST (user_data));
	}

	static void
	remove_pad (const GValue * item, gpointer user_data)
	{
	GstElement *self = user_data;
	GstPad *pad = g_value_get_object (item);

	gst_element_remove_pad (self, pad);
	}

	static GstStateChangeReturn
	gst_sdp_src_change_state (GstElement * element, GstStateChange transition)
	{
	GstSdpSrc *self = GST_SDP_SRC_CAST (element);
	GstStateChangeReturn ret = GST_STATE_CHANGE_SUCCESS;

	switch (transition) {
	case GST_STATE_CHANGE_NULL_TO_READY:
	GST_OBJECT_LOCK (self);
	if (self->sdp_buffer)
	gst_buffer_unref (self->sdp_buffer);
	self->sdp_buffer = NULL;

	if (self->location && strcmp (self->location, "sdp://") != 0) {
	/* Do nothing */
	} else if (self->sdp) {
	self->sdp_buffer =
	gst_buffer_new_wrapped (self->sdp, strlen (self->sdp) + 1);
	} else {
	ret = GST_STATE_CHANGE_FAILURE;
	}
	GST_OBJECT_UNLOCK (self);

	if (ret != GST_STATE_CHANGE_FAILURE) {
	if (self->sdp_buffer) {
	GstCaps *caps = gst_caps_new_empty_simple ("application/sdp");

	self->src = gst_element_factory_make ("appsrc", NULL);
	g_object_set (self->src, "caps", caps, "emit-signals", FALSE, NULL);
	gst_caps_unref (caps);
	} else {
	self->src = gst_element_factory_make ("filesrc", NULL);
	g_object_set (self->src, "location", self->location + 6, NULL);
	}

	self->demux = gst_element_factory_make ("sdpdemux", NULL);
	g_signal_connect (self->demux, "pad-added", G_CALLBACK (pad_added_cb),
	self);
	g_signal_connect (self->demux, "pad-removed",
	G_CALLBACK (pad_removed_cb), self);
	g_signal_connect (self->demux, "no-more-pads",
	G_CALLBACK (no_more_pads_cb), self);
	gst_bin_add_many (GST_BIN_CAST (self), self->src, self->demux, NULL);
	gst_element_link_pads (self->src, "src", self->demux, "sink");
	}
	break;
	default:
	break;
	}

	if (ret == GST_STATE_CHANGE_FAILURE)
	return ret;
	ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
	if (ret == GST_STATE_CHANGE_FAILURE)
	return ret;

	switch (transition) {
	case GST_STATE_CHANGE_READY_TO_NULL:{
	GstIterator *it;

	it = gst_element_iterate_src_pads (GST_ELEMENT_CAST (self));
	while (gst_iterator_foreach (it, remove_pad, self) == GST_ITERATOR_RESYNC)
	gst_iterator_resync (it);
	gst_iterator_free (it);

	if (self->src) {
	gst_bin_remove (GST_BIN_CAST (self), self->src);
	self->src = NULL;
	}
	if (self->demux) {
	gst_bin_remove (GST_BIN_CAST (self), self->demux);
	self->demux = NULL;
	}
	break;
	}
	case GST_STATE_CHANGE_READY_TO_PAUSED:
	if (ret != GST_STATE_CHANGE_FAILURE)
	ret = GST_STATE_CHANGE_NO_PREROLL;
	if (self->sdp_buffer) {
	if (gst_app_src_push_buffer (GST_APP_SRC_CAST (self->src),
	gst_buffer_ref (self->sdp_buffer)) != GST_FLOW_OK)
	ret = GST_STATE_CHANGE_FAILURE;
	else
	gst_app_src_end_of_stream (GST_APP_SRC_CAST (self->src));
	}
	break;
	default:
	break;
	}

	return ret;
	}

	static void
	gst_sdp_src_class_init (GstSdpSrcClass * klass)
	{
	GObjectClass gobject_class = (GObjectClass ) klass;
	GstElementClass element_class = (GstElementClass ) klass;

	GST_DEBUG_CATEGORY_INIT (sdp_src_debug, "sdpsrc", 0, "SDP Source");

	gobject_class->finalize = gst_sdp_src_finalize;
	gobject_class->set_property = gst_sdp_src_set_property;
	gobject_class->get_property = gst_sdp_src_get_property;

	g_object_class_install_property (gobject_class, PROP_LOCATION,
	g_param_spec_string ("location",
	"Location",
	"URI to SDP file (sdp:///path/to/file)", NULL,
	G_PARAM_READWRITE \| G_PARAM_STATIC_STRINGS));

	g_object_class_install_property (gobject_class, PROP_SDP,
	g_param_spec_string ("sdp",
	"SDP",
	"SDP description used instead of location", NULL,
	G_PARAM_READWRITE \| G_PARAM_STATIC_STRINGS));

	gst_element_class_add_pad_template (element_class,
	gst_static_pad_template_get (&src_template));

	gst_element_class_set_static_metadata (element_class, "SDP Source",
	"Source/Network/RTP",
	"Stream RTP based on an SDP",
	"Sebastian Dröge <sebastian@centricular.com>");

	element_class->change_state = GST_DEBUG_FUNCPTR (gst_sdp_src_change_state);
	}

	static void
	gst_sdp_src_init (GstSdpSrc * self)
	{
	}

	static GstURIType
	gst_sdp_src_get_uri_type (GType type)
	{
	return GST_URI_SRC;
	}

	static const gchar const
	gst_sdp_src_get_protocols (GType type)
	{
	static const gchar *protocols[] = { "sdp", 0 };

	return protocols;
	}

	static gchar *
	gst_sdp_src_get_uri (GstURIHandler * handler)
	{
	gchar *uri = NULL;

	g_object_get (handler, "location", &uri, NULL);

	return uri;
	}

	static gboolean
	gst_sdp_src_set_uri (GstURIHandler * handler, const gchar * uri,
	GError ** error)
	{
	if (uri && !g_str_has_prefix (uri, "sdp://")) {
	g_set_error (error, GST_URI_ERROR, GST_URI_ERROR_BAD_URI,
	"Invalid SDP URI");
	return FALSE;
	}

	g_object_set (handler, "location", uri, NULL);

	return TRUE;
	}

	static void
	gst_sdp_src_handler_init (gpointer g_iface, gpointer iface_data)
	{
	GstURIHandlerInterface iface = (GstURIHandlerInterface ) g_iface;

	iface->get_type = gst_sdp_src_get_uri_type;
	iface->get_protocols = gst_sdp_src_get_protocols;
	iface->get_uri = gst_sdp_src_get_uri;
	iface->set_uri = gst_sdp_src_set_uri;
	}