sink_mem.h

Go to the documentation of this file.

#ifndef CVMFS_NETWORK_SINK_MEM_H_
#define CVMFS_NETWORK_SINK_MEM_H_

#include <cassert>
#include <cstring>
#include <string>

#include "sink.h"
#include "util/posix.h"
#include "util/smalloc.h"

namespace cvmfs {

class MemSink : public Sink {
 public:
  MemSink() : Sink(true), size_(0), pos_(0),
              data_(NULL), max_size_(kMaxMemSize) { }
  explicit MemSink(size_t size);
  virtual ~MemSink() { FreeData(); }

  virtual int64_t Write(const void *buf, uint64_t sz);

  virtual int Reset();

  virtual int Purge() {
    return Reset();
  }

  virtual bool IsValid();

  virtual int Flush() {
    return 0;
  }

  virtual bool Reserve(size_t size);

  virtual bool RequiresReserve() {
    return true;
  }

  virtual std::string Describe();

  void Adopt(size_t size, size_t pos, unsigned char *data,
             bool is_owner = true);

  size_t size() { return size_; }
  size_t pos() { return pos_; }
  unsigned char* data() { return data_; }

  static const size_t kMaxMemSize = 1024ul * 1024ul;

 private:
  void FreeData() {
    if (is_owner_) {
      free(data_);
    }
  }

  size_t size_;
  size_t pos_;
  unsigned char *data_;
  const size_t max_size_;
};

}  // namespace cvmfs

#endif  // CVMFS_NETWORK_SINK_MEM_H_