/*

 * Copyright (C) 2006, 2007 Apple Inc.  All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 *

 * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY

 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE

 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE COMPUTER, INC. OR

 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,

 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,

 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR

 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY

 * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE

 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 

 */

#ifndef MemoryStream_H

#define MemoryStream_H

#include <objidl.h>

#include "COMPtr.h"

#include <WebCore/SharedBuffer.h>

#include <WTF/PassRefPtr.h>

#include <WTF/RefPtr.h>

class MemoryStream : public IStream

{

public:

    static MemoryStream* MemoryStream::createInstance(PassRefPtr<WebCore::SharedBuffer> buffer);

protected:

    MemoryStream(PassRefPtr<WebCore::SharedBuffer> buffer);

    ~MemoryStream();

public:

    // IUnknown

    virtual HRESULT STDMETHODCALLTYPE QueryInterface(REFIID riid, void** ppvObject);

    virtual ULONG STDMETHODCALLTYPE AddRef(void);

    virtual ULONG STDMETHODCALLTYPE Release(void);

    // ISequentialStream

    virtual /* [local] */ HRESULT STDMETHODCALLTYPE Read( 

        /* [length_is][size_is][out] */ void* pv,

        /* [in] */ ULONG cb,

        /* [out] */ ULONG* pcbRead);

    virtual /* [local] */ HRESULT STDMETHODCALLTYPE Write( 

        /* [size_is][in] */ const void* pv,

        /* [in] */ ULONG cb,

        /* [out] */ ULONG* pcbWritten);

    // IStream

    virtual /* [local] */ HRESULT STDMETHODCALLTYPE Seek( 

        /* [in] */ LARGE_INTEGER dlibMove,

        /* [in] */ DWORD dwOrigin,

        /* [out] */ ULARGE_INTEGER* plibNewPosition);

    virtual HRESULT STDMETHODCALLTYPE SetSize( 

        /* [in] */ ULARGE_INTEGER libNewSize);

    virtual /* [local] */ HRESULT STDMETHODCALLTYPE CopyTo( 

        /* [unique][in] */ IStream* pstm,

        /* [in] */ ULARGE_INTEGER cb,

        /* [out] */ ULARGE_INTEGER* pcbRead,

        /* [out] */ ULARGE_INTEGER* pcbWritten);

    virtual HRESULT STDMETHODCALLTYPE Commit( 

        /* [in] */ DWORD grfCommitFlags);

    virtual HRESULT STDMETHODCALLTYPE Revert( void);

    virtual HRESULT STDMETHODCALLTYPE LockRegion( 

        /* [in] */ ULARGE_INTEGER libOffset,

        /* [in] */ ULARGE_INTEGER cb,

        /* [in] */ DWORD dwLockType);

    virtual HRESULT STDMETHODCALLTYPE UnlockRegion( 

        /* [in] */ ULARGE_INTEGER libOffset,

        /* [in] */ ULARGE_INTEGER cb,

        /* [in] */ DWORD dwLockType);

    virtual HRESULT STDMETHODCALLTYPE Stat( 

        /* [out] */ STATSTG* pstatstg,

        /* [in] */ DWORD grfStatFlag);

    virtual HRESULT STDMETHODCALLTYPE Clone( 

        /* [out] */ IStream** ppstm);

protected:

    ULONG m_refCount;

    RefPtr<WebCore::SharedBuffer> m_buffer;

    size_t m_pos;

};

#endif