Fix an issue where new UDP streams would be without HTTP headers.

[cubemap] / udpinput.h

diff --git a/udpinput.h b/udpinput.h
index 0f89d6828de79b9be3e27db77ed2d7efcf93fb70..481b006549f9f249a7ddd971874a78353c26dcd9 100644 (file)
--- a/udpinput.h
+++ b/udpinput.h
@@ -1,8 +1,8 @@
 #ifndef _UDPINPUT_H
 #define _UDPINPUT_H 1
 
-#include <vector>
 #include <string>
+#include <vector>
 
 #include "input.h"
 
@@ -10,7 +10,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 +19,8 @@ public:
        virtual std::string get_url() const { return url; }
        virtual void close_socket();
 
+       virtual void add_destination(const std::string &stream_id);
+
 private:
        // Actually gets the packets.
        virtual void do_work();
@@ -26,12 +28,15 @@ private:
        // Create the HTTP header.
        void construct_header();
 
-       std::string stream_id;
+       std::vector<std::string> stream_ids;
 
        // The URL and its parsed components.
        std::string url;
        std::string 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;
 };