git.sesse.net Git - cubemap/blob - input.h

   1 #ifndef _INPUT_H
   2 #define _INPUT_H 1
   3
   4 #include <stddef.h>
   5 #include <time.h>
   6 #include <string>
   7
   8 #include "thread.h"
   9
  10 class Input;
  11 class InputProto;
  12
  13 // Digested statistics for writing to logs etc.
  14 struct InputStats {
  15         std::string url;
  16
  17         // The number of bytes we have received so far, including any Metacube headers.
  18         //
  19         // Not reset across connections.
  20         size_t bytes_received;
  21
  22         // The number of data bytes we have received so far (or more precisely,
  23         // number of data bytes we have sent on to the stream). This excludes Metacube
  24         // headers and corrupted data we've skipped.
  25         //
  26         // Not reset across connections.
  27         size_t data_bytes_received;
  28
  29         // When the current connection was initiated. -1 if we are not currently connected.
  30         time_t connect_time;
  31
  32         // TODO: Number of loss events might both be useful,
  33         // similar to for clients. Also, per-connection byte counters.
  34 };
  35
  36 class Input : public Thread {
  37 public:
  38         // Must be in sync with StreamConfig::Encoding.
  39         enum Encoding { INPUT_ENCODING_RAW = 0, INPUT_ENCODING_METACUBE };
  40
  41         virtual ~Input();
  42         virtual InputProto serialize() const = 0;
  43         virtual std::string get_url() const = 0;
  44         virtual void close_socket() = 0;
  45         virtual void add_destination(int stream_index) = 0;
  46
  47         // Note: May be called from a different thread, so must be thread-safe.
  48         virtual InputStats get_stats() const = 0;
  49 };
  50
  51 // Extremely rudimentary URL parsing.
  52 bool parse_url(const std::string &url, std::string *protocol, std::string *user, std::string *host, std::string *port, std::string *path);
  53
  54 // Figure out the right type of input based on the URL, and create a new Input of the right type.
  55 // Will return NULL if unknown.
  56 Input *create_input(const std::string &url, Input::Encoding encoding);
  57 Input *create_input(const InputProto &serialized);
  58
  59 #endif  // !defined(_INPUT_H)