gst/gstnicesrc.c

   1 /*
   2  * This file is part of the Nice GLib ICE library.
   3  *
   4  * (C) 2006, 2007 Collabora Ltd.
   5  *  Contact: Dafydd Harries
   6  * (C) 2006, 2007 Nokia Corporation. All rights reserved.
   7  *  Contact: Kai Vehmanen
   8  *
   9  * The contents of this file are subject to the Mozilla Public License Version
  10  * 1.1 (the "License"); you may not use this file except in compliance with
  11  * the License. You may obtain a copy of the License at
  12  * http://www.mozilla.org/MPL/
  13  *
  14  * Software distributed under the License is distributed on an "AS IS" basis,
  15  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  16  * for the specific language governing rights and limitations under the
  17  * License.
  18  *
  19  * The Original Code is the Nice GLib ICE library.
  20  *
  21  * The Initial Developers of the Original Code are Collabora Ltd and Nokia
  22  * Corporation. All Rights Reserved.
  23  *
  24  * Contributors:
  25  *   Dafydd Harries, Collabora Ltd.
  26  *
  27  * Alternatively, the contents of this file may be used under the terms of the
  28  * the GNU Lesser General Public License Version 2.1 (the "LGPL"), in which
  29  * case the provisions of LGPL are applicable instead of those above. If you
  30  * wish to allow use of your version of this file only under the terms of the
  31  * LGPL and not to allow others to use your version of this file under the
  32  * MPL, indicate your decision by deleting the provisions above and replace
  33  * them with the notice and other provisions required by the LGPL. If you do
  34  * not delete the provisions above, a recipient may use your version of this
  35  * file under either the MPL or the LGPL.
  36  */
  37 #ifdef HAVE_CONFIG_H
  38 # include "config.h"
  39 #endif
  40
  41 #include <string.h>
  42
  43 #include "gstnicesrc.h"
  44
  45 GST_DEBUG_CATEGORY_STATIC (nicesrc_debug);
  46 #define GST_CAT_DEFAULT nicesrc_debug
  47
  48
  49 #define BUFFER_SIZE (65536)
  50
  51 static GstFlowReturn
  52 gst_nice_src_create (
  53   GstPushSrc *basesrc,
  54   GstBuffer **buffer);
  55
  56 static gboolean
  57 gst_nice_src_unlock (
  58     GstBaseSrc *basesrc);
  59
  60 static gboolean
  61 gst_nice_src_unlock_stop (
  62     GstBaseSrc *basesrc);
  63
  64 static void
  65 gst_nice_src_set_property (
  66   GObject *object,
  67   guint prop_id,
  68   const GValue *value,
  69   GParamSpec *pspec);
  70
  71 static void
  72 gst_nice_src_get_property (
  73   GObject *object,
  74   guint prop_id,
  75   GValue *value,
  76   GParamSpec *pspec);
  77
  78
  79 static void
  80 gst_nice_src_dispose (GObject *object);
  81
  82 static GstStateChangeReturn
  83 gst_nice_src_change_state (
  84     GstElement * element,
  85     GstStateChange transition);
  86
  87 static GstStaticPadTemplate gst_nice_src_src_template =
  88 GST_STATIC_PAD_TEMPLATE (
  89     "src",
  90     GST_PAD_SRC,
  91     GST_PAD_ALWAYS,
  92     GST_STATIC_CAPS_ANY);
  93
  94 G_DEFINE_TYPE (GstNiceSrc, gst_nice_src, GST_TYPE_PUSH_SRC);
  95
  96 enum
  97 {
  98   PROP_AGENT = 1,
  99   PROP_STREAM,
 100   PROP_COMPONENT
 101 };
 102
 103
 104 static void
 105 gst_nice_src_class_init (GstNiceSrcClass *klass)
 106 {
 107   GstPushSrcClass *gstpushsrc_class;
 108   GstBaseSrcClass *gstbasesrc_class;
 109   GstElementClass *gstelement_class;
 110   GObjectClass *gobject_class;
 111
 112   GST_DEBUG_CATEGORY_INIT (nicesrc_debug, "nicesrc",
 113       0, "libnice source");
 114
 115   gstpushsrc_class = (GstPushSrcClass *) klass;
 116   gstpushsrc_class->create = GST_DEBUG_FUNCPTR (gst_nice_src_create);
 117
 118   gstbasesrc_class = (GstBaseSrcClass *) klass;
 119   gstbasesrc_class->unlock = GST_DEBUG_FUNCPTR (gst_nice_src_unlock);
 120   gstbasesrc_class->unlock_stop = GST_DEBUG_FUNCPTR (gst_nice_src_unlock_stop);
 121
 122   gobject_class = (GObjectClass *) klass;
 123   gobject_class->set_property = gst_nice_src_set_property;
 124   gobject_class->get_property = gst_nice_src_get_property;
 125   gobject_class->dispose = gst_nice_src_dispose;
 126
 127   gstelement_class = (GstElementClass *) klass;
 128   gstelement_class->change_state = gst_nice_src_change_state;
 129
 130   gst_element_class_add_pad_template (gstelement_class,
 131       gst_static_pad_template_get (&gst_nice_src_src_template));
 132 #if GST_CHECK_VERSION (1,0,0)
 133   gst_element_class_set_metadata (gstelement_class,
 134 #else
 135   gst_element_class_set_details_simple (gstelement_class,
 136 #endif
 137       "ICE source",
 138       "Source",
 139       "Interactive UDP connectivity establishment",
 140       "Dafydd Harries <dafydd.harries@collabora.co.uk>");
 141
 142   g_object_class_install_property (gobject_class, PROP_AGENT,
 143       g_param_spec_object (
 144          "agent",
 145          "Agent",
 146          "The NiceAgent this source is bound to",
 147          NICE_TYPE_AGENT,
 148          G_PARAM_READWRITE));
 149
 150   g_object_class_install_property (gobject_class, PROP_STREAM,
 151       g_param_spec_uint (
 152          "stream",
 153          "Stream ID",
 154          "The ID of the stream to read from",
 155          0,
 156          G_MAXUINT,
 157          0,
 158          G_PARAM_READWRITE));
 159
 160   g_object_class_install_property (gobject_class, PROP_COMPONENT,
 161       g_param_spec_uint (
 162          "component",
 163          "Component ID",
 164          "The ID of the component to read from",
 165          0,
 166          G_MAXUINT,
 167          0,
 168          G_PARAM_READWRITE));
 169 }
 170
 171 static void
 172 gst_nice_src_init (GstNiceSrc *src)
 173 {
 174   gst_base_src_set_live (GST_BASE_SRC (src), TRUE);
 175   gst_base_src_set_format (GST_BASE_SRC (src), GST_FORMAT_TIME);
 176   gst_base_src_set_do_timestamp (GST_BASE_SRC (src), TRUE);
 177   src->agent = NULL;
 178   src->stream_id = 0;
 179   src->component_id = 0;
 180   src->mainctx = g_main_context_new ();
 181   src->mainloop = g_main_loop_new (src->mainctx, FALSE);
 182   src->unlocked = FALSE;
 183   src->idle_source = NULL;
 184   src->outbufs = g_queue_new ();
 185 }
 186
 187 static void
 188 gst_nice_src_read_callback (NiceAgent *agent,
 189     guint stream_id,
 190     guint component_id,
 191     guint len,
 192     gchar *buf,
 193     gpointer data)
 194 {
 195   GstBaseSrc *basesrc = GST_BASE_SRC (data);
 196   GstNiceSrc *nicesrc = GST_NICE_SRC (basesrc);
 197   GstBuffer *buffer = NULL;
 198
 199   GST_LOG_OBJECT (agent, "Got buffer, getting out of the main loop");
 200
 201 #if GST_CHECK_VERSION (1,0,0)
 202   buffer = gst_buffer_new_allocate (NULL, len, NULL);
 203   gst_buffer_fill (buffer, 0, buf, len);
 204 #else
 205   buffer = gst_buffer_new_and_alloc (len);
 206   memcpy (GST_BUFFER_DATA (buffer), buf, len);
 207 #endif
 208   g_queue_push_tail (nicesrc->outbufs, buffer);
 209
 210   g_main_loop_quit (nicesrc->mainloop);
 211 }
 212
 213 static gboolean
 214 gst_nice_src_unlock_idler (gpointer data)
 215 {
 216   GstNiceSrc *nicesrc = GST_NICE_SRC (data);
 217
 218   GST_OBJECT_LOCK (nicesrc);
 219   if (nicesrc->unlocked)
 220     g_main_loop_quit (nicesrc->mainloop);
 221
 222   if (nicesrc->idle_source) {
 223     g_source_destroy (nicesrc->idle_source);
 224     g_source_unref (nicesrc->idle_source);
 225     nicesrc->idle_source = NULL;
 226   }
 227   GST_OBJECT_UNLOCK (nicesrc);
 228
 229   return FALSE;
 230 }
 231
 232 static gboolean
 233 gst_nice_src_unlock (GstBaseSrc *src)
 234 {
 235   GstNiceSrc *nicesrc = GST_NICE_SRC (src);
 236
 237   GST_OBJECT_LOCK (src);
 238   nicesrc->unlocked = TRUE;
 239
 240   g_main_loop_quit (nicesrc->mainloop);
 241
 242   if (!nicesrc->idle_source) {
 243     nicesrc->idle_source = g_idle_source_new ();
 244     g_source_set_priority (nicesrc->idle_source, G_PRIORITY_HIGH);
 245     g_source_set_callback (nicesrc->idle_source, gst_nice_src_unlock_idler, src, NULL);
 246     g_source_attach (nicesrc->idle_source, g_main_loop_get_context (nicesrc->mainloop));
 247   }
 248   GST_OBJECT_UNLOCK (src);
 249
 250   return TRUE;
 251 }
 252
 253 static gboolean
 254 gst_nice_src_unlock_stop (GstBaseSrc *src)
 255 {
 256   GstNiceSrc *nicesrc = GST_NICE_SRC (src);
 257
 258   GST_OBJECT_LOCK (src);
 259   nicesrc->unlocked = FALSE;
 260   if (nicesrc->idle_source) {
 261     g_source_destroy (nicesrc->idle_source);
 262     g_source_unref(nicesrc->idle_source);
 263   }
 264   nicesrc->idle_source = NULL;
 265   GST_OBJECT_UNLOCK (src);
 266
 267   return TRUE;
 268 }
 269
 270 static GstFlowReturn
 271 gst_nice_src_create (
 272   GstPushSrc *basesrc,
 273   GstBuffer **buffer)
 274 {
 275   GstNiceSrc *nicesrc = GST_NICE_SRC (basesrc);
 276
 277   GST_LOG_OBJECT (nicesrc, "create called");
 278
 279   GST_OBJECT_LOCK (basesrc);
 280   if (nicesrc->unlocked) {
 281     GST_OBJECT_UNLOCK (basesrc);
 282 #if GST_CHECK_VERSION (1,0,0)
 283     return GST_FLOW_FLUSHING;
 284 #else
 285     return GST_FLOW_WRONG_STATE;
 286 #endif
 287   }
 288   GST_OBJECT_UNLOCK (basesrc);
 289
 290   if (g_queue_is_empty (nicesrc->outbufs))
 291     g_main_loop_run (nicesrc->mainloop);
 292
 293   *buffer = g_queue_pop_head (nicesrc->outbufs);
 294   if (*buffer != NULL) {
 295     GST_LOG_OBJECT (nicesrc, "Got buffer, pushing");
 296     return GST_FLOW_OK;
 297   } else {
 298     GST_LOG_OBJECT (nicesrc, "Got interrupting, returning wrong-state");
 299 #if GST_CHECK_VERSION (1,0,0)
 300     return GST_FLOW_FLUSHING;
 301 #else
 302     return GST_FLOW_WRONG_STATE;
 303 #endif
 304   }
 305
 306 }
 307
 308 static void
 309 gst_nice_src_dispose (GObject *object)
 310 {
 311   GstNiceSrc *src = GST_NICE_SRC (object);
 312
 313   if (src->agent)
 314     g_object_unref (src->agent);
 315   src->agent = NULL;
 316
 317   if (src->mainloop)
 318     g_main_loop_unref (src->mainloop);
 319   src->mainloop = NULL;
 320
 321   if (src->mainctx)
 322     g_main_context_unref (src->mainctx);
 323   src->mainctx = NULL;
 324
 325   if (src->outbufs)
 326     g_queue_free (src->outbufs);
 327   src->outbufs = NULL;
 328
 329   G_OBJECT_CLASS (gst_nice_src_parent_class)->dispose (object);
 330 }
 331
 332 static void
 333 gst_nice_src_set_property (
 334   GObject *object,
 335   guint prop_id,
 336   const GValue *value,
 337   GParamSpec *pspec)
 338 {
 339   GstNiceSrc *src = GST_NICE_SRC (object);
 340
 341   switch (prop_id)
 342     {
 343     case PROP_AGENT:
 344       if (src->agent)
 345         GST_ERROR_OBJECT (object,
 346             "Changing the agent on a nice src not allowed");
 347       else
 348         src->agent = g_value_dup_object (value);
 349       break;
 350
 351     case PROP_STREAM:
 352       src->stream_id = g_value_get_uint (value);
 353       break;
 354
 355     case PROP_COMPONENT:
 356       src->component_id = g_value_get_uint (value);
 357       break;
 358
 359     default:
 360       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 361       break;
 362     }
 363 }
 364
 365 static void
 366 gst_nice_src_get_property (
 367   GObject *object,
 368   guint prop_id,
 369   GValue *value,
 370   GParamSpec *pspec)
 371 {
 372   GstNiceSrc *src = GST_NICE_SRC (object);
 373
 374   switch (prop_id)
 375     {
 376     case PROP_AGENT:
 377       g_value_set_object (value, src->agent);
 378       break;
 379
 380     case PROP_STREAM:
 381       g_value_set_uint (value, src->stream_id);
 382       break;
 383
 384     case PROP_COMPONENT:
 385       g_value_set_uint (value, src->component_id);
 386       break;
 387
 388     default:
 389       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, prop_id, pspec);
 390       break;
 391     }
 392 }
 393
 394 static GstStateChangeReturn
 395 gst_nice_src_change_state (GstElement * element, GstStateChange transition)
 396 {
 397   GstNiceSrc *src;
 398   GstStateChangeReturn ret;
 399
 400   src = GST_NICE_SRC (element);
 401
 402   switch (transition) {
 403     case GST_STATE_CHANGE_NULL_TO_READY:
 404       if (src->agent == NULL || src->stream_id == 0 || src->component_id == 0)
 405         {
 406           GST_ERROR_OBJECT (element,
 407               "Trying to start Nice source without an agent set");
 408           return GST_STATE_CHANGE_FAILURE;
 409         }
 410       else
 411         {
 412           nice_agent_attach_recv (src->agent, src->stream_id, src->component_id,
 413               src->mainctx, gst_nice_src_read_callback, (gpointer) src);
 414         }
 415       break;
 416     case GST_STATE_CHANGE_READY_TO_NULL:
 417       nice_agent_attach_recv (src->agent, src->stream_id, src->component_id,
 418           src->mainctx, NULL, NULL);
 419       break;
 420     default:
 421       break;
 422   }
 423
 424   ret = GST_ELEMENT_CLASS (gst_nice_src_parent_class)->change_state (element,
 425       transition);
 426
 427   return ret;
 428 }
 429
 430