+#ifndef LTTNG_RELAYD_H
+#define LTTNG_RELAYD_H
+
/*
* Copyright (C) 2012 - Julien Desfossez <jdesfossez@efficios.com>
* David Goulet <dgoulet@efficios.com>
+ * 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,
* 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>
+
+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,
-};
-
/*
- * 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 {
- uint64_t id;
- struct lttcomm_sock *sock;
- unsigned int version_check_done:1;
-};
+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 seq;
- struct lttng_ht_node_ulong stream_n;
- struct relay_session *session;
- int fd;
-};
+extern char *opt_output_path;
+extern const char *tracing_group_name;
+extern const char * const config_section_name;
-/*
- * 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;
- enum connection_type type;
-};
+extern int thread_quit_pipe[2];
+
+void lttng_relay_notify_ready(void);
+int lttng_relay_stop_threads(void);
#endif /* LTTNG_RELAYD_H */