X-Git-Url: http://developer.intra2net.com/git/?p=libt2n;a=blobdiff_plain;f=src%2Fsocket_server.hxx;h=a9c7b3d0433d8d15d83c4784e5393d8acba43c6e;hp=8f1b84c3306d4b25c03c026a89fe26015418988c;hb=644c4d262aab14ee4ec1cfade83a2e99568098d3;hpb=0cf4dc9bf7fa527751fd7dc425f882fc86888132

diff --git a/src/socket_server.hxx b/src/socket_server.hxx
index 8f1b84c..a9c7b3d 100644
--- a/src/socket_server.hxx
+++ b/src/socket_server.hxx
@@ -20,53 +20,66 @@
 #define __LIBT2N_SOCKET_SERVER
 
 #include <sys/types.h>
+#include <string>
 
 #include "server.hxx"
+#include "socket_handler.hxx"
+#include "types.hxx"
 
 namespace libt2n
 {
 
-class socket_server : public server
+/**
+    Socket based server class
+*/
+class socket_server : public socket_handler, public server
 {
-    public:
-        enum socket_type_value { tcp_s, unix_s };
-
     private:
-        int sock;
         fd_set connection_set;
-        socket_type_value socket_type;
         std::string unix_path;
 
-        void set_socket_options(int sock);
         void new_connection();
 
+    protected:
+        void log(log_level_values level, const std::string& message)
+            { log(level,message.c_str()); }
+        void log(log_level_values level, const char* message)
+            { server::log(level,message); }
+
     public:
-        socket_server(int port, const char* ip="0.0.0.0");
-        socket_server(const char* path, mode_t filemode, const char* user="", const char* group="");
+        socket_server(int port, const std::string& ip="0.0.0.0");
+        socket_server(const std::string& path, mode_t filemode=00770, const std::string& user="", const std::string& group="");
 
         ~socket_server();
 
-        socket_type_value get_type()
-            { return socket_type; }
-
         void fill_buffer(long long usec_timeout=-1);
+        void fill_connection_buffers();
+
+        void remove_connection_socket(int sock);
 };
 
-class socket_connection : public connection
+/**
+    Socket based connection class
+*/
+class socket_server_connection : public socket_handler, public server_connection
 {
+    friend class socket_server;
+
     private:
-        int socket;
+        socket_server_connection(int _sock, socket_type_value _stype, int _timeout)
+           : server_connection(_timeout), socket_handler(_sock,_stype)
+           { }
+
+        void log(log_level_values level, const char* message);
 
-        friend void socket_server::fill_buffer(long long usec_timeout);
-        socket_connection(int _socket, int _timeout);
+        void real_write(const std::string& data)
+            { socket_write(data); }
 
     public:
+        void fill_buffer(long long usec_timeout=-1)
+            { socket_handler::fill_buffer(buffer,usec_timeout); }
 
         void close();
-
-        void fill_buffer(void);
-
-        void write(const std::string& data);
 };
 
 }