udp_socket.h

Go to the documentation of this file.

#ifndef ODVA_ETHERNETIP_SOCKET_UDP_SOCKET_H
#define ODVA_ETHERNETIP_SOCKET_UDP_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::udp;

namespace eip {
namespace socket {

class UDPSocket : public Socket
{
public:

  UDPSocket(io_service& io_serv, unsigned short local_port, std::string local_ip="0.0.0.0")
    : socket_(io_serv), local_endpoint_(boost::asio::ip::address::from_string(local_ip), local_port) { }

  virtual void open(string hostname, string port)
  {
    udp::resolver resolver(socket_.get_io_service());
    udp::resolver::query query(udp::v4(), hostname, port);
    remote_endpoint_ = *resolver.resolve(query);
    socket_.open(udp::v4());
    socket_.bind(local_endpoint_);
  }

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

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

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

private:
  udp::socket socket_;
  udp::endpoint local_endpoint_;
  udp::endpoint remote_endpoint_;
};

} // namespace socket
} // namespace eip
#endif  // ODVA_ETHERNETIP_SOCKET_UDP_SOCKET_H