Index: chrome_frame/urlmon_bind_status_callback.cc |
diff --git a/chrome_frame/urlmon_bind_status_callback.cc b/chrome_frame/urlmon_bind_status_callback.cc |
deleted file mode 100644 |
index 6001538a1b75c87b7549dfc9d591e5b3b4acde2a..0000000000000000000000000000000000000000 |
--- a/chrome_frame/urlmon_bind_status_callback.cc |
+++ /dev/null |
@@ -1,430 +0,0 @@ |
-// Copyright (c) 2011 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. |
- |
-#include "chrome_frame/urlmon_bind_status_callback.h" |
- |
-#include <mshtml.h> |
-#include <shlguid.h> |
- |
-#include "base/logging.h" |
-#include "base/strings/string_util.h" |
-#include "base/strings/stringprintf.h" |
-#include "base/strings/utf_string_conversions.h" |
-#include "base/threading/platform_thread.h" |
-#include "chrome_frame/bind_context_info.h" |
-#include "chrome_frame/chrome_tab.h" |
-#include "chrome_frame/exception_barrier.h" |
-#include "chrome_frame/urlmon_moniker.h" |
- |
- |
-// A helper to given feed data to the specified |bscb| using |
-// CacheStream instance. |
-HRESULT CacheStream::BSCBFeedData(IBindStatusCallback* bscb, const char* data, |
- size_t size, CLIPFORMAT clip_format, |
- size_t flags, bool eof) { |
- if (!bscb) { |
- NOTREACHED() << "invalid IBindStatusCallback"; |
- return E_INVALIDARG; |
- } |
- |
- // We can't use a CComObjectStackEx here since mshtml will hold |
- // onto the stream pointer. |
- CComObject<CacheStream>* cache_stream = NULL; |
- HRESULT hr = CComObject<CacheStream>::CreateInstance(&cache_stream); |
- if (FAILED(hr)) { |
- NOTREACHED(); |
- return hr; |
- } |
- |
- scoped_refptr<CacheStream> cache_ref = cache_stream; |
- hr = cache_stream->Initialize(data, size, eof); |
- if (FAILED(hr)) |
- return hr; |
- |
- FORMATETC format_etc = { clip_format, NULL, DVASPECT_CONTENT, -1, |
- TYMED_ISTREAM }; |
- STGMEDIUM medium = {0}; |
- medium.tymed = TYMED_ISTREAM; |
- medium.pstm = cache_stream; |
- |
- hr = bscb->OnDataAvailable(flags, size, &format_etc, &medium); |
- return hr; |
-} |
- |
-HRESULT CacheStream::Initialize(const char* cache, size_t size, bool eof) { |
- position_ = 0; |
- eof_ = eof; |
- |
- HRESULT hr = S_OK; |
- cache_.reset(new char[size]); |
- if (cache_.get()) { |
- memcpy(cache_.get(), cache, size); |
- size_ = size; |
- } else { |
- DLOG(ERROR) << "failed to allocate cache stream."; |
- hr = E_OUTOFMEMORY; |
- } |
- |
- return hr; |
-} |
- |
-// Read is the only call that we expect. Return E_PENDING if there |
-// is no more data to serve. Otherwise this will result in a |
-// read with 0 bytes indicating that no more data is available. |
-STDMETHODIMP CacheStream::Read(void* pv, ULONG cb, ULONG* read) { |
- if (!pv || !read) |
- return E_INVALIDARG; |
- |
- if (!cache_.get()) { |
- *read = 0; |
- return S_FALSE; |
- } |
- |
- // Default to E_PENDING to signal that this is a partial data. |
- HRESULT hr = eof_ ? S_FALSE : E_PENDING; |
- if (position_ < size_) { |
- *read = std::min(size_ - position_, size_t(cb)); |
- memcpy(pv, cache_ .get() + position_, *read); |
- position_ += *read; |
- hr = S_OK; |
- } |
- |
- return hr; |
-} |
- |
- |
-///////////////////////////////////////////////////////////////////// |
- |
-HRESULT SniffData::InitializeCache(const std::wstring& url) { |
- url_ = url; |
- renderer_type_ = UNDETERMINED; |
- |
- const int kInitialSize = 4 * 1024; // 4K |
- HGLOBAL mem = GlobalAlloc(0, kInitialSize); |
- DCHECK(mem) << "GlobalAlloc failed: " << GetLastError(); |
- |
- HRESULT hr = CreateStreamOnHGlobal(mem, TRUE, cache_.Receive()); |
- if (SUCCEEDED(hr)) { |
- ULARGE_INTEGER size = {0}; |
- cache_->SetSize(size); |
- } else { |
- DLOG(ERROR) << "CreateStreamOnHGlobal failed: " << hr; |
- } |
- |
- return hr; |
-} |
- |
-HRESULT SniffData::ReadIntoCache(IStream* stream, bool force_determination) { |
- if (!stream) { |
- NOTREACHED(); |
- return E_INVALIDARG; |
- } |
- |
- HRESULT hr = S_OK; |
- while (SUCCEEDED(hr)) { |
- const size_t kChunkSize = 4 * 1024; |
- char buffer[kChunkSize]; |
- DWORD read = 0; |
- hr = stream->Read(buffer, sizeof(buffer), &read); |
- if (read) { |
- DWORD written = 0; |
- cache_->Write(buffer, read, &written); |
- size_ += written; |
- } |
- |
- if ((S_FALSE == hr) || !read) |
- break; |
- } |
- |
- bool last_chance = force_determination || (size() >= kMaxSniffSize); |
- eof_ = force_determination; |
- DetermineRendererType(last_chance); |
- return hr; |
-} |
- |
-HRESULT SniffData::DrainCache(IBindStatusCallback* bscb, DWORD bscf, |
- CLIPFORMAT clip_format) { |
- if (!is_cache_valid()) { |
- return S_OK; |
- } |
- |
- // Ideally we could just use the cache_ IStream implementation but |
- // can't use it here since we have to return E_PENDING for the |
- // last call |
- HGLOBAL memory = NULL; |
- HRESULT hr = GetHGlobalFromStream(cache_, &memory); |
- if (SUCCEEDED(hr) && memory) { |
- char* buffer = reinterpret_cast<char*>(GlobalLock(memory)); |
- hr = CacheStream::BSCBFeedData(bscb, buffer, size_, clip_format, bscf, |
- eof_); |
- GlobalUnlock(memory); |
- } |
- |
- size_ = 0; |
- cache_.Release(); |
- return hr; |
-} |
- |
-// Scan the buffer or OptIn URL list and decide if the renderer is |
-// to be switched. Last chance means there's no more data. |
-void SniffData::DetermineRendererType(bool last_chance) { |
- if (is_undetermined()) { |
- if (last_chance) |
- renderer_type_ = OTHER; |
- if (IsChrome(RendererTypeForUrl(url_))) { |
- renderer_type_ = CHROME; |
- } else { |
- if (is_cache_valid() && cache_) { |
- HGLOBAL memory = NULL; |
- GetHGlobalFromStream(cache_, &memory); |
- const char* buffer = reinterpret_cast<const char*>(GlobalLock(memory)); |
- |
- std::wstring html_contents; |
- // TODO(joshia): detect and handle different content encodings |
- if (buffer && size_) { |
- base::UTF8ToWide(buffer, std::min(size_, kMaxSniffSize), |
- &html_contents); |
- GlobalUnlock(memory); |
- } |
- |
- // Note that document_contents_ may have NULL characters in it. While |
- // browsers may handle this properly, we don't and will stop scanning |
- // for the XUACompat content value if we encounter one. |
- std::wstring xua_compat_content; |
- UtilGetXUACompatContentValue(html_contents, &xua_compat_content); |
- if (StrStrI(xua_compat_content.c_str(), kChromeContentPrefix)) { |
- renderer_type_ = CHROME; |
- } |
- } |
- } |
- DVLOG(1) << __FUNCTION__ << "Url: " << url_ << base::StringPrintf( |
- "Renderer type: %s", renderer_type_ == CHROME ? "CHROME" : "OTHER"); |
- } |
-} |
- |
-///////////////////////////////////////////////////////////////////// |
- |
-BSCBStorageBind::BSCBStorageBind() : clip_format_(CF_NULL) { |
-} |
- |
-BSCBStorageBind::~BSCBStorageBind() { |
- std::for_each(saved_progress_.begin(), saved_progress_.end(), |
- utils::DeleteObject()); |
-} |
- |
-HRESULT BSCBStorageBind::Initialize(IMoniker* moniker, IBindCtx* bind_ctx) { |
- DVLOG(1) << __FUNCTION__ << me() |
- << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); |
- |
- std::wstring url = GetActualUrlFromMoniker(moniker, bind_ctx, |
- std::wstring()); |
- HRESULT hr = data_sniffer_.InitializeCache(url); |
- if (FAILED(hr)) |
- return hr; |
- |
- hr = AttachToBind(bind_ctx); |
- if (FAILED(hr)) { |
- NOTREACHED() << __FUNCTION__ << me() << "AttachToBind error: " << hr; |
- return hr; |
- } |
- |
- if (!delegate()) { |
- NOTREACHED() << __FUNCTION__ << me() << "No existing callback: " << hr; |
- return E_FAIL; |
- } |
- |
- return hr; |
-} |
- |
-STDMETHODIMP BSCBStorageBind::OnProgress(ULONG progress, ULONG progress_max, |
- ULONG status_code, LPCWSTR status_text) { |
- DVLOG(1) << __FUNCTION__ << me() |
- << base::StringPrintf(" status=%i tid=%i %ls", status_code, |
- base::PlatformThread::CurrentId(), |
- status_text); |
- // Report all crashes in the exception handler if we wrap the callback. |
- // Note that this avoids having the VEH report a crash if an SEH earlier in |
- // the chain handles the exception. |
- ExceptionBarrier barrier; |
- |
- HRESULT hr = S_OK; |
- |
- // TODO(ananta) |
- // ChromeFrame will not be informed of any redirects which occur while we |
- // switch into Chrome. This will only break the moniker patch which is |
- // legacy and needs to be deleted. |
- |
- if (ShouldCacheProgress(status_code)) { |
- saved_progress_.push_back(new Progress(progress, progress_max, status_code, |
- status_text)); |
- } else { |
- hr = CallbackImpl::OnProgress(progress, progress_max, status_code, |
- status_text); |
- } |
- |
- return hr; |
-} |
- |
-// Refer to urlmon_moniker.h for explanation of how things work. |
-STDMETHODIMP BSCBStorageBind::OnDataAvailable(DWORD flags, DWORD size, |
- FORMATETC* format_etc, |
- STGMEDIUM* stgmed) { |
- DVLOG(1) << __FUNCTION__ |
- << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); |
- // Report all crashes in the exception handler if we wrap the callback. |
- // Note that this avoids having the VEH report a crash if an SEH earlier in |
- // the chain handles the exception. |
- ExceptionBarrier barrier; |
- // Do not touch anything other than text/html. |
- bool is_interesting = (format_etc && stgmed && stgmed->pstm && |
- stgmed->tymed == TYMED_ISTREAM && |
- IsTextHtmlClipFormat(format_etc->cfFormat)); |
- |
- if (!is_interesting) { |
- // Play back report progress so far. |
- MayPlayBack(flags); |
- return CallbackImpl::OnDataAvailable(flags, size, format_etc, stgmed); |
- } |
- |
- HRESULT hr = S_OK; |
- if (!clip_format_) |
- clip_format_ = format_etc->cfFormat; |
- |
- if (data_sniffer_.is_undetermined()) { |
- bool force_determination = !!(flags & |
- (BSCF_LASTDATANOTIFICATION | BSCF_DATAFULLYAVAILABLE)); |
- hr = data_sniffer_.ReadIntoCache(stgmed->pstm, force_determination); |
- // If we don't have sufficient data to determine renderer type |
- // wait for the next data notification. |
- if (data_sniffer_.is_undetermined()) |
- return S_OK; |
- } |
- |
- DCHECK(!data_sniffer_.is_undetermined()); |
- |
- if (data_sniffer_.is_cache_valid()) { |
- hr = MayPlayBack(flags); |
- DCHECK(!data_sniffer_.is_cache_valid()); |
- } else { |
- hr = CallbackImpl::OnDataAvailable(flags, size, format_etc, stgmed); |
- } |
- return hr; |
-} |
- |
-STDMETHODIMP BSCBStorageBind::OnStopBinding(HRESULT hresult, LPCWSTR error) { |
- DVLOG(1) << __FUNCTION__ |
- << base::StringPrintf(" tid=%i", base::PlatformThread::CurrentId()); |
- // Report all crashes in the exception handler if we wrap the callback. |
- // Note that this avoids having the VEH report a crash if an SEH earlier in |
- // the chain handles the exception. |
- ExceptionBarrier barrier; |
- |
- HRESULT hr = MayPlayBack(BSCF_LASTDATANOTIFICATION); |
- if (FAILED(hr)) |
- return hr; |
- hr = CallbackImpl::OnStopBinding(hresult, error); |
- ReleaseBind(); |
- return hr; |
-} |
- |
-// Play back the cached data to the delegate. Normally this would happen |
-// when we have read enough data to determine the renderer. In this case |
-// we first play back the data from the cache and then go into a 'pass |
-// through' mode. In some cases we may end up getting OnStopBinding |
-// before we get a chance to determine. Also it's possible that the |
-// BindToStorage call will return before OnStopBinding is sent. Hence |
-// This is called from 3 places and it's important to maintain the |
-// exact sequence of calls. |
-// Once the data is played back, calling this again is a no op. |
-HRESULT BSCBStorageBind::MayPlayBack(DWORD flags) { |
- // Force renderer type determination if not already done since |
- // we want to play back data now. |
- data_sniffer_.DetermineRendererType(true); |
- DCHECK(!data_sniffer_.is_undetermined()); |
- |
- HRESULT hr = S_OK; |
- if (data_sniffer_.is_chrome()) { |
- // Remember clip format. If we are switching to chrome, then in order |
- // to make mshtml return INET_E_TERMINATED_BIND and reissue navigation |
- // with the same bind context, we have to return a mime type that is |
- // special cased by mshtml. |
- static const CLIPFORMAT kMagicClipFormat = |
- RegisterClipboardFormat(CFSTR_MIME_MPEG); |
- clip_format_ = kMagicClipFormat; |
- } else { |
- if (!saved_progress_.empty()) { |
- for (ProgressVector::iterator i = saved_progress_.begin(); |
- i != saved_progress_.end(); i++) { |
- Progress* p = (*i); |
- // We don't really expect a race condition here but just for sake |
- // of completeness we check. |
- if (p) { |
- (*i) = NULL; |
- CallbackImpl::OnProgress(p->progress(), p->progress_max(), |
- p->status_code(), p->status_text()); |
- delete p; |
- } |
- } |
- saved_progress_.clear(); |
- } |
- } |
- |
- if (data_sniffer_.is_cache_valid()) { |
- if (data_sniffer_.is_chrome()) { |
- base::win::ScopedComPtr<BindContextInfo> info; |
- BindContextInfo::FromBindContext(bind_ctx_, info.Receive()); |
- DCHECK(info); |
- if (info) { |
- info->SetToSwitch(data_sniffer_.cache_); |
- } |
- } |
- |
- hr = data_sniffer_.DrainCache(delegate(), |
- flags | BSCF_FIRSTDATANOTIFICATION, clip_format_); |
- DLOG_IF(WARNING, INET_E_TERMINATED_BIND != hr) << __FUNCTION__ << |
- " mshtml OnDataAvailable returned: " << std::hex << hr; |
- } |
- |
- return hr; |
-} |
- |
-// We cache and suppress sending progress notifications till |
-// we get the first OnDataAvailable. This is to prevent |
-// mshtml from making up its mind about the mime type. |
-// However, this is the invasive part of the patch and |
-// could trip other software that's due to mistimed progress |
-// notifications. It is probably not a good idea to hide redirects |
-// and some cookie notifications. |
-// |
-// We only need to suppress data notifications like |
-// BINDSTATUS_MIMETYPEAVAILABLE, |
-// BINDSTATUS_CACHEFILENAMEAVAILABLE etc. |
-// |
-// This is an atempt to reduce the exposure by starting to |
-// cache only when we receive one of the interesting progress |
-// notification. |
-bool BSCBStorageBind::ShouldCacheProgress(unsigned long status_code) const { |
- // We need to cache progress notifications only if we haven't yet figured |
- // out which way the request is going. |
- if (data_sniffer_.is_undetermined()) { |
- // If we are already caching then continue. |
- if (!saved_progress_.empty()) |
- return true; |
- // Start caching only if we see one of the interesting progress |
- // notifications. |
- switch (status_code) { |
- case BINDSTATUS_BEGINDOWNLOADDATA: |
- case BINDSTATUS_DOWNLOADINGDATA: |
- case BINDSTATUS_USINGCACHEDCOPY: |
- case BINDSTATUS_MIMETYPEAVAILABLE: |
- case BINDSTATUS_CACHEFILENAMEAVAILABLE: |
- case BINDSTATUS_SERVER_MIMETYPEAVAILABLE: |
- return true; |
- default: |
- break; |
- } |
- } |
- |
- return false; |
-} |