From 5192bd1945f59254b3d260ded15dd9f2b8cc2de7 Mon Sep 17 00:00:00 2001 From: Lennart Poettering Date: Tue, 17 Aug 2010 03:29:46 +0200 Subject: auto-getty: rework auto console getty logic to work in conjunction with single user mode --- src/auto-console-getty.c | 189 +++++++++++++++++++++++++++++++++++++++++++++++ src/initctl.c | 2 +- src/main.c | 39 +--------- 3 files changed, 193 insertions(+), 37 deletions(-) create mode 100644 src/auto-console-getty.c (limited to 'src') diff --git a/src/auto-console-getty.c b/src/auto-console-getty.c new file mode 100644 index 0000000000..44d2eff732 --- /dev/null +++ b/src/auto-console-getty.c @@ -0,0 +1,189 @@ +/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/ + +/*** + This file is part of systemd. + + Copyright 2010 Lennart Poettering + + systemd is free software; you can redistribute it and/or modify it + under the terms of the GNU General Public License as published by + the Free Software Foundation; either version 2 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 + General Public License for more details. + + You should have received a copy of the GNU General Public License + along with systemd; If not, see . +***/ + +#include +#include +#include + +#include + +#include "util.h" +#include "log.h" +#include "dbus-common.h" + +static int spawn_getty(DBusConnection *bus, const char *console) { + DBusMessage *m = NULL, *reply = NULL; + DBusError error; + const char *fail = "fail"; + char *name; + int r = -EIO; + + dbus_error_init(&error); + + assert(bus); + assert(console); + + /* FIXME: we probably should escape the tty name properly here */ + if (asprintf(&name, "getty@%s.service", console) < 0) + return -ENOMEM; + + if (!(m = dbus_message_new_method_call("org.freedesktop.systemd1", "/org/freedesktop/systemd1", "org.freedesktop.systemd1.Manager", "StartUnit"))) { + log_error("Could not allocate message."); + goto finish; + } + + if (!dbus_message_append_args(m, + DBUS_TYPE_STRING, &name, + DBUS_TYPE_STRING, &fail, + DBUS_TYPE_INVALID)) { + log_error("Could not attach target and flag information to message."); + goto finish; + } + + if (!(reply = dbus_connection_send_with_reply_and_block(bus, m, -1, &error))) { + log_error("Failed to start unit: %s", error.message); + goto finish; + } + + r = 0; + +finish: + if (m) + dbus_message_unref(m); + + if (reply) + dbus_message_unref(reply); + + dbus_error_free(&error); + + free(name); + + return r; +} + +static int parse_proc_cmdline_word(const char *word, char **console) { + assert(word); + + if (startswith(word, "console=")) { + const char *k; + size_t l; + char *w = NULL; + + k = word + 8; + l = strcspn(k, ","); + + if (l < 4 || + !startswith(k, "tty") || + k[3+strspn(k+3, "0123456789")] != 0) { + + if (!(w = strndup(k, l))) + return -ENOMEM; + + } + + free(*console); + *console = w; + } + + return 0; +} + +static int parse_proc_cmdline(char **console) { + char *line; + int r; + char *w; + size_t l; + char *state; + + assert(console); + + if ((r = read_one_line_file("/tmp/cmdline", &line)) < 0) { + log_warning("Failed to read /proc/cmdline, ignoring: %s", strerror(-r)); + return 0; + } + + FOREACH_WORD_QUOTED(w, l, line, state) { + char *word; + + if (!(word = strndup(w, l))) { + r = -ENOMEM; + goto finish; + } + + r = parse_proc_cmdline_word(word, console); + free(word); + + if (r < 0) + goto finish; + } + + r = 0; + +finish: + free(line); + return r; +} + +int main(int argc, char *argv[]) { + DBusError error; + int r = 1; + char *console = NULL; + DBusConnection *bus = NULL; + + dbus_error_init(&error); + + if (argc > 1) { + log_error("This program does not take arguments."); + return 1; + } + + log_set_target(LOG_TARGET_SYSLOG_OR_KMSG); + log_parse_environment(); + log_open(); + + if (bus_connect(DBUS_BUS_SYSTEM, &bus, NULL, &error) < 0) { + log_error("Failed to get D-Bus connection: %s", error.message); + goto finish; + } + + if (parse_proc_cmdline(&console) < 0) + goto finish; + + if (console) + if (spawn_getty(bus, console) < 0) + goto finish; + + r = 0; + +finish: + free(console); + + if (bus) { + dbus_connection_close(bus); + dbus_connection_unref(bus); + } + + dbus_error_free(&error); + + dbus_shutdown(); + + return r; +} diff --git a/src/initctl.c b/src/initctl.c index 115452e86b..7b478a75d5 100644 --- a/src/initctl.c +++ b/src/initctl.c @@ -119,7 +119,7 @@ static void change_runlevel(Server *s, int runlevel) { DBUS_TYPE_STRING, &target, DBUS_TYPE_STRING, &replace, DBUS_TYPE_INVALID)) { - log_error("Could not attach target and flag information to signal message."); + log_error("Could not attach target and flag information to message."); goto finish; } diff --git a/src/main.c b/src/main.c index 5b5c05acbc..51253c66c8 100644 --- a/src/main.c +++ b/src/main.c @@ -57,7 +57,6 @@ static enum { } arg_action = ACTION_RUN; static char *arg_default_unit = NULL; -static char *arg_console = NULL; static ManagerRunningAs arg_running_as = _MANAGER_RUNNING_AS_INVALID; static bool arg_dump_core = true; @@ -245,6 +244,8 @@ static int parse_proc_cmdline_word(const char *word) { "5", SPECIAL_RUNLEVEL5_TARGET }; + assert(word); + if (startswith(word, "systemd.unit=")) return set_default_unit(word + 13); @@ -336,26 +337,8 @@ static int parse_proc_cmdline_word(const char *word) { } else if (streq(word, "nomodules")) arg_nomodules = true; - else if (startswith(word, "console=")) { - const char *k; - size_t l; - char *w = NULL; - - k = word + 8; - l = strcspn(k, ","); - - if (l < 4 || - !startswith(k, "tty") || - k[3+strspn(k+3, "0123456789")] != 0) { - - if (!(w = strndup(k, l))) - return -ENOMEM; - } - - free(arg_console); - arg_console = w; - } else if (streq(word, "quiet")) { + else if (streq(word, "quiet")) { arg_show_status = false; arg_sysv_console = false; } else { @@ -1061,21 +1044,6 @@ int main(int argc, char *argv[]) { goto finish; } - if (arg_console && arg_running_as == MANAGER_SYSTEM) { - char *name; - - if (asprintf(&name, "getty@%s.service", arg_console) < 0) - log_error("Out of memory while generating console getty service name."); - else { - if ((r = manager_add_job_by_name(m, JOB_START, name, JOB_FAIL, false, &error, NULL)) < 0) { - log_error("Failed to start console getty target: %s", bus_error(&error, r)); - dbus_error_free(&error); - } - - free(name); - } - } - if (arg_action == ACTION_TEST) { printf("-> By jobs:\n"); manager_dump_jobs(m, stdout, "\t"); @@ -1121,7 +1089,6 @@ finish: manager_free(m); free(arg_default_unit); - free(arg_console); dbus_shutdown(); -- cgit v1.2.3-54-g00ecf