X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/8569ecdc2c9368dc0fe650b901cce7b37337ffec..ebd170a764ab9660adee464588cda1801c7986b4:/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 29e4974e..b52a5e1c 100644 --- a/projects/stargazer/plugins/other/ping/ping.h +++ b/projects/stargazer/plugins/other/ping/ping.h @@ -1,157 +1,95 @@ - /* - $Revision: 1.16 $ - $Date: 2009/06/23 11:32:28 $ - $Author: faust $ - */ +#pragma once -#ifndef PING_H -#define PING_H - -#include - -#include -#include - -#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" -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) - : PROPERTY_NOTIFIER_BASE(), user(u), ping(p) {} - CHG_CURRIP_NOTIFIER_PING(const CHG_CURRIP_NOTIFIER_PING & rvalue) - : PROPERTY_NOTIFIER_BASE(), - user(rvalue.user), ping(rvalue.ping) {} - 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 & rvalue); - - 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) - : PROPERTY_NOTIFIER_BASE(), user(u), ping(p) {} - CHG_IPS_NOTIFIER_PING(const CHG_IPS_NOTIFIER_PING & rvalue) - : PROPERTY_NOTIFIER_BASE(), - user(rvalue.user), ping(rvalue.ping) {} - 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 & rvalue); - - USER_PTR user; - const PING & ping; -}; -//----------------------------------------------------------------------------- -class ADD_USER_NONIFIER_PING: public NOTIFIER_BASE { -public: - ADD_USER_NONIFIER_PING(PING & p) : NOTIFIER_BASE(), ping(p) {} - virtual ~ADD_USER_NONIFIER_PING() {} - void Notify(const USER_PTR & user); - -private: - ADD_USER_NONIFIER_PING(const ADD_USER_NONIFIER_PING & rvalue); - ADD_USER_NONIFIER_PING & operator=(const ADD_USER_NONIFIER_PING & rvalue); - - PING & ping; -}; -//----------------------------------------------------------------------------- -class DEL_USER_NONIFIER_PING: public NOTIFIER_BASE { -public: - DEL_USER_NONIFIER_PING(PING & p) : NOTIFIER_BASE(), ping(p) {} - virtual ~DEL_USER_NONIFIER_PING() {} - void Notify(const USER_PTR & user); - -private: - DEL_USER_NONIFIER_PING(const DEL_USER_NONIFIER_PING & rvalue); - DEL_USER_NONIFIER_PING & operator=(const DEL_USER_NONIFIER_PING & rvalue); - - PING & ping; -}; +#include +#include +#include +#include +#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 PING_SETTINGS { -public: - PING_SETTINGS() : pingDelay(0), errorStr() {} - virtual ~PING_SETTINGS() {} - 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_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 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() { 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: - 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; +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; }; -//----------------------------------------------------------------------------- -#endif +}