/* * Machine interface * Print a list of channel * */ static int mi_list_channels(struct lttng_channel *channels, int count, const char *channel_name) { int i, ret; unsigned int chan_found = 0; /* Open channels element */ ret = mi_lttng_channels_open(writer); if (ret) { goto error; } for (i = 0; i < count; i++) { if (channel_name != NULL) { if (strncmp(channels[i].name, channel_name, NAME_MAX) == 0) { chan_found = 1; } else { continue; } } /* Write channel element and leave it open */ ret = mi_lttng_channel(writer, &channels[i], 1); if (ret) { goto error; } /* Listing events per channel */ ret = list_events(channels[i].name); if (ret) { goto error; } /* Closing the channel element we opened earlier */ ret = mi_lttng_writer_close_element(writer); if (ret) { goto error; } if (chan_found) { break; } } /* Close channels element */ ret = mi_lttng_writer_close_element(writer); if (ret) { goto error; } error: return ret; }
LTTNG_HIDDEN int mi_lttng_pid(struct mi_writer *writer, pid_t pid , const char *cmdline, int is_open) { int ret; /* Open element pid */ ret = mi_lttng_writer_open_element(writer, mi_lttng_element_pid); if (ret) { goto end; } /* Writing pid number */ ret = mi_lttng_writer_write_element_signed_int(writer, mi_lttng_element_pid_id, (int)pid); if (ret) { goto end; } /* Writing name of the process */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, cmdline); if (ret) { goto end; } if (!is_open) { /* Closing Pid */ ret = mi_lttng_writer_close_element(writer); } end: return ret; }
static int mi_lttng_app_context(struct mi_writer *writer, const char *provider_name, const char *ctx_name) { int ret; /* Open app */ ret = mi_lttng_writer_open_element(writer, config_element_context_app); if (ret) { goto end; } /* provider_name */ ret = mi_lttng_writer_write_element_string(writer, config_element_context_app_provider_name, provider_name); if (ret) { goto end; } /* ctx_name */ ret = mi_lttng_writer_write_element_string(writer, config_element_context_app_ctx_name, ctx_name); if (ret) { goto end; } /* Close app */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
/* * Machine interface * List all availables session */ static int mi_list_sessions(struct lttng_session *sessions, int count) { int ret, i; /* Opening sessions element */ ret = mi_lttng_sessions_open(writer); if (ret) { goto end; } /* Listing sessions */ for (i = 0; i < count; i++) { ret = mi_lttng_session(writer, &sessions[i], 0); if (ret) { goto end; } } /* Closing sessions element */ ret = mi_lttng_writer_close_element(writer); if (ret) { goto end; } end: return ret; }
/* * Mi print of partial session */ static int mi_print_session(char *session_name, int enabled) { int ret; assert(writer); assert(session_name); /* Open session element */ ret = mi_lttng_writer_open_element(writer, config_element_session); if (ret) { goto end; } /* Print session name element */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, session_name); if (ret) { goto end; } /* Is enabled ? */ ret = mi_lttng_writer_write_element_bool(writer, config_element_enabled, enabled); if (ret) { goto end; } /* Close session element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
/* * Machine interface * Print a list of system calls. */ static int mi_list_syscalls(struct lttng_event *events, int count) { int ret, i; /* Open events */ ret = mi_lttng_events_open(writer); if (ret) { goto end; } for (i = 0; i < count; i++) { ret = mi_lttng_event(writer, &events[i], 0, handle->domain.type); if (ret) { goto end; } } /* Close events. */ ret = mi_lttng_writer_close_element(writer); if (ret) { goto end; } end: return ret; }
/* * Mi print exlcusion list */ static int mi_print_exclusion(char **names) { int i, ret; int count = names ? strutils_array_of_strings_len(names) : 0; assert(writer); if (count == 0) { ret = 0; goto end; } ret = mi_lttng_writer_open_element(writer, config_element_exclusions); if (ret) { goto end; } for (i = 0; i < count; i++) { ret = mi_lttng_writer_write_element_string(writer, config_element_exclusion, names[i]); if (ret) { goto end; } } /* Close exclusions element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
/* * List all tracker of a domain */ static int list_trackers(void) { int ret; /* Trackers listing */ if (lttng_opt_mi) { ret = mi_lttng_trackers_open(writer); if (ret) { goto end; } } /* pid tracker */ ret = list_tracker_pids(); if (ret) { goto end; } if (lttng_opt_mi) { /* Close trackers element */ ret = mi_lttng_writer_close_element(writer); if (ret) { goto end; } } end: return ret; }
LTTNG_HIDDEN int mi_lttng_session(struct mi_writer *writer, struct lttng_session *session, int is_open) { int ret; assert(session); /* Open sessions element */ ret = mi_lttng_writer_open_element(writer, config_element_session); if (ret) { goto end; } /* Name of the session */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, session->name); if (ret) { goto end; } /* Path */ ret = mi_lttng_writer_write_element_string(writer, config_element_path, session->path); if (ret) { goto end; } /* Enabled ? */ ret = mi_lttng_writer_write_element_bool(writer, config_element_enabled, session->enabled); if (ret) { goto end; } /* Snapshot mode */ ret = mi_lttng_writer_write_element_unsigned_int(writer, config_element_snapshot_mode, session->snapshot_mode); if (ret) { goto end; } /* Live timer interval in usec */ ret = mi_lttng_writer_write_element_unsigned_int(writer, config_element_live_timer_interval, session->live_timer_interval); if (ret) { goto end; } if (!is_open) { /* Closing session element */ ret = mi_lttng_writer_close_element(writer); } end: return ret; }
static int mi_list_output(void) { int ret; struct lttng_snapshot_output *s_iter; struct lttng_snapshot_output_list *list; assert(writer); ret = lttng_snapshot_list_output(current_session_name, &list); if (ret < 0) { goto error; } ret = mi_lttng_snapshot_output_session_name(writer, current_session_name); if (ret) { ret = CMD_ERROR; goto end; } while ((s_iter = lttng_snapshot_output_list_get_next(list)) != NULL) { ret = mi_lttng_snapshot_list_output(writer, s_iter); if (ret) { ret = CMD_ERROR; goto end; } } /* Close snapshot snapshots element */ ret = mi_lttng_writer_close_element(writer); if (ret) { ret = CMD_ERROR; goto end; } /* Close snapshot session element */ ret = mi_lttng_writer_close_element(writer); if (ret) { ret = CMD_ERROR; } end: lttng_snapshot_output_list_destroy(list); error: return ret; }
static int write_event_exclusions(struct mi_writer *writer, struct lttng_event *event) { int i; int ret; int exclusion_count; /* Open event exclusions */ ret = mi_lttng_writer_open_element(writer, config_element_exclusions); if (ret) { goto end; } exclusion_count = lttng_event_get_exclusion_name_count(event); if (exclusion_count < 0) { ret = exclusion_count; goto end; } for (i = 0; i < exclusion_count; i++) { const char *name; ret = lttng_event_get_exclusion_name(event, i, &name); if (ret) { /* Close exclusions */ mi_lttng_writer_close_element(writer); goto end; } ret = mi_lttng_writer_write_element_string(writer, config_element_exclusion, name); if (ret) { /* Close exclusions */ mi_lttng_writer_close_element(writer); goto end; } } /* Close exclusions */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
LTTNG_HIDDEN int mi_lttng_snapshot_add_output(struct mi_writer *writer, const char *current_session_name, const char *n_ptr, struct lttng_snapshot_output *output) { int ret; /* Open element snapshot */ ret = mi_lttng_writer_open_element(writer, mi_lttng_element_command_snapshot); if (ret) { goto end; } /* Snapshot output id */ ret = mi_lttng_writer_write_element_unsigned_int(writer, mi_lttng_element_id, output->id); if (ret) { goto end; } /* Snapshot output names */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, n_ptr); if (ret) { goto end; } /* Destination of the output (ctrl_url)*/ ret = mi_lttng_writer_write_element_string(writer, mi_lttng_element_snapshot_ctrl_url, output->ctrl_url); if (ret) { goto end; } /* Snapshot added for session "current_session_name"*/ ret = mi_lttng_writer_write_element_string(writer, mi_lttng_element_snapshot_session_name, current_session_name); if (ret) { goto end; } /* total size of all stream combined */ ret = mi_lttng_writer_write_element_unsigned_int(writer, mi_lttng_element_snapshot_max_size, output->max_size); if (ret) { goto end; } /* Close snapshot element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
LTTNG_HIDDEN int mi_lttng_snapshot_list_output(struct mi_writer *writer, struct lttng_snapshot_output *output) { int ret; /* Open element snapshot output */ ret = mi_lttng_writer_open_element(writer, mi_lttng_element_command_snapshot); if (ret) { goto end; } /* ID of the snapshot output */ ret = mi_lttng_writer_write_element_unsigned_int(writer, mi_lttng_element_id, output->id); if (ret) { goto end; } /* Name of the output */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, output->name); if (ret) { goto end; } /* Destination of the output (ctrl_url)*/ ret = mi_lttng_writer_write_element_string(writer, mi_lttng_element_snapshot_ctrl_url, output->ctrl_url); if (ret) { goto end; } /* Destination of the output (data_url) */ ret = mi_lttng_writer_write_element_string(writer, mi_lttng_element_snapshot_data_url, output->data_url); if (ret) { goto end; } /* total size of all stream combined */ ret = mi_lttng_writer_write_element_unsigned_int(writer, mi_lttng_element_snapshot_max_size, output->max_size); if (ret) { goto end; } /* Close snapshot output element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
LTTNG_HIDDEN int mi_lttng_snapshot_del_output(struct mi_writer *writer, int id, const char *name, const char *current_session_name) { int ret; /* Open element del_snapshot */ ret = mi_lttng_writer_open_element(writer, mi_lttng_element_command_snapshot); if (ret) { goto end; } if (id != UINT32_MAX) { /* "Snapshot output "id" successfully deleted * for "current_session_name" * ID of the snapshot output */ ret = mi_lttng_writer_write_element_unsigned_int(writer, mi_lttng_element_id, id); if (ret) { goto end; } } else { /* "Snapshot output "name" successfully deleted * for session "current_session_name" * Name of the output */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, name); if (ret) { goto end; } } /* Snapshot was deleted for session "current_session_name"*/ ret = mi_lttng_writer_write_element_string(writer, mi_lttng_element_snapshot_session_name, current_session_name); if (ret) { goto end; } /* Close snapshot element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
LTTNG_HIDDEN int mi_lttng_event_field(struct mi_writer *writer, struct lttng_event_field *field) { int ret; if (!field->field_name[0]) { ret = 0; goto end; } /* Open field */ ret = mi_lttng_writer_open_element(writer, mi_lttng_element_event_field); if (ret) { goto end; } if (!field->field_name[0]) { goto close; } /* Name */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, field->field_name); if (ret) { goto end; } /* Type */ ret = mi_lttng_writer_write_element_string(writer, config_element_type, mi_lttng_eventfieldtype_string(field->type)); if (ret) { goto end; } /* nowrite */ ret = mi_lttng_writer_write_element_signed_int(writer, mi_lttng_element_nowrite, field->nowrite); if (ret) { goto end; } close: /* Close field element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
LTTNG_HIDDEN int mi_lttng_domain(struct mi_writer *writer, struct lttng_domain *domain, int is_open) { int ret = 0; const char *str_domain; const char *str_buffer; assert(domain); /* Open domain element */ ret = mi_lttng_writer_open_element(writer, config_element_domain); if (ret) { goto end; } /* Domain Type */ str_domain = mi_lttng_domaintype_string(domain->type); ret = mi_lttng_writer_write_element_string(writer, config_element_type, str_domain); if (ret) { goto end; } /* Buffer Type */ str_buffer= mi_lttng_buffertype_string(domain->buf_type); ret = mi_lttng_writer_write_element_string(writer, config_element_buffer_type, str_buffer); if (ret) { goto end; } /* TODO: union attr * This union is not currently used and was added for * future ust domain support. * Date: 25-06-2014 * */ if (!is_open) { /* Closing domain element */ ret = mi_lttng_writer_close_element(writer); } end: return ret; }
LTTNG_HIDDEN int mi_lttng_context(struct mi_writer *writer, struct lttng_event_context *context, int is_open) { int ret; const char *type_string; struct lttng_event_perf_counter_ctx *perf_context; /* Open context */ ret = mi_lttng_writer_open_element(writer , config_element_context); if (ret) { goto end; } type_string = mi_lttng_event_contexttype_string(context->ctx); if (!type_string) { ret = -LTTNG_ERR_INVALID; goto end; } /* Print context type */ ret = mi_lttng_writer_write_element_string(writer, config_element_type, type_string); /* Special case for PERF_*_COUNTER * print the lttng_event_perf_counter_ctx*/ switch (context->ctx) { case LTTNG_EVENT_CONTEXT_PERF_COUNTER: case LTTNG_EVENT_CONTEXT_PERF_THREAD_COUNTER: case LTTNG_EVENT_CONTEXT_PERF_CPU_COUNTER: perf_context = &context->u.perf_counter; ret = mi_lttng_perf_counter_context(writer, perf_context); if (ret) { goto end; } break; default: break; } /* Close context */ if (!is_open) { ret = mi_lttng_writer_close_element(writer); } end: return ret; }
LTTNG_HIDDEN int mi_lttng_snapshot_record(struct mi_writer *writer, const char *current_session_name, const char *url, const char *cmdline_ctrl_url, const char *cmdline_data_url) { int ret; /* Open element snapshot */ ret = mi_lttng_writer_open_element(writer, mi_lttng_element_command_snapshot); if (ret) { goto end; } /* * If a valid an URL was given, serialize it, * else take the command line data and ctrl urls*/ if (url) { /* Destination of the output (ctrl_url)*/ ret = mi_lttng_writer_write_element_string(writer, mi_lttng_element_snapshot_ctrl_url, url); if (ret) { goto end; } } else if (cmdline_ctrl_url) { /* Destination of the output (ctrl_url)*/ ret = mi_lttng_writer_write_element_string(writer, mi_lttng_element_snapshot_ctrl_url, cmdline_ctrl_url); if (ret) { goto end; } /* Destination of the output (data_url) */ ret = mi_lttng_writer_write_element_string(writer, mi_lttng_element_snapshot_data_url, cmdline_data_url); if (ret) { goto end; } } /* Close record_snapshot element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
LTTNG_HIDDEN int mi_lttng_channel(struct mi_writer *writer, struct lttng_channel *channel, int is_open) { int ret = 0; assert(channel); /* Opening channel element */ ret = mi_lttng_writer_open_element(writer, config_element_channel); if (ret) { goto end; } /* Name */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, channel->name); if (ret) { goto end; } /* Enabled ? */ ret = mi_lttng_writer_write_element_bool(writer, config_element_enabled, channel->enabled); if (ret) { goto end; } /* Attribute */ ret = mi_lttng_channel_attr(writer, &channel->attr); if (ret) { goto end; } if (!is_open) { /* Closing channel element */ ret = mi_lttng_writer_close_element(writer); if (ret) { goto end; } } end: return ret; }
LTTNG_HIDDEN int mi_lttng_close_multi_element(struct mi_writer *writer, unsigned int nb_element) { int ret, i; if (nb_element < 1) { ret = 0; goto end; } for (i = 0; i < nb_element; i++) { ret = mi_lttng_writer_close_element(writer); if (ret) { goto end; } } end: return ret; }
LTTNG_HIDDEN int mi_lttng_event(struct mi_writer *writer, struct lttng_event *event, int is_open, enum lttng_domain_type domain) { int ret; ret = mi_lttng_event_common_attributes(writer, event); if (ret) { goto end; } switch (event->type) { case LTTNG_EVENT_TRACEPOINT: { if (event->loglevel != -1) { ret = mi_lttng_event_tracepoint_loglevel(writer, event, domain); } else { ret = mi_lttng_event_tracepoint_no_loglevel(writer, event); } break; } case LTTNG_EVENT_FUNCTION: /* Fallthrough */ case LTTNG_EVENT_PROBE: ret = mi_lttng_event_function_probe(writer, event); break; case LTTNG_EVENT_FUNCTION_ENTRY: ret = mi_lttng_event_function_entry(writer, event); break; case LTTNG_EVENT_ALL: /* Fallthrough */ default: break; } if (!is_open) { ret = mi_lttng_writer_close_element(writer); } end: return ret; }
/* Mi print a partial event. * enabled is 0 or 1 * success is 0 or 1 */ static int mi_print_event(char *event_name, int enabled, int success) { int ret; assert(writer); assert(event_name); /* Open event element */ ret = mi_lttng_writer_open_element(writer, config_element_event); if (ret) { goto end; } /* Print the name of event */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, event_name); if (ret) { goto end; } /* Print enabled ? */ ret = mi_lttng_writer_write_element_bool(writer, config_element_enabled, enabled); if (ret) { goto end; } /* Success ? */ ret = mi_lttng_writer_write_element_bool(writer, mi_lttng_element_command_success, success); if (ret) { goto end; } /* Close event element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
LTTNG_HIDDEN int mi_lttng_perf_counter_context(struct mi_writer *writer, struct lttng_event_perf_counter_ctx *perf_context) { int ret; /* Open perf_counter_context */ ret = mi_lttng_writer_open_element(writer, mi_lttng_element_perf_counter_context); if (ret) { goto end; } /* Type */ ret = mi_lttng_writer_write_element_unsigned_int(writer, config_element_type, perf_context->type); if (ret) { goto end; } /* Config */ ret = mi_lttng_writer_write_element_unsigned_int(writer, config_element_config, perf_context->config); if (ret) { goto end; } /* Name of the perf counter */ ret = mi_lttng_writer_write_element_string(writer, config_element_name, perf_context->name); if (ret) { goto end; } /* Close perf_counter_context */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
static int mi_partial_session(const char *session_name) { int ret; assert(writer); assert(session_name); /* Open session element */ ret = mi_lttng_writer_open_element(writer, config_element_session); if (ret) { goto end; } ret = mi_lttng_writer_write_element_string(writer, config_element_name, session_name); if (ret) { goto end; } /* Closing session element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
LTTNG_HIDDEN int mi_lttng_pid_target(struct mi_writer *writer, pid_t pid, int is_open) { int ret; ret = mi_lttng_writer_open_element(writer, config_element_target_pid); if (ret) { goto end; } /* Writing pid number * Special case for element all on track untrack command * All pid is represented as wildcard * */ if ((int) pid == -1) { ret = mi_lttng_writer_write_element_string(writer, config_element_pid, mi_lttng_element_track_untrack_all_wildcard); } else { ret = mi_lttng_writer_write_element_signed_int(writer, config_element_pid, (int) pid); } if (ret) { goto end; } if (!is_open) { ret = mi_lttng_writer_close_element(writer); if (ret) { goto end; } } end: return ret; }
LTTNG_HIDDEN int mi_lttng_calibrate(struct mi_writer *writer, struct lttng_calibrate *calibrate) { int ret; /* Open calibrate element */ ret = mi_lttng_writer_open_element(writer, mi_lttng_element_calibrate); if (ret) { goto end; } /* Calibration type */ ret = mi_lttng_writer_write_element_string(writer, config_element_type, mi_lttng_calibratetype_string(calibrate->type)); if (ret) { goto end; } /* Closing calibrate element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
/* * Mi print of save command */ static int mi_save_print(const char *session_name) { int ret; assert(writer); if (opt_save_all) { /* We use a wildcard to represent all sessions */ session_name = "*"; } /* Print save element */ ret = mi_lttng_writer_open_element(writer, mi_lttng_element_save); if (ret) { goto end; } /* Print session element */ ret = mi_partial_session(session_name); if (ret) { goto end; } /* Path element */ if (opt_output_path) { ret = mi_lttng_writer_write_element_string(writer, config_element_path, opt_output_path); if (ret) { goto end; } } /* Close save element */ ret = mi_lttng_writer_close_element(writer); end: return ret; }
/* * Machine Interface * list available domain(s) for a session. */ static int mi_list_domains(struct lttng_domain *domains, int count) { int i, ret; /* Open domains element */ ret = mi_lttng_domains_open(writer); if (ret) { goto end; } for (i = 0; i < count; i++) { ret = mi_lttng_domain(writer, &domains[i] , 0); if (ret) { goto end; } } /* Closing domains element */ ret = mi_lttng_writer_close_element(writer); if (ret) { goto end; } end: return ret; }
/* * 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; }
/* * 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; }