buffer_writer.h

Go to the documentation of this file.

#ifndef ODVA_ETHERNETIP_SERIALIZATION_BUFFER_WRITER_H
#define ODVA_ETHERNETIP_SERIALIZATION_BUFFER_WRITER_H

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

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

using boost::asio::const_buffer;
using boost::asio::mutable_buffer;

namespace eip {
namespace serialization {

class BufferWriter : public Writer
{
public:

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

  void writeBytes(const void* p, size_t n)
  {
    writeBuffer(boost::asio::buffer(p, n));
  }

  void writeBuffer(const_buffer b)
  {
    using boost::asio::buffer_size;
    if (buffer_size(buf_) < buffer_size(b))
    {
      throw std::length_error("Buffer to small to serialize value");
    }
    boost::asio::buffer_copy(buf_, b);
    byte_count_ += buffer_size(b);
    buf_ = buf_ + buffer_size(b);
  }

  size_t getByteCount()
  {
    return byte_count_;
  }

private:
  mutable_buffer buf_;
  size_t byte_count_;
};

} // namespace serialization
} // namespace eip
#endif  // ODVA_ETHERNETIP_SERIALIZATION_BUFFER_WRITER_H