[babeltrace.git] / src / lib / trace-ir / trace.c

/*
 * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
 * Copyright 2014 Jérémie Galarneau <jeremie.galarneau@efficios.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * 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.
 */

#define BT_LOG_TAG "LIB/TRACE"
#include "lib/logging.h"

#include "lib/assert-pre.h"
#include "lib/assert-post.h"
#include <babeltrace2/trace-ir/trace.h>
#include <babeltrace2/trace-ir/trace-const.h>
#include <babeltrace2/trace-ir/event-class.h>
#include "ctf-writer/functor.h"
#include "ctf-writer/clock.h"
#include "compat/compiler.h"
#include <babeltrace2/value.h>
#include <babeltrace2/value-const.h>
#include "lib/value.h"
#include <babeltrace2/types.h>
#include "compat/endian.h"
#include "common/assert.h"
#include "compat/glib.h"
#include <inttypes.h>
#include <stdint.h>
#include <string.h>
#include <stdbool.h>
#include <stdlib.h>

#include "attributes.h"
#include "clock-class.h"
#include "event-class.h"
#include "event.h"
#include "field-class.h"
#include "field-wrapper.h"
#include "resolve-field-path.h"
#include "stream-class.h"
#include "stream.h"
#include "trace-class.h"
#include "trace.h"
#include "utils.h"
#include "lib/value.h"
#include "lib/func-status.h"

struct bt_trace_destruction_listener_elem {
	bt_trace_destruction_listener_func func;
	void *data;
};

#define BT_ASSERT_PRE_DEV_TRACE_HOT(_trace) \
	BT_ASSERT_PRE_DEV_HOT((_trace), "Trace", ": %!+t", (_trace))

static
void destroy_trace(struct bt_object *obj)
{
	struct bt_trace *trace = (void *) obj;

	BT_LIB_LOGD("Destroying trace object: %!+t", trace);
	BT_OBJECT_PUT_REF_AND_RESET(trace->user_attributes);

	/*
	 * Call destruction listener functions so that everything else
	 * still exists in the trace.
	 */
	if (trace->destruction_listeners) {
		uint64_t i;
		BT_LIB_LOGD("Calling trace destruction listener(s): %!+t", trace);

		/*
		* The trace's reference count is 0 if we're here. Increment
		* it to avoid a double-destroy (possibly infinitely recursive).
		* This could happen for example if a destruction listener did
		* bt_object_get_ref() (or anything that causes
		* bt_object_get_ref() to be called) on the trace (ref.
		* count goes from 0 to 1), and then bt_object_put_ref(): the
		* reference count would go from 1 to 0 again and this function
		* would be called again.
		*/
		trace->base.ref_count++;

		/* Call all the trace destruction listeners */
		for (i = 0; i < trace->destruction_listeners->len; i++) {
			struct bt_trace_destruction_listener_elem elem =
				g_array_index(trace->destruction_listeners,
						struct bt_trace_destruction_listener_elem, i);

			if (elem.func) {
				elem.func(trace, elem.data);
			}

			/*
			 * The destruction listener should not have kept a
			 * reference to the trace.
			 */
			BT_ASSERT_POST(trace->base.ref_count == 1, "Destruction listener kept a reference to the trace being destroyed: %![trace-]+t", trace);
		}
		g_array_free(trace->destruction_listeners, TRUE);
		trace->destruction_listeners = NULL;
	}

	if (trace->name.str) {
		g_string_free(trace->name.str, TRUE);
		trace->name.str = NULL;
		trace->name.value = NULL;
	}

	if (trace->environment) {
		BT_LOGD_STR("Destroying environment attributes.");
		bt_attributes_destroy(trace->environment);
		trace->environment = NULL;
	}

	if (trace->streams) {
		BT_LOGD_STR("Destroying streams.");
		g_ptr_array_free(trace->streams, TRUE);
		trace->streams = NULL;
	}

	if (trace->stream_classes_stream_count) {
		g_hash_table_destroy(trace->stream_classes_stream_count);
		trace->stream_classes_stream_count = NULL;
	}

	BT_LOGD_STR("Putting trace's class.");
	bt_object_put_ref(trace->class);
	trace->class = NULL;
	g_free(trace);
}

struct bt_trace *bt_trace_create(struct bt_trace_class *tc)
{
	struct bt_trace *trace = NULL;

	BT_LIB_LOGD("Creating trace object: %![tc-]+T", tc);
	trace = g_new0(struct bt_trace, 1);
	if (!trace) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one trace.");
		goto error;
	}

	bt_object_init_shared(&trace->base, destroy_trace);
	trace->user_attributes = bt_value_map_create();
	if (!trace->user_attributes) {
		BT_LIB_LOGE_APPEND_CAUSE(
			"Failed to create a map value object.");
		goto error;
	}

	trace->streams = g_ptr_array_new_with_free_func(
		(GDestroyNotify) bt_object_try_spec_release);
	if (!trace->streams) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GPtrArray.");
		goto error;
	}

	trace->stream_classes_stream_count = g_hash_table_new(g_direct_hash,
		g_direct_equal);
	if (!trace->stream_classes_stream_count) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GHashTable.");
		goto error;
	}

	trace->name.str = g_string_new(NULL);
	if (!trace->name.str) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GString.");
		goto error;
	}

	trace->environment = bt_attributes_create();
	if (!trace->environment) {
		BT_LIB_LOGE_APPEND_CAUSE("Cannot create empty attributes object.");
		goto error;
	}

	trace->destruction_listeners = g_array_new(FALSE, TRUE,
		sizeof(struct bt_trace_destruction_listener_elem));
	if (!trace->destruction_listeners) {
		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GArray.");
		goto error;
	}

	trace->class = tc;
	bt_object_get_ref_no_null_check(trace->class);
	BT_LIB_LOGD("Created trace object: %!+t", trace);
	goto end;

error:
	BT_OBJECT_PUT_REF_AND_RESET(trace);

end:
	return trace;
}

const char *bt_trace_get_name(const struct bt_trace *trace)
{
	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	return trace->name.value;
}

enum bt_trace_set_name_status bt_trace_set_name(struct bt_trace *trace,
		const char *name)
{
	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE_NON_NULL(name, "Name");
	BT_ASSERT_PRE_DEV_TRACE_HOT(trace);
	g_string_assign(trace->name.str, name);
	trace->name.value = trace->name.str->str;
	BT_LIB_LOGD("Set trace's name: %!+t", trace);
	return BT_FUNC_STATUS_OK;
}

bt_uuid bt_trace_get_uuid(const struct bt_trace *trace)
{
	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	return trace->uuid.value;
}

void bt_trace_set_uuid(struct bt_trace *trace, bt_uuid uuid)
{
	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE_NON_NULL(uuid, "UUID");
	BT_ASSERT_PRE_DEV_TRACE_HOT(trace);
	bt_uuid_copy(trace->uuid.uuid, uuid);
	trace->uuid.value = trace->uuid.uuid;
	BT_LIB_LOGD("Set trace's UUID: %!+t", trace);
}

static
bool trace_has_environment_entry(const struct bt_trace *trace, const char *name)
{
	BT_ASSERT(trace);

	return bt_attributes_borrow_field_value_by_name(
		trace->environment, name);
}

static
enum bt_trace_set_environment_entry_status set_environment_entry(
		struct bt_trace *trace,
		const char *name, struct bt_value *value)
{
	int ret;

	BT_ASSERT(trace);
	BT_ASSERT(name);
	BT_ASSERT(value);
	BT_ASSERT_PRE(!trace->frozen ||
		!trace_has_environment_entry(trace, name),
		"Trace is frozen: cannot replace environment entry: "
		"%![trace-]+t, entry-name=\"%s\"", trace, name);
	ret = bt_attributes_set_field_value(trace->environment, name,
		value);
	if (ret) {
		ret = BT_FUNC_STATUS_MEMORY_ERROR;
		BT_LIB_LOGE_APPEND_CAUSE(
			"Cannot set trace's environment entry: "
			"%![trace-]+t, entry-name=\"%s\"", trace, name);
	} else {
		bt_value_freeze(value);
		BT_LIB_LOGD("Set trace's environment entry: "
			"%![trace-]+t, entry-name=\"%s\"", trace, name);
	}

	return ret;
}

enum bt_trace_set_environment_entry_status
bt_trace_set_environment_entry_string(
		struct bt_trace *trace, const char *name, const char *value)
{
	int ret;
	struct bt_value *value_obj;
	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE_NON_NULL(name, "Name");
	BT_ASSERT_PRE_NON_NULL(value, "Value");
	value_obj = bt_value_string_create_init(value);
	if (!value_obj) {
		BT_LIB_LOGE_APPEND_CAUSE(
			"Cannot create a string value object.");
		ret = -1;
		goto end;
	}

	/* set_environment_entry() logs errors */
	ret = set_environment_entry(trace, name, value_obj);

end:
	bt_object_put_ref(value_obj);
	return ret;
}

enum bt_trace_set_environment_entry_status
bt_trace_set_environment_entry_integer(
		struct bt_trace *trace, const char *name, int64_t value)
{
	int ret;
	struct bt_value *value_obj;
	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE_NON_NULL(name, "Name");
	value_obj = bt_value_integer_signed_create_init(value);
	if (!value_obj) {
		BT_LIB_LOGE_APPEND_CAUSE(
			"Cannot create an integer value object.");
		ret = BT_FUNC_STATUS_MEMORY_ERROR;
		goto end;
	}

	/* set_environment_entry() logs errors */
	ret = set_environment_entry(trace, name, value_obj);

end:
	bt_object_put_ref(value_obj);
	return ret;
}

uint64_t bt_trace_get_environment_entry_count(const struct bt_trace *trace)
{
	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	return bt_attributes_get_count(trace->environment);
}

void bt_trace_borrow_environment_entry_by_index_const(
		const struct bt_trace *trace, uint64_t index,
		const char **name, const struct bt_value **value)
{
	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE_DEV_NON_NULL(name, "Name");
	BT_ASSERT_PRE_DEV_NON_NULL(value, "Value");
	BT_ASSERT_PRE_DEV_VALID_INDEX(index,
		bt_attributes_get_count(trace->environment));
	*value = bt_attributes_borrow_field_value(trace->environment, index);
	BT_ASSERT(*value);
	*name = bt_attributes_get_field_name(trace->environment, index);
	BT_ASSERT(*name);
}

const struct bt_value *bt_trace_borrow_environment_entry_value_by_name_const(
		const struct bt_trace *trace, const char *name)
{
	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE_DEV_NON_NULL(name, "Name");
	return bt_attributes_borrow_field_value_by_name(trace->environment,
		name);
}

uint64_t bt_trace_get_stream_count(const struct bt_trace *trace)
{
	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	return (uint64_t) trace->streams->len;
}

struct bt_stream *bt_trace_borrow_stream_by_index(
		struct bt_trace *trace, uint64_t index)
{
	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE_DEV_VALID_INDEX(index, trace->streams->len);
	return g_ptr_array_index(trace->streams, index);
}

const struct bt_stream *bt_trace_borrow_stream_by_index_const(
		const struct bt_trace *trace, uint64_t index)
{
	return bt_trace_borrow_stream_by_index((void *) trace, index);
}

struct bt_stream *bt_trace_borrow_stream_by_id(struct bt_trace *trace,
		uint64_t id)
{
	struct bt_stream *stream = NULL;
	uint64_t i;

	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");

	for (i = 0; i < trace->streams->len; i++) {
		struct bt_stream *stream_candidate =
			g_ptr_array_index(trace->streams, i);

		if (stream_candidate->id == id) {
			stream = stream_candidate;
			goto end;
		}
	}

end:
	return stream;
}

const struct bt_stream *bt_trace_borrow_stream_by_id_const(
		const struct bt_trace *trace, uint64_t id)
{
	return bt_trace_borrow_stream_by_id((void *) trace, id);
}

enum bt_trace_add_listener_status bt_trace_add_destruction_listener(
		const struct bt_trace *c_trace,
		bt_trace_destruction_listener_func listener,
		void *data, bt_listener_id *listener_id)
{
	struct bt_trace *trace = (void *) c_trace;
	uint64_t i;
	struct bt_trace_destruction_listener_elem new_elem = {
		.func = listener,
		.data = data,
	};

	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE_NON_NULL(listener, "Listener");

	/* Find the next available spot */
	for (i = 0; i < trace->destruction_listeners->len; i++) {
		struct bt_trace_destruction_listener_elem elem =
			g_array_index(trace->destruction_listeners,
				struct bt_trace_destruction_listener_elem, i);

		if (!elem.func) {
			break;
		}
	}

	if (i == trace->destruction_listeners->len) {
		g_array_append_val(trace->destruction_listeners, new_elem);
	} else {
		g_array_insert_val(trace->destruction_listeners, i, new_elem);
	}

	if (listener_id) {
		*listener_id = i;
	}

	BT_LIB_LOGD("Added destruction listener: " "%![trace-]+t, "
			"listener-id=%" PRIu64, trace, i);
	return BT_FUNC_STATUS_OK;
}

static
bool has_listener_id(const struct bt_trace *trace, uint64_t listener_id)
{
	BT_ASSERT(listener_id < trace->destruction_listeners->len);
	return (&g_array_index(trace->destruction_listeners,
			struct bt_trace_destruction_listener_elem,
			listener_id))->func;
}

enum bt_trace_remove_listener_status bt_trace_remove_destruction_listener(
		const struct bt_trace *c_trace, bt_listener_id listener_id)
{
	struct bt_trace *trace = (void *) c_trace;
	struct bt_trace_destruction_listener_elem *elem;

	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE(has_listener_id(trace, listener_id),
		"Trace has no such trace destruction listener ID: "
		"%![trace-]+t, %" PRIu64, trace, listener_id);
	elem = &g_array_index(trace->destruction_listeners,
			struct bt_trace_destruction_listener_elem,
			listener_id);
	BT_ASSERT(elem->func);

	elem->func = NULL;
	elem->data = NULL;
	BT_LIB_LOGD("Removed \"trace destruction listener: "
		"%![trace-]+t, listener-id=%" PRIu64,
		trace, listener_id);
	return BT_FUNC_STATUS_OK;
}

BT_HIDDEN
void _bt_trace_freeze(const struct bt_trace *trace)
{
	BT_ASSERT(trace);
	BT_LIB_LOGD("Freezing trace's class: %!+T", trace->class);
	bt_trace_class_freeze(trace->class);
	BT_LIB_LOGD("Freezing trace's user attributes: %!+v",
		trace->user_attributes);
	bt_value_freeze(trace->user_attributes);
	BT_LIB_LOGD("Freezing trace: %!+t", trace);
	((struct bt_trace *) trace)->frozen = true;
}

BT_HIDDEN
void bt_trace_add_stream(struct bt_trace *trace, struct bt_stream *stream)
{
	guint count = 0;

	bt_object_set_parent(&stream->base, &trace->base);
	g_ptr_array_add(trace->streams, stream);
	bt_trace_freeze(trace);

	if (bt_g_hash_table_contains(trace->stream_classes_stream_count,
			stream->class)) {
		count = GPOINTER_TO_UINT(g_hash_table_lookup(
			trace->stream_classes_stream_count, stream->class));
	}

	g_hash_table_insert(trace->stream_classes_stream_count,
		stream->class, GUINT_TO_POINTER(count + 1));
}

BT_HIDDEN
uint64_t bt_trace_get_automatic_stream_id(const struct bt_trace *trace,
		const struct bt_stream_class *stream_class)
{
	gpointer orig_key;
	gpointer value;
	uint64_t id = 0;

	BT_ASSERT(stream_class);
	BT_ASSERT(trace);
	if (g_hash_table_lookup_extended(trace->stream_classes_stream_count,
			stream_class, &orig_key, &value)) {
		id = (uint64_t) GPOINTER_TO_UINT(value);
	}

	return id;
}

struct bt_trace_class *bt_trace_borrow_class(struct bt_trace *trace)
{
	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	return trace->class;
}

const struct bt_trace_class *bt_trace_borrow_class_const(
		const struct bt_trace *trace)
{
	return bt_trace_borrow_class((void *) trace);
}

const struct bt_value *bt_trace_borrow_user_attributes_const(
		const struct bt_trace *trace)
{
	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	return trace->user_attributes;
}

struct bt_value *bt_trace_borrow_user_attributes(struct bt_trace *trace)
{
	return (void *) bt_trace_borrow_user_attributes_const((void *) trace);
}

void bt_trace_set_user_attributes(
		struct bt_trace *trace,
		const struct bt_value *user_attributes)
{
	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	BT_ASSERT_PRE_NON_NULL(user_attributes, "User attributes");
	BT_ASSERT_PRE(user_attributes->type == BT_VALUE_TYPE_MAP,
		"User attributes object is not a map value object.");
	BT_ASSERT_PRE_DEV_TRACE_HOT(trace);
	bt_object_put_ref_no_null_check(trace->user_attributes);
	trace->user_attributes = (void *) user_attributes;
	bt_object_get_ref_no_null_check(trace->user_attributes);
}

void bt_trace_get_ref(const struct bt_trace *trace)
{
	bt_object_get_ref(trace);
}

void bt_trace_put_ref(const struct bt_trace *trace)
{
	bt_object_put_ref(trace);
}
Commit	Line	Data
bc37ae52	1	/*
e2f7325d	2	* Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
bc37ae52 JG	3	* Copyright 2014 Jérémie Galarneau <jeremie.galarneau@efficios.com>
bc37ae52 JG	4	*
bc37ae52 JG	5	* Permission is hereby granted, free of charge, to any person obtaining a copy
	6	* of this software and associated documentation files (the "Software"), to deal
	7	* in the Software without restriction, including without limitation the rights
	8	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	9	* copies of the Software, and to permit persons to whom the Software is
	10	* furnished to do so, subject to the following conditions:
	11	*
	12	* The above copyright notice and this permission notice shall be included in
	13	* all copies or substantial portions of the Software.
	14	*
	15	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	16	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	17	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
	18	* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	19	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	20	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
	21	* SOFTWARE.
	22	*/
	23
350ad6c1	24	#define BT_LOG_TAG "LIB/TRACE"
c2d9d9cf	25	#include "lib/logging.h"
e011d2c1	26
578e048b	27	#include "lib/assert-pre.h"
3c634c85	28	#include "lib/assert-post.h"
3fadfbc0	29	#include <babeltrace2/trace-ir/trace.h>
3fadfbc0	30	#include <babeltrace2/trace-ir/trace-const.h>
3fadfbc0	31	#include <babeltrace2/trace-ir/event-class.h>
578e048b MJ	32	#include "ctf-writer/functor.h"
	33	#include "ctf-writer/clock.h"
	34	#include "compat/compiler.h"
3fadfbc0 MJ	35	#include <babeltrace2/value.h>
3fadfbc0 MJ	36	#include <babeltrace2/value-const.h>
578e048b	37	#include "lib/value.h"
3fadfbc0	38	#include <babeltrace2/types.h>
578e048b MJ	39	#include "compat/endian.h"
	40	#include "common/assert.h"
	41	#include "compat/glib.h"
dc3fffef	42	#include <inttypes.h>
544d0515	43	#include <stdint.h>
4a32fda0	44	#include <string.h>
c4f23e30	45	#include <stdbool.h>
0fbb9a9f	46	#include <stdlib.h>
bc37ae52	47
578e048b MJ	48	#include "attributes.h"
	49	#include "clock-class.h"
	50	#include "event-class.h"
	51	#include "event.h"
	52	#include "field-class.h"
	53	#include "field-wrapper.h"
	54	#include "resolve-field-path.h"
	55	#include "stream-class.h"
	56	#include "stream.h"
	57	#include "trace-class.h"
	58	#include "trace.h"
	59	#include "utils.h"
c6962c96	60	#include "lib/value.h"
d24d5663	61	#include "lib/func-status.h"
578e048b	62
ad5268b5 FD	63	struct bt_trace_destruction_listener_elem {
ad5268b5 FD	64	bt_trace_destruction_listener_func func;
3602afb0 PP	65	void *data;
	66	};
	67
bdb288b3 PP	68	#define BT_ASSERT_PRE_DEV_TRACE_HOT(_trace) \
bdb288b3 PP	69	BT_ASSERT_PRE_DEV_HOT((_trace), "Trace", ": %!+t", (_trace))
cb6f1f7d	70
bc37ae52	71	static
44c440bc	72	void destroy_trace(struct bt_object *obj)
3dca2276 PP	73	{
3dca2276 PP	74	struct bt_trace trace = (void ) obj;
bc37ae52	75
44c440bc	76	BT_LIB_LOGD("Destroying trace object: %!+t", trace);
c6962c96	77	BT_OBJECT_PUT_REF_AND_RESET(trace->user_attributes);
bc37ae52	78
3dca2276	79	/*
ad5268b5 FD	80	* Call destruction listener functions so that everything else
ad5268b5 FD	81	* still exists in the trace.
3dca2276	82	*/
ad5268b5 FD	83	if (trace->destruction_listeners) {
ad5268b5 FD	84	uint64_t i;
3f7d4d90	85	BT_LIB_LOGD("Calling trace destruction listener(s): %!+t", trace);
c47a6bec SM	86
	87	/*
	88	* The trace's reference count is 0 if we're here. Increment
	89	* it to avoid a double-destroy (possibly infinitely recursive).
	90	* This could happen for example if a destruction listener did
	91	* bt_object_get_ref() (or anything that causes
	92	* bt_object_get_ref() to be called) on the trace (ref.
	93	* count goes from 0 to 1), and then bt_object_put_ref(): the
	94	* reference count would go from 1 to 0 again and this function
	95	* would be called again.
	96	*/
	97	trace->base.ref_count++;
	98
ad5268b5 FD	99	/* Call all the trace destruction listeners */
	100	for (i = 0; i < trace->destruction_listeners->len; i++) {
	101	struct bt_trace_destruction_listener_elem elem =
	102	g_array_index(trace->destruction_listeners,
	103	struct bt_trace_destruction_listener_elem, i);
	104
	105	if (elem.func) {
	106	elem.func(trace, elem.data);
3dca2276	107	}
c47a6bec SM	108
	109	/*
	110	* The destruction listener should not have kept a
	111	* reference to the trace.
	112	*/
3c634c85	113	BT_ASSERT_POST(trace->base.ref_count == 1, "Destruction listener kept a reference to the trace being destroyed: %![trace-]+t", trace);
3dca2276	114	}
ad5268b5 FD	115	g_array_free(trace->destruction_listeners, TRUE);
ad5268b5 FD	116	trace->destruction_listeners = NULL;
bc37ae52 JG	117	}
bc37ae52 JG	118
44c440bc PP	119	if (trace->name.str) {
44c440bc PP	120	g_string_free(trace->name.str, TRUE);
238b7404 PP	121	trace->name.str = NULL;
238b7404 PP	122	trace->name.value = NULL;
95c09b3a PP	123	}
95c09b3a PP	124
335a2da5 PP	125	if (trace->environment) {
	126	BT_LOGD_STR("Destroying environment attributes.");
	127	bt_attributes_destroy(trace->environment);
	128	trace->environment = NULL;
	129	}
	130
44c440bc PP	131	if (trace->streams) {
	132	BT_LOGD_STR("Destroying streams.");
	133	g_ptr_array_free(trace->streams, TRUE);
238b7404	134	trace->streams = NULL;
bc37ae52 JG	135	}
bc37ae52 JG	136
44c440bc PP	137	if (trace->stream_classes_stream_count) {
44c440bc PP	138	g_hash_table_destroy(trace->stream_classes_stream_count);
238b7404	139	trace->stream_classes_stream_count = NULL;
44c440bc	140	}
3dca2276	141
862ca4ed PP	142	BT_LOGD_STR("Putting trace's class.");
	143	bt_object_put_ref(trace->class);
	144	trace->class = NULL;
44c440bc	145	g_free(trace);
3dca2276 PP	146	}
3dca2276 PP	147
862ca4ed	148	struct bt_trace bt_trace_create(struct bt_trace_class tc)
3dca2276 PP	149	{
3dca2276 PP	150	struct bt_trace *trace = NULL;
3dca2276	151
862ca4ed	152	BT_LIB_LOGD("Creating trace object: %![tc-]+T", tc);
3dca2276 PP	153	trace = g_new0(struct bt_trace, 1);
3dca2276 PP	154	if (!trace) {
870631a2	155	BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one trace.");
3dca2276 PP	156	goto error;
	157	}
	158
862ca4ed	159	bt_object_init_shared(&trace->base, destroy_trace);
c6962c96 PP	160	trace->user_attributes = bt_value_map_create();
	161	if (!trace->user_attributes) {
	162	BT_LIB_LOGE_APPEND_CAUSE(
	163	"Failed to create a map value object.");
	164	goto error;
	165	}
	166
44c440bc PP	167	trace->streams = g_ptr_array_new_with_free_func(
	168	(GDestroyNotify) bt_object_try_spec_release);
	169	if (!trace->streams) {
870631a2	170	BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GPtrArray.");
3dca2276 PP	171	goto error;
	172	}
	173
44c440bc PP	174	trace->stream_classes_stream_count = g_hash_table_new(g_direct_hash,
	175	g_direct_equal);
	176	if (!trace->stream_classes_stream_count) {
870631a2	177	BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GHashTable.");
44c440bc PP	178	goto error;
	179	}
	180
	181	trace->name.str = g_string_new(NULL);
	182	if (!trace->name.str) {
870631a2	183	BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GString.");
44c440bc PP	184	goto error;
	185	}
	186
335a2da5 PP	187	trace->environment = bt_attributes_create();
	188	if (!trace->environment) {
	189	BT_LIB_LOGE_APPEND_CAUSE("Cannot create empty attributes object.");
	190	goto error;
	191	}
	192
ad5268b5 FD	193	trace->destruction_listeners = g_array_new(FALSE, TRUE,
	194	sizeof(struct bt_trace_destruction_listener_elem));
	195	if (!trace->destruction_listeners) {
870631a2	196	BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate one GArray.");
3602afb0 PP	197	goto error;
	198	}
	199
862ca4ed	200	trace->class = tc;
6871026b	201	bt_object_get_ref_no_null_check(trace->class);
44c440bc PP	202	BT_LIB_LOGD("Created trace object: %!+t", trace);
44c440bc PP	203	goto end;
bc37ae52	204
bc37ae52	205	error:
65300d60	206	BT_OBJECT_PUT_REF_AND_RESET(trace);
44c440bc PP	207
44c440bc PP	208	end:
862ca4ed	209	return trace;
bc37ae52 JG	210	}
bc37ae52 JG	211
40f4ba76	212	const char bt_trace_get_name(const struct bt_trace trace)
e96045d4	213	{
bdb288b3	214	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
44c440bc	215	return trace->name.value;
e96045d4 JG	216	}
e96045d4 JG	217
d24d5663 PP	218	enum bt_trace_set_name_status bt_trace_set_name(struct bt_trace *trace,
d24d5663 PP	219	const char *name)
e96045d4	220	{
44c440bc PP	221	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
44c440bc PP	222	BT_ASSERT_PRE_NON_NULL(name, "Name");
bdb288b3	223	BT_ASSERT_PRE_DEV_TRACE_HOT(trace);
44c440bc PP	224	g_string_assign(trace->name.str, name);
44c440bc PP	225	trace->name.value = trace->name.str->str;
3f7d4d90	226	BT_LIB_LOGD("Set trace's name: %!+t", trace);
d24d5663	227	return BT_FUNC_STATUS_OK;
e96045d4 JG	228	}
e96045d4 JG	229
335a2da5 PP	230	bt_uuid bt_trace_get_uuid(const struct bt_trace *trace)
335a2da5 PP	231	{
bdb288b3	232	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
335a2da5 PP	233	return trace->uuid.value;
	234	}
	235
	236	void bt_trace_set_uuid(struct bt_trace *trace, bt_uuid uuid)
	237	{
	238	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	239	BT_ASSERT_PRE_NON_NULL(uuid, "UUID");
bdb288b3	240	BT_ASSERT_PRE_DEV_TRACE_HOT(trace);
6162e6b7	241	bt_uuid_copy(trace->uuid.uuid, uuid);
335a2da5 PP	242	trace->uuid.value = trace->uuid.uuid;
	243	BT_LIB_LOGD("Set trace's UUID: %!+t", trace);
	244	}
	245
335a2da5 PP	246	static
	247	bool trace_has_environment_entry(const struct bt_trace trace, const char name)
	248	{
	249	BT_ASSERT(trace);
	250
	251	return bt_attributes_borrow_field_value_by_name(
5084732e	252	trace->environment, name);
335a2da5 PP	253	}
	254
	255	static
	256	enum bt_trace_set_environment_entry_status set_environment_entry(
	257	struct bt_trace *trace,
	258	const char name, struct bt_value value)
	259	{
	260	int ret;
	261
	262	BT_ASSERT(trace);
	263	BT_ASSERT(name);
	264	BT_ASSERT(value);
	265	BT_ASSERT_PRE(!trace->frozen \|\|
	266	!trace_has_environment_entry(trace, name),
	267	"Trace is frozen: cannot replace environment entry: "
	268	"%![trace-]+t, entry-name=\"%s\"", trace, name);
	269	ret = bt_attributes_set_field_value(trace->environment, name,
	270	value);
	271	if (ret) {
	272	ret = BT_FUNC_STATUS_MEMORY_ERROR;
	273	BT_LIB_LOGE_APPEND_CAUSE(
	274	"Cannot set trace's environment entry: "
	275	"%![trace-]+t, entry-name=\"%s\"", trace, name);
	276	} else {
	277	bt_value_freeze(value);
	278	BT_LIB_LOGD("Set trace's environment entry: "
	279	"%![trace-]+t, entry-name=\"%s\"", trace, name);
	280	}
	281
	282	return ret;
	283	}
	284
	285	enum bt_trace_set_environment_entry_status
	286	bt_trace_set_environment_entry_string(
	287	struct bt_trace trace, const char name, const char *value)
	288	{
	289	int ret;
	290	struct bt_value *value_obj;
	291	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	292	BT_ASSERT_PRE_NON_NULL(name, "Name");
	293	BT_ASSERT_PRE_NON_NULL(value, "Value");
	294	value_obj = bt_value_string_create_init(value);
	295	if (!value_obj) {
	296	BT_LIB_LOGE_APPEND_CAUSE(
	297	"Cannot create a string value object.");
	298	ret = -1;
	299	goto end;
	300	}
	301
	302	/* set_environment_entry() logs errors */
	303	ret = set_environment_entry(trace, name, value_obj);
	304
	305	end:
	306	bt_object_put_ref(value_obj);
	307	return ret;
	308	}
	309
	310	enum bt_trace_set_environment_entry_status
	311	bt_trace_set_environment_entry_integer(
	312	struct bt_trace trace, const char name, int64_t value)
	313	{
	314	int ret;
	315	struct bt_value *value_obj;
	316	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
317	BT_ASSERT_PRE_NON_NULL(name, "Name");
9c08c816	318	value_obj = bt_value_integer_signed_create_init(value);
335a2da5 PP	319	if (!value_obj) {
	320	BT_LIB_LOGE_APPEND_CAUSE(
	321	"Cannot create an integer value object.");
	322	ret = BT_FUNC_STATUS_MEMORY_ERROR;
	323	goto end;
	324	}
	325
	326	/* set_environment_entry() logs errors */
	327	ret = set_environment_entry(trace, name, value_obj);
	328
	329	end:
	330	bt_object_put_ref(value_obj);
	331	return ret;
	332	}
	333
	334	uint64_t bt_trace_get_environment_entry_count(const struct bt_trace *trace)
	335	{
bdb288b3	336	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
99b4b64b	337	return bt_attributes_get_count(trace->environment);
335a2da5 PP	338	}
	339
	340	void bt_trace_borrow_environment_entry_by_index_const(
	341	const struct bt_trace *trace, uint64_t index,
	342	const char name, const struct bt_value value)
	343	{
bdb288b3 PP	344	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	345	BT_ASSERT_PRE_DEV_NON_NULL(name, "Name");
	346	BT_ASSERT_PRE_DEV_NON_NULL(value, "Value");
	347	BT_ASSERT_PRE_DEV_VALID_INDEX(index,
335a2da5 PP	348	bt_attributes_get_count(trace->environment));
	349	*value = bt_attributes_borrow_field_value(trace->environment, index);
	350	BT_ASSERT(*value);
	351	*name = bt_attributes_get_field_name(trace->environment, index);
	352	BT_ASSERT(*name);
	353	}
	354
	355	const struct bt_value *bt_trace_borrow_environment_entry_value_by_name_const(
	356	const struct bt_trace trace, const char name)
	357	{
bdb288b3 PP	358	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
bdb288b3 PP	359	BT_ASSERT_PRE_DEV_NON_NULL(name, "Name");
335a2da5 PP	360	return bt_attributes_borrow_field_value_by_name(trace->environment,
	361	name);
	362	}
	363
40f4ba76	364	uint64_t bt_trace_get_stream_count(const struct bt_trace *trace)
bc37ae52	365	{
bdb288b3	366	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
44c440bc PP	367	return (uint64_t) trace->streams->len;
44c440bc PP	368	}
95c09b3a	369
44c440bc PP	370	struct bt_stream *bt_trace_borrow_stream_by_index(
	371	struct bt_trace *trace, uint64_t index)
	372	{
bdb288b3 PP	373	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
bdb288b3 PP	374	BT_ASSERT_PRE_DEV_VALID_INDEX(index, trace->streams->len);
44c440bc PP	375	return g_ptr_array_index(trace->streams, index);
44c440bc PP	376	}
cb6f1f7d	377
40f4ba76 PP	378	const struct bt_stream *bt_trace_borrow_stream_by_index_const(
40f4ba76 PP	379	const struct bt_trace *trace, uint64_t index)
e5be10ef	380	{
40f4ba76	381	return bt_trace_borrow_stream_by_index((void *) trace, index);
e5be10ef PP	382	}
e5be10ef PP	383
40f4ba76 PP	384	struct bt_stream bt_trace_borrow_stream_by_id(struct bt_trace trace,
40f4ba76 PP	385	uint64_t id)
44c440bc PP	386	{
	387	struct bt_stream *stream = NULL;
	388	uint64_t i;
bc37ae52	389
bdb288b3	390	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
bc37ae52	391
44c440bc PP	392	for (i = 0; i < trace->streams->len; i++) {
	393	struct bt_stream *stream_candidate =
	394	g_ptr_array_index(trace->streams, i);
cfeb617e	395
44c440bc PP	396	if (stream_candidate->id == id) {
	397	stream = stream_candidate;
	398	goto end;
	399	}
6474e016	400	}
95c09b3a	401
bc37ae52	402	end:
44c440bc	403	return stream;
bc37ae52 JG	404	}
bc37ae52 JG	405
40f4ba76 PP	406	const struct bt_stream *bt_trace_borrow_stream_by_id_const(
40f4ba76 PP	407	const struct bt_trace *trace, uint64_t id)
e5be10ef	408	{
40f4ba76	409	return bt_trace_borrow_stream_by_id((void *) trace, id);
e5be10ef PP	410	}
e5be10ef PP	411
d24d5663	412	enum bt_trace_add_listener_status bt_trace_add_destruction_listener(
40f4ba76	413	const struct bt_trace *c_trace,
ad5268b5	414	bt_trace_destruction_listener_func listener,
2054a0d1	415	void data, bt_listener_id listener_id)
3602afb0	416	{
40f4ba76	417	struct bt_trace trace = (void ) c_trace;
44c440bc	418	uint64_t i;
ad5268b5	419	struct bt_trace_destruction_listener_elem new_elem = {
3602afb0 PP	420	.func = listener,
	421	.data = data,
	422	};
	423
44c440bc PP	424	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
44c440bc PP	425	BT_ASSERT_PRE_NON_NULL(listener, "Listener");
8480c8cc	426
3602afb0	427	/* Find the next available spot */
ad5268b5 FD	428	for (i = 0; i < trace->destruction_listeners->len; i++) {
	429	struct bt_trace_destruction_listener_elem elem =
	430	g_array_index(trace->destruction_listeners,
	431	struct bt_trace_destruction_listener_elem, i);
3602afb0 PP	432
	433	if (!elem.func) {
	434	break;
	435	}
	436	}
	437
ad5268b5 FD	438	if (i == trace->destruction_listeners->len) {
ad5268b5 FD	439	g_array_append_val(trace->destruction_listeners, new_elem);
3602afb0	440	} else {
ad5268b5	441	g_array_insert_val(trace->destruction_listeners, i, new_elem);
3602afb0 PP	442	}
3602afb0 PP	443
44c440bc PP	444	if (listener_id) {
	445	*listener_id = i;
	446	}
3602afb0	447
3f7d4d90	448	BT_LIB_LOGD("Added destruction listener: " "%![trace-]+t, "
ad5268b5	449	"listener-id=%" PRIu64, trace, i);
d24d5663	450	return BT_FUNC_STATUS_OK;
44c440bc PP	451	}
44c440bc PP	452
44c440bc	453	static
40f4ba76	454	bool has_listener_id(const struct bt_trace *trace, uint64_t listener_id)
44c440bc	455	{
ad5268b5 FD	456	BT_ASSERT(listener_id < trace->destruction_listeners->len);
	457	return (&g_array_index(trace->destruction_listeners,
	458	struct bt_trace_destruction_listener_elem,
5084732e	459	listener_id))->func;
3602afb0 PP	460	}
3602afb0 PP	461
d24d5663	462	enum bt_trace_remove_listener_status bt_trace_remove_destruction_listener(
2054a0d1	463	const struct bt_trace *c_trace, bt_listener_id listener_id)
3602afb0	464	{
40f4ba76	465	struct bt_trace trace = (void ) c_trace;
ad5268b5	466	struct bt_trace_destruction_listener_elem *elem;
3602afb0	467
44c440bc	468	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
44c440bc	469	BT_ASSERT_PRE(has_listener_id(trace, listener_id),
ad5268b5	470	"Trace has no such trace destruction listener ID: "
44c440bc	471	"%![trace-]+t, %" PRIu64, trace, listener_id);
ad5268b5 FD	472	elem = &g_array_index(trace->destruction_listeners,
ad5268b5 FD	473	struct bt_trace_destruction_listener_elem,
3602afb0	474	listener_id);
44c440bc	475	BT_ASSERT(elem->func);
3602afb0 PP	476
	477	elem->func = NULL;
	478	elem->data = NULL;
3f7d4d90	479	BT_LIB_LOGD("Removed \"trace destruction listener: "
44c440bc PP	480	"%![trace-]+t, listener-id=%" PRIu64,
44c440bc PP	481	trace, listener_id);
d24d5663	482	return BT_FUNC_STATUS_OK;
44c440bc	483	}
3602afb0	484
44c440bc	485	BT_HIDDEN
40f4ba76	486	void _bt_trace_freeze(const struct bt_trace *trace)
44c440bc	487	{
44c440bc	488	BT_ASSERT(trace);
862ca4ed PP	489	BT_LIB_LOGD("Freezing trace's class: %!+T", trace->class);
862ca4ed PP	490	bt_trace_class_freeze(trace->class);
c6962c96 PP	491	BT_LIB_LOGD("Freezing trace's user attributes: %!+v",
	492	trace->user_attributes);
	493	bt_value_freeze(trace->user_attributes);
44c440bc	494	BT_LIB_LOGD("Freezing trace: %!+t", trace);
40f4ba76	495	((struct bt_trace *) trace)->frozen = true;
5acf2ae6	496	}
312c056a	497
44c440bc PP	498	BT_HIDDEN
	499	void bt_trace_add_stream(struct bt_trace trace, struct bt_stream stream)
	500	{
	501	guint count = 0;
	502
	503	bt_object_set_parent(&stream->base, &trace->base);
	504	g_ptr_array_add(trace->streams, stream);
cb6f1f7d	505	bt_trace_freeze(trace);
312c056a	506
44c440bc PP	507	if (bt_g_hash_table_contains(trace->stream_classes_stream_count,
	508	stream->class)) {
	509	count = GPOINTER_TO_UINT(g_hash_table_lookup(
	510	trace->stream_classes_stream_count, stream->class));
312c056a PP	511	}
312c056a PP	512
44c440bc PP	513	g_hash_table_insert(trace->stream_classes_stream_count,
	514	stream->class, GUINT_TO_POINTER(count + 1));
	515	}
	516
	517	BT_HIDDEN
40f4ba76 PP	518	uint64_t bt_trace_get_automatic_stream_id(const struct bt_trace *trace,
40f4ba76 PP	519	const struct bt_stream_class *stream_class)
44c440bc PP	520	{
	521	gpointer orig_key;
	522	gpointer value;
	523	uint64_t id = 0;
	524
	525	BT_ASSERT(stream_class);
	526	BT_ASSERT(trace);
	527	if (g_hash_table_lookup_extended(trace->stream_classes_stream_count,
	528	stream_class, &orig_key, &value)) {
	529	id = (uint64_t) GPOINTER_TO_UINT(value);
	530	}
	531
	532	return id;
312c056a	533	}
862ca4ed PP	534
	535	struct bt_trace_class bt_trace_borrow_class(struct bt_trace trace)
	536	{
bdb288b3	537	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
862ca4ed PP	538	return trace->class;
	539	}
	540
	541	const struct bt_trace_class *bt_trace_borrow_class_const(
	542	const struct bt_trace *trace)
	543	{
	544	return bt_trace_borrow_class((void *) trace);
	545	}
c5b9b441	546
c6962c96 PP	547	const struct bt_value *bt_trace_borrow_user_attributes_const(
	548	const struct bt_trace *trace)
	549	{
	550	BT_ASSERT_PRE_DEV_NON_NULL(trace, "Trace");
	551	return trace->user_attributes;
	552	}
	553
	554	struct bt_value bt_trace_borrow_user_attributes(struct bt_trace trace)
	555	{
	556	return (void ) bt_trace_borrow_user_attributes_const((void ) trace);
	557	}
	558
	559	void bt_trace_set_user_attributes(
	560	struct bt_trace *trace,
	561	const struct bt_value *user_attributes)
	562	{
	563	BT_ASSERT_PRE_NON_NULL(trace, "Trace");
	564	BT_ASSERT_PRE_NON_NULL(user_attributes, "User attributes");
	565	BT_ASSERT_PRE(user_attributes->type == BT_VALUE_TYPE_MAP,
	566	"User attributes object is not a map value object.");
	567	BT_ASSERT_PRE_DEV_TRACE_HOT(trace);
6871026b	568	bt_object_put_ref_no_null_check(trace->user_attributes);
c6962c96	569	trace->user_attributes = (void *) user_attributes;
6871026b	570	bt_object_get_ref_no_null_check(trace->user_attributes);
c6962c96 PP	571	}
c6962c96 PP	572
c5b9b441 PP	573	void bt_trace_get_ref(const struct bt_trace *trace)
	574	{
	575	bt_object_get_ref(trace);
	576	}
	577
	578	void bt_trace_put_ref(const struct bt_trace *trace)
	579	{
	580	bt_object_put_ref(trace);
	581	}