X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=server.cpp;h=e4ea647c64d05555505f3f18358f4e9b642d5370;hp=8c3382d4e8c6506d4c6a99aa686c2bc5121fa436;hb=8f44468bfe4a1d1607b0ab7044c3071605ae1fa7;hpb=340489a8e732519182ecbc92116e7dfa2997143c

diff --git a/server.cpp b/server.cpp
index 8c3382d..e4ea647 100644
--- a/server.cpp
+++ b/server.cpp
@@ -1,3 +1,4 @@
+#include <arpa/inet.h>
 #include <assert.h>
 #include <errno.h>
 #include <pthread.h>
@@ -17,6 +18,7 @@
 #include "accesslog.h"
 #include "log.h"
 #include "markpool.h"
+#include "metacube.h"
 #include "mutexlock.h"
 #include "parse.h"
 #include "server.h"
@@ -194,10 +196,10 @@ void Server::add_client_from_serialized(const ClientProto &client)
 	}
 }
 
-void Server::add_stream(const string &stream_id, size_t backlog_size)
+void Server::add_stream(const string &stream_id, size_t backlog_size, Stream::Encoding encoding)
 {
 	MutexLock lock(&mutex);
-	streams.insert(make_pair(stream_id, new Stream(stream_id, backlog_size)));
+	streams.insert(make_pair(stream_id, new Stream(stream_id, backlog_size, encoding)));
 }
 
 void Server::add_stream_from_serialized(const StreamProto &stream)
@@ -206,17 +208,25 @@ void Server::add_stream_from_serialized(const StreamProto &stream)
 	streams.insert(make_pair(stream.stream_id(), new Stream(stream)));
 }
 	
-void Server::set_backlog_size(const std::string &stream_id, size_t new_size)
+void Server::set_backlog_size(const string &stream_id, size_t new_size)
 {
 	MutexLock lock(&mutex);
 	assert(streams.count(stream_id) != 0);
 	streams[stream_id]->set_backlog_size(new_size);
 }
 	
-void Server::set_header(const string &stream_id, const string &header)
+void Server::set_encoding(const string &stream_id, Stream::Encoding encoding)
 {
 	MutexLock lock(&mutex);
-	find_stream(stream_id)->header = header;
+	assert(streams.count(stream_id) != 0);
+	streams[stream_id]->encoding = encoding;
+}
+	
+void Server::set_header(const string &stream_id, const string &http_header, const string &stream_header)
+{
+	MutexLock lock(&mutex);
+	find_stream(stream_id)->http_header = http_header;
+	find_stream(stream_id)->stream_header = stream_header;
 
 	// If there are clients we haven't sent anything to yet, we should give
 	// them the header, so push back into the SENDING_HEADER state.
@@ -231,7 +241,7 @@ void Server::set_header(const string &stream_id, const string &header)
 	}
 }
 	
-void Server::set_mark_pool(const std::string &stream_id, MarkPool *mark_pool)
+void Server::set_mark_pool(const string &stream_id, MarkPool *mark_pool)
 {
 	MutexLock lock(&mutex);
 	assert(clients.empty());
@@ -454,7 +464,25 @@ int Server::parse_request(Client *client)
 
 void Server::construct_header(Client *client)
 {
-	client->header_or_error = find_stream(client->stream_id)->header;
+	Stream *stream = find_stream(client->stream_id);
+	if (stream->encoding == Stream::STREAM_ENCODING_RAW) {
+		client->header_or_error = stream->http_header +
+			"\r\n" +
+			stream->stream_header;
+	} else if (stream->encoding == Stream::STREAM_ENCODING_METACUBE) {
+		metacube_block_header hdr;
+		memcpy(hdr.sync, METACUBE_SYNC, sizeof(hdr.sync));
+		hdr.size = htonl(stream->stream_header.size());
+		hdr.flags = htonl(METACUBE_FLAGS_HEADER);
+
+		client->header_or_error = stream->http_header +
+			"Content-encoding: metacube\r\n" +
+			"\r\n" +
+			string(reinterpret_cast<char *>(&hdr), sizeof(hdr)) +
+			stream->stream_header;
+	} else {
+		assert(false);
+	}
 
 	// Switch states.
 	client->state = Client::SENDING_HEADER;