satdump/src-core/plugin.cpp

#define SATDUMP_DLL_EXPORT 1
#include "plugin.h"
#ifdef _WIN32
#include "libs/dlfcn/dlfcn.h"
#else
#include <dlfcn.h>
#endif
#include "logger.h"
#include <filesystem>
#include "satdump_vars.h"

std::shared_ptr<satdump::Plugin> loadPlugin(std::string plugin)
{
    logger->trace("Loading plugin " + plugin + "...");

    void *dynlib = dlopen(plugin.c_str(), RTLD_LAZY);
    if (!dynlib)
        throw std::runtime_error("Error loading " + plugin + "!");

    void *create = dlsym(dynlib, "loader");
    const char *dlsym_error = dlerror();
    if (dlsym_error != NULL)
        logger->warn("Possible error loading symbols from plugin!");

    satdump::Plugin *pluginObject = reinterpret_cast<satdump::Plugin *(*)()>(create)();
    pluginObject->init();
    logger->trace("Plugin " + pluginObject->getID() + " loaded!");

    return std::shared_ptr<satdump::Plugin>(pluginObject);
}

namespace satdump
{
    SATDUMP_DLL std::map<std::string, std::shared_ptr<satdump::Plugin>> loaded_plugins;
    SATDUMP_DLL std::shared_ptr<dp::event_bus> eventBus = std::make_shared<dp::event_bus>();
};

void loadPlugins()
{
    std::string plugins_path = (std::string)RESOURCES_PATH + "plugins";

    if (std::filesystem::exists("plugins"))
        plugins_path = "./plugins";

#if defined(_WIN32)
    std::string extension = ".dll";
#elif defined(__APPLE__)
    std::string extension = ".dylib";
#else
    std::string extension = ".so";
#endif

    logger->info("Loading plugins from " + plugins_path);

    std::filesystem::recursive_directory_iterator pluginIterator(plugins_path);
    std::error_code iteratorError;
    while (pluginIterator != std::filesystem::recursive_directory_iterator())
    {
        if (!std::filesystem::is_directory(pluginIterator->path()))
        {
            if (pluginIterator->path().filename().string().find(extension) != std::string::npos)
            {
                std::string path = pluginIterator->path().string();

                if (std::filesystem::exists(path) && std::filesystem::is_regular_file(path))
                {
                    std::shared_ptr<satdump::Plugin> pl = loadPlugin(path);
                    satdump::loaded_plugins.insert({pl->getID(), pl});
                }
                else
                {
                    logger->error("File " + path + " is not a valid plugin!");
                }
            }
        }

        pluginIterator.increment(iteratorError);
        if (iteratorError)
            logger->critical(iteratorError.message());
    }

    if (satdump::loaded_plugins.size() > 0)
    {
        logger->debug("Loaded plugins (" + std::to_string(satdump::loaded_plugins.size()) + ") : ");
        for (std::pair<const std::string, std::shared_ptr<satdump::Plugin>> &it : satdump::loaded_plugins)
            logger->debug(" - " + it.first);
    }
}