X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?a=blobdiff_plain;f=src%2Fshutdown.c;h=d157e0fbfeaf4db91cea33666f9e2ab5c59c3633;hb=1124fe6f01b1d59d016c238026f20380f38d98dc;hp=528b30b98f0ea6e3d66318f38bf810336fe9d2fd;hpb=7cb1094acefff2a89dc842d84eb94a8afcac67fd;p=elogind.git diff --git a/src/shutdown.c b/src/shutdown.c index 528b30b98..d157e0fbf 100644 --- a/src/shutdown.c +++ b/src/shutdown.c @@ -41,35 +41,53 @@ #include "log.h" #include "umount.h" #include "util.h" +#include "virt.h" #define TIMEOUT_USEC (5 * USEC_PER_SEC) #define FINALIZE_ATTEMPTS 50 -#define pivot_root(new_root,put_old) syscall(SYS_pivot_root,new_root,put_old) static bool ignore_proc(pid_t pid) { + char buf[PATH_MAX]; + FILE *f; + char c; + size_t count; + uid_t uid; + int r; + + /* We are PID 1, let's not commit suicide */ if (pid == 1) return true; - /* TODO: add more ignore rules here: device-mapper, etc */ + r = get_process_uid(pid, &uid); + if (r < 0) + return true; /* not really, but better safe than sorry */ - return false; -} + /* Non-root processes otherwise are always subject to be killed */ + if (uid != 0) + return false; -static bool is_kernel_thread(pid_t pid) -{ - char buf[PATH_MAX]; - FILE *f; - char c; - size_t count; + snprintf(buf, sizeof(buf), "/proc/%lu/cmdline", (unsigned long) pid); + char_array_0(buf); - snprintf(buf, sizeof(buf), "/proc/%lu/cmdline", (unsigned long)pid); f = fopen(buf, "re"); if (!f) return true; /* not really, but has the desired effect */ count = fread(&c, 1, 1, f); fclose(f); - return count != 1; + + /* Kernel threads have an empty cmdline */ + if (count <= 0) + return true; + + /* Processes with argv[0][0] = '@' we ignore from the killing + * spree. + * + * http://www.freedesktop.org/wiki/Software/systemd/RootStorageDaemons */ + if (count == 1 && c == '@') + return true; + + return false; } static int killall(int sign) { @@ -77,7 +95,8 @@ static int killall(int sign) { struct dirent *d; unsigned int n_processes = 0; - if ((dir = opendir("/proc")) == NULL) + dir = opendir("/proc"); + if (!dir) return -errno; while ((d = readdir(dir))) { @@ -86,9 +105,6 @@ static int killall(int sign) { if (parse_pid(d->d_name, &pid) < 0) continue; - if (is_kernel_thread(pid)) - continue; - if (ignore_proc(pid)) continue; @@ -199,57 +215,58 @@ finish: sigprocmask(SIG_SETMASK, &oldmask, NULL); } -static bool prepare_new_root(void) { - int r = false; - const char *dirs[] = { "/run/initramfs/oldroot", - "/run/initramfs/proc", - "/run/initramfs/sys", - "/run/initramfs/dev", - "/run/initramfs/run", - NULL }; - const char **dir; - const char *msg; - - msg = "Failed to mount bind /run/initramfs on /run/initramfs"; - if (mount("/run/initramfs", "/run/initramfs", NULL, MS_BIND, NULL) != 0) - goto out; - - msg="Failed to make /run/initramfs private mount %m:"; - if (mount(NULL, "/run/initramfs", NULL, MS_PRIVATE, NULL) != 0) - goto out; - - for (dir = &dirs[0]; *dir != NULL; dir++) { - asprintf((char **) &msg, "mkdir %s: %%m", *dir); - if (mkdir(*dir, 0755) != 0) { - if (errno != EEXIST) - goto out; +static int prepare_new_root(void) { + static const char dirs[] = + "/run/initramfs/oldroot\0" + "/run/initramfs/proc\0" + "/run/initramfs/sys\0" + "/run/initramfs/dev\0" + "/run/initramfs/run\0"; + + const char *dir; + + if (mount("/run/initramfs", "/run/initramfs", NULL, MS_BIND, NULL) < 0) { + log_error("Failed to mount bind /run/initramfs on /run/initramfs: %m"); + return -errno; + } + + if (mount(NULL, "/run/initramfs", NULL, MS_PRIVATE, NULL) < 0) { + log_error("Failed to make /run/initramfs private mount: %m"); + return -errno; + } + + NULSTR_FOREACH(dir, dirs) + if (mkdir_p(dir, 0755) < 0 && errno != EEXIST) { + log_error("Failed to mkdir %s: %m", dir); + return -errno; } - free((char *) msg); + + if (mount("/sys", "/run/initramfs/sys", NULL, MS_BIND, NULL) < 0) { + log_error("Failed to mount bind /sys on /run/initramfs/sys: %m"); + return -errno; + } + + if (mount("/proc", "/run/initramfs/proc", NULL, MS_BIND, NULL) < 0) { + log_error("Failed to mount bind /proc on /run/initramfs/proc: %m"); + return -errno; + } + + if (mount("/dev", "/run/initramfs/dev", NULL, MS_BIND, NULL) < 0) { + log_error("Failed to mount bind /dev on /run/initramfs/dev: %m"); + return -errno; } - msg = "Failed to mount bind /sys on /run/initramfs/sys"; - if (mount("/sys", "/run/initramfs/sys", NULL, MS_BIND, NULL) != 0) - goto out; - msg = "Failed to mount bind /proc on /run/initramfs/proc"; - if (mount("/proc", "/run/initramfs/proc", NULL, MS_BIND, NULL) != 0) - goto out; - msg = "Failed to mount bind /dev on /run/initramfs/dev"; - if (mount("/dev", "/run/initramfs/dev", NULL, MS_BIND, NULL) != 0) - goto out; - msg = "Failed to mount bind /run on /run/initramfs/run"; - if (mount("/run", "/run/initramfs/run", NULL, MS_BIND, NULL) != 0) - goto out; - - r = true; - out: - if (!r) - log_error("%s: %m", msg); - return r; + if (mount("/run", "/run/initramfs/run", NULL, MS_BIND, NULL) < 0) { + log_error("Failed to mount bind /run on /run/initramfs/run: %m"); + return -errno; + } + + return 0; } -static bool pivot_to_new_root(void) { +static int pivot_to_new_root(void) { int fd; - int r = 0; + chdir("/run/initramfs"); /* @@ -257,30 +274,32 @@ static bool pivot_to_new_root(void) { It works for pivot_root, but the ref count for the root device is not decreasing :-/ */ - if (mount(NULL, "/", NULL, MS_PRIVATE, NULL) != 0) { - log_error("Failed to make \"/\" private mount %m: "); - return false; + if (mount(NULL, "/", NULL, MS_PRIVATE, NULL) < 0) { + log_error("Failed to make \"/\" private mount %m"); + return -errno; } - r = pivot_root(".", "oldroot"); - if (r!=0) { + if (pivot_root(".", "oldroot") < 0) { log_error("pivot failed: %m"); - /* only chroot, if pivot root succeded */ - return false; + /* only chroot if pivot root succeded */ + return -errno; } + chroot("."); - log_info("pivot rooted"); - - fd = open("dev/console", O_RDONLY); - dup2(fd, STDIN_FILENO); - close_nointr_nofail(fd); - fd = open("dev/console", O_WRONLY); - dup2(fd, STDOUT_FILENO); - close_nointr_nofail(fd); - fd = open("dev/console", O_WRONLY); - dup2(fd, STDERR_FILENO); - close_nointr_nofail(fd); - return true; + log_info("Successfully changed into root pivot."); + + fd = open("/dev/console", O_RDWR); + if (fd < 0) + log_error("Failed to open /dev/console: %m"); + else { + make_stdio(fd); + + /* Initialize the controlling terminal */ + setsid(); + ioctl(STDIN_FILENO, TIOCSCTTY, NULL); + } + + return 0; } int main(int argc, char *argv[]) { @@ -293,6 +312,8 @@ int main(int argc, char *argv[]) { log_set_target(LOG_TARGET_CONSOLE); /* syslog will die if not gone yet */ log_open(); + umask(0022); + if (getpid() != 1) { log_error("Not executed by init (pid 1)."); r = -EPERM; @@ -382,9 +403,12 @@ int main(int argc, char *argv[]) { log_error("Failed to detach DM devices: %s", strerror(-r)); } - if (!need_umount && !need_swapoff && !need_loop_detach && !need_dm_detach) + if (!need_umount && !need_swapoff && !need_loop_detach && !need_dm_detach) { + if (retries > 0) + log_info("All filesystems, swaps, loop devices, DM devices detached."); /* Yay, done */ break; + } /* If in this iteration we didn't manage to * unmount/deactivate anything, we either kill more @@ -419,19 +443,17 @@ int main(int argc, char *argv[]) { exit(0); } - sync(); - if (access("/run/initramfs/shutdown", X_OK) == 0) { - char *new_argv[3]; - new_argv[0] = strdup(argv[0]); - new_argv[1] = strdup(argv[1]); - new_argv[2] = NULL; - if (prepare_new_root() && pivot_to_new_root()) { - execv("/shutdown", new_argv); + + if (prepare_new_root() >= 0 && + pivot_to_new_root() >= 0) { + execv("/shutdown", argv); log_error("Failed to execute shutdown binary: %m"); } } + sync(); + if (cmd == LINUX_REBOOT_CMD_KEXEC) { /* We cheat and exec kexec to avoid doing all its work */ pid_t pid = fork();