examples/test-cgroups.c

   1 /* GStreamer
   2  * Copyright (C) 2013 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 /* Runs a pipeline and clasifies the media pipelines based on the
  21  * authenticated user.
  22  *
  23  * This test requires 2 cpu cgroups to exist named 'user' and 'admin'.
  24  * The rtsp server should have permission to add its threads to the
  25  * cgroups.
  26  *
  27  *   sudo cgcreate -t uid:gid -g cpu:/user
  28  *   sudo cgcreate -t uid:gid -g cpu:/admin
  29  *
  30  * With -t you can give the user and group access to the task file to
  31  * write the thread ids. The user running the server can be used.
  32  *
  33  * Then you would want to change the cpu shares assigned to each group:
  34  *
  35  *   sudo cgset -r cpu.shares=100 user
  36  *   sudo cgset -r cpu.shares=1024 admin
  37  *
  38  * Then start clients for 'user' until the stream is degraded because of
  39  * lack of CPU. Then start a client for 'admin' and check that the stream
  40  * is not degraded.
  41  */
  42
  43 #include <libcgroup.h>
  44
  45 #include <gst/gst.h>
  46 #include <gst/rtsp-server/rtsp-server.h>
  47
  48 typedef struct _GstRTSPCGroupPool GstRTSPCGroupPool;
  49 typedef struct _GstRTSPCGroupPoolClass GstRTSPCGroupPoolClass;
  50
  51 #define GST_TYPE_RTSP_CGROUP_POOL              (gst_rtsp_cgroup_pool_get_type ())
  52 #define GST_IS_RTSP_CGROUP_POOL(obj)           (G_TYPE_CHECK_INSTANCE_TYPE ((obj), GST_TYPE_RTSP_CGROUP_POOL))
  53 #define GST_IS_RTSP_CGROUP_POOL_CLASS(klass)   (G_TYPE_CHECK_CLASS_TYPE ((klass), GST_TYPE_RTSP_CGROUP_POOL))
  54 #define GST_RTSP_CGROUP_POOL_GET_CLASS(obj)    (G_TYPE_INSTANCE_GET_CLASS ((obj), GST_TYPE_RTSP_CGROUP_POOL, GstRTSPCGroupPoolClass))
  55 #define GST_RTSP_CGROUP_POOL(obj)              (G_TYPE_CHECK_INSTANCE_CAST ((obj), GST_TYPE_RTSP_CGROUP_POOL, GstRTSPCGroupPool))
  56 #define GST_RTSP_CGROUP_POOL_CLASS(klass)      (G_TYPE_CHECK_CLASS_CAST ((klass), GST_TYPE_RTSP_CGROUP_POOL, GstRTSPCGroupPoolClass))
  57 #define GST_RTSP_CGROUP_POOL_CAST(obj)         ((GstRTSPCGroupPool*)(obj))
  58 #define GST_RTSP_CGROUP_POOL_CLASS_CAST(klass) ((GstRTSPCGroupPoolClass*)(klass))
  59
  60 struct _GstRTSPCGroupPool
  61 {
  62   GstRTSPThreadPool parent;
  63
  64   struct cgroup *user;
  65   struct cgroup *admin;
  66 };
  67
  68 struct _GstRTSPCGroupPoolClass
  69 {
  70   GstRTSPThreadPoolClass parent_class;
  71 };
  72
  73 static GQuark thread_cgroup;
  74
  75 static void gst_rtsp_cgroup_pool_finalize (GObject * obj);
  76
  77 static void default_thread_enter (GstRTSPThreadPool * pool,
  78     GstRTSPThread * thread);
  79 static void default_configure_thread (GstRTSPThreadPool * pool,
  80     GstRTSPThread * thread, GstRTSPContext * ctx);
  81
  82 static GType gst_rtsp_cgroup_pool_get_type (void);
  83
  84 G_DEFINE_TYPE (GstRTSPCGroupPool, gst_rtsp_cgroup_pool,
  85     GST_TYPE_RTSP_THREAD_POOL);
  86
  87 static void
  88 gst_rtsp_cgroup_pool_class_init (GstRTSPCGroupPoolClass * klass)
  89 {
  90   GObjectClass *gobject_class;
  91   GstRTSPThreadPoolClass *tpool_class;
  92
  93   gobject_class = G_OBJECT_CLASS (klass);
  94   tpool_class = GST_RTSP_THREAD_POOL_CLASS (klass);
  95
  96   gobject_class->finalize = gst_rtsp_cgroup_pool_finalize;
  97
  98   tpool_class->configure_thread = default_configure_thread;
  99   tpool_class->thread_enter = default_thread_enter;
 100
 101   thread_cgroup = g_quark_from_string ("cgroup.pool.thread.cgroup");
 102
 103   cgroup_init ();
 104 }
 105
 106 static void
 107 gst_rtsp_cgroup_pool_init (GstRTSPCGroupPool * pool)
 108 {
 109   pool->user = cgroup_new_cgroup ("user");
 110   g_assert (cgroup_add_controller (pool->user, "cpu") != NULL);
 111   pool->admin = cgroup_new_cgroup ("admin");
 112   g_assert (cgroup_add_controller (pool->admin, "cpu") != NULL);
 113 }
 114
 115 static void
 116 gst_rtsp_cgroup_pool_finalize (GObject * obj)
 117 {
 118   GstRTSPCGroupPool *pool = GST_RTSP_CGROUP_POOL (obj);
 119
 120   GST_INFO ("finalize pool %p", pool);
 121
 122   cgroup_free (&pool->user);
 123   cgroup_free (&pool->admin);
 124
 125   G_OBJECT_CLASS (gst_rtsp_cgroup_pool_parent_class)->finalize (obj);
 126 }
 127
 128 static void
 129 default_thread_enter (GstRTSPThreadPool * pool, GstRTSPThread * thread)
 130 {
 131   struct cgroup *cgroup;
 132
 133   cgroup = gst_mini_object_get_qdata (GST_MINI_OBJECT (thread), thread_cgroup);
 134   if (cgroup) {
 135     gint res = 0;
 136
 137     res = cgroup_attach_task (cgroup);
 138
 139     if (res != 0)
 140       GST_ERROR ("error: %d (%s)", res, cgroup_strerror (res));
 141   }
 142 }
 143
 144 static void
 145 default_configure_thread (GstRTSPThreadPool * pool,
 146     GstRTSPThread * thread, GstRTSPContext * ctx)
 147 {
 148   GstRTSPCGroupPool *cpool = GST_RTSP_CGROUP_POOL (pool);
 149   const gchar *cls;
 150   struct cgroup *cgroup;
 151
 152   if (ctx->token)
 153     cls = gst_rtsp_token_get_string (ctx->token, "cgroup.pool.media.class");
 154   else
 155     cls = NULL;
 156
 157   GST_DEBUG ("manage cgroup %s", cls);
 158
 159   if (!g_strcmp0 (cls, "admin"))
 160     cgroup = cpool->admin;
 161   else
 162     cgroup = cpool->user;
 163
 164   /* attach the cgroup to the thread */
 165   gst_mini_object_set_qdata (GST_MINI_OBJECT (thread), thread_cgroup,
 166       cgroup, NULL);
 167 }
 168
 169 static gboolean
 170 timeout (GstRTSPServer * server)
 171 {
 172   GstRTSPSessionPool *pool;
 173
 174   pool = gst_rtsp_server_get_session_pool (server);
 175   gst_rtsp_session_pool_cleanup (pool);
 176   g_object_unref (pool);
 177
 178   return TRUE;
 179 }
 180
 181 int
 182 main (int argc, char *argv[])
 183 {
 184   GMainLoop *loop;
 185   GstRTSPServer *server;
 186   GstRTSPMountPoints *mounts;
 187   GstRTSPMediaFactory *factory;
 188   GstRTSPAuth *auth;
 189   GstRTSPToken *token;
 190   gchar *basic;
 191   GstRTSPThreadPool *thread_pool;
 192
 193   gst_init (&argc, &argv);
 194
 195   loop = g_main_loop_new (NULL, FALSE);
 196
 197   /* create a server instance */
 198   server = gst_rtsp_server_new ();
 199
 200   /* get the mounts for this server, every server has a default mapper object
 201    * that be used to map uri mount points to media factories */
 202   mounts = gst_rtsp_server_get_mount_points (server);
 203
 204   /* make a media factory for a test stream. The default media factory can use
 205    * gst-launch syntax to create pipelines.
 206    * any launch line works as long as it contains elements named pay%d. Each
 207    * element with pay%d names will be a stream */
 208   factory = gst_rtsp_media_factory_new ();
 209   gst_rtsp_media_factory_set_launch (factory, "( "
 210       "videotestsrc ! video/x-raw,width=640,height=480,framerate=50/1 ! "
 211       "x264enc ! rtph264pay name=pay0 pt=96 "
 212       "audiotestsrc ! audio/x-raw,rate=8000 ! "
 213       "alawenc ! rtppcmapay name=pay1 pt=97 " ")");
 214   /* attach the test factory to the /test url */
 215   gst_rtsp_mount_points_add_factory (mounts, "/test", factory);
 216
 217   /* allow user and admin to access this resource */
 218   gst_rtsp_media_factory_add_role (factory, "user",
 219       "media.factory.access", G_TYPE_BOOLEAN, TRUE,
 220       "media.factory.construct", G_TYPE_BOOLEAN, TRUE, NULL);
 221   gst_rtsp_media_factory_add_role (factory, "admin",
 222       "media.factory.access", G_TYPE_BOOLEAN, TRUE,
 223       "media.factory.construct", G_TYPE_BOOLEAN, TRUE, NULL);
 224
 225   /* don't need the ref to the mapper anymore */
 226   g_object_unref (mounts);
 227
 228   /* make a new authentication manager */
 229   auth = gst_rtsp_auth_new ();
 230
 231   /* make user token */
 232   token = gst_rtsp_token_new ("cgroup.pool.media.class", G_TYPE_STRING, "user",
 233       "media.factory.role", G_TYPE_STRING, "user", NULL);
 234   basic = gst_rtsp_auth_make_basic ("user", "password");
 235   gst_rtsp_auth_add_basic (auth, basic, token);
 236   g_free (basic);
 237   gst_rtsp_token_unref (token);
 238
 239   /* make admin token */
 240   token = gst_rtsp_token_new ("cgroup.pool.media.class", G_TYPE_STRING, "admin",
 241       "media.factory.role", G_TYPE_STRING, "admin", NULL);
 242   basic = gst_rtsp_auth_make_basic ("admin", "power");
 243   gst_rtsp_auth_add_basic (auth, basic, token);
 244   g_free (basic);
 245   gst_rtsp_token_unref (token);
 246
 247   /* set as the server authentication manager */
 248   gst_rtsp_server_set_auth (server, auth);
 249   g_object_unref (auth);
 250
 251   thread_pool = g_object_new (GST_TYPE_RTSP_CGROUP_POOL, NULL);
 252   gst_rtsp_server_set_thread_pool (server, thread_pool);
 253   g_object_unref (thread_pool);
 254
 255   /* attach the server to the default maincontext */
 256   if (gst_rtsp_server_attach (server, NULL) == 0)
 257     goto failed;
 258
 259   g_timeout_add_seconds (2, (GSourceFunc) timeout, server);
 260
 261   /* start serving */
 262   g_print ("stream with user:password ready at rtsp://127.0.0.1:8554/test\n");
 263   g_print ("stream with admin:power ready at rtsp://127.0.0.1:8554/test\n");
 264   g_main_loop_run (loop);
 265
 266   return 0;
 267
 268   /* ERRORS */
 269 failed:
 270   {
 271     g_print ("failed to attach the server\n");
 272     return -1;
 273   }
 274 }