Fix: honor base path for network URIs
[lttng-tools.git] / src / bin / lttng-relayd / session.c
CommitLineData
2f8f53af
DG
1/*
2 * Copyright (C) 2013 - Julien Desfossez <jdesfossez@efficios.com>
3 * David Goulet <dgoulet@efficios.com>
7591bab1 4 * 2015 - Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
2f8f53af
DG
5 *
6 * This program is free software; you can redistribute it and/or modify it
7 * under the terms of the GNU General Public License, version 2 only, as
8 * published by the Free Software Foundation.
9 *
10 * This program is distributed in the hope that it will be useful, but WITHOUT
11 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
12 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
13 * more details.
14 *
15 * You should have received a copy of the GNU General Public License along with
16 * this program; if not, write to the Free Software Foundation, Inc., 51
17 * Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
18 */
19
6c1c0768 20#define _LGPL_SOURCE
2a174661 21#include <common/common.h>
a620c891 22#include <common/utils.h>
1e791a74 23#include <common/compat/uuid.h>
7591bab1 24#include <urcu/rculist.h>
2a174661
DG
25
26#include "ctf-trace.h"
a620c891 27#include "lttng-relayd.h"
2f8f53af 28#include "session.h"
23c8ff50 29#include "sessiond-trace-chunks.h"
a620c891 30#include "stream.h"
2a174661
DG
31
32/* Global session id used in the session creation. */
33static uint64_t last_relay_session_id;
7591bab1 34static pthread_mutex_t last_relay_session_id_lock = PTHREAD_MUTEX_INITIALIZER;
2a174661 35
1e791a74
JG
36static int session_set_anonymous_chunk(struct relay_session *session)
37{
38 int ret = 0;
39 struct lttng_trace_chunk *chunk = NULL;
40 enum lttng_trace_chunk_status status;
41 struct lttng_directory_handle output_directory;
4f00620d 42 const char *base_path = opt_output_path;
a620c891
JR
43
44 if (base_path == NULL) {
45 /* No output path defined */
46 base_path = utils_get_home_dir();
47 if (base_path == NULL) {
48 ERR("Home path not found.\n \
49 Please specify an output path using -o, --output PATH");
50 ret = -1;
51 goto end;
52 }
53 }
1e791a74 54
a620c891 55 ret = lttng_directory_handle_init(&output_directory, base_path);
1e791a74
JG
56 if (ret) {
57 goto end;
58 }
59
60 chunk = lttng_trace_chunk_create_anonymous();
61 if (!chunk) {
62 goto end;
63 }
64
65 status = lttng_trace_chunk_set_credentials_current_user(chunk);
66 if (status != LTTNG_TRACE_CHUNK_STATUS_OK) {
67 ret = -1;
68 goto end;
69 }
70
71 status = lttng_trace_chunk_set_as_owner(chunk, &output_directory);
72 if (status != LTTNG_TRACE_CHUNK_STATUS_OK) {
73 ret = -1;
74 goto end;
75 }
76 session->current_trace_chunk = chunk;
77 chunk = NULL;
78end:
79 lttng_trace_chunk_put(chunk);
80 lttng_directory_handle_fini(&output_directory);
81 return ret;
82}
83
2a174661
DG
84/*
85 * Create a new session by assigning a new session ID.
86 *
87 * Return allocated session or else NULL.
88 */
7591bab1 89struct relay_session *session_create(const char *session_name,
6fa5fe7c 90 const char *hostname, const char *base_path,
db1da059
JG
91 uint32_t live_timer,
92 bool snapshot,
93 const lttng_uuid sessiond_uuid,
94 const uint64_t *id_sessiond,
95 const uint64_t *current_chunk_id,
96 const time_t *creation_time,
97 uint32_t major,
98 uint32_t minor)
2a174661 99{
23c8ff50 100 int ret;
2a174661
DG
101 struct relay_session *session;
102
103 session = zmalloc(sizeof(*session));
104 if (!session) {
1e791a74 105 PERROR("Failed to allocate session");
2a174661
DG
106 goto error;
107 }
bb5d54e7
MD
108 if (lttng_strncpy(session->session_name, session_name,
109 sizeof(session->session_name))) {
1e791a74 110 WARN("Session name exceeds maximal allowed length");
bb5d54e7
MD
111 goto error;
112 }
113 if (lttng_strncpy(session->hostname, hostname,
114 sizeof(session->hostname))) {
1e791a74 115 WARN("Hostname exceeds maximal allowed length");
bb5d54e7
MD
116 goto error;
117 }
6fa5fe7c
MD
118 if (lttng_strncpy(session->base_path, base_path,
119 sizeof(session->base_path))) {
120 WARN("Base path exceeds maximal allowed length");
121 goto error;
122 }
123
2a174661
DG
124 session->ctf_traces_ht = lttng_ht_new(0, LTTNG_HT_TYPE_STRING);
125 if (!session->ctf_traces_ht) {
2a174661
DG
126 goto error;
127 }
128
7591bab1 129 pthread_mutex_lock(&last_relay_session_id_lock);
2a174661 130 session->id = ++last_relay_session_id;
7591bab1
MD
131 pthread_mutex_unlock(&last_relay_session_id_lock);
132
133 session->major = major;
134 session->minor = minor;
2a174661 135 lttng_ht_node_init_u64(&session->session_n, session->id);
7591bab1
MD
136 urcu_ref_init(&session->ref);
137 CDS_INIT_LIST_HEAD(&session->recv_list);
138 pthread_mutex_init(&session->lock, NULL);
7591bab1
MD
139 pthread_mutex_init(&session->recv_list_lock, NULL);
140
7591bab1
MD
141 session->live_timer = live_timer;
142 session->snapshot = snapshot;
23c8ff50
JG
143 lttng_uuid_copy(session->sessiond_uuid, sessiond_uuid);
144
1e791a74
JG
145 if (id_sessiond) {
146 LTTNG_OPTIONAL_SET(&session->id_sessiond, *id_sessiond);
147 }
148
23c8ff50
JG
149 ret = sessiond_trace_chunk_registry_session_created(
150 sessiond_trace_chunk_registry, sessiond_uuid);
151 if (ret) {
152 goto error;
153 }
7591bab1 154
1e791a74
JG
155 if (id_sessiond && current_chunk_id) {
156 session->current_trace_chunk =
157 sessiond_trace_chunk_registry_get_chunk(
158 sessiond_trace_chunk_registry,
159 session->sessiond_uuid,
160 session->id_sessiond.value,
161 *current_chunk_id);
162 if (!session->current_trace_chunk) {
163 char uuid_str[UUID_STR_LEN];
164
165 lttng_uuid_to_str(sessiond_uuid, uuid_str);
166 ERR("Could not find trace chunk: sessiond = {%s}, sessiond session id = %" PRIu64 ", trace chunk id = %" PRIu64,
167 uuid_str, *id_sessiond,
168 *current_chunk_id);
169 }
170 } else if (!id_sessiond) {
171 /*
172 * Pre-2.11 peers will not announce trace chunks. An
173 * anonymous trace chunk which will remain set for the
174 * duration of the session is created.
175 */
176 ret = session_set_anonymous_chunk(session);
177 if (ret) {
178 goto error;
179 }
180 }
181
7591bab1 182 lttng_ht_add_unique_u64(sessions_ht, &session->session_n);
db1da059
JG
183 if (creation_time) {
184 LTTNG_OPTIONAL_SET(&session->creation_time, *creation_time);
185 }
bb5d54e7 186 return session;
2a174661
DG
187
188error:
1e791a74 189 session_put(session);
bb5d54e7 190 return NULL;
2a174661 191}
2f8f53af 192
7591bab1
MD
193/* Should be called with RCU read-side lock held. */
194bool session_get(struct relay_session *session)
195{
ce4d4083 196 return urcu_ref_get_unless_zero(&session->ref);
7591bab1
MD
197}
198
2f8f53af 199/*
7591bab1
MD
200 * Lookup a session within the session hash table using the session id
201 * as key. A session reference is taken when a session is returned.
202 * session_put() must be called on that session.
2f8f53af
DG
203 *
204 * Return session or NULL if not found.
205 */
7591bab1 206struct relay_session *session_get_by_id(uint64_t id)
2f8f53af
DG
207{
208 struct relay_session *session = NULL;
2a174661 209 struct lttng_ht_node_u64 *node;
2f8f53af
DG
210 struct lttng_ht_iter iter;
211
7591bab1
MD
212 rcu_read_lock();
213 lttng_ht_lookup(sessions_ht, &id, &iter);
2a174661 214 node = lttng_ht_iter_get_node_u64(&iter);
2f8f53af 215 if (!node) {
2a174661 216 DBG("Session find by ID %" PRIu64 " id NOT found", id);
2f8f53af
DG
217 goto end;
218 }
219 session = caa_container_of(node, struct relay_session, session_n);
2a174661 220 DBG("Session find by ID %" PRIu64 " id found", id);
7591bab1
MD
221 if (!session_get(session)) {
222 session = NULL;
223 }
2f8f53af 224end:
7591bab1 225 rcu_read_unlock();
2f8f53af
DG
226 return session;
227}
2a174661 228
7591bab1
MD
229static void rcu_destroy_session(struct rcu_head *rcu_head)
230{
231 struct relay_session *session =
232 caa_container_of(rcu_head, struct relay_session,
233 rcu_node);
49e614cb
MD
234 /*
235 * Since each trace has a reference on the session, it means
236 * that if we are at the point where we teardown the session, no
237 * trace belonging to that session exist at this point.
238 * Calling lttng_ht_destroy in call_rcu worker thread so we
239 * don't hold the RCU read-side lock while calling it.
240 */
241 lttng_ht_destroy(session->ctf_traces_ht);
7591bab1
MD
242 free(session);
243}
244
2a174661
DG
245/*
246 * Delete session from the given hash table.
247 *
248 * Return lttng ht del error code being 0 on success and 1 on failure.
249 */
7591bab1 250static int session_delete(struct relay_session *session)
2a174661
DG
251{
252 struct lttng_ht_iter iter;
253
2a174661 254 iter.iter.node = &session->session_n.node;
7591bab1 255 return lttng_ht_del(sessions_ht, &iter);
2a174661
DG
256}
257
7591bab1
MD
258
259static void destroy_session(struct relay_session *session)
260{
261 int ret;
262
263 ret = session_delete(session);
264 assert(!ret);
639ddf68 265 lttng_trace_chunk_put(session->current_trace_chunk);
c35f9726 266 session->current_trace_chunk = NULL;
62bad3bf
JG
267 lttng_trace_chunk_put(session->pending_closure_trace_chunk);
268 session->pending_closure_trace_chunk = NULL;
23c8ff50
JG
269 ret = sessiond_trace_chunk_registry_session_destroyed(
270 sessiond_trace_chunk_registry, session->sessiond_uuid);
271 assert(!ret);
7591bab1
MD
272 call_rcu(&session->rcu_node, rcu_destroy_session);
273}
274
275void session_release(struct urcu_ref *ref)
2a174661 276{
7591bab1
MD
277 struct relay_session *session =
278 caa_container_of(ref, struct relay_session, ref);
2a174661 279
7591bab1
MD
280 destroy_session(session);
281}
2a174661 282
7591bab1
MD
283void session_put(struct relay_session *session)
284{
285 rcu_read_lock();
7591bab1 286 urcu_ref_put(&session->ref, session_release);
7591bab1 287 rcu_read_unlock();
2a174661
DG
288}
289
7591bab1 290int session_close(struct relay_session *session)
2a174661
DG
291{
292 int ret = 0;
7591bab1
MD
293 struct ctf_trace *trace;
294 struct lttng_ht_iter iter;
295 struct relay_stream *stream;
296
297 pthread_mutex_lock(&session->lock);
298 DBG("closing session %" PRIu64 ": is conn already closed %d",
299 session->id, session->connection_closed);
7591bab1 300 session->connection_closed = true;
7591bab1 301 pthread_mutex_unlock(&session->lock);
2a174661 302
7591bab1
MD
303 rcu_read_lock();
304 cds_lfht_for_each_entry(session->ctf_traces_ht->ht,
305 &iter.iter, trace, node.node) {
306 ret = ctf_trace_close(trace);
307 if (ret) {
308 goto rcu_unlock;
2a174661
DG
309 }
310 }
7591bab1
MD
311 cds_list_for_each_entry_rcu(stream, &session->recv_list,
312 recv_node) {
bda7c7b9
JG
313 /* Close streams which have not been published yet. */
314 try_stream_close(stream);
7591bab1
MD
315 }
316rcu_unlock:
317 rcu_read_unlock();
318 if (ret) {
319 return ret;
320 }
321 /* Put self-reference from create. */
322 session_put(session);
323 return ret;
2a174661
DG
324}
325
98ba050e
JR
326int session_abort(struct relay_session *session)
327{
328 int ret = 0;
329
330 if (!session) {
331 return 0;
332 }
333
334 pthread_mutex_lock(&session->lock);
335 DBG("aborting session %" PRIu64, session->id);
98ba050e 336 session->aborted = true;
98ba050e
JR
337 pthread_mutex_unlock(&session->lock);
338 return ret;
339}
340
7591bab1 341void print_sessions(void)
2a174661 342{
2a174661 343 struct lttng_ht_iter iter;
7591bab1 344 struct relay_session *session;
2a174661 345
ce3f3ba3
JG
346 if (!sessions_ht) {
347 return;
348 }
349
2a174661 350 rcu_read_lock();
7591bab1
MD
351 cds_lfht_for_each_entry(sessions_ht->ht, &iter.iter, session,
352 session_n.node) {
353 if (!session_get(session)) {
354 continue;
355 }
356 DBG("session %p refcount %ld session %" PRIu64,
357 session,
358 session->ref.refcount,
359 session->id);
360 session_put(session);
2a174661 361 }
2a174661 362 rcu_read_unlock();
2a174661 363}
This page took 0.053318 seconds and 4 git commands to generate.