tests/src/test-session.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3 #include <string.h>
   4 #include <sys/types.h>
   5 #include <unistd.h>
   6 #include <fcntl.h>
   7 #include <grp.h>
   8 #include <xcb/xcb.h>
   9 #include <glib.h>
  10 #include <glib-object.h>
  11 #include <gio/gio.h>
  12 #include <glib-unix.h>
  13 #include <glib/gstdio.h>
  14
  15 #include "status.h"
  16
  17 static gchar *session_id;
  18
  19 static GMainLoop *loop;
  20
  21 static GString *open_fds;
  22
  23 static GKeyFile *config;
  24
  25 static xcb_connection_t *connection;
  26
  27 static gboolean
  28 sigint_cb (gpointer user_data)
  29 {
  30     status_notify ("%s TERMINATE SIGNAL=%d", session_id, SIGINT);
  31     g_main_loop_quit (loop);
  32     return TRUE;
  33 }
  34
  35 static gboolean
  36 sigterm_cb (gpointer user_data)
  37 {
  38     status_notify ("%s TERMINATE SIGNAL=%d", session_id, SIGTERM);
  39     g_main_loop_quit (loop);
  40     return TRUE;
  41 }
  42
  43 static void
  44 request_cb (const gchar *request)
  45 {
  46     gchar *r;
  47
  48     if (!request)
  49     {
  50         g_main_loop_quit (loop);
  51         return;
  52     }
  53
  54     r = g_strdup_printf ("%s LOGOUT", session_id);
  55     if (strcmp (request, r) == 0)
  56         exit (EXIT_SUCCESS);
  57     g_free (r);
  58
  59     r = g_strdup_printf ("%s CRASH", session_id);
  60     if (strcmp (request, r) == 0)
  61         kill (getpid (), SIGSEGV);
  62     g_free (r);
  63
  64     r = g_strdup_printf ("%s LOCK-SEAT", session_id);
  65     if (strcmp (request, r) == 0)
  66     {
  67         status_notify ("%s LOCK-SEAT", session_id);
  68         g_dbus_connection_call_sync (g_bus_get_sync (G_BUS_TYPE_SYSTEM, NULL, NULL),
  69                                      "org.freedesktop.DisplayManager",
  70                                      getenv ("XDG_SEAT_PATH"),
  71                                      "org.freedesktop.DisplayManager.Seat",
  72                                      "Lock",
  73                                      g_variant_new ("()"),
  74                                      G_VARIANT_TYPE ("()"),
  75                                      G_DBUS_CALL_FLAGS_NONE,
  76                                      1000,
  77                                      NULL,
  78                                      NULL);
  79     }
  80     g_free (r);
  81
  82     r = g_strdup_printf ("%s LOCK-SESSION", session_id);
  83     if (strcmp (request, r) == 0)
  84     {
  85         status_notify ("%s LOCK-SESSION", session_id);
  86         g_dbus_connection_call_sync (g_bus_get_sync (G_BUS_TYPE_SYSTEM, NULL, NULL),
  87                                      "org.freedesktop.DisplayManager",
  88                                      getenv ("XDG_SESSION_PATH"),
  89                                      "org.freedesktop.DisplayManager.Session",
  90                                      "Lock",
  91                                      g_variant_new ("()"),
  92                                      G_VARIANT_TYPE ("()"),
  93                                      G_DBUS_CALL_FLAGS_NONE,
  94                                      1000,
  95                                      NULL,
  96                                      NULL);
  97     }
  98     g_free (r);
  99
 100     r = g_strdup_printf ("%s LIST-GROUPS", session_id);
 101     if (strcmp (request, r) == 0)
 102     {
 103         int n_groups, i;
 104         gid_t *groups;
 105         GString *group_list;
 106
 107         n_groups = getgroups (0, NULL);
 108         if (n_groups < 0)
 109         {
 110             g_printerr ("Failed to get groups: %s", strerror (errno));
 111             n_groups = 0;
 112         }
 113         groups = malloc (sizeof (gid_t) * n_groups);
 114         n_groups = getgroups (n_groups, groups);
 115         group_list = g_string_new ("");
 116         for (i = 0; i < n_groups; i++)
 117         {
 118             struct group *group;
 119
 120             if (i != 0)
 121                 g_string_append (group_list, ",");
 122             group = getgrgid (groups[i]);
 123             if (group)
 124                 g_string_append (group_list, group->gr_name);
 125             else
 126                 g_string_append_printf (group_list, "%d", groups[i]);
 127         }
 128         status_notify ("%s LIST-GROUPS GROUPS=%s", session_id, group_list->str);
 129         g_string_free (group_list, TRUE);
 130         free (groups);
 131     }
 132
 133     r = g_strdup_printf ("%s READ-ENV NAME=", session_id);
 134     if (g_str_has_prefix (request, r))
 135     {
 136         const gchar *name = request + strlen (r);
 137         const gchar *value = g_getenv (name);
 138         status_notify ("%s READ-ENV NAME=%s VALUE=%s", session_id, name, value ? value : "");
 139     }
 140     g_free (r);
 141
 142     r = g_strdup_printf ("%s WRITE-STDOUT TEXT=", session_id);
 143     if (g_str_has_prefix (request, r))
 144         g_print ("%s", request + strlen (r));
 145     g_free (r);
 146
 147     r = g_strdup_printf ("%s WRITE-STDERR TEXT=", session_id);
 148     if (g_str_has_prefix (request, r))
 149         g_printerr ("%s", request + strlen (r));
 150     g_free (r);
 151
 152     r = g_strdup_printf ("%s READ FILE=", session_id);
 153     if (g_str_has_prefix (request, r))
 154     {
 155         const gchar *name = request + strlen (r);
 156         gchar *contents;
 157         GError *error = NULL;
 158
 159         if (g_file_get_contents (name, &contents, NULL, &error))
 160             status_notify ("%s READ FILE=%s TEXT=%s", session_id, name, contents);
 161         else
 162             status_notify ("%s READ FILE=%s ERROR=%s", session_id, name, error->message);
 163         g_clear_error (&error);
 164     }
 165     g_free (r);
 166
 167     r = g_strdup_printf ("%s LIST-UNKNOWN-FILE-DESCRIPTORS", session_id);
 168     if (strcmp (request, r) == 0)
 169         status_notify ("%s LIST-UNKNOWN-FILE-DESCRIPTORS FDS=%s", session_id, open_fds->str);
 170     g_free (r);
 171
 172     r = g_strdup_printf ("%s CHECK-X-AUTHORITY", session_id);
 173     if (strcmp (request, r) == 0)
 174     {
 175         gchar *xauthority;
 176         GStatBuf file_info;
 177         GString *mode_string;
 178
 179         xauthority = g_strdup (g_getenv ("XAUTHORITY"));
 180         if (!xauthority)
 181             xauthority = g_build_filename (g_get_home_dir (), ".Xauthority", NULL);
 182
 183         g_stat (xauthority, &file_info);
 184         g_free (xauthority);
 185
 186         mode_string = g_string_new ("");
 187         g_string_append_c (mode_string, file_info.st_mode & S_IRUSR ? 'r' : '-');
 188         g_string_append_c (mode_string, file_info.st_mode & S_IWUSR ? 'w' : '-');
 189         g_string_append_c (mode_string, file_info.st_mode & S_IXUSR ? 'x' : '-');
 190         g_string_append_c (mode_string, file_info.st_mode & S_IRGRP ? 'r' : '-');
 191         g_string_append_c (mode_string, file_info.st_mode & S_IWGRP ? 'w' : '-');
 192         g_string_append_c (mode_string, file_info.st_mode & S_IXGRP ? 'x' : '-');
 193         g_string_append_c (mode_string, file_info.st_mode & S_IROTH ? 'r' : '-');
 194         g_string_append_c (mode_string, file_info.st_mode & S_IWOTH ? 'w' : '-');
 195         g_string_append_c (mode_string, file_info.st_mode & S_IXOTH ? 'x' : '-');
 196         status_notify ("%s CHECK-X-AUTHORITY MODE=%s", session_id, mode_string->str);
 197         g_string_free (mode_string, TRUE);
 198     }
 199     g_free (r);
 200 }
 201
 202 int
 203 main (int argc, char **argv)
 204 {
 205     gchar *display, *xdg_seat, *xdg_vtnr, *xdg_current_desktop, *xdg_session_cookie, *mir_socket, *mir_vt, *mir_id;
 206     GString *status_text;
 207     int fd, open_max;
 208
 209     display = getenv ("DISPLAY");
 210     xdg_seat = getenv ("XDG_SEAT");
 211     xdg_vtnr = getenv ("XDG_VTNR");
 212     xdg_current_desktop = getenv ("XDG_CURRENT_DESKTOP");
 213     xdg_session_cookie = getenv ("XDG_SESSION_COOKIE");
 214     mir_socket = getenv ("MIR_SOCKET");
 215     mir_vt = getenv ("MIR_SERVER_VT");
 216     mir_id = getenv ("MIR_ID");
 217     if (display)
 218     {
 219         if (display[0] == ':')
 220             session_id = g_strdup_printf ("SESSION-X-%s", display + 1);
 221         else
 222             session_id = g_strdup_printf ("SESSION-X-%s", display);
 223     }
 224     else if (mir_id)
 225         session_id = g_strdup_printf ("SESSION-MIR-%s", mir_id);
 226     else if (mir_socket || mir_vt)
 227         session_id = g_strdup ("SESSION-MIR");
 228     else
 229         session_id = g_strdup ("SESSION-UNKNOWN");
 230
 231     open_fds = g_string_new ("");
 232     open_max = sysconf (_SC_OPEN_MAX);
 233     for (fd = STDERR_FILENO + 1; fd < open_max; fd++)
 234     {
 235         if (fcntl (fd, F_GETFD) >= 0)
 236             g_string_append_printf (open_fds, "%d,", fd);
 237     }
 238     if (g_str_has_suffix (open_fds->str, ","))
 239         open_fds->str[strlen (open_fds->str) - 1] = '\0';
 240
 241 #if !defined(GLIB_VERSION_2_36)
 242     g_type_init ();
 243 #endif
 244
 245     loop = g_main_loop_new (NULL, FALSE);
 246
 247     g_unix_signal_add (SIGINT, sigint_cb, NULL);
 248     g_unix_signal_add (SIGTERM, sigterm_cb, NULL);
 249
 250     status_connect (request_cb);
 251
 252     status_text = g_string_new ("");
 253     g_string_printf (status_text, "%s START", session_id);
 254     if (xdg_seat)
 255         g_string_append_printf (status_text, " XDG_SEAT=%s", xdg_seat);
 256     if (xdg_vtnr)
 257         g_string_append_printf (status_text, " XDG_VTNR=%s", xdg_vtnr);
 258     if (xdg_current_desktop)
 259         g_string_append_printf (status_text, " XDG_CURRENT_DESKTOP=%s", xdg_current_desktop);
 260     if (xdg_session_cookie)
 261         g_string_append_printf (status_text, " XDG_SESSION_COOKIE=%s", xdg_session_cookie);
 262     if (mir_vt > 0)
 263         g_string_append_printf (status_text, " MIR_SERVER_VT=%s", mir_vt);
 264     if (argc > 1)
 265         g_string_append_printf (status_text, " NAME=%s", argv[1]);
 266     g_string_append_printf (status_text, " USER=%s", getenv ("USER"));
 267     status_notify (status_text->str);
 268     g_string_free (status_text, TRUE);
 269
 270     config = g_key_file_new ();
 271     g_key_file_load_from_file (config, g_build_filename (g_getenv ("LIGHTDM_TEST_ROOT"), "script", NULL), G_KEY_FILE_NONE, NULL);
 272
 273     if (display)
 274     {
 275         connection = xcb_connect (NULL, NULL);
 276         if (xcb_connection_has_error (connection))
 277         {
 278             status_notify ("%s CONNECT-XSERVER-ERROR", session_id);
 279             return EXIT_FAILURE;
 280         }
 281         status_notify ("%s CONNECT-XSERVER", session_id);
 282     }
 283
 284     g_main_loop_run (loop);
 285
 286     return EXIT_SUCCESS;
 287 }