Network streaming support

[lttng-tools.git] / src / bin / lttng-sessiond / consumer.h

diff --git a/src/bin/lttng-sessiond/consumer.h b/src/bin/lttng-sessiond/consumer.h
index 21d762a1582028a8ac206104943f8e57a16d9cfa..2eb9d7433455f8156d83678029628ab7a969c41c 100644 (file)
--- a/src/bin/lttng-sessiond/consumer.h
+++ b/src/bin/lttng-sessiond/consumer.h
@@ -21,6 +21,12 @@
 #include <semaphore.h>
 
 #include <common/consumer.h>
+#include <lttng/lttng.h>
+
+enum consumer_dst_type {
+       CONSUMER_DST_LOCAL,
+       CONSUMER_DST_NET,
+};
 
 struct consumer_data {
        enum lttng_consumer_type type;
@@ -40,4 +46,81 @@ struct consumer_data {
        char cmd_unix_sock_path[PATH_MAX];
 };
 
+/*
+ * Network URIs
+ */
+struct consumer_net {
+       /*
+        * Indicate if URI type is set. Those flags should only be set when the
+        * created URI is done AND valid.
+        */
+       int control_isset;
+       int data_isset;
+
+       /*
+        * The following two URIs MUST have the same destination address for
+        * network streaming to work. Network hop are not yet supported.
+        */
+
+       /* Control path for network streaming. */
+       struct lttng_uri control;
+
+       /* Data path for network streaming. */
+       struct lttng_uri data;
+};
+
+/*
+ * Consumer output object describing where and how to send data.
+ */
+struct consumer_output {
+       /* Consumer socket file descriptor */
+       int sock;
+       /* If the consumer is enabled meaning that should be used */
+       unsigned int enabled;
+       enum consumer_dst_type type;
+       /*
+        * The net_seq_index is the index of the network stream on the consumer
+        * side. It's basically the relayd socket file descriptor value so the
+        * consumer can identify which streams goes with which socket.
+        */
+       int net_seq_index;
+       /*
+        * Subdirectory path name used for both local and network consumer.
+        */
+       char subdir[PATH_MAX];
+       union {
+               char trace_path[PATH_MAX];
+               struct consumer_net net;
+       } dst;
+};
+
+struct consumer_output *consumer_create_output(enum consumer_dst_type type);
+struct consumer_output *consumer_copy_output(struct consumer_output *obj);
+void consumer_destroy_output(struct consumer_output *obj);
+int consumer_set_network_uri(struct consumer_output *obj,
+               struct lttng_uri *uri);
+int consumer_send_fds(int sock, int *fds, size_t nb_fd);
+int consumer_send_stream(int sock, struct consumer_output *dst,
+               struct lttcomm_consumer_msg *msg, int *fds, size_t nb_fd);
+int consumer_send_channel(int sock, struct lttcomm_consumer_msg *msg);
+void consumer_init_stream_comm_msg(struct lttcomm_consumer_msg *msg,
+               enum lttng_consumer_command cmd,
+               int channel_key,
+               int stream_key,
+               uint32_t state,
+               enum lttng_event_output output,
+               uint64_t mmap_len,
+               uid_t uid,
+               gid_t gid,
+               int net_index,
+               unsigned int metadata_flag,
+               const char *name,
+               const char *pathname);
+void consumer_init_channel_comm_msg(struct lttcomm_consumer_msg *msg,
+               enum lttng_consumer_command cmd,
+               int channel_key,
+               uint64_t max_sb_size,
+               uint64_t mmap_len,
+               const char *name);
+
 #endif /* _CONSUMER_H */