]> git.stg.codes - stg.git/blob - projects/stargazer/plugin_runner.cpp
Merge branch 'master' into full-month-stats
[stg.git] / 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 return plugin->Stop();
182 }
183 //-----------------------------------------------------------------------------
184 int PLUGIN_RUNNER::Reload()
185 {
186 if (!isPluginLoaded)
187     {
188     errorStr = "Plugin '" + pluginFileName + "' was not loaded yet!";
189     printfd(__FILE__, "PLUGIN_LOADER::Reload() - %s\n", errorStr.c_str());
190     return -1;
191     }
192
193 if (!plugin)
194     {
195     errorStr = "Plugin '" + pluginFileName + "' was not created!";
196     printfd(__FILE__, "PLUGIN_LOADER::Reload() - %s\n", errorStr.c_str());
197     return -1;
198     }
199
200 int res = plugin->Reload();
201 errorStr = plugin->GetStrError();
202 return res;
203 }
204 //-----------------------------------------------------------------------------
205 bool PLUGIN_RUNNER::IsRunning()
206 {
207 if (!isPluginLoaded)
208     {
209     errorStr = "Plugin '" + pluginFileName + "' was not loaded yet!";
210     printfd(__FILE__, "PLUGIN_LOADER::IsRunning() - %s\n", errorStr.c_str());
211     return false;
212     }
213
214 if (!plugin)
215     {
216     errorStr = "Plugin '" + pluginFileName + "' was not created!";
217     printfd(__FILE__, "PLUGIN_LOADER::IsRunning() - %s\n", errorStr.c_str());
218     return false;
219     }
220
221 return plugin->IsRunning();
222 }
223 //-----------------------------------------------------------------------------
224 int PLUGIN_RUNNER::Load()
225 {
226 if (isPluginLoaded)
227     {
228     errorStr = "Plugin '" + pluginFileName + "' was already loaded!";
229     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
230     return -1;
231     }
232
233 if (pluginFileName.empty())
234     {
235     errorStr = "Empty plugin file name!";
236     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
237     return -1;
238     }
239
240 libHandle = dlopen(pluginFileName.c_str(), RTLD_NOW);
241
242 if (!libHandle)
243     {
244     errorStr = "Error loading plugin '"
245         + pluginFileName + "': '" + dlerror() + "'";
246     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
247     return -1;
248     }
249
250 isPluginLoaded = true;
251
252 PLUGIN * (*GetPlugin)();
253 GetPlugin = (PLUGIN * (*)())dlsym(libHandle, "GetPlugin");
254 if (!GetPlugin)
255     {
256     errorStr = std::string("GetPlugin() not found. ") + dlerror();
257     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
258     return -1;
259     }
260 plugin = GetPlugin();
261
262 if (!plugin)
263     {
264     errorStr = "Plugin was not created!";
265     printfd(__FILE__, "PLUGIN_LOADER::Load() - %s\n", errorStr.c_str());
266     return -1;
267     }
268
269 plugin->SetSettings(modSettings);
270 if (plugin->ParseSettings())
271     {
272     errorStr = plugin->GetStrError();
273     printfd(__FILE__, "PLUGIN_LOADER::Load() - Failed to parse settings. Plugin reports: '%s'\n", errorStr.c_str());
274     return -1;
275     }
276
277 return 0;
278 }
279 //-----------------------------------------------------------------------------
280 int PLUGIN_RUNNER::Unload()
281 {
282 if (isPluginLoaded)
283     {
284     if (dlclose(libHandle))
285         {
286         errorStr = "Failed to unload plugin '";
287         errorStr += pluginFileName + "': ";
288         errorStr += dlerror();
289         printfd(__FILE__, "PLUGIN_LOADER::Unload() - %s", errorStr.c_str());
290         return -1;
291         }
292     plugin = NULL;
293     isPluginLoaded = false;
294     }
295 return 0;
296 }
297 //-----------------------------------------------------------------------------