Support joining multicast addresses (both ASM and SSM).

[cubemap] / udpinput.h

diff --git a/udpinput.h b/udpinput.h
index 0f89d6828de79b9be3e27db77ed2d7efcf93fb70..b3f504aa20255943302a77e5673489fd89ba4831 100644 (file)
--- a/udpinput.h
+++ b/udpinput.h
@@ -1,8 +1,9 @@
 #ifndef _UDPINPUT_H
 #define _UDPINPUT_H 1
 
-#include <vector>
+#include <pthread.h>
 #include <string>
+#include <vector>
 
 #include "input.h"
 
@@ -10,7 +11,7 @@ class InputProto;
 
 class UDPInput : public Input {
 public:
-       UDPInput(const std::string &stream_id, const std::string &url);
+       UDPInput(const std::string &url);
 
        // Serialization/deserialization.
        UDPInput(const InputProto &serialized);
@@ -19,6 +20,10 @@ public:
        virtual std::string get_url() const { return url; }
        virtual void close_socket();
 
+       virtual void add_destination(int stream_index);
+
+       virtual InputStats get_stats() const;
+
 private:
        // Actually gets the packets.
        virtual void do_work();
@@ -26,14 +31,26 @@ private:
        // Create the HTTP header.
        void construct_header();
 
-       std::string stream_id;
+       std::vector<int> stream_indices;
 
        // The URL and its parsed components.
        std::string url;
-       std::string host, port, path;
+       std::string user, host, port, path;
+
+       // The HTTP header we're sending to clients.
+       std::string http_header;
 
        // The socket we are receiving on (or -1).
        int sock;
+
+       // Temporary buffer, sized for the maximum size of an UDP packet.
+       char packet_buf[65536];
+
+       // Mutex protecting <stats>.
+       mutable pthread_mutex_t stats_mutex;
+
+       // The current statistics for this connection. Protected by <stats_mutex>.
+       InputStats stats;
 };
 
 #endif  // !defined(_UDPINPUT_H)