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

diff --git a/src/socket_server.hxx b/src/socket_server.hxx
index ca2f03c..a9c7b3d 100644
--- a/src/socket_server.hxx
+++ b/src/socket_server.hxx
@@ -20,50 +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 { tcp, unix };
-
     private:
-        int socket;
         fd_set connection_set;
-        socket_type type;
-        string unix_path;
+        std::string unix_path;
+
+        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 chmod, 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 get_type()
-            { return 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);
 };
 
 }