-#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.h>
-#include <common/sessiond-comm/sessiond-comm.h>
-#include <common/sessiond-comm/relayd.h>
-#include <common/dynamic-buffer.h>
-
-#include "session.h"
-
-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 */