From b83f54f1591fdef8347fceef9ab0ca8c70f7ef74 Mon Sep 17 00:00:00 2001
From: Wim Taymans <wim.taymans@collabora.co.uk>
Date: Fri, 15 May 2009 17:58:44 +0200
Subject: [PATCH] media: link the RTP udpsrc to the session manager

Link the RTP udpsrc and the appsrc to the session manager so that they don't
shut down when the client sends a packet to open firewalls.
---
 gst/rtsp-server/rtsp-media.c | 38 +++++++++++++++++++++++++++++++-------
 gst/rtsp-server/rtsp-media.h |  3 +++
 2 files changed, 34 insertions(+), 7 deletions(-)

diff --git a/gst/rtsp-server/rtsp-media.c b/gst/rtsp-server/rtsp-media.c
index f95848e..0e5fec6 100644
--- a/gst/rtsp-server/rtsp-media.c
+++ b/gst/rtsp-server/rtsp-media.c
@@ -761,7 +761,7 @@ setup_stream (GstRTSPMediaStream *stream, guint idx, GstRTSPMedia *media)
   GstPad *pad, *teepad, *selpad;
   GstPadLinkReturn ret;
   gint i;
-  GstElement *tee, *selector;
+  GstElement *tee;
 
   /* allocate udp ports, we will have 4 of them, 2 for receiving RTP/RTCP and 2
    * for sending RTP/RTCP. The sender and receiver ports are shared between the
@@ -801,6 +801,9 @@ setup_stream (GstRTSPMediaStream *stream, guint idx, GstRTSPMedia *media)
   name = g_strdup_printf ("recv_rtcp_sink_%d", idx);
   stream->recv_rtcp_sink = gst_element_get_request_pad (media->rtpbin, name);
   g_free (name);
+  name = g_strdup_printf ("recv_rtp_sink_%d", idx);
+  stream->recv_rtp_sink = gst_element_get_request_pad (media->rtpbin, name);
+  g_free (name);
 
   /* get the session */
   g_signal_emit_by_name (media->rtpbin, "get-internal-session", idx,
@@ -864,22 +867,43 @@ setup_stream (GstRTSPMediaStream *stream, guint idx, GstRTSPMedia *media)
   gst_object_unref (pad);
   gst_object_unref (teepad);
 
+  /* make selector for the RTP receivers */
+  stream->selector[0] = gst_element_factory_make ("input-selector", NULL);
+  g_object_set (stream->selector[0], "select-all", TRUE, NULL);
+  gst_bin_add (GST_BIN_CAST (media->pipeline), stream->selector[0]);
+
+  pad = gst_element_get_static_pad (stream->selector[0], "src");
+  gst_pad_link (pad, stream->recv_rtp_sink);
+  gst_object_unref (pad);
+
+  selpad = gst_element_get_request_pad (stream->selector[0], "sink%d");
+  pad = gst_element_get_static_pad (stream->udpsrc[0], "src");
+  gst_pad_link (pad, selpad);
+  gst_object_unref (pad);
+  gst_object_unref (selpad);
+
+  selpad = gst_element_get_request_pad (stream->selector[0], "sink%d");
+  pad = gst_element_get_static_pad (stream->appsrc[0], "src");
+  gst_pad_link (pad, selpad);
+  gst_object_unref (pad);
+  gst_object_unref (selpad);
+
   /* make selector for the RTCP receivers */
-  selector = gst_element_factory_make ("input-selector", NULL);
-  g_object_set (selector, "select-all", TRUE, NULL);
-  gst_bin_add (GST_BIN_CAST (media->pipeline), selector);
+  stream->selector[1] = gst_element_factory_make ("input-selector", NULL);
+  g_object_set (stream->selector[1], "select-all", TRUE, NULL);
+  gst_bin_add (GST_BIN_CAST (media->pipeline), stream->selector[1]);
 
-  pad = gst_element_get_static_pad (selector, "src");
+  pad = gst_element_get_static_pad (stream->selector[1], "src");
   gst_pad_link (pad, stream->recv_rtcp_sink);
   gst_object_unref (pad);
 
-  selpad = gst_element_get_request_pad (selector, "sink%d");
+  selpad = gst_element_get_request_pad (stream->selector[1], "sink%d");
   pad = gst_element_get_static_pad (stream->udpsrc[1], "src");
   gst_pad_link (pad, selpad);
   gst_object_unref (pad);
   gst_object_unref (selpad);
 
-  selpad = gst_element_get_request_pad (selector, "sink%d");
+  selpad = gst_element_get_request_pad (stream->selector[1], "sink%d");
   pad = gst_element_get_static_pad (stream->appsrc[1], "src");
   gst_pad_link (pad, selpad);
   gst_object_unref (pad);
diff --git a/gst/rtsp-server/rtsp-media.h b/gst/rtsp-server/rtsp-media.h
index 3273f53..a7e9a1c 100644
--- a/gst/rtsp-server/rtsp-media.h
+++ b/gst/rtsp-server/rtsp-media.h
@@ -94,6 +94,7 @@ struct _GstRTSPMediaStream {
 
   /* pads on the rtpbin */
   GstPad       *recv_rtcp_sink;
+  GstPad       *recv_rtp_sink;
   GstPad       *send_rtp_sink;
   GstPad       *send_rtp_src;
   GstPad       *send_rtcp_src;
@@ -109,6 +110,8 @@ struct _GstRTSPMediaStream {
   GstElement   *appsrc[2];
   GstElement   *appsink[2];
 
+  GstElement   *selector[2];
+
   /* server ports for sending/receiving */
   GstRTSPRange  server_port;
 
-- 
2.7.4