X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/ede91934442fd804d7b818971a44e3ad795cb01f..8c6fa3fbaccc22127280bf77a48fab5a3ee0716e:/projects/stargazer/plugins/other/radius/radius.h diff --git a/projects/stargazer/plugins/other/radius/radius.h b/projects/stargazer/plugins/other/radius/radius.h index 26f70aa4..52da138e 100644 --- a/projects/stargazer/plugins/other/radius/radius.h +++ b/projects/stargazer/plugins/other/radius/radius.h @@ -18,160 +18,93 @@ * Author : Maxim Mamontov */ -/* - * Radius data access plugin for Stargazer - * - * $Revision: 1.10 $ - * $Date: 2009/12/13 14:17:13 $ - * - */ +#ifndef __STG_RADIUS_H__ +#define __STG_RADIUS_H__ -#ifndef RADIUS_H -#define RADIUS_H +#include "stg/os_int.h" +#include "stg/auth.h" +#include "stg/module_settings.h" +#include "stg/logger.h" -#include +#include "config.h" +#include "conn.h" -#include -#include #include -#include -#include -#include - -#include "os_int.h" -#include "auth.h" -#include "module_settings.h" -#include "notifer.h" -#include "user_ips.h" -#include "user.h" -#include "users.h" -#include "blowfish.h" -#include "rad_packets.h" +#include +#include + +#include +#include +#include +#include extern "C" PLUGIN * GetPlugin(); -#define RAD_DEBUG (1) +class STORE; +class USERS; -class RADIUS; -//----------------------------------------------------------------------------- -class RAD_SETTINGS { +class RADIUS : public AUTH { public: - RAD_SETTINGS() : port(0) {} - virtual ~RAD_SETTINGS() {} - const string & GetStrError() const { return errorStr; } - int ParseSettings(const MODULE_SETTINGS & s); - uint16_t GetPort() const { return port; } - const std::string & GetPassword() const { return password; } - const std::list & GetAuthServices() const { return authServices; } - const std::list & GetAcctServices() const { return acctServices; } + RADIUS(); + virtual ~RADIUS() {} -private: - int ParseIntInRange(const std::string & str, int min, int max, int * val); - int ParseServices(const std::vector & str, std::list * lst); - - uint16_t port; - std::string errorStr; - std::string password; - std::list authServices; - std::list acctServices; -}; -//----------------------------------------------------------------------------- -struct RAD_SESSION { - std::string userName; - std::string serviceType; -}; -//----------------------------------------------------------------------------- -class RADIUS :public AUTH { -public: - RADIUS(); - virtual ~RADIUS() {}; - - void SetUsers(USERS * u); - void SetTariffs(TARIFFS *) {} - void SetAdmins(ADMINS *) {} - void SetTraffcounter(TRAFFCOUNTER *) {} - void SetStore(STORE * ); - void SetStgSettings(const SETTINGS * s); - void SetSettings(const MODULE_SETTINGS & s); - int ParseSettings(); - - int Start(); - int Stop(); - int Reload() { return 0; } - bool IsRunning(); - - const std::string & GetStrError() const { return errorStr; } - const std::string GetVersion() const; - uint16_t GetStartPosition() const; - uint16_t GetStopPosition() const; - - int SendMessage(const STG_MSG &, uint32_t) const { return 0; } + void SetUsers(USERS* u) { m_users = u; } + void SetStore(STORE* s) { m_store = s; } + void SetStgSettings(const SETTINGS*) {} + void SetSettings(const MODULE_SETTINGS& s) { m_settings = s; } + int ParseSettings(); + + int Start(); + int Stop(); + int Reload(const MODULE_SETTINGS & /*ms*/) { return 0; } + bool IsRunning() { return m_running; } + + const std::string& GetStrError() const { return m_error; } + std::string GetVersion() const { return "RADIUS data access plugin v. 2.0"; } + uint16_t GetStartPosition() const { return 30; } + uint16_t GetStopPosition() const { return 30; } + + int SendMessage(const STG_MSG&, uint32_t) const { return 0; } + + void authorize(const USER& user); + void unauthorize(const std::string& login, const std::string& reason); private: - static void * Run(void *); - int PrepareNet(); - int FinalizeNet(); - - int Send(const RAD_PACKET & packet, struct sockaddr_in * outerAddr); - int RecvData(RAD_PACKET * packet, struct sockaddr_in * outerAddr); - int ProcessData(RAD_PACKET * packet); - - int ProcessAutzPacket(RAD_PACKET * packet); - int ProcessAuthPacket(RAD_PACKET * packet); - int ProcessPostAuthPacket(RAD_PACKET * packet); - int ProcessAcctStartPacket(RAD_PACKET * packet); - int ProcessAcctStopPacket(RAD_PACKET * packet); - int ProcessAcctUpdatePacket(RAD_PACKET * packet); - int ProcessAcctOtherPacket(RAD_PACKET * packet); - - bool FindUser(USER_PTR * ui, const std::string & login) const; - bool CanAuthService(const std::string & svc) const; - bool CanAcctService(const std::string & svc) const; - bool IsAllowedService(const std::string & svc) const; - - bool WaitPackets(int sd) const; - - void PrintServices(const std::list & svcs); - - struct Printer : public unary_function - { - void operator()(const std::string & line) - { - printfd("radius.cpp", "'%s'\n", line.c_str()); - }; - }; - struct SPrinter : public unary_function, void> - { - void operator()(const std::pair & it) - { - printfd("radius.cpp", "%s - ('%s', '%s')\n", it.first.c_str(), it.second.userName.c_str(), it.second.serviceType.c_str()); - }; - }; - - BLOWFISH_CTX ctx; - - mutable std::string errorStr; - RAD_SETTINGS radSettings; - MODULE_SETTINGS settings; - std::list authServices; - std::list acctServices; - std::map sessions; - - bool nonstop; - bool isRunning; - - USERS * users; - const SETTINGS * stgSettings; - const STORE * store; - - pthread_t thread; - pthread_mutex_t mutex; - - int sock; - - RAD_PACKET packet; + RADIUS(const RADIUS & rvalue); + RADIUS & operator=(const RADIUS & rvalue); + + static void* run(void*); + + bool reconnect(); + int createUNIX() const; + int createTCP() const; + void runImpl(); + int maxFD() const; + void buildFDSet(fd_set & fds) const; + void cleanupConns(); + void handleEvents(const fd_set & fds); + void acceptConnection(); + void acceptUNIX(); + void acceptTCP(); + + mutable std::string m_error; + STG::Config m_config; + + MODULE_SETTINGS m_settings; + + bool m_running; + bool m_stopped; + + USERS* m_users; + const STORE* m_store; + + int m_listenSocket; + std::deque m_conns; + std::set m_logins; + + pthread_t m_thread; + PLUGIN_LOGGER m_logger; }; -//----------------------------------------------------------------------------- #endif