1 /*

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

     3  *

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

     5  * modification, are permitted provided that the following conditions

     6  * are met:

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

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

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

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

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

    12  *

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

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

    15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR

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

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

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

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

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

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

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

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

    24  */

    25 

    26 #ifndef MemoryStream_H

    27 #define MemoryStream_H

    28 

    29 #include <objidl.h>

    30 

    31 #include "COMPtr.h"

    32 #include <WebCore/SharedBuffer.h>

    33 #include <WTF/PassRefPtr.h>

    34 #include <WTF/RefPtr.h>

    35 

    36 class MemoryStream : public IStream

    37 {

    38 public:

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

    40 

    41 protected:

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

    43     ~MemoryStream();

    44 public:

    45 

    46     // IUnknown

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

    48     virtual ULONG STDMETHODCALLTYPE AddRef(void);

    49     virtual ULONG STDMETHODCALLTYPE Release(void);

    50 

    51     // ISequentialStream

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

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

    54         /* [in] */ ULONG cb,

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

    56     

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

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

    59         /* [in] */ ULONG cb,

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

    61 

    62     // IStream

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

    64         /* [in] */ LARGE_INTEGER dlibMove,

    65         /* [in] */ DWORD dwOrigin,

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

    67     

    68     virtual HRESULT STDMETHODCALLTYPE SetSize( 

    69         /* [in] */ ULARGE_INTEGER libNewSize);

    70     

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

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

    73         /* [in] */ ULARGE_INTEGER cb,

    74         /* [out] */ ULARGE_INTEGER* pcbRead,

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

    76     

    77     virtual HRESULT STDMETHODCALLTYPE Commit( 

    78         /* [in] */ DWORD grfCommitFlags);

    79     

    80     virtual HRESULT STDMETHODCALLTYPE Revert( void);

    81     

    82     virtual HRESULT STDMETHODCALLTYPE LockRegion( 

    83         /* [in] */ ULARGE_INTEGER libOffset,

    84         /* [in] */ ULARGE_INTEGER cb,

    85         /* [in] */ DWORD dwLockType);

    86     

    87     virtual HRESULT STDMETHODCALLTYPE UnlockRegion( 

    88         /* [in] */ ULARGE_INTEGER libOffset,

    89         /* [in] */ ULARGE_INTEGER cb,

    90         /* [in] */ DWORD dwLockType);

    91     

    92     virtual HRESULT STDMETHODCALLTYPE Stat( 

    93         /* [out] */ STATSTG* pstatstg,

    94         /* [in] */ DWORD grfStatFlag);

    95     

    96     virtual HRESULT STDMETHODCALLTYPE Clone( 

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

    98 

    99 protected:

   100     ULONG m_refCount;

   101     RefPtr<WebCore::SharedBuffer> m_buffer;

   102     size_t m_pos;

   103 };

   104 

   105 #endif