lib: make public reference count functions have strict types

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

diff --git a/include/babeltrace/graph/port-output-const.h b/include/babeltrace/graph/port-output-const.h
index 4d44914e86faaaf848ecae036f8cd090ec57b2d6..d13c7395c777e81c0611c7b0e5fd77359cf87e0e 100644 (file)
--- a/include/babeltrace/graph/port-output-const.h
+++ b/include/babeltrace/graph/port-output-const.h
@@ -2,10 +2,9 @@
 #define BABELTRACE_GRAPH_PORT_OUTPUT_CONST_H
 
 /*
+ * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
  * Copyright 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
  *
- * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
- *
  * Permission is hereby granted, free of charge, to any person obtaining a copy
  * of this software and associated documentation files (the "Software"), to deal
  * in the Software without restriction, including without limitation the rights
@@ -41,6 +40,23 @@ const struct bt_port *bt_port_output_as_port_const(
        return (const void *) port_output;
 }
 
+extern void bt_port_output_get_ref(const struct bt_port_output *port_output);
+
+extern void bt_port_output_put_ref(const struct bt_port_output *port_output);
+
+#define BT_PORT_OUTPUT_PUT_REF_AND_RESET(_var)         \
+       do {                                            \
+               bt_port_output_put_ref(_var);           \
+               (_var) = NULL;                          \
+       } while (0)
+
+#define BT_PORT_OUTPUT_MOVE_REF(_var_dst, _var_src)    \
+       do {                                            \
+               bt_port_output_put_ref(_var_dst);       \
+               (_var_dst) = (_var_src);                \
+               (_var_src) = NULL;                      \
+       } while (0)
+
 #ifdef __cplusplus
 }
 #endif