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 7e8ea45fc4226709c51bb438e9d5b52d87e5f558..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;
@@ -99,17 +104,52 @@ const char *bt_graph_status_string(enum bt_graph_status status)
                return "BT_GRAPH_STATUS_END";
        case BT_GRAPH_STATUS_OK:
                return "BT_GRAPH_STATUS_OK";
-       case BT_GRAPH_STATUS_ALREADY_IN_A_GRAPH:
-               return "BT_GRAPH_STATUS_ALREADY_IN_A_GRAPH";
        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 */