projects/stargazer/plugin_runner.cpp

   1 /*
   2  *    This program is free software; you can redistribute it and/or modify
   3  *    it under the terms of the GNU General Public License as published by
   4  *    the Free Software Foundation; either version 2 of the License, or
   5  *    (at your option) any later version.
   6  *
   7  *    This program is distributed in the hope that it will be useful,
   8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
   9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  10  *    GNU General Public License for more details.
  11  *
  12  *    You should have received a copy of the GNU General Public License
  13  *    along with this program; if not, write to the Free Software
  14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  15  */
  16
  17 /*
  18  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
  19  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  20  */
  21
  22 #include "plugin_runner.h"
  23
  24 #include "stg/common.h"
  25
  26 #include <dlfcn.h>
  27 #include <unistd.h>
  28
  29 //-----------------------------------------------------------------------------
  30 PLUGIN_RUNNER::PLUGIN_RUNNER(const std::string & fileName,
  31                              const MODULE_SETTINGS & ms,
  32                              ADMINS & admins,
  33                              TARIFFS & tariffs,
  34                              USERS & users,
  35                              SERVICES & services,
  36                              CORPORATIONS & corporations,
  37                              TRAFFCOUNTER & traffcounter,
  38                              STORE & store,
  39                              const SETTINGS & settings)
  40     : pluginFileName(fileName),
  41       libHandle(NULL),
  42       m_plugin(Load(ms, admins, tariffs, users, services, corporations,
  43                     traffcounter, store, settings))
  44 {
  45 }
  46 //-----------------------------------------------------------------------------
  47 PLUGIN_RUNNER::~PLUGIN_RUNNER()
  48 {
  49 if (dlclose(libHandle))
  50     {
  51     errorStr = "Failed to unload plugin '" + pluginFileName + "': " + dlerror();
  52     printfd(__FILE__, "PLUGIN_RUNNER::Unload() - %s", errorStr.c_str());
  53     }
  54 }
  55 //-----------------------------------------------------------------------------
  56 int PLUGIN_RUNNER::Start()
  57 {
  58 int res = m_plugin.Start();
  59 errorStr = m_plugin.GetStrError();
  60 return res;
  61 }
  62 //-----------------------------------------------------------------------------
  63 int PLUGIN_RUNNER::Stop()
  64 {
  65 int res = m_plugin.Stop();
  66 errorStr = m_plugin.GetStrError();
  67 return res;
  68 }
  69 //-----------------------------------------------------------------------------
  70 int PLUGIN_RUNNER::Reload()
  71 {
  72 int res = m_plugin.Reload();
  73 errorStr = m_plugin.GetStrError();
  74 return res;
  75 }
  76 //-----------------------------------------------------------------------------
  77 PLUGIN & PLUGIN_RUNNER::Load(const MODULE_SETTINGS & ms,
  78                              ADMINS & admins,
  79                              TARIFFS & tariffs,
  80                              USERS & users,
  81                              SERVICES & services,
  82                              CORPORATIONS & corporations,
  83                              TRAFFCOUNTER & traffcounter,
  84                              STORE & store,
  85                              const SETTINGS & settings)
  86 {
  87 if (pluginFileName.empty())
  88     {
  89     errorStr = "Empty plugin file name.";
  90     printfd(__FILE__, "PLUGIN_RUNNER::Load() - %s\n", errorStr.c_str());
  91     throw Error(errorStr);
  92     }
  93
  94 libHandle = dlopen(pluginFileName.c_str(), RTLD_NOW);
  95
  96 if (!libHandle)
  97     {
  98     errorStr = "Error loading plugin '" + pluginFileName + "': '" + dlerror() + "'";
  99     printfd(__FILE__, "PLUGIN_RUNNER::Load() - %s\n", errorStr.c_str());
 100     throw Error(errorStr);
 101     }
 102
 103 PLUGIN * (*GetPlugin)();
 104 GetPlugin = (PLUGIN * (*)())dlsym(libHandle, "GetPlugin");
 105 if (!GetPlugin)
 106     {
 107     errorStr = "Plugin '" + pluginFileName + "' does not have GetPlugin() function. " + dlerror();
 108     printfd(__FILE__, "PLUGIN_RUNNER::Load() - %s\n", errorStr.c_str());
 109     throw Error(errorStr);
 110     }
 111 PLUGIN * plugin = GetPlugin();
 112
 113 if (!plugin)
 114     {
 115     errorStr = "Failed to create an instance of plugin '" + pluginFileName + "'.";
 116     printfd(__FILE__, "PLUGIN_RUNNER::Load() - %s\n", errorStr.c_str());
 117     throw Error(errorStr);
 118     }
 119
 120 plugin->SetSettings(ms);
 121 plugin->SetTariffs(&tariffs);
 122 plugin->SetAdmins(&admins);
 123 plugin->SetUsers(&users);
 124 plugin->SetServices(&services);
 125 plugin->SetCorporations(&corporations);
 126 plugin->SetTraffcounter(&traffcounter);
 127 plugin->SetStore(&store);
 128 plugin->SetStgSettings(&settings);
 129
 130 if (plugin->ParseSettings())
 131     {
 132     errorStr = "Plugin '" + pluginFileName + "' is unable to parse settings. " + plugin->GetStrError();
 133     printfd(__FILE__, "PLUGIN_RUNNER::Load() - %s\n", errorStr.c_str());
 134     throw Error(errorStr);
 135     }
 136
 137 return *plugin;
 138 }