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/>.
25 #include <sys/epoll.h>
28 #include "sd-daemon.h"
37 #include "bus-error.h"
39 #include "formats-util.h"
41 #define SERVER_FD_MAX 16
42 #define TIMEOUT_MSEC ((int) (DEFAULT_EXIT_USEC/USEC_PER_MSEC))
44 typedef struct Fifo Fifo
;
46 typedef struct Server
{
49 LIST_HEAD(Fifo
, fifos
);
62 struct init_request buffer
;
65 LIST_FIELDS(Fifo
, fifo
);
68 static const char *translate_runlevel(int runlevel
, bool *isolate
) {
74 { '0', SPECIAL_POWEROFF_TARGET
, false },
75 { '1', SPECIAL_RESCUE_TARGET
, true },
76 { 's', SPECIAL_RESCUE_TARGET
, true },
77 { 'S', SPECIAL_RESCUE_TARGET
, true },
78 { '2', SPECIAL_MULTI_USER_TARGET
, true },
79 { '3', SPECIAL_MULTI_USER_TARGET
, true },
80 { '4', SPECIAL_MULTI_USER_TARGET
, true },
81 { '5', SPECIAL_GRAPHICAL_TARGET
, true },
82 { '6', SPECIAL_REBOOT_TARGET
, false },
89 for (i
= 0; i
< ELEMENTSOF(table
); i
++)
90 if (table
[i
].runlevel
== runlevel
) {
91 *isolate
= table
[i
].isolate
;
92 if (runlevel
== '6' && kexec_loaded())
93 return SPECIAL_KEXEC_TARGET
;
94 return table
[i
].special
;
100 static void change_runlevel(Server
*s
, int runlevel
) {
102 _cleanup_bus_error_free_ sd_bus_error error
= SD_BUS_ERROR_NULL
;
104 bool isolate
= false;
109 target
= translate_runlevel(runlevel
, &isolate
);
111 log_warning("Got request for unknown runlevel %c, ignoring.", runlevel
);
118 mode
= "replace-irreversibly";
120 log_debug("Running request %s/start/%s", target
, mode
);
122 r
= sd_bus_call_method(
124 "org.freedesktop.systemd1",
125 "/org/freedesktop/systemd1",
126 "org.freedesktop.systemd1.Manager",
132 log_error("Failed to change runlevel: %s", bus_error_message(&error
, -r
));
137 static void request_process(Server
*s
, const struct init_request
*req
) {
141 if (req
->magic
!= INIT_MAGIC
) {
142 log_error("Got initctl request with invalid magic. Ignoring.");
148 case INIT_CMD_RUNLVL
:
149 if (!isprint(req
->runlevel
))
150 log_error("Got invalid runlevel. Ignoring.");
152 switch (req
->runlevel
) {
154 /* we are async anyway, so just use kill for reexec/reload */
157 if (kill(1, SIGTERM
) < 0)
158 log_error_errno(errno
, "kill() failed: %m");
160 /* The bus connection will be
161 * terminated if PID 1 is reexecuted,
162 * hence let's just exit here, and
163 * rely on that we'll be restarted on
164 * the next request */
170 if (kill(1, SIGHUP
) < 0)
171 log_error_errno(errno
, "kill() failed: %m");
175 change_runlevel(s
, req
->runlevel
);
179 case INIT_CMD_POWERFAIL
:
180 case INIT_CMD_POWERFAILNOW
:
181 case INIT_CMD_POWEROK
:
182 log_warning("Received UPS/power initctl request. This is not implemented in systemd. Upgrade your UPS daemon!");
185 case INIT_CMD_CHANGECONS
:
186 log_warning("Received console change initctl request. This is not implemented in systemd.");
189 case INIT_CMD_SETENV
:
190 case INIT_CMD_UNSETENV
:
191 log_warning("Received environment initctl request. This is not implemented in systemd.");
195 log_warning("Received unknown initctl request. Ignoring.");
200 static int fifo_process(Fifo
*f
) {
207 ((uint8_t*) &f
->buffer
) + f
->bytes_read
,
208 sizeof(f
->buffer
) - f
->bytes_read
);
213 log_warning_errno(errno
, "Failed to read from fifo: %m");
218 assert(f
->bytes_read
<= sizeof(f
->buffer
));
220 if (f
->bytes_read
== sizeof(f
->buffer
)) {
221 request_process(f
->server
, &f
->buffer
);
228 static void fifo_free(Fifo
*f
) {
232 assert(f
->server
->n_fifos
> 0);
233 f
->server
->n_fifos
--;
234 LIST_REMOVE(fifo
, f
->server
->fifos
, f
);
239 epoll_ctl(f
->server
->epoll_fd
, EPOLL_CTL_DEL
, f
->fd
, NULL
);
247 static void server_done(Server
*s
) {
253 safe_close(s
->epoll_fd
);
256 sd_bus_flush(s
->bus
);
257 sd_bus_unref(s
->bus
);
261 static int server_init(Server
*s
, unsigned n_sockets
) {
266 assert(n_sockets
> 0);
270 s
->epoll_fd
= epoll_create1(EPOLL_CLOEXEC
);
271 if (s
->epoll_fd
< 0) {
272 r
= log_error_errno(errno
,
273 "Failed to create epoll object: %m");
277 for (i
= 0; i
< n_sockets
; i
++) {
278 struct epoll_event ev
;
282 fd
= SD_LISTEN_FDS_START
+i
;
284 r
= sd_is_fifo(fd
, NULL
);
286 log_error_errno(r
, "Failed to determine file descriptor type: %m");
291 log_error("Wrong file descriptor type.");
299 log_error_errno(errno
, "Failed to create fifo object: %m");
308 if (epoll_ctl(s
->epoll_fd
, EPOLL_CTL_ADD
, fd
, &ev
) < 0) {
311 log_error_errno(errno
, "Failed to add fifo fd to epoll object: %m");
316 LIST_PREPEND(fifo
, s
->fifos
, f
);
321 r
= bus_connect_system_systemd(&s
->bus
);
323 log_error_errno(r
, "Failed to get D-Bus connection: %m");
336 static int process_event(Server
*s
, struct epoll_event
*ev
) {
342 if (!(ev
->events
& EPOLLIN
)) {
343 log_info("Got invalid event from epoll. (3)");
347 f
= (Fifo
*) ev
->data
.ptr
;
350 log_info_errno(r
, "Got error on fifo: %m");
358 int main(int argc
, char *argv
[]) {
360 int r
= EXIT_FAILURE
, n
;
362 if (getppid() != 1) {
363 log_error("This program should be invoked by init only.");
368 log_error("This program does not take arguments.");
372 log_set_target(LOG_TARGET_AUTO
);
373 log_parse_environment();
378 n
= sd_listen_fds(true);
380 log_error_errno(r
, "Failed to read listening file descriptors from environment: %m");
384 if (n
<= 0 || n
> SERVER_FD_MAX
) {
385 log_error("No or too many file descriptors passed.");
389 if (server_init(&server
, (unsigned) n
) < 0)
392 log_debug("systemd-initctl running as pid "PID_FMT
, getpid());
396 "STATUS=Processing requests...");
398 while (!server
.quit
) {
399 struct epoll_event event
;
402 k
= epoll_wait(server
.epoll_fd
, &event
, 1, TIMEOUT_MSEC
);
406 log_error_errno(errno
, "epoll_wait() failed: %m");
413 if (process_event(&server
, &event
) < 0)
419 log_debug("systemd-initctl stopped as pid "PID_FMT
, getpid());
424 "STATUS=Shutting down...");
426 server_done(&server
);