API Fix : give access to trace_handle and context

[babeltrace.git] / lib / context.c

diff --git a/lib/context.c b/lib/context.c
index 0c20ed53de6b247ff996a85e0c49ca9b91fd0793..dbab33754cab346ba6c8a2054d72398338ffde23 100644 (file)
--- a/lib/context.c
+++ b/lib/context.c
@@ -50,6 +50,7 @@ struct bt_context *bt_context_create(void)
                                g_direct_equal, NULL,
                                (GDestroyNotify) bt_trace_handle_destroy);
 
+       ctx->current_iterator = NULL;
        ctx->tc = g_new0(struct trace_collection, 1);
        init_trace_collection(ctx->tc);
 
@@ -106,6 +107,11 @@ int bt_context_add_trace(struct bt_context *ctx, const char *path,
        strncpy(handle->path, path, PATH_MAX);
        handle->path[PATH_MAX - 1] = '\0';
 
+       if (fmt->set_handle)
+               fmt->set_handle(td, handle);
+       if (fmt->set_context)
+               fmt->set_context(td, ctx);
+
        /* Add new handle to container */
        g_hash_table_insert(ctx->trace_handles,
                (gpointer) (unsigned long) handle->id,