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