lib: graph: disallow recursive consuming

[babeltrace.git] / include / babeltrace / graph / graph-internal.h

diff --git a/include/babeltrace/graph/graph-internal.h b/include/babeltrace/graph/graph-internal.h
index 405b8c38e2659d7b7f5d6238e0afa204a64afe2f..7fdca46ae8cdfb105ad7b30cd3188c95f9c36045 100644 (file)
--- a/include/babeltrace/graph/graph-internal.h
+++ b/include/babeltrace/graph/graph-internal.h
@@ -28,8 +28,10 @@
  */
 
 #include <babeltrace/graph/graph.h>
+#include <babeltrace/graph/component-status.h>
 #include <babeltrace/babeltrace-internal.h>
 #include <babeltrace/object-internal.h>
+#include <stdlib.h>
 #include <glib.h>
 
 struct bt_component;
@@ -56,6 +58,9 @@ struct bt_graph {
        GQueue *sinks_to_consume;
 
        bt_bool canceled;
+       bt_bool in_remove_listener;
+       bt_bool has_sink;
+       bt_bool can_consume;
 
        struct {
                GArray *port_added;
@@ -87,4 +92,64 @@ BT_HIDDEN
 void bt_graph_remove_connection(struct bt_graph *graph,
                struct bt_connection *connection);
 
+static inline
+const char *bt_graph_status_string(enum bt_graph_status status)
+{
+       switch (status) {
+       case BT_GRAPH_STATUS_CANCELED:
+               return "BT_GRAPH_STATUS_CANCELED";
+       case BT_GRAPH_STATUS_AGAIN:
+               return "BT_GRAPH_STATUS_AGAIN";
+       case BT_GRAPH_STATUS_END:
+               return "BT_GRAPH_STATUS_END";
+       case BT_GRAPH_STATUS_OK:
+               return "BT_GRAPH_STATUS_OK";
+       case BT_GRAPH_STATUS_INVALID:
+               return "BT_GRAPH_STATUS_INVALID";
+       case BT_GRAPH_STATUS_NO_SINK:
+               return "BT_GRAPH_STATUS_NO_SINK";
+       case BT_GRAPH_STATUS_ERROR:
+               return "BT_GRAPH_STATUS_ERROR";
+       case BT_GRAPH_STATUS_COMPONENT_REFUSES_PORT_CONNECTION:
+               return "BT_GRAPH_STATUS_COMPONENT_REFUSES_PORT_CONNECTION";
+       case BT_GRAPH_STATUS_NOMEM:
+               return "BT_GRAPH_STATUS_NOMEM";
+       case BT_GRAPH_STATUS_CANNOT_CONSUME:
+               return "BT_GRAPH_STATUS_CANNOT_CONSUME";
+       default:
+               return "(unknown)";
+       }
+}
+
+static inline
+enum bt_graph_status bt_graph_status_from_component_status(
+               enum bt_component_status comp_status)
+{
+       switch (comp_status) {
+       case BT_COMPONENT_STATUS_OK:
+               return BT_GRAPH_STATUS_OK;
+       case BT_COMPONENT_STATUS_END:
+               return BT_GRAPH_STATUS_END;
+       case BT_COMPONENT_STATUS_AGAIN:
+               return BT_GRAPH_STATUS_AGAIN;
+       case BT_COMPONENT_STATUS_REFUSE_PORT_CONNECTION:
+               return BT_GRAPH_STATUS_COMPONENT_REFUSES_PORT_CONNECTION;
+       case BT_COMPONENT_STATUS_ERROR:
+               return BT_GRAPH_STATUS_ERROR;
+       case BT_COMPONENT_STATUS_UNSUPPORTED:
+               return BT_GRAPH_STATUS_ERROR;
+       case BT_COMPONENT_STATUS_INVALID:
+               return BT_GRAPH_STATUS_INVALID;
+       case BT_COMPONENT_STATUS_NOMEM:
+               return BT_GRAPH_STATUS_NOMEM;
+       case BT_COMPONENT_STATUS_NOT_FOUND:
+               return BT_GRAPH_STATUS_ERROR;
+       default:
+#ifdef BT_LOGF
+               BT_LOGF("Unknown component status: status=%d", comp_status);
+#endif
+               abort();
+       }
+}
+
 #endif /* BABELTRACE_COMPONENT_COMPONENT_GRAPH_INTERNAL_H */