558b058af3bb059b0f198a8010b49a92f67c352b
[libt2n] / src / socket_server.hxx
1 /***************************************************************************
2  *   Copyright (C) 2006 by Gerd v. Egidy                                   *
3  *   gve@intra2net.com                                                     *
4  *                                                                         *
5  *   This library is free software; you can redistribute it and/or modify  *
6  *   it under the terms of the GNU Lesser General Public License version   *
7  *   2.1 as published by the Free Software Foundation.                     *
8  *                                                                         *
9  *   This library is distributed in the hope that it will be useful,       *
10  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
11  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
12  *   GNU Lesser General Public License for more details.                   *
13  *                                                                         *
14  *   You should have received a copy of the GNU Lesser General Public      *
15  *   License along with this program; if not, write to the                 *
16  *   Free Software Foundation, Inc.,                                       *
17  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
18  ***************************************************************************/
19 #ifndef __LIBT2N_SOCKET_SERVER
20 #define __LIBT2N_SOCKET_SERVER
21
22 #include <sys/types.h>
23
24 #include "server.hxx"
25 #include "socket_handler.hxx"
26 #include "types.hxx"
27
28 namespace libt2n
29 {
30
31 /**
32     Socket based server class
33 */
34 class socket_server : public socket_handler, server
35 {
36     private:
37         fd_set connection_set;
38         std::string unix_path;
39
40         void new_connection();
41
42     protected:
43         void log(log_level_values level, const std::string& message)
44             { log(level,message.c_str()); }
45         void log(log_level_values level, const char* message)
46             { server::log(level,message); }
47
48     public:
49         socket_server(int port, const char* ip="0.0.0.0");
50         socket_server(const char* path, mode_t filemode, const char* user="", const char* group="");
51
52         ~socket_server();
53
54         void fill_buffer(long long usec_timeout=-1);
55         void fill_connection_buffers();
56
57         void remove_connection_socket(int sock);
58 };
59
60 /**
61     Socket based connection class
62 */
63 class socket_server_connection : public socket_handler, server_connection
64 {
65     friend class socket_server;
66
67     private:
68         socket_server_connection(int _sock, socket_type_value _stype, int _timeout)
69            : server_connection(_timeout), socket_handler(_sock,_stype)
70            { }
71
72         void log(log_level_values level, const char* message)
73             { if(my_server) my_server->log(level,message); }
74
75         void real_write(const std::string& data)
76             { socket_handler::write(data); }
77
78     public:
79         void fill_buffer(long long usec_timeout=-1)
80             { socket_handler::fill_buffer(buffer,usec_timeout); }
81
82         void close();
83 };
84
85 }
86
87 #endif