b9aa175ddaaa9a7bf3e475cf999e10bd3229f247
[libt2n] / src / socket_handler.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_HANDLER
20 #define __LIBT2N_SOCKET_HANDLER
21
22 #include <iostream>
23
24 #include "types.hxx"
25
26 namespace libt2n
27 {
28
29 /** @brief handles socket based communication.
30     Don't use directly, use socket_server or socket_client_connection instead.
31 */
32 class socket_handler
33 {
34     private:
35         static const unsigned int default_recv_buffer_size=2048;
36         static const unsigned int default_write_block_size=4096;
37
38         socket_type_value socket_type;
39
40         bool data_waiting(long long usec_timeout,long long *timeout_remaining=NULL);
41
42     protected:
43         int sock;
44         unsigned int recv_buffer_size;
45         unsigned int write_block_size;
46
47         socket_handler(int _sock, socket_type_value _socket_type);
48
49         void set_socket_options(int sock);
50
51         virtual std::ostream* get_logstream(log_level_values level)
52             { return NULL; }
53
54         void socket_write(const std::string& data);
55
56         virtual void close();
57
58         bool fill_buffer(std::string& buffer, long long usec_timeout, long long*timeout_remaining=NULL);
59         bool fill_buffer(std::string& buffer);
60
61     public:
62         /// is this a tcp or unix socket connection
63         socket_type_value get_type()
64             { return socket_type; }
65
66         bool is_closed();
67
68         void set_recv_buffer_size(unsigned int new_recv_buffer_size);
69         void set_write_block_size(unsigned int new_write_block_size);
70
71         unsigned int get_recv_buffer_size() const { return recv_buffer_size; }
72         unsigned int get_write_block_size() const { return write_block_size; }
73 };
74
75 }
76
77 #endif