tests/src/test-gobject-greeter.c

   1 /* -*- Mode: C; indent-tabs-mode: nil; tab-width: 4 -*- */
   2
   3 #include <stdlib.h>
   4 #include <stdio.h>
   5 #include <string.h>
   6 #include <xcb/xcb.h>
   7 #include <lightdm.h>
   8
   9 #include "status.h"
  10
  11 static gchar *greeter_id;
  12 static GMainLoop *loop;
  13 static LightDMGreeter *greeter;
  14 static xcb_connection_t *connection = NULL;
  15 static GKeyFile *config;
  16
  17 static void
  18 show_message_cb (LightDMGreeter *greeter, const gchar *text, LightDMMessageType type)
  19 {
  20     status_notify ("%s SHOW-MESSAGE TEXT=\"%s\"", greeter_id, text);
  21 }
  22
  23 static void
  24 show_prompt_cb (LightDMGreeter *greeter, const gchar *text, LightDMPromptType type)
  25 {
  26     status_notify ("%s SHOW-PROMPT TEXT=\"%s\"", greeter_id, text);
  27 }
  28
  29 static void
  30 authentication_complete_cb (LightDMGreeter *greeter)
  31 {
  32     if (lightdm_greeter_get_authentication_user (greeter))
  33         status_notify ("%s AUTHENTICATION-COMPLETE USERNAME=%s AUTHENTICATED=%s",
  34                        greeter_id,
  35                        lightdm_greeter_get_authentication_user (greeter),
  36                        lightdm_greeter_get_is_authenticated (greeter) ? "TRUE" : "FALSE");
  37     else
  38         status_notify ("%s AUTHENTICATION-COMPLETE AUTHENTICATED=%s",
  39                        greeter_id,
  40                        lightdm_greeter_get_is_authenticated (greeter) ? "TRUE" : "FALSE");
  41 }
  42
  43 static void
  44 autologin_timer_expired_cb (LightDMGreeter *greeter)
  45 {
  46     status_notify ("%s AUTOLOGIN-TIMER-EXPIRED", greeter_id);
  47 }
  48
  49 static void
  50 signal_cb (int signum)
  51 {
  52     status_notify ("%s TERMINATE SIGNAL=%d", greeter_id, signum);
  53     exit (EXIT_SUCCESS);
  54 }
  55
  56 static void
  57 request_cb (const gchar *request)
  58 {
  59     gchar *r;
  60
  61     if (!request)
  62     {
  63         g_main_loop_quit (loop);
  64         return;
  65     }
  66
  67     r = g_strdup_printf ("%s AUTHENTICATE", greeter_id);
  68     if (strcmp (request, r) == 0)
  69         lightdm_greeter_authenticate (greeter, NULL);
  70     g_free (r);
  71
  72     r = g_strdup_printf ("%s AUTHENTICATE USERNAME=", greeter_id);
  73     if (g_str_has_prefix (request, r))
  74         lightdm_greeter_authenticate (greeter, request + strlen (r));
  75     g_free (r);
  76
  77     r = g_strdup_printf ("%s AUTHENTICATE-GUEST", greeter_id);
  78     if (strcmp (request, r) == 0)
  79         lightdm_greeter_authenticate_as_guest (greeter);
  80     g_free (r);
  81
  82     r = g_strdup_printf ("%s AUTHENTICATE-AUTOLOGIN", greeter_id);
  83     if (strcmp (request, r) == 0)
  84         lightdm_greeter_authenticate_autologin (greeter);
  85     g_free (r);
  86
  87     r = g_strdup_printf ("%s AUTHENTICATE-REMOTE SESSION=", greeter_id);
  88     if (g_str_has_prefix (request, r))
  89         lightdm_greeter_authenticate_remote (greeter, request + strlen (r), NULL);
  90     g_free (r);
  91
  92     r = g_strdup_printf ("%s RESPOND TEXT=\"", greeter_id);
  93     if (g_str_has_prefix (request, r))
  94     {
  95         gchar *text = g_strdup (request + strlen (r));
  96         text[strlen (text) - 1] = '\0';
  97         lightdm_greeter_respond (greeter, text);
  98         g_free (text);
  99     }
 100     g_free (r);
 101
 102     r = g_strdup_printf ("%s CANCEL-AUTHENTICATION", greeter_id);
 103     if (strcmp (request, r) == 0)
 104         lightdm_greeter_cancel_authentication (greeter);
 105     g_free (r);
 106
 107     r = g_strdup_printf ("%s START-SESSION", greeter_id);
 108     if (strcmp (request, r) == 0)
 109     {
 110         if (!lightdm_greeter_start_session_sync (greeter, NULL, NULL))
 111             status_notify ("%s SESSION-FAILED", greeter_id);
 112     }
 113     g_free (r);
 114
 115     r = g_strdup_printf ("%s START-SESSION SESSION=", greeter_id);
 116     if (g_str_has_prefix (request, r))
 117     {
 118         if (!lightdm_greeter_start_session_sync (greeter, request + strlen (r), NULL))
 119             status_notify ("%s SESSION-FAILED", greeter_id);
 120     }
 121     g_free (r);
 122
 123     r = g_strdup_printf ("%s LOG-LAYOUT USERNAME=", greeter_id);
 124     if (g_str_has_prefix (request, r))
 125     {
 126         LightDMUser *user;
 127         const gchar *username, *layout;
 128
 129         username = request + strlen (r);
 130         user = lightdm_user_list_get_user_by_name (lightdm_user_list_get_instance (), username);
 131         layout = lightdm_user_get_layout (user);
 132
 133         status_notify ("%s LOG-LAYOUT USERNAME=%s LAYOUT='%s'", greeter_id, username, layout ? layout : "");
 134     }
 135     g_free (r);
 136
 137     r = g_strdup_printf ("%s LOG-LAYOUTS USERNAME=", greeter_id);
 138     if (g_str_has_prefix (request, r))
 139     {
 140         LightDMUser *user;
 141         const gchar *username;
 142         const gchar * const *layouts;
 143         int i;
 144
 145         username = request + strlen (r);
 146         user = lightdm_user_list_get_user_by_name (lightdm_user_list_get_instance (), username);
 147         layouts = lightdm_user_get_layouts (user);
 148
 149         for (i = 0; layouts[i]; i++)
 150             status_notify ("%s LOG-LAYOUTS USERNAME=%s LAYOUT='%s'", greeter_id, username, layouts[i]);
 151     }
 152     g_free (r);
 153
 154     r = g_strdup_printf ("%s LOG-LANGUAGE USERNAME=", greeter_id);
 155     if (g_str_has_prefix (request, r))
 156     {
 157         LightDMUser *user;
 158         const gchar *username, *language;
 159
 160         username = request + strlen (r);
 161         user = lightdm_user_list_get_user_by_name (lightdm_user_list_get_instance (), username);
 162         language = lightdm_user_get_language (user);
 163
 164         status_notify ("%s LOG-LANGUAGE USERNAME=%s LANGUAGE=%s", greeter_id, username, language ? language : "");
 165     }
 166     g_free (r);
 167
 168     r = g_strdup_printf ("%s GET-CAN-SUSPEND", greeter_id);
 169     if (strcmp (request, r) == 0)
 170     {
 171         gboolean can_suspend = lightdm_get_can_suspend ();
 172         status_notify ("%s CAN-SUSPEND ALLOWED=%s", greeter_id, can_suspend ? "TRUE" : "FALSE");
 173     }
 174     g_free (r);
 175
 176     r = g_strdup_printf ("%s SUSPEND", greeter_id);
 177     if (strcmp (request, r) == 0)
 178     {
 179         GError *error = NULL;
 180         if (!lightdm_suspend (&error))
 181             status_notify ("%s FAIL-SUSPEND", greeter_id);
 182         g_clear_error (&error);
 183     }
 184     g_free (r);
 185
 186     r = g_strdup_printf ("%s GET-CAN-HIBERNATE", greeter_id);
 187     if (strcmp (request, r) == 0)
 188     {
 189         gboolean can_hibernate = lightdm_get_can_hibernate ();
 190         status_notify ("%s CAN-HIBERNATE ALLOWED=%s", greeter_id, can_hibernate ? "TRUE" : "FALSE");
 191     }
 192     g_free (r);
 193
 194     r = g_strdup_printf ("%s HIBERNATE", greeter_id);
 195     if (strcmp (request, r) == 0)
 196     {
 197         GError *error = NULL;
 198         if (!lightdm_hibernate (&error))
 199             status_notify ("%s FAIL-HIBERNATE", greeter_id);
 200         g_clear_error (&error);
 201     }
 202     g_free (r);
 203
 204     r = g_strdup_printf ("%s GET-CAN-RESTART", greeter_id);
 205     if (strcmp (request, r) == 0)
 206     {
 207         gboolean can_restart = lightdm_get_can_restart ();
 208         status_notify ("%s CAN-RESTART ALLOWED=%s", greeter_id, can_restart ? "TRUE" : "FALSE");
 209     }
 210     g_free (r);
 211
 212     r = g_strdup_printf ("%s RESTART", greeter_id);
 213     if (strcmp (request, r) == 0)
 214     {
 215         GError *error = NULL;
 216         if (!lightdm_restart (&error))
 217             status_notify ("%s FAIL-RESTART", greeter_id);
 218         g_clear_error (&error);
 219     }
 220     g_free (r);
 221
 222     r = g_strdup_printf ("%s GET-CAN-SHUTDOWN", greeter_id);
 223     if (strcmp (request, r) == 0)
 224     {
 225         gboolean can_shutdown = lightdm_get_can_shutdown ();
 226         status_notify ("%s CAN-SHUTDOWN ALLOWED=%s", greeter_id, can_shutdown ? "TRUE" : "FALSE");
 227     }
 228     g_free (r);
 229
 230     r = g_strdup_printf ("%s SHUTDOWN", greeter_id);
 231     if (strcmp (request, r) == 0)
 232     {
 233         GError *error = NULL;
 234         if (!lightdm_shutdown (&error))
 235             status_notify ("%s FAIL-SHUTDOWN", greeter_id);
 236         g_clear_error (&error);
 237     }
 238     g_free (r);
 239 }
 240
 241 int
 242 main (int argc, char **argv)
 243 {
 244     gchar *display;
 245
 246     signal (SIGINT, signal_cb);
 247     signal (SIGTERM, signal_cb);
 248
 249 #if !defined(GLIB_VERSION_2_36)
 250     g_type_init ();
 251 #endif
 252
 253     display = getenv ("DISPLAY");
 254     if (display == NULL)
 255         greeter_id = g_strdup ("GREETER-?");
 256     else if (display[0] == ':')
 257         greeter_id = g_strdup_printf ("GREETER-X-%s", display + 1);
 258     else
 259         greeter_id = g_strdup_printf ("GREETER-X-%s", display);
 260
 261     loop = g_main_loop_new (NULL, FALSE);
 262
 263     status_connect (request_cb);
 264
 265     status_notify ("%s START", greeter_id);
 266
 267     config = g_key_file_new ();
 268     g_key_file_load_from_file (config, g_build_filename (g_getenv ("LIGHTDM_TEST_ROOT"), "script", NULL), G_KEY_FILE_NONE, NULL);
 269
 270     if (g_key_file_has_key (config, "test-greeter-config", "return-value", NULL))
 271     {
 272         int return_value = g_key_file_get_integer (config, "test-greeter-config", "return-value", NULL);
 273         status_notify ("%s EXIT CODE=%d", greeter_id, return_value);
 274         return return_value;
 275     }
 276
 277     connection = xcb_connect (NULL, NULL);
 278
 279     if (xcb_connection_has_error (connection))
 280     {
 281         status_notify ("%s FAIL-CONNECT-XSERVER", greeter_id);
 282         return EXIT_FAILURE;
 283     }
 284
 285     status_notify ("%s CONNECT-XSERVER", greeter_id);
 286
 287     greeter = lightdm_greeter_new ();
 288     g_signal_connect (greeter, "show-message", G_CALLBACK (show_message_cb), NULL);
 289     g_signal_connect (greeter, "show-prompt", G_CALLBACK (show_prompt_cb), NULL);
 290     g_signal_connect (greeter, "authentication-complete", G_CALLBACK (authentication_complete_cb), NULL);
 291     g_signal_connect (greeter, "autologin-timer-expired", G_CALLBACK (autologin_timer_expired_cb), NULL);
 292
 293     status_notify ("%s CONNECT-TO-DAEMON", greeter_id);
 294     if (!lightdm_greeter_connect_sync (greeter, NULL))
 295     {
 296         status_notify ("%s FAIL-CONNECT-DAEMON", greeter_id);
 297         return EXIT_FAILURE;
 298     }
 299
 300     status_notify ("%s CONNECTED-TO-DAEMON", greeter_id);
 301
 302     if (lightdm_greeter_get_select_user_hint (greeter))
 303         status_notify ("%s SELECT-USER-HINT USERNAME=%s", greeter_id, lightdm_greeter_get_select_user_hint (greeter));
 304     if (lightdm_greeter_get_lock_hint (greeter))
 305         status_notify ("%s LOCK-HINT", greeter_id);
 306
 307     g_main_loop_run (loop);
 308
 309     return EXIT_SUCCESS;
 310 }