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