1 /* -*- mode: C; c-file-style: "gnu" -*- */
2 /* dbus-launch.c dbus-launch utility
4 * Copyright (C) 2003 Red Hat, Inc.
6 * Licensed under the Academic Free License version 1.2
8 * This program is free software; you can redistribute it and/or modify
9 * it under the terms of the GNU General Public License as published by
10 * the Free Software Foundation; either version 2 of the License, or
11 * (at your option) any later version.
13 * This program is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 * GNU General Public License for more details.
18 * You should have received a copy of the GNU General Public License
19 * along with this program; if not, write to the Free Software
20 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
47 #define MAX(a, b) (((a) > (b)) ? (a) : (b))
50 verbose (const char *format,
54 static int verbose = TRUE;
55 static int verbose_initted = FALSE;
57 /* things are written a bit oddly here so that
58 * in the non-verbose case we just have the one
59 * conditional and return immediately.
66 verbose = getenv ("DBUS_VERBOSE") != NULL;
67 verbose_initted = TRUE;
72 fprintf (stderr, "%lu: ", (unsigned long) getpid ());
74 va_start (args, format);
75 vfprintf (stderr, format, args);
82 fprintf (stderr, "dbus-launch [--version] [--help] [--sh-syntax] [--csh-syntax] [--auto-syntax] [--exit-with-session]\n");
89 printf ("D-BUS Message Bus Launcher %s\n"
90 "Copyright (C) 2003 Red Hat, Inc.\n"
91 "This is free software; see the source for copying conditions.\n"
92 "There is NO warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n",
98 xstrdup (const char *str)
108 copy = malloc (len + 1);
112 memcpy (copy, str, len + 1);
119 READ_STATUS_OK, /**< Read succeeded */
120 READ_STATUS_ERROR, /**< Some kind of error */
121 READ_STATUS_EOF /**< EOF returned */
132 memset (buf, '\0', maxlen);
133 maxlen -= 1; /* ensure nul term */
135 retval = READ_STATUS_OK;
143 to_read = maxlen - bytes;
151 if (chunk < 0 && errno == EINTR)
156 retval = READ_STATUS_ERROR;
161 retval = READ_STATUS_EOF;
168 if (retval == READ_STATUS_EOF &&
170 retval = READ_STATUS_OK;
173 if (retval != READ_STATUS_ERROR &&
175 buf[bytes-1] == '\n')
188 retval = READ_STATUS_OK;
196 to_read = sizeof (pid_t) - bytes;
202 ((char*)buf) + bytes,
204 if (chunk < 0 && errno == EINTR)
209 retval = READ_STATUS_ERROR;
214 retval = READ_STATUS_EOF;
225 do_write (int fd, const void *buf, size_t count)
227 size_t bytes_written;
234 ret = write (fd, ((const char*)buf) + bytes_written, count - bytes_written);
242 fprintf (stderr, "Failed to write data to pipe!\n");
243 exit (1); /* give up, we suck */
247 bytes_written += ret;
249 if (bytes_written < count)
257 do_write (fd, &pid, sizeof (pid));
261 do_waitpid (pid_t pid)
266 ret = waitpid (pid, NULL, 0);
275 static pid_t bus_pid_to_kill = -1;
278 kill_bus_and_exit (void)
280 verbose ("Killing message bus and exiting babysitter\n");
282 /* in case these point to any NFS mounts, get rid of them immediately */
287 kill (bus_pid_to_kill, SIGTERM);
289 kill (bus_pid_to_kill, SIGKILL);
294 #ifdef DBUS_BUILD_X11
296 x_io_error_handler (Display *xdisplay)
298 verbose ("X IO error\n");
299 kill_bus_and_exit ();
304 static int got_sighup = FALSE;
307 signal_handler (int sig)
318 kill_bus_when_session_ends (void)
325 struct sigaction act;
327 #ifdef DBUS_BUILD_X11
331 /* install SIGHUP handler */
333 sigemptyset (&empty_mask);
334 act.sa_handler = signal_handler;
335 act.sa_mask = empty_mask;
337 sigaction (SIGHUP, &act, 0);
339 #ifdef DBUS_BUILD_X11
340 xdisplay = XOpenDisplay (NULL);
341 if (xdisplay != NULL)
343 verbose ("Successfully opened X display\n");
344 x_fd = ConnectionNumber (xdisplay);
345 XSetIOErrorHandler (x_io_error_handler);
350 verbose ("Compiled without X11 support\n");
360 verbose ("stdin isatty(), monitoring it\n");
362 verbose ("stdin was not a TTY, not monitoring it\n");
364 if (tty_fd < 0 && x_fd < 0)
366 fprintf (stderr, "No terminal on standard input and no X display; cannot attach message bus to session lifetime\n");
377 FD_SET (tty_fd, &read_set);
378 FD_SET (tty_fd, &err_set);
383 FD_SET (x_fd, &read_set);
384 FD_SET (x_fd, &err_set);
387 ret = select (MAX (tty_fd, x_fd) + 1,
388 &read_set, NULL, &err_set, NULL);
392 verbose ("Got SIGHUP, exiting\n");
393 kill_bus_and_exit ();
396 #ifdef DBUS_BUILD_X11
397 /* Dump events on the floor, and let
398 * IO error handler run if we lose
402 verbose ("X fd condition reading = %d error = %d\n",
403 FD_ISSET (x_fd, &read_set),
404 FD_ISSET (x_fd, &err_set));
406 if (xdisplay != NULL)
408 while (XPending (xdisplay))
411 XNextEvent (xdisplay, &ignored);
418 if (FD_ISSET (tty_fd, &read_set))
423 verbose ("TTY ready for reading\n");
425 bytes_read = read (tty_fd, discard, sizeof (discard));
427 verbose ("Read %d bytes from TTY errno = %d\n",
431 kill_bus_and_exit (); /* EOF */
432 else if (bytes_read < 0 && errno != EINTR)
434 /* This shouldn't happen I don't think; to avoid
435 * spinning on the fd forever we exit.
437 fprintf (stderr, "dbus-launch: error reading from stdin: %s\n",
439 kill_bus_and_exit ();
442 else if (FD_ISSET (tty_fd, &err_set))
444 verbose ("TTY has error condition\n");
446 kill_bus_and_exit ();
453 babysit (int exit_with_session,
455 int read_bus_pid_fd, /* read pid from here */
456 int write_bus_pid_fd) /* forward pid to here */
459 #define MAX_PID_LEN 64
460 char buf[MAX_PID_LEN];
465 /* We chdir ("/") since we are persistent and daemon-like, and fork
466 * again so dbus-launch can reap the parent. However, we don't
467 * setsid() or close fd 0 because the idea is to remain attached
468 * to the tty and the X server in order to kill the message bus
469 * when the session ends.
474 fprintf (stderr, "Could not change to root directory: %s\n",
479 /* Move stdout/stderr so we don't block an "eval" or otherwise
482 dev_null_fd = open ("/dev/null", O_RDWR);
483 if (dev_null_fd >= 0)
485 dup2 (dev_null_fd, 1);
486 dup2 (dev_null_fd, 2);
490 fprintf (stderr, "Failed to open /dev/null: %s\n",
492 /* continue, why not */
499 fprintf (stderr, "fork() failed in babysitter: %s\n",
506 /* Parent reaps pre-fork part of bus daemon, then exits and is
507 * reaped so the babysitter isn't a zombie
510 verbose ("=== Babysitter's intermediate parent continues again\n");
512 if (do_waitpid (child_pid) < 0)
514 /* shouldn't happen */
515 fprintf (stderr, "Failed waitpid() waiting for bus daemon's parent\n");
519 verbose ("Babysitter's intermediate parent exiting\n");
524 /* Child continues */
525 verbose ("=== Babysitter process created\n");
527 verbose ("Reading PID from daemon\n");
529 switch (read_line (read_bus_pid_fd, buf, MAX_PID_LEN))
533 case READ_STATUS_EOF:
534 fprintf (stderr, "EOF reading PID from bus daemon\n");
537 case READ_STATUS_ERROR:
538 fprintf (stderr, "Error reading PID from bus daemon: %s\n",
545 val = strtol (buf, &end, 0);
546 if (buf == end || end == NULL)
548 fprintf (stderr, "Failed to parse bus PID \"%s\": %s\n",
549 buf, strerror (errno));
553 bus_pid_to_kill = val;
555 verbose ("Got PID %ld from daemon\n",
556 (long) bus_pid_to_kill);
558 /* Write data to launcher */
559 write_pid (write_bus_pid_fd, bus_pid_to_kill);
560 close (write_bus_pid_fd);
562 if (exit_with_session)
564 /* Bus is now started and launcher has needed info;
565 * we connect to X display and tty and wait to
566 * kill bus if requested.
569 kill_bus_when_session_ends ();
572 verbose ("Babysitter exiting\n");
581 main (int argc, char **argv)
583 const char *prev_arg;
585 const char *runprog = NULL;
586 int remaining_args = 0;
587 int exit_with_session;
588 int c_shell_syntax = FALSE;
589 int bourne_shell_syntax = FALSE;
590 int auto_shell_syntax = FALSE;
593 int bus_pid_to_launcher_pipe[2];
594 int bus_pid_to_babysitter_pipe[2];
595 int bus_address_to_launcher_pipe[2];
598 exit_with_session = FALSE;
605 const char *arg = argv[i];
607 if (strcmp (arg, "--help") == 0 ||
608 strcmp (arg, "-h") == 0 ||
609 strcmp (arg, "-?") == 0)
611 else if (strcmp (arg, "--auto-syntax") == 0)
612 auto_shell_syntax = TRUE;
613 else if (strcmp (arg, "-c") == 0 ||
614 strcmp (arg, "--csh-syntax") == 0)
615 c_shell_syntax = TRUE;
616 else if (strcmp (arg, "-s") == 0 ||
617 strcmp (arg, "--sh-syntax") == 0)
618 bourne_shell_syntax = TRUE;
619 else if (strcmp (arg, "--version") == 0)
621 else if (strcmp (arg, "--exit-with-session") == 0)
622 exit_with_session = TRUE;
623 else if (strstr (arg, "--config-file=") == arg)
627 if (config_file != NULL)
629 fprintf (stderr, "--config-file given twice\n");
633 file = strchr (arg, '=');
636 config_file = xstrdup (file);
639 strcmp (prev_arg, "--config-file") == 0)
641 if (config_file != NULL)
643 fprintf (stderr, "--config-file given twice\n");
647 config_file = xstrdup (arg);
649 else if (strcmp (arg, "--config-file") == 0)
650 ; /* wait for next arg */
654 remaining_args = i+1;
663 if (exit_with_session)
664 verbose ("--exit-with-session enabled\n");
666 if (auto_shell_syntax)
668 if ((shname = getenv ("SHELL")) != NULL)
670 if (!strncmp (shname + strlen (shname) - 3, "csh", 3))
671 c_shell_syntax = TRUE;
673 bourne_shell_syntax = TRUE;
676 bourne_shell_syntax = TRUE;
679 if (pipe (bus_pid_to_launcher_pipe) < 0 ||
680 pipe (bus_address_to_launcher_pipe) < 0)
683 "Failed to create pipe: %s\n",
688 bus_pid_to_babysitter_pipe[READ_END] = -1;
689 bus_pid_to_babysitter_pipe[WRITE_END] = -1;
694 fprintf (stderr, "Failed to fork: %s\n",
702 #define MAX_FD_LEN 64
703 char write_pid_fd_as_string[MAX_FD_LEN];
704 char write_address_fd_as_string[MAX_FD_LEN];
706 verbose ("=== Babysitter's intermediate parent created\n");
708 /* Fork once more to create babysitter */
710 if (pipe (bus_pid_to_babysitter_pipe) < 0)
713 "Failed to create pipe: %s\n",
721 fprintf (stderr, "Failed to fork: %s\n",
729 verbose ("=== Babysitter's intermediate parent continues\n");
731 close (bus_pid_to_launcher_pipe[READ_END]);
732 close (bus_address_to_launcher_pipe[READ_END]);
733 close (bus_address_to_launcher_pipe[WRITE_END]);
734 close (bus_pid_to_babysitter_pipe[WRITE_END]);
736 /* babysit() will fork *again*
737 * and will also reap the pre-forked bus
740 babysit (exit_with_session, ret,
741 bus_pid_to_babysitter_pipe[READ_END],
742 bus_pid_to_launcher_pipe[WRITE_END]);
746 verbose ("=== Bus exec process created\n");
748 /* Now we are the bus process (well, almost;
749 * dbus-daemon-1 itself forks again)
751 close (bus_pid_to_launcher_pipe[READ_END]);
752 close (bus_address_to_launcher_pipe[READ_END]);
753 close (bus_pid_to_babysitter_pipe[READ_END]);
754 close (bus_pid_to_launcher_pipe[WRITE_END]);
756 sprintf (write_pid_fd_as_string,
757 "%d", bus_pid_to_babysitter_pipe[WRITE_END]);
759 sprintf (write_address_fd_as_string,
760 "%d", bus_address_to_launcher_pipe[WRITE_END]);
762 verbose ("Calling exec()\n");
764 execlp ("dbus-daemon-1",
767 "--print-pid", write_pid_fd_as_string,
768 "--print-address", write_address_fd_as_string,
769 config_file ? "--config-file" : "--session",
770 config_file, /* has to be last in this varargs list */
774 "Failed to execute message bus daemon: %s\n",
781 #define MAX_ADDR_LEN 512
783 char bus_address[MAX_ADDR_LEN];
785 verbose ("=== Parent dbus-launch continues\n");
787 close (bus_pid_to_launcher_pipe[WRITE_END]);
788 close (bus_address_to_launcher_pipe[WRITE_END]);
790 verbose ("Waiting for babysitter's intermediate parent\n");
792 /* Immediately reap parent of babysitter
793 * (which was created just for us to reap)
795 if (do_waitpid (ret) < 0)
797 fprintf (stderr, "Failed to waitpid() for babysitter intermediate process: %s\n",
802 verbose ("Reading address from bus\n");
804 /* Read the pipe data, print, and exit */
805 switch (read_line (bus_address_to_launcher_pipe[READ_END],
806 bus_address, MAX_ADDR_LEN))
810 case READ_STATUS_EOF:
811 fprintf (stderr, "EOF in dbus-launch reading address from bus daemon\n");
814 case READ_STATUS_ERROR:
815 fprintf (stderr, "Error in dbus-launch reading address from bus daemon: %s\n",
821 close (bus_address_to_launcher_pipe[READ_END]);
823 verbose ("Reading PID from babysitter\n");
825 switch (read_pid (bus_pid_to_launcher_pipe[READ_END], &bus_pid))
829 case READ_STATUS_EOF:
830 fprintf (stderr, "EOF in dbus-launch reading address from bus daemon\n");
833 case READ_STATUS_ERROR:
834 fprintf (stderr, "Error in dbus-launch reading address from bus daemon: %s\n",
840 close (bus_pid_to_launcher_pipe[READ_END]);
847 envvar = malloc (strlen ("DBUS_SESSION_BUS_ADDRESS=") + strlen (bus_address) + 1);
848 args = malloc (sizeof (char *) * ((argc-remaining_args)+2));
850 if (envvar == NULL || args == NULL)
853 args[0] = xstrdup (runprog);
856 for (i = 1; i <= (argc-remaining_args); i++)
858 size_t len = strlen (argv[remaining_args+i-1])+1;
859 args[i] = malloc (len);
862 strncpy (args[i], argv[remaining_args+i-1], len);
866 strcpy (envvar, "DBUS_SESSION_BUS_ADDRESS=");
867 strcat (envvar, bus_address);
870 execvp (runprog, args);
871 fprintf (stderr, "Couldn't exec %s: %s\n", runprog, strerror (errno));
877 printf ("setenv DBUS_SESSION_BUS_ADDRESS '%s'\n", bus_address);
880 printf ("DBUS_SESSION_BUS_ADDRESS='%s'\n", bus_address);
881 if (bourne_shell_syntax)
882 printf ("export DBUS_SESSION_BUS_ADDRESS\n");
885 printf ("set DBUS_SESSION_BUS_PID=%ld\n", (long) bus_pid);
887 printf ("DBUS_SESSION_BUS_PID=%ld\n", (long) bus_pid);
890 verbose ("dbus-launch exiting\n");
897 fprintf (stderr, "Out of memory!");