1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2010 Lennart Poettering
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
22 #include <sys/socket.h>
23 #include <sys/types.h>
31 #include <sys/epoll.h>
36 #include "sd-daemon.h"
45 #include "bus-error.h"
48 #define SERVER_FD_MAX 16
49 #define TIMEOUT_MSEC ((int) (DEFAULT_EXIT_USEC/USEC_PER_MSEC))
51 typedef struct Fifo Fifo;
53 typedef struct Server {
56 LIST_HEAD(Fifo, fifos);
69 struct init_request buffer;
72 LIST_FIELDS(Fifo, fifo);
75 static const char *translate_runlevel(int runlevel, bool *isolate) {
81 { '0', SPECIAL_POWEROFF_TARGET, false },
82 { '1', SPECIAL_RESCUE_TARGET, true },
83 { 's', SPECIAL_RESCUE_TARGET, true },
84 { 'S', SPECIAL_RESCUE_TARGET, true },
85 { '2', SPECIAL_RUNLEVEL2_TARGET, true },
86 { '3', SPECIAL_RUNLEVEL3_TARGET, true },
87 { '4', SPECIAL_RUNLEVEL4_TARGET, true },
88 { '5', SPECIAL_RUNLEVEL5_TARGET, true },
89 { '6', SPECIAL_REBOOT_TARGET, false },
96 for (i = 0; i < ELEMENTSOF(table); i++)
97 if (table[i].runlevel == runlevel) {
98 *isolate = table[i].isolate;
99 if (runlevel == '6' && kexec_loaded())
100 return SPECIAL_KEXEC_TARGET;
101 return table[i].special;
107 static void change_runlevel(Server *s, int runlevel) {
109 _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
111 bool isolate = false;
116 target = translate_runlevel(runlevel, &isolate);
118 log_warning("Got request for unknown runlevel %c, ignoring.", runlevel);
125 mode = "replace-irreversibly";
127 log_debug("Running request %s/start/%s", target, mode);
129 r = sd_bus_call_method(
131 "org.freedesktop.systemd1",
132 "/org/freedesktop/systemd1",
133 "org.freedesktop.systemd1.Manager",
139 log_error("Failed to change runlevel: %s", bus_error_message(&error, -r));
144 static void request_process(Server *s, const struct init_request *req) {
148 if (req->magic != INIT_MAGIC) {
149 log_error("Got initctl request with invalid magic. Ignoring.");
155 case INIT_CMD_RUNLVL:
156 if (!isprint(req->runlevel))
157 log_error("Got invalid runlevel. Ignoring.");
159 switch (req->runlevel) {
161 /* we are async anyway, so just use kill for reexec/reload */
164 if (kill(1, SIGTERM) < 0)
165 log_error("kill() failed: %m");
167 /* The bus connection will be
168 * terminated if PID 1 is reexecuted,
169 * hence let's just exit here, and
170 * rely on that we'll be restarted on
171 * the next request */
177 if (kill(1, SIGHUP) < 0)
178 log_error("kill() failed: %m");
182 change_runlevel(s, req->runlevel);
186 case INIT_CMD_POWERFAIL:
187 case INIT_CMD_POWERFAILNOW:
188 case INIT_CMD_POWEROK:
189 log_warning("Received UPS/power initctl request. This is not implemented in systemd. Upgrade your UPS daemon!");
192 case INIT_CMD_CHANGECONS:
193 log_warning("Received console change initctl request. This is not implemented in systemd.");
196 case INIT_CMD_SETENV:
197 case INIT_CMD_UNSETENV:
198 log_warning("Received environment initctl request. This is not implemented in systemd.");
202 log_warning("Received unknown initctl request. Ignoring.");
207 static int fifo_process(Fifo *f) {
214 ((uint8_t*) &f->buffer) + f->bytes_read,
215 sizeof(f->buffer) - f->bytes_read);
220 log_warning("Failed to read from fifo: %m");
225 assert(f->bytes_read <= sizeof(f->buffer));
227 if (f->bytes_read == sizeof(f->buffer)) {
228 request_process(f->server, &f->buffer);
235 static void fifo_free(Fifo *f) {
239 assert(f->server->n_fifos > 0);
240 f->server->n_fifos--;
241 LIST_REMOVE(fifo, f->server->fifos, f);
246 epoll_ctl(f->server->epoll_fd, EPOLL_CTL_DEL, f->fd, NULL);
248 close_nointr_nofail(f->fd);
254 static void server_done(Server *s) {
260 if (s->epoll_fd >= 0)
261 close_nointr_nofail(s->epoll_fd);
264 sd_bus_flush(s->bus);
265 sd_bus_unref(s->bus);
269 static int server_init(Server *s, unsigned n_sockets) {
274 assert(n_sockets > 0);
278 s->epoll_fd = epoll_create1(EPOLL_CLOEXEC);
279 if (s->epoll_fd < 0) {
281 log_error("Failed to create epoll object: %m");
285 for (i = 0; i < n_sockets; i++) {
286 struct epoll_event ev;
290 fd = SD_LISTEN_FDS_START+i;
292 r = sd_is_fifo(fd, NULL);
294 log_error("Failed to determine file descriptor type: %s",
300 log_error("Wrong file descriptor type.");
308 log_error("Failed to create fifo object: %m");
317 if (epoll_ctl(s->epoll_fd, EPOLL_CTL_ADD, fd, &ev) < 0) {
320 log_error("Failed to add fifo fd to epoll object: %m");
325 LIST_PREPEND(fifo, s->fifos, f);
330 r = bus_open_system_systemd(&s->bus);
332 log_error("Failed to get D-Bus connection: %s", strerror(-r));
345 static int process_event(Server *s, struct epoll_event *ev) {
351 if (!(ev->events & EPOLLIN)) {
352 log_info("Got invalid event from epoll. (3)");
356 f = (Fifo*) ev->data.ptr;
359 log_info("Got error on fifo: %s", strerror(-r));
367 int main(int argc, char *argv[]) {
369 int r = EXIT_FAILURE, n;
371 if (getppid() != 1) {
372 log_error("This program should be invoked by init only.");
377 log_error("This program does not take arguments.");
381 log_set_target(LOG_TARGET_AUTO);
382 log_parse_environment();
387 if ((n = sd_listen_fds(true)) < 0) {
388 log_error("Failed to read listening file descriptors from environment: %s", strerror(-r));
392 if (n <= 0 || n > SERVER_FD_MAX) {
393 log_error("No or too many file descriptors passed.");
397 if (server_init(&server, (unsigned) n) < 0)
400 log_debug("systemd-initctl running as pid %lu", (unsigned long) getpid());
404 "STATUS=Processing requests...");
406 while (!server.quit) {
407 struct epoll_event event;
410 if ((k = epoll_wait(server.epoll_fd,
412 TIMEOUT_MSEC)) < 0) {
417 log_error("epoll_wait() failed: %m");
424 if (process_event(&server, &event) < 0)
430 log_debug("systemd-initctl stopped as pid %lu", (unsigned long) getpid());
434 "STATUS=Shutting down...");
436 server_done(&server);