X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=client.h;h=e3fb26bd895698af8fcca68b36934c827eba7116;hp=f1d14e65cf2cbc321770d069aeed8f93fc087f94;hb=ed218ece51e5be77fd8bd8f014b5f05708e4be5b;hpb=019d476c729a6c1b6ac001e345d44339ba423221

diff --git a/client.h b/client.h
index f1d14e6..e3fb26b 100644
--- a/client.h
+++ b/client.h
@@ -5,7 +5,11 @@
 
 #include <stddef.h>
 #include <time.h>
+
+#include <memory>
 #include <string>
+#include <unordered_map>
+#include <vector>
 
 #include "tlse.h"
 
@@ -24,27 +28,37 @@ struct ClientStats {
 	size_t bytes_sent;
 	size_t bytes_lost;
 	size_t num_loss_events;
+	std::string hls_zombie_key;
 };
 
 struct Client {
 	Client(int sock);
 
 	// Serialization/deserialization.
-	Client(const ClientProto &serialized, Stream *stream);
-	ClientProto serialize() const;
+	Client(const ClientProto &serialized, const std::vector<std::shared_ptr<const std::string>> &short_responses, Stream *stream);
+	ClientProto serialize(std::unordered_map<const std::string *, size_t> *short_response_pool) const;
 
 	ClientStats get_stats() const;
 
+	std::string get_hls_zombie_key() const {
+		if (x_playback_session_id.empty()) {
+			return remote_addr;
+		} else {
+			return x_playback_session_id;
+		}
+	}
+
 	// The file descriptor associated with this socket.
 	int sock;
 
-	// When the client connected, in terms of CLOCK_MONOTONIC_COARSE.
+	// When the client connected (or went into keepalive), in terms of CLOCK_MONOTONIC_COARSE.
 	timespec connect_time;
 
 	// Some information only used for logging.
 	std::string remote_addr;
 	std::string referer;
 	std::string user_agent;
+	std::string x_playback_session_id;
 
 	enum State { READING_REQUEST, SENDING_HEADER, SENDING_DATA, SENDING_SHORT_RESPONSE, WAITING_FOR_KEYFRAME, PREBUFFERING };
 	State state = READING_REQUEST;
@@ -58,6 +72,9 @@ struct Client {
 	std::string url;
 	Stream *stream = nullptr;
 
+	// If true, we don't actually serve the stream, but its HLS playlist.
+	bool serving_hls_playlist = false;
+
 	// Whether we should close the connection after sending the response.
 	// Not relevant for READING_REQUEST. Must be true if http_11 is false.
 	bool close_after_response;
@@ -73,7 +90,20 @@ struct Client {
 	//
 	// Must start with the string "HTTP/1.0 ", which will be changed to 1.1
 	// if relevant.
-	std::string header_or_short_response;
+	const std::string *header_or_short_response = nullptr;
+
+	// <header_or_short_response> can come from two distinct places; it can be
+	// local to the Client object, or it can be shared between many Clients
+	// (typically HLS playlists, that can be so large that they are expensive
+	// to hold in many copies). <header_or_short_response> will point to exactly
+	// one of these, which should be cleared out/dereferenced when it is
+	// no longer needed.
+	//
+	// The use of shared_ptr is somewhat overkill since we don't need
+	// to access the HLS playlists from multiple threads, but it's not a
+	// big deal for us.
+	std::string header_or_short_response_holder;
+	std::shared_ptr<const std::string> header_or_short_response_ref;
 
 	// Number of bytes we've sent of the header. Only relevant for SENDING_HEADER
 	// or SENDING_SHORT_RESPONSE.
@@ -83,20 +113,26 @@ struct Client {
 	// -1 means we want to send from the end of the backlog (the normal case),
 	// although only at a keyframe.
 	// -2 means we want to send from the _beginning_ of the backlog.
+	// -3 means we sent the header only.
+	static const uint64_t STREAM_POS_AT_END = -1;
+	static const uint64_t STREAM_POS_AT_START = -2;
+	static const uint64_t STREAM_POS_HEADER_ONLY = -3;
+
 	// Once we go into WAITING_FOR_KEYFRAME, PREBUFFERING or SENDING_DATA,
 	// these negative values will be translated to real numbers.
-	size_t stream_pos = 0;
+	uint64_t stream_pos = 0;
 
 	// Position at which to end the stream (one-past-the-end, used for fragments).
 	// -1 means never to end; this is the common case.
-	size_t stream_pos_end = 0;
+	static const uint64_t STREAM_POS_NO_END = -1;
+	uint64_t stream_pos_end = 0;
 
 	// Number of bytes we've sent of data. Only relevant for SENDING_DATA.
-	size_t bytes_sent = 0;
+	uint64_t bytes_sent = 0;
 
 	// Number of times we've skipped forward due to the backlog being too big,
 	// and how many bytes we've skipped over in all. Only relevant for SENDING_DATA.
-	size_t bytes_lost = 0, num_loss_events = 0;
+	uint64_t bytes_lost = 0, num_loss_events = 0;
 
 	TLSContext *tls_context = nullptr;
 	const unsigned char *tls_data_to_send = nullptr;