Fix API: allow packet_seek override

[babeltrace.git] / lib / context.c

diff --git a/lib/context.c b/lib/context.c
index 90d77df88f6840af5a954616048f5fbc0f81c631..5021d7add1c56ab92386e4f93cdd43c520509086 100644 (file)
--- a/lib/context.c
+++ b/lib/context.c
@@ -26,14 +26,14 @@
 #include <babeltrace/trace-handle-internal.h>
 #include <babeltrace/trace-collection.h>
 #include <babeltrace/format.h>
+#include <babeltrace/babeltrace-internal.h>
 #include <stdlib.h>
+#include <string.h>
+#include <assert.h>
 
 #include <fts.h>
 #include <fcntl.h> /* For O_RDONLY */
 
-/* TODO ybrosseau: should be hidden in the CTF format */
-#include <babeltrace/ctf/types.h> /* for ctf_move_pos_slow */
-
 #include <glib.h>
 
 struct bt_context *bt_context_create(void)
@@ -57,16 +57,23 @@ struct bt_context *bt_context_create(void)
 }
 
 int bt_context_add_trace(struct bt_context *ctx, const char *path,
-               const char *format_str)
+               const char *format_name,
+               void (*packet_seek)(struct stream_pos *pos,
+                       size_t offset, int whence))
 {
        struct trace_descriptor *td;
        struct format *fmt;
        struct bt_trace_handle *handle;
        int ret;
 
-       fmt = bt_lookup_format(g_quark_from_string(format_str));
-       td = fmt->open_trace(path, O_RDONLY,
-                            ctf_move_pos_slow, NULL);
+       fmt = bt_lookup_format(g_quark_from_string(format_name));
+       if (!fmt) {
+               fprintf(stderr, "[error] [Context] Format \"%s\" unknown.\n\n",
+                       format_name);
+               ret = -1;
+               goto end;
+       }
+       td = fmt->open_trace(path, O_RDONLY, packet_seek, NULL);
        if (!td) {
                fprintf(stderr, "[error] [Context] Cannot open_trace of the format %s .\n\n",
                                path);