Fix: IR visitor: error msg printing

[babeltrace.git] / lib / ref.c

diff --git a/lib/ref.c b/lib/ref.c
index e642e266de87543c24f9f3819e6c8b7a577c1b46..d2ec688e7d3d1d3f31ebd4586bb28ffbceefaf73 100644 (file)
--- a/lib/ref.c
+++ b/lib/ref.c
@@ -27,21 +27,40 @@
 #include <babeltrace/ref-internal.h>
 #include <babeltrace/object-internal.h>
 
-void *bt_get(void *obj)
+#define BT_LOG_TAG "REF"
+#include <babeltrace/lib-logging-internal.h>
+
+void *bt_get(void *ptr)
 {
-       if (obj) {
-               struct bt_object *base = obj;
+       struct bt_object *obj = ptr;
+
+       if (unlikely(!obj)) {
+               goto end;
+       }
 
-               bt_ref_get(&base->ref_count);
+       if (unlikely(obj->parent && bt_object_get_ref_count(obj) == 0)) {
+               BT_LOGV("Incrementing object's parent's reference count: "
+                       "addr=%p, parent-addr=%p", ptr, obj->parent);
+               bt_get(obj->parent);
        }
+       bt_ref_get(&obj->ref_count);
+       BT_LOGV("Incremented object's reference count: %lu -> %lu: "
+               "addr=%p, new-count=%lu", obj->ref_count.count - 1,
+               obj->ref_count.count, ptr, obj->ref_count.count);
+end:
        return obj;
 }
 
-void bt_put(void *obj)
+void bt_put(void *ptr)
 {
-       if (obj) {
-               struct bt_object *base = obj;
+       struct bt_object *obj = ptr;
 
-               bt_ref_put(&base->ref_count);
+       if (unlikely(!obj)) {
+               return;
        }
+
+       BT_LOGV("Decrementing object's reference count: %lu -> %lu: "
+               "addr=%p, cur-count=%lu", obj->ref_count.count,
+               obj->ref_count.count - 1, ptr, obj->ref_count.count);
+       bt_ref_put(&obj->ref_count);
 }