* Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
-#define _GNU_SOURCE
#define _LGPL_SOURCE
#include <common/common.h>
#include <urcu/rculist.h>
{
int ret = 0;
+ ASSERT_LOCKED(session->lock);
+
/* Will not fail, as per the ownership guarantee. */
if (!session_get(session)) {
ret = -1;
goto end;
}
- pthread_mutex_lock(&session->lock);
if (session->viewer_attached) {
ret = -1;
} else {
/* Put our local ref. */
session_put(session);
}
- /* Safe since we know the session exists. */
- pthread_mutex_unlock(&session->lock);
end:
return ret;
}
void viewer_session_destroy(struct relay_viewer_session *vsession)
{
+ lttng_trace_chunk_put(vsession->current_trace_chunk);
free(vsession);
}
+/*
+ * Release ownership of all the streams of one session and detach the viewer.
+ */
+void viewer_session_close_one_session(struct relay_viewer_session *vsession,
+ struct relay_session *session)
+{
+ struct lttng_ht_iter iter;
+ struct relay_viewer_stream *vstream;
+
+ /*
+ * TODO: improvement: create more efficient list of
+ * vstream per session.
+ */
+ cds_lfht_for_each_entry(viewer_streams_ht->ht, &iter.iter,
+ vstream, stream_n.node) {
+ if (!viewer_stream_get(vstream)) {
+ continue;
+ }
+ if (vstream->stream->trace->session != session) {
+ viewer_stream_put(vstream);
+ continue;
+ }
+ /* Put local reference. */
+ viewer_stream_put(vstream);
+ /*
+ * We have reached one of the viewer stream's lifetime
+ * end condition. This "put" will cause the proper
+ * teardown of the viewer stream.
+ */
+ viewer_stream_put(vstream);
+ }
+
+ viewer_session_detach(vsession, session);
+}
+
void viewer_session_close(struct relay_viewer_session *vsession)
{
struct relay_session *session;
rcu_read_lock();
cds_list_for_each_entry_rcu(session,
&vsession->session_list, viewer_session_node) {
- struct lttng_ht_iter iter;
- struct relay_viewer_stream *vstream;
-
- /*
- * TODO: improvement: create more efficient list of
- * vstream per session.
- */
- cds_lfht_for_each_entry(viewer_streams_ht->ht, &iter.iter,
- vstream, stream_n.node) {
- if (!viewer_stream_get(vstream)) {
- continue;
- }
- if (vstream->stream->trace->session != session) {
- viewer_stream_put(vstream);
- continue;
- }
- /* Put local reference. */
- viewer_stream_put(vstream);
- /*
- * We have reached one of the viewer stream's lifetime
- * end condition.
- */
- viewer_stream_put(vstream);
- }
-
- viewer_session_detach(vsession, session);
+ viewer_session_close_one_session(vsession, session);
}
rcu_read_unlock();
}
pthread_mutex_unlock(&session->lock);
return found;
}
+
+int viewer_session_set_trace_chunk(struct relay_viewer_session *vsession,
+ struct lttng_trace_chunk *relay_session_trace_chunk)
+{
+ int ret = 0;
+ struct lttng_trace_chunk *viewer_chunk;
+
+ assert(relay_session_trace_chunk);
+ assert(!vsession->current_trace_chunk);
+
+ DBG("Copying relay session's current trace chunk to the viewer session");
+ viewer_chunk = lttng_trace_chunk_copy(relay_session_trace_chunk);
+ if (!viewer_chunk) {
+ ERR("Failed to create a viewer trace chunk from the relay session's current chunk");
+ ret = -1;
+ goto end;
+ }
+
+ vsession->current_trace_chunk = viewer_chunk;
+end:
+ return ret;
+}