X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/1347f3d1e04bedd1508589173f577673ee2c5554..df4e8a09fef4d521167151e90080da07f166bb0b:/projects/stargazer/plugins/other/rscript/rscript.h diff --git a/projects/stargazer/plugins/other/rscript/rscript.h b/projects/stargazer/plugins/other/rscript/rscript.h index 8d22b458..4fb0ac73 100644 --- a/projects/stargazer/plugins/other/rscript/rscript.h +++ b/projects/stargazer/plugins/other/rscript/rscript.h @@ -22,14 +22,6 @@ #ifndef RSCRIPT_H #define RSCRIPT_H -#include - -#include -#include -#include -#include -#include - #include "stg/plugin.h" #include "stg/module_settings.h" #include "stg/os_int.h" @@ -41,6 +33,14 @@ #include "nrmap_parser.h" +#include +#include +#include +#include +#include + +#include + extern "C" PLUGIN * GetPlugin(); #define RS_DEBUG (1) @@ -60,7 +60,7 @@ class DisconnectUser; //----------------------------------------------------------------------------- class ADD_USER_NONIFIER: public NOTIFIER_BASE { public: - ADD_USER_NONIFIER(REMOTE_SCRIPT & r) + explicit ADD_USER_NONIFIER(REMOTE_SCRIPT & r) : NOTIFIER_BASE(), rs(r) {} virtual ~ADD_USER_NONIFIER() {} void Notify(const USER_PTR & user); @@ -74,7 +74,7 @@ private: //----------------------------------------------------------------------------- class DEL_USER_NONIFIER: public NOTIFIER_BASE { public: - DEL_USER_NONIFIER(REMOTE_SCRIPT & r) + explicit DEL_USER_NONIFIER(REMOTE_SCRIPT & r) : NOTIFIER_BASE(), rs(r) {} virtual ~DEL_USER_NONIFIER() {} void Notify(const USER_PTR & user); @@ -138,8 +138,10 @@ private: //----------------------------------------------------------------------------- struct USER { USER(const std::vector & r, USER_PTR it) - : user(it), + : lastSentTime(0), + user(it), routers(r), + shortPacketsCount(0), ip(user->GetCurrIP()) {} @@ -184,7 +186,7 @@ public: int Start(); int Stop(); - int Reload(); + int Reload(const MODULE_SETTINGS & ms); bool IsRunning() { return isRunning; } const std::string & GetStrError() const { return errorStr; } @@ -213,7 +215,6 @@ private: std::vector IP2Routers(uint32_t ip); bool GetUsers(); - std::string GetUserParam(USER_PTR u, const std::string & paramName) const; void SetUserNotifiers(USER_PTR u); void UnSetUserNotifiers(USER_PTR u); @@ -257,7 +258,7 @@ private: //----------------------------------------------------------------------------- class DisconnectUser : public std::unary_function &, void> { public: - DisconnectUser(REMOTE_SCRIPT & rs) : rscript(rs) {} + explicit DisconnectUser(REMOTE_SCRIPT & rs) : rscript(rs) {} void operator()(std::pair & p) { rscript.Send(p.second, true);