#include "alloc-util.h"

#include "mkdir.h"

#include "parse-util.h"

#include "proc-cmdline.h"

#include "special.h"

#include "string-util.h"

#include "strv.h"

#include "unit-name.h"

#include "util.h"

static char *arg_default_unit = NULL;

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;

r = unit_name_mangle(value, UNIT_NAME_NOGLOB, &n);

if (r < 0)

return log_error_errno(r, "Failed to glob unit name: %m");

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;

r = unit_name_mangle(value, UNIT_NAME_NOGLOB, &n);

if (r < 0)

return log_error_errno(r, "Failed to glob unit name: %m");

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;

} else if (streq(key, "systemd.unit")) {

if (!value)

log_error("Missing argument for systemd.unit= kernel command line parameter.");

else {

r = free_and_strdup(&arg_default_unit, value);

if (r < 0)

return log_error_errno(r, "Failed to set default unit %s: %m", value);

}

} else if (!value) {

const char *target;

target = runlevel_to_target(key);

if (target) {

r = free_and_strdup(&arg_default_unit, target);

if (r < 0)

return log_error_errno(r, "Failed to set default unit %s: %m", target);

}

}

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)

r = log_error_errno(errno,

"Failed to create mask symlink %s: %m",

p);

}

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, "/", arg_default_unit, ".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)

r = log_error_errno(errno,

"Failed to create wants symlink %s: %m",

p);

}

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);

r = free_and_strdup(&arg_default_unit, SPECIAL_DEFAULT_TARGET);

if (r < 0) {

log_error_errno(r, "Failed to set default unit %s: %m", SPECIAL_DEFAULT_TARGET);

goto finish;

}

r = parse_proc_cmdline(parse_proc_cmdline_item);

if (r < 0)

log_warning_errno(r, "Failed to parse kernel command line, ignoring: %m");

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;

}