]> git.stg.codes - stg.git/blob - projects/sgconf/main.cpp
[NY] Code deduplication.
[stg.git] / projects / sgconf / main.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.25 $
24  $Date: 2010/03/25 14:37:43 $
25  */
26
27 #include <unistd.h>
28 #include <getopt.h>
29 #include <iconv.h>
30 #include <langinfo.h>
31
32 #include <cerrno>
33 #include <clocale>
34 #include <cstdio>
35 #include <cstring>
36 #include <string>
37 #include <list>
38 #include <sstream>
39
40 #include "stg/common.h"
41 #include "stg/netunit.h"
42 #include "request.h"
43 #include "common_sg.h"
44 #include "sg_error_codes.h"
45
46 namespace
47 {
48
49 template <typename T>
50 struct ARRAY_TYPE;
51
52 template <typename T>
53 struct ARRAY_TYPE<T[]>
54 {
55 typedef T type;
56 };
57
58 template <typename T, size_t N>
59 struct ARRAY_TYPE<T[N]>
60 {
61 typedef T type;
62 };
63
64 template <typename T>
65 bool SetArrayItem(T & array, const char * index, const typename ARRAY_TYPE<T>::type & value)
66 {
67 size_t pos = 0;
68 if (str2x(index, pos))
69     return false;
70 array[pos] = value;
71 return true;
72 }
73
74 } // namespace anonymous
75
76 time_t stgTime;
77
78 int ParseReplyGet(void * data, list<string> * ans);
79 //int ParseReplySet(void * data, list<string> * ans);
80
81 struct option long_options_get[] = {
82 {"server",      1, 0, 's'},  //Server
83 {"port",        1, 0, 'p'},  //Port
84 {"admin",       1, 0, 'a'},  //Admin
85 {"admin_pass",  1, 0, 'w'},  //passWord
86 {"user",        1, 0, 'u'},  //User
87 {"addcash",     0, 0, 'c'},  //Add Cash
88 //{"setcash",     0, 0, 'v'},  //Set Cash
89 {"credit",      0, 0, 'r'},  //cRedit
90 {"tariff",      0, 0, 't'},  //Tariff
91 {"message",     0, 0, 'm'},  //message
92 {"password",    0, 0, 'o'},  //password
93 {"down",        0, 0, 'd'},  //down
94 {"passive",     0, 0, 'i'},  //passive
95 {"disable-stat",0, 0, 'S'},  //disable detail stat
96 {"always-online",0, 0, 'O'}, //always online
97 {"session-upload",   1, 0, 500},  //SU0
98 {"session-download", 1, 0, 501},  //SD0
99 {"month-upload",     1, 0, 502},  //MU0
100 {"month-download",   1, 0, 503},  //MD0
101
102 {"user-data",   1, 0, 700},  //UserData0
103
104 {"prepaid",     0, 0, 'e'},  //prepaid traff
105 {"create",      0, 0, 'n'},  //create
106 {"delete",      0, 0, 'l'},  //delete
107
108 {"note",        0, 0, 'N'},  //Note
109 {"name",        0, 0, 'A'},  //nAme
110 {"address",     0, 0, 'D'},  //aDdress
111 {"email",       0, 0, 'L'},  //emaiL
112 {"phone",       0, 0, 'P'},  //phone
113 {"group",       0, 0, 'G'},  //Group
114 {"ip",          0, 0, 'I'},  //IP-address of user
115 {"authorized-by",0, 0, 800}, //always online
116
117 {0, 0, 0, 0}};
118
119 struct option long_options_set[] = {
120 {"server",      1, 0, 's'},  //Server
121 {"port",        1, 0, 'p'},  //Port
122 {"admin",       1, 0, 'a'},  //Admin
123 {"admin_pass",  1, 0, 'w'},  //passWord
124 {"user",        1, 0, 'u'},  //User
125 {"addcash",     1, 0, 'c'},  //Add Cash
126 {"setcash",     1, 0, 'v'},  //Set Cash
127 {"credit",      1, 0, 'r'},  //cRedit
128 {"tariff",      1, 0, 't'},  //Tariff
129 {"message",     1, 0, 'm'},  //message
130 {"password",    1, 0, 'o'},  //password
131 {"down",        1, 0, 'd'},  //down
132 {"passive",     1, 0, 'i'},  //passive
133 {"disable-stat",1, 0, 'S'},  //disable detail stat
134 {"always-online",1, 0, 'O'},  //always online
135 {"session-upload",   1, 0, 500},  //U0
136 {"session-download", 1, 0, 501},  //U1
137 {"month-upload",     1, 0, 502},  //U2
138 {"month-download",   1, 0, 503},  //U3
139
140 {"user-data",        1, 0, 700},  //UserData
141
142 {"prepaid",     1, 0, 'e'},  //prepaid traff
143 {"create",      1, 0, 'n'},  //create
144 {"delete",      1, 0, 'l'},  //delete
145
146 {"note",        1, 0, 'N'},  //Note
147 {"name",        1, 0, 'A'},  //nAme
148 {"address",     1, 0, 'D'},  //aDdress
149 {"email",       1, 0, 'L'},  //emaiL
150 {"phone",       1, 0, 'P'},  //phone
151 {"group",       1, 0, 'G'},  //Group
152 {"ip",          0, 0, 'I'},  //IP-address of user
153
154 {0, 0, 0, 0}};
155
156 //-----------------------------------------------------------------------------
157 double ParseCash(const char * c, string * message)
158 {
159 //-c 123.45:log message
160 double cash;
161 char * msg;
162 char * str;
163 str = new char[strlen(c) + 1];
164
165 strncpy(str, c, strlen(c));
166 str[strlen(c)] = 0;
167
168 msg = strchr(str, ':');
169
170 if (msg)
171     {
172     *message =  msg + 1;
173     str[msg - str] = 0;
174     }
175 else
176     *message = "";
177
178 if (strtodouble2(str, cash) != 0)
179     {
180     printf("Incorrect cash value %s\n", c);
181     exit(PARAMETER_PARSING_ERR_CODE);
182     }
183
184 delete[] str;
185 return cash;
186 }
187 //-----------------------------------------------------------------------------
188 double ParseCredit(const char * c)
189 {
190 double credit;
191 if (strtodouble2(c, credit) != 0)
192     {
193     printf("Incorrect credit value %s\n", c);
194     exit(PARAMETER_PARSING_ERR_CODE);
195     }
196
197 return credit;
198 }
199 //-----------------------------------------------------------------------------
200 double ParsePrepaidTraffic(const char * c)
201 {
202 double credit;
203 if (strtodouble2(c, credit) != 0)
204     {
205     printf("Incorrect prepaid traffic value %s\n", c);
206     exit(PARAMETER_PARSING_ERR_CODE);
207     }
208
209 return credit;
210 }
211 //-----------------------------------------------------------------------------
212 int64_t ParseTraff(const char * c)
213 {
214 int64_t traff;
215 if (str2x(c, traff) != 0)
216     {
217     printf("Incorrect credit value %s\n", c);
218     exit(PARAMETER_PARSING_ERR_CODE);
219     }
220
221 return traff;
222 }
223 //-----------------------------------------------------------------------------
224 bool ParseDownPassive(const char * dp)
225 {
226 if (!(dp[1] == 0 && (dp[0] == '1' || dp[0] == '0')))
227     {
228     printf("Incorrect value %s\n", dp);
229     exit(PARAMETER_PARSING_ERR_CODE);
230     }
231
232 return dp[0] - '0';
233 }
234 //-----------------------------------------------------------------------------
235 string ParseTariff(const char * t, int &chgType)
236 {
237 int l = strlen(t);
238 char * s;
239 s = new char[l];
240 char * s1, * s2;
241 string ss;
242
243 strcpy(s, t);
244
245 s1 = strtok(s, ":");
246
247 if (strlen(s1) >= TARIFF_NAME_LEN)
248     {
249     printf("Tariff name too big %s\n", s1);
250     exit(PARAMETER_PARSING_ERR_CODE);
251     }
252
253 //*tariff = s;
254
255 if (CheckLogin(s1))
256     {
257     printf("Incorrect tariff value %s\n", t);
258     exit(PARAMETER_PARSING_ERR_CODE);
259     }
260
261 s2 = strtok(NULL, ":");
262
263 chgType = -1;
264
265 if (s2 == NULL)
266     {
267     chgType = TARIFF_NOW;
268     ss = s;
269     delete[] s;
270     return ss;
271     }
272
273
274 if (strcmp(s2, "now") == 0)
275     chgType = TARIFF_NOW;
276
277 if (strcmp(s2, "delayed") == 0)
278     chgType = TARIFF_DEL;
279
280 if (strcmp(s2, "recalc") == 0)
281     chgType = TARIFF_REC;
282
283 if (chgType < 0)
284     {
285     printf("Incorrect tariff value %s\n", t);
286     exit(PARAMETER_PARSING_ERR_CODE);
287     }
288
289 ss = s;
290 delete[] s;
291 return ss;
292 }
293 //-----------------------------------------------------------------------------
294 time_t ParseCreditExpire(const char * str)
295 {
296 struct tm brokenTime;
297
298 brokenTime.tm_wday = 0;
299 brokenTime.tm_yday = 0;
300 brokenTime.tm_isdst = 0;
301 brokenTime.tm_hour = 0;
302 brokenTime.tm_min = 0;
303 brokenTime.tm_sec = 0;
304
305 stg_strptime(str, "%Y-%m-%d", &brokenTime);
306
307 return stg_timegm(&brokenTime);
308 }
309 //-----------------------------------------------------------------------------
310 void ParseAnyString(const char * c, string * msg, const char * enc)
311 {
312 iconv_t cd;
313 char * ob = new char[strlen(c) + 1];
314 char * ib = new char[strlen(c) + 1];
315
316 strcpy(ib, c);
317
318 char * outbuf = ob;
319 char * inbuf = ib;
320
321 setlocale(LC_ALL, "");
322
323 char charsetF[255];
324 strncpy(charsetF, nl_langinfo(CODESET), 255);
325
326 const char * charsetT = enc;
327
328 size_t nconv = 1;
329
330 size_t insize = strlen(ib);
331 size_t outsize = strlen(ib);
332
333 insize = strlen(c);
334
335 cd = iconv_open(charsetT, charsetF);
336 if (cd == (iconv_t) -1)
337     {
338     if (errno == EINVAL)
339         {
340         printf("Warning: iconv from %s to %s failed\n", charsetF, charsetT);
341         *msg = c;
342         return;
343         }
344     else
345         printf("error iconv_open\n");
346
347     exit(ICONV_ERR_CODE);
348     }
349
350 #if defined(FREE_BSD) || defined(FREE_BSD5)
351 nconv = iconv (cd, (const char**)&inbuf, &insize, &outbuf, &outsize);
352 #else
353 nconv = iconv (cd, &inbuf, &insize, &outbuf, &outsize);
354 #endif
355 //printf("nconv=%d outsize=%d\n", nconv, outsize);
356 if (nconv == (size_t) -1)
357     {
358     if (errno != EINVAL)
359         {
360         printf("iconv error\n");
361         exit(ICONV_ERR_CODE);
362         }
363     }
364
365 *outbuf = L'\0';
366
367 iconv_close(cd);
368 *msg = ob;
369
370 delete[] ob;
371 delete[] ib;
372 }
373 //-----------------------------------------------------------------------------
374 void CreateRequestSet(REQUEST * req, char * r)
375 {
376 const int strLen = 10024;
377 char str[strLen];
378 memset(str, 0, strLen);
379
380 r[0] = 0;
381
382 if (!req->usrMsg.res_empty())
383     {
384     string msg;
385     Encode12str(msg, req->usrMsg);
386     sprintf(str, "<Message login=\"%s\" msgver=\"1\" msgtype=\"1\" repeat=\"0\" repeatperiod=\"0\" showtime=\"0\" text=\"%s\"/>", req->login.const_data().c_str(), msg.c_str());
387     //sprintf(str, "<message login=\"%s\" priority=\"0\" text=\"%s\"/>\n", req->login, msg);
388     strcat(r, str);
389     return;
390     }
391
392 if (req->deleteUser)
393     {
394     sprintf(str, "<DelUser login=\"%s\"/>", req->login.const_data().c_str());
395     strcat(r, str);
396     //printf("%s\n", r);
397     return;
398     }
399
400 if (req->createUser)
401     {
402     sprintf(str, "<AddUser> <login value=\"%s\"/> </AddUser>", req->login.const_data().c_str());
403     strcat(r, str);
404     //printf("%s\n", r);
405     return;
406     }
407
408 strcat(r, "<SetUser>\n");
409 sprintf(str, "<login value=\"%s\"/>\n", req->login.const_data().c_str());
410 strcat(r, str);
411 if (!req->credit.res_empty())
412     {
413     sprintf(str, "<credit value=\"%f\"/>\n", req->credit.const_data());
414     strcat(r, str);
415     }
416
417 if (!req->creditExpire.res_empty())
418     {
419     sprintf(str, "<creditExpire value=\"%ld\"/>\n", req->creditExpire.const_data());
420     strcat(r, str);
421     }
422
423 if (!req->prepaidTraff.res_empty())
424     {
425     sprintf(str, "<FreeMb value=\"%f\"/>\n", req->prepaidTraff.const_data());
426     strcat(r, str);
427     }
428
429 if (!req->cash.res_empty())
430     {
431     string msg;
432     Encode12str(msg, req->message);
433     sprintf(str, "<cash add=\"%f\" msg=\"%s\"/>\n", req->cash.const_data(), msg.c_str());
434     strcat(r, str);
435     }
436
437 if (!req->setCash.res_empty())
438     {
439     string msg;
440     Encode12str(msg, req->message);
441     sprintf(str, "<cash set=\"%f\" msg=\"%s\"/>\n", req->setCash.const_data(), msg.c_str());
442     strcat(r, str);
443     }
444
445 if (!req->usrPasswd.res_empty())
446     {
447     sprintf(str, "<password value=\"%s\" />\n", req->usrPasswd.const_data().c_str());
448     strcat(r, str);
449     }
450
451 if (!req->down.res_empty())
452     {
453     sprintf(str, "<down value=\"%d\" />\n", req->down.const_data());
454     strcat(r, str);
455     }
456
457 if (!req->passive.res_empty())
458     {
459     sprintf(str, "<passive value=\"%d\" />\n", req->passive.const_data());
460     strcat(r, str);
461     }
462
463 if (!req->disableDetailStat.res_empty())
464     {
465     sprintf(str, "<disableDetailStat value=\"%d\" />\n", req->disableDetailStat.const_data());
466     strcat(r, str);
467     }
468
469 if (!req->alwaysOnline.res_empty())
470     {
471     sprintf(str, "<aonline value=\"%d\" />\n", req->alwaysOnline.const_data());
472     strcat(r, str);
473     }
474
475 // IP-address of user
476 if (!req->ips.res_empty())
477     {
478     sprintf(str, "<ip value=\"%s\" />\n", req->ips.const_data().c_str());
479     strcat(r, str);
480     }
481
482 int uPresent = false;
483 int dPresent = false;
484 for (int i = 0; i < DIR_NUM; i++)
485     {
486     if (!req->monthUpload[i].res_empty())
487         {
488         if (!uPresent && !dPresent)
489             {
490             sprintf(str, "<traff ");
491             strcat(r, str);
492             uPresent = true;
493             }
494
495         stringstream ss;
496         ss << req->monthUpload[i].const_data();
497         //sprintf(str, "MU%d=\"%lld\" ", i, req->u[i].const_data());
498         sprintf(str, "MU%d=\"%s\" ", i, ss.str().c_str());
499         strcat(r, str);
500         }
501     if (!req->monthDownload[i].res_empty())
502         {
503         if (!uPresent && !dPresent)
504             {
505             sprintf(str, "<traff ");
506             strcat(r, str);
507             dPresent = true;
508             }
509
510         stringstream ss;
511         ss << req->monthDownload[i].const_data();
512         sprintf(str, "MD%d=\"%s\" ", i, ss.str().c_str());
513         strcat(r, str);
514         }
515     if (!req->sessionUpload[i].res_empty())
516         {
517         if (!uPresent && !dPresent)
518             {
519             sprintf(str, "<traff ");
520             strcat(r, str);
521             uPresent = true;
522             }
523
524         stringstream ss;
525         ss << req->sessionUpload[i].const_data();
526         //sprintf(str, "MU%d=\"%lld\" ", i, req->u[i].const_data());
527         sprintf(str, "MU%d=\"%s\" ", i, ss.str().c_str());
528         strcat(r, str);
529         }
530     if (!req->sessionDownload[i].res_empty())
531         {
532         if (!uPresent && !dPresent)
533             {
534             sprintf(str, "<traff ");
535             strcat(r, str);
536             dPresent = true;
537             }
538
539         stringstream ss;
540         ss << req->sessionDownload[i].const_data();
541         sprintf(str, "MD%d=\"%s\" ", i, ss.str().c_str());
542         strcat(r, str);
543         }
544     }
545 if (uPresent || dPresent)
546     {
547     strcat(r, "/>");
548     }
549
550 //printf("%s\n", r);
551
552 if (!req->tariff.res_empty())
553     {
554     switch (req->chgTariff)
555         {
556         case TARIFF_NOW:
557             sprintf(str, "<tariff now=\"%s\"/>\n", req->tariff.const_data().c_str());
558             strcat(r, str);
559             break;
560         case TARIFF_REC:
561             sprintf(str, "<tariff recalc=\"%s\"/>\n", req->tariff.const_data().c_str());
562             strcat(r, str);
563             break;
564         case TARIFF_DEL:
565             sprintf(str, "<tariff delayed=\"%s\"/>\n", req->tariff.const_data().c_str());
566             strcat(r, str);
567             break;
568         }
569
570     }
571
572 if (!req->note.res_empty())
573     {
574     string note;
575     Encode12str(note, req->note);
576     sprintf(str, "<note value=\"%s\"/>", note.c_str());
577     strcat(r, str);
578     }
579
580 if (!req->name.res_empty())
581     {
582     string name;
583     Encode12str(name, req->name);
584     sprintf(str, "<name value=\"%s\"/>", name.c_str());
585     strcat(r, str);
586     }
587
588 if (!req->address.res_empty())
589     {
590     string address;
591     Encode12str(address, req->address);
592     sprintf(str, "<address value=\"%s\"/>", address.c_str());
593     strcat(r, str);
594     }
595
596 if (!req->email.res_empty())
597     {
598     string email;
599     Encode12str(email, req->email);
600     sprintf(str, "<email value=\"%s\"/>", email.c_str());
601     strcat(r, str);
602     }
603
604 if (!req->phone.res_empty())
605     {
606     string phone;
607     Encode12str(phone, req->phone);
608     sprintf(str, "<phone value=\"%s\"/>", phone.c_str());
609     strcat(r, str);
610     }
611
612 if (!req->group.res_empty())
613     {
614     string group;
615     Encode12str(group, req->group);
616     sprintf(str, "<group value=\"%s\"/>", group.c_str());
617     strcat(r, str);
618     }
619
620 for (int i = 0; i < USERDATA_NUM; i++)
621     {
622     if (!req->userData[i].res_empty())
623         {
624         string ud;
625         Encode12str(ud, req->userData[i]);
626         sprintf(str, "<userdata%d value=\"%s\"/>", i, ud.c_str());
627         strcat(r, str);
628         }
629     }
630
631 strcat(r, "</SetUser>\n");
632 }
633 //-----------------------------------------------------------------------------
634 int CheckParameters(REQUEST * req)
635 {
636 bool su = false;
637 bool sd = false;
638 bool mu = false;
639 bool md = false;
640 bool ud = false;
641 bool a = !req->admLogin.res_empty()
642     && !req->admPasswd.res_empty()
643     && !req->server.res_empty()
644     && !req->port.res_empty()
645     && !req->login.res_empty();
646
647 bool b = !req->cash.res_empty()
648     || !req->setCash.res_empty()
649     || !req->credit.res_empty()
650     || !req->prepaidTraff.res_empty()
651     || !req->tariff.res_empty()
652     || !req->usrMsg.res_empty()
653     || !req->usrPasswd.res_empty()
654
655     || !req->note.res_empty()
656     || !req->name.res_empty()
657     || !req->address.res_empty()
658     || !req->email.res_empty()
659     || !req->phone.res_empty()
660     || !req->group.res_empty()
661     || !req->ips.res_empty()    // IP-address of user
662
663     || !req->createUser
664     || !req->deleteUser;
665
666
667 for (int i = 0; i < DIR_NUM; i++)
668     {
669     if (req->sessionUpload[i].res_empty())
670         {
671         su = true;
672         break;
673         }
674     }
675
676 for (int i = 0; i < DIR_NUM; i++)
677     {
678     if (req->sessionDownload[i].res_empty())
679         {
680         sd = true;
681         break;
682         }
683     }
684
685 for (int i = 0; i < DIR_NUM; i++)
686     {
687     if (req->monthUpload[i].res_empty())
688         {
689         mu = true;
690         break;
691         }
692     }
693
694 for (int i = 0; i < DIR_NUM; i++)
695     {
696     if (req->monthDownload[i].res_empty())
697         {
698         md = true;
699         break;
700         }
701     }
702
703 for (int i = 0; i < DIR_NUM; i++)
704     {
705     if (req->userData[i].res_empty())
706         {
707         ud = true;
708         break;
709         }
710     }
711
712
713 //printf("a=%d, b=%d, u=%d, d=%d ud=%d\n", a, b, u, d, ud);
714 return a && (b || su || sd || mu || md || ud);
715 }
716 //-----------------------------------------------------------------------------
717 int CheckParametersGet(REQUEST * req)
718 {
719 return CheckParameters(req);
720 }
721 //-----------------------------------------------------------------------------
722 int CheckParametersSet(REQUEST * req)
723 {
724 return CheckParameters(req);
725 }
726 //-----------------------------------------------------------------------------
727 bool mainGet(int argc, char **argv)
728 {
729 int c;
730 REQUEST req;
731 RESETABLE<string>   t1;
732 int missedOptionArg = false;
733
734 const char * short_options_get = "s:p:a:w:u:crtmodieNADLPGISOE";
735 int option_index = -1;
736
737 while (1)
738     {
739     option_index = -1;
740     c = getopt_long(argc, argv, short_options_get, long_options_get, &option_index);
741     if (c == -1)
742         break;
743
744     switch (c)
745         {
746         case 's': //server
747             req.server = optarg;
748             break;
749
750         case 'p': //port
751             req.port = ParseServerPort(optarg);
752             //req.portReq = 1;
753             break;
754
755         case 'a': //admin
756             req.admLogin = ParseAdminLogin(optarg);
757             break;
758
759         case 'w': //admin password
760             req.admPasswd = ParsePassword(optarg);
761             break;
762
763         case 'o': //change user password
764             req.usrPasswd = " ";
765             break;
766
767         case 'u': //user
768             req.login = ParseUser(optarg);
769             break;
770
771         case 'c': //get cash
772             req.cash = 1;
773             break;
774
775         case 'r': //credit
776             req.credit = 1;
777             break;
778
779         case 'E': //credit expire
780             req.creditExpire = 1;
781             break;
782
783         case 'd': //down
784             req.down = 1;
785             break;
786
787         case 'i': //passive
788             req.passive = 1;
789             break;
790
791         case 't': //tariff
792             req.tariff = " ";
793             break;
794
795         case 'e': //Prepaid Traffic
796             req.prepaidTraff = 1;
797             break;
798
799         case 'N': //Note
800             req.note = " ";
801             break;
802
803         case 'A': //nAme
804             req.name = " ";
805             break;
806
807         case 'D': //aDdress
808             req.address =" ";
809             break;
810
811         case 'L': //emaiL
812             req.email = " ";
813             break;
814
815         case 'P': //phone
816             req.phone = " ";
817             break;
818
819         case 'G': //Group
820             req.group = " ";
821             break;
822
823         case 'I': //IP-address of user
824             req.ips = " ";
825             break;
826
827         case 'S': //Detail stat status
828             req.disableDetailStat = " ";
829             break;
830
831         case 'O': //Always online status
832             req.alwaysOnline = " ";
833             break;
834
835         case 500: //U
836             SetArrayItem(req.sessionUpload, optarg, 1);
837             //req.sessionUpload[optarg] = 1;
838             break;
839         case 501:
840             SetArrayItem(req.sessionDownload, optarg, 1);
841             //req.sessionDownload[optarg] = 1;
842             break;
843         case 502:
844             SetArrayItem(req.monthUpload, optarg, 1);
845             //req.monthUpload[optarg] = 1;
846             break;
847         case 503:
848             SetArrayItem(req.monthDownload, optarg, 1);
849             //req.monthDownload[optarg] = 1;
850             break;
851
852         case 700: //UserData
853             SetArrayItem(req.userData, optarg, std::string(" "));
854             //req.userData[optarg] = " ";
855             break;
856
857         case 800:
858             req.authBy = true;
859             break;
860
861         case '?':
862         case ':':
863             missedOptionArg = true;
864             break;
865
866         default:
867             printf ("?? getopt returned character code 0%o ??\n", c);
868         }
869     }
870
871 if (optind < argc)
872     {
873     printf ("non-option ARGV-elements: ");
874     while (optind < argc)
875         printf ("%s ", argv[optind++]);
876     UsageInfo();
877     exit(PARAMETER_PARSING_ERR_CODE);
878     }
879
880 if (missedOptionArg || !CheckParametersGet(&req))
881     {
882     //printf("Parameter needed\n");
883     UsageInfo();
884     exit(PARAMETER_PARSING_ERR_CODE);
885     }
886
887 if (req.authBy)
888     return ProcessAuthBy(req.server, req.port, req.admLogin, req.admPasswd, req.login);
889 else
890     return ProcessGetUser(req.server, req.port, req.admLogin, req.admPasswd, req.login, req);
891 }
892 //-----------------------------------------------------------------------------
893 bool mainSet(int argc, char **argv)
894 {
895 string str;
896
897 int c;
898 bool isMessage = false;
899 REQUEST req;
900
901 RESETABLE<string>   t1;
902
903 const char * short_options_set = "s:p:a:w:u:c:r:t:m:o:d:i:e:v:nlN:A:D:L:P:G:I:S:O:E:";
904
905 int missedOptionArg = false;
906
907 while (1)
908     {
909     int option_index = -1;
910
911     c = getopt_long(argc, argv, short_options_set, long_options_set, &option_index);
912
913     if (c == -1)
914         break;
915
916     switch (c)
917         {
918         case 's': //server
919             req.server = optarg;
920             break;
921
922         case 'p': //port
923             req.port = ParseServerPort(optarg);
924             //req.portReq = 1;
925             break;
926
927         case 'a': //admin
928             req.admLogin = ParseAdminLogin(optarg);
929             break;
930
931         case 'w': //admin password
932             req.admPasswd = ParsePassword(optarg);
933             break;
934
935         case 'o': //change user password
936             req.usrPasswd = ParsePassword(optarg);
937             break;
938
939         case 'u': //user
940             req.login = ParseUser(optarg);
941             break;
942
943         case 'c': //add cash
944             req.cash = ParseCash(optarg, &req.message);
945             break;
946
947         case 'v': //set cash
948             req.setCash = ParseCash(optarg, &req.message);
949             break;
950
951         case 'r': //credit
952             req.credit = ParseCredit(optarg);
953             break;
954
955         case 'E': //credit expire
956             req.creditExpire = ParseCreditExpire(optarg);
957             break;
958
959         case 'd': //down
960             req.down = ParseDownPassive(optarg);
961             break;
962
963         case 'i': //passive
964             req.passive = ParseDownPassive(optarg);
965             break;
966
967         case 't': //tariff
968             req.tariff = ParseTariff(optarg, req.chgTariff);
969             break;
970
971         case 'm': //message
972             ParseAnyString(optarg, &str);
973             req.usrMsg = str;
974             isMessage = true;
975             break;
976
977         case 'e': //Prepaid Traffic
978             req.prepaidTraff = ParsePrepaidTraffic(optarg);
979             break;
980
981         case 'n': //Create User
982             req.createUser = true;
983             break;
984
985         case 'l': //Delete User
986             req.deleteUser = true;
987             break;
988
989         case 'N': //Note
990             ParseAnyString(optarg, &str, "koi8-ru");
991             req.note = str;
992             break;
993
994         case 'A': //nAme
995             ParseAnyString(optarg, &str, "koi8-ru");
996             req.name = str;
997             break;
998
999         case 'D': //aDdress
1000             ParseAnyString(optarg, &str, "koi8-ru");
1001             req.address = str;
1002             break;
1003
1004         case 'L': //emaiL
1005             ParseAnyString(optarg, &str, "koi8-ru");
1006             req.email = str;
1007             //printf("EMAIL=%s\n", optarg);
1008             break;
1009
1010         case 'P': //phone
1011             ParseAnyString(optarg, &str);
1012             req.phone = str;
1013             break;
1014
1015         case 'G': //Group
1016             ParseAnyString(optarg, &str, "koi8-ru");
1017             req.group = str;
1018             break;
1019
1020         case 'I': //IP-address of user
1021             ParseAnyString(optarg, &str);
1022             req.ips = str;
1023             break;
1024
1025         case 'S':
1026             req.disableDetailStat = ParseDownPassive(optarg);
1027             break;
1028
1029         case 'O':
1030             req.alwaysOnline = ParseDownPassive(optarg);
1031             break;
1032
1033         case 500: //U
1034             SetArrayItem(req.sessionUpload, optarg, ParseTraff(argv[optind++]));
1035             //req.sessionUpload[optarg] = ParseTraff(argv[optind++]);
1036             break;
1037         case 501:
1038             SetArrayItem(req.sessionDownload, optarg, ParseTraff(argv[optind++]));
1039             //req.sessionDownload[optarg] = ParseTraff(argv[optind++]);
1040             break;
1041         case 502:
1042             SetArrayItem(req.monthUpload, optarg, ParseTraff(argv[optind++]));
1043             //req.monthUpload[optarg] = ParseTraff(argv[optind++]);
1044             break;
1045         case 503:
1046             SetArrayItem(req.monthDownload, optarg, ParseTraff(argv[optind++]));
1047             //req.monthDownload[optarg] = ParseTraff(argv[optind++]);
1048             break;
1049
1050         case 700: //UserData
1051             ParseAnyString(argv[optind++], &str);
1052             SetArrayItem(req.userData, optarg, str);
1053             //req.userData[optarg] = str;
1054             break;
1055
1056         case '?':
1057             missedOptionArg = true;
1058             break;
1059
1060         case ':':
1061             missedOptionArg = true;
1062             break;
1063
1064         default:
1065             printf("?? getopt returned character code 0%o ??\n", c);
1066         }
1067     }
1068
1069 if (optind < argc)
1070     {
1071     printf ("non-option ARGV-elements: ");
1072     while (optind < argc)
1073         printf ("%s ", argv[optind++]);
1074     UsageConf();
1075     exit(PARAMETER_PARSING_ERR_CODE);
1076     }
1077
1078 if (missedOptionArg || !CheckParametersSet(&req))
1079     {
1080     //printf("Parameter needed\n");
1081     UsageConf();
1082     exit(PARAMETER_PARSING_ERR_CODE);
1083     }
1084
1085 const int rLen = 20000;
1086 char rstr[rLen];
1087 memset(rstr, 0, rLen);
1088
1089 CreateRequestSet(&req, rstr);
1090 if (isMessage)
1091     return ProcessSendMessage(req.server, req.port, req.admLogin, req.admPasswd, rstr);
1092
1093 return ProcessSetUser(req.server, req.port, req.admLogin, req.admPasswd, rstr);
1094 }
1095 //-----------------------------------------------------------------------------
1096 int main(int argc, char **argv)
1097 {
1098 if (argc <= 2)
1099     {
1100     UsageConf();
1101     exit(PARAMETER_PARSING_ERR_CODE);
1102     }
1103
1104 if (strcmp(argv[1], "get") == 0)
1105     {
1106     //printf("get\n");
1107     return mainGet(argc - 1, argv + 1);
1108     }
1109 else if (strcmp(argv[1], "set") == 0)
1110     {
1111     //printf("set\n");
1112     if (mainSet(argc - 1, argv + 1) )
1113         return 0;
1114     return -1;
1115     }
1116 else
1117     {
1118     UsageConf();
1119     exit(PARAMETER_PARSING_ERR_CODE);
1120     }
1121 return UNKNOWN_ERR_CODE;
1122 }
1123 //-----------------------------------------------------------------------------
1124