urcu_ref: do error checking on urcu_ref_put

[urcu.git] / urcu / urcu_ref.h

diff --git a/urcu/urcu_ref.h b/urcu/urcu_ref.h
index bce711619a8b77dfb5795e5aee8421964d6028ac..75620d14f576b30f38bcc4836d4c2b7e88a56594 100644 (file)
--- a/urcu/urcu_ref.h
+++ b/urcu/urcu_ref.h
@@ -33,14 +33,15 @@ static inline void urcu_ref_init(struct urcu_ref *ref)
 
 static inline void urcu_ref_get(struct urcu_ref *ref)
 {
-       long res = uatomic_add_return(&ref->refcount, 1);
-       assert(res != 0);
+       uatomic_add(&ref->refcount, 1);
 }
 
 static inline void urcu_ref_put(struct urcu_ref *ref,
                                void (*release)(struct urcu_ref *))
 {
-       if (!uatomic_sub_return(&ref->refcount, 1))
+       long res = uatomic_sub_return(&ref->refcount, 1);
+       assert (res >= 0);
+       if (res == 0)
                release(ref);
 }