pdnsutil: also load modules through the load-modules directive

diff --git a/pdns/pdnsutil.cc b/pdns/pdnsutil.cc
index 59f5955b0ab2a5dcc115b7978ac3eb423e056b5e..d617105403bc32ecbc29a52abefab00db9fb29ff 100644 (file)
--- a/pdns/pdnsutil.cc
+++ b/pdns/pdnsutil.cc
@@ -58,6 +58,32 @@ string humanTime(time_t t)
   return ret;
 }
 
+static void loadModules()
+{
+  if(!::arg()["load-modules"].empty()) {
+    vector<string>modules;
+
+    stringtok(modules,::arg()["load-modules"],", ");
+
+    for(vector<string>::const_iterator i=modules.begin();i!=modules.end();++i) {
+      bool res;
+      const string &module=*i;
+
+      if(module.find(".")==string::npos)
+        res=UeberBackend::loadmodule(::arg()["module-dir"]+"/lib"+module+"backend.so");
+      else if(module[0]=='/' || (module[0]=='.' && module[1]=='/') || (module[0]=='.' && module[1]=='.'))    // absolute or current path
+        res=UeberBackend::loadmodule(module);
+      else
+        res=UeberBackend::loadmodule(::arg()["module-dir"]+"/"+module);
+
+      if(res==false) {
+        g_log<<Logger::Error<<"Receiver unable to load module "<<module<<endl;
+        exit(1);
+      }
+    }
+  }
+}
+
 void loadMainConfig(const std::string& configdir)
 {
   ::arg().set("config-dir","Location of configuration directory (pdns.conf)")=configdir;
@@ -66,6 +92,7 @@ void loadMainConfig(const std::string& configdir)
   ::arg().set("dnssec","if we should do dnssec")="true";
   ::arg().set("config-name","Name of this virtual configuration - will rename the binary image")=g_vm["config-name"].as<string>();
   ::arg().setCmd("help","Provide a helpful message");
+  ::arg().set("load-modules","Load this module - supply absolute or relative path")="";
   //::arg().laxParse(argc,argv);
 
   if(::arg().mustDo("help")) {
@@ -98,6 +125,8 @@ void loadMainConfig(const std::string& configdir)
   ::arg().set("rng", "Specify random number generator to use. Valid values are auto,sodium,openssl,getrandom,arc4random,urandom.")="auto";
   ::arg().laxFile(configname.c_str());
 
+  loadModules();
+
   g_log.toConsole(Logger::Error);   // so we print any errors
   BackendMakers().launch(::arg()["launch"]); // vrooooom!
   if(::arg().asNum("loglevel") >= 3) // so you can't kill our errors