/mordor/streams/memory.h

#ifndef __MORDOR_MEMORY_STREAM_H__
#define __MORDOR_MEMORY_STREAM_H__
// Copyright (c) 2009 - Mozy, Inc.

#include "buffer.h"
#include "stream.h"

namespace Mordor {

class MemoryStream : public Stream
{
public:
    typedef boost::shared_ptr<MemoryStream> ptr;
public:
    MemoryStream();
    MemoryStream(const Buffer &b);

    bool supportsRead() { return true; }
    bool supportsWrite() { return true; }
    bool supportsSeek() { return true; }
    bool supportsSize() { return true; }
    bool supportsTruncate() { return true; }
    bool supportsFind() { return true; }

    size_t read(Buffer &b, size_t len);
    size_t read(void *buffer, size_t length);
    size_t write(const Buffer &b, size_t len);
    size_t write(const void *b, size_t len);
    long long seek(long long offset, Anchor anchor = BEGIN);
    long long size();
    void truncate(long long size);
    ptrdiff_t find(char delim, size_t sanitySize = ~0, bool throwIfNotFound = true);
    ptrdiff_t find(const std::string &str, size_t sanitySize = ~0, bool throwIfNotFound = true);

    // Direct access to memory
    const Buffer &buffer() const { return m_original; }
    const Buffer &readBuffer() const { return m_read; }

private:
    template <class T> size_t readInternal(T &buffer, size_t length);
    template <class T> size_t writeInternal(const T &buffer, size_t length);

private:
    Buffer m_read;
    Buffer m_original;
    size_t m_offset;
};

}

#endif