X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/26d5d878270d01123669ba6a38c306c398ed7b8d..908dee0db9288bb455e9f1b68dd81ab50897a677:/stglibs/srvconf.lib/servconf.cpp?ds=inline diff --git a/stglibs/srvconf.lib/servconf.cpp b/stglibs/srvconf.lib/servconf.cpp index 7b2c85ec..e0c2e58b 100644 --- a/stglibs/srvconf.lib/servconf.cpp +++ b/stglibs/srvconf.lib/servconf.cpp @@ -16,430 +16,400 @@ /* * Author : Boris Mikhailenko + * Author : Maxim Mamontov */ - /* - $Revision: 1.8 $ - $Date: 2010/08/04 00:40:38 $ - $Author: faust $ - */ +#include "stg/servconf.h" + +#include "netunit.h" + +#include "parsers/simple.h" +#include "parsers/get_container.h" + +#include "parsers/server_info.h" + +#include "parsers/get_admin.h" +#include "parsers/chg_admin.h" + +#include "parsers/get_tariff.h" +#include "parsers/chg_tariff.h" + +#include "parsers/auth_by.h" +#include "parsers/get_user.h" +#include "parsers/chg_user.h" + +#include "parsers/get_service.h" +#include "parsers/chg_service.h" + +#include "parsers/get_corp.h" +#include "parsers/chg_corp.h" + +#include "parsers/base.h" + +#include "stg/common.h" #include #include -#include "stg/common.h" -#include "stg/servconf.h" +#include -using namespace std; +using namespace STG; -//----------------------------------------------------------------------------- -int AnsRecv(void * data, list * list1) +class SERVCONF::IMPL { -//NODE * node; -SERVCONF * sc; -char ans[ENC_MSG_LEN + 1]; -int len, done = 0; +public: + IMPL(const std::string & server, uint16_t port, + const std::string & login, const std::string & password); + IMPL(const std::string & server, uint16_t port, + const std::string & localAddress, uint16_t localPort, + const std::string & login, const std::string & password); + ~IMPL() { XML_ParserFree(parser); } + + const std::string & GetStrError() const; + static void Start(void * data, const char * el, const char ** attr); + static void End(void * data, const char * el); + + int RawXML(const std::string & request, RAW_XML::CALLBACK f, void * data); + + template + int Exec(const std::string & request, C callback, void * data) + { + P cp(callback, data, encoding); + return ExecImpl(request, cp); + } + + template + int Exec(const std::string & tag, const std::string & request, C callback, void * data) + { + P cp(tag, callback, data, encoding); + return ExecImpl(request, cp); + } + +private: + NETTRANSACT nt; -sc = (SERVCONF*)data; + std::string encoding; + std::string errorMsg; + XML_Parser parser; -XML_ParserReset(sc->parser, NULL); -XML_SetElementHandler(sc->parser, Start, End); -XML_SetUserData(sc->parser, data); + static bool ParserRecv(const std::string & chunk, bool final, void * data); + static bool SimpleRecv(const std::string & chunk, bool final, void * data); + int ExecImpl(const std::string & request, PARSER & cp); +}; -//loop parsing -list::iterator node; -node = list1->begin(); +bool SERVCONF::IMPL::ParserRecv(const std::string & chunk, bool final, void * data) +{ +SERVCONF::IMPL * sc = static_cast(data); -if (node == list1->end()) +if (XML_Parse(sc->parser, chunk.c_str(), chunk.length(), final) == XML_STATUS_ERROR) { - return st_ok; + strprintf(&sc->errorMsg, "XML parse error at line %d: %s", + static_cast(XML_GetCurrentLineNumber(sc->parser)), + XML_ErrorString(XML_GetErrorCode(sc->parser))); + printf("%s\n", sc->errorMsg.c_str()); + return false; } -while (node != list1->end()) - { - strncpy(ans, node->c_str(), ENC_MSG_LEN); - ans[ENC_MSG_LEN] = 0; - //printf("---> %s\n", ans); - len = strlen(ans); - - if (XML_Parse(sc->parser, ans, len, done) == XML_STATUS_ERROR) - { - strprintf(&sc->errorMsg, "XML parse error at line %d: %s", - static_cast(XML_GetCurrentLineNumber(sc->parser)), - XML_ErrorString(XML_GetErrorCode(sc->parser))); - return st_xml_parse_error; - } - ++node; +return true; +} - } +bool SERVCONF::IMPL::SimpleRecv(const std::string & chunk, bool /*final*/, void * data) +{ +*static_cast(data) += chunk; +return true; +} -return 0; +SERVCONF::SERVCONF(const std::string & server, uint16_t port, + const std::string & login, const std::string & password) + : pImpl(new IMPL(server, port, login, password)) +{ } -//----------------------------------------------------------------------------- -void Start(void *data, const char *el, const char **attr) + +SERVCONF::SERVCONF(const std::string & server, uint16_t port, + const std::string & localAddress, uint16_t localPort, + const std::string & login, const std::string & password) + : pImpl(new IMPL(server, port, localAddress, localPort, login, password)) { -SERVCONF * sc; -sc = (SERVCONF*)data; -sc->Start(el, attr); } -//----------------------------------------------------------------------------- -void End(void *data, const char *el) + +SERVCONF::~SERVCONF() { -SERVCONF * sc; -sc = (SERVCONF*)data; -sc->End(el); +delete pImpl; } -//----------------------------------------------------------------------------- -SERVCONF::SERVCONF() - : currParser(NULL), - parseDepth(0), - error(0), - RecvUserDataCb(NULL), - RecvGetUserDataCb(NULL), - authByCallback(NULL), - serverInfoCallback(NULL), - RecvChgUserCb(NULL), - checkUserCallback(NULL), - RecvSendMessageCb(NULL), - getUserDataDataCb(NULL), - authByData(NULL), - getUsersDataDataCb(NULL), - serverInfoData(NULL), - chgUserDataCb(NULL), - checkUserData(NULL), - sendMessageDataCb(NULL) + +int SERVCONF::ServerInfo(SERVER_INFO::CALLBACK f, void * data) { -parser = XML_ParserCreate(NULL); +return pImpl->Exec("", f, data); } -//----------------------------------------------------------------------------- -void SERVCONF::SetServer(const char * server) + +int SERVCONF::RawXML(const std::string & request, RAW_XML::CALLBACK f, void * data) { -nt.SetServer(server); +return pImpl->RawXML(request, f, data); } -//----------------------------------------------------------------------------- -void SERVCONF::SetPort(uint16_t port) + +// -- Admins -- + +int SERVCONF::GetAdmins(GET_CONTAINER::CALLBACK::TYPE f, void * data) { -nt.SetServerPort(port); +return pImpl->Exec >("admins", "", f, data); } -//----------------------------------------------------------------------------- -void SERVCONF::SetAdmLogin(const char * login) + +int SERVCONF::GetAdmin(const std::string & login, GET_ADMIN::CALLBACK f, void * data) { -nt.SetLogin(login); +return pImpl->Exec("", f, data); } -//----------------------------------------------------------------------------- -void SERVCONF::SetAdmPassword(const char * password) + +int SERVCONF::ChgAdmin(const ADMIN_CONF_RES & conf, SIMPLE::CALLBACK f, void * data) { -nt.SetPassword(password); +return pImpl->Exec("ChgAdmin", "", f, data); } -//----------------------------------------------------------------------------- -int SERVCONF::GetUser(const char * l) + +int SERVCONF::AddAdmin(const std::string & login, + const ADMIN_CONF_RES & conf, + SIMPLE::CALLBACK f, void * data) { -char request[255]; -snprintf(request, 255, "", l); -int ret; +int res = pImpl->Exec("AddAdmin", "", f, data); +if (res != st_ok) + return res; +return pImpl->Exec("ChgAdmin", "", f, data); +} -currParser = &parserGetUser; -((PARSER_GET_USER*)currParser)->SetCallback(getUserCallback, getUserData); +int SERVCONF::DelAdmin(const std::string & login, SIMPLE::CALLBACK f, void * data) +{ +return pImpl->Exec("DelAdmin", "", f, data); +} -nt.Reset(); -nt.SetRxCallback(this, AnsRecv); +// -- Tariffs -- -if ((ret = nt.Connect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Transact(request)) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Disconnect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } +int SERVCONF::GetTariffs(GET_CONTAINER::CALLBACK::TYPE f, void * data) +{ +return pImpl->Exec >("tariffs", "", f, data); +} -return st_ok; +int SERVCONF::GetTariff(const std::string & name, GET_TARIFF::CALLBACK f, void * data) +{ +return pImpl->Exec("", f, data); } -//----------------------------------------------------------------------------- -int SERVCONF::AuthBy(const char * l) + +int SERVCONF::ChgTariff(const TARIFF_DATA_RES & tariffData, SIMPLE::CALLBACK f, void * data) { -char request[255]; -snprintf(request, 255, "", l); +return pImpl->Exec("SetTariff", "" + CHG_TARIFF::Serialize(tariffData, encoding) + "", f, data); +} -currParser = &parserAuthBy; -((PARSER_AUTH_BY*)currParser)->SetCallback(authByCallback, authByData); +int SERVCONF::AddTariff(const std::string & name, + const TARIFF_DATA_RES & tariffData, + SIMPLE::CALLBACK f, void * data) +{ +int res = pImpl->Exec("AddTariff", "", f, data); +if (res != st_ok) + return res; +return pImpl->Exec("SetTariff", "" + CHG_TARIFF::Serialize(tariffData, encoding) + "", f, data); +} -nt.Reset(); -nt.SetRxCallback(this, AnsRecv); +int SERVCONF::DelTariff(const std::string & name, SIMPLE::CALLBACK f, void * data) +{ +return pImpl->Exec("DelTariff", "", f, data); +} -int ret; -if ((ret = nt.Connect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Transact(request)) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Disconnect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } +// -- Users -- -return st_ok; +int SERVCONF::GetUsers(GET_CONTAINER::CALLBACK::TYPE f, void * data) +{ +return pImpl->Exec >("users", "", f, data); } -//----------------------------------------------------------------------------- -int SERVCONF::GetUsers() + +int SERVCONF::GetUser(const std::string & login, GET_USER::CALLBACK f, void * data) { -char request[] = ""; -int ret; +return pImpl->Exec("", f, data); +} -currParser = &parserGetUsers; -((PARSER_GET_USERS*)currParser)->SetCallback(getUsersCallback, getUsersData); +int SERVCONF::ChgUser(const std::string & login, + const USER_CONF_RES & conf, + const USER_STAT_RES & stat, + SIMPLE::CALLBACK f, void * data) +{ +return pImpl->Exec("" + CHG_USER::Serialize(conf, stat, encoding) + "", f, data); +} -nt.Reset(); -nt.SetRxCallback(this, AnsRecv); +int SERVCONF::DelUser(const std::string & login, SIMPLE::CALLBACK f, void * data) +{ +return pImpl->Exec("DelUser", "", f, data); +} -if ((ret = nt.Connect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Transact(request)) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Disconnect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } +int SERVCONF::AddUser(const std::string & login, + const USER_CONF_RES & conf, + const USER_STAT_RES & stat, + SIMPLE::CALLBACK f, void * data) +{ +int res = pImpl->Exec("AddUser", "", f, data); +if (res != st_ok) + return res; +return pImpl->Exec("" + CHG_USER::Serialize(conf, stat, encoding) + "", f, data); +} -return st_ok; +int SERVCONF::AuthBy(const std::string & login, AUTH_BY::CALLBACK f, void * data) +{ +return pImpl->Exec("", f, data); } -//----------------------------------------------------------------------------- -int SERVCONF::SendMessage(const char * login, const char * message, int prio) + +int SERVCONF::SendMessage(const std::string & login, const std::string & text, SIMPLE::CALLBACK f, void * data) { -char request[1000]; -char msg[500]; -Encode12(msg, message, strlen(message)); -snprintf(request, 1000, "", login, prio, msg); -int ret; +return pImpl->Exec("SendMessageResult", "", f, data); +} -currParser = &parserSendMessage; -parserSendMessage.SetSendMessageRecvCb(RecvSendMessageCb, sendMessageDataCb); +int SERVCONF::CheckUser(const std::string & login, const std::string & password, SIMPLE::CALLBACK f, void * data) +{ +return pImpl->Exec("CheckUser", "", f, data); +} -nt.Reset(); -nt.SetRxCallback(this, AnsRecv); +// -- Services -- -if ((ret = nt.Connect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Transact(request)) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Disconnect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } +int SERVCONF::GetServices(GET_CONTAINER::CALLBACK::TYPE f, void * data) +{ +return pImpl->Exec >("services", "", f, data); +} -return st_ok; +int SERVCONF::GetService(const std::string & name, GET_SERVICE::CALLBACK f, void * data) +{ +return pImpl->Exec("", f, data); } -//----------------------------------------------------------------------------- -int SERVCONF::ServerInfo() + +int SERVCONF::ChgService(const SERVICE_CONF_RES & conf, SIMPLE::CALLBACK f, void * data) { -char request[] = ""; -int ret; +return pImpl->Exec("SetService", "" + CHG_SERVICE::Serialize(conf, encoding) + "", f, data); +} -currParser = &parserServerInfo; -((PARSER_SERVER_INFO*)currParser)->SetCallback(serverInfoCallback, serverInfoData); +int SERVCONF::AddService(const std::string & name, + const SERVICE_CONF_RES & conf, + SIMPLE::CALLBACK f, void * data) +{ +int res = pImpl->Exec("AddService", "", f, data); +if (res != st_ok) + return res; +return pImpl->Exec("SetService", "" + CHG_SERVICE::Serialize(conf, encoding) + "", f, data); +} -nt.Reset(); -nt.SetRxCallback(this, AnsRecv); +int SERVCONF::DelService(const std::string & name, SIMPLE::CALLBACK f, void * data) +{ +return pImpl->Exec("DelService", "", f, data); +} -if ((ret = nt.Connect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Transact(request)) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Disconnect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } +// -- Corporations -- -return st_ok; +int SERVCONF::GetCorporations(GET_CONTAINER::CALLBACK::TYPE f, void * data) +{ +return pImpl->Exec >("corporations", "", f, data); } -//----------------------------------------------------------------------------- -int SERVCONF::ChgUser(const char * request) + +int SERVCONF::GetCorp(const std::string & name, GET_CORP::CALLBACK f, void * data) +{ +return pImpl->Exec("", f, data); +} + +int SERVCONF::ChgCorp(const CORP_CONF_RES & conf, SIMPLE::CALLBACK f, void * data) { -int ret; +return pImpl->Exec("SetCorp", "" + CHG_CORP::Serialize(conf, encoding) + "", f, data); +} -currParser = &parserChgUser; -((PARSER_CHG_USER*)currParser)->SetChgUserRecvCb(RecvChgUserCb, chgUserDataCb); +int SERVCONF::AddCorp(const std::string & name, + const CORP_CONF_RES & conf, + SIMPLE::CALLBACK f, void * data) +{ +int res = pImpl->Exec("AddCorp", "", f, data); +if (res != st_ok) + return res; +return pImpl->Exec("SetCorp", "" + CHG_CORP::Serialize(conf, encoding) + "", f, data); +} -nt.Reset(); -nt.SetRxCallback(this, AnsRecv); +int SERVCONF::DelCorp(const std::string & name, SIMPLE::CALLBACK f, void * data) +{ +return pImpl->Exec("DelCorp", "", f, data); +} -if ((ret = nt.Connect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Transact(request)) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Disconnect()) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } +const std::string & SERVCONF::GetStrError() const +{ +return pImpl->GetStrError(); +} -return st_ok; +//----------------------------------------------------------------------------- +SERVCONF::IMPL::IMPL(const std::string & server, uint16_t port, + const std::string & login, const std::string & password) + : nt(server, port, login, password) +{ +setlocale(LC_ALL, ""); +encoding = nl_langinfo(CODESET); +parser = XML_ParserCreate(NULL); } //----------------------------------------------------------------------------- -// TODO: remove this shit! +SERVCONF::IMPL::IMPL(const std::string & server, uint16_t port, + const std::string & localAddress, uint16_t localPort, + const std::string & login, const std::string & password) + : nt(server, port, localAddress, localPort, login, password) +{ +setlocale(LC_ALL, ""); +encoding = nl_langinfo(CODESET); +parser = XML_ParserCreate(NULL); +} //----------------------------------------------------------------------------- -int SERVCONF::MsgUser(const char * request) +void SERVCONF::IMPL::Start(void * data, const char * el, const char ** attr) { -int ret; - -currParser = &parserSendMessage; -parserSendMessage.SetSendMessageRecvCb(RecvSendMessageCb, sendMessageDataCb); - -nt.Reset(); -nt.SetRxCallback(this, AnsRecv); +PARSER * currParser = static_cast(data); +currParser->ParseStart(el, attr); +} +//----------------------------------------------------------------------------- +void SERVCONF::IMPL::End(void * data, const char * el) +{ +PARSER * currParser = static_cast(data); +currParser->ParseEnd(el); +} +//----------------------------------------------------------------------------- +const std::string & SERVCONF::IMPL::GetStrError() const +{ +return errorMsg; +} +//----------------------------------------------------------------------------- +int SERVCONF::IMPL::ExecImpl(const std::string & request, PARSER & cp) +{ +XML_ParserReset(parser, NULL); +XML_SetElementHandler(parser, Start, End); +XML_SetUserData(parser, &cp); +int ret = 0; if ((ret = nt.Connect()) != st_ok) { errorMsg = nt.GetError(); + cp.Failure(errorMsg); return ret; } -if ((ret = nt.Transact(request)) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Disconnect()) != st_ok) +if ((ret = nt.Transact(request, ParserRecv, this)) != st_ok) { errorMsg = nt.GetError(); + cp.Failure(errorMsg); return ret; } +nt.Disconnect(); return st_ok; } -//----------------------------------------------------------------------------- -int SERVCONF::CheckUser(const char * login, const char * password) -{ -char request[255]; -snprintf(request, 255, "", login, password); -int ret; - -currParser = &parserCheckUser; -((PARSER_CHECK_USER*)currParser)->SetCallback(checkUserCallback, checkUserData); - -nt.Reset(); -nt.SetRxCallback(this, AnsRecv); +int SERVCONF::IMPL::RawXML(const std::string & request, RAW_XML::CALLBACK callback, void * data) +{ +int ret = 0; if ((ret = nt.Connect()) != st_ok) { errorMsg = nt.GetError(); + callback(false, errorMsg, "", data); return ret; } -if ((ret = nt.Transact(request)) != st_ok) - { - errorMsg = nt.GetError(); - return ret; - } -if ((ret = nt.Disconnect()) != st_ok) +std::string response; +if ((ret = nt.Transact(request, SimpleRecv, &response)) != st_ok) { errorMsg = nt.GetError(); + callback(false, errorMsg, "", data); return ret; } +nt.Disconnect(); +callback(true, "", response, data); return st_ok; } -//----------------------------------------------------------------------------- -int SERVCONF::Start(const char *el, const char **attr) -{ -currParser->ParseStart(el, attr); -return 0; -} -//----------------------------------------------------------------------------- -void SERVCONF::End(const char *el) -{ -currParser->ParseEnd(el); -} -//----------------------------------------------------------------------------- -void SERVCONF::SetGetUsersCallback(PARSER_GET_USERS::CALLBACK f, void * data) -{ -getUsersCallback = f; -getUsersData = data; -} -//----------------------------------------------------------------------------- -void SERVCONF::SetGetUserCallback(PARSER_GET_USER::CALLBACK f, void * data) -{ -getUserCallback = f; -getUserData = data; -} -//----------------------------------------------------------------------------- -void SERVCONF::SetAuthByCallback(PARSER_AUTH_BY::CALLBACK f, void * data) -{ -authByCallback = f; -authByData = data; -} -//----------------------------------------------------------------------------- -void SERVCONF::SetServerInfoCallback(PARSER_SERVER_INFO::CALLBACK f, void * data) -{ -serverInfoCallback = f; -serverInfoData = data; -} -//----------------------------------------------------------------------------- -void SERVCONF::SetChgUserCb(RecvChgUserCb_t f, void * data) -{ -RecvChgUserCb = f; -chgUserDataCb = data; -} -//----------------------------------------------------------------------------- -void SERVCONF::SetCheckUserCallback(PARSER_CHECK_USER::CALLBACK f, void * data) -{ -checkUserCallback = f; -checkUserData = data; -} -//----------------------------------------------------------------------------- -void SERVCONF::SetSendMessageCb(RecvSendMessageCb_t f, void * data) -{ -RecvSendMessageCb = f; -sendMessageDataCb = data; -} -//----------------------------------------------------------------------------- -const std::string & SERVCONF::GetStrError() const -{ -return errorMsg; -} -//----------------------------------------------------------------------------- -int SERVCONF::GetError() -{ -int e = error; -error = 0; -return e; -} -//-----------------------------------------------------------------------------