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=da23240341a52b5f99dd1eaa0628b9be77b047eb;hb=d7b377ed1acd4043bde353d99122e0e56fa4e975;hpb=0a6518b0685cb9e07cf156a4c882dc6ec40db35a diff --git a/src/bin/lttng-relayd/lttng-relayd.h b/src/bin/lttng-relayd/lttng-relayd.h index da2324034..e64bb373f 100644 --- a/src/bin/lttng-relayd/lttng-relayd.h +++ b/src/bin/lttng-relayd/lttng-relayd.h @@ -1,173 +1,59 @@ +#ifndef LTTNG_RELAYD_H +#define LTTNG_RELAYD_H + /* - * Copyright (C) 2012 - Julien Desfossez - * David Goulet - * - * 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. + * Copyright (C) 2012 Julien Desfossez + * Copyright (C) 2012 David Goulet + * Copyright (C) 2015 Mathieu Desnoyers * - * 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. + * SPDX-License-Identifier: GPL-2.0-only * - * 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 #include -#include +#include #include -#include +#include -#include "ctf-trace.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 = 1, - RELAY_CONTROL = 2, - RELAY_VIEWER_COMMAND = 3, - RELAY_VIEWER_NOTIFICATION = 4, -}; - -/* - * Represents a session for the relay point of view - */ -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; - char session_name[NAME_MAX]; - char hostname[HOST_NAME_MAX]; - uint32_t live_timer; - struct lttng_ht_node_ulong session_n; - struct rcu_head rcu_node; - uint32_t viewer_attached; - uint32_t stream_count; -}; - -/* - * 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; - /* FD on which to write the index data. */ - int index_fd; - /* FD on which to read the index data for the viewer. */ - int read_index_fd; - - 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; - - uint64_t total_index_received; - struct relay_viewer_stream *viewer_stream; - uint64_t last_net_seq_num; - struct lttng_ht_node_str ctf_trace_node; - - /* - * To protect from concurrent read/update between the - * streaming-side and the viewer-side. - * This lock must be held, we reading/updating the - * ctf_trace pointer. - */ - pthread_mutex_t lock; - - struct ctf_trace *ctf_trace; - /* - * If the stream is inactive, this field is updated with the live beacon - * timestamp end, when it is active, this field == -1ULL. - */ - uint64_t beacon_ts_end; - - /* Information telling us when to close the stream */ - unsigned int close_flag:1; - /* Indicate if the stream was initialized for a data pending command. */ - unsigned int data_pending_check_done:1; - unsigned int metadata_flag:1; -}; - -/* - * Shadow copy of the relay_stream structure for the viewer side. The only - * fields updated by the writer (streaming side) after allocation are : - * total_index_received and close_flag. Everything else is updated by the - * reader (viewer side). - */ -struct relay_viewer_stream { - uint64_t stream_handle; - uint64_t session_id; - int read_fd; - int index_read_fd; - char *path_name; - char *channel_name; - uint64_t last_sent_index; - uint64_t total_index_received; - uint64_t tracefile_size; - uint64_t tracefile_size_current; - uint64_t tracefile_count; - uint64_t tracefile_count_current; - struct lttng_ht_node_u64 stream_n; - struct rcu_head rcu_node; - struct ctf_trace *ctf_trace; - /* Information telling us if the stream is a metadata stream. */ - unsigned int metadata_flag:1; +enum relay_group_output_by { + RELAYD_GROUP_OUTPUT_BY_UNKNOWN, + RELAYD_GROUP_OUTPUT_BY_HOST, + RELAYD_GROUP_OUTPUT_BY_SESSION, }; /* - * Internal structure to map a socket with the corresponding session. - * A hashtable indexed on the socket FD is used for the lookups. + * 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_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; - struct lttng_ht *ctf_traces_ht; /* indexed by path name */ -}; - -struct relay_local_data { - struct lttng_ht *sessions_ht; -}; +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; 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; -extern struct lttng_ht *relay_streams_ht; -extern struct lttng_ht *viewer_streams_ht; -extern struct lttng_ht *indexes_ht; +extern int thread_quit_pipe[2]; + +extern struct fd_tracker *the_fd_tracker; -struct relay_stream *relay_stream_find_by_id(uint64_t stream_id); +void lttng_relay_notify_ready(void); +int lttng_relay_stop_threads(void); #endif /* LTTNG_RELAYD_H */