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 "modprobe-setup.h"
  42 #include "load-fragment.h"
  43 #include "fdset.h"
  44 #include "special.h"
  45 #include "conf-parser.h"
  46 #include "bus-errors.h"
  47 #include "missing.h"
  48 #include "label.h"
  49 #include "build.h"
  50
  51 static enum {
  52         ACTION_RUN,
  53         ACTION_HELP,
  54         ACTION_TEST,
  55         ACTION_DUMP_CONFIGURATION_ITEMS,
  56         ACTION_DONE
  57 } arg_action = ACTION_RUN;
  58
  59 static char *arg_default_unit = NULL;
  60 static char *arg_console = 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_nomodules = false;
  68 static bool arg_show_status = true;
  69 static bool arg_sysv_console = true;
  70
  71 static FILE* serialization = NULL;
  72
  73 _noreturn_ static void freeze(void) {
  74         for (;;)
  75                 pause();
  76 }
  77
  78 static void nop_handler(int sig) {
  79 }
  80
  81 _noreturn_ static void crash(int sig) {
  82
  83         if (!arg_dump_core)
  84                 log_error("Caught <%s>, not dumping core.", signal_to_string(sig));
  85         else {
  86                 struct sigaction sa;
  87                 pid_t pid;
  88
  89                 /* We want to wait for the core process, hence let's enable SIGCHLD */
  90                 zero(sa);
  91                 sa.sa_handler = nop_handler;
  92                 sa.sa_flags = SA_NOCLDSTOP|SA_RESTART;
  93                 assert_se(sigaction(SIGCHLD, &sa, NULL) == 0);
  94
  95                 if ((pid = fork()) < 0)
  96                         log_error("Caught <%s>, cannot fork for core dump: %s", signal_to_string(sig), strerror(errno));
  97
  98                 else if (pid == 0) {
  99                         struct rlimit rl;
 100
 101                         /* Enable default signal handler for core dump */
 102                         zero(sa);
 103                         sa.sa_handler = SIG_DFL;
 104                         assert_se(sigaction(sig, &sa, NULL) == 0);
 105
 106                         /* Don't limit the core dump size */
 107                         zero(rl);
 108                         rl.rlim_cur = RLIM_INFINITY;
 109                         rl.rlim_max = RLIM_INFINITY;
 110                         setrlimit(RLIMIT_CORE, &rl);
 111
 112                         /* Just to be sure... */
 113                         assert_se(chdir("/") == 0);
 114
 115                         /* Raise the signal again */
 116                         raise(sig);
 117
 118                         assert_not_reached("We shouldn't be here...");
 119                         _exit(1);
 120
 121                 } else {
 122                         int status;
 123
 124                         /* Order things nicely. */
 125                         if (waitpid(pid, &status, 0) < 0)
 126                                 log_error("Caught <%s>, waitpid() failed: %s", signal_to_string(sig), strerror(errno));
 127                         else if (!WCOREDUMP(status))
 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 make_null_stdio(void) {
 185         int null_fd, r;
 186
 187         if ((null_fd = open("/dev/null", O_RDWR|O_NOCTTY)) < 0) {
 188                 log_error("Failed to open /dev/null: %m");
 189                 return -errno;
 190         }
 191
 192         if ((r = make_stdio(null_fd)) < 0)
 193                 log_warning("Failed to dup2() device: %s", strerror(-r));
 194
 195         return r;
 196 }
 197
 198 static int console_setup(bool do_reset) {
 199         int tty_fd, r;
 200
 201         /* If we are init, we connect stdin/stdout/stderr to /dev/null
 202          * and make sure we don't have a controlling tty. */
 203
 204         release_terminal();
 205
 206         if (!do_reset)
 207                 return 0;
 208
 209         if ((tty_fd = open_terminal("/dev/console", O_WRONLY|O_NOCTTY|O_CLOEXEC)) < 0) {
 210                 log_error("Failed to open /dev/console: %s", strerror(-tty_fd));
 211                 return -tty_fd;
 212         }
 213
 214         if ((r = reset_terminal(tty_fd)) < 0)
 215                 log_error("Failed to reset /dev/console: %s", strerror(-r));
 216
 217         close_nointr_nofail(tty_fd);
 218         return r;
 219 }
 220
 221 static int set_default_unit(const char *u) {
 222         char *c;
 223
 224         assert(u);
 225
 226         if (!(c = strdup(u)))
 227                 return -ENOMEM;
 228
 229         free(arg_default_unit);
 230         arg_default_unit = c;
 231         return 0;
 232 }
 233
 234 static int parse_proc_cmdline_word(const char *word) {
 235
 236         static const char * const rlmap[] = {
 237                 "single", SPECIAL_RESCUE_TARGET,
 238                 "-s",     SPECIAL_RESCUE_TARGET,
 239                 "s",      SPECIAL_RESCUE_TARGET,
 240                 "S",      SPECIAL_RESCUE_TARGET,
 241                 "1",      SPECIAL_RESCUE_TARGET,
 242                 "2",      SPECIAL_RUNLEVEL2_TARGET,
 243                 "3",      SPECIAL_RUNLEVEL3_TARGET,
 244                 "4",      SPECIAL_RUNLEVEL4_TARGET,
 245                 "5",      SPECIAL_RUNLEVEL5_TARGET
 246         };
 247
 248         if (startswith(word, "systemd.unit="))
 249                 return set_default_unit(word + 13);
 250
 251         else if (startswith(word, "systemd.log_target=")) {
 252
 253                 if (log_set_target_from_string(word + 19) < 0)
 254                         log_warning("Failed to parse log target %s. Ignoring.", word + 19);
 255
 256         } else if (startswith(word, "systemd.log_level=")) {
 257
 258                 if (log_set_max_level_from_string(word + 18) < 0)
 259                         log_warning("Failed to parse log level %s. Ignoring.", word + 18);
 260
 261         } else if (startswith(word, "systemd.log_color=")) {
 262
 263                 if (log_show_color_from_string(word + 18) < 0)
 264                         log_warning("Failed to parse log color setting %s. Ignoring.", word + 18);
 265
 266         } else if (startswith(word, "systemd.log_location=")) {
 267
 268                 if (log_show_location_from_string(word + 21) < 0)
 269                         log_warning("Failed to parse log location setting %s. Ignoring.", word + 21);
 270
 271         } else if (startswith(word, "systemd.dump_core=")) {
 272                 int r;
 273
 274                 if ((r = parse_boolean(word + 18)) < 0)
 275                         log_warning("Failed to parse dump core switch %s, Ignoring.", word + 18);
 276                 else
 277                         arg_dump_core = r;
 278
 279         } else if (startswith(word, "systemd.crash_shell=")) {
 280                 int r;
 281
 282                 if ((r = parse_boolean(word + 20)) < 0)
 283                         log_warning("Failed to parse crash shell switch %s, Ignoring.", word + 20);
 284                 else
 285                         arg_crash_shell = r;
 286
 287         } else if (startswith(word, "systemd.confirm_spawn=")) {
 288                 int r;
 289
 290                 if ((r = parse_boolean(word + 22)) < 0)
 291                         log_warning("Failed to parse confirm spawn switch %s, Ignoring.", word + 22);
 292                 else
 293                         arg_confirm_spawn = r;
 294
 295         } else if (startswith(word, "systemd.crash_chvt=")) {
 296                 int k;
 297
 298                 if (safe_atoi(word + 19, &k) < 0)
 299                         log_warning("Failed to parse crash chvt switch %s, Ignoring.", word + 19);
 300                 else
 301                         arg_crash_chvt = k;
 302
 303         } else if (startswith(word, "systemd.show_status=")) {
 304                 int r;
 305
 306                 if ((r = parse_boolean(word + 20)) < 0)
 307                         log_warning("Failed to parse show status switch %s, Ignoring.", word + 20);
 308                 else
 309                         arg_show_status = r;
 310
 311         } else if (startswith(word, "systemd.sysv_console=")) {
 312                 int r;
 313
 314                 if ((r = parse_boolean(word + 21)) < 0)
 315                         log_warning("Failed to parse SysV console switch %s, Ignoring.", word + 20);
 316                 else
 317                         arg_sysv_console = r;
 318
 319         } else if (startswith(word, "systemd.")) {
 320
 321                 log_warning("Unknown kernel switch %s. Ignoring.", word);
 322
 323                 log_info("Supported kernel switches:\n"
 324                          "systemd.unit=UNIT                        Default unit to start\n"
 325                          "systemd.dump_core=0|1                    Dump core on crash\n"
 326                          "systemd.crash_shell=0|1                  Run shell on crash\n"
 327                          "systemd.crash_chvt=N                     Change to VT #N on crash\n"
 328                          "systemd.confirm_spawn=0|1                Confirm every process spawn\n"
 329                          "systemd.show_status=0|1                  Show status updates on the console during bootup\n"
 330                          "systemd.sysv_console=0|1                 Connect output of SysV scripts to console\n"
 331                          "systemd.log_target=console|kmsg|syslog|syslog-org-kmsg|null\n"
 332                          "                                         Log target\n"
 333                          "systemd.log_level=LEVEL                  Log level\n"
 334                          "systemd.log_color=0|1                    Highlight important log messages\n"
 335                          "systemd.log_location=0|1                 Include code location in log messages\n");
 336
 337         } else if (streq(word, "nomodules"))
 338                 arg_nomodules = true;
 339         else if (startswith(word, "console=")) {
 340                 const char *k;
 341                 size_t l;
 342                 char *w = NULL;
 343
 344                 k = word + 8;
 345                 l = strcspn(k, ",");
 346
 347                 if (l < 4 ||
 348                     !startswith(k, "tty") ||
 349                     k[3+strspn(k+3, "0123456789")] != 0) {
 350
 351                         if (!(w = strndup(k, l)))
 352                                 return -ENOMEM;
 353                 }
 354
 355                 free(arg_console);
 356                 arg_console = w;
 357
 358         } else if (streq(word, "quiet")) {
 359                 arg_show_status = false;
 360                 arg_sysv_console = false;
 361         } else {
 362                 unsigned i;
 363
 364                 /* SysV compatibility */
 365                 for (i = 0; i < ELEMENTSOF(rlmap); i += 2)
 366                         if (streq(word, rlmap[i]))
 367                                 return set_default_unit(rlmap[i+1]);
 368         }
 369
 370         return 0;
 371 }
 372
 373 static int config_parse_level(
 374                 const char *filename,
 375                 unsigned line,
 376                 const char *section,
 377                 const char *lvalue,
 378                 const char *rvalue,
 379                 void *data,
 380                 void *userdata) {
 381
 382         assert(filename);
 383         assert(lvalue);
 384         assert(rvalue);
 385
 386         log_set_max_level_from_string(rvalue);
 387         return 0;
 388 }
 389
 390 static int config_parse_target(
 391                 const char *filename,
 392                 unsigned line,
 393                 const char *section,
 394                 const char *lvalue,
 395                 const char *rvalue,
 396                 void *data,
 397                 void *userdata) {
 398
 399         assert(filename);
 400         assert(lvalue);
 401         assert(rvalue);
 402
 403         log_set_target_from_string(rvalue);
 404         return 0;
 405 }
 406
 407 static int config_parse_color(
 408                 const char *filename,
 409                 unsigned line,
 410                 const char *section,
 411                 const char *lvalue,
 412                 const char *rvalue,
 413                 void *data,
 414                 void *userdata) {
 415
 416         assert(filename);
 417         assert(lvalue);
 418         assert(rvalue);
 419
 420         log_show_color_from_string(rvalue);
 421         return 0;
 422 }
 423
 424 static int config_parse_location(
 425                 const char *filename,
 426                 unsigned line,
 427                 const char *section,
 428                 const char *lvalue,
 429                 const char *rvalue,
 430                 void *data,
 431                 void *userdata) {
 432
 433         assert(filename);
 434         assert(lvalue);
 435         assert(rvalue);
 436
 437         log_show_location_from_string(rvalue);
 438         return 0;
 439 }
 440
 441 static int config_parse_cpu_affinity(
 442                 const char *filename,
 443                 unsigned line,
 444                 const char *section,
 445                 const char *lvalue,
 446                 const char *rvalue,
 447                 void *data,
 448                 void *userdata) {
 449
 450         char *w;
 451         size_t l;
 452         char *state;
 453         cpu_set_t *c = NULL;
 454         unsigned ncpus = 0;
 455
 456         assert(filename);
 457         assert(lvalue);
 458         assert(rvalue);
 459
 460         FOREACH_WORD_QUOTED(w, l, rvalue, state) {
 461                 char *t;
 462                 int r;
 463                 unsigned cpu;
 464
 465                 if (!(t = strndup(w, l)))
 466                         return -ENOMEM;
 467
 468                 r = safe_atou(t, &cpu);
 469                 free(t);
 470
 471                 if (!c)
 472                         if (!(c = cpu_set_malloc(&ncpus)))
 473                                 return -ENOMEM;
 474
 475                 if (r < 0 || cpu >= ncpus) {
 476                         log_error("[%s:%u] Failed to parse CPU affinity: %s", filename, line, rvalue);
 477                         CPU_FREE(c);
 478                         return -EBADMSG;
 479                 }
 480
 481                 CPU_SET_S(cpu, CPU_ALLOC_SIZE(ncpus), c);
 482         }
 483
 484         if (c) {
 485                 if (sched_setaffinity(0, CPU_ALLOC_SIZE(ncpus), c) < 0)
 486                         log_warning("Failed to set CPU affinity: %m");
 487
 488                 CPU_FREE(c);
 489         }
 490
 491         return 0;
 492 }
 493
 494 static int parse_config_file(void) {
 495
 496         const ConfigItem items[] = {
 497                 { "LogLevel",    config_parse_level,        NULL,             "Manager" },
 498                 { "LogTarget",   config_parse_target,       NULL,             "Manager" },
 499                 { "LogColor",    config_parse_color,        NULL,             "Manager" },
 500                 { "LogLocation", config_parse_location,     NULL,             "Manager" },
 501                 { "DumpCore",    config_parse_bool,         &arg_dump_core,   "Manager" },
 502                 { "CrashShell",  config_parse_bool,         &arg_crash_shell, "Manager" },
 503                 { "ShowStatus",  config_parse_bool,         &arg_show_status, "Manager" },
 504                 { "SysVConsole", config_parse_bool,         &arg_sysv_console,"Manager" },
 505                 { "CrashChVT",   config_parse_int,          &arg_crash_chvt,  "Manager" },
 506                 { "CPUAffinity", config_parse_cpu_affinity, NULL,             "Manager" },
 507                 { NULL, NULL, NULL, NULL }
 508         };
 509
 510         static const char * const sections[] = {
 511                 "Manager",
 512                 NULL
 513         };
 514
 515         FILE *f;
 516         const char *fn;
 517         int r;
 518
 519         fn = arg_running_as == MANAGER_SYSTEM ? SYSTEM_CONFIG_FILE : SESSION_CONFIG_FILE;
 520
 521         if (!(f = fopen(fn, "re"))) {
 522                 if (errno == ENOENT)
 523                         return 0;
 524
 525                 log_warning("Failed to open configuration file '%s': %m", fn);
 526                 return 0;
 527         }
 528
 529         if ((r = config_parse(fn, f, sections, items, false, NULL)) < 0)
 530                 log_warning("Failed to parse configuration file: %s", strerror(-r));
 531
 532         fclose(f);
 533
 534         return 0;
 535 }
 536
 537 static int parse_proc_cmdline(void) {
 538         char *line;
 539         int r;
 540         char *w;
 541         size_t l;
 542         char *state;
 543
 544         if ((r = read_one_line_file("/proc/cmdline", &line)) < 0) {
 545                 log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r));
 546                 return 0;
 547         }
 548
 549         FOREACH_WORD_QUOTED(w, l, line, state) {
 550                 char *word;
 551
 552                 if (!(word = strndup(w, l))) {
 553                         r = -ENOMEM;
 554                         goto finish;
 555                 }
 556
 557                 r = parse_proc_cmdline_word(word);
 558                 free(word);
 559
 560                 if (r < 0)
 561                         goto finish;
 562         }
 563
 564         r = 0;
 565
 566 finish:
 567         free(line);
 568         return r;
 569 }
 570
 571 static int parse_argv(int argc, char *argv[]) {
 572
 573         enum {
 574                 ARG_LOG_LEVEL = 0x100,
 575                 ARG_LOG_TARGET,
 576                 ARG_LOG_COLOR,
 577                 ARG_LOG_LOCATION,
 578                 ARG_UNIT,
 579                 ARG_SYSTEM,
 580                 ARG_SESSION,
 581                 ARG_TEST,
 582                 ARG_DUMP_CONFIGURATION_ITEMS,
 583                 ARG_DUMP_CORE,
 584                 ARG_CRASH_SHELL,
 585                 ARG_CONFIRM_SPAWN,
 586                 ARG_SHOW_STATUS,
 587                 ARG_SYSV_CONSOLE,
 588                 ARG_DESERIALIZE,
 589                 ARG_INTROSPECT
 590         };
 591
 592         static const struct option options[] = {
 593                 { "log-level",                required_argument, NULL, ARG_LOG_LEVEL                },
 594                 { "log-target",               required_argument, NULL, ARG_LOG_TARGET               },
 595                 { "log-color",                optional_argument, NULL, ARG_LOG_COLOR                },
 596                 { "log-location",             optional_argument, NULL, ARG_LOG_LOCATION             },
 597                 { "unit",                     required_argument, NULL, ARG_UNIT                     },
 598                 { "system",                   no_argument,       NULL, ARG_SYSTEM                   },
 599                 { "session",                  no_argument,       NULL, ARG_SESSION                  },
 600                 { "test",                     no_argument,       NULL, ARG_TEST                     },
 601                 { "help",                     no_argument,       NULL, 'h'                          },
 602                 { "dump-configuration-items", no_argument,       NULL, ARG_DUMP_CONFIGURATION_ITEMS },
 603                 { "dump-core",                no_argument,       NULL, ARG_DUMP_CORE                },
 604                 { "crash-shell",              no_argument,       NULL, ARG_CRASH_SHELL              },
 605                 { "confirm-spawn",            no_argument,       NULL, ARG_CONFIRM_SPAWN            },
 606                 { "show-status",              optional_argument, NULL, ARG_SHOW_STATUS              },
 607                 { "sysv-console",             optional_argument, NULL, ARG_SYSV_CONSOLE             },
 608                 { "deserialize",              required_argument, NULL, ARG_DESERIALIZE              },
 609                 { "introspect",               optional_argument, NULL, ARG_INTROSPECT               },
 610                 { NULL,                       0,                 NULL, 0                            }
 611         };
 612
 613         int c, r;
 614
 615         assert(argc >= 1);
 616         assert(argv);
 617
 618         while ((c = getopt_long(argc, argv, "hD", options, NULL)) >= 0)
 619
 620                 switch (c) {
 621
 622                 case ARG_LOG_LEVEL:
 623                         if ((r = log_set_max_level_from_string(optarg)) < 0) {
 624                                 log_error("Failed to parse log level %s.", optarg);
 625                                 return r;
 626                         }
 627
 628                         break;
 629
 630                 case ARG_LOG_TARGET:
 631
 632                         if ((r = log_set_target_from_string(optarg)) < 0) {
 633                                 log_error("Failed to parse log target %s.", optarg);
 634                                 return r;
 635                         }
 636
 637                         break;
 638
 639                 case ARG_LOG_COLOR:
 640
 641                         if (optarg) {
 642                                 if ((r = log_show_color_from_string(optarg)) < 0) {
 643                                         log_error("Failed to parse log color setting %s.", optarg);
 644                                         return r;
 645                                 }
 646                         } else
 647                                 log_show_color(true);
 648
 649                         break;
 650
 651                 case ARG_LOG_LOCATION:
 652
 653                         if (optarg) {
 654                                 if ((r = log_show_location_from_string(optarg)) < 0) {
 655                                         log_error("Failed to parse log location setting %s.", optarg);
 656                                         return r;
 657                                 }
 658                         } else
 659                                 log_show_location(true);
 660
 661                         break;
 662
 663                 case ARG_UNIT:
 664
 665                         if ((r = set_default_unit(optarg)) < 0) {
 666                                 log_error("Failed to set default unit %s: %s", optarg, strerror(-r));
 667                                 return r;
 668                         }
 669
 670                         break;
 671
 672                 case ARG_SYSTEM:
 673                         arg_running_as = MANAGER_SYSTEM;
 674                         break;
 675
 676                 case ARG_SESSION:
 677                         arg_running_as = MANAGER_SESSION;
 678                         break;
 679
 680                 case ARG_TEST:
 681                         arg_action = ACTION_TEST;
 682                         break;
 683
 684                 case ARG_DUMP_CONFIGURATION_ITEMS:
 685                         arg_action = ACTION_DUMP_CONFIGURATION_ITEMS;
 686                         break;
 687
 688                 case ARG_DUMP_CORE:
 689                         arg_dump_core = true;
 690                         break;
 691
 692                 case ARG_CRASH_SHELL:
 693                         arg_crash_shell = true;
 694                         break;
 695
 696                 case ARG_CONFIRM_SPAWN:
 697                         arg_confirm_spawn = true;
 698                         break;
 699
 700                 case ARG_SHOW_STATUS:
 701
 702                         if (optarg) {
 703                                 if ((r = parse_boolean(optarg)) < 0) {
 704                                         log_error("Failed to show status boolean %s.", optarg);
 705                                         return r;
 706                                 }
 707                                 arg_show_status = r;
 708                         } else
 709                                 arg_show_status = true;
 710                         break;
 711
 712                 case ARG_SYSV_CONSOLE:
 713
 714                         if (optarg) {
 715                                 if ((r = parse_boolean(optarg)) < 0) {
 716                                         log_error("Failed to SysV console boolean %s.", optarg);
 717                                         return r;
 718                                 }
 719                                 arg_sysv_console = r;
 720                         } else
 721                                 arg_sysv_console = true;
 722                         break;
 723
 724                 case ARG_DESERIALIZE: {
 725                         int fd;
 726                         FILE *f;
 727
 728                         if ((r = safe_atoi(optarg, &fd)) < 0 || fd < 0) {
 729                                 log_error("Failed to parse deserialize option %s.", optarg);
 730                                 return r;
 731                         }
 732
 733                         if (!(f = fdopen(fd, "r"))) {
 734                                 log_error("Failed to open serialization fd: %m");
 735                                 return r;
 736                         }
 737
 738                         if (serialization)
 739                                 fclose(serialization);
 740
 741                         serialization = f;
 742
 743                         break;
 744                 }
 745
 746                 case ARG_INTROSPECT: {
 747                         const char * const * i = NULL;
 748
 749                         for (i = bus_interface_table; *i; i += 2)
 750                                 if (!optarg || streq(i[0], optarg)) {
 751                                         fputs(DBUS_INTROSPECT_1_0_XML_DOCTYPE_DECL_NODE
 752                                               "<node>\n", stdout);
 753                                         fputs(i[1], stdout);
 754                                         fputs("</node>\n", stdout);
 755
 756                                         if (optarg)
 757                                                 break;
 758                                 }
 759
 760                         if (!i[0] && optarg)
 761                                 log_error("Unknown interface %s.", optarg);
 762
 763                         arg_action = ACTION_DONE;
 764                         break;
 765                 }
 766
 767                 case 'h':
 768                         arg_action = ACTION_HELP;
 769                         break;
 770
 771                 case 'D':
 772                         log_set_max_level(LOG_DEBUG);
 773                         break;
 774
 775                 case '?':
 776                         return -EINVAL;
 777
 778                 default:
 779                         log_error("Unknown option code %c", c);
 780                         return -EINVAL;
 781                 }
 782
 783         /* PID 1 will get the kernel arguments as parameters, which we
 784          * ignore and unconditionally read from
 785          * /proc/cmdline. However, we need to ignore those arguments
 786          * here. */
 787         if (arg_running_as != MANAGER_SYSTEM && optind < argc) {
 788                 log_error("Excess arguments.");
 789                 return -EINVAL;
 790         }
 791
 792         return 0;
 793 }
 794
 795 static int help(void) {
 796
 797         printf("%s [OPTIONS...]\n\n"
 798                "Starts up and maintains the system or a session.\n\n"
 799                "  -h --help                      Show this help\n"
 800                "     --test                      Determine startup sequence, dump it and exit\n"
 801                "     --dump-configuration-items  Dump understood unit configuration items\n"
 802                "     --introspect[=INTERFACE]    Extract D-Bus interface data\n"
 803                "     --unit=UNIT                 Set default unit\n"
 804                "     --system                    Run a system instance, even if PID != 1\n"
 805                "     --session                   Run a session instance\n"
 806                "     --dump-core                 Dump core on crash\n"
 807                "     --crash-shell               Run shell on crash\n"
 808                "     --confirm-spawn             Ask for confirmation when spawning processes\n"
 809                "     --show-status[=0|1]         Show status updates on the console during bootup\n"
 810                "     --sysv-console[=0|1]        Connect output of SysV scripts to console\n"
 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         Unit *target = NULL;
 877         int r, retval = 1;
 878         FDSet *fds = NULL;
 879         bool reexecute = false;
 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                 if (label_init() < 0)
 900                         goto finish;
 901         } else {
 902                 arg_running_as = MANAGER_SESSION;
 903                 log_set_target(LOG_TARGET_CONSOLE);
 904         }
 905
 906         if (set_default_unit(SPECIAL_DEFAULT_TARGET) < 0)
 907                 goto finish;
 908
 909         /* Mount /proc, /sys and friends, so that /proc/cmdline and
 910          * /proc/$PID/fd is available. */
 911         if (geteuid() == 0 && !getenv("SYSTEMD_SKIP_API_MOUNTS"))
 912                 if (mount_setup() < 0)
 913                         goto finish;
 914
 915         /* Reset all signal handlers. */
 916         assert_se(reset_all_signal_handlers() == 0);
 917
 918         /* If we are init, we can block sigkill. Yay. */
 919         ignore_signals(SIGNALS_IGNORE, -1);
 920
 921         if (parse_config_file() < 0)
 922                 goto finish;
 923
 924         if (arg_running_as == MANAGER_SYSTEM)
 925                 if (parse_proc_cmdline() < 0)
 926                         goto finish;
 927
 928         log_parse_environment();
 929
 930         if (parse_argv(argc, argv) < 0)
 931                 goto finish;
 932
 933         if (arg_action == ACTION_HELP) {
 934                 retval = help();
 935                 goto finish;
 936         } else if (arg_action == ACTION_DUMP_CONFIGURATION_ITEMS) {
 937                 unit_dump_config_items(stdout);
 938                 retval = 0;
 939                 goto finish;
 940         } else if (arg_action == ACTION_DONE) {
 941                 retval = 0;
 942                 goto finish;
 943         }
 944
 945         assert_se(arg_action == ACTION_RUN || arg_action == ACTION_TEST);
 946
 947         /* Remember open file descriptors for later deserialization */
 948         if (serialization) {
 949                 if ((r = fdset_new_fill(&fds)) < 0) {
 950                         log_error("Failed to allocate fd set: %s", strerror(-r));
 951                         goto finish;
 952                 }
 953
 954                 assert_se(fdset_remove(fds, fileno(serialization)) >= 0);
 955         } else
 956                 close_all_fds(NULL, 0);
 957
 958         /* Set up PATH unless it is already set */
 959         setenv("PATH",
 960                "/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin",
 961                arg_running_as == MANAGER_SYSTEM);
 962
 963         /* Move out of the way, so that we won't block unmounts */
 964         assert_se(chdir("/")  == 0);
 965
 966         if (arg_running_as == MANAGER_SYSTEM) {
 967                 /* Become a session leader if we aren't one yet. */
 968                 setsid();
 969
 970                 /* Disable the umask logic */
 971                 umask(0);
 972         }
 973
 974         /* Make sure D-Bus doesn't fiddle with the SIGPIPE handlers */
 975         dbus_connection_set_change_sigpipe(FALSE);
 976
 977         /* Reset the console, but only if this is really init and we
 978          * are freshly booted */
 979         if (arg_running_as == MANAGER_SYSTEM && arg_action == ACTION_RUN) {
 980                 console_setup(getpid() == 1 && !serialization);
 981                 make_null_stdio();
 982         }
 983
 984         /* Open the logging devices, if possible and necessary */
 985         log_open();
 986
 987         /* Make sure we leave a core dump without panicing the
 988          * kernel. */
 989         if (getpid() == 1)
 990                 install_crash_handler();
 991
 992         log_full(arg_running_as == MANAGER_SYSTEM ? LOG_INFO : LOG_DEBUG,
 993                  PACKAGE_STRING " running in %s mode. (" SYSTEMD_FEATURES ")", manager_running_as_to_string(arg_running_as));
 994
 995         if (arg_running_as == MANAGER_SYSTEM) {
 996
 997                 /* Disable nscd, to avoid deadlocks when systemd uses
 998                  * NSS and the nscd socket is maintained by us. */
 999                 nss_disable_nscd();
1000
1001                 if (!serialization) {
1002                         if (arg_show_status)
1003                                 status_welcome();
1004                         modprobe_setup(arg_nomodules);
1005                         kmod_setup();
1006                         hostname_setup();
1007                         loopback_setup();
1008                 }
1009         }
1010
1011         if ((r = manager_new(arg_running_as, &m)) < 0) {
1012                 log_error("Failed to allocate manager object: %s", strerror(-r));
1013                 goto finish;
1014         }
1015
1016         m->confirm_spawn = arg_confirm_spawn;
1017         m->show_status = arg_show_status;
1018         m->sysv_console = arg_sysv_console;
1019
1020         if ((r = manager_startup(m, serialization, fds)) < 0)
1021                 log_error("Failed to fully start up daemon: %s", strerror(-r));
1022
1023         if (fds) {
1024                 /* This will close all file descriptors that were opened, but
1025                  * not claimed by any unit. */
1026
1027                 fdset_free(fds);
1028                 fds = NULL;
1029         }
1030
1031         if (serialization) {
1032                 fclose(serialization);
1033                 serialization = NULL;
1034         } else {
1035                 DBusError error;
1036
1037                 dbus_error_init(&error);
1038
1039                 log_debug("Activating default unit: %s", arg_default_unit);
1040
1041                 if ((r = manager_load_unit(m, arg_default_unit, NULL, &error, &target)) < 0) {
1042                         log_error("Failed to load default target: %s", bus_error(&error, r));
1043                         dbus_error_free(&error);
1044
1045                         log_info("Trying to load rescue target...");
1046                         if ((r = manager_load_unit(m, SPECIAL_RESCUE_TARGET, NULL, &error, &target)) < 0) {
1047                                 log_error("Failed to load rescue target: %s", bus_error(&error, r));
1048                                 dbus_error_free(&error);
1049                                 goto finish;
1050                         }
1051                 }
1052
1053                 if (arg_action == ACTION_TEST) {
1054                         printf("-> By units:\n");
1055                         manager_dump_units(m, stdout, "\t");
1056                 }
1057
1058                 if ((r = manager_add_job(m, JOB_START, target, JOB_REPLACE, false, &error, NULL)) < 0) {
1059                         log_error("Failed to start default target: %s", bus_error(&error, r));
1060                         dbus_error_free(&error);
1061                         goto finish;
1062                 }
1063
1064                 if (arg_console && arg_running_as == MANAGER_SYSTEM) {
1065                         char *name;
1066
1067                         if (asprintf(&name, "getty@%s.service", arg_console) < 0)
1068                                 log_error("Out of memory while generating console getty service name.");
1069                         else {
1070                                 if ((r = manager_add_job_by_name(m, JOB_START, name, JOB_FAIL, false, &error, NULL)) < 0) {
1071                                         log_error("Failed to start console getty target: %s", bus_error(&error, r));
1072                                         dbus_error_free(&error);
1073                                 }
1074
1075                                 free(name);
1076                         }
1077                 }
1078
1079                 if (arg_action == ACTION_TEST) {
1080                         printf("-> By jobs:\n");
1081                         manager_dump_jobs(m, stdout, "\t");
1082                         retval = 0;
1083                         goto finish;
1084                 }
1085         }
1086
1087         for (;;) {
1088                 if ((r = manager_loop(m)) < 0) {
1089                         log_error("Failed to run mainloop: %s", strerror(-r));
1090                         goto finish;
1091                 }
1092
1093                 switch (m->exit_code) {
1094
1095                 case MANAGER_EXIT:
1096                         retval = 0;
1097                         log_debug("Exit.");
1098                         goto finish;
1099
1100                 case MANAGER_RELOAD:
1101                         log_info("Reloading.");
1102                         if ((r = manager_reload(m)) < 0)
1103                                 log_error("Failed to reload: %s", strerror(-r));
1104                         break;
1105
1106                 case MANAGER_REEXECUTE:
1107                         if (prepare_reexecute(m, &serialization, &fds) < 0)
1108                                 goto finish;
1109
1110                         reexecute = true;
1111                         log_notice("Reexecuting.");
1112                         goto finish;
1113
1114                 default:
1115                         assert_not_reached("Unknown exit code.");
1116                 }
1117         }
1118
1119 finish:
1120         if (m)
1121                 manager_free(m);
1122
1123         free(arg_default_unit);
1124         free(arg_console);
1125
1126         dbus_shutdown();
1127
1128         if (reexecute) {
1129                 const char *args[15];
1130                 unsigned i = 0;
1131                 char sfd[16];
1132
1133                 assert(serialization);
1134                 assert(fds);
1135
1136                 args[i++] = SYSTEMD_BINARY_PATH;
1137
1138                 args[i++] = "--log-level";
1139                 args[i++] = log_level_to_string(log_get_max_level());
1140
1141                 args[i++] = "--log-target";
1142                 args[i++] = log_target_to_string(log_get_target());
1143
1144                 if (arg_running_as == MANAGER_SYSTEM)
1145                         args[i++] = "--system";
1146                 else
1147                         args[i++] = "--session";
1148
1149                 if (arg_dump_core)
1150                         args[i++] = "--dump-core";
1151
1152                 if (arg_crash_shell)
1153                         args[i++] = "--crash-shell";
1154
1155                 if (arg_confirm_spawn)
1156                         args[i++] = "--confirm-spawn";
1157
1158                 if (arg_show_status)
1159                         args[i++] = "--show-status=1";
1160                 else
1161                         args[i++] = "--show-status=0";
1162
1163                 if (arg_sysv_console)
1164                         args[i++] = "--sysv-console=1";
1165                 else
1166                         args[i++] = "--sysv-console=0";
1167
1168                 snprintf(sfd, sizeof(sfd), "%i", fileno(serialization));
1169                 char_array_0(sfd);
1170
1171                 args[i++] = "--deserialize";
1172                 args[i++] = sfd;
1173
1174                 args[i++] = NULL;
1175
1176                 assert(i <= ELEMENTSOF(args));
1177
1178                 execv(args[0], (char* const*) args);
1179
1180                 log_error("Failed to reexecute: %m");
1181         }
1182
1183         if (serialization)
1184                 fclose(serialization);
1185
1186         if (fds)
1187                 fdset_free(fds);
1188
1189         if (getpid() == 1)
1190                 freeze();
1191
1192         label_finish();
1193
1194         return retval;
1195 }