Implement snapshot commands in lttng-sessiond

[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 09f4545a03fdc1e40fb777aaff1c4f9cdf278b2d..3c5cf155d07ed14764d9504bbb4f4e0f70d69d95 100644 (file)
--- a/src/bin/lttng-sessiond/consumer.h
+++ b/src/bin/lttng-sessiond/consumer.h
@@ -22,6 +22,11 @@
 #include <common/hashtable/hashtable.h>
 #include <lttng/lttng.h>
 
+#include "snapshot.h"
+
+struct snapshot;
+struct snapshot_output;
+
 enum consumer_dst_type {
        CONSUMER_DST_LOCAL,
        CONSUMER_DST_NET,
@@ -48,6 +53,8 @@ struct consumer_socket {
        unsigned int data_sock_sent;
 
        struct lttng_ht_node_ulong node;
+
+       enum lttng_consumer_type type;
 };
 
 struct consumer_data {
@@ -156,6 +163,8 @@ void consumer_add_socket(struct consumer_socket *sock,
 void consumer_del_socket(struct consumer_socket *sock,
                struct consumer_output *consumer);
 void consumer_destroy_socket(struct consumer_socket *sock);
+int consumer_copy_sockets(struct consumer_output *dst,
+               struct consumer_output *src);
 
 struct consumer_output *consumer_create_output(enum consumer_dst_type type);
 struct consumer_output *consumer_copy_output(struct consumer_output *obj);
@@ -233,4 +242,9 @@ int consumer_push_metadata(struct consumer_socket *socket,
                size_t target_offset);
 int consumer_flush_channel(struct consumer_socket *socket, uint64_t key);
 
+/* Snapshot command. */
+int consumer_snapshot_channel(struct consumer_socket *socket, uint64_t key,
+               struct snapshot_output *output, int metadata, uid_t uid, gid_t gid,
+               int wait);
+
 #endif /* _CONSUMER_H */