Added "plugin" command

2 files changed, 118 insertions, 0 deletions

diff --git a/passes/cmds/Makefile.inc b/passes/cmds/Makefile.inc
index d6e45f2d1..4cf61150e 100644
--- a/passes/cmds/Makefile.inc
+++ b/passes/cmds/Makefile.inc
@@ -21,4 +21,5 @@ OBJS += passes/cmds/connwrappers.o
 OBJS += passes/cmds/cover.o
 OBJS += passes/cmds/trace.o
 OBJS += passes/cmds/wreduce.o
+OBJS += passes/cmds/plugin.o
 
diff --git a/passes/cmds/plugin.cc b/passes/cmds/plugin.cc
new file mode 100644
index 000000000..c73684f1b
--- /dev/null
+++ b/passes/cmds/plugin.cc
@@ -0,0 +1,117 @@
+/*
+ *  yosys -- Yosys Open SYnthesis Suite
+ *
+ *  Copyright (C) 2014  Clifford Wolf <clifford@clifford.at>
+ *
+ *  Permission to use, copy, modify, and/or distribute this software for any
+ *  purpose with or without fee is hereby granted, provided that the above
+ *  copyright notice and this permission notice appear in all copies.
+ *
+ *  THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ *  WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ *  MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ *  ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ *  WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ *  ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ *  OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ *
+ */
+
+#include "kernel/yosys.h"
+#include <dlfcn.h>
+
+YOSYS_NAMESPACE_BEGIN
+
+std::map<std::string, void*> loaded_plugins;
+std::map<std::string, std::string> loaded_plugin_aliases;
+
+void load_plugin(std::string filename, std::vector<std::string> aliases)
+{
+	if (filename.find('/') == std::string::npos)
+		filename = "./" + filename;
+
+	if (!loaded_plugins.count(filename)) {
+		void *hdl = dlopen(filename.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[filename] = hdl;
+		Pass::init_register();
+	}
+
+	for (auto &alias : aliases)
+		loaded_plugin_aliases[alias] = filename;
+}
+
+struct PluginPass : public Pass {
+	PluginPass() : Pass("plugin", "load and list loaded plugins") { }
+	virtual void help()
+	{
+		//   |---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|---v---|
+		log("\n");
+		log("    plugin [options]\n");
+		log("\n");
+		log("Load and list loaded plugins.\n");
+		log("\n");
+		log("    -i <plugin_filename>\n");
+		log("        Load (install) the specified plugin.\n");
+		log("\n");
+		log("    -a <alias_name>\n");
+		log("        Register the specified alias name for the loaded plugin\n");
+		log("\n");
+		log("    -l\n");
+		log("        List loaded plugins\n");
+		log("\n");
+	}
+	virtual void execute(std::vector<std::string> args, RTLIL::Design *design)
+	{
+		std::string plugin_filename;
+		std::vector<std::string> plugin_aliases;
+		bool list_mode = false;
+
+		size_t argidx;
+		for (argidx = 1; argidx < args.size(); argidx++)
+		{
+			if ((args[argidx] == "-i") && argidx+1 < args.size() && plugin_filename.empty()) {
+				plugin_filename = args[++argidx];
+				continue;
+			}
+			if ((args[argidx] == "-a") && argidx+1 < args.size()) {
+				plugin_aliases.push_back(args[++argidx]);
+				continue;
+			}
+			if (args[argidx] == "-l") {
+				list_mode = true;
+				continue;
+			}
+			break;
+		}
+		extra_args(args, argidx, design, false);
+
+		if (!plugin_filename.empty())
+			load_plugin(plugin_filename, plugin_aliases);
+
+		if (list_mode)
+		{
+			log("\n");
+			if (loaded_plugins.empty())
+				log("No plugins loaded.\n");
+			else
+				log("Loaded plugins:\n");
+
+			for (auto &it : loaded_plugins)
+				log("  %s\n", it.first.c_str());
+
+			if (!loaded_plugin_aliases.empty()) {
+				log("\n");
+				int max_alias_len = 1;
+				for (auto &it : loaded_plugin_aliases)
+					max_alias_len = std::max(max_alias_len, SIZE(it.first));
+				for (auto &it : loaded_plugin_aliases)
+					log("Alias: %-*s %s\n", max_alias_len, it.first.c_str(), it.second.c_str());
+			}
+		}
+	}
+} PluginPass;
+
+YOSYS_NAMESPACE_END
+