Radiusю (#12)

[stg.git] / 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 4cc84e264004cd28a53c5a0777b3a9dc4fc05190..80e032a8c90b09c5a64715729744ec2ceb310c51 100644 (file)
--- a/projects/stargazer/plugins/other/radius/radius.h
+++ b/projects/stargazer/plugins/other/radius/radius.h
@@ -1,30 +1,114 @@
 #pragma once
 
 #include "stg/auth.h"
+#include "stg/plugin.h"
+#include "stg/module_settings.h"
+#include "stg/subscriptions.h"
+#include "stg/logger.h"
+#include "server.h"
+
+#include <boost/asio.hpp>
 #include <string>
+#include <memory>
+#include <mutex>
+#include <jthread.hpp>
+#include <cstdint> //uint8_t, uint32_t
 
 namespace STG
 {
+    struct Settings;
+
+    class Users;
+
+    class RAD_SETTINGS
+    {
+        public:
+            RAD_SETTINGS();
+            virtual ~RAD_SETTINGS() {}
+
+            struct AttrValue
+            {
+                enum class Type
+                {
+                    PARAM_NAME,
+                    VALUE
+                };
+                std::string value;
+                Type type;
+            };
+
+            struct ASection
+            {
+                using Pairs = std::vector<std::pair<std::string, AttrValue>>;
+                Pairs match;
+                Pairs send;
+            };
+
+            const std::string& GetStrError() const { return m_errorStr; }
+            int ParseSettings(const ModuleSettings& s);
+
+            uint16_t GetPort() const { return m_port; }
+            const std::string& GetDictionaries() const { return m_dictionaries; }
+            const std::string& GetSecret() const { return m_secret; }
+            const ASection& getAuth() const { return m_auth; }
+            const ASection& getAutz() const { return m_autz; }
+
+        private:
+            std::vector<std::pair<std::string, AttrValue>> ParseRules(const std::string& value, const std::string& paramName);
+            ASection parseASection(const std::vector<ParamValue>& conf);
+
+            std::string m_errorStr;
+            uint16_t m_port;
+            std::string m_dictionaries;
+            std::string m_secret;
+
+            ASection m_auth;
+            ASection m_autz;
+
+            PluginLogger m_logger;
+    };
+
     class RADIUS : public Auth
     {
         public:
             RADIUS();
+            RADIUS(const RADIUS&) = delete;
+            RADIUS& operator=(const RADIUS&) = delete;
+
+            void SetUsers(Users* u) override { m_users = u; }
+            void SetSettings(const ModuleSettings& s) override { m_settings = s; }
+            int ParseSettings() override;
+
+            int Start() override;
+            int Stop() override;
+            int Reload(const ModuleSettings& /*ms*/) override { return 0; }
+            bool IsRunning() override;
+            void SetRunning(bool val);
 
-            int Start() override { return 0; }
-            int Stop() override { return 0; }
-            int Reload(const ModuleSettings & /*ms*/) override { return 0; }
-            bool IsRunning() override { return isRunning; }
-            int ParseSettings() override { return 0; }
-            const std::string & GetStrError() const override { return errorStr; }
+            const std::string& GetStrError() const override { return m_errorStr; }
             std::string GetVersion() const override;
+
             uint16_t GetStartPosition() const override { return 0; }
             uint16_t GetStopPosition() const override { return 0; }
 
-            int SendMessage(const Message & msg, uint32_t ip) const override { return 0; }
+            int SendMessage(const Message& /*msg*/, uint32_t /*ip*/) const override { return 0; }
 
         private:
-            mutable std::string errorStr;
-            bool isRunning;
+            std::mutex m_mutex;
+
+            boost::asio::io_context m_ioContext;
+            int Run(std::stop_token token);
+
+            mutable std::string m_errorStr;
+            RAD_SETTINGS m_radSettings;
+            ModuleSettings m_settings;
+
+            bool m_running;
+
+            std::jthread m_thread;
+            Users* m_users;
+            PluginLogger m_logger;
 
+            std::unique_ptr<Server> m_server;
     };
 }