X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=lib%2Fcontext.c;h=87901b3c61027aadea02e0ddf0f9aa042b4aaac1;hp=5820fb3e5533ba97693044ce0e419ca0382545d8;hb=bb35f032b52330cfb17c144bfed81f2b1569f308;hpb=9a7312d380ce32824cb0856ef1333f9fa3b9f77f

diff --git a/lib/context.c b/lib/context.c
index 5820fb3e..87901b3c 100644
--- a/lib/context.c
+++ b/lib/context.c
@@ -115,7 +115,7 @@ int bt_context_add_trace(struct bt_context *ctx, const char *path,
 		fprintf(stderr, "[error] [Context] Creating trace handle %s .\n\n",
 				path);
 		ret = -1;
-		goto error;
+		goto error_close;
 	}
 	handle->format = fmt;
 	handle->td = td;
@@ -124,31 +124,47 @@ int bt_context_add_trace(struct bt_context *ctx, const char *path,
 		handle->path[PATH_MAX - 1] = '\0';
 	}
 
+	ret = bt_trace_collection_add(ctx->tc, td);
+	if (ret != 0)
+		goto error_destroy_handle;
+
 	if (fmt->set_handle)
 		fmt->set_handle(td, handle);
 	if (fmt->set_context)
 		fmt->set_context(td, ctx);
 
+	if (fmt->convert_index_timestamp) {
+		ret = fmt->convert_index_timestamp(td);
+		if (ret < 0)
+			goto error_collection_del;
+	}
+
+	if (fmt->timestamp_begin)
+		handle->real_timestamp_begin = fmt->timestamp_begin(td,
+				handle, BT_CLOCK_REAL);
+	if (fmt->timestamp_end)
+		handle->real_timestamp_end = fmt->timestamp_end(td, handle,
+				BT_CLOCK_REAL);
+	if (fmt->timestamp_begin)
+		handle->cycles_timestamp_begin = fmt->timestamp_begin(td,
+				handle, BT_CLOCK_CYCLES);
+	if (fmt->timestamp_end)
+		handle->cycles_timestamp_end = fmt->timestamp_end(td, handle,
+				BT_CLOCK_CYCLES);
+
 	/* Add new handle to container */
 	g_hash_table_insert(ctx->trace_handles,
 		(gpointer) (unsigned long) handle->id,
 		handle);
-	ret = bt_trace_collection_add(ctx->tc, td);
-	if (ret != 0)
-		goto error;
-
-	ret = fmt->convert_index_timestamp(td);
-	if (ret < 0)
-		goto error;
-
-	handle->real_timestamp_begin = fmt->timestamp_begin(td, handle, BT_CLOCK_REAL);
-	handle->real_timestamp_end = fmt->timestamp_end(td, handle, BT_CLOCK_REAL);
-	handle->cycles_timestamp_begin = fmt->timestamp_begin(td, handle, BT_CLOCK_CYCLES);
-	handle->cycles_timestamp_end = fmt->timestamp_end(td, handle, BT_CLOCK_CYCLES);
 
 	return handle->id;
 
-error:
+error_collection_del:
+	/* Remove from containers */
+	bt_trace_collection_remove(handle->td->ctx->tc, handle->td);
+error_destroy_handle:
+	bt_trace_handle_destroy(handle);
+error_close:
 	closeret = fmt->close_trace(td);
 	if (closeret) {
 		fprintf(stderr, "Error in close_trace callback\n");
@@ -217,6 +233,8 @@ void remove_trace_handle(struct bt_trace_handle *handle)
 {
 	int ret;
 
+	if (!handle->td->ctx)
+		return;
 	/* Remove from containers */
 	bt_trace_collection_remove(handle->td->ctx->tc, handle->td);
 	/* Close the trace */