TCP.h

Go to the documentation of this file.

#ifndef TCP_H_
#define TCP_H_

#include <iostream>
#include <string>
#include <boost/asio.hpp>
#include <boost/system/system_error.hpp>
#include <boost/bind.hpp>

#define ERROR_CODE_LOST_SOCKET  2


namespace apps
{

using boost::asio::ip::tcp;

class TCP
{
private:        //data elements

        enum _mode {server,client};
        _mode _choosenMode;

        unsigned int _delay;

        bool _requestNewConnection;

        boost::asio::io_service _ioService;
        unsigned int _port;
        std::string _ip;

        boost::asio::ip::tcp::endpoint* _targetServer;  //used when client

        boost::asio::ip::tcp::socket* _socket;

        tcp::acceptor* _server;                                                 //used when server

    // -- async_read + timeout --
    boost::asio::deadline_timer* _timer;

public:
        TCP(unsigned int port);                                         //server
        TCP(std::string ip, unsigned int port);         //client
        virtual ~TCP();

        int connectOnce();

        int read(void* data, unsigned int size);

        int readAll(void* data, unsigned int size);

        int read(void* data, unsigned int size, unsigned int delay_us);

        int write(void* data, unsigned int size);

private:    //functions
    void read_callback(bool& data_available, boost::asio::deadline_timer& timeout,
                       const boost::system::error_code& error,
                       std::size_t bytes_transferred);

    void wait_callback(boost::asio::ip::tcp::socket& ser_port,
                       const boost::system::error_code& error);

};

} /* namespace apps */
#endif /* TCP_H_ */