Merge from trunk

[sojka/lightdm.git] / tests / src / test-session.c

diff --git a/tests/src/test-session.c b/tests/src/test-session.c
index 5208ffb0dcc139f8121fe64fe3dfbb1f701d2969..5f2a04d444ac5c35571fdd8e141332e835012ce7 100644 (file)
--- a/tests/src/test-session.c
+++ b/tests/src/test-session.c
@@ -9,6 +9,8 @@
 #include <glib.h>
 #include <glib-object.h>
 #include <gio/gio.h>
+#include <glib-unix.h>
+#include <glib/gstdio.h>
 
 #include "status.h"
 
@@ -103,6 +105,11 @@ request_cb (const gchar *request)
         GString *group_list;
 
         n_groups = getgroups (0, NULL);
+        if (n_groups < 0)
+        {
+            g_printerr ("Failed to get groups: %s", strerror (errno));
+            n_groups = 0;
+        }
         groups = malloc (sizeof (gid_t) * n_groups);
         n_groups = getgroups (n_groups, groups);
         group_list = g_string_new ("");
@@ -161,21 +168,67 @@ request_cb (const gchar *request)
     if (strcmp (request, r) == 0)
         status_notify ("%s LIST-UNKNOWN-FILE-DESCRIPTORS FDS=%s", session_id, open_fds->str);
     g_free (r);
+
+    r = g_strdup_printf ("%s CHECK-X-AUTHORITY", session_id);
+    if (strcmp (request, r) == 0)
+    {
+        gchar *xauthority;
+        GStatBuf file_info;
+        GString *mode_string;
+
+        xauthority = g_strdup (g_getenv ("XAUTHORITY"));
+        if (!xauthority)
+            xauthority = g_build_filename (g_get_home_dir (), ".Xauthority", NULL);
+
+        g_stat (xauthority, &file_info);
+        g_free (xauthority);
+
+        mode_string = g_string_new ("");
+        g_string_append_c (mode_string, file_info.st_mode & S_IRUSR ? 'r' : '-');
+        g_string_append_c (mode_string, file_info.st_mode & S_IWUSR ? 'w' : '-');
+        g_string_append_c (mode_string, file_info.st_mode & S_IXUSR ? 'x' : '-');
+        g_string_append_c (mode_string, file_info.st_mode & S_IRGRP ? 'r' : '-');
+        g_string_append_c (mode_string, file_info.st_mode & S_IWGRP ? 'w' : '-');
+        g_string_append_c (mode_string, file_info.st_mode & S_IXGRP ? 'x' : '-');
+        g_string_append_c (mode_string, file_info.st_mode & S_IROTH ? 'r' : '-');
+        g_string_append_c (mode_string, file_info.st_mode & S_IWOTH ? 'w' : '-');
+        g_string_append_c (mode_string, file_info.st_mode & S_IXOTH ? 'x' : '-');
+        status_notify ("%s CHECK-X-AUTHORITY MODE=%s", session_id, mode_string->str);
+        g_string_free (mode_string, TRUE);
+    }
+    g_free (r);
 }
 
 int
 main (int argc, char **argv)
 {
-    gchar *display;
+    gchar *display, *xdg_seat, *xdg_vtnr, *xdg_current_desktop, *xdg_session_cookie, *xdg_session_class, *desktop_session, *mir_socket, *mir_vt, *mir_id;
+    GString *status_text;
     int fd, open_max;
 
     display = getenv ("DISPLAY");
-    if (display == NULL)
-        session_id = g_strdup ("SESSION-?");
-    else if (display[0] == ':')
-        session_id = g_strdup_printf ("SESSION-X-%s", display + 1);
+    xdg_seat = getenv ("XDG_SEAT");
+    xdg_vtnr = getenv ("XDG_VTNR");
+    xdg_current_desktop = getenv ("XDG_CURRENT_DESKTOP");
+    xdg_session_cookie = getenv ("XDG_SESSION_COOKIE");
+    xdg_session_class = getenv ("XDG_SESSION_CLASS");
+    desktop_session = getenv ("DESKTOP_SESSION");
+    mir_socket = getenv ("MIR_SOCKET");
+    mir_vt = getenv ("MIR_SERVER_VT");
+    mir_id = getenv ("MIR_SERVER_NAME");
+    if (display)
+    {
+        if (display[0] == ':')
+            session_id = g_strdup_printf ("SESSION-X-%s", display + 1);
+        else
+            session_id = g_strdup_printf ("SESSION-X-%s", display);
+    }
+    else if (mir_id)
+        session_id = g_strdup_printf ("SESSION-MIR-%s", mir_id);
+    else if (mir_socket || mir_vt)
+        session_id = g_strdup ("SESSION-MIR");
     else
-        session_id = g_strdup_printf ("SESSION-X-%s", display);
+        session_id = g_strdup ("SESSION-UNKNOWN");
 
     open_fds = g_string_new ("");
     open_max = sysconf (_SC_OPEN_MAX);
@@ -198,24 +251,42 @@ main (int argc, char **argv)
 
     status_connect (request_cb);
 
+    status_text = g_string_new ("");
+    g_string_printf (status_text, "%s START", session_id);
+    if (xdg_seat)
+        g_string_append_printf (status_text, " XDG_SEAT=%s", xdg_seat);
+    if (xdg_vtnr)
+        g_string_append_printf (status_text, " XDG_VTNR=%s", xdg_vtnr);
+    if (xdg_current_desktop)
+        g_string_append_printf (status_text, " XDG_CURRENT_DESKTOP=%s", xdg_current_desktop);
+    if (xdg_session_cookie)
+        g_string_append_printf (status_text, " XDG_SESSION_COOKIE=%s", xdg_session_cookie);
+    if (xdg_session_class)
+        g_string_append_printf (status_text, " XDG_SESSION_CLASS=%s", xdg_session_class);
+    if (desktop_session)
+        g_string_append_printf (status_text, " DESKTOP_SESSION=%s", desktop_session);
+    if (mir_vt > 0)
+        g_string_append_printf (status_text, " MIR_SERVER_VT=%s", mir_vt);
     if (argc > 1)
-        status_notify ("%s START NAME=%s USER=%s", session_id, argv[1], getenv ("USER"));
-    else
-        status_notify ("%s START USER=%s", session_id, getenv ("USER"));
+        g_string_append_printf (status_text, " NAME=%s", argv[1]);
+    g_string_append_printf (status_text, " USER=%s", getenv ("USER"));
+    status_notify (status_text->str);
+    g_string_free (status_text, TRUE);
 
     config = g_key_file_new ();
     g_key_file_load_from_file (config, g_build_filename (g_getenv ("LIGHTDM_TEST_ROOT"), "script", NULL), G_KEY_FILE_NONE, NULL);
 
-    connection = xcb_connect (NULL, NULL);
-
-    if (xcb_connection_has_error (connection))
+    if (display)
     {
-        status_notify ("%s CONNECT-XSERVER-ERROR", session_id);
-        return EXIT_FAILURE;
+        connection = xcb_connect (NULL, NULL);
+        if (xcb_connection_has_error (connection))
+        {
+            status_notify ("%s CONNECT-XSERVER-ERROR", session_id);
+            return EXIT_FAILURE;
+        }
+        status_notify ("%s CONNECT-XSERVER", session_id);
     }
 
-    status_notify ("%s CONNECT-XSERVER", session_id);
-
     g_main_loop_run (loop);
 
     return EXIT_SUCCESS;