buffer_reader.h

Go to the documentation of this file.

#ifndef ODVA_ETHERNETIP_SERIALIZATION_BUFFER_READER_H
#define ODVA_ETHERNETIP_SERIALIZATION_BUFFER_READER_H

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

#include "odva_ethernetip/eip_types.h"
#include "odva_ethernetip/serialization/reader.h"

using boost::asio::buffer;
using boost::asio::mutable_buffer;
using boost::asio::buffer_size;

namespace eip {
namespace serialization {

class BufferReader : public Reader
{
public:

  BufferReader(mutable_buffer buf) : buf_(buf), byte_count_(0) { }

  virtual void readBytes(void* buf, size_t n)
  {
    readBuffer(buffer(buf, n));
  }

  virtual void* readBytes(size_t n)
  {
    return boost::asio::buffer_cast<void*>(readBuffer(n));
  }

  virtual mutable_buffer readBuffer(size_t n)
  {
    if (buffer_size(buf_) < n)
    {
      throw std::length_error("Buffer too small to deserialize value");
    }
    mutable_buffer b = buffer(buf_, n);
    byte_count_ += n;
    buf_ = buf_ + n;
    return b;
  }

  virtual void readBuffer(mutable_buffer buf)
  {
    size_t n = buffer_size(buf);
    if (buffer_size(buf_) < n)
    {
      throw std::length_error("Buffer too small to deserialize value");
    }
    boost::asio::buffer_copy(buf, buf_);
    byte_count_ += n;
    buf_ = buf_ + n;
  }

  virtual size_t getByteCount()
  {
    return byte_count_;
  }

  virtual void skip(size_t n)
  {
    if (buffer_size(buf_) < n)
    {
      throw std::length_error("End of buffer reached skipping forward");
    }
    byte_count_ += n;
    buf_ = buf_ + n;
  }
private:
  mutable_buffer buf_;
  size_t byte_count_;
};

} // namespace serialization
} // namespace eip
#endif  // ODVA_ETHERNETIP_SERIALIZATION_BUFFER_READER_H