trace{-class}.c: BT_ASSERT_PRE() -> BT_ASSERT_POST() (ref. count check)

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

diff --git a/src/lib/trace-ir/trace.c b/src/lib/trace-ir/trace.c
index 9891d12b224dc53131fc94f0f863b11e0802dd11..184fc0705f2dd6ea125aad2e73fb23a5748a1cb5 100644 (file)
--- a/src/lib/trace-ir/trace.c
+++ b/src/lib/trace-ir/trace.c
@@ -25,6 +25,7 @@
 #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>
@@ -108,7 +109,7 @@ void destroy_trace(struct bt_object *obj)
                         * The destruction listener should not have kept a
                         * reference to the trace.
                         */
-                       BT_ASSERT_PRE(trace->base.ref_count == 1, "Destruction listener kept a reference to the trace being destroyed: %![trace-]+t", 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;
@@ -196,7 +197,7 @@ struct bt_trace *bt_trace_create(struct bt_trace_class *tc)
        }
 
        trace->class = tc;
-       bt_object_get_no_null_check(trace->class);
+       bt_object_get_ref_no_null_check(trace->class);
        BT_LIB_LOGD("Created trace object: %!+t", trace);
        goto end;
 
@@ -568,9 +569,9 @@ void bt_trace_set_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_no_null_check(trace->user_attributes);
+       bt_object_put_ref_no_null_check(trace->user_attributes);
        trace->user_attributes = (void *) user_attributes;
-       bt_object_get_no_null_check(trace->user_attributes);
+       bt_object_get_ref_no_null_check(trace->user_attributes);
 }
 
 void bt_trace_get_ref(const struct bt_trace *trace)