OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_ | 5 #ifndef CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_ |
6 #define CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_ | 6 #define CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_ |
7 | 7 |
8 #include <urlmon.h> | 8 #include <urlmon.h> |
9 #include <atlbase.h> | 9 #include <atlbase.h> |
10 #include <atlcom.h> | 10 #include <atlcom.h> |
11 | 11 |
12 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
13 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
14 #include "chrome_frame/stream_impl.h" | 14 #include "chrome_frame/stream_impl.h" |
15 #include "net/base/upload_data.h" | 15 #include "net/base/upload_data.h" |
16 #include "net/base/upload_data_stream.h" | 16 #include "net/base/upload_data_stream.h" |
17 | 17 |
18 // Provides an IStream interface to the very different UploadDataStream | 18 // Provides an IStream interface to the very different UploadDataStream |
19 // implementation. | 19 // implementation. |
20 class UrlmonUploadDataStream : public CComObjectRootEx<CComMultiThreadModel>, | 20 class UrlmonUploadDataStream : public CComObjectRootEx<CComMultiThreadModel>, |
21 public StreamImpl { | 21 public StreamImpl { |
22 public: | 22 public: |
23 UrlmonUploadDataStream() {} | 23 UrlmonUploadDataStream() {} |
24 | 24 |
25 BEGIN_COM_MAP(UrlmonUploadDataStream) | 25 BEGIN_COM_MAP(UrlmonUploadDataStream) |
26 COM_INTERFACE_ENTRY(ISequentialStream) | 26 COM_INTERFACE_ENTRY(ISequentialStream) |
27 COM_INTERFACE_ENTRY(IStream) | 27 COM_INTERFACE_ENTRY(IStream) |
28 END_COM_MAP() | 28 END_COM_MAP() |
29 | 29 |
30 void Initialize(net::UploadData* upload_data); | 30 bool Initialize(net::UploadData* upload_data); |
31 | 31 |
32 // Partial implementation of IStream. | 32 // Partial implementation of IStream. |
33 STDMETHOD(Read)(void* pv, ULONG cb, ULONG* read); | 33 STDMETHOD(Read)(void* pv, ULONG cb, ULONG* read); |
34 STDMETHOD(Seek)(LARGE_INTEGER move, DWORD origin, ULARGE_INTEGER* new_pos); | 34 STDMETHOD(Seek)(LARGE_INTEGER move, DWORD origin, ULARGE_INTEGER* new_pos); |
35 STDMETHOD(Stat)(STATSTG *pstatstg, DWORD grfStatFlag); | 35 STDMETHOD(Stat)(STATSTG *pstatstg, DWORD grfStatFlag); |
36 | 36 |
37 private: | 37 private: |
38 scoped_refptr<net::UploadData> upload_data_; | 38 scoped_refptr<net::UploadData> upload_data_; |
39 scoped_ptr<net::UploadDataStream> request_body_stream_; | 39 scoped_ptr<net::UploadDataStream> request_body_stream_; |
40 }; | 40 }; |
41 | 41 |
42 #endif // CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_ | 42 #endif // CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_ |
OLD | NEW |