527b7a421f
Change cunescape() to return a normal error code, so that we can distuingish OOM errors from parse errors. This also adds a flags parameter to control whether "relaxed" or normal parsing shall be done. If set no parse failures are generated, and the only reason why cunescape() can fail is OOM.
472 lines
12 KiB
C
472 lines
12 KiB
C
/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
|
|
|
|
/***
|
|
This file is part of systemd.
|
|
|
|
Copyright 2012 Lennart Poettering
|
|
|
|
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 <errno.h>
|
|
#include <fcntl.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include "util.h"
|
|
#include "mkdir.h"
|
|
#include "logind-inhibit.h"
|
|
#include "fileio.h"
|
|
|
|
Inhibitor* inhibitor_new(Manager *m, const char* id) {
|
|
Inhibitor *i;
|
|
|
|
assert(m);
|
|
|
|
i = new0(Inhibitor, 1);
|
|
if (!i)
|
|
return NULL;
|
|
|
|
i->state_file = strappend("/run/systemd/inhibit/", id);
|
|
if (!i->state_file) {
|
|
free(i);
|
|
return NULL;
|
|
}
|
|
|
|
i->id = basename(i->state_file);
|
|
|
|
if (hashmap_put(m->inhibitors, i->id, i) < 0) {
|
|
free(i->state_file);
|
|
free(i);
|
|
return NULL;
|
|
}
|
|
|
|
i->manager = m;
|
|
i->fifo_fd = -1;
|
|
|
|
return i;
|
|
}
|
|
|
|
void inhibitor_free(Inhibitor *i) {
|
|
assert(i);
|
|
|
|
hashmap_remove(i->manager->inhibitors, i->id);
|
|
|
|
inhibitor_remove_fifo(i);
|
|
|
|
free(i->who);
|
|
free(i->why);
|
|
|
|
if (i->state_file) {
|
|
unlink(i->state_file);
|
|
free(i->state_file);
|
|
}
|
|
|
|
free(i);
|
|
}
|
|
|
|
int inhibitor_save(Inhibitor *i) {
|
|
_cleanup_free_ char *temp_path = NULL;
|
|
_cleanup_fclose_ FILE *f = NULL;
|
|
int r;
|
|
|
|
assert(i);
|
|
|
|
r = mkdir_safe_label("/run/systemd/inhibit", 0755, 0, 0);
|
|
if (r < 0)
|
|
goto finish;
|
|
|
|
r = fopen_temporary(i->state_file, &f, &temp_path);
|
|
if (r < 0)
|
|
goto finish;
|
|
|
|
fchmod(fileno(f), 0644);
|
|
|
|
fprintf(f,
|
|
"# This is private data. Do not parse.\n"
|
|
"WHAT=%s\n"
|
|
"MODE=%s\n"
|
|
"UID="UID_FMT"\n"
|
|
"PID="PID_FMT"\n",
|
|
inhibit_what_to_string(i->what),
|
|
inhibit_mode_to_string(i->mode),
|
|
i->uid,
|
|
i->pid);
|
|
|
|
if (i->who) {
|
|
_cleanup_free_ char *cc = NULL;
|
|
|
|
cc = cescape(i->who);
|
|
if (!cc)
|
|
r = -ENOMEM;
|
|
else
|
|
fprintf(f, "WHO=%s\n", cc);
|
|
}
|
|
|
|
if (i->why) {
|
|
_cleanup_free_ char *cc = NULL;
|
|
|
|
cc = cescape(i->why);
|
|
if (!cc)
|
|
r = -ENOMEM;
|
|
else
|
|
fprintf(f, "WHY=%s\n", cc);
|
|
}
|
|
|
|
if (i->fifo_path)
|
|
fprintf(f, "FIFO=%s\n", i->fifo_path);
|
|
|
|
fflush(f);
|
|
|
|
if (ferror(f) || rename(temp_path, i->state_file) < 0) {
|
|
r = -errno;
|
|
unlink(i->state_file);
|
|
unlink(temp_path);
|
|
}
|
|
|
|
finish:
|
|
if (r < 0)
|
|
log_error_errno(r, "Failed to save inhibit data %s: %m", i->state_file);
|
|
|
|
return r;
|
|
}
|
|
|
|
int inhibitor_start(Inhibitor *i) {
|
|
assert(i);
|
|
|
|
if (i->started)
|
|
return 0;
|
|
|
|
dual_timestamp_get(&i->since);
|
|
|
|
log_debug("Inhibitor %s (%s) pid="PID_FMT" uid="UID_FMT" mode=%s started.",
|
|
strna(i->who), strna(i->why),
|
|
i->pid, i->uid,
|
|
inhibit_mode_to_string(i->mode));
|
|
|
|
inhibitor_save(i);
|
|
|
|
i->started = true;
|
|
|
|
manager_send_changed(i->manager, i->mode == INHIBIT_BLOCK ? "BlockInhibited" : "DelayInhibited", NULL);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int inhibitor_stop(Inhibitor *i) {
|
|
assert(i);
|
|
|
|
if (i->started)
|
|
log_debug("Inhibitor %s (%s) pid="PID_FMT" uid="UID_FMT" mode=%s stopped.",
|
|
strna(i->who), strna(i->why),
|
|
i->pid, i->uid,
|
|
inhibit_mode_to_string(i->mode));
|
|
|
|
if (i->state_file)
|
|
unlink(i->state_file);
|
|
|
|
i->started = false;
|
|
|
|
manager_send_changed(i->manager, i->mode == INHIBIT_BLOCK ? "BlockInhibited" : "DelayInhibited", NULL);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int inhibitor_load(Inhibitor *i) {
|
|
|
|
_cleanup_free_ char
|
|
*what = NULL,
|
|
*uid = NULL,
|
|
*pid = NULL,
|
|
*who = NULL,
|
|
*why = NULL,
|
|
*mode = NULL;
|
|
|
|
InhibitWhat w;
|
|
InhibitMode mm;
|
|
char *cc;
|
|
int r;
|
|
|
|
r = parse_env_file(i->state_file, NEWLINE,
|
|
"WHAT", &what,
|
|
"UID", &uid,
|
|
"PID", &pid,
|
|
"WHO", &who,
|
|
"WHY", &why,
|
|
"MODE", &mode,
|
|
"FIFO", &i->fifo_path,
|
|
NULL);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
w = what ? inhibit_what_from_string(what) : 0;
|
|
if (w >= 0)
|
|
i->what = w;
|
|
|
|
mm = mode ? inhibit_mode_from_string(mode) : INHIBIT_BLOCK;
|
|
if (mm >= 0)
|
|
i->mode = mm;
|
|
|
|
if (uid) {
|
|
r = parse_uid(uid, &i->uid);
|
|
if (r < 0)
|
|
return r;
|
|
}
|
|
|
|
if (pid) {
|
|
r = parse_pid(pid, &i->pid);
|
|
if (r < 0)
|
|
return r;
|
|
}
|
|
|
|
if (who) {
|
|
r = cunescape(who, 0, &cc);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
free(i->who);
|
|
i->who = cc;
|
|
}
|
|
|
|
if (why) {
|
|
r = cunescape(why, 0, &cc);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
free(i->why);
|
|
i->why = cc;
|
|
}
|
|
|
|
if (i->fifo_path) {
|
|
int fd;
|
|
|
|
fd = inhibitor_create_fifo(i);
|
|
safe_close(fd);
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static int inhibitor_dispatch_fifo(sd_event_source *s, int fd, uint32_t revents, void *userdata) {
|
|
Inhibitor *i = userdata;
|
|
|
|
assert(s);
|
|
assert(fd == i->fifo_fd);
|
|
assert(i);
|
|
|
|
inhibitor_stop(i);
|
|
inhibitor_free(i);
|
|
|
|
return 0;
|
|
}
|
|
|
|
int inhibitor_create_fifo(Inhibitor *i) {
|
|
int r;
|
|
|
|
assert(i);
|
|
|
|
/* Create FIFO */
|
|
if (!i->fifo_path) {
|
|
r = mkdir_safe_label("/run/systemd/inhibit", 0755, 0, 0);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
i->fifo_path = strjoin("/run/systemd/inhibit/", i->id, ".ref", NULL);
|
|
if (!i->fifo_path)
|
|
return -ENOMEM;
|
|
|
|
if (mkfifo(i->fifo_path, 0600) < 0 && errno != EEXIST)
|
|
return -errno;
|
|
}
|
|
|
|
/* Open reading side */
|
|
if (i->fifo_fd < 0) {
|
|
i->fifo_fd = open(i->fifo_path, O_RDONLY|O_CLOEXEC|O_NDELAY);
|
|
if (i->fifo_fd < 0)
|
|
return -errno;
|
|
}
|
|
|
|
if (!i->event_source) {
|
|
r = sd_event_add_io(i->manager->event, &i->event_source, i->fifo_fd, 0, inhibitor_dispatch_fifo, i);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
r = sd_event_source_set_priority(i->event_source, SD_EVENT_PRIORITY_IDLE);
|
|
if (r < 0)
|
|
return r;
|
|
}
|
|
|
|
/* Open writing side */
|
|
r = open(i->fifo_path, O_WRONLY|O_CLOEXEC|O_NDELAY);
|
|
if (r < 0)
|
|
return -errno;
|
|
|
|
return r;
|
|
}
|
|
|
|
void inhibitor_remove_fifo(Inhibitor *i) {
|
|
assert(i);
|
|
|
|
i->event_source = sd_event_source_unref(i->event_source);
|
|
i->fifo_fd = safe_close(i->fifo_fd);
|
|
|
|
if (i->fifo_path) {
|
|
unlink(i->fifo_path);
|
|
free(i->fifo_path);
|
|
i->fifo_path = NULL;
|
|
}
|
|
}
|
|
|
|
InhibitWhat manager_inhibit_what(Manager *m, InhibitMode mm) {
|
|
Inhibitor *i;
|
|
Iterator j;
|
|
InhibitWhat what = 0;
|
|
|
|
assert(m);
|
|
|
|
HASHMAP_FOREACH(i, m->inhibitors, j)
|
|
if (i->mode == mm)
|
|
what |= i->what;
|
|
|
|
return what;
|
|
}
|
|
|
|
static int pid_is_active(Manager *m, pid_t pid) {
|
|
Session *s;
|
|
int r;
|
|
|
|
r = manager_get_session_by_pid(m, pid, &s);
|
|
if (r < 0)
|
|
return r;
|
|
|
|
/* If there's no session assigned to it, then it's globally
|
|
* active on all ttys */
|
|
if (r == 0)
|
|
return 1;
|
|
|
|
return session_is_active(s);
|
|
}
|
|
|
|
bool manager_is_inhibited(
|
|
Manager *m,
|
|
InhibitWhat w,
|
|
InhibitMode mm,
|
|
dual_timestamp *since,
|
|
bool ignore_inactive,
|
|
bool ignore_uid,
|
|
uid_t uid,
|
|
Inhibitor **offending) {
|
|
|
|
Inhibitor *i;
|
|
Iterator j;
|
|
struct dual_timestamp ts = { 0, 0 };
|
|
bool inhibited = false;
|
|
|
|
assert(m);
|
|
assert(w > 0 && w < _INHIBIT_WHAT_MAX);
|
|
|
|
HASHMAP_FOREACH(i, m->inhibitors, j) {
|
|
if (!(i->what & w))
|
|
continue;
|
|
|
|
if (i->mode != mm)
|
|
continue;
|
|
|
|
if (ignore_inactive && pid_is_active(m, i->pid) <= 0)
|
|
continue;
|
|
|
|
if (ignore_uid && i->uid == uid)
|
|
continue;
|
|
|
|
if (!inhibited ||
|
|
i->since.monotonic < ts.monotonic)
|
|
ts = i->since;
|
|
|
|
inhibited = true;
|
|
|
|
if (offending)
|
|
*offending = i;
|
|
}
|
|
|
|
if (since)
|
|
*since = ts;
|
|
|
|
return inhibited;
|
|
}
|
|
|
|
const char *inhibit_what_to_string(InhibitWhat w) {
|
|
static thread_local char buffer[97];
|
|
char *p;
|
|
|
|
if (w < 0 || w >= _INHIBIT_WHAT_MAX)
|
|
return NULL;
|
|
|
|
p = buffer;
|
|
if (w & INHIBIT_SHUTDOWN)
|
|
p = stpcpy(p, "shutdown:");
|
|
if (w & INHIBIT_SLEEP)
|
|
p = stpcpy(p, "sleep:");
|
|
if (w & INHIBIT_IDLE)
|
|
p = stpcpy(p, "idle:");
|
|
if (w & INHIBIT_HANDLE_POWER_KEY)
|
|
p = stpcpy(p, "handle-power-key:");
|
|
if (w & INHIBIT_HANDLE_SUSPEND_KEY)
|
|
p = stpcpy(p, "handle-suspend-key:");
|
|
if (w & INHIBIT_HANDLE_HIBERNATE_KEY)
|
|
p = stpcpy(p, "handle-hibernate-key:");
|
|
if (w & INHIBIT_HANDLE_LID_SWITCH)
|
|
p = stpcpy(p, "handle-lid-switch:");
|
|
|
|
if (p > buffer)
|
|
*(p-1) = 0;
|
|
else
|
|
*p = 0;
|
|
|
|
return buffer;
|
|
}
|
|
|
|
InhibitWhat inhibit_what_from_string(const char *s) {
|
|
InhibitWhat what = 0;
|
|
const char *word, *state;
|
|
size_t l;
|
|
|
|
FOREACH_WORD_SEPARATOR(word, l, s, ":", state) {
|
|
if (l == 8 && strneq(word, "shutdown", l))
|
|
what |= INHIBIT_SHUTDOWN;
|
|
else if (l == 5 && strneq(word, "sleep", l))
|
|
what |= INHIBIT_SLEEP;
|
|
else if (l == 4 && strneq(word, "idle", l))
|
|
what |= INHIBIT_IDLE;
|
|
else if (l == 16 && strneq(word, "handle-power-key", l))
|
|
what |= INHIBIT_HANDLE_POWER_KEY;
|
|
else if (l == 18 && strneq(word, "handle-suspend-key", l))
|
|
what |= INHIBIT_HANDLE_SUSPEND_KEY;
|
|
else if (l == 20 && strneq(word, "handle-hibernate-key", l))
|
|
what |= INHIBIT_HANDLE_HIBERNATE_KEY;
|
|
else if (l == 17 && strneq(word, "handle-lid-switch", l))
|
|
what |= INHIBIT_HANDLE_LID_SWITCH;
|
|
else
|
|
return _INHIBIT_WHAT_INVALID;
|
|
}
|
|
|
|
return what;
|
|
}
|
|
|
|
static const char* const inhibit_mode_table[_INHIBIT_MODE_MAX] = {
|
|
[INHIBIT_BLOCK] = "block",
|
|
[INHIBIT_DELAY] = "delay"
|
|
};
|
|
|
|
DEFINE_STRING_TABLE_LOOKUP(inhibit_mode, InhibitMode);
|