X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/f30167d5f98aa53c747b1e119110fb0acff2651e..73bee151e048054bde9a39c5ad2efec25af9c77b:/projects/stargazer/plugins/other/ping/ping.h diff --git a/projects/stargazer/plugins/other/ping/ping.h b/projects/stargazer/plugins/other/ping/ping.h index 8b58e7ee..b52a5e1c 100644 --- a/projects/stargazer/plugins/other/ping/ping.h +++ b/projects/stargazer/plugins/other/ping/ping.h @@ -1,148 +1,95 @@ - /* - $Revision: 1.16 $ - $Date: 2009/06/23 11:32:28 $ - $Author: faust $ - */ +#pragma once -#ifndef PING_H -#define PING_H - -#include "stg/os_int.h" #include "stg/plugin.h" #include "stg/module_settings.h" -#include "stg/notifer.h" +#include "stg/subscriptions.h" #include "stg/user_ips.h" #include "stg/pinger.h" #include "stg/users.h" #include "stg/logger.h" #include +#include +#include #include - -#include - -extern "C" PLUGIN * GetPlugin(); - -class PING; -class USER; -class SETTINGS; -//-----------------------------------------------------------------------------*/ -class CHG_CURRIP_NOTIFIER_PING: public PROPERTY_NOTIFIER_BASE { -public: - CHG_CURRIP_NOTIFIER_PING(const PING & p, USER_PTR u) - : user(u), ping(p) {} - void Notify(const uint32_t & oldIP, const uint32_t & newIP); - USER_PTR GetUser() const { return user; } - -private: - CHG_CURRIP_NOTIFIER_PING & operator=(const CHG_CURRIP_NOTIFIER_PING &); - - USER_PTR user; - const PING & ping; -}; -//----------------------------------------------------------------------------- -class CHG_IPS_NOTIFIER_PING: public PROPERTY_NOTIFIER_BASE { -public: - CHG_IPS_NOTIFIER_PING(const PING & p, USER_PTR u) - : user(u), ping(p) {} - void Notify(const USER_IPS & oldIPS, const USER_IPS & newIPS); - USER_PTR GetUser() const { return user; } - -private: - CHG_IPS_NOTIFIER_PING & operator=(const CHG_IPS_NOTIFIER_PING &); - - USER_PTR user; - const PING & ping; -}; +#include +#pragma GCC diagnostic push +#pragma GCC diagnostic ignored "-Wshadow" +#include +#pragma GCC diagnostic pop +#include + +namespace STG +{ +struct USER; +struct SETTINGS; + +using UserPtr = User*; //----------------------------------------------------------------------------- -class ADD_USER_NONIFIER_PING: public NOTIFIER_BASE { -public: - explicit ADD_USER_NONIFIER_PING(PING & p) : ping(p) {} - void Notify(const USER_PTR & user); - -private: - ADD_USER_NONIFIER_PING(const ADD_USER_NONIFIER_PING &); - ADD_USER_NONIFIER_PING & operator=(const ADD_USER_NONIFIER_PING &); - - PING & ping; +class PING_SETTINGS +{ + public: + PING_SETTINGS() : pingDelay(0) {} + const std::string & GetStrError() const { return errorStr; } + int ParseSettings(const ModuleSettings & s); + int GetPingDelay() const { return pingDelay; } + private: + int pingDelay; + std::string errorStr; }; //----------------------------------------------------------------------------- -class DEL_USER_NONIFIER_PING: public NOTIFIER_BASE { -public: - explicit DEL_USER_NONIFIER_PING(PING & p) : ping(p) {} - void Notify(const USER_PTR & user); - -private: - DEL_USER_NONIFIER_PING(const DEL_USER_NONIFIER_PING &); - DEL_USER_NONIFIER_PING & operator=(const DEL_USER_NONIFIER_PING &); - - PING & ping; +class PING : public Plugin +{ + public: + PING(); + + void SetUsers(Users * u) override { users = u; } + void SetSettings(const ModuleSettings & s) override { settings = s; } + int ParseSettings() override; + + int Start() override; + int Stop() override; + int Reload(const ModuleSettings & /*ms*/) override { return 0; } + bool IsRunning() override; + + const std::string & GetStrError() const override { return errorStr; } + std::string GetVersion() const override { return "Pinger v.1.01"; } + uint16_t GetStartPosition() const override { return 10; } + uint16_t GetStopPosition() const override { return 10; } + + void AddUser(UserPtr u); + void DelUser(UserPtr u); + + private: + explicit PING(const PING & rvalue); + PING & operator=(const PING & rvalue); + + void GetUsers(); + void SetUserNotifiers(UserPtr u); + void UnSetUserNotifiers(UserPtr u); + void Run(std::stop_token token); + + std::string errorStr; + PING_SETTINGS pingSettings; + ModuleSettings settings; + Users * users; + std::list usersList; + + std::jthread m_thread; + std::mutex m_mutex; + bool isRunning; + STG_PINGER m_pinger; + + void updateCurrIP(uint32_t oldVal, uint32_t newVal); + void updateIPs(const UserIPs& oldVal, const UserIPs& newVal); + + ScopedConnection m_onAddUserConn; + ScopedConnection m_onDelUserConn; + + using ConnHolder = std::tuple; + std::vector m_conns; + + PluginLogger logger; }; -//----------------------------------------------------------------------------- -class PING_SETTINGS { -public: - PING_SETTINGS() : pingDelay(0) {} - const std::string & GetStrError() const { return errorStr; } - int ParseSettings(const MODULE_SETTINGS & s); - int GetPingDelay() const { return pingDelay; } -private: - int pingDelay; - mutable std::string errorStr; -}; -//----------------------------------------------------------------------------- -class PING : public PLUGIN { -friend class CHG_CURRIP_NOTIFIER_PING; -friend class CHG_IPS_NOTIFIER_PING; -public: - PING(); - virtual ~PING(); - - void SetUsers(USERS * u) { users = u; } - void SetSettings(const MODULE_SETTINGS & s) { settings = s; } - int ParseSettings(); - - int Start(); - int Stop(); - int Reload(const MODULE_SETTINGS & /*ms*/) { return 0; } - bool IsRunning(); - - const std::string & GetStrError() const { return errorStr; } - std::string GetVersion() const { return "Pinger v.1.01"; } - uint16_t GetStartPosition() const { return 10; } - uint16_t GetStopPosition() const { return 10; } - - void AddUser(USER_PTR u); - void DelUser(USER_PTR u); - -private: - explicit PING(const PING & rvalue); - PING & operator=(const PING & rvalue); - - void GetUsers(); - void SetUserNotifiers(USER_PTR u); - void UnSetUserNotifiers(USER_PTR u); - static void * Run(void * d); - - mutable std::string errorStr; - PING_SETTINGS pingSettings; - MODULE_SETTINGS settings; - USERS * users; - std::list usersList; - - pthread_t thread; - pthread_mutex_t mutex; - bool nonstop; - bool isRunning; - mutable STG_PINGER pinger; - - std::list ChgCurrIPNotifierList; - std::list ChgIPNotifierList; - - ADD_USER_NONIFIER_PING onAddUserNotifier; - DEL_USER_NONIFIER_PING onDelUserNotifier; - - PLUGIN_LOGGER logger; -}; -//----------------------------------------------------------------------------- -#endif +}