X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Flttng-relayd.h;h=e64bb373fa4200964eae2995374d2d8556766e4c;hp=61db23a7752930ac5b351236f90d3e255181fb27;hb=d7b377ed1acd4043bde353d99122e0e56fa4e975;hpb=309167d2a6f59d0c8cbf64eb23ba912cdea76a34

diff --git a/src/bin/lttng-relayd/lttng-relayd.h b/src/bin/lttng-relayd/lttng-relayd.h
index 61db23a77..e64bb373f 100644
--- a/src/bin/lttng-relayd/lttng-relayd.h
+++ b/src/bin/lttng-relayd/lttng-relayd.h
@@ -1,99 +1,59 @@
+#ifndef LTTNG_RELAYD_H
+#define LTTNG_RELAYD_H
+
 /*
- * Copyright (C) 2012 - Julien Desfossez <jdesfossez@efficios.com>
- *                      David Goulet <dgoulet@efficios.com>
+ * Copyright (C) 2012 Julien Desfossez <jdesfossez@efficios.com>
+ * Copyright (C) 2012 David Goulet <dgoulet@efficios.com>
+ * Copyright (C) 2015 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
  *
- * This program is free software; you can redistribute it and/or modify
- * it under the terms of the GNU General Public License, version 2 only,
- * as published by the Free Software Foundation.
+ * SPDX-License-Identifier: GPL-2.0-only
  *
- * This program is distributed in the hope that it will be useful, but WITHOUT
- * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
- * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
- * more details.
- *
- * You should have received a copy of the GNU General Public License along
- * with this program; if not, write to the Free Software Foundation, Inc.,
- * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  */
 
-#ifndef LTTNG_RELAYD_H
-#define LTTNG_RELAYD_H
-
-#define _LGPL_SOURCE
+#include <limits.h>
 #include <urcu.h>
-#include <urcu/wfqueue.h>
+#include <urcu/wfcqueue.h>
+
 #include <common/hashtable/hashtable.h>
+#include <common/fd-tracker/fd-tracker.h>
+
+struct sessiond_trace_chunk_registry;
 
 /*
  * Queue used to enqueue relay requests
  */
-struct relay_cmd_queue {
-	struct cds_wfq_queue queue;
+struct relay_conn_queue {
+	struct cds_wfcq_head head;
+	struct cds_wfcq_tail tail;
 	int32_t futex;
 };
 
-enum connection_type {
-	RELAY_DATA,
-	RELAY_CONTROL,
+enum relay_group_output_by {
+	RELAYD_GROUP_OUTPUT_BY_UNKNOWN,
+	RELAYD_GROUP_OUTPUT_BY_HOST,
+	RELAYD_GROUP_OUTPUT_BY_SESSION,
 };
 
 /*
- * Represents a session for the relay point of view
+ * Contains stream indexed by ID. This is important since many commands lookup
+ * streams only by ID thus also keeping them in this hash table makes the
+ * search O(1).
  */
-struct relay_session {
-	/*
-	 * This session id is used to identify a set of stream to a tracing session
-	 * but also make sure we have a unique session id associated with a session
-	 * daemon which can provide multiple data source.
-	 */
-	uint64_t id;
-	struct lttcomm_sock *sock;
-};
+extern struct lttng_ht *sessions_ht;
+extern struct lttng_ht *relay_streams_ht;
+extern struct lttng_ht *viewer_streams_ht;
+extern struct sessiond_trace_chunk_registry *sessiond_trace_chunk_registry;
 
-/*
- * Represents a stream in the relay
- */
-struct relay_stream {
-	uint64_t stream_handle;
-	uint64_t prev_seq;	/* previous data sequence number encountered */
-	struct lttng_ht_node_ulong stream_n;
-	struct relay_session *session;
-	struct rcu_head rcu_node;
-	int fd;
-	int index_fd;
+extern char *opt_output_path;
+extern const char *tracing_group_name;
+extern const char * const config_section_name;
+extern enum relay_group_output_by opt_group_output_by;
 
-	char *path_name;
-	char *channel_name;
-	/* on-disk circular buffer of tracefiles */
-	uint64_t tracefile_size;
-	uint64_t tracefile_size_current;
-	uint64_t tracefile_count;
-	uint64_t tracefile_count_current;
+extern int thread_quit_pipe[2];
 
-	/* Information telling us when to close the stream  */
-	unsigned int close_flag:1;
-	uint64_t last_net_seq_num;
-	/* Indicate if the stream was initialized for a data pending command. */
-	unsigned int data_pending_check_done:1;
-};
+extern struct fd_tracker *the_fd_tracker;
 
-/*
- * Internal structure to map a socket with the corresponding session.
- * A hashtable indexed on the socket FD is used for the lookups.
- */
-struct relay_command {
-	struct lttcomm_sock *sock;
-	struct relay_session *session;
-	struct cds_wfq_node node;
-	struct lttng_ht_node_ulong sock_n;
-	struct rcu_head rcu_node;
-	enum connection_type type;
-	unsigned int version_check_done:1;
-	/* protocol version to use for this session */
-	uint32_t major;
-	uint32_t minor;
-};
-
-extern char *opt_output_path;
+void lttng_relay_notify_ready(void);
+int lttng_relay_stop_threads(void);
 
 #endif /* LTTNG_RELAYD_H */