tcp_socket.h

Go to the documentation of this file.

#ifndef ODVA_ETHERNETIP_SOCKET_TCP_SOCKET_H
#define ODVA_ETHERNETIP_SOCKET_TCP_SOCKET_H

#include <string>
#include <boost/asio.hpp>

#include "odva_ethernetip/socket/socket.h"

using std::size_t;
using std::string;
using namespace boost::asio;
using boost::asio::ip::tcp;

namespace eip {
namespace socket {

class TCPSocket : public Socket
{
public:

  TCPSocket(io_service& io_serv) : socket_(io_serv) { }

  virtual void open(string hostname, string port)
  {
    tcp::resolver resolver(socket_.get_io_service());
    tcp::resolver::query query(hostname, port);
    tcp::resolver::iterator endpoint_iterator = resolver.resolve(query);
    connect(socket_, endpoint_iterator);
  }

  virtual inline void close()
  {
    socket_.close();
  }

  virtual size_t send(const_buffer buf)
  {
    return socket_.send(const_buffers_1(buf));
  }

  virtual size_t receive(mutable_buffer buf)
  {
    // have to make this into a sequence or the templating engine throws a fit
    return socket_.receive(mutable_buffers_1(buf));
  }

private:
  tcp::socket socket_;
};

} // namespace socket
} // namespace eip
#endif  // ODVA_ETHERNETIP_SOCKET_TCP_SOCKET_H