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