#include "common/common.h"
#include <babeltrace2/plugin/plugin-const.h>
#include <babeltrace2/graph/component-class-const.h>
+#include <babeltrace2/current-thread.h>
#include "lib/graph/component-class.h"
#include <babeltrace2/types.h>
#include <glib.h>
#include "plugin-so.h"
#include "lib/func-status.h"
-#define PYTHON_PLUGIN_PROVIDER_FILENAME "libbabeltrace2-python-plugin-provider." G_MODULE_SUFFIX
+#define PYTHON_PLUGIN_PROVIDER_FILENAME "babeltrace2-python-plugin-provider." G_MODULE_SUFFIX
+#define PYTHON_PLUGIN_PROVIDER_DIR BABELTRACE_PLUGIN_PROVIDERS_DIR
#define PYTHON_PLUGIN_PROVIDER_SYM_NAME bt_plugin_python_create_all_from_file
#define PYTHON_PLUGIN_PROVIDER_SYM_NAME_STR G_STRINGIFY(PYTHON_PLUGIN_PROVIDER_SYM_NAME)
bt_plugin_python_create_all_from_file;
static
-void init_python_plugin_provider(void)
+enum bt_plugin_status init_python_plugin_provider(void)
{
}
#else /* BT_BUILT_IN_PYTHON_PLUGIN_SUPPORT */
struct bt_plugin_set **plugin_set_out);
static
-void init_python_plugin_provider(void) {
- if (bt_plugin_python_create_all_from_file_sym != NULL) {
- return;
+int init_python_plugin_provider(void) {
+ int status = BT_FUNC_STATUS_OK;
+ const char *provider_dir_envvar;
+ static const char * const provider_dir_envvar_name = "LIBBABELTRACE2_PLUGIN_PROVIDER_DIR";
+ char *provider_path = NULL;
+
+ if (bt_plugin_python_create_all_from_file_sym) {
+ goto end;
}
BT_LOGI_STR("Loading Python plugin provider module.");
- python_plugin_provider_module =
- g_module_open(PYTHON_PLUGIN_PROVIDER_FILENAME, 0);
+
+ provider_dir_envvar = getenv(provider_dir_envvar_name);
+ if (provider_dir_envvar) {
+ provider_path = g_build_filename(provider_dir_envvar,
+ PYTHON_PLUGIN_PROVIDER_FILENAME, NULL);
+ BT_LOGI("Using `%s` environment variable to find the Python "
+ "plugin provider: path=\"%s\"", provider_dir_envvar_name,
+ provider_path);
+ } else {
+ provider_path = g_build_filename(PYTHON_PLUGIN_PROVIDER_DIR,
+ PYTHON_PLUGIN_PROVIDER_FILENAME, NULL);
+ BT_LOGI("Using default path (`%s` environment variable is not "
+ "set) to find the Python plugin provider: path=\"%s\"",
+ provider_dir_envvar_name, provider_path);
+ }
+
+ python_plugin_provider_module = g_module_open(provider_path, 0);
if (!python_plugin_provider_module) {
/*
* This is not an error. The whole point of having an
* missing and the Babeltrace library still works.
*/
BT_LOGI("Cannot open `%s`: %s: continuing without Python plugin support.",
- PYTHON_PLUGIN_PROVIDER_FILENAME, g_module_error());
- return;
+ provider_path, g_module_error());
+ goto end;
}
if (!g_module_symbol(python_plugin_provider_module,
* plugin provider shared object, we expect this symbol
* to exist.
*/
- BT_LOGE("Cannot find the Python plugin provider loading symbol: "
+ BT_LIB_LOGE_APPEND_CAUSE(
+ "Cannot find the Python plugin provider loading symbol: "
"%s: continuing without Python plugin support: "
"file=\"%s\", symbol=\"%s\"",
g_module_error(),
- PYTHON_PLUGIN_PROVIDER_FILENAME,
+ provider_path,
PYTHON_PLUGIN_PROVIDER_SYM_NAME_STR);
- return;
+ status = BT_FUNC_STATUS_ERROR;
+ goto end;
}
BT_LOGI("Loaded Python plugin provider module: addr=%p",
python_plugin_provider_module);
+
+end:
+ g_free(provider_path);
+
+ return status;
}
__attribute__((destructor)) static
BT_LOGI("Unloading Python plugin provider module.");
if (!g_module_close(python_plugin_provider_module)) {
+ /*
+ * This occurs when the library is finalized: do
+ * NOT append an error cause.
+ */
BT_LOGE("Failed to close the Python plugin provider module: %s.",
g_module_error());
}
}
#endif
-uint64_t bt_plugin_set_get_plugin_count(struct bt_plugin_set *plugin_set)
+uint64_t bt_plugin_set_get_plugin_count(const struct bt_plugin_set *plugin_set)
{
- BT_ASSERT_PRE_NON_NULL(plugin_set, "Plugin set");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin_set, "Plugin set");
return (uint64_t) plugin_set->plugins->len;
}
const struct bt_plugin *bt_plugin_set_borrow_plugin_by_index_const(
const struct bt_plugin_set *plugin_set, uint64_t index)
{
- BT_ASSERT_PRE_NON_NULL(plugin_set, "Plugin set");
- BT_ASSERT_PRE_VALID_INDEX(index, plugin_set->plugins->len);
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin_set, "Plugin set");
+ BT_ASSERT_PRE_DEV_VALID_INDEX(index, plugin_set->plugins->len);
return g_ptr_array_index(plugin_set->plugins, index);
}
BT_ASSERT(!*plugin_set_out);
/* Try Python plugins if support is available */
- init_python_plugin_provider();
+ status = init_python_plugin_provider();
+ if (status < 0) {
+ /* init_python_plugin_provider() logs errors */
+ goto end;
+ }
+
+ BT_ASSERT(status == BT_FUNC_STATUS_OK);
+ status = BT_FUNC_STATUS_NOT_FOUND;
+
if (bt_plugin_python_create_all_from_file_sym) {
+ /* Python plugin provider exists */
status = bt_plugin_python_create_all_from_file_sym(path,
fail_on_load_error, (void *) plugin_set_out);
if (status == BT_FUNC_STATUS_OK) {
BT_ASSERT((*plugin_set_out)->plugins->len > 0);
goto end;
} else if (status < 0) {
+ /*
+ * bt_plugin_python_create_all_from_file_sym()
+ * handles `fail_on_load_error` itself, so this
+ * is a "real" error.
+ */
BT_ASSERT(!*plugin_set_out);
goto end;
}
return status;
}
-static void destroy_gstring(void *data)
+static
+void destroy_gstring(void *data)
{
g_string_free(data, TRUE);
}
-enum bt_plugin_find_status bt_plugin_find(const char *plugin_name,
- bt_bool fail_on_load_error, const struct bt_plugin **plugin_out)
+enum bt_plugin_find_all_status bt_plugin_find_all(bt_bool find_in_std_env_var,
+ bt_bool find_in_user_dir, bt_bool find_in_sys_dir,
+ bt_bool find_in_static, bt_bool fail_on_load_error,
+ const struct bt_plugin_set **plugin_set_out)
{
- const char *system_plugin_dir;
char *home_plugin_dir = NULL;
- const char *envvar;
- const struct bt_plugin *plugin = NULL;
const struct bt_plugin_set *plugin_set = NULL;
GPtrArray *dirs = NULL;
int ret;
int status = BT_FUNC_STATUS_OK;
- size_t i, j;
+ uint64_t dir_i, plugin_i;
- BT_ASSERT_PRE_NON_NULL(plugin_name, "Name");
- BT_ASSERT_PRE_NON_NULL(plugin_out, "Plugin (output)");
- BT_LOGI("Finding named plugin in standard directories and built-in plugins: "
- "name=\"%s\"", plugin_name);
+ BT_ASSERT_PRE_NON_NULL(plugin_set_out, "Plugin set (output)");
+ BT_LOGI("Finding all plugins in standard directories and built-in plugins: "
+ "find-in-std-env-var=%d, find-in-user-dir=%d, "
+ "find-in-sys-dir=%d, find-in-static=%d",
+ find_in_std_env_var, find_in_user_dir, find_in_sys_dir,
+ find_in_static);
dirs = g_ptr_array_new_with_free_func((GDestroyNotify) destroy_gstring);
if (!dirs) {
- BT_LOGE_STR("Failed to allocate a GPtrArray.");
+ BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GPtrArray.");
+ status = BT_FUNC_STATUS_MEMORY_ERROR;
+ goto end;
+ }
+
+ *plugin_set_out = bt_plugin_set_create();
+ if (!*plugin_set_out) {
+ BT_LIB_LOGE_APPEND_CAUSE("Cannot create empty plugin set.");
status = BT_FUNC_STATUS_MEMORY_ERROR;
goto end;
}
/*
* Search order is:
*
- * 1. BABELTRACE_PLUGIN_PATH environment variable
+ * 1. `BABELTRACE_PLUGIN_PATH` environment variable
* (colon-separated list of directories)
- * 2. ~/.local/lib/babeltrace2/plugins
+ * 2. `~/.local/lib/babeltrace2/plugins`
* 3. Default system directory for Babeltrace plugins, usually
- * /usr/lib/babeltrace2/plugins or
- * /usr/local/lib/babeltrace2/plugins if installed
- * locally
+ * `/usr/lib/babeltrace2/plugins` or
+ * `/usr/local/lib/babeltrace2/plugins` if installed locally
* 4. Built-in plugins (static)
*
* Directories are searched non-recursively.
*/
- envvar = getenv("BABELTRACE_PLUGIN_PATH");
- if (envvar) {
- ret = bt_common_append_plugin_path_dirs(envvar, dirs);
- if (ret) {
- BT_LOGE_STR("Failed to append plugin path to array of directories.");
- status = BT_FUNC_STATUS_MEMORY_ERROR;
- goto end;
+ if (find_in_std_env_var) {
+ const char *envvar = getenv("BABELTRACE_PLUGIN_PATH");
+
+ if (envvar) {
+ ret = bt_common_append_plugin_path_dirs(envvar, dirs);
+ if (ret) {
+ BT_LIB_LOGE_APPEND_CAUSE(
+ "Failed to append plugin path to array of directories.");
+ status = BT_FUNC_STATUS_MEMORY_ERROR;
+ goto end;
+ }
}
}
- home_plugin_dir = bt_common_get_home_plugin_path(BT_LOG_OUTPUT_LEVEL);
- if (home_plugin_dir) {
- GString *home_plugin_dir_str = g_string_new(home_plugin_dir);
+ if (find_in_user_dir) {
+ home_plugin_dir = bt_common_get_home_plugin_path(
+ BT_LOG_OUTPUT_LEVEL);
+ if (home_plugin_dir) {
+ GString *home_plugin_dir_str = g_string_new(
+ home_plugin_dir);
- if (!home_plugin_dir_str) {
- BT_LOGE_STR("Failed to allocate a GString.");
- status = BT_FUNC_STATUS_MEMORY_ERROR;
- goto end;
- }
+ if (!home_plugin_dir_str) {
+ BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
+ status = BT_FUNC_STATUS_MEMORY_ERROR;
+ goto end;
+ }
- g_ptr_array_add(dirs, home_plugin_dir_str);
+ g_ptr_array_add(dirs, home_plugin_dir_str);
+ }
}
- system_plugin_dir = bt_common_get_system_plugin_path();
- if (system_plugin_dir) {
- GString *system_plugin_dir_str =
- g_string_new(system_plugin_dir);
+ if (find_in_sys_dir) {
+ const char *system_plugin_dir =
+ bt_common_get_system_plugin_path();
- if (!system_plugin_dir_str) {
- BT_LOGE_STR("Failed to allocate a GString.");
- status = BT_FUNC_STATUS_MEMORY_ERROR;
- goto end;
- }
+ if (system_plugin_dir) {
+ GString *system_plugin_dir_str =
+ g_string_new(system_plugin_dir);
- g_ptr_array_add(dirs, system_plugin_dir_str);
+ if (!system_plugin_dir_str) {
+ BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
+ status = BT_FUNC_STATUS_MEMORY_ERROR;
+ goto end;
+ }
+
+ g_ptr_array_add(dirs, system_plugin_dir_str);
+ }
}
- for (i = 0; i < dirs->len; i++) {
- GString *dir = g_ptr_array_index(dirs, i);
+ for (dir_i = 0; dir_i < dirs->len; dir_i++) {
+ GString *dir = dirs->pdata[dir_i];
BT_OBJECT_PUT_REF_AND_RESET(plugin_set);
BT_ASSERT(status == BT_FUNC_STATUS_OK);
BT_ASSERT(plugin_set);
+ BT_LOGI("Found plugins in directory: path=\"%s\", count=%u",
+ dir->str, plugin_set->plugins->len);
+
+ for (plugin_i = 0; plugin_i < plugin_set->plugins->len;
+ plugin_i++) {
+ bt_plugin_set_add_plugin((void *) *plugin_set_out,
+ plugin_set->plugins->pdata[plugin_i]);
+ }
+ }
- for (j = 0; j < plugin_set->plugins->len; j++) {
- const struct bt_plugin *candidate_plugin =
- g_ptr_array_index(plugin_set->plugins, j);
+ if (find_in_static) {
+ BT_OBJECT_PUT_REF_AND_RESET(plugin_set);
+ status = bt_plugin_find_all_from_static(fail_on_load_error,
+ &plugin_set);
+ if (status < 0) {
+ BT_ASSERT(!plugin_set);
+ goto end;
+ } else if (status == BT_FUNC_STATUS_NOT_FOUND) {
+ BT_ASSERT(!plugin_set);
+ BT_LOGI_STR("No plugins found in built-in plugins.");
+ goto end;
+ }
- if (strcmp(bt_plugin_get_name(candidate_plugin),
- plugin_name) == 0) {
- BT_LOGI("Plugin found in directory: name=\"%s\", path=\"%s\"",
- plugin_name, dir->str);
- *plugin_out = candidate_plugin;
- bt_object_get_no_null_check(*plugin_out);
- goto end;
- }
+ BT_ASSERT(status == BT_FUNC_STATUS_OK);
+ BT_ASSERT(plugin_set);
+ BT_LOGI("Found built-in plugins: count=%u",
+ plugin_set->plugins->len);
+
+ for (plugin_i = 0; plugin_i < plugin_set->plugins->len;
+ plugin_i++) {
+ bt_plugin_set_add_plugin((void *) *plugin_set_out,
+ plugin_set->plugins->pdata[plugin_i]);
}
+ }
- BT_LOGI("Plugin not found in directory: name=\"%s\", path=\"%s\"",
- plugin_name, dir->str);
+end:
+ free(home_plugin_dir);
+ bt_object_put_ref(plugin_set);
+
+ if (dirs) {
+ g_ptr_array_free(dirs, TRUE);
}
- BT_OBJECT_PUT_REF_AND_RESET(plugin_set);
- status = bt_plugin_find_all_from_static(fail_on_load_error,
- &plugin_set);
if (status < 0) {
- BT_ASSERT(!plugin_set);
- goto end;
- } else if (status == BT_FUNC_STATUS_NOT_FOUND) {
- BT_ASSERT(!plugin_set);
- BT_LOGI_STR("No plugins found in built-in plugins.");
+ BT_OBJECT_PUT_REF_AND_RESET(*plugin_set_out);
+ } else {
+ BT_ASSERT(*plugin_set_out);
+
+ if ((*plugin_set_out)->plugins->len > 0) {
+ BT_LOGI("Found plugins in standard directories and built-in plugins: "
+ "count=%u", (*plugin_set_out)->plugins->len);
+ status = BT_FUNC_STATUS_OK;
+ } else {
+ BT_LOGI_STR("No plugins found in standard directories and built-in plugins.");
+ status = BT_FUNC_STATUS_NOT_FOUND;
+ BT_OBJECT_PUT_REF_AND_RESET(*plugin_set_out);
+ }
+ }
+
+ return status;
+}
+
+enum bt_plugin_find_status bt_plugin_find(const char *plugin_name,
+ bt_bool find_in_std_env_var, bt_bool find_in_user_dir,
+ bt_bool find_in_sys_dir, bt_bool find_in_static,
+ bt_bool fail_on_load_error, const struct bt_plugin **plugin_out)
+{
+ enum bt_plugin_find_status status;
+ const struct bt_plugin_set *plugin_set = NULL;
+ uint64_t i;
+
+ BT_ASSERT_PRE_NON_NULL(plugin_name, "Name");
+ BT_ASSERT_PRE_NON_NULL(plugin_out, "Plugin (output)");
+ BT_LOGI("Finding named plugin in standard directories and built-in plugins: "
+ "name=\"%s\", find-in-std-env-var=%d, find-in-user-dir=%d, "
+ "find-in-sys-dir=%d, find-in-static=%d",
+ plugin_name, find_in_std_env_var, find_in_user_dir,
+ find_in_sys_dir, find_in_static);
+ status = (enum bt_plugin_find_status) bt_plugin_find_all(find_in_std_env_var, find_in_user_dir,
+ find_in_sys_dir, find_in_static, fail_on_load_error,
+ &plugin_set);
+ if (status != BT_FUNC_STATUS_OK) {
goto end;
}
- BT_ASSERT(status == BT_FUNC_STATUS_OK);
BT_ASSERT(plugin_set);
- for (j = 0; j < plugin_set->plugins->len; j++) {
- const struct bt_plugin *candidate_plugin =
- g_ptr_array_index(plugin_set->plugins, j);
+ for (i = 0; i < plugin_set->plugins->len; i++) {
+ const struct bt_plugin *plugin = plugin_set->plugins->pdata[i];
- if (strcmp(bt_plugin_get_name(candidate_plugin),
- plugin_name) == 0) {
- BT_LOGI("Plugin found in built-in plugins: "
- "name=\"%s\"", plugin_name);
- *plugin_out = candidate_plugin;
+ if (strcmp(plugin->info.name->str, plugin_name) == 0) {
+ *plugin_out = plugin;
bt_object_get_no_null_check(*plugin_out);
goto end;
}
status = BT_FUNC_STATUS_NOT_FOUND;
end:
- free(home_plugin_dir);
- bt_object_put_ref(plugin_set);
-
- if (dirs) {
- g_ptr_array_free(dirs, TRUE);
- }
-
if (status == BT_FUNC_STATUS_OK) {
+ BT_ASSERT(*plugin_out);
BT_LIB_LOGI("Found plugin in standard directories and built-in plugins: "
- "%!+l", plugin);
- } else if (status == BT_FUNC_STATUS_NOT_FOUND) {
+ "%!+l", *plugin_out);
+ } else if (status == BT_FUNC_STATUS_NOT_FOUND) {
BT_LOGI("No plugin found in standard directories and built-in plugins: "
"name=\"%s\"", plugin_name);
}
BT_LOGW_ERRNO("Cannot open directory",
": path=\"%s\", recurse=%d",
path, recurse);
+ (void) BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(
+ "Babeltrace library",
+ "Cannot open directory: path=\"%s\", recurse=%d",
+ path, recurse);
status = BT_FUNC_STATUS_ERROR;
goto end;
}
status = append_all_from_dir_info.status;
pthread_mutex_unlock(&append_all_from_dir_info.lock);
if (ret) {
- BT_LOGW_ERRNO("Failed to walk directory",
+ BT_LIB_LOGW_APPEND_CAUSE("Failed to walk directory",
": path=\"%s\", recurse=%d",
path, recurse);
status = BT_FUNC_STATUS_ERROR;
path, recurse);
*plugin_set_out = bt_plugin_set_create();
if (!*plugin_set_out) {
- BT_LOGE_STR("Cannot create empty plugin set.");
+ BT_LIB_LOGE_APPEND_CAUSE("Cannot create empty plugin set.");
status = BT_FUNC_STATUS_MEMORY_ERROR;
goto error;
}
* bt_plugin_create_append_all_from_dir() handles
* `fail_on_load_error`, so this is a "real" error.
*/
- BT_LOGW("Cannot append plugins found in directory: "
+ BT_LIB_LOGE_APPEND_CAUSE(
+ "Cannot append plugins found in directory: "
"path=\"%s\", status=%s",
path, bt_common_func_status_string(status));
goto error;
const char *bt_plugin_get_name(const struct bt_plugin *plugin)
{
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
return plugin->info.name_set ? plugin->info.name->str : NULL;
}
const char *bt_plugin_get_author(const struct bt_plugin *plugin)
{
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
return plugin->info.author_set ? plugin->info.author->str : NULL;
}
const char *bt_plugin_get_license(const struct bt_plugin *plugin)
{
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
return plugin->info.license_set ? plugin->info.license->str : NULL;
}
const char *bt_plugin_get_path(const struct bt_plugin *plugin)
{
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
return plugin->info.path_set ? plugin->info.path->str : NULL;
}
const char *bt_plugin_get_description(const struct bt_plugin *plugin)
{
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
return plugin->info.description_set ?
plugin->info.description->str : NULL;
}
enum bt_property_availability avail =
BT_PROPERTY_AVAILABILITY_AVAILABLE;
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
if (!plugin->info.version_set) {
BT_LIB_LOGD("Plugin's version is not set: %!+l", plugin);
uint64_t bt_plugin_get_source_component_class_count(const struct bt_plugin *plugin)
{
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
return (uint64_t) plugin->src_comp_classes->len;
}
uint64_t bt_plugin_get_filter_component_class_count(const struct bt_plugin *plugin)
{
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
return (uint64_t) plugin->flt_comp_classes->len;
}
uint64_t bt_plugin_get_sink_component_class_count(const struct bt_plugin *plugin)
{
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
return (uint64_t) plugin->sink_comp_classes->len;
}
const struct bt_plugin *plugin, GPtrArray *comp_classes,
uint64_t index)
{
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
- BT_ASSERT_PRE_VALID_INDEX(index, comp_classes->len);
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_VALID_INDEX(index, comp_classes->len);
return g_ptr_array_index(comp_classes, index);
}
struct bt_component_class *comp_class = NULL;
size_t i;
- BT_ASSERT_PRE_NON_NULL(plugin, "Plugin");
- BT_ASSERT_PRE_NON_NULL(name, "Name");
+ BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
+ BT_ASSERT_PRE_DEV_NON_NULL(name, "Name");
for (i = 0; i < comp_classes->len; i++) {
struct bt_component_class *comp_class_candidate =