X-Git-Url: https://git.lttng.org/?p=lttng-tools.git;a=blobdiff_plain;f=src%2Fbin%2Flttng-relayd%2Flttng-relayd.h;h=0af8e0f1d963199b5a6bb82c28b244a33daa6b6c;hp=c60280e0fd9c717dc0bd0d1e975236ca0e73bed5;hb=cef0f7d51b8025d3ba04e6496242c1cca1641aa6;hpb=1c20f0e29cbf8627bfb1ff444572d52d6655c4e2 diff --git a/src/bin/lttng-relayd/lttng-relayd.h b/src/bin/lttng-relayd/lttng-relayd.h index c60280e0f..0af8e0f1d 100644 --- a/src/bin/lttng-relayd/lttng-relayd.h +++ b/src/bin/lttng-relayd/lttng-relayd.h @@ -20,11 +20,15 @@ #define LTTNG_RELAYD_H #define _LGPL_SOURCE +#include #include #include + #include #include +#include "ctf-trace.h" + /* * Queue used to enqueue relay requests */ @@ -34,8 +38,10 @@ struct relay_cmd_queue { }; enum connection_type { - RELAY_DATA, - RELAY_CONTROL, + RELAY_DATA = 1, + RELAY_CONTROL = 2, + RELAY_VIEWER_COMMAND = 3, + RELAY_VIEWER_NOTIFICATION = 4, }; /* @@ -49,6 +55,23 @@ struct relay_session { */ 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; + /* Tell if this session is for a snapshot or not. */ + unsigned int snapshot:1; + + /* + * Indicate version protocol for this session. This is especially useful + * for the data thread that has no idea which version it operates on since + * linking control/data sockets is non trivial. + */ + uint64_t minor; + uint64_t major; }; /* @@ -63,6 +86,8 @@ struct relay_stream { 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; @@ -71,12 +96,93 @@ struct relay_stream { uint64_t tracefile_size_current; uint64_t tracefile_count; uint64_t tracefile_count_current; + /* To inform the viewer up to where it can go back in time. */ + uint64_t oldest_tracefile_id; + + uint64_t total_index_received; + struct relay_viewer_stream *viewer_stream; + uint64_t last_net_seq_num; + + /* + * This node is added to the *control* connection hash table and the + * pointer is copied in here so we can access it when deleting this object. + * When deleting this, the ctf trace ht MUST NOT be destroyed. This happens + * at connection deletion. + */ + struct lttng_ht_node_str ctf_trace_node; + struct lttng_ht *ctf_traces_ht; + + /* + * 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; + /* + * To protect the update of the close_write_flag and the checks of + * the tracefile_count_current. + * It is taken before checking whenever we need to know if the + * writer and reader are working in the same tracefile. + */ + pthread_mutex_t viewer_stream_rotation_lock; /* Information telling us when to close the stream */ unsigned int close_flag:1; - uint64_t last_net_seq_num; /* Indicate if the stream was initialized for a data pending command. */ unsigned int data_pending_check_done:1; + unsigned int metadata_flag:1; + /* + * To detect when we start overwriting old data, it is used to + * update the oldest_tracefile_id. + */ + unsigned int tracefile_overwrite: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_count; + uint64_t tracefile_count_current; + struct lttng_ht_node_u64 stream_n; + struct rcu_head rcu_node; + struct ctf_trace *ctf_trace; + /* + * This lock blocks only when the writer is about to start overwriting + * a file currently read by the reader. + */ + pthread_mutex_t overwrite_lock; + /* Information telling us if the stream is a metadata stream. */ + unsigned int metadata_flag:1; + /* + * Information telling us that the stream is closed in write, so + * we don't expect new indexes and we can read up to EOF. + */ + unsigned int close_write_flag:1; + /* + * If the streaming side closes a FD in use in the viewer side, + * it sets this flag to inform that it is a normal error. + */ + unsigned int abort_flag:1; }; /* @@ -94,8 +200,22 @@ struct relay_command { /* protocol version to use for this session */ uint32_t major; uint32_t minor; + struct lttng_ht *ctf_traces_ht; /* indexed by path name */ + uint64_t session_id; +}; + +struct relay_local_data { + struct lttng_ht *sessions_ht; }; extern char *opt_output_path; +extern struct lttng_ht *relay_streams_ht; +extern struct lttng_ht *viewer_streams_ht; +extern struct lttng_ht *indexes_ht; + +extern const char *tracing_group_name; + +struct relay_stream *relay_stream_find_by_id(uint64_t stream_id); + #endif /* LTTNG_RELAYD_H */