X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/52b4afb2c72468a55fb49aeed3f7dda1277319df..355911106bc27f5504eb904a064367ef1b5f0aba:/projects/stargazer/plugins/other/smux/smux.h?ds=sidebyside diff --git a/projects/stargazer/plugins/other/smux/smux.h b/projects/stargazer/plugins/other/smux/smux.h index c5b94d4c..7337deaf 100644 --- a/projects/stargazer/plugins/other/smux/smux.h +++ b/projects/stargazer/plugins/other/smux/smux.h @@ -5,23 +5,31 @@ #include #include +#include -#include "asn1/SMUX-PDUs.h" -#include "asn1/ObjectSyntax.h" +#include "stg/SMUX-PDUs.h" +#include "stg/ObjectSyntax.h" #include "stg/os_int.h" #include "stg/plugin.h" #include "stg/module_settings.h" -#include "stg/users.h" -#include "stg/tariffs.h" +#include "stg/notifer.h" +#include "stg/noncopyable.h" #include "sensors.h" +#include "tables.h" +#include "types.h" extern "C" PLUGIN * GetPlugin(); class USER; class SETTINGS; class SMUX; +class USERS; +class TARIFFS; +class SERVICES; +class CORPORATIONS; +class TRAFFCOUNTER; typedef bool (SMUX::*SMUXPacketHandler)(const SMUX_PDUs_t * pdus); typedef bool (SMUX::*PDUsHandler)(const PDUs_t * pdus); @@ -47,6 +55,47 @@ private: std::string password; }; //----------------------------------------------------------------------------- +class CHG_AFTER_NOTIFIER : public PROPERTY_NOTIFIER_BASE { +public: + CHG_AFTER_NOTIFIER(SMUX & s, const USER_PTR & u) : smux(s), userPtr(u) {} + CHG_AFTER_NOTIFIER(const CHG_AFTER_NOTIFIER & rvalue) : smux(rvalue.smux), userPtr(rvalue.userPtr) {} + void Notify(const std::string &, const std::string &); + + USER_PTR GetUserPtr() { return userPtr; } + +private: + CHG_AFTER_NOTIFIER & operator=(const CHG_AFTER_NOTIFIER & rvalue); + SMUX & smux; + USER_PTR userPtr; +}; +//----------------------------------------------------------------------------- +class ADD_DEL_TARIFF_NOTIFIER : public NOTIFIER_BASE, private NONCOPYABLE { +public: + ADD_DEL_TARIFF_NOTIFIER(SMUX & s) : smux(s) {} + void Notify(const TARIFF_DATA &); + +private: + SMUX & smux; +}; +//----------------------------------------------------------------------------- +class ADD_USER_NOTIFIER : public NOTIFIER_BASE, private NONCOPYABLE { +public: + ADD_USER_NOTIFIER(SMUX & s) : smux(s) {} + void Notify(const USER_PTR &); + +private: + SMUX & smux; +}; +//----------------------------------------------------------------------------- +class DEL_USER_NOTIFIER : public NOTIFIER_BASE, private NONCOPYABLE { +public: + DEL_USER_NOTIFIER(SMUX & s) : smux(s) {} + void Notify(const USER_PTR &); + +private: + SMUX & smux; +}; +//----------------------------------------------------------------------------- class SMUX : public PLUGIN { public: SMUX(); @@ -54,24 +103,32 @@ public: void SetUsers(USERS * u) { users = u; } void SetTariffs(TARIFFS * t) { tariffs = t; } - void SetAdmins(ADMINS *) {} - void SetTraffcounter(TRAFFCOUNTER *) {} - void SetStore(STORE *) {} - void SetStgSettings(const SETTINGS *) {} + void SetAdmins(ADMINS * a) { admins = a; } + void SetServices(SERVICES * s) { services = s; } + void SetTraffcounter(TRAFFCOUNTER * tc) { traffcounter = tc; } + void SetCorporations(CORPORATIONS * c) { corporations = c; } void SetSettings(const MODULE_SETTINGS & s) { settings = s; } int ParseSettings(); int Start(); int Stop(); - int Reload() { return 0; } + int Reload(); bool IsRunning() { return running && !stopped; } const std::string & GetStrError() const { return errorStr; } - const std::string GetVersion() const { return "Stg SMUX Plugin 1.0"; } + const std::string GetVersion() const { return "Stg SMUX Plugin 1.1"; } uint16_t GetStartPosition() const { return 100; } uint16_t GetStopPosition() const { return 100; } + bool UpdateTables(); + + void SetNotifier(USER_PTR userPtr); + void UnsetNotifier(USER_PTR userPtr); + private: + SMUX(const SMUX & rvalue); + SMUX & operator=(const SMUX & rvalue); + static void * Runner(void * d); void Run(); bool PrepareNet(); @@ -80,15 +137,22 @@ private: bool CloseHandler(const SMUX_PDUs_t * pdus); bool RegisterResponseHandler(const SMUX_PDUs_t * pdus); - bool PDUsHandler(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(); + USERS * users; TARIFFS * tariffs; + ADMINS * admins; + SERVICES * services; + CORPORATIONS * corporations; + TRAFFCOUNTER * traffcounter; mutable std::string errorStr; SMUX_SETTINGS smuxSettings; @@ -104,10 +168,35 @@ private: SMUXHandlers smuxHandlers; PDUsHandlers pdusHandlers; Sensors sensors; + Tables tables; + std::list notifiers; + ADD_USER_NOTIFIER addUserNotifier; + DEL_USER_NOTIFIER delUserNotifier; + ADD_DEL_TARIFF_NOTIFIER addDelTariffNotifier; }; //----------------------------------------------------------------------------- +inline +void ADD_DEL_TARIFF_NOTIFIER::Notify(const TARIFF_DATA &) +{ +smux.UpdateTables(); +} + +inline +void ADD_USER_NOTIFIER::Notify(const USER_PTR & userPtr) +{ +smux.SetNotifier(userPtr); +smux.UpdateTables(); +} + +inline +void DEL_USER_NOTIFIER::Notify(const USER_PTR & userPtr) +{ +smux.UnsetNotifier(userPtr); +smux.UpdateTables(); +} + extern "C" PLUGIN * GetPlugin(); #endif