mirror of
https://gitlab.freedesktop.org/gstreamer/gstreamer.git
synced 2024-10-06 02:32:23 +00:00
facc91a942
Avoid passing GstStructure in the add_role method, use varargs instead to construct the structure behind the scenes. We can then also use the structure name as the role and simplify some more logic.
259 lines
8.2 KiB
C
259 lines
8.2 KiB
C
/* GStreamer
|
|
* Copyright (C) 2013 Wim Taymans <wim.taymans at gmail.com>
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Library General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Library General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Library General Public
|
|
* License along with this library; if not, write to the
|
|
* Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#include <libcgroup.h>
|
|
|
|
#include <gst/gst.h>
|
|
#include <gst/rtsp-server/rtsp-server.h>
|
|
|
|
typedef struct _GstRTSPCGroupPool GstRTSPCGroupPool;
|
|
typedef struct _GstRTSPCGroupPoolClass GstRTSPCGroupPoolClass;
|
|
|
|
#define GST_TYPE_RTSP_CGROUP_POOL (gst_rtsp_cgroup_pool_get_type ())
|
|
#define GST_IS_RTSP_CGROUP_POOL(obj) (G_TYPE_CHECK_INSTANCE_TYPE ((obj), GST_TYPE_RTSP_CGROUP_POOL))
|
|
#define GST_IS_RTSP_CGROUP_POOL_CLASS(klass) (G_TYPE_CHECK_CLASS_TYPE ((klass), GST_TYPE_RTSP_CGROUP_POOL))
|
|
#define GST_RTSP_CGROUP_POOL_GET_CLASS(obj) (G_TYPE_INSTANCE_GET_CLASS ((obj), GST_TYPE_RTSP_CGROUP_POOL, GstRTSPCGroupPoolClass))
|
|
#define GST_RTSP_CGROUP_POOL(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), GST_TYPE_RTSP_CGROUP_POOL, GstRTSPCGroupPool))
|
|
#define GST_RTSP_CGROUP_POOL_CLASS(klass) (G_TYPE_CHECK_CLASS_CAST ((klass), GST_TYPE_RTSP_CGROUP_POOL, GstRTSPCGroupPoolClass))
|
|
#define GST_RTSP_CGROUP_POOL_CAST(obj) ((GstRTSPCGroupPool*)(obj))
|
|
#define GST_RTSP_CGROUP_POOL_CLASS_CAST(klass) ((GstRTSPCGroupPoolClass*)(klass))
|
|
|
|
struct _GstRTSPCGroupPool
|
|
{
|
|
GstRTSPThreadPool parent;
|
|
|
|
struct cgroup *user;
|
|
struct cgroup *admin;
|
|
};
|
|
|
|
struct _GstRTSPCGroupPoolClass
|
|
{
|
|
GstRTSPThreadPoolClass parent_class;
|
|
};
|
|
|
|
static GQuark thread_cgroup;
|
|
|
|
static void gst_rtsp_cgroup_pool_finalize (GObject * obj);
|
|
|
|
static void default_thread_enter (GstRTSPThreadPool * pool,
|
|
GstRTSPThread * thread);
|
|
static void default_configure_thread (GstRTSPThreadPool * pool,
|
|
GstRTSPThread * thread, GstRTSPClientState * state);
|
|
|
|
G_DEFINE_TYPE (GstRTSPCGroupPool, gst_rtsp_cgroup_pool,
|
|
GST_TYPE_RTSP_THREAD_POOL);
|
|
|
|
static void
|
|
gst_rtsp_cgroup_pool_class_init (GstRTSPCGroupPoolClass * klass)
|
|
{
|
|
GObjectClass *gobject_class;
|
|
GstRTSPThreadPoolClass *tpool_class;
|
|
|
|
gobject_class = G_OBJECT_CLASS (klass);
|
|
tpool_class = GST_RTSP_THREAD_POOL_CLASS (klass);
|
|
|
|
gobject_class->finalize = gst_rtsp_cgroup_pool_finalize;
|
|
|
|
tpool_class->configure_thread = default_configure_thread;
|
|
tpool_class->thread_enter = default_thread_enter;
|
|
|
|
thread_cgroup = g_quark_from_string ("cgroup.pool.thread.cgroup");
|
|
|
|
cgroup_init ();
|
|
}
|
|
|
|
static void
|
|
gst_rtsp_cgroup_pool_init (GstRTSPCGroupPool * pool)
|
|
{
|
|
pool->user = cgroup_new_cgroup ("user");
|
|
g_assert (cgroup_add_controller (pool->user, "cpu") != NULL);
|
|
pool->admin = cgroup_new_cgroup ("admin");
|
|
g_assert (cgroup_add_controller (pool->admin, "cpu") != NULL);
|
|
}
|
|
|
|
static void
|
|
gst_rtsp_cgroup_pool_finalize (GObject * obj)
|
|
{
|
|
GstRTSPCGroupPool *pool = GST_RTSP_CGROUP_POOL (obj);
|
|
|
|
GST_INFO ("finalize pool %p", pool);
|
|
|
|
cgroup_free (&pool->user);
|
|
cgroup_free (&pool->admin);
|
|
|
|
G_OBJECT_CLASS (gst_rtsp_cgroup_pool_parent_class)->finalize (obj);
|
|
}
|
|
|
|
static void
|
|
default_thread_enter (GstRTSPThreadPool * pool, GstRTSPThread * thread)
|
|
{
|
|
struct cgroup *cgroup;
|
|
|
|
cgroup = gst_mini_object_get_qdata (GST_MINI_OBJECT (thread), thread_cgroup);
|
|
if (cgroup) {
|
|
gint res = 0;
|
|
|
|
res = cgroup_attach_task (cgroup);
|
|
|
|
if (res != 0)
|
|
GST_ERROR ("error: %d (%s)", res, cgroup_strerror (res));
|
|
}
|
|
}
|
|
|
|
static void
|
|
default_configure_thread (GstRTSPThreadPool * pool,
|
|
GstRTSPThread * thread, GstRTSPClientState * state)
|
|
{
|
|
GstRTSPCGroupPool *cpool = GST_RTSP_CGROUP_POOL (pool);
|
|
const gchar *cls;
|
|
struct cgroup *cgroup;
|
|
|
|
if (state->token)
|
|
cls = gst_rtsp_token_get_string (state->token, "cgroup.pool.media.class");
|
|
else
|
|
cls = NULL;
|
|
|
|
GST_DEBUG ("manage cgroup %s", cls);
|
|
|
|
if (!g_strcmp0 (cls, "admin"))
|
|
cgroup = cpool->admin;
|
|
else
|
|
cgroup = cpool->user;
|
|
|
|
/* attach the cgroup to the thread */
|
|
gst_mini_object_set_qdata (GST_MINI_OBJECT (thread), thread_cgroup,
|
|
cgroup, NULL);
|
|
}
|
|
|
|
static gboolean
|
|
timeout (GstRTSPServer * server)
|
|
{
|
|
GstRTSPSessionPool *pool;
|
|
|
|
pool = gst_rtsp_server_get_session_pool (server);
|
|
gst_rtsp_session_pool_cleanup (pool);
|
|
g_object_unref (pool);
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
int
|
|
main (int argc, char *argv[])
|
|
{
|
|
GMainLoop *loop;
|
|
GstRTSPServer *server;
|
|
GstRTSPMountPoints *mounts;
|
|
GstRTSPMediaFactory *factory;
|
|
GstRTSPAuth *auth;
|
|
GstRTSPToken *token;
|
|
gchar *basic;
|
|
GstStructure *s;
|
|
GstRTSPPermissions *permissions;
|
|
GstRTSPThreadPool *thread_pool;
|
|
|
|
gst_init (&argc, &argv);
|
|
|
|
loop = g_main_loop_new (NULL, FALSE);
|
|
|
|
/* create a server instance */
|
|
server = gst_rtsp_server_new ();
|
|
|
|
/* get the mounts for this server, every server has a default mapper object
|
|
* that be used to map uri mount points to media factories */
|
|
mounts = gst_rtsp_server_get_mount_points (server);
|
|
|
|
/* make a media factory for a test stream. The default media factory can use
|
|
* gst-launch syntax to create pipelines.
|
|
* any launch line works as long as it contains elements named pay%d. Each
|
|
* element with pay%d names will be a stream */
|
|
factory = gst_rtsp_media_factory_new ();
|
|
gst_rtsp_media_factory_set_launch (factory, "( "
|
|
"videotestsrc ! video/x-raw,width=640,height=480,framerate=50/1 ! "
|
|
"x264enc ! rtph264pay name=pay0 pt=96 "
|
|
"audiotestsrc ! audio/x-raw,rate=8000 ! "
|
|
"alawenc ! rtppcmapay name=pay1 pt=97 " ")");
|
|
/* attach the test factory to the /test url */
|
|
gst_rtsp_mount_points_add_factory (mounts, "/test", factory);
|
|
|
|
/* allow user and admin to access this resource */
|
|
permissions = gst_rtsp_permissions_new ();
|
|
gst_rtsp_permissions_add_role (permissions, "user",
|
|
"media.factory.access", G_TYPE_BOOLEAN, TRUE,
|
|
"media.factory.construct", G_TYPE_BOOLEAN, TRUE, NULL);
|
|
gst_rtsp_permissions_add_role (permissions, "admin",
|
|
"media.factory.access", G_TYPE_BOOLEAN, TRUE,
|
|
"media.factory.construct", G_TYPE_BOOLEAN, TRUE, NULL);
|
|
gst_rtsp_media_factory_set_permissions (factory, permissions);
|
|
gst_rtsp_permissions_unref (permissions);
|
|
|
|
/* don't need the ref to the mapper anymore */
|
|
g_object_unref (mounts);
|
|
|
|
/* make a new authentication manager */
|
|
auth = gst_rtsp_auth_new ();
|
|
|
|
/* make user token */
|
|
token = gst_rtsp_token_new ();
|
|
s = gst_rtsp_token_writable_structure (token);
|
|
gst_structure_set (s, "cgroup.pool.media.class", G_TYPE_STRING, "user", NULL);
|
|
gst_structure_set (s, "media.factory.role", G_TYPE_STRING, "user", NULL);
|
|
basic = gst_rtsp_auth_make_basic ("user", "password");
|
|
gst_rtsp_auth_add_basic (auth, basic, token);
|
|
g_free (basic);
|
|
gst_rtsp_token_unref (token);
|
|
|
|
/* make admin token */
|
|
token = gst_rtsp_token_new ();
|
|
s = gst_rtsp_token_writable_structure (token);
|
|
gst_structure_set (s, "cgroup.pool.media.class", G_TYPE_STRING, "admin",
|
|
NULL);
|
|
gst_structure_set (s, "media.factory.role", G_TYPE_STRING, "admin", NULL);
|
|
basic = gst_rtsp_auth_make_basic ("admin", "power");
|
|
gst_rtsp_auth_add_basic (auth, basic, token);
|
|
g_free (basic);
|
|
gst_rtsp_token_unref (token);
|
|
|
|
/* set as the server authentication manager */
|
|
gst_rtsp_server_set_auth (server, auth);
|
|
g_object_unref (auth);
|
|
|
|
thread_pool = g_object_new (GST_TYPE_RTSP_CGROUP_POOL, NULL);
|
|
gst_rtsp_server_set_thread_pool (server, thread_pool);
|
|
g_object_unref (thread_pool);
|
|
|
|
/* attach the server to the default maincontext */
|
|
if (gst_rtsp_server_attach (server, NULL) == 0)
|
|
goto failed;
|
|
|
|
g_timeout_add_seconds (2, (GSourceFunc) timeout, server);
|
|
|
|
/* start serving */
|
|
g_print ("stream with user:password ready at rtsp://127.0.0.1:8554/test\n");
|
|
g_print ("stream with admin:power ready at rtsp://127.0.0.1:8554/test\n");
|
|
g_main_loop_run (loop);
|
|
|
|
return 0;
|
|
|
|
/* ERRORS */
|
|
failed:
|
|
{
|
|
g_print ("failed to attach the server\n");
|
|
return -1;
|
|
}
|
|
}
|