X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=src%2Flib%2Ftrace-ir%2Fevent-class.c;h=464dd00f3d26416da025407b497c929e5d6b8328;hb=6871026b82224d83bb63cbb44cc33c16c766d96d;hp=c5e5230e160d049b46f80da2445716de274762bb;hpb=0de083a28d550811823a5f61d9ed1989d785f48d;p=babeltrace.git

diff --git a/src/lib/trace-ir/event-class.c b/src/lib/trace-ir/event-class.c
index c5e5230e..464dd00f 100644
--- a/src/lib/trace-ir/event-class.c
+++ b/src/lib/trace-ir/event-class.c
@@ -59,6 +59,7 @@ void destroy_event_class(struct bt_object *obj)
 	struct bt_event_class *event_class = (void *) obj;
 
 	BT_LIB_LOGD("Destroying event class: %!+E", event_class);
+	BT_OBJECT_PUT_REF_AND_RESET(event_class->user_attributes);
 
 	if (event_class->name.str) {
 		g_string_free(event_class->name.str, TRUE);
@@ -127,21 +128,26 @@ struct bt_event_class *create_event_class_with_id(
 
 	bt_object_init_shared_with_parent(&event_class->base,
 		destroy_event_class);
+	event_class->user_attributes = bt_value_map_create();
+	if (!event_class->user_attributes) {
+		BT_LIB_LOGE_APPEND_CAUSE(
+			"Failed to create a map value object.");
+		goto error;
+	}
+
 	event_class->id = id;
 	bt_property_uint_init(&event_class->log_level,
 			BT_PROPERTY_AVAILABILITY_NOT_AVAILABLE, 0);
 	event_class->name.str = g_string_new(NULL);
 	if (!event_class->name.str) {
 		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
-		ret = -1;
-		goto end;
+		goto error;
 	}
 
 	event_class->emf_uri.str = g_string_new(NULL);
 	if (!event_class->emf_uri.str) {
 		BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
-		ret = -1;
-		goto end;
+		goto error;
 	}
 
 	ret = bt_object_pool_initialize(&event_class->event_pool,
@@ -324,7 +330,7 @@ bt_event_class_set_specific_context_field_class(
 	bt_field_class_make_part_of_trace_class(field_class);
 	bt_object_put_ref(event_class->specific_context_fc);
 	event_class->specific_context_fc = field_class;
-	bt_object_get_no_null_check(event_class->specific_context_fc);
+	bt_object_get_ref_no_null_check(event_class->specific_context_fc);
 	bt_field_class_freeze(field_class);
 	BT_LIB_LOGD("Set event class's specific context field class: %!+E",
 		event_class);
@@ -389,7 +395,7 @@ bt_event_class_set_payload_field_class(
 	bt_field_class_make_part_of_trace_class(field_class);
 	bt_object_put_ref(event_class->payload_fc);
 	event_class->payload_fc = field_class;
-	bt_object_get_no_null_check(event_class->payload_fc);
+	bt_object_get_ref_no_null_check(event_class->payload_fc);
 	bt_field_class_freeze(field_class);
 	BT_LIB_LOGD("Set event class's payload field class: %!+E", event_class);
 
@@ -402,10 +408,41 @@ void _bt_event_class_freeze(const struct bt_event_class *event_class)
 {
 	/* The field classes are already frozen */
 	BT_ASSERT(event_class);
+	BT_LIB_LOGD("Freezing event class's user attributes: %!+v",
+		event_class->user_attributes);
+	bt_value_freeze(event_class->user_attributes);
 	BT_LIB_LOGD("Freezing event class: %!+E", event_class);
 	((struct bt_event_class *) event_class)->frozen = true;
 }
 
+const struct bt_value *bt_event_class_borrow_user_attributes_const(
+		const struct bt_event_class *event_class)
+{
+	BT_ASSERT_PRE_DEV_NON_NULL(event_class, "Event class");
+	return event_class->user_attributes;
+}
+
+struct bt_value *bt_event_class_borrow_user_attributes(
+		struct bt_event_class *event_class)
+{
+	return (void *) bt_event_class_borrow_user_attributes_const(
+		(void *) event_class);
+}
+
+void bt_event_class_set_user_attributes(
+		struct bt_event_class *event_class,
+		const struct bt_value *user_attributes)
+{
+	BT_ASSERT_PRE_NON_NULL(event_class, "Event class");
+	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_EVENT_CLASS_HOT(event_class);
+	bt_object_put_ref_no_null_check(event_class->user_attributes);
+	event_class->user_attributes = (void *) user_attributes;
+	bt_object_get_ref_no_null_check(event_class->user_attributes);
+}
+
 void bt_event_class_get_ref(const struct bt_event_class *event_class)
 {
 	bt_object_get_ref(event_class);