Commit | Line | Data |
---|---|---|
ac7fdc22 GE |
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> | |
644c4d26 | 23 | #include <string> |
ac7fdc22 GE |
24 | |
25 | #include "server.hxx" | |
a11e19b7 GE |
26 | #include "socket_handler.hxx" |
27 | #include "types.hxx" | |
ac7fdc22 GE |
28 | |
29 | namespace libt2n | |
30 | { | |
31 | ||
94247295 GE |
32 | class socket_server_connection; |
33 | ||
34 | /** @brief Socket based server class | |
35 | ||
36 | Use this class to instantiate a server listening for client connections. | |
37 | Call fill_buffer() to read data from the network and get_packet() to retrieve | |
38 | this data. Don't forget to call cleanup() from time to time to remove closed | |
39 | connections and close idle ones. | |
59adb9e2 | 40 | */ |
644c4d26 | 41 | class socket_server : public socket_handler, public server |
ac7fdc22 | 42 | { |
94247295 GE |
43 | friend class socket_server_connection; |
44 | ||
ac7fdc22 | 45 | private: |
ac7fdc22 | 46 | fd_set connection_set; |
0cf4dc9b GE |
47 | std::string unix_path; |
48 | ||
0cf4dc9b | 49 | void new_connection(); |
ac7fdc22 | 50 | |
94247295 GE |
51 | bool fill_connection_buffers(); |
52 | void remove_connection_socket(int sock); | |
53 | ||
a11e19b7 GE |
54 | protected: |
55 | void log(log_level_values level, const std::string& message) | |
56 | { log(level,message.c_str()); } | |
57 | void log(log_level_values level, const char* message) | |
58 | { server::log(level,message); } | |
59 | ||
ac7fdc22 | 60 | public: |
644c4d26 GE |
61 | socket_server(int port, const std::string& ip="0.0.0.0"); |
62 | socket_server(const std::string& path, mode_t filemode=00770, const std::string& user="", const std::string& group=""); | |
ac7fdc22 GE |
63 | |
64 | ~socket_server(); | |
65 | ||
94247295 | 66 | bool fill_buffer(long long usec_timeout=-1); |
ac7fdc22 GE |
67 | }; |
68 | ||
94247295 GE |
69 | /** @brief Socket based connection |
70 | ||
71 | This class is used within a socket_server to represent the connection to each client. | |
59adb9e2 | 72 | */ |
644c4d26 | 73 | class socket_server_connection : public socket_handler, public server_connection |
ac7fdc22 | 74 | { |
04e6b271 GE |
75 | friend class socket_server; |
76 | ||
ac7fdc22 | 77 | private: |
a11e19b7 GE |
78 | socket_server_connection(int _sock, socket_type_value _stype, int _timeout) |
79 | : server_connection(_timeout), socket_handler(_sock,_stype) | |
80 | { } | |
aa499d20 | 81 | |
644c4d26 | 82 | void log(log_level_values level, const char* message); |
ac7fdc22 | 83 | |
a11e19b7 | 84 | void real_write(const std::string& data) |
644c4d26 | 85 | { socket_write(data); } |
ac7fdc22 GE |
86 | |
87 | public: | |
07e98688 GE |
88 | bool fill_buffer(long long usec_timeout=-1) |
89 | { return socket_handler::fill_buffer(buffer,usec_timeout); } | |
ac7fdc22 GE |
90 | |
91 | void close(); | |
ac7fdc22 GE |
92 | }; |
93 | ||
94 | } | |
95 | ||
96 | #endif |