Experimental implementation of SGCP using Boost.Asio.

[stg.git] / projects / rlm_stg / stg_client.h

diff --git a/projects/rlm_stg/stg_client.h b/projects/rlm_stg/stg_client.h
index 680d088387907fede8851bde53e843642db9ef96..d57af7dacfc7c2201d0404750e0d31d89d213a8c 100644 (file)
--- a/projects/rlm_stg/stg_client.h
+++ b/projects/rlm_stg/stg_client.h
@@ -18,44 +18,74 @@
  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  */
 
  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  */
 

-/*
- *  Header file for client part of data access via Stargazer for RADIUS
- *
- *  $Revision: 1.4 $
- *  $Date: 2010/04/16 12:30:02 $
- *
- */
-

 #ifndef STG_CLIENT_H
 #define STG_CLIENT_H
 
 #ifndef STG_CLIENT_H
 #define STG_CLIENT_H
 

-#include <string>
+#include "stg/sgcp_proto.h" // Proto
+#include "stg/sgcp_types.h" // TransportType
+#include "stg/os_int.h"

 
 

-#include <netinet/in.h>
-#include <arpa/inet.h>
-#include <sys/socket.h> // socklen_t
+#include <boost/thread.hpp>

 
 

-#include "stg/blowfish.h"
-#include "stg/rad_packets.h"
+#include <string>
+#include <vector>
+#include <utility>

 
 typedef std::vector<std::pair<std::string, std::string> > PAIRS;
 
 
 typedef std::vector<std::pair<std::string, std::string> > PAIRS;
 

+struct RESULT
+{
+    PAIRS modify;
+    PAIRS reply;
+};
+
+struct ChannelConfig {
+    struct Error : std::runtime_error {
+        Error(const std::string& message) : runtime_error(message) {}
+    };
+
+    ChannelConfig(std::string address);
+
+    STG::SGCP::TransportType transport;
+    std::string key;
+    std::string address;
+    uint16_t port;
+};
+

 class STG_CLIENT
 {
 public:
 class STG_CLIENT
 {
 public:

-    STG_CLIENT(const std::string & host, uint16_t port, const std::string & password);
+    enum TYPE {
+        AUTHORIZE,
+        AUTHENTICATE,
+        POST_AUTH,
+        PRE_ACCT,
+        ACCOUNT
+    };
+    struct Error : std::runtime_error {
+        Error(const std::string& message) : runtime_error(message) {}
+    };
+
+    STG_CLIENT(const std::string& address);

     ~STG_CLIENT();
 
     ~STG_CLIENT();
 

+    bool stop();
+

     static STG_CLIENT* get();
     static STG_CLIENT* get();

-    static void configure(const std::string& server, uint16_t port, const std::string& password);
+    static bool configure(const std::string& address);

 
 

-    PAIRS authorize(const PAIRS& pairs);
-    PAIRS authenticate(const PAIRS& pairs);
-    PAIRS postAuth(const PAIRS& pairs);
-    PAIRS preAcct(const PAIRS& pairs);
-    PAIRS account(const PAIRS& pairs);
+    RESULT request(TYPE type, const std::string& userName, const std::string& password, const PAIRS& pairs);

 
 private:
 
 private:

+    ChannelConfig m_config;
+    STG::SGCP::Proto m_proto;
+    boost::thread m_thread;
+
+    void m_writeHeader(TYPE type, const std::string& userName, const std::string& password);
+    void m_writePairBlock(const PAIRS& source);
+    PAIRS m_readPairBlock();
+
+    void m_run();

 };
 
 #endif
 };
 
 #endif