1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
|
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_FRAME_PROTOCOL_SINK_WRAP_H_
#define CHROME_FRAME_PROTOCOL_SINK_WRAP_H_
#include <exdisp.h>
#include <urlmon.h>
#include <atlbase.h>
#include <atlcom.h>
#include <map>
#include <string>
#include "base/basictypes.h"
#include "base/ref_counted.h"
#include "base/win/scoped_comptr.h"
#include "base/win/scoped_bstr.h"
#include "googleurl/src/gurl.h"
#include "chrome_frame/chrome_frame_delegate.h"
#include "chrome_frame/http_negotiate.h"
#include "chrome_frame/ie8_types.h"
#include "chrome_frame/utils.h"
#include "chrome_frame/vtable_patch_manager.h"
// Typedefs for IInternetProtocol and related methods that we patch.
typedef HRESULT (STDMETHODCALLTYPE* InternetProtocol_Start_Fn)(
IInternetProtocol* this_object, LPCWSTR url,
IInternetProtocolSink* prot_sink, IInternetBindInfo* bind_info,
DWORD flags, HANDLE_PTR reserved);
typedef HRESULT (STDMETHODCALLTYPE* InternetProtocol_Read_Fn)(
IInternetProtocol* this_object, void* buffer, ULONG size,
ULONG* size_read);
typedef HRESULT (STDMETHODCALLTYPE* InternetProtocol_StartEx_Fn)(
IInternetProtocolEx* this_object, IUri* uri,
IInternetProtocolSink* prot_sink, IInternetBindInfo* bind_info,
DWORD flags, HANDLE_PTR reserved);
typedef HRESULT (STDMETHODCALLTYPE* InternetProtocol_LockRequest_Fn)(
IInternetProtocol* this_object, DWORD options);
typedef HRESULT (STDMETHODCALLTYPE* InternetProtocol_UnlockRequest_Fn)(
IInternetProtocol* this_object);
class ProtData;
// A class to wrap protocol sink in IInternetProtocol::Start[Ex] for
// HTTP and HTTPS protocols.
//
// This is an alternative to a mime filter and we have to do this in order
// to inspect initial portion of HTML for 'chrome' meta tag and report
// a different mime type in that case.
//
// We implement several documented interfaces
// supported by the original sink provided by urlmon. There are a few
// undocumented interfaces that we have chosen not to implement
// but delegate simply the QI.
class ProtocolSinkWrap
: public CComObjectRootEx<CComMultiThreadModel>,
public IServiceProvider,
public UserAgentAddOn, // implements IHttpNegotiate
public IInternetProtocolSink {
public:
BEGIN_COM_MAP(ProtocolSinkWrap)
COM_INTERFACE_ENTRY(IServiceProvider)
COM_INTERFACE_ENTRY(IInternetProtocolSink)
COM_INTERFACE_BLIND_DELEGATE()
END_COM_MAP()
static base::win::ScopedComPtr<IInternetProtocolSink> CreateNewSink(
IInternetProtocolSink* sink, ProtData* prot_data);
// Apparently this has to be public, to satisfy COM_INTERFACE_BLIND_DELEGATE
IInternetProtocolSink* delegate() {
return delegate_;
}
protected:
ProtocolSinkWrap();
~ProtocolSinkWrap();
private:
// IInternetProtocolSink methods
STDMETHOD(Switch)(PROTOCOLDATA* protocol_data);
STDMETHOD(ReportProgress)(ULONG status_code, LPCWSTR status_text);
STDMETHOD(ReportData)(DWORD flags, ULONG progress, ULONG max_progress);
STDMETHOD(ReportResult)(HRESULT result, DWORD error, LPCWSTR result_text);
// IServiceProvider - return our HttpNegotiate or forward to delegate
STDMETHOD(QueryService)(REFGUID guidService, REFIID riid, void** ppvObject);
// Helpers.
HRESULT ObtainHttpNegotiate();
HRESULT ObtainServiceProvider();
// Remember original sink
base::win::ScopedComPtr<IInternetProtocolSink> delegate_;
base::win::ScopedComPtr<IServiceProvider> delegate_service_provider_;
scoped_refptr<ProtData> prot_data_;
DISALLOW_COPY_AND_ASSIGN(ProtocolSinkWrap);
};
class ProtData : public base::RefCounted<ProtData> {
public:
ProtData(IInternetProtocol* protocol, InternetProtocol_Read_Fn read_fun,
const wchar_t* url);
~ProtData();
HRESULT Read(void* buffer, ULONG size, ULONG* size_read);
HRESULT ReportProgress(IInternetProtocolSink* delegate,
ULONG status_code,
LPCWSTR status_text);
HRESULT ReportData(IInternetProtocolSink* delegate,
DWORD flags, ULONG progress, ULONG max_progress);
HRESULT ReportResult(IInternetProtocolSink* delegate, HRESULT result,
DWORD error, LPCWSTR result_text);
void UpdateUrl(const wchar_t* url);
static scoped_refptr<ProtData> DataFromProtocol(IInternetProtocol* protocol);
RendererType renderer_type() {
return renderer_type_;
}
// Valid only if renderer_type_ is CHROME.
const std::string& referrer() const {
return referrer_;
}
bool is_attach_external_tab_request() const {
return read_fun_ == NULL;
}
// Removes the mapping between the protocol and the ProtData.
void Invalidate();
private:
typedef std::map<IInternetProtocol*, ProtData*> ProtocolDataMap;
static ProtocolDataMap datamap_;
static Lock datamap_lock_;
// Url we are retrieving. Used for RendererTypeForUrl() only.
std::wstring url_;
// HTTP "Referrer" header if we detect are going to switch.
// We have to save and pass it to Chrome, so scripts can read it via DOM.
std::string referrer_;
// Our gate to IInternetProtocol::Read()
IInternetProtocol* protocol_;
InternetProtocol_Read_Fn read_fun_;
// What BINDSTATUS_MIMETYPEAVAILABLE and Co. tells us.
base::win::ScopedBstr suggested_mime_type_;
// At least one of the following has been received:
// BINDSTATUS_MIMETYPEAVAILABLE,
// MIMESTATUS_VERIFIEDMIMETYPEAVAILABLE
// BINDSTATUS_SERVER_MIMETYPEAVAILABLE
bool has_suggested_mime_type_;
// BINDSTATUS_SERVER_MIMETYPEAVAILABLE received, so we shall fire one.
bool has_server_mime_type_;
RendererType renderer_type_;
// Buffer for accumulated data including 1 extra for NULL-terminator
static const size_t kMaxContentSniffLength = 2 * 1024;
char buffer_[kMaxContentSniffLength + 1];
unsigned long buffer_size_; // NOLINT
unsigned long buffer_pos_; // NOLINT
HRESULT FillBuffer();
void SaveSuggestedMimeType(LPCWSTR status_text);
void FireSuggestedMimeType(IInternetProtocolSink* delegate);
void SaveReferrer(IInternetProtocolSink* delegate);
};
struct TransactionHooks {
void InstallHooks();
void RevertHooks();
};
DECLSPEC_SELECTANY struct TransactionHooks g_trans_hooks;
#endif // CHROME_FRAME_PROTOCOL_SINK_WRAP_H_
|
