implement multi-url setup

This reverts commit 6fb3c6f193.

src/server.c

@@ -36,12 +36,16 @@
 #include <string.h>
 #include <pthread.h>
 #include <sys/stat.h>
+#include <malloc.h>
 
 #include <curl/curl.h>
 #include <glib.h>
 
 GStaticMutex dupes_mutex = G_STATIC_MUTEX_INIT;
 
+char *urls[2] = {"http://debuginfo.clearlinux.org/debuginfo/", "http://debuginfo.fenrus.org/debuginfo/" };
+int urlcounter = 0;
+
 
 static GHashTable *hash;
 
@@ -118,6 +122,9 @@ static int curl_get_file(const char *url, const char *prefix, time_t timestamp)
 	fflush(file);
 //	printf("HTTP return code is %i\n", ret);
 
+	if (ret != 200 && ret != 404)
+		urlcounter++;
+
 	if (ret == 200) {
 		char *command = NULL;
 //		printf("Filename is %s\n", filename);
@@ -185,7 +192,7 @@ static void *server_thread(void *arg)
 	path++;
 
 
-	if (strstr(path, "..") || strstr(prefix, "..")) {
+	if (strstr(path, "..") || strstr(prefix, "..") || strstr(path, "'") || strstr(path, ";")) {
 		close(fd);
 		return NULL;
 	}
@@ -196,7 +203,7 @@ static void *server_thread(void *arg)
 		return NULL; 
 	}
 	url = NULL;
-	if (asprintf(&url, "http://debuginfo.clearlinux.org/debuginfo/%s%s.tar", prefix, path) < 0) {
+	if (asprintf(&url, "%s%s%s.tar", urls[urlcounter % 2], prefix, path) < 0) {
 		close(fd);
 		return NULL;
 	}
@@ -204,14 +211,19 @@ static void *server_thread(void *arg)
 //	printf("Getting url %s    %i:%06i\n", url, before.tv_sec, before.tv_usec);
 	ret = curl_get_file(url, prefix, timestamp);
 
-	/* tell the other side we're done with the download */
-	write(fd, "ok", 3);
-	close(fd);
+	if (ret != 200)
+		printf("Request for %s resulted in error %i\n", url, ret);
 
 	gettimeofday(&after, NULL);
 	if (timedelta(before, after) > 0.6) 
 		printf("Request for %s took %5.2f seconds (%i - %i)\n", url,
 			after.tv_sec - before.tv_sec + (1.0*after.tv_usec - before.tv_usec) / 1000000.0, ret, (int)timestamp);
+
+	/* tell the other side we're done with the download */
+	write(fd, "ok", 3);
+	close(fd);
+
+			
 	free(url);
 	return NULL;
 }
@@ -222,11 +234,11 @@ int main(int argc, char **argv)
 	int sockfd;
 	struct sockaddr_un sun;
 	int ret;
+	int curl_done = 0;
 
 	if (argc < -1)
 		(void)argv;
 
-	curl_global_init(CURL_GLOBAL_ALL);
 
 	signal(SIGPIPE,SIG_IGN);
 
@@ -235,7 +247,6 @@ int main(int argc, char **argv)
 	if (access("/var/cache/debuginfo/src/", F_OK))	
 		system("mkdir -p /var/cache/debuginfo/src/ &> /dev/null");
 
-  
 	sockfd = socket(AF_UNIX, SOCK_STREAM, 0);
 	if (sockfd < 0)
 		return EXIT_FAILURE;
@@ -260,8 +271,14 @@ int main(int argc, char **argv)
 		int clientsock;
 		pthread_t thread;
 
+		malloc_trim(0);
 		clientsock = accept(sockfd, NULL, NULL);
 
+		if (!curl_done) {
+			curl_global_init(CURL_GLOBAL_ALL);
+			curl_done = 1;
+		}
+
 		pthread_create(&thread, NULL, server_thread, (void *)(unsigned long)clientsock);
 		pthread_detach(thread);
 	}