media: convert_range replaces get_range_times
[platform/upstream/gstreamer.git] / gst / rtsp-server / rtsp-client.c
1 /* GStreamer
2  * Copyright (C) 2008 Wim Taymans <wim.taymans at gmail.com>
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Library General Public
6  * License as published by the Free Software Foundation; either
7  * version 2 of the License, or (at your option) any later version.
8  *
9  * This library is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
12  * Library General Public License for more details.
13  *
14  * You should have received a copy of the GNU Library General Public
15  * License along with this library; if not, write to the
16  * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
17  * Boston, MA 02110-1301, USA.
18  */
19
20 #include <stdio.h>
21 #include <string.h>
22
23 #include "rtsp-client.h"
24 #include "rtsp-sdp.h"
25 #include "rtsp-params.h"
26
27 #define GST_RTSP_CLIENT_GET_PRIVATE(obj)  \
28    (G_TYPE_INSTANCE_GET_PRIVATE ((obj), GST_TYPE_RTSP_CLIENT, GstRTSPClientPrivate))
29
30 /* locking order:
31  * send_lock, lock, tunnels_lock
32  */
33
34 struct _GstRTSPClientPrivate
35 {
36   GMutex lock;                  /* protects everything else */
37   GMutex send_lock;
38   GstRTSPConnection *connection;
39   GstRTSPWatch *watch;
40   guint close_seq;
41   gchar *server_ip;
42   gboolean is_ipv6;
43   gboolean use_client_settings;
44
45   GstRTSPClientSendFunc send_func;      /* protected by send_lock */
46   gpointer send_data;           /* protected by send_lock */
47   GDestroyNotify send_notify;   /* protected by send_lock */
48
49   GstRTSPSessionPool *session_pool;
50   GstRTSPMountPoints *mount_points;
51   GstRTSPAuth *auth;
52
53   GstRTSPUrl *uri;
54   GstRTSPMedia *media;
55
56   GList *transports;
57   GList *sessions;
58 };
59
60 static GMutex tunnels_lock;
61 static GHashTable *tunnels;     /* protected by tunnels_lock */
62
63 #define DEFAULT_SESSION_POOL            NULL
64 #define DEFAULT_MOUNT_POINTS            NULL
65 #define DEFAULT_USE_CLIENT_SETTINGS     FALSE
66
67 enum
68 {
69   PROP_0,
70   PROP_SESSION_POOL,
71   PROP_MOUNT_POINTS,
72   PROP_USE_CLIENT_SETTINGS,
73   PROP_LAST
74 };
75
76 enum
77 {
78   SIGNAL_CLOSED,
79   SIGNAL_NEW_SESSION,
80   SIGNAL_OPTIONS_REQUEST,
81   SIGNAL_DESCRIBE_REQUEST,
82   SIGNAL_SETUP_REQUEST,
83   SIGNAL_PLAY_REQUEST,
84   SIGNAL_PAUSE_REQUEST,
85   SIGNAL_TEARDOWN_REQUEST,
86   SIGNAL_SET_PARAMETER_REQUEST,
87   SIGNAL_GET_PARAMETER_REQUEST,
88   SIGNAL_LAST
89 };
90
91 GST_DEBUG_CATEGORY_STATIC (rtsp_client_debug);
92 #define GST_CAT_DEFAULT rtsp_client_debug
93
94 static guint gst_rtsp_client_signals[SIGNAL_LAST] = { 0 };
95
96 static void gst_rtsp_client_get_property (GObject * object, guint propid,
97     GValue * value, GParamSpec * pspec);
98 static void gst_rtsp_client_set_property (GObject * object, guint propid,
99     const GValue * value, GParamSpec * pspec);
100 static void gst_rtsp_client_finalize (GObject * obj);
101
102 static GstSDPMessage *create_sdp (GstRTSPClient * client, GstRTSPMedia * media);
103 static void client_session_finalized (GstRTSPClient * client,
104     GstRTSPSession * session);
105 static void unlink_session_transports (GstRTSPClient * client,
106     GstRTSPSession * session, GstRTSPSessionMedia * media);
107
108 G_DEFINE_TYPE (GstRTSPClient, gst_rtsp_client, G_TYPE_OBJECT);
109
110 static void
111 gst_rtsp_client_class_init (GstRTSPClientClass * klass)
112 {
113   GObjectClass *gobject_class;
114
115   g_type_class_add_private (klass, sizeof (GstRTSPClientPrivate));
116
117   gobject_class = G_OBJECT_CLASS (klass);
118
119   gobject_class->get_property = gst_rtsp_client_get_property;
120   gobject_class->set_property = gst_rtsp_client_set_property;
121   gobject_class->finalize = gst_rtsp_client_finalize;
122
123   klass->create_sdp = create_sdp;
124
125   g_object_class_install_property (gobject_class, PROP_SESSION_POOL,
126       g_param_spec_object ("session-pool", "Session Pool",
127           "The session pool to use for client session",
128           GST_TYPE_RTSP_SESSION_POOL,
129           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
130
131   g_object_class_install_property (gobject_class, PROP_MOUNT_POINTS,
132       g_param_spec_object ("mount-points", "Mount Points",
133           "The mount points to use for client session",
134           GST_TYPE_RTSP_MOUNT_POINTS,
135           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
136
137   g_object_class_install_property (gobject_class, PROP_USE_CLIENT_SETTINGS,
138       g_param_spec_boolean ("use-client-settings", "Use Client Settings",
139           "Use client settings for ttl and destination in multicast",
140           DEFAULT_USE_CLIENT_SETTINGS,
141           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
142
143   gst_rtsp_client_signals[SIGNAL_CLOSED] =
144       g_signal_new ("closed", G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
145       G_STRUCT_OFFSET (GstRTSPClientClass, closed), NULL, NULL,
146       g_cclosure_marshal_VOID__VOID, G_TYPE_NONE, 0, G_TYPE_NONE);
147
148   gst_rtsp_client_signals[SIGNAL_NEW_SESSION] =
149       g_signal_new ("new-session", G_TYPE_FROM_CLASS (klass), G_SIGNAL_RUN_LAST,
150       G_STRUCT_OFFSET (GstRTSPClientClass, new_session), NULL, NULL,
151       g_cclosure_marshal_VOID__OBJECT, G_TYPE_NONE, 1, GST_TYPE_RTSP_SESSION);
152
153   gst_rtsp_client_signals[SIGNAL_OPTIONS_REQUEST] =
154       g_signal_new ("options-request", G_TYPE_FROM_CLASS (klass),
155       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPClientClass, options_request),
156       NULL, NULL, g_cclosure_marshal_VOID__POINTER, G_TYPE_NONE, 1,
157       G_TYPE_POINTER);
158
159   gst_rtsp_client_signals[SIGNAL_DESCRIBE_REQUEST] =
160       g_signal_new ("describe-request", G_TYPE_FROM_CLASS (klass),
161       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPClientClass, describe_request),
162       NULL, NULL, g_cclosure_marshal_VOID__POINTER, G_TYPE_NONE, 1,
163       G_TYPE_POINTER);
164
165   gst_rtsp_client_signals[SIGNAL_SETUP_REQUEST] =
166       g_signal_new ("setup-request", G_TYPE_FROM_CLASS (klass),
167       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPClientClass, setup_request),
168       NULL, NULL, g_cclosure_marshal_VOID__POINTER, G_TYPE_NONE, 1,
169       G_TYPE_POINTER);
170
171   gst_rtsp_client_signals[SIGNAL_PLAY_REQUEST] =
172       g_signal_new ("play-request", G_TYPE_FROM_CLASS (klass),
173       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPClientClass, play_request),
174       NULL, NULL, g_cclosure_marshal_VOID__POINTER, G_TYPE_NONE, 1,
175       G_TYPE_POINTER);
176
177   gst_rtsp_client_signals[SIGNAL_PAUSE_REQUEST] =
178       g_signal_new ("pause-request", G_TYPE_FROM_CLASS (klass),
179       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPClientClass, pause_request),
180       NULL, NULL, g_cclosure_marshal_VOID__POINTER, G_TYPE_NONE, 1,
181       G_TYPE_POINTER);
182
183   gst_rtsp_client_signals[SIGNAL_TEARDOWN_REQUEST] =
184       g_signal_new ("teardown-request", G_TYPE_FROM_CLASS (klass),
185       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPClientClass, teardown_request),
186       NULL, NULL, g_cclosure_marshal_VOID__POINTER, G_TYPE_NONE, 1,
187       G_TYPE_POINTER);
188
189   gst_rtsp_client_signals[SIGNAL_SET_PARAMETER_REQUEST] =
190       g_signal_new ("set-parameter-request", G_TYPE_FROM_CLASS (klass),
191       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPClientClass,
192           set_parameter_request), NULL, NULL, g_cclosure_marshal_VOID__POINTER,
193       G_TYPE_NONE, 1, G_TYPE_POINTER);
194
195   gst_rtsp_client_signals[SIGNAL_GET_PARAMETER_REQUEST] =
196       g_signal_new ("get-parameter-request", G_TYPE_FROM_CLASS (klass),
197       G_SIGNAL_RUN_LAST, G_STRUCT_OFFSET (GstRTSPClientClass,
198           get_parameter_request), NULL, NULL, g_cclosure_marshal_VOID__POINTER,
199       G_TYPE_NONE, 1, G_TYPE_POINTER);
200
201   tunnels =
202       g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_object_unref);
203   g_mutex_init (&tunnels_lock);
204
205   GST_DEBUG_CATEGORY_INIT (rtsp_client_debug, "rtspclient", 0, "GstRTSPClient");
206 }
207
208 static void
209 gst_rtsp_client_init (GstRTSPClient * client)
210 {
211   GstRTSPClientPrivate *priv = GST_RTSP_CLIENT_GET_PRIVATE (client);
212
213   client->priv = priv;
214
215   g_mutex_init (&priv->lock);
216   g_mutex_init (&priv->send_lock);
217   priv->use_client_settings = DEFAULT_USE_CLIENT_SETTINGS;
218   priv->close_seq = 0;
219 }
220
221 static GstRTSPFilterResult
222 filter_session (GstRTSPSession * sess, GstRTSPSessionMedia * media,
223     gpointer user_data)
224 {
225   GstRTSPClient *client = GST_RTSP_CLIENT (user_data);
226
227   gst_rtsp_session_media_set_state (media, GST_STATE_NULL);
228   unlink_session_transports (client, sess, media);
229
230   /* unmanage the media in the session */
231   return GST_RTSP_FILTER_REMOVE;
232 }
233
234 static void
235 client_unlink_session (GstRTSPClient * client, GstRTSPSession * session)
236 {
237   /* unlink all media managed in this session */
238   gst_rtsp_session_filter (session, filter_session, client);
239 }
240
241 static void
242 client_cleanup_sessions (GstRTSPClient * client)
243 {
244   GstRTSPClientPrivate *priv = client->priv;
245   GList *sessions;
246
247   /* remove weak-ref from sessions */
248   for (sessions = priv->sessions; sessions; sessions = g_list_next (sessions)) {
249     GstRTSPSession *session = (GstRTSPSession *) sessions->data;
250     g_object_weak_unref (G_OBJECT (session),
251         (GWeakNotify) client_session_finalized, client);
252     client_unlink_session (client, session);
253   }
254   g_list_free (priv->sessions);
255   priv->sessions = NULL;
256 }
257
258 /* A client is finalized when the connection is broken */
259 static void
260 gst_rtsp_client_finalize (GObject * obj)
261 {
262   GstRTSPClient *client = GST_RTSP_CLIENT (obj);
263   GstRTSPClientPrivate *priv = client->priv;
264
265   GST_INFO ("finalize client %p", client);
266
267   gst_rtsp_client_set_send_func (client, NULL, NULL, NULL);
268
269   if (priv->watch)
270     g_source_destroy ((GSource *) priv->watch);
271
272   client_cleanup_sessions (client);
273
274   if (priv->connection)
275     gst_rtsp_connection_free (priv->connection);
276   if (priv->session_pool)
277     g_object_unref (priv->session_pool);
278   if (priv->mount_points)
279     g_object_unref (priv->mount_points);
280   if (priv->auth)
281     g_object_unref (priv->auth);
282
283   if (priv->uri)
284     gst_rtsp_url_free (priv->uri);
285   if (priv->media) {
286     gst_rtsp_media_unprepare (priv->media);
287     g_object_unref (priv->media);
288   }
289
290   g_free (priv->server_ip);
291   g_mutex_clear (&priv->lock);
292   g_mutex_clear (&priv->send_lock);
293
294   G_OBJECT_CLASS (gst_rtsp_client_parent_class)->finalize (obj);
295 }
296
297 static void
298 gst_rtsp_client_get_property (GObject * object, guint propid,
299     GValue * value, GParamSpec * pspec)
300 {
301   GstRTSPClient *client = GST_RTSP_CLIENT (object);
302
303   switch (propid) {
304     case PROP_SESSION_POOL:
305       g_value_take_object (value, gst_rtsp_client_get_session_pool (client));
306       break;
307     case PROP_MOUNT_POINTS:
308       g_value_take_object (value, gst_rtsp_client_get_mount_points (client));
309       break;
310     case PROP_USE_CLIENT_SETTINGS:
311       g_value_set_boolean (value,
312           gst_rtsp_client_get_use_client_settings (client));
313       break;
314     default:
315       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
316   }
317 }
318
319 static void
320 gst_rtsp_client_set_property (GObject * object, guint propid,
321     const GValue * value, GParamSpec * pspec)
322 {
323   GstRTSPClient *client = GST_RTSP_CLIENT (object);
324
325   switch (propid) {
326     case PROP_SESSION_POOL:
327       gst_rtsp_client_set_session_pool (client, g_value_get_object (value));
328       break;
329     case PROP_MOUNT_POINTS:
330       gst_rtsp_client_set_mount_points (client, g_value_get_object (value));
331       break;
332     case PROP_USE_CLIENT_SETTINGS:
333       gst_rtsp_client_set_use_client_settings (client,
334           g_value_get_boolean (value));
335       break;
336     default:
337       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
338   }
339 }
340
341 /**
342  * gst_rtsp_client_new:
343  *
344  * Create a new #GstRTSPClient instance.
345  *
346  * Returns: a new #GstRTSPClient
347  */
348 GstRTSPClient *
349 gst_rtsp_client_new (void)
350 {
351   GstRTSPClient *result;
352
353   result = g_object_new (GST_TYPE_RTSP_CLIENT, NULL);
354
355   return result;
356 }
357
358 static void
359 send_response (GstRTSPClient * client, GstRTSPSession * session,
360     GstRTSPMessage * response, gboolean close)
361 {
362   GstRTSPClientPrivate *priv = client->priv;
363
364   gst_rtsp_message_add_header (response, GST_RTSP_HDR_SERVER,
365       "GStreamer RTSP server");
366
367   /* remove any previous header */
368   gst_rtsp_message_remove_header (response, GST_RTSP_HDR_SESSION, -1);
369
370   /* add the new session header for new session ids */
371   if (session) {
372     gst_rtsp_message_take_header (response, GST_RTSP_HDR_SESSION,
373         gst_rtsp_session_get_header (session));
374   }
375
376   if (gst_debug_category_get_threshold (rtsp_client_debug) >= GST_LEVEL_LOG) {
377     gst_rtsp_message_dump (response);
378   }
379
380   if (close)
381     gst_rtsp_message_add_header (response, GST_RTSP_HDR_CONNECTION, "close");
382
383   g_mutex_lock (&priv->send_lock);
384   if (priv->send_func)
385     priv->send_func (client, response, close, priv->send_data);
386   g_mutex_unlock (&priv->send_lock);
387
388   gst_rtsp_message_unset (response);
389 }
390
391 static void
392 send_generic_response (GstRTSPClient * client, GstRTSPStatusCode code,
393     GstRTSPClientState * state)
394 {
395   gst_rtsp_message_init_response (state->response, code,
396       gst_rtsp_status_as_text (code), state->request);
397
398   send_response (client, NULL, state->response, FALSE);
399 }
400
401 static void
402 handle_unauthorized_request (GstRTSPClient * client, GstRTSPAuth * auth,
403     GstRTSPClientState * state)
404 {
405   gst_rtsp_message_init_response (state->response, GST_RTSP_STS_UNAUTHORIZED,
406       gst_rtsp_status_as_text (GST_RTSP_STS_UNAUTHORIZED), state->request);
407
408   if (auth) {
409     /* and let the authentication manager setup the auth tokens */
410     gst_rtsp_auth_setup_auth (auth, client, 0, state);
411   }
412
413   send_response (client, state->session, state->response, FALSE);
414 }
415
416
417 static gboolean
418 compare_uri (const GstRTSPUrl * uri1, const GstRTSPUrl * uri2)
419 {
420   if (uri1 == NULL || uri2 == NULL)
421     return FALSE;
422
423   if (strcmp (uri1->abspath, uri2->abspath))
424     return FALSE;
425
426   return TRUE;
427 }
428
429 /* this function is called to initially find the media for the DESCRIBE request
430  * but is cached for when the same client (without breaking the connection) is
431  * doing a setup for the exact same url. */
432 static GstRTSPMedia *
433 find_media (GstRTSPClient * client, GstRTSPClientState * state)
434 {
435   GstRTSPClientPrivate *priv = client->priv;
436   GstRTSPMediaFactory *factory;
437   GstRTSPMedia *media;
438   GstRTSPAuth *auth;
439
440   if (!compare_uri (priv->uri, state->uri)) {
441     /* remove any previously cached values before we try to construct a new
442      * media for uri */
443     if (priv->uri)
444       gst_rtsp_url_free (priv->uri);
445     priv->uri = NULL;
446     if (priv->media) {
447       gst_rtsp_media_unprepare (priv->media);
448       g_object_unref (priv->media);
449     }
450     priv->media = NULL;
451
452     if (!priv->mount_points)
453       goto no_mount_points;
454
455     /* find the factory for the uri first */
456     if (!(factory =
457             gst_rtsp_mount_points_find_factory (priv->mount_points,
458                 state->uri)))
459       goto no_factory;
460
461     /* check if we have access to the factory */
462     if ((auth = gst_rtsp_media_factory_get_auth (factory))) {
463       state->factory = factory;
464
465       if (!gst_rtsp_auth_check (auth, client, 0, state))
466         goto not_allowed;
467
468       state->factory = NULL;
469       g_object_unref (auth);
470     }
471
472     /* prepare the media and add it to the pipeline */
473     if (!(media = gst_rtsp_media_factory_construct (factory, state->uri)))
474       goto no_media;
475
476     g_object_unref (factory);
477     factory = NULL;
478
479     /* prepare the media */
480     if (!(gst_rtsp_media_prepare (media)))
481       goto no_prepare;
482
483     /* now keep track of the uri and the media */
484     priv->uri = gst_rtsp_url_copy (state->uri);
485     priv->media = media;
486     state->media = media;
487   } else {
488     /* we have seen this uri before, used cached media */
489     media = priv->media;
490     state->media = media;
491     GST_INFO ("reusing cached media %p", media);
492   }
493
494   if (media)
495     g_object_ref (media);
496
497   return media;
498
499   /* ERRORS */
500 no_mount_points:
501   {
502     GST_ERROR ("client %p: no mount points configured", client);
503     send_generic_response (client, GST_RTSP_STS_NOT_FOUND, state);
504     return NULL;
505   }
506 no_factory:
507   {
508     GST_ERROR ("client %p: no factory for uri", client);
509     send_generic_response (client, GST_RTSP_STS_NOT_FOUND, state);
510     return NULL;
511   }
512 not_allowed:
513   {
514     GST_ERROR ("client %p: unauthorized request", client);
515     handle_unauthorized_request (client, auth, state);
516     g_object_unref (factory);
517     state->factory = NULL;
518     g_object_unref (auth);
519     return NULL;
520   }
521 no_media:
522   {
523     GST_ERROR ("client %p: can't create media", client);
524     send_generic_response (client, GST_RTSP_STS_SERVICE_UNAVAILABLE, state);
525     g_object_unref (factory);
526     return NULL;
527   }
528 no_prepare:
529   {
530     GST_ERROR ("client %p: can't prepare media", client);
531     send_generic_response (client, GST_RTSP_STS_SERVICE_UNAVAILABLE, state);
532     g_object_unref (media);
533     return NULL;
534   }
535 }
536
537 static gboolean
538 do_send_data (GstBuffer * buffer, guint8 channel, GstRTSPClient * client)
539 {
540   GstRTSPClientPrivate *priv = client->priv;
541   GstRTSPMessage message = { 0 };
542   GstMapInfo map_info;
543   guint8 *data;
544   guint usize;
545
546   gst_rtsp_message_init_data (&message, channel);
547
548   /* FIXME, need some sort of iovec RTSPMessage here */
549   if (!gst_buffer_map (buffer, &map_info, GST_MAP_READ))
550     return FALSE;
551
552   gst_rtsp_message_take_body (&message, map_info.data, map_info.size);
553
554   g_mutex_lock (&priv->send_lock);
555   if (priv->send_func)
556     priv->send_func (client, &message, FALSE, priv->send_data);
557   g_mutex_unlock (&priv->send_lock);
558
559   gst_rtsp_message_steal_body (&message, &data, &usize);
560   gst_buffer_unmap (buffer, &map_info);
561
562   gst_rtsp_message_unset (&message);
563
564   return TRUE;
565 }
566
567 static void
568 link_transport (GstRTSPClient * client, GstRTSPSession * session,
569     GstRTSPStreamTransport * trans)
570 {
571   GstRTSPClientPrivate *priv = client->priv;
572
573   GST_DEBUG ("client %p: linking transport %p", client, trans);
574
575   gst_rtsp_stream_transport_set_callbacks (trans,
576       (GstRTSPSendFunc) do_send_data,
577       (GstRTSPSendFunc) do_send_data, client, NULL);
578
579   priv->transports = g_list_prepend (priv->transports, trans);
580
581   /* make sure our session can't expire */
582   gst_rtsp_session_prevent_expire (session);
583 }
584
585 static void
586 unlink_transport (GstRTSPClient * client, GstRTSPSession * session,
587     GstRTSPStreamTransport * trans)
588 {
589   GstRTSPClientPrivate *priv = client->priv;
590
591   GST_DEBUG ("client %p: unlinking transport %p", client, trans);
592
593   gst_rtsp_stream_transport_set_callbacks (trans, NULL, NULL, NULL, NULL);
594
595   priv->transports = g_list_remove (priv->transports, trans);
596
597   /* our session can now expire */
598   gst_rtsp_session_allow_expire (session);
599 }
600
601 static void
602 unlink_session_transports (GstRTSPClient * client, GstRTSPSession * session,
603     GstRTSPSessionMedia * media)
604 {
605   guint n_streams, i;
606
607   n_streams =
608       gst_rtsp_media_n_streams (gst_rtsp_session_media_get_media (media));
609   for (i = 0; i < n_streams; i++) {
610     GstRTSPStreamTransport *trans;
611     const GstRTSPTransport *tr;
612
613     /* get the transport, if there is no transport configured, skip this stream */
614     trans = gst_rtsp_session_media_get_transport (media, i);
615     if (trans == NULL)
616       continue;
617
618     tr = gst_rtsp_stream_transport_get_transport (trans);
619
620     if (tr->lower_transport == GST_RTSP_LOWER_TRANS_TCP) {
621       /* for TCP, unlink the stream from the TCP connection of the client */
622       unlink_transport (client, session, trans);
623     }
624   }
625 }
626
627 static void
628 close_connection (GstRTSPClient * client)
629 {
630   GstRTSPClientPrivate *priv = client->priv;
631   const gchar *tunnelid;
632
633   GST_DEBUG ("client %p: closing connection", client);
634
635   if ((tunnelid = gst_rtsp_connection_get_tunnelid (priv->connection))) {
636     g_mutex_lock (&tunnels_lock);
637     /* remove from tunnelids */
638     g_hash_table_remove (tunnels, tunnelid);
639     g_mutex_unlock (&tunnels_lock);
640   }
641
642   gst_rtsp_connection_close (priv->connection);
643 }
644
645 static gboolean
646 handle_teardown_request (GstRTSPClient * client, GstRTSPClientState * state)
647 {
648   GstRTSPClientPrivate *priv = client->priv;
649   GstRTSPSession *session;
650   GstRTSPSessionMedia *media;
651   GstRTSPStatusCode code;
652
653   if (!state->session)
654     goto no_session;
655
656   session = state->session;
657
658   /* get a handle to the configuration of the media in the session */
659   media = gst_rtsp_session_get_media (session, state->uri);
660   if (!media)
661     goto not_found;
662
663   state->sessmedia = media;
664
665   /* we emit the signal before closing the connection */
666   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_TEARDOWN_REQUEST],
667       0, state);
668
669   /* unlink the all TCP callbacks */
670   unlink_session_transports (client, session, media);
671
672   /* remove the session from the watched sessions */
673   g_object_weak_unref (G_OBJECT (session),
674       (GWeakNotify) client_session_finalized, client);
675   priv->sessions = g_list_remove (priv->sessions, session);
676
677   gst_rtsp_session_media_set_state (media, GST_STATE_NULL);
678
679   /* unmanage the media in the session, returns false if all media session
680    * are torn down. */
681   if (!gst_rtsp_session_release_media (session, media)) {
682     /* remove the session */
683     gst_rtsp_session_pool_remove (priv->session_pool, session);
684   }
685   /* construct the response now */
686   code = GST_RTSP_STS_OK;
687   gst_rtsp_message_init_response (state->response, code,
688       gst_rtsp_status_as_text (code), state->request);
689
690   send_response (client, session, state->response, TRUE);
691
692   return TRUE;
693
694   /* ERRORS */
695 no_session:
696   {
697     GST_ERROR ("client %p: no session", client);
698     send_generic_response (client, GST_RTSP_STS_SESSION_NOT_FOUND, state);
699     return FALSE;
700   }
701 not_found:
702   {
703     GST_ERROR ("client %p: no media for uri", client);
704     send_generic_response (client, GST_RTSP_STS_NOT_FOUND, state);
705     return FALSE;
706   }
707 }
708
709 static gboolean
710 handle_get_param_request (GstRTSPClient * client, GstRTSPClientState * state)
711 {
712   GstRTSPResult res;
713   guint8 *data;
714   guint size;
715
716   res = gst_rtsp_message_get_body (state->request, &data, &size);
717   if (res != GST_RTSP_OK)
718     goto bad_request;
719
720   if (size == 0) {
721     /* no body, keep-alive request */
722     send_generic_response (client, GST_RTSP_STS_OK, state);
723   } else {
724     /* there is a body, handle the params */
725     res = gst_rtsp_params_get (client, state);
726     if (res != GST_RTSP_OK)
727       goto bad_request;
728
729     send_response (client, state->session, state->response, FALSE);
730   }
731
732   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_GET_PARAMETER_REQUEST],
733       0, state);
734
735   return TRUE;
736
737   /* ERRORS */
738 bad_request:
739   {
740     GST_ERROR ("client %p: bad request", client);
741     send_generic_response (client, GST_RTSP_STS_BAD_REQUEST, state);
742     return FALSE;
743   }
744 }
745
746 static gboolean
747 handle_set_param_request (GstRTSPClient * client, GstRTSPClientState * state)
748 {
749   GstRTSPResult res;
750   guint8 *data;
751   guint size;
752
753   res = gst_rtsp_message_get_body (state->request, &data, &size);
754   if (res != GST_RTSP_OK)
755     goto bad_request;
756
757   if (size == 0) {
758     /* no body, keep-alive request */
759     send_generic_response (client, GST_RTSP_STS_OK, state);
760   } else {
761     /* there is a body, handle the params */
762     res = gst_rtsp_params_set (client, state);
763     if (res != GST_RTSP_OK)
764       goto bad_request;
765
766     send_response (client, state->session, state->response, FALSE);
767   }
768
769   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_SET_PARAMETER_REQUEST],
770       0, state);
771
772   return TRUE;
773
774   /* ERRORS */
775 bad_request:
776   {
777     GST_ERROR ("client %p: bad request", client);
778     send_generic_response (client, GST_RTSP_STS_BAD_REQUEST, state);
779     return FALSE;
780   }
781 }
782
783 static gboolean
784 handle_pause_request (GstRTSPClient * client, GstRTSPClientState * state)
785 {
786   GstRTSPSession *session;
787   GstRTSPSessionMedia *media;
788   GstRTSPStatusCode code;
789   GstRTSPState rtspstate;
790
791   if (!(session = state->session))
792     goto no_session;
793
794   /* get a handle to the configuration of the media in the session */
795   media = gst_rtsp_session_get_media (session, state->uri);
796   if (!media)
797     goto not_found;
798
799   state->sessmedia = media;
800
801   rtspstate = gst_rtsp_session_media_get_rtsp_state (media);
802   /* the session state must be playing or recording */
803   if (rtspstate != GST_RTSP_STATE_PLAYING &&
804       rtspstate != GST_RTSP_STATE_RECORDING)
805     goto invalid_state;
806
807   /* unlink the all TCP callbacks */
808   unlink_session_transports (client, session, media);
809
810   /* then pause sending */
811   gst_rtsp_session_media_set_state (media, GST_STATE_PAUSED);
812
813   /* construct the response now */
814   code = GST_RTSP_STS_OK;
815   gst_rtsp_message_init_response (state->response, code,
816       gst_rtsp_status_as_text (code), state->request);
817
818   send_response (client, session, state->response, FALSE);
819
820   /* the state is now READY */
821   gst_rtsp_session_media_set_rtsp_state (media, GST_RTSP_STATE_READY);
822
823   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_PAUSE_REQUEST],
824       0, state);
825
826   return TRUE;
827
828   /* ERRORS */
829 no_session:
830   {
831     GST_ERROR ("client %p: no seesion", client);
832     send_generic_response (client, GST_RTSP_STS_SESSION_NOT_FOUND, state);
833     return FALSE;
834   }
835 not_found:
836   {
837     GST_ERROR ("client %p: no media for uri", client);
838     send_generic_response (client, GST_RTSP_STS_NOT_FOUND, state);
839     return FALSE;
840   }
841 invalid_state:
842   {
843     GST_ERROR ("client %p: not PLAYING or RECORDING", client);
844     send_generic_response (client, GST_RTSP_STS_METHOD_NOT_VALID_IN_THIS_STATE,
845         state);
846     return FALSE;
847   }
848 }
849
850 static gboolean
851 handle_play_request (GstRTSPClient * client, GstRTSPClientState * state)
852 {
853   GstRTSPSession *session;
854   GstRTSPSessionMedia *media;
855   GstRTSPStatusCode code;
856   GString *rtpinfo;
857   guint n_streams, i, infocount;
858   gchar *str;
859   GstRTSPTimeRange *range;
860   GstRTSPResult res;
861   GstRTSPState rtspstate;
862   GstRTSPRangeUnit unit = GST_RTSP_RANGE_NPT;
863
864   if (!(session = state->session))
865     goto no_session;
866
867   /* get a handle to the configuration of the media in the session */
868   media = gst_rtsp_session_get_media (session, state->uri);
869   if (!media)
870     goto not_found;
871
872   state->sessmedia = media;
873
874   /* the session state must be playing or ready */
875   rtspstate = gst_rtsp_session_media_get_rtsp_state (media);
876   if (rtspstate != GST_RTSP_STATE_PLAYING && rtspstate != GST_RTSP_STATE_READY)
877     goto invalid_state;
878
879   /* parse the range header if we have one */
880   res =
881       gst_rtsp_message_get_header (state->request, GST_RTSP_HDR_RANGE, &str, 0);
882   if (res == GST_RTSP_OK) {
883     if (gst_rtsp_range_parse (str, &range) == GST_RTSP_OK) {
884       /* we have a range, seek to the position */
885       unit = range->unit;
886       gst_rtsp_media_seek (gst_rtsp_session_media_get_media (media), range);
887       gst_rtsp_range_free (range);
888     }
889   }
890
891   /* grab RTPInfo from the payloaders now */
892   rtpinfo = g_string_new ("");
893
894   n_streams =
895       gst_rtsp_media_n_streams (gst_rtsp_session_media_get_media (media));
896   for (i = 0, infocount = 0; i < n_streams; i++) {
897     GstRTSPStreamTransport *trans;
898     GstRTSPStream *stream;
899     const GstRTSPTransport *tr;
900     gchar *uristr;
901     guint rtptime, seq;
902
903     /* get the transport, if there is no transport configured, skip this stream */
904     trans = gst_rtsp_session_media_get_transport (media, i);
905     if (trans == NULL) {
906       GST_INFO ("stream %d is not configured", i);
907       continue;
908     }
909     tr = gst_rtsp_stream_transport_get_transport (trans);
910
911     if (tr->lower_transport == GST_RTSP_LOWER_TRANS_TCP) {
912       /* for TCP, link the stream to the TCP connection of the client */
913       link_transport (client, session, trans);
914     }
915
916     stream = gst_rtsp_stream_transport_get_stream (trans);
917     if (gst_rtsp_stream_get_rtpinfo (stream, &rtptime, &seq)) {
918       if (infocount > 0)
919         g_string_append (rtpinfo, ", ");
920
921       uristr = gst_rtsp_url_get_request_uri (state->uri);
922       g_string_append_printf (rtpinfo, "url=%s/stream=%d;seq=%u;rtptime=%u",
923           uristr, i, seq, rtptime);
924       g_free (uristr);
925
926       infocount++;
927     } else {
928       GST_WARNING ("RTP-Info cannot be determined for stream %d", i);
929     }
930   }
931
932   /* construct the response now */
933   code = GST_RTSP_STS_OK;
934   gst_rtsp_message_init_response (state->response, code,
935       gst_rtsp_status_as_text (code), state->request);
936
937   /* add the RTP-Info header */
938   if (infocount > 0) {
939     str = g_string_free (rtpinfo, FALSE);
940     gst_rtsp_message_take_header (state->response, GST_RTSP_HDR_RTP_INFO, str);
941   } else {
942     g_string_free (rtpinfo, TRUE);
943   }
944
945   /* add the range */
946   str =
947       gst_rtsp_media_get_range_string (gst_rtsp_session_media_get_media (media),
948       TRUE, unit);
949   gst_rtsp_message_take_header (state->response, GST_RTSP_HDR_RANGE, str);
950
951   send_response (client, session, state->response, FALSE);
952
953   /* start playing after sending the request */
954   gst_rtsp_session_media_set_state (media, GST_STATE_PLAYING);
955
956   gst_rtsp_session_media_set_rtsp_state (media, GST_RTSP_STATE_PLAYING);
957
958   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_PLAY_REQUEST],
959       0, state);
960
961   return TRUE;
962
963   /* ERRORS */
964 no_session:
965   {
966     GST_ERROR ("client %p: no session", client);
967     send_generic_response (client, GST_RTSP_STS_SESSION_NOT_FOUND, state);
968     return FALSE;
969   }
970 not_found:
971   {
972     GST_ERROR ("client %p: media not found", client);
973     send_generic_response (client, GST_RTSP_STS_NOT_FOUND, state);
974     return FALSE;
975   }
976 invalid_state:
977   {
978     GST_ERROR ("client %p: not PLAYING or READY", client);
979     send_generic_response (client, GST_RTSP_STS_METHOD_NOT_VALID_IN_THIS_STATE,
980         state);
981     return FALSE;
982   }
983 }
984
985 static void
986 do_keepalive (GstRTSPSession * session)
987 {
988   GST_INFO ("keep session %p alive", session);
989   gst_rtsp_session_touch (session);
990 }
991
992 /* parse @transport and return a valid transport in @tr. only transports
993  * from @supported are returned. Returns FALSE if no valid transport
994  * was found. */
995 static gboolean
996 parse_transport (const char *transport, GstRTSPLowerTrans supported,
997     GstRTSPTransport * tr)
998 {
999   gint i;
1000   gboolean res;
1001   gchar **transports;
1002
1003   res = FALSE;
1004   gst_rtsp_transport_init (tr);
1005
1006   GST_DEBUG ("parsing transports %s", transport);
1007
1008   transports = g_strsplit (transport, ",", 0);
1009
1010   /* loop through the transports, try to parse */
1011   for (i = 0; transports[i]; i++) {
1012     res = gst_rtsp_transport_parse (transports[i], tr);
1013     if (res != GST_RTSP_OK) {
1014       /* no valid transport, search some more */
1015       GST_WARNING ("could not parse transport %s", transports[i]);
1016       goto next;
1017     }
1018
1019     /* we have a transport, see if it's RTP/AVP */
1020     if (tr->trans != GST_RTSP_TRANS_RTP || tr->profile != GST_RTSP_PROFILE_AVP) {
1021       GST_WARNING ("invalid transport %s", transports[i]);
1022       goto next;
1023     }
1024
1025     if (!(tr->lower_transport & supported)) {
1026       GST_WARNING ("unsupported transport %s", transports[i]);
1027       goto next;
1028     }
1029
1030     /* we have a valid transport */
1031     GST_INFO ("found valid transport %s", transports[i]);
1032     res = TRUE;
1033     break;
1034
1035   next:
1036     gst_rtsp_transport_init (tr);
1037   }
1038   g_strfreev (transports);
1039
1040   return res;
1041 }
1042
1043 static gboolean
1044 handle_blocksize (GstRTSPMedia * media, GstRTSPStream * stream,
1045     GstRTSPMessage * request)
1046 {
1047   gchar *blocksize_str;
1048   gboolean ret = TRUE;
1049
1050   if (gst_rtsp_message_get_header (request, GST_RTSP_HDR_BLOCKSIZE,
1051           &blocksize_str, 0) == GST_RTSP_OK) {
1052     guint64 blocksize;
1053     gchar *end;
1054
1055     blocksize = g_ascii_strtoull (blocksize_str, &end, 10);
1056     if (end == blocksize_str) {
1057       GST_ERROR ("failed to parse blocksize");
1058       ret = FALSE;
1059     } else {
1060       /* we don't want to change the mtu when this media
1061        * can be shared because it impacts other clients */
1062       if (gst_rtsp_media_is_shared (media))
1063         return TRUE;
1064
1065       if (blocksize > G_MAXUINT)
1066         blocksize = G_MAXUINT;
1067       gst_rtsp_stream_set_mtu (stream, blocksize);
1068     }
1069   }
1070   return ret;
1071 }
1072
1073 static gboolean
1074 configure_client_transport (GstRTSPClient * client, GstRTSPClientState * state,
1075     GstRTSPTransport * ct)
1076 {
1077   GstRTSPClientPrivate *priv = client->priv;
1078
1079   /* we have a valid transport now, set the destination of the client. */
1080   if (ct->lower_transport == GST_RTSP_LOWER_TRANS_UDP_MCAST) {
1081     if (ct->destination && priv->use_client_settings) {
1082       GstRTSPAddress *addr;
1083
1084       addr = gst_rtsp_stream_reserve_address (state->stream, ct->destination,
1085           ct->port.min, ct->port.max - ct->port.min + 1, ct->ttl);
1086
1087       if (addr == NULL)
1088         goto no_address;
1089
1090       gst_rtsp_address_free (addr);
1091     } else {
1092       GstRTSPAddress *addr;
1093
1094       addr = gst_rtsp_stream_get_address (state->stream);
1095       if (addr == NULL)
1096         goto no_address;
1097
1098       g_free (ct->destination);
1099       ct->destination = g_strdup (addr->address);
1100       ct->port.min = addr->port;
1101       ct->port.max = addr->port + addr->n_ports - 1;
1102       ct->ttl = addr->ttl;
1103
1104       gst_rtsp_address_free (addr);
1105     }
1106   } else {
1107     GstRTSPUrl *url;
1108
1109     url = gst_rtsp_connection_get_url (priv->connection);
1110     g_free (ct->destination);
1111     ct->destination = g_strdup (url->host);
1112
1113     if (ct->lower_transport & GST_RTSP_LOWER_TRANS_TCP) {
1114       /* check if the client selected channels for TCP */
1115       if (ct->interleaved.min == -1 || ct->interleaved.max == -1) {
1116         gst_rtsp_session_media_alloc_channels (state->sessmedia,
1117             &ct->interleaved);
1118       }
1119     }
1120   }
1121   return TRUE;
1122
1123   /* ERRORS */
1124 no_address:
1125   {
1126     GST_ERROR_OBJECT (client, "failed to acquire address for stream");
1127     return FALSE;
1128   }
1129 }
1130
1131 static GstRTSPTransport *
1132 make_server_transport (GstRTSPClient * client, GstRTSPClientState * state,
1133     GstRTSPTransport * ct)
1134 {
1135   GstRTSPTransport *st;
1136   GInetAddress *addr;
1137   GSocketFamily family;
1138
1139   /* prepare the server transport */
1140   gst_rtsp_transport_new (&st);
1141
1142   st->trans = ct->trans;
1143   st->profile = ct->profile;
1144   st->lower_transport = ct->lower_transport;
1145
1146   addr = g_inet_address_new_from_string (ct->destination);
1147
1148   if (!addr) {
1149     GST_ERROR ("failed to get inet addr from client destination");
1150     family = G_SOCKET_FAMILY_IPV4;
1151   } else {
1152     family = g_inet_address_get_family (addr);
1153     g_object_unref (addr);
1154     addr = NULL;
1155   }
1156
1157   switch (st->lower_transport) {
1158     case GST_RTSP_LOWER_TRANS_UDP:
1159       st->client_port = ct->client_port;
1160       gst_rtsp_stream_get_server_port (state->stream, &st->server_port, family);
1161       break;
1162     case GST_RTSP_LOWER_TRANS_UDP_MCAST:
1163       st->port = ct->port;
1164       st->destination = g_strdup (ct->destination);
1165       st->ttl = ct->ttl;
1166       break;
1167     case GST_RTSP_LOWER_TRANS_TCP:
1168       st->interleaved = ct->interleaved;
1169     default:
1170       break;
1171   }
1172
1173   gst_rtsp_stream_get_ssrc (state->stream, &st->ssrc);
1174
1175   return st;
1176 }
1177
1178 static gboolean
1179 handle_setup_request (GstRTSPClient * client, GstRTSPClientState * state)
1180 {
1181   GstRTSPClientPrivate *priv = client->priv;
1182   GstRTSPResult res;
1183   GstRTSPUrl *uri;
1184   gchar *transport;
1185   GstRTSPTransport *ct, *st;
1186   GstRTSPLowerTrans supported;
1187   GstRTSPStatusCode code;
1188   GstRTSPSession *session;
1189   GstRTSPStreamTransport *trans;
1190   gchar *trans_str, *pos;
1191   guint streamid;
1192   GstRTSPSessionMedia *sessmedia;
1193   GstRTSPMedia *media;
1194   GstRTSPStream *stream;
1195   GstRTSPState rtspstate;
1196
1197   uri = state->uri;
1198
1199   /* the uri contains the stream number we added in the SDP config, which is
1200    * always /stream=%d so we need to strip that off
1201    * parse the stream we need to configure, look for the stream in the abspath
1202    * first and then in the query. */
1203   if (uri->abspath == NULL || !(pos = strstr (uri->abspath, "/stream="))) {
1204     if (uri->query == NULL || !(pos = strstr (uri->query, "/stream=")))
1205       goto bad_request;
1206   }
1207
1208   /* we can mofify the parsed uri in place */
1209   *pos = '\0';
1210
1211   pos += strlen ("/stream=");
1212   if (sscanf (pos, "%u", &streamid) != 1)
1213     goto bad_request;
1214
1215   /* parse the transport */
1216   res =
1217       gst_rtsp_message_get_header (state->request, GST_RTSP_HDR_TRANSPORT,
1218       &transport, 0);
1219   if (res != GST_RTSP_OK)
1220     goto no_transport;
1221
1222   gst_rtsp_transport_new (&ct);
1223
1224   /* our supported transports */
1225   supported = GST_RTSP_LOWER_TRANS_UDP |
1226       GST_RTSP_LOWER_TRANS_UDP_MCAST | GST_RTSP_LOWER_TRANS_TCP;
1227
1228   /* parse and find a usable supported transport */
1229   if (!parse_transport (transport, supported, ct))
1230     goto unsupported_transports;
1231
1232   /* we create the session after parsing stuff so that we don't make
1233    * a session for malformed requests */
1234   if (priv->session_pool == NULL)
1235     goto no_pool;
1236
1237   session = state->session;
1238
1239   if (session) {
1240     g_object_ref (session);
1241     /* get a handle to the configuration of the media in the session, this can
1242      * return NULL if this is a new url to manage in this session. */
1243     sessmedia = gst_rtsp_session_get_media (session, uri);
1244   } else {
1245     /* create a session if this fails we probably reached our session limit or
1246      * something. */
1247     if (!(session = gst_rtsp_session_pool_create (priv->session_pool)))
1248       goto service_unavailable;
1249
1250     state->session = session;
1251
1252     /* we need a new media configuration in this session */
1253     sessmedia = NULL;
1254   }
1255
1256   /* we have no media, find one and manage it */
1257   if (sessmedia == NULL) {
1258     /* get a handle to the configuration of the media in the session */
1259     if ((media = find_media (client, state))) {
1260       /* manage the media in our session now */
1261       sessmedia = gst_rtsp_session_manage_media (session, uri, media);
1262     }
1263   }
1264
1265   /* if we stil have no media, error */
1266   if (sessmedia == NULL)
1267     goto not_found;
1268
1269   state->sessmedia = sessmedia;
1270   state->media = media = gst_rtsp_session_media_get_media (sessmedia);
1271
1272   /* now get the stream */
1273   stream = gst_rtsp_media_get_stream (media, streamid);
1274   if (stream == NULL)
1275     goto not_found;
1276
1277   state->stream = stream;
1278
1279   /* set blocksize on this stream */
1280   if (!handle_blocksize (media, stream, state->request))
1281     goto invalid_blocksize;
1282
1283   /* update the client transport */
1284   if (!configure_client_transport (client, state, ct))
1285     goto unsupported_client_transport;
1286
1287   /* set in the session media transport */
1288   trans = gst_rtsp_session_media_set_transport (sessmedia, stream, ct);
1289
1290   /* configure keepalive for this transport */
1291   gst_rtsp_stream_transport_set_keepalive (trans,
1292       (GstRTSPKeepAliveFunc) do_keepalive, session, NULL);
1293
1294   /* create and serialize the server transport */
1295   st = make_server_transport (client, state, ct);
1296   trans_str = gst_rtsp_transport_as_text (st);
1297   gst_rtsp_transport_free (st);
1298
1299   /* construct the response now */
1300   code = GST_RTSP_STS_OK;
1301   gst_rtsp_message_init_response (state->response, code,
1302       gst_rtsp_status_as_text (code), state->request);
1303
1304   gst_rtsp_message_add_header (state->response, GST_RTSP_HDR_TRANSPORT,
1305       trans_str);
1306   g_free (trans_str);
1307
1308   send_response (client, session, state->response, FALSE);
1309
1310   /* update the state */
1311   rtspstate = gst_rtsp_session_media_get_rtsp_state (sessmedia);
1312   switch (rtspstate) {
1313     case GST_RTSP_STATE_PLAYING:
1314     case GST_RTSP_STATE_RECORDING:
1315     case GST_RTSP_STATE_READY:
1316       /* no state change */
1317       break;
1318     default:
1319       gst_rtsp_session_media_set_rtsp_state (sessmedia, GST_RTSP_STATE_READY);
1320       break;
1321   }
1322   g_object_unref (session);
1323
1324   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_SETUP_REQUEST],
1325       0, state);
1326
1327   return TRUE;
1328
1329   /* ERRORS */
1330 bad_request:
1331   {
1332     GST_ERROR ("client %p: bad request", client);
1333     send_generic_response (client, GST_RTSP_STS_BAD_REQUEST, state);
1334     return FALSE;
1335   }
1336 not_found:
1337   {
1338     GST_ERROR ("client %p: media not found", client);
1339     send_generic_response (client, GST_RTSP_STS_NOT_FOUND, state);
1340     g_object_unref (session);
1341     gst_rtsp_transport_free (ct);
1342     return FALSE;
1343   }
1344 invalid_blocksize:
1345   {
1346     GST_ERROR ("client %p: invalid blocksize", client);
1347     send_generic_response (client, GST_RTSP_STS_BAD_REQUEST, state);
1348     g_object_unref (session);
1349     gst_rtsp_transport_free (ct);
1350     return FALSE;
1351   }
1352 unsupported_client_transport:
1353   {
1354     GST_ERROR ("client %p: unsupported client transport", client);
1355     send_generic_response (client, GST_RTSP_STS_UNSUPPORTED_TRANSPORT, state);
1356     g_object_unref (session);
1357     gst_rtsp_transport_free (ct);
1358     return FALSE;
1359   }
1360 no_transport:
1361   {
1362     GST_ERROR ("client %p: no transport", client);
1363     send_generic_response (client, GST_RTSP_STS_UNSUPPORTED_TRANSPORT, state);
1364     return FALSE;
1365   }
1366 unsupported_transports:
1367   {
1368     GST_ERROR ("client %p: unsupported transports", client);
1369     send_generic_response (client, GST_RTSP_STS_UNSUPPORTED_TRANSPORT, state);
1370     gst_rtsp_transport_free (ct);
1371     return FALSE;
1372   }
1373 no_pool:
1374   {
1375     GST_ERROR ("client %p: no session pool configured", client);
1376     send_generic_response (client, GST_RTSP_STS_SESSION_NOT_FOUND, state);
1377     gst_rtsp_transport_free (ct);
1378     return FALSE;
1379   }
1380 service_unavailable:
1381   {
1382     GST_ERROR ("client %p: can't create session", client);
1383     send_generic_response (client, GST_RTSP_STS_SERVICE_UNAVAILABLE, state);
1384     gst_rtsp_transport_free (ct);
1385     return FALSE;
1386   }
1387 }
1388
1389 static GstSDPMessage *
1390 create_sdp (GstRTSPClient * client, GstRTSPMedia * media)
1391 {
1392   GstRTSPClientPrivate *priv = client->priv;
1393   GstSDPMessage *sdp;
1394   GstSDPInfo info;
1395   const gchar *proto;
1396
1397   gst_sdp_message_new (&sdp);
1398
1399   /* some standard things first */
1400   gst_sdp_message_set_version (sdp, "0");
1401
1402   if (priv->is_ipv6)
1403     proto = "IP6";
1404   else
1405     proto = "IP4";
1406
1407   gst_sdp_message_set_origin (sdp, "-", "1188340656180883", "1", "IN", proto,
1408       priv->server_ip);
1409
1410   gst_sdp_message_set_session_name (sdp, "Session streamed with GStreamer");
1411   gst_sdp_message_set_information (sdp, "rtsp-server");
1412   gst_sdp_message_add_time (sdp, "0", "0", NULL);
1413   gst_sdp_message_add_attribute (sdp, "tool", "GStreamer");
1414   gst_sdp_message_add_attribute (sdp, "type", "broadcast");
1415   gst_sdp_message_add_attribute (sdp, "control", "*");
1416
1417   info.is_ipv6 = priv->is_ipv6;
1418   info.server_ip = priv->server_ip;
1419
1420   /* create an SDP for the media object */
1421   if (!gst_rtsp_sdp_from_media (sdp, &info, media))
1422     goto no_sdp;
1423
1424   return sdp;
1425
1426   /* ERRORS */
1427 no_sdp:
1428   {
1429     GST_ERROR ("client %p: could not create SDP", client);
1430     gst_sdp_message_free (sdp);
1431     return NULL;
1432   }
1433 }
1434
1435 /* for the describe we must generate an SDP */
1436 static gboolean
1437 handle_describe_request (GstRTSPClient * client, GstRTSPClientState * state)
1438 {
1439   GstRTSPResult res;
1440   GstSDPMessage *sdp;
1441   guint i, str_len;
1442   gchar *str, *content_base;
1443   GstRTSPMedia *media;
1444   GstRTSPClientClass *klass;
1445
1446   klass = GST_RTSP_CLIENT_GET_CLASS (client);
1447
1448   /* check what kind of format is accepted, we don't really do anything with it
1449    * and always return SDP for now. */
1450   for (i = 0; i++;) {
1451     gchar *accept;
1452
1453     res =
1454         gst_rtsp_message_get_header (state->request, GST_RTSP_HDR_ACCEPT,
1455         &accept, i);
1456     if (res == GST_RTSP_ENOTIMPL)
1457       break;
1458
1459     if (g_ascii_strcasecmp (accept, "application/sdp") == 0)
1460       break;
1461   }
1462
1463   /* find the media object for the uri */
1464   if (!(media = find_media (client, state)))
1465     goto no_media;
1466
1467   /* create an SDP for the media object on this client */
1468   if (!(sdp = klass->create_sdp (client, media)))
1469     goto no_sdp;
1470
1471   g_object_unref (media);
1472
1473   gst_rtsp_message_init_response (state->response, GST_RTSP_STS_OK,
1474       gst_rtsp_status_as_text (GST_RTSP_STS_OK), state->request);
1475
1476   gst_rtsp_message_add_header (state->response, GST_RTSP_HDR_CONTENT_TYPE,
1477       "application/sdp");
1478
1479   /* content base for some clients that might screw up creating the setup uri */
1480   str = gst_rtsp_url_get_request_uri (state->uri);
1481   str_len = strlen (str);
1482
1483   /* check for trailing '/' and append one */
1484   if (str[str_len - 1] != '/') {
1485     content_base = g_malloc (str_len + 2);
1486     memcpy (content_base, str, str_len);
1487     content_base[str_len] = '/';
1488     content_base[str_len + 1] = '\0';
1489     g_free (str);
1490   } else {
1491     content_base = str;
1492   }
1493
1494   GST_INFO ("adding content-base: %s", content_base);
1495
1496   gst_rtsp_message_add_header (state->response, GST_RTSP_HDR_CONTENT_BASE,
1497       content_base);
1498   g_free (content_base);
1499
1500   /* add SDP to the response body */
1501   str = gst_sdp_message_as_text (sdp);
1502   gst_rtsp_message_take_body (state->response, (guint8 *) str, strlen (str));
1503   gst_sdp_message_free (sdp);
1504
1505   send_response (client, state->session, state->response, FALSE);
1506
1507   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_DESCRIBE_REQUEST],
1508       0, state);
1509
1510   return TRUE;
1511
1512   /* ERRORS */
1513 no_media:
1514   {
1515     GST_ERROR ("client %p: no media", client);
1516     /* error reply is already sent */
1517     return FALSE;
1518   }
1519 no_sdp:
1520   {
1521     GST_ERROR ("client %p: can't create SDP", client);
1522     send_generic_response (client, GST_RTSP_STS_SERVICE_UNAVAILABLE, state);
1523     g_object_unref (media);
1524     return FALSE;
1525   }
1526 }
1527
1528 static gboolean
1529 handle_options_request (GstRTSPClient * client, GstRTSPClientState * state)
1530 {
1531   GstRTSPMethod options;
1532   gchar *str;
1533
1534   options = GST_RTSP_DESCRIBE |
1535       GST_RTSP_OPTIONS |
1536       GST_RTSP_PAUSE |
1537       GST_RTSP_PLAY |
1538       GST_RTSP_SETUP |
1539       GST_RTSP_GET_PARAMETER | GST_RTSP_SET_PARAMETER | GST_RTSP_TEARDOWN;
1540
1541   str = gst_rtsp_options_as_text (options);
1542
1543   gst_rtsp_message_init_response (state->response, GST_RTSP_STS_OK,
1544       gst_rtsp_status_as_text (GST_RTSP_STS_OK), state->request);
1545
1546   gst_rtsp_message_add_header (state->response, GST_RTSP_HDR_PUBLIC, str);
1547   g_free (str);
1548
1549   send_response (client, state->session, state->response, FALSE);
1550
1551   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_OPTIONS_REQUEST],
1552       0, state);
1553
1554   return TRUE;
1555 }
1556
1557 /* remove duplicate and trailing '/' */
1558 static void
1559 sanitize_uri (GstRTSPUrl * uri)
1560 {
1561   gint i, len;
1562   gchar *s, *d;
1563   gboolean have_slash, prev_slash;
1564
1565   s = d = uri->abspath;
1566   len = strlen (uri->abspath);
1567
1568   prev_slash = FALSE;
1569
1570   for (i = 0; i < len; i++) {
1571     have_slash = s[i] == '/';
1572     *d = s[i];
1573     if (!have_slash || !prev_slash)
1574       d++;
1575     prev_slash = have_slash;
1576   }
1577   len = d - uri->abspath;
1578   /* don't remove the first slash if that's the only thing left */
1579   if (len > 1 && *(d - 1) == '/')
1580     d--;
1581   *d = '\0';
1582 }
1583
1584 static void
1585 client_session_finalized (GstRTSPClient * client, GstRTSPSession * session)
1586 {
1587   GstRTSPClientPrivate *priv = client->priv;
1588
1589   GST_INFO ("client %p: session %p finished", client, session);
1590
1591   /* unlink all media managed in this session */
1592   client_unlink_session (client, session);
1593
1594   /* remove the session */
1595   if (!(priv->sessions = g_list_remove (priv->sessions, session))) {
1596     GST_INFO ("client %p: all sessions finalized, close the connection",
1597         client);
1598     close_connection (client);
1599   }
1600 }
1601
1602 static void
1603 client_watch_session (GstRTSPClient * client, GstRTSPSession * session)
1604 {
1605   GstRTSPClientPrivate *priv = client->priv;
1606   GList *walk;
1607
1608   for (walk = priv->sessions; walk; walk = g_list_next (walk)) {
1609     GstRTSPSession *msession = (GstRTSPSession *) walk->data;
1610
1611     /* we already know about this session */
1612     if (msession == session)
1613       return;
1614   }
1615
1616   GST_INFO ("watching session %p", session);
1617
1618   g_object_weak_ref (G_OBJECT (session), (GWeakNotify) client_session_finalized,
1619       client);
1620   priv->sessions = g_list_prepend (priv->sessions, session);
1621
1622   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_NEW_SESSION], 0,
1623       session);
1624 }
1625
1626 static void
1627 handle_request (GstRTSPClient * client, GstRTSPMessage * request)
1628 {
1629   GstRTSPClientPrivate *priv = client->priv;
1630   GstRTSPMethod method;
1631   const gchar *uristr;
1632   GstRTSPUrl *uri = NULL;
1633   GstRTSPVersion version;
1634   GstRTSPResult res;
1635   GstRTSPSession *session = NULL;
1636   GstRTSPClientState state = { NULL };
1637   GstRTSPMessage response = { 0 };
1638   gchar *sessid;
1639
1640   state.request = request;
1641   state.response = &response;
1642
1643   if (gst_debug_category_get_threshold (rtsp_client_debug) >= GST_LEVEL_LOG) {
1644     gst_rtsp_message_dump (request);
1645   }
1646
1647   GST_INFO ("client %p: received a request", client);
1648
1649   gst_rtsp_message_parse_request (request, &method, &uristr, &version);
1650
1651   /* we can only handle 1.0 requests */
1652   if (version != GST_RTSP_VERSION_1_0)
1653     goto not_supported;
1654
1655   state.method = method;
1656
1657   /* we always try to parse the url first */
1658   if (gst_rtsp_url_parse (uristr, &uri) != GST_RTSP_OK)
1659     goto bad_request;
1660
1661   /* get the session if there is any */
1662   res = gst_rtsp_message_get_header (request, GST_RTSP_HDR_SESSION, &sessid, 0);
1663   if (res == GST_RTSP_OK) {
1664     if (priv->session_pool == NULL)
1665       goto no_pool;
1666
1667     /* we had a session in the request, find it again */
1668     if (!(session = gst_rtsp_session_pool_find (priv->session_pool, sessid)))
1669       goto session_not_found;
1670
1671     /* we add the session to the client list of watched sessions. When a session
1672      * disappears because it times out, we will be notified. If all sessions are
1673      * gone, we will close the connection */
1674     client_watch_session (client, session);
1675   }
1676
1677   /* sanitize the uri */
1678   sanitize_uri (uri);
1679   state.uri = uri;
1680   state.session = session;
1681
1682   if (priv->auth) {
1683     if (!gst_rtsp_auth_check (priv->auth, client, 0, &state))
1684       goto not_authorized;
1685   }
1686
1687   /* now see what is asked and dispatch to a dedicated handler */
1688   switch (method) {
1689     case GST_RTSP_OPTIONS:
1690       handle_options_request (client, &state);
1691       break;
1692     case GST_RTSP_DESCRIBE:
1693       handle_describe_request (client, &state);
1694       break;
1695     case GST_RTSP_SETUP:
1696       handle_setup_request (client, &state);
1697       break;
1698     case GST_RTSP_PLAY:
1699       handle_play_request (client, &state);
1700       break;
1701     case GST_RTSP_PAUSE:
1702       handle_pause_request (client, &state);
1703       break;
1704     case GST_RTSP_TEARDOWN:
1705       handle_teardown_request (client, &state);
1706       break;
1707     case GST_RTSP_SET_PARAMETER:
1708       handle_set_param_request (client, &state);
1709       break;
1710     case GST_RTSP_GET_PARAMETER:
1711       handle_get_param_request (client, &state);
1712       break;
1713     case GST_RTSP_ANNOUNCE:
1714     case GST_RTSP_RECORD:
1715     case GST_RTSP_REDIRECT:
1716       goto not_implemented;
1717     case GST_RTSP_INVALID:
1718     default:
1719       goto bad_request;
1720   }
1721
1722 done:
1723   if (session)
1724     g_object_unref (session);
1725   if (uri)
1726     gst_rtsp_url_free (uri);
1727   return;
1728
1729   /* ERRORS */
1730 not_supported:
1731   {
1732     GST_ERROR ("client %p: version %d not supported", client, version);
1733     send_generic_response (client, GST_RTSP_STS_RTSP_VERSION_NOT_SUPPORTED,
1734         &state);
1735     goto done;
1736   }
1737 bad_request:
1738   {
1739     GST_ERROR ("client %p: bad request", client);
1740     send_generic_response (client, GST_RTSP_STS_BAD_REQUEST, &state);
1741     goto done;
1742   }
1743 no_pool:
1744   {
1745     GST_ERROR ("client %p: no pool configured", client);
1746     send_generic_response (client, GST_RTSP_STS_SESSION_NOT_FOUND, &state);
1747     goto done;
1748   }
1749 session_not_found:
1750   {
1751     GST_ERROR ("client %p: session not found", client);
1752     send_generic_response (client, GST_RTSP_STS_SESSION_NOT_FOUND, &state);
1753     goto done;
1754   }
1755 not_authorized:
1756   {
1757     GST_ERROR ("client %p: not allowed", client);
1758     handle_unauthorized_request (client, priv->auth, &state);
1759     goto done;
1760   }
1761 not_implemented:
1762   {
1763     GST_ERROR ("client %p: method %d not implemented", client, method);
1764     send_generic_response (client, GST_RTSP_STS_NOT_IMPLEMENTED, &state);
1765     goto done;
1766   }
1767 }
1768
1769 static void
1770 handle_data (GstRTSPClient * client, GstRTSPMessage * message)
1771 {
1772   GstRTSPClientPrivate *priv = client->priv;
1773   GstRTSPResult res;
1774   guint8 channel;
1775   GList *walk;
1776   guint8 *data;
1777   guint size;
1778   GstBuffer *buffer;
1779   gboolean handled;
1780
1781   /* find the stream for this message */
1782   res = gst_rtsp_message_parse_data (message, &channel);
1783   if (res != GST_RTSP_OK)
1784     return;
1785
1786   gst_rtsp_message_steal_body (message, &data, &size);
1787
1788   buffer = gst_buffer_new_wrapped (data, size);
1789
1790   handled = FALSE;
1791   for (walk = priv->transports; walk; walk = g_list_next (walk)) {
1792     GstRTSPStreamTransport *trans;
1793     GstRTSPStream *stream;
1794     const GstRTSPTransport *tr;
1795
1796     trans = walk->data;
1797
1798     tr = gst_rtsp_stream_transport_get_transport (trans);
1799     stream = gst_rtsp_stream_transport_get_stream (trans);
1800
1801     /* check for TCP transport */
1802     if (tr->lower_transport == GST_RTSP_LOWER_TRANS_TCP) {
1803       /* dispatch to the stream based on the channel number */
1804       if (tr->interleaved.min == channel) {
1805         gst_rtsp_stream_recv_rtp (stream, buffer);
1806         handled = TRUE;
1807         break;
1808       } else if (tr->interleaved.max == channel) {
1809         gst_rtsp_stream_recv_rtcp (stream, buffer);
1810         handled = TRUE;
1811         break;
1812       }
1813     }
1814   }
1815   if (!handled)
1816     gst_buffer_unref (buffer);
1817 }
1818
1819 /**
1820  * gst_rtsp_client_set_session_pool:
1821  * @client: a #GstRTSPClient
1822  * @pool: a #GstRTSPSessionPool
1823  *
1824  * Set @pool as the sessionpool for @client which it will use to find
1825  * or allocate sessions. the sessionpool is usually inherited from the server
1826  * that created the client but can be overridden later.
1827  */
1828 void
1829 gst_rtsp_client_set_session_pool (GstRTSPClient * client,
1830     GstRTSPSessionPool * pool)
1831 {
1832   GstRTSPSessionPool *old;
1833   GstRTSPClientPrivate *priv;
1834
1835   g_return_if_fail (GST_IS_RTSP_CLIENT (client));
1836
1837   priv = client->priv;
1838
1839   if (pool)
1840     g_object_ref (pool);
1841
1842   g_mutex_lock (&priv->lock);
1843   old = priv->session_pool;
1844   priv->session_pool = pool;
1845   g_mutex_unlock (&priv->lock);
1846
1847   if (old)
1848     g_object_unref (old);
1849 }
1850
1851 /**
1852  * gst_rtsp_client_get_session_pool:
1853  * @client: a #GstRTSPClient
1854  *
1855  * Get the #GstRTSPSessionPool object that @client uses to manage its sessions.
1856  *
1857  * Returns: (transfer full): a #GstRTSPSessionPool, unref after usage.
1858  */
1859 GstRTSPSessionPool *
1860 gst_rtsp_client_get_session_pool (GstRTSPClient * client)
1861 {
1862   GstRTSPClientPrivate *priv;
1863   GstRTSPSessionPool *result;
1864
1865   g_return_val_if_fail (GST_IS_RTSP_CLIENT (client), NULL);
1866
1867   priv = client->priv;
1868
1869   g_mutex_lock (&priv->lock);
1870   if ((result = priv->session_pool))
1871     g_object_ref (result);
1872   g_mutex_unlock (&priv->lock);
1873
1874   return result;
1875 }
1876
1877 /**
1878  * gst_rtsp_client_set_mount_points:
1879  * @client: a #GstRTSPClient
1880  * @mounts: a #GstRTSPMountPoints
1881  *
1882  * Set @mounts as the mount points for @client which it will use to map urls
1883  * to media streams. These mount points are usually inherited from the server that
1884  * created the client but can be overriden later.
1885  */
1886 void
1887 gst_rtsp_client_set_mount_points (GstRTSPClient * client,
1888     GstRTSPMountPoints * mounts)
1889 {
1890   GstRTSPClientPrivate *priv;
1891   GstRTSPMountPoints *old;
1892
1893   g_return_if_fail (GST_IS_RTSP_CLIENT (client));
1894
1895   priv = client->priv;
1896
1897   if (mounts)
1898     g_object_ref (mounts);
1899
1900   g_mutex_lock (&priv->lock);
1901   old = priv->mount_points;
1902   priv->mount_points = mounts;
1903   g_mutex_unlock (&priv->lock);
1904
1905   if (old)
1906     g_object_unref (old);
1907 }
1908
1909 /**
1910  * gst_rtsp_client_get_mount_points:
1911  * @client: a #GstRTSPClient
1912  *
1913  * Get the #GstRTSPMountPoints object that @client uses to manage its sessions.
1914  *
1915  * Returns: (transfer full): a #GstRTSPMountPoints, unref after usage.
1916  */
1917 GstRTSPMountPoints *
1918 gst_rtsp_client_get_mount_points (GstRTSPClient * client)
1919 {
1920   GstRTSPClientPrivate *priv;
1921   GstRTSPMountPoints *result;
1922
1923   g_return_val_if_fail (GST_IS_RTSP_CLIENT (client), NULL);
1924
1925   priv = client->priv;
1926
1927   g_mutex_lock (&priv->lock);
1928   if ((result = priv->mount_points))
1929     g_object_ref (result);
1930   g_mutex_unlock (&priv->lock);
1931
1932   return result;
1933 }
1934
1935 /**
1936  * gst_rtsp_client_set_use_client_settings:
1937  * @client: a #GstRTSPClient
1938  * @use_client_settings: whether to use client settings for multicast
1939  *
1940  * Use client transport settings (destination and ttl) for multicast.
1941  * When @use_client_settings is %FALSE, the server settings will be
1942  * used.
1943  */
1944 void
1945 gst_rtsp_client_set_use_client_settings (GstRTSPClient * client,
1946     gboolean use_client_settings)
1947 {
1948   GstRTSPClientPrivate *priv;
1949
1950   g_return_if_fail (GST_IS_RTSP_CLIENT (client));
1951
1952   priv = client->priv;
1953
1954   g_mutex_lock (&priv->lock);
1955   priv->use_client_settings = use_client_settings;
1956   g_mutex_unlock (&priv->lock);
1957 }
1958
1959 /**
1960  * gst_rtsp_client_get_use_client_settings:
1961  * @client: a #GstRTSPClient
1962  *
1963  * Check if client transport settings (destination and ttl) for multicast
1964  * will be used.
1965  */
1966 gboolean
1967 gst_rtsp_client_get_use_client_settings (GstRTSPClient * client)
1968 {
1969   GstRTSPClientPrivate *priv;
1970   gboolean res;
1971
1972   g_return_val_if_fail (GST_IS_RTSP_CLIENT (client), FALSE);
1973
1974   priv = client->priv;
1975
1976   g_mutex_lock (&priv->lock);
1977   res = priv->use_client_settings;
1978   g_mutex_unlock (&priv->lock);
1979
1980   return res;
1981 }
1982
1983 /**
1984  * gst_rtsp_client_set_auth:
1985  * @client: a #GstRTSPClient
1986  * @auth: a #GstRTSPAuth
1987  *
1988  * configure @auth to be used as the authentication manager of @client.
1989  */
1990 void
1991 gst_rtsp_client_set_auth (GstRTSPClient * client, GstRTSPAuth * auth)
1992 {
1993   GstRTSPClientPrivate *priv;
1994   GstRTSPAuth *old;
1995
1996   g_return_if_fail (GST_IS_RTSP_CLIENT (client));
1997
1998   priv = client->priv;
1999
2000   if (auth)
2001     g_object_ref (auth);
2002
2003   g_mutex_lock (&priv->lock);
2004   old = priv->auth;
2005   priv->auth = auth;
2006   g_mutex_unlock (&priv->lock);
2007
2008   if (old)
2009     g_object_unref (old);
2010 }
2011
2012
2013 /**
2014  * gst_rtsp_client_get_auth:
2015  * @client: a #GstRTSPClient
2016  *
2017  * Get the #GstRTSPAuth used as the authentication manager of @client.
2018  *
2019  * Returns: (transfer full): the #GstRTSPAuth of @client. g_object_unref() after
2020  * usage.
2021  */
2022 GstRTSPAuth *
2023 gst_rtsp_client_get_auth (GstRTSPClient * client)
2024 {
2025   GstRTSPClientPrivate *priv;
2026   GstRTSPAuth *result;
2027
2028   g_return_val_if_fail (GST_IS_RTSP_CLIENT (client), NULL);
2029
2030   priv = client->priv;
2031
2032   g_mutex_lock (&priv->lock);
2033   if ((result = priv->auth))
2034     g_object_ref (result);
2035   g_mutex_unlock (&priv->lock);
2036
2037   return result;
2038 }
2039
2040 /**
2041  * gst_rtsp_client_get_uri:
2042  * @client: a #GstRTSPClient
2043  *
2044  * Get the #GstRTSPUrl of @client.
2045  *
2046  * Returns: (transfer full): the #GstRTSPUrl of @client. Free with
2047  * gst_rtsp_url_free () after usage.
2048  */
2049 GstRTSPUrl *
2050 gst_rtsp_client_get_uri (GstRTSPClient * client)
2051 {
2052   GstRTSPClientPrivate *priv;
2053   GstRTSPUrl *result = NULL;
2054
2055   g_return_val_if_fail (GST_IS_RTSP_CLIENT (client), NULL);
2056
2057   priv = client->priv;
2058
2059   g_mutex_lock (&priv->lock);
2060   if (priv->uri != NULL)
2061     result = gst_rtsp_url_copy (priv->uri);
2062   g_mutex_unlock (&priv->lock);
2063
2064   return result;
2065 }
2066
2067 /**
2068  * gst_rtsp_client_set_connection:
2069  * @client: a #GstRTSPClient
2070  * @conn: (transfer full): a #GstRTSPConnection
2071  *
2072  * Set the #GstRTSPConnection of @client. This function takes ownership of
2073  * @conn.
2074  *
2075  * Returns: %TRUE on success.
2076  */
2077 gboolean
2078 gst_rtsp_client_set_connection (GstRTSPClient * client,
2079     GstRTSPConnection * conn)
2080 {
2081   GstRTSPClientPrivate *priv;
2082   GSocket *read_socket;
2083   GSocketAddress *address;
2084   GstRTSPUrl *url;
2085   GError *error = NULL;
2086
2087   g_return_val_if_fail (GST_IS_RTSP_CLIENT (client), FALSE);
2088   g_return_val_if_fail (conn != NULL, FALSE);
2089
2090   priv = client->priv;
2091
2092   read_socket = gst_rtsp_connection_get_read_socket (conn);
2093
2094   if (!(address = g_socket_get_local_address (read_socket, &error)))
2095     goto no_address;
2096
2097   g_free (priv->server_ip);
2098   /* keep the original ip that the client connected to */
2099   if (G_IS_INET_SOCKET_ADDRESS (address)) {
2100     GInetAddress *iaddr;
2101
2102     iaddr = g_inet_socket_address_get_address (G_INET_SOCKET_ADDRESS (address));
2103
2104     /* socket might be ipv6 but adress still ipv4 */
2105     priv->is_ipv6 = g_inet_address_get_family (iaddr) == G_SOCKET_FAMILY_IPV6;
2106     priv->server_ip = g_inet_address_to_string (iaddr);
2107     g_object_unref (address);
2108   } else {
2109     priv->is_ipv6 = g_socket_get_family (read_socket) == G_SOCKET_FAMILY_IPV6;
2110     priv->server_ip = g_strdup ("unknown");
2111   }
2112
2113   GST_INFO ("client %p connected to server ip %s, ipv6 = %d", client,
2114       priv->server_ip, priv->is_ipv6);
2115
2116   url = gst_rtsp_connection_get_url (conn);
2117   GST_INFO ("added new client %p ip %s:%d", client, url->host, url->port);
2118
2119   priv->connection = conn;
2120
2121   return TRUE;
2122
2123   /* ERRORS */
2124 no_address:
2125   {
2126     GST_ERROR ("could not get remote address %s", error->message);
2127     g_error_free (error);
2128     return FALSE;
2129   }
2130 }
2131
2132 /**
2133  * gst_rtsp_client_get_connection:
2134  * @client: a #GstRTSPClient
2135  *
2136  * Get the #GstRTSPConnection of @client.
2137  *
2138  * Returns: (transfer none): the #GstRTSPConnection of @client.
2139  * The connection object returned remains valid until the client is freed.
2140  */
2141 GstRTSPConnection *
2142 gst_rtsp_client_get_connection (GstRTSPClient * client)
2143 {
2144   g_return_val_if_fail (GST_IS_RTSP_CLIENT (client), NULL);
2145
2146   return client->priv->connection;
2147 }
2148
2149 /**
2150  * gst_rtsp_client_set_send_func:
2151  * @client: a #GstRTSPClient
2152  * @func: a #GstRTSPClientSendFunc
2153  * @user_data: user data passed to @func
2154  * @notify: called when @user_data is no longer in use
2155  *
2156  * Set @func as the callback that will be called when a new message needs to be
2157  * sent to the client. @user_data is passed to @func and @notify is called when
2158  * @user_data is no longer in use.
2159  */
2160 void
2161 gst_rtsp_client_set_send_func (GstRTSPClient * client,
2162     GstRTSPClientSendFunc func, gpointer user_data, GDestroyNotify notify)
2163 {
2164   GstRTSPClientPrivate *priv;
2165   GDestroyNotify old_notify;
2166   gpointer old_data;
2167
2168   g_return_if_fail (GST_IS_RTSP_CLIENT (client));
2169
2170   priv = client->priv;
2171
2172   g_mutex_lock (&priv->send_lock);
2173   priv->send_func = func;
2174   old_notify = priv->send_notify;
2175   old_data = priv->send_data;
2176   priv->send_notify = notify;
2177   priv->send_data = user_data;
2178   g_mutex_unlock (&priv->send_lock);
2179
2180   if (old_notify)
2181     old_notify (old_data);
2182 }
2183
2184 /**
2185  * gst_rtsp_client_handle_message:
2186  * @client: a #GstRTSPClient
2187  * @message: an #GstRTSPMessage
2188  *
2189  * Let the client handle @message.
2190  *
2191  * Returns: a #GstRTSPResult.
2192  */
2193 GstRTSPResult
2194 gst_rtsp_client_handle_message (GstRTSPClient * client,
2195     GstRTSPMessage * message)
2196 {
2197   g_return_val_if_fail (GST_IS_RTSP_CLIENT (client), GST_RTSP_EINVAL);
2198   g_return_val_if_fail (message != NULL, GST_RTSP_EINVAL);
2199
2200   switch (message->type) {
2201     case GST_RTSP_MESSAGE_REQUEST:
2202       handle_request (client, message);
2203       break;
2204     case GST_RTSP_MESSAGE_RESPONSE:
2205       break;
2206     case GST_RTSP_MESSAGE_DATA:
2207       handle_data (client, message);
2208       break;
2209     default:
2210       break;
2211   }
2212   return GST_RTSP_OK;
2213 }
2214
2215 static GstRTSPResult
2216 do_send_message (GstRTSPClient * client, GstRTSPMessage * message,
2217     gboolean close, gpointer user_data)
2218 {
2219   GstRTSPClientPrivate *priv = client->priv;
2220
2221   /* send the response and store the seq number so we can wait until it's
2222    * written to the client to close the connection */
2223   return gst_rtsp_watch_send_message (priv->watch, message, close ?
2224       &priv->close_seq : NULL);
2225 }
2226
2227 static GstRTSPResult
2228 message_received (GstRTSPWatch * watch, GstRTSPMessage * message,
2229     gpointer user_data)
2230 {
2231   return gst_rtsp_client_handle_message (GST_RTSP_CLIENT (user_data), message);
2232 }
2233
2234 static GstRTSPResult
2235 message_sent (GstRTSPWatch * watch, guint cseq, gpointer user_data)
2236 {
2237   GstRTSPClient *client = GST_RTSP_CLIENT (user_data);
2238   GstRTSPClientPrivate *priv = client->priv;
2239
2240   if (priv->close_seq && priv->close_seq == cseq) {
2241     priv->close_seq = 0;
2242     close_connection (client);
2243   }
2244
2245   return GST_RTSP_OK;
2246 }
2247
2248 static GstRTSPResult
2249 closed (GstRTSPWatch * watch, gpointer user_data)
2250 {
2251   GstRTSPClient *client = GST_RTSP_CLIENT (user_data);
2252   GstRTSPClientPrivate *priv = client->priv;
2253   const gchar *tunnelid;
2254
2255   GST_INFO ("client %p: connection closed", client);
2256
2257   if ((tunnelid = gst_rtsp_connection_get_tunnelid (priv->connection))) {
2258     g_mutex_lock (&tunnels_lock);
2259     /* remove from tunnelids */
2260     g_hash_table_remove (tunnels, tunnelid);
2261     g_mutex_unlock (&tunnels_lock);
2262   }
2263
2264   gst_rtsp_client_set_send_func (client, NULL, NULL, NULL);
2265
2266   return GST_RTSP_OK;
2267 }
2268
2269 static GstRTSPResult
2270 error (GstRTSPWatch * watch, GstRTSPResult result, gpointer user_data)
2271 {
2272   GstRTSPClient *client = GST_RTSP_CLIENT (user_data);
2273   gchar *str;
2274
2275   str = gst_rtsp_strresult (result);
2276   GST_INFO ("client %p: received an error %s", client, str);
2277   g_free (str);
2278
2279   return GST_RTSP_OK;
2280 }
2281
2282 static GstRTSPResult
2283 error_full (GstRTSPWatch * watch, GstRTSPResult result,
2284     GstRTSPMessage * message, guint id, gpointer user_data)
2285 {
2286   GstRTSPClient *client = GST_RTSP_CLIENT (user_data);
2287   gchar *str;
2288
2289   str = gst_rtsp_strresult (result);
2290   GST_INFO
2291       ("client %p: error when handling message %p with id %d: %s",
2292       client, message, id, str);
2293   g_free (str);
2294
2295   return GST_RTSP_OK;
2296 }
2297
2298 static gboolean
2299 remember_tunnel (GstRTSPClient * client)
2300 {
2301   GstRTSPClientPrivate *priv = client->priv;
2302   const gchar *tunnelid;
2303
2304   /* store client in the pending tunnels */
2305   tunnelid = gst_rtsp_connection_get_tunnelid (priv->connection);
2306   if (tunnelid == NULL)
2307     goto no_tunnelid;
2308
2309   GST_INFO ("client %p: inserting tunnel session %s", client, tunnelid);
2310
2311   /* we can't have two clients connecting with the same tunnelid */
2312   g_mutex_lock (&tunnels_lock);
2313   if (g_hash_table_lookup (tunnels, tunnelid))
2314     goto tunnel_existed;
2315
2316   g_hash_table_insert (tunnels, g_strdup (tunnelid), g_object_ref (client));
2317   g_mutex_unlock (&tunnels_lock);
2318
2319   return TRUE;
2320
2321   /* ERRORS */
2322 no_tunnelid:
2323   {
2324     GST_ERROR ("client %p: no tunnelid provided", client);
2325     return FALSE;
2326   }
2327 tunnel_existed:
2328   {
2329     g_mutex_unlock (&tunnels_lock);
2330     GST_ERROR ("client %p: tunnel session %s already existed", client,
2331         tunnelid);
2332     return FALSE;
2333   }
2334 }
2335
2336 static GstRTSPStatusCode
2337 tunnel_start (GstRTSPWatch * watch, gpointer user_data)
2338 {
2339   GstRTSPClient *client = GST_RTSP_CLIENT (user_data);
2340   GstRTSPClientPrivate *priv = client->priv;
2341
2342   GST_INFO ("client %p: tunnel start (connection %p)", client,
2343       priv->connection);
2344
2345   if (!remember_tunnel (client))
2346     goto tunnel_error;
2347
2348   return GST_RTSP_STS_OK;
2349
2350   /* ERRORS */
2351 tunnel_error:
2352   {
2353     GST_ERROR ("client %p: error starting tunnel", client);
2354     return GST_RTSP_STS_SERVICE_UNAVAILABLE;
2355   }
2356 }
2357
2358 static GstRTSPResult
2359 tunnel_lost (GstRTSPWatch * watch, gpointer user_data)
2360 {
2361   GstRTSPClient *client = GST_RTSP_CLIENT (user_data);
2362   GstRTSPClientPrivate *priv = client->priv;
2363
2364   GST_WARNING ("client %p: tunnel lost (connection %p)", client,
2365       priv->connection);
2366
2367   /* ignore error, it'll only be a problem when the client does a POST again */
2368   remember_tunnel (client);
2369
2370   return GST_RTSP_OK;
2371 }
2372
2373 static GstRTSPResult
2374 tunnel_complete (GstRTSPWatch * watch, gpointer user_data)
2375 {
2376   const gchar *tunnelid;
2377   GstRTSPClient *client = GST_RTSP_CLIENT (user_data);
2378   GstRTSPClientPrivate *priv = client->priv;
2379   GstRTSPClient *oclient;
2380   GstRTSPClientPrivate *opriv;
2381
2382   GST_INFO ("client %p: tunnel complete", client);
2383
2384   /* find previous tunnel */
2385   tunnelid = gst_rtsp_connection_get_tunnelid (priv->connection);
2386   if (tunnelid == NULL)
2387     goto no_tunnelid;
2388
2389   g_mutex_lock (&tunnels_lock);
2390   if (!(oclient = g_hash_table_lookup (tunnels, tunnelid)))
2391     goto no_tunnel;
2392
2393   /* remove the old client from the table. ref before because removing it will
2394    * remove the ref to it. */
2395   g_object_ref (oclient);
2396   g_hash_table_remove (tunnels, tunnelid);
2397
2398   opriv = oclient->priv;
2399
2400   if (opriv->watch == NULL)
2401     goto tunnel_closed;
2402   g_mutex_unlock (&tunnels_lock);
2403
2404   GST_INFO ("client %p: found tunnel %p (old %p, new %p)", client, oclient,
2405       opriv->connection, priv->connection);
2406
2407   /* merge the tunnels into the first client */
2408   gst_rtsp_connection_do_tunnel (opriv->connection, priv->connection);
2409   gst_rtsp_watch_reset (opriv->watch);
2410   g_object_unref (oclient);
2411
2412   return GST_RTSP_OK;
2413
2414   /* ERRORS */
2415 no_tunnelid:
2416   {
2417     GST_ERROR ("client %p: no tunnelid provided", client);
2418     return GST_RTSP_ERROR;
2419   }
2420 no_tunnel:
2421   {
2422     g_mutex_unlock (&tunnels_lock);
2423     GST_ERROR ("client %p: tunnel session %s not found", client, tunnelid);
2424     return GST_RTSP_ERROR;
2425   }
2426 tunnel_closed:
2427   {
2428     g_mutex_unlock (&tunnels_lock);
2429     GST_ERROR ("client %p: tunnel session %s was closed", client, tunnelid);
2430     g_object_unref (oclient);
2431     return GST_RTSP_ERROR;
2432   }
2433 }
2434
2435 static GstRTSPWatchFuncs watch_funcs = {
2436   message_received,
2437   message_sent,
2438   closed,
2439   error,
2440   tunnel_start,
2441   tunnel_complete,
2442   error_full,
2443   tunnel_lost
2444 };
2445
2446 static void
2447 client_watch_notify (GstRTSPClient * client)
2448 {
2449   GstRTSPClientPrivate *priv = client->priv;
2450
2451   GST_INFO ("client %p: watch destroyed", client);
2452   priv->watch = NULL;
2453   g_signal_emit (client, gst_rtsp_client_signals[SIGNAL_CLOSED], 0, NULL);
2454   g_object_unref (client);
2455 }
2456
2457 /**
2458  * gst_rtsp_client_attach:
2459  * @client: a #GstRTSPClient
2460  * @context: (allow-none): a #GMainContext
2461  *
2462  * Attaches @client to @context. When the mainloop for @context is run, the
2463  * client will be dispatched. When @context is NULL, the default context will be
2464  * used).
2465  *
2466  * This function should be called when the client properties and urls are fully
2467  * configured and the client is ready to start.
2468  *
2469  * Returns: the ID (greater than 0) for the source within the GMainContext.
2470  */
2471 guint
2472 gst_rtsp_client_attach (GstRTSPClient * client, GMainContext * context)
2473 {
2474   GstRTSPClientPrivate *priv;
2475   guint res;
2476
2477   g_return_val_if_fail (GST_IS_RTSP_CLIENT (client), 0);
2478   priv = client->priv;
2479   g_return_val_if_fail (priv->watch == NULL, 0);
2480
2481   /* create watch for the connection and attach */
2482   priv->watch = gst_rtsp_watch_new (priv->connection, &watch_funcs,
2483       g_object_ref (client), (GDestroyNotify) client_watch_notify);
2484   gst_rtsp_client_set_send_func (client, do_send_message, priv->watch,
2485       (GDestroyNotify) gst_rtsp_watch_unref);
2486
2487   /* FIXME make this configurable. We don't want to do this yet because it will
2488    * be superceeded by a cache object later */
2489   gst_rtsp_watch_set_send_backlog (priv->watch, 0, 100);
2490
2491   GST_INFO ("attaching to context %p", context);
2492   res = gst_rtsp_watch_attach (priv->watch, context);
2493
2494   return res;
2495 }