]> git.stg.codes - stg.git/blob - projects/stargazer/settings_impl.cpp
Added new files to build.
[stg.git] / projects / stargazer / settings_impl.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 #include "settings_impl.h"
22
23 #include "stg/logger.h"
24 #include "stg/dotconfpp.h"
25 #include "stg/common.h"
26
27 #include <stdexcept>
28 #include <cstring>
29 #include <cerrno>
30
31 namespace
32 {
33
34 struct Error : public std::runtime_error
35 {
36     Error(const std::string& message) : runtime_error(message) {}
37 };
38
39 std::vector<std::string> toValues(const DOTCONFDocumentNode& node)
40 {
41     std::vector<std::string> values;
42
43     size_t i = 0;
44     const char* value = NULL;
45     while ((value = node.getValue(i++)) != NULL)
46         values.push_back(value);
47
48     return values;
49 }
50
51 std::vector<PARAM_VALUE> toPVS(const DOTCONFDocumentNode& node)
52 {
53     std::vector<PARAM_VALUE> pvs;
54
55     const DOTCONFDocumentNode* child = node.getChildNode();
56     while (child != NULL)
57         {
58         if (child->getName() == NULL)
59             continue;
60
61         if (child->getChildNode() == NULL)
62             pvs.push_back(PARAM_VALUE(child->getName(), toValues(*child)));
63         else
64             pvs.push_back(PARAM_VALUE(child->getName(), toValues(*child), toPVS(*child)));
65
66         child = child->getNextNode();
67         }
68
69     return pvs;
70 }
71
72 unsigned toPeriod(const char* value)
73 {
74     if (value == NULL)
75         throw Error("No detail stat period value.");
76
77     std::string period(value);
78     if (period == "1")
79         return dsPeriod_1;
80     else if (period == "1/2")
81         return dsPeriod_1_2;
82     else if (period == "1/4")
83         return dsPeriod_1_4;
84     else if (period == "1/6")
85         return dsPeriod_1_6;
86
87     throw Error("Invalid detail stat period value: '" + period + "'. Should be one of '1', '1/2', '1/4' or '1/6'.");
88 }
89
90 }
91
92 //-----------------------------------------------------------------------------
93 SETTINGS_IMPL::SETTINGS_IMPL(const std::string & cd)
94     : modulesPath("/usr/lib/stg"),
95       dirName(DIR_NUM),
96       confDir(cd.empty() ? "/etc/stargazer" : cd),
97       scriptsDir(confDir),
98       rules(confDir + "/rules"),
99       logFile("/var/log/stargazer.log"),
100       pidFile("/var/run/stargazer.pid"),
101       monitorDir("/var/stargazer/monitoring"),
102       monitoring(false),
103       detailStatWritePeriod(dsPeriod_1_6),
104       statWritePeriod(10),
105       stgExecMsgKey(5555),
106       executersNum(1),
107       fullFee(false),
108       dayFee(0),
109       dayResetTraff(0),
110       spreadFee(false),
111       freeMbAllowInet(false),
112       dayFeeIsLastDay(false),
113       writeFreeMbTraffCost(false),
114       showFeeInCash(true),
115       messageTimeout(0),
116       feeChargeType(0),
117       reconnectOnTariffChange(false),
118       logger(GetStgLogger())
119 {
120 }
121 //-----------------------------------------------------------------------------
122 void SETTINGS_IMPL::ErrorCallback(void * data, const char * buf)
123 {
124     printfd(__FILE__, "SETTINGS_IMPL::ErrorCallback() - %s\n", buf);
125     SETTINGS_IMPL * settings = static_cast<SETTINGS_IMPL *>(data);
126     settings->logger("%s", buf);
127 }
128 //-----------------------------------------------------------------------------
129 int SETTINGS_IMPL::ReadSettings()
130 {
131 const char * requiredOptions[] = {
132     "ModulesPath",
133     "Modules",
134     "StoreModule",
135     "Rules",
136     "LogFile",
137     "DetailStatWritePeriod",
138     "DayFee",
139     "DayResetTraff",
140     "SpreadFee",
141     "FreeMbAllowInet",
142     "DayFeeIsLastDay",
143     "WriteFreeMbTraffCost",
144     NULL
145     };
146 int storeModulesCount = 0;
147 modulesSettings.clear();
148
149 DOTCONFDocument conf(DOTCONFDocument::CASEINSENSITIVE);
150 conf.setErrorCallback(SETTINGS_IMPL::ErrorCallback, this);
151 conf.setRequiredOptionNames(requiredOptions);
152 std::string confFile = confDir + "/stargazer.conf";
153
154 if(conf.setContent(confFile.c_str()) != 0)
155     {
156     strError = "Cannot read file " + confFile;
157     return -1;
158     }
159
160 const DOTCONFDocumentNode * node = conf.getFirstNode();
161
162 while (node)
163     {
164     if (strcasecmp(node->getName(), "ScriptDir") == 0)
165         {
166         scriptsDir = node->getValue(0);
167         }
168
169     if (strcasecmp(node->getName(), "LogFile") == 0)
170         {
171         logFile = node->getValue(0);
172         }
173
174     if (strcasecmp(node->getName(), "PIDFile") == 0)
175         {
176         pidFile = node->getValue(0);
177         }
178
179     if (strcasecmp(node->getName(), "ModulesPath") == 0)
180         {
181         modulesPath = node->getValue(0);
182         }
183
184     if (strcasecmp(node->getName(), "Rules") == 0)
185         {
186         rules = node->getValue(0);
187         }
188
189     if (strcasecmp(node->getName(), "DetailStatWritePeriod") == 0)
190         {
191         try
192         {
193             detailStatWritePeriod = toPeriod(node->getValue(0));
194         }
195         catch (const Error& error)
196         {
197             strError = error.what();
198             return -1;
199         }
200         }
201
202     if (strcasecmp(node->getName(), "StatWritePeriod") == 0)
203         {
204         if (ParseUnsignedInRange(node->getValue(0), 1, 1440, &statWritePeriod) != 0)
205             {
206             strError = "Incorrect StatWritePeriod value: \'" + std::string(node->getValue(0)) + "\'";
207             return -1;
208             }
209         }
210
211     if (strcasecmp(node->getName(), "ExecMsgKey") == 0)
212         {
213         if (ParseInt(node->getValue(0), &stgExecMsgKey) != 0)
214             {
215             strError = "Incorrect ExecMsgKey value: \'" + std::string(node->getValue(0)) + "\'";
216             return -1;
217             }
218         }
219
220     if (strcasecmp(node->getName(), "ExecutersNum") == 0)
221         {
222         if (ParseUnsignedInRange(node->getValue(0), 1, 1024, &executersNum) != 0)
223             {
224             strError = "Incorrect ExecutersNum value: \'" + std::string(node->getValue(0)) + "\'";
225             return -1;
226             }
227         }
228
229     if (strcasecmp(node->getName(), "DayFee") == 0)
230         {
231         if (ParseUnsignedInRange(node->getValue(0), 0, 31, &dayFee) != 0)
232             {
233             strError = "Incorrect DayFee value: \'" + std::string(node->getValue(0)) + "\'";
234             return -1;
235             }
236         }
237
238     if (strcasecmp(node->getName(), "FullFee") == 0)
239         {
240         if (ParseYesNo(node->getValue(0), &fullFee) != 0)
241             {
242             strError = "Incorrect FullFee value: \'" + std::string(node->getValue(0)) + "\'";
243             return -1;
244             }
245         }
246
247     if (strcasecmp(node->getName(), "DayResetTraff") == 0)
248         {
249         if (ParseUnsignedInRange(node->getValue(0), 0, 31, &dayResetTraff) != 0)
250             {
251             strError = "Incorrect DayResetTraff value: \'" + std::string(node->getValue(0)) + "\'";
252             return -1;
253             }
254         }
255
256     if (strcasecmp(node->getName(), "SpreadFee") == 0)
257         {
258         if (ParseYesNo(node->getValue(0), &spreadFee) != 0)
259             {
260             strError = "Incorrect SpreadFee value: \'" + std::string(node->getValue(0)) + "\'";
261             return -1;
262             }
263         }
264
265     if (strcasecmp(node->getName(), "FreeMbAllowInet") == 0)
266         {
267         if (ParseYesNo(node->getValue(0), &freeMbAllowInet) != 0)
268             {
269             strError = "Incorrect FreeMbAllowInet value: \'" + std::string(node->getValue(0)) + "\'";
270             return -1;
271             }
272         }
273
274     if (strcasecmp(node->getName(), "DayFeeIsLastDay") == 0)
275         {
276         if (ParseYesNo(node->getValue(0), &dayFeeIsLastDay) != 0)
277             {
278             strError = "Incorrect DayFeeIsLastDay value: \'" + std::string(node->getValue(0)) + "\'";
279             return -1;
280             }
281         }
282
283     if (strcasecmp(node->getName(), "WriteFreeMbTraffCost") == 0)
284         {
285         if (ParseYesNo(node->getValue(0), &writeFreeMbTraffCost) != 0)
286             {
287             strError = "Incorrect WriteFreeMbTraffCost value: \'" + std::string(node->getValue(0)) + "\'";
288             return -1;
289             }
290         }
291
292     if (strcasecmp(node->getName(), "ShowFeeInCash") == 0)
293         {
294         if (ParseYesNo(node->getValue(0), &showFeeInCash) != 0)
295             {
296             strError = "Incorrect ShowFeeInCash value: \'" + std::string(node->getValue(0)) + "\'";
297             return -1;
298             }
299         }
300
301     if (strcasecmp(node->getName(), "MonitorDir") == 0)
302         {
303         monitorDir = node->getValue(0);
304         struct stat stat;
305         monitoring = false;
306
307         if (!lstat(monitorDir.c_str(), &stat) && S_ISDIR(stat.st_mode))
308             {
309             monitoring = true;
310             }
311         }
312
313     if (strcasecmp(node->getName(), "MessageTimeout") == 0)
314         {
315         if (ParseUnsigned(node->getValue(0), &messageTimeout) != 0)
316             {
317             strError = "Incorrect MessageTimeout value: \'" + std::string(node->getValue(0)) + "\'";
318             return -1;
319             }
320         }
321
322     if (strcasecmp(node->getName(), "FeeChargeType") == 0)
323         {
324         if (ParseUnsignedInRange(node->getValue(0), 0, 3, &feeChargeType) != 0)
325             {
326             strError = "Incorrect FeeChargeType value: \'" + std::string(node->getValue(0)) + "\'";
327             return -1;
328             }
329         }
330
331     if (strcasecmp(node->getName(), "ReconnectOnTariffChange") == 0)
332         {
333         if (ParseYesNo(node->getValue(0), &reconnectOnTariffChange) != 0)
334             {
335             strError = "Incorrect ReconnectOnTariffChange value: \'" + std::string(node->getValue(0)) + "\'";
336             return -1;
337             }
338         }
339
340     if (strcasecmp(node->getName(), "DirNames") == 0)
341         {
342         const DOTCONFDocumentNode * child = node->getChildNode();
343         if (child)
344             {
345             const DOTCONFDocumentNode * dirNameNode;
346             dirName.reserve(DIR_NUM);
347             for (int i = 0; i < DIR_NUM; i++)
348                 {
349                 char strDirName[12];
350                 sprintf(strDirName, "DirName%d", i);
351                 dirNameNode = conf.findNode(strDirName, node);
352                 if (dirNameNode && dirNameNode->getValue(0))
353                     {
354                     dirName[i] = dirNameNode->getValue(0);
355                     }
356                 }
357             }
358         }
359
360     if (strcasecmp(node->getName(), "StoreModule") == 0)
361         {
362         if (node->getValue(1))
363             {
364             strError = "Unexpected \'" + std::string(node->getValue(1)) + "\'.";
365             return -1;
366             }
367
368         if (storeModulesCount)
369             {
370             strError = "Should be only one StoreModule.";
371             return -1;
372             }
373         storeModulesCount++;
374
375         if (node->getValue(0) == NULL)
376             {
377             strError = "No module name in the StoreModule section.";
378             return -1;
379             }
380         storeModuleSettings.moduleName = node->getValue(0);
381         storeModuleSettings.moduleParams = toPVS(*node);
382         }
383
384     if (strcasecmp(node->getName(), "Modules") == 0)
385         {
386         if (node->getValue(0))
387             {
388             strError = "Unexpected \'" + std::string(node->getValue(0)) + "\'.";
389             return -1;
390             }
391         const DOTCONFDocumentNode * child = node->getChildNode();
392         while (child)
393             {
394             if (strcasecmp(child->getName(), "Module") != 0)
395                 {
396                 child = child->getNextNode();
397                 continue;
398                 }
399
400             if (child->getValue(0) == NULL)
401                 {
402                 strError = "No module name in the Module section.";
403                 return -1;
404                 }
405
406             modulesSettings.push_back(MODULE_SETTINGS(child->getValue(0), toPVS(*child)));
407
408             child = child->getNextNode();
409             }
410         }
411
412     if (strcasecmp(node->getName(), "ScriptParams") == 0)
413         {
414         for (int i = 0; node->getValue(i) != NULL; ++i)
415             {
416             scriptParams.push_back(node->getValue(i));
417             }
418         }
419     node = node->getNextNode();
420     }
421
422 return 0;
423 }
424 //-----------------------------------------------------------------------------