X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=serverpool.cpp;h=c7ff10cf2866057e3ddc7a996a6d999da967720e;hp=e9fea12944d13455ee7315685a32c7f2ed6fc9ae;hb=HEAD;hpb=7ec54788f88dde7b083ba6cfd30732b32295b54a

diff --git a/serverpool.cpp b/serverpool.cpp
index e9fea12..7b790c2 100644
--- a/serverpool.cpp
+++ b/serverpool.cpp
@@ -1,4 +1,5 @@
 #include <assert.h>
+#include <fcntl.h>
 #include <stdlib.h>
 #include <sys/types.h>
 
@@ -180,10 +181,11 @@ void ServerPool::add_data(int stream_index, const char *data, size_t bytes, uint
 
 void ServerPool::set_unavailable(int stream_index)
 {
-	assert(stream_index >= 0 && stream_index < ssize_t(num_http_streams));
-
-	for (int i = 0; i < num_servers; ++i) {
-		servers[i].set_unavailable(stream_index);
+	assert(stream_index >= 0 && stream_index < ssize_t(num_http_streams + udp_streams.size()));
+	if (stream_index < ssize_t(num_http_streams)) {
+		for (int i = 0; i < num_servers; ++i) {
+			servers[i].set_unavailable(stream_index);
+		}
 	}
 }