]> git.stg.codes - stg.git/blob - stargazer/plugins/configuration/rpcconfig/user_helper.cpp
Public interfaces: part 2
[stg.git] / stargazer / plugins / configuration / rpcconfig / user_helper.cpp
1 #include <cmath>
2
3 #include "stg/tariffs.h"
4 #include "stg/admin.h"
5 #include "stg/store.h"
6 #include "stg/user_ips.h"
7 #include "stg/common.h"
8 #include "stg/user_property.h"
9 #include "user_helper.h"
10
11 //------------------------------------------------------------------------------
12
13 void USER_HELPER::GetUserInfo(xmlrpc_c::value * info,
14                               bool hidePassword)
15 {
16 std::map<std::string, xmlrpc_c::value> structVal;
17
18 structVal["result"] = xmlrpc_c::value_boolean(true);
19 structVal["login"] = xmlrpc_c::value_string(ptr->GetLogin());
20
21 if (!hidePassword)
22     {
23     structVal["password"] = xmlrpc_c::value_string(ptr->GetProperty().password.Get());
24     }
25 else
26     {
27     structVal["password"] = xmlrpc_c::value_string("++++++++");
28     }
29
30 structVal["cash"] = xmlrpc_c::value_double(ptr->GetProperty().cash.Get());
31 structVal["freemb"] = xmlrpc_c::value_double(ptr->GetProperty().freeMb.Get());
32 structVal["credit"] = xmlrpc_c::value_double(ptr->GetProperty().credit.Get());
33
34 if (ptr->GetProperty().nextTariff.Get() != "")
35     {
36     structVal["tariff"] = xmlrpc_c::value_string(
37             ptr->GetProperty().tariffName.Get() +
38             "/" +
39             ptr->GetProperty().nextTariff.Get()
40             );
41     }
42 else
43     {
44     structVal["tariff"] = xmlrpc_c::value_string(ptr->GetProperty().tariffName.Get());
45     }
46
47 structVal["note"] = xmlrpc_c::value_string(IconvString(ptr->GetProperty().note, "KOI8-RU", "UTF-8"));
48
49 structVal["phone"] = xmlrpc_c::value_string(IconvString(ptr->GetProperty().phone, "KOI8-RU", "UTF-8"));
50
51 structVal["address"] = xmlrpc_c::value_string(IconvString(ptr->GetProperty().address, "KOI8-RU", "UTF-8"));
52
53 structVal["email"] = xmlrpc_c::value_string(IconvString(ptr->GetProperty().email, "KOI8-RU", "UTF-8"));
54
55 std::vector<xmlrpc_c::value> userdata;
56
57 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata0.Get(), "KOI8-RU", "UTF-8")));
58 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata1.Get(), "KOI8-RU", "UTF-8")));
59 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata2.Get(), "KOI8-RU", "UTF-8")));
60 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata3.Get(), "KOI8-RU", "UTF-8")));
61 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata4.Get(), "KOI8-RU", "UTF-8")));
62 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata5.Get(), "KOI8-RU", "UTF-8")));
63 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata6.Get(), "KOI8-RU", "UTF-8")));
64 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata7.Get(), "KOI8-RU", "UTF-8")));
65 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata8.Get(), "KOI8-RU", "UTF-8")));
66 userdata.push_back(xmlrpc_c::value_string(IconvString(ptr->GetProperty().userdata9.Get(), "KOI8-RU", "UTF-8")));
67
68 structVal["userdata"] = xmlrpc_c::value_array(userdata);
69
70 structVal["name"] = xmlrpc_c::value_string(IconvString(ptr->GetProperty().realName, "KOI8-RU", "UTF-8"));
71
72 structVal["group"] = xmlrpc_c::value_string(IconvString(ptr->GetProperty().group, "KOI8-RU", "UTF-8"));
73
74 structVal["status"] = xmlrpc_c::value_boolean(ptr->GetConnected());
75 structVal["aonline"] = xmlrpc_c::value_boolean(ptr->GetProperty().alwaysOnline.Get());
76 structVal["currip"] = xmlrpc_c::value_string(inet_ntostring(ptr->GetCurrIP()));
77 structVal["pingtime"] = xmlrpc_c::value_int(static_cast<int>(ptr->GetPingTime()));
78 structVal["ips"] = xmlrpc_c::value_string(ptr->GetProperty().ips.Get().GetIpStr());
79
80 std::map<std::string, xmlrpc_c::value> traffInfo;
81 std::vector<xmlrpc_c::value> mu(DIR_NUM);
82 std::vector<xmlrpc_c::value> md(DIR_NUM);
83 std::vector<xmlrpc_c::value> su(DIR_NUM);
84 std::vector<xmlrpc_c::value> sd(DIR_NUM);
85
86 DIR_TRAFF upload;
87 DIR_TRAFF download;
88 DIR_TRAFF supload;
89 DIR_TRAFF sdownload;
90 download = ptr->GetProperty().down.Get();
91 upload = ptr->GetProperty().up.Get();
92 sdownload = ptr->GetSessionUpload();
93 supload = ptr->GetSessionDownload();
94
95 for (int j = 0; j < DIR_NUM; j++)
96     {
97     mu[j] = xmlrpc_c::value_string(std::to_string(upload[j]));
98     md[j] = xmlrpc_c::value_string(std::to_string(download[j]));
99     su[j] = xmlrpc_c::value_string(std::to_string(supload[j]));
100     sd[j] = xmlrpc_c::value_string(std::to_string(sdownload[j]));
101     }
102
103 traffInfo["mu"] = xmlrpc_c::value_array(mu);
104 traffInfo["md"] = xmlrpc_c::value_array(md);
105 traffInfo["su"] = xmlrpc_c::value_array(su);
106 traffInfo["sd"] = xmlrpc_c::value_array(sd);
107
108 structVal["traff"] = xmlrpc_c::value_struct(traffInfo);
109
110 structVal["down"] = xmlrpc_c::value_boolean(ptr->GetProperty().disabled.Get());
111 structVal["disableddetailstat"] = xmlrpc_c::value_boolean(ptr->GetProperty().disabledDetailStat.Get());
112 structVal["passive"] = xmlrpc_c::value_boolean(ptr->GetProperty().passive.Get());
113 structVal["lastcash"] = xmlrpc_c::value_double(ptr->GetProperty().lastCashAdd.Get());
114 structVal["lasttimecash"] = xmlrpc_c::value_int(static_cast<int>(ptr->GetProperty().lastCashAddTime.Get()));
115 structVal["lastactivitytime"] = xmlrpc_c::value_int(static_cast<int>(ptr->GetProperty().lastActivityTime.Get()));
116 structVal["creditexpire"] = xmlrpc_c::value_int(static_cast<int>(ptr->GetProperty().creditExpire.Get()));
117
118 *info = xmlrpc_c::value_struct(structVal);
119 }
120
121 //------------------------------------------------------------------------------
122
123 bool USER_HELPER::SetUserInfo(const xmlrpc_c::value & info,
124                               const ADMIN * admin,
125                               const std::string & login,
126                               const STORE & store,
127                               TARIFFS * tariffs)
128 {
129 std::map<std::string, xmlrpc_c::value> structVal(
130     static_cast<std::map<std::string, xmlrpc_c::value> >(xmlrpc_c::value_struct(info))
131     );
132
133 std::map<std::string, xmlrpc_c::value>::iterator it;
134
135 bool check = false;
136 bool alwaysOnline = ptr->GetProperty().alwaysOnline;
137 if ((it = structVal.find("aonline")) != structVal.end())
138     {
139     check = true;
140     alwaysOnline = xmlrpc_c::value_boolean(it->second);
141     }
142 bool onlyOneIP = ptr->GetProperty().ips.ConstData().OnlyOneIP();
143 if ((it = structVal.find("ips")) != structVal.end())
144     {
145     check = true;
146     onlyOneIP = StrToIPS(xmlrpc_c::value_string(it->second)).OnlyOneIP();
147     }
148
149 if (check && alwaysOnline && !onlyOneIP)
150     {
151     printfd(__FILE__, "Requested change leads to a forbidden state: AlwaysOnline with multiple IP's\n");
152     return true;
153     }
154
155 if ((it = structVal.find("ips")) != structVal.end())
156     {
157     USER_IPS ips;
158     ips = StrToIPS(xmlrpc_c::value_string(it->second));
159
160     for (size_t i = 0; i < ips.Count(); ++i)
161         {
162         CONST_USER_PTR user;
163         uint32_t ip = ips[i].ip;
164         if (users.IsIPInUse(ip, login, &user))
165             {
166             printfd(__FILE__, "Trying to assign an IP %s to '%s' that is already in use by '%s'\n", inet_ntostring(ip).c_str(), login.c_str(), user->GetLogin().c_str());
167             return true;
168             }
169         }
170
171     if (!ptr->GetProperty().ips.Set(ips,
172                                 admin,
173                                 login,
174                                 &store))
175         return true;
176     }
177
178 if ((it = structVal.find("aonline")) != structVal.end())
179     {
180     bool value(xmlrpc_c::value_boolean(it->second));
181     if (ptr->GetProperty().alwaysOnline.Get() != value)
182         if (!ptr->GetProperty().alwaysOnline.Set(value,
183                                              admin,
184                                              login,
185                                              &store))
186             return true;
187     }
188
189 if ((it = structVal.find("password")) != structVal.end())
190     {
191     std::string value(xmlrpc_c::value_string(it->second));
192     if (ptr->GetProperty().password.Get() != value)
193         if (!ptr->GetProperty().password.Set(value,
194                                          admin,
195                                          login,
196                                          &store))
197             return true;
198     }
199
200 if ((it = structVal.find("address")) != structVal.end())
201     {
202     std::string value(IconvString(xmlrpc_c::value_string(it->second), "UTF-8", "KOI8-RU"));
203     if (ptr->GetProperty().address.Get() != value)
204         if (!ptr->GetProperty().address.Set(value,
205                                         admin,
206                                         login,
207                                         &store))
208             return true;
209     }
210
211 if ((it = structVal.find("phone")) != structVal.end())
212     {
213     std::string value(IconvString(xmlrpc_c::value_string(it->second), "UTF-8", "KOI8-RU"));
214     if (ptr->GetProperty().phone.Get() != value)
215         if (!ptr->GetProperty().phone.Set(value,
216                                       admin,
217                                       login,
218                                       &store))
219             return true;
220     }
221
222 if ((it = structVal.find("email")) != structVal.end())
223     {
224     std::string value(IconvString(xmlrpc_c::value_string(it->second), "UTF-8", "KOI8-RU"));
225     if (ptr->GetProperty().email.Get() != value)
226         if (!ptr->GetProperty().email.Set(value,
227                                       admin,
228                                       login,
229                                       &store))
230             return true;
231     }
232
233 if ((it = structVal.find("cash")) != structVal.end())
234     {
235     double value(xmlrpc_c::value_double(it->second));
236     if (std::fabs(ptr->GetProperty().cash.Get() - value) > 1.0e-3)
237         if (!ptr->GetProperty().cash.Set(value,
238                                      admin,
239                                      login,
240                                      &store))
241             return true;
242     }
243
244 if ((it = structVal.find("creditexpire")) != structVal.end())
245     {
246     time_t value(xmlrpc_c::value_int(it->second));
247     if (ptr->GetProperty().creditExpire.Get() != value)
248         if (!ptr->GetProperty().creditExpire.Set(value,
249                                              admin,
250                                              login,
251                                              &store))
252             return true;
253     }
254
255 if ((it = structVal.find("credit")) != structVal.end())
256     {
257     double value(xmlrpc_c::value_double(it->second));
258     if (std::fabs(ptr->GetProperty().credit.Get() - value) > 1.0e-3)
259         if (!ptr->GetProperty().credit.Set(value,
260                                        admin,
261                                        login,
262                                        &store))
263             return true;
264     }
265
266 if ((it = structVal.find("freemb")) != structVal.end())
267     {
268     double value(xmlrpc_c::value_double(it->second));
269     if (std::fabs(ptr->GetProperty().freeMb.Get() - value) > 1.0e-3)
270         if (!ptr->GetProperty().freeMb.Set(value,
271                                        admin,
272                                        login,
273                                        &store))
274             return true;
275     }
276
277 if ((it = structVal.find("down")) != structVal.end())
278     {
279     bool value(xmlrpc_c::value_boolean(it->second));
280     if (ptr->GetProperty().disabled.Get() != value)
281         if (!ptr->GetProperty().disabled.Set(value,
282                                          admin,
283                                          login,
284                                          &store))
285             return true;
286     }
287
288 if ((it = structVal.find("passive")) != structVal.end())
289     {
290     bool value(xmlrpc_c::value_boolean(it->second));
291     if (ptr->GetProperty().passive.Get() != value)
292         if (!ptr->GetProperty().passive.Set(value,
293                                         admin,
294                                         login,
295                                         &store))
296             return true;
297     }
298
299 if ((it = structVal.find("disableddetailstat")) != structVal.end())
300     {
301     bool value(xmlrpc_c::value_boolean(it->second));
302     if (ptr->GetProperty().disabledDetailStat.Get() != value)
303         if (!ptr->GetProperty().disabledDetailStat.Set(value,
304                                                    admin,
305                                                    login,
306                                                    &store))
307             return true;
308     }
309
310 if ((it = structVal.find("name")) != structVal.end())
311     {
312     std::string value(IconvString(xmlrpc_c::value_string(it->second), "UTF-8", "KOI8-RU"));
313     if (ptr->GetProperty().realName.Get() != value)
314         if (!ptr->GetProperty().realName.Set(value,
315                                          admin,
316                                          login,
317                                          &store))
318             return true;
319     }
320
321 if ((it = structVal.find("group")) != structVal.end())
322     {
323     std::string value(IconvString(xmlrpc_c::value_string(it->second), "UTF-8", "KOI8-RU"));
324     if (ptr->GetProperty().group.Get() != value)
325         if (!ptr->GetProperty().group.Set(value,
326                                       admin,
327                                       login,
328                                       &store))
329             return true;
330     }
331
332 if ((it = structVal.find("note")) != structVal.end())
333     {
334     std::string value(IconvString(xmlrpc_c::value_string(it->second), "UTF-8", "KOI8-RU"));
335     if (ptr->GetProperty().note.Get() != value)
336         if (!ptr->GetProperty().note.Set(value,
337                                      admin,
338                                      login,
339                                      &store))
340             return true;
341     }
342
343 if ((it = structVal.find("userdata")) != structVal.end())
344     {
345     std::vector<USER_PROPERTY_LOGGED<std::string> *> userdata;
346     userdata.push_back(ptr->GetProperty().userdata0.GetPointer());
347     userdata.push_back(ptr->GetProperty().userdata1.GetPointer());
348     userdata.push_back(ptr->GetProperty().userdata2.GetPointer());
349     userdata.push_back(ptr->GetProperty().userdata3.GetPointer());
350     userdata.push_back(ptr->GetProperty().userdata4.GetPointer());
351     userdata.push_back(ptr->GetProperty().userdata5.GetPointer());
352     userdata.push_back(ptr->GetProperty().userdata6.GetPointer());
353     userdata.push_back(ptr->GetProperty().userdata7.GetPointer());
354     userdata.push_back(ptr->GetProperty().userdata8.GetPointer());
355     userdata.push_back(ptr->GetProperty().userdata9.GetPointer());
356
357     std::vector<xmlrpc_c::value> udata(
358         xmlrpc_c::value_array(it->second).vectorValueValue()
359         );
360
361     for (unsigned i = 0; i < userdata.size(); ++i)
362         {
363         std::string value(IconvString(xmlrpc_c::value_string(udata[i]), "UTF-8", "KOI8-RU"));
364         if (userdata[i]->Get() != value)
365             if (!userdata[i]->Set(value,
366                                   admin,
367                                   login,
368                                   &store))
369                 return true;
370         }
371     }
372
373 if ((it = structVal.find("traff")) != structVal.end())
374     {
375     std::map<std::string, xmlrpc_c::value> traff(
376         static_cast<std::map<std::string, xmlrpc_c::value> >(xmlrpc_c::value_struct(it->second))
377         );
378
379     DIR_TRAFF dtData;
380     dtData = ptr->GetProperty().up.Get();
381     if ((it = traff.find("mu")) != traff.end())
382         {
383         std::vector<xmlrpc_c::value> data(xmlrpc_c::value_array(it->second).vectorValueValue());
384
385         for (int i = 0; i < std::min(DIR_NUM, static_cast<int>(data.size())); ++i)
386             {
387             int64_t value;
388             if (str2x(xmlrpc_c::value_string(data[i]), value))
389                 printfd(__FILE__, "USER_HELPER::SetUserInfo(): 'Invalid month upload value'\n");
390             else
391                 dtData[i] = value;
392             }
393         if (!ptr->GetProperty().up.Set(dtData,
394                                    admin,
395                                    login,
396                                    &store))
397             return true;
398         }
399     dtData = ptr->GetProperty().down.Get();
400     if ((it = traff.find("md")) != traff.end())
401         {
402         std::vector<xmlrpc_c::value> data(xmlrpc_c::value_array(it->second).vectorValueValue());
403
404         for (int i = 0; i < std::min(DIR_NUM, static_cast<int>(data.size())); ++i)
405             {
406             int64_t value;
407             if (str2x(xmlrpc_c::value_string(data[i]), value))
408                 printfd(__FILE__, "USER_HELPER::SetUserInfo(): 'Invalid month download value'\n");
409             else
410                 dtData[i] = value;
411             }
412         if (!ptr->GetProperty().down.Set(dtData,
413                                      admin,
414                                      login,
415                                      &store))
416             return true;
417         }
418     }
419
420 if ((it = structVal.find("tariff")) != structVal.end())
421     {
422     std::string tariff(xmlrpc_c::value_string(it->second));
423     size_t pos = tariff.find('/');
424     std::string nextTariff;
425     if (pos != std::string::npos)
426         {
427         nextTariff = tariff.substr(pos + 1);
428         tariff = tariff.substr(0, pos);
429         }
430
431     const TARIFF * newTariff = tariffs->FindByName(tariff);
432     if (newTariff)
433         {
434         const TARIFF * currentTariff = ptr->GetTariff();
435         std::string message = currentTariff->TariffChangeIsAllowed(*newTariff, stgTime);
436         if (message.empty())
437             {
438             if (ptr->GetProperty().tariffName.Get() != tariff)
439                 {
440                 if (!ptr->GetProperty().tariffName.Set(tariff,
441                                                    admin,
442                                                    login,
443                                                    &store))
444                     return true;
445                 }
446             }
447         else
448             {
449                 PluginLogger::get("conf_rpc")("Tariff change is prohibited for user %s. %s", ptr->GetLogin().c_str(), message.c_str());
450             }
451         }
452
453     if (nextTariff != "" &&
454         tariffs->FindByName(nextTariff))
455         if (ptr->GetProperty().nextTariff.Get() != nextTariff)
456             if (!ptr->GetProperty().nextTariff.Set(tariff,
457                                                admin,
458                                                login,
459                                                &store))
460                 return true;
461     }
462
463 return false;
464 }