Rename: field type -> field class

[babeltrace.git] / lib / ref.c

diff --git a/lib/ref.c b/lib/ref.c
index d39a24d0c1bc5121f4082b5f3eb7ed039a4faae1..ae2cef017745e5aacac0e50e67b997478ae32720 100644 (file)
--- a/lib/ref.c
+++ b/lib/ref.c
@@ -27,7 +27,7 @@
 #define BT_LOG_TAG "REF"
 #include <babeltrace/lib-logging-internal.h>
 
-#include <babeltrace/ref-internal.h>
+#include <babeltrace/assert-pre-internal.h>
 #include <babeltrace/object-internal.h>
 
 void *bt_get(void *ptr)
@@ -38,17 +38,11 @@ void *bt_get(void *ptr)
                goto end;
        }
 
-       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);
+       BT_ASSERT_PRE(obj->is_shared, "Object is not shared: addr=%p", obj);
+       bt_object_get_no_null_check(obj);
+
 end:
-       return obj;
+       return ptr;
 }
 
 void bt_put(void *ptr)
@@ -59,8 +53,8 @@ void bt_put(void *ptr)
                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);
+       BT_ASSERT_PRE(obj->is_shared, "Object is not shared: addr=%p", obj);
+       BT_ASSERT_PRE(bt_object_get_ref_count(obj) > 0,
+               "Decrementing a reference count set to 0: addr=%p", ptr);
+       bt_object_put_no_null_check(obj);
 }