X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=httpinput.cpp;h=274cc2687b160d45e365fc7f4daafb665b9d2660;hp=d0e1c5c82135f42bf7ac1b75e77d90af5d4a5402;hb=e3f2936e3c9ff3b5569759c1aaed16f03bf728f8;hpb=6d34c5b6d8e5bec5d1421eadc103f38d206f34f1

diff --git a/httpinput.cpp b/httpinput.cpp
index d0e1c5c..274cc26 100644
--- a/httpinput.cpp
+++ b/httpinput.cpp
@@ -11,9 +11,12 @@
 #include <sys/ioctl.h>
 #include <sys/socket.h>
 #include <sys/time.h>
+#include <sys/types.h>
+#include <sys/wait.h>
 #include <time.h>
 #include <unistd.h>
 #include <math.h>
+#include <spawn.h>
 #include <map>
 #include <string>
 #include <utility>
@@ -220,6 +223,52 @@ int HTTPInput::lookup_and_connect(const string &host, const string &port)
 	freeaddrinfo(base_ai);
 	return -1;
 }
+
+int HTTPInput::open_child_process(const string &cmdline)
+{
+	int devnullfd = open("/dev/null", O_RDONLY | O_CLOEXEC);
+	if (devnullfd == -1) {
+		log_perror("/dev/null");
+		return -1;
+	}
+
+	int pipefd[2];
+	if (pipe2(pipefd, O_CLOEXEC) == -1) {
+		log_perror("pipe2()");
+		close(devnullfd);
+		return -1;
+	}
+
+	// Point stdout to us, stdin to /dev/null, and stderr remains where it is
+	// (probably the systemd log). All other file descriptors should be marked
+	// as close-on-exec, and should thus not leak into the child.
+	posix_spawn_file_actions_t actions;
+	posix_spawn_file_actions_init(&actions);
+	posix_spawn_file_actions_adddup2(&actions, devnullfd, 0);
+	posix_spawn_file_actions_adddup2(&actions, pipefd[1], 1);
+
+	pid_t child_pid;
+	char * const argv[] = {
+		strdup("/bin/sh"),
+		strdup("-c"),
+		strdup(path.c_str()),
+		nullptr
+	};
+	int err = posix_spawn(&child_pid, "/bin/sh", &actions, /*attrp=*/nullptr, argv, /*envp=*/nullptr);
+	posix_spawn_file_actions_destroy(&actions);
+	free(argv[0]);
+	free(argv[1]);
+	free(argv[2]);
+	close(pipefd[1]);
+
+	if (err == 0) {
+		return pipefd[0];
+	} else {
+		log_perror(cmdline.c_str());
+		close(pipefd[0]);
+		return -1;
+	}
+}
 	
 bool HTTPInput::parse_response(const string &request)
 {
@@ -332,7 +381,11 @@ void HTTPInput::do_work()
 		}
 
 		switch (state) {
-		case NOT_CONNECTED:
+		case NOT_CONNECTED: {
+			// Reap any exited children.
+			int wstatus;
+			while (waitpid(-1, &wstatus, WNOHANG) != -1 || errno == EINTR) ;
+
 			request.clear();
 			request_bytes_sent = 0;
 			response.clear();
@@ -343,8 +396,9 @@ void HTTPInput::do_work()
 				servers->set_unavailable(stream_index);
 			}
 
+			string protocol;
 			{
-				string protocol, user;  // Thrown away.
+				string user;  // Thrown away.
 				if (!parse_url(url, &protocol, &user, &host, &port, &path)) {
 					if (!suppress_logging) {
 						log(WARNING, "[%s] Failed to parse URL '%s'", url.c_str(), url.c_str());
@@ -378,18 +432,33 @@ void HTTPInput::do_work()
 				assert(err != -1);
 			}
 			++num_connection_attempts;
-			sock = lookup_and_connect(host, port);
+			if (protocol == "pipe") {
+				sock = open_child_process(path.c_str());
+
+				if (sock != -1) {
+					// Construct a minimal HTTP header.
+					http_header = "HTTP/1.0 200 OK\r\n";
+					for (int stream_index : stream_indices) {
+						servers->set_header(stream_index, http_header, stream_header);
+					}
+					state = RECEIVING_DATA;
+				}
+			} else {
+				sock = lookup_and_connect(host, port);
+				if (sock != -1) {
+					// Yay, successful connect.
+					state = SENDING_REQUEST;
+					request = "GET " + path + " HTTP/1.0\r\nHost: " + host_header(host, port) + "\r\nUser-Agent: cubemap\r\n\r\n";
+					request_bytes_sent = 0;
+				}
+			}
 			if (sock != -1) {
-				// Yay, successful connect.
-				state = SENDING_REQUEST;
-				request = "GET " + path + " HTTP/1.0\r\nHost: " + host_header(host, port) + "\r\nUser-Agent: cubemap\r\n\r\n";
-				request_bytes_sent = 0;
-
 				lock_guard<mutex> lock(stats_mutex);
 				stats.connect_time = time(nullptr);
 				clock_gettime(CLOCK_MONOTONIC_COARSE, &last_activity);
 			}
 			break;
+		}
 		case SENDING_REQUEST: {
 			size_t to_send = request.size() - request_bytes_sent;
 			int ret;