Move to kernel style SPDX license identifiers
[babeltrace.git] / src / python-plugin-provider / python-plugin-provider.c
index b4bbff475421326b97983da83cc01022d6d8322a..78026d6a1fd87e6ddb6883c0ff61fae4eecc964b 100644 (file)
@@ -1,37 +1,25 @@
 /*
- * python-plugin-provider.c
- *
- * Babeltrace Python plugin provider
+ * SPDX-License-Identifier: MIT
  *
  * Copyright 2017 Philippe Proulx <pproulx@efficios.com>
  *
- * Permission is hereby granted, free of charge, to any person obtaining a copy
- * of this software and associated documentation files (the "Software"), to deal
- * in the Software without restriction, including without limitation the rights
- * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- * copies of the Software, and to permit persons to whom the Software is
- * furnished to do so, subject to the following conditions:
- *
- * The above copyright notice and this permission notice shall be included in
- * all copies or substantial portions of the Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- * SOFTWARE.
+ * Babeltrace Python plugin provider
  */
 
-#define BT_LOG_TAG "PLUGIN-PY"
+#define BT_LOG_TAG "LIB/PLUGIN-PY"
+#include "lib/logging.h"
+
+#include "python-plugin-provider.h"
 
 #include "common/macros.h"
 #include "compat/compiler.h"
-#include <babeltrace2/plugin/plugin-const.h>
+#include <babeltrace2/plugin/plugin-loading.h>
 #include "lib/plugin/plugin.h"
 #include <babeltrace2/graph/component-class.h>
+#include <babeltrace2/error-reporting.h>
 #include "lib/graph/component-class.h"
+#include "py-common/py-common.h"
+#include <stdbool.h>
 #include <stdlib.h>
 #include <signal.h>
 #include <Python.h>
@@ -43,7 +31,7 @@
 #define PYTHON_PLUGIN_FILE_EXT         ".py"
 #define PYTHON_PLUGIN_FILE_EXT_LEN     (sizeof(PYTHON_PLUGIN_FILE_EXT) - 1)
 
-enum python_state {
+static enum python_state {
        /* init_python() not called yet */
        PYTHON_STATE_NOT_INITED,
 
@@ -52,17 +40,58 @@ enum python_state {
 
        /* init_python() called once without success */
        PYTHON_STATE_CANNOT_INITIALIZE,
+
+       /*
+        * init_python() called, but environment variable asks the
+        * Python interpreter not to be loaded.
+        */
+       PYTHON_STATE_WONT_INITIALIZE,
 } python_state = PYTHON_STATE_NOT_INITED;
 
 static PyObject *py_try_load_plugin_module_func = NULL;
 static bool python_was_initialized_by_us;
 
 static
-void print_python_traceback_warn(void)
+void append_python_traceback_error_cause(void)
 {
-       if (BT_LOG_ON_WARN && Py_IsInitialized() && PyErr_Occurred()) {
-               BT_LOGW_STR("Exception occured: traceback: ");
-               PyErr_Print();
+       GString *exc = NULL;
+
+       if (Py_IsInitialized() && PyErr_Occurred()) {
+               exc = bt_py_common_format_current_exception(BT_LOG_OUTPUT_LEVEL);
+               if (!exc) {
+                       BT_LOGE_STR("Failed to format Python exception.");
+                       goto end;
+               }
+
+               (void) BT_CURRENT_THREAD_ERROR_APPEND_CAUSE_FROM_UNKNOWN(
+                       BT_LIB_LOG_LIBBABELTRACE2_NAME, "%s", exc->str);
+       }
+
+end:
+       if (exc) {
+               g_string_free(exc, TRUE);
+       }
+}
+
+static
+void log_python_traceback(int log_level)
+{
+       GString *exc = NULL;
+
+       if (Py_IsInitialized() && PyErr_Occurred()) {
+               exc = bt_py_common_format_current_exception(BT_LOG_OUTPUT_LEVEL);
+               if (!exc) {
+                       BT_LOGE_STR("Failed to format Python exception.");
+                       goto end;
+               }
+
+               BT_LOG_WRITE(log_level, BT_LOG_TAG,
+                       "Exception occurred: Python traceback:\n%s", exc->str);
+       }
+
+end:
+       if (exc) {
+               g_string_free(exc, TRUE);
        }
 }
 
@@ -75,27 +104,42 @@ void pyerr_clear(void)
 }
 
 static
-void init_python(void)
+int init_python(void)
 {
+       int ret = BT_FUNC_STATUS_OK;
        PyObject *py_bt2_py_plugin_mod = NULL;
        const char *dis_python_env;
 #ifndef __MINGW32__
-       sighandler_t old_sigint = signal(SIGINT, SIG_DFL);
+       sig_t old_sigint = signal(SIGINT, SIG_DFL);
 #endif
 
-       if (python_state != PYTHON_STATE_NOT_INITED) {
+       switch (python_state) {
+       case PYTHON_STATE_NOT_INITED:
+               break;
+       case PYTHON_STATE_FULLY_INITIALIZED:
+               goto end;
+       case PYTHON_STATE_WONT_INITIALIZE:
+               ret = BT_FUNC_STATUS_NOT_FOUND;
                goto end;
+       case PYTHON_STATE_CANNOT_INITIALIZE:
+               ret = BT_FUNC_STATUS_ERROR;
+               goto end;
+       default:
+               bt_common_abort();
        }
 
        /*
         * User can disable Python plugin support with the
-        * BABELTRACE_DISABLE_PYTHON_PLUGINS environment variable set to
-        * 1.
+        * `LIBBABELTRACE2_DISABLE_PYTHON_PLUGINS` environment variable
+        * set to 1.
         */
-       dis_python_env = getenv("BABELTRACE_DISABLE_PYTHON_PLUGINS");
+       dis_python_env = getenv("LIBBABELTRACE2_DISABLE_PYTHON_PLUGINS");
        if (dis_python_env && strcmp(dis_python_env, "1") == 0) {
-               BT_LOGI_STR("Python plugin support is disabled because `BABELTRACE_DISABLE_PYTHON_PLUGINS=1`.");
-               python_state = PYTHON_STATE_CANNOT_INITIALIZE;
+               BT_LOGI_STR("Python plugin support is disabled because the "
+                       "`LIBBABELTRACE2_DISABLE_PYTHON_PLUGINS` environment "
+                       "variable is set to `1`.");
+               python_state = PYTHON_STATE_WONT_INITIALIZE;
+               ret = BT_FUNC_STATUS_NOT_FOUND;
                goto end;
        }
 
@@ -112,16 +156,24 @@ void init_python(void)
 
        py_bt2_py_plugin_mod = PyImport_ImportModule("bt2.py_plugin");
        if (!py_bt2_py_plugin_mod) {
-               BT_LOGI_STR("Cannot import bt2.py_plugin Python module: Python plugin support is disabled.");
+               append_python_traceback_error_cause();
+               BT_LIB_LOGW_APPEND_CAUSE(
+                       "Cannot import `bt2.py_plugin` Python module: "
+                       "Python plugin support is disabled.");
                python_state = PYTHON_STATE_CANNOT_INITIALIZE;
+               ret = BT_FUNC_STATUS_ERROR;
                goto end;
        }
 
        py_try_load_plugin_module_func =
                PyObject_GetAttrString(py_bt2_py_plugin_mod, "_try_load_plugin_module");
        if (!py_try_load_plugin_module_func) {
-               BT_LOGI_STR("Cannot get _try_load_plugin_module attribute from bt2.py_plugin Python module: Python plugin support is disabled.");
+               append_python_traceback_error_cause();
+               BT_LIB_LOGW_APPEND_CAUSE(
+                       "Cannot get `_try_load_plugin_module` attribute from `bt2.py_plugin` Python module: "
+                       "Python plugin support is disabled.");
                python_state = PYTHON_STATE_CANNOT_INITIALIZE;
+               ret = BT_FUNC_STATUS_ERROR;
                goto end;
        }
 
@@ -134,10 +186,11 @@ end:
        }
 #endif
 
-       print_python_traceback_warn();
+       log_python_traceback(ret == BT_FUNC_STATUS_ERROR ?
+               BT_LOG_WARNING : BT_LOG_INFO);
        pyerr_clear();
        Py_XDECREF(py_bt2_py_plugin_mod);
-       return;
+       return ret;
 }
 
 __attribute__((destructor)) static
@@ -156,9 +209,10 @@ void fini_python(void) {
 }
 
 static
-bt_plugin *bt_plugin_from_python_plugin_info(PyObject *plugin_info)
+int bt_plugin_from_python_plugin_info(PyObject *plugin_info,
+               bool fail_on_load_error, bt_plugin **plugin_out)
 {
-       bt_plugin *plugin = NULL;
+       int status = BT_FUNC_STATUS_OK;
        PyObject *py_name = NULL;
        PyObject *py_author = NULL;
        PyObject *py_description = NULL;
@@ -171,72 +225,172 @@ bt_plugin *bt_plugin_from_python_plugin_info(PyObject *plugin_info)
        const char *license = NULL;
        unsigned int major = 0, minor = 0, patch = 0;
        const char *version_extra = NULL;
-       int ret;
 
+       BT_ASSERT(plugin_out);
+       *plugin_out = NULL;
        BT_ASSERT(plugin_info);
        BT_ASSERT(python_state == PYTHON_STATE_FULLY_INITIALIZED);
        py_name = PyObject_GetAttrString(plugin_info, "name");
        if (!py_name) {
-               BT_LOGW("Cannot find `name` attribute in Python plugin info object: "
-                       "py-plugin-info-addr=%p", plugin_info);
+               if (fail_on_load_error) {
+                       append_python_traceback_error_cause();
+                       BT_LIB_LOGW_APPEND_CAUSE(
+                               "Cannot find `name` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_ERROR;
+               } else {
+                       BT_LIB_LOGW(
+                               "Cannot find `name` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_NOT_FOUND;
+               }
+
                goto error;
        }
 
        py_author = PyObject_GetAttrString(plugin_info, "author");
        if (!py_author) {
-               BT_LOGW("Cannot find `author` attribute in Python plugin info object: "
-                       "py-plugin-info-addr=%p", plugin_info);
+               if (fail_on_load_error) {
+                       append_python_traceback_error_cause();
+                       BT_LIB_LOGW_APPEND_CAUSE(
+                               "Cannot find `author` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_ERROR;
+               } else {
+                       BT_LIB_LOGW(
+                               "Cannot find `author` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_NOT_FOUND;
+               }
+
                goto error;
        }
 
        py_description = PyObject_GetAttrString(plugin_info, "description");
        if (!py_description) {
-               BT_LOGW("Cannot find `desciption` attribute in Python plugin info object: "
-                       "py-plugin-info-addr=%p", plugin_info);
+               if (fail_on_load_error) {
+                       append_python_traceback_error_cause();
+                       BT_LIB_LOGW_APPEND_CAUSE(
+                               "Cannot find `description` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_ERROR;
+               } else {
+                       BT_LIB_LOGW(
+                               "Cannot find `description` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_NOT_FOUND;
+               }
+
                goto error;
        }
 
        py_license = PyObject_GetAttrString(plugin_info, "license");
        if (!py_license) {
-               BT_LOGW("Cannot find `license` attribute in Python plugin info object: "
-                       "py-plugin-info-addr=%p", plugin_info);
+               if (fail_on_load_error) {
+                       append_python_traceback_error_cause();
+                       BT_LIB_LOGW_APPEND_CAUSE(
+                               "Cannot find `license` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_ERROR;
+               } else {
+                       BT_LIB_LOGW(
+                               "Cannot find `license` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_NOT_FOUND;
+               }
+
                goto error;
        }
 
        py_version = PyObject_GetAttrString(plugin_info, "version");
        if (!py_version) {
-               BT_LOGW("Cannot find `version` attribute in Python plugin info object: "
-                       "py-plugin-info-addr=%p", plugin_info);
+               if (fail_on_load_error) {
+                       append_python_traceback_error_cause();
+                       BT_LIB_LOGW_APPEND_CAUSE(
+                               "Cannot find `version` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_ERROR;
+               } else {
+                       BT_LIB_LOGW(
+                               "Cannot find `version` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_NOT_FOUND;
+               }
+
                goto error;
        }
 
        py_comp_class_addrs = PyObject_GetAttrString(plugin_info,
                "comp_class_addrs");
        if (!py_comp_class_addrs) {
-               BT_LOGW("Cannot find `comp_class_addrs` attribute in Python plugin info object: "
-                       "py-plugin-info-addr=%p", plugin_info);
+               if (fail_on_load_error) {
+                       append_python_traceback_error_cause();
+                       BT_LIB_LOGW_APPEND_CAUSE(
+                               "Cannot find `comp_class_addrs` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_ERROR;
+               } else {
+                       BT_LIB_LOGW(
+                               "Cannot find `comp_class_addrs` attribute in Python plugin info object: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_NOT_FOUND;
+               }
+
                goto error;
        }
 
        if (PyUnicode_Check(py_name)) {
                name = PyUnicode_AsUTF8(py_name);
                if (!name) {
-                       BT_LOGW("Cannot decode Python plugin name string: "
-                               "py-plugin-info-addr=%p", plugin_info);
+                       if (fail_on_load_error) {
+                               append_python_traceback_error_cause();
+                               BT_LIB_LOGW_APPEND_CAUSE(
+                                       "Cannot decode Python plugin name string: "
+                                       "py-plugin-info-addr=%p", plugin_info);
+                               status = BT_FUNC_STATUS_ERROR;
+                       } else {
+                               BT_LIB_LOGW(
+                                       "Cannot decode Python plugin name string: "
+                                       "py-plugin-info-addr=%p", plugin_info);
+                               status = BT_FUNC_STATUS_NOT_FOUND;
+                       }
+
                        goto error;
                }
        } else {
                /* Plugin name is mandatory */
-               BT_LOGW("Plugin name is not a string: "
-                       "py-plugin-info-addr=%p", plugin_info);
+               if (fail_on_load_error) {
+                       append_python_traceback_error_cause();
+                       BT_LIB_LOGW_APPEND_CAUSE(
+                               "Plugin name is not a string: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_ERROR;
+               } else {
+                       BT_LIB_LOGW(
+                               "Plugin name is not a string: "
+                               "py-plugin-info-addr=%p", plugin_info);
+                       status = BT_FUNC_STATUS_NOT_FOUND;
+               }
+
                goto error;
        }
 
        if (PyUnicode_Check(py_author)) {
                author = PyUnicode_AsUTF8(py_author);
                if (!author) {
-                       BT_LOGW("Cannot decode Python plugin author string: "
-                               "py-plugin-info-addr=%p", plugin_info);
+                       if (fail_on_load_error) {
+                               append_python_traceback_error_cause();
+                               BT_LIB_LOGW_APPEND_CAUSE(
+                                       "Cannot decode Python plugin author string: "
+                                       "py-plugin-info-addr=%p", plugin_info);
+                               status = BT_FUNC_STATUS_ERROR;
+                       } else {
+                               BT_LIB_LOGW(
+                                       "Cannot decode Python plugin author string: "
+                                       "py-plugin-info-addr=%p", plugin_info);
+                               status = BT_FUNC_STATUS_NOT_FOUND;
+                       }
+
                        goto error;
                }
        }
@@ -244,8 +398,19 @@ bt_plugin *bt_plugin_from_python_plugin_info(PyObject *plugin_info)
        if (PyUnicode_Check(py_description)) {
                description = PyUnicode_AsUTF8(py_description);
                if (!description) {
-                       BT_LOGW("Cannot decode Python plugin description string: "
-                               "py-plugin-info-addr=%p", plugin_info);
+                       if (fail_on_load_error) {
+                               append_python_traceback_error_cause();
+                               BT_LIB_LOGW_APPEND_CAUSE(
+                                       "Cannot decode Python plugin description string: "
+                                       "py-plugin-info-addr=%p", plugin_info);
+                               status = BT_FUNC_STATUS_ERROR;
+                       } else {
+                               BT_LIB_LOGW(
+                                       "Cannot decode Python plugin description string: "
+                                       "py-plugin-info-addr=%p", plugin_info);
+                               status = BT_FUNC_STATUS_NOT_FOUND;
+                       }
+
                        goto error;
                }
        }
@@ -253,8 +418,19 @@ bt_plugin *bt_plugin_from_python_plugin_info(PyObject *plugin_info)
        if (PyUnicode_Check(py_license)) {
                license = PyUnicode_AsUTF8(py_license);
                if (!license) {
-                       BT_LOGW("Cannot decode Python plugin license string: "
-                               "py-plugin-info-addr=%p", plugin_info);
+                       if (fail_on_load_error) {
+                               append_python_traceback_error_cause();
+                               BT_LIB_LOGW_APPEND_CAUSE(
+                                       "Cannot decode Python plugin license string: "
+                                       "py-plugin-info-addr=%p", plugin_info);
+                               status = BT_FUNC_STATUS_ERROR;
+                       } else {
+                               BT_LIB_LOGW(
+                                       "Cannot decode Python plugin license string: "
+                                       "py-plugin-info-addr=%p", plugin_info);
+                               status = BT_FUNC_STATUS_NOT_FOUND;
+                       }
+
                        goto error;
                }
        }
@@ -283,8 +459,19 @@ bt_plugin *bt_plugin_from_python_plugin_info(PyObject *plugin_info)
 
                        if (PyErr_Occurred()) {
                                /* Overflow error, most probably */
-                               BT_LOGW("Invalid Python plugin version format: "
-                                       "py-plugin-info-addr=%p", plugin_info);
+                               if (fail_on_load_error) {
+                                       append_python_traceback_error_cause();
+                                       BT_LIB_LOGW_APPEND_CAUSE(
+                                               "Invalid Python plugin version format: "
+                                               "py-plugin-info-addr=%p", plugin_info);
+                                       status = BT_FUNC_STATUS_ERROR;
+                               } else {
+                                       BT_LIB_LOGW(
+                                               "Invalid Python plugin version format: "
+                                               "py-plugin-info-addr=%p", plugin_info);
+                                       status = BT_FUNC_STATUS_NOT_FOUND;
+                               }
+
                                goto error;
                        }
                }
@@ -297,35 +484,47 @@ bt_plugin *bt_plugin_from_python_plugin_info(PyObject *plugin_info)
                        if (PyUnicode_Check(py_extra)) {
                                version_extra = PyUnicode_AsUTF8(py_extra);
                                if (!version_extra) {
-                               BT_LOGW("Cannot decode Python plugin version's extra string: "
-                                       "py-plugin-info-addr=%p", plugin_info);
+                                       if (fail_on_load_error) {
+                                               append_python_traceback_error_cause();
+                                               BT_LIB_LOGW_APPEND_CAUSE(
+                                                       "Cannot decode Python plugin version's extra string: "
+                                                       "py-plugin-info-addr=%p", plugin_info);
+                                               status = BT_FUNC_STATUS_ERROR;
+                                       } else {
+                                               BT_LIB_LOGW(
+                                                       "Cannot decode Python plugin version's extra string: "
+                                                       "py-plugin-info-addr=%p", plugin_info);
+                                               status = BT_FUNC_STATUS_NOT_FOUND;
+                                       }
+
                                        goto error;
                                }
                        }
                }
        }
 
-       plugin = bt_plugin_create_empty(BT_PLUGIN_TYPE_PYTHON);
-       if (!plugin) {
-               BT_LOGE_STR("Cannot create empty plugin object.");
+       *plugin_out = bt_plugin_create_empty(BT_PLUGIN_TYPE_PYTHON);
+       if (!*plugin_out) {
+               BT_LIB_LOGE_APPEND_CAUSE("Cannot create empty plugin object.");
+               status = BT_FUNC_STATUS_MEMORY_ERROR;
                goto error;
        }
 
-       bt_plugin_set_name(plugin, name);
+       bt_plugin_set_name(*plugin_out, name);
 
        if (description) {
-               bt_plugin_set_description(plugin, description);
+               bt_plugin_set_description(*plugin_out, description);
        }
 
        if (author) {
-               bt_plugin_set_author(plugin, author);
+               bt_plugin_set_author(*plugin_out, author);
        }
 
        if (license) {
-               bt_plugin_set_license(plugin, license);
+               bt_plugin_set_license(*plugin_out, license);
        }
 
-       bt_plugin_set_version(plugin, major, minor, patch, version_extra);
+       bt_plugin_set_version(*plugin_out, major, minor, patch, version_extra);
 
        if (PyList_Check(py_comp_class_addrs)) {
                size_t i;
@@ -340,27 +539,41 @@ bt_plugin *bt_plugin_from_python_plugin_info(PyObject *plugin_info)
                        if (PyLong_Check(py_comp_class_addr)) {
                                comp_class = PyLong_AsVoidPtr(py_comp_class_addr);
                        } else {
-                               BT_LOGW("Component class address is not an integer in Python plugin info object: "
-                                       "py-plugin-info-addr=%p, index=%zu",
-                                       plugin_info, i);
+                               if (fail_on_load_error) {
+                                       append_python_traceback_error_cause();
+                                       BT_LIB_LOGW_APPEND_CAUSE(
+                                               "Component class address is not an integer in Python plugin info object: "
+                                               "py-plugin-info-addr=%p, index=%zu",
+                                               plugin_info, i);
+                                       status = BT_FUNC_STATUS_ERROR;
+                               } else {
+                                       BT_LIB_LOGW(
+                                               "Component class address is not an integer in Python plugin info object: "
+                                               "py-plugin-info-addr=%p, index=%zu",
+                                               plugin_info, i);
+                                       status = BT_FUNC_STATUS_NOT_FOUND;
+                               }
+
                                continue;
                        }
 
-                       ret = bt_plugin_add_component_class(plugin, comp_class);
-                       if (ret < 0) {
-                               BT_LOGE("Cannot add component class to plugin: "
+                       status = bt_plugin_add_component_class(*plugin_out,
+                               comp_class);
+                       if (status < 0) {
+                               BT_LIB_LOGE_APPEND_CAUSE(
+                                       "Cannot add component class to plugin: "
                                        "py-plugin-info-addr=%p, "
                                        "plugin-addr=%p, plugin-name=\"%s\", "
                                        "comp-class-addr=%p, "
                                        "comp-class-name=\"%s\", "
                                        "comp-class-type=%s",
-                                       plugin_info,
-                                       plugin, bt_plugin_get_name(plugin),
+                                       plugin_info, *plugin_out,
+                                       bt_plugin_get_name(*plugin_out),
                                        comp_class,
                                        bt_component_class_get_name(comp_class),
                                        bt_component_class_type_string(
                                                bt_component_class_get_type(comp_class)));
-                               continue;
+                               goto error;
                        }
                }
        }
@@ -368,9 +581,10 @@ bt_plugin *bt_plugin_from_python_plugin_info(PyObject *plugin_info)
        goto end;
 
 error:
-       print_python_traceback_warn();
+       BT_ASSERT(status != BT_FUNC_STATUS_OK);
+       log_python_traceback(fail_on_load_error ? BT_LOG_WARNING : BT_LOG_INFO);
        pyerr_clear();
-       BT_OBJECT_PUT_REF_AND_RESET(plugin);
+       BT_OBJECT_PUT_REF_AND_RESET(*plugin_out);
 
 end:
        Py_XDECREF(py_name);
@@ -379,17 +593,18 @@ end:
        Py_XDECREF(py_license);
        Py_XDECREF(py_version);
        Py_XDECREF(py_comp_class_addrs);
-       return plugin;
+       return status;
 }
 
 G_MODULE_EXPORT
-bt_plugin_set *bt_plugin_python_create_all_from_file(const char *path)
+int bt_plugin_python_create_all_from_file(const char *path,
+               bool fail_on_load_error, struct bt_plugin_set **plugin_set_out)
 {
-       bt_plugin_set *plugin_set = NULL;
        bt_plugin *plugin = NULL;
        PyObject *py_plugin_info = NULL;
        gchar *basename = NULL;
        size_t path_len;
+       int status = BT_FUNC_STATUS_OK;
 
        BT_ASSERT(path);
 
@@ -399,6 +614,20 @@ bt_plugin_set *bt_plugin_python_create_all_from_file(const char *path)
                 * here because we already know Python cannot be fully
                 * initialized.
                 */
+               BT_LIB_LOGE_APPEND_CAUSE(
+                       "Python interpreter could not be initialized previously.");
+               status = BT_FUNC_STATUS_ERROR;
+               goto error;
+       } else if (python_state == PYTHON_STATE_WONT_INITIALIZE) {
+               /*
+                * This is not an error: the environment requires that
+                * Python plugins are disabled, so it's simply not
+                * found.
+                */
+               BT_LOGI_STR("Python plugin support was disabled previously "
+                       "because the `LIBBABELTRACE2_DISABLE_PYTHON_PLUGINS` "
+                       "environment variable is set to `1`.");
+               status = BT_FUNC_STATUS_NOT_FOUND;
                goto error;
        }
 
@@ -411,13 +640,16 @@ bt_plugin_set *bt_plugin_python_create_all_from_file(const char *path)
                        PYTHON_PLUGIN_FILE_EXT,
                        PYTHON_PLUGIN_FILE_EXT_LEN) != 0) {
                BT_LOGI("Skipping non-Python file: path=\"%s\"", path);
+               status = BT_FUNC_STATUS_NOT_FOUND;
                goto error;
        }
 
        /* File name starts with `bt_plugin_` */
        basename = g_path_get_basename(path);
        if (!basename) {
-               BT_LOGW("Cannot get path's basename: path=\"%s\"", path);
+               BT_LIB_LOGE_APPEND_CAUSE(
+                       "Cannot get path's basename: path=\"%s\"", path);
+               status = BT_FUNC_STATUS_ERROR;
                goto error;
        }
 
@@ -425,6 +657,7 @@ bt_plugin_set *bt_plugin_python_create_all_from_file(const char *path)
                        PYTHON_PLUGIN_FILE_PREFIX_LEN) != 0) {
                BT_LOGI("Skipping Python file not starting with `%s`: "
                        "path=\"%s\"", PYTHON_PLUGIN_FILE_PREFIX, path);
+               status = BT_FUNC_STATUS_NOT_FOUND;
                goto error;
        }
 
@@ -436,14 +669,9 @@ bt_plugin_set *bt_plugin_python_create_all_from_file(const char *path)
         * have any potential Python plugins, you don't need to endure
         * this waiting time everytime you load the library.
         */
-       init_python();
-       if (python_state != PYTHON_STATE_FULLY_INITIALIZED) {
-               /*
-                * For some reason we cannot initialize Python,
-                * import the required modules, and get the required
-                * attributes from them.
-                */
-               BT_LOGI("Failed to initialize Python interpreter.");
+       status = init_python();
+       if (status != BT_FUNC_STATUS_OK) {
+               /* init_python() logs and append errors */
                goto error;
        }
 
@@ -457,42 +685,69 @@ bt_plugin_set *bt_plugin_python_create_all_from_file(const char *path)
        py_plugin_info = PyObject_CallFunction(py_try_load_plugin_module_func,
                "(s)", path);
        if (!py_plugin_info || py_plugin_info == Py_None) {
-               BT_LOGW("Cannot load Python plugin: path=\"%s\"", path);
-               print_python_traceback_warn();
-               PyErr_Clear();
+               if (fail_on_load_error) {
+                       append_python_traceback_error_cause();
+                       BT_LIB_LOGW_APPEND_CAUSE(
+                               "Cannot load Python plugin: path=\"%s\"", path);
+                       status = BT_FUNC_STATUS_ERROR;
+               } else {
+                       BT_LIB_LOGW(
+                               "Cannot load Python plugin: path=\"%s\"", path);
+                       status = BT_FUNC_STATUS_NOT_FOUND;
+               }
+
                goto error;
        }
 
        /*
         * Get bt_plugin from plugin info object.
         */
-       plugin = bt_plugin_from_python_plugin_info(py_plugin_info);
-       if (!plugin) {
-               BT_LOGW("Cannot create plugin object from Python plugin info object: "
+       plugin = NULL;
+       status = bt_plugin_from_python_plugin_info(py_plugin_info,
+               fail_on_load_error, &plugin);
+       if (status < 0) {
+               /*
+                * bt_plugin_from_python_plugin_info() handles
+                * `fail_on_load_error`, so this is a "real" error.
+                */
+               BT_LIB_LOGW_APPEND_CAUSE(
+                       "Cannot create plugin object from Python plugin info object: "
                        "path=\"%s\", py-plugin-info-addr=%p",
                        path, py_plugin_info);
+               BT_ASSERT(!plugin);
+               goto error;
+       } else if (status == BT_FUNC_STATUS_NOT_FOUND) {
+               BT_ASSERT(!plugin);
                goto error;
        }
 
+       BT_ASSERT(status == BT_FUNC_STATUS_OK);
+       BT_ASSERT(plugin);
        bt_plugin_set_path(plugin, path);
-       plugin_set = bt_plugin_set_create();
-       if (!plugin_set) {
-               BT_LOGE_STR("Cannot create empty plugin set.");
+       *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 error;
        }
 
-       bt_plugin_set_add_plugin(plugin_set, plugin);
+       bt_plugin_set_add_plugin(*plugin_set_out, plugin);
        BT_LOGD("Created all Python plugins from file: path=\"%s\", "
                "plugin-addr=%p, plugin-name=\"%s\"",
                path, plugin, bt_plugin_get_name(plugin));
        goto end;
 
 error:
-       BT_OBJECT_PUT_REF_AND_RESET(plugin_set);
+       BT_ASSERT(status != BT_FUNC_STATUS_OK);
+       log_python_traceback(BT_LOG_WARNING);
+       pyerr_clear();
+       BT_OBJECT_PUT_REF_AND_RESET(*plugin_set_out);
 
 end:
        bt_plugin_put_ref(plugin);
        Py_XDECREF(py_plugin_info);
+
        g_free(basename);
-       return plugin_set;
+
+       return status;
 }
This page took 0.031998 seconds and 4 git commands to generate.