projects/sgconf/common_sg.cpp

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