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