media: add lock to protect state changes
[platform/upstream/gstreamer.git] / gst / rtsp-server / rtsp-server.c
index 25f3666..0edf635 100644 (file)
@@ -28,6 +28,7 @@
 /* #define DEFAULT_ADDRESS         "::0" */
 #define DEFAULT_SERVICE         "8554"
 #define DEFAULT_BACKLOG         5
+#define DEFAULT_MAX_THREADS     0
 
 /* Define to use the SO_LINGER option so that the server sockets can be resused
  * sooner. Disabled for now because it is not very well implemented by various
@@ -44,6 +45,7 @@ enum
 
   PROP_SESSION_POOL,
   PROP_MEDIA_MAPPING,
+  PROP_MAX_THREADS,
   PROP_LAST
 };
 
@@ -58,6 +60,8 @@ G_DEFINE_TYPE (GstRTSPServer, gst_rtsp_server, G_TYPE_OBJECT);
 GST_DEBUG_CATEGORY_STATIC (rtsp_server_debug);
 #define GST_CAT_DEFAULT rtsp_server_debug
 
+typedef struct _ClientContext ClientContext;
+
 static guint gst_rtsp_server_signals[SIGNAL_LAST] = { 0 };
 
 static void gst_rtsp_server_get_property (GObject * object, guint propid,
@@ -66,6 +70,7 @@ static void gst_rtsp_server_set_property (GObject * object, guint propid,
     const GValue * value, GParamSpec * pspec);
 static void gst_rtsp_server_finalize (GObject * object);
 
+static gpointer do_loop (ClientContext * ctx);
 static GstRTSPClient *default_create_client (GstRTSPServer * server);
 static gboolean default_accept_client (GstRTSPServer * server,
     GstRTSPClient * client, GSocket * socket, GError ** error);
@@ -150,6 +155,18 @@ gst_rtsp_server_class_init (GstRTSPServerClass * klass)
           "The media mapping to use for client session",
           GST_TYPE_RTSP_MEDIA_MAPPING,
           G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
+  /**
+   * GstRTSPServer::max-threads:
+   *
+   * The maximum amount of threads to use for client connections. A value of
+   * 0 means to use only the mainloop, -1 means an unlimited amount of
+   * threads.
+   */
+  g_object_class_install_property (gobject_class, PROP_MAX_THREADS,
+      g_param_spec_int ("max-threads", "Max Threads",
+          "The maximum amount of threads to use for client connections "
+          "(0 = only mainloop, -1 = unlimited)", -1, G_MAXINT,
+          DEFAULT_MAX_THREADS, G_PARAM_READWRITE | G_PARAM_STATIC_STRINGS));
 
   gst_rtsp_server_signals[SIGNAL_CLIENT_CONNECTED] =
       g_signal_new ("client-connected", G_TYPE_FROM_CLASS (gobject_class),
@@ -160,6 +177,8 @@ gst_rtsp_server_class_init (GstRTSPServerClass * klass)
   klass->create_client = default_create_client;
   klass->accept_client = default_accept_client;
 
+  klass->pool = g_thread_pool_new ((GFunc) do_loop, klass, -1, FALSE, NULL);
+
   GST_DEBUG_CATEGORY_INIT (rtsp_server_debug, "rtspserver", 0, "GstRTSPServer");
 }
 
@@ -526,6 +545,49 @@ gst_rtsp_server_get_auth (GstRTSPServer * server)
   return result;
 }
 
+/**
+ * gst_rtsp_server_set_max_threads:
+ * @server: a #GstRTSPServer
+ * @max_threads: maximum threads
+ *
+ * Set the maximum threads used by the server to handle client requests.
+ * A value of 0 will use the server mainloop, a value of -1 will use an
+ * unlimited number of threads.
+ */
+void
+gst_rtsp_server_set_max_threads (GstRTSPServer * server, gint max_threads)
+{
+  g_return_if_fail (GST_IS_RTSP_SERVER (server));
+
+  GST_RTSP_SERVER_LOCK (server);
+  server->max_threads = max_threads;
+  GST_RTSP_SERVER_UNLOCK (server);
+}
+
+/**
+ * gst_rtsp_server_get_max_threads:
+ * @server: a #GstRTSPServer
+ *
+ * Get the maximum number of threads used for client connections.
+ * See gst_rtsp_server_set_max_threads().
+ *
+ * Returns: the maximum number of threads.
+ */
+gint
+gst_rtsp_server_get_max_threads (GstRTSPServer * server)
+{
+  gint res;
+
+  g_return_val_if_fail (GST_IS_RTSP_SERVER (server), -1);
+
+  GST_RTSP_SERVER_LOCK (server);
+  res = server->max_threads;
+  GST_RTSP_SERVER_UNLOCK (server);
+
+  return res;
+}
+
+
 static void
 gst_rtsp_server_get_property (GObject * object, guint propid,
     GValue * value, GParamSpec * pspec)
@@ -551,6 +613,9 @@ gst_rtsp_server_get_property (GObject * object, guint propid,
     case PROP_MEDIA_MAPPING:
       g_value_take_object (value, gst_rtsp_server_get_media_mapping (server));
       break;
+    case PROP_MAX_THREADS:
+      g_value_set_int (value, gst_rtsp_server_get_max_threads (server));
+      break;
     default:
       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
   }
@@ -578,6 +643,9 @@ gst_rtsp_server_set_property (GObject * object, guint propid,
     case PROP_MEDIA_MAPPING:
       gst_rtsp_server_set_media_mapping (server, g_value_get_object (value));
       break;
+    case PROP_MAX_THREADS:
+      gst_rtsp_server_set_max_threads (server, g_value_get_int (value));
+      break;
     default:
       G_OBJECT_WARN_INVALID_PROPERTY_ID (object, propid, pspec);
   }
@@ -749,13 +817,13 @@ close_error:
   }
 }
 
-typedef struct
+struct _ClientContext
 {
   GstRTSPServer *server;
   GMainLoop *loop;
   GMainContext *context;
   GstRTSPClient *client;
-} ClientContext;
+};
 
 static void
 free_client_context (ClientContext * ctx)
@@ -814,8 +882,7 @@ manage_client (GstRTSPServer * server, GstRTSPClient * client)
   ctx = g_slice_new0 (ClientContext);
   ctx->server = server;
   ctx->client = client;
-#if 1
-  {
+  if (server->max_threads == 0) {
     GSource *source;
 
     /* find the context to add the watch */
@@ -823,12 +890,10 @@ manage_client (GstRTSPServer * server, GstRTSPClient * client)
       ctx->context = g_main_context_ref (g_source_get_context (source));
     else
       ctx->context = NULL;
+  } else {
+    ctx->context = g_main_context_new ();
+    ctx->loop = g_main_loop_new (ctx->context, TRUE);
   }
-#else
-  ctx->context = g_main_context_new ();
-  ctx->loop = g_main_loop_new (ctx->context, TRUE);
-  ctx->dothread = TRUE;
-#endif
   gst_rtsp_client_attach (client, ctx->context);
 
   GST_RTSP_SERVER_LOCK (server);
@@ -837,10 +902,9 @@ manage_client (GstRTSPServer * server, GstRTSPClient * client)
   GST_RTSP_SERVER_UNLOCK (server);
 
   if (ctx->loop) {
-    GThread *thread;
+    GstRTSPServerClass *klass = GST_RTSP_SERVER_GET_CLASS (server);
 
-    thread = g_thread_new ("MainLoop Thread", (GThreadFunc) do_loop, ctx);
-    g_thread_unref (thread);
+    g_thread_pool_push (klass->pool, ctx, NULL);
   }
 }
 
@@ -1033,7 +1097,7 @@ GSource *
 gst_rtsp_server_create_source (GstRTSPServer * server,
     GCancellable * cancellable, GError ** error)
 {
-  GSocket *socket;
+  GSocket *socket, *old;
   GSource *source;
 
   g_return_val_if_fail (GST_IS_RTSP_SERVER (server), NULL);
@@ -1041,7 +1105,14 @@ gst_rtsp_server_create_source (GstRTSPServer * server,
   socket = gst_rtsp_server_create_socket (server, NULL, error);
   if (socket == NULL)
     goto no_socket;
+
+  GST_RTSP_SERVER_LOCK (server);
+  old = server->socket;
   server->socket = g_object_ref (socket);
+  GST_RTSP_SERVER_UNLOCK (server);
+
+  if (old)
+    g_object_unref (old);
 
   /* create a watch for reads (new connections) and possible errors */
   source = g_socket_create_source (socket, G_IO_IN |