]> git.stg.codes - stg.git/blob - projects/stargazer/plugins/configuration/rpcconfig/users_methods.cpp
Add std namespace to the string in rpcconfig.cpp
[stg.git] / projects / stargazer / plugins / configuration / rpcconfig / users_methods.cpp
1 #include <cerrno>
2
3 #include "users_methods.h"
4
5 #include "users.h"
6 #include "admins.h"
7 #include "tariffs.h"
8
9 #include "rpcconfig.h"
10 #include "user_helper.h"
11 #include "user_ips.h"
12 #include "utils.h"
13 #include "common.h"
14
15 //------------------------------------------------------------------------------
16
17 void METHOD_USER_GET::execute(xmlrpc_c::paramList const & paramList,
18                               xmlrpc_c::value *   const   retvalPtr)
19 {
20 std::string cookie = paramList.getString(0);
21 std::string login = paramList.getString(1);
22 std::string enc;
23 paramList.verifyEnd(2);
24
25 std::map<std::string, xmlrpc_c::value> structVal;
26 ADMIN_INFO adminInfo;
27
28 if (config->GetAdminInfo(cookie, &adminInfo))
29     {
30     structVal["result"] = xmlrpc_c::value_boolean(false);
31     *retvalPtr = xmlrpc_c::value_struct(structVal);
32     return;
33     }
34
35 user_iter u;
36
37 if (users->FindByName(login, &u))
38     {
39     structVal["result"] = xmlrpc_c::value_boolean(false);
40     *retvalPtr = xmlrpc_c::value_struct(structVal);
41     return;
42     }
43
44 USER_HELPER uhelper(u);
45
46 if (!adminInfo.priviledges.userConf || !adminInfo.priviledges.userPasswd)
47     {
48     uhelper.GetUserInfo(retvalPtr, true);
49     return;
50     }
51
52 uhelper.GetUserInfo(retvalPtr);
53 }
54
55 //------------------------------------------------------------------------------
56
57 void METHOD_USER_ADD::execute(xmlrpc_c::paramList const & paramList,
58                               xmlrpc_c::value *   const   retvalPtr)
59 {
60 std::string cookie = paramList.getString(0);
61 std::string login = paramList.getString(1);
62 std::string enc;
63 paramList.verifyEnd(2);
64
65 ADMIN_INFO adminInfo;
66
67 if (config->GetAdminInfo(cookie, &adminInfo))
68     {
69     *retvalPtr = xmlrpc_c::value_boolean(false);
70     return;
71     }
72
73 ADMIN admin;
74
75 if (admins->FindAdmin(adminInfo.admin, &admin))
76     {
77     *retvalPtr = xmlrpc_c::value_boolean(false);
78     return;
79     }
80
81 user_iter u;
82
83 if (users->FindByName(login, &u))
84     {
85     if (users->Add(login, admin))
86         {
87         *retvalPtr = xmlrpc_c::value_boolean(false);
88         return;
89         }
90
91     *retvalPtr = xmlrpc_c::value_boolean(true);
92     return;
93     }
94     
95 *retvalPtr = xmlrpc_c::value_boolean(false);
96 return;
97 }
98
99 //------------------------------------------------------------------------------
100
101 void METHOD_USER_DEL::execute(xmlrpc_c::paramList const & paramList,
102                               xmlrpc_c::value *   const   retvalPtr)
103 {
104 std::string cookie = paramList.getString(0);
105 std::string login = paramList.getString(1);
106 std::string enc;
107 paramList.verifyEnd(2);
108
109 ADMIN_INFO adminInfo;
110
111 if (config->GetAdminInfo(cookie, &adminInfo))
112     {
113     *retvalPtr = xmlrpc_c::value_boolean(false);
114     return;
115     }
116
117 ADMIN admin;
118
119 if (admins->FindAdmin(adminInfo.admin, &admin))
120     {
121     *retvalPtr = xmlrpc_c::value_boolean(false);
122     return;
123     }
124
125 user_iter u;
126
127 if (users->FindByName(login, &u))
128     {
129     users->Del(login, admin);
130     *retvalPtr = xmlrpc_c::value_boolean(true);
131     return;
132     }
133
134 *retvalPtr = xmlrpc_c::value_boolean(false);
135 return;
136 }
137
138 //------------------------------------------------------------------------------
139
140 void METHOD_USERS_GET::execute(xmlrpc_c::paramList const & paramList,
141                                xmlrpc_c::value *   const   retvalPtr)
142 {
143 std::string cookie = paramList.getString(0);
144 std::string enc;
145 paramList.verifyEnd(1);
146
147 std::map<std::string, xmlrpc_c::value> structVal;
148 std::vector<xmlrpc_c::value> retval;
149 ADMIN_INFO adminInfo;
150
151 if (config->GetAdminInfo(cookie, &adminInfo))
152     {
153     structVal["result"] = xmlrpc_c::value_boolean(false);
154     *retvalPtr = xmlrpc_c::value_struct(structVal);
155     return;
156     }
157
158 bool hidePassword = !adminInfo.priviledges.userConf ||
159                     !adminInfo.priviledges.userPasswd;
160
161 user_iter u;
162
163 int h = users->OpenSearch();
164 if (!h)
165     {
166     printfd(__FILE__, "users->OpenSearch() error\n");
167     users->CloseSearch(h);
168     return;
169     }
170
171 while (1)
172     {
173     if (users->SearchNext(h, &u))
174         {
175         break;
176         }
177
178     xmlrpc_c::value info;
179
180     USER_HELPER uhelper(u);
181
182     uhelper.GetUserInfo(&info, hidePassword);
183
184     retval.push_back(info);
185     }
186
187 *retvalPtr = xmlrpc_c::value_array(retval);
188 }
189
190 //------------------------------------------------------------------------------
191
192 void METHOD_USER_CHG::execute(xmlrpc_c::paramList const & paramList,
193                               xmlrpc_c::value *   const   retvalPtr)
194 {
195 std::string cookie = paramList.getString(0);
196 std::string login = paramList.getString(1);
197 xmlrpc_c::value_struct info(paramList.getStruct(2));
198 std::string enc;
199 paramList.verifyEnd(3);
200
201 ADMIN_INFO adminInfo;
202
203 if (config->GetAdminInfo(cookie, &adminInfo))
204     {
205     *retvalPtr = xmlrpc_c::value_boolean(false);
206     return;
207     }
208
209 ADMIN admin;
210
211 if (admins->FindAdmin(adminInfo.admin, &admin))
212     {
213     *retvalPtr = xmlrpc_c::value_boolean(false);
214     return;
215     }
216
217 user_iter u;
218
219 if (users->FindByName(login, &u))
220     {
221     *retvalPtr = xmlrpc_c::value_boolean(false);
222     return;
223     }
224
225 USER_HELPER uhelper(u);
226
227 if (!adminInfo.priviledges.userConf || !adminInfo.priviledges.userPasswd)
228     {
229     uhelper.SetUserInfo(info, admin, login, *store, tariffs);
230     }
231 else
232     {
233     uhelper.SetUserInfo(info, admin, login, *store, tariffs);
234     }
235
236 u->WriteConf();
237 u->WriteStat();
238
239 *retvalPtr = xmlrpc_c::value_boolean(true);
240 }
241
242 //------------------------------------------------------------------------------
243
244 void METHOD_USER_CASH_ADD::execute(xmlrpc_c::paramList const & paramList,
245                                    xmlrpc_c::value *   const   retvalPtr)
246 {
247 std::string cookie = paramList.getString(0);
248 std::string login = paramList.getString(1);
249 double amount = paramList.getDouble(2);
250 std::string comment = IconvString(paramList.getString(3), "UTF-8", "KOI8-R");
251 std::string enc;
252 paramList.verifyEnd(4);
253
254 ADMIN_INFO adminInfo;
255
256 if (config->GetAdminInfo(cookie, &adminInfo))
257     {
258     *retvalPtr = xmlrpc_c::value_boolean(false);
259     return;
260     }
261
262 ADMIN admin;
263
264 if (admins->FindAdmin(adminInfo.admin, &admin))
265     {
266     *retvalPtr = xmlrpc_c::value_boolean(false);
267     return;
268     }
269
270 user_iter u;
271
272 if (users->FindByName(login, &u))
273     {
274     *retvalPtr = xmlrpc_c::value_boolean(false);
275     return;
276     }
277
278 double cash = u->property.cash.Get();
279 cash += amount;
280
281 if (!u->property.cash.Set(cash, admin, login, store, comment))
282     {
283     *retvalPtr = xmlrpc_c::value_boolean(false);
284     return;
285     }
286
287 u->WriteStat();
288
289 *retvalPtr = xmlrpc_c::value_boolean(true);
290 }
291
292 //------------------------------------------------------------------------------
293
294 void METHOD_USER_CASH_SET::execute(xmlrpc_c::paramList const & paramList,
295                                    xmlrpc_c::value *   const   retvalPtr)
296 {
297 std::string cookie = paramList.getString(0);
298 std::string login = paramList.getString(1);
299 double cash = paramList.getDouble(2);
300 std::string comment = IconvString(paramList.getString(3), "UTF-8", "KOI8-R");
301 std::string enc;
302 paramList.verifyEnd(4);
303
304 ADMIN_INFO adminInfo;
305
306 if (config->GetAdminInfo(cookie, &adminInfo))
307     {
308     *retvalPtr = xmlrpc_c::value_boolean(false);
309     return;
310     }
311
312 ADMIN admin;
313
314 if (admins->FindAdmin(adminInfo.admin, &admin))
315     {
316     *retvalPtr = xmlrpc_c::value_boolean(false);
317     return;
318     }
319
320 user_iter u;
321
322 if (users->FindByName(login, &u))
323     {
324     *retvalPtr = xmlrpc_c::value_boolean(false);
325     return;
326     }
327
328 if (!u->property.cash.Set(cash, admin, login, store, comment))
329     {
330     *retvalPtr = xmlrpc_c::value_boolean(false);
331     return;
332     }
333
334 u->WriteStat();
335
336 *retvalPtr = xmlrpc_c::value_boolean(true);
337 }
338
339 //------------------------------------------------------------------------------
340
341 void METHOD_USER_TARIFF_CHANGE::execute(xmlrpc_c::paramList const & paramList,
342                                         xmlrpc_c::value *   const   retvalPtr)
343 {
344 std::string cookie = paramList.getString(0);
345 std::string login = paramList.getString(1);
346 std::string tariff = paramList.getString(2);
347 bool delayed = paramList.getBoolean(3);
348 std::string comment = IconvString(paramList.getString(4), "UTF-8", "KOI8-R");
349 std::string enc;
350 paramList.verifyEnd(5);
351
352 ADMIN_INFO adminInfo;
353
354 if (config->GetAdminInfo(cookie, &adminInfo))
355     {
356     *retvalPtr = xmlrpc_c::value_boolean(false);
357     return;
358     }
359
360 ADMIN admin;
361
362 if (admins->FindAdmin(adminInfo.admin, &admin))
363     {
364     *retvalPtr = xmlrpc_c::value_boolean(false);
365     return;
366     }
367
368 user_iter u;
369
370 if (users->FindByName(login, &u))
371     {
372     *retvalPtr = xmlrpc_c::value_boolean(false);
373     return;
374     }
375
376 if (tariffs->FindByName(tariff))
377     {
378     if (delayed)
379         {
380         if (u->property.nextTariff.Set(tariff,
381                                        admin,
382                                        login,
383                                        store))
384             {
385             u->WriteConf();
386             *retvalPtr = xmlrpc_c::value_boolean(true);
387             return;
388             }
389         }
390     else
391         {
392         if (u->property.tariffName.Set(tariff,
393                                        admin,
394                                        login,
395                                        store))
396             {
397             u->WriteConf();
398             *retvalPtr = xmlrpc_c::value_boolean(true);
399             return;
400             }
401         }
402     }
403
404 *retvalPtr = xmlrpc_c::value_boolean(false);
405 }
406
407 //------------------------------------------------------------------------------
408
409 void METHOD_GET_ONLINE_IPS::execute(xmlrpc_c::paramList const & paramList,
410                                     xmlrpc_c::value *   const   retvalPtr)
411 {
412 std::string cookie = paramList.getString(0);
413 std::vector<xmlrpc_c::value> subnetsStr = paramList.getArray(1);
414 paramList.verifyEnd(2);
415
416 std::vector<IP_MASK> subnets;
417
418 std::vector<xmlrpc_c::value>::iterator it;
419
420 for (it = subnetsStr.begin(); it != subnetsStr.end(); ++it)
421     {
422     IP_MASK ipm;
423     if (ParseNet(xmlrpc_c::value_string(*it), ipm))
424         {
425         printfd(__FILE__, "METHOD_GET_ONLINE_IPS::execute(): Failed to parse subnet ('%s')\n", std::string(xmlrpc_c::value_string(*it)).c_str());
426         }
427     else
428         {
429         subnets.push_back(ipm);
430         }
431     }
432
433 std::map<std::string, xmlrpc_c::value> structVal;
434 ADMIN_INFO adminInfo;
435
436 if (config->GetAdminInfo(cookie, &adminInfo))
437     {
438     structVal["result"] = xmlrpc_c::value_boolean(false);
439     *retvalPtr = xmlrpc_c::value_struct(structVal);
440     return;
441     }
442
443 std::vector<xmlrpc_c::value> ips;
444
445 user_iter u;
446
447 int handle = users->OpenSearch();
448 if (!handle)
449     {
450     printfd(__FILE__, "users->OpenSearch() error\n");
451     users->CloseSearch(handle);
452     return;
453     }
454
455 while (1)
456     {
457     if (users->SearchNext(handle, &u))
458         {
459         break;
460         }
461
462     if (u->GetAuthorized())
463         {
464         uint32_t ip = u->GetCurrIP();
465
466         std::vector<IP_MASK>::iterator it;
467         for (it = subnets.begin(); it != subnets.end(); ++it)
468             {
469             if ((it->ip & it->mask) == (ip & it->mask))
470                 {
471                 ips.push_back(xmlrpc_c::value_string(inet_ntostring(u->GetCurrIP())));
472                 break;
473                 }
474             }
475         }
476     }
477
478 structVal["ips"] = xmlrpc_c::value_array(ips);
479
480 *retvalPtr = xmlrpc_c::value_struct(structVal);
481 }
482
483 bool METHOD_GET_ONLINE_IPS::ParseNet(const std::string & net, IP_MASK & ipm) const
484 {
485 size_t pos = net.find_first_of('/');
486
487 if (pos == std::string::npos)
488     {
489     printfd(__FILE__, "METHOD_GET_ONLINE_IPS::ParseNet(): Network address is not in CIDR-notation\n");
490     return true;
491     }
492
493 int res = inet_pton(AF_INET, net.substr(0, pos).c_str(), &ipm.ip);
494
495 if (res < 0)
496     {
497     printfd(__FILE__, "METHOD_GET_ONLINE_IPS::ParseNet(): '%s'\n", strerror(errno));
498     return true;
499     }
500 else if (res == 0)
501     {
502     printfd(__FILE__, "METHOD_GET_ONLINE_IPS::ParseNet(): Invalid network address\n", strerror(errno));
503     return true;
504     }
505
506 if (str2x(net.substr(pos + 1, net.length() - pos - 1), ipm.mask))
507     {
508     printfd(__FILE__, "METHOD_GET_ONLINE_IPS::ParseNet(): Invalid network mask\n");
509     return true;
510     }
511 if (ipm.mask > 32)
512     {
513     printfd(__FILE__, "METHOD_GET_ONLINE_IPS::ParseNet(): Network mask is out of range\n");
514     return true;
515     }
516 ipm.mask = htonl(0xffFFffFF << (32 - ipm.mask));
517
518 return false;
519 }