]> git.stg.codes - stg.git/blob - projects/stargazer/plugin_runner.cpp
bcf68bee861197e69d3535733c7849eafce624ad
[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  *    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 }