*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+ * SOFTWARE.
*/
#include <babeltrace/babeltrace.h>
#include <babeltrace/trace-handle-internal.h>
#include <babeltrace/trace-collection.h>
#include <babeltrace/format.h>
+#include <babeltrace/format-internal.h>
+#include <babeltrace/babeltrace-internal.h>
#include <stdlib.h>
+#include <string.h>
+#include <assert.h>
+#include <errno.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>
+static
+void remove_trace_handle(struct bt_trace_handle *handle);
+
struct bt_context *bt_context_create(void)
{
struct bt_context *ctx;
/* Instanciate the trace handle container */
ctx->trace_handles = g_hash_table_new_full(g_direct_hash,
g_direct_equal, NULL,
- (GDestroyNotify) bt_trace_handle_destroy);
+ (GDestroyNotify) remove_trace_handle);
+ ctx->current_iterator = NULL;
ctx->tc = g_new0(struct trace_collection, 1);
- init_trace_collection(ctx->tc);
+ bt_init_trace_collection(ctx->tc);
return ctx;
}
int bt_context_add_trace(struct bt_context *ctx, const char *path,
- const char *format_str)
+ const char *format_name,
+ void (*packet_seek)(struct bt_stream_pos *pos, size_t index,
+ int whence),
+ struct bt_mmap_stream_list *stream_list,
+ FILE *metadata)
{
- struct trace_descriptor *td;
- struct format *fmt;
+ struct bt_trace_descriptor *td;
+ struct bt_format *fmt;
struct bt_trace_handle *handle;
+ int ret, closeret;
- 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(stdout, "[error] [Context] Cannot Open_trace of the format %s .\n\n",
- path);
- return 0;
+ if (!ctx || !format_name || (!path && !stream_list))
+ return -EINVAL;
+
+ 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, "[warning] [Context] Cannot open_trace of format %s at path %s.\n",
+ format_name, path);
+ ret = -1;
+ goto end;
+ }
+ } else {
+ td = fmt->open_mmap_trace(stream_list, packet_seek, metadata);
+ if (!td) {
+ fprintf(stderr, "[error] [Context] Cannot open_mmap_trace of format %s.\n\n",
+ format_name);
+ ret = -1;
+ goto end;
+ }
}
/* Create an handle for the trace */
handle = bt_trace_handle_create(ctx);
- if (handle < 0) {
- fprintf(stdout, "[error] [Context] Creating trace handle %s .\n\n",
+ if (!handle) {
+ fprintf(stderr, "[error] [Context] Creating trace handle %s .\n\n",
path);
- return 0;
+ ret = -1;
+ goto error;
}
handle->format = fmt;
handle->td = td;
- strncpy(handle->path, path, PATH_MAX);
- handle->path[PATH_MAX - 1] = '\0';
+ if (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,
handle);
- trace_collection_add(ctx->tc, td);
- return handle->id;
-}
-
-/*
- * Unable to open toplevel: failure.
- * Unable to open some subdirectory or file: warn and continue;
- */
-int bt_context_add_traces(struct bt_context *ctx, const char *path,
- const char *format_str)
-{
- FTS *tree;
- FTSENT *node;
- GArray *trace_ids;
- char lpath[PATH_MAX];
- char * const paths[2] = { lpath, NULL };
- int ret;
-
- /*
- * Need to copy path, because fts_open can change it.
- * It is the pointer array, not the strings, that are constant.
- */
- strncpy(lpath, path, PATH_MAX);
- lpath[PATH_MAX - 1] = '\0';
-
- tree = fts_open(paths, FTS_NOCHDIR | FTS_LOGICAL, 0);
- if (tree == NULL) {
- fprintf(stdout, "[error] Cannot traverse \"%s\" for reading.\n\n",
- path);
- return -EINVAL;
+ ret = bt_trace_collection_add(ctx->tc, td);
+ if (ret != 0)
+ goto error;
+
+ if (fmt->convert_index_timestamp) {
+ ret = fmt->convert_index_timestamp(td);
+ if (ret < 0)
+ goto error;
}
- trace_ids = g_array_new(FALSE, TRUE, sizeof(int));
-
- while ((node = fts_read(tree))) {
- int dirfd, metafd;
+ 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);
- if (!(node->fts_info & FTS_D))
- continue;
-
- dirfd = open(node->fts_accpath, 0);
- if (dirfd < 0) {
- fprintf(stdout, "[warning] unable to open trace "
- "directory file descriptor.\n");
- continue;
- }
- metafd = openat(dirfd, "metadata", O_RDONLY);
- if (metafd < 0) {
- ret = close(dirfd);
- if (ret < 0) {
- perror("close");
- goto error;
- }
- } else {
- int trace_id;
-
- ret = close(metafd);
- if (ret < 0) {
- perror("close");
- goto error;
- }
- ret = close(dirfd);
- if (ret < 0) {
- perror("close");
- goto error;
- }
-
- trace_id = bt_context_add_trace(ctx,
- node->fts_accpath, format_str);
- if (trace_id < 0) {
- fprintf(stdout, "[warning] CTX opening trace \"%s\"from %s "
- "for reading.\n\n", node->fts_accpath, path);
- continue;
- }
- g_array_append_val(trace_ids, trace_id);
- }
- }
-
- g_array_free(trace_ids, TRUE);
- return 0;
+ return handle->id;
error:
+ closeret = fmt->close_trace(td);
+ if (closeret) {
+ fprintf(stderr, "Error in close_trace callback\n");
+ }
+end:
return ret;
}
-void bt_context_remove_trace(struct bt_context *ctx, int handle_id)
+int bt_context_remove_trace(struct bt_context *ctx, int handle_id)
{
- struct bt_trace_handle *handle;
-
- handle = g_hash_table_lookup(ctx->trace_handles,
- (gpointer) (unsigned long) handle_id);
- assert(handle != NULL);
-
- /* Remove from containers */
- trace_collection_remove(ctx->tc, handle->td);
- g_hash_table_remove(ctx->trace_handles,
- (gpointer) (unsigned long) handle_id);
+ int ret = 0;
- /* Close the trace */
- handle->format->close_trace(handle->td);
+ if (!ctx) {
+ ret = -EINVAL;
+ goto end;
+ }
- /* Destory the handle */
- bt_trace_handle_destroy(handle);
+ /*
+ * Remove the handle. remove_trace_handle will be called
+ * automatically.
+ */
+ if (!g_hash_table_remove(ctx->trace_handles,
+ (gpointer) (unsigned long) handle_id)) {
+ ret = -ENOENT;
+ goto end;
+ }
+end:
+ return ret;
}
static
void bt_context_destroy(struct bt_context *ctx)
{
- finalize_trace_collection(ctx->tc);
+ assert(ctx);
- /* Remote all traces. The g_hash_table_destroy will call
- * bt_trace_handle_destroy on each elements.
+ /*
+ * Remove all traces. The g_hash_table_destroy will call
+ * remove_trace_handle on each element.
*/
g_hash_table_destroy(ctx->trace_handles);
+ bt_finalize_trace_collection(ctx->tc);
+
/* ctx->tc should always be valid */
assert(ctx->tc != NULL);
g_free(ctx->tc);
void bt_context_get(struct bt_context *ctx)
{
+ assert(ctx);
ctx->refcount++;
}
void bt_context_put(struct bt_context *ctx)
{
+ assert(ctx);
ctx->refcount--;
if (ctx->refcount == 0)
bt_context_destroy(ctx);
}
+
+static
+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 */
+ ret = handle->format->close_trace(handle->td);
+ if (ret) {
+ fprintf(stderr, "Error in close_trace callback\n");
+ }
+
+ bt_trace_handle_destroy(handle);
+}