X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=include%2Flttng%2Flttng-consumer.h;h=e5672d7b78ef81f45d7ea8134bb288f78253f8dc;hp=7751570efef38cfde7319b97bb3f114381c7b793;hb=e4baff1ed855ca743e95068ca6449fd906d2d833;hpb=b5c5fc2947ec79834e054e0117414b717faeee4d

diff --git a/include/lttng/lttng-consumer.h b/include/lttng/lttng-consumer.h
index 7751570ef..e5672d7b7 100644
--- a/include/lttng/lttng-consumer.h
+++ b/include/lttng/lttng-consumer.h
@@ -68,7 +68,8 @@ struct lttng_consumer_stream_list {
 enum lttng_consumer_type {
 	LTTNG_CONSUMER_UNKNOWN = 0,
 	LTTNG_CONSUMER_KERNEL,
-	LTTNG_CONSUMER_UST,
+	LTTNG_CONSUMER64_UST,
+	LTTNG_CONSUMER32_UST,
 };
 
 struct lttng_consumer_channel {
@@ -85,6 +86,7 @@ struct lttng_consumer_channel {
 	int nr_streams;
 	int shm_fd_is_copy;
 	int wait_fd_is_copy;
+	int cpucount;
 };
 
 /* Forward declaration for UST. */
@@ -117,6 +119,7 @@ struct lttng_consumer_stream {
 	/* For UST */
 	struct lttng_ust_lib_ring_buffer *buf;
 	int cpu;
+	int hangup_flush_done;
 };
 
 /*
@@ -125,7 +128,8 @@ struct lttng_consumer_stream {
  */
 struct lttng_consumer_local_data {
 	/* function to call when data is available on a buffer */
-	int (*on_buffer_ready)(struct lttng_consumer_stream *stream);
+	int (*on_buffer_ready)(struct lttng_consumer_stream *stream,
+			struct lttng_consumer_local_data *ctx);
 	/*
 	 * function to call when we receive a new channel, it receives a
 	 * newly allocated channel, depending on the return code of this
@@ -276,7 +280,8 @@ int consumer_add_channel(struct lttng_consumer_channel *channel);
 
 extern struct lttng_consumer_local_data *lttng_consumer_create(
 		enum lttng_consumer_type type,
-		int (*buffer_ready)(struct lttng_consumer_stream *stream),
+		int (*buffer_ready)(struct lttng_consumer_stream *stream,
+			struct lttng_consumer_local_data *ctx),
 		int (*recv_channel)(struct lttng_consumer_channel *channel),
 		int (*recv_stream)(struct lttng_consumer_stream *stream),
 		int (*update_stream)(int sessiond_key, uint32_t state));
@@ -298,4 +303,8 @@ extern void *lttng_consumer_thread_receive_fds(void *data);
 extern int lttng_consumer_recv_cmd(struct lttng_consumer_local_data *ctx,
 		int sock, struct pollfd *consumer_sockpoll);
 
+int lttng_consumer_read_subbuffer(struct lttng_consumer_stream *stream,
+		struct lttng_consumer_local_data *ctx);
+int lttng_consumer_on_recv_stream(struct lttng_consumer_stream *stream);
+
 #endif /* _LTTNG_CONSUMER_H */