Fix: make sure ctrl & data url are set for post creation configuration
[lttng-tools.git] / src / bin / lttng / commands / create.c
CommitLineData
f3ed775e
DG
1/*
2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
3 *
d14d33bf
AM
4 * This program is free software; you can redistribute it and/or modify
5 * it under the terms of the GNU General Public License, version 2 only,
6 * as published by the Free Software Foundation.
f3ed775e
DG
7 *
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
12 *
d14d33bf
AM
13 * You should have received a copy of the GNU General Public License along
14 * with this program; if not, write to the Free Software Foundation, Inc.,
15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
f3ed775e
DG
16 */
17
6c1c0768 18#define _LGPL_SOURCE
a4b92340 19#include <assert.h>
ecc48a90 20#include <ctype.h>
f3ed775e
DG
21#include <popt.h>
22#include <stdio.h>
23#include <stdlib.h>
24#include <string.h>
25#include <sys/stat.h>
26#include <sys/types.h>
27#include <time.h>
28#include <unistd.h>
92360082 29#include <signal.h>
bbd44cae 30#include <sys/wait.h>
f3ed775e 31
37d03ff7
JRJ
32#include <common/mi-lttng.h>
33
c399183f 34#include "../command.h"
679b4943 35#include "../utils.h"
f3ed775e 36
00e2e675 37#include <common/defaults.h>
42224349 38#include <common/sessiond-comm/sessiond-comm.h>
00e2e675 39#include <common/uri.h>
81b86775 40#include <common/utils.h>
16f6f820 41#include <lttng/snapshot.h>
42224349 42
f3ed775e
DG
43static char *opt_output_path;
44static char *opt_session_name;
a4b92340
DG
45static char *opt_url;
46static char *opt_ctrl_url;
47static char *opt_data_url;
d7ba1388 48static char *opt_shm_path;
a4b92340 49static int opt_no_consumer;
96fe6b8d 50static int opt_no_output;
16f6f820 51static int opt_snapshot;
ecc48a90 52static unsigned int opt_live_timer;
f3ed775e
DG
53
54enum {
55 OPT_HELP = 1,
679b4943 56 OPT_LIST_OPTIONS,
ecc48a90 57 OPT_LIVE_TIMER,
f3ed775e
DG
58};
59
227bc702 60enum {
101fa438 61 OUTPUT_UNKNOWN = -1,
227bc702
JR
62 OUTPUT_NONE,
63 OUTPUT_LOCAL,
64 OUTPUT_NET,
65};
66enum {
101fa438 67 SESSION_UNKNOWN = -1,
227bc702
JR
68 SESSION_NORMAL,
69 SESSION_LIVE,
70 SESSION_SNAPSHOT,
71};
72
73
37d03ff7
JRJ
74static struct mi_writer *writer;
75
f3ed775e
DG
76static struct poptOption long_options[] = {
77 /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */
679b4943
SM
78 {"help", 'h', POPT_ARG_NONE, NULL, OPT_HELP, NULL, NULL},
79 {"output", 'o', POPT_ARG_STRING, &opt_output_path, 0, NULL, NULL},
80 {"list-options", 0, POPT_ARG_NONE, NULL, OPT_LIST_OPTIONS, NULL, NULL},
23d14dff
DG
81 {"set-url", 'U', POPT_ARG_STRING, &opt_url, 0, 0, 0},
82 {"ctrl-url", 'C', POPT_ARG_STRING, &opt_ctrl_url, 0, 0, 0},
83 {"data-url", 'D', POPT_ARG_STRING, &opt_data_url, 0, 0, 0},
96fe6b8d 84 {"no-output", 0, POPT_ARG_VAL, &opt_no_output, 1, 0, 0},
2bba9e53 85 {"no-consumer", 0, POPT_ARG_VAL, &opt_no_consumer, 1, 0, 0},
16f6f820 86 {"snapshot", 0, POPT_ARG_VAL, &opt_snapshot, 1, 0, 0},
d73c5802 87 {"live", 0, POPT_ARG_INT | POPT_ARGFLAG_OPTIONAL, 0, OPT_LIVE_TIMER, 0, 0},
d7ba1388 88 {"shm-path", 0, POPT_ARG_STRING, &opt_shm_path, 0, 0, 0},
f3ed775e
DG
89 {0, 0, 0, 0, 0, 0, 0}
90};
91
16de1a24
DG
92/*
93 * Please have a look at src/lib/lttng-ctl/lttng-ctl.c for more information on
94 * why this declaration exists and used ONLY in for this command.
95 */
07424f16 96extern int _lttng_create_session_ext(const char *name, const char *url,
d1edd8a8 97 const char *datetime);
07424f16 98
37d03ff7 99/*
485ca16f 100 * Retrieve the created session and mi output it based on provided argument
37d03ff7
JRJ
101 * This is currently a summary of what was pretty printed and is subject to
102 * enhancements.
37d03ff7
JRJ
103 */
104static int mi_created_session(const char *session_name)
105{
106 int ret, i, count, found;
107 struct lttng_session *sessions;
108
109 /* session_name should not be null */
110 assert(session_name);
111 assert(writer);
112
113 count = lttng_list_sessions(&sessions);
114 if (count < 0) {
115 ret = count;
116 ERR("%s", lttng_strerror(ret));
117 goto error;
118 }
119
120 if (count == 0) {
121 ERR("Error session creation failed: session %s not found", session_name);
122 ret = -LTTNG_ERR_SESS_NOT_FOUND;
123 goto end;
124 }
125
126 found = 0;
127 for (i = 0; i < count; i++) {
128 if (strncmp(sessions[i].name, session_name, NAME_MAX) == 0) {
129 found = 1;
130 ret = mi_lttng_session(writer, &sessions[i], 0);
131 if (ret) {
132 goto error;
133 }
134 break;
135 }
136 }
137
138 if (!found) {
139 ret = -LTTNG_ERR_SESS_NOT_FOUND;
140 } else {
141 ret = CMD_SUCCESS;
142 }
143
144error:
145 free(sessions);
146end:
147 return ret;
148}
149
00e2e675 150/*
a4b92340 151 * For a session name, set the consumer URLs.
00e2e675 152 */
a4b92340
DG
153static int set_consumer_url(const char *session_name, const char *ctrl_url,
154 const char *data_url)
00e2e675 155{
a4b92340
DG
156 int ret;
157 struct lttng_handle *handle;
a4b92340
DG
158
159 assert(session_name);
160
161 /*
95681498
JG
162 * Set handle with the session_name, but no domain. This implies that
163 * the actions taken with this handle apply on the tracing session
164 * rather then the domain-specific session.
a4b92340 165 */
95681498 166 handle = lttng_create_handle(session_name, NULL);
a4b92340
DG
167 if (handle == NULL) {
168 ret = CMD_FATAL;
169 goto error;
170 }
00e2e675 171
a4b92340
DG
172 ret = lttng_set_consumer_url(handle, ctrl_url, data_url);
173 if (ret < 0) {
174 goto error;
00e2e675
DG
175 }
176
a4b92340
DG
177error:
178 lttng_destroy_handle(handle);
179 return ret;
180}
181
16f6f820
DG
182static int add_snapshot_output(const char *session_name, const char *ctrl_url,
183 const char *data_url)
184{
185 int ret;
186 struct lttng_snapshot_output *output = NULL;
187
188 assert(session_name);
189
190 output = lttng_snapshot_output_create();
191 if (!output) {
192 ret = CMD_FATAL;
193 goto error_create;
194 }
195
196 if (ctrl_url) {
197 ret = lttng_snapshot_output_set_ctrl_url(ctrl_url, output);
198 if (ret < 0) {
199 goto error;
200 }
201 }
202
203 if (data_url) {
204 ret = lttng_snapshot_output_set_data_url(data_url, output);
205 if (ret < 0) {
206 goto error;
207 }
208 }
209
210 /* This call, if successful, populates the id of the output object. */
211 ret = lttng_snapshot_add_output(session_name, output);
212 if (ret < 0) {
213 goto error;
214 }
215
216error:
217 lttng_snapshot_output_destroy(output);
218error_create:
219 return ret;
220}
221
227bc702
JR
222/*
223 * Validate the combinations of passed options
224 *
225 * CMD_ERROR on error
226 * CMD_SUCCESS on success
227 */
228static int validate_command_options(void)
229{
230 int ret = CMD_SUCCESS;
231 if (opt_snapshot && opt_live_timer) {
232 ERR("Snapshot and live modes are mutually exclusive.");
233 ret = CMD_ERROR;
234 goto error;
235 }
236
237 if ((!opt_ctrl_url && opt_data_url) || (opt_ctrl_url && !opt_data_url)) {
238 ERR("You need both control and data URL.");
239 ret = CMD_ERROR;
240 goto error;
241 }
242
243error:
244 return ret;
245}
246
247/*
248 * Create a session via direct calls to liblttng-ctl.
249 *
250 * Return CMD_SUCCESS on success, negative value on internal lttng errors and positive
251 * value on command errors.
252 */
253static int create_session_basic (const char *session_name,
254 int session_type,
255 int live_timer,
256 int output_type,
257 const char* url,
258 const char* ctrl_url,
259 const char* data_url,
260 const char* shm_path,
261 const char* datetime)
262{
263 /* Create session based on session creation */
264 int ret = CMD_SUCCESS;
265 const char *pathname;
266
267 assert(datetime);
268
269 if (opt_relayd_path) {
270 pathname = opt_relayd_path;
271 } else {
272 pathname = INSTALL_BIN_PATH "/lttng-relayd";
273 }
274
275 switch (session_type) {
276 case SESSION_NORMAL:
277 ret = _lttng_create_session_ext(session_name, url, datetime);
278 break;
279 case SESSION_SNAPSHOT:
280 if (output_type == OUTPUT_NONE) {
281 ERR("--no-output on a snapshot session is invalid");
282 ret = CMD_UNSUPPORTED;
283 goto error;
284 }
285 ret = lttng_create_session_snapshot(session_name, url);
286 break;
287 case SESSION_LIVE:
288 if (output_type == OUTPUT_NONE) {
289 ERR("--no-output on a live session is invalid");
290 ret = CMD_UNSUPPORTED;
291 goto error;
292 }
293
294 if (output_type == OUTPUT_LOCAL) {
295 ERR("Local file output on a live session is invalid");
296 ret = CMD_UNSUPPORTED;
297 goto error;
298 }
299 if (output_type != OUTPUT_NET && !check_relayd() &&
300 spawn_relayd(pathname, 0) < 0) {
301 ret = CMD_FATAL;
302 goto error;
303 }
304 ret = lttng_create_session_live(session_name, url, live_timer);
305 break;
306 default:
307 ERR("Unknown session type");
308 ret = CMD_UNDEFINED;
309 goto error;
310 }
311
312 if (ret < 0) {
8ad05258 313 /* Don't set ret so the sessiond error is propagated. */
227bc702
JR
314 switch (-ret) {
315 case LTTNG_ERR_EXIST_SESS:
316 WARN("Session %s already exists", session_name);
317 break;
318 default:
319 break;
320 }
321 goto error;
322 }
323
324 /* Configure the session based on the output type */
325 switch (output_type) {
326 case OUTPUT_LOCAL:
327 break;
328 case OUTPUT_NET:
19ed6174
JR
329 if (!ctrl_url || !data_url) {
330 break;
331 }
332
227bc702
JR
333 if (session_type == SESSION_SNAPSHOT) {
334 ret = add_snapshot_output(session_name, ctrl_url,
335 data_url);
19ed6174 336 } else {
227bc702
JR
337 /*
338 * Normal sessions and live sessions behave the same way
339 * regarding consumer url.
340 */
341 ret = set_consumer_url(session_name, ctrl_url, data_url);
342 }
343 if (ret < 0) {
344 /* Destroy created session on errors */
345 lttng_destroy_session(session_name);
346 goto error;
347 }
348 break;
349 case OUTPUT_NONE:
350 break;
351 default:
352 ERR("Unknown output type");
353 ret = CMD_UNDEFINED;
354 goto error;
355 }
356
357 /*
358 * Set the session shared memory path
359 */
360 if (shm_path) {
361 ret = lttng_set_session_shm_path(session_name, shm_path);
362 if (ret < 0) {
363 lttng_destroy_session(session_name);
364 goto error;
365 }
366 }
367error:
368 return ret;
369}
370
371static int generate_output(const char *session_name,
372 int session_type,
373 int live_timer,
374 int output_type,
375 const char* url,
376 const char* ctrl_url,
377 const char* data_url,
378 const char* shm_path)
379{
380 int ret = CMD_SUCCESS;
381
382 /*
383 * TODO move this to after session name
384 * for now we only emulate previous behaviour.
385 */
386 if (session_type != SESSION_SNAPSHOT) {
387 if (ctrl_url) {
388 MSG("Control URL %s set for session %s", ctrl_url, session_name);
389 }
390
391 if (data_url) {
392 MSG("Data URL %s set for session %s", data_url, session_name);
393 }
394 }
395
396 if (url && output_type == OUTPUT_LOCAL) {
397 /* Remove the file:// */
398 if (strlen(url) > strlen("file://")){
399 url = url + strlen("file://");
400 }
401 }
402
403 MSG("Session %s created.", session_name);
404 if (url && session_type != SESSION_SNAPSHOT) {
405 MSG("Traces will be written in %s", url);
406
407 if (live_timer) {
408 MSG("Live timer set to %u usec", live_timer);
409 }
410 } else if (session_type == SESSION_SNAPSHOT) {
411 if (url) {
412 MSG("Default snapshot output set to: %s", url);
413 }
414 MSG("Snapshot mode set. Every channel enabled for that session will "
415 "be set to mmap output, and default to overwrite mode.");
416 }
417
418 if (shm_path) {
419 MSG("Session %s set to shm_path: %s.", session_name,
420 shm_path);
421 }
422
423 /* Mi output */
424 if (lttng_opt_mi) {
425 ret = mi_created_session(session_name);
426 if (ret) {
427 ret = CMD_ERROR;
428 goto error;
429 }
430 }
431error:
432 return ret;
433}
434
f3ed775e 435/*
1c8d13c8
TD
436 * Create a tracing session.
437 * If no name is specified, a default name is generated.
f3ed775e 438 *
1c8d13c8 439 * Returns one of the CMD_* result constants.
f3ed775e 440 */
a4b92340 441static int create_session(void)
f3ed775e 442{
a4b92340 443 int ret;
227bc702
JR
444
445 /* Base data */
446 int base_session_type = SESSION_UNKNOWN;
447 int base_output_type = OUTPUT_UNKNOWN;
448 char *base_session_name = NULL;
449 char *base_url = NULL;
450 char *base_ctrl_url = NULL;
451 char *base_data_url = NULL;
452 char *base_shm_path = NULL;
453 int base_live_timer = 0;
454
455 /* Time data */
456 char datetime[16];
f3ed775e 457 time_t rawtime;
227bc702
JR
458 struct tm *timeinfo = NULL;
459
460 /* Temporary variables */
461 char *traces_path = NULL;
462 char *temp_url = NULL;
463 char *session_name_date = NULL;
464 char *tmp_url = NULL;
465 char *tmp_home_path = NULL;
466 struct lttng_uri *uris = NULL;
467 ssize_t uri_array_size = 0;
468
469 /* Option validation */
470 if (validate_command_options() != CMD_SUCCESS) {
471 ret = CMD_ERROR;
472 goto error;
473 }
f3ed775e 474
d6175221
DG
475 /* Get date and time for automatic session name/path */
476 time(&rawtime);
477 timeinfo = localtime(&rawtime);
478 strftime(datetime, sizeof(datetime), "%Y%m%d-%H%M%S", timeinfo);
479
227bc702
JR
480 /* Find the session type based on options */
481 if(base_session_type == SESSION_UNKNOWN) {
482 if (opt_snapshot) {
483 base_session_type = SESSION_SNAPSHOT;
484 } else if (opt_live_timer) {
485 base_session_type = SESSION_LIVE;
486 } else {
487 base_session_type = SESSION_NORMAL;
07424f16 488 }
227bc702
JR
489 }
490
491 /*
492 * Session name handling
493 */
494 if (opt_session_name) {
495 /* Override the session name */
487b253b
DG
496 if (strlen(opt_session_name) > NAME_MAX) {
497 ERR("Session name too long. Length must be lower or equal to %d",
498 NAME_MAX);
499 ret = LTTNG_ERR_SESSION_FAIL;
227bc702 500 free(session_name_date);
487b253b
DG
501 goto error;
502 }
4b861950
DG
503 /*
504 * Check if the session name begins with "auto-" or is exactly "auto".
505 * Both are reserved for the default session name. See bug #449 to
506 * understand why we need to check both here.
507 */
508 if ((strncmp(opt_session_name, DEFAULT_SESSION_NAME "-",
227bc702
JR
509 strlen(DEFAULT_SESSION_NAME) + 1) == 0) ||
510 (strncmp(opt_session_name, DEFAULT_SESSION_NAME,
511 strlen(DEFAULT_SESSION_NAME)) == 0 &&
512 strlen(opt_session_name) == strlen(DEFAULT_SESSION_NAME))) {
61b35a5a
DG
513 ERR("%s is a reserved keyword for default session(s)",
514 DEFAULT_SESSION_NAME);
227bc702
JR
515
516 ret = CMD_ERROR;
517 goto error;
518 }
519
520 base_session_name = strndup(opt_session_name, NAME_MAX);
521 if (!base_session_name) {
522 PERROR("Strdup session name");
61b35a5a
DG
523 ret = CMD_ERROR;
524 goto error;
525 }
227bc702
JR
526
527 ret = asprintf(&session_name_date, "%s-%s", base_session_name, datetime);
528 if (ret < 0) {
529 PERROR("Asprintf session name");
530 goto error;
531 }
532 DBG("Session name from command option set to %s", base_session_name);
533 } else if (base_session_name) {
534 ret = asprintf(&session_name_date, "%s-%s", base_session_name, datetime);
535 if (ret < 0) {
536 PERROR("Asprintf session name");
537 goto error;
538 }
539 } else {
540 /* Generate a name */
541 /* TODO: use asprint */
542 ret = asprintf(&base_session_name, DEFAULT_SESSION_NAME "-%s", datetime);
07424f16 543 if (ret < 0) {
227bc702 544 PERROR("Asprintf generated session name");
07424f16
DG
545 goto error;
546 }
227bc702
JR
547 session_name_date = strdup(base_session_name);
548 DBG("Auto session name set to %s", base_session_name);
f3ed775e
DG
549 }
550
227bc702
JR
551
552 /*
553 * Output handling
554 */
555
556 /*
557 * If any of those options are present clear all output related data.
558 */
559 if (opt_output_path || opt_url || (opt_ctrl_url && opt_data_url) || opt_no_output) {
560 /* Overwrite output */
561 free(base_url);
562 free(base_ctrl_url);
563 free(base_data_url);
564 base_url = NULL;
565 base_ctrl_url = NULL;
566 base_data_url = NULL;
1a241656
DG
567 }
568
227bc702
JR
569 if (opt_output_path) {
570
81b86775 571 traces_path = utils_expand_path(opt_output_path);
227bc702 572 if (!traces_path) {
00e2e675
DG
573 ret = CMD_ERROR;
574 goto error;
575 }
576
37d03ff7 577 /* Create URL string from the local file system path */
227bc702 578 ret = asprintf(&temp_url, "file://%s", traces_path);
00e2e675 579 if (ret < 0) {
a4b92340 580 PERROR("asprintf url path");
00e2e675
DG
581 ret = CMD_FATAL;
582 goto error;
583 }
227bc702
JR
584
585 base_url = temp_url;
a4b92340 586 } else if (opt_url) { /* Handling URL (-U opt) */
227bc702 587 base_url = strdup(opt_url);
1a241656
DG
588 } else if (opt_data_url && opt_ctrl_url) {
589 /*
227bc702
JR
590 * With both control and data, we'll be setting the consumer URL
591 * after session creation thus use no URL.
1a241656 592 */
227bc702
JR
593 base_ctrl_url = strdup(opt_ctrl_url);
594 base_data_url = strdup(opt_data_url);
595 } else if (!(opt_no_output || base_output_type == OUTPUT_NONE ||
596 base_url || base_ctrl_url || base_data_url)) {
597 /* Generate default output depending on the session type */
598 switch (base_session_type) {
599 case SESSION_NORMAL:
600 /* fallthrough */
601 case SESSION_SNAPSHOT:
602 /* Default to a local path */
603 tmp_home_path = utils_get_home_dir();
604 if (tmp_home_path == NULL) {
605 ERR("HOME path not found.\n \
606 Please specify an output path using -o, --output PATH");
607 ret = CMD_FATAL;
608 goto error;
609 }
ecc48a90 610
227bc702
JR
611 ret = asprintf(&tmp_url,
612 "file://%s/" DEFAULT_TRACE_DIR_NAME "/%s",
613 tmp_home_path, session_name_date);
16f6f820 614
227bc702
JR
615 if (ret < 0) {
616 PERROR("asprintf trace dir name");
617 ret = CMD_FATAL;
618 goto error;
619 }
8960e9cd 620
227bc702 621 base_url = tmp_url ;
42224349 622 break;
227bc702
JR
623 case SESSION_LIVE:
624 /* Default to a net output */
625 ret = asprintf(&tmp_url, "net://127.0.0.1");
626 if (ret < 0) {
627 PERROR("asprintf default live URL");
628 ret = CMD_FATAL;
629 goto error;
630 }
631 base_url = tmp_url ;
60e835ca 632 break;
227bc702
JR
633 default:
634 ERR("Unknown session type");
635 ret = CMD_FATAL;
a4b92340
DG
636 goto error;
637 }
4f50c803
DG
638 }
639
227bc702
JR
640 /*
641 * Shared memory path handling
642 */
d7ba1388 643 if (opt_shm_path) {
227bc702 644 ret = asprintf(&base_shm_path, "%s/%s", opt_shm_path, session_name_date);
d7ba1388 645 if (ret < 0) {
227bc702 646 PERROR("asprintf shm_path");
d7ba1388
MD
647 goto error;
648 }
649 }
650
227bc702
JR
651 /*
652 * Live timer handling
653 */
654 if (opt_live_timer) {
655 base_live_timer = opt_live_timer;
656 }
d73c5802 657
227bc702
JR
658 /* Get output type from urls */
659 if (base_url) {
660 /* Get lttng uris from single url */
661 uri_array_size = uri_parse_str_urls(base_url, NULL, &uris);
662 if (uri_array_size < 0) {
663 ret = CMD_ERROR;
664 goto error;
d73c5802 665 }
227bc702
JR
666 } else if (base_ctrl_url && base_data_url) {
667 uri_array_size = uri_parse_str_urls(base_ctrl_url, base_data_url, &uris);
668 if (uri_array_size < 0) {
669 ret = CMD_ERROR;
670 goto error;
16f6f820 671 }
227bc702
JR
672 } else {
673 /* --no-output */
674 uri_array_size = 0;
675 }
676
677 switch (uri_array_size) {
678 case 0:
679 base_output_type = OUTPUT_NONE;
680 break;
681 case 1:
682 base_output_type = OUTPUT_LOCAL;
683 break;
684 case 2:
685 base_output_type = OUTPUT_NET;
686 break;
687 default:
688 ret = CMD_ERROR;
689 goto error;
a4b92340 690 }
227bc702
JR
691
692 ret = create_session_basic (base_session_name,
693 base_session_type,
694 base_live_timer,
695 base_output_type,
696 base_url,
697 base_ctrl_url,
698 base_data_url,
699 base_shm_path,
700 datetime);
701 if (ret) {
702 goto error;
d7ba1388 703 }
a4b92340 704
227bc702
JR
705 ret = generate_output (base_session_name,
706 base_session_type,
707 base_live_timer,
708 base_output_type,
709 base_url,
710 base_ctrl_url,
711 base_data_url,
712 base_shm_path);
713 if (ret) {
714 goto error;
37d03ff7
JRJ
715 }
716
58a97671 717 /* Init lttng session config */
227bc702 718 ret = config_init(base_session_name);
f3ed775e 719 if (ret < 0) {
27089920 720 ret = CMD_ERROR;
f3ed775e
DG
721 goto error;
722 }
723
f3ed775e
DG
724 ret = CMD_SUCCESS;
725
726error:
227bc702
JR
727
728 /* Session temp stuff */
729 free(session_name_date);
730
731 free(uris);
a4b92340
DG
732
733 if (ret < 0) {
734 ERR("%s", lttng_strerror(ret));
735 }
227bc702
JR
736 free(base_session_name);
737 free(base_url);
738 free(base_ctrl_url);
739 free(base_data_url);
740 free(base_shm_path);
f3ed775e
DG
741 return ret;
742}
743
92360082
JG
744/*
745 * spawn_sessiond
746 *
747 * Spawn a session daemon by forking and execv.
748 */
749static int spawn_sessiond(char *pathname)
750{
751 int ret = 0;
752 pid_t pid;
753
754 MSG("Spawning a session daemon");
92360082
JG
755 pid = fork();
756 if (pid == 0) {
757 /*
bbd44cae 758 * Spawn session daemon in daemon mode.
92360082 759 */
bbd44cae
PP
760 execlp(pathname, "lttng-sessiond",
761 "--daemonize", NULL);
92360082
JG
762 /* execlp only returns if error happened */
763 if (errno == ENOENT) {
764 ERR("No session daemon found. Use --sessiond-path.");
765 } else {
766 PERROR("execlp");
767 }
768 kill(getppid(), SIGTERM); /* wake parent */
769 exit(EXIT_FAILURE);
770 } else if (pid > 0) {
92360082 771 /*
bbd44cae
PP
772 * In daemon mode (--daemonize), sessiond only exits when
773 * it's ready to accept commands.
92360082 774 */
bbd44cae 775 for (;;) {
81527d36
JG
776 int status;
777 pid_t wait_pid_ret = waitpid(pid, &status, 0);
778
779 if (wait_pid_ret < 0) {
780 if (errno == EINTR) {
781 continue;
782 }
783 PERROR("waitpid");
784 ret = -errno;
785 goto end;
786 }
bbd44cae
PP
787
788 if (WIFSIGNALED(status)) {
789 ERR("Session daemon was killed by signal %d",
790 WTERMSIG(status));
791 ret = -1;
792 goto end;
793 } else if (WIFEXITED(status)) {
794 DBG("Session daemon terminated normally (exit status: %d)",
795 WEXITSTATUS(status));
796
797 if (WEXITSTATUS(status) != 0) {
798 ERR("Session daemon terminated with an error (exit status: %d)",
799 WEXITSTATUS(status));
800 ret = -1;
801 goto end;
802 }
803 break;
804 }
92360082 805 }
bbd44cae 806
92360082
JG
807 goto end;
808 } else {
809 PERROR("fork");
810 ret = -1;
811 goto end;
812 }
813
814end:
815 return ret;
816}
817
818/*
819 * launch_sessiond
820 *
821 * Check if the session daemon is available using
822 * the liblttngctl API for the check. If not, try to
823 * spawn a daemon.
824 */
825static int launch_sessiond(void)
826{
827 int ret;
828 char *pathname = NULL;
829
830 ret = lttng_session_daemon_alive();
831 if (ret) {
832 /* Sessiond is alive, not an error */
833 ret = 0;
834 goto end;
835 }
836
837 /* Try command line option path */
838 pathname = opt_sessiond_path;
839
840 /* Try LTTNG_SESSIOND_PATH env variable */
841 if (pathname == NULL) {
842 pathname = getenv(DEFAULT_SESSIOND_PATH_ENV);
843 }
844
845 /* Try with configured path */
846 if (pathname == NULL) {
847 if (CONFIG_SESSIOND_BIN[0] != '\0') {
848 pathname = CONFIG_SESSIOND_BIN;
849 }
850 }
851
852 /* Try the default path */
853 if (pathname == NULL) {
854 pathname = INSTALL_BIN_PATH "/lttng-sessiond";
855 }
856
857 DBG("Session daemon binary path: %s", pathname);
858
859 /* Check existence and permissions */
860 ret = access(pathname, F_OK | X_OK);
861 if (ret < 0) {
862 ERR("No such file or access denied: %s", pathname);
863 goto end;
864 }
865
866 ret = spawn_sessiond(pathname);
92360082 867end:
0f4fa0d2
JG
868 if (ret) {
869 ERR("Problem occurred while launching session daemon (%s)",
870 pathname);
871 }
92360082
JG
872 return ret;
873}
874
f3ed775e 875/*
74cc1d0f 876 * The 'create <options>' first level command
1c8d13c8
TD
877 *
878 * Returns one of the CMD_* result constants.
f3ed775e
DG
879 */
880int cmd_create(int argc, const char **argv)
881{
37d03ff7 882 int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS, success = 1;
ecc48a90 883 char *opt_arg = NULL;
f3ed775e
DG
884 static poptContext pc;
885
886 pc = poptGetContext(NULL, argc, argv, long_options, 0);
887 poptReadDefaultConfig(pc, 0);
888
889 while ((opt = poptGetNextOpt(pc)) != -1) {
890 switch (opt) {
891 case OPT_HELP:
4ba92f18 892 SHOW_HELP();
f3ed775e 893 goto end;
679b4943
SM
894 case OPT_LIST_OPTIONS:
895 list_cmd_options(stdout, long_options);
679b4943 896 goto end;
ecc48a90
JD
897 case OPT_LIVE_TIMER:
898 {
899 unsigned long v;
900
901 errno = 0;
902 opt_arg = poptGetOptArg(pc);
d73c5802
DG
903 if (!opt_arg) {
904 /* Set up default values. */
905 opt_live_timer = (uint32_t) DEFAULT_LTTNG_LIVE_TIMER;
906 DBG("Session live timer interval set to default value %d",
907 opt_live_timer);
908 break;
909 }
910
ecc48a90
JD
911 v = strtoul(opt_arg, NULL, 0);
912 if (errno != 0 || !isdigit(opt_arg[0])) {
913 ERR("Wrong value in --live parameter: %s", opt_arg);
914 ret = CMD_ERROR;
915 goto end;
916 }
917 if (v != (uint32_t) v) {
918 ERR("32-bit overflow in --live parameter: %s", opt_arg);
919 ret = CMD_ERROR;
920 goto end;
921 }
0ed9e0be
JG
922 if (v == 0) {
923 ERR("Live timer interval must be greater than zero");
924 ret = CMD_ERROR;
925 goto end;
926 }
ecc48a90
JD
927 opt_live_timer = (uint32_t) v;
928 DBG("Session live timer interval set to %d", opt_live_timer);
929 break;
930 }
f3ed775e 931 default:
f3ed775e
DG
932 ret = CMD_UNDEFINED;
933 goto end;
934 }
935 }
936
785d2d0d 937 if (opt_no_consumer) {
96fe6b8d 938 MSG("The option --no-consumer is obsolete. Use --no-output now.");
785d2d0d
DG
939 ret = CMD_WARNING;
940 goto end;
941 }
942
92360082
JG
943 /* Spawn a session daemon if needed */
944 if (!opt_no_sessiond) {
945 ret = launch_sessiond();
946 if (ret) {
947 ret = CMD_ERROR;
948 goto end;
949 }
950 }
951
acc09215 952 /* MI initialization */
37d03ff7
JRJ
953 if (lttng_opt_mi) {
954 writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi);
955 if (!writer) {
956 ret = -LTTNG_ERR_NOMEM;
957 goto end;
958 }
959
960 /* Open command element */
961 ret = mi_lttng_writer_command_open(writer,
962 mi_lttng_element_command_create);
963 if (ret) {
964 ret = CMD_ERROR;
965 goto end;
966 }
967
968 /* Open output element */
969 ret = mi_lttng_writer_open_element(writer,
970 mi_lttng_element_command_output);
971 if (ret) {
972 ret = CMD_ERROR;
973 goto end;
974 }
975 }
f3ed775e
DG
976 opt_session_name = (char*) poptGetArg(pc);
977
37d03ff7 978 command_ret = create_session();
227bc702 979
37d03ff7
JRJ
980 if (command_ret) {
981 success = 0;
982 }
983
984 if (lttng_opt_mi) {
985 /* Close output element */
986 ret = mi_lttng_writer_close_element(writer);
987 if (ret) {
988 ret = CMD_ERROR;
989 goto end;
990 }
991
992 /* Success ? */
993 ret = mi_lttng_writer_write_element_bool(writer,
994 mi_lttng_element_command_success, success);
995 if (ret) {
996 ret = CMD_ERROR;
997 goto end;
998 }
999
1000 /* Command element close */
1001 ret = mi_lttng_writer_command_close(writer);
1002 if (ret) {
1003 ret = CMD_ERROR;
1004 goto end;
1005 }
1006 }
f3ed775e
DG
1007
1008end:
37d03ff7
JRJ
1009 /* Mi clean-up */
1010 if (writer && mi_lttng_writer_destroy(writer)) {
1011 /* Preserve original error code */
1012 ret = ret ? ret : -LTTNG_ERR_MI_IO_FAIL;
1013 }
1014
acc09215 1015 /* Overwrite ret if an error occurred in create_session() */
37d03ff7
JRJ
1016 ret = command_ret ? command_ret : ret;
1017
ca1c3607 1018 poptFreeContext(pc);
f3ed775e
DG
1019 return ret;
1020}
This page took 0.108063 seconds and 5 git commands to generate.