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