libt2n: (gerd) fixes & testcase
[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 #include <string>
24
25 #include "server.hxx"
26 #include "socket_handler.hxx"
27 #include "types.hxx"
28
29 namespace libt2n
30 {
31
32 /**
33     Socket based server class
34 */
35 class socket_server : public socket_handler, public server
36 {
37     private:
38         fd_set connection_set;
39         std::string unix_path;
40
41         void new_connection();
42
43     protected:
44         void log(log_level_values level, const std::string& message)
45             { log(level,message.c_str()); }
46         void log(log_level_values level, const char* message)
47             { server::log(level,message); }
48
49     public:
50         socket_server(int port, const std::string& ip="0.0.0.0");
51         socket_server(const std::string& path, mode_t filemode=00770, const std::string& user="", const std::string& group="");
52
53         ~socket_server();
54
55         void fill_buffer(long long usec_timeout=-1);
56         void fill_connection_buffers();
57
58         void remove_connection_socket(int sock);
59 };
60
61 /**
62     Socket based connection class
63 */
64 class socket_server_connection : public socket_handler, public server_connection
65 {
66     friend class socket_server;
67
68     private:
69         socket_server_connection(int _sock, socket_type_value _stype, int _timeout)
70            : server_connection(_timeout), socket_handler(_sock,_stype)
71            { }
72
73         void log(log_level_values level, const char* message);
74
75         void real_write(const std::string& data)
76             { socket_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