src/main.c

   1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
   2
   3 /***
   4   This file is part of systemd.
   5
   6   Copyright 2010 Lennart Poettering
   7
   8   systemd is free software; you can redistribute it and/or modify it
   9   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   systemd is distributed in the hope that it will be useful, but
  14   WITHOUT ANY WARRANTY; without even the implied warranty of
  15   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16   General Public License for more details.
  17
  18   You should have received a copy of the GNU General Public License
  19   along with systemd; If not, see <http://www.gnu.org/licenses/>.
  20 ***/
  21
  22 #include <dbus/dbus.h>
  23
  24 #include <stdio.h>
  25 #include <errno.h>
  26 #include <string.h>
  27 #include <unistd.h>
  28 #include <sys/types.h>
  29 #include <sys/stat.h>
  30 #include <getopt.h>
  31 #include <signal.h>
  32 #include <sys/wait.h>
  33 #include <fcntl.h>
  34
  35 #include "manager.h"
  36 #include "log.h"
  37 #include "mount-setup.h"
  38 #include "hostname-setup.h"
  39 #include "loopback-setup.h"
  40 #include "kmod-setup.h"
  41 #include "locale-setup.h"
  42 #include "selinux-setup.h"
  43 #include "load-fragment.h"
  44 #include "fdset.h"
  45 #include "special.h"
  46 #include "conf-parser.h"
  47 #include "bus-errors.h"
  48 #include "missing.h"
  49 #include "label.h"
  50 #include "build.h"
  51
  52 static enum {
  53         ACTION_RUN,
  54         ACTION_HELP,
  55         ACTION_TEST,
  56         ACTION_DUMP_CONFIGURATION_ITEMS,
  57         ACTION_DONE
  58 } arg_action = ACTION_RUN;
  59
  60 static char *arg_default_unit = NULL;
  61 static ManagerRunningAs arg_running_as = _MANAGER_RUNNING_AS_INVALID;
  62
  63 static bool arg_dump_core = true;
  64 static bool arg_crash_shell = false;
  65 static int arg_crash_chvt = -1;
  66 static bool arg_confirm_spawn = false;
  67 static bool arg_show_status = true;
  68 #ifdef HAVE_SYSV_COMPAT
  69 static bool arg_sysv_console = true;
  70 #endif
  71 static bool arg_mount_auto = true;
  72 static bool arg_swap_auto = true;
  73 static char *arg_console = NULL;
  74
  75 static FILE* serialization = NULL;
  76
  77 static void nop_handler(int sig) {
  78 }
  79
  80 _noreturn_ static void crash(int sig) {
  81
  82         if (!arg_dump_core)
  83                 log_error("Caught <%s>, not dumping core.", signal_to_string(sig));
  84         else {
  85                 struct sigaction sa;
  86                 pid_t pid;
  87
  88                 /* We want to wait for the core process, hence let's enable SIGCHLD */
  89                 zero(sa);
  90                 sa.sa_handler = nop_handler;
  91                 sa.sa_flags = SA_NOCLDSTOP|SA_RESTART;
  92                 assert_se(sigaction(SIGCHLD, &sa, NULL) == 0);
  93
  94                 if ((pid = fork()) < 0)
  95                         log_error("Caught <%s>, cannot fork for core dump: %s", signal_to_string(sig), strerror(errno));
  96
  97                 else if (pid == 0) {
  98                         struct rlimit rl;
  99
 100                         /* Enable default signal handler for core dump */
 101                         zero(sa);
 102                         sa.sa_handler = SIG_DFL;
 103                         assert_se(sigaction(sig, &sa, NULL) == 0);
 104
 105                         /* Don't limit the core dump size */
 106                         zero(rl);
 107                         rl.rlim_cur = RLIM_INFINITY;
 108                         rl.rlim_max = RLIM_INFINITY;
 109                         setrlimit(RLIMIT_CORE, &rl);
 110
 111                         /* Just to be sure... */
 112                         assert_se(chdir("/") == 0);
 113
 114                         /* Raise the signal again */
 115                         raise(sig);
 116
 117                         assert_not_reached("We shouldn't be here...");
 118                         _exit(1);
 119
 120                 } else {
 121                         siginfo_t status;
 122                         int r;
 123
 124                         /* Order things nicely. */
 125                         if ((r = wait_for_terminate(pid, &status)) < 0)
 126                                 log_error("Caught <%s>, waitpid() failed: %s", signal_to_string(sig), strerror(-r));
 127                         else if (status.si_code != CLD_DUMPED)
 128                                 log_error("Caught <%s>, core dump failed.", signal_to_string(sig));
 129                         else
 130                                 log_error("Caught <%s>, dumped core as pid %lu.", signal_to_string(sig), (unsigned long) pid);
 131                 }
 132         }
 133
 134         if (arg_crash_chvt)
 135                 chvt(arg_crash_chvt);
 136
 137         if (arg_crash_shell) {
 138                 struct sigaction sa;
 139                 pid_t pid;
 140
 141                 log_info("Executing crash shell in 10s...");
 142                 sleep(10);
 143
 144                 /* Let the kernel reap children for us */
 145                 zero(sa);
 146                 sa.sa_handler = SIG_IGN;
 147                 sa.sa_flags = SA_NOCLDSTOP|SA_NOCLDWAIT|SA_RESTART;
 148                 assert_se(sigaction(SIGCHLD, &sa, NULL) == 0);
 149
 150                 if ((pid = fork()) < 0)
 151                         log_error("Failed to fork off crash shell: %s", strerror(errno));
 152                 else if (pid == 0) {
 153                         int fd, r;
 154
 155                         if ((fd = acquire_terminal("/dev/console", false, true, true)) < 0)
 156                                 log_error("Failed to acquire terminal: %s", strerror(-fd));
 157                         else if ((r = make_stdio(fd)) < 0)
 158                                 log_error("Failed to duplicate terminal fd: %s", strerror(-r));
 159
 160                         execl("/bin/sh", "/bin/sh", NULL);
 161
 162                         log_error("execl() failed: %s", strerror(errno));
 163                         _exit(1);
 164                 }
 165
 166                 log_info("Successfully spawned crash shall as pid %lu.", (unsigned long) pid);
 167         }
 168
 169         log_info("Freezing execution.");
 170         freeze();
 171 }
 172
 173 static void install_crash_handler(void) {
 174         struct sigaction sa;
 175
 176         zero(sa);
 177
 178         sa.sa_handler = crash;
 179         sa.sa_flags = SA_NODEFER;
 180
 181         sigaction_many(&sa, SIGNALS_CRASH_HANDLER, -1);
 182 }
 183
 184 static int console_setup(bool do_reset) {
 185         int tty_fd, r;
 186
 187         /* If we are init, we connect stdin/stdout/stderr to /dev/null
 188          * and make sure we don't have a controlling tty. */
 189
 190         release_terminal();
 191
 192         if (!do_reset)
 193                 return 0;
 194
 195         if ((tty_fd = open_terminal("/dev/console", O_WRONLY|O_NOCTTY|O_CLOEXEC)) < 0) {
 196                 log_error("Failed to open /dev/console: %s", strerror(-tty_fd));
 197                 return -tty_fd;
 198         }
 199
 200         if ((r = reset_terminal(tty_fd)) < 0)
 201                 log_error("Failed to reset /dev/console: %s", strerror(-r));
 202
 203         close_nointr_nofail(tty_fd);
 204         return r;
 205 }
 206
 207 static int set_default_unit(const char *u) {
 208         char *c;
 209
 210         assert(u);
 211
 212         if (!(c = strdup(u)))
 213                 return -ENOMEM;
 214
 215         free(arg_default_unit);
 216         arg_default_unit = c;
 217         return 0;
 218 }
 219
 220 static int parse_proc_cmdline_word(const char *word) {
 221
 222         static const char * const rlmap[] = {
 223                 "emergency", SPECIAL_EMERGENCY_TARGET,
 224                 "single",    SPECIAL_RESCUE_TARGET,
 225                 "-s",        SPECIAL_RESCUE_TARGET,
 226                 "s",         SPECIAL_RESCUE_TARGET,
 227                 "S",         SPECIAL_RESCUE_TARGET,
 228                 "1",         SPECIAL_RESCUE_TARGET,
 229                 "2",         SPECIAL_RUNLEVEL2_TARGET,
 230                 "3",         SPECIAL_RUNLEVEL3_TARGET,
 231                 "4",         SPECIAL_RUNLEVEL4_TARGET,
 232                 "5",         SPECIAL_RUNLEVEL5_TARGET,
 233         };
 234
 235         assert(word);
 236
 237         if (startswith(word, "systemd.unit="))
 238                 return set_default_unit(word + 13);
 239
 240         else if (startswith(word, "systemd.log_target=")) {
 241
 242                 if (log_set_target_from_string(word + 19) < 0)
 243                         log_warning("Failed to parse log target %s. Ignoring.", word + 19);
 244
 245         } else if (startswith(word, "systemd.log_level=")) {
 246
 247                 if (log_set_max_level_from_string(word + 18) < 0)
 248                         log_warning("Failed to parse log level %s. Ignoring.", word + 18);
 249
 250         } else if (startswith(word, "systemd.log_color=")) {
 251
 252                 if (log_show_color_from_string(word + 18) < 0)
 253                         log_warning("Failed to parse log color setting %s. Ignoring.", word + 18);
 254
 255         } else if (startswith(word, "systemd.log_location=")) {
 256
 257                 if (log_show_location_from_string(word + 21) < 0)
 258                         log_warning("Failed to parse log location setting %s. Ignoring.", word + 21);
 259
 260         } else if (startswith(word, "systemd.dump_core=")) {
 261                 int r;
 262
 263                 if ((r = parse_boolean(word + 18)) < 0)
 264                         log_warning("Failed to parse dump core switch %s, Ignoring.", word + 18);
 265                 else
 266                         arg_dump_core = r;
 267
 268         } else if (startswith(word, "systemd.crash_shell=")) {
 269                 int r;
 270
 271                 if ((r = parse_boolean(word + 20)) < 0)
 272                         log_warning("Failed to parse crash shell switch %s, Ignoring.", word + 20);
 273                 else
 274                         arg_crash_shell = r;
 275
 276         } else if (startswith(word, "systemd.confirm_spawn=")) {
 277                 int r;
 278
 279                 if ((r = parse_boolean(word + 22)) < 0)
 280                         log_warning("Failed to parse confirm spawn switch %s, Ignoring.", word + 22);
 281                 else
 282                         arg_confirm_spawn = r;
 283
 284         } else if (startswith(word, "systemd.crash_chvt=")) {
 285                 int k;
 286
 287                 if (safe_atoi(word + 19, &k) < 0)
 288                         log_warning("Failed to parse crash chvt switch %s, Ignoring.", word + 19);
 289                 else
 290                         arg_crash_chvt = k;
 291
 292         } else if (startswith(word, "systemd.show_status=")) {
 293                 int r;
 294
 295                 if ((r = parse_boolean(word + 20)) < 0)
 296                         log_warning("Failed to parse show status switch %s, Ignoring.", word + 20);
 297                 else
 298                         arg_show_status = r;
 299 #ifdef HAVE_SYSV_COMPAT
 300         } else if (startswith(word, "systemd.sysv_console=")) {
 301                 int r;
 302
 303                 if ((r = parse_boolean(word + 21)) < 0)
 304                         log_warning("Failed to parse SysV console switch %s, Ignoring.", word + 20);
 305                 else
 306                         arg_sysv_console = r;
 307 #endif
 308
 309         } else if (startswith(word, "systemd.")) {
 310
 311                 log_warning("Unknown kernel switch %s. Ignoring.", word);
 312
 313                 log_info("Supported kernel switches:\n"
 314                          "systemd.unit=UNIT                        Default unit to start\n"
 315                          "systemd.dump_core=0|1                    Dump core on crash\n"
 316                          "systemd.crash_shell=0|1                  Run shell on crash\n"
 317                          "systemd.crash_chvt=N                     Change to VT #N on crash\n"
 318                          "systemd.confirm_spawn=0|1                Confirm every process spawn\n"
 319                          "systemd.show_status=0|1                  Show status updates on the console during bootup\n"
 320 #ifdef HAVE_SYSV_COMPAT
 321                          "systemd.sysv_console=0|1                 Connect output of SysV scripts to console\n"
 322 #endif
 323                          "systemd.log_target=console|kmsg|syslog|syslog-or-kmsg|null\n"
 324                          "                                         Log target\n"
 325                          "systemd.log_level=LEVEL                  Log level\n"
 326                          "systemd.log_color=0|1                    Highlight important log messages\n"
 327                          "systemd.log_location=0|1                 Include code location in log messages\n");
 328
 329         } else if (startswith(word, "console=")) {
 330                 const char *k;
 331                 size_t l;
 332                 char *w = NULL;
 333
 334                 k = word + 8;
 335                 l = strcspn(k, ",");
 336
 337                 /* Ignore the console setting if set to a VT */
 338                 if (l < 4 ||
 339                     !startswith(k, "tty") ||
 340                     k[3+strspn(k+3, "0123456789")] != 0) {
 341
 342                         if (!(w = strndup(k, l)))
 343                                 return -ENOMEM;
 344                 }
 345
 346                 free(arg_console);
 347                 arg_console = w;
 348
 349         } else if (streq(word, "quiet")) {
 350                 arg_show_status = false;
 351 #ifdef HAVE_SYSV_COMPAT
 352                 arg_sysv_console = false;
 353 #endif
 354         } else {
 355                 unsigned i;
 356
 357                 /* SysV compatibility */
 358                 for (i = 0; i < ELEMENTSOF(rlmap); i += 2)
 359                         if (streq(word, rlmap[i]))
 360                                 return set_default_unit(rlmap[i+1]);
 361         }
 362
 363         return 0;
 364 }
 365
 366 static int config_parse_level(
 367                 const char *filename,
 368                 unsigned line,
 369                 const char *section,
 370                 const char *lvalue,
 371                 const char *rvalue,
 372                 void *data,
 373                 void *userdata) {
 374
 375         assert(filename);
 376         assert(lvalue);
 377         assert(rvalue);
 378
 379         log_set_max_level_from_string(rvalue);
 380         return 0;
 381 }
 382
 383 static int config_parse_target(
 384                 const char *filename,
 385                 unsigned line,
 386                 const char *section,
 387                 const char *lvalue,
 388                 const char *rvalue,
 389                 void *data,
 390                 void *userdata) {
 391
 392         assert(filename);
 393         assert(lvalue);
 394         assert(rvalue);
 395
 396         log_set_target_from_string(rvalue);
 397         return 0;
 398 }
 399
 400 static int config_parse_color(
 401                 const char *filename,
 402                 unsigned line,
 403                 const char *section,
 404                 const char *lvalue,
 405                 const char *rvalue,
 406                 void *data,
 407                 void *userdata) {
 408
 409         assert(filename);
 410         assert(lvalue);
 411         assert(rvalue);
 412
 413         log_show_color_from_string(rvalue);
 414         return 0;
 415 }
 416
 417 static int config_parse_location(
 418                 const char *filename,
 419                 unsigned line,
 420                 const char *section,
 421                 const char *lvalue,
 422                 const char *rvalue,
 423                 void *data,
 424                 void *userdata) {
 425
 426         assert(filename);
 427         assert(lvalue);
 428         assert(rvalue);
 429
 430         log_show_location_from_string(rvalue);
 431         return 0;
 432 }
 433
 434 static int config_parse_cpu_affinity(
 435                 const char *filename,
 436                 unsigned line,
 437                 const char *section,
 438                 const char *lvalue,
 439                 const char *rvalue,
 440                 void *data,
 441                 void *userdata) {
 442
 443         char *w;
 444         size_t l;
 445         char *state;
 446         cpu_set_t *c = NULL;
 447         unsigned ncpus = 0;
 448
 449         assert(filename);
 450         assert(lvalue);
 451         assert(rvalue);
 452
 453         FOREACH_WORD_QUOTED(w, l, rvalue, state) {
 454                 char *t;
 455                 int r;
 456                 unsigned cpu;
 457
 458                 if (!(t = strndup(w, l)))
 459                         return -ENOMEM;
 460
 461                 r = safe_atou(t, &cpu);
 462                 free(t);
 463
 464                 if (!c)
 465                         if (!(c = cpu_set_malloc(&ncpus)))
 466                                 return -ENOMEM;
 467
 468                 if (r < 0 || cpu >= ncpus) {
 469                         log_error("[%s:%u] Failed to parse CPU affinity: %s", filename, line, rvalue);
 470                         CPU_FREE(c);
 471                         return -EBADMSG;
 472                 }
 473
 474                 CPU_SET_S(cpu, CPU_ALLOC_SIZE(ncpus), c);
 475         }
 476
 477         if (c) {
 478                 if (sched_setaffinity(0, CPU_ALLOC_SIZE(ncpus), c) < 0)
 479                         log_warning("Failed to set CPU affinity: %m");
 480
 481                 CPU_FREE(c);
 482         }
 483
 484         return 0;
 485 }
 486
 487 static int parse_config_file(void) {
 488
 489         const ConfigItem items[] = {
 490                 { "LogLevel",    config_parse_level,        NULL,               "Manager" },
 491                 { "LogTarget",   config_parse_target,       NULL,               "Manager" },
 492                 { "LogColor",    config_parse_color,        NULL,               "Manager" },
 493                 { "LogLocation", config_parse_location,     NULL,               "Manager" },
 494                 { "DumpCore",    config_parse_bool,         &arg_dump_core,     "Manager" },
 495                 { "CrashShell",  config_parse_bool,         &arg_crash_shell,   "Manager" },
 496                 { "ShowStatus",  config_parse_bool,         &arg_show_status,   "Manager" },
 497 #ifdef HAVE_SYSV_COMPAT
 498                 { "SysVConsole", config_parse_bool,         &arg_sysv_console,  "Manager" },
 499 #endif
 500                 { "CrashChVT",   config_parse_int,          &arg_crash_chvt,    "Manager" },
 501                 { "CPUAffinity", config_parse_cpu_affinity, NULL,               "Manager" },
 502                 { "MountAuto",   config_parse_bool,         &arg_mount_auto,    "Manager" },
 503                 { "SwapAuto",    config_parse_bool,         &arg_swap_auto,     "Manager" },
 504                 { NULL, NULL, NULL, NULL }
 505         };
 506
 507         static const char * const sections[] = {
 508                 "Manager",
 509                 NULL
 510         };
 511
 512         FILE *f;
 513         const char *fn;
 514         int r;
 515
 516         fn = arg_running_as == MANAGER_SYSTEM ? SYSTEM_CONFIG_FILE : SESSION_CONFIG_FILE;
 517
 518         if (!(f = fopen(fn, "re"))) {
 519                 if (errno == ENOENT)
 520                         return 0;
 521
 522                 log_warning("Failed to open configuration file '%s': %m", fn);
 523                 return 0;
 524         }
 525
 526         if ((r = config_parse(fn, f, sections, items, false, NULL)) < 0)
 527                 log_warning("Failed to parse configuration file: %s", strerror(-r));
 528
 529         fclose(f);
 530
 531         return 0;
 532 }
 533
 534 static int parse_proc_cmdline(void) {
 535         char *line, *w, *state;
 536         int r;
 537         size_t l;
 538
 539         if ((r = read_one_line_file("/proc/cmdline", &line)) < 0) {
 540                 log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r));
 541                 return 0;
 542         }
 543
 544         FOREACH_WORD_QUOTED(w, l, line, state) {
 545                 char *word;
 546
 547                 if (!(word = strndup(w, l))) {
 548                         r = -ENOMEM;
 549                         goto finish;
 550                 }
 551
 552                 r = parse_proc_cmdline_word(word);
 553                 free(word);
 554
 555                 if (r < 0)
 556                         goto finish;
 557         }
 558
 559         r = 0;
 560
 561 finish:
 562         free(line);
 563         return r;
 564 }
 565
 566 static int parse_argv(int argc, char *argv[]) {
 567
 568         enum {
 569                 ARG_LOG_LEVEL = 0x100,
 570                 ARG_LOG_TARGET,
 571                 ARG_LOG_COLOR,
 572                 ARG_LOG_LOCATION,
 573                 ARG_UNIT,
 574                 ARG_SYSTEM,
 575                 ARG_SESSION,
 576                 ARG_TEST,
 577                 ARG_DUMP_CONFIGURATION_ITEMS,
 578                 ARG_DUMP_CORE,
 579                 ARG_CRASH_SHELL,
 580                 ARG_CONFIRM_SPAWN,
 581                 ARG_SHOW_STATUS,
 582                 ARG_SYSV_CONSOLE,
 583                 ARG_DESERIALIZE,
 584                 ARG_INTROSPECT
 585         };
 586
 587         static const struct option options[] = {
 588                 { "log-level",                required_argument, NULL, ARG_LOG_LEVEL                },
 589                 { "log-target",               required_argument, NULL, ARG_LOG_TARGET               },
 590                 { "log-color",                optional_argument, NULL, ARG_LOG_COLOR                },
 591                 { "log-location",             optional_argument, NULL, ARG_LOG_LOCATION             },
 592                 { "unit",                     required_argument, NULL, ARG_UNIT                     },
 593                 { "system",                   no_argument,       NULL, ARG_SYSTEM                   },
 594                 { "session",                  no_argument,       NULL, ARG_SESSION                  },
 595                 { "test",                     no_argument,       NULL, ARG_TEST                     },
 596                 { "help",                     no_argument,       NULL, 'h'                          },
 597                 { "dump-configuration-items", no_argument,       NULL, ARG_DUMP_CONFIGURATION_ITEMS },
 598                 { "dump-core",                no_argument,       NULL, ARG_DUMP_CORE                },
 599                 { "crash-shell",              no_argument,       NULL, ARG_CRASH_SHELL              },
 600                 { "confirm-spawn",            no_argument,       NULL, ARG_CONFIRM_SPAWN            },
 601                 { "show-status",              optional_argument, NULL, ARG_SHOW_STATUS              },
 602 #ifdef HAVE_SYSV_COMPAT
 603                 { "sysv-console",             optional_argument, NULL, ARG_SYSV_CONSOLE             },
 604 #endif
 605                 { "deserialize",              required_argument, NULL, ARG_DESERIALIZE              },
 606                 { "introspect",               optional_argument, NULL, ARG_INTROSPECT               },
 607                 { NULL,                       0,                 NULL, 0                            }
 608         };
 609
 610         int c, r;
 611
 612         assert(argc >= 1);
 613         assert(argv);
 614
 615         while ((c = getopt_long(argc, argv, "hD", options, NULL)) >= 0)
 616
 617                 switch (c) {
 618
 619                 case ARG_LOG_LEVEL:
 620                         if ((r = log_set_max_level_from_string(optarg)) < 0) {
 621                                 log_error("Failed to parse log level %s.", optarg);
 622                                 return r;
 623                         }
 624
 625                         break;
 626
 627                 case ARG_LOG_TARGET:
 628
 629                         if ((r = log_set_target_from_string(optarg)) < 0) {
 630                                 log_error("Failed to parse log target %s.", optarg);
 631                                 return r;
 632                         }
 633
 634                         break;
 635
 636                 case ARG_LOG_COLOR:
 637
 638                         if (optarg) {
 639                                 if ((r = log_show_color_from_string(optarg)) < 0) {
 640                                         log_error("Failed to parse log color setting %s.", optarg);
 641                                         return r;
 642                                 }
 643                         } else
 644                                 log_show_color(true);
 645
 646                         break;
 647
 648                 case ARG_LOG_LOCATION:
 649
 650                         if (optarg) {
 651                                 if ((r = log_show_location_from_string(optarg)) < 0) {
 652                                         log_error("Failed to parse log location setting %s.", optarg);
 653                                         return r;
 654                                 }
 655                         } else
 656                                 log_show_location(true);
 657
 658                         break;
 659
 660                 case ARG_UNIT:
 661
 662                         if ((r = set_default_unit(optarg)) < 0) {
 663                                 log_error("Failed to set default unit %s: %s", optarg, strerror(-r));
 664                                 return r;
 665                         }
 666
 667                         break;
 668
 669                 case ARG_SYSTEM:
 670                         arg_running_as = MANAGER_SYSTEM;
 671                         break;
 672
 673                 case ARG_SESSION:
 674                         arg_running_as = MANAGER_SESSION;
 675                         break;
 676
 677                 case ARG_TEST:
 678                         arg_action = ACTION_TEST;
 679                         break;
 680
 681                 case ARG_DUMP_CONFIGURATION_ITEMS:
 682                         arg_action = ACTION_DUMP_CONFIGURATION_ITEMS;
 683                         break;
 684
 685                 case ARG_DUMP_CORE:
 686                         arg_dump_core = true;
 687                         break;
 688
 689                 case ARG_CRASH_SHELL:
 690                         arg_crash_shell = true;
 691                         break;
 692
 693                 case ARG_CONFIRM_SPAWN:
 694                         arg_confirm_spawn = true;
 695                         break;
 696
 697                 case ARG_SHOW_STATUS:
 698
 699                         if (optarg) {
 700                                 if ((r = parse_boolean(optarg)) < 0) {
 701                                         log_error("Failed to show status boolean %s.", optarg);
 702                                         return r;
 703                                 }
 704                                 arg_show_status = r;
 705                         } else
 706                                 arg_show_status = true;
 707                         break;
 708 #ifdef HAVE_SYSV_COMPAT
 709                 case ARG_SYSV_CONSOLE:
 710
 711                         if (optarg) {
 712                                 if ((r = parse_boolean(optarg)) < 0) {
 713                                         log_error("Failed to SysV console boolean %s.", optarg);
 714                                         return r;
 715                                 }
 716                                 arg_sysv_console = r;
 717                         } else
 718                                 arg_sysv_console = true;
 719                         break;
 720 #endif
 721
 722                 case ARG_DESERIALIZE: {
 723                         int fd;
 724                         FILE *f;
 725
 726                         if ((r = safe_atoi(optarg, &fd)) < 0 || fd < 0) {
 727                                 log_error("Failed to parse deserialize option %s.", optarg);
 728                                 return r;
 729                         }
 730
 731                         if (!(f = fdopen(fd, "r"))) {
 732                                 log_error("Failed to open serialization fd: %m");
 733                                 return r;
 734                         }
 735
 736                         if (serialization)
 737                                 fclose(serialization);
 738
 739                         serialization = f;
 740
 741                         break;
 742                 }
 743
 744                 case ARG_INTROSPECT: {
 745                         const char * const * i = NULL;
 746
 747                         for (i = bus_interface_table; *i; i += 2)
 748                                 if (!optarg || streq(i[0], optarg)) {
 749                                         fputs(DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE
 750                                               "<node>\n", stdout);
 751                                         fputs(i[1], stdout);
 752                                         fputs("</node>\n", stdout);
 753
 754                                         if (optarg)
 755                                                 break;
 756                                 }
 757
 758                         if (!i[0] && optarg)
 759                                 log_error("Unknown interface %s.", optarg);
 760
 761                         arg_action = ACTION_DONE;
 762                         break;
 763                 }
 764
 765                 case 'h':
 766                         arg_action = ACTION_HELP;
 767                         break;
 768
 769                 case 'D':
 770                         log_set_max_level(LOG_DEBUG);
 771                         break;
 772
 773                 case '?':
 774                         return -EINVAL;
 775
 776                 default:
 777                         log_error("Unknown option code %c", c);
 778                         return -EINVAL;
 779                 }
 780
 781         /* PID 1 will get the kernel arguments as parameters, which we
 782          * ignore and unconditionally read from
 783          * /proc/cmdline. However, we need to ignore those arguments
 784          * here. */
 785         if (arg_running_as != MANAGER_SYSTEM && optind < argc) {
 786                 log_error("Excess arguments.");
 787                 return -EINVAL;
 788         }
 789
 790         return 0;
 791 }
 792
 793 static int help(void) {
 794
 795         printf("%s [OPTIONS...]\n\n"
 796                "Starts up and maintains the system or a session.\n\n"
 797                "  -h --help                      Show this help\n"
 798                "     --test                      Determine startup sequence, dump it and exit\n"
 799                "     --dump-configuration-items  Dump understood unit configuration items\n"
 800                "     --introspect[=INTERFACE]    Extract D-Bus interface data\n"
 801                "     --unit=UNIT                 Set default unit\n"
 802                "     --system                    Run a system instance, even if PID != 1\n"
 803                "     --session                   Run a session instance\n"
 804                "     --dump-core                 Dump core on crash\n"
 805                "     --crash-shell               Run shell on crash\n"
 806                "     --confirm-spawn             Ask for confirmation when spawning processes\n"
 807                "     --show-status[=0|1]         Show status updates on the console during bootup\n"
 808 #ifdef HAVE_SYSV_COMPAT
 809                "     --sysv-console[=0|1]        Connect output of SysV scripts to console\n"
 810 #endif
 811                "     --log-target=TARGET         Set log target (console, syslog, kmsg, syslog-or-kmsg, null)\n"
 812                "     --log-level=LEVEL           Set log level (debug, info, notice, warning, err, crit, alert, emerg)\n"
 813                "     --log-color[=0|1]           Highlight important log messages\n"
 814                "     --log-location[=0|1]        Include code location in log messages\n",
 815                program_invocation_short_name);
 816
 817         return 0;
 818 }
 819
 820 static int prepare_reexecute(Manager *m, FILE **_f, FDSet **_fds) {
 821         FILE *f = NULL;
 822         FDSet *fds = NULL;
 823         int r;
 824
 825         assert(m);
 826         assert(_f);
 827         assert(_fds);
 828
 829         if ((r = manager_open_serialization(m, &f)) < 0) {
 830                 log_error("Failed to create serialization faile: %s", strerror(-r));
 831                 goto fail;
 832         }
 833
 834         if (!(fds = fdset_new())) {
 835                 r = -ENOMEM;
 836                 log_error("Failed to allocate fd set: %s", strerror(-r));
 837                 goto fail;
 838         }
 839
 840         if ((r = manager_serialize(m, f, fds)) < 0) {
 841                 log_error("Failed to serialize state: %s", strerror(-r));
 842                 goto fail;
 843         }
 844
 845         if (fseeko(f, 0, SEEK_SET) < 0) {
 846                 log_error("Failed to rewind serialization fd: %m");
 847                 goto fail;
 848         }
 849
 850         if ((r = fd_cloexec(fileno(f), false)) < 0) {
 851                 log_error("Failed to disable O_CLOEXEC for serialization: %s", strerror(-r));
 852                 goto fail;
 853         }
 854
 855         if ((r = fdset_cloexec(fds, false)) < 0) {
 856                 log_error("Failed to disable O_CLOEXEC for serialization fds: %s", strerror(-r));
 857                 goto fail;
 858         }
 859
 860         *_f = f;
 861         *_fds = fds;
 862
 863         return 0;
 864
 865 fail:
 866         fdset_free(fds);
 867
 868         if (f)
 869                 fclose(f);
 870
 871         return r;
 872 }
 873
 874 int main(int argc, char *argv[]) {
 875         Manager *m = NULL;
 876         int r, retval = EXIT_FAILURE;
 877         FDSet *fds = NULL;
 878         bool reexecute = false;
 879         const char *shutdown_verb = NULL;
 880
 881         if (getpid() != 1 && strstr(program_invocation_short_name, "init")) {
 882                 /* This is compatbility support for SysV, where
 883                  * calling init as a user is identical to telinit. */
 884
 885                 errno = -ENOENT;
 886                 execv(SYSTEMCTL_BINARY_PATH, argv);
 887                 log_error("Failed to exec " SYSTEMCTL_BINARY_PATH ": %m");
 888                 return 1;
 889         }
 890
 891         log_show_color(isatty(STDERR_FILENO) > 0);
 892         log_show_location(false);
 893         log_set_max_level(LOG_INFO);
 894
 895         if (getpid() == 1) {
 896                 arg_running_as = MANAGER_SYSTEM;
 897                 log_set_target(LOG_TARGET_SYSLOG_OR_KMSG);
 898
 899                 /* This might actually not return, but cause a
 900                  * reexecution */
 901                 if (selinux_setup(argv) < 0)
 902                         goto finish;
 903
 904                 if (label_init() < 0)
 905                         goto finish;
 906         } else {
 907                 arg_running_as = MANAGER_SESSION;
 908                 log_set_target(LOG_TARGET_CONSOLE);
 909         }
 910
 911         if (set_default_unit(SPECIAL_DEFAULT_TARGET) < 0)
 912                 goto finish;
 913
 914         /* Mount /proc, /sys and friends, so that /proc/cmdline and
 915          * /proc/$PID/fd is available. */
 916         if (geteuid() == 0 && !getenv("SYSTEMD_SKIP_API_MOUNTS"))
 917                 if (mount_setup() < 0)
 918                         goto finish;
 919
 920         /* Reset all signal handlers. */
 921         assert_se(reset_all_signal_handlers() == 0);
 922
 923         /* If we are init, we can block sigkill. Yay. */
 924         ignore_signals(SIGNALS_IGNORE, -1);
 925
 926         if (parse_config_file() < 0)
 927                 goto finish;
 928
 929         if (arg_running_as == MANAGER_SYSTEM)
 930                 if (parse_proc_cmdline() < 0)
 931                         goto finish;
 932
 933         log_parse_environment();
 934
 935         if (parse_argv(argc, argv) < 0)
 936                 goto finish;
 937
 938         if (arg_action == ACTION_HELP) {
 939                 retval = help();
 940                 goto finish;
 941         } else if (arg_action == ACTION_DUMP_CONFIGURATION_ITEMS) {
 942                 unit_dump_config_items(stdout);
 943                 retval = EXIT_SUCCESS;
 944                 goto finish;
 945         } else if (arg_action == ACTION_DONE) {
 946                 retval = EXIT_SUCCESS;
 947                 goto finish;
 948         }
 949
 950         assert_se(arg_action == ACTION_RUN || arg_action == ACTION_TEST);
 951
 952         /* Remember open file descriptors for later deserialization */
 953         if (serialization) {
 954                 if ((r = fdset_new_fill(&fds)) < 0) {
 955                         log_error("Failed to allocate fd set: %s", strerror(-r));
 956                         goto finish;
 957                 }
 958
 959                 assert_se(fdset_remove(fds, fileno(serialization)) >= 0);
 960         } else
 961                 close_all_fds(NULL, 0);
 962
 963         /* Set up PATH unless it is already set */
 964         setenv("PATH",
 965                "/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin",
 966                arg_running_as == MANAGER_SYSTEM);
 967
 968         /* Unset some environment variables passed in from the kernel
 969          * that don't really make sense for us. */
 970         if (arg_running_as == MANAGER_SYSTEM) {
 971                 unsetenv("HOME");
 972                 unsetenv("TERM");
 973         }
 974
 975         /* Move out of the way, so that we won't block unmounts */
 976         assert_se(chdir("/")  == 0);
 977
 978         if (arg_running_as == MANAGER_SYSTEM) {
 979                 /* Become a session leader if we aren't one yet. */
 980                 setsid();
 981
 982                 /* Disable the umask logic */
 983                 umask(0);
 984         }
 985
 986         /* Make sure D-Bus doesn't fiddle with the SIGPIPE handlers */
 987         dbus_connection_set_change_sigpipe(FALSE);
 988
 989         /* Reset the console, but only if this is really init and we
 990          * are freshly booted */
 991         if (arg_running_as == MANAGER_SYSTEM && arg_action == ACTION_RUN) {
 992                 console_setup(getpid() == 1 && !serialization);
 993                 make_null_stdio();
 994         }
 995
 996         /* Open the logging devices, if possible and necessary */
 997         log_open();
 998
 999         /* Make sure we leave a core dump without panicing the
1000          * kernel. */
1001         if (getpid() == 1)
1002                 install_crash_handler();
1003
1004         log_full(arg_running_as == MANAGER_SYSTEM ? LOG_INFO : LOG_DEBUG,
1005                  PACKAGE_STRING " running in %s mode. (" SYSTEMD_FEATURES "; " DISTRIBUTION ")", manager_running_as_to_string(arg_running_as));
1006
1007         if (arg_running_as == MANAGER_SYSTEM && !serialization) {
1008                 locale_setup();
1009
1010                 if (arg_show_status)
1011                         status_welcome();
1012
1013                 kmod_setup();
1014                 hostname_setup();
1015                 loopback_setup();
1016
1017                 mkdir_p("/dev/.systemd/ask-password/", 0755);
1018         }
1019
1020         if ((r = manager_new(arg_running_as, &m)) < 0) {
1021                 log_error("Failed to allocate manager object: %s", strerror(-r));
1022                 goto finish;
1023         }
1024
1025         m->confirm_spawn = arg_confirm_spawn;
1026         m->show_status = arg_show_status;
1027 #ifdef HAVE_SYSV_COMPAT
1028         m->sysv_console = arg_sysv_console;
1029 #endif
1030         m->mount_auto = arg_mount_auto;
1031         m->swap_auto = arg_swap_auto;
1032
1033         if (arg_console)
1034                 manager_set_console(m, arg_console);
1035
1036         if ((r = manager_startup(m, serialization, fds)) < 0)
1037                 log_error("Failed to fully start up daemon: %s", strerror(-r));
1038
1039         if (fds) {
1040                 /* This will close all file descriptors that were opened, but
1041                  * not claimed by any unit. */
1042
1043                 fdset_free(fds);
1044                 fds = NULL;
1045         }
1046
1047         if (serialization) {
1048                 fclose(serialization);
1049                 serialization = NULL;
1050         } else {
1051                 DBusError error;
1052                 Unit *target = NULL;
1053
1054                 dbus_error_init(&error);
1055
1056                 log_debug("Activating default unit: %s", arg_default_unit);
1057
1058                 if ((r = manager_load_unit(m, arg_default_unit, NULL, &error, &target)) < 0) {
1059                         log_error("Failed to load default target: %s", bus_error(&error, r));
1060                         dbus_error_free(&error);
1061                 } else if (target->meta.load_state == UNIT_ERROR)
1062                         log_error("Failed to load default target: %s", strerror(-target->meta.load_error));
1063                 else if (target->meta.load_state == UNIT_MASKED)
1064                         log_error("Default target masked.");
1065
1066                 if (!target || target->meta.load_state != UNIT_LOADED) {
1067                         log_info("Trying to load rescue target...");
1068
1069                         if ((r = manager_load_unit(m, SPECIAL_RESCUE_TARGET, NULL, &error, &target)) < 0) {
1070                                 log_error("Failed to load rescue target: %s", bus_error(&error, r));
1071                                 dbus_error_free(&error);
1072                                 goto finish;
1073                         } else if (target->meta.load_state == UNIT_ERROR) {
1074                                 log_error("Failed to load rescue target: %s", strerror(-target->meta.load_error));
1075                                 goto finish;
1076                         } else if (target->meta.load_state == UNIT_MASKED) {
1077                                 log_error("Rescue target masked.");
1078                                 goto finish;
1079                         }
1080                 }
1081
1082                 assert(target->meta.load_state == UNIT_LOADED);
1083
1084                 if (arg_action == ACTION_TEST) {
1085                         printf("-> By units:\n");
1086                         manager_dump_units(m, stdout, "\t");
1087                 }
1088
1089                 if ((r = manager_add_job(m, JOB_START, target, JOB_REPLACE, false, &error, NULL)) < 0) {
1090                         log_error("Failed to start default target: %s", bus_error(&error, r));
1091                         dbus_error_free(&error);
1092                         goto finish;
1093                 }
1094
1095                 if (arg_action == ACTION_TEST) {
1096                         printf("-> By jobs:\n");
1097                         manager_dump_jobs(m, stdout, "\t");
1098                         retval = EXIT_SUCCESS;
1099                         goto finish;
1100                 }
1101         }
1102
1103         for (;;) {
1104                 if ((r = manager_loop(m)) < 0) {
1105                         log_error("Failed to run mainloop: %s", strerror(-r));
1106                         goto finish;
1107                 }
1108
1109                 switch (m->exit_code) {
1110
1111                 case MANAGER_EXIT:
1112                         retval = EXIT_SUCCESS;
1113                         log_debug("Exit.");
1114                         goto finish;
1115
1116                 case MANAGER_RELOAD:
1117                         log_info("Reloading.");
1118                         if ((r = manager_reload(m)) < 0)
1119                                 log_error("Failed to reload: %s", strerror(-r));
1120                         break;
1121
1122                 case MANAGER_REEXECUTE:
1123                         if (prepare_reexecute(m, &serialization, &fds) < 0)
1124                                 goto finish;
1125
1126                         reexecute = true;
1127                         log_notice("Reexecuting.");
1128                         goto finish;
1129
1130                 case MANAGER_REBOOT:
1131                 case MANAGER_POWEROFF:
1132                 case MANAGER_HALT:
1133                 case MANAGER_KEXEC: {
1134                         static const char * const table[_MANAGER_EXIT_CODE_MAX] = {
1135                                 [MANAGER_REBOOT] = "reboot",
1136                                 [MANAGER_POWEROFF] = "poweroff",
1137                                 [MANAGER_HALT] = "halt",
1138                                 [MANAGER_KEXEC] = "kexec"
1139                         };
1140
1141                         assert_se(shutdown_verb = table[m->exit_code]);
1142
1143                         log_notice("Shutting down.");
1144                         goto finish;
1145                 }
1146
1147                 default:
1148                         assert_not_reached("Unknown exit code.");
1149                 }
1150         }
1151
1152 finish:
1153         if (m)
1154                 manager_free(m);
1155
1156         free(arg_default_unit);
1157         free(arg_console);
1158
1159         dbus_shutdown();
1160
1161         label_finish();
1162
1163         if (reexecute) {
1164                 const char *args[15];
1165                 unsigned i = 0;
1166                 char sfd[16];
1167
1168                 assert(serialization);
1169                 assert(fds);
1170
1171                 args[i++] = SYSTEMD_BINARY_PATH;
1172
1173                 args[i++] = "--log-level";
1174                 args[i++] = log_level_to_string(log_get_max_level());
1175
1176                 args[i++] = "--log-target";
1177                 args[i++] = log_target_to_string(log_get_target());
1178
1179                 if (arg_running_as == MANAGER_SYSTEM)
1180                         args[i++] = "--system";
1181                 else
1182                         args[i++] = "--session";
1183
1184                 if (arg_dump_core)
1185                         args[i++] = "--dump-core";
1186
1187                 if (arg_crash_shell)
1188                         args[i++] = "--crash-shell";
1189
1190                 if (arg_confirm_spawn)
1191                         args[i++] = "--confirm-spawn";
1192
1193                 if (arg_show_status)
1194                         args[i++] = "--show-status=1";
1195                 else
1196                         args[i++] = "--show-status=0";
1197
1198 #ifdef HAVE_SYSV_COMPAT
1199                 if (arg_sysv_console)
1200                         args[i++] = "--sysv-console=1";
1201                 else
1202                         args[i++] = "--sysv-console=0";
1203 #endif
1204
1205                 snprintf(sfd, sizeof(sfd), "%i", fileno(serialization));
1206                 char_array_0(sfd);
1207
1208                 args[i++] = "--deserialize";
1209                 args[i++] = sfd;
1210
1211                 args[i++] = NULL;
1212
1213                 assert(i <= ELEMENTSOF(args));
1214
1215                 execv(args[0], (char* const*) args);
1216
1217                 log_error("Failed to reexecute: %m");
1218         }
1219
1220         if (serialization)
1221                 fclose(serialization);
1222
1223         if (fds)
1224                 fdset_free(fds);
1225
1226         if (shutdown_verb) {
1227                 const char * command_line[] = {
1228                         SYSTEMD_SHUTDOWN_BINARY_PATH,
1229                         shutdown_verb,
1230                         NULL
1231                 };
1232
1233                 execv(SYSTEMD_SHUTDOWN_BINARY_PATH, (char **) command_line);
1234                 log_error("Failed to execute shutdown binary, freezing: %m");
1235         }
1236
1237         if (getpid() == 1)
1238                 freeze();
1239
1240         return retval;
1241 }