added support for emperor protocol in the fork server

1 files changed, 19 insertions, 0 deletions

diff --git a/core/fork_server.c b/core/fork_server.c
index 232678c4..18beb88c 100644
--- a/core/fork_server.c
+++ b/core/fork_server.c
@@ -33,8 +33,27 @@ void uwsgi_fork_server(char *socket) {
 	// automatically receive credentials (TODO make something useful with them, like checking the pid is from the Emperor)
 	if (uwsgi_socket_passcred(fd)) exit(1);
 
+	// initialize the event queue
+	int eq = event_queue_init();
+	if (uwsgi.has_emperor) {
+		event_queue_add_fd_read(eq, uwsgi.emperor_fd);
+	}
+	event_queue_add_fd_read(eq, fd);
+
 	// now start waiting for connections
 	for(;;) {
+		int interesting_fd = -1;
+		int rlen = event_queue_wait(eq, -1, &interesting_fd);
+		if (rlen <= 0) continue;
+		if (uwsgi.has_emperor && interesting_fd == uwsgi.emperor_fd) {
+			char byte;
+        		ssize_t rlen = read(uwsgi.emperor_fd, &byte, 1);
+        		if (rlen > 0) {
+                		uwsgi_log_verbose("received message %d from emperor\n", byte);
+			}
+			exit(0);
+		}
+		if (interesting_fd != fd) continue;
 		struct sockaddr_un client_src;
         	socklen_t client_src_len = 0;
         	int client_fd = accept(fd, (struct sockaddr *) &client_src, &client_src_len);