lib: make public reference count functions have strict types

[babeltrace.git] / include / babeltrace / graph / connection-const.h

diff --git a/include/babeltrace/graph/connection-const.h b/include/babeltrace/graph/connection-const.h
index 59959063f43b7e19c4cc593e4ffe800bd119dc85..52f605e054fa9fad55504aad6471137234d30410 100644 (file)
--- a/include/babeltrace/graph/connection-const.h
+++ b/include/babeltrace/graph/connection-const.h
@@ -43,6 +43,23 @@ extern const struct bt_port_output *bt_connection_borrow_upstream_port_const(
 
 extern bt_bool bt_connection_is_ended(const struct bt_connection *connection);
 
+extern void bt_connection_get_ref(const struct bt_connection *connection);
+
+extern void bt_connection_put_ref(const struct bt_connection *connection);
+
+#define BT_CONNECTION_PUT_REF_AND_RESET(_var)          \
+       do {                                            \
+               bt_connection_put_ref(_var);            \
+               (_var) = NULL;                          \
+       } while (0)
+
+#define BT_CONNECTION_MOVE_REF(_var_dst, _var_src)     \
+       do {                                            \
+               bt_connection_put_ref(_var_dst);        \
+               (_var_dst) = (_var_src);                \
+               (_var_src) = NULL;                      \
+       } while (0)
+
 #ifdef __cplusplus
 }
 #endif