gst/rtsp-server/rtsp-session-pool.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., 59 Temple Place - Suite 330,
  17  * Boston, MA 02111-1307, USA.
  18  */
  19
  20 #include "rtsp-session-pool.h"
  21
  22 #undef DEBUG
  23
  24 static void gst_rtsp_session_pool_finalize (GObject * object);
  25
  26
  27 G_DEFINE_TYPE (GstRTSPSessionPool, gst_rtsp_session_pool, G_TYPE_OBJECT);
  28
  29 static void
  30 gst_rtsp_session_pool_class_init (GstRTSPSessionPoolClass * klass)
  31 {
  32   GObjectClass *gobject_class;
  33
  34   gobject_class = G_OBJECT_CLASS (klass);
  35
  36   gobject_class->finalize = gst_rtsp_session_pool_finalize;
  37 }
  38
  39 static void
  40 gst_rtsp_session_pool_init (GstRTSPSessionPool * pool)
  41 {
  42   pool->lock = g_mutex_new ();
  43   pool->sessions = g_hash_table_new_full (g_str_hash, g_str_equal,
  44                   g_free, g_object_unref);
  45 }
  46
  47 static void
  48 gst_rtsp_session_pool_finalize (GObject * object)
  49 {
  50   GstRTSPSessionPool * pool = GST_RTSP_SESSION_POOL (object);
  51
  52   g_mutex_free (pool->lock);
  53   g_hash_table_unref (pool->sessions);
  54
  55   G_OBJECT_CLASS (gst_rtsp_session_pool_parent_class)->finalize (object);
  56 }
  57
  58 /**
  59  * gst_rtsp_session_pool_new:
  60  *
  61  * Create a new #GstRTSPSessionPool instance.
  62  */
  63 GstRTSPSessionPool *
  64 gst_rtsp_session_pool_new (void)
  65 {
  66   GstRTSPSessionPool *result;
  67
  68   result = g_object_new (GST_TYPE_RTSP_SESSION_POOL, NULL);
  69
  70   return result;
  71 }
  72
  73 /**
  74  * gst_rtsp_session_pool_find:
  75  * @pool: the pool to search
  76  * @sessionid: the session id
  77  *
  78  * Find the session with @sessionid in @pool.
  79  *
  80  * Returns: the #GstRTSPSession with @sessionid or %NULL when the session did
  81  * not exist. g_object_unref() after usage.
  82  */
  83 GstRTSPSession *
  84 gst_rtsp_session_pool_find (GstRTSPSessionPool *pool, const gchar *sessionid)
  85 {
  86   GstRTSPSession *result;
  87
  88   g_return_val_if_fail (GST_IS_RTSP_SESSION_POOL (pool), NULL);
  89   g_return_val_if_fail (sessionid != NULL, NULL);
  90
  91   g_mutex_lock (pool->lock);
  92   result = g_hash_table_lookup (pool->sessions, sessionid);
  93   if (result)
  94     g_object_ref (result);
  95   g_mutex_unlock (pool->lock);
  96
  97   return result;
  98 }
  99
 100 static gchar *
 101 create_session_id (void)
 102 {
 103   gchar id[16];
 104   gint i;
 105
 106   for (i = 0; i < 16; i++) {
 107     id[i] = g_random_int_range ('a', 'z');
 108   }
 109
 110   return g_strndup (id, 16);
 111 }
 112
 113 /**
 114  * gst_rtsp_session_pool_create:
 115  * @pool: a #GstRTSPSessionPool
 116  *
 117  * Create a new #GstRTSPSession object in @pool.
 118  *
 119  * Returns: a new #GstRTSPSession.
 120  */
 121 GstRTSPSession *
 122 gst_rtsp_session_pool_create (GstRTSPSessionPool *pool)
 123 {
 124   GstRTSPSession *result = NULL;
 125   gchar *id;
 126
 127   g_return_val_if_fail (GST_IS_RTSP_SESSION_POOL (pool), NULL);
 128
 129   do {
 130     /* start by creating a new random session id, we assume that this is random
 131      * enough to not cause a collision, which we will check later  */
 132     id = create_session_id ();
 133
 134     g_mutex_lock (pool->lock);
 135     /* check if the sessionid existed */
 136     result = g_hash_table_lookup (pool->sessions, id);
 137     if (result) {
 138       /* found, retry with a different session id*/
 139       result = NULL;
 140     }
 141     else {
 142       /* not found, create session and insert it in the pool */
 143       result = gst_rtsp_session_new (id);
 144       /* take additional ref for the pool */
 145       g_object_ref (result);
 146       g_hash_table_insert (pool->sessions, result->sessionid, result);
 147     }
 148     g_mutex_unlock (pool->lock);
 149
 150     g_free (id);
 151   } while (result == NULL);
 152
 153   return result;
 154 }
 155
 156 /**
 157  * gst_rtsp_session_pool_remove:
 158  * @pool: a #GstRTSPSessionPool
 159  * @sess: a #GstRTSPSession
 160  *
 161  * Remove @sess from @pool and Clean it up.
 162  *
 163  * Returns: a new #GstRTSPSession.
 164  */
 165 void
 166 gst_rtsp_session_pool_remove (GstRTSPSessionPool *pool, GstRTSPSession *sess)
 167 {
 168   gboolean found;
 169
 170   g_return_if_fail (GST_IS_RTSP_SESSION_POOL (pool));
 171   g_return_if_fail (GST_IS_RTSP_SESSION (sess));
 172
 173   g_mutex_lock (pool->lock);
 174   found = g_hash_table_remove (pool->sessions, sess);
 175   g_mutex_unlock (pool->lock);
 176 }
 177