X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/bb0f5269e5987bcd09eaa0afc74d08dcd5dd29b7..e6c0bd53164bd30bbe038c80a370707847fc266c:/projects/stargazer/plugins/configuration/sgconfig/configproto.h diff --git a/projects/stargazer/plugins/configuration/sgconfig/configproto.h b/projects/stargazer/plugins/configuration/sgconfig/configproto.h index d73498ff..c5ddd579 100644 --- a/projects/stargazer/plugins/configuration/sgconfig/configproto.h +++ b/projects/stargazer/plugins/configuration/sgconfig/configproto.h @@ -16,104 +16,93 @@ /* * Author : Boris Mikhailenko + * Author : Maxim Mamontov */ - /* - $Revision: 1.14 $ - $Date: 2010/10/04 20:24:14 $ - $Author: faust $ - */ +#pragma once +#include "parser.h" -#ifndef CONFIGPROTO_H -#define CONFIGPROTO_H +#include "stg/module_settings.h" #include -#include -#include - -#include "stg/os_int.h" - -#include -#include - -#define STG_HEADER "SG04" -#define OK_HEADER "OKHD" -#define ERR_HEADER "ERHD" -#define OK_LOGIN "OKLG" -#define ERR_LOGIN "ERLG" -#define OK_LOGINS "OKLS" -#define ERR_LOGINS "ERLS" - -class BASE_PARSER; -class USERS; -class ADMINS; -class ADMIN; -class TARIFFS; -class PLUGIN_LOGGER; -class STORE; -class SETTINGS; - -//----------------------------------------------------------------------------- +#include +#pragma GCC diagnostic push +#pragma GCC diagnostic ignored "-Wshadow" +#include +#pragma GCC diagnostic pop +#include + +#include +#include +#include + +namespace STG +{ + +struct Settings; +struct Admins; +struct Tariffs; +struct Users; +struct Services; +struct Corporations; +struct Store; +class PluginLogger; + +class Conn; + +} + class CONFIGPROTO { public: - CONFIGPROTO(PLUGIN_LOGGER & l); + explicit CONFIGPROTO(STG::PluginLogger & l); ~CONFIGPROTO(); - void SetPort(uint16_t p) { port = p; } - void SetAdmins(ADMINS * a); - void SetUsers(USERS * u); - void SetTariffs(TARIFFS * t); - void SetStore(STORE * s); - void SetStgSettings(const SETTINGS * s); - uint32_t GetAdminIP() const { return adminIP; } - int Prepare(); - int Stop(); - const std::string & GetStrError() const { return errorStr; } - void Run(); + void SetPort(uint16_t port) { m_port = port; } + void SetBindAddress(const std::string & address) { m_bindAddress = address; } + void SetSettings(const STG::Settings * settings) { m_settings = settings; } + void SetAdmins(STG::Admins * admins) { m_admins = admins; } + void SetTariffs(STG::Tariffs * tariffs) { m_tariffs = tariffs; } + void SetUsers(STG::Users * users) { m_users = users; } + void SetStore(STG::Store * store) { m_store = store; } + void SetServices(STG::Services * services) { m_services = services; } + void SetCorporations(STG::Corporations * corporations) { m_corporations = corporations; } + + int Prepare(); + int Stop(); + const std::string & GetStrError() const { return m_errorStr; } + void Run(std::stop_token token); private: CONFIGPROTO(const CONFIGPROTO & rvalue); CONFIGPROTO & operator=(const CONFIGPROTO & rvalue); - int RecvHdr(int sock); - int RecvLogin(int sock); - int SendLoginAnswer(int sock); - int SendHdrAnswer(int sock, int err); - int RecvLoginS(int sock); - int SendLoginSAnswer(int sock, int err); - int RecvData(int sock); - int SendDataAnswer(int sock); - void SendError(const char * text); - void WriteLogAccessFailed(uint32_t ip); - - int ParseCommand(); - - std::list answerList; - std::list requestList; - uint32_t adminIP; - std::string adminLogin; - std::string adminPassword; - uint16_t port; - pthread_t thrReciveSendConf; - bool nonstop; - int state; - ADMIN * currAdmin; - PLUGIN_LOGGER & logger; + const STG::Settings * m_settings; + STG::Admins * m_admins; + STG::Tariffs * m_tariffs; + STG::Users * m_users; + STG::Services * m_services; + STG::Corporations * m_corporations; + STG::Store * m_store; - int listenSocket; + uint16_t m_port; + std::string m_bindAddress; + bool m_stopped; + STG::PluginLogger & m_logger; + int m_listenSocket; - ADMINS * admins; + std::string m_errorStr; - BASE_PARSER * currParser; - std::vector dataParser; + BASE_PARSER::REGISTRY m_registry; + std::deque m_conns; - XML_Parser xmlParser; + bool Bind(); - std::string errorStr; + void RegisterParsers(); - friend void ParseXMLStart(void *data, const char *el, const char **attr); - friend void ParseXMLEnd(void *data, const char *el); + int MaxFD() const; + void BuildFDSet(fd_set & fds) const; + void CleanupConns(); + void HandleEvents(const fd_set & fds); + void AcceptConnection(); }; -//----------------------------------------------------------------------------- -#endif //CONFIGPROTO_H