OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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/ref_counted.h" | 12 #include "base/ref_counted.h" |
| 13 #include "base/win/atlcheck.h" |
13 #include "chrome_frame/stream_impl.h" | 14 #include "chrome_frame/stream_impl.h" |
14 #include "net/base/upload_data.h" | 15 #include "net/base/upload_data.h" |
15 #include "net/base/upload_data_stream.h" | 16 #include "net/base/upload_data_stream.h" |
16 | 17 |
17 // Provides an IStream interface to the very different UploadDataStream | 18 // Provides an IStream interface to the very different UploadDataStream |
18 // implementation. | 19 // implementation. |
19 class UrlmonUploadDataStream : public CComObjectRootEx<CComMultiThreadModel>, | 20 class UrlmonUploadDataStream : public CComObjectRootEx<CComMultiThreadModel>, |
20 public StreamImpl { | 21 public StreamImpl { |
21 public: | 22 public: |
22 UrlmonUploadDataStream() {} | 23 UrlmonUploadDataStream() {} |
23 | 24 |
24 BEGIN_COM_MAP(UrlmonUploadDataStream) | 25 BEGIN_COM_MAP(UrlmonUploadDataStream) |
25 COM_INTERFACE_ENTRY(ISequentialStream) | 26 COM_INTERFACE_ENTRY(ISequentialStream) |
26 COM_INTERFACE_ENTRY(IStream) | 27 COM_INTERFACE_ENTRY(IStream) |
27 END_COM_MAP() | 28 END_COM_MAP() |
28 | 29 |
29 void Initialize(net::UploadData* upload_data); | 30 void Initialize(net::UploadData* upload_data); |
30 | 31 |
31 // Partial implementation of IStream. | 32 // Partial implementation of IStream. |
32 STDMETHOD(Read)(void* pv, ULONG cb, ULONG* read); | 33 STDMETHOD(Read)(void* pv, ULONG cb, ULONG* read); |
33 STDMETHOD(Seek)(LARGE_INTEGER move, DWORD origin, ULARGE_INTEGER* new_pos); | 34 STDMETHOD(Seek)(LARGE_INTEGER move, DWORD origin, ULARGE_INTEGER* new_pos); |
34 STDMETHOD(Stat)(STATSTG *pstatstg, DWORD grfStatFlag); | 35 STDMETHOD(Stat)(STATSTG *pstatstg, DWORD grfStatFlag); |
35 | 36 |
36 private: | 37 private: |
37 scoped_refptr<net::UploadData> upload_data_; | 38 scoped_refptr<net::UploadData> upload_data_; |
38 scoped_ptr<net::UploadDataStream> request_body_stream_; | 39 scoped_ptr<net::UploadDataStream> request_body_stream_; |
39 }; | 40 }; |
40 | 41 |
41 #endif // CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_ | 42 #endif // CHROME_FRAME_URLMON_UPLOAD_DATA_STREAM_H_ |
OLD | NEW |