]> git.stg.codes - stg.git/blob - projects/sgauthstress/settings.cpp
9055ceed24df59e9b11190f92a79ca2782653ed1
[stg.git] / projects / sgauthstress / settings.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 #include <cstring>
22 #include <cassert>
23 #include <vector>
24
25 #include "stg/dotconfpp.h"
26 #include "stg/module_settings.h"
27 #include "stg/common.h"
28
29 #include "settings.h"
30
31 SETTINGS::SETTINGS()
32     : port(0),
33       localPort(0),
34       confFile("/etc/sgauth.conf")
35 {
36 }
37 //-----------------------------------------------------------------------------
38 int ParseYesNo(const string & value, bool * val)
39 {
40 if (0 == strcasecmp(value.c_str(), "yes"))
41     {
42     *val = true;
43     return 0;
44     }
45 if (0 == strcasecmp(value.c_str(), "no"))
46     {
47     *val = false;
48     return 0;
49     }
50
51 return -1;
52 }
53 //-----------------------------------------------------------------------------
54 int ParseInt(const string & value, int * val)
55 {
56 if (str2x<int>(value, *val))
57     return -1;
58
59 return 0;
60 }
61 //-----------------------------------------------------------------------------
62 int ParseUnsigned(const string & value, unsigned * val)
63 {
64 if (str2x<unsigned>(value, *val))
65     return -1;
66
67 return 0;
68 }
69 //-----------------------------------------------------------------------------
70 int ParseIntInRange(const string & value, int min, int max, int * val)
71 {
72 if (ParseInt(value, val) != 0)
73     return -1;
74
75 if (*val < min || *val > max)
76     return -1;
77
78 return 0;
79 }
80 //-----------------------------------------------------------------------------
81 int ParseUnsignedInRange(const string & value, unsigned min, unsigned max, unsigned * val)
82 {
83 if (ParseUnsigned(value, val) != 0)
84     return -1;
85
86 if (*val < min || *val > max)
87     return -1;
88
89 return 0;
90 }
91 //-----------------------------------------------------------------------------
92 int ParseModuleSettings(const DOTCONFDocumentNode * node, std::vector<PARAM_VALUE> * params)
93 {
94 assert(node && "DOTCONFDocumentNode must not be NULL!");
95
96 const DOTCONFDocumentNode * childNode = node->getChildNode();
97 while (childNode)
98     {
99     PARAM_VALUE pv;
100     pv.param = childNode->getName();
101     int i = 0;
102     while ((value = childNode->getValue(i++)) != NULL)
103         pv.value.push_back(value);
104     params->push_back(pv);
105     childNode = childNode->getNextNode();
106     }
107
108 return 0;
109 }
110 //-----------------------------------------------------------------------------
111 int SETTINGS::ReadSettings()
112 {
113 const char * requiredOptions[] = {
114     "ModulesPath",
115     "StoreModule",
116     "Login",
117     "Password",
118     "ServerName",
119     "ServerPort"
120     NULL
121     };
122
123 DOTCONFDocument conf(DOTCONFDocument::CASEINSENSITIVE);
124 conf.setRequiredOptionNames(requiredOptions);
125
126 if(conf.setContent(confFile.c_str()) != 0)
127     {
128     strError = "Cannot read file " + confFile + ".";
129     printfd(__FILE__, "SETTINGS::ReadSettings() - %s\n", strError.c_str());
130     return -1;
131     }
132
133 const DOTCONFDocumentNode * node = conf.getFirstNode();
134
135 int storeModulesCount = 0;
136 while (node)
137     {
138     if (strcasecmp(node->getName(), "ModulesPath") == 0)
139         modulesPath = node->getValue(0);
140     else if (strcasecmp(node->getName(), "Login") == 0)
141         login = node->getValue(0);
142     else if (strcasecmp(node->getName(), "Password") == 0)
143         password = node->getValue(0);
144     else if (strcasecmp(node->getName(), "ServerName") == 0)
145         serverName = node->getValue(0);
146     else if (strcasecmp(node->getName(), "ServerPort") == 0)
147         if (ParseIntInRange(node->getValue(0), 1, 65535, &port))
148             {
149             strError = "Parameter 'ServerPort' is not valid.";
150             printfd(__FILE__, "SETTINGS::ReadSettings() - %s\n", strError.c_str());
151             return -1;
152             }
153     else if (strcasecmp(node->getName(), "LocalPort") == 0)
154         if (ParseIntInRange(node->getValue(0), 0, 65535, &localPort))
155             {
156             strError = "Parameter 'LocalPort' is not valid.";
157             printfd(__FILE__, "SETTINGS::ReadSettings() - %s\n", strError.c_str());
158             return -1;
159             }
160     else if (strcasecmp(node->getName(), "StoreModule") == 0)
161         {
162         if (node->getValue(1))
163             {
164             strError = "Unexpected \'" + std::string(node->getValue(1)) + "\'.";
165             printfd(__FILE__, "SETTINGS::ReadSettings() - %s\n", strError.c_str());
166             return -1;
167             }
168
169         if (storeModulesCount)
170             {
171             strError = "Should be only one source StoreModule.";
172             printfd(__FILE__, "SETTINGS::ReadSettings() - %s\n", strError.c_str());
173             return -1;
174             }
175         ++storeModulesCount;
176
177         storeModuleSettings.moduleName = node->getValue(0);
178         if (ParseModuleSettings(node, &storeModuleSettings.moduleParams))
179             {
180             strError = "Failed to parse store module settings";
181             printfd(__FILE__, "SETTINGS::ReadSettings() - %s\n", strError.c_str());
182             return -1;
183             }
184         }
185
186     node = node->getNextNode();
187     }
188
189 return 0;
190 }