X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/d230cb2411b6d2d6603a930f165bbd3138831eeb..90b793eb07dc190eeb1144a1da079de760d17c36:/projects/stargazer/plugins/configuration/sgconfig/configproto.h?ds=inline diff --git a/projects/stargazer/plugins/configuration/sgconfig/configproto.h b/projects/stargazer/plugins/configuration/sgconfig/configproto.h index 4b93d675..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 +#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; + +} -#include "stg/module_settings.h" -#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; - -//----------------------------------------------------------------------------- class CONFIGPROTO { public: - CONFIGPROTO(PLUGIN_LOGGER & l); + explicit CONFIGPROTO(STG::PluginLogger & l); ~CONFIGPROTO(); - void SetPort(uint16_t port); - 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, const std::string & answer); - int SendError(int sock, const std::string & text); - void WriteLogAccessFailed(uint32_t ip); - - int ParseCommand(); - - 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