bus/main.c

   1 /* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*- */
   2 /* main.c  main() for message bus
   3  *
   4  * Copyright (C) 2003 Red Hat, Inc.
   5  *
   6  * Licensed under the Academic Free License version 2.1
   7  *
   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.
  12  *
  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.
  17  *
  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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  21  *
  22  */
  23 #include "bus.h"
  24 #include "driver.h"
  25 #include <dbus/dbus-internals.h>
  26 #include <dbus/dbus-watch.h>
  27 #include <stdio.h>
  28 #include <stdlib.h>
  29 #include <string.h>
  30 #include <signal.h>
  31 #ifdef HAVE_ERRNO_H
  32 #include <errno.h>
  33 #endif
  34 #include "selinux.h"
  35
  36 static BusContext *context;
  37
  38 static int reload_pipe[2];
  39 #define RELOAD_READ_END 0
  40 #define RELOAD_WRITE_END 1
  41
  42 static void close_reload_pipe (void);
  43
  44 static void
  45 signal_handler (int sig)
  46 {
  47
  48   switch (sig)
  49     {
  50 #ifdef DBUS_BUS_ENABLE_DNOTIFY_ON_LINUX
  51     case SIGIO:
  52       /* explicit fall-through */
  53 #endif /* DBUS_BUS_ENABLE_DNOTIFY_ON_LINUX  */
  54 #ifdef SIGHUP
  55     case SIGHUP:
  56       {
  57         DBusString str;
  58         _dbus_string_init_const (&str, "foo");
  59         if ((reload_pipe[RELOAD_WRITE_END] > 0) &&
  60             !_dbus_write_socket (reload_pipe[RELOAD_WRITE_END], &str, 0, 1))
  61           {
  62             _dbus_warn ("Unable to write to reload pipe.\n");
  63             close_reload_pipe ();
  64           }
  65       }
  66       break;
  67 #endif
  68     case SIGTERM:
  69       _dbus_loop_quit (bus_context_get_loop (context));
  70       break;
  71     }
  72 }
  73
  74 static void
  75 usage (void)
  76 {
  77   fprintf (stderr, DBUS_DAEMON_NAME " [--version] [--session] [--system] [--config-file=FILE] [--print-address[=DESCRIPTOR]] [--print-pid[=DESCRIPTOR]] [--fork] [--nofork] [--introspect]\n");
  78   exit (1);
  79 }
  80
  81 static void
  82 version (void)
  83 {
  84   printf ("D-Bus Message Bus Daemon %s\n"
  85           "Copyright (C) 2002, 2003 Red Hat, Inc., CodeFactory AB, and others\n"
  86           "This is free software; see the source for copying conditions.\n"
  87           "There is NO warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n",
  88           DBUS_VERSION_STRING);
  89   exit (0);
  90 }
  91
  92 static void
  93 introspect (void)
  94 {
  95   DBusString xml;
  96   const char *v_STRING;
  97
  98   if (!_dbus_string_init (&xml))
  99     goto oom;
 100
 101   if (!bus_driver_generate_introspect_string (&xml))
 102     {
 103       _dbus_string_free (&xml);
 104       goto oom;
 105     }
 106
 107   v_STRING = _dbus_string_get_const_data (&xml);
 108   printf ("%s\n", v_STRING);
 109
 110   exit (0);
 111
 112  oom:
 113   _dbus_warn ("Can not introspect - Out of memory\n");
 114   exit (1);
 115 }
 116 static void
 117 check_two_config_files (const DBusString *config_file,
 118                         const char       *extra_arg)
 119 {
 120   if (_dbus_string_get_length (config_file) > 0)
 121     {
 122       fprintf (stderr, "--%s specified but configuration file %s already requested\n",
 123                extra_arg, _dbus_string_get_const_data (config_file));
 124       exit (1);
 125     }
 126 }
 127
 128 static void
 129 check_two_addr_descriptors (const DBusString *addr_fd,
 130                             const char       *extra_arg)
 131 {
 132   if (_dbus_string_get_length (addr_fd) > 0)
 133     {
 134       fprintf (stderr, "--%s specified but printing address to %s already requested\n",
 135                extra_arg, _dbus_string_get_const_data (addr_fd));
 136       exit (1);
 137     }
 138 }
 139
 140 static void
 141 check_two_pid_descriptors (const DBusString *pid_fd,
 142                            const char       *extra_arg)
 143 {
 144   if (_dbus_string_get_length (pid_fd) > 0)
 145     {
 146       fprintf (stderr, "--%s specified but printing pid to %s already requested\n",
 147                extra_arg, _dbus_string_get_const_data (pid_fd));
 148       exit (1);
 149     }
 150 }
 151
 152 static dbus_bool_t
 153 handle_reload_watch (DBusWatch    *watch,
 154                      unsigned int  flags,
 155                      void         *data)
 156 {
 157   DBusError error;
 158   DBusString str;
 159
 160   while (!_dbus_string_init (&str))
 161     _dbus_wait_for_memory ();
 162
 163   if ((reload_pipe[RELOAD_READ_END] > 0) &&
 164       _dbus_read_socket (reload_pipe[RELOAD_READ_END], &str, 1) != 1)
 165     {
 166       _dbus_warn ("Couldn't read from reload pipe.\n");
 167       close_reload_pipe ();
 168       return TRUE;
 169     }
 170   _dbus_string_free (&str);
 171
 172   /* this can only fail if we don't understand the config file
 173    * or OOM.  Either way we should just stick with the currently
 174    * loaded config.
 175    */
 176   dbus_error_init (&error);
 177   if (! bus_context_reload_config (context, &error))
 178     {
 179       _DBUS_ASSERT_ERROR_IS_SET (&error);
 180       _dbus_assert (dbus_error_has_name (&error, DBUS_ERROR_FAILED) ||
 181                     dbus_error_has_name (&error, DBUS_ERROR_NO_MEMORY));
 182       _dbus_warn ("Unable to reload configuration: %s\n",
 183                   error.message);
 184       dbus_error_free (&error);
 185     }
 186   return TRUE;
 187 }
 188
 189 static dbus_bool_t
 190 reload_watch_callback (DBusWatch    *watch,
 191                        unsigned int  condition,
 192                        void         *data)
 193 {
 194   return dbus_watch_handle (watch, condition);
 195 }
 196
 197 static void
 198 setup_reload_pipe (DBusLoop *loop)
 199 {
 200   DBusError error;
 201   DBusWatch *watch;
 202
 203   dbus_error_init (&error);
 204
 205   if (!_dbus_full_duplex_pipe (&reload_pipe[0], &reload_pipe[1],
 206                                TRUE, &error))
 207     {
 208       _dbus_warn ("Unable to create reload pipe: %s\n",
 209                   error.message);
 210       dbus_error_free (&error);
 211       exit (1);
 212     }
 213
 214   watch = _dbus_watch_new (reload_pipe[RELOAD_READ_END],
 215                            DBUS_WATCH_READABLE, TRUE,
 216                            handle_reload_watch, NULL, NULL);
 217
 218   if (watch == NULL)
 219     {
 220       _dbus_warn ("Unable to create reload watch: %s\n",
 221                   error.message);
 222       dbus_error_free (&error);
 223       exit (1);
 224     }
 225
 226   if (!_dbus_loop_add_watch (loop, watch, reload_watch_callback,
 227                              NULL, NULL))
 228     {
 229       _dbus_warn ("Unable to add reload watch to main loop: %s\n",
 230                   error.message);
 231       dbus_error_free (&error);
 232       exit (1);
 233     }
 234
 235 }
 236
 237 static void
 238 close_reload_pipe (void)
 239 {
 240     _dbus_close_socket (reload_pipe[RELOAD_READ_END], NULL);
 241     reload_pipe[RELOAD_READ_END] = -1;
 242
 243     _dbus_close_socket (reload_pipe[RELOAD_WRITE_END], NULL);
 244     reload_pipe[RELOAD_WRITE_END] = -1;
 245 }
 246
 247 int
 248 main (int argc, char **argv)
 249 {
 250   DBusError error;
 251   DBusString config_file;
 252   DBusString addr_fd;
 253   DBusString pid_fd;
 254   const char *prev_arg;
 255   DBusPipe print_addr_pipe;
 256   DBusPipe print_pid_pipe;
 257   int i;
 258   dbus_bool_t print_address;
 259   dbus_bool_t print_pid;
 260   dbus_bool_t is_session_bus;
 261   int force_fork;
 262
 263   if (!_dbus_string_init (&config_file))
 264     return 1;
 265
 266   if (!_dbus_string_init (&addr_fd))
 267     return 1;
 268
 269   if (!_dbus_string_init (&pid_fd))
 270     return 1;
 271
 272   print_address = FALSE;
 273   print_pid = FALSE;
 274   is_session_bus = FALSE;
 275   force_fork = FORK_FOLLOW_CONFIG_FILE;
 276
 277   prev_arg = NULL;
 278   i = 1;
 279   while (i < argc)
 280     {
 281       const char *arg = argv[i];
 282
 283       if (strcmp (arg, "--help") == 0 ||
 284           strcmp (arg, "-h") == 0 ||
 285           strcmp (arg, "-?") == 0)
 286         usage ();
 287       else if (strcmp (arg, "--version") == 0)
 288         version ();
 289       else if (strcmp (arg, "--introspect") == 0)
 290         introspect ();
 291       else if (strcmp (arg, "--nofork") == 0)
 292         force_fork = FORK_NEVER;
 293       else if (strcmp (arg, "--fork") == 0)
 294         force_fork = FORK_ALWAYS;
 295       else if (strcmp (arg, "--system") == 0)
 296         {
 297           check_two_config_files (&config_file, "system");
 298
 299           if (!_dbus_append_system_config_file (&config_file))
 300             exit (1);
 301         }
 302       else if (strcmp (arg, "--session") == 0)
 303         {
 304           check_two_config_files (&config_file, "session");
 305
 306           if (!_dbus_append_session_config_file (&config_file))
 307             exit (1);
 308         }
 309       else if (strstr (arg, "--config-file=") == arg)
 310         {
 311           const char *file;
 312
 313           check_two_config_files (&config_file, "config-file");
 314
 315           file = strchr (arg, '=');
 316           ++file;
 317
 318           if (!_dbus_string_append (&config_file, file))
 319             exit (1);
 320         }
 321       else if (prev_arg &&
 322                strcmp (prev_arg, "--config-file") == 0)
 323         {
 324           check_two_config_files (&config_file, "config-file");
 325
 326           if (!_dbus_string_append (&config_file, arg))
 327             exit (1);
 328         }
 329       else if (strcmp (arg, "--config-file") == 0)
 330         ; /* wait for next arg */
 331       else if (strstr (arg, "--print-address=") == arg)
 332         {
 333           const char *desc;
 334
 335           check_two_addr_descriptors (&addr_fd, "print-address");
 336
 337           desc = strchr (arg, '=');
 338           ++desc;
 339
 340           if (!_dbus_string_append (&addr_fd, desc))
 341             exit (1);
 342
 343           print_address = TRUE;
 344         }
 345       else if (prev_arg &&
 346                strcmp (prev_arg, "--print-address") == 0)
 347         {
 348           check_two_addr_descriptors (&addr_fd, "print-address");
 349
 350           if (!_dbus_string_append (&addr_fd, arg))
 351             exit (1);
 352
 353           print_address = TRUE;
 354         }
 355       else if (strcmp (arg, "--print-address") == 0)
 356         print_address = TRUE; /* and we'll get the next arg if appropriate */
 357       else if (strstr (arg, "--print-pid=") == arg)
 358         {
 359           const char *desc;
 360
 361           check_two_pid_descriptors (&pid_fd, "print-pid");
 362
 363           desc = strchr (arg, '=');
 364           ++desc;
 365
 366           if (!_dbus_string_append (&pid_fd, desc))
 367             exit (1);
 368
 369           print_pid = TRUE;
 370         }
 371       else if (prev_arg &&
 372                strcmp (prev_arg, "--print-pid") == 0)
 373         {
 374           check_two_pid_descriptors (&pid_fd, "print-pid");
 375
 376           if (!_dbus_string_append (&pid_fd, arg))
 377             exit (1);
 378
 379           print_pid = TRUE;
 380         }
 381       else if (strcmp (arg, "--print-pid") == 0)
 382         print_pid = TRUE; /* and we'll get the next arg if appropriate */
 383       else
 384         usage ();
 385
 386       prev_arg = arg;
 387
 388       ++i;
 389     }
 390
 391   if (_dbus_string_get_length (&config_file) == 0)
 392     {
 393       fprintf (stderr, "No configuration file specified.\n");
 394       usage ();
 395     }
 396
 397   _dbus_pipe_invalidate (&print_addr_pipe);
 398   if (print_address)
 399     {
 400       _dbus_pipe_init_stdout (&print_addr_pipe);
 401       if (_dbus_string_get_length (&addr_fd) > 0)
 402         {
 403           long val;
 404           int end;
 405           if (!_dbus_string_parse_int (&addr_fd, 0, &val, &end) ||
 406               end != _dbus_string_get_length (&addr_fd) ||
 407               val < 0 || val > _DBUS_INT_MAX)
 408             {
 409               fprintf (stderr, "Invalid file descriptor: \"%s\"\n",
 410                        _dbus_string_get_const_data (&addr_fd));
 411               exit (1);
 412             }
 413
 414           _dbus_pipe_init (&print_addr_pipe, val);
 415         }
 416     }
 417   _dbus_string_free (&addr_fd);
 418
 419   _dbus_pipe_invalidate (&print_pid_pipe);
 420   if (print_pid)
 421     {
 422       _dbus_pipe_init_stdout (&print_pid_pipe);
 423       if (_dbus_string_get_length (&pid_fd) > 0)
 424         {
 425           long val;
 426           int end;
 427           if (!_dbus_string_parse_int (&pid_fd, 0, &val, &end) ||
 428               end != _dbus_string_get_length (&pid_fd) ||
 429               val < 0 || val > _DBUS_INT_MAX)
 430             {
 431               fprintf (stderr, "Invalid file descriptor: \"%s\"\n",
 432                        _dbus_string_get_const_data (&pid_fd));
 433               exit (1);
 434             }
 435
 436           _dbus_pipe_init (&print_pid_pipe, val);
 437         }
 438     }
 439   _dbus_string_free (&pid_fd);
 440
 441   if (!bus_selinux_pre_init ())
 442     {
 443       _dbus_warn ("SELinux pre-initialization failed\n");
 444       exit (1);
 445     }
 446
 447   dbus_error_init (&error);
 448   context = bus_context_new (&config_file, force_fork,
 449                              &print_addr_pipe, &print_pid_pipe,
 450                              &error);
 451   _dbus_string_free (&config_file);
 452   if (context == NULL)
 453     {
 454       _dbus_warn ("Failed to start message bus: %s\n",
 455                   error.message);
 456       dbus_error_free (&error);
 457       exit (1);
 458     }
 459
 460   is_session_bus = bus_context_get_type(context) != NULL
 461       && strcmp(bus_context_get_type(context),"session") == 0;
 462
 463   if (is_session_bus)
 464     _dbus_daemon_publish_session_bus_address (bus_context_get_address (context));
 465
 466   /* bus_context_new() closes the print_addr_pipe and
 467    * print_pid_pipe
 468    */
 469
 470   setup_reload_pipe (bus_context_get_loop (context));
 471
 472 #ifdef SIGHUP
 473   _dbus_set_signal_handler (SIGHUP, signal_handler);
 474 #endif
 475   _dbus_set_signal_handler (SIGTERM, signal_handler);
 476 #ifdef DBUS_BUS_ENABLE_DNOTIFY_ON_LINUX
 477   _dbus_set_signal_handler (SIGIO, signal_handler);
 478 #endif /* DBUS_BUS_ENABLE_DNOTIFY_ON_LINUX */
 479
 480   _dbus_verbose ("We are on D-Bus...\n");
 481   _dbus_loop_run (bus_context_get_loop (context));
 482
 483   bus_context_shutdown (context);
 484   bus_context_unref (context);
 485   bus_selinux_shutdown ();
 486
 487   if (is_session_bus)
 488     _dbus_daemon_unpublish_session_bus_address ();
 489
 490   return 0;
 491 }