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