lib: make public reference count functions have strict types

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

diff --git a/include/babeltrace/graph/port-const.h b/include/babeltrace/graph/port-const.h
index 414c5ccde842163e279d4566631005b6b3d276c8..826ab02b034a456589e2807d42c798cc792b87fc 100644 (file)
--- a/include/babeltrace/graph/port-const.h
+++ b/include/babeltrace/graph/port-const.h
@@ -66,6 +66,23 @@ bt_bool bt_port_is_output(const struct bt_port *port)
        return bt_port_get_type(port) == BT_PORT_TYPE_OUTPUT;
 }
 
+extern void bt_port_get_ref(const struct bt_port *port);
+
+extern void bt_port_put_ref(const struct bt_port *port);
+
+#define BT_PORT_PUT_REF_AND_RESET(_var)                \
+       do {                                    \
+               bt_port_put_ref(_var);          \
+               (_var) = NULL;                  \
+       } while (0)
+
+#define BT_PORT_MOVE_REF(_var_dst, _var_src)   \
+       do {                                    \
+               bt_port_put_ref(_var_dst);      \
+               (_var_dst) = (_var_src);        \
+               (_var_src) = NULL;              \
+       } while (0)
+
 #ifdef __cplusplus
 }
 #endif