examples/test-auth.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 <gst/gst.h>
  21
  22 #include <gst/rtsp-server/rtsp-server.h>
  23
  24 static gboolean
  25 remove_func (GstRTSPSessionPool * pool, GstRTSPSession * session,
  26     GstRTSPServer * server)
  27 {
  28   return GST_RTSP_FILTER_REMOVE;
  29 }
  30
  31 static gboolean
  32 remove_sessions (GstRTSPServer * server)
  33 {
  34   GstRTSPSessionPool *pool;
  35
  36   g_print ("removing all sessions\n");
  37   pool = gst_rtsp_server_get_session_pool (server);
  38   gst_rtsp_session_pool_filter (pool,
  39       (GstRTSPSessionPoolFilterFunc) remove_func, server);
  40   g_object_unref (pool);
  41
  42   return FALSE;
  43 }
  44
  45 static gboolean
  46 timeout (GstRTSPServer * server)
  47 {
  48   GstRTSPSessionPool *pool;
  49
  50   pool = gst_rtsp_server_get_session_pool (server);
  51   gst_rtsp_session_pool_cleanup (pool);
  52   g_object_unref (pool);
  53
  54   return TRUE;
  55 }
  56
  57 int
  58 main (int argc, char *argv[])
  59 {
  60   GMainLoop *loop;
  61   GstRTSPServer *server;
  62   GstRTSPMountPoints *mounts;
  63   GstRTSPMediaFactory *factory;
  64   GstRTSPAuth *auth;
  65   GstRTSPToken *token;
  66   gchar *basic;
  67   GstStructure *s;
  68   GstRTSPPermissions *permissions;
  69
  70   gst_init (&argc, &argv);
  71
  72   loop = g_main_loop_new (NULL, FALSE);
  73
  74   /* create a server instance */
  75   server = gst_rtsp_server_new ();
  76
  77   /* get the mounts for this server, every server has a default mapper object
  78    * that be used to map uri mount points to media factories */
  79   mounts = gst_rtsp_server_get_mount_points (server);
  80
  81
  82   /* make a media factory for a test stream. The default media factory can use
  83    * gst-launch syntax to create pipelines.
  84    * any launch line works as long as it contains elements named pay%d. Each
  85    * element with pay%d names will be a stream */
  86   factory = gst_rtsp_media_factory_new ();
  87   gst_rtsp_media_factory_set_launch (factory, "( "
  88       "videotestsrc ! video/x-raw,width=352,height=288,framerate=15/1 ! "
  89       "x264enc ! rtph264pay name=pay0 pt=96 "
  90       "audiotestsrc ! audio/x-raw,rate=8000 ! "
  91       "alawenc ! rtppcmapay name=pay1 pt=97 " ")");
  92   /* attach the test factory to the /test url */
  93   gst_rtsp_mount_points_add_factory (mounts, "/test", factory);
  94
  95   /* allow user and admin to access this resource */
  96   permissions = gst_rtsp_permissions_new ();
  97   gst_rtsp_permissions_add_role (permissions, "user",
  98       "media.factory.access", G_TYPE_BOOLEAN, TRUE,
  99       "media.factory.construct", G_TYPE_BOOLEAN, TRUE, NULL);
 100   gst_rtsp_permissions_add_role (permissions, "admin",
 101       "media.factory.access", G_TYPE_BOOLEAN, TRUE,
 102       "media.factory.construct", G_TYPE_BOOLEAN, TRUE, NULL);
 103   /* admin2 can look at the media but not construct so he gets a
 104    * 401 Unauthorized */
 105   gst_rtsp_permissions_add_role (permissions, "admin2",
 106       "media.factory.access", G_TYPE_BOOLEAN, TRUE,
 107       "media.factory.construct", G_TYPE_BOOLEAN, FALSE, NULL);
 108   gst_rtsp_media_factory_set_permissions (factory, permissions);
 109   gst_rtsp_permissions_unref (permissions);
 110
 111   /* make another factory */
 112   factory = gst_rtsp_media_factory_new ();
 113   gst_rtsp_media_factory_set_launch (factory, "( "
 114       "videotestsrc ! video/x-raw,width=352,height=288,framerate=30/1 ! "
 115       "x264enc ! rtph264pay name=pay0 pt=96 )");
 116   /* attach the test factory to the /test url */
 117   gst_rtsp_mount_points_add_factory (mounts, "/test2", factory);
 118
 119   /* allow admin2 to access this resource */
 120   permissions = gst_rtsp_permissions_new ();
 121   /* user and admin have no permissions so they can't even see the
 122    * media and get a 404 Not Found */
 123   gst_rtsp_permissions_add_role (permissions, "admin2",
 124       "media.factory.access", G_TYPE_BOOLEAN, TRUE,
 125       "media.factory.construct", G_TYPE_BOOLEAN, TRUE, NULL);
 126   gst_rtsp_media_factory_set_permissions (factory, permissions);
 127   gst_rtsp_permissions_unref (permissions);
 128
 129   /* don't need the ref to the mapper anymore */
 130   g_object_unref (mounts);
 131
 132   /* make a new authentication manager */
 133   auth = gst_rtsp_auth_new ();
 134
 135   /* make user token */
 136   token = gst_rtsp_token_new ();
 137   s = gst_rtsp_token_writable_structure (token);
 138   gst_structure_set (s, "resources.class", G_TYPE_STRING, "user", NULL);
 139   gst_structure_set (s, "media.factory.role", G_TYPE_STRING, "user", NULL);
 140   basic = gst_rtsp_auth_make_basic ("user", "password");
 141   gst_rtsp_auth_add_basic (auth, basic, token);
 142   g_free (basic);
 143   gst_rtsp_token_unref (token);
 144
 145   /* make admin token */
 146   token = gst_rtsp_token_new ();
 147   s = gst_rtsp_token_writable_structure (token);
 148   gst_structure_set (s, "resources.class", G_TYPE_STRING, "admin", NULL);
 149   gst_structure_set (s, "media.factory.role", G_TYPE_STRING, "admin", NULL);
 150   basic = gst_rtsp_auth_make_basic ("admin", "power");
 151   gst_rtsp_auth_add_basic (auth, basic, token);
 152   g_free (basic);
 153   gst_rtsp_token_unref (token);
 154
 155   /* make admin2 token */
 156   token = gst_rtsp_token_new ();
 157   s = gst_rtsp_token_writable_structure (token);
 158   gst_structure_set (s, "resources.class", G_TYPE_STRING, "admin", NULL);
 159   gst_structure_set (s, "media.factory.role", G_TYPE_STRING, "admin2", NULL);
 160   basic = gst_rtsp_auth_make_basic ("admin2", "power2");
 161   gst_rtsp_auth_add_basic (auth, basic, token);
 162   g_free (basic);
 163   gst_rtsp_token_unref (token);
 164
 165   /* set as the server authentication manager */
 166   gst_rtsp_server_set_auth (server, auth);
 167   g_object_unref (auth);
 168
 169   /* attach the server to the default maincontext */
 170   if (gst_rtsp_server_attach (server, NULL) == 0)
 171     goto failed;
 172
 173   g_timeout_add_seconds (2, (GSourceFunc) timeout, server);
 174   g_timeout_add_seconds (10, (GSourceFunc) remove_sessions, server);
 175
 176   /* start serving */
 177   g_print ("stream with user:password ready at rtsp://127.0.0.1:8554/test\n");
 178   g_print ("stream with admin:power ready at rtsp://127.0.0.1:8554/test\n");
 179   g_print ("stream with admin2:power2 ready at rtsp://127.0.0.1:8554/test2\n");
 180   g_main_loop_run (loop);
 181
 182   return 0;
 183
 184   /* ERRORS */
 185 failed:
 186   {
 187     g_print ("failed to attach the server\n");
 188     return -1;
 189   }
 190 }