From: Jérémie Galarneau <jeremie.galarneau@efficios.com>
Date: Tue, 19 Jan 2016 18:22:59 +0000 (-0500)
Subject: Object: Acquire reference to parent on first acquired reference
X-Git-Url: http://git.efficios.com/?p=babeltrace.git;a=commitdiff_plain;h=47601e5f2145c7a23260a4cbdfc4286b56a77365;ds=sidebyside

Object: Acquire reference to parent on first acquired reference

See doc/ref-counting.md

Signed-off-by: Jérémie Galarneau <jeremie.galarneau@efficios.com>
---

diff --git a/lib/ref.c b/lib/ref.c
index e642e266..97460e93 100644
--- a/lib/ref.c
+++ b/lib/ref.c
@@ -27,13 +27,19 @@
 #include <babeltrace/ref-internal.h>
 #include <babeltrace/object-internal.h>
 
-void *bt_get(void *obj)
+void *bt_get(void *ptr)
 {
-	if (obj) {
-		struct bt_object *base = obj;
+	struct bt_object *obj = ptr;
+
+	if (!obj) {
+		goto end;
+	}
 
-		bt_ref_get(&base->ref_count);
+	if (obj->parent && bt_object_get_ref_count(obj) == 0) {
+		bt_get(obj->parent);
 	}
+	bt_ref_get(&obj->ref_count);
+end:
 	return obj;
 }