X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Fconnection.hpp;fp=src%2Fbin%2Flttng-relayd%2Fconnection.hpp;h=de25255a7aab10057d6f0b7eb8631618283ef771;hp=0000000000000000000000000000000000000000;hb=c9e313bc594f40a86eed237dce222c0fc99c957f;hpb=4878de5c7deb512bbdac4fdfc498907efa06fb7c

diff --git a/src/bin/lttng-relayd/connection.hpp b/src/bin/lttng-relayd/connection.hpp
new file mode 100644
index 000000000..de25255a7
--- /dev/null
+++ b/src/bin/lttng-relayd/connection.hpp
@@ -0,0 +1,144 @@
+#ifndef _CONNECTION_H
+#define _CONNECTION_H
+
+/*
+ * Copyright (C) 2013 Julien Desfossez <jdesfossez@efficios.com>
+ * Copyright (C) 2013 David Goulet <dgoulet@efficios.com>
+ * Copyright (C) 2015 Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
+ *
+ * SPDX-License-Identifier: GPL-2.0-only
+ *
+ */
+
+#include <limits.h>
+#include <inttypes.h>
+#include <pthread.h>
+#include <urcu.h>
+#include <urcu/wfcqueue.h>
+#include <urcu/list.h>
+
+#include <common/hashtable/hashtable.hpp>
+#include <common/sessiond-comm/sessiond-comm.hpp>
+#include <common/sessiond-comm/relayd.hpp>
+#include <common/dynamic-buffer.hpp>
+
+#include "session.hpp"
+
+enum connection_type {
+	RELAY_CONNECTION_UNKNOWN    = 0,
+	RELAY_DATA                  = 1,
+	RELAY_CONTROL               = 2,
+	RELAY_VIEWER_COMMAND        = 3,
+	RELAY_VIEWER_NOTIFICATION   = 4,
+};
+
+enum data_connection_state {
+	DATA_CONNECTION_STATE_RECEIVE_HEADER = 0,
+	DATA_CONNECTION_STATE_RECEIVE_PAYLOAD = 1,
+};
+
+enum ctrl_connection_state {
+	CTRL_CONNECTION_STATE_RECEIVE_HEADER = 0,
+	CTRL_CONNECTION_STATE_RECEIVE_PAYLOAD = 1,
+};
+
+struct data_connection_state_receive_header {
+	uint64_t received, left_to_receive;
+	char header_reception_buffer[sizeof(struct lttcomm_relayd_data_hdr)];
+};
+
+struct data_connection_state_receive_payload {
+	uint64_t received, left_to_receive;
+	struct lttcomm_relayd_data_hdr header;
+	bool rotate_index;
+};
+
+struct ctrl_connection_state_receive_header {
+	uint64_t received, left_to_receive;
+};
+
+struct ctrl_connection_state_receive_payload {
+	uint64_t received, left_to_receive;
+	struct lttcomm_relayd_hdr header;
+};
+
+/*
+ * Internal structure to map a socket with the corresponding session.
+ * A hashtable indexed on the socket FD is used for the lookups.
+ *
+ * Connections are assumed to be accessed from a single thread. Live
+ * connections between the relay and a live client are only accessed
+ * from the live worker thread.
+ *
+ * The connections between the consumerd/sessiond and the relayd are only
+ * handled by the "main" worker thread (as in, the worker thread in main.c).
+ *
+ * This is why there are no back references to connections from the
+ * sessions and session list.
+ */
+struct relay_connection {
+	struct lttcomm_sock *sock;
+	struct cds_wfcq_node qnode;
+
+	enum connection_type type;
+	/*
+	 * session is only ever set for RELAY_CONTROL connection type.
+	 */
+	struct relay_session *session;
+	/*
+	 * viewer_session is only ever set for RELAY_VIEWER_COMMAND
+	 * connection type.
+	 */
+	struct relay_viewer_session *viewer_session;
+
+	/*
+	 * Protocol version to use for this connection. Only valid for
+	 * RELAY_CONTROL connection type.
+	 */
+	uint32_t major;
+	uint32_t minor;
+
+	struct urcu_ref ref;
+
+	bool version_check_done;
+
+	/*
+	 * Node member of connection within global socket hash table.
+	 */
+	struct lttng_ht_node_ulong sock_n;
+	bool in_socket_ht;
+	struct lttng_ht *socket_ht;	/* HACK: Contained within this hash table. */
+	struct rcu_head rcu_node;	/* For call_rcu teardown. */
+
+	union {
+		struct {
+			enum data_connection_state state_id;
+			union {
+				struct data_connection_state_receive_header receive_header;
+				struct data_connection_state_receive_payload receive_payload;
+			} state;
+		} data;
+		struct {
+			enum ctrl_connection_state state_id;
+			union {
+				struct ctrl_connection_state_receive_header receive_header;
+				struct ctrl_connection_state_receive_payload receive_payload;
+			} state;
+			struct lttng_dynamic_buffer reception_buffer;
+		} ctrl;
+	} protocol;
+};
+
+struct relay_connection *connection_create(struct lttcomm_sock *sock,
+		enum connection_type type);
+struct relay_connection *connection_get_by_sock(struct lttng_ht *relay_connections_ht,
+		int sock);
+int connection_reset_protocol_state(struct relay_connection *connection);
+bool connection_get(struct relay_connection *connection);
+void connection_put(struct relay_connection *connection);
+void connection_ht_add(struct lttng_ht *relay_connections_ht,
+		struct relay_connection *conn);
+int connection_set_session(struct relay_connection *conn,
+		struct relay_session *session);
+
+#endif /* _CONNECTION_H */