d1fab3fe88ae873b26b50359758776ad9e31968eZbigniew Jędrzejewski-Szmek

#include <sys/socket.h>

#include <sys/types.h>

#include <assert.h>

#include <time.h>

#include <string.h>

#include <stdio.h>

#include <errno.h>

#include <unistd.h>

#include <sys/poll.h>

#include <sys/epoll.h>

#include <sys/un.h>

#include <fcntl.h>

#include <ctype.h>

#include <dbus/dbus.h>

#include "util.h"

#include "log.h"

#include "list.h"

#include "initreq.h"

#include "special.h"

#include "sd-daemon.h"

#define SERVER_FD_MAX 16

#define TIMEOUT ((int) (10*MSEC_PER_SEC))

typedef struct Fifo Fifo;

typedef struct Server {

int epoll_fd;

LIST_HEAD(Fifo, fifos);

unsigned n_fifos;

DBusConnection *bus;

} Server;

struct Fifo {

Server *server;

int fd;

struct init_request buffer;

size_t bytes_read;

LIST_FIELDS(Fifo, fifo);

};

static const char *translate_runlevel(int runlevel) {

static const struct {

const int runlevel;

const char *special;

} table[] = {

{ '0', SPECIAL_POWEROFF_TARGET },

{ '1', SPECIAL_RESCUE_TARGET },

{ 's', SPECIAL_RESCUE_TARGET },

{ 'S', SPECIAL_RESCUE_TARGET },

{ '2', SPECIAL_RUNLEVEL2_TARGET },

{ '3', SPECIAL_RUNLEVEL3_TARGET },

{ '4', SPECIAL_RUNLEVEL4_TARGET },

{ '5', SPECIAL_RUNLEVEL5_TARGET },

{ '6', SPECIAL_REBOOT_TARGET },

};

unsigned i;

for (i = 0; i < ELEMENTSOF(table); i++)

if (table[i].runlevel == runlevel)

return table[i].special;

return NULL;

}

static void change_runlevel(Server *s, int runlevel) {

const char *target;

DBusMessage *m = NULL, *reply = NULL;

DBusError error;

const char *replace = "replace";

assert(s);

dbus_error_init(&error);

if (!(target = translate_runlevel(runlevel))) {

log_warning("Got request for unknown runlevel %c, ignoring.", runlevel);

goto finish;

}

log_debug("Running request %s", target);

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, &target,

DBUS_TYPE_STRING, &replace,

DBUS_TYPE_INVALID)) {

log_error("Could not attach target and flag information to signal message.");

goto finish;

}

if (!(reply = dbus_connection_send_with_reply_and_block(s->bus, m, -1, &error))) {

log_error("Failed to start unit: %s", error.message);

goto finish;

}

finish:

if (m)

dbus_message_unref(m);

if (reply)

dbus_message_unref(reply);

dbus_error_free(&error);

}

static void request_process(Server *s, const struct init_request *req) {

assert(s);

assert(req);

if (req->magic != INIT_MAGIC) {

log_error("Got initctl request with invalid magic. Ignoring.");

return;

}

switch (req->cmd) {

case INIT_CMD_RUNLVL:

if (!isprint(req->runlevel))

log_error("Got invalid runlevel. Ignoring.");

else

change_runlevel(s, req->runlevel);

return;

case INIT_CMD_POWERFAIL:

case INIT_CMD_POWERFAILNOW:

case INIT_CMD_POWEROK:

log_warning("Received UPS/power initctl request. This is not implemented in systemd. Upgrade your UPS daemon!");

return;

case INIT_CMD_CHANGECONS:

log_warning("Received console change initctl request. This is not implemented in systemd.");

return;

case INIT_CMD_SETENV:

case INIT_CMD_UNSETENV:

log_warning("Received environment initctl request. This is not implemented in systemd.");

return;

default:

log_warning("Received unknown initctl request. Ignoring.");

return;

}

}

static int fifo_process(Fifo *f) {

ssize_t l;

assert(f);

errno = EIO;

if ((l = read(f->fd, ((uint8_t*) &f->buffer) + f->bytes_read, sizeof(f->buffer) - f->bytes_read)) <= 0) {

if (errno == EAGAIN)

return 0;

log_warning("Failed to read from fifo: %s", strerror(errno));

return -1;

}

f->bytes_read += l;

assert(f->bytes_read <= sizeof(f->buffer));

if (f->bytes_read == sizeof(f->buffer)) {

request_process(f->server, &f->buffer);

f->bytes_read = 0;

}

return 0;

}

static void fifo_free(Fifo *f) {

assert(f);

if (f->server) {

assert(f->server->n_fifos > 0);

f->server->n_fifos--;

LIST_REMOVE(Fifo, fifo, f->server->fifos, f);

}

if (f->fd >= 0) {

if (f->server)

epoll_ctl(f->server->epoll_fd, EPOLL_CTL_DEL, f->fd, NULL);

close_nointr_nofail(f->fd);

}

free(f);

}

static void server_done(Server *s) {

assert(s);

while (s->fifos)

fifo_free(s->fifos);

if (s->epoll_fd >= 0)

close_nointr_nofail(s->epoll_fd);

if (s->bus) {

dbus_connection_set_exit_on_disconnect(s->bus, FALSE);

dbus_connection_unref(s->bus);

}

}

static int server_init(Server *s, unsigned n_sockets) {

int r;

unsigned i;

DBusError error;

assert(s);

assert(n_sockets > 0);

dbus_error_init(&error);

zero(*s);

if ((s->epoll_fd = epoll_create1(EPOLL_CLOEXEC)) < 0) {

r = -errno;

log_error("Failed to create epoll object: %s", strerror(errno));

goto fail;

}

for (i = 0; i < n_sockets; i++) {

struct epoll_event ev;

Fifo *f;

int fd;

fd = SD_LISTEN_FDS_START+i;

if ((r = sd_is_fifo(fd, NULL)) < 0) {

log_error("Failed to determine file descriptor type: %s", strerror(-r));

goto fail;

}

if (!r) {

log_error("Wrong file descriptor type.");

r = -EINVAL;

goto fail;

}

if (!(f = new0(Fifo, 1))) {

r = -ENOMEM;

log_error("Failed to create fifo object: %s", strerror(errno));

goto fail;

}

f->fd = -1;

zero(ev);

ev.events = EPOLLIN;

ev.data.ptr = f;

if (epoll_ctl(s->epoll_fd, EPOLL_CTL_ADD, fd, &ev) < 0) {

r = -errno;

fifo_free(f);

log_error("Failed to add fifo fd to epoll object: %s", strerror(errno));

goto fail;

}

f->fd = SD_LISTEN_FDS_START+i;

LIST_PREPEND(Fifo, fifo, s->fifos, f);

f->server = s;

s->n_fifos ++;

}

if (!(s->bus = dbus_bus_get(DBUS_BUS_SYSTEM, &error))) {

log_error("Failed to get D-Bus connection: %s", error.message);

goto fail;

}

return 0;

fail:

server_done(s);

dbus_error_free(&error);

return r;

}

static int process_event(Server *s, struct epoll_event *ev) {

int r;

Fifo *f;

assert(s);

if (!(ev->events & EPOLLIN)) {

log_info("Got invalid event from epoll. (3)");

return -EIO;

}

f = (Fifo*) ev->data.ptr;

if ((r = fifo_process(f)) < 0) {

log_info("Got error on fifo: %s", strerror(-r));

fifo_free(f);

return r;

}

return 0;

}

int main(int argc, char *argv[]) {

Server server;

int r = 3, n;

log_set_target(LOG_TARGET_SYSLOG_OR_KMSG);

log_parse_environment();

log_info("systemd-initctl running as pid %llu", (unsigned long long) getpid());

if ((n = sd_listen_fds(true)) < 0) {

log_error("Failed to read listening file descriptors from environment: %s", strerror(-r));

return 1;

}

if (n <= 0 || n > SERVER_FD_MAX) {

log_error("No or too many file descriptors passed.");

return 2;

}

if (server_init(&server, (unsigned) n) < 0)

return 2;

sd_notify(false,

"READY=1\n"

"STATUS=Processing requests...");

for (;;) {

struct epoll_event event;

int k;

if ((k = epoll_wait(server.epoll_fd,

&event, 1,

TIMEOUT)) < 0) {

if (errno == EINTR)

continue;

log_error("epoll_wait() failed: %s", strerror(errno));

goto fail;

}

if (k <= 0)

break;

if ((k = process_event(&server, &event)) < 0)

goto fail;

}

r = 0;

fail:

sd_notify(false,

"STATUS=Shutting down...");

server_done(&server);

log_info("systemd-initctl stopped as pid %llu", (unsigned long long) getpid());

dbus_shutdown();

return r;

}