2 * Copyright (C) 2011 EfficiOS Inc.
4 * SPDX-License-Identifier: GPL-2.0-only
14 #include <sys/types.h>
17 #include <common/mi-lttng.hpp>
19 #include "../command.hpp"
21 #ifdef LTTNG_EMBED_HELP
22 static const char help_msg
[] =
23 #include <lttng-set-session.1.h>
32 static struct mi_writer
*writer
;
34 static struct poptOption long_options
[] = {
35 /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */
36 {"help", 'h', POPT_ARG_NONE
, 0, OPT_HELP
, 0, 0},
37 {"list-options", 0, POPT_ARG_NONE
, NULL
, OPT_LIST_OPTIONS
, NULL
, NULL
},
42 * Print the necessary mi for a session and name.
44 static int mi_print(const char *session_name
)
49 LTTNG_ASSERT(session_name
);
52 * Open a sessions element
53 * This is purely for validation purpose
55 ret
= mi_lttng_sessions_open(writer
);
60 /* Open a session element */
61 ret
= mi_lttng_writer_open_element(writer
, config_element_session
);
67 ret
= mi_lttng_writer_write_element_string(writer
, config_element_name
,
73 /* Close session and sessions element */
74 ret
= mi_lttng_close_multi_element(writer
, 2);
85 static int set_session(const char *session_name
)
87 int ret
= CMD_SUCCESS
;
89 unsigned int session_found
= 0;
90 struct lttng_session
*sessions
;
92 if (session_name
&& strlen(session_name
) > NAME_MAX
) {
93 ERR("Session name too long. Length must be lower or equal to %d",
99 count
= lttng_list_sessions(&sessions
);
102 ERR("%s", lttng_strerror(count
));
106 for (i
= 0; i
< count
; i
++) {
107 if (strncmp(sessions
[i
].name
, session_name
, NAME_MAX
) == 0) {
113 if (!session_found
) {
114 ERR("Session '%s' not found", session_name
);
119 ret
= config_init(session_name
);
121 ERR("Unable to set session name");
126 MSG("Session set to %s", session_name
);
128 ret
= mi_print(session_name
);
146 int cmd_set_session(int argc
, const char **argv
)
148 int opt
, ret
= CMD_SUCCESS
, command_ret
= CMD_SUCCESS
, success
= 1;
149 static poptContext pc
;
150 const char *arg_session_name
= NULL
;
152 pc
= poptGetContext(NULL
, argc
, argv
, long_options
, 0);
153 poptReadDefaultConfig(pc
, 0);
155 while ((opt
= poptGetNextOpt(pc
)) != -1) {
160 case OPT_LIST_OPTIONS
:
161 list_cmd_options(stdout
, long_options
);
169 arg_session_name
= poptGetArg(pc
);
170 if (arg_session_name
== NULL
) {
171 ERR("Missing session name");
178 writer
= mi_lttng_writer_create(fileno(stdout
), lttng_opt_mi
);
180 ret
= -LTTNG_ERR_NOMEM
;
184 /* Open command element */
185 ret
= mi_lttng_writer_command_open(writer
,
186 mi_lttng_element_command_set_session
);
192 /* Open output element */
193 ret
= mi_lttng_writer_open_element(writer
,
194 mi_lttng_element_command_output
);
201 command_ret
= set_session(arg_session_name
);
208 /* Close output element */
209 ret
= mi_lttng_writer_close_element(writer
);
216 ret
= mi_lttng_writer_write_element_bool(writer
,
217 mi_lttng_element_command_success
, success
);
223 /* Command element close */
224 ret
= mi_lttng_writer_command_close(writer
);
233 if (writer
&& mi_lttng_writer_destroy(writer
)) {
234 /* Preserve original error code */
235 ret
= ret
? ret
: LTTNG_ERR_MI_IO_FAIL
;
238 /* Overwrite ret if an error occurred during set_session() */
239 ret
= command_ret
? command_ret
: ret
;