lib: make public reference count functions have strict types

[babeltrace.git] / include / babeltrace / graph / component-sink-const.h

diff --git a/include/babeltrace/graph/component-sink-const.h b/include/babeltrace/graph/component-sink-const.h
index 0afeb5148efc19f2758d3859c51690f67353ae54..bda32846b2e440b4af9b50af603fd833bc833a41 100644 (file)
--- a/include/babeltrace/graph/component-sink-const.h
+++ b/include/babeltrace/graph/component-sink-const.h
@@ -52,6 +52,25 @@ extern const struct bt_port_input *
 bt_component_sink_borrow_input_port_by_index_const(
                const struct bt_component_sink *component, uint64_t index);
 
+extern void bt_component_sink_get_ref(
+               const struct bt_component_sink *component_sink);
+
+extern void bt_component_sink_put_ref(
+               const struct bt_component_sink *component_sink);
+
+#define BT_COMPONENT_SINK_PUT_REF_AND_RESET(_var)              \
+       do {                                                    \
+               bt_component_sink_put_ref(_var);                \
+               (_var) = NULL;                                  \
+       } while (0)
+
+#define BT_COMPONENT_SINK_MOVE_REF(_var_dst, _var_src)         \
+       do {                                                    \
+               bt_component_sink_put_ref(_var_dst);            \
+               (_var_dst) = (_var_src);                        \
+               (_var_src) = NULL;                              \
+       } while (0)
+
 #ifdef __cplusplus
 }
 #endif