serialize_buffer_accessor.h

Go to the documentation of this file.

#ifndef SERIALIZE_ACCESSOR_H
#define SERIALIZE_ACCESSOR_H

#include <cstring>
#include <sprockit/errors.h>

namespace sprockit {
namespace pvt {

class ser_buffer_overrun : public spkt_error {
 public:
  ser_buffer_overrun(int maxsize) :
    spkt_error(sprockit::printf("serialization overrun buffer of size %d", maxsize))
  {
  }
};

class ser_buffer_accessor {
 public:
  template <class T>
  T*
  next(){
    T* ser_buffer = reinterpret_cast<T*>(bufptr_);
    bufptr_ += sizeof(T);
    size_ += sizeof(T);
    if (size_ > max_size_) throw ser_buffer_overrun(max_size_);
    return ser_buffer;
  }

  char*
  next_str(size_t size){
    char* ser_buffer = reinterpret_cast<char*>(bufptr_);
    bufptr_ += size;
    size_ += size;
    if (size_ > max_size_) throw ser_buffer_overrun(max_size_);
    return ser_buffer;
  }

  size_t
  size() const {
    return size_;
  }

  size_t
  max_size() const {
    return max_size_;
  }

  void
  init(void* buffer, size_t size){
    bufstart_ = reinterpret_cast<char*>(buffer);
    max_size_ = size;
    reset();
  }

  void
  clear(){
    bufstart_ = bufptr_ = 0;
    max_size_ = size_ = 0;
  }

  void
  reset(){
    bufptr_ = bufstart_;
    size_ = 0;
  }

 protected:
  ser_buffer_accessor() :
    bufstart_(0),
    bufptr_(0),
    size_(0)
  {
  }

 protected:
  char* bufstart_;
  char* bufptr_;
  size_t size_;
  size_t max_size_;

};

} }

#endif // SERIALIZE_ACCESSOR_H