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 | ||
cc68aabb GE |
49 | void start_listening(); |
50 | ||
0cf4dc9b | 51 | void new_connection(); |
ac7fdc22 | 52 | |
94247295 GE |
53 | bool fill_connection_buffers(); |
54 | void remove_connection_socket(int sock); | |
55 | ||
a11e19b7 | 56 | protected: |
a7170401 GE |
57 | std::ostream* get_logstream(log_level_values level) |
58 | { return server::get_logstream(level); } | |
a11e19b7 | 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 | ||
45a2ebc9 | 66 | bool fill_buffer(long long usec_timeout=-1,long long* usec_timeout_remaining=NULL); |
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 | |
a7170401 GE |
82 | std::ostream* get_logstream(log_level_values level) |
83 | { return server_connection::get_logstream(level); } | |
ac7fdc22 | 84 | |
a11e19b7 | 85 | void real_write(const std::string& data) |
644c4d26 | 86 | { socket_write(data); } |
ac7fdc22 GE |
87 | |
88 | public: | |
45a2ebc9 GE |
89 | bool fill_buffer(long long usec_timeout=-1,long long* usec_timeout_remaining=NULL) |
90 | { return socket_handler::fill_buffer(buffer,usec_timeout,usec_timeout_remaining); } | |
ac7fdc22 GE |
91 | |
92 | void close(); | |
ac7fdc22 GE |
93 | }; |
94 | ||
95 | } | |
96 | ||
97 | #endif |