/*
* cmd_disable_events
*
* Disable event to trace session
*/
int cmd_disable_events(int argc, const char **argv)
{
int opt, ret = CMD_SUCCESS;
static poptContext pc;
char *session_name = NULL;
pc = poptGetContext(NULL, argc, argv, long_options, 0);
poptReadDefaultConfig(pc, 0);
while ((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case OPT_HELP:
usage(stdout);
goto end;
case OPT_USERSPACE:
opt_userspace = 1;
break;
case OPT_LIST_OPTIONS:
list_cmd_options(stdout, long_options);
goto end;
default:
usage(stderr);
ret = CMD_UNDEFINED;
goto end;
}
}
opt_event_list = (char*) poptGetArg(pc);
if (opt_event_list == NULL && opt_disable_all == 0) {
ERR("Missing event name(s).\n");
usage(stderr);
ret = CMD_ERROR;
goto end;
}
if (!opt_session_name) {
session_name = get_session_name();
if (session_name == NULL) {
ret = CMD_ERROR;
goto end;
}
} else {
session_name = opt_session_name;
}
ret = disable_events(session_name);
end:
if (!opt_session_name && session_name) {
free(session_name);
}
poptFreeContext(pc);
return ret;
}
/*
* Add channel to trace session
*/
int cmd_enable_channels(int argc, const char **argv)
{
int opt, ret = CMD_SUCCESS;
static poptContext pc;
char *session_name = NULL;
char *opt_arg = NULL;
init_channel_config();
pc = poptGetContext(NULL, argc, argv, long_options, 0);
poptReadDefaultConfig(pc, 0);
while ((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case OPT_HELP:
usage(stdout);
goto end;
case OPT_DISCARD:
chan.attr.overwrite = 0;
DBG("Channel set to discard");
break;
case OPT_OVERWRITE:
chan.attr.overwrite = 1;
DBG("Channel set to overwrite");
break;
case OPT_SUBBUF_SIZE:
{
uint64_t rounded_size;
int order;
/* Parse the size */
opt_arg = poptGetOptArg(pc);
if (utils_parse_size_suffix(opt_arg, &chan.attr.subbuf_size) < 0 || !chan.attr.subbuf_size) {
ERR("Wrong value in --subbuf-size parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
order = get_count_order_u64(chan.attr.subbuf_size);
assert(order >= 0);
rounded_size = 1ULL << order;
if (rounded_size != chan.attr.subbuf_size) {
WARN("The subbuf size (%" PRIu64 ") is rounded to the next power of 2 (%" PRIu64 ")",
chan.attr.subbuf_size, rounded_size);
chan.attr.subbuf_size = rounded_size;
}
/* Should now be power of 2 */
assert(!((chan.attr.subbuf_size - 1) & chan.attr.subbuf_size));
DBG("Channel subbuf size set to %" PRIu64, chan.attr.subbuf_size);
break;
}
case OPT_NUM_SUBBUF:
{
uint64_t rounded_size;
int order;
errno = 0;
opt_arg = poptGetOptArg(pc);
chan.attr.num_subbuf = strtoull(opt_arg, NULL, 0);
if (errno != 0 || !chan.attr.num_subbuf || !isdigit(opt_arg[0])) {
ERR("Wrong value in --num-subbuf parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
order = get_count_order_u64(chan.attr.num_subbuf);
assert(order >= 0);
rounded_size = 1ULL << order;
if (rounded_size != chan.attr.num_subbuf) {
WARN("The number of subbuffers (%" PRIu64 ") is rounded to the next power of 2 (%" PRIu64 ")",
chan.attr.num_subbuf, rounded_size);
chan.attr.num_subbuf = rounded_size;
}
/* Should now be power of 2 */
assert(!((chan.attr.num_subbuf - 1) & chan.attr.num_subbuf));
DBG("Channel subbuf num set to %" PRIu64, chan.attr.num_subbuf);
break;
}
case OPT_SWITCH_TIMER:
{
unsigned long v;
errno = 0;
opt_arg = poptGetOptArg(pc);
v = strtoul(opt_arg, NULL, 0);
if (errno != 0 || !isdigit(opt_arg[0])) {
ERR("Wrong value in --switch-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
if (v != (uint32_t) v) {
ERR("32-bit overflow in --switch-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
chan.attr.switch_timer_interval = (uint32_t) v;
DBG("Channel switch timer interval set to %d", chan.attr.switch_timer_interval);
break;
}
case OPT_READ_TIMER:
{
unsigned long v;
errno = 0;
opt_arg = poptGetOptArg(pc);
v = strtoul(opt_arg, NULL, 0);
if (errno != 0 || !isdigit(opt_arg[0])) {
ERR("Wrong value in --read-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
if (v != (uint32_t) v) {
ERR("32-bit overflow in --read-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
chan.attr.read_timer_interval = (uint32_t) v;
DBG("Channel read timer interval set to %d", chan.attr.read_timer_interval);
break;
}
case OPT_USERSPACE:
opt_userspace = 1;
break;
case OPT_TRACEFILE_SIZE:
opt_arg = poptGetOptArg(pc);
if (utils_parse_size_suffix(opt_arg, &chan.attr.tracefile_size) < 0) {
ERR("Wrong value in --tracefile-size parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
DBG("Maximum tracefile size set to %" PRIu64,
chan.attr.tracefile_size);
break;
case OPT_TRACEFILE_COUNT:
{
unsigned long v;
errno = 0;
opt_arg = poptGetOptArg(pc);
v = strtoul(opt_arg, NULL, 0);
if (errno != 0 || !isdigit(opt_arg[0])) {
ERR("Wrong value in --tracefile-count parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
if (v != (uint32_t) v) {
ERR("32-bit overflow in --tracefile-count parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
chan.attr.tracefile_count = (uint32_t) v;
DBG("Maximum tracefile count set to %" PRIu64,
chan.attr.tracefile_count);
break;
}
case OPT_LIST_OPTIONS:
list_cmd_options(stdout, long_options);
goto end;
default:
usage(stderr);
ret = CMD_UNDEFINED;
goto end;
}
}
opt_channels = (char*) poptGetArg(pc);
if (opt_channels == NULL) {
ERR("Missing channel name.\n");
usage(stderr);
ret = CMD_ERROR;
goto end;
}
if (!opt_session_name) {
session_name = get_session_name();
if (session_name == NULL) {
ret = CMD_ERROR;
goto end;
}
} else {
session_name = opt_session_name;
}
ret = enable_channel(session_name);
end:
if (!opt_session_name && session_name) {
free(session_name);
}
poptFreeContext(pc);
return ret;
}
/********************************************************************
* FUNCTION yangcli_notification_handler
*
* matches callback template mgr_not_cbfn_t
*
* INPUTS:
* scb == session receiving RPC reply
* msg == notification msg that was parsed
* consumed == address of return consumed message flag
*
* OUTPUTS:
* *consumed == TRUE if msg has been consumed so
* it will not be freed by mgr_not_dispatch
* == FALSE if msg has been not consumed so
* it will be freed by mgr_not_dispatch
*********************************************************************/
void
yangcli_notification_handler (ses_cb_t *scb,
mgr_not_msg_t *msg,
boolean *consumed)
{
assert(scb && "scb is NULL!");
assert(msg && "msg is NULL!");
assert(consumed && "consumed is NULL!");
*consumed = FALSE;
mgr_scb_t *mgrcb = scb->mgrcb;
server_cb_t *server_cb = get_cur_server_cb();
session_cb_t *save_session_cb = server_cb->cur_session_cb;
session_cb_t *session_cb = NULL;
if (mgrcb) {
session_cb = mgrcb->session_cb;
}
if (session_cb == NULL) {
log_error("\nError: session no longer valid; dropping notification");
return;
}
const xmlChar *sesname = get_session_name(session_cb);
if (session_cb != save_session_cb) {
set_cur_session_cb(server_cb, session_cb);
}
/* check the contents of the notification */
if (msg && msg->notification) {
/* dispatch this event to any handler registered for
* this event; !!! not session-specific !!! */
uint32 retcnt = dispatch_notif_event(session_cb, msg);
/* display only unhandled notifications, and only if
* the session is configured to show them */
log_debug_t loglevel = log_get_debug_level();
log_debug_t base_loglevel = loglevel + 1;
if (loglevel == LOG_DEBUG_INFO) {
base_loglevel++;
}
if (retcnt == 0 && session_cb->echo_notifs &&
base_loglevel >= session_cb->echo_notif_loglevel) {
gl_normal_io(server_cb->cli_gl);
if (loglevel >= session_cb->echo_notif_loglevel) {
log_write("\n\nIncoming notification for session %u [%s]:",
scb->sid, sesname);
val_dump_value_max(msg->notification,
0, session_cb->defindent,
DUMP_VAL_LOG,
session_cb->display_mode,
FALSE, FALSE);
log_write_append("\n\n");
} else if (msg->eventType) {
log_write("\n\nIncoming <%s> notification for session "
"%u [%s]\n\n", msg->eventType->name,
scb->sid, sesname);
}
}
/* store msg in the session notification log */
dlq_enque(msg, &session_cb->notificationQ);
*consumed = TRUE;
}
if (session_cb != save_session_cb) {
set_cur_session_cb(server_cb, save_session_cb);
}
} /* yangcli_notification_handler */
/*
* The 'snapshot <cmd> <options>' first level command
*/
int cmd_snapshot(int argc, const char **argv)
{
int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS, success = 1;
char *session_name = NULL;
static poptContext pc;
pc = poptGetContext(NULL, argc, argv, snapshot_opts, 0);
poptReadDefaultConfig(pc, 0);
/* Mi check */
if (lttng_opt_mi) {
writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi);
if (!writer) {
ret = -LTTNG_ERR_NOMEM;
goto end;
}
/* Open command element */
ret = mi_lttng_writer_command_open(writer,
mi_lttng_element_command_snapshot);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Open output element */
ret = mi_lttng_writer_open_element(writer,
mi_lttng_element_command_output);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
while ((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case OPT_HELP:
usage(stdout);
goto end;
case OPT_LIST_OPTIONS:
list_cmd_options(stdout, snapshot_opts);
goto end;
case OPT_LIST_COMMANDS:
list_commands(actions, stdout);
goto end;
case OPT_MAX_SIZE:
{
uint64_t val;
const char *opt = poptGetOptArg(pc);
if (utils_parse_size_suffix((char *) opt, &val) < 0) {
ERR("Unable to handle max-size value %s", opt);
ret = CMD_ERROR;
goto end;
}
opt_max_size = val;
break;
}
default:
usage(stderr);
ret = CMD_UNDEFINED;
goto end;
}
}
if (!opt_session_name) {
session_name = get_session_name();
if (session_name == NULL) {
ret = CMD_ERROR;
goto end;
}
current_session_name = session_name;
} else {
current_session_name = opt_session_name;
}
command_ret = handle_command(poptGetArgs(pc));
if (command_ret) {
switch (-command_ret) {
case LTTNG_ERR_EPERM:
ERR("The session needs to be set in no output mode (--no-output)");
break;
case LTTNG_ERR_SNAPSHOT_NODATA:
WARN("%s", lttng_strerror(command_ret));
break;
default:
ERR("%s", lttng_strerror(command_ret));
break;
}
success = 0;
}
if (lttng_opt_mi) {
/* Close output element */
ret = mi_lttng_writer_close_element(writer);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Success ? */
ret = mi_lttng_writer_write_element_bool(writer,
mi_lttng_element_command_success, success);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Command element close */
ret = mi_lttng_writer_command_close(writer);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
end:
/* Mi clean-up */
if (writer && mi_lttng_writer_destroy(writer)) {
/* Preserve original error code */
ret = ret ? ret : -LTTNG_ERR_MI_IO_FAIL;
}
if (!opt_session_name) {
free(session_name);
}
/* Overwrite ret if an error occured during handle_command */
ret = command_ret ? command_ret : ret;
poptFreeContext(pc);
return ret;
}
/*
* Exec viewer if found and use session name path.
*/
static int view_trace(void)
{
int ret;
char *session_name, *trace_path = NULL;
struct lttng_session *sessions = NULL;
/*
* Safety net. If lttng is suid at some point for *any* useless reasons,
* this prevent any bad execution of binaries.
*/
if (getuid() != 0) {
if (getuid() != geteuid()) {
ERR("UID does not match effective UID.");
ret = CMD_ERROR;
goto error;
} else if (getgid() != getegid()) {
ERR("GID does not match effective GID.");
ret = CMD_ERROR;
goto error;
}
}
/* User define trace path override the session name */
if (opt_trace_path) {
session_name = NULL;
} else if(opt_session_name == NULL) {
session_name = get_session_name();
if (session_name == NULL) {
ret = CMD_ERROR;
goto error;
}
} else {
session_name = opt_session_name;
}
DBG("Viewing trace for session %s", session_name);
if (session_name) {
int i, count, found = 0;
/* Getting all sessions */
count = lttng_list_sessions(&sessions);
if (count < 0) {
ERR("Unable to list sessions. Session name %s not found.",
session_name);
MSG("Is there a session daemon running?");
ret = CMD_ERROR;
goto free_error;
}
/* Find our session listed by the session daemon */
for (i = 0; i < count; i++) {
if (strncmp(sessions[i].name, session_name, NAME_MAX) == 0) {
found = 1;
break;
}
}
if (!found) {
MSG("Session name %s not found", session_name);
ret = CMD_ERROR;
goto free_sessions;
}
session_live_mode = sessions[i].live_timer_interval;
DBG("Session live mode set to %d", session_live_mode);
if (sessions[i].enabled && !session_live_mode) {
WARN("Session %s is running. Please stop it before reading it.",
session_name);
ret = CMD_ERROR;
goto free_sessions;
}
/* If the timer interval is set we are in live mode. */
if (session_live_mode) {
trace_path = build_live_path(session_name);
if (!trace_path) {
ret = CMD_ERROR;
goto free_sessions;
}
} else {
/* Get file system session path. */
trace_path = sessions[i].path;
}
} else {
trace_path = opt_trace_path;
}
MSG("Trace directory: %s\n", trace_path);
ret = spawn_viewer(trace_path);
if (ret < 0) {
/* Don't set ret so lttng can interpret the sessiond error. */
goto free_sessions;
}
free_sessions:
if (session_live_mode) {
free(trace_path);
}
free(sessions);
free_error:
if (opt_session_name == NULL) {
free(session_name);
}
error:
return ret;
}
/*
* Add channel to trace session
*/
int cmd_enable_channels(int argc, const char **argv)
{
int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS, success = 1;
static poptContext pc;
char *session_name = NULL;
char *opt_arg = NULL;
init_channel_config();
pc = poptGetContext(NULL, argc, argv, long_options, 0);
poptReadDefaultConfig(pc, 0);
while ((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case OPT_HELP:
usage(stdout);
goto end;
case OPT_DISCARD:
chan.attr.overwrite = 0;
DBG("Channel set to discard");
break;
case OPT_OVERWRITE:
chan.attr.overwrite = 1;
DBG("Channel set to overwrite");
break;
case OPT_SUBBUF_SIZE:
{
uint64_t rounded_size;
int order;
/* Parse the size */
opt_arg = poptGetOptArg(pc);
if (utils_parse_size_suffix(opt_arg, &chan.attr.subbuf_size) < 0 || !chan.attr.subbuf_size) {
ERR("Wrong value in --subbuf-size parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
order = get_count_order_u64(chan.attr.subbuf_size);
assert(order >= 0);
rounded_size = 1ULL << order;
if (rounded_size < chan.attr.subbuf_size) {
ERR("The subbuf size (%" PRIu64 ") is rounded and overflows!",
chan.attr.subbuf_size);
ret = CMD_ERROR;
goto end;
}
if (rounded_size != chan.attr.subbuf_size) {
WARN("The subbuf size (%" PRIu64 ") is rounded to the next power of 2 (%" PRIu64 ")",
chan.attr.subbuf_size, rounded_size);
chan.attr.subbuf_size = rounded_size;
}
/* Should now be power of 2 */
assert(!((chan.attr.subbuf_size - 1) & chan.attr.subbuf_size));
DBG("Channel subbuf size set to %" PRIu64, chan.attr.subbuf_size);
break;
}
case OPT_NUM_SUBBUF:
{
uint64_t rounded_size;
int order;
errno = 0;
opt_arg = poptGetOptArg(pc);
chan.attr.num_subbuf = strtoull(opt_arg, NULL, 0);
if (errno != 0 || !chan.attr.num_subbuf || !isdigit(opt_arg[0])) {
ERR("Wrong value in --num-subbuf parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
order = get_count_order_u64(chan.attr.num_subbuf);
assert(order >= 0);
rounded_size = 1ULL << order;
if (rounded_size < chan.attr.num_subbuf) {
ERR("The number of subbuffers (%" PRIu64 ") is rounded and overflows!",
chan.attr.num_subbuf);
ret = CMD_ERROR;
goto end;
}
if (rounded_size != chan.attr.num_subbuf) {
WARN("The number of subbuffers (%" PRIu64 ") is rounded to the next power of 2 (%" PRIu64 ")",
chan.attr.num_subbuf, rounded_size);
chan.attr.num_subbuf = rounded_size;
}
/* Should now be power of 2 */
assert(!((chan.attr.num_subbuf - 1) & chan.attr.num_subbuf));
DBG("Channel subbuf num set to %" PRIu64, chan.attr.num_subbuf);
break;
}
case OPT_SWITCH_TIMER:
{
unsigned long v;
errno = 0;
opt_arg = poptGetOptArg(pc);
v = strtoul(opt_arg, NULL, 0);
if (errno != 0 || !isdigit(opt_arg[0])) {
ERR("Wrong value in --switch-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
if (v != (uint32_t) v) {
ERR("32-bit overflow in --switch-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
chan.attr.switch_timer_interval = (uint32_t) v;
DBG("Channel switch timer interval set to %d", chan.attr.switch_timer_interval);
break;
}
case OPT_READ_TIMER:
{
unsigned long v;
errno = 0;
opt_arg = poptGetOptArg(pc);
v = strtoul(opt_arg, NULL, 0);
if (errno != 0 || !isdigit(opt_arg[0])) {
ERR("Wrong value in --read-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
if (v != (uint32_t) v) {
ERR("32-bit overflow in --read-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
chan.attr.read_timer_interval = (uint32_t) v;
DBG("Channel read timer interval set to %d", chan.attr.read_timer_interval);
break;
}
case OPT_USERSPACE:
opt_userspace = 1;
break;
case OPT_TRACEFILE_SIZE:
opt_arg = poptGetOptArg(pc);
if (utils_parse_size_suffix(opt_arg, &chan.attr.tracefile_size) < 0) {
ERR("Wrong value in --tracefile-size parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
DBG("Maximum tracefile size set to %" PRIu64,
chan.attr.tracefile_size);
break;
case OPT_TRACEFILE_COUNT:
{
unsigned long v;
errno = 0;
opt_arg = poptGetOptArg(pc);
v = strtoul(opt_arg, NULL, 0);
if (errno != 0 || !isdigit(opt_arg[0])) {
ERR("Wrong value in --tracefile-count parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
if (v != (uint32_t) v) {
ERR("32-bit overflow in --tracefile-count parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
chan.attr.tracefile_count = (uint32_t) v;
DBG("Maximum tracefile count set to %" PRIu64,
chan.attr.tracefile_count);
break;
}
case OPT_LIST_OPTIONS:
list_cmd_options(stdout, long_options);
goto end;
default:
usage(stderr);
ret = CMD_UNDEFINED;
goto end;
}
}
/* Mi check */
if (lttng_opt_mi) {
writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi);
if (!writer) {
ret = -LTTNG_ERR_NOMEM;
goto end;
}
/* Open command element */
ret = mi_lttng_writer_command_open(writer,
mi_lttng_element_command_enable_channels);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Open output element */
ret = mi_lttng_writer_open_element(writer,
mi_lttng_element_command_output);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
opt_channels = (char*) poptGetArg(pc);
if (opt_channels == NULL) {
ERR("Missing channel name.\n");
usage(stderr);
ret = CMD_ERROR;
success = 0;
goto mi_closing;
}
if (!opt_session_name) {
session_name = get_session_name();
if (session_name == NULL) {
command_ret = CMD_ERROR;
success = 0;
goto mi_closing;
}
} else {
session_name = opt_session_name;
}
command_ret = enable_channel(session_name);
if (command_ret) {
success = 0;
}
mi_closing:
/* Mi closing */
if (lttng_opt_mi) {
/* Close output element */
ret = mi_lttng_writer_close_element(writer);
if (ret) {
goto end;
}
/* Success ? */
ret = mi_lttng_writer_write_element_bool(writer,
mi_lttng_element_command_success, success);
if (ret) {
goto end;
}
/* Command element close */
ret = mi_lttng_writer_command_close(writer);
if (ret) {
goto end;
}
}
end:
/* Mi clean-up */
if (writer && mi_lttng_writer_destroy(writer)) {
/* Preserve original error code */
ret = ret ? ret : LTTNG_ERR_MI_IO_FAIL;
}
if (!opt_session_name && session_name) {
free(session_name);
}
/* Overwrite ret if an error occurred when enable_channel */
ret = command_ret ? command_ret : ret;
poptFreeContext(pc);
return ret;
}
/*
* The 'destroy <options>' first level command
*/
int cmd_destroy(int argc, const char **argv)
{
int opt;
int ret = CMD_SUCCESS , i, command_ret = CMD_SUCCESS, success = 1;
static poptContext pc;
char *session_name = NULL;
const char *leftover = NULL;
struct lttng_session *sessions;
int count;
int found;
pc = poptGetContext(NULL, argc, argv, long_options, 0);
poptReadDefaultConfig(pc, 0);
while ((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case OPT_HELP:
SHOW_HELP();
break;
case OPT_LIST_OPTIONS:
list_cmd_options(stdout, long_options);
break;
default:
ret = CMD_UNDEFINED;
break;
}
goto end;
}
/* Mi preparation */
if (lttng_opt_mi) {
writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi);
if (!writer) {
ret = -LTTNG_ERR_NOMEM;
goto end;
}
/* Open command element */
ret = mi_lttng_writer_command_open(writer,
mi_lttng_element_command_destroy);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Open output element */
ret = mi_lttng_writer_open_element(writer,
mi_lttng_element_command_output);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* For validation and semantic purpose we open a sessions element */
ret = mi_lttng_sessions_open(writer);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
/* Recuperate all sessions for further operation */
count = lttng_list_sessions(&sessions);
if (count < 0) {
ERR("%s", lttng_strerror(count));
command_ret = CMD_ERROR;
success = 0;
goto mi_closing;
}
/* Ignore session name in case all sessions are to be destroyed */
if (opt_destroy_all) {
command_ret = destroy_all_sessions(sessions, count);
if (command_ret) {
success = 0;
}
} else {
opt_session_name = (char *) poptGetArg(pc);
if (!opt_session_name) {
/* No session name specified, lookup default */
session_name = get_session_name();
if (session_name == NULL) {
command_ret = CMD_ERROR;
success = 0;
goto mi_closing;
}
} else {
session_name = opt_session_name;
}
/* Find the corresponding lttng_session struct */
found = 0;
for (i = 0; i < count; i++) {
if (strncmp(sessions[i].name, session_name, NAME_MAX) == 0) {
found = 1;
command_ret = destroy_session(&sessions[i]);
if (command_ret) {
success = 0;
}
}
}
if (!found) {
ERR("Session name %s not found", session_name);
command_ret = LTTNG_ERR_SESS_NOT_FOUND;
success = 0;
goto mi_closing;
}
}
leftover = poptGetArg(pc);
if (leftover) {
ERR("Unknown argument: %s", leftover);
ret = CMD_ERROR;
success = 0;
goto mi_closing;
}
mi_closing:
/* Mi closing */
if (lttng_opt_mi) {
/* Close sessions and output element element */
ret = mi_lttng_close_multi_element(writer, 2);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Success ? */
ret = mi_lttng_writer_write_element_bool(writer,
mi_lttng_element_command_success, success);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Command element close */
ret = mi_lttng_writer_command_close(writer);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
end:
/* Mi clean-up */
if (writer && mi_lttng_writer_destroy(writer)) {
/* Preserve original error code */
ret = ret ? ret : -LTTNG_ERR_MI_IO_FAIL;
}
if (opt_session_name == NULL) {
free(session_name);
}
/* Overwrite ret if an error occurred during destroy_session/all */
ret = command_ret ? command_ret : ret;
poptFreeContext(pc);
return ret;
}
/*
* cmd_disable_events
*
* Disable event to trace session
*/
int cmd_disable_events(int argc, const char **argv)
{
int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS, success = 1;
static poptContext pc;
char *session_name = NULL;
int event_type = -1;
pc = poptGetContext(NULL, argc, argv, long_options, 0);
poptReadDefaultConfig(pc, 0);
/* Default event type */
opt_event_type = LTTNG_EVENT_ALL;
while ((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case OPT_HELP:
SHOW_HELP();
goto end;
case OPT_TYPE_SYSCALL:
opt_event_type = LTTNG_EVENT_SYSCALL;
break;
case OPT_TYPE_TRACEPOINT:
opt_event_type = LTTNG_EVENT_TRACEPOINT;
break;
case OPT_TYPE_PROBE:
opt_event_type = LTTNG_EVENT_PROBE;
break;
case OPT_TYPE_FUNCTION:
opt_event_type = LTTNG_EVENT_FUNCTION;
break;
case OPT_TYPE_ALL:
opt_event_type = LTTNG_EVENT_ALL;
break;
case OPT_LIST_OPTIONS:
list_cmd_options(stdout, long_options);
goto end;
default:
ret = CMD_UNDEFINED;
goto end;
}
/* Validate event type. Multiple event type are not supported. */
if (event_type == -1) {
event_type = opt_event_type;
} else {
if (event_type != opt_event_type) {
ERR("Multiple event type not supported.");
ret = CMD_ERROR;
goto end;
}
}
}
ret = print_missing_or_multiple_domains(
opt_kernel + opt_userspace + opt_jul + opt_log4j + opt_python);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Ust and agent only support ALL event type */
if ((opt_userspace || opt_jul || opt_log4j || opt_python)
&& opt_event_type != LTTNG_EVENT_ALL) {
ERR("Disabling userspace and agent (-j | -l | -p) event(s) based on instrumentation type is not supported.\n");
ret = CMD_ERROR;
goto end;
}
opt_event_list = (char*) poptGetArg(pc);
if (opt_event_list == NULL && opt_disable_all == 0) {
ERR("Missing event name(s).\n");
ret = CMD_ERROR;
goto end;
}
if (!opt_session_name) {
session_name = get_session_name();
if (session_name == NULL) {
ret = CMD_ERROR;
goto end;
}
} else {
session_name = opt_session_name;
}
/* Mi check */
if (lttng_opt_mi) {
writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi);
if (!writer) {
ret = -LTTNG_ERR_NOMEM;
goto end;
}
/* Open command element */
ret = mi_lttng_writer_command_open(writer,
mi_lttng_element_command_disable_event);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Open output element */
ret = mi_lttng_writer_open_element(writer,
mi_lttng_element_command_output);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
command_ret = disable_events(session_name);
if (command_ret) {
success = 0;
}
/* Mi closing */
if (lttng_opt_mi) {
/* Close output element */
ret = mi_lttng_writer_close_element(writer);
if (ret) {
ret = CMD_ERROR;
goto end;
}
ret = mi_lttng_writer_write_element_bool(writer,
mi_lttng_element_command_success, success);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Command element close */
ret = mi_lttng_writer_command_close(writer);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
end:
if (!opt_session_name && session_name) {
free(session_name);
}
/* Mi clean-up */
if (writer && mi_lttng_writer_destroy(writer)) {
/* Preserve original error code */
ret = ret ? ret : LTTNG_ERR_MI_IO_FAIL;
}
/* Overwrite ret if an error occurred in disable_events */
ret = command_ret ? command_ret : ret;
poptFreeContext(pc);
return ret;
}
/*
* Add event to trace session
*/
int cmd_enable_events(int argc, const char **argv)
{
int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS, success = 1;
static poptContext pc;
char *session_name = NULL;
const char *leftover = NULL;
int event_type = -1;
pc = poptGetContext(NULL, argc, argv, long_options, 0);
poptReadDefaultConfig(pc, 0);
/* Default event type */
opt_event_type = LTTNG_EVENT_ALL;
while ((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case OPT_HELP:
SHOW_HELP();
goto end;
case OPT_TRACEPOINT:
opt_event_type = LTTNG_EVENT_TRACEPOINT;
break;
case OPT_PROBE:
opt_event_type = LTTNG_EVENT_PROBE;
break;
case OPT_USERSPACE_PROBE:
opt_event_type = LTTNG_EVENT_USERSPACE_PROBE;
break;
case OPT_FUNCTION:
opt_event_type = LTTNG_EVENT_FUNCTION;
break;
case OPT_SYSCALL:
opt_event_type = LTTNG_EVENT_SYSCALL;
break;
case OPT_USERSPACE:
opt_userspace = 1;
break;
case OPT_LOGLEVEL:
opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_RANGE;
opt_loglevel = poptGetOptArg(pc);
break;
case OPT_LOGLEVEL_ONLY:
opt_loglevel_type = LTTNG_EVENT_LOGLEVEL_SINGLE;
opt_loglevel = poptGetOptArg(pc);
break;
case OPT_LIST_OPTIONS:
list_cmd_options(stdout, long_options);
goto end;
case OPT_FILTER:
break;
case OPT_EXCLUDE:
break;
default:
ret = CMD_UNDEFINED;
goto end;
}
/* Validate event type. Multiple event type are not supported. */
if (event_type == -1) {
event_type = opt_event_type;
} else {
if (event_type != opt_event_type) {
ERR("Multiple event type not supported.");
ret = CMD_ERROR;
goto end;
}
}
}
ret = print_missing_or_multiple_domains(
opt_kernel + opt_userspace + opt_jul + opt_log4j + opt_python);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Mi check */
if (lttng_opt_mi) {
writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi);
if (!writer) {
ret = -LTTNG_ERR_NOMEM;
goto end;
}
/* Open command element */
ret = mi_lttng_writer_command_open(writer,
mi_lttng_element_command_enable_event);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Open output element */
ret = mi_lttng_writer_open_element(writer,
mi_lttng_element_command_output);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
opt_event_list = (char*) poptGetArg(pc);
if (opt_event_list == NULL && opt_enable_all == 0) {
ERR("Missing event name(s).\n");
ret = CMD_ERROR;
goto end;
}
leftover = poptGetArg(pc);
if (leftover) {
ERR("Unknown argument: %s", leftover);
ret = CMD_ERROR;
goto end;
}
if (!opt_session_name) {
session_name = get_session_name();
if (session_name == NULL) {
command_ret = CMD_ERROR;
success = 0;
goto mi_closing;
}
} else {
session_name = opt_session_name;
}
command_ret = enable_events(session_name);
if (command_ret) {
success = 0;
goto mi_closing;
}
mi_closing:
/* Mi closing */
if (lttng_opt_mi) {
/* Close output element */
ret = mi_lttng_writer_close_element(writer);
if (ret) {
ret = CMD_ERROR;
goto end;
}
ret = mi_lttng_writer_write_element_bool(writer,
mi_lttng_element_command_success, success);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Command element close */
ret = mi_lttng_writer_command_close(writer);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
end:
/* Mi clean-up */
if (writer && mi_lttng_writer_destroy(writer)) {
/* Preserve original error code */
ret = ret ? ret : LTTNG_ERR_MI_IO_FAIL;
}
if (opt_session_name == NULL) {
free(session_name);
}
/* Overwrite ret if an error occurred in enable_events */
ret = command_ret ? command_ret : ret;
poptFreeContext(pc);
return ret;
}
/*
* Add/remove tracker to/from session.
*/
static
int cmd_track_untrack(enum cmd_type cmd_type, const char *cmd_str,
int argc, const char **argv)
{
int opt, ret = 0;
enum cmd_error_code command_ret = CMD_SUCCESS;
int success = 1;
static poptContext pc;
char *session_name = NULL;
struct mi_writer *writer = NULL;
if (argc < 1) {
command_ret = CMD_ERROR;
goto end;
}
pc = poptGetContext(NULL, argc, argv, long_options, 0);
poptReadDefaultConfig(pc, 0);
while ((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case OPT_HELP:
SHOW_HELP();
goto end;
case OPT_LIST_OPTIONS:
list_cmd_options(stdout, long_options);
goto end;
case OPT_SESSION:
case OPT_PID:
opt_pid = 1;
break;
default:
command_ret = CMD_UNDEFINED;
goto end;
}
}
ret = print_missing_or_multiple_domains(opt_kernel + opt_userspace);
if (ret) {
ret = CMD_ERROR;
goto end;
}
if (!opt_session_name) {
session_name = get_session_name();
if (session_name == NULL) {
command_ret = CMD_ERROR;
goto end;
}
} else {
session_name = opt_session_name;
}
/* Currently only PID tracker is supported */
if (!opt_pid) {
ERR("Please specify at least one tracker with its expected arguments");
command_ret = CMD_ERROR;
goto end;
}
/* Mi check */
if (lttng_opt_mi) {
writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi);
if (!writer) {
command_ret = CMD_ERROR;
goto end;
}
}
if (writer) {
/* Open command element */
ret = mi_lttng_writer_command_open(writer,
get_mi_element_command(cmd_type));
if (ret) {
command_ret = CMD_ERROR;
goto end;
}
/* Open output element */
ret = mi_lttng_writer_open_element(writer,
mi_lttng_element_command_output);
if (ret) {
command_ret = CMD_ERROR;
goto end;
}
}
command_ret = track_untrack_pid(cmd_type,
cmd_str, session_name, opt_pid_string,
opt_all, writer);
if (command_ret != CMD_SUCCESS) {
success = 0;
}
/* Mi closing */
if (writer) {
/* Close output element */
ret = mi_lttng_writer_close_element(writer);
if (ret) {
command_ret = CMD_ERROR;
goto end;
}
/* Success ? */
ret = mi_lttng_writer_write_element_bool(writer,
mi_lttng_element_command_success, success);
if (ret) {
command_ret = CMD_ERROR;
goto end;
}
/* Command element close */
ret = mi_lttng_writer_command_close(writer);
if (ret) {
command_ret = CMD_ERROR;
goto end;
}
}
end:
if (!opt_session_name) {
free(session_name);
}
/* Mi clean-up */
if (writer && mi_lttng_writer_destroy(writer)) {
/* Preserve original error code */
command_ret = CMD_ERROR;
}
poptFreeContext(pc);
return (int) command_ret;
}
/*
* Start tracing for all trace of the session.
*/
static int stop_tracing(void)
{
int ret;
char *session_name;
if (opt_session_name == NULL) {
session_name = get_session_name();
if (session_name == NULL) {
ret = CMD_ERROR;
goto error;
}
} else {
session_name = opt_session_name;
}
ret = lttng_stop_tracing_no_wait(session_name);
if (ret < 0) {
switch (-ret) {
case LTTNG_ERR_TRACE_ALREADY_STOPPED:
WARN("Tracing already stopped for session %s", session_name);
break;
default:
ERR("%s", lttng_strerror(ret));
break;
}
goto free_name;
}
if (!opt_no_wait) {
_MSG("Waiting for data availability");
fflush(stdout);
do {
ret = lttng_data_pending(session_name);
if (ret < 0) {
/* Return the data available call error. */
goto free_name;
}
/*
* Data sleep time before retrying (in usec). Don't sleep if the call
* returned value indicates availability.
*/
if (ret) {
usleep(DEFAULT_DATA_AVAILABILITY_WAIT_TIME);
_MSG(".");
fflush(stdout);
}
} while (ret != 0);
MSG("");
}
ret = CMD_SUCCESS;
print_session_stats(session_name);
MSG("Tracing stopped for session %s", session_name);
if (lttng_opt_mi) {
ret = mi_print_session(session_name, 0);
if (ret) {
goto free_name;
}
}
free_name:
if (opt_session_name == NULL) {
free(session_name);
}
error:
return ret;
}
/*
* Add channel to trace session
*/
int cmd_enable_channels(int argc, const char **argv)
{
int opt, ret = CMD_SUCCESS, command_ret = CMD_SUCCESS, success = 1;
static poptContext pc;
char *session_name = NULL;
char *opt_arg = NULL;
const char *leftover = NULL;
init_channel_config();
pc = poptGetContext(NULL, argc, argv, long_options, 0);
poptReadDefaultConfig(pc, 0);
while ((opt = poptGetNextOpt(pc)) != -1) {
switch (opt) {
case OPT_HELP:
SHOW_HELP();
goto end;
case OPT_DISCARD:
chan_opts.attr.overwrite = 0;
DBG("Channel set to discard");
break;
case OPT_OVERWRITE:
chan_opts.attr.overwrite = 1;
DBG("Channel set to overwrite");
break;
case OPT_SUBBUF_SIZE:
{
uint64_t rounded_size;
int order;
/* Parse the size */
opt_arg = poptGetOptArg(pc);
if (utils_parse_size_suffix(opt_arg, &chan_opts.attr.subbuf_size) < 0 || !chan_opts.attr.subbuf_size) {
ERR("Wrong value in --subbuf-size parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
order = get_count_order_u64(chan_opts.attr.subbuf_size);
assert(order >= 0);
rounded_size = 1ULL << order;
if (rounded_size < chan_opts.attr.subbuf_size) {
ERR("The subbuf size (%" PRIu64 ") is rounded and overflows!",
chan_opts.attr.subbuf_size);
ret = CMD_ERROR;
goto end;
}
if (rounded_size != chan_opts.attr.subbuf_size) {
WARN("The subbuf size (%" PRIu64 ") is rounded to the next power of 2 (%" PRIu64 ")",
chan_opts.attr.subbuf_size, rounded_size);
chan_opts.attr.subbuf_size = rounded_size;
}
/* Should now be power of 2 */
assert(!((chan_opts.attr.subbuf_size - 1) & chan_opts.attr.subbuf_size));
DBG("Channel subbuf size set to %" PRIu64, chan_opts.attr.subbuf_size);
break;
}
case OPT_NUM_SUBBUF:
{
uint64_t rounded_size;
int order;
errno = 0;
opt_arg = poptGetOptArg(pc);
chan_opts.attr.num_subbuf = strtoull(opt_arg, NULL, 0);
if (errno != 0 || !chan_opts.attr.num_subbuf || !isdigit(opt_arg[0])) {
ERR("Wrong value in --num-subbuf parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
order = get_count_order_u64(chan_opts.attr.num_subbuf);
assert(order >= 0);
rounded_size = 1ULL << order;
if (rounded_size < chan_opts.attr.num_subbuf) {
ERR("The number of subbuffers (%" PRIu64 ") is rounded and overflows!",
chan_opts.attr.num_subbuf);
ret = CMD_ERROR;
goto end;
}
if (rounded_size != chan_opts.attr.num_subbuf) {
WARN("The number of subbuffers (%" PRIu64 ") is rounded to the next power of 2 (%" PRIu64 ")",
chan_opts.attr.num_subbuf, rounded_size);
chan_opts.attr.num_subbuf = rounded_size;
}
/* Should now be power of 2 */
assert(!((chan_opts.attr.num_subbuf - 1) & chan_opts.attr.num_subbuf));
DBG("Channel subbuf num set to %" PRIu64, chan_opts.attr.num_subbuf);
break;
}
case OPT_SWITCH_TIMER:
{
uint64_t v;
errno = 0;
opt_arg = poptGetOptArg(pc);
if (utils_parse_time_suffix(opt_arg, &v) < 0) {
ERR("Wrong value for --switch-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
if (v != (uint32_t) v) {
ERR("32-bit overflow in --switch-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
chan_opts.attr.switch_timer_interval = (uint32_t) v;
DBG("Channel switch timer interval set to %d %s",
chan_opts.attr.switch_timer_interval,
USEC_UNIT);
break;
}
case OPT_READ_TIMER:
{
uint64_t v;
errno = 0;
opt_arg = poptGetOptArg(pc);
if (utils_parse_time_suffix(opt_arg, &v) < 0) {
ERR("Wrong value for --read-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
if (v != (uint32_t) v) {
ERR("32-bit overflow in --read-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
chan_opts.attr.read_timer_interval = (uint32_t) v;
DBG("Channel read timer interval set to %d %s",
chan_opts.attr.read_timer_interval,
USEC_UNIT);
break;
}
case OPT_MONITOR_TIMER:
{
uint64_t v;
errno = 0;
opt_arg = poptGetOptArg(pc);
if (utils_parse_time_suffix(opt_arg, &v) < 0) {
ERR("Wrong value for --monitor-timer parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
opt_monitor_timer.interval = (uint64_t) v;
opt_monitor_timer.set = true;
DBG("Channel monitor timer interval set to %" PRIu64 " %s",
opt_monitor_timer.interval,
USEC_UNIT);
break;
}
case OPT_BLOCKING_TIMEOUT:
{
uint64_t v;
long long v_msec;
errno = 0;
opt_arg = poptGetOptArg(pc);
if (strcmp(opt_arg, "inf") == 0) {
opt_blocking_timeout.value = (int64_t) -1;
opt_blocking_timeout.set = true;
DBG("Channel blocking timeout set to infinity");
break;
}
if (utils_parse_time_suffix(opt_arg, &v) < 0) {
ERR("Wrong value for --blocking-timeout parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
/*
* While LTTng-UST and LTTng-tools will accept a
* blocking timeout expressed in µs, the current
* tracer implementation relies on poll() which
* takes an "int timeout" parameter expressed in
* msec.
*
* Since the error reporting from the tracer is
* not precise, we perform this check here to
* provide a helpful error message in case of
* overflow.
*
* The setter (liblttng-ctl) also performs an
* equivalent check.
*/
v_msec = v / 1000;
if (v_msec != (int32_t) v_msec) {
ERR("32-bit milliseconds overflow in --blocking-timeout parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
opt_blocking_timeout.value = (int64_t) v;
opt_blocking_timeout.set = true;
DBG("Channel blocking timeout set to %" PRId64 " %s%s",
opt_blocking_timeout.value,
USEC_UNIT,
opt_blocking_timeout.value == 0 ?
" (non-blocking)" : "");
break;
}
case OPT_USERSPACE:
opt_userspace = 1;
break;
case OPT_TRACEFILE_SIZE:
opt_arg = poptGetOptArg(pc);
if (utils_parse_size_suffix(opt_arg, &chan_opts.attr.tracefile_size) < 0) {
ERR("Wrong value in --tracefile-size parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
DBG("Maximum tracefile size set to %" PRIu64,
chan_opts.attr.tracefile_size);
break;
case OPT_TRACEFILE_COUNT:
{
unsigned long v;
errno = 0;
opt_arg = poptGetOptArg(pc);
v = strtoul(opt_arg, NULL, 0);
if (errno != 0 || !isdigit(opt_arg[0])) {
ERR("Wrong value in --tracefile-count parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
if (v != (uint32_t) v) {
ERR("32-bit overflow in --tracefile-count parameter: %s", opt_arg);
ret = CMD_ERROR;
goto end;
}
chan_opts.attr.tracefile_count = (uint32_t) v;
DBG("Maximum tracefile count set to %" PRIu64,
chan_opts.attr.tracefile_count);
break;
}
case OPT_LIST_OPTIONS:
list_cmd_options(stdout, long_options);
goto end;
default:
ret = CMD_UNDEFINED;
goto end;
}
}
ret = print_missing_or_multiple_domains(opt_kernel + opt_userspace);
if (ret) {
ret = CMD_ERROR;
goto end;
}
if (chan_opts.attr.overwrite == 1 && opt_blocking_timeout.set &&
opt_blocking_timeout.value != 0) {
ERR("You cannot specify --overwrite and --blocking-timeout=N, "
"where N is different than 0");
ret = CMD_ERROR;
goto end;
}
/* Mi check */
if (lttng_opt_mi) {
writer = mi_lttng_writer_create(fileno(stdout), lttng_opt_mi);
if (!writer) {
ret = -LTTNG_ERR_NOMEM;
goto end;
}
/* Open command element */
ret = mi_lttng_writer_command_open(writer,
mi_lttng_element_command_enable_channels);
if (ret) {
ret = CMD_ERROR;
goto end;
}
/* Open output element */
ret = mi_lttng_writer_open_element(writer,
mi_lttng_element_command_output);
if (ret) {
ret = CMD_ERROR;
goto end;
}
}
opt_channels = (char*) poptGetArg(pc);
if (opt_channels == NULL) {
ERR("Missing channel name.\n");
ret = CMD_ERROR;
success = 0;
goto mi_closing;
}
leftover = poptGetArg(pc);
if (leftover) {
ERR("Unknown argument: %s", leftover);
ret = CMD_ERROR;
success = 0;
goto mi_closing;
}
if (!opt_session_name) {
session_name = get_session_name();
if (session_name == NULL) {
command_ret = CMD_ERROR;
success = 0;
goto mi_closing;
}
} else {
session_name = opt_session_name;
}
command_ret = enable_channel(session_name);
if (command_ret) {
success = 0;
}
mi_closing:
/* Mi closing */
if (lttng_opt_mi) {
/* Close output element */
ret = mi_lttng_writer_close_element(writer);
if (ret) {
goto end;
}
/* Success ? */
ret = mi_lttng_writer_write_element_bool(writer,
mi_lttng_element_command_success, success);
if (ret) {
goto end;
}
/* Command element close */
ret = mi_lttng_writer_command_close(writer);
if (ret) {
goto end;
}
}
end:
/* Mi clean-up */
if (writer && mi_lttng_writer_destroy(writer)) {
/* Preserve original error code */
ret = ret ? ret : LTTNG_ERR_MI_IO_FAIL;
}
if (!opt_session_name && session_name) {
free(session_name);
}
/* Overwrite ret if an error occurred when enable_channel */
ret = command_ret ? command_ret : ret;
poptFreeContext(pc);
return ret;
}