X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=httpinput.cpp;h=875ee62a29d45ca31858dc0fd142ed197f404d55;hp=46f1358927312ba210cfa2f518e29991595d14ed;hb=9bb20b7dd0bcea9de1daf4cac29263d74924ce5a;hpb=d60b2c17baf9b71e4a70a499fa1d34753821c822

diff --git a/httpinput.cpp b/httpinput.cpp
index 46f1358..875ee62 100644
--- a/httpinput.cpp
+++ b/httpinput.cpp
@@ -1,30 +1,26 @@
+#include <assert.h>
+#include <errno.h>
+#include <netdb.h>
+#include <netinet/in.h>
+#include <poll.h>
+#include <stdint.h>
 #include <stdio.h>
+#include <stdlib.h>
 #include <string.h>
-#include <stdint.h>
-#include <assert.h>
-#include <arpa/inet.h>
-#include <sys/socket.h>
-#include <pthread.h>
-#include <sys/types.h>
 #include <sys/ioctl.h>
-#include <sys/types.h>
 #include <sys/socket.h>
-#include <netdb.h>
-#include <poll.h>
-#include <signal.h>
-#include <errno.h>
-#include <vector>
-#include <string>
+#include <unistd.h>
 #include <map>
+#include <string>
+#include <utility>
+#include <vector>
 
-#include "metacube.h"
-#include "mutexlock.h"
 #include "httpinput.h"
-#include "server.h"
-#include "serverpool.h"
+#include "metacube.h"
 #include "parse.h"
-#include "version.h"
+#include "serverpool.h"
 #include "state.pb.h"
+#include "version.h"
 
 using namespace std;
 
@@ -57,6 +53,18 @@ HTTPInput::HTTPInput(const InputProto &serialized)
 	parse_url(url, &protocol, &host, &port, &path);  // Don't care if it fails.
 }
 
+void HTTPInput::close_socket()
+{
+	int ret;
+	do {
+		ret = close(sock);
+	} while (ret == -1 && errno == EINTR);
+
+	if (ret == -1) {
+		perror("close()");
+	}
+}
+
 InputProto HTTPInput::serialize() const
 {
 	InputProto serialized;
@@ -101,6 +109,15 @@ int HTTPInput::lookup_and_connect(const string &host, const string &port)
 			return sock;
 		}
 
+		do {
+			err = close(sock);
+		} while (err == -1 && errno == EINTR);
+
+		if (err == -1) {
+			perror("close");
+			// Can still continue.
+		}
+
 		ai = ai->ai_next;
 	}