test_socket.h

Go to the documentation of this file.

#ifndef ODVA_ETHERNETIP_SOCKET_TEST_SOCKET_H
#define ODVA_ETHERNETIP_SOCKET_TEST_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;

namespace eip {
namespace socket {

class TestSocket : public Socket
{
public:

  bool is_open;

  string hostname;

  string port;

  unsigned char tx_buffer[4096];
  size_t tx_count;

  const_buffer rx_buffer;
  size_t rx_count;

  TestSocket() : tx_count(0), rx_count(0), is_open(false) { }

  TestSocket(const_buffer rbuf) : tx_count(0), rx_count(0), rx_buffer(rbuf), is_open(false) { }

  virtual void open(string hostname, string port)
  {
    this->hostname = hostname;
    this->port = port;
    is_open = true;
  }

  virtual void close()
  {
    is_open = false;
  }

  virtual size_t send(const_buffer buf)
  {
    size_t n = buffer_copy(buffer(tx_buffer) + tx_count, buf);
    tx_count += n;
    return n;
  }

  virtual size_t receive(mutable_buffer buf)
  {
    size_t n = buffer_copy(buf, rx_buffer);
    rx_count += n;
    rx_buffer = rx_buffer + n;
    return n;
  }

  void clearTxBuffer()
  {
    tx_count = 0;
    memset(tx_buffer, 0, sizeof(tx_count));
  }

};

} // namespace socket
} // namespace eip
#endif  // ODVA_ETHERNETIP_SOCKET_TEST_SOCKET_H