stream: fix docs for recently-added get/set_buffer_size API
[platform/upstream/gstreamer.git] / gst / rtsp-server / rtsp-stream.c
1 /* GStreamer
2  * Copyright (C) 2008 Wim Taymans <wim.taymans at gmail.com>
3  * Copyright (C) 2015 Centricular Ltd
4  *     Author: Sebastian Dröge <sebastian@centricular.com>
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., 51 Franklin St, Fifth Floor,
19  * Boston, MA 02110-1301, USA.
20  */
21 /**
22  * SECTION:rtsp-stream
23  * @short_description: A media stream
24  * @see_also: #GstRTSPMedia
25  *
26  * The #GstRTSPStream object manages the data transport for one stream. It
27  * is created from a payloader element and a source pad that produce the RTP
28  * packets for the stream.
29  *
30  * With gst_rtsp_stream_join_bin() the streaming elements are added to the bin
31  * and rtpbin. gst_rtsp_stream_leave_bin() removes the elements again.
32  *
33  * The #GstRTSPStream will use the configured addresspool, as set with
34  * gst_rtsp_stream_set_address_pool(), to allocate multicast addresses for the
35  * stream. With gst_rtsp_stream_get_multicast_address() you can get the
36  * configured address.
37  *
38  * With gst_rtsp_stream_get_server_port () you can get the port that the server
39  * will use to receive RTCP. This is the part that the clients will use to send
40  * RTCP to.
41  *
42  * With gst_rtsp_stream_add_transport() destinations can be added where the
43  * stream should be sent to. Use gst_rtsp_stream_remove_transport() to remove
44  * the destination again.
45  *
46  * Last reviewed on 2013-07-16 (1.0.0)
47  */
48
49 #include <stdlib.h>
50 #include <stdio.h>
51 #include <string.h>
52
53 #include <gio/gio.h>
54
55 #include <gst/app/gstappsrc.h>
56 #include <gst/app/gstappsink.h>
57
58 #include <gst/rtp/gstrtpbuffer.h>
59
60 #include "rtsp-stream.h"
61
62 #define GST_RTSP_STREAM_GET_PRIVATE(obj)  \
63      (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GST_TYPE_RTSP_STREAM, GstRTSPStreamPrivate))
64
65 typedef struct
66 {
67   GstRTSPStreamTransport *transport;
68
69   /* RTP and RTCP source */
70   GstElement *udpsrc[2];
71   GstPad *selpad[2];
72 } GstRTSPMulticastTransportSource;
73
74 struct _GstRTSPStreamPrivate
75 {
76   GMutex lock;
77   guint idx;
78   /* Only one pad is ever set */
79   GstPad *srcpad, *sinkpad;
80   GstElement *payloader;
81   guint buffer_size;
82   gboolean is_joined;
83   gchar *control;
84
85   GstRTSPProfile profiles;
86   GstRTSPLowerTrans protocols;
87
88   /* pads on the rtpbin */
89   GstPad *send_rtp_sink;
90   GstPad *recv_rtp_src;
91   GstPad *recv_sink[2];
92   GstPad *send_src[2];
93
94   /* the RTPSession object */
95   GObject *session;
96
97   /* SRTP encoder/decoder */
98   GstElement *srtpenc;
99   GstElement *srtpdec;
100   GHashTable *keys;
101
102   /* sinks used for sending and receiving RTP and RTCP over ipv4, they share
103    * sockets */
104   GstElement *udpsrc_v4[2];
105
106   /* sinks used for sending and receiving RTP and RTCP over ipv6, they share
107    * sockets */
108   GstElement *udpsrc_v6[2];
109
110   GstElement *udpqueue[2];
111   GstElement *udpsink[2];
112
113   /* for TCP transport */
114   GstElement *appsrc[2];
115   GstClockTime appsrc_base_time[2];
116   GstElement *appqueue[2];
117   GstElement *appsink[2];
118
119   GstElement *tee[2];
120   GstElement *funnel[2];
121
122   /* retransmission */
123   GstElement *rtxsend;
124   guint rtx_pt;
125   GstClockTime rtx_time;
126
127   /* server ports for sending/receiving over ipv4 */
128   GstRTSPRange server_port_v4;
129   GstRTSPAddress *server_addr_v4;
130   gboolean have_ipv4;
131
132   /* server ports for sending/receiving over ipv6 */
133   GstRTSPRange server_port_v6;
134   GstRTSPAddress *server_addr_v6;
135   gboolean have_ipv6;
136
137   /* multicast addresses */
138   GstRTSPAddressPool *pool;
139   GstRTSPAddress *addr_v4;
140   GstRTSPAddress *addr_v6;
141
142   /* the caps of the stream */
143   gulong caps_sig;
144   GstCaps *caps;
145
146   /* transports we stream to */
147   guint n_active;
148   GList *transports;
149   guint transports_cookie;
150   GList *tr_cache_rtp;
151   GList *tr_cache_rtcp;
152   guint tr_cache_cookie_rtp;
153   guint tr_cache_cookie_rtcp;
154
155
156   /* UDP sources for UDP multicast transports */
157   GList *transport_sources;
158
159   gint dscp_qos;
160
161   /* stream blocking */
162   gulong blocked_id;
163   gboolean blocking;
164
165   /* pt->caps map for RECORD streams */
166   GHashTable *ptmap;
167 };
168
169 #define DEFAULT_CONTROL         NULL
170 #define DEFAULT_PROFILES        GST_RTSP_PROFILE_AVP
171 #define DEFAULT_PROTOCOLS       GST_RTSP_LOWER_TRANS_UDP | GST_RTSP_LOWER_TRANS_UDP_MCAST | \
172                                         GST_RTSP_LOWER_TRANS_TCP
173
174 enum
175 {
176   PROP_0,
177   PROP_CONTROL,
178   PROP_PROFILES,
179   PROP_PROTOCOLS,
180   PROP_LAST
181 };
182
183 enum
184 {
185   SIGNAL_NEW_RTP_ENCODER,
186   SIGNAL_NEW_RTCP_ENCODER,
187   SIGNAL_LAST
188 };
189
190 GST_DEBUG_CATEGORY_STATIC (rtsp_stream_debug);
191 #define GST_CAT_DEFAULT rtsp_stream_debug
192
193 static GQuark ssrc_stream_map_key;
194
195 static void gst_rtsp_stream_get_property (GObject * object, guint propid,
196     GValue * value, GParamSpec * pspec);
197 static void gst_rtsp_stream_set_property (GObject * object, guint propid,
198     const GValue * value, GParamSpec * pspec);
199
200 static void gst_rtsp_stream_finalize (GObject * obj);
201
202 static guint gst_rtsp_stream_signals[SIGNAL_LAST] = { 0 };
203
204 G_DEFINE_TYPE (GstRTSPStream, gst_rtsp_stream, G_TYPE_OBJECT);
205
206 static void
207 gst_rtsp_stream_class_init (GstRTSPStreamClass * klass)
208 {
209   GObjectClass *gobject_class;
210
211   g_type_class_add_private (klass, sizeof (GstRTSPStreamPrivate));
212
213   gobject_class = G_OBJECT_CLASS (klass);
214
215   gobject_class->get_property = gst_rtsp_stream_get_property;
216   gobject_class->set_property = gst_rtsp_stream_set_property;
217   gobject_class->finalize = gst_rtsp_stream_finalize;
218
219   g_object_class_install_property (gobject_class, PROP_CONTROL,
220       g_param_spec_string ("control", "Control",
221           "The control string for this stream", DEFAULT_CONTROL,
222           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
223
224   g_object_class_install_property (gobject_class, PROP_PROFILES,
225       g_param_spec_flags ("profiles", "Profiles",
226           "Allowed transfer profiles", GST_TYPE_RTSP_PROFILE,
227           DEFAULT_PROFILES, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
228
229   g_object_class_install_property (gobject_class, PROP_PROTOCOLS,
230       g_param_spec_flags ("protocols", "Protocols",
231           "Allowed lower transport protocols", GST_TYPE_RTSP_LOWER_TRANS,
232           DEFAULT_PROTOCOLS, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
233
234   gst_rtsp_stream_signals[SIGNAL_NEW_RTP_ENCODER] =
235       g_signal_new ("new-rtp-encoder", G_TYPE_FROM_CLASS (klass),
236       G_SIGNAL_RUN_LAST, 0, NULL, NULL, g_cclosure_marshal_generic,
237       G_TYPE_NONE, 1, GST_TYPE_ELEMENT);
238
239   gst_rtsp_stream_signals[SIGNAL_NEW_RTCP_ENCODER] =
240       g_signal_new ("new-rtcp-encoder", G_TYPE_FROM_CLASS (klass),
241       G_SIGNAL_RUN_LAST, 0, NULL, NULL, g_cclosure_marshal_generic,
242       G_TYPE_NONE, 1, GST_TYPE_ELEMENT);
243
244   GST_DEBUG_CATEGORY_INIT (rtsp_stream_debug, "rtspstream", 0, "GstRTSPStream");
245
246   ssrc_stream_map_key = g_quark_from_static_string ("GstRTSPServer.stream");
247 }
248
249 static void
250 gst_rtsp_stream_init (GstRTSPStream * stream)
251 {
252   GstRTSPStreamPrivate *priv = GST_RTSP_STREAM_GET_PRIVATE (stream);
253
254   GST_DEBUG ("new stream %p", stream);
255
256   stream->priv = priv;
257
258   priv->dscp_qos = -1;
259   priv->control = g_strdup (DEFAULT_CONTROL);
260   priv->profiles = DEFAULT_PROFILES;
261   priv->protocols = DEFAULT_PROTOCOLS;
262
263   g_mutex_init (&priv->lock);
264
265   priv->keys = g_hash_table_new_full (g_direct_hash, g_direct_equal,
266       NULL, (GDestroyNotify) gst_caps_unref);
267   priv->ptmap = g_hash_table_new_full (NULL, NULL, NULL,
268       (GDestroyNotify) gst_caps_unref);
269 }
270
271 static void
272 gst_rtsp_stream_finalize (GObject * obj)
273 {
274   GstRTSPStream *stream;
275   GstRTSPStreamPrivate *priv;
276
277   stream = GST_RTSP_STREAM (obj);
278   priv = stream->priv;
279
280   GST_DEBUG ("finalize stream %p", stream);
281
282   /* we really need to be unjoined now */
283   g_return_if_fail (!priv->is_joined);
284
285   if (priv->addr_v4)
286     gst_rtsp_address_free (priv->addr_v4);
287   if (priv->addr_v6)
288     gst_rtsp_address_free (priv->addr_v6);
289   if (priv->server_addr_v4)
290     gst_rtsp_address_free (priv->server_addr_v4);
291   if (priv->server_addr_v6)
292     gst_rtsp_address_free (priv->server_addr_v6);
293   if (priv->pool)
294     g_object_unref (priv->pool);
295   if (priv->rtxsend)
296     g_object_unref (priv->rtxsend);
297
298   gst_object_unref (priv->payloader);
299   if (priv->srcpad)
300     gst_object_unref (priv->srcpad);
301   if (priv->sinkpad)
302     gst_object_unref (priv->sinkpad);
303   g_free (priv->control);
304   g_mutex_clear (&priv->lock);
305
306   g_hash_table_unref (priv->keys);
307   g_hash_table_destroy (priv->ptmap);
308
309   G_OBJECT_CLASS (gst_rtsp_stream_parent_class)->finalize (obj);
310 }
311
312 static void
313 gst_rtsp_stream_get_property (GObject * object, guint propid,
314     GValue * value, GParamSpec * pspec)
315 {
316   GstRTSPStream *stream = GST_RTSP_STREAM (object);
317
318   switch (propid) {
319     case PROP_CONTROL:
320       g_value_take_string (value, gst_rtsp_stream_get_control (stream));
321       break;
322     case PROP_PROFILES:
323       g_value_set_flags (value, gst_rtsp_stream_get_profiles (stream));
324       break;
325     case PROP_PROTOCOLS:
326       g_value_set_flags (value, gst_rtsp_stream_get_protocols (stream));
327       break;
328     default:
329       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
330   }
331 }
332
333 static void
334 gst_rtsp_stream_set_property (GObject * object, guint propid,
335     const GValue * value, GParamSpec * pspec)
336 {
337   GstRTSPStream *stream = GST_RTSP_STREAM (object);
338
339   switch (propid) {
340     case PROP_CONTROL:
341       gst_rtsp_stream_set_control (stream, g_value_get_string (value));
342       break;
343     case PROP_PROFILES:
344       gst_rtsp_stream_set_profiles (stream, g_value_get_flags (value));
345       break;
346     case PROP_PROTOCOLS:
347       gst_rtsp_stream_set_protocols (stream, g_value_get_flags (value));
348       break;
349     default:
350       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
351   }
352 }
353
354 /**
355  * gst_rtsp_stream_new:
356  * @idx: an index
357  * @pad: a #GstPad
358  * @payloader: a #GstElement
359  *
360  * Create a new media stream with index @idx that handles RTP data on
361  * @pad and has a payloader element @payloader if @pad is a source pad
362  * or a depayloader element @payloader if @pad is a sink pad.
363  *
364  * Returns: (transfer full): a new #GstRTSPStream
365  */
366 GstRTSPStream *
367 gst_rtsp_stream_new (guint idx, GstElement * payloader, GstPad * pad)
368 {
369   GstRTSPStreamPrivate *priv;
370   GstRTSPStream *stream;
371
372   g_return_val_if_fail (GST_IS_ELEMENT (payloader), NULL);
373   g_return_val_if_fail (GST_IS_PAD (pad), NULL);
374
375   stream = g_object_new (GST_TYPE_RTSP_STREAM, NULL);
376   priv = stream->priv;
377   priv->idx = idx;
378   priv->payloader = gst_object_ref (payloader);
379   if (GST_PAD_IS_SRC (pad))
380     priv->srcpad = gst_object_ref (pad);
381   else
382     priv->sinkpad = gst_object_ref (pad);
383
384   return stream;
385 }
386
387 /**
388  * gst_rtsp_stream_get_index:
389  * @stream: a #GstRTSPStream
390  *
391  * Get the stream index.
392  *
393  * Return: the stream index.
394  */
395 guint
396 gst_rtsp_stream_get_index (GstRTSPStream * stream)
397 {
398   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), -1);
399
400   return stream->priv->idx;
401 }
402
403 /**
404  * gst_rtsp_stream_get_pt:
405  * @stream: a #GstRTSPStream
406  *
407  * Get the stream payload type.
408  *
409  * Return: the stream payload type.
410  */
411 guint
412 gst_rtsp_stream_get_pt (GstRTSPStream * stream)
413 {
414   GstRTSPStreamPrivate *priv;
415   guint pt;
416
417   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), -1);
418
419   priv = stream->priv;
420
421   g_object_get (G_OBJECT (priv->payloader), "pt", &pt, NULL);
422
423   return pt;
424 }
425
426 /**
427  * gst_rtsp_stream_get_srcpad:
428  * @stream: a #GstRTSPStream
429  *
430  * Get the srcpad associated with @stream.
431  *
432  * Returns: (transfer full): the srcpad. Unref after usage.
433  */
434 GstPad *
435 gst_rtsp_stream_get_srcpad (GstRTSPStream * stream)
436 {
437   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
438
439   if (!stream->priv->srcpad)
440     return NULL;
441
442   return gst_object_ref (stream->priv->srcpad);
443 }
444
445 /**
446  * gst_rtsp_stream_get_sinkpad:
447  * @stream: a #GstRTSPStream
448  *
449  * Get the sinkpad associated with @stream.
450  *
451  * Returns: (transfer full): the sinkpad. Unref after usage.
452  */
453 GstPad *
454 gst_rtsp_stream_get_sinkpad (GstRTSPStream * stream)
455 {
456   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
457
458   if (!stream->priv->sinkpad)
459     return NULL;
460
461   return gst_object_ref (stream->priv->sinkpad);
462 }
463
464 /**
465  * gst_rtsp_stream_get_control:
466  * @stream: a #GstRTSPStream
467  *
468  * Get the control string to identify this stream.
469  *
470  * Returns: (transfer full): the control string. g_free() after usage.
471  */
472 gchar *
473 gst_rtsp_stream_get_control (GstRTSPStream * stream)
474 {
475   GstRTSPStreamPrivate *priv;
476   gchar *result;
477
478   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
479
480   priv = stream->priv;
481
482   g_mutex_lock (&priv->lock);
483   if ((result = g_strdup (priv->control)) == NULL)
484     result = g_strdup_printf ("stream=%u", priv->idx);
485   g_mutex_unlock (&priv->lock);
486
487   return result;
488 }
489
490 /**
491  * gst_rtsp_stream_set_control:
492  * @stream: a #GstRTSPStream
493  * @control: a control string
494  *
495  * Set the control string in @stream.
496  */
497 void
498 gst_rtsp_stream_set_control (GstRTSPStream * stream, const gchar * control)
499 {
500   GstRTSPStreamPrivate *priv;
501
502   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
503
504   priv = stream->priv;
505
506   g_mutex_lock (&priv->lock);
507   g_free (priv->control);
508   priv->control = g_strdup (control);
509   g_mutex_unlock (&priv->lock);
510 }
511
512 /**
513  * gst_rtsp_stream_has_control:
514  * @stream: a #GstRTSPStream
515  * @control: a control string
516  *
517  * Check if @stream has the control string @control.
518  *
519  * Returns: %TRUE is @stream has @control as the control string
520  */
521 gboolean
522 gst_rtsp_stream_has_control (GstRTSPStream * stream, const gchar * control)
523 {
524   GstRTSPStreamPrivate *priv;
525   gboolean res;
526
527   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
528
529   priv = stream->priv;
530
531   g_mutex_lock (&priv->lock);
532   if (priv->control)
533     res = (g_strcmp0 (priv->control, control) == 0);
534   else {
535     guint streamid;
536
537     if (sscanf (control, "stream=%u", &streamid) > 0)
538       res = (streamid == priv->idx);
539     else
540       res = FALSE;
541   }
542   g_mutex_unlock (&priv->lock);
543
544   return res;
545 }
546
547 /**
548  * gst_rtsp_stream_set_mtu:
549  * @stream: a #GstRTSPStream
550  * @mtu: a new MTU
551  *
552  * Configure the mtu in the payloader of @stream to @mtu.
553  */
554 void
555 gst_rtsp_stream_set_mtu (GstRTSPStream * stream, guint mtu)
556 {
557   GstRTSPStreamPrivate *priv;
558
559   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
560
561   priv = stream->priv;
562
563   GST_LOG_OBJECT (stream, "set MTU %u", mtu);
564
565   g_object_set (G_OBJECT (priv->payloader), "mtu", mtu, NULL);
566 }
567
568 /**
569  * gst_rtsp_stream_get_mtu:
570  * @stream: a #GstRTSPStream
571  *
572  * Get the configured MTU in the payloader of @stream.
573  *
574  * Returns: the MTU of the payloader.
575  */
576 guint
577 gst_rtsp_stream_get_mtu (GstRTSPStream * stream)
578 {
579   GstRTSPStreamPrivate *priv;
580   guint mtu;
581
582   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), 0);
583
584   priv = stream->priv;
585
586   g_object_get (G_OBJECT (priv->payloader), "mtu", &mtu, NULL);
587
588   return mtu;
589 }
590
591 /* Update the dscp qos property on the udp sinks */
592 static void
593 update_dscp_qos (GstRTSPStream * stream)
594 {
595   GstRTSPStreamPrivate *priv;
596
597   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
598
599   priv = stream->priv;
600
601   if (priv->udpsink[0]) {
602     g_object_set (G_OBJECT (priv->udpsink[0]), "qos-dscp", priv->dscp_qos,
603         NULL);
604   }
605
606   if (priv->udpsink[1]) {
607     g_object_set (G_OBJECT (priv->udpsink[1]), "qos-dscp", priv->dscp_qos,
608         NULL);
609   }
610 }
611
612 /**
613  * gst_rtsp_stream_set_dscp_qos:
614  * @stream: a #GstRTSPStream
615  * @dscp_qos: a new dscp qos value (0-63, or -1 to disable)
616  *
617  * Configure the dscp qos of the outgoing sockets to @dscp_qos.
618  */
619 void
620 gst_rtsp_stream_set_dscp_qos (GstRTSPStream * stream, gint dscp_qos)
621 {
622   GstRTSPStreamPrivate *priv;
623
624   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
625
626   priv = stream->priv;
627
628   GST_LOG_OBJECT (stream, "set DSCP QoS %d", dscp_qos);
629
630   if (dscp_qos < -1 || dscp_qos > 63) {
631     GST_WARNING_OBJECT (stream, "trying to set illegal dscp qos %d", dscp_qos);
632     return;
633   }
634
635   priv->dscp_qos = dscp_qos;
636
637   update_dscp_qos (stream);
638 }
639
640 /**
641  * gst_rtsp_stream_get_dscp_qos:
642  * @stream: a #GstRTSPStream
643  *
644  * Get the configured DSCP QoS in of the outgoing sockets.
645  *
646  * Returns: the DSCP QoS value of the outgoing sockets, or -1 if disbled.
647  */
648 gint
649 gst_rtsp_stream_get_dscp_qos (GstRTSPStream * stream)
650 {
651   GstRTSPStreamPrivate *priv;
652
653   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), -1);
654
655   priv = stream->priv;
656
657   return priv->dscp_qos;
658 }
659
660 /**
661  * gst_rtsp_stream_is_transport_supported:
662  * @stream: a #GstRTSPStream
663  * @transport: (transfer none): a #GstRTSPTransport
664  *
665  * Check if @transport can be handled by stream
666  *
667  * Returns: %TRUE if @transport can be handled by @stream.
668  */
669 gboolean
670 gst_rtsp_stream_is_transport_supported (GstRTSPStream * stream,
671     GstRTSPTransport * transport)
672 {
673   GstRTSPStreamPrivate *priv;
674
675   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
676
677   priv = stream->priv;
678
679   g_mutex_lock (&priv->lock);
680   if (transport->trans != GST_RTSP_TRANS_RTP)
681     goto unsupported_transmode;
682
683   if (!(transport->profile & priv->profiles))
684     goto unsupported_profile;
685
686   if (!(transport->lower_transport & priv->protocols))
687     goto unsupported_ltrans;
688
689   g_mutex_unlock (&priv->lock);
690
691   return TRUE;
692
693   /* ERRORS */
694 unsupported_transmode:
695   {
696     GST_DEBUG ("unsupported transport mode %d", transport->trans);
697     g_mutex_unlock (&priv->lock);
698     return FALSE;
699   }
700 unsupported_profile:
701   {
702     GST_DEBUG ("unsupported profile %d", transport->profile);
703     g_mutex_unlock (&priv->lock);
704     return FALSE;
705   }
706 unsupported_ltrans:
707   {
708     GST_DEBUG ("unsupported lower transport %d", transport->lower_transport);
709     g_mutex_unlock (&priv->lock);
710     return FALSE;
711   }
712 }
713
714 /**
715  * gst_rtsp_stream_set_profiles:
716  * @stream: a #GstRTSPStream
717  * @profiles: the new profiles
718  *
719  * Configure the allowed profiles for @stream.
720  */
721 void
722 gst_rtsp_stream_set_profiles (GstRTSPStream * stream, GstRTSPProfile profiles)
723 {
724   GstRTSPStreamPrivate *priv;
725
726   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
727
728   priv = stream->priv;
729
730   g_mutex_lock (&priv->lock);
731   priv->profiles = profiles;
732   g_mutex_unlock (&priv->lock);
733 }
734
735 /**
736  * gst_rtsp_stream_get_profiles:
737  * @stream: a #GstRTSPStream
738  *
739  * Get the allowed profiles of @stream.
740  *
741  * Returns: a #GstRTSPProfile
742  */
743 GstRTSPProfile
744 gst_rtsp_stream_get_profiles (GstRTSPStream * stream)
745 {
746   GstRTSPStreamPrivate *priv;
747   GstRTSPProfile res;
748
749   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), GST_RTSP_PROFILE_UNKNOWN);
750
751   priv = stream->priv;
752
753   g_mutex_lock (&priv->lock);
754   res = priv->profiles;
755   g_mutex_unlock (&priv->lock);
756
757   return res;
758 }
759
760 /**
761  * gst_rtsp_stream_set_protocols:
762  * @stream: a #GstRTSPStream
763  * @protocols: the new flags
764  *
765  * Configure the allowed lower transport for @stream.
766  */
767 void
768 gst_rtsp_stream_set_protocols (GstRTSPStream * stream,
769     GstRTSPLowerTrans protocols)
770 {
771   GstRTSPStreamPrivate *priv;
772
773   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
774
775   priv = stream->priv;
776
777   g_mutex_lock (&priv->lock);
778   priv->protocols = protocols;
779   g_mutex_unlock (&priv->lock);
780 }
781
782 /**
783  * gst_rtsp_stream_get_protocols:
784  * @stream: a #GstRTSPStream
785  *
786  * Get the allowed protocols of @stream.
787  *
788  * Returns: a #GstRTSPLowerTrans
789  */
790 GstRTSPLowerTrans
791 gst_rtsp_stream_get_protocols (GstRTSPStream * stream)
792 {
793   GstRTSPStreamPrivate *priv;
794   GstRTSPLowerTrans res;
795
796   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream),
797       GST_RTSP_LOWER_TRANS_UNKNOWN);
798
799   priv = stream->priv;
800
801   g_mutex_lock (&priv->lock);
802   res = priv->protocols;
803   g_mutex_unlock (&priv->lock);
804
805   return res;
806 }
807
808 /**
809  * gst_rtsp_stream_set_address_pool:
810  * @stream: a #GstRTSPStream
811  * @pool: (transfer none): a #GstRTSPAddressPool
812  *
813  * configure @pool to be used as the address pool of @stream.
814  */
815 void
816 gst_rtsp_stream_set_address_pool (GstRTSPStream * stream,
817     GstRTSPAddressPool * pool)
818 {
819   GstRTSPStreamPrivate *priv;
820   GstRTSPAddressPool *old;
821
822   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
823
824   priv = stream->priv;
825
826   GST_LOG_OBJECT (stream, "set address pool %p", pool);
827
828   g_mutex_lock (&priv->lock);
829   if ((old = priv->pool) != pool)
830     priv->pool = pool ? g_object_ref (pool) : NULL;
831   else
832     old = NULL;
833   g_mutex_unlock (&priv->lock);
834
835   if (old)
836     g_object_unref (old);
837 }
838
839 /**
840  * gst_rtsp_stream_get_address_pool:
841  * @stream: a #GstRTSPStream
842  *
843  * Get the #GstRTSPAddressPool used as the address pool of @stream.
844  *
845  * Returns: (transfer full): the #GstRTSPAddressPool of @stream. g_object_unref() after
846  * usage.
847  */
848 GstRTSPAddressPool *
849 gst_rtsp_stream_get_address_pool (GstRTSPStream * stream)
850 {
851   GstRTSPStreamPrivate *priv;
852   GstRTSPAddressPool *result;
853
854   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
855
856   priv = stream->priv;
857
858   g_mutex_lock (&priv->lock);
859   if ((result = priv->pool))
860     g_object_ref (result);
861   g_mutex_unlock (&priv->lock);
862
863   return result;
864 }
865
866 /**
867  * gst_rtsp_stream_get_multicast_address:
868  * @stream: a #GstRTSPStream
869  * @family: the #GSocketFamily
870  *
871  * Get the multicast address of @stream for @family.
872  *
873  * Returns: (transfer full) (nullable): the #GstRTSPAddress of @stream
874  * or %NULL when no address could be allocated. gst_rtsp_address_free()
875  * after usage.
876  */
877 GstRTSPAddress *
878 gst_rtsp_stream_get_multicast_address (GstRTSPStream * stream,
879     GSocketFamily family)
880 {
881   GstRTSPStreamPrivate *priv;
882   GstRTSPAddress *result;
883   GstRTSPAddress **addrp;
884   GstRTSPAddressFlags flags;
885
886   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
887
888   priv = stream->priv;
889
890   if (family == G_SOCKET_FAMILY_IPV6) {
891     flags = GST_RTSP_ADDRESS_FLAG_IPV6;
892     addrp = &priv->addr_v6;
893   } else {
894     flags = GST_RTSP_ADDRESS_FLAG_IPV4;
895     addrp = &priv->addr_v4;
896   }
897
898   g_mutex_lock (&priv->lock);
899   if (*addrp == NULL) {
900     if (priv->pool == NULL)
901       goto no_pool;
902
903     flags |= GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_MULTICAST;
904
905     *addrp = gst_rtsp_address_pool_acquire_address (priv->pool, flags, 2);
906     if (*addrp == NULL)
907       goto no_address;
908   }
909   result = gst_rtsp_address_copy (*addrp);
910   g_mutex_unlock (&priv->lock);
911
912   return result;
913
914   /* ERRORS */
915 no_pool:
916   {
917     GST_ERROR_OBJECT (stream, "no address pool specified");
918     g_mutex_unlock (&priv->lock);
919     return NULL;
920   }
921 no_address:
922   {
923     GST_ERROR_OBJECT (stream, "failed to acquire address from pool");
924     g_mutex_unlock (&priv->lock);
925     return NULL;
926   }
927 }
928
929 /**
930  * gst_rtsp_stream_reserve_address:
931  * @stream: a #GstRTSPStream
932  * @address: an address
933  * @port: a port
934  * @n_ports: n_ports
935  * @ttl: a TTL
936  *
937  * Reserve @address and @port as the address and port of @stream.
938  *
939  * Returns: (nullable): the #GstRTSPAddress of @stream or %NULL when
940  * the address could be reserved. gst_rtsp_address_free() after usage.
941  */
942 GstRTSPAddress *
943 gst_rtsp_stream_reserve_address (GstRTSPStream * stream,
944     const gchar * address, guint port, guint n_ports, guint ttl)
945 {
946   GstRTSPStreamPrivate *priv;
947   GstRTSPAddress *result;
948   GInetAddress *addr;
949   GSocketFamily family;
950   GstRTSPAddress **addrp;
951
952   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
953   g_return_val_if_fail (address != NULL, NULL);
954   g_return_val_if_fail (port > 0, NULL);
955   g_return_val_if_fail (n_ports > 0, NULL);
956   g_return_val_if_fail (ttl > 0, NULL);
957
958   priv = stream->priv;
959
960   addr = g_inet_address_new_from_string (address);
961   if (!addr) {
962     GST_ERROR ("failed to get inet addr from %s", address);
963     family = G_SOCKET_FAMILY_IPV4;
964   } else {
965     family = g_inet_address_get_family (addr);
966     g_object_unref (addr);
967   }
968
969   if (family == G_SOCKET_FAMILY_IPV6)
970     addrp = &priv->addr_v6;
971   else
972     addrp = &priv->addr_v4;
973
974   g_mutex_lock (&priv->lock);
975   if (*addrp == NULL) {
976     GstRTSPAddressPoolResult res;
977
978     if (priv->pool == NULL)
979       goto no_pool;
980
981     res = gst_rtsp_address_pool_reserve_address (priv->pool, address,
982         port, n_ports, ttl, addrp);
983     if (res != GST_RTSP_ADDRESS_POOL_OK)
984       goto no_address;
985   } else {
986     if (strcmp ((*addrp)->address, address) ||
987         (*addrp)->port != port || (*addrp)->n_ports != n_ports ||
988         (*addrp)->ttl != ttl)
989       goto different_address;
990   }
991   result = gst_rtsp_address_copy (*addrp);
992   g_mutex_unlock (&priv->lock);
993
994   return result;
995
996   /* ERRORS */
997 no_pool:
998   {
999     GST_ERROR_OBJECT (stream, "no address pool specified");
1000     g_mutex_unlock (&priv->lock);
1001     return NULL;
1002   }
1003 no_address:
1004   {
1005     GST_ERROR_OBJECT (stream, "failed to acquire address %s from pool",
1006         address);
1007     g_mutex_unlock (&priv->lock);
1008     return NULL;
1009   }
1010 different_address:
1011   {
1012     GST_ERROR_OBJECT (stream, "address %s is not the same that was already"
1013         " reserved", address);
1014     g_mutex_unlock (&priv->lock);
1015     return NULL;
1016   }
1017 }
1018
1019 static gboolean
1020 alloc_ports_one_family (GstRTSPStream * stream, GstRTSPAddressPool * pool,
1021     gint buffer_size, GSocketFamily family, GstElement * udpsrc_out[2],
1022     GstElement * udpsink_out[2], GstRTSPRange * server_port_out,
1023     GstRTSPAddress ** server_addr_out)
1024 {
1025   GstRTSPStreamPrivate *priv = stream->priv;
1026   GstStateChangeReturn ret;
1027   GstElement *udpsrc0, *udpsrc1;
1028   GstElement *udpsink0, *udpsink1;
1029   GSocket *rtp_socket = NULL;
1030   GSocket *rtcp_socket;
1031   gint tmp_rtp, tmp_rtcp;
1032   guint count;
1033   gint rtpport, rtcpport;
1034   GList *rejected_addresses = NULL;
1035   GstRTSPAddress *addr = NULL;
1036   GInetAddress *inetaddr = NULL;
1037   GSocketAddress *rtp_sockaddr = NULL;
1038   GSocketAddress *rtcp_sockaddr = NULL;
1039   const gchar *multisink_socket;
1040
1041   if (family == G_SOCKET_FAMILY_IPV6)
1042     multisink_socket = "socket-v6";
1043   else
1044     multisink_socket = "socket";
1045
1046   udpsrc0 = NULL;
1047   udpsrc1 = NULL;
1048   udpsink0 = NULL;
1049   udpsink1 = NULL;
1050   count = 0;
1051
1052   /* Start with random port */
1053   tmp_rtp = 0;
1054
1055   rtcp_socket = g_socket_new (family, G_SOCKET_TYPE_DATAGRAM,
1056       G_SOCKET_PROTOCOL_UDP, NULL);
1057   if (!rtcp_socket)
1058     goto no_udp_protocol;
1059
1060   if (*server_addr_out)
1061     gst_rtsp_address_free (*server_addr_out);
1062
1063   /* try to allocate 2 UDP ports, the RTP port should be an even
1064    * number and the RTCP port should be the next (uneven) port */
1065 again:
1066
1067   if (rtp_socket == NULL) {
1068     rtp_socket = g_socket_new (family, G_SOCKET_TYPE_DATAGRAM,
1069         G_SOCKET_PROTOCOL_UDP, NULL);
1070     if (!rtp_socket)
1071       goto no_udp_protocol;
1072   }
1073
1074   if (pool && gst_rtsp_address_pool_has_unicast_addresses (pool)) {
1075     GstRTSPAddressFlags flags;
1076
1077     if (addr)
1078       rejected_addresses = g_list_prepend (rejected_addresses, addr);
1079
1080     flags = GST_RTSP_ADDRESS_FLAG_EVEN_PORT | GST_RTSP_ADDRESS_FLAG_UNICAST;
1081     if (family == G_SOCKET_FAMILY_IPV6)
1082       flags |= GST_RTSP_ADDRESS_FLAG_IPV6;
1083     else
1084       flags |= GST_RTSP_ADDRESS_FLAG_IPV4;
1085
1086     addr = gst_rtsp_address_pool_acquire_address (pool, flags, 2);
1087
1088     if (addr == NULL)
1089       goto no_ports;
1090
1091     tmp_rtp = addr->port;
1092
1093     g_clear_object (&inetaddr);
1094     inetaddr = g_inet_address_new_from_string (addr->address);
1095   } else {
1096     if (tmp_rtp != 0) {
1097       tmp_rtp += 2;
1098       if (++count > 20)
1099         goto no_ports;
1100     }
1101
1102     if (inetaddr == NULL)
1103       inetaddr = g_inet_address_new_any (family);
1104   }
1105
1106   rtp_sockaddr = g_inet_socket_address_new (inetaddr, tmp_rtp);
1107   if (!g_socket_bind (rtp_socket, rtp_sockaddr, FALSE, NULL)) {
1108     g_object_unref (rtp_sockaddr);
1109     goto again;
1110   }
1111   g_object_unref (rtp_sockaddr);
1112
1113   rtp_sockaddr = g_socket_get_local_address (rtp_socket, NULL);
1114   if (rtp_sockaddr == NULL || !G_IS_INET_SOCKET_ADDRESS (rtp_sockaddr)) {
1115     g_clear_object (&rtp_sockaddr);
1116     goto socket_error;
1117   }
1118
1119   tmp_rtp =
1120       g_inet_socket_address_get_port (G_INET_SOCKET_ADDRESS (rtp_sockaddr));
1121   g_object_unref (rtp_sockaddr);
1122
1123   /* check if port is even */
1124   if ((tmp_rtp & 1) != 0) {
1125     /* port not even, close and allocate another */
1126     tmp_rtp++;
1127     g_clear_object (&rtp_socket);
1128     goto again;
1129   }
1130
1131   /* set port */
1132   tmp_rtcp = tmp_rtp + 1;
1133
1134   rtcp_sockaddr = g_inet_socket_address_new (inetaddr, tmp_rtcp);
1135   if (!g_socket_bind (rtcp_socket, rtcp_sockaddr, FALSE, NULL)) {
1136     g_object_unref (rtcp_sockaddr);
1137     g_clear_object (&rtp_socket);
1138     goto again;
1139   }
1140   g_object_unref (rtcp_sockaddr);
1141
1142   g_clear_object (&inetaddr);
1143
1144   udpsrc0 = gst_element_factory_make ("udpsrc", NULL);
1145   udpsrc1 = gst_element_factory_make ("udpsrc", NULL);
1146
1147   if (udpsrc0 == NULL || udpsrc1 == NULL)
1148     goto no_udp_protocol;
1149
1150   g_object_set (G_OBJECT (udpsrc0), "socket", rtp_socket, NULL);
1151   g_object_set (G_OBJECT (udpsrc1), "socket", rtcp_socket, NULL);
1152
1153   ret = gst_element_set_state (udpsrc0, GST_STATE_READY);
1154   if (ret == GST_STATE_CHANGE_FAILURE)
1155     goto element_error;
1156   ret = gst_element_set_state (udpsrc1, GST_STATE_READY);
1157   if (ret == GST_STATE_CHANGE_FAILURE)
1158     goto element_error;
1159
1160   /* all fine, do port check */
1161   g_object_get (G_OBJECT (udpsrc0), "port", &rtpport, NULL);
1162   g_object_get (G_OBJECT (udpsrc1), "port", &rtcpport, NULL);
1163
1164   /* this should not happen... */
1165   if (rtpport != tmp_rtp || rtcpport != tmp_rtcp)
1166     goto port_error;
1167
1168   if (udpsink_out[0])
1169     udpsink0 = udpsink_out[0];
1170   else
1171     udpsink0 = gst_element_factory_make ("multiudpsink", NULL);
1172
1173   if (!udpsink0)
1174     goto no_udp_protocol;
1175
1176   g_object_set (G_OBJECT (udpsink0), "close-socket", FALSE, NULL);
1177   g_object_set (G_OBJECT (udpsink0), multisink_socket, rtp_socket, NULL);
1178
1179   if (udpsink_out[1])
1180     udpsink1 = udpsink_out[1];
1181   else
1182     udpsink1 = gst_element_factory_make ("multiudpsink", NULL);
1183
1184   if (!udpsink1)
1185     goto no_udp_protocol;
1186
1187   g_object_set (G_OBJECT (udpsink0), "send-duplicates", FALSE, NULL);
1188   g_object_set (G_OBJECT (udpsink1), "send-duplicates", FALSE, NULL);
1189   g_object_set (G_OBJECT (udpsink0), "buffer-size", buffer_size, NULL);
1190
1191   g_object_set (G_OBJECT (udpsink1), "close-socket", FALSE, NULL);
1192   g_object_set (G_OBJECT (udpsink1), multisink_socket, rtcp_socket, NULL);
1193   g_object_set (G_OBJECT (udpsink1), "sync", FALSE, NULL);
1194   /* Needs to be async for RECORD streams, otherwise we will never go to
1195    * PLAYING because the sinks will wait for data while the udpsrc can't
1196    * provide data with timestamps in PAUSED. */
1197   if (priv->sinkpad)
1198     g_object_set (G_OBJECT (udpsink0), "async", FALSE, NULL);
1199   g_object_set (G_OBJECT (udpsink1), "async", FALSE, NULL);
1200   g_object_set (G_OBJECT (udpsink0), "auto-multicast", FALSE, NULL);
1201   g_object_set (G_OBJECT (udpsink0), "loop", FALSE, NULL);
1202   g_object_set (G_OBJECT (udpsink1), "auto-multicast", FALSE, NULL);
1203   g_object_set (G_OBJECT (udpsink1), "loop", FALSE, NULL);
1204
1205   /* we keep these elements, we will further configure them when the
1206    * client told us to really use the UDP ports. */
1207   udpsrc_out[0] = udpsrc0;
1208   udpsrc_out[1] = udpsrc1;
1209   udpsink_out[0] = udpsink0;
1210   udpsink_out[1] = udpsink1;
1211
1212   server_port_out->min = rtpport;
1213   server_port_out->max = rtcpport;
1214
1215   *server_addr_out = addr;
1216   g_list_free_full (rejected_addresses, (GDestroyNotify) gst_rtsp_address_free);
1217
1218   g_object_unref (rtp_socket);
1219   g_object_unref (rtcp_socket);
1220
1221   return TRUE;
1222
1223   /* ERRORS */
1224 no_udp_protocol:
1225   {
1226     goto cleanup;
1227   }
1228 no_ports:
1229   {
1230     goto cleanup;
1231   }
1232 port_error:
1233   {
1234     goto cleanup;
1235   }
1236 socket_error:
1237   {
1238     goto cleanup;
1239   }
1240 element_error:
1241   {
1242     goto cleanup;
1243   }
1244 cleanup:
1245   {
1246     if (udpsrc0) {
1247       gst_element_set_state (udpsrc0, GST_STATE_NULL);
1248       gst_object_unref (udpsrc0);
1249     }
1250     if (udpsrc1) {
1251       gst_element_set_state (udpsrc1, GST_STATE_NULL);
1252       gst_object_unref (udpsrc1);
1253     }
1254     if (udpsink0) {
1255       gst_element_set_state (udpsink0, GST_STATE_NULL);
1256       gst_object_unref (udpsink0);
1257     }
1258     if (inetaddr)
1259       g_object_unref (inetaddr);
1260     g_list_free_full (rejected_addresses,
1261         (GDestroyNotify) gst_rtsp_address_free);
1262     if (addr)
1263       gst_rtsp_address_free (addr);
1264     if (rtp_socket)
1265       g_object_unref (rtp_socket);
1266     if (rtcp_socket)
1267       g_object_unref (rtcp_socket);
1268     return FALSE;
1269   }
1270 }
1271
1272 /* must be called with lock */
1273 static gboolean
1274 alloc_ports (GstRTSPStream * stream)
1275 {
1276   GstRTSPStreamPrivate *priv = stream->priv;
1277
1278   priv->have_ipv4 =
1279       alloc_ports_one_family (stream, priv->pool, priv->buffer_size,
1280       G_SOCKET_FAMILY_IPV4, priv->udpsrc_v4, priv->udpsink,
1281       &priv->server_port_v4, &priv->server_addr_v4);
1282
1283   priv->have_ipv6 =
1284       alloc_ports_one_family (stream, priv->pool, priv->buffer_size,
1285       G_SOCKET_FAMILY_IPV6, priv->udpsrc_v6, priv->udpsink,
1286       &priv->server_port_v6, &priv->server_addr_v6);
1287
1288   return priv->have_ipv4 || priv->have_ipv6;
1289 }
1290
1291 /**
1292  * gst_rtsp_stream_get_server_port:
1293  * @stream: a #GstRTSPStream
1294  * @server_port: (out): result server port
1295  * @family: the port family to get
1296  *
1297  * Fill @server_port with the port pair used by the server. This function can
1298  * only be called when @stream has been joined.
1299  */
1300 void
1301 gst_rtsp_stream_get_server_port (GstRTSPStream * stream,
1302     GstRTSPRange * server_port, GSocketFamily family)
1303 {
1304   GstRTSPStreamPrivate *priv;
1305
1306   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
1307   priv = stream->priv;
1308   g_return_if_fail (priv->is_joined);
1309
1310   g_mutex_lock (&priv->lock);
1311   if (family == G_SOCKET_FAMILY_IPV4) {
1312     if (server_port)
1313       *server_port = priv->server_port_v4;
1314   } else {
1315     if (server_port)
1316       *server_port = priv->server_port_v6;
1317   }
1318   g_mutex_unlock (&priv->lock);
1319 }
1320
1321 /**
1322  * gst_rtsp_stream_get_rtpsession:
1323  * @stream: a #GstRTSPStream
1324  *
1325  * Get the RTP session of this stream.
1326  *
1327  * Returns: (transfer full): The RTP session of this stream. Unref after usage.
1328  */
1329 GObject *
1330 gst_rtsp_stream_get_rtpsession (GstRTSPStream * stream)
1331 {
1332   GstRTSPStreamPrivate *priv;
1333   GObject *session;
1334
1335   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
1336
1337   priv = stream->priv;
1338
1339   g_mutex_lock (&priv->lock);
1340   if ((session = priv->session))
1341     g_object_ref (session);
1342   g_mutex_unlock (&priv->lock);
1343
1344   return session;
1345 }
1346
1347 /**
1348  * gst_rtsp_stream_get_ssrc:
1349  * @stream: a #GstRTSPStream
1350  * @ssrc: (out): result ssrc
1351  *
1352  * Get the SSRC used by the RTP session of this stream. This function can only
1353  * be called when @stream has been joined.
1354  */
1355 void
1356 gst_rtsp_stream_get_ssrc (GstRTSPStream * stream, guint * ssrc)
1357 {
1358   GstRTSPStreamPrivate *priv;
1359
1360   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
1361   priv = stream->priv;
1362   g_return_if_fail (priv->is_joined);
1363
1364   g_mutex_lock (&priv->lock);
1365   if (ssrc && priv->session)
1366     g_object_get (priv->session, "internal-ssrc", ssrc, NULL);
1367   g_mutex_unlock (&priv->lock);
1368 }
1369
1370 /**
1371  * gst_rtsp_stream_set_retransmission_time:
1372  * @stream: a #GstRTSPStream
1373  * @time: a #GstClockTime
1374  *
1375  * Set the amount of time to store retransmission packets.
1376  */
1377 void
1378 gst_rtsp_stream_set_retransmission_time (GstRTSPStream * stream,
1379     GstClockTime time)
1380 {
1381   GST_DEBUG_OBJECT (stream, "set retransmission time %" G_GUINT64_FORMAT, time);
1382
1383   g_mutex_lock (&stream->priv->lock);
1384   stream->priv->rtx_time = time;
1385   if (stream->priv->rtxsend)
1386     g_object_set (stream->priv->rtxsend, "max-size-time",
1387         GST_TIME_AS_MSECONDS (time), NULL);
1388   g_mutex_unlock (&stream->priv->lock);
1389 }
1390
1391 /**
1392  * gst_rtsp_stream_get_retransmission_time:
1393  * @stream: a #GstRTSPStream
1394  *
1395  * Get the amount of time to store retransmission data.
1396  *
1397  * Returns: the amount of time to store retransmission data.
1398  */
1399 GstClockTime
1400 gst_rtsp_stream_get_retransmission_time (GstRTSPStream * stream)
1401 {
1402   GstClockTime ret;
1403
1404   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), 0);
1405
1406   g_mutex_lock (&stream->priv->lock);
1407   ret = stream->priv->rtx_time;
1408   g_mutex_unlock (&stream->priv->lock);
1409
1410   return ret;
1411 }
1412
1413 /**
1414  * gst_rtsp_stream_set_retransmission_pt:
1415  * @stream: a #GstRTSPStream
1416  * @rtx_pt: a #guint
1417  *
1418  * Set the payload type (pt) for retransmission of this stream.
1419  */
1420 void
1421 gst_rtsp_stream_set_retransmission_pt (GstRTSPStream * stream, guint rtx_pt)
1422 {
1423   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
1424
1425   GST_DEBUG_OBJECT (stream, "set retransmission pt %u", rtx_pt);
1426
1427   g_mutex_lock (&stream->priv->lock);
1428   stream->priv->rtx_pt = rtx_pt;
1429   if (stream->priv->rtxsend) {
1430     guint pt = gst_rtsp_stream_get_pt (stream);
1431     gchar *pt_s = g_strdup_printf ("%d", pt);
1432     GstStructure *rtx_pt_map = gst_structure_new ("application/x-rtp-pt-map",
1433         pt_s, G_TYPE_UINT, rtx_pt, NULL);
1434     g_object_set (stream->priv->rtxsend, "payload-type-map", rtx_pt_map, NULL);
1435     g_free (pt_s);
1436     gst_structure_free (rtx_pt_map);
1437   }
1438   g_mutex_unlock (&stream->priv->lock);
1439 }
1440
1441 /**
1442  * gst_rtsp_stream_get_retransmission_pt:
1443  * @stream: a #GstRTSPStream
1444  *
1445  * Get the payload-type used for retransmission of this stream
1446  *
1447  * Returns: The retransmission PT.
1448  */
1449 guint
1450 gst_rtsp_stream_get_retransmission_pt (GstRTSPStream * stream)
1451 {
1452   guint rtx_pt;
1453
1454   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), 0);
1455
1456   g_mutex_lock (&stream->priv->lock);
1457   rtx_pt = stream->priv->rtx_pt;
1458   g_mutex_unlock (&stream->priv->lock);
1459
1460   return rtx_pt;
1461 }
1462
1463 /**
1464  * gst_rtsp_stream_set_buffer_size:
1465  * @stream: a #GstRTSPStream
1466  * @size: the buffer size
1467  *
1468  * Set the size of the UDP transmission buffer (in bytes)
1469  * Needs to be set before the stream is joined to a bin.
1470  *
1471  * Since: 1.6
1472  */
1473 void
1474 gst_rtsp_stream_set_buffer_size (GstRTSPStream * stream, guint size)
1475 {
1476   g_mutex_lock (&stream->priv->lock);
1477   stream->priv->buffer_size = size;
1478   g_mutex_unlock (&stream->priv->lock);
1479 }
1480
1481 /**
1482  * gst_rtsp_stream_get_buffer_size:
1483  * @stream: a #GstRTSPStream
1484  *
1485  * Get the size of the UDP transmission buffer (in bytes)
1486  *
1487  * Returns: the size of the UDP TX buffer
1488  *
1489  * Since: 1.6
1490  */
1491 guint
1492 gst_rtsp_stream_get_buffer_size (GstRTSPStream * stream)
1493 {
1494   guint buffer_size;
1495
1496   g_mutex_lock (&stream->priv->lock);
1497   buffer_size = stream->priv->buffer_size;
1498   g_mutex_unlock (&stream->priv->lock);
1499
1500   return buffer_size;
1501 }
1502
1503 /* executed from streaming thread */
1504 static void
1505 caps_notify (GstPad * pad, GParamSpec * unused, GstRTSPStream * stream)
1506 {
1507   GstRTSPStreamPrivate *priv = stream->priv;
1508   GstCaps *newcaps, *oldcaps;
1509
1510   newcaps = gst_pad_get_current_caps (pad);
1511
1512   GST_INFO ("stream %p received caps %p, %" GST_PTR_FORMAT, stream, newcaps,
1513       newcaps);
1514
1515   g_mutex_lock (&priv->lock);
1516   oldcaps = priv->caps;
1517   priv->caps = newcaps;
1518   g_mutex_unlock (&priv->lock);
1519
1520   if (oldcaps)
1521     gst_caps_unref (oldcaps);
1522 }
1523
1524 static void
1525 dump_structure (const GstStructure * s)
1526 {
1527   gchar *sstr;
1528
1529   sstr = gst_structure_to_string (s);
1530   GST_INFO ("structure: %s", sstr);
1531   g_free (sstr);
1532 }
1533
1534 static GstRTSPStreamTransport *
1535 find_transport (GstRTSPStream * stream, const gchar * rtcp_from)
1536 {
1537   GstRTSPStreamPrivate *priv = stream->priv;
1538   GList *walk;
1539   GstRTSPStreamTransport *result = NULL;
1540   const gchar *tmp;
1541   gchar *dest;
1542   guint port;
1543
1544   if (rtcp_from == NULL)
1545     return NULL;
1546
1547   tmp = g_strrstr (rtcp_from, ":");
1548   if (tmp == NULL)
1549     return NULL;
1550
1551   port = atoi (tmp + 1);
1552   dest = g_strndup (rtcp_from, tmp - rtcp_from);
1553
1554   g_mutex_lock (&priv->lock);
1555   GST_INFO ("finding %s:%d in %d transports", dest, port,
1556       g_list_length (priv->transports));
1557
1558   for (walk = priv->transports; walk; walk = g_list_next (walk)) {
1559     GstRTSPStreamTransport *trans = walk->data;
1560     const GstRTSPTransport *tr;
1561     gint min, max;
1562
1563     tr = gst_rtsp_stream_transport_get_transport (trans);
1564
1565     min = tr->client_port.min;
1566     max = tr->client_port.max;
1567
1568     if ((strcmp (tr->destination, dest) == 0) && (min == port || max == port)) {
1569       result = trans;
1570       break;
1571     }
1572   }
1573   if (result)
1574     g_object_ref (result);
1575   g_mutex_unlock (&priv->lock);
1576
1577   g_free (dest);
1578
1579   return result;
1580 }
1581
1582 static GstRTSPStreamTransport *
1583 check_transport (GObject * source, GstRTSPStream * stream)
1584 {
1585   GstStructure *stats;
1586   GstRTSPStreamTransport *trans;
1587
1588   /* see if we have a stream to match with the origin of the RTCP packet */
1589   trans = g_object_get_qdata (source, ssrc_stream_map_key);
1590   if (trans == NULL) {
1591     g_object_get (source, "stats", &stats, NULL);
1592     if (stats) {
1593       const gchar *rtcp_from;
1594
1595       dump_structure (stats);
1596
1597       rtcp_from = gst_structure_get_string (stats, "rtcp-from");
1598       if ((trans = find_transport (stream, rtcp_from))) {
1599         GST_INFO ("%p: found transport %p for source  %p", stream, trans,
1600             source);
1601         g_object_set_qdata_full (source, ssrc_stream_map_key, trans,
1602             g_object_unref);
1603       }
1604       gst_structure_free (stats);
1605     }
1606   }
1607   return trans;
1608 }
1609
1610
1611 static void
1612 on_new_ssrc (GObject * session, GObject * source, GstRTSPStream * stream)
1613 {
1614   GstRTSPStreamTransport *trans;
1615
1616   GST_INFO ("%p: new source %p", stream, source);
1617
1618   trans = check_transport (source, stream);
1619
1620   if (trans)
1621     GST_INFO ("%p: source %p for transport %p", stream, source, trans);
1622 }
1623
1624 static void
1625 on_ssrc_sdes (GObject * session, GObject * source, GstRTSPStream * stream)
1626 {
1627   GST_INFO ("%p: new SDES %p", stream, source);
1628 }
1629
1630 static void
1631 on_ssrc_active (GObject * session, GObject * source, GstRTSPStream * stream)
1632 {
1633   GstRTSPStreamTransport *trans;
1634
1635   trans = check_transport (source, stream);
1636
1637   if (trans) {
1638     GST_INFO ("%p: source %p in transport %p is active", stream, source, trans);
1639     gst_rtsp_stream_transport_keep_alive (trans);
1640   }
1641 #ifdef DUMP_STATS
1642   {
1643     GstStructure *stats;
1644     g_object_get (source, "stats", &stats, NULL);
1645     if (stats) {
1646       dump_structure (stats);
1647       gst_structure_free (stats);
1648     }
1649   }
1650 #endif
1651 }
1652
1653 static void
1654 on_bye_ssrc (GObject * session, GObject * source, GstRTSPStream * stream)
1655 {
1656   GST_INFO ("%p: source %p bye", stream, source);
1657 }
1658
1659 static void
1660 on_bye_timeout (GObject * session, GObject * source, GstRTSPStream * stream)
1661 {
1662   GstRTSPStreamTransport *trans;
1663
1664   GST_INFO ("%p: source %p bye timeout", stream, source);
1665
1666   if ((trans = g_object_get_qdata (source, ssrc_stream_map_key))) {
1667     gst_rtsp_stream_transport_set_timed_out (trans, TRUE);
1668     g_object_set_qdata (source, ssrc_stream_map_key, NULL);
1669   }
1670 }
1671
1672 static void
1673 on_timeout (GObject * session, GObject * source, GstRTSPStream * stream)
1674 {
1675   GstRTSPStreamTransport *trans;
1676
1677   GST_INFO ("%p: source %p timeout", stream, source);
1678
1679   if ((trans = g_object_get_qdata (source, ssrc_stream_map_key))) {
1680     gst_rtsp_stream_transport_set_timed_out (trans, TRUE);
1681     g_object_set_qdata (source, ssrc_stream_map_key, NULL);
1682   }
1683 }
1684
1685 static void
1686 clear_tr_cache (GstRTSPStreamPrivate * priv, gboolean is_rtp)
1687 {
1688   if (is_rtp) {
1689     g_list_foreach (priv->tr_cache_rtp, (GFunc) g_object_unref, NULL);
1690     g_list_free (priv->tr_cache_rtp);
1691     priv->tr_cache_rtp = NULL;
1692   } else {
1693     g_list_foreach (priv->tr_cache_rtcp, (GFunc) g_object_unref, NULL);
1694     g_list_free (priv->tr_cache_rtcp);
1695     priv->tr_cache_rtcp = NULL;
1696   }
1697 }
1698
1699 static GstFlowReturn
1700 handle_new_sample (GstAppSink * sink, gpointer user_data)
1701 {
1702   GstRTSPStreamPrivate *priv;
1703   GList *walk;
1704   GstSample *sample;
1705   GstBuffer *buffer;
1706   GstRTSPStream *stream;
1707   gboolean is_rtp;
1708
1709   sample = gst_app_sink_pull_sample (sink);
1710   if (!sample)
1711     return GST_FLOW_OK;
1712
1713   stream = (GstRTSPStream *) user_data;
1714   priv = stream->priv;
1715   buffer = gst_sample_get_buffer (sample);
1716
1717   is_rtp = GST_ELEMENT_CAST (sink) == priv->appsink[0];
1718
1719   g_mutex_lock (&priv->lock);
1720   if (is_rtp) {
1721     if (priv->tr_cache_cookie_rtp != priv->transports_cookie) {
1722       clear_tr_cache (priv, is_rtp);
1723       for (walk = priv->transports; walk; walk = g_list_next (walk)) {
1724         GstRTSPStreamTransport *tr = (GstRTSPStreamTransport *) walk->data;
1725         priv->tr_cache_rtp =
1726             g_list_prepend (priv->tr_cache_rtp, g_object_ref (tr));
1727       }
1728       priv->tr_cache_cookie_rtp = priv->transports_cookie;
1729     }
1730   } else {
1731     if (priv->tr_cache_cookie_rtcp != priv->transports_cookie) {
1732       clear_tr_cache (priv, is_rtp);
1733       for (walk = priv->transports; walk; walk = g_list_next (walk)) {
1734         GstRTSPStreamTransport *tr = (GstRTSPStreamTransport *) walk->data;
1735         priv->tr_cache_rtcp =
1736             g_list_prepend (priv->tr_cache_rtcp, g_object_ref (tr));
1737       }
1738       priv->tr_cache_cookie_rtcp = priv->transports_cookie;
1739     }
1740   }
1741   g_mutex_unlock (&priv->lock);
1742
1743   if (is_rtp) {
1744     for (walk = priv->tr_cache_rtp; walk; walk = g_list_next (walk)) {
1745       GstRTSPStreamTransport *tr = (GstRTSPStreamTransport *) walk->data;
1746       gst_rtsp_stream_transport_send_rtp (tr, buffer);
1747     }
1748   } else {
1749     for (walk = priv->tr_cache_rtcp; walk; walk = g_list_next (walk)) {
1750       GstRTSPStreamTransport *tr = (GstRTSPStreamTransport *) walk->data;
1751       gst_rtsp_stream_transport_send_rtcp (tr, buffer);
1752     }
1753   }
1754   gst_sample_unref (sample);
1755
1756   return GST_FLOW_OK;
1757 }
1758
1759 static GstAppSinkCallbacks sink_cb = {
1760   NULL,                         /* not interested in EOS */
1761   NULL,                         /* not interested in preroll samples */
1762   handle_new_sample,
1763 };
1764
1765 static GstElement *
1766 get_rtp_encoder (GstRTSPStream * stream, guint session)
1767 {
1768   GstRTSPStreamPrivate *priv = stream->priv;
1769
1770   if (priv->srtpenc == NULL) {
1771     gchar *name;
1772
1773     name = g_strdup_printf ("srtpenc_%u", session);
1774     priv->srtpenc = gst_element_factory_make ("srtpenc", name);
1775     g_free (name);
1776
1777     g_object_set (priv->srtpenc, "random-key", TRUE, NULL);
1778   }
1779   return gst_object_ref (priv->srtpenc);
1780 }
1781
1782 static GstElement *
1783 request_rtp_encoder (GstElement * rtpbin, guint session, GstRTSPStream * stream)
1784 {
1785   GstRTSPStreamPrivate *priv = stream->priv;
1786   GstElement *oldenc, *enc;
1787   GstPad *pad;
1788   gchar *name;
1789
1790   if (priv->idx != session)
1791     return NULL;
1792
1793   GST_DEBUG_OBJECT (stream, "make RTP encoder for session %u", session);
1794
1795   oldenc = priv->srtpenc;
1796   enc = get_rtp_encoder (stream, session);
1797   name = g_strdup_printf ("rtp_sink_%d", session);
1798   pad = gst_element_get_request_pad (enc, name);
1799   g_free (name);
1800   gst_object_unref (pad);
1801
1802   if (oldenc == NULL)
1803     g_signal_emit (stream, gst_rtsp_stream_signals[SIGNAL_NEW_RTP_ENCODER], 0,
1804         enc);
1805
1806   return enc;
1807 }
1808
1809 static GstElement *
1810 request_rtcp_encoder (GstElement * rtpbin, guint session,
1811     GstRTSPStream * stream)
1812 {
1813   GstRTSPStreamPrivate *priv = stream->priv;
1814   GstElement *oldenc, *enc;
1815   GstPad *pad;
1816   gchar *name;
1817
1818   if (priv->idx != session)
1819     return NULL;
1820
1821   GST_DEBUG_OBJECT (stream, "make RTCP encoder for session %u", session);
1822
1823   oldenc = priv->srtpenc;
1824   enc = get_rtp_encoder (stream, session);
1825   name = g_strdup_printf ("rtcp_sink_%d", session);
1826   pad = gst_element_get_request_pad (enc, name);
1827   g_free (name);
1828   gst_object_unref (pad);
1829
1830   if (oldenc == NULL)
1831     g_signal_emit (stream, gst_rtsp_stream_signals[SIGNAL_NEW_RTCP_ENCODER], 0,
1832         enc);
1833
1834   return enc;
1835 }
1836
1837 static GstCaps *
1838 request_key (GstElement * srtpdec, guint ssrc, GstRTSPStream * stream)
1839 {
1840   GstRTSPStreamPrivate *priv = stream->priv;
1841   GstCaps *caps;
1842
1843   GST_DEBUG ("request key %08x", ssrc);
1844
1845   g_mutex_lock (&priv->lock);
1846   if ((caps = g_hash_table_lookup (priv->keys, GINT_TO_POINTER (ssrc))))
1847     gst_caps_ref (caps);
1848   g_mutex_unlock (&priv->lock);
1849
1850   return caps;
1851 }
1852
1853 static GstElement *
1854 request_rtp_rtcp_decoder (GstElement * rtpbin, guint session,
1855     GstRTSPStream * stream)
1856 {
1857   GstRTSPStreamPrivate *priv = stream->priv;
1858
1859   if (priv->idx != session)
1860     return NULL;
1861
1862   if (priv->srtpdec == NULL) {
1863     gchar *name;
1864
1865     name = g_strdup_printf ("srtpdec_%u", session);
1866     priv->srtpdec = gst_element_factory_make ("srtpdec", name);
1867     g_free (name);
1868
1869     g_signal_connect (priv->srtpdec, "request-key",
1870         (GCallback) request_key, stream);
1871   }
1872   return gst_object_ref (priv->srtpdec);
1873 }
1874
1875 /**
1876  * gst_rtsp_stream_request_aux_sender:
1877  * @stream: a #GstRTSPStream
1878  * @sessid: the session id
1879  *
1880  * Creating a rtxsend bin
1881  *
1882  * Returns: (transfer full): a #GstElement.
1883  *
1884  * Since: 1.6
1885  */
1886 GstElement *
1887 gst_rtsp_stream_request_aux_sender (GstRTSPStream * stream, guint sessid)
1888 {
1889   GstElement *bin;
1890   GstPad *pad;
1891   GstStructure *pt_map;
1892   gchar *name;
1893   guint pt, rtx_pt;
1894   gchar *pt_s;
1895
1896   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
1897
1898   pt = gst_rtsp_stream_get_pt (stream);
1899   pt_s = g_strdup_printf ("%u", pt);
1900   rtx_pt = stream->priv->rtx_pt;
1901
1902   GST_INFO ("creating rtxsend with pt %u to %u", pt, rtx_pt);
1903
1904   bin = gst_bin_new (NULL);
1905   stream->priv->rtxsend = gst_element_factory_make ("rtprtxsend", NULL);
1906   pt_map = gst_structure_new ("application/x-rtp-pt-map",
1907       pt_s, G_TYPE_UINT, rtx_pt, NULL);
1908   g_object_set (stream->priv->rtxsend, "payload-type-map", pt_map,
1909       "max-size-time", GST_TIME_AS_MSECONDS (stream->priv->rtx_time), NULL);
1910   g_free (pt_s);
1911   gst_structure_free (pt_map);
1912   gst_bin_add (GST_BIN (bin), gst_object_ref (stream->priv->rtxsend));
1913
1914   pad = gst_element_get_static_pad (stream->priv->rtxsend, "src");
1915   name = g_strdup_printf ("src_%u", sessid);
1916   gst_element_add_pad (bin, gst_ghost_pad_new (name, pad));
1917   g_free (name);
1918   gst_object_unref (pad);
1919
1920   pad = gst_element_get_static_pad (stream->priv->rtxsend, "sink");
1921   name = g_strdup_printf ("sink_%u", sessid);
1922   gst_element_add_pad (bin, gst_ghost_pad_new (name, pad));
1923   g_free (name);
1924   gst_object_unref (pad);
1925
1926   return bin;
1927 }
1928
1929 /**
1930  * gst_rtsp_stream_set_pt_map:
1931  * @stream: a #GstRTSPStream
1932  * @pt: the pt
1933  * @caps: a #GstCaps
1934  *
1935  * Configure a pt map between @pt and @caps.
1936  */
1937 void
1938 gst_rtsp_stream_set_pt_map (GstRTSPStream * stream, guint pt, GstCaps * caps)
1939 {
1940   GstRTSPStreamPrivate *priv = stream->priv;
1941
1942   g_mutex_lock (&priv->lock);
1943   g_hash_table_insert (priv->ptmap, GINT_TO_POINTER (pt), gst_caps_ref (caps));
1944   g_mutex_unlock (&priv->lock);
1945 }
1946
1947 static GstCaps *
1948 request_pt_map (GstElement * rtpbin, guint session, guint pt,
1949     GstRTSPStream * stream)
1950 {
1951   GstRTSPStreamPrivate *priv = stream->priv;
1952   GstCaps *caps = NULL;
1953
1954   g_mutex_lock (&priv->lock);
1955
1956   if (priv->idx == session) {
1957     caps = g_hash_table_lookup (priv->ptmap, GINT_TO_POINTER (pt));
1958     if (caps) {
1959       GST_DEBUG ("Stream %p, pt %u: caps %" GST_PTR_FORMAT, stream, pt, caps);
1960       gst_caps_ref (caps);
1961     } else {
1962       GST_DEBUG ("Stream %p, pt %u: no caps", stream, pt);
1963     }
1964   }
1965
1966   g_mutex_unlock (&priv->lock);
1967
1968   return caps;
1969 }
1970
1971 static void
1972 pad_added (GstElement * rtpbin, GstPad * pad, GstRTSPStream * stream)
1973 {
1974   GstRTSPStreamPrivate *priv = stream->priv;
1975   gchar *name;
1976   GstPadLinkReturn ret;
1977   guint sessid;
1978
1979   GST_DEBUG ("Stream %p added pad %s:%s for pad %s:%s", stream,
1980       GST_DEBUG_PAD_NAME (pad), GST_DEBUG_PAD_NAME (priv->sinkpad));
1981
1982   name = gst_pad_get_name (pad);
1983   if (sscanf (name, "recv_rtp_src_%u", &sessid) != 1) {
1984     g_free (name);
1985     return;
1986   }
1987   g_free (name);
1988
1989   if (priv->idx != sessid)
1990     return;
1991
1992   if (gst_pad_is_linked (priv->sinkpad)) {
1993     GST_WARNING ("Stream %p: Pad %s:%s is linked already", stream,
1994         GST_DEBUG_PAD_NAME (priv->sinkpad));
1995     return;
1996   }
1997
1998   /* link the RTP pad to the session manager, it should not really fail unless
1999    * this is not really an RTP pad */
2000   ret = gst_pad_link (pad, priv->sinkpad);
2001   if (ret != GST_PAD_LINK_OK)
2002     goto link_failed;
2003   priv->recv_rtp_src = gst_object_ref (pad);
2004
2005   return;
2006
2007 /* ERRORS */
2008 link_failed:
2009   {
2010     GST_ERROR ("Stream %p: Failed to link pads %s:%s and %s:%s", stream,
2011         GST_DEBUG_PAD_NAME (pad), GST_DEBUG_PAD_NAME (priv->sinkpad));
2012   }
2013 }
2014
2015 static void
2016 on_npt_stop (GstElement * rtpbin, guint session, guint ssrc,
2017     GstRTSPStream * stream)
2018 {
2019   /* TODO: What to do here other than this? */
2020   GST_DEBUG ("Stream %p: Got EOS", stream);
2021   gst_pad_send_event (stream->priv->sinkpad, gst_event_new_eos ());
2022 }
2023
2024 /**
2025  * gst_rtsp_stream_join_bin:
2026  * @stream: a #GstRTSPStream
2027  * @bin: (transfer none): a #GstBin to join
2028  * @rtpbin: (transfer none): a rtpbin element in @bin
2029  * @state: the target state of the new elements
2030  *
2031  * Join the #GstBin @bin that contains the element @rtpbin.
2032  *
2033  * @stream will link to @rtpbin, which must be inside @bin. The elements
2034  * added to @bin will be set to the state given in @state.
2035  *
2036  * Returns: %TRUE on success.
2037  */
2038 gboolean
2039 gst_rtsp_stream_join_bin (GstRTSPStream * stream, GstBin * bin,
2040     GstElement * rtpbin, GstState state)
2041 {
2042   GstRTSPStreamPrivate *priv;
2043   gint i;
2044   guint idx;
2045   gchar *name;
2046   GstPad *pad, *sinkpad, *selpad;
2047   GstPadLinkReturn ret;
2048
2049   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
2050   g_return_val_if_fail (GST_IS_BIN (bin), FALSE);
2051   g_return_val_if_fail (GST_IS_ELEMENT (rtpbin), FALSE);
2052
2053   priv = stream->priv;
2054
2055   g_mutex_lock (&priv->lock);
2056   if (priv->is_joined)
2057     goto was_joined;
2058
2059   /* create a session with the same index as the stream */
2060   idx = priv->idx;
2061
2062   GST_INFO ("stream %p joining bin as session %u", stream, idx);
2063
2064   if (!alloc_ports (stream))
2065     goto no_ports;
2066
2067   /* update the dscp qos field in the sinks */
2068   update_dscp_qos (stream);
2069
2070   if (priv->profiles & GST_RTSP_PROFILE_SAVP
2071       || priv->profiles & GST_RTSP_PROFILE_SAVPF) {
2072     /* For SRTP */
2073     g_signal_connect (rtpbin, "request-rtp-encoder",
2074         (GCallback) request_rtp_encoder, stream);
2075     g_signal_connect (rtpbin, "request-rtcp-encoder",
2076         (GCallback) request_rtcp_encoder, stream);
2077     g_signal_connect (rtpbin, "request-rtp-decoder",
2078         (GCallback) request_rtp_rtcp_decoder, stream);
2079     g_signal_connect (rtpbin, "request-rtcp-decoder",
2080         (GCallback) request_rtp_rtcp_decoder, stream);
2081   }
2082
2083   if (priv->sinkpad) {
2084     g_signal_connect (rtpbin, "request-pt-map",
2085         (GCallback) request_pt_map, stream);
2086   }
2087
2088   /* get a pad for sending RTP */
2089   name = g_strdup_printf ("send_rtp_sink_%u", idx);
2090   priv->send_rtp_sink = gst_element_get_request_pad (rtpbin, name);
2091   g_free (name);
2092
2093   if (priv->srcpad) {
2094     /* link the RTP pad to the session manager, it should not really fail unless
2095      * this is not really an RTP pad */
2096     ret = gst_pad_link (priv->srcpad, priv->send_rtp_sink);
2097     if (ret != GST_PAD_LINK_OK)
2098       goto link_failed;
2099   } else {
2100     /* Need to connect our sinkpad from here */
2101     g_signal_connect (rtpbin, "pad-added", (GCallback) pad_added, stream);
2102     /* EOS */
2103     g_signal_connect (rtpbin, "on-npt-stop", (GCallback) on_npt_stop, stream);
2104   }
2105
2106   /* get pads from the RTP session element for sending and receiving
2107    * RTP/RTCP*/
2108   name = g_strdup_printf ("send_rtp_src_%u", idx);
2109   priv->send_src[0] = gst_element_get_static_pad (rtpbin, name);
2110   g_free (name);
2111   name = g_strdup_printf ("recv_rtp_sink_%u", idx);
2112   priv->recv_sink[0] = gst_element_get_request_pad (rtpbin, name);
2113   g_free (name);
2114
2115   name = g_strdup_printf ("send_rtcp_src_%u", idx);
2116   priv->send_src[1] = gst_element_get_request_pad (rtpbin, name);
2117   g_free (name);
2118   name = g_strdup_printf ("recv_rtcp_sink_%u", idx);
2119   priv->recv_sink[1] = gst_element_get_request_pad (rtpbin, name);
2120   g_free (name);
2121
2122   /* get the session */
2123   g_signal_emit_by_name (rtpbin, "get-internal-session", idx, &priv->session);
2124
2125   g_signal_connect (priv->session, "on-new-ssrc", (GCallback) on_new_ssrc,
2126       stream);
2127   g_signal_connect (priv->session, "on-ssrc-sdes", (GCallback) on_ssrc_sdes,
2128       stream);
2129   g_signal_connect (priv->session, "on-ssrc-active",
2130       (GCallback) on_ssrc_active, stream);
2131   g_signal_connect (priv->session, "on-bye-ssrc", (GCallback) on_bye_ssrc,
2132       stream);
2133   g_signal_connect (priv->session, "on-bye-timeout",
2134       (GCallback) on_bye_timeout, stream);
2135   g_signal_connect (priv->session, "on-timeout", (GCallback) on_timeout,
2136       stream);
2137
2138   for (i = 0; i < 2; i++) {
2139     GstPad *teepad, *queuepad;
2140     /* For the sender we create this bit of pipeline for both
2141      * RTP and RTCP. Sync and preroll are enabled on udpsink so
2142      * we need to add a queue before appsink and udpsink to make
2143      * the pipeline not block. For the TCP case, we want to pump
2144      * data to the client as fast as possible.
2145      *
2146      * .--------.      .-----.    .---------.    .---------.
2147      * | rtpbin |      | tee |    |  queue  |    | udpsink |
2148      * |       send->sink   src->sink      src->sink       |
2149      * '--------'      |     |    '---------'    '---------'
2150      *                 |     |    .---------.    .---------.
2151      *                 |     |    |  queue  |    | appsink |
2152      *                 |    src->sink      src->sink       |
2153      *                 '-----'    '---------'    '---------'
2154      *
2155      * When only UDP is allowed, we skip the tee, queue and appsink and link the
2156      * udpsink directly to the session.
2157      */
2158     /* add udpsink */
2159     gst_bin_add (bin, priv->udpsink[i]);
2160     sinkpad = gst_element_get_static_pad (priv->udpsink[i], "sink");
2161
2162     if (priv->protocols & GST_RTSP_LOWER_TRANS_TCP) {
2163       /* make tee for RTP/RTCP */
2164       priv->tee[i] = gst_element_factory_make ("tee", NULL);
2165       gst_bin_add (bin, priv->tee[i]);
2166
2167       /* and link to rtpbin send pad */
2168       pad = gst_element_get_static_pad (priv->tee[i], "sink");
2169       gst_pad_link (priv->send_src[i], pad);
2170       gst_object_unref (pad);
2171
2172       priv->udpqueue[i] = gst_element_factory_make ("queue", NULL);
2173       g_object_set (priv->udpqueue[i], "max-size-buffers",
2174           1, "max-size-bytes", 0, "max-size-time", G_GINT64_CONSTANT (0), NULL);
2175       gst_bin_add (bin, priv->udpqueue[i]);
2176       /* link tee to udpqueue */
2177       teepad = gst_element_get_request_pad (priv->tee[i], "src_%u");
2178       pad = gst_element_get_static_pad (priv->udpqueue[i], "sink");
2179       gst_pad_link (teepad, pad);
2180       gst_object_unref (pad);
2181       gst_object_unref (teepad);
2182
2183       /* link udpqueue to udpsink */
2184       queuepad = gst_element_get_static_pad (priv->udpqueue[i], "src");
2185       gst_pad_link (queuepad, sinkpad);
2186       gst_object_unref (queuepad);
2187
2188       /* make queue */
2189       priv->appqueue[i] = gst_element_factory_make ("queue", NULL);
2190       g_object_set (priv->appqueue[i], "max-size-buffers",
2191           1, "max-size-bytes", 0, "max-size-time", G_GINT64_CONSTANT (0), NULL);
2192       gst_bin_add (bin, priv->appqueue[i]);
2193       /* and link to tee */
2194       teepad = gst_element_get_request_pad (priv->tee[i], "src_%u");
2195       pad = gst_element_get_static_pad (priv->appqueue[i], "sink");
2196       gst_pad_link (teepad, pad);
2197       gst_object_unref (pad);
2198       gst_object_unref (teepad);
2199
2200       /* make appsink */
2201       priv->appsink[i] = gst_element_factory_make ("appsink", NULL);
2202       g_object_set (priv->appsink[i], "async", FALSE, "sync", FALSE, NULL);
2203       g_object_set (priv->appsink[i], "emit-signals", FALSE, NULL);
2204       gst_bin_add (bin, priv->appsink[i]);
2205       gst_app_sink_set_callbacks (GST_APP_SINK_CAST (priv->appsink[i]),
2206           &sink_cb, stream, NULL);
2207       /* and link to queue */
2208       queuepad = gst_element_get_static_pad (priv->appqueue[i], "src");
2209       pad = gst_element_get_static_pad (priv->appsink[i], "sink");
2210       gst_pad_link (queuepad, pad);
2211       gst_object_unref (pad);
2212       gst_object_unref (queuepad);
2213     } else {
2214       /* else only udpsink needed, link it to the session */
2215       gst_pad_link (priv->send_src[i], sinkpad);
2216     }
2217     gst_object_unref (sinkpad);
2218
2219     /* For the receiver we create this bit of pipeline for both
2220      * RTP and RTCP. We receive RTP/RTCP on appsrc and udpsrc
2221      * and it is all funneled into the rtpbin receive pad.
2222      *
2223      * .--------.     .--------.    .--------.
2224      * | udpsrc |     | funnel |    | rtpbin |
2225      * |       src->sink      src->sink      |
2226      * '--------'     |        |    '--------'
2227      * .--------.     |        |
2228      * | appsrc |     |        |
2229      * |       src->sink       |
2230      * '--------'     '--------'
2231      */
2232     /* make funnel for the RTP/RTCP receivers */
2233     priv->funnel[i] = gst_element_factory_make ("funnel", NULL);
2234     gst_bin_add (bin, priv->funnel[i]);
2235
2236     pad = gst_element_get_static_pad (priv->funnel[i], "src");
2237     gst_pad_link (pad, priv->recv_sink[i]);
2238     gst_object_unref (pad);
2239
2240     if (priv->udpsrc_v4[i]) {
2241       if (priv->srcpad) {
2242         /* we set and keep these to playing so that they don't cause NO_PREROLL return
2243          * values. This is only relevant for PLAY pipelines */
2244         gst_element_set_state (priv->udpsrc_v4[i], GST_STATE_PLAYING);
2245         gst_element_set_locked_state (priv->udpsrc_v4[i], TRUE);
2246       }
2247       /* add udpsrc */
2248       gst_bin_add (bin, priv->udpsrc_v4[i]);
2249
2250       /* and link to the funnel v4 */
2251       selpad = gst_element_get_request_pad (priv->funnel[i], "sink_%u");
2252       pad = gst_element_get_static_pad (priv->udpsrc_v4[i], "src");
2253       gst_pad_link (pad, selpad);
2254       gst_object_unref (pad);
2255       gst_object_unref (selpad);
2256     }
2257
2258     if (priv->udpsrc_v6[i]) {
2259       if (priv->srcpad) {
2260         gst_element_set_state (priv->udpsrc_v6[i], GST_STATE_PLAYING);
2261         gst_element_set_locked_state (priv->udpsrc_v6[i], TRUE);
2262       }
2263       gst_bin_add (bin, priv->udpsrc_v6[i]);
2264
2265       /* and link to the funnel v6 */
2266       selpad = gst_element_get_request_pad (priv->funnel[i], "sink_%u");
2267       pad = gst_element_get_static_pad (priv->udpsrc_v6[i], "src");
2268       gst_pad_link (pad, selpad);
2269       gst_object_unref (pad);
2270       gst_object_unref (selpad);
2271     }
2272
2273     if (priv->protocols & GST_RTSP_LOWER_TRANS_TCP) {
2274       /* make and add appsrc */
2275       priv->appsrc[i] = gst_element_factory_make ("appsrc", NULL);
2276       priv->appsrc_base_time[i] = -1;
2277       g_object_set (priv->appsrc[i], "format", GST_FORMAT_TIME, NULL);
2278       gst_bin_add (bin, priv->appsrc[i]);
2279       /* and link to the funnel */
2280       selpad = gst_element_get_request_pad (priv->funnel[i], "sink_%u");
2281       pad = gst_element_get_static_pad (priv->appsrc[i], "src");
2282       gst_pad_link (pad, selpad);
2283       gst_object_unref (pad);
2284       gst_object_unref (selpad);
2285     }
2286
2287     /* check if we need to set to a special state */
2288     if (state != GST_STATE_NULL) {
2289       if (priv->udpsink[i])
2290         gst_element_set_state (priv->udpsink[i], state);
2291       if (priv->appsink[i])
2292         gst_element_set_state (priv->appsink[i], state);
2293       if (priv->appqueue[i])
2294         gst_element_set_state (priv->appqueue[i], state);
2295       if (priv->udpqueue[i])
2296         gst_element_set_state (priv->udpqueue[i], state);
2297       if (priv->tee[i])
2298         gst_element_set_state (priv->tee[i], state);
2299       if (priv->funnel[i])
2300         gst_element_set_state (priv->funnel[i], state);
2301       if (priv->appsrc[i])
2302         gst_element_set_state (priv->appsrc[i], state);
2303     }
2304   }
2305
2306   /* be notified of caps changes */
2307   priv->caps_sig = g_signal_connect (priv->send_src[0], "notify::caps",
2308       (GCallback) caps_notify, stream);
2309
2310   priv->is_joined = TRUE;
2311   g_mutex_unlock (&priv->lock);
2312
2313   return TRUE;
2314
2315   /* ERRORS */
2316 was_joined:
2317   {
2318     g_mutex_unlock (&priv->lock);
2319     return TRUE;
2320   }
2321 no_ports:
2322   {
2323     g_mutex_unlock (&priv->lock);
2324     GST_WARNING ("failed to allocate ports %u", idx);
2325     return FALSE;
2326   }
2327 link_failed:
2328   {
2329     GST_WARNING ("failed to link stream %u", idx);
2330     gst_object_unref (priv->send_rtp_sink);
2331     priv->send_rtp_sink = NULL;
2332     g_mutex_unlock (&priv->lock);
2333     return FALSE;
2334   }
2335 }
2336
2337 /**
2338  * gst_rtsp_stream_leave_bin:
2339  * @stream: a #GstRTSPStream
2340  * @bin: (transfer none): a #GstBin
2341  * @rtpbin: (transfer none): a rtpbin #GstElement
2342  *
2343  * Remove the elements of @stream from @bin.
2344  *
2345  * Return: %TRUE on success.
2346  */
2347 gboolean
2348 gst_rtsp_stream_leave_bin (GstRTSPStream * stream, GstBin * bin,
2349     GstElement * rtpbin)
2350 {
2351   GstRTSPStreamPrivate *priv;
2352   gint i;
2353   GList *l;
2354
2355   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
2356   g_return_val_if_fail (GST_IS_BIN (bin), FALSE);
2357   g_return_val_if_fail (GST_IS_ELEMENT (rtpbin), FALSE);
2358
2359   priv = stream->priv;
2360
2361   g_mutex_lock (&priv->lock);
2362   if (!priv->is_joined)
2363     goto was_not_joined;
2364
2365   /* all transports must be removed by now */
2366   if (priv->transports != NULL)
2367     goto transports_not_removed;
2368
2369   clear_tr_cache (priv, TRUE);
2370   clear_tr_cache (priv, FALSE);
2371
2372   GST_INFO ("stream %p leaving bin", stream);
2373
2374   if (priv->srcpad) {
2375     gst_pad_unlink (priv->srcpad, priv->send_rtp_sink);
2376   } else if (priv->recv_rtp_src) {
2377     gst_pad_unlink (priv->recv_rtp_src, priv->sinkpad);
2378     gst_object_unref (priv->recv_rtp_src);
2379     priv->recv_rtp_src = NULL;
2380   }
2381   g_signal_handler_disconnect (priv->send_src[0], priv->caps_sig);
2382   gst_element_release_request_pad (rtpbin, priv->send_rtp_sink);
2383   gst_object_unref (priv->send_rtp_sink);
2384   priv->send_rtp_sink = NULL;
2385
2386   for (i = 0; i < 2; i++) {
2387     if (priv->udpsink[i])
2388       gst_element_set_state (priv->udpsink[i], GST_STATE_NULL);
2389     if (priv->appsink[i])
2390       gst_element_set_state (priv->appsink[i], GST_STATE_NULL);
2391     if (priv->appqueue[i])
2392       gst_element_set_state (priv->appqueue[i], GST_STATE_NULL);
2393     if (priv->udpqueue[i])
2394       gst_element_set_state (priv->udpqueue[i], GST_STATE_NULL);
2395     if (priv->tee[i])
2396       gst_element_set_state (priv->tee[i], GST_STATE_NULL);
2397     if (priv->funnel[i])
2398       gst_element_set_state (priv->funnel[i], GST_STATE_NULL);
2399     if (priv->appsrc[i])
2400       gst_element_set_state (priv->appsrc[i], GST_STATE_NULL);
2401     if (priv->udpsrc_v4[i]) {
2402       /* and set udpsrc to NULL now before removing */
2403       gst_element_set_locked_state (priv->udpsrc_v4[i], FALSE);
2404       gst_element_set_state (priv->udpsrc_v4[i], GST_STATE_NULL);
2405       /* removing them should also nicely release the request
2406        * pads when they finalize */
2407       gst_bin_remove (bin, priv->udpsrc_v4[i]);
2408     }
2409     if (priv->udpsrc_v6[i]) {
2410       gst_element_set_locked_state (priv->udpsrc_v6[i], FALSE);
2411       gst_element_set_state (priv->udpsrc_v6[i], GST_STATE_NULL);
2412       gst_bin_remove (bin, priv->udpsrc_v6[i]);
2413     }
2414
2415     for (l = priv->transport_sources; l; l = l->next) {
2416       GstRTSPMulticastTransportSource *s = l->data;
2417
2418       if (!s->udpsrc[i])
2419         continue;
2420
2421       gst_element_set_locked_state (s->udpsrc[i], FALSE);
2422       gst_element_set_state (s->udpsrc[i], GST_STATE_NULL);
2423       gst_bin_remove (bin, s->udpsrc[i]);
2424     }
2425
2426     if (priv->udpsink[i])
2427       gst_bin_remove (bin, priv->udpsink[i]);
2428     if (priv->appsrc[i])
2429       gst_bin_remove (bin, priv->appsrc[i]);
2430     if (priv->appsink[i])
2431       gst_bin_remove (bin, priv->appsink[i]);
2432     if (priv->appqueue[i])
2433       gst_bin_remove (bin, priv->appqueue[i]);
2434     if (priv->udpqueue[i])
2435       gst_bin_remove (bin, priv->udpqueue[i]);
2436     if (priv->tee[i])
2437       gst_bin_remove (bin, priv->tee[i]);
2438     if (priv->funnel[i])
2439       gst_bin_remove (bin, priv->funnel[i]);
2440
2441     gst_element_release_request_pad (rtpbin, priv->recv_sink[i]);
2442     gst_object_unref (priv->recv_sink[i]);
2443     priv->recv_sink[i] = NULL;
2444
2445     priv->udpsrc_v4[i] = NULL;
2446     priv->udpsrc_v6[i] = NULL;
2447     priv->udpsink[i] = NULL;
2448     priv->appsrc[i] = NULL;
2449     priv->appsink[i] = NULL;
2450     priv->appqueue[i] = NULL;
2451     priv->udpqueue[i] = NULL;
2452     priv->tee[i] = NULL;
2453     priv->funnel[i] = NULL;
2454   }
2455
2456   for (l = priv->transport_sources; l; l = l->next) {
2457     GstRTSPMulticastTransportSource *s = l->data;
2458     g_slice_free (GstRTSPMulticastTransportSource, s);
2459   }
2460   g_list_free (priv->transport_sources);
2461   priv->transport_sources = NULL;
2462
2463   gst_object_unref (priv->send_src[0]);
2464   priv->send_src[0] = NULL;
2465
2466   gst_element_release_request_pad (rtpbin, priv->send_src[1]);
2467   gst_object_unref (priv->send_src[1]);
2468   priv->send_src[1] = NULL;
2469
2470   g_object_unref (priv->session);
2471   priv->session = NULL;
2472   if (priv->caps)
2473     gst_caps_unref (priv->caps);
2474   priv->caps = NULL;
2475
2476   if (priv->srtpenc)
2477     gst_object_unref (priv->srtpenc);
2478   if (priv->srtpdec)
2479     gst_object_unref (priv->srtpdec);
2480
2481   priv->is_joined = FALSE;
2482   g_mutex_unlock (&priv->lock);
2483
2484   return TRUE;
2485
2486 was_not_joined:
2487   {
2488     g_mutex_unlock (&priv->lock);
2489     return TRUE;
2490   }
2491 transports_not_removed:
2492   {
2493     GST_ERROR_OBJECT (stream, "can't leave bin (transports not removed)");
2494     g_mutex_unlock (&priv->lock);
2495     return FALSE;
2496   }
2497 }
2498
2499 /**
2500  * gst_rtsp_stream_get_rtpinfo:
2501  * @stream: a #GstRTSPStream
2502  * @rtptime: (allow-none): result RTP timestamp
2503  * @seq: (allow-none): result RTP seqnum
2504  * @clock_rate: (allow-none): the clock rate
2505  * @running_time: (allow-none): result running-time
2506  *
2507  * Retrieve the current rtptime, seq and running-time. This is used to
2508  * construct a RTPInfo reply header.
2509  *
2510  * Returns: %TRUE when rtptime, seq and running-time could be determined.
2511  */
2512 gboolean
2513 gst_rtsp_stream_get_rtpinfo (GstRTSPStream * stream,
2514     guint * rtptime, guint * seq, guint * clock_rate,
2515     GstClockTime * running_time)
2516 {
2517   GstRTSPStreamPrivate *priv;
2518   GstStructure *stats;
2519   GObjectClass *payobjclass;
2520
2521   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
2522
2523   priv = stream->priv;
2524
2525   payobjclass = G_OBJECT_GET_CLASS (priv->payloader);
2526
2527   g_mutex_lock (&priv->lock);
2528
2529   /* First try to extract the information from the last buffer on the sinks.
2530    * This will have a more accurate sequence number and timestamp, as between
2531    * the payloader and the sink there can be some queues
2532    */
2533   if (priv->udpsink[0] || priv->appsink[0]) {
2534     GstSample *last_sample;
2535
2536     if (priv->udpsink[0])
2537       g_object_get (priv->udpsink[0], "last-sample", &last_sample, NULL);
2538     else
2539       g_object_get (priv->appsink[0], "last-sample", &last_sample, NULL);
2540
2541     if (last_sample) {
2542       GstCaps *caps;
2543       GstBuffer *buffer;
2544       GstSegment *segment;
2545       GstRTPBuffer rtp_buffer = GST_RTP_BUFFER_INIT;
2546
2547       caps = gst_sample_get_caps (last_sample);
2548       buffer = gst_sample_get_buffer (last_sample);
2549       segment = gst_sample_get_segment (last_sample);
2550
2551       if (gst_rtp_buffer_map (buffer, GST_MAP_READ, &rtp_buffer)) {
2552         if (seq) {
2553           *seq = gst_rtp_buffer_get_seq (&rtp_buffer);
2554         }
2555
2556         if (rtptime) {
2557           *rtptime = gst_rtp_buffer_get_timestamp (&rtp_buffer);
2558         }
2559
2560         gst_rtp_buffer_unmap (&rtp_buffer);
2561
2562         if (running_time) {
2563           *running_time =
2564               gst_segment_to_running_time (segment, GST_FORMAT_TIME,
2565               GST_BUFFER_TIMESTAMP (buffer));
2566         }
2567
2568         if (clock_rate) {
2569           GstStructure *s = gst_caps_get_structure (caps, 0);
2570
2571           gst_structure_get_int (s, "clock-rate", (gint *) clock_rate);
2572
2573           if (*clock_rate == 0 && running_time)
2574             *running_time = GST_CLOCK_TIME_NONE;
2575         }
2576         gst_sample_unref (last_sample);
2577
2578         goto done;
2579       } else {
2580         gst_sample_unref (last_sample);
2581       }
2582     }
2583   }
2584
2585   if (g_object_class_find_property (payobjclass, "stats")) {
2586     g_object_get (priv->payloader, "stats", &stats, NULL);
2587     if (stats == NULL)
2588       goto no_stats;
2589
2590     if (seq)
2591       gst_structure_get_uint (stats, "seqnum", seq);
2592
2593     if (rtptime)
2594       gst_structure_get_uint (stats, "timestamp", rtptime);
2595
2596     if (running_time)
2597       gst_structure_get_clock_time (stats, "running-time", running_time);
2598
2599     if (clock_rate) {
2600       gst_structure_get_uint (stats, "clock-rate", clock_rate);
2601       if (*clock_rate == 0 && running_time)
2602         *running_time = GST_CLOCK_TIME_NONE;
2603     }
2604     gst_structure_free (stats);
2605   } else {
2606     if (!g_object_class_find_property (payobjclass, "seqnum") ||
2607         !g_object_class_find_property (payobjclass, "timestamp"))
2608       goto no_stats;
2609
2610     if (seq)
2611       g_object_get (priv->payloader, "seqnum", seq, NULL);
2612
2613     if (rtptime)
2614       g_object_get (priv->payloader, "timestamp", rtptime, NULL);
2615
2616     if (running_time)
2617       *running_time = GST_CLOCK_TIME_NONE;
2618   }
2619
2620 done:
2621   g_mutex_unlock (&priv->lock);
2622
2623   return TRUE;
2624
2625   /* ERRORS */
2626 no_stats:
2627   {
2628     GST_WARNING ("Could not get payloader stats");
2629     g_mutex_unlock (&priv->lock);
2630     return FALSE;
2631   }
2632 }
2633
2634 /**
2635  * gst_rtsp_stream_get_caps:
2636  * @stream: a #GstRTSPStream
2637  *
2638  * Retrieve the current caps of @stream.
2639  *
2640  * Returns: (transfer full): the #GstCaps of @stream. use gst_caps_unref()
2641  * after usage.
2642  */
2643 GstCaps *
2644 gst_rtsp_stream_get_caps (GstRTSPStream * stream)
2645 {
2646   GstRTSPStreamPrivate *priv;
2647   GstCaps *result;
2648
2649   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
2650
2651   priv = stream->priv;
2652
2653   g_mutex_lock (&priv->lock);
2654   if ((result = priv->caps))
2655     gst_caps_ref (result);
2656   g_mutex_unlock (&priv->lock);
2657
2658   return result;
2659 }
2660
2661 /**
2662  * gst_rtsp_stream_recv_rtp:
2663  * @stream: a #GstRTSPStream
2664  * @buffer: (transfer full): a #GstBuffer
2665  *
2666  * Handle an RTP buffer for the stream. This method is usually called when a
2667  * message has been received from a client using the TCP transport.
2668  *
2669  * This function takes ownership of @buffer.
2670  *
2671  * Returns: a GstFlowReturn.
2672  */
2673 GstFlowReturn
2674 gst_rtsp_stream_recv_rtp (GstRTSPStream * stream, GstBuffer * buffer)
2675 {
2676   GstRTSPStreamPrivate *priv;
2677   GstFlowReturn ret;
2678   GstElement *element;
2679
2680   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), GST_FLOW_ERROR);
2681   priv = stream->priv;
2682   g_return_val_if_fail (GST_IS_BUFFER (buffer), GST_FLOW_ERROR);
2683   g_return_val_if_fail (priv->is_joined, FALSE);
2684
2685   g_mutex_lock (&priv->lock);
2686   if (priv->appsrc[0])
2687     element = gst_object_ref (priv->appsrc[0]);
2688   else
2689     element = NULL;
2690   g_mutex_unlock (&priv->lock);
2691
2692   if (element) {
2693     if (priv->appsrc_base_time[0] == -1) {
2694       /* Take current running_time. This timestamp will be put on
2695        * the first buffer of each stream because we are a live source and so we
2696        * timestamp with the running_time. When we are dealing with TCP, we also
2697        * only timestamp the first buffer (using the DISCONT flag) because a server
2698        * typically bursts data, for which we don't want to compensate by speeding
2699        * up the media. The other timestamps will be interpollated from this one
2700        * using the RTP timestamps. */
2701       GST_OBJECT_LOCK (element);
2702       if (GST_ELEMENT_CLOCK (element)) {
2703         GstClockTime now;
2704         GstClockTime base_time;
2705
2706         now = gst_clock_get_time (GST_ELEMENT_CLOCK (element));
2707         base_time = GST_ELEMENT_CAST (element)->base_time;
2708
2709         priv->appsrc_base_time[0] = now - base_time;
2710         GST_BUFFER_TIMESTAMP (buffer) = priv->appsrc_base_time[0];
2711         GST_DEBUG ("stream %p: first buffer at time %" GST_TIME_FORMAT
2712             ", base %" GST_TIME_FORMAT, stream, GST_TIME_ARGS (now),
2713             GST_TIME_ARGS (base_time));
2714       }
2715       GST_OBJECT_UNLOCK (element);
2716     }
2717
2718     ret = gst_app_src_push_buffer (GST_APP_SRC_CAST (element), buffer);
2719     gst_object_unref (element);
2720   } else {
2721     ret = GST_FLOW_OK;
2722   }
2723   return ret;
2724 }
2725
2726 /**
2727  * gst_rtsp_stream_recv_rtcp:
2728  * @stream: a #GstRTSPStream
2729  * @buffer: (transfer full): a #GstBuffer
2730  *
2731  * Handle an RTCP buffer for the stream. This method is usually called when a
2732  * message has been received from a client using the TCP transport.
2733  *
2734  * This function takes ownership of @buffer.
2735  *
2736  * Returns: a GstFlowReturn.
2737  */
2738 GstFlowReturn
2739 gst_rtsp_stream_recv_rtcp (GstRTSPStream * stream, GstBuffer * buffer)
2740 {
2741   GstRTSPStreamPrivate *priv;
2742   GstFlowReturn ret;
2743   GstElement *element;
2744
2745   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), GST_FLOW_ERROR);
2746   priv = stream->priv;
2747   g_return_val_if_fail (GST_IS_BUFFER (buffer), GST_FLOW_ERROR);
2748
2749   if (!priv->is_joined) {
2750     gst_buffer_unref (buffer);
2751     return GST_FLOW_NOT_LINKED;
2752   }
2753   g_mutex_lock (&priv->lock);
2754   if (priv->appsrc[1])
2755     element = gst_object_ref (priv->appsrc[1]);
2756   else
2757     element = NULL;
2758   g_mutex_unlock (&priv->lock);
2759
2760   if (element) {
2761     if (priv->appsrc_base_time[1] == -1) {
2762       /* Take current running_time. This timestamp will be put on
2763        * the first buffer of each stream because we are a live source and so we
2764        * timestamp with the running_time. When we are dealing with TCP, we also
2765        * only timestamp the first buffer (using the DISCONT flag) because a server
2766        * typically bursts data, for which we don't want to compensate by speeding
2767        * up the media. The other timestamps will be interpollated from this one
2768        * using the RTP timestamps. */
2769       GST_OBJECT_LOCK (element);
2770       if (GST_ELEMENT_CLOCK (element)) {
2771         GstClockTime now;
2772         GstClockTime base_time;
2773
2774         now = gst_clock_get_time (GST_ELEMENT_CLOCK (element));
2775         base_time = GST_ELEMENT_CAST (element)->base_time;
2776
2777         priv->appsrc_base_time[1] = now - base_time;
2778         GST_BUFFER_TIMESTAMP (buffer) = priv->appsrc_base_time[1];
2779         GST_DEBUG ("stream %p: first buffer at time %" GST_TIME_FORMAT
2780             ", base %" GST_TIME_FORMAT, stream, GST_TIME_ARGS (now),
2781             GST_TIME_ARGS (base_time));
2782       }
2783       GST_OBJECT_UNLOCK (element);
2784     }
2785
2786     ret = gst_app_src_push_buffer (GST_APP_SRC_CAST (element), buffer);
2787     gst_object_unref (element);
2788   } else {
2789     ret = GST_FLOW_OK;
2790     gst_buffer_unref (buffer);
2791   }
2792   return ret;
2793 }
2794
2795 /* must be called with lock */
2796 static gboolean
2797 update_transport (GstRTSPStream * stream, GstRTSPStreamTransport * trans,
2798     gboolean add)
2799 {
2800   GstRTSPStreamPrivate *priv = stream->priv;
2801   const GstRTSPTransport *tr;
2802
2803   tr = gst_rtsp_stream_transport_get_transport (trans);
2804
2805   switch (tr->lower_transport) {
2806     case GST_RTSP_LOWER_TRANS_UDP_MCAST:
2807     {
2808       GstRTSPMulticastTransportSource *source;
2809       GstBin *bin;
2810
2811       bin = GST_BIN (gst_object_get_parent (GST_OBJECT (priv->funnel[0])));
2812
2813       if (add) {
2814         gchar *host;
2815         gint i;
2816         GstPad *selpad, *pad;
2817
2818         source = g_slice_new0 (GstRTSPMulticastTransportSource);
2819         source->transport = trans;
2820
2821         for (i = 0; i < 2; i++) {
2822           host =
2823               g_strdup_printf ("udp://%s:%d", tr->destination,
2824               (i == 0) ? tr->port.min : tr->port.max);
2825           source->udpsrc[i] =
2826               gst_element_make_from_uri (GST_URI_SRC, host, NULL, NULL);
2827           g_free (host);
2828
2829           if (priv->srcpad) {
2830             /* we set and keep these to playing so that they don't cause NO_PREROLL return
2831              * values. This is only relevant for PLAY pipelines */
2832             gst_element_set_state (source->udpsrc[i], GST_STATE_PLAYING);
2833             gst_element_set_locked_state (source->udpsrc[i], TRUE);
2834           }
2835           /* add udpsrc */
2836           gst_bin_add (bin, source->udpsrc[i]);
2837
2838           /* and link to the funnel v4 */
2839           source->selpad[i] = selpad =
2840               gst_element_get_request_pad (priv->funnel[i], "sink_%u");
2841           pad = gst_element_get_static_pad (source->udpsrc[i], "src");
2842           gst_pad_link (pad, selpad);
2843           gst_object_unref (pad);
2844           gst_object_unref (selpad);
2845         }
2846         gst_object_unref (bin);
2847
2848         priv->transport_sources =
2849             g_list_prepend (priv->transport_sources, source);
2850       } else {
2851         GList *l;
2852
2853         for (l = priv->transport_sources; l; l = l->next) {
2854           source = l->data;
2855
2856           if (source->transport == trans) {
2857             priv->transport_sources =
2858                 g_list_delete_link (priv->transport_sources, l);
2859             break;
2860           }
2861         }
2862
2863         if (l != NULL) {
2864           gint i;
2865
2866           for (i = 0; i < 2; i++) {
2867             /* Will automatically unlink everything */
2868             gst_bin_remove (bin,
2869                 GST_ELEMENT (gst_object_ref (source->udpsrc[i])));
2870
2871             gst_element_set_state (source->udpsrc[i], GST_STATE_NULL);
2872             gst_object_unref (source->udpsrc[i]);
2873
2874             gst_element_release_request_pad (priv->funnel[i],
2875                 source->selpad[i]);
2876           }
2877
2878           g_slice_free (GstRTSPMulticastTransportSource, source);
2879         }
2880       }
2881
2882       /* fall through for the generic case */
2883     }
2884     case GST_RTSP_LOWER_TRANS_UDP:
2885     {
2886       gchar *dest;
2887       gint min, max;
2888       guint ttl = 0;
2889
2890       dest = tr->destination;
2891       if (tr->lower_transport == GST_RTSP_LOWER_TRANS_UDP_MCAST) {
2892         min = tr->port.min;
2893         max = tr->port.max;
2894         ttl = tr->ttl;
2895       } else {
2896         min = tr->client_port.min;
2897         max = tr->client_port.max;
2898       }
2899
2900       if (add) {
2901         if (ttl > 0) {
2902           GST_INFO ("setting ttl-mc %d", ttl);
2903           g_object_set (G_OBJECT (priv->udpsink[0]), "ttl-mc", ttl, NULL);
2904           g_object_set (G_OBJECT (priv->udpsink[1]), "ttl-mc", ttl, NULL);
2905         }
2906         GST_INFO ("adding %s:%d-%d", dest, min, max);
2907         g_signal_emit_by_name (priv->udpsink[0], "add", dest, min, NULL);
2908         g_signal_emit_by_name (priv->udpsink[1], "add", dest, max, NULL);
2909         priv->transports = g_list_prepend (priv->transports, trans);
2910       } else {
2911         GST_INFO ("removing %s:%d-%d", dest, min, max);
2912         g_signal_emit_by_name (priv->udpsink[0], "remove", dest, min, NULL);
2913         g_signal_emit_by_name (priv->udpsink[1], "remove", dest, max, NULL);
2914         priv->transports = g_list_remove (priv->transports, trans);
2915       }
2916       priv->transports_cookie++;
2917       break;
2918     }
2919     case GST_RTSP_LOWER_TRANS_TCP:
2920       if (add) {
2921         GST_INFO ("adding TCP %s", tr->destination);
2922         priv->transports = g_list_prepend (priv->transports, trans);
2923       } else {
2924         GST_INFO ("removing TCP %s", tr->destination);
2925         priv->transports = g_list_remove (priv->transports, trans);
2926       }
2927       priv->transports_cookie++;
2928       break;
2929     default:
2930       goto unknown_transport;
2931   }
2932   return TRUE;
2933
2934   /* ERRORS */
2935 unknown_transport:
2936   {
2937     GST_INFO ("Unknown transport %d", tr->lower_transport);
2938     return FALSE;
2939   }
2940 }
2941
2942
2943 /**
2944  * gst_rtsp_stream_add_transport:
2945  * @stream: a #GstRTSPStream
2946  * @trans: (transfer none): a #GstRTSPStreamTransport
2947  *
2948  * Add the transport in @trans to @stream. The media of @stream will
2949  * then also be send to the values configured in @trans.
2950  *
2951  * @stream must be joined to a bin.
2952  *
2953  * @trans must contain a valid #GstRTSPTransport.
2954  *
2955  * Returns: %TRUE if @trans was added
2956  */
2957 gboolean
2958 gst_rtsp_stream_add_transport (GstRTSPStream * stream,
2959     GstRTSPStreamTransport * trans)
2960 {
2961   GstRTSPStreamPrivate *priv;
2962   gboolean res;
2963
2964   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
2965   priv = stream->priv;
2966   g_return_val_if_fail (GST_IS_RTSP_STREAM_TRANSPORT (trans), FALSE);
2967   g_return_val_if_fail (priv->is_joined, FALSE);
2968
2969   g_mutex_lock (&priv->lock);
2970   res = update_transport (stream, trans, TRUE);
2971   g_mutex_unlock (&priv->lock);
2972
2973   return res;
2974 }
2975
2976 /**
2977  * gst_rtsp_stream_remove_transport:
2978  * @stream: a #GstRTSPStream
2979  * @trans: (transfer none): a #GstRTSPStreamTransport
2980  *
2981  * Remove the transport in @trans from @stream. The media of @stream will
2982  * not be sent to the values configured in @trans.
2983  *
2984  * @stream must be joined to a bin.
2985  *
2986  * @trans must contain a valid #GstRTSPTransport.
2987  *
2988  * Returns: %TRUE if @trans was removed
2989  */
2990 gboolean
2991 gst_rtsp_stream_remove_transport (GstRTSPStream * stream,
2992     GstRTSPStreamTransport * trans)
2993 {
2994   GstRTSPStreamPrivate *priv;
2995   gboolean res;
2996
2997   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
2998   priv = stream->priv;
2999   g_return_val_if_fail (GST_IS_RTSP_STREAM_TRANSPORT (trans), FALSE);
3000   g_return_val_if_fail (priv->is_joined, FALSE);
3001
3002   g_mutex_lock (&priv->lock);
3003   res = update_transport (stream, trans, FALSE);
3004   g_mutex_unlock (&priv->lock);
3005
3006   return res;
3007 }
3008
3009 /**
3010  * gst_rtsp_stream_update_crypto:
3011  * @stream: a #GstRTSPStream
3012  * @ssrc: the SSRC
3013  * @crypto: (transfer none) (allow-none): a #GstCaps with crypto info
3014  *
3015  * Update the new crypto information for @ssrc in @stream. If information
3016  * for @ssrc did not exist, it will be added. If information
3017  * for @ssrc existed, it will be replaced. If @crypto is %NULL, it will
3018  * be removed from @stream.
3019  *
3020  * Returns: %TRUE if @crypto could be updated
3021  */
3022 gboolean
3023 gst_rtsp_stream_update_crypto (GstRTSPStream * stream,
3024     guint ssrc, GstCaps * crypto)
3025 {
3026   GstRTSPStreamPrivate *priv;
3027
3028   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
3029   g_return_val_if_fail (crypto == NULL || GST_IS_CAPS (crypto), FALSE);
3030
3031   priv = stream->priv;
3032
3033   GST_DEBUG_OBJECT (stream, "update key for %08x", ssrc);
3034
3035   g_mutex_lock (&priv->lock);
3036   if (crypto)
3037     g_hash_table_insert (priv->keys, GINT_TO_POINTER (ssrc),
3038         gst_caps_ref (crypto));
3039   else
3040     g_hash_table_remove (priv->keys, GINT_TO_POINTER (ssrc));
3041   g_mutex_unlock (&priv->lock);
3042
3043   return TRUE;
3044 }
3045
3046 /**
3047  * gst_rtsp_stream_get_rtp_socket:
3048  * @stream: a #GstRTSPStream
3049  * @family: the socket family
3050  *
3051  * Get the RTP socket from @stream for a @family.
3052  *
3053  * @stream must be joined to a bin.
3054  *
3055  * Returns: (transfer full) (nullable): the RTP socket or %NULL if no
3056  * socket could be allocated for @family. Unref after usage
3057  */
3058 GSocket *
3059 gst_rtsp_stream_get_rtp_socket (GstRTSPStream * stream, GSocketFamily family)
3060 {
3061   GstRTSPStreamPrivate *priv = GST_RTSP_STREAM_GET_PRIVATE (stream);
3062   GSocket *socket;
3063   const gchar *name;
3064
3065   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
3066   g_return_val_if_fail (family == G_SOCKET_FAMILY_IPV4 ||
3067       family == G_SOCKET_FAMILY_IPV6, NULL);
3068   g_return_val_if_fail (priv->udpsink[0], NULL);
3069
3070   if (family == G_SOCKET_FAMILY_IPV6)
3071     name = "socket-v6";
3072   else
3073     name = "socket";
3074
3075   g_object_get (priv->udpsink[0], name, &socket, NULL);
3076
3077   return socket;
3078 }
3079
3080 /**
3081  * gst_rtsp_stream_get_rtcp_socket:
3082  * @stream: a #GstRTSPStream
3083  * @family: the socket family
3084  *
3085  * Get the RTCP socket from @stream for a @family.
3086  *
3087  * @stream must be joined to a bin.
3088  *
3089  * Returns: (transfer full) (nullable): the RTCP socket or %NULL if no
3090  * socket could be allocated for @family. Unref after usage
3091  */
3092 GSocket *
3093 gst_rtsp_stream_get_rtcp_socket (GstRTSPStream * stream, GSocketFamily family)
3094 {
3095   GstRTSPStreamPrivate *priv = GST_RTSP_STREAM_GET_PRIVATE (stream);
3096   GSocket *socket;
3097   const gchar *name;
3098
3099   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
3100   g_return_val_if_fail (family == G_SOCKET_FAMILY_IPV4 ||
3101       family == G_SOCKET_FAMILY_IPV6, NULL);
3102   g_return_val_if_fail (priv->udpsink[1], NULL);
3103
3104   if (family == G_SOCKET_FAMILY_IPV6)
3105     name = "socket-v6";
3106   else
3107     name = "socket";
3108
3109   g_object_get (priv->udpsink[1], name, &socket, NULL);
3110
3111   return socket;
3112 }
3113
3114 /**
3115  * gst_rtsp_stream_set_seqnum:
3116  * @stream: a #GstRTSPStream
3117  * @seqnum: a new sequence number
3118  *
3119  * Configure the sequence number in the payloader of @stream to @seqnum.
3120  */
3121 void
3122 gst_rtsp_stream_set_seqnum_offset (GstRTSPStream * stream, guint16 seqnum)
3123 {
3124   GstRTSPStreamPrivate *priv;
3125
3126   g_return_if_fail (GST_IS_RTSP_STREAM (stream));
3127
3128   priv = stream->priv;
3129
3130   g_object_set (G_OBJECT (priv->payloader), "seqnum-offset", seqnum, NULL);
3131 }
3132
3133 /**
3134  * gst_rtsp_stream_get_seqnum:
3135  * @stream: a #GstRTSPStream
3136  *
3137  * Get the configured sequence number in the payloader of @stream.
3138  *
3139  * Returns: the sequence number of the payloader.
3140  */
3141 guint16
3142 gst_rtsp_stream_get_current_seqnum (GstRTSPStream * stream)
3143 {
3144   GstRTSPStreamPrivate *priv;
3145   guint seqnum;
3146
3147   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), 0);
3148
3149   priv = stream->priv;
3150
3151   g_object_get (G_OBJECT (priv->payloader), "seqnum", &seqnum, NULL);
3152
3153   return seqnum;
3154 }
3155
3156 /**
3157  * gst_rtsp_stream_transport_filter:
3158  * @stream: a #GstRTSPStream
3159  * @func: (scope call) (allow-none): a callback
3160  * @user_data: (closure): user data passed to @func
3161  *
3162  * Call @func for each transport managed by @stream. The result value of @func
3163  * determines what happens to the transport. @func will be called with @stream
3164  * locked so no further actions on @stream can be performed from @func.
3165  *
3166  * If @func returns #GST_RTSP_FILTER_REMOVE, the transport will be removed from
3167  * @stream.
3168  *
3169  * If @func returns #GST_RTSP_FILTER_KEEP, the transport will remain in @stream.
3170  *
3171  * If @func returns #GST_RTSP_FILTER_REF, the transport will remain in @stream but
3172  * will also be added with an additional ref to the result #GList of this
3173  * function..
3174  *
3175  * When @func is %NULL, #GST_RTSP_FILTER_REF will be assumed for each transport.
3176  *
3177  * Returns: (element-type GstRTSPStreamTransport) (transfer full): a #GList with all
3178  * transports for which @func returned #GST_RTSP_FILTER_REF. After usage, each
3179  * element in the #GList should be unreffed before the list is freed.
3180  */
3181 GList *
3182 gst_rtsp_stream_transport_filter (GstRTSPStream * stream,
3183     GstRTSPStreamTransportFilterFunc func, gpointer user_data)
3184 {
3185   GstRTSPStreamPrivate *priv;
3186   GList *result, *walk, *next;
3187   GHashTable *visited = NULL;
3188   guint cookie;
3189
3190   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), NULL);
3191
3192   priv = stream->priv;
3193
3194   result = NULL;
3195   if (func)
3196     visited = g_hash_table_new_full (NULL, NULL, g_object_unref, NULL);
3197
3198   g_mutex_lock (&priv->lock);
3199 restart:
3200   cookie = priv->transports_cookie;
3201   for (walk = priv->transports; walk; walk = next) {
3202     GstRTSPStreamTransport *trans = walk->data;
3203     GstRTSPFilterResult res;
3204     gboolean changed;
3205
3206     next = g_list_next (walk);
3207
3208     if (func) {
3209       /* only visit each transport once */
3210       if (g_hash_table_contains (visited, trans))
3211         continue;
3212
3213       g_hash_table_add (visited, g_object_ref (trans));
3214       g_mutex_unlock (&priv->lock);
3215
3216       res = func (stream, trans, user_data);
3217
3218       g_mutex_lock (&priv->lock);
3219     } else
3220       res = GST_RTSP_FILTER_REF;
3221
3222     changed = (cookie != priv->transports_cookie);
3223
3224     switch (res) {
3225       case GST_RTSP_FILTER_REMOVE:
3226         update_transport (stream, trans, FALSE);
3227         break;
3228       case GST_RTSP_FILTER_REF:
3229         result = g_list_prepend (result, g_object_ref (trans));
3230         break;
3231       case GST_RTSP_FILTER_KEEP:
3232       default:
3233         break;
3234     }
3235     if (changed)
3236       goto restart;
3237   }
3238   g_mutex_unlock (&priv->lock);
3239
3240   if (func)
3241     g_hash_table_unref (visited);
3242
3243   return result;
3244 }
3245
3246 static GstPadProbeReturn
3247 pad_blocking (GstPad * pad, GstPadProbeInfo * info, gpointer user_data)
3248 {
3249   GstRTSPStreamPrivate *priv;
3250   GstRTSPStream *stream;
3251
3252   stream = user_data;
3253   priv = stream->priv;
3254
3255   GST_DEBUG_OBJECT (pad, "now blocking");
3256
3257   g_mutex_lock (&priv->lock);
3258   priv->blocking = TRUE;
3259   g_mutex_unlock (&priv->lock);
3260
3261   gst_element_post_message (priv->payloader,
3262       gst_message_new_element (GST_OBJECT_CAST (priv->payloader),
3263           gst_structure_new_empty ("GstRTSPStreamBlocking")));
3264
3265   return GST_PAD_PROBE_OK;
3266 }
3267
3268 /**
3269  * gst_rtsp_stream_set_blocked:
3270  * @stream: a #GstRTSPStream
3271  * @blocked: boolean indicating we should block or unblock
3272  *
3273  * Blocks or unblocks the dataflow on @stream.
3274  *
3275  * Returns: %TRUE on success
3276  */
3277 gboolean
3278 gst_rtsp_stream_set_blocked (GstRTSPStream * stream, gboolean blocked)
3279 {
3280   GstRTSPStreamPrivate *priv;
3281
3282   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
3283
3284   priv = stream->priv;
3285
3286   g_mutex_lock (&priv->lock);
3287   if (blocked) {
3288     priv->blocking = FALSE;
3289     if (priv->blocked_id == 0) {
3290       priv->blocked_id = gst_pad_add_probe (priv->srcpad,
3291           GST_PAD_PROBE_TYPE_BLOCK | GST_PAD_PROBE_TYPE_BUFFER |
3292           GST_PAD_PROBE_TYPE_BUFFER_LIST, pad_blocking,
3293           g_object_ref (stream), g_object_unref);
3294     }
3295   } else {
3296     if (priv->blocked_id != 0) {
3297       gst_pad_remove_probe (priv->srcpad, priv->blocked_id);
3298       priv->blocked_id = 0;
3299       priv->blocking = FALSE;
3300     }
3301   }
3302   g_mutex_unlock (&priv->lock);
3303
3304   return TRUE;
3305 }
3306
3307 /**
3308  * gst_rtsp_stream_is_blocking:
3309  * @stream: a #GstRTSPStream
3310  *
3311  * Check if @stream is blocking on a #GstBuffer.
3312  *
3313  * Returns: %TRUE if @stream is blocking
3314  */
3315 gboolean
3316 gst_rtsp_stream_is_blocking (GstRTSPStream * stream)
3317 {
3318   GstRTSPStreamPrivate *priv;
3319   gboolean result;
3320
3321   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
3322
3323   priv = stream->priv;
3324
3325   g_mutex_lock (&priv->lock);
3326   result = priv->blocking;
3327   g_mutex_unlock (&priv->lock);
3328
3329   return result;
3330 }
3331
3332 /**
3333  * gst_rtsp_stream_query_position:
3334  * @stream: a #GstRTSPStream
3335  *
3336  * Query the position of the stream in %GST_FORMAT_TIME. This only considers
3337  * the RTP parts of the pipeline and not the RTCP parts.
3338  *
3339  * Returns: %TRUE if the position could be queried
3340  */
3341 gboolean
3342 gst_rtsp_stream_query_position (GstRTSPStream * stream, gint64 * position)
3343 {
3344   GstRTSPStreamPrivate *priv;
3345   GstElement *sink;
3346   gboolean ret;
3347
3348   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
3349
3350   priv = stream->priv;
3351
3352   g_mutex_lock (&priv->lock);
3353   if ((sink = priv->udpsink[0]))
3354     gst_object_ref (sink);
3355   g_mutex_unlock (&priv->lock);
3356
3357   if (!sink)
3358     return FALSE;
3359
3360   ret = gst_element_query_position (sink, GST_FORMAT_TIME, position);
3361   gst_object_unref (sink);
3362
3363   return ret;
3364 }
3365
3366 /**
3367  * gst_rtsp_stream_query_stop:
3368  * @stream: a #GstRTSPStream
3369  *
3370  * Query the stop of the stream in %GST_FORMAT_TIME. This only considers
3371  * the RTP parts of the pipeline and not the RTCP parts.
3372  *
3373  * Returns: %TRUE if the stop could be queried
3374  */
3375 gboolean
3376 gst_rtsp_stream_query_stop (GstRTSPStream * stream, gint64 * stop)
3377 {
3378   GstRTSPStreamPrivate *priv;
3379   GstElement *sink;
3380   GstQuery *query;
3381   gboolean ret;
3382
3383   g_return_val_if_fail (GST_IS_RTSP_STREAM (stream), FALSE);
3384
3385   priv = stream->priv;
3386
3387   g_mutex_lock (&priv->lock);
3388   if ((sink = priv->udpsink[0]))
3389     gst_object_ref (sink);
3390   g_mutex_unlock (&priv->lock);
3391
3392   if (!sink)
3393     return FALSE;
3394
3395   query = gst_query_new_segment (GST_FORMAT_TIME);
3396   if ((ret = gst_element_query (sink, query))) {
3397     GstFormat format;
3398
3399     gst_query_parse_segment (query, NULL, &format, NULL, stop);
3400     if (format != GST_FORMAT_TIME)
3401       *stop = -1;
3402   }
3403   gst_query_unref (query);
3404   gst_object_unref (sink);
3405
3406   return ret;
3407
3408 }