2 * Copyright (C) 2011 - David Goulet <david.goulet@polymtl.ca>
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.
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.
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.
24 #include <sys/types.h>
27 #include <common/sessiond-comm/sessiond-comm.h>
28 #include <common/mi-lttng.h>
30 #include "../command.h"
33 static char *opt_session_name
;
34 static struct mi_writer
*writer
;
41 static struct poptOption long_options
[] = {
42 /* longName, shortName, argInfo, argPtr, value, descrip, argDesc */
43 {"help", 'h', POPT_ARG_NONE
, 0, OPT_HELP
, 0, 0},
44 {"list-options", 0, POPT_ARG_NONE
, NULL
, OPT_LIST_OPTIONS
, NULL
, NULL
},
48 static int mi_print_session(char *session_name
, int enabled
)
52 /* Open session element */
53 ret
= mi_lttng_writer_open_element(writer
, config_element_session
);
58 /* Print session name element */
59 ret
= mi_lttng_writer_write_element_string(writer
, config_element_name
,
65 ret
= mi_lttng_writer_write_element_bool(writer
, config_element_enabled
,
71 /* Close session element */
72 ret
= mi_lttng_writer_close_element(writer
);
81 * Start tracing for all trace of the session.
83 static int start_tracing(void)
88 if (opt_session_name
== NULL
) {
89 session_name
= get_session_name();
90 if (session_name
== NULL
) {
95 session_name
= opt_session_name
;
98 DBG("Starting tracing for session %s", session_name
);
100 ret
= lttng_start_tracing(session_name
);
103 case LTTNG_ERR_TRACE_ALREADY_STARTED
:
104 WARN("Tracing already started for session %s", session_name
);
107 ERR("%s", lttng_strerror(ret
));
115 MSG("Tracing started for session %s", session_name
);
117 ret
= mi_print_session(session_name
, 1);
125 if (opt_session_name
== NULL
) {
135 * The 'start <options>' first level command
137 int cmd_start(int argc
, const char **argv
)
139 int opt
, ret
= CMD_SUCCESS
, command_ret
= CMD_SUCCESS
, success
= 1;
140 static poptContext pc
;
142 pc
= poptGetContext(NULL
, argc
, argv
, long_options
, 0);
143 poptReadDefaultConfig(pc
, 0);
145 while ((opt
= poptGetNextOpt(pc
)) != -1) {
150 case OPT_LIST_OPTIONS
:
151 list_cmd_options(stdout
, long_options
);
159 opt_session_name
= (char*) poptGetArg(pc
);
163 writer
= mi_lttng_writer_create(fileno(stdout
), lttng_opt_mi
);
165 ret
= -LTTNG_ERR_NOMEM
;
169 /* Open command element */
170 ret
= mi_lttng_writer_command_open(writer
,
171 mi_lttng_element_command_start
);
177 /* Open output element */
178 ret
= mi_lttng_writer_open_element(writer
,
179 mi_lttng_element_command_output
);
186 * Open sessions element
187 * For validation purpose
189 ret
= mi_lttng_writer_open_element(writer
,
190 config_element_sessions
);
197 command_ret
= start_tracing();
204 /* Close sessions and output element */
205 ret
= mi_lttng_close_multi_element(writer
, 2);
212 ret
= mi_lttng_writer_write_element_bool(writer
,
213 mi_lttng_element_command_success
, success
);
219 /* Command element close */
220 ret
= mi_lttng_writer_command_close(writer
);
229 if (writer
&& mi_lttng_writer_destroy(writer
)) {
230 /* Preserve original error code */
231 ret
= ret
? ret
: -LTTNG_ERR_MI_IO_FAIL
;
234 /* Overwrite ret if an error occurred with start_tracing */
235 ret
= command_ret
? command_ret
: ret
;