CTF trace open/close

[babeltrace.git] / converter / babeltrace.c

diff --git a/converter/babeltrace.c b/converter/babeltrace.c
index 63ade113c81f719a1cb4b1a5b0178a5567fb826b..b603dee53e22a555b0f7b94c9dda7e3651c71ddf 100644 (file)
--- a/converter/babeltrace.c
+++ b/converter/babeltrace.c
@@ -21,15 +21,27 @@
 #include <popt.h>
 #include <errno.h>
 #include <stdlib.h>
+#include <ctype.h>
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <fcntl.h>
 
-static const char *opt_input_format;
-static const char *opt_output_format;
+static char *opt_input_format;
+static char *opt_output_format;
 
 static const char *opt_input_path;
 static const char *opt_output_path;
 
 int babeltrace_verbose, babeltrace_debug;
 
+void strlower(char *str)
+{
+       while (*str) {
+               *str = tolower(*str);
+               str++;
+       }
+}
+
 enum {
        OPT_NONE = 0,
        OPT_HELP,
@@ -79,12 +91,12 @@ static void usage(FILE *fp)
  * 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, const char **argv)
+static int parse_options(int argc, char **argv)
 {
        poptContext pc;
        int opt, ret = 0;
 
-       pc = poptGetContext(NULL, argc, argv, long_options, 0);
+       pc = poptGetContext(NULL, argc, (const char **) argv, long_options, 0);
        poptReadDefaultConfig(pc, 0);
 
        while ((opt = poptGetNextOpt(pc)) != -1) {
@@ -126,13 +138,15 @@ end:
        return ret;
 }
 
-int main(int argc, const char **argv)
+int main(int argc, char **argv)
 {
        int ret;
+       struct format *fmt_read, *fmt_write;
+       struct trace_descriptor *td_read, *td_write;
 
        ret = parse_options(argc, argv);
        if (ret < 0) {
-               fprintf(stdout, "Error parsing options.\n");
+               fprintf(stdout, "Error parsing options.\n\n");
                usage(stdout);
                exit(EXIT_FAILURE);
        } else if (ret > 0) {
@@ -141,12 +155,60 @@ int main(int argc, const char **argv)
        printf_verbose("Verbose mode active.\n");
        printf_debug("Debug mode active.\n");
 
+       if (opt_input_format)
+               strlower(opt_input_format);
+       if (opt_output_format)
+               strlower(opt_output_format);
+
        printf_verbose("Converting from file: %s\n", opt_input_path);
        printf_verbose("Converting from format: %s\n",
                opt_input_format ? : "<autodetect>");
        printf_verbose("Converting to file: %s\n", opt_output_path);
        printf_verbose("Converting to format: %s\n",
-               opt_output_format ? : "CTF");
+               opt_output_format ? : "ctf");
+
+       if (!opt_input_format) {
+               fprintf(stdout, "Error: input format autodetection not implemented yet.\n\n");
+               usage(stdout);
+               exit(EXIT_FAILURE);
+       }
+       fmt_read = bt_lookup_format(g_quark_from_static_string(opt_input_format));
+       if (!fmt_read) {
+               fprintf(stdout, "Error: format \"%s\" is not supported.\n\n",
+                       opt_input_format);
+               exit(EXIT_FAILURE);
+       }
+       if (!opt_output_format)
+               opt_output_format = "ctf";
+       fmt_write = bt_lookup_format(g_quark_from_static_string(opt_output_format));
+       if (!fmt_write) {
+               fprintf(stdout, "Error: format \"%s\" is not supported.\n\n",
+                       opt_output_format);
+               exit(EXIT_FAILURE);
+       }
+
+       td_read = fmt_read->open_trace(opt_input_path, O_RDONLY);
+       if (!td_read) {
+               fprintf(stdout, "Error opening trace \"%s\" for reading.\n\n",
+                       opt_input_path);
+               goto error_td_read;
+       }
+
+       td_write = fmt_write->open_trace(opt_output_path, O_WRONLY);
+       if (!td_write) {
+               fprintf(stdout, "Error opening trace \"%s\" for writing.\n\n",
+                       opt_output_path);
+               goto error_td_write;
+       }
+
+       fmt_write->close_trace(td_write);
+       fmt_read->close_trace(td_read);
+       exit(EXIT_SUCCESS);
 
-       return 0;
+       /* Error handling */
+       fmt_write->close_trace(td_write);
+error_td_write:
+       fmt_read->close_trace(td_read);
+error_td_read:
+       exit(EXIT_FAILURE);
 }