1 /* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- */
2 /* dbus-launch.c dbus-launch utility
4 * Copyright (C) 2003, 2006 Red Hat, Inc.
5 * Copyright (C) 2006 Thiago Macieira <thiago@kde.org>
7 * Licensed under the Academic Free License version 2.1
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License
20 * along with this program; if not, write to the Free Software
21 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
24 #include "dbus-launch.h"
35 #include <sys/select.h>
40 extern Display *xdisplay;
43 static char* machine_uuid = NULL;
46 get_machine_uuid (void)
52 save_machine_uuid (const char *uuid_arg)
54 if (strlen (uuid_arg) != 32)
56 fprintf (stderr, "machine ID '%s' looks like it's the wrong length, should be 32 hex digits",
61 machine_uuid = xstrdup (uuid_arg);
65 verbose (const char *format,
69 static int verbose = TRUE;
70 static int verbose_initted = FALSE;
72 /* things are written a bit oddly here so that
73 * in the non-verbose case we just have the one
74 * conditional and return immediately.
81 verbose = getenv ("DBUS_VERBOSE") != NULL;
82 verbose_initted = TRUE;
87 fprintf (stderr, "%lu: ", (unsigned long) getpid ());
89 va_start (args, format);
90 vfprintf (stderr, format, args);
97 fprintf (stderr, "dbus-launch [--version] [--help] [--sh-syntax] [--csh-syntax] [--auto-syntax] [--exit-with-session]\n");
104 printf ("D-Bus Message Bus Launcher %s\n"
105 "Copyright (C) 2003 Red Hat, Inc.\n"
106 "This is free software; see the source for copying conditions.\n"
107 "There is NO warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n",
113 xstrdup (const char *str)
123 copy = malloc (len + 1);
127 memcpy (copy, str, len + 1);
134 READ_STATUS_OK, /**< Read succeeded */
135 READ_STATUS_ERROR, /**< Some kind of error */
136 READ_STATUS_EOF /**< EOF returned */
147 memset (buf, '\0', maxlen);
148 maxlen -= 1; /* ensure nul term */
150 retval = READ_STATUS_OK;
158 to_read = maxlen - bytes;
166 if (chunk < 0 && errno == EINTR)
171 retval = READ_STATUS_ERROR;
176 retval = READ_STATUS_EOF;
183 if (retval == READ_STATUS_EOF &&
185 retval = READ_STATUS_OK;
188 if (retval != READ_STATUS_ERROR &&
190 buf[bytes-1] == '\n')
203 retval = READ_STATUS_OK;
211 to_read = sizeof (pid_t) - bytes;
217 ((char*)buf) + bytes,
219 if (chunk < 0 && errno == EINTR)
224 retval = READ_STATUS_ERROR;
229 retval = READ_STATUS_EOF;
240 do_write (int fd, const void *buf, size_t count)
242 size_t bytes_written;
249 ret = write (fd, ((const char*)buf) + bytes_written, count - bytes_written);
257 fprintf (stderr, "Failed to write data to pipe! %s\n",
259 exit (1); /* give up, we suck */
263 bytes_written += ret;
265 if (bytes_written < count)
273 do_write (fd, &pid, sizeof (pid));
277 do_waitpid (pid_t pid)
282 ret = waitpid (pid, NULL, 0);
291 static pid_t bus_pid_to_kill = -1;
294 kill_bus_and_exit (int exitcode)
296 verbose ("Killing message bus and exiting babysitter\n");
298 /* in case these point to any NFS mounts, get rid of them immediately */
303 kill (bus_pid_to_kill, SIGTERM);
305 kill (bus_pid_to_kill, SIGKILL);
311 print_variables (const char *bus_address, pid_t bus_pid, long bus_wid,
312 int c_shell_syntax, int bourne_shell_syntax,
317 write (1, bus_address, strlen (bus_address) + 1);
318 write (1, &bus_pid, sizeof bus_pid);
319 write (1, &bus_wid, sizeof bus_wid);
322 else if (c_shell_syntax)
324 printf ("setenv DBUS_SESSION_BUS_ADDRESS '%s';\n", bus_address);
325 printf ("set DBUS_SESSION_BUS_PID=%ld;\n", (long) bus_pid);
327 printf ("set DBUS_SESSION_BUS_WINDOWID=%ld;\n", (long) bus_wid);
330 else if (bourne_shell_syntax)
332 printf ("DBUS_SESSION_BUS_ADDRESS='%s';\n", bus_address);
333 if (bourne_shell_syntax)
334 printf ("export DBUS_SESSION_BUS_ADDRESS;\n");
335 printf ("DBUS_SESSION_BUS_PID=%ld;\n", (long) bus_pid);
337 printf ("DBUS_SESSION_BUS_WINDOWID=%ld;\n", (long) bus_wid);
342 printf ("DBUS_SESSION_BUS_ADDRESS=%s\n", bus_address);
343 printf ("DBUS_SESSION_BUS_PID=%ld\n", (long) bus_pid);
345 printf ("DBUS_SESSION_BUS_WINDOWID=%ld\n", (long) bus_wid);
350 static int got_sighup = FALSE;
353 signal_handler (int sig)
365 kill_bus_when_session_ends (void)
371 struct sigaction act;
374 /* install SIGHUP handler */
376 sigemptyset (&empty_mask);
377 act.sa_handler = signal_handler;
378 act.sa_mask = empty_mask;
380 sigaction (SIGHUP, &act, NULL);
381 sigaction (SIGTERM, &act, NULL);
383 #ifdef DBUS_BUILD_X11
385 if (xdisplay != NULL)
387 x_fd = ConnectionNumber (xdisplay);
401 verbose ("stdin isatty(), monitoring it\n");
403 verbose ("stdin was not a TTY, not monitoring it\n");
405 if (tty_fd < 0 && x_fd < 0)
407 fprintf (stderr, "No terminal on standard input and no X display; cannot attach message bus to session lifetime\n");
418 FD_SET (tty_fd, &read_set);
419 FD_SET (tty_fd, &err_set);
424 FD_SET (x_fd, &read_set);
425 FD_SET (x_fd, &err_set);
428 select (MAX (tty_fd, x_fd) + 1,
429 &read_set, NULL, &err_set, NULL);
433 verbose ("Got SIGHUP, exiting\n");
434 kill_bus_and_exit (0);
437 #ifdef DBUS_BUILD_X11
438 /* Dump events on the floor, and let
439 * IO error handler run if we lose
443 verbose ("X fd condition reading = %d error = %d\n",
444 FD_ISSET (x_fd, &read_set),
445 FD_ISSET (x_fd, &err_set));
451 if (FD_ISSET (tty_fd, &read_set))
456 verbose ("TTY ready for reading\n");
458 bytes_read = read (tty_fd, discard, sizeof (discard));
460 verbose ("Read %d bytes from TTY errno = %d\n",
464 kill_bus_and_exit (0); /* EOF */
465 else if (bytes_read < 0 && errno != EINTR)
467 /* This shouldn't happen I don't think; to avoid
468 * spinning on the fd forever we exit.
470 fprintf (stderr, "dbus-launch: error reading from stdin: %s\n",
472 kill_bus_and_exit (0);
475 else if (FD_ISSET (tty_fd, &err_set))
477 verbose ("TTY has error condition\n");
479 kill_bus_and_exit (0);
486 babysit (int exit_with_session,
488 int read_bus_pid_fd) /* read pid from here */
494 verbose ("babysitting, exit_with_session = %d, child_pid = %ld, read_bus_pid_fd = %d\n",
495 exit_with_session, (long) child_pid, read_bus_pid_fd);
497 /* We chdir ("/") since we are persistent and daemon-like, and fork
498 * again so dbus-launch can reap the parent. However, we don't
499 * setsid() or close fd 0 because the idea is to remain attached
500 * to the tty and the X server in order to kill the message bus
501 * when the session ends.
506 fprintf (stderr, "Could not change to root directory: %s\n",
511 /* Close stdout/stderr so we don't block an "eval" or otherwise
512 * lock up. stdout is still chaining through to dbus-launch
513 * and in turn to the parent shell.
515 dev_null_fd = open ("/dev/null", O_RDWR);
516 if (dev_null_fd >= 0)
518 if (!exit_with_session)
519 dup2 (dev_null_fd, 0);
520 dup2 (dev_null_fd, 1);
521 s = getenv ("DBUS_DEBUG_OUTPUT");
522 if (s == NULL || *s == '\0')
523 dup2 (dev_null_fd, 2);
527 fprintf (stderr, "Failed to open /dev/null: %s\n",
529 /* continue, why not */
536 fprintf (stderr, "fork() failed in babysitter: %s\n",
543 /* Parent reaps pre-fork part of bus daemon, then exits and is
544 * reaped so the babysitter isn't a zombie
547 verbose ("=== Babysitter's intermediate parent continues again\n");
549 if (do_waitpid (child_pid) < 0)
551 /* shouldn't happen */
552 fprintf (stderr, "Failed waitpid() waiting for bus daemon's parent\n");
556 verbose ("Babysitter's intermediate parent exiting\n");
561 /* Child continues */
562 verbose ("=== Babysitter process created\n");
564 verbose ("Reading PID from bus\n");
566 switch (read_pid (read_bus_pid_fd, &bus_pid_to_kill))
570 case READ_STATUS_EOF:
571 fprintf (stderr, "EOF in dbus-launch reading PID from bus daemon\n");
574 case READ_STATUS_ERROR:
575 fprintf (stderr, "Error in dbus-launch reading PID from bus daemon: %s\n",
581 verbose ("Got PID %ld from daemon\n",
582 (long) bus_pid_to_kill);
584 if (exit_with_session)
586 /* Bus is now started and launcher has needed info;
587 * we connect to X display and tty and wait to
588 * kill bus if requested.
591 kill_bus_when_session_ends ();
594 verbose ("Babysitter exiting\n");
600 do_close_stderr (void)
606 /* dbus-launch is a Unix-only program, so we can rely on /dev/null being there.
607 * We're including unistd.h and we're dealing with sh/csh launch sequences...
609 fd = open ("/dev/null", O_RDWR);
612 fprintf (stderr, "Internal error: cannot open /dev/null: %s", strerror (errno));
617 if (dup2 (fd, 2) == -1)
618 // error; we can't report an error anymore...
627 main (int argc, char **argv)
629 const char *prev_arg;
631 const char *runprog = NULL;
632 int remaining_args = 0;
633 int exit_with_session;
634 int binary_syntax = FALSE;
635 int c_shell_syntax = FALSE;
636 int bourne_shell_syntax = FALSE;
637 int auto_shell_syntax = FALSE;
638 int autolaunch = FALSE;
639 int requires_arg = FALSE;
640 int close_stderr = FALSE;
643 int bus_pid_to_launcher_pipe[2];
644 int bus_pid_to_babysitter_pipe[2];
645 int bus_address_to_launcher_pipe[2];
648 exit_with_session = FALSE;
655 const char *arg = argv[i];
657 if (strcmp (arg, "--help") == 0 ||
658 strcmp (arg, "-h") == 0 ||
659 strcmp (arg, "-?") == 0)
661 else if (strcmp (arg, "--auto-syntax") == 0)
662 auto_shell_syntax = TRUE;
663 else if (strcmp (arg, "-c") == 0 ||
664 strcmp (arg, "--csh-syntax") == 0)
665 c_shell_syntax = TRUE;
666 else if (strcmp (arg, "-s") == 0 ||
667 strcmp (arg, "--sh-syntax") == 0)
668 bourne_shell_syntax = TRUE;
669 else if (strcmp (arg, "--binary-syntax") == 0)
670 binary_syntax = TRUE;
671 else if (strcmp (arg, "--version") == 0)
673 else if (strcmp (arg, "--exit-with-session") == 0)
674 exit_with_session = TRUE;
675 else if (strcmp (arg, "--close-stderr") == 0)
677 else if (strstr (arg, "--autolaunch=") == arg)
683 fprintf (stderr, "--autolaunch given twice\n");
689 s = strchr (arg, '=');
692 save_machine_uuid (s);
695 strcmp (prev_arg, "--autolaunch") == 0)
699 fprintf (stderr, "--autolaunch given twice\n");
705 save_machine_uuid (arg);
706 requires_arg = FALSE;
708 else if (strcmp (arg, "--autolaunch") == 0)
710 else if (strstr (arg, "--config-file=") == arg)
714 if (config_file != NULL)
716 fprintf (stderr, "--config-file given twice\n");
720 file = strchr (arg, '=');
723 config_file = xstrdup (file);
726 strcmp (prev_arg, "--config-file") == 0)
728 if (config_file != NULL)
730 fprintf (stderr, "--config-file given twice\n");
734 config_file = xstrdup (arg);
735 requires_arg = FALSE;
737 else if (strcmp (arg, "--config-file") == 0)
739 else if (arg[0] == '-')
741 if (strcmp (arg, "--") != 0)
743 fprintf (stderr, "Option `%s' is unknown.\n", arg);
749 remaining_args = i+2;
756 remaining_args = i+1;
766 fprintf (stderr, "Option `%s' requires an argument.\n", prev_arg);
770 if (auto_shell_syntax)
772 if ((shname = getenv ("SHELL")) != NULL)
774 if (!strncmp (shname + strlen (shname) - 3, "csh", 3))
775 c_shell_syntax = TRUE;
777 bourne_shell_syntax = TRUE;
780 bourne_shell_syntax = TRUE;
783 if (exit_with_session)
784 verbose ("--exit-with-session enabled\n");
788 #ifndef DBUS_BUILD_X11
789 fprintf (stderr, "Autolaunch requested, but X11 support not compiled in.\n"
790 "Cannot continue.\n");
797 if (get_machine_uuid () == NULL)
799 fprintf (stderr, "Machine UUID not provided as arg to --autolaunch\n");
803 /* FIXME right now autolaunch always does print_variables(), but it should really
804 * exec the child program instead if a child program was specified. For now
805 * we just exit if this conflict arises.
809 fprintf (stderr, "Currently --autolaunch does not support running a program\n");
813 verbose ("Autolaunch enabled (using X11).\n");
814 if (!exit_with_session)
816 verbose ("--exit-with-session automatically enabled\n");
817 exit_with_session = TRUE;
822 fprintf (stderr, "Autolaunch error: X11 initialization failed.\n");
826 if (!x11_get_address (&address, &pid, &wid))
828 fprintf (stderr, "Autolaunch error: X11 communication error.\n");
834 verbose ("dbus-daemon is already running. Returning existing parameters.\n");
835 print_variables (address, pid, wid, c_shell_syntax,
836 bourne_shell_syntax, binary_syntax);
842 if (pipe (bus_pid_to_launcher_pipe) < 0 ||
843 pipe (bus_address_to_launcher_pipe) < 0 ||
844 pipe (bus_pid_to_babysitter_pipe) < 0)
847 "Failed to create pipe: %s\n",
855 fprintf (stderr, "Failed to fork: %s\n",
863 #define MAX_FD_LEN 64
864 char write_pid_fd_as_string[MAX_FD_LEN];
865 char write_address_fd_as_string[MAX_FD_LEN];
870 verbose ("=== Babysitter's intermediate parent created\n");
872 /* Fork once more to create babysitter */
877 fprintf (stderr, "Failed to fork: %s\n",
885 verbose ("=== Babysitter's intermediate parent continues\n");
887 close (bus_pid_to_launcher_pipe[READ_END]);
888 close (bus_pid_to_launcher_pipe[WRITE_END]);
889 close (bus_address_to_launcher_pipe[READ_END]);
890 close (bus_address_to_launcher_pipe[WRITE_END]);
891 close (bus_pid_to_babysitter_pipe[WRITE_END]);
893 /* babysit() will fork *again*
894 * and will also reap the pre-forked bus
897 babysit (exit_with_session, ret,
898 bus_pid_to_babysitter_pipe[READ_END]);
902 verbose ("=== Bus exec process created\n");
904 /* Now we are the bus process (well, almost;
905 * dbus-daemon itself forks again)
907 close (bus_pid_to_launcher_pipe[READ_END]);
908 close (bus_address_to_launcher_pipe[READ_END]);
909 close (bus_pid_to_babysitter_pipe[READ_END]);
910 close (bus_pid_to_babysitter_pipe[WRITE_END]);
912 sprintf (write_pid_fd_as_string,
913 "%d", bus_pid_to_launcher_pipe[WRITE_END]);
915 sprintf (write_address_fd_as_string,
916 "%d", bus_address_to_launcher_pipe[WRITE_END]);
918 verbose ("Calling exec()\n");
920 #ifdef DBUS_BUILD_TESTS
921 /* exec from testdir */
922 if (getenv("DBUS_USE_TEST_BINARY") != NULL)
924 execl (TEST_BUS_BINARY,
927 "--print-pid", write_pid_fd_as_string,
928 "--print-address", write_address_fd_as_string,
929 config_file ? "--config-file" : "--session",
930 config_file, /* has to be last in this varargs list */
934 "Failed to execute test message bus daemon %s: %s. Will try again with the system path.\n",
935 TEST_BUS_BINARY, strerror (errno));
937 #endif /* DBUS_BUILD_TESTS */
939 execl (DBUS_DAEMONDIR"/dbus-daemon",
940 DBUS_DAEMONDIR"/dbus-daemon",
942 "--print-pid", write_pid_fd_as_string,
943 "--print-address", write_address_fd_as_string,
944 config_file ? "--config-file" : "--session",
945 config_file, /* has to be last in this varargs list */
949 "Failed to execute message bus daemon %s: %s. Will try again without full path.\n",
950 DBUS_DAEMONDIR"/dbus-daemon", strerror (errno));
953 * If it failed, try running without full PATH. Note this is needed
954 * because the build process builds the run-with-tmp-session-bus.conf
955 * file and the dbus-daemon will not be in the install location during
958 execlp ("dbus-daemon",
961 "--print-pid", write_pid_fd_as_string,
962 "--print-address", write_address_fd_as_string,
963 config_file ? "--config-file" : "--session",
964 config_file, /* has to be last in this varargs list */
968 "Failed to execute message bus daemon: %s\n",
975 #define MAX_PID_LEN 64
977 char bus_address[MAX_ADDR_LEN];
978 char buf[MAX_PID_LEN];
984 verbose ("=== Parent dbus-launch continues\n");
986 close (bus_pid_to_launcher_pipe[WRITE_END]);
987 close (bus_address_to_launcher_pipe[WRITE_END]);
988 close (bus_pid_to_babysitter_pipe[READ_END]);
990 verbose ("Waiting for babysitter's intermediate parent\n");
992 /* Immediately reap parent of babysitter
993 * (which was created just for us to reap)
995 if (do_waitpid (ret) < 0)
997 fprintf (stderr, "Failed to waitpid() for babysitter intermediate process: %s\n",
1002 verbose ("Reading address from bus\n");
1004 /* Read the pipe data, print, and exit */
1005 switch (read_line (bus_address_to_launcher_pipe[READ_END],
1006 bus_address, MAX_ADDR_LEN))
1008 case READ_STATUS_OK:
1010 case READ_STATUS_EOF:
1011 fprintf (stderr, "EOF in dbus-launch reading address from bus daemon\n");
1014 case READ_STATUS_ERROR:
1015 fprintf (stderr, "Error in dbus-launch reading address from bus daemon: %s\n",
1021 close (bus_address_to_launcher_pipe[READ_END]);
1023 verbose ("Reading PID from daemon\n");
1025 switch (read_line (bus_pid_to_launcher_pipe[READ_END], buf, MAX_PID_LEN))
1027 case READ_STATUS_OK:
1029 case READ_STATUS_EOF:
1030 fprintf (stderr, "EOF reading PID from bus daemon\n");
1033 case READ_STATUS_ERROR:
1034 fprintf (stderr, "Error reading PID from bus daemon: %s\n",
1041 val = strtol (buf, &end, 0);
1042 if (buf == end || end == NULL)
1044 fprintf (stderr, "Failed to parse bus PID \"%s\": %s\n",
1045 buf, strerror (errno));
1051 close (bus_pid_to_launcher_pipe[READ_END]);
1053 #ifdef DBUS_BUILD_X11
1054 /* FIXME the runprog == NULL is broken - we need to launch the runprog with the existing bus,
1055 * instead of just doing print_variables() if there's an existing bus.
1057 if (xdisplay != NULL && runprog == NULL)
1059 ret2 = x11_save_address (bus_address, bus_pid, &wid);
1062 /* another window got added. Return its address */
1067 if (x11_get_address (&address, &pid, &wid) && address != NULL)
1069 verbose ("dbus-daemon is already running. Returning existing parameters.\n");
1070 print_variables (address, pid, wid, c_shell_syntax,
1071 bourne_shell_syntax, binary_syntax);
1074 bus_pid_to_kill = bus_pid;
1075 kill_bus_and_exit (0);
1078 /* if failed, fall through */
1082 fprintf (stderr, "Error saving bus information.\n");
1083 bus_pid_to_kill = bus_pid;
1084 kill_bus_and_exit (1);
1089 /* Forward the pid to the babysitter */
1090 write_pid (bus_pid_to_babysitter_pipe[WRITE_END], bus_pid);
1091 close (bus_pid_to_babysitter_pipe[WRITE_END]);
1098 envvar = malloc (strlen ("DBUS_SESSION_BUS_ADDRESS=") + strlen (bus_address) + 1);
1099 args = malloc (sizeof (char *) * ((argc-remaining_args)+2));
1101 if (envvar == NULL || args == NULL)
1104 args[0] = xstrdup (runprog);
1107 for (i = 1; i <= (argc-remaining_args); i++)
1109 size_t len = strlen (argv[remaining_args+i-1])+1;
1110 args[i] = malloc (len);
1113 strncpy (args[i], argv[remaining_args+i-1], len);
1117 strcpy (envvar, "DBUS_SESSION_BUS_ADDRESS=");
1118 strcat (envvar, bus_address);
1121 execvp (runprog, args);
1122 fprintf (stderr, "Couldn't exec %s: %s\n", runprog, strerror (errno));
1127 print_variables (bus_address, bus_pid, wid, c_shell_syntax,
1128 bourne_shell_syntax, binary_syntax);
1131 verbose ("dbus-launch exiting\n");
1143 fprintf (stderr, "Out of memory!");