Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code
Issues
All Issues

Side by Side Diff: src/plugin/PluginWbPassThrough.cpp

Issue 5231751647461376: Issue 1264 - Generate custom response for subdocument on subdomain request (Closed)
Patch Set: move opening braces on the new line Created Sept. 8, 2014, 8:42 a.m.
Left:
Right:
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View unified diff | Download patch
« no previous file with comments | « src/plugin/PluginWbPassThrough.h ('k') | src/plugin/SinkPolicy.inl » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
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 #ifdef SUPPORT_FILTER 6 #ifdef SUPPORT_FILTER
7 #include "PluginFilter.h" 7 #include "PluginFilter.h"
8 #endif 8 #endif
9 #include "PluginSettings.h" 9 #include "PluginSettings.h"
10 #include "PluginClass.h" 10 #include "PluginClass.h"
11 #include "PluginSystem.h" 11 #include "PluginSystem.h"
12 12
13 #include "wtypes.h" 13 #include "wtypes.h"
14 14
15 EXTERN_C IMAGE_DOS_HEADER __ImageBase; 15 EXTERN_C IMAGE_DOS_HEADER __ImageBase;
16 16
17 17 namespace
18 {
19 std::string g_myPageBlocked = "<!DOCTYPE html>"
20 "<html>"
21 "<body>"
22 "<!-- blocked by AdblockPlus -->"
23 "</body>"
24 "</html>";
25 }
18 26
19 int WBPassthruSink::GetContentTypeFromMimeType(CString mimeType) 27 int WBPassthruSink::GetContentTypeFromMimeType(CString mimeType)
20 { 28 {
21 if (mimeType.Find(L"image/") >= 0) 29 if (mimeType.Find(L"image/") >= 0)
22 { 30 {
23 return CFilter::contentTypeImage; 31 return CFilter::contentTypeImage;
24 } 32 }
25 if (mimeType.Find(L"text/css") >= 0) 33 if (mimeType.Find(L"text/css") >= 0)
26 { 34 {
27 return CFilter::contentTypeStyleSheet; 35 return CFilter::contentTypeStyleSheet;
(...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after
108 return contentType; 116 return contentType;
109 } 117 }
110 118
111 //////////////////////////////////////////////////////////////////////////////// //////// 119 //////////////////////////////////////////////////////////////////////////////// ////////
112 //WBPassthruSink 120 //WBPassthruSink
113 //Monitor and/or cancel every request and responde 121 //Monitor and/or cancel every request and responde
114 //WB makes, including images, sounds, scripts, etc 122 //WB makes, including images, sounds, scripts, etc
115 //////////////////////////////////////////////////////////////////////////////// //////// 123 //////////////////////////////////////////////////////////////////////////////// ////////
116 HRESULT WBPassthruSink::OnStart(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSin k, 124 HRESULT WBPassthruSink::OnStart(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSin k,
117 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HAN DLE_PTR dwReserved, 125 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HAN DLE_PTR dwReserved,
118 IInternetProtocol* pTargetProtocol) 126 IInternetProtocol* pTargetProtocol, bool& handle d)
119 { 127 {
120
121 m_pTargetProtocol = pTargetProtocol; 128 m_pTargetProtocol = pTargetProtocol;
122 bool isBlocked = false; 129 bool isBlocked = false;
123 m_shouldBlock = false; 130 m_currentPositionOfSentPage = 0;
124 m_lastDataReported = false;
125 CString src; 131 CString src;
126 src.Append(szUrl); 132 src.Append(szUrl);
127 DEBUG_GENERAL(src); 133 DEBUG_GENERAL(src);
128 CPluginClient::UnescapeUrl(src); 134 CPluginClient::UnescapeUrl(src);
129 135
130 CString boundDomain; 136 CString boundDomain;
131 CString mimeType; 137 CString mimeType;
132 LPOLESTR mime[10]; 138 LPOLESTR mime[10];
133 if (pOIBindInfo) 139 if (pOIBindInfo)
134 { 140 {
(...skipping 95 matching lines...) Expand 10 before | Expand all | Expand 10 after
230 } 236 }
231 else 237 else
232 { 238 {
233 CPluginDebug::DebugResultIgnoring(type, src, boundDomain); 239 CPluginDebug::DebugResultIgnoring(type, src, boundDomain);
234 } 240 }
235 #endif 241 #endif
236 242
237 //Fixes the iframe back button issue 243 //Fixes the iframe back button issue
238 if (client->GetIEVersion() > 6) 244 if (client->GetIEVersion() > 6)
239 { 245 {
240 if ((contentType == CFilter::contentTypeImage) && (isBlocked)) 246 if (contentType == CFilter::contentTypeImage && isBlocked)
241 { 247 {
242 m_shouldBlock = true;
243 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol); 248 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol);
244 249 handled = true;
250 // IE shows a cross that img is not loaded
245 return INET_E_REDIRECT_FAILED; 251 return INET_E_REDIRECT_FAILED;
246
247 } 252 }
248 if (((contentType == CFilter::contentTypeSubdocument))&& (isBlocked)) 253 if (contentType == CFilter::contentTypeSubdocument && isBlocked)
249 { 254 {
250 m_shouldBlock = true; 255 PassthroughAPP::CustomSinkStartPolicy<WBPassthru, WBPassthruSink>::GetProt ocol(this)->m_shouldSupplyCustomContent = true;
251 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol); 256 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol);
252 257
253 m_spInternetProtocolSink->ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE, L"t ext/html"); 258 m_spInternetProtocolSink->ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE, L"t ext/html");
254 259 m_spInternetProtocolSink->ReportData(BSCF_FIRSTDATANOTIFICATION, 0, static _cast<ULONG>(g_myPageBlocked.size()));
255 //Here we check if we are running on Windows 8 Consumer Preview. 260 handled = true;
256 //For some reason on that environment the next line causes IE to crash 261 return S_OK;
257 if (CPluginSettings::GetInstance()->GetWindowsBuildNumber() != 8250)
258 {
259 m_spInternetProtocolSink->ReportResult(INET_E_REDIRECT_FAILED, 0, szUrl) ;
260 }
261
262 return INET_E_REDIRECT_FAILED;
263 } 262 }
264 if (((contentType == CFilter::contentTypeScript)) && (isBlocked)) 263 if (contentType == CFilter::contentTypeScript && isBlocked)
265 { 264 {
266 m_shouldBlock = true;
267 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol); 265 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol);
268 m_spInternetProtocolSink->ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE, L"t ext/javascript"); 266 m_spInternetProtocolSink->ReportProgress(BINDSTATUS_MIMETYPEAVAILABLE, L"t ext/javascript");
269 m_spInternetProtocolSink->ReportResult(INET_E_REDIRECTING, 301, L"data:"); 267 m_spInternetProtocolSink->ReportResult(INET_E_REDIRECTING, 301, L"data:");
268 handled = true;
270 return INET_E_REDIRECT_FAILED; 269 return INET_E_REDIRECT_FAILED;
271 } 270 }
272 if ((contentType == CFilter::contentTypeXmlHttpRequest) && (isBlocked)) 271 if (contentType == CFilter::contentTypeXmlHttpRequest && isBlocked)
273 { 272 {
274 m_shouldBlock = true;
275 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol); 273 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol);
276 m_spInternetProtocolSink->ReportResult(INET_E_REDIRECTING, 301, L"data:"); 274 m_spInternetProtocolSink->ReportResult(INET_E_REDIRECTING, 301, L"data:");
275 handled = true;
277 return INET_E_REDIRECT_FAILED; 276 return INET_E_REDIRECT_FAILED;
278 } 277 }
279 if ((isBlocked)) 278 if (isBlocked)
280 { 279 {
281 m_shouldBlock = true;
282 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol); 280 BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, pTa rgetProtocol);
283 m_spInternetProtocolSink->ReportResult(S_FALSE, 0, L""); 281 m_spInternetProtocolSink->ReportResult(S_FALSE, 0, L"");
284 282 handled = true;
285 return INET_E_REDIRECT_FAILED; 283 return INET_E_REDIRECT_FAILED;
286 } 284 }
287 } 285 }
288 #endif // SUPPORT_FILTER 286 #endif // SUPPORT_FILTER
289 287
290 return isBlocked ? S_FALSE : BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInf o, grfPI, dwReserved, pTargetProtocol); 288 return isBlocked ? S_FALSE : BaseClass::OnStart(szUrl, pOIProtSink, pOIBindInf o, grfPI, dwReserved, pTargetProtocol);
291 } 289 }
292 290
293 291
294 HRESULT WBPassthruSink::Read(void *pv, ULONG cb, ULONG* pcbRead) 292 HRESULT WBPassthruSink::OnRead(void* pv, ULONG cb, ULONG* pcbRead)
295 { 293 {
296 if (m_shouldBlock) 294 if (nullptr == pv)
297 { 295 {
298 *pcbRead = 0; 296 return E_POINTER;
299 if (!m_lastDataReported) 297 }
298 if (nullptr == pcbRead)
299 {
300 return E_POINTER;
301 }
302 *pcbRead = 0;
303
304 if (PassthroughAPP::CustomSinkStartPolicy<WBPassthru, WBPassthruSink>::GetProt ocol(this)->m_shouldSupplyCustomContent)
305 {
306 ULONG myPageSize = static_cast<ULONG>(g_myPageBlocked.size());
307 auto positionGrow = std::min<ULONG>(cb, static_cast<ULONG>(g_myPageBlocked.s ize() - m_currentPositionOfSentPage));
308 if (0 == positionGrow) {
309 return S_FALSE;
310 }
311 std::copy(g_myPageBlocked.begin(), g_myPageBlocked.begin() + positionGrow,
312 stdext::make_checked_array_iterator(static_cast<char*>(pv), cb));
313 *pcbRead = positionGrow;
314 m_currentPositionOfSentPage += positionGrow;
315
316 if (m_spInternetProtocolSink)
300 { 317 {
301 if (cb <= 1) 318 m_spInternetProtocolSink->ReportData(BSCF_INTERMEDIATEDATANOTIFICATION,
302 { 319 static_cast<ULONG>(m_currentPositionOfSentPage), myPageSize);
303 //IE must've gone nuts if this happened, but let's be cool about it and report we have no more data 320 }
304 m_spInternetProtocolSink->ReportResult(S_FALSE, 0, NULL); 321 if (myPageSize == m_currentPositionOfSentPage && m_spInternetProtocolSink)
305 return S_FALSE; 322 {
306 } 323 m_spInternetProtocolSink->ReportData(BSCF_DATAFULLYAVAILABLE, myPageSize, myPageSize);
307 *pcbRead = 1; 324 m_spInternetProtocolSink->ReportResult(S_OK, 0, nullptr);
308 memcpy(pv, " ", 1);
309
310 if (m_spInternetProtocolSink != NULL)
311 {
312 m_spInternetProtocolSink->ReportResult(S_OK, 0, NULL);
313 }
314 m_lastDataReported = true;
315 m_shouldBlock = false;
316 return S_OK;
317 } 325 }
318 return S_OK; 326 return S_OK;
319 } 327 }
320 else 328 return m_pTargetProtocol->Read(pv, cb, pcbRead);
321 {
322
323 return m_pTargetProtocol->Read(pv, cb, pcbRead);
324 }
325 return S_OK;
326 } 329 }
327 STDMETHODIMP WBPassthruSink::Switch( 330 STDMETHODIMP WBPassthruSink::Switch(
328 /* [in] */ PROTOCOLDATA *pProtocolData) 331 /* [in] */ PROTOCOLDATA *pProtocolData)
329 { 332 {
330 ATLASSERT(m_spInternetProtocolSink != 0); 333 ATLASSERT(m_spInternetProtocolSink != 0);
331 334
332 /* 335 /*
333 From Igor Tandetnik "itandetnik@mvps.org" 336 From Igor Tandetnik "itandetnik@mvps.org"
334 " 337 "
335 Beware multithreading. URLMon has this nasty habit of spinning worker 338 Beware multithreading. URLMon has this nasty habit of spinning worker
336 threads, not even bothering to initialize COM on them, and calling APP 339 threads, not even bothering to initialize COM on them, and calling APP
337 methods on those threads. If you try to raise COM events directly from 340 methods on those threads. If you try to raise COM events directly from
338 such a thread, bad things happen (random crashes, events being lost). 341 such a thread, bad things happen (random crashes, events being lost).
339 You are only guaranteed to be on the main STA thread in two cases. 342 You are only guaranteed to be on the main STA thread in two cases.
340 First, in methods of interfaces that were obtained with 343 First, in methods of interfaces that were obtained with
341 IServiceProvider, such as IHttpNegotiage::BeginningTransaction or 344 IServiceProvider, such as IHttpNegotiage::BeginningTransaction or
342 IAuthenticate::Authenticate. Second, you can call 345 IAuthenticate::Authenticate. Second, you can call
343 IInternetProtocolSink::Switch with PD_FORCE_SWITCH flag in 346 IInternetProtocolSink::Switch with PD_FORCE_SWITCH flag in
344 PROTOCOLDATA::grfFlags, eventually URLMon will turn around and call 347 PROTOCOLDATA::grfFlags, eventually URLMon will turn around and call
345 IInternetProtocol::Continue on the main thread. 348 IInternetProtocol::Continue on the main thread.
346 349
347 Or, if you happen to have a window handy that was created on the main 350 Or, if you happen to have a window handy that was created on the main
348 thread, you can post yourself a message. 351 thread, you can post yourself a message.
349 " 352 "
350 */ 353 */
351 return m_spInternetProtocolSink ? m_spInternetProtocolSink->Switch(pProtocolDa ta) : E_UNEXPECTED; 354 return m_spInternetProtocolSink ? m_spInternetProtocolSink->Switch(pProtocolDa ta) : E_UNEXPECTED;
352 } 355 }
353 356
354
355 STDMETHODIMP WBPassthruSink::BeginningTransaction(LPCWSTR szURL, LPCWSTR szHeade rs, DWORD dwReserved, LPWSTR *pszAdditionalHeaders) 357 STDMETHODIMP WBPassthruSink::BeginningTransaction(LPCWSTR szURL, LPCWSTR szHeade rs, DWORD dwReserved, LPWSTR *pszAdditionalHeaders)
356 { 358 {
357 if (pszAdditionalHeaders) 359 if (pszAdditionalHeaders)
358 { 360 {
359 *pszAdditionalHeaders = 0; 361 *pszAdditionalHeaders = 0;
360 } 362 }
361 363
362 CComPtr<IHttpNegotiate> spHttpNegotiate; 364 CComPtr<IHttpNegotiate> spHttpNegotiate;
363 QueryServiceFromClient(&spHttpNegotiate); 365 QueryServiceFromClient(&spHttpNegotiate);
364 return spHttpNegotiate ? spHttpNegotiate->BeginningTransaction(szURL, szHeader s,dwReserved, pszAdditionalHeaders) : S_OK; 366 return spHttpNegotiate ? spHttpNegotiate->BeginningTransaction(szURL, szHeader s,dwReserved, pszAdditionalHeaders) : S_OK;
(...skipping 10 matching lines...) Expand all
375 QueryServiceFromClient(&spHttpNegotiate); 377 QueryServiceFromClient(&spHttpNegotiate);
376 378
377 return spHttpNegotiate ? spHttpNegotiate->OnResponse(dwResponseCode, szRespons eHeaders, szRequestHeaders, pszAdditionalRequestHeaders) : S_OK; 379 return spHttpNegotiate ? spHttpNegotiate->OnResponse(dwResponseCode, szRespons eHeaders, szRequestHeaders, pszAdditionalRequestHeaders) : S_OK;
378 } 380 }
379 381
380 STDMETHODIMP WBPassthruSink::ReportProgress(ULONG ulStatusCode, LPCWSTR szStatus Text) 382 STDMETHODIMP WBPassthruSink::ReportProgress(ULONG ulStatusCode, LPCWSTR szStatus Text)
381 { 383 {
382 return m_spInternetProtocolSink ? m_spInternetProtocolSink->ReportProgress(ulS tatusCode, szStatusText) : S_OK; 384 return m_spInternetProtocolSink ? m_spInternetProtocolSink->ReportProgress(ulS tatusCode, szStatusText) : S_OK;
383 } 385 }
384 386
387 WBPassthru::WBPassthru()
388 : m_shouldSupplyCustomContent(false)
389 {
390 }
385 391
386 STDMETHODIMP WBPassthru::Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink , 392 STDMETHODIMP WBPassthru::Start(LPCWSTR szUrl, IInternetProtocolSink *pOIProtSink ,
387 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved) 393 IInternetBindInfo *pOIBindInfo, DWORD grfPI, HANDLE_PTR dwReserved)
388 { 394 {
389 ATLASSERT(m_spInternetProtocol != 0); 395 ATLASSERT(m_spInternetProtocol != 0);
390 if (!m_spInternetProtocol) 396 if (!m_spInternetProtocol)
391 { 397 {
392 return E_UNEXPECTED; 398 return E_UNEXPECTED;
393 } 399 }
394 400
395 return OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, 401 return OnStart(szUrl, pOIProtSink, pOIBindInfo, grfPI, dwReserved, m_spInterne tProtocol);
396 dwReserved, m_spInternetProtocol);
397 } 402 }
398 403
399 STDMETHODIMP WBPassthru::Read(»/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [out] */ ULONG *pcbRead) 404 STDMETHODIMP WBPassthru::Read(/* [in, out] */ void *pv,/* [in] */ ULONG cb,/* [o ut] */ ULONG *pcbRead)
400 { 405 {
401 406 WBPassthruSink* pSink = GetSink();
402 WBPassthruSink* pSink = GetSink(); 407 return pSink->OnRead(pv, cb, pcbRead);
403 return pSink->Read(pv, cb, pcbRead); 408 }
404 } 409
410 STDMETHODIMP WBPassthru::LockRequest(/* [in] */ DWORD options)
411 {
412 if (m_shouldSupplyCustomContent)
413 {
414 return S_OK;
415 }
416 return PassthroughAPP::CInternetProtocol<WBStartPolicy>::LockRequest(options);
417 }
418
419 STDMETHODIMP WBPassthru::UnlockRequest()
420 {
421 if (m_shouldSupplyCustomContent)
422 {
423 return S_OK;
424 }
425 return PassthroughAPP::CInternetProtocol<WBStartPolicy>::UnlockRequest();
426 }
OLDNEW
« no previous file with comments | « src/plugin/PluginWbPassThrough.h ('k') | src/plugin/SinkPolicy.inl » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld