+ if (cmd_run_ctx_connect_upstream_port(ctx, port)) {
+ BT_LOGF_STR("Cannot connect upstream port.");
+ fprintf(stderr, "Added port could not be connected: aborting\n");
+ abort();
+ }
+
+end:
+ bt_put(comp);
+ return;
+}
+
+static
+void graph_port_removed_listener(struct bt_component *component,
+ struct bt_port *port, void *data)
+{
+ BT_LOGI("Port removed from a graph's component: comp-addr=%p, "
+ "comp-name=\"%s\", port-addr=%p, port-name=\"%s\"",
+ component, bt_component_get_name(component),
+ port, bt_port_get_name(port));
+}
+
+static
+void graph_ports_connected_listener(struct bt_port *upstream_port,
+ struct bt_port *downstream_port, void *data)
+{
+ struct bt_component *upstream_comp = bt_port_get_component(upstream_port);
+ struct bt_component *downstream_comp = bt_port_get_component(downstream_port);
+
+ assert(upstream_comp);
+ assert(downstream_comp);
+ BT_LOGI("Graph's component ports connected: "
+ "upstream-comp-addr=%p, upstream-comp-name=\"%s\", "
+ "upstream-port-addr=%p, upstream-port-name=\"%s\", "
+ "downstream-comp-addr=%p, downstream-comp-name=\"%s\", "
+ "downstream-port-addr=%p, downstream-port-name=\"%s\"",
+ upstream_comp, bt_component_get_name(upstream_comp),
+ upstream_port, bt_port_get_name(upstream_port),
+ downstream_comp, bt_component_get_name(downstream_comp),
+ downstream_port, bt_port_get_name(downstream_port));
+ bt_put(upstream_comp);
+ bt_put(downstream_comp);
+}
+
+static
+void graph_ports_disconnected_listener(
+ struct bt_component *upstream_component,
+ struct bt_component *downstream_component,
+ struct bt_port *upstream_port, struct bt_port *downstream_port,
+ void *data)
+{
+ BT_LOGI("Graph's component ports disconnected: "
+ "upstream-port-addr=%p, upstream-port-name=\"%s\", "
+ "downstream-port-addr=%p, downstream-port-name=\"%s\"",
+ upstream_port, bt_port_get_name(upstream_port),
+ downstream_port, bt_port_get_name(downstream_port));
+}
+
+static
+void cmd_run_ctx_destroy(struct cmd_run_ctx *ctx)
+{
+ if (!ctx) {
+ return;
+ }
+
+ if (ctx->components) {
+ g_hash_table_destroy(ctx->components);
+ ctx->components = NULL;
+ }
+
+ if (ctx->intersections) {
+ g_hash_table_destroy(ctx->intersections);
+ ctx->intersections = NULL;
+ }
+
+ BT_PUT(ctx->graph);
+ the_graph = NULL;
+ ctx->cfg = NULL;
+}
+
+static
+int cmd_run_ctx_init(struct cmd_run_ctx *ctx, struct bt_config *cfg)
+{
+ int ret = 0;
+
+ ctx->cfg = cfg;
+ ctx->connect_ports = false;
+ ctx->components = g_hash_table_new_full(g_direct_hash, g_direct_equal,
+ NULL, bt_put);
+ if (!ctx->components) {
+ goto error;
+ }
+
+ if (cfg->cmd_data.run.stream_intersection_mode) {
+ ctx->stream_intersection_mode = true;
+ ctx->intersections = g_hash_table_new_full(port_id_hash,
+ port_id_equal, port_id_destroy, trace_range_destroy);
+ if (!ctx->intersections) {
+ goto error;
+ }
+ }
+
+ ctx->graph = bt_graph_create();
+ if (!ctx->graph) {
+ goto error;
+ }
+
+ the_graph = ctx->graph;
+ ret = bt_graph_add_port_added_listener(ctx->graph,
+ graph_port_added_listener, NULL, ctx);
+ if (ret < 0) {
+ BT_LOGE_STR("Cannot add \"port added\" listener to graph.");
+ goto error;
+ }
+
+ ret = bt_graph_add_port_removed_listener(ctx->graph,
+ graph_port_removed_listener, NULL, ctx);
+ if (ret < 0) {
+ BT_LOGE_STR("Cannot add \"port removed\" listener to graph.");
+ goto error;
+ }
+
+ ret = bt_graph_add_ports_connected_listener(ctx->graph,
+ graph_ports_connected_listener, NULL, ctx);
+ if (ret < 0) {
+ BT_LOGE_STR("Cannot add \"ports connected\" listener to graph.");
+ goto error;
+ }
+
+ ret = bt_graph_add_ports_disconnected_listener(ctx->graph,
+ graph_ports_disconnected_listener, NULL, ctx);
+ if (ret < 0) {
+ BT_LOGE_STR("Cannot add \"ports disconnected\" listener to graph.");
+ goto error;
+ }
+
+ goto end;
+
+error:
+ cmd_run_ctx_destroy(ctx);
+ ret = -1;
+
+end:
+ return ret;
+}
+
+static
+int set_stream_intersections(struct cmd_run_ctx *ctx,
+ struct bt_config_component *cfg_comp,
+ struct bt_component_class *comp_cls)
+{
+ int ret = 0;
+ uint64_t trace_idx;
+ int64_t trace_count;
+ enum bt_value_status value_status;
+ const char *path = NULL;
+ struct bt_value *component_path_value = NULL;
+ struct bt_value *query_params = NULL;
+ struct bt_value *query_result = NULL;
+ struct bt_value *trace_info = NULL;
+ struct bt_value *intersection_range = NULL;
+ struct bt_value *intersection_begin = NULL;
+ struct bt_value *intersection_end = NULL;
+ struct bt_value *stream_path_value = NULL;
+ struct bt_value *stream_paths = NULL;
+ struct bt_value *stream_infos = NULL;
+ struct bt_value *stream_info = NULL;
+ struct port_id *port_id = NULL;
+ struct trace_range *trace_range = NULL;
+ const char *fail_reason = NULL;
+
+ component_path_value = bt_value_map_get(cfg_comp->params, "path");
+ if (!bt_value_is_string(component_path_value)) {
+ BT_LOGD("Cannot get path parameter: component-name=%s",
+ cfg_comp->instance_name->str);
+ ret = -1;
+ goto error;
+ }
+
+ value_status = bt_value_string_get(component_path_value, &path);
+ if (value_status != BT_VALUE_STATUS_OK) {
+ BT_LOGD("Cannot get path string value: component-name=%s",
+ cfg_comp->instance_name->str);
+ ret = -1;
+ goto error;
+ }
+
+ query_params = bt_value_map_create();
+ if (!query_params) {
+ BT_LOGE_STR("Cannot create query parameters.");
+ ret = -1;
+ goto error;
+ }
+
+ value_status = bt_value_map_insert(query_params, "path", component_path_value);
+ if (value_status != BT_VALUE_STATUS_OK) {
+ BT_LOGE_STR("Cannot insert path parameter in query parameter map.");
+ ret = -1;
+ goto error;
+ }
+
+ ret = query(comp_cls, "trace-info", query_params, &query_result,
+ &fail_reason);
+ if (ret) {
+ BT_LOGD("Component class does not support the `trace-info` query: %s: "
+ "comp-class-name=\"%s\"", fail_reason,
+ bt_component_class_get_name(comp_cls));
+ ret = -1;
+ goto error;
+ }
+
+ if (!bt_value_is_array(query_result)) {
+ BT_LOGD("Unexpected format of \'trace-info\' query result: "
+ "component-class-name=%s",
+ bt_component_class_get_name(comp_cls));
+ ret = -1;
+ goto error;
+ }
+
+ trace_count = bt_value_array_size(query_result);
+ if (trace_count < 0) {
+ ret = -1;
+ goto error;
+ }
+
+ for (trace_idx = 0; trace_idx < trace_count; trace_idx++) {
+ int64_t begin, end;
+ uint64_t stream_idx;
+ int64_t stream_count;
+
+ trace_info = bt_value_array_get(query_result, trace_idx);
+ if (!trace_info || !bt_value_is_map(trace_info)) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve trace from query result.");
+ goto error;
+ }
+
+ intersection_range = bt_value_map_get(trace_info,
+ "intersection-range-ns");
+ if (!intersection_range) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve \'intersetion-range-ns\' field from query result.");
+ goto error;
+ }
+
+ intersection_begin = bt_value_map_get(intersection_range,
+ "begin");
+ if (!intersection_begin) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve intersection-range-ns \'begin\' field from query result.");
+ goto error;
+ }
+
+ intersection_end = bt_value_map_get(intersection_range,
+ "end");
+ if (!intersection_end) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve intersection-range-ns \'end\' field from query result.");
+ goto error;
+ }
+
+ value_status = bt_value_integer_get(intersection_begin, &begin);
+ if (value_status != BT_VALUE_STATUS_OK) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve value of intersection-range-ns \'begin\' field from query result.");
+ goto error;
+ }
+
+ value_status = bt_value_integer_get(intersection_end, &end);
+ if (value_status != BT_VALUE_STATUS_OK) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve value of intersection-range-ns \'end\' field from query result.");
+ goto error;
+ }
+
+ if (begin < 0 || end < 0 || end < begin) {
+ BT_LOGW("Invalid trace stream intersection values: "
+ "intersection-range-ns:begin=%" PRId64
+ ", intersection-range-ns:end=%" PRId64,
+ begin, end);
+ ret = -1;
+ goto error;
+ }
+
+ stream_infos = bt_value_map_get(trace_info, "streams");
+ if (!stream_infos || !bt_value_is_array(stream_infos)) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve stream informations from trace in query result.");
+ goto error;
+ }
+
+ stream_count = bt_value_array_size(stream_infos);
+ if (stream_count < 0) {
+ ret = -1;
+ goto error;
+ }
+
+ /*
+ * FIXME
+ *
+ * The first path of a stream's "paths" is currently used to
+ * associate streams/ports to a given trace intersection.
+ *
+ * This is a fragile hack as it relies on the port names
+ * being set to the various streams path.
+ *
+ * A stream name should be introduced as part of the trace-info
+ * query result.
+ */
+ for (stream_idx = 0; stream_idx < stream_count; stream_idx++) {
+ const char *stream_path;
+
+ port_id = g_new0(struct port_id, 1);
+ if (!port_id) {
+ ret = -1;
+ BT_LOGE_STR("Cannot allocate memory for port_id structure.");
+ goto error;
+ }
+ port_id->instance_name = strdup(cfg_comp->instance_name->str);
+ if (!port_id->instance_name) {
+ ret = -1;
+ BT_LOGE_STR("Cannot allocate memory for port_id component instance name.");
+ goto error;
+ }
+
+ trace_range = g_new0(struct trace_range, 1);
+ if (!trace_range) {
+ ret = -1;
+ BT_LOGE_STR("Cannot allocate memory for trace_range structure.");
+ goto error;
+ }
+ trace_range->intersection_range_begin_ns = begin;
+ trace_range->intersection_range_end_ns = end;
+
+ stream_info = bt_value_array_get(stream_infos,
+ stream_idx);
+ if (!stream_info || !bt_value_is_map(stream_info)) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve stream informations from trace in query result.");
+ goto error;
+ }
+
+ stream_paths = bt_value_map_get(stream_info, "paths");
+ if (!stream_paths || !bt_value_is_array(stream_paths)) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve stream paths from trace in query result.");
+ goto error;
+ }
+
+ stream_path_value = bt_value_array_get(stream_paths, 0);
+ if (!stream_path_value ||
+ !bt_value_is_string(stream_path_value)) {
+ ret = -1;
+ BT_LOGD_STR("Cannot retrieve stream path value from trace in query result.");
+ goto error;
+ }
+
+ value_status = bt_value_string_get(stream_path_value,
+ &stream_path);
+ if (value_status != BT_VALUE_STATUS_OK) {
+ ret = -1;
+ goto error;
+ }
+
+ port_id->port_name = strdup(stream_path);
+ if (!port_id->port_name) {
+ ret = -1;
+ BT_LOGE_STR("Cannot allocate memory for port_id port_name.");
+ goto error;
+ }
+
+ BT_LOGD("Inserting stream intersection ");
+
+ g_hash_table_insert(ctx->intersections, port_id, trace_range);
+
+ port_id = NULL;
+ trace_range = NULL;
+ BT_PUT(stream_info);
+ BT_PUT(stream_paths);
+ BT_PUT(stream_path_value);
+ }
+
+ BT_PUT(trace_info);
+ BT_PUT(stream_paths);
+ BT_PUT(stream_path_value);
+ BT_PUT(intersection_range);
+ BT_PUT(intersection_begin);
+ BT_PUT(intersection_end);
+ BT_PUT(stream_paths);
+ BT_PUT(stream_path_value);
+ }
+
+ goto end;
+
+error:
+ fprintf(stderr, "%s%sCannot determine stream intersection of trace at path \'%s\'.%s\n",
+ bt_common_color_bold(),
+ bt_common_color_fg_yellow(),
+ path ? path : "(unknown)",
+ bt_common_color_reset());
+end:
+ bt_put(component_path_value);
+ bt_put(query_params);
+ bt_put(query_result);
+ bt_put(trace_info);
+ bt_put(intersection_range);
+ bt_put(intersection_begin);
+ bt_put(intersection_end);
+ bt_put(stream_infos);
+ bt_put(stream_info);
+ bt_put(stream_paths);
+ bt_put(stream_path_value);
+ g_free(port_id);
+ g_free(trace_range);
+ return ret;
+}
+
+static
+int cmd_run_ctx_create_components_from_config_components(
+ struct cmd_run_ctx *ctx, GPtrArray *cfg_components)
+{
+ size_t i;
+ struct bt_component_class *comp_cls = NULL;
+ struct bt_component *comp = NULL;
+ int ret = 0;
+
+ for (i = 0; i < cfg_components->len; i++) {
+ struct bt_config_component *cfg_comp =
+ g_ptr_array_index(cfg_components, i);
+ GQuark quark;
+
+ comp_cls = find_component_class(cfg_comp->plugin_name->str,
+ cfg_comp->comp_cls_name->str, cfg_comp->type);
+ if (!comp_cls) {
+ BT_LOGE("Cannot find component class: plugin-name=\"%s\", "
+ "comp-cls-name=\"%s\", comp-cls-type=%d",
+ cfg_comp->plugin_name->str,
+ cfg_comp->comp_cls_name->str,
+ cfg_comp->type);
+ fprintf(stderr, "%s%sCannot find component class %s",
+ bt_common_color_bold(),
+ bt_common_color_fg_red(),
+ bt_common_color_reset());
+ print_plugin_comp_cls_opt(stderr,
+ cfg_comp->plugin_name->str,
+ cfg_comp->comp_cls_name->str,
+ cfg_comp->type);
+ fprintf(stderr, "\n");
+ goto error;
+ }
+
+ ret = bt_graph_add_component(ctx->graph, comp_cls,
+ cfg_comp->instance_name->str, cfg_comp->params, &comp);
+ if (ret) {
+ BT_LOGE("Cannot create component: plugin-name=\"%s\", "
+ "comp-cls-name=\"%s\", comp-cls-type=%d, "
+ "comp-name=\"%s\"",
+ cfg_comp->plugin_name->str,
+ cfg_comp->comp_cls_name->str,
+ cfg_comp->type, cfg_comp->instance_name->str);
+ fprintf(stderr, "%s%sCannot create component `%s`%s\n",
+ bt_common_color_bold(),
+ bt_common_color_fg_red(),
+ cfg_comp->instance_name->str,
+ bt_common_color_reset());
+ goto error;
+ }
+
+ if (ctx->stream_intersection_mode &&
+ cfg_comp->type == BT_COMPONENT_CLASS_TYPE_SOURCE) {
+ ret = set_stream_intersections(ctx, cfg_comp, comp_cls);
+ if (ret) {
+ goto error;
+ }
+ }
+
+ BT_LOGI("Created and inserted component: comp-addr=%p, comp-name=\"%s\"",
+ comp, cfg_comp->instance_name->str);
+ quark = g_quark_from_string(cfg_comp->instance_name->str);
+ assert(quark > 0);
+ g_hash_table_insert(ctx->components,
+ GUINT_TO_POINTER(quark), comp);
+ comp = NULL;
+ BT_PUT(comp_cls);
+ }
+
+ goto end;
+
+error:
+ ret = -1;
+
+end:
+ bt_put(comp);
+ bt_put(comp_cls);
+ return ret;
+}
+
+static
+int cmd_run_ctx_create_components(struct cmd_run_ctx *ctx)
+{
+ int ret = 0;
+
+ /*
+ * Make sure that, during this phase, our graph's "port added"
+ * listener does not connect ports while we are creating the
+ * components because we have a special, initial phase for
+ * this.
+ */
+ ctx->connect_ports = false;
+
+ ret = cmd_run_ctx_create_components_from_config_components(
+ ctx, ctx->cfg->cmd_data.run.sources);
+ if (ret) {