1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2010 ProFUSION embedded systems
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 2 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 General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
23 #include <sys/types.h>
24 #include <sys/reboot.h>
25 #include <linux/reboot.h>
39 #define TIMEOUT_USEC (5 * USEC_PER_SEC)
40 #define FINALIZE_ATTEMPTS 50
42 static bool ignore_proc(pid_t pid) {
46 /* TODO: add more ignore rules here: device-mapper, etc */
51 static bool is_kernel_thread(pid_t pid)
58 snprintf(buf, sizeof(buf), "/proc/%lu/cmdline", (unsigned long)pid);
61 return true; /* not really, but has the desired effect */
63 count = fread(&c, 1, 1, f);
68 static int killall(int sign) {
71 unsigned int n_processes = 0;
73 if ((dir = opendir("/proc")) == NULL)
76 while ((d = readdir(dir))) {
79 if (parse_pid(d->d_name, &pid) < 0)
82 if (is_kernel_thread(pid))
88 if (kill(pid, sign) == 0)
91 log_warning("Could not kill %d: %m", pid);
99 static int send_signal(int sign) {
100 sigset_t mask, oldmask;
105 assert_se(sigemptyset(&mask) == 0);
106 assert_se(sigaddset(&mask, SIGCHLD) == 0);
107 if (sigprocmask(SIG_BLOCK, &mask, &oldmask) != 0)
110 if (kill(-1, SIGSTOP) < 0)
111 log_warning("Failed kill(-1, SIGSTOP): %m");
113 n_processes = killall(sign);
115 if (kill(-1, SIGCONT) < 0)
116 log_warning("Failed kill(-1, SIGCONT): %m");
118 if (n_processes <= 0)
121 until = now(CLOCK_MONOTONIC) + TIMEOUT_USEC;
123 usec_t n = now(CLOCK_MONOTONIC);
125 pid_t pid = waitpid(-1, NULL, WNOHANG);
129 else if (pid < 0 && errno == ECHILD) {
134 if (--n_processes == 0)
141 timespec_store(&ts, until - n);
142 if (sigtimedwait(&mask, NULL, &ts) != SIGCHLD)
144 log_warning("Failed: sigtimedwait did not return SIGCHLD: %m");
148 sigprocmask(SIG_SETMASK, &oldmask, NULL);
153 static int rescue_send_signal(int sign) {
154 sigset_t mask, oldmask;
160 sigaddset(&mask, SIGCHLD);
161 if (sigprocmask(SIG_BLOCK, &mask, &oldmask) != 0)
164 if (kill(-1, SIGSTOP) < 0)
165 log_warning("Failed kill(-1, SIGSTOP): %m");
169 log_warning("Failed kill(-1, %d): %m", sign);
171 if (kill(-1, SIGCONT) < 0)
172 log_warning("Failed kill(-1, SIGCONT): %m");
177 until = now(CLOCK_MONOTONIC) + TIMEOUT_USEC;
179 usec_t n = now(CLOCK_MONOTONIC);
181 pid_t pid = waitpid(-1, NULL, WNOHANG);
184 else if (pid < 0 && errno == ECHILD)
191 timespec_store(&ts, until - n);
192 if (sigtimedwait(&mask, NULL, &ts) != SIGCHLD)
194 log_warning("Failed: sigtimedwait did not return SIGCHLD: %m");
198 sigprocmask(SIG_SETMASK, &oldmask, NULL);
203 int main(int argc, char *argv[]) {
206 bool need_umount = true, need_swapoff = true, need_loop_detach = true, need_dm_detach = true;
207 bool killed_everbody = false;
209 log_parse_environment();
210 log_set_target(LOG_TARGET_CONSOLE); /* syslog will die if not gone yet */
214 log_error("Not executed by init (pid 1).");
220 log_error("Invalid number of arguments.");
225 if (streq(argv[1], "reboot"))
227 else if (streq(argv[1], "poweroff"))
229 else if (streq(argv[1], "halt"))
230 cmd = RB_HALT_SYSTEM;
231 else if (streq(argv[1], "kexec"))
232 cmd = LINUX_REBOOT_CMD_KEXEC;
234 log_error("Unknown action '%s'.", argv[1]);
239 /* lock us into memory */
240 if (mlockall(MCL_CURRENT|MCL_FUTURE) != 0)
241 log_warning("Cannot lock process memory: %m");
243 log_info("Sending SIGTERM to processes");
244 r = send_signal(SIGTERM);
246 log_warning("Cannot send SIGTERM to all process: %s", strerror(r));
248 log_info("Sending SIGKILL to processes");
249 r = send_signal(SIGKILL);
251 log_warning("Cannot send SIGKILL to all process: %s", strerror(r));
253 /* Unmount all mountpoints, swaps, and loopback devices */
254 for (retries = 0; retries < FINALIZE_ATTEMPTS; retries++) {
255 bool changed = false;
258 log_info("Unmounting filesystems.");
259 r = umount_all(&changed);
263 log_warning("Not all filesystems unmounted, %d left.", r);
265 log_error("Error unmounting filesystems: %s", strerror(-r));
269 log_info("Disabling swaps.");
270 r = swapoff_all(&changed);
272 need_swapoff = false;
274 log_warning("Not all swaps are off, %d left.", r);
276 log_error("Error turning off swaps: %s", strerror(-r));
279 if (need_loop_detach) {
280 log_info("Detaching loop devices.");
281 r = loopback_detach_all(&changed);
283 need_loop_detach = false;
285 log_warning("Not all loop devices detached, %d left.", r);
287 log_error("Error detaching loop devices: %s", strerror(-r));
290 if (need_dm_detach) {
291 log_info("Detaching DM devices.");
292 r = dm_detach_all(&changed);
294 need_dm_detach = false;
296 log_warning("Not all dm devices detached, %d left.", r);
298 log_error("Error detaching dm devices: %s", strerror(-r));
301 if (!need_umount && !need_swapoff && !need_loop_detach && !need_dm_detach)
305 /* If in this iteration we didn't manage to
306 * unmount/deactivate anything, we either kill more
307 * processes, or simply give up */
310 if (killed_everbody) {
311 /* Hmm, we already killed everybody,
312 * let's just give up */
313 log_error("Cannot finalize all filesystems and devices, giving up.");
317 log_warning("Cannot finalize filesystems and devices, trying to kill remaining processes.");
318 rescue_send_signal(SIGTERM);
319 rescue_send_signal(SIGKILL);
320 killed_everbody = true;
323 log_debug("Couldn't finalize filesystems and devices after %u retries, trying again.", retries+1);
326 if (retries >= FINALIZE_ATTEMPTS)
327 log_error("Too many interations, giving up.");
331 if (cmd == LINUX_REBOOT_CMD_KEXEC) {
332 /* We cheat and exec kexec to avoid doing all its work */
336 log_error("Could not fork: %m. Falling back to normal reboot.");
338 wait_for_terminate_and_warn("kexec", pid);
339 log_warning("kexec failed. Falling back to normal reboot.");
342 const char *args[5] = { KEXEC_BINARY_PATH, "-e", "-f", "-x", NULL };
343 execv(args[0], (char * const *) args);
351 log_error("Failed to invoke reboot(): %m");
356 log_error("Critical error while doing system shutdown: %s", strerror(-r));