riscv
/
yosys
mirror of https://github.com/YosysHQ/yosys.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

plugin: Re-vamped how plugin lookup was done to make it more consistent with the rest of yosys, and prevented a case where you could end up with `.so.so` on the end

This commit is contained in:
Aki Van Ness 2023-05-03 02:22:46 -04:00
parent cee3cb31b9
commit 572c8df9a8
No known key found for this signature in database
GPG Key ID: 58B2EF8A70E7F62C
1 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
passes/cmds/plugin.cc
View File

@ -21,12 +21,12 @@
#ifdef YOSYS_ENABLE_PLUGINS
# include <dlfcn.h>
# include <boost/filesystem.hpp>
# include <boost/algorithm/string/predicate.hpp>
#endif
#ifdef WITH_PYTHON
# include <boost/algorithm/string/predicate.hpp>
# include <Python.h>
# include <boost/filesystem.hpp>
#endif
YOSYS_NAMESPACE_BEGIN
@ -41,20 +41,18 @@ std::map<std::string, std::string> loaded_plugin_aliases;
void load_plugin(std::string filename, std::vector<std::string> aliases)
{
std::string orig_filename = filename;
rewrite_filename(filename);
boost::filesystem::path full_path(filename);
if (filename.find('/') == std::string::npos)
filename = "./" + filename;
#ifdef WITH_PYTHON
if (!loaded_plugins.count(filename) && !loaded_python_plugins.count(filename)) {
if (!loaded_plugins.count(orig_filename) && !loaded_python_plugins.count(orig_filename)) {
#else
if (!loaded_plugins.count(filename)) {
if (!loaded_plugins.count(orig_filename)) {
#endif
#ifdef WITH_PYTHON
boost::filesystem::path full_path(filename);
if(strcmp(full_path.extension().c_str(), ".py") == 0)
{
std::string path(full_path.parent_path().c_str());
@ -75,10 +73,23 @@ void load_plugin(std::string filename, std::vector<std::string> aliases)
#endif
void *hdl = dlopen(filename.c_str(), RTLD_LAZY|RTLD_LOCAL);
if (hdl == NULL && orig_filename.find('/') == std::string::npos)
hdl = dlopen((proc_share_dirname() + "plugins/" + orig_filename + ".so").c_str(), RTLD_LAZY|RTLD_LOCAL);
// We were unable to open the file, try to do so from the plugin directory
if (hdl == NULL && filename.find('/') == std::string::npos) {
hdl = dlopen([filename]() {
std::string new_path = proc_share_dirname() + "plugins/" + filename;
// Check if we need to append .so
if (new_path.find(".so") == std::string::npos)
new_path.append(".so");
return new_path;
}().c_str(), RTLD_LAZY|RTLD_LOCAL);
}
if (hdl == NULL)
log_cmd_error("Can't load module `%s': %s\n", filename.c_str(), dlerror());
loaded_plugins[orig_filename] = hdl;
Pass::init_register();
@ -182,4 +193,3 @@ struct PluginPass : public Pass {
} PluginPass;
YOSYS_NAMESPACE_END