]> git.stg.codes - stg.git/blob - projects/stargazer/store_loader.cpp
Виправлено помилку що призводила до масових дисконнектів після масових
[stg.git] / projects / stargazer / store_loader.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 : Maxim Mamontov <faust@stargazer.dp.ua>
19  */
20
21 /*
22  $Revision: 1.6 $
23  $Date: 2010/03/04 12:24:19 $
24  $Author: faust $
25  */
26
27 /*
28  *  An implementation of RAII store plugin loader
29  */
30
31 #include <dlfcn.h>
32
33 #include "store_loader.h"
34 #include "common.h"
35
36 STORE_LOADER::STORE_LOADER(const SETTINGS & settings)
37     : isLoaded(false),
38       handle(NULL),
39       plugin(NULL),
40       errorStr(),
41       storeSettings(settings.GetStoreModuleSettings()),
42       pluginFileName(settings.GetModulesPath() + "/mod_" + storeSettings.moduleName + ".so")
43 {
44 }
45
46 STORE_LOADER::~STORE_LOADER()
47 {
48 Unload();
49 }
50
51 bool STORE_LOADER::Load()
52 {
53 printfd(__FILE__, "STORE_LOADER::Load()\n");
54 if (isLoaded)
55     {
56     return false;
57     }
58
59 if (pluginFileName.empty())
60     {
61     errorStr = "Empty store plugin filename";
62     printfd(__FILE__, "STORE_LOADER::Load - %s\n", errorStr.c_str());
63     return true;
64     }
65
66 handle = dlopen(pluginFileName.c_str(), RTLD_NOW);
67
68 if (!handle)
69     {
70     errorStr = "Error loading plugin '"
71         + pluginFileName + "': '" + dlerror() + "'";
72     printfd(__FILE__, "STORE_LOADER::Load - %s\n", errorStr.c_str());
73     return true;
74     }
75
76 isLoaded = true;
77
78 BASE_STORE * (*GetStore)();
79 GetStore = (BASE_STORE * (*)())dlsym(handle, "GetStore");
80 if (!GetStore)
81     {
82     errorStr = "GetStore not found.";
83     printfd(__FILE__, "STORE_LOADER::Load - %s\n", errorStr.c_str());
84     return true;
85     }
86
87 plugin = GetStore();
88
89 if (!plugin)
90     {
91     errorStr = "NULL store plugin";
92     printfd(__FILE__, "STORE_LOADER::Load - %s\n");
93     return true;
94     }
95
96 plugin->SetSettings(storeSettings);
97 if (plugin->ParseSettings())
98     {
99     errorStr = plugin->GetStrError();
100     printfd(__FILE__, "Failed to parse settings. Plugin reports: '%s'\n", errorStr.c_str());
101     return true;
102     }
103
104 return false;
105 }
106
107 bool STORE_LOADER::Unload()
108 {
109 printfd(__FILE__, "STORE_LOADER::Unload()\n");
110 if (!isLoaded)
111     {
112     return false;
113     }
114
115 if (dlclose(handle))
116     {
117     errorStr = "Failed to unload plugin: '";
118     errorStr += dlerror();
119     errorStr += "'";
120     printfd(__FILE__, "STORE_LOADER::Unload - %s\n", errorStr.c_str());
121     return true;
122     }
123
124 isLoaded = false;
125
126 return false;
127 }