Tests: ns_contexts: discarded events result in test failure

[lttng-tools.git] / src / common / condition.c

diff --git a/src/common/condition.c b/src/common/condition.c
index 367f4a38b6b977045c3e97a7572d9e607f462ef8..427c49e093ec3c4621161d32f94eefba730e7bf7 100644 (file)
--- a/src/common/condition.c
+++ b/src/common/condition.c
@@ -23,15 +23,36 @@ enum lttng_condition_type lttng_condition_get_type(
 }
 
 void lttng_condition_destroy(struct lttng_condition *condition)
+{
+       lttng_condition_put(condition);
+}
+
+static void condition_destroy_ref(struct urcu_ref *ref)
+{
+       struct lttng_condition *condition =
+               container_of(ref, struct lttng_condition, ref);
+
+       condition->destroy(condition);
+}
+
+LTTNG_HIDDEN
+void lttng_condition_get(struct lttng_condition *condition)
+{
+       urcu_ref_get(&condition->ref);
+}
+
+LTTNG_HIDDEN
+void lttng_condition_put(struct lttng_condition *condition)
 {
        if (!condition) {
                return;
        }
 
        assert(condition->destroy);
-       condition->destroy(condition);
+       urcu_ref_put(&condition->ref, condition_destroy_ref);
 }
 
+
 LTTNG_HIDDEN
 bool lttng_condition_validate(const struct lttng_condition *condition)
 {
@@ -171,4 +192,5 @@ void lttng_condition_init(struct lttng_condition *condition,
                enum lttng_condition_type type)
 {
        condition->type = type;
+       urcu_ref_init(&condition->ref);
 }