X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=udpinput.cpp;h=3e4b9b79c0cd872bcd140401b0f9a7a16238b21f;hp=07bf63d344e8ba6e38d7d8d399acafad3016058c;hb=845934ca50eee40884e8cc85ea81eb310efa5ca3;hpb=3fd8650ccf3da3960a946d8ac9abc305aec399ce

diff --git a/udpinput.cpp b/udpinput.cpp
index 07bf63d..3e4b9b7 100644
--- a/udpinput.cpp
+++ b/udpinput.cpp
@@ -1,7 +1,7 @@
 #include <assert.h>
 #include <errno.h>
 #include <poll.h>
-#include <stdio.h>
+#include <stddef.h>
 #include <stdlib.h>
 #include <sys/socket.h>
 #include <unistd.h>
@@ -67,10 +67,10 @@ void UDPInput::construct_header()
 	        "Connection: close\r\n";
 }
 	
-void UDPInput::add_destination(const string &stream_id)
+void UDPInput::add_destination(int stream_index)
 {
-	stream_ids.push_back(stream_id);
-	servers->set_header(stream_id, http_header, "");
+	stream_indices.push_back(stream_index);
+	servers->set_header(stream_index, http_header, "");
 }
 
 void UDPInput::do_work()
@@ -94,10 +94,9 @@ void UDPInput::do_work()
 			continue;
 		}
 
-		char buf[4096];
 		int ret;
 		do {
-			ret = recv(sock, buf, sizeof(buf), 0);
+			ret = recv(sock, packet_buf, sizeof(packet_buf), 0);
 		} while (ret == -1 && errno == EINTR);
 
 		if (ret <= 0) {
@@ -106,8 +105,8 @@ void UDPInput::do_work()
 			continue;
 		}
 		
-		for (size_t i = 0; i < stream_ids.size(); ++i) {
-			servers->add_data(stream_ids[i], buf, ret);
+		for (size_t i = 0; i < stream_indices.size(); ++i) {
+			servers->add_data(stream_indices[i], packet_buf, ret);
 		}
 	}
 }