Zbigniew Jędrzejewski-Szmek 930e1b
From 546f518baf5c1cb083030fe2b3de751f160e5eea Mon Sep 17 00:00:00 2001
Zbigniew Jędrzejewski-Szmek 930e1b
From: Lennart Poettering <lennart@poettering.net>
Zbigniew Jędrzejewski-Szmek 930e1b
Date: Wed, 6 Nov 2013 02:05:06 +0100
Zbigniew Jędrzejewski-Szmek 930e1b
Subject: [PATCH] nspawn: explicitly terminate machines when we exit nspawn
Zbigniew Jędrzejewski-Szmek 930e1b
Zbigniew Jędrzejewski-Szmek 930e1b
https://bugs.freedesktop.org/show_bug.cgi?id=68370
Zbigniew Jędrzejewski-Szmek 930e1b
https://bugzilla.redhat.com/show_bug.cgi?id=988883
Zbigniew Jędrzejewski-Szmek 930e1b
Zbigniew Jędrzejewski-Szmek 930e1b
Conflicts:
Zbigniew Jędrzejewski-Szmek 930e1b
	src/nspawn/nspawn.c
Zbigniew Jędrzejewski-Szmek 930e1b
---
Zbigniew Jędrzejewski-Szmek 930e1b
 src/nspawn/nspawn.c | 70 +++++++++++++++++++++++++++++++++++++++++++++++++----
Zbigniew Jędrzejewski-Szmek 930e1b
 1 file changed, 65 insertions(+), 5 deletions(-)
Zbigniew Jędrzejewski-Szmek 930e1b
Zbigniew Jędrzejewski-Szmek 930e1b
diff --git a/src/nspawn/nspawn.c b/src/nspawn/nspawn.c
Zbigniew Jędrzejewski-Szmek 930e1b
index cda6c82..85bbadf 100644
Zbigniew Jędrzejewski-Szmek 930e1b
--- a/src/nspawn/nspawn.c
Zbigniew Jędrzejewski-Szmek 930e1b
+++ b/src/nspawn/nspawn.c
Zbigniew Jędrzejewski-Szmek 930e1b
@@ -43,9 +43,9 @@
Zbigniew Jędrzejewski-Szmek 930e1b
 #include <sys/socket.h>
Zbigniew Jędrzejewski-Szmek 930e1b
 #include <linux/netlink.h>
Zbigniew Jędrzejewski-Szmek 930e1b
 
Zbigniew Jędrzejewski-Szmek 930e1b
-#include <systemd/sd-daemon.h>
Zbigniew Jędrzejewski-Szmek 930e1b
-#include <systemd/sd-bus.h>
Zbigniew Jędrzejewski-Szmek 930e1b
-
Zbigniew Jędrzejewski-Szmek 930e1b
+#include "sd-daemon.h"
Zbigniew Jędrzejewski-Szmek 930e1b
+#include "sd-bus.h"
Zbigniew Jędrzejewski-Szmek 930e1b
+#include "sd-id128.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "log.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "util.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "mkdir.h"
Zbigniew Jędrzejewski-Szmek 930e1b
@@ -56,13 +56,13 @@
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "strv.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "path-util.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "loopback-setup.h"
Zbigniew Jędrzejewski-Szmek 930e1b
-#include "sd-id128.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "dev-setup.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "fdset.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "build.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "fileio.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "bus-internal.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "bus-message.h"
Zbigniew Jędrzejewski-Szmek 930e1b
+#include "bus-error.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 #include "ptyfwd.h"
Zbigniew Jędrzejewski-Szmek 930e1b
 
Zbigniew Jędrzejewski-Szmek 930e1b
 #ifndef TTY_GID
Zbigniew Jędrzejewski-Szmek 930e1b
@@ -966,10 +966,64 @@ static int register_machine(void) {
Zbigniew Jędrzejewski-Szmek 930e1b
                         strempty(arg_directory),
Zbigniew Jędrzejewski-Szmek 930e1b
                         !isempty(arg_slice), "Slice", "s", arg_slice);
Zbigniew Jędrzejewski-Szmek 930e1b
         if (r < 0) {
Zbigniew Jędrzejewski-Szmek 930e1b
-                log_error("Failed to register machine: %s", error.message ? error.message : strerror(-r));
Zbigniew Jędrzejewski-Szmek 930e1b
+                log_error("Failed to register machine: %s", bus_error_message(&error, r));
Zbigniew Jędrzejewski-Szmek 930e1b
+                return r;
Zbigniew Jędrzejewski-Szmek 930e1b
+        }
Zbigniew Jędrzejewski-Szmek 930e1b
+
Zbigniew Jędrzejewski-Szmek 930e1b
+        return 0;
Zbigniew Jędrzejewski-Szmek 930e1b
+}
Zbigniew Jędrzejewski-Szmek 930e1b
+
Zbigniew Jędrzejewski-Szmek 930e1b
+static int terminate_machine(pid_t pid) {
Zbigniew Jędrzejewski-Szmek 930e1b
+        _cleanup_bus_error_free_ sd_bus_error error = SD_BUS_ERROR_NULL;
Zbigniew Jędrzejewski-Szmek 930e1b
+        _cleanup_bus_message_unref_ sd_bus_message *reply = NULL;
Zbigniew Jędrzejewski-Szmek 930e1b
+        _cleanup_bus_unref_ sd_bus *bus = NULL;
Zbigniew Jędrzejewski-Szmek 930e1b
+        const char *path;
Zbigniew Jędrzejewski-Szmek 930e1b
+        int r;
Zbigniew Jędrzejewski-Szmek 930e1b
+
Zbigniew Jędrzejewski-Szmek 930e1b
+        r = sd_bus_open_system(&bus;;
Zbigniew Jędrzejewski-Szmek 930e1b
+        if (r < 0) {
Zbigniew Jędrzejewski-Szmek 930e1b
+                log_error("Failed to open system bus: %s", strerror(-r));
Zbigniew Jędrzejewski-Szmek 930e1b
+                return r;
Zbigniew Jędrzejewski-Szmek 930e1b
+        }
Zbigniew Jędrzejewski-Szmek 930e1b
+
Zbigniew Jędrzejewski-Szmek 930e1b
+        r = sd_bus_call_method(
Zbigniew Jędrzejewski-Szmek 930e1b
+                        bus,
Zbigniew Jędrzejewski-Szmek 930e1b
+                        "org.freedesktop.machine1",
Zbigniew Jędrzejewski-Szmek 930e1b
+                        "/org/freedesktop/machine1",
Zbigniew Jędrzejewski-Szmek 930e1b
+                        "org.freedesktop.machine1.Manager",
Zbigniew Jędrzejewski-Szmek 930e1b
+                        "GetMachineByPID",
Zbigniew Jędrzejewski-Szmek 930e1b
+                        &error,
Zbigniew Jędrzejewski-Szmek 930e1b
+                        &reply,
Zbigniew Jędrzejewski-Szmek 930e1b
+                        "u",
Zbigniew Jędrzejewski-Szmek 930e1b
+                        (uint32_t) pid);
Zbigniew Jędrzejewski-Szmek 930e1b
+        if (r < 0) {
Zbigniew Jędrzejewski-Szmek 930e1b
+                /* Note that the machine might already have been
Zbigniew Jędrzejewski-Szmek 930e1b
+                 * cleaned up automatically, hence don't consider it a
Zbigniew Jędrzejewski-Szmek 930e1b
+                 * failure if we cannot get the machine object. */
Zbigniew Jędrzejewski-Szmek 930e1b
+                log_debug("Failed to get machine: %s", bus_error_message(&error, r));
Zbigniew Jędrzejewski-Szmek 930e1b
+                return 0;
Zbigniew Jędrzejewski-Szmek 930e1b
+        }
Zbigniew Jędrzejewski-Szmek 930e1b
+
Zbigniew Jędrzejewski-Szmek 930e1b
+        r = sd_bus_message_read(reply, "o", &path);
Zbigniew Jędrzejewski-Szmek 930e1b
+        if (r < 0) {
Zbigniew Jędrzejewski-Szmek 930e1b
+                log_error("Failed to parse GetMachineByPID() reply: %s", bus_error_message(&error, r));
Zbigniew Jędrzejewski-Szmek 930e1b
                 return r;
Zbigniew Jędrzejewski-Szmek 930e1b
         }
Zbigniew Jędrzejewski-Szmek 930e1b
 
Zbigniew Jędrzejewski-Szmek 930e1b
+        r = sd_bus_call_method(
Zbigniew Jędrzejewski-Szmek 930e1b
+                        bus,
Zbigniew Jędrzejewski-Szmek 930e1b
+                        "org.freedesktop.machine1",
Zbigniew Jędrzejewski-Szmek 930e1b
+                        path,
Zbigniew Jędrzejewski-Szmek 930e1b
+                        "org.freedesktop.machine1.Machine",
Zbigniew Jędrzejewski-Szmek 930e1b
+                        "Terminate",
Zbigniew Jędrzejewski-Szmek 930e1b
+                        &error,
Zbigniew Jędrzejewski-Szmek 930e1b
+                        NULL,
Zbigniew Jędrzejewski-Szmek 930e1b
+                        NULL);
Zbigniew Jędrzejewski-Szmek 930e1b
+        if (r < 0) {
Zbigniew Jędrzejewski-Szmek 930e1b
+                log_debug("Failed to terminate machine: %s", bus_error_message(&error, r));
Zbigniew Jędrzejewski-Szmek 930e1b
+                return 0;
Zbigniew Jędrzejewski-Szmek 930e1b
+        }
Zbigniew Jędrzejewski-Szmek 930e1b
+
Zbigniew Jędrzejewski-Szmek 930e1b
         return 0;
Zbigniew Jędrzejewski-Szmek 930e1b
 }
Zbigniew Jędrzejewski-Szmek 930e1b
 
Zbigniew Jędrzejewski-Szmek 930e1b
@@ -1448,6 +1502,12 @@ int main(int argc, char *argv[]) {
Zbigniew Jędrzejewski-Szmek 930e1b
                 if (saved_attr_valid)
Zbigniew Jędrzejewski-Szmek 930e1b
                         tcsetattr(STDIN_FILENO, TCSANOW, &saved_attr);
Zbigniew Jędrzejewski-Szmek 930e1b
 
Zbigniew Jędrzejewski-Szmek 930e1b
+                /* Kill if it is not dead yet anyway */
Zbigniew Jędrzejewski-Szmek 930e1b
+                terminate_machine(pid);
Zbigniew Jędrzejewski-Szmek 930e1b
+
Zbigniew Jędrzejewski-Szmek 930e1b
+                /* Redundant, but better safe than sorry */
Zbigniew Jędrzejewski-Szmek 930e1b
+                kill(pid, SIGKILL);
Zbigniew Jędrzejewski-Szmek 930e1b
+
Zbigniew Jędrzejewski-Szmek 930e1b
                 k = wait_for_terminate(pid, &status);
Zbigniew Jędrzejewski-Szmek 930e1b
                 if (k < 0) {
Zbigniew Jędrzejewski-Szmek 930e1b
                         r = EXIT_FAILURE;