X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/5b474acc0450b4d2db9ebb50e279eef038b48778..96f8ba7ed42706e7d875ffd0dac3f62ec4c43ba5:/projects/stargazer/plugins/authorization/inetaccess/inetaccess.h?ds=sidebyside diff --git a/projects/stargazer/plugins/authorization/inetaccess/inetaccess.h b/projects/stargazer/plugins/authorization/inetaccess/inetaccess.h index 92cd29fe..e16c0257 100644 --- a/projects/stargazer/plugins/authorization/inetaccess/inetaccess.h +++ b/projects/stargazer/plugins/authorization/inetaccess/inetaccess.h @@ -34,24 +34,23 @@ #include #include #include +#include #include #include -#include "os_int.h" -#include "base_auth.h" -#include "base_store.h" -#include "notifer.h" -#include "user_ips.h" -#include "../../../user.h" -#include "../../../users.h" -#include "ia_packets.h" -#include "blowfish.h" -#include "stg_logger.h" -#include "utime.h" +#include "stg/os_int.h" +#include "stg/auth.h" +#include "stg/store.h" +#include "stg/notifer.h" +#include "stg/user_ips.h" +#include "stg/user.h" +#include "stg/users.h" +#include "stg/ia_packets.h" +#include "stg/blowfish.h" +#include "stg/logger.h" +#include "stg/utime.h" -using namespace std; - -extern "C" BASE_PLUGIN * GetPlugin(); +extern "C" PLUGIN * GetPlugin(); #define IA_PROTO_VER (6) @@ -60,8 +59,7 @@ extern "C" BASE_PLUGIN * GetPlugin(); class AUTH_IA; //----------------------------------------------------------------------------- -enum FREEMB -{ +enum FREEMB { freeMb0 = 0, freeMb1, freeMb2, @@ -86,8 +84,7 @@ enum FREEMB freeMbNone = 101 }; //----------------------------------------------------------------------------- -class IA_PHASE -{ +class IA_PHASE { public: IA_PHASE(); ~IA_PHASE(); @@ -103,8 +100,8 @@ public: const UTIME & GetTime() const; #ifdef IA_PHASE_DEBUG - void SetUserLogin(const string & login); - void SetLogFileName(const string & logFileName); + void SetUserLogin(const std::string & login); + void SetLogFileName(const std::string & logFileName); #endif private: @@ -113,16 +110,16 @@ private: #ifdef IA_PHASE_DEBUG void WritePhaseChange(int newPhase); - string log; - string login; + std::string log; + std::string login; FILE * flog; #endif }; //----------------------------------------------------------------------------- -struct IA_USER -{ +struct IA_USER { IA_USER() - : lastSendAlive(0), + : user(NULL), + lastSendAlive(0), rnd(random()), port(0), protoVer(0), @@ -155,28 +152,25 @@ struct IA_USER memcpy(&ctx, &u.ctx, sizeof(BLOWFISH_CTX)); }; - user_iter user; - //int phase; - //UTIME phaseTime; + USER_PTR user; IA_PHASE phase; UTIME lastSendAlive; uint32_t rnd; uint16_t port; BLOWFISH_CTX ctx; - list messagesToSend; + std::list messagesToSend; int protoVer; - string password; + std::string password; #ifdef IA_DEBUG bool aliveSent; #endif }; //----------------------------------------------------------------------------- -class AUTH_IA_SETTINGS -{ +class AUTH_IA_SETTINGS { public: AUTH_IA_SETTINGS(); virtual ~AUTH_IA_SETTINGS() {}; - const string& GetStrError() const { return errorStr; }; + const std::string & GetStrError() const { return errorStr; }; int ParseSettings(const MODULE_SETTINGS & s); int GetUserDelay() const { return userDelay; }; int GetUserTimeout() const { return userTimeout; }; @@ -184,38 +178,50 @@ public: FREEMB GetFreeMbShowType() const { return freeMbShowType; }; private: - int ParseIntInRange(const string & str, int min, int max, int * val); + int ParseIntInRange(const std::string & str, int min, int max, int * val); int userDelay; int userTimeout; uint16_t port; - string errorStr; + std::string errorStr; FREEMB freeMbShowType; }; //----------------------------------------------------------------------------- -class AUTH_IA :public BASE_AUTH -{ +class AUTH_IA; +//----------------------------------------------------------------------------- +class DEL_USER_NOTIFIER: public NOTIFIER_BASE { +public: + DEL_USER_NOTIFIER(AUTH_IA & a) : auth(a) {} + virtual ~DEL_USER_NOTIFIER() {} + + void Notify(const USER_PTR & user); +private: + AUTH_IA & auth; +}; +//----------------------------------------------------------------------------- +class AUTH_IA :public AUTH { +friend class DEL_USER_NOTIFIER; public: AUTH_IA(); virtual ~AUTH_IA(); - void SetUsers(USERS * u) { users = u; }; - void SetTariffs(TARIFFS *) {}; - void SetAdmins(ADMINS *) {}; - void SetTraffcounter(TRAFFCOUNTER *) {}; - void SetStore(BASE_STORE *) {}; - void SetStgSettings(const SETTINGS * s) { stgSettings = s; }; - void SetSettings(const MODULE_SETTINGS & s) { settings = s; }; + void SetUsers(USERS * u) { users = u; } + void SetTariffs(TARIFFS *) {} + void SetAdmins(ADMINS *) {} + void SetTraffcounter(TRAFFCOUNTER *) {} + void SetStore(STORE *) {} + void SetStgSettings(const SETTINGS * s) { stgSettings = s; } + void SetSettings(const MODULE_SETTINGS & s) { settings = s; } int ParseSettings(); int Start(); int Stop(); - int Reload() { return 0; }; - bool IsRunning() { return isRunningRunTimeouter || isRunningRun; }; + int Reload() { return 0; } + bool IsRunning() { return isRunningRunTimeouter || isRunningRun; } - const string & GetStrError() const { return errorStr; }; - const string GetVersion() const { return "InetAccess authorization plugin v.1.4"; }; - uint16_t GetStartPosition() const { return 50; }; - uint16_t GetStopPosition() const { return 50; }; + const std::string & GetStrError() const { return errorStr; } + const std::string GetVersion() const { return "InetAccess authorization plugin v.1.4"; } + uint16_t GetStartPosition() const { return 50; } + uint16_t GetStopPosition() const { return 50; } int SendMessage(const STG_MSG & msg, uint32_t ip) const; @@ -224,46 +230,43 @@ private: static void * RunTimeouter(void * d); int PrepareNet(); int FinalizeNet(); - void DelUser(user_iter u); + void DelUser(USER_PTR u); int RecvData(char * buffer, int bufferSize); int CheckHeader(const char * buffer, int * protoVer); - int PacketProcessor(char * buff, int dataLen, uint32_t sip, uint16_t sport, int protoVer, user_iter * user); + int PacketProcessor(char * buff, int dataLen, uint32_t sip, uint16_t sport, int protoVer, USER_PTR * user); - int Process_CONN_SYN_6(CONN_SYN_6 * connSyn, IA_USER * iaUser, user_iter * user, uint32_t sip); - int Process_CONN_SYN_7(CONN_SYN_7 * connSyn, IA_USER * iaUser, user_iter * user, uint32_t sip); - int Process_CONN_SYN_8(CONN_SYN_8 * connSyn, IA_USER * iaUser, user_iter * user, uint32_t sip); + int Process_CONN_SYN_6(CONN_SYN_6 * connSyn, IA_USER * iaUser, uint32_t sip); + int Process_CONN_SYN_7(CONN_SYN_7 * connSyn, IA_USER * iaUser, uint32_t sip); + int Process_CONN_SYN_8(CONN_SYN_8 * connSyn, IA_USER * iaUser, uint32_t sip); - int Process_CONN_ACK_6(CONN_ACK_6 * connAck, IA_USER * iaUser, user_iter * user, uint32_t sip); - int Process_CONN_ACK_7(CONN_ACK_7 * connAck, IA_USER * iaUser, user_iter * user, uint32_t sip); - int Process_CONN_ACK_8(CONN_ACK_8 * connAck, IA_USER * iaUser, user_iter * user, uint32_t sip); + int Process_CONN_ACK_6(CONN_ACK_6 * connAck, IA_USER * iaUser, uint32_t sip); + int Process_CONN_ACK_7(CONN_ACK_7 * connAck, IA_USER * iaUser, uint32_t sip); + int Process_CONN_ACK_8(CONN_ACK_8 * connAck, IA_USER * iaUser, uint32_t sip); - int Process_ALIVE_ACK_6(ALIVE_ACK_6 * aliveAck, IA_USER * iaUser, user_iter * user, uint32_t sip); - int Process_ALIVE_ACK_7(ALIVE_ACK_7 * aliveAck, IA_USER * iaUser, user_iter * user, uint32_t sip); - int Process_ALIVE_ACK_8(ALIVE_ACK_8 * aliveAck, IA_USER * iaUser, user_iter * user, uint32_t sip); + int Process_ALIVE_ACK_6(ALIVE_ACK_6 * aliveAck, IA_USER * iaUser, uint32_t sip); + int Process_ALIVE_ACK_7(ALIVE_ACK_7 * aliveAck, IA_USER * iaUser, uint32_t sip); + int Process_ALIVE_ACK_8(ALIVE_ACK_8 * aliveAck, IA_USER * iaUser, uint32_t sip); - int Process_DISCONN_SYN_6(DISCONN_SYN_6 * disconnSyn, IA_USER * iaUser, user_iter * user, uint32_t sip); - int Process_DISCONN_SYN_7(DISCONN_SYN_7 * disconnSyn, IA_USER * iaUser, user_iter * user, uint32_t sip); - int Process_DISCONN_SYN_8(DISCONN_SYN_8 * disconnSyn, IA_USER * iaUser, user_iter * user, uint32_t sip); + int Process_DISCONN_SYN_6(DISCONN_SYN_6 * disconnSyn, IA_USER * iaUser, uint32_t sip); + int Process_DISCONN_SYN_7(DISCONN_SYN_7 * disconnSyn, IA_USER * iaUser, uint32_t sip); + int Process_DISCONN_SYN_8(DISCONN_SYN_8 * disconnSyn, IA_USER * iaUser, uint32_t sip); int Process_DISCONN_ACK_6(DISCONN_ACK_6 * disconnSyn, IA_USER * iaUser, - user_iter * user, uint32_t sip, map::iterator it); int Process_DISCONN_ACK_7(DISCONN_ACK_7 * disconnSyn, IA_USER * iaUser, - user_iter * user, uint32_t sip, map::iterator it); int Process_DISCONN_ACK_8(DISCONN_ACK_8 * disconnSyn, IA_USER * iaUser, - user_iter * user, uint32_t sip, map::iterator it); - int Send_CONN_SYN_ACK_6(IA_USER * iaUser, user_iter * user, uint32_t sip); - int Send_CONN_SYN_ACK_7(IA_USER * iaUser, user_iter * user, uint32_t sip); - int Send_CONN_SYN_ACK_8(IA_USER * iaUser, user_iter * user, uint32_t sip); + int Send_CONN_SYN_ACK_6(IA_USER * iaUser, uint32_t sip); + int Send_CONN_SYN_ACK_7(IA_USER * iaUser, uint32_t sip); + int Send_CONN_SYN_ACK_8(IA_USER * iaUser, uint32_t sip); int Send_ALIVE_SYN_6(IA_USER * iaUser, uint32_t sip); int Send_ALIVE_SYN_7(IA_USER * iaUser, uint32_t sip); @@ -279,7 +282,7 @@ private: int Timeouter(); - int SendError(uint32_t ip, uint16_t port, int protoVer, const string & text); + int SendError(uint32_t ip, uint16_t port, int protoVer, const std::string & text); int Send(uint32_t ip, uint16_t port, const char * buffer, int len); int RealSendMessage6(const STG_MSG & msg, uint32_t ip, IA_USER & user); int RealSendMessage7(const STG_MSG & msg, uint32_t ip, IA_USER & user); @@ -289,7 +292,7 @@ private: BLOWFISH_CTX ctxS; //for loginS - mutable string errorStr; + mutable std::string errorStr; AUTH_IA_SETTINGS iaSettings; MODULE_SETTINGS settings; @@ -301,7 +304,7 @@ private: USERS * users; const SETTINGS * stgSettings; - mutable map ip2user; + mutable std::map ip2user; pthread_t recvThread; pthread_t timeouterThread; @@ -320,30 +323,17 @@ private: FIN_6 fin6; FIN_8 fin8; - map packetTypes; + std::map packetTypes; STG_LOGGER & WriteServLog; uint32_t enabledDirs; - class DEL_USER_NONIFIER: public NOTIFIER_BASE - { - public: - DEL_USER_NONIFIER(AUTH_IA & a) : auth(a) {}; - virtual ~DEL_USER_NONIFIER(){}; - - void Notify(const user_iter & user) - { - auth.DelUser(user); - } - - private: - AUTH_IA & auth; - } onDelUserNotifier; + DEL_USER_NOTIFIER onDelUserNotifier; class UnauthorizeUser : std::unary_function &, void> { public: - UnauthorizeUser(AUTH_IA * a) : auth(a) {}; + UnauthorizeUser(AUTH_IA * a) : auth(a) {} void operator()(const std::pair & p) { p.second.user->Unauthorize(auth); @@ -354,7 +344,10 @@ private: }; //----------------------------------------------------------------------------- +inline +void DEL_USER_NOTIFIER::Notify(const USER_PTR & user) +{ + auth.DelUser(user); +} #endif - -