Left: | ||
Right: |
OLD | NEW |
---|---|
1 #include "PluginStdAfx.h" | 1 #include "PluginStdAfx.h" |
2 | 2 |
3 #include "PluginWbPassThrough.h" | 3 #include "PluginWbPassThrough.h" |
4 #include "PluginClient.h" | 4 #include "PluginClient.h" |
5 #include "PluginClientFactory.h" | 5 #include "PluginClientFactory.h" |
6 #include "PluginFilter.h" | 6 #include "PluginFilter.h" |
7 #include "PluginSettings.h" | 7 #include "PluginSettings.h" |
8 #include "PluginClass.h" | 8 #include "PluginClass.h" |
9 #include "PluginSystem.h" | 9 #include "PluginSystem.h" |
10 #include <WinInet.h> | 10 #include <WinInet.h> |
11 #include "wtypes.h" | 11 #include "wtypes.h" |
12 #include "../shared/Utils.h" | |
12 | 13 |
13 namespace | 14 namespace |
14 { | 15 { |
15 std::string g_blockedByABPPage = "<!DOCTYPE html>" | 16 std::string g_blockedByABPPage = "<!DOCTYPE html>" |
16 "<html>" | 17 "<html>" |
17 "<body>" | 18 "<body>" |
18 "<!-- blocked by AdblockPlus -->" | 19 "<!-- blocked by AdblockPlus -->" |
19 "</body>" | 20 "</body>" |
20 "</html>"; | 21 "</html>"; |
22 | |
23 template <class T> | |
24 T ExtractHttpHeader(const T& allHeaders, const T& targetHeaderName, const T& d elimiter) | |
25 { | |
26 auto targetHeaderBeginsAt = allHeaders.find(targetHeaderName); | |
Felix Dahlke
2014/11/05 09:04:52
I still find it unintuitive that the colon is part
Oleksandr
2014/11/05 10:01:58
1. Colon is part of header name because this is a
sergei
2014/11/05 10:28:59
1. T is std::string or std::wstring. The variant w
Oleksandr
2014/11/05 16:42:50
Created https://issues.adblockplus.org/ticket/1529
| |
27 if (targetHeaderBeginsAt == T::npos) | |
28 { | |
29 return T(); | |
30 } | |
31 targetHeaderBeginsAt += targetHeaderName.length(); | |
32 auto targetHeaderEndsAt = allHeaders.find(delimiter, targetHeaderBeginsAt); | |
33 if (targetHeaderEndsAt == T::npos) | |
34 { | |
35 return T(); | |
36 } | |
37 return allHeaders.substr(targetHeaderBeginsAt, targetHeaderEndsAt - targetHe aderBeginsAt); | |
38 } | |
39 | |
40 std::string ExtractHttpAcceptHeader(IInternetProtocol* internetProtocol) | |
41 { | |
42 // Despite there being HTTP_QUERY_ACCEPT and other query info flags, they do n't work here, | |
43 // only HTTP_QUERY_RAW_HEADERS_CRLF | HTTP_QUERY_FLAG_REQUEST_HEADERS does w ork. | |
44 ATL::CComPtr<IWinInetHttpInfo> winInetHttpInfo; | |
45 HRESULT hr = internetProtocol->QueryInterface(&winInetHttpInfo); | |
46 if (FAILED(hr)) | |
47 { | |
48 return ""; | |
49 } | |
50 DWORD size = 0; | |
51 DWORD flags = 0; | |
52 DWORD queryOption = HTTP_QUERY_RAW_HEADERS_CRLF | HTTP_QUERY_FLAG_REQUEST_HE ADERS; | |
53 hr = winInetHttpInfo->QueryInfo(queryOption, /*buffer*/ nullptr, /*get size* / &size, &flags, /*reserved*/ 0); | |
54 if (FAILED(hr)) | |
55 { | |
56 return ""; | |
57 } | |
58 std::string buf(size, '\0'); | |
59 hr = winInetHttpInfo->QueryInfo(queryOption, &buf[0], &size, &flags, 0); | |
60 if (FAILED(hr)) | |
61 { | |
62 return ""; | |
63 } | |
64 return ExtractHttpHeader<std::string>(buf, "Accept:", "\r\n"); | |
65 } | |
21 } | 66 } |
22 | 67 |
23 WBPassthruSink::WBPassthruSink() | 68 WBPassthruSink::WBPassthruSink() |
24 : m_currentPositionOfSentPage(0) | 69 : m_currentPositionOfSentPage(0) |
25 , m_contentType(CFilter::EContentType::contentTypeAny) | 70 , m_contentType(CFilter::EContentType::contentTypeAny) |
26 , m_blockedInTransaction(false) | 71 , m_blockedInTransaction(false) |
27 { | 72 { |
28 } | 73 } |
29 | 74 |
30 int WBPassthruSink::GetContentTypeFromMimeType(const CString& mimeType) | 75 int WBPassthruSink::GetContentTypeFromMimeType(const CString& mimeType) |
(...skipping 87 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
118 //////////////////////////////////////////////////////////////////////////////// //////// | 163 //////////////////////////////////////////////////////////////////////////////// //////// |
119 //WBPassthruSink | 164 //WBPassthruSink |
120 //Monitor and/or cancel every request and responde | 165 //Monitor and/or cancel every request and responde |
121 //WB makes, including images, sounds, scripts, etc | 166 //WB makes, including images, sounds, scripts, etc |
122 //////////////////////////////////////////////////////////////////////////////// //////// | 167 //////////////////////////////////////////////////////////////////////////////// //////// |
123 HRESULT WBPassthruSink::OnStart(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSin k, | 168 HRESULT WBPassthruSink::OnStart(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSin k, |
124 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HAN DLE_PTR dwReserved, | 169 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HAN DLE_PTR dwReserved, |
125 IInternetProtocol* pTargetProtocol, bool& handle d) | 170 IInternetProtocol* pTargetProtocol, bool& handle d) |
126 { | 171 { |
127 m_pTargetProtocol = pTargetProtocol; | 172 m_pTargetProtocol = pTargetProtocol; |
128 bool isBlocked = false; | 173 return BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTargetProtocol); |
129 CString src = szUrl; | |
130 DEBUG_GENERAL(src); | |
131 CPluginClient::UnescapeUrl(src); | |
132 | |
133 // call the impl of the base class as soon as possible because it initializes the base class | |
134 // members, used by this method. It queries for the required interfaces. | |
135 HRESULT hr = BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwRese rved, pTargetProtocol); | |
136 if (FAILED(hr)) | |
137 { | |
138 return hr; | |
139 } | |
140 | |
141 CString mimeType; | |
142 if (pOIBindInfo) | |
143 { | |
144 ULONG resLen = 0; | |
145 | |
146 // Apparently IE will report random mime type if there's more then 1 in the list. | |
147 // So we get the whole list and just use the first one (top priority one) | |
148 LPOLESTR mime[10]; | |
149 pOIBindInfo->GetBindString(BINDSTRING_ACCEPT_MIMES, mime, 10, &resLen); | |
150 if (mime && resLen > 0) | |
151 { | |
152 mimeType.SetString(mime[0]); | |
153 } | |
154 LPOLESTR bindString = nullptr; | |
155 pOIBindInfo->GetBindString(BINDSTRING_FLAG_BIND_TO_OBJECT, &bindString, 1, & resLen); | |
156 LPOLESTR domainRetrieved = nullptr; | |
157 if (resLen == 0 || wcscmp(bindString, L"FALSE") == 0) | |
158 { | |
159 HRESULT hr = pOIBindInfo->GetBindString(BINDSTRING_XDR_ORIGIN, &domainRetr ieved, 1, &resLen); | |
160 if ((hr == S_OK) && domainRetrieved && (resLen > 0)) | |
161 { | |
162 m_boundDomain = domainRetrieved; | |
163 } | |
164 } | |
165 // We can obtain IBindCtx* here, but IEnumString obtained via IBindCtx::Enum ObjectParam | |
166 // does not return any parameter, so it's useless. | |
167 } | |
168 | |
169 CString cookie; | |
170 ULONG len1 = 2048; | |
171 ULONG len2 = 2048; | |
172 | |
173 CPluginTab* tab = CPluginClass::GetTab(::GetCurrentThreadId()); | |
174 CPluginClient* client = CPluginClient::GetInstance(); | |
175 | |
176 if (tab && client) | |
177 { | |
178 CString documentUrl = tab->GetDocumentUrl(); | |
179 // Page is identical to document => don't block | |
180 if (documentUrl == src) | |
181 { | |
182 // fall through | |
183 } | |
184 else if (CPluginSettings::GetInstance()->IsPluginEnabled() && !client->IsWhi telistedUrl(std::wstring(documentUrl))) | |
185 { | |
186 m_boundDomain = tab->GetDocumentUrl(); | |
187 m_contentType = CFilter::contentTypeAny; | |
188 if (tab != nullptr && tab->IsFrameCached(src)) | |
189 { | |
190 m_contentType = CFilter::contentTypeSubdocument; | |
191 } | |
192 else | |
193 { | |
194 m_contentType = GetContentType(mimeType, m_boundDomain, src); | |
195 } | |
196 } | |
197 } | |
198 | |
199 if (tab == nullptr) | |
200 { | |
201 m_contentType = GetContentType(mimeType, m_boundDomain, src); | |
202 } | |
203 | |
204 { | |
205 // Here is the heuristic which detects the requests issued by Flash.ocx. | |
206 // It turned out that the implementation from ''Flash.ocx'' (tested version is 15.0.0.152) | |
207 // returns quite minimal configuration in comparison with the implementation from Microsofts' | |
208 // libraries (see grfBINDF and bindInfo.dwOptions). The impl from MS often i ncludes something | |
209 // else. | |
210 ATL::CComPtr<IBindStatusCallback> bscb; | |
211 if (SUCCEEDED(QueryServiceFromClient(&bscb)) && !!bscb) | |
212 { | |
213 DWORD grfBINDF = 0; | |
214 BINDINFO bindInfo = {}; | |
215 bindInfo.cbSize = sizeof(bindInfo); | |
216 if (SUCCEEDED(bscb->GetBindInfo(&grfBINDF, &bindInfo)) | |
217 && (BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE| BINDF_PULLDATA) == grfBINDF | |
218 && (BINDINFO_OPTIONS_ENABLE_UTF8 | BINDINFO_OPTIONS_USE_IE_ENCODING) == bindInfo.dwOptions | |
219 ) | |
220 { | |
221 m_contentType = CFilter::EContentType::contentTypeObjectSubrequest; | |
222 } | |
223 } | |
224 } | |
225 | |
226 // The descision about EContentType::contentTypeAny is made later in | |
227 // WBPassthruSink::BeginningTransaction. Sometimes here we cannot detect the r equest type, but | |
228 // in WBPassthruSink::BeginningTransaction the header Accept is available whic h allows to | |
229 // obtain the "request type" in our terminology. | |
230 if (nullptr != client | |
231 && CFilter::EContentType::contentTypeAny != m_contentType | |
232 && client->ShouldBlock(static_cast<const wchar_t*>(src), m_contentType, m_bo undDomain, true)) | |
233 { | |
234 isBlocked = true; | |
235 } | |
236 | |
237 // For IE6 and earlier there is iframe back button issue, so avoid it. | |
238 if (isBlocked && client->GetIEVersion() > 6) | |
239 { | |
240 handled = true; | |
241 if (CFilter::EContentType::contentTypeImage == m_contentType) | |
242 { | |
243 // IE shows a cross that img is not loaded | |
244 return INET_E_REDIRECT_FAILED; | |
245 } | |
246 if (CFilter::EContentType::contentTypeSubdocument == m_contentType) | |
247 { | |
248 PassthroughAPP::CustomSinkStartPolicy<WBPassthru, WBPassthruSink>::GetProt ocol(this)->m_shouldSupplyCustomContent = true; | |
249 m_spInternetProtocolSink->ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE, L"t ext/html"); | |
250 m_spInternetProtocolSink->ReportData(BSCF_FIRSTDATANOTIFICATION, 0, static _cast<ULONG>(g_blockedByABPPage.size())); | |
251 return S_OK; | |
252 } | |
253 if (CFilter::EContentType::contentTypeScript == m_contentType) | |
254 { | |
255 m_spInternetProtocolSink->ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE, L"t ext/javascript"); | |
256 m_spInternetProtocolSink->ReportResult(INET_E_REDIRECTING, 301, L"data:"); | |
257 return INET_E_REDIRECT_FAILED; | |
258 } | |
259 if (CFilter::EContentType::contentTypeXmlHttpRequest == m_contentType) | |
260 { | |
261 m_spInternetProtocolSink->ReportResult(INET_E_REDIRECTING, 301, L"data:"); | |
262 return INET_E_REDIRECT_FAILED; | |
263 } | |
264 if (CFilter::EContentType::contentTypeAny != m_contentType) | |
265 { | |
266 m_spInternetProtocolSink->ReportResult(INET_E_REDIRECTING, 301, L"data:"); | |
267 return INET_E_REDIRECT_FAILED; | |
268 } | |
269 } | |
270 | |
271 return isBlocked ? S_FALSE : hr; | |
272 } | 174 } |
273 | 175 |
274 HRESULT WBPassthruSink::OnRead(void* pv, ULONG cb, ULONG* pcbRead) | 176 HRESULT WBPassthruSink::OnRead(void* pv, ULONG cb, ULONG* pcbRead) |
275 { | 177 { |
276 if (pv == nullptr) | 178 if (pv == nullptr) |
277 { | 179 { |
278 return E_POINTER; | 180 return E_POINTER; |
279 } | 181 } |
280 if (pcbRead == nullptr) | 182 if (pcbRead == nullptr) |
281 { | 183 { |
(...skipping 47 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
329 PROTOCOLDATA::grfFlags, eventually URLMon will turn around and call | 231 PROTOCOLDATA::grfFlags, eventually URLMon will turn around and call |
330 IInternetProtocol::Continue on the main thread. | 232 IInternetProtocol::Continue on the main thread. |
331 | 233 |
332 Or, if you happen to have a window handy that was created on the main | 234 Or, if you happen to have a window handy that was created on the main |
333 thread, you can post yourself a message. | 235 thread, you can post yourself a message. |
334 " | 236 " |
335 */ | 237 */ |
336 return m_spInternetProtocolSink ? m_spInternetProtocolSink->Switch(pProtocolDa ta) : E_UNEXPECTED; | 238 return m_spInternetProtocolSink ? m_spInternetProtocolSink->Switch(pProtocolDa ta) : E_UNEXPECTED; |
337 } | 239 } |
338 | 240 |
241 // This is the heuristic which detects the requests issued by Flash.ocx. | |
242 // It turned out that the implementation from ''Flash.ocx'' (tested version is 1 5.0.0.152) | |
243 // returns quite minimal configuration in comparison with the implementation fro m Microsofts' | |
244 // libraries (see grfBINDF and bindInfo.dwOptions). The impl from MS often inclu des something | |
245 // else. | |
246 bool WBPassthruSink::IsFlashRequest() | |
247 { | |
248 ATL::CComPtr<IBindStatusCallback> bscb; | |
249 if (SUCCEEDED(QueryServiceFromClient(&bscb)) && !!bscb) | |
250 { | |
251 DWORD grfBINDF = 0; | |
252 BINDINFO bindInfo = {}; | |
253 bindInfo.cbSize = sizeof(bindInfo); | |
254 if (SUCCEEDED(bscb->GetBindInfo(&grfBINDF, &bindInfo)) && | |
255 (BINDF_ASYNCHRONOUS | BINDF_ASYNCSTORAGE| BINDF_PULLDATA) == grfBINDF && | |
256 (BINDINFO_OPTIONS_ENABLE_UTF8 | BINDINFO_OPTIONS_USE_IE_ENCODING) == bindI nfo.dwOptions | |
257 ) | |
258 { | |
259 return true; | |
260 } | |
261 } | |
262 return false; | |
263 } | |
264 | |
339 STDMETHODIMP WBPassthruSink::BeginningTransaction(LPCWSTR szURL, LPCWSTR szHeade rs, DWORD dwReserved, LPWSTR* pszAdditionalHeaders) | 265 STDMETHODIMP WBPassthruSink::BeginningTransaction(LPCWSTR szURL, LPCWSTR szHeade rs, DWORD dwReserved, LPWSTR* pszAdditionalHeaders) |
340 { | 266 { |
341 if (pszAdditionalHeaders) | 267 if (pszAdditionalHeaders) |
342 { | 268 { |
343 *pszAdditionalHeaders = nullptr; | 269 *pszAdditionalHeaders = nullptr; |
344 } | 270 } |
271 std::wstring src = szURL; | |
272 DEBUG_GENERAL(ToCString(src)); | |
Felix Dahlke
2014/11/05 09:04:52
Hehe, still not quite what I meant :D Wouldn't it
Oleksandr
2014/11/05 10:01:58
That makes sense, yes.
| |
345 | 273 |
346 CPluginClient* client = nullptr; | 274 std::string acceptHeader = ExtractHttpAcceptHeader(m_spTargetProtocol); |
347 if (CFilter::EContentType::contentTypeAny == m_contentType && (client = CPlugi nClient::GetInstance())) | 275 m_contentType = GetContentTypeFromMimeType(ATL::CString(acceptHeader.c_str())) ; |
276 | |
277 CComPtr<IHttpNegotiate> httpNegotiate; | |
278 QueryServiceFromClient(&httpNegotiate); | |
279 // This fills the pszAdditionalHeaders with more headers. One of which is the Referer header, which we need. | |
280 // There doesn't seem to be any other way to get this header before the reques t has been made. | |
281 HRESULT nativeHr = httpNegotiate ? httpNegotiate->BeginningTransaction(szURL, szHeaders, dwReserved, pszAdditionalHeaders) : S_OK; | |
282 | |
283 if (*pszAdditionalHeaders != 0) | |
348 { | 284 { |
349 auto acceptHeader = [&]() -> std::string | 285 m_boundDomain = ExtractHttpHeader<std::wstring>(*pszAdditionalHeaders, L"Ref erer:", L"\n").c_str(); |
286 } | |
287 m_boundDomain = TrimString(m_boundDomain); | |
288 CPluginTab* tab = CPluginClass::GetTab(::GetCurrentThreadId()); | |
289 CPluginClient* client = CPluginClient::GetInstance(); | |
290 | |
291 if (tab && client) | |
292 { | |
293 CString documentUrl = tab->GetDocumentUrl(); | |
294 // Page is identical to document => don't block | |
295 if (documentUrl == ToCString(src)) | |
350 { | 296 { |
351 // Despite there is HTTP_QUERY_ACCEPT and other query info flags, they don 't work here, | 297 return nativeHr; |
352 // only HTTP_QUERY_RAW_HEADERS_CRLF | HTTP_QUERY_FLAG_REQUEST_HEADERS does dork. | 298 } |
353 ATL::CComPtr<IWinInetHttpInfo> winInetHttpInfo; | 299 else if (CPluginSettings::GetInstance()->IsPluginEnabled() && !client->IsWhi telistedUrl(std::wstring(documentUrl))) |
354 HRESULT hr = m_spTargetProtocol->QueryInterface(&winInetHttpInfo); | 300 { |
355 if(FAILED(hr)) | 301 if (tab->IsFrameCached(ToCString(src))) |
356 { | 302 { |
357 return ""; | 303 m_contentType = CFilter::contentTypeSubdocument; |
358 } | 304 } |
359 DWORD size = 0; | |
360 DWORD flags = 0; | |
361 hr = winInetHttpInfo->QueryInfo(HTTP_QUERY_RAW_HEADERS_CRLF | HTTP_QUERY_F LAG_REQUEST_HEADERS, | |
362 /*buffer*/nullptr, /* get size */&size, &flags, /*reserved*/ 0); | |
363 if(FAILED(hr)) | |
364 { | |
365 return ""; | |
366 } | |
367 std::string buf(size, '\0'); | |
368 hr = winInetHttpInfo->QueryInfo(HTTP_QUERY_RAW_HEADERS_CRLF | HTTP_QUERY_F LAG_REQUEST_HEADERS, | |
369 &buf[0], &size, &flags, 0); | |
370 if(FAILED(hr)) | |
371 { | |
372 return ""; | |
373 } | |
374 char acceptHeader[] = "Accept:"; | |
375 auto acceptHeaderBeginsAt = buf.find(acceptHeader); | |
376 if (std::string::npos == acceptHeaderBeginsAt) | |
377 { | |
378 return ""; | |
379 } | |
380 acceptHeaderBeginsAt += sizeof(acceptHeader); | |
381 auto acceptHeaderEndsAt = buf.find("\n", acceptHeaderBeginsAt); | |
382 if (std::string::npos == acceptHeaderEndsAt) | |
383 { | |
384 return ""; | |
385 } | |
386 return buf.substr(acceptHeaderBeginsAt, acceptHeaderEndsAt - acceptHeaderB eginsAt); | |
387 }(); | |
388 m_contentType = GetContentTypeFromMimeType(ATL::CString(acceptHeader.c_str() )); | |
389 bool isBlocked = client->ShouldBlock(szURL, m_contentType, m_boundDomain, /* debug flag but must be set*/true); | |
390 if (isBlocked) | |
391 { | |
392 m_blockedInTransaction = true; | |
393 return E_ABORT; | |
394 } | 305 } |
395 } | 306 } |
396 CComPtr<IHttpNegotiate> spHttpNegotiate; | 307 |
397 QueryServiceFromClient(&spHttpNegotiate); | 308 if (IsFlashRequest()) |
398 return spHttpNegotiate ? spHttpNegotiate->BeginningTransaction(szURL, szHeader s,dwReserved, pszAdditionalHeaders) : S_OK; | 309 { |
310 m_contentType = CFilter::EContentType::contentTypeObjectSubrequest; | |
311 } | |
312 | |
313 m_blockedInTransaction = client->ShouldBlock(szURL, m_contentType, m_boundDoma in, /*debug flag but must be set*/true); | |
314 if (m_blockedInTransaction) | |
315 { | |
316 return E_ABORT; | |
317 } | |
318 return nativeHr; | |
399 } | 319 } |
400 | 320 |
401 STDMETHODIMP WBPassthruSink::OnResponse(DWORD dwResponseCode, LPCWSTR szResponse Headers, LPCWSTR szRequestHeaders, LPWSTR *pszAdditionalRequestHeaders) | 321 STDMETHODIMP WBPassthruSink::OnResponse(DWORD dwResponseCode, LPCWSTR szResponse Headers, LPCWSTR szRequestHeaders, LPWSTR *pszAdditionalRequestHeaders) |
402 { | 322 { |
403 if (pszAdditionalRequestHeaders) | 323 if (pszAdditionalRequestHeaders) |
404 { | 324 { |
405 *pszAdditionalRequestHeaders = 0; | 325 *pszAdditionalRequestHeaders = 0; |
406 } | 326 } |
407 | 327 |
408 CComPtr<IHttpNegotiate> spHttpNegotiate; | 328 CComPtr<IHttpNegotiate> spHttpNegotiate; |
(...skipping 15 matching lines...) Expand all Loading... | |
424 // Current method is called by the original protocol implementation and we a re intercepting the | 344 // Current method is called by the original protocol implementation and we a re intercepting the |
425 // call here and eating it, we will call the proper ReportResult later by ou rself. | 345 // call here and eating it, we will call the proper ReportResult later by ou rself. |
426 return S_OK; | 346 return S_OK; |
427 } | 347 } |
428 return BaseClass::ReportResult(hrResult, dwError, szResult); | 348 return BaseClass::ReportResult(hrResult, dwError, szResult); |
429 } | 349 } |
430 | 350 |
431 | 351 |
432 WBPassthru::WBPassthru() | 352 WBPassthru::WBPassthru() |
433 : m_shouldSupplyCustomContent(false) | 353 : m_shouldSupplyCustomContent(false) |
434 , m_hasOriginalStartCalled(false) | |
435 { | 354 { |
436 } | 355 } |
437 | 356 |
438 STDMETHODIMP WBPassthru::Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink , | 357 STDMETHODIMP WBPassthru::Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink , |
439 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) | 358 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) |
440 { | 359 { |
441 ATLASSERT(m_spInternetProtocol != 0); | 360 ATLASSERT(m_spInternetProtocol != 0); |
442 if (!m_spInternetProtocol) | 361 if (!m_spInternetProtocol) |
443 { | 362 { |
444 return E_UNEXPECTED; | 363 return E_UNEXPECTED; |
445 } | 364 } |
446 | 365 |
447 return OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, m_spInterne tProtocol); | 366 return OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, m_spInterne tProtocol); |
448 } | 367 } |
449 | 368 |
450 STDMETHODIMP WBPassthru::Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [o ut] */ ULONG *pcbRead) | 369 STDMETHODIMP WBPassthru::Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [o ut] */ ULONG *pcbRead) |
451 { | 370 { |
452 WBPassthruSink* pSink = GetSink(); | 371 WBPassthruSink* pSink = GetSink(); |
453 return pSink->OnRead(pv, cb, pcbRead); | 372 return pSink->OnRead(pv, cb, pcbRead); |
454 } | 373 } |
455 | 374 |
456 STDMETHODIMP WBPassthru::LockRequest(/* [in] */ DWORD options) | 375 STDMETHODIMP WBPassthru::LockRequest(/* [in] */ DWORD options) |
457 { | 376 { |
458 if (!m_hasOriginalStartCalled) | |
459 { | |
460 return S_OK; | |
461 } | |
462 return BaseClass::LockRequest(options); | 377 return BaseClass::LockRequest(options); |
463 } | 378 } |
464 | 379 |
465 STDMETHODIMP WBPassthru::UnlockRequest() | 380 STDMETHODIMP WBPassthru::UnlockRequest() |
466 { | 381 { |
467 if (!m_hasOriginalStartCalled) | |
468 { | |
469 return S_OK; | |
470 } | |
471 return BaseClass::UnlockRequest(); | 382 return BaseClass::UnlockRequest(); |
472 } | 383 } |
OLD | NEW |