#define BT_AUTODISC_LOGE_APPEND_CAUSE(_fmt, ...) \
BT_AUTODISC_LOG_AND_APPEND(BT_LOG_ERROR, _fmt, ##__VA_ARGS__)
+/*
+ * Define a status enum for inside the auto source discovery code,
+ * as we don't want to return `NO_MATCH` to the caller.
+ */
+typedef enum auto_source_discovery_internal_status {
+ AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_OK = AUTO_SOURCE_DISCOVERY_STATUS_OK,
+ AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_ERROR = AUTO_SOURCE_DISCOVERY_STATUS_ERROR,
+ AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_MEMORY_ERROR = AUTO_SOURCE_DISCOVERY_STATUS_MEMORY_ERROR,
+ AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_INTERRUPTED = AUTO_SOURCE_DISCOVERY_STATUS_INTERRUPTED,
+ AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_NO_MATCH = __BT_FUNC_STATUS_NO_MATCH,
+} auto_source_discovery_internal_status;
+
/* Finalize and free a `struct auto_source_discovery_result`. */
static
goto end;
error:
auto_source_discovery_result_destroy(res);
+ res = NULL;
end:
return res;
*/
static
-int auto_source_discovery_add(struct auto_source_discovery *auto_disc,
+auto_source_discovery_internal_status auto_source_discovery_add(
+ struct auto_source_discovery *auto_disc,
const char *plugin_name,
const char *source_cc_name,
const char *group,
uint64_t original_input_index,
bt_logging_level log_level)
{
- int status;
+ auto_source_discovery_internal_status status;
bt_value_array_append_element_status append_status;
guint len;
guint i;
}
}
- status = 0;
+ status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_OK;
goto end;
error:
- status = -1;
+ status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_ERROR;
end:
return status;
*
* If `component_class_restrict` is non-NULL, only query source component classes
* with that name.
- *
- * Return:
- *
- * - > 0 on success, if no source component class has reported that it handles `input`
- * - 0 on success, if a source component class has reported that it handles `input`
- * - < 0 on failure (e.g. memory error)
*/
static
-int support_info_query_all_sources(const char *input,
+auto_source_discovery_internal_status support_info_query_all_sources(
+ const char *input,
const char *input_type,
uint64_t original_input_index,
const bt_plugin **plugins,
size_t plugin_count,
const char *component_class_restrict,
enum bt_logging_level log_level,
- struct auto_source_discovery *auto_disc)
+ struct auto_source_discovery *auto_disc,
+ const bt_interrupter *interrupter)
{
bt_value_map_insert_entry_status insert_status;
bt_value *query_params = NULL;
- int status;
+ auto_source_discovery_internal_status status;
size_t i_plugins;
const struct bt_value *query_result = NULL;
struct {
double weigth;
} winner = { NULL, NULL, NULL, 0 };
+ if (interrupter && bt_interrupter_is_set(interrupter)) {
+ status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_INTERRUPTED;
+ goto end;
+ }
+
query_params = bt_value_map_create();
if (!query_params) {
BT_AUTODISC_LOGE_APPEND_CAUSE("Failed to allocate a map value.");
}
if (bt_value_map_has_entry(query_result, "group")) {
- enum bt_value_type group_value_type;
-
group_value = bt_value_map_borrow_entry_value_const(query_result, "group");
BT_ASSERT(group_value);
- group_value_type = bt_value_get_type(group_value);
-
- if (group_value_type == BT_VALUE_TYPE_NULL) {
- /* Do as if no value was passed. */
- group_value = NULL;
- } else if (bt_value_get_type(group_value) != BT_VALUE_TYPE_STRING) {
+ if (bt_value_get_type(group_value) != BT_VALUE_TYPE_STRING) {
BT_LOGW("babeltrace.support-info query: unexpected type for entry `group`: "
"component-class-name=source.%s.%s, input=%s, input-type=%s, "
"expected-entry-type=%s,%s, actual-entry-type=%s",
}
BT_LOGD("babeltrace.support-info query: success: component-class-name=source.%s.%s, input=%s, "
- "type=%s, weight=%f\n",
+ "type=%s, weight=%f, group=%s\n",
bt_plugin_get_name(plugin), bt_component_class_get_name(cc), input,
- input_type, weight);
+ input_type, weight, group_value ? bt_value_string_get(group_value) : "(none)");
if (weight > winner.weigth) {
winner.source = source_cc;
plugin_name = bt_plugin_get_name(winner.plugin);
group = winner.group ? bt_value_string_get(winner.group) : NULL;
- BT_LOGI("Input %s is awarded to component class source.%s.%s with weight %f",
- input, plugin_name, source_name, winner.weigth);
+ BT_LOGI("Input awarded: input=%s, type=%s, component-class-name=source.%s.%s, weight=%f, group=%s",
+ input, input_type, plugin_name, source_name, winner.weigth, group ? group : "(none)");
status = auto_source_discovery_add(auto_disc, plugin_name,
source_name, group, input, original_input_index, log_level);
- if (status != 0) {
+ if (status != AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_OK) {
goto error;
}
} else {
- BT_LOGI("Input %s (%s) was not recognized by any source component class.",
+ BT_LOGI("Input not recognized: input=%s, type=%s",
input, input_type);
- status = 1;
+ status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_NO_MATCH;
}
goto end;
error:
- status = -1;
+ status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_ERROR;
end:
bt_value_put_ref(query_result);
*/
static
-int auto_discover_source_for_input_as_string(const char *input,
+auto_source_discovery_internal_status auto_discover_source_for_input_as_string(
+ const char *input,
uint64_t original_input_index,
const bt_plugin **plugins,
size_t plugin_count,
const char *component_class_restrict,
enum bt_logging_level log_level,
- struct auto_source_discovery *auto_disc)
+ struct auto_source_discovery *auto_disc,
+ const bt_interrupter *interrupter)
{
return support_info_query_all_sources(input, "string",
original_input_index, plugins, plugin_count,
- component_class_restrict, log_level, auto_disc);
+ component_class_restrict, log_level, auto_disc,
+ interrupter);
}
static
-int auto_discover_source_for_input_as_dir_or_file_rec(GString *input,
+auto_source_discovery_internal_status auto_discover_source_for_input_as_dir_or_file_rec(
+ GString *input,
uint64_t original_input_index,
const bt_plugin **plugins,
size_t plugin_count,
const char *component_class_restrict,
enum bt_logging_level log_level,
- struct auto_source_discovery *auto_disc)
+ struct auto_source_discovery *auto_disc,
+ const bt_interrupter *interrupter)
{
- int status;
+ auto_source_discovery_internal_status status;
GError *error = NULL;
if (g_file_test(input->str, G_FILE_TEST_IS_REGULAR)) {
/* It's a file. */
status = support_info_query_all_sources(input->str,
"file", original_input_index, plugins, plugin_count,
- component_class_restrict, log_level, auto_disc);
+ component_class_restrict, log_level, auto_disc,
+ interrupter);
} else if (g_file_test(input->str, G_FILE_TEST_IS_DIR)) {
GDir *dir;
const gchar *dirent;
gsize saved_input_len;
- int dir_status = 1;
+ int dir_status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_NO_MATCH;
/* It's a directory. */
status = support_info_query_all_sources(input->str,
"directory", original_input_index, plugins,
plugin_count, component_class_restrict, log_level,
- auto_disc);
+ auto_disc, interrupter);
if (status < 0) {
/* Fatal error. */
goto error;
- } else if (status == 0) {
+ } else if (status == AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_OK ||
+ status == AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_INTERRUPTED) {
/*
* A component class claimed this input as a directory,
- * don't recurse.
+ * don't recurse. Or, we got interrupted.
*/
goto end;
}
if (error->code == G_FILE_ERROR_ACCES) {
/* This is not a fatal error, we just skip it. */
- status = 1;
+ status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_NO_MATCH;
goto end;
} else {
BT_AUTODISC_LOGE_APPEND_CAUSE(fmt, input->str,
status = auto_discover_source_for_input_as_dir_or_file_rec(
input, original_input_index, plugins, plugin_count,
- component_class_restrict, log_level, auto_disc);
+ component_class_restrict, log_level, auto_disc,
+ interrupter);
g_string_truncate(input, saved_input_len);
if (status < 0) {
/* Fatal error. */
goto error;
- } else if (status == 0) {
- dir_status = 0;
+ } else if (status == AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_INTERRUPTED) {
+ goto end;
+ } else if (status == AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_OK) {
+ dir_status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_OK;
}
} else if (errno != 0) {
BT_LOGW_ERRNO("Failed to read directory entry", ": dir=%s", input->str);
g_dir_close(dir);
} else {
BT_LOGD("Skipping %s, not a file or directory", input->str);
- status = 1;
+ status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_NO_MATCH;
}
goto end;
error:
- status = -1;
+ status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_ERROR;
end:
*/
static
-int auto_discover_source_for_input_as_dir_or_file(const char *input,
+auto_source_discovery_internal_status auto_discover_source_for_input_as_dir_or_file(
+ const char *input,
uint64_t original_input_index,
const bt_plugin **plugins,
size_t plugin_count,
const char *component_class_restrict,
enum bt_logging_level log_level,
- struct auto_source_discovery *auto_disc)
+ struct auto_source_discovery *auto_disc,
+ const bt_interrupter *interrupter)
{
GString *mutable_input;
- int status;
+ auto_source_discovery_internal_status status;
mutable_input = g_string_new(input);
if (!mutable_input) {
- status = -1;
+ status = AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_ERROR;
goto end;
}
status = auto_discover_source_for_input_as_dir_or_file_rec(
mutable_input, original_input_index, plugins, plugin_count,
- component_class_restrict, log_level, auto_disc);
+ component_class_restrict, log_level, auto_disc,
+ interrupter);
g_string_free(mutable_input, TRUE);
end:
return status;
}
-int auto_discover_source_components(
+auto_source_discovery_status auto_discover_source_components(
const bt_value *inputs,
const bt_plugin **plugins,
size_t plugin_count,
const char *component_class_restrict,
enum bt_logging_level log_level,
- struct auto_source_discovery *auto_disc)
+ struct auto_source_discovery *auto_disc,
+ const bt_interrupter *interrupter)
{
uint64_t i_inputs, input_count;
- int status;
+ auto_source_discovery_internal_status internal_status;
+ auto_source_discovery_status status;
input_count = bt_value_array_get_length(inputs);
input_value = bt_value_array_borrow_element_by_index_const(inputs, i_inputs);
input = bt_value_string_get(input_value);
- status = auto_discover_source_for_input_as_string(input, i_inputs,
+ internal_status = auto_discover_source_for_input_as_string(input, i_inputs,
plugins, plugin_count, component_class_restrict,
- log_level, auto_disc);
- if (status < 0) {
- /* Fatal error. */
+ log_level, auto_disc, interrupter);
+ if (internal_status < 0 || internal_status == AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_INTERRUPTED) {
+ /* Fatal error or we got interrupted. */
+ status = (auto_source_discovery_status) internal_status;
goto end;
- } else if (status == 0) {
+ } else if (internal_status == AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_OK) {
/* A component class has claimed this input as an arbitrary string. */
continue;
}
- status = auto_discover_source_for_input_as_dir_or_file(input,
+ internal_status = auto_discover_source_for_input_as_dir_or_file(input,
i_inputs, plugins, plugin_count,
- component_class_restrict, log_level, auto_disc);
- if (status < 0) {
- /* Fatal error. */
+ component_class_restrict, log_level, auto_disc, interrupter);
+ if (internal_status < 0 || internal_status == AUTO_SOURCE_DISCOVERY_INTERNAL_STATUS_INTERRUPTED) {
+ /* Fatal error or we got interrupted. */
+ status = (auto_source_discovery_status) internal_status;
goto end;
- } else if (status == 0) {
+ } else if (internal_status == 0) {
/*
* This input (or something under it) was recognized.
*/
BT_LOGW("No trace was found based on input `%s`.", input);
}
- status = 0;
+ status = AUTO_SOURCE_DISCOVERY_STATUS_OK;
end:
return status;
}