X-Git-Url: https://git.sesse.net/?p=cubemap;a=blobdiff_plain;f=client.h;h=1a7f2a46216dd4f48849860cd4bd6ad2ef1c7286;hp=e82b48758d677740132498f2ca1049151fd34250;hb=31d5c83783aa338e589029a4e6f3f3a4159372b2;hpb=488f28bf7070f44469a006ed4a9d4c423788d175 diff --git a/client.h b/client.h index e82b487..1a7f2a4 100644 --- a/client.h +++ b/client.h @@ -5,41 +5,63 @@ #include #include + +#include #include +#include +#include + +#include "tlse.h" class ClientProto; struct Stream; // Digested statistics for writing to logs etc. +// Note that "referer" and "user_agent" here are already escaped for scary characters. struct ClientStats { - std::string stream_id; + std::string url; + int sock; std::string remote_addr; - time_t connect_time; + std::string referer; + std::string user_agent; + timespec connect_time; size_t bytes_sent; + size_t bytes_lost; + size_t num_loss_events; + std::string hls_zombie_key; }; struct Client { - Client() {} Client(int sock); // Serialization/deserialization. - Client(const ClientProto &serialized, Stream *stream); - ClientProto serialize() const; + Client(const ClientProto &serialized, const std::vector> &short_responses, Stream *stream); + ClientProto serialize(std::unordered_map *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; - // The fwmark associated with this socket (or 0). - int fwmark; + // 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; - time_t connect_time; + std::string referer; + std::string user_agent; + std::string x_playback_session_id; - enum State { READING_REQUEST, SENDING_HEADER, SENDING_DATA, SENDING_ERROR }; - State state; + enum State { READING_REQUEST, SENDING_HEADER, SENDING_DATA, SENDING_SHORT_RESPONSE, WAITING_FOR_KEYFRAME, PREBUFFERING }; + State state = READING_REQUEST; // The HTTP request, as sent by the client. If we are in READING_REQUEST, // this might not be finished. @@ -47,20 +69,78 @@ struct Client { // What stream we're connecting to; parsed from . // Not relevant for READING_REQUEST. - std::string stream_id; - Stream *stream; - - // The header we want to send. This is nominally a copy of Stream::header, - // but since that might change on reconnects etc., we keep a local copy here. - // Only relevant for SENDING_HEADER or SENDING_ERROR; blank otherwise. - std::string header_or_error; + std::string url; + Stream *stream = nullptr; + + // If true, we don't actually serve the stream, but its HLS playlist. + // TODO: Maybe this shouldn't be part of Client, since it's only + // really used in communicating once from parse_request() to + // process_client(); it's not permanent state (and is not serialized). + 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; + + // Whether the request came in over HTTP/1.1 or higher. + bool http_11; + + // The header we want to send, or the response with headers if we know + // it in its entirety after reading the request (typically an error). + // This is nominally a copy of Stream::header, but since that might + // change on reconnects etc., we keep a local copy here. Only relevant + // for SENDING_HEADER or SENDING_SHORT_RESPONSE; blank otherwise. + // + // Must start with the string "HTTP/1.0 ", which will be changed to 1.1 + // if relevant. + const std::string *header_or_short_response = nullptr; + + // 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). 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 header_or_short_response_ref; // Number of bytes we've sent of the header. Only relevant for SENDING_HEADER - // or SENDING_ERROR. - size_t header_or_error_bytes_sent; + // or SENDING_SHORT_RESPONSE. + size_t header_or_short_response_bytes_sent = 0; + + // Number of bytes we are into the stream (ie., the end of last send). + // -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. + 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. + 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; + 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. + uint64_t bytes_lost = 0, num_loss_events = 0; + + TLSContext *tls_context = nullptr; + const unsigned char *tls_data_to_send = nullptr; + unsigned tls_data_left_to_send = 0; + bool in_ktls_mode = false; }; #endif // !defined(_CLIENT_H)