include/babeltrace/graph/graph-internal.h

   1 #ifndef BABELTRACE_GRAPH_GRAPH_INTERNAL_H
   2 #define BABELTRACE_GRAPH_GRAPH_INTERNAL_H
   3
   4 /*
   5  * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
   6  * Copyright 2017 Jérémie Galarneau <jeremie.galarneau@efficios.com>
   7  *
   8  * Permission is hereby granted, free of charge, to any person obtaining a copy
   9  * of this software and associated documentation files (the "Software"), to deal
  10  * in the Software without restriction, including without limitation the rights
  11  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  12  * copies of the Software, and to permit persons to whom the Software is
  13  * furnished to do so, subject to the following conditions:
  14  *
  15  * The above copyright notice and this permission notice shall be included in
  16  * all copies or substantial portions of the Software.
  17  *
  18  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  19  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  20  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  21  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  22  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  23  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  24  * SOFTWARE.
  25  */
  26
  27 #include <babeltrace/graph/graph.h>
  28 #include <babeltrace/graph/connection-internal.h>
  29 #include <babeltrace/graph/message-const.h>
  30 #include <babeltrace/graph/component-internal.h>
  31 #include <babeltrace/graph/component-sink-internal.h>
  32 #include <babeltrace/babeltrace-internal.h>
  33 #include <babeltrace/object-internal.h>
  34 #include <babeltrace/object-pool-internal.h>
  35 #include <babeltrace/assert-internal.h>
  36 #include <stdlib.h>
  37 #include <glib.h>
  38
  39 struct bt_component;
  40 struct bt_port;
  41
  42 enum bt_graph_configuration_state {
  43         BT_GRAPH_CONFIGURATION_STATE_CONFIGURING,
  44         BT_GRAPH_CONFIGURATION_STATE_PARTIALLY_CONFIGURED,
  45         BT_GRAPH_CONFIGURATION_STATE_CONFIGURED,
  46         BT_GRAPH_CONFIGURATION_STATE_FAULTY,
  47 };
  48
  49 struct bt_graph {
  50         /**
  51          * A component graph contains components and point-to-point connection
  52          * between these components.
  53          *
  54          * In terms of ownership:
  55          * 1) The graph is the components' parent,
  56          * 2) The graph is the connnections' parent,
  57          * 3) Components share the ownership of their connections,
  58          * 4) A connection holds weak references to its two component endpoints.
  59          */
  60         struct bt_object base;
  61
  62         /* Array of pointers to bt_connection. */
  63         GPtrArray *connections;
  64         /* Array of pointers to bt_component. */
  65         GPtrArray *components;
  66         /* Queue of pointers (weak references) to sink bt_components. */
  67         GQueue *sinks_to_consume;
  68
  69         bool canceled;
  70         bool in_remove_listener;
  71         bool has_sink;
  72
  73         /*
  74          * If this is false, then the public API's consuming
  75          * functions (bt_graph_consume() and bt_graph_run()) return
  76          * BT_GRAPH_STATUS_CANNOT_CONSUME. The internal "no check"
  77          * functions always work.
  78          *
  79          * In bt_port_output_message_iterator_create(), on success,
  80          * this flag is cleared so that the iterator remains the only
  81          * consumer for the graph's lifetime.
  82          */
  83         bool can_consume;
  84
  85         enum bt_graph_configuration_state config_state;
  86
  87         struct {
  88                 GArray *source_output_port_added;
  89                 GArray *filter_output_port_added;
  90                 GArray *filter_input_port_added;
  91                 GArray *sink_input_port_added;
  92                 GArray *source_filter_ports_connected;
  93                 GArray *source_sink_ports_connected;
  94                 GArray *filter_filter_ports_connected;
  95                 GArray *filter_sink_ports_connected;
  96         } listeners;
  97
  98         /* Pool of `struct bt_message_event *` */
  99         struct bt_object_pool event_msg_pool;
 100
 101         /* Pool of `struct bt_message_packet_beginning *` */
 102         struct bt_object_pool packet_begin_msg_pool;
 103
 104         /* Pool of `struct bt_message_packet_end *` */
 105         struct bt_object_pool packet_end_msg_pool;
 106
 107         /*
 108          * Array of `struct bt_message *` (weak).
 109          *
 110          * This is an array of all the messages ever created from
 111          * this graph. Some of them can be in one of the pools above,
 112          * some of them can be at large. Because each message has a
 113          * weak pointer to the graph containing its pool, we need to
 114          * notify each message that the graph is gone on graph
 115          * destruction.
 116          *
 117          * TODO: When we support a maximum size for object pools,
 118          * add a way for a message to remove itself from this
 119          * array (on destruction).
 120          */
 121         GPtrArray *messages;
 122 };
 123
 124 static inline
 125 void _bt_graph_set_can_consume(struct bt_graph *graph, bool can_consume)
 126 {
 127         BT_ASSERT(graph);
 128         graph->can_consume = can_consume;
 129 }
 130
 131 #ifdef BT_DEV_MODE
 132 # define bt_graph_set_can_consume       _bt_graph_set_can_consume
 133 #else
 134 # define bt_graph_set_can_consume(_graph, _can_consume)
 135 #endif
 136
 137 BT_HIDDEN
 138 enum bt_graph_status bt_graph_consume_sink_no_check(struct bt_graph *graph,
 139                 struct bt_component_sink *sink);
 140
 141 BT_HIDDEN
 142 void bt_graph_notify_port_added(struct bt_graph *graph, struct bt_port *port);
 143
 144 BT_HIDDEN
 145 void bt_graph_notify_ports_connected(struct bt_graph *graph,
 146                 struct bt_port *upstream_port, struct bt_port *downstream_port);
 147
 148 BT_HIDDEN
 149 void bt_graph_remove_connection(struct bt_graph *graph,
 150                 struct bt_connection *connection);
 151
 152 /*
 153  * This only works with a component which is not connected at this
 154  * point.
 155  *
 156  * Also the reference count of `component` should be 0 when you call
 157  * this function, which means only `graph` owns the component, so it
 158  * is safe to destroy.
 159  */
 160 BT_HIDDEN
 161 int bt_graph_remove_unconnected_component(struct bt_graph *graph,
 162                 struct bt_component *component);
 163
 164 BT_HIDDEN
 165 void bt_graph_add_message(struct bt_graph *graph,
 166                 struct bt_message *msg);
 167
 168 static inline
 169 const char *bt_graph_status_string(enum bt_graph_status status)
 170 {
 171         switch (status) {
 172         case BT_GRAPH_STATUS_CANCELED:
 173                 return "BT_GRAPH_STATUS_CANCELED";
 174         case BT_GRAPH_STATUS_AGAIN:
 175                 return "BT_GRAPH_STATUS_AGAIN";
 176         case BT_GRAPH_STATUS_END:
 177                 return "BT_GRAPH_STATUS_END";
 178         case BT_GRAPH_STATUS_OK:
 179                 return "BT_GRAPH_STATUS_OK";
 180         case BT_GRAPH_STATUS_ERROR:
 181                 return "BT_GRAPH_STATUS_ERROR";
 182         case BT_GRAPH_STATUS_COMPONENT_REFUSES_PORT_CONNECTION:
 183                 return "BT_GRAPH_STATUS_COMPONENT_REFUSES_PORT_CONNECTION";
 184         case BT_GRAPH_STATUS_NOMEM:
 185                 return "BT_GRAPH_STATUS_NOMEM";
 186         default:
 187                 return "(unknown)";
 188         }
 189 }
 190
 191 static inline
 192 const char *bt_graph_configuration_state_string(
 193                 enum bt_graph_configuration_state state)
 194 {
 195         switch (state) {
 196         case BT_GRAPH_CONFIGURATION_STATE_CONFIGURING:
 197                 return "BT_GRAPH_CONFIGURATION_STATE_CONFIGURING";
 198         case BT_GRAPH_CONFIGURATION_STATE_PARTIALLY_CONFIGURED:
 199                 return "BT_GRAPH_CONFIGURATION_STATE_PARTIALLY_CONFIGURED";
 200         case BT_GRAPH_CONFIGURATION_STATE_CONFIGURED:
 201                 return "BT_GRAPH_CONFIGURATION_STATE_CONFIGURED";
 202         default:
 203                 return "(unknown)";
 204         }
 205 }
 206
 207 static inline
 208 enum bt_graph_status bt_graph_configure(struct bt_graph *graph)
 209 {
 210         enum bt_graph_status status = BT_GRAPH_STATUS_OK;
 211         uint64_t i;
 212
 213         BT_ASSERT(graph->config_state != BT_GRAPH_CONFIGURATION_STATE_FAULTY);
 214
 215         if (likely(graph->config_state ==
 216                         BT_GRAPH_CONFIGURATION_STATE_CONFIGURED)) {
 217                 goto end;
 218         }
 219
 220 #ifdef BT_ASSERT_PRE
 221         BT_ASSERT_PRE(graph->has_sink, "Graph has no sink component: %!+g", graph);
 222 #endif
 223
 224         graph->config_state = BT_GRAPH_CONFIGURATION_STATE_PARTIALLY_CONFIGURED;
 225
 226         for (i = 0; i < graph->components->len; i++) {
 227                 struct bt_component *comp = graph->components->pdata[i];
 228                 struct bt_component_sink *comp_sink = (void *) comp;
 229                 struct bt_component_class_sink *comp_cls_sink =
 230                         (void *) comp->class;
 231
 232                 if (comp->class->type != BT_COMPONENT_CLASS_TYPE_SINK) {
 233                         continue;
 234                 }
 235
 236                 if (comp_sink->graph_is_configured_method_called) {
 237                         continue;
 238                 }
 239
 240                 if (comp_cls_sink->methods.graph_is_configured) {
 241                         enum bt_self_component_status comp_status;
 242
 243 #ifdef BT_LIB_LOGD
 244                         BT_LIB_LOGD("Calling user's \"graph is configured\" method: "
 245                                 "%![graph-]+g, %![comp-]+c",
 246                                 graph, comp);
 247 #endif
 248
 249                         comp_status = comp_cls_sink->methods.graph_is_configured(
 250                                 (void *) comp_sink);
 251
 252 #ifdef BT_LIB_LOGD
 253                         BT_LIB_LOGD("User method returned: status=%s",
 254                                 bt_self_component_status_string(comp_status));
 255 #endif
 256
 257 #ifdef BT_ASSERT_PRE
 258                         BT_ASSERT_PRE(comp_status == BT_SELF_COMPONENT_STATUS_OK ||
 259                                 comp_status == BT_SELF_COMPONENT_STATUS_ERROR ||
 260                                 comp_status == BT_SELF_COMPONENT_STATUS_NOMEM,
 261                                 "Unexpected returned status: status=%s",
 262                                 bt_self_component_status_string(comp_status));
 263 #endif
 264
 265                         if (comp_status != BT_SELF_COMPONENT_STATUS_OK) {
 266                                 status = BT_GRAPH_STATUS_ERROR;
 267 #ifdef BT_LIB_LOGW
 268                                 BT_LIB_LOGW("User's \"graph is configured\" method failed: "
 269                                         "%![comp-]+c, status=%s",
 270                                         comp,
 271                                         bt_self_component_status_string(
 272                                                 comp_status));
 273 #endif
 274
 275                                 goto end;
 276                         }
 277                 }
 278
 279                 comp_sink->graph_is_configured_method_called = true;
 280         }
 281
 282         graph->config_state = BT_GRAPH_CONFIGURATION_STATE_CONFIGURED;
 283
 284 end:
 285         return status;
 286 }
 287
 288 #endif /* BABELTRACE_GRAPH_GRAPH_INTERNAL_H */