]> git.stg.codes - stg.git/blob - projects/sgconf/common_sg.cpp
0f5c9cc9ba23566c012c3e2a6ce853d021617cb9
[stg.git] / projects / sgconf / common_sg.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  /*
22  $Author: faust $
23  $Revision: 1.12 $
24  $Date: 2009/06/08 10:02:28 $
25  */
26
27
28 #include <stdio.h>
29 #include <string.h>
30 #include <stdlib.h>
31 #include <errno.h>
32 #include <locale.h>
33 #include <langinfo.h>
34 #include <iostream>
35 #include <iconv.h>
36
37 #include "stg/common.h"
38 #include "sg_error_codes.h"
39 #include "common_sg.h"
40 #include "version_sg.h"
41
42 using namespace std;
43
44 const int usageConf = 0;
45 const int usageInfo = 1;
46
47 const int TO_KOI8 = 0;
48 const int FROM_KOI8 = 1;
49 //-----------------------------------------------------------------------------
50 struct GetUserData
51 {
52     REQUEST & request;
53     bool result;
54 };
55 //---------------------------------------------------------------------------
56 struct HelpParams
57 {
58     string setActionName;
59     string getActionName;
60     string valueName;
61     string valueParam;
62 };
63 //---------------------------------------------------------------------------
64 void Usage(int usageType)
65 {
66 printf("Sgconf version: %s\n\n", VERSION_SG);
67
68 char action[4];
69 if (usageType == usageConf)
70     strcpy(action, "set");
71 else
72     strcpy(action, "get");
73
74 printf("To add or to set cash use:\n");
75 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> -c <add_cash[:log message]>\n");
76 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> -v <set_cash[:log message]>\n");
77 printf("To get cash use:\n");
78 printf("sgconf get -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> -c\n\n");
79
80 HelpParams hp[] =
81 {
82     {"set tariff",              "get tariff",           "-t",   "<tariff:now|delayed>"},
83     {"set credit",              "get credit",           "-r",   "<credit>"},
84     {"set credit expire",       "get credit expire",    "-E",   "<credit_expire_date>"},
85     {"set password",            "get password",         "-o",   "<new_password>"},
86     {"set prepaid traffic",     "get prepaid traffic",  "-e",   "<prepaid>"},
87     {"set IP-addresses",        "get IP-addresses",     "-I",   "<*|ip_addr[,ip_addr...]>"},
88     {"set name",                "get name",             "-A",   "<name>"},
89     {"set note",                "get note",             "-N",   "<note>"},
90     {"set street address",      "get street address",   "-D",   "<address>"},
91     {"set email",               "get email",            "-L",   "<email>"},
92     {"set phone",               "get phone",            "-P",   "<phone>"},
93     {"set group",               "get group",            "-G",   "<group>"},
94     {"set/unset down",          "get down",             "-d",   "<0/1>"},
95     {"set/unset \'passive\'",   "get \'passive\'",      "-i",   "<0/1>"},
96     {"set/unset \'disableDetailStat\'",   "get \'disableDetailStat\'",      "--disable-stat",   "<0/1>"},
97     {"set/unset \'alwaysOnline\'",   "get \'alwaysOnline\'",      "--always-online",   "<0/1>"},
98 };
99
100 for (unsigned i = 0; i < sizeof(hp) / sizeof(HelpParams); i++)
101     {
102     printf("To %s use:\n", hp[i].setActionName.c_str());
103     printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> %s %s\n",
104            hp[i].valueName.c_str(), hp[i].valueParam.c_str());
105     printf("To %s use:\n", hp[i].getActionName.c_str());
106     printf("sgconf get -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> %s\n\n",
107            hp[i].valueName.c_str());
108     }
109
110 printf("To set user\'s upload traffic value use:\n");
111 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --u0 <traff> [--u1<traff> ...]\n");
112 printf("To get user\'s upload traffic value use:\n");
113 printf("sgconf get -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --u0 [--u1 ...]\n\n");
114
115 printf("To set user\'s download traffic value use:\n");
116 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --d0 <traff> [--d1<traff> ...]\n");
117 printf("To get user\'s download traffic value use:\n");
118 printf("sgconf get -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --d0 [--d1 ...]\n\n");
119
120 printf("To set userdata<0...9> use:\n");
121 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --ud0 <userdata> [--ud1<userdata> ...]\n");
122 printf("To get userdata<0...9> use:\n");
123 printf("sgconf get -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --ud0 [--ud1 ...]\n\n");
124
125 printf("To get user's authorizers list use:\n");
126 printf("sgconf get -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --authorized-by\n\n");
127
128 printf("To send message use:\n");
129 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> -m <message>\n\n");
130
131 printf("To create user use:\n");
132 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> -n\n\n");
133
134 printf("To delete user use:\n");
135 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> -l\n\n");
136 }
137 //---------------------------------------------------------------------------
138 void UsageConf()
139 {
140 Usage(usageConf);
141 }
142 //---------------------------------------------------------------------------
143 void UsageInfo()
144 {
145 Usage(usageInfo);
146 }
147 //---------------------------------------------------------------------------
148 int CheckLogin(const char * login)
149 {
150 for (int i = 0; i < (int)strlen(login); i++)
151     {
152     if (!(( login[i] >= 'a' && login[i] <= 'z')
153         || (login[i] >= 'A' && login[i] <= 'Z')
154         || (login[i] >= '0' && login[i] <= '9')
155         ||  login[i] == '.'
156         ||  login[i] == '_'
157         ||  login[i] == '-'))
158         {
159         return 1;
160         }
161     }
162 return 0;
163 }
164 //-----------------------------------------------------------------------------
165 short int ParseServerPort(const char * p)
166 {
167 int port;
168 if (str2x(p, port) != 0)
169     {
170     printf("Incorresct server port %s\n", p);
171     exit(NETWORK_ERR_CODE);
172     }
173 return (short)port;
174 }
175 //-----------------------------------------------------------------------------
176 char * ParseAdminLogin(char * adm)
177 {
178 if (CheckLogin(adm))
179     {
180     printf("Incorresct admin login %s\n", adm);
181     exit(PARAMETER_PARSING_ERR_CODE);
182     }
183 return adm;
184 }
185 //-----------------------------------------------------------------------------
186 char * ParsePassword(char * pass)
187 {
188 if (strlen(pass) >= ADM_PASSWD_LEN)
189     {
190     printf("Password too big %s\n", pass);
191     exit(PARAMETER_PARSING_ERR_CODE);
192     }
193
194 return pass;
195 }
196 //-----------------------------------------------------------------------------
197 char * ParseUser(char * usr)
198 {
199 if (CheckLogin(usr))
200     {
201     printf("Incorresct user login %s\n", usr);
202     exit(PARAMETER_PARSING_ERR_CODE);
203     }
204 return usr;
205 }
206 //-----------------------------------------------------------------------------
207 void ConvertKOI8(const string & src, string * dst, int encType)
208 {
209 iconv_t cd;
210 char * ob = new char[src.size() * 2 + 1];
211 char * ib = new char[src.size() + 1];
212
213 strcpy(ib, src.c_str());
214
215 char * outbuf = ob;
216 char * inbuf = ib;
217
218 setlocale(LC_ALL, "");
219
220 char charsetF[100];
221 char charsetT[100];
222
223 if (encType == TO_KOI8)
224     {
225     strcpy(charsetF, nl_langinfo(CODESET));
226     strcpy(charsetT, "koi8-ru");
227     }
228 else
229     {
230     strcpy(charsetT, nl_langinfo(CODESET));
231     strcpy(charsetF, "koi8-ru");
232     }
233
234 size_t nconv = 1;
235
236 size_t insize = strlen(ib);
237 size_t outsize = insize * 2 + 1;
238
239 insize = src.size();
240
241 cd = iconv_open(charsetT, charsetF);
242 if (cd == (iconv_t) -1)
243     {
244     if (errno != EINVAL)
245         printf("error iconv_open\n");
246     else
247         {
248         printf("Warning: iconv from %s to %s failed\n", charsetF, charsetT);
249         *dst = src;
250         return;
251         }
252
253     exit(ICONV_ERR_CODE);
254     }
255
256 #if defined(FREE_BSD) || defined(FREE_BSD5)
257 nconv = iconv(cd, (const char **)&inbuf, &insize, &outbuf, &outsize);
258 #else
259 nconv = iconv(cd, &inbuf, &insize, &outbuf, &outsize);
260 #endif
261 //printf("charsetT=%s charsetF=%s\n", charsetT, charsetF);
262 //printf("ib=%s ob=%s\n", ib, ob);
263 //printf("nconv=%d outsize=%d\n", nconv, outsize);
264 if (nconv == (size_t) -1)
265     {
266     if (errno != EINVAL)
267         {
268         printf("iconv error\n");
269         exit(ICONV_ERR_CODE);
270         }
271     }
272
273 *outbuf = L'\0';
274
275 iconv_close(cd);
276 *dst = ob;
277
278 delete[] ob;
279 delete[] ib;
280 }
281 //-----------------------------------------------------------------------------
282 void ConvertFromKOI8(const string & src, string * dst)
283 {
284 ConvertKOI8(src, dst, FROM_KOI8);
285 }
286 //-----------------------------------------------------------------------------
287 void ConvertToKOI8(const string & src, string * dst)
288 {
289 ConvertKOI8(src, dst, TO_KOI8);
290 }
291 //-----------------------------------------------------------------------------
292 int RecvSetUserAnswer(const char * ans, void * d)
293 {
294 GetUserCbData * gucbd;
295 gucbd = (GetUserCbData *)d;
296
297 bool * result = gucbd->result;
298
299 //REQUEST * req = (REQUEST *)gucbd->data;
300
301 //printf("ans=%s\n", ans);
302 if (strcasecmp("Ok", ans) == 0)
303     *result = true;
304 else
305     *result = false;
306
307 return 0;
308 }
309 //-----------------------------------------------------------------------------
310 struct StringReqParams
311 {
312     string name;
313     RESETABLE<string> reqParam;
314     string * value;
315 };
316 //-----------------------------------------------------------------------------
317 void GetUserCallback(const PARSER_GET_USER::INFO & info, void * d)
318 {
319 GetUserData * data = static_cast<GetUserData *>(d);
320
321 REQUEST * req = (REQUEST *)gucbd->data;
322
323 if (info.login == "")
324     {
325     data->result = false;
326     return;
327     }
328
329 if (!data->request.cash.res_empty())
330     cout << "cash=" << info.cash << endl;
331
332 if (!data->requst.credit.res_empty())
333     cout << "credit=" << info.credit << endl;
334
335 if (!data->requst.creditExpire.res_empty())
336     {
337     char buf[32];
338     struct tm brokenTime;
339     time_t tt = info.creditExpire;
340
341     brokenTime.tm_wday = 0;
342     brokenTime.tm_yday = 0;
343     brokenTime.tm_isdst = 0;
344     brokenTime.tm_hour = 0;
345     brokenTime.tm_min = 0;
346     brokenTime.tm_sec = 0;
347
348     gmtime_r(&tt, &brokenTime);
349
350     strftime(buf, 32, "%Y-%m-%d", &brokenTime);
351
352     cout << "creditExpire=" << buf << endl;
353     }
354
355 if (!data->requst.down.res_empty())
356     cout << "down=" << info.down << endl;
357
358 if (!data->requst.passive.res_empty())
359     cout << "passive=" << info.passive << endl;
360
361 if (!data->requst.disableDetailStat.res_empty())
362     cout << "disableDetailStat=" << info.disableDetailStat << endl;
363
364 if (!data->requst.alwaysOnline.res_empty())
365     cout << "alwaysOnline=" << info.alwaysOnline << endl;
366
367 if (!data->requst.prepaidTraff.res_empty())
368     cout << "prepaidTraff=" << info.prepaidTraff << endl;
369
370 for (int i = 0; i < DIR_NUM; i++)
371     {
372     if (!data->requst.sessionUpload[i].res_empty())
373         cout << "session upload for dir" << i << "=" << info.stat.su[i] << endl;
374     if (!data->requst.sessionDownload[i].res_empty())
375         cout << "session download for dir" << i << "=" << info.stat.sd[i] << endl;
376     }
377
378 for (int i = 0; i < DIR_NUM; i++)
379     {
380     if (!data->requst.monthUpload[i].res_empty())
381         cout << "month upload for dir" << i << "=" << info.stat.mu[i] << endl;
382     if (!data->requst.monthDownload[i].res_empty())
383         cout << "month download for dir" << i << "=" << info.stat.md[i] << endl;
384     }
385
386 for (int i = 0; i < USERDATA_NUM; i++)
387     {
388     if (!data->requst.ud[i].res_empty())
389         {
390         string str;
391         ConvertFromKOI8(info.userData[i], &str);
392         cout << "userdata" << i << "=" << str << endl;
393         }
394     }
395
396 StringReqParams strReqParams[] =
397 {
398     {"note",     data->requst.note,        &info.note},
399     {"name",     data->requst.name,        &info.name},
400     {"address",  data->requst.address,     &info.address},
401     {"email",    data->requst.email,       &info.email},
402     {"phone",    data->requst.phone,       &info.phone},
403     {"group",    data->requst.group,       &info.group},
404     {"tariff",   data->requst.tariff,      &info.tariff},
405     {"password", data->requst.usrPasswd,   &info.password},
406     {"ip",       data->requst.ips,         &info.ips} // IP-address of user
407 };
408 for (unsigned i = 0; i < sizeof(strReqParams) / sizeof(StringReqParams); i++)
409     {
410     if (!strReqParams[i].reqParam.res_empty())
411         {
412         string str;
413         ConvertFromKOI8(*strReqParams[i].value, &str);
414         cout << strReqParams[i].name << "=" << str << endl;
415         }
416     }
417 data->result = true;
418 }
419 //-----------------------------------------------------------------------------
420 void RecvAuthByData(const PARSER_AUTH_BY::INFO & list, void *)
421 {
422 for (std::vector<std::string>::const_iterator it = list.begin(); it != list.end(); ++it)
423     cout << *it << "\n";
424 cout << endl;
425 }
426 //-----------------------------------------------------------------------------
427 int ProcessSetUser(const std::string &server,
428                    int port,
429                    const std::string &admLogin,
430                    const std::string &admPasswd,
431                    const std::string &str,
432                    void * data,
433                    bool isMessage)
434 {
435 SERVCONF sc;
436
437 bool result = false;
438
439
440 sc.SetServer(server.c_str());
441 sc.SetPort(port);
442 sc.SetAdmLogin(admLogin.c_str());
443 sc.SetAdmPassword(admPasswd.c_str());
444
445 // TODO Good variable name :)
446 GetUserCbData gucbd;
447
448 gucbd.data = data;
449 gucbd.result = &result;
450
451 if (isMessage)
452     {
453     sc.SetSendMessageCb(RecvSetUserAnswer, &gucbd);
454     sc.MsgUser(str.c_str());
455     }
456 else
457     {
458     sc.SetChgUserCb(RecvSetUserAnswer, &gucbd);
459     sc.ChgUser(str.c_str());
460     }
461
462 if (result)
463     {
464     printf("Ok\n");
465     return 0;
466     }
467 else
468     {
469     printf("Error\n");
470     return -1;
471     }
472
473 return 0;
474 }
475 //-----------------------------------------------------------------------------
476 int ProcessGetUser(const std::string &server,
477                    int port,
478                    const std::string &admLogin,
479                    const std::string &admPasswd,
480                    const std::string &login,
481                    REQUEST & request)
482 {
483 SERVCONF sc;
484
485 bool result = false;
486
487 sc.SetServer(server.c_str());
488 sc.SetPort(port);
489 sc.SetAdmLogin(admLogin.c_str());
490 sc.SetAdmPassword(admPasswd.c_str());
491
492 // TODO Good variable name :)
493 GetUserCbData gucbd;
494
495 gucbd.data = &request;
496 gucbd.result = &result;
497
498 sc.SetGetUserCallback(GetUserCallback, &gucbd);
499 sc.GetUser(login.c_str());
500
501 if (result)
502     {
503     printf("Ok\n");
504     return 0;
505     }
506 else
507     {
508     printf("Error\n");
509     return -1;
510     }
511
512 return 0;
513 }
514 //-----------------------------------------------------------------------------
515 int ProcessAuthBy(const std::string &server,
516                   int port,
517                   const std::string &admLogin,
518                   const std::string &admPasswd,
519                   const std::string &login,
520                   REQUEST & request)
521 {
522 SERVCONF sc;
523
524 sc.SetServer(server.c_str());
525 sc.SetPort(port);
526 sc.SetAdmLogin(admLogin.c_str());
527 sc.SetAdmPassword(admPasswd.c_str());
528
529 sc.SetAuthByCallback(RecvAuthByData, NULL);
530 sc.AuthBy(login.c_str());
531
532 if (sc.GetError())
533     {
534     printf("Error\n");
535     return -1;
536     }
537
538 printf("Ok\n");
539 return 0;
540 }
541 //-----------------------------------------------------------------------------