/*** This file is part of systemd. Copyright 2014 Lennart Poettering Copyright 2012 Michael Olbrich 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 <sys/reboot.h> #include <linux/reboot.h> #include "bus-error.h" #include "bus-util.h" #include "failure-action.h" #include "special.h" #include "string-table.h" #include "terminal-util.h" static void log_and_status(Manager *m, const char *message) { log_warning("%s", message); manager_status_printf(m, STATUS_TYPE_EMERGENCY, ANSI_HIGHLIGHT_RED " !! " ANSI_NORMAL, "%s", message); } int failure_action( Manager *m, FailureAction action, const char *reboot_arg) { assert(m); assert(action >= 0); assert(action < _FAILURE_ACTION_MAX); if (action == FAILURE_ACTION_NONE) return -ECANCELED; if (!MANAGER_IS_SYSTEM(m)) { /* Downgrade all options to simply exiting if we run * in user mode */ log_warning("Exiting as result of failure."); m->exit_code = MANAGER_EXIT; return -ECANCELED; } switch (action) { case FAILURE_ACTION_REBOOT: log_and_status(m, "Rebooting as result of failure."); (void) update_reboot_parameter_and_warn(reboot_arg); (void) manager_add_job_by_name_and_warn(m, JOB_START, SPECIAL_REBOOT_TARGET, JOB_REPLACE_IRREVERSIBLY, NULL); break; case FAILURE_ACTION_REBOOT_FORCE: log_and_status(m, "Forcibly rebooting as result of failure."); (void) update_reboot_parameter_and_warn(reboot_arg); m->exit_code = MANAGER_REBOOT; break; case FAILURE_ACTION_REBOOT_IMMEDIATE: log_and_status(m, "Rebooting immediately as result of failure."); sync(); if (!isempty(reboot_arg)) { log_info("Rebooting with argument '%s'.", reboot_arg); syscall(SYS_reboot, LINUX_REBOOT_MAGIC1, LINUX_REBOOT_MAGIC2, LINUX_REBOOT_CMD_RESTART2, reboot_arg); log_warning_errno(errno, "Failed to reboot with parameter, retrying without: %m"); } log_info("Rebooting."); reboot(RB_AUTOBOOT); break; case FAILURE_ACTION_POWEROFF: log_and_status(m, "Powering off as result of failure."); (void) manager_add_job_by_name_and_warn(m, JOB_START, SPECIAL_POWEROFF_TARGET, JOB_REPLACE_IRREVERSIBLY, NULL); break; case FAILURE_ACTION_POWEROFF_FORCE: log_and_status(m, "Forcibly powering off as result of failure."); m->exit_code = MANAGER_POWEROFF; break; case FAILURE_ACTION_POWEROFF_IMMEDIATE: log_and_status(m, "Powering off immediately as result of failure."); sync(); log_info("Powering off."); reboot(RB_POWER_OFF); break; default: assert_not_reached("Unknown failure action"); } return -ECANCELED; } static const char* const failure_action_table[_FAILURE_ACTION_MAX] = { [FAILURE_ACTION_NONE] = "none", [FAILURE_ACTION_REBOOT] = "reboot", [FAILURE_ACTION_REBOOT_FORCE] = "reboot-force", [FAILURE_ACTION_REBOOT_IMMEDIATE] = "reboot-immediate", [FAILURE_ACTION_POWEROFF] = "poweroff", [FAILURE_ACTION_POWEROFF_FORCE] = "poweroff-force", [FAILURE_ACTION_POWEROFF_IMMEDIATE] = "poweroff-immediate" }; DEFINE_STRING_TABLE_LOOKUP(failure_action, FailureAction);