From be2dea3c51c0622f50bb41a0e660e27afbb791c0 Mon Sep 17 00:00:00 2001
From: Unbit <info@unbit.it>
Date: Sat, 3 May 2014 12:53:04 +0200
Subject: [PATCH] preparing for fork server connection

---
 core/emperor.c | 86 +++++++++++++++++++++++++++++++++++++++++++++++---
 core/uwsgi.c   |  1 +
 uwsgi.h        |  2 ++
 3 files changed, 84 insertions(+), 5 deletions(-)

diff --git a/core/emperor.c b/core/emperor.c
index 9c7ac340..73f01cec 100644
--- a/core/emperor.c
+++ b/core/emperor.c
@@ -929,6 +929,68 @@ void emperor_add(struct uwsgi_emperor_scanner *ues, char *name, time_t born, cha
 
 static void uwsgi_emperor_spawn_vassal(struct uwsgi_instance *);
 
+/*
+	there are max 3 file descriptors we need to pass to the fork server:
+
+	n_ui->pipe[1]
+	n_ui->pipe_config[1]
+	n_ui->on_demand_fd[1]
+
+*/
+static pid_t emperor_connect_to_fork_server(char *socket, struct uwsgi_instance *n_ui) {
+	int fd = uwsgi_connect(socket, uwsgi.socket_timeout, 0);
+
+        int num_fds = 1;
+	if (n_ui->use_config) num_fds ++;	
+	if (n_ui->on_demand_fd > -1) num_fds++;
+
+        struct msghdr ep_msg;
+        void *ep_msg_control = uwsgi_malloc(CMSG_SPACE(sizeof(int) * num_fds));
+        struct iovec ep_iov[2];
+        struct cmsghdr *cmsg;
+
+	// build the parameters (passed as a uwsgi array)
+
+        ep_iov[0].iov_base = "uwsgi-emperor";
+        ep_iov[0].iov_len = 13;
+        ep_iov[1].iov_base = &num_fds;
+        ep_iov[1].iov_len = sizeof(int);
+
+        ep_msg.msg_name = NULL;
+        ep_msg.msg_namelen = 0;
+
+        ep_msg.msg_iov = ep_iov;
+        ep_msg.msg_iovlen = 2;
+
+        ep_msg.msg_flags = 0;
+        ep_msg.msg_control = ep_msg_control;
+        ep_msg.msg_controllen = CMSG_SPACE(sizeof(int) * num_fds);
+
+        cmsg = CMSG_FIRSTHDR(&ep_msg);
+        cmsg->cmsg_len = CMSG_LEN(sizeof(int) * num_fds);
+        cmsg->cmsg_level = SOL_SOCKET;
+        cmsg->cmsg_type = SCM_RIGHTS;
+
+        unsigned char *ep_fd_ptr = CMSG_DATA(cmsg);
+
+        memcpy(ep_fd_ptr, &uwsgi.emperor_fd, sizeof(int));
+        if (num_fds > 1) {
+                memcpy(ep_fd_ptr + sizeof(int), &uwsgi.emperor_fd_config, sizeof(int));
+        }
+
+        if (sendmsg(fd, &ep_msg, 0) < 0) {
+                uwsgi_error("emperor_connect_to_fork_server()/sendmsg()");
+        }
+
+        free(ep_msg_control);
+
+	// now wait for the response (the pid number)
+
+        close(fd);
+
+	return -1;
+}
+
 int uwsgi_emperor_vassal_start(struct uwsgi_instance *n_ui) {
 
 	pid_t pid;
@@ -956,17 +1018,20 @@ int uwsgi_emperor_vassal_start(struct uwsgi_instance *n_ui) {
 	// TODO pre-start hook
 
 	// a new uWSGI instance will start 
+	if (uwsgi.emperor_use_fork_server) {
+		// pid can only be > 0 or -1
+		pid = emperor_connect_to_fork_server(uwsgi.emperor_use_fork_server, n_ui);
+	}
 #if defined(__linux__) && !defined(OBSOLETE_LINUX_KERNEL) && !defined(__ia64__)
-	if (uwsgi.emperor_clone) {
+	else if (uwsgi.emperor_clone) {
 		char stack[PTHREAD_STACK_MIN];
 		pid = clone((int (*)(void *))uwsgi_emperor_spawn_vassal, stack + PTHREAD_STACK_MIN, SIGCHLD | uwsgi.emperor_clone, (void *) n_ui);
 	}
+#endif
 	else {
-#endif
-	pid = fork();
-#if defined(__linux__) && !defined(OBSOLETE_LINUX_KERNEL) && !defined(__ia64__)
+		pid = fork();
 	}
-#endif
+
 	if (pid < 0) {
 		uwsgi_error("uwsgi_emperor_spawn_vassal()/fork()")
 	}
@@ -1854,6 +1919,10 @@ recheck:
 				uwsgi_error("waitpid()");
 			}
 		}
+
+		if (diedpid > 0) {
+			uwsgi_log("DIEDPID = %d\n", diedpid);
+		}
 		ui_current = ui;
 		while (ui_current->ui_next) {
 			ui_current = ui_current->ui_next;
@@ -2133,6 +2202,13 @@ end:
 
 void uwsgi_emperor_start() {
 
+#ifdef __linux__
+        if (prctl(PR_SET_CHILD_SUBREAPER, 1, 0, 0, 0)) {
+                uwsgi_error("uwsgi_fork_server()/fork()");
+                exit(1);
+        }
+#endif
+
 	if (!uwsgi.sockets && !ushared->gateways_cnt && !uwsgi.master_process) {
 		if (uwsgi.emperor_procname) {
 			uwsgi_set_processname(uwsgi.emperor_procname);
diff --git a/core/uwsgi.c b/core/uwsgi.c
index c660c3dd..e7866609 100644
--- a/core/uwsgi.c
+++ b/core/uwsgi.c
@@ -1952,6 +1952,7 @@ static char *uwsgi_at_file_read(char *filename) {
 }
 
 void uwsgi_setup(int argc, char *argv[], char *envp[]) {
+
 #ifdef UWSGI_AS_SHARED_LIBRARY
 #ifdef __APPLE__
 	char ***envPtr = _NSGetEnviron();
diff --git a/uwsgi.h b/uwsgi.h
index a6ec0740..ec2c1047 100644
--- a/uwsgi.h
+++ b/uwsgi.h
@@ -2710,9 +2710,11 @@ struct uwsgi_server {
 	int mule_reload_mercy;
 	int alarm_cheap;
 
+	// uWSGI 2.0.5
 	char *fork_socket;
 	int new_argc;
 	char **new_argv;
+	char *emperor_use_fork_server;
 };
 
 struct uwsgi_rpc {