rtsp-media: Don't set the transport mode based on what elements we find
[platform/upstream/gstreamer.git] / gst / rtsp-server / rtsp-media.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-media
23  * @short_description: The media pipeline
24  * @see_also: #GstRTSPMediaFactory, #GstRTSPStream, #GstRTSPSession,
25  *     #GstRTSPSessionMedia
26  *
27  * a #GstRTSPMedia contains the complete GStreamer pipeline to manage the
28  * streaming to the clients. The actual data transfer is done by the
29  * #GstRTSPStream objects that are created and exposed by the #GstRTSPMedia.
30  *
31  * The #GstRTSPMedia is usually created from a #GstRTSPMediaFactory when the
32  * client does a DESCRIBE or SETUP of a resource.
33  *
34  * A media is created with gst_rtsp_media_new() that takes the element that will
35  * provide the streaming elements. For each of the streams, a new #GstRTSPStream
36  * object needs to be made with the gst_rtsp_media_create_stream() which takes
37  * the payloader element and the source pad that produces the RTP stream.
38  *
39  * The pipeline of the media is set to PAUSED with gst_rtsp_media_prepare(). The
40  * prepare method will add rtpbin and sinks and sources to send and receive RTP
41  * and RTCP packets from the clients. Each stream srcpad is connected to an
42  * input into the internal rtpbin.
43  *
44  * It is also possible to dynamically create #GstRTSPStream objects during the
45  * prepare phase. With gst_rtsp_media_get_status() you can check the status of
46  * the prepare phase.
47  *
48  * After the media is prepared, it is ready for streaming. It will usually be
49  * managed in a session with gst_rtsp_session_manage_media(). See
50  * #GstRTSPSession and #GstRTSPSessionMedia.
51  *
52  * The state of the media can be controlled with gst_rtsp_media_set_state ().
53  * Seeking can be done with gst_rtsp_media_seek().
54  *
55  * With gst_rtsp_media_unprepare() the pipeline is stopped and shut down. When
56  * gst_rtsp_media_set_eos_shutdown() an EOS will be sent to the pipeline to
57  * cleanly shut down.
58  *
59  * With gst_rtsp_media_set_shared(), the media can be shared between multiple
60  * clients. With gst_rtsp_media_set_reusable() you can control if the pipeline
61  * can be prepared again after an unprepare.
62  *
63  * Last reviewed on 2013-07-11 (1.0.0)
64  */
65
66 #include <stdio.h>
67 #include <string.h>
68 #include <stdlib.h>
69
70 #include <gst/app/gstappsrc.h>
71 #include <gst/app/gstappsink.h>
72
73 #include <gst/sdp/gstmikey.h>
74 #include <gst/rtp/gstrtppayloads.h>
75
76 #define AES_128_KEY_LEN 16
77 #define AES_256_KEY_LEN 32
78
79 #define HMAC_32_KEY_LEN 4
80 #define HMAC_80_KEY_LEN 10
81
82 #include "rtsp-media.h"
83
84 #define GST_RTSP_MEDIA_GET_PRIVATE(obj)  \
85      (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GST_TYPE_RTSP_MEDIA, GstRTSPMediaPrivate))
86
87 struct _GstRTSPMediaPrivate
88 {
89   GMutex lock;
90   GCond cond;
91
92   /* protected by lock */
93   GstRTSPPermissions *permissions;
94   gboolean shared;
95   gboolean suspend_mode;
96   gboolean reusable;
97   GstRTSPProfile profiles;
98   GstRTSPLowerTrans protocols;
99   gboolean reused;
100   gboolean eos_shutdown;
101   guint buffer_size;
102   GstRTSPAddressPool *pool;
103   gboolean blocked;
104   GstRTSPTransportMode transport_mode;
105
106   GstElement *element;
107   GRecMutex state_lock;         /* locking order: state lock, lock */
108   GPtrArray *streams;           /* protected by lock */
109   GList *dynamic;               /* protected by lock */
110   GstRTSPMediaStatus status;    /* protected by lock */
111   gint prepare_count;
112   gint n_active;
113   gboolean adding;
114
115   /* the pipeline for the media */
116   GstElement *pipeline;
117   GstElement *fakesink;         /* protected by lock */
118   GSource *source;
119   guint id;
120   GstRTSPThread *thread;
121
122   gboolean time_provider;
123   GstNetTimeProvider *nettime;
124
125   gboolean is_live;
126   gboolean seekable;
127   gboolean buffering;
128   GstState target_state;
129
130   /* RTP session manager */
131   GstElement *rtpbin;
132
133   /* the range of media */
134   GstRTSPTimeRange range;       /* protected by lock */
135   GstClockTime range_start;
136   GstClockTime range_stop;
137
138   GList *payloads;              /* protected by lock */
139   GstClockTime rtx_time;        /* protected by lock */
140   guint latency;                /* protected by lock */
141 };
142
143 #define DEFAULT_SHARED          FALSE
144 #define DEFAULT_SUSPEND_MODE    GST_RTSP_SUSPEND_MODE_NONE
145 #define DEFAULT_REUSABLE        FALSE
146 #define DEFAULT_PROFILES        GST_RTSP_PROFILE_AVP
147 #define DEFAULT_PROTOCOLS       GST_RTSP_LOWER_TRANS_UDP | GST_RTSP_LOWER_TRANS_UDP_MCAST | \
148                                         GST_RTSP_LOWER_TRANS_TCP
149 #define DEFAULT_EOS_SHUTDOWN    FALSE
150 #define DEFAULT_BUFFER_SIZE     0x80000
151 #define DEFAULT_TIME_PROVIDER   FALSE
152 #define DEFAULT_LATENCY         200
153 #define DEFAULT_TRANSPORT_MODE  GST_RTSP_TRANSPORT_MODE_PLAY
154
155 /* define to dump received RTCP packets */
156 #undef DUMP_STATS
157
158 enum
159 {
160   PROP_0,
161   PROP_SHARED,
162   PROP_SUSPEND_MODE,
163   PROP_REUSABLE,
164   PROP_PROFILES,
165   PROP_PROTOCOLS,
166   PROP_EOS_SHUTDOWN,
167   PROP_BUFFER_SIZE,
168   PROP_ELEMENT,
169   PROP_TIME_PROVIDER,
170   PROP_LATENCY,
171   PROP_TRANSPORT_MODE,
172   PROP_LAST
173 };
174
175 enum
176 {
177   SIGNAL_NEW_STREAM,
178   SIGNAL_REMOVED_STREAM,
179   SIGNAL_PREPARED,
180   SIGNAL_UNPREPARED,
181   SIGNAL_TARGET_STATE,
182   SIGNAL_NEW_STATE,
183   SIGNAL_LAST
184 };
185
186 GST_DEBUG_CATEGORY_STATIC (rtsp_media_debug);
187 #define GST_CAT_DEFAULT rtsp_media_debug
188
189 static void gst_rtsp_media_get_property (GObject * object, guint propid,
190     GValue * value, GParamSpec * pspec);
191 static void gst_rtsp_media_set_property (GObject * object, guint propid,
192     const GValue * value, GParamSpec * pspec);
193 static void gst_rtsp_media_finalize (GObject * obj);
194
195 static gboolean default_handle_message (GstRTSPMedia * media,
196     GstMessage * message);
197 static void finish_unprepare (GstRTSPMedia * media);
198 static gboolean default_prepare (GstRTSPMedia * media, GstRTSPThread * thread);
199 static gboolean default_unprepare (GstRTSPMedia * media);
200 static gboolean default_suspend (GstRTSPMedia * media);
201 static gboolean default_unsuspend (GstRTSPMedia * media);
202 static gboolean default_convert_range (GstRTSPMedia * media,
203     GstRTSPTimeRange * range, GstRTSPRangeUnit unit);
204 static gboolean default_query_position (GstRTSPMedia * media,
205     gint64 * position);
206 static gboolean default_query_stop (GstRTSPMedia * media, gint64 * stop);
207 static GstElement *default_create_rtpbin (GstRTSPMedia * media);
208 static gboolean default_setup_sdp (GstRTSPMedia * media, GstSDPMessage * sdp,
209     GstSDPInfo * info);
210 static gboolean default_handle_sdp (GstRTSPMedia * media, GstSDPMessage * sdp);
211
212 static gboolean wait_preroll (GstRTSPMedia * media);
213
214 static guint gst_rtsp_media_signals[SIGNAL_LAST] = { 0 };
215
216 #define C_ENUM(v) ((gint) v)
217
218 GType
219 gst_rtsp_suspend_mode_get_type (void)
220 {
221   static gsize id = 0;
222   static const GEnumValue values[] = {
223     {C_ENUM (GST_RTSP_SUSPEND_MODE_NONE), "GST_RTSP_SUSPEND_MODE_NONE", "none"},
224     {C_ENUM (GST_RTSP_SUSPEND_MODE_PAUSE), "GST_RTSP_SUSPEND_MODE_PAUSE",
225         "pause"},
226     {C_ENUM (GST_RTSP_SUSPEND_MODE_RESET), "GST_RTSP_SUSPEND_MODE_RESET",
227         "reset"},
228     {0, NULL, NULL}
229   };
230
231   if (g_once_init_enter (&id)) {
232     GType tmp = g_enum_register_static ("GstRTSPSuspendMode", values);
233     g_once_init_leave (&id, tmp);
234   }
235   return (GType) id;
236 }
237
238 #define C_FLAGS(v) ((guint) v)
239
240 GType
241 gst_rtsp_transport_mode_get_type (void)
242 {
243   static gsize id = 0;
244   static const GFlagsValue values[] = {
245     {C_FLAGS (GST_RTSP_TRANSPORT_MODE_PLAY), "GST_RTSP_TRANSPORT_MODE_PLAY",
246         "play"},
247     {C_FLAGS (GST_RTSP_TRANSPORT_MODE_RECORD), "GST_RTSP_TRANSPORT_MODE_RECORD",
248         "record"},
249     {0, NULL, NULL}
250   };
251
252   if (g_once_init_enter (&id)) {
253     GType tmp = g_flags_register_static ("GstRTSPTransportMode", values);
254     g_once_init_leave (&id, tmp);
255   }
256   return (GType) id;
257 }
258
259 G_DEFINE_TYPE (GstRTSPMedia, gst_rtsp_media, G_TYPE_OBJECT);
260
261 static void
262 gst_rtsp_media_class_init (GstRTSPMediaClass * klass)
263 {
264   GObjectClass *gobject_class;
265
266   g_type_class_add_private (klass, sizeof (GstRTSPMediaPrivate));
267
268   gobject_class = G_OBJECT_CLASS (klass);
269
270   gobject_class->get_property = gst_rtsp_media_get_property;
271   gobject_class->set_property = gst_rtsp_media_set_property;
272   gobject_class->finalize = gst_rtsp_media_finalize;
273
274   g_object_class_install_property (gobject_class, PROP_SHARED,
275       g_param_spec_boolean ("shared", "Shared",
276           "If this media pipeline can be shared", DEFAULT_SHARED,
277           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
278
279   g_object_class_install_property (gobject_class, PROP_SUSPEND_MODE,
280       g_param_spec_enum ("suspend-mode", "Suspend Mode",
281           "How to suspend the media in PAUSED", GST_TYPE_RTSP_SUSPEND_MODE,
282           DEFAULT_SUSPEND_MODE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
283
284   g_object_class_install_property (gobject_class, PROP_REUSABLE,
285       g_param_spec_boolean ("reusable", "Reusable",
286           "If this media pipeline can be reused after an unprepare",
287           DEFAULT_REUSABLE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
288
289   g_object_class_install_property (gobject_class, PROP_PROFILES,
290       g_param_spec_flags ("profiles", "Profiles",
291           "Allowed transfer profiles", GST_TYPE_RTSP_PROFILE,
292           DEFAULT_PROFILES, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
293
294   g_object_class_install_property (gobject_class, PROP_PROTOCOLS,
295       g_param_spec_flags ("protocols", "Protocols",
296           "Allowed lower transport protocols", GST_TYPE_RTSP_LOWER_TRANS,
297           DEFAULT_PROTOCOLS, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
298
299   g_object_class_install_property (gobject_class, PROP_EOS_SHUTDOWN,
300       g_param_spec_boolean ("eos-shutdown", "EOS Shutdown",
301           "Send an EOS event to the pipeline before unpreparing",
302           DEFAULT_EOS_SHUTDOWN, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
303
304   g_object_class_install_property (gobject_class, PROP_BUFFER_SIZE,
305       g_param_spec_uint ("buffer-size", "Buffer Size",
306           "The kernel UDP buffer size to use", 0, G_MAXUINT,
307           DEFAULT_BUFFER_SIZE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
308
309   g_object_class_install_property (gobject_class, PROP_ELEMENT,
310       g_param_spec_object ("element", "The Element",
311           "The GstBin to use for streaming the media", GST_TYPE_ELEMENT,
312           G_PARAM_CONSTRUCT_ONLY | G_PARAM_READWRITE));
313
314   g_object_class_install_property (gobject_class, PROP_TIME_PROVIDER,
315       g_param_spec_boolean ("time-provider", "Time Provider",
316           "Use a NetTimeProvider for clients",
317           DEFAULT_TIME_PROVIDER, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
318
319   g_object_class_install_property (gobject_class, PROP_LATENCY,
320       g_param_spec_uint ("latency", "Latency",
321           "Latency used for receiving media in milliseconds", 0, G_MAXUINT,
322           DEFAULT_BUFFER_SIZE, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
323
324   g_object_class_install_property (gobject_class, PROP_TRANSPORT_MODE,
325       g_param_spec_flags ("transport-mode", "Transport Mode",
326           "If this media pipeline can be used for PLAY or RECORD",
327           GST_TYPE_RTSP_TRANSPORT_MODE, DEFAULT_TRANSPORT_MODE,
328           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
329
330   gst_rtsp_media_signals[SIGNAL_NEW_STREAM] =
331       g_signal_new ("new-stream", G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
332       G_STRUCT_OFFSET (GstRTSPMediaClass, new_stream), NULL, NULL,
333       g_cclosure_marshal_generic, G_TYPE_NONE, 1, GST_TYPE_RTSP_STREAM);
334
335   gst_rtsp_media_signals[SIGNAL_REMOVED_STREAM] =
336       g_signal_new ("removed-stream", G_TYPE_FROM_CLASS (klass),
337       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPMediaClass, removed_stream),
338       NULL, NULL, g_cclosure_marshal_generic, G_TYPE_NONE, 1,
339       GST_TYPE_RTSP_STREAM);
340
341   gst_rtsp_media_signals[SIGNAL_PREPARED] =
342       g_signal_new ("prepared", G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
343       G_STRUCT_OFFSET (GstRTSPMediaClass, prepared), NULL, NULL,
344       g_cclosure_marshal_generic, G_TYPE_NONE, 0, G_TYPE_NONE);
345
346   gst_rtsp_media_signals[SIGNAL_UNPREPARED] =
347       g_signal_new ("unprepared", G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
348       G_STRUCT_OFFSET (GstRTSPMediaClass, unprepared), NULL, NULL,
349       g_cclosure_marshal_generic, G_TYPE_NONE, 0, G_TYPE_NONE);
350
351   gst_rtsp_media_signals[SIGNAL_TARGET_STATE] =
352       g_signal_new ("target-state", G_TYPE_FROM_CLASS (klass),
353       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPMediaClass, target_state),
354       NULL, NULL, g_cclosure_marshal_generic, G_TYPE_NONE, 1, G_TYPE_INT);
355
356   gst_rtsp_media_signals[SIGNAL_NEW_STATE] =
357       g_signal_new ("new-state", G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
358       G_STRUCT_OFFSET (GstRTSPMediaClass, new_state), NULL, NULL,
359       g_cclosure_marshal_generic, G_TYPE_NONE, 1, G_TYPE_INT);
360
361   GST_DEBUG_CATEGORY_INIT (rtsp_media_debug, "rtspmedia", 0, "GstRTSPMedia");
362
363   klass->handle_message = default_handle_message;
364   klass->prepare = default_prepare;
365   klass->unprepare = default_unprepare;
366   klass->suspend = default_suspend;
367   klass->unsuspend = default_unsuspend;
368   klass->convert_range = default_convert_range;
369   klass->query_position = default_query_position;
370   klass->query_stop = default_query_stop;
371   klass->create_rtpbin = default_create_rtpbin;
372   klass->setup_sdp = default_setup_sdp;
373   klass->handle_sdp = default_handle_sdp;
374 }
375
376 static void
377 gst_rtsp_media_init (GstRTSPMedia * media)
378 {
379   GstRTSPMediaPrivate *priv = GST_RTSP_MEDIA_GET_PRIVATE (media);
380
381   media->priv = priv;
382
383   priv->streams = g_ptr_array_new_with_free_func (g_object_unref);
384   g_mutex_init (&priv->lock);
385   g_cond_init (&priv->cond);
386   g_rec_mutex_init (&priv->state_lock);
387
388   priv->shared = DEFAULT_SHARED;
389   priv->suspend_mode = DEFAULT_SUSPEND_MODE;
390   priv->reusable = DEFAULT_REUSABLE;
391   priv->profiles = DEFAULT_PROFILES;
392   priv->protocols = DEFAULT_PROTOCOLS;
393   priv->eos_shutdown = DEFAULT_EOS_SHUTDOWN;
394   priv->buffer_size = DEFAULT_BUFFER_SIZE;
395   priv->time_provider = DEFAULT_TIME_PROVIDER;
396   priv->transport_mode = DEFAULT_TRANSPORT_MODE;
397 }
398
399 static void
400 gst_rtsp_media_finalize (GObject * obj)
401 {
402   GstRTSPMediaPrivate *priv;
403   GstRTSPMedia *media;
404
405   media = GST_RTSP_MEDIA (obj);
406   priv = media->priv;
407
408   GST_INFO ("finalize media %p", media);
409
410   if (priv->permissions)
411     gst_rtsp_permissions_unref (priv->permissions);
412
413   g_ptr_array_unref (priv->streams);
414
415   g_list_free_full (priv->dynamic, gst_object_unref);
416
417   if (priv->pipeline)
418     gst_object_unref (priv->pipeline);
419   if (priv->nettime)
420     gst_object_unref (priv->nettime);
421   gst_object_unref (priv->element);
422   if (priv->pool)
423     g_object_unref (priv->pool);
424   if (priv->payloads)
425     g_list_free (priv->payloads);
426   g_mutex_clear (&priv->lock);
427   g_cond_clear (&priv->cond);
428   g_rec_mutex_clear (&priv->state_lock);
429
430   G_OBJECT_CLASS (gst_rtsp_media_parent_class)->finalize (obj);
431 }
432
433 static void
434 gst_rtsp_media_get_property (GObject * object, guint propid,
435     GValue * value, GParamSpec * pspec)
436 {
437   GstRTSPMedia *media = GST_RTSP_MEDIA (object);
438
439   switch (propid) {
440     case PROP_ELEMENT:
441       g_value_set_object (value, media->priv->element);
442       break;
443     case PROP_SHARED:
444       g_value_set_boolean (value, gst_rtsp_media_is_shared (media));
445       break;
446     case PROP_SUSPEND_MODE:
447       g_value_set_enum (value, gst_rtsp_media_get_suspend_mode (media));
448       break;
449     case PROP_REUSABLE:
450       g_value_set_boolean (value, gst_rtsp_media_is_reusable (media));
451       break;
452     case PROP_PROFILES:
453       g_value_set_flags (value, gst_rtsp_media_get_profiles (media));
454       break;
455     case PROP_PROTOCOLS:
456       g_value_set_flags (value, gst_rtsp_media_get_protocols (media));
457       break;
458     case PROP_EOS_SHUTDOWN:
459       g_value_set_boolean (value, gst_rtsp_media_is_eos_shutdown (media));
460       break;
461     case PROP_BUFFER_SIZE:
462       g_value_set_uint (value, gst_rtsp_media_get_buffer_size (media));
463       break;
464     case PROP_TIME_PROVIDER:
465       g_value_set_boolean (value, gst_rtsp_media_is_time_provider (media));
466       break;
467     case PROP_LATENCY:
468       g_value_set_uint (value, gst_rtsp_media_get_latency (media));
469       break;
470     case PROP_TRANSPORT_MODE:
471       g_value_set_flags (value, gst_rtsp_media_get_transport_mode (media));
472       break;
473     default:
474       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
475   }
476 }
477
478 static void
479 gst_rtsp_media_set_property (GObject * object, guint propid,
480     const GValue * value, GParamSpec * pspec)
481 {
482   GstRTSPMedia *media = GST_RTSP_MEDIA (object);
483
484   switch (propid) {
485     case PROP_ELEMENT:
486       media->priv->element = g_value_get_object (value);
487       gst_object_ref_sink (media->priv->element);
488       break;
489     case PROP_SHARED:
490       gst_rtsp_media_set_shared (media, g_value_get_boolean (value));
491       break;
492     case PROP_SUSPEND_MODE:
493       gst_rtsp_media_set_suspend_mode (media, g_value_get_enum (value));
494       break;
495     case PROP_REUSABLE:
496       gst_rtsp_media_set_reusable (media, g_value_get_boolean (value));
497       break;
498     case PROP_PROFILES:
499       gst_rtsp_media_set_profiles (media, g_value_get_flags (value));
500       break;
501     case PROP_PROTOCOLS:
502       gst_rtsp_media_set_protocols (media, g_value_get_flags (value));
503       break;
504     case PROP_EOS_SHUTDOWN:
505       gst_rtsp_media_set_eos_shutdown (media, g_value_get_boolean (value));
506       break;
507     case PROP_BUFFER_SIZE:
508       gst_rtsp_media_set_buffer_size (media, g_value_get_uint (value));
509       break;
510     case PROP_TIME_PROVIDER:
511       gst_rtsp_media_use_time_provider (media, g_value_get_boolean (value));
512       break;
513     case PROP_LATENCY:
514       gst_rtsp_media_set_latency (media, g_value_get_uint (value));
515       break;
516     case PROP_TRANSPORT_MODE:
517       gst_rtsp_media_set_transport_mode (media, g_value_get_flags (value));
518       break;
519     default:
520       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
521   }
522 }
523
524 typedef struct
525 {
526   gint64 position;
527   gboolean ret;
528 } DoQueryPositionData;
529
530 static void
531 do_query_position (GstRTSPStream * stream, DoQueryPositionData * data)
532 {
533   gint64 tmp;
534
535   if (gst_rtsp_stream_query_position (stream, &tmp)) {
536     data->position = MAX (data->position, tmp);
537     data->ret = TRUE;
538   }
539 }
540
541 static gboolean
542 default_query_position (GstRTSPMedia * media, gint64 * position)
543 {
544   GstRTSPMediaPrivate *priv;
545   DoQueryPositionData data;
546
547   priv = media->priv;
548
549   data.position = -1;
550   data.ret = FALSE;
551
552   g_ptr_array_foreach (priv->streams, (GFunc) do_query_position, &data);
553
554   *position = data.position;
555
556   return data.ret;
557 }
558
559 typedef struct
560 {
561   gint64 stop;
562   gboolean ret;
563 } DoQueryStopData;
564
565 static void
566 do_query_stop (GstRTSPStream * stream, DoQueryStopData * data)
567 {
568   gint64 tmp;
569
570   if (gst_rtsp_stream_query_stop (stream, &tmp)) {
571     data->stop = MAX (data->stop, tmp);
572     data->ret = TRUE;
573   }
574 }
575
576 static gboolean
577 default_query_stop (GstRTSPMedia * media, gint64 * stop)
578 {
579   GstRTSPMediaPrivate *priv;
580   DoQueryStopData data;
581
582   priv = media->priv;
583
584   data.stop = -1;
585   data.ret = FALSE;
586
587   g_ptr_array_foreach (priv->streams, (GFunc) do_query_stop, &data);
588
589   *stop = data.stop;
590
591   return data.ret;
592 }
593
594 static GstElement *
595 default_create_rtpbin (GstRTSPMedia * media)
596 {
597   GstElement *rtpbin;
598
599   rtpbin = gst_element_factory_make ("rtpbin", NULL);
600
601   return rtpbin;
602 }
603
604 /* must be called with state lock */
605 static void
606 collect_media_stats (GstRTSPMedia * media)
607 {
608   GstRTSPMediaPrivate *priv = media->priv;
609   gint64 position = 0, stop = -1;
610
611   if (priv->status != GST_RTSP_MEDIA_STATUS_PREPARED &&
612       priv->status != GST_RTSP_MEDIA_STATUS_PREPARING)
613     return;
614
615   priv->range.unit = GST_RTSP_RANGE_NPT;
616
617   GST_INFO ("collect media stats");
618
619   if (priv->is_live) {
620     priv->range.min.type = GST_RTSP_TIME_NOW;
621     priv->range.min.seconds = -1;
622     priv->range_start = -1;
623     priv->range.max.type = GST_RTSP_TIME_END;
624     priv->range.max.seconds = -1;
625     priv->range_stop = -1;
626   } else {
627     GstRTSPMediaClass *klass;
628     gboolean ret;
629
630     klass = GST_RTSP_MEDIA_GET_CLASS (media);
631
632     /* get the position */
633     ret = FALSE;
634     if (klass->query_position)
635       ret = klass->query_position (media, &position);
636
637     if (!ret) {
638       GST_INFO ("position query failed");
639       position = 0;
640     }
641
642     /* get the current segment stop */
643     ret = FALSE;
644     if (klass->query_stop)
645       ret = klass->query_stop (media, &stop);
646
647     if (!ret) {
648       GST_INFO ("stop query failed");
649       stop = -1;
650     }
651
652     GST_INFO ("stats: position %" GST_TIME_FORMAT ", stop %"
653         GST_TIME_FORMAT, GST_TIME_ARGS (position), GST_TIME_ARGS (stop));
654
655     if (position == -1) {
656       priv->range.min.type = GST_RTSP_TIME_NOW;
657       priv->range.min.seconds = -1;
658       priv->range_start = -1;
659     } else {
660       priv->range.min.type = GST_RTSP_TIME_SECONDS;
661       priv->range.min.seconds = ((gdouble) position) / GST_SECOND;
662       priv->range_start = position;
663     }
664     if (stop == -1) {
665       priv->range.max.type = GST_RTSP_TIME_END;
666       priv->range.max.seconds = -1;
667       priv->range_stop = -1;
668     } else {
669       priv->range.max.type = GST_RTSP_TIME_SECONDS;
670       priv->range.max.seconds = ((gdouble) stop) / GST_SECOND;
671       priv->range_stop = stop;
672     }
673   }
674 }
675
676 /**
677  * gst_rtsp_media_new:
678  * @element: (transfer full): a #GstElement
679  *
680  * Create a new #GstRTSPMedia instance. @element is the bin element that
681  * provides the different streams. The #GstRTSPMedia object contains the
682  * element to produce RTP data for one or more related (audio/video/..)
683  * streams.
684  *
685  * Ownership is taken of @element.
686  *
687  * Returns: (transfer full): a new #GstRTSPMedia object.
688  */
689 GstRTSPMedia *
690 gst_rtsp_media_new (GstElement * element)
691 {
692   GstRTSPMedia *result;
693
694   g_return_val_if_fail (GST_IS_ELEMENT (element), NULL);
695
696   result = g_object_new (GST_TYPE_RTSP_MEDIA, "element", element, NULL);
697
698   return result;
699 }
700
701 /**
702  * gst_rtsp_media_get_element:
703  * @media: a #GstRTSPMedia
704  *
705  * Get the element that was used when constructing @media.
706  *
707  * Returns: (transfer full): a #GstElement. Unref after usage.
708  */
709 GstElement *
710 gst_rtsp_media_get_element (GstRTSPMedia * media)
711 {
712   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), NULL);
713
714   return gst_object_ref (media->priv->element);
715 }
716
717 /**
718  * gst_rtsp_media_take_pipeline:
719  * @media: a #GstRTSPMedia
720  * @pipeline: (transfer full): a #GstPipeline
721  *
722  * Set @pipeline as the #GstPipeline for @media. Ownership is
723  * taken of @pipeline.
724  */
725 void
726 gst_rtsp_media_take_pipeline (GstRTSPMedia * media, GstPipeline * pipeline)
727 {
728   GstRTSPMediaPrivate *priv;
729   GstElement *old;
730   GstNetTimeProvider *nettime;
731
732   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
733   g_return_if_fail (GST_IS_PIPELINE (pipeline));
734
735   priv = media->priv;
736
737   g_mutex_lock (&priv->lock);
738   old = priv->pipeline;
739   priv->pipeline = GST_ELEMENT_CAST (pipeline);
740   nettime = priv->nettime;
741   priv->nettime = NULL;
742   g_mutex_unlock (&priv->lock);
743
744   if (old)
745     gst_object_unref (old);
746
747   if (nettime)
748     gst_object_unref (nettime);
749
750   gst_bin_add (GST_BIN_CAST (pipeline), priv->element);
751 }
752
753 /**
754  * gst_rtsp_media_set_permissions:
755  * @media: a #GstRTSPMedia
756  * @permissions: (transfer none): a #GstRTSPPermissions
757  *
758  * Set @permissions on @media.
759  */
760 void
761 gst_rtsp_media_set_permissions (GstRTSPMedia * media,
762     GstRTSPPermissions * permissions)
763 {
764   GstRTSPMediaPrivate *priv;
765
766   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
767
768   priv = media->priv;
769
770   g_mutex_lock (&priv->lock);
771   if (priv->permissions)
772     gst_rtsp_permissions_unref (priv->permissions);
773   if ((priv->permissions = permissions))
774     gst_rtsp_permissions_ref (permissions);
775   g_mutex_unlock (&priv->lock);
776 }
777
778 /**
779  * gst_rtsp_media_get_permissions:
780  * @media: a #GstRTSPMedia
781  *
782  * Get the permissions object from @media.
783  *
784  * Returns: (transfer full): a #GstRTSPPermissions object, unref after usage.
785  */
786 GstRTSPPermissions *
787 gst_rtsp_media_get_permissions (GstRTSPMedia * media)
788 {
789   GstRTSPMediaPrivate *priv;
790   GstRTSPPermissions *result;
791
792   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), NULL);
793
794   priv = media->priv;
795
796   g_mutex_lock (&priv->lock);
797   if ((result = priv->permissions))
798     gst_rtsp_permissions_ref (result);
799   g_mutex_unlock (&priv->lock);
800
801   return result;
802 }
803
804 /**
805  * gst_rtsp_media_set_suspend_mode:
806  * @media: a #GstRTSPMedia
807  * @mode: the new #GstRTSPSuspendMode
808  *
809  * Control how @ media will be suspended after the SDP has been generated and
810  * after a PAUSE request has been performed.
811  *
812  * Media must be unprepared when setting the suspend mode.
813  */
814 void
815 gst_rtsp_media_set_suspend_mode (GstRTSPMedia * media, GstRTSPSuspendMode mode)
816 {
817   GstRTSPMediaPrivate *priv;
818
819   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
820
821   priv = media->priv;
822
823   g_rec_mutex_lock (&priv->state_lock);
824   if (priv->status == GST_RTSP_MEDIA_STATUS_PREPARED)
825     goto was_prepared;
826   priv->suspend_mode = mode;
827   g_rec_mutex_unlock (&priv->state_lock);
828
829   return;
830
831   /* ERRORS */
832 was_prepared:
833   {
834     GST_WARNING ("media %p was prepared", media);
835     g_rec_mutex_unlock (&priv->state_lock);
836   }
837 }
838
839 /**
840  * gst_rtsp_media_get_suspend_mode:
841  * @media: a #GstRTSPMedia
842  *
843  * Get how @media will be suspended.
844  *
845  * Returns: #GstRTSPSuspendMode.
846  */
847 GstRTSPSuspendMode
848 gst_rtsp_media_get_suspend_mode (GstRTSPMedia * media)
849 {
850   GstRTSPMediaPrivate *priv;
851   GstRTSPSuspendMode res;
852
853   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), GST_RTSP_SUSPEND_MODE_NONE);
854
855   priv = media->priv;
856
857   g_rec_mutex_lock (&priv->state_lock);
858   res = priv->suspend_mode;
859   g_rec_mutex_unlock (&priv->state_lock);
860
861   return res;
862 }
863
864 /**
865  * gst_rtsp_media_set_shared:
866  * @media: a #GstRTSPMedia
867  * @shared: the new value
868  *
869  * Set or unset if the pipeline for @media can be shared will multiple clients.
870  * When @shared is %TRUE, client requests for this media will share the media
871  * pipeline.
872  */
873 void
874 gst_rtsp_media_set_shared (GstRTSPMedia * media, gboolean shared)
875 {
876   GstRTSPMediaPrivate *priv;
877
878   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
879
880   priv = media->priv;
881
882   g_mutex_lock (&priv->lock);
883   priv->shared = shared;
884   g_mutex_unlock (&priv->lock);
885 }
886
887 /**
888  * gst_rtsp_media_is_shared:
889  * @media: a #GstRTSPMedia
890  *
891  * Check if the pipeline for @media can be shared between multiple clients.
892  *
893  * Returns: %TRUE if the media can be shared between clients.
894  */
895 gboolean
896 gst_rtsp_media_is_shared (GstRTSPMedia * media)
897 {
898   GstRTSPMediaPrivate *priv;
899   gboolean res;
900
901   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
902
903   priv = media->priv;
904
905   g_mutex_lock (&priv->lock);
906   res = priv->shared;
907   g_mutex_unlock (&priv->lock);
908
909   return res;
910 }
911
912 /**
913  * gst_rtsp_media_set_reusable:
914  * @media: a #GstRTSPMedia
915  * @reusable: the new value
916  *
917  * Set or unset if the pipeline for @media can be reused after the pipeline has
918  * been unprepared.
919  */
920 void
921 gst_rtsp_media_set_reusable (GstRTSPMedia * media, gboolean reusable)
922 {
923   GstRTSPMediaPrivate *priv;
924
925   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
926
927   priv = media->priv;
928
929   g_mutex_lock (&priv->lock);
930   priv->reusable = reusable;
931   g_mutex_unlock (&priv->lock);
932 }
933
934 /**
935  * gst_rtsp_media_is_reusable:
936  * @media: a #GstRTSPMedia
937  *
938  * Check if the pipeline for @media can be reused after an unprepare.
939  *
940  * Returns: %TRUE if the media can be reused
941  */
942 gboolean
943 gst_rtsp_media_is_reusable (GstRTSPMedia * media)
944 {
945   GstRTSPMediaPrivate *priv;
946   gboolean res;
947
948   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
949
950   priv = media->priv;
951
952   g_mutex_lock (&priv->lock);
953   res = priv->reusable;
954   g_mutex_unlock (&priv->lock);
955
956   return res;
957 }
958
959 static void
960 do_set_profiles (GstRTSPStream * stream, GstRTSPProfile * profiles)
961 {
962   gst_rtsp_stream_set_profiles (stream, *profiles);
963 }
964
965 /**
966  * gst_rtsp_media_set_profiles:
967  * @media: a #GstRTSPMedia
968  * @profiles: the new flags
969  *
970  * Configure the allowed lower transport for @media.
971  */
972 void
973 gst_rtsp_media_set_profiles (GstRTSPMedia * media, GstRTSPProfile profiles)
974 {
975   GstRTSPMediaPrivate *priv;
976
977   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
978
979   priv = media->priv;
980
981   g_mutex_lock (&priv->lock);
982   priv->profiles = profiles;
983   g_ptr_array_foreach (priv->streams, (GFunc) do_set_profiles, &profiles);
984   g_mutex_unlock (&priv->lock);
985 }
986
987 /**
988  * gst_rtsp_media_get_profiles:
989  * @media: a #GstRTSPMedia
990  *
991  * Get the allowed profiles of @media.
992  *
993  * Returns: a #GstRTSPProfile
994  */
995 GstRTSPProfile
996 gst_rtsp_media_get_profiles (GstRTSPMedia * media)
997 {
998   GstRTSPMediaPrivate *priv;
999   GstRTSPProfile res;
1000
1001   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), GST_RTSP_PROFILE_UNKNOWN);
1002
1003   priv = media->priv;
1004
1005   g_mutex_lock (&priv->lock);
1006   res = priv->profiles;
1007   g_mutex_unlock (&priv->lock);
1008
1009   return res;
1010 }
1011
1012 static void
1013 do_set_protocols (GstRTSPStream * stream, GstRTSPLowerTrans * protocols)
1014 {
1015   gst_rtsp_stream_set_protocols (stream, *protocols);
1016 }
1017
1018 /**
1019  * gst_rtsp_media_set_protocols:
1020  * @media: a #GstRTSPMedia
1021  * @protocols: the new flags
1022  *
1023  * Configure the allowed lower transport for @media.
1024  */
1025 void
1026 gst_rtsp_media_set_protocols (GstRTSPMedia * media, GstRTSPLowerTrans protocols)
1027 {
1028   GstRTSPMediaPrivate *priv;
1029
1030   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
1031
1032   priv = media->priv;
1033
1034   g_mutex_lock (&priv->lock);
1035   priv->protocols = protocols;
1036   g_ptr_array_foreach (priv->streams, (GFunc) do_set_protocols, &protocols);
1037   g_mutex_unlock (&priv->lock);
1038 }
1039
1040 /**
1041  * gst_rtsp_media_get_protocols:
1042  * @media: a #GstRTSPMedia
1043  *
1044  * Get the allowed protocols of @media.
1045  *
1046  * Returns: a #GstRTSPLowerTrans
1047  */
1048 GstRTSPLowerTrans
1049 gst_rtsp_media_get_protocols (GstRTSPMedia * media)
1050 {
1051   GstRTSPMediaPrivate *priv;
1052   GstRTSPLowerTrans res;
1053
1054   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media),
1055       GST_RTSP_LOWER_TRANS_UNKNOWN);
1056
1057   priv = media->priv;
1058
1059   g_mutex_lock (&priv->lock);
1060   res = priv->protocols;
1061   g_mutex_unlock (&priv->lock);
1062
1063   return res;
1064 }
1065
1066 /**
1067  * gst_rtsp_media_set_eos_shutdown:
1068  * @media: a #GstRTSPMedia
1069  * @eos_shutdown: the new value
1070  *
1071  * Set or unset if an EOS event will be sent to the pipeline for @media before
1072  * it is unprepared.
1073  */
1074 void
1075 gst_rtsp_media_set_eos_shutdown (GstRTSPMedia * media, gboolean eos_shutdown)
1076 {
1077   GstRTSPMediaPrivate *priv;
1078
1079   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
1080
1081   priv = media->priv;
1082
1083   g_mutex_lock (&priv->lock);
1084   priv->eos_shutdown = eos_shutdown;
1085   g_mutex_unlock (&priv->lock);
1086 }
1087
1088 /**
1089  * gst_rtsp_media_is_eos_shutdown:
1090  * @media: a #GstRTSPMedia
1091  *
1092  * Check if the pipeline for @media will send an EOS down the pipeline before
1093  * unpreparing.
1094  *
1095  * Returns: %TRUE if the media will send EOS before unpreparing.
1096  */
1097 gboolean
1098 gst_rtsp_media_is_eos_shutdown (GstRTSPMedia * media)
1099 {
1100   GstRTSPMediaPrivate *priv;
1101   gboolean res;
1102
1103   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
1104
1105   priv = media->priv;
1106
1107   g_mutex_lock (&priv->lock);
1108   res = priv->eos_shutdown;
1109   g_mutex_unlock (&priv->lock);
1110
1111   return res;
1112 }
1113
1114 /**
1115  * gst_rtsp_media_set_buffer_size:
1116  * @media: a #GstRTSPMedia
1117  * @size: the new value
1118  *
1119  * Set the kernel UDP buffer size.
1120  */
1121 void
1122 gst_rtsp_media_set_buffer_size (GstRTSPMedia * media, guint size)
1123 {
1124   GstRTSPMediaPrivate *priv;
1125
1126   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
1127
1128   GST_LOG_OBJECT (media, "set buffer size %u", size);
1129
1130   priv = media->priv;
1131
1132   g_mutex_lock (&priv->lock);
1133   priv->buffer_size = size;
1134   g_mutex_unlock (&priv->lock);
1135 }
1136
1137 /**
1138  * gst_rtsp_media_get_buffer_size:
1139  * @media: a #GstRTSPMedia
1140  *
1141  * Get the kernel UDP buffer size.
1142  *
1143  * Returns: the kernel UDP buffer size.
1144  */
1145 guint
1146 gst_rtsp_media_get_buffer_size (GstRTSPMedia * media)
1147 {
1148   GstRTSPMediaPrivate *priv;
1149   guint res;
1150
1151   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
1152
1153   priv = media->priv;
1154
1155   g_mutex_unlock (&priv->lock);
1156   res = priv->buffer_size;
1157   g_mutex_unlock (&priv->lock);
1158
1159   return res;
1160 }
1161
1162 /**
1163  * gst_rtsp_media_set_retransmission_time:
1164  * @media: a #GstRTSPMedia
1165  * @time: the new value
1166  *
1167  * Set the amount of time to store retransmission packets.
1168  */
1169 void
1170 gst_rtsp_media_set_retransmission_time (GstRTSPMedia * media, GstClockTime time)
1171 {
1172   GstRTSPMediaPrivate *priv;
1173   guint i;
1174
1175   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
1176
1177   GST_LOG_OBJECT (media, "set retransmission time %" G_GUINT64_FORMAT, time);
1178
1179   priv = media->priv;
1180
1181   g_mutex_lock (&priv->lock);
1182   priv->rtx_time = time;
1183   for (i = 0; i < priv->streams->len; i++) {
1184     GstRTSPStream *stream = g_ptr_array_index (priv->streams, i);
1185
1186     gst_rtsp_stream_set_retransmission_time (stream, time);
1187   }
1188
1189   if (priv->rtpbin)
1190     g_object_set (priv->rtpbin, "do-retransmission", time > 0, NULL);
1191   g_mutex_unlock (&priv->lock);
1192 }
1193
1194 /**
1195  * gst_rtsp_media_get_retransmission_time:
1196  * @media: a #GstRTSPMedia
1197  *
1198  * Get the amount of time to store retransmission data.
1199  *
1200  * Returns: the amount of time to store retransmission data.
1201  */
1202 GstClockTime
1203 gst_rtsp_media_get_retransmission_time (GstRTSPMedia * media)
1204 {
1205   GstRTSPMediaPrivate *priv;
1206   GstClockTime res;
1207
1208   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
1209
1210   priv = media->priv;
1211
1212   g_mutex_unlock (&priv->lock);
1213   res = priv->rtx_time;
1214   g_mutex_unlock (&priv->lock);
1215
1216   return res;
1217 }
1218
1219 /**
1220  * gst_rtsp_media_set_latncy:
1221  * @media: a #GstRTSPMedia
1222  * @latency: latency in milliseconds
1223  *
1224  * Configure the latency used for receiving media.
1225  */
1226 void
1227 gst_rtsp_media_set_latency (GstRTSPMedia * media, guint latency)
1228 {
1229   GstRTSPMediaPrivate *priv;
1230
1231   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
1232
1233   GST_LOG_OBJECT (media, "set latency %ums", latency);
1234
1235   priv = media->priv;
1236
1237   g_mutex_lock (&priv->lock);
1238   priv->latency = latency;
1239   if (priv->rtpbin)
1240     g_object_set (priv->rtpbin, "latency", latency, NULL);
1241   g_mutex_unlock (&priv->lock);
1242 }
1243
1244 /**
1245  * gst_rtsp_media_get_latency:
1246  * @media: a #GstRTSPMedia
1247  *
1248  * Get the latency that is used for receiving media.
1249  *
1250  * Returns: latency in milliseconds
1251  */
1252 guint
1253 gst_rtsp_media_get_latency (GstRTSPMedia * media)
1254 {
1255   GstRTSPMediaPrivate *priv;
1256   guint res;
1257
1258   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
1259
1260   priv = media->priv;
1261
1262   g_mutex_unlock (&priv->lock);
1263   res = priv->latency;
1264   g_mutex_unlock (&priv->lock);
1265
1266   return res;
1267 }
1268
1269 /**
1270  * gst_rtsp_media_use_time_provider:
1271  * @media: a #GstRTSPMedia
1272  * @time_provider: if a #GstNetTimeProvider should be used
1273  *
1274  * Set @media to provide a #GstNetTimeProvider.
1275  */
1276 void
1277 gst_rtsp_media_use_time_provider (GstRTSPMedia * media, gboolean time_provider)
1278 {
1279   GstRTSPMediaPrivate *priv;
1280
1281   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
1282
1283   priv = media->priv;
1284
1285   g_mutex_lock (&priv->lock);
1286   priv->time_provider = time_provider;
1287   g_mutex_unlock (&priv->lock);
1288 }
1289
1290 /**
1291  * gst_rtsp_media_is_time_provider:
1292  * @media: a #GstRTSPMedia
1293  *
1294  * Check if @media can provide a #GstNetTimeProvider for its pipeline clock.
1295  *
1296  * Use gst_rtsp_media_get_time_provider() to get the network clock.
1297  *
1298  * Returns: %TRUE if @media can provide a #GstNetTimeProvider.
1299  */
1300 gboolean
1301 gst_rtsp_media_is_time_provider (GstRTSPMedia * media)
1302 {
1303   GstRTSPMediaPrivate *priv;
1304   gboolean res;
1305
1306   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
1307
1308   priv = media->priv;
1309
1310   g_mutex_unlock (&priv->lock);
1311   res = priv->time_provider;
1312   g_mutex_unlock (&priv->lock);
1313
1314   return res;
1315 }
1316
1317 /**
1318  * gst_rtsp_media_set_address_pool:
1319  * @media: a #GstRTSPMedia
1320  * @pool: (transfer none): a #GstRTSPAddressPool
1321  *
1322  * configure @pool to be used as the address pool of @media.
1323  */
1324 void
1325 gst_rtsp_media_set_address_pool (GstRTSPMedia * media,
1326     GstRTSPAddressPool * pool)
1327 {
1328   GstRTSPMediaPrivate *priv;
1329   GstRTSPAddressPool *old;
1330
1331   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
1332
1333   priv = media->priv;
1334
1335   GST_LOG_OBJECT (media, "set address pool %p", pool);
1336
1337   g_mutex_lock (&priv->lock);
1338   if ((old = priv->pool) != pool)
1339     priv->pool = pool ? g_object_ref (pool) : NULL;
1340   else
1341     old = NULL;
1342   g_ptr_array_foreach (priv->streams, (GFunc) gst_rtsp_stream_set_address_pool,
1343       pool);
1344   g_mutex_unlock (&priv->lock);
1345
1346   if (old)
1347     g_object_unref (old);
1348 }
1349
1350 /**
1351  * gst_rtsp_media_get_address_pool:
1352  * @media: a #GstRTSPMedia
1353  *
1354  * Get the #GstRTSPAddressPool used as the address pool of @media.
1355  *
1356  * Returns: (transfer full): the #GstRTSPAddressPool of @media. g_object_unref() after
1357  * usage.
1358  */
1359 GstRTSPAddressPool *
1360 gst_rtsp_media_get_address_pool (GstRTSPMedia * media)
1361 {
1362   GstRTSPMediaPrivate *priv;
1363   GstRTSPAddressPool *result;
1364
1365   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), NULL);
1366
1367   priv = media->priv;
1368
1369   g_mutex_lock (&priv->lock);
1370   if ((result = priv->pool))
1371     g_object_ref (result);
1372   g_mutex_unlock (&priv->lock);
1373
1374   return result;
1375 }
1376
1377 static GList *
1378 _find_payload_types (GstRTSPMedia * media)
1379 {
1380   gint i, n;
1381   GQueue queue = G_QUEUE_INIT;
1382
1383   n = media->priv->streams->len;
1384   for (i = 0; i < n; i++) {
1385     GstRTSPStream *stream = g_ptr_array_index (media->priv->streams, i);
1386     guint pt = gst_rtsp_stream_get_pt (stream);
1387
1388     g_queue_push_tail (&queue, GUINT_TO_POINTER (pt));
1389   }
1390
1391   return queue.head;
1392 }
1393
1394 static guint
1395 _next_available_pt (GList * payloads)
1396 {
1397   guint i;
1398
1399   for (i = 96; i <= 127; i++) {
1400     GList *iter = g_list_find (payloads, GINT_TO_POINTER (i));
1401     if (!iter)
1402       return GPOINTER_TO_UINT (i);
1403   }
1404
1405   return 0;
1406 }
1407
1408 /**
1409  * gst_rtsp_media_collect_streams:
1410  * @media: a #GstRTSPMedia
1411  *
1412  * Find all payloader elements, they should be named pay\%d in the
1413  * element of @media, and create #GstRTSPStreams for them.
1414  *
1415  * Collect all dynamic elements, named dynpay\%d, and add them to
1416  * the list of dynamic elements.
1417  *
1418  * Find all depayloader elements, they should be named depay\%d in the
1419  * element of @media, and create #GstRTSPStreams for them.
1420  */
1421 void
1422 gst_rtsp_media_collect_streams (GstRTSPMedia * media)
1423 {
1424   GstRTSPMediaPrivate *priv;
1425   GstElement *element, *elem;
1426   GstPad *pad;
1427   gint i;
1428   gboolean have_elem;
1429   GstRTSPTransportMode mode = 0;
1430
1431   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
1432
1433   priv = media->priv;
1434   element = priv->element;
1435
1436   have_elem = TRUE;
1437   for (i = 0; have_elem; i++) {
1438     gchar *name;
1439
1440     have_elem = FALSE;
1441
1442     name = g_strdup_printf ("pay%d", i);
1443     if ((elem = gst_bin_get_by_name (GST_BIN (element), name))) {
1444       GST_INFO ("found stream %d with payloader %p", i, elem);
1445
1446       /* take the pad of the payloader */
1447       pad = gst_element_get_static_pad (elem, "src");
1448       /* create the stream */
1449       gst_rtsp_media_create_stream (media, elem, pad);
1450       gst_object_unref (pad);
1451       gst_object_unref (elem);
1452
1453       have_elem = TRUE;
1454       mode |= GST_RTSP_TRANSPORT_MODE_PLAY;
1455     }
1456     g_free (name);
1457
1458     name = g_strdup_printf ("dynpay%d", i);
1459     if ((elem = gst_bin_get_by_name (GST_BIN (element), name))) {
1460       /* a stream that will dynamically create pads to provide RTP packets */
1461       GST_INFO ("found dynamic element %d, %p", i, elem);
1462
1463       g_mutex_lock (&priv->lock);
1464       priv->dynamic = g_list_prepend (priv->dynamic, elem);
1465       g_mutex_unlock (&priv->lock);
1466
1467       have_elem = TRUE;
1468       mode |= GST_RTSP_TRANSPORT_MODE_PLAY;
1469     }
1470     g_free (name);
1471
1472     name = g_strdup_printf ("depay%d", i);
1473     if ((elem = gst_bin_get_by_name (GST_BIN (element), name))) {
1474       GST_INFO ("found stream %d with depayloader %p", i, elem);
1475
1476       /* take the pad of the payloader */
1477       pad = gst_element_get_static_pad (elem, "sink");
1478       /* create the stream */
1479       gst_rtsp_media_create_stream (media, elem, pad);
1480       gst_object_unref (pad);
1481       gst_object_unref (elem);
1482
1483       have_elem = TRUE;
1484       mode |= GST_RTSP_TRANSPORT_MODE_RECORD;
1485     }
1486     g_free (name);
1487   }
1488
1489   if (have_elem) {
1490     if (priv->transport_mode != mode)
1491       GST_WARNING ("found different mode than expected (0x%02x != 0x%02d)",
1492           priv->transport_mode, mode);
1493   }
1494 }
1495
1496 /**
1497  * gst_rtsp_media_create_stream:
1498  * @media: a #GstRTSPMedia
1499  * @payloader: a #GstElement
1500  * @pad: a #GstPad
1501  *
1502  * Create a new stream in @media that provides RTP data on @pad.
1503  * @pad should be a pad of an element inside @media->element.
1504  *
1505  * Returns: (transfer none): a new #GstRTSPStream that remains valid for as long
1506  * as @media exists.
1507  */
1508 GstRTSPStream *
1509 gst_rtsp_media_create_stream (GstRTSPMedia * media, GstElement * payloader,
1510     GstPad * pad)
1511 {
1512   GstRTSPMediaPrivate *priv;
1513   GstRTSPStream *stream;
1514   GstPad *ghostpad;
1515   gchar *name;
1516   gint idx;
1517
1518   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), NULL);
1519   g_return_val_if_fail (GST_IS_ELEMENT (payloader), NULL);
1520   g_return_val_if_fail (GST_IS_PAD (pad), NULL);
1521
1522   priv = media->priv;
1523
1524   g_mutex_lock (&priv->lock);
1525   idx = priv->streams->len;
1526
1527   GST_DEBUG ("media %p: creating stream with index %d", media, idx);
1528
1529   if (GST_PAD_IS_SRC (pad))
1530     name = g_strdup_printf ("src_%u", idx);
1531   else
1532     name = g_strdup_printf ("sink_%u", idx);
1533
1534   ghostpad = gst_ghost_pad_new (name, pad);
1535   gst_pad_set_active (ghostpad, TRUE);
1536   gst_element_add_pad (priv->element, ghostpad);
1537   g_free (name);
1538
1539   stream = gst_rtsp_stream_new (idx, payloader, ghostpad);
1540   if (priv->pool)
1541     gst_rtsp_stream_set_address_pool (stream, priv->pool);
1542   gst_rtsp_stream_set_profiles (stream, priv->profiles);
1543   gst_rtsp_stream_set_protocols (stream, priv->protocols);
1544   gst_rtsp_stream_set_retransmission_time (stream, priv->rtx_time);
1545
1546   g_ptr_array_add (priv->streams, stream);
1547
1548   if (GST_PAD_IS_SRC (pad)) {
1549     gint i, n;
1550
1551     if (priv->payloads)
1552       g_list_free (priv->payloads);
1553     priv->payloads = _find_payload_types (media);
1554
1555     n = priv->streams->len;
1556     for (i = 0; i < n; i++) {
1557       GstRTSPStream *stream = g_ptr_array_index (priv->streams, i);
1558       guint rtx_pt = _next_available_pt (priv->payloads);
1559
1560       if (rtx_pt == 0) {
1561         GST_WARNING ("Ran out of space of dynamic payload types");
1562         break;
1563       }
1564
1565       gst_rtsp_stream_set_retransmission_pt (stream, rtx_pt);
1566
1567       priv->payloads =
1568           g_list_append (priv->payloads, GUINT_TO_POINTER (rtx_pt));
1569     }
1570   }
1571   g_mutex_unlock (&priv->lock);
1572
1573   g_signal_emit (media, gst_rtsp_media_signals[SIGNAL_NEW_STREAM], 0, stream,
1574       NULL);
1575
1576   return stream;
1577 }
1578
1579 static void
1580 gst_rtsp_media_remove_stream (GstRTSPMedia * media, GstRTSPStream * stream)
1581 {
1582   GstRTSPMediaPrivate *priv;
1583   GstPad *srcpad;
1584
1585   priv = media->priv;
1586
1587   g_mutex_lock (&priv->lock);
1588   /* remove the ghostpad */
1589   srcpad = gst_rtsp_stream_get_srcpad (stream);
1590   gst_element_remove_pad (priv->element, srcpad);
1591   gst_object_unref (srcpad);
1592   /* now remove the stream */
1593   g_object_ref (stream);
1594   g_ptr_array_remove (priv->streams, stream);
1595   g_mutex_unlock (&priv->lock);
1596
1597   g_signal_emit (media, gst_rtsp_media_signals[SIGNAL_REMOVED_STREAM], 0,
1598       stream, NULL);
1599
1600   g_object_unref (stream);
1601 }
1602
1603 /**
1604  * gst_rtsp_media_n_streams:
1605  * @media: a #GstRTSPMedia
1606  *
1607  * Get the number of streams in this media.
1608  *
1609  * Returns: The number of streams.
1610  */
1611 guint
1612 gst_rtsp_media_n_streams (GstRTSPMedia * media)
1613 {
1614   GstRTSPMediaPrivate *priv;
1615   guint res;
1616
1617   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), 0);
1618
1619   priv = media->priv;
1620
1621   g_mutex_lock (&priv->lock);
1622   res = priv->streams->len;
1623   g_mutex_unlock (&priv->lock);
1624
1625   return res;
1626 }
1627
1628 /**
1629  * gst_rtsp_media_get_stream:
1630  * @media: a #GstRTSPMedia
1631  * @idx: the stream index
1632  *
1633  * Retrieve the stream with index @idx from @media.
1634  *
1635  * Returns: (nullable) (transfer none): the #GstRTSPStream at index
1636  * @idx or %NULL when a stream with that index did not exist.
1637  */
1638 GstRTSPStream *
1639 gst_rtsp_media_get_stream (GstRTSPMedia * media, guint idx)
1640 {
1641   GstRTSPMediaPrivate *priv;
1642   GstRTSPStream *res;
1643
1644   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), NULL);
1645
1646   priv = media->priv;
1647
1648   g_mutex_lock (&priv->lock);
1649   if (idx < priv->streams->len)
1650     res = g_ptr_array_index (priv->streams, idx);
1651   else
1652     res = NULL;
1653   g_mutex_unlock (&priv->lock);
1654
1655   return res;
1656 }
1657
1658 /**
1659  * gst_rtsp_media_find_stream:
1660  * @media: a #GstRTSPMedia
1661  * @control: the control of the stream
1662  *
1663  * Find a stream in @media with @control as the control uri.
1664  *
1665  * Returns: (nullable) (transfer none): the #GstRTSPStream with
1666  * control uri @control or %NULL when a stream with that control did
1667  * not exist.
1668  */
1669 GstRTSPStream *
1670 gst_rtsp_media_find_stream (GstRTSPMedia * media, const gchar * control)
1671 {
1672   GstRTSPMediaPrivate *priv;
1673   GstRTSPStream *res;
1674   gint i;
1675
1676   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), NULL);
1677   g_return_val_if_fail (control != NULL, NULL);
1678
1679   priv = media->priv;
1680
1681   res = NULL;
1682
1683   g_mutex_lock (&priv->lock);
1684   for (i = 0; i < priv->streams->len; i++) {
1685     GstRTSPStream *test;
1686
1687     test = g_ptr_array_index (priv->streams, i);
1688     if (gst_rtsp_stream_has_control (test, control)) {
1689       res = test;
1690       break;
1691     }
1692   }
1693   g_mutex_unlock (&priv->lock);
1694
1695   return res;
1696 }
1697
1698 /* called with state-lock */
1699 static gboolean
1700 default_convert_range (GstRTSPMedia * media, GstRTSPTimeRange * range,
1701     GstRTSPRangeUnit unit)
1702 {
1703   return gst_rtsp_range_convert_units (range, unit);
1704 }
1705
1706 /**
1707  * gst_rtsp_media_get_range_string:
1708  * @media: a #GstRTSPMedia
1709  * @play: for the PLAY request
1710  * @unit: the unit to use for the string
1711  *
1712  * Get the current range as a string. @media must be prepared with
1713  * gst_rtsp_media_prepare ().
1714  *
1715  * Returns: (transfer full): The range as a string, g_free() after usage.
1716  */
1717 gchar *
1718 gst_rtsp_media_get_range_string (GstRTSPMedia * media, gboolean play,
1719     GstRTSPRangeUnit unit)
1720 {
1721   GstRTSPMediaClass *klass;
1722   GstRTSPMediaPrivate *priv;
1723   gchar *result;
1724   GstRTSPTimeRange range;
1725
1726   klass = GST_RTSP_MEDIA_GET_CLASS (media);
1727   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), NULL);
1728   g_return_val_if_fail (klass->convert_range != NULL, FALSE);
1729
1730   priv = media->priv;
1731
1732   g_rec_mutex_lock (&priv->state_lock);
1733   if (priv->status != GST_RTSP_MEDIA_STATUS_PREPARED &&
1734       priv->status != GST_RTSP_MEDIA_STATUS_SUSPENDED)
1735     goto not_prepared;
1736
1737   g_mutex_lock (&priv->lock);
1738
1739   /* Update the range value with current position/duration */
1740   collect_media_stats (media);
1741
1742   /* make copy */
1743   range = priv->range;
1744
1745   if (!play && priv->n_active > 0) {
1746     range.min.type = GST_RTSP_TIME_NOW;
1747     range.min.seconds = -1;
1748   }
1749   g_mutex_unlock (&priv->lock);
1750   g_rec_mutex_unlock (&priv->state_lock);
1751
1752   if (!klass->convert_range (media, &range, unit))
1753     goto conversion_failed;
1754
1755   result = gst_rtsp_range_to_string (&range);
1756
1757   return result;
1758
1759   /* ERRORS */
1760 not_prepared:
1761   {
1762     GST_WARNING ("media %p was not prepared", media);
1763     g_rec_mutex_unlock (&priv->state_lock);
1764     return NULL;
1765   }
1766 conversion_failed:
1767   {
1768     GST_WARNING ("range conversion to unit %d failed", unit);
1769     return NULL;
1770   }
1771 }
1772
1773 static void
1774 stream_update_blocked (GstRTSPStream * stream, GstRTSPMedia * media)
1775 {
1776   gst_rtsp_stream_set_blocked (stream, media->priv->blocked);
1777 }
1778
1779 static void
1780 media_streams_set_blocked (GstRTSPMedia * media, gboolean blocked)
1781 {
1782   GstRTSPMediaPrivate *priv = media->priv;
1783
1784   GST_DEBUG ("media %p set blocked %d", media, blocked);
1785   priv->blocked = blocked;
1786   g_ptr_array_foreach (priv->streams, (GFunc) stream_update_blocked, media);
1787 }
1788
1789 static void
1790 gst_rtsp_media_set_status (GstRTSPMedia * media, GstRTSPMediaStatus status)
1791 {
1792   GstRTSPMediaPrivate *priv = media->priv;
1793
1794   g_mutex_lock (&priv->lock);
1795   priv->status = status;
1796   GST_DEBUG ("setting new status to %d", status);
1797   g_cond_broadcast (&priv->cond);
1798   g_mutex_unlock (&priv->lock);
1799 }
1800
1801 /**
1802  * gst_rtsp_media_get_status:
1803  * @media: a #GstRTSPMedia
1804  *
1805  * Get the status of @media. When @media is busy preparing, this function waits
1806  * until @media is prepared or in error.
1807  *
1808  * Returns: the status of @media.
1809  */
1810 GstRTSPMediaStatus
1811 gst_rtsp_media_get_status (GstRTSPMedia * media)
1812 {
1813   GstRTSPMediaPrivate *priv = media->priv;
1814   GstRTSPMediaStatus result;
1815   gint64 end_time;
1816
1817   g_mutex_lock (&priv->lock);
1818   end_time = g_get_monotonic_time () + 20 * G_TIME_SPAN_SECOND;
1819   /* while we are preparing, wait */
1820   while (priv->status == GST_RTSP_MEDIA_STATUS_PREPARING) {
1821     GST_DEBUG ("waiting for status change");
1822     if (!g_cond_wait_until (&priv->cond, &priv->lock, end_time)) {
1823       GST_DEBUG ("timeout, assuming error status");
1824       priv->status = GST_RTSP_MEDIA_STATUS_ERROR;
1825     }
1826   }
1827   /* could be success or error */
1828   result = priv->status;
1829   GST_DEBUG ("got status %d", result);
1830   g_mutex_unlock (&priv->lock);
1831
1832   return result;
1833 }
1834
1835 /**
1836  * gst_rtsp_media_seek:
1837  * @media: a #GstRTSPMedia
1838  * @range: (transfer none): a #GstRTSPTimeRange
1839  *
1840  * Seek the pipeline of @media to @range. @media must be prepared with
1841  * gst_rtsp_media_prepare().
1842  *
1843  * Returns: %TRUE on success.
1844  */
1845 gboolean
1846 gst_rtsp_media_seek (GstRTSPMedia * media, GstRTSPTimeRange * range)
1847 {
1848   GstRTSPMediaClass *klass;
1849   GstRTSPMediaPrivate *priv;
1850   gboolean res;
1851   GstClockTime start, stop;
1852   GstSeekType start_type, stop_type;
1853   GstQuery *query;
1854
1855   klass = GST_RTSP_MEDIA_GET_CLASS (media);
1856
1857   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
1858   g_return_val_if_fail (range != NULL, FALSE);
1859   g_return_val_if_fail (klass->convert_range != NULL, FALSE);
1860
1861   priv = media->priv;
1862
1863   g_rec_mutex_lock (&priv->state_lock);
1864   if (priv->status != GST_RTSP_MEDIA_STATUS_PREPARED)
1865     goto not_prepared;
1866
1867   /* Update the seekable state of the pipeline in case it changed */
1868   if ((priv->transport_mode & GST_RTSP_TRANSPORT_MODE_RECORD)) {
1869     /* TODO: Seeking for RECORD? */
1870     priv->seekable = FALSE;
1871   } else {
1872     query = gst_query_new_seeking (GST_FORMAT_TIME);
1873     if (gst_element_query (priv->pipeline, query)) {
1874       GstFormat format;
1875       gboolean seekable;
1876       gint64 start, end;
1877
1878       gst_query_parse_seeking (query, &format, &seekable, &start, &end);
1879       priv->seekable = seekable;
1880     }
1881     gst_query_unref (query);
1882   }
1883
1884   if (!priv->seekable)
1885     goto not_seekable;
1886
1887   start_type = stop_type = GST_SEEK_TYPE_NONE;
1888
1889   if (!klass->convert_range (media, range, GST_RTSP_RANGE_NPT))
1890     goto not_supported;
1891   gst_rtsp_range_get_times (range, &start, &stop);
1892
1893   GST_INFO ("got %" GST_TIME_FORMAT " - %" GST_TIME_FORMAT,
1894       GST_TIME_ARGS (start), GST_TIME_ARGS (stop));
1895   GST_INFO ("current %" GST_TIME_FORMAT " - %" GST_TIME_FORMAT,
1896       GST_TIME_ARGS (priv->range_start), GST_TIME_ARGS (priv->range_stop));
1897
1898   if (start != GST_CLOCK_TIME_NONE)
1899     start_type = GST_SEEK_TYPE_SET;
1900
1901   if (priv->range_stop == stop)
1902     stop = GST_CLOCK_TIME_NONE;
1903   else if (stop != GST_CLOCK_TIME_NONE)
1904     stop_type = GST_SEEK_TYPE_SET;
1905
1906   if (start != GST_CLOCK_TIME_NONE || stop != GST_CLOCK_TIME_NONE) {
1907     GstSeekFlags flags;
1908
1909     GST_INFO ("seeking to %" GST_TIME_FORMAT " - %" GST_TIME_FORMAT,
1910         GST_TIME_ARGS (start), GST_TIME_ARGS (stop));
1911
1912     gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_PREPARING);
1913     if (priv->blocked)
1914       media_streams_set_blocked (media, TRUE);
1915
1916     /* depends on the current playing state of the pipeline. We might need to
1917      * queue this until we get EOS. */
1918     flags = GST_SEEK_FLAG_FLUSH;
1919
1920     /* if range start was not supplied we must continue from current position.
1921      * but since we're doing a flushing seek, let us query the current position
1922      * so we end up at exactly the same position after the seek. */
1923     if (range->min.type == GST_RTSP_TIME_END) { /* Yepp, that's right! */
1924       gint64 position;
1925       gboolean ret = FALSE;
1926
1927       if (klass->query_position)
1928         ret = klass->query_position (media, &position);
1929
1930       if (!ret) {
1931         GST_WARNING ("position query failed");
1932       } else {
1933         GST_DEBUG ("doing accurate seek to %" GST_TIME_FORMAT,
1934             GST_TIME_ARGS (position));
1935         start = position;
1936         start_type = GST_SEEK_TYPE_SET;
1937         flags |= GST_SEEK_FLAG_ACCURATE;
1938       }
1939     } else {
1940       /* only set keyframe flag when modifying start */
1941       if (start_type != GST_SEEK_TYPE_NONE)
1942         flags |= GST_SEEK_FLAG_KEY_UNIT;
1943     }
1944
1945     /* FIXME, we only do forwards playback, no trick modes yet */
1946     res = gst_element_seek (priv->pipeline, 1.0, GST_FORMAT_TIME,
1947         flags, start_type, start, stop_type, stop);
1948
1949     /* and block for the seek to complete */
1950     GST_INFO ("done seeking %d", res);
1951     g_rec_mutex_unlock (&priv->state_lock);
1952
1953     /* wait until pipeline is prerolled again, this will also collect stats */
1954     if (!wait_preroll (media))
1955       goto preroll_failed;
1956
1957     g_rec_mutex_lock (&priv->state_lock);
1958     GST_INFO ("prerolled again");
1959   } else {
1960     GST_INFO ("no seek needed");
1961     res = TRUE;
1962   }
1963   g_rec_mutex_unlock (&priv->state_lock);
1964
1965   return res;
1966
1967   /* ERRORS */
1968 not_prepared:
1969   {
1970     g_rec_mutex_unlock (&priv->state_lock);
1971     GST_INFO ("media %p is not prepared", media);
1972     return FALSE;
1973   }
1974 not_seekable:
1975   {
1976     g_rec_mutex_unlock (&priv->state_lock);
1977     GST_INFO ("pipeline is not seekable");
1978     return FALSE;
1979   }
1980 not_supported:
1981   {
1982     g_rec_mutex_unlock (&priv->state_lock);
1983     GST_WARNING ("conversion to npt not supported");
1984     return FALSE;
1985   }
1986 preroll_failed:
1987   {
1988     GST_WARNING ("failed to preroll after seek");
1989     return FALSE;
1990   }
1991 }
1992
1993 static void
1994 stream_collect_blocking (GstRTSPStream * stream, gboolean * blocked)
1995 {
1996   *blocked &= gst_rtsp_stream_is_blocking (stream);
1997 }
1998
1999 static gboolean
2000 media_streams_blocking (GstRTSPMedia * media)
2001 {
2002   gboolean blocking = TRUE;
2003
2004   g_ptr_array_foreach (media->priv->streams, (GFunc) stream_collect_blocking,
2005       &blocking);
2006
2007   return blocking;
2008 }
2009
2010 static GstStateChangeReturn
2011 set_state (GstRTSPMedia * media, GstState state)
2012 {
2013   GstRTSPMediaPrivate *priv = media->priv;
2014   GstStateChangeReturn ret;
2015
2016   GST_INFO ("set state to %s for media %p", gst_element_state_get_name (state),
2017       media);
2018   ret = gst_element_set_state (priv->pipeline, state);
2019
2020   return ret;
2021 }
2022
2023 static GstStateChangeReturn
2024 set_target_state (GstRTSPMedia * media, GstState state, gboolean do_state)
2025 {
2026   GstRTSPMediaPrivate *priv = media->priv;
2027   GstStateChangeReturn ret;
2028
2029   GST_INFO ("set target state to %s for media %p",
2030       gst_element_state_get_name (state), media);
2031   priv->target_state = state;
2032
2033   g_signal_emit (media, gst_rtsp_media_signals[SIGNAL_TARGET_STATE], 0,
2034       priv->target_state, NULL);
2035
2036   if (do_state)
2037     ret = set_state (media, state);
2038   else
2039     ret = GST_STATE_CHANGE_SUCCESS;
2040
2041   return ret;
2042 }
2043
2044 /* called with state-lock */
2045 static gboolean
2046 default_handle_message (GstRTSPMedia * media, GstMessage * message)
2047 {
2048   GstRTSPMediaPrivate *priv = media->priv;
2049   GstMessageType type;
2050
2051   type = GST_MESSAGE_TYPE (message);
2052
2053   switch (type) {
2054     case GST_MESSAGE_STATE_CHANGED:
2055     {
2056       GstState old, new, pending;
2057
2058       if (GST_MESSAGE_SRC (message) != GST_OBJECT (priv->pipeline))
2059         break;
2060
2061       gst_message_parse_state_changed (message, &old, &new, &pending);
2062
2063       GST_DEBUG ("%p: went from %s to %s (pending %s)", media,
2064           gst_element_state_get_name (old), gst_element_state_get_name (new),
2065           gst_element_state_get_name (pending));
2066       if ((priv->transport_mode & GST_RTSP_TRANSPORT_MODE_RECORD)
2067           && old == GST_STATE_READY && new == GST_STATE_PAUSED) {
2068         GST_INFO ("%p: went to PAUSED, prepared now", media);
2069         collect_media_stats (media);
2070
2071         if (priv->status == GST_RTSP_MEDIA_STATUS_PREPARING)
2072           gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_PREPARED);
2073       }
2074
2075       break;
2076     }
2077     case GST_MESSAGE_BUFFERING:
2078     {
2079       gint percent;
2080
2081       gst_message_parse_buffering (message, &percent);
2082
2083       /* no state management needed for live pipelines */
2084       if (priv->is_live)
2085         break;
2086
2087       if (percent == 100) {
2088         /* a 100% message means buffering is done */
2089         priv->buffering = FALSE;
2090         /* if the desired state is playing, go back */
2091         if (priv->target_state == GST_STATE_PLAYING) {
2092           GST_INFO ("Buffering done, setting pipeline to PLAYING");
2093           set_state (media, GST_STATE_PLAYING);
2094         } else {
2095           GST_INFO ("Buffering done");
2096         }
2097       } else {
2098         /* buffering busy */
2099         if (priv->buffering == FALSE) {
2100           if (priv->target_state == GST_STATE_PLAYING) {
2101             /* we were not buffering but PLAYING, PAUSE  the pipeline. */
2102             GST_INFO ("Buffering, setting pipeline to PAUSED ...");
2103             set_state (media, GST_STATE_PAUSED);
2104           } else {
2105             GST_INFO ("Buffering ...");
2106           }
2107         }
2108         priv->buffering = TRUE;
2109       }
2110       break;
2111     }
2112     case GST_MESSAGE_LATENCY:
2113     {
2114       gst_bin_recalculate_latency (GST_BIN_CAST (priv->pipeline));
2115       break;
2116     }
2117     case GST_MESSAGE_ERROR:
2118     {
2119       GError *gerror;
2120       gchar *debug;
2121
2122       gst_message_parse_error (message, &gerror, &debug);
2123       GST_WARNING ("%p: got error %s (%s)", media, gerror->message, debug);
2124       g_error_free (gerror);
2125       g_free (debug);
2126
2127       gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_ERROR);
2128       break;
2129     }
2130     case GST_MESSAGE_WARNING:
2131     {
2132       GError *gerror;
2133       gchar *debug;
2134
2135       gst_message_parse_warning (message, &gerror, &debug);
2136       GST_WARNING ("%p: got warning %s (%s)", media, gerror->message, debug);
2137       g_error_free (gerror);
2138       g_free (debug);
2139       break;
2140     }
2141     case GST_MESSAGE_ELEMENT:
2142     {
2143       const GstStructure *s;
2144
2145       s = gst_message_get_structure (message);
2146       if (gst_structure_has_name (s, "GstRTSPStreamBlocking")) {
2147         GST_DEBUG ("media received blocking message");
2148         if (priv->blocked && media_streams_blocking (media)) {
2149           GST_DEBUG ("media is blocking");
2150           collect_media_stats (media);
2151
2152           if (priv->status == GST_RTSP_MEDIA_STATUS_PREPARING)
2153             gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_PREPARED);
2154         }
2155       }
2156       break;
2157     }
2158     case GST_MESSAGE_STREAM_STATUS:
2159       break;
2160     case GST_MESSAGE_ASYNC_DONE:
2161       if (priv->adding) {
2162         /* when we are dynamically adding pads, the addition of the udpsrc will
2163          * temporarily produce ASYNC_DONE messages. We have to ignore them and
2164          * wait for the final ASYNC_DONE after everything prerolled */
2165         GST_INFO ("%p: ignoring ASYNC_DONE", media);
2166       } else {
2167         GST_INFO ("%p: got ASYNC_DONE", media);
2168         collect_media_stats (media);
2169
2170         if (priv->status == GST_RTSP_MEDIA_STATUS_PREPARING)
2171           gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_PREPARED);
2172       }
2173       break;
2174     case GST_MESSAGE_EOS:
2175       GST_INFO ("%p: got EOS", media);
2176
2177       if (priv->status == GST_RTSP_MEDIA_STATUS_UNPREPARING) {
2178         GST_DEBUG ("shutting down after EOS");
2179         finish_unprepare (media);
2180       }
2181       break;
2182     default:
2183       GST_INFO ("%p: got message type %d (%s)", media, type,
2184           gst_message_type_get_name (type));
2185       break;
2186   }
2187   return TRUE;
2188 }
2189
2190 static gboolean
2191 bus_message (GstBus * bus, GstMessage * message, GstRTSPMedia * media)
2192 {
2193   GstRTSPMediaPrivate *priv = media->priv;
2194   GstRTSPMediaClass *klass;
2195   gboolean ret;
2196
2197   klass = GST_RTSP_MEDIA_GET_CLASS (media);
2198
2199   g_rec_mutex_lock (&priv->state_lock);
2200   if (klass->handle_message)
2201     ret = klass->handle_message (media, message);
2202   else
2203     ret = FALSE;
2204   g_rec_mutex_unlock (&priv->state_lock);
2205
2206   return ret;
2207 }
2208
2209 static void
2210 watch_destroyed (GstRTSPMedia * media)
2211 {
2212   GST_DEBUG_OBJECT (media, "source destroyed");
2213   g_object_unref (media);
2214 }
2215
2216 static GstElement *
2217 find_payload_element (GstElement * payloader)
2218 {
2219   GstElement *pay = NULL;
2220
2221   if (GST_IS_BIN (payloader)) {
2222     GstIterator *iter;
2223     GValue item = { 0 };
2224
2225     iter = gst_bin_iterate_recurse (GST_BIN (payloader));
2226     while (gst_iterator_next (iter, &item) == GST_ITERATOR_OK) {
2227       GstElement *element = (GstElement *) g_value_get_object (&item);
2228       GstElementClass *eclass = GST_ELEMENT_GET_CLASS (element);
2229       const gchar *klass;
2230
2231       klass =
2232           gst_element_class_get_metadata (eclass, GST_ELEMENT_METADATA_KLASS);
2233       if (klass == NULL)
2234         continue;
2235
2236       if (strstr (klass, "Payloader") && strstr (klass, "RTP")) {
2237         pay = gst_object_ref (element);
2238         g_value_unset (&item);
2239         break;
2240       }
2241       g_value_unset (&item);
2242     }
2243     gst_iterator_free (iter);
2244   } else {
2245     pay = g_object_ref (payloader);
2246   }
2247
2248   return pay;
2249 }
2250
2251 /* called from streaming threads */
2252 static void
2253 pad_added_cb (GstElement * element, GstPad * pad, GstRTSPMedia * media)
2254 {
2255   GstRTSPMediaPrivate *priv = media->priv;
2256   GstRTSPStream *stream;
2257   GstElement *pay;
2258
2259   /* find the real payload element */
2260   pay = find_payload_element (element);
2261   stream = gst_rtsp_media_create_stream (media, pay, pad);
2262   gst_object_unref (pay);
2263
2264   GST_INFO ("pad added %s:%s, stream %p", GST_DEBUG_PAD_NAME (pad), stream);
2265
2266   g_rec_mutex_lock (&priv->state_lock);
2267   if (priv->status != GST_RTSP_MEDIA_STATUS_PREPARING)
2268     goto not_preparing;
2269
2270   g_object_set_data (G_OBJECT (pad), "gst-rtsp-dynpad-stream", stream);
2271
2272   /* we will be adding elements below that will cause ASYNC_DONE to be
2273    * posted in the bus. We want to ignore those messages until the
2274    * pipeline really prerolled. */
2275   priv->adding = TRUE;
2276
2277   /* join the element in the PAUSED state because this callback is
2278    * called from the streaming thread and it is PAUSED */
2279   if (!gst_rtsp_stream_join_bin (stream, GST_BIN (priv->pipeline),
2280           priv->rtpbin, GST_STATE_PAUSED)) {
2281     GST_WARNING ("failed to join bin element");
2282   }
2283
2284   priv->adding = FALSE;
2285   g_rec_mutex_unlock (&priv->state_lock);
2286
2287   return;
2288
2289   /* ERRORS */
2290 not_preparing:
2291   {
2292     gst_rtsp_media_remove_stream (media, stream);
2293     g_rec_mutex_unlock (&priv->state_lock);
2294     GST_INFO ("ignore pad because we are not preparing");
2295     return;
2296   }
2297 }
2298
2299 static void
2300 pad_removed_cb (GstElement * element, GstPad * pad, GstRTSPMedia * media)
2301 {
2302   GstRTSPMediaPrivate *priv = media->priv;
2303   GstRTSPStream *stream;
2304
2305   stream = g_object_get_data (G_OBJECT (pad), "gst-rtsp-dynpad-stream");
2306   if (stream == NULL)
2307     return;
2308
2309   GST_INFO ("pad removed %s:%s, stream %p", GST_DEBUG_PAD_NAME (pad), stream);
2310
2311   g_rec_mutex_lock (&priv->state_lock);
2312   gst_rtsp_stream_leave_bin (stream, GST_BIN (priv->pipeline), priv->rtpbin);
2313   g_rec_mutex_unlock (&priv->state_lock);
2314
2315   gst_rtsp_media_remove_stream (media, stream);
2316 }
2317
2318 static void
2319 remove_fakesink (GstRTSPMediaPrivate * priv)
2320 {
2321   GstElement *fakesink;
2322
2323   g_mutex_lock (&priv->lock);
2324   if ((fakesink = priv->fakesink))
2325     gst_object_ref (fakesink);
2326   priv->fakesink = NULL;
2327   g_mutex_unlock (&priv->lock);
2328
2329   if (fakesink) {
2330     gst_bin_remove (GST_BIN (priv->pipeline), fakesink);
2331     gst_element_set_state (fakesink, GST_STATE_NULL);
2332     gst_object_unref (fakesink);
2333     GST_INFO ("removed fakesink");
2334   }
2335 }
2336
2337 static void
2338 no_more_pads_cb (GstElement * element, GstRTSPMedia * media)
2339 {
2340   GstRTSPMediaPrivate *priv = media->priv;
2341
2342   GST_INFO ("no more pads");
2343   remove_fakesink (priv);
2344 }
2345
2346 typedef struct _DynPaySignalHandlers DynPaySignalHandlers;
2347
2348 struct _DynPaySignalHandlers
2349 {
2350   gulong pad_added_handler;
2351   gulong pad_removed_handler;
2352   gulong no_more_pads_handler;
2353 };
2354
2355 static gboolean
2356 start_preroll (GstRTSPMedia * media)
2357 {
2358   GstRTSPMediaPrivate *priv = media->priv;
2359   GstStateChangeReturn ret;
2360
2361   GST_INFO ("setting pipeline to PAUSED for media %p", media);
2362   /* first go to PAUSED */
2363   ret = set_target_state (media, GST_STATE_PAUSED, TRUE);
2364
2365   switch (ret) {
2366     case GST_STATE_CHANGE_SUCCESS:
2367       GST_INFO ("SUCCESS state change for media %p", media);
2368       priv->seekable = TRUE;
2369       break;
2370     case GST_STATE_CHANGE_ASYNC:
2371       GST_INFO ("ASYNC state change for media %p", media);
2372       priv->seekable = TRUE;
2373       break;
2374     case GST_STATE_CHANGE_NO_PREROLL:
2375       /* we need to go to PLAYING */
2376       GST_INFO ("NO_PREROLL state change: live media %p", media);
2377       /* FIXME we disable seeking for live streams for now. We should perform a
2378        * seeking query in preroll instead */
2379       priv->seekable = FALSE;
2380       priv->is_live = TRUE;
2381       if (!(priv->transport_mode & GST_RTSP_TRANSPORT_MODE_RECORD)) {
2382         /* start blocked  to make sure nothing goes to the sink */
2383         media_streams_set_blocked (media, TRUE);
2384       }
2385       ret = set_state (media, GST_STATE_PLAYING);
2386       if (ret == GST_STATE_CHANGE_FAILURE)
2387         goto state_failed;
2388       break;
2389     case GST_STATE_CHANGE_FAILURE:
2390       goto state_failed;
2391   }
2392
2393   return TRUE;
2394
2395 state_failed:
2396   {
2397     GST_WARNING ("failed to preroll pipeline");
2398     return FALSE;
2399   }
2400 }
2401
2402 static gboolean
2403 wait_preroll (GstRTSPMedia * media)
2404 {
2405   GstRTSPMediaStatus status;
2406
2407   GST_DEBUG ("wait to preroll pipeline");
2408
2409   /* wait until pipeline is prerolled */
2410   status = gst_rtsp_media_get_status (media);
2411   if (status == GST_RTSP_MEDIA_STATUS_ERROR)
2412     goto preroll_failed;
2413
2414   return TRUE;
2415
2416 preroll_failed:
2417   {
2418     GST_WARNING ("failed to preroll pipeline");
2419     return FALSE;
2420   }
2421 }
2422
2423 static gboolean
2424 start_prepare (GstRTSPMedia * media)
2425 {
2426   GstRTSPMediaPrivate *priv = media->priv;
2427   guint i;
2428   GList *walk;
2429
2430   /* link streams we already have, other streams might appear when we have
2431    * dynamic elements */
2432   for (i = 0; i < priv->streams->len; i++) {
2433     GstRTSPStream *stream;
2434
2435     stream = g_ptr_array_index (priv->streams, i);
2436
2437     if (!gst_rtsp_stream_join_bin (stream, GST_BIN (priv->pipeline),
2438             priv->rtpbin, GST_STATE_NULL)) {
2439       goto join_bin_failed;
2440     }
2441   }
2442
2443   if (priv->rtpbin)
2444     g_object_set (priv->rtpbin, "do-retransmission", priv->rtx_time > 0, NULL);
2445
2446   for (walk = priv->dynamic; walk; walk = g_list_next (walk)) {
2447     GstElement *elem = walk->data;
2448     DynPaySignalHandlers *handlers = g_slice_new (DynPaySignalHandlers);
2449
2450     GST_INFO ("adding callbacks for dynamic element %p", elem);
2451
2452     handlers->pad_added_handler = g_signal_connect (elem, "pad-added",
2453         (GCallback) pad_added_cb, media);
2454     handlers->pad_removed_handler = g_signal_connect (elem, "pad-removed",
2455         (GCallback) pad_removed_cb, media);
2456     handlers->no_more_pads_handler = g_signal_connect (elem, "no-more-pads",
2457         (GCallback) no_more_pads_cb, media);
2458
2459     g_object_set_data (G_OBJECT (elem), "gst-rtsp-dynpay-handlers", handlers);
2460
2461     /* we add a fakesink here in order to make the state change async. We remove
2462      * the fakesink again in the no-more-pads callback. */
2463     priv->fakesink = gst_element_factory_make ("fakesink", "fakesink");
2464     gst_bin_add (GST_BIN (priv->pipeline), priv->fakesink);
2465   }
2466
2467   if (!start_preroll (media))
2468     goto preroll_failed;
2469
2470   return FALSE;
2471
2472 join_bin_failed:
2473   {
2474     GST_WARNING ("failed to join bin element");
2475     gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_ERROR);
2476     return FALSE;
2477   }
2478 preroll_failed:
2479   {
2480     GST_WARNING ("failed to preroll pipeline");
2481     gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_ERROR);
2482     return FALSE;
2483   }
2484 }
2485
2486 static gboolean
2487 default_prepare (GstRTSPMedia * media, GstRTSPThread * thread)
2488 {
2489   GstRTSPMediaPrivate *priv;
2490   GstRTSPMediaClass *klass;
2491   GstBus *bus;
2492   GMainContext *context;
2493   GSource *source;
2494
2495   priv = media->priv;
2496
2497   klass = GST_RTSP_MEDIA_GET_CLASS (media);
2498
2499   if (!klass->create_rtpbin)
2500     goto no_create_rtpbin;
2501
2502   priv->rtpbin = klass->create_rtpbin (media);
2503   if (priv->rtpbin != NULL) {
2504     gboolean success = TRUE;
2505
2506     g_object_set (priv->rtpbin, "latency", priv->latency, NULL);
2507
2508     if (klass->setup_rtpbin)
2509       success = klass->setup_rtpbin (media, priv->rtpbin);
2510
2511     if (success == FALSE) {
2512       gst_object_unref (priv->rtpbin);
2513       priv->rtpbin = NULL;
2514     }
2515   }
2516   if (priv->rtpbin == NULL)
2517     goto no_rtpbin;
2518
2519   priv->thread = thread;
2520   context = (thread != NULL) ? (thread->context) : NULL;
2521
2522   bus = gst_pipeline_get_bus (GST_PIPELINE_CAST (priv->pipeline));
2523
2524   /* add the pipeline bus to our custom mainloop */
2525   priv->source = gst_bus_create_watch (bus);
2526   gst_object_unref (bus);
2527
2528   g_source_set_callback (priv->source, (GSourceFunc) bus_message,
2529       g_object_ref (media), (GDestroyNotify) watch_destroyed);
2530
2531   priv->id = g_source_attach (priv->source, context);
2532
2533   /* add stuff to the bin */
2534   gst_bin_add (GST_BIN (priv->pipeline), priv->rtpbin);
2535
2536   /* do remainder in context */
2537   source = g_idle_source_new ();
2538   g_source_set_callback (source, (GSourceFunc) start_prepare, media, NULL);
2539   g_source_attach (source, context);
2540   g_source_unref (source);
2541
2542   return TRUE;
2543
2544   /* ERRORS */
2545 no_create_rtpbin:
2546   {
2547     GST_ERROR ("no create_rtpbin function");
2548     g_critical ("no create_rtpbin vmethod function set");
2549     return FALSE;
2550   }
2551 no_rtpbin:
2552   {
2553     GST_WARNING ("no rtpbin element");
2554     g_warning ("failed to create element 'rtpbin', check your installation");
2555     return FALSE;
2556   }
2557 }
2558
2559 /**
2560  * gst_rtsp_media_prepare:
2561  * @media: a #GstRTSPMedia
2562  * @thread: (transfer full) (allow-none): a #GstRTSPThread to run the
2563  *   bus handler or %NULL
2564  *
2565  * Prepare @media for streaming. This function will create the objects
2566  * to manage the streaming. A pipeline must have been set on @media with
2567  * gst_rtsp_media_take_pipeline().
2568  *
2569  * It will preroll the pipeline and collect vital information about the streams
2570  * such as the duration.
2571  *
2572  * Returns: %TRUE on success.
2573  */
2574 gboolean
2575 gst_rtsp_media_prepare (GstRTSPMedia * media, GstRTSPThread * thread)
2576 {
2577   GstRTSPMediaPrivate *priv;
2578   GstRTSPMediaClass *klass;
2579
2580   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
2581
2582   priv = media->priv;
2583
2584   g_rec_mutex_lock (&priv->state_lock);
2585   priv->prepare_count++;
2586
2587   if (priv->status == GST_RTSP_MEDIA_STATUS_PREPARED ||
2588       priv->status == GST_RTSP_MEDIA_STATUS_SUSPENDED)
2589     goto was_prepared;
2590
2591   if (priv->status == GST_RTSP_MEDIA_STATUS_PREPARING)
2592     goto is_preparing;
2593
2594   if (priv->status != GST_RTSP_MEDIA_STATUS_UNPREPARED)
2595     goto not_unprepared;
2596
2597   if (!priv->reusable && priv->reused)
2598     goto is_reused;
2599
2600   GST_INFO ("preparing media %p", media);
2601
2602   /* reset some variables */
2603   priv->is_live = FALSE;
2604   priv->seekable = FALSE;
2605   priv->buffering = FALSE;
2606
2607   /* we're preparing now */
2608   gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_PREPARING);
2609
2610   klass = GST_RTSP_MEDIA_GET_CLASS (media);
2611   if (klass->prepare) {
2612     if (!klass->prepare (media, thread))
2613       goto prepare_failed;
2614   }
2615
2616 wait_status:
2617   g_rec_mutex_unlock (&priv->state_lock);
2618
2619   /* now wait for all pads to be prerolled, FIXME, we should somehow be
2620    * able to do this async so that we don't block the server thread. */
2621   if (!wait_preroll (media))
2622     goto preroll_failed;
2623
2624   g_signal_emit (media, gst_rtsp_media_signals[SIGNAL_PREPARED], 0, NULL);
2625
2626   GST_INFO ("object %p is prerolled", media);
2627
2628   return TRUE;
2629
2630   /* OK */
2631 is_preparing:
2632   {
2633     /* we are not going to use the giving thread, so stop it. */
2634     if (thread)
2635       gst_rtsp_thread_stop (thread);
2636     goto wait_status;
2637   }
2638 was_prepared:
2639   {
2640     GST_LOG ("media %p was prepared", media);
2641     /* we are not going to use the giving thread, so stop it. */
2642     if (thread)
2643       gst_rtsp_thread_stop (thread);
2644     g_rec_mutex_unlock (&priv->state_lock);
2645     return TRUE;
2646   }
2647   /* ERRORS */
2648 not_unprepared:
2649   {
2650     /* we are not going to use the giving thread, so stop it. */
2651     if (thread)
2652       gst_rtsp_thread_stop (thread);
2653     GST_WARNING ("media %p was not unprepared", media);
2654     priv->prepare_count--;
2655     g_rec_mutex_unlock (&priv->state_lock);
2656     return FALSE;
2657   }
2658 is_reused:
2659   {
2660     /* we are not going to use the giving thread, so stop it. */
2661     if (thread)
2662       gst_rtsp_thread_stop (thread);
2663     priv->prepare_count--;
2664     g_rec_mutex_unlock (&priv->state_lock);
2665     GST_WARNING ("can not reuse media %p", media);
2666     return FALSE;
2667   }
2668 prepare_failed:
2669   {
2670     /* we are not going to use the giving thread, so stop it. */
2671     if (thread)
2672       gst_rtsp_thread_stop (thread);
2673     priv->prepare_count--;
2674     g_rec_mutex_unlock (&priv->state_lock);
2675     GST_ERROR ("failed to prepare media");
2676     return FALSE;
2677   }
2678 preroll_failed:
2679   {
2680     GST_WARNING ("failed to preroll pipeline");
2681     gst_rtsp_media_unprepare (media);
2682     return FALSE;
2683   }
2684 }
2685
2686 /* must be called with state-lock */
2687 static void
2688 finish_unprepare (GstRTSPMedia * media)
2689 {
2690   GstRTSPMediaPrivate *priv = media->priv;
2691   gint i;
2692   GList *walk;
2693
2694   GST_DEBUG ("shutting down");
2695
2696   /* release the lock on shutdown, otherwise pad_added_cb might try to
2697    * acquire the lock and then we deadlock */
2698   g_rec_mutex_unlock (&priv->state_lock);
2699   set_state (media, GST_STATE_NULL);
2700   g_rec_mutex_lock (&priv->state_lock);
2701   remove_fakesink (priv);
2702
2703   for (i = 0; i < priv->streams->len; i++) {
2704     GstRTSPStream *stream;
2705
2706     GST_INFO ("Removing elements of stream %d from pipeline", i);
2707
2708     stream = g_ptr_array_index (priv->streams, i);
2709
2710     gst_rtsp_stream_leave_bin (stream, GST_BIN (priv->pipeline), priv->rtpbin);
2711   }
2712
2713   /* remove the pad signal handlers */
2714   for (walk = priv->dynamic; walk; walk = g_list_next (walk)) {
2715     GstElement *elem = walk->data;
2716     DynPaySignalHandlers *handlers;
2717
2718     handlers =
2719         g_object_steal_data (G_OBJECT (elem), "gst-rtsp-dynpay-handlers");
2720     g_assert (handlers != NULL);
2721
2722     g_signal_handler_disconnect (G_OBJECT (elem), handlers->pad_added_handler);
2723     g_signal_handler_disconnect (G_OBJECT (elem),
2724         handlers->pad_removed_handler);
2725     g_signal_handler_disconnect (G_OBJECT (elem),
2726         handlers->no_more_pads_handler);
2727
2728     g_slice_free (DynPaySignalHandlers, handlers);
2729   }
2730
2731   gst_bin_remove (GST_BIN (priv->pipeline), priv->rtpbin);
2732   priv->rtpbin = NULL;
2733
2734   if (priv->nettime)
2735     gst_object_unref (priv->nettime);
2736   priv->nettime = NULL;
2737
2738   priv->reused = TRUE;
2739   gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_UNPREPARED);
2740
2741   /* when the media is not reusable, this will effectively unref the media and
2742    * recreate it */
2743   g_signal_emit (media, gst_rtsp_media_signals[SIGNAL_UNPREPARED], 0, NULL);
2744
2745   /* the source has the last ref to the media */
2746   if (priv->source) {
2747     GST_DEBUG ("destroy source");
2748     g_source_destroy (priv->source);
2749     g_source_unref (priv->source);
2750   }
2751   if (priv->thread) {
2752     GST_DEBUG ("stop thread");
2753     gst_rtsp_thread_stop (priv->thread);
2754   }
2755 }
2756
2757 /* called with state-lock */
2758 static gboolean
2759 default_unprepare (GstRTSPMedia * media)
2760 {
2761   GstRTSPMediaPrivate *priv = media->priv;
2762
2763   gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_UNPREPARING);
2764
2765   if (priv->eos_shutdown) {
2766     GST_DEBUG ("sending EOS for shutdown");
2767     /* ref so that we don't disappear */
2768     gst_element_send_event (priv->pipeline, gst_event_new_eos ());
2769     /* we need to go to playing again for the EOS to propagate, normally in this
2770      * state, nothing is receiving data from us anymore so this is ok. */
2771     set_state (media, GST_STATE_PLAYING);
2772   } else {
2773     finish_unprepare (media);
2774   }
2775   return TRUE;
2776 }
2777
2778 /**
2779  * gst_rtsp_media_unprepare:
2780  * @media: a #GstRTSPMedia
2781  *
2782  * Unprepare @media. After this call, the media should be prepared again before
2783  * it can be used again. If the media is set to be non-reusable, a new instance
2784  * must be created.
2785  *
2786  * Returns: %TRUE on success.
2787  */
2788 gboolean
2789 gst_rtsp_media_unprepare (GstRTSPMedia * media)
2790 {
2791   GstRTSPMediaPrivate *priv;
2792   gboolean success;
2793
2794   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
2795
2796   priv = media->priv;
2797
2798   g_rec_mutex_lock (&priv->state_lock);
2799   if (priv->status == GST_RTSP_MEDIA_STATUS_UNPREPARED)
2800     goto was_unprepared;
2801
2802   priv->prepare_count--;
2803   if (priv->prepare_count > 0)
2804     goto is_busy;
2805
2806   GST_INFO ("unprepare media %p", media);
2807   if (priv->blocked)
2808     media_streams_set_blocked (media, FALSE);
2809   set_target_state (media, GST_STATE_NULL, FALSE);
2810   success = TRUE;
2811
2812   if (priv->status == GST_RTSP_MEDIA_STATUS_PREPARED) {
2813     GstRTSPMediaClass *klass;
2814
2815     klass = GST_RTSP_MEDIA_GET_CLASS (media);
2816     if (klass->unprepare)
2817       success = klass->unprepare (media);
2818   } else {
2819     finish_unprepare (media);
2820   }
2821   g_rec_mutex_unlock (&priv->state_lock);
2822
2823   return success;
2824
2825 was_unprepared:
2826   {
2827     g_rec_mutex_unlock (&priv->state_lock);
2828     GST_INFO ("media %p was already unprepared", media);
2829     return TRUE;
2830   }
2831 is_busy:
2832   {
2833     GST_INFO ("media %p still prepared %d times", media, priv->prepare_count);
2834     g_rec_mutex_unlock (&priv->state_lock);
2835     return TRUE;
2836   }
2837 }
2838
2839 /* should be called with state-lock */
2840 static GstClock *
2841 get_clock_unlocked (GstRTSPMedia * media)
2842 {
2843   if (media->priv->status != GST_RTSP_MEDIA_STATUS_PREPARED) {
2844     GST_DEBUG_OBJECT (media, "media was not prepared");
2845     return NULL;
2846   }
2847   return gst_pipeline_get_clock (GST_PIPELINE_CAST (media->priv->pipeline));
2848 }
2849
2850 /**
2851  * gst_rtsp_media_get_clock:
2852  * @media: a #GstRTSPMedia
2853  *
2854  * Get the clock that is used by the pipeline in @media.
2855  *
2856  * @media must be prepared before this method returns a valid clock object.
2857  *
2858  * Returns: (transfer full): the #GstClock used by @media. unref after usage.
2859  */
2860 GstClock *
2861 gst_rtsp_media_get_clock (GstRTSPMedia * media)
2862 {
2863   GstClock *clock;
2864   GstRTSPMediaPrivate *priv;
2865
2866   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), NULL);
2867
2868   priv = media->priv;
2869
2870   g_rec_mutex_lock (&priv->state_lock);
2871   clock = get_clock_unlocked (media);
2872   g_rec_mutex_unlock (&priv->state_lock);
2873
2874   return clock;
2875 }
2876
2877 /**
2878  * gst_rtsp_media_get_base_time:
2879  * @media: a #GstRTSPMedia
2880  *
2881  * Get the base_time that is used by the pipeline in @media.
2882  *
2883  * @media must be prepared before this method returns a valid base_time.
2884  *
2885  * Returns: the base_time used by @media.
2886  */
2887 GstClockTime
2888 gst_rtsp_media_get_base_time (GstRTSPMedia * media)
2889 {
2890   GstClockTime result;
2891   GstRTSPMediaPrivate *priv;
2892
2893   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), GST_CLOCK_TIME_NONE);
2894
2895   priv = media->priv;
2896
2897   g_rec_mutex_lock (&priv->state_lock);
2898   if (media->priv->status != GST_RTSP_MEDIA_STATUS_PREPARED)
2899     goto not_prepared;
2900
2901   result = gst_element_get_base_time (media->priv->pipeline);
2902   g_rec_mutex_unlock (&priv->state_lock);
2903
2904   return result;
2905
2906   /* ERRORS */
2907 not_prepared:
2908   {
2909     g_rec_mutex_unlock (&priv->state_lock);
2910     GST_DEBUG_OBJECT (media, "media was not prepared");
2911     return GST_CLOCK_TIME_NONE;
2912   }
2913 }
2914
2915 /**
2916  * gst_rtsp_media_get_time_provider:
2917  * @media: a #GstRTSPMedia
2918  * @address: (allow-none): an address or %NULL
2919  * @port: a port or 0
2920  *
2921  * Get the #GstNetTimeProvider for the clock used by @media. The time provider
2922  * will listen on @address and @port for client time requests.
2923  *
2924  * Returns: (transfer full): the #GstNetTimeProvider of @media.
2925  */
2926 GstNetTimeProvider *
2927 gst_rtsp_media_get_time_provider (GstRTSPMedia * media, const gchar * address,
2928     guint16 port)
2929 {
2930   GstRTSPMediaPrivate *priv;
2931   GstNetTimeProvider *provider = NULL;
2932
2933   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), NULL);
2934
2935   priv = media->priv;
2936
2937   g_rec_mutex_lock (&priv->state_lock);
2938   if (priv->time_provider) {
2939     if ((provider = priv->nettime) == NULL) {
2940       GstClock *clock;
2941
2942       if (priv->time_provider && (clock = get_clock_unlocked (media))) {
2943         provider = gst_net_time_provider_new (clock, address, port);
2944         gst_object_unref (clock);
2945
2946         priv->nettime = provider;
2947       }
2948     }
2949   }
2950   g_rec_mutex_unlock (&priv->state_lock);
2951
2952   if (provider)
2953     gst_object_ref (provider);
2954
2955   return provider;
2956 }
2957
2958 static gboolean
2959 default_setup_sdp (GstRTSPMedia * media, GstSDPMessage * sdp, GstSDPInfo * info)
2960 {
2961   return gst_rtsp_sdp_from_media (sdp, info, media);
2962 }
2963
2964 /**
2965  * gst_rtsp_media_setup_sdp:
2966  * @media: a #GstRTSPMedia
2967  * @sdp: (transfer none): a #GstSDPMessage
2968  * @info: (transfer none): a #GstSDPInfo
2969  *
2970  * Add @media specific info to @sdp. @info is used to configure the connection
2971  * information in the SDP.
2972  *
2973  * Returns: TRUE on success.
2974  */
2975 gboolean
2976 gst_rtsp_media_setup_sdp (GstRTSPMedia * media, GstSDPMessage * sdp,
2977     GstSDPInfo * info)
2978 {
2979   GstRTSPMediaPrivate *priv;
2980   GstRTSPMediaClass *klass;
2981   gboolean res;
2982
2983   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
2984   g_return_val_if_fail (sdp != NULL, FALSE);
2985   g_return_val_if_fail (info != NULL, FALSE);
2986
2987   priv = media->priv;
2988
2989   g_rec_mutex_lock (&priv->state_lock);
2990
2991   klass = GST_RTSP_MEDIA_GET_CLASS (media);
2992
2993   if (!klass->setup_sdp)
2994     goto no_setup_sdp;
2995
2996   res = klass->setup_sdp (media, sdp, info);
2997
2998   g_rec_mutex_unlock (&priv->state_lock);
2999
3000   return res;
3001
3002   /* ERRORS */
3003 no_setup_sdp:
3004   {
3005     g_rec_mutex_unlock (&priv->state_lock);
3006     GST_ERROR ("no setup_sdp function");
3007     g_critical ("no setup_sdp vmethod function set");
3008     return FALSE;
3009   }
3010 }
3011
3012 static const gchar *
3013 rtsp_get_attribute_for_pt (const GstSDPMedia * media, const gchar * name,
3014     gint pt)
3015 {
3016   guint i;
3017
3018   for (i = 0;; i++) {
3019     const gchar *attr;
3020     gint val;
3021
3022     if ((attr = gst_sdp_media_get_attribute_val_n (media, name, i)) == NULL)
3023       break;
3024
3025     if (sscanf (attr, "%d ", &val) != 1)
3026       continue;
3027
3028     if (val == pt)
3029       return attr;
3030   }
3031   return NULL;
3032 }
3033
3034 #define PARSE_INT(p, del, res)          \
3035 G_STMT_START {                          \
3036   gchar *t = p;                         \
3037   p = strstr (p, del);                  \
3038   if (p == NULL)                        \
3039     res = -1;                           \
3040   else {                                \
3041     *p = '\0';                          \
3042     p++;                                \
3043     res = atoi (t);                     \
3044   }                                     \
3045 } G_STMT_END
3046
3047 #define PARSE_STRING(p, del, res)       \
3048 G_STMT_START {                          \
3049   gchar *t = p;                         \
3050   p = strstr (p, del);                  \
3051   if (p == NULL) {                      \
3052     res = NULL;                         \
3053     p = t;                              \
3054   }                                     \
3055   else {                                \
3056     *p = '\0';                          \
3057     p++;                                \
3058     res = t;                            \
3059   }                                     \
3060 } G_STMT_END
3061
3062 #define SKIP_SPACES(p)                  \
3063   while (*p && g_ascii_isspace (*p))    \
3064     p++;
3065
3066 /* rtpmap contains:
3067  *
3068  *  <payload> <encoding_name>/<clock_rate>[/<encoding_params>]
3069  */
3070 static gboolean
3071 parse_rtpmap (const gchar * rtpmap, gint * payload, gchar ** name,
3072     gint * rate, gchar ** params)
3073 {
3074   gchar *p, *t;
3075
3076   p = (gchar *) rtpmap;
3077
3078   PARSE_INT (p, " ", *payload);
3079   if (*payload == -1)
3080     return FALSE;
3081
3082   SKIP_SPACES (p);
3083   if (*p == '\0')
3084     return FALSE;
3085
3086   PARSE_STRING (p, "/", *name);
3087   if (*name == NULL) {
3088     GST_DEBUG ("no rate, name %s", p);
3089     /* no rate, assume -1 then, this is not supposed to happen but RealMedia
3090      * streams seem to omit the rate. */
3091     *name = p;
3092     *rate = -1;
3093     return TRUE;
3094   }
3095
3096   t = p;
3097   p = strstr (p, "/");
3098   if (p == NULL) {
3099     *rate = atoi (t);
3100     return TRUE;
3101   }
3102   *p = '\0';
3103   p++;
3104   *rate = atoi (t);
3105
3106   t = p;
3107   if (*p == '\0')
3108     return TRUE;
3109   *params = t;
3110
3111   return TRUE;
3112 }
3113
3114 /*
3115  *  Mapping of caps to and from SDP fields:
3116  *
3117  *   a=rtpmap:<payload> <encoding_name>/<clock_rate>[/<encoding_params>]
3118  *   a=fmtp:<payload> <param>[=<value>];...
3119  */
3120 static GstCaps *
3121 media_to_caps (gint pt, const GstSDPMedia * media)
3122 {
3123   GstCaps *caps;
3124   const gchar *rtpmap;
3125   const gchar *fmtp;
3126   gchar *name = NULL;
3127   gint rate = -1;
3128   gchar *params = NULL;
3129   gchar *tmp;
3130   GstStructure *s;
3131   gint payload = 0;
3132   gboolean ret;
3133
3134   /* get and parse rtpmap */
3135   rtpmap = rtsp_get_attribute_for_pt (media, "rtpmap", pt);
3136
3137   if (rtpmap) {
3138     ret = parse_rtpmap (rtpmap, &payload, &name, &rate, &params);
3139     if (!ret) {
3140       g_warning ("error parsing rtpmap, ignoring");
3141       rtpmap = NULL;
3142     }
3143   }
3144   /* dynamic payloads need rtpmap or we fail */
3145   if (rtpmap == NULL && pt >= 96)
3146     goto no_rtpmap;
3147
3148   /* check if we have a rate, if not, we need to look up the rate from the
3149    * default rates based on the payload types. */
3150   if (rate == -1) {
3151     const GstRTPPayloadInfo *info;
3152
3153     if (GST_RTP_PAYLOAD_IS_DYNAMIC (pt)) {
3154       /* dynamic types, use media and encoding_name */
3155       tmp = g_ascii_strdown (media->media, -1);
3156       info = gst_rtp_payload_info_for_name (tmp, name);
3157       g_free (tmp);
3158     } else {
3159       /* static types, use payload type */
3160       info = gst_rtp_payload_info_for_pt (pt);
3161     }
3162
3163     if (info) {
3164       if ((rate = info->clock_rate) == 0)
3165         rate = -1;
3166     }
3167     /* we fail if we cannot find one */
3168     if (rate == -1)
3169       goto no_rate;
3170   }
3171
3172   tmp = g_ascii_strdown (media->media, -1);
3173   caps = gst_caps_new_simple ("application/x-unknown",
3174       "media", G_TYPE_STRING, tmp, "payload", G_TYPE_INT, pt, NULL);
3175   g_free (tmp);
3176   s = gst_caps_get_structure (caps, 0);
3177
3178   gst_structure_set (s, "clock-rate", G_TYPE_INT, rate, NULL);
3179
3180   /* encoding name must be upper case */
3181   if (name != NULL) {
3182     tmp = g_ascii_strup (name, -1);
3183     gst_structure_set (s, "encoding-name", G_TYPE_STRING, tmp, NULL);
3184     g_free (tmp);
3185   }
3186
3187   /* params must be lower case */
3188   if (params != NULL) {
3189     tmp = g_ascii_strdown (params, -1);
3190     gst_structure_set (s, "encoding-params", G_TYPE_STRING, tmp, NULL);
3191     g_free (tmp);
3192   }
3193
3194   /* parse optional fmtp: field */
3195   if ((fmtp = rtsp_get_attribute_for_pt (media, "fmtp", pt))) {
3196     gchar *p;
3197     gint payload = 0;
3198
3199     p = (gchar *) fmtp;
3200
3201     /* p is now of the format <payload> <param>[=<value>];... */
3202     PARSE_INT (p, " ", payload);
3203     if (payload != -1 && payload == pt) {
3204       gchar **pairs;
3205       gint i;
3206
3207       /* <param>[=<value>] are separated with ';' */
3208       pairs = g_strsplit (p, ";", 0);
3209       for (i = 0; pairs[i]; i++) {
3210         gchar *valpos;
3211         const gchar *val, *key;
3212
3213         /* the key may not have a '=', the value can have other '='s */
3214         valpos = strstr (pairs[i], "=");
3215         if (valpos) {
3216           /* we have a '=' and thus a value, remove the '=' with \0 */
3217           *valpos = '\0';
3218           /* value is everything between '=' and ';'. We split the pairs at ;
3219            * boundaries so we can take the remainder of the value. Some servers
3220            * put spaces around the value which we strip off here. Alternatively
3221            * we could strip those spaces in the depayloaders should these spaces
3222            * actually carry any meaning in the future. */
3223           val = g_strstrip (valpos + 1);
3224         } else {
3225           /* simple <param>;.. is translated into <param>=1;... */
3226           val = "1";
3227         }
3228         /* strip the key of spaces, convert key to lowercase but not the value. */
3229         key = g_strstrip (pairs[i]);
3230         if (strlen (key) > 1) {
3231           tmp = g_ascii_strdown (key, -1);
3232           gst_structure_set (s, tmp, G_TYPE_STRING, val, NULL);
3233           g_free (tmp);
3234         }
3235       }
3236       g_strfreev (pairs);
3237     }
3238   }
3239   return caps;
3240
3241   /* ERRORS */
3242 no_rtpmap:
3243   {
3244     g_warning ("rtpmap type not given for dynamic payload %d", pt);
3245     return NULL;
3246   }
3247 no_rate:
3248   {
3249     g_warning ("rate unknown for payload type %d", pt);
3250     return NULL;
3251   }
3252 }
3253
3254 static gboolean
3255 parse_keymgmt (const gchar * keymgmt, GstCaps * caps)
3256 {
3257   gboolean res = FALSE;
3258   gchar *p, *kmpid;
3259   gsize size;
3260   guchar *data;
3261   GstMIKEYMessage *msg;
3262   const GstMIKEYPayload *payload;
3263   const gchar *srtp_cipher;
3264   const gchar *srtp_auth;
3265
3266   p = (gchar *) keymgmt;
3267
3268   SKIP_SPACES (p);
3269   if (*p == '\0')
3270     return FALSE;
3271
3272   PARSE_STRING (p, " ", kmpid);
3273   if (!g_str_equal (kmpid, "mikey"))
3274     return FALSE;
3275
3276   data = g_base64_decode (p, &size);
3277   if (data == NULL)
3278     return FALSE;
3279
3280   msg = gst_mikey_message_new_from_data (data, size, NULL, NULL);
3281   g_free (data);
3282   if (msg == NULL)
3283     return FALSE;
3284
3285   srtp_cipher = "aes-128-icm";
3286   srtp_auth = "hmac-sha1-80";
3287
3288   /* check the Security policy if any */
3289   if ((payload = gst_mikey_message_find_payload (msg, GST_MIKEY_PT_SP, 0))) {
3290     GstMIKEYPayloadSP *p = (GstMIKEYPayloadSP *) payload;
3291     guint len, i;
3292
3293     if (p->proto != GST_MIKEY_SEC_PROTO_SRTP)
3294       goto done;
3295
3296     len = gst_mikey_payload_sp_get_n_params (payload);
3297     for (i = 0; i < len; i++) {
3298       const GstMIKEYPayloadSPParam *param =
3299           gst_mikey_payload_sp_get_param (payload, i);
3300
3301       switch (param->type) {
3302         case GST_MIKEY_SP_SRTP_ENC_ALG:
3303           switch (param->val[0]) {
3304             case 0:
3305               srtp_cipher = "null";
3306               break;
3307             case 2:
3308             case 1:
3309               srtp_cipher = "aes-128-icm";
3310               break;
3311             default:
3312               break;
3313           }
3314           break;
3315         case GST_MIKEY_SP_SRTP_ENC_KEY_LEN:
3316           switch (param->val[0]) {
3317             case AES_128_KEY_LEN:
3318               srtp_cipher = "aes-128-icm";
3319               break;
3320             case AES_256_KEY_LEN:
3321               srtp_cipher = "aes-256-icm";
3322               break;
3323             default:
3324               break;
3325           }
3326           break;
3327         case GST_MIKEY_SP_SRTP_AUTH_ALG:
3328           switch (param->val[0]) {
3329             case 0:
3330               srtp_auth = "null";
3331               break;
3332             case 2:
3333             case 1:
3334               srtp_auth = "hmac-sha1-80";
3335               break;
3336             default:
3337               break;
3338           }
3339           break;
3340         case GST_MIKEY_SP_SRTP_AUTH_KEY_LEN:
3341           switch (param->val[0]) {
3342             case HMAC_32_KEY_LEN:
3343               srtp_auth = "hmac-sha1-32";
3344               break;
3345             case HMAC_80_KEY_LEN:
3346               srtp_auth = "hmac-sha1-80";
3347               break;
3348             default:
3349               break;
3350           }
3351           break;
3352         case GST_MIKEY_SP_SRTP_SRTP_ENC:
3353           break;
3354         case GST_MIKEY_SP_SRTP_SRTCP_ENC:
3355           break;
3356         default:
3357           break;
3358       }
3359     }
3360   }
3361
3362   if (!(payload = gst_mikey_message_find_payload (msg, GST_MIKEY_PT_KEMAC, 0)))
3363     goto done;
3364   else {
3365     GstMIKEYPayloadKEMAC *p = (GstMIKEYPayloadKEMAC *) payload;
3366     const GstMIKEYPayload *sub;
3367     GstMIKEYPayloadKeyData *pkd;
3368     GstBuffer *buf;
3369
3370     if (p->enc_alg != GST_MIKEY_ENC_NULL || p->mac_alg != GST_MIKEY_MAC_NULL)
3371       goto done;
3372
3373     if (!(sub = gst_mikey_payload_kemac_get_sub (payload, 0)))
3374       goto done;
3375
3376     if (sub->type != GST_MIKEY_PT_KEY_DATA)
3377       goto done;
3378
3379     pkd = (GstMIKEYPayloadKeyData *) sub;
3380     buf =
3381         gst_buffer_new_wrapped (g_memdup (pkd->key_data, pkd->key_len),
3382         pkd->key_len);
3383     gst_caps_set_simple (caps, "srtp-key", GST_TYPE_BUFFER, buf, NULL);
3384   }
3385
3386   gst_caps_set_simple (caps,
3387       "srtp-cipher", G_TYPE_STRING, srtp_cipher,
3388       "srtp-auth", G_TYPE_STRING, srtp_auth,
3389       "srtcp-cipher", G_TYPE_STRING, srtp_cipher,
3390       "srtcp-auth", G_TYPE_STRING, srtp_auth, NULL);
3391
3392   res = TRUE;
3393 done:
3394   gst_mikey_message_unref (msg);
3395
3396   return res;
3397 }
3398
3399 /*
3400  * Mapping SDP attributes to caps
3401  *
3402  * prepend 'a-' to IANA registered sdp attributes names
3403  * (ie: not prefixed with 'x-') in order to avoid
3404  * collision with gstreamer standard caps properties names
3405  */
3406 static void
3407 sdp_attributes_to_caps (GArray * attributes, GstCaps * caps)
3408 {
3409   if (attributes->len > 0) {
3410     GstStructure *s;
3411     guint i;
3412
3413     s = gst_caps_get_structure (caps, 0);
3414
3415     for (i = 0; i < attributes->len; i++) {
3416       GstSDPAttribute *attr = &g_array_index (attributes, GstSDPAttribute, i);
3417       gchar *tofree, *key;
3418
3419       key = attr->key;
3420
3421       /* skip some of the attribute we already handle */
3422       if (!strcmp (key, "fmtp"))
3423         continue;
3424       if (!strcmp (key, "rtpmap"))
3425         continue;
3426       if (!strcmp (key, "control"))
3427         continue;
3428       if (!strcmp (key, "range"))
3429         continue;
3430       if (g_str_equal (key, "key-mgmt")) {
3431         parse_keymgmt (attr->value, caps);
3432         continue;
3433       }
3434
3435       /* string must be valid UTF8 */
3436       if (!g_utf8_validate (attr->value, -1, NULL))
3437         continue;
3438
3439       if (!g_str_has_prefix (key, "x-"))
3440         tofree = key = g_strdup_printf ("a-%s", key);
3441       else
3442         tofree = NULL;
3443
3444       GST_DEBUG ("adding caps: %s=%s", key, attr->value);
3445       gst_structure_set (s, key, G_TYPE_STRING, attr->value, NULL);
3446       g_free (tofree);
3447     }
3448   }
3449 }
3450
3451 static gboolean
3452 default_handle_sdp (GstRTSPMedia * media, GstSDPMessage * sdp)
3453 {
3454   GstRTSPMediaPrivate *priv = media->priv;
3455   gint i, medias_len;
3456
3457   medias_len = gst_sdp_message_medias_len (sdp);
3458   if (medias_len != priv->streams->len) {
3459     GST_ERROR ("%p: Media has more or less streams than SDP (%d /= %d)", media,
3460         priv->streams->len, medias_len);
3461     return FALSE;
3462   }
3463
3464   for (i = 0; i < medias_len; i++) {
3465     const gchar *proto, *media_type;
3466     const GstSDPMedia *sdp_media = gst_sdp_message_get_media (sdp, i);
3467     GstRTSPStream *stream;
3468     gint j, formats_len;
3469     const gchar *control;
3470     GstRTSPProfile profile, profiles;
3471
3472     stream = g_ptr_array_index (priv->streams, i);
3473
3474     /* TODO: Should we do something with the other SDP information? */
3475
3476     /* get proto */
3477     proto = gst_sdp_media_get_proto (sdp_media);
3478     if (proto == NULL) {
3479       GST_ERROR ("%p: SDP media %d has no proto", media, i);
3480       return FALSE;
3481     }
3482
3483     if (g_str_equal (proto, "RTP/AVP")) {
3484       media_type = "application/x-rtp";
3485       profile = GST_RTSP_PROFILE_AVP;
3486     } else if (g_str_equal (proto, "RTP/SAVP")) {
3487       media_type = "application/x-srtp";
3488       profile = GST_RTSP_PROFILE_SAVP;
3489     } else if (g_str_equal (proto, "RTP/AVPF")) {
3490       media_type = "application/x-rtp";
3491       profile = GST_RTSP_PROFILE_AVPF;
3492     } else if (g_str_equal (proto, "RTP/SAVPF")) {
3493       media_type = "application/x-srtp";
3494       profile = GST_RTSP_PROFILE_SAVPF;
3495     } else {
3496       GST_ERROR ("%p: unsupported profile '%s' for stream %d", media, proto, i);
3497       return FALSE;
3498     }
3499
3500     profiles = gst_rtsp_stream_get_profiles (stream);
3501     if ((profiles & profile) == 0) {
3502       GST_ERROR ("%p: unsupported profile '%s' for stream %d", media, proto, i);
3503       return FALSE;
3504     }
3505
3506     formats_len = gst_sdp_media_formats_len (sdp_media);
3507     for (j = 0; j < formats_len; j++) {
3508       gint pt;
3509       GstCaps *caps;
3510       GstStructure *s;
3511
3512       pt = atoi (gst_sdp_media_get_format (sdp_media, j));
3513
3514       GST_DEBUG (" looking at %d pt: %d", j, pt);
3515
3516       /* convert caps */
3517       caps = media_to_caps (pt, sdp_media);
3518       if (caps == NULL) {
3519         GST_WARNING (" skipping pt %d without caps", pt);
3520         continue;
3521       }
3522
3523       /* do some tweaks */
3524       GST_DEBUG ("mapping sdp session level attributes to caps");
3525       sdp_attributes_to_caps (sdp->attributes, caps);
3526       GST_DEBUG ("mapping sdp media level attributes to caps");
3527       sdp_attributes_to_caps (sdp_media->attributes, caps);
3528
3529       s = gst_caps_get_structure (caps, 0);
3530       gst_structure_set_name (s, media_type);
3531
3532       gst_rtsp_stream_set_pt_map (stream, pt, caps);
3533       gst_caps_unref (caps);
3534     }
3535
3536     control = gst_sdp_media_get_attribute_val (sdp_media, "control");
3537     if (control)
3538       gst_rtsp_stream_set_control (stream, control);
3539
3540   }
3541
3542   return TRUE;
3543 }
3544
3545 /**
3546  * gst_rtsp_media_handle_sdp:
3547  * @media: a #GstRTSPMedia
3548  * @sdp: (transfer none): a #GstSDPMessage
3549  *
3550  * Configure an SDP on @media for receiving streams
3551  *
3552  * Returns: TRUE on success.
3553  */
3554 gboolean
3555 gst_rtsp_media_handle_sdp (GstRTSPMedia * media, GstSDPMessage * sdp)
3556 {
3557   GstRTSPMediaPrivate *priv;
3558   GstRTSPMediaClass *klass;
3559   gboolean res;
3560
3561   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
3562   g_return_val_if_fail (sdp != NULL, FALSE);
3563
3564   priv = media->priv;
3565
3566   g_rec_mutex_lock (&priv->state_lock);
3567
3568   klass = GST_RTSP_MEDIA_GET_CLASS (media);
3569
3570   if (!klass->handle_sdp)
3571     goto no_handle_sdp;
3572
3573   res = klass->handle_sdp (media, sdp);
3574
3575   g_rec_mutex_unlock (&priv->state_lock);
3576
3577   return res;
3578
3579   /* ERRORS */
3580 no_handle_sdp:
3581   {
3582     g_rec_mutex_unlock (&priv->state_lock);
3583     GST_ERROR ("no handle_sdp function");
3584     g_critical ("no handle_sdp vmethod function set");
3585     return FALSE;
3586   }
3587 }
3588
3589 static void
3590 do_set_seqnum (GstRTSPStream * stream)
3591 {
3592   guint16 seq_num;
3593   seq_num = gst_rtsp_stream_get_current_seqnum (stream);
3594   gst_rtsp_stream_set_seqnum_offset (stream, seq_num + 1);
3595 }
3596
3597 /* call with state_lock */
3598 gboolean
3599 default_suspend (GstRTSPMedia * media)
3600 {
3601   GstRTSPMediaPrivate *priv = media->priv;
3602   GstStateChangeReturn ret;
3603
3604   switch (priv->suspend_mode) {
3605     case GST_RTSP_SUSPEND_MODE_NONE:
3606       GST_DEBUG ("media %p no suspend", media);
3607       break;
3608     case GST_RTSP_SUSPEND_MODE_PAUSE:
3609       GST_DEBUG ("media %p suspend to PAUSED", media);
3610       ret = set_target_state (media, GST_STATE_PAUSED, TRUE);
3611       if (ret == GST_STATE_CHANGE_FAILURE)
3612         goto state_failed;
3613       break;
3614     case GST_RTSP_SUSPEND_MODE_RESET:
3615       GST_DEBUG ("media %p suspend to NULL", media);
3616       ret = set_target_state (media, GST_STATE_NULL, TRUE);
3617       if (ret == GST_STATE_CHANGE_FAILURE)
3618         goto state_failed;
3619       /* Because payloader needs to set the sequence number as
3620        * monotonic, we need to preserve the sequence number
3621        * after pause. (otherwise going from pause to play,  which
3622        * is actually from NULL to PLAY will create a new sequence
3623        * number. */
3624       g_ptr_array_foreach (priv->streams, (GFunc) do_set_seqnum, NULL);
3625       break;
3626     default:
3627       break;
3628   }
3629
3630   /* let the streams do the state changes freely, if any */
3631   media_streams_set_blocked (media, FALSE);
3632
3633   return TRUE;
3634
3635   /* ERRORS */
3636 state_failed:
3637   {
3638     GST_WARNING ("failed changing pipeline's state for media %p", media);
3639     return FALSE;
3640   }
3641 }
3642
3643 /**
3644  * gst_rtsp_media_suspend:
3645  * @media: a #GstRTSPMedia
3646  *
3647  * Suspend @media. The state of the pipeline managed by @media is set to
3648  * GST_STATE_NULL but all streams are kept. @media can be prepared again
3649  * with gst_rtsp_media_unsuspend()
3650  *
3651  * @media must be prepared with gst_rtsp_media_prepare();
3652  *
3653  * Returns: %TRUE on success.
3654  */
3655 gboolean
3656 gst_rtsp_media_suspend (GstRTSPMedia * media)
3657 {
3658   GstRTSPMediaPrivate *priv = media->priv;
3659   GstRTSPMediaClass *klass;
3660
3661   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
3662
3663   GST_FIXME ("suspend for dynamic pipelines needs fixing");
3664
3665   g_rec_mutex_lock (&priv->state_lock);
3666   if (priv->status != GST_RTSP_MEDIA_STATUS_PREPARED)
3667     goto not_prepared;
3668
3669   /* don't attempt to suspend when something is busy */
3670   if (priv->n_active > 0)
3671     goto done;
3672
3673   klass = GST_RTSP_MEDIA_GET_CLASS (media);
3674   if (klass->suspend) {
3675     if (!klass->suspend (media))
3676       goto suspend_failed;
3677   }
3678
3679   gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_SUSPENDED);
3680 done:
3681   g_rec_mutex_unlock (&priv->state_lock);
3682
3683   return TRUE;
3684
3685   /* ERRORS */
3686 not_prepared:
3687   {
3688     g_rec_mutex_unlock (&priv->state_lock);
3689     GST_WARNING ("media %p was not prepared", media);
3690     return FALSE;
3691   }
3692 suspend_failed:
3693   {
3694     g_rec_mutex_unlock (&priv->state_lock);
3695     gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_ERROR);
3696     GST_WARNING ("failed to suspend media %p", media);
3697     return FALSE;
3698   }
3699 }
3700
3701 /* call with state_lock */
3702 gboolean
3703 default_unsuspend (GstRTSPMedia * media)
3704 {
3705   GstRTSPMediaPrivate *priv = media->priv;
3706
3707   switch (priv->suspend_mode) {
3708     case GST_RTSP_SUSPEND_MODE_NONE:
3709       gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_PREPARED);
3710       break;
3711     case GST_RTSP_SUSPEND_MODE_PAUSE:
3712       gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_PREPARED);
3713       break;
3714     case GST_RTSP_SUSPEND_MODE_RESET:
3715     {
3716       gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_PREPARING);
3717       if (!start_preroll (media))
3718         goto start_failed;
3719       g_rec_mutex_unlock (&priv->state_lock);
3720
3721       if (!wait_preroll (media))
3722         goto preroll_failed;
3723
3724       g_rec_mutex_lock (&priv->state_lock);
3725     }
3726     default:
3727       break;
3728   }
3729
3730   return TRUE;
3731
3732   /* ERRORS */
3733 start_failed:
3734   {
3735     GST_WARNING ("failed to preroll pipeline");
3736     return FALSE;
3737   }
3738 preroll_failed:
3739   {
3740     GST_WARNING ("failed to preroll pipeline");
3741     return FALSE;
3742   }
3743 }
3744
3745 /**
3746  * gst_rtsp_media_unsuspend:
3747  * @media: a #GstRTSPMedia
3748  *
3749  * Unsuspend @media if it was in a suspended state. This method does nothing
3750  * when the media was not in the suspended state.
3751  *
3752  * Returns: %TRUE on success.
3753  */
3754 gboolean
3755 gst_rtsp_media_unsuspend (GstRTSPMedia * media)
3756 {
3757   GstRTSPMediaPrivate *priv = media->priv;
3758   GstRTSPMediaClass *klass;
3759
3760   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
3761
3762   g_rec_mutex_lock (&priv->state_lock);
3763   if (priv->status != GST_RTSP_MEDIA_STATUS_SUSPENDED)
3764     goto done;
3765
3766   klass = GST_RTSP_MEDIA_GET_CLASS (media);
3767   if (klass->unsuspend) {
3768     if (!klass->unsuspend (media))
3769       goto unsuspend_failed;
3770   }
3771
3772 done:
3773   g_rec_mutex_unlock (&priv->state_lock);
3774
3775   return TRUE;
3776
3777   /* ERRORS */
3778 unsuspend_failed:
3779   {
3780     g_rec_mutex_unlock (&priv->state_lock);
3781     GST_WARNING ("failed to unsuspend media %p", media);
3782     gst_rtsp_media_set_status (media, GST_RTSP_MEDIA_STATUS_ERROR);
3783     return FALSE;
3784   }
3785 }
3786
3787 /* must be called with state-lock */
3788 static void
3789 media_set_pipeline_state_locked (GstRTSPMedia * media, GstState state)
3790 {
3791   GstRTSPMediaPrivate *priv = media->priv;
3792
3793   if (state == GST_STATE_NULL) {
3794     gst_rtsp_media_unprepare (media);
3795   } else {
3796     GST_INFO ("state %s media %p", gst_element_state_get_name (state), media);
3797     set_target_state (media, state, FALSE);
3798     /* when we are buffering, don't update the state yet, this will be done
3799      * when buffering finishes */
3800     if (priv->buffering) {
3801       GST_INFO ("Buffering busy, delay state change");
3802     } else {
3803       if (state == GST_STATE_PLAYING)
3804         /* make sure pads are not blocking anymore when going to PLAYING */
3805         media_streams_set_blocked (media, FALSE);
3806
3807       set_state (media, state);
3808
3809       /* and suspend after pause */
3810       if (state == GST_STATE_PAUSED)
3811         gst_rtsp_media_suspend (media);
3812     }
3813   }
3814 }
3815
3816 /**
3817  * gst_rtsp_media_set_pipeline_state:
3818  * @media: a #GstRTSPMedia
3819  * @state: the target state of the pipeline
3820  *
3821  * Set the state of the pipeline managed by @media to @state
3822  */
3823 void
3824 gst_rtsp_media_set_pipeline_state (GstRTSPMedia * media, GstState state)
3825 {
3826   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
3827
3828   g_rec_mutex_lock (&media->priv->state_lock);
3829   media_set_pipeline_state_locked (media, state);
3830   g_rec_mutex_unlock (&media->priv->state_lock);
3831 }
3832
3833 /**
3834  * gst_rtsp_media_set_state:
3835  * @media: a #GstRTSPMedia
3836  * @state: the target state of the media
3837  * @transports: (transfer none) (element-type GstRtspServer.RTSPStreamTransport):
3838  * a #GPtrArray of #GstRTSPStreamTransport pointers
3839  *
3840  * Set the state of @media to @state and for the transports in @transports.
3841  *
3842  * @media must be prepared with gst_rtsp_media_prepare();
3843  *
3844  * Returns: %TRUE on success.
3845  */
3846 gboolean
3847 gst_rtsp_media_set_state (GstRTSPMedia * media, GstState state,
3848     GPtrArray * transports)
3849 {
3850   GstRTSPMediaPrivate *priv;
3851   gint i;
3852   gboolean activate, deactivate, do_state;
3853   gint old_active;
3854
3855   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
3856   g_return_val_if_fail (transports != NULL, FALSE);
3857
3858   priv = media->priv;
3859
3860   g_rec_mutex_lock (&priv->state_lock);
3861   if (priv->status == GST_RTSP_MEDIA_STATUS_ERROR)
3862     goto error_status;
3863   if (priv->status != GST_RTSP_MEDIA_STATUS_PREPARED &&
3864       priv->status != GST_RTSP_MEDIA_STATUS_SUSPENDED)
3865     goto not_prepared;
3866
3867   /* NULL and READY are the same */
3868   if (state == GST_STATE_READY)
3869     state = GST_STATE_NULL;
3870
3871   activate = deactivate = FALSE;
3872
3873   GST_INFO ("going to state %s media %p, target state %s",
3874       gst_element_state_get_name (state), media,
3875       gst_element_state_get_name (priv->target_state));
3876
3877   switch (state) {
3878     case GST_STATE_NULL:
3879       /* we're going from PLAYING or PAUSED to READY or NULL, deactivate */
3880       if (priv->target_state >= GST_STATE_PAUSED)
3881         deactivate = TRUE;
3882       break;
3883     case GST_STATE_PAUSED:
3884       /* we're going from PLAYING to PAUSED, READY or NULL, deactivate */
3885       if (priv->target_state == GST_STATE_PLAYING)
3886         deactivate = TRUE;
3887       break;
3888     case GST_STATE_PLAYING:
3889       /* we're going to PLAYING, activate */
3890       activate = TRUE;
3891       break;
3892     default:
3893       break;
3894   }
3895   old_active = priv->n_active;
3896
3897   GST_DEBUG ("%d transports, activate %d, deactivate %d", transports->len,
3898       activate, deactivate);
3899   for (i = 0; i < transports->len; i++) {
3900     GstRTSPStreamTransport *trans;
3901
3902     /* we need a non-NULL entry in the array */
3903     trans = g_ptr_array_index (transports, i);
3904     if (trans == NULL)
3905       continue;
3906
3907     if (activate) {
3908       if (gst_rtsp_stream_transport_set_active (trans, TRUE))
3909         priv->n_active++;
3910     } else if (deactivate) {
3911       if (gst_rtsp_stream_transport_set_active (trans, FALSE))
3912         priv->n_active--;
3913     }
3914   }
3915
3916   /* we just activated the first media, do the playing state change */
3917   if (old_active == 0 && activate)
3918     do_state = TRUE;
3919   /* if we have no more active media, do the downward state changes */
3920   else if (priv->n_active == 0)
3921     do_state = TRUE;
3922   else
3923     do_state = FALSE;
3924
3925   GST_INFO ("state %d active %d media %p do_state %d", state, priv->n_active,
3926       media, do_state);
3927
3928   if (priv->target_state != state) {
3929     if (do_state)
3930       media_set_pipeline_state_locked (media, state);
3931
3932     g_signal_emit (media, gst_rtsp_media_signals[SIGNAL_NEW_STATE], 0, state,
3933         NULL);
3934   }
3935
3936   /* remember where we are */
3937   if (state != GST_STATE_NULL && (state == GST_STATE_PAUSED ||
3938           old_active != priv->n_active))
3939     collect_media_stats (media);
3940
3941   g_rec_mutex_unlock (&priv->state_lock);
3942
3943   return TRUE;
3944
3945   /* ERRORS */
3946 not_prepared:
3947   {
3948     GST_WARNING ("media %p was not prepared", media);
3949     g_rec_mutex_unlock (&priv->state_lock);
3950     return FALSE;
3951   }
3952 error_status:
3953   {
3954     GST_WARNING ("media %p in error status while changing to state %d",
3955         media, state);
3956     if (state == GST_STATE_NULL) {
3957       for (i = 0; i < transports->len; i++) {
3958         GstRTSPStreamTransport *trans;
3959
3960         /* we need a non-NULL entry in the array */
3961         trans = g_ptr_array_index (transports, i);
3962         if (trans == NULL)
3963           continue;
3964
3965         gst_rtsp_stream_transport_set_active (trans, FALSE);
3966       }
3967       priv->n_active = 0;
3968     }
3969     g_rec_mutex_unlock (&priv->state_lock);
3970     return FALSE;
3971   }
3972 }
3973
3974 /**
3975  * gst_rtsp_media_set_transport_mode:
3976  * @media: a #GstRTSPMedia
3977  * @mode: the new value
3978  *
3979  * Sets if the media pipeline can work in PLAY or RECORD mode
3980  */
3981 void
3982 gst_rtsp_media_set_transport_mode (GstRTSPMedia * media,
3983     GstRTSPTransportMode mode)
3984 {
3985   GstRTSPMediaPrivate *priv;
3986
3987   g_return_if_fail (GST_IS_RTSP_MEDIA (media));
3988
3989   priv = media->priv;
3990
3991   g_mutex_lock (&priv->lock);
3992   priv->transport_mode = mode;
3993   g_mutex_unlock (&priv->lock);
3994 }
3995
3996 /**
3997  * gst_rtsp_media_get_transport_mode:
3998  * @media: a #GstRTSPMedia
3999  *
4000  * Check if the pipeline for @media can be used for PLAY or RECORD methods.
4001  *
4002  * Returns: The transport mode.
4003  */
4004 GstRTSPTransportMode
4005 gst_rtsp_media_get_transport_mode (GstRTSPMedia * media)
4006 {
4007   GstRTSPMediaPrivate *priv;
4008   GstRTSPTransportMode res;
4009
4010   g_return_val_if_fail (GST_IS_RTSP_MEDIA (media), FALSE);
4011
4012   priv = media->priv;
4013
4014   g_mutex_lock (&priv->lock);
4015   res = priv->transport_mode;
4016   g_mutex_unlock (&priv->lock);
4017
4018   return res;
4019 }