Home | History | Annotate | Download | only in chrome_frame 
      1 // Copyright (c) 2012 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_URLMON_UPLOAD_DATA_STREAM_H_
      6 #define CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_
      7 
      8 #include <urlmon.h>
      9 #include <atlbase.h>
     10 #include <atlcom.h>
     11 
     12 #include "base/memory/ref_counted.h"
     13 #include "base/memory/scoped_ptr.h"
     14 #include "chrome_frame/stream_impl.h"
     15 #include "net/base/upload_data.h"
     16 #include "net/base/upload_data_stream.h"
     17 
     18 // Provides an IStream interface to the very different UploadDataStream
     19 // implementation.
     20 class UrlmonUploadDataStream : public CComObjectRootEx<CComMultiThreadModel>,
     21                                public StreamImpl {
     22  public:
     23   UrlmonUploadDataStream() {}
     24 
     25   BEGIN_COM_MAP(UrlmonUploadDataStream)
     26     COM_INTERFACE_ENTRY(ISequentialStream)
     27     COM_INTERFACE_ENTRY(IStream)
     28   END_COM_MAP()
     29 
     30   bool Initialize(net::UploadData* upload_data);
     31 
     32   // Partial implementation of IStream.
     33   STDMETHOD(Read)(void* pv, ULONG cb, ULONG* read);
     34   STDMETHOD(Seek)(LARGE_INTEGER move, DWORD origin, ULARGE_INTEGER* new_pos);
     35   STDMETHOD(Stat)(STATSTG *pstatstg, DWORD grfStatFlag);
     36 
     37  private:
     38   scoped_refptr<net::UploadData> upload_data_;
     39   scoped_ptr<net::UploadDataStream> request_body_stream_;
     40 };
     41 
     42 #endif  // CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_
     43