* Author : Maxim Mamontov <faust@stargazer.dp.ua>
*/
-/*
- * Radius data access plugin for Stargazer
- *
- * $Revision: 1.10 $
- * $Date: 2009/12/13 14:17:13 $
- *
- */
+#ifndef __STG_RADIUS_H__
+#define __STG_RADIUS_H__
+
+#include "stg/os_int.h"
+#include "stg/auth.h"
+#include "stg/module_settings.h"
+#include "stg/logger.h"
-#ifndef RADIUS_H
-#define RADIUS_H
+#include "config.h"
+#include "conn.h"
#include <string>
-#include <list>
+#include <deque>
+#include <set>
+
#include <pthread.h>
-#include <string.h>
-#include <stdlib.h>
+#include <unistd.h>
+#include <sys/select.h>
+#include <sys/types.h>
-#include "os_int.h"
-#include "base_auth.h"
-#include "notifer.h"
-#include "user_ips.h"
-#include "../../../user.h"
-#include "../../../users.h"
-#include "blowfish.h"
-#include "rad_packets.h"
+extern "C" PLUGIN * GetPlugin();
-using namespace std;
+class STORE;
+class USERS;
-extern "C" BASE_PLUGIN * GetPlugin();
+class RADIUS : public AUTH {
+public:
+ RADIUS();
+ virtual ~RADIUS() {}
-#define RAD_DEBUG (1)
+ 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();
-class RADIUS;
-//-----------------------------------------------------------------------------
-class RAD_SETTINGS
-{
-public:
- virtual ~RAD_SETTINGS(){};
- const string& GetStrError() const { return errorStr; };
- int ParseSettings(const MODULE_SETTINGS & s);
- uint16_t GetPort() const;
- int GetPassword(string * password) const;
- int GetAuthServices(list<string> * svcs) const;
- int GetAcctServices(list<string> * svcs) const;
+ int Start();
+ int Stop();
+ int Reload() { return 0; }
+ bool IsRunning() { return m_running; }
-private:
- int ParseIntInRange(const string & str, int min, int max, int * val);
- int ParseIP(const string & str, uint32_t * routerIP);
- int ParseServices(const vector<string> & str, list<string> * lst);
-
- uint16_t port;
- string errorStr;
- string password;
- list<string> authServices;
- list<string> acctServices;
-};
-//-----------------------------------------------------------------------------
-struct RAD_SESSION {
- std::string userName;
- std::string serviceType;
-};
-//-----------------------------------------------------------------------------
-class RADIUS :public BASE_AUTH
-{
-public:
- RADIUS();
- virtual ~RADIUS(){};
-
- void SetUsers(USERS * u);
- void SetTariffs(TARIFFS *){};
- void SetAdmins(ADMINS *){};
- void SetTraffcounter(TRAFFCOUNTER *){};
- void SetStore(BASE_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 string & GetStrError() const { return errorStr; };
- const string GetVersion() const;
- uint16_t GetStartPosition() const;
- uint16_t GetStopPosition() const;
-
- int SendMessage(const STG_MSG &, uint32_t) const { return 0; };
+ 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();
-
- void InitEncrypt(BLOWFISH_CTX * ctx, const string & password);
- void Decrypt(BLOWFISH_CTX * ctx, char * dst, const char * src, int len8);
- void Encrypt(BLOWFISH_CTX * ctx, char * dst, const char * src, int len8);
-
- 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_iter * 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;
-
- void SetUserNotifier(user_iter u);
- void UnSetUserNotifier(user_iter u);
-
- bool WaitPackets(int sd) const;
-
- void PrintServices(const std::list<std::string> & svcs);
-
- struct Printer : public unary_function<std::string, void>
- {
- void operator()(const std::string & line)
- {
- printfd("radius.cpp", "'%s'\n", line.c_str());
- };
- };
- struct SPrinter : public unary_function<std::pair<std::string, RAD_SESSION>, void>
- {
- void operator()(const std::pair<std::string, RAD_SESSION> & 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 string errorStr;
- RAD_SETTINGS radSettings;
- MODULE_SETTINGS settings;
- list<string> authServices;
- list<string> acctServices;
- map<string, RAD_SESSION> sessions;
-
- bool nonstop;
- bool isRunning;
-
- USERS * users;
- const SETTINGS * stgSettings;
- const BASE_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<STG::Conn*> m_conns;
+ std::set<std::string> m_logins;
+
+ pthread_t m_thread;
+ PLUGIN_LOGGER m_logger;
};
-//-----------------------------------------------------------------------------
#endif
git.stg.codes / stg.git / blobdiff