Tcp.h

Go to the documentation of this file.

/*
 * Tcp.h
 *
 *  Created on: Jun 8, 2011
 *      Author: jtedesco
 */

#ifndef TCP_H_
#define TCP_H_

#include <boost/asio.hpp>
#include <boost/thread.hpp>
#include <boost/make_shared.hpp>
#include <boost/lexical_cast.hpp>
#include <boost/optional.hpp>
#include <iostream>
#include <set>
#include <map>

#if defined(__UNIX__)

#include <netinet/tcp.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <errno.h>

#elif defined(__OSX__)

#include <netinet/tcp.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <errno.h>

#else

#define WIN32_LEAN_AND_MEAN
#include <winsock2.h>
#include <Windows.h>
#include <MSTcpIP.h>

#endif


class Tcp;

#include "TcpEvent.h"
#include "Logger.h"

using boost::optional;
using boost::asio::ip::tcp;
using boost::asio::buffer_size;
using std::map;

class Tcp
{
    public:

        Tcp(string host, int port, boost::asio::io_service & io_service);

        virtual ~Tcp() {}

        friend class TcpEvent;

    protected:

        bool set_tcp_keepalive(boost::shared_ptr<tcp::socket> socket);

        inline string bool_option_to_string(optional<bool> &arg, string iftrue, string iffalse);

        template <class T> inline string option_to_string(optional<T> &arg);

        void inline parse_string_bool_arg(map<string, string> &options, string arg, optional<bool> &arg_value);

        void inline parse_string_int_arg(map<string, string> &options, string arg, optional<int> &arg_value);

        void parse_args(map<string, string> options);

        void log_options();

        virtual void close() = 0;

        virtual void send_handler(const boost::system::error_code & error_code, std::size_t bytes_transferred,
                string data, string host, int port, boost::shared_ptr<tcp::socket> connection);

        virtual void receive_handler(const boost::system::error_code & error_code, std::size_t bytes_transferred,
                boost::shared_ptr<tcp::socket> connection, string host, int port);

        virtual void fire_error_event(const string & message) = 0;

        virtual void fire_disconnect_event(const string & message) = 0;

        virtual void fire_data_event(const string data, boost::shared_ptr<tcp::socket> connection) = 0;

        static const int BUFFER_SIZE = 4096;

        static const int MAX_DATA_SIZE = 65536;

        boost::array<char, BUFFER_SIZE> receive_buffer;

        boost::asio::io_service & io_service;

        bool waiting_to_shutdown;

        optional<bool> using_ipv6;

        optional<bool> no_delay;

        optional<bool> do_not_route;

        optional<bool> keep_alive;

        optional<int> keep_alive_timeout;

        int active_jobs;

        boost::mutex active_jobs_mutex;

        string host;

        int port;

        std::set<boost::system::error_code> disconnect_errors;

        bool failed;
};

#endif /* TCP_H_ */