Cleanup: add `#include <stdbool.h>` whenever `bool` type is used
[babeltrace.git] / src / lib / plugin / plugin.c
CommitLineData
33b34c43 1/*
e2f7325d 2 * Copyright 2017-2018 Philippe Proulx <pproulx@efficios.com>
33b34c43 3 * Copyright 2016 Jérémie Galarneau <jeremie.galarneau@efficios.com>
33b34c43
PP
4 *
5 * Author: Jérémie Galarneau <jeremie.galarneau@efficios.com>
6 *
7 * Permission is hereby granted, free of charge, to any person obtaining a copy
8 * of this software and associated documentation files (the "Software"), to deal
9 * in the Software without restriction, including without limitation the rights
10 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11 * copies of the Software, and to permit persons to whom the Software is
12 * furnished to do so, subject to the following conditions:
13 *
14 * The above copyright notice and this permission notice shall be included in
15 * all copies or substantial portions of the Software.
16 *
17 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
23 * SOFTWARE.
24 */
25
350ad6c1 26#define BT_LOG_TAG "LIB/PLUGIN"
c2d9d9cf 27#include "lib/logging.h"
8c1a3187 28
578e048b
MJ
29#include "common/assert.h"
30#include "lib/assert-pre.h"
91d81473 31#include "common/macros.h"
578e048b
MJ
32#include "compat/compiler.h"
33#include "common/common.h"
3fadfbc0
MJ
34#include <babeltrace2/plugin/plugin-const.h>
35#include <babeltrace2/graph/component-class-const.h>
870631a2 36#include <babeltrace2/current-thread.h>
578e048b 37#include "lib/graph/component-class.h"
3fadfbc0 38#include <babeltrace2/types.h>
33b34c43 39#include <glib.h>
33b34c43 40#include <unistd.h>
c4f23e30 41#include <stdbool.h>
33b34c43 42#include <stdlib.h>
9ac68eb1 43#include <stdint.h>
8c1a3187 44#include <inttypes.h>
33b34c43 45#include <sys/stat.h>
e1f4c4f7
MJ
46#include <ftw.h>
47#include <pthread.h>
33b34c43 48
578e048b
MJ
49#include "plugin.h"
50#include "plugin-so.h"
d24d5663 51#include "lib/func-status.h"
578e048b 52
b14c7bf1
MJ
53#define PYTHON_PLUGIN_PROVIDER_FILENAME "babeltrace2-python-plugin-provider." G_MODULE_SUFFIX
54#define PYTHON_PLUGIN_PROVIDER_DIR BABELTRACE_PLUGIN_PROVIDERS_DIR
6fbd4105 55#define PYTHON_PLUGIN_PROVIDER_SYM_NAME bt_plugin_python_create_all_from_file
91d81473 56#define PYTHON_PLUGIN_PROVIDER_SYM_NAME_STR G_STRINGIFY(PYTHON_PLUGIN_PROVIDER_SYM_NAME)
33b34c43 57
e1f4c4f7
MJ
58#define APPEND_ALL_FROM_DIR_NFDOPEN_MAX 8
59
7c7301d5
SM
60/* Declare here to make sure definition in both ifdef branches are in sync. */
61static
62int init_python_plugin_provider(void);
63typedef int (*create_all_from_file_sym_type)(
64 const char *path,
65 bool fail_on_load_error,
66 struct bt_plugin_set **plugin_set_out);
67
6fbd4105 68#ifdef BT_BUILT_IN_PYTHON_PLUGIN_SUPPORT
578e048b 69#include <plugin/python-plugin-provider.h>
e1f4c4f7 70
33b34c43 71static
7c7301d5
SM
72create_all_from_file_sym_type
73 bt_plugin_python_create_all_from_file_sym =
74 bt_plugin_python_create_all_from_file;
95ef44ce
MJ
75
76static
7c7301d5 77int init_python_plugin_provider(void)
9736d991
PP
78{
79}
6fbd4105
PP
80#else /* BT_BUILT_IN_PYTHON_PLUGIN_SUPPORT */
81static GModule *python_plugin_provider_module;
9736d991 82
6fbd4105 83static
7c7301d5 84create_all_from_file_sym_type bt_plugin_python_create_all_from_file_sym;
33b34c43 85
95ef44ce 86static
870631a2
PP
87int init_python_plugin_provider(void) {
88 int status = BT_FUNC_STATUS_OK;
b14c7bf1
MJ
89 const char *provider_dir_envvar;
90 static const char * const provider_dir_envvar_name = "LIBBABELTRACE2_PLUGIN_PROVIDER_DIR";
91 char *provider_path = NULL;
870631a2 92
5084732e 93 if (bt_plugin_python_create_all_from_file_sym) {
870631a2 94 goto end;
95ef44ce
MJ
95 }
96
3f7d4d90 97 BT_LOGI_STR("Loading Python plugin provider module.");
b14c7bf1
MJ
98
99 provider_dir_envvar = getenv(provider_dir_envvar_name);
100 if (provider_dir_envvar) {
101 provider_path = g_build_filename(provider_dir_envvar,
102 PYTHON_PLUGIN_PROVIDER_FILENAME, NULL);
103 BT_LOGI("Using `%s` environment variable to find the Python "
104 "plugin provider: path=\"%s\"", provider_dir_envvar_name,
105 provider_path);
106 } else {
107 provider_path = g_build_filename(PYTHON_PLUGIN_PROVIDER_DIR,
108 PYTHON_PLUGIN_PROVIDER_FILENAME, NULL);
109 BT_LOGI("Using default path (`%s` environment variable is not "
110 "set) to find the Python plugin provider: path=\"%s\"",
111 provider_dir_envvar_name, provider_path);
112 }
113
114 python_plugin_provider_module = g_module_open(provider_path, 0);
6fbd4105 115 if (!python_plugin_provider_module) {
9736d991
PP
116 /*
117 * This is not an error. The whole point of having an
118 * external Python plugin provider is that it can be
119 * missing and the Babeltrace library still works.
120 */
a77aed14 121 BT_LOGI("Cannot open `%s`: %s: continuing without Python plugin support.",
b14c7bf1 122 provider_path, g_module_error());
870631a2 123 goto end;
33b34c43
PP
124 }
125
6fbd4105
PP
126 if (!g_module_symbol(python_plugin_provider_module,
127 PYTHON_PLUGIN_PROVIDER_SYM_NAME_STR,
128 (gpointer) &bt_plugin_python_create_all_from_file_sym)) {
9736d991
PP
129 /*
130 * This is an error because, since we found the Python
131 * plugin provider shared object, we expect this symbol
132 * to exist.
133 */
870631a2
PP
134 BT_LIB_LOGE_APPEND_CAUSE(
135 "Cannot find the Python plugin provider loading symbol: "
9736d991 136 "%s: continuing without Python plugin support: "
8c1a3187 137 "file=\"%s\", symbol=\"%s\"",
9736d991 138 g_module_error(),
b14c7bf1 139 provider_path,
8c1a3187 140 PYTHON_PLUGIN_PROVIDER_SYM_NAME_STR);
870631a2
PP
141 status = BT_FUNC_STATUS_ERROR;
142 goto end;
33b34c43 143 }
8c1a3187 144
9e0bf9b0 145 BT_LOGI("Loaded Python plugin provider module: addr=%p",
8c1a3187 146 python_plugin_provider_module);
870631a2
PP
147
148end:
b14c7bf1
MJ
149 g_free(provider_path);
150
870631a2 151 return status;
33b34c43
PP
152}
153
6fbd4105
PP
154__attribute__((destructor)) static
155void fini_python_plugin_provider(void) {
156 if (python_plugin_provider_module) {
3f7d4d90 157 BT_LOGI("Unloading Python plugin provider module.");
8c1a3187
PP
158
159 if (!g_module_close(python_plugin_provider_module)) {
870631a2
PP
160 /*
161 * This occurs when the library is finalized: do
162 * NOT append an error cause.
163 */
8c1a3187
PP
164 BT_LOGE("Failed to close the Python plugin provider module: %s.",
165 g_module_error());
166 }
167
6fbd4105 168 python_plugin_provider_module = NULL;
6ba0b073 169 }
6ba0b073 170}
6fbd4105 171#endif
6ba0b073 172
e58aeae2 173uint64_t bt_plugin_set_get_plugin_count(const struct bt_plugin_set *plugin_set)
a8ff38ef 174{
bdb288b3 175 BT_ASSERT_PRE_DEV_NON_NULL(plugin_set, "Plugin set");
d94d92ac 176 return (uint64_t) plugin_set->plugins->len;
a8ff38ef
PP
177}
178
92fed4e1
PP
179const struct bt_plugin *bt_plugin_set_borrow_plugin_by_index_const(
180 const struct bt_plugin_set *plugin_set, uint64_t index)
a8ff38ef 181{
bdb288b3
PP
182 BT_ASSERT_PRE_DEV_NON_NULL(plugin_set, "Plugin set");
183 BT_ASSERT_PRE_DEV_VALID_INDEX(index, plugin_set->plugins->len);
d94d92ac 184 return g_ptr_array_index(plugin_set->plugins, index);
a8ff38ef
PP
185}
186
d24d5663 187enum bt_plugin_find_all_from_static_status bt_plugin_find_all_from_static(
9736d991
PP
188 bt_bool fail_on_load_error,
189 const struct bt_plugin_set **plugin_set_out)
6ba0b073 190{
8c1a3187 191 /* bt_plugin_so_create_all_from_static() logs errors */
9736d991
PP
192 return bt_plugin_so_create_all_from_static(fail_on_load_error,
193 (void *) plugin_set_out);
6ba0b073
PP
194}
195
d24d5663
PP
196enum bt_plugin_find_all_from_file_status bt_plugin_find_all_from_file(
197 const char *path, bt_bool fail_on_load_error,
9736d991 198 const struct bt_plugin_set **plugin_set_out)
33b34c43 199{
d24d5663 200 enum bt_plugin_find_all_from_file_status status;
33b34c43 201
d94d92ac 202 BT_ASSERT_PRE_NON_NULL(path, "Path");
9736d991 203 BT_ASSERT_PRE_NON_NULL(path, "Plugin set (output)");
3f7d4d90 204 BT_LOGI("Creating plugins from file: path=\"%s\"", path);
6ba0b073 205
55bb57e0 206 /* Try shared object plugins */
9736d991
PP
207 status = bt_plugin_so_create_all_from_file(path, fail_on_load_error,
208 (void *) plugin_set_out);
d24d5663 209 if (status == BT_FUNC_STATUS_OK) {
9736d991
PP
210 BT_ASSERT(*plugin_set_out);
211 BT_ASSERT((*plugin_set_out)->plugins->len > 0);
212 goto end;
213 } else if (status < 0) {
214 BT_ASSERT(!*plugin_set_out);
6ba0b073
PP
215 goto end;
216 }
217
d24d5663 218 BT_ASSERT(status == BT_FUNC_STATUS_NOT_FOUND);
9736d991
PP
219 BT_ASSERT(!*plugin_set_out);
220
6fbd4105 221 /* Try Python plugins if support is available */
870631a2
PP
222 status = init_python_plugin_provider();
223 if (status < 0) {
224 /* init_python_plugin_provider() logs errors */
225 goto end;
226 }
227
228 BT_ASSERT(status == BT_FUNC_STATUS_OK);
229 status = BT_FUNC_STATUS_NOT_FOUND;
230
6fbd4105 231 if (bt_plugin_python_create_all_from_file_sym) {
870631a2 232 /* Python plugin provider exists */
9736d991
PP
233 status = bt_plugin_python_create_all_from_file_sym(path,
234 fail_on_load_error, (void *) plugin_set_out);
d24d5663 235 if (status == BT_FUNC_STATUS_OK) {
9736d991
PP
236 BT_ASSERT(*plugin_set_out);
237 BT_ASSERT((*plugin_set_out)->plugins->len > 0);
238 goto end;
239 } else if (status < 0) {
870631a2
PP
240 /*
241 * bt_plugin_python_create_all_from_file_sym()
242 * handles `fail_on_load_error` itself, so this
243 * is a "real" error.
244 */
9736d991 245 BT_ASSERT(!*plugin_set_out);
6fbd4105
PP
246 goto end;
247 }
9736d991 248
d24d5663 249 BT_ASSERT(status == BT_FUNC_STATUS_NOT_FOUND);
9736d991 250 BT_ASSERT(!*plugin_set_out);
6ba0b073
PP
251 }
252
33b34c43 253end:
d24d5663 254 if (status == BT_FUNC_STATUS_OK) {
3f7d4d90 255 BT_LOGI("Created %u plugins from file: "
8c1a3187 256 "path=\"%s\", count=%u, plugin-set-addr=%p",
9736d991
PP
257 (*plugin_set_out)->plugins->len, path,
258 (*plugin_set_out)->plugins->len,
259 *plugin_set_out);
d24d5663 260 } else if (status == BT_FUNC_STATUS_NOT_FOUND) {
3f7d4d90 261 BT_LOGI("Found no plugins in file: path=\"%s\"", path);
8c1a3187
PP
262 }
263
9736d991 264 return status;
33b34c43
PP
265}
266
577fa92f
PP
267static
268void destroy_gstring(void *data)
1670bffd
PP
269{
270 g_string_free(data, TRUE);
271}
272
577fa92f
PP
273enum bt_plugin_find_all_status bt_plugin_find_all(bt_bool find_in_std_env_var,
274 bt_bool find_in_user_dir, bt_bool find_in_sys_dir,
275 bt_bool find_in_static, bt_bool fail_on_load_error,
276 const struct bt_plugin_set **plugin_set_out)
1670bffd 277{
1670bffd 278 char *home_plugin_dir = NULL;
92fed4e1 279 const struct bt_plugin_set *plugin_set = NULL;
1670bffd
PP
280 GPtrArray *dirs = NULL;
281 int ret;
d24d5663 282 int status = BT_FUNC_STATUS_OK;
577fa92f
PP
283 uint64_t dir_i, plugin_i;
284
285 BT_ASSERT_PRE_NON_NULL(plugin_set_out, "Plugin set (output)");
286 BT_LOGI("Finding all plugins in standard directories and built-in plugins: "
287 "find-in-std-env-var=%d, find-in-user-dir=%d, "
288 "find-in-sys-dir=%d, find-in-static=%d",
289 find_in_std_env_var, find_in_user_dir, find_in_sys_dir,
290 find_in_static);
1670bffd
PP
291 dirs = g_ptr_array_new_with_free_func((GDestroyNotify) destroy_gstring);
292 if (!dirs) {
870631a2 293 BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GPtrArray.");
d24d5663 294 status = BT_FUNC_STATUS_MEMORY_ERROR;
1670bffd
PP
295 goto end;
296 }
297
577fa92f
PP
298 *plugin_set_out = bt_plugin_set_create();
299 if (!*plugin_set_out) {
300 BT_LIB_LOGE_APPEND_CAUSE("Cannot create empty plugin set.");
301 status = BT_FUNC_STATUS_MEMORY_ERROR;
302 goto end;
303 }
304
1670bffd
PP
305 /*
306 * Search order is:
307 *
577fa92f 308 * 1. `BABELTRACE_PLUGIN_PATH` environment variable
1670bffd 309 * (colon-separated list of directories)
577fa92f 310 * 2. `~/.local/lib/babeltrace2/plugins`
1670bffd 311 * 3. Default system directory for Babeltrace plugins, usually
577fa92f
PP
312 * `/usr/lib/babeltrace2/plugins` or
313 * `/usr/local/lib/babeltrace2/plugins` if installed locally
1670bffd
PP
314 * 4. Built-in plugins (static)
315 *
316 * Directories are searched non-recursively.
317 */
577fa92f
PP
318 if (find_in_std_env_var) {
319 const char *envvar = getenv("BABELTRACE_PLUGIN_PATH");
320
321 if (envvar) {
322 ret = bt_common_append_plugin_path_dirs(envvar, dirs);
323 if (ret) {
324 BT_LIB_LOGE_APPEND_CAUSE(
325 "Failed to append plugin path to array of directories.");
326 status = BT_FUNC_STATUS_MEMORY_ERROR;
327 goto end;
328 }
1670bffd
PP
329 }
330 }
331
577fa92f
PP
332 if (find_in_user_dir) {
333 home_plugin_dir = bt_common_get_home_plugin_path(
334 BT_LOG_OUTPUT_LEVEL);
335 if (home_plugin_dir) {
336 GString *home_plugin_dir_str = g_string_new(
337 home_plugin_dir);
1670bffd 338
577fa92f
PP
339 if (!home_plugin_dir_str) {
340 BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
341 status = BT_FUNC_STATUS_MEMORY_ERROR;
342 goto end;
343 }
1670bffd 344
577fa92f
PP
345 g_ptr_array_add(dirs, home_plugin_dir_str);
346 }
1670bffd
PP
347 }
348
577fa92f
PP
349 if (find_in_sys_dir) {
350 const char *system_plugin_dir =
351 bt_common_get_system_plugin_path();
1670bffd 352
577fa92f
PP
353 if (system_plugin_dir) {
354 GString *system_plugin_dir_str =
355 g_string_new(system_plugin_dir);
356
357 if (!system_plugin_dir_str) {
358 BT_LIB_LOGE_APPEND_CAUSE("Failed to allocate a GString.");
359 status = BT_FUNC_STATUS_MEMORY_ERROR;
360 goto end;
361 }
1670bffd 362
577fa92f
PP
363 g_ptr_array_add(dirs, system_plugin_dir_str);
364 }
1670bffd
PP
365 }
366
577fa92f
PP
367 for (dir_i = 0; dir_i < dirs->len; dir_i++) {
368 GString *dir = dirs->pdata[dir_i];
1670bffd 369
65300d60 370 BT_OBJECT_PUT_REF_AND_RESET(plugin_set);
8c1a3187 371
50ad9320
PP
372 /*
373 * Skip this if the directory does not exist because
c8db3219 374 * bt_plugin_find_all_from_dir() would log a warning.
50ad9320
PP
375 */
376 if (!g_file_test(dir->str, G_FILE_TEST_IS_DIR)) {
3f7d4d90 377 BT_LOGI("Skipping nonexistent directory path: "
50ad9320
PP
378 "path=\"%s\"", dir->str);
379 continue;
380 }
381
c8db3219 382 /* bt_plugin_find_all_from_dir() logs details/errors */
9736d991
PP
383 status = bt_plugin_find_all_from_dir(dir->str, BT_FALSE,
384 fail_on_load_error, &plugin_set);
385 if (status < 0) {
386 BT_ASSERT(!plugin_set);
387 goto end;
d24d5663 388 } else if (status == BT_FUNC_STATUS_NOT_FOUND) {
9736d991 389 BT_ASSERT(!plugin_set);
3f7d4d90 390 BT_LOGI("No plugins found in directory: path=\"%s\"",
8c1a3187 391 dir->str);
1670bffd
PP
392 continue;
393 }
394
d24d5663 395 BT_ASSERT(status == BT_FUNC_STATUS_OK);
9736d991 396 BT_ASSERT(plugin_set);
577fa92f
PP
397 BT_LOGI("Found plugins in directory: path=\"%s\", count=%u",
398 dir->str, plugin_set->plugins->len);
9736d991 399
577fa92f
PP
400 for (plugin_i = 0; plugin_i < plugin_set->plugins->len;
401 plugin_i++) {
402 bt_plugin_set_add_plugin((void *) *plugin_set_out,
403 plugin_set->plugins->pdata[plugin_i]);
404 }
405 }
1670bffd 406
577fa92f
PP
407 if (find_in_static) {
408 BT_OBJECT_PUT_REF_AND_RESET(plugin_set);
409 status = bt_plugin_find_all_from_static(fail_on_load_error,
410 &plugin_set);
411 if (status < 0) {
412 BT_ASSERT(!plugin_set);
413 goto end;
414 } else if (status == BT_FUNC_STATUS_NOT_FOUND) {
415 BT_ASSERT(!plugin_set);
416 BT_LOGI_STR("No plugins found in built-in plugins.");
417 goto end;
1670bffd 418 }
8c1a3187 419
577fa92f
PP
420 BT_ASSERT(status == BT_FUNC_STATUS_OK);
421 BT_ASSERT(plugin_set);
422 BT_LOGI("Found built-in plugins: count=%u",
423 plugin_set->plugins->len);
424
425 for (plugin_i = 0; plugin_i < plugin_set->plugins->len;
426 plugin_i++) {
427 bt_plugin_set_add_plugin((void *) *plugin_set_out,
428 plugin_set->plugins->pdata[plugin_i]);
429 }
430 }
431
432end:
433 free(home_plugin_dir);
434 bt_object_put_ref(plugin_set);
435
436 if (dirs) {
437 g_ptr_array_free(dirs, TRUE);
1670bffd
PP
438 }
439
9736d991 440 if (status < 0) {
577fa92f
PP
441 BT_OBJECT_PUT_REF_AND_RESET(*plugin_set_out);
442 } else {
443 BT_ASSERT(*plugin_set_out);
444
445 if ((*plugin_set_out)->plugins->len > 0) {
446 BT_LOGI("Found plugins in standard directories and built-in plugins: "
447 "count=%u", (*plugin_set_out)->plugins->len);
448 status = BT_FUNC_STATUS_OK;
449 } else {
450 BT_LOGI_STR("No plugins found in standard directories and built-in plugins.");
451 status = BT_FUNC_STATUS_NOT_FOUND;
452 BT_OBJECT_PUT_REF_AND_RESET(*plugin_set_out);
453 }
454 }
455
456 return status;
457}
458
459enum bt_plugin_find_status bt_plugin_find(const char *plugin_name,
460 bt_bool find_in_std_env_var, bt_bool find_in_user_dir,
461 bt_bool find_in_sys_dir, bt_bool find_in_static,
462 bt_bool fail_on_load_error, const struct bt_plugin **plugin_out)
463{
464 enum bt_plugin_find_status status;
465 const struct bt_plugin_set *plugin_set = NULL;
466 uint64_t i;
467
468 BT_ASSERT_PRE_NON_NULL(plugin_name, "Name");
469 BT_ASSERT_PRE_NON_NULL(plugin_out, "Plugin (output)");
470 BT_LOGI("Finding named plugin in standard directories and built-in plugins: "
471 "name=\"%s\", find-in-std-env-var=%d, find-in-user-dir=%d, "
472 "find-in-sys-dir=%d, find-in-static=%d",
473 plugin_name, find_in_std_env_var, find_in_user_dir,
474 find_in_sys_dir, find_in_static);
7aa2e7df 475 status = (enum bt_plugin_find_status) bt_plugin_find_all(find_in_std_env_var, find_in_user_dir,
577fa92f
PP
476 find_in_sys_dir, find_in_static, fail_on_load_error,
477 &plugin_set);
478 if (status != BT_FUNC_STATUS_OK) {
f600f713 479 BT_ASSERT(!plugin_set);
9736d991
PP
480 goto end;
481 }
482
9736d991
PP
483 BT_ASSERT(plugin_set);
484
577fa92f
PP
485 for (i = 0; i < plugin_set->plugins->len; i++) {
486 const struct bt_plugin *plugin = plugin_set->plugins->pdata[i];
9736d991 487
577fa92f
PP
488 if (strcmp(plugin->info.name->str, plugin_name) == 0) {
489 *plugin_out = plugin;
6871026b 490 bt_object_get_ref_no_null_check(*plugin_out);
9736d991 491 goto end;
1670bffd 492 }
1670bffd
PP
493 }
494
d24d5663 495 status = BT_FUNC_STATUS_NOT_FOUND;
9736d991 496
1670bffd 497end:
d24d5663 498 if (status == BT_FUNC_STATUS_OK) {
577fa92f 499 BT_ASSERT(*plugin_out);
3f7d4d90 500 BT_LIB_LOGI("Found plugin in standard directories and built-in plugins: "
577fa92f
PP
501 "%!+l", *plugin_out);
502 } else if (status == BT_FUNC_STATUS_NOT_FOUND) {
3f7d4d90 503 BT_LOGI("No plugin found in standard directories and built-in plugins: "
8c1a3187
PP
504 "name=\"%s\"", plugin_name);
505 }
506
f600f713
FD
507 bt_plugin_set_put_ref(plugin_set);
508
9736d991 509 return status;
1670bffd
PP
510}
511
e1f4c4f7
MJ
512static struct {
513 pthread_mutex_t lock;
514 struct bt_plugin_set *plugin_set;
d94d92ac 515 bool recurse;
9736d991 516 bool fail_on_load_error;
d24d5663 517 int status;
e1f4c4f7
MJ
518} append_all_from_dir_info = {
519 .lock = PTHREAD_MUTEX_INITIALIZER
520};
521
33b34c43 522static
9736d991
PP
523int nftw_append_all_from_dir(const char *file,
524 const struct stat *sb, int flag, struct FTW *s)
33b34c43 525{
e1f4c4f7
MJ
526 int ret = 0;
527 const char *name = file + s->base;
528
529 /* Check for recursion */
530 if (!append_all_from_dir_info.recurse && s->level > 1) {
531 goto end;
532 }
533
534 switch (flag) {
535 case FTW_F:
52238017 536 {
9736d991 537 const struct bt_plugin_set *plugins_from_file = NULL;
52238017 538
e1f4c4f7
MJ
539 if (name[0] == '.') {
540 /* Skip hidden files */
3f7d4d90 541 BT_LOGI("Skipping hidden file: path=\"%s\"", file);
e1f4c4f7
MJ
542 goto end;
543 }
52238017 544
9736d991
PP
545 append_all_from_dir_info.status =
546 bt_plugin_find_all_from_file(file,
547 append_all_from_dir_info.fail_on_load_error,
548 &plugins_from_file);
d24d5663 549 if (append_all_from_dir_info.status == BT_FUNC_STATUS_OK) {
e1f4c4f7
MJ
550 size_t j;
551
9736d991
PP
552 BT_ASSERT(plugins_from_file);
553
e1f4c4f7
MJ
554 for (j = 0; j < plugins_from_file->plugins->len; j++) {
555 struct bt_plugin *plugin =
556 g_ptr_array_index(plugins_from_file->plugins, j);
557
3f7d4d90 558 BT_LIB_LOGI("Adding plugin to plugin set: "
d94d92ac
PP
559 "plugin-path=\"%s\", %![plugin-]+l",
560 file, plugin);
92fed4e1
PP
561 bt_plugin_set_add_plugin(
562 append_all_from_dir_info.plugin_set,
563 plugin);
e1f4c4f7 564 }
33b34c43 565
65300d60 566 bt_object_put_ref(plugins_from_file);
9736d991
PP
567 goto end;
568 } else if (append_all_from_dir_info.status < 0) {
569 /* bt_plugin_find_all_from_file() logs errors */
570 BT_ASSERT(!plugins_from_file);
571 ret = -1;
572 goto end;
e1f4c4f7 573 }
9736d991
PP
574
575 /*
576 * Not found in this file: this is no an error; continue
577 * walking the directories.
578 */
579 BT_ASSERT(!plugins_from_file);
580 BT_ASSERT(append_all_from_dir_info.status ==
d24d5663 581 BT_FUNC_STATUS_NOT_FOUND);
e1f4c4f7 582 break;
52238017 583 }
e1f4c4f7
MJ
584 case FTW_DNR:
585 /* Continue to next file / directory. */
9736d991 586 BT_LOGI("Cannot enter directory: continuing: path=\"%s\"", file);
e1f4c4f7
MJ
587 break;
588 case FTW_NS:
589 /* Continue to next file / directory. */
3f7d4d90 590 BT_LOGI("Cannot get file information: continuing: path=\"%s\"", file);
e1f4c4f7 591 break;
33b34c43 592 }
e1f4c4f7
MJ
593
594end:
595 return ret;
33b34c43
PP
596}
597
598static
d24d5663
PP
599int bt_plugin_create_append_all_from_dir(struct bt_plugin_set *plugin_set,
600 const char *path, bt_bool recurse, bt_bool fail_on_load_error)
33b34c43 601{
e1f4c4f7 602 int nftw_flags = FTW_PHYS;
9736d991 603 int ret;
d24d5663 604 int status;
037cbd1a 605 struct stat sb;
33b34c43 606
d94d92ac
PP
607 BT_ASSERT(plugin_set);
608 BT_ASSERT(path);
609 BT_ASSERT(strlen(path) < PATH_MAX);
037cbd1a
JR
610
611 /*
612 * Make sure that path exists and is accessible.
613 * This is necessary since Cygwin implementation of nftw() is not POSIX
614 * compliant. Cygwin nftw() implementation does not fail on non-existent
615 * path with ENOENT. Instead, it flags the directory as FTW_NS. FTW_NS during
616 * nftw_append_all_from_dir is not treated as an error since we are
617 * traversing the tree for plugin discovery.
618 */
619 if (stat(path, &sb)) {
620 BT_LOGW_ERRNO("Cannot open directory",
621 ": path=\"%s\", recurse=%d",
622 path, recurse);
870631a2
PP
623 (void) BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(
624 "Babeltrace library",
625 "Cannot open directory: path=\"%s\", recurse=%d",
626 path, recurse);
d24d5663 627 status = BT_FUNC_STATUS_ERROR;
037cbd1a
JR
628 goto end;
629 }
630
e1f4c4f7 631 pthread_mutex_lock(&append_all_from_dir_info.lock);
e1f4c4f7
MJ
632 append_all_from_dir_info.plugin_set = plugin_set;
633 append_all_from_dir_info.recurse = recurse;
d24d5663 634 append_all_from_dir_info.status = BT_FUNC_STATUS_OK;
9736d991 635 append_all_from_dir_info.fail_on_load_error = fail_on_load_error;
e1f4c4f7
MJ
636 ret = nftw(path, nftw_append_all_from_dir,
637 APPEND_ALL_FROM_DIR_NFDOPEN_MAX, nftw_flags);
9736d991
PP
638 append_all_from_dir_info.plugin_set = NULL;
639 status = append_all_from_dir_info.status;
e1f4c4f7 640 pthread_mutex_unlock(&append_all_from_dir_info.lock);
9736d991 641 if (ret) {
870631a2 642 BT_LIB_LOGW_APPEND_CAUSE("Failed to walk directory",
9736d991
PP
643 ": path=\"%s\", recurse=%d",
644 path, recurse);
d24d5663 645 status = BT_FUNC_STATUS_ERROR;
9736d991
PP
646 goto end;
647 }
648
d24d5663 649 if (status == BT_FUNC_STATUS_NOT_FOUND) {
9736d991
PP
650 /*
651 * We're just appending in this function; even if
652 * nothing was found, it's still okay from the caller's
653 * perspective.
654 */
d24d5663 655 status = BT_FUNC_STATUS_OK;
33b34c43 656 }
8c1a3187 657
9736d991 658end:
060cf347 659 return status;
33b34c43
PP
660}
661
d24d5663
PP
662enum bt_plugin_find_all_from_dir_status bt_plugin_find_all_from_dir(
663 const char *path, bt_bool recurse, bt_bool fail_on_load_error,
9736d991 664 const struct bt_plugin_set **plugin_set_out)
33b34c43 665{
d24d5663
PP
666 enum bt_plugin_find_all_from_dir_status status =
667 BT_FUNC_STATUS_OK;
33b34c43 668
9736d991 669 BT_ASSERT_PRE_NON_NULL(plugin_set_out, "Plugin set (output)");
3f7d4d90 670 BT_LOGI("Creating all plugins in directory: path=\"%s\", recurse=%d",
8c1a3187 671 path, recurse);
9736d991
PP
672 *plugin_set_out = bt_plugin_set_create();
673 if (!*plugin_set_out) {
870631a2 674 BT_LIB_LOGE_APPEND_CAUSE("Cannot create empty plugin set.");
d24d5663 675 status = BT_FUNC_STATUS_MEMORY_ERROR;
a8ff38ef
PP
676 goto error;
677 }
678
9736d991
PP
679 /*
680 * Append found plugins to array (never returns
d24d5663 681 * `BT_FUNC_STATUS_NOT_FOUND`)
9736d991
PP
682 */
683 status = bt_plugin_create_append_all_from_dir((void *) *plugin_set_out,
684 path, recurse, fail_on_load_error);
33b34c43 685 if (status < 0) {
9736d991
PP
686 /*
687 * bt_plugin_create_append_all_from_dir() handles
688 * `fail_on_load_error`, so this is a "real" error.
689 */
870631a2
PP
690 BT_LIB_LOGE_APPEND_CAUSE(
691 "Cannot append plugins found in directory: "
dbd7f7e9 692 "path=\"%s\", status=%s",
d24d5663 693 path, bt_common_func_status_string(status));
33b34c43
PP
694 goto error;
695 }
696
d24d5663 697 BT_ASSERT(status == BT_FUNC_STATUS_OK);
9736d991
PP
698
699 if ((*plugin_set_out)->plugins->len == 0) {
700 /* Nothing was appended: not found */
701 BT_LOGI("No plugins found in directory: path=\"%s\"", path);
d24d5663 702 status = BT_FUNC_STATUS_NOT_FOUND;
9736d991
PP
703 goto error;
704 }
705
3f7d4d90 706 BT_LOGI("Created %u plugins from directory: count=%u, path=\"%s\"",
9736d991
PP
707 (*plugin_set_out)->plugins->len,
708 (*plugin_set_out)->plugins->len, path);
33b34c43
PP
709 goto end;
710
711error:
d24d5663 712 BT_ASSERT(status != BT_FUNC_STATUS_OK);
9736d991 713 BT_OBJECT_PUT_REF_AND_RESET(*plugin_set_out);
33b34c43
PP
714
715end:
9736d991 716 return status;
33b34c43
PP
717}
718
92fed4e1 719const char *bt_plugin_get_name(const struct bt_plugin *plugin)
33b34c43 720{
bdb288b3 721 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
d94d92ac 722 return plugin->info.name_set ? plugin->info.name->str : NULL;
33b34c43
PP
723}
724
92fed4e1 725const char *bt_plugin_get_author(const struct bt_plugin *plugin)
33b34c43 726{
bdb288b3 727 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
d94d92ac 728 return plugin->info.author_set ? plugin->info.author->str : NULL;
33b34c43
PP
729}
730
92fed4e1 731const char *bt_plugin_get_license(const struct bt_plugin *plugin)
33b34c43 732{
bdb288b3 733 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
d94d92ac 734 return plugin->info.license_set ? plugin->info.license->str : NULL;
33b34c43
PP
735}
736
92fed4e1 737const char *bt_plugin_get_path(const struct bt_plugin *plugin)
33b34c43 738{
bdb288b3 739 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
d94d92ac 740 return plugin->info.path_set ? plugin->info.path->str : NULL;
33b34c43
PP
741}
742
92fed4e1 743const char *bt_plugin_get_description(const struct bt_plugin *plugin)
33b34c43 744{
bdb288b3 745 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
d94d92ac
PP
746 return plugin->info.description_set ?
747 plugin->info.description->str : NULL;
33b34c43
PP
748}
749
92fed4e1 750enum bt_property_availability bt_plugin_get_version(const struct bt_plugin *plugin,
b6de043b
PP
751 unsigned int *major, unsigned int *minor, unsigned int *patch,
752 const char **extra)
753{
d94d92ac
PP
754 enum bt_property_availability avail =
755 BT_PROPERTY_AVAILABILITY_AVAILABLE;
b6de043b 756
bdb288b3 757 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
8c1a3187
PP
758
759 if (!plugin->info.version_set) {
3f7d4d90 760 BT_LIB_LOGD("Plugin's version is not set: %!+l", plugin);
d94d92ac 761 avail = BT_PROPERTY_AVAILABILITY_NOT_AVAILABLE;
b6de043b
PP
762 goto end;
763 }
764
765 if (major) {
55bb57e0 766 *major = plugin->info.version.major;
b6de043b
PP
767 }
768
769 if (minor) {
55bb57e0 770 *minor = plugin->info.version.minor;
b6de043b
PP
771 }
772
773 if (patch) {
55bb57e0 774 *patch = plugin->info.version.patch;
b6de043b
PP
775 }
776
777 if (extra) {
55bb57e0 778 *extra = plugin->info.version.extra->str;
b6de043b
PP
779 }
780
781end:
d94d92ac 782 return avail;
b6de043b
PP
783}
784
92fed4e1 785uint64_t bt_plugin_get_source_component_class_count(const struct bt_plugin *plugin)
33b34c43 786{
bdb288b3 787 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
d94d92ac 788 return (uint64_t) plugin->src_comp_classes->len;
33b34c43
PP
789}
790
92fed4e1 791uint64_t bt_plugin_get_filter_component_class_count(const struct bt_plugin *plugin)
33b34c43 792{
bdb288b3 793 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
d94d92ac
PP
794 return (uint64_t) plugin->flt_comp_classes->len;
795}
33b34c43 796
92fed4e1 797uint64_t bt_plugin_get_sink_component_class_count(const struct bt_plugin *plugin)
d94d92ac 798{
bdb288b3 799 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
d94d92ac
PP
800 return (uint64_t) plugin->sink_comp_classes->len;
801}
8c1a3187 802
d94d92ac
PP
803static inline
804struct bt_component_class *borrow_component_class_by_index(
92fed4e1 805 const struct bt_plugin *plugin, GPtrArray *comp_classes,
d94d92ac
PP
806 uint64_t index)
807{
bdb288b3
PP
808 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
809 BT_ASSERT_PRE_DEV_VALID_INDEX(index, comp_classes->len);
d94d92ac
PP
810 return g_ptr_array_index(comp_classes, index);
811}
33b34c43 812
0d72b8c3 813const struct bt_component_class_source *
3f79b5cf 814bt_plugin_borrow_source_component_class_by_index_const(
92fed4e1 815 const struct bt_plugin *plugin, uint64_t index)
d94d92ac 816{
0d72b8c3 817 return (const void *) borrow_component_class_by_index(plugin,
d94d92ac
PP
818 plugin->src_comp_classes, index);
819}
33b34c43 820
0d72b8c3 821const struct bt_component_class_filter *
92fed4e1
PP
822bt_plugin_borrow_filter_component_class_by_index_const(
823 const struct bt_plugin *plugin, uint64_t index)
d94d92ac 824{
0d72b8c3 825 return (const void *) borrow_component_class_by_index(plugin,
d94d92ac
PP
826 plugin->flt_comp_classes, index);
827}
828
0d72b8c3 829const struct bt_component_class_sink *
92fed4e1
PP
830bt_plugin_borrow_sink_component_class_by_index_const(
831 const struct bt_plugin *plugin, uint64_t index)
d94d92ac 832{
0d72b8c3 833 return (const void *) borrow_component_class_by_index(plugin,
d94d92ac 834 plugin->sink_comp_classes, index);
33b34c43
PP
835}
836
d94d92ac
PP
837static inline
838struct bt_component_class *borrow_component_class_by_name(
92fed4e1 839 const struct bt_plugin *plugin, GPtrArray *comp_classes,
d94d92ac 840 const char *name)
33b34c43
PP
841{
842 struct bt_component_class *comp_class = NULL;
843 size_t i;
844
bdb288b3
PP
845 BT_ASSERT_PRE_DEV_NON_NULL(plugin, "Plugin");
846 BT_ASSERT_PRE_DEV_NON_NULL(name, "Name");
33b34c43 847
d94d92ac 848 for (i = 0; i < comp_classes->len; i++) {
33b34c43 849 struct bt_component_class *comp_class_candidate =
d94d92ac 850 g_ptr_array_index(comp_classes, i);
33b34c43
PP
851 const char *comp_class_cand_name =
852 bt_component_class_get_name(comp_class_candidate);
33b34c43 853
98b15851 854 BT_ASSERT_DBG(comp_class_cand_name);
33b34c43 855
d94d92ac
PP
856 if (strcmp(name, comp_class_cand_name) == 0) {
857 comp_class = comp_class_candidate;
33b34c43
PP
858 break;
859 }
860 }
861
33b34c43
PP
862 return comp_class;
863}
864
0d72b8c3 865const struct bt_component_class_source *
92fed4e1
PP
866bt_plugin_borrow_source_component_class_by_name_const(
867 const struct bt_plugin *plugin, const char *name)
33b34c43 868{
0d72b8c3 869 return (const void *) borrow_component_class_by_name(plugin,
d94d92ac
PP
870 plugin->src_comp_classes, name);
871}
33b34c43 872
0d72b8c3 873const struct bt_component_class_filter *
92fed4e1
PP
874bt_plugin_borrow_filter_component_class_by_name_const(
875 const struct bt_plugin *plugin, const char *name)
d94d92ac 876{
0d72b8c3 877 return (const void *) borrow_component_class_by_name(plugin,
d94d92ac
PP
878 plugin->flt_comp_classes, name);
879}
33b34c43 880
0d72b8c3 881const struct bt_component_class_sink *
92fed4e1
PP
882bt_plugin_borrow_sink_component_class_by_name_const(
883 const struct bt_plugin *plugin, const char *name)
d94d92ac 884{
0d72b8c3 885 return (const void *) borrow_component_class_by_name(plugin,
d94d92ac 886 plugin->sink_comp_classes, name);
33b34c43 887}
c5b9b441
PP
888
889void bt_plugin_get_ref(const struct bt_plugin *plugin)
890{
891 bt_object_get_ref(plugin);
892}
893
894void bt_plugin_put_ref(const struct bt_plugin *plugin)
895{
896 bt_object_put_ref(plugin);
897}
898
899void bt_plugin_set_get_ref(const struct bt_plugin_set *plugin_set)
900{
901 bt_object_get_ref(plugin_set);
902}
903
904void bt_plugin_set_put_ref(const struct bt_plugin_set *plugin_set)
905{
906 bt_object_put_ref(plugin_set);
907}
This page took 0.116028 seconds and 4 git commands to generate.