gst/rtpmanager/gstrtpssrcdemux.c

   1 /* GStreamer
   2  * Copyright (C) <2007> Wim Taymans <wim.taymans@gmail.com>
   3  *
   4  * RTP SSRC demuxer
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Library General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Library General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Library General Public
  17  * License along with this library; if not, write to the
  18  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  19  * Boston, MA 02111-1307, USA.
  20  */
  21
  22 /**
  23  * SECTION:element-gstrtpssrcdemux
  24  *
  25  * gstrtpssrcdemux acts as a demuxer for RTP packets based on the SSRC of the
  26  * packets. Its main purpose is to allow an application to easily receive and
  27  * decode an RTP stream with multiple SSRCs.
  28  *
  29  * For each SSRC that is detected, a new pad will be created and the
  30  * #GstRtpSsrcDemux::new-ssrc-pad signal will be emitted.
  31  *
  32  * <refsect2>
  33  * <title>Example pipelines</title>
  34  * |[
  35  * gst-launch udpsrc caps="application/x-rtp" ! gstrtpssrcdemux ! fakesink
  36  * ]| Takes an RTP stream and send the RTP packets with the first detected SSRC
  37  * to fakesink, discarding the other SSRCs.
  38  * </refsect2>
  39  *
  40  * Last reviewed on 2007-05-28 (0.10.5)
  41  */
  42
  43 #ifdef HAVE_CONFIG_H
  44 #include "config.h"
  45 #endif
  46
  47 #include <string.h>
  48 #include <gst/rtp/gstrtpbuffer.h>
  49 #include <gst/rtp/gstrtcpbuffer.h>
  50
  51 #include "gstrtpbin-marshal.h"
  52 #include "gstrtpssrcdemux.h"
  53
  54 GST_DEBUG_CATEGORY_STATIC (gst_rtp_ssrc_demux_debug);
  55 #define GST_CAT_DEFAULT gst_rtp_ssrc_demux_debug
  56
  57 /* generic templates */
  58 static GstStaticPadTemplate rtp_ssrc_demux_sink_template =
  59 GST_STATIC_PAD_TEMPLATE ("sink",
  60     GST_PAD_SINK,
  61     GST_PAD_ALWAYS,
  62     GST_STATIC_CAPS ("application/x-rtp")
  63     );
  64
  65 static GstStaticPadTemplate rtp_ssrc_demux_rtcp_sink_template =
  66 GST_STATIC_PAD_TEMPLATE ("rtcp_sink",
  67     GST_PAD_SINK,
  68     GST_PAD_ALWAYS,
  69     GST_STATIC_CAPS ("application/x-rtcp")
  70     );
  71
  72 static GstStaticPadTemplate rtp_ssrc_demux_src_template =
  73 GST_STATIC_PAD_TEMPLATE ("src_%u",
  74     GST_PAD_SRC,
  75     GST_PAD_SOMETIMES,
  76     GST_STATIC_CAPS ("application/x-rtp")
  77     );
  78
  79 static GstStaticPadTemplate rtp_ssrc_demux_rtcp_src_template =
  80 GST_STATIC_PAD_TEMPLATE ("rtcp_src_%u",
  81     GST_PAD_SRC,
  82     GST_PAD_SOMETIMES,
  83     GST_STATIC_CAPS ("application/x-rtcp")
  84     );
  85
  86 #define GST_PAD_LOCK(obj)   (g_rec_mutex_lock (&(obj)->padlock))
  87 #define GST_PAD_UNLOCK(obj) (g_rec_mutex_unlock (&(obj)->padlock))
  88
  89 /* signals */
  90 enum
  91 {
  92   SIGNAL_NEW_SSRC_PAD,
  93   SIGNAL_REMOVED_SSRC_PAD,
  94   SIGNAL_CLEAR_SSRC,
  95   LAST_SIGNAL
  96 };
  97
  98 #define gst_rtp_ssrc_demux_parent_class parent_class
  99 G_DEFINE_TYPE (GstRtpSsrcDemux, gst_rtp_ssrc_demux, GST_TYPE_ELEMENT);
 100
 101 /* GObject vmethods */
 102 static void gst_rtp_ssrc_demux_dispose (GObject * object);
 103 static void gst_rtp_ssrc_demux_finalize (GObject * object);
 104
 105 /* GstElement vmethods */
 106 static GstStateChangeReturn gst_rtp_ssrc_demux_change_state (GstElement *
 107     element, GstStateChange transition);
 108
 109 static void gst_rtp_ssrc_demux_clear_ssrc (GstRtpSsrcDemux * demux,
 110     guint32 ssrc);
 111
 112 /* sinkpad stuff */
 113 static GstFlowReturn gst_rtp_ssrc_demux_chain (GstPad * pad, GstObject * parent,
 114     GstBuffer * buf);
 115 static gboolean gst_rtp_ssrc_demux_sink_event (GstPad * pad, GstObject * parent,
 116     GstEvent * event);
 117
 118 static GstFlowReturn gst_rtp_ssrc_demux_rtcp_chain (GstPad * pad,
 119     GstObject * parent, GstBuffer * buf);
 120 static gboolean gst_rtp_ssrc_demux_rtcp_sink_event (GstPad * pad,
 121     GstObject * parent, GstEvent * event);
 122 static GstIterator *gst_rtp_ssrc_demux_iterate_internal_links_sink (GstPad *
 123     pad, GstObject * parent);
 124
 125 /* srcpad stuff */
 126 static gboolean gst_rtp_ssrc_demux_src_event (GstPad * pad, GstObject * parent,
 127     GstEvent * event);
 128 static GstIterator *gst_rtp_ssrc_demux_iterate_internal_links_src (GstPad * pad,
 129     GstObject * parent);
 130 static gboolean gst_rtp_ssrc_demux_src_query (GstPad * pad, GstObject * parent,
 131     GstQuery * query);
 132
 133 static guint gst_rtp_ssrc_demux_signals[LAST_SIGNAL] = { 0 };
 134
 135 /*
 136  * Item for storing GstPad <-> SSRC pairs.
 137  */
 138 struct _GstRtpSsrcDemuxPad
 139 {
 140   guint32 ssrc;
 141   GstPad *rtp_pad;
 142   GstCaps *caps;
 143   GstPad *rtcp_pad;
 144 };
 145
 146 /* find a src pad for a given SSRC, returns NULL if the SSRC was not found
 147  */
 148 static GstRtpSsrcDemuxPad *
 149 find_demux_pad_for_ssrc (GstRtpSsrcDemux * demux, guint32 ssrc)
 150 {
 151   GSList *walk;
 152
 153   for (walk = demux->srcpads; walk; walk = g_slist_next (walk)) {
 154     GstRtpSsrcDemuxPad *pad = (GstRtpSsrcDemuxPad *) walk->data;
 155
 156     if (pad->ssrc == ssrc)
 157       return pad;
 158   }
 159   return NULL;
 160 }
 161
 162 static GstEvent *
 163 add_ssrc_and_ref (GstEvent * event, guint32 ssrc)
 164 {
 165   /* Set the ssrc on the output caps */
 166   switch (GST_EVENT_TYPE (event)) {
 167     case GST_EVENT_CAPS:
 168     {
 169       GstCaps *caps;
 170       GstCaps *newcaps;
 171       GstStructure *s;
 172
 173       gst_event_parse_caps (event, &caps);
 174       newcaps = gst_caps_copy (caps);
 175
 176       s = gst_caps_get_structure (newcaps, 0);
 177       gst_structure_set (s, "ssrc", G_TYPE_UINT, ssrc, NULL);
 178       event = gst_event_new_caps (newcaps);
 179       gst_caps_unref (newcaps);
 180       break;
 181     }
 182     default:
 183       gst_event_ref (event);
 184       break;
 185   }
 186
 187   return event;
 188 }
 189
 190 struct ForwardEventData
 191 {
 192   GstPad *pad;
 193   guint32 ssrc;
 194 };
 195
 196 static gboolean
 197 forward_sticky_events (GstPad * pad, GstEvent ** event, gpointer user_data)
 198 {
 199   struct ForwardEventData *data = user_data;
 200   GstEvent *newevent;
 201
 202   newevent = add_ssrc_and_ref (*event, data->ssrc);
 203
 204   gst_pad_push_event (data->pad, newevent);
 205
 206   return TRUE;
 207 }
 208
 209
 210 /* with PAD_LOCK */
 211 static GstRtpSsrcDemuxPad *
 212 find_or_create_demux_pad_for_ssrc (GstRtpSsrcDemux * demux, guint32 ssrc)
 213 {
 214   GstPad *rtp_pad, *rtcp_pad;
 215   GstElementClass *klass;
 216   GstPadTemplate *templ;
 217   gchar *padname;
 218   GstRtpSsrcDemuxPad *demuxpad;
 219   GstCaps *caps;
 220   struct ForwardEventData fdata;
 221
 222   GST_DEBUG_OBJECT (demux, "creating pad for SSRC %08x", ssrc);
 223
 224   demuxpad = find_demux_pad_for_ssrc (demux, ssrc);
 225   if (demuxpad != NULL) {
 226     return demuxpad;
 227   }
 228
 229   klass = GST_ELEMENT_GET_CLASS (demux);
 230   templ = gst_element_class_get_pad_template (klass, "src_%u");
 231   padname = g_strdup_printf ("src_%u", ssrc);
 232   rtp_pad = gst_pad_new_from_template (templ, padname);
 233   g_free (padname);
 234
 235   templ = gst_element_class_get_pad_template (klass, "rtcp_src_%u");
 236   padname = g_strdup_printf ("rtcp_src_%u", ssrc);
 237   rtcp_pad = gst_pad_new_from_template (templ, padname);
 238   g_free (padname);
 239
 240   /* wrap in structure and add to list */
 241   demuxpad = g_new0 (GstRtpSsrcDemuxPad, 1);
 242   demuxpad->ssrc = ssrc;
 243   demuxpad->rtp_pad = rtp_pad;
 244   demuxpad->rtcp_pad = rtcp_pad;
 245
 246   fdata.ssrc = ssrc;
 247
 248   gst_pad_set_element_private (rtp_pad, demuxpad);
 249   gst_pad_set_element_private (rtcp_pad, demuxpad);
 250
 251   demux->srcpads = g_slist_prepend (demux->srcpads, demuxpad);
 252
 253   gst_pad_set_query_function (rtp_pad, gst_rtp_ssrc_demux_src_query);
 254   gst_pad_set_iterate_internal_links_function (rtp_pad,
 255       gst_rtp_ssrc_demux_iterate_internal_links_src);
 256   gst_pad_set_event_function (rtp_pad, gst_rtp_ssrc_demux_src_event);
 257   gst_pad_use_fixed_caps (rtp_pad);
 258   gst_pad_set_active (rtp_pad, TRUE);
 259   fdata.pad = rtp_pad;
 260   gst_pad_sticky_events_foreach (demux->rtp_sink, forward_sticky_events,
 261       &fdata);
 262
 263   gst_pad_set_event_function (rtcp_pad, gst_rtp_ssrc_demux_src_event);
 264   gst_pad_set_iterate_internal_links_function (rtcp_pad,
 265       gst_rtp_ssrc_demux_iterate_internal_links_src);
 266   gst_pad_use_fixed_caps (rtcp_pad);
 267   gst_pad_set_active (rtcp_pad, TRUE);
 268   fdata.pad = rtcp_pad;
 269   gst_pad_sticky_events_foreach (demux->rtcp_sink, forward_sticky_events,
 270       &fdata);
 271
 272   /* copy caps from input */
 273   if ((caps = gst_pad_get_current_caps (demux->rtp_sink))) {
 274     gst_pad_set_caps (rtp_pad, caps);
 275     gst_caps_unref (caps);
 276   }
 277   if ((caps = gst_pad_get_current_caps (demux->rtcp_sink))) {
 278     gst_pad_set_caps (rtcp_pad, caps);
 279     gst_caps_unref (caps);
 280   }
 281   gst_element_add_pad (GST_ELEMENT_CAST (demux), rtp_pad);
 282   gst_element_add_pad (GST_ELEMENT_CAST (demux), rtcp_pad);
 283
 284   g_signal_emit (G_OBJECT (demux),
 285       gst_rtp_ssrc_demux_signals[SIGNAL_NEW_SSRC_PAD], 0, ssrc, rtp_pad);
 286
 287   return demuxpad;
 288 }
 289
 290 static void
 291 gst_rtp_ssrc_demux_class_init (GstRtpSsrcDemuxClass * klass)
 292 {
 293   GObjectClass *gobject_klass;
 294   GstElementClass *gstelement_klass;
 295   GstRtpSsrcDemuxClass *gstrtpssrcdemux_klass;
 296
 297   gobject_klass = (GObjectClass *) klass;
 298   gstelement_klass = (GstElementClass *) klass;
 299   gstrtpssrcdemux_klass = (GstRtpSsrcDemuxClass *) klass;
 300
 301   gobject_klass->dispose = gst_rtp_ssrc_demux_dispose;
 302   gobject_klass->finalize = gst_rtp_ssrc_demux_finalize;
 303
 304   /**
 305    * GstRtpSsrcDemux::new-ssrc-pad:
 306    * @demux: the object which received the signal
 307    * @ssrc: the SSRC of the pad
 308    * @pad: the new pad.
 309    *
 310    * Emited when a new SSRC pad has been created.
 311    */
 312   gst_rtp_ssrc_demux_signals[SIGNAL_NEW_SSRC_PAD] =
 313       g_signal_new ("new-ssrc-pad",
 314       G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
 315       G_STRUCT_OFFSET (GstRtpSsrcDemuxClass, new_ssrc_pad),
 316       NULL, NULL, gst_rtp_bin_marshal_VOID__UINT_OBJECT,
 317       G_TYPE_NONE, 2, G_TYPE_UINT, GST_TYPE_PAD);
 318
 319   /**
 320    * GstRtpSsrcDemux::removed-ssrc-pad:
 321    * @demux: the object which received the signal
 322    * @ssrc: the SSRC of the pad
 323    * @pad: the removed pad.
 324    *
 325    * Emited when a SSRC pad has been removed.
 326    */
 327   gst_rtp_ssrc_demux_signals[SIGNAL_REMOVED_SSRC_PAD] =
 328       g_signal_new ("removed-ssrc-pad",
 329       G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
 330       G_STRUCT_OFFSET (GstRtpSsrcDemuxClass, removed_ssrc_pad),
 331       NULL, NULL, gst_rtp_bin_marshal_VOID__UINT_OBJECT,
 332       G_TYPE_NONE, 2, G_TYPE_UINT, GST_TYPE_PAD);
 333
 334   /**
 335    * GstRtpSsrcDemux::clear-ssrc:
 336    * @demux: the object which received the signal
 337    * @ssrc: the SSRC of the pad
 338    *
 339    * Action signal to remove the pad for SSRC.
 340    */
 341   gst_rtp_ssrc_demux_signals[SIGNAL_CLEAR_SSRC] =
 342       g_signal_new ("clear-ssrc",
 343       G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
 344       G_STRUCT_OFFSET (GstRtpSsrcDemuxClass, clear_ssrc),
 345       NULL, NULL, gst_rtp_bin_marshal_VOID__UINT, G_TYPE_NONE, 1, G_TYPE_UINT);
 346
 347   gstelement_klass->change_state =
 348       GST_DEBUG_FUNCPTR (gst_rtp_ssrc_demux_change_state);
 349   gstrtpssrcdemux_klass->clear_ssrc =
 350       GST_DEBUG_FUNCPTR (gst_rtp_ssrc_demux_clear_ssrc);
 351
 352   gst_element_class_add_pad_template (gstelement_klass,
 353       gst_static_pad_template_get (&rtp_ssrc_demux_sink_template));
 354   gst_element_class_add_pad_template (gstelement_klass,
 355       gst_static_pad_template_get (&rtp_ssrc_demux_rtcp_sink_template));
 356   gst_element_class_add_pad_template (gstelement_klass,
 357       gst_static_pad_template_get (&rtp_ssrc_demux_src_template));
 358   gst_element_class_add_pad_template (gstelement_klass,
 359       gst_static_pad_template_get (&rtp_ssrc_demux_rtcp_src_template));
 360
 361   gst_element_class_set_details_simple (gstelement_klass, "RTP SSRC Demux",
 362       "Demux/Network/RTP",
 363       "Splits RTP streams based on the SSRC",
 364       "Wim Taymans <wim.taymans@gmail.com>");
 365
 366   GST_DEBUG_CATEGORY_INIT (gst_rtp_ssrc_demux_debug,
 367       "rtpssrcdemux", 0, "RTP SSRC demuxer");
 368 }
 369
 370 static void
 371 gst_rtp_ssrc_demux_init (GstRtpSsrcDemux * demux)
 372 {
 373   GstElementClass *klass = GST_ELEMENT_GET_CLASS (demux);
 374
 375   demux->rtp_sink =
 376       gst_pad_new_from_template (gst_element_class_get_pad_template (klass,
 377           "sink"), "sink");
 378   gst_pad_set_chain_function (demux->rtp_sink, gst_rtp_ssrc_demux_chain);
 379   gst_pad_set_event_function (demux->rtp_sink, gst_rtp_ssrc_demux_sink_event);
 380   gst_pad_set_iterate_internal_links_function (demux->rtp_sink,
 381       gst_rtp_ssrc_demux_iterate_internal_links_sink);
 382   gst_element_add_pad (GST_ELEMENT_CAST (demux), demux->rtp_sink);
 383
 384   demux->rtcp_sink =
 385       gst_pad_new_from_template (gst_element_class_get_pad_template (klass,
 386           "rtcp_sink"), "rtcp_sink");
 387   gst_pad_set_chain_function (demux->rtcp_sink, gst_rtp_ssrc_demux_rtcp_chain);
 388   gst_pad_set_event_function (demux->rtcp_sink,
 389       gst_rtp_ssrc_demux_rtcp_sink_event);
 390   gst_pad_set_iterate_internal_links_function (demux->rtcp_sink,
 391       gst_rtp_ssrc_demux_iterate_internal_links_sink);
 392   gst_element_add_pad (GST_ELEMENT_CAST (demux), demux->rtcp_sink);
 393
 394   g_rec_mutex_init (&demux->padlock);
 395
 396   gst_segment_init (&demux->segment, GST_FORMAT_UNDEFINED);
 397 }
 398
 399 static void
 400 gst_rtp_ssrc_demux_reset (GstRtpSsrcDemux * demux)
 401 {
 402   GSList *walk;
 403
 404   for (walk = demux->srcpads; walk; walk = g_slist_next (walk)) {
 405     GstRtpSsrcDemuxPad *dpad = (GstRtpSsrcDemuxPad *) walk->data;
 406
 407     gst_pad_set_active (dpad->rtp_pad, FALSE);
 408     gst_pad_set_active (dpad->rtcp_pad, FALSE);
 409
 410     gst_element_remove_pad (GST_ELEMENT_CAST (demux), dpad->rtp_pad);
 411     gst_element_remove_pad (GST_ELEMENT_CAST (demux), dpad->rtcp_pad);
 412     g_free (dpad);
 413   }
 414   g_slist_free (demux->srcpads);
 415   demux->srcpads = NULL;
 416 }
 417
 418 static void
 419 gst_rtp_ssrc_demux_dispose (GObject * object)
 420 {
 421   GstRtpSsrcDemux *demux;
 422
 423   demux = GST_RTP_SSRC_DEMUX (object);
 424
 425   gst_rtp_ssrc_demux_reset (demux);
 426
 427   G_OBJECT_CLASS (parent_class)->dispose (object);
 428 }
 429
 430 static void
 431 gst_rtp_ssrc_demux_finalize (GObject * object)
 432 {
 433   GstRtpSsrcDemux *demux;
 434
 435   demux = GST_RTP_SSRC_DEMUX (object);
 436   g_rec_mutex_clear (&demux->padlock);
 437
 438   G_OBJECT_CLASS (parent_class)->finalize (object);
 439 }
 440
 441 static void
 442 gst_rtp_ssrc_demux_clear_ssrc (GstRtpSsrcDemux * demux, guint32 ssrc)
 443 {
 444   GstRtpSsrcDemuxPad *dpad;
 445
 446   GST_PAD_LOCK (demux);
 447   dpad = find_demux_pad_for_ssrc (demux, ssrc);
 448   if (dpad == NULL) {
 449     GST_PAD_UNLOCK (demux);
 450     goto unknown_pad;
 451   }
 452
 453   GST_DEBUG_OBJECT (demux, "clearing pad for SSRC %08x", ssrc);
 454
 455   demux->srcpads = g_slist_remove (demux->srcpads, dpad);
 456   GST_PAD_UNLOCK (demux);
 457
 458   gst_pad_set_active (dpad->rtp_pad, FALSE);
 459   gst_pad_set_active (dpad->rtcp_pad, FALSE);
 460
 461   g_signal_emit (G_OBJECT (demux),
 462       gst_rtp_ssrc_demux_signals[SIGNAL_REMOVED_SSRC_PAD], 0, ssrc,
 463       dpad->rtp_pad);
 464
 465   gst_element_remove_pad (GST_ELEMENT_CAST (demux), dpad->rtp_pad);
 466   gst_element_remove_pad (GST_ELEMENT_CAST (demux), dpad->rtcp_pad);
 467
 468   g_free (dpad);
 469
 470   return;
 471
 472   /* ERRORS */
 473 unknown_pad:
 474   {
 475     GST_WARNING_OBJECT (demux, "unknown SSRC %08x", ssrc);
 476     return;
 477   }
 478 }
 479
 480 static gboolean
 481 gst_rtp_ssrc_demux_sink_event (GstPad * pad, GstObject * parent,
 482     GstEvent * event)
 483 {
 484   GstRtpSsrcDemux *demux;
 485   gboolean res = FALSE;
 486
 487   demux = GST_RTP_SSRC_DEMUX (parent);
 488
 489   switch (GST_EVENT_TYPE (event)) {
 490     case GST_EVENT_FLUSH_STOP:
 491       gst_segment_init (&demux->segment, GST_FORMAT_UNDEFINED);
 492       /* fallthrough */
 493     default:
 494     {
 495       GSList *walk;
 496       GSList *pads = NULL;
 497
 498       res = TRUE;
 499       /* need local snapshot of pads;
 500        * should not push downstream while holding lock as that might deadlock
 501        * with stuff traveling upstream tyring to get this lock while holding
 502        * other (stream)lock */
 503       GST_PAD_LOCK (demux);
 504       for (walk = demux->srcpads; walk; walk = g_slist_next (walk)) {
 505         GstRtpSsrcDemuxPad *pad = (GstRtpSsrcDemuxPad *) walk->data;
 506
 507         pad = g_slice_dup (GstRtpSsrcDemuxPad, pad);
 508         gst_object_ref (pad->rtp_pad);
 509
 510         pads = g_slist_prepend (pads, pad);
 511       }
 512       GST_PAD_UNLOCK (demux);
 513
 514       for (walk = pads; walk; walk = g_slist_next (walk)) {
 515         GstRtpSsrcDemuxPad *dpad = walk->data;
 516         GstEvent *newevent;
 517
 518         newevent = add_ssrc_and_ref (event, dpad->ssrc);
 519
 520         res &= gst_pad_push_event (dpad->rtp_pad, newevent);
 521         gst_object_unref (dpad->rtp_pad);
 522         g_slice_free (GstRtpSsrcDemuxPad, dpad);
 523       }
 524       g_slist_free (pads);
 525       gst_event_unref (event);
 526       break;
 527     }
 528   }
 529
 530   return res;
 531 }
 532
 533 static gboolean
 534 gst_rtp_ssrc_demux_rtcp_sink_event (GstPad * pad, GstObject * parent,
 535     GstEvent * event)
 536 {
 537   GstRtpSsrcDemux *demux;
 538   gboolean res = TRUE;
 539   GSList *walk;
 540   GSList *pads = NULL;
 541
 542   demux = GST_RTP_SSRC_DEMUX (parent);
 543
 544   GST_PAD_LOCK (demux);
 545   for (walk = demux->srcpads; walk; walk = g_slist_next (walk)) {
 546     GstRtpSsrcDemuxPad *pad = (GstRtpSsrcDemuxPad *) walk->data;
 547
 548     pad = g_slice_dup (GstRtpSsrcDemuxPad, pad);
 549     gst_object_ref (pad->rtcp_pad);
 550
 551     pads = g_slist_prepend (pads, pad);
 552   }
 553   GST_PAD_UNLOCK (demux);
 554
 555   for (walk = pads; walk; walk = g_slist_next (walk)) {
 556     GstRtpSsrcDemuxPad *dpad = walk->data;
 557     GstEvent *newevent;
 558
 559     newevent = add_ssrc_and_ref (event, dpad->ssrc);
 560
 561     res &= gst_pad_push_event (dpad->rtcp_pad, newevent);
 562     gst_object_unref (dpad->rtcp_pad);
 563     g_slice_free (GstRtpSsrcDemuxPad, dpad);
 564   }
 565   g_slist_free (pads);
 566   gst_event_unref (event);
 567
 568   return res;
 569 }
 570
 571 static GstFlowReturn
 572 gst_rtp_ssrc_demux_chain (GstPad * pad, GstObject * parent, GstBuffer * buf)
 573 {
 574   GstFlowReturn ret;
 575   GstRtpSsrcDemux *demux;
 576   guint32 ssrc;
 577   GstRtpSsrcDemuxPad *dpad;
 578   GstRTPBuffer rtp = { NULL };
 579   GstPad *srcpad;
 580
 581   demux = GST_RTP_SSRC_DEMUX (parent);
 582
 583   if (!gst_rtp_buffer_validate (buf))
 584     goto invalid_payload;
 585
 586   gst_rtp_buffer_map (buf, GST_MAP_READ, &rtp);
 587   ssrc = gst_rtp_buffer_get_ssrc (&rtp);
 588   gst_rtp_buffer_unmap (&rtp);
 589
 590   GST_DEBUG_OBJECT (demux, "received buffer of SSRC %08x", ssrc);
 591
 592   GST_PAD_LOCK (demux);
 593   dpad = find_or_create_demux_pad_for_ssrc (demux, ssrc);
 594   if (dpad == NULL) {
 595     GST_PAD_UNLOCK (demux);
 596     goto create_failed;
 597   }
 598   srcpad = gst_object_ref (dpad->rtp_pad);
 599   GST_PAD_UNLOCK (demux);
 600
 601   /* push to srcpad */
 602   ret = gst_pad_push (srcpad, buf);
 603
 604   gst_object_unref (srcpad);
 605
 606   return ret;
 607
 608   /* ERRORS */
 609 invalid_payload:
 610   {
 611     /* this is fatal and should be filtered earlier */
 612     GST_ELEMENT_ERROR (demux, STREAM, DECODE, (NULL),
 613         ("Dropping invalid RTP payload"));
 614     gst_buffer_unref (buf);
 615     return GST_FLOW_ERROR;
 616   }
 617 create_failed:
 618   {
 619     GST_ELEMENT_ERROR (demux, STREAM, DECODE, (NULL),
 620         ("Could not create new pad"));
 621     gst_buffer_unref (buf);
 622     return GST_FLOW_ERROR;
 623   }
 624 }
 625
 626 static GstFlowReturn
 627 gst_rtp_ssrc_demux_rtcp_chain (GstPad * pad, GstObject * parent,
 628     GstBuffer * buf)
 629 {
 630   GstFlowReturn ret;
 631   GstRtpSsrcDemux *demux;
 632   guint32 ssrc;
 633   GstRtpSsrcDemuxPad *dpad;
 634   GstRTCPPacket packet;
 635   GstRTCPBuffer rtcp = { NULL, };
 636   GstPad *srcpad;
 637
 638   demux = GST_RTP_SSRC_DEMUX (parent);
 639
 640   if (!gst_rtcp_buffer_validate (buf))
 641     goto invalid_rtcp;
 642
 643   gst_rtcp_buffer_map (buf, GST_MAP_READ, &rtcp);
 644   if (!gst_rtcp_buffer_get_first_packet (&rtcp, &packet)) {
 645     gst_rtcp_buffer_unmap (&rtcp);
 646     goto invalid_rtcp;
 647   }
 648
 649   /* first packet must be SR or RR or else the validate would have failed */
 650   switch (gst_rtcp_packet_get_type (&packet)) {
 651     case GST_RTCP_TYPE_SR:
 652       /* get the ssrc so that we can route it to the right source pad */
 653       gst_rtcp_packet_sr_get_sender_info (&packet, &ssrc, NULL, NULL, NULL,
 654           NULL);
 655       break;
 656     default:
 657       goto unexpected_rtcp;
 658   }
 659   gst_rtcp_buffer_unmap (&rtcp);
 660
 661   GST_DEBUG_OBJECT (demux, "received RTCP of SSRC %08x", ssrc);
 662
 663   GST_PAD_LOCK (demux);
 664   dpad = find_or_create_demux_pad_for_ssrc (demux, ssrc);
 665   if (dpad == NULL) {
 666     GST_PAD_UNLOCK (demux);
 667     goto create_failed;
 668   }
 669   srcpad = gst_object_ref (dpad->rtcp_pad);
 670   GST_PAD_UNLOCK (demux);
 671
 672   /* push to srcpad */
 673   ret = gst_pad_push (srcpad, buf);
 674
 675   gst_object_unref (srcpad);
 676
 677   return ret;
 678
 679   /* ERRORS */
 680 invalid_rtcp:
 681   {
 682     /* this is fatal and should be filtered earlier */
 683     GST_ELEMENT_ERROR (demux, STREAM, DECODE, (NULL),
 684         ("Dropping invalid RTCP packet"));
 685     gst_buffer_unref (buf);
 686     return GST_FLOW_ERROR;
 687   }
 688 unexpected_rtcp:
 689   {
 690     GST_DEBUG_OBJECT (demux, "dropping unexpected RTCP packet");
 691     gst_buffer_unref (buf);
 692     return GST_FLOW_OK;
 693   }
 694 create_failed:
 695   {
 696     GST_ELEMENT_ERROR (demux, STREAM, DECODE, (NULL),
 697         ("Could not create new pad"));
 698     gst_buffer_unref (buf);
 699     return GST_FLOW_ERROR;
 700   }
 701 }
 702
 703 static GstRtpSsrcDemuxPad *
 704 find_demux_pad_for_pad (GstRtpSsrcDemux * demux, GstPad * pad)
 705 {
 706   GSList *walk;
 707
 708   for (walk = demux->srcpads; walk; walk = g_slist_next (walk)) {
 709     GstRtpSsrcDemuxPad *dpad = (GstRtpSsrcDemuxPad *) walk->data;
 710     if (dpad->rtp_pad == pad || dpad->rtcp_pad == pad) {
 711       return dpad;
 712     }
 713   }
 714
 715   return NULL;
 716 }
 717
 718
 719 static gboolean
 720 gst_rtp_ssrc_demux_src_event (GstPad * pad, GstObject * parent,
 721     GstEvent * event)
 722 {
 723   GstRtpSsrcDemux *demux;
 724   const GstStructure *s;
 725
 726   demux = GST_RTP_SSRC_DEMUX (parent);
 727
 728   switch (GST_EVENT_TYPE (event)) {
 729     case GST_EVENT_CUSTOM_UPSTREAM:
 730     case GST_EVENT_CUSTOM_BOTH:
 731     case GST_EVENT_CUSTOM_BOTH_OOB:
 732       s = gst_event_get_structure (event);
 733       if (s && !gst_structure_has_field (s, "ssrc")) {
 734         GstRtpSsrcDemuxPad *dpad = find_demux_pad_for_pad (demux, pad);
 735
 736         if (dpad) {
 737           GstStructure *ws;
 738
 739           event = gst_event_make_writable (event);
 740           ws = gst_event_writable_structure (event);
 741           gst_structure_set (ws, "ssrc", G_TYPE_UINT, dpad->ssrc, NULL);
 742         }
 743       }
 744       break;
 745     default:
 746       break;
 747   }
 748
 749   return gst_pad_event_default (pad, parent, event);
 750 }
 751
 752 static GstIterator *
 753 gst_rtp_ssrc_demux_iterate_internal_links_src (GstPad * pad, GstObject * parent)
 754 {
 755   GstRtpSsrcDemux *demux;
 756   GstPad *otherpad = NULL;
 757   GstIterator *it = NULL;
 758   GSList *current;
 759
 760   demux = GST_RTP_SSRC_DEMUX (parent);
 761
 762   GST_PAD_LOCK (demux);
 763   for (current = demux->srcpads; current; current = g_slist_next (current)) {
 764     GstRtpSsrcDemuxPad *dpad = (GstRtpSsrcDemuxPad *) current->data;
 765
 766     if (pad == dpad->rtp_pad) {
 767       otherpad = demux->rtp_sink;
 768       break;
 769     } else if (pad == dpad->rtcp_pad) {
 770       otherpad = demux->rtcp_sink;
 771       break;
 772     }
 773   }
 774   if (otherpad) {
 775     GValue val = { 0, };
 776
 777     g_value_init (&val, GST_TYPE_PAD);
 778     g_value_set_object (&val, otherpad);
 779     it = gst_iterator_new_single (GST_TYPE_PAD, &val);
 780     g_value_unset (&val);
 781
 782   }
 783   GST_PAD_UNLOCK (demux);
 784
 785   return it;
 786 }
 787
 788 /* Should return 0 for elements to be included */
 789 static gint
 790 src_pad_compare_func (gconstpointer a, gconstpointer b)
 791 {
 792   GstPad *pad = GST_PAD (g_value_get_object (a));
 793   const gchar *prefix = g_value_get_string (b);
 794   gint res = 1;
 795
 796   GST_OBJECT_LOCK (pad);
 797   res = !GST_PAD_NAME (pad) || g_str_has_prefix (GST_PAD_NAME (pad), prefix);
 798   GST_OBJECT_UNLOCK (pad);
 799
 800   return res;
 801 }
 802
 803 static GstIterator *
 804 gst_rtp_ssrc_demux_iterate_internal_links_sink (GstPad * pad,
 805     GstObject * parent)
 806 {
 807   GstRtpSsrcDemux *demux;
 808   GstIterator *it = NULL;
 809   GValue gval = { 0, };
 810
 811   demux = GST_RTP_SSRC_DEMUX (parent);
 812
 813   g_value_init (&gval, G_TYPE_STRING);
 814   if (pad == demux->rtp_sink)
 815     g_value_set_static_string (&gval, "src_");
 816   else if (pad == demux->rtcp_sink)
 817     g_value_set_static_string (&gval, "rtcp_src_");
 818   else
 819     g_assert_not_reached ();
 820
 821   it = gst_element_iterate_src_pads (GST_ELEMENT_CAST (demux));
 822   it = gst_iterator_filter (it, src_pad_compare_func, &gval);
 823
 824   return it;
 825 }
 826
 827
 828 static gboolean
 829 gst_rtp_ssrc_demux_src_query (GstPad * pad, GstObject * parent,
 830     GstQuery * query)
 831 {
 832   GstRtpSsrcDemux *demux;
 833   gboolean res = FALSE;
 834
 835   demux = GST_RTP_SSRC_DEMUX (parent);
 836
 837   switch (GST_QUERY_TYPE (query)) {
 838     case GST_QUERY_LATENCY:
 839     {
 840
 841       if ((res = gst_pad_peer_query (demux->rtp_sink, query))) {
 842         gboolean live;
 843         GstClockTime min_latency, max_latency;
 844         GstRtpSsrcDemuxPad *demuxpad;
 845
 846         demuxpad = gst_pad_get_element_private (pad);
 847
 848         gst_query_parse_latency (query, &live, &min_latency, &max_latency);
 849
 850         GST_DEBUG_OBJECT (demux, "peer min latency %" GST_TIME_FORMAT,
 851             GST_TIME_ARGS (min_latency));
 852
 853         GST_DEBUG_OBJECT (demux, "latency for SSRC %08x", demuxpad->ssrc);
 854
 855         gst_query_set_latency (query, live, min_latency, max_latency);
 856       }
 857       break;
 858     }
 859     default:
 860       res = gst_pad_query_default (pad, parent, query);
 861       break;
 862   }
 863
 864   return res;
 865 }
 866
 867 static GstStateChangeReturn
 868 gst_rtp_ssrc_demux_change_state (GstElement * element,
 869     GstStateChange transition)
 870 {
 871   GstStateChangeReturn ret;
 872   GstRtpSsrcDemux *demux;
 873
 874   demux = GST_RTP_SSRC_DEMUX (element);
 875
 876   switch (transition) {
 877     case GST_STATE_CHANGE_NULL_TO_READY:
 878     case GST_STATE_CHANGE_READY_TO_PAUSED:
 879     case GST_STATE_CHANGE_PAUSED_TO_PLAYING:
 880     default:
 881       break;
 882   }
 883
 884   ret = GST_ELEMENT_CLASS (parent_class)->change_state (element, transition);
 885
 886   switch (transition) {
 887     case GST_STATE_CHANGE_PLAYING_TO_PAUSED:
 888       break;
 889     case GST_STATE_CHANGE_PAUSED_TO_READY:
 890       gst_rtp_ssrc_demux_reset (demux);
 891       break;
 892     case GST_STATE_CHANGE_READY_TO_NULL:
 893     default:
 894       break;
 895   }
 896   return ret;
 897 }