Don't assume that the mime type for documents that mshtml rejects is CF

chrome_frame/urlmon_bind_status_callback.cc

 // Copyright (c) 2010 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 "base/logging.h"
 #include "base/string_util.h"
 
 CFUrlmonBindStatusCallback::CFUrlmonBindStatusCallback()
@@ skipping 142 matching lines @@
 HRESULT CFUrlmonBindStatusCallback::GetBindInfo(DWORD* bindf,
                                                 BINDINFO* bind_info) {
   DLOG(INFO) << __FUNCTION__ << me() << StringPrintf(" tid=%i",
       PlatformThread::CurrentId());
   return delegate_->GetBindInfo(bindf, bind_info);
 }
 
 HRESULT CFUrlmonBindStatusCallback::OnDataAvailable(DWORD bscf, DWORD size,
                                                     FORMATETC* format_etc,
                                                     STGMEDIUM* stgmed) {
+  DCHECK(format_etc);
 #ifndef NDEBUG
   wchar_t clip_fmt_name[MAX_PATH] = {0};
   if (format_etc) {
     ::GetClipboardFormatNameW(format_etc->cfFormat, clip_fmt_name,
                               arraysize(clip_fmt_name));
   }
   DLOG(INFO) << __FUNCTION__ << me()
       << StringPrintf(" tid=%i original fmt=%ls",
                       PlatformThread::CurrentId(), clip_fmt_name);
 
   DCHECK(stgmed);
   DCHECK(stgmed->tymed == TYMED_ISTREAM);
 
   if (bscf & BSCF_FIRSTDATANOTIFICATION) {
     DLOG(INFO) << "first data notification";
   }
 #endif
 
   HRESULT hr = S_OK;
   size_t bytes_read = 0;
   if (!only_buffer_) {
     hr = data_->DelegateDataRead(delegate_, bscf, size, format_etc, stgmed,
                                  &bytes_read);
   }
 
   DLOG(INFO) << __FUNCTION__ << StringPrintf(" - 0x%08x", hr);
   if (hr == INET_E_TERMINATED_BIND) {
-    // We want to complete fetching the entire document even though the
-    // delegate isn't interested in continuing.
-    // This happens when we switch from mshtml to CF.
-    // We take over and buffer the document and once we're done, we report
-    // INET_E_TERMINATED to mshtml so that it will continue as usual.
-    hr = S_OK;
-    only_buffer_ = true;
-    binding_delegate_->OverrideBindResults(INET_E_TERMINATED_BIND);
+    // Check if the content type is CF's mime type.
+    wchar_t format_name[MAX_PATH] = {0};
+    ::GetClipboardFormatNameW(format_etc->cfFormat, format_name,

[amit, 2010/03/26 00:13:23]

nit: would it be better to compare the return value of RegisterClipboardFormat?

+                              arraysize(format_name));
+    DLOG(INFO) << StringPrintf("INET_E_TERMINATED_BIND for cf=%ls",
+        format_name);
+    if (lstrcmpiW(format_name, kChromeMimeType) == 0) {
+      // We want to complete fetching the entire document even though the
+      // delegate isn't interested in continuing.
+      // This happens when we switch from mshtml to CF.
+      // We take over and buffer the document and once we're done, we report
+      // INET_E_TERMINATED to mshtml so that it will continue as usual.
+      hr = S_OK;
+      only_buffer_ = true;
+      binding_delegate_->OverrideBindResults(INET_E_TERMINATED_BIND);
+    }
   }
 
   if (only_buffer_) {
     data_->CacheAll(stgmed->pstm);
     DCHECK(hr == S_OK);
   }
 
   return hr;
 }
 
@@ skipping 69 matching lines @@
                                            reserved);
 }
 
 HRESULT CFUrlmonBindStatusCallback::GetSerializedClientCertContext(
     BYTE** cert,
     DWORD* cert_size) {
   ScopedComPtr<IHttpNegotiate3> http_negotiate;
   http_negotiate.QueryFrom(delegate_);
   return http_negotiate->GetSerializedClientCertContext(cert, cert_size);
 }