[libt2n] / src / socket_handler.hxx

/***************************************************************************
 *   Copyright (C) 2006 by Gerd v. Egidy                                   *
 *   gve@intra2net.com                                                     *
 *                                                                         *
 *   This library is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU Lesser General Public License version   *
 *   2.1 as published by the Free Software Foundation.                     *
 *                                                                         *
 *   This library is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU Lesser General Public License for more details.                   *
 *                                                                         *
 *   You should have received a copy of the GNU Lesser General Public      *
 *   License along with this program; if not, write to the                 *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/
#ifndef __LIBT2N_SOCKET_HANDLER
#define __LIBT2N_SOCKET_HANDLER

#include <iostream>

#include "types.hxx"

namespace libt2n
{

/** @brief handles socket based communication.
    Don't use directly, use socket_server or socket_client_connection instead.
*/
class socket_handler
{
    private:
        static const unsigned int default_recv_buffer_size=2048;
        static const unsigned int default_write_block_size=4096;

        socket_type_value socket_type;

        bool data_waiting(long long usec_timeout,long long *timeout_remaining=NULL);

    protected:
        int sock;
        unsigned int recv_buffer_size;
        unsigned int write_block_size;

        socket_handler(int _sock, socket_type_value _socket_type);

        void set_socket_options(int sock);

        virtual std::ostream* get_logstream(log_level_values level)
            { return NULL; }

        void socket_write(const std::string& data);

        virtual void close();

        bool fill_buffer(std::string& buffer, long long usec_timeout, long long*timeout_remaining=NULL);
        bool fill_buffer(std::string& buffer);

    public:
        /// is this a tcp or unix socket connection
        socket_type_value get_type()
            { return socket_type; }

        bool is_closed();

        void set_recv_buffer_size(unsigned int new_recv_buffer_size);
        void set_write_block_size(unsigned int new_write_block_size);

        unsigned int get_recv_buffer_size() const { return recv_buffer_size; }
        unsigned int get_write_block_size() const { return write_block_size; }
};

}

#endif
Commit	Line	Data
	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