a336a7912ecb62fb9310c4f3e50dc622aea4951cKnut Anders Hatlen

a336a7912ecb62fb9310c4f3e50dc622aea4951cKnut Anders Hatlen

a336a7912ecb62fb9310c4f3e50dc622aea4951cKnut Anders Hatlen#include "mkdir.h"

#include "parse-util.h"

#include "proc-cmdline.h"

#include "string-util.h"

#include "strv.h"

#include "unit-name.h"

#include "util.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;

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;

}

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

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

}