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  */
  20
  21 /*
  22  $Revision: 1.17 $
  23  $Date: 2010/09/13 05:52:46 $
  24  $Author: faust $
  25  */
  26
  27 #include <dlfcn.h>
  28 #include <unistd.h>
  29
  30 #include "stg/common.h"
  31 #include "stg/traffcounter.h"
  32 #include "plugin_runner.h"
  33 #include "settings_impl.h"
  34 #include "admins_impl.h"
  35 #include "tariffs_impl.h"
  36 #include "users_impl.h"
  37 #include "services_impl.h"
  38 #include "corps_impl.h"
  39
  40 //-----------------------------------------------------------------------------
  41 PLUGIN_RUNNER::PLUGIN_RUNNER(const std::string & pFileName,
  42                              const MODULE_SETTINGS & ms,
  43                              ADMINS_IMPL * a,
  44                              TARIFFS_IMPL * t,
  45                              USERS_IMPL * u,
  46                              SERVICES_IMPL * svc,
  47                              CORPORATIONS_IMPL * crp,
  48                              TRAFFCOUNTER * tc,
  49                              STORE * st,
  50                              const SETTINGS_IMPL * s)
  51     : pluginFileName(pFileName),
  52       pluginSettingFileName(),
  53       plugin(NULL),
  54       isPluginLoaded(false),
  55       errorStr(),
  56       libHandle(NULL),
  57       isRunning(false),
  58       admins(a),
  59       tariffs(t),
  60       users(u),
  61       services(svc),
  62       corps(crp),
  63       store(st),
  64       traffCnt(tc),
  65       stgSettings(s),
  66       modSettings(ms)
  67 {
  68 }
  69 //-----------------------------------------------------------------------------
  70 PLUGIN_RUNNER::PLUGIN_RUNNER(const PLUGIN_RUNNER & rvalue)
  71     : pluginFileName(rvalue.pluginFileName),
  72       pluginSettingFileName(rvalue.pluginSettingFileName),
  73       plugin(rvalue.plugin),
  74       isPluginLoaded(rvalue.isPluginLoaded),
  75       errorStr(rvalue.errorStr),
  76       libHandle(rvalue.libHandle),
  77       isRunning(rvalue.isRunning),
  78       admins(rvalue.admins),
  79       tariffs(rvalue.tariffs),
  80       users(rvalue.users),
  81       services(rvalue.services),
  82       corps(rvalue.corps),
  83       store(rvalue.store),
  84       traffCnt(rvalue.traffCnt),
  85       stgSettings(rvalue.stgSettings),
  86       modSettings(rvalue.modSettings)
  87 {
  88 }
  89 //-----------------------------------------------------------------------------
  90 PLUGIN_RUNNER & PLUGIN_RUNNER::operator=(const PLUGIN_RUNNER & rvalue)
  91 {
  92 pluginFileName = rvalue.pluginFileName;
  93 pluginSettingFileName = rvalue.pluginSettingFileName;
  94 plugin = rvalue.plugin;
  95 isPluginLoaded = rvalue.isPluginLoaded;
  96 errorStr = rvalue.errorStr;
  97 libHandle = rvalue.libHandle;
  98 isRunning = rvalue.isRunning;
  99 admins = rvalue.admins;
 100 tariffs = rvalue.tariffs;
 101 users = rvalue.users;
 102 services = rvalue.services;
 103 corps = rvalue.corps;
 104 store = rvalue.store;
 105 traffCnt = rvalue.traffCnt;
 106 stgSettings = rvalue.stgSettings;
 107 modSettings = rvalue.modSettings;
 108
 109 return *this;
 110 }
 111 //-----------------------------------------------------------------------------
 112 PLUGIN_RUNNER::~PLUGIN_RUNNER()
 113 {
 114 if (isPluginLoaded)
 115     {
 116     Unload();
 117     }
 118
 119 isPluginLoaded = false;
 120 }
 121 //-----------------------------------------------------------------------------
 122 PLUGIN * PLUGIN_RUNNER::GetPlugin()
 123 {
 124 if (!isPluginLoaded)
 125     {
 126     errorStr = "Plugin '" + pluginFileName + "' is not loaded yet!";
 127     printfd(__FILE__, "PLUGIN_LOADER::GetPlugin() - %s\n", errorStr.c_str());
 128     return NULL;
 129     }
 130
 131 return plugin;
 132 }
 133 //-----------------------------------------------------------------------------
 134 int PLUGIN_RUNNER::Start()
 135 {
 136 if (!isPluginLoaded)
 137     if (Load())
 138         return -1;
 139
 140 if (!plugin)
 141     {
 142     errorStr = "Plugin '" + pluginFileName + "' was not created!";
 143     printfd(__FILE__, "PLUGIN_LOADER::Start() - %s\n", errorStr.c_str());
 144     return -1;
 145     }
 146
 147 plugin->SetTariffs(tariffs);
 148 plugin->SetAdmins(admins);
 149 plugin->SetUsers(users);
 150 plugin->SetServices(services);
 151 plugin->SetCorporations(corps);
 152 plugin->SetTraffcounter(traffCnt);
 153 plugin->SetStore(store);
 154 plugin->SetStgSettings(stgSettings);
 155
 156 if (plugin->Start())
 157     {
 158     errorStr = plugin->GetStrError();
 159     return -1;
 160     }
 161
 162 return 0;
 163 }
 164 //-----------------------------------------------------------------------------
 165 int PLUGIN_RUNNER::Stop()
 166 {
 167 if (!isPluginLoaded)
 168     {
 169     errorStr = "Plugin '" + pluginFileName + "' was not loaded yet!";
 170     printfd(__FILE__, "PLUGIN_LOADER::Stop() - %s\n", errorStr.c_str());
 171     return -1;
 172     }
 173
 174 if (!plugin)
 175     {
 176     errorStr = "Plugin '" + pluginFileName + "' was not created!";
 177     printfd(__FILE__, "PLUGIN_LOADER::Stop() - %s\n", errorStr.c_str());
 178     return -1;
 179     }
 180
 181 plugin->Stop();
 182
 183 return 0;
 184 }
 185 //-----------------------------------------------------------------------------
 186 int PLUGIN_RUNNER::Reload()
 187 {
 188 if (!isPluginLoaded)
 189     {
 190     errorStr = "Plugin '" + pluginFileName + "' was not loaded yet!";
 191     printfd(__FILE__, "PLUGIN_LOADER::Reload() - %s\n", errorStr.c_str());
 192     return -1;
 193     }
 194
 195 if (!plugin)
 196     {
 197     errorStr = "Plugin '" + pluginFileName + "' was not created!";
 198     printfd(__FILE__, "PLUGIN_LOADER::Reload() - %s\n", errorStr.c_str());
 199     return -1;
 200     }
 201
 202 int res = plugin->Reload();
 203 errorStr = plugin->GetStrError();
 204 return res;
 205 }
 206 //-----------------------------------------------------------------------------
 207 bool PLUGIN_RUNNER::IsRunning()
 208 {
 209 if (!isPluginLoaded)
 210     {
 211     errorStr = "Plugin '" + pluginFileName + "' was not loaded yet!";
 212     printfd(__FILE__, "PLUGIN_LOADER::IsRunning() - %s\n", errorStr.c_str());
 213     return false;
 214     }
 215
 216 if (!plugin)
 217     {
 218     errorStr = "Plugin '" + pluginFileName + "' was not created!";
 219     printfd(__FILE__, "PLUGIN_LOADER::IsRunning() - %s\n", errorStr.c_str());
 220     return false;
 221     }
 222
 223 return plugin->IsRunning();
 224 }
 225 //-----------------------------------------------------------------------------
 226 int PLUGIN_RUNNER::Load()
 227 {
 228 if (isPluginLoaded)
 229     {
 230     errorStr = "Plugin '" + pluginFileName + "' was already loaded!";
 231     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
 232     return -1;
 233     }
 234
 235 if (pluginFileName.empty())
 236     {
 237     errorStr = "Empty plugin file name!";
 238     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
 239     return -1;
 240     }
 241
 242 libHandle = dlopen(pluginFileName.c_str(), RTLD_NOW);
 243
 244 if (!libHandle)
 245     {
 246     errorStr = "Error loading plugin '"
 247         + pluginFileName + "': '" + dlerror() + "'";
 248     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
 249     return -1;
 250     }
 251
 252 isPluginLoaded = true;
 253
 254 PLUGIN * (*GetPlugin)();
 255 GetPlugin = (PLUGIN * (*)())dlsym(libHandle, "GetPlugin");
 256 if (!GetPlugin)
 257     {
 258     errorStr = std::string("GetPlugin() not found. ") + dlerror();
 259     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
 260     return -1;
 261     }
 262 plugin = GetPlugin();
 263
 264 if (!plugin)
 265     {
 266     errorStr = "Plugin was not created!";
 267     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
 268     return -1;
 269     }
 270
 271 plugin->SetSettings(modSettings);
 272 if (plugin->ParseSettings())
 273     {
 274     errorStr = plugin->GetStrError();
 275     printfd(__FILE__, "PLUGIN_LOADER::Load() - Failed to parse settings. Plugin reports: '%s'\n", errorStr.c_str());
 276     return -1;
 277     }
 278
 279 return 0;
 280 }
 281 //-----------------------------------------------------------------------------
 282 int PLUGIN_RUNNER::Unload()
 283 {
 284 if (isPluginLoaded)
 285     {
 286     if (dlclose(libHandle))
 287         {
 288         errorStr = "Failed to unload plugin '";
 289         errorStr += pluginFileName + "': ";
 290         errorStr += dlerror();
 291         printfd(__FILE__, "PLUGIN_LOADER::Unload() - %s", errorStr.c_str());
 292         return -1;
 293         }
 294     plugin = NULL;
 295     isPluginLoaded = false;
 296     }
 297 return 0;
 298 }
 299 //-----------------------------------------------------------------------------