]> rtime.felk.cvut.cz Git - sojka/lightdm.git/blob - src/session-child.c
Be extra careful not to call any non thread safe function after a fork
[sojka/lightdm.git] / src / session-child.c
1 #include <config.h>
2
3 #include <stdlib.h>
4 #include <stdio.h>
5 #include <unistd.h>
6 #include <string.h>
7 #include <errno.h>
8 #include <sys/types.h>
9 #include <sys/stat.h>
10 #include <sys/wait.h>
11 #include <fcntl.h>
12 #include <pwd.h>
13 #include <grp.h>
14 #include <glib.h>
15 #include <security/pam_appl.h>
16 #include <utmpx.h>
17 #include <sys/mman.h>
18
19 #include "configuration.h"
20 #include "session-child.h"
21 #include "session.h"
22 #include "console-kit.h"
23 #include "login1.h"
24 #include "privileges.h"
25 #include "x-authority.h"
26 #include "configuration.h"
27
28 /* Child process being run */
29 static GPid child_pid = 0;
30
31 /* Pipe to communicate with daemon */
32 static int from_daemon_output = 0;
33 static int to_daemon_input = 0;
34
35 static gboolean is_interactive;
36 static gboolean do_authenticate;
37 static gboolean authentication_complete = FALSE;
38 static pam_handle_t *pam_handle;
39
40 /* Maximum length of a string to pass between daemon and session */
41 #define MAX_STRING_LENGTH 65535
42
43 static void
44 write_data (const void *buf, size_t count)
45 {
46     if (write (to_daemon_input, buf, count) != count)
47         g_printerr ("Error writing to daemon: %s\n", strerror (errno));
48 }
49
50 static void
51 write_string (const char *value)
52 {
53     int length;
54
55     length = value ? strlen (value) : -1;
56     write_data (&length, sizeof (length));
57     if (value)
58         write_data (value, sizeof (char) * length);
59 }
60
61 static ssize_t
62 read_data (void *buf, size_t count)
63 {
64     ssize_t n_read;
65
66     n_read = read (from_daemon_output, buf, count);
67     if (n_read < 0)
68         g_printerr ("Error reading from daemon: %s\n", strerror (errno));
69   
70     return n_read;
71 }
72
73 static gchar *
74 read_string_full (void* (*alloc_fn)(size_t n))
75 {
76     int length;
77     char *value;
78
79     if (read_data (&length, sizeof (length)) <= 0)
80         return NULL;
81     if (length < 0)
82         return NULL;
83     if (length > MAX_STRING_LENGTH)
84     {
85         g_printerr ("Invalid string length %d from daemon\n", length);
86         return NULL;
87     }
88   
89     value = (*alloc_fn) (sizeof (char) * (length + 1));
90     read_data (value, length);
91     value[length] = '\0';      
92
93     return value;
94 }
95
96 static gchar *
97 read_string (void)
98 {
99     return read_string_full (g_malloc);
100 }
101
102 static int
103 pam_conv_cb (int msg_length, const struct pam_message **msg, struct pam_response **resp, void *app_data)
104 {
105     int i, error;
106     gboolean auth_complete = FALSE;
107     struct pam_response *response;
108     gchar *username = NULL;
109
110     /* FIXME: We don't support communication after pam_authenticate completes */
111     if (authentication_complete)
112         return PAM_SUCCESS;
113
114     /* Cancel authentication if requiring input */
115     if (!is_interactive)
116     {
117         for (i = 0; i < msg_length; i++)
118         {
119             if (msg[i]->msg_style == PAM_PROMPT_ECHO_ON || msg[i]->msg_style == PAM_PROMPT_ECHO_OFF)
120             {
121                 g_printerr ("Stopping PAM conversation, interaction requested but not supported\n");
122                 return PAM_CONV_ERR;
123             }
124         }
125
126         /* Ignore informational messages */
127         return PAM_SUCCESS;
128     }
129
130     /* Check if we changed user */
131     pam_get_item (pam_handle, PAM_USER, (const void **) &username);
132
133     /* Notify the daemon */
134     write_string (username);
135     write_data (&auth_complete, sizeof (auth_complete));
136     write_data (&msg_length, sizeof (msg_length));
137     for (i = 0; i < msg_length; i++)
138     {
139         const struct pam_message *m = msg[i];
140         write_data (&m->msg_style, sizeof (m->msg_style));
141         write_string (m->msg);
142     }
143
144     /* Get response */
145     read_data (&error, sizeof (error));
146     if (error != PAM_SUCCESS)
147         return error;
148     response = calloc (msg_length, sizeof (struct pam_response));
149     for (i = 0; i < msg_length; i++)
150     {
151         struct pam_response *r = &response[i];
152         // callers of this function inside pam will expect to be able to call
153         // free() on the strings we give back.  So alloc with malloc.
154         r->resp = read_string_full (malloc);
155         read_data (&r->resp_retcode, sizeof (r->resp_retcode));
156     }
157
158     *resp = response;
159     return PAM_SUCCESS;
160 }
161
162 static void
163 signal_cb (int signum)
164 {
165     /* Pass on signal to child, otherwise just quit */
166     if (child_pid > 0)
167         kill (child_pid, signum);
168     else
169         exit (EXIT_SUCCESS);
170 }
171
172 static XAuthority *
173 read_xauth (void)
174 {
175     gchar *x_authority_name;
176     guint16 x_authority_family;
177     guint8 *x_authority_address;
178     gsize x_authority_address_length;
179     gchar *x_authority_number;
180     guint8 *x_authority_data;
181     gsize x_authority_data_length;
182
183     x_authority_name = read_string ();
184     if (!x_authority_name)
185         return NULL;
186
187     read_data (&x_authority_family, sizeof (x_authority_family));
188     read_data (&x_authority_address_length, sizeof (x_authority_address_length));
189     x_authority_address = g_malloc (x_authority_address_length);
190     read_data (x_authority_address, x_authority_address_length);
191     x_authority_number = read_string ();
192     read_data (&x_authority_data_length, sizeof (x_authority_data_length));
193     x_authority_data = g_malloc (x_authority_data_length);
194     read_data (x_authority_data, x_authority_data_length);
195
196     return x_authority_new (x_authority_family, x_authority_address, x_authority_address_length, x_authority_number, x_authority_name, x_authority_data, x_authority_data_length);
197 }
198
199 int
200 session_child_run (int argc, char **argv)
201 {
202     struct pam_conv conversation = { pam_conv_cb, NULL };
203     int i, version, fd, result;
204     gboolean auth_complete = TRUE;
205     User *user = NULL;
206     gchar *log_filename, *log_backup_filename = NULL;
207     gsize env_length;
208     gsize command_argc;
209     gchar **command_argv;
210     GVariantBuilder ck_parameters;
211     int return_code;
212     int authentication_result;
213     gchar *authentication_result_string;
214     gchar *service;
215     gchar *username;
216     gchar *tty;
217     gchar *remote_host_name;
218     gchar *xdisplay;
219     XAuthority *x_authority = NULL;
220     gchar *x_authority_filename;
221     GDBusConnection *bus;
222     gchar *console_kit_cookie = NULL;
223     gchar *login1_session = NULL;
224     const gchar *locale_value;
225     gchar *locale_var;
226     static const gchar * const locale_var_names[] = {
227         "LC_COLLATE",
228         "LC_CTYPE",
229         "LC_MONETARY",
230         "LC_NUMERIC",
231         "LC_TIME",
232         "LC_MESSAGES",
233         "LC_ALL",
234         "LANG",
235         NULL
236     };
237     gid_t gid;
238     uid_t uid;
239     const gchar *home_directory;
240     GError *error = NULL;
241
242 #if !defined(GLIB_VERSION_2_36)
243     g_type_init ();
244 #endif
245
246     if (config_get_boolean (config_get_instance (), "LightDM", "lock-memory"))
247     {
248         /* Protect memory from being paged to disk, as we deal with passwords */
249         mlockall (MCL_CURRENT | MCL_FUTURE);
250     }
251
252     /* Make input non-blocking */
253     fd = open ("/dev/null", O_RDONLY);
254     dup2 (fd, STDIN_FILENO);
255     close (fd);
256
257     /* Close stdout */
258     fd = open ("/dev/null", O_WRONLY);
259     dup2 (fd, STDOUT_FILENO);
260     close (fd);
261
262     /* Get the pipe from the daemon */
263     if (argc != 4)
264     {
265         g_printerr ("Usage: lightdm --session-child INPUTFD OUTPUTFD\n");
266         return EXIT_FAILURE;
267     }
268     from_daemon_output = atoi (argv[2]);
269     to_daemon_input = atoi (argv[3]);
270     if (from_daemon_output == 0 || to_daemon_input == 0)
271     {
272         g_printerr ("Invalid file descriptors %s %s\n", argv[2], argv[3]);
273         return EXIT_FAILURE;
274     }
275
276     /* Don't let these pipes leak to the command we will run */
277     fcntl (from_daemon_output, F_SETFD, FD_CLOEXEC);
278     fcntl (to_daemon_input, F_SETFD, FD_CLOEXEC);
279
280     /* Read a version number so we can handle upgrades (i.e. a newer version of session child is run for an old daemon */
281     read_data (&version, sizeof (version));
282
283     service = read_string ();
284     username = read_string ();
285     read_data (&do_authenticate, sizeof (do_authenticate));
286     read_data (&is_interactive, sizeof (is_interactive));
287     read_string (); /* Used to be class, now we just use the environment variable */
288     tty = read_string ();
289     remote_host_name = read_string ();
290     xdisplay = read_string ();
291     x_authority = read_xauth ();
292
293     /* Setup PAM */
294     result = pam_start (service, username, &conversation, &pam_handle);
295     if (result != PAM_SUCCESS)
296     {
297         g_printerr ("Failed to start PAM: %s", pam_strerror (NULL, result));
298         return EXIT_FAILURE;
299     }
300     if (xdisplay)
301     {
302 #ifdef PAM_XDISPLAY
303         pam_set_item (pam_handle, PAM_XDISPLAY, xdisplay);
304 #endif
305         pam_set_item (pam_handle, PAM_TTY, xdisplay);
306     }
307     else if (tty)
308         pam_set_item (pam_handle, PAM_TTY, tty);    
309
310 #ifdef PAM_XAUTHDATA
311     if (x_authority)
312     {
313         struct pam_xauth_data value;
314
315         value.name = (char *) x_authority_get_authorization_name (x_authority);
316         value.namelen = strlen (x_authority_get_authorization_name (x_authority));
317         value.data = (char *) x_authority_get_authorization_data (x_authority);
318         value.datalen = x_authority_get_authorization_data_length (x_authority);
319         pam_set_item (pam_handle, PAM_XAUTHDATA, &value);
320     }
321 #endif
322
323     /* Authenticate */
324     if (do_authenticate)
325     {
326         const gchar *new_username;
327
328         authentication_result = pam_authenticate (pam_handle, 0);
329
330         /* See what user we ended up as */
331         if (pam_get_item (pam_handle, PAM_USER, (const void **) &new_username) != PAM_SUCCESS)
332             return EXIT_FAILURE;
333         g_free (username);
334         username = g_strdup (new_username);
335
336         /* Check account is valid */
337         if (authentication_result == PAM_SUCCESS)
338             authentication_result = pam_acct_mgmt (pam_handle, 0);
339         if (authentication_result == PAM_NEW_AUTHTOK_REQD)
340             authentication_result = pam_chauthtok (pam_handle, PAM_CHANGE_EXPIRED_AUTHTOK);
341     }
342     else
343         authentication_result = PAM_SUCCESS;
344     authentication_complete = TRUE;
345
346     if (authentication_result == PAM_SUCCESS)
347     {
348         /* Fail authentication if user doesn't actually exist */
349         user = accounts_get_user_by_name (username);
350         if (!user)
351         {
352             g_printerr ("Failed to get information on user %s: %s\n", username, strerror (errno));
353             authentication_result = PAM_USER_UNKNOWN;
354         }
355         else
356         {
357             /* Set POSIX variables */
358             pam_putenv (pam_handle, "PATH=/usr/local/bin:/usr/bin:/bin");
359             pam_putenv (pam_handle, g_strdup_printf ("USER=%s", username));
360             pam_putenv (pam_handle, g_strdup_printf ("LOGNAME=%s", username));
361             pam_putenv (pam_handle, g_strdup_printf ("HOME=%s", user_get_home_directory (user)));
362             pam_putenv (pam_handle, g_strdup_printf ("SHELL=%s", user_get_shell (user)));
363
364             /* Let the greeter and user session inherit the system default locale */
365             for (i = 0; locale_var_names[i] != NULL; i++)
366             {
367                 if ((locale_value = g_getenv (locale_var_names[i])) != NULL)
368                 {
369                     locale_var = g_strdup_printf ("%s=%s", locale_var_names[i], locale_value);
370                     pam_putenv (pam_handle, locale_var);
371                     g_free (locale_var);
372                 }
373             }
374         }
375     }
376
377     authentication_result_string = g_strdup (pam_strerror (pam_handle, authentication_result));
378
379     /* Report authentication result */
380     write_string (username);
381     write_data (&auth_complete, sizeof (auth_complete));
382     write_data (&authentication_result, sizeof (authentication_result));
383     write_string (authentication_result_string);
384
385     /* Check we got a valid user */
386     if (!username)
387     {
388         g_printerr ("No user selected during authentication\n");
389         return EXIT_FAILURE;
390     }
391
392     /* Stop if we didn't authenticated */
393     if (authentication_result != PAM_SUCCESS)
394         return EXIT_FAILURE;
395
396     /* Get the command to run (blocks) */
397     log_filename = read_string ();
398     if (version >= 1)
399     {
400         g_free (tty);
401         tty = read_string ();      
402     }
403     x_authority_filename = read_string ();
404     if (version >= 1)
405     {
406         g_free (xdisplay);
407         xdisplay = read_string ();
408         if (x_authority)
409             g_object_unref (x_authority);
410         x_authority = read_xauth ();
411     }
412     read_data (&env_length, sizeof (env_length));
413     for (i = 0; i < env_length; i++)
414         pam_putenv (pam_handle, read_string ());
415     read_data (&command_argc, sizeof (command_argc));
416     command_argv = g_malloc (sizeof (gchar *) * (command_argc + 1));
417     for (i = 0; i < command_argc; i++)
418         command_argv[i] = read_string ();
419     command_argv[i] = NULL;
420
421     /* Redirect stderr to a log file */
422     if (log_filename)
423     {
424         log_backup_filename = g_strdup_printf ("%s.old", log_filename);
425         if (g_path_is_absolute (log_filename))
426         {
427             rename (log_filename, log_backup_filename);
428             fd = open (log_filename, O_WRONLY | O_APPEND | O_CREAT, 0600);
429             dup2 (fd, STDERR_FILENO);
430             close (fd);
431             g_free (log_filename);
432             log_filename = NULL;
433         }
434     }
435     else
436     {
437         fd = open ("/dev/null", O_WRONLY);   
438         dup2 (fd, STDERR_FILENO);
439         close (fd);
440     }
441
442     /* Set group membership - these can be overriden in pam_setcred */
443     if (getuid () == 0)
444     {
445         if (initgroups (username, user_get_gid (user)) < 0)
446         {
447             g_printerr ("Failed to initialize supplementary groups for %s: %s\n", username, strerror (errno));
448             _exit (EXIT_FAILURE);
449         }
450     }
451
452     /* Set credentials */
453     result = pam_setcred (pam_handle, PAM_ESTABLISH_CRED);
454     if (result != PAM_SUCCESS)
455     {
456         g_printerr ("Failed to establish PAM credentials: %s\n", pam_strerror (pam_handle, result));
457         return EXIT_FAILURE;
458     }
459      
460     /* Open the session */
461     result = pam_open_session (pam_handle, 0);
462     if (result != PAM_SUCCESS)
463     {
464         g_printerr ("Failed to open PAM session: %s\n", pam_strerror (pam_handle, result));
465         return EXIT_FAILURE;
466     }
467
468     /* Open a connection to the system bus for ConsoleKit - we must keep it open or CK will close the session */
469     bus = g_bus_get_sync (G_BUS_TYPE_SYSTEM, NULL, &error);
470     if (error)
471         g_printerr ("Unable to contact system bus: %s", error->message);
472     if (!bus)
473         return EXIT_FAILURE;
474
475     if (login1_is_running ())
476     {
477         login1_session = login1_get_session_id ();
478         write_string (login1_session);
479     }
480
481     if (!login1_session)
482     {
483         /* Open a Console Kit session */
484         g_variant_builder_init (&ck_parameters, G_VARIANT_TYPE ("(a(sv))"));
485         g_variant_builder_open (&ck_parameters, G_VARIANT_TYPE ("a(sv)"));
486         g_variant_builder_add (&ck_parameters, "(sv)", "unix-user", g_variant_new_int32 (user_get_uid (user)));
487         if (g_strcmp0 (pam_getenv (pam_handle, "XDG_SESSION_CLASS"), "greeter") == 0)
488             g_variant_builder_add (&ck_parameters, "(sv)", "session-type", g_variant_new_string ("LoginWindow"));
489         if (xdisplay)
490         {
491             g_variant_builder_add (&ck_parameters, "(sv)", "x11-display", g_variant_new_string (xdisplay));
492             if (tty)
493                 g_variant_builder_add (&ck_parameters, "(sv)", "x11-display-device", g_variant_new_string (tty));
494         }
495         if (remote_host_name)
496         {
497             g_variant_builder_add (&ck_parameters, "(sv)", "is-local", g_variant_new_boolean (FALSE));
498             g_variant_builder_add (&ck_parameters, "(sv)", "remote-host-name", g_variant_new_string (remote_host_name));
499         }
500         else
501             g_variant_builder_add (&ck_parameters, "(sv)", "is-local", g_variant_new_boolean (TRUE));
502         console_kit_cookie = ck_open_session (&ck_parameters);
503         write_string (console_kit_cookie);
504         if (console_kit_cookie)
505         {
506             gchar *value;
507             value = g_strdup_printf ("XDG_SESSION_COOKIE=%s", console_kit_cookie);
508             pam_putenv (pam_handle, value);
509             g_free (value);
510         }
511     }
512
513     /* Write X authority */
514     if (x_authority)
515     {
516         gboolean drop_privileges, result;
517         gchar *value;
518         GError *error = NULL;
519
520         drop_privileges = geteuid () == 0;
521         if (drop_privileges)
522             privileges_drop (user_get_uid (user), user_get_gid (user));
523         result = x_authority_write (x_authority, XAUTH_WRITE_MODE_REPLACE, x_authority_filename, &error);
524         if (drop_privileges)
525             privileges_reclaim ();
526
527         if (error)
528             g_printerr ("Error writing X authority: %s\n", error->message);
529         g_clear_error (&error);
530         if (!result)
531             return EXIT_FAILURE;
532
533         value = g_strdup_printf ("XAUTHORITY=%s", x_authority_filename);
534         pam_putenv (pam_handle, value);
535         g_free (value);
536     }
537
538     /* Catch terminate signal and pass it to the child */
539     signal (SIGTERM, signal_cb);
540
541     /* Run the command as the authenticated user */
542     uid = user_get_uid (user);
543     gid = user_get_gid (user);
544     home_directory = user_get_home_directory (user);
545     child_pid = fork ();
546     if (child_pid == 0)
547     {
548         /* Make this process its own session */
549         if (setsid () < 0)
550             _exit (errno);
551
552         /* Change to this user */
553         if (getuid () == 0)
554         {
555             if (setgid (gid) != 0)
556                 _exit (errno);
557
558             if (setuid (uid) != 0)
559                 _exit (errno);
560         }
561
562         /* Change working directory */
563         /* NOTE: This must be done after the permissions are changed because NFS filesystems can
564          * be setup so the local root user accesses the NFS files as 'nobody'.  If the home directories
565          * are not system readable then the chdir can fail */
566         if (chdir (home_directory) != 0)
567             _exit (errno);
568
569         if (log_filename)
570         {
571             rename (log_filename, log_backup_filename);
572             fd = open (log_filename, O_WRONLY | O_APPEND | O_CREAT, 0600);
573             if (fd >= 0)
574             {
575                 dup2 (fd, STDERR_FILENO);
576                 close (fd);
577             }
578         }
579
580         /* Run the command */
581         execve (command_argv[0], command_argv, pam_getenvlist (pam_handle));
582         _exit (EXIT_FAILURE);
583     }
584
585     /* Bail out if failed to fork */
586     if (child_pid < 0)
587     {
588         g_printerr ("Failed to fork session child process: %s\n", strerror (errno));
589         return_code = EXIT_FAILURE;
590     }
591
592     /* Wait for the command to complete (blocks) */
593     if (child_pid > 0)
594     {
595         /* Log to utmp */
596         if (g_strcmp0 (pam_getenv (pam_handle, "XDG_SESSION_CLASS"), "greeter") != 0)
597         {
598             struct utmpx ut;
599             struct timeval tv;
600
601             memset (&ut, 0, sizeof (ut));
602             ut.ut_type = USER_PROCESS;
603             ut.ut_pid = child_pid;
604             if (tty)
605                 strncpy (ut.ut_line, tty + strlen ("/dev/"), sizeof (ut.ut_line));
606             if (xdisplay)
607                 strncpy (ut.ut_id, xdisplay, sizeof (ut.ut_id));
608             strncpy (ut.ut_user, username, sizeof (ut.ut_user));
609             if (xdisplay)
610                 strncpy (ut.ut_host, xdisplay, sizeof (ut.ut_host));
611             else if (remote_host_name)
612                 strncpy (ut.ut_host, remote_host_name, sizeof (ut.ut_host));
613             gettimeofday (&tv, NULL);
614             ut.ut_tv.tv_sec = tv.tv_sec;
615             ut.ut_tv.tv_usec = tv.tv_usec;
616
617             setutxent ();
618             if (!pututxline (&ut))
619                 g_printerr ("Failed to write utmpx: %s\n", strerror (errno));
620             endutxent ();
621         }
622
623         waitpid (child_pid, &return_code, 0);
624         child_pid = 0;
625
626         /* Log to utmp */
627         if (g_strcmp0 (pam_getenv (pam_handle, "XDG_SESSION_CLASS"), "greeter") != 0)
628         {
629             struct utmpx ut;
630             struct timeval tv;
631
632             memset (&ut, 0, sizeof (ut));
633             ut.ut_type = DEAD_PROCESS;
634             ut.ut_pid = child_pid;
635             if (tty)
636                 strncpy (ut.ut_line, tty + strlen ("/dev/"), sizeof (ut.ut_line));
637             if (xdisplay)
638                 strncpy (ut.ut_id, xdisplay, sizeof (ut.ut_id));
639             strncpy (ut.ut_user, username, sizeof (ut.ut_user));
640             if (xdisplay)
641                 strncpy (ut.ut_host, xdisplay, sizeof (ut.ut_host));
642             else if (remote_host_name)
643                 strncpy (ut.ut_host, remote_host_name, sizeof (ut.ut_host));
644             gettimeofday (&tv, NULL);
645             ut.ut_tv.tv_sec = tv.tv_sec;
646             ut.ut_tv.tv_usec = tv.tv_usec;
647
648             setutxent ();
649             if (!pututxline (&ut))
650                 g_printerr ("Failed to write utmpx: %s\n", strerror (errno));
651             endutxent ();
652         }
653     }
654
655     /* Remove X authority */
656     if (x_authority)
657     {
658         gboolean drop_privileges, result;
659         GError *error = NULL;
660
661         drop_privileges = geteuid () == 0;
662         if (drop_privileges)
663             privileges_drop (user_get_uid (user), user_get_gid (user));
664         result = x_authority_write (x_authority, XAUTH_WRITE_MODE_REMOVE, x_authority_filename, &error);
665         if (drop_privileges)
666             privileges_reclaim ();
667
668         if (error)
669             g_printerr ("Error removing X authority: %s\n", error->message);
670         g_clear_error (&error);
671         if (!result)
672             _exit (EXIT_FAILURE);
673     }
674
675     /* Close the Console Kit session */
676     if (console_kit_cookie)
677         ck_close_session (console_kit_cookie);
678
679     /* Close the session */
680     pam_close_session (pam_handle, 0);
681
682     /* Remove credentials */
683     result = pam_setcred (pam_handle, PAM_DELETE_CRED);
684
685     pam_end (pam_handle, 0);
686     pam_handle = NULL;
687
688     /* Return result of session process to the daemon */
689     return return_code;
690 }