#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)
}
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 index,
+ int whence),
+ struct mmap_stream_list *stream_list,
+ FILE *metadata)
{
struct trace_descriptor *td;
struct format *fmt;
struct bt_trace_handle *handle;
-
- fmt = bt_lookup_format(g_quark_from_string(format_str));
- td = fmt->open_trace(path, O_RDONLY,
- ctf_move_pos_slow, NULL);
- if (!td) {
- fprintf(stderr, "[error] [Context] Cannot open_trace of the format %s .\n\n",
- path);
- return -1;
+ int ret;
+
+ 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;
+ }
+ if (path) {
+ 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);
+ ret = -1;
+ goto end;
+ }
+ } else {
+ td = fmt->open_mmap_trace(stream_list, packet_seek, metadata);
+ if (!td) {
+ fprintf(stderr, "[error] [Context] Cannot open_trace of the format %s .\n\n",
+ path);
+ ret = -1;
+ goto end;
+ }
}
/* Create an handle for the trace */
if (handle < 0) {
fprintf(stderr, "[error] [Context] Creating trace handle %s .\n\n",
path);
- return -1;
+ ret = -1;
+ goto end;
}
handle->format = fmt;
handle->td = td;
g_hash_table_insert(ctx->trace_handles,
(gpointer) (unsigned long) handle->id,
handle);
- return trace_collection_add(ctx->tc, td);
+ ret = trace_collection_add(ctx->tc, td);
+ if (ret == 0)
+ return handle->id;
+end:
+ return ret;
}
void bt_context_remove_trace(struct bt_context *ctx, int handle_id)