/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ /*** This file is part of systemd. Copyright 2014 Lennart Poettering systemd is free software; you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 2.1 of the License, or (at your option) any later version. systemd is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with systemd; If not, see <http://www.gnu.org/licenses/>. ***/ #include "util.h" #include "strv.h" #include "unit-name.h" #include "mkdir.h" static const char *arg_dest = "/tmp"; static char **arg_mask = NULL; static char **arg_wants = NULL; static bool arg_debug_shell = false; static int parse_proc_cmdline_item(const char *key, const char *value) { int r; assert(key); if (streq(key, "systemd.mask")) { if (!value) log_error("Missing argument for systemd.mask= kernel command line parameter."); else { char *n; n = unit_name_mangle(value, MANGLE_NOGLOB); if (!n) return log_oom(); r = strv_consume(&arg_mask, n); if (r < 0) return log_oom(); } } else if (streq(key, "systemd.wants")) { if (!value) log_error("Missing argument for systemd.want= kernel command line parameter."); else { char *n; n = unit_name_mangle(value, MANGLE_NOGLOB); if (!n) return log_oom(); r = strv_consume(&arg_wants, n); if (r < 0) return log_oom(); } } else if (streq(key, "systemd.debug-shell")) { if (value) { r = parse_boolean(value); if (r < 0) log_error("Failed to parse systemd.debug-shell= argument '%s', ignoring.", value); else arg_debug_shell = r; } else arg_debug_shell = true; } return 0; } static int generate_mask_symlinks(void) { char **u; int r = 0; if (strv_isempty(arg_mask)) return 0; STRV_FOREACH(u, arg_mask) { _cleanup_free_ char *p = NULL; p = strjoin(arg_dest, "/", *u, NULL); if (!p) return log_oom(); if (symlink("/dev/null", p) < 0) { log_error("Failed to create mask symlink %s: %m", p); r = -errno; } } return r; } static int generate_wants_symlinks(void) { char **u; int r = 0; if (strv_isempty(arg_wants)) return 0; STRV_FOREACH(u, arg_wants) { _cleanup_free_ char *p = NULL, *f = NULL; p = strjoin(arg_dest, "/default.target.wants/", *u, NULL); if (!p) return log_oom(); f = strappend(SYSTEM_DATA_UNIT_PATH "/", *u); if (!f) return log_oom(); mkdir_parents_label(p, 0755); if (symlink(f, p) < 0) { log_error("Failed to create wants symlink %s: %m", p); r = -errno; } } return r; } int main(int argc, char *argv[]) { int r, q; if (argc > 1 && argc != 4) { log_error("This program takes three or no arguments."); return EXIT_FAILURE; } if (argc > 1) arg_dest = argv[2]; log_set_target(LOG_TARGET_SAFE); log_parse_environment(); log_open(); umask(0022); if (parse_proc_cmdline(parse_proc_cmdline_item) < 0) return EXIT_FAILURE; if (arg_debug_shell) { r = strv_extend(&arg_wants, "debug-shell.service"); if (r < 0) { r = log_oom(); goto finish; } } r = generate_mask_symlinks(); q = generate_wants_symlinks(); if (q < 0) r = q; finish: return r < 0 ? EXIT_FAILURE : EXIT_SUCCESS; }