Update version to 1.0.0-pre2
[babeltrace.git] / lib / context.c
CommitLineData
b469d2dd
JD
1/*
2 * context.c
3 *
4 * Babeltrace Library
5 *
6 * Copyright 2011-2012 EfficiOS Inc. and Linux Foundation
7 *
8 * Author: Mathieu Desnoyers <mathieu.desnoyers@efficios.com>
9 * Julien Desfossez <julien.desfossez@efficios.com>
10 *
11 * Permission is hereby granted, free of charge, to any person obtaining a copy
12 * of this software and associated documentation files (the "Software"), to deal
13 * in the Software without restriction, including without limitation the rights
14 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
15 * copies of the Software, and to permit persons to whom the Software is
16 * furnished to do so, subject to the following conditions:
17 *
18 * The above copyright notice and this permission notice shall be included in
19 * all copies or substantial portions of the Software.
20 */
21
22#include <babeltrace/babeltrace.h>
23#include <babeltrace/context.h>
08c22d05 24#include <babeltrace/context-internal.h>
6cba487f
MD
25#include <babeltrace/trace-handle.h>
26#include <babeltrace/trace-handle-internal.h>
27#include <babeltrace/trace-collection.h>
28#include <babeltrace/format.h>
e1d01c39 29#include <babeltrace/babeltrace-internal.h>
b469d2dd 30#include <stdlib.h>
e1d01c39
MD
31#include <string.h>
32#include <assert.h>
b469d2dd 33
6cba487f
MD
34#include <fts.h>
35#include <fcntl.h> /* For O_RDONLY */
36
6cba487f
MD
37#include <glib.h>
38
39struct bt_context *bt_context_create(void)
b469d2dd
JD
40{
41 struct bt_context *ctx;
42
6cba487f 43 ctx = g_new0(struct bt_context, 1);
b469d2dd 44 ctx->refcount = 1;
6cba487f 45 /* Negative handle id are errors. */
842c2b97 46 ctx->last_trace_handle_id = 0;
b469d2dd 47
6cba487f
MD
48 /* Instanciate the trace handle container */
49 ctx->trace_handles = g_hash_table_new_full(g_direct_hash,
50 g_direct_equal, NULL,
51 (GDestroyNotify) bt_trace_handle_destroy);
52
53 ctx->tc = g_new0(struct trace_collection, 1);
54 init_trace_collection(ctx->tc);
55
b469d2dd 56 return ctx;
6cba487f 57}
b469d2dd 58
6cba487f 59int bt_context_add_trace(struct bt_context *ctx, const char *path,
613f532b 60 const char *format_name,
20d0dcf9 61 void (*packet_seek)(struct stream_pos *pos, size_t index,
0d4c669f
MD
62 int whence),
63 struct mmap_stream_list *stream_list,
64 FILE *metadata)
6cba487f
MD
65{
66 struct trace_descriptor *td;
67 struct format *fmt;
68 struct bt_trace_handle *handle;
1059a2bf 69 int ret;
6cba487f 70
282e1952
MD
71 fmt = bt_lookup_format(g_quark_from_string(format_name));
72 if (!fmt) {
73 fprintf(stderr, "[error] [Context] Format \"%s\" unknown.\n\n",
74 format_name);
75 ret = -1;
76 goto end;
77 }
0d4c669f
MD
78 if (path) {
79 td = fmt->open_trace(path, O_RDONLY, packet_seek, NULL);
80 if (!td) {
81 fprintf(stderr, "[error] [Context] Cannot open_trace of the format %s .\n\n",
82 path);
83 ret = -1;
84 goto end;
85 }
86 } else {
87 td = fmt->open_mmap_trace(stream_list, packet_seek, metadata);
88 if (!td) {
89 fprintf(stderr, "[error] [Context] Cannot open_trace of the format %s .\n\n",
90 path);
91 ret = -1;
92 goto end;
93 }
6cba487f
MD
94 }
95
96 /* Create an handle for the trace */
97 handle = bt_trace_handle_create(ctx);
98 if (handle < 0) {
02dc4610 99 fprintf(stderr, "[error] [Context] Creating trace handle %s .\n\n",
6cba487f 100 path);
1059a2bf
JD
101 ret = -1;
102 goto end;
6cba487f
MD
103 }
104 handle->format = fmt;
105 handle->td = td;
106 strncpy(handle->path, path, PATH_MAX);
107 handle->path[PATH_MAX - 1] = '\0';
108
109 /* Add new handle to container */
110 g_hash_table_insert(ctx->trace_handles,
111 (gpointer) (unsigned long) handle->id,
112 handle);
1059a2bf
JD
113 ret = trace_collection_add(ctx->tc, td);
114 if (ret == 0)
115 return handle->id;
116end:
117 return ret;
b469d2dd
JD
118}
119
6cba487f 120void bt_context_remove_trace(struct bt_context *ctx, int handle_id)
b469d2dd 121{
6cba487f
MD
122 struct bt_trace_handle *handle;
123
124 handle = g_hash_table_lookup(ctx->trace_handles,
125 (gpointer) (unsigned long) handle_id);
126 assert(handle != NULL);
127
128 /* Remove from containers */
129 trace_collection_remove(ctx->tc, handle->td);
130 g_hash_table_remove(ctx->trace_handles,
131 (gpointer) (unsigned long) handle_id);
132
133 /* Close the trace */
134 handle->format->close_trace(handle->td);
135
136 /* Destory the handle */
137 bt_trace_handle_destroy(handle);
138}
139
140static
141void bt_context_destroy(struct bt_context *ctx)
142{
143 finalize_trace_collection(ctx->tc);
144
145 /* Remote all traces. The g_hash_table_destroy will call
146 * bt_trace_handle_destroy on each elements.
147 */
148 g_hash_table_destroy(ctx->trace_handles);
149
150 /* ctx->tc should always be valid */
151 assert(ctx->tc != NULL);
152 g_free(ctx->tc);
153 g_free(ctx);
b469d2dd
JD
154}
155
6cba487f 156void bt_context_get(struct bt_context *ctx)
b469d2dd 157{
6cba487f
MD
158 ctx->refcount++;
159}
b469d2dd 160
6cba487f
MD
161void bt_context_put(struct bt_context *ctx)
162{
b469d2dd
JD
163 ctx->refcount--;
164 if (ctx->refcount == 0)
6cba487f 165 bt_context_destroy(ctx);
b469d2dd 166}
This page took 0.02956 seconds and 4 git commands to generate.