X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/4271ab433cd55bbd2612292bcf39e4dc3d7274f1..0907aa4037b12b6b88ee24495d4577a064d4f8db:/stargazer/plugins/other/smux/smux.h diff --git a/stargazer/plugins/other/smux/smux.h b/stargazer/plugins/other/smux/smux.h deleted file mode 100644 index 4954bd09..00000000 --- a/stargazer/plugins/other/smux/smux.h +++ /dev/null @@ -1,223 +0,0 @@ -#ifndef __SMUX_H__ -#define __SMUX_H__ - -#include - -#include -#include -#include -#include - -#include "stg/SMUX-PDUs.h" -#include "stg/ObjectSyntax.h" - -#include "stg/plugin.h" -#include "stg/module_settings.h" -#include "stg/notifer.h" -#include "stg/noncopyable.h" -#include "stg/logger.h" - -#include "sensors.h" -#include "tables.h" -#include "types.h" - -namespace STG -{ -struct User; -struct Settings; -struct Users; -struct Tariffs; -struct Services; -struct Corporations; -struct TraffCounter; -} - -class SMUX; - -typedef bool (SMUX::*SMUXPacketHandler)(const SMUX_PDUs_t * pdus); -typedef bool (SMUX::*PDUsHandler)(const PDUs_t * pdus); -typedef std::map SMUXHandlers; -typedef std::map PDUsHandlers; - -using UserPtr = STG::User*; -//----------------------------------------------------------------------------- -class SMUX_SETTINGS { -public: - SMUX_SETTINGS(); - virtual ~SMUX_SETTINGS() {} - const std::string & GetStrError() const { return errorStr; } - int ParseSettings(const STG::ModuleSettings & s); - - uint32_t GetIP() const { return ip; } - uint16_t GetPort() const { return port; } - const std::string GetPassword() const { return password; } - -private: - mutable std::string errorStr; - - uint32_t ip; - uint16_t port; - std::string password; -}; -//----------------------------------------------------------------------------- -class CHG_AFTER_NOTIFIER : public STG::PropertyNotifierBase { -public: - CHG_AFTER_NOTIFIER(SMUX & s, const UserPtr & u) - : STG::PropertyNotifierBase(), - smux(s), userPtr(u) {} - CHG_AFTER_NOTIFIER(const CHG_AFTER_NOTIFIER & rvalue) - : STG::PropertyNotifierBase(), - smux(rvalue.smux), userPtr(rvalue.userPtr) {} - void Notify(const std::string &, const std::string &); - - UserPtr GetUserPtr() const { return userPtr; } - -private: - CHG_AFTER_NOTIFIER & operator=(const CHG_AFTER_NOTIFIER & rvalue); - SMUX & smux; - UserPtr userPtr; -}; -//----------------------------------------------------------------------------- -class ADD_DEL_TARIFF_NOTIFIER : public STG::NotifierBase { -public: - explicit ADD_DEL_TARIFF_NOTIFIER(SMUX & s) - : STG::NotifierBase(), smux(s) {} - void Notify(const STG::TariffData &); - -private: - SMUX & smux; -}; -//----------------------------------------------------------------------------- -class ADD_USER_NOTIFIER : public STG::NotifierBase { -public: - explicit ADD_USER_NOTIFIER(SMUX & s) : STG::NotifierBase(), smux(s) {} - void Notify(const UserPtr &); - -private: - SMUX & smux; -}; -//----------------------------------------------------------------------------- -class DEL_USER_NOTIFIER : public STG::NotifierBase { -public: - explicit DEL_USER_NOTIFIER(SMUX & s) : STG::NotifierBase(), smux(s) {} - void Notify(const UserPtr &); - -private: - SMUX & smux; -}; -//----------------------------------------------------------------------------- -class SMUX : public STG::Plugin { -public: - SMUX(); - virtual ~SMUX(); - - void SetUsers(STG::Users * u) { users = u; } - void SetTariffs(STG::Tariffs * t) { tariffs = t; } - void SetAdmins(STG::Admins * a) { admins = a; } - void SetServices(STG::Services * s) { services = s; } - void SetTraffcounter(STG::TraffCounter * tc) { traffcounter = tc; } - void SetCorporations(STG::Corporations * c) { corporations = c; } - void SetSettings(const STG::ModuleSettings & s) { settings = s; } - int ParseSettings(); - - int Start(); - int Stop(); - int Reload(const STG::ModuleSettings & ms); - bool IsRunning() { return running && !stopped; } - - const std::string & GetStrError() const { return errorStr; } - std::string GetVersion() const { return "Stg SMUX Plugin 1.1"; } - uint16_t GetStartPosition() const { return 10; } - uint16_t GetStopPosition() const { return 10; } - - bool UpdateTables(); - - void SetNotifier(UserPtr userPtr); - void UnsetNotifier(UserPtr userPtr); - -private: - SMUX(const SMUX & rvalue); - SMUX & operator=(const SMUX & rvalue); - - static void * Runner(void * d); - void Run(); - bool PrepareNet(); - bool Reconnect(); - - bool DispatchPDUs(const SMUX_PDUs_t * pdus); - - bool CloseHandler(const SMUX_PDUs_t * pdus); - bool RegisterResponseHandler(const SMUX_PDUs_t * pdus); - bool PDUsRequestHandler(const SMUX_PDUs_t * pdus); - bool CommitOrRollbackHandler(const SMUX_PDUs_t * pdus); - - bool GetRequestHandler(const PDUs_t * pdus); - bool GetNextRequestHandler(const PDUs_t * pdus); - bool SetRequestHandler(const PDUs_t * pdus); - - void SetNotifiers(); - void ResetNotifiers(); - - STG::Users * users; - STG::Tariffs * tariffs; - STG::Admins * admins; - STG::Services * services; - STG::Corporations * corporations; - STG::TraffCounter * traffcounter; - - mutable std::string errorStr; - SMUX_SETTINGS smuxSettings; - STG::ModuleSettings settings; - - pthread_t thread; - pthread_mutex_t mutex; - bool running; - bool stopped; - bool needReconnect; - - time_t lastReconnectTry; - unsigned reconnectTimeout; - - int sock; - - SMUXHandlers smuxHandlers; - PDUsHandlers pdusHandlers; - Sensors sensors; - Tables tables; - - std::list notifiers; - ADD_USER_NOTIFIER addUserNotifier; - DEL_USER_NOTIFIER delUserNotifier; - ADD_DEL_TARIFF_NOTIFIER addDelTariffNotifier; - - STG::PluginLogger logger; -}; -//----------------------------------------------------------------------------- - -inline -void CHG_AFTER_NOTIFIER::Notify(const std::string &, const std::string &) -{ -smux.UpdateTables(); -} - -inline -void ADD_DEL_TARIFF_NOTIFIER::Notify(const STG::TariffData &) -{ -smux.UpdateTables(); -} - -inline -void ADD_USER_NOTIFIER::Notify(const UserPtr & userPtr) -{ -smux.SetNotifier(userPtr); -smux.UpdateTables(); -} - -inline -void DEL_USER_NOTIFIER::Notify(const UserPtr & userPtr) -{ -smux.UnsetNotifier(userPtr); -smux.UpdateTables(); -} - -#endif