/* SPDX-License-Identifier: LGPL-2.1-or-later */

#include <unistd.h>

#include "sd-bus.h"
#include "sd-messages.h"

#include "alloc-util.h"
#include "bitfield.h"
#include "bus-error.h"
#include "bus-unit-util.h"
#include "conf-parser.h"
#include "extract-word.h"
#include "format-util.h"
#include "hashmap.h"
#include "logind.h"
#include "logind-action.h"
#include "logind-dbus.h"
#include "logind-seat-dbus.h"
#include "logind-session-dbus.h"
#include "process-util.h"
#include "special.h"
#include "string-table.h"
#include "strv.h"
#include "user-util.h"

static const HandleActionData handle_action_data_table[_HANDLE_ACTION_MAX] = {
        [HANDLE_POWEROFF] = {
                .handle                          = HANDLE_POWEROFF,
                .target                          = SPECIAL_POWEROFF_TARGET,
                .inhibit_what                    = INHIBIT_SHUTDOWN,
                .polkit_action                   = "org.freedesktop.login1.power-off",
                .polkit_action_multiple_sessions = "org.freedesktop.login1.power-off-multiple-sessions",
                .polkit_action_ignore_inhibit    = "org.freedesktop.login1.power-off-ignore-inhibit",
                .sleep_operation                 = _SLEEP_OPERATION_INVALID,
                .message_id                      = SD_MESSAGE_SHUTDOWN_STR,
                .message                         = "System is powering down",
                .log_verb                        = "power-off",
        },
        [HANDLE_REBOOT] = {
                .handle                          = HANDLE_REBOOT,
                .target                          = SPECIAL_REBOOT_TARGET,
                .inhibit_what                    = INHIBIT_SHUTDOWN,
                .polkit_action                   = "org.freedesktop.login1.reboot",
                .polkit_action_multiple_sessions = "org.freedesktop.login1.reboot-multiple-sessions",
                .polkit_action_ignore_inhibit    = "org.freedesktop.login1.reboot-ignore-inhibit",
                .sleep_operation                 = _SLEEP_OPERATION_INVALID,
                .message_id                      = SD_MESSAGE_SHUTDOWN_STR,
                .message                         = "System is rebooting",
                .log_verb                        = "reboot",
        },
        [HANDLE_HALT] = {
                .handle                          = HANDLE_HALT,
                .target                          = SPECIAL_HALT_TARGET,
                .inhibit_what                    = INHIBIT_SHUTDOWN,
                .polkit_action                   = "org.freedesktop.login1.halt",
                .polkit_action_multiple_sessions = "org.freedesktop.login1.halt-multiple-sessions",
                .polkit_action_ignore_inhibit    = "org.freedesktop.login1.halt-ignore-inhibit",
                .sleep_operation                 = _SLEEP_OPERATION_INVALID,
                .message_id                      = SD_MESSAGE_SHUTDOWN_STR,
                .message                         = "System is halting",
                .log_verb                        = "halt",
        },
        [HANDLE_KEXEC] = {
                .handle                          = HANDLE_KEXEC,
                .target                          = SPECIAL_KEXEC_TARGET,
                .inhibit_what                    = INHIBIT_SHUTDOWN,
                .polkit_action                   = "org.freedesktop.login1.reboot",
                .polkit_action_multiple_sessions = "org.freedesktop.login1.reboot-multiple-sessions",
                .polkit_action_ignore_inhibit    = "org.freedesktop.login1.reboot-ignore-inhibit",
                .sleep_operation                 = _SLEEP_OPERATION_INVALID,
                .message_id                      = SD_MESSAGE_SHUTDOWN_STR,
                .message                         = "System is rebooting with kexec",
                .log_verb                        = "kexec",
        },
        [HANDLE_SOFT_REBOOT] = {
                .handle                          = HANDLE_SOFT_REBOOT,
                .target                          = SPECIAL_SOFT_REBOOT_TARGET,
                .inhibit_what                    = INHIBIT_SHUTDOWN,
                .polkit_action                   = "org.freedesktop.login1.reboot",
                .polkit_action_multiple_sessions = "org.freedesktop.login1.reboot-multiple-sessions",
                .polkit_action_ignore_inhibit    = "org.freedesktop.login1.reboot-ignore-inhibit",
                .sleep_operation                 = _SLEEP_OPERATION_INVALID,
                .message_id                      = SD_MESSAGE_SHUTDOWN_STR,
                .message                         = "System userspace is rebooting",
                .log_verb                        = "soft-reboot",
        },
        [HANDLE_SUSPEND] = {
                .handle                          = HANDLE_SUSPEND,
                .target                          = SPECIAL_SUSPEND_TARGET,
                .inhibit_what                    = INHIBIT_SLEEP,
                .polkit_action                   = "org.freedesktop.login1.suspend",
                .polkit_action_multiple_sessions = "org.freedesktop.login1.suspend-multiple-sessions",
                .polkit_action_ignore_inhibit    = "org.freedesktop.login1.suspend-ignore-inhibit",
                .sleep_operation                 = SLEEP_SUSPEND,
        },
        [HANDLE_HIBERNATE] = {
                .handle                          = HANDLE_HIBERNATE,
                .target                          = SPECIAL_HIBERNATE_TARGET,
                .inhibit_what                    = INHIBIT_SLEEP,
                .polkit_action                   = "org.freedesktop.login1.hibernate",
                .polkit_action_multiple_sessions = "org.freedesktop.login1.hibernate-multiple-sessions",
                .polkit_action_ignore_inhibit    = "org.freedesktop.login1.hibernate-ignore-inhibit",
                .sleep_operation                 = SLEEP_HIBERNATE,
        },
        [HANDLE_HYBRID_SLEEP] = {
                .handle                          = HANDLE_HYBRID_SLEEP,
                .target                          = SPECIAL_HYBRID_SLEEP_TARGET,
                .inhibit_what                    = INHIBIT_SLEEP,
                .polkit_action                   = "org.freedesktop.login1.hibernate",
                .polkit_action_multiple_sessions = "org.freedesktop.login1.hibernate-multiple-sessions",
                .polkit_action_ignore_inhibit    = "org.freedesktop.login1.hibernate-ignore-inhibit",
                .sleep_operation                 = SLEEP_HYBRID_SLEEP,
        },
        [HANDLE_SUSPEND_THEN_HIBERNATE] = {
                .handle                          = HANDLE_SUSPEND_THEN_HIBERNATE,
                .target                          = SPECIAL_SUSPEND_THEN_HIBERNATE_TARGET,
                .inhibit_what                    = INHIBIT_SLEEP,
                .polkit_action                   = "org.freedesktop.login1.hibernate",
                .polkit_action_multiple_sessions = "org.freedesktop.login1.hibernate-multiple-sessions",
                .polkit_action_ignore_inhibit    = "org.freedesktop.login1.hibernate-ignore-inhibit",
                .sleep_operation                 = SLEEP_SUSPEND_THEN_HIBERNATE,
        },
        [HANDLE_FACTORY_RESET] = {
                .handle                          = HANDLE_FACTORY_RESET,
                .target                          = SPECIAL_FACTORY_RESET_TARGET,
                .inhibit_what                    = _INHIBIT_WHAT_INVALID,
                .sleep_operation                 = _SLEEP_OPERATION_INVALID,
                .message_id                      = SD_MESSAGE_FACTORY_RESET_STR,
                .message                         = "System is performing factory reset",
        },
};

const HandleActionData* handle_action_lookup(HandleAction action) {

        if (action < 0 || (size_t) action >= ELEMENTSOF(handle_action_data_table))
                return NULL;

        return &handle_action_data_table[action];
}

static bool handle_action_sleep_supported(HandleAction action) {
        assert(HANDLE_ACTION_IS_SLEEP(action) && action != HANDLE_SLEEP);
        return sleep_supported(ASSERT_PTR(handle_action_lookup(action))->sleep_operation) > 0;
}

/* The order in which we try each sleep operation. We should typically prefer operations without a delay,
 * i.e. s2h and suspend, and use hibernation at last since it requires minimum hardware support.
 * hybrid-sleep is disabled by default, and thus should be ordered before suspend if manually chosen by user,
 * since it implies suspend and will probably never be selected by us otherwise. */
static const HandleAction sleep_actions[] = {
        HANDLE_SUSPEND_THEN_HIBERNATE,
        HANDLE_HYBRID_SLEEP,
        HANDLE_SUSPEND,
        HANDLE_HIBERNATE,
};

int handle_action_get_enabled_sleep_actions(HandleActionSleepMask mask, char ***ret) {
        _cleanup_strv_free_ char **actions = NULL;
        int r;

        assert(ret);

        FOREACH_ELEMENT(i, sleep_actions)
                if (BIT_SET(mask, *i)) {
                        r = strv_extend(&actions, handle_action_to_string(*i));
                        if (r < 0)
                                return r;
                }

        *ret = TAKE_PTR(actions);
        return 0;
}

HandleAction handle_action_sleep_select(Manager *m) {
        assert(m);

        FOREACH_ELEMENT(i, sleep_actions) {
                const HandleActionData *a;
                _cleanup_free_ char *load_state = NULL;

                if (!BIT_SET(m->handle_action_sleep_mask, *i))
                        continue;

                a = ASSERT_PTR(handle_action_lookup(*i));

                if (sleep_supported(a->sleep_operation) <= 0)
                        continue;

                (void) unit_load_state(m->bus, a->target, &load_state);
                if (streq_ptr(load_state, "loaded"))
                        return *i;
        }

        return _HANDLE_ACTION_INVALID;
}

static int handle_action_execute(
                Manager *m,
                HandleAction handle,
                bool ignore_inhibited,
                bool is_edge) {

        static const char * const message_table[_HANDLE_ACTION_MAX] = {
                [HANDLE_POWEROFF]               = "Powering off...",
                [HANDLE_REBOOT]                 = "Rebooting...",
                [HANDLE_HALT]                   = "Halting...",
                [HANDLE_KEXEC]                  = "Rebooting via kexec...",
                [HANDLE_SOFT_REBOOT]            = "Rebooting userspace...",
                [HANDLE_SUSPEND]                = "Suspending...",
                [HANDLE_HIBERNATE]              = "Hibernating...",
                [HANDLE_HYBRID_SLEEP]           = "Hibernating and suspending...",
                [HANDLE_SUSPEND_THEN_HIBERNATE] = "Suspending, then hibernating...",
                [HANDLE_FACTORY_RESET]          = "Performing factory reset...",
        };

        _cleanup_(sd_bus_error_free) sd_bus_error error = SD_BUS_ERROR_NULL;
        InhibitWhat inhibit_operation;
        Inhibitor *offending = NULL;
        int r;

        assert(m);
        assert(!IN_SET(handle, HANDLE_IGNORE, HANDLE_LOCK, HANDLE_SLEEP));

        if (m->delayed_action)
                return log_debug_errno(SYNTHETIC_ERRNO(EALREADY),
                                       "Action %s already in progress, ignoring requested %s operation.",
                                       handle_action_to_string(m->delayed_action->handle),
                                       handle_action_to_string(handle));

        inhibit_operation = ASSERT_PTR(handle_action_lookup(handle))->inhibit_what;

        /* If the actual operation is inhibited, warn and fail */
        if (inhibit_what_is_valid(inhibit_operation) &&
            !ignore_inhibited &&
            manager_is_inhibited(m, inhibit_operation, NULL, /* flags= */ 0, UID_INVALID, &offending)) {
                _cleanup_free_ char *comm = NULL, *u = NULL;

                (void) pidref_get_comm(&offending->pid, &comm);
                u = uid_to_name(offending->uid);

                /* If this is just a recheck of the lid switch then don't warn about anything */
                log_full(is_edge ? LOG_ERR : LOG_DEBUG,
                         "Refusing %s operation, %s is inhibited by UID "UID_FMT"/%s, PID "PID_FMT"/%s.",
                         handle_action_to_string(handle),
                         inhibit_what_to_string(inhibit_operation),
                         offending->uid, strna(u),
                         offending->pid.pid, strna(comm));

                return is_edge ? -EPERM : 0;
        }

        log_info("%s", message_table[handle]);

        r = bus_manager_shutdown_or_sleep_now_or_later(m, handle_action_lookup(handle), &error);
        if (r < 0)
                return log_error_errno(r, "Failed to execute %s operation: %s",
                                       handle_action_to_string(handle),
                                       bus_error_message(&error, r));

        return 1;
}

static int handle_action_sleep_execute(
                Manager *m,
                HandleAction handle,
                bool ignore_inhibited,
                bool is_edge) {

        assert(m);
        assert(HANDLE_ACTION_IS_SLEEP(handle));

        if (handle == HANDLE_SLEEP) {
                HandleAction a;

                a = handle_action_sleep_select(m);
                if (a < 0)
                        return log_warning_errno(SYNTHETIC_ERRNO(EOPNOTSUPP),
                                                 "None of the configured sleep operations are supported, ignoring.");

                return handle_action_sleep_execute(m, a, ignore_inhibited, is_edge);
        }

        bool supported = handle_action_sleep_supported(handle);

        if (!supported && handle != HANDLE_SUSPEND) {
                supported = sleep_supported(SLEEP_SUSPEND) > 0;
                if (supported) {
                        log_notice("Requested %s operation is not supported, using regular suspend instead.",
                                   handle_action_to_string(handle));
                        handle = HANDLE_SUSPEND;
                }
        }

        if (!supported)
                return log_warning_errno(SYNTHETIC_ERRNO(EOPNOTSUPP),
                                         "Requested %s operation not supported, ignoring.", handle_action_to_string(handle));

        return handle_action_execute(m, handle, ignore_inhibited, is_edge);
}

static int manager_handle_action_secure_attention_key(
                Manager *m,
                bool is_edge,
                const char *seat) {

        int r;
        Seat *o;
        _cleanup_free_ char *p = NULL;

        assert(m);

        if (!is_edge)
                return 0;

        if (!seat)
                return 0;

        o = hashmap_get(m->seats, seat);
        if (!o)
                return 0;

        p = seat_bus_path(o);
        if (!p)
                return log_oom();

        log_struct(LOG_INFO,
                   LOG_MESSAGE("Secure Attention Key sequence pressed on seat %s", seat),
                   LOG_MESSAGE_ID(SD_MESSAGE_SECURE_ATTENTION_KEY_PRESS_STR),
                   LOG_ITEM("SEAT_ID=%s", seat));

        r = sd_bus_emit_signal(
                        m->bus,
                        "/org/freedesktop/login1",
                        "org.freedesktop.login1.Manager",
                        "SecureAttentionKey",
                        "so", seat, p);
        if (r < 0)
                log_warning_errno(r, "Failed to emit SecureAttentionKey signal, ignoring: %m");

        return 0;
}

int manager_handle_action(
                Manager *m,
                InhibitWhat inhibit_key,
                HandleAction action,
                bool ignore_inhibited,
                bool is_edge,
                const char *action_seat) {

        assert(m);
        assert(handle_action_valid(action));

        /* If the key handling is turned off, don't do anything */
        if (action == HANDLE_IGNORE) {
                log_debug("Handling of %s (%s) is disabled, taking no action.",
                          inhibit_key == 0 ? "idle timeout" : inhibit_what_to_string(inhibit_key),
                          is_edge ? "edge" : "level");
                return 0;
        }

        if (inhibit_key == INHIBIT_HANDLE_LID_SWITCH) {
                /* If the last system suspend or startup is too close, let's not suspend for now, to give
                 * USB docking stations some time to settle so that we can properly watch its displays. */
                if (m->lid_switch_ignore_event_source) {
                        log_debug("Ignoring lid switch request, system startup or resume too close.");
                        return 0;
                }
        }

        /* If the key handling is inhibited, don't do anything */
        if (inhibit_key > 0) {
                if (manager_is_inhibited(m, inhibit_key, NULL, MANAGER_IS_INHIBITED_IGNORE_INACTIVE, UID_INVALID, NULL)) {
                        log_debug("Refusing %s operation, %s is inhibited.",
                                  handle_action_to_string(action),
                                  inhibit_what_to_string(inhibit_key));
                        return 0;
                }
        }

        /* Locking and greeter activation is handled differently from the rest. */
        if (action == HANDLE_LOCK) {
                if (!is_edge)
                        return 0;

                log_info("Locking sessions...");
                session_send_lock_all(m, true);
                return 1;
        }

        if (action == HANDLE_SECURE_ATTENTION_KEY)
                return manager_handle_action_secure_attention_key(m, is_edge, action_seat);

        if (HANDLE_ACTION_IS_SLEEP(action))
                return handle_action_sleep_execute(m, action, ignore_inhibited, is_edge);

        return handle_action_execute(m, action, ignore_inhibited, is_edge);
}

static const char* const handle_action_verb_table[_HANDLE_ACTION_MAX] = {
        [HANDLE_IGNORE]                 = "do nothing",
        [HANDLE_POWEROFF]               = "power off",
        [HANDLE_REBOOT]                 = "reboot",
        [HANDLE_HALT]                   = "halt",
        [HANDLE_KEXEC]                  = "kexec",
        [HANDLE_SOFT_REBOOT]            = "soft-reboot",
        [HANDLE_SUSPEND]                = "suspend",
        [HANDLE_HIBERNATE]              = "hibernate",
        [HANDLE_HYBRID_SLEEP]           = "hybrid sleep",
        [HANDLE_SUSPEND_THEN_HIBERNATE] = "suspend and later hibernate",
        [HANDLE_SLEEP]                  = "sleep",
        [HANDLE_FACTORY_RESET]          = "perform a factory reset",
        [HANDLE_LOCK]                   = "be locked",
        [HANDLE_SECURE_ATTENTION_KEY]   = "handle the secure attention key",
};

DEFINE_STRING_TABLE_LOOKUP_TO_STRING(handle_action_verb, HandleAction);

/* These strings are sent out by PrepareForShutdownWithMetadata signals as metadata, so the values cannot
 * change as they are public APIs. */
static const char* const handle_action_table[_HANDLE_ACTION_MAX] = {
        [HANDLE_IGNORE]                 = "ignore",
        [HANDLE_POWEROFF]               = "poweroff",
        [HANDLE_REBOOT]                 = "reboot",
        [HANDLE_HALT]                   = "halt",
        [HANDLE_KEXEC]                  = "kexec",
        [HANDLE_SOFT_REBOOT]            = "soft-reboot",
        [HANDLE_SUSPEND]                = "suspend",
        [HANDLE_HIBERNATE]              = "hibernate",
        [HANDLE_HYBRID_SLEEP]           = "hybrid-sleep",
        [HANDLE_SUSPEND_THEN_HIBERNATE] = "suspend-then-hibernate",
        [HANDLE_SLEEP]                  = "sleep",
        [HANDLE_FACTORY_RESET]          = "factory-reset",
        [HANDLE_LOCK]                   = "lock",
        [HANDLE_SECURE_ATTENTION_KEY]   = "secure-attention-key",
};

DEFINE_STRING_TABLE_LOOKUP(handle_action, HandleAction);
DEFINE_CONFIG_PARSE_ENUM(config_parse_handle_action, handle_action, HandleAction);

int config_parse_handle_action_sleep(
                const char *unit,
                const char *filename,
                unsigned line,
                const char *section,
                unsigned section_line,
                const char *lvalue,
                int ltype,
                const char *rvalue,
                void *data,
                void *userdata) {

        HandleActionSleepMask *mask = ASSERT_PTR(data);
        _cleanup_strv_free_ char **actions = NULL;

        assert(filename);
        assert(lvalue);
        assert(rvalue);

        if (isempty(rvalue))
                goto empty;

        if (strv_split_full(&actions, rvalue, NULL, EXTRACT_UNQUOTE|EXTRACT_RETAIN_ESCAPE) < 0)
                return log_oom();

        *mask = 0;

        STRV_FOREACH(action, actions) {
                HandleAction a;

                a = handle_action_from_string(*action);
                if (a < 0) {
                        log_syntax(unit, LOG_WARNING, filename, line, a,
                                   "Failed to parse SleepOperation '%s', ignoring: %m", *action);
                        continue;
                }

                if (!HANDLE_ACTION_IS_SLEEP(a) || a == HANDLE_SLEEP) {
                        log_syntax(unit, LOG_WARNING, filename, line, 0,
                                   "HandleAction '%s' is not a sleep operation, ignoring: %m", *action);
                        continue;
                }

                SET_BIT(*mask, a);
        }

        if (*mask == 0)
                goto empty;

        return 0;

empty:
        *mask = HANDLE_ACTION_SLEEP_MASK_DEFAULT;
        return 0;
}
