+
+/*
+ * Return 0 if caller should continue, < 0 if caller should return
+ * error, > 0 if caller should exit without reporting error.
+ */
+static int parse_options(int argc, char **argv)
+{
+ poptContext pc;
+ int opt, ret = 0;
+
+ if (argc == 1) {
+ usage(stdout);
+ return 1; /* exit cleanly */
+ }
+
+ pc = poptGetContext(NULL, argc, (const char **) argv, long_options, 0);
+ poptReadDefaultConfig(pc, 0);
+
+ /* set default */
+ opt_context_field_names = 1;
+ opt_payload_field_names = 1;
+
+ while ((opt = poptGetNextOpt(pc)) != -1) {
+ switch (opt) {
+ case OPT_HELP:
+ usage(stdout);
+ ret = 1; /* exit cleanly */
+ goto end;
+ case OPT_LIST:
+ list_formats(stdout);
+ ret = 1;
+ goto end;
+ case OPT_VERBOSE:
+ babeltrace_verbose = 1;
+ break;
+ case OPT_NAMES:
+ if (get_names_args(&pc)) {
+ ret = -EINVAL;
+ goto end;
+ }
+ break;
+ case OPT_FIELDS:
+ if (get_fields_args(&pc)) {
+ ret = -EINVAL;
+ goto end;
+ }
+ break;
+ case OPT_DEBUG:
+ babeltrace_debug = 1;
+ break;
+ case OPT_NO_DELTA:
+ opt_delta_field = 0;
+ break;
+ case OPT_CLOCK_RAW:
+ opt_clock_raw = 1;
+ break;
+ case OPT_CLOCK_OFFSET:
+ {
+ const char *str;
+ char *endptr;
+
+ str = poptGetOptArg(pc);
+ if (!str) {
+ fprintf(stderr, "[error] Missing --clock-offset argument\n");
+ ret = -EINVAL;
+ goto end;
+ }
+ errno = 0;
+ opt_clock_offset = strtoull(str, &endptr, 0);
+ if (*endptr != '\0' || str == endptr || errno != 0) {
+ fprintf(stderr, "[error] Incorrect --clock-offset argument: %s\n", str);
+ ret = -EINVAL;
+ goto end;
+ }
+ break;
+ }
+ case OPT_CLOCK_SECONDS:
+ opt_clock_seconds = 1;
+ break;
+ case OPT_CLOCK_DATE:
+ opt_clock_date = 1;
+ break;
+ case OPT_CLOCK_GMT:
+ opt_clock_gmt = 1;
+ break;
+ case OPT_CLOCK_FORCE_CORRELATE:
+ opt_clock_force_correlate = 1;
+ break;
+
+ default:
+ ret = -EINVAL;
+ goto end;
+ }
+ }
+
+ opt_input_path = poptGetArg(pc);
+ if (!opt_input_path) {
+ ret = -EINVAL;
+ goto end;
+ }
+ opt_output_path = poptGetArg(pc);
+
+end:
+ if (pc) {
+ poptFreeContext(pc);
+ }
+ return ret;
+}
+
+
+/*
+ * bt_context_add_traces_recursive: Open a trace recursively
+ *
+ * Find each trace present in the subdirectory starting from the given
+ * path, and add them to the context. The packet_seek parameter can be
+ * NULL: this specify to use the default format packet_seek.
+ *
+ * Return: 0 on success, < 0 on failure, > 0 on partial failure.
+ * Unable to open toplevel: failure.
+ * Unable to open some subdirectory or file: warn and continue (partial
+ * failure);
+ */
+int bt_context_add_traces_recursive(struct bt_context *ctx, const char *path,
+ const char *format_str,
+ void (*packet_seek)(struct stream_pos *pos,
+ size_t offset, int whence))
+{
+ FTS *tree;
+ FTSENT *node;
+ GArray *trace_ids;
+ char lpath[PATH_MAX];
+ char * const paths[2] = { lpath, NULL };
+ int ret = 0;
+
+ /*
+ * Need to copy path, because fts_open can change it.
+ * It is the pointer array, not the strings, that are constant.
+ */
+ strncpy(lpath, path, PATH_MAX);
+ lpath[PATH_MAX - 1] = '\0';
+
+ tree = fts_open(paths, FTS_NOCHDIR | FTS_LOGICAL, 0);
+ if (tree == NULL) {
+ fprintf(stderr, "[error] [Context] Cannot traverse \"%s\" for reading.\n",
+ path);
+ return -EINVAL;
+ }
+
+ trace_ids = g_array_new(FALSE, TRUE, sizeof(int));
+
+ while ((node = fts_read(tree))) {
+ int dirfd, metafd;
+ int closeret;
+
+ if (!(node->fts_info & FTS_D))
+ continue;
+
+ dirfd = open(node->fts_accpath, 0);
+ if (dirfd < 0) {
+ fprintf(stderr, "[error] [Context] Unable to open trace "
+ "directory file descriptor.\n");
+ ret = 1; /* partial error */
+ goto error;
+ }
+ metafd = openat(dirfd, "metadata", O_RDONLY);
+ if (metafd < 0) {
+ closeret = close(dirfd);
+ if (closeret < 0) {
+ perror("close");
+ ret = -1; /* failure */
+ goto error;
+ }
+ continue;
+ } else {
+ int trace_id;
+
+ closeret = close(metafd);
+ if (closeret < 0) {
+ perror("close");
+ ret = -1; /* failure */
+ goto error;
+ }
+ closeret = close(dirfd);
+ if (closeret < 0) {
+ perror("close");
+ ret = -1; /* failure */
+ goto error;
+ }
+
+ trace_id = bt_context_add_trace(ctx,
+ node->fts_accpath, format_str,
+ packet_seek, NULL, NULL);
+ if (trace_id < 0) {
+ fprintf(stderr, "[warning] [Context] opening trace \"%s\" from %s "
+ "for reading.\n", node->fts_accpath, path);
+ /* Allow to skip erroneous traces. */
+ ret = 1; /* partial error */
+ continue;
+ }
+ g_array_append_val(trace_ids, trace_id);
+ }
+ }
+
+error:
+ /*
+ * Return an error if no trace can be opened.
+ */
+ if (trace_ids->len == 0) {
+ fprintf(stderr, "[error] Cannot open any trace for reading.\n\n");
+ ret = -ENOENT; /* failure */
+ }
+ g_array_free(trace_ids, TRUE);
+ return ret;
+}
+
+
+
+int convert_trace(struct trace_descriptor *td_write,
+ struct bt_context *ctx)
+{
+ struct bt_ctf_iter *iter;
+ struct ctf_text_stream_pos *sout;
+ struct bt_iter_pos begin_pos;
+ struct bt_ctf_event *ctf_event;
+ int ret;
+
+ sout = container_of(td_write, struct ctf_text_stream_pos,
+ trace_descriptor);
+
+ begin_pos.type = BT_SEEK_BEGIN;
+ iter = bt_ctf_iter_create(ctx, &begin_pos, NULL);
+ if (!iter) {
+ ret = -1;
+ goto error_iter;
+ }
+ while ((ctf_event = bt_ctf_iter_read_event(iter))) {
+ ret = sout->parent.event_cb(&sout->parent, ctf_event->parent->stream);
+ if (ret) {
+ fprintf(stderr, "[error] Writing event failed.\n");
+ goto end;
+ }
+ ret = bt_iter_next(bt_ctf_get_iter(iter));
+ if (ret < 0)
+ goto end;
+ }
+ ret = 0;
+
+end:
+ bt_ctf_iter_destroy(iter);
+error_iter:
+ return ret;
+}
+
+int main(int argc, char **argv)
+{
+ int ret, partial_error = 0;
+ struct format *fmt_write;
+ struct trace_descriptor *td_write;
+ struct bt_context *ctx;
+
+ ret = parse_options(argc, argv);
+ if (ret < 0) {
+ fprintf(stderr, "Error parsing options.\n\n");
+ usage(stderr);
+ exit(EXIT_FAILURE);
+ } else if (ret > 0) {
+ exit(EXIT_SUCCESS);
+ }
+ printf_verbose("Verbose mode active.\n");
+ printf_debug("Debug mode active.\n");
+
+ if (opt_input_format_arg) {
+ opt_input_format = strdup(opt_input_format_arg);
+ if (!opt_input_format) {
+ partial_error = 1;
+ goto end;
+ }
+ strlower(opt_input_format);
+ }
+ if (opt_output_format) {
+ opt_output_format = strdup(opt_output_format_arg);
+ if (!opt_output_format) {
+ partial_error = 1;
+ goto end;
+ }
+ strlower(opt_output_format);
+ }
+
+ printf_verbose("Converting from directory: %s\n", opt_input_path);
+ printf_verbose("Converting from format: %s\n",
+ opt_input_format ? : "ctf <default>");
+ printf_verbose("Converting to directory: %s\n",
+ opt_output_path ? : "<stdout>");
+ printf_verbose("Converting to format: %s\n",
+ opt_output_format ? : "text <default>");
+
+ if (!opt_input_format) {
+ opt_input_format = strdup("ctf");
+ if (!opt_input_format) {
+ partial_error = 1;
+ goto end;
+ }
+ }
+ if (!opt_output_format) {
+ opt_output_format = strdup("text");
+ if (!opt_output_format) {
+ partial_error = 1;
+ goto end;
+ }
+ }
+ fmt_read = bt_lookup_format(g_quark_from_static_string(opt_input_format));
+ if (!fmt_read) {
+ fprintf(stderr, "[error] Format \"%s\" is not supported.\n\n",
+ opt_input_format);
+ partial_error = 1;
+ goto end;
+ }
+ fmt_write = bt_lookup_format(g_quark_from_static_string(opt_output_format));
+ if (!fmt_write) {
+ fprintf(stderr, "[error] format \"%s\" is not supported.\n\n",
+ opt_output_format);
+ partial_error = 1;
+ goto end;
+ }
+
+ ctx = bt_context_create();
+
+ ret = bt_context_add_traces_recursive(ctx, opt_input_path,
+ opt_input_format, NULL);
+ if (ret < 0) {
+ fprintf(stderr, "[error] opening trace \"%s\" for reading.\n\n",
+ opt_input_path);
+ goto error_td_read;
+ } else if (ret > 0) {
+ fprintf(stderr, "[warning] errors occurred when opening trace \"%s\" for reading, continuing anyway.\n\n",
+ opt_input_path);
+ partial_error = 1;
+ }
+
+ td_write = fmt_write->open_trace(opt_output_path, O_RDWR, NULL, NULL);
+ if (!td_write) {
+ fprintf(stderr, "Error opening trace \"%s\" for writing.\n\n",
+ opt_output_path ? : "<none>");
+ goto error_td_write;
+ }
+
+ ret = convert_trace(td_write, ctx);
+ if (ret) {
+ fprintf(stderr, "Error printing trace.\n\n");
+ goto error_copy_trace;
+ }
+
+ fmt_write->close_trace(td_write);
+
+ bt_context_put(ctx);
+ printf_verbose("finished converting. Output written to:\n%s\n",
+ opt_output_path ? : "<stdout>");
+ goto end;
+
+ /* Error handling */
+error_copy_trace:
+ fmt_write->close_trace(td_write);
+error_td_write:
+ bt_context_put(ctx);
+error_td_read:
+ partial_error = 1;
+
+ /* teardown and exit */
+end:
+ free(opt_input_format);
+ free(opt_output_format);
+ if (partial_error)
+ exit(EXIT_FAILURE);
+ else
+ exit(EXIT_SUCCESS);
+}