[CRIU] [PATCH 3/3] ns: Make pid_ns helpers as children of criu main process

[Kirill Tkhai]

If a task, holding userns_sync_lock unexpectedly exits,
criu will hang on error path in restore_root_task(),
because it can't use usernsd to destroy them.

Lets remove the intermediary: we'll create pid_ns helpers
as children of criu main task, and criu main task will
be able to use simple kill() to stop them.

Signed-off-by: Kirill Tkhai <ktkhai at virtuozzo.com>
---
 criu/cr-restore.c |    8 ++++++++
 criu/namespaces.c |   35 +++--------------------------------
 2 files changed, 11 insertions(+), 32 deletions(-)

diff --git a/criu/cr-restore.c b/criu/cr-restore.c
index 2c505f6ea..68576241b 100644
--- a/criu/cr-restore.c
+++ b/criu/cr-restore.c
@@ -1446,6 +1446,7 @@ static inline int fork_with_pid(struct pstree_item *item)
 static void sigchld_handler(int signal, siginfo_t *siginfo, void *data)
 {
 	int status, pid, exit;
+	struct ns_id *ns;
 
 	while (1) {
 		pid = waitpid(-1, &status, WNOHANG);
@@ -1465,6 +1466,13 @@ static void sigchld_handler(int signal, siginfo_t *siginfo, void *data)
 		break;
 	}
 
+	if (!current) {
+		ns = handle_pid_ns_helper_exit(pid);
+		if (ns)
+			pr_err("Unexpected pid ns helper: pid=%d, ns_id=%d\n",
+				pid, ns->id);
+	}
+
 	if (exit)
 		pr_err("%d exited, status=%d\n", pid, status);
 	else
diff --git a/criu/namespaces.c b/criu/namespaces.c
index c44480b37..2c60da1ce 100644
--- a/criu/namespaces.c
+++ b/criu/namespaces.c
@@ -1597,26 +1597,6 @@ struct ns_id *handle_pid_ns_helper_exit(pid_t real_pid)
 	return ns;
 }
 
-static void usernsd_handler(int signal, siginfo_t *siginfo, void *data)
-{
-	pid_t pid = siginfo->si_pid;
-	struct ns_id *ns;
-	int status;
-
-	while (pid) {
-		pid = waitpid(-1, &status, WNOHANG);
-		if (pid <= 0)
-			return;
-
-		ns = handle_pid_ns_helper_exit(pid);
-		if (!ns)
-			pr_err("Spurious pid ns helper: pid=%d\n", pid);
-
-		pr_err("%d finished unexpected: status=%d\n", pid, status);
-		futex_abort_and_wake(&task_entries->nr_in_progress);
-	}
-}
-
 static int usernsd_recv_transport(void *arg, int fd, pid_t pid)
 {
 	if (install_service_fd(TRANSPORT_FD_OFF, fd) < 0) {
@@ -1660,11 +1640,6 @@ static int usernsd(int sk)
 {
 	pr_info("uns: Daemon started\n");
 
-	if (criu_signals_setup(usernsd_handler) < 0) {
-		pr_err("Can't setup handler\n");
-		return -1;
-	}
-
 	while (1) {
 		struct unsc_msg um;
 		static char msg[MAX_UNSFD_MSG_SIZE];
@@ -2719,7 +2694,7 @@ static int do_create_pid_ns_helper(void *arg, int sk, pid_t unused_pid)
 			unlock_last_pid();
 			goto restore_ns;
 		}
-	child = fork();
+	child = sys_clone_unified(CLONE_PARENT|SIGCHLD, NULL, NULL, NULL, 0);
 	if (!child)
 		exit(pid_ns_helper(ns, sk));
 	saved_errno = errno;
@@ -2781,7 +2756,7 @@ int create_pid_ns_helper(struct ns_id *ns)
 	return 0;
 }
 
-static int do_destroy_pid_ns_helpers(void *arg, int fd, pid_t unused)
+static int do_destroy_pid_ns_helpers(void)
 {
 	int status, sig_blocked = true, ret = 0;
 	sigset_t sig_mask;
@@ -2830,11 +2805,7 @@ int destroy_pid_ns_helpers(void)
 	if (!(root_ns_mask & CLONE_NEWPID))
 		return 0;
 
-	if (userns_call(do_destroy_pid_ns_helpers, 0, NULL, 0, -1) < 0) {
-		pr_err("Can't create pid_ns helper\n");
-		return -1;
-	}
-	return 0;
+	return do_destroy_pid_ns_helpers();
 }
 
 struct ns_desc pid_ns_desc = NS_DESC_ENTRY(CLONE_NEWPID, "pid", "pid_for_children");