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