bus/main.c

   1 /* -*- mode: C; c-file-style: "gnu" -*- */
   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.0
   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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  21  *
  22  */
  23 #include "bus.h"
  24 #include <dbus/dbus-internals.h>
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <signal.h>
  29 #include <errno.h>
  30
  31 static BusContext *context;
  32
  33 static void
  34 signal_handler (int sig)
  35 {
  36   DBusError error;
  37
  38   switch (sig)
  39     {
  40     case SIGHUP:
  41       /* FIXME: We shouldn't be reloading the config in the
  42          signal handler.  We should use a pipe or something to
  43          make the reload happen in the main loop. */
  44       dbus_error_init (&error);
  45       if (!bus_context_reload_config (context, &error))
  46         {
  47           _dbus_warn ("Unable to reload configuration: %s\n",
  48                       error.message);
  49           dbus_error_free (&error);
  50           exit (1);
  51         }
  52       break;
  53
  54     case SIGTERM:
  55       _dbus_loop_quit (bus_context_get_loop (context));
  56       break;
  57     }
  58 }
  59
  60 static void
  61 usage (void)
  62 {
  63   fprintf (stderr, "dbus-daemon-1 [--version] [--session] [--system] [--config-file=FILE] [--print-address[=DESCRIPTOR]] [--print-pid[=DESCRIPTOR]] [--fork]\n");
  64   exit (1);
  65 }
  66
  67 static void
  68 version (void)
  69 {
  70   printf ("D-BUS Message Bus Daemon %s\n"
  71           "Copyright (C) 2002, 2003 Red Hat, Inc., CodeFactory AB, and others\n"
  72           "This is free software; see the source for copying conditions.\n"
  73           "There is NO warranty; not even for MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.\n",
  74           VERSION);
  75   exit (0);
  76 }
  77
  78 static void
  79 check_two_config_files (const DBusString *config_file,
  80                         const char       *extra_arg)
  81 {
  82   if (_dbus_string_get_length (config_file) > 0)
  83     {
  84       fprintf (stderr, "--%s specified but configuration file %s already requested\n",
  85                extra_arg, _dbus_string_get_const_data (config_file));
  86       exit (1);
  87     }
  88 }
  89
  90 static void
  91 check_two_addr_descriptors (const DBusString *addr_fd,
  92                             const char       *extra_arg)
  93 {
  94   if (_dbus_string_get_length (addr_fd) > 0)
  95     {
  96       fprintf (stderr, "--%s specified but printing address to %s already requested\n",
  97                extra_arg, _dbus_string_get_const_data (addr_fd));
  98       exit (1);
  99     }
 100 }
 101
 102 static void
 103 check_two_pid_descriptors (const DBusString *pid_fd,
 104                            const char       *extra_arg)
 105 {
 106   if (_dbus_string_get_length (pid_fd) > 0)
 107     {
 108       fprintf (stderr, "--%s specified but printing pid to %s already requested\n",
 109                extra_arg, _dbus_string_get_const_data (pid_fd));
 110       exit (1);
 111     }
 112 }
 113
 114 int
 115 main (int argc, char **argv)
 116 {
 117   DBusError error;
 118   DBusString config_file;
 119   DBusString addr_fd;
 120   DBusString pid_fd;
 121   const char *prev_arg;
 122   int print_addr_fd;
 123   int print_pid_fd;
 124   int i;
 125   dbus_bool_t print_address;
 126   dbus_bool_t print_pid;
 127   dbus_bool_t force_fork;
 128
 129   if (!_dbus_string_init (&config_file))
 130     return 1;
 131
 132   if (!_dbus_string_init (&addr_fd))
 133     return 1;
 134
 135   if (!_dbus_string_init (&pid_fd))
 136     return 1;
 137
 138   print_address = FALSE;
 139   print_pid = FALSE;
 140   force_fork = FALSE;
 141
 142   prev_arg = NULL;
 143   i = 1;
 144   while (i < argc)
 145     {
 146       const char *arg = argv[i];
 147
 148       if (strcmp (arg, "--help") == 0 ||
 149           strcmp (arg, "-h") == 0 ||
 150           strcmp (arg, "-?") == 0)
 151         usage ();
 152       else if (strcmp (arg, "--version") == 0)
 153         version ();
 154       else if (strcmp (arg, "--fork") == 0)
 155         force_fork = TRUE;
 156       else if (strcmp (arg, "--system") == 0)
 157         {
 158           check_two_config_files (&config_file, "system");
 159
 160           if (!_dbus_string_append (&config_file, DBUS_SYSTEM_CONFIG_FILE))
 161             exit (1);
 162         }
 163       else if (strcmp (arg, "--session") == 0)
 164         {
 165           check_two_config_files (&config_file, "session");
 166
 167           if (!_dbus_string_append (&config_file, DBUS_SESSION_CONFIG_FILE))
 168             exit (1);
 169         }
 170       else if (strstr (arg, "--config-file=") == arg)
 171         {
 172           const char *file;
 173
 174           check_two_config_files (&config_file, "config-file");
 175
 176           file = strchr (arg, '=');
 177           ++file;
 178
 179           if (!_dbus_string_append (&config_file, file))
 180             exit (1);
 181         }
 182       else if (prev_arg &&
 183                strcmp (prev_arg, "--config-file") == 0)
 184         {
 185           check_two_config_files (&config_file, "config-file");
 186
 187           if (!_dbus_string_append (&config_file, arg))
 188             exit (1);
 189         }
 190       else if (strcmp (arg, "--config-file") == 0)
 191         ; /* wait for next arg */
 192       else if (strstr (arg, "--print-address=") == arg)
 193         {
 194           const char *desc;
 195
 196           check_two_addr_descriptors (&addr_fd, "print-address");
 197
 198           desc = strchr (arg, '=');
 199           ++desc;
 200
 201           if (!_dbus_string_append (&addr_fd, desc))
 202             exit (1);
 203
 204           print_address = TRUE;
 205         }
 206       else if (prev_arg &&
 207                strcmp (prev_arg, "--print-address") == 0)
 208         {
 209           check_two_addr_descriptors (&addr_fd, "print-address");
 210
 211           if (!_dbus_string_append (&addr_fd, arg))
 212             exit (1);
 213
 214           print_address = TRUE;
 215         }
 216       else if (strcmp (arg, "--print-address") == 0)
 217         print_address = TRUE; /* and we'll get the next arg if appropriate */
 218       else if (strstr (arg, "--print-pid=") == arg)
 219         {
 220           const char *desc;
 221
 222           check_two_pid_descriptors (&pid_fd, "print-pid");
 223
 224           desc = strchr (arg, '=');
 225           ++desc;
 226
 227           if (!_dbus_string_append (&pid_fd, desc))
 228             exit (1);
 229
 230           print_pid = TRUE;
 231         }
 232       else if (prev_arg &&
 233                strcmp (prev_arg, "--print-pid") == 0)
 234         {
 235           check_two_pid_descriptors (&pid_fd, "print-pid");
 236
 237           if (!_dbus_string_append (&pid_fd, arg))
 238             exit (1);
 239
 240           print_pid = TRUE;
 241         }
 242       else if (strcmp (arg, "--print-pid") == 0)
 243         print_pid = TRUE; /* and we'll get the next arg if appropriate */
 244       else
 245         usage ();
 246
 247       prev_arg = arg;
 248
 249       ++i;
 250     }
 251
 252   if (_dbus_string_get_length (&config_file) == 0)
 253     {
 254       fprintf (stderr, "No configuration file specified.\n");
 255       usage ();
 256     }
 257
 258   print_addr_fd = -1;
 259   if (print_address)
 260     {
 261       print_addr_fd = 1; /* stdout */
 262       if (_dbus_string_get_length (&addr_fd) > 0)
 263         {
 264           long val;
 265           int end;
 266           if (!_dbus_string_parse_int (&addr_fd, 0, &val, &end) ||
 267               end != _dbus_string_get_length (&addr_fd) ||
 268               val < 0 || val > _DBUS_INT_MAX)
 269             {
 270               fprintf (stderr, "Invalid file descriptor: \"%s\"\n",
 271                        _dbus_string_get_const_data (&addr_fd));
 272               exit (1);
 273             }
 274
 275           print_addr_fd = val;
 276         }
 277     }
 278
 279   print_pid_fd = -1;
 280   if (print_pid)
 281     {
 282       print_pid_fd = 1; /* stdout */
 283       if (_dbus_string_get_length (&pid_fd) > 0)
 284         {
 285           long val;
 286           int end;
 287           if (!_dbus_string_parse_int (&pid_fd, 0, &val, &end) ||
 288               end != _dbus_string_get_length (&pid_fd) ||
 289               val < 0 || val > _DBUS_INT_MAX)
 290             {
 291               fprintf (stderr, "Invalid file descriptor: \"%s\"\n",
 292                        _dbus_string_get_const_data (&pid_fd));
 293               exit (1);
 294             }
 295
 296           print_pid_fd = val;
 297         }
 298     }
 299
 300   dbus_error_init (&error);
 301   context = bus_context_new (&config_file, force_fork,
 302                              print_addr_fd, print_pid_fd,
 303                              &error);
 304   _dbus_string_free (&config_file);
 305   if (context == NULL)
 306     {
 307       _dbus_warn ("Failed to start message bus: %s\n",
 308                   error.message);
 309       dbus_error_free (&error);
 310       exit (1);
 311     }
 312
 313   _dbus_set_signal_handler (SIGHUP, signal_handler);
 314   _dbus_set_signal_handler (SIGTERM, signal_handler);
 315
 316   _dbus_verbose ("We are on D-Bus...\n");
 317   _dbus_loop_run (bus_context_get_loop (context));
 318
 319   bus_context_shutdown (context);
 320   bus_context_unref (context);
 321
 322   return 0;
 323 }