2 * Copyright (C) 2012 - Julien Desfossez <jdesfossez@efficios.com>
3 * David Goulet <dgoulet@efficios.com>
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License, version 2 only,
7 * as published by the Free Software Foundation.
9 * This program is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
19 #ifndef LTTNG_RELAYD_H
20 #define LTTNG_RELAYD_H
25 #include <urcu/wfqueue.h>
27 #include <common/hashtable/hashtable.h>
28 #include <common/index/lttng-index.h>
30 #include "ctf-trace.h"
33 * Queue used to enqueue relay requests
35 struct relay_cmd_queue
{
36 struct cds_wfq_queue queue
;
40 enum connection_type
{
43 RELAY_VIEWER_COMMAND
= 3,
44 RELAY_VIEWER_NOTIFICATION
= 4,
48 * Represents a session for the relay point of view
50 struct relay_session
{
52 * This session id is used to identify a set of stream to a tracing session
53 * but also make sure we have a unique session id associated with a session
54 * daemon which can provide multiple data source.
57 struct lttcomm_sock
*sock
;
58 char session_name
[NAME_MAX
];
59 char hostname
[HOST_NAME_MAX
];
61 struct lttng_ht_node_ulong session_n
;
62 struct rcu_head rcu_node
;
63 uint32_t viewer_attached
;
64 uint32_t stream_count
;
68 * Represents a stream in the relay
71 uint64_t stream_handle
;
72 uint64_t prev_seq
; /* previous data sequence number encountered */
73 struct lttng_ht_node_ulong stream_n
;
74 struct relay_session
*session
;
75 struct rcu_head rcu_node
;
77 /* FD on which to write the index data. */
79 /* FD on which to read the index data for the viewer. */
84 /* on-disk circular buffer of tracefiles */
85 uint64_t tracefile_size
;
86 uint64_t tracefile_size_current
;
87 uint64_t tracefile_count
;
88 uint64_t tracefile_count_current
;
90 uint64_t total_index_received
;
91 struct relay_viewer_stream
*viewer_stream
;
92 uint64_t last_net_seq_num
;
93 struct lttng_ht_node_str ctf_trace_node
;
96 * To protect from concurrent read/update between the
97 * streaming-side and the viewer-side.
98 * This lock must be held, we reading/updating the
101 pthread_mutex_t lock
;
103 struct ctf_trace
*ctf_trace
;
105 * If the stream is inactive, this field is updated with the live beacon
106 * timestamp end, when it is active, this field == -1ULL.
108 uint64_t beacon_ts_end
;
110 /* Information telling us when to close the stream */
111 unsigned int close_flag
:1;
112 /* Indicate if the stream was initialized for a data pending command. */
113 unsigned int data_pending_check_done
:1;
114 unsigned int metadata_flag
:1;
118 * Shadow copy of the relay_stream structure for the viewer side. The only
119 * fields updated by the writer (streaming side) after allocation are :
120 * total_index_received and close_flag. Everything else is updated by the
121 * reader (viewer side).
123 struct relay_viewer_stream
{
124 uint64_t stream_handle
;
130 uint64_t last_sent_index
;
131 uint64_t total_index_received
;
132 uint64_t tracefile_size
;
133 uint64_t tracefile_size_current
;
134 uint64_t tracefile_count
;
135 uint64_t tracefile_count_current
;
136 struct lttng_ht_node_u64 stream_n
;
137 struct rcu_head rcu_node
;
138 struct ctf_trace
*ctf_trace
;
139 /* Information telling us if the stream is a metadata stream. */
140 unsigned int metadata_flag
:1;
144 * Internal structure to map a socket with the corresponding session.
145 * A hashtable indexed on the socket FD is used for the lookups.
147 struct relay_command
{
148 struct lttcomm_sock
*sock
;
149 struct relay_session
*session
;
150 struct cds_wfq_node node
;
151 struct lttng_ht_node_ulong sock_n
;
152 struct rcu_head rcu_node
;
153 enum connection_type type
;
154 unsigned int version_check_done
:1;
155 /* protocol version to use for this session */
158 struct lttng_ht
*ctf_traces_ht
; /* indexed by path name */
161 struct relay_local_data
{
162 struct lttng_ht
*sessions_ht
;
165 extern char *opt_output_path
;
167 extern struct lttng_ht
*relay_streams_ht
;
168 extern struct lttng_ht
*viewer_streams_ht
;
170 struct relay_stream
*relay_stream_find_by_id(uint64_t stream_id
);
172 #endif /* LTTNG_RELAYD_H */