]> rtime.felk.cvut.cz Git - sojka/lightdm.git/blob - src/seat.c
Fix up final tests
[sojka/lightdm.git] / src / seat.c
1 /*
2  * Copyright (C) 2010-2011 Robert Ancell.
3  * Author: Robert Ancell <robert.ancell@canonical.com>
4  * 
5  * This program is free software: you can redistribute it and/or modify it under
6  * the terms of the GNU General Public License as published by the Free Software
7  * Foundation, either version 3 of the License, or (at your option) any later
8  * version. See http://www.gnu.org/copyleft/gpl.html the full text of the
9  * license.
10  */
11
12 #include <stdlib.h>
13 #include <string.h>
14 #include <sys/wait.h>
15
16 #include "seat.h"
17 #include "configuration.h"
18 #include "guest-account.h"
19 #include "greeter.h"
20
21 enum {
22     SESSION_ADDED,
23     SESSION_REMOVED,
24     STOPPED,
25     LAST_SIGNAL
26 };
27 static guint signals[LAST_SIGNAL] = { 0 };
28
29 struct SeatPrivate
30 {
31     /* Configuration for this seat */
32     GHashTable *properties;
33
34     /* TRUE if able to switch users */
35     gboolean can_switch;
36
37     /* TRUE if display server can be shared for sessions */
38     gboolean share_display_server;
39
40     /* The display servers on this seat */
41     GList *display_servers;
42
43     /* The sessions on this seat */
44     GList *sessions;
45
46     /* The session to set active when it starts */
47     Session *session_to_activate;
48
49     /* TRUE if stopping this seat (waiting for displays to stop) */
50     gboolean stopping;
51
52     /* TRUE if stopped */
53     gboolean stopped;
54 };
55
56 /* PAM services to use */
57 #define GREETER_SERVICE   "lightdm-greeter"
58 #define USER_SERVICE      "lightdm"
59 #define AUTOLOGIN_SERVICE "lightdm-autologin"
60
61 G_DEFINE_TYPE (Seat, seat, G_TYPE_OBJECT);
62
63 typedef struct
64 {
65     const gchar *name;
66     GType type;
67 } SeatModule;
68 static GHashTable *seat_modules = NULL;
69
70 // FIXME: Make a get_display_server() that re-uses display servers if supported
71 static DisplayServer *create_display_server (Seat *seat);
72 static Greeter *create_greeter_session (Seat *seat);
73
74 void
75 seat_register_module (const gchar *name, GType type)
76 {
77     SeatModule *module;
78
79     if (!seat_modules)
80         seat_modules = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, NULL);
81
82     g_debug ("Registered seat module %s", name);
83
84     module = g_malloc0 (sizeof (SeatModule));
85     module->name = g_strdup (name);
86     module->type = type;
87     g_hash_table_insert (seat_modules, g_strdup (name), module);
88 }
89
90 Seat *
91 seat_new (const gchar *module_name)
92 {
93     Seat *seat;
94     SeatModule *m = NULL;
95   
96     g_return_val_if_fail (module_name != NULL, NULL);
97
98     if (seat_modules)
99         m = g_hash_table_lookup (seat_modules, module_name);
100     if (!m)
101         return NULL;
102
103     seat = g_object_new (m->type, NULL);
104
105     return seat;
106 }
107
108 void
109 seat_set_property (Seat *seat, const gchar *name, const gchar *value)
110 {
111     g_return_if_fail (seat != NULL);
112     g_hash_table_insert (seat->priv->properties, g_strdup (name), g_strdup (value));
113 }
114
115 const gchar *
116 seat_get_string_property (Seat *seat, const gchar *name)
117 {
118     g_return_val_if_fail (seat != NULL, NULL);
119     return g_hash_table_lookup (seat->priv->properties, name);
120 }
121
122 gboolean
123 seat_get_boolean_property (Seat *seat, const gchar *name)
124 {
125     return g_strcmp0 (seat_get_string_property (seat, name), "true") == 0;
126 }
127
128 gint
129 seat_get_integer_property (Seat *seat, const gchar *name)
130 {
131     const gchar *value;
132
133     value = seat_get_string_property (seat, name);
134     return value ? atoi (value) : 0;
135 }
136
137 void
138 seat_set_can_switch (Seat *seat, gboolean can_switch)
139 {
140     g_return_if_fail (seat != NULL);
141
142     seat->priv->can_switch = can_switch;
143 }
144
145 void
146 seat_set_share_display_server (Seat *seat, gboolean share_display_server)
147 {
148     g_return_if_fail (seat != NULL);
149
150     seat->priv->share_display_server = share_display_server;
151 }
152
153 gboolean
154 seat_start (Seat *seat)
155 {
156     g_return_val_if_fail (seat != NULL, FALSE);
157   
158     SEAT_GET_CLASS (seat)->setup (seat);
159     return SEAT_GET_CLASS (seat)->start (seat);
160 }
161
162 GList *
163 seat_get_sessions (Seat *seat)
164 {
165     g_return_val_if_fail (seat != NULL, NULL);
166     return seat->priv->sessions;
167 }
168
169 void
170 seat_set_active_session (Seat *seat, Session *session)
171 {
172     GList *link;
173
174     g_return_if_fail (seat != NULL);
175
176     SEAT_GET_CLASS (seat)->set_active_session (seat, session);
177   
178     /* Stop any greeters */
179     for (link = seat->priv->sessions; link; link = link->next)
180     {
181         Session *s = link->data;
182
183         if (s == session || session_get_is_stopping (s))
184             continue;
185
186         if (IS_GREETER (s))
187         {
188             g_debug ("Stopping greeter");
189             session_stop (s);
190         }
191     }
192 }
193
194 Session *
195 seat_get_active_session (Seat *seat)
196 {
197     g_return_val_if_fail (seat != NULL, NULL);
198     return SEAT_GET_CLASS (seat)->get_active_session (seat);
199 }
200
201 gboolean
202 seat_get_can_switch (Seat *seat)
203 {
204     g_return_val_if_fail (seat != NULL, FALSE);
205     return seat->priv->can_switch;
206 }
207
208 gboolean
209 seat_get_allow_guest (Seat *seat)
210 {
211     g_return_val_if_fail (seat != NULL, FALSE);
212     return seat_get_boolean_property (seat, "allow-guest") && guest_account_is_installed ();
213 }
214
215 gboolean
216 seat_get_greeter_allow_guest (Seat *seat)
217 {
218     g_return_val_if_fail (seat != NULL, FALSE);  
219     return seat_get_allow_guest (seat) && seat_get_boolean_property (seat, "greeter-allow-guest");
220 }
221
222 static gboolean
223 run_script (Seat *seat, DisplayServer *display_server, const gchar *script_name, User *user)
224 {
225     Process *script;
226     gboolean result = FALSE;
227   
228     script = process_new ();
229
230     process_set_command (script, script_name);
231
232     /* Set POSIX variables */
233     process_set_clear_environment (script, TRUE);
234     process_set_env (script, "SHELL", "/bin/sh");
235
236     /* Variables required for regression tests */
237     if (g_getenv ("LIGHTDM_TEST_ROOT"))
238     {
239         process_set_env (script, "LIGHTDM_TEST_ROOT", g_getenv ("LIGHTDM_TEST_ROOT"));
240         process_set_env (script, "LD_PRELOAD", g_getenv ("LD_PRELOAD"));
241         process_set_env (script, "LD_LIBRARY_PATH", g_getenv ("LD_LIBRARY_PATH"));
242         process_set_env (script, "PATH", g_getenv ("PATH"));
243     }
244     else
245         process_set_env (script, "PATH", "/usr/local/bin:/usr/bin:/bin");
246
247     if (user)
248     {
249         process_set_env (script, "USER", user_get_name (user));
250         process_set_env (script, "LOGNAME", user_get_name (user));
251         process_set_env (script, "HOME", user_get_home_directory (user));
252     }
253     else
254         process_set_env (script, "HOME", "/");
255
256     SEAT_GET_CLASS (seat)->run_script (seat, display_server, script);
257
258     if (process_start (script, TRUE))
259     {
260         int exit_status;
261
262         exit_status = process_get_exit_status (script);
263         if (WIFEXITED (exit_status))
264         {
265             g_debug ("Exit status of %s: %d", script_name, WEXITSTATUS (exit_status));
266             result = WEXITSTATUS (exit_status) == EXIT_SUCCESS;
267         }
268     }
269
270     g_object_unref (script);
271
272     return result;
273 }
274
275 static void
276 seat_real_run_script (Seat *seat, DisplayServer *display_server, Process *process)
277 {  
278 }
279
280 static void
281 emit_upstart_signal (const gchar *signal)
282 {
283     g_return_if_fail (signal != NULL);
284     g_return_if_fail (signal[0] != 0);
285
286     if (getuid () != 0)
287         return;
288
289     gchar *cmd = g_strdup_printf ("initctl -q emit %s DISPLAY_MANAGER=lightdm", signal);
290     g_spawn_command_line_async (cmd, NULL); /* OK if it fails, probably not installed */
291     g_free (cmd);
292 }
293
294 static void
295 check_stopped (Seat *seat)
296 {
297     if (seat->priv->stopping &&
298         !seat->priv->stopped &&
299         g_list_length (seat->priv->display_servers) == 0 &&
300         g_list_length (seat->priv->sessions) == 0)
301     {
302         seat->priv->stopped = TRUE;
303         g_debug ("Seat stopped");
304         g_signal_emit (seat, signals[STOPPED], 0);
305     }
306 }
307
308 static void
309 display_server_stopped_cb (DisplayServer *display_server, Seat *seat)
310 {
311     GList *link;
312     Session *active_session;
313
314     g_debug ("Display server stopped");
315
316     g_signal_handlers_disconnect_matched (display_server, G_SIGNAL_MATCH_DATA, 0, 0, NULL, NULL, seat);
317     seat->priv->display_servers = g_list_remove (seat->priv->display_servers, display_server);
318
319     if (seat->priv->stopping)
320     {
321         check_stopped (seat);
322         g_object_unref (display_server);
323         return;
324     }
325
326     /* Stop all sessions on this display server */
327     for (link = seat->priv->sessions; link; link = link->next)
328     {
329         Session *session = link->data;
330
331         if (session_get_display_server (session) != display_server || session_get_is_stopping (session))
332             continue;
333
334         /* Stop seat if this is the only display server and it failed to start a greeter */
335         if (IS_GREETER (session) &&
336             !session_get_is_started (session) &&
337             g_list_length (seat->priv->display_servers) == 0)
338         {
339             g_debug ("Stopping seat, greeter display server failed to start");
340             seat_stop (seat);
341         }
342
343         g_debug ("Stopping session");
344         session_stop (session);
345     }
346
347     /* If we were the active session, switch to a greeter */
348     active_session = seat_get_active_session (seat);
349     if (!active_session || session_get_display_server (active_session) == display_server)
350     {
351         g_debug ("Active display server stopped, starting greeter");
352         seat_switch_to_greeter (seat);
353     }
354
355     g_object_unref (display_server);
356 }
357
358 static void
359 start_session (Seat *seat, Session *session)
360 {
361     Greeter *greeter_session;
362
363     if (session_start (session))
364         return;
365
366     if (IS_GREETER (session))
367     {
368         g_debug ("Failed to start greeter");
369         display_server_stop (session_get_display_server (session));
370         return;
371     }
372
373     g_debug ("Failed to start session, starting greeter");
374     greeter_session = create_greeter_session (seat);
375     if (session_get_is_guest (session))
376         greeter_set_hint (greeter_session, "select-guest", "true");
377     else
378         greeter_set_hint (greeter_session, "select-user", session_get_username (session));
379     if (seat->priv->session_to_activate)
380         g_object_unref (seat->priv->session_to_activate);
381     seat->priv->session_to_activate = g_object_ref (greeter_session);
382
383     if (seat->priv->share_display_server)
384         session_set_display_server (SESSION (greeter_session), session_get_display_server (session));
385     else
386     {
387         DisplayServer *display_server;
388
389         display_server = create_display_server (seat);
390         if (!display_server_start (display_server))
391         {
392             g_debug ("Failed to start display server for greeter");
393             seat_stop (seat);
394         }
395
396         session_set_display_server (session, display_server);
397     }
398
399     start_session (seat, SESSION (greeter_session));
400 }
401
402 static void
403 run_session (Seat *seat, Session *session)
404 {
405     const gchar *script;
406
407     if (IS_GREETER (session))
408         script = seat_get_string_property (seat, "greeter-setup-script");
409     else
410         script = seat_get_string_property (seat, "session-setup-script");
411     if (script && !run_script (seat, session_get_display_server (session), script, NULL))
412     {
413         Greeter *greeter_session;
414
415         g_debug ("Switching to greeter due to failed setup script");
416
417         greeter_session = create_greeter_session (seat);
418         if (seat->priv->session_to_activate)
419             g_object_unref (seat->priv->session_to_activate);
420         seat->priv->session_to_activate = g_object_ref (greeter_session);
421
422         if (seat->priv->share_display_server)
423             session_set_display_server (SESSION (greeter_session), session_get_display_server (session));
424         else
425         {
426             DisplayServer *display_server;
427
428             display_server = create_display_server (seat);
429             if (!display_server_start (display_server))
430             {
431                 g_debug ("Failed to start display server for greeter");
432                 seat_stop (seat);
433             }
434
435             session_set_display_server (SESSION (greeter_session), display_server);         
436         }
437
438         start_session (seat, SESSION (greeter_session));
439
440         /* Stop failed session */
441         session_stop (session);
442       
443         return;
444     }
445
446     session_run (session);
447
448     // FIXME: Wait until the session is ready
449
450     if (session == seat->priv->session_to_activate)
451     {
452         seat_set_active_session (seat, session);
453         g_object_unref (seat->priv->session_to_activate);
454         seat->priv->session_to_activate = NULL;
455     }
456 }
457
458 static void
459 session_authentication_complete_cb (Session *session, Seat *seat)
460 {
461     if (session_get_is_authenticated (session))
462     {
463         g_debug ("Session authenticated, running command");
464         run_session (seat, session);
465     }
466     else if (!IS_GREETER (session))
467     {
468         Greeter *greeter_session;
469
470         g_debug ("Switching to greeter due to failed authentication");
471
472         greeter_session = create_greeter_session (seat);
473         if (seat->priv->session_to_activate)
474             g_object_unref (seat->priv->session_to_activate);
475         seat->priv->session_to_activate = g_object_ref (greeter_session);
476         if (session_get_is_guest (session))
477             greeter_set_hint (greeter_session, "select-guest", "true");
478         else
479             greeter_set_hint (greeter_session, "select-user", session_get_username (session));
480
481         if (seat->priv->share_display_server)
482             session_set_display_server (SESSION (greeter_session), session_get_display_server (session));
483         else
484         {
485             DisplayServer *display_server;
486
487             display_server = create_display_server (seat);
488             if (!display_server_start (display_server))
489             {
490                 g_debug ("Failed to start display server for greeter");
491                 seat_stop (seat);
492             }
493
494             session_set_display_server (SESSION (greeter_session), display_server);         
495         }
496
497         start_session (seat, SESSION (greeter_session));
498
499         /* Stop failed session */
500         session_stop (session);
501     }
502     else
503     {
504         g_debug ("Stopping session that failed authentication");
505         session_stop (session);
506     }
507 }
508
509 static void
510 session_stopped_cb (Session *session, Seat *seat)
511 {
512     DisplayServer *display_server;
513
514     g_debug ("Session stopped");
515
516     display_server = session_get_display_server (session);
517
518     /* Cleanup */
519     if (!IS_GREETER (session))
520     {
521         const gchar *script;
522         script = seat_get_string_property (seat, "session-cleanup-script");
523         if (script)
524             run_script (seat, display_server, script, session_get_user (session));
525     }
526
527     g_signal_handlers_disconnect_matched (session, G_SIGNAL_MATCH_DATA, 0, 0, NULL, NULL, seat);
528     seat->priv->sessions = g_list_remove (seat->priv->sessions, session);
529
530     /* We were waiting for this session, but it didn't start :( */
531     // FIXME: Start a greeter on this?
532     if (session == seat->priv->session_to_activate)
533     {
534         g_object_unref (seat->priv->session_to_activate);
535         seat->priv->session_to_activate = NULL;
536     }
537
538     if (seat->priv->stopping)
539     {
540         check_stopped (seat);
541         g_object_unref (session);
542         return;
543     }
544
545     /* If this is the greeter session then re-use this display server */
546     if (IS_GREETER (session) && seat->priv->share_display_server &&
547         greeter_get_start_session (GREETER (session)))
548     {
549         GList *link;
550
551         for (link = seat->priv->sessions; link; link = link->next)
552         {
553             Session *s = link->data;
554
555             /* Skip this session and sessions on other display servers */
556             if (s == session || session_get_display_server (s) != display_server || session_get_is_stopping (s))
557                 continue;
558
559             if (session_get_is_authenticated (s))
560             {
561                 g_debug ("Greeter stopped, running session");
562                 run_session (seat, s);
563             }
564             else
565             {
566                 g_debug ("Greeter stopped, starting session authentication");
567                 start_session (seat, s);
568             }
569             break;
570         }
571     }
572     /* If this is the greeter and nothing else is running then stop the seat */
573     else if (IS_GREETER (session) &&
574         !greeter_get_start_session (GREETER (session)) &&
575         g_list_length (seat->priv->display_servers) == 1 &&
576         g_list_nth_data (seat->priv->display_servers, 0) == display_server)
577     {
578         g_debug ("Stopping seat, failed to start a greeter");
579         seat_stop (seat);
580     }
581     /* If we were the active session, switch to a greeter */
582     else if (!IS_GREETER (session) && session == seat_get_active_session (seat))
583     {
584         g_debug ("Active session stopped, starting greeter");
585         seat_switch_to_greeter (seat);
586     }
587
588     /* Stop the display server if no-longer required */
589     if (display_server && !display_server_get_is_stopping (display_server))
590     {
591         GList *link;
592         int n_sessions = 0;
593
594         for (link = seat->priv->sessions; link; link = link->next)
595         {
596             Session *s = link->data;
597             if (s == session)
598                 continue;
599             if (session_get_display_server (s) == display_server)
600                 n_sessions++;
601         }
602         if (n_sessions == 0)
603         {
604             g_debug ("Stopping display server, no sessions require it");
605             display_server_stop (display_server);
606         }
607     }
608
609     g_object_unref (session);
610 }
611
612 static void
613 set_session_env (Session *session)
614 {
615     /* Variables required for regression tests */
616     if (g_getenv ("LIGHTDM_TEST_ROOT"))
617     {
618         session_set_env (session, "LIGHTDM_TEST_ROOT", g_getenv ("LIGHTDM_TEST_ROOT"));
619         session_set_env (session, "DBUS_SYSTEM_BUS_ADDRESS", g_getenv ("DBUS_SYSTEM_BUS_ADDRESS"));
620         session_set_env (session, "DBUS_SESSION_BUS_ADDRESS", g_getenv ("DBUS_SESSION_BUS_ADDRESS"));
621         session_set_env (session, "LD_PRELOAD", g_getenv ("LD_PRELOAD"));
622         session_set_env (session, "LD_LIBRARY_PATH", g_getenv ("LD_LIBRARY_PATH"));
623         session_set_env (session, "GI_TYPELIB_PATH", g_getenv ("GI_TYPELIB_PATH"));
624     }
625 }
626
627 static Session *
628 create_session (Seat *seat, gboolean autostart)
629 {
630     Session *session;
631
632     session = SEAT_GET_CLASS (seat)->create_session (seat);
633     seat->priv->sessions = g_list_append (seat->priv->sessions, session);
634     if (autostart)
635         g_signal_connect (session, "authentication-complete", G_CALLBACK (session_authentication_complete_cb), seat);
636     g_signal_connect (session, "stopped", G_CALLBACK (session_stopped_cb), seat);
637
638     set_session_env (session);
639
640     return session;
641 }
642
643 static gchar **
644 get_session_argv_from_filename (const gchar *filename, const gchar *session_wrapper)
645 {
646     GKeyFile *session_desktop_file;
647     gboolean result;
648     int argc;
649     gchar *command = NULL, **argv, *path;
650     GError *error = NULL;
651
652     /* Read the command from the .desktop file */
653     session_desktop_file = g_key_file_new ();
654     result = g_key_file_load_from_file (session_desktop_file, filename, G_KEY_FILE_NONE, &error);
655     if (error)
656         g_debug ("Failed to load session file %s: %s", filename, error->message);
657     g_clear_error (&error);
658     if (result)
659     {
660         command = g_key_file_get_string (session_desktop_file, G_KEY_FILE_DESKTOP_GROUP, G_KEY_FILE_DESKTOP_KEY_EXEC, NULL);
661         if (!command)
662             g_debug ("No command in session file %s", filename);
663     }
664     g_key_file_free (session_desktop_file);
665
666     if (!command)
667         return NULL;
668
669     /* If configured, run sessions through a wrapper */
670     if (session_wrapper)
671     {
672         argv = g_malloc (sizeof (gchar *) * 3);
673         path = g_find_program_in_path (session_wrapper);
674         argv[0] = path ? path : g_strdup (session_wrapper);
675         argv[1] = command;
676         argv[2] = NULL;
677         return argv;
678     }
679
680     /* Split command into an array listing and make command absolute */
681     result = g_shell_parse_argv (command, &argc, &argv, &error);
682     if (error)
683         g_debug ("Invalid session command '%s': %s", command, error->message);
684     g_clear_error (&error);
685     g_free (command);
686     if (!result)
687         return NULL;
688     path = g_find_program_in_path (argv[0]);
689     if (path)
690     {
691         g_free (argv[0]);
692         argv[0] = path;
693     }
694   
695     return argv;
696 }
697
698 static gchar **
699 get_session_argv (const gchar *sessions_dir, const gchar *session_name, const gchar *session_wrapper)
700 {
701     gchar **dirs, **argv;
702     int i;
703
704     g_return_val_if_fail (sessions_dir != NULL, NULL);
705     g_return_val_if_fail (session_name != NULL, NULL);
706
707     dirs = g_strsplit (sessions_dir, ":", -1);
708     for (i = 0; dirs[i]; i++)
709     {
710         gchar *filename, *path;
711
712         filename = g_strdup_printf ("%s.desktop", session_name);
713         path = g_build_filename (dirs[i], filename, NULL);
714         g_free (filename);
715         argv = get_session_argv_from_filename (path, session_wrapper);
716         g_free (path);
717         if (argv)
718             break;
719     }
720     g_strfreev (dirs);
721
722     return argv;
723 }
724
725 static Session *
726 create_user_session (Seat *seat, const gchar *username)
727 {
728     User *user;
729     gchar *sessions_dir, **argv;
730     const gchar *session_name;
731     Session *session;
732   
733     user = accounts_get_user_by_name (username);
734     if (!user)
735     {
736         g_debug ("Can't login unknown user '%s'", username);
737         return NULL;
738     }
739
740     session_name = user_get_xsession (user);
741     g_object_unref (user);
742     if (!session_name)
743         session_name = seat_get_string_property (seat, "user-session");
744     sessions_dir = config_get_string (config_get_instance (), "LightDM", "sessions-directory");
745     argv = get_session_argv (sessions_dir, session_name, seat_get_string_property (seat, "session-wrapper"));
746     g_free (sessions_dir);
747     if (!argv)
748     {
749         g_debug ("Can't find session '%s'", seat_get_string_property (seat, "user-session"));
750         return NULL;
751     }
752
753     session = create_session (seat, TRUE);
754     session_set_pam_service (session, AUTOLOGIN_SERVICE);
755     session_set_username (session, username);
756     session_set_do_authenticate (session, TRUE);
757     session_set_argv (session, argv);
758
759     return session;
760 }
761
762 static Session *
763 create_guest_session (Seat *seat)
764 {
765     gchar *sessions_dir, **argv;
766     Session *session;
767
768     sessions_dir = config_get_string (config_get_instance (), "LightDM", "sessions-directory");
769     argv = get_session_argv (sessions_dir,
770                              seat_get_string_property (seat, "user-session"),
771                              seat_get_string_property (seat, "session-wrapper"));
772     g_free (sessions_dir);
773     if (!argv)
774     {
775         g_debug ("Can't find session '%s'", seat_get_string_property (seat, "user-session"));
776         return NULL;
777     }
778
779     session = create_session (seat, TRUE);
780     session_set_do_authenticate (session, TRUE);
781     session_set_is_guest (session, TRUE);
782     session_set_argv (session, argv);
783   
784     return session;
785 }
786
787 static Session *
788 greeter_create_session_cb (Greeter *greeter, Seat *seat)
789 {
790     Session *session;
791
792     session = create_session (seat, FALSE);
793     session_set_display_server (session, session_get_display_server (SESSION (greeter)));
794
795     return g_object_ref (session);
796 }
797
798 static void
799 prepend_argv (gchar ***argv, const gchar *value)
800 {
801     gchar **old_argv, **new_argv;
802     gint i;
803
804     old_argv = *argv;
805     new_argv = g_malloc (sizeof (gchar *) * (g_strv_length (*argv) + 2));
806     new_argv[0] = g_strdup (value);
807     for (i = 0; old_argv[i]; i++)
808         new_argv[i + 1] = old_argv[i];
809     new_argv[i + 1] = NULL;
810
811     g_free (*argv);
812     *argv = new_argv;
813 }
814
815 static Session *
816 find_user_session (Seat *seat, const gchar *username)
817 {
818     GList *link;
819
820     if (!username)
821         return NULL;
822
823     for (link = seat->priv->sessions; link; link = link->next)
824     {
825         Session *session = link->data;
826
827         if (!session_get_is_stopping (session) && strcmp (session_get_username (session), username) == 0)
828             return session;
829     }
830
831     return NULL;
832 }
833
834 static gboolean
835 greeter_start_session_cb (Greeter *greeter, SessionType type, const gchar *session_name, Seat *seat)
836 {
837     Session *session, *existing_session;
838     const gchar *username;
839     User *user;
840     gchar *sessions_dir = NULL;
841     gchar **argv;
842
843     /* Get the session to use */
844     if (greeter_get_guest_authenticated (greeter))
845     {
846         session = create_guest_session (seat);
847         session_set_pam_service (session, AUTOLOGIN_SERVICE);
848     }
849     else
850         session = greeter_get_authentication_session (greeter);
851
852     /* Switch to this session when it is ready */
853     if (seat->priv->session_to_activate)
854         g_object_unref (seat->priv->session_to_activate);
855     seat->priv->session_to_activate = g_object_ref (session);
856
857     /* Return to existing session if it is open */
858     username = session_get_username (session);
859     existing_session = find_user_session (seat, username);
860     if (existing_session && session != existing_session)
861     {
862         g_debug ("Returning to existing user session %s", username);
863         session_stop (session);
864         seat_set_active_session (seat, existing_session);
865         return TRUE;
866     }
867
868     /* Get session command to run */
869     switch (type)
870     {
871     case SESSION_TYPE_LOCAL:
872         sessions_dir = config_get_string (config_get_instance (), "LightDM", "sessions-directory");
873         break;
874     case SESSION_TYPE_REMOTE:
875         sessions_dir = config_get_string (config_get_instance (), "LightDM", "remote-sessions-directory");
876         break;
877     }
878     user = session_get_user (session);
879     if (!session_name && user)
880         session_name = user_get_xsession (user);
881     if (!session_name)
882         session_name = seat_get_string_property (seat, "user-session");
883     if (user)
884         user_set_xsession (session_get_user (session), session_name);
885     argv = get_session_argv (sessions_dir, session_name, seat_get_string_property (seat, "session-wrapper"));
886     g_free (sessions_dir);
887   
888     if (!argv)
889     {
890         g_debug ("Can't find session '%s'", seat_get_string_property (seat, "user-session"));
891         return FALSE;
892     }
893
894     session_set_argv (session, argv);
895     g_strfreev (argv);
896
897     /* If no session information found, then can't start the session */
898     if (!argv)
899     {
900         g_debug ("Can't find greeter session '%s'", session_name);
901         return FALSE;
902     }
903
904     /* If can re-use the display server, stop the greeter first */
905     if (seat->priv->share_display_server)
906     {
907         /* Run on the same display server after the greeter has stopped */
908         session_set_display_server (session, session_get_display_server (SESSION (greeter)));
909
910         g_debug ("Stopping greeter");
911         session_stop (SESSION (greeter));
912
913         return TRUE;
914     }
915     /* Otherwise start a new display server for this session */
916     else
917     {
918         DisplayServer *display_server;
919
920         display_server = create_display_server (seat);
921         if (!display_server_start (display_server))
922             return FALSE;
923
924         session_set_display_server (session, display_server);
925
926         return TRUE;
927     }
928 }
929
930 static Greeter *
931 create_greeter_session (Seat *seat)
932 {
933     gchar *sessions_dir, **argv;
934     Greeter *greeter_session;
935     gchar *greeter_user;
936     const gchar *greeter_wrapper;
937
938     sessions_dir = config_get_string (config_get_instance (), "LightDM", "greeters-directory");
939     argv = get_session_argv (sessions_dir,
940                              seat_get_string_property (seat, "greeter-session"),
941                              seat_get_string_property (seat, "session-wrapper"));
942     g_free (sessions_dir);
943     if (!argv)
944         return NULL;
945
946     greeter_wrapper = seat_get_string_property (seat, "greeter-wrapper");
947     if (greeter_wrapper)
948     {
949         gchar *path;
950         path = g_find_program_in_path (greeter_wrapper);
951         prepend_argv (&argv, path ? path : greeter_wrapper);
952         g_free (path);
953     }
954
955     greeter_session = SEAT_GET_CLASS (seat)->create_greeter_session (seat);
956     seat->priv->sessions = g_list_append (seat->priv->sessions, SESSION (greeter_session));
957     g_signal_connect (greeter_session, "authentication-complete", G_CALLBACK (session_authentication_complete_cb), seat);
958     g_signal_connect (greeter_session, "stopped", G_CALLBACK (session_stopped_cb), seat);
959   
960     set_session_env (SESSION (greeter_session));
961
962     session_set_pam_service (SESSION (greeter_session), GREETER_SERVICE);
963     greeter_user = config_get_string (config_get_instance (), "LightDM", "greeter-user");
964     session_set_username (SESSION (greeter_session), greeter_user);
965     g_free (greeter_user);
966     session_set_argv (SESSION (greeter_session), argv);
967
968     greeter_set_pam_services (greeter_session, USER_SERVICE, AUTOLOGIN_SERVICE);
969     greeter_set_allow_guest (greeter_session, seat_get_allow_guest (seat));
970     g_signal_connect (greeter_session, "create-session", G_CALLBACK (greeter_create_session_cb), seat);
971     g_signal_connect (greeter_session, "start-session", G_CALLBACK (greeter_start_session_cb), seat);
972
973     return greeter_session;
974 }
975
976 static Session *
977 find_session_for_display_server (Seat *seat, DisplayServer *display_server)
978 {
979     GList *link;
980
981     for (link = seat->priv->sessions; link; link = link->next)
982     {
983         Session *session = link->data;
984         if (session_get_display_server (session) == display_server && !session_get_is_stopping (session))
985             return session;
986     }
987
988     return NULL;
989 }
990
991 static void
992 display_server_ready_cb (DisplayServer *display_server, Seat *seat)
993 {
994     const gchar *script;
995     Session *session;
996
997     /* Run setup script */
998     script = seat_get_string_property (seat, "display-setup-script");
999     if (script && !run_script (seat, display_server, script, NULL))
1000     {
1001         g_debug ("Stopping display server due to failed setup script");
1002         display_server_stop (display_server);
1003         return;
1004     }
1005
1006     emit_upstart_signal ("login-session-start");
1007
1008     /* Start the session waiting for this display server */
1009     session = find_session_for_display_server (seat, display_server);
1010     if (session)
1011     {
1012         if (session_get_is_authenticated (session))
1013         {
1014             g_debug ("Display server ready, running session");
1015             run_session (seat, session);
1016         }
1017         else
1018         {
1019             g_debug ("Display server ready, starting session authentication");
1020             start_session (seat, session);
1021         }
1022     }
1023     else
1024     {
1025         g_debug ("Stopping not required display server");
1026         display_server_stop (display_server);
1027     }
1028 }
1029
1030 static DisplayServer *
1031 create_display_server (Seat *seat)
1032 {
1033     DisplayServer *display_server;
1034
1035     display_server = SEAT_GET_CLASS (seat)->create_display_server (seat);
1036     seat->priv->display_servers = g_list_append (seat->priv->display_servers, display_server);
1037     g_signal_connect (display_server, "ready", G_CALLBACK (display_server_ready_cb), seat);
1038     g_signal_connect (display_server, "stopped", G_CALLBACK (display_server_stopped_cb), seat);
1039
1040     return display_server;
1041 }
1042
1043 static Greeter *
1044 find_greeter_session (Seat *seat)
1045 {
1046     GList *link;
1047
1048     for (link = seat->priv->sessions; link; link = link->next)
1049     {
1050         Session *session = link->data;
1051         if (!session_get_is_stopping (session) && IS_GREETER (session))
1052             return GREETER (session);
1053     }
1054
1055     return NULL;
1056 }
1057
1058 gboolean
1059 seat_switch_to_greeter (Seat *seat)
1060 {
1061     Greeter *greeter_session;
1062     DisplayServer *display_server;
1063
1064     g_return_val_if_fail (seat != NULL, FALSE);
1065
1066     if (!seat->priv->can_switch)
1067         return FALSE;
1068
1069     /* Switch to greeter if one open (shouldn't be though) */
1070     greeter_session = find_greeter_session (seat);
1071     if (greeter_session)
1072     {
1073         g_debug ("Switching to existing greeter");
1074         seat_set_active_session (seat, SESSION (greeter_session));
1075         return TRUE;
1076     }
1077
1078     greeter_session = create_greeter_session (seat);
1079     if (seat->priv->session_to_activate)
1080         g_object_unref (seat->priv->session_to_activate);
1081     seat->priv->session_to_activate = g_object_ref (greeter_session);
1082
1083     display_server = create_display_server (seat);
1084     session_set_display_server (SESSION (greeter_session), display_server);
1085     if (!display_server_start (display_server))
1086         return FALSE;
1087
1088     return TRUE;
1089 }
1090
1091 gboolean
1092 seat_switch_to_user (Seat *seat, const gchar *username, const gchar *session_name)
1093 {
1094     Session *session;
1095     DisplayServer *display_server;
1096
1097     g_return_val_if_fail (seat != NULL, FALSE);
1098     g_return_val_if_fail (username != NULL, FALSE);
1099
1100     if (!seat->priv->can_switch)
1101         return FALSE;
1102
1103     g_debug ("Switching to user %s", username);
1104
1105     session = find_user_session (seat, username);
1106     if (session)
1107     {
1108         g_debug ("Switching to existing user session %s", username);
1109         seat_set_active_session (seat, session);
1110         return TRUE;
1111     }
1112
1113     session = create_user_session (seat, username);
1114     if (seat->priv->session_to_activate)
1115         g_object_unref (seat->priv->session_to_activate);
1116     seat->priv->session_to_activate = g_object_ref (session);
1117     session_set_pam_service (session, USER_SERVICE);
1118
1119     display_server = create_display_server (seat);
1120     session_set_display_server (session, display_server);
1121     if (!display_server_start (display_server))
1122         return FALSE;
1123
1124     return FALSE;
1125 }
1126
1127 static Session *
1128 find_guest_session (Seat *seat)
1129 {
1130     GList *link;
1131
1132     for (link = seat->priv->sessions; link; link = link->next)
1133     {
1134         Session *session = link->data;
1135         if (!session_get_is_stopping (session) && session_get_is_guest (session))
1136             return session;
1137     }
1138
1139     return NULL;
1140 }
1141
1142 gboolean
1143 seat_switch_to_guest (Seat *seat, const gchar *session_name)
1144 {
1145     Session *session;
1146     DisplayServer *display_server;
1147     GList *link;
1148
1149     g_return_val_if_fail (seat != NULL, FALSE);
1150
1151     if (!seat->priv->can_switch || !seat_get_allow_guest (seat))
1152         return FALSE;
1153
1154     /* Switch to session if one open */
1155     session = find_guest_session (seat);
1156     if (session)
1157     {
1158         g_debug ("Switching to existing guest account %s", session_get_username (session));
1159         seat_set_active_session (seat, session);
1160         return TRUE;
1161     }
1162
1163     display_server = create_display_server (seat);
1164     if (!display_server_start (display_server))
1165         return FALSE;
1166
1167     session = create_guest_session (seat);
1168     if (seat->priv->session_to_activate)
1169         g_object_unref (seat->priv->session_to_activate);
1170     seat->priv->session_to_activate = g_object_ref (session);
1171     session_set_pam_service (session, AUTOLOGIN_SERVICE);
1172     session_set_display_server (session, display_server);
1173
1174     return TRUE;
1175 }
1176
1177 gboolean
1178 seat_lock (Seat *seat, const gchar *username)
1179 {
1180     Greeter *greeter_session;
1181     DisplayServer *display_server;
1182
1183     g_return_val_if_fail (seat != NULL, FALSE);
1184
1185     if (!seat->priv->can_switch)
1186         return FALSE;
1187
1188     g_debug ("Locking seat");
1189
1190     /* Switch to greeter if one open (shouldn't be though) */
1191     greeter_session = find_greeter_session (seat);
1192     if (greeter_session)
1193     {
1194         g_debug ("Switching to existing greeter");
1195         seat_set_active_session (seat, SESSION (greeter_session));
1196         return TRUE;
1197     }
1198
1199     display_server = create_display_server (seat);
1200     if (!display_server_start (display_server))
1201         return FALSE;
1202
1203     greeter_session = create_greeter_session (seat);
1204     if (seat->priv->session_to_activate)
1205         g_object_unref (seat->priv->session_to_activate);
1206     seat->priv->session_to_activate = g_object_ref (greeter_session);
1207     greeter_set_hint (greeter_session, "lock-screen", "true");
1208     if (username)
1209         greeter_set_hint (greeter_session, "select-user", username);
1210     session_set_display_server (SESSION (greeter_session), display_server);
1211
1212     return TRUE;
1213 }
1214
1215 void
1216 seat_stop (Seat *seat)
1217 {
1218     g_return_if_fail (seat != NULL);
1219
1220     if (seat->priv->stopping)
1221         return;
1222
1223     g_debug ("Stopping seat");
1224     seat->priv->stopping = TRUE;
1225     SEAT_GET_CLASS (seat)->stop (seat);
1226 }
1227
1228 gboolean
1229 seat_get_is_stopping (Seat *seat)
1230 {
1231     g_return_val_if_fail (seat != NULL, FALSE);
1232     return seat->priv->stopping;
1233 }
1234
1235 static void
1236 seat_real_setup (Seat *seat)
1237 {
1238 }
1239
1240 static gboolean
1241 seat_real_start (Seat *seat)
1242 {
1243     const gchar *autologin_username;
1244     int autologin_timeout;
1245     gboolean autologin_guest;
1246     gboolean autologin_in_background;
1247     const gchar *user_session;
1248     Session *session = NULL;
1249     DisplayServer *display_server;
1250     User *user;
1251     gchar *sessions_dir;
1252     const gchar *wrapper = NULL;
1253     const gchar *session_name = NULL;
1254
1255     g_debug ("Starting seat");
1256
1257     display_server = create_display_server (seat);
1258
1259     /* Get autologin settings */
1260     autologin_username = seat_get_string_property (seat, "autologin-user");
1261     if (g_strcmp0 (autologin_username, "") == 0)
1262         autologin_username = NULL;
1263     autologin_timeout = seat_get_integer_property (seat, "autologin-user-timeout");
1264     autologin_guest = seat_get_boolean_property (seat, "autologin-guest");
1265     autologin_in_background = seat_get_boolean_property (seat, "autologin-in-background");
1266
1267     /* Autologin if configured */
1268     if (autologin_timeout == 0 || autologin_in_background)
1269     {
1270         if (autologin_guest)
1271         {
1272             session = create_guest_session (seat);
1273             if (seat->priv->session_to_activate)
1274                 g_object_unref (seat->priv->session_to_activate);
1275             seat->priv->session_to_activate = g_object_ref (session);
1276             session_set_pam_service (session, AUTOLOGIN_SERVICE);
1277         }
1278         else if (autologin_username != NULL)
1279         {
1280             session = create_user_session (seat, autologin_username);
1281             if (seat->priv->session_to_activate)
1282                 g_object_unref (seat->priv->session_to_activate);
1283             seat->priv->session_to_activate = g_object_ref (session);
1284             session_set_pam_service (session, AUTOLOGIN_SERVICE);
1285         }
1286
1287         /* Load in background if required */
1288         if (autologin_in_background && session)
1289         {
1290             DisplayServer *background_display_server;
1291
1292             background_display_server = create_display_server (seat);
1293             session_set_display_server (session, background_display_server);
1294             if (!display_server_start (background_display_server))
1295                 return FALSE;
1296
1297             /* Start a greeter as well */
1298             session = NULL;
1299         }
1300     }
1301
1302     /* Fallback to a greeter */
1303     if (!session)
1304     {
1305         Greeter *greeter_session;
1306
1307         greeter_session = create_greeter_session (seat);
1308         if (seat->priv->session_to_activate)
1309             g_object_unref (seat->priv->session_to_activate);
1310         seat->priv->session_to_activate = g_object_ref (greeter_session);
1311         session = SESSION (greeter_session);
1312
1313         if (autologin_timeout)
1314         {
1315             gchar *value;
1316
1317             value = g_strdup_printf ("%d", autologin_timeout);
1318             greeter_set_hint (greeter_session, "autologin-timeout", value);
1319             g_free (value);
1320             if (autologin_username)
1321                 greeter_set_hint (greeter_session, "autologin-user", autologin_username);
1322             if (autologin_guest)
1323                 greeter_set_hint (greeter_session, "autologin-guest", "true");
1324         }
1325     }
1326
1327     /* Fail if can't start a session */
1328     if (!session)
1329     {
1330         seat_stop (seat);
1331         return FALSE;
1332     }
1333
1334     /* Start display server to show session on */
1335     session_set_display_server (session, display_server);
1336     if (!display_server_start (display_server))
1337         return FALSE;
1338
1339     return TRUE;
1340 }
1341
1342 static void
1343 seat_real_set_active_session (Seat *seat, Session *session)
1344 {
1345 }
1346
1347 static Session *
1348 seat_real_get_active_session (Seat *seat)
1349 {
1350     return NULL;
1351 }
1352
1353 static void
1354 seat_real_stop (Seat *seat)
1355 {
1356     GList *list, *link;
1357
1358     check_stopped (seat);
1359     if (seat->priv->stopped)
1360         return;
1361
1362     /* Stop all the display servers and sessions on the seat. Copy the list as
1363      * it might be modified if a display server / session stops during this loop */
1364     list = g_list_copy (seat->priv->display_servers);
1365     for (link = list; link; link = link->next)
1366     {
1367         DisplayServer *display_server = link->data;
1368         if (!display_server_get_is_stopping (display_server))
1369         {
1370             g_debug ("Stopping display server");
1371             display_server_stop (display_server);
1372         }
1373     }
1374     g_list_free (list);
1375     list = g_list_copy (seat->priv->sessions);
1376     for (link = list; link; link = link->next)
1377     {
1378         Session *session = link->data;
1379         if (!session_get_is_stopping (session))
1380         {
1381             g_debug ("Stopping session");
1382             session_stop (session);
1383         }
1384     }
1385     g_list_free (list);
1386 }
1387
1388 static void
1389 seat_init (Seat *seat)
1390 {
1391     seat->priv = G_TYPE_INSTANCE_GET_PRIVATE (seat, SEAT_TYPE, SeatPrivate);
1392     seat->priv->properties = g_hash_table_new_full (g_str_hash, g_str_equal, g_free, g_free);
1393     seat->priv->share_display_server = TRUE;
1394 }
1395
1396 static void
1397 seat_finalize (GObject *object)
1398 {
1399     Seat *self;
1400     GList *link;
1401
1402     self = SEAT (object);
1403
1404     g_hash_table_unref (self->priv->properties);
1405     for (link = self->priv->display_servers; link; link = link->next)
1406     {
1407         DisplayServer *display_server = link->data;
1408         g_signal_handlers_disconnect_matched (display_server, G_SIGNAL_MATCH_DATA, 0, 0, NULL, NULL, self);
1409     }  
1410     g_list_free_full (self->priv->display_servers, g_object_unref);
1411     for (link = self->priv->sessions; link; link = link->next)
1412     {
1413         Session *session = link->data;
1414         g_signal_handlers_disconnect_matched (session, G_SIGNAL_MATCH_DATA, 0, 0, NULL, NULL, self);
1415     }
1416     g_list_free_full (self->priv->sessions, g_object_unref);
1417     if (self->priv->session_to_activate)
1418         g_object_unref (self->priv->session_to_activate);
1419
1420     G_OBJECT_CLASS (seat_parent_class)->finalize (object);
1421 }
1422
1423 static void
1424 seat_class_init (SeatClass *klass)
1425 {
1426     GObjectClass *object_class = G_OBJECT_CLASS (klass);
1427
1428     klass->setup = seat_real_setup;
1429     klass->start = seat_real_start;
1430     klass->set_active_session = seat_real_set_active_session;
1431     klass->get_active_session = seat_real_get_active_session;
1432     klass->run_script = seat_real_run_script;
1433     klass->stop = seat_real_stop;
1434
1435     object_class->finalize = seat_finalize;
1436
1437     g_type_class_add_private (klass, sizeof (SeatPrivate));
1438
1439     signals[SESSION_ADDED] =
1440         g_signal_new ("session-added",
1441                       G_TYPE_FROM_CLASS (klass),
1442                       G_SIGNAL_RUN_LAST,
1443                       G_STRUCT_OFFSET (SeatClass, session_added),
1444                       NULL, NULL,
1445                       g_cclosure_marshal_VOID__OBJECT,
1446                       G_TYPE_NONE, 1, SESSION_TYPE);
1447     signals[SESSION_REMOVED] =
1448         g_signal_new ("session-removed",
1449                       G_TYPE_FROM_CLASS (klass),
1450                       G_SIGNAL_RUN_LAST,
1451                       G_STRUCT_OFFSET (SeatClass, session_removed),
1452                       NULL, NULL,
1453                       g_cclosure_marshal_VOID__OBJECT,
1454                       G_TYPE_NONE, 1, SESSION_TYPE);
1455     signals[STOPPED] =
1456         g_signal_new ("stopped",
1457                       G_TYPE_FROM_CLASS (klass),
1458                       G_SIGNAL_RUN_LAST,
1459                       G_STRUCT_OFFSET (SeatClass, stopped),
1460                       NULL, NULL,
1461                       g_cclosure_marshal_VOID__VOID,
1462                       G_TYPE_NONE, 0);
1463 }