From: Jérémie Galarneau <jeremie.galarneau@efficios.com>
Date: Fri, 28 Oct 2016 21:27:09 +0000 (-0400)
Subject: Validate ref count opt-in in bt_ref_get
X-Git-Url: http://drtracing.org/?a=commitdiff_plain;h=f2a27391d8b70e8e1c4a2b3df19cfda8a5b087ff;p=deliverable%2Fbabeltrace.git

Validate ref count opt-in in bt_ref_get

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

diff --git a/include/babeltrace/ref-internal.h b/include/babeltrace/ref-internal.h
index 8578de97b..3a8b39f41 100644
--- a/include/babeltrace/ref-internal.h
+++ b/include/babeltrace/ref-internal.h
@@ -50,9 +50,12 @@ static inline
 void bt_ref_get(struct bt_ref *ref)
 {
 	assert(ref);
-	ref->count++;
-	/* Overflow check. */
-	assert(ref->count);
+	if (ref->release) {
+		/* Object has opted-in reference counting. */
+		ref->count++;
+		/* Overflow check. */
+		assert(ref->count);
+	}
 }
 
 static inline