1// Copyright (c) 2011 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#ifndef CHROME_FRAME_STREAM_IMPL_H_
6#define CHROME_FRAME_STREAM_IMPL_H_
7
8#include <atlbase.h>
9#include <atlcom.h>
10
11#include "base/win/scoped_comptr.h"
12
13// A generic base class for IStream implementation. If provided
14// with a delegate, it delegated calls to it otherwise can be
15// used a as a base class for an IStream implementation.
16class StreamImpl : public IStream {
17 public:
18  StreamImpl() {}
19
20  void Initialize(IStream* delegate);
21
22  STDMETHOD(Write)(const void * buffer, ULONG size, ULONG* size_written);
23  STDMETHOD(Read)(void* pv, ULONG cb, ULONG* read);
24  STDMETHOD(Seek)(LARGE_INTEGER move, DWORD origin, ULARGE_INTEGER* new_pos);
25  STDMETHOD(SetSize)(ULARGE_INTEGER new_size);
26  STDMETHOD(CopyTo)(IStream* stream, ULARGE_INTEGER cb, ULARGE_INTEGER* read,
27                    ULARGE_INTEGER* written);
28  STDMETHOD(Commit)(DWORD flags);
29  STDMETHOD(Revert)();
30  STDMETHOD(LockRegion)(ULARGE_INTEGER offset, ULARGE_INTEGER cb,
31                        DWORD type);
32  STDMETHOD(UnlockRegion)(ULARGE_INTEGER offset, ULARGE_INTEGER cb,
33                          DWORD type);
34  STDMETHOD(Stat)(STATSTG* statstg, DWORD flag);
35  STDMETHOD(Clone)(IStream** stream);
36
37 protected:
38  base::win::ScopedComPtr<IStream> delegate_;
39
40 private:
41  DISALLOW_COPY_AND_ASSIGN(StreamImpl);
42};
43
44#endif  // CHROME_FRAME_STREAM_IMPL_H_
45