X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/e39f173d25ae650ee843e3d1c311abe9c1cb5ee9..c57f51df2406db09088196b6adf78e373c97591f:/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 52da138e..59cbf0d5 100644 --- a/projects/stargazer/plugins/other/radius/radius.h +++ b/projects/stargazer/plugins/other/radius/radius.h @@ -1,110 +1,65 @@ -/* - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License as published by - * the Free Software Foundation; either version 2 of the License, or - * (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA - */ +#pragma once -/* - * Author : Maxim Mamontov - */ - -#ifndef __STG_RADIUS_H__ -#define __STG_RADIUS_H__ - -#include "stg/os_int.h" #include "stg/auth.h" +#include "stg/plugin.h" +#include "config.h" #include "stg/module_settings.h" +#include "stg/subscriptions.h" #include "stg/logger.h" +#include "server.h" -#include "config.h" -#include "conn.h" - +#include #include -#include -#include - -#include -#include -#include -#include - -extern "C" PLUGIN * GetPlugin(); - -class STORE; -class USERS; - -class RADIUS : public AUTH { -public: - RADIUS(); - virtual ~RADIUS() {} - - 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; } +#include +#include +#include +#include //uint8_t, uint32_t - int SendMessage(const STG_MSG&, uint32_t) const { return 0; } +namespace STG +{ + class Users; - void authorize(const USER& user); - void unauthorize(const std::string& login, const std::string& reason); + class RADIUS : public Auth + { + public: + RADIUS(); + RADIUS(const RADIUS&) = delete; + RADIUS& operator=(const RADIUS&) = delete; -private: - RADIUS(const RADIUS & rvalue); - RADIUS & operator=(const RADIUS & rvalue); + void SetUsers(Users* u) override { m_users = u; } + void SetSettings(const ModuleSettings& s) override { m_settings = s; } + int ParseSettings() override; - static void* run(void*); + int Start() override; + int Stop() override; + int Reload(const ModuleSettings& /*ms*/) override { return 0; } + bool IsRunning() override; - 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(); + const std::string& GetStrError() const override { return m_errorStr; } + std::string GetVersion() const override; - mutable std::string m_error; - STG::Config m_config; + uint16_t GetStartPosition() const override { return 0; } + uint16_t GetStopPosition() const override { return 0; } - MODULE_SETTINGS m_settings; + int SendMessage(const Message& /*msg*/, uint32_t /*ip*/) const override { return 0; } - bool m_running; - bool m_stopped; + private: + std::mutex m_mutex; - USERS* m_users; - const STORE* m_store; + boost::asio::io_context m_ioContext; + void SetRunning(bool val); + int Run(std::stop_token token); - int m_listenSocket; - std::deque m_conns; - std::set m_logins; + mutable std::string m_errorStr; + Config m_config; + ModuleSettings m_settings; - pthread_t m_thread; + bool m_running; - PLUGIN_LOGGER m_logger; -}; + std::jthread m_thread; + Users* m_users; + PluginLogger m_logger; -#endif + std::unique_ptr m_server; + }; +}