]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/other/radius/config.cpp
68b8760f1347c29f240d08e97e5abfdd1342cfc1
[stg.git] / projects / stargazer / plugins / other / radius / config.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 "config.h"
22
23 #include "stg/common.h"
24
25 #include <vector>
26 #include <stdexcept>
27
28 #include <strings.h> // strncasecmp
29
30 using STG::Config;
31
32 namespace
33 {
34
35 struct ParserError : public std::runtime_error
36 {
37     ParserError(size_t pos, const std::string& message)
38         : runtime_error("Parsing error at position " + x2str(pos) + ". " + message),
39           position(pos),
40           error(message)
41     {}
42     virtual ~ParserError() throw() {}
43
44     size_t position;
45     std::string error;
46 };
47
48 size_t skipSpaces(const std::string& value, size_t start)
49 {
50     while (start < value.length() && std::isspace(value[start]))
51         ++start;
52     return start;
53 }
54
55 size_t checkChar(const std::string& value, size_t start, char ch)
56 {
57     if (start >= value.length())
58         throw ParserError(start, "Unexpected end of string. Expected '" + std::string(1, ch) + "'.");
59     if (value[start] != ch)
60         throw ParserError(start, "Expected '" + std::string(1, ch) + "', got '" + std::string(1, value[start]) + "'.");
61     return start + 1;
62 }
63
64 std::pair<size_t, std::string> readString(const std::string& value, size_t start)
65 {
66     std::string dest;
67     while (start < value.length() && !std::isspace(value[start]) &&
68            value[start] != ',' && value[start] != '(' && value[start] != ')')
69         dest.push_back(value[start++]);
70     if (dest.empty()) {
71         if (start == value.length())
72             throw ParserError(start, "Unexpected end of string. Expected string.");
73         else
74             throw ParserError(start, "Unexpected whitespace. Expected string.");
75     }
76     return std::make_pair(start, dest);
77 }
78
79 Config::Pairs toPairs(const std::vector<std::string>& values)
80 {
81     if (values.empty())
82         return Config::Pairs();
83     std::string value(values[0]);
84     Config::Pairs res;
85     size_t start = 0;
86     while (start < value.size()) {
87         Config::Pair pair;
88         start = skipSpaces(value, start);
89         if (!res.empty())
90         {
91             start = checkChar(value, start, ',');
92             start = skipSpaces(value, start);
93         }
94         size_t pairStart = start;
95         start = checkChar(value, start, '(');
96         const std::pair<size_t, std::string> key = readString(value, start);
97         start = key.first;
98         pair.first = key.second;
99         start = skipSpaces(value, start);
100         start = checkChar(value, start, ',');
101         start = skipSpaces(value, start);
102         const std::pair<size_t, std::string> val = readString(value, start);
103         start = val.first;
104         pair.second = val.second;
105         start = skipSpaces(value, start);
106         start = checkChar(value, start, ')');
107         if (res.find(pair.first) != res.end())
108             throw ParserError(pairStart, "Duplicate field.");
109         res.insert(pair);
110     }
111     return res;
112 }
113
114 bool toBool(const std::vector<std::string>& values)
115 {
116     if (values.empty())
117         return false;
118     std::string value(values[0]);
119     return strncasecmp(value.c_str(), "yes", 3) == 0;
120 }
121
122 std::string toString(const std::vector<std::string>& values)
123 {
124     if (values.empty())
125         return "";
126     return values[0];
127 }
128
129 uid_t toUID(const std::vector<std::string>& values)
130 {
131     if (values.empty())
132         return -1;
133     uid_t res = str2uid(values[0]);
134     if (res == static_cast<uid_t>(-1))
135         throw ParserError(0, "Invalid user name: '" + values[0] + "'");
136     return res;
137 }
138
139 gid_t toGID(const std::vector<std::string>& values)
140 {
141     if (values.empty())
142         return -1;
143     gid_t res = str2gid(values[0]);
144     if (res == static_cast<gid_t>(-1))
145         throw ParserError(0, "Invalid group name: '" + values[0] + "'");
146     return res;
147 }
148
149 mode_t toMode(const std::vector<std::string>& values)
150 {
151     if (values.empty())
152         return -1;
153     mode_t res = str2mode(values[0]);
154     if (res == static_cast<mode_t>(-1))
155         throw ParserError(0, "Invalid mode: '" + values[0] + "'");
156     return res;
157 }
158
159 template <typename T>
160 T toInt(const std::vector<std::string>& values)
161 {
162     if (values.empty())
163         return 0;
164     T res = 0;
165     if (str2x(values[0], res) == 0)
166         return res;
167     return 0;
168 }
169
170 uint16_t toPort(const std::string& value)
171 {
172     if (value.empty())
173         return 0;
174     uint16_t res = 0;
175     if (str2x(value, res) == 0)
176         return res;
177     throw ParserError(0, "'" + value + "' is not a valid port number.");
178 }
179
180 typedef std::map<std::string, Config::ReturnCode> Codes;
181
182 // One-time call to initialize the list of codes.
183 Codes getCodes()
184 {
185     Codes res;
186     res["reject"]   = Config::REJECT;
187     res["fail"]     = Config::FAIL;
188     res["ok"]       = Config::OK;
189     res["handled"]  = Config::HANDLED;
190     res["invalid"]  = Config::INVALID;
191     res["userlock"] = Config::USERLOCK;
192     res["notfound"] = Config::NOTFOUND;
193     res["noop"]     = Config::NOOP;
194     res["updated"]  = Config::UPDATED;
195     return res;
196 }
197
198 Config::ReturnCode toReturnCode(const std::vector<std::string>& values)
199 {
200     static Codes codes(getCodes());
201     if (values.empty())
202         return Config::REJECT;
203     std::string code = ToLower(values[0]);
204     const Codes::const_iterator it = codes.find(code);
205     if (it == codes.end())
206         return Config::REJECT;
207     return it->second;
208 }
209
210 Config::Pairs parseVector(const std::string& paramName, const std::vector<PARAM_VALUE>& params)
211 {
212     for (size_t i = 0; i < params.size(); ++i)
213         if (params[i].param == paramName)
214             return toPairs(params[i].value);
215     return Config::Pairs();
216 }
217
218 Config::ReturnCode parseReturnCode(const std::string& paramName, const std::vector<PARAM_VALUE>& params)
219 {
220     for (size_t i = 0; i < params.size(); ++i)
221         if (params[i].param == paramName)
222             return toReturnCode(params[i].value);
223     return Config::REJECT;
224 }
225
226 bool parseBool(const std::string& paramName, const std::vector<PARAM_VALUE>& params)
227 {
228     for (size_t i = 0; i < params.size(); ++i)
229         if (params[i].param == paramName)
230             return toBool(params[i].value);
231     return false;
232 }
233
234 std::string parseString(const std::string& paramName, const std::vector<PARAM_VALUE>& params)
235 {
236     for (size_t i = 0; i < params.size(); ++i)
237         if (params[i].param == paramName)
238             return toString(params[i].value);
239     return "";
240 }
241
242 std::string parseAddress(Config::Type connectionType, const std::string& value)
243 {
244     size_t pos = value.find_first_of(':');
245     if (pos == std::string::npos)
246         throw ParserError(0, "Connection type is not specified. Should be either 'unix' or 'tcp'.");
247     if (connectionType == Config::UNIX)
248         return value.substr(pos + 1);
249     std::string address(value.substr(pos + 1));
250     pos = address.find_first_of(':', pos + 1);
251     if (pos == std::string::npos)
252         throw ParserError(0, "Port is not specified.");
253     return address.substr(0, pos - 1);
254 }
255
256 std::string parsePort(Config::Type connectionType, const std::string& value)
257 {
258     size_t pos = value.find_first_of(':');
259     if (pos == std::string::npos)
260         throw ParserError(0, "Connection type is not specified. Should be either 'unix' or 'tcp'.");
261     if (connectionType == Config::UNIX)
262         return "";
263     std::string address(value.substr(pos + 1));
264     pos = address.find_first_of(':', pos + 1);
265     if (pos == std::string::npos)
266         throw ParserError(0, "Port is not specified.");
267     return address.substr(pos + 1);
268 }
269
270 Config::Type parseConnectionType(const std::string& address)
271 {
272     size_t pos = address.find_first_of(':');
273     if (pos == std::string::npos)
274         throw ParserError(0, "Connection type is not specified. Should be either 'unix' or 'tcp'.");
275     std::string type = ToLower(address.substr(0, pos));
276     if (type == "unix")
277         return Config::UNIX;
278     else if (type == "tcp")
279         return Config::TCP;
280     throw ParserError(0, "Invalid connection type. Should be either 'unix' or 'tcp', got '" + type + "'");
281 }
282
283 Config::Section parseSection(const std::string& paramName, const std::vector<PARAM_VALUE>& params)
284 {
285     for (size_t i = 0; i < params.size(); ++i)
286         if (params[i].param == paramName)
287             return Config::Section(parseVector("match", params[i].sections),
288                                    parseVector("modify", params[i].sections),
289                                    parseVector("reply", params[i].sections),
290                                    parseReturnCode("no_match", params[i].sections));
291     return Config::Section();
292 }
293
294 uid_t parseUID(const std::string& paramName, const std::vector<PARAM_VALUE>& params)
295 {
296     for (size_t i = 0; i < params.size(); ++i)
297         if (params[i].param == paramName)
298             return toUID(params[i].value);
299     return -1;
300 }
301
302 gid_t parseGID(const std::string& paramName, const std::vector<PARAM_VALUE>& params)
303 {
304     for (size_t i = 0; i < params.size(); ++i)
305         if (params[i].param == paramName)
306             return toGID(params[i].value);
307     return -1;
308 }
309
310 mode_t parseMode(const std::string& paramName, const std::vector<PARAM_VALUE>& params)
311 {
312     for (size_t i = 0; i < params.size(); ++i)
313         if (params[i].param == paramName)
314             return toMode(params[i].value);
315     return -1;
316 }
317
318 } // namespace anonymous
319
320 Config::Config(const MODULE_SETTINGS& settings)
321     : autz(parseSection("autz", settings.moduleParams)),
322       auth(parseSection("auth", settings.moduleParams)),
323       postauth(parseSection("postauth", settings.moduleParams)),
324       preacct(parseSection("preacct", settings.moduleParams)),
325       acct(parseSection("acct", settings.moduleParams)),
326       verbose(parseBool("verbose", settings.moduleParams)),
327       address(parseString("bind_address", settings.moduleParams)),
328       connectionType(parseConnectionType(address)),
329       bindAddress(parseAddress(connectionType, address)),
330       portStr(parsePort(connectionType, address)),
331       port(toPort(portStr)),
332       key(parseString("key", settings.moduleParams)),
333       sockUID(parseUID("sock_owner", settings.moduleParams)),
334       sockGID(parseGID("sock_group", settings.moduleParams)),
335       sockMode(parseMode("sock_mode", settings.moduleParams))
336 {
337 }