X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=blobdiff_plain;f=lib%2Fcontext.c;h=87901b3c61027aadea02e0ddf0f9aa042b4aaac1;hp=48cf216e11780f99a20b6dea27264368c9c350c3;hb=4a6f6ed3aeb9845dec4b8ec9ca8573ed1c064187;hpb=842c2b97eab577484edae763770dfd1440490818 diff --git a/lib/context.c b/lib/context.c index 48cf216e..87901b3c 100644 --- a/lib/context.c +++ b/lib/context.c @@ -17,61 +17,231 @@ * * 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 #include +#include +#include +#include +#include +#include +#include +#include #include +#include +#include +#include + +#include /* For O_RDONLY */ + +#include -struct bt_context *bt_context_create(struct trace_collection *tc) +static +void remove_trace_handle(struct bt_trace_handle *handle); + +struct bt_context *bt_context_create(void) { struct bt_context *ctx; - ctx = calloc(1, sizeof(struct bt_context)); - if (ctx == NULL) { - perror("allocating context"); - goto error; - } - - ctx->tc = tc; + ctx = g_new0(struct bt_context, 1); ctx->refcount = 1; + /* Negative handle id are errors. */ ctx->last_trace_handle_id = 0; + /* Instanciate the trace handle container */ + ctx->trace_handles = g_hash_table_new_full(g_direct_hash, + g_direct_equal, NULL, + (GDestroyNotify) remove_trace_handle); + + ctx->current_iterator = NULL; + ctx->tc = g_new0(struct trace_collection, 1); + bt_init_trace_collection(ctx->tc); + return ctx; +} + +int bt_context_add_trace(struct bt_context *ctx, const char *path, + 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 bt_trace_descriptor *td; + struct bt_format *fmt; + struct bt_trace_handle *handle; + int ret, closeret; + + 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) { + fprintf(stderr, "[error] [Context] Creating trace handle %s .\n\n", + path); + ret = -1; + goto error_close; + } + handle->format = fmt; + handle->td = td; + if (path) { + strncpy(handle->path, path, PATH_MAX); + 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); -error: - return NULL; + 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); + + return handle->id; + +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"); + } +end: + return ret; } -int bt_context_destroy(struct bt_context *ctx) +int bt_context_remove_trace(struct bt_context *ctx, int handle_id) { - if (ctx) { - if (ctx->refcount >= 1) - goto ctx_used; + int ret = 0; - free(ctx); + if (!ctx) { + ret = -EINVAL; + goto end; } - return 0; -ctx_used: - return -1; + /* + * 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; } -int bt_context_get(struct bt_context *ctx) +static +void bt_context_destroy(struct bt_context *ctx) { - if (!ctx) - return -1; - ctx->refcount++; - return 0; + assert(ctx); + + /* + * 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); + g_free(ctx); } -int bt_context_put(struct bt_context *ctx) +void bt_context_get(struct bt_context *ctx) { - if (!ctx) - return -1; + assert(ctx); + ctx->refcount++; +} +void bt_context_put(struct bt_context *ctx) +{ + assert(ctx); ctx->refcount--; if (ctx->refcount == 0) - return bt_context_destroy(ctx); - return 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); }