X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/61b5cf35c82a5850ceb07fca5232a409e249444c..cd0bf130b165970760884e279bb4301d3f6946ca:/projects/sgconf/common_sg.cpp diff --git a/projects/sgconf/common_sg.cpp b/projects/sgconf/common_sg.cpp index 898c334f..e407a716 100644 --- a/projects/sgconf/common_sg.cpp +++ b/projects/sgconf/common_sg.cpp @@ -43,17 +43,26 @@ #include #include +using namespace STG; + const int usageConf = 0; const int usageInfo = 1; const int TO_KOI8 = 0; const int FROM_KOI8 = 1; //----------------------------------------------------------------------------- +struct ResultData +{ + 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 @@ -287,15 +296,27 @@ void ConvertFromKOI8(const std::string & src, std::string * dst) ConvertKOI8(src, dst, FROM_KOI8); } //----------------------------------------------------------------------------- -int RecvSetUserAnswer(const char * ans, void * d) +void ResultCallback(bool result, const std::string & reason, void * d) { -GetUserData * data = static_cast(d); +ResultData * data = static_cast(d); +data->result = result; +data->reason = reason; +} +//----------------------------------------------------------------------------- +void RecvAuthByData(bool result, const std::string & reason, + const AUTH_BY::INFO & list, void * d) +{ +ResultData * data = static_cast(d); +data->result = result; +data->reason = reason; -std::cout << ans << std::endl; +if (!result) + return; -data->result = (strcasecmp("Ok", ans) == 0); +for (std::vector::const_iterator it = list.begin(); it != list.end(); ++it) + std::cout << *it << "\n"; -return 0; +std::cout << std::endl; } //----------------------------------------------------------------------------- struct StringReqParams @@ -305,23 +326,29 @@ struct StringReqParams const std::string * value; }; //----------------------------------------------------------------------------- -void GetUserCallback(const PARSER_GET_USER::INFO & info, void * d) +void GetUserCallback(bool result, const std::string& reason, const GET_USER::INFO & info, void * d) { GetUserData * data = static_cast(d); +data->result = false; +data->reason = reason; + +if (!result) + return; if (info.login == "") { data->result = false; + data->reason = "Invalid login."; return; } -if (!data->request.cash.res_empty()) +if (!data->request.cash.empty()) cout << "cash = " << info.cash << endl; -if (!data->request.credit.res_empty()) +if (!data->request.credit.empty()) cout << "credit = " << info.credit << endl; -if (!data->request.creditExpire.res_empty()) +if (!data->request.creditExpire.empty()) { char buf[32]; struct tm brokenTime; @@ -341,40 +368,40 @@ if (!data->request.creditExpire.res_empty()) cout << "creditExpire = " << buf << endl; } -if (!data->request.down.res_empty()) +if (!data->request.down.empty()) cout << "down = " << info.down << endl; -if (!data->request.passive.res_empty()) +if (!data->request.passive.empty()) cout << "passive = " << info.passive << endl; -if (!data->request.disableDetailStat.res_empty()) +if (!data->request.disableDetailStat.empty()) cout << "disableDetailStat = " << info.disableDetailStat << endl; -if (!data->request.alwaysOnline.res_empty()) +if (!data->request.alwaysOnline.empty()) cout << "alwaysOnline = " << info.alwaysOnline << endl; -if (!data->request.prepaidTraff.res_empty()) +if (!data->request.prepaidTraff.empty()) cout << "prepaidTraff = " << info.prepaidTraff << endl; for (int i = 0; i < DIR_NUM; i++) { - if (!data->request.sessionUpload[i].res_empty()) + if (!data->request.sessionUpload[i].empty()) cout << "session upload for dir " << i << " = " << info.stat.su[i] << endl; - if (!data->request.sessionDownload[i].res_empty()) + if (!data->request.sessionDownload[i].empty()) cout << "session download for dir " << i << "=" << info.stat.sd[i] << endl; } for (int i = 0; i < DIR_NUM; i++) { - if (!data->request.monthUpload[i].res_empty()) + if (!data->request.monthUpload[i].empty()) cout << "month upload for dir " << i << " = " << info.stat.mu[i] << endl; - if (!data->request.monthDownload[i].res_empty()) + if (!data->request.monthDownload[i].empty()) cout << "month download for dir " << i << " = " << info.stat.md[i] << endl; } for (int i = 0; i < USERDATA_NUM; i++) { - if (!data->request.userData[i].res_empty()) + if (!data->request.userData[i].empty()) { std::string str; ConvertFromKOI8(info.userData[i], &str); @@ -396,7 +423,7 @@ StringReqParams strReqParams[] = }; for (unsigned i = 0; i < sizeof(strReqParams) / sizeof(StringReqParams); i++) { - if (!strReqParams[i].reqParam.res_empty()) + if (!strReqParams[i].reqParam.empty()) { string str; ConvertFromKOI8(*strReqParams[i].value, &str); @@ -406,114 +433,104 @@ for (unsigned i = 0; i < sizeof(strReqParams) / sizeof(StringReqParams); i++) data->result = true; } //----------------------------------------------------------------------------- -void RecvAuthByData(const PARSER_AUTH_BY::INFO & list, void *) +bool ProcessSetUser(const std::string & server, + int port, + const std::string & login, + const std::string & password, + const std::string & user, + const USER_CONF_RES & conf, + const USER_STAT_RES & stat) { -for (std::vector::const_iterator it = list.begin(); it != list.end(); ++it) - cout << *it << "\n"; -cout << endl; -} -//----------------------------------------------------------------------------- -int ProcessSetUser(const std::string &server, - int port, - const std::string &admLogin, - const std::string &admPasswd, - const std::string &str, - bool isMessage) -{ -SERVCONF sc; +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(user, conf, stat, ResultCallback, &data); -REQUEST request; -GetUserData cbdata(request, false); - -int res = 0; -if (isMessage) +if (res == st_ok && data.result) { - sc.SetSendMessageCallback(RecvSetUserAnswer, &cbdata); - res = sc.SendMessage(str.c_str()); + printf("Ok\n"); + return false; } + +printf("Error\n"); +if (res != st_ok) + printf("%s\n", sc.GetStrError().c_str()); else - { - sc.SetChgUserCallback(RecvSetUserAnswer, &cbdata); - res = 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 & user, const std::string & text) +{ +SERVCONF sc(server, port, login, password); + +ResultData data; +int res = sc.SendMessage(user, text, ResultCallback, &data); -if (res == st_ok && cbdata.result) +if (res == st_ok && data.result) { printf("Ok\n"); - return 0; - } -else - { - printf("Error\n"); - if (res != st_ok) - printf("%s\n", sc.GetStrError().c_str()); - return -1; + return true; } -return 0; +printf("Error\n"); +if (res != st_ok) + printf("%s\n", sc.GetStrError().c_str()); +else + printf("%s\n", data.reason.c_str()); +return false; } //----------------------------------------------------------------------------- -int ProcessGetUser(const std::string &server, - int port, - const std::string &admLogin, - const std::string &admPasswd, - const std::string &login, - REQUEST & request) +bool ProcessGetUser(const std::string &server, + int port, + const std::string &admLogin, + const std::string &admPasswd, + const std::string &login, + REQUEST & request) { -SERVCONF sc; - -sc.SetServer(server.c_str()); -sc.SetPort(port); -sc.SetAdmLogin(admLogin.c_str()); -sc.SetAdmPassword(admPasswd.c_str()); +SERVCONF sc(server, port, admLogin, admPasswd); GetUserData data(request, false); - -sc.SetGetUserCallback(GetUserCallback, &data); -bool res = (sc.GetUser(login.c_str()) == st_ok); +bool res = (sc.GetUser(login.c_str(), GetUserCallback, &data) == st_ok); if (res && data.result) { 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; } //----------------------------------------------------------------------------- -int ProcessAuthBy(const std::string &server, - int port, - const std::string &admLogin, - const std::string &admPasswd, - const std::string &login) +bool ProcessAuthBy(const std::string &server, + int port, + const std::string &admLogin, + const std::string &admPasswd, + const std::string &login) { -SERVCONF sc; +SERVCONF sc(server, port, admLogin, admPasswd); -sc.SetServer(server.c_str()); -sc.SetPort(port); -sc.SetAdmLogin(admLogin.c_str()); -sc.SetAdmPassword(admPasswd.c_str()); +ResultData data; +bool res = (sc.AuthBy(login.c_str(), RecvAuthByData, &data) == st_ok); -sc.SetAuthByCallback(RecvAuthByData, NULL); -bool res = (sc.AuthBy(login.c_str()) == st_ok); - -if (!res) +if (res && data.result) { - printf("Error\n"); - return -1; + printf("Ok\n"); + return true; } -printf("Ok\n"); -return 0; +printf("Error\n"); +if (!res) + printf("%s\n", sc.GetStrError().c_str()); +else + printf("%s\n", data.reason.c_str()); +return false; } //-----------------------------------------------------------------------------