2 * Copyright (C) 2013 - Julien Desfossez <jdesfossez@efficios.com>
3 * David Goulet <dgoulet@efficios.com>
5 * This program is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License, version 2 only, as
7 * 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 with
15 * this program; if not, write to the Free Software Foundation, Inc., 51
16 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
21 #include <common/common.h>
25 #include "viewer-stream.h"
27 static void rcu_destroy_stream(struct rcu_head
*head
)
29 struct relay_stream
*stream
=
30 caa_container_of(head
, struct relay_stream
, rcu_node
);
32 free(stream
->path_name
);
33 free(stream
->channel_name
);
38 * Get stream from stream id from the given hash table. Return stream if found
41 * Need to be called with RCU read-side lock held.
43 struct relay_stream
*stream_find_by_id(struct lttng_ht
*ht
,
46 struct lttng_ht_node_u64
*node
;
47 struct lttng_ht_iter iter
;
48 struct relay_stream
*stream
= NULL
;
52 lttng_ht_lookup(ht
, &stream_id
, &iter
);
53 node
= lttng_ht_iter_get_node_u64(&iter
);
55 DBG("Relay stream %" PRIu64
" not found", stream_id
);
58 stream
= caa_container_of(node
, struct relay_stream
, node
);
65 * Close a given stream. If an assosiated viewer stream exists it is updated.
67 * RCU read side lock MUST be acquired.
69 * Return 0 if close was successful or 1 if already closed.
71 int stream_close(struct relay_session
*session
, struct relay_stream
*stream
)
74 struct relay_viewer_stream
*vstream
;
75 struct ctf_trace
*ctf_trace
;
79 pthread_mutex_lock(&stream
->lock
);
81 if (stream
->terminated_flag
) {
82 /* This stream is already closed. Ignore. */
87 DBG("Closing stream id %" PRIu64
, stream
->stream_handle
);
89 if (stream
->fd
>= 0) {
90 delret
= close(stream
->fd
);
92 PERROR("close stream");
96 if (stream
->index_fd
>= 0) {
97 delret
= close(stream
->index_fd
);
99 PERROR("close stream index_fd");
103 vstream
= viewer_stream_find_by_id(stream
->stream_handle
);
106 * Set the last good value into the viewer stream. This is done
107 * right before the stream gets deleted from the hash table. The
108 * lookup failure on the live thread side of a stream indicates
109 * that the viewer stream index received value should be used.
111 pthread_mutex_lock(&stream
->viewer_stream_rotation_lock
);
112 vstream
->total_index_received
= stream
->total_index_received
;
113 vstream
->tracefile_count_last
= stream
->tracefile_count_current
;
114 vstream
->close_write_flag
= 1;
115 pthread_mutex_unlock(&stream
->viewer_stream_rotation_lock
);
118 /* Cleanup index of that stream. */
119 relay_index_destroy_by_stream_id(stream
->stream_handle
);
121 ctf_trace
= ctf_trace_find_by_path(session
->ctf_traces_ht
,
124 ctf_trace_put_ref(ctf_trace
);
126 stream
->close_flag
= 1;
127 stream
->terminated_flag
= 1;
131 pthread_mutex_unlock(&stream
->lock
);
135 void stream_delete(struct lttng_ht
*ht
, struct relay_stream
*stream
)
138 struct lttng_ht_iter iter
;
143 iter
.iter
.node
= &stream
->node
.node
;
144 ret
= lttng_ht_del(ht
, &iter
);
147 cds_list_del(&stream
->trace_list
);
150 void stream_destroy(struct relay_stream
*stream
)
154 call_rcu(&stream
->rcu_node
, rcu_destroy_stream
);