Added missing header.

[stg.git] / projects / sgconf / common_sg.cpp

diff --git a/projects/sgconf/common_sg.cpp b/projects/sgconf/common_sg.cpp
index 77d7476fd02c33feff1c88c0153e6b4b16a829ab..6ccd4fe5128beedef5f058c8e6652e77b1ef3a8d 100644 (file)
--- a/projects/sgconf/common_sg.cpp
+++ b/projects/sgconf/common_sg.cpp
@@ -25,21 +25,25 @@
  */
 
 
  */
 
 

-#include <stdio.h>
-#include <string.h>
-#include <stdlib.h>
-#include <errno.h>
-#include <locale.h>
-#include <langinfo.h>
-#include <iostream>
-#include <iconv.h>
-
+#include "sg_error_codes.h"

 #include "common_sg.h"
 #include "version_sg.h"
 #include "common_sg.h"
 #include "version_sg.h"

-#include "common.h"
-#include "sg_error_codes.h"

 
 

-using namespace std;
+#include "stg/common.h"
+
+#include <iostream>
+#include <vector>
+
+#include <cstdio>
+#include <cstring>
+#include <cstdlib>
+#include <cerrno>
+#include <clocale>
+
+#include <langinfo.h>
+#include <iconv.h>
+
+using namespace STG;

 
 const int usageConf = 0;
 const int usageInfo = 1;
 
 const int usageConf = 0;
 const int usageInfo = 1;


@@ -47,18 +51,26 @@ const int usageInfo = 1;
 const int TO_KOI8 = 0;
 const int FROM_KOI8 = 1;
 //-----------------------------------------------------------------------------
 const int TO_KOI8 = 0;
 const int FROM_KOI8 = 1;
 //-----------------------------------------------------------------------------

-struct GetUserCbData
+struct ResultData

 {
 {

-    void * data;
-    bool * result;
+    bool result;
+    std::string reason;
+};
+//-----------------------------------------------------------------------------
+struct GetUserData
+{
+    GetUserData(REQUEST & req, bool res) : request(req), result(res) {}
+    REQUEST & request;
+    bool result;
+    std::string reason;

 };
 //---------------------------------------------------------------------------
 struct HelpParams
 {
 };
 //---------------------------------------------------------------------------
 struct HelpParams
 {

-    string setActionName;
-    string getActionName;
-    string valueName;
-    string valueParam;
+    std::string setActionName;
+    std::string getActionName;
+    std::string valueName;
+    std::string valueParam;

 };
 //---------------------------------------------------------------------------
 void Usage(int usageType)
 };
 //---------------------------------------------------------------------------
 void Usage(int usageType)


@@ -81,9 +93,10 @@ HelpParams hp[] =
 {
     {"set tariff",              "get tariff",           "-t",   "<tariff:now|delayed>"},
     {"set credit",              "get credit",           "-r",   "<credit>"},
 {
     {"set tariff",              "get tariff",           "-t",   "<tariff:now|delayed>"},
     {"set credit",              "get credit",           "-r",   "<credit>"},

+    {"set credit expire",       "get credit expire",    "-E",   "<credit_expire_date>"},

     {"set password",            "get password",         "-o",   "<new_password>"},
     {"set prepaid traffic",     "get prepaid traffic",  "-e",   "<prepaid>"},
     {"set password",            "get password",         "-o",   "<new_password>"},
     {"set prepaid traffic",     "get prepaid traffic",  "-e",   "<prepaid>"},

-    {"set IP-addresses",       "get IP-addresses",     "-I",   "<*|ip_addr[,ip_addr...]>"},
+    {"set IP-addresses",        "get IP-addresses",     "-I",   "<*|ip_addr[,ip_addr...]>"},

     {"set name",                "get name",             "-A",   "<name>"},
     {"set note",                "get note",             "-N",   "<note>"},
     {"set street address",      "get street address",   "-D",   "<address>"},
     {"set name",                "get name",             "-A",   "<name>"},
     {"set note",                "get note",             "-N",   "<note>"},
     {"set street address",      "get street address",   "-D",   "<address>"},


@@ -121,6 +134,9 @@ printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --
 printf("To get userdata<0...9> use:\n");
 printf("sgconf get -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --ud0 [--ud1 ...]\n\n");
 
 printf("To get userdata<0...9> use:\n");
 printf("sgconf get -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --ud0 [--ud1 ...]\n\n");
 

+printf("To get user's authorizers list use:\n");
+printf("sgconf get -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> --authorized-by\n\n");
+

 printf("To send message use:\n");
 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> -m <message>\n\n");
 
 printf("To send message use:\n");
 printf("sgconf set -s <server> -p <port> -a <admin> -w <admin_pass> -u <user> -m <message>\n\n");
 


@@ -148,6 +164,7 @@ for (int i = 0; i < (int)strlen(login); i++)
     if (!(( login[i] >= 'a' && login[i] <= 'z')
         || (login[i] >= 'A' && login[i] <= 'Z')
         || (login[i] >= '0' && login[i] <= '9')
     if (!(( login[i] >= 'a' && login[i] <= 'z')
         || (login[i] >= 'A' && login[i] <= 'Z')
         || (login[i] >= '0' && login[i] <= '9')

+        ||  login[i] == '.'

         ||  login[i] == '_'
         ||  login[i] == '-'))
         {
         ||  login[i] == '_'
         ||  login[i] == '-'))
         {


@@ -162,7 +179,7 @@ short int ParseServerPort(const char * p)
 int port;
 if (str2x(p, port) != 0)
     {
 int port;
 if (str2x(p, port) != 0)
     {

-    printf("Incorresct server port %s\n", p);
+    printf("Incorrect server port %s\n", p);

     exit(NETWORK_ERR_CODE);
     }
 return (short)port;
     exit(NETWORK_ERR_CODE);
     }
 return (short)port;


@@ -172,7 +189,7 @@ char * ParseAdminLogin(char * adm)
 {
 if (CheckLogin(adm))
     {
 {
 if (CheckLogin(adm))
     {

-    printf("Incorresct admin login %s\n", adm);
+    printf("Incorrect admin login %s\n", adm);

     exit(PARAMETER_PARSING_ERR_CODE);
     }
 return adm;
     exit(PARAMETER_PARSING_ERR_CODE);
     }
 return adm;


@@ -193,13 +210,13 @@ char * ParseUser(char * usr)
 {
 if (CheckLogin(usr))
     {
 {
 if (CheckLogin(usr))
     {

-    printf("Incorresct user login %s\n", usr);
+    printf("Incorrect user login %s\n", usr);

     exit(PARAMETER_PARSING_ERR_CODE);
     }
 return usr;
 }
 //-----------------------------------------------------------------------------
     exit(PARAMETER_PARSING_ERR_CODE);
     }
 return usr;
 }
 //-----------------------------------------------------------------------------

-void ConvertKOI8(const string & src, string * dst, int encType)
+void ConvertKOI8(const std::string & src, std::string * dst, int encType)

 {
 iconv_t cd;
 char * ob = new char[src.size() * 2 + 1];
 {
 iconv_t cd;
 char * ob = new char[src.size() * 2 + 1];


@@ -218,12 +235,12 @@ char charsetT[100];
 if (encType == TO_KOI8)
     {
     strcpy(charsetF, nl_langinfo(CODESET));
 if (encType == TO_KOI8)
     {
     strcpy(charsetF, nl_langinfo(CODESET));

-    strcpy(charsetT, "koi8-r");
+    strcpy(charsetT, "koi8-ru");

     }
 else
     {
     strcpy(charsetT, nl_langinfo(CODESET));
     }
 else
     {
     strcpy(charsetT, nl_langinfo(CODESET));

-    strcpy(charsetF, "koi8-r");
+    strcpy(charsetF, "koi8-ru");

     }
 
 size_t nconv = 1;
     }
 
 size_t nconv = 1;


@@ -274,106 +291,135 @@ delete[] ob;
 delete[] ib;
 }
 //-----------------------------------------------------------------------------
 delete[] ib;
 }
 //-----------------------------------------------------------------------------

-void ConvertFromKOI8(const string & src, string * dst)
+void ConvertFromKOI8(const std::string & src, std::string * dst)

 {
 ConvertKOI8(src, dst, FROM_KOI8);
 }
 //-----------------------------------------------------------------------------
 {
 ConvertKOI8(src, dst, FROM_KOI8);
 }
 //-----------------------------------------------------------------------------

-void ConvertToKOI8(const string & src, string * dst)
+void ResultCallback(bool result, const std::string & reason, void * d)

 {
 {

-ConvertKOI8(src, dst, TO_KOI8);
+ResultData * data = static_cast<ResultData *>(d);
+data->result = result;
+data->reason = reason;

 }
 //-----------------------------------------------------------------------------
 }
 //-----------------------------------------------------------------------------

-int RecvSetUserAnswer(const char * ans, void * d)
+void RecvAuthByData(bool result, const std::string & reason,
+                    const AUTH_BY::INFO & list, void * d)

 {
 {

-GetUserCbData * gucbd;
-gucbd = (GetUserCbData *)d;
-
-bool * result = gucbd->result;
+ResultData * data = static_cast<ResultData *>(d);
+data->result = result;
+data->reason = reason;

 
 

-//REQUEST * req = (REQUEST *)gucbd->data;
+if (!result)
+    return;

 
 

-//printf("ans=%s\n", ans);
-if (strcasecmp("Ok", ans) == 0)
-    *result = true;
-else
-    *result = false;
+for (std::vector<std::string>::const_iterator it = list.begin(); it != list.end(); ++it)
+    std::cout << *it << "\n";

 
 

-return 0;
+std::cout << std::endl;

 }
 //-----------------------------------------------------------------------------
 struct StringReqParams
 {
 }
 //-----------------------------------------------------------------------------
 struct StringReqParams
 {

-    string name;
-    RESETABLE<string> reqParam;
-    string * value;
+    std::string name;
+    RESETABLE<std::string> reqParam;
+    const std::string * value;

 };
 //-----------------------------------------------------------------------------
 };
 //-----------------------------------------------------------------------------

-void RecvUserData(USERDATA * ud, void * d)
+void GetUserCallback(bool result, const std::string& reason, const GET_USER::INFO & info, void * d)

 {
 {

-GetUserCbData * gucbd;
-gucbd = (GetUserCbData *)d;
+GetUserData * data = static_cast<GetUserData *>(d);
+data->result = false;
+data->reason = reason;

 
 

-bool * result = gucbd->result;
-
-REQUEST * req = (REQUEST *)gucbd->data;
+if (!result)
+    return;

 
 

-if (ud->login == "")
+if (info.login == "")

     {
     {

-    *result = false;
+    data->result = false;
+    data->reason = "Invalid login.";

     return;
     }
 
     return;
     }
 

-if (!req->cash.res_empty())
-    cout << "cash=" << ud->cash << endl;
+if (!data->request.cash.res_empty())
+    cout << "cash = " << info.cash << endl;

 
 

-if (!req->credit.res_empty())
-    cout << "credit=" << ud->credit << endl;
+if (!data->request.credit.res_empty())
+    cout << "credit = " << info.credit << endl;

 
 

-if (!req->down.res_empty())
-    cout << "down=" << ud->down << endl;
+if (!data->request.creditExpire.res_empty())
+    {
+    char buf[32];
+    struct tm brokenTime;
+    time_t tt = info.creditExpire;
+
+    brokenTime.tm_wday = 0;
+    brokenTime.tm_yday = 0;
+    brokenTime.tm_isdst = 0;
+    brokenTime.tm_hour = 0;
+    brokenTime.tm_min = 0;
+    brokenTime.tm_sec = 0;

 
 

-if (!req->passive.res_empty())
-    cout << "passive=" << ud->passive << endl;
+    gmtime_r(&tt, &brokenTime);

 
 

-if (!req->disableDetailStat.res_empty())
-    cout << "disableDetailStat=" << ud->disableDetailStat << endl;
+    strftime(buf, 32, "%Y-%m-%d", &brokenTime);
+
+    cout << "creditExpire = " << buf << endl;
+    }

 
 

-if (!req->alwaysOnline.res_empty())
-    cout << "alwaysOnline=" << ud->alwaysOnline << endl;
+if (!data->request.down.res_empty())
+    cout << "down = " << info.down << endl;

 
 

-if (!req->prepaidTraff.res_empty())
-    cout << "prepaidTraff=" << ud->prepaidTraff << endl;
+if (!data->request.passive.res_empty())
+    cout << "passive = " << info.passive << endl;
+
+if (!data->request.disableDetailStat.res_empty())
+    cout << "disableDetailStat = " << info.disableDetailStat << endl;
+
+if (!data->request.alwaysOnline.res_empty())
+    cout << "alwaysOnline = " << info.alwaysOnline << endl;
+
+if (!data->request.prepaidTraff.res_empty())
+    cout << "prepaidTraff = " << info.prepaidTraff << endl;
+
+for (int i = 0; i < DIR_NUM; i++)
+    {
+    if (!data->request.sessionUpload[i].res_empty())
+        cout << "session upload for dir " << i << " = " << info.stat.su[i] << endl;
+    if (!data->request.sessionDownload[i].res_empty())
+        cout << "session download for dir " << i << "=" << info.stat.sd[i] << endl;
+    }

 
 for (int i = 0; i < DIR_NUM; i++)
     {
 
 for (int i = 0; i < DIR_NUM; i++)
     {

-    if (!req->u[i].res_empty())
-        cout << "u" << i << "=" << ud->stat.mu[i] << endl;
-    if (!req->d[i].res_empty())
-        cout << "d" << i << "=" << ud->stat.md[i] << endl;
+    if (!data->request.monthUpload[i].res_empty())
+        cout << "month upload for dir " << i << " = " << info.stat.mu[i] << endl;
+    if (!data->request.monthDownload[i].res_empty())
+        cout << "month download for dir " << i << " = " << info.stat.md[i] << endl;

     }
 
 for (int i = 0; i < USERDATA_NUM; i++)
     {
     }
 
 for (int i = 0; i < USERDATA_NUM; i++)
     {

-    if (!req->ud[i].res_empty())
+    if (!data->request.userData[i].res_empty())

         {
         {

-        string str;
-        ConvertFromKOI8(ud->userData[i], &str);
-        cout << "userdata" << i << "=" << str << endl;
+        std::string str;
+        ConvertFromKOI8(info.userData[i], &str);
+        cout << "user data " << i << " = " << str << endl;

         }
     }
 
 StringReqParams strReqParams[] =
 {
         }
     }
 
 StringReqParams strReqParams[] =
 {

-    {"note",     req->note,        &ud->note},
-    {"name",     req->name,        &ud->name},
-    {"address",  req->address,     &ud->address},
-    {"email",    req->email,       &ud->email},
-    {"phone",    req->phone,       &ud->phone},
-    {"group",    req->group,       &ud->group},
-    {"tariff",   req->tariff,      &ud->tariff},
-    {"password", req->usrPasswd,   &ud->password},
-    {"ip",      req->ips,         &ud->ips}    // IP-address of user
+    {"note",     data->request.note,        &info.note},
+    {"name",     data->request.name,        &info.name},
+    {"address",  data->request.address,     &info.address},
+    {"email",    data->request.email,       &info.email},
+    {"phone",    data->request.phone,       &info.phone},
+    {"group",    data->request.group,       &info.group},
+    {"tariff",   data->request.tariff,      &info.tariff},
+    {"password", data->request.usrPasswd,   &info.password},
+    {"ip",       data->request.ips,         &info.ips} // IP-address of user

 };
 for (unsigned i = 0; i < sizeof(strReqParams) / sizeof(StringReqParams); i++)
     {
 };
 for (unsigned i = 0; i < sizeof(strReqParams) / sizeof(StringReqParams); i++)
     {


@@ -381,99 +427,108 @@ for (unsigned i = 0; i < sizeof(strReqParams) / sizeof(StringReqParams); i++)
         {
         string str;
         ConvertFromKOI8(*strReqParams[i].value, &str);
         {
         string str;
         ConvertFromKOI8(*strReqParams[i].value, &str);

-        cout << strReqParams[i].name << "=" << str << endl;
+        cout << strReqParams[i].name << " = " << str << endl;

         }
     }
         }
     }

-*result = true;
+data->result = true;

 }
 //-----------------------------------------------------------------------------
 }
 //-----------------------------------------------------------------------------

-int ProcessSetUser(const std::string &server,
-                   int port,
-                   const std::string &admLogin,
-                   const std::string &admPasswd,
-                   const std::string &str,
-                   void * data,
-                   bool isMessage)
+bool ProcessSetUser(const std::string & server,
+                    int port,
+                    const std::string & login,
+                    const std::string & password,
+                    const std::string & str)

 {
 {

-SERVCONF sc;
-
-bool result = false;
-
+SERVCONF sc(server, port, login, password);

 
 

-sc.SetServer(server.c_str());  // õÓÔÁÎÁ×ÌÉ×ÁÅÍ ÉÍÑ ÓÅÒ×ÅÒÁ Ó ËÏÔÏÒÇÏ ÚÁÂÉÒÁÔØ ÉÎÆÕ
-sc.SetPort(port);           // ÁÄÍÉÎÓËÉÊ ÐÏÒÔ ÓÅÒ×ÅÒÁÐÏÒÔ
-sc.SetAdmLogin(admLogin.c_str());    // ÷ÙÓÔÁ×ÌÑÅÍ ÌÏÇÉÎ É ÐÁÒÏÌØ ÁÄÍÉÎÁ
-sc.SetAdmPassword(admPasswd.c_str());
+ResultData data;
+int res = sc.ChgUser(str.c_str(), ResultCallback, &data);

 
 

-// TODO Good variable name :)
-GetUserCbData gucbd;
-
-gucbd.data = data;
-gucbd.result = &result;
-
-if (isMessage)
+if (res == st_ok && data.result)

     {
     {

-    sc.SetSendMessageCb(RecvSetUserAnswer, &gucbd);
-    sc.MsgUser(str.c_str());
+    printf("Ok\n");
+    return false;

     }
     }

+
+printf("Error\n");
+if (res != st_ok)
+    printf("%s\n", sc.GetStrError().c_str());

 else
 else

-    {
-    sc.SetChgUserCb(RecvSetUserAnswer, &gucbd);
-    sc.ChgUser(str.c_str());
-    }
+    printf("%s\n", data.reason.c_str());
+return true;
+}
+//-----------------------------------------------------------------------------
+bool ProcessSendMessage(const std::string & server, uint16_t port,
+                        const std::string & login, const std::string & password,
+                        const std::string & requestString)
+{
+SERVCONF sc(server, port, login, password);

 
 

-if (result)
+ResultData data;
+int res = sc.SendMessage(requestString.c_str(), ResultCallback, &data);
+
+if (res == st_ok && data.result)

     {
     printf("Ok\n");
     {
     printf("Ok\n");

-    return 0;
+    return true;

     }
     }

+
+printf("Error\n");
+if (res != st_ok)
+    printf("%s\n", sc.GetStrError().c_str());

 else
 else

+    printf("%s\n", data.reason.c_str());
+return false;
+}
+//-----------------------------------------------------------------------------
+bool ProcessGetUser(const std::string &server,
+                    int port,
+                    const std::string &admLogin,
+                    const std::string &admPasswd,
+                    const std::string &login,
+                    REQUEST & request)
+{
+SERVCONF sc(server, port, admLogin, admPasswd);
+
+GetUserData data(request, false);
+bool res = (sc.GetUser(login.c_str(), GetUserCallback, &data) == st_ok);
+
+if (res && data.result)

     {
     {

-    printf("Error\n");
-    return -1;
+    printf("Ok\n");
+    return true;

     }
 
     }
 

-return 0;
+printf("Error\n");
+if (!res)
+    printf("%s\n", sc.GetStrError().c_str());
+else
+    printf("%s\n", data.reason.c_str());
+return false;

 }
 //-----------------------------------------------------------------------------
 }
 //-----------------------------------------------------------------------------

-int ProcessGetUser(const std::string &server,
+bool ProcessAuthBy(const std::string &server,

                    int port,
                    const std::string &admLogin,
                    const std::string &admPasswd,
                    int port,
                    const std::string &admLogin,
                    const std::string &admPasswd,

-                   const std::string &login,
-                   void * data)
+                   const std::string &login)

 {
 {

-SERVCONF sc;
-
-bool result = false;
-
-sc.SetServer(server.c_str());  // õÓÔÁÎÁ×ÌÉ×ÁÅÍ ÉÍÑ ÓÅÒ×ÅÒÁ Ó ËÏÔÏÒÇÏ ÚÁÂÉÒÁÔØ ÉÎÆÕ
-sc.SetPort(port);           // ÁÄÍÉÎÓËÉÊ ÐÏÒÔ ÓÅÒ×ÅÒÁÐÏÒÔ
-sc.SetAdmLogin(admLogin.c_str());    // ÷ÙÓÔÁ×ÌÑÅÍ ÌÏÇÉÎ É ÐÁÒÏÌØ ÁÄÍÉÎÁ
-sc.SetAdmPassword(admPasswd.c_str());
-
-// TODO Good variable name :)
-GetUserCbData gucbd;
+SERVCONF sc(server, port, admLogin, admPasswd);

 
 

-gucbd.data = data;
-gucbd.result = &result;
+ResultData data;
+bool res = (sc.AuthBy(login.c_str(), RecvAuthByData, &data) == st_ok);

 
 

-sc.SetGetUserDataRecvCb(RecvUserData, &gucbd);
-sc.GetUser(login.c_str());
-
-if (result)
+if (res && data.result)

     {
     printf("Ok\n");
     {
     printf("Ok\n");

-    return 0;
-    }
-else
-    {
-    printf("Error\n");
-    return -1;
+    return true;

     }
 
     }
 

-return 0;
+printf("Error\n");
+if (!res)
+    printf("%s\n", sc.GetStrError().c_str());
+else
+    printf("%s\n", data.reason.c_str());
+return false;

 }
 //-----------------------------------------------------------------------------
 }
 //-----------------------------------------------------------------------------

-
-